/*
* Copyright (c) 2007-2009 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of the License "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description: 
* Common implementation of SHA224 and SHA256
* RFC 4634 (US Secure Hash Algorithms (SHA and HMAC-SHA))
*
*/


/**
 @file
*/


#include "sha224and256.h"

/**
 * SHA256 Constants
 * 
 * SHA-256 uses a sequence of sixty-four constant 32-bit words. 
 * These words represent the first thirty-two bits of the fractional 
 * parts of the cube roots of the first sixtyfour prime numbers.
 * 
 * FIPS 180-2 Section 4.2.2
 */
const TUint K[64] = 
	{
	0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5,	
	0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
	0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3,
	0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
	0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc,
	0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
	0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7,
	0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
	0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13,
	0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
	0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3, 
	0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
	0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5, 
	0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
	0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208,
	0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2
	};

/**
 * Define the SHA SIGMA and sigma macros 
 * 
 * FIPS 180-2 section 4.1.2
 */
// Equation 4.4
inline TUint SHA256_SIGMA0(TUint aWord)
	{
	return (SHA_ROTR<TUint>( 2,aWord) ^ SHA_ROTR<TUint>(13,aWord) ^ SHA_ROTR<TUint>(22,aWord));
	}
// Equation 4.5
inline TUint SHA256_SIGMA1(TUint aWord)
	{
	return (SHA_ROTR<TUint>( 6,aWord) ^ SHA_ROTR<TUint>(11,aWord) ^ SHA_ROTR<TUint>(25,aWord));
	}
// Equation 4.6
inline TUint SHA256_sigma0(TUint aWord)
	{
	return (SHA_ROTR<TUint>( 7,aWord) ^ SHA_ROTR<TUint>(18,aWord) ^ SHA_SHR<TUint>( 3,aWord));
	}
// Equation 4.7
inline TUint SHA256_sigma1(TUint aWord)
	{
	return (SHA_ROTR<TUint>(17,aWord) ^ SHA_ROTR<TUint>(19,aWord) ^ SHA_SHR<TUint>(10,aWord));
	}


// Macros
inline TUint MakeWord(const TUint8* aData)
	{
	return (aData[0] << 24 | aData[1] << 16 | aData[2] << 8 | aData[3]);
	}

	
CSHA224And256* CSHA224And256::NewL()
	{
	CSHA224And256* self=new (ELeave) CSHA224And256();
	return self;						
	}
														
CSHA224And256::CSHA224And256() : iHash(KSHA256HashSize)
	{		
	}
	
CSHA224And256::CSHA224And256(const CSHA224And256& aSHAImpl)
				: 	iHash(aSHAImpl.iHash),
					iA(aSHAImpl.iA),
					iB(aSHAImpl.iB),
					iC(aSHAImpl.iC),
					iD(aSHAImpl.iD),
					iE(aSHAImpl.iE),
					iF(aSHAImpl.iF),
					iG(aSHAImpl.iG),
					iH(aSHAImpl.iH),
					iNl(aSHAImpl.iNl),
					iNh(aSHAImpl.iNh)
	{
	Mem::Copy(iData, aSHAImpl.iData, KSHA256BlockSize*sizeof(TUint));
	}
	
void CSHA224And256::Reset(const TAny* aValArray)
	{
	const TUint* values = static_cast<const TUint*>(aValArray);
	iA=values[0];
	iB=values[1];
	iC=values[2];
	iD=values[3];
	iE=values[4];
	iF=values[5];
	iG=values[6];
	iH=values[7];
	iNh=0;
	iNl=0;
	}

// This assumes a big-endian architecture
void CSHA224And256::Update(const TUint8* aData,TUint aLength)
	{
	while((aLength / 4) > 0 && (iNl % 4 == 0))
		{
		iData[iNl>>2] = MakeWord(aData);
		iNl+=4;
		aData+=4;
		aLength-=4;
		if(iNl==KSHA256BlockSize) 
			{
			Block();
			AddLength(KSHA256BlockSize);
			}
		}

	while(aLength--)
		{
		if(!(iNl&0x03))
			{
			iData[iNl >> 2] = 0;
			}
		iData[iNl >> 2] |= *aData << ((3 - iNl&0x03) << 3) ;
		++aData;
		++iNl;
		if(iNl==KSHA256BlockSize) 
			{
			Block();
			AddLength(KSHA256BlockSize);
			}
		}
	}

//This function will panic if the total input length is longer than 2^64 in bits
_LIT(KPanicString, "Message length exceeds supported length");
inline void CSHA224And256::AddLength(const TUint aLength)
	{
	TUint64 temp = iNh;
	iNh += aLength << 3;
	__ASSERT_ALWAYS((temp <= iNh), User::Panic(KPanicString, KErrOverflow));
	}


static inline void CSHA256_16(	const TUint aA, 
								const TUint aB, 
								const TUint aC,
								TUint& aD, 
								const TUint aE, 
								const TUint aF,
								const TUint aG, 
								TUint& aH,
								TUint aTemp1,
								TUint aTemp2,
								const TUint aK,
								const TUint aWord)
	{
	aTemp1 = aH + SHA256_SIGMA1(aE) + SHA_Ch(aE,aF,aG) + aK + aWord;
	aTemp2 = SHA256_SIGMA0(aA) + SHA_Maj(aA,aB,aC);
	aD = aD + aTemp1;
	aH = aTemp1 + aTemp2;
	}

static inline void CSHA256_48(	const TUint aA, 
								const TUint aB, 
								const TUint aC,
								TUint& aD, 
								const TUint aE, 
								const TUint aF,
								const TUint aG, 
								TUint& aH,
								TUint aTemp1,
								TUint aTemp2,
								const TUint aK,
								TUint& aWord0,
								const TUint aWord2,
								const TUint aWord7,
								const TUint aWord15,
								const TUint aWord16)
	{
	aWord0 = SHA256_sigma1(aWord2) + aWord7 + SHA256_sigma0(aWord15) + aWord16;
	CSHA256_16(aA, aB, aC, aD, aE, aF, aG, aH, aTemp1, aTemp2, aK, aWord0);
	}

/**
 * This function actually calculates the hash.
 * Function is defined in FIPS 180-2 section 6.2.2
 * 
 * This function is the expanded version of the following loop.
 *	for(TUint i = 0; i < 64; ++i)
 *		{
 *		if(i >= 16)
 *			{
 * 			iData[i] = SHA256_sigma1(iData[i-2]) + iData[i-7] + SHA256_sigma0(iData[i-15]) + iData[i-16];
 *			}
 *
 *		temp1 = tempH + SHA256_SIGMA1(tempE) + SHA_Ch(tempE,tempF,tempG) + K[i] + iData[i];
 *		temp2 = SHA256_SIGMA0(tempA) + SHA_Maj(tempA,tempB,tempC);
 *	    tempH = tempG;
 *	    tempG = tempF;
 *	    tempF = tempE;
 *	    tempE = tempD + temp1;
 *	    tempD = tempC;
 *	    tempC = tempB;
 *	    tempB = tempA;
 *	    tempA = temp1 + temp2;		
 *		}
 */
void CSHA224And256::Block()
	{
	TUint tempA=iA;
	TUint tempB=iB;
	TUint tempC=iC;
	TUint tempD=iD;
	TUint tempE=iE;
	TUint tempF=iF;
	TUint tempG=iG;
	TUint tempH=iH;
	TUint temp1=0;
	TUint temp2=0;
	
	CSHA256_16(tempA,tempB,tempC,tempD,tempE,tempF,tempG,tempH,temp1,temp2,K[0],iData[0]);
	CSHA256_16(tempH,tempA,tempB,tempC,tempD,tempE,tempF,tempG,temp1,temp2,K[1],iData[1]);
	CSHA256_16(tempG,tempH,tempA,tempB,tempC,tempD,tempE,tempF,temp1,temp2,K[2],iData[2]);
	CSHA256_16(tempF,tempG,tempH,tempA,tempB,tempC,tempD,tempE,temp1,temp2,K[3],iData[3]);
	CSHA256_16(tempE,tempF,tempG,tempH,tempA,tempB,tempC,tempD,temp1,temp2,K[4],iData[4]);
	CSHA256_16(tempD,tempE,tempF,tempG,tempH,tempA,tempB,tempC,temp1,temp2,K[5],iData[5]);
	CSHA256_16(tempC,tempD,tempE,tempF,tempG,tempH,tempA,tempB,temp1,temp2,K[6],iData[6]);
	CSHA256_16(tempB,tempC,tempD,tempE,tempF,tempG,tempH,tempA,temp1,temp2,K[7],iData[7]);

	CSHA256_16(tempA,tempB,tempC,tempD,tempE,tempF,tempG,tempH,temp1,temp2,K[8],iData[8]);
	CSHA256_16(tempH,tempA,tempB,tempC,tempD,tempE,tempF,tempG,temp1,temp2,K[9],iData[9]);
	CSHA256_16(tempG,tempH,tempA,tempB,tempC,tempD,tempE,tempF,temp1,temp2,K[10],iData[10]);
	CSHA256_16(tempF,tempG,tempH,tempA,tempB,tempC,tempD,tempE,temp1,temp2,K[11],iData[11]);
	CSHA256_16(tempE,tempF,tempG,tempH,tempA,tempB,tempC,tempD,temp1,temp2,K[12],iData[12]);
	CSHA256_16(tempD,tempE,tempF,tempG,tempH,tempA,tempB,tempC,temp1,temp2,K[13],iData[13]);
	CSHA256_16(tempC,tempD,tempE,tempF,tempG,tempH,tempA,tempB,temp1,temp2,K[14],iData[14]);
	CSHA256_16(tempB,tempC,tempD,tempE,tempF,tempG,tempH,tempA,temp1,temp2,K[15],iData[15]);

	CSHA256_48(	tempA, tempB, tempC, tempD, tempE, tempF, tempG, tempH, temp1, temp2,
				K[16], iData[16], iData[14], iData[9], iData[1], iData[0]);
	CSHA256_48(	tempH, tempA, tempB, tempC, tempD, tempE, tempF, tempG, temp1, temp2,
				K[17], iData[17], iData[15], iData[10], iData[2], iData[1]);
	CSHA256_48(	tempG, tempH, tempA, tempB, tempC, tempD, tempE, tempF, temp1, temp2,
				K[18], iData[18], iData[16], iData[11], iData[3], iData[2]);
	CSHA256_48(	tempF, tempG, tempH, tempA, tempB, tempC, tempD, tempE, temp1, temp2,
				K[19], iData[19], iData[17], iData[12], iData[4], iData[3]);
	CSHA256_48(	tempE, tempF, tempG, tempH, tempA, tempB, tempC, tempD, temp1, temp2,
				K[20], iData[20], iData[18], iData[13], iData[5], iData[4]);
	CSHA256_48(	tempD, tempE, tempF, tempG, tempH, tempA, tempB, tempC, temp1, temp2,
				K[21], iData[21], iData[19], iData[14], iData[6], iData[5]);
	CSHA256_48(	tempC, tempD, tempE, tempF, tempG, tempH, tempA, tempB, temp1, temp2,
				K[22], iData[22], iData[20], iData[15], iData[7], iData[6]);
	CSHA256_48(	tempB, tempC, tempD, tempE, tempF, tempG, tempH, tempA, temp1, temp2,
				K[23], iData[23], iData[21], iData[16], iData[8], iData[7]);

	CSHA256_48(	tempA, tempB, tempC, tempD, tempE, tempF, tempG, tempH, temp1, temp2,
				K[24], iData[24], iData[22], iData[17], iData[9], iData[8]);
	CSHA256_48(	tempH, tempA, tempB, tempC, tempD, tempE, tempF, tempG, temp1, temp2,
				K[25], iData[25], iData[23], iData[18], iData[10], iData[9]);
	CSHA256_48(	tempG, tempH, tempA, tempB, tempC, tempD, tempE, tempF, temp1, temp2,
				K[26], iData[26], iData[24], iData[19], iData[11], iData[10]);
	CSHA256_48(	tempF, tempG, tempH, tempA, tempB, tempC, tempD, tempE, temp1, temp2,
				K[27], iData[27], iData[25], iData[20], iData[12], iData[11]);
	CSHA256_48(	tempE, tempF, tempG, tempH, tempA, tempB, tempC, tempD, temp1, temp2,
				K[28], iData[28], iData[26], iData[21], iData[13], iData[12]);
	CSHA256_48(	tempD, tempE, tempF, tempG, tempH, tempA, tempB, tempC, temp1, temp2,
				K[29], iData[29], iData[27], iData[22], iData[14], iData[13]);
	CSHA256_48(	tempC, tempD, tempE, tempF, tempG, tempH, tempA, tempB, temp1, temp2,
				K[30], iData[30], iData[28], iData[23], iData[15], iData[14]);
	CSHA256_48(	tempB, tempC, tempD, tempE, tempF, tempG, tempH, tempA, temp1, temp2,
				K[31], iData[31], iData[29], iData[24], iData[16], iData[15]);

	CSHA256_48(	tempA, tempB, tempC, tempD, tempE, tempF, tempG, tempH, temp1, temp2,
				K[32], iData[32], iData[30], iData[25], iData[17], iData[16]);
	CSHA256_48(	tempH, tempA, tempB, tempC, tempD, tempE, tempF, tempG, temp1, temp2,
				K[33], iData[33], iData[31], iData[26], iData[18], iData[17]);
	CSHA256_48(	tempG, tempH, tempA, tempB, tempC, tempD, tempE, tempF, temp1, temp2,
				K[34], iData[34], iData[32], iData[27], iData[19], iData[18]);
	CSHA256_48(	tempF, tempG, tempH, tempA, tempB, tempC, tempD, tempE, temp1, temp2,
				K[35], iData[35], iData[33], iData[28], iData[20], iData[19]);
	CSHA256_48(	tempE, tempF, tempG, tempH, tempA, tempB, tempC, tempD, temp1, temp2,
				K[36], iData[36], iData[34], iData[29], iData[21], iData[20]);
	CSHA256_48(	tempD, tempE, tempF, tempG, tempH, tempA, tempB, tempC, temp1, temp2,
				K[37], iData[37], iData[35], iData[30], iData[22], iData[21]);
	CSHA256_48(	tempC, tempD, tempE, tempF, tempG, tempH, tempA, tempB, temp1, temp2,
				K[38], iData[38], iData[36], iData[31], iData[23], iData[22]);
	CSHA256_48(	tempB, tempC, tempD, tempE, tempF, tempG, tempH, tempA, temp1, temp2,
				K[39], iData[39], iData[37], iData[32], iData[24], iData[23]);

	CSHA256_48(	tempA, tempB, tempC, tempD, tempE, tempF, tempG, tempH, temp1, temp2,
				K[40], iData[40], iData[38], iData[33], iData[25], iData[24]);
	CSHA256_48(	tempH, tempA, tempB, tempC, tempD, tempE, tempF, tempG, temp1, temp2,
				K[41], iData[41], iData[39], iData[34], iData[26], iData[25]);
	CSHA256_48(	tempG, tempH, tempA, tempB, tempC, tempD, tempE, tempF, temp1, temp2,
				K[42], iData[42], iData[40], iData[35], iData[27], iData[26]);
	CSHA256_48(	tempF, tempG, tempH, tempA, tempB, tempC, tempD, tempE, temp1, temp2,
				K[43], iData[43], iData[41], iData[36], iData[28], iData[27]);
	CSHA256_48(	tempE, tempF, tempG, tempH, tempA, tempB, tempC, tempD, temp1, temp2,
				K[44], iData[44], iData[42], iData[37], iData[29], iData[28]);
	CSHA256_48(	tempD, tempE, tempF, tempG, tempH, tempA, tempB, tempC, temp1, temp2,
				K[45], iData[45], iData[43], iData[38], iData[30], iData[29]);
	CSHA256_48(	tempC, tempD, tempE, tempF, tempG, tempH, tempA, tempB, temp1, temp2,
				K[46], iData[46], iData[44], iData[39], iData[31], iData[30]);
	CSHA256_48(	tempB, tempC, tempD, tempE, tempF, tempG, tempH, tempA, temp1, temp2,
				K[47], iData[47], iData[45], iData[40], iData[32], iData[31]);

	CSHA256_48(	tempA, tempB, tempC, tempD, tempE, tempF, tempG, tempH, temp1, temp2,
				K[48], iData[48], iData[46], iData[41], iData[33], iData[32]);
	CSHA256_48(	tempH, tempA, tempB, tempC, tempD, tempE, tempF, tempG, temp1, temp2,
				K[49], iData[49], iData[47], iData[42], iData[34], iData[33]);
	CSHA256_48(	tempG, tempH, tempA, tempB, tempC, tempD, tempE, tempF, temp1, temp2,
				K[50], iData[50], iData[48], iData[43], iData[35], iData[34]);
	CSHA256_48(	tempF, tempG, tempH, tempA, tempB, tempC, tempD, tempE, temp1, temp2,
				K[51], iData[51], iData[49], iData[44], iData[36], iData[35]);
	CSHA256_48(	tempE, tempF, tempG, tempH, tempA, tempB, tempC, tempD, temp1, temp2,
				K[52], iData[52], iData[50], iData[45], iData[37], iData[36]);
	CSHA256_48(	tempD, tempE, tempF, tempG, tempH, tempA, tempB, tempC, temp1, temp2,
				K[53], iData[53], iData[51], iData[46], iData[38], iData[37]);
	CSHA256_48(	tempC, tempD, tempE, tempF, tempG, tempH, tempA, tempB, temp1, temp2,
				K[54], iData[54], iData[52], iData[47], iData[39], iData[38]);
	CSHA256_48(	tempB, tempC, tempD, tempE, tempF, tempG, tempH, tempA, temp1, temp2,
				K[55], iData[55], iData[53], iData[48], iData[40], iData[39]);

	CSHA256_48(	tempA, tempB, tempC, tempD, tempE, tempF, tempG, tempH, temp1, temp2,
				K[56], iData[56], iData[54], iData[49], iData[41], iData[40]);
	CSHA256_48(	tempH, tempA, tempB, tempC, tempD, tempE, tempF, tempG, temp1, temp2,
				K[57], iData[57], iData[55], iData[50], iData[42], iData[41]);
	CSHA256_48(	tempG, tempH, tempA, tempB, tempC, tempD, tempE, tempF, temp1, temp2,
				K[58], iData[58], iData[56], iData[51], iData[43], iData[42]);
	CSHA256_48(	tempF, tempG, tempH, tempA, tempB, tempC, tempD, tempE, temp1, temp2,
				K[59], iData[59], iData[57], iData[52], iData[44], iData[43]);
	CSHA256_48(	tempE, tempF, tempG, tempH, tempA, tempB, tempC, tempD, temp1, temp2,
				K[60], iData[60], iData[58], iData[53], iData[45], iData[44]);
	CSHA256_48(	tempD, tempE, tempF, tempG, tempH, tempA, tempB, tempC, temp1, temp2,
				K[61], iData[61], iData[59], iData[54], iData[46], iData[45]);
	CSHA256_48(	tempC, tempD, tempE, tempF, tempG, tempH, tempA, tempB, temp1, temp2,
				K[62], iData[62], iData[60], iData[55], iData[47], iData[46]);
	CSHA256_48(	tempB, tempC, tempD, tempE, tempF, tempG, tempH, tempA, temp1, temp2,
				K[63], iData[63], iData[61], iData[56], iData[48], iData[47]);

	iA+=tempA;
	iB+=tempB;
	iC+=tempC;
	iD+=tempD;
	iE+=tempE;
	iF+=tempF;
	iG+=tempG;
	iH+=tempH;

	iNl=0;
	}

/**
 * According to the standard, the message must be padded to an
 * even 512 bits. The first padding bit must be a '1'. The last
 * 64 bits represent the length of the original message. All bits 
 * in between should be 0. This helper function will pad the 
 * message according to those rules by filling the iData array 
 * accordingly. 
 */ 
void CSHA224And256::PadMessage()
	{
	const TUint padByte = 0x80;
	
	if(!(iNl&0x03))
		{
		iData[iNl >> 2] = 0;
		}
	iData[iNl >> 2] |= padByte << ((3 - iNl&0x03) << 3) ;

	if (iNl >= (KSHA256BlockSize - 2*sizeof(TUint))) 
		{
		if (iNl < (KSHA256BlockSize - sizeof(TUint)))
			iData[(KSHA256BlockSize >> 2) - 1]=0;		
		Block();
		Mem::FillZ(iData, KSHA256BlockSize);
		} 
	else
		{
		const TUint offset=(iNl+4)>>2; //+4 to account for the word added in the
		//switch statement above
		Mem::FillZ(iData+offset,(KSHA256BlockSize - offset*sizeof(TUint)));
		}

	//Length in bits
	TUint64 msgLength = iNh;

	iData[(KSHA256BlockSize >> 2) - 2] = (msgLength) >> 32;
	iData[(KSHA256BlockSize >> 2) - 1] = (msgLength & 0xFFFFFFFF);	
	}

inline void CSHA224And256::CopyWordToHash(TUint aVal, TUint aIndex)
	{
	TUint value = MakeWord(reinterpret_cast<TUint8*>(&aVal));
	Mem::Copy(const_cast<TUint8*>(iHash.Ptr())+ (4*aIndex), &value, sizeof(aVal));
	}

const TDes8& CSHA224And256::Final()
	{
	AddLength(iNl);
	PadMessage();
	Block();
	//
	// Generate hash value into iHash
	//
	CopyWordToHash(iA, 0);
	CopyWordToHash(iB, 1);
	CopyWordToHash(iC, 2);
	CopyWordToHash(iD, 3);
	CopyWordToHash(iE, 4);
	CopyWordToHash(iF, 5);
	CopyWordToHash(iG, 6);
	CopyWordToHash(iH, 7);
	
	return iHash;
	}

void CSHA224And256::RestoreState()
	{
	iA = iACopy;
	iB = iBCopy;
	iC = iCCopy;
	iD = iDCopy;
	iE = iECopy;
	iF = iFCopy;
	iG = iGCopy;
	iH = iHCopy;
	iNl = iNlCopy;
	iNh = iNhCopy;	
	Mem::Copy(iData, iDataCopy, KSHA256BlockSize*sizeof(TUint)); 
	}

void CSHA224And256::StoreState()
	{
	iACopy = iA;
	iBCopy = iB;
	iCCopy = iC;
	iDCopy = iD;
	iECopy = iE;
	iFCopy = iF;
	iGCopy = iG;
	iHCopy = iH;
	iNlCopy = iNl;
	iNhCopy = iNh;	
	Mem::Copy(iDataCopy, iData, KSHA256BlockSize*sizeof(TUint));
	}

// Implemented in hmacimpl.cpp or softwarehashbase.cpp
// but required as derived from MHash. No coverage here.
#ifdef _BullseyeCoverage
#pragma suppress_warnings on
#pragma BullseyeCoverage off
#pragma suppress_warnings off
#endif

