File content as of revision 7:838d7ea07e18:

#include "HTTPServer.h"

#define INVALID_FORMATTER "No valid formatter specified"

bool cmp(char* a, char* b)
{
    return strcmp(a,b) < 0;
}

HTTPServer::HTTPServer(Formatter *f):
socket(),
handlers(&cmp),
formatter(f),
reply(),
command()
{
}

HTTPServer::~HTTPServer()
{
    for(std::map<char*, RequestHandler*>::iterator itor = handlers.begin();
        itor != handlers.end();
        ++itor)
        delete itor->second;
    
    if(formatter)
        delete formatter;
}

bool HTTPServer::init(int port)
{
    socket.set_blocking(true);
    return !socket.bind(port) && !socket.listen();
}

void HTTPServer::run()
{
    TCPSocketConnection c;
    while(true)
    {
        while(socket.accept(c));
        c.set_blocking(false, 1000);
        while(c.is_connected())
        {
            char buffer[512];
            int n = c.receive_all(buffer, sizeof(buffer)-1);
            if(n == 0)
            {
                c.close();
                break;
            }
            else if(n != -1)
            {
                printf("Received data\n");
                buffer[n] = '\0';
                handle_request(buffer);
                if(formatter != NULL)
                {
                    printf("Sending data...");
                    char *page = formatter->get_page(reply);
                    do
                    {
                        c.send(page, strlen(page)+1);
                        page = formatter->get_page(reply);
                    }while(strlen(page)>0);
                    printf("done\n");
                }
                else
                    c.send(INVALID_FORMATTER, strlen(INVALID_FORMATTER)+1);
            }
            else
                printf("Error while receiving data\n");
        }
    }
}

void HTTPServer::handle_request(char *buffer)
{
    char *request_type = strtok(buffer, " ");
    char *request = strtok(NULL, " ");

    reply[0] = '\0';
    if(!strcmp(request, "/"))
        return;    
    
    if(!command.decode(request))
    {
        strcat(reply, "Malformed request");
        return;
    }
    
    std::map<char*, RequestHandler*>::iterator itor = handlers.find(request_type);
    if(itor == handlers.end())
    {
        strcat(reply, "Only PUT, DELETE and GET requests are accepted.");
        return;
    }
    if(itor->second != NULL)
        itor->second->handle(command, reply);
    else
        strcat(reply, "Invalid request handler");
}

void HTTPServer::add_request_handler(char *name, RequestHandler* handler)
{
    handlers[name] = handler;
}