Line data Source code
1 : /*
2 : * http_server - HTTP server
3 : * Copyright (c) 2009, Jouni Malinen <j@w1.fi>
4 : *
5 : * This software may be distributed under the terms of the BSD license.
6 : * See README for more details.
7 : */
8 :
9 : #include "includes.h"
10 : #include <fcntl.h>
11 :
12 : #include "common.h"
13 : #include "eloop.h"
14 : #include "httpread.h"
15 : #include "http_server.h"
16 :
17 : #define HTTP_SERVER_TIMEOUT 30
18 : #define HTTP_SERVER_MAX_REQ_LEN 8000
19 : #define HTTP_SERVER_MAX_CONNECTIONS 10
20 :
21 : struct http_request {
22 : struct http_request *next;
23 : struct http_server *srv;
24 : int fd;
25 : struct sockaddr_in cli;
26 : struct httpread *hread;
27 : };
28 :
29 : struct http_server {
30 : void (*cb)(void *ctx, struct http_request *req);
31 : void *cb_ctx;
32 :
33 : int fd;
34 : int port;
35 :
36 : struct http_request *requests;
37 : unsigned int request_count;
38 : };
39 :
40 :
41 179 : static void http_request_cb(struct httpread *handle, void *cookie,
42 : enum httpread_event en)
43 : {
44 179 : struct http_request *req = cookie;
45 179 : struct http_server *srv = req->srv;
46 :
47 179 : if (en == HTTPREAD_EVENT_FILE_READY) {
48 178 : wpa_printf(MSG_DEBUG, "HTTP: Request from %s:%d received",
49 : inet_ntoa(req->cli.sin_addr),
50 178 : ntohs(req->cli.sin_port));
51 178 : srv->cb(srv->cb_ctx, req);
52 357 : return;
53 : }
54 1 : wpa_printf(MSG_DEBUG, "HTTP: Request from %s:%d could not be received "
55 : "completely", inet_ntoa(req->cli.sin_addr),
56 1 : ntohs(req->cli.sin_port));
57 1 : http_request_deinit(req);
58 : }
59 :
60 :
61 179 : static struct http_request * http_request_init(struct http_server *srv, int fd,
62 : struct sockaddr_in *cli)
63 : {
64 : struct http_request *req;
65 :
66 179 : if (srv->request_count >= HTTP_SERVER_MAX_CONNECTIONS) {
67 0 : wpa_printf(MSG_DEBUG, "HTTP: Too many concurrent requests");
68 0 : return NULL;
69 : }
70 :
71 179 : req = os_zalloc(sizeof(*req));
72 179 : if (req == NULL)
73 0 : return NULL;
74 :
75 179 : req->srv = srv;
76 179 : req->fd = fd;
77 179 : req->cli = *cli;
78 :
79 179 : req->hread = httpread_create(req->fd, http_request_cb, req,
80 : HTTP_SERVER_MAX_REQ_LEN,
81 : HTTP_SERVER_TIMEOUT);
82 179 : if (req->hread == NULL) {
83 0 : http_request_deinit(req);
84 0 : return NULL;
85 : }
86 :
87 179 : return req;
88 : }
89 :
90 :
91 179 : void http_request_deinit(struct http_request *req)
92 : {
93 : struct http_request *r, *p;
94 : struct http_server *srv;
95 :
96 179 : if (req == NULL)
97 179 : return;
98 :
99 179 : srv = req->srv;
100 179 : p = NULL;
101 179 : r = srv->requests;
102 361 : while (r) {
103 182 : if (r == req) {
104 179 : if (p)
105 3 : p->next = r->next;
106 : else
107 176 : srv->requests = r->next;
108 179 : srv->request_count--;
109 179 : break;
110 : }
111 3 : p = r;
112 3 : r = r->next;
113 : }
114 :
115 179 : httpread_destroy(req->hread);
116 179 : close(req->fd);
117 179 : os_free(req);
118 : }
119 :
120 :
121 24 : static void http_request_free_all(struct http_request *req)
122 : {
123 : struct http_request *prev;
124 48 : while (req) {
125 0 : prev = req;
126 0 : req = req->next;
127 0 : http_request_deinit(prev);
128 : }
129 24 : }
130 :
131 :
132 178 : void http_request_send(struct http_request *req, struct wpabuf *resp)
133 : {
134 : int res;
135 :
136 178 : wpa_printf(MSG_DEBUG, "HTTP: Send %lu byte response to %s:%d",
137 : (unsigned long) wpabuf_len(resp),
138 : inet_ntoa(req->cli.sin_addr),
139 178 : ntohs(req->cli.sin_port));
140 :
141 178 : res = send(req->fd, wpabuf_head(resp), wpabuf_len(resp), 0);
142 178 : if (res < 0) {
143 0 : wpa_printf(MSG_DEBUG, "HTTP: Send failed: %s",
144 0 : strerror(errno));
145 178 : } else if ((size_t) res < wpabuf_len(resp)) {
146 0 : wpa_printf(MSG_DEBUG, "HTTP: Sent only %d of %lu bytes",
147 : res, (unsigned long) wpabuf_len(resp));
148 : /* TODO: add eloop handler for sending rest of the data */
149 : }
150 :
151 178 : wpabuf_free(resp);
152 178 : }
153 :
154 :
155 178 : void http_request_send_and_deinit(struct http_request *req,
156 : struct wpabuf *resp)
157 : {
158 178 : http_request_send(req, resp);
159 178 : http_request_deinit(req);
160 178 : }
161 :
162 :
163 178 : enum httpread_hdr_type http_request_get_type(struct http_request *req)
164 : {
165 178 : return httpread_hdr_type_get(req->hread);
166 : }
167 :
168 :
169 232 : char * http_request_get_uri(struct http_request *req)
170 : {
171 232 : return httpread_uri_get(req->hread);
172 : }
173 :
174 :
175 38 : char * http_request_get_hdr(struct http_request *req)
176 : {
177 38 : return httpread_hdr_get(req->hread);
178 : }
179 :
180 :
181 180 : char * http_request_get_data(struct http_request *req)
182 : {
183 180 : return httpread_data_get(req->hread);
184 : }
185 :
186 :
187 71 : char * http_request_get_hdr_line(struct http_request *req, const char *tag)
188 : {
189 71 : return httpread_hdr_line_get(req->hread, tag);
190 : }
191 :
192 :
193 178 : struct sockaddr_in * http_request_get_cli_addr(struct http_request *req)
194 : {
195 178 : return &req->cli;
196 : }
197 :
198 :
199 179 : static void http_server_cb(int sd, void *eloop_ctx, void *sock_ctx)
200 : {
201 : struct sockaddr_in addr;
202 179 : socklen_t addr_len = sizeof(addr);
203 179 : struct http_server *srv = eloop_ctx;
204 : int conn;
205 : struct http_request *req;
206 :
207 179 : conn = accept(srv->fd, (struct sockaddr *) &addr, &addr_len);
208 179 : if (conn < 0) {
209 0 : wpa_printf(MSG_DEBUG, "HTTP: Failed to accept new connection: "
210 0 : "%s", strerror(errno));
211 0 : return;
212 : }
213 179 : wpa_printf(MSG_DEBUG, "HTTP: Connection from %s:%d",
214 179 : inet_ntoa(addr.sin_addr), ntohs(addr.sin_port));
215 :
216 179 : req = http_request_init(srv, conn, &addr);
217 179 : if (req == NULL) {
218 0 : close(conn);
219 0 : return;
220 : }
221 :
222 179 : req->next = srv->requests;
223 179 : srv->requests = req;
224 179 : srv->request_count++;
225 : }
226 :
227 :
228 24 : struct http_server * http_server_init(struct in_addr *addr, int port,
229 : void (*cb)(void *ctx,
230 : struct http_request *req),
231 : void *cb_ctx)
232 : {
233 : struct sockaddr_in sin;
234 : struct http_server *srv;
235 24 : int on = 1;
236 :
237 24 : srv = os_zalloc(sizeof(*srv));
238 24 : if (srv == NULL)
239 0 : return NULL;
240 24 : srv->cb = cb;
241 24 : srv->cb_ctx = cb_ctx;
242 :
243 24 : srv->fd = socket(AF_INET, SOCK_STREAM, 0);
244 24 : if (srv->fd < 0)
245 0 : goto fail;
246 :
247 24 : if (setsockopt(srv->fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0)
248 : {
249 0 : wpa_printf(MSG_DEBUG,
250 : "HTTP: setsockopt(SO_REUSEADDR) failed: %s",
251 0 : strerror(errno));
252 : /* try to continue anyway */
253 : }
254 :
255 24 : if (fcntl(srv->fd, F_SETFL, O_NONBLOCK) < 0)
256 0 : goto fail;
257 24 : if (port < 0)
258 24 : srv->port = 49152;
259 : else
260 0 : srv->port = port;
261 :
262 24 : os_memset(&sin, 0, sizeof(sin));
263 24 : sin.sin_family = AF_INET;
264 24 : sin.sin_addr.s_addr = addr->s_addr;
265 :
266 : for (;;) {
267 33 : sin.sin_port = htons(srv->port);
268 33 : if (bind(srv->fd, (struct sockaddr *) &sin, sizeof(sin)) == 0)
269 24 : break;
270 9 : if (errno == EADDRINUSE) {
271 : /* search for unused port */
272 9 : if (++srv->port == 65535 || port >= 0)
273 : goto fail;
274 9 : continue;
275 : }
276 0 : wpa_printf(MSG_DEBUG, "HTTP: Failed to bind server port %d: "
277 0 : "%s", srv->port, strerror(errno));
278 0 : goto fail;
279 9 : }
280 24 : if (listen(srv->fd, 10 /* max backlog */) < 0)
281 0 : goto fail;
282 24 : if (fcntl(srv->fd, F_SETFL, O_NONBLOCK) < 0)
283 0 : goto fail;
284 24 : if (eloop_register_sock(srv->fd, EVENT_TYPE_READ, http_server_cb,
285 : srv, NULL))
286 0 : goto fail;
287 :
288 24 : wpa_printf(MSG_DEBUG, "HTTP: Started server on %s:%d",
289 : inet_ntoa(*addr), srv->port);
290 :
291 24 : return srv;
292 :
293 : fail:
294 0 : http_server_deinit(srv);
295 0 : return NULL;
296 : }
297 :
298 :
299 24 : void http_server_deinit(struct http_server *srv)
300 : {
301 24 : if (srv == NULL)
302 24 : return;
303 24 : if (srv->fd >= 0) {
304 24 : eloop_unregister_sock(srv->fd, EVENT_TYPE_READ);
305 24 : close(srv->fd);
306 : }
307 24 : http_request_free_all(srv->requests);
308 :
309 24 : os_free(srv);
310 : }
311 :
312 :
313 24 : int http_server_get_port(struct http_server *srv)
314 : {
315 24 : return srv->port;
316 : }
|
