From 8b02191cd885f9f81a15f3d2ad99f1def87234cc Mon Sep 17 00:00:00 2001 From: pippijn Date: Wed, 9 Jul 2008 11:24:58 +0000 Subject: - disabled m_nopartmsg for now, since it requires API changes to work - fixed m_httpd_stats so it does not send negative numerical xml entities git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@9950 e03df62e-2008-0410-955e-edbf42e46eb7 --- src/modules/m_httpd_stats.cpp | 358 +++++++++++++++++++++--------------------- 1 file changed, 183 insertions(+), 175 deletions(-) (limited to 'src') diff --git a/src/modules/m_httpd_stats.cpp b/src/modules/m_httpd_stats.cpp index d2205b80b..d3a5ae90d 100644 --- a/src/modules/m_httpd_stats.cpp +++ b/src/modules/m_httpd_stats.cpp @@ -6,7 +6,7 @@ * See: http://www.inspircd.org/wiki/index.php/Credits * * This program is free but copyrighted software; see - * the file COPYING for details. + * the file COPYING for details. * * --------------------------------------------------- */ @@ -20,183 +20,191 @@ class ModuleHttpStats : public Module { - - std::string stylesheet; - bool changed; + static std::map const &entities; + std::string stylesheet; + bool changed; public: - void ReadConfig() - { - ConfigReader c(ServerInstance); - this->stylesheet = c.ReadValue("httpstats", "stylesheet", 0); - } - - ModuleHttpStats(InspIRCd* Me) : Module(Me) - { - ReadConfig(); - this->changed = true; - Implementation eventlist[] = { I_OnEvent, I_OnRequest }; - ServerInstance->Modules->Attach(eventlist, this, 2); - } - - std::string Sanitize(const std::string &str) - { - std::string ret; - - for (std::string::const_iterator x = str.begin(); x != str.end(); ++x) - { - switch (*x) - { - case '<': - ret += "<"; - break; - case '>': - ret += ">"; - break; - case '&': - ret += "&"; - break; - case '"': - ret += """; - break; - default: - if (*x < 32 || *x > 126) - { - int n = *x; - ret += ("&#" + ConvToStr(n) + ";"); - } - else - ret += *x; - break; - } - } - return ret; - } - - void OnEvent(Event* event) - { - std::stringstream data(""); - - if (event->GetEventID() == "httpd_url") - { - ServerInstance->Logs->Log("m_http_stats", DEBUG,"Handling httpd event"); - HTTPRequest* http = (HTTPRequest*)event->GetData(); - - if ((http->GetURI() == "/stats") || (http->GetURI() == "/stats/")) - { - data << ""; - - data << "" << ServerInstance->Config->ServerName << "" << Sanitize(ServerInstance->Config->ServerDesc) << ""; - - data << ""; - data << "" << ServerInstance->Users->clientlist->size() << ""; - data << "" << ServerInstance->chanlist->size() << ""; - data << "" << ServerInstance->Users->all_opers.size() << ""; - data << "" << (ServerInstance->SE->GetMaxFds() - ServerInstance->SE->GetRemainingFds()) << "" << ServerInstance->SE->GetMaxFds() << - "" << ServerInstance->SE->GetName() << ""; - - time_t current_time = 0; - current_time = ServerInstance->Time(); - time_t server_uptime = current_time - ServerInstance->startup_time; - struct tm* stime; - stime = gmtime(&server_uptime); - data << "" << stime->tm_yday << "" << stime->tm_hour << "" << stime->tm_min << "" << stime->tm_sec << "" << ServerInstance->startup_time << ""; - - - data << ""; - data << ""; - std::vector module_names = ServerInstance->Modules->GetAllModuleNames(0); - for (std::vector::iterator i = module_names.begin(); i != module_names.end(); ++i) - { - Module* m = ServerInstance->Modules->Find(i->c_str()); - Version v = m->GetVersion(); - data << "" << *i << "" << v.Major << "." << v.Minor << "." << v.Revision << "." << v.Build << ""; - } - data << ""; - data << ""; - - for (chan_hash::const_iterator a = ServerInstance->chanlist->begin(); a != ServerInstance->chanlist->end(); ++a) - { - Channel* c = a->second; - - data << ""; - data << "" << c->GetUsers()->size() << "" << c->name << ""; - data << "" << c->GetOppedUsers()->size() << ""; - data << "" << c->GetHalfoppedUsers()->size() << ""; - data << "" << c->GetVoicedUsers()->size() << ""; - data << "" << Sanitize(c->topic) << ""; - data << "" << Sanitize(c->ChanModes(true)) << ""; - CUList* ulist = c->GetUsers(); - - for (CUList::iterator x = ulist->begin(); x != ulist->end(); ++x) - { - data << "" << x->first->uuid << "" << Sanitize(c->GetAllPrefixChars(x->first)) << ""; - } - data << ""; - } - - data << ""; - - for (user_hash::const_iterator a = ServerInstance->Users->clientlist->begin(); a != ServerInstance->Users->clientlist->end(); ++a) - { - User* u = a->second; - - data << ""; - data << "" << u->nick << "" << u->uuid << "" << u->host << "" << u->dhost << ""; - data << "" << Sanitize(u->fullname) << "" << u->server << "" << Sanitize(u->awaymsg) << "" << Sanitize(u->oper) << ""; - std::string modes; - for (unsigned char n = 'A'; n <= 'z'; ++n) - if (u->IsModeSet(n)) - modes += n; - - data << modes << "" << Sanitize(u->ident) << "" << u->GetPort() << "" << u->GetIPString() << ""; - data << ""; - } - - data << ""; - - ProtoServerList sl; - ServerInstance->PI->GetServerList(sl); - - for (ProtoServerList::iterator b = sl.begin(); b != sl.end(); ++b) - { - data << ""; - data << "" << b->servername << ""; - data << "" << b->parentname << ""; - data << "" << b->usercount << ""; - data << "" << b->opercount << ""; - data << "" << b->latencyms << ""; - data << ""; - } - data << ""; - - data << ""; - - /* Send the document back to m_httpd */ - HTTPDocument response(http->sock, &data, 200); - response.headers.SetHeader("X-Powered-By", "m_httpd_stats.so"); - response.headers.SetHeader("Content-Type", "text/xml"); - Request req((char*)&response, (Module*)this, event->GetSource()); - req.Send(); - } - } - } - - const char* OnRequest(Request* request) - { - return NULL; - } - - - virtual ~ModuleHttpStats() - { - } - - virtual Version GetVersion() - { - return Version(1, 2, 0, 0, VF_VENDOR, API_VERSION); - } + void ReadConfig() + { + ConfigReader c(ServerInstance); + this->stylesheet = c.ReadValue("httpstats", "stylesheet", 0); + } + + ModuleHttpStats(InspIRCd* Me) : Module(Me) + { + ReadConfig(); + this->changed = true; + Implementation eventlist[] = { I_OnEvent, I_OnRequest }; + ServerInstance->Modules->Attach(eventlist, this, 2); + } + + std::string Sanitize(const std::string &str) + { + std::string ret; + ret.reserve(str.length() * 2); + + for (std::string::const_iterator x = str.begin(); x != str.end(); ++x) + { + std::map::const_iterator it = entities.find(*x); + if (it != entities.end()) + { + ret += '&'; + ret += it->second; + ret += ';'; + } + else if (*x < 32 || *x > 126) + { + int n = (unsigned char)*x; + ret += ("&#" + ConvToStr(n) + ";"); + } + else + { + ret += *x; + } + } + return ret; + } + + void OnEvent(Event* event) + { + std::stringstream data(""); + + if (event->GetEventID() == "httpd_url") + { + ServerInstance->Logs->Log("m_http_stats", DEBUG,"Handling httpd event"); + HTTPRequest* http = (HTTPRequest*)event->GetData(); + + if ((http->GetURI() == "/stats") || (http->GetURI() == "/stats/")) + { + data << ""; + + data << "" << ServerInstance->Config->ServerName << "" << Sanitize(ServerInstance->Config->ServerDesc) << ""; + + data << ""; + data << "" << ServerInstance->Users->clientlist->size() << ""; + data << "" << ServerInstance->chanlist->size() << ""; + data << "" << ServerInstance->Users->all_opers.size() << ""; + data << "" << (ServerInstance->SE->GetMaxFds() - ServerInstance->SE->GetRemainingFds()) << "" << ServerInstance->SE->GetMaxFds() << + "" << ServerInstance->SE->GetName() << ""; + + time_t current_time = 0; + current_time = ServerInstance->Time(); + time_t server_uptime = current_time - ServerInstance->startup_time; + struct tm* stime; + stime = gmtime(&server_uptime); + data << "" << stime->tm_yday << "" << stime->tm_hour << "" << stime->tm_min << "" << stime->tm_sec << "" << ServerInstance->startup_time << ""; + + + data << ""; + data << ""; + std::vector module_names = ServerInstance->Modules->GetAllModuleNames(0); + for (std::vector::iterator i = module_names.begin(); i != module_names.end(); ++i) + { + Module* m = ServerInstance->Modules->Find(i->c_str()); + Version v = m->GetVersion(); + data << "" << *i << "" << v.Major << "." << v.Minor << "." << v.Revision << "." << v.Build << ""; + } + data << ""; + data << ""; + + for (chan_hash::const_iterator a = ServerInstance->chanlist->begin(); a != ServerInstance->chanlist->end(); ++a) + { + Channel* c = a->second; + + data << ""; + data << "" << c->GetUsers()->size() << "" << c->name << ""; + data << "" << c->GetOppedUsers()->size() << ""; + data << "" << c->GetHalfoppedUsers()->size() << ""; + data << "" << c->GetVoicedUsers()->size() << ""; + data << "" << Sanitize(c->topic) << ""; + data << "" << Sanitize(c->ChanModes(true)) << ""; + CUList* ulist = c->GetUsers(); + + for (CUList::iterator x = ulist->begin(); x != ulist->end(); ++x) + { + data << "" << x->first->uuid << "" << Sanitize(c->GetAllPrefixChars(x->first)) << ""; + } + data << ""; + } + + data << ""; + + for (user_hash::const_iterator a = ServerInstance->Users->clientlist->begin(); a != ServerInstance->Users->clientlist->end(); ++a) + { + User* u = a->second; + + data << ""; + data << "" << u->nick << "" << u->uuid << "" << u->host << "" << u->dhost << ""; + data << "" << Sanitize(u->fullname) << "" << u->server << "" << Sanitize(u->awaymsg) << "" << Sanitize(u->oper) << ""; + std::string modes; + for (unsigned char n = 'A'; n <= 'z'; ++n) + if (u->IsModeSet(n)) + modes += n; + + data << modes << "" << Sanitize(u->ident) << "" << u->GetPort() << "" << u->GetIPString() << ""; + data << ""; + } + + data << ""; + + ProtoServerList sl; + ServerInstance->PI->GetServerList(sl); + + for (ProtoServerList::iterator b = sl.begin(); b != sl.end(); ++b) + { + data << ""; + data << "" << b->servername << ""; + data << "" << b->parentname << ""; + data << "" << b->usercount << ""; + data << "" << b->opercount << ""; + data << "" << b->latencyms << ""; + data << ""; + } + data << ""; + + data << ""; + + /* Send the document back to m_httpd */ + HTTPDocument response(http->sock, &data, 200); + response.headers.SetHeader("X-Powered-By", "m_httpd_stats.so"); + response.headers.SetHeader("Content-Type", "text/xml"); + Request req((char*)&response, (Module*)this, event->GetSource()); + req.Send(); + } + } + } + + const char* OnRequest(Request* request) + { + return NULL; + } + + + virtual ~ModuleHttpStats() + { + } + + virtual Version GetVersion() + { + return Version(1, 2, 0, 0, VF_VENDOR, API_VERSION); + } }; +static std::map const &init_entities() +{ + static std::map entities; + entities['<'] = "lt"; + entities['>'] = "gt"; + entities['&'] = "amp"; + entities['"'] = "quot"; + entities[246] = "ouml"; + entities[228] = "auml"; + entities[252] = "uuml"; + return entities; +} + +std::map const &ModuleHttpStats::entities = init_entities (); + MODULE_INIT(ModuleHttpStats) -- cgit v1.2.3