webserver.c

   1 /* depend: */
   2 /* cflags: */
   3 /* linker: color.o debug.o http.o server.o */
   4
   5 #include <assert.h>
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <unistd.h>
   9
  10 #include "debug.h"
  11 #include "http.h"
  12 #include "server.h"
  13
  14 /* types */
  15
  16 /* constants */
  17
  18 #define BUFFER_SIZE 4096
  19
  20 /* macros */
  21
  22 /* gobal variables */
  23
  24 char *progname = NULL;
  25 int port = 8080;
  26
  27 /* help function */
  28
  29 int usage (int ret)
  30 {
  31     FILE *fid = ret ? stderr : stdout;
  32     fprintf (fid, "usage: %s\n", progname);
  33     fprintf (fid, " -h : help message\n");
  34     fprintf (fid, " -p : port number (%d)\n", port);
  35     fprintf (fid, " -v : verbose level (%d)\n", verbose);
  36
  37     return ret;
  38 }
  39
  40 /* main function */
  41
  42 int main (int argc, char *argv[])
  43 {
  44     int i = 0;
  45
  46     /* program name */
  47
  48     progname = argv[0];
  49     while (progname[i] != '\0') {
  50         if ((progname[i] == '/') || (progname[i] == '\\')) {
  51             progname += i + 1;
  52             i = 0;
  53         } else {
  54             i++;
  55         }
  56     }
  57
  58     /* argument processing */
  59
  60      while (argc-- > 1) {
  61         char *arg = *(++argv);
  62         if (arg[0] != '-') {
  63             VERBOSE (ERROR, PERROR ("%s: invalid option -- '%s'\n", progname, arg); usage (1));
  64             return 1;
  65         }
  66         char c = arg[1];
  67         switch (c) {
  68             case 'p':
  69             arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
  70             if (arg == NULL) {
  71                 VERBOSE (ERROR, PERROR ("%s: missing port number\n", progname); usage (1));
  72                 return 1;
  73             }
  74             port = atoi (arg);
  75             if (port <= 0) {
  76                 VERBOSE (ERROR, PERROR ("%s: incorrect port number (%s)\n", progname, arg); usage (1));
  77                 return 1;
  78             }
  79             break;
  80         case 'v':
  81             arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
  82             if (arg == NULL) {
  83                 VERBOSE (ERROR, PERROR ("%s: missing verbose level\n", progname); usage (1));
  84                 return 1;
  85             }
  86             verbose = atoi (arg);
  87             break;
  88         case 'h':
  89         default:
  90             return usage (c != 'h');
  91         }
  92     }
  93
  94     VERBOSE (DEBUG, PRINT ("Initializing socket\n"));
  95     init_network_context ();
  96     if (open_listening_socket (port) == 0) {
  97         VERBOSE (ERROR, PERROR ("Can't open listening socket\n"));
  98         return 1;
  99     }
 100     VERBOSE (INFO, PRINT ("Listening socket on port %d\n", port));
 101
 102     /* main loop */
 103     while (1) {
 104         if (accept_incoming_connection () == 0) {
 105             usleep (1e5);
 106             continue;
 107         }
 108
 109         VERBOSE (DEBUG, PRINT ("Server connected, waiting for data\n"));
 110
 111         unsigned char *data = {0};
 112
 113         int len = receive_data (&data);
 114         if (len == 0) {
 115             VERBOSE (WARNING, PRINT ("Connection closed by peer (rx)\n"));
 116         } else if (len < 0) {
 117             VERBOSE (WARNING, PRINT ("Connection in error (rx)\n"));
 118         } else {
 119             VERBOSE (DEBUG, PRINT ("Received %d bytes\n", len));
 120
 121             // processing
 122             VERBOSE (DEBUG, PRINT ("Processing %s\n", data));
 123             len = processing ((char *)data, len, (char **)&data);
 124
 125             int rc = send_data (data, len);
 126             if (rc == 0) {
 127                 VERBOSE (WARNING, PRINT ("Connection closed by peer (tx)\n"));
 128             } else if (rc < 0) {
 129                 VERBOSE (WARNING, PRINT ("Connection in error (tx)\n"));
 130             }
 131         }
 132
 133         VERBOSE (DEBUG, PRINT ("Closing connection\n"));
 134         if (data) {
 135             free (data);
 136         }
 137         close_connection ();
 138     }
 139
 140     VERBOSE (DEBUG, PRINT ("Closing socket\n"));
 141     terminate_network_context ();
 142
 143     return 2;
 144 }
 145
 146 // test: webserver.exe -h
 147 // test: webserver.exe -h | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
 148 // test: webserver.exe -_ 2> /dev/null | wc -l | xargs test 0 =
 149 // test: webserver.exe -_ 2>&1 | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
 150 // test: webserver.exe error 2>&1 | grep -q 'invalid option'
 151 // test: webserver.exe -v 2>&1 | grep -q 'missing verbose level'
 152 // test: webserver.exe -p 2>&1 | grep -q 'missing port number'
 153 // test: webserver.exe -p -1 2>&1 | grep -q 'incorrect port number'
 154 // test: webserver.exe > test.log & pid=$!; sleep 1; kill -QUIT $pid; grep -q 'Listening socket on port 8080' test.log
 155 // test: webserver.exe -p 8000 > test.log & pid=$!; sleep 1; kill -ABRT $pid; grep -q 'Listening socket on port 8000' test.log
 156 // test: webserver.exe & pid=$!; sleep 1; kill -TERM $pid; ps aux | grep -q [w]ebserver.exe && kill -9 $pid || rc=1; test x$rc = x1
 157
 158 /* vim: set ts=4 sw=4 et: */