src/http/httpthread.cpp

   1 #include "../../include/http/httpthread.h"\r
   2 #include "../../include/option.h"\r
   3 #include "../../include/stringfunctions.h"\r
   4 #include "../../include/http/pages/homepage.h"\r
   5 #include "../../include/http/pages/optionspage.h"\r
   6 #include "../../include/http/pages/showcaptchapage.h"\r
   7 #include "../../include/http/pages/createidentitypage.h"\r
   8 #include "../../include/http/pages/localidentitiespage.h"\r
   9 #include "../../include/http/pages/announceidentitypage.h"\r
  10 #include "../../include/http/pages/addpeerpage.h"\r
  11 #include "../../include/http/pages/peertrustpage.h"\r
  12 #include "../../include/http/pages/controlboardpage.h"\r
  13 #include "../../include/http/pages/peerdetailspage.h"\r
  14 \r
  15 #include <iostream>\r
  16 \r
  17 #ifdef XMEM\r
  18         #include <xmem.h>\r
  19 #endif\r
  20 \r
  21 HTTPThread::HTTPThread()\r
  22 {\r
  23         std::string templatestr;\r
  24         int port;\r
  25         std::string portstr;\r
  26         Option::Instance()->Get("HTTPListenPort",portstr);\r
  27         StringFunctions::Convert(portstr,port);\r
  28 \r
  29         // set template\r
  30         templatestr="<html><head></head><body><a href=\"home.htm\">Home</a><br>[CONTENT]</body></html>";\r
  31         FILE *infile=fopen("template.htm","r+b");\r
  32         if(infile)\r
  33         {\r
  34                 fseek(infile,0,SEEK_END);\r
  35                 long len=ftell(infile);\r
  36                 std::vector<char> data(len,0);\r
  37                 fseek(infile,0,SEEK_SET);\r
  38                 fread(&data[0],1,len,infile);\r
  39                 fclose(infile);\r
  40                 templatestr.assign(data.begin(),data.end());\r
  41         }\r
  42         else\r
  43         {\r
  44                 m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"HTTPThread::HTTPThread could not open template.htm");\r
  45         }\r
  46 \r
  47         // push back page handlers\r
  48         m_pagehandlers.push_back(new ShowCaptchaPage());\r
  49         m_pagehandlers.push_back(new OptionsPage(templatestr));\r
  50         m_pagehandlers.push_back(new LocalIdentitiesPage(templatestr));\r
  51         m_pagehandlers.push_back(new CreateIdentityPage(templatestr));\r
  52         m_pagehandlers.push_back(new AnnounceIdentityPage(templatestr));\r
  53         m_pagehandlers.push_back(new AddPeerPage(templatestr));\r
  54         m_pagehandlers.push_back(new PeerTrustPage(templatestr));\r
  55         m_pagehandlers.push_back(new ControlBoardPage(templatestr));\r
  56         m_pagehandlers.push_back(new PeerDetailsPage(templatestr));\r
  57         // homepage must be last - catch all page handler\r
  58         m_pagehandlers.push_back(new HomePage(templatestr));\r
  59 \r
  60         m_ctx=0;\r
  61         m_ctx=shttpd_init(NULL,"listen_ports",portstr.c_str(),NULL);\r
  62         shttpd_listen(m_ctx,port,false);\r
  63 \r
  64         shttpd_register_uri(m_ctx,"*",HTTPThread::PageCallback,this);\r
  65 \r
  66 }\r
  67 \r
  68 HTTPThread::~HTTPThread()\r
  69 {\r
  70         shttpd_fini(m_ctx);\r
  71 \r
  72         for(std::vector<IPageHandler *>::iterator i=m_pagehandlers.begin(); i!=m_pagehandlers.end(); i++)\r
  73         {\r
  74                 delete (*i);\r
  75         }\r
  76 \r
  77 }\r
  78 \r
  79 void HTTPThread::PageCallback(shttpd_arg *arg)\r
  80 {\r
  81 \r
  82         HTTPThread *thread=(HTTPThread *)arg->user_data;\r
  83 \r
  84         for(std::vector<IPageHandler *>::iterator i=thread->m_pagehandlers.begin(); i!=thread->m_pagehandlers.end(); )\r
  85         {\r
  86                 if((*i)->Handle(arg)==true)\r
  87                 {\r
  88                         i=thread->m_pagehandlers.end();\r
  89                 }\r
  90                 else\r
  91                 {\r
  92                         i++;\r
  93                 }\r
  94         }\r
  95 \r
  96 }\r
  97 \r
  98 void HTTPThread::Run()\r
  99 {\r
 100         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"HTTPThread::run thread started.");\r
 101 \r
 102         do\r
 103         {\r
 104                 shttpd_poll(m_ctx,1000);\r
 105         }while(!IsCancelled());\r
 106 \r
 107         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"HTTPThread::run thread exiting.");\r
 108 \r
 109 }\r