| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef NET_TOOLS_QUIC_QUIC_SPDY_CLIENT_STREAM_H_ | |
| 6 #define NET_TOOLS_QUIC_QUIC_SPDY_CLIENT_STREAM_H_ | |
| 7 | |
| 8 #include <sys/types.h> | |
| 9 #include <string> | |
| 10 | |
| 11 #include "base/basictypes.h" | |
| 12 #include "base/strings/string_piece.h" | |
| 13 #include "net/base/io_buffer.h" | |
| 14 #include "net/quic/quic_data_stream.h" | |
| 15 #include "net/quic/quic_protocol.h" | |
| 16 #include "net/tools/balsa/balsa_frame.h" | |
| 17 #include "net/tools/balsa/balsa_headers.h" | |
| 18 | |
| 19 namespace net { | |
| 20 | |
| 21 namespace tools { | |
| 22 | |
| 23 class QuicClientSession; | |
| 24 | |
| 25 // All this does right now is send an SPDY request, and aggregate the | |
| 26 // SPDY response. | |
| 27 class QuicSpdyClientStream : public QuicDataStream { | |
| 28 public: | |
| 29 QuicSpdyClientStream(QuicStreamId id, QuicClientSession* session); | |
| 30 ~QuicSpdyClientStream() override; | |
| 31 | |
| 32 // Override the base class to close the write side as soon as we get a | |
| 33 // response. | |
| 34 // SPDY/HTTP does not support bidirectional streaming. | |
| 35 void OnStreamFrame(const QuicStreamFrame& frame) override; | |
| 36 | |
| 37 // Override the base class to store the size of the headers. | |
| 38 void OnStreamHeadersComplete(bool fin, size_t frame_len) override; | |
| 39 | |
| 40 // ReliableQuicStream implementation called by the session when there's | |
| 41 // data for us. | |
| 42 uint32 ProcessData(const char* data, uint32 data_len) override; | |
| 43 | |
| 44 void OnFinRead() override; | |
| 45 | |
| 46 // Serializes the headers and body, sends it to the server, and | |
| 47 // returns the number of bytes sent. | |
| 48 ssize_t SendRequest(const BalsaHeaders& headers, | |
| 49 base::StringPiece body, | |
| 50 bool fin); | |
| 51 | |
| 52 // Sends body data to the server, or buffers if it can't be sent immediately. | |
| 53 void SendBody(const std::string& data, bool fin); | |
| 54 // As above, but |delegate| will be notified once |data| is ACKed. | |
| 55 void SendBody(const std::string& data, | |
| 56 bool fin, | |
| 57 QuicAckNotifier::DelegateInterface* delegate); | |
| 58 | |
| 59 // Returns the response data. | |
| 60 const std::string& data() { return data_; } | |
| 61 | |
| 62 // Returns whatever headers have been received for this stream. | |
| 63 const BalsaHeaders& headers() { return headers_; } | |
| 64 | |
| 65 size_t header_bytes_read() const { return header_bytes_read_; } | |
| 66 | |
| 67 size_t header_bytes_written() const { return header_bytes_written_; } | |
| 68 | |
| 69 // While the server's set_priority shouldn't be called externally, the creator | |
| 70 // of client-side streams should be able to set the priority. | |
| 71 using QuicDataStream::set_priority; | |
| 72 | |
| 73 private: | |
| 74 int ParseResponseHeaders(); | |
| 75 | |
| 76 BalsaHeaders headers_; | |
| 77 std::string data_; | |
| 78 | |
| 79 scoped_refptr<GrowableIOBuffer> read_buf_; | |
| 80 bool response_headers_received_; | |
| 81 size_t header_bytes_read_; | |
| 82 size_t header_bytes_written_; | |
| 83 | |
| 84 DISALLOW_COPY_AND_ASSIGN(QuicSpdyClientStream); | |
| 85 }; | |
| 86 | |
| 87 } // namespace tools | |
| 88 } // namespace net | |
| 89 | |
| 90 #endif // NET_TOOLS_QUIC_QUIC_SPDY_CLIENT_STREAM_H_ | |
| OLD | NEW |