/*

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

 #include <cryptospi/cryptoasymmetriccipherapi.h>

 #include <cryptospi/cryptokeypairgeneratorapi.h>

 #include <cryptospi/keypair.h>

 #include "filereader.h"

 using namespace CryptoSpi;

 CASymmetricCipherEncryptedDataCheckStep::~CASymmetricCipherEncryptedDataCheckStep()

 	{

 	}

 CASymmetricCipherEncryptedDataCheckStep::CASymmetricCipherEncryptedDataCheckStep()

 	{

 	SetTestStepName(KASymmetricCipherEncryptedDataCheckStep);

 	}

 TVerdict CASymmetricCipherEncryptedDataCheckStep::doTestStepPreambleL()

 	{

 	SetTestStepResult(EPass);

 	return TestStepResult();

 	}

 TVerdict CASymmetricCipherEncryptedDataCheckStep::doTestStepL()

 	{

 	INFO_PRINTF1(_L("*** Asymmetric Cipher - Encrypted Data Check ***"));

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

   	if (TestStepResult()==EPass)

 		{

 		//Assume faliure, unless all is successful

 		SetTestStepResult(EFail);

 		TVariantPtrC algorithm;

 		TVariantPtrC srcPath;

 		TVariantPtrC encrypted;

 		TVariantPtrC paddingMode;

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

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

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

 			!GetStringFromConfig(ConfigSection(),KConfigExEncryptedData, encrypted))

 			{

 			User::Leave(KErrNotFound);

 			}

 		else

 			{

 			//assume failure, unless all is successful

 			SetTestStepResult(EFail);

 			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 a Symmetric Cipher 

 			CryptoSpi::CAsymmetricCipher * impl = NULL;	

 			TRAP(err,CAsymmetricCipherFactory::CreateAsymmetricCipherL(impl,

 														algorithm,

 														keyPair->PublicKey(),

 														KCryptoModeEncryptUid,

 														paddingMode,

 														NULL));

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

 				{

 				CleanupStack::PushL(impl);

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

 				//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(TPtrC8(*srcData), encryptedPtr));

 				if(err == KErrNone)

 					{

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

 					//Check that the source data matches the data thats 

 					//been encrypted then decrypted

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

 						{

 						INFO_PRINTF1(_L("PASS : Encrypted Data Matches Expected"));

 						SetTestStepResult(EPass);

 						}

 					else

 						{

 						INFO_PRINTF1(_L("FAIL : Encrypted Data and Expected Mismatch"));	

 						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 CASymmetricCipherEncryptedDataCheckStep::doTestStepPostambleL()

 	{

 	return TestStepResult();

 	}