1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
/* -*- 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 "LockManagerChild.h"
#include "LockRequestChild.h"
#include "mozilla/dom/Document.h"
#include "mozilla/dom/RemoteWorkerChild.h"
#include "mozilla/dom/WindowGlobalChild.h"
#include "mozilla/dom/WorkerCommon.h"
#include "mozilla/dom/WorkerPrivate.h"
#include "mozilla/dom/WorkerRunnable.h"
namespace mozilla::dom::locks {
LockManagerChild::LockManagerChild(nsIGlobalObject* aOwner) : mOwner(aOwner) {
if (!NS_IsMainThread()) {
mWorkerRef = IPCWorkerRef::Create(GetCurrentThreadWorkerPrivate(),
"LockManagerChild");
}
}
void LockManagerChild::NotifyBFCacheOnMainThread(nsPIDOMWindowInner* aInner,
bool aCreated) {
AssertIsOnMainThread();
if (!aInner) {
return;
}
if (aCreated) {
aInner->RemoveFromBFCacheSync();
}
uint32_t count = aInner->UpdateLockCount(aCreated);
// It's okay for WindowGlobalChild to not exist, as it should mean it already
// is destroyed and can't enter bfcache anyway.
if (WindowGlobalChild* child = aInner->GetWindowGlobalChild()) {
if (aCreated && count == 1) {
// The first lock is active.
child->BlockBFCacheFor(BFCacheStatus::ACTIVE_LOCK);
} else if (count == 0) {
child->UnblockBFCacheFor(BFCacheStatus::ACTIVE_LOCK);
}
}
}
class BFCacheNotifyLockRunnable final : public WorkerProxyToMainThreadRunnable {
public:
explicit BFCacheNotifyLockRunnable(bool aCreated) : mCreated(aCreated) {}
void RunOnMainThread(WorkerPrivate* aWorkerPrivate) override {
MOZ_ASSERT(aWorkerPrivate);
AssertIsOnMainThread();
if (aWorkerPrivate->IsDedicatedWorker()) {
LockManagerChild::NotifyBFCacheOnMainThread(
aWorkerPrivate->GetAncestorWindow(), mCreated);
return;
}
if (aWorkerPrivate->IsSharedWorker()) {
aWorkerPrivate->GetRemoteWorkerController()->NotifyLock(mCreated);
return;
}
MOZ_ASSERT_UNREACHABLE("Unexpected worker type");
}
void RunBackOnWorkerThreadForCleanup(WorkerPrivate* aWorkerPrivate) override {
MOZ_ASSERT(aWorkerPrivate);
aWorkerPrivate->AssertIsOnWorkerThread();
}
private:
bool mCreated;
};
NS_IMPL_CYCLE_COLLECTION(LockManagerChild, mOwner)
void LockManagerChild::RequestLock(const LockRequest& aRequest,
const LockOptions& aOptions) {
auto requestActor = MakeRefPtr<LockRequestChild>(aRequest, aOptions.mSignal);
requestActor->MaybeSetWorkerRef();
SendPLockRequestConstructor(
requestActor, IPCLockRequest(nsString(aRequest.mName), aOptions.mMode,
aOptions.mIfAvailable, aOptions.mSteal));
NotifyToWindow(true);
}
void LockManagerChild::NotifyRequestDestroy() const { NotifyToWindow(false); }
void LockManagerChild::NotifyToWindow(bool aCreated) const {
if (NS_IsMainThread()) {
NotifyBFCacheOnMainThread(GetParentObject()->AsInnerWindow(), aCreated);
return;
}
WorkerPrivate* wp = GetCurrentThreadWorkerPrivate();
if (wp->IsDedicatedWorker() || wp->IsSharedWorker()) {
RefPtr<BFCacheNotifyLockRunnable> runnable =
new BFCacheNotifyLockRunnable(aCreated);
runnable->Dispatch(wp);
}
};
} // namespace mozilla::dom::locks
|
