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