/*

 * 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 "hmacincrementalhmacwithreplicatestep.h"

 #include <cryptospi/cryptohashapi.h>

 #include <cryptospi/keys.h>

 #include <cryptospi/plugincharacteristics.h>

 using namespace CryptoSpi;

 CHmacIncrementalHmacWithReplicateStep::~CHmacIncrementalHmacWithReplicateStep()

 	{

 	}

 CHmacIncrementalHmacWithReplicateStep::CHmacIncrementalHmacWithReplicateStep()

 	{

 	SetTestStepName(KHmacIncrementalHmacWithReplicateStep);

 	}

 TVerdict CHmacIncrementalHmacWithReplicateStep::doTestStepPreambleL()

 	{

 	SetTestStepResult(EPass);

 	return TestStepResult();

 	}

 TVerdict CHmacIncrementalHmacWithReplicateStep::doTestStepL()

 	{

 	if (TestStepResult()==EPass)

 		{

 		//Assume faliure, unless all is successful

 		SetTestStepResult(EFail);

 		INFO_PRINTF1(_L("*** Hmac - Incremental Hash with Replicate ***"));

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

 		TVariantPtrC algorithmUid;

 		TVariantPtrC operationModeUid;

 		TPtrC sourcePath;

 		TPtrC expectedHash;

 		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(),KConfigSourcePath,sourcePath) ||

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

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

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

 			{

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

 			SetTestStepResult(EFail);

 			}

 		else

 			{

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

 			CHash* hmacImpl = 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 key type and secret key string

 			CCryptoParams* keyParams = CCryptoParams::NewL();

 			CleanupStack::PushL(keyParams);

 			keyParams->AddL(*keyStr,keyType);

 			//Create Key Object

 			TKeyProperty keyProperty;

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

 			CleanupStack::PushL(key);

 			//Retrieve a Hmac Factory Object			

 			TRAPD(err,CHashFactory::CreateHashL(hmacImpl,

 												algorithmUid,

 												operationModeUid,

 												key,

 												NULL));											

 			if(hmacImpl && (err == KErrNone))

 				{

 				//Push the Hmac Implementation Object onto the Cleanup Stack

 				CleanupStack::PushL(hmacImpl);

 				RFs fsSession;

 				//Create a connection to the file server	

 				err = fsSession.Connect();

 				if(err != KErrNone)

 					{

 					ERR_PRINTF2(_L("*** Error: File Server Connection - %d ***"), err);

 					SetTestStepResult(EFail);

 					}	

 				else

 					{

 					RFile sourceFile;

 					CleanupClosePushL(sourceFile);

 	    			//Open the specified source file		

 	    			err = sourceFile.Open(fsSession,sourcePath, EFileRead);

 	    			if(err != KErrNone)

 						{

 						ERR_PRINTF2(_L("*** Error: Opening Source File - %d ***"), err);

 						SetTestStepResult(EFail);

 						}

 					else

 						{

 						TInt sourceLength = 0;

 						TInt readPosition = 0;

 						TInt readIncrement = 0;

 						TBool hashComplete = EFalse;

 						TBool hashReplicated = EFalse;

 						TPtrC8 hashStr;

 						CHash* hmacReplicateImpl = NULL;

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

 						//Divide the total size of the source file up into individual equal sized blocks to read

 						//over several increments

 						readIncrement = sourceLength/KDataReadBlocks;

 						do 

 							{							

 							//Create a heap based descriptor to store the data

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

 							CleanupStack::PushL(sourceData);

 							TPtr8 sourcePtr = sourceData->Des();

 							//Read in a block of data from the source file from the current position

 							err = sourceFile.Read(readPosition,sourcePtr,readIncrement);

 							//Update the read position by adding the number of bytes read

 							readPosition += readIncrement;

 							if(readPosition == readIncrement)

 								{

 								//Read in the first block from the data file into the Hmac implementation object

 								if(hashReplicated == EFalse)

 									{

 									hmacImpl->Hash(*sourceData);

 									INFO_PRINTF2(_L("Intial Hmac - Bytes Read: %d"), readPosition);

 									}

 								else

 									{

 									hmacReplicateImpl->Hash(*sourceData);

 									INFO_PRINTF2(_L("Intial Hmac (Replicate) - Bytes Read: %d"), readPosition);	

 									}

 								}

 							else if(readPosition >= sourceLength)

 								{

 								//Reading in the final block, constructs the complete hash value and returns it within a TPtrC8

 								hashStr.Set(hmacReplicateImpl->Final(*sourceData));

 								//Sets the Complete Flag to ETrue in order to drop out of the loop

 								hashComplete = ETrue;

 								TInt totalRead = (readPosition - readIncrement) + (*sourceData).Length();

 								INFO_PRINTF2(_L("Final Hmac - Bytes Read: %d"),totalRead);

 								}

 							//If the read position is half the source length and the implementation

 							//object hasn't already been replicated

 							else if((readPosition >= sourceLength/2) && (hashReplicated == EFalse))

 								{

 								INFO_PRINTF1(_L("Replicating Hmac Object..."));

 								//Create a Copy of the existing Hmac Object with NO internal message state

 								hmacReplicateImpl = hmacImpl->ReplicateL();

 								hashReplicated = ETrue;

 								//Sets the read position back to 0 inorder to restart the file read from the beginning

 								readPosition =0;

 								INFO_PRINTF2(_L("*** HMAC REPLICATE - Bytes Read: %d ***"), readPosition);

 								}

 							else

 								{

 								//Update the message data within the Hmac object with the new block

 								if(hashReplicated == EFalse)

 									{

 									hmacImpl->Update(*sourceData);

 									INFO_PRINTF2(_L("Hmac Update - Bytes Read: %d"), readPosition);		

 									}

 								else

 									{

 									hmacReplicateImpl->Update(*sourceData);

 									INFO_PRINTF2(_L("Hmac Update (Replicate) - Bytes Read: %d"), readPosition);		

 									}

 								}

 							CleanupStack::PopAndDestroy(sourceData);

 							}while(hashComplete == EFalse);

 						//Create a NULL TCharacteristics pointer

 						const TCharacteristics* charsPtr(NULL);

 						//Retrieve the characteristics for the hash implementation object

 						TRAP_LOG(err, hmacImpl->GetCharacteristicsL(charsPtr));

 						//Static cast the characteristics to type THashCharacteristics

 						const THashCharacteristics* hashCharsPtr = static_cast<const THashCharacteristics*>(charsPtr);

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

 						TInt hashSize = hashCharsPtr->iOutputSize/8;

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

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

 						TPtr hashPtr = hashData->Des();

 						hashPtr.Copy(hashStr);

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

 						TVariantPtrC convertHash;

 						convertHash.Set(hashPtr);

 						HBufC* hmacResult = convertHash.HexStringLC();								

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

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

 						//If the returned hash value matches the expected hash, Pass the test	

 						if(*hmacResult == expectedHash)							

 							{

 							INFO_PRINTF1(_L("*** Hmac - Incremental Hash with Replicate : PASS ***"));

 							SetTestStepResult(EPass);	

 							}

 						else

 							{

 							ERR_PRINTF2(_L("*** FAIL: Hashed and Expected Value Mismatch  ***"), err);

 							SetTestStepResult(EFail);	

 							}						

 						CleanupStack::PopAndDestroy(hmacResult);

 						CleanupStack::PopAndDestroy(hashData);

 						delete hmacReplicateImpl;

 						}

 					//Cleanup the Source RFile	

 					CleanupStack::PopAndDestroy();	

 					}

 				fsSession.Close();

 				CleanupStack::PopAndDestroy(hmacImpl);	

 				}

 			else

 				{

 				ERR_PRINTF2(_L("*** FAIL: Failed to Create Hmac Object - %d ***"), err);

 				SetTestStepResult(EFail);	

 				}

 			CleanupStack::PopAndDestroy(key);

 			CleanupStack::PopAndDestroy(keyParams);

 			CleanupStack::PopAndDestroy(keyStr);

 			}

 		}

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

 	return TestStepResult();  

 	}

 TVerdict CHmacIncrementalHmacWithReplicateStep::doTestStepPostambleL()

 	{

 	return TestStepResult();

 	}