webserver.c

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