src/net/tools/flip_server/http_interface.h

   1 // Copyright (c) 2011 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_FLIP_SERVER_HTTP_INTERFACE_H_
   6 #define NET_TOOLS_FLIP_SERVER_HTTP_INTERFACE_H_
   7
   8 #include <string>
   9
  10 #include "base/compiler_specific.h"
  11 #include "net/tools/balsa/balsa_headers.h"
  12 #include "net/tools/balsa/balsa_visitor_interface.h"
  13 #include "net/tools/flip_server/output_ordering.h"
  14 #include "net/tools/flip_server/sm_connection.h"
  15 #include "net/tools/flip_server/sm_interface.h"
  16
  17 namespace net {
  18
  19 class BalsaFrame;
  20 class DataFrame;
  21 class EpollServer;
  22 class FlipAcceptor;
  23 class MemoryCache;
  24
  25 class HttpSM : public BalsaVisitorInterface,
  26                public SMInterface {
  27  public:
  28   HttpSM(SMConnection* connection,
  29          SMInterface* sm_spdy_interface,
  30          MemoryCache* memory_cache,
  31          FlipAcceptor* acceptor);
  32   virtual ~HttpSM();
  33
  34  private:
  35   // BalsaVisitorInterface:
  36   virtual void ProcessBodyInput(const char *input, size_t size) OVERRIDE {}
  37   virtual void ProcessBodyData(const char *input, size_t size) OVERRIDE;
  38   virtual void ProcessHeaderInput(const char *input, size_t size) OVERRIDE {}
  39   virtual void ProcessTrailerInput(const char *input, size_t size) OVERRIDE {}
  40   virtual void ProcessHeaders(const BalsaHeaders& headers) OVERRIDE;
  41   virtual void ProcessRequestFirstLine(const char* line_input,
  42                                        size_t line_length,
  43                                        const char* method_input,
  44                                        size_t method_length,
  45                                        const char* request_uri_input,
  46                                        size_t request_uri_length,
  47                                        const char* version_input,
  48                                        size_t version_length) OVERRIDE {}
  49   virtual void ProcessResponseFirstLine(const char *line_input,
  50                                         size_t line_length,
  51                                         const char *version_input,
  52                                         size_t version_length,
  53                                         const char *status_input,
  54                                         size_t status_length,
  55                                         const char *reason_input,
  56                                         size_t reason_length) OVERRIDE {}
  57   virtual void ProcessChunkLength(size_t chunk_length) OVERRIDE {}
  58   virtual void ProcessChunkExtensions(const char *input,
  59                                       size_t size) OVERRIDE {}
  60   virtual void HeaderDone() OVERRIDE {}
  61   virtual void MessageDone() OVERRIDE;
  62   virtual void HandleHeaderError(BalsaFrame* framer) OVERRIDE;
  63   virtual void HandleHeaderWarning(BalsaFrame* framer) OVERRIDE {}
  64   virtual void HandleChunkingError(BalsaFrame* framer) OVERRIDE;
  65   virtual void HandleBodyError(BalsaFrame* framer) OVERRIDE;
  66
  67   void HandleError();
  68
  69  public:
  70   void AddToOutputOrder(const MemCacheIter& mci);
  71   BalsaFrame* spdy_framer() { return http_framer_; }
  72   virtual void set_is_request() OVERRIDE {}
  73   const OutputOrdering& output_ordering() const { return output_ordering_; }
  74
  75   // SMInterface:
  76   virtual void InitSMInterface(SMInterface* sm_spdy_interface,
  77                                int32 server_idx) OVERRIDE;
  78   virtual void InitSMConnection(SMConnectionPoolInterface* connection_pool,
  79                                 SMInterface* sm_interface,
  80                                 EpollServer* epoll_server,
  81                                 int fd,
  82                                 std::string server_ip,
  83                                 std::string server_port,
  84                                 std::string remote_ip,
  85                                 bool use_ssl) OVERRIDE;
  86   virtual size_t ProcessReadInput(const char* data, size_t len) OVERRIDE;
  87   virtual size_t ProcessWriteInput(const char* data, size_t len) OVERRIDE;
  88   virtual bool MessageFullyRead() const OVERRIDE;
  89   virtual void SetStreamID(uint32 stream_id) OVERRIDE;
  90   virtual bool Error() const OVERRIDE;
  91   virtual const char* ErrorAsString() const OVERRIDE;
  92   virtual void Reset() OVERRIDE;
  93   virtual void ResetForNewInterface(int32 server_idx) OVERRIDE {}
  94   virtual void ResetForNewConnection() OVERRIDE;
  95   virtual void Cleanup() OVERRIDE;
  96   virtual int PostAcceptHook() OVERRIDE;
  97
  98   virtual void NewStream(uint32 stream_id, uint32 priority,
  99                          const std::string& filename) OVERRIDE;
 100   virtual void SendEOF(uint32 stream_id) OVERRIDE;
 101   virtual void SendErrorNotFound(uint32 stream_id) OVERRIDE;
 102   virtual size_t SendSynStream(uint32 stream_id,
 103                                const BalsaHeaders& headers) OVERRIDE;
 104   virtual size_t SendSynReply(uint32 stream_id,
 105                               const BalsaHeaders& headers) OVERRIDE;
 106   virtual void SendDataFrame(uint32 stream_id, const char* data, int64 len,
 107                              uint32 flags, bool compress) OVERRIDE;
 108
 109  private:
 110   void SendEOFImpl(uint32 stream_id);
 111   void SendErrorNotFoundImpl(uint32 stream_id);
 112   void SendOKResponseImpl(uint32 stream_id, const std::string& output);
 113   size_t SendSynReplyImpl(uint32 stream_id, const BalsaHeaders& headers);
 114   size_t SendSynStreamImpl(uint32 stream_id, const BalsaHeaders& headers);
 115   void SendDataFrameImpl(uint32 stream_id, const char* data, int64 len,
 116                          uint32 flags, bool compress);
 117   void EnqueueDataFrame(DataFrame* df);
 118   virtual void GetOutput() OVERRIDE;
 119
 120  private:
 121   BalsaFrame* http_framer_;
 122   BalsaHeaders headers_;
 123   uint32 stream_id_;
 124   int32 server_idx_;
 125
 126   SMConnection* connection_;
 127   SMInterface* sm_spdy_interface_;
 128   OutputList* output_list_;
 129   OutputOrdering output_ordering_;
 130   MemoryCache* memory_cache_;
 131   FlipAcceptor* acceptor_;
 132 };
 133
 134 }  // namespace net
 135
 136 #endif  // NET_TOOLS_FLIP_SERVER_HTTP_INTERFACE_H_