#include "md5.h"
#include <stdint.h>
#include <string.h>
#define CCMD5_BLOCK_SIZE 64
#define F(x,y,z) (z ^ (x & (y ^ z)))
#define G(x,y,z) (y ^ (z & (y ^ x)))
#define H(x,y,z) (x^y^z)
#define I(x,y,z) (y^(x|(~z)))
#define CC_ROLc(X,s) (((X) << (s)) | ((X) >> (32 - (s))))
#define FF(a,b,c,d,M,s,t) \
a = (a + F(b,c,d) + M + t); a = CC_ROLc(a, s) + b;
#define GG(a,b,c,d,M,s,t) \
a = (a + G(b,c,d) + M + t); a = CC_ROLc(a, s) + b;
#define HH(a,b,c,d,M,s,t) \
a = (a + H(b,c,d) + M + t); a = CC_ROLc(a, s) + b;
#define II(a,b,c,d,M,s,t) \
a = (a + I(b,c,d) + M + t); a = CC_ROLc(a, s) + b;
static void md5_compress(uint32_t *state, unsigned long nblocks, const void *in)
{
uint32_t i, W[16], a, b, c, d;
uint32_t *s = state;
const unsigned char *buf = in;
while(nblocks--) {
for (i = 0; i < 16; i++) {
W[i] = ((uint32_t*)buf)[i];
}
a = s[0];
b = s[1];
c = s[2];
d = s[3];
FF(a,b,c,d,W[0],7,0xd76aa478)
FF(d,a,b,c,W[1],12,0xe8c7b756)
FF(c,d,a,b,W[2],17,0x242070db)
FF(b,c,d,a,W[3],22,0xc1bdceee)
FF(a,b,c,d,W[4],7,0xf57c0faf)
FF(d,a,b,c,W[5],12,0x4787c62a)
FF(c,d,a,b,W[6],17,0xa8304613)
FF(b,c,d,a,W[7],22,0xfd469501)
FF(a,b,c,d,W[8],7,0x698098d8)
FF(d,a,b,c,W[9],12,0x8b44f7af)
FF(c,d,a,b,W[10],17,0xffff5bb1)
FF(b,c,d,a,W[11],22,0x895cd7be)
FF(a,b,c,d,W[12],7,0x6b901122)
FF(d,a,b,c,W[13],12,0xfd987193)
FF(c,d,a,b,W[14],17,0xa679438e)
FF(b,c,d,a,W[15],22,0x49b40821)
GG(a,b,c,d,W[1],5,0xf61e2562)
GG(d,a,b,c,W[6],9,0xc040b340)
GG(c,d,a,b,W[11],14,0x265e5a51)
GG(b,c,d,a,W[0],20,0xe9b6c7aa)
GG(a,b,c,d,W[5],5,0xd62f105d)
GG(d,a,b,c,W[10],9,0x02441453)
GG(c,d,a,b,W[15],14,0xd8a1e681)
GG(b,c,d,a,W[4],20,0xe7d3fbc8)
GG(a,b,c,d,W[9],5,0x21e1cde6)
GG(d,a,b,c,W[14],9,0xc33707d6)
GG(c,d,a,b,W[3],14,0xf4d50d87)
GG(b,c,d,a,W[8],20,0x455a14ed)
GG(a,b,c,d,W[13],5,0xa9e3e905)
GG(d,a,b,c,W[2],9,0xfcefa3f8)
GG(c,d,a,b,W[7],14,0x676f02d9)
GG(b,c,d,a,W[12],20,0x8d2a4c8a)
HH(a,b,c,d,W[5],4,0xfffa3942)
HH(d,a,b,c,W[8],11,0x8771f681)
HH(c,d,a,b,W[11],16,0x6d9d6122)
HH(b,c,d,a,W[14],23,0xfde5380c)
HH(a,b,c,d,W[1],4,0xa4beea44)
HH(d,a,b,c,W[4],11,0x4bdecfa9)
HH(c,d,a,b,W[7],16,0xf6bb4b60)
HH(b,c,d,a,W[10],23,0xbebfbc70)
HH(a,b,c,d,W[13],4,0x289b7ec6)
HH(d,a,b,c,W[0],11,0xeaa127fa)
HH(c,d,a,b,W[3],16,0xd4ef3085)
HH(b,c,d,a,W[6],23,0x04881d05)
HH(a,b,c,d,W[9],4,0xd9d4d039)
HH(d,a,b,c,W[12],11,0xe6db99e5)
HH(c,d,a,b,W[15],16,0x1fa27cf8)
HH(b,c,d,a,W[2],23,0xc4ac5665)
II(a,b,c,d,W[0],6,0xf4292244)
II(d,a,b,c,W[7],10,0x432aff97)
II(c,d,a,b,W[14],15,0xab9423a7)
II(b,c,d,a,W[5],21,0xfc93a039)
II(a,b,c,d,W[12],6,0x655b59c3)
II(d,a,b,c,W[3],10,0x8f0ccc92)
II(c,d,a,b,W[10],15,0xffeff47d)
II(b,c,d,a,W[1],21,0x85845dd1)
II(a,b,c,d,W[8],6,0x6fa87e4f)
II(d,a,b,c,W[15],10,0xfe2ce6e0)
II(c,d,a,b,W[6],15,0xa3014314)
II(b,c,d,a,W[13],21,0x4e0811a1)
II(a,b,c,d,W[4],6,0xf7537e82)
II(d,a,b,c,W[11],10,0xbd3af235)
II(c,d,a,b,W[2],15,0x2ad7d2bb)
II(b,c,d,a,W[9],21,0xeb86d391)
s[0] += a;
s[1] += b;
s[2] += c;
s[3] += d;
buf+=CCMD5_BLOCK_SIZE;
}
}
void md5_hash(uint8_t *message, uint64_t len, uint32_t *hash) {
hash[0] = 0x67452301;
hash[1] = 0xEFCDAB89;
hash[2] = 0x98BADCFE;
hash[3] = 0x10325476;
md5_compress(hash, len/64, message);
uint32_t blockbuff[16];
uint8_t *byteptr = (uint8_t*)blockbuff;
int left = len % 64;
memcpy(byteptr, message + len-left, left);
byteptr[left] = 0x80;
left++;
if (64 - left >= 8)
bzero(byteptr + left, 56 - left);
else {
memset(byteptr + left, 0, 64 - left);
md5_compress(hash, 1, blockbuff);
bzero(blockbuff, 56);
}
blockbuff[14] = (uint32_t)(len << 3);
blockbuff[15] = (uint32_t)(len >> 29);
md5_compress(hash, 1, blockbuff);
}