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
|
/* -*- 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 "ClientHandleOpParent.h"
#include "ClientHandleParent.h"
#include "ClientSourceParent.h"
#include "mozilla/dom/ipc/StructuredCloneData.h"
#include "mozilla/dom/PClientManagerParent.h"
namespace mozilla::dom {
ClientSourceParent* ClientHandleOpParent::GetSource() const {
auto handle = static_cast<ClientHandleParent*>(Manager());
return handle->GetSource();
}
void ClientHandleOpParent::ActorDestroy(ActorDestroyReason aReason) {
mPromiseRequestHolder.DisconnectIfExists();
mSourcePromiseRequestHolder.DisconnectIfExists();
}
void ClientHandleOpParent::Init(ClientOpConstructorArgs&& aArgs) {
RefPtr<ClientHandleParent> handle =
static_cast<ClientHandleParent*>(Manager());
handle->EnsureSource()
->Then(
GetCurrentSerialEventTarget(), __func__,
[this, handle, args = std::move(aArgs)](bool) mutable {
mSourcePromiseRequestHolder.Complete();
auto source = handle->GetSource();
if (!source) {
CopyableErrorResult rv;
rv.ThrowAbortError("Client has been destroyed");
Unused << PClientHandleOpParent::Send__delete__(this, rv);
return;
}
RefPtr<ClientOpPromise> p;
// ClientPostMessageArgs can contain PBlob actors. This means we
// can't just forward the args from one PBackground manager to
// another. Instead, unpack the structured clone data and repack
// it into a new set of arguments.
if (args.type() ==
ClientOpConstructorArgs::TClientPostMessageArgs) {
const ClientPostMessageArgs& orig =
args.get_ClientPostMessageArgs();
ClientPostMessageArgs rebuild;
rebuild.serviceWorker() = orig.serviceWorker();
ipc::StructuredCloneData data;
data.BorrowFromClonedMessageData(orig.clonedData());
if (!data.BuildClonedMessageData(rebuild.clonedData())) {
CopyableErrorResult rv;
rv.ThrowAbortError("Aborting client operation");
Unused << PClientHandleOpParent::Send__delete__(this, rv);
return;
}
p = source->StartOp(std::move(rebuild));
}
// Other argument types can just be forwarded straight through.
else {
p = source->StartOp(std::move(args));
}
// Capturing 'this' is safe here because we disconnect the promise
// in ActorDestroy() which ensures neither lambda is called if the
// actor is destroyed before the source operation completes.
p->Then(
GetCurrentSerialEventTarget(), __func__,
[this](const ClientOpResult& aResult) {
mPromiseRequestHolder.Complete();
Unused << PClientHandleOpParent::Send__delete__(this,
aResult);
},
[this](const CopyableErrorResult& aRv) {
mPromiseRequestHolder.Complete();
Unused << PClientHandleOpParent::Send__delete__(this, aRv);
})
->Track(mPromiseRequestHolder);
},
[=](const CopyableErrorResult& failure) {
mSourcePromiseRequestHolder.Complete();
Unused << PClientHandleOpParent::Send__delete__(this, failure);
return;
})
->Track(mSourcePromiseRequestHolder);
}
} // namespace mozilla::dom
|
