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
|
/* 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 "EarlyHintPreconnect.h"
#include "mozilla/CORSMode.h"
#include "mozilla/dom/Element.h"
#include "mozilla/StaticPrefs_network.h"
#include "nsIOService.h"
#include "nsIURI.h"
#include "SpeculativeTransaction.h"
namespace mozilla::net {
namespace {
class EarlyHintsPreConnectOverride : public nsIInterfaceRequestor,
public nsISpeculativeConnectionOverrider {
public:
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSISPECULATIVECONNECTIONOVERRIDER
NS_DECL_NSIINTERFACEREQUESTOR
explicit EarlyHintsPreConnectOverride(uint32_t aConnectionLimit)
: mConnectionLimit(aConnectionLimit) {}
private:
virtual ~EarlyHintsPreConnectOverride() = default;
// Only set once on main thread and can be read on multiple threads.
const uint32_t mConnectionLimit;
};
NS_IMPL_ISUPPORTS(EarlyHintsPreConnectOverride, nsIInterfaceRequestor,
nsISpeculativeConnectionOverrider)
NS_IMETHODIMP
EarlyHintsPreConnectOverride::GetInterface(const nsIID& iid, void** result) {
if (NS_SUCCEEDED(QueryInterface(iid, result)) && *result) {
return NS_OK;
}
return NS_ERROR_NO_INTERFACE;
}
NS_IMETHODIMP
EarlyHintsPreConnectOverride::GetIgnoreIdle(bool* ignoreIdle) {
*ignoreIdle = true;
return NS_OK;
}
NS_IMETHODIMP
EarlyHintsPreConnectOverride::GetParallelSpeculativeConnectLimit(
uint32_t* parallelSpeculativeConnectLimit) {
*parallelSpeculativeConnectLimit = mConnectionLimit;
return NS_OK;
}
NS_IMETHODIMP
EarlyHintsPreConnectOverride::GetIsFromPredictor(bool* isFromPredictor) {
*isFromPredictor = false;
return NS_OK;
}
NS_IMETHODIMP
EarlyHintsPreConnectOverride::GetAllow1918(bool* allow) {
*allow = false;
return NS_OK;
}
} // namespace
void EarlyHintPreconnect::MaybePreconnect(
const LinkHeader& aHeader, nsIURI* aBaseURI,
OriginAttributes&& aOriginAttributes) {
if (!StaticPrefs::network_early_hints_preconnect_enabled()) {
return;
}
if (!gIOService) {
return;
}
nsCOMPtr<nsIURI> uri;
if (NS_FAILED(aHeader.NewResolveHref(getter_AddRefs(uri), aBaseURI))) {
return;
}
// only preconnect secure context urls
if (!uri->SchemeIs("https")) {
return;
}
uint32_t maxPreconnectCount =
StaticPrefs::network_early_hints_preconnect_max_connections();
nsCOMPtr<nsIInterfaceRequestor> callbacks =
new EarlyHintsPreConnectOverride(maxPreconnectCount);
// Note that the http connection manager will limit the number of
// connections we can make, so it should be fine we don't check duplicate
// preconnect attempts here.
CORSMode corsMode = dom::Element::StringToCORSMode(aHeader.mCrossOrigin);
gIOService->SpeculativeConnectWithOriginAttributesNative(
uri, std::move(aOriginAttributes), callbacks, corsMode == CORS_ANONYMOUS);
}
} // namespace mozilla::net
|
