OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_ | 5 #ifndef NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_ |
6 #define NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_ | 6 #define NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
56 QuicPacketWriterWrapper* test_writer_; | 56 QuicPacketWriterWrapper* test_writer_; |
57 | 57 |
58 DISALLOW_COPY_AND_ASSIGN(MockableQuicClient); | 58 DISALLOW_COPY_AND_ASSIGN(MockableQuicClient); |
59 }; | 59 }; |
60 | 60 |
61 // A toy QUIC client used for testing, mostly following the SimpleClient APIs. | 61 // A toy QUIC client used for testing, mostly following the SimpleClient APIs. |
62 class QuicTestClient : public SimpleClient, | 62 class QuicTestClient : public SimpleClient, |
63 public QuicDataStream::Visitor { | 63 public QuicDataStream::Visitor { |
64 public: | 64 public: |
65 QuicTestClient(IPEndPoint server_address, | 65 QuicTestClient(IPEndPoint server_address, |
66 const string& server_hostname, | 66 const std::string& server_hostname, |
67 bool secure, | 67 bool secure, |
68 const QuicVersionVector& supported_versions); | 68 const QuicVersionVector& supported_versions); |
69 QuicTestClient(IPEndPoint server_address, | 69 QuicTestClient(IPEndPoint server_address, |
70 const string& server_hostname, | 70 const std::string& server_hostname, |
71 bool secure, | 71 bool secure, |
72 const QuicConfig& config, | 72 const QuicConfig& config, |
73 const QuicVersionVector& supported_versions); | 73 const QuicVersionVector& supported_versions); |
74 | 74 |
75 ~QuicTestClient() override; | 75 ~QuicTestClient() override; |
76 | 76 |
77 // ExpectCertificates controls whether the server is expected to provide | 77 // ExpectCertificates controls whether the server is expected to provide |
78 // certificates. The certificates, if any, are not verified, but the common | 78 // certificates. The certificates, if any, are not verified, but the common |
79 // name is recorded and available with |cert_common_name()|. | 79 // name is recorded and available with |cert_common_name()|. |
80 void ExpectCertificates(bool on); | 80 void ExpectCertificates(bool on); |
81 | 81 |
82 // Sets the |user_agent_id| of the |client_|. | 82 // Sets the |user_agent_id| of the |client_|. |
83 void SetUserAgentID(const string& user_agent_id); | 83 void SetUserAgentID(const std::string& user_agent_id); |
84 | 84 |
85 // Wraps data in a quic packet and sends it. | 85 // Wraps data in a quic packet and sends it. |
86 ssize_t SendData(string data, bool last_data); | 86 ssize_t SendData(std::string data, bool last_data); |
87 | 87 |
88 // From SimpleClient | 88 // From SimpleClient |
89 // Clears any outstanding state and sends a simple GET of 'uri' to the | 89 // Clears any outstanding state and sends a simple GET of 'uri' to the |
90 // server. Returns 0 if the request failed and no bytes were written. | 90 // server. Returns 0 if the request failed and no bytes were written. |
91 ssize_t SendRequest(const string& uri) override; | 91 ssize_t SendRequest(const std::string& uri) override; |
92 ssize_t SendMessage(const HTTPMessage& message) override; | 92 ssize_t SendMessage(const HTTPMessage& message) override; |
93 string SendCustomSynchronousRequest(const HTTPMessage& message) override; | 93 std::string SendCustomSynchronousRequest(const HTTPMessage& message) override; |
94 string SendSynchronousRequest(const string& uri) override; | 94 std::string SendSynchronousRequest(const std::string& uri) override; |
95 void Connect() override; | 95 void Connect() override; |
96 void ResetConnection() override; | 96 void ResetConnection() override; |
97 void Disconnect() override; | 97 void Disconnect() override; |
98 IPEndPoint LocalSocketAddress() const override; | 98 IPEndPoint LocalSocketAddress() const override; |
99 void ClearPerRequestState() override; | 99 void ClearPerRequestState() override; |
100 void WaitForResponseForMs(int timeout_ms) override; | 100 void WaitForResponseForMs(int timeout_ms) override; |
101 void WaitForInitialResponseForMs(int timeout_ms) override; | 101 void WaitForInitialResponseForMs(int timeout_ms) override; |
102 ssize_t Send(const void* buffer, size_t size) override; | 102 ssize_t Send(const void* buffer, size_t size) override; |
103 bool response_complete() const override; | 103 bool response_complete() const override; |
104 bool response_headers_complete() const override; | 104 bool response_headers_complete() const override; |
105 const BalsaHeaders* response_headers() const override; | 105 const BalsaHeaders* response_headers() const override; |
106 int64 response_size() const override; | 106 int64 response_size() const override; |
107 int response_header_size() const override; | 107 int response_header_size() const override; |
108 int64 response_body_size() const override; | 108 int64 response_body_size() const override; |
109 size_t bytes_read() const override; | 109 size_t bytes_read() const override; |
110 size_t bytes_written() const override; | 110 size_t bytes_written() const override; |
111 bool buffer_body() const override; | 111 bool buffer_body() const override; |
112 void set_buffer_body(bool buffer_body) override; | 112 void set_buffer_body(bool buffer_body) override; |
113 bool ServerInLameDuckMode() const override; | 113 bool ServerInLameDuckMode() const override; |
114 const string& response_body() override; | 114 const std::string& response_body() override; |
115 bool connected() const override; | 115 bool connected() const override; |
116 // These functions are all unimplemented functions from SimpleClient, and log | 116 // These functions are all unimplemented functions from SimpleClient, and log |
117 // DFATAL if called by users of SimpleClient. | 117 // DFATAL if called by users of SimpleClient. |
118 ssize_t SendAndWaitForResponse(const void* buffer, size_t size) override; | 118 ssize_t SendAndWaitForResponse(const void* buffer, size_t size) override; |
119 void Bind(IPEndPoint* local_address) override; | 119 void Bind(IPEndPoint* local_address) override; |
120 string SerializeMessage(const HTTPMessage& message) override; | 120 std::string SerializeMessage(const HTTPMessage& message) override; |
121 IPAddressNumber bind_to_address() const override; | 121 IPAddressNumber bind_to_address() const override; |
122 void set_bind_to_address(IPAddressNumber address) override; | 122 void set_bind_to_address(IPAddressNumber address) override; |
123 const IPEndPoint& address() const override; | 123 const IPEndPoint& address() const override; |
124 size_t requests_sent() const override; | 124 size_t requests_sent() const override; |
125 | 125 |
126 // From QuicDataStream::Visitor | 126 // From QuicDataStream::Visitor |
127 void OnClose(QuicDataStream* stream) override; | 127 void OnClose(QuicDataStream* stream) override; |
128 | 128 |
129 // Configures client_ to take ownership of and use the writer. | 129 // Configures client_ to take ownership of and use the writer. |
130 // Must be called before initial connect. | 130 // Must be called before initial connect. |
131 void UseWriter(QuicPacketWriterWrapper* writer); | 131 void UseWriter(QuicPacketWriterWrapper* writer); |
132 // If the given ConnectionId is nonzero, configures client_ to use a specific | 132 // If the given ConnectionId is nonzero, configures client_ to use a specific |
133 // ConnectionId instead of a random one. | 133 // ConnectionId instead of a random one. |
134 void UseConnectionId(QuicConnectionId connection_id); | 134 void UseConnectionId(QuicConnectionId connection_id); |
135 | 135 |
136 // Returns nullptr if the maximum number of streams have already been created. | 136 // Returns nullptr if the maximum number of streams have already been created. |
137 QuicSpdyClientStream* GetOrCreateStream(); | 137 QuicSpdyClientStream* GetOrCreateStream(); |
138 | 138 |
139 QuicRstStreamErrorCode stream_error() { return stream_error_; } | 139 QuicRstStreamErrorCode stream_error() { return stream_error_; } |
140 QuicErrorCode connection_error(); | 140 QuicErrorCode connection_error(); |
141 | 141 |
142 MockableQuicClient* client(); | 142 MockableQuicClient* client(); |
143 | 143 |
144 // cert_common_name returns the common name value of the server's certificate, | 144 // cert_common_name returns the common name value of the server's certificate, |
145 // or the empty string if no certificate was presented. | 145 // or the empty string if no certificate was presented. |
146 const string& cert_common_name() const; | 146 const std::string& cert_common_name() const; |
147 | 147 |
148 // Get the server config map. | 148 // Get the server config map. |
149 QuicTagValueMap GetServerConfig() const; | 149 QuicTagValueMap GetServerConfig() const; |
150 | 150 |
151 void set_auto_reconnect(bool reconnect) { auto_reconnect_ = reconnect; } | 151 void set_auto_reconnect(bool reconnect) { auto_reconnect_ = reconnect; } |
152 | 152 |
153 void set_priority(QuicPriority priority) { priority_ = priority; } | 153 void set_priority(QuicPriority priority) { priority_ = priority; } |
154 | 154 |
155 // Sets client's FEC policy. This policy applies to the data stream(s), and | 155 // Sets client's FEC policy. This policy applies to the data stream(s), and |
156 // also to the headers and crypto streams. | 156 // also to the headers and crypto streams. |
(...skipping 14 matching lines...) Expand all Loading... |
171 EpollServer epoll_server_; | 171 EpollServer epoll_server_; |
172 scoped_ptr<MockableQuicClient> client_; // The actual client | 172 scoped_ptr<MockableQuicClient> client_; // The actual client |
173 QuicSpdyClientStream* stream_; | 173 QuicSpdyClientStream* stream_; |
174 | 174 |
175 QuicRstStreamErrorCode stream_error_; | 175 QuicRstStreamErrorCode stream_error_; |
176 | 176 |
177 bool response_complete_; | 177 bool response_complete_; |
178 bool response_headers_complete_; | 178 bool response_headers_complete_; |
179 BalsaHeaders headers_; | 179 BalsaHeaders headers_; |
180 QuicPriority priority_; | 180 QuicPriority priority_; |
181 string response_; | 181 std::string response_; |
182 uint64 bytes_read_; | 182 uint64 bytes_read_; |
183 uint64 bytes_written_; | 183 uint64 bytes_written_; |
184 // The number of uncompressed HTTP header bytes received. | 184 // The number of uncompressed HTTP header bytes received. |
185 int response_header_size_; | 185 int response_header_size_; |
186 // The number of HTTP body bytes received. | 186 // The number of HTTP body bytes received. |
187 int64 response_body_size_; | 187 int64 response_body_size_; |
188 // True if we tried to connect already since the last call to Disconnect(). | 188 // True if we tried to connect already since the last call to Disconnect(). |
189 bool connect_attempted_; | 189 bool connect_attempted_; |
190 bool secure_; | 190 bool secure_; |
191 // The client will auto-connect exactly once before sending data. If | 191 // The client will auto-connect exactly once before sending data. If |
(...skipping 10 matching lines...) Expand all Loading... |
202 | 202 |
203 DISALLOW_COPY_AND_ASSIGN(QuicTestClient); | 203 DISALLOW_COPY_AND_ASSIGN(QuicTestClient); |
204 }; | 204 }; |
205 | 205 |
206 } // namespace test | 206 } // namespace test |
207 | 207 |
208 } // namespace tools | 208 } // namespace tools |
209 } // namespace net | 209 } // namespace net |
210 | 210 |
211 #endif // NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_ | 211 #endif // NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_ |
OLD | NEW |