src/modules/m_md5.cpp

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