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
|
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set sw=2 ts=8 et tw=80 ft=cpp : */
/* 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 webrtc_tcp_socket_h__
#define webrtc_tcp_socket_h__
#include <list>
#include "nsCOMPtr.h"
#include "nsIAsyncInputStream.h"
#include "nsIAsyncOutputStream.h"
#include "nsIAuthPromptProvider.h"
#include "nsIHttpChannelInternal.h"
#include "nsIInterfaceRequestor.h"
#include "nsIStreamListener.h"
#include "nsStringFwd.h"
#include "nsTArray.h"
#include "nsIProtocolProxyCallback.h"
#include "mozilla/net/WebrtcProxyConfig.h"
class nsISocketTransport;
namespace mozilla::net {
class WebrtcTCPSocketCallback;
class WebrtcTCPData;
class WebrtcTCPSocket : public nsIHttpUpgradeListener,
public nsIStreamListener,
public nsIInputStreamCallback,
public nsIOutputStreamCallback,
public nsIInterfaceRequestor,
public nsIAuthPromptProvider,
public nsIProtocolProxyCallback {
public:
NS_DECL_NSIHTTPUPGRADELISTENER
NS_DECL_NSIINPUTSTREAMCALLBACK
NS_DECL_NSIINTERFACEREQUESTOR
NS_DECL_NSIOUTPUTSTREAMCALLBACK
NS_DECL_NSIREQUESTOBSERVER
NS_DECL_NSISTREAMLISTENER
NS_DECL_THREADSAFE_ISUPPORTS
NS_FORWARD_SAFE_NSIAUTHPROMPTPROVIDER(mAuthProvider)
NS_DECL_NSIPROTOCOLPROXYCALLBACK
explicit WebrtcTCPSocket(WebrtcTCPSocketCallback* aCallbacks);
void SetTabId(dom::TabId aTabId);
nsresult Open(const nsACString& aHost, const int& aPort,
const nsACString& aLocalAddress, const int& aLocalPort,
bool aUseTls,
const Maybe<net::WebrtcProxyConfig>& aProxyConfig);
nsresult Write(nsTArray<uint8_t>&& aBytes);
nsresult Close();
size_t CountUnwrittenBytes() const;
protected:
virtual ~WebrtcTCPSocket();
// protected for gtests
virtual void InvokeOnClose(nsresult aReason);
virtual void InvokeOnConnected();
virtual void InvokeOnRead(nsTArray<uint8_t>&& aReadData);
RefPtr<WebrtcTCPSocketCallback> mProxyCallbacks;
private:
bool mClosed;
bool mOpened;
nsCOMPtr<nsIURI> mURI;
bool mTls = false;
Maybe<WebrtcProxyConfig> mProxyConfig;
nsCString mLocalAddress;
uint16_t mLocalPort = 0;
nsCString mProxyType;
nsresult DoProxyConfigLookup();
nsresult OpenWithHttpProxy();
void OpenWithoutHttpProxy(nsIProxyInfo* aSocksProxyInfo);
void FinishOpen();
void EnqueueWrite_s(nsTArray<uint8_t>&& aWriteData);
void CloseWithReason(nsresult aReason);
size_t mWriteOffset;
std::list<WebrtcTCPData> mWriteQueue;
nsCOMPtr<nsIAuthPromptProvider> mAuthProvider;
// Indicates that the channel is CONNECTed
nsCOMPtr<nsISocketTransport> mTransport;
nsCOMPtr<nsIAsyncInputStream> mSocketIn;
nsCOMPtr<nsIAsyncOutputStream> mSocketOut;
nsCOMPtr<nsIEventTarget> mMainThread;
nsCOMPtr<nsIEventTarget> mSocketThread;
nsCOMPtr<nsICancelable> mProxyRequest;
};
} // namespace mozilla::net
#endif // webrtc_tcp_socket_h__
|
