From 26a029d407be480d791972afb5975cf62c9360a6 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 19 Apr 2024 02:47:55 +0200 Subject: Adding upstream version 124.0.1. Signed-off-by: Daniel Baumann --- dom/cache/CacheWorkerRef.cpp | 130 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 130 insertions(+) create mode 100644 dom/cache/CacheWorkerRef.cpp (limited to 'dom/cache/CacheWorkerRef.cpp') 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::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(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::PreferBehavior( + SafeRefPtr aCurrentRef, Behavior aBehavior) { + if (!aCurrentRef) { + return nullptr; + } + + SafeRefPtr 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 replace = Create(workerPrivate, aBehavior); + return static_cast(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 -- cgit v1.2.3