Index: net/quic/quic_data_stream.h |
diff --git a/net/quic/quic_data_stream.h b/net/quic/quic_data_stream.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..76ebea8d4353e627f4ea5a53520c25e0a510aa52 |
--- /dev/null |
+++ b/net/quic/quic_data_stream.h |
@@ -0,0 +1,140 @@ |
+// 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. |
+// |
+// The base class for streams which deliver data to/from an application. |
+// In each direction, the data on such a stream first contains compressed |
+// headers then body data. |
+ |
+#ifndef NET_QUIC_QUIC_DATA_STREAM_H_ |
+#define NET_QUIC_QUIC_DATA_STREAM_H_ |
+ |
+#include <sys/types.h> |
+ |
+#include <list> |
+ |
+#include "base/strings/string_piece.h" |
+#include "net/base/iovec.h" |
+#include "net/base/net_export.h" |
+#include "net/quic/quic_ack_notifier.h" |
+#include "net/quic/quic_spdy_compressor.h" |
+#include "net/quic/quic_spdy_decompressor.h" |
+#include "net/quic/quic_stream_sequencer.h" |
+#include "net/quic/reliable_quic_stream.h" |
+ |
+namespace net { |
+ |
+namespace test { |
+class QuicDataStreamPeer; |
+class ReliableQuicStreamPeer; |
+} // namespace test |
+ |
+class IPEndPoint; |
+class QuicSession; |
+class SSLInfo; |
+ |
+// All this does right now is send data to subclasses via the sequencer. |
+class NET_EXPORT_PRIVATE QuicDataStream : public ReliableQuicStream, |
+ public QuicSpdyDecompressor::Visitor { |
+ public: |
+ // Visitor receives callbacks from the stream. |
+ class Visitor { |
+ public: |
+ Visitor() {} |
+ |
+ // Called when the stream is closed. |
+ virtual void OnClose(QuicDataStream* stream) = 0; |
+ |
+ protected: |
+ virtual ~Visitor() {} |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(Visitor); |
+ }; |
+ |
+ QuicDataStream(QuicStreamId id, QuicSession* session); |
+ |
+ virtual ~QuicDataStream(); |
+ |
+ // ReliableQuicStream implementation |
+ virtual void OnClose() OVERRIDE; |
+ // By default, this is the same as priority(), however it allows streams |
+ // to temporarily alter effective priority. For example if a SPDY stream has |
+ // compressed but not written headers it can write the headers with a higher |
+ // priority. |
+ virtual QuicPriority EffectivePriority() const OVERRIDE; |
+ virtual uint32 ProcessRawData(const char* data, uint32 data_len) OVERRIDE; |
+ |
+ // QuicSpdyDecompressor::Visitor implementation. |
+ virtual bool OnDecompressedData(base::StringPiece data) OVERRIDE; |
+ virtual void OnDecompressionError() OVERRIDE; |
+ |
+ virtual uint32 ProcessData(const char* data, uint32 data_len) = 0; |
+ |
+ // This block of functions wraps the sequencer's functions of the same |
+ // name. These methods return uncompressed data until that has |
+ // been fully processed. Then they simply delegate to the sequencer. |
+ virtual size_t Readv(const struct iovec* iov, size_t iov_len); |
+ virtual int GetReadableRegions(iovec* iov, size_t iov_len); |
+ // Returns true when all data has been read from the peer, including the fin. |
+ virtual bool IsDoneReading() const; |
+ virtual bool HasBytesToRead() const; |
+ |
+ // Called by the session when a decompression blocked stream |
+ // becomes unblocked. |
+ virtual void OnDecompressorAvailable(); |
+ |
+ void set_visitor(Visitor* visitor) { visitor_ = visitor; } |
+ |
+ bool headers_decompressed() const { return headers_decompressed_; } |
+ |
+ const IPEndPoint& GetPeerAddress(); |
+ |
+ QuicSpdyCompressor* compressor(); |
+ |
+ // Gets the SSL connection information. |
+ bool GetSSLInfo(SSLInfo* ssl_info); |
+ |
+ protected: |
+ // Sets priority_ to priority. This should only be called before bytes are |
+ // written to the server. |
+ void set_priority(QuicPriority priority); |
+ // This is protected because external classes should use EffectivePriority |
+ // instead. |
+ QuicPriority priority() const { return priority_; } |
+ |
+ private: |
+ friend class test::QuicDataStreamPeer; |
+ friend class test::ReliableQuicStreamPeer; |
+ friend class QuicStreamUtils; |
+ |
+ uint32 ProcessHeaderData(); |
+ |
+ uint32 StripPriorityAndHeaderId(const char* data, uint32 data_len); |
+ |
+ bool FinishedReadingHeaders(); |
+ |
+ Visitor* visitor_; |
+ // True if the headers have been completely decompresssed. |
+ bool headers_decompressed_; |
+ // The priority of the stream, once parsed. |
+ QuicPriority priority_; |
+ // ID of the header block sent by the peer, once parsed. |
+ QuicHeaderId headers_id_; |
+ // Buffer into which we write bytes from priority_ and headers_id_ |
+ // until each is fully parsed. |
+ string headers_id_and_priority_buffer_; |
+ // Contains a copy of the decompressed headers until they are consumed |
+ // via ProcessData or Readv. |
+ string decompressed_headers_; |
+ // True if an error was encountered during decompression. |
+ bool decompression_failed_; |
+ // True if the priority has been read, false otherwise. |
+ bool priority_parsed_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(QuicDataStream); |
+}; |
+ |
+} // namespace net |
+ |
+#endif // NET_QUIC_QUIC_DATA_STREAM_H_ |