OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef NET_SERVER_HTTP_SERVER_H_ | 5 #ifndef NET_SERVER_HTTP_SERVER_H_ |
6 #define NET_SERVER_HTTP_SERVER_H_ | 6 #define NET_SERVER_HTTP_SERVER_H_ |
7 | 7 |
8 #include <list> | |
9 #include <map> | 8 #include <map> |
10 | 9 |
11 #include "base/basictypes.h" | |
12 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
| 11 #include "base/memory/weak_ptr.h" |
13 #include "net/http/http_status_code.h" | 12 #include "net/http/http_status_code.h" |
14 #include "net/socket/stream_listen_socket.h" | 13 #include "net/server/http_connection.h" |
15 | 14 |
16 namespace net { | 15 namespace net { |
17 | 16 |
18 class HttpConnection; | |
19 class HttpServerRequestInfo; | 17 class HttpServerRequestInfo; |
20 class HttpServerResponseInfo; | 18 class HttpServerResponseInfo; |
21 class IPEndPoint; | 19 class IPEndPoint; |
| 20 class ServerSocket; |
| 21 class StreamSocket; |
22 class WebSocket; | 22 class WebSocket; |
23 | 23 |
24 class HttpServer : public StreamListenSocket::Delegate, | 24 class NET_EXPORT HttpServer { |
25 public base::RefCountedThreadSafe<HttpServer> { | |
26 public: | 25 public: |
27 class Delegate { | 26 class Delegate { |
28 public: | 27 public: |
29 virtual void OnHttpRequest(int connection_id, | 28 virtual void OnHttpRequest(int connection_id, |
30 const HttpServerRequestInfo& info) = 0; | 29 const HttpServerRequestInfo& info) = 0; |
31 | 30 |
32 virtual void OnWebSocketRequest(int connection_id, | 31 virtual void OnWebSocketRequest(int connection_id, |
33 const HttpServerRequestInfo& info) = 0; | 32 const HttpServerRequestInfo& info) = 0; |
34 | 33 |
35 virtual void OnWebSocketMessage(int connection_id, | 34 virtual void OnWebSocketMessage(int connection_id, |
36 const std::string& data) = 0; | 35 const std::string& data) = 0; |
37 | 36 |
38 virtual void OnClose(int connection_id) = 0; | 37 virtual void OnClose(int connection_id) = 0; |
39 | |
40 protected: | |
41 virtual ~Delegate() {} | |
42 }; | 38 }; |
43 | 39 |
44 HttpServer(const StreamListenSocketFactory& socket_factory, | 40 HttpServer(scoped_ptr<ServerSocket> server_socket, |
45 HttpServer::Delegate* delegate); | 41 HttpServer::Delegate* delegate); |
| 42 ~HttpServer(); |
46 | 43 |
47 void AcceptWebSocket(int connection_id, | 44 void AcceptWebSocket(int connection_id, |
48 const HttpServerRequestInfo& request); | 45 const HttpServerRequestInfo& request); |
49 void SendOverWebSocket(int connection_id, const std::string& data); | 46 void SendOverWebSocket(int connection_id, const std::string& data); |
50 // Sends the provided data directly to the given connection. No validation is | 47 // Sends the provided data directly to the given connection. No validation is |
51 // performed that data constitutes a valid HTTP response. A valid HTTP | 48 // performed that data constitutes a valid HTTP response. A valid HTTP |
52 // response may be split across multiple calls to SendRaw. | 49 // response may be split across multiple calls to SendRaw. |
53 void SendRaw(int connection_id, const std::string& data); | 50 void SendRaw(int connection_id, const std::string& data); |
54 void SendResponse(int connection_id, const HttpServerResponseInfo& response); | 51 void SendResponse(int connection_id, const HttpServerResponseInfo& response); |
55 void Send(int connection_id, | 52 void Send(int connection_id, |
56 HttpStatusCode status_code, | 53 HttpStatusCode status_code, |
57 const std::string& data, | 54 const std::string& data, |
58 const std::string& mime_type); | 55 const std::string& mime_type); |
59 void Send200(int connection_id, | 56 void Send200(int connection_id, |
60 const std::string& data, | 57 const std::string& data, |
61 const std::string& mime_type); | 58 const std::string& mime_type); |
62 void Send404(int connection_id); | 59 void Send404(int connection_id); |
63 void Send500(int connection_id, const std::string& message); | 60 void Send500(int connection_id, const std::string& message); |
64 | 61 |
65 void Close(int connection_id); | 62 void Close(int connection_id); |
66 | 63 |
| 64 void SetReceiveBufferSize(int connection_id, int32 size); |
| 65 void SetSendBufferSize(int connection_id, int32 size); |
| 66 |
67 // Copies the local address to |address|. Returns a network error code. | 67 // Copies the local address to |address|. Returns a network error code. |
68 int GetLocalAddress(IPEndPoint* address); | 68 int GetLocalAddress(IPEndPoint* address); |
69 | 69 |
70 // ListenSocketDelegate | 70 base::WeakPtr<HttpServer> GetWeakPtr(); |
71 virtual void DidAccept(StreamListenSocket* server, | |
72 scoped_ptr<StreamListenSocket> socket) OVERRIDE; | |
73 virtual void DidRead(StreamListenSocket* socket, | |
74 const char* data, | |
75 int len) OVERRIDE; | |
76 virtual void DidClose(StreamListenSocket* socket) OVERRIDE; | |
77 | |
78 protected: | |
79 virtual ~HttpServer(); | |
80 | 71 |
81 private: | 72 private: |
82 friend class base::RefCountedThreadSafe<HttpServer>; | 73 friend class HttpServerTest; |
83 friend class HttpConnection; | 74 |
| 75 typedef std::map<int, HttpConnection*> IdToConnectionMap; |
| 76 |
| 77 void DoAcceptLoop(); |
| 78 void OnAcceptCompleted(int rv); |
| 79 int DidAccept(int rv); |
| 80 |
| 81 void DoReadLoop(HttpConnection* connection); |
| 82 void OnReadCompleted(int connection_id, int rv); |
| 83 int DidRead(HttpConnection* conn, int rv); |
| 84 |
| 85 void DoWriteLoop(HttpConnection* connection); |
| 86 void OnWriteCompleted(int connection_id, int rv); |
| 87 int DidWrite(HttpConnection* conn, int rv); |
84 | 88 |
85 // Expects the raw data to be stored in recv_data_. If parsing is successful, | 89 // Expects the raw data to be stored in recv_data_. If parsing is successful, |
86 // will remove the data parsed from recv_data_, leaving only the unused | 90 // will remove the data parsed from recv_data_, leaving only the unused |
87 // recv data. | 91 // recv data. |
88 bool ParseHeaders(HttpConnection* connection, | 92 bool ParseHeaders(const char* data, |
| 93 size_t data_len, |
89 HttpServerRequestInfo* info, | 94 HttpServerRequestInfo* info, |
90 size_t* pos); | 95 size_t* pos); |
91 | 96 |
92 HttpConnection* FindConnection(int connection_id); | 97 HttpConnection* FindConnection(int connection_id); |
93 HttpConnection* FindConnection(StreamListenSocket* socket); | |
94 | 98 |
95 HttpServer::Delegate* delegate_; | 99 const scoped_ptr<ServerSocket> server_socket_; |
96 scoped_ptr<StreamListenSocket> server_; | 100 scoped_ptr<StreamSocket> accepted_socket_; |
97 typedef std::map<int, HttpConnection*> IdToConnectionMap; | 101 HttpServer::Delegate* const delegate_; |
| 102 |
| 103 int last_id_; |
98 IdToConnectionMap id_to_connection_; | 104 IdToConnectionMap id_to_connection_; |
99 typedef std::map<StreamListenSocket*, HttpConnection*> SocketToConnectionMap; | 105 |
100 SocketToConnectionMap socket_to_connection_; | 106 base::WeakPtrFactory<HttpServer> weak_ptr_factory_; |
101 | 107 |
102 DISALLOW_COPY_AND_ASSIGN(HttpServer); | 108 DISALLOW_COPY_AND_ASSIGN(HttpServer); |
103 }; | 109 }; |
104 | 110 |
105 } // namespace net | 111 } // namespace net |
106 | 112 |
107 #endif // NET_SERVER_HTTP_SERVER_H_ | 113 #endif // NET_SERVER_HTTP_SERVER_H_ |
OLD | NEW |