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
|
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* 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_WebTransportStreamProxy_h
#define mozilla_net_WebTransportStreamProxy_h
#include "nsIAsyncInputStream.h"
#include "nsIAsyncOutputStream.h"
#include "nsIWebTransportStream.h"
#include "nsCOMPtr.h"
namespace mozilla::net {
class Http3WebTransportStream;
class WebTransportStreamProxy final : public nsIWebTransportReceiveStream,
public nsIWebTransportSendStream,
public nsIWebTransportBidirectionalStream,
public nsIAsyncInputStream,
public nsIAsyncOutputStream {
public:
NS_DECL_THREADSAFE_ISUPPORTS
explicit WebTransportStreamProxy(Http3WebTransportStream* aStream);
NS_IMETHOD SendStopSending(uint8_t aError) override;
NS_IMETHOD SendFin() override;
NS_IMETHOD Reset(uint8_t aErrorCode) override;
NS_IMETHOD GetSendStreamStats(
nsIWebTransportStreamStatsCallback* aCallback) override;
NS_IMETHOD GetReceiveStreamStats(
nsIWebTransportStreamStatsCallback* aCallback) override;
NS_IMETHOD Close() override;
NS_IMETHOD Available(uint64_t* aAvailable) override;
NS_IMETHOD Read(char* aBuf, uint32_t aCount, uint32_t* aResult) override;
NS_IMETHOD ReadSegments(nsWriteSegmentFun aWriter, void* aClosure,
uint32_t aCount, uint32_t* aResult) override;
NS_IMETHOD IsNonBlocking(bool* aResult) override;
NS_IMETHOD CloseWithStatus(nsresult aStatus) override;
NS_IMETHOD AsyncWait(nsIInputStreamCallback* aCallback, uint32_t aFlags,
uint32_t aRequestedCount,
nsIEventTarget* aEventTarget) override;
// XPCConnect seems to be confused about the other AsyncWait (with
// nsIOutputStreamCallback), so we need this function for reading.
NS_IMETHOD AsyncWaitForRead(nsIInputStreamCallback* aCallback,
uint32_t aFlags, uint32_t aRequestedCount,
nsIEventTarget* aTarget) override;
NS_IMETHOD Flush() override;
NS_IMETHOD Write(const char* aBuf, uint32_t aCount,
uint32_t* aResult) override;
NS_IMETHOD WriteFrom(nsIInputStream* aFromStream, uint32_t aCount,
uint32_t* aResult) override;
NS_IMETHOD WriteSegments(nsReadSegmentFun aReader, void* aClosure,
uint32_t aCount, uint32_t* aResult) override;
NS_IMETHOD AsyncWait(nsIOutputStreamCallback* aCallback, uint32_t aFlags,
uint32_t aRequestedCount,
nsIEventTarget* aEventTarget) override;
private:
virtual ~WebTransportStreamProxy();
static nsresult WriteFromSegments(nsIInputStream*, void*, const char*,
uint32_t offset, uint32_t count,
uint32_t* countRead);
RefPtr<Http3WebTransportStream> mWebTransportStream;
nsCOMPtr<nsIAsyncOutputStream> mWriter;
nsCOMPtr<nsIAsyncInputStream> mReader;
};
} // namespace mozilla::net
#endif
|
