http_server.c revision c5ec7f57ead87efa365800228aa0b09a12d9e6c4
1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver/* 2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * http_server - HTTP server 3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Copyright (c) 2009, Jouni Malinen <j@w1.fi> 4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * This software may be distributed under the terms of the BSD license. 6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * See README for more details. 7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver */ 8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "includes.h" 10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include <fcntl.h> 11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "common.h" 13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "eloop.h" 14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "httpread.h" 15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "http_server.h" 16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#define HTTP_SERVER_TIMEOUT 30 18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#define HTTP_SERVER_MAX_REQ_LEN 8000 19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#define HTTP_SERVER_MAX_CONNECTIONS 10 20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverstruct http_request { 22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver struct http_request *next; 23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver struct http_server *srv; 24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver int fd; 25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver struct sockaddr_in cli; 26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver struct httpread *hread; 27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver}; 28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverstruct http_server { 30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver void (*cb)(void *ctx, struct http_request *req); 31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver void *cb_ctx; 32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver int fd; 34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver int port; 35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver struct http_request *requests; 37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver unsigned int request_count; 38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver}; 39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverstatic void http_request_cb(struct httpread *handle, void *cookie, 42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver enum httpread_event en) 43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver struct http_request *req = cookie; 45324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver struct http_server *srv = req->srv; 46324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 47324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (en == HTTPREAD_EVENT_FILE_READY) { 48324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver wpa_printf(MSG_DEBUG, "HTTP: Request from %s:%d received", 49324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver inet_ntoa(req->cli.sin_addr), 50324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver ntohs(req->cli.sin_port)); 51324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver srv->cb(srv->cb_ctx, req); 52324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return; 53324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 54324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver wpa_printf(MSG_DEBUG, "HTTP: Request from %s:%d could not be received " 55324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver "completely", inet_ntoa(req->cli.sin_addr), 56324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver ntohs(req->cli.sin_port)); 57324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver http_request_deinit(req); 58324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 59324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 60324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 61324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverstatic struct http_request * http_request_init(struct http_server *srv, int fd, 62324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver struct sockaddr_in *cli) 63324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 64324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver struct http_request *req; 65324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 66324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (srv->request_count >= HTTP_SERVER_MAX_CONNECTIONS) { 67324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver wpa_printf(MSG_DEBUG, "HTTP: Too many concurrent requests"); 68324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return NULL; 69324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 70324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 71324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver req = os_zalloc(sizeof(*req)); 72324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (req == NULL) 73324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return NULL; 74324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 75324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver req->srv = srv; 76324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver req->fd = fd; 77324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver req->cli = *cli; 78324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 79324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver req->hread = httpread_create(req->fd, http_request_cb, req, 80324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver HTTP_SERVER_MAX_REQ_LEN, 81324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver HTTP_SERVER_TIMEOUT); 82324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (req->hread == NULL) { 83324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver http_request_deinit(req); 84324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return NULL; 85324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 86324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 87324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return req; 88324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 89324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 90 91void http_request_deinit(struct http_request *req) 92{ 93 struct http_request *r, *p; 94 struct http_server *srv; 95 96 if (req == NULL) 97 return; 98 99 srv = req->srv; 100 p = NULL; 101 r = srv->requests; 102 while (r) { 103 if (r == req) { 104 if (p) 105 p->next = r->next; 106 else 107 srv->requests = r->next; 108 srv->request_count--; 109 break; 110 } 111 p = r; 112 r = r->next; 113 } 114 115 httpread_destroy(req->hread); 116 close(req->fd); 117 os_free(req); 118} 119 120 121static void http_request_free_all(struct http_request *req) 122{ 123 struct http_request *prev; 124 while (req) { 125 prev = req; 126 req = req->next; 127 http_request_deinit(prev); 128 } 129} 130 131 132void http_request_send(struct http_request *req, struct wpabuf *resp) 133{ 134 int res; 135 136 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 ntohs(req->cli.sin_port)); 140 141 res = send(req->fd, wpabuf_head(resp), wpabuf_len(resp), 0); 142 if (res < 0) { 143 wpa_printf(MSG_DEBUG, "HTTP: Send failed: %s", 144 strerror(errno)); 145 } else if ((size_t) res < wpabuf_len(resp)) { 146 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 wpabuf_free(resp); 152} 153 154 155void http_request_send_and_deinit(struct http_request *req, 156 struct wpabuf *resp) 157{ 158 http_request_send(req, resp); 159 http_request_deinit(req); 160} 161 162 163enum httpread_hdr_type http_request_get_type(struct http_request *req) 164{ 165 return httpread_hdr_type_get(req->hread); 166} 167 168 169char * http_request_get_uri(struct http_request *req) 170{ 171 return httpread_uri_get(req->hread); 172} 173 174 175char * http_request_get_hdr(struct http_request *req) 176{ 177 return httpread_hdr_get(req->hread); 178} 179 180 181char * http_request_get_data(struct http_request *req) 182{ 183 return httpread_data_get(req->hread); 184} 185 186 187char * http_request_get_hdr_line(struct http_request *req, const char *tag) 188{ 189 return httpread_hdr_line_get(req->hread, tag); 190} 191 192 193struct sockaddr_in * http_request_get_cli_addr(struct http_request *req) 194{ 195 return &req->cli; 196} 197 198 199static void http_server_cb(int sd, void *eloop_ctx, void *sock_ctx) 200{ 201 struct sockaddr_in addr; 202 socklen_t addr_len = sizeof(addr); 203 struct http_server *srv = eloop_ctx; 204 int conn; 205 struct http_request *req; 206 207 conn = accept(srv->fd, (struct sockaddr *) &addr, &addr_len); 208 if (conn < 0) { 209 wpa_printf(MSG_DEBUG, "HTTP: Failed to accept new connection: " 210 "%s", strerror(errno)); 211 return; 212 } 213 wpa_printf(MSG_DEBUG, "HTTP: Connection from %s:%d", 214 inet_ntoa(addr.sin_addr), ntohs(addr.sin_port)); 215 216 req = http_request_init(srv, conn, &addr); 217 if (req == NULL) { 218 close(conn); 219 return; 220 } 221 222 req->next = srv->requests; 223 srv->requests = req; 224 srv->request_count++; 225} 226 227 228struct 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 236 srv = os_zalloc(sizeof(*srv)); 237 if (srv == NULL) 238 return NULL; 239 srv->cb = cb; 240 srv->cb_ctx = cb_ctx; 241 242 srv->fd = socket(AF_INET, SOCK_STREAM, 0); 243 if (srv->fd < 0) 244 goto fail; 245 if (fcntl(srv->fd, F_SETFL, O_NONBLOCK) < 0) 246 goto fail; 247 if (port < 0) 248 srv->port = 49152; 249 else 250 srv->port = port; 251 252 os_memset(&sin, 0, sizeof(sin)); 253 sin.sin_family = AF_INET; 254 sin.sin_addr.s_addr = addr->s_addr; 255 256 for (;;) { 257 sin.sin_port = htons(srv->port); 258 if (bind(srv->fd, (struct sockaddr *) &sin, sizeof(sin)) == 0) 259 break; 260 if (errno == EADDRINUSE) { 261 /* search for unused port */ 262 if (++srv->port == 65535 || port >= 0) 263 goto fail; 264 continue; 265 } 266 wpa_printf(MSG_DEBUG, "HTTP: Failed to bind server port %d: " 267 "%s", srv->port, strerror(errno)); 268 goto fail; 269 } 270 if (listen(srv->fd, 10 /* max backlog */) < 0) 271 goto fail; 272 if (fcntl(srv->fd, F_SETFL, O_NONBLOCK) < 0) 273 goto fail; 274 if (eloop_register_sock(srv->fd, EVENT_TYPE_READ, http_server_cb, 275 srv, NULL)) 276 goto fail; 277 278 wpa_printf(MSG_DEBUG, "HTTP: Started server on %s:%d", 279 inet_ntoa(*addr), srv->port); 280 281 return srv; 282 283fail: 284 http_server_deinit(srv); 285 return NULL; 286} 287 288 289void http_server_deinit(struct http_server *srv) 290{ 291 if (srv == NULL) 292 return; 293 if (srv->fd >= 0) { 294 eloop_unregister_sock(srv->fd, EVENT_TYPE_READ); 295 close(srv->fd); 296 } 297 http_request_free_all(srv->requests); 298 299 os_free(srv); 300} 301 302 303int http_server_get_port(struct http_server *srv) 304{ 305 return srv->port; 306} 307