blob: 541080fba76c8602ef574791927ff9fe9368004a [file] [log] [blame]
henrike@webrtc.orgf0488722014-05-13 18:00:261/*
2 * Copyright 2004 The WebRTC Project Authors. All rights reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Steve Anton10542f22019-01-11 17:11:0011#ifndef RTC_BASE_ASYNC_TCP_SOCKET_H_
12#define RTC_BASE_ASYNC_TCP_SOCKET_H_
henrike@webrtc.orgf0488722014-05-13 18:00:2613
Yves Gerey988cc082018-10-23 10:03:0114#include <stddef.h>
Jonas Olssona4d87372019-07-05 17:08:3315
Henrik Kjellanderec78f1c2017-06-29 05:52:5016#include <memory>
jbauch555604a2016-04-26 10:13:2217
Steve Anton10542f22019-01-11 17:11:0018#include "rtc_base/async_packet_socket.h"
Yves Gerey988cc082018-10-23 10:03:0119#include "rtc_base/buffer.h"
Yves Gerey988cc082018-10-23 10:03:0120#include "rtc_base/socket.h"
Steve Anton10542f22019-01-11 17:11:0021#include "rtc_base/socket_address.h"
Henrik Kjellanderec78f1c2017-06-29 05:52:5022
23namespace rtc {
24
25// Simulates UDP semantics over TCP. Send and Recv packet sizes
26// are preserved, and drops packets silently on Send, rather than
27// buffer them in user space.
28class AsyncTCPSocketBase : public AsyncPacketSocket {
29 public:
Niels Möllerd30ece12021-10-19 08:11:0230 AsyncTCPSocketBase(Socket* socket, size_t max_packet_size);
Henrik Kjellanderec78f1c2017-06-29 05:52:5031 ~AsyncTCPSocketBase() override;
32
Byoungchan Lee14af7622022-01-11 20:24:5833 AsyncTCPSocketBase(const AsyncTCPSocketBase&) = delete;
34 AsyncTCPSocketBase& operator=(const AsyncTCPSocketBase&) = delete;
35
Henrik Kjellanderec78f1c2017-06-29 05:52:5036 // Pure virtual methods to send and recv data.
Yves Gerey665174f2018-06-19 13:03:0537 int Send(const void* pv,
38 size_t cb,
39 const rtc::PacketOptions& options) override = 0;
Henrik Kjellanderec78f1c2017-06-29 05:52:5040 virtual void ProcessInput(char* data, size_t* len) = 0;
Henrik Kjellanderec78f1c2017-06-29 05:52:5041
42 SocketAddress GetLocalAddress() const override;
43 SocketAddress GetRemoteAddress() const override;
44 int SendTo(const void* pv,
45 size_t cb,
46 const SocketAddress& addr,
47 const rtc::PacketOptions& options) override;
48 int Close() override;
49
50 State GetState() const override;
51 int GetOption(Socket::Option opt, int* value) override;
52 int SetOption(Socket::Option opt, int value) override;
53 int GetError() const override;
54 void SetError(int error) override;
55
56 protected:
Artem Titov96e3b992021-07-26 14:03:1457 // Binds and connects `socket` and creates AsyncTCPSocket for
58 // it. Takes ownership of `socket`. Returns null if bind() or
59 // connect() fail (`socket` is destroyed in that case).
Niels Möllerd0b88792021-08-12 08:32:3060 static Socket* ConnectSocket(Socket* socket,
61 const SocketAddress& bind_address,
62 const SocketAddress& remote_address);
Henrik Kjellanderec78f1c2017-06-29 05:52:5063 int FlushOutBuffer();
Artem Titov96e3b992021-07-26 14:03:1464 // Add data to `outbuf_`.
Henrik Kjellanderec78f1c2017-06-29 05:52:5065 void AppendToOutBuffer(const void* pv, size_t cb);
66
Artem Titov96e3b992021-07-26 14:03:1467 // Helper methods for `outpos_`.
Henrik Kjellanderec78f1c2017-06-29 05:52:5068 bool IsOutBufferEmpty() const { return outbuf_.size() == 0; }
69 void ClearOutBuffer() { outbuf_.Clear(); }
70
71 private:
72 // Called by the underlying socket
Niels Möllerd0b88792021-08-12 08:32:3073 void OnConnectEvent(Socket* socket);
74 void OnReadEvent(Socket* socket);
75 void OnWriteEvent(Socket* socket);
76 void OnCloseEvent(Socket* socket, int error);
Henrik Kjellanderec78f1c2017-06-29 05:52:5077
Niels Möllerd0b88792021-08-12 08:32:3078 std::unique_ptr<Socket> socket_;
Henrik Kjellanderec78f1c2017-06-29 05:52:5079 Buffer inbuf_;
80 Buffer outbuf_;
81 size_t max_insize_;
82 size_t max_outsize_;
Henrik Kjellanderec78f1c2017-06-29 05:52:5083};
84
85class AsyncTCPSocket : public AsyncTCPSocketBase {
86 public:
Artem Titov96e3b992021-07-26 14:03:1487 // Binds and connects `socket` and creates AsyncTCPSocket for
88 // it. Takes ownership of `socket`. Returns null if bind() or
89 // connect() fail (`socket` is destroyed in that case).
Niels Möllerd0b88792021-08-12 08:32:3090 static AsyncTCPSocket* Create(Socket* socket,
Henrik Kjellanderec78f1c2017-06-29 05:52:5091 const SocketAddress& bind_address,
92 const SocketAddress& remote_address);
Niels Möllerd30ece12021-10-19 08:11:0293 explicit AsyncTCPSocket(Socket* socket);
Henrik Kjellanderec78f1c2017-06-29 05:52:5094 ~AsyncTCPSocket() override {}
95
Byoungchan Lee14af7622022-01-11 20:24:5896 AsyncTCPSocket(const AsyncTCPSocket&) = delete;
97 AsyncTCPSocket& operator=(const AsyncTCPSocket&) = delete;
98
Henrik Kjellanderec78f1c2017-06-29 05:52:5099 int Send(const void* pv,
100 size_t cb,
101 const rtc::PacketOptions& options) override;
102 void ProcessInput(char* data, size_t* len) override;
Henrik Kjellanderec78f1c2017-06-29 05:52:50103};
104
Niels Möllerd30ece12021-10-19 08:11:02105class AsyncTcpListenSocket : public AsyncListenSocket {
106 public:
107 explicit AsyncTcpListenSocket(std::unique_ptr<Socket> socket);
108
109 State GetState() const override;
110 SocketAddress GetLocalAddress() const override;
111
Niels Möllerd30ece12021-10-19 08:11:02112 virtual void HandleIncomingConnection(rtc::Socket* socket);
113
114 private:
115 // Called by the underlying socket
116 void OnReadEvent(Socket* socket);
117
118 std::unique_ptr<Socket> socket_;
119};
120
Henrik Kjellanderec78f1c2017-06-29 05:52:50121} // namespace rtc
henrike@webrtc.orgf0488722014-05-13 18:00:26122
Steve Anton10542f22019-01-11 17:11:00123#endif // RTC_BASE_ASYNC_TCP_SOCKET_H_