/* +------------------------------------+
* | Inspire Internet Relay Chat Daemon |
* +------------------------------------+
*
* 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.
*
* ---------------------------------------------------
*/
#include "inspircd.h"
#include "httpd.h"
/* $ModDesc: Provides HTTP serving facilities to modules */
/* $ModDep: httpd.h */
class ModuleHttpServer;
static ModuleHttpServer* HttpModule;
static bool claimed;
/** HTTP socket states
*/
enum HttpState
{
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 */
};
/** A socket used for HTTP transport
*/
class HttpServerSocket : public BufferedSocket
{
HttpState InternalState;
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:
HttpServerSocket(int newfd, const std::string& IP, ListenSocket* via, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server)
: BufferedSocket(newfd), ip(IP), postsize(0)
{
InternalState = HTTP_SERVE_WAIT_REQUEST;
FOREACH_MOD(I_OnHookIO, OnHookIO(this, via));
if (GetIOHook())
GetIOHook()->OnStreamSocketAccept(this, client, server);
}
virtual void OnError(BufferedSocketError)
{
}
std::string Response(int response)
{
switch (response)
{
case 100:
return "CONTINUE";
case 101:
return "SWITCHING PROTOCOLS";
case 200:
return "OK";
case 201:
return "CREATED";
case 202:
return "ACCEPTED";
case 203:
return "NON-AUTHORITATIVE INFORMATION";
case 204:
return "NO CONTENT";
case 205:
return "RESET CONTENT";
case 206:
return "PARTIAL CONTENT";
case 300:
return "MULTIPLE CHOICES";
case 301:
return "MOVED PERMENANTLY";
case 302:
return "FOUND";
case 303:
return "SEE OTHER";
case 304:
return "NOT MODIFIED";
case 305:
return "USE PROXY";
case 307:
return "TEMPORARY REDIRECT";
case 400:
return "BAD REQUEST";
case 401:
return "UNAUTHORIZED";
case 402:
return "PAYMENT REQUIRED";
case 403:
return "FORBIDDEN";
case 404:
return "NOT FOUND";
case 405:
return "METHOD NOT ALLOWED";
case 406:
return "NOT ACCEPTABLE";
case 407:
return "PROXY AUTHENTICATION REQUIRED";
case 408:
return "REQUEST TIMEOUT";
case 409:
return "CONFLICT";
case 410:
return "GONE";
case 411:
return "LENGTH REQUIRED";
case 412:
return "PRECONDITION FAILED";
case 413:
return "REQUEST ENTITY TOO LARGE";
case 414:
return "REQUEST-URI TOO LONG";
case 415:
return "UNSUPPORTED MEDIA TYPE";
case 416:
return "REQUESTED RANGE NOT SATISFIABLE";
case 417:
return "EXPECTATION FAILED";
case 500:
return "INTERNAL SERVER ERROR";
case 501:
return "NOT IMPLEMENTED";
case 502:
return "BAD GATEWAY";
case 503:
return "SERVICE UNAVAILABLE";
case 504:
return "GATEWAY TIMEOUT";
case 505:
return "HTTP VERSION NOT SUPPORTED";
default:
return "WTF";
break;
}
}
void SendHTTPError(int response)
{
HTTPHeaders empty;
std::string data = "Server error "+ConvToStr(response)+": "+Response(response)+"
"+
"Powered by InspIRCd";
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)
{
postdata.append(recvq);
if (postdata.length() >= postsize)
ServeData();
}
else
{
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();
}
}
void CheckRequestBuffer()
{
std::string::size_type reqend = reqbuffer.find("\r\n\r\n");
if (reqend == std::string::npos)
return;
// We have the headers; parse them all
std::string::size_type hbegin = 0, hend;
while ((hend = reqbuffer.find("\r\n", hbegin)) != std::string::npos)
{
if (hbegin == hend)
break;
if (request_type.empty())
{
std::istringstream cheader(std::string(reqbuffer, hbegin, hend - hbegin));
cheader >> request_type;
cheader >> uri;
cheader >> http_version;
if (request_type.empty() || uri.empty() || http_version.empty())
{
SendHTTPError(400);
return;
}
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();
}
if (postdata.length() >= postsize)
ServeData();
return;
}
ServeData();
}
void ServeData()
{
InternalState = HTTP_SERVE_SEND_DATA;
claimed = false;
HTTPRequest acl((Module*)HttpModule, "httpd_acl", request_type, uri, &headers, this, ip, postdata);
acl.Send();
if (!claimed)
{
HTTPRequest url((Module*)HttpModule, "httpd_url", request_type, uri, &headers, this, ip, postdata);
url.Send();
if (!claimed)
{
SendHTTPError(404);
}
}
}
void Page(std::stringstream* n, int response, HTTPHeaders *hheaders)
{
SendHeaders(n->str().length(), response, *hheaders);
WriteData(n->str());
}
};
class ModuleHttpServer : public Module
{
std::vector httpsocks;
public:
ModuleHttpServer() {
HttpModule = this;
}
void OnRequest(Request& request)
{
if (strcmp(request.id, "HTTP-DOC") != 0)
return;
HTTPDocumentResponse& resp = static_cast(request);
claimed = true;
resp.src.sock->Page(resp.document, resp.responsecode, &resp.headers);
}
ModResult OnAcceptConnection(int nfd, ListenSocket* from, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server)
{
if (from->bind_tag->getString("type") != "httpd");
return MOD_RES_PASSTHRU;
int port;
std::string incomingip;
irc::sockets::satoap(*client, incomingip, port);
new HttpServerSocket(nfd, incomingip, from, client, server);
return MOD_RES_ALLOW;
}
virtual ~ModuleHttpServer()
{
for (size_t i = 0; i < httpsocks.size(); i++)
{
httpsocks[i]->cull();
delete httpsocks[i];
}
}
virtual Version GetVersion()
{
return Version("Provides HTTP serving facilities to modules", VF_VENDOR);
}
};
MODULE_INIT(ModuleHttpServer)