]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_hostchange.cpp
Replace hardcoded mode letters, part 3
[user/henk/code/inspircd.git] / src / modules / m_hostchange.cpp
index f7ff58fa17aba57b79e9723f2b867c3a11ebb961..ac53e793c9e89d93269c0a945e11c2e231a5064d 100644 (file)
@@ -1 +1,170 @@
-/*       +------------------------------------+\r *       | Inspire Internet Relay Chat Daemon |\r *       +------------------------------------+\r *\r *  InspIRCd: (C) 2002-2007 InspIRCd Development Team\r * See: http://www.inspircd.org/wiki/index.php/Credits\r *\r * This program is free but copyrighted software; see\r *            the file COPYING for details.\r *\r * ---------------------------------------------------\r */\r\r#include "inspircd.h"\r#include "users.h"\r#include "channels.h"\r#include "modules.h"\r\r/* $ModDesc: Provides masking of user hostnames in a different way to m_cloaking */\r\r/** Holds information on a host set by m_hostchange\r */\rclass Host : public classbase\r{\r public:\r        std::string action;\r    std::string newhost;\r};\r\rtypedef std::map<std::string,Host*> hostchanges_t;\r\rclass ModuleHostChange : public Module\r{\r private:\r        hostchanges_t hostchanges;\r     std::string MySuffix;\r  std::string MyPrefix;\r  std::string MySeparator;\r        \r public:\r     ModuleHostChange(InspIRCd* Me)\r         : Module(Me)\r   {\r              OnRehash(NULL,"");\r     }\r      \r       virtual ~ModuleHostChange()\r    {\r              for (hostchanges_t::iterator i = hostchanges.begin(); i != hostchanges.end(); i++)\r             {\r                      DELETE(i->second);\r             }\r              hostchanges.clear();\r   }\r\r     Priority Prioritize()\r  {\r              return (Priority)ServerInstance->PriorityAfter("m_cloaking.so");\r       }\r\r     void Implements(char* List)\r    {\r              List[I_OnRehash] = List[I_OnUserConnect] = 1;\r  }\r\r     virtual void OnRehash(userrec* user, const std::string &parameter)\r     {\r              ConfigReader Conf(ServerInstance);\r             MySuffix = Conf.ReadValue("host","suffix",0);\r          MyPrefix = Conf.ReadValue("host","prefix","",0);\r               MySeparator = Conf.ReadValue("host","separator",".",0);\r                for (hostchanges_t::iterator i = hostchanges.begin(); i != hostchanges.end(); i++)\r             {\r                      DELETE(i->second);\r             }\r              hostchanges.clear();\r           for (int index = 0; index < Conf.Enumerate("hostchange"); index++)\r             {\r                      std::string mask = Conf.ReadValue("hostchange","mask",index);\r                  std::string action = Conf.ReadValue("hostchange","action",index);\r                      std::string newhost = Conf.ReadValue("hostchange","value",index);\r                      Host* x = new Host;\r                    x->action = action;\r                    x->newhost = newhost;\r                  hostchanges[mask] = x;\r         }\r      }\r      \r       virtual Version GetVersion()\r   {\r              // returns the version number of the module to be\r              // listed in /MODULES\r          return Version(1,1,0,1,VF_VENDOR,API_VERSION);\r }\r      \r       virtual void OnUserConnect(userrec* user)\r      {\r              for (hostchanges_t::iterator i = hostchanges.begin(); i != hostchanges.end(); i++)\r             {\r                      if (ServerInstance->MatchText(std::string(user->ident)+"@"+std::string(user->host),i->first))\r                  {\r                              Host* h = (Host*)i->second;\r                            // host of new user matches a hostchange tag's mask\r                            std::string newhost;\r                           if (h->action == "set")\r                                {\r                                      newhost = h->newhost;\r                          }\r                              else if (h->action == "suffix")\r                                {\r                                      newhost = MySuffix;\r                            }\r                              else if (h->action == "addnick")\r                               {\r                                      // first take their nick and strip out non-dns, leaving just [A-Z0-9\-]\r                                        std::string complete;\r                                  std::string old = user->nick;\r                                  for (unsigned int j = 0; j < old.length(); j++)\r                                        {\r                                              if  (((old[j] >= 'A') && (old[j] <= 'Z')) ||\r                                               ((old[j] >= 'a') && (old[j] <= 'z')) ||\r                                                ((old[j] >= '0') && (old[j] <= '9')) ||\r                                                (old[j] == '-'))\r                                           {\r                                                      complete = complete + old[j];\r                                          }\r                                      }\r                                      if (complete.empty())\r                                          complete = "i-have-a-lame-nick";\r                                               \r                                       if (!MyPrefix.empty())\r                                         newhost = MyPrefix + MySeparator + complete;\r                                   else\r                                           newhost = complete + MySeparator + MySuffix;\r                           }\r                              if (!newhost.empty())\r                          {\r                                      user->WriteServ("NOTICE "+std::string(user->nick)+" :Setting your virtual host: " + newhost);\r                                  if (!user->ChangeDisplayedHost(newhost.c_str()))\r                                               user->WriteServ("NOTICE "+std::string(user->nick)+" :Could not set your virtual host: " + newhost);\r                                    return;\r                                }\r                      }\r              }\r      }\r};\r\rMODULE_INIT(ModuleHostChange)\r
\ No newline at end of file
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
+ *
+ *   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/>.
+ */
+
+
+#include "inspircd.h"
+
+/** Holds information on a host set by m_hostchange
+ */
+class Host
+{
+ public:
+       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::vector<std::pair<std::string, Host> > hostchanges_t;
+
+class ModuleHostChange : public Module
+{
+       hostchanges_t hostchanges;
+       std::string MySuffix;
+       std::string MyPrefix;
+       std::string MySeparator;
+
+ public:
+       void init() CXX11_OVERRIDE
+       {
+               OnRehash(NULL);
+               Implementation eventlist[] = { I_OnRehash, I_OnUserConnect };
+               ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
+       }
+
+       void OnRehash(User* user) CXX11_OVERRIDE
+       {
+               ConfigTag* host = ServerInstance->Config->ConfValue("host");
+               MySuffix = host->getString("suffix");
+               MyPrefix = host->getString("prefix");
+               MySeparator = host->getString("separator", ".");
+               hostchanges.clear();
+
+               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);
+
+                       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);
+
+                       hostchanges.push_back(std::make_pair(mask, Host(act, newhost, tag->getString("ports"))));
+               }
+       }
+
+       Version GetVersion() CXX11_OVERRIDE
+       {
+               // returns the version number of the module to be
+               // listed in /MODULES
+               return Version("Provides masking of user hostnames in a different way to m_cloaking", VF_VENDOR);
+       }
+
+       void OnUserConnect(LocalUser* user) CXX11_OVERRIDE
+       {
+               for (hostchanges_t::iterator i = hostchanges.begin(); i != hostchanges.end(); i++)
+               {
+                       if (((InspIRCd::MatchCIDR(user->MakeHost(), i->first)) || (InspIRCd::MatchCIDR(user->MakeHostIP(), i->first))))
+                       {
+                               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 == Host::HCA_SET)
+                               {
+                                       newhost = h.newhost;
+                               }
+                               else if (h.action == Host::HCA_SUFFIX)
+                               {
+                                       newhost = MySuffix;
+                               }
+                               else if (h.action == Host::HCA_ADDNICK)
+                               {
+                                       // first take their nick and strip out non-dns, leaving just [A-Z0-9\-]
+                                       std::string complete;
+                                       for (std::string::const_iterator j = user->nick.begin(); j != user->nick.end(); ++j)
+                                       {
+                                               if  (((*j >= 'A') && (*j <= 'Z')) ||
+                                                   ((*j >= 'a') && (*j <= 'z')) ||
+                                                   ((*j >= '0') && (*j <= '9')) ||
+                                                   (*j == '-'))
+                                               {
+                                                       complete = complete + *j;
+                                               }
+                                       }
+                                       if (complete.empty())
+                                               complete = "i-have-a-lame-nick";
+
+                                       if (!MyPrefix.empty())
+                                               newhost = MyPrefix + MySeparator + complete;
+                                       else
+                                               newhost = complete + MySeparator + MySuffix;
+                               }
+                               if (!newhost.empty())
+                               {
+                                       user->WriteNotice("Setting your virtual host: " + newhost);
+                                       if (!user->ChangeDisplayedHost(newhost.c_str()))
+                                               user->WriteNotice("Could not set your virtual host: " + newhost);
+                                       return;
+                               }
+                       }
+               }
+       }
+};
+
+MODULE_INIT(ModuleHostChange)