Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(168)

Unified Diff: webrtc/base/asynctcpsocket.cc

Issue 2877023002: Move webrtc/{base => rtc_base} (Closed)
Patch Set: update presubmit.py and DEPS include rules Created 3 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « webrtc/base/asynctcpsocket.h ('k') | webrtc/base/asynctcpsocket_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: webrtc/base/asynctcpsocket.cc
diff --git a/webrtc/base/asynctcpsocket.cc b/webrtc/base/asynctcpsocket.cc
deleted file mode 100644
index 658541cdde1282ca1405d7250bf69fcf7afe8f31..0000000000000000000000000000000000000000
--- a/webrtc/base/asynctcpsocket.cc
+++ /dev/null
@@ -1,331 +0,0 @@
-/*
- * 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.
- */
-
-#include "webrtc/base/asynctcpsocket.h"
-
-#include <string.h>
-
-#include <algorithm>
-#include <memory>
-
-#include "webrtc/base/byteorder.h"
-#include "webrtc/base/checks.h"
-#include "webrtc/base/logging.h"
-
-#if defined(WEBRTC_POSIX)
-#include <errno.h>
-#endif // WEBRTC_POSIX
-
-namespace rtc {
-
-static const size_t kMaxPacketSize = 64 * 1024;
-
-typedef uint16_t PacketLength;
-static const size_t kPacketLenSize = sizeof(PacketLength);
-
-static const size_t kBufSize = kMaxPacketSize + kPacketLenSize;
-
-// The input buffer will be resized so that at least kMinimumRecvSize bytes can
-// be received (but it will not grow above the maximum size passed to the
-// constructor).
-static const size_t kMinimumRecvSize = 128;
-
-static const int kListenBacklog = 5;
-
-// Binds and connects |socket|
-AsyncSocket* AsyncTCPSocketBase::ConnectSocket(
- rtc::AsyncSocket* socket,
- const rtc::SocketAddress& bind_address,
- const rtc::SocketAddress& remote_address) {
- std::unique_ptr<rtc::AsyncSocket> owned_socket(socket);
- if (socket->Bind(bind_address) < 0) {
- LOG(LS_ERROR) << "Bind() failed with error " << socket->GetError();
- return nullptr;
- }
- if (socket->Connect(remote_address) < 0) {
- LOG(LS_ERROR) << "Connect() failed with error " << socket->GetError();
- return nullptr;
- }
- return owned_socket.release();
-}
-
-AsyncTCPSocketBase::AsyncTCPSocketBase(AsyncSocket* socket, bool listen,
- size_t max_packet_size)
- : socket_(socket),
- listen_(listen),
- max_insize_(max_packet_size),
- max_outsize_(max_packet_size) {
- if (!listen_) {
- // Listening sockets don't send/receive data, so they don't need buffers.
- inbuf_.EnsureCapacity(kMinimumRecvSize);
- }
-
- RTC_DCHECK(socket_.get() != nullptr);
- socket_->SignalConnectEvent.connect(
- this, &AsyncTCPSocketBase::OnConnectEvent);
- socket_->SignalReadEvent.connect(this, &AsyncTCPSocketBase::OnReadEvent);
- socket_->SignalWriteEvent.connect(this, &AsyncTCPSocketBase::OnWriteEvent);
- socket_->SignalCloseEvent.connect(this, &AsyncTCPSocketBase::OnCloseEvent);
-
- if (listen_) {
- if (socket_->Listen(kListenBacklog) < 0) {
- LOG(LS_ERROR) << "Listen() failed with error " << socket_->GetError();
- }
- }
-}
-
-AsyncTCPSocketBase::~AsyncTCPSocketBase() {}
-
-SocketAddress AsyncTCPSocketBase::GetLocalAddress() const {
- return socket_->GetLocalAddress();
-}
-
-SocketAddress AsyncTCPSocketBase::GetRemoteAddress() const {
- return socket_->GetRemoteAddress();
-}
-
-int AsyncTCPSocketBase::Close() {
- return socket_->Close();
-}
-
-AsyncTCPSocket::State AsyncTCPSocketBase::GetState() const {
- switch (socket_->GetState()) {
- case Socket::CS_CLOSED:
- return STATE_CLOSED;
- case Socket::CS_CONNECTING:
- if (listen_) {
- return STATE_BOUND;
- } else {
- return STATE_CONNECTING;
- }
- case Socket::CS_CONNECTED:
- return STATE_CONNECTED;
- default:
- RTC_NOTREACHED();
- return STATE_CLOSED;
- }
-}
-
-int AsyncTCPSocketBase::GetOption(Socket::Option opt, int* value) {
- return socket_->GetOption(opt, value);
-}
-
-int AsyncTCPSocketBase::SetOption(Socket::Option opt, int value) {
- return socket_->SetOption(opt, value);
-}
-
-int AsyncTCPSocketBase::GetError() const {
- return socket_->GetError();
-}
-
-void AsyncTCPSocketBase::SetError(int error) {
- return socket_->SetError(error);
-}
-
-int AsyncTCPSocketBase::SendTo(const void *pv, size_t cb,
- const SocketAddress& addr,
- const rtc::PacketOptions& options) {
- const SocketAddress& remote_address = GetRemoteAddress();
- if (addr == remote_address)
- return Send(pv, cb, options);
- // Remote address may be empty if there is a sudden network change.
- RTC_DCHECK(remote_address.IsNil());
- socket_->SetError(ENOTCONN);
- return -1;
-}
-
-int AsyncTCPSocketBase::SendRaw(const void * pv, size_t cb) {
- if (outbuf_.size() + cb > max_outsize_) {
- socket_->SetError(EMSGSIZE);
- return -1;
- }
-
- RTC_DCHECK(!listen_);
- outbuf_.AppendData(static_cast<const uint8_t*>(pv), cb);
-
- return FlushOutBuffer();
-}
-
-int AsyncTCPSocketBase::FlushOutBuffer() {
- RTC_DCHECK(!listen_);
- int res = socket_->Send(outbuf_.data(), outbuf_.size());
- if (res <= 0) {
- return res;
- }
- if (static_cast<size_t>(res) > outbuf_.size()) {
- RTC_NOTREACHED();
- return -1;
- }
- size_t new_size = outbuf_.size() - res;
- if (new_size > 0) {
- memmove(outbuf_.data(), outbuf_.data() + res, new_size);
- }
- outbuf_.SetSize(new_size);
- return res;
-}
-
-void AsyncTCPSocketBase::AppendToOutBuffer(const void* pv, size_t cb) {
- RTC_DCHECK(outbuf_.size() + cb <= max_outsize_);
- RTC_DCHECK(!listen_);
- outbuf_.AppendData(static_cast<const uint8_t*>(pv), cb);
-}
-
-void AsyncTCPSocketBase::OnConnectEvent(AsyncSocket* socket) {
- SignalConnect(this);
-}
-
-void AsyncTCPSocketBase::OnReadEvent(AsyncSocket* socket) {
- RTC_DCHECK(socket_.get() == socket);
-
- if (listen_) {
- rtc::SocketAddress address;
- rtc::AsyncSocket* new_socket = socket->Accept(&address);
- if (!new_socket) {
- // TODO(stefan): Do something better like forwarding the error
- // to the user.
- LOG(LS_ERROR) << "TCP accept failed with error " << socket_->GetError();
- return;
- }
-
- HandleIncomingConnection(new_socket);
-
- // Prime a read event in case data is waiting.
- new_socket->SignalReadEvent(new_socket);
- } else {
- size_t total_recv = 0;
- while (true) {
- size_t free_size = inbuf_.capacity() - inbuf_.size();
- if (free_size < kMinimumRecvSize && inbuf_.capacity() < max_insize_) {
- inbuf_.EnsureCapacity(std::min(max_insize_, inbuf_.capacity() * 2));
- free_size = inbuf_.capacity() - inbuf_.size();
- }
-
- int len =
- socket_->Recv(inbuf_.data() + inbuf_.size(), free_size, nullptr);
- if (len < 0) {
- // TODO(stefan): Do something better like forwarding the error to the
- // user.
- if (!socket_->IsBlocking()) {
- LOG(LS_ERROR) << "Recv() returned error: " << socket_->GetError();
- }
- break;
- }
-
- total_recv += len;
- inbuf_.SetSize(inbuf_.size() + len);
- if (!len || static_cast<size_t>(len) < free_size) {
- break;
- }
- }
-
- if (!total_recv) {
- return;
- }
-
- size_t size = inbuf_.size();
- ProcessInput(inbuf_.data<char>(), &size);
-
- if (size > inbuf_.size()) {
- LOG(LS_ERROR) << "input buffer overflow";
- RTC_NOTREACHED();
- inbuf_.Clear();
- } else {
- inbuf_.SetSize(size);
- }
- }
-}
-
-void AsyncTCPSocketBase::OnWriteEvent(AsyncSocket* socket) {
- RTC_DCHECK(socket_.get() == socket);
-
- if (outbuf_.size() > 0) {
- FlushOutBuffer();
- }
-
- if (outbuf_.size() == 0) {
- SignalReadyToSend(this);
- }
-}
-
-void AsyncTCPSocketBase::OnCloseEvent(AsyncSocket* socket, int error) {
- SignalClose(this, error);
-}
-
-// AsyncTCPSocket
-// 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).
-AsyncTCPSocket* AsyncTCPSocket::Create(
- AsyncSocket* socket,
- const SocketAddress& bind_address,
- const SocketAddress& remote_address) {
- return new AsyncTCPSocket(AsyncTCPSocketBase::ConnectSocket(
- socket, bind_address, remote_address), false);
-}
-
-AsyncTCPSocket::AsyncTCPSocket(AsyncSocket* socket, bool listen)
- : AsyncTCPSocketBase(socket, listen, kBufSize) {
-}
-
-int AsyncTCPSocket::Send(const void *pv, size_t cb,
- const rtc::PacketOptions& options) {
- if (cb > kBufSize) {
- SetError(EMSGSIZE);
- return -1;
- }
-
- // If we are blocking on send, then silently drop this packet
- if (!IsOutBufferEmpty())
- return static_cast<int>(cb);
-
- PacketLength pkt_len = HostToNetwork16(static_cast<PacketLength>(cb));
- AppendToOutBuffer(&pkt_len, kPacketLenSize);
- AppendToOutBuffer(pv, cb);
-
- int res = FlushOutBuffer();
- if (res <= 0) {
- // drop packet if we made no progress
- ClearOutBuffer();
- return res;
- }
-
- rtc::SentPacket sent_packet(options.packet_id, rtc::TimeMillis());
- SignalSentPacket(this, sent_packet);
-
- // We claim to have sent the whole thing, even if we only sent partial
- return static_cast<int>(cb);
-}
-
-void AsyncTCPSocket::ProcessInput(char * data, size_t* len) {
- SocketAddress remote_addr(GetRemoteAddress());
-
- while (true) {
- if (*len < kPacketLenSize)
- return;
-
- PacketLength pkt_len = rtc::GetBE16(data);
- if (*len < kPacketLenSize + pkt_len)
- return;
-
- SignalReadPacket(this, data + kPacketLenSize, pkt_len, remote_addr,
- CreatePacketTime(0));
-
- *len -= kPacketLenSize + pkt_len;
- if (*len > 0) {
- memmove(data, data + kPacketLenSize + pkt_len, *len);
- }
- }
-}
-
-void AsyncTCPSocket::HandleIncomingConnection(AsyncSocket* socket) {
- SignalNewConnection(this, new AsyncTCPSocket(socket, false));
-}
-
-} // namespace rtc
« no previous file with comments | « webrtc/base/asynctcpsocket.h ('k') | webrtc/base/asynctcpsocket_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698