/*

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

* testinteger.cpp

* Implementation for testing TInteger encoding/decoding

*

*/

#include "testbigint.h"

#include "tasn1normaltest.h"

#include <asn1enc.h>

#include <asn1dec.h>

#include <e32cons.h>

CTestBigInt* CTestBigInt::NewL(CASN1NormalTest &aASN1Action)

	{

	CTestBigInt* test = new (ELeave) CTestBigInt(aASN1Action);

	return test;

	}

CTestBigInt::CTestBigInt(CASN1NormalTest &aASN1Action) : CTestBase(aASN1Action)

	{

	};

void CTestBigInt::GetName(TDes& aBuf)

	{

	aBuf.Copy(_L("Test BigInteger"));

	}

void CTestBigInt::FillParameterArray(void)

	{

	iParameters->Append(CTestParameter::EInt);

	}

//	This function can leave

TBool CTestBigInt::PerformTest(CConsoleBase& aConsole, const RInteger& aTest, const TInt &aTestNumber, const TInt &aTotalTests)

	{

	// Make the value to encode

	RInteger encodedValue = RInteger::NewL(0);

	CleanupStack::PushL(encodedValue);

	encodedValue.CopyL(aTest);

	// Make the encoder

	CASN1EncBigInt* encoder = CASN1EncBigInt::NewLC(encodedValue);

	// Prepare a buffer

	HBufC8* buf = HBufC8::NewMaxLC(encoder->LengthDER());

	TPtr8 tBuf = buf->Des();

	// Write into the buffer

	TUint writeLength = 0;

	encoder->WriteDERL(tBuf, writeLength);

	// Read it out again

	TASN1DecInteger decoder;

	TInt readLength = 0;

	RInteger decodedValue = decoder.DecodeDERLongL(tBuf, readLength);

	CleanupStack::PushL(decodedValue);

	//Check that a positive integer has not been encoded as negative - DEF038956

	TBool negEncodeErr = EFalse;

	RInteger null = RInteger::NewL(0);

	CleanupStack::PushL(null);

	if (aTest > null)

		{

		HBufC8* originalValue = aTest.BufferLC();

		TPtrC8 tOriginalValue = originalValue->Des();

		//determine which byte to check (should a leading zero byte have been added)

		if ((*originalValue)[0] & 0x80)

			{

			//for example 000000FF, aTest value FF, *buf 020200FF need to confirm (*buf)[2] is 00

			//encoder->LengthDER() = 4, tOriginalValue.Length() = 1

			if ((*buf)[encoder->LengthDER() - tOriginalValue.Length() - 1] != 0)

				{

				negEncodeErr = ETrue;

				}

			}

		else

			{

			//for example 0000007F, aTest value 7F, *buf 02027F need to confirm leading bit (*buf)[2] not set

			//encoder->LengthDER() = 3, tOriginalValue.Length() = 1

			if ((*buf)[encoder->LengthDER() - tOriginalValue.Length()] & 0x80)

				{

				negEncodeErr = ETrue;

				}

			}

		CleanupStack::PopAndDestroy(); //originalValue

		}

	CleanupStack::PopAndDestroy(&null); //null

	// Check lengths of reads + values

	if ((writeLength != STATIC_CAST(TUint, readLength)) || !(decodedValue == encodedValue) || (negEncodeErr))

	//if(1)

		{

		aConsole.Printf(_L("\nERROR!  Problem integer: \n"));

		OutputIntegerL(aConsole, encodedValue);

		OutputIntegerL(aConsole, decodedValue);

		OutputEncodingL(aConsole, tBuf);

		iASN1Action.ReportProgressL(KErrASN1EncodingError, aTestNumber, aTotalTests);

		CleanupStack::PopAndDestroy(4, &encodedValue); // decodedValue, buf,  encoder, encodedValue

		return(EFalse);

		}

	else

		{

		iASN1Action.ReportProgressL(KErrNone, aTestNumber, aTotalTests);

		CleanupStack::PopAndDestroy(4, &encodedValue); // decodedValue, buf,  encoder, encodedValue

		return(ETrue);

		}

	}

void CTestBigInt::OutputIntegerL(CConsoleBase& aConsole, RInteger& aInt)

	{

	aConsole.Printf(_L("Bytes() = "));

	TBuf<10> bytes;

	bytes.AppendNum(aInt.ByteCount());

	bytes.Append(_L(", "));

	aConsole.Printf(bytes);

	aConsole.Printf(_L("Bits() = "));

	TBuf<10> bits;

	bits.AppendNum(aInt.BitCount());

	bits.Append(_L(", "));

	aConsole.Printf(bits);

	HBufC8* buf = aInt.BufferLC();

	TPtr8 ptr = buf->Des();

	TInt size = ptr.Length();

	for (TInt i = 0; i < size; ++i)

		{

		TBuf<10> tbuf;

		tbuf.AppendNumFixedWidth(ptr[i], EHex, 2);

		aConsole.Printf(tbuf);

		}

	RInteger null = RInteger::NewL(0);

	if (aInt < null)

		{

		ASSERT(EFalse);	//	Shouldn't happen for new crypto api - no signed integers

		aConsole.Printf(_L(" (-ve)"));

		}

	else

		{

		aConsole.Printf(_L(" (+ve)"));

		}

	null.Close();

	aConsole.Printf(_L("\n"));

	CleanupStack::PopAndDestroy(); // buf;

	}

TBool CTestBigInt::PerformTestsL(CConsoleBase& aConsole)

	{

	const TInt KMaxBits = 2048;

	CTestParameter* test;

	TInt totalTests, currentTest=0;

	if(!CountTests(totalTests)) return(EFalse);

	for(TInt pos = 0; pos < iValues->Count(); pos++)

		{

		test = (*iValues)[pos];

		switch(test->GetType())

			{

			case CTestParameter::EInt :

				{

				CIntTestParameter *rangeInt = REINTERPRET_CAST(CIntTestParameter*, test);

				RInteger encodedValue;

				encodedValue = RInteger::NewL(rangeInt->Value());

				CleanupStack::PushL(encodedValue);

				if(PerformTest(aConsole, encodedValue, currentTest, totalTests))

					CleanupStack::PopAndDestroy(&encodedValue);

				else

					{

					CleanupStack::PopAndDestroy(&encodedValue);

					return(EFalse);

					};

				currentTest++;

				break;

				}

			case CTestParameter::EIntRange :

				{

				CIntRangeTestParameter *rangeInt = REINTERPRET_CAST(CIntRangeTestParameter*, test);

				for(TInt test = rangeInt->Start(); test <= rangeInt->Finish(); test++)

					{

					RInteger encodedValue;

					encodedValue = RInteger::NewL(test);						

					CleanupStack::PushL(encodedValue);

					if(PerformTest(aConsole, encodedValue, currentTest, totalTests))

						CleanupStack::PopAndDestroy(&encodedValue);

					else

						{

						CleanupStack::PopAndDestroy(&encodedValue);

						return(EFalse);

						}

					currentTest++;

					};

				break;

				}

			case CTestParameter::ERandom :

				{

				CRandomTestParameter *randomInt = REINTERPRET_CAST(CRandomTestParameter*, test);

				for(TInt test = 0; test <= randomInt->Interations(); test++)

					{

					RInteger encodedValue;

					encodedValue = RInteger::NewRandomL((test % KMaxBits) + 1, TInteger::EAllBitsRandom);

					CleanupStack::PushL(encodedValue);

					if(PerformTest(aConsole, encodedValue, currentTest, totalTests))

						CleanupStack::PopAndDestroy(&encodedValue);

					else

						{	

						CleanupStack::PopAndDestroy(&encodedValue);

						return(EFalse);

						}

					currentTest++;

					}

				break;

				}

			default:

				{

				return EFalse;

				}

			}

		}

	iASN1Action.ReportProgressL(KErrNone, totalTests, totalTests);

	return(ETrue);

	};