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
|
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* vim:set ts=4 sw=4 sts=4 et cin: */
/* 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/. */
// HttpLog.h should generally be included first
#include "HttpLog.h"
#include "AltServiceChild.h"
#include "mozilla/ClearOnShutdown.h"
#include "mozilla/StaticPtr.h"
#include "mozilla/net/SocketProcessChild.h"
#include "nsHttpConnectionInfo.h"
#include "nsProxyInfo.h"
#include "nsThreadUtils.h"
namespace mozilla {
namespace net {
StaticRefPtr<AltServiceChild> sAltServiceChild;
AltServiceChild::AltServiceChild() {
LOG(("AltServiceChild ctor [%p]\n", this));
}
AltServiceChild::~AltServiceChild() {
LOG(("AltServiceChild dtor [%p]\n", this));
}
// static
bool AltServiceChild::EnsureAltServiceChild() {
MOZ_ASSERT(XRE_IsSocketProcess());
MOZ_ASSERT(NS_IsMainThread());
if (sAltServiceChild) {
return true;
}
SocketProcessChild* socketChild = SocketProcessChild::GetSingleton();
if (!socketChild || socketChild->IsShuttingDown()) {
return false;
}
sAltServiceChild = new AltServiceChild();
ClearOnShutdown(&sAltServiceChild);
if (!socketChild->SendPAltServiceConstructor(sAltServiceChild)) {
sAltServiceChild = nullptr;
return false;
}
return true;
}
// static
void AltServiceChild::ClearHostMapping(nsHttpConnectionInfo* aCi) {
LOG(("AltServiceChild::ClearHostMapping ci=%s", aCi->HashKey().get()));
MOZ_ASSERT(aCi);
RefPtr<nsHttpConnectionInfo> ci = aCi->Clone();
auto task = [ci{std::move(ci)}]() {
if (!EnsureAltServiceChild()) {
return;
}
if (!ci->GetOrigin().IsEmpty() && sAltServiceChild->CanSend()) {
Unused << sAltServiceChild->SendClearHostMapping(
ci->GetOrigin(), ci->OriginPort(), ci->GetOriginAttributes());
}
};
if (!NS_IsMainThread()) {
MOZ_ALWAYS_SUCCEEDS(NS_DispatchToMainThread(NS_NewRunnableFunction(
"net::AltServiceChild::ClearHostMapping", task)));
return;
}
task();
}
// static
void AltServiceChild::ProcessHeader(
const nsCString& aBuf, const nsCString& aOriginScheme,
const nsCString& aOriginHost, int32_t aOriginPort,
const nsCString& aUsername, bool aPrivateBrowsing,
nsIInterfaceRequestor* aCallbacks, nsProxyInfo* aProxyInfo, uint32_t aCaps,
const OriginAttributes& aOriginAttributes) {
LOG(("AltServiceChild::ProcessHeader"));
MOZ_ASSERT(NS_IsMainThread());
if (!EnsureAltServiceChild()) {
return;
}
if (!sAltServiceChild->CanSend()) {
return;
}
nsTArray<ProxyInfoCloneArgs> proxyInfoArray;
if (aProxyInfo) {
nsProxyInfo::SerializeProxyInfo(aProxyInfo, proxyInfoArray);
}
Unused << sAltServiceChild->SendProcessHeader(
aBuf, aOriginScheme, aOriginHost, aOriginPort, aUsername,
aPrivateBrowsing, proxyInfoArray, aCaps, aOriginAttributes);
}
} // namespace net
} // namespace mozilla
|
