webserver.c

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