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
|
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* 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 DOM_WEBTRANSPORT_PARENT_WEBTRANSPORTPARENT_H_
#define DOM_WEBTRANSPORT_PARENT_WEBTRANSPORTPARENT_H_
#include "ErrorList.h"
#include "mozilla/dom/ClientIPCTypes.h"
#include "mozilla/dom/FlippedOnce.h"
#include "mozilla/dom/PWebTransportParent.h"
#include "mozilla/ipc/Endpoint.h"
#include "mozilla/ipc/PBackgroundSharedTypes.h"
#include "nsISupports.h"
#include "nsIPrincipal.h"
#include "nsIWebTransport.h"
#include "nsIWebTransportStream.h"
#include "nsTHashMap.h"
namespace mozilla::dom {
enum class WebTransportReliabilityMode : uint8_t;
class WebTransportParent : public PWebTransportParent,
public WebTransportSessionEventListener {
using IPCResult = mozilla::ipc::IPCResult;
public:
WebTransportParent() = default;
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_WEBTRANSPORTSESSIONEVENTLISTENER
void Create(const nsAString& aURL, nsIPrincipal* aPrincipal,
const mozilla::Maybe<IPCClientInfo>& aClientInfo,
const bool& aDedicated, const bool& aRequireUnreliable,
const uint32_t& aCongestionControl,
nsTArray<WebTransportHash>&& aServerCertHashes,
Endpoint<PWebTransportParent>&& aParentEndpoint,
std::function<void(std::tuple<const nsresult&, const uint8_t&>)>&&
aResolver);
IPCResult RecvClose(const uint32_t& aCode, const nsACString& aReason);
IPCResult RecvSetSendOrder(uint64_t aStreamId, Maybe<int64_t> aSendOrder);
IPCResult RecvCreateUnidirectionalStream(
Maybe<int64_t> aSendOrder,
CreateUnidirectionalStreamResolver&& aResolver);
IPCResult RecvCreateBidirectionalStream(
Maybe<int64_t> aSendOrder, CreateBidirectionalStreamResolver&& aResolver);
::mozilla::ipc::IPCResult RecvOutgoingDatagram(
nsTArray<uint8_t>&& aData, const TimeStamp& aExpirationTime,
OutgoingDatagramResolver&& aResolver);
::mozilla::ipc::IPCResult RecvGetMaxDatagramSize(
GetMaxDatagramSizeResolver&& aResolver);
void ActorDestroy(ActorDestroyReason aWhy) override;
class OnResetOrStopSendingCallback final {
public:
explicit OnResetOrStopSendingCallback(
std::function<void(nsresult)>&& aCallback)
: mCallback(std::move(aCallback)) {}
~OnResetOrStopSendingCallback() = default;
void OnResetOrStopSending(nsresult aError) { mCallback(aError); }
private:
std::function<void(nsresult)> mCallback;
};
protected:
virtual ~WebTransportParent();
private:
void NotifyRemoteClosed(bool aCleanly, uint32_t aErrorCode,
const nsACString& aReason);
using ResolveType = std::tuple<const nsresult&, const uint8_t&>;
nsCOMPtr<nsISerialEventTarget> mSocketThread;
Atomic<bool> mSessionReady{false};
mozilla::Mutex mMutex{"WebTransportParent::mMutex"};
std::function<void(ResolveType)> mResolver MOZ_GUARDED_BY(mMutex);
// This is needed because mResolver is resolved on the background thread and
// OnSessionClosed is called on the socket thread.
std::function<void()> mExecuteAfterResolverCallback MOZ_GUARDED_BY(mMutex);
OutgoingDatagramResolver mOutgoingDatagramResolver;
GetMaxDatagramSizeResolver mMaxDatagramSizeResolver;
FlippedOnce<false> mClosed MOZ_GUARDED_BY(mMutex);
nsCOMPtr<nsIWebTransport> mWebTransport;
nsCOMPtr<nsIEventTarget> mOwningEventTarget;
// What we need to be able to lookup by streamId
template <typename T>
struct StreamHash {
OnResetOrStopSendingCallback mCallback;
nsCOMPtr<T> mStream;
};
nsTHashMap<nsUint64HashKey, StreamHash<nsIWebTransportBidirectionalStream>>
mBidiStreamCallbackMap;
nsTHashMap<nsUint64HashKey, StreamHash<nsIWebTransportSendStream>>
mUniStreamCallbackMap;
};
} // namespace mozilla::dom
#endif // DOM_WEBTRANSPORT_PARENT_WEBTRANSPORTPARENT_H_
|
