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
107
108
109
110
111
112
113
114
115
|
/* -*- 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 "ClientHandleParent.h"
#include "ClientHandleOpParent.h"
#include "ClientManagerService.h"
#include "ClientPrincipalUtils.h"
#include "ClientSourceParent.h"
#include "mozilla/dom/ClientIPCTypes.h"
#include "mozilla/Unused.h"
namespace mozilla::dom {
using mozilla::ipc::IPCResult;
IPCResult ClientHandleParent::RecvTeardown() {
Unused << Send__delete__(this);
return IPC_OK();
}
void ClientHandleParent::ActorDestroy(ActorDestroyReason aReason) {
if (mSource) {
mSource->DetachHandle(this);
mSource = nullptr;
} else {
if (!mSourcePromiseHolder.IsEmpty()) {
CopyableErrorResult rv;
rv.ThrowAbortError("Client aborted");
mSourcePromiseHolder.Reject(rv, __func__);
}
mSourcePromiseRequestHolder.DisconnectIfExists();
}
}
PClientHandleOpParent* ClientHandleParent::AllocPClientHandleOpParent(
const ClientOpConstructorArgs& aArgs) {
return new ClientHandleOpParent();
}
bool ClientHandleParent::DeallocPClientHandleOpParent(
PClientHandleOpParent* aActor) {
delete aActor;
return true;
}
IPCResult ClientHandleParent::RecvPClientHandleOpConstructor(
PClientHandleOpParent* aActor, const ClientOpConstructorArgs& aArgs) {
auto actor = static_cast<ClientHandleOpParent*>(aActor);
actor->Init(std::move(const_cast<ClientOpConstructorArgs&>(aArgs)));
return IPC_OK();
}
ClientHandleParent::ClientHandleParent()
: mService(ClientManagerService::GetOrCreateInstance()), mSource(nullptr) {}
ClientHandleParent::~ClientHandleParent() { MOZ_DIAGNOSTIC_ASSERT(!mSource); }
void ClientHandleParent::Init(const IPCClientInfo& aClientInfo) {
mClientId = aClientInfo.id();
mPrincipalInfo = aClientInfo.principalInfo();
// Callbacks are disconnected in ActorDestroy, so capturing `this` is safe.
mService->FindSource(aClientInfo.id(), aClientInfo.principalInfo())
->Then(
GetCurrentSerialEventTarget(), __func__,
[self = RefPtr{this}](bool) {
self->mSourcePromiseRequestHolder.Complete();
ClientSourceParent* source = self->mService->FindExistingSource(
self->mClientId, self->mPrincipalInfo);
if (source) {
self->FoundSource(source);
}
},
[self = RefPtr{this}](const CopyableErrorResult&) {
self->mSourcePromiseRequestHolder.Complete();
Unused << Send__delete__(self);
})
->Track(mSourcePromiseRequestHolder);
}
ClientSourceParent* ClientHandleParent::GetSource() const { return mSource; }
RefPtr<SourcePromise> ClientHandleParent::EnsureSource() {
if (mSource) {
return SourcePromise::CreateAndResolve(mSource, __func__);
}
return mSourcePromiseHolder.Ensure(__func__);
}
void ClientHandleParent::FoundSource(ClientSourceParent* aSource) {
MOZ_ASSERT(aSource);
MOZ_ASSERT(aSource->Info().Id() == mClientId);
if (!ClientMatchPrincipalInfo(aSource->Info().PrincipalInfo(),
mPrincipalInfo)) {
if (mSourcePromiseHolder.IsEmpty()) {
CopyableErrorResult rv;
rv.ThrowAbortError("Client aborted");
mSourcePromiseHolder.Reject(rv, __func__);
}
Unused << Send__delete__(this);
return;
}
mSource = aSource;
mSource->AttachHandle(this);
mSourcePromiseHolder.ResolveIfExists(true, __func__);
}
} // namespace mozilla::dom
|
