src/modules/m_md5.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
   6  *                       E-mail:
   7  *                <brain@chatspike.net>
   8  *                <Craig@chatspike.net>
   9  *
  10  * Written by Craig Edwards, Craig McLure, and others.
  11  * This program is free but copyrighted software; see
  12  *            the file COPYING for details.
  13  *
  14  * ---------------------------------------------------
  15  */
  16
  17 /* $ModDesc: Allows for MD5 encrypted oper passwords */
  18 /* $ModDep: m_md5.h */
  19
  20 using namespace std;
  21
  22 #include "inspircd_config.h"
  23 #ifdef HAS_STDINT
  24 #include <stdint.h>
  25 #endif
  26 #include "users.h"
  27 #include "channels.h"
  28 #include "modules.h"
  29 #include "inspircd.h"
  30
  31 #include "m_md5.h"
  32
  33 /* The four core functions - F1 is optimized somewhat */
  34 #define F1(x, y, z) (z ^ (x & (y ^ z)))
  35 #define F2(x, y, z) F1(z, x, y)
  36 #define F3(x, y, z) (x ^ y ^ z)
  37 #define F4(x, y, z) (y ^ (x | ~z))
  38
  39 /* This is the central step in the MD5 algorithm. */
  40 #define MD5STEP(f,w,x,y,z,in,s) \
  41          (w += f(x,y,z) + in, w = (w<<s | w>>(32-s)) + x)
  42
  43 #ifndef HAS_STDINT
  44 typedef unsigned int uint32_t;
  45 #endif
  46
  47 typedef uint32_t word32; /* NOT unsigned long. We don't support 16 bit platforms, anyway. */
  48 typedef unsigned char byte;
  49
  50 /** An MD5 context, used by m_opermd5
  51  */
  52 class MD5Context : public classbase
  53 {
  54  public:
  55         word32 buf[4];
  56         word32 bytes[2];
  57         word32 in[16];
  58 };
  59
  60 class ModuleMD5 : public Module
  61 {
  62         void byteSwap(word32 *buf, unsigned words)
  63         {
  64                 byte *p = (byte *)buf;
  65
  66                 do
  67                 {
  68                         *buf++ = (word32)((unsigned)p[3] << 8 | p[2]) << 16 |
  69                                 ((unsigned)p[1] << 8 | p[0]);
  70                         p += 4;
  71                 } while (--words);
  72         }
  73
  74         /* XXX - maybe if we had an md5/encryption moduletype? *shrug* */
  75         void MD5Init(struct MD5Context *ctx, unsigned int* key = NULL)
  76         {
  77                 /* These are the defaults for md5 */
  78                 if (!key)
  79                 {
  80                         ctx->buf[0] = 0x67452301;
  81                         ctx->buf[1] = 0xefcdab89;
  82                         ctx->buf[2] = 0x98badcfe;
  83                         ctx->buf[3] = 0x10325476;
  84                 }
  85                 else
  86                 {
  87                         ctx->buf[0] = key[0];
  88                         ctx->buf[1] = key[1];
  89                         ctx->buf[2] = key[2];
  90                         ctx->buf[3] = key[3];
  91                 }
  92
  93                 ctx->bytes[0] = 0;
  94                 ctx->bytes[1] = 0;
  95         }
  96
  97         void MD5Update(struct MD5Context *ctx, byte const *buf, int len)
  98         {
  99                 word32 t;
 100
 101                 /* Update byte count */
 102
 103                 t = ctx->bytes[0];
 104                 if ((ctx->bytes[0] = t + len) < t)
 105                         ctx->bytes[1]++;        /* Carry from low to high */
 106
 107                 t = 64 - (t & 0x3f);    /* Space available in ctx->in (at least 1) */
 108                 if ((unsigned)t > (unsigned)len)
 109                 {
 110                         memcpy((byte *)ctx->in + 64 - (unsigned)t, buf, len);
 111                         return;
 112                 }
 113                 /* First chunk is an odd size */
 114                 memcpy((byte *)ctx->in + 64 - (unsigned)t, buf, (unsigned)t);
 115                 byteSwap(ctx->in, 16);
 116                 MD5Transform(ctx->buf, ctx->in);
 117                 buf += (unsigned)t;
 118                 len -= (unsigned)t;
 119
 120                 /* Process data in 64-byte chunks */
 121                 while (len >= 64)
 122                 {
 123                         memcpy(ctx->in, buf, 64);
 124                         byteSwap(ctx->in, 16);
 125                         MD5Transform(ctx->buf, ctx->in);
 126                         buf += 64;
 127                         len -= 64;
 128                 }
 129
 130                 /* Handle any remaining bytes of data. */
 131                 memcpy(ctx->in, buf, len);
 132         }
 133
 134         void MD5Final(byte digest[16], struct MD5Context *ctx)
 135         {
 136                 int count = (int)(ctx->bytes[0] & 0x3f); /* Bytes in ctx->in */
 137                 byte *p = (byte *)ctx->in + count;      /* First unused byte */
 138
 139                 /* Set the first char of padding to 0x80.  There is always room. */
 140                 *p++ = 0x80;
 141
 142                 /* Bytes of padding needed to make 56 bytes (-8..55) */
 143                 count = 56 - 1 - count;
 144
 145                 if (count < 0)
 146                 {       /* Padding forces an extra block */
 147                         memset(p, 0, count+8);
 148                         byteSwap(ctx->in, 16);
 149                         MD5Transform(ctx->buf, ctx->in);
 150                         p = (byte *)ctx->in;
 151                         count = 56;
 152                 }
 153                 memset(p, 0, count+8);
 154                 byteSwap(ctx->in, 14);
 155
 156                 /* Append length in bits and transform */
 157                 ctx->in[14] = ctx->bytes[0] << 3;
 158                 ctx->in[15] = ctx->bytes[1] << 3 | ctx->bytes[0] >> 29;
 159                 MD5Transform(ctx->buf, ctx->in);
 160
 161                 byteSwap(ctx->buf, 4);
 162                 memcpy(digest, ctx->buf, 16);
 163                 memset(ctx, 0, sizeof(ctx));
 164         }
 165
 166         void MD5Transform(word32 buf[4], word32 const in[16])
 167         {
 168                 register word32 a, b, c, d;
 169
 170                 a = buf[0];
 171                 b = buf[1];
 172                 c = buf[2];
 173                 d = buf[3];
 174
 175                 MD5STEP(F1, a, b, c, d, in[0] + 0xd76aa478, 7);
 176                 MD5STEP(F1, d, a, b, c, in[1] + 0xe8c7b756, 12);
 177                 MD5STEP(F1, c, d, a, b, in[2] + 0x242070db, 17);
 178                 MD5STEP(F1, b, c, d, a, in[3] + 0xc1bdceee, 22);
 179                 MD5STEP(F1, a, b, c, d, in[4] + 0xf57c0faf, 7);
 180                 MD5STEP(F1, d, a, b, c, in[5] + 0x4787c62a, 12);
 181                 MD5STEP(F1, c, d, a, b, in[6] + 0xa8304613, 17);
 182                 MD5STEP(F1, b, c, d, a, in[7] + 0xfd469501, 22);
 183                 MD5STEP(F1, a, b, c, d, in[8] + 0x698098d8, 7);
 184                 MD5STEP(F1, d, a, b, c, in[9] + 0x8b44f7af, 12);
 185                 MD5STEP(F1, c, d, a, b, in[10] + 0xffff5bb1, 17);
 186                 MD5STEP(F1, b, c, d, a, in[11] + 0x895cd7be, 22);
 187                 MD5STEP(F1, a, b, c, d, in[12] + 0x6b901122, 7);
 188                 MD5STEP(F1, d, a, b, c, in[13] + 0xfd987193, 12);
 189                 MD5STEP(F1, c, d, a, b, in[14] + 0xa679438e, 17);
 190                 MD5STEP(F1, b, c, d, a, in[15] + 0x49b40821, 22);
 191
 192                 MD5STEP(F2, a, b, c, d, in[1] + 0xf61e2562, 5);
 193                 MD5STEP(F2, d, a, b, c, in[6] + 0xc040b340, 9);
 194                 MD5STEP(F2, c, d, a, b, in[11] + 0x265e5a51, 14);
 195                 MD5STEP(F2, b, c, d, a, in[0] + 0xe9b6c7aa, 20);
 196                 MD5STEP(F2, a, b, c, d, in[5] + 0xd62f105d, 5);
 197                 MD5STEP(F2, d, a, b, c, in[10] + 0x02441453, 9);
 198                 MD5STEP(F2, c, d, a, b, in[15] + 0xd8a1e681, 14);
 199                 MD5STEP(F2, b, c, d, a, in[4] + 0xe7d3fbc8, 20);
 200                 MD5STEP(F2, a, b, c, d, in[9] + 0x21e1cde6, 5);
 201                 MD5STEP(F2, d, a, b, c, in[14] + 0xc33707d6, 9);
 202                 MD5STEP(F2, c, d, a, b, in[3] + 0xf4d50d87, 14);
 203                 MD5STEP(F2, b, c, d, a, in[8] + 0x455a14ed, 20);
 204                 MD5STEP(F2, a, b, c, d, in[13] + 0xa9e3e905, 5);
 205                 MD5STEP(F2, d, a, b, c, in[2] + 0xfcefa3f8, 9);
 206                 MD5STEP(F2, c, d, a, b, in[7] + 0x676f02d9, 14);
 207                 MD5STEP(F2, b, c, d, a, in[12] + 0x8d2a4c8a, 20);
 208
 209                 MD5STEP(F3, a, b, c, d, in[5] + 0xfffa3942, 4);
 210                 MD5STEP(F3, d, a, b, c, in[8] + 0x8771f681, 11);
 211                 MD5STEP(F3, c, d, a, b, in[11] + 0x6d9d6122, 16);
 212                 MD5STEP(F3, b, c, d, a, in[14] + 0xfde5380c, 23);
 213                 MD5STEP(F3, a, b, c, d, in[1] + 0xa4beea44, 4);
 214                 MD5STEP(F3, d, a, b, c, in[4] + 0x4bdecfa9, 11);
 215                 MD5STEP(F3, c, d, a, b, in[7] + 0xf6bb4b60, 16);
 216                 MD5STEP(F3, b, c, d, a, in[10] + 0xbebfbc70, 23);
 217                 MD5STEP(F3, a, b, c, d, in[13] + 0x289b7ec6, 4);
 218                 MD5STEP(F3, d, a, b, c, in[0] + 0xeaa127fa, 11);
 219                 MD5STEP(F3, c, d, a, b, in[3] + 0xd4ef3085, 16);
 220                 MD5STEP(F3, b, c, d, a, in[6] + 0x04881d05, 23);
 221                 MD5STEP(F3, a, b, c, d, in[9] + 0xd9d4d039, 4);
 222                 MD5STEP(F3, d, a, b, c, in[12] + 0xe6db99e5, 11);
 223                 MD5STEP(F3, c, d, a, b, in[15] + 0x1fa27cf8, 16);
 224                 MD5STEP(F3, b, c, d, a, in[2] + 0xc4ac5665, 23);
 225
 226                 MD5STEP(F4, a, b, c, d, in[0] + 0xf4292244, 6);
 227                 MD5STEP(F4, d, a, b, c, in[7] + 0x432aff97, 10);
 228                 MD5STEP(F4, c, d, a, b, in[14] + 0xab9423a7, 15);
 229                 MD5STEP(F4, b, c, d, a, in[5] + 0xfc93a039, 21);
 230                 MD5STEP(F4, a, b, c, d, in[12] + 0x655b59c3, 6);
 231                 MD5STEP(F4, d, a, b, c, in[3] + 0x8f0ccc92, 10);
 232                 MD5STEP(F4, c, d, a, b, in[10] + 0xffeff47d, 15);
 233                 MD5STEP(F4, b, c, d, a, in[1] + 0x85845dd1, 21);
 234                 MD5STEP(F4, a, b, c, d, in[8] + 0x6fa87e4f, 6);
 235                 MD5STEP(F4, d, a, b, c, in[15] + 0xfe2ce6e0, 10);
 236                 MD5STEP(F4, c, d, a, b, in[6] + 0xa3014314, 15);
 237                 MD5STEP(F4, b, c, d, a, in[13] + 0x4e0811a1, 21);
 238                 MD5STEP(F4, a, b, c, d, in[4] + 0xf7537e82, 6);
 239                 MD5STEP(F4, d, a, b, c, in[11] + 0xbd3af235, 10);
 240                 MD5STEP(F4, c, d, a, b, in[2] + 0x2ad7d2bb, 15);
 241                 MD5STEP(F4, b, c, d, a, in[9] + 0xeb86d391, 21);
 242
 243                 buf[0] += a;
 244                 buf[1] += b;
 245                 buf[2] += c;
 246                 buf[3] += d;
 247         }
 248
 249
 250         void MyMD5(void *dest, void *orig, int len, unsigned int* key)
 251         {
 252                 MD5Context context;
 253                 MD5Init(&context, key);
 254                 MD5Update(&context, (const unsigned char*)orig, len);
 255                 MD5Final((unsigned char*)dest, &context);
 256         }
 257
 258
 259         void GenHash(const char* src, char* dest, const char* xtab, unsigned int* key)
 260         {
 261                 unsigned char bytes[16];
 262
 263                 MyMD5((char*)bytes, (void*)src, strlen(src), key);
 264
 265                 for (int i = 0; i < 16; i++)
 266                 {
 267                         *dest++ = xtab[bytes[i] / 16];
 268                         *dest++ = xtab[bytes[i] % 16];
 269                 }
 270                 *dest++ = 0;
 271         }
 272
 273         unsigned int *key;
 274         char* chars;
 275
 276  public:
 277
 278         ModuleMD5(InspIRCd* Me)
 279                 : Module::Module(Me), key(NULL), chars(NULL)
 280         {
 281         }
 282
 283         virtual ~ModuleMD5()
 284         {
 285         }
 286
 287         void Implements(char* List)
 288         {
 289                 List[I_OnRequest] = 1;
 290         }
 291
 292         virtual char* OnRequest(Request* request)
 293         {
 294                 MD5Request* MD5 = (MD5Request*)request;
 295                 if (strcmp("MD5_KEY", request->GetId()) == 0)
 296                 {
 297                         this->key = (unsigned int*)MD5->GetKeyData();
 298                 }
 299                 else if (strcmp("MD5_HEX", request->GetId()) == 0)
 300                 {
 301                         this->chars = (char*)MD5->GetOutputs();
 302                 }
 303                 else if (strcmp("MD5_SUM", request->GetId()) == 0)
 304                 {
 305                         static char data[MAXBUF];
 306                         GenHash((const char*)MD5->GetHashData(), data, chars ? chars : "0123456789abcdef", key);
 307                         return data;
 308                 }
 309                 else if (strcmp("MD5_RESET", request->GetId()) == 0)
 310                 {
 311                         this->chars = NULL;
 312                         this->key = NULL;
 313                 }
 314                 return NULL;
 315         }
 316
 317         virtual Version GetVersion()
 318         {
 319                 return Version(1,1,0,1,VF_VENDOR,API_VERSION);
 320         }
 321 };
 322
 323
 324 class ModuleMD5Factory : public ModuleFactory
 325 {
 326  public:
 327         ModuleMD5Factory()
 328         {
 329         }
 330
 331         ~ModuleMD5Factory()
 332         {
 333         }
 334
 335         virtual Module * CreateModule(InspIRCd* Me)
 336         {
 337                 return new ModuleMD5(Me);
 338         }
 339
 340 };
 341
 342
 343 extern "C" void * init_module( void )
 344 {
 345         return new ModuleMD5Factory;
 346 }