/*

 * 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 "apiteststeps.h"

 #include "cleanuputils.h"

 CPkixCertAddSupportedOidsStep::CPkixCertAddSupportedOidsStep()

 	{

 	SetTestStepName(KAddSupportedOidsTestStep);

 	}

 //1. make a copy(A) of the supported oids at the start

 //2. add in the oids from the config file(B), make a copy of these and add to A

 //3. get the supported oids now(C)

 //4. iterate over C, for each oid find copies in A and remove them.  if it is not found we error

 //5. if A is non empty we error

 void CPkixCertAddSupportedOidsStep::PerformTestL()

 	{

 	CPkixCertStepBase::PerformTestL();

 	//1. make a copy(A) of the supported oids at the start

 	const RPointerArray<TDesC>& suppOids = iCertChain->SupportedCriticalExtensions();	

 	//CleanupClosePushL(suppOids);	//don't need to do this as a ref and done in stepbase destructor

 	RPointerArray<HBufC> startSuppOids;

 	CleanupResetAndDestroy<RPointerArray<HBufC> >::PushL(startSuppOids);

 	HBufC* des;

 	TInt i;

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

 		{

 		des = (suppOids[i])->AllocLC();

 		startSuppOids.AppendL(des);

 		CleanupStack::Pop(des);

 		}

 	//2. add in the oids from the config file(B)

 	iCertChain->AddSupportedCriticalExtensionsL(iProcessedOids);

 	//make a copy of these and add to A

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

 		{

 		des = (iProcessedOids[i])->AllocLC();

 		startSuppOids.AppendL(des);

 		CleanupStack::Pop(des);

 		}

 	//3. get the supported oids now(C)

 	const RPointerArray<TDesC>& suppOids2 = iCertChain->SupportedCriticalExtensions();

 	//4. iterate over C, for each oid find copies in A and remove them.  if it is not found we error

 	TBool found;

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

 		{

 		found = EFalse;

 		for (TInt j=0; j < startSuppOids.Count(); ++j)

 			{

 			if ((*startSuppOids[j]) == (*suppOids2[i]))

 				{

 				found = ETrue;

 				delete (startSuppOids[j]);

 				startSuppOids.Remove(j);

 				--j;

 				}

 			}	//startSuppOids loop

 		if (!found)	

 			{

 			ERR_PRINTF2(_L("Extra ERROR: OID in result: %S"), suppOids2[i]);

 			User::Leave(KErrGeneral);

 			}			

 		}	//suppOids2 loop

 	//5. if A is non empty we error

 	if (startSuppOids.Count() != 0)

 		{

 		for (TInt j=0; j < startSuppOids.Count(); ++j)

 			{	

 			ERR_PRINTF2(_L("ERROR: OID missing from result: %S"), startSuppOids[j]);

 			}

 		User::Leave(KErrGeneral);

 		}

 	CleanupStack::PopAndDestroy(2, iCertChain);

 	}

 ///////////////////////////////////////////////////////////////////

 CPkixCertRemoveSupportedOidsStep::CPkixCertRemoveSupportedOidsStep()

 	{

 	SetTestStepName(KRemoveSupportedOidsTestStep);

 	}

 //1. make a copy(A) of the supported oids at the start

 //2. remove the oids read from the config file(B)

 //3. make a copy(C) of the supported oids now

 //4. add B to C

 //5. iterate over A, for each oid find and remove first occurence only in C.  if not found we error

 //6. iterate over B removing each from C

 //7. if C is non empty we error	

 void CPkixCertRemoveSupportedOidsStep::PerformTestL()

 	{

 	CPkixCertStepBase::PerformTestL();

 	//1. make a copy(A) of the supported oids at the start

 	const RPointerArray<TDesC>& suppOids = iCertChain->SupportedCriticalExtensions();	

 	//CleanupClosePushL(suppOids);	//don't need to do this as a ref and done in stepbase destructor

 	RPointerArray<HBufC> startSuppOids;

 	CleanupResetAndDestroy<RPointerArray<HBufC> >::PushL(startSuppOids);

 	HBufC* des;

 	TInt i;

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

 		{

 		des = (suppOids[i])->AllocLC();

 		startSuppOids.AppendL(des);

 		CleanupStack::Pop(des);

 		}

 	//2. remove the oids read from the config file(B)

 	iCertChain->RemoveSupportedCriticalExtensions(iProcessedOids);

 	//3. make a copy(C) of the supported oids now

 	const RPointerArray<TDesC>& suppOids2 = iCertChain->SupportedCriticalExtensions();

 	RPointerArray<HBufC> endSuppOids;

 	CleanupResetAndDestroy<RPointerArray<HBufC> >::PushL(endSuppOids);

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

 		{

 		des = (suppOids2[i])->AllocLC();

 		endSuppOids.AppendL(des);

 		CleanupStack::Pop(des);

 		}

 	////////////

 	//4. add B to C

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

 		{

 		des = (iProcessedOids[i])->AllocLC();

 		endSuppOids.AppendL(des);

 		CleanupStack::Pop(des);

 		}

 	///////////

 	//5. iterate over A, for each oid find and remove first occurence only in C.  if not found we error

 	TBool found;

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

 		{

 		found = EFalse;

 		for (TInt j=0; j < endSuppOids.Count(); ++j)

 			{

 			if ((*endSuppOids[j]) == (*startSuppOids[i]))

 				{

 				found = ETrue;

 				delete (endSuppOids[j]);

 				endSuppOids.Remove(j);

 				break;

 				}

 			}

 		if (!found)

 			{

 			ERR_PRINTF2(_L("ERROR: Extra OID removed from result: %S"), startSuppOids[i]);

 			User::Leave(KErrGeneral);

 			}

 		}

 	//6. iterate over B removing each from C

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

 		{

 		for (TInt j=0; j < endSuppOids.Count(); ++j)

 			{

 			if ((*endSuppOids[j]) == (*iProcessedOids[i]))

 				{

 				delete (endSuppOids[j]);

 				endSuppOids.Remove(j);

 				break;

 				}

 			}

 		}

 	//7. if C is non empty we error		

 	if (endSuppOids.Count() != 0)

 		{	

 		for (TInt j=0; j < endSuppOids.Count(); ++j)

 			{	

 			ERR_PRINTF2(_L("ERROR: Extra/duplicate OID found in result: %S"), endSuppOids[j]);

 			}

 		User::Leave(KErrGeneral);

 		}

 	CleanupStack::PopAndDestroy(3, iCertChain);

 	}

 /////////////////////////////////////////////////////////////////////

 CPkixCertSetSupportedOidsStep::CPkixCertSetSupportedOidsStep()

 	{

 	SetTestStepName(KSetSupportedOidsTestStep);

 	}

 //1. set the supported oids to the ones from the config(A)

 //2. make a copy(B) of supported oids

 //3. iterate over B, remove every match found in copy(C) taken from A.  if not found in C we error

 //4. if C not empty we error

 void CPkixCertSetSupportedOidsStep::PerformTestL()

 	{

 	CPkixCertStepBase::PerformTestL();

 	//1. set the supported oids to the ones from the config(A)

 	iCertChain->SetSupportedCriticalExtensionsL(iProcessedOids);

 	//2. make a copy(B) of supported oids

 	const RPointerArray<TDesC>& suppOids = iCertChain->SupportedCriticalExtensions();

 	RPointerArray<HBufC> endSuppOids;

 	CleanupResetAndDestroy<RPointerArray<HBufC> >::PushL(endSuppOids);

 	HBufC* des;

 	TInt i;	

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

 		{

 		des = (suppOids[i])->AllocLC();

 		endSuppOids.AppendL(des);

 		CleanupStack::Pop(des);

 		}

 	//3. iterate over B, remove every match found in copy(C) taken from A.  if not found in C we error

 	RPointerArray<TDesC> oidsCopy;

 	CleanupClosePushL(oidsCopy);

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

 		{

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

 		}

 	TBool found;

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

 		{

 		found = EFalse;

 		for (TInt j=(oidsCopy.Count() - 1); j >= 0; --j)

 			{

 			if ((*oidsCopy[j]) == (*endSuppOids[i]))

 				{

 				found = ETrue;

 				oidsCopy.Remove(j);

 				}

 			}

 		if (!found)	

 			{

 			ERR_PRINTF2(_L("ERROR: Extra OID found in result: %S"), endSuppOids[i]);

 			User::Leave(KErrGeneral);

 			}

 		}

 	//4. if C not empty we error	

 	if (oidsCopy.Count() != 0)

 		{

 		for (TInt j=0; j < oidsCopy.Count(); ++j)

 			{	

 			ERR_PRINTF2(_L("ERROR: OID missing from result: %S"), oidsCopy[j]);

 			}		

 		User::Leave(KErrGeneral);

 		}	

 	CleanupStack::PopAndDestroy(3, iCertChain);

 	}

 ////////////////////////////////////////////////

 CPkixCertResetSupportedOidsStep::CPkixCertResetSupportedOidsStep()

 	{

 	SetTestStepName(KResetSupportedOidsTestStep);

 	}

 //1. make a copy(A) of supported oids

 //2. set supported oids to list from config

 //3. reset supported oids

 //4. get supported oids(B)

 //5. iterate over B, find and remove first match in A.  if not found we error

 //6. if A is not empty we error

 void CPkixCertResetSupportedOidsStep::PerformTestL()

 	{

 	CPkixCertStepBase::PerformTestL();

 	//1. make a copy(A) of supported oids

 	const RPointerArray<TDesC>& suppOids = iCertChain->SupportedCriticalExtensions();	

 	//CleanupClosePushL(suppOids);	//don't need to do this as a ref and done in stepbase destructor

 	RPointerArray<HBufC> startSuppOids;

 	CleanupResetAndDestroy<RPointerArray<HBufC> >::PushL(startSuppOids);

 	HBufC* des;

 	TInt i;

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

 		{

 		des = (suppOids[i])->AllocLC();

 		startSuppOids.AppendL(des);

 		CleanupStack::Pop(des);

 		}

 	//2. set supported oids to list from config

 	iCertChain->SetSupportedCriticalExtensionsL(iProcessedOids);

 	//3. reset supported oids

 	iCertChain->ResetSupportedCriticalExtsToDefaultL();

 	//4. get supported oids(B)

 	const RPointerArray<TDesC>& suppOids2 = iCertChain->SupportedCriticalExtensions();

 	//5. iterate over B, find and remove first match in A.  if not found we error

 	TBool found;

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

 		{

 		found = EFalse;

 		for (TInt j=0; j < startSuppOids.Count(); ++j)

 			{

 			if ((*suppOids2[i]) == (*startSuppOids[j]))

 				{

 				found = ETrue;

 				delete (startSuppOids[j]);

 				startSuppOids.Remove(j);

 				break;

 				}

 			}

 		if (!found)	

 			{

 			ERR_PRINTF2(_L("ERROR: Extra OID found in result: %S"), suppOids2[i]);

 			User::Leave(KErrGeneral);

 			}

 		}

 	//6. if A is not empty we error

 	if (startSuppOids.Count() != 0)

 		{

 		for (TInt j=0; j < startSuppOids.Count(); ++j)

 			{	

 			ERR_PRINTF2(_L("ERROR: OID missing from result: %S"), startSuppOids[j]);

 			}

 		User::Leave(KErrGeneral);

 		}		

 	CleanupStack::PopAndDestroy(2, iCertChain);	

 	}