// Copyright (c) 2003-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:

//

#include "TlsProvTestStep.h"

#include <test/testexecutelog.h>

#include <hash.h>

#define KTLSCachingTimeout  200 

TVerdict CTlsProvTestActive::doTest4_0L( CTlsProvStep* aStep )

	{

	CTlsTestRandom rnd;

	SetThreadRandomL( &rnd );

	CTLSProvider* tlsProvider = CTLSProvider::ConnectL();

	RArray<TTLSCipherSuite> userCipherSuiteList;

	iStatus = KRequestPending;

	tlsProvider->CipherSuitesL( userCipherSuiteList ,iStatus);	

	SetActive();

	CActiveScheduler::Start();

	CTLSSession* sessionObj = NULL;

	CTlsCryptoAttributes* tlsCryptoAttributes = tlsProvider->Attributes();

	HBufC8* encServerCert = NULL;

	TInt res = StandardTestInitL(aStep, tlsCryptoAttributes, encServerCert);

	if( res  < 0 )

		{

		iLogInfo.Copy( _L("	4.0:  problems with test data files") );

		return EFail;

		}

	CX509Certificate* serverCert = NULL;

	iStatus = KRequestPending;

	tlsProvider->VerifyServerCertificate( encServerCert->Des(), serverCert, iStatus);

	SetActive();

	CActiveScheduler::Start();

	serverCert = CX509Certificate::NewL(encServerCert->Des()); //- remove when VerifyServerCertificate ready

	HBufC8* clientKeyExch = NULL;

	iStatus = KRequestPending;

	tlsProvider->CreateL( sessionObj,

						iStatus);	

	SetActive();

	CActiveScheduler::Start();	

	// check that ClientKeyExch is not NULL

	if ( (iStatus.Int() != KErrNone ) || (NULL == sessionObj) )

		{

		iLogInfo.Copy( _L("	4.0:  CTLSProvider::Create failed") );

		return EInconclusive;

		}

	iStatus = KRequestPending;

	sessionObj->ClientKeyExchange(clientKeyExch, iStatus);

	SetActive();

	CActiveScheduler::Start();

	// check that ClientKeyExch is not NULL

	if ( (NULL == clientKeyExch) || ( 0 == clientKeyExch->Length()) ) 

		{

		iLogInfo.Copy( _L("	4.0:  CTLSSession::ClientKeyExchange failed") );

		return EInconclusive;

		}

	if( EFalse == CacheSessionL( aStep, sessionObj) )

		return EInconclusive;

	iStatus = KRequestPending;

	CX509Certificate* cert;

	sessionObj->ServerCertificate( cert, iStatus );

	SetActive();

	CActiveScheduler::Start();

	if ( KErrNone != iStatus.Int() )

		{

		iLogInfo.Copy( _L("	4.0:  CTLSSession::ServerCertificte - returned with error") );

		return EFail;

		}

	//compare retrieved cert with original one:

	if ( cert->IsEqualL( *serverCert ) == EFalse  ) 

		{

		iLogInfo.Copy( _L("	4.0:  retrieved certificate doesn't match") );

		return EFail;

		}

	iLogInfo.Copy( _L("	4.0:  OK") );

	delete tlsProvider;	

	delete sessionObj;

	return EPass;

	} 	

TVerdict CTlsProvTestActive::doTest4_1L( CTlsProvStep* aStep )

	{

	CTlsTestRandom rnd;

	SetThreadRandomL( &rnd );

	RArray<TTLSCipherSuite> userCipherSuiteList;

	CTLSProvider* tlsProvider = CTLSProvider::ConnectL();

	iStatus = KRequestPending;

	tlsProvider->CipherSuitesL( userCipherSuiteList ,iStatus);	

	SetActive();

	CActiveScheduler::Start();

	CTLSSession* sessionObj;

	CTlsCryptoAttributes* tlsCryptoAttributes = tlsProvider->Attributes();

	HBufC8* encServerCert = NULL;

	TInt res = StandardTestInitL(aStep, tlsCryptoAttributes, encServerCert);

	if( res  < 0 )

		{

		iLogInfo.Copy( _L("	4.1:  problems with test data files") );

		return EFail;

		}

	tlsCryptoAttributes->iSessionNameAndID.iServerName.iAddress.Copy( KServer2 );

	tlsCryptoAttributes->iNegotiatedProtocol.iMajor = 3;

	tlsCryptoAttributes->iNegotiatedProtocol.iMinor = 0; 

	tlsCryptoAttributes->iProposedProtocol.iMajor = 3;

	tlsCryptoAttributes->iProposedProtocol.iMinor = 0; 

	CX509Certificate* serverCert = NULL;

	iStatus = KRequestPending;

	tlsProvider->VerifyServerCertificate( encServerCert->Des(), serverCert, iStatus);

	SetActive();

	CActiveScheduler::Start();

	tlsCryptoAttributes->iClientAuthenticate = EFalse; 

	tlsCryptoAttributes->iDialogNonAttendedMode = ETrue;

	HBufC8* clientKeyExch = NULL;

	iStatus = KRequestPending;

	tlsProvider->CreateL( sessionObj,

						iStatus);	

	SetActive();

	CActiveScheduler::Start();	

	// check that ClientKeyExch is not NULL

	if ( (iStatus.Int() != KErrNone ) || (NULL == sessionObj) )

		{

		iLogInfo.Copy( _L("	4.1:  CTLSProvider::Create failed") );

		return EInconclusive;

		}

	iStatus = KRequestPending;

	sessionObj->ClientKeyExchange(clientKeyExch, iStatus);

	SetActive();

	CActiveScheduler::Start();

	// check that ClientKeyExch is not NULL

	if ( (NULL == clientKeyExch) || ( 0 == clientKeyExch->Length()) ) 

		{

		iLogInfo.Copy( _L("	4.1:  CTLSSession::ClientKeyExchange failed") );

		return EInconclusive;

		}

	if( EFalse == CacheSessionL( aStep, sessionObj) )

		return EInconclusive;

	TTLSSessionId sessionId;

	iStatus = KRequestPending;

	tlsProvider->GetSessionL(

		tlsCryptoAttributes->iSessionNameAndID.iServerName,

		sessionId,

		iStatus );

	SetActive();

	CActiveScheduler::Start();

	if ( KErrNone != iStatus.Int() )

		{

		iLogInfo.Copy( _L("	4.1:  CTLSProvider::GetSession - returned with error") );

		return EFail;

		}

	if (  0 != sessionId.Compare( 

								tlsCryptoAttributes->iSessionNameAndID.iSessionId ) ) 

		{

		iLogInfo.Copy( _L("	4.1:  CTLSProvider::GetSession - BAD session id") );

		return EFail;

		}

	iLogInfo.Copy( _L("	4.1:  OK") );

	return EPass;

	} 

TVerdict CTlsProvTestActive::doTest4_2L( CTlsProvStep* aStep )

	{

	CTlsTestRandom rnd;

	SetThreadRandomL( &rnd );

	RArray<TTLSCipherSuite> userCipherSuiteList;

	CTLSProvider* tlsProvider = CTLSProvider::ConnectL();

	CTLSProvider* tlsProvider2 = CTLSProvider::ConnectL();

	iStatus = KRequestPending;

	tlsProvider->CipherSuitesL( userCipherSuiteList ,iStatus);	

	SetActive();

	CActiveScheduler::Start();

	CTLSSession* sessionObj;

	CTLSSession* sessionObj2;

	CTlsCryptoAttributes* tlsCryptoAttributes = tlsProvider->Attributes();

	HBufC8* encServerCert = NULL;

	TInt res = StandardTestInitL(aStep, tlsCryptoAttributes, encServerCert);

	if( res  < 0 )

		{

		iLogInfo.Copy( _L("	4.2:  problems with test data files (1)") );

		return EFail;

		}

	CX509Certificate* serverCert = NULL;

	iStatus = KRequestPending;

	tlsProvider->VerifyServerCertificate( encServerCert->Des(), serverCert, iStatus);

	SetActive();

	CActiveScheduler::Start();

	HBufC8* clientKeyExch = NULL;

	iStatus = KRequestPending;

	tlsProvider->CreateL( sessionObj,

						iStatus);	

	SetActive();

	CActiveScheduler::Start();	

	// check that ClientKeyExch is not NULL

	if ( (iStatus.Int() != KErrNone ) || (NULL == sessionObj) )

		{

		iLogInfo.Copy( _L("	4.2:  CTLSProvider::Create failed") );

		return EInconclusive;

		}

	iStatus = KRequestPending;

	sessionObj->ClientKeyExchange(clientKeyExch, iStatus);

	SetActive();

	CActiveScheduler::Start();

	// check that ClientKeyExch is not NULL

	if ( (NULL == clientKeyExch) || ( 0 == clientKeyExch->Length()) ) 

		{

		iLogInfo.Copy( _L("	4.2:  CTLSSession::ClientKeyExchange failed") );

		return EInconclusive;

		}

	if( EFalse == CacheSessionL( aStep, sessionObj) )

		return EInconclusive;

	//

	// entry 2

	delete clientKeyExch;	

	clientKeyExch = NULL;

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SSLServerRnd"),aStep->iServerRnd);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SSLClientRnd"),aStep->iClientRnd);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SSLKeyParams1"),aStep->iKeyParam1);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SSLKeyParams2"),aStep->iKeyParam2);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SSLKeyParams3"),aStep->iKeyParam3);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SSLServerCert"),aStep->iServerCertChain);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SSLHandshakeMsgsServer"),aStep->iHandshakeMsgsServer);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SSLSrvFinishedCheckOutput"),aStep->iSrvFinishedCheckOutput);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SSLPremaster"),aStep->iPremaster);

	RFs fs;

	if ( KErrNone != fs.Connect() )

		{

		iLogInfo.Copy( _L("	4.2:  problems with file system (2)") );

		return EFail;

		}

	CFileMan* fMan = CFileMan::NewL( fs );

	TDriveUnit sysDrive (RFs::GetSystemDrive());

	TBuf<128> fileName (sysDrive.Name());

	fileName.Append(_L("\\Premaster.bin"));

	fMan->Delete(fileName);

	fMan->Copy( aStep->iPremaster,	fileName, CFileMan::EOverWrite );

	fs.Close();	

	RArray<TTLSCipherSuite> userCipherSuiteList2;

	iStatus = KRequestPending;

	tlsProvider2->CipherSuitesL( userCipherSuiteList2 ,iStatus);	

	SetActive();

	CActiveScheduler::Start();

	CTlsCryptoAttributes* tlsCryptoAttributes2 = tlsProvider2->Attributes();

	tlsCryptoAttributes2->iSessionNameAndID.iServerName.iAddress.Copy( KServer1 );

	tlsCryptoAttributes2->iSessionNameAndID.iServerName.iPort = 10;

	tlsCryptoAttributes2->iSessionNameAndID.iSessionId.Copy( KSessionId2 );

	tlsCryptoAttributes2->iCompressionMethod = ENullCompression;

	tlsCryptoAttributes2->iCurrentCipherSuite.iHiByte = 0;

	tlsCryptoAttributes2->iCurrentCipherSuite.iLoByte = 3;

	tlsCryptoAttributes2->iNegotiatedProtocol.iMajor = 3;

	tlsCryptoAttributes2->iNegotiatedProtocol.iMinor = 0; 

	tlsCryptoAttributes2->iProposedProtocol.iMajor = 3;

	tlsCryptoAttributes2->iProposedProtocol.iMinor = 0; 

	tlsCryptoAttributes2->iPublicKeyParams->iKeyType = ERsa;

	tlsCryptoAttributes2->iClientAuthenticate = EFalse; 

	tlsCryptoAttributes2->iDialogNonAttendedMode = ETrue;

	delete encServerCert;

	encServerCert = NULL;

	res = aStep->ReadDataForCreateL(tlsCryptoAttributes2, encServerCert);

	if( res  < 0 )

		{

		iLogInfo.Copy( _L("	4.2:  problems with test data files (2)") );

		return EFail;

		}

	delete serverCert;

	serverCert = NULL;

	iStatus = KRequestPending;

	tlsProvider2->VerifyServerCertificate( encServerCert->Des(), serverCert, iStatus);

	SetActive();

	CActiveScheduler::Start();

	iStatus = KRequestPending;

	tlsProvider2->CreateL( sessionObj2,

						iStatus);	

	SetActive();

	CActiveScheduler::Start();	

	// check that ClientKeyExch is not NULL

	if ( (iStatus.Int() != KErrNone ) || (NULL == sessionObj2) ) 

		{

		iLogInfo.Copy( _L("	4.2:  CTLSProvider::Create failed") );

		return EInconclusive;

		}

	iStatus = KRequestPending;

	sessionObj2->ClientKeyExchange(clientKeyExch, iStatus);

	SetActive();

	CActiveScheduler::Start();

	// check that ClientKeyExch is not NULL

	if ( (NULL == clientKeyExch) || ( 0 == clientKeyExch->Length()) ) 

		{

		iLogInfo.Copy( _L("	4.2:  CTLSSession::ClientKeyExchange failed (4)") );

		return EInconclusive;

		}

	delete clientKeyExch;	

	clientKeyExch = NULL;

	if( EFalse == CacheSessionL( aStep, sessionObj2) )

		return EInconclusive;

	// entry 2 inserted

	//

	//

	TTLSSessionId sessionId;

	CTLSProvider* tlsProvider3 = CTLSProvider::ConnectL();

	RArray<TTLSCipherSuite> userCipherSuiteList3;

	iStatus = KRequestPending;

	tlsProvider3->CipherSuitesL( userCipherSuiteList3 ,iStatus);	

	SetActive();

	CActiveScheduler::Start();

	CTlsCryptoAttributes* tlsCryptoAttributes3 = tlsProvider3->Attributes();

	tlsCryptoAttributes3->iCurrentCipherSuite.iHiByte = 0;

	tlsCryptoAttributes3->iCurrentCipherSuite.iLoByte = 3;

	tlsCryptoAttributes3->iNegotiatedProtocol.iMajor = 3;

	tlsCryptoAttributes3->iNegotiatedProtocol.iMinor = 0; 

	tlsCryptoAttributes3->iProposedProtocol.iMajor = 3;

	tlsCryptoAttributes3->iProposedProtocol.iMinor = 0; 

	TTLSSessionNameAndID sessionNameAndID;

	sessionNameAndID.iServerName.iAddress.Copy( KServer1 );

	sessionNameAndID.iServerName.iPort = 10;

	iStatus = KRequestPending;

	tlsProvider3->GetSessionL(

		sessionNameAndID.iServerName,

		sessionId,

		iStatus );

	SetActive();

	CActiveScheduler::Start();

	if ( KErrNone != iStatus.Int() )

		{

		iLogInfo.Copy( _L("	4.2:  CTLSProvider::GetSession - returned with error") );

		return EFail;

		}

	if (  0 != sessionId.Compare( tlsCryptoAttributes2->iSessionNameAndID.iSessionId ) ) 

		{

		iLogInfo.Copy( _L("	4.2:  CTLSProvider::GetSession - BAD session id") );

		return EFail;

		}

	return EPass;

	} 	

TVerdict CTlsProvTestActive::doTest4_3L( CTlsProvStep* aStep )

	{

	CTlsTestRandom rnd;

	SetThreadRandomL( &rnd );

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("ServerRnd"),aStep->iServerRnd);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("ClientRnd"),aStep->iClientRnd);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("KeyParams1"),aStep->iKeyParam1);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("KeyParams2"),aStep->iKeyParam2);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("KeyParams3"),aStep->iKeyParam3);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("ServerCert"),aStep->iServerCertChain);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("HandshakeMsgsServer"),aStep->iHandshakeMsgsServer);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("SrvFinishedCheckOutput"),aStep->iSrvFinishedCheckOutput);

	aStep->GetStringFromConfig(aStep->ConfigSection(),_L("Premaster"),aStep->iPremaster);

	RArray<TTLSCipherSuite> userCipherSuiteList;

	CTLSProvider* tlsProvider = CTLSProvider::ConnectL();

	iStatus = KRequestPending;

	tlsProvider->CipherSuitesL( userCipherSuiteList ,iStatus);	

	SetActive();

	CActiveScheduler::Start();

	TTLSSessionNameAndID sessionNameAndID;

	sessionNameAndID.iServerName.iAddress.Copy( KServer3 );

	sessionNameAndID.iServerName.iPort = 20;

	TTLSSessionId sessionId;

	iStatus = KRequestPending;

	tlsProvider->GetSessionL(

		sessionNameAndID.iServerName,

		sessionId,

		iStatus );

	SetActive();

	CActiveScheduler::Start();

	if ( 0 != sessionId.Length()  )