| Index: net/socket/tcp_client_socket_win.cc
|
| diff --git a/net/socket/tcp_client_socket_win.cc b/net/socket/tcp_client_socket_win.cc
|
| deleted file mode 100644
|
| index f1334e78a468b144e8783e9fbae5a812dba50af4..0000000000000000000000000000000000000000
|
| --- a/net/socket/tcp_client_socket_win.cc
|
| +++ /dev/null
|
| @@ -1,956 +0,0 @@
|
| -// Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
| -// Use of this source code is governed by a BSD-style license that can be
|
| -// found in the LICENSE file.
|
| -
|
| -#include "net/socket/tcp_client_socket_win.h"
|
| -
|
| -#include <mstcpip.h>
|
| -
|
| -#include "base/basictypes.h"
|
| -#include "base/compiler_specific.h"
|
| -#include "base/metrics/stats_counters.h"
|
| -#include "base/strings/string_util.h"
|
| -#include "base/win/object_watcher.h"
|
| -#include "base/win/windows_version.h"
|
| -#include "net/base/connection_type_histograms.h"
|
| -#include "net/base/io_buffer.h"
|
| -#include "net/base/ip_endpoint.h"
|
| -#include "net/base/net_errors.h"
|
| -#include "net/base/net_log.h"
|
| -#include "net/base/net_util.h"
|
| -#include "net/base/network_change_notifier.h"
|
| -#include "net/base/winsock_init.h"
|
| -#include "net/base/winsock_util.h"
|
| -#include "net/socket/socket_descriptor.h"
|
| -#include "net/socket/socket_net_log_params.h"
|
| -
|
| -namespace net {
|
| -
|
| -namespace {
|
| -
|
| -const int kTCPKeepAliveSeconds = 45;
|
| -bool g_disable_overlapped_reads = false;
|
| -
|
| -bool SetSocketReceiveBufferSize(SOCKET socket, int32 size) {
|
| - int rv = setsockopt(socket, SOL_SOCKET, SO_RCVBUF,
|
| - reinterpret_cast<const char*>(&size), sizeof(size));
|
| - DCHECK(!rv) << "Could not set socket receive buffer size: " << GetLastError();
|
| - return rv == 0;
|
| -}
|
| -
|
| -bool SetSocketSendBufferSize(SOCKET socket, int32 size) {
|
| - int rv = setsockopt(socket, SOL_SOCKET, SO_SNDBUF,
|
| - reinterpret_cast<const char*>(&size), sizeof(size));
|
| - DCHECK(!rv) << "Could not set socket send buffer size: " << GetLastError();
|
| - return rv == 0;
|
| -}
|
| -
|
| -// Disable Nagle.
|
| -// The Nagle implementation on windows is governed by RFC 896. The idea
|
| -// behind Nagle is to reduce small packets on the network. When Nagle is
|
| -// enabled, if a partial packet has been sent, the TCP stack will disallow
|
| -// further *partial* packets until an ACK has been received from the other
|
| -// side. Good applications should always strive to send as much data as
|
| -// possible and avoid partial-packet sends. However, in most real world
|
| -// applications, there are edge cases where this does not happen, and two
|
| -// partial packets may be sent back to back. For a browser, it is NEVER
|
| -// a benefit to delay for an RTT before the second packet is sent.
|
| -//
|
| -// As a practical example in Chromium today, consider the case of a small
|
| -// POST. I have verified this:
|
| -// Client writes 649 bytes of header (partial packet #1)
|
| -// Client writes 50 bytes of POST data (partial packet #2)
|
| -// In the above example, with Nagle, a RTT delay is inserted between these
|
| -// two sends due to nagle. RTTs can easily be 100ms or more. The best
|
| -// fix is to make sure that for POSTing data, we write as much data as
|
| -// possible and minimize partial packets. We will fix that. But disabling
|
| -// Nagle also ensure we don't run into this delay in other edge cases.
|
| -// See also:
|
| -// http://technet.microsoft.com/en-us/library/bb726981.aspx
|
| -bool DisableNagle(SOCKET socket, bool disable) {
|
| - BOOL val = disable ? TRUE : FALSE;
|
| - int rv = setsockopt(socket, IPPROTO_TCP, TCP_NODELAY,
|
| - reinterpret_cast<const char*>(&val),
|
| - sizeof(val));
|
| - DCHECK(!rv) << "Could not disable nagle";
|
| - return rv == 0;
|
| -}
|
| -
|
| -// Enable TCP Keep-Alive to prevent NAT routers from timing out TCP
|
| -// connections. See http://crbug.com/27400 for details.
|
| -bool SetTCPKeepAlive(SOCKET socket, BOOL enable, int delay_secs) {
|
| - int delay = delay_secs * 1000;
|
| - struct tcp_keepalive keepalive_vals = {
|
| - enable ? 1 : 0, // TCP keep-alive on.
|
| - delay, // Delay seconds before sending first TCP keep-alive packet.
|
| - delay, // Delay seconds between sending TCP keep-alive packets.
|
| - };
|
| - DWORD bytes_returned = 0xABAB;
|
| - int rv = WSAIoctl(socket, SIO_KEEPALIVE_VALS, &keepalive_vals,
|
| - sizeof(keepalive_vals), NULL, 0,
|
| - &bytes_returned, NULL, NULL);
|
| - DCHECK(!rv) << "Could not enable TCP Keep-Alive for socket: " << socket
|
| - << " [error: " << WSAGetLastError() << "].";
|
| -
|
| - // Disregard any failure in disabling nagle or enabling TCP Keep-Alive.
|
| - return rv == 0;
|
| -}
|
| -
|
| -// Sets socket parameters. Returns the OS error code (or 0 on
|
| -// success).
|
| -int SetupSocket(SOCKET socket) {
|
| - // Increase the socket buffer sizes from the default sizes for WinXP. In
|
| - // performance testing, there is substantial benefit by increasing from 8KB
|
| - // to 64KB.
|
| - // See also:
|
| - // http://support.microsoft.com/kb/823764/EN-US
|
| - // On Vista, if we manually set these sizes, Vista turns off its receive
|
| - // window auto-tuning feature.
|
| - // http://blogs.msdn.com/wndp/archive/2006/05/05/Winhec-blog-tcpip-2.aspx
|
| - // Since Vista's auto-tune is better than any static value we can could set,
|
| - // only change these on pre-vista machines.
|
| - if (base::win::GetVersion() < base::win::VERSION_VISTA) {
|
| - const int32 kSocketBufferSize = 64 * 1024;
|
| - SetSocketReceiveBufferSize(socket, kSocketBufferSize);
|
| - SetSocketSendBufferSize(socket, kSocketBufferSize);
|
| - }
|
| -
|
| - DisableNagle(socket, true);
|
| - SetTCPKeepAlive(socket, true, kTCPKeepAliveSeconds);
|
| - return 0;
|
| -}
|
| -
|
| -// Creates a new socket and sets default parameters for it. Returns
|
| -// the OS error code (or 0 on success).
|
| -int CreateSocket(int family, SOCKET* socket) {
|
| - *socket = CreatePlatformSocket(family, SOCK_STREAM, IPPROTO_TCP);
|
| - if (*socket == INVALID_SOCKET) {
|
| - int os_error = WSAGetLastError();
|
| - LOG(ERROR) << "CreatePlatformSocket failed: " << os_error;
|
| - return os_error;
|
| - }
|
| - int error = SetupSocket(*socket);
|
| - if (error) {
|
| - if (closesocket(*socket) < 0)
|
| - PLOG(ERROR) << "closesocket";
|
| - *socket = INVALID_SOCKET;
|
| - return error;
|
| - }
|
| - return 0;
|
| -}
|
| -
|
| -int MapConnectError(int os_error) {
|
| - switch (os_error) {
|
| - // connect fails with WSAEACCES when Windows Firewall blocks the
|
| - // connection.
|
| - case WSAEACCES:
|
| - return ERR_NETWORK_ACCESS_DENIED;
|
| - case WSAETIMEDOUT:
|
| - return ERR_CONNECTION_TIMED_OUT;
|
| - default: {
|
| - int net_error = MapSystemError(os_error);
|
| - if (net_error == ERR_FAILED)
|
| - return ERR_CONNECTION_FAILED; // More specific than ERR_FAILED.
|
| -
|
| - // Give a more specific error when the user is offline.
|
| - if (net_error == ERR_ADDRESS_UNREACHABLE &&
|
| - NetworkChangeNotifier::IsOffline()) {
|
| - return ERR_INTERNET_DISCONNECTED;
|
| - }
|
| -
|
| - return net_error;
|
| - }
|
| - }
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -//-----------------------------------------------------------------------------
|
| -
|
| -// This class encapsulates all the state that has to be preserved as long as
|
| -// there is a network IO operation in progress. If the owner TCPClientSocketWin
|
| -// is destroyed while an operation is in progress, the Core is detached and it
|
| -// lives until the operation completes and the OS doesn't reference any resource
|
| -// declared on this class anymore.
|
| -class TCPClientSocketWin::Core : public base::RefCounted<Core> {
|
| - public:
|
| - explicit Core(TCPClientSocketWin* socket);
|
| -
|
| - // Start watching for the end of a read or write operation.
|
| - void WatchForRead();
|
| - void WatchForWrite();
|
| -
|
| - // The TCPClientSocketWin is going away.
|
| - void Detach() { socket_ = NULL; }
|
| -
|
| - // The separate OVERLAPPED variables for asynchronous operation.
|
| - // |read_overlapped_| is used for both Connect() and Read().
|
| - // |write_overlapped_| is only used for Write();
|
| - OVERLAPPED read_overlapped_;
|
| - OVERLAPPED write_overlapped_;
|
| -
|
| - // The buffers used in Read() and Write().
|
| - scoped_refptr<IOBuffer> read_iobuffer_;
|
| - scoped_refptr<IOBuffer> write_iobuffer_;
|
| - int read_buffer_length_;
|
| - int write_buffer_length_;
|
| -
|
| - bool non_blocking_reads_initialized_;
|
| -
|
| - private:
|
| - friend class base::RefCounted<Core>;
|
| -
|
| - class ReadDelegate : public base::win::ObjectWatcher::Delegate {
|
| - public:
|
| - explicit ReadDelegate(Core* core) : core_(core) {}
|
| - virtual ~ReadDelegate() {}
|
| -
|
| - // base::ObjectWatcher::Delegate methods:
|
| - virtual void OnObjectSignaled(HANDLE object);
|
| -
|
| - private:
|
| - Core* const core_;
|
| - };
|
| -
|
| - class WriteDelegate : public base::win::ObjectWatcher::Delegate {
|
| - public:
|
| - explicit WriteDelegate(Core* core) : core_(core) {}
|
| - virtual ~WriteDelegate() {}
|
| -
|
| - // base::ObjectWatcher::Delegate methods:
|
| - virtual void OnObjectSignaled(HANDLE object);
|
| -
|
| - private:
|
| - Core* const core_;
|
| - };
|
| -
|
| - ~Core();
|
| -
|
| - // The socket that created this object.
|
| - TCPClientSocketWin* socket_;
|
| -
|
| - // |reader_| handles the signals from |read_watcher_|.
|
| - ReadDelegate reader_;
|
| - // |writer_| handles the signals from |write_watcher_|.
|
| - WriteDelegate writer_;
|
| -
|
| - // |read_watcher_| watches for events from Connect() and Read().
|
| - base::win::ObjectWatcher read_watcher_;
|
| - // |write_watcher_| watches for events from Write();
|
| - base::win::ObjectWatcher write_watcher_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(Core);
|
| -};
|
| -
|
| -TCPClientSocketWin::Core::Core(
|
| - TCPClientSocketWin* socket)
|
| - : read_buffer_length_(0),
|
| - write_buffer_length_(0),
|
| - non_blocking_reads_initialized_(false),
|
| - socket_(socket),
|
| - reader_(this),
|
| - writer_(this) {
|
| - memset(&read_overlapped_, 0, sizeof(read_overlapped_));
|
| - memset(&write_overlapped_, 0, sizeof(write_overlapped_));
|
| -
|
| - read_overlapped_.hEvent = WSACreateEvent();
|
| - write_overlapped_.hEvent = WSACreateEvent();
|
| -}
|
| -
|
| -TCPClientSocketWin::Core::~Core() {
|
| - // Make sure the message loop is not watching this object anymore.
|
| - read_watcher_.StopWatching();
|
| - write_watcher_.StopWatching();
|
| -
|
| - WSACloseEvent(read_overlapped_.hEvent);
|
| - memset(&read_overlapped_, 0xaf, sizeof(read_overlapped_));
|
| - WSACloseEvent(write_overlapped_.hEvent);
|
| - memset(&write_overlapped_, 0xaf, sizeof(write_overlapped_));
|
| -}
|
| -
|
| -void TCPClientSocketWin::Core::WatchForRead() {
|
| - // We grab an extra reference because there is an IO operation in progress.
|
| - // Balanced in ReadDelegate::OnObjectSignaled().
|
| - AddRef();
|
| - read_watcher_.StartWatching(read_overlapped_.hEvent, &reader_);
|
| -}
|
| -
|
| -void TCPClientSocketWin::Core::WatchForWrite() {
|
| - // We grab an extra reference because there is an IO operation in progress.
|
| - // Balanced in WriteDelegate::OnObjectSignaled().
|
| - AddRef();
|
| - write_watcher_.StartWatching(write_overlapped_.hEvent, &writer_);
|
| -}
|
| -
|
| -void TCPClientSocketWin::Core::ReadDelegate::OnObjectSignaled(
|
| - HANDLE object) {
|
| - DCHECK_EQ(object, core_->read_overlapped_.hEvent);
|
| - if (core_->socket_) {
|
| - if (core_->socket_->waiting_connect())
|
| - core_->socket_->DidCompleteConnect();
|
| - else
|
| - core_->socket_->DidSignalRead();
|
| - }
|
| -
|
| - core_->Release();
|
| -}
|
| -
|
| -void TCPClientSocketWin::Core::WriteDelegate::OnObjectSignaled(
|
| - HANDLE object) {
|
| - DCHECK_EQ(object, core_->write_overlapped_.hEvent);
|
| - if (core_->socket_)
|
| - core_->socket_->DidCompleteWrite();
|
| -
|
| - core_->Release();
|
| -}
|
| -
|
| -//-----------------------------------------------------------------------------
|
| -
|
| -TCPClientSocketWin::TCPClientSocketWin(const AddressList& addresses,
|
| - net::NetLog* net_log,
|
| - const net::NetLog::Source& source)
|
| - : socket_(INVALID_SOCKET),
|
| - bound_socket_(INVALID_SOCKET),
|
| - addresses_(addresses),
|
| - current_address_index_(-1),
|
| - waiting_read_(false),
|
| - waiting_write_(false),
|
| - next_connect_state_(CONNECT_STATE_NONE),
|
| - connect_os_error_(0),
|
| - net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_SOCKET)),
|
| - previously_disconnected_(false) {
|
| - net_log_.BeginEvent(NetLog::TYPE_SOCKET_ALIVE,
|
| - source.ToEventParametersCallback());
|
| - EnsureWinsockInit();
|
| -}
|
| -
|
| -TCPClientSocketWin::~TCPClientSocketWin() {
|
| - Disconnect();
|
| - net_log_.EndEvent(NetLog::TYPE_SOCKET_ALIVE);
|
| -}
|
| -
|
| -int TCPClientSocketWin::AdoptSocket(SOCKET socket) {
|
| - DCHECK_EQ(socket_, INVALID_SOCKET);
|
| -
|
| - int error = SetupSocket(socket);
|
| - if (error)
|
| - return MapSystemError(error);
|
| -
|
| - socket_ = socket;
|
| - SetNonBlocking(socket_);
|
| -
|
| - core_ = new Core(this);
|
| - current_address_index_ = 0;
|
| - use_history_.set_was_ever_connected();
|
| -
|
| - return OK;
|
| -}
|
| -
|
| -int TCPClientSocketWin::Bind(const IPEndPoint& address) {
|
| - if (current_address_index_ >= 0 || bind_address_.get()) {
|
| - // Cannot bind the socket if we are already connected or connecting.
|
| - return ERR_UNEXPECTED;
|
| - }
|
| -
|
| - SockaddrStorage storage;
|
| - if (!address.ToSockAddr(storage.addr, &storage.addr_len))
|
| - return ERR_INVALID_ARGUMENT;
|
| -
|
| - // Create |bound_socket_| and try to bind it to |address|.
|
| - int error = CreateSocket(address.GetSockAddrFamily(), &bound_socket_);
|
| - if (error)
|
| - return MapSystemError(error);
|
| -
|
| - if (bind(bound_socket_, storage.addr, storage.addr_len)) {
|
| - error = errno;
|
| - if (closesocket(bound_socket_) < 0)
|
| - PLOG(ERROR) << "closesocket";
|
| - bound_socket_ = INVALID_SOCKET;
|
| - return MapSystemError(error);
|
| - }
|
| -
|
| - bind_address_.reset(new IPEndPoint(address));
|
| -
|
| - return 0;
|
| -}
|
| -
|
| -
|
| -int TCPClientSocketWin::Connect(const CompletionCallback& callback) {
|
| - DCHECK(CalledOnValidThread());
|
| -
|
| - // If already connected, then just return OK.
|
| - if (socket_ != INVALID_SOCKET)
|
| - return OK;
|
| -
|
| - base::StatsCounter connects("tcp.connect");
|
| - connects.Increment();
|
| -
|
| - net_log_.BeginEvent(NetLog::TYPE_TCP_CONNECT,
|
| - addresses_.CreateNetLogCallback());
|
| -
|
| - // We will try to connect to each address in addresses_. Start with the
|
| - // first one in the list.
|
| - next_connect_state_ = CONNECT_STATE_CONNECT;
|
| - current_address_index_ = 0;
|
| -
|
| - int rv = DoConnectLoop(OK);
|
| - if (rv == ERR_IO_PENDING) {
|
| - // Synchronous operation not supported.
|
| - DCHECK(!callback.is_null());
|
| - // TODO(ajwong): Is setting read_callback_ the right thing to do here??
|
| - read_callback_ = callback;
|
| - } else {
|
| - LogConnectCompletion(rv);
|
| - }
|
| -
|
| - return rv;
|
| -}
|
| -
|
| -int TCPClientSocketWin::DoConnectLoop(int result) {
|
| - DCHECK_NE(next_connect_state_, CONNECT_STATE_NONE);
|
| -
|
| - int rv = result;
|
| - do {
|
| - ConnectState state = next_connect_state_;
|
| - next_connect_state_ = CONNECT_STATE_NONE;
|
| - switch (state) {
|
| - case CONNECT_STATE_CONNECT:
|
| - DCHECK_EQ(OK, rv);
|
| - rv = DoConnect();
|
| - break;
|
| - case CONNECT_STATE_CONNECT_COMPLETE:
|
| - rv = DoConnectComplete(rv);
|
| - break;
|
| - default:
|
| - LOG(DFATAL) << "bad state " << state;
|
| - rv = ERR_UNEXPECTED;
|
| - break;
|
| - }
|
| - } while (rv != ERR_IO_PENDING && next_connect_state_ != CONNECT_STATE_NONE);
|
| -
|
| - return rv;
|
| -}
|
| -
|
| -int TCPClientSocketWin::DoConnect() {
|
| - DCHECK_GE(current_address_index_, 0);
|
| - DCHECK_LT(current_address_index_, static_cast<int>(addresses_.size()));
|
| - DCHECK_EQ(0, connect_os_error_);
|
| -
|
| - const IPEndPoint& endpoint = addresses_[current_address_index_];
|
| -
|
| - if (previously_disconnected_) {
|
| - use_history_.Reset();
|
| - previously_disconnected_ = false;
|
| - }
|
| -
|
| - net_log_.BeginEvent(NetLog::TYPE_TCP_CONNECT_ATTEMPT,
|
| - CreateNetLogIPEndPointCallback(&endpoint));
|
| -
|
| - next_connect_state_ = CONNECT_STATE_CONNECT_COMPLETE;
|
| -
|
| - if (bound_socket_ != INVALID_SOCKET) {
|
| - DCHECK(bind_address_.get());
|
| - socket_ = bound_socket_;
|
| - bound_socket_ = INVALID_SOCKET;
|
| - } else {
|
| - connect_os_error_ = CreateSocket(endpoint.GetSockAddrFamily(), &socket_);
|
| - if (connect_os_error_ != 0)
|
| - return MapSystemError(connect_os_error_);
|
| -
|
| - if (bind_address_.get()) {
|
| - SockaddrStorage storage;
|
| - if (!bind_address_->ToSockAddr(storage.addr, &storage.addr_len))
|
| - return ERR_INVALID_ARGUMENT;
|
| - if (bind(socket_, storage.addr, storage.addr_len))
|
| - return MapSystemError(errno);
|
| - }
|
| - }
|
| -
|
| - DCHECK(!core_);
|
| - core_ = new Core(this);
|
| - // WSAEventSelect sets the socket to non-blocking mode as a side effect.
|
| - // Our connect() and recv() calls require that the socket be non-blocking.
|
| - WSAEventSelect(socket_, core_->read_overlapped_.hEvent, FD_CONNECT);
|
| -
|
| - SockaddrStorage storage;
|
| - if (!endpoint.ToSockAddr(storage.addr, &storage.addr_len))
|
| - return ERR_INVALID_ARGUMENT;
|
| - if (!connect(socket_, storage.addr, storage.addr_len)) {
|
| - // Connected without waiting!
|
| - //
|
| - // The MSDN page for connect says:
|
| - // With a nonblocking socket, the connection attempt cannot be completed
|
| - // immediately. In this case, connect will return SOCKET_ERROR, and
|
| - // WSAGetLastError will return WSAEWOULDBLOCK.
|
| - // which implies that for a nonblocking socket, connect never returns 0.
|
| - // It's not documented whether the event object will be signaled or not
|
| - // if connect does return 0. So the code below is essentially dead code
|
| - // and we don't know if it's correct.
|
| - NOTREACHED();
|
| -
|
| - if (ResetEventIfSignaled(core_->read_overlapped_.hEvent))
|
| - return OK;
|
| - } else {
|
| - int os_error = WSAGetLastError();
|
| - if (os_error != WSAEWOULDBLOCK) {
|
| - LOG(ERROR) << "connect failed: " << os_error;
|
| - connect_os_error_ = os_error;
|
| - return MapConnectError(os_error);
|
| - }
|
| - }
|
| -
|
| - core_->WatchForRead();
|
| - return ERR_IO_PENDING;
|
| -}
|
| -
|
| -int TCPClientSocketWin::DoConnectComplete(int result) {
|
| - // Log the end of this attempt (and any OS error it threw).
|
| - int os_error = connect_os_error_;
|
| - connect_os_error_ = 0;
|
| - if (result != OK) {
|
| - net_log_.EndEvent(NetLog::TYPE_TCP_CONNECT_ATTEMPT,
|
| - NetLog::IntegerCallback("os_error", os_error));
|
| - } else {
|
| - net_log_.EndEvent(NetLog::TYPE_TCP_CONNECT_ATTEMPT);
|
| - }
|
| -
|
| - if (result == OK) {
|
| - use_history_.set_was_ever_connected();
|
| - return OK; // Done!
|
| - }
|
| -
|
| - // Close whatever partially connected socket we currently have.
|
| - DoDisconnect();
|
| -
|
| - // Try to fall back to the next address in the list.
|
| - if (current_address_index_ + 1 < static_cast<int>(addresses_.size())) {
|
| - next_connect_state_ = CONNECT_STATE_CONNECT;
|
| - ++current_address_index_;
|
| - return OK;
|
| - }
|
| -
|
| - // Otherwise there is nothing to fall back to, so give up.
|
| - return result;
|
| -}
|
| -
|
| -void TCPClientSocketWin::Disconnect() {
|
| - DCHECK(CalledOnValidThread());
|
| -
|
| - DoDisconnect();
|
| - current_address_index_ = -1;
|
| - bind_address_.reset();
|
| -}
|
| -
|
| -void TCPClientSocketWin::DoDisconnect() {
|
| - DCHECK(CalledOnValidThread());
|
| -
|
| - if (socket_ == INVALID_SOCKET)
|
| - return;
|
| -
|
| - // Note: don't use CancelIo to cancel pending IO because it doesn't work
|
| - // when there is a Winsock layered service provider.
|
| -
|
| - // In most socket implementations, closing a socket results in a graceful
|
| - // connection shutdown, but in Winsock we have to call shutdown explicitly.
|
| - // See the MSDN page "Graceful Shutdown, Linger Options, and Socket Closure"
|
| - // at http://msdn.microsoft.com/en-us/library/ms738547.aspx
|
| - shutdown(socket_, SD_SEND);
|
| -
|
| - // This cancels any pending IO.
|
| - closesocket(socket_);
|
| - socket_ = INVALID_SOCKET;
|
| -
|
| - if (waiting_connect()) {
|
| - // We closed the socket, so this notification will never come.
|
| - // From MSDN' WSAEventSelect documentation:
|
| - // "Closing a socket with closesocket also cancels the association and
|
| - // selection of network events specified in WSAEventSelect for the socket".
|
| - core_->Release();
|
| - }
|
| -
|
| - waiting_read_ = false;
|
| - waiting_write_ = false;
|
| -
|
| - core_->Detach();
|
| - core_ = NULL;
|
| -
|
| - previously_disconnected_ = true;
|
| -}
|
| -
|
| -bool TCPClientSocketWin::IsConnected() const {
|
| - DCHECK(CalledOnValidThread());
|
| -
|
| - if (socket_ == INVALID_SOCKET || waiting_connect())
|
| - return false;
|
| -
|
| - if (waiting_read_)
|
| - return true;
|
| -
|
| - // Check if connection is alive.
|
| - char c;
|
| - int rv = recv(socket_, &c, 1, MSG_PEEK);
|
| - if (rv == 0)
|
| - return false;
|
| - if (rv == SOCKET_ERROR && WSAGetLastError() != WSAEWOULDBLOCK)
|
| - return false;
|
| -
|
| - return true;
|
| -}
|
| -
|
| -bool TCPClientSocketWin::IsConnectedAndIdle() const {
|
| - DCHECK(CalledOnValidThread());
|
| -
|
| - if (socket_ == INVALID_SOCKET || waiting_connect())
|
| - return false;
|
| -
|
| - if (waiting_read_)
|
| - return true;
|
| -
|
| - // Check if connection is alive and we haven't received any data
|
| - // unexpectedly.
|
| - char c;
|
| - int rv = recv(socket_, &c, 1, MSG_PEEK);
|
| - if (rv >= 0)
|
| - return false;
|
| - if (WSAGetLastError() != WSAEWOULDBLOCK)
|
| - return false;
|
| -
|
| - return true;
|
| -}
|
| -
|
| -int TCPClientSocketWin::GetPeerAddress(IPEndPoint* address) const {
|
| - DCHECK(CalledOnValidThread());
|
| - DCHECK(address);
|
| - if (!IsConnected())
|
| - return ERR_SOCKET_NOT_CONNECTED;
|
| - *address = addresses_[current_address_index_];
|
| - return OK;
|
| -}
|
| -
|
| -int TCPClientSocketWin::GetLocalAddress(IPEndPoint* address) const {
|
| - DCHECK(CalledOnValidThread());
|
| - DCHECK(address);
|
| - if (socket_ == INVALID_SOCKET) {
|
| - if (bind_address_.get()) {
|
| - *address = *bind_address_;
|
| - return OK;
|
| - }
|
| - return ERR_SOCKET_NOT_CONNECTED;
|
| - }
|
| -
|
| - struct sockaddr_storage addr_storage;
|
| - socklen_t addr_len = sizeof(addr_storage);
|
| - struct sockaddr* addr = reinterpret_cast<struct sockaddr*>(&addr_storage);
|
| - if (getsockname(socket_, addr, &addr_len))
|
| - return MapSystemError(WSAGetLastError());
|
| - if (!address->FromSockAddr(addr, addr_len))
|
| - return ERR_FAILED;
|
| - return OK;
|
| -}
|
| -
|
| -void TCPClientSocketWin::SetSubresourceSpeculation() {
|
| - use_history_.set_subresource_speculation();
|
| -}
|
| -
|
| -void TCPClientSocketWin::SetOmniboxSpeculation() {
|
| - use_history_.set_omnibox_speculation();
|
| -}
|
| -
|
| -bool TCPClientSocketWin::WasEverUsed() const {
|
| - return use_history_.was_used_to_convey_data();
|
| -}
|
| -
|
| -bool TCPClientSocketWin::UsingTCPFastOpen() const {
|
| - // Not supported on windows.
|
| - return false;
|
| -}
|
| -
|
| -bool TCPClientSocketWin::WasNpnNegotiated() const {
|
| - return false;
|
| -}
|
| -
|
| -NextProto TCPClientSocketWin::GetNegotiatedProtocol() const {
|
| - return kProtoUnknown;
|
| -}
|
| -
|
| -bool TCPClientSocketWin::GetSSLInfo(SSLInfo* ssl_info) {
|
| - return false;
|
| -}
|
| -
|
| -int TCPClientSocketWin::Read(IOBuffer* buf,
|
| - int buf_len,
|
| - const CompletionCallback& callback) {
|
| - DCHECK(CalledOnValidThread());
|
| - DCHECK_NE(socket_, INVALID_SOCKET);
|
| - DCHECK(!waiting_read_);
|
| - DCHECK(read_callback_.is_null());
|
| - DCHECK(!core_->read_iobuffer_);
|
| -
|
| - return DoRead(buf, buf_len, callback);
|
| -}
|
| -
|
| -int TCPClientSocketWin::Write(IOBuffer* buf,
|
| - int buf_len,
|
| - const CompletionCallback& callback) {
|
| - DCHECK(CalledOnValidThread());
|
| - DCHECK_NE(socket_, INVALID_SOCKET);
|
| - DCHECK(!waiting_write_);
|
| - DCHECK(write_callback_.is_null());
|
| - DCHECK_GT(buf_len, 0);
|
| - DCHECK(!core_->write_iobuffer_);
|
| -
|
| - base::StatsCounter writes("tcp.writes");
|
| - writes.Increment();
|
| -
|
| - WSABUF write_buffer;
|
| - write_buffer.len = buf_len;
|
| - write_buffer.buf = buf->data();
|
| -
|
| - // TODO(wtc): Remove the assertion after enough testing.
|
| - AssertEventNotSignaled(core_->write_overlapped_.hEvent);
|
| - DWORD num;
|
| - int rv = WSASend(socket_, &write_buffer, 1, &num, 0,
|
| - &core_->write_overlapped_, NULL);
|
| - if (rv == 0) {
|
| - if (ResetEventIfSignaled(core_->write_overlapped_.hEvent)) {
|
| - rv = static_cast<int>(num);
|
| - if (rv > buf_len || rv < 0) {
|
| - // It seems that some winsock interceptors report that more was written
|
| - // than was available. Treat this as an error. http://crbug.com/27870
|
| - LOG(ERROR) << "Detected broken LSP: Asked to write " << buf_len
|
| - << " bytes, but " << rv << " bytes reported.";
|
| - return ERR_WINSOCK_UNEXPECTED_WRITTEN_BYTES;
|
| - }
|
| - base::StatsCounter write_bytes("tcp.write_bytes");
|
| - write_bytes.Add(rv);
|
| - if (rv > 0)
|
| - use_history_.set_was_used_to_convey_data();
|
| - net_log_.AddByteTransferEvent(NetLog::TYPE_SOCKET_BYTES_SENT, rv,
|
| - buf->data());
|
| - return rv;
|
| - }
|
| - } else {
|
| - int os_error = WSAGetLastError();
|
| - if (os_error != WSA_IO_PENDING) {
|
| - int net_error = MapSystemError(os_error);
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKET_WRITE_ERROR,
|
| - CreateNetLogSocketErrorCallback(net_error, os_error));
|
| - return net_error;
|
| - }
|
| - }
|
| - waiting_write_ = true;
|
| - write_callback_ = callback;
|
| - core_->write_iobuffer_ = buf;
|
| - core_->write_buffer_length_ = buf_len;
|
| - core_->WatchForWrite();
|
| - return ERR_IO_PENDING;
|
| -}
|
| -
|
| -bool TCPClientSocketWin::SetReceiveBufferSize(int32 size) {
|
| - DCHECK(CalledOnValidThread());
|
| - return SetSocketReceiveBufferSize(socket_, size);
|
| -}
|
| -
|
| -bool TCPClientSocketWin::SetSendBufferSize(int32 size) {
|
| - DCHECK(CalledOnValidThread());
|
| - return SetSocketSendBufferSize(socket_, size);
|
| -}
|
| -
|
| -bool TCPClientSocketWin::SetKeepAlive(bool enable, int delay) {
|
| - return SetTCPKeepAlive(socket_, enable, delay);
|
| -}
|
| -
|
| -bool TCPClientSocketWin::SetNoDelay(bool no_delay) {
|
| - return DisableNagle(socket_, no_delay);
|
| -}
|
| -
|
| -void TCPClientSocketWin::LogConnectCompletion(int net_error) {
|
| - if (net_error == OK)
|
| - UpdateConnectionTypeHistograms(CONNECTION_ANY);
|
| -
|
| - if (net_error != OK) {
|
| - net_log_.EndEventWithNetErrorCode(NetLog::TYPE_TCP_CONNECT, net_error);
|
| - return;
|
| - }
|
| -
|
| - struct sockaddr_storage source_address;
|
| - socklen_t addrlen = sizeof(source_address);
|
| - int rv = getsockname(
|
| - socket_, reinterpret_cast<struct sockaddr*>(&source_address), &addrlen);
|
| - if (rv != 0) {
|
| - LOG(ERROR) << "getsockname() [rv: " << rv
|
| - << "] error: " << WSAGetLastError();
|
| - NOTREACHED();
|
| - net_log_.EndEventWithNetErrorCode(NetLog::TYPE_TCP_CONNECT, rv);
|
| - return;
|
| - }
|
| -
|
| - net_log_.EndEvent(
|
| - NetLog::TYPE_TCP_CONNECT,
|
| - CreateNetLogSourceAddressCallback(
|
| - reinterpret_cast<const struct sockaddr*>(&source_address),
|
| - sizeof(source_address)));
|
| -}
|
| -
|
| -int TCPClientSocketWin::DoRead(IOBuffer* buf, int buf_len,
|
| - const CompletionCallback& callback) {
|
| - if (!core_->non_blocking_reads_initialized_) {
|
| - WSAEventSelect(socket_, core_->read_overlapped_.hEvent,
|
| - FD_READ | FD_CLOSE);
|
| - core_->non_blocking_reads_initialized_ = true;
|
| - }
|
| - int rv = recv(socket_, buf->data(), buf_len, 0);
|
| - if (rv == SOCKET_ERROR) {
|
| - int os_error = WSAGetLastError();
|
| - if (os_error != WSAEWOULDBLOCK) {
|
| - int net_error = MapSystemError(os_error);
|
| - net_log_.AddEvent(
|
| - NetLog::TYPE_SOCKET_READ_ERROR,
|
| - CreateNetLogSocketErrorCallback(net_error, os_error));
|
| - return net_error;
|
| - }
|
| - } else {
|
| - base::StatsCounter read_bytes("tcp.read_bytes");
|
| - if (rv > 0) {
|
| - use_history_.set_was_used_to_convey_data();
|
| - read_bytes.Add(rv);
|
| - }
|
| - net_log_.AddByteTransferEvent(NetLog::TYPE_SOCKET_BYTES_RECEIVED, rv,
|
| - buf->data());
|
| - return rv;
|
| - }
|
| -
|
| - waiting_read_ = true;
|
| - read_callback_ = callback;
|
| - core_->read_iobuffer_ = buf;
|
| - core_->read_buffer_length_ = buf_len;
|
| - core_->WatchForRead();
|
| - return ERR_IO_PENDING;
|
| -}
|
| -
|
| -void TCPClientSocketWin::DoReadCallback(int rv) {
|
| - DCHECK_NE(rv, ERR_IO_PENDING);
|
| - DCHECK(!read_callback_.is_null());
|
| -
|
| - // Since Run may result in Read being called, clear read_callback_ up front.
|
| - CompletionCallback c = read_callback_;
|
| - read_callback_.Reset();
|
| - c.Run(rv);
|
| -}
|
| -
|
| -void TCPClientSocketWin::DoWriteCallback(int rv) {
|
| - DCHECK_NE(rv, ERR_IO_PENDING);
|
| - DCHECK(!write_callback_.is_null());
|
| -
|
| - // Since Run may result in Write being called, clear write_callback_ up front.
|
| - CompletionCallback c = write_callback_;
|
| - write_callback_.Reset();
|
| - c.Run(rv);
|
| -}
|
| -
|
| -void TCPClientSocketWin::DidCompleteConnect() {
|
| - DCHECK_EQ(next_connect_state_, CONNECT_STATE_CONNECT_COMPLETE);
|
| - int result;
|
| -
|
| - WSANETWORKEVENTS events;
|
| - int rv = WSAEnumNetworkEvents(socket_, core_->read_overlapped_.hEvent,
|
| - &events);
|
| - int os_error = 0;
|
| - if (rv == SOCKET_ERROR) {
|
| - NOTREACHED();
|
| - os_error = WSAGetLastError();
|
| - result = MapSystemError(os_error);
|
| - } else if (events.lNetworkEvents & FD_CONNECT) {
|
| - os_error = events.iErrorCode[FD_CONNECT_BIT];
|
| - result = MapConnectError(os_error);
|
| - } else {
|
| - NOTREACHED();
|
| - result = ERR_UNEXPECTED;
|
| - }
|
| -
|
| - connect_os_error_ = os_error;
|
| - rv = DoConnectLoop(result);
|
| - if (rv != ERR_IO_PENDING) {
|
| - LogConnectCompletion(rv);
|
| - DoReadCallback(rv);
|
| - }
|
| -}
|
| -
|
| -void TCPClientSocketWin::DidCompleteWrite() {
|
| - DCHECK(waiting_write_);
|
| -
|
| - DWORD num_bytes, flags;
|
| - BOOL ok = WSAGetOverlappedResult(socket_, &core_->write_overlapped_,
|
| - &num_bytes, FALSE, &flags);
|
| - WSAResetEvent(core_->write_overlapped_.hEvent);
|
| - waiting_write_ = false;
|
| - int rv;
|
| - if (!ok) {
|
| - int os_error = WSAGetLastError();
|
| - rv = MapSystemError(os_error);
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKET_WRITE_ERROR,
|
| - CreateNetLogSocketErrorCallback(rv, os_error));
|
| - } else {
|
| - rv = static_cast<int>(num_bytes);
|
| - if (rv > core_->write_buffer_length_ || rv < 0) {
|
| - // It seems that some winsock interceptors report that more was written
|
| - // than was available. Treat this as an error. http://crbug.com/27870
|
| - LOG(ERROR) << "Detected broken LSP: Asked to write "
|
| - << core_->write_buffer_length_ << " bytes, but " << rv
|
| - << " bytes reported.";
|
| - rv = ERR_WINSOCK_UNEXPECTED_WRITTEN_BYTES;
|
| - } else {
|
| - base::StatsCounter write_bytes("tcp.write_bytes");
|
| - write_bytes.Add(num_bytes);
|
| - if (num_bytes > 0)
|
| - use_history_.set_was_used_to_convey_data();
|
| - net_log_.AddByteTransferEvent(NetLog::TYPE_SOCKET_BYTES_SENT, num_bytes,
|
| - core_->write_iobuffer_->data());
|
| - }
|
| - }
|
| - core_->write_iobuffer_ = NULL;
|
| - DoWriteCallback(rv);
|
| -}
|
| -
|
| -void TCPClientSocketWin::DidSignalRead() {
|
| - DCHECK(waiting_read_);
|
| - int os_error = 0;
|
| - WSANETWORKEVENTS network_events;
|
| - int rv = WSAEnumNetworkEvents(socket_, core_->read_overlapped_.hEvent,
|
| - &network_events);
|
| - if (rv == SOCKET_ERROR) {
|
| - os_error = WSAGetLastError();
|
| - rv = MapSystemError(os_error);
|
| - } else if (network_events.lNetworkEvents) {
|
| - DCHECK_EQ(network_events.lNetworkEvents & ~(FD_READ | FD_CLOSE), 0);
|
| - // If network_events.lNetworkEvents is FD_CLOSE and
|
| - // network_events.iErrorCode[FD_CLOSE_BIT] is 0, it is a graceful
|
| - // connection closure. It is tempting to directly set rv to 0 in
|
| - // this case, but the MSDN pages for WSAEventSelect and
|
| - // WSAAsyncSelect recommend we still call DoRead():
|
| - // FD_CLOSE should only be posted after all data is read from a
|
| - // socket, but an application should check for remaining data upon
|
| - // receipt of FD_CLOSE to avoid any possibility of losing data.
|
| - //
|
| - // If network_events.iErrorCode[FD_READ_BIT] or
|
| - // network_events.iErrorCode[FD_CLOSE_BIT] is nonzero, still call
|
| - // DoRead() because recv() reports a more accurate error code
|
| - // (WSAECONNRESET vs. WSAECONNABORTED) when the connection was
|
| - // reset.
|
| - rv = DoRead(core_->read_iobuffer_, core_->read_buffer_length_,
|
| - read_callback_);
|
| - if (rv == ERR_IO_PENDING)
|
| - return;
|
| - } else {
|
| - // This may happen because Read() may succeed synchronously and
|
| - // consume all the received data without resetting the event object.
|
| - core_->WatchForRead();
|
| - return;
|
| - }
|
| - waiting_read_ = false;
|
| - core_->read_iobuffer_ = NULL;
|
| - core_->read_buffer_length_ = 0;
|
| - DoReadCallback(rv);
|
| -}
|
| -
|
| -} // namespace net
|
|
|