OLD | NEW |
| (Empty) |
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_SERVER_HTTP_SERVER_H_ | |
6 #define NET_SERVER_HTTP_SERVER_H_ | |
7 | |
8 #include <map> | |
9 #include <string> | |
10 | |
11 #include "base/basictypes.h" | |
12 #include "base/macros.h" | |
13 #include "base/memory/scoped_ptr.h" | |
14 #include "base/memory/weak_ptr.h" | |
15 #include "net/http/http_status_code.h" | |
16 | |
17 namespace net { | |
18 | |
19 class HttpConnection; | |
20 class HttpServerRequestInfo; | |
21 class HttpServerResponseInfo; | |
22 class IPEndPoint; | |
23 class ServerSocket; | |
24 class StreamSocket; | |
25 class WebSocket; | |
26 | |
27 class HttpServer { | |
28 public: | |
29 // Delegate to handle http/websocket events. Beware that it is not safe to | |
30 // destroy the HttpServer in any of these callbacks. | |
31 class Delegate { | |
32 public: | |
33 virtual void OnConnect(int connection_id) = 0; | |
34 virtual void OnHttpRequest(int connection_id, | |
35 const HttpServerRequestInfo& info) = 0; | |
36 virtual void OnWebSocketRequest(int connection_id, | |
37 const HttpServerRequestInfo& info) = 0; | |
38 virtual void OnWebSocketMessage(int connection_id, | |
39 const std::string& data) = 0; | |
40 virtual void OnClose(int connection_id) = 0; | |
41 }; | |
42 | |
43 // Instantiates a http server with |server_socket| which already started | |
44 // listening, but not accepting. This constructor schedules accepting | |
45 // connections asynchronously in case when |delegate| is not ready to get | |
46 // callbacks yet. | |
47 HttpServer(scoped_ptr<ServerSocket> server_socket, | |
48 HttpServer::Delegate* delegate); | |
49 ~HttpServer(); | |
50 | |
51 void AcceptWebSocket(int connection_id, | |
52 const HttpServerRequestInfo& request); | |
53 void SendOverWebSocket(int connection_id, const std::string& data); | |
54 // Sends the provided data directly to the given connection. No validation is | |
55 // performed that data constitutes a valid HTTP response. A valid HTTP | |
56 // response may be split across multiple calls to SendRaw. | |
57 void SendRaw(int connection_id, const std::string& data); | |
58 // TODO(byungchul): Consider replacing function name with SendResponseInfo | |
59 void SendResponse(int connection_id, const HttpServerResponseInfo& response); | |
60 void Send(int connection_id, | |
61 HttpStatusCode status_code, | |
62 const std::string& data, | |
63 const std::string& mime_type); | |
64 void Send200(int connection_id, | |
65 const std::string& data, | |
66 const std::string& mime_type); | |
67 void Send404(int connection_id); | |
68 void Send500(int connection_id, const std::string& message); | |
69 | |
70 void Close(int connection_id); | |
71 | |
72 void SetReceiveBufferSize(int connection_id, int32 size); | |
73 void SetSendBufferSize(int connection_id, int32 size); | |
74 | |
75 // Copies the local address to |address|. Returns a network error code. | |
76 int GetLocalAddress(IPEndPoint* address); | |
77 | |
78 private: | |
79 friend class HttpServerTest; | |
80 | |
81 typedef std::map<int, HttpConnection*> IdToConnectionMap; | |
82 | |
83 void DoAcceptLoop(); | |
84 void OnAcceptCompleted(int rv); | |
85 int HandleAcceptResult(int rv); | |
86 | |
87 void DoReadLoop(HttpConnection* connection); | |
88 void OnReadCompleted(int connection_id, int rv); | |
89 int HandleReadResult(HttpConnection* connection, int rv); | |
90 | |
91 void DoWriteLoop(HttpConnection* connection); | |
92 void OnWriteCompleted(int connection_id, int rv); | |
93 int HandleWriteResult(HttpConnection* connection, int rv); | |
94 | |
95 // Expects the raw data to be stored in recv_data_. If parsing is successful, | |
96 // will remove the data parsed from recv_data_, leaving only the unused | |
97 // recv data. | |
98 bool ParseHeaders(const char* data, | |
99 size_t data_len, | |
100 HttpServerRequestInfo* info, | |
101 size_t* pos); | |
102 | |
103 HttpConnection* FindConnection(int connection_id); | |
104 | |
105 // Whether or not Close() has been called during delegate callback processing. | |
106 bool HasClosedConnection(HttpConnection* connection); | |
107 | |
108 const scoped_ptr<ServerSocket> server_socket_; | |
109 scoped_ptr<StreamSocket> accepted_socket_; | |
110 HttpServer::Delegate* const delegate_; | |
111 | |
112 int last_id_; | |
113 IdToConnectionMap id_to_connection_; | |
114 | |
115 base::WeakPtrFactory<HttpServer> weak_ptr_factory_; | |
116 | |
117 DISALLOW_COPY_AND_ASSIGN(HttpServer); | |
118 }; | |
119 | |
120 } // namespace net | |
121 | |
122 #endif // NET_SERVER_HTTP_SERVER_H_ | |
OLD | NEW |