Mercurial > urweb

annotate src/c/http.c @ 1924:b44138e6a3bf

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add activeHead tag. activeHead is similar to active but produces xhead instead of xbody
author Sergey Mironov <grrwlf@gmail.com>
date Fri, 06 Dec 2013 13:59:39 +0400
parents 398298ca82b9
children 5a7ae5acdcea
rev   line source
adamc@1268 1 #include "config.h"
adamc@1268 2
adamc@116 3 #include <stdio.h>
adamc@116 4 #include <string.h>
adamc@502 5 #include <stdlib.h>
adamc@116 6 #include <sys/types.h>
adamc@116 7 #include <sys/socket.h>
adamc@116 8 #include <netinet/in.h>
adam@1917 9 #include <netinet/tcp.h>
adam@1779 10 #include <arpa/inet.h>
adamc@472 11 #include <unistd.h>
adamc@502 12 #include <signal.h>
adamc@856 13 #include <stdarg.h>
adamc@116 14
adamc@138 15 #include <pthread.h>
adamc@138 16
adamc@244 17 #include "urweb.h"
adamc@853 18 #include "request.h"
adamc@859 19 #include "queue.h"
adamc@117 20
adamc@1094 21 extern uw_app uw_application;
adamc@1094 22
adam@1914 23 int uw_backlog = SOMAXCONN;
adam@1917 24 static int keepalive = 0;
adamc@116 25
adamc@854 26 static char *get_header(void *data, const char *h) {
adamc@854 27 char *s = data;
adamc@854 28 int len = strlen(h);
adamc@854 29 char *p;
adamc@854 30
adamc@1134 31 while ((p = strchr(s, ':'))) {
adamc@854 32 if (p - s == len && !strncasecmp(s, h, len)) {
adamc@854 33 return p + 2;
adamc@854 34 } else {
adamc@854 35 if ((s = strchr(p, 0)) && s[1] != 0)
adamc@854 36 s += 2;
adamc@854 37 else
adamc@854 38 return NULL;
adamc@854 39 }
adamc@854 40 }
adamc@854 41
adamc@854 42 return NULL;
adamc@854 43 }
adamc@854 44
adam@1799 45 static char *get_env(void *data, const char *name) {
adam@1799 46 return getenv(name);
adam@1799 47 }
adam@1799 48
adamc@856 49 static void on_success(uw_context ctx) {
adamc@856 50 uw_write_header(ctx, "HTTP/1.1 200 OK\r\n");
adamc@856 51 }
adamc@856 52
adamc@856 53 static void on_failure(uw_context ctx) {
adamc@856 54 uw_write_header(ctx, "HTTP/1.1 500 Internal Server Error\r\n");
adamc@856 55 }
adamc@856 56
adamc@856 57 static void log_error(void *data, const char *fmt, ...) {
adamc@856 58 va_list ap;
adamc@856 59 va_start(ap, fmt);
adamc@856 60
adamc@856 61 vfprintf(stderr, fmt, ap);
adamc@856 62 }
adamc@856 63
adamc@856 64 static void log_debug(void *data, const char *fmt, ...) {
adamc@856 65 va_list ap;
adamc@856 66 va_start(ap, fmt);
adamc@856 67
adamc@856 68 vprintf(fmt, ap);
adamc@856 69 }
adamc@856 70
adamc@853 71 static void *worker(void *data) {
adamc@853 72 int me = *(int *)data;
adam@1446 73 uw_context ctx = uw_request_new_context(me, &uw_application, NULL, log_error, log_debug);
adamc@853 74 size_t buf_size = 2;
adam@1917 75 char *buf = malloc(buf_size), *back = buf;
adamc@853 76 uw_request_context rc = uw_new_request_context();
adam@1917 77 int sock = 0;
adamc@698 78
adamc@272 79 while (1) {
adam@1917 80 if (sock == 0) {
adam@1917 81 back = buf;
adam@1917 82 sock = uw_dequeue();
adam@1917 83 }
adamc@138 84
adamc@138 85 printf("Handling connection with thread #%d.\n", me);
adamc@138 86
adamc@138 87 while (1) {
adamc@737 88 int r;
adam@1917 89 char *method, *path, *query_string, *headers, *body, *after, *s, *s2;
adamc@737 90
adamc@742 91 if (back - buf == buf_size - 1) {
adamc@737 92 char *new_buf;
adamc@737 93 buf_size *= 2;
adamc@737 94 new_buf = realloc(buf, buf_size);
adamc@737 95 back = new_buf + (back - buf);
adamc@737 96 buf = new_buf;
adamc@737 97 }
adamc@737 98
adamc@742 99 r = recv(sock, back, buf_size - 1 - (back - buf), 0);
adamc@138 100
adamc@138 101 if (r < 0) {
adamc@138 102 fprintf(stderr, "Recv failed\n");
adam@1917 103 close(sock);
adam@1917 104 sock = 0;
adamc@138 105 break;
adamc@138 106 }
adamc@138 107
adamc@138 108 if (r == 0) {
adamc@138 109 printf("Connection closed.\n");
adam@1917 110 close(sock);
adam@1917 111 sock = 0;
adamc@138 112 break;
adamc@138 113 }
adamc@138 114
adamc@138 115 back += r;
adamc@138 116 *back = 0;
adamc@730 117
adamc@854 118 if ((body = strstr(buf, "\r\n\r\n"))) {
adamc@853 119 request_result rr;
adam@1917 120 int should_keepalive = 0;
adamc@138 121
adamc@854 122 body[0] = body[1] = 0;
adamc@854 123 body += 4;
adamc@854 124
adamc@854 125 if ((s = strcasestr(buf, "\r\nContent-Length: ")) && s < body) {
adamc@853 126 int clen;
adamc@737 127
adamc@854 128 if (sscanf(s + 18, "%d\r\n", &clen) != 1) {
adamc@853 129 fprintf(stderr, "Malformed Content-Length header\n");
adam@1917 130 close(sock);
adam@1917 131 sock = 0;
adamc@167 132 break;
adamc@167 133 }
adamc@167 134
adamc@854 135 while (back - body < clen) {
adamc@854 136 if (back - buf == buf_size - 1) {
adamc@854 137 char *new_buf;
adamc@854 138 buf_size *= 2;
adamc@854 139 new_buf = realloc(buf, buf_size);
adamc@854 140
adamc@854 141 back = new_buf + (back - buf);
adamc@854 142 body = new_buf + (body - buf);
adamc@854 143 s = new_buf + (s - buf);
adamc@854 144
adamc@854 145 buf = new_buf;
adamc@854 146 }
adamc@854 147
adamc@854 148 r = recv(sock, back, buf_size - 1 - (back - buf), 0);
adamc@854 149
adamc@854 150 if (r < 0) {
adamc@854 151 fprintf(stderr, "Recv failed\n");
adamc@854 152 close(sock);
adam@1917 153 sock = 0;
adamc@854 154 goto done;
adamc@854 155 }
adamc@854 156
adamc@854 157 if (r == 0) {
adamc@854 158 fprintf(stderr, "Connection closed.\n");
adamc@854 159 close(sock);
adam@1917 160 sock = 0;
adamc@854 161 goto done;
adamc@854 162 }
adamc@854 163
adamc@854 164 back += r;
adamc@854 165 *back = 0;
adamc@854 166 }
adam@1917 167
adam@1917 168 after = body + clen;
adam@1917 169 } else
adam@1917 170 after = body;
adamc@138 171
adamc@1119 172 body[-4] = '\r';
adamc@1119 173 body[-3] = '\n';
adamc@1119 174
adamc@854 175 if (!(s = strstr(buf, "\r\n"))) {
adamc@854 176 fprintf(stderr, "No newline in request\n");
adamc@854 177 close(sock);
adam@1917 178 sock = 0;
adamc@854 179 goto done;
adamc@854 180 }
adamc@854 181
adamc@1119 182 body[-4] = body[-3] = 0;
adamc@1119 183
adamc@854 184 *s = 0;
adamc@854 185 headers = s + 2;
adamc@854 186 method = s = buf;
adamc@854 187
adam@1426 188 strsep(&s, " ");
adam@1426 189 if (!s) {
adamc@854 190 fprintf(stderr, "No first space in HTTP command\n");
adamc@854 191 close(sock);
adam@1917 192 sock = 0;
adamc@854 193 goto done;
adamc@854 194 }
adamc@854 195 path = s;
adamc@854 196
adamc@1134 197 if ((s = strchr(path, ' ')))
adamc@854 198 *s = 0;
adamc@854 199
adamc@1134 200 if ((s = strchr(path, '?'))) {
adamc@854 201 *s = 0;
adamc@854 202 query_string = s+1;
adamc@854 203 }
adamc@854 204 else
adamc@854 205 query_string = NULL;
adamc@854 206
adamc@854 207 s = headers;
adamc@1134 208 while ((s2 = strchr(s, '\r'))) {
adamc@854 209 s = s2;
adamc@854 210
adamc@854 211 if (s[1] == 0)
adamc@854 212 break;
adamc@854 213
adamc@854 214 *s = 0;
adamc@854 215 s += 2;
adamc@854 216 }
adamc@854 217
adamc@854 218 uw_set_headers(ctx, get_header, headers);
adam@1799 219 uw_set_env(ctx, get_env, NULL);
adamc@854 220
adam@1415 221 printf("Serving URI %s....\n", path);
adamc@856 222 rr = uw_request(rc, ctx, method, path, query_string, body, back - body,
adamc@856 223 on_success, on_failure,
adamc@856 224 NULL, log_error, log_debug,
adamc@863 225 sock, uw_really_send, close);
adam@1917 226
adam@1915 227 if (rr != KEEP_OPEN) {
adam@1915 228 char clen[100];
adam@1915 229
adam@1917 230 if (keepalive) {
adam@1917 231 char *connection = uw_Basis_requestHeader(ctx, "Connection");
adam@1917 232
adam@1917 233 should_keepalive = !(connection && !strcmp(connection, "close"));
adam@1917 234 }
adam@1917 235
adam@1917 236 if (!should_keepalive)
adam@1917 237 uw_write_header(ctx, "Connection: close\r\n");
adam@1917 238
adam@1915 239 sprintf(clen, "Content-length: %d\r\n", uw_pagelen(ctx));
adam@1915 240 uw_write_header(ctx, clen);
adam@1915 241 uw_send(ctx, sock);
adam@1915 242 }
adamc@138 243
adam@1917 244 if (rr == SERVED || rr == FAILED) {
adam@1917 245 if (should_keepalive) {
adam@1917 246 // In case any other requests are queued up, shift
adam@1917 247 // unprocessed part of buffer to front.
adam@1917 248 int kept = back - after;
adam@1917 249 memmove(buf, after, kept);
adam@1917 250 back = buf + kept;
adam@1917 251 } else {
adam@1917 252 close(sock);
adam@1917 253 sock = 0;
adam@1917 254 }
adam@1917 255 } else if (rr != KEEP_OPEN)
adamc@853 256 fprintf(stderr, "Illegal uw_request return code: %d\n", rr);
adamc@853 257
adamc@138 258 break;
adamc@138 259 }
adamc@116 260 }
adamc@116 261
adamc@854 262 done:
adamc@311 263 uw_reset(ctx);
adamc@116 264 }
adamc@1138 265
adamc@1138 266 return NULL;
adamc@116 267 }
adamc@102 268
adamc@477 269 static void help(char *cmd) {
adam@1917 270 printf("Usage: %s [-p <port>] [-a <IP address>] [-t <thread count>] [-k]\nThe '-k' option turns on HTTP keepalive.\n", cmd);
adamc@477 271 }
adamc@477 272
adamc@502 273 static void sigint(int signum) {
adamc@502 274 printf("Exiting....\n");
adamc@502 275 exit(0);
adamc@502 276 }
adamc@502 277
adamc@1094 278 static loggers ls = {&uw_application, NULL, log_error, log_debug};
adamc@856 279
adamc@138 280 int main(int argc, char *argv[]) {
adamc@116 281 // The skeleton for this function comes from Beej's sockets tutorial.
adamc@138 282 int sockfd; // listen on sock_fd
adamc@116 283 struct sockaddr_in my_addr;
adamc@116 284 struct sockaddr_in their_addr; // connector's address information
adamc@1134 285 socklen_t sin_size;
adamc@1134 286 int yes = 1, uw_port = 8080, nthreads = 1, i, *names, opt;
adamc@502 287
adamc@502 288 signal(SIGINT, sigint);
adamc@505 289 signal(SIGPIPE, SIG_IGN);
adamc@505 290
adam@1779 291 my_addr.sin_addr.s_addr = INADDR_ANY; // auto-fill with my IP
adam@1779 292 memset(my_addr.sin_zero, '\0', sizeof my_addr.sin_zero);
adam@1779 293
adam@1917 294 while ((opt = getopt(argc, argv, "hp:a:t:k")) != -1) {
adamc@472 295 switch (opt) {
adamc@472 296 case '?':
adamc@472 297 fprintf(stderr, "Unknown command-line option");
adamc@477 298 help(argv[0]);
adamc@472 299 return 1;
adamc@138 300
adamc@477 301 case 'h':
adamc@477 302 help(argv[0]);
adamc@477 303 return 0;
adamc@477 304
adamc@472 305 case 'p':
adamc@472 306 uw_port = atoi(optarg);
adamc@472 307 if (uw_port <= 0) {
adamc@472 308 fprintf(stderr, "Invalid port number\n");
adamc@477 309 help(argv[0]);
adamc@472 310 return 1;
adamc@472 311 }
adamc@472 312 break;
adamc@472 313
adam@1779 314 case 'a':
adam@1779 315 if (!inet_pton(AF_INET, optarg, &my_addr.sin_addr)) {
adam@1779 316 fprintf(stderr, "Invalid IP address\n");
adam@1779 317 help(argv[0]);
adam@1779 318 return 1;
adam@1779 319 }
adam@1779 320 break;
adam@1779 321
adamc@472 322 case 't':
adamc@472 323 nthreads = atoi(optarg);
adamc@472 324 if (nthreads <= 0) {
adamc@472 325 fprintf(stderr, "Invalid thread count\n");
adamc@477 326 help(argv[0]);
adamc@472 327 return 1;
adamc@472 328 }
adamc@472 329 break;
adamc@472 330
adam@1917 331 case 'k':
adam@1917 332 keepalive = 1;
adam@1917 333 break;
adam@1917 334
adamc@472 335 default:
adamc@472 336 fprintf(stderr, "Unexpected getopt() behavior\n");
adamc@472 337 return 1;
adamc@472 338 }
adamc@138 339 }
adamc@138 340
adamc@1094 341 uw_request_init(&uw_application, NULL, log_error, log_debug);
adamc@687 342
adamc@138 343 names = calloc(nthreads, sizeof(int));
adamc@116 344
adamc@116 345 sockfd = socket(PF_INET, SOCK_STREAM, 0); // do some error checking!
adamc@116 346
adamc@116 347 if (sockfd < 0) {
adamc@116 348 fprintf(stderr, "Listener socket creation failed\n");
adamc@116 349 return 1;
adamc@116 350 }
adamc@116 351
adamc@116 352 if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) < 0) {
adamc@116 353 fprintf(stderr, "Listener socket option setting failed\n");
adamc@116 354 return 1;
adamc@116 355 }
adamc@116 356
adamc@116 357 my_addr.sin_family = AF_INET; // host byte order
adamc@311 358 my_addr.sin_port = htons(uw_port); // short, network byte order
adamc@116 359
adamc@116 360 if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof my_addr) < 0) {
adamc@116 361 fprintf(stderr, "Listener socket bind failed\n");
adamc@116 362 return 1;
adamc@116 363 }
adamc@116 364
adamc@311 365 if (listen(sockfd, uw_backlog) < 0) {
adamc@116 366 fprintf(stderr, "Socket listen failed\n");
adamc@116 367 return 1;
adamc@116 368 }
adamc@116 369
adamc@116 370 sin_size = sizeof their_addr;
adamc@116 371
adamc@311 372 printf("Listening on port %d....\n", uw_port);
adamc@116 373
adamc@667 374 {
adamc@667 375 pthread_t thread;
adamc@667 376
adam@1522 377 if (pthread_create_big(&thread, NULL, client_pruner, &ls)) {
adamc@667 378 fprintf(stderr, "Error creating pruner thread\n");
adamc@667 379 return 1;
adamc@667 380 }
adamc@667 381 }
adamc@667 382
adamc@138 383 for (i = 0; i < nthreads; ++i) {
adamc@138 384 pthread_t thread;
adamc@138 385 names[i] = i;
adam@1522 386 if (pthread_create_big(&thread, NULL, worker, &names[i])) {
adamc@138 387 fprintf(stderr, "Error creating worker thread #%d\n", i);
adamc@138 388 return 1;
adamc@138 389 }
adamc@138 390 }
adamc@138 391
adamc@116 392 while (1) {
adamc@138 393 int new_fd = accept(sockfd, (struct sockaddr *)&their_addr, &sin_size);
adamc@116 394
adamc@116 395 if (new_fd < 0) {
adamc@116 396 fprintf(stderr, "Socket accept failed\n");
adamc@116 397 return 1;
adamc@116 398 }
adamc@116 399
adamc@116 400 printf("Accepted connection.\n");
adamc@138 401
adam@1917 402 if (keepalive) {
adam@1917 403 int flag = 1;
adam@1917 404 setsockopt(new_fd, IPPROTO_TCP, TCP_NODELAY, (char *) &flag, sizeof(int));
adam@1917 405 }
adam@1917 406
adamc@859 407 uw_enqueue(new_fd);
adamc@116 408 }
adamc@102 409 }
adamc@1121 410
adamc@1121 411 void *uw_init_client_data() {
adamc@1121 412 return NULL;
adamc@1121 413 }
adamc@1121 414
adamc@1121 415 void uw_free_client_data(void *data) {
adamc@1121 416 }
adamc@1121 417
adamc@1121 418 void uw_copy_client_data(void *dst, void *src) {
adamc@1121 419 }
adamc@1121 420
adamc@1121 421 void uw_do_expunge(uw_context ctx, uw_Basis_client cli, void *data) {
adamc@1121 422 if (uw_get_app(ctx)->db_begin(ctx))
adamc@1121 423 uw_error(ctx, FATAL, "Error running SQL BEGIN");
adamc@1121 424 uw_get_app(ctx)->expunger(ctx, cli);
adam@1672 425 uw_commit(ctx);
adamc@1121 426 }
adamc@1121 427
adamc@1121 428 void uw_post_expunge(uw_context ctx, void *data) {
adamc@1121 429 }
adam@1320 430
adam@1320 431 int uw_supports_direct_status = 1;