// MD5.cpp: implementation of the MD5 class.
/*
* Copyright (c) 2006 - 2007 All Rights Reserved
*
* 程序作者:
* 张鲁夺(zhangluduo) : 为所有爱我的人和我爱的人努力!
*
* 联系方式:
* zhangluduo@msn.com
* QQ群:34064264
*
* 更新时间:
* 2007-10-24
*
* 功能描述:
* MD5算法实现, 根据rfc1321.txt中的源码封装成C++类, 详细算法流程请参
* 见rfc1321.txt
*
* 授权声明:
* 许可任何单位,个人随意使用,拷贝,修改,散布及出售这份代码,及其相关的
* 开发文档,但是必须保留此版权信息,以慰藉作者辛勤的劳动,及表明此代码
* 的来源,如若此份代码有任何BUG或者您有更好的修改建议, 请通知作者,以
* 便弥补作者由于水平所限而导致的一些错误和不足,谢谢!
*/
#include "stdafx.h"
#include "MD5.h"
MD5::MD5()
{
PADDING[0] = 0x80;
for(int i = 1; i PADDING[i] = 0;
}
MD5::~MD5()
{
}
// MD5 initialization. Begins an MD5 operation, writing a new context.
void MD5::MD5Init (struct MD5_CTX *context)
{
context->count[0] = context->count[1] = 0;
context->state[0] = 0x67452301;
context->state[1] = 0xEFCDAB89;
context->state[2] = 0x98BADCFE;
context->state[3] = 0x10325476;
}
/** MD5 block update operation. Continues an MD5 message-digest
operation, processing another message block, and updating the
context.
*/
void MD5::MD5Update (struct MD5_CTX *context, /* context */ unsigned char *input, /* input block */ unsigned int inputLen /* length of input block */)
{
unsigned int i, index, partLen;
// Compute number of bytes mod 64
index = (unsigned int)((context->count[0] >> 3) & 0x3F);
// Update number of bits
if ((context->count[0] += ((unsigned long)inputLen context->count[1]++;
context->count[1] += ((unsigned long)inputLen >> 29);
partLen = 64 - index;
// Transform as many times as possible.
if (inputLen >= partLen)
{
MD5_memcpy
((unsigned char*)&context->buffer[index], (unsigned char*)input, partLen);
MD5Transform (context->state, context->buffer);
for (i = partLen; i + 63 < inputLen; i += 64)
MD5Transform (context->state, &input[i]);
index = 0;
}
else
i = 0;
// Buffer remaining input
MD5_memcpy ((unsigned char*)&context->buffer[index], (unsigned char*)&input[i], inputLen-i);
}
// Note: Replace "for loop" with standard memcpy if possible.
void MD5::MD5_memcpy (unsigned char* output, unsigned char* input, unsigned int len)
{
for (unsigned int i = 0; i < len; i++)
output[i] = input[i];
}
// MD5 basic transformation. Transforms state based on block.
void MD5::MD5Transform (unsigned long state[4], unsigned char block[64])
{
unsigned long a = state[0], b = state[1], c = state[2], d = state[3], x[16];
Decode (x, block, 64);
/* Round 1 */
FF (a, b, c, d, x[ 0], S11, 0xD76AA478); /* 1 */
FF (d, a, b, c, x[ 1], S12, 0xE8C7B756); /* 2 */
FF (c, d, a, b, x[ 2], S13, 0x242070DB); /* 3 */
FF (b, c, d, a, x[ 3], S14, 0xC1BDCEEE); /* 4 */
FF (a, b, c, d, x[ 4], S11, 0xF57C0FAF); /* 5 */
FF (d, a, b, c, x[ 5], S12, 0x4787C62A); /* 6 */
FF (c, d, a, b, x[ 6], S13, 0xA8304613); /* 7 */
FF (b, c, d, a, x[ 7], S14, 0xFD469501); /* 8 */
FF (a, b, c, d, x[ 8], S11, 0x698098D8); /* 9 */
FF (d, a, b, c, x[ 9], S12, 0x8B44F7AF); /* 10 */
FF (c, d, a, b, x[10], S13, 0xFFFF5BB1); /* 11 */
FF (b, c, d, a, x[11], S14, 0x895CD7BE); /* 12 */
FF (a, b, c, d, x[12], S11, 0x6B901122); /* 13 */
FF (d, a, b, c, x[13], S12, 0xFD987193); /* 14 */
FF (c, d, a, b, x[14], S13, 0xA679438E); /* 15 */
FF (b, c, d, a, x[15], S14, 0x49B40821); /* 16 */
/* Round 2 */
GG (a, b, c, d, x[ 1], S21, 0xF61E2562); /* 17 */
GG (d, a, b, c, x[ 6], S22, 0xC040B340); /* 18 */
GG (c, d, a, b, x[11], S23, 0x265E5A51); /* 19 */
GG (b, c, d, a, x[ 0], S24, 0xE9B6C7AA); /* 20 */
GG (a, b, c, d, x[ 5], S21, 0xD62F105D); /* 21 */
GG (d, a, b, c, x[10], S22, 0x02441453); /* 22 */
GG (c, d, a, b, x[15], S23, 0xD8A1E681); /* 23 */
GG (b, c, d, a, x[ 4], S24, 0xE7D3FBC8); /* 24 */
GG (a, b, c, d, x[ 9], S21, 0x21E1CDE6); /* 25 */
GG (d, a, b, c, x[14], S22, 0xC33707D6); /* 26 */
GG (c, d, a, b, x[ 3], S23, 0xF4D50D87); /* 27 */
GG (b, c, d, a, x[ 8], S24, 0x455A14ED); /* 28 */
GG (a, b, c, d, x[13], S21, 0xA9E3E905); /* 29 */
GG (d, a, b, c, x[ 2], S22, 0xFCEFA3F8); /* 30 */
GG (c, d, a, b, x[ 7], S23, 0x676F02D9); /* 31 */
GG (b, c, d, a, x[12], S24, 0x8D2A4C8A); /* 32 */
/* Round 3 */
HH (a, b, c, d, x[ 5], S31, 0xFFFA3942); /* 33 */
HH (d, a, b, c, x[ 8], S32, 0x8771F681); /* 34 */
HH (c, d, a, b, x[11], S33, 0x6D9D6122); /* 35 */
HH (b, c, d, a, x[14], S34, 0xFDE5380C); /* 36 */
HH (a, b, c, d, x[ 1], S31, 0xA4BEEA44); /* 37 */
HH (d, a, b, c, x[ 4], S32, 0x4BDECFA9); /* 38 */
HH (c, d, a, b, x[ 7], S33, 0xF6BB4B60); /* 39 */
HH (b, c, d, a, x[10], S34, 0xBEBFBC70); /* 40 */
HH (a, b, c, d, x[13], S31, 0x289B7EC6); /* 41 */
HH (d, a, b, c, x[ 0], S32, 0xEAA127FA); /* 42 */
HH (c, d, a, b, x[ 3], S33, 0xD4EF3085); /* 43 */
HH (b, c, d, a, x[ 6], S34, 0x04881D05); /* 44 */
HH (a, b, c, d, x[ 9], S31, 0xD9D4D039); /* 45 */
HH (d, a, b, c, x[12], S32, 0xE6DB99E5); /* 46 */
HH (c, d, a, b, x[15], S33, 0x1FA27CF8); /* 47 */
HH (b, c, d, a, x[ 2], S34, 0xC4AC5665); /* 48 */
/* Round 4 */
II (a, b, c, d, x[ 0], S41, 0xF4292244); /* 49 */
II (d, a, b, c, x[ 7], S42, 0x432AFF97); /* 50 */
II (c, d, a, b, x[14], S43, 0xAB9423A7); /* 51 */
II (b, c, d, a, x[ 5], S44, 0xFC93A039); /* 52 */
II (a, b, c, d, x[12], S41, 0x655B59C3); /* 53 */
II (d, a, b, c, x[ 3], S42, 0x8F0CCC92); /* 54 */
II (c, d, a, b, x[10], S43, 0xFFEFF47D); /* 55 */
II (b, c, d, a, x[ 1], S44, 0x85845DD1); /* 56 */
II (a, b, c, d, x[ 8], S41, 0x6FA87E4F); /* 57 */
II (d, a, b, c, x[15], S42, 0xFE2CE6E0); /* 58 */
II (c, d, a, b, x[ 6], S43, 0xA3014314); /* 59 */
II (b, c, d, a, x[13], S44, 0x4E0811A1); /* 60 */
II (a, b, c, d, x[ 4], S41, 0xF7537E82); /* 61 */
II (d, a, b, c, x[11], S42, 0xBD3AF235); /* 62 */
II (c, d, a, b, x[ 2], S43, 0x2AD7D2BB); /* 63 */
II (b, c, d, a, x[ 9], S44, 0xEB86D391); /* 64 */
state[0] += a;
state[1] += b;
state[2] += c;
state[3] += d;
MD5_memset ((unsigned char*)x, 0, sizeof (x));
}
/** Encodes input (UINT4) into output (unsigned char). Assumes len is
a multiple of 4.
*/
void MD5::Encode (unsigned char *output, unsigned long *input, unsigned int len)
{
for (unsigned int i = 0, j = 0; j < len; i++, j += 4)
{
output[j] = (unsigned char)(input[i] & 0xff);
output[j+1] = (unsigned char)((input[i] >> 8) & 0xff);
output[j+2] = (unsigned char)((input[i] >> 16) & 0xff);
output[j+3] = (unsigned char)((input[i] >> 24) & 0xff);
}
}
/** Encodes input (UINT4) into output (unsigned char). Assumes len is
a multiple of 4.
*/
void MD5::Decode (unsigned long* output, unsigned char *input, unsigned int len)
{
for (unsigned int i = 0, j = 0; j < len; i++, j += 4)
output[i] = ((unsigned long)input[j]) | (((unsigned long)input[j+1]) }
/** Decodes input (unsigned char) into output (UINT4). Assumes len is
a multiple of 4.
*/
void MD5::MD5_memset (unsigned char* output, int value, unsigned int len)
{
for (unsigned int i = 0; i < len; i++)
((char *)output)[i] = (char)value;
}
/** MD5 finalization. Ends an MD5 message-digest operation, writing the
the message digest and zeroizing the context.
*/
void MD5::MD5Final (unsigned char digest[16], /* message digest */ MD5_CTX *context /* context */)
{
unsigned char bits[8];
unsigned int index, padLen;
// Save number of bits
Encode (bits, context->count, 8);
// Pad out to 56 mod 64.
index = (unsigned int)((context->count[0] >> 3) & 0x3f);
padLen = (index < 56) ? (56 - index) : (120 - index);
MD5Update (context, PADDING, padLen);
// Append length (before padding)
MD5Update (context, bits, 8);
// Store state in digest
Encode (digest, context->state, 16);
// Zeroize sensitive information.
MD5_memset ((unsigned char*)context, 0, sizeof (*context));
}
string MD5::ResultToHex(unsigned char digest[16], bool bUppercase)
{
string strRetVal = "";
for(int i = 0 ; i < 16; i++)
{
char buf[3] = {0};
if(bUppercase)
sprintf(buf, "%02X", digest[i]);
else
sprintf(buf, "%02x", digest[i]);
strRetVal += buf;
}
return strRetVal;
}
void MD5::OnFileProcessing(int nProgress, FunctionAddr addr)
{
void* pThis = addr.GetThisPtr();
unsigned long Addr = addr.GetAddr();
// ! the Calling Convention must be __stdcall
__asm
{
push nProgress ;
mov ecx, pThis ;
call Addr ;
}
}
string MD5::MD5String(unsigned char *pData, unsigned long DataLen, bool bUppercase)
{
MD5_CTX context;
unsigned char digest[16];
MD5Init (&context);
MD5Update (&context, pData, DataLen);
MD5Final (digest, &context);
return ResultToHex(digest, bUppercase);
}
string MD5::MD5File(const char *FileName, bool bUppercase, FunctionAddr addr)
{
MD5_CTX context;
FILE *file = 0;
int nFileBufSize = 10240; // 文件缓冲区大小
unsigned char *buffer = new unsigned char[nFileBufSize + 1];
unsigned char digest[16] = {0};
m_bStop = false;
try
{
if ((file = fopen (FileName, "rb")) == NULL)
{
m_strErr = "Open file failed!";
goto ERR;
}
else
{
int len = 0;
int ReadCount = 0;
unsigned long nFileSize = 0;
// 获取文件大小
{
HANDLE hFile = CreateFile(FileName, NULL, FILE_SHARE_READ, NULL, OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL, 0);
if(hFile== NULL)
{
m_strErr = "Get file size failed!";
goto ERR;
}
nFileSize=::GetFileSize(hFile, NULL);
CloseHandle(hFile);
}
if(!addr.IsNull())
OnFileProcessing(0, addr);
MD5Init(&context);
while (len = fread (buffer, 1, nFileBufSize, file))
{
ReadCount++;
if(m_bStop)
{
m_strErr = "User terminate calculate!";
goto ERR;
}
MD5Update(&context, buffer, len);
if(!addr.IsNull())
OnFileProcessing((int)(nFileBufSize * ReadCount / (nFileSize / 100)), addr);
memset(buffer, nFileBufSize + 1, 0);
}
MD5Final(digest, &context);
if(!addr.IsNull())
OnFileProcessing(100, addr);
fclose (file);
delete[] buffer;
return ResultToHex(digest, bUppercase);
}
}
catch(...)
{
m_strErr = "UnKnow error!";
goto ERR;
}
ERR:
if(file != 0)
fclose (file);
delete[] buffer;
return "";
}
void MD5::FileTerminate()
{
m_bStop = true;
}
string MD5::GetError()
{
return m_strErr;
}
