// 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 multicancelstep.cpp
 @internalTechnology
*/
#include "multicancelstep.h"
#include <tlsprovinterface.h>


CMultiCancelStep::CMultiCancelStep()
	{
	SetTestStepName(KMultiCancelStep);
	}
	
TVerdict CMultiCancelStep::doTestStepL()
	{
	INFO_PRINTF1(_L("Calling TLS Provider to fetch cipher suites."));
	
	// Gets cipher suite with cancellation to increase test code coverage.
	TInt err = GetCipherSuitesWithCancelL();
	if (err != KErrNone && err != KErrCancel)
		{
		INFO_PRINTF2(_L("Unnexpected error when retrieving supported cipher suites! (with cancellation) (Error %d)"),			err);
		SetTestStepResult(EFail);
		return TestStepResult();
		}
	
	// first we have to retrieve the available cipher suites
	err = 0;
	err = GetCipherSuitesL();
	
	if (err != KErrNone)
		{
		INFO_PRINTF2(_L("Failed! Cannot retrieve supported cipher suites! (Error %d)"),
			err);
		SetTestStepResult(EFail);
		return TestStepResult();
		}
		
	// verifies certificate if is not a PSK cipher suite
  	if( !UsePsk() )
		{
			// we have to verify the server certificate, to supply the certificate
		// and its parameters to the TLS provider.

		INFO_PRINTF1(_L("Calling TLS Provider to verify server certificate, followed of cancellation of request."));

		CX509Certificate* cert = NULL;
		
		// Important part of the test: cancellation
 		err = VerifyServerCertificateWithCancelL(cert);
		delete cert;
		
			// make sure it returned was cancelled.
		if (err != KErrCancel)
			{
			INFO_PRINTF2(_L("Failed! Server Certificate cancelled returned incorrect Error: %d"),
				err);
			SetTestStepResult(EFail);
			return TestStepResult();
			}
				
		// Reissue same request but this time it does not cancel request.
		err = VerifyServerCertificateL(cert);
		delete cert;
				
		// make sure it completed sucessfully.
		if (err != KErrNone)
			{
			INFO_PRINTF2(_L("Failed! Server Certificate did not verify correctly! (Error %d)"),
					err);
			SetTestStepResult(EFail);
			return TestStepResult();
			}
		}   
	
  	INFO_PRINTF1(_L("Creating TLS Session."));	
  	// now, create a session with followed of a cancel of request
  	err = CreateSessionWithCancelL();
  	if (err != KErrCancel)
		{
		INFO_PRINTF2(_L("Failed! CreateSession cancelled returned incorrect Error: %d"),
			err);
		SetTestStepResult(EFail);
		return TestStepResult();
		}
  	
	// now, create a session with the parameters set in the preamble
	err = CreateSessionL();
	
	// Ensure we succeeded
	if (err != KErrNone)
		{
		INFO_PRINTF2(_L("Failed! Create Session failed! (Error %d)"), err);
		SetTestStepResult(EFail);
		return TestStepResult();
		}
	
	INFO_PRINTF1(_L("Calling TLS session key exchange with cancellation."));
	HBufC8* keyExMessage = NULL;
	
	// Important part of the test: cancellation
	err = ClientKeyExchangeWithCancel(keyExMessage);
	
	if (err != KErrCancel)
		{
		INFO_PRINTF2(_L("Client Key Exchange cancellation returned incorrect Error %d)"), err);
		delete keyExMessage;
		SetTestStepResult(EFail);
		return TestStepResult();
		}
	
	// Makes same request but this time does not cancel it
	err = ClientKeyExchange(keyExMessage);
		
	if (err != KErrNone)
		{
		INFO_PRINTF2(_L("Failed! Key exchange failed! (Error %d)"), err);
		delete keyExMessage;
		SetTestStepResult(EFail);
		return TestStepResult();
		}
	
	INFO_PRINTF1(_L("Deriving premaster secret."));
			
	// derive the premaster secret from the key exchange method	
	CleanupStack::PushL(keyExMessage);
	HBufC8* premaster = DerivePreMasterSecretL(*keyExMessage);
	CleanupStack::PopAndDestroy(keyExMessage);
	
	INFO_PRINTF1(_L("Deriving master secret."));
	
	// compute the master secret from the premaster.
	CleanupStack::PushL(premaster);
	HBufC8* master = ComputeMasterSecretL(*premaster);
	CleanupStack::PopAndDestroy(premaster);
	CleanupStack::PushL(master);
	
	ValidateServerFinishedL(*master);
		
	CleanupStack::PopAndDestroy(master);
			
	return TestStepResult();
		
	}