| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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.h" | 5 #include "net/socket/tcp_client_socket.h" |
| 6 | 6 |
| 7 #include "base/callback_helpers.h" | 7 #include "base/callback_helpers.h" |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "base/profiler/scoped_tracker.h" | 9 #include "base/profiler/scoped_tracker.h" |
| 10 #include "net/base/io_buffer.h" | 10 #include "net/base/io_buffer.h" |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 56 | 56 |
| 57 result = socket_->Bind(address); | 57 result = socket_->Bind(address); |
| 58 if (result != OK) | 58 if (result != OK) |
| 59 return result; | 59 return result; |
| 60 | 60 |
| 61 bind_address_.reset(new IPEndPoint(address)); | 61 bind_address_.reset(new IPEndPoint(address)); |
| 62 return OK; | 62 return OK; |
| 63 } | 63 } |
| 64 | 64 |
| 65 int TCPClientSocket::Connect(const CompletionCallback& callback) { | 65 int TCPClientSocket::Connect(const CompletionCallback& callback) { |
| 66 // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed. | |
| 67 tracked_objects::ScopedTracker tracking_profile( | |
| 68 FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPClientSocket::Connect")); | |
| 69 | |
| 70 DCHECK(!callback.is_null()); | 66 DCHECK(!callback.is_null()); |
| 71 | 67 |
| 72 // If connecting or already connected, then just return OK. | 68 // If connecting or already connected, then just return OK. |
| 73 if (socket_->IsValid() && current_address_index_ >= 0) | 69 if (socket_->IsValid() && current_address_index_ >= 0) |
| 74 return OK; | 70 return OK; |
| 75 | 71 |
| 76 socket_->StartLoggingMultipleConnectAttempts(addresses_); | 72 socket_->StartLoggingMultipleConnectAttempts(addresses_); |
| 77 | 73 |
| 78 // We will try to connect to each address in addresses_. Start with the | 74 // We will try to connect to each address in addresses_. Start with the |
| 79 // first one in the list. | 75 // first one in the list. |
| (...skipping 29 matching lines...) Expand all Loading... |
| 109 NOTREACHED() << "bad state " << state; | 105 NOTREACHED() << "bad state " << state; |
| 110 rv = ERR_UNEXPECTED; | 106 rv = ERR_UNEXPECTED; |
| 111 break; | 107 break; |
| 112 } | 108 } |
| 113 } while (rv != ERR_IO_PENDING && next_connect_state_ != CONNECT_STATE_NONE); | 109 } while (rv != ERR_IO_PENDING && next_connect_state_ != CONNECT_STATE_NONE); |
| 114 | 110 |
| 115 return rv; | 111 return rv; |
| 116 } | 112 } |
| 117 | 113 |
| 118 int TCPClientSocket::DoConnect() { | 114 int TCPClientSocket::DoConnect() { |
| 119 // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed. | |
| 120 tracked_objects::ScopedTracker tracking_profile1( | |
| 121 FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPClientSocket::DoConnect1")); | |
| 122 | |
| 123 DCHECK_GE(current_address_index_, 0); | 115 DCHECK_GE(current_address_index_, 0); |
| 124 DCHECK_LT(current_address_index_, static_cast<int>(addresses_.size())); | 116 DCHECK_LT(current_address_index_, static_cast<int>(addresses_.size())); |
| 125 | 117 |
| 126 const IPEndPoint& endpoint = addresses_[current_address_index_]; | 118 const IPEndPoint& endpoint = addresses_[current_address_index_]; |
| 127 | 119 |
| 128 if (previously_disconnected_) { | 120 { |
| 129 use_history_.Reset(); | 121 // TODO(ricea): Remove ScopedTracker below once crbug.com/436634 is fixed. |
| 130 previously_disconnected_ = false; | 122 tracked_objects::ScopedTracker tracking_profile( |
| 131 } | 123 FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPClientSocket::DoConnect")); |
| 132 | 124 |
| 133 next_connect_state_ = CONNECT_STATE_CONNECT_COMPLETE; | 125 if (previously_disconnected_) { |
| 126 use_history_.Reset(); |
| 127 previously_disconnected_ = false; |
| 128 } |
| 134 | 129 |
| 135 if (socket_->IsValid()) { | 130 next_connect_state_ = CONNECT_STATE_CONNECT_COMPLETE; |
| 136 DCHECK(bind_address_); | |
| 137 } else { | |
| 138 int result = OpenSocket(endpoint.GetFamily()); | |
| 139 if (result != OK) | |
| 140 return result; | |
| 141 | 131 |
| 142 if (bind_address_) { | 132 if (socket_->IsValid()) { |
| 143 result = socket_->Bind(*bind_address_); | 133 DCHECK(bind_address_); |
| 144 if (result != OK) { | 134 } else { |
| 145 socket_->Close(); | 135 int result = OpenSocket(endpoint.GetFamily()); |
| 136 if (result != OK) |
| 146 return result; | 137 return result; |
| 138 |
| 139 if (bind_address_) { |
| 140 result = socket_->Bind(*bind_address_); |
| 141 if (result != OK) { |
| 142 socket_->Close(); |
| 143 return result; |
| 144 } |
| 147 } | 145 } |
| 148 } | 146 } |
| 149 } | 147 } |
| 150 | 148 |
| 151 // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed. | |
| 152 tracked_objects::ScopedTracker tracking_profile2( | |
| 153 FROM_HERE_WITH_EXPLICIT_FUNCTION("436634 TCPClientSocket::DoConnect2")); | |
| 154 | |
| 155 // |socket_| is owned by this class and the callback won't be run once | 149 // |socket_| is owned by this class and the callback won't be run once |
| 156 // |socket_| is gone. Therefore, it is safe to use base::Unretained() here. | 150 // |socket_| is gone. Therefore, it is safe to use base::Unretained() here. |
| 157 return socket_->Connect(endpoint, | 151 return socket_->Connect(endpoint, |
| 158 base::Bind(&TCPClientSocket::DidCompleteConnect, | 152 base::Bind(&TCPClientSocket::DidCompleteConnect, |
| 159 base::Unretained(this))); | 153 base::Unretained(this))); |
| 160 } | 154 } |
| 161 | 155 |
| 162 int TCPClientSocket::DoConnectComplete(int result) { | 156 int TCPClientSocket::DoConnectComplete(int result) { |
| 163 // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed. | |
| 164 tracked_objects::ScopedTracker tracking_profile( | |
| 165 FROM_HERE_WITH_EXPLICIT_FUNCTION( | |
| 166 "436634 TCPClientSocket::DoConnectComplete")); | |
| 167 | |
| 168 if (result == OK) { | 157 if (result == OK) { |
| 169 use_history_.set_was_ever_connected(); | 158 use_history_.set_was_ever_connected(); |
| 170 return OK; // Done! | 159 return OK; // Done! |
| 171 } | 160 } |
| 172 | 161 |
| 173 // Close whatever partially connected socket we currently have. | 162 // Close whatever partially connected socket we currently have. |
| 174 DoDisconnect(); | 163 DoDisconnect(); |
| 175 | 164 |
| 176 // Try to fall back to the next address in the list. | 165 // Try to fall back to the next address in the list. |
| 177 if (current_address_index_ + 1 < static_cast<int>(addresses_.size())) { | 166 if (current_address_index_ + 1 < static_cast<int>(addresses_.size())) { |
| (...skipping 159 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 337 int result = socket_->Open(family); | 326 int result = socket_->Open(family); |
| 338 if (result != OK) | 327 if (result != OK) |
| 339 return result; | 328 return result; |
| 340 | 329 |
| 341 socket_->SetDefaultOptionsForClient(); | 330 socket_->SetDefaultOptionsForClient(); |
| 342 | 331 |
| 343 return OK; | 332 return OK; |
| 344 } | 333 } |
| 345 | 334 |
| 346 } // namespace net | 335 } // namespace net |
| OLD | NEW |