net/tools/quic/quic_spdy_server_stream.h

   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_SERVER_STREAM_H_
   6 #define NET_TOOLS_QUIC_QUIC_SPDY_SERVER_STREAM_H_
   7
   8 #include <string>
   9
  10 #include "base/basictypes.h"
  11 #include "net/quic/quic_data_stream.h"
  12 #include "net/quic/quic_protocol.h"
  13 #include "net/spdy/spdy_framer.h"
  14
  15 namespace net {
  16
  17 class QuicSpdySession;
  18
  19 namespace tools {
  20
  21 namespace test {
  22 class QuicSpdyServerStreamPeer;
  23 }  // namespace test
  24
  25 // All this does right now is aggregate data, and on fin, send an HTTP
  26 // response.
  27 class QuicSpdyServerStream : public QuicDataStream {
  28  public:
  29   QuicSpdyServerStream(QuicStreamId id, QuicSpdySession* session);
  30   ~QuicSpdyServerStream() override;
  31
  32   // QuicDataStream
  33   void OnStreamHeadersComplete(bool fin, size_t frame_len) override;
  34
  35   // ReliableQuicStream implementation called by the sequencer when there is
  36   // data (or a FIN) to be read.
  37   void OnDataAvailable() override;
  38
  39  protected:
  40   // Sends a basic 200 response using SendHeaders for the headers and WriteData
  41   // for the body.
  42   virtual void SendResponse();
  43
  44   void SendHeadersAndBody(const SpdyHeaderBlock& response_headers,
  45                           base::StringPiece body);
  46
  47   SpdyHeaderBlock* request_headers() { return &request_headers_; }
  48
  49   const std::string& body() { return body_; }
  50
  51  private:
  52   friend class test::QuicSpdyServerStreamPeer;
  53
  54   // Parses the request headers from |data| to |request_headers_|.
  55   // Returns false if there was an error parsing the headers.
  56   bool ParseRequestHeaders(const char* data, uint32 data_len);
  57
  58   // Sends a basic 500 response using SendHeaders for the headers and WriteData
  59   // for the body
  60   void SendErrorResponse();
  61
  62   // Returns the key for |request_headers_| which identifies the host.
  63   const std::string GetHostKey();
  64
  65   // The parsed headers received from the client.
  66   SpdyHeaderBlock request_headers_;
  67   int content_length_;
  68   std::string body_;
  69
  70   DISALLOW_COPY_AND_ASSIGN(QuicSpdyServerStream);
  71 };
  72
  73 }  // namespace tools
  74 }  // namespace net
  75
  76 #endif  // NET_TOOLS_QUIC_QUIC_SPDY_SERVER_STREAM_H_