Index: net/tools/quic/quic_client.h |
diff --git a/net/tools/quic/quic_client.h b/net/tools/quic/quic_client.h |
index 8146764bab93f60624a2b9518487ef7336016901..88268159e4ef713b26538903347980be27b7c6aa 100644 |
--- a/net/tools/quic/quic_client.h |
+++ b/net/tools/quic/quic_client.h |
@@ -34,10 +34,13 @@ namespace test { |
class QuicClientPeer; |
} // namespace test |
-class QuicClient : public EpollCallbackInterface { |
+class QuicClient : public EpollCallbackInterface, |
+ public ReliableQuicStream::Visitor { |
public: |
- QuicClient(IPEndPoint server_address, const std::string& server_hostname, |
- const QuicVersion version); |
+ QuicClient(IPEndPoint server_address, |
+ const string& server_hostname, |
+ const QuicVersion version, |
+ bool print_response); |
QuicClient(IPEndPoint server_address, |
const std::string& server_hostname, |
const QuicConfig& config, |
@@ -96,6 +99,9 @@ class QuicClient : public EpollCallbackInterface { |
virtual void OnUnregistration(int fd, bool replaced) OVERRIDE {} |
virtual void OnShutdown(EpollServer* eps, int fd) OVERRIDE {} |
+ // ReliableQuicStream::Visitor |
+ virtual void OnClose(ReliableQuicStream* stream) OVERRIDE; |
+ |
QuicPacketCreator::Options* options(); |
QuicClientSession* session() { return session_.get(); } |
@@ -190,6 +196,10 @@ class QuicClient : public EpollCallbackInterface { |
// Which QUIC version does this client talk? |
QuicVersion version_; |
+ // If true, then the contents of each response will be printed to stdout |
+ // when the stream is closed (in OnClose). |
+ bool print_response_; |
+ |
DISALLOW_COPY_AND_ASSIGN(QuicClient); |
}; |