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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
|
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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/. */
#ifndef HttpConnectionBase_h__
#define HttpConnectionBase_h__
#include "nsHttpConnectionInfo.h"
#include "nsHttpResponseHead.h"
#include "nsAHttpTransaction.h"
#include "nsCOMPtr.h"
#include "nsProxyRelease.h"
#include "prinrval.h"
#include "mozilla/Mutex.h"
#include "ARefBase.h"
#include "TimingStruct.h"
#include "HttpTrafficAnalyzer.h"
#include "mozilla/net/DNS.h"
#include "mozilla/WeakPtr.h"
#include "nsIAsyncInputStream.h"
#include "nsIAsyncOutputStream.h"
#include "nsIInterfaceRequestor.h"
#include "nsITimer.h"
class nsISocketTransport;
class nsITLSSocketControl;
namespace mozilla {
namespace net {
class ConnectionEntry;
class nsHttpHandler;
class ASpdySession;
class WebTransportSessionBase;
enum class ConnectionState : uint32_t {
HALF_OPEN = 0,
INITED,
TLS_HANDSHAKING,
ZERORTT,
TRANSFERING,
CLOSED
};
enum class ConnectionExperienceState : uint32_t {
Not_Experienced = 0,
First_Request_Sent = (1 << 0),
First_Response_Received = (1 << 1),
Experienced = (1 << 2),
};
MOZ_MAKE_ENUM_CLASS_BITWISE_OPERATORS(ConnectionExperienceState);
// 1dcc863e-db90-4652-a1fe-13fea0b54e46
#define HTTPCONNECTIONBASE_IID \
{0x437e7d26, 0xa2fd, 0x49f2, {0xb3, 0x7c, 0x84, 0x23, 0xf0, 0x94, 0x72, 0x36}}
//-----------------------------------------------------------------------------
// nsHttpConnection - represents a connection to a HTTP server (or proxy)
//
// NOTE: this objects lives on the socket thread only. it should not be
// accessed from any other thread.
//-----------------------------------------------------------------------------
class HttpConnectionBase : public nsSupportsWeakReference {
public:
NS_INLINE_DECL_STATIC_IID(HTTPCONNECTIONBASE_IID)
HttpConnectionBase();
// Activate causes the given transaction to be processed on this
// connection. It fails if there is already an existing transaction unless
// a multiplexing protocol such as SPDY is being used
[[nodiscard]] virtual nsresult Activate(nsAHttpTransaction*, uint32_t caps,
int32_t pri) = 0;
// Close the underlying socket transport.
virtual void Close(nsresult reason, bool aIsShutdown = false) = 0;
virtual bool CanReuse() = 0; // can this connection be reused?
virtual bool CanDirectlyActivate() = 0;
virtual void DontReuse() = 0;
virtual nsAHttpTransaction* Transaction() = 0;
nsHttpConnectionInfo* ConnectionInfo() { return mConnInfo; }
virtual void CloseTransaction(nsAHttpTransaction*, nsresult,
bool aIsShutdown = false) = 0;
[[nodiscard]] virtual nsresult OnHeadersAvailable(nsAHttpTransaction*,
nsHttpRequestHead*,
nsHttpResponseHead*,
bool* reset) = 0;
[[nodiscard]] virtual nsresult TakeTransport(nsISocketTransport**,
nsIAsyncInputStream**,
nsIAsyncOutputStream**) = 0;
virtual WebTransportSessionBase* GetWebTransportSession(
nsAHttpTransaction* aTransaction) {
return nullptr;
}
virtual bool UsingSpdy() { return false; }
virtual bool UsingHttp3() { return false; }
virtual void SetTransactionCaps(uint32_t aCaps) { mTransactionCaps = aCaps; }
virtual void PrintDiagnostics(nsCString& log) = 0;
// IsExperienced() returns true when the connection has started at least one
// non null HTTP transaction of any version.
bool IsExperienced() { return mExperienced; }
virtual bool TestJoinConnection(const nsACString& hostname, int32_t port) = 0;
virtual bool JoinConnection(const nsACString& hostname, int32_t port) = 0;
// Return true when the socket this connection is using has not been
// authenticated using a client certificate. Before SSL negotiation
// has finished this returns false.
virtual bool NoClientCertAuth() const { return true; }
// HTTP/2 websocket support
virtual ExtendedCONNECTSupport GetExtendedCONNECTSupport() {
return ExtendedCONNECTSupport::NO_SUPPORT;
}
void GetConnectionInfo(nsHttpConnectionInfo** ci) {
*ci = do_AddRef(mConnInfo).take();
}
virtual void GetTLSSocketControl(nsITLSSocketControl** result) = 0;
[[nodiscard]] virtual nsresult ResumeSend() = 0;
[[nodiscard]] virtual nsresult ResumeRecv() = 0;
[[nodiscard]] virtual nsresult ForceSend() = 0;
[[nodiscard]] virtual nsresult ForceRecv() = 0;
virtual HttpVersion Version() = 0;
virtual bool LastTransactionExpectedNoContent() = 0;
virtual void SetLastTransactionExpectedNoContent(bool) = 0;
virtual int64_t BytesWritten() = 0; // includes TLS
void SetSecurityCallbacks(nsIInterfaceRequestor* aCallbacks);
void SetTrafficCategory(HttpTrafficCategory);
void BootstrapTimings(TimingStruct times);
virtual bool IsPersistent() = 0;
virtual bool IsReused() = 0;
[[nodiscard]] virtual nsresult PushBack(const char* data,
uint32_t length) = 0;
PRIntervalTime Rtt() { return mRtt; }
virtual void SetEvent(nsresult aStatus) = 0;
virtual nsISocketTransport* Transport() { return nullptr; }
virtual nsresult GetSelfAddr(NetAddr* addr) = 0;
virtual nsresult GetPeerAddr(NetAddr* addr) = 0;
virtual bool ResolvedByTRR() = 0;
virtual nsIRequest::TRRMode EffectiveTRRMode() = 0;
virtual TRRSkippedReason TRRSkipReason() = 0;
virtual bool GetEchConfigUsed() = 0;
virtual PRIntervalTime LastWriteTime() = 0;
void ChangeConnectionState(ConnectionState aState);
ConnectionCloseReason CloseReason() const { return mCloseReason; }
void SetCloseReason(ConnectionCloseReason aReason) {
if (mCloseReason == ConnectionCloseReason::UNSET) {
mCloseReason = aReason;
}
}
void RecordConnectionCloseTelemetry(nsresult aReason);
void RecordConnectionAddressType();
bool IsProxyConnectInProgress() { return mState == SETTING_UP_TUNNEL; }
virtual nsresult CreateTunnelStream(nsAHttpTransaction* httpTransaction,
HttpConnectionBase** aHttpConnection,
bool aIsExtendedCONNECT = false) = 0;
virtual void SetInTunnel() {};
void SetOwner(ConnectionEntry* aEntry);
ConnectionEntry* OwnerEntry() const;
protected:
// The capabailities associated with the most recent transaction
uint32_t mTransactionCaps{0};
RefPtr<nsHttpConnectionInfo> mConnInfo;
// Set when this connection is added to the active connections list,
// cleared when it is removed.
WeakPtr<ConnectionEntry> mOwnerEntry;
bool mExperienced{false};
// Used to track whether this connection is serving the first request.
bool mHasFirstHttpTransaction{false};
bool mBootstrappedTimingsSet{false};
TimingStruct mBootstrappedTimings;
Mutex mCallbacksLock MOZ_UNANNOTATED{"nsHttpConnection::mCallbacksLock"};
nsMainThreadPtrHandle<nsIInterfaceRequestor> mCallbacks;
nsTArray<HttpTrafficCategory> mTrafficCategory;
PRIntervalTime mRtt{0};
nsresult mErrorBeforeConnect = NS_OK;
ConnectionState mConnectionState = ConnectionState::HALF_OPEN;
// Represent if the connection has served more than one request.
ConnectionExperienceState mExperienceState =
ConnectionExperienceState::Not_Experienced;
ConnectionCloseReason mCloseReason = ConnectionCloseReason::UNSET;
bool mAddressTypeReported{false};
// Tunnel retated functions:
enum HttpConnectionState {
UNINITIALIZED,
SETTING_UP_TUNNEL,
REQUEST,
} mState{HttpConnectionState::UNINITIALIZED};
void ChangeState(HttpConnectionState newState);
bool TunnelSetupInProgress() { return mState == SETTING_UP_TUNNEL; }
virtual void SetTunnelSetupDone() {}
virtual nsresult SetupProxyConnectStream() { return NS_OK; }
nsresult CheckTunnelIsNeeded(nsAHttpTransaction* aTransaction);
};
#define NS_DECL_HTTPCONNECTIONBASE \
[[nodiscard]] nsresult Activate(nsAHttpTransaction*, uint32_t, int32_t) \
override; \
[[nodiscard]] nsresult OnHeadersAvailable( \
nsAHttpTransaction*, nsHttpRequestHead*, nsHttpResponseHead*, \
bool* reset) override; \
[[nodiscard]] nsresult TakeTransport( \
nsISocketTransport**, nsIAsyncInputStream**, nsIAsyncOutputStream**) \
override; \
void Close(nsresult, bool aIsShutdown = false) override; \
bool CanReuse() override; \
bool CanDirectlyActivate() override; \
void DontReuse() override; \
void CloseTransaction(nsAHttpTransaction*, nsresult, \
bool aIsShutdown = false) override; \
void PrintDiagnostics(nsCString&) override; \
bool TestJoinConnection(const nsACString&, int32_t) override; \
bool JoinConnection(const nsACString&, int32_t) override; \
void GetTLSSocketControl(nsITLSSocketControl** result) override; \
[[nodiscard]] nsresult ResumeSend() override; \
[[nodiscard]] nsresult ResumeRecv() override; \
[[nodiscard]] nsresult ForceSend() override; \
[[nodiscard]] nsresult ForceRecv() override; \
HttpVersion Version() override; \
bool LastTransactionExpectedNoContent() override; \
void SetLastTransactionExpectedNoContent(bool val) override; \
bool IsPersistent() override; \
bool IsReused() override; \
[[nodiscard]] nsresult PushBack(const char* data, uint32_t length) override; \
void SetEvent(nsresult aStatus) override; \
virtual nsAHttpTransaction* Transaction() override; \
PRIntervalTime LastWriteTime() override;
} // namespace net
} // namespace mozilla
#endif // HttpConnectionBase_h__
|
