/* -*- 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__
