| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/transport_client_socket_pool.h" | 5 #include "net/socket/transport_client_socket_pool.h" |
| 6 | 6 |
| 7 #include "base/callback.h" | 7 #include "base/callback.h" |
| 8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
| 9 #include "base/logging.h" | 9 #include "base/logging.h" |
| 10 #include "base/message_loop.h" | 10 #include "base/message_loop.h" |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 45 *address = IPEndPoint(number, 80); | 45 *address = IPEndPoint(number, 80); |
| 46 } | 46 } |
| 47 | 47 |
| 48 class MockClientSocket : public StreamSocket { | 48 class MockClientSocket : public StreamSocket { |
| 49 public: | 49 public: |
| 50 MockClientSocket(const AddressList& addrlist) | 50 MockClientSocket(const AddressList& addrlist) |
| 51 : connected_(false), | 51 : connected_(false), |
| 52 addrlist_(addrlist) {} | 52 addrlist_(addrlist) {} |
| 53 | 53 |
| 54 // StreamSocket implementation. | 54 // StreamSocket implementation. |
| 55 virtual int Connect(OldCompletionCallback* callback) { | |
| 56 connected_ = true; | |
| 57 return OK; | |
| 58 } | |
| 59 virtual int Connect(const CompletionCallback& callback) { | 55 virtual int Connect(const CompletionCallback& callback) { |
| 60 connected_ = true; | 56 connected_ = true; |
| 61 return OK; | 57 return OK; |
| 62 } | 58 } |
| 63 virtual void Disconnect() { | 59 virtual void Disconnect() { |
| 64 connected_ = false; | 60 connected_ = false; |
| 65 } | 61 } |
| 66 virtual bool IsConnected() const { | 62 virtual bool IsConnected() const { |
| 67 return connected_; | 63 return connected_; |
| 68 } | 64 } |
| (...skipping 20 matching lines...) Expand all Loading... |
| 89 virtual void SetOmniboxSpeculation() {} | 85 virtual void SetOmniboxSpeculation() {} |
| 90 virtual bool WasEverUsed() const { return false; } | 86 virtual bool WasEverUsed() const { return false; } |
| 91 virtual bool UsingTCPFastOpen() const { return false; } | 87 virtual bool UsingTCPFastOpen() const { return false; } |
| 92 virtual int64 NumBytesRead() const { return -1; } | 88 virtual int64 NumBytesRead() const { return -1; } |
| 93 virtual base::TimeDelta GetConnectTimeMicros() const { | 89 virtual base::TimeDelta GetConnectTimeMicros() const { |
| 94 return base::TimeDelta::FromMicroseconds(-1); | 90 return base::TimeDelta::FromMicroseconds(-1); |
| 95 } | 91 } |
| 96 | 92 |
| 97 // Socket implementation. | 93 // Socket implementation. |
| 98 virtual int Read(IOBuffer* buf, int buf_len, | 94 virtual int Read(IOBuffer* buf, int buf_len, |
| 99 OldCompletionCallback* callback) { | |
| 100 return ERR_FAILED; | |
| 101 } | |
| 102 virtual int Read(IOBuffer* buf, int buf_len, | |
| 103 const CompletionCallback& callback) { | 95 const CompletionCallback& callback) { |
| 104 return ERR_FAILED; | 96 return ERR_FAILED; |
| 105 } | 97 } |
| 106 virtual int Write(IOBuffer* buf, int buf_len, | 98 virtual int Write(IOBuffer* buf, int buf_len, |
| 107 OldCompletionCallback* callback) { | 99 const CompletionCallback& callback) { |
| 108 return ERR_FAILED; | 100 return ERR_FAILED; |
| 109 } | 101 } |
| 110 virtual bool SetReceiveBufferSize(int32 size) { return true; } | 102 virtual bool SetReceiveBufferSize(int32 size) { return true; } |
| 111 virtual bool SetSendBufferSize(int32 size) { return true; } | 103 virtual bool SetSendBufferSize(int32 size) { return true; } |
| 112 | 104 |
| 113 private: | 105 private: |
| 114 bool connected_; | 106 bool connected_; |
| 115 const AddressList addrlist_; | 107 const AddressList addrlist_; |
| 116 BoundNetLog net_log_; | 108 BoundNetLog net_log_; |
| 117 }; | 109 }; |
| 118 | 110 |
| 119 class MockFailingClientSocket : public StreamSocket { | 111 class MockFailingClientSocket : public StreamSocket { |
| 120 public: | 112 public: |
| 121 MockFailingClientSocket(const AddressList& addrlist) : addrlist_(addrlist) {} | 113 MockFailingClientSocket(const AddressList& addrlist) : addrlist_(addrlist) {} |
| 122 | 114 |
| 123 // StreamSocket implementation. | 115 // StreamSocket implementation. |
| 124 virtual int Connect(OldCompletionCallback* callback) { | 116 virtual int Connect(const CompletionCallback& callback) { |
| 125 return ERR_CONNECTION_FAILED; | |
| 126 } | |
| 127 virtual int Connect(const net::CompletionCallback& callback) { | |
| 128 return ERR_CONNECTION_FAILED; | 117 return ERR_CONNECTION_FAILED; |
| 129 } | 118 } |
| 130 | 119 |
| 131 virtual void Disconnect() {} | 120 virtual void Disconnect() {} |
| 132 | 121 |
| 133 virtual bool IsConnected() const { | 122 virtual bool IsConnected() const { |
| 134 return false; | 123 return false; |
| 135 } | 124 } |
| 136 virtual bool IsConnectedAndIdle() const { | 125 virtual bool IsConnectedAndIdle() const { |
| 137 return false; | 126 return false; |
| (...skipping 12 matching lines...) Expand all Loading... |
| 150 virtual void SetOmniboxSpeculation() {} | 139 virtual void SetOmniboxSpeculation() {} |
| 151 virtual bool WasEverUsed() const { return false; } | 140 virtual bool WasEverUsed() const { return false; } |
| 152 virtual bool UsingTCPFastOpen() const { return false; } | 141 virtual bool UsingTCPFastOpen() const { return false; } |
| 153 virtual int64 NumBytesRead() const { return -1; } | 142 virtual int64 NumBytesRead() const { return -1; } |
| 154 virtual base::TimeDelta GetConnectTimeMicros() const { | 143 virtual base::TimeDelta GetConnectTimeMicros() const { |
| 155 return base::TimeDelta::FromMicroseconds(-1); | 144 return base::TimeDelta::FromMicroseconds(-1); |
| 156 } | 145 } |
| 157 | 146 |
| 158 // Socket implementation. | 147 // Socket implementation. |
| 159 virtual int Read(IOBuffer* buf, int buf_len, | 148 virtual int Read(IOBuffer* buf, int buf_len, |
| 160 OldCompletionCallback* callback) { | |
| 161 return ERR_FAILED; | |
| 162 } | |
| 163 virtual int Read(IOBuffer* buf, int buf_len, | |
| 164 const CompletionCallback& callback) { | 149 const CompletionCallback& callback) { |
| 165 return ERR_FAILED; | 150 return ERR_FAILED; |
| 166 } | 151 } |
| 167 | 152 |
| 168 virtual int Write(IOBuffer* buf, int buf_len, | 153 virtual int Write(IOBuffer* buf, int buf_len, |
| 169 OldCompletionCallback* callback) { | 154 const CompletionCallback& callback) { |
| 170 return ERR_FAILED; | 155 return ERR_FAILED; |
| 171 } | 156 } |
| 172 virtual bool SetReceiveBufferSize(int32 size) { return true; } | 157 virtual bool SetReceiveBufferSize(int32 size) { return true; } |
| 173 virtual bool SetSendBufferSize(int32 size) { return true; } | 158 virtual bool SetSendBufferSize(int32 size) { return true; } |
| 174 | 159 |
| 175 private: | 160 private: |
| 176 const AddressList addrlist_; | 161 const AddressList addrlist_; |
| 177 BoundNetLog net_log_; | 162 BoundNetLog net_log_; |
| 178 }; | 163 }; |
| 179 | 164 |
| 180 class MockPendingClientSocket : public StreamSocket { | 165 class MockPendingClientSocket : public StreamSocket { |
| 181 public: | 166 public: |
| 182 // |should_connect| indicates whether the socket should successfully complete | 167 // |should_connect| indicates whether the socket should successfully complete |
| 183 // or fail. | 168 // or fail. |
| 184 // |should_stall| indicates that this socket should never connect. | 169 // |should_stall| indicates that this socket should never connect. |
| 185 // |delay_ms| is the delay, in milliseconds, before simulating a connect. | 170 // |delay_ms| is the delay, in milliseconds, before simulating a connect. |
| 186 MockPendingClientSocket( | 171 MockPendingClientSocket( |
| 187 const AddressList& addrlist, | 172 const AddressList& addrlist, |
| 188 bool should_connect, | 173 bool should_connect, |
| 189 bool should_stall, | 174 bool should_stall, |
| 190 int delay_ms) | 175 int delay_ms) |
| 191 : ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)), | 176 : ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)), |
| 192 should_connect_(should_connect), | 177 should_connect_(should_connect), |
| 193 should_stall_(should_stall), | 178 should_stall_(should_stall), |
| 194 delay_ms_(delay_ms), | 179 delay_ms_(delay_ms), |
| 195 is_connected_(false), | 180 is_connected_(false), |
| 196 addrlist_(addrlist) {} | 181 addrlist_(addrlist) {} |
| 197 | 182 |
| 198 // StreamSocket implementation. | 183 // StreamSocket implementation. |
| 199 virtual int Connect(OldCompletionCallback* callback) { | |
| 200 MessageLoop::current()->PostDelayedTask( | |
| 201 FROM_HERE, | |
| 202 base::Bind(&MockPendingClientSocket::DoOldCallback, | |
| 203 weak_factory_.GetWeakPtr(), callback), | |
| 204 delay_ms_); | |
| 205 return ERR_IO_PENDING; | |
| 206 } | |
| 207 virtual int Connect(const CompletionCallback& callback) { | 184 virtual int Connect(const CompletionCallback& callback) { |
| 208 MessageLoop::current()->PostDelayedTask( | 185 MessageLoop::current()->PostDelayedTask( |
| 209 FROM_HERE, | 186 FROM_HERE, |
| 210 base::Bind(&MockPendingClientSocket::DoCallback, | 187 base::Bind(&MockPendingClientSocket::DoCallback, |
| 211 weak_factory_.GetWeakPtr(), callback), | 188 weak_factory_.GetWeakPtr(), callback), |
| 212 delay_ms_); | 189 delay_ms_); |
| 213 return ERR_IO_PENDING; | 190 return ERR_IO_PENDING; |
| 214 } | 191 } |
| 215 | 192 |
| 216 virtual void Disconnect() {} | 193 virtual void Disconnect() {} |
| (...skipping 24 matching lines...) Expand all Loading... |
| 241 virtual void SetOmniboxSpeculation() {} | 218 virtual void SetOmniboxSpeculation() {} |
| 242 virtual bool WasEverUsed() const { return false; } | 219 virtual bool WasEverUsed() const { return false; } |
| 243 virtual bool UsingTCPFastOpen() const { return false; } | 220 virtual bool UsingTCPFastOpen() const { return false; } |
| 244 virtual int64 NumBytesRead() const { return -1; } | 221 virtual int64 NumBytesRead() const { return -1; } |
| 245 virtual base::TimeDelta GetConnectTimeMicros() const { | 222 virtual base::TimeDelta GetConnectTimeMicros() const { |
| 246 return base::TimeDelta::FromMicroseconds(-1); | 223 return base::TimeDelta::FromMicroseconds(-1); |
| 247 } | 224 } |
| 248 | 225 |
| 249 // Socket implementation. | 226 // Socket implementation. |
| 250 virtual int Read(IOBuffer* buf, int buf_len, | 227 virtual int Read(IOBuffer* buf, int buf_len, |
| 251 OldCompletionCallback* callback) { | |
| 252 return ERR_FAILED; | |
| 253 } | |
| 254 virtual int Read(IOBuffer* buf, int buf_len, | |
| 255 const CompletionCallback& callback) { | 228 const CompletionCallback& callback) { |
| 256 return ERR_FAILED; | 229 return ERR_FAILED; |
| 257 } | 230 } |
| 258 | 231 |
| 259 virtual int Write(IOBuffer* buf, int buf_len, | 232 virtual int Write(IOBuffer* buf, int buf_len, |
| 260 OldCompletionCallback* callback) { | 233 const CompletionCallback& callback) { |
| 261 return ERR_FAILED; | 234 return ERR_FAILED; |
| 262 } | 235 } |
| 263 virtual bool SetReceiveBufferSize(int32 size) { return true; } | 236 virtual bool SetReceiveBufferSize(int32 size) { return true; } |
| 264 virtual bool SetSendBufferSize(int32 size) { return true; } | 237 virtual bool SetSendBufferSize(int32 size) { return true; } |
| 265 | 238 |
| 266 private: | 239 private: |
| 267 void DoOldCallback(OldCompletionCallback* callback) { | |
| 268 if (should_stall_) | |
| 269 return; | |
| 270 | |
| 271 if (should_connect_) { | |
| 272 is_connected_ = true; | |
| 273 callback->Run(OK); | |
| 274 } else { | |
| 275 is_connected_ = false; | |
| 276 callback->Run(ERR_CONNECTION_FAILED); | |
| 277 } | |
| 278 } | |
| 279 void DoCallback(const CompletionCallback& callback) { | 240 void DoCallback(const CompletionCallback& callback) { |
| 280 if (should_stall_) | 241 if (should_stall_) |
| 281 return; | 242 return; |
| 282 | 243 |
| 283 if (should_connect_) { | 244 if (should_connect_) { |
| 284 is_connected_ = true; | 245 is_connected_ = true; |
| 285 callback.Run(OK); | 246 callback.Run(OK); |
| 286 } else { | 247 } else { |
| 287 is_connected_ = false; | 248 is_connected_ = false; |
| 288 callback.Run(ERR_CONNECTION_FAILED); | 249 callback.Run(ERR_CONNECTION_FAILED); |
| (...skipping 978 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1267 EXPECT_TRUE(handle.socket()); | 1228 EXPECT_TRUE(handle.socket()); |
| 1268 IPEndPoint endpoint; | 1229 IPEndPoint endpoint; |
| 1269 handle.socket()->GetLocalAddress(&endpoint); | 1230 handle.socket()->GetLocalAddress(&endpoint); |
| 1270 EXPECT_EQ(kIPv4AddressSize, endpoint.address().size()); | 1231 EXPECT_EQ(kIPv4AddressSize, endpoint.address().size()); |
| 1271 EXPECT_EQ(1, client_socket_factory_.allocation_count()); | 1232 EXPECT_EQ(1, client_socket_factory_.allocation_count()); |
| 1272 } | 1233 } |
| 1273 | 1234 |
| 1274 } // namespace | 1235 } // namespace |
| 1275 | 1236 |
| 1276 } // namespace net | 1237 } // namespace net |
| OLD | NEW |