Cleanup of QUIC stream classes.

net/tools/quic/quic_spdy_server_stream.h

 // 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_SERVER_STREAM_H_
 #define NET_TOOLS_QUIC_QUIC_SPDY_SERVER_STREAM_H_
 
 #include <string>
 
 #include "net/base/io_buffer.h"
-#include "net/tools/quic/quic_reliable_server_stream.h"
+#include "net/quic/quic_protocol.h"
+#include "net/quic/reliable_quic_stream.h"
+#include "net/tools/balsa/balsa_headers.h"
 
 namespace net {
 
 class QuicSession;
 
 namespace tools {
 
-// All this does right now is aggregate data, and on fin, send a cached
+namespace test {
+class QuicSpdyServerStreamPeer;
+}  // namespace test
+
+// All this does right now is aggregate data, and on fin, send an HTTP
 // response.
-class QuicSpdyServerStream : public QuicReliableServerStream {
+class QuicSpdyServerStream : public ReliableQuicStream {
  public:
   QuicSpdyServerStream(QuicStreamId id, QuicSession* session);
   virtual ~QuicSpdyServerStream();
 
   // ReliableQuicStream implementation called by the session when there's
   // data for us.
   virtual uint32 ProcessData(const char* data, uint32 data_len) OVERRIDE;
 
-  virtual void SendHeaders(const BalsaHeaders& response_headers) OVERRIDE;
+  virtual void SendHeaders(const BalsaHeaders& response_headers);
 
   int ParseRequestHeaders();
 
+  // Sends a basic 200 response using SendHeaders for the headers and WriteData
+  // for the body.
+  void SendResponse();
+  // Sends a basic 500 response using SendHeaders for the headers and WriteData
+  // for the body
+  void SendErrorResponse();
+  // Make sure that as soon as we start writing data, we stop reading.
+  virtual QuicConsumedData WriteData(base::StringPiece data, bool fin) OVERRIDE;
+
+  // Returns whatever headers have been received for this stream.
+  const BalsaHeaders& headers() { return headers_; }
+
+  const string& body() { return body_; }
+
  protected:
   virtual void OnFinRead() OVERRIDE;
 
+ private:
+  friend class test::QuicSpdyServerStreamPeer;
+
+  BalsaHeaders headers_;
+  string body_;
+
   // Buffer into which response header data is read.
   scoped_refptr<GrowableIOBuffer> read_buf_;
   bool request_headers_received_;
 };
 
 }  // namespace tools
 }  // namespace net
 
 #endif  // NET_TOOLS_QUIC_QUIC_SPDY_SERVER_STREAM_H_