/*
* 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: 
*
*/


/**
 @file
 @internalTechnology
*/
 
#include "plugincharskeyagreestep.h"
#include "plugincharschecker.h"

#include "keypair.h"
#include "cryptokeypairgeneratorapi.h"
#include "cryptokeyagreementapi.h"

using namespace CryptoSpi;

CPluginCharsKeyAgreeStep::~CPluginCharsKeyAgreeStep()
	{
	}

CPluginCharsKeyAgreeStep::CPluginCharsKeyAgreeStep()
	{
	SetTestStepName(KPluginCharsKeyAgreeStep);
	}

TVerdict CPluginCharsKeyAgreeStep::doTestStepPreambleL()
	{
	SetTestStepResult(EPass);
	return TestStepResult();
	}
				
TVerdict CPluginCharsKeyAgreeStep::doTestStepL()
	{
	
	INFO_PRINTF1(_L("Plugin Characteristics - Key Agreement Chracteristics"));
	INFO_PRINTF2(_L("HEAP CELLS: %d"), User::CountAllocCells());
		
	if (TestStepResult()==EPass)
		{
		
		//Assume faliure, unless all is successful
		SetTestStepResult(EFail);
		
		TVariantPtrC algorithmUid;
		
		//Each of the individual parameters required to create the Key Agreement object
		//are read in from the specified INI configuration file					
		if(!GetStringFromConfig(ConfigSection(),KConfigAlgorithmUid,algorithmUid))
			{
			ERR_PRINTF1(_L("** .INI Error: Key Agreement Object Arguments Not Located **"));
			SetTestStepResult(EFail);
			}
		else
			{
			INFO_PRINTF1(_L("Creating Primes and Base Integers...")); 
	 		 
			RInteger DH_N = RInteger::NewPrimeL(64);
			CleanupClosePushL(DH_N);
			
			RInteger DH_N_MinusOne = RInteger::NewL(DH_N);
			CleanupClosePushL(DH_N_MinusOne);
			DH_N_MinusOne-=1;
			
			RInteger DH_G = RInteger::NewRandomL(TInteger::Two(), DH_N_MinusOne);	
			CleanupClosePushL(DH_G);
			
			CCryptoParams* keyParams = CCryptoParams::NewLC(); 
						
			TRAPD_LOG(err,keyParams->AddL(DH_N, KDhKeyParameterNUid));
			TRAP_LOG(err,keyParams->AddL(DH_G, KDhKeyParameterGUid));
			
			//****************************************************
			//Create Key Pair and Key Pair Generator Objects
   			CKeyPair* keyPair = NULL;
			CKeyPairGenerator * keypairImpl = NULL;
			
			INFO_PRINTF1(_L("Generating Key Pair..."));
			
			// create a key pair generator implementation interface
			TRAP_LOG(err,CKeyPairGeneratorFactory::CreateKeyPairGeneratorL(keypairImpl, 
																			KDHKeyPairGeneratorUid, 
																			keyParams));								
			CleanupStack::PushL(keypairImpl);

			// Create a Key Pair	
			TRAP_LOG(err,keypairImpl->GenerateKeyPairL(NULL, *keyParams, keyPair));
			
			CleanupStack::PushL(keyPair);
			
			//*****************************************************
			
			CKeyAgreement* keyAgreementImpl = NULL;	
				
			TRAP(err,CKeyAgreementFactory::CreateKeyAgreementL(keyAgreementImpl,
																algorithmUid,
																keyPair->PrivateKey(),
																keyParams));
			
			if(keyAgreementImpl && (err == KErrNone))
				{
				
				CleanupStack::PushL(keyAgreementImpl);
				
				INFO_PRINTF1(_L("** Successfully Loaded Key Agreement Object **"));
				
				//Define a pointer of type TCharacteristics in order to store the key agreement
				//object's characterisctics	
				const TCharacteristics* chars(NULL);
							
				//Retrieve the characteristics for the key agreement signer implementation object
				TRAP_LOG(err, keyAgreementImpl->GetCharacteristicsL(chars));
				
				//Static cast the characteristics to type TKeyAgreementCharacteristics
				const TKeyAgreementCharacteristics* keyagreeChars = static_cast<const TKeyAgreementCharacteristics*>(chars);

				//Retrieve all the Common characteristics that are required for the test
				TVariantPtrC exInterfaceUid;
				TVariantPtrC exAlgorithmUid;
				TVariantPtrC exImplementationUid;
				TVariantPtrC exCreatorName;
				TBool exFIPSApproved;
				TBool exHardwareSupported;
				TInt exMaxConcurrencySupported;
				TVariantPtrC exAlgorithmName;
				TInt exLatency;
				TInt exThroughput;
				
				if(!GetStringFromConfig(ConfigSection(),KConfigExInterfaceUid,exInterfaceUid) ||
					!GetStringFromConfig(ConfigSection(),KConfigExAlgorithmUid,exAlgorithmUid) ||
					!GetStringFromConfig(ConfigSection(),KConfigExImplementationUid,exImplementationUid) ||
					!GetStringFromConfig(ConfigSection(),KConfigExCreatorName,exCreatorName) ||
					!GetBoolFromConfig(ConfigSection(),KConfigExFIPSApproved,exFIPSApproved) ||
					!GetBoolFromConfig(ConfigSection(),KConfigExHardwareSupport,exHardwareSupported) ||
					!GetIntFromConfig(ConfigSection(),KConfigExMaxConcurrency,exMaxConcurrencySupported) ||
					!GetStringFromConfig(ConfigSection(),KConfigExAlgorithmName,exAlgorithmName) ||
					!GetIntFromConfig(ConfigSection(),KConfigExLatency,exLatency) ||
					!GetIntFromConfig(ConfigSection(),KConfigExThroughput,exThroughput))
					{
					ERR_PRINTF1(_L("** .INI Error: Expected Key Agreement/Common Characteristics Not Located **"));
					SetTestStepResult(EFail);
					}
				else
					{
			
					INFO_PRINTF1(_L("** Checking Key Agreement/Common Characteristics.... **"));
					
					CPluginCharsChecker* pluginCheck = CPluginCharsChecker::NewLC();
					
					//Retrieve the Common Characteristics from TKeyAgreementCharacteristics 
					const TCommonCharacteristics* keyagreeCommonChars = &keyagreeChars->cmn;
					
					TPtrC errorMessage;
					
					//Perform Key Agreement/Common Characteristic Checks
					if(pluginCheck->checkCommonCharacteristics(keyagreeCommonChars,
																exInterfaceUid,
																exAlgorithmUid,
																exImplementationUid,
																exCreatorName,
																exFIPSApproved,
																exHardwareSupported,
																exMaxConcurrencySupported,
																exAlgorithmName,
																exLatency,
																exThroughput,
																errorMessage))
						{
						INFO_PRINTF1(_L("** PASS : Key Agreement/Common characteristics successfully match expected values **"));
						SetTestStepResult(EPass);	
						}
					else
						{
						ERR_PRINTF2(_L("** FAIL: Characteristic Mismatch - %S **"),&errorMessage);
						}
					
					CleanupStack::PopAndDestroy(pluginCheck);
					}
				
				CleanupStack::PopAndDestroy(keyAgreementImpl);
				}
			else
				{
				ERR_PRINTF2(_L("*** FAIL: Failed to Create Key Agreement Object - %d ***"), err);	
				}
				
			CleanupStack::PopAndDestroy(6,&DH_N);
			}
		
		}
		
	INFO_PRINTF2(_L("HEAP CELLS: %d"), User::CountAllocCells());	
	return TestStepResult();
	}

TVerdict CPluginCharsKeyAgreeStep::doTestStepPostambleL()
	{
	return TestStepResult();
	}
