15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef NET_SOCKET_SOCKET_TEST_UTIL_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define NET_SOCKET_SOCKET_TEST_UTIL_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <cstring> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <deque> 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector> 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/callback.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 163551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)#include "base/memory/ref_counted.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_vector.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/weak_ptr.h" 205e3f23d412006dc4db4e659864679f29341e113fTorne (Richard Coles)#include "base/strings/string16.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/address_list.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/io_buffer.h" 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_errors.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_log.h" 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/test_completion_callback.h" 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/http/http_auth_controller.h" 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/http/http_proxy_client_socket_pool.h" 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/socket/client_socket_factory.h" 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/socket/client_socket_handle.h" 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/socket/socks_client_socket_pool.h" 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/socket/ssl_client_socket.h" 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/socket/ssl_client_socket_pool.h" 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/socket/transport_client_socket_pool.h" 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/ssl/ssl_config_service.h" 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/udp/datagram_client_socket.h" 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)enum { 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A private network error code used by the socket test utility classes. 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If the |result| member of a MockRead is 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ERR_TEST_PEER_CLOSE_AFTER_NEXT_MOCK_READ, that MockRead is just a 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // marker that indicates the peer will close the connection after the next 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MockRead. The other members of that MockRead are ignored. 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ERR_TEST_PEER_CLOSE_AFTER_NEXT_MOCK_READ = -10000, 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class AsyncSocket; 505f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)class ChannelIDService; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockClientSocket; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SSLClientSocket; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class StreamSocket; 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)enum IoMode { 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASYNC, 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SYNCHRONOUS 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct MockConnect { 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Asynchronous connection success. 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Creates a MockConnect with |mode| ASYC, |result| OK, and 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |peer_addr| 192.0.2.33. 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockConnect(); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Creates a MockConnect with the specified mode and result, with 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |peer_addr| 192.0.2.33. 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockConnect(IoMode io_mode, int r); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockConnect(IoMode io_mode, int r, IPEndPoint addr); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~MockConnect(); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IoMode mode; 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int result; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IPEndPoint peer_addr; 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MockRead and MockWrite shares the same interface and members, but we'd like 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// to have distinct types because we don't want to have them used 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// interchangably. To do this, a struct template is defined, and MockRead and 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MockWrite are instantiated by using this template. Template parameter |type| 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// is not used in the struct definition (it purely exists for creating a new 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// type). 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |data| in MockRead and MockWrite has different meanings: |data| in MockRead 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// is the data returned from the socket when MockTCPClientSocket::Read() is 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// attempted, while |data| in MockWrite is the expected data that should be 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// given in MockTCPClientSocket::Write(). 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)enum MockReadWriteType { 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MOCK_READ, 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MOCK_WRITE 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)template <MockReadWriteType type> 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct MockReadWrite { 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Flag to indicate that the message loop should be terminated. 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) enum { 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) STOPLOOP = 1 << 31 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Default 1005d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockReadWrite() 1015d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : mode(SYNCHRONOUS), 1025d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result(0), 1035d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data(NULL), 1045d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data_len(0), 1055d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) sequence_number(0), 1065d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) time_stamp(base::Time::Now()) {} 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read/write failure (no data). 1095d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockReadWrite(IoMode io_mode, int result) 1105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : mode(io_mode), 1115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result(result), 1125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data(NULL), 1135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data_len(0), 1145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) sequence_number(0), 1155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) time_stamp(base::Time::Now()) {} 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read/write failure (no data), with sequence information. 1185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockReadWrite(IoMode io_mode, int result, int seq) 1195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : mode(io_mode), 1205d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result(result), 1215d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data(NULL), 1225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data_len(0), 1235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) sequence_number(seq), 1245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) time_stamp(base::Time::Now()) {} 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Asynchronous read/write success (inferred data length). 1275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) explicit MockReadWrite(const char* data) 1285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : mode(ASYNC), 1295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result(0), 1305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data(data), 1315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data_len(strlen(data)), 1325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) sequence_number(0), 1335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) time_stamp(base::Time::Now()) {} 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read/write success (inferred data length). 1365d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockReadWrite(IoMode io_mode, const char* data) 1375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : mode(io_mode), 1385d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result(0), 1395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data(data), 1405d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data_len(strlen(data)), 1415d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) sequence_number(0), 1425d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) time_stamp(base::Time::Now()) {} 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read/write success. 1455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockReadWrite(IoMode io_mode, const char* data, int data_len) 1465d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : mode(io_mode), 1475d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result(0), 1485d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data(data), 1495d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data_len(data_len), 1505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) sequence_number(0), 1515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) time_stamp(base::Time::Now()) {} 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read/write success (inferred data length) with sequence information. 1545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockReadWrite(IoMode io_mode, int seq, const char* data) 1555d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : mode(io_mode), 1565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result(0), 1575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data(data), 1585d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data_len(strlen(data)), 1595d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) sequence_number(seq), 1605d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) time_stamp(base::Time::Now()) {} 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read/write success with sequence information. 1635d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockReadWrite(IoMode io_mode, const char* data, int data_len, int seq) 1645d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : mode(io_mode), 1655d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) result(0), 1665d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data(data), 1675d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) data_len(data_len), 1685d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) sequence_number(seq), 1695d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) time_stamp(base::Time::Now()) {} 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IoMode mode; 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int result; 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const char* data; 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int data_len; 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For OrderedSocketData, which only allows reads to occur in a particular 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // sequence. If a read occurs before the given |sequence_number| is reached, 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // an ERR_IO_PENDING is returned. 1795d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int sequence_number; // The sequence number at which a read is allowed 1805d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // to occur. 1815d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::Time time_stamp; // The time stamp at which the operation occurred. 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef MockReadWrite<MOCK_READ> MockRead; 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef MockReadWrite<MOCK_WRITE> MockWrite; 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct MockWriteResult { 1885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockWriteResult(IoMode io_mode, int result) : mode(io_mode), result(result) {} 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IoMode mode; 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int result; 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The SocketDataProvider is an interface used by the MockClientSocket 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// for getting data about individual reads and writes on the socket. 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SocketDataProvider { 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProvider() : socket_(NULL) {} 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~SocketDataProvider() {} 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns the buffer and result code for the next simulated read. 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If the |MockRead.result| is ERR_IO_PENDING, it informs the caller 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // that it will be called via the AsyncSocket::OnReadComplete() 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // function at a later time. 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockRead GetNextRead() = 0; 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockWriteResult OnWrite(const std::string& data) = 0; 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Reset() = 0; 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Accessor for the socket which is using the SocketDataProvider. 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AsyncSocket* socket() { return socket_; } 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void set_socket(AsyncSocket* socket) { socket_ = socket; } 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockConnect connect_data() const { return connect_; } 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void set_connect_data(const MockConnect& connect) { connect_ = connect; } 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockConnect connect_; 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AsyncSocket* socket_; 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(SocketDataProvider); 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The AsyncSocket is an interface used by the SocketDataProvider to 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// complete the asynchronous read operation. 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class AsyncSocket { 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If an async IO is pending because the SocketDataProvider returned 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ERR_IO_PENDING, then the AsyncSocket waits until this OnReadComplete 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // is called to complete the asynchronous read operation. 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // data.async is ignored, and this read is completed synchronously as 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // part of this call. 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnReadComplete(const MockRead& data) = 0; 234eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch virtual void OnConnectComplete(const MockConnect& data) = 0; 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SocketDataProvider which responds based on static tables of mock reads and 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// writes. 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class StaticSocketDataProvider : public SocketDataProvider { 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StaticSocketDataProvider(); 2425d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) StaticSocketDataProvider(MockRead* reads, 2435d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t reads_count, 2445d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockWrite* writes, 2455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t writes_count); 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~StaticSocketDataProvider(); 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // These functions get access to the next available read and write data. 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const MockRead& PeekRead() const; 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const MockWrite& PeekWrite() const; 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // These functions get random access to the read and write data, for timing. 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const MockRead& PeekRead(size_t index) const; 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const MockWrite& PeekWrite(size_t index) const; 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t read_index() const { return read_index_; } 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t write_index() const { return write_index_; } 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t read_count() const { return read_count_; } 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t write_count() const { return write_count_; } 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool at_read_eof() const { return read_index_ >= read_count_; } 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool at_write_eof() const { return write_index_ >= write_count_; } 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void CompleteRead() {} 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SocketDataProvider implementation. 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockRead GetNextRead() OVERRIDE; 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockWriteResult OnWrite(const std::string& data) OVERRIDE; 2678bcbed890bc3ce4d7a057a8f32cab53fa534672eTorne (Richard Coles) virtual void Reset() OVERRIDE; 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead* reads_; 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t read_index_; 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t read_count_; 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite* writes_; 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t write_index_; 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t write_count_; 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(StaticSocketDataProvider); 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SocketDataProvider which can make decisions about next mock reads based on 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// received writes. It can also be used to enforce order of operations, for 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// example that tested code must send the "Hello!" message before receiving 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// response. This is useful for testing conversation-like protocols like FTP. 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DynamicSocketDataProvider : public SocketDataProvider { 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DynamicSocketDataProvider(); 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DynamicSocketDataProvider(); 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int short_read_limit() const { return short_read_limit_; } 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void set_short_read_limit(int limit) { short_read_limit_ = limit; } 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void allow_unconsumed_reads(bool allow) { allow_unconsumed_reads_ = allow; } 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SocketDataProvider implementation. 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockRead GetNextRead() OVERRIDE; 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockWriteResult OnWrite(const std::string& data) = 0; 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Reset() OVERRIDE; 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The next time there is a read from this socket, it will return |data|. 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Before calling SimulateRead next time, the previous data must be consumed. 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SimulateRead(const char* data, size_t length); 3035d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void SimulateRead(const char* data) { SimulateRead(data, std::strlen(data)); } 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::deque<MockRead> reads_; 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Max number of bytes we will read at a time. 0 means no limit. 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int short_read_limit_; 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If true, we'll not require the client to consume all data before we 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // mock the next read. 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool allow_unconsumed_reads_; 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(DynamicSocketDataProvider); 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SSLSocketDataProviders only need to keep track of the return code from calls 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// to Connect(). 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct SSLSocketDataProvider { 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SSLSocketDataProvider(IoMode mode, int result); 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~SSLSocketDataProvider(); 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetNextProto(NextProto proto); 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockConnect connect; 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SSLClientSocket::NextProtoStatus next_proto_status; 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string next_proto; 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool was_npn_negotiated; 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NextProto protocol_negotiated; 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool client_cert_sent; 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SSLCertRequestInfo* cert_request_info; 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<X509Certificate> cert; 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool channel_id_sent; 3355f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) ChannelIDService* channel_id_service; 336cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) int connection_status; 3376e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) // Indicates that the socket should pause in the Connect method. 3386e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) bool should_pause_on_connect; 3395f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // Whether or not the Socket should behave like there is a pre-existing 3405f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // session to resume. Whether or not such a session is reported as 3415f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // resumed is controlled by |connection_status|. 3425f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) bool is_in_session_cache; 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A DataProvider where the client must write a request before the reads (e.g. 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the response) will complete. 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DelayedSocketData : public StaticSocketDataProvider { 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |write_delay| the number of MockWrites to complete before allowing 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // a MockRead to complete. 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |reads| the list of MockRead completions. 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |writes| the list of MockWrite completions. 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: For stream sockets, the MockRead list must end with a EOF, e.g., a 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MockRead(true, 0, 0); 3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DelayedSocketData(int write_delay, 3565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockRead* reads, 3575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t reads_count, 3585d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockWrite* writes, 3595d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t writes_count); 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |connect| the result for the connect phase. 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |reads| the list of MockRead completions. 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |write_delay| the number of MockWrites to complete before allowing 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // a MockRead to complete. 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |writes| the list of MockWrite completions. 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: For stream sockets, the MockRead list must end with a EOF, e.g., a 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MockRead(true, 0, 0); 3685d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) DelayedSocketData(const MockConnect& connect, 3695d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int write_delay, 3705d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockRead* reads, 3715d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t reads_count, 3725d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockWrite* writes, 3735d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t writes_count); 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DelayedSocketData(); 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ForceNextRead(); 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // StaticSocketDataProvider: 3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockRead GetNextRead() OVERRIDE; 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockWriteResult OnWrite(const std::string& data) OVERRIDE; 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Reset() OVERRIDE; 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void CompleteRead() OVERRIDE; 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int write_delay_; 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool read_in_progress_; 387f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WeakPtrFactory<DelayedSocketData> weak_factory_; 389f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 390f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(DelayedSocketData); 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A DataProvider where the reads are ordered. 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// If a read is requested before its sequence number is reached, we return an 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ERR_IO_PENDING (that way we don't have to explicitly add a MockRead just to 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// wait). 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The sequence number is incremented on every read and write operation. 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The message loop may be interrupted by setting the high bit of the sequence 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// number in the MockRead's sequence number. When that MockRead is reached, 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// we post a Quit message to the loop. This allows us to interrupt the reading 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// of data before a complete message has arrived, and provides support for 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// testing server push when the request is issued while the response is in the 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// middle of being received. 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class OrderedSocketData : public StaticSocketDataProvider { 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |reads| the list of MockRead completions. 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |writes| the list of MockWrite completions. 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: All MockReads and MockWrites must be async. 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: For stream sockets, the MockRead list must end with a EOF, e.g., a 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MockRead(true, 0, 0); 4115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) OrderedSocketData(MockRead* reads, 4125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t reads_count, 4135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockWrite* writes, 4145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t writes_count); 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~OrderedSocketData(); 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |connect| the result for the connect phase. 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |reads| the list of MockRead completions. 4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |writes| the list of MockWrite completions. 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: All MockReads and MockWrites must be async. 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: For stream sockets, the MockRead list must end with a EOF, e.g., a 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MockRead(true, 0, 0); 4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OrderedSocketData(const MockConnect& connect, 4245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockRead* reads, 4255d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t reads_count, 4265d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockWrite* writes, 4275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t writes_count); 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Posts a quit message to the current message loop, if one is running. 4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void EndLoop(); 4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // StaticSocketDataProvider: 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockRead GetNextRead() OVERRIDE; 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockWriteResult OnWrite(const std::string& data) OVERRIDE; 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Reset() OVERRIDE; 4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void CompleteRead() OVERRIDE; 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int sequence_number_; 4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int loop_stop_stage_; 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool blocked_; 442f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WeakPtrFactory<OrderedSocketData> weak_factory_; 444f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 445f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(OrderedSocketData); 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DeterministicMockTCPClientSocket; 4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This class gives the user full control over the network activity, 4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// specifically the timing of the COMPLETION of I/O operations. Regardless of 4525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the order in which I/O operations are initiated, this class ensures that they 4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// complete in the correct order. 4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Network activity is modeled as a sequence of numbered steps which is 4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// incremented whenever an I/O operation completes. This can happen under two 4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// different circumstances: 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 1) Performing a synchronous I/O operation. (Invoking Read() or Write() 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// when the corresponding MockRead or MockWrite is marked !async). 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 2) Running the Run() method of this class. The run method will invoke 4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the current MessageLoop, running all pending events, and will then 4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// invoke any pending IO callbacks. 4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// In addition, this class allows for I/O processing to "stop" at a specified 4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// step, by calling SetStop(int) or StopAfter(int). Initiating an I/O operation 4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// by calling Read() or Write() while stopped is permitted if the operation is 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// asynchronous. It is an error to perform synchronous I/O while stopped. 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// When creating the MockReads and MockWrites, note that the sequence number 4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// refers to the number of the step in which the I/O will complete. In the 4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// case of synchronous I/O, this will be the same step as the I/O is initiated. 4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// However, in the case of asynchronous I/O, this I/O may be initiated in 4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// a much earlier step. Furthermore, when the a Read() or Write() is separated 4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// from its completion by other Read() or Writes()'s, it can not be marked 4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// synchronous. If it is, ERR_UNUEXPECTED will be returned indicating that a 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// synchronous Read() or Write() could not be completed synchronously because of 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the specific ordering constraints. 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Sequence numbers are preserved across both reads and writes. There should be 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// no gaps in sequence numbers, and no repeated sequence numbers. i.e. 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MockRead reads[] = { 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MockRead(false, "first read", length, 0) // sync 4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MockRead(true, "second read", length, 2) // async 4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// }; 4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MockWrite writes[] = { 4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MockWrite(true, "first write", length, 1), // async 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MockWrite(false, "second write", length, 3), // sync 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// }; 4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Example control flow: 4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Read() is called. The current step is 0. The first available read is 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// synchronous, so the call to Read() returns length. The current step is 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// now 1. Next, Read() is called again. The next available read can 4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// not be completed until step 2, so Read() returns ERR_IO_PENDING. The current 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// step is still 1. Write is called(). The first available write is able to 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// complete in this step, but is marked asynchronous. Write() returns 4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ERR_IO_PENDING. The current step is still 1. At this point RunFor(1) is 4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// called which will cause the write callback to be invoked, and will then 5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// stop. The current state is now 2. RunFor(1) is called again, which 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// causes the read callback to be invoked, and will then stop. Then current 5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// step is 2. Write() is called again. Then next available write is 5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// synchronous so the call to Write() returns length. 5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// For examples of how to use this class, see: 5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// deterministic_socket_data_unittests.cc 5075d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)class DeterministicSocketData : public StaticSocketDataProvider { 5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 509b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // The Delegate is an abstract interface which handles the communication from 510b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // the DeterministicSocketData to the Deterministic MockSocket. The 511b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // MockSockets directly store a pointer to the DeterministicSocketData, 512b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // whereas the DeterministicSocketData only stores a pointer to the 513b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // abstract Delegate interface. 514b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) class Delegate { 515b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) public: 516b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // Returns true if there is currently a write pending. That is to say, if 517b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // an asynchronous write has been started but the callback has not been 518b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // invoked. 519b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual bool WritePending() const = 0; 520b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // Returns true if there is currently a read pending. That is to say, if 521b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // an asynchronous read has been started but the callback has not been 522b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // invoked. 523b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual bool ReadPending() const = 0; 524b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // Called to complete an asynchronous write to execute the write callback. 525b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual void CompleteWrite() = 0; 526b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // Called to complete an asynchronous read to execute the read callback. 527b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual int CompleteRead() = 0; 528b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 529b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) protected: 530b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual ~Delegate() {} 531b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) }; 532b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |reads| the list of MockRead completions. 5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |writes| the list of MockWrite completions. 5355d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) DeterministicSocketData(MockRead* reads, 5365d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t reads_count, 5375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockWrite* writes, 5385d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t writes_count); 5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DeterministicSocketData(); 5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Consume all the data up to the give stop point (via SetStop()). 5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Run(); 5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set the stop point to be |steps| from now, and then invoke Run(). 5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RunFor(int steps); 5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stop at step |seq|, which must be in the future. 5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetStop(int seq); 5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stop |seq| steps after the current step. 5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void StopAfter(int seq); 5525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool stopped() const { return stopped_; } 5535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetStopped(bool val) { stopped_ = val; } 5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead& current_read() { return current_read_; } 5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite& current_write() { return current_write_; } 5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int sequence_number() const { return sequence_number_; } 5575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void set_delegate(base::WeakPtr<Delegate> delegate) { delegate_ = delegate; } 5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // StaticSocketDataProvider: 5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // When the socket calls Read(), that calls GetNextRead(), and expects either 5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ERR_IO_PENDING or data. 5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockRead GetNextRead() OVERRIDE; 5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // When the socket calls Write(), it always completes synchronously. OnWrite() 5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // checks to make sure the written data matches the expected data. The 5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // callback will not be invoked until its sequence number is reached. 5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockWriteResult OnWrite(const std::string& data) OVERRIDE; 5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Reset() OVERRIDE; 5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void CompleteRead() OVERRIDE {} 5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Invoke the read and write callbacks, if the timing is appropriate. 5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void InvokeCallbacks(); 5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void NextStep(); 5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5785d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void VerifyCorrectSequenceNumbers(MockRead* reads, 5795d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t reads_count, 5805d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockWrite* writes, 5815d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) size_t writes_count); 5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int sequence_number_; 5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead current_read_; 5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite current_write_; 5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int stopping_sequence_number_; 5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool stopped_; 588b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) base::WeakPtr<Delegate> delegate_; 5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool print_debug_; 5902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool is_running_; 5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 5925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Holds an array of SocketDataProvider elements. As Mock{TCP,SSL}StreamSocket 5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// objects get instantiated, they take their data from the i'th element of this 5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// array. 5965d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)template <typename T> 5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SocketDataProviderArray { 5985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProviderArray() : next_index_(0) {} 6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) T* GetNext() { 6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_LT(next_index_, data_providers_.size()); 6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return data_providers_[next_index_++]; 6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Add(T* data_provider) { 6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(data_provider); 6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_providers_.push_back(data_provider); 6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 611c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) size_t next_index() { return next_index_; } 612c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 6135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void ResetNextIndex() { next_index_ = 0; } 6145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Index of the next |data_providers_| element to use. Not an iterator 6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // because those are invalidated on vector reallocation. 6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t next_index_; 6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SocketDataProviders to be returned. 6215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<T*> data_providers_; 6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockUDPClientSocket; 6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockTCPClientSocket; 6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockSSLClientSocket; 6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ClientSocketFactory which contains arrays of sockets of each type. 6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// You should first fill the arrays using AddMock{SSL,}Socket. When the factory 6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// is asked to create a socket, it takes next entry from appropriate array. 6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// You can use ResetNextMockIndexes to reset that next entry index for all mock 6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// socket types. 6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockClientSocketFactory : public ClientSocketFactory { 6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockClientSocketFactory(); 6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockClientSocketFactory(); 6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddSocketDataProvider(SocketDataProvider* socket); 6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddSSLSocketDataProvider(SSLSocketDataProvider* socket); 6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ResetNextMockIndexes(); 6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProviderArray<SocketDataProvider>& mock_data() { 6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return mock_data_; 6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6465f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // Note: this method is unsafe; the elements of the returned vector 6475f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // are not necessarily valid. 6485f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) const std::vector<MockSSLClientSocket*>& ssl_client_sockets() const { 6495f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) return ssl_client_sockets_; 6505f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) } 6515f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) 6525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ClientSocketFactory 6533551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) virtual scoped_ptr<DatagramClientSocket> CreateDatagramClientSocket( 6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatagramSocket::BindType bind_type, 6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const RandIntCallback& rand_int_cb, 6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NetLog* net_log, 6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const NetLog::Source& source) OVERRIDE; 6583551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) virtual scoped_ptr<StreamSocket> CreateTransportClientSocket( 6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const AddressList& addresses, 6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NetLog* net_log, 6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const NetLog::Source& source) OVERRIDE; 6623551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) virtual scoped_ptr<SSLClientSocket> CreateSSLClientSocket( 6633551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) scoped_ptr<ClientSocketHandle> transport_socket, 6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const HostPortPair& host_and_port, 6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const SSLConfig& ssl_config, 6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const SSLClientSocketContext& context) OVERRIDE; 6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void ClearSSLSessionCache() OVERRIDE; 6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProviderArray<SocketDataProvider> mock_data_; 6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProviderArray<SSLSocketDataProvider> mock_ssl_data_; 6725f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) std::vector<MockSSLClientSocket*> ssl_client_sockets_; 6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockClientSocket : public SSLClientSocket { 6765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Value returned by GetTLSUniqueChannelBinding(). 6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const char kTlsUnique[]; 6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // The BoundNetLog is needed to test LoadTimingInfo, which uses NetLog IDs as 6812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // unique socket IDs. 6822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) explicit MockClientSocket(const BoundNetLog& net_log); 6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Socket implementation. 6855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Read(IOBuffer* buf, 6865d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) = 0; 6885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Write(IOBuffer* buf, 6895d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) = 0; 691c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual int SetReceiveBufferSize(int32 size) OVERRIDE; 692c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual int SetSendBufferSize(int32 size) OVERRIDE; 6935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // StreamSocket implementation. 6955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int Connect(const CompletionCallback& callback) = 0; 6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Disconnect() OVERRIDE; 6975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool IsConnected() const OVERRIDE; 6985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool IsConnectedAndIdle() const OVERRIDE; 6995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE; 7005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int GetLocalAddress(IPEndPoint* address) const OVERRIDE; 7015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual const BoundNetLog& NetLog() const OVERRIDE; 7025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetSubresourceSpeculation() OVERRIDE {} 7035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetOmniboxSpeculation() OVERRIDE {} 7045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SSLClientSocket implementation. 70603b57e008b61dfcb1fbad3aea950ae0e001748b0Torne (Richard Coles) virtual std::string GetSessionCacheKey() const OVERRIDE; 7075f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) virtual bool InSessionCache() const OVERRIDE; 7085f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) virtual void SetHandshakeCompletionCallback(const base::Closure& cb) OVERRIDE; 7095d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual void GetSSLCertRequestInfo(SSLCertRequestInfo* cert_request_info) 7105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) OVERRIDE; 7115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int ExportKeyingMaterial(const base::StringPiece& label, 7125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool has_context, 7135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::StringPiece& context, 7145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsigned char* out, 7155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsigned int outlen) OVERRIDE; 7165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int GetTLSUniqueChannelBinding(std::string* out) OVERRIDE; 7175f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) virtual NextProtoStatus GetNextProto(std::string* proto) OVERRIDE; 7185f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) virtual ChannelIDService* GetChannelIDService() const OVERRIDE; 7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockClientSocket(); 7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RunCallbackAsync(const CompletionCallback& callback, int result); 7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RunCallback(const CompletionCallback& callback, int result); 7245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) // SSLClientSocket implementation. 726a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) virtual scoped_refptr<X509Certificate> GetUnverifiedServerCertificateChain() 727a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const OVERRIDE; 728a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 7295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // True if Connect completed successfully and Disconnect hasn't been called. 7305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool connected_; 7315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Address of the "remote" peer we're connected to. 7335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IPEndPoint peer_addr_; 7345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BoundNetLog net_log_; 736f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 7375f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) private: 738f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) base::WeakPtrFactory<MockClientSocket> weak_factory_; 739f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 740f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockClientSocket); 7415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 7425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockTCPClientSocket : public MockClientSocket, public AsyncSocket { 7445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 7455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockTCPClientSocket(const AddressList& addresses, 7465d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) net::NetLog* net_log, 7475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProvider* socket); 7485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockTCPClientSocket(); 7495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const AddressList& addresses() const { return addresses_; } 7515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Socket implementation. 7535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Read(IOBuffer* buf, 7545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 7555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 7565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Write(IOBuffer* buf, 7575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 7585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 7595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // StreamSocket implementation. 7615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int Connect(const CompletionCallback& callback) OVERRIDE; 7625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Disconnect() OVERRIDE; 7635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool IsConnected() const OVERRIDE; 7645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool IsConnectedAndIdle() const OVERRIDE; 7655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE; 7665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool WasEverUsed() const OVERRIDE; 7675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool UsingTCPFastOpen() const OVERRIDE; 7685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool WasNpnNegotiated() const OVERRIDE; 7695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool GetSSLInfo(SSLInfo* ssl_info) OVERRIDE; 7705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // AsyncSocket: 7725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnReadComplete(const MockRead& data) OVERRIDE; 773eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch virtual void OnConnectComplete(const MockConnect& data) OVERRIDE; 7745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 7765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int CompleteRead(); 7775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddressList addresses_; 7795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProvider* data_; 7815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int read_offset_; 7825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead read_data_; 7835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool need_read_data_; 7845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // True if the peer has closed the connection. This allows us to simulate 7865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the recv(..., MSG_PEEK) call in the IsConnectedAndIdle method of the real 7875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TCPClientSocket. 7885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool peer_closed_connection_; 7895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // While an asynchronous IO is pending, we save our user-buffer state. 7918bcbed890bc3ce4d7a057a8f32cab53fa534672eTorne (Richard Coles) scoped_refptr<IOBuffer> pending_buf_; 7925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pending_buf_len_; 7935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompletionCallback pending_callback_; 7945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool was_used_to_convey_data_; 795f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 796f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockTCPClientSocket); 7975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 7985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 799b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)// DeterministicSocketHelper is a helper class that can be used 800b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)// to simulate net::Socket::Read() and net::Socket::Write() 801b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)// using deterministic |data|. 802b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)// Note: This is provided as a common helper class because 803b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)// of the inheritance hierarchy of DeterministicMock[UDP,TCP]ClientSocket and a 804b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)// desire not to introduce an additional common base class. 805b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)class DeterministicSocketHelper { 806b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) public: 807b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DeterministicSocketHelper(net::NetLog* net_log, 808b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DeterministicSocketData* data); 809b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual ~DeterministicSocketHelper(); 810b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 811b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) bool write_pending() const { return write_pending_; } 812b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) bool read_pending() const { return read_pending_; } 813b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 814b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) void CompleteWrite(); 815b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) int CompleteRead(); 816b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 8175d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int Write(IOBuffer* buf, int buf_len, const CompletionCallback& callback); 8185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int Read(IOBuffer* buf, int buf_len, const CompletionCallback& callback); 819b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 820b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) const BoundNetLog& net_log() const { return net_log_; } 821b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 822b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) bool was_used_to_convey_data() const { return was_used_to_convey_data_; } 823b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 824b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) bool peer_closed_connection() const { return peer_closed_connection_; } 825b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 826b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DeterministicSocketData* data() const { return data_; } 827b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 828b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) private: 829b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) bool write_pending_; 830b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) CompletionCallback write_callback_; 831b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) int write_result_; 832b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 833b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) MockRead read_data_; 834b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 835b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) IOBuffer* read_buf_; 836b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) int read_buf_len_; 837b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) bool read_pending_; 838b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) CompletionCallback read_callback_; 839b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DeterministicSocketData* data_; 840b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) bool was_used_to_convey_data_; 841b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) bool peer_closed_connection_; 842b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) BoundNetLog net_log_; 843b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)}; 844b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 845b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)// Mock UDP socket to be used in conjunction with DeterministicSocketData. 846b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)class DeterministicMockUDPClientSocket 847b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) : public DatagramClientSocket, 848b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) public AsyncSocket, 849b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) public DeterministicSocketData::Delegate, 850b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) public base::SupportsWeakPtr<DeterministicMockUDPClientSocket> { 851b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) public: 852b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DeterministicMockUDPClientSocket(net::NetLog* net_log, 853b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DeterministicSocketData* data); 854b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual ~DeterministicMockUDPClientSocket(); 855b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 856b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // DeterministicSocketData::Delegate: 857b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual bool WritePending() const OVERRIDE; 858b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual bool ReadPending() const OVERRIDE; 859b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual void CompleteWrite() OVERRIDE; 860b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual int CompleteRead() OVERRIDE; 861b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 862b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // Socket implementation. 8635d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Read(IOBuffer* buf, 8645d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 865b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 8665d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Write(IOBuffer* buf, 8675d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 868b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 869c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual int SetReceiveBufferSize(int32 size) OVERRIDE; 870c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual int SetSendBufferSize(int32 size) OVERRIDE; 871b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 872b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // DatagramSocket implementation. 873b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual void Close() OVERRIDE; 874b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE; 875b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual int GetLocalAddress(IPEndPoint* address) const OVERRIDE; 876b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual const BoundNetLog& NetLog() const OVERRIDE; 877b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 878b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // DatagramClientSocket implementation. 879b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual int Connect(const IPEndPoint& address) OVERRIDE; 880b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 881b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // AsyncSocket implementation. 882b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual void OnReadComplete(const MockRead& data) OVERRIDE; 883eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch virtual void OnConnectComplete(const MockConnect& data) OVERRIDE; 884b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 8855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void set_source_port(int port) { source_port_ = port; } 8865d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 887b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) private: 888b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) bool connected_; 889b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) IPEndPoint peer_address_; 890b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DeterministicSocketHelper helper_; 8915d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int source_port_; // Ephemeral source port. 892f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 893f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(DeterministicMockUDPClientSocket); 894b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)}; 895b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) 896b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)// Mock TCP socket to be used in conjunction with DeterministicSocketData. 8975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DeterministicMockTCPClientSocket 8985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : public MockClientSocket, 8995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public AsyncSocket, 900b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) public DeterministicSocketData::Delegate, 9015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public base::SupportsWeakPtr<DeterministicMockTCPClientSocket> { 9025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 9035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DeterministicMockTCPClientSocket(net::NetLog* net_log, 9045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DeterministicSocketData* data); 9055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DeterministicMockTCPClientSocket(); 9065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 907b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // DeterministicSocketData::Delegate: 908b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual bool WritePending() const OVERRIDE; 909b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual bool ReadPending() const OVERRIDE; 910b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual void CompleteWrite() OVERRIDE; 911b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) virtual int CompleteRead() OVERRIDE; 9125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Socket: 9145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Write(IOBuffer* buf, 9155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 9165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 9175d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Read(IOBuffer* buf, 9185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 9195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 9205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 921b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) // StreamSocket: 9225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int Connect(const CompletionCallback& callback) OVERRIDE; 9235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Disconnect() OVERRIDE; 9245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool IsConnected() const OVERRIDE; 9255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool IsConnectedAndIdle() const OVERRIDE; 9265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool WasEverUsed() const OVERRIDE; 9275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool UsingTCPFastOpen() const OVERRIDE; 9285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool WasNpnNegotiated() const OVERRIDE; 9295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool GetSSLInfo(SSLInfo* ssl_info) OVERRIDE; 9305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // AsyncSocket: 9325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnReadComplete(const MockRead& data) OVERRIDE; 933eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch virtual void OnConnectComplete(const MockConnect& data) OVERRIDE; 9345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 936b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DeterministicSocketHelper helper_; 937f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 938f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(DeterministicMockTCPClientSocket); 9395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 9405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockSSLClientSocket : public MockClientSocket, public AsyncSocket { 9425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 9435d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockSSLClientSocket(scoped_ptr<ClientSocketHandle> transport_socket, 9445d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const HostPortPair& host_and_port, 9455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const SSLConfig& ssl_config, 9465d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) SSLSocketDataProvider* socket); 9475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockSSLClientSocket(); 9485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Socket implementation. 9505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Read(IOBuffer* buf, 9515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 9525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 9535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Write(IOBuffer* buf, 9545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 9555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 9565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // StreamSocket implementation. 9585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int Connect(const CompletionCallback& callback) OVERRIDE; 9595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Disconnect() OVERRIDE; 9605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool IsConnected() const OVERRIDE; 9615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool WasEverUsed() const OVERRIDE; 9625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool UsingTCPFastOpen() const OVERRIDE; 9635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE; 9645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool WasNpnNegotiated() const OVERRIDE; 9655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool GetSSLInfo(SSLInfo* ssl_info) OVERRIDE; 9665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SSLClientSocket implementation. 96803b57e008b61dfcb1fbad3aea950ae0e001748b0Torne (Richard Coles) virtual std::string GetSessionCacheKey() const OVERRIDE; 9695f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) virtual bool InSessionCache() const OVERRIDE; 9705f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) virtual void SetHandshakeCompletionCallback(const base::Closure& cb) OVERRIDE; 9715d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual void GetSSLCertRequestInfo(SSLCertRequestInfo* cert_request_info) 9725d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) OVERRIDE; 9735f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) virtual NextProtoStatus GetNextProto(std::string* proto) OVERRIDE; 9745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool set_was_npn_negotiated(bool negotiated) OVERRIDE; 9755d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual void set_protocol_negotiated(NextProto protocol_negotiated) OVERRIDE; 9765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual NextProto GetNegotiatedProtocol() const OVERRIDE; 9775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This MockSocket does not implement the manual async IO feature. 9795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnReadComplete(const MockRead& data) OVERRIDE; 980eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch virtual void OnConnectComplete(const MockConnect& data) OVERRIDE; 9815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool WasChannelIDSent() const OVERRIDE; 9835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void set_channel_id_sent(bool channel_id_sent) OVERRIDE; 9845f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) virtual ChannelIDService* GetChannelIDService() const OVERRIDE; 9855f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) 9865f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) bool reached_connect() const { return reached_connect_; } 9875f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) 9885f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // Resumes the connection of a socket that was paused for testing. 9895f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // |connect_callback_| should be set before invoking this method. 9905f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) void RestartPausedConnect(); 9915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 9935f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) enum ConnectState { 9945f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) STATE_NONE, 9955f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) STATE_SSL_CONNECT, 9965f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) STATE_SSL_CONNECT_COMPLETE, 9975f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) }; 9985f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) 9995f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) void OnIOComplete(int result); 10005f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) 10015f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // Runs the state transistion loop. 10025f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) int DoConnectLoop(int result); 10035f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) 10045f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) int DoSSLConnect(); 10055f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) int DoSSLConnectComplete(int result); 10065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<ClientSocketHandle> transport_; 100803b57e008b61dfcb1fbad3aea950ae0e001748b0Torne (Richard Coles) HostPortPair host_port_pair_; 10095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SSLSocketDataProvider* data_; 10105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool is_npn_state_set_; 10115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool new_npn_value_; 10125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool is_protocol_negotiated_set_; 10135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NextProto protocol_negotiated_; 1014f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 10155f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) CompletionCallback connect_callback_; 10165f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // Indicates what state of Connect the socket should enter. 10175f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) ConnectState next_connect_state_; 10185f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) // True if the Connect method has been called on the socket. 10195f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) bool reached_connect_; 10205f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) 10215f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) base::Closure handshake_completion_callback_; 10225f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) 10235f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) base::WeakPtrFactory<MockSSLClientSocket> weak_factory_; 10245f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) 1025f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockSSLClientSocket); 10265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 10275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)class MockUDPClientSocket : public DatagramClientSocket, public AsyncSocket { 10295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 10305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockUDPClientSocket(SocketDataProvider* data, net::NetLog* net_log); 10315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockUDPClientSocket(); 10325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Socket implementation. 10345d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Read(IOBuffer* buf, 10355d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 10365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 10375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual int Write(IOBuffer* buf, 10385d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int buf_len, 10395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) OVERRIDE; 1040c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual int SetReceiveBufferSize(int32 size) OVERRIDE; 1041c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch virtual int SetSendBufferSize(int32 size) OVERRIDE; 10425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // DatagramSocket implementation. 10445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Close() OVERRIDE; 10455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE; 10465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int GetLocalAddress(IPEndPoint* address) const OVERRIDE; 10475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual const BoundNetLog& NetLog() const OVERRIDE; 10485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // DatagramClientSocket implementation. 10505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int Connect(const IPEndPoint& address) OVERRIDE; 10515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // AsyncSocket implementation. 10535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnReadComplete(const MockRead& data) OVERRIDE; 1054eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch virtual void OnConnectComplete(const MockConnect& data) OVERRIDE; 10555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) void set_source_port(int port) { source_port_ = port;} 10575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 10585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 10595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int CompleteRead(); 10605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RunCallbackAsync(const CompletionCallback& callback, int result); 10625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RunCallback(const CompletionCallback& callback, int result); 10635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool connected_; 10655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProvider* data_; 10665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int read_offset_; 10675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead read_data_; 10685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool need_read_data_; 10695d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int source_port_; // Ephemeral source port. 10705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Address of the "remote" peer we're connected to. 10722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IPEndPoint peer_addr_; 10732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 10745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // While an asynchronous IO is pending, we save our user-buffer state. 10758bcbed890bc3ce4d7a057a8f32cab53fa534672eTorne (Richard Coles) scoped_refptr<IOBuffer> pending_buf_; 10765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pending_buf_len_; 10775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompletionCallback pending_callback_; 10785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BoundNetLog net_log_; 10805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WeakPtrFactory<MockUDPClientSocket> weak_factory_; 10825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockUDPClientSocket); 10845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 10855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class TestSocketRequest : public TestCompletionCallbackBase { 10875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 10885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestSocketRequest(std::vector<TestSocketRequest*>* request_order, 10895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t* completion_count); 10905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~TestSocketRequest(); 10915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ClientSocketHandle* handle() { return &handle_; } 10935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const net::CompletionCallback& callback() const { return callback_; } 10955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 10975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnComplete(int result); 10985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ClientSocketHandle handle_; 11005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<TestSocketRequest*>* request_order_; 11015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t* completion_count_; 11025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompletionCallback callback_; 11035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(TestSocketRequest); 11055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 11065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ClientSocketPoolTest { 11085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 11095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) enum KeepAlive { 11105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) KEEP_ALIVE, 11115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A socket will be disconnected in addition to handle being reset. 11135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NO_KEEP_ALIVE, 11145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 11155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const int kIndexOutOfBounds; 11175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const int kRequestNotFound; 11185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ClientSocketPoolTest(); 11205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~ClientSocketPoolTest(); 11215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11223551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) template <typename PoolType> 11233551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) int StartRequestUsingPool( 11243551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) PoolType* socket_pool, 11253551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) const std::string& group_name, 11263551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) RequestPriority priority, 11273551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) const scoped_refptr<typename PoolType::SocketParams>& socket_params) { 11285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(socket_pool); 11295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) TestSocketRequest* request = 11305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) new TestSocketRequest(&request_order_, &completion_count_); 11315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) requests_.push_back(request); 11325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int rv = request->handle()->Init(group_name, 11335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) socket_params, 11345d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) priority, 11355d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) request->callback(), 11365d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) socket_pool, 11375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) BoundNetLog()); 11385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (rv != ERR_IO_PENDING) 11395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) request_order_.push_back(request); 11405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return rv; 11415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 11425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Provided there were n requests started, takes |index| in range 1..n 11445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and returns order in which that request completed, in range 1..n, 11455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // or kIndexOutOfBounds if |index| is out of bounds, or kRequestNotFound 11465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // if that request did not complete (for example was canceled). 11475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int GetOrderOfRequest(size_t index) const; 11485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Resets first initialized socket handle from |requests_|. If found such 11505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // a handle, returns true. 11515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool ReleaseOneConnection(KeepAlive keep_alive); 11525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Releases connections until there is nothing to release. 11545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ReleaseAllConnections(KeepAlive keep_alive); 11555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11565e3f23d412006dc4db4e659864679f29341e113fTorne (Richard Coles) // Note that this uses 0-based indices, while GetOrderOfRequest takes and 11575e3f23d412006dc4db4e659864679f29341e113fTorne (Richard Coles) // returns 0-based indices. 11585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestSocketRequest* request(int i) { return requests_[i]; } 11595e3f23d412006dc4db4e659864679f29341e113fTorne (Richard Coles) 11605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t requests_size() const { return requests_.size(); } 11615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedVector<TestSocketRequest>* requests() { return &requests_; } 11625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t completion_count() const { return completion_count_; } 11635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 11655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedVector<TestSocketRequest> requests_; 11665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<TestSocketRequest*> request_order_; 11675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t completion_count_; 1168f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 1169f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ClientSocketPoolTest); 11705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 11715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11723551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)class MockTransportSocketParams 11733551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) : public base::RefCounted<MockTransportSocketParams> { 11743551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) private: 11753551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) friend class base::RefCounted<MockTransportSocketParams>; 11763551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) ~MockTransportSocketParams() {} 1177f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 1178f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockTransportSocketParams); 11793551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)}; 11803551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 11815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockTransportClientSocketPool : public TransportClientSocketPool { 11825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 11833551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) typedef MockTransportSocketParams SocketParams; 11843551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) 11855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) class MockConnectJob { 11865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 11875d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockConnectJob(scoped_ptr<StreamSocket> socket, 11885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ClientSocketHandle* handle, 11895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback); 11905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~MockConnectJob(); 11915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int Connect(); 11935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool CancelHandle(const ClientSocketHandle* handle); 11945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 11965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnConnect(int rv); 11975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<StreamSocket> socket_; 11995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ClientSocketHandle* handle_; 12005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompletionCallback user_callback_; 12015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockConnectJob); 12035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 12045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12055d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockTransportClientSocketPool(int max_sockets, 12065d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int max_sockets_per_group, 12075d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ClientSocketPoolHistograms* histograms, 12085d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ClientSocketFactory* socket_factory); 12095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockTransportClientSocketPool(); 12115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12123551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) RequestPriority last_request_priority() const { 12133551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) return last_request_priority_; 12143551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) } 12155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int release_count() const { return release_count_; } 12165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int cancel_count() const { return cancel_count_; } 12175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TransportClientSocketPool implementation. 12195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int RequestSocket(const std::string& group_name, 12205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const void* socket_params, 12215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RequestPriority priority, 12225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ClientSocketHandle* handle, 12235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback, 12245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const BoundNetLog& net_log) OVERRIDE; 12255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void CancelRequest(const std::string& group_name, 12275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ClientSocketHandle* handle) OVERRIDE; 12285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void ReleaseSocket(const std::string& group_name, 12293551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) scoped_ptr<StreamSocket> socket, 12303551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) int id) OVERRIDE; 12315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 12335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ClientSocketFactory* client_socket_factory_; 12345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedVector<MockConnectJob> job_list_; 12353551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) RequestPriority last_request_priority_; 12365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int release_count_; 12375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int cancel_count_; 12385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockTransportClientSocketPool); 12405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 12415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DeterministicMockClientSocketFactory : public ClientSocketFactory { 12435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 12445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DeterministicMockClientSocketFactory(); 12455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DeterministicMockClientSocketFactory(); 12465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddSocketDataProvider(DeterministicSocketData* socket); 12485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddSSLSocketDataProvider(SSLSocketDataProvider* socket); 12495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ResetNextMockIndexes(); 12505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Return |index|-th MockSSLClientSocket (starting from 0) that the factory 12525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // created. 12535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockSSLClientSocket* GetMockSSLClientSocket(size_t index) const; 12545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProviderArray<DeterministicSocketData>& mock_data() { 12565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return mock_data_; 12575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 12585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<DeterministicMockTCPClientSocket*>& tcp_client_sockets() { 12595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return tcp_client_sockets_; 12605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1261b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) std::vector<DeterministicMockUDPClientSocket*>& udp_client_sockets() { 1262b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) return udp_client_sockets_; 1263b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) } 12645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ClientSocketFactory 12663551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) virtual scoped_ptr<DatagramClientSocket> CreateDatagramClientSocket( 12675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatagramSocket::BindType bind_type, 12685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const RandIntCallback& rand_int_cb, 12695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NetLog* net_log, 12705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const NetLog::Source& source) OVERRIDE; 12713551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) virtual scoped_ptr<StreamSocket> CreateTransportClientSocket( 12725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const AddressList& addresses, 12735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NetLog* net_log, 12745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const NetLog::Source& source) OVERRIDE; 12753551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) virtual scoped_ptr<SSLClientSocket> CreateSSLClientSocket( 12763551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) scoped_ptr<ClientSocketHandle> transport_socket, 12775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const HostPortPair& host_and_port, 12785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const SSLConfig& ssl_config, 12795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const SSLClientSocketContext& context) OVERRIDE; 12805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void ClearSSLSessionCache() OVERRIDE; 12815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 12835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProviderArray<DeterministicSocketData> mock_data_; 12845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDataProviderArray<SSLSocketDataProvider> mock_ssl_data_; 12855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Store pointers to handed out sockets in case the test wants to get them. 12875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<DeterministicMockTCPClientSocket*> tcp_client_sockets_; 1288b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) std::vector<DeterministicMockUDPClientSocket*> udp_client_sockets_; 12895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<MockSSLClientSocket*> ssl_client_sockets_; 1290f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 1291f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(DeterministicMockClientSocketFactory); 12925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 12935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 12945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockSOCKSClientSocketPool : public SOCKSClientSocketPool { 12955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 12965d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) MockSOCKSClientSocketPool(int max_sockets, 12975d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int max_sockets_per_group, 12985d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ClientSocketPoolHistograms* histograms, 12995d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) TransportClientSocketPool* transport_pool); 13005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockSOCKSClientSocketPool(); 13025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SOCKSClientSocketPool implementation. 13045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int RequestSocket(const std::string& group_name, 13055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const void* socket_params, 13065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RequestPriority priority, 13075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ClientSocketHandle* handle, 13085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback, 13095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const BoundNetLog& net_log) OVERRIDE; 13105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void CancelRequest(const std::string& group_name, 13125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ClientSocketHandle* handle) OVERRIDE; 13135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void ReleaseSocket(const std::string& group_name, 13143551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) scoped_ptr<StreamSocket> socket, 13153551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) int id) OVERRIDE; 13165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 13185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TransportClientSocketPool* const transport_pool_; 13195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockSOCKSClientSocketPool); 13215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 13225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Constants for a successful SOCKS v5 handshake. 13245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const char kSOCKS5GreetRequest[]; 13255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const int kSOCKS5GreetRequestLength; 13265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const char kSOCKS5GreetResponse[]; 13285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const int kSOCKS5GreetResponseLength; 13295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const char kSOCKS5OkRequest[]; 13315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const int kSOCKS5OkRequestLength; 13325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const char kSOCKS5OkResponse[]; 13345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern const int kSOCKS5OkResponseLength; 13355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 13375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 13385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // NET_SOCKET_SOCKET_TEST_UTIL_H_ 1339