/*

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

 #include <cryptospi/keypair.h>

 #include <cryptospi/cryptosignatureapi.h>

 #include <cryptospi/cryptokeypairgeneratorapi.h>

 using namespace CryptoSpi;

 CSignerVerifierStep::~CSignerVerifierStep()

 	{

 	}

 CSignerVerifierStep::CSignerVerifierStep()

 	{

 	SetTestStepName(KSignerVerifierStep);

 	}

 TVerdict CSignerVerifierStep::doTestStepPreambleL()

 	{

 	SetTestStepResult(EPass);

 	return TestStepResult();

 	}

 TVerdict CSignerVerifierStep::doTestStepL()

 	{

 	INFO_PRINTF1(_L("*** Signer/Verifier - Signature and Verification ***"));

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

   	if (TestStepResult()==EPass)

 		{

 		//Assume faliure, unless all is successful

 		SetTestStepResult(EFail);

 		TVariantPtrC testVariant;

 		TVariantPtrC typeVariant;

 		TVariantPtrC dataVariant;

 		if(	!GetStringFromConfig(ConfigSection(),KConfigSignVerifyType, typeVariant) 

 			|| !GetStringFromConfig(ConfigSection(),KConfigSignVerifyData, dataVariant))

 			{

 			// Leave if there's any error.

 			User::Leave(KErrNotFound);

 			}

 		else

 			{

 			TUid sigType = typeVariant;

 			//Construct a Crypto Parameters object to store the necessary key pair generator parameters

 			CCryptoParams* keyParams = CCryptoParams::NewL();

 			CleanupStack::PushL(keyParams);

 			//Create Key Pair Generator Object

 			CKeyPairGenerator * keypairImpl = NULL;

 			//Define error code as first TRAP is beyond scope

 			TInt err = 0;

 			if (sigType == KRsaSignerUid)			

 				{

 				INFO_PRINTF1(_L("Creating Key Pair Generator (RSA)..."));

 				keyParams->AddL(KKeyExponent, KRsaKeyParameterEUid);

 				keyParams->AddL(KRsaPrivateKeyStandard, KRsaKeyTypeUid);

 				//Construct an 'RSA' Key Pair Generator Implementation Object

 				TRAP_LOG(err,CKeyPairGeneratorFactory::CreateKeyPairGeneratorL(keypairImpl, 

 																KRSAKeyPairGeneratorUid, 

 																keyParams));

 				}

 			else

 				{

 				INFO_PRINTF1(_L("Creating Key Pair Generator (DSA)..."));

 				//Construct a 'DSA' Key Pair Generator Implementation Object

 				TRAP_LOG(err,CKeyPairGeneratorFactory::CreateKeyPairGeneratorL(keypairImpl,

                                             					KDSAKeyPairGeneratorUid,

                                             					NULL));

 				}

 			CleanupStack::PushL(keypairImpl);

 			// Generate a Key Pair 

 			INFO_PRINTF1(_L("Generating Key Pair..."));

 			CKeyPair* keyPair = NULL;

 			TRAP_LOG(err,keypairImpl->GenerateKeyPairL(1024, 

 													*keyParams, 

 													keyPair));

 			CleanupStack::PushL(keyPair);

 			// Create a Signer Object	

 			INFO_PRINTF1(_L("Creating Signer Object..."));

 			CCryptoParams* svParams = CCryptoParams::NewL();

 			CleanupStack::PushL(svParams);

 			CSigner * implsig = NULL;	

 		    TRAP_LOG(err,CSignatureFactory::CreateSignerL(implsig,

 													   sigType,

 													   keyPair->PublicKey(),

 													   KPaddingModePkcs1_v1_5_SignatureUid,

 													   svParams));

 			CleanupStack::PushL(implsig);

 			// Create a Verifier

 			INFO_PRINTF1(_L("Creating Verifier Object..."));

 			CVerifier * implver = NULL;	

 		    TRAP_LOG(err,CSignatureFactory::CreateVerifierL(implver,

 										   				sigType == KRsaSignerUid ? KRsaVerifierUid : KDsaVerifierUid,

 										   				keyPair->PrivateKey(),

 										   				KPaddingModePkcs1_v1_5_SignatureUid,

 										   				svParams));

 			CleanupStack::PushL(implver);

 			//Define a new signature object

 			CCryptoParams *signature = CCryptoParams::NewL();

 			CleanupStack::PushL(signature);

 			//Convert the source data to an 8 Bit Descriptor

 			HBufC8* sourceData = HBufC8::NewLC(dataVariant.Length());

 			TPtr8 sourceDataPtr = sourceData->Des();

 			sourceDataPtr.Copy(dataVariant);

 			//Set the signer object to use the 'Private' Key of the key pair

 			INFO_PRINTF1(_L("Setting Signer Private Key..."));

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

 			//Generate a signature for the source data

 			INFO_PRINTF1(_L("Signing..."));

 			TRAP_LOG(err,implsig->SignL(*sourceData, *signature));

     		TBool bResult = EFalse;	

     		//Set the verifier object to use the 'Public' Key of the key pair

     		INFO_PRINTF1(_L("Setting Verifier Public Key..."));

 			TRAP_LOG(err,implver->SetKeyL(keyPair->PublicKey()));

 			//Verify that the generated signature is valid for the given data

 			INFO_PRINTF1(_L("Verifying..."));

 			TRAP_LOG(err,implver->VerifyL(*sourceData, *signature, bResult));

 			if (bResult)

 				{

 				// verifier agrees signature is valid

 				INFO_PRINTF1(_L("PASS: Verification of Generated Signature Successful"));

 				SetTestStepResult(EPass);

 				}

 			else

 				{

 				//verification failure, signature is invalid

 				ERR_PRINTF2(_L("FAIL: Verification Failure of Generated Signature - %d"), err);

 				SetTestStepResult(EFail);

 				}

 			CleanupStack::PopAndDestroy(sourceData);	

 			CleanupStack::PopAndDestroy(signature);

 			CleanupStack::PopAndDestroy(implver);

 			CleanupStack::PopAndDestroy(implsig);

 			CleanupStack::PopAndDestroy(svParams);

 			CleanupStack::PopAndDestroy(keyPair);

 			CleanupStack::PopAndDestroy(keypairImpl);

 			CleanupStack::PopAndDestroy(keyParams);

 			}

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

 		}

 	  return TestStepResult();

 	}

 TVerdict CSignerVerifierStep::doTestStepPostambleL()

 	{

 	return TestStepResult();

 	}