| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/socket_test_util.h" | 5 #include "net/socket/socket_test_util.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 #include <utility> | 8 #include <utility> |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| (...skipping 737 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 748 next_ssl_data->next_protos_expected_in_ssl_config.end(), | 748 next_ssl_data->next_protos_expected_in_ssl_config.end(), |
| 749 ssl_config.alpn_protos.begin())); | 749 ssl_config.alpn_protos.begin())); |
| 750 } | 750 } |
| 751 return std::unique_ptr<SSLClientSocket>(new MockSSLClientSocket( | 751 return std::unique_ptr<SSLClientSocket>(new MockSSLClientSocket( |
| 752 std::move(transport_socket), host_and_port, ssl_config, next_ssl_data)); | 752 std::move(transport_socket), host_and_port, ssl_config, next_ssl_data)); |
| 753 } | 753 } |
| 754 | 754 |
| 755 void MockClientSocketFactory::ClearSSLSessionCache() { | 755 void MockClientSocketFactory::ClearSSLSessionCache() { |
| 756 } | 756 } |
| 757 | 757 |
| 758 MockClientSocket::MockClientSocket(const BoundNetLog& net_log) | 758 MockClientSocket::MockClientSocket(const NetLogWithSource& net_log) |
| 759 : connected_(false), | 759 : connected_(false), net_log_(net_log), weak_factory_(this) { |
| 760 net_log_(net_log), | |
| 761 weak_factory_(this) { | |
| 762 peer_addr_ = IPEndPoint(IPAddress(192, 0, 2, 33), 0); | 760 peer_addr_ = IPEndPoint(IPAddress(192, 0, 2, 33), 0); |
| 763 } | 761 } |
| 764 | 762 |
| 765 int MockClientSocket::SetReceiveBufferSize(int32_t size) { | 763 int MockClientSocket::SetReceiveBufferSize(int32_t size) { |
| 766 return OK; | 764 return OK; |
| 767 } | 765 } |
| 768 | 766 |
| 769 int MockClientSocket::SetSendBufferSize(int32_t size) { | 767 int MockClientSocket::SetSendBufferSize(int32_t size) { |
| 770 return OK; | 768 return OK; |
| 771 } | 769 } |
| (...skipping 15 matching lines...) Expand all Loading... |
| 787 return ERR_SOCKET_NOT_CONNECTED; | 785 return ERR_SOCKET_NOT_CONNECTED; |
| 788 *address = peer_addr_; | 786 *address = peer_addr_; |
| 789 return OK; | 787 return OK; |
| 790 } | 788 } |
| 791 | 789 |
| 792 int MockClientSocket::GetLocalAddress(IPEndPoint* address) const { | 790 int MockClientSocket::GetLocalAddress(IPEndPoint* address) const { |
| 793 *address = IPEndPoint(IPAddress(192, 0, 2, 33), 123); | 791 *address = IPEndPoint(IPAddress(192, 0, 2, 33), 123); |
| 794 return OK; | 792 return OK; |
| 795 } | 793 } |
| 796 | 794 |
| 797 const BoundNetLog& MockClientSocket::NetLog() const { | 795 const NetLogWithSource& MockClientSocket::NetLog() const { |
| 798 return net_log_; | 796 return net_log_; |
| 799 } | 797 } |
| 800 | 798 |
| 801 bool MockClientSocket::WasNpnNegotiated() const { | 799 bool MockClientSocket::WasNpnNegotiated() const { |
| 802 return false; | 800 return false; |
| 803 } | 801 } |
| 804 | 802 |
| 805 NextProto MockClientSocket::GetNegotiatedProtocol() const { | 803 NextProto MockClientSocket::GetNegotiatedProtocol() const { |
| 806 return kProtoUnknown; | 804 return kProtoUnknown; |
| 807 } | 805 } |
| (...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 856 | 854 |
| 857 void MockClientSocket::RunCallback(const CompletionCallback& callback, | 855 void MockClientSocket::RunCallback(const CompletionCallback& callback, |
| 858 int result) { | 856 int result) { |
| 859 if (!callback.is_null()) | 857 if (!callback.is_null()) |
| 860 callback.Run(result); | 858 callback.Run(result); |
| 861 } | 859 } |
| 862 | 860 |
| 863 MockTCPClientSocket::MockTCPClientSocket(const AddressList& addresses, | 861 MockTCPClientSocket::MockTCPClientSocket(const AddressList& addresses, |
| 864 net::NetLog* net_log, | 862 net::NetLog* net_log, |
| 865 SocketDataProvider* data) | 863 SocketDataProvider* data) |
| 866 : MockClientSocket(BoundNetLog::Make(net_log, NetLogSourceType::NONE)), | 864 : MockClientSocket(NetLogWithSource::Make(net_log, NetLogSourceType::NONE)), |
| 867 addresses_(addresses), | 865 addresses_(addresses), |
| 868 data_(data), | 866 data_(data), |
| 869 read_offset_(0), | 867 read_offset_(0), |
| 870 read_data_(SYNCHRONOUS, ERR_UNEXPECTED), | 868 read_data_(SYNCHRONOUS, ERR_UNEXPECTED), |
| 871 need_read_data_(true), | 869 need_read_data_(true), |
| 872 peer_closed_connection_(false), | 870 peer_closed_connection_(false), |
| 873 pending_read_buf_(NULL), | 871 pending_read_buf_(NULL), |
| 874 pending_read_buf_len_(0), | 872 pending_read_buf_len_(0), |
| 875 was_used_to_convey_data_(false) { | 873 was_used_to_convey_data_(false) { |
| 876 DCHECK(data_); | 874 DCHECK(data_); |
| (...skipping 251 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1128 ssl_client_socket->connected_ = true; | 1126 ssl_client_socket->connected_ = true; |
| 1129 callback.Run(rv); | 1127 callback.Run(rv); |
| 1130 } | 1128 } |
| 1131 | 1129 |
| 1132 MockSSLClientSocket::MockSSLClientSocket( | 1130 MockSSLClientSocket::MockSSLClientSocket( |
| 1133 std::unique_ptr<ClientSocketHandle> transport_socket, | 1131 std::unique_ptr<ClientSocketHandle> transport_socket, |
| 1134 const HostPortPair& host_port_pair, | 1132 const HostPortPair& host_port_pair, |
| 1135 const SSLConfig& ssl_config, | 1133 const SSLConfig& ssl_config, |
| 1136 SSLSocketDataProvider* data) | 1134 SSLSocketDataProvider* data) |
| 1137 : MockClientSocket( | 1135 : MockClientSocket( |
| 1138 // Have to use the right BoundNetLog for LoadTimingInfo regression | 1136 // Have to use the right NetLogWithSource for LoadTimingInfo |
| 1137 // regression |
| 1139 // tests. | 1138 // tests. |
| 1140 transport_socket->socket()->NetLog()), | 1139 transport_socket->socket()->NetLog()), |
| 1141 transport_(std::move(transport_socket)), | 1140 transport_(std::move(transport_socket)), |
| 1142 data_(data) { | 1141 data_(data) { |
| 1143 DCHECK(data_); | 1142 DCHECK(data_); |
| 1144 peer_addr_ = data->connect.peer_addr; | 1143 peer_addr_ = data->connect.peer_addr; |
| 1145 } | 1144 } |
| 1146 | 1145 |
| 1147 MockSSLClientSocket::~MockSSLClientSocket() { | 1146 MockSSLClientSocket::~MockSSLClientSocket() { |
| 1148 Disconnect(); | 1147 Disconnect(); |
| (...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1253 net::NetLog* net_log) | 1252 net::NetLog* net_log) |
| 1254 : connected_(false), | 1253 : connected_(false), |
| 1255 data_(data), | 1254 data_(data), |
| 1256 read_offset_(0), | 1255 read_offset_(0), |
| 1257 read_data_(SYNCHRONOUS, ERR_UNEXPECTED), | 1256 read_data_(SYNCHRONOUS, ERR_UNEXPECTED), |
| 1258 need_read_data_(true), | 1257 need_read_data_(true), |
| 1259 source_port_(123), | 1258 source_port_(123), |
| 1260 network_(NetworkChangeNotifier::kInvalidNetworkHandle), | 1259 network_(NetworkChangeNotifier::kInvalidNetworkHandle), |
| 1261 pending_read_buf_(NULL), | 1260 pending_read_buf_(NULL), |
| 1262 pending_read_buf_len_(0), | 1261 pending_read_buf_len_(0), |
| 1263 net_log_(BoundNetLog::Make(net_log, NetLogSourceType::NONE)), | 1262 net_log_(NetLogWithSource::Make(net_log, NetLogSourceType::NONE)), |
| 1264 weak_factory_(this) { | 1263 weak_factory_(this) { |
| 1265 DCHECK(data_); | 1264 DCHECK(data_); |
| 1266 data_->Initialize(this); | 1265 data_->Initialize(this); |
| 1267 peer_addr_ = data->connect_data().peer_addr; | 1266 peer_addr_ = data->connect_data().peer_addr; |
| 1268 } | 1267 } |
| 1269 | 1268 |
| 1270 MockUDPClientSocket::~MockUDPClientSocket() { | 1269 MockUDPClientSocket::~MockUDPClientSocket() { |
| 1271 if (data_) | 1270 if (data_) |
| 1272 data_->DetachSocket(); | 1271 data_->DetachSocket(); |
| 1273 } | 1272 } |
| (...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1346 return OK; | 1345 return OK; |
| 1347 } | 1346 } |
| 1348 | 1347 |
| 1349 int MockUDPClientSocket::GetLocalAddress(IPEndPoint* address) const { | 1348 int MockUDPClientSocket::GetLocalAddress(IPEndPoint* address) const { |
| 1350 *address = IPEndPoint(IPAddress(192, 0, 2, 33), source_port_); | 1349 *address = IPEndPoint(IPAddress(192, 0, 2, 33), source_port_); |
| 1351 return OK; | 1350 return OK; |
| 1352 } | 1351 } |
| 1353 | 1352 |
| 1354 void MockUDPClientSocket::UseNonBlockingIO() {} | 1353 void MockUDPClientSocket::UseNonBlockingIO() {} |
| 1355 | 1354 |
| 1356 const BoundNetLog& MockUDPClientSocket::NetLog() const { | 1355 const NetLogWithSource& MockUDPClientSocket::NetLog() const { |
| 1357 return net_log_; | 1356 return net_log_; |
| 1358 } | 1357 } |
| 1359 | 1358 |
| 1360 int MockUDPClientSocket::Connect(const IPEndPoint& address) { | 1359 int MockUDPClientSocket::Connect(const IPEndPoint& address) { |
| 1361 if (!data_) | 1360 if (!data_) |
| 1362 return ERR_UNEXPECTED; | 1361 return ERR_UNEXPECTED; |
| 1363 connected_ = true; | 1362 connected_ = true; |
| 1364 peer_addr_ = address; | 1363 peer_addr_ = address; |
| 1365 return data_->connect_data().result; | 1364 return data_->connect_data().result; |
| 1366 } | 1365 } |
| (...skipping 254 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1621 | 1620 |
| 1622 MockTransportClientSocketPool::~MockTransportClientSocketPool() {} | 1621 MockTransportClientSocketPool::~MockTransportClientSocketPool() {} |
| 1623 | 1622 |
| 1624 int MockTransportClientSocketPool::RequestSocket( | 1623 int MockTransportClientSocketPool::RequestSocket( |
| 1625 const std::string& group_name, | 1624 const std::string& group_name, |
| 1626 const void* socket_params, | 1625 const void* socket_params, |
| 1627 RequestPriority priority, | 1626 RequestPriority priority, |
| 1628 RespectLimits respect_limits, | 1627 RespectLimits respect_limits, |
| 1629 ClientSocketHandle* handle, | 1628 ClientSocketHandle* handle, |
| 1630 const CompletionCallback& callback, | 1629 const CompletionCallback& callback, |
| 1631 const BoundNetLog& net_log) { | 1630 const NetLogWithSource& net_log) { |
| 1632 last_request_priority_ = priority; | 1631 last_request_priority_ = priority; |
| 1633 std::unique_ptr<StreamSocket> socket = | 1632 std::unique_ptr<StreamSocket> socket = |
| 1634 client_socket_factory_->CreateTransportClientSocket( | 1633 client_socket_factory_->CreateTransportClientSocket( |
| 1635 AddressList(), NULL, net_log.net_log(), NetLog::Source()); | 1634 AddressList(), NULL, net_log.net_log(), NetLog::Source()); |
| 1636 MockConnectJob* job = new MockConnectJob(std::move(socket), handle, callback); | 1635 MockConnectJob* job = new MockConnectJob(std::move(socket), handle, callback); |
| 1637 job_list_.push_back(base::WrapUnique(job)); | 1636 job_list_.push_back(base::WrapUnique(job)); |
| 1638 handle->set_pool_id(1); | 1637 handle->set_pool_id(1); |
| 1639 return job->Connect(); | 1638 return job->Connect(); |
| 1640 } | 1639 } |
| 1641 | 1640 |
| (...skipping 28 matching lines...) Expand all Loading... |
| 1670 transport_pool_(transport_pool) {} | 1669 transport_pool_(transport_pool) {} |
| 1671 | 1670 |
| 1672 MockSOCKSClientSocketPool::~MockSOCKSClientSocketPool() {} | 1671 MockSOCKSClientSocketPool::~MockSOCKSClientSocketPool() {} |
| 1673 | 1672 |
| 1674 int MockSOCKSClientSocketPool::RequestSocket(const std::string& group_name, | 1673 int MockSOCKSClientSocketPool::RequestSocket(const std::string& group_name, |
| 1675 const void* socket_params, | 1674 const void* socket_params, |
| 1676 RequestPriority priority, | 1675 RequestPriority priority, |
| 1677 RespectLimits respect_limits, | 1676 RespectLimits respect_limits, |
| 1678 ClientSocketHandle* handle, | 1677 ClientSocketHandle* handle, |
| 1679 const CompletionCallback& callback, | 1678 const CompletionCallback& callback, |
| 1680 const BoundNetLog& net_log) { | 1679 const NetLogWithSource& net_log) { |
| 1681 return transport_pool_->RequestSocket(group_name, socket_params, priority, | 1680 return transport_pool_->RequestSocket(group_name, socket_params, priority, |
| 1682 respect_limits, handle, callback, | 1681 respect_limits, handle, callback, |
| 1683 net_log); | 1682 net_log); |
| 1684 } | 1683 } |
| 1685 | 1684 |
| 1686 void MockSOCKSClientSocketPool::CancelRequest( | 1685 void MockSOCKSClientSocketPool::CancelRequest( |
| 1687 const std::string& group_name, | 1686 const std::string& group_name, |
| 1688 ClientSocketHandle* handle) { | 1687 ClientSocketHandle* handle) { |
| 1689 return transport_pool_->CancelRequest(group_name, handle); | 1688 return transport_pool_->CancelRequest(group_name, handle); |
| 1690 } | 1689 } |
| (...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1733 } | 1732 } |
| 1734 | 1733 |
| 1735 int64_t CountWriteBytes(const MockWrite writes[], size_t writes_size) { | 1734 int64_t CountWriteBytes(const MockWrite writes[], size_t writes_size) { |
| 1736 int64_t total = 0; | 1735 int64_t total = 0; |
| 1737 for (const MockWrite* write = writes; write != writes + writes_size; ++write) | 1736 for (const MockWrite* write = writes; write != writes + writes_size; ++write) |
| 1738 total += write->data_len; | 1737 total += write->data_len; |
| 1739 return total; | 1738 return total; |
| 1740 } | 1739 } |
| 1741 | 1740 |
| 1742 } // namespace net | 1741 } // namespace net |
| OLD | NEW |