Diff: HTTPServer.cpp

Revision:

0:c385e589a779

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/HTTPServer.cpp	Tue Apr 08 12:13:32 2014 +0000
@@ -0,0 +1,122 @@
+#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);
+    if(socket.bind(port))
+    {
+        printf("Could not bind on port %d.\n", port);
+        return false; 
+    }
+    
+    if(socket.listen())
+    {
+        printf("Could not listen %d\n", port);
+        return false;
+    }
+    
+    return true;
+}
+
+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");
+        break;//***new
+        }
+    break;//***new
+    }
+}
+
+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, "No request handler found for this type of request.");
+        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;
+}
+