* | Inspire Internet Relay Chat Daemon |
* +------------------------------------+
*
- * InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- * E-mail:
- * <brain@chatspike.net>
- * <Craig@chatspike.net>
- *
- * Written by Craig Edwards, Craig McLure, and others.
+ * InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
+ *
* This program is free but copyrighted software; see
- * the file COPYING for details.
+ * the file COPYING for details.
*
* ---------------------------------------------------
*/
-using namespace std;
-
-#include <stdio.h>
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-#include "inspsocket.h"
-#include "helperfuncs.h"
+#include "inspircd.h"
#include "httpd.h"
/* $ModDesc: Provides HTTP serving facilities to modules */
+/* $ModDep: httpd.h */
-class ModuleHttp;
+class ModuleHttpServer;
-static Server *Srv;
-static ModuleHttp* HttpModule;
-extern time_t TIME;
+static ModuleHttpServer* HttpModule;
static bool claimed;
+/** HTTP socket states
+ */
enum HttpState
{
- HTTP_LISTEN = 0,
- HTTP_SERVE_WAIT_REQUEST = 1,
- HTTP_SERVE_SEND_DATA = 2
+ HTTP_SERVE_WAIT_REQUEST = 0, /* Waiting for a full request */
+ HTTP_SERVE_RECV_POSTDATA = 1, /* Waiting to finish recieving POST data */
+ HTTP_SERVE_SEND_DATA = 2 /* Sending response */
};
-class HttpSocket : public InspSocket
+/** A socket used for HTTP transport
+ */
+class HttpServerSocket : public BufferedSocket
{
FileReader* index;
HttpState InternalState;
- std::stringstream headers;
+ std::string ip;
+
+ HTTPHeaders headers;
+ std::string reqbuffer;
+ std::string postdata;
+ unsigned int postsize;
+ std::string request_type;
+ std::string uri;
+ std::string http_version;
public:
- HttpSocket(std::string host, int port, bool listening, unsigned long maxtime, FileReader* index_page) : InspSocket(host, port, listening, maxtime), index(index_page)
+ HttpServerSocket(int newfd, const char* IP, FileReader* ind)
+ : BufferedSocket(newfd), index(ind), ip(IP), postsize(0)
{
- log(DEBUG,"HttpSocket constructor");
- InternalState = HTTP_LISTEN;
+ InternalState = HTTP_SERVE_WAIT_REQUEST;
}
- HttpSocket(int newfd, char* ip, FileReader* ind) : InspSocket(newfd, ip), index(ind)
+ FileReader* GetIndex()
{
- InternalState = HTTP_SERVE_WAIT_REQUEST;
+ return index;
}
- virtual int OnIncomingConnection(int newsock, char* ip)
+ ~HttpServerSocket()
{
- if (InternalState == HTTP_LISTEN)
- {
- HttpSocket* s = new HttpSocket(newsock, ip, index);
- Srv->AddSocket(s);
- }
- return true;
}
- virtual void OnClose()
+ virtual void OnError(BufferedSocketError)
{
}
default:
return "WTF";
break;
-
+
}
}
- void SendHeaders(unsigned long size, int response, const std::string &extraheaders)
+ void SendHTTPError(int response)
{
- struct tm *timeinfo = localtime(&TIME);
- this->Write("HTTP/1.1 "+ConvToStr(response)+" "+Response(response)+"\r\nDate: ");
- this->Write(asctime(timeinfo));
- if (extraheaders.empty())
+ HTTPHeaders empty;
+ std::string data = "<html><head></head><body>Server error "+ConvToStr(response)+": "+Response(response)+"<br>"+
+ "<small>Powered by <a href='http://www.inspircd.org'>InspIRCd</a></small></body></html>";
+
+ SendHeaders(data.length(), response, empty);
+ WriteData(data);
+ }
+
+ void SendHeaders(unsigned long size, int response, HTTPHeaders &rheaders)
+ {
+
+ WriteData(http_version + " "+ConvToStr(response)+" "+Response(response)+"\r\n");
+
+ time_t local = ServerInstance->Time();
+ struct tm *timeinfo = gmtime(&local);
+ char *date = asctime(timeinfo);
+ date[strlen(date) - 1] = '\0';
+ rheaders.CreateHeader("Date", date);
+
+ rheaders.CreateHeader("Server", "InspIRCd/m_httpd.so/1.2");
+ rheaders.SetHeader("Content-Length", ConvToStr(size));
+
+ if (size)
+ rheaders.CreateHeader("Content-Type", "text/html");
+ else
+ rheaders.RemoveHeader("Content-Type");
+
+ /* Supporting Connection: keep-alive causes a whole world of hurt syncronizing timeouts,
+ * so remove it, its not essential for what we need.
+ */
+ rheaders.SetHeader("Connection", "Close");
+
+ WriteData(rheaders.GetFormattedHeaders());
+ WriteData("\r\n");
+ }
+
+ void OnDataReady()
+ {
+ if (InternalState == HTTP_SERVE_RECV_POSTDATA)
{
- this->Write("Content-Type: text/html\r\n");
+ postdata.append(recvq);
+ if (postdata.length() >= postsize)
+ ServeData();
}
else
{
- this->Write(extraheaders);
+ reqbuffer.append(recvq);
+
+ if (reqbuffer.length() >= 8192)
+ {
+ ServerInstance->Logs->Log("m_httpd",DEBUG, "m_httpd dropped connection due to an oversized request buffer");
+ reqbuffer.clear();
+ SetError("Buffer");
+ }
+
+ if (InternalState == HTTP_SERVE_WAIT_REQUEST)
+ CheckRequestBuffer();
}
- this->Write("Server: InspIRCd/m_httpd.so/1.1\r\nContent-Length: "+ConvToStr(size)+
- "\r\nConnection: close\r\n\r\n");
}
- virtual bool OnDataReady()
+ void CheckRequestBuffer()
{
- char* data = this->Read();
- std::string request_type;
- std::string uri;
- std::string http_version;
+ std::string::size_type reqend = reqbuffer.find("\r\n\r\n");
+ if (reqend == std::string::npos)
+ return;
- /* Check that the data read is a valid pointer and it has some content */
- if (data && *data)
+ // We have the headers; parse them all
+ std::string::size_type hbegin = 0, hend;
+ while ((hend = reqbuffer.find("\r\n", hbegin)) != std::string::npos)
{
- headers << data;
+ if (hbegin == hend)
+ break;
- if (headers.str().find("\r\n\r\n") != std::string::npos)
+ if (request_type.empty())
{
- /* Headers are complete */
- InternalState = HTTP_SERVE_SEND_DATA;
+ std::istringstream cheader(std::string(reqbuffer, hbegin, hend - hbegin));
+ cheader >> request_type;
+ cheader >> uri;
+ cheader >> http_version;
- headers >> request_type;
- headers >> uri;
- headers >> http_version;
-
- if ((http_version != "HTTP/1.1") && (http_version != "HTTP/1.0"))
- {
- SendHeaders(0, 505, "");
- }
- else
+ if (request_type.empty() || uri.empty() || http_version.empty())
{
- if ((request_type == "GET") && (uri == "/"))
- {
- SendHeaders(index->ContentSize(), 200, "");
- this->Write(index->Contents());
- }
- else
- {
- claimed = false;
- HTTPRequest httpr(request_type,uri,&headers,this,this->GetIP());
- Event e((char*)&httpr, (Module*)HttpModule, "httpd_url");
- e.Send();
-
- if (!claimed)
- {
- SendHeaders(0, 404, "");
- log(DEBUG,"Page not claimed, 404");
- }
- }
+ SendHTTPError(400);
+ return;
}
- return false;
+ hbegin = hend + 2;
+ continue;
+ }
+
+ std::string cheader = reqbuffer.substr(hbegin, hend - hbegin);
+
+ std::string::size_type fieldsep = cheader.find(':');
+ if ((fieldsep == std::string::npos) || (fieldsep == 0) || (fieldsep == cheader.length() - 1))
+ {
+ SendHTTPError(400);
+ return;
+ }
+
+ headers.SetHeader(cheader.substr(0, fieldsep), cheader.substr(fieldsep + 2));
+
+ hbegin = hend + 2;
+ }
+
+ reqbuffer.erase(0, reqend + 4);
+
+ std::transform(request_type.begin(), request_type.end(), request_type.begin(), ::toupper);
+ std::transform(http_version.begin(), http_version.end(), http_version.begin(), ::toupper);
+
+ if ((http_version != "HTTP/1.1") && (http_version != "HTTP/1.0"))
+ {
+ SendHTTPError(505);
+ return;
+ }
+
+ if (headers.IsSet("Content-Length") && (postsize = atoi(headers.GetHeader("Content-Length").c_str())) != 0)
+ {
+ InternalState = HTTP_SERVE_RECV_POSTDATA;
+
+ if (reqbuffer.length() >= postsize)
+ {
+ postdata = reqbuffer.substr(0, postsize);
+ reqbuffer.erase(0, postsize);
+ }
+ else if (!reqbuffer.empty())
+ {
+ postdata = reqbuffer;
+ reqbuffer.clear();
}
- return true;
+
+ if (postdata.length() >= postsize)
+ ServeData();
+
+ return;
+ }
+
+ ServeData();
+ }
+
+ void ServeData()
+ {
+ InternalState = HTTP_SERVE_SEND_DATA;
+
+ if ((request_type == "GET") && (uri == "/"))
+ {
+ HTTPHeaders empty;
+ SendHeaders(index->ContentSize(), 200, empty);
+ WriteData(index->Contents());
}
else
{
- /* Bastard client closed the socket on us!
- * Oh wait, theyre SUPPOSED to do that!
- */
- return false;
+ claimed = false;
+ HTTPRequest httpr(request_type,uri,&headers,this,ip,postdata);
+ Event acl((char*)&httpr, (Module*)HttpModule, "httpd_acl");
+ acl.Send();
+ if (!claimed)
+ {
+ Event e((char*)&httpr, (Module*)HttpModule, "httpd_url");
+ e.Send();
+ if (!claimed)
+ {
+ SendHTTPError(404);
+ }
+ }
}
}
- void Page(std::stringstream* n, int response, std::string& extraheaders)
+ void Page(std::stringstream* n, int response, HTTPHeaders *hheaders)
{
- log(DEBUG,"Sending page");
- SendHeaders(n->str().length(), response, extraheaders);
- this->Write(n->str());
+ SendHeaders(n->str().length(), response, *hheaders);
+ WriteData(n->str());
}
};
-class ModuleHttp : public Module
+/** Spawn HTTP sockets from a listener
+ */
+class HttpListener : public ListenSocketBase
{
- int port;
- std::string host;
- std::string bindip;
- std::string indexfile;
+ FileReader* index;
- FileReader index;
+ public:
+ HttpListener(FileReader *idx, int port, const std::string &addr) : ListenSocketBase(port, addr)
+ {
+ this->index = idx;
+ }
- HttpSocket* http;
+ virtual void OnAcceptReady(int nfd)
+ {
+ int port;
+ std::string incomingip;
+ irc::sockets::satoap(&client, incomingip, port);
+ new HttpServerSocket(nfd, incomingip.c_str(), index);
+ }
+};
+class ModuleHttpServer : public Module
+{
+ std::vector<HttpServerSocket *> httpsocks;
+ std::vector<HttpListener *> httplisteners;
public:
void ReadConfig()
{
+ int port;
+ std::string host;
+ std::string bindip;
+ std::string indexfile;
+ FileReader* index;
+ HttpListener *http;
ConfigReader c;
- this->host = c.ReadValue("http", "host", 0);
- this->bindip = c.ReadValue("http", "ip", 0);
- this->port = c.ReadInteger("http", "port", 0, true);
- this->indexfile = c.ReadValue("http", "index", 0);
- index.LoadFile(this->indexfile);
- }
+ httpsocks.clear(); // XXX this will BREAK if this module is made rehashable
+ httplisteners.clear();
- void CreateListener()
- {
- http = new HttpSocket(this->bindip, this->port, true, 0, &index);
- if ((http) && (http->GetState() == I_LISTENING))
+ for (int i = 0; i < c.Enumerate("http"); i++)
{
- Srv->AddSocket(http);
+ host = c.ReadValue("http", "host", i);
+ bindip = c.ReadValue("http", "ip", i);
+ port = c.ReadInteger("http", "port", i, true);
+ indexfile = c.ReadValue("http", "index", i);
+ index = new FileReader(indexfile);
+ if (!index->Exists())
+ throw ModuleException("Can't read index file: "+indexfile);
+ http = new HttpListener(index, port, (char *)bindip.c_str()); // XXX this cast SUCKS.
+ httplisteners.push_back(http);
}
}
- ModuleHttp(Server* Me) : Module::Module(Me)
- {
- Srv = Me;
+ ModuleHttpServer() {
ReadConfig();
- CreateListener();
- }
-
- void OnEvent(Event* event)
- {
+ HttpModule = this;
+ Implementation eventlist[] = { I_OnRequest };
+ ServerInstance->Modules->Attach(eventlist, this, 1);
}
- char* OnRequest(Request* request)
+ virtual const char* OnRequest(Request* request)
{
- log(DEBUG,"Got HTTPDocument object");
claimed = true;
HTTPDocument* doc = (HTTPDocument*)request->GetData();
- HttpSocket* sock = (HttpSocket*)doc->sock;
- sock->Page(doc->GetDocument(), doc->GetResponseCode(), doc->GetExtraHeaders());
+ HttpServerSocket* sock = (HttpServerSocket*)doc->sock;
+ sock->Page(doc->GetDocument(), doc->GetResponseCode(), &doc->headers);
return NULL;
}
- void Implements(char* List)
- {
- List[I_OnEvent] = List[I_OnRequest] = 1;
- }
- virtual ~ModuleHttp()
+ virtual ~ModuleHttpServer()
{
- Srv->DelSocket(http);
- }
+ for (size_t i = 0; i < httplisteners.size(); i++)
+ {
+ delete httplisteners[i];
+ }
- virtual Version GetVersion()
- {
- return Version(1,0,0,0,VF_STATIC|VF_VENDOR|VF_SERVICEPROVIDER);
+ for (size_t i = 0; i < httpsocks.size(); i++)
+ {
+ ServerInstance->SE->DelFd(httpsocks[i]);
+ httpsocks[i]->Close();
+ delete httpsocks[i]->GetIndex();
+ }
}
-};
-
-class ModuleHttpFactory : public ModuleFactory
-{
- public:
- ModuleHttpFactory()
- {
- }
-
- ~ModuleHttpFactory()
- {
- }
-
- virtual Module * CreateModule(Server* Me)
+ virtual Version GetVersion()
{
- HttpModule = new ModuleHttp(Me);
- return HttpModule;
+ return Version("Provides HTTP serving facilities to modules", VF_VENDOR | VF_SERVICEPROVIDER, API_VERSION);
}
};
-
-extern "C" void * init_module( void )
-{
- return new ModuleHttpFactory;
-}
+MODULE_INIT(ModuleHttpServer)
projects / user / henk / code / inspircd.git / blobdiff