HTTPServer.cpp

00001 #include <string.h>
00002 #include "HTTPServer.h"
00003 #include "MbedJSONValue.h"
00004 
00005 extern MbedJSONValue WIZwikiREST;
00006 
00007 bool cmp(char* a, char* b)
00008 {
00009     return strcmp(a,b) < 0;
00010 }
00011 
00012 HTTPServer::HTTPServer():
00013 socket(),
00014 handlers(&cmp)
00015 {
00016 }
00017 
00018 HTTPServer::~HTTPServer()
00019 {
00020     for(std::map<char*, RequestHandler*>::iterator itor = handlers.begin();
00021         itor != handlers.end();
00022         ++itor)
00023         delete itor->second;
00024 }
00025 
00026 bool HTTPServer::init(int port)
00027 {
00028     socket.set_blocking(true);
00029     if(socket.bind(port))
00030     {
00031         printf("Could not bind on port %d.\n", port);
00032         return false; 
00033     }
00034     
00035     if(socket.listen())
00036     {
00037         printf("Could not listen %d\n", port);
00038         return false;
00039     }
00040     
00041     return true;
00042 }
00043 
00044 void HTTPServer::run()
00045 {
00046     TCPSocketConnection c;
00047     while(true)
00048     {
00049         while(socket.accept(c));
00050         c.set_blocking(false, 1000);
00051         while(c.is_connected())
00052         {
00053 
00054             int n = c.receive_all(HTTPBUF, sizeof(HTTPBUF)-1);
00055                                 
00056                         if(n == 0)
00057             {
00058                 c.close();
00059                 break;
00060             }
00061             else if(n != -1)
00062             {
00063 #ifdef DEBUG_HTTP                                                
00064                                 printf("Received data : %d\r\n",n);
00065 #endif                                
00066                             
00067                                     HTTPBUF[n] = '\0';
00068                                     if(handle_request(HTTPBUF) == HTTP_SUCCESS)
00069                                {
00070                                        c.send(rest_result, strlen(rest_result));
00071                                        //c.send((char*)rest_result.c_str(), 159);
00072                                  }
00073                                  else
00074                                  {
00075                                        //printf("send fail : %s\r\n",(char*)rest_result.c_str());
00076                                        c.send(rest_result, strlen(rest_result));
00077                                  }
00078             }
00079             else
00080                 printf("Error while receiving data\n");
00081                 
00082         }
00083     }
00084 }
00085  
00086 
00087 HTTP_RESULT HTTPServer::handle_request(char *buffer)
00088 {
00089     char *request_type;
00090     char *request;
00091     
00092 //{"Name":"WIZwiki-REST-01","Network":{"IP":"192.168.100.100","SN":"255.255.255.0","GW":"192.168.100.1"},"User":{"Name":"Lawrence","ID":"law","PSWD":"law1234"}}   159
00093     if(buffer)
00094     {
00095 #ifdef DEBUG_HTTP                
00096                 // buffer check
00097                 printf("*******************************************************\r\n");
00098         printf("buffer=%s\r\n",buffer);
00099                 printf("*******************************************************\r\n");
00100 #endif                                
00101               // type parsing
00102               request_type = strtok(buffer," \r\n");
00103 #ifdef DEBUG_HTTP                              
00104         printf("Type = %s\r\n", request_type);
00105 #endif        
00106             
00107                 if(request_type)
00108         {
00109                       request = strtok(NULL, "  \r\n");                                                 // corrested  " " -> " /"     : /Name -> Name
00110             if(request)
00111             {
00112 #ifdef DEBUG_HTTP                                    
00113                     printf("URI = %s\r\n", request);
00114 #endif                    
00115                         }
00116                         else
00117                         {
00118               strcpy(rest_result, "Invaild URI");
00119 #ifdef DEBUG_HTTP                              
00120                             printf("%s\r\n",rest_result);
00121 #endif                            
00122               return HTTP_INVALID_URI;
00123             }
00124         }
00125     }
00126     
00127     std::map<char*, RequestHandler*>::iterator itor = handlers.find(request_type);
00128     if(itor == handlers.end())
00129     {
00130         strcpy(rest_result, "No request handler found for this type of request.");
00131         return HTTP_INVALID_HANDLE;
00132     }
00133     //if(itor != NULL)
00134         //itor->handle(request, rest_result.c_str());
00135     if(itor->second != NULL)
00136         itor->second->handle(request, rest_result);
00137     else
00138     {
00139         strcpy(rest_result, "Invalid request handler");
00140         return HTTP_INVALID_HANDLE;
00141     }
00142     return HTTP_SUCCESS;
00143 }
00144 
00145 void HTTPServer::add_request_handler(char *name, RequestHandler* handler)
00146 {
00147     handlers[name] = handler;
00148 }
00149