| OLD | NEW |
| 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/socket/tcp_client_socket_win.h" | 5 #include "net/socket/tcp_client_socket_win.h" |
| 6 | 6 |
| 7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
| 8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
| 9 #include "base/memory_debug.h" | 9 #include "base/memory_debug.h" |
| 10 #include "base/stats_counters.h" | 10 #include "base/stats_counters.h" |
| 11 #include "base/string_util.h" | 11 #include "base/string_util.h" |
| 12 #include "base/sys_info.h" | 12 #include "base/sys_info.h" |
| 13 #include "base/trace_event.h" | 13 #include "base/trace_event.h" |
| 14 #include "net/base/connection_type_histograms.h" | 14 #include "net/base/connection_type_histograms.h" |
| 15 #include "net/base/io_buffer.h" | 15 #include "net/base/io_buffer.h" |
| 16 #include "net/base/load_log.h" | 16 #include "net/base/net_log.h" |
| 17 #include "net/base/net_errors.h" | 17 #include "net/base/net_errors.h" |
| 18 #include "net/base/sys_addrinfo.h" | 18 #include "net/base/sys_addrinfo.h" |
| 19 #include "net/base/winsock_init.h" | 19 #include "net/base/winsock_init.h" |
| 20 | 20 |
| 21 namespace net { | 21 namespace net { |
| 22 | 22 |
| 23 namespace { | 23 namespace { |
| 24 | 24 |
| 25 // If the (manual-reset) event object is signaled, resets it and returns true. | 25 // If the (manual-reset) event object is signaled, resets it and returns true. |
| 26 // Otherwise, does nothing and returns false. Called after a Winsock function | 26 // Otherwise, does nothing and returns false. Called after a Winsock function |
| (...skipping 254 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 281 read_callback_(NULL), | 281 read_callback_(NULL), |
| 282 write_callback_(NULL) { | 282 write_callback_(NULL) { |
| 283 EnsureWinsockInit(); | 283 EnsureWinsockInit(); |
| 284 } | 284 } |
| 285 | 285 |
| 286 TCPClientSocketWin::~TCPClientSocketWin() { | 286 TCPClientSocketWin::~TCPClientSocketWin() { |
| 287 Disconnect(); | 287 Disconnect(); |
| 288 } | 288 } |
| 289 | 289 |
| 290 int TCPClientSocketWin::Connect(CompletionCallback* callback, | 290 int TCPClientSocketWin::Connect(CompletionCallback* callback, |
| 291 LoadLog* load_log) { | 291 const BoundNetLog& net_log) { |
| 292 // If already connected, then just return OK. | 292 // If already connected, then just return OK. |
| 293 if (socket_ != INVALID_SOCKET) | 293 if (socket_ != INVALID_SOCKET) |
| 294 return OK; | 294 return OK; |
| 295 | 295 |
| 296 DCHECK(!load_log_); | 296 DCHECK(!net_log_.net_log()); |
| 297 | 297 |
| 298 static StatsCounter connects("tcp.connect"); | 298 static StatsCounter connects("tcp.connect"); |
| 299 connects.Increment(); | 299 connects.Increment(); |
| 300 | 300 |
| 301 TRACE_EVENT_BEGIN("socket.connect", this, ""); | 301 TRACE_EVENT_BEGIN("socket.connect", this, ""); |
| 302 | 302 |
| 303 LoadLog::BeginEvent(load_log, LoadLog::TYPE_TCP_CONNECT); | 303 net_log.BeginEvent(NetLog::TYPE_TCP_CONNECT); |
| 304 | 304 |
| 305 int rv = DoConnect(); | 305 int rv = DoConnect(); |
| 306 | 306 |
| 307 if (rv == ERR_IO_PENDING) { | 307 if (rv == ERR_IO_PENDING) { |
| 308 // Synchronous operation not supported. | 308 // Synchronous operation not supported. |
| 309 DCHECK(callback); | 309 DCHECK(callback); |
| 310 | 310 |
| 311 load_log_ = load_log; | 311 net_log_ = net_log; |
| 312 waiting_connect_ = true; | 312 waiting_connect_ = true; |
| 313 read_callback_ = callback; | 313 read_callback_ = callback; |
| 314 } else { | 314 } else { |
| 315 TRACE_EVENT_END("socket.connect", this, ""); | 315 TRACE_EVENT_END("socket.connect", this, ""); |
| 316 LoadLog::EndEvent(load_log, LoadLog::TYPE_TCP_CONNECT); | 316 net_log.EndEvent(NetLog::TYPE_TCP_CONNECT); |
| 317 if (rv == OK) | 317 if (rv == OK) |
| 318 UpdateConnectionTypeHistograms(CONNECTION_ANY); | 318 UpdateConnectionTypeHistograms(CONNECTION_ANY); |
| 319 } | 319 } |
| 320 | 320 |
| 321 return rv; | 321 return rv; |
| 322 } | 322 } |
| 323 | 323 |
| 324 int TCPClientSocketWin::DoConnect() { | 324 int TCPClientSocketWin::DoConnect() { |
| 325 const struct addrinfo* ai = current_ai_; | 325 const struct addrinfo* ai = current_ai_; |
| 326 DCHECK(ai); | 326 DCHECK(ai); |
| (...skipping 324 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 651 if (rv == SOCKET_ERROR) { | 651 if (rv == SOCKET_ERROR) { |
| 652 NOTREACHED(); | 652 NOTREACHED(); |
| 653 result = MapWinsockError(WSAGetLastError()); | 653 result = MapWinsockError(WSAGetLastError()); |
| 654 } else if (events.lNetworkEvents & FD_CONNECT) { | 654 } else if (events.lNetworkEvents & FD_CONNECT) { |
| 655 int os_error = events.iErrorCode[FD_CONNECT_BIT]; | 655 int os_error = events.iErrorCode[FD_CONNECT_BIT]; |
| 656 if (current_ai_->ai_next && ShouldTryNextAddress(os_error)) { | 656 if (current_ai_->ai_next && ShouldTryNextAddress(os_error)) { |
| 657 // Try using the next address. | 657 // Try using the next address. |
| 658 const struct addrinfo* next = current_ai_->ai_next; | 658 const struct addrinfo* next = current_ai_->ai_next; |
| 659 Disconnect(); | 659 Disconnect(); |
| 660 current_ai_ = next; | 660 current_ai_ = next; |
| 661 scoped_refptr<LoadLog> load_log; | 661 BoundNetLog net_log(net_log_); |
| 662 load_log.swap(load_log_); | 662 net_log_ = BoundNetLog(); |
| 663 TRACE_EVENT_END("socket.connect", this, ""); | 663 TRACE_EVENT_END("socket.connect", this, ""); |
| 664 LoadLog::EndEvent(load_log, LoadLog::TYPE_TCP_CONNECT); | 664 net_log.EndEvent(NetLog::TYPE_TCP_CONNECT); |
| 665 result = Connect(read_callback_, load_log); | 665 result = Connect(read_callback_, net_log); |
| 666 } else { | 666 } else { |
| 667 result = MapConnectError(os_error); | 667 result = MapConnectError(os_error); |
| 668 TRACE_EVENT_END("socket.connect", this, ""); | 668 TRACE_EVENT_END("socket.connect", this, ""); |
| 669 LoadLog::EndEvent(load_log_, LoadLog::TYPE_TCP_CONNECT); | 669 net_log_.EndEvent(NetLog::TYPE_TCP_CONNECT); |
| 670 load_log_ = NULL; | 670 net_log_ = BoundNetLog(); |
| 671 } | 671 } |
| 672 } else { | 672 } else { |
| 673 NOTREACHED(); | 673 NOTREACHED(); |
| 674 result = ERR_UNEXPECTED; | 674 result = ERR_UNEXPECTED; |
| 675 TRACE_EVENT_END("socket.connect", this, ""); | 675 TRACE_EVENT_END("socket.connect", this, ""); |
| 676 LoadLog::EndEvent(load_log_, LoadLog::TYPE_TCP_CONNECT); | 676 net_log_.EndEvent(NetLog::TYPE_TCP_CONNECT); |
| 677 load_log_ = NULL; | 677 net_log_ = BoundNetLog(); |
| 678 } | 678 } |
| 679 | 679 |
| 680 if (result != ERR_IO_PENDING) { | 680 if (result != ERR_IO_PENDING) { |
| 681 if (result == OK) | 681 if (result == OK) |
| 682 UpdateConnectionTypeHistograms(CONNECTION_ANY); | 682 UpdateConnectionTypeHistograms(CONNECTION_ANY); |
| 683 DoReadCallback(result); | 683 DoReadCallback(result); |
| 684 } | 684 } |
| 685 } | 685 } |
| 686 | 686 |
| 687 void TCPClientSocketWin::DidCompleteRead() { | 687 void TCPClientSocketWin::DidCompleteRead() { |
| (...skipping 29 matching lines...) Expand all Loading... |
| 717 << core_->write_buffer_length_ << " bytes, but " << rv | 717 << core_->write_buffer_length_ << " bytes, but " << rv |
| 718 << " bytes reported."; | 718 << " bytes reported."; |
| 719 rv = ERR_WINSOCK_UNEXPECTED_WRITTEN_BYTES; | 719 rv = ERR_WINSOCK_UNEXPECTED_WRITTEN_BYTES; |
| 720 } | 720 } |
| 721 } | 721 } |
| 722 core_->write_iobuffer_ = NULL; | 722 core_->write_iobuffer_ = NULL; |
| 723 DoWriteCallback(rv); | 723 DoWriteCallback(rv); |
| 724 } | 724 } |
| 725 | 725 |
| 726 } // namespace net | 726 } // namespace net |
| OLD | NEW |