antkeeper
/
superbuild

/*
 * This is a quick and dirty HTTP server that uses PhysicsFS to retrieve *  files. It is not robust at all, probably buggy, and definitely poorly *  designed. It's just meant to show that it can be done. * * Basically, you compile this code, and run it: *   ./physfshttpd archive1.zip archive2.zip /path/to/a/real/dir etc... * * The files are appended in order to the PhysicsFS search path, and when *  a client request comes in, it looks for the file in said search path. * * My goal was to make this work in less than 300 lines of C, so again, it's *  not to be used for any serious purpose. Patches to make this application *  suck less will be readily and gratefully accepted. * * Command line I used to build this on Linux: *  gcc -Wall -Werror -g -o bin/physfshttpd extras/physfshttpd.c -lphysfs * * License: this code is public domain. I make no warranty that it is useful, *  correct, harmless, or environmentally safe. * * This particular file may be used however you like, including copying it *  verbatim into a closed-source project, exploiting it commercially, and *  removing any trace of my name from the source (although I hope you won't *  do that). I welcome enhancements and corrections to this file, but I do *  not require you to send me patches if you make changes. This code has *  NO WARRANTY. * * Unless otherwise stated, the rest of PhysicsFS falls under the zlib license. *  Please see LICENSE.txt in the root of the source tree. * *  This file was written by Ryan C. Gordon. (icculus@icculus.org). */
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <ctype.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#ifndef LACKING_SIGNALS
#include <signal.h>
#endif

#ifndef LACKING_PROTOENT
#include <netdb.h>
#endif

#include "physfs.h"


#define DEFAULT_PORTNUM 8080

typedef struct{    int sock;    struct sockaddr *addr;    socklen_t addrlen;} http_args;

#define txt404 \
    "HTTP/1.0 404 Not Found\n" \    "Connection: close\n" \    "Content-Type: text/html; charset=utf-8\n" \    "\n" \    "<html><head><title>404 Not Found</title></head>\n" \    "<body>Can't find '%s'.</body></html>\n\n" \
#define txt200 \
    "HTTP/1.0 200 OK\n" \    "Connection: close\n" \    "Content-Type: %s\n" \    "\n"
static const char *lastError(void){    return PHYSFS_getErrorByCode(PHYSFS_getLastErrorCode());} /* lastError */
static int writeAll(const char *ipstr, const int sock, void *buf, const size_t len){    if (write(sock, buf, len) != len)    {        printf("%s: Write error to socket.\n", ipstr);        return 0;    } /* if */
    return 1;} /* writeAll */
static int writeString(const char *ipstr, const int sock, const char *fmt, ...){    /* none of this is robust against large strings or HTML escaping. */    char buffer[1024];    int len;    va_list ap;    va_start(ap, fmt);    len = vsnprintf(buffer, sizeof (buffer), fmt, ap);    va_end(ap);    if (len < 0)    {        printf("uhoh, vsnprintf() failed!\n");        return 0;    } /* if */
    return writeAll(ipstr, sock, buffer, len);} /* writeString */

static void feed_file_http(const char *ipstr, int sock, const char *fname){    PHYSFS_File *in = PHYSFS_openRead(fname);
    if (in == NULL)    {        printf("%s: Can't open [%s]: %s.\n", ipstr, fname, lastError());        writeString(ipstr, sock, txt404, fname);        return;    } /* if */
    /* !!! FIXME: mimetype */    if (writeString(ipstr, sock, txt200, "text/plain; charset=utf-8"))    {        do        {            char buffer[1024];            PHYSFS_sint64 br = PHYSFS_readBytes(in, buffer, sizeof (buffer));            if (br == -1)            {                printf("%s: Read error: %s.\n", ipstr, lastError());                break;            } /* if */
            else if (!writeAll(ipstr, sock, buffer, (size_t) br))            {                break;            } /* else if */        } while (!PHYSFS_eof(in));    } /* if */
    PHYSFS_close(in);} /* feed_file_http */

static void feed_dirlist_http(const char *ipstr, int sock,                              const char *dname, char **list){    int i;
    if (!writeString(ipstr, sock, txt200, "text/html; charset=utf-8"))        return;
    else if (!writeString(ipstr, sock,                    "<html><head><title>Directory %s</title></head>"                    "<body><p><h1>Directory %s</h1></p><p><ul>\n",                    dname, dname))        return;
    if (strcmp(dname, "/") == 0)        dname = "";
    for (i = 0; list[i]; i++)    {        const char *fname = list[i];        if (!writeString(ipstr, sock,            "<li><a href='%s/%s'>%s</a></li>\n", dname, fname, fname))            break;    } /* for */
    writeString(ipstr, sock, "</ul></body></html>\n");} /* feed_dirlist_http */
static void feed_dir_http(const char *ipstr, int sock, const char *dname){    char **list = PHYSFS_enumerateFiles(dname);    if (list == NULL)    {        printf("%s: Can't enumerate directory [%s]: %s.\n",               ipstr, dname, lastError());        writeString(ipstr, sock, txt404, dname);        return;    } /* if */
    feed_dirlist_http(ipstr, sock, dname, list);    PHYSFS_freeList(list);} /* feed_dir_http */
static void feed_http_request(const char *ipstr, int sock, const char *fname){    PHYSFS_Stat statbuf;
    printf("%s: requested [%s].\n", ipstr, fname);
    if (!PHYSFS_stat(fname, &statbuf))    {        printf("%s: Can't stat [%s]: %s.\n", ipstr, fname, lastError());        writeString(ipstr, sock, txt404, fname);        return;    } /* if */
    if (statbuf.filetype == PHYSFS_FILETYPE_DIRECTORY)        feed_dir_http(ipstr, sock, fname);    else        feed_file_http(ipstr, sock, fname);} /* feed_http_request */

static void *do_http(void *_args){    http_args *args = (http_args *) _args;    char ipstr[128];    char buffer[512];    char *ptr;    strncpy(ipstr, inet_ntoa(((struct sockaddr_in *) args->addr)->sin_addr),            sizeof (ipstr));    ipstr[sizeof (ipstr) - 1] = '\0';
    printf("%s: connected.\n", ipstr);    read(args->sock, buffer, sizeof (buffer));    buffer[sizeof (buffer) - 1] = '\0';    ptr = strchr(buffer, '\n');    if (!ptr)        printf("%s: potentially bogus request.\n", ipstr);    else    {        *ptr = '\0';        ptr = strchr(buffer, '\r');        if (ptr != NULL)            *ptr = '\0';
        if ((toupper(buffer[0]) == 'G') &&            (toupper(buffer[1]) == 'E') &&            (toupper(buffer[2]) == 'T') &&            (toupper(buffer[3]) == ' ') &&            (toupper(buffer[4]) == '/'))        {            ptr = strchr(buffer + 5, ' ');            if (ptr != NULL)                *ptr = '\0';            feed_http_request(ipstr, args->sock, buffer + 4);        } /* if */    } /* else */
    /* !!! FIXME: Time the transfer. */    printf("%s: closing connection.\n", ipstr);    close(args->sock);    free(args->addr);    free(args);    return NULL;} /* do_http */

static void serve_http_request(int sock, struct sockaddr *addr,                               socklen_t addrlen){    http_args *args = (http_args *) malloc(sizeof (http_args));    if (args == NULL)    {        printf("out of memory.\n");        return;    } /* if */    args->addr = (struct sockaddr *) malloc(addrlen);    if (args->addr == NULL)    {        free(args);        printf("out of memory.\n");        return;    } /* if */
    args->sock = sock;    args->addrlen = addrlen;    memcpy(args->addr, addr, addrlen);
    /* !!! FIXME: optionally spin a thread... */    do_http((void *) args);} /* server_http_request */

static int create_listen_socket(short portnum){    int retval = -1;    int protocol = 0;  /* pray this is right. */
#ifndef LACKING_PROTOENT
    struct protoent *prot;    setprotoent(0);    prot = getprotobyname("tcp");    if (prot != NULL)        protocol = prot->p_proto;#endif

    retval = socket(PF_INET, SOCK_STREAM, protocol);    if (retval >= 0)    {        struct sockaddr_in addr;        addr.sin_family = AF_INET;        addr.sin_port = htons(portnum);        addr.sin_addr.s_addr = INADDR_ANY;        if ((bind(retval, (struct sockaddr *) &addr, (socklen_t) sizeof (addr)) == -1) ||            (listen(retval, 5) == -1))        {            close(retval);            retval = -1;        } /* if */    } /* if */
    return retval;} /* create_listen_socket */

static int listensocket = -1;
void at_exit_cleanup(void){    /*
     * !!! FIXME: If thread support, signal threads to terminate and     * !!! FIXME:  wait for them to clean up.     */
    if (listensocket >= 0)        close(listensocket);
    if (!PHYSFS_deinit())        printf("PHYSFS_deinit() failed: %s\n", lastError());} /* at_exit_cleanup */

int main(int argc, char **argv){    int i;    int portnum = DEFAULT_PORTNUM;
    setbuf(stdout, NULL);    setbuf(stderr, NULL);
#ifndef LACKING_SIGNALS
    /* I'm not sure if this qualifies as a cheap trick... */    signal(SIGTERM, exit);    signal(SIGINT, exit);    signal(SIGFPE, exit);    signal(SIGSEGV, exit);    signal(SIGPIPE, exit);    signal(SIGILL, exit);#endif

    if (argc == 1)    {        printf("USAGE: %s <archive1> [archive2 [... archiveN]]\n", argv[0]);        return 42;    } /* if */
    if (!PHYSFS_init(argv[0]))    {        printf("PHYSFS_init() failed: %s\n", lastError());        return 42;    } /* if */
    /* normally, this is bad practice, but oh well. */    atexit(at_exit_cleanup);
    for (i = 1; i < argc; i++)    {        if (!PHYSFS_mount(argv[i], NULL, 1))            printf(" WARNING: failed to add [%s] to search path.\n", argv[i]);    } /* else */
    listensocket = create_listen_socket(portnum);    if (listensocket < 0)    {        printf("listen socket failed to create.\n");        return 42;    } /* if */
    while (1)  /* infinite loop for now. */    {        struct sockaddr addr;        socklen_t len;        int s = accept(listensocket, &addr, &len);        if (s < 0)        {            printf("accept() failed: %s\n", strerror(errno));            close(listensocket);            return 42;        } /* if */
        serve_http_request(s, &addr, len);    } /* while */
    return 0;} /* main */
/* end of physfshttpd.c ... */