src/modules/m_httpd.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
   6  *                     E-mail:
   7  *              <brain@chatspike.net>
   8  *                <Craig@chatspike.net>
   9  *
  10  * Written by Craig Edwards, Craig McLure, and others.
  11  * This program is free but copyrighted software; see
  12  *          the file COPYING for details.
  13  *
  14  * ---------------------------------------------------
  15  */
  16
  17 using namespace std;
  18
  19 #include <stdio.h>
  20 #include "users.h"
  21 #include "channels.h"
  22 #include "modules.h"
  23 #include "inspsocket.h"
  24 #include "helperfuncs.h"
  25 #include "httpd.h"
  26
  27 /* $ModDesc: Provides HTTP serving facilities to modules */
  28
  29 class ModuleHttp;
  30
  31 static Server *Srv;
  32 static ModuleHttp* HttpModule;
  33 extern time_t TIME;
  34 static bool claimed;
  35
  36 enum HttpState
  37 {
  38         HTTP_LISTEN = 0,
  39         HTTP_SERVE_WAIT_REQUEST = 1,
  40         HTTP_SERVE_SEND_DATA = 2
  41 };
  42
  43 class HttpSocket : public InspSocket
  44 {
  45         FileReader* index;
  46         HttpState InternalState;
  47         std::stringstream headers;
  48
  49  public:
  50
  51         HttpSocket(std::string host, int port, bool listening, unsigned long maxtime, FileReader* index_page) : InspSocket(host, port, listening, maxtime), index(index_page)
  52         {
  53                 log(DEBUG,"HttpSocket constructor");
  54                 InternalState = HTTP_LISTEN;
  55         }
  56
  57         HttpSocket(int newfd, char* ip, FileReader* ind) : InspSocket(newfd, ip), index(ind)
  58         {
  59                 InternalState = HTTP_SERVE_WAIT_REQUEST;
  60         }
  61
  62         virtual int OnIncomingConnection(int newsock, char* ip)
  63         {
  64                 if (InternalState == HTTP_LISTEN)
  65                 {
  66                         HttpSocket* s = new HttpSocket(newsock, ip, index);
  67                         Srv->AddSocket(s);
  68                 }
  69                 return true;
  70         }
  71
  72         virtual void OnClose()
  73         {
  74         }
  75
  76         std::string Response(int response)
  77         {
  78                 switch (response)
  79                 {
  80                         case 100:
  81                                 return "CONTINUE";
  82                         case 101:
  83                                 return "SWITCHING PROTOCOLS";
  84                         case 200:
  85                                 return "OK";
  86                         case 201:
  87                                 return "CREATED";
  88                         case 202:
  89                                 return "ACCEPTED";
  90                         case 203:
  91                                 return "NON-AUTHORITATIVE INFORMATION";
  92                         case 204:
  93                                 return "NO CONTENT";
  94                         case 205:
  95                                 return "RESET CONTENT";
  96                         case 206:
  97                                 return "PARTIAL CONTENT";
  98                         case 300:
  99                                 return "MULTIPLE CHOICES";
 100                         case 301:
 101                                 return "MOVED PERMENANTLY";
 102                         case 302:
 103                                 return "FOUND";
 104                         case 303:
 105                                 return "SEE OTHER";
 106                         case 304:
 107                                 return "NOT MODIFIED";
 108                         case 305:
 109                                 return "USE PROXY";
 110                         case 307:
 111                                 return "TEMPORARY REDIRECT";
 112                         case 400:
 113                                 return "BAD REQUEST";
 114                         case 401:
 115                                 return "UNAUTHORIZED";
 116                         case 402:
 117                                 return "PAYMENT REQUIRED";
 118                         case 403:
 119                                 return "FORBIDDEN";
 120                         case 404:
 121                                 return "NOT FOUND";
 122                         case 405:
 123                                 return "METHOD NOT ALLOWED";
 124                         case 406:
 125                                 return "NOT ACCEPTABLE";
 126                         case 407:
 127                                 return "PROXY AUTHENTICATION REQUIRED";
 128                         case 408:
 129                                 return "REQUEST TIMEOUT";
 130                         case 409:
 131                                 return "CONFLICT";
 132                         case 410:
 133                                 return "GONE";
 134                         case 411:
 135                                 return "LENGTH REQUIRED";
 136                         case 412:
 137                                 return "PRECONDITION FAILED";
 138                         case 413:
 139                                 return "REQUEST ENTITY TOO LARGE";
 140                         case 414:
 141                                 return "REQUEST-URI TOO LONG";
 142                         case 415:
 143                                 return "UNSUPPORTED MEDIA TYPE";
 144                         case 416:
 145                                 return "REQUESTED RANGE NOT SATISFIABLE";
 146                         case 417:
 147                                 return "EXPECTATION FAILED";
 148                         case 500:
 149                                 return "INTERNAL SERVER ERROR";
 150                         case 501:
 151                                 return "NOT IMPLEMENTED";
 152                         case 502:
 153                                 return "BAD GATEWAY";
 154                         case 503:
 155                                 return "SERVICE UNAVAILABLE";
 156                         case 504:
 157                                 return "GATEWAY TIMEOUT";
 158                         case 505:
 159                                 return "HTTP VERSION NOT SUPPORTED";
 160                         default:
 161                                 return "WTF";
 162                         break;
 163
 164                 }
 165         }
 166
 167         void SendHeaders(unsigned long size, int response)
 168         {
 169                 struct tm *timeinfo = localtime(&TIME);
 170                 this->Write("HTTP/1.1 "+ConvToStr(response)+" "+Response(response)+"\r\nDate: ");
 171                 this->Write(asctime(timeinfo));
 172                 this->Write("Server: InspIRCd/m_httpd.so/1.1\r\nContent-Length: "+ConvToStr(size)+
 173                                 "\r\nConnection: close\r\nContent-Type: text/html\r\n\r\n");
 174         }
 175
 176         virtual bool OnDataReady()
 177         {
 178                 char* data = this->Read();
 179                 std::string request_type;
 180                 std::string uri;
 181                 std::string http_version;
 182
 183                 /* Check that the data read is a valid pointer and it has some content */
 184                 if (data && *data)
 185                 {
 186                         headers << data;
 187
 188                         if (headers.str().find("\r\n\r\n") != std::string::npos)
 189                         {
 190                                 /* Headers are complete */
 191                                 InternalState = HTTP_SERVE_SEND_DATA;
 192
 193                                 headers >> request_type;
 194                                 headers >> uri;
 195                                 headers >> http_version;
 196
 197                                 if ((http_version != "HTTP/1.1") && (http_version != "HTTP/1.0"))
 198                                 {
 199                                         SendHeaders(0, 505);
 200                                 }
 201                                 else
 202                                 {
 203                                         if ((request_type == "GET") && (uri == "/"))
 204                                         {
 205                                                 SendHeaders(index->ContentSize(),200);
 206                                                 this->Write(index->Contents());
 207                                         }
 208                                         else
 209                                         {
 210                                                 claimed = false;
 211                                                 HTTPRequest httpr(request_type,uri,&headers,this,this->GetIP());
 212                                                 Event e((char*)&httpr, (Module*)HttpModule, "httpd_url");
 213                                                 e.Send();
 214
 215                                                 if (!claimed)
 216                                                 {
 217                                                         SendHeaders(0, 404);
 218                                                         log(DEBUG,"Page not claimed, 404");
 219                                                 }
 220                                         }
 221                                 }
 222
 223                                 return false;
 224                         }
 225                         return true;
 226                 }
 227                 else
 228                 {
 229                         /* Bastard client closed the socket on us!
 230                          * Oh wait, theyre SUPPOSED to do that!
 231                          */
 232                         return false;
 233                 }
 234         }
 235
 236         void Page(std::stringstream* n, int response)
 237         {
 238                 log(DEBUG,"Sending page");
 239                 SendHeaders(n->str().length(),response);
 240                 this->Write(n->str());
 241         }
 242 };
 243
 244 class ModuleHttp : public Module
 245 {
 246         int port;
 247         std::string host;
 248         std::string bindip;
 249         std::string indexfile;
 250
 251         FileReader index;
 252
 253         HttpSocket* http;
 254
 255  public:
 256
 257         void ReadConfig()
 258         {
 259                 ConfigReader c;
 260                 this->host = c.ReadValue("http", "host", 0);
 261                 this->bindip = c.ReadValue("http", "ip", 0);
 262                 this->port = c.ReadInteger("http", "port", 0, true);
 263                 this->indexfile = c.ReadValue("http", "index", 0);
 264
 265                 index.LoadFile(this->indexfile);
 266         }
 267
 268         void CreateListener()
 269         {
 270                 http = new HttpSocket(this->bindip, this->port, true, 0, &index);
 271                 if ((http) && (http->GetState() == I_LISTENING))
 272                 {
 273                         Srv->AddSocket(http);
 274                 }
 275         }
 276
 277         ModuleHttp(Server* Me) : Module::Module(Me)
 278         {
 279                 Srv = Me;
 280                 ReadConfig();
 281                 CreateListener();
 282         }
 283
 284         void OnEvent(Event* event)
 285         {
 286         }
 287
 288         char* OnRequest(Request* request)
 289         {
 290                 log(DEBUG,"Got HTTPDocument object");
 291                 claimed = true;
 292                 HTTPDocument* doc = (HTTPDocument*)request->GetData();
 293                 HttpSocket* sock = (HttpSocket*)doc->sock;
 294                 sock->Page(doc->GetDocument(), doc->GetResponseCode());
 295                 return NULL;
 296         }
 297
 298         void Implements(char* List)
 299         {
 300                 List[I_OnEvent] = List[I_OnRequest] = 1;
 301         }
 302
 303         virtual ~ModuleHttp()
 304         {
 305                 Srv->DelSocket(http);
 306         }
 307
 308         virtual Version GetVersion()
 309         {
 310                 return Version(1,0,0,0,VF_STATIC|VF_VENDOR);
 311         }
 312 };
 313
 314
 315 class ModuleHttpFactory : public ModuleFactory
 316 {
 317  public:
 318         ModuleHttpFactory()
 319         {
 320         }
 321
 322         ~ModuleHttpFactory()
 323         {
 324         }
 325
 326         virtual Module * CreateModule(Server* Me)
 327         {
 328                 HttpModule = new ModuleHttp(Me);
 329                 return HttpModule;
 330         }
 331 };
 332
 333
 334 extern "C" void * init_module( void )
 335 {
 336         return new ModuleHttpFactory;
 337 }