/*

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

* test tool step implementation

*

*/

/**

 @file

*/

#include <e32std.h>

#include "ttesttoolstep.h"

_LIT8(KLabel,"label");

const TChar KCarriageReturn = '\r';

const TChar KLineReturn = '\n';

const TChar KDelemeter = ':';

HBufC8* ParseOutputFileLC(const TPtrC& aFileName);

TInt ReadWordL(const TDesC8& aBuffer, TInt& aPos, TPtrC8& aLine);

void ConstuctExpectedArrayL(RPointerArray<HBufC>& aArray, const TDesC8& aBuffer,TBool aCheck, TBool aException = EFalse);

void ResetAndDestroyHBufCList(TAny* aList);

//

// CTestToolListCertStep

//

CTestToolListCertStep::~CTestToolListCertStep()

	{

	}

CTestToolListCertStep::CTestToolListCertStep()

	{

	// Call base class method to set up the human readable name for logging

	SetTestStepName(KTestToolListCertStep);

	}

/**

 * Override of base class virtual. Prepares for the test run of Filetokens

 * @return TVerdict code

 */

TVerdict CTestToolListCertStep::doTestStepPreambleL()

	{

	_LIT(KActualOutput, "actualoutput");

	GetStringFromConfig(ConfigSection(), KActualOutput, iActualOutput);	

	// construct name of the cert

	if (GetStringFromConfig(ConfigSection(), KExpectedOwner, iExpectedOwner))

		{

		iOwnerExist = 1;	

		}

	if (!GetIntFromConfig(ConfigSection(), KExpectedListStore, iListStoreExist))

		{

		iListStoreExist = 0;	

		}

	if(!GetIntFromConfig(ConfigSection(), KStore, iExpectedStore))

		{

		iExpectedStore = -1;

		}

	if(!GetIntFromConfig(ConfigSection(), KExpectedNumLabel, iExpectedNumLabel))

		{

		iExpectedNumLabel = 1;

		}

	if (GetStringFromConfig(ConfigSection(), KExpectedLabel1, iExpectedLabel1))

		{

		iLabel1Exist = 1;		

		}

	if (GetStringFromConfig(ConfigSection(), KExpectedLabel2, iExpectedLabel2))

		{

		iLabel2Exist = 1;		

		}

	return TestStepResult();

	}

/**

 * Override of base class pure virtual

 * Demonstrates reading configuration parameters from an ini file section

 * @return TVerdict code

 */

TVerdict CTestToolListCertStep::doTestStepL()

	{

	if (TestStepResult() != EPass)

		{

		return TestStepResult();

		}

	CActiveScheduler* sched = NULL;

	sched = new (ELeave) CActiveScheduler;

	CleanupStack::PushL(sched);

	CActiveScheduler::Install(sched);	

	CTestToolEngine* activeEngine = CTestToolEngine::NewLC();

	if (iLabel1Exist && !iLabel2Exist)

		{

		iVerdict = activeEngine->ListCerts(iExpectedLabel1, iExpectedNumLabel);

		}

	else if (iLabel2Exist && iLabel1Exist)

		{

		iVerdict = activeEngine->ListCerts(iExpectedLabel1, iExpectedLabel2, iExpectedNumLabel);

		}

	else

		{

		RPointerArray<HBufC> CertList;

		CleanupStack::PushL(TCleanupItem(ResetAndDestroyHBufCList, &CertList));

		HBufC8* fileContents = ParseOutputFileLC(iActualOutput);

		if (iListStoreExist)

			{

			_LIT8(KLabel1, "label:");

			TLex8 lex(*fileContents);

			while(!lex.Eos())

                {

                TPtrC8 token = lex.NextToken();

                if(token.CompareF(KLabel1) == KErrNone)

                	{

                	TInt offset = lex.Offset();

                	TChar c = lex.Get();

                	while(c != KCarriageReturn && c != KLineReturn)

                		{

                		c = lex.Get();

                		} 

                	TInt end = lex.Offset();

                	TPtrC8 newtoken;

                	TInt tokenLen = end-offset-1;

					newtoken.Set(&fileContents->Des()[offset], tokenLen);

                	HBufC* tmp = HBufC::NewLC(tokenLen);

                	tmp->Des().Copy(newtoken);

                	tmp->Des().Trim();

                	CertList.Append(tmp);

                	CleanupStack::Pop(tmp);

                	}

                }

			CleanupStack::PopAndDestroy(fileContents);

			iVerdict = activeEngine->ListStore(CertList);

			}

		else

			{

			ConstuctExpectedArrayL(CertList, fileContents->Des(), ETrue);

			CleanupStack::PopAndDestroy(fileContents);

			if (iOwnerExist)

				{

				iVerdict = activeEngine->ListCerts(CertList, iExpectedOwner);

				}

			else if (iExpectedStore != -1)

				{

				iVerdict = activeEngine->ListCerts(CertList, iExpectedStore);

				}

			else

				{

				iVerdict = activeEngine->ListCerts(CertList);	

				}

			}

		CleanupStack::PopAndDestroy(&CertList);

		}

	CleanupStack::PopAndDestroy(2, sched);

	if (iVerdict == EFail)

		{

		INFO_PRINTF1(_L("Expected Certificate does not Exist"));

		}

	SetTestStepResult(iVerdict);

	return TestStepResult();

	}

/**

 * Override of base class virtual

 * @return TVerdict code

 */

TVerdict CTestToolListCertStep::doTestStepPostambleL()

	{

	CActiveScheduler::Install(NULL);

	return TestStepResult();

	}

//

// CTestToolGetTrustAppsStep

//

CTestToolGetTrustAppsStep::~CTestToolGetTrustAppsStep()

	{

	iApps.ResetAndDestroy();

	}

CTestToolGetTrustAppsStep::CTestToolGetTrustAppsStep()

	{

	// Call base class method to set up the human readable name for logging

	SetTestStepName(KTestToolGetTrustAppsStep);

	}

/**

 * Override of base class virtual. Prepares for the test run of Filetokens

 * @return TVerdict code

 */

TVerdict CTestToolGetTrustAppsStep::doTestStepPreambleL()

	{

	_LIT(KNumApps, "numapps"); // this specifies how many commands to read

	_LIT(KAppsBase, "apps"); // + number (0-based) = file to check for

	if (!GetStringFromConfig(ConfigSection(), KExpectedLabel1, iExpectedLabel))

		{

		INFO_PRINTF1(_L("label name is missing"));

		SetTestStepResult(EFail);

		}

	TInt numApps = 0;

	if (GetIntFromConfig(ConfigSection(), KNumApps, numApps) && numApps!=0)

		{

		TPtrC appName;

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

			{

			// construct name of the key

			const TInt KKeyBufSize=64;

			TBuf<KKeyBufSize> keyBuf(KAppsBase);

			keyBuf.AppendNum(i);

			if (GetStringFromConfig(ConfigSection(), keyBuf, appName))

				{			

				HBufC* apps = appName.AllocLC();

				iApps.AppendL(apps);

				CleanupStack::Pop(apps);								

				}

			else

				{

				SetTestStepResult(EFail);

				// the string must exist, otherwise the config is invalid

				ERR_PRINTF2(_L("Missing apps name for key '%S'"), &keyBuf);

				}

			}

		}

	return TestStepResult();

	}

/**

 * Override of base class pure virtual

 * Demonstrates reading configuration parameters from an ini file section

 * @return TVerdict code

 */

TVerdict CTestToolGetTrustAppsStep::doTestStepL()

	{

	if (TestStepResult() != EPass)

		{

		return TestStepResult();

		}

	CActiveScheduler* sched = NULL;

	sched = new (ELeave) CActiveScheduler;

	CleanupStack::PushL(sched);

	CActiveScheduler::Install(sched);	

	CTestToolEngine* activeEngine = CTestToolEngine::NewLC();

	iVerdict = activeEngine->CheckApps(iApps, iExpectedLabel);

	CleanupStack::PopAndDestroy(2, sched);

	if (iVerdict == EFail)

		{

		INFO_PRINTF1(_L("Certificate is not trusted for the applicaion"));

		}

	SetTestStepResult(iVerdict);

	return TestStepResult();

	}

/**

 * Override of base class virtual

 * @return TVerdict code

 */

TVerdict CTestToolGetTrustAppsStep::doTestStepPostambleL()

	{

	CActiveScheduler::Install(NULL);

	return TestStepResult();

	}

//

// CTestToolListKeyStep

//

CTestToolListKeyStep::~CTestToolListKeyStep()

	{

	}

CTestToolListKeyStep::CTestToolListKeyStep()

	{

	// Call base class method to set up the human readable name for logging

	SetTestStepName(KTestToolListKeyStep);

	}

/**

 * Override of base class virtual. Prepares for the test run of Filetokens

 * @return TVerdict code

 */

TVerdict CTestToolListKeyStep::doTestStepPreambleL()

	{

	// construct name of the key

	GetStringFromConfig(ConfigSection(), KActualOutput, iActualOutput);	

	if (!GetIntFromConfig(ConfigSection(), KExpectedListStore, iListStoreExist))

		{

		iListStoreExist = 0;	

		}

	if(!GetIntFromConfig(ConfigSection(), KStore, iExpectedStore))

		{

		iExpectedStore = -1;

		}

	if(!GetIntFromConfig(ConfigSection(), KExpectedNumLabel, iExpectedNumLabel))

		{

		iExpectedNumLabel = 1;

		}

	if (GetStringFromConfig(ConfigSection(), KExpectedLabel1, iExpectedLabel1))

		{

		iLabel1Exist = 1;		

		}

	if (GetStringFromConfig(ConfigSection(), KExpectedLabel2, iExpectedLabel2))

		{

		iLabel2Exist = 1;		

		}

	return TestStepResult();

	}

/**

 * Override of base class pure virtual

 * Demonstrates reading configuration parameters from an ini file section

 * @return TVerdict code

 */

TVerdict CTestToolListKeyStep::doTestStepL()

	{

	if (TestStepResult() != EPass)

		{

		return TestStepResult();

		}

	CActiveScheduler* sched = NULL;

	sched = new (ELeave) CActiveScheduler;

	CleanupStack::PushL(sched);

	CActiveScheduler::Install(sched);	

	CTestToolEngine* activeEngine = CTestToolEngine::NewLC();

	if (iLabel1Exist && !iLabel2Exist)

		{

		iVerdict = activeEngine->ListKeys(iExpectedLabel1, iExpectedNumLabel);

		}

	else if (iLabel2Exist && iLabel1Exist)

		{

		iVerdict = activeEngine->ListKeys(iExpectedLabel1, iExpectedLabel2, iExpectedNumLabel);

		}

	else

		{

		RPointerArray<HBufC> KeyList;

		CleanupStack::PushL(TCleanupItem(ResetAndDestroyHBufCList, &KeyList));

		HBufC8* fileContents = ParseOutputFileLC(iActualOutput);

		if (iListStoreExist)

			{

			_LIT8(KLabel1, "label:");

			TLex8 lex(*fileContents);

			while(!lex.Eos())

                {

                TPtrC8 token = lex.NextToken();

                if(token.CompareF(KLabel1) == KErrNone)

                	{

                	TInt offset = lex.Offset();

                	TChar c = lex.Get();

                	while(c != KCarriageReturn && c != KLineReturn)

                		{

                		c = lex.Get();

                		} 

                	TInt end = lex.Offset();

                	TPtrC8 newtoken;

                	TInt tokenLen = end-offset-1;

					newtoken.Set(&fileContents->Des()[offset], tokenLen);

                	HBufC* tmp = HBufC::NewLC(tokenLen);

                	tmp->Des().Copy(newtoken);

                	tmp->Des().Trim();

                	KeyList.Append(tmp);

                	CleanupStack::Pop(tmp);

                	}

                }

			CleanupStack::PopAndDestroy(fileContents);

			iVerdict = activeEngine->ListStoreKey(KeyList);

			}

		else

			{

			ConstuctExpectedArrayL(KeyList, fileContents->Des(), EFalse);

			CleanupStack::PopAndDestroy(fileContents);

			if (iExpectedStore != -1)

				{

				iVerdict = activeEngine->ListKeys(KeyList, iExpectedStore);

				}

			else

				{

				iVerdict = activeEngine->ListKeys(KeyList);	

				}

			}

		CleanupStack::PopAndDestroy(&KeyList); 

		}

	CleanupStack::PopAndDestroy(2, sched);

	if (iVerdict == EFail)

		{

		INFO_PRINTF1(_L("Expected Keys do not Exist"));

		}

	SetTestStepResult(iVerdict);

	return TestStepResult();

	}

/**

 * Override of base class virtual

 * @return TVerdict code

 */

TVerdict CTestToolListKeyStep::doTestStepPostambleL()

	{

	CActiveScheduler::Install(NULL);

	return TestStepResult();

	}

//

// CTestToolGetPolicyStep

//

CTestToolGetPolicyStep::~CTestToolGetPolicyStep()

	{

	}

CTestToolGetPolicyStep::CTestToolGetPolicyStep()

	{

	// Call base class method to set up the human readable name for logging

	SetTestStepName(KTestToolGetPolicyStep);

	}

/**

 * Override of base class virtual. Prepares for the test run of Filetokens

 * @return TVerdict code

 */

TVerdict CTestToolGetPolicyStep::doTestStepPreambleL()

	{

	// construct name of the key

	if (GetStringFromConfig(ConfigSection(), KExpectedLabel1, iExpectedLabel))

		{

		iLabelExist = 1;		

		}

	if (!GetStringFromConfig(ConfigSection(), KExpectedUser, iExpectedUser))

		{

		INFO_PRINTF1(_L("policy user name is missing"));

		SetTestStepResult(EFail);

		}

	if (!GetIntFromConfig(ConfigSection(), KExpectedUserExist, iExpectedUserExist))

		{

		iExpectedUserExist = 0;

		}

	return TestStepResult();

	}

/**

 * Override of base class pure virtual

 * Demonstrates reading configuration parameters from an ini file section

 * @return TVerdict code

 */

TVerdict CTestToolGetPolicyStep::doTestStepL()

	{

	if (TestStepResult() != EPass)

		{

		return TestStepResult();

		}

	CActiveScheduler* sched = NULL;

	sched = new (ELeave) CActiveScheduler;

	CleanupStack::PushL(sched);

	CActiveScheduler::Install(sched);	

	CTestToolEngine* activeEngine = CTestToolEngine::NewLC();

    if (iLabelExist)

		{

		iVerdict = activeEngine->GetPolicy(iExpectedLabel, iExpectedUser, iExpectedUserExist);