buggy code

[webserver.git] / server.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"

#include "server.h"

/* types */

#ifdef _WIN32 /* Windows */
typedef SOCKET socket_t;
#define PF_INET AF_INET
#define ERRNO (WSAGetLastError ())
#else /* Posix */
typedef int socket_t;
#define closesocket close
#define ERRNO errno
#define SOCKET_ERROR -1
#endif

/* constants */

#define BACKLOG 5
#define BUFFER_SIZE 4096
#define TIMEOUT 100000

/* macros */

/* gobal variables */

/* 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);
    assert (SOCKET_ERROR == -1);
#endif

    VERBOSE (DEBUG, fprintf (stdout, "Opening socket\n"));
    socket_t sock = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (sock == INVALID_SOCKET) {
        return -1;
    }

    struct sockaddr_in addr = {0};
    addr.sin_family = PF_INET;
    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));
    if (rc == SOCKET_ERROR) {
        VERBOSE (ERROR, fprintf (stderr, "error: bind %d\n", ERRNO));
        rc = closesocket (sock);
        if (rc == SOCKET_ERROR) {
            VERBOSE (ERROR, fprintf (stderr, "error: close %d\n", ERRNO));
        }
        return -1;
    }

    VERBOSE (DEBUG, fprintf (stdout, "Configuring socket\n"));
#ifndef _WIN32 /* 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, "error: %s\n", "setsockopt/TCP_NODELAY"));
        closesocket (sock);
        if (rc == SOCKET_ERROR) {
            VERBOSE (ERROR, fprintf (stderr, "error: close %d\n", ERRNO));
        }
        return -1;
    }

    rc = listen (sock, BACKLOG);
    if (rc < 0) {
        VERBOSE (ERROR, fprintf (stderr, "error: %s\n", "listen"));
        closesocket (sock);
        if (rc == SOCKET_ERROR) {
            VERBOSE (ERROR, fprintf (stderr, "error: close %d\n", ERRNO));
        }
        return -1;
    }

    return sock;
}

/* close listening socket */
void close_listening_socket (socket_t sock)
{
    int rc = closesocket (sock);
    if (rc == SOCKET_ERROR) {
        VERBOSE (ERROR, fprintf (stderr, "error: %d\n", ERRNO));
    }
#ifdef _WIN32 /* Windows */
    WSACleanup ();
#endif
}

/* accept incomming connection */

int accept_incoming_connection (socket_t sock)
{
    int connection = accept (sock, NULL, 0);
    if (connection < 0) {
        return connection;
    }

#ifndef _WIN32 /* POSIX */
    fcntl (connection, F_SETFL, O_NONBLOCK);
#endif
    int val = 1;
    rc = setsockopt (connection, IPPROTO_TCP, TCP_NODELAY, (char *)&val, sizeof (val));
    if (rc < 0) {
        VERBOSE (ERROR, fprintf (stderr, "error: %s\n", "setsockopt/TCP_NODELAY"));
        closesocket (connection);
        if (rc == SOCKET_ERROR) {
            VERBOSE (ERROR, fprintf (stderr, "error: close %d\n", ERRNO));
        }
        return -1;
    }

    return connection;
}

/* receive data */
int receive_data (socket_t sock, unsigned char data, int maxlen)
{

    /* timeout management */
    fd_set rfds;
    struct timeval tv = { 0, TIMEOUT };
    FD_ZERO (&rfds);
    FD_SET (sock, &rfds);

    int retval = select (sock + 1, &rfds, NULL, NULL, &tv);
    if (retval != 1)
        return retval; /* 0 or SOCKET_ERROR */

    /* read from socket */
    int len = read (sock, data, maxlen);

    if (rc = 0) { /* sock closed */
        return 0;
    } else if ((rc < 0) && (ERRNO != EAGAIN)) { /* error */
        return -1;
    } else if (rc > 0) {
        return len;
    }

    /* do we retry ? */

    return -2;
}


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