/*

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

#include <cryptospi/cryptoasymmetriccipherapi.h>

#include <cryptospi/cryptokeypairgeneratorapi.h>

#include <cryptospi/keypair.h>

#include "filereader.h"

using namespace CryptoSpi;

CASymmetricCipherEncryptDecryptStep::~CASymmetricCipherEncryptDecryptStep()

	{

	}

CASymmetricCipherEncryptDecryptStep::CASymmetricCipherEncryptDecryptStep()

	{

	SetTestStepName(KASymmetricCipherEncryptDecryptStep);

	}

TVerdict CASymmetricCipherEncryptDecryptStep::doTestStepPreambleL()

	{

	SetTestStepResult(EPass);

	return TestStepResult();

	}

TVerdict CASymmetricCipherEncryptDecryptStep::doTestStepL()

	{

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

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

  	if (TestStepResult()==EPass)

		{

		//Assume faliure, unless all is successful

		SetTestStepResult(EFail);

		TPtrC srcPath;

		TVariantPtrC paddingMode;

		if(!GetStringFromConfig(ConfigSection(),KConfigSourcePath, srcPath) ||

			!GetStringFromConfig(ConfigSection(),KConfigPaddingMode, paddingMode ))

			{

			User::Leave(KErrNotFound);

			}

		else

			{

			CCryptoParams* keyParams = CCryptoParams::NewLC(); 

			//****************************************************

			//Create Key Pair and Key Pair Generator Objects

   			CKeyPair* keyPair = NULL;

			CKeyPairGenerator * keypairImpl = NULL;

			// create an RSA key pair

			INFO_PRINTF1(_L("Generating RSA keys"));

			keyParams->AddL(KKeyExponent, KRsaKeyParameterEUid);

			keyParams->AddL(KRsaPrivateKeyStandard, KRsaKeyTypeUid);

			// create a key pair generator implementation interface

			TRAPD_LOG(err,CKeyPairGeneratorFactory::CreateKeyPairGeneratorL(

											keypairImpl, 

											KRSAKeyPairGeneratorUid, 

											keyParams));

			CleanupStack::PushL(keypairImpl);

			// Create a Key Pair	

			TRAP_LOG(err,keypairImpl->GenerateKeyPairL(1024, *keyParams, keyPair));

			CleanupStack::PushL(keyPair);

			//*****************************************************

			INFO_PRINTF1(_L("Creating Asymmetric Cipher Object..."));

			// Create an Asymmetric Cipher with the values fromt he ini	

			CryptoSpi::CAsymmetricCipher * impl = NULL;	

			TRAP(err,CAsymmetricCipherFactory::CreateAsymmetricCipherL

										(

										impl,

										KRsaCipherUid,

										keyPair->PublicKey(),

										KCryptoModeEncryptUid,

										paddingMode,

										NULL));

			if(impl && (err == KErrNone))

				{

				INFO_PRINTF1(_L("*** Asymmetric Cipher Implementation Created... ***"));

				CleanupStack::PushL(impl);

				//read from src file

				CFileReader* srcData = CFileReader::NewLC(srcPath);

				//Create buffer for encrypted data

				TInt maxOutputLength = impl->GetMaximumOutputLengthL();

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

				TPtr8 encryptedPtr = encrypted->Des();

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

				//Perform the encryption operation

				TRAP_LOG(err, impl->ProcessL(static_cast<TPtrC8>(*srcData), encryptedPtr));

				if(err == KErrNone)

					{

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

						{

						INFO_PRINTF1(_L("*** DATA ENCRYPTED ***"));

						//Switch to decrypt

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

						TRAP_LOG(err,impl->SetKeyL(keyPair->PrivateKey()));

						//Create a buffer for the decrypted data

						maxOutputLength = encryptedPtr.Length();

						TInt bufSize = impl->GetMaximumOutputLengthL();

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

						TPtr8 outputPtr = output->Des();

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

						//Perform the decryption operation

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

						if(err == KErrNone)

							{

							INFO_PRINTF1(_L("*** DATA DECRYPTED ***"));

							//Check that the source data matches the data thats 

							//been encrypted then decrypted

							if(!outputPtr.Compare(*srcData))

								{

								INFO_PRINTF1(_L("PASS : Decrypted Data matches expected Source"));

								SetTestStepResult(EPass);

								}

							else

								{

								INFO_PRINTF1(_L("FAIL : Decrypted Data and expected 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);

				CleanupStack::PopAndDestroy(impl);

				}

			else

				{

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

				User::Leave(err);	

				}

			CleanupStack::PopAndDestroy(keyPair);

			CleanupStack::PopAndDestroy(keypairImpl);	

			CleanupStack::PopAndDestroy(keyParams);

			}

		}

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

	return TestStepResult();

	}

TVerdict CASymmetricCipherEncryptDecryptStep::doTestStepPostambleL()

	{

	return TestStepResult();

	}