/*
* 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);	
	}