/*

* Copyright (c) 2008-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 "pkixcertstepbase.h"

#include "cleanuputils.h"

#include <s32file.h>

CPkixCertStepBase::~CPkixCertStepBase()

	{

	delete iConcatenatedChain;	

	iConcatenatedChain = NULL;

	iFileServer.Close();

	iProcessedOids.Reset();

	iOids.Reset();

	iRootCerts.ResetAndDestroy();

	CActiveScheduler::Install(NULL);

	delete iScheduler;

	}

TVerdict CPkixCertStepBase::doTestStepPreambleL()

	{

	//read in stuff from config

	GetBoolFromConfig(ConfigSection(), KPerformOom, iPerformOom);

	iFileServer.Connect();	

	TPtrC configString;

	//temporary storage of the DER encoded certs so we can calculate the combined size for iConcatenatedChain

	RPointerArray<HBufC8> rawCerts;

	CleanupResetAndDestroy<RPointerArray<HBufC8> >::PushL(rawCerts);

	TInt chainSize = 0;

	GetStringFromConfig(ConfigSection(), KEndEntity, configString);

	HBufC8* certBuf = ReadFileLC(configString);

	rawCerts.AppendL(certBuf);

	CleanupStack::Pop(certBuf);

	chainSize += certBuf->Size();

	RArray<TPtrC> certFileNames;

	CleanupClosePushL(certFileNames);

	GetStringArrayFromConfigL(ConfigSection(), KIntermediateCert, certFileNames);

	TInt i;

	for (i = 0; i < certFileNames.Count(); ++i)

		{

		certBuf = ReadFileLC(certFileNames[i]);

		rawCerts.AppendL(certBuf);

		CleanupStack::Pop(certBuf);

		chainSize += certBuf->Size();

		}

	CleanupStack::PopAndDestroy(&certFileNames);	

	iConcatenatedChain = HBufC8::NewL(chainSize);

	TPtr8 modPtr = iConcatenatedChain->Des();

	for (i = 0; i < rawCerts.Count(); i++)

		{

		modPtr.Append(*rawCerts[i]);

		}	

	CleanupStack::PopAndDestroy(&rawCerts);

	///////////

	//read in oid values.  the usage of these is dependant on the test step

	GetStringArrayFromConfigL(ConfigSection(), KOid, iOids);

	//convert into argument format used by APIs

	for (i=0; i < iOids.Count(); ++i)

		{

		iProcessedOids.AppendL(&(iOids[i]));

		}

	RArray<TPtrC> certFileNames2;

	CleanupClosePushL(certFileNames2);

	GetStringArrayFromConfigL(ConfigSection(), KRootCert, certFileNames2);

	CX509Certificate* cert;

	for (i = 0; i < certFileNames2.Count(); ++i)

		{

		certBuf = ReadFileLC(certFileNames2[i]);

		cert = CX509Certificate::NewLC(*certBuf);

		iRootCerts.AppendL(cert);

		CleanupStack::Pop(cert);

		CleanupStack::PopAndDestroy(certBuf);

		}

	CleanupStack::PopAndDestroy(&certFileNames2);	

	TInt uid;

	iUseUidOverload = GetIntFromConfig(ConfigSection(), KUid, uid);

	iUid.iUid = uid;

    iScheduler=new(ELeave) CActiveScheduler(); 

	CActiveScheduler::Install(iScheduler);

	return EPass;	

	}

//N.B. iCertChain must be popped and destroyed at the end of any deriving class's PerformTestL() method

void CPkixCertStepBase::PerformTestL()

	{

	iPtr.Set(*iConcatenatedChain);

	//split on which of the NewLs to use

	if (iUseUidOverload)

		{

		iCertChain = CPKIXCertChain::NewLC(iFileServer, iPtr, iUid);

		}

	else

		{		

		iCertChain = CPKIXCertChain::NewLC(iFileServer, iPtr, iRootCerts);			

		}

	}

HBufC8* CPkixCertStepBase::ReadFileLC(const TDesC& aFileName)

	{

	RFile file;

	User::LeaveIfError(file.Open(iFileServer, aFileName, EFileRead));

	CleanupClosePushL(file);

    TInt size;

    file.Size(size);

    CleanupStack::PopAndDestroy();//fileClose

    HBufC8* result = HBufC8::NewLC(size);

    TPtr8 ptr(result->Des());

    ptr.SetLength(size);

    RFileReadStream stream;

    User::LeaveIfError(stream.Open(iFileServer, aFileName, EFileStream));

    CleanupClosePushL(stream);

    stream.ReadL(ptr, size);

    CleanupStack::PopAndDestroy();//streamClose

    return result;	

	}

void CPkixCertStepBase::GetStringArrayFromConfigL(const TDesC& aSectName, const TDesC& aKeyName, RArray<TPtrC>& aArray)

	{

    HBufC* buf = HBufC::NewLC(aKeyName.Length() + KKeyFormat().Length());

    TPtr ptr(buf->Des());

    INFO_PRINTF2(_L("Parsing attribute: %S"), &aKeyName);

    TInt i = 0;

    TBool cont = ETrue;

    do

        {

        ++i;

        ptr = aKeyName;

        ptr.AppendFormat(KKeyFormat(), i);

        TPtrC val;

        cont = GetStringFromConfig(aSectName, ptr, val);

        if (cont)

            {

            User::LeaveIfError(aArray.Append(val));

            }

        } while (cont);

    INFO_PRINTF2(_L("Element count: %d"), i-1);

    CleanupStack::PopAndDestroy(buf);

	}

void CPkixCertStepBase::GetIntArrayFromConfigL(const TDesC& aSectName, const TDesC& aKeyName, RArray<TInt>& aArray)

	{

	HBufC* buf = HBufC::NewLC(aKeyName.Length() + KKeyFormat().Length());

    TPtr ptr(buf->Des());

    INFO_PRINTF2(_L("Parsing attribute: %S"), &aKeyName);

    TInt i = 0;

    TBool cont = ETrue;

    do

        {

        ++i;

        ptr = aKeyName;

        ptr.AppendFormat(KKeyFormat(), i);

        TInt val;

        cont = GetIntFromConfig(aSectName, ptr, val);

        if (cont)

            {

            User::LeaveIfError(aArray.Append(val));

            }

        } while (cont);

    INFO_PRINTF2(_L("Element count: %d"), i-1);

    CleanupStack::PopAndDestroy(buf);

	}

TVerdict CPkixCertStepBase::doTestStepL()

	{

	if (!iPerformOom)	

		{

		TRAPD(err, PerformTestL());

		if (err == KErrNone)

			{

			SetTestStepResult(EPass);

			}

		else{

			SetTestStepResult(EFail);

			}

		}

	else

		{

 		PerformOomTestL();

	    }	

   	return TestStepResult();	

	}

void CPkixCertStepBase::PerformOomTestL()

	{

 	for (TInt oomCount = 0; ; oomCount++)

 		{

 		__UHEAP_RESET;

 		__UHEAP_SETFAIL(RHeap::EDeterministic, oomCount);

 		TInt countBefore = User::CountAllocCells();

 		TRAPD(error, PerformTestL());// ----> This is the actual test that runs under OOM conditions.

 		TInt countAfter = User::CountAllocCells();

 		if (countBefore != countAfter)

 			{

 			INFO_PRINTF2(_L("OOM Failed at %d"), oomCount);

 			SetTestStepResult(EFail);  

 			break;

 			}

 		INFO_PRINTF2(_L("Result: %d"), error); 			

 		if (error == KErrNone)

 			{

			INFO_PRINTF1(_L("Test outcome : Passed"));

			SetTestStepResult(EPass); 

 			break;

 			}

		}	 	

	}