adapt to linux

[webserver.git] / webserver.c

/* depend: */
/* cflags: */
/* linker: color.o debug.o */

#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#ifdef _WIN32 /* Windows */
#include <winsock2.h>
#include <ws2tcpip.h>
#else /* Posix */
#include <errno.h>
#include <unistd.h>
#include <fcntl.h>
#include <netinet/ip.h>
#include <netinet/tcp.h>
#include <sys/types.h>
#include <sys/socket.h>
#endif

#include "debug.h"

/* types */

#ifdef _WIN32 /* Windows */
typedef SOCKET socket_t;
#else /* Posix */
typedef int socket_t;
#endif

/* constants */

#define BUFFER_SIZE 4096

/* macros */

/* gobal variables */

char *progname = NULL;
int port = 8080;

/* 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, " -v : verbose level (%d)\n", verbose);

    return ret;
}

/* open listening socket */

socket_t open_listening_socket (int port)
{
#ifdef _WIN32 /* Windows */
    WSADATA WSAData;
    WSAStartup (MAKEWORD(2,0), &WSAData);
    assert (INVALID_SOCKET == (socket_t)-1);
#endif

    VERBOSE (DEBUG, fprintf (stdout, "Opening socket\n"));
#ifdef _WIN32 /* Windows */
    socket_t sock = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (sock == INVALID_SOCKET)
        return -1;
#else /* Posix */
    socket_t sock = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (sock < 0)
        return -1;
#endif

    struct sockaddr_in addr = {0};
#ifdef _WIN32 /* Windows */
    addr.sin_family = AF_INET;
#else /* Posix */
    addr.sin_family = PF_INET;
#endif
    addr.sin_port = htons (port);
    addr.sin_addr.s_addr = htonl (INADDR_ANY);

    VERBOSE (DEBUG, fprintf (stdout, "Binding socket\n"));
    int rc = bind (sock, (struct sockaddr *)&addr, sizeof (addr));
#ifdef _WIN32 /* Windows */
    if (rc == SOCKET_ERROR) {
        VERBOSE (ERROR, fprintf (stderr, "error: %d\n", WSAGetLastError ()));
        if (closesocket (sock) == SOCKET_ERROR) {
            VERBOSE (ERROR, fprintf (stderr, "error: %d\n", WSAGetLastError ()));
        }
#else /* Posix */
    if (rc < 0) {
        VERBOSE (ERROR, fprintf (stderr, "error: %d\n", errno));
        close (sock);
#endif
        return -1;
    }

    VERBOSE (DEBUG, fprintf (stdout, "Configuring socket\n"));
#ifdef _WIN32 /* Windows */
#else /* Posix */
    fcntl (sock, F_SETFL, O_NONBLOCK);
#endif
    int val = 1;
    rc = setsockopt (sock, IPPROTO_TCP, TCP_NODELAY, (char *)&val, sizeof (val));
    if (rc < 0) {
        VERBOSE (ERROR, fprintf (stderr, "%s\n", "setsockopt/TCP_NODELAY"));
#ifdef _WIN32 /* Windows */
        closesocket (sock);
#else /* Posix */
        close (sock);
#endif
        return -1;
    }

    return sock;
}

/* close listening socket */
void close_listening_socket (socket_t sock)
{
#ifdef _WIN32 /* Windows */
    closesocket (sock);
    WSACleanup ();
#else /* Posix */
    close (sock);
#endif
}

/* main function */

int main (int argc, char *argv[])
{
    int i = 0;
    int ret = 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, fprintf (stderr, "%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, fprintf (stderr, "%s: missing port number\n", progname); usage (1));
                return 1;
            }
            port = atoi (arg);
            if (port <= 0) {
                VERBOSE (ERROR, fprintf (stderr, "%s: incorrect port number (%s)\n", progname, arg); usage (1));
                return 1;
            }
            break;
        case 'v':
            arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
            if (arg == NULL) {
                VERBOSE (ERROR, fprintf (stderr, "%s: missing verbose level\n", progname); usage (1));
                return 1;
            }
            verbose = atoi (arg);
            break;
        case 'h':
        default:
            return usage (c != 'h');
        }
    }

    VERBOSE (DEBUG, fprintf (stdout, "Initializing socket\n"));
    socket_t sock = open_listening_socket (port);
    if (sock == (socket_t)-1) {
        VERBOSE (ERROR, fprintf (stderr, "Can't open listening socket\n"));
        return 1;
    }

    VERBOSE (INFO, fprintf (stdout, "Listening socket on port %d\n", port));
    sleep (2);

    VERBOSE (DEBUG, fprintf (stdout, "Closing socket\n"));
    close_listening_socket (sock);

    return ret;
}

// 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 | grep -q 'Listening socket on port 8080'
// test: webserver.exe -p 8000 | grep -q 'Listening socket on port 8000'

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