2019-07-12 14:17:59 +03:00
|
|
|
/* -*- 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 {
|
|
|
|
namespace dom {
|
|
|
|
namespace cache {
|
|
|
|
|
2020-05-11 15:12:22 +03:00
|
|
|
namespace {
|
|
|
|
// XXX Move this to mfbt, or do we already have something like this? Or remove
|
|
|
|
// the need for that by changing StrongWorkerRef/IPCWorkerRef?
|
|
|
|
|
|
|
|
template <class T>
|
|
|
|
class FakeCopyable {
|
|
|
|
public:
|
|
|
|
explicit FakeCopyable(T&& aTarget) : mTarget(std::forward<T>(aTarget)) {}
|
|
|
|
|
|
|
|
FakeCopyable(FakeCopyable&&) = default;
|
|
|
|
|
|
|
|
FakeCopyable(const FakeCopyable& aOther)
|
|
|
|
: mTarget(std::move(const_cast<FakeCopyable&>(aOther).mTarget)) {
|
|
|
|
MOZ_CRASH("Do not copy.");
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename... Args>
|
|
|
|
auto operator()(Args&&... aArgs) {
|
|
|
|
return mTarget(std::forward<Args>(aArgs)...);
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
T mTarget;
|
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace
|
|
|
|
|
2019-07-12 14:17:59 +03:00
|
|
|
// static
|
2020-05-11 15:12:22 +03:00
|
|
|
SafeRefPtr<CacheWorkerRef> CacheWorkerRef::Create(WorkerPrivate* aWorkerPrivate,
|
|
|
|
Behavior aBehavior) {
|
2019-07-12 14:17:59 +03:00
|
|
|
MOZ_DIAGNOSTIC_ASSERT(aWorkerPrivate);
|
|
|
|
|
2020-05-11 15:12:22 +03:00
|
|
|
// 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 =
|
|
|
|
FakeCopyable([workerRef = workerRef.clonePtr()] { workerRef->Notify(); });
|
2019-07-12 14:17:59 +03:00
|
|
|
if (aBehavior == eStrongWorkerRef) {
|
2020-05-11 15:12:22 +03:00
|
|
|
workerRef->mStrongWorkerRef = StrongWorkerRef::Create(
|
|
|
|
aWorkerPrivate, "CacheWorkerRef-Strong", std::move(notify));
|
2019-07-12 14:17:59 +03:00
|
|
|
} else {
|
|
|
|
MOZ_ASSERT(aBehavior == eIPCWorkerRef);
|
2020-05-11 15:12:22 +03:00
|
|
|
workerRef->mIPCWorkerRef = IPCWorkerRef::Create(
|
|
|
|
aWorkerPrivate, "CacheWorkerRef-IPC", std::move(notify));
|
2019-07-12 14:17:59 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if (NS_WARN_IF(!workerRef->mIPCWorkerRef && !workerRef->mStrongWorkerRef)) {
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
2020-05-11 15:12:22 +03:00
|
|
|
return workerRef;
|
2019-07-12 14:17:59 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// static
|
2020-05-11 15:12:22 +03:00
|
|
|
SafeRefPtr<CacheWorkerRef> CacheWorkerRef::PreferBehavior(
|
|
|
|
SafeRefPtr<CacheWorkerRef> aCurrentRef, Behavior aBehavior) {
|
2019-07-12 14:17:59 +03:00
|
|
|
if (!aCurrentRef) {
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
2020-05-11 15:12:22 +03:00
|
|
|
SafeRefPtr<CacheWorkerRef> orig = std::move(aCurrentRef);
|
2019-07-12 14:17:59 +03:00
|
|
|
if (orig->mBehavior == aBehavior) {
|
2020-05-11 15:12:22 +03:00
|
|
|
return orig;
|
2019-07-12 14:17:59 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
WorkerPrivate* workerPrivate = nullptr;
|
|
|
|
if (orig->mBehavior == eStrongWorkerRef) {
|
|
|
|
workerPrivate = orig->mStrongWorkerRef->Private();
|
|
|
|
} else {
|
|
|
|
MOZ_ASSERT(orig->mBehavior == eIPCWorkerRef);
|
|
|
|
workerPrivate = orig->mIPCWorkerRef->Private();
|
|
|
|
}
|
|
|
|
|
|
|
|
MOZ_ASSERT(workerPrivate);
|
|
|
|
|
2020-05-11 15:12:22 +03:00
|
|
|
SafeRefPtr<CacheWorkerRef> replace = Create(workerPrivate, aBehavior);
|
|
|
|
return static_cast<bool>(replace) ? std::move(replace) : std::move(orig);
|
2019-07-12 14:17:59 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
void CacheWorkerRef::AddActor(ActorChild* aActor) {
|
|
|
|
NS_ASSERT_OWNINGTHREAD(CacheWorkerRef);
|
|
|
|
MOZ_DIAGNOSTIC_ASSERT(aActor);
|
|
|
|
MOZ_ASSERT(!mActorList.Contains(aActor));
|
|
|
|
|
|
|
|
mActorList.AppendElement(aActor);
|
|
|
|
|
|
|
|
// 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);
|
|
|
|
MOZ_DIAGNOSTIC_ASSERT(aActor);
|
|
|
|
|
|
|
|
#if defined(RELEASE_OR_BETA)
|
|
|
|
mActorList.RemoveElement(aActor);
|
|
|
|
#else
|
|
|
|
MOZ_DIAGNOSTIC_ASSERT(mActorList.RemoveElement(aActor));
|
|
|
|
#endif
|
|
|
|
|
|
|
|
MOZ_ASSERT(!mActorList.Contains(aActor));
|
|
|
|
|
|
|
|
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 (uint32_t i = 0; i < mActorList.Length(); ++i) {
|
|
|
|
MOZ_DIAGNOSTIC_ASSERT(mActorList[i]);
|
|
|
|
mActorList[i]->StartDestroy();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-05-11 15:12:22 +03:00
|
|
|
CacheWorkerRef::CacheWorkerRef(Behavior aBehavior, ConstructorGuard)
|
2019-07-12 14:17:59 +03:00
|
|
|
: mBehavior(aBehavior), mNotified(false) {}
|
|
|
|
|
|
|
|
CacheWorkerRef::~CacheWorkerRef() {
|
|
|
|
NS_ASSERT_OWNINGTHREAD(CacheWorkerRef);
|
|
|
|
MOZ_DIAGNOSTIC_ASSERT(mActorList.IsEmpty());
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace cache
|
|
|
|
} // namespace dom
|
|
|
|
} // namespace mozilla
|