#include "StdAfx.h"
#include ".\newhash.h"
/**
 * 散列構造器，有機器碼
 * 本類可根據 機器碼，用戶名，口令或只用用戶名，口令散列出指定位的序列
 * 參數：
 *  pcstr 機器碼
 *  username 用戶名
 *  psw 用戶口令
 *  len1 pcstr有效長度
 *  len2 username有效長度
 *  len3 psw有效長度
 *  keyLen 須散列后序列的位數，缺省為 128位
 */
NewHash::NewHash(unsigned char *pcstr, unsigned char *username, unsigned char *psw, int len1, int len2, int len3, int keyLen)
{
	if(keyLen!=128 && keyLen!=192 && keyLen!=256) throw exception("密鑰長度無效");	//密鑰長度無效異常
																					//密鑰長度必須是128或192或256位
	try{
		genKey = new GenKey(pcstr,username,psw,len1,len2,len3);
		buffer = new unsigned char[len1+len2+len3];
		unsigned char *tp = buffer;
		memcpy(tp,pcstr,len1);
		tp+=len1;
		memcpy(tp,username,len2);
		tp+=len2;
		memcpy(tp,psw,len3);
		count=len1+len2+len3;
		this->keyLen=keyLen;
		initial();						//基本數據初始化
		flg=false;
	}catch(exception& ex)
	{
		delete (genKey);
		delete[] buffer;
		throw ex;
	}
}
/**
 * 散列構造器，無機器碼
 * 參數：
 *  pcstr 機器碼
 *  username 用戶名
 *  len1 pcstr有效長度
 *  len2 username有效長度
 *  keyLen 須散列后序列的位數，缺省為 128位
 */
NewHash::NewHash(unsigned char *pcstr, unsigned char *username, int len1, int len2, int keyLen)
{
	if(keyLen!=128 && keyLen!=192 && keyLen!=256) throw exception("密鑰長度無效");
	try{
		genKey = new GenKey(pcstr,username,len1,len2);
		buffer = new unsigned char[len1+len2];
		unsigned char *tp = buffer;
		memcpy(tp,pcstr,len1);
		tp+=len1;
		memcpy(tp,username,len2);
		count=len1+len2;
		this->keyLen=keyLen;
		initial();
		flg=false;
	}catch(exception& ex)
	{
		delete (genKey);
		delete[] buffer;
		throw ex;
	}
}
/**
 * 取的散列后指定位數的序列，進行此方法調用時必須進行iniKey方法的調用初始化，否則函數體內將拋出異常
 * 參數：
 *  finkey 存儲最終散列序列，要求大小務必大于等于keyLen
 * 返回：finkey 存儲最終散列序列指針
 */
unsigned char *NewHash::getKey(unsigned char *finkey)
{
	if(!flg) throw exception("從未初始化的序列！");
	if(strlen((char*)((void*)finkey))<(size_t)keyLen/8) throw exception("密鑰參數數組越界！");
	unsigned char *reh; 
	switch(keyLen)							//根據指定的不同的散列序列位數進行處理
	{
	case 128: //finkey= new unsigned char[SL];
		if(ha[SL-1]%2) memcpy(finkey,ha,SL);
		else memcpy(finkey,hb,SL);
		return finkey;
	case 192:
		//finkey= new unsigned char[SL+8];
		reh=finkey;
		memcpy(reh,ha,16);
		reh+=16;
		memcpy(reh,hb,8);
		return finkey;
	case 256:
		//finkey= new unsigned char[2*SL];
		reh=finkey;
		memcpy(reh,ha,16);
		reh+=16;
		memcpy(reh,hb,16);
		return finkey;
	default: throw exception("密鑰長度無效");
	}
}
/**
 * 進行基本數據結構及數據的初始化，在構造函數里調用
 */
void NewHash::initial()
{
	unsigned char *tp,*tk;
	int n;
	mecnt=0;
	tp=buffer;
	genKey->getKey(key);
	memset(m,0,sizeof(m));
	memset(me,0,sizeof(me));
	for(int i=0; i<count; i+=16,tp+=16)					//初始化me（由m經過aes加密得）
	{
		n =( count -i > 16 ) ? 16 : ( count - i );
		memcpy(m[mecnt],tp,n);
		AesCrypt::rijCrypt(m[mecnt],me[mecnt],key,16);
		mecnt++;
	}
	tk=key;
	memcpy(k1,tk,SUBKL);
	tk+=SUBKL;
	memcpy(k2,tk,SUBKL);
	//printf("-=%d=-\n",mecnt);
	/*for(int i=0; i<mecnt; i++)
	{
		for(int j=0; j<16; j++) printf("%x-",me[i][j]);
		printf("\n");
	}*/
}
/**
 * 散列的核心，進行散列的基本過程，調用getKey方法前必須先調用此方法
 */
void NewHash::iniKey()
{
	flg=true;
	unsigned char tmp[SL];
	memset(ha,0,sizeof(ha));
	memset(hb,0,sizeof(hb));
	memcpy(ha,k1,SUBKL);
	memcpy(hb,k2,SUBKL);
	for(int i=0; i<mecnt; i++)
	{
		xor(tmp,ha,me[i],SL);
		AesCrypt::rijCrypt(m[i],r,tmp,16);		//對稱加密，使其更隨機
		xor(r,r,me[i],SL);

		xor(tmp,hb,m[i],SL);
		xor(tmp,tmp,r,SL);
		AesCrypt::rijCrypt(me[i],re,tmp,16);	//對稱加密，使其更隨機
		xor(re,re,m[i],SL);
		
		memcpy(tmp,ha,SL);
		xor(ha,ha,hb,SL);
		xor(ha,ha,re,SL);
		xor(ha,ha,r,SL);
		
		xor(hb,tmp,hb,SL);
		xor(hb,hb,r,SL);

		//memcpy(tmp,ha,SL);
		//ha^=me[i];
		//AesCrypt::rijCrypt(m[i],r,ha,16);
		//r^=me[i];

		//memcpy(tmp1,hb,SL);
		//hb^=m[i];
		//hb^=r;
		//AesCrypt::rijCrypt(me[i],re,hb,16);
		//re^=m[i];

		//memcpy(ha,tmp,SL);
		//ha^=hb;
		//ha^=re;
		//ha^=r;

		//memcpy(hb,tmp1,SL);
		//hb^=tmp, hb^=r;
	}
}
/**
 * 進行兩個序列的異或運算
 */
void NewHash::xor(unsigned char *buf, unsigned char *a, unsigned char *b, int l)
{
	for(int i=0; i<l; i++)
	{
		buf[i]=a[i]^b[i];			//逐個元素異或
	}
}

NewHash::~NewHash(void)
{
	delete (genKey);
	delete[] buffer;
}