Source/WebKit/StorageProcess/ServiceWorker/WebSWOriginStore.cpp

   1 /*
   2  * Copyright (C) 2017 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  23  * THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27
  28 #if ENABLE(SERVICE_WORKER)
  29 #include "WebSWOriginStore.h"
  30
  31 #include "WebSWClientConnectionMessages.h"
  32 #include "WebSWServerConnection.h"
  33 #include <WebCore/SecurityOrigin.h>
  34
  35 namespace WebKit {
  36
  37 using namespace WebCore;
  38
  39 WebSWOriginStore::WebSWOriginStore()
  40     : m_store(*this)
  41 {
  42 }
  43
  44 void WebSWOriginStore::addToStore(const SecurityOrigin& origin)
  45 {
  46     m_store.scheduleAddition(computeSharedStringHash(origin.toString()));
  47     m_store.flushPendingChanges();
  48 }
  49
  50 void WebSWOriginStore::removeFromStore(const SecurityOrigin& origin)
  51 {
  52     m_store.scheduleRemoval(computeSharedStringHash(origin.toString()));
  53     m_store.flushPendingChanges();
  54 }
  55
  56 void WebSWOriginStore::clearStore()
  57 {
  58     m_store.clear();
  59 }
  60
  61 void WebSWOriginStore::registerSWServerConnection(WebSWServerConnection& connection)
  62 {
  63     m_webSWServerConnections.add(&connection);
  64
  65     if (m_store.isEmpty()) {
  66         connection.send(Messages::WebSWClientConnection::InitializeSWOriginTableAsEmpty());
  67         return;
  68     }
  69
  70     sendStoreHandle(connection);
  71 }
  72
  73 void WebSWOriginStore::unregisterSWServerConnection(WebSWServerConnection& connection)
  74 {
  75     m_webSWServerConnections.remove(&connection);
  76 }
  77
  78 void WebSWOriginStore::sendStoreHandle(WebSWServerConnection& connection)
  79 {
  80     SharedMemory::Handle handle;
  81     if (!m_store.createSharedMemoryHandle(handle))
  82         return;
  83
  84     connection.send(Messages::WebSWClientConnection::SetSWOriginTableSharedMemory(handle));
  85 }
  86
  87 void WebSWOriginStore::didInvalidateSharedMemory()
  88 {
  89     for (auto* connection : m_webSWServerConnections)
  90         sendStoreHandle(*connection);
  91 }
  92
  93 } // namespace WebKit
  94
  95 #endif // ENABLE(SERVICE_WORKER)