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