src/modules/m_sha256.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 /* m_sha256 - Based on m_opersha256 written by Special <john@yarbbles.com>
  15  * Modified and improved by Craig Edwards, December 2006.
  16  *
  17  *
  18  * FIPS 180-2 SHA-224/256/384/512 implementation
  19  * Last update: 05/23/2005
  20  * Issue date:  04/30/2005
  21  *
  22  * Copyright (C) 2005 Olivier Gay <olivier.gay@a3.epfl.ch>
  23  * All rights reserved.
  24  *
  25  * Redistribution and use in source and binary forms, with or without
  26  * modification, are permitted provided that the following conditions
  27  * are met:
  28  * 1. Redistributions of source code must retain the above copyright
  29  *    notice, this list of conditions and the following disclaimer.
  30  * 2. Redistributions in binary form must reproduce the above copyright
  31  *    notice, this list of conditions and the following disclaimer in the
  32  *    documentation and/or other materials provided with the distribution.
  33  * 3. Neither the name of the project nor the names of its contributors
  34  *    may be used to endorse or promote products derived from this software
  35  *    without specific prior written permission.
  36  *
  37  * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
  38  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  39  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  40  * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
  41  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  42  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  43  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  44  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  45  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  46  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  47  * SUCH DAMAGE.
  48  */
  49
  50 /* $ModDesc: Allows for SHA-256 encrypted oper passwords */
  51 /* $ModDep: m_hash.h */
  52
  53 #include "inspircd.h"
  54 #ifdef HAS_STDINT
  55 #include <stdint.h>
  56 #endif
  57 #include "users.h"
  58 #include "channels.h"
  59 #include "modules.h"
  60 #include "m_hash.h"
  61
  62 #ifndef HAS_STDINT
  63 typedef unsigned int uint32_t;
  64 #endif
  65
  66 /** An sha 256 context, used by m_opersha256
  67  */
  68 class SHA256Context : public classbase
  69 {
  70  public:
  71         unsigned int tot_len;
  72         unsigned int len;
  73         unsigned char block[2 * SHA256_BLOCK_SIZE];
  74         uint32_t h[8];
  75 };
  76
  77 #define SHFR(x, n)    (x >> n)
  78 #define ROTR(x, n)   ((x >> n) | (x << ((sizeof(x) << 3) - n)))
  79 #define ROTL(x, n)   ((x << n) | (x >> ((sizeof(x) << 3) - n)))
  80 #define CH(x, y, z)  ((x & y) ^ (~x & z))
  81 #define MAJ(x, y, z) ((x & y) ^ (x & z) ^ (y & z))
  82
  83 #define SHA256_F1(x) (ROTR(x,  2) ^ ROTR(x, 13) ^ ROTR(x, 22))
  84 #define SHA256_F2(x) (ROTR(x,  6) ^ ROTR(x, 11) ^ ROTR(x, 25))
  85 #define SHA256_F3(x) (ROTR(x,  7) ^ ROTR(x, 18) ^ SHFR(x,  3))
  86 #define SHA256_F4(x) (ROTR(x, 17) ^ ROTR(x, 19) ^ SHFR(x, 10))
  87
  88 #define UNPACK32(x, str)                       \
  89 {                                             \
  90         *((str) + 3) = (uint8_t) ((x)      );      \
  91         *((str) + 2) = (uint8_t) ((x) >>  8);      \
  92         *((str) + 1) = (uint8_t) ((x) >> 16);      \
  93         *((str) + 0) = (uint8_t) ((x) >> 24);      \
  94 }
  95
  96 #define PACK32(str, x)                   \
  97 {                                             \
  98         *(x) = ((uint32_t) *((str) + 3)      )     \
  99         | ((uint32_t) *((str) + 2) <<  8)     \
 100         | ((uint32_t) *((str) + 1) << 16)     \
 101         | ((uint32_t) *((str) + 0) << 24);    \
 102 }
 103
 104 /* Macros used for loops unrolling */
 105
 106 #define SHA256_SCR(i)                     \
 107 {                                             \
 108         w[i] =  SHA256_F4(w[i - 2]) + w[i - 7]     \
 109         + SHA256_F3(w[i - 15]) + w[i - 16];  \
 110 }
 111
 112 const unsigned int sha256_h0[8] =
 113 {
 114         0x6a09e667, 0xbb67ae85, 0x3c6ef372, 0xa54ff53a,
 115         0x510e527f, 0x9b05688c, 0x1f83d9ab, 0x5be0cd19
 116 };
 117
 118 uint32_t sha256_k[64] =
 119 {
 120         0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5,
 121         0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
 122         0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3,
 123         0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
 124         0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc,
 125         0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
 126         0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7,
 127         0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
 128         0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13,
 129         0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
 130         0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3,
 131         0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
 132         0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5,
 133         0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
 134         0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208,
 135         0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2
 136 };
 137
 138 class ModuleSHA256 : public Module
 139 {
 140         void SHA256Init(SHA256Context *ctx, const unsigned int* key)
 141         {
 142                 if (key)
 143                 {
 144                         for (int i = 0; i < 8; i++)
 145                                 ctx->h[i] = key[i];
 146                 }
 147                 else
 148                 {
 149                         for (int i = 0; i < 8; i++)
 150                                 ctx->h[i] = sha256_h0[i];
 151                 }
 152                 ctx->len = 0;
 153                 ctx->tot_len = 0;
 154         }
 155
 156         void SHA256Transform(SHA256Context *ctx, unsigned char *message, unsigned int block_nb)
 157         {
 158                 uint32_t w[64];
 159                 uint32_t wv[8];
 160                 unsigned char *sub_block;
 161                 for (unsigned int i = 1; i <= block_nb; i++)
 162                 {
 163                         int j;
 164                         sub_block = message + ((i - 1) << 6);
 165
 166                         for (j = 0; j < 16; j++)
 167                                 PACK32(&sub_block[j << 2], &w[j]);
 168                         for (j = 16; j < 64; j++)
 169                                 SHA256_SCR(j);
 170                         for (j = 0; j < 8; j++)
 171                                 wv[j] = ctx->h[j];
 172                         for (j = 0; j < 64; j++)
 173                         {
 174                                 uint32_t t1 = wv[7] + SHA256_F2(wv[4]) + CH(wv[4], wv[5], wv[6]) + sha256_k[j] + w[j];
 175                                 uint32_t t2 = SHA256_F1(wv[0]) + MAJ(wv[0], wv[1], wv[2]);
 176                                 wv[7] = wv[6];
 177                                 wv[6] = wv[5];
 178                                 wv[5] = wv[4];
 179                                 wv[4] = wv[3] + t1;
 180                                 wv[3] = wv[2];
 181                                 wv[2] = wv[1];
 182                                 wv[1] = wv[0];
 183                                 wv[0] = t1 + t2;
 184                         }
 185                         for (j = 0; j < 8; j++)
 186                                 ctx->h[j] += wv[j];
 187                 }
 188         }
 189
 190         void SHA256Update(SHA256Context *ctx, unsigned char *message, unsigned int len)
 191         {
 192                 unsigned int rem_len = SHA256_BLOCK_SIZE - ctx->len;
 193                 memcpy(&ctx->block[ctx->len], message, rem_len);
 194                 if (ctx->len + len < SHA256_BLOCK_SIZE)
 195                 {
 196                         ctx->len += len;
 197                         return;
 198                 }
 199                 unsigned int new_len = len - rem_len;
 200                 unsigned int block_nb = new_len / SHA256_BLOCK_SIZE;
 201                 unsigned char *shifted_message = message + rem_len;
 202                 SHA256Transform(ctx, ctx->block, 1);
 203                 SHA256Transform(ctx, shifted_message, block_nb);
 204                 rem_len = new_len % SHA256_BLOCK_SIZE;
 205                 memcpy(ctx->block, &shifted_message[block_nb << 6],rem_len);
 206                 ctx->len = rem_len;
 207                 ctx->tot_len += (block_nb + 1) << 6;
 208         }
 209
 210         void SHA256Final(SHA256Context *ctx, unsigned char *digest)
 211         {
 212                 unsigned int block_nb = (1 + ((SHA256_BLOCK_SIZE - 9) < (ctx->len % SHA256_BLOCK_SIZE)));
 213                 unsigned int len_b = (ctx->tot_len + ctx->len) << 3;
 214                 unsigned int pm_len = block_nb << 6;
 215                 memset(ctx->block + ctx->len, 0, pm_len - ctx->len);
 216                 ctx->block[ctx->len] = 0x80;
 217                 UNPACK32(len_b, ctx->block + pm_len - 4);
 218                 SHA256Transform(ctx, ctx->block, block_nb);
 219                 for (int i = 0 ; i < 8; i++)
 220                         UNPACK32(ctx->h[i], &digest[i << 2]);
 221         }
 222
 223         void SHA256(const char *src, char *dest, int len, const char* hxc, const unsigned int* key = NULL)
 224         {
 225                 // Generate the hash
 226                 unsigned char bytehash[SHA256_DIGEST_SIZE];
 227                 SHA256Context ctx;
 228                 SHA256Init(&ctx, key);
 229                 SHA256Update(&ctx, (unsigned char *)src, (unsigned int)len);
 230                 SHA256Final(&ctx, bytehash);
 231                 // Convert it to hex
 232                 for (int i = 0, j = 0; i < SHA256_DIGEST_SIZE; i++)
 233                 {
 234                         dest[j++] = hxc[bytehash[i] / 16];
 235                         dest[j++] = hxc[bytehash[i] % 16];
 236                         dest[j] = '\0';
 237                 }
 238         }
 239
 240         unsigned int* key;
 241         char* chars;
 242
 243  public:
 244
 245         ModuleSHA256(InspIRCd* Me) : Module(Me), key(NULL), chars(NULL)
 246         {
 247                 ServerInstance->PublishInterface("HashRequest", this);
 248         }
 249
 250         virtual ~ModuleSHA256()
 251         {
 252                 ServerInstance->UnpublishInterface("HashRequest", this);
 253         }
 254
 255         void Implements(char *List)
 256         {
 257                 List[I_OnRequest] = 1;
 258         }
 259
 260         virtual char* OnRequest(Request* request)
 261         {
 262                 HashRequest* SHA = (HashRequest*)request;
 263                 if (strcmp("KEY", request->GetId()) == 0)
 264                 {
 265                         this->key = (unsigned int*)SHA->GetKeyData();
 266                 }
 267                 else if (strcmp("HEX", request->GetId()) == 0)
 268                 {
 269                         this->chars = (char*)SHA->GetOutputs();
 270                 }
 271                 else if (strcmp("SUM", request->GetId()) == 0)
 272                 {
 273                         static char data[MAXBUF];
 274                         SHA256((const char*)SHA->GetHashData(), data, strlen(SHA->GetHashData()), chars ? chars : "0123456789abcdef", key);
 275                         return data;
 276                 }
 277                 else if (strcmp("NAME", request->GetId()) == 0)
 278                 {
 279                         return "sha256";
 280                 }
 281                 else if (strcmp("RESET", request->GetId()) == 0)
 282                 {
 283                         this->chars = NULL;
 284                         this->key = NULL;
 285                 }
 286                 return NULL;
 287         }
 288
 289         virtual Version GetVersion()
 290         {
 291                 return Version(1, 1, 0, 1, VF_VENDOR|VF_SERVICEPROVIDER, API_VERSION);
 292         }
 293 };
 294
 295
 296 class ModuleSHA256Factory : public ModuleFactory
 297 {
 298 public:
 299         ModuleSHA256Factory()
 300         {
 301         }
 302
 303         ~ModuleSHA256Factory()
 304         {
 305         }
 306
 307         virtual Module *CreateModule(InspIRCd* Me)
 308         {
 309                 return new ModuleSHA256(Me);
 310         }
 311
 312 };
 313
 314 extern "C" DllExport void * init_module( void )
 315 {
 316         return new ModuleSHA256Factory;
 317 }