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 // A toy client, which connects to a specified port and sends QUIC | 5 // A toy client, which connects to a specified port and sends QUIC |
6 // request to that endpoint. | 6 // request to that endpoint. |
7 | 7 |
8 #ifndef NET_TOOLS_QUIC_QUIC_SIMPLE_CLIENT_H_ | 8 #ifndef NET_TOOLS_QUIC_QUIC_SIMPLE_CLIENT_H_ |
9 #define NET_TOOLS_QUIC_QUIC_SIMPLE_CLIENT_H_ | 9 #define NET_TOOLS_QUIC_QUIC_SIMPLE_CLIENT_H_ |
10 | 10 |
11 #include <string> | 11 #include <string> |
12 | 12 |
13 #include "base/basictypes.h" | 13 #include "base/basictypes.h" |
14 #include "base/command_line.h" | 14 #include "base/command_line.h" |
15 #include "base/memory/scoped_ptr.h" | 15 #include "base/memory/scoped_ptr.h" |
16 #include "base/strings/string_piece.h" | 16 #include "base/strings/string_piece.h" |
17 #include "net/base/io_buffer.h" | 17 #include "net/base/io_buffer.h" |
18 #include "net/base/ip_endpoint.h" | 18 #include "net/base/ip_endpoint.h" |
19 #include "net/base/net_log.h" | 19 #include "net/base/net_log.h" |
20 #include "net/http/http_response_headers.h" | 20 #include "net/http/http_response_headers.h" |
21 #include "net/quic/crypto/crypto_handshake.h" | 21 #include "net/quic/crypto/crypto_handshake.h" |
22 #include "net/quic/quic_config.h" | 22 #include "net/quic/quic_config.h" |
23 #include "net/quic/quic_framer.h" | 23 #include "net/quic/quic_framer.h" |
24 #include "net/quic/quic_packet_creator.h" | 24 #include "net/quic/quic_packet_creator.h" |
25 #include "net/quic/quic_packet_reader.h" | 25 #include "net/quic/quic_packet_reader.h" |
26 #include "net/tools/quic/quic_simple_client_session.h" | 26 #include "net/tools/quic/quic_client_session.h" |
27 #include "net/tools/quic/quic_simple_client_stream.h" | 27 #include "net/tools/quic/quic_spdy_client_stream.h" |
28 | 28 |
29 namespace net { | 29 namespace net { |
30 | 30 |
31 struct HttpRequestInfo; | 31 struct HttpRequestInfo; |
32 class ProofVerifier; | 32 class ProofVerifier; |
33 class QuicServerId; | 33 class QuicServerId; |
34 class QuicConnectionHelper; | 34 class QuicConnectionHelper; |
35 class UDPClientSocket; | 35 class UDPClientSocket; |
36 | 36 |
37 namespace tools { | 37 namespace tools { |
(...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
94 // Sends an HTTP request and waits for response before returning. | 94 // Sends an HTTP request and waits for response before returning. |
95 void SendRequestAndWaitForResponse(const HttpRequestInfo& headers, | 95 void SendRequestAndWaitForResponse(const HttpRequestInfo& headers, |
96 base::StringPiece body, | 96 base::StringPiece body, |
97 bool fin); | 97 bool fin); |
98 | 98 |
99 // Sends a request simple GET for each URL in |args|, and then waits for | 99 // Sends a request simple GET for each URL in |args|, and then waits for |
100 // each to complete. | 100 // each to complete. |
101 void SendRequestsAndWaitForResponse( | 101 void SendRequestsAndWaitForResponse( |
102 const base::CommandLine::StringVector& url_list); | 102 const base::CommandLine::StringVector& url_list); |
103 | 103 |
104 // Returns a newly created QuicSimpleClientStream, owned by the | 104 // Returns a newly created QuicSpdyClientStream, owned by the |
105 // QuicSimpleClient. | 105 // QuicSimpleClient. |
106 QuicSimpleClientStream* CreateReliableClientStream(); | 106 QuicSpdyClientStream* CreateReliableClientStream(); |
107 | 107 |
108 // Wait for events until the stream with the given ID is closed. | 108 // Wait for events until the stream with the given ID is closed. |
109 void WaitForStreamToClose(QuicStreamId id); | 109 void WaitForStreamToClose(QuicStreamId id); |
110 | 110 |
111 // Wait for events until the handshake is confirmed. | 111 // Wait for events until the handshake is confirmed. |
112 void WaitForCryptoHandshakeConfirmed(); | 112 void WaitForCryptoHandshakeConfirmed(); |
113 | 113 |
114 // Wait up to 50ms, and handle any events which occur. | 114 // Wait up to 50ms, and handle any events which occur. |
115 // Returns true if there are any outstanding requests. | 115 // Returns true if there are any outstanding requests. |
116 bool WaitForEvents(); | 116 bool WaitForEvents(); |
117 | 117 |
118 // QuicPacketReader::Visitor | 118 // QuicPacketReader::Visitor |
119 void OnReadError(int result) override; | 119 void OnReadError(int result) override; |
120 bool OnPacket(const QuicEncryptedPacket& packet, | 120 bool OnPacket(const QuicEncryptedPacket& packet, |
121 IPEndPoint local_address, | 121 IPEndPoint local_address, |
122 IPEndPoint peer_address) override; | 122 IPEndPoint peer_address) override; |
123 | 123 |
124 // QuicDataStream::Visitor | 124 // QuicDataStream::Visitor |
125 void OnClose(QuicDataStream* stream) override; | 125 void OnClose(QuicDataStream* stream) override; |
126 | 126 |
127 QuicSimpleClientSession* session() { return session_.get(); } | 127 QuicClientSession* session() { return session_.get(); } |
128 | 128 |
129 bool connected() const; | 129 bool connected() const; |
130 bool goaway_received() const; | 130 bool goaway_received() const; |
131 | 131 |
132 void set_bind_to_address(IPAddressNumber address) { | 132 void set_bind_to_address(IPAddressNumber address) { |
133 bind_to_address_ = address; | 133 bind_to_address_ = address; |
134 } | 134 } |
135 | 135 |
136 IPAddressNumber bind_to_address() const { return bind_to_address_; } | 136 IPAddressNumber bind_to_address() const { return bind_to_address_; } |
137 | 137 |
(...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
231 // If initialized, the address to bind to. | 231 // If initialized, the address to bind to. |
232 IPAddressNumber bind_to_address_; | 232 IPAddressNumber bind_to_address_; |
233 // Local port to bind to. Initialize to 0. | 233 // Local port to bind to. Initialize to 0. |
234 int local_port_; | 234 int local_port_; |
235 | 235 |
236 // Writer used to actually send packets to the wire. Needs to outlive | 236 // Writer used to actually send packets to the wire. Needs to outlive |
237 // |session_|. | 237 // |session_|. |
238 scoped_ptr<QuicPacketWriter> writer_; | 238 scoped_ptr<QuicPacketWriter> writer_; |
239 | 239 |
240 // Session which manages streams. | 240 // Session which manages streams. |
241 scoped_ptr<QuicSimpleClientSession> session_; | 241 scoped_ptr<QuicClientSession> session_; |
242 | 242 |
243 // UDP socket connected to the server. | 243 // UDP socket connected to the server. |
244 scoped_ptr<UDPClientSocket> socket_; | 244 scoped_ptr<UDPClientSocket> socket_; |
245 | 245 |
246 // Connection on the socket. Owned by |session_|. | 246 // Connection on the socket. Owned by |session_|. |
247 QuicConnection* connection_; | 247 QuicConnection* connection_; |
248 | 248 |
249 // Helper to be used by created connections. | 249 // Helper to be used by created connections. |
250 scoped_ptr<QuicConnectionHelper> helper_; | 250 scoped_ptr<QuicConnectionHelper> helper_; |
251 | 251 |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
286 | 286 |
287 base::WeakPtrFactory<QuicSimpleClient> weak_factory_; | 287 base::WeakPtrFactory<QuicSimpleClient> weak_factory_; |
288 | 288 |
289 DISALLOW_COPY_AND_ASSIGN(QuicSimpleClient); | 289 DISALLOW_COPY_AND_ASSIGN(QuicSimpleClient); |
290 }; | 290 }; |
291 | 291 |
292 } // namespace tools | 292 } // namespace tools |
293 } // namespace net | 293 } // namespace net |
294 | 294 |
295 #endif // NET_TOOLS_QUIC_QUIC_SIMPLE_CLIENT_H_ | 295 #endif // NET_TOOLS_QUIC_QUIC_SIMPLE_CLIENT_H_ |
OLD | NEW |