]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_cloaking.cpp
More cleanup of quit servers
[user/henk/code/inspircd.git] / src / modules / m_cloaking.cpp
index 381a2dc4009c76f5f753b1a4321747e50146db72..0405e31dd2d93e530e656db0f47ed879504e7152 100644 (file)
@@ -2,7 +2,7 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
  * See: http://wiki.inspircd.org/Credits
  *
  * This program is free but copyrighted software; see
@@ -57,7 +57,7 @@ class CloakUser : public ModeHandler
                }
 
                /* don't allow this user to spam modechanges */
-               dest->IncreasePenalty(5);
+               IS_LOCAL(dest)->CommandFloodPenalty += 5000;
 
                if (adding)
                {
@@ -74,10 +74,10 @@ class CloakUser : public ModeHandler
 
                                std::string* cloak = ext.get(dest);
 
-                               if (!cloak)
+                               if (!cloak && IS_LOCAL(dest))
                                {
                                        /* Force creation of missing cloak */
-                                       creator->OnUserConnect(dest);
+                                       creator->OnUserConnect(IS_LOCAL(dest));
                                        cloak = ext.get(dest);
                                }
                                if (cloak)
@@ -116,29 +116,25 @@ class ModuleCloaking : public Module
        std::string key;
        unsigned int compatkey[4];
        const char* xtab[4];
-       Module* HashProvider;
+       dynamic_reference<HashProvider> Hash;
 
  public:
-       ModuleCloaking() : cu(this)
+       ModuleCloaking() : cu(this), Hash(this, "hash/md5")
        {
-               /* Attempt to locate the md5 service provider, bail if we can't find it */
-               HashProvider = ServerInstance->Modules->Find("m_md5.so");
-               if (!HashProvider)
-                       throw ModuleException("Can't find m_md5.so. Please load m_md5.so before m_cloaking.so.");
+       }
 
+       void init()
+       {
                OnRehash(NULL);
 
                /* Register it with the core */
                if (!ServerInstance->Modes->AddMode(&cu))
                        throw ModuleException("Could not add new modes!");
 
-               ServerInstance->Modules->UseInterface("HashRequest");
-               Extensible::Register(&cu.ext);
+               ServerInstance->Extensions.Register(&cu.ext);
 
                Implementation eventlist[] = { I_OnRehash, I_OnCheckBan, I_OnUserConnect };
                ServerInstance->Modules->Attach(eventlist, this, 3);
-
-               CloakExistingUsers();
        }
 
        /** This function takes a domain name string and returns just the last two domain parts,
@@ -195,8 +191,7 @@ class ModuleCloaking : public Module
                /* Send the Hash module a different hex table for each octet group's Hash sum */
                for (int k = 0; k < 4; k++)
                {
-                       HashRequestIV hash(this, HashProvider, compatkey, xtab[(compatkey[k]+i[k]) % 4], octet[k]);
-                       rv.append(hash.result.substr(0,6));
+                       rv.append(Hash->sumIV(compatkey, xtab[(compatkey[k]+i[k]) % 4], octet[k]).substr(0,6));
                        if (k < 3)
                                rv.append(".");
                }
@@ -217,67 +212,76 @@ class ModuleCloaking : public Module
                        item += *input;
                        if (item.length() > 7)
                        {
-                               HashRequestIV hash(this, HashProvider, compatkey, xtab[(compatkey[1]+rounds) % 4], item);
-                               hashies.push_back(hash.result.substr(0,8));
+                               hashies.push_back(Hash->sumIV(compatkey, xtab[(compatkey[1]+rounds) % 4], item).substr(0,8));
                                item.clear();
                        }
                        rounds++;
                }
                if (!item.empty())
                {
-                       HashRequestIV hash(this, HashProvider, compatkey, xtab[(compatkey[1]+rounds) % 4], item);
-                       hashies.push_back(hash.result.substr(0,8));
+                       hashies.push_back(Hash->sumIV(compatkey, xtab[(compatkey[1]+rounds) % 4], item).substr(0,8));
                }
                /* Stick them all together */
                return irc::stringjoiner(":", hashies, 0, hashies.size() - 1).GetJoined();
        }
 
-       std::string ReversePartialIP(const irc::sockets::sockaddrs& ip)
-       {
-               char rv[50];
-               if (ip.sa.sa_family == AF_INET6)
-               {
-                       snprintf(rv, 50, ".%02x%02x.%02x%02x.%02x%02x.IP",
-                               ip.in6.sin6_addr.s6_addr[4], ip.in6.sin6_addr.s6_addr[5],
-                               ip.in6.sin6_addr.s6_addr[2], ip.in6.sin6_addr.s6_addr[3],
-                               ip.in6.sin6_addr.s6_addr[0], ip.in6.sin6_addr.s6_addr[1]);
-               }
-               else
-               {
-                       const char* ip4 = (const char*)&ip.in4.sin_addr;
-                       snprintf(rv, 50, ".%d.%d.IP", ip4[1], ip4[0]);
-               }
-               return rv;
-       }
-
-       std::string SegmentIP(const irc::sockets::sockaddrs& ip)
+       std::string SegmentIP(const irc::sockets::sockaddrs& ip, bool full)
        {
                std::string bindata;
-               int hop1, hop2;
+               int hop1, hop2, hop3;
+               std::string rv;
                if (ip.sa.sa_family == AF_INET6)
                {
                        bindata = std::string((const char*)ip.in6.sin6_addr.s6_addr, 16);
                        hop1 = 8;
                        hop2 = 6;
+                       hop3 = 4;
+                       rv.reserve(prefix.length() + 37);
                }
                else
                {
                        bindata = std::string((const char*)&ip.in4.sin_addr, 4);
                        hop1 = 3;
-                       hop2 = 2;
+                       hop2 = 0;
+                       hop3 = 2;
+                       rv.reserve(prefix.length() + 30);
                }
 
-               std::string rv;
-               rv.reserve(prefix.length() + 30);
                rv.append(prefix);
-               rv.append(SegmentCloak(bindata, 2));
+               rv.append(SegmentCloak(bindata, 10));
                rv.append(1, '.');
                bindata.erase(hop1);
-               rv.append(SegmentCloak(bindata, 3));
-               rv.append(1, '.');
-               bindata.erase(hop2);
-               rv.append(SegmentCloak(bindata, 4));
-               rv.append(".IP");
+               rv.append(SegmentCloak(bindata, 11));
+               if (hop2)
+               {
+                       rv.append(1, '.');
+                       bindata.erase(hop2);
+                       rv.append(SegmentCloak(bindata, 12));
+               }
+
+               if (full)
+               {
+                       rv.append(1, '.');
+                       bindata.erase(hop3);
+                       rv.append(SegmentCloak(bindata, 13));
+                       rv.append(".IP");
+               }
+               else
+               {
+                       char buf[50];
+                       if (ip.sa.sa_family == AF_INET6)
+                       {
+                               snprintf(buf, 50, ".%02x%02x.%02x%02x.IP",
+                                       ip.in6.sin6_addr.s6_addr[2], ip.in6.sin6_addr.s6_addr[3],
+                                       ip.in6.sin6_addr.s6_addr[0], ip.in6.sin6_addr.s6_addr[1]);
+                       }
+                       else
+                       {
+                               const unsigned char* ip4 = (const unsigned char*)&ip.in4.sin_addr;
+                               snprintf(buf, 50, ".%d.%d.IP", ip4[1], ip4[0]);
+                       }
+                       rv.append(buf);
+               }
                return rv;
        }
 
@@ -290,8 +294,7 @@ class ModuleCloaking : public Module
                input.append(1, 0); // null does not terminate a C++ string
                input.append(item);
 
-               HashRequest hash(this, HashProvider, input);
-               std::string rv = hash.binresult.substr(0,6);
+               std::string rv = Hash->sum(input).substr(0,6);
                for(int i=0; i < 6; i++)
                {
                        // this discards 3 bits per byte. We have an
@@ -302,26 +305,18 @@ class ModuleCloaking : public Module
                return rv;
        }
 
-       void CloakExistingUsers()
-       {
-               std::string* cloak;
-               for (std::vector<User*>::iterator u = ServerInstance->Users->local_users.begin(); u != ServerInstance->Users->local_users.end(); u++)
-               {
-                       cloak = cu.ext.get(*u);
-                       if (!cloak)
-                       {
-                               OnUserConnect(*u);
-                       }
-               }
-       }
-
        ModResult OnCheckBan(User* user, Channel* chan, const std::string& mask)
        {
-               char cmask[MAXBUF];
+               LocalUser* lu = IS_LOCAL(user);
+               if (!lu)
+                       return MOD_RES_PASSTHRU;
+
+               OnUserConnect(lu);
                std::string* cloak = cu.ext.get(user);
                /* Check if they have a cloaked host, but are not using it */
                if (cloak && *cloak != user->dhost)
                {
+                       char cmask[MAXBUF];
                        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;
@@ -337,14 +332,27 @@ class ModuleCloaking : public Module
 
        ~ModuleCloaking()
        {
-               ServerInstance->Modules->DoneWithInterface("HashRequest");
        }
 
        Version GetVersion()
        {
-               // returns the version number of the module to be
-               // listed in /MODULES
-               return Version("Provides masking of user hostnames", VF_COMMON|VF_VENDOR,API_VERSION);
+               std::string testcloak;
+               switch (mode)
+               {
+                       case MODE_COMPAT_HOST:
+                               testcloak = prefix + "-" + Hash->sumIV(compatkey, xtab[0], "*").substr(0,10);
+                               break;
+                       case MODE_COMPAT_IPONLY:
+                               testcloak = Hash->sumIV(compatkey, xtab[0], "*").substr(0,10);
+                               break;
+                       case MODE_HALF_CLOAK:
+                               testcloak = prefix + SegmentCloak("*", 3);
+                               break;
+                       case MODE_OPAQUE:
+                       default:
+                               testcloak = prefix + SegmentCloak("*", 4);
+               }
+               return Version("Provides masking of user hostnames", VF_COMMON|VF_VENDOR, testcloak);
        }
 
        void OnRehash(User* user)
@@ -377,10 +385,10 @@ class ModuleCloaking : public Module
                         * We must limit the keys or else we get different results on
                         * amd64/x86 boxes. - psychon */
                        const unsigned int limit = 0x80000000;
-                       compatkey[1] = (unsigned int) Conf.ReadInteger("cloak","key1",0,false);
-                       compatkey[2] = (unsigned int) Conf.ReadInteger("cloak","key2",0,false);
-                       compatkey[3] = (unsigned int) Conf.ReadInteger("cloak","key3",0,false);
-                       compatkey[4] = (unsigned int) Conf.ReadInteger("cloak","key4",0,false);
+                       compatkey[0] = (unsigned int) Conf.ReadInteger("cloak","key1",0,false);
+                       compatkey[1] = (unsigned int) Conf.ReadInteger("cloak","key2",0,false);
+                       compatkey[2] = (unsigned int) Conf.ReadInteger("cloak","key3",0,false);
+                       compatkey[3] = (unsigned int) Conf.ReadInteger("cloak","key4",0,false);
 
                        if (!lowercase)
                        {
@@ -400,17 +408,17 @@ class ModuleCloaking : public Module
                        if (prefix.empty())
                                prefix = ServerInstance->Config->Network;
 
-                       if (!compatkey[1] || !compatkey[2] || !compatkey[3] || !compatkey[4] ||
-                               compatkey[1] >= limit || compatkey[2] >= limit || compatkey[3] >= limit || compatkey[4] >= limit)
+                       if (!compatkey[0] || !compatkey[1] || !compatkey[2] || !compatkey[3] ||
+                               compatkey[0] >= limit || compatkey[1] >= limit || compatkey[2] >= limit || compatkey[3] >= limit)
                        {
                                std::string detail;
-                               if (!compatkey[1] || compatkey[1] >= limit)
+                               if (!compatkey[0] || compatkey[0] >= limit)
                                        detail = "<cloak:key1> is not valid, it may be set to a too high/low value, or it may not exist.";
-                               else if (!compatkey[2] || compatkey[2] >= limit)
+                               else if (!compatkey[1] || compatkey[1] >= limit)
                                        detail = "<cloak:key2> is not valid, it may be set to a too high/low value, or it may not exist.";
-                               else if (!compatkey[3] || compatkey[3] >= limit)
+                               else if (!compatkey[2] || compatkey[2] >= limit)
                                        detail = "<cloak:key3> is not valid, it may be set to a too high/low value, or it may not exist.";
-                               else if (!compatkey[4] || compatkey[4] >= limit)
+                               else if (!compatkey[3] || compatkey[3] >= limit)
                                        detail = "<cloak:key4> is not valid, it may be set to a too high/low value, or it may not exist.";
 
                                throw ModuleException("You have not defined cloak keys for m_cloaking!!! THIS IS INSECURE AND SHOULD BE CHECKED! - " + detail);
@@ -424,15 +432,12 @@ class ModuleCloaking : public Module
                }
        }
 
-       void OnUserConnect(User* dest)
+       void OnUserConnect(LocalUser* dest)
        {
                std::string* cloak = cu.ext.get(dest);
                if (cloak)
                        return;
 
-               if (dest->host.find('.') == std::string::npos && dest->host.find(':') == std::string::npos)
-                       return;
-
                std::string ipstr = dest->GetIPString();
                std::string chost;
 
@@ -444,11 +449,8 @@ class ModuleCloaking : public Module
                                {
                                        std::string tail = LastTwoDomainParts(dest->host);
 
-                                       /** Reset the Hash module, and send it our IV and hex table */
-                                       HashRequestIV hash(this, HashProvider, compatkey, xtab[(dest->host[0]) % 4], dest->host);
-
                                        /* Generate a cloak using specialized Hash */
-                                       chost = prefix + "-" + hash.result.substr(0,8) + tail;
+                                       chost = prefix + "-" + Hash->sumIV(compatkey, xtab[(dest->host[0]) % 4], dest->host).substr(0,8) + tail;
 
                                        /* Fix by brain - if the cloaked host is > the max length of a host (64 bytes
                                         * according to the DNS RFC) then they get cloaked as an IP.
@@ -466,17 +468,15 @@ class ModuleCloaking : public Module
                                break;
                        case MODE_HALF_CLOAK:
                        {
-                               std::string tail;
                                if (ipstr != dest->host)
-                                       tail = LastTwoDomainParts(dest->host);
-                               if (tail.empty() || tail.length() > 50)
-                                       tail = ReversePartialIP(dest->client_sa);
-                               chost = prefix + SegmentCloak(dest->host, 1) + tail;
+                                       chost = prefix + SegmentCloak(dest->host, 1) + LastTwoDomainParts(dest->host);
+                               if (chost.empty() || chost.length() > 50)
+                                       chost = SegmentIP(dest->client_sa, false);
                                break;
                        }
                        case MODE_OPAQUE:
                        default:
-                               chost = SegmentIP(dest->client_sa);
+                               chost = SegmentIP(dest->client_sa, true);
                }
                cu.ext.set(dest,chost);
        }