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
|
/* -*- 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_Http3WebTransportStream_h
#define mozilla_net_Http3WebTransportStream_h
#include <functional>
#include "Http3StreamBase.h"
#include "mozilla/net/NeqoHttp3Conn.h"
#include "mozilla/Maybe.h"
#include "mozilla/Result.h"
#include "mozilla/ResultVariant.h"
#include "nsIAsyncInputStream.h"
#include "nsIAsyncOutputStream.h"
class nsIWebTransportSendStreamStats;
class nsIWebTransportReceiveStreamStats;
namespace mozilla::net {
class Http3WebTransportSession;
class Http3WebTransportStream final : public Http3StreamBase,
public nsAHttpSegmentWriter,
public nsAHttpSegmentReader,
public nsIInputStreamCallback {
public:
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSAHTTPSEGMENTWRITER
NS_DECL_NSAHTTPSEGMENTREADER
NS_DECL_NSIINPUTSTREAMCALLBACK
explicit Http3WebTransportStream(
Http3Session* aSession, uint64_t aSessionId, WebTransportStreamType aType,
std::function<void(Result<RefPtr<Http3WebTransportStream>, nsresult>&&)>&&
aCallback);
explicit Http3WebTransportStream(Http3Session* aSession, uint64_t aSessionId,
WebTransportStreamType aType,
uint64_t aStreamId);
Http3WebTransportSession* GetHttp3WebTransportSession() override {
return nullptr;
}
Http3WebTransportStream* GetHttp3WebTransportStream() override {
return this;
}
Http3Stream* GetHttp3Stream() override { return nullptr; }
[[nodiscard]] nsresult ReadSegments() override;
[[nodiscard]] nsresult WriteSegments() override;
bool Done() const override;
void Close(nsresult aResult) override;
void SetResponseHeaders(nsTArray<uint8_t>& aResponseHeaders, bool fin,
bool interim) override {}
uint64_t SessionId() const { return mSessionId; }
WebTransportStreamType StreamType() const { return mStreamType; }
void SendFin();
void Reset(uint64_t aErrorCode);
void SendStopSending(uint8_t aErrorCode);
already_AddRefed<nsIWebTransportSendStreamStats> GetSendStreamStats();
already_AddRefed<nsIWebTransportReceiveStreamStats> GetReceiveStreamStats();
void GetWriterAndReader(nsIAsyncOutputStream** aOutOutputStream,
nsIAsyncInputStream** aOutInputStream);
// When mRecvState is RECV_DONE, this means we already received the FIN.
bool RecvDone() const { return mRecvState == RECV_DONE; }
private:
friend class Http3WebTransportSession;
virtual ~Http3WebTransportStream();
nsresult TryActivating();
static nsresult ReadRequestSegment(nsIInputStream*, void*, const char*,
uint32_t, uint32_t, uint32_t*);
static nsresult WritePipeSegment(nsIOutputStream*, void*, char*, uint32_t,
uint32_t, uint32_t*);
nsresult InitOutputPipe();
nsresult InitInputPipe();
uint64_t mSessionId{UINT64_MAX};
WebTransportStreamType mStreamType{WebTransportStreamType::BiDi};
enum StreamRole {
INCOMING,
OUTGOING,
} mStreamRole{INCOMING};
enum SendStreamState {
WAITING_TO_ACTIVATE,
WAITING_DATA,
SENDING,
SEND_DONE,
} mSendState{WAITING_TO_ACTIVATE};
enum RecvStreamState { BEFORE_READING, READING, RECEIVED_FIN, RECV_DONE };
Atomic<RecvStreamState> mRecvState{BEFORE_READING};
nsresult mSocketOutCondition = NS_ERROR_NOT_INITIALIZED;
nsresult mSocketInCondition = NS_ERROR_NOT_INITIALIZED;
std::function<void(Result<RefPtr<Http3WebTransportStream>, nsresult>&&)>
mStreamReadyCallback;
Mutex mMutex{"Http3WebTransportStream::mMutex"};
nsCOMPtr<nsIAsyncInputStream> mSendStreamPipeIn;
nsCOMPtr<nsIAsyncOutputStream> mSendStreamPipeOut MOZ_GUARDED_BY(mMutex);
nsCOMPtr<nsIAsyncInputStream> mReceiveStreamPipeIn MOZ_GUARDED_BY(mMutex);
nsCOMPtr<nsIAsyncOutputStream> mReceiveStreamPipeOut;
uint64_t mTotalSent = 0;
uint64_t mTotalReceived = 0;
// TODO: neqo doesn't expose this information for now.
uint64_t mTotalAcknowledged = 0;
bool mSendFin{false};
// The error code used to reset the stream. Should be only set once.
Maybe<uint64_t> mResetError;
// The error code used for STOP_SENDING. Should be only set once.
Maybe<uint8_t> mStopSendingError;
// This is used when SendFin or Reset is called when mSendState is SENDING.
nsTArray<std::function<void()>> mPendingTasks;
};
} // namespace mozilla::net
#endif // mozilla_net_Http3WebTransportStream_h
|
