| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef NET_TOOLS_QUIC_QUIC_SPDY_CLIENT_STREAM_H_ |
| #define NET_TOOLS_QUIC_QUIC_SPDY_CLIENT_STREAM_H_ |
| |
| #include <sys/types.h> |
| #include <string> |
| |
| #include "base/basictypes.h" |
| #include "base/strings/string_piece.h" |
| #include "net/quic/quic_protocol.h" |
| #include "net/quic/quic_spdy_stream.h" |
| #include "net/spdy/spdy_framer.h" |
| |
| namespace net { |
| namespace tools { |
| |
| class QuicClientSession; |
| |
| // All this does right now is send an SPDY request, and aggregate the |
| // SPDY response. |
| class QuicSpdyClientStream : public QuicSpdyStream { |
| public: |
| QuicSpdyClientStream(QuicStreamId id, QuicClientSession* session); |
| ~QuicSpdyClientStream() override; |
| |
| // Override the base class to close the write side as soon as we get a |
| // response. |
| // SPDY/HTTP does not support bidirectional streaming. |
| void OnStreamFrame(const QuicStreamFrame& frame) override; |
| |
| // Override the base class to store the size of the headers. |
| void OnStreamHeadersComplete(bool fin, size_t frame_len) override; |
| |
| // ReliableQuicStream implementation called by the session when there's |
| // data for us. |
| void OnDataAvailable() override; |
| |
| // Serializes the headers and body, sends it to the server, and |
| // returns the number of bytes sent. |
| size_t SendRequest(const SpdyHeaderBlock& headers, |
| base::StringPiece body, |
| bool fin); |
| |
| // Sends body data to the server, or buffers if it can't be sent immediately. |
| void SendBody(const std::string& data, bool fin); |
| // As above, but |delegate| will be notified once |data| is ACKed. |
| void SendBody(const std::string& data, |
| bool fin, |
| QuicAckListenerInterface* listener); |
| |
| // Returns the response data. |
| const std::string& data() { return data_; } |
| |
| // Returns whatever headers have been received for this stream. |
| const SpdyHeaderBlock& headers() { return response_headers_; } |
| |
| size_t header_bytes_read() const { return header_bytes_read_; } |
| |
| size_t header_bytes_written() const { return header_bytes_written_; } |
| |
| int response_code() const { return response_code_; } |
| |
| // While the server's set_priority shouldn't be called externally, the creator |
| // of client-side streams should be able to set the priority. |
| using QuicSpdyStream::set_priority; |
| |
| void set_allow_bidirectional_data(bool value) { |
| allow_bidirectional_data_ = value; |
| } |
| |
| bool allow_bidirectional_data() const { return allow_bidirectional_data_; } |
| |
| private: |
| bool ParseResponseHeaders(const char* data, uint32 data_len); |
| |
| // The parsed headers received from the server. |
| SpdyHeaderBlock response_headers_; |
| // The parsed content-length, or -1 if none is specified. |
| int content_length_; |
| int response_code_; |
| std::string data_; |
| size_t header_bytes_read_; |
| size_t header_bytes_written_; |
| // When true allows the sending of a request to continue while the response is |
| // arriving. |
| bool allow_bidirectional_data_; |
| |
| DISALLOW_COPY_AND_ASSIGN(QuicSpdyClientStream); |
| }; |
| |
| } // namespace tools |
| } // namespace net |
| |
| #endif // NET_TOOLS_QUIC_QUIC_SPDY_CLIENT_STREAM_H_ |