/*

* 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 "tdsavector.h"

#include "tvectorutils.h"

#include "t_input.h"

#include <bigint.h>

#include "tbrokenrandom.h"

////////////////////////////////////////////////////////////////////////////////

// CDSASignVector

////////////////////////////////////////////////////////////////////////////////

CTestAction* CDSASignVector::NewL(RFs& aFs,

                                  CConsoleBase& aConsole,

                                  Output& aOut, 

                                  const TTestActionSpec& aTestActionSpec)

	{

	CTestAction* self = CDSASignVector::NewLC(aFs, aConsole, aOut, aTestActionSpec);

	CleanupStack::Pop();

	return self;

	}

CTestAction* CDSASignVector::NewLC(RFs& aFs,

                                   CConsoleBase& aConsole,

                                   Output& aOut, 

                                   const TTestActionSpec& aTestActionSpec)

	{

	CDSASignVector* self = new(ELeave) CDSASignVector(aFs, aConsole, aOut);

	CleanupStack::PushL(self);

	self->ConstructL(aTestActionSpec);

	return self;

	}

CDSASignVector::~CDSASignVector()

	{

    delete iPrivKey;

	delete iSignature;

	delete iK;

	delete iSigInput;

	}

CDSASignVector::CDSASignVector(RFs& /*aFs*/, 

                               CConsoleBase& aConsole,

                               Output& aOut)					 

    : CVectorTest(aConsole, aOut)

	{

	}

void CDSASignVector::ConstructL(const TTestActionSpec& aTestActionSpec)

	{

	CVectorTest::ConstructL(aTestActionSpec);

    iPrivKey = VectorUtils::ReadDSAPrivateKeyL(aTestActionSpec.iActionBody);

	iMessage.Set(Input::ParseElement(aTestActionSpec.iActionBody, _L8("<m>")));

	iK = Input::ParseElementHexL(aTestActionSpec.iActionBody, _L8("<k>"));

	iSignature = VectorUtils::ReadDSASignatureL(aTestActionSpec.iActionBody);

	iSigInput = CHashingSignatureInput::NewL(CMessageDigest::ESHA1);

	iSigInput->Update(iMessage);

	}

void CDSASignVector::DoPerformActionL()

	{

	__UHEAP_MARK;

	CRandomSetSource* random = new(ELeave)CRandomSetSource(*iK);

	SetThreadRandomLC(random);

    CDSASigner* signer = CDSASigner::NewLC(*iPrivKey);

	const CDSASignature* testSig = signer->SignL(iSigInput->Final());

    iResult = (*testSig == *iSignature);

	delete testSig;

    CleanupStack::PopAndDestroy(signer);

	CleanupStack::PopAndDestroy(); //SetThreadRandomLC

	__UHEAP_MARKEND;

	}

void CDSASignVector::DoPerformanceTestActionL()

	{

	iResult = ETrue;

	}

void CDSASignVector::DoCheckResult(TInt /*aError*/)

	{

//	If using Keith's fixed up testframework for testing failures, iResult will 

//	have already been corrected for a deliberate fail result.

//	If not using Keith's testframework iResult is still a fail result at this

//	point, so now's the time to adjust for deliberate failure.

	if (!iResult)

		iResult = (iResult && iExpectedResult) || (!iResult && !iExpectedResult);

	if( iResult == EFalse )

		{

		iConsole.Printf(_L("X"));

		}

	else 

		{

		iConsole.Printf(_L("."));

		}

	}

////////////////////////////////////////////////////////////////////////////////

// CDSAVerifyVector

////////////////////////////////////////////////////////////////////////////////

CTestAction* CDSAVerifyVector::NewL(RFs& aFs,

                                  CConsoleBase& aConsole,

                                  Output& aOut, 

                                  const TTestActionSpec& aTestActionSpec)

	{

	CTestAction* self = CDSAVerifyVector::NewLC(aFs, aConsole, aOut, aTestActionSpec);

	CleanupStack::Pop();

	return self;

	}

CTestAction* CDSAVerifyVector::NewLC(RFs& aFs,

                                   CConsoleBase& aConsole,

                                   Output& aOut, 

                                   const TTestActionSpec& aTestActionSpec)

	{

	CDSAVerifyVector* self = new(ELeave) CDSAVerifyVector(aFs, aConsole, aOut);

	CleanupStack::PushL(self);

	self->ConstructL(aTestActionSpec);

	return self;

	}

CDSAVerifyVector::~CDSAVerifyVector()

	{

    delete iPubKey;

	delete iSignature;

	delete iMessage;

	delete iSigInput;

	}

CDSAVerifyVector::CDSAVerifyVector(RFs& /*aFs*/, 

                                   CConsoleBase& aConsole,

                                   Output& aOut)					 

    : CVectorTest(aConsole, aOut)

	{

	}

void CDSAVerifyVector::ConstructL(const TTestActionSpec& aTestActionSpec)

	{

	CVectorTest::ConstructL(aTestActionSpec);

    iPubKey = VectorUtils::ReadDSAPublicKeyL(aTestActionSpec.iActionBody);

	TPtrC8 message(Input::ParseElement(aTestActionSpec.iActionBody, _L8("<m>")));

	if (message.Length()==0)	

		iMessage = Input::ParseElementHexL(aTestActionSpec.iActionBody, _L8("<hexm>"));

	else

		iMessage = message.AllocL();

	iSignature = VectorUtils::ReadDSASignatureL(aTestActionSpec.iActionBody);

	iSigInput = CHashingSignatureInput::NewL(CMessageDigest::ESHA1);

	iSigInput->Update(*iMessage);

	}	

void CDSAVerifyVector::DoPerformActionL()

	{

	__UHEAP_MARK;

    CDSAVerifier* verifier = CDSAVerifier::NewLC(*iPubKey);

    iResult = verifier->VerifyL(iSigInput->Final(), *iSignature);

    CleanupStack::PopAndDestroy(verifier);

	__UHEAP_MARKEND;

	}

void CDSAVerifyVector::DoPerformanceTestActionL()

	{

	iResult = ETrue;

	}

void CDSAVerifyVector::DoCheckResult(TInt /*aError*/)

	{

//	If using Keith's fixed up testframework for testing failures, iResult will 

//	have already been corrected for a deliberate fail result.

//	If not using Keith's testframework iResult is still a fail result at this

//	point, so now's the time to adjust for deliberate failure.

	if (!iResult)

		iResult = (iResult && iExpectedResult) || (!iResult && !iExpectedResult);

	if( iResult == EFalse )

		{

		iConsole.Printf(_L("X"));

		}

	else 

		{

		iConsole.Printf(_L("."));

		}

	}