/*

* Copyright (c) 2007-2010 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: 

* Example CTestStep derived implementation

*

*/

/**

 @file

 @internalTechnology

*/

#include "hmacsetoperationmodecheckingstep.h"

#include <cryptospi/cryptohashapi.h>

#include <cryptospi/keys.h>

#include <cryptospi/plugincharacteristics.h>

using namespace CryptoSpi;

CHmacSetOperationModeCheckingStep::~CHmacSetOperationModeCheckingStep()

	{

	}

CHmacSetOperationModeCheckingStep::CHmacSetOperationModeCheckingStep()

	{

	SetTestStepName(KHmacSetOperationModeCheckingStep);

	}

TVerdict CHmacSetOperationModeCheckingStep::doTestStepPreambleL()

	{

	SetTestStepResult(EPass);

	return TestStepResult();

	}

TVerdict CHmacSetOperationModeCheckingStep::doTestStepL()

	{

	if (TestStepResult()==EPass)

		{

		//Assume faliure, unless all is successful

		SetTestStepResult(EFail);

		INFO_PRINTF1(_L("*** Hmac - Set Operation Mode Checking ***"));

		INFO_PRINTF2(_L("HEAP CELLS: %d"), User::CountAllocCells());

		TVariantPtrC algorithmUid;

		TVariantPtrC operationModeUid;

		TVariantPtrC secondOperationModeUid;

		TPtrC sourcePath;

		TPtrC expectedHash;

		TPtrC expectedHmac;

		TPtrC encryptKey;

		TVariantPtrC keyType;

		//Extract the Test Case ID parameter from the specified INI file

		if(!GetStringFromConfig(ConfigSection(),KConfigAlgorithmUid,algorithmUid) ||

			!GetStringFromConfig(ConfigSection(),KConfigOperationMode,operationModeUid) ||

			!GetStringFromConfig(ConfigSection(),KConfigSecondOperationMode,secondOperationModeUid) ||

			!GetStringFromConfig(ConfigSection(),KConfigSourcePath,sourcePath) ||

			!GetStringFromConfig(ConfigSection(),KConfigExHashHmacValue,expectedHash) ||

			!GetStringFromConfig(ConfigSection(),KConfigExSecondHashHmacValue,expectedHmac) ||

			!GetStringFromConfig(ConfigSection(),KConfigEncryptKey,encryptKey) ||

			!GetStringFromConfig(ConfigSection(),KConfigEncryptKeyType,keyType))

			{

			ERR_PRINTF1(_L("** Error: Failed to Load Configuration Parameters **"));

			SetTestStepResult(EFail);

			}

		else

			{	

			RFs fsSession;

			//Create a connection to the file server	

			User::LeaveIfError(fsSession.Connect());

			RFile sourceFile;

	    	CleanupClosePushL(sourceFile);

	    	//Open the specified source file		

	    	User::LeaveIfError(sourceFile.Open(fsSession,sourcePath, EFileRead));

			TInt sourceLength = 0;

			User::LeaveIfError(sourceFile.Size(sourceLength));

			//Create a heap based descriptor to store the data

			HBufC8* sourceData = HBufC8::NewL(sourceLength);						

			CleanupStack::PushL(sourceData);

			TPtr8 sourcePtr = sourceData->Des();

			sourceFile.Read(sourcePtr);

			if(sourcePtr.Length() != sourceLength)

				{

				ERR_PRINTF1(_L("*** Error: Reading Source File ***"));

				SetTestStepResult(EFail);	

				}

			else

				{

				//Create a pointer for the Hash + Key (Hmac) Implementation Object

				CHash* hashHmacImpl = NULL;

				//Convert encryption key to an 8 Bit Descriptor

				HBufC8* keyStr = HBufC8::NewLC(encryptKey.Length());

				TPtr8 keyStrPtr = keyStr->Des();

				keyStrPtr.Copy(encryptKey);

				//Create an new CryptoParams object to encapsulate the invalid key type and key string

				CCryptoParams* keyParams = CCryptoParams::NewL();

				CleanupStack::PushL(keyParams);

				keyParams->AddL(*keyStr,keyType);

				//Create a valid CKey Object

				TKeyProperty keyProperty;

				CKey* key = CKey::NewL(keyProperty,*keyParams);

				CleanupStack::PushL(key);

				//Construct an initial hash object with NO key, Catching any possible Leaves			

				TRAPD(err,CHashFactory::CreateHashL(

									hashHmacImpl,

									algorithmUid,

									operationModeUid,

									NULL,

									NULL));

				if(hashHmacImpl && (err == KErrNone))

					{

					//Push the Implementation Object onto the Cleanup Stack

					CleanupStack::PushL(hashHmacImpl);

					//Create a NULL TCharacteristics pointer

					const TCharacteristics* charsPtrA(NULL);

					//Retrieve the characteristics for the hash implementation object

					TRAP_LOG(err, hashHmacImpl->GetCharacteristicsL(charsPtrA));

					//Static cast the characteristics to type THashCharacteristics

					const THashCharacteristics* hashCharsPtrA = static_cast<const THashCharacteristics*>(charsPtrA);

					//The hash output size is returned in Bits, divide by 8 to get the Byte size

					TInt hashSize = hashCharsPtrA->iOutputSize/8;

					//Retrieve the final 8bit hash value and convert to 16bit												

					HBufC* hashDataA = HBufC::NewLC(hashSize);

					TPtr hashPtrA = hashDataA->Des();

					hashPtrA.Copy(hashHmacImpl->Hash(*sourceData));

					//Take the 16bit descriptor and convert the string to hexadecimal

					TVariantPtrC convertHash;

					convertHash.Set(hashPtrA);

					HBufC* resultA = convertHash.HexStringLC();							

					INFO_PRINTF2(_L("*** Hashed Data: %S ***"),&*resultA);

					INFO_PRINTF2(_L("*** Expected Hash: %S ***"),&expectedHash);

					if(*resultA == expectedHash)

						{

						INFO_PRINTF1(_L("*** PRIMARY HASH VALID - STAGE 1 PASS ***"));

						//Set the valid key within the Hmac Implementation Object

						TRAP(err,hashHmacImpl->SetKeyL(*key));

						if(err!=KErrNone)

							{

							ERR_PRINTF2(_L("*** ERROR %d: Setting Key ***"),err);

							User::Leave(err);	

							}

						else

							{

							INFO_PRINTF1(_L("*** HMAC KEY SET ***"));	

							}

						//Set the Operation Mode of the Hmac Implementation Object

						hashHmacImpl->SetOperationModeL(secondOperationModeUid);

						if(err!=KErrNone)

							{

							ERR_PRINTF3(_L("*** ERROR %d: Setting Operation Mode %S ***"),err,&secondOperationModeUid);	

							User::Leave(err);

							}

						else

							{

							INFO_PRINTF2(_L("*** OPERATION MODE SET : %S ***"),&secondOperationModeUid);	

							}

						//Create a NULL TCharacteristics pointer

						const TCharacteristics* charsPtrB(NULL);

						//Retrieve the characteristics for the hash implementation object

						TRAP_LOG(err, hashHmacImpl->GetCharacteristicsL(charsPtrB));

						//Static cast the characteristics to type THashCharacteristics

						const THashCharacteristics* hashCharsPtrB = static_cast<const THashCharacteristics*>(charsPtrB);

						//The hash output size is returned in Bits, divide by 8 to get the Byte size

						hashSize = hashCharsPtrB->iOutputSize/8;

						//Retrieve the final 8bit hash value and convert to 16bit												

						HBufC* hashDataB = HBufC::NewLC(hashSize);

						TPtr hashPtrB = hashDataB->Des();

						hashPtrB.Copy(hashHmacImpl->Hash(*sourceData));

						//Take the 16bit descriptor and convert the string to hexadecimal

						convertHash.Set(hashPtrB);

						HBufC* resultB = convertHash.HexStringLC();						

						INFO_PRINTF2(_L("*** Hashed Data: %S ***"),&*resultB);

						INFO_PRINTF2(_L("*** Expected Hash: %S ***"),&expectedHmac);

						if(*resultB == expectedHmac)

							{

							INFO_PRINTF1(_L("*** SECONDARY HASH VALID - STAGE 2 PASS ***"));

							//Set the Operation Mode of the Hmac Implementation Object

							TRAP(err,hashHmacImpl->SetOperationModeL(operationModeUid));

							if(err!=KErrNone)

								{

								ERR_PRINTF3(_L("*** ERROR %d: Setting Operation Mode %S ***"),err,&operationModeUid);	

								User::Leave(err);

								}

							else

								{

								INFO_PRINTF2(_L("*** OPERATION MODE SET : %S ***"),&operationModeUid);	

								}

							//Create a NULL TCharacteristics pointer

							const TCharacteristics* charsPtrC(NULL);

							//Retrieve the characteristics for the hash implementation object

							TRAP_LOG(err, hashHmacImpl->GetCharacteristicsL(charsPtrC));

							//Static cast the characteristics to type THashCharacteristics

							const THashCharacteristics* hashCharsPtrC = static_cast<const THashCharacteristics*>(charsPtrC);

							//The hash output size is returned in Bits, divide by 8 to get the Byte size

							hashSize = hashCharsPtrC->iOutputSize/8;

							//Retrieve the final 8bit hash value and convert to 16bit												

							HBufC* hashDataC = HBufC::NewLC(hashSize);

							TPtr hashPtrC = hashDataC->Des();

							hashPtrC.Copy(hashHmacImpl->Hash(*sourceData));

							//Take the 16bit descriptor and convert the string to hexadecimal

							convertHash.Set(hashPtrC);

							HBufC* resultC = convertHash.HexStringLC();							

							INFO_PRINTF2(_L("*** Hashed Data: %S ***"),&*resultC);

							INFO_PRINTF2(_L("*** Expected Hash: %S ***"),&expectedHash);

							if(*resultC == expectedHash)

								{

								INFO_PRINTF1(_L("*** FINAL HASH VALID - STAGE 3 PASS ***"));

								INFO_PRINTF1(_L("*** Hmac - Set Operation Mode Checking : PASS ***"));

								SetTestStepResult(EPass);

								}

							else

								{

								ERR_PRINTF1(_L("*** STAGE 3 FAIL: Hash and Expected Value Mismatch ***"));

								SetTestStepResult(EFail);	

								}

							CleanupStack::PopAndDestroy(resultC);

							CleanupStack::PopAndDestroy(hashDataC);

							}

						else

							{

							ERR_PRINTF1(_L("*** STAGE 2 FAIL: Hash and Expected Value Mismatch ***"));

							SetTestStepResult(EFail);	

							}

						CleanupStack::PopAndDestroy(resultB);

						CleanupStack::PopAndDestroy(hashDataB);	

						}

					else

						{

						ERR_PRINTF1(_L("*** STAGE 1 FAIL: Hash and Expected Value Match ***"));

						SetTestStepResult(EFail);	

						}

					CleanupStack::PopAndDestroy(resultA);

					CleanupStack::PopAndDestroy(hashDataA);

					CleanupStack::PopAndDestroy(hashHmacImpl);										

					}

				else if(err==KErrNotSupported)

					{

					if((((TUid)operationModeUid != KHashModeUid) && ((TUid)operationModeUid != KHmacModeUid)) ||

						((TUid)algorithmUid != KMd2Uid) && (TUid)algorithmUid != KMd5Uid && (TUid)algorithmUid != KSha1Uid && (TUid)algorithmUid != KMd4Uid && (TUid)algorithmUid != KSha224Uid && (TUid)algorithmUid != KSha256Uid && (TUid)algorithmUid != KSha384Uid && (TUid)algorithmUid != KSha512Uid)

						{

						ERR_PRINTF2(_L("*** Object Load Failure - Invalid Operation Mode : %d ***"), err);

						User::Leave(err);	

						}

					else

						{

						SetTestStepResult(EFail);	

						}

					}

				else

					{

					ERR_PRINTF2(_L("*** Hash/Hmac Facotry Object Load Failure : %d ***"), err);

					User::Leave(err);	

					}

				CleanupStack::PopAndDestroy(key);

				CleanupStack::PopAndDestroy(keyParams);

				CleanupStack::PopAndDestroy(keyStr);

				}

			CleanupStack::PopAndDestroy(sourceData);

			//Cleanup the Source RFile	

			CleanupStack::PopAndDestroy();	

			fsSession.Close();	

			}

		}

		else

		{

			SetTestStepResult(EFail);	

		}		

	INFO_PRINTF2(_L("HEAP CELLS: %d"), User::CountAllocCells());	

	return TestStepResult();

	}

TVerdict CHmacSetOperationModeCheckingStep::doTestStepPostambleL()

	{

	return TestStepResult();

	}