adamc@1268: #include "config.h"
adamc@1268: 
adamc@116: #include <stdio.h>
adamc@116: #include <string.h>
adamc@502: #include <stdlib.h>
adamc@116: #include <sys/types.h>
adamc@116: #include <sys/socket.h>
adamc@116: #include <netinet/in.h>
adam@1917: #include <netinet/tcp.h>
adam@1779: #include <arpa/inet.h>
adamc@472: #include <unistd.h>
adamc@502: #include <signal.h>
adamc@856: #include <stdarg.h>
adamc@116: 
adamc@138: #include <pthread.h>
adamc@138: 
adamc@244: #include "urweb.h"
adamc@853: #include "request.h"
adamc@859: #include "queue.h"
adamc@117: 
adamc@1094: extern uw_app uw_application;
adamc@1094: 
adam@1914: int uw_backlog = SOMAXCONN;
adam@1930: static int keepalive = 0, quiet = 0;
adamc@116: 
grrwlf@2069: #define qfprintf(f, fmt, args...) do { if(!quiet) fprintf(f, fmt, ##args); } while(0)
grrwlf@2069: #define qprintf(fmt, args...) do { if(!quiet) printf(fmt, ##args); } while(0)
grrwlf@2069: 
adamc@854: static char *get_header(void *data, const char *h) {
adamc@854:   char *s = data;
adamc@854:   int len = strlen(h);
adamc@854:   char *p;
adamc@854: 
adamc@1134:   while ((p = strchr(s, ':'))) {
adamc@854:     if (p - s == len && !strncasecmp(s, h, len)) {
adamc@854:       return p + 2;
adamc@854:     } else {
adamc@854:       if ((s = strchr(p, 0)) && s[1] != 0)
adamc@854:         s += 2;
adamc@854:       else
adamc@854:         return NULL;
adamc@854:     }
adamc@854:   }
adamc@854:   
adamc@854:   return NULL;
adamc@854: }
adamc@854: 
adam@1799: static char *get_env(void *data, const char *name) {
adam@1799:   return getenv(name);
adam@1799: }
adam@1799: 
adamc@856: static void on_success(uw_context ctx) {
adamc@856:   uw_write_header(ctx, "HTTP/1.1 200 OK\r\n");
adamc@856: }
adamc@856: 
adamc@856: static void on_failure(uw_context ctx) {
adamc@856:   uw_write_header(ctx, "HTTP/1.1 500 Internal Server Error\r\n");
adamc@856: }
adamc@856: 
adamc@856: static void log_error(void *data, const char *fmt, ...) {
adamc@856:   va_list ap;
adamc@856:   va_start(ap, fmt);
adamc@856: 
adamc@856:   vfprintf(stderr, fmt, ap);
adamc@856: }
adamc@856: 
adamc@856: static void log_debug(void *data, const char *fmt, ...) {
adam@1930:   if (!quiet) {
adam@1930:     va_list ap;
adam@1930:     va_start(ap, fmt);
adamc@856: 
adam@1930:     vprintf(fmt, ap);
adam@1930:   }
adamc@856: }
adamc@856: 
grrwlf@1997: static uw_loggers ls = {NULL, log_error, log_debug};
grrwlf@1997: 
adamc@853: static void *worker(void *data) {
adamc@853:   int me = *(int *)data;
grrwlf@1997:   uw_context ctx = uw_request_new_context(me, &uw_application, &ls);
adam@1937:   size_t buf_size = 1024;
adam@1917:   char *buf = malloc(buf_size), *back = buf;
adamc@853:   uw_request_context rc = uw_new_request_context();
adam@1917:   int sock = 0;
adamc@698: 
adamc@272:   while (1) {
adam@1917:     if (sock == 0) {
adam@1917:       back = buf;
adam@1917:       sock = uw_dequeue();
adam@1917:     }
adamc@138: 
adam@2073:     uw_set_remoteSock(ctx, sock);
adam@2073: 
grrwlf@2069:     qprintf("Handling connection with thread #%d.\n", me);
adamc@138: 
adamc@138:     while (1) {
adamc@737:       int r;
adam@1917:       char *method, *path, *query_string, *headers, *body, *after, *s, *s2;
adamc@737: 
adamc@742:       if (back - buf == buf_size - 1) {
adamc@737:         char *new_buf;
grrwlf@2070:         size_t new_buf_size = buf_size*2;
grrwlf@2070:         new_buf = realloc(buf, new_buf_size);
grrwlf@2070:         if(!new_buf) {
grrwlf@2070:           qfprintf(stderr, "Realloc failed while receiving header\n");
grrwlf@2070:           close(sock);
grrwlf@2070:           sock = 0;
grrwlf@2070:           break;
grrwlf@2070:         }
grrwlf@2070:         buf_size = new_buf_size;
adamc@737:         back = new_buf + (back - buf);
adamc@737:         buf = new_buf;
adamc@737:       }
adamc@737: 
adam@1937:       *back = 0;
adam@1937:       body = strstr(buf, "\r\n\r\n");
adam@1937:       if (body == NULL) {
adam@1937:         r = recv(sock, back, buf_size - 1 - (back - buf), 0);
adamc@138: 
adam@1937:         if (r < 0) {
grrwlf@2071:           qfprintf(stderr, "Recv failed while receiving header, retcode %d errno %m\n", r);
adam@1937:           close(sock);
adam@1937:           sock = 0;
adam@1937:           break;
adam@1937:         }
adam@1937: 
adam@1937:         if (r == 0) {
grrwlf@2069:           qprintf("Connection closed.\n");
adam@1937:           close(sock);
adam@1937:           sock = 0;
adam@1937:           break;
adam@1937:         }
adam@1937: 
adam@1937:         back += r;
adam@1937:         *back = 0;
adamc@138:       }
adamc@138: 
adam@1937:       if (body != NULL || (body = strstr(buf, "\r\n\r\n"))) {
adamc@853:         request_result rr;
adam@1917:         int should_keepalive = 0;
adamc@138: 
adamc@854:         body[0] = body[1] = 0;
adamc@854:         body += 4;
adamc@854: 
adamc@854:         if ((s = strcasestr(buf, "\r\nContent-Length: ")) && s < body) {
adamc@853:           int clen;
adamc@737: 
adamc@854:           if (sscanf(s + 18, "%d\r\n", &clen) != 1) {
adamc@853:             fprintf(stderr, "Malformed Content-Length header\n");
adam@1917:             close(sock);
adam@1917:             sock = 0;
adamc@167:             break;
adamc@167:           }
adamc@167: 
adamc@854:           while (back - body < clen) {
adamc@854:             if (back - buf == buf_size - 1) {
adamc@854:               char *new_buf;
grrwlf@2070:               size_t new_buf_size = buf_size * 2;
grrwlf@2070:               new_buf = realloc(buf, new_buf_size);
grrwlf@2070:               if(!new_buf) {
grrwlf@2070:                 qfprintf(stderr, "Realloc failed while receiving content\n");
grrwlf@2070:                 close(sock);
grrwlf@2070:                 sock = 0;
grrwlf@2070:                 goto done;
grrwlf@2070:               }
adamc@854: 
grrwlf@2070:               buf_size = new_buf_size;
adamc@854:               back = new_buf + (back - buf);
adamc@854:               body = new_buf + (body - buf);
adamc@854:               s = new_buf + (s - buf);
adamc@854: 
adamc@854:               buf = new_buf;
adamc@854:             }
adamc@854: 
adamc@854:             r = recv(sock, back, buf_size - 1 - (back - buf), 0);
adamc@854: 
adamc@854:             if (r < 0) {
grrwlf@2071:               qfprintf(stderr, "Recv failed while receiving content, retcode %d errno %m\n", r);
adamc@854:               close(sock);
adam@1917:               sock = 0;
adamc@854:               goto done;
adamc@854:             }
adamc@854: 
adamc@854:             if (r == 0) {
grrwlf@2069:               qfprintf(stderr, "Connection closed.\n");
adamc@854:               close(sock);
adam@1917:               sock = 0;
adamc@854:               goto done;
adamc@854:             }
adamc@854: 
adamc@854:             back += r;
adamc@854:             *back = 0;      
adamc@854:           }
adam@1917: 
adam@1917:           after = body + clen;
adam@1917:         } else
adam@1917:           after = body;
adamc@138: 
adamc@1119:         body[-4] = '\r';
adamc@1119:         body[-3] = '\n';
adamc@1119: 
adamc@854:         if (!(s = strstr(buf, "\r\n"))) {
adamc@854:           fprintf(stderr, "No newline in request\n");
adamc@854:           close(sock);
adam@1917:           sock = 0;
adamc@854:           goto done;
adamc@854:         }
adamc@854: 
adamc@1119:         body[-4] = body[-3] = 0;
adamc@1119: 
adamc@854:         *s = 0;
adamc@854:         headers = s + 2;
adamc@854:         method = s = buf;
adamc@854: 
adam@1426:         strsep(&s, " ");
adam@1426:         if (!s) {
adamc@854:           fprintf(stderr, "No first space in HTTP command\n");
adamc@854:           close(sock);
adam@1917:           sock = 0;
adamc@854:           goto done;
adamc@854:         }
adamc@854:         path = s;
adamc@854: 
adamc@1134:         if ((s = strchr(path, ' ')))
adamc@854:           *s = 0;
adamc@854: 
adamc@1134:         if ((s = strchr(path, '?'))) {
adamc@854:           *s = 0;
adamc@854:           query_string = s+1;
adamc@854:         }
adamc@854:         else
adamc@854:           query_string = NULL;
adamc@854: 
adamc@854:         s = headers;
adamc@1134:         while ((s2 = strchr(s, '\r'))) {
adam@1937:           if (s2 == s) {
adam@1937:             *s = 0;
adam@1937:             break;
adam@1937:           }
adam@1937: 
adamc@854:           s = s2;
adamc@854: 
adamc@854:           if (s[1] == 0)
adamc@854:             break;
adamc@854: 
adamc@854:           *s = 0;
adamc@854:           s += 2;
adamc@854:         }
adamc@854: 
adamc@854:         uw_set_headers(ctx, get_header, headers);
adam@1799:         uw_set_env(ctx, get_env, NULL);
adamc@854: 
grrwlf@2069:         qprintf("Serving URI %s....\n", path);
adamc@856:         rr = uw_request(rc, ctx, method, path, query_string, body, back - body,
adamc@856:                         on_success, on_failure,
adamc@856:                         NULL, log_error, log_debug,
adamc@863:                         sock, uw_really_send, close);
adam@1917: 
adam@1915:         if (rr != KEEP_OPEN) {
adam@1917:           if (keepalive) {
adam@1917:             char *connection = uw_Basis_requestHeader(ctx, "Connection");
adam@1917: 
adam@1917:             should_keepalive = !(connection && !strcmp(connection, "close"));
adam@1917:           }
adam@1917: 
adam@1917:           if (!should_keepalive)
adam@1917:             uw_write_header(ctx, "Connection: close\r\n");
adam@1917: 
adam@1934:           if (!uw_has_contentLength(ctx)) {
adam@1934:             char clen[100];
adam@1934: 
adam@1934:             sprintf(clen, "Content-length: %d\r\n", uw_pagelen(ctx));
adam@1934:             uw_write_header(ctx, clen);
adam@1934:           }
adam@1934: 
adam@1915:           uw_send(ctx, sock);
adam@1915:         }
adamc@138: 
adam@1917:         if (rr == SERVED || rr == FAILED) {
adam@1917:           if (should_keepalive) {
adam@1917:             // In case any other requests are queued up, shift
adam@1917:             // unprocessed part of buffer to front.
adam@1917:             int kept = back - after;
adam@1944: 
adam@1945:             if (kept == 0) {
adam@1944:               // No pipelining going on here.
adam@1944:               // We'd might as well try to switch to a different connection,
adam@1944:               // while we wait for more input on this one.
adam@1944:               uw_enqueue(sock);
adam@1944:               sock = 0;
adam@1944:             } else {
adam@1944:               // More input!  Move it to the front and continue in this loop.
adam@1944:               memmove(buf, after, kept);
adam@1944:               back = buf + kept;
adam@1944:             }
adam@1917:           } else {
adam@1917:             close(sock);
adam@1917:             sock = 0;
adam@1917:           }
adam@1931:         } else if (rr == KEEP_OPEN)
adam@1931:           sock = 0;
adam@1931:         else
adamc@853:           fprintf(stderr, "Illegal uw_request return code: %d\n", rr);
adamc@853: 
adamc@138:         break;
adamc@138:       }
adamc@116:     }
adamc@116: 
adamc@854:   done:
adamc@311:     uw_reset(ctx);
adamc@116:   }
adamc@1138: 
adamc@1138:   return NULL;
adamc@116: }
adamc@102: 
adamc@477: static void help(char *cmd) {
adam@2172:   printf("Usage: %s [-p <port>] [-a <IP address>] [-t <thread count>] [-k] [-q] [-T SEC]\nThe '-k' option turns on HTTP keepalive.\nThe '-q' option turns off some chatter on stdout.\nThe '-T' option sets socket recv timeout (0 disables timeout, default is 5 sec).\n", cmd);
adamc@477: }
adamc@477: 
adamc@502: static void sigint(int signum) {
adamc@502:   printf("Exiting....\n");
adamc@502:   exit(0);
adamc@502: }
adamc@502: 
adamc@138: int main(int argc, char *argv[]) {
adamc@116:   // The skeleton for this function comes from Beej's sockets tutorial.
adamc@138:   int sockfd;  // listen on sock_fd
adamc@116:   struct sockaddr_in my_addr;
adamc@116:   struct sockaddr_in their_addr; // connector's address information
adamc@1134:   socklen_t sin_size;
adamc@1134:   int yes = 1, uw_port = 8080, nthreads = 1, i, *names, opt;
grrwlf@2071:   int recv_timeout_sec = 5;
adamc@502:  
adamc@502:   signal(SIGINT, sigint);
adamc@505:   signal(SIGPIPE, SIG_IGN); 
adamc@505: 
adam@1779:   my_addr.sin_addr.s_addr = INADDR_ANY; // auto-fill with my IP
adam@1779:   memset(my_addr.sin_zero, '\0', sizeof my_addr.sin_zero);
adam@1779: 
grrwlf@2071:   while ((opt = getopt(argc, argv, "hp:a:t:kqT:")) != -1) {
adamc@472:     switch (opt) {
adamc@472:     case '?':
adam@1930:       fprintf(stderr, "Unknown command-line option\n");
adamc@477:       help(argv[0]);
adamc@472:       return 1;
adamc@138: 
adamc@477:     case 'h':
adamc@477:       help(argv[0]);
adamc@477:       return 0;
adamc@477: 
adamc@472:     case 'p':
adamc@472:       uw_port = atoi(optarg);
adamc@472:       if (uw_port <= 0) {
adamc@472:         fprintf(stderr, "Invalid port number\n");
adamc@477:         help(argv[0]);
adamc@472:         return 1;
adamc@472:       }
adamc@472:       break;
adamc@472: 
adam@1779:     case 'a':
adam@1779:       if (!inet_pton(AF_INET, optarg, &my_addr.sin_addr)) {
adam@1779:         fprintf(stderr, "Invalid IP address\n");
adam@1779:         help(argv[0]);
adam@1779:         return 1;
adam@1779:       }
adam@1779:       break;
adam@1779: 
adamc@472:     case 't':
adamc@472:       nthreads = atoi(optarg);
adamc@472:       if (nthreads <= 0) {
adamc@472:         fprintf(stderr, "Invalid thread count\n");
adamc@477:         help(argv[0]);
adamc@472:         return 1;
adamc@472:       }
adamc@472:       break;
adamc@472: 
adam@1917:     case 'k':
adam@1917:       keepalive = 1;
adam@1917:       break;
adam@1917: 
grrwlf@2071:     case 'T':
grrwlf@2071:       recv_timeout_sec = atoi(optarg);
grrwlf@2071:       if (recv_timeout_sec < 0) {
grrwlf@2071:         fprintf(stderr, "Invalid recv timeout\n");
grrwlf@2071:         help(argv[0]);
grrwlf@2071:         return 1;
grrwlf@2071:       }
grrwlf@2071:       break;
grrwlf@2071: 
adam@1930:     case 'q':
adam@1930:       quiet = 1;
adam@1930:       break;
adam@1930: 
adamc@472:     default:
adamc@472:       fprintf(stderr, "Unexpected getopt() behavior\n");
adamc@472:       return 1;
adamc@472:     }
adamc@138:   }
adamc@138: 
grrwlf@1997:   uw_request_init(&uw_application, &ls);
adamc@687: 
adamc@138:   names = calloc(nthreads, sizeof(int));
adamc@116: 
adamc@116:   sockfd = socket(PF_INET, SOCK_STREAM, 0); // do some error checking!
adamc@116: 
adamc@116:   if (sockfd < 0) {
adamc@116:     fprintf(stderr, "Listener socket creation failed\n");
adamc@116:     return 1;
adamc@116:   }
adamc@116: 
adamc@116:   if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) < 0) {
adamc@116:     fprintf(stderr, "Listener socket option setting failed\n");
adamc@116:     return 1;
adamc@116:   }
adamc@116: 
adamc@116:   my_addr.sin_family = AF_INET;         // host byte order
adamc@311:   my_addr.sin_port = htons(uw_port);    // short, network byte order
adamc@116: 
adamc@116:   if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof my_addr) < 0) {
adamc@116:     fprintf(stderr, "Listener socket bind failed\n");
adamc@116:     return 1;
adamc@116:   }
adamc@116: 
adamc@311:   if (listen(sockfd, uw_backlog) < 0) {
adamc@116:     fprintf(stderr, "Socket listen failed\n");
adamc@116:     return 1;
adamc@116:   }
adamc@116: 
adamc@116:   sin_size = sizeof their_addr;
adamc@116: 
grrwlf@2069:   qprintf("Listening on port %d....\n", uw_port);
adamc@116: 
adamc@667:   {
adamc@667:     pthread_t thread;
adamc@667: 
grrwlf@1997:     pruner_data *pd = (pruner_data *)malloc(sizeof(pruner_data));
grrwlf@1997:     pd->app = &uw_application;
grrwlf@1997:     pd->loggers = &ls;
grrwlf@1997: 
grrwlf@1997:     if (pthread_create_big(&thread, NULL, client_pruner, pd)) {
adamc@667:       fprintf(stderr, "Error creating pruner thread\n");
adamc@667:       return 1;
adamc@667:     }
adamc@667:   }
adamc@667: 
adamc@138:   for (i = 0; i < nthreads; ++i) {
adamc@138:     pthread_t thread;    
adamc@138:     names[i] = i;
adam@1522:     if (pthread_create_big(&thread, NULL, worker, &names[i])) {
adamc@138:       fprintf(stderr, "Error creating worker thread #%d\n", i);
adamc@138:       return 1;
adamc@138:     }
adamc@138:   }
adamc@138: 
adamc@116:   while (1) {
adamc@138:     int new_fd = accept(sockfd, (struct sockaddr *)&their_addr, &sin_size);
adamc@116: 
adamc@116:     if (new_fd < 0) {
grrwlf@2069:       qfprintf(stderr, "Socket accept failed\n");
adam@1948:     } else {
grrwlf@2069:       qprintf("Accepted connection.\n");
adam@1948: 
adam@1948:       if (keepalive) {
adam@1948:         int flag = 1; 
adam@1948:         setsockopt(new_fd, IPPROTO_TCP, TCP_NODELAY, (char *) &flag, sizeof(int));
adam@1948:       }
adam@1948: 
grrwlf@2071:       if(recv_timeout_sec>0) {
grrwlf@2071:         int ret;
grrwlf@2071:         struct timeval tv;
grrwlf@2071:         memset(&tv, 0, sizeof(struct timeval));
grrwlf@2071:         tv.tv_sec = recv_timeout_sec;
grrwlf@2071:         ret = setsockopt(new_fd, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof(struct timeval));
grrwlf@2071:         if(ret != 0) {
grrwlf@2071:           qfprintf(stderr, "Timeout setting failed, errcode %d errno '%m'\n", ret);
grrwlf@2071:         }
grrwlf@2071:       }
grrwlf@2071: 
adam@1948:       uw_enqueue(new_fd);
adamc@116:     }
adamc@116:   }
adamc@102: }
adamc@1121: 
adamc@1121: void *uw_init_client_data() {
adamc@1121:   return NULL;
adamc@1121: }
adamc@1121: 
adamc@1121: void uw_free_client_data(void *data) {
adamc@1121: }
adamc@1121: 
adamc@1121: void uw_copy_client_data(void *dst, void *src) {
adamc@1121: }
adamc@1121: 
adamc@1121: void uw_do_expunge(uw_context ctx, uw_Basis_client cli, void *data) {
adam@1941:   uw_ensure_transaction(ctx);
adam@1941:   uw_get_app(ctx)->expunger(ctx, cli);
adam@1941: 
adam@1941:   if (uw_commit(ctx))
adam@1941:     uw_error(ctx, UNLIMITED_RETRY, "Rerunning expunge transaction");
adamc@1121: }
adamc@1121: 
adamc@1121: void uw_post_expunge(uw_context ctx, void *data) {
adamc@1121: }
adam@1320: 
adam@1320: int uw_supports_direct_status = 1;