minimal mime management

[webserver.git] / webserver.c

/* depend: */
/* cflags: */
/* linker: color.o debug.o file.o http.o server.o */

#include <assert.h>
#include <dirent.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include "debug.h"
#include "http.h"
#include "server.h"

/* types */

/* constants */

#define BUFFER_SIZE 4096
#define ROOT_DIR "webroot"
#define SERVER_NAME "localhost"
#define CHARSET "iso-8859-1"

/* macros */

/* gobal variables */

char *progname = NULL;
int port = 8080;
char *root = ROOT_DIR;
char *servername = SERVER_NAME;
char *charset = CHARSET;

/* help function */

int usage (int ret)
{
    FILE *fid = ret ? stderr : stdout;
    fprintf (fid, "usage: %s\n", progname);
    fprintf (fid, " -h : help message\n");
    fprintf (fid, " -p : port number (%d)\n", port);
    fprintf (fid, " -r : web root directory (%s)\n", root);
    fprintf (fid, " -v : verbose level (%d)\n", verbose);

    return ret;
}

/* main function */

int main (int argc, char *argv[])
{
    int i = 0;

    /* program name */

    progname = argv[0];
    while (progname[i] != '\0') {
        if ((progname[i] == '/') || (progname[i] == '\\')) {
            progname += i + 1;
            i = 0;
        } else {
            i++;
        }
    }

    /* argument processing */

    while (argc-- > 1) {
        char *arg = *(++argv);
        if (arg[0] != '-') {
            VERBOSE (ERROR, PERROR ("%s: invalid option -- '%s'\n", progname, arg); usage (1));
            return 1;
        }
        char c = arg[1];
        switch (c) {
            case 'p':
            arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
            if (arg == NULL) {
                VERBOSE (ERROR, PERROR ("%s: missing port number\n", progname); usage (1));
                return 1;
            }
            port = atoi (arg);
            if (port <= 0) {
                VERBOSE (ERROR, PERROR ("%s: incorrect port number (%s)\n", progname, arg); usage (1));
                return 1;
            }
            break;
        case 'r':
            arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
            if (arg == NULL) {
                VERBOSE (ERROR, PERROR ("%s: missing directory name\n", progname); usage (1));
                return 1;
            }
            root = arg;
            break;
        case 'v':
            arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
            if (arg == NULL) {
                VERBOSE (ERROR, PERROR ("%s: missing verbose level\n", progname); usage (1));
                return 1;
            }
            verbose = atoi (arg);
            break;
        case 'h':
        default:
            return usage (c != 'h');
        }
    }

    /* check root directory */
    VERBOSE (DEBUG, PRINT ("Check web root\n"));
    DIR *pdir = opendir (root);
    if (pdir == NULL) {
        VERBOSE (ERROR, PERROR ("Can't read directory (%s)\n", root));
        return 1;
    }
    closedir (pdir);

    /* configuration */
    conf_t conf = {root, servername, charset};

    /* init network stack */
    VERBOSE (DEBUG, PRINT ("Initializing socket\n"));
    init_network_context ();
    if (open_listening_socket (port) == 0) {
        VERBOSE (ERROR, PERROR ("Can't open listening socket\n"));
        return 1;
    }
    VERBOSE (INFO, PRINT ("Listening socket on port %d\n", port));

    /* main loop */
    while (1) {
        if (accept_incoming_connection () == 0) {
            usleep (1e5);
            continue;
        }

        VERBOSE (DEBUG, PRINT ("Server connected, waiting for data\n"));

        char *data = NULL;
        char *output = NULL;

        int len = receive_data (&data);
        if (len == 0) {
            VERBOSE (WARNING, PRINT ("Connection closed by peer (rx)\n"));
        } else if (len < 0) {
            VERBOSE (WARNING, PRINT ("Connection in error (rx)\n"));
        } else {
            VERBOSE (DEBUG, PRINT ("Received %d bytes\n", len));

            // processing
            VERBOSE (DEBUG, PRINT ("Processing %s\n", data));
            len = processing (data, len, &conf, &output);

            VERBOSE (DEBUG, PRINT ("Sending data (%d)\n%s\n", len, data));
            int rc = send_data (output, len);
            if (rc == 0) {
                VERBOSE (WARNING, PRINT ("Connection closed by peer (tx)\n"));
            } else if (rc < 0) {
                VERBOSE (WARNING, PRINT ("Connection in error (tx)\n"));
            }
        }

        VERBOSE (DEBUG, PRINT ("Closing connection\n"));
        if (data) {
            free (data);
        }
        if (output) {
            free (output);
        }
        close_connection ();
    }
    
    VERBOSE (DEBUG, PRINT ("Closing socket\n"));
    terminate_network_context ();

    return 2;
}

// test: webserver.exe -h
// test: webserver.exe -h | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
// test: webserver.exe -_ 2> /dev/null | wc -l | xargs test 0 =
// test: webserver.exe -_ 2>&1 | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
// test: webserver.exe error 2>&1 | grep -q 'invalid option'
// test: webserver.exe -v 2>&1 | grep -q 'missing verbose level'
// test: webserver.exe -p 2>&1 | grep -q 'missing port number'
// test: webserver.exe -p -1 2>&1 | grep -q 'incorrect port number'
// test: webserver.exe > test.log & pid=$!; sleep 1; kill -QUIT $pid; grep -q 'Listening socket on port 8080' test.log
// test: webserver.exe -p 8000 > test.log & pid=$!; sleep 1; kill -ABRT $pid; grep -q 'Listening socket on port 8000' test.log
// 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

/* vim: set ts=4 sw=4 et: */