md5.cpp

MD5算法的C++版本

// MD5.cpp: implementation of the MD5 class.

/*
 *	Copyright (c) 2006 - 2007 All Rights Reserved
 *
 *	程序作者:
 *		张鲁夺(zhangluduo) : 为所有爱我的人和我爱的人努力!
 *
 *	联系方式:
 *		[e-mail]
 *		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 <= 63; 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 << 3)) < ((unsigned long)inputLen << 3))
		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]) << 8) | (((unsigned long)input[j+2]) << 16) | (((unsigned long)input[j+3]) << 24);
}

/** 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;
}