bertos/sec/hash/md5.c

   1 /*
   2  * This code implements the MD5 message-digest algorithm.
   3  * The algorithm is due to Ron Rivest.  This code was
   4  * written by Colin Plumb in 1993, no copyright is claimed.
   5  * This code is in the public domain; do with it what you wish.
   6  *
   7  * Equivalent code is available from RSA Data Security, Inc.
   8  * This code has been tested against that, and is equivalent,
   9  * except that you don't need to include two pages of legalese
  10  * with every copy.
  11  *
  12  * To compute the message digest of a chunk of bytes, declare an
  13  * MD5Context structure, pass it to MD5Init, call MD5Update as
  14  * needed on buffers full of bytes, and then call MD5Final, which
  15  * will fill a supplied 16-byte array with the digest.
  16  */
  17
  18 #include "md5.h"
  19 #include <sec/util.h>
  20 #include <cpu/byteorder.h>
  21 #include <string.h>
  22
  23 static void MD5Transform(uint32_t buf[4], uint32_t in[16]);
  24
  25 static void byteReverse(uint32_t *buf, unsigned longs)
  26 {
  27     do {
  28                 *buf = le32_to_cpu(*buf);
  29         ++buf;
  30     } while (--longs);
  31 }
  32
  33 /*
  34  * Start MD5 accumulation.  Set bit count to 0 and buffer to mysterious
  35  * initialization constants.
  36  */
  37 static void MD5_begin(Hash *h)
  38 {
  39         MD5_Context *ctx = (MD5_Context *)h;
  40
  41     ctx->buf[0] = 0x67452301;
  42     ctx->buf[1] = 0xefcdab89;
  43     ctx->buf[2] = 0x98badcfe;
  44     ctx->buf[3] = 0x10325476;
  45
  46         ctx->bits = 0;
  47 }
  48
  49 /*
  50  * Update context to reflect the concatenation of another buffer full
  51  * of bytes.
  52  */
  53 static void MD5_update(Hash *h, const void* vbuf, size_t len)
  54 {
  55         MD5_Context *ctx = (MD5_Context *)h;
  56         const char *buf = (const char *)vbuf;
  57         uint32_t *aligned_ptr = NULL;
  58         if (is_aligned(ctx->in, sizeof(uint32_t)))
  59                 aligned_ptr = (uint32_t *)((size_t)ctx->in);
  60         else
  61                 ASSERT2(0, "Unaligned memory");
  62     uint32_t t;
  63
  64     /* Update bitcount */
  65     t = (ctx->bits >> 3) & 0x3f;        /* Bytes already in shsInfo->data */
  66         ctx->bits += len*8;
  67
  68     /* Handle any leading odd-sized chunks */
  69
  70     if (t) {
  71         uint8_t *p = (uint8_t*) ctx->in + t;
  72
  73         t = 64 - t;
  74         if (len < t) {
  75             memcpy(p, buf, len);
  76             return;
  77         }
  78         memcpy(p, buf, t);
  79         byteReverse(aligned_ptr, 16);
  80         MD5Transform(ctx->buf, aligned_ptr);
  81         buf += t;
  82         len -= t;
  83     }
  84     /* Process data in 64-byte chunks */
  85
  86     while (len >= 64) {
  87         memcpy(ctx->in, buf, 64);
  88         byteReverse(aligned_ptr, 16);
  89         MD5Transform(ctx->buf, aligned_ptr);
  90         buf += 64;
  91         len -= 64;
  92     }
  93
  94     /* Handle any remaining bytes of data. */
  95     memcpy(ctx->in, buf, len);
  96 }
  97
  98 /*
  99  * Final wrapup - pad to 64-byte boundary with the bit pattern
 100  * 1 0* (64-bit count of bits processed, MSB-first)
 101  */
 102 static uint8_t* MD5_final(struct Hash *h)
 103 {
 104         MD5_Context *ctx = (MD5_Context *)h;
 105     unsigned count;
 106     unsigned char *p;
 107         uint32_t *aligned_ptr = NULL;
 108         if (is_aligned(ctx->in, sizeof(uint32_t)))
 109                 aligned_ptr = (uint32_t *)((size_t)ctx->in);
 110         else
 111                 ASSERT2(0, "Unaligned memory");
 112
 113     /* Compute number of bytes mod 64 */
 114     count = (ctx->bits >> 3) & 0x3F;
 115
 116     /* Set the first char of padding to 0x80.  This is safe since there is
 117        always at least one byte free */
 118     p = ctx->in + count;
 119     *p++ = 0x80;
 120
 121     /* Bytes of padding needed to make 64 bytes */
 122     count = 64 - 1 - count;
 123
 124     /* Pad out to 56 mod 64 */
 125     if (count < 8) {
 126         /* Two lots of padding:  Pad the first block to 64 bytes */
 127         memset(p, 0, count);
 128         byteReverse(aligned_ptr, 16);
 129         MD5Transform(ctx->buf, aligned_ptr);
 130
 131         /* Now fill the next block with 56 bytes */
 132         memset(ctx->in, 0, 56);
 133     } else {
 134         /* Pad block to 56 bytes */
 135         memset(p, 0, count - 8);
 136     }
 137     byteReverse(aligned_ptr, 14);
 138
 139     /* Append length in bits and transform */
 140     aligned_ptr[14] = (uint32_t)ctx->bits;
 141     aligned_ptr[15] = (uint32_t)(ctx->bits >> 32);
 142
 143     MD5Transform(ctx->buf, aligned_ptr);
 144     byteReverse((uint32_t*)ctx->buf, 4);
 145
 146         PURGE(ctx->in);
 147         PURGE(ctx->bits);
 148
 149         return (uint8_t *)ctx->buf;
 150 }
 151
 152
 153 /* The four core functions - F1 is optimized somewhat */
 154
 155 /* #define F1(x, y, z) (x & y | ~x & z) */
 156 #define F1(x, y, z) (z ^ (x & (y ^ z)))
 157 #define F2(x, y, z) F1(z, x, y)
 158 #define F3(x, y, z) (x ^ y ^ z)
 159 #define F4(x, y, z) (y ^ (x | ~z))
 160
 161 /* This is the central step in the MD5 algorithm. */
 162 #define MD5STEP(f, w, x, y, z, data, s) \
 163         ( w += f(x, y, z) + data,  w = w<<s | w>>(32-s),  w += x )
 164
 165 /*
 166  * The core of the MD5 algorithm, this alters an existing MD5 hash to
 167  * reflect the addition of 16 longwords of new data.  MD5Update blocks
 168  * the data and converts bytes into longwords for this routine.
 169  */
 170 static void MD5Transform(uint32_t buf[4], uint32_t in[16])
 171 {
 172     register uint32_t a, b, c, d;
 173
 174     a = buf[0];
 175     b = buf[1];
 176     c = buf[2];
 177     d = buf[3];
 178
 179     MD5STEP(F1, a, b, c, d, in[0] + 0xd76aa478, 7);
 180     MD5STEP(F1, d, a, b, c, in[1] + 0xe8c7b756, 12);
 181     MD5STEP(F1, c, d, a, b, in[2] + 0x242070db, 17);
 182     MD5STEP(F1, b, c, d, a, in[3] + 0xc1bdceee, 22);
 183     MD5STEP(F1, a, b, c, d, in[4] + 0xf57c0faf, 7);
 184     MD5STEP(F1, d, a, b, c, in[5] + 0x4787c62a, 12);
 185     MD5STEP(F1, c, d, a, b, in[6] + 0xa8304613, 17);
 186     MD5STEP(F1, b, c, d, a, in[7] + 0xfd469501, 22);
 187     MD5STEP(F1, a, b, c, d, in[8] + 0x698098d8, 7);
 188     MD5STEP(F1, d, a, b, c, in[9] + 0x8b44f7af, 12);
 189     MD5STEP(F1, c, d, a, b, in[10] + 0xffff5bb1, 17);
 190     MD5STEP(F1, b, c, d, a, in[11] + 0x895cd7be, 22);
 191     MD5STEP(F1, a, b, c, d, in[12] + 0x6b901122, 7);
 192     MD5STEP(F1, d, a, b, c, in[13] + 0xfd987193, 12);
 193     MD5STEP(F1, c, d, a, b, in[14] + 0xa679438e, 17);
 194     MD5STEP(F1, b, c, d, a, in[15] + 0x49b40821, 22);
 195
 196     MD5STEP(F2, a, b, c, d, in[1] + 0xf61e2562, 5);
 197     MD5STEP(F2, d, a, b, c, in[6] + 0xc040b340, 9);
 198     MD5STEP(F2, c, d, a, b, in[11] + 0x265e5a51, 14);
 199     MD5STEP(F2, b, c, d, a, in[0] + 0xe9b6c7aa, 20);
 200     MD5STEP(F2, a, b, c, d, in[5] + 0xd62f105d, 5);
 201     MD5STEP(F2, d, a, b, c, in[10] + 0x02441453, 9);
 202     MD5STEP(F2, c, d, a, b, in[15] + 0xd8a1e681, 14);
 203     MD5STEP(F2, b, c, d, a, in[4] + 0xe7d3fbc8, 20);
 204     MD5STEP(F2, a, b, c, d, in[9] + 0x21e1cde6, 5);
 205     MD5STEP(F2, d, a, b, c, in[14] + 0xc33707d6, 9);
 206     MD5STEP(F2, c, d, a, b, in[3] + 0xf4d50d87, 14);
 207     MD5STEP(F2, b, c, d, a, in[8] + 0x455a14ed, 20);
 208     MD5STEP(F2, a, b, c, d, in[13] + 0xa9e3e905, 5);
 209     MD5STEP(F2, d, a, b, c, in[2] + 0xfcefa3f8, 9);
 210     MD5STEP(F2, c, d, a, b, in[7] + 0x676f02d9, 14);
 211     MD5STEP(F2, b, c, d, a, in[12] + 0x8d2a4c8a, 20);
 212
 213     MD5STEP(F3, a, b, c, d, in[5] + 0xfffa3942, 4);
 214     MD5STEP(F3, d, a, b, c, in[8] + 0x8771f681, 11);
 215     MD5STEP(F3, c, d, a, b, in[11] + 0x6d9d6122, 16);
 216     MD5STEP(F3, b, c, d, a, in[14] + 0xfde5380c, 23);
 217     MD5STEP(F3, a, b, c, d, in[1] + 0xa4beea44, 4);
 218     MD5STEP(F3, d, a, b, c, in[4] + 0x4bdecfa9, 11);
 219     MD5STEP(F3, c, d, a, b, in[7] + 0xf6bb4b60, 16);
 220     MD5STEP(F3, b, c, d, a, in[10] + 0xbebfbc70, 23);
 221     MD5STEP(F3, a, b, c, d, in[13] + 0x289b7ec6, 4);
 222     MD5STEP(F3, d, a, b, c, in[0] + 0xeaa127fa, 11);
 223     MD5STEP(F3, c, d, a, b, in[3] + 0xd4ef3085, 16);
 224     MD5STEP(F3, b, c, d, a, in[6] + 0x04881d05, 23);
 225     MD5STEP(F3, a, b, c, d, in[9] + 0xd9d4d039, 4);
 226     MD5STEP(F3, d, a, b, c, in[12] + 0xe6db99e5, 11);
 227     MD5STEP(F3, c, d, a, b, in[15] + 0x1fa27cf8, 16);
 228     MD5STEP(F3, b, c, d, a, in[2] + 0xc4ac5665, 23);
 229
 230     MD5STEP(F4, a, b, c, d, in[0] + 0xf4292244, 6);
 231     MD5STEP(F4, d, a, b, c, in[7] + 0x432aff97, 10);
 232     MD5STEP(F4, c, d, a, b, in[14] + 0xab9423a7, 15);
 233     MD5STEP(F4, b, c, d, a, in[5] + 0xfc93a039, 21);
 234     MD5STEP(F4, a, b, c, d, in[12] + 0x655b59c3, 6);
 235     MD5STEP(F4, d, a, b, c, in[3] + 0x8f0ccc92, 10);
 236     MD5STEP(F4, c, d, a, b, in[10] + 0xffeff47d, 15);
 237     MD5STEP(F4, b, c, d, a, in[1] + 0x85845dd1, 21);
 238     MD5STEP(F4, a, b, c, d, in[8] + 0x6fa87e4f, 6);
 239     MD5STEP(F4, d, a, b, c, in[15] + 0xfe2ce6e0, 10);
 240     MD5STEP(F4, c, d, a, b, in[6] + 0xa3014314, 15);
 241     MD5STEP(F4, b, c, d, a, in[13] + 0x4e0811a1, 21);
 242     MD5STEP(F4, a, b, c, d, in[4] + 0xf7537e82, 6);
 243     MD5STEP(F4, d, a, b, c, in[11] + 0xbd3af235, 10);
 244     MD5STEP(F4, c, d, a, b, in[2] + 0x2ad7d2bb, 15);
 245     MD5STEP(F4, b, c, d, a, in[9] + 0xeb86d391, 21);
 246
 247     buf[0] += a;
 248     buf[1] += b;
 249     buf[2] += c;
 250     buf[3] += d;
 251 }
 252
 253 /*******************************************************************/
 254
 255 void MD5_init(MD5_Context *ctx)
 256 {
 257         ctx->h.begin = MD5_begin;
 258         ctx->h.update = MD5_update;
 259         ctx->h.final = MD5_final;
 260         ctx->h.digest_len = 16;
 261         ctx->h.block_len = 64;
 262 }