/*

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

	}