From bab14f0dd2345c9d7dcbc47c918563709e1ac094 Mon Sep 17 00:00:00 2001 From: peavey Date: Mon, 16 Jul 2007 17:14:45 +0000 Subject: 'svn propset -R svn:eol-style CR *' Set to UNIX-style always. Binaries are auto skipped by svn. git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@7454 e03df62e-2008-0410-955e-edbf42e46eb7 --- src/modules/m_http_client.cpp | 347 +----------------------------------------- 1 file changed, 1 insertion(+), 346 deletions(-) (limited to 'src/modules/m_http_client.cpp') diff --git a/src/modules/m_http_client.cpp b/src/modules/m_http_client.cpp index 35b93b581..3f9875caf 100644 --- a/src/modules/m_http_client.cpp +++ b/src/modules/m_http_client.cpp @@ -1,346 +1 @@ -/* +------------------------------------+ - * | Inspire Internet Relay Chat Daemon | - * +------------------------------------+ - * - * InspIRCd: (C) 2002-2007 InspIRCd Development Team - * See: http://www.inspircd.org/wiki/index.php/Credits - * - * This program is free but copyrighted software; see - * the file COPYING for details. - * - * --------------------------------------------------- - */ - -#include "inspircd.h" -#include "httpclient.h" - -/* $ModDesc: HTTP client service provider */ - -class URL -{ - public: - std::string url; - std::string protocol, username, password, domain, request; - int port; -}; - -class HTTPSocket : public InspSocket -{ - private: - InspIRCd *Server; - class ModuleHTTPClient *Mod; - HTTPClientRequest req; - HTTPClientResponse *response; - URL url; - enum { HTTP_CLOSED, HTTP_REQSENT, HTTP_HEADERS, HTTP_DATA } status; - std::string data; - std::string buffer; - - public: - HTTPSocket(InspIRCd *Instance, class ModuleHTTPClient *Mod); - virtual ~HTTPSocket(); - virtual bool DoRequest(HTTPClientRequest *req); - virtual bool ParseURL(const std::string &url); - virtual void Connect(const std::string &ip); - virtual bool OnConnected(); - virtual bool OnDataReady(); - virtual void OnClose(); -}; - -class HTTPResolver : public Resolver -{ - private: - HTTPSocket *socket; - public: - HTTPResolver(HTTPSocket *socket, InspIRCd *Instance, const string &hostname, bool &cached, Module* me) : Resolver(Instance, hostname, DNS_QUERY_FORWARD, cached, me), socket(socket) - { - } - - void OnLookupComplete(const string &result, unsigned int ttl, bool cached) - { - socket->Connect(result); - } - - void OnError(ResolverError e, const string &errmsg) - { - delete socket; - } -}; - -typedef vector HTTPList; - -class ModuleHTTPClient : public Module -{ - public: - HTTPList sockets; - - ModuleHTTPClient(InspIRCd *Me) - : Module(Me) - { - } - - virtual ~ModuleHTTPClient() - { - for (HTTPList::iterator i = sockets.begin(); i != sockets.end(); i++) - delete *i; - } - - virtual Version GetVersion() - { - return Version(1, 0, 0, 0, VF_SERVICEPROVIDER | VF_VENDOR, API_VERSION); - } - - void Implements(char* List) - { - List[I_OnRequest] = 1; - } - - char* OnRequest(Request *req) - { - HTTPClientRequest *httpreq = (HTTPClientRequest *)req; - if (!strcmp(httpreq->GetId(), HTTP_CLIENT_REQUEST)) - { - HTTPSocket *sock = new HTTPSocket(ServerInstance, this); - sock->DoRequest(httpreq); - // No return value - } - return NULL; - } -}; - -HTTPSocket::HTTPSocket(InspIRCd *Instance, ModuleHTTPClient *Mod) - : InspSocket(Instance), Server(Instance), Mod(Mod), status(HTTP_CLOSED) -{ - this->ClosePending = false; - this->port = 80; -} - -HTTPSocket::~HTTPSocket() -{ - Close(); - for (HTTPList::iterator i = Mod->sockets.begin(); i != Mod->sockets.end(); i++) - { - if (*i == this) - { - Mod->sockets.erase(i); - break; - } - } -} - -bool HTTPSocket::DoRequest(HTTPClientRequest *req) -{ - /* Tweak by brain - we take a copy of this, - * so that the caller doesnt need to leave - * pointers knocking around, less chance of - * a memory leak. - */ - this->req = *req; - - if (!ParseURL(this->req.GetURL())) - return false; - - this->port = url.port; - strlcpy(this->host, url.domain.c_str(), MAXBUF); - - in_addr addy1; -#ifdef IPV6 - in6_addr addy2; - if ((inet_aton(this->host, &addy1) > 0) || (inet_pton(AF_INET6, this->host, &addy2) > 0)) -#else - if (inet_aton(this->host, &addy1) > 0) -#endif - { - bool cached; - HTTPResolver* r = new HTTPResolver(this, Server, url.domain, cached, (Module*)Mod); - Instance->AddResolver(r, cached); - return true; - } - else - { - this->Connect(url.domain); - } - - return true; -} - -bool HTTPSocket::ParseURL(const std::string &iurl) -{ - url.url = iurl; - url.port = 80; - url.protocol = "http"; - - irc::sepstream tokenizer(iurl, '/'); - - for (int p = 0;; p++) - { - std::string part = tokenizer.GetToken(); - if (part.empty() && tokenizer.StreamEnd()) - break; - - if ((p == 0) && (part[part.length() - 1] == ':')) - { - // Protocol ('http:') - url.protocol = part.substr(0, part.length() - 1); - } - else if ((p == 1) && (part.empty())) - { - continue; - } - else if (url.domain.empty()) - { - // Domain part: [user[:pass]@]domain[:port] - std::string::size_type usrpos = part.find('@'); - if (usrpos != std::string::npos) - { - // Have a user (and possibly password) part - std::string::size_type ppos = part.find(':'); - if ((ppos != std::string::npos) && (ppos < usrpos)) - { - // Have password too - url.password = part.substr(ppos + 1, usrpos - ppos - 1); - url.username = part.substr(0, ppos); - } - else - { - url.username = part.substr(0, usrpos); - } - - part = part.substr(usrpos + 1); - } - - std::string::size_type popos = part.rfind(':'); - if (popos != std::string::npos) - { - url.port = atoi(part.substr(popos + 1).c_str()); - url.domain = part.substr(0, popos); - } - else - { - url.domain = part; - } - } - else - { - // Request (part of it).. - url.request.append("/"); - url.request.append(part); - } - } - - if (url.request.empty()) - url.request = "/"; - - if ((url.domain.empty()) || (!url.port) || (url.protocol.empty())) - { - Instance->Log(DEFAULT, "Invalid URL (%s): Missing required value", iurl.c_str()); - return false; - } - - if (url.protocol != "http") - { - Instance->Log(DEFAULT, "Invalid URL (%s): Unsupported protocol '%s'", iurl.c_str(), url.protocol.c_str()); - return false; - } - - return true; -} - -void HTTPSocket::Connect(const string &ip) -{ - strlcpy(this->IP, ip.c_str(), MAXBUF); - - if (!this->DoConnect()) - { - delete this; - } -} - -bool HTTPSocket::OnConnected() -{ - std::string request = "GET " + url.request + " HTTP/1.1\r\n"; - - // Dump headers into the request - HeaderMap headers = req.GetHeaders(); - - for (HeaderMap::iterator i = headers.begin(); i != headers.end(); i++) - request += i->first + ": " + i->second + "\r\n"; - - // The Host header is required for HTTP 1.1 and isn't known when the request is created; if they didn't overload it - // manually, add it here - if (headers.find("Host") == headers.end()) - request += "Host: " + url.domain + "\r\n"; - - request += "\r\n"; - - this->status = HTTP_REQSENT; - - return this->Write(request); -} - -bool HTTPSocket::OnDataReady() -{ - char *data = this->Read(); - - if (!data) - { - this->Close(); - return false; - } - - if (this->status < HTTP_DATA) - { - std::string line; - std::string::size_type pos; - - this->buffer += data; - while ((pos = buffer.find("\r\n")) != std::string::npos) - { - line = buffer.substr(0, pos); - buffer = buffer.substr(pos + 2); - if (line.empty()) - { - this->status = HTTP_DATA; - this->data += this->buffer; - this->buffer.clear(); - break; - } - - if (this->status == HTTP_REQSENT) - { - // HTTP reply (HTTP/1.1 200 msg) - char const* data = line.c_str(); - data += 9; - response = new HTTPClientResponse((Module*)Mod, req.GetSource() , url.url, atoi(data), data + 4); - this->status = HTTP_HEADERS; - continue; - } - - if ((pos = line.find(':')) != std::string::npos) - { - response->AddHeader(line.substr(0, pos), line.substr(pos + 1)); - } - else - { - continue; - } - } - } - else - { - this->data += data; - } - return true; -} - -void HTTPSocket::OnClose() -{ - if (data.empty()) - return; // notification that request failed? - - response->data = data; - response->Send(); - delete response; -} - -MODULE_INIT(ModuleHTTPClient) +/* +------------------------------------+ * | Inspire Internet Relay Chat Daemon | * +------------------------------------+ * * InspIRCd: (C) 2002-2007 InspIRCd Development Team * See: http://www.inspircd.org/wiki/index.php/Credits * * This program is free but copyrighted software; see * the file COPYING for details. * * --------------------------------------------------- */ #include "inspircd.h" #include "httpclient.h" /* $ModDesc: HTTP client service provider */ class URL { public: std::string url; std::string protocol, username, password, domain, request; int port; }; class HTTPSocket : public InspSocket { private: InspIRCd *Server; class ModuleHTTPClient *Mod; HTTPClientRequest req; HTTPClientResponse *response; URL url; enum { HTTP_CLOSED, HTTP_REQSENT, HTTP_HEADERS, HTTP_DATA } status; std::string data; std::string buffer; public: HTTPSocket(InspIRCd *Instance, class ModuleHTTPClient *Mod); virtual ~HTTPSocket(); virtual bool DoRequest(HTTPClientRequest *req); virtual bool ParseURL(const std::string &url); virtual void Connect(const std::string &ip); virtual bool OnConnected(); virtual bool OnDataReady(); virtual void OnClose(); }; class HTTPResolver : public Resolver { private: HTTPSocket *socket; public: HTTPResolver(HTTPSocket *socket, InspIRCd *Instance, const string &hostname, bool &cached, Module* me) : Resolver(Instance, hostname, DNS_QUERY_FORWARD, cached, me), socket(socket) { } void OnLookupComplete(const string &result, unsigned int ttl, bool cached) { socket->Connect(result); } void OnError(ResolverError e, const string &errmsg) { delete socket; } }; typedef vector HTTPList; class ModuleHTTPClient : public Module { public: HTTPList sockets; ModuleHTTPClient(InspIRCd *Me) : Module(Me) { } virtual ~ModuleHTTPClient() { for (HTTPList::iterator i = sockets.begin(); i != sockets.end(); i++) delete *i; } virtual Version GetVersion() { return Version(1, 0, 0, 0, VF_SERVICEPROVIDER | VF_VENDOR, API_VERSION); } void Implements(char* List) { List[I_OnRequest] = 1; } char* OnRequest(Request *req) { HTTPClientRequest *httpreq = (HTTPClientRequest *)req; if (!strcmp(httpreq->GetId(), HTTP_CLIENT_REQUEST)) { HTTPSocket *sock = new HTTPSocket(ServerInstance, this); sock->DoRequest(httpreq); // No return value } return NULL; } }; HTTPSocket::HTTPSocket(InspIRCd *Instance, ModuleHTTPClient *Mod) : InspSocket(Instance), Server(Instance), Mod(Mod), status(HTTP_CLOSED) { this->ClosePending = false; this->port = 80; } HTTPSocket::~HTTPSocket() { Close(); for (HTTPList::iterator i = Mod->sockets.begin(); i != Mod->sockets.end(); i++) { if (*i == this) { Mod->sockets.erase(i); break; } } } bool HTTPSocket::DoRequest(HTTPClientRequest *req) { /* Tweak by brain - we take a copy of this, * so that the caller doesnt need to leave * pointers knocking around, less chance of * a memory leak. */ this->req = *req; if (!ParseURL(this->req.GetURL())) return false; this->port = url.port; strlcpy(this->host, url.domain.c_str(), MAXBUF); in_addr addy1; #ifdef IPV6 in6_addr addy2; if ((inet_aton(this->host, &addy1) > 0) || (inet_pton(AF_INET6, this->host, &addy2) > 0)) #else if (inet_aton(this->host, &addy1) > 0) #endif { bool cached; HTTPResolver* r = new HTTPResolver(this, Server, url.domain, cached, (Module*)Mod); Instance->AddResolver(r, cached); return true; } else { this->Connect(url.domain); } return true; } bool HTTPSocket::ParseURL(const std::string &iurl) { url.url = iurl; url.port = 80; url.protocol = "http"; irc::sepstream tokenizer(iurl, '/'); for (int p = 0;; p++) { std::string part = tokenizer.GetToken(); if (part.empty() && tokenizer.StreamEnd()) break; if ((p == 0) && (part[part.length() - 1] == ':')) { // Protocol ('http:') url.protocol = part.substr(0, part.length() - 1); } else if ((p == 1) && (part.empty())) { continue; } else if (url.domain.empty()) { // Domain part: [user[:pass]@]domain[:port] std::string::size_type usrpos = part.find('@'); if (usrpos != std::string::npos) { // Have a user (and possibly password) part std::string::size_type ppos = part.find(':'); if ((ppos != std::string::npos) && (ppos < usrpos)) { // Have password too url.password = part.substr(ppos + 1, usrpos - ppos - 1); url.username = part.substr(0, ppos); } else { url.username = part.substr(0, usrpos); } part = part.substr(usrpos + 1); } std::string::size_type popos = part.rfind(':'); if (popos != std::string::npos) { url.port = atoi(part.substr(popos + 1).c_str()); url.domain = part.substr(0, popos); } else { url.domain = part; } } else { // Request (part of it).. url.request.append("/"); url.request.append(part); } } if (url.request.empty()) url.request = "/"; if ((url.domain.empty()) || (!url.port) || (url.protocol.empty())) { Instance->Log(DEFAULT, "Invalid URL (%s): Missing required value", iurl.c_str()); return false; } if (url.protocol != "http") { Instance->Log(DEFAULT, "Invalid URL (%s): Unsupported protocol '%s'", iurl.c_str(), url.protocol.c_str()); return false; } return true; } void HTTPSocket::Connect(const string &ip) { strlcpy(this->IP, ip.c_str(), MAXBUF); if (!this->DoConnect()) { delete this; } } bool HTTPSocket::OnConnected() { std::string request = "GET " + url.request + " HTTP/1.1\r\n"; // Dump headers into the request HeaderMap headers = req.GetHeaders(); for (HeaderMap::iterator i = headers.begin(); i != headers.end(); i++) request += i->first + ": " + i->second + "\r\n"; // The Host header is required for HTTP 1.1 and isn't known when the request is created; if they didn't overload it // manually, add it here if (headers.find("Host") == headers.end()) request += "Host: " + url.domain + "\r\n"; request += "\r\n"; this->status = HTTP_REQSENT; return this->Write(request); } bool HTTPSocket::OnDataReady() { char *data = this->Read(); if (!data) { this->Close(); return false; } if (this->status < HTTP_DATA) { std::string line; std::string::size_type pos; this->buffer += data; while ((pos = buffer.find("\r\n")) != std::string::npos) { line = buffer.substr(0, pos); buffer = buffer.substr(pos + 2); if (line.empty()) { this->status = HTTP_DATA; this->data += this->buffer; this->buffer.clear(); break; } if (this->status == HTTP_REQSENT) { // HTTP reply (HTTP/1.1 200 msg) char const* data = line.c_str(); data += 9; response = new HTTPClientResponse((Module*)Mod, req.GetSource() , url.url, atoi(data), data + 4); this->status = HTTP_HEADERS; continue; } if ((pos = line.find(':')) != std::string::npos) { response->AddHeader(line.substr(0, pos), line.substr(pos + 1)); } else { continue; } } } else { this->data += data; } return true; } void HTTPSocket::OnClose() { if (data.empty()) return; // notification that request failed? response->data = data; response->Send(); delete response; } MODULE_INIT(ModuleHTTPClient) \ No newline at end of file -- cgit v1.3.1-10-gc9f91