/*
* Copyright (c) 2003-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 "tdhvector.h"
#include <t_input.h>
#include "performancetest.h"

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

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

CDHVector::~CDHVector()
	{
	delete iN;
	delete iG;
	}

CDHVector::CDHVector(RFs& /*aFs*/, 
                                     CConsoleBase& aConsole,
                                     Output& aOut)					 
: CVectorTest(aConsole, aOut)
	{
	}

void CDHVector::ConstructL(const TTestActionSpec& aTestActionSpec)
	{
	CVectorTest::ConstructL(aTestActionSpec);

    iN = Input::ParseElementHexL(aTestActionSpec.iActionBody, _L8("<n>"));
    iG = Input::ParseElementHexL(aTestActionSpec.iActionBody, _L8("<g>"));

	}


void CDHVector::DoPerformPrerequisite(TRequestStatus& aStatus)
	{
	iN1 = RInteger::NewL(*iN);
	iN2 = RInteger::NewL(*iN);
	iG1 = RInteger::NewL(*iG);
	iG2 = RInteger::NewL(*iG);
	TRequestStatus* status = &aStatus;
	User::RequestComplete(status, KErrNone);
	iActionState = CTestAction::EAction;
	}

void CDHVector::DoPerformanceTestActionL()
	{
	iResult = ETrue;
	__UHEAP_MARK;

	CDHKeyPair* keyPair1 = CDHKeyPair::NewLC(iN1, iG1);
	CDHKeyPair* keyPair2 = CDHKeyPair::NewLC(iN2, iG2);
	CDH* dh1 = CDH::NewLC(keyPair1->PrivateKey());
	
	TTimeIntervalMicroSeconds agreeTime(0);
	TTime start, end;
	TTimeIntervalSeconds diff(0);
	const TTimeIntervalSeconds KIterationTime(iPerfTestIterations);

	TInt index = 0;

	start.UniversalTime();

	while (diff < KIterationTime)
		{
		delete const_cast<HBufC8*>(dh1->AgreeL(keyPair2->PublicKey()));
		end.UniversalTime();
		end.SecondsFrom(start, diff);
		index++;
		}
	end.UniversalTime();
	
	agreeTime = end.MicroSecondsFrom(start);
	
    CleanupStack::PopAndDestroy(dh1);
    CleanupStack::PopAndDestroy(keyPair2);
    CleanupStack::PopAndDestroy(keyPair1);

	__UHEAP_MARKEND;

	if (iResult)
		{
		TReal rate = I64REAL(agreeTime.Int64()) / index;
		TReal agreetime = I64REAL(agreeTime.Int64());
		TBuf<256> buf;
		_LIT(KEncryptTime, "\tKey Agreements: %f us/agreement (%i in %f us)\r\n");
		buf.Format(KEncryptTime, rate, index, agreetime);
		iOut.writeString(buf);
		}
	else
		{
		_LIT(KNoTimingInfo, "\tTest Failed! No benchmark data\n");
		iOut.writeString(KNoTimingInfo);
		}
}

void CDHVector::DoPerformActionL()
	{
	iResult = ETrue;
	__UHEAP_MARK;

	CDHKeyPair* keyPair1 = CDHKeyPair::NewLC(iN1, iG1);
	CDHKeyPair* keyPair2 = CDHKeyPair::NewLC(iN2, iG2);
	CDH* dh1 = CDH::NewLC(keyPair1->PrivateKey());
	CDH* dh2 = CDH::NewLC(keyPair2->PrivateKey());
	const HBufC8* key1 = dh1->AgreeL(keyPair2->PublicKey());
	CleanupStack::PushL(const_cast<HBufC8*>(key1));
	const HBufC8* key2 = dh2->AgreeL(keyPair1->PublicKey());

	if(*key1 != *key2)
		{
		iResult = EFalse;
		}

	delete const_cast<HBufC8*>(key2);
	CleanupStack::PopAndDestroy(const_cast<HBufC8*>(key1));
	CleanupStack::PopAndDestroy(dh2);
	CleanupStack::PopAndDestroy(dh1);
	CleanupStack::PopAndDestroy(keyPair2);
	CleanupStack::PopAndDestroy(keyPair1);

	__UHEAP_MARKEND;
	}