// 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 "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:

//

/**

 @file entrystatusstep.cpp

 @internalTechnology	

*/

#include "entrystatusstep.h"

#include <x509cert.h>

// Default delay before checking state

const TInt KStateCheckDelay = 1000000;

CEntryStatusStep::CEntryStatusStep()

	{

	SetTestStepName(KEntryStatus);

	}

TVerdict CEntryStatusStep::doTestStepPreambleL()

	{

	InitializeL();

	SetTestStepResult(EPass);

	_LIT(KRequestChangeNotify, "requestchangenotify");

	_LIT(KCertEntryState, "state");

	_LIT(KNewEntryString, "ENewEntry");

	_LIT(KEntryAwaitingApprovalString, "EEntryAwaitingApproval");

	_LIT(KEntryDeniedString, "EEntryDenied");

	_LIT(KEntryApprovedString, "EEntryApproved");

	_LIT(KCancelPoint, "cancelpoint");

	_LIT(KOpen, "Open");	

	_LIT(KGetState, "GetStateL");

	_LIT(KChangeNotify, "ChangeNotify");

	TPtrC cancelPoint;

	if (!GetStringFromConfig(ConfigSection(), KCancelPoint, cancelPoint))

		{

		iCancelPoint = ENoCancel;

		}

	else if (cancelPoint.CompareF(KOpen) == 0)

		{

		iCancelPoint = EAfterOpen;

		INFO_PRINTF2(_L("This test step will call Cancel() on the cert cache session after %S()"),&KOpen);

		}

	else if (cancelPoint.CompareF(KGetState) == 0)

		{

		iCancelPoint = EAfterGetState;

		INFO_PRINTF2(_L("This test step will call Cancel() on the cert cache session after %S()"),&KGetState);

		}

	else if (cancelPoint.CompareF(KChangeNotify) == 0)

		{

		iCancelPoint = EAfterChangeNotify;

		INFO_PRINTF2(_L("This test step will call Cancel() on the cert cache session after %S()"),&KChangeNotify);

		}

	// Check if this step should wait for change notification.

	if (!GetBoolFromConfig(ConfigSection(), KRequestChangeNotify, iRequestChangeNotify))

		{

		iRequestChangeNotify = EFalse;

		}

	else if (iRequestChangeNotify)

		{

		if (iCancelPoint == ENoCancel)

			{

			INFO_PRINTF1(_L("This test step will wait for change notification."));

			}

		else if (iCancelPoint != EAfterChangeNotify)

			{

			INFO_PRINTF1(_L("Invalid test config, requesting notification but cancelling earlier."));

			SetTestStepResult(EAbort);

			return EAbort;

			}

		_LIT(KRequirePendingApproval, "requirependingapproval");

		if (!GetBoolFromConfig(ConfigSection(), KRequirePendingApproval, iRequirePendingApproval))

			{

			iRequirePendingApproval = ETrue;

			}

		if (iRequirePendingApproval)

			{

			INFO_PRINTF1(_L("This step will fail if the state is not initially EEntryAwaitingApproval."));

			}

		else

			{

			INFO_PRINTF1(_L("Notification will be requested even if the state is not initially EEntryAwaitingApproval."));

			}

		}

	TPtrC expectedState;

	if (!GetStringFromConfig(ConfigSection(), KCertEntryState, expectedState))

		{

		INFO_PRINTF1(_L("Could not read expected certificate approval state from INI, abort."));

		SetTestStepResult(EAbort);

		}

	else

		{

		if (expectedState.CompareF(KNewEntryString) == 0)

			{

			iExpectedState = ENewEntry;

			INFO_PRINTF2(_L("Certificate state is expected to be %S."),&KNewEntryString);

		 	}

		else if (expectedState.CompareF(KEntryAwaitingApprovalString) == 0)

			{

			iExpectedState = EEntryAwaitingApproval;

			INFO_PRINTF2(_L("Certificate state is expected to be %S."),&KEntryAwaitingApprovalString);

			}

		else if (expectedState.CompareF(KEntryApprovedString) == 0)

			{

			iExpectedState = EEntryApproved;

			INFO_PRINTF2(_L("Certificate state is expected to be %S."),&KEntryApprovedString);

			}

		else if (expectedState.CompareF(KEntryDeniedString) == 0)

			{

			iExpectedState = EEntryDenied;

			INFO_PRINTF2(_L("Certificate state is expected to be %S."),&KEntryDeniedString);

			}

		else

			{

			INFO_PRINTF1(_L("Invalid expected certificate state, abort."));

			SetTestStepResult(EAbort);

			}

		}

	return TestStepResult();

	}

TVerdict CEntryStatusStep::doTestStepL()

	{

	// don't continue if previous phases have aborted

	if (TestStepResult() != EPass)

		{

		return TestStepResult();

		}

	// Delay briefly to ensure that any update entry steps in concurrent tests can

	// check first (which sets the state to EAwaitingApproval.)

	User::After(KStateCheckDelay);

	// Cancel if set to do so before checking state.

	if (iCancelPoint == EAfterOpen)

		{

		INFO_PRINTF1(_L("Cancelling..."));

		Session().Cancel();

		}

	iState = Session().GetStateL();

	// log the action

	INFO_PRINTF3(_L("State of cache entry for certificate '%S' is %d."), SubjectLC(), iState);

	CleanupStack::PopAndDestroy(1); // subject

	if (iCancelPoint == EAfterGetState)

		{

		INFO_PRINTF1(_L("Cancelling..."));

		Session().Cancel();

		iState = Session().GetStateL();

		INFO_PRINTF3(_L("State of cache entry for certificate '%S' is %d."), SubjectLC(), iState);

		CleanupStack::PopAndDestroy(1); // subject

		}

	else if (iRequestChangeNotify)

		{

		if (iState == EEntryAwaitingApproval || !iRequirePendingApproval)

			{

			TRequestStatus status;

			Session().RequestNotify(status);

			if (iCancelPoint == EAfterChangeNotify)

				{

				INFO_PRINTF1(_L("Cancelling..."));

				Session().Cancel();

				}

			User::WaitForRequest(status);

			User::LeaveIfError(status.Int());

			iState = Session().GetStateL();

			// log the action

			INFO_PRINTF3(_L("Got cache change notify for certificate '%S', state = %d."), SubjectLC(), iState);

			CleanupStack::PopAndDestroy(1); // certificate status

			}

		else

			{

			// log the action

			INFO_PRINTF2(_L("Cannot wait for change notify, entry state is not %d (EEntryAwaitingApproval.)"), EEntryAwaitingApproval);

			SetTestStepResult(EFail)		;

			}

		}

	return TestStepResult();

	}

TVerdict CEntryStatusStep::doTestStepPostambleL()

	{

	if (TestStepResult() == EPass)

		{

		INFO_PRINTF1(_L("Step suceeded. Checking result..."));

		}

	else

		{

		INFO_PRINTF1(_L("Step failed."));

		return TestStepResult();

		}

	if (iExpectedState == iState)

		{

		INFO_PRINTF1(_L("Expected state %d matches actual state."));

		SetTestStepResult(EPass);

		return EPass;

		}

	else

		{

		INFO_PRINTF3(_L("Expected state %d does not match actual state %d!"), iExpectedState, iState);

		SetTestStepResult(EFail);

		return EFail;

		}

	}