/*
* Copyright (c) 1998-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: 
*
*/


#include "tdsaprimegen.h"
#include "t_input.h"
#include <asymmetric.h>
#include <random.h>
#include <padding.h>
#include <bigint.h>
#include "tbrokenrandom.h"

CTestAction* CDSAPrimeGen::NewL(RFs& aFs, CConsoleBase& aConsole, Output& aOut, 
	const TTestActionSpec& aTestActionSpec)
	{
	CTestAction* self = CDSAPrimeGen::NewLC(aFs, aConsole,
		aOut, aTestActionSpec);
	CleanupStack::Pop();
	return self;
	}

CTestAction* CDSAPrimeGen::NewLC(RFs& aFs, CConsoleBase& aConsole, Output& aOut, 
	const TTestActionSpec& aTestActionSpec)
	{
	CDSAPrimeGen* self = new(ELeave) CDSAPrimeGen(aFs, aConsole, aOut);
	CleanupStack::PushL(self);
	self->ConstructL(aTestActionSpec);
	return self;
	}

CDSAPrimeGen::~CDSAPrimeGen()
	{
	delete iBody;
	iP.Close();
	iQ.Close();
	delete const_cast<CDSAPrimeCertificate*>(iPrimeCert);
	}

CDSAPrimeGen::CDSAPrimeGen(RFs& aFs, CConsoleBase& aConsole, Output& aOut)
	: CTestAction(aConsole, aOut), iFs(aFs)
	{
	}

void CDSAPrimeGen::ConstructL(const TTestActionSpec& aTestActionSpec)
	{
	CTestAction::ConstructL(aTestActionSpec);
	iBody = HBufC8::NewL(aTestActionSpec.iActionBody.Length());
	iBody->Des().Copy(aTestActionSpec.iActionBody);
	

	iKeyBits = Input::ParseIntElement(*iBody, _L8("<keybits>"), _L8("</keybits>"));

	HBufC8* p = Input::ParseElementHexL(*iBody, _L8("<p>"));
	CleanupStack::PushL(p);
	iP = RInteger::NewL(*p);
	CleanupStack::PopAndDestroy(p);

	HBufC8* q = Input::ParseElementHexL(*iBody, _L8("<q>"));
	CleanupStack::PushL(q);
	iQ = RInteger::NewL(*q);
	CleanupStack::PopAndDestroy(q);

	TUint counter = Input::ParseIntElement(*iBody, _L8("<c>"), _L8("</c>"));
	HBufC8* seed = Input::ParseElementHexL(*iBody, _L8("<seed>"));
	CleanupStack::PushL(seed);
	
	iExpectedResult = Input::ParseIntElement(*iBody, _L8("<result>"), _L8("</result>"));
	
	iPrimeCert = CDSAPrimeCertificate::NewL(*seed, counter);
	CleanupStack::PopAndDestroy(seed);
	}

void CDSAPrimeGen::DoPerformPrerequisite(TRequestStatus& aStatus)
	{
	TRequestStatus* status = &aStatus;
	User::RequestComplete(status, KErrNone);
	iActionState = CTestAction::EAction;
	}

void CDSAPrimeGen::DoPerformPostrequisite(TRequestStatus& aStatus)
	{
	TRequestStatus* status = &aStatus;

	iFinished = ETrue;
	User::RequestComplete(status, KErrNone);
	}

void CDSAPrimeGen::DoReportAction(void)
	{
	}

void CDSAPrimeGen::DoCheckResult(TInt)
	{
	if (iResult)
		{
		iConsole.Printf(_L("."));
		}
	else if (iResult == iExpectedResult)
		{
		iKnownFailure = ETrue;
		iConsole.Printf(_L("."));
		}
	else
		{
		iConsole.Printf(_L("X"));
		}
	}

void CDSAPrimeGen::PerformAction(TRequestStatus& aStatus)
	{
	__UHEAP_MARK;
	TRequestStatus* status = &aStatus;
	iResult = EFalse;

	CRandomSetSource* rng = new(ELeave)CRandomSetSource(iPrimeCert->Seed());
	SetThreadRandomLC(rng);
	CDSAKeyPair* dsaPair = CDSAKeyPair::NewLC(iKeyBits);
	
	//this is more or less just calling the prime generation routine over again
	//but it does test if ValidatePrimesL and the NewLC generate the same primes
	iResult = dsaPair->PublicKey().ValidatePrimesL(*iPrimeCert);

	//we've already checked that the ValidatePrime + NewLC give same results
	//this now checks against the vectors in the test scripts
	if(iP != dsaPair->PublicKey().P() ||
		iQ != dsaPair->PublicKey().Q() || 
		iP != dsaPair->PrivateKey().P() ||
		iQ != dsaPair->PrivateKey().Q() )
		{
		iResult = EFalse;
		}
	
	// Get the prime certificate and compare the seed and counter values with the initial value supplied.
	const CDSAPrimeCertificate& primeCert = dsaPair->PrimeCertificate();

	if (((iPrimeCert->Seed()).Compare(primeCert.Seed()) != 0) ||
		(iPrimeCert->Counter() != primeCert.Counter()))
		{
		iResult = EFalse;
		}
	
	// Validate the prime length values. 
	if (CDSAParameters::ValidPrimeLength(256) ||  /* for <512 */
		CDSAParameters::ValidPrimeLength(1088) || /* for >1024 */
		CDSAParameters::ValidPrimeLength(775))    /* for non multiple of 64 */
		{
		iResult = EFalse;
		}

	CleanupStack::PopAndDestroy(dsaPair); 
	CleanupStack::PopAndDestroy(); //SetThreadRandomLC
	User::RequestComplete(status, KErrNone);
	iActionState = CTestAction::EPostrequisite;
	__UHEAP_MARKEND;
	}