* | Inspire Internet Relay Chat Daemon |
* +------------------------------------+
*
- * InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * InspIRCd: (C) 2002-2008 InspIRCd Development Team
* See: http://www.inspircd.org/wiki/index.php/Credits
*
* This program is free but copyrighted software; see
*/
#include "inspircd.h"
-#include <algorithm>
-#include "modules.h"
#include "httpd.h"
/* $ModDesc: Provides HTTP serving facilities to modules */
HTTP_SERVE_SEND_DATA = 3 /* Sending response */
};
-class HttpServerSocket;
-
-/** This class is used to handle HTTP socket timeouts
- */
-class HttpServerTimeout : public InspTimer
-{
- private:
- /** HttpServerSocket we are attached to
- */
- HttpServerSocket* s;
- /** Socketengine the file descriptor is in
- */
- SocketEngine* SE;
- public:
- /** Attach timeout to HttpServerSocket
- */
- HttpServerTimeout(HttpServerSocket* sock, SocketEngine* engine);
- /** Handle timer tick
- */
- void Tick(time_t TIME);
-};
-
/** A socket used for HTTP transport
*/
-class HttpServerSocket : public InspSocket
+class HttpServerSocket : public BufferedSocket
{
FileReader* index;
HttpState InternalState;
-
+
HTTPHeaders headers;
std::string reqbuffer;
std::string postdata;
std::string request_type;
std::string uri;
std::string http_version;
- bool keepalive;
-
- HttpServerTimeout* Timeout;
- bool DataSinceLastTick;
- friend class HttpServerTimeout;
-
+
public:
- HttpServerSocket(InspIRCd* SI, std::string host, int port, bool listening, unsigned long maxtime, FileReader* index_page) : InspSocket(SI, host, port, listening, maxtime), index(index_page), postsize(0)
+ HttpServerSocket(InspIRCd* SI, std::string shost, int iport, bool listening, unsigned long maxtime, FileReader* index_page) : BufferedSocket(SI, shost, iport, listening, maxtime), index(index_page), postsize(0)
{
InternalState = HTTP_LISTEN;
- Timeout = NULL;
}
- HttpServerSocket(InspIRCd* SI, int newfd, char* ip, FileReader* ind) : InspSocket(SI, newfd, ip), index(ind), postsize(0), keepalive(false), DataSinceLastTick(false)
+ HttpServerSocket(InspIRCd* SI, int newfd, char* ip, FileReader* ind) : BufferedSocket(SI, newfd, ip), index(ind), postsize(0)
{
InternalState = HTTP_SERVE_WAIT_REQUEST;
- Timeout = new HttpServerTimeout(this, Instance->SE);
- Instance->Timers->AddTimer(Timeout);
}
FileReader* GetIndex()
~HttpServerSocket()
{
- if (Timeout)
- {
- if (Instance->Time() < Timeout->GetTimer())
- Instance->Timers->DelTimer(Timeout);
- Timeout = NULL;
- }
}
-
- void ResetRequest()
- {
- headers.Clear();
- postdata.clear();
- postsize = 0;
- request_type.clear();
- uri.clear();
- http_version.clear();
- InternalState = HTTP_SERVE_WAIT_REQUEST;
-
- if (reqbuffer.size())
- CheckRequestBuffer();
- }
-
+
virtual int OnIncomingConnection(int newsock, char* ip)
{
if (InternalState == HTTP_LISTEN)
default:
return "WTF";
break;
-
+
}
}
-
+
void SendHTTPError(int response)
{
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);
this->Write(data);
-
- if (keepalive)
- ResetRequest();
+ this->FlushWriteBuffer();
}
-
+
void SendHeaders(unsigned long size, int response, HTTPHeaders &rheaders)
{
char *date = asctime(timeinfo);
date[strlen(date) - 1] = '\0';
rheaders.CreateHeader("Date", date);
-
- rheaders.CreateHeader("Server", "InspIRCd/m_httpd.so/1.1");
+
+ 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");
-
- if (rheaders.GetHeader("Connection") == "Close")
- keepalive = false;
- else if (rheaders.GetHeader("Connection") == "Keep-Alive" && !headers.IsSet("Connection"))
- keepalive = true;
- else if (!rheaders.IsSet("Connection") && !keepalive)
- rheaders.SetHeader("Connection", "Close");
-
+
+ /* 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");
+
this->Write(rheaders.GetFormattedHeaders());
this->Write("\r\n");
-
- if (!size && keepalive)
- ResetRequest();
}
virtual bool OnDataReady()
{
- char* data = this->Read();
+ const char* data = this->Read();
/* Check that the data read is a valid pointer and it has some content */
if (!data || !*data)
return false;
-
- DataSinceLastTick = true;
-
+
if (InternalState == HTTP_SERVE_RECV_POSTDATA)
{
postdata.append(data);
else
{
reqbuffer.append(data);
-
+
if (reqbuffer.length() >= 8192)
{
- Instance->Log(DEBUG, "m_httpd dropped connection due to an oversized request buffer");
+ Instance->Logs->Log("m_httpd",DEBUG, "m_httpd dropped connection due to an oversized request buffer");
reqbuffer.clear();
return false;
}
-
+
if (InternalState == HTTP_SERVE_WAIT_REQUEST)
CheckRequestBuffer();
}
-
+
return true;
}
-
+
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);
+ SetWrite();
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);
+ SetWrite();
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);
+ SetWrite();
return;
}
-
- if (strcasecmp(headers.GetHeader("Connection").c_str(), "keep-alive") == 0)
- keepalive = true;
-
+
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);
postdata = reqbuffer;
reqbuffer.clear();
}
-
+
if (postdata.length() >= postsize)
ServeData();
-
+
return;
}
-
+
ServeData();
}
HTTPHeaders empty;
SendHeaders(index->ContentSize(), 200, empty);
this->Write(index->Contents());
+ this->FlushWriteBuffer();
+ SetWrite();
}
else
{
claimed = false;
HTTPRequest httpr(request_type,uri,&headers,this,this->GetIP(),postdata);
- Event e((char*)&httpr, (Module*)HttpModule, "httpd_url");
- e.Send(this->Instance);
+ Event acl((char*)&httpr, (Module*)HttpModule, "httpd_acl");
+ acl.Send(this->Instance);
if (!claimed)
{
- SendHTTPError(404);
+ Event e((char*)&httpr, (Module*)HttpModule, "httpd_url");
+ e.Send(this->Instance);
+ if (!claimed)
+ {
+ SendHTTPError(404);
+ SetWrite();
+ }
}
}
}
- void Page(std::stringstream* n, int response, HTTPHeaders *headers)
+
+ bool OnWriteReady()
{
- SendHeaders(n->str().length(), response, *headers);
- this->Write(n->str());
-
- if (!keepalive)
- {
- Instance->SE->DelFd(this);
- this->Close();
- }
- else
- this->ResetRequest();
+ Instance->Logs->Log("m_httpd",DEBUG,"OnWriteReady()");
+ return false;
}
-};
-HttpServerTimeout::HttpServerTimeout(HttpServerSocket* sock, SocketEngine* engine) : InspTimer(15, time(NULL), true), s(sock), SE(engine)
-{
-}
+ void Page(std::stringstream* n, int response, HTTPHeaders *hheaders)
+ {
+ SendHeaders(n->str().length(), response, *hheaders);
+ this->Write(n->str());
+ this->FlushWriteBuffer();
+ SetWrite();
+ }
-void HttpServerTimeout::Tick(time_t TIME)
-{
- if (!s->DataSinceLastTick)
+ void SetWrite()
{
- SE->DelFd(s);
- s->Close();
- s->Timeout = NULL;
- this->CancelRepeat();
+ Instance->Logs->Log("m_httpd",DEBUG,"SetWrite()");
+ this->WaitingForWriteEvent = true;
+ Instance->SE->WantWrite(this);
}
- else
- s->DataSinceLastTick = false;
-}
+};
class ModuleHttpServer : public Module
{
{
ReadConfig();
HttpModule = this;
+ Implementation eventlist[] = { I_OnRequest };
+ ServerInstance->Modules->Attach(eventlist, this, 1);
}
- char* OnRequest(Request* request)
+ virtual const char* OnRequest(Request* request)
{
claimed = true;
HTTPDocument* doc = (HTTPDocument*)request->GetData();
return NULL;
}
- void Implements(char* List)
- {
- List[I_OnRequest] = 1;
- }
virtual ~ModuleHttpServer()
{
httpsocks[i]->Close();
delete httpsocks[i]->GetIndex();
}
- ServerInstance->InspSocketCull();
+ ServerInstance->BufferedSocketCull();
}
virtual Version GetVersion()
{
- return Version(1,1,0,0,VF_VENDOR|VF_SERVICEPROVIDER,API_VERSION);
+ return Version(1,2,0,0,VF_VENDOR|VF_SERVICEPROVIDER,API_VERSION);
}
};
projects / user / henk / code / inspircd.git / blobdiff