]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_hostchange.cpp
Merge insp20
[user/henk/code/inspircd.git] / src / modules / m_hostchange.cpp
index 354eea3ed3489659727c3224de596e1828885f71..6d5896ef51bd73ada8bedd4d875224030cc6a613 100644 (file)
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *               <Craig@chatspike.net>
- *     
- * Written by Craig Edwards, Craig McLure, and others.
- * This program is free but copyrighted software; see
- *            the file COPYING for details.
+ *   Copyright (C) 2005-2007 Craig Edwards <craigedwards@brainbox.cc>
+ *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
  *
- * ---------------------------------------------------
+ * This file is part of InspIRCd.  InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-using namespace std;
 
-#include <stdio.h>
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
 #include "inspircd.h"
 
-/* $ModDesc: Provides masking of user hostnames in a different way to m_cloaking */
-
-class Host : public classbase
+/** Holds information on a host set by m_hostchange
+ */
+class Host
 {
  public:
-       std::string action;
+       enum HostChangeAction
+       {
+               HCA_SET,
+               HCA_SUFFIX,
+               HCA_ADDNICK
+       };
+
+       HostChangeAction action;
        std::string newhost;
+       std::string ports;
+
+       Host(HostChangeAction Action, const std::string& Newhost, const std::string& Ports) :
+               action(Action), newhost(Newhost), ports(Ports) {}
 };
 
-typedef std::map<std::string,Host*> hostchanges_t;
+typedef std::vector<std::pair<std::string, Host> > hostchanges_t;
 
 class ModuleHostChange : public Module
 {
- private:
-
-       Server *Srv;
-       ConfigReader *Conf;
        hostchanges_t hostchanges;
        std::string MySuffix;
-        
+       std::string MyPrefix;
+       std::string MySeparator;
+
  public:
-       ModuleHostChange(Server* Me)
-               : Module::Module(Me)
-       {
-               Srv = Me;
-               Conf = new ConfigReader;
-               OnRehash("");
-       }
-       
-       virtual ~ModuleHostChange()
+       void ReadConfig(ConfigStatus& status) CXX11_OVERRIDE
        {
-               DELETE(Conf);
-       }
+               ConfigTag* host = ServerInstance->Config->ConfValue("host");
+               MySuffix = host->getString("suffix");
+               MyPrefix = host->getString("prefix");
+               MySeparator = host->getString("separator", ".");
+               hostchanges.clear();
 
-       Priority Prioritize()
-       {
-               return (Priority)Srv->PriorityAfter("m_cloaking.so");
-       }
+               std::set<std::string> dupecheck;
+               ConfigTagList tags = ServerInstance->Config->ConfTags("hostchange");
+               for (ConfigIter i = tags.first; i != tags.second; ++i)
+               {
+                       ConfigTag* tag = i->second;
+                       std::string mask = tag->getString("mask");
+                       if (!dupecheck.insert(mask).second)
+                               throw ModuleException("Duplicate hostchange entry: " + mask);
 
-       void Implements(char* List)
-       {
-               List[I_OnRehash] = List[I_OnUserConnect] = 1;
-       }
+                       Host::HostChangeAction act;
+                       std::string newhost;
+                       std::string action = tag->getString("action");
+                       if (!strcasecmp(action.c_str(), "set"))
+                       {
+                               act = Host::HCA_SET;
+                               newhost = tag->getString("value");
+                       }
+                       else if (!strcasecmp(action.c_str(), "suffix"))
+                               act = Host::HCA_SUFFIX;
+                       else if (!strcasecmp(action.c_str(), "addnick"))
+                               act = Host::HCA_ADDNICK;
+                       else
+                               throw ModuleException("Invalid hostchange action: " + action);
 
-       virtual void OnRehash(const std::string &parameter)
-       {
-               DELETE(Conf);
-               Conf = new ConfigReader;
-               MySuffix = Conf->ReadValue("host","suffix",0);
-               for (hostchanges_t::iterator i = hostchanges.begin(); i != hostchanges.end(); i++)
-               {
-                       DELETE(i->second);
-               }
-               hostchanges.clear();
-               for (int index = 0; index < Conf->Enumerate("hostchange"); index++)
-               {
-                       std::string mask = Conf->ReadValue("hostchange","mask",index);
-                       std::string action = Conf->ReadValue("hostchange","action",index);
-                       std::string newhost = Conf->ReadValue("hostchange","value",index);
-                       Host* x = new Host;
-                       x->action = action;
-                       x->newhost = newhost;
-                       hostchanges[mask] = x;
+                       hostchanges.push_back(std::make_pair(mask, Host(act, newhost, tag->getString("ports"))));
                }
        }
-       
-       virtual Version GetVersion()
+
+       Version GetVersion() CXX11_OVERRIDE
        {
                // returns the version number of the module to be
                // listed in /MODULES
-               return Version(1,0,0,1,VF_VENDOR);
+               return Version("Provides masking of user hostnames in a different way to m_cloaking", VF_VENDOR);
        }
-       
-       virtual void OnUserConnect(userrec* user)
+
+       void OnUserConnect(LocalUser* user) CXX11_OVERRIDE
        {
                for (hostchanges_t::iterator i = hostchanges.begin(); i != hostchanges.end(); i++)
                {
-                       if (Srv->MatchText(std::string(user->ident)+"@"+std::string(user->host),i->first))
+                       if (((InspIRCd::MatchCIDR(user->MakeHost(), i->first)) || (InspIRCd::MatchCIDR(user->MakeHostIP(), i->first))))
                        {
-                               Host* h = (Host*)i->second;
+                               const Host& h = i->second;
+
+                               if (!h.ports.empty())
+                               {
+                                       irc::portparser portrange(h.ports, false);
+                                       long portno = -1;
+                                       bool foundany = false;
+
+                                       while ((portno = portrange.GetToken()))
+                                               if (portno == user->GetServerPort())
+                                                       foundany = true;
+
+                                       if (!foundany)
+                                               continue;
+                               }
+
                                // host of new user matches a hostchange tag's mask
-                               std::string newhost = "";
-                               if (h->action == "set")
+                               std::string newhost;
+                               if (h.action == Host::HCA_SET)
                                {
-                                       newhost = h->newhost;
+                                       newhost = h.newhost;
                                }
-                               else if (h->action == "suffix")
+                               else if (h.action == Host::HCA_SUFFIX)
                                {
                                        newhost = MySuffix;
                                }
-                               else if (h->action == "addnick")
+                               else if (h.action == Host::HCA_ADDNICK)
                                {
                                        // first take their nick and strip out non-dns, leaving just [A-Z0-9\-]
-                                       std::string complete = "";
-                                       std::string old = user->nick;
-                                       for (unsigned int j = 0; j < old.length(); j++)
+                                       std::string complete;
+                                       for (std::string::const_iterator j = user->nick.begin(); j != user->nick.end(); ++j)
                                        {
-                                               if  (((old[j] >= 'A') && (old[j] <= 'Z')) ||
-                                                   ((old[j] >= 'a') && (old[j] <= 'z')) ||
-                                                   ((old[j] >= '0') && (old[j] <= '9')) ||
-                                                   (old[j] == '-'))
+                                               if  (((*j >= 'A') && (*j <= 'Z')) ||
+                                                   ((*j >= 'a') && (*j <= 'z')) ||
+                                                   ((*j >= '0') && (*j <= '9')) ||
+                                                   (*j == '-'))
                                                {
-                                                       complete = complete + old[j];
+                                                       complete = complete + *j;
                                                }
                                        }
-                                       if (complete == "")
+                                       if (complete.empty())
                                                complete = "i-have-a-lame-nick";
-                                       newhost = complete + "." + MySuffix;
+
+                                       if (!MyPrefix.empty())
+                                               newhost = MyPrefix + MySeparator + complete;
+                                       else
+                                               newhost = complete + MySeparator + MySuffix;
                                }
-                               if (newhost != "")
+                               if (!newhost.empty())
                                {
-                                       user->WriteServ("NOTICE "+std::string(user->nick)+" :Setting your virtual host: " + newhost);
-                                       if (!user->ChangeDisplayedHost(newhost.c_str()))
-                                               user->WriteServ("NOTICE "+std::string(user->nick)+" :Could not set your virtual host: " + newhost);
+                                       user->WriteNotice("Setting your virtual host: " + newhost);
+                                       if (!user->ChangeDisplayedHost(newhost))
+                                               user->WriteNotice("Could not set your virtual host: " + newhost);
                                        return;
                                }
                        }
@@ -143,29 +160,4 @@ class ModuleHostChange : public Module
        }
 };
 
-// stuff down here is the module-factory stuff. For basic modules you can ignore this.
-
-class ModuleHostChangeFactory : public ModuleFactory
-{
- public:
-       ModuleHostChangeFactory()
-       {
-       }
-       
-       ~ModuleHostChangeFactory()
-       {
-       }
-       
-       virtual Module * CreateModule(Server* Me)
-       {
-               return new ModuleHostChange(Me);
-       }
-       
-};
-
-
-extern "C" void * init_module( void )
-{
-       return new ModuleHostChangeFactory;
-}
-
+MODULE_INIT(ModuleHostChange)