/*
* 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: 
* Implementation for testing octet string encoding/decoding
*
*/


#include "testoctetstr.h"
#include "tasn1normaltest.h"
#include <asn1enc.h>
#include <asn1dec.h>

#include <e32math.h>
#include <e32cons.h>

const TInt KMaxStringLength = 1000;

CTestOctetString* CTestOctetString::NewL(CASN1NormalTest &aASN1Action)
	{
	CTestOctetString* test = new (ELeave) CTestOctetString(aASN1Action);
	test->ConstructL();
	return test;
	}

CTestOctetString::CTestOctetString(CASN1NormalTest &aASN1Action) : CTestBase(aASN1Action)
	{
	};

void CTestOctetString::ConstructL()	
	{
	toEncodeBuf = HBufC8::NewMaxLC(KMaxStringLength);
	encodedBuf = HBufC8::NewMaxLC(KMaxStringLength + 10); // extra 10 for tag/length
	CleanupStack::Pop(2);
	};

void CTestOctetString::FillParameterArray(void)
	{
	iParameters->Append(CTestParameter::EString);
	}


CTestOctetString::~CTestOctetString()	
	{
	delete toEncodeBuf;
	delete encodedBuf;
	}

void CTestOctetString::GetName(TDes& aBuf)
	{
	aBuf.Copy(_L("Test Octet String"));
	}


TBool CTestOctetString::PerformTest(CConsoleBase& aConsole, const TPtr8 &toEncodePtr, const TInt &aTestNumber, const TInt &aTotalTests)
	{
	TPtr8 encodedPtr = encodedBuf->Des();

	// Make the encoder
	CASN1EncOctetString* encoder = CASN1EncOctetString::NewLC(toEncodePtr);

	// Do the encoding
	TUint writeLength = 0;
	encoder->WriteDERL(encodedPtr, writeLength);
	
	// Read it out again
	TASN1DecOctetString decoder;
	TInt readLength = 0;
	HBufC8* readBuf = decoder.DecodeDERL(encodedPtr, readLength);
	CleanupStack::PushL(readBuf);
	
	// Check lengths of reads + values
	if ((writeLength != STATIC_CAST(TUint, readLength)) || (*readBuf != toEncodePtr))
		{
		aConsole.Printf(_L("\nERROR!\n"));
		iASN1Action.ReportProgressL(KErrASN1EncodingError, aTestNumber, aTotalTests);
		CleanupStack::PopAndDestroy(2); // encoder, readBuf
		return(ETrue);
		}
	else
		{
		iASN1Action.ReportProgressL(KErrNone, aTestNumber, aTotalTests);
		CleanupStack::PopAndDestroy(2); // encoder, readBuf
		return(ETrue);
		}
	}


TBool CTestOctetString::PerformTestsL(CConsoleBase& aConsole)
	{
	TInt nLow = Math::Random();
	TInt64 nHigh((TInt)Math::Random());	
	TInt64 seed((nHigh << 32) + nLow);
	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::EString :
				{
				CStringTestParameter *testString = REINTERPRET_CAST(CStringTestParameter*, test);
				HBufC* value = HBufC::NewLC(KMaxStringLength);
				TPtr tBuf = value->Des();

				testString->GetValue(tBuf);

				if((tBuf.Length() % 2) != 0)
					User::Leave(KErrArgument);

				TPtr8 toEncodePtr(toEncodeBuf->Des());
				TUint theOctet;
				TInt octetPos=0;

				for(TInt octet = 0; octet < tBuf.Length(); octet+=2)
					{
					TLex lex(tBuf.Mid(octet, 2));
					
					if(lex.Val(theOctet, EHex)!=KErrNone)
						User::Leave(KErrArgument);
					toEncodePtr[octetPos++] = STATIC_CAST(TUint8, theOctet);
					};
				if(PerformTest(aConsole, toEncodePtr, currentTest, totalTests))
					{
					CleanupStack::PopAndDestroy();
					currentTest++;
					}
				else
					{
					CleanupStack::PopAndDestroy();
					return(EFalse);
					}
				break;
				}
			case CTestParameter::ERandom :
				{
				CRandomTestParameter *randomInt = REINTERPRET_CAST(CRandomTestParameter*, test);

				for (TInt count = 0; count <= randomInt->Interations(); ++count)
					{
					// Descriptor for part of buffer we'll use; fill with noise
					TInt toEncodeLength = count % KMaxStringLength;
					TPtr8 toEncodePtr(toEncodeBuf->Des());
					for (TInt i = 0; i < toEncodeLength; ++i)
						{
						toEncodePtr[i] = STATIC_CAST(TUint8, Math::Rand(seed));
						}
					if(!PerformTest(aConsole, toEncodePtr, currentTest, totalTests))
						return(EFalse);
					currentTest++;
					}

				break;
				}
			default:
				{
				return EFalse;
				}
			}
		}
	iASN1Action.ReportProgressL(KErrNone, totalTests, totalTests);
	return(ETrue);
	};