Mercurial > urweb

annotate src/c/http.c @ 1517:27b8c0a460cf

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