henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 1 | /* |
| 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 Anton | 10542f2 | 2019-01-11 17:11:00 | [diff] [blame] | 11 | #include "rtc_base/async_tcp_socket.h" |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 12 | |
Yves Gerey | 3e70781 | 2018-11-28 15:47:49 | [diff] [blame] | 13 | #include <stdint.h> |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 14 | #include <string.h> |
Jonas Olsson | a4d8737 | 2019-07-05 17:08:33 | [diff] [blame] | 15 | |
jbauch | 313afba | 2016-03-03 11:41:05 | [diff] [blame] | 16 | #include <algorithm> |
jbauch | 555604a | 2016-04-26 10:13:22 | [diff] [blame] | 17 | #include <memory> |
jbauch | 313afba | 2016-03-03 11:41:05 | [diff] [blame] | 18 | |
Steve Anton | 3fa2b80 | 2020-01-31 18:31:53 | [diff] [blame] | 19 | #include "api/array_view.h" |
Steve Anton | 10542f2 | 2019-01-11 17:11:00 | [diff] [blame] | 20 | #include "rtc_base/byte_order.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 04:47:31 | [diff] [blame] | 21 | #include "rtc_base/checks.h" |
| 22 | #include "rtc_base/logging.h" |
Yves Gerey | 3e70781 | 2018-11-28 15:47:49 | [diff] [blame] | 23 | #include "rtc_base/network/sent_packet.h" |
Steve Anton | 10542f2 | 2019-01-11 17:11:00 | [diff] [blame] | 24 | #include "rtc_base/time_utils.h" // for TimeMillis |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 25 | |
| 26 | #if defined(WEBRTC_POSIX) |
| 27 | #include <errno.h> |
| 28 | #endif // WEBRTC_POSIX |
| 29 | |
| 30 | namespace rtc { |
| 31 | |
| 32 | static const size_t kMaxPacketSize = 64 * 1024; |
| 33 | |
Peter Boström | 0c4e06b | 2015-10-07 10:23:21 | [diff] [blame] | 34 | typedef uint16_t PacketLength; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 35 | static const size_t kPacketLenSize = sizeof(PacketLength); |
| 36 | |
| 37 | static const size_t kBufSize = kMaxPacketSize + kPacketLenSize; |
| 38 | |
jbauch | 313afba | 2016-03-03 11:41:05 | [diff] [blame] | 39 | // The input buffer will be resized so that at least kMinimumRecvSize bytes can |
| 40 | // be received (but it will not grow above the maximum size passed to the |
| 41 | // constructor). |
| 42 | static const size_t kMinimumRecvSize = 128; |
| 43 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 44 | static const int kListenBacklog = 5; |
| 45 | |
Artem Titov | 96e3b99 | 2021-07-26 14:03:14 | [diff] [blame] | 46 | // Binds and connects `socket` |
Niels Möller | d0b8879 | 2021-08-12 08:32:30 | [diff] [blame] | 47 | Socket* AsyncTCPSocketBase::ConnectSocket( |
| 48 | rtc::Socket* socket, |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 49 | const rtc::SocketAddress& bind_address, |
| 50 | const rtc::SocketAddress& remote_address) { |
Niels Möller | d0b8879 | 2021-08-12 08:32:30 | [diff] [blame] | 51 | std::unique_ptr<rtc::Socket> owned_socket(socket); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 52 | if (socket->Bind(bind_address) < 0) { |
Mirko Bonadei | 675513b | 2017-11-09 10:09:25 | [diff] [blame] | 53 | RTC_LOG(LS_ERROR) << "Bind() failed with error " << socket->GetError(); |
deadbeef | 37f5ecf | 2017-02-27 22:06:41 | [diff] [blame] | 54 | return nullptr; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 55 | } |
| 56 | if (socket->Connect(remote_address) < 0) { |
Mirko Bonadei | 675513b | 2017-11-09 10:09:25 | [diff] [blame] | 57 | RTC_LOG(LS_ERROR) << "Connect() failed with error " << socket->GetError(); |
deadbeef | 37f5ecf | 2017-02-27 22:06:41 | [diff] [blame] | 58 | return nullptr; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 59 | } |
| 60 | return owned_socket.release(); |
| 61 | } |
| 62 | |
Jared Siskin | 802e8e5 | 2023-04-20 00:35:28 | [diff] [blame] | 63 | AsyncTCPSocketBase::AsyncTCPSocketBase(Socket* socket, size_t max_packet_size) |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 64 | : socket_(socket), |
jbauch | 313afba | 2016-03-03 11:41:05 | [diff] [blame] | 65 | max_insize_(max_packet_size), |
jbauch | 250fc65 | 2016-02-28 23:06:40 | [diff] [blame] | 66 | max_outsize_(max_packet_size) { |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 67 | inbuf_.EnsureCapacity(kMinimumRecvSize); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 68 | |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 69 | socket_->SignalConnectEvent.connect(this, |
| 70 | &AsyncTCPSocketBase::OnConnectEvent); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 71 | socket_->SignalReadEvent.connect(this, &AsyncTCPSocketBase::OnReadEvent); |
| 72 | socket_->SignalWriteEvent.connect(this, &AsyncTCPSocketBase::OnWriteEvent); |
| 73 | socket_->SignalCloseEvent.connect(this, &AsyncTCPSocketBase::OnCloseEvent); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 74 | } |
| 75 | |
jbauch | 3c16576 | 2016-02-26 17:31:32 | [diff] [blame] | 76 | AsyncTCPSocketBase::~AsyncTCPSocketBase() {} |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 77 | |
| 78 | SocketAddress AsyncTCPSocketBase::GetLocalAddress() const { |
| 79 | return socket_->GetLocalAddress(); |
| 80 | } |
| 81 | |
| 82 | SocketAddress AsyncTCPSocketBase::GetRemoteAddress() const { |
| 83 | return socket_->GetRemoteAddress(); |
| 84 | } |
| 85 | |
| 86 | int AsyncTCPSocketBase::Close() { |
| 87 | return socket_->Close(); |
| 88 | } |
| 89 | |
| 90 | AsyncTCPSocket::State AsyncTCPSocketBase::GetState() const { |
| 91 | switch (socket_->GetState()) { |
| 92 | case Socket::CS_CLOSED: |
| 93 | return STATE_CLOSED; |
| 94 | case Socket::CS_CONNECTING: |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 95 | return STATE_CONNECTING; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 96 | case Socket::CS_CONNECTED: |
| 97 | return STATE_CONNECTED; |
| 98 | default: |
Artem Titov | d325196 | 2021-11-15 15:57:07 | [diff] [blame] | 99 | RTC_DCHECK_NOTREACHED(); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 100 | return STATE_CLOSED; |
| 101 | } |
| 102 | } |
| 103 | |
| 104 | int AsyncTCPSocketBase::GetOption(Socket::Option opt, int* value) { |
| 105 | return socket_->GetOption(opt, value); |
| 106 | } |
| 107 | |
| 108 | int AsyncTCPSocketBase::SetOption(Socket::Option opt, int value) { |
| 109 | return socket_->SetOption(opt, value); |
| 110 | } |
| 111 | |
| 112 | int AsyncTCPSocketBase::GetError() const { |
| 113 | return socket_->GetError(); |
| 114 | } |
| 115 | |
| 116 | void AsyncTCPSocketBase::SetError(int error) { |
| 117 | return socket_->SetError(error); |
| 118 | } |
| 119 | |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 120 | int AsyncTCPSocketBase::SendTo(const void* pv, |
| 121 | size_t cb, |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 122 | const SocketAddress& addr, |
| 123 | const rtc::PacketOptions& options) { |
honghaiz | b19eba3 | 2015-08-03 17:23:31 | [diff] [blame] | 124 | const SocketAddress& remote_address = GetRemoteAddress(); |
| 125 | if (addr == remote_address) |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 126 | return Send(pv, cb, options); |
honghaiz | b19eba3 | 2015-08-03 17:23:31 | [diff] [blame] | 127 | // Remote address may be empty if there is a sudden network change. |
jbauch | 3c16576 | 2016-02-26 17:31:32 | [diff] [blame] | 128 | RTC_DCHECK(remote_address.IsNil()); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 129 | socket_->SetError(ENOTCONN); |
| 130 | return -1; |
| 131 | } |
| 132 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 133 | int AsyncTCPSocketBase::FlushOutBuffer() { |
Steve Anton | 3fa2b80 | 2020-01-31 18:31:53 | [diff] [blame] | 134 | RTC_DCHECK_GT(outbuf_.size(), 0); |
| 135 | rtc::ArrayView<uint8_t> view = outbuf_; |
| 136 | int res; |
| 137 | while (view.size() > 0) { |
| 138 | res = socket_->Send(view.data(), view.size()); |
| 139 | if (res <= 0) { |
| 140 | break; |
| 141 | } |
| 142 | if (static_cast<size_t>(res) > view.size()) { |
Artem Titov | d325196 | 2021-11-15 15:57:07 | [diff] [blame] | 143 | RTC_DCHECK_NOTREACHED(); |
Steve Anton | 3fa2b80 | 2020-01-31 18:31:53 | [diff] [blame] | 144 | res = -1; |
| 145 | break; |
| 146 | } |
| 147 | view = view.subview(res); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 148 | } |
Steve Anton | 3fa2b80 | 2020-01-31 18:31:53 | [diff] [blame] | 149 | if (res > 0) { |
| 150 | // The output buffer may have been written out over multiple partial Send(), |
| 151 | // so reconstruct the total written length. |
| 152 | RTC_DCHECK_EQ(view.size(), 0); |
| 153 | res = outbuf_.size(); |
| 154 | outbuf_.Clear(); |
| 155 | } else { |
| 156 | // There was an error when calling Send(), so there will still be data left |
| 157 | // to send at a later point. |
| 158 | RTC_DCHECK_GT(view.size(), 0); |
| 159 | // In the special case of EWOULDBLOCK, signal that we had a partial write. |
| 160 | if (socket_->GetError() == EWOULDBLOCK) { |
| 161 | res = outbuf_.size() - view.size(); |
| 162 | } |
| 163 | if (view.size() < outbuf_.size()) { |
| 164 | memmove(outbuf_.data(), view.data(), view.size()); |
| 165 | outbuf_.SetSize(view.size()); |
| 166 | } |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 167 | } |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 168 | return res; |
| 169 | } |
| 170 | |
| 171 | void AsyncTCPSocketBase::AppendToOutBuffer(const void* pv, size_t cb) { |
jbauch | 250fc65 | 2016-02-28 23:06:40 | [diff] [blame] | 172 | RTC_DCHECK(outbuf_.size() + cb <= max_outsize_); |
jbauch | 250fc65 | 2016-02-28 23:06:40 | [diff] [blame] | 173 | outbuf_.AppendData(static_cast<const uint8_t*>(pv), cb); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 174 | } |
| 175 | |
Niels Möller | d0b8879 | 2021-08-12 08:32:30 | [diff] [blame] | 176 | void AsyncTCPSocketBase::OnConnectEvent(Socket* socket) { |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 177 | SignalConnect(this); |
| 178 | } |
| 179 | |
Niels Möller | d0b8879 | 2021-08-12 08:32:30 | [diff] [blame] | 180 | void AsyncTCPSocketBase::OnReadEvent(Socket* socket) { |
jbauch | 3c16576 | 2016-02-26 17:31:32 | [diff] [blame] | 181 | RTC_DCHECK(socket_.get() == socket); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 182 | |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 183 | size_t total_recv = 0; |
| 184 | while (true) { |
| 185 | size_t free_size = inbuf_.capacity() - inbuf_.size(); |
| 186 | if (free_size < kMinimumRecvSize && inbuf_.capacity() < max_insize_) { |
| 187 | inbuf_.EnsureCapacity(std::min(max_insize_, inbuf_.capacity() * 2)); |
| 188 | free_size = inbuf_.capacity() - inbuf_.size(); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 189 | } |
| 190 | |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 191 | int len = socket_->Recv(inbuf_.data() + inbuf_.size(), free_size, nullptr); |
| 192 | if (len < 0) { |
| 193 | // TODO(stefan): Do something better like forwarding the error to the |
| 194 | // user. |
| 195 | if (!socket_->IsBlocking()) { |
| 196 | RTC_LOG(LS_ERROR) << "Recv() returned error: " << socket_->GetError(); |
| 197 | } |
| 198 | break; |
| 199 | } |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 200 | |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 201 | total_recv += len; |
| 202 | inbuf_.SetSize(inbuf_.size() + len); |
| 203 | if (!len || static_cast<size_t>(len) < free_size) { |
| 204 | break; |
| 205 | } |
| 206 | } |
| 207 | |
| 208 | if (!total_recv) { |
| 209 | return; |
| 210 | } |
| 211 | |
| 212 | size_t size = inbuf_.size(); |
| 213 | ProcessInput(inbuf_.data<char>(), &size); |
| 214 | |
| 215 | if (size > inbuf_.size()) { |
| 216 | RTC_LOG(LS_ERROR) << "input buffer overflow"; |
Artem Titov | d325196 | 2021-11-15 15:57:07 | [diff] [blame] | 217 | RTC_DCHECK_NOTREACHED(); |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 218 | inbuf_.Clear(); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 219 | } else { |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 220 | inbuf_.SetSize(size); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 221 | } |
| 222 | } |
| 223 | |
Niels Möller | d0b8879 | 2021-08-12 08:32:30 | [diff] [blame] | 224 | void AsyncTCPSocketBase::OnWriteEvent(Socket* socket) { |
jbauch | 3c16576 | 2016-02-26 17:31:32 | [diff] [blame] | 225 | RTC_DCHECK(socket_.get() == socket); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 226 | |
jbauch | 250fc65 | 2016-02-28 23:06:40 | [diff] [blame] | 227 | if (outbuf_.size() > 0) { |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 228 | FlushOutBuffer(); |
| 229 | } |
| 230 | |
jbauch | 250fc65 | 2016-02-28 23:06:40 | [diff] [blame] | 231 | if (outbuf_.size() == 0) { |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 232 | SignalReadyToSend(this); |
| 233 | } |
| 234 | } |
| 235 | |
Niels Möller | d0b8879 | 2021-08-12 08:32:30 | [diff] [blame] | 236 | void AsyncTCPSocketBase::OnCloseEvent(Socket* socket, int error) { |
Tomas Gunnarsson | f15189d | 2022-04-13 09:03:52 | [diff] [blame] | 237 | NotifyClosed(error); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 238 | } |
| 239 | |
| 240 | // AsyncTCPSocket |
Artem Titov | 96e3b99 | 2021-07-26 14:03:14 | [diff] [blame] | 241 | // Binds and connects `socket` and creates AsyncTCPSocket for |
| 242 | // it. Takes ownership of `socket`. Returns null if bind() or |
| 243 | // connect() fail (`socket` is destroyed in that case). |
Niels Möller | d0b8879 | 2021-08-12 08:32:30 | [diff] [blame] | 244 | AsyncTCPSocket* AsyncTCPSocket::Create(Socket* socket, |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 245 | const SocketAddress& bind_address, |
| 246 | const SocketAddress& remote_address) { |
| 247 | return new AsyncTCPSocket( |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 248 | AsyncTCPSocketBase::ConnectSocket(socket, bind_address, remote_address)); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 249 | } |
| 250 | |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 251 | AsyncTCPSocket::AsyncTCPSocket(Socket* socket) |
| 252 | : AsyncTCPSocketBase(socket, kBufSize) {} |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 253 | |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 254 | int AsyncTCPSocket::Send(const void* pv, |
| 255 | size_t cb, |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 256 | const rtc::PacketOptions& options) { |
| 257 | if (cb > kBufSize) { |
| 258 | SetError(EMSGSIZE); |
| 259 | return -1; |
| 260 | } |
| 261 | |
| 262 | // If we are blocking on send, then silently drop this packet |
| 263 | if (!IsOutBufferEmpty()) |
| 264 | return static_cast<int>(cb); |
| 265 | |
| 266 | PacketLength pkt_len = HostToNetwork16(static_cast<PacketLength>(cb)); |
| 267 | AppendToOutBuffer(&pkt_len, kPacketLenSize); |
| 268 | AppendToOutBuffer(pv, cb); |
| 269 | |
| 270 | int res = FlushOutBuffer(); |
| 271 | if (res <= 0) { |
| 272 | // drop packet if we made no progress |
| 273 | ClearOutBuffer(); |
| 274 | return res; |
| 275 | } |
| 276 | |
Qingsi Wang | 6e641e6 | 2018-04-12 03:14:17 | [diff] [blame] | 277 | rtc::SentPacket sent_packet(options.packet_id, rtc::TimeMillis(), |
| 278 | options.info_signaled_after_sent); |
Qingsi Wang | 4ea53b3 | 2018-04-17 01:22:31 | [diff] [blame] | 279 | CopySocketInformationToPacketInfo(cb, *this, false, &sent_packet.info); |
stefan | c1aeaf0 | 2015-10-15 14:26:07 | [diff] [blame] | 280 | SignalSentPacket(this, sent_packet); |
| 281 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 282 | // We claim to have sent the whole thing, even if we only sent partial |
| 283 | return static_cast<int>(cb); |
| 284 | } |
| 285 | |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 286 | void AsyncTCPSocket::ProcessInput(char* data, size_t* len) { |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 287 | SocketAddress remote_addr(GetRemoteAddress()); |
| 288 | |
| 289 | while (true) { |
| 290 | if (*len < kPacketLenSize) |
| 291 | return; |
| 292 | |
| 293 | PacketLength pkt_len = rtc::GetBE16(data); |
| 294 | if (*len < kPacketLenSize + pkt_len) |
| 295 | return; |
| 296 | |
| 297 | SignalReadPacket(this, data + kPacketLenSize, pkt_len, remote_addr, |
Niels Möller | 15ca5a9 | 2018-11-01 13:32:47 | [diff] [blame] | 298 | TimeMicros()); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 299 | |
| 300 | *len -= kPacketLenSize + pkt_len; |
| 301 | if (*len > 0) { |
| 302 | memmove(data, data + kPacketLenSize + pkt_len, *len); |
| 303 | } |
| 304 | } |
| 305 | } |
| 306 | |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 307 | AsyncTcpListenSocket::AsyncTcpListenSocket(std::unique_ptr<Socket> socket) |
| 308 | : socket_(std::move(socket)) { |
| 309 | RTC_DCHECK(socket_.get() != nullptr); |
| 310 | socket_->SignalReadEvent.connect(this, &AsyncTcpListenSocket::OnReadEvent); |
| 311 | if (socket_->Listen(kListenBacklog) < 0) { |
| 312 | RTC_LOG(LS_ERROR) << "Listen() failed with error " << socket_->GetError(); |
| 313 | } |
| 314 | } |
| 315 | |
| 316 | AsyncTcpListenSocket::State AsyncTcpListenSocket::GetState() const { |
| 317 | switch (socket_->GetState()) { |
| 318 | case Socket::CS_CLOSED: |
| 319 | return State::kClosed; |
| 320 | case Socket::CS_CONNECTING: |
| 321 | return State::kBound; |
| 322 | default: |
Artem Titov | d325196 | 2021-11-15 15:57:07 | [diff] [blame] | 323 | RTC_DCHECK_NOTREACHED(); |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 324 | return State::kClosed; |
| 325 | } |
| 326 | } |
| 327 | |
| 328 | SocketAddress AsyncTcpListenSocket::GetLocalAddress() const { |
| 329 | return socket_->GetLocalAddress(); |
| 330 | } |
| 331 | |
Niels Möller | d30ece1 | 2021-10-19 08:11:02 | [diff] [blame] | 332 | void AsyncTcpListenSocket::OnReadEvent(Socket* socket) { |
| 333 | RTC_DCHECK(socket_.get() == socket); |
| 334 | |
| 335 | rtc::SocketAddress address; |
| 336 | rtc::Socket* new_socket = socket->Accept(&address); |
| 337 | if (!new_socket) { |
| 338 | // TODO(stefan): Do something better like forwarding the error |
| 339 | // to the user. |
| 340 | RTC_LOG(LS_ERROR) << "TCP accept failed with error " << socket_->GetError(); |
| 341 | return; |
| 342 | } |
| 343 | |
| 344 | HandleIncomingConnection(new_socket); |
| 345 | |
| 346 | // Prime a read event in case data is waiting. |
| 347 | new_socket->SignalReadEvent(new_socket); |
| 348 | } |
| 349 | |
| 350 | void AsyncTcpListenSocket::HandleIncomingConnection(Socket* socket) { |
| 351 | SignalNewConnection(this, new AsyncTCPSocket(socket)); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 | [diff] [blame] | 352 | } |
| 353 | |
| 354 | } // namespace rtc |