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