diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
commit | 26a029d407be480d791972afb5975cf62c9360a6 (patch) | |
tree | f435a8308119effd964b339f76abb83a57c29483 /third_party/libwebrtc/rtc_base/proxy_server.h | |
parent | Initial commit. (diff) | |
download | firefox-26a029d407be480d791972afb5975cf62c9360a6.tar.xz firefox-26a029d407be480d791972afb5975cf62c9360a6.zip |
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/libwebrtc/rtc_base/proxy_server.h')
-rw-r--r-- | third_party/libwebrtc/rtc_base/proxy_server.h | 110 |
1 files changed, 110 insertions, 0 deletions
diff --git a/third_party/libwebrtc/rtc_base/proxy_server.h b/third_party/libwebrtc/rtc_base/proxy_server.h new file mode 100644 index 0000000000..0b9b655a5e --- /dev/null +++ b/third_party/libwebrtc/rtc_base/proxy_server.h @@ -0,0 +1,110 @@ +/* + * 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_PROXY_SERVER_H_ +#define RTC_BASE_PROXY_SERVER_H_ + +#include <memory> +#include <vector> + +#include "absl/memory/memory.h" +#include "rtc_base/memory/fifo_buffer.h" +#include "rtc_base/server_socket_adapters.h" +#include "rtc_base/socket.h" +#include "rtc_base/socket_address.h" + +namespace rtc { + +class SocketFactory; + +// ProxyServer is a base class that allows for easy construction of proxy +// servers. With its helper class ProxyBinding, it contains all the necessary +// logic for receiving and bridging connections. The specific client-server +// proxy protocol is implemented by an instance of the AsyncProxyServerSocket +// class; children of ProxyServer implement WrapSocket appropriately to return +// the correct protocol handler. + +class ProxyBinding : public sigslot::has_slots<> { + public: + ProxyBinding(AsyncProxyServerSocket* in_socket, Socket* out_socket); + ~ProxyBinding() override; + + ProxyBinding(const ProxyBinding&) = delete; + ProxyBinding& operator=(const ProxyBinding&) = delete; + + sigslot::signal1<ProxyBinding*> SignalDestroyed; + + private: + void OnConnectRequest(AsyncProxyServerSocket* socket, + const SocketAddress& addr); + void OnInternalRead(Socket* socket); + void OnInternalWrite(Socket* socket); + void OnInternalClose(Socket* socket, int err); + void OnExternalConnect(Socket* socket); + void OnExternalRead(Socket* socket); + void OnExternalWrite(Socket* socket); + void OnExternalClose(Socket* socket, int err); + + static void Read(Socket* socket, FifoBuffer* buffer); + static void Write(Socket* socket, FifoBuffer* buffer); + void Destroy(); + + static const int kBufferSize = 4096; + std::unique_ptr<AsyncProxyServerSocket> int_socket_; + std::unique_ptr<Socket> ext_socket_; + bool connected_; + FifoBuffer out_buffer_; + FifoBuffer in_buffer_; +}; + +class ProxyServer : public sigslot::has_slots<> { + public: + ProxyServer(SocketFactory* int_factory, + const SocketAddress& int_addr, + SocketFactory* ext_factory, + const SocketAddress& ext_ip); + ~ProxyServer() override; + + ProxyServer(const ProxyServer&) = delete; + ProxyServer& operator=(const ProxyServer&) = delete; + + // Returns the address to which the proxy server is bound + SocketAddress GetServerAddress(); + + protected: + void OnAcceptEvent(Socket* socket); + virtual AsyncProxyServerSocket* WrapSocket(Socket* socket) = 0; + + private: + SocketFactory* ext_factory_; + SocketAddress ext_ip_; + std::unique_ptr<Socket> server_socket_; + std::vector<std::unique_ptr<ProxyBinding>> bindings_; +}; + +// SocksProxyServer is a simple extension of ProxyServer to implement SOCKS. +class SocksProxyServer : public ProxyServer { + public: + SocksProxyServer(SocketFactory* int_factory, + const SocketAddress& int_addr, + SocketFactory* ext_factory, + const SocketAddress& ext_ip) + : ProxyServer(int_factory, int_addr, ext_factory, ext_ip) {} + + SocksProxyServer(const SocksProxyServer&) = delete; + SocksProxyServer& operator=(const SocksProxyServer&) = delete; + + protected: + AsyncProxyServerSocket* WrapSocket(Socket* socket) override; +}; + +} // namespace rtc + +#endif // RTC_BASE_PROXY_SERVER_H_ |