--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/crypto/weakcrypto/test/tasymmetric/tdsaprimegen.cpp	Fri Jun 11 15:32:35 2010 +0300
@@ -0,0 +1,141 @@
+/*
+* 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);
+	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
+		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;
+		}
+
+	CleanupStack::PopAndDestroy(dsaPair); 
+	CleanupStack::PopAndDestroy(); //SetThreadRandomLC
+	User::RequestComplete(status, KErrNone);
+	iActionState = CTestAction::EPostrequisite;
+	__UHEAP_MARKEND;
+	}
+