/*

 * 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: 

 * Example CTestStep derived implementation

 *

 */

 /**

  @file

  @internalTechnology

 */

 #include "symmetriccipherencryptdecryptstep.h"

 using namespace CryptoSpi;

 CSymmetricCipherEncryptDecryptStep::~CSymmetricCipherEncryptDecryptStep()

 	{

 	}

 CSymmetricCipherEncryptDecryptStep::CSymmetricCipherEncryptDecryptStep()

 	{

 	SetTestStepName(KSymmetricCipherEncryptDecryptStep);

 	}

 TVerdict CSymmetricCipherEncryptDecryptStep::doTestStepPreambleL()

 	{

 	SetTestStepResult(EPass);

 	return TestStepResult();

 	}

 TVerdict CSymmetricCipherEncryptDecryptStep::doTestStepL()

 	{

 	INFO_PRINTF1(_L("*** Symmetric Cipher - Encrypt/Decrypt ***"));

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

   	if (TestStepResult()==EPass)

 		{

 		//Assume faliure, unless all is successful

 		SetTestStepResult(EFail);

 		TVariantPtrC operationMode;

 		// Create a Symmetric Cipher with the values from the ini file	

 		CryptoSpi::CSymmetricCipher * impl = NULL;	

 		CKey* key = NULL;

 		SetupCipherL(ETrue, EFalse, operationMode, impl, key);

 		INFO_PRINTF1(_L("Plugin loaded."));

 		CleanupStack::PushL(key);

 		CleanupStack::PushL(impl);

 		HBufC8* iv = NULL;

 		TInt err(0);

 		if((TUid(operationMode) == KOperationModeCBCUid) || (TUid(operationMode) == KOperationModeCTRUid))

 			{

 			TInt blockSize(0);

 			if (TUid(operationMode) == KOperationModeCTRUid)

 				{

 				blockSize = CtrModeCalcBlockSizeL(*impl);

 				}

 			else

 				{

 				blockSize = impl->BlockSize();

 				}

 			// (blocksize is in bits so to allocate the correct number of bytes divide by 8)

 			// iv is left on the cleanup stack for the duration of the test and deleted in a conditional at the end of the outer block.

 			// If this conditional block changes, take care to update the condition for deleting this allocated IV, near the end of this function.

 			iv = HBufC8::NewLC(blockSize/8);	

 			// blocksize is in bits so to allocate the correct number of 8 byte chunks divide by 64

 			for(TInt i = 0 ; i <blockSize/64 ; i++)

 				{

 				iv->Des().Append(_L8("12345678"));

 				}

 			TRAP_LOG(err,impl->SetIvL(iv->Des()));

 			}

 		//read from src file

 		HBufC8* srcData = ReadInPlaintextL();

 		CleanupStack::PushL(srcData);

 		//Create buffer for encrypted data

 		TInt maxOutputLength = impl->MaxFinalOutputLength(srcData->Length());

 		HBufC8* encrypted =	HBufC8::NewLC(maxOutputLength);

 		TPtr8 encryptedPtr = encrypted->Des();

 		INFO_PRINTF1(_L("Encrypting Source Data..."));

 		//Perform the encryption operation

 		TRAP_LOG(err, impl->ProcessFinalL((*srcData), encryptedPtr));

 		if(encryptedPtr.Compare((*srcData)) != 0)

 			{

 			//Switch to decrypt

 			TRAP_LOG(err,impl->SetCryptoModeL(KCryptoModeDecryptUid));

 			//If in CTR mode need to reset the keystream to the start of the sequence used for encryption

 			if(TUid(operationMode) == KOperationModeCTRUid)

 				{

 				impl->SetIvL(iv->Des());

 				}

 			//Create a buffer for the decrypted data

 			maxOutputLength = encryptedPtr.Length();

 			TInt bufSize = impl->MaxFinalOutputLength(maxOutputLength);

 			HBufC8* output = HBufC8::NewLC(bufSize);

 			TPtr8 outputPtr = output->Des();

 			INFO_PRINTF1(_L("Decrypting Data..."));

 			//Perform the decryption operation

 			TRAP_LOG(err, impl->ProcessFinalL(encryptedPtr, outputPtr));

 			if(err == KErrNone)

 				{

 				//Check that the source data matches the data thats 

 				//been encrypted then decrypted

 				if(	!outputPtr.Compare(*srcData) )

 					{

 					INFO_PRINTF1(_L("PASS : Decrypted Data and Source Match"));

 					SetTestStepResult(EPass);

 					}

 				else

 					{

 					INFO_PRINTF1(_L("FAIL : Decrypted Data and Source Mismatch"));	

 					SetTestStepResult(EFail);

 					}

 				}

 			CleanupStack::PopAndDestroy(output); 

 			}

 		else

 			{

 			INFO_PRINTF1(_L("FAIL : Encrpyted Data and Source Data length and content is the same"));	

 			SetTestStepResult(EFail);	

 			}

 		CleanupStack::PopAndDestroy(encrypted); 

 		CleanupStack::PopAndDestroy(srcData);

 		if((TUid(operationMode) == KOperationModeCBCUid) || (TUid(operationMode) == KOperationModeCTRUid))

 			{

 			// Iv is left on the cleanupstack at creation.  

 			// If it becomes possible for operationMode to be modified during

 			// the test this needs to be re-engineered.

 			CleanupStack::PopAndDestroy(iv);	

 			}

 		CleanupStack::PopAndDestroy(2, key);		

 		}

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

 	return TestStepResult();

 	}

 TVerdict CSymmetricCipherEncryptDecryptStep::doTestStepPostambleL()

 	{

 	return TestStepResult();

 	}