From d1b20f95198858c0ffd2974fb7fd622a0d945a81 Mon Sep 17 00:00:00 2001 From: brain Date: Mon, 24 Mar 2008 16:24:52 +0000 Subject: This module uses io hooking too, fix it up git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@9190 e03df62e-2008-0410-955e-edbf42e46eb7 --- src/configreader.cpp | 31 ------------------------- src/modules/m_xmlsocket.cpp | 55 +++++++++++++++++++++++++++------------------ 2 files changed, 33 insertions(+), 53 deletions(-) (limited to 'src') diff --git a/src/configreader.cpp b/src/configreader.cpp index 83b7e14d9..cba015ec9 100644 --- a/src/configreader.cpp +++ b/src/configreader.cpp @@ -63,32 +63,12 @@ void ServerConfig::ClearStack() include_stack.clear(); } -Module* ServerConfig::GetIOHook(int port) -{ - std::map::iterator x = IOHookModule.find(port); - return (x != IOHookModule.end() ? x->second : NULL); -} - Module* ServerConfig::GetIOHook(BufferedSocket* is) { std::map::iterator x = SocketIOHookModule.find(is); return (x != SocketIOHookModule.end() ? x->second : NULL); } -bool ServerConfig::AddIOHook(int port, Module* iomod) -{ - if (!GetIOHook(port)) - { - IOHookModule[port] = iomod; - return true; - } - else - { - throw ModuleException("Port already hooked by another module"); - return false; - } -} - bool ServerConfig::AddIOHook(Module* iomod, BufferedSocket* is) { if (!GetIOHook(is)) @@ -104,17 +84,6 @@ bool ServerConfig::AddIOHook(Module* iomod, BufferedSocket* is) } } -bool ServerConfig::DelIOHook(int port) -{ - std::map::iterator x = IOHookModule.find(port); - if (x != IOHookModule.end()) - { - IOHookModule.erase(x); - return true; - } - return false; -} - bool ServerConfig::DelIOHook(BufferedSocket* is) { std::map::iterator x = SocketIOHookModule.find(is); diff --git a/src/modules/m_xmlsocket.cpp b/src/modules/m_xmlsocket.cpp index cfe22bca9..2e4b17fc6 100644 --- a/src/modules/m_xmlsocket.cpp +++ b/src/modules/m_xmlsocket.cpp @@ -18,7 +18,7 @@ class ModuleXMLSocket : public Module { ConfigReader* Conf; - std::vector listenports; + std::vector listenports; public: @@ -26,8 +26,13 @@ class ModuleXMLSocket : public Module : Module(Me) { OnRehash(NULL,""); - Implementation eventlist[] = { I_OnUnloadModule, I_OnRawSocketRead, I_OnRawSocketWrite, I_OnRehash }; - ServerInstance->Modules->Attach(eventlist, this, 4); + 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 &hostandport, const std::vector &portlist) + { + return std::find(portlist.begin(), portlist.end(), hostandport) != portlist.end(); } virtual void OnRehash(User* user, const std::string ¶m) @@ -35,11 +40,6 @@ class ModuleXMLSocket : public Module 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++) @@ -50,27 +50,21 @@ class ModuleXMLSocket : public Module { // 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 { - if (ServerInstance->Config->AddIOHook(portno, this)) - { - listenports.push_back(portno); - for (size_t j = 0; j < ServerInstance->Config->ports.size(); j++) - if (ServerInstance->Config->ports[j]->GetPort() == portno) - ServerInstance->Config->ports[j]->SetDescription("xml"); - } - else - { - ServerInstance->Logs->Log("m_xmlsocket",DEFAULT, "m_xmlsocket.so: FAILED to enable XMLSocket on port %d, maybe you have another similar module loaded?", portno); - } + listenports.push_back(addr + ":" + ConvToStr(portno)); + for (size_t j = 0; j < ServerInstance->Config->ports.size(); j++) + if ((ServerInstance->Config->ports[i]->GetPort() == portno) && (ServerInstance->Config->ports[i]->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 %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()); + ServerInstance->Logs->Log("m_xmlsocket",DEFAULT, "m_xmlsocket.so: FAILED to enable XMLSocket on port %d: %s. Maybe you have another similar module loaded?", portno, e.GetReason()); } } } @@ -89,19 +83,36 @@ class ModuleXMLSocket : public Module { 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]) + 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+":"+ConvToStr(user->GetPort()),listenports)) + { + /* Hook the user with our module */ + user->io = this; + } + } virtual int OnRawSocketRead(int fd, char* buffer, unsigned int count, int &readresult) { -- cgit v1.2.3