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
|
/* -*- 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/JSValidatorParent.h"
#include "HttpBaseChannel.h"
#include "mozilla/RefPtr.h"
#include "mozilla/dom/JSOracleParent.h"
#include "mozilla/dom/JSValidatorUtils.h"
#include "mozilla/net/OpaqueResponseUtils.h"
#include "nsCOMPtr.h"
namespace mozilla::dom {
/* static */
already_AddRefed<JSValidatorParent> JSValidatorParent::Create() {
RefPtr<JSValidatorParent> validator = new JSValidatorParent();
JSOracleParent::WithJSOracle([validator](JSOracleParent* aParent) {
MOZ_ASSERT_IF(aParent, aParent->CanSend());
if (aParent) {
MOZ_ALWAYS_TRUE(aParent->SendPJSValidatorConstructor(validator));
}
});
return validator.forget();
}
void JSValidatorParent::IsOpaqueResponseAllowed(
const std::function<void(Maybe<Shmem>, ValidatorResult)>& aCallback) {
JSOracleParent::WithJSOracle([=, self = RefPtr{this}](const auto* aParent) {
if (aParent) {
MOZ_DIAGNOSTIC_ASSERT(self->CanSend());
self->SendIsOpaqueResponseAllowed()->Then(
GetMainThreadSerialEventTarget(), __func__,
[aCallback](
const IsOpaqueResponseAllowedPromise::ResolveOrRejectValue&
aResult) {
if (aResult.IsResolve()) {
auto [data, result] = aResult.ResolveValue();
aCallback(std::move(data), result);
} else {
// For cases like the Utility Process crashes, the promise will be
// rejected due to sending failures, and we'll block the request
// since we can't validate it.
aCallback(Nothing(), ValidatorResult::Failure);
}
});
} else {
aCallback(Nothing(), ValidatorResult::Failure);
}
});
}
void JSValidatorParent::OnDataAvailable(const nsACString& aData) {
JSOracleParent::WithJSOracle(
[self = RefPtr{this}, data = nsCString{aData}](const auto* aParent) {
if (!aParent) {
return;
}
if (self->CanSend()) {
Shmem sharedData;
nsresult rv =
JSValidatorUtils::CopyCStringToShmem(self, data, sharedData);
if (NS_FAILED(rv)) {
return;
}
(void)self->SendOnDataAvailable(std::move(sharedData));
}
});
}
void JSValidatorParent::OnStopRequest(nsresult aResult, nsIRequest& aRequest) {
JSOracleParent::WithJSOracle(
[self = RefPtr{this}, aResult,
request = nsCOMPtr{&aRequest}](const auto* aParent) {
if (!aParent) {
return;
}
if (self->CanSend() && request) {
nsCOMPtr<net::HttpBaseChannel> httpBaseChannel =
do_QueryInterface(request);
MOZ_ASSERT(httpBaseChannel);
nsAutoCString contentCharset;
(void)httpBaseChannel->GetContentCharset(contentCharset);
nsAutoString hintCharset;
(void)httpBaseChannel->GetClassicScriptHintCharset(hintCharset);
nsAutoString documentCharset;
(void)httpBaseChannel->GetDocumentCharacterSet(documentCharset);
(void)self->SendOnStopRequest(aResult, contentCharset, hintCharset,
documentCharset);
}
});
}
} // namespace mozilla::dom
|
