1 files changed, 130 insertions, 0 deletions
diff --git a/dom/cache/CacheWorkerRef.cpp b/dom/cache/CacheWorkerRef.cpp
new file mode 100644
index 0000000000..7679da8b05
--- /dev/null
+++ b/dom/cache/CacheWorkerRef.cpp
@@ -0,0 +1,130 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "mozilla/dom/cache/CacheWorkerRef.h"
+
+#include "mozilla/dom/cache/ActorChild.h"
+#include "mozilla/dom/WorkerPrivate.h"
+#include "mozilla/dom/WorkerRef.h"
+
+namespace mozilla::dom::cache {
+
+// static
+SafeRefPtr<CacheWorkerRef> CacheWorkerRef::Create(WorkerPrivate* aWorkerPrivate,
+                                                  Behavior aBehavior) {
+  MOZ_DIAGNOSTIC_ASSERT(aWorkerPrivate);
+
+  // XXX This looks as if this could be simplified now by moving into the ctor
+  // of CacheWorkerRef, since we can now use SafeRefPtrFromThis in the ctor
+  auto workerRef =
+      MakeSafeRefPtr<CacheWorkerRef>(aBehavior, ConstructorGuard{});
+  auto notify = [workerRef = workerRef.clonePtr()] { workerRef->Notify(); };
+  if (aBehavior == eStrongWorkerRef) {
+    workerRef->mStrongWorkerRef = StrongWorkerRef::Create(
+        aWorkerPrivate, "CacheWorkerRef-Strong", std::move(notify));
+  } else {
+    MOZ_ASSERT(aBehavior == eIPCWorkerRef);
+    workerRef->mIPCWorkerRef = IPCWorkerRef::Create(
+        aWorkerPrivate, "CacheWorkerRef-IPC", std::move(notify));
+  }
+
+  if (NS_WARN_IF(!workerRef->mIPCWorkerRef && !workerRef->mStrongWorkerRef)) {
+    return nullptr;
+  }
+
+  return workerRef;
+}
+
+// static
+SafeRefPtr<CacheWorkerRef> CacheWorkerRef::PreferBehavior(
+    SafeRefPtr<CacheWorkerRef> aCurrentRef, Behavior aBehavior) {
+  if (!aCurrentRef) {
+    return nullptr;
+  }
+
+  SafeRefPtr<CacheWorkerRef> orig = std::move(aCurrentRef);
+  if (orig->mBehavior == aBehavior) {
+    return orig;
+  }
+
+  WorkerPrivate* workerPrivate = nullptr;
+  if (orig->mBehavior == eStrongWorkerRef) {
+    workerPrivate = orig->mStrongWorkerRef->Private();
+  } else {
+    MOZ_ASSERT(orig->mBehavior == eIPCWorkerRef);
+    workerPrivate = orig->mIPCWorkerRef->Private();
+  }
+
+  MOZ_ASSERT(workerPrivate);
+
+  SafeRefPtr<CacheWorkerRef> replace = Create(workerPrivate, aBehavior);
+  return static_cast<bool>(replace) ? std::move(replace) : std::move(orig);
+}
+
+void CacheWorkerRef::AddActor(ActorChild& aActor) {
+  NS_ASSERT_OWNINGTHREAD(CacheWorkerRef);
+  MOZ_ASSERT(!mActorList.Contains(&aActor));
+
+  mActorList.AppendElement(WrapNotNullUnchecked(&aActor));
+  if (mBehavior == eIPCWorkerRef) {
+    MOZ_ASSERT(mIPCWorkerRef);
+    mIPCWorkerRef->SetActorCount(mActorList.Length());
+  }
+
+  // Allow an actor to be added after we've entered the Notifying case.  We
+  // can't stop the actor creation from racing with out destruction of the
+  // other actors and we need to wait for this extra one to close as well.
+  // Signal it should destroy itself right away.
+  if (mNotified) {
+    aActor.StartDestroy();
+  }
+}
+
+void CacheWorkerRef::RemoveActor(ActorChild& aActor) {
+  NS_ASSERT_OWNINGTHREAD(CacheWorkerRef);
+
+#if defined(RELEASE_OR_BETA)
+  mActorList.RemoveElement(&aActor);
+#else
+  MOZ_DIAGNOSTIC_ASSERT(mActorList.RemoveElement(&aActor));
+#endif
+
+  MOZ_ASSERT(!mActorList.Contains(&aActor));
+
+  if (mBehavior == eIPCWorkerRef) {
+    MOZ_ASSERT(mIPCWorkerRef);
+    mIPCWorkerRef->SetActorCount(mActorList.Length());
+  }
+
+  if (mActorList.IsEmpty()) {
+    mStrongWorkerRef = nullptr;
+    mIPCWorkerRef = nullptr;
+  }
+}
+
+bool CacheWorkerRef::Notified() const { return mNotified; }
+
+void CacheWorkerRef::Notify() {
+  NS_ASSERT_OWNINGTHREAD(CacheWorkerRef);
+
+  mNotified = true;
+
+  // Start the asynchronous destruction of our actors.  These will call back
+  // into RemoveActor() once the actor is destroyed.
+  for (const auto& actor : mActorList) {
+    actor->StartDestroy();
+  }
+}
+
+CacheWorkerRef::CacheWorkerRef(Behavior aBehavior, ConstructorGuard)
+    : mBehavior(aBehavior), mNotified(false) {}
+
+CacheWorkerRef::~CacheWorkerRef() {
+  NS_ASSERT_OWNINGTHREAD(CacheWorkerRef);
+  MOZ_DIAGNOSTIC_ASSERT(mActorList.IsEmpty());
+}
+
+}  // namespace mozilla::dom::cache