/*

* 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();

	}