webserver.c

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