]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_cloaking.cpp
Get rid of a bunch of memory-wasting C-style strings
[user/henk/code/inspircd.git] / src / modules / m_cloaking.cpp
index 7a500c1b30695b01570cc6960c64f0de0beaa75d..2bd8df80eea37f9dc6b3126e44c85cd0ba8e0b31 100644 (file)
@@ -2,8 +2,8 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
+ *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
  *
  * This program is free but copyrighted software; see
  *         the file COPYING for details.
@@ -12,7 +12,6 @@
  */
 
 #include "inspircd.h"
-#include "wildcard.h"
 #include "m_hash.h"
 
 /* $ModDesc: Provides masking of user hostnames */
@@ -29,9 +28,9 @@ class CloakUser : public ModeHandler
        unsigned int key3;
        unsigned int key4;
        bool ipalways;
-       Module* Sender;
        Module* HashProvider;
        const char *xtab[4];
+       LocalStringExt ext;
 
        /** This function takes a domain name string and returns just the last two domain parts,
         * or the last domain part if only two are available. Failing that it just returns what it was given.
@@ -39,7 +38,7 @@ class CloakUser : public ModeHandler
         * For example, if it is passed "svn.inspircd.org" it will return ".inspircd.org".
         * If it is passed "brainbox.winbot.co.uk" it will return ".co.uk",
         * and if it is passed "localhost.localdomain" it will return ".localdomain".
-        * 
+        *
         * This is used to ensure a significant part of the host is always cloaked (see Bug #216)
         */
        std::string LastTwoDomainParts(const std::string &host)
@@ -64,15 +63,14 @@ class CloakUser : public ModeHandler
                        return host.substr(splitdot);
        }
 
-       CloakUser(InspIRCd* Instance, Module* source, Module* Hash) : ModeHandler(Instance, 'x', 0, 0, false, MODETYPE_USER, false), Sender(source), HashProvider(Hash)
+       CloakUser(Module* source, Module* Hash)
+               : ModeHandler(source, 'x', PARAM_NONE, MODETYPE_USER), HashProvider(Hash),
+               ext("cloaked_host", source)
        {
        }
 
-       ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding, bool)
+       ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
        {
-               if (source != dest)
-                       return MODEACTION_DENY;
-
                /* For remote clients, we dont take any action, we just allow it.
                 * The local server where they are will set their cloak instead.
                 * This is fine, as we will recieve it later.
@@ -99,11 +97,16 @@ class CloakUser : public ModeHandler
                                 * are connecting via localhost) -- this doesnt matter much.
                                 */
 
-                               std::string* cloak;
+                               std::string* cloak = ext.get(dest);
 
-                               if (dest->GetExt("cloaked_host", cloak))
+                               if (!cloak)
+                               {
+                                       /* Force creation of missing cloak */
+                                       creator->OnUserConnect(dest);
+                                       cloak = ext.get(dest);
+                               }
+                               if (cloak)
                                {
-                                       /* Cloaked host has been set before on this user, don't bother to recalculate and waste cpu */
                                        dest->ChangeDisplayedHost(cloak->c_str());
                                        dest->SetMode('x',true);
                                        return MODEACTION_ALLOW;
@@ -111,13 +114,13 @@ class CloakUser : public ModeHandler
                        }
                }
                else
-               {
+               {
                        if (dest->IsModeSet('x'))
                        {
-                               /* User is removing the mode, so just restore their real host
-                                * and make it match the displayed one.
+                               /* User is removing the mode, so just restore their real host
+                                * and make it match the displayed one.
                                 */
-                               dest->ChangeDisplayedHost(dest->host);
+                               dest->ChangeDisplayedHost(dest->host.c_str());
                                dest->SetMode('x',false);
                                return MODEACTION_ALLOW;
                        }
@@ -145,14 +148,14 @@ class CloakUser : public ModeHandler
                octet[1] = octet[0] + "." + octet[1];
 
                /* Reset the Hash module and send it our IV */
-               HashResetRequest(Sender, HashProvider).Send();
-               HashKeyRequest(Sender, HashProvider, iv).Send();
+               HashResetRequest(creator, HashProvider).Send();
+               HashKeyRequest(creator, HashProvider, iv).Send();
 
                /* Send the Hash module a different hex table for each octet group's Hash sum */
                for (int k = 0; k < 4; k++)
                {
-                       HashHexRequest(Sender, HashProvider, xtab[(iv[k]+i[k]) % 4]).Send();
-                       ra[k] = std::string(HashSumRequest(Sender, HashProvider, octet[k]).Send()).substr(0,6);
+                       HashHexRequest(creator, HashProvider, xtab[(iv[k]+i[k]) % 4]).Send();
+                       ra[k] = std::string(HashSumRequest(creator, HashProvider, octet[k]).Send()).substr(0,6);
                }
                /* Stick them all together */
                return std::string().append(ra[0]).append(".").append(ra[1]).append(".").append(ra[2]).append(".").append(ra[3]);
@@ -160,19 +163,14 @@ class CloakUser : public ModeHandler
 
        std::string Cloak6(const char* ip)
        {
-               /* Theyre using 4in6 (YUCK). Translate as ipv4 cloak */
-               if (!strncmp(ip, "0::ffff:", 8))
-                       return Cloak4(ip + 8);
-
-               /* If we get here, yes it really is an ipv6 ip */
                unsigned int iv[] = { key1, key2, key3, key4 };
                std::vector<std::string> hashies;
                std::string item;
                int rounds = 0;
 
                /* Reset the Hash module and send it our IV */
-               HashResetRequest(Sender, HashProvider).Send();
-               HashKeyRequest(Sender, HashProvider, iv).Send();
+               HashResetRequest(creator, HashProvider).Send();
+               HashKeyRequest(creator, HashProvider, iv).Send();
 
                for (const char* input = ip; *input; input++)
                {
@@ -180,8 +178,8 @@ class CloakUser : public ModeHandler
                        if (item.length() > 7)
                        {
                                /* Send the Hash module a different hex table for each octet group's Hash sum */
-                               HashHexRequest(Sender, HashProvider, xtab[(key1+rounds) % 4]).Send();
-                               hashies.push_back(std::string(HashSumRequest(Sender, HashProvider, item).Send()).substr(0,8));
+                               HashHexRequest(creator, HashProvider, xtab[(key1+rounds) % 4]).Send();
+                               hashies.push_back(std::string(HashSumRequest(creator, HashProvider, item).Send()).substr(0,8));
                                item.clear();
                        }
                        rounds++;
@@ -189,20 +187,20 @@ class CloakUser : public ModeHandler
                if (!item.empty())
                {
                        /* Send the Hash module a different hex table for each octet group's Hash sum */
-                       HashHexRequest(Sender, HashProvider, xtab[(key1+rounds) % 4]).Send();
-                       hashies.push_back(std::string(HashSumRequest(Sender, HashProvider, item).Send()).substr(0,8));
+                       HashHexRequest(creator, HashProvider, xtab[(key1+rounds) % 4]).Send();
+                       hashies.push_back(std::string(HashSumRequest(creator, HashProvider, item).Send()).substr(0,8));
                        item.clear();
                }
                /* Stick them all together */
                return irc::stringjoiner(":", hashies, 0, hashies.size() - 1).GetJoined();
        }
-       
+
        void DoRehash()
        {
-               ConfigReader Conf(ServerInstance);
+               ConfigReader Conf;
                bool lowercase;
-               
-               /* These are *not* using the need_positive parameter of ReadInteger - 
+
+               /* These are *not* using the need_positive parameter of ReadInteger -
                 * that will limit the valid values to only the positive values in a
                 * signed int. Instead, accept any value that fits into an int and
                 * cast it to an unsigned int. That will, a bit oddly, give us the full
@@ -215,7 +213,7 @@ class CloakUser : public ModeHandler
                prefix = Conf.ReadValue("cloak","prefix",0);
                ipalways = Conf.ReadFlag("cloak", "ipalways", 0);
                lowercase = Conf.ReadFlag("cloak", "lowercase", 0);
-               
+
                if (!lowercase)
                {
                        xtab[0] = "F92E45D871BCA630";
@@ -255,24 +253,21 @@ class CloakUser : public ModeHandler
 class ModuleCloaking : public Module
 {
  private:
-       
        CloakUser* cu;
-       Module* HashModule;
 
  public:
-       ModuleCloaking(InspIRCd* Me)
-               : Module(Me)
+       ModuleCloaking()
        {
                /* Attempt to locate the md5 service provider, bail if we can't find it */
-               HashModule = ServerInstance->Modules->Find("m_md5.so");
+               Module* HashModule = ServerInstance->Modules->Find("m_md5.so");
                if (!HashModule)
                        throw ModuleException("Can't find m_md5.so. Please load m_md5.so before m_cloaking.so.");
 
-               cu = new CloakUser(ServerInstance, this, HashModule);
+               cu = new CloakUser(this, HashModule);
 
                try
                {
-                       OnRehash(NULL,"");
+                       OnRehash(NULL);
                }
                catch (ModuleException &e)
                {
@@ -288,83 +283,77 @@ class ModuleCloaking : public Module
                }
 
                ServerInstance->Modules->UseInterface("HashRequest");
+               Extensible::Register(&cu->ext);
+
+               Implementation eventlist[] = { I_OnRehash, I_OnCheckBan, I_OnUserConnect };
+               ServerInstance->Modules->Attach(eventlist, this, 3);
 
-               Implementation eventlist[] = { I_OnRehash, I_OnUserDisconnect, I_OnCleanup, I_OnCheckBan, I_OnUserConnect, I_OnSyncUserMetaData };
-               ServerInstance->Modules->Attach(eventlist, this, 6);
+               CloakExistingUsers();
        }
 
-       void OnSyncUserMetaData(User* user, Module* proto,void* opaque, const std::string &extname, bool displayable)
+       void CloakExistingUsers()
        {
-               if ((displayable) && (extname == "cloaked_host"))
+               std::string* cloak;
+               for (std::vector<User*>::iterator u = ServerInstance->Users->local_users.begin(); u != ServerInstance->Users->local_users.end(); u++)
                {
-                       std::string* cloak;
-                       if (user->GetExt("cloaked_host", cloak))
-                               proto->ProtoSendMetaData(opaque, TYPE_USER, user, extname, *cloak);
+                       cloak = cu->ext.get(*u);
+                       if (!cloak)
+                       {
+                               OnUserConnect(*u);
+                       }
                }
        }
 
-
-       virtual int OnCheckBan(User* user, Channel* chan)
+       ModResult OnCheckBan(User* user, Channel* chan, const std::string& mask)
        {
-               char mask[MAXBUF];
-               std::string* tofree;
+               char cmask[MAXBUF];
+               std::string* cloak = cu->ext.get(user);
                /* Check if they have a cloaked host, but are not using it */
-               if (user->GetExt("cloaked_host", tofree) && *tofree != user->dhost)
+               if (cloak && *cloak != user->dhost)
                {
-                       snprintf(mask, MAXBUF, "%s!%s@%s", user->nick, user->ident, tofree->c_str());
-                       for (BanList::iterator i = chan->bans.begin(); i != chan->bans.end(); i++)
-                       {
-                               if (match(mask,i->data))
-                                       return -1;
-                       }
+                       snprintf(cmask, MAXBUF, "%s!%s@%s", user->nick.c_str(), user->ident.c_str(), cloak->c_str());
+                       if (InspIRCd::Match(cmask,mask))
+                               return MOD_RES_DENY;
                }
-               return 0;
+               return MOD_RES_PASSTHRU;
        }
 
        void Prioritize()
        {
                /* Needs to be after m_banexception etc. */
-               ServerInstance->Modules->SetPriority(this, I_OnCheckBan, PRIO_LAST);
-       }
+               ServerInstance->Modules->SetPriority(this, I_OnCheckBan, PRIORITY_LAST);
 
-       virtual void OnUserDisconnect(User* user)
-       {
-               std::string* tofree;
-               if (user->GetExt("cloaked_host", tofree))
-                       delete tofree;
+               /* but before m_conn_umodes, so host is generated ready to apply */
+               Module *um = ServerInstance->Modules->Find("m_conn_umodes.so");
+               ServerInstance->Modules->SetPriority(this, I_OnUserConnect, PRIORITY_AFTER, &um);
        }
 
-       virtual void OnCleanup(int target_type, void* item)
-       {
-               if (target_type == TYPE_USER)
-                       OnUserDisconnect((User*)item);
-       }
-       
-       virtual ~ModuleCloaking()
+       ~ModuleCloaking()
        {
                ServerInstance->Modes->DelMode(cu);
                delete cu;
                ServerInstance->Modules->DoneWithInterface("HashRequest");
        }
-       
-       virtual Version GetVersion()
+
+       Version GetVersion()
        {
                // returns the version number of the module to be
                // listed in /MODULES
-               return Version(1,2,0,2,VF_COMMON|VF_VENDOR,API_VERSION);
+               return Version("Provides masking of user hostnames", VF_COMMON|VF_VENDOR,API_VERSION);
        }
 
-       virtual void OnRehash(User* user, const std::string &parameter)
+       void OnRehash(User* user)
        {
                cu->DoRehash();
        }
 
-       virtual void OnUserConnect(User* dest)
+       void OnUserConnect(User* dest)
        {
-               char* n1 = strchr(dest->host,'.');
-               char* n2 = strchr(dest->host,':');
+               std::string* cloak = cu->ext.get(dest);
+               if (cloak)
+                       return;
 
-               if (n1 || n2)
+               if (dest->host.find('.') != std::string::npos || dest->host.find(':') != std::string::npos)
                {
                        unsigned int iv[] = { cu->key1, cu->key2, cu->key3, cu->key4 };
                        std::string a = cu->LastTwoDomainParts(dest->host);
@@ -384,49 +373,35 @@ class ModuleCloaking : public Module
                                /** Reset the Hash module, and send it our IV and hex table */
                                HashResetRequest(this, cu->HashProvider).Send();
                                HashKeyRequest(this, cu->HashProvider, iv).Send();
-                               HashHexRequest(this, cu->HashProvider, cu->xtab[(*dest->host) % 4]);
+                               HashHexRequest(this, cu->HashProvider, cu->xtab[(dest->host[0]) % 4]);
 
                                /* Generate a cloak using specialized Hash */
-                               std::string hostcloak = cu->prefix + "-" + std::string(HashSumRequest(this, cu->HashProvider, dest->host).Send()).substr(0,8) + a;
+                               std::string hostcloak = cu->prefix + "-" + std::string(HashSumRequest(this, cu->HashProvider, dest->host.c_str()).Send()).substr(0,8) + a;
 
                                /* Fix by brain - if the cloaked host is > the max length of a host (64 bytes
-                                * according to the DNS RFC) then tough titty, they get cloaked as an IP. 
+                                * according to the DNS RFC) then tough titty, they get cloaked as an IP.
                                 * Their ISP shouldnt go to town on subdomains, or they shouldnt have a kiddie
                                 * vhost.
                                 */
-#ifdef IPV6
-                               in6_addr testaddr;
-                               in_addr testaddr2;
-                               if ((dest->GetProtocolFamily() == AF_INET6) && (inet_pton(AF_INET6,dest->host,&testaddr) < 1) && (hostcloak.length() <= 64))
-                                       /* Invalid ipv6 address, and ipv6 user (resolved host) */
-                                       b = hostcloak;
-                               else if ((dest->GetProtocolFamily() == AF_INET) && (inet_aton(dest->host,&testaddr2) < 1) && (hostcloak.length() <= 64))
-                                       /* Invalid ipv4 address, and ipv4 user (resolved host) */
-                                       b = hostcloak;
-                               else
-                                       /* Valid ipv6 or ipv4 address (not resolved) ipv4 or ipv6 user */
-                                       b = ((!strchr(dest->host,':')) ? cu->Cloak4(dest->host) : cu->Cloak6(dest->host));
-#else
-                               in_addr testaddr;
-                               if ((inet_aton(dest->host,&testaddr) < 1) && (hostcloak.length() <= 64))
-                                       /* Invalid ipv4 address, and ipv4 user (resolved host) */
+                               std::string testaddr;
+                               int testport;
+                               if (!irc::sockets::satoap(&dest->client_sa, testaddr, testport) && (hostcloak.length() <= 64))
+                                       /* not a valid address, must have been a host, so cloak as a host */
                                        b = hostcloak;
+                               else if (dest->client_sa.sa.sa_family == AF_INET6)
+                                       b = cu->Cloak6(dest->GetIPString());
                                else
-                                       /* Valid ipv4 address (not resolved) ipv4 user */
-                                       b = cu->Cloak4(dest->host);
-#endif
+                                       b = cu->Cloak4(dest->GetIPString());
                        }
                        else
                        {
-#ifdef IPV6
-                               if (dest->GetProtocolFamily() == AF_INET6)
+                               if (dest->client_sa.sa.sa_family == AF_INET6)
                                        b = cu->Cloak6(dest->GetIPString());
-#endif
-                               if (dest->GetProtocolFamily() == AF_INET)
+                               else
                                        b = cu->Cloak4(dest->GetIPString());
                        }
 
-                       dest->Extend("cloaked_host", new std::string(b));
+                       cu->ext.set(dest,b);
                }
        }