]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_cloaking.cpp
MetaData rework
[user/henk/code/inspircd.git] / src / modules / m_cloaking.cpp
index fb7ac134f1941a85206ae662eb34a0129eb098e8..458d833dfab28258c9f992d802765860a6166089 100644 (file)
  *       | Inspire 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.
+ *  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.
+ *         the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-#include "inspircd_config.h"
-#ifdef HAS_STDINT
-#include <stdint.h>
-#endif
-#include "configreader.h"
 #include "inspircd.h"
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
+#include "m_hash.h"
 
 /* $ModDesc: Provides masking of user hostnames */
-
-/* The four core functions - F1 is optimized somewhat */
-
-#define F1(x, y, z) (z ^ (x & (y ^ z)))
-#define F2(x, y, z) F1(z, x, y)
-#define F3(x, y, z) (x ^ y ^ z)
-#define F4(x, y, z) (y ^ (x | ~z))
-#define MD5STEP(f,w,x,y,z,in,s) (w += f(x,y,z) + in, w = (w<<s | w>>(32-s)) + x)
-
-#ifndef HAS_STDINT
-typedef unsigned int uint32_t;
-#endif
-
-typedef uint32_t word32; /* NOT unsigned long. We don't support 16 bit platforms, anyway. */
-typedef unsigned char byte;
-
-/** An MD5 context, used by m_cloaking
- */
-class xMD5Context : public classbase
-{
- public:
-       word32 buf[4];
-       word32 bytes[2];
-       word32 in[16];
-};
+/* $ModDep: m_hash.h */
 
 /** Handles user mode +x
  */
 class CloakUser : public ModeHandler
 {
-       
+ public:
        std::string prefix;
-       word32 key1;
-       word32 key2;
-       word32 key3;
-       word32 key4;
-       
-       void byteSwap(word32 *buf, unsigned words)
-       {
-               byte *p = (byte *)buf;
-       
-               do
-               {
-                       *buf++ = (word32)((unsigned)p[3] << 8 | p[2]) << 16 |
-                               ((unsigned)p[1] << 8 | p[0]);
-                       p += 4;
-               } while (--words);
-       }
-
-       void xMD5Init(struct xMD5Context *ctx)
-       {
-               ctx->buf[0] = key1;
-               ctx->buf[1] = key2;
-               ctx->buf[2] = key3;
-               ctx->buf[3] = key4;
-
-               ctx->bytes[0] = 0;
-               ctx->bytes[1] = 0;
-       }
-
-       void xMD5Update(struct xMD5Context *ctx, byte const *buf, int len)
+       unsigned int key1;
+       unsigned int key2;
+       unsigned int key3;
+       unsigned int key4;
+       bool ipalways;
+       Module* Sender;
+       Module* HashProvider;
+       const char *xtab[4];
+
+       /** 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.
+        *
+        * 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)
        {
-               word32 t;
-
-               /* Update byte count */
+               int dots = 0;
+               std::string::size_type splitdot = host.length();
 
-               t = ctx->bytes[0];
-               if ((ctx->bytes[0] = t + len) < t)
-                       ctx->bytes[1]++;        /* Carry from low to high */
-
-               t = 64 - (t & 0x3f);    /* Space available in ctx->in (at least 1) */
-               if ((unsigned)t > (unsigned)len)
+               for (std::string::size_type x = host.length() - 1; x; --x)
                {
-                       memcpy((byte *)ctx->in + 64 - (unsigned)t, buf, len);
-                       return;
-               }
-               /* First chunk is an odd size */
-               memcpy((byte *)ctx->in + 64 - (unsigned)t, buf, (unsigned)t);
-               byteSwap(ctx->in, 16);
-               xMD5Transform(ctx->buf, ctx->in);
-               buf += (unsigned)t;
-               len -= (unsigned)t;
-
-               /* Process data in 64-byte chunks */
-               while (len >= 64)
-               {
-                       memcpy(ctx->in, buf, 64);
-                       byteSwap(ctx->in, 16);
-                       xMD5Transform(ctx->buf, ctx->in);
-                       buf += 64;
-                       len -= 64;
-               }
-
-               /* Handle any remaining bytes of data. */
-               memcpy(ctx->in, buf, len);
-       }
-
-       void xMD5Final(byte digest[16], struct xMD5Context *ctx)
-       {
-               int count = (int)(ctx->bytes[0] & 0x3f); /* Bytes in ctx->in */
-               byte *p = (byte *)ctx->in + count;      /* First unused byte */
-
-               /* Set the first char of padding to 0x80.  There is always room. */
-               *p++ = 0x80;
-
-               /* Bytes of padding needed to make 56 bytes (-8..55) */
-               count = 56 - 1 - count;
-
-               if (count < 0)
-               {       /* Padding forces an extra block */
-                       memset(p, 0, count+8);
-                       byteSwap(ctx->in, 16);
-                       xMD5Transform(ctx->buf, ctx->in);
-                       p = (byte *)ctx->in;
-                       count = 56;
+                       if (host[x] == '.')
+                       {
+                               splitdot = x;
+                               dots++;
+                       }
+                       if (dots >= 3)
+                               break;
                }
-               memset(p, 0, count+8);
-               byteSwap(ctx->in, 14);
-
-               /* Append length in bits and transform */
-               ctx->in[14] = ctx->bytes[0] << 3;
-               ctx->in[15] = ctx->bytes[1] << 3 | ctx->bytes[0] >> 29;
-               xMD5Transform(ctx->buf, ctx->in);
-
-               byteSwap(ctx->buf, 4);
-               memcpy(digest, ctx->buf, 16);
-               memset(ctx, 0, sizeof(ctx));
-       }
 
-       void xMD5Transform(word32 buf[4], word32 const in[16])
-       {
-               register word32 a, b, c, d;
-
-               a = buf[0];
-               b = buf[1];
-               c = buf[2];
-               d = buf[3];
-
-               MD5STEP(F1, a, b, c, d, in[0] + 0xd76aa478, 7);
-               MD5STEP(F1, d, a, b, c, in[1] + 0xe8c7b756, 12);
-               MD5STEP(F1, c, d, a, b, in[2] + 0x242070db, 17);
-               MD5STEP(F1, b, c, d, a, in[3] + 0xc1bdceee, 22);
-               MD5STEP(F1, a, b, c, d, in[4] + 0xf57c0faf, 7);
-               MD5STEP(F1, d, a, b, c, in[5] + 0x4787c62a, 12);
-               MD5STEP(F1, c, d, a, b, in[6] + 0xa8304613, 17);
-               MD5STEP(F1, b, c, d, a, in[7] + 0xfd469501, 22);
-               MD5STEP(F1, a, b, c, d, in[8] + 0x698098d8, 7);
-               MD5STEP(F1, d, a, b, c, in[9] + 0x8b44f7af, 12);
-               MD5STEP(F1, c, d, a, b, in[10] + 0xffff5bb1, 17);
-               MD5STEP(F1, b, c, d, a, in[11] + 0x895cd7be, 22);
-               MD5STEP(F1, a, b, c, d, in[12] + 0x6b901122, 7);
-               MD5STEP(F1, d, a, b, c, in[13] + 0xfd987193, 12);
-               MD5STEP(F1, c, d, a, b, in[14] + 0xa679438e, 17);
-               MD5STEP(F1, b, c, d, a, in[15] + 0x49b40821, 22);
-
-               MD5STEP(F2, a, b, c, d, in[1] + 0xf61e2562, 5);
-               MD5STEP(F2, d, a, b, c, in[6] + 0xc040b340, 9);
-               MD5STEP(F2, c, d, a, b, in[11] + 0x265e5a51, 14);
-               MD5STEP(F2, b, c, d, a, in[0] + 0xe9b6c7aa, 20);
-               MD5STEP(F2, a, b, c, d, in[5] + 0xd62f105d, 5);
-               MD5STEP(F2, d, a, b, c, in[10] + 0x02441453, 9);
-               MD5STEP(F2, c, d, a, b, in[15] + 0xd8a1e681, 14);
-               MD5STEP(F2, b, c, d, a, in[4] + 0xe7d3fbc8, 20);
-               MD5STEP(F2, a, b, c, d, in[9] + 0x21e1cde6, 5);
-               MD5STEP(F2, d, a, b, c, in[14] + 0xc33707d6, 9);
-               MD5STEP(F2, c, d, a, b, in[3] + 0xf4d50d87, 14);
-               MD5STEP(F2, b, c, d, a, in[8] + 0x455a14ed, 20);
-               MD5STEP(F2, a, b, c, d, in[13] + 0xa9e3e905, 5);
-               MD5STEP(F2, d, a, b, c, in[2] + 0xfcefa3f8, 9);
-               MD5STEP(F2, c, d, a, b, in[7] + 0x676f02d9, 14);
-               MD5STEP(F2, b, c, d, a, in[12] + 0x8d2a4c8a, 20);
-
-               MD5STEP(F3, a, b, c, d, in[5] + 0xfffa3942, 4);
-               MD5STEP(F3, d, a, b, c, in[8] + 0x8771f681, 11);
-               MD5STEP(F3, c, d, a, b, in[11] + 0x6d9d6122, 16);
-               MD5STEP(F3, b, c, d, a, in[14] + 0xfde5380c, 23);
-               MD5STEP(F3, a, b, c, d, in[1] + 0xa4beea44, 4);
-               MD5STEP(F3, d, a, b, c, in[4] + 0x4bdecfa9, 11);
-               MD5STEP(F3, c, d, a, b, in[7] + 0xf6bb4b60, 16);
-               MD5STEP(F3, b, c, d, a, in[10] + 0xbebfbc70, 23);
-               MD5STEP(F3, a, b, c, d, in[13] + 0x289b7ec6, 4);
-               MD5STEP(F3, d, a, b, c, in[0] + 0xeaa127fa, 11);
-               MD5STEP(F3, c, d, a, b, in[3] + 0xd4ef3085, 16);
-               MD5STEP(F3, b, c, d, a, in[6] + 0x04881d05, 23);
-               MD5STEP(F3, a, b, c, d, in[9] + 0xd9d4d039, 4);
-               MD5STEP(F3, d, a, b, c, in[12] + 0xe6db99e5, 11);
-               MD5STEP(F3, c, d, a, b, in[15] + 0x1fa27cf8, 16);
-               MD5STEP(F3, b, c, d, a, in[2] + 0xc4ac5665, 23);
-
-               MD5STEP(F4, a, b, c, d, in[0] + 0xf4292244, 6);
-               MD5STEP(F4, d, a, b, c, in[7] + 0x432aff97, 10);
-               MD5STEP(F4, c, d, a, b, in[14] + 0xab9423a7, 15);
-               MD5STEP(F4, b, c, d, a, in[5] + 0xfc93a039, 21);
-               MD5STEP(F4, a, b, c, d, in[12] + 0x655b59c3, 6);
-               MD5STEP(F4, d, a, b, c, in[3] + 0x8f0ccc92, 10);
-               MD5STEP(F4, c, d, a, b, in[10] + 0xffeff47d, 15);
-               MD5STEP(F4, b, c, d, a, in[1] + 0x85845dd1, 21);
-               MD5STEP(F4, a, b, c, d, in[8] + 0x6fa87e4f, 6);
-               MD5STEP(F4, d, a, b, c, in[15] + 0xfe2ce6e0, 10);
-               MD5STEP(F4, c, d, a, b, in[6] + 0xa3014314, 15);
-               MD5STEP(F4, b, c, d, a, in[13] + 0x4e0811a1, 21);
-               MD5STEP(F4, a, b, c, d, in[4] + 0xf7537e82, 6);
-               MD5STEP(F4, d, a, b, c, in[11] + 0xbd3af235, 10);
-               MD5STEP(F4, c, d, a, b, in[2] + 0x2ad7d2bb, 15);
-               MD5STEP(F4, b, c, d, a, in[9] + 0xeb86d391, 21);
-
-               buf[0] += a;
-               buf[1] += b;
-               buf[2] += c;
-               buf[3] += d;
+               if (splitdot == host.length())
+                       return host;
+               else
+                       return host.substr(splitdot);
        }
 
-
-       void MyMD5(void *dest, const void *orig, int len)
+       CloakUser(InspIRCd* Instance, Module* source, Module* Hash) : ModeHandler(Instance, 'x', 0, 0, false, MODETYPE_USER, false), Sender(source), HashProvider(Hash)
        {
-               struct xMD5Context context;
-       
-               xMD5Init(&context);
-               xMD5Update(&context, (const unsigned char*)orig, len);
-               xMD5Final((unsigned char*)dest, &context);
        }
 
-
-       void GenHash(const char* src, char* dest, int xtable = 0)
+       ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding, bool)
        {
-               // purposefully lossy md5 - only gives them the most significant 4 bits
-               // of every md5 output byte.
-               int i = 0;
-               unsigned char bytes[16];
-               char hash[MAXBUF];
-               *hash = 0;
-               MyMD5(bytes, src, strlen(src));
-               for (i = 0; i < 16; i++)
-               {
-                       const char* xtab[] = {"F92E45D871BCA630", "A1B9D80C72E653F4", "1ABC078934DEF562", "ABCDEF5678901234"};
-                       char hx[2] = { xtab[xtable][bytes[i] / 16], 0};
-                       strlcat(hash,hx,MAXBUF);
-               }
-               strlcpy(dest,hash,MAXBUF);
-       }
-       
- public:
-       CloakUser(InspIRCd* Instance) : ModeHandler(Instance, 'x', 0, 0, false, MODETYPE_USER, false) { }
-
-       ModeAction OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string &parameter, bool adding)
-       {
-               /* Only opers can change other users modes */
-               if ((source != dest) && (!*source->oper))
-                       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.
                 */
                if (!IS_LOCAL(dest))
+               {
+                       dest->SetMode('x',adding);
                        return MODEACTION_ALLOW;
+               }
+
+               /* don't allow this user to spam modechanges */
+               dest->IncreasePenalty(5);
 
                if (adding)
                {
@@ -293,58 +94,30 @@ class CloakUser : public ModeHandler
                                 * naming in their hostname (e.g. if they are on a lan or
                                 * are connecting via localhost) -- this doesnt matter much.
                                 */
-                       
-                               if (strchr(dest->host,'.') || strchr(dest->host,':'))
+
+                               std::string* cloak;
+
+                               if (!dest->GetExt("cloaked_host", cloak))
                                {
-                                       /* InspIRCd users have two hostnames; A displayed
-                                        * hostname which can be modified by modules (e.g.
-                                        * to create vhosts, implement chghost, etc) and a
-                                        * 'real' hostname which you shouldnt write to.
-                                        */
-
-                                       char* n = strstr(dest->host,".");
-                                       if (!n)
-                                               n = strstr(dest->host,":");
-
-                                       std::string a = n;
-                                       
-                                       char ra[64];
-                                       this->GenHash(dest->host,ra);
-                                       std::string b;
-                                       insp_inaddr testaddr;
-                                       std::string hostcloak = prefix + "-" + ra + 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. 
-                                        * Their ISP shouldnt go to town on subdomains, or they shouldnt have a kiddie
-                                        * vhost.
-                                        */
-
-                                       if ((insp_aton(dest->host,&testaddr) < 1) && (hostcloak.length() <= 64))
-                                       {
-                                               // if they have a hostname, make something appropriate
-                                               b = hostcloak;
-                                       }
-                                       else
-                                       {
-                                               b = ((b.find(':') == std::string::npos) ? Cloak4(dest->host) : Cloak6(dest->host));
-                                       }
-                                       ServerInstance->Log(DEBUG,"cloak: allocated "+b);
-                                       dest->ChangeDisplayedHost(b.c_str());
+                                       /* Force creation of missing cloak */
+                                       Sender->OnUserConnect(dest);
+                               }
+                               if (dest->GetExt("cloaked_host", cloak))
+                               {
+                                       dest->ChangeDisplayedHost(cloak->c_str());
+                                       dest->SetMode('x',true);
+                                       return MODEACTION_ALLOW;
                                }
-                               
-                               dest->SetMode('x',true);
-                               return MODEACTION_ALLOW;
                        }
                }
                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;
                        }
@@ -355,101 +128,120 @@ class CloakUser : public ModeHandler
 
        std::string Cloak4(const char* ip)
        {
+               unsigned int iv[] = { key1, key2, key3, key4 };
                irc::sepstream seps(ip, '.');
-               char ra1[64], ra2[64], ra3[64], ra4[64];
-               int i1, i2, i3, i4;
-               std::string octet1 = seps.GetToken();
-               std::string octet2 = seps.GetToken();
-               std::string octet3 = seps.GetToken();
-               std::string octet4 = seps.GetToken();
-               i1 = atoi(octet1.c_str());
-               i2 = atoi(octet2.c_str());
-               i3 = atoi(octet3.c_str());
-               i4 = atoi(octet4.c_str());
-               octet4 = octet1 + "." + octet2 + "." + octet3 + "." + octet4;
-               octet3 = octet1 + "." + octet2 + "." + octet3;
-               octet2 = octet1 + "." + octet2;
-               this->GenHash(octet1.c_str(),ra1, (key1+i1) % 4);
-               this->GenHash(octet2.c_str(),ra2, (key2+i2) % 4);
-               this->GenHash(octet3.c_str(),ra3, (key3+i3) % 4);
-               this->GenHash(octet4.c_str(),ra4, (key4+i4) % 4);
-               /* This is safe as we know the length generated by our genhash is always 16 */
-               ra1[8] = ra2[8] = ra3[8] = ra4[8] = 0;
-               return std::string().append(ra1).append(".").append(ra2).append(".").append(ra3).append(".").append(ra4);
+               std::string ra[4];;
+               std::string octet[4];
+               int i[4];
+
+               for (int j = 0; j < 4; j++)
+               {
+                       seps.GetToken(octet[j]);
+                       i[j] = atoi(octet[j].c_str());
+               }
+
+               octet[3] = octet[0] + "." + octet[1] + "." + octet[2] + "." + octet[3];
+               octet[2] = octet[0] + "." + octet[1] + "." + octet[2];
+               octet[1] = octet[0] + "." + octet[1];
+
+               /* Reset the Hash module and send it our IV */
+               HashResetRequest(Sender, HashProvider).Send();
+               HashKeyRequest(Sender, 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);
+               }
+               /* Stick them all together */
+               return std::string().append(ra[0]).append(".").append(ra[1]).append(".").append(ra[2]).append(".").append(ra[3]);
        }
 
        std::string Cloak6(const char* ip)
        {
+               unsigned int iv[] = { key1, key2, key3, key4 };
                std::vector<std::string> hashies;
-               std::string item = "";
+               std::string item;
                int rounds = 0;
+
+               /* Reset the Hash module and send it our IV */
+               HashResetRequest(Sender, HashProvider).Send();
+               HashKeyRequest(Sender, HashProvider, iv).Send();
+
                for (const char* input = ip; *input; input++)
                {
                        item += *input;
-                       if (item.length() > 5)
+                       if (item.length() > 7)
                        {
-                               char ra[64];
-                               this->GenHash(item.c_str(), ra, (key1+rounds) % 4);
-                               ra[9] = 0;
-                               hashies.push_back(ra);
-                               item = "";
+                               /* 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));
+                               item.clear();
                        }
                        rounds++;
                }
                if (!item.empty())
                {
-                       char ra[64];
-                       this->GenHash(item.c_str(), ra, (key1+rounds) % 4);
-                       ra[9] = 0;
-                       hashies.push_back(ra);
+                       /* 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));
+                       item.clear();
                }
+               /* Stick them all together */
                return irc::stringjoiner(":", hashies, 0, hashies.size() - 1).GetJoined();
        }
 
-       /* XXX: Uncomment and call to use the test suite
-       void TestSuite()
-       {
-               printf("%s %s\n", "192.168.1.1", Cloak4("192.168.1.1").c_str());
-               printf("%s %s\n", "192.168.1.2", Cloak4("192.168.1.2").c_str());
-               printf("%s %s\n", "192.168.10.1", Cloak4("192.168.10.1").c_str());
-               printf("%s %s\n", "192.168.10.1", Cloak4("192.168.10.2").c_str());
-               printf("%s %s\n", "192.169.1.1", Cloak4("192.169.1.1").c_str());
-               printf("%s %s\n", "192.169.2.1", Cloak4("192.169.2.1").c_str());
-               printf("%s %s\n", "200.168.1.1", Cloak4("200.168.1.1").c_str());
-               printf("%s %s\n", "200.168.1.3", Cloak4("200.168.1.3").c_str());
-               printf("%s %s\n", "200.168.3.3", Cloak4("200.168.3.3").c_str());
-               printf("%s %s\n", "200.169.4.3", Cloak4("200.169.4.3").c_str());
-               printf("---\n");
-               printf("%s %s\n", "9a05:2f00:3f11::5f12::1", Cloak6("9a05:2f00:3f11::5f12::1").c_str());
-               printf("%s %s\n", "9a05:2f00:3f11::5f12::2", Cloak6("9a05:2f00:3f11::5f12::2").c_str());
-               printf("%s %s\n", "9a05:2f00:3f11::5a12::1", Cloak6("9a05:2f00:3f11::5a12::1").c_str());
-               printf("%s %s\n", "9a05:2f00:3f11::5a12::2", Cloak6("9a05:2f00:3f11::5a12::2").c_str());
-               printf("%s %s\n", "9a05:3f01:3f11::5f12::1", Cloak6("9a05:3f01:3f11::5f12::1").c_str());
-               printf("%s %s\n", "9a05:4f00:3f11::5f13::2", Cloak6("9a05:4f00:3f11::5f13::2").c_str());
-               printf("%s %s\n", "ffff:2f00:3f11::5f12::1", Cloak6("ffff:2f00:3f11::5f12::1").c_str());
-               printf("%s %s\n", "ffff:2f00:3f11::5f13::2", Cloak6("ffff:2f00:3f11::5f13::2").c_str());
-               exit(0);
-       }
-       */
-       
        void DoRehash()
        {
                ConfigReader Conf(ServerInstance);
+               bool lowercase;
+
+               /* 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
+                * spectrum of an unsigned integer. - Special */
                key1 = key2 = key3 = key4 = 0;
-               key1 = Conf.ReadInteger("cloak","key1",0,false);
-               key2 = Conf.ReadInteger("cloak","key2",0,false);
-               key3 = Conf.ReadInteger("cloak","key3",0,false);
-               key4 = Conf.ReadInteger("cloak","key4",0,false);
-               
+               key1 = (unsigned int) Conf.ReadInteger("cloak","key1",0,false);
+               key2 = (unsigned int) Conf.ReadInteger("cloak","key2",0,false);
+               key3 = (unsigned int) Conf.ReadInteger("cloak","key3",0,false);
+               key4 = (unsigned int) Conf.ReadInteger("cloak","key4",0,false);
                prefix = Conf.ReadValue("cloak","prefix",0);
-               if (prefix == "")
+               ipalways = Conf.ReadFlag("cloak", "ipalways", 0);
+               lowercase = Conf.ReadFlag("cloak", "lowercase", 0);
+
+               if (!lowercase)
                {
-                       prefix = ServerInstance->Config->Network;
+                       xtab[0] = "F92E45D871BCA630";
+                       xtab[1] = "A1B9D80C72E653F4";
+                       xtab[2] = "1ABC078934DEF562";
+                       xtab[3] = "ABCDEF5678901234";
                }
-               if (!key1 && !key2 && !key3 && !key4)
+               else
                {
-                       ModuleException ex("You have not defined cloak keys for m_cloaking!!! THIS IS INSECURE AND SHOULD BE CHECKED!");
-                       throw (ex);
+                       xtab[0] = "f92e45d871bca630";
+                       xtab[1] = "a1b9d80c72e653f4";
+                       xtab[2] = "1abc078934def562";
+                       xtab[3] = "abcdef5678901234";
+               }
+
+               if (prefix.empty())
+                       prefix = ServerInstance->Config->Network;
+
+               if (!key1 || !key2 || !key3 || !key4)
+               {
+                       std::string detail;
+                       if (!key1)
+                               detail = "<cloak:key1> is not valid, it may be set to a too high/low value, or it may not exist.";
+                       else if (!key2)
+                               detail = "<cloak:key2> is not valid, it may be set to a too high/low value, or it may not exist.";
+                       else if (!key3)
+                               detail = "<cloak:key3> is not valid, it may be set to a too high/low value, or it may not exist.";
+                       else if (!key4)
+                               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);
                }
        }
 };
@@ -458,68 +250,183 @@ class CloakUser : public ModeHandler
 class ModuleCloaking : public Module
 {
  private:
-       
        CloakUser* cu;
 
  public:
        ModuleCloaking(InspIRCd* Me)
-               : Module::Module(Me)
+               : Module(Me)
        {
-               /* Create new mode handler object */
-               cu = new CloakUser(ServerInstance);
+               /* Attempt to locate the md5 service provider, bail if we can't find it */
+               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);
+
+               try
+               {
+                       OnRehash(NULL);
+               }
+               catch (ModuleException &e)
+               {
+                       delete cu;
+                       throw e;
+               }
+
+               /* Register it with the core */
+               if (!ServerInstance->Modes->AddMode(cu))
+               {
+                       delete cu;
+                       throw ModuleException("Could not add new modes!");
+               }
+
+               ServerInstance->Modules->UseInterface("HashRequest");
 
-               /* Register it with the core */         
-               ServerInstance->AddMode(cu, 'x');
+               Implementation eventlist[] = { I_OnRehash, I_OnUserDisconnect, I_OnCleanup, I_OnCheckBan, I_OnUserConnect, I_OnSyncUser, I_OnCleanup };
+               ServerInstance->Modules->Attach(eventlist, this, 6);
 
-               OnRehash("");
+               CloakExistingUsers();
        }
-       
-       virtual ~ModuleCloaking()
+
+       void OnSyncUser(User* user, Module* proto,void* opaque)
        {
-               ServerInstance->Modes->DelMode(cu);
-               DELETE(cu);
+               std::string* cloak;
+               if (user->GetExt("cloaked_host", cloak) && proto->ProtoTranslate(NULL) == "?")
+                       proto->ProtoSendMetaData(opaque, user, "cloaked_host", *cloak);
        }
-       
-       virtual Version GetVersion()
+
+       void CloakExistingUsers()
        {
-               // returns the version number of the module to be
-               // listed in /MODULES
-               return Version(1,1,0,2,VF_COMMON|VF_VENDOR,API_VERSION);
+               std::string* cloak;
+               for (std::vector<User*>::iterator u = ServerInstance->Users->local_users.begin(); u != ServerInstance->Users->local_users.end(); u++)
+               {
+                       if (!(*u)->GetExt("cloaked_host", cloak))
+                       {
+                               OnUserConnect(*u);
+                       }
+               }
        }
 
-       virtual void OnRehash(const std::string &parameter)
+       virtual int OnCheckBan(User* user, Channel* chan)
        {
-               cu->DoRehash();
+               char mask[MAXBUF];
+               std::string* tofree;
+               /* Check if they have a cloaked host, but are not using it */
+               if (user->GetExt("cloaked_host", tofree) && *tofree != user->dhost)
+               {
+                       snprintf(mask, MAXBUF, "%s!%s@%s", user->nick.c_str(), user->ident.c_str(), tofree->c_str());
+                       for (BanList::iterator i = chan->bans.begin(); i != chan->bans.end(); i++)
+                       {
+                               if (InspIRCd::Match(mask,i->data))
+                                       return -1;
+                       }
+               }
+               return 0;
+       }
+
+       void Prioritize()
+       {
+               /* Needs to be after m_banexception etc. */
+               ServerInstance->Modules->SetPriority(this, I_OnCheckBan, PRIORITY_LAST);
+
+               /* 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);
        }
 
-       void Implements(char* List)
+       virtual void OnUserDisconnect(User* user)
        {
-               List[I_OnRehash] = 1;
+               std::string* tofree;
+               if (user->GetExt("cloaked_host", tofree))
+               {
+                       delete tofree;
+                       user->Shrink("cloaked_host");
+               }
        }
-};
 
-// stuff down here is the module-factory stuff. For basic modules you can ignore this.
+       virtual void OnCleanup(int target_type, void* item)
+       {
+               if (target_type == TYPE_USER)
+                       OnUserDisconnect((User*)item);
+       }
 
-class ModuleCloakingFactory : public ModuleFactory
-{
- public:
-       ModuleCloakingFactory()
+       virtual ~ModuleCloaking()
+       {
+               ServerInstance->Modes->DelMode(cu);
+               delete cu;
+               ServerInstance->Modules->DoneWithInterface("HashRequest");
+       }
+
+       virtual Version GetVersion()
        {
+               // returns the version number of the module to be
+               // listed in /MODULES
+               return Version("$Id$", VF_COMMON|VF_VENDOR,API_VERSION);
        }
-       
-       ~ModuleCloakingFactory()
+
+       virtual void OnRehash(User* user)
        {
+               cu->DoRehash();
        }
-       
-       virtual Module * CreateModule(InspIRCd* Me)
+
+       virtual void OnUserConnect(User* dest)
        {
-               return new ModuleCloaking(Me);
+               std::string* tofree;
+               if (dest->GetExt("cloaked_host", tofree))
+                       return;
+
+               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);
+                       std::string b;
+
+                       /* InspIRCd users have two hostnames; A displayed
+                        * hostname which can be modified by modules (e.g.
+                        * to create vhosts, implement chghost, etc) and a
+                        * 'real' hostname which you shouldnt write to.
+                        */
+
+                       /* 2008/08/18: add <cloak:ipalways> which always cloaks
+                        * the IP, for anonymity. --nenolod
+                        */
+                       if (!cu->ipalways)
+                       {
+                               /** 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[0]) % 4]);
+
+                               /* Generate a cloak using specialized Hash */
+                               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.
+                                * Their ISP shouldnt go to town on subdomains, or they shouldnt have a kiddie
+                                * vhost.
+                                */
+                               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
+                                       b = cu->Cloak4(dest->GetIPString());
+                       }
+                       else
+                       {
+                               if (dest->client_sa.sa.sa_family == AF_INET6)
+                                       b = cu->Cloak6(dest->GetIPString());
+                               else
+                                       b = cu->Cloak4(dest->GetIPString());
+                       }
+
+                       dest->Extend("cloaked_host", new std::string(b));
+               }
        }
-       
-};
 
+};
 
-extern "C" void * init_module( void )
-{
-       return new ModuleCloakingFactory;
-}
+MODULE_INIT(ModuleCloaking)