dom/serviceworkers/ServiceWorkerProxy.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "ServiceWorkerProxy.h"
   8 #include "ServiceWorkerCloneData.h"
   9 #include "ServiceWorkerManager.h"
  10 #include "ServiceWorkerParent.h"
  11
  12 #include "mozilla/SchedulerGroup.h"
  13 #include "mozilla/ScopeExit.h"
  14 #include "mozilla/dom/ClientState.h"
  15 #include "mozilla/ipc/BackgroundParent.h"
  16 #include "ServiceWorkerInfo.h"
  17
  18 namespace mozilla::dom {
  19
  20 using mozilla::ipc::AssertIsOnBackgroundThread;
  21
  22 ServiceWorkerProxy::~ServiceWorkerProxy() {
  23   // Any thread
  24   MOZ_DIAGNOSTIC_ASSERT(!mActor);
  25   MOZ_DIAGNOSTIC_ASSERT(!mInfo);
  26 }
  27
  28 void ServiceWorkerProxy::MaybeShutdownOnBGThread() {
  29   AssertIsOnBackgroundThread();
  30   if (!mActor) {
  31     return;
  32   }
  33   mActor->MaybeSendDelete();
  34 }
  35
  36 void ServiceWorkerProxy::InitOnMainThread() {
  37   AssertIsOnMainThread();
  38
  39   auto scopeExit = MakeScopeExit([&] { MaybeShutdownOnMainThread(); });
  40
  41   RefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
  42   NS_ENSURE_TRUE_VOID(swm);
  43
  44   RefPtr<ServiceWorkerRegistrationInfo> reg =
  45       swm->GetRegistration(mDescriptor.PrincipalInfo(), mDescriptor.Scope());
  46   NS_ENSURE_TRUE_VOID(reg);
  47
  48   RefPtr<ServiceWorkerInfo> info = reg->GetByDescriptor(mDescriptor);
  49   NS_ENSURE_TRUE_VOID(info);
  50
  51   scopeExit.release();
  52
  53   mInfo = new nsMainThreadPtrHolder<ServiceWorkerInfo>(
  54       "ServiceWorkerProxy::mInfo", info);
  55 }
  56
  57 void ServiceWorkerProxy::MaybeShutdownOnMainThread() {
  58   AssertIsOnMainThread();
  59
  60   nsCOMPtr<nsIRunnable> r = NewRunnableMethod(
  61       __func__, this, &ServiceWorkerProxy::MaybeShutdownOnBGThread);
  62
  63   MOZ_ALWAYS_SUCCEEDS(mEventTarget->Dispatch(r.forget(), NS_DISPATCH_NORMAL));
  64 }
  65
  66 void ServiceWorkerProxy::StopListeningOnMainThread() {
  67   AssertIsOnMainThread();
  68   mInfo = nullptr;
  69 }
  70
  71 ServiceWorkerProxy::ServiceWorkerProxy(
  72     const ServiceWorkerDescriptor& aDescriptor)
  73     : mEventTarget(GetCurrentSerialEventTarget()), mDescriptor(aDescriptor) {}
  74
  75 void ServiceWorkerProxy::Init(ServiceWorkerParent* aActor) {
  76   AssertIsOnBackgroundThread();
  77   MOZ_DIAGNOSTIC_ASSERT(aActor);
  78   MOZ_DIAGNOSTIC_ASSERT(!mActor);
  79   MOZ_DIAGNOSTIC_ASSERT(mEventTarget);
  80
  81   mActor = aActor;
  82
  83   // Note, this must be done from a separate Init() method and not in
  84   // the constructor.  If done from the constructor the runnable can
  85   // execute, complete, and release its reference before the constructor
  86   // returns.
  87   nsCOMPtr<nsIRunnable> r = NewRunnableMethod(
  88       "ServiceWorkerProxy::Init", this, &ServiceWorkerProxy::InitOnMainThread);
  89   MOZ_ALWAYS_SUCCEEDS(
  90       SchedulerGroup::Dispatch(TaskCategory::Other, r.forget()));
  91 }
  92
  93 void ServiceWorkerProxy::RevokeActor(ServiceWorkerParent* aActor) {
  94   AssertIsOnBackgroundThread();
  95   MOZ_DIAGNOSTIC_ASSERT(mActor);
  96   MOZ_DIAGNOSTIC_ASSERT(mActor == aActor);
  97   mActor = nullptr;
  98
  99   nsCOMPtr<nsIRunnable> r = NewRunnableMethod(
 100       __func__, this, &ServiceWorkerProxy::StopListeningOnMainThread);
 101   MOZ_ALWAYS_SUCCEEDS(
 102       SchedulerGroup::Dispatch(TaskCategory::Other, r.forget()));
 103 }
 104
 105 void ServiceWorkerProxy::PostMessage(RefPtr<ServiceWorkerCloneData>&& aData,
 106                                      const ClientInfo& aClientInfo,
 107                                      const ClientState& aClientState) {
 108   AssertIsOnBackgroundThread();
 109   RefPtr<ServiceWorkerProxy> self = this;
 110   nsCOMPtr<nsIRunnable> r = NS_NewRunnableFunction(
 111       __func__,
 112       [self, data = std::move(aData), aClientInfo, aClientState]() mutable {
 113         if (!self->mInfo) {
 114           return;
 115         }
 116         self->mInfo->PostMessage(std::move(data), aClientInfo, aClientState);
 117       });
 118   MOZ_ALWAYS_SUCCEEDS(
 119       SchedulerGroup::Dispatch(TaskCategory::Other, r.forget()));
 120 }
 121
 122 }  // namespace mozilla::dom