]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_cloaking.cpp
Merge pull request #1219 from SaberUK/master+directive
[user/henk/code/inspircd.git] / src / modules / m_cloaking.cpp
index f26e16e0e2edb7cb25a204ef0cba60f88d08c138..890c799f2b2b55d2e96b4c099db9bdc1163a9f75 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) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2006-2008 Robin Burchell <robin+git@viroteck.net>
+ *   Copyright (C) 2008 Pippijn van Steenhoven <pip88nl@gmail.com>
+ *   Copyright (C) 2003-2008 Craig Edwards <craigedwards@brainbox.cc>
+ *   Copyright (C) 2007 John Brooks <john.brooks@dereferenced.net>
+ *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ *   Copyright (C) 2006 Oliver Lupton <oliverlupton@gmail.com>
  *
- * ---------------------------------------------------
+ * 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/>.
  */
 
-// Hostname cloaking (+x mode) module for inspircd.
-// version 1.0.0.1 by brain (C. J. Edwards) Mar 2004.
-//
-// When loaded this module will automatically set the
-// +x mode on all connecting clients.
-//
-// Setting +x on a client causes the module to change the
-// dhost entry (displayed host) for each user who has the
-// mode, cloaking their host. Unlike unreal, the algorithm
-// is non-reversible as uncloaked hosts are passed along
-// the server->server link, and all encoding of hosts is
-// done locally on the server by this module.
-
-#include <sys/types.h>
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <arpa/inet.h>
-#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"
-
-/* $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;
+#include "inspircd.h"
+#include "modules/hash.h"
 
-class xMD5Context : public classbase
+enum CloakMode
 {
- public:
-       word32 buf[4];
-       word32 bytes[2];
-       word32 in[16];
+       /** 2.0 cloak of "half" of the hostname plus the full IP hash */
+       MODE_HALF_CLOAK,
+       /** 2.0 cloak of IP hash, split at 2 common CIDR range points */
+       MODE_OPAQUE
 };
 
+// lowercase-only encoding similar to base64, used for hash output
+static const char base32[] = "0123456789abcdefghijklmnopqrstuv";
+
+/** Handles user mode +x
+ */
 class CloakUser : public ModeHandler
 {
-       
-       std::string prefix;
-       word32 key1;
-       word32 key2;
-       word32 key3;
-       word32 key4;
-       
-       void byteSwap(word32 *buf, unsigned words)
+ public:
+       LocalStringExt ext;
+       std::string debounce_uid;
+       time_t debounce_ts;
+       int debounce_count;
+
+       CloakUser(Module* source)
+               : ModeHandler(source, "cloak", 'x', PARAM_NONE, MODETYPE_USER),
+               ext("cloaked_host", ExtensionItem::EXT_USER, source), debounce_ts(0), debounce_count(0)
        {
-               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)
+       ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
        {
-               ctx->buf[0] = key1;
-               ctx->buf[1] = key2;
-               ctx->buf[2] = key3;
-               ctx->buf[3] = key4;
+               LocalUser* user = IS_LOCAL(dest);
 
-               ctx->bytes[0] = 0;
-               ctx->bytes[1] = 0;
-       }
+               /* For remote clients, we don't take any action, we just allow it.
+                * The local server where they are will set their cloak instead.
+                * This is fine, as we will receive it later.
+                */
+               if (!user)
+               {
+                       dest->SetMode(this, adding);
+                       return MODEACTION_ALLOW;
+               }
 
-       void xMD5Update(struct xMD5Context *ctx, byte const *buf, int len)
-       {
-               word32 t;
+               if (user->uuid == debounce_uid && debounce_ts == ServerInstance->Time())
+               {
+                       // prevent spamming using /mode user +x-x+x-x+x-x
+                       if (++debounce_count > 2)
+                               return MODEACTION_DENY;
+               }
+               else
+               {
+                       debounce_uid = user->uuid;
+                       debounce_count = 1;
+                       debounce_ts = ServerInstance->Time();
+               }
 
-               /* Update byte count */
+               if (adding == user->IsModeSet(this))
+                       return MODEACTION_DENY;
 
-               t = ctx->bytes[0];
-               if ((ctx->bytes[0] = t + len) < t)
-                       ctx->bytes[1]++;        /* Carry from low to high */
+               /* don't allow this user to spam modechanges */
+               if (source == dest)
+                       user->CommandFloodPenalty += 5000;
 
-               t = 64 - (t & 0x3f);    /* Space available in ctx->in (at least 1) */
-               if ((unsigned)t > (unsigned)len)
+               if (adding)
                {
-                       memcpy((byte *)ctx->in + 64 - (unsigned)t, buf, len);
-                       return;
+                       std::string* cloak = ext.get(user);
+
+                       if (!cloak)
+                       {
+                               /* Force creation of missing cloak */
+                               creator->OnUserConnect(user);
+                               cloak = ext.get(user);
+                       }
+                       if (cloak)
+                       {
+                               user->ChangeDisplayedHost(*cloak);
+                               user->SetMode(this, true);
+                               return MODEACTION_ALLOW;
+                       }
+                       else
+                               return MODEACTION_DENY;
                }
-               /* 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)
+               else
                {
-                       memcpy(ctx->in, buf, 64);
-                       byteSwap(ctx->in, 16);
-                       xMD5Transform(ctx->buf, ctx->in);
-                       buf += 64;
-                       len -= 64;
+                       /* User is removing the mode, so restore their real host
+                        * and make it match the displayed one.
+                        */
+                       user->SetMode(this, false);
+                       user->ChangeDisplayedHost(user->host.c_str());
+                       return MODEACTION_ALLOW;
                }
-
-               /* Handle any remaining bytes of data. */
-               memcpy(ctx->in, buf, len);
        }
+};
 
-       void xMD5Final(byte digest[16], struct xMD5Context *ctx)
+class CommandCloak : public Command
+{
+ public:
+       CommandCloak(Module* Creator) : Command(Creator, "CLOAK", 1)
        {
-               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;
-               }
-               memset(p, 0, count+8);
-               byteSwap(ctx->in, 14);
+               flags_needed = 'o';
+               syntax = "<host>";
+       }
+
+       CmdResult Handle(const std::vector<std::string> &parameters, User *user);
+};
 
-               /* 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);
+class ModuleCloaking : public Module
+{
+ public:
+       CloakUser cu;
+       CloakMode mode;
+       CommandCloak ck;
+       std::string prefix;
+       std::string suffix;
+       std::string key;
+       const char* xtab[4];
+       dynamic_reference<HashProvider> Hash;
 
-               byteSwap(ctx->buf, 4);
-               memcpy(digest, ctx->buf, 16);
-               memset(ctx, 0, sizeof(ctx));
+       ModuleCloaking() : cu(this), mode(MODE_OPAQUE), ck(this), Hash(this, "hash/md5")
+       {
        }
 
-       void xMD5Transform(word32 buf[4], word32 const in[16])
+       /** 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)
        {
-               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;
-       }
+               int dots = 0;
+               std::string::size_type splitdot = host.length();
 
+               for (std::string::size_type x = host.length() - 1; x; --x)
+               {
+                       if (host[x] == '.')
+                       {
+                               splitdot = x;
+                               dots++;
+                       }
+                       if (dots >= 3)
+                               break;
+               }
 
-       void MyMD5(void *dest, void *orig, int len)
-       {
-               struct xMD5Context context;
-       
-               xMD5Init(&context);
-               xMD5Update(&context, (const unsigned char*)orig, len);
-               xMD5Final((unsigned char*)dest, &context);
+               if (splitdot == host.length())
+                       return "";
+               else
+                       return host.substr(splitdot);
        }
 
-
-       void GenHash(char* src, char* dest)
+       /**
+        * 2.0-style cloaking function
+        * @param item The item to cloak (part of an IP or hostname)
+        * @param id A unique ID for this type of item (to make it unique if the item matches)
+        * @param len The length of the output. Maximum for MD5 is 16 characters.
+        */
+       std::string SegmentCloak(const std::string& item, char id, int len)
        {
-               // 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((char*)bytes,src,strlen(src));
-               for (i = 0; i < 16; i++)
+               std::string input;
+               input.reserve(key.length() + 3 + item.length());
+               input.append(1, id);
+               input.append(key);
+               input.append(1, '\0'); // null does not terminate a C++ string
+               input.append(item);
+
+               std::string rv = Hash->GenerateRaw(input).substr(0,len);
+               for(int i=0; i < len; i++)
                {
-                       const char* xtab = "F92E45D871BCA630";
-                       unsigned char hi = xtab[bytes[i] / 16];
-                       char hx[2];
-                       hx[0] = hi;
-                       hx[1] = '\0';
-                       strlcat(hash,hx,MAXBUF);
+                       // this discards 3 bits per byte. We have an
+                       // overabundance of bits in the hash output, doesn't
+                       // matter which ones we are discarding.
+                       rv[i] = base32[rv[i] & 0x1F];
                }
-               strlcpy(dest,hash,MAXBUF);
+               return rv;
        }
-       
- 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)
+       std::string SegmentIP(const irc::sockets::sockaddrs& ip, bool full)
        {
-               /* Only opers can change other users modes */
-               if ((source != dest) && (!*source->oper))
-                       return MODEACTION_DENY;
+               std::string bindata;
+               int hop1, hop2, hop3;
+               int len1, len2;
+               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;
+                       len1 = 6;
+                       len2 = 4;
+                       // pfx s1.s2.s3. (xxxx.xxxx or s4) sfx
+                       //     6  4  4    9/6
+                       rv.reserve(prefix.length() + 26 + suffix.length());
+               }
+               else
+               {
+                       bindata = std::string((const char*)&ip.in4.sin_addr, 4);
+                       hop1 = 3;
+                       hop2 = 0;
+                       hop3 = 2;
+                       len1 = len2 = 3;
+                       // pfx s1.s2. (xxx.xxx or s3) sfx
+                       rv.reserve(prefix.length() + 15 + suffix.length());
+               }
 
-               /* For remote clients, we dont take any action, we just allow it.
-                * The local server where they are will set their cloak instead.
-                */
-               if (!IS_LOCAL(dest))
-                       return MODEACTION_ALLOW;
+               rv.append(prefix);
+               rv.append(SegmentCloak(bindata, 10, len1));
+               rv.append(1, '.');
+               bindata.erase(hop1);
+               rv.append(SegmentCloak(bindata, 11, len2));
+               if (hop2)
+               {
+                       rv.append(1, '.');
+                       bindata.erase(hop2);
+                       rv.append(SegmentCloak(bindata, 12, len2));
+               }
 
-               if (adding)
+               if (full)
                {
-                       if(!dest->IsModeSet('x'))
-                       {
-                               /* The mode is being turned on - so attempt to
-                                * allocate the user a cloaked host using a non-reversible
-                                * algorithm (its simple, but its non-reversible so the
-                                * simplicity doesnt really matter). This algorithm
-                                * will not work if the user has only one level of domain
-                                * 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,':'))
-                               {
-                                       /* 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 + "-" + std::string(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
-                                       {
-                                               // else, they have an ip
-                                               b = std::string(ra) + "." + prefix + ".cloak";
-                                       }
-                                       ServerInstance->Log(DEBUG,"cloak: allocated "+b);
-                                       dest->ChangeDisplayedHost(b.c_str());
-                               }
-                               
-                               dest->SetMode('x',true);
-                               return MODEACTION_ALLOW;
-                       }
+                       rv.append(1, '.');
+                       bindata.erase(hop3);
+                       rv.append(SegmentCloak(bindata, 13, 6));
+                       rv.append(suffix);
                }
                else
-               {
-                       if (dest->IsModeSet('x'))
+               {
+                       if (ip.sa.sa_family == AF_INET6)
                        {
-                               /* User is removing the mode, so just restore their real host
-                                * and make it match the displayed one.
-                                */
-                               dest->ChangeDisplayedHost(dest->host);
-                               dest->SetMode('x',false);
-                               return MODEACTION_ALLOW;
+                               rv.append(InspIRCd::Format(".%02x%02x.%02x%02x%s",
+                                       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], suffix.c_str()));
+                       }
+                       else
+                       {
+                               const unsigned char* ip4 = (const unsigned char*)&ip.in4.sin_addr;
+                               rv.append(InspIRCd::Format(".%d.%d%s", ip4[1], ip4[0], suffix.c_str()));
                        }
                }
-
-               return MODEACTION_DENY;
+               return rv;
        }
-       
-       void DoRehash()
+
+       ModResult OnCheckBan(User* user, Channel* chan, const std::string& mask) CXX11_OVERRIDE
        {
-               ConfigReader Conf(ServerInstance);
-               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);
-               
-               prefix = Conf.ReadValue("cloak","prefix",0);
-               if (prefix == "")
-               {
-                       prefix = ServerInstance->Config->Network;
-               }
-               if (!key1 && !key2 && !key3 && !key4)
+               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)
                {
-                       ModuleException ex("You have not defined cloak keys for m_cloaking!!! THIS IS INSECURE AND SHOULD BE CHECKED!");
-                       throw (ex);
+                       const std::string cloakMask = user->nick + "!" + user->ident + "@" + *cloak;
+                       if (InspIRCd::Match(cloakMask, mask))
+                               return MOD_RES_DENY;
                }
+               return MOD_RES_PASSTHRU;
        }
-};
 
-
-class ModuleCloaking : public Module
-{
- private:
-       
-       CloakUser* cu;
-
- public:
-       ModuleCloaking(InspIRCd* Me)
-               : Module::Module(Me)
+       void Prioritize()
        {
-               /* Create new mode handler object */
-               cu = new CloakUser(ServerInstance);
-
-               /* Register it with the core */         
-               ServerInstance->AddMode(cu, 'x');
-
-               OnRehash("");
+               /* Needs to be after m_banexception etc. */
+               ServerInstance->Modules->SetPriority(this, I_OnCheckBan, PRIORITY_LAST);
        }
-       
-       virtual ~ModuleCloaking()
+
+       // this unsets umode +x on every host change. If we are actually doing a +x
+       // mode change, we will call SetMode back to true AFTER the host change is done.
+       void OnChangeHost(User* u, const std::string& host) CXX11_OVERRIDE
        {
-               ServerInstance->Modes->DelMode(cu);
-               DELETE(cu);
+               if (u->IsModeSet(cu))
+               {
+                       u->SetMode(cu, false);
+                       u->WriteCommand("MODE", "-" + ConvToStr(cu.GetModeChar()));
+               }
        }
-       
-       virtual Version GetVersion()
+
+       Version GetVersion() CXX11_OVERRIDE
        {
-               // returns the version number of the module to be
-               // listed in /MODULES
-               return Version(1,0,0,2,VF_COMMON|VF_VENDOR);
+               std::string testcloak = "broken";
+               if (Hash)
+               {
+                       switch (mode)
+                       {
+                               case MODE_HALF_CLOAK:
+                                       testcloak = prefix + SegmentCloak("*", 3, 8) + suffix;
+                                       break;
+                               case MODE_OPAQUE:
+                                       testcloak = prefix + SegmentCloak("*", 4, 8) + suffix;
+                       }
+               }
+               return Version("Provides masking of user hostnames", VF_COMMON|VF_VENDOR, testcloak);
        }
 
-       virtual void OnRehash(const std::string &parameter)
+       void ReadConfig(ConfigStatus& status) CXX11_OVERRIDE
        {
-               cu->DoRehash();
+               ConfigTag* tag = ServerInstance->Config->ConfValue("cloak");
+               prefix = tag->getString("prefix");
+               suffix = tag->getString("suffix", ".IP");
+
+               std::string modestr = tag->getString("mode");
+               if (modestr == "half")
+                       mode = MODE_HALF_CLOAK;
+               else if (modestr == "full")
+                       mode = MODE_OPAQUE;
+               else
+                       throw ModuleException("Bad value for <cloak:mode>; must be half or full");
+
+               key = tag->getString("key");
+               if (key.empty() || key == "secret")
+                       throw ModuleException("You have not defined cloak keys for m_cloaking. Define <cloak:key> as a network-wide secret.");
        }
 
-       void Implements(char* List)
+       std::string GenCloak(const irc::sockets::sockaddrs& ip, const std::string& ipstr, const std::string& host)
        {
-               List[I_OnRehash] = List[I_OnUserConnect] = 1;
+               std::string chost;
+
+               switch (mode)
+               {
+                       case MODE_HALF_CLOAK:
+                       {
+                               if (ipstr != host)
+                                       chost = prefix + SegmentCloak(host, 1, 6) + LastTwoDomainParts(host);
+                               if (chost.empty() || chost.length() > 50)
+                                       chost = SegmentIP(ip, false);
+                               break;
+                       }
+                       case MODE_OPAQUE:
+                       default:
+                               chost = SegmentIP(ip, true);
+               }
+               return chost;
        }
 
-       virtual void OnUserConnect(userrec* user)
+       void OnUserConnect(LocalUser* dest) CXX11_OVERRIDE
        {
-               // Heres the weird bit. When a user connects we must set +x on them, so
-               // we're going to use the SendMode method of the Server class to send
-               // the mode to the client. This is basically the same as sending an
-               // SAMODE in unreal. Note that to the user it will appear as if they set
-               // the mode on themselves.
-               
-               const char* modes[2];           // only two parameters
-               modes[0] = user->nick;          // first parameter is the nick
-               modes[1] = "+x";                // second parameter is the mode
-               ServerInstance->SendMode(modes,2,user); // send these, forming the command "MODE <nick> +x"
-       }
+               std::string* cloak = cu.ext.get(dest);
+               if (cloak)
+                       return;
 
+               cu.ext.set(dest, GenCloak(dest->client_sa, dest->GetIPString(), dest->host));
+       }
 };
 
-// stuff down here is the module-factory stuff. For basic modules you can ignore this.
-
-class ModuleCloakingFactory : public ModuleFactory
+CmdResult CommandCloak::Handle(const std::vector<std::string> &parameters, User *user)
 {
- public:
-       ModuleCloakingFactory()
-       {
-       }
-       
-       ~ModuleCloakingFactory()
-       {
-       }
-       
-       virtual Module * CreateModule(InspIRCd* Me)
-       {
-               return new ModuleCloaking(Me);
-       }
-       
-};
+       ModuleCloaking* mod = (ModuleCloaking*)(Module*)creator;
+       irc::sockets::sockaddrs sa;
+       std::string cloak;
 
+       if (irc::sockets::aptosa(parameters[0], 0, sa))
+               cloak = mod->GenCloak(sa, parameters[0], parameters[0]);
+       else
+               cloak = mod->GenCloak(sa, "", parameters[0]);
 
-extern "C" void * init_module( void )
-{
-       return new ModuleCloakingFactory;
+       user->WriteNotice("*** Cloak for " + parameters[0] + " is " + cloak);
+
+       return CMD_SUCCESS;
 }
+
+MODULE_INIT(ModuleCloaking)