diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/libwebrtc/rtc_base/socket_adapters.h | |
parent | Initial commit. (diff) | |
download | firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/libwebrtc/rtc_base/socket_adapters.h')
-rw-r--r-- | third_party/libwebrtc/rtc_base/socket_adapters.h | 178 |
1 files changed, 178 insertions, 0 deletions
diff --git a/third_party/libwebrtc/rtc_base/socket_adapters.h b/third_party/libwebrtc/rtc_base/socket_adapters.h new file mode 100644 index 0000000000..e78ee18a27 --- /dev/null +++ b/third_party/libwebrtc/rtc_base/socket_adapters.h @@ -0,0 +1,178 @@ +/* + * Copyright 2004 The WebRTC Project Authors. All rights reserved. + * + * Use of this source code is governed by a BSD-style license + * that can be found in the LICENSE file in the root of the source + * tree. An additional intellectual property rights grant can be found + * in the file PATENTS. All contributing project authors may + * be found in the AUTHORS file in the root of the source tree. + */ + +#ifndef RTC_BASE_SOCKET_ADAPTERS_H_ +#define RTC_BASE_SOCKET_ADAPTERS_H_ + +#include <string> + +#include "absl/strings/string_view.h" +#include "api/array_view.h" +#include "rtc_base/async_socket.h" +#include "rtc_base/crypt_string.h" + +namespace rtc { + +struct HttpAuthContext; +class ByteBufferReader; +class ByteBufferWriter; + +/////////////////////////////////////////////////////////////////////////////// + +// Implements a socket adapter that can buffer and process data internally, +// as in the case of connecting to a proxy, where you must speak the proxy +// protocol before commencing normal socket behavior. +class BufferedReadAdapter : public AsyncSocketAdapter { + public: + BufferedReadAdapter(Socket* socket, size_t buffer_size); + ~BufferedReadAdapter() override; + + BufferedReadAdapter(const BufferedReadAdapter&) = delete; + BufferedReadAdapter& operator=(const BufferedReadAdapter&) = delete; + + int Send(const void* pv, size_t cb) override; + int Recv(void* pv, size_t cb, int64_t* timestamp) override; + + protected: + int DirectSend(const void* pv, size_t cb) { + return AsyncSocketAdapter::Send(pv, cb); + } + + void BufferInput(bool on = true); + virtual void ProcessInput(char* data, size_t* len) = 0; + + void OnReadEvent(Socket* socket) override; + + private: + char* buffer_; + size_t buffer_size_, data_len_; + bool buffering_; +}; + +/////////////////////////////////////////////////////////////////////////////// + +// Implements a socket adapter that performs the client side of a +// fake SSL handshake. Used for "ssltcp" P2P functionality. +class AsyncSSLSocket : public BufferedReadAdapter { + public: + static ArrayView<const uint8_t> SslClientHello(); + static ArrayView<const uint8_t> SslServerHello(); + + explicit AsyncSSLSocket(Socket* socket); + + AsyncSSLSocket(const AsyncSSLSocket&) = delete; + AsyncSSLSocket& operator=(const AsyncSSLSocket&) = delete; + + int Connect(const SocketAddress& addr) override; + + protected: + void OnConnectEvent(Socket* socket) override; + void ProcessInput(char* data, size_t* len) override; +}; + +/////////////////////////////////////////////////////////////////////////////// + +// Implements a socket adapter that speaks the HTTP/S proxy protocol. +class AsyncHttpsProxySocket : public BufferedReadAdapter { + public: + AsyncHttpsProxySocket(Socket* socket, + absl::string_view user_agent, + const SocketAddress& proxy, + absl::string_view username, + const CryptString& password); + ~AsyncHttpsProxySocket() override; + + AsyncHttpsProxySocket(const AsyncHttpsProxySocket&) = delete; + AsyncHttpsProxySocket& operator=(const AsyncHttpsProxySocket&) = delete; + + // If connect is forced, the adapter will always issue an HTTP CONNECT to the + // target address. Otherwise, it will connect only if the destination port + // is not port 80. + void SetForceConnect(bool force) { force_connect_ = force; } + + int Connect(const SocketAddress& addr) override; + SocketAddress GetRemoteAddress() const override; + int Close() override; + ConnState GetState() const override; + + protected: + void OnConnectEvent(Socket* socket) override; + void OnCloseEvent(Socket* socket, int err) override; + void ProcessInput(char* data, size_t* len) override; + + bool ShouldIssueConnect() const; + void SendRequest(); + void ProcessLine(char* data, size_t len); + void EndResponse(); + void Error(int error); + + private: + SocketAddress proxy_, dest_; + std::string agent_, user_, headers_; + CryptString pass_; + bool force_connect_; + size_t content_length_; + int defer_error_; + bool expect_close_; + enum ProxyState { + PS_INIT, + PS_LEADER, + PS_AUTHENTICATE, + PS_SKIP_HEADERS, + PS_ERROR_HEADERS, + PS_TUNNEL_HEADERS, + PS_SKIP_BODY, + PS_TUNNEL, + PS_WAIT_CLOSE, + PS_ERROR + } state_; + HttpAuthContext* context_; + std::string unknown_mechanisms_; +}; + +/////////////////////////////////////////////////////////////////////////////// + +// Implements a socket adapter that speaks the SOCKS proxy protocol. +class AsyncSocksProxySocket : public BufferedReadAdapter { + public: + AsyncSocksProxySocket(Socket* socket, + const SocketAddress& proxy, + absl::string_view username, + const CryptString& password); + ~AsyncSocksProxySocket() override; + + AsyncSocksProxySocket(const AsyncSocksProxySocket&) = delete; + AsyncSocksProxySocket& operator=(const AsyncSocksProxySocket&) = delete; + + int Connect(const SocketAddress& addr) override; + SocketAddress GetRemoteAddress() const override; + int Close() override; + ConnState GetState() const override; + + protected: + void OnConnectEvent(Socket* socket) override; + void ProcessInput(char* data, size_t* len) override; + + void SendHello(); + void SendConnect(); + void SendAuth(); + void Error(int error); + + private: + enum State { SS_INIT, SS_HELLO, SS_AUTH, SS_CONNECT, SS_TUNNEL, SS_ERROR }; + State state_; + SocketAddress proxy_, dest_; + std::string user_; + CryptString pass_; +}; + +} // namespace rtc + +#endif // RTC_BASE_SOCKET_ADAPTERS_H_ |