Source/WebCore/workers/service/server/SWServer.h

   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 #pragma once
  27
  28 #if ENABLE(SERVICE_WORKER)
  29
  30 #include "ServiceWorkerJob.h"
  31 #include "ServiceWorkerRegistrationKey.h"
  32 #include <wtf/CrossThreadQueue.h>
  33 #include <wtf/CrossThreadTask.h>
  34 #include <wtf/HashMap.h>
  35 #include <wtf/HashSet.h>
  36 #include <wtf/Identified.h>
  37 #include <wtf/RunLoop.h>
  38 #include <wtf/ThreadSafeRefCounted.h>
  39 #include <wtf/Threading.h>
  40
  41 namespace WebCore {
  42
  43 class SWServerRegistration;
  44 class SWServerWorker;
  45 struct ExceptionData;
  46 struct ServiceWorkerContextData;
  47 struct ServiceWorkerFetchResult;
  48 struct ServiceWorkerRegistrationData;
  49
  50 class SWServer {
  51 public:
  52     class Connection : public Identified<Connection> {
  53     friend class SWServer;
  54     public:
  55         WEBCORE_EXPORT virtual ~Connection();
  56
  57         WEBCORE_EXPORT void scriptContextFailedToStart(const ServiceWorkerRegistrationKey&, const String& workerID, const String& message);
  58         WEBCORE_EXPORT void scriptContextStarted(const ServiceWorkerRegistrationKey&, uint64_t identifier, const String& workerID);
  59
  60     protected:
  61         WEBCORE_EXPORT Connection(SWServer&, uint64_t identifier);
  62         SWServer& server() { return m_server; }
  63
  64         WEBCORE_EXPORT void scheduleJobInServer(const ServiceWorkerJobData&);
  65         WEBCORE_EXPORT void finishFetchingScriptInServer(const ServiceWorkerFetchResult&);
  66
  67     private:
  68         // Messages to the client WebProcess
  69         virtual void rejectJobInClient(uint64_t jobIdentifier, const ExceptionData&) = 0;
  70         virtual void resolveJobInClient(uint64_t jobIdentifier, const ServiceWorkerRegistrationData&) = 0;
  71         virtual void startScriptFetchInClient(uint64_t jobIdentifier) = 0;
  72
  73         // Messages to the SW host WebProcess
  74         virtual void startServiceWorkerContext(const ServiceWorkerContextData&) = 0;
  75
  76         SWServer& m_server;
  77     };
  78
  79     WEBCORE_EXPORT SWServer();
  80     WEBCORE_EXPORT ~SWServer();
  81
  82     void scheduleJob(const ServiceWorkerJobData&);
  83     void rejectJob(const ServiceWorkerJobData&, const ExceptionData&);
  84     void resolveJob(const ServiceWorkerJobData&, const ServiceWorkerRegistrationData&);
  85     void startScriptFetch(const ServiceWorkerJobData&);
  86
  87     void postTask(CrossThreadTask&&);
  88     void postTaskReply(CrossThreadTask&&);
  89
  90     Ref<SWServerWorker> createWorker(Connection&, const ServiceWorkerRegistrationKey&, const URL&, const String& script, WorkerType);
  91
  92 private:
  93     void registerConnection(Connection&);
  94     void unregisterConnection(Connection&);
  95
  96     void taskThreadEntryPoint();
  97     void handleTaskRepliesOnMainThread();
  98
  99     void scriptFetchFinished(Connection&, const ServiceWorkerFetchResult&);
 100     void scriptContextFailedToStart(Connection&, const ServiceWorkerRegistrationKey&, const String& workerID, const String& message);
 101     void scriptContextStarted(Connection&, const ServiceWorkerRegistrationKey&, uint64_t identifier, const String& workerID);
 102
 103     HashMap<uint64_t, Connection*> m_connections;
 104     HashMap<ServiceWorkerRegistrationKey, std::unique_ptr<SWServerRegistration>> m_registrations;
 105
 106     HashMap<String, Ref<SWServerWorker>> m_workersByID;
 107
 108     RefPtr<Thread> m_taskThread;
 109     Lock m_taskThreadLock;
 110
 111     CrossThreadQueue<CrossThreadTask> m_taskQueue;
 112     CrossThreadQueue<CrossThreadTask> m_taskReplyQueue;
 113
 114     Lock m_mainThreadReplyLock;
 115     bool m_mainThreadReplyScheduled { false };
 116 };
 117
 118 } // namespace WebCore
 119
 120 #endif // ENABLE(SERVICE_WORKER)