X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_xmlsocket.cpp;h=0e29cb99347432d81a47007bdc15f9caba4e5ef4;hb=31b232eec07d0b85ce2cc7c4ebc5d5387f781dcb;hp=12da3762ef01527edaf175eb75c1afa1d4c2a53f;hpb=bab14f0dd2345c9d7dcbc47c918563709e1ac094;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/modules/m_xmlsocket.cpp b/src/modules/m_xmlsocket.cpp index 12da3762e..0e29cb993 100644 --- a/src/modules/m_xmlsocket.cpp +++ b/src/modules/m_xmlsocket.cpp @@ -1 +1,182 @@ -/* +------------------------------------+ * | 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 "users.h" #include "channels.h" #include "modules.h" #include "hashcomp.h" /* $ModDesc: Provides XMLSocket support for clients */ class ModuleXMLSocket : public Module { ConfigReader* Conf; std::vector listenports; public: ModuleXMLSocket(InspIRCd* Me) : Module(Me) { OnRehash(NULL,""); } virtual void OnRehash(userrec* user, const std::string ¶m) { Conf = new ConfigReader(ServerInstance); for (unsigned int i = 0; i < listenports.size(); i++) { ServerInstance->Config->DelIOHook(listenports[i]); } listenports.clear(); for (int i = 0; i < Conf->Enumerate("bind"); i++) { // For each tag std::string x = Conf->ReadValue("bind", "type", i); if (((x.empty()) || (x == "clients")) && (Conf->ReadFlag("bind", "xmlsocket", i))) { // Get the port we're meant to be listening on with SSL std::string port = Conf->ReadValue("bind", "port", i); irc::portparser portrange(port, false); long portno = -1; while ((portno = portrange.GetToken())) { try { if (ServerInstance->Config->AddIOHook(portno, this)) { listenports.push_back(portno); for (size_t i = 0; i < ServerInstance->Config->ports.size(); i++) if (ServerInstance->Config->ports[i]->GetPort() == portno) ServerInstance->Config->ports[i]->SetDescription("xml"); } else { ServerInstance->Log(DEFAULT, "m_xmlsocket.so: FAILED to enable XMLSocket on port %d, maybe you have another similar module loaded?", portno); } } catch (ModuleException &e) { ServerInstance->Log(DEFAULT, "m_xmlsocket.so: FAILED to enable XMLSocket on port %d: %s. Maybe it's already hooked by the same port on a different IP, or you have another similar module loaded?", portno, e.GetReason()); } } } } DELETE(Conf); } virtual ~ModuleXMLSocket() { } virtual void OnUnloadModule(Module* mod, const std::string &name) { if (mod == this) { for(unsigned int i = 0; i < listenports.size(); i++) { ServerInstance->Config->DelIOHook(listenports[i]); for (size_t j = 0; j < ServerInstance->Config->ports.size(); j++) if (ServerInstance->Config->ports[j]->GetPort() == listenports[i]) ServerInstance->Config->ports[j]->SetDescription("plaintext"); } } } virtual Version GetVersion() { return Version(1, 1, 0, 0, VF_VENDOR, API_VERSION); } void Implements(char* List) { List[I_OnUnloadModule] = List[I_OnRawSocketRead] = List[I_OnRawSocketWrite] = List[I_OnRehash] = 1; } virtual int OnRawSocketRead(int fd, char* buffer, unsigned int count, int &readresult) { userrec* user = dynamic_cast(ServerInstance->FindDescriptor(fd)); if (user == NULL) return -1; int result = user->ReadData(buffer, count); if ((result == -1) && (errno == EAGAIN)) return -1; else if (result < 1) return 0; /* XXX: The core is more than happy to split lines purely on an \n * rather than a \r\n. This is good for us as it means that the size * of data we are receiving is exactly the same as the size of data * we asked for, and we dont need to re-implement our own socket * buffering (See below) */ for (int n = 0; n < result; n++) if (buffer[n] == 0) buffer[n] = '\n'; readresult = result; return result; } virtual int OnRawSocketWrite(int fd, const char* buffer, int count) { userrec* user = dynamic_cast(ServerInstance->FindDescriptor(fd)); if (user == NULL) return -1; /* We want to alter the buffer, so we have to make a copy */ char * tmpbuffer = new char[count + 1]; memcpy(tmpbuffer, buffer, count); /* XXX: This will actually generate lines "looking\0\0like\0\0this" * rather than lines "looking\0like\0this". This shouldnt be a problem * to the client, but it saves us a TON of processing and the need * to re-implement socket buffering, as the data we are sending is * exactly the same length as the data we are receiving. */ for (int n = 0; n < count; n++) if ((tmpbuffer[n] == '\r') || (tmpbuffer[n] == '\n')) tmpbuffer[n] = 0; user->AddWriteBuf(std::string(tmpbuffer,count)); delete [] tmpbuffer; return 1; } }; MODULE_INIT(ModuleXMLSocket) \ No newline at end of file +/* +------------------------------------+ + * | Inspire Internet Relay Chat Daemon | + * +------------------------------------+ + * + * InspIRCd: (C) 2002-2008 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" + +/* $ModDesc: Provides XMLSocket support for clients */ + +class ModuleXMLSocket : public Module +{ + ConfigReader* Conf; + std::vector listenports; + + public: + + ModuleXMLSocket(InspIRCd* Me) + : Module(Me) + { + OnRehash(NULL,""); + Implementation eventlist[] = { I_OnUnloadModule, I_OnRawSocketRead, I_OnRawSocketWrite, I_OnRehash, I_OnHookUserIO, I_OnCleanup }; + ServerInstance->Modules->Attach(eventlist, this, 6); + } + + bool isin(const std::string &host, int port, const std::vector &portlist) + { + if (std::find(portlist.begin(), portlist.end(), "*:" + ConvToStr(port)) != portlist.end()) + return true; + + if (std::find(portlist.begin(), portlist.end(), ":" + ConvToStr(port)) != portlist.end()) + return true; + + return std::find(portlist.begin(), portlist.end(), host + ":" + ConvToStr(port)) != portlist.end(); + } + + virtual void OnRehash(User* user, const std::string ¶m) + { + + Conf = new ConfigReader(ServerInstance); + + listenports.clear(); + + for (int i = 0; i < Conf->Enumerate("bind"); i++) + { + // For each tag + std::string x = Conf->ReadValue("bind", "type", i); + if (((x.empty()) || (x == "clients")) && (Conf->ReadFlag("bind", "xmlsocket", i))) + { + // Get the port we're meant to be listening on with SSL + std::string port = Conf->ReadValue("bind", "port", i); + std::string addr = Conf->ReadValue("bind", "address", i); + + irc::portparser portrange(port, false); + long portno = -1; + while ((portno = portrange.GetToken())) + { + try + { + listenports.push_back(addr + ":" + ConvToStr(portno)); + for (size_t j = 0; j < ServerInstance->Config->ports.size(); j++) + if ((ServerInstance->Config->ports[j]->GetPort() == portno) && (ServerInstance->Config->ports[j]->GetIP() == addr)) + ServerInstance->Config->ports[j]->SetDescription("xml"); + } + catch (ModuleException &e) + { + ServerInstance->Logs->Log("m_xmlsocket",DEFAULT, "m_xmlsocket.so: FAILED to enable XMLSocket on port %ld: %s. Maybe you have another similar module loaded?", portno, e.GetReason()); + } + } + } + } + + delete Conf; + } + + virtual ~ModuleXMLSocket() + { + } + + virtual void OnUnloadModule(Module* mod, const std::string &name) + { + if (mod == this) + { + for(unsigned int i = 0; i < listenports.size(); i++) + { + for (size_t j = 0; j < ServerInstance->Config->ports.size(); j++) + if (listenports[i] == (ServerInstance->Config->ports[j]->GetIP()+":"+ConvToStr(ServerInstance->Config->ports[j]->GetPort()))) + ServerInstance->Config->ports[j]->SetDescription("plaintext"); + } + } + } + + virtual void OnCleanup(int target_type, void* item) + { + if(target_type == TYPE_USER) + { + User* user = (User*)item; + if(user->io == this) + user->io = NULL; + } + } + + virtual Version GetVersion() + { + return Version(1, 2, 0, 0, VF_VENDOR, API_VERSION); + } + + virtual void OnHookUserIO(User* user, const std::string &targetip) + { + if (!user->io && isin(targetip,user->GetPort(),listenports)) + { + /* Hook the user with our module */ + user->io = this; + } + } + + virtual int OnRawSocketRead(int fd, char* buffer, unsigned int count, int &readresult) + { + User* user = dynamic_cast(ServerInstance->FindDescriptor(fd)); + + if (user == NULL) + return -1; + + int result = user->ReadData(buffer, count); + + if ((result == -1) && (errno == EAGAIN)) + return -1; + else if (result < 1) + return 0; + + /* XXX: The core is more than happy to split lines purely on an \n + * rather than a \r\n. This is good for us as it means that the size + * of data we are receiving is exactly the same as the size of data + * we asked for, and we dont need to re-implement our own socket + * buffering (See below) + */ + for (int n = 0; n < result; n++) + if (buffer[n] == 0) + buffer[n] = '\n'; + + readresult = result; + return result; + } + + virtual int OnRawSocketWrite(int fd, const char* buffer, int count) + { + User* user = dynamic_cast(ServerInstance->FindDescriptor(fd)); + + if (user == NULL) + return -1; + + /* We want to alter the buffer, so we have to make a copy */ + char * tmpbuffer = new char[count + 1]; + memcpy(tmpbuffer, buffer, count); + + /* XXX: This will actually generate lines "looking\0\0like\0\0this" + * rather than lines "looking\0like\0this". This shouldnt be a problem + * to the client, but it saves us a TON of processing and the need + * to re-implement socket buffering, as the data we are sending is + * exactly the same length as the data we are receiving. + */ + for (int n = 0; n < count; n++) + if ((tmpbuffer[n] == '\r') || (tmpbuffer[n] == '\n')) + tmpbuffer[n] = 0; + + user->AddWriteBuf(std::string(tmpbuffer,count)); + delete [] tmpbuffer; + + return 1; + } + +}; + +MODULE_INIT(ModuleXMLSocket) +