/*

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

* This file is used for the verification function of the pkcs12 file

*

*/

/**

 @file 

 @internalTechnology

*/

// System Include

#include <test/testexecutelog.h>

//User Include 

#include "tpkcs12libteststep.h"

#include "tpkcs12data.h"

#include "tpkcs12common.h"

using namespace PKCS12;

const TInt KBagDataLength = 7;

const TInt KContentInfoLength = 11;

/**

Description:constructor:used for initialization of the variables declared      

@test

@internalTechnology

*/

CPKCS12LibTestStep::CPKCS12LibTestStep()

	{

	iActualResult=ETrue;

	iUnitTest=EFalse;

	iUnitTestContinueFlag = ETrue;

	iExpectedError = ETrue;

	iExpectedResult=KPositiveTest;

	}

/**

Description:Destructor

@test

@internalTechnology

*/

CPKCS12LibTestStep::~CPKCS12LibTestStep()

	{

	iFileSession.Close();

	delete iParser;	

	iPrivacyPassword.ResetAndDestroy();	

	}

/**

Function is intended to parse the pkcs12 file, compare the results and 

log the same

@internalTechnology:

@return:TVerdict -returns EFail if there is a failure else EPass

@test

@leave:- if memory allocation fails

*/

TVerdict CPKCS12LibTestStep::doTestStepL()

	{

	//__UHEAP_MARK;

	TPtrC integrityMode(KNullDesC);

	TPtrC integrityPassword(KNullDesC);

	TPtrC privacyPassword(KNullDesC);

	TPtrC dataPath(KNullDesC);

	TBool flagReturn=ETrue;

	if(iFileSession.Connect())

		{

		INFO_PRINTF1(_L("File server session failed"));

		flagReturn=EFalse;

		}

	if(flagReturn)	

		{

		if((!GetStringFromConfig(KMainSec,KFilePath,dataPath)) ||

		   !GetStringFromConfig(KMainSec,KIntegrityMode,integrityMode))

			{

			SetTestStepResult(EAbort);

			ERR_PRINTF1(_L("Input Data not found in ini file"));

			flagReturn=EFalse;

			}	

		}

	//Check the input

	if(flagReturn)

		{

		if(!GetBoolFromConfig(KMainSec,KUnitTest,iUnitTest))

			{

			iUnitTest=EFalse;

			}

		//Password Integrity Mode	

		if(integrityMode.CompareF(KPassword)==0)

			{

			if(!GetStringFromConfig(KMainSec,KIntegrityPassword,integrityPassword))

				{

				ERR_PRINTF1(_L("Problem in the Main section of the ini file"));

				SetTestStepResult(EFail);

				flagReturn=EFalse;

				}

			else

				{

				iParser	= CPkcs12Parser::NewL();

				// set the integrity password 

				iParser->SetIntegrityPasswordL(integrityPassword);

				// set the datapath

				iParser->SetDataL(dataPath);

				//set the privacy password

				ParsePrivacyPwdL();

				//parse the pkcs12 file

				iParser->ParseL();

				if(!GetBoolFromConfig(KMainSec,KIntegrityValid,iExpectedResult))

					{

					WARN_PRINTF1(_L("Integrity valid not found in ini file"));

					iExpectedResult=KPositiveTest;

					}

				if(!VerifyMainSecL())

					{

					ERR_PRINTF1(_L("Main section failed"));

					flagReturn=EFalse;

					}

				else if(!iUnitTestContinueFlag)

					{

					SetTestStepResult(EPass);

					delete iParser;

					iParser = NULL;

					flagReturn = EFalse;

					}

				}

			}

		else if(integrityMode.CompareF(KPublicIntegrity)==0)

			{

			//public Integrity Mode	

			}

		else

			{

			ERR_PRINTF1(_L("Integrity pwd not known in ini file"));

			SetTestStepResult(EFail);

			flagReturn=EFalse;

			}

		}

	if(flagReturn)

		{

		//Check the Authenticated safe section	

		if(!VerifyAuthSafe())

			{

			ERR_PRINTF1(_L("Authenticated safe section failed"));

			SetTestStepResult(EFail);

			flagReturn=EFalse;

			}

		else if(!iUnitTestContinueFlag )

			{

			SetTestStepResult(EPass);

			delete iParser;

			iParser = NULL;

			flagReturn = EFalse;

			}

		}

	if(flagReturn)	

		{

		//calculate the totalsafebags in the pkcs12 file	

		TInt totalsafebags = iParser->PfxHeader().TotalSafeBagCount();

		//calculate the totalbags(including the safecontentsbag) in the pkcs12 file						

		TInt totalbags = totalsafebags + iParser->PfxHeader().SafecontentBagCount();

		//Check the BagData section	

		if(!ChkBagDataL(totalsafebags,totalbags))

			{

			ERR_PRINTF1(_L("Bag Data section failed"));

			SetTestStepResult(EFail);

			flagReturn=EFalse;

			}

		else if(!iUnitTestContinueFlag)

			{

			delete iParser;

			iParser = NULL;

			flagReturn = EFalse;

			}

		}

	if(flagReturn)

		{

		SetTestStepResult(EPass);		

		}

	//__UHEAP_MARKEND; 

	return TestStepResult();

	}

/**

Function is intended to verify the expected error in each section of the ini file

@internalTechnology:

@return:TBool:returns EFalse if there is a failure else ETrue

@test:

*/

TBool CPKCS12LibTestStep::VerifyExpectedError(const TDesC& aSecName)

	{

	TBool flagReturn = ETrue;

	if ( iUnitTest ) 

		{

		if(GetIntFromConfig(aSecName,KExpectedError,iExpectedError)==EFalse)

			{

			INFO_PRINTF1(_L("ExpectedError not found in INI File"));

			}	

		// Checks the error returned by the library with the expected error read 

		// from the INI file.

		TInt actualError = 0;

		if ( (!aSecName.Compare(KMainSec) || !aSecName.Compare(KAuthSafeSec) ||

		     !aSecName.Compare(_L("MacData")) || (!aSecName.Left(KBagDataLength).Compare(KBagDataSec)) ||

		     (!aSecName.Left(KContentInfoLength).CompareF(KContentInfoKey))) && !iExpectedResult )

			{

			actualError = iParser->PfxHeader().Pkcs12ActualError();

			}	

		if (actualError == iExpectedError)

			{

			INFO_PRINTF1(_L("ExpectedError matchs"));

			iUnitTestContinueFlag = EFalse;

			iExpectedResult = ETrue;

			}

		}

	if(iExpectedResult!=iActualResult)

		{

		SetTestStepResult(EFail);

		flagReturn= EFalse;

		}

	iActualResult = ETrue;

	return flagReturn;	

	}

/**

Function is intended to set the privacy passwords of the contentinfotype

@internalTechnology:

@test:

@leave: if leave occurs if memory allocation fails

*/		

void CPKCS12LibTestStep::ParsePrivacyPwdL()

	{

	TPtrC pass(KNullDesC);

	//to append the number to get the actual key value

	TInt count=0;

	//return value

	TBool privacyPwdPresent=EFalse;

	//Put all the content info privacy passwords in an array 

	do

		{

		TBuf<KPrivacyPwdStrLen> privacyPwdBuf(KPrivacyPassword);

		pass.Set(KNullDesC);

		count++;

		//appending the count to the Keyvalue

		privacyPwdBuf.AppendNum(count);

		privacyPwdPresent=GetStringFromConfig(KMainSec,privacyPwdBuf,pass);

		if(privacyPwdPresent)

			{

			HBufC *temp=pass.AllocLC();

			iPrivacyPassword.AppendL(temp);

			CleanupStack::Pop(temp);

			}

		}while(privacyPwdPresent);

	iParser->SetPrivacyPassword(iPrivacyPassword);	

	}

/**

Function is intended to parse the main section of the ini file

@internalTechnology:

@return:returns EFalse if there is a failure else ETrue

@test:

*/			

TBool CPKCS12LibTestStep::VerifyMainSecL()

	{

	TBool mac=EFalse;

	TBool flagReturn=ETrue;

	//retrieve the expected result:PositiveTest/NegativeTest	

	if(!GetBoolFromConfig(KMainSec,KExpectedResult,iExpectedResult))

		{

		WARN_PRINTF1(_L("ExpectedResult not found in INI file:Default:PositiveTest"));

		iExpectedResult=KPositiveTest;

		}

	if(!GetBoolFromConfig(KMainSec,KMacPresent,mac))

		{

		WARN_PRINTF1(_L("Mac present not found in ini file"));

		//default only integration test

		mac=EFalse;

		}

	else if(mac && iUnitTest)

		{

		//call the mac data function and check for all the macdata section values

		iActualResult=ChkMacValueL();

		if (!iUnitTestContinueFlag)	

			{

			return flagReturn;

			}

		}

	if(!VerifyExpectedError(KMainSec))

		{

		SetTestStepResult(EFail);

		flagReturn = EFalse;

		}

	return flagReturn;	

	}

/**

Function is intended to parse the AuthSafe section of the ini file

@internalTechnology:

@return:returns EFalse if there is a failure else ETrue

@test:

*/			

TBool CPKCS12LibTestStep::VerifyAuthSafe()

	{

	TInt contentInfoCount=0;

	TPtrC contentSection(KNullDesC);

	TInt contentIndex=0;

	TBool flagReturn=ETrue;

	if(!GetIntFromConfig(KAuthSafeSec,KContentInfoNum,contentInfoCount))

		{

		ERR_PRINTF1(_L("Count not available in the inifile"));

		SetTestStepResult(EFail);

		flagReturn=EFalse;

		}

	//retrieve the expected result:PositiveTest/NegativeTest	

	if(GetIntFromConfig(KAuthSafeSec,KExpectedResult,iExpectedResult)==EFalse)

		{

		INFO_PRINTF1(_L("ExpectedResult not found in INI file:Default:PositiveTest"));

		}

	//verify positive or negative test

	if(!VerifyExpectedError(KAuthSafeSec()))

		{

		INFO_PRINTF1(_L("Content Info Failure"));

		SetTestStepResult(EFail);

		flagReturn = EFalse;

		}

	else if(!iUnitTestContinueFlag)

		{

		return flagReturn;

		}

	for(TInt i=1;i<=contentInfoCount;i++)

		{

		TBuf<KContentInfoBufLen> contentBuf(KContentInfoKey);

		//Append the counter to the keyname		

		contentBuf.AppendNum(i);

		if(!GetStringFromConfig(KAuthSafeSec,contentBuf,contentSection))

			{

			ERR_PRINTF1(_L("ContentSection not available in the inifile"));

			SetTestStepResult(EFail);

			flagReturn=EFalse;

			break;

			}

		//retrieve the expected result:PositiveTest/NegativeTest	

		if(GetIntFromConfig(contentSection,KExpectedResult,iExpectedResult)==EFalse)

			{

			INFO_PRINTF1(_L("ExpectedResult not found in INI file:Default:PositiveTest"));

			iExpectedResult=KPositiveTest;

			}

		// Verify before calling array to access the data.

		if(!VerifyExpectedError(contentSection))

			{

			INFO_PRINTF1(_L("Content Info Failure"));

			SetTestStepResult(EFail);

			flagReturn = EFalse;

			break;

			}

		else if(!iUnitTestContinueFlag)

			{

			flagReturn = ETrue;

			break;

			}

		//Call the function for checking the contentinfo,pass the contntsction as parameter

		iBagData=iParser->ArraySafeBagData()[contentIndex];

		contentIndex++;

		//Check the contentinfo type

		iActualResult=ChkCIType(contentSection,*iBagData);

		}

	return flagReturn;

	}

/**

Function is intended to parse the BagData section of the ini file

@param-aTotalsafebags:total number of safebags

@param-aTotalbags:total number of bags including the safecontentbags

@internalTechnology:

@return:returns EFalse if  there is a failure else return ETrue

@test:

@leave: if leave occurs at the ChkBagValL function

*/			

TBool CPKCS12LibTestStep::ChkBagDataL(TInt aTotalsafebags,TInt aTotalbags)	

	{

	TInt bagType=0;

	TInt safeContentIndex=0;

	TInt safeBagIndex=0;

	CSafeContentBag* safeContentBag;

	TBool flagReturn=ETrue;

	TInt totalSafeContentsBag = iParser->PfxHeader().SafecontentBagCount();

	//loop for the totalbags

	for(TInt bags=0;bags<aTotalbags;bags++)

		{

		TBuf<KBagDataSecBufLen> bagSection(KBagDataSec);

		//Append the bags to the sectionname :BagData1

		bagSection.AppendNum(bags+1);

		if(!GetIntFromConfig(bagSection,KBagType,bagType))

			{

			ERR_PRINTF1(_L("Not able to get the BagType from the ini file"));

			flagReturn=EFalse;

			break;

			}

		if(bagType == CDecPkcs12SafeBag::ESafeContentsBag && flagReturn)

			{

			//loop for the safecontents bags

			while(safeContentIndex < totalSafeContentsBag && flagReturn)

				{

				safeContentBag=iParser->ArraySafeContentBag()[safeContentIndex];

				if(!ChkSafeContsBagData(bagSection,*safeContentBag))

					{

					ERR_PRINTF1(_L("SafeContent Bag Failed"));

					flagReturn=EFalse;

					}

				safeContentIndex++;

				break;

				} 

			}

		else

			{

			while(safeBagIndex < aTotalsafebags && flagReturn)

				{

				//retrieve the safebag sructure pointer

				iBagData=iParser->ArraySafeBagData()[safeBagIndex];

				//Chk the BagType

				if(!ChkBagType(bagSection,*iBagData,bagType))

					{

					ERR_PRINTF1(_L("Bag Type does not match"));

					flagReturn=EFalse;

					break;

					}

				if(!iUnitTestContinueFlag)

					{

					break;

					}

				iBagAttrNumVal=0;	

				if(!ChkBagAttbutesL(bagSection,*iBagData))

					{

					ERR_PRINTF1(_L("Failure in Attributes"));

					flagReturn=EFalse;

					break;

					}	

				switch(bagType)

					{

					case CDecPkcs12SafeBag::ECertBag:

						{

						//check the bagvalue

						if(!ChkBagValL(bagSection,*iBagData))

							{

							flagReturn=EFalse;

							ERR_PRINTF1(_L("Certificate Bag Value comparison Failed"));

							}

						//check the certificate type

						if(flagReturn)	

							{

							if(!ChkCertId(bagSection,*iBagData))

								{

								flagReturn=EFalse;

								ERR_PRINTF1(_L("Certificate Bag certID comparison Failed"));

								}

							}	

						break;	

						}

					case  CDecPkcs12SafeBag::EKeyBag:

						{

						if(!ChkBagValL(bagSection,*iBagData))

							{

							flagReturn = EFalse;

							ERR_PRINTF1(_L("Key Bag Failed"));

							}	

						break;		

						}

					case CDecPkcs12SafeBag::EShroudedKeyBag:

						{

							if(!ChkEncodedBagValL(bagSection,*iBagData))

							{

							flagReturn = EFalse;

							ERR_PRINTF1(_L("Shrouded Bag Failed"));

							}