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

Unified Diff: net/socket/tcp_socket_win.cc

Issue 992733002: Remove //net (except for Android test stuff) and sdch (Closed) Base URL: git@github.com:domokit/mojo.git@master
Patch Set: Created 5 years, 9 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 | « net/socket/tcp_socket_win.h ('k') | net/socket/transport_client_socket_pool.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: net/socket/tcp_socket_win.cc
diff --git a/net/socket/tcp_socket_win.cc b/net/socket/tcp_socket_win.cc
deleted file mode 100644
index d9ec0d54ca6eb8d27f85b6d386fd6f4b15cdb736..0000000000000000000000000000000000000000
--- a/net/socket/tcp_socket_win.cc
+++ /dev/null
@@ -1,1095 +0,0 @@
-// Copyright 2013 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_socket.h"
-#include "net/socket/tcp_socket_win.h"
-
-#include <mstcpip.h>
-
-#include "base/callback_helpers.h"
-#include "base/logging.h"
-#include "base/profiler/scoped_tracker.h"
-#include "base/win/windows_version.h"
-#include "net/base/address_list.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_util.h"
-#include "net/base/network_activity_monitor.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;
-
-int SetSocketReceiveBufferSize(SOCKET socket, int32 size) {
- int rv = setsockopt(socket, SOL_SOCKET, SO_RCVBUF,
- reinterpret_cast<const char*>(&size), sizeof(size));
- int net_error = (rv == 0) ? OK : MapSystemError(WSAGetLastError());
- DCHECK(!rv) << "Could not set socket receive buffer size: " << net_error;
- return net_error;
-}
-
-int SetSocketSendBufferSize(SOCKET socket, int32 size) {
- int rv = setsockopt(socket, SOL_SOCKET, SO_SNDBUF,
- reinterpret_cast<const char*>(&size), sizeof(size));
- int net_error = (rv == 0) ? OK : MapSystemError(WSAGetLastError());
- DCHECK(!rv) << "Could not set socket send buffer size: " << net_error;
- return net_error;
-}
-
-// 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;
-}
-
-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
-
-//-----------------------------------------------------------------------------
-
-// Nothing to do for Windows since it doesn't support TCP FastOpen.
-// TODO(jri): Remove these along with the corresponding global variables.
-bool IsTCPFastOpenSupported() { return false; }
-bool IsTCPFastOpenUserEnabled() { return false; }
-void CheckSupportAndMaybeEnableTCPFastOpen(bool user_enabled) {}
-
-// 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 TCPSocketWin 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 TCPSocketWin::Core : public base::RefCounted<Core> {
- public:
- explicit Core(TCPSocketWin* socket);
-
- // Start watching for the end of a read or write operation.
- void WatchForRead();
- void WatchForWrite();
-
- // The TCPSocketWin 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.
- TCPSocketWin* 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);
-};
-
-TCPSocketWin::Core::Core(TCPSocketWin* 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();
-}
-
-TCPSocketWin::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 TCPSocketWin::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 TCPSocketWin::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 TCPSocketWin::Core::ReadDelegate::OnObjectSignaled(HANDLE object) {
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/418183 is fixed.
- tracked_objects::ScopedTracker tracking_profile(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "418183 TCPSocketWin::Core::ReadDelegate::OnObjectSignaled"));
-
- 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 TCPSocketWin::Core::WriteDelegate::OnObjectSignaled(
- HANDLE object) {
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/418183 is fixed.
- tracked_objects::ScopedTracker tracking_profile(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "418183 TCPSocketWin::Core::WriteDelegate::OnObjectSignaled"));
-
- DCHECK_EQ(object, core_->write_overlapped_.hEvent);
- if (core_->socket_)
- core_->socket_->DidCompleteWrite();
-
- core_->Release();
-}
-
-//-----------------------------------------------------------------------------
-
-TCPSocketWin::TCPSocketWin(net::NetLog* net_log,
- const net::NetLog::Source& source)
- : socket_(INVALID_SOCKET),
- accept_event_(WSA_INVALID_EVENT),
- accept_socket_(NULL),
- accept_address_(NULL),
- waiting_connect_(false),
- waiting_read_(false),
- waiting_write_(false),
- connect_os_error_(0),
- logging_multiple_connect_attempts_(false),
- net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_SOCKET)) {
- net_log_.BeginEvent(NetLog::TYPE_SOCKET_ALIVE,
- source.ToEventParametersCallback());
- EnsureWinsockInit();
-}
-
-TCPSocketWin::~TCPSocketWin() {
- Close();
- net_log_.EndEvent(NetLog::TYPE_SOCKET_ALIVE);
-}
-
-int TCPSocketWin::Open(AddressFamily family) {
- DCHECK(CalledOnValidThread());
- DCHECK_EQ(socket_, INVALID_SOCKET);
-
- socket_ = CreatePlatformSocket(ConvertAddressFamily(family), SOCK_STREAM,
- IPPROTO_TCP);
- if (socket_ == INVALID_SOCKET) {
- PLOG(ERROR) << "CreatePlatformSocket() returned an error";
- return MapSystemError(WSAGetLastError());
- }
-
- if (SetNonBlocking(socket_)) {
- int result = MapSystemError(WSAGetLastError());
- Close();
- return result;
- }
-
- return OK;
-}
-
-int TCPSocketWin::AdoptConnectedSocket(SOCKET socket,
- const IPEndPoint& peer_address) {
- DCHECK(CalledOnValidThread());
- DCHECK_EQ(socket_, INVALID_SOCKET);
- DCHECK(!core_.get());
-
- socket_ = socket;
-
- if (SetNonBlocking(socket_)) {
- int result = MapSystemError(WSAGetLastError());
- Close();
- return result;
- }
-
- core_ = new Core(this);
- peer_address_.reset(new IPEndPoint(peer_address));
-
- return OK;
-}
-
-int TCPSocketWin::AdoptListenSocket(SOCKET socket) {
- DCHECK(CalledOnValidThread());
- DCHECK_EQ(socket_, INVALID_SOCKET);
-
- socket_ = socket;
-
- if (SetNonBlocking(socket_)) {
- int result = MapSystemError(WSAGetLastError());
- Close();
- return result;
- }
-
- // |core_| is not needed for sockets that are used to accept connections.
- // The operation here is more like Open but with an existing socket.
-
- return OK;
-}
-
-int TCPSocketWin::Bind(const IPEndPoint& address) {
- DCHECK(CalledOnValidThread());
- DCHECK_NE(socket_, INVALID_SOCKET);
-
- SockaddrStorage storage;
- if (!address.ToSockAddr(storage.addr, &storage.addr_len))
- return ERR_ADDRESS_INVALID;
-
- int result = bind(socket_, storage.addr, storage.addr_len);
- if (result < 0) {
- PLOG(ERROR) << "bind() returned an error";
- return MapSystemError(WSAGetLastError());
- }
-
- return OK;
-}
-
-int TCPSocketWin::Listen(int backlog) {
- DCHECK(CalledOnValidThread());
- DCHECK_GT(backlog, 0);
- DCHECK_NE(socket_, INVALID_SOCKET);
- DCHECK_EQ(accept_event_, WSA_INVALID_EVENT);
-
- accept_event_ = WSACreateEvent();
- if (accept_event_ == WSA_INVALID_EVENT) {
- PLOG(ERROR) << "WSACreateEvent()";
- return MapSystemError(WSAGetLastError());
- }
-
- int result = listen(socket_, backlog);
- if (result < 0) {
- PLOG(ERROR) << "listen() returned an error";
- return MapSystemError(WSAGetLastError());
- }
-
- return OK;
-}
-
-int TCPSocketWin::Accept(scoped_ptr<TCPSocketWin>* socket,
- IPEndPoint* address,
- const CompletionCallback& callback) {
- DCHECK(CalledOnValidThread());
- DCHECK(socket);
- DCHECK(address);
- DCHECK(!callback.is_null());
- DCHECK(accept_callback_.is_null());
-
- net_log_.BeginEvent(NetLog::TYPE_TCP_ACCEPT);
-
- int result = AcceptInternal(socket, address);
-
- if (result == ERR_IO_PENDING) {
- // Start watching.
- WSAEventSelect(socket_, accept_event_, FD_ACCEPT);
- accept_watcher_.StartWatching(accept_event_, this);
-
- accept_socket_ = socket;
- accept_address_ = address;
- accept_callback_ = callback;
- }
-
- return result;
-}
-
-int TCPSocketWin::Connect(const IPEndPoint& address,
- const CompletionCallback& callback) {
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed.
- tracked_objects::ScopedTracker tracking_profile(
- FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPSocketWin::Connect"));
-
- DCHECK(CalledOnValidThread());
- DCHECK_NE(socket_, INVALID_SOCKET);
- DCHECK(!waiting_connect_);
-
- // |peer_address_| and |core_| will be non-NULL if Connect() has been called.
- // Unless Close() is called to reset the internal state, a second call to
- // Connect() is not allowed.
- // Please note that we enforce this even if the previous Connect() has
- // completed and failed. Although it is allowed to connect the same |socket_|
- // again after a connection attempt failed on Windows, it results in
- // unspecified behavior according to POSIX. Therefore, we make it behave in
- // the same way as TCPSocketLibevent.
- DCHECK(!peer_address_ && !core_.get());
-
- if (!logging_multiple_connect_attempts_)
- LogConnectBegin(AddressList(address));
-
- peer_address_.reset(new IPEndPoint(address));
-
- int rv = DoConnect();
- if (rv == ERR_IO_PENDING) {
- // Synchronous operation not supported.
- DCHECK(!callback.is_null());
- read_callback_ = callback;
- waiting_connect_ = true;
- } else {
- DoConnectComplete(rv);
- }
-
- return rv;
-}
-
-bool TCPSocketWin::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 TCPSocketWin::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 TCPSocketWin::Read(IOBuffer* buf,
- int buf_len,
- const CompletionCallback& callback) {
- DCHECK(CalledOnValidThread());
- DCHECK_NE(socket_, INVALID_SOCKET);
- DCHECK(!waiting_read_);
- CHECK(read_callback_.is_null());
- DCHECK(!core_->read_iobuffer_.get());
-
- return DoRead(buf, buf_len, callback);
-}
-
-int TCPSocketWin::Write(IOBuffer* buf,
- int buf_len,
- const CompletionCallback& callback) {
- DCHECK(CalledOnValidThread());
- DCHECK_NE(socket_, INVALID_SOCKET);
- DCHECK(!waiting_write_);
- CHECK(write_callback_.is_null());
- DCHECK_GT(buf_len, 0);
- DCHECK(!core_->write_iobuffer_.get());
-
- 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;
- }
- net_log_.AddByteTransferEvent(NetLog::TYPE_SOCKET_BYTES_SENT, rv,
- buf->data());
- NetworkActivityMonitor::GetInstance()->IncrementBytesSent(rv);
- 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;
-}
-
-int TCPSocketWin::GetLocalAddress(IPEndPoint* address) const {
- DCHECK(CalledOnValidThread());
- DCHECK(address);
-
- SockaddrStorage storage;
- if (getsockname(socket_, storage.addr, &storage.addr_len))
- return MapSystemError(WSAGetLastError());
- if (!address->FromSockAddr(storage.addr, storage.addr_len))
- return ERR_ADDRESS_INVALID;
-
- return OK;
-}
-
-int TCPSocketWin::GetPeerAddress(IPEndPoint* address) const {
- DCHECK(CalledOnValidThread());
- DCHECK(address);
- if (!IsConnected())
- return ERR_SOCKET_NOT_CONNECTED;
- *address = *peer_address_;
- return OK;
-}
-
-int TCPSocketWin::SetDefaultOptionsForServer() {
- return SetExclusiveAddrUse();
-}
-
-void TCPSocketWin::SetDefaultOptionsForClient() {
- // 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);
-}
-
-int TCPSocketWin::SetExclusiveAddrUse() {
- // On Windows, a bound end point can be hijacked by another process by
- // setting SO_REUSEADDR. Therefore a Windows-only option SO_EXCLUSIVEADDRUSE
- // was introduced in Windows NT 4.0 SP4. If the socket that is bound to the
- // end point has SO_EXCLUSIVEADDRUSE enabled, it is not possible for another
- // socket to forcibly bind to the end point until the end point is unbound.
- // It is recommend that all server applications must use SO_EXCLUSIVEADDRUSE.
- // MSDN: http://goo.gl/M6fjQ.
- //
- // Unlike on *nix, on Windows a TCP server socket can always bind to an end
- // point in TIME_WAIT state without setting SO_REUSEADDR, therefore it is not
- // needed here.
- //
- // SO_EXCLUSIVEADDRUSE will prevent a TCP client socket from binding to an end
- // point in TIME_WAIT status. It does not have this effect for a TCP server
- // socket.
-
- BOOL true_value = 1;
- int rv = setsockopt(socket_, SOL_SOCKET, SO_EXCLUSIVEADDRUSE,
- reinterpret_cast<const char*>(&true_value),
- sizeof(true_value));
- if (rv < 0)
- return MapSystemError(errno);
- return OK;
-}
-
-int TCPSocketWin::SetReceiveBufferSize(int32 size) {
- DCHECK(CalledOnValidThread());
- return SetSocketReceiveBufferSize(socket_, size);
-}
-
-int TCPSocketWin::SetSendBufferSize(int32 size) {
- DCHECK(CalledOnValidThread());
- return SetSocketSendBufferSize(socket_, size);
-}
-
-bool TCPSocketWin::SetKeepAlive(bool enable, int delay) {
- return SetTCPKeepAlive(socket_, enable, delay);
-}
-
-bool TCPSocketWin::SetNoDelay(bool no_delay) {
- return DisableNagle(socket_, no_delay);
-}
-
-void TCPSocketWin::Close() {
- DCHECK(CalledOnValidThread());
-
- if (socket_ != INVALID_SOCKET) {
- // Only log the close event if there's actually a socket to close.
- net_log_.AddEvent(NetLog::EventType::TYPE_SOCKET_CLOSED);
-
- // 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.
- if (closesocket(socket_) < 0)
- PLOG(ERROR) << "closesocket";
- socket_ = INVALID_SOCKET;
- }
-
- if (!accept_callback_.is_null()) {
- accept_watcher_.StopWatching();
- accept_socket_ = NULL;
- accept_address_ = NULL;
- accept_callback_.Reset();
- }
-
- if (accept_event_) {
- WSACloseEvent(accept_event_);
- accept_event_ = WSA_INVALID_EVENT;
- }
-
- if (core_.get()) {
- 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();
- }
- core_->Detach();
- core_ = NULL;
- }
-
- waiting_connect_ = false;
- waiting_read_ = false;
- waiting_write_ = false;
-
- read_callback_.Reset();
- write_callback_.Reset();
- peer_address_.reset();
- connect_os_error_ = 0;
-}
-
-void TCPSocketWin::StartLoggingMultipleConnectAttempts(
- const AddressList& addresses) {
- if (!logging_multiple_connect_attempts_) {
- logging_multiple_connect_attempts_ = true;
- LogConnectBegin(addresses);
- } else {
- NOTREACHED();
- }
-}
-
-void TCPSocketWin::EndLoggingMultipleConnectAttempts(int net_error) {
- if (logging_multiple_connect_attempts_) {
- LogConnectEnd(net_error);
- logging_multiple_connect_attempts_ = false;
- } else {
- NOTREACHED();
- }
-}
-
-int TCPSocketWin::AcceptInternal(scoped_ptr<TCPSocketWin>* socket,
- IPEndPoint* address) {
- SockaddrStorage storage;
- int new_socket = accept(socket_, storage.addr, &storage.addr_len);
- if (new_socket < 0) {
- int net_error = MapSystemError(WSAGetLastError());
- if (net_error != ERR_IO_PENDING)
- net_log_.EndEventWithNetErrorCode(NetLog::TYPE_TCP_ACCEPT, net_error);
- return net_error;
- }
-
- IPEndPoint ip_end_point;
- if (!ip_end_point.FromSockAddr(storage.addr, storage.addr_len)) {
- NOTREACHED();
- if (closesocket(new_socket) < 0)
- PLOG(ERROR) << "closesocket";
- int net_error = ERR_ADDRESS_INVALID;
- net_log_.EndEventWithNetErrorCode(NetLog::TYPE_TCP_ACCEPT, net_error);
- return net_error;
- }
- scoped_ptr<TCPSocketWin> tcp_socket(new TCPSocketWin(
- net_log_.net_log(), net_log_.source()));
- int adopt_result = tcp_socket->AdoptConnectedSocket(new_socket, ip_end_point);
- if (adopt_result != OK) {
- net_log_.EndEventWithNetErrorCode(NetLog::TYPE_TCP_ACCEPT, adopt_result);
- return adopt_result;
- }
- *socket = tcp_socket.Pass();
- *address = ip_end_point;
- net_log_.EndEvent(NetLog::TYPE_TCP_ACCEPT,
- CreateNetLogIPEndPointCallback(&ip_end_point));
- return OK;
-}
-
-void TCPSocketWin::OnObjectSignaled(HANDLE object) {
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/418183 is fixed.
- tracked_objects::ScopedTracker tracking_profile(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "418383 TCPSocketWin::OnObjectSignaled"));
-
- WSANETWORKEVENTS ev;
- if (WSAEnumNetworkEvents(socket_, accept_event_, &ev) == SOCKET_ERROR) {
- PLOG(ERROR) << "WSAEnumNetworkEvents()";
- return;
- }
-
- if (ev.lNetworkEvents & FD_ACCEPT) {
- int result = AcceptInternal(accept_socket_, accept_address_);
- if (result != ERR_IO_PENDING) {
- accept_socket_ = NULL;
- accept_address_ = NULL;
- base::ResetAndReturn(&accept_callback_).Run(result);
- }
- } else {
- // This happens when a client opens a connection and closes it before we
- // have a chance to accept it.
- DCHECK(ev.lNetworkEvents == 0);
-
- // Start watching the next FD_ACCEPT event.
- WSAEventSelect(socket_, accept_event_, FD_ACCEPT);
- accept_watcher_.StartWatching(accept_event_, this);
- }
-}
-
-int TCPSocketWin::DoConnect() {
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed.
- tracked_objects::ScopedTracker tracking_profile(
- FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPSocketWin::DoConnect"));
-
- DCHECK_EQ(connect_os_error_, 0);
- DCHECK(!core_.get());
-
- net_log_.BeginEvent(NetLog::TYPE_TCP_CONNECT_ATTEMPT,
- CreateNetLogIPEndPointCallback(peer_address_.get()));
-
- core_ = new Core(this);
-
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed.
- tracked_objects::ScopedTracker tracking_profile1(
- FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPSocketWin::DoConnect1"));
-
- // 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 (!peer_address_->ToSockAddr(storage.addr, &storage.addr_len))
- return ERR_ADDRESS_INVALID;
-
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed.
- tracked_objects::ScopedTracker tracking_profile2(
- FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPSocketWin::DoConnect2"));
-
- if (!connect(socket_, storage.addr, storage.addr_len)) {
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed.
- tracked_objects::ScopedTracker tracking_profile3(
- FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPSocketWin::DoConnect3"));
-
- // 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();
-
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed.
- tracked_objects::ScopedTracker tracking_profile4(
- FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPSocketWin::DoConnect4"));
-
- if (os_error != WSAEWOULDBLOCK) {
- LOG(ERROR) << "connect failed: " << os_error;
- connect_os_error_ = os_error;
- int rv = MapConnectError(os_error);
- CHECK_NE(ERR_IO_PENDING, rv);
- return rv;
- }
- }
-
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed.
- tracked_objects::ScopedTracker tracking_profile5(
- FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPSocketWin::DoConnect5"));
-
- core_->WatchForRead();
- return ERR_IO_PENDING;
-}
-
-void TCPSocketWin::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 (!logging_multiple_connect_attempts_)
- LogConnectEnd(result);
-}
-
-void TCPSocketWin::LogConnectBegin(const AddressList& addresses) {
- net_log_.BeginEvent(NetLog::TYPE_TCP_CONNECT,
- addresses.CreateNetLogCallback());
-}
-
-void TCPSocketWin::LogConnectEnd(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 TCPSocketWin::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 {
- net_log_.AddByteTransferEvent(NetLog::TYPE_SOCKET_BYTES_RECEIVED, rv,
- buf->data());
- NetworkActivityMonitor::GetInstance()->IncrementBytesReceived(rv);
- 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 TCPSocketWin::DidCompleteConnect() {
- DCHECK(waiting_connect_);
- DCHECK(!read_callback_.is_null());
- int result;
-
- // TODO(pkasting): Remove ScopedTracker below once crbug.com/418183 is fixed.
- tracked_objects::ScopedTracker tracking_profile1(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "418183 TCPSocketWin::DidCompleteConnect1"));
- 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) {
- // TODO(pkasting): Remove ScopedTracker below once crbug.com/418183 is
- // fixed.
- tracked_objects::ScopedTracker tracking_profile2(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "418183 TCPSocketWin::DidCompleteConnect2"));
- os_error = events.iErrorCode[FD_CONNECT_BIT];
- result = MapConnectError(os_error);
- } else {
- NOTREACHED();
- result = ERR_UNEXPECTED;
- }
-
- // TODO(pkasting): Remove ScopedTracker below once crbug.com/418183 is fixed.
- tracked_objects::ScopedTracker tracking_profile3(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "418183 TCPSocketWin::DidCompleteConnect3"));
- connect_os_error_ = os_error;
- DoConnectComplete(result);
- waiting_connect_ = false;
-
- // TODO(pkasting): Remove ScopedTracker below once crbug.com/418183 is fixed.
- tracked_objects::ScopedTracker tracking_profile4(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "418183 TCPSocketWin::DidCompleteConnect4"));
- DCHECK_NE(result, ERR_IO_PENDING);
- base::ResetAndReturn(&read_callback_).Run(result);
-}
-
-void TCPSocketWin::DidCompleteWrite() {
- DCHECK(waiting_write_);
- DCHECK(!write_callback_.is_null());
-
- 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 {
- net_log_.AddByteTransferEvent(NetLog::TYPE_SOCKET_BYTES_SENT, num_bytes,
- core_->write_iobuffer_->data());
- NetworkActivityMonitor::GetInstance()->IncrementBytesSent(num_bytes);
- }
- }
-
- core_->write_iobuffer_ = NULL;
-
- DCHECK_NE(rv, ERR_IO_PENDING);
- base::ResetAndReturn(&write_callback_).Run(rv);
-}
-
-void TCPSocketWin::DidSignalRead() {
- DCHECK(waiting_read_);
- DCHECK(!read_callback_.is_null());
-
- // TODO(pkasting): Remove ScopedTracker below once crbug.com/418183 is fixed.
- tracked_objects::ScopedTracker tracking_profile1(
- FROM_HERE_WITH_EXPLICIT_FUNCTION("418183 TCPSocketWin::DidSignalRead1"));
- 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) {
- // TODO(pkasting): Remove ScopedTracker below once crbug.com/418183 is
- // fixed.
- tracked_objects::ScopedTracker tracking_profile2(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "418183 TCPSocketWin::DidSignalRead2"));
- 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_.get(), core_->read_buffer_length_,
- read_callback_);
- if (rv == ERR_IO_PENDING)
- return;
- } else {
- // TODO(pkasting): Remove ScopedTracker below once crbug.com/418183 is
- // fixed.
- tracked_objects::ScopedTracker tracking_profile3(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "418183 TCPSocketWin::DidSignalRead3"));
- // 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;
-
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/418183 is fixed.
- tracked_objects::ScopedTracker tracking_profile4(
- FROM_HERE_WITH_EXPLICIT_FUNCTION("418183 TCPSocketWin::DidSignalRead4"));
- DCHECK_NE(rv, ERR_IO_PENDING);
- base::ResetAndReturn(&read_callback_).Run(rv);
-}
-
-} // namespace net
« no previous file with comments | « net/socket/tcp_socket_win.h ('k') | net/socket/transport_client_socket_pool.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698