diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 09:22:09 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 09:22:09 +0000 |
commit | 43a97878ce14b72f0981164f87f2e35e14151312 (patch) | |
tree | 620249daf56c0258faa40cbdcf9cfba06de2a846 /third_party/libwebrtc/rtc_base/async_tcp_socket.h | |
parent | Initial commit. (diff) | |
download | firefox-43a97878ce14b72f0981164f87f2e35e14151312.tar.xz firefox-43a97878ce14b72f0981164f87f2e35e14151312.zip |
Adding upstream version 110.0.1.upstream/110.0.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/libwebrtc/rtc_base/async_tcp_socket.h')
-rw-r--r-- | third_party/libwebrtc/rtc_base/async_tcp_socket.h | 123 |
1 files changed, 123 insertions, 0 deletions
diff --git a/third_party/libwebrtc/rtc_base/async_tcp_socket.h b/third_party/libwebrtc/rtc_base/async_tcp_socket.h new file mode 100644 index 0000000000..541080fba7 --- /dev/null +++ b/third_party/libwebrtc/rtc_base/async_tcp_socket.h @@ -0,0 +1,123 @@ +/* + * 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_ASYNC_TCP_SOCKET_H_ +#define RTC_BASE_ASYNC_TCP_SOCKET_H_ + +#include <stddef.h> + +#include <memory> + +#include "rtc_base/async_packet_socket.h" +#include "rtc_base/buffer.h" +#include "rtc_base/socket.h" +#include "rtc_base/socket_address.h" + +namespace rtc { + +// Simulates UDP semantics over TCP. Send and Recv packet sizes +// are preserved, and drops packets silently on Send, rather than +// buffer them in user space. +class AsyncTCPSocketBase : public AsyncPacketSocket { + public: + AsyncTCPSocketBase(Socket* socket, size_t max_packet_size); + ~AsyncTCPSocketBase() override; + + AsyncTCPSocketBase(const AsyncTCPSocketBase&) = delete; + AsyncTCPSocketBase& operator=(const AsyncTCPSocketBase&) = delete; + + // Pure virtual methods to send and recv data. + int Send(const void* pv, + size_t cb, + const rtc::PacketOptions& options) override = 0; + virtual void ProcessInput(char* data, size_t* len) = 0; + + SocketAddress GetLocalAddress() const override; + SocketAddress GetRemoteAddress() const override; + int SendTo(const void* pv, + size_t cb, + const SocketAddress& addr, + const rtc::PacketOptions& options) override; + int Close() override; + + State GetState() const override; + int GetOption(Socket::Option opt, int* value) override; + int SetOption(Socket::Option opt, int value) override; + int GetError() const override; + void SetError(int error) override; + + protected: + // Binds and connects `socket` and creates AsyncTCPSocket for + // it. Takes ownership of `socket`. Returns null if bind() or + // connect() fail (`socket` is destroyed in that case). + static Socket* ConnectSocket(Socket* socket, + const SocketAddress& bind_address, + const SocketAddress& remote_address); + int FlushOutBuffer(); + // Add data to `outbuf_`. + void AppendToOutBuffer(const void* pv, size_t cb); + + // Helper methods for `outpos_`. + bool IsOutBufferEmpty() const { return outbuf_.size() == 0; } + void ClearOutBuffer() { outbuf_.Clear(); } + + private: + // Called by the underlying socket + void OnConnectEvent(Socket* socket); + void OnReadEvent(Socket* socket); + void OnWriteEvent(Socket* socket); + void OnCloseEvent(Socket* socket, int error); + + std::unique_ptr<Socket> socket_; + Buffer inbuf_; + Buffer outbuf_; + size_t max_insize_; + size_t max_outsize_; +}; + +class AsyncTCPSocket : public AsyncTCPSocketBase { + public: + // Binds and connects `socket` and creates AsyncTCPSocket for + // it. Takes ownership of `socket`. Returns null if bind() or + // connect() fail (`socket` is destroyed in that case). + static AsyncTCPSocket* Create(Socket* socket, + const SocketAddress& bind_address, + const SocketAddress& remote_address); + explicit AsyncTCPSocket(Socket* socket); + ~AsyncTCPSocket() override {} + + AsyncTCPSocket(const AsyncTCPSocket&) = delete; + AsyncTCPSocket& operator=(const AsyncTCPSocket&) = delete; + + int Send(const void* pv, + size_t cb, + const rtc::PacketOptions& options) override; + void ProcessInput(char* data, size_t* len) override; +}; + +class AsyncTcpListenSocket : public AsyncListenSocket { + public: + explicit AsyncTcpListenSocket(std::unique_ptr<Socket> socket); + + State GetState() const override; + SocketAddress GetLocalAddress() const override; + + virtual void HandleIncomingConnection(rtc::Socket* socket); + + private: + // Called by the underlying socket + void OnReadEvent(Socket* socket); + + std::unique_ptr<Socket> socket_; +}; + +} // namespace rtc + +#endif // RTC_BASE_ASYNC_TCP_SOCKET_H_ |