/*

 * 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;

  			}

 		}	 	

 	}