sl@0: /*
sl@0: * Copyright (c) 1998-2009 Nokia Corporation and/or its subsidiary(-ies).
sl@0: * All rights reserved.
sl@0: * This component and the accompanying materials are made available
sl@0: * under the terms of the License "Eclipse Public License v1.0"
sl@0: * which accompanies this distribution, and is available
sl@0: * at the URL "http://www.eclipse.org/legal/epl-v10.html".
sl@0: *
sl@0: * Initial Contributors:
sl@0: * Nokia Corporation - initial contribution.
sl@0: *
sl@0: * Contributors:
sl@0: *
sl@0: * Description: 
sl@0: *
sl@0: */
sl@0: 
sl@0: 
sl@0: #include "tdsasignfb.h"
sl@0: #include "t_input.h"
sl@0: #include "t_output.h"
sl@0: #include <asymmetric.h>
sl@0: #include <random.h>
sl@0: #include <padding.h>
sl@0: #include <bigint.h>
sl@0: #include "tbrokenrandom.h"
sl@0: 
sl@0: CTestAction* CDSASignFB::NewL(RFs& aFs, CConsoleBase& aConsole, Output& aOut, 
sl@0: 	const TTestActionSpec& aTestActionSpec)
sl@0: 	{
sl@0: 	CTestAction* self = CDSASignFB::NewLC(aFs, aConsole,
sl@0: 		aOut, aTestActionSpec);
sl@0: 	CleanupStack::Pop();
sl@0: 	return self;
sl@0: 	}
sl@0: 
sl@0: CTestAction* CDSASignFB::NewLC(RFs& aFs, CConsoleBase& aConsole, Output& aOut, 
sl@0: 	const TTestActionSpec& aTestActionSpec)
sl@0: 	{
sl@0: 	CDSASignFB* self = new(ELeave) CDSASignFB(aFs, aConsole, aOut);
sl@0: 	CleanupStack::PushL(self);
sl@0: 	self->ConstructL(aTestActionSpec);
sl@0: 	return self;
sl@0: 	}
sl@0: 
sl@0: CDSASignFB::~CDSASignFB()
sl@0: 	{
sl@0: 	delete iBody;
sl@0: 	delete iSeed;
sl@0: 	}
sl@0: 
sl@0: CDSASignFB::CDSASignFB(RFs& aFs, CConsoleBase& aConsole, Output& aOut)
sl@0: 	: CTestAction(aConsole, aOut), iFs(aFs)
sl@0: 	{
sl@0: 	}
sl@0: 
sl@0: void CDSASignFB::ConstructL(const TTestActionSpec& aTestActionSpec)
sl@0: 	{
sl@0: 	CTestAction::ConstructL(aTestActionSpec);
sl@0: 	iBody = HBufC8::NewL(aTestActionSpec.iActionBody.Length());
sl@0: 	iBody->Des().Copy(aTestActionSpec.iActionBody);
sl@0: 	
sl@0: 
sl@0: 	iSeed = Input::ParseElementHexL(*iBody, _L8("<seed>"));
sl@0: 	iKeyBits = Input::ParseIntElement(*iBody, _L8("<keybits>"), _L8("</keybits>"));
sl@0: 	}
sl@0: 
sl@0: void CDSASignFB::DoPerformPrerequisite(TRequestStatus& aStatus)
sl@0: 	{
sl@0: 	TRequestStatus* status = &aStatus;
sl@0: 	User::RequestComplete(status, KErrNone);
sl@0: 	iActionState = CTestAction::EAction;
sl@0: 	}
sl@0: 
sl@0: void CDSASignFB::DoPerformPostrequisite(TRequestStatus& aStatus)
sl@0: 	{
sl@0: 	TRequestStatus* status = &aStatus;
sl@0: 
sl@0: 	iFinished = ETrue;
sl@0: 	User::RequestComplete(status, KErrNone);
sl@0: 	}
sl@0: 
sl@0: void CDSASignFB::DoReportAction(void)
sl@0: 	{
sl@0: 	}
sl@0: 
sl@0: void CDSASignFB::DoCheckResult(TInt)
sl@0: 	{
sl@0: 	if (iResult)
sl@0: 		iConsole.Printf(_L("."));
sl@0: 	else
sl@0: 		iConsole.Printf(_L("X"));
sl@0: 	}
sl@0: 
sl@0: void CDSASignFB::PerformAction(TRequestStatus& aStatus)
sl@0: 	{
sl@0: 	__UHEAP_MARK;
sl@0: 	TRequestStatus* status = &aStatus;
sl@0: 	iResult = EFalse;
sl@0: 
sl@0: 	CRandomSetSource* rng = new(ELeave)CRandomSetSource(*iSeed);
sl@0: 	SetThreadRandomLC(rng);
sl@0: 	CDSAKeyPair* dsaPair = CDSAKeyPair::NewLC(iKeyBits);
sl@0: 	CDSASigner* signer = CDSASigner::NewLC(dsaPair->PrivateKey());
sl@0: 	CDSAVerifier* verifier = CDSAVerifier::NewLC(dsaPair->PublicKey());
sl@0: 
sl@0: 	TBuf8<128> message(128);
sl@0: 	TRandom::RandomL(message);
sl@0: 
sl@0: 	const CDSASignature* signature = signer->SignL(message);
sl@0: 	CleanupStack::PushL(const_cast<CDSASignature*>(signature));
sl@0: 
sl@0: 	if(verifier->VerifyL(message, *signature))
sl@0: 		{
sl@0: 		iResult = ETrue;
sl@0: 		}
sl@0: 	else
sl@0: 		{
sl@0: 		iOut.writeString(_L("Failure in dsasignfb with following data:\n"));
sl@0: 		iOut.writeString(_L("input data: "));
sl@0: 		iOut.writeOctetStringL(message);
sl@0: 		iOut.writeString(_L("\n"));
sl@0: 		iOut.writeString(_L("seed: "));
sl@0: 		iOut.writeOctetStringL(*iSeed);
sl@0: 		iOut.writeString(_L("\n"));
sl@0: 		}
sl@0: 	
sl@0: 	CleanupStack::PopAndDestroy(const_cast<CDSASignature*>(signature));
sl@0: 	CleanupStack::PopAndDestroy(verifier);
sl@0: 	CleanupStack::PopAndDestroy(signer);
sl@0: 	CleanupStack::PopAndDestroy(dsaPair);
sl@0: 	CleanupStack::PopAndDestroy(); //SetThreadRandomLC
sl@0: 	
sl@0: 	User::RequestComplete(status, KErrNone);
sl@0: 	iActionState = CTestAction::EPostrequisite;
sl@0: 	__UHEAP_MARKEND;
sl@0: 	}
sl@0: 
sl@0: void CDSASignFB::Hex(HBufC8& aString)
sl@0:     {
sl@0:     TPtr8 ptr=aString.Des();
sl@0:     if (aString.Length()%2)
sl@0:         {
sl@0:         ptr.SetLength(0);
sl@0:         return;
sl@0:         }
sl@0:     TInt i;
sl@0:     for (i=0;i<aString.Length();i+=2)
sl@0:         {
sl@0:         TUint8 tmp;
sl@0:         tmp=(TUint8)(aString[i]-(aString[i]>'9'?('A'-10):'0'));
sl@0:         tmp*=16;
sl@0:         tmp|=(TUint8)(aString[i+1]-(aString[i+1]>'9'?('A'-10):'0'));
sl@0:         ptr[i/2]=tmp;
sl@0:         }
sl@0:     ptr.SetLength(aString.Length()/2);
sl@0:     }
sl@0: