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
|
/* -*- Mode: C++; tab-width: 8; 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 mozilla_net_Http3WebTransportSession_h
#define mozilla_net_Http3WebTransportSession_h
#include "WebTransportSessionBase.h"
#include "Http3StreamBase.h"
#include "nsIWebTransport.h"
#include "mozilla/net/NeqoHttp3Conn.h"
namespace mozilla::net {
class Http3Session;
class Http3TunnelStreamBase : public Http3StreamBase,
public nsAHttpSegmentWriter,
public nsAHttpSegmentReader {
public:
NS_DECL_NSAHTTPSEGMENTWRITER
NS_DECL_NSAHTTPSEGMENTREADER
explicit Http3TunnelStreamBase(nsAHttpTransaction* trans,
Http3SessionBase* aHttp3Session);
[[nodiscard]] nsresult ReadSegments() override;
[[nodiscard]] nsresult WriteSegments() override;
void SetResponseHeaders(nsTArray<uint8_t>& aResponseHeaders, bool fin,
bool interim) override;
virtual nsresult TryActivating();
bool Done() const override { return mRecvState == RECV_DONE; }
virtual void OnDatagramReceived(nsTArray<uint8_t>&& aData) = 0;
void TransactionIsDone(nsresult aResult);
virtual bool OnActivated() { return true; }
virtual nsresult OnProcessDatagram() { return NS_OK; }
protected:
bool ConsumeHeaders(const char* buf, uint32_t avail, uint32_t* countUsed);
virtual void OnClosePending() = 0;
enum RecvStreamState {
BEFORE_HEADERS,
READING_HEADERS,
READING_INTERIM_HEADERS,
ACTIVE,
CLOSE_PENDING,
RECV_DONE
} mRecvState{BEFORE_HEADERS};
enum SendStreamState {
PREPARING_HEADERS,
WAITING_TO_ACTIVATE,
PROCESSING_DATAGRAM,
SEND_DONE,
} mSendState{PREPARING_HEADERS};
nsCString mFlatHttpRequestHeaders;
nsTArray<uint8_t> mFlatResponseHeaders;
nsresult mSocketInCondition = NS_ERROR_NOT_INITIALIZED;
nsresult mSocketOutCondition = NS_ERROR_NOT_INITIALIZED;
};
// TODO Http3WebTransportSession is very similar to Http3Stream. It should
// be built on top of it with a couple of small changes. The docs will be added
// when this is implemented.
class Http3WebTransportSession final : public WebTransportSessionBase,
public Http3TunnelStreamBase {
public:
NS_INLINE_DECL_THREADSAFE_REFCOUNTING(Http3WebTransportSession, override)
Http3WebTransportSession(nsAHttpTransaction*, Http3Session*);
Http3WebTransportSession* GetHttp3WebTransportSession() override {
return this;
}
Http3WebTransportStream* GetHttp3WebTransportStream() override {
return nullptr;
}
Http3Stream* GetHttp3Stream() override { return nullptr; }
Http3ConnectUDPStream* GetHttp3ConnectUDPStream() override { return nullptr; }
Http3StreamTunnel* GetHttp3StreamTunnel() override { return nullptr; }
void Close(nsresult aResult) override;
void CloseSession(uint32_t aStatus, const nsACString& aReason) override;
void OnSessionClosed(bool aCleanly, uint32_t aStatus,
const nsACString& aReason);
uint64_t GetStreamId() const override;
void CreateOutgoingBidirectionalStream(
std::function<void(Result<RefPtr<WebTransportStreamBase>, nsresult>&&)>&&
aCallback) override;
void CreateOutgoingUnidirectionalStream(
std::function<void(Result<RefPtr<WebTransportStreamBase>, nsresult>&&)>&&
aCallback) override;
void RemoveWebTransportStream(Http3WebTransportStream* aStream);
already_AddRefed<Http3WebTransportStream> OnIncomingWebTransportStream(
WebTransportStreamType aType, uint64_t aId);
void SendDatagram(nsTArray<uint8_t>&& aData, uint64_t aTrackingId) override;
void OnDatagramReceived(nsTArray<uint8_t>&& aData) override;
void GetMaxDatagramSize() override;
void OnOutgoingDatagramOutCome(
uint64_t aId, WebTransportSessionEventListener::DatagramOutcome aOutCome);
void OnStreamStopSending(uint64_t aId, nsresult aError);
void OnStreamReset(uint64_t aId, nsresult aError);
private:
virtual ~Http3WebTransportSession();
void CreateStreamInternal(
bool aBidi,
std::function<void(Result<RefPtr<WebTransportStreamBase>, nsresult>&&)>&&
aCallback);
void OnClosePending() override;
nsTArray<RefPtr<Http3WebTransportStream>> mStreams;
uint32_t mStatus{0};
nsCString mReason;
};
} // namespace mozilla::net
#endif // mozilla_net_Http3WebTransportSession_h
|
