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 session when there's
  36   // data for us.
  37   uint32 ProcessData(const char* data, uint32 data_len) override;
  38   void OnFinRead() override;
  39
  40  protected:
  41   // Sends a basic 200 response using SendHeaders for the headers and WriteData
  42   // for the body.
  43   virtual void SendResponse();
  44
  45   void SendHeadersAndBody(const SpdyHeaderBlock& response_headers,
  46                           base::StringPiece body);
  47
  48   SpdyHeaderBlock* request_headers() { return &request_headers_; }
  49
  50   const std::string& body() { return body_; }
  51
  52  private:
  53   friend class test::QuicSpdyServerStreamPeer;
  54
  55   // Parses the request headers from |data| to |request_headers_|.
  56   // Returns false if there was an error parsing the headers.
  57   bool ParseRequestHeaders(const char* data, uint32 data_len);
  58
  59   // Sends a basic 500 response using SendHeaders for the headers and WriteData
  60   // for the body
  61   void SendErrorResponse();
  62
  63   // Returns the key for |request_headers_| which identifies the host.
  64   const std::string GetHostKey();
  65
  66   // The parsed headers received from the client.
  67   SpdyHeaderBlock request_headers_;
  68   int content_length_;
  69   std::string body_;
  70
  71   DISALLOW_COPY_AND_ASSIGN(QuicSpdyServerStream);
  72 };
  73
  74 }  // namespace tools
  75 }  // namespace net
  76
  77 #endif  // NET_TOOLS_QUIC_QUIC_SPDY_SERVER_STREAM_H_