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

// System includes
#include <f32file.h>

// User includes
#include "WspTestTransactions.h"
#include "csrvaddrval.h"

// Proxy information properties
_LIT(KItemWspProxyAddress,			"proxy_address");
_LIT(KItemWspBearer,				"wsp_bearer");
_LIT(KItemWspLocalPort,				"local_port");
_LIT(KItemWspRemotePort,			"remote_port");
_LIT(KItemWspServiceNumber,			"service_number");
_LIT(KItemWspSessionType,			"session_type");
_LIT(KItemWspSecurity,				"security");

// Capability properties
_LIT(KItemWspClientMsgSize,			"client_message_size");
_LIT(KItemWspServerMsgSize,			"server_message_size");
_LIT(KItemWspUseAcknowledgements,	"use_acknowledgements");
_LIT(KItemWspSupportSuspendResume,	"support_suspend_resume");
_LIT(KItemWspMaxOutstandingRequests,"max_outstanding_requests");
_LIT(KItemExpectedError,			"test_validation_param_expectederror");

// Security policy properties
_LIT(KSecurityPolicy,				"securitypolicy");
_LIT(KSecurityPolicyFileName,		"securitypolicy.ini");

_LIT(KValidateUntrustedServerCert,	"ValidateUntrustedServerCert");
_LIT(KWTLSCipherSuites,				"WTLSCipherSuites");
_LIT(KWTLSKeyExchangeSuites,		"WTLSKeyExchangeSuites");
_LIT(KTLSCipherSuites,				"TLSCipherSuites");
_LIT(KPolicy,						"Policy");


CWspTestTransactions* CWspTestTransactions::NewL(CScriptFile& aIniFile, CScriptFile* aIniSettingsFile, const TInt aSectionName)
	{
	CWspTestTransactions* self = new(ELeave) CWspTestTransactions(aIniSettingsFile);
	CleanupStack::PushL(self);
	self->ConstructL(aIniFile, aIniSettingsFile, aSectionName);
	CleanupStack::Pop(self);
	return self;
	}


CWspTestTransactions::CWspTestTransactions(CScriptFile* aIniSettingsFile) :
	iIniSettingsFile(aIniSettingsFile)
	{
	}


void CWspTestTransactions::ConstructL(CScriptFile& aIniFile, CScriptFile* aIniSettingsFile, const TInt aSectionName)
	{
	CHttpTestTransactions::ConstructL(aIniFile, aIniSettingsFile, aSectionName);
	}


CWspTestTransactions::~CWspTestTransactions()
	{
	}

void CWspTestTransactions::DoRunL()
	{
	OpenTestSessionL(); // set up session and transaction here
	if ((iSessionEvent.iStatus) == THTTPSessionEvent::EDisconnected || (iSessionEvent.iStatus) == THTTPSessionEvent::EConnectionTimedOut)	
		{
		DoCloseTestSession(); // close any strings, transactiona and session

		// Check if the expected status is returned
		if( (iSessionEvent.iStatus == THTTPSessionEvent::EConnectionTimedOut) && (iExpectedStatusCode == 504) )
			{
			// No fail...
			iTestFail = 0;
			}
		else if( (iSessionEvent.iStatus == THTTPSessionEvent::EDisconnected) && (iExpectedError == THTTPSessionEvent::EAuthenticationFailure) && iAuthenticationFailure )
			{
			// No fail...
			iTestFail = 0;
			}
		else
			{
			// Fail!
			iTestFail = 1;
			}

		iEngine->Utils().LogIt(_L("Warning: Connection to proxy closed\n"));
		}
	else
		{
		CActiveScheduler::Start();
		CloseTestSession(); // disconnect from gateway and then close any 
							// strings, transactions and sessions
		}
	if (iTestFail==1)
		{
		User::Leave(-1);
		}
	}


void CWspTestTransactions::OpenTestSessionL()
	{
	// Iterate through the TestCases.ini file, beginning at the first element 
	// i.e. index 0, and write the security policy properties to the 
	// securitypolicy.ini file.
	SetWspSecurityPolicyPropertiesL(0);

	iSession.OpenL(_L8("WSP/WSP"));

	iMyStrP = iSession.StringPool();

	// Set the session callback
	iSession.SetSessionEventCallback(this);

	SetWspSessionPropertiesL();

	ConnectToGatewayL();
	}


void CWspTestTransactions::CloseTestSession()
	{
	LogCertificate();
	// Disconnect from the gateway first
	TRAP_IGNORE( DisconnectFromGatewayL() );
	DoCloseTestSession();
	}

void CWspTestTransactions::LogCertificate()
	{
	if(iLogCert)
		{
		TRAPD(error, LogCertL());
		if(error != KErrNone)
			iEngine->Utils().LogIt(_L("Unable to log server certificate.  Error code=%d\n"), error);
		}
	}

void CWspTestTransactions::LogCertL()
	{
	TCertInfo certInfo;
	User::LeaveIfError(iSession.ServerCert(certInfo));

	LogCertificateL(certInfo);
	}

void CWspTestTransactions::DoCloseTestSession()
	{
	CHttpTestTransactions::CloseTestSession();
	iTransaction.Close();
	iEngine->Utils().LogIt(_L("Transaction closed\n"));
	iSession.Close();
	iEngine->Utils().LogIt(_L("Session closed"));
	}


void CWspTestTransactions::ConnectToGatewayL()
	{
	iSession.ConnectL();
	CActiveScheduler::Start();
	}


void CWspTestTransactions::DisconnectFromGatewayL()
	{
	iSession.DisconnectL();
	CActiveScheduler::Start();
	}


void CWspTestTransactions::CreateSingleTransactionL()
	{
	// Iterate through the TestCases.ini file, beginning at the first element
	// i.e. index 0, and set up the transaction properties
	iTransaction = CreateTransactionL(0);
	iTransaction.SubmitL();
	}


void CWspTestTransactions::MHFSessionRunL(const THTTPSessionEvent& aEvent)
	{
	iSessionEvent = aEvent;
	switch(aEvent.iStatus)
		{
	case THTTPSessionEvent::EConnectedOK:
		{
		iEngine->Utils().LogIt(_L("WSP Session connected. No reduced capabilities\n"));
		CreateSingleTransactionL();
		CActiveScheduler::Stop();
		} break;
	case THTTPSessionEvent::EConnectedWithReducedCapabilities:
		{
		iEngine->Utils().LogIt(_L("WSP Session connected with reduced capabilities\n"));
		CreateSingleTransactionL();
		CActiveScheduler::Stop();
		} break;
	case THTTPSessionEvent::EConnectionTimedOut:
		{
		// Need to create the transaction even though its timed out to read
		// the ini file for the time out tests in servertestcases.ini
		iTransaction = CreateTransactionL(0);
		iEngine->Utils().LogIt(_L("Proxy connection timed out\n"));
		CActiveScheduler::Stop();
		} break;
	case THTTPSessionEvent::EDisconnected:
		{
		iEngine->Utils().LogIt(_L("WSP Session disconnected\n"));
		CActiveScheduler::Stop();
		} break;
	case THTTPSessionEvent::ERedirected:
		{
		iEngine->Utils().LogIt(_L("Redirected\n"));
		} break;
	case THTTPSessionEvent::EAuthenticatedOK:
		{
		iEngine->Utils().LogIt(_L("Authenticated OK\n"));
		} break;
	case THTTPSessionEvent::EAuthenticationFailure:
		{
		iEngine->Utils().LogIt(_L("Warning: Authentication Failure\n"));
		iAuthenticationFailure = ETrue;
		} break;
	case THTTPSessionEvent::ENotConnected:
		{
		iEngine->Utils().LogIt(_L("Warning: Not Connected\n"));
		} break;
	case THTTPSessionEvent::EExceptionInfo:
		{
		iEngine->Utils().LogIt(_L("Warning: Exception Info\n"));
		} break;
	case THTTPSessionEvent::EAlreadyConnecting:
		{
		iEngine->Utils().LogIt(_L("Warning: Already Connecting\n"));
		} break;
	case THTTPSessionEvent::EAlreadyConnected:
		{
		iEngine->Utils().LogIt(_L("Warning: Already Connected\n"));
		} break;
	case THTTPSessionEvent::EAlreadyDisconnecting:
		{
		iEngine->Utils().LogIt(_L("Warning: Already Disconnecting\n"));
		} break;
	case THTTPSessionEvent::EAlreadyDisconnected:
		{
		iEngine->Utils().LogIt(_L("Warning: Already Disconnected\n"));
		} break;
	default:
		CActiveScheduler::Stop();
		break;
		}
	}


TInt CWspTestTransactions::MHFSessionRunError(TInt aError, const THTTPSessionEvent& /*aEvent*/)
	{
	return aError;
	}


void CWspTestTransactions::SetWspSessionPropertiesL()
	{
	// Iterate through the TestCases.ini file, beginning at the first element
	// i.e. index 0, and set up the WSP proxy properties and the
	// WSP capability properties
	SetWspProxyPropertiesL(0);
	SetWspCapabilityPropertiesL(0);
	}


void CWspTestTransactions::SetWspProxyPropertiesL(TInt aStartItem)
	{
	RHTTPConnectionInfo connInfo = iSession.ConnectionInfo();
	TInt endItem = FindEndOfTransaction(aStartItem);	
	for (TInt ii = aStartItem; ii < endItem; ++ii)
		{
		TPtrC itemPtr(iIniFile->Section(iSectionNumber).Item(ii).Item());
		HBufC8* itemBuff8 = HBufC8::NewLC(itemPtr.Length());
		TPtr8  itemPtr8 = itemBuff8->Des();
		itemPtr8.Copy(itemPtr);
		RStringF item = iMyStrP.OpenFStringL(itemPtr8);
		CleanupClosePushL(item);
		

		TPtrC valuePtr(iIniFile->Section(iSectionNumber).Item(ii).Value());

		// Retrieve the actual value from the settings.ini file for
		// KItemWspProxyAddress ("proxy_address").  Other values like
		// uri value are retrieved in trans.cpp file.
		
		// Replace the host name in the URI
		HBufC16* valueBuff16 = TSrvAddrVal::ReplaceHostNameL(valuePtr, iIniSettingsFile);
		CleanupStack::PushL(valueBuff16);
		TPtr16 valuePtr16 = valueBuff16->Des();

		// Convert 16 bit to 8 bit.
		HBufC8* valueBuff8 = HBufC8::NewLC(valuePtr16.Length());
		TPtr8 valuePtr8 = valueBuff8->Des();
		valuePtr8.Copy(valuePtr16);

		RStringF value = iMyStrP.OpenFStringL(valuePtr8);
		CleanupClosePushL(value);

		if (itemPtr.CompareF(KItemWspProxyAddress)==0)
			{
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspProxyAddress, RHTTPSession::GetTable()), value);
			}
		if (itemPtr.Compare(KItemWspBearer)==0)
			{
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspBearer, RHTTPSession::GetTable()), value);
			}
		if (itemPtr.Compare(KItemWspLocalPort)==0)
			{
			TLex lex(valuePtr);
			TInt localPort = 0;
			lex.Val(localPort);
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspLocalPort, RHTTPSession::GetTable()), localPort);
			}
		if (itemPtr.Compare(KItemWspRemotePort)==0)
			{
			TLex lex(valuePtr);
			TInt remotePort = 0;
			lex.Val(remotePort);
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspRemotePort, RHTTPSession::GetTable()), remotePort);
			}
		if (itemPtr.Compare(KItemWspServiceNumber)==0)
			{
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspServiceNumber, RHTTPSession::GetTable()), value);
			}
		if (itemPtr.Compare(KItemWspSessionType)==0)
			{
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspSessionType, RHTTPSession::GetTable()), value);
			}
		if (itemPtr.Compare(KItemWspSecurity)==0)
			{
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspSecurity, RHTTPSession::GetTable()), value);
			}

		CleanupStack::PopAndDestroy(5, itemBuff8); // item, valueBuff8, value, valueBuff16
		}
	}

void CWspTestTransactions::SetWspCapabilityPropertiesL(TInt aStartItem)
	{
	RHTTPConnectionInfo connInfo = iSession.ConnectionInfo();
	TInt endItem = FindEndOfTransaction(aStartItem);
	for (TInt ii = aStartItem; ii < endItem; ++ii)
		{
		TPtrC itemPtr(iIniFile->Section(iSectionNumber).Item(ii).Item());
		HBufC8* itemBuff8 = HBufC8::NewLC(itemPtr.Length());
		TPtr8  itemPtr8= itemBuff8->Des();
		itemPtr8.Copy(itemPtr);
		RStringF item = iMyStrP.OpenFStringL(itemPtr8);
		CleanupClosePushL(item);
		

		TPtrC valuePtr(iIniFile->Section(iSectionNumber).Item(ii).Value());
		HBufC8* valueBuff8 = HBufC8::NewLC(valuePtr.Length());
		TPtr8 valuePtr8= valueBuff8->Des();
		valuePtr8.Copy(valuePtr);
		RStringF value = iMyStrP.OpenFStringL(valuePtr8);
		CleanupClosePushL(value);

		if (itemPtr.CompareF(KItemWspClientMsgSize)==0)
			{
			TLex lex(valuePtr);
			TInt cliMsgSize = 0;
			lex.Val(cliMsgSize);
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspCapClientMessageSize, RHTTPSession::GetTable()), cliMsgSize);
			}
		if (itemPtr.CompareF(KItemWspServerMsgSize)==0)
			{
			TLex lex(valuePtr);
			TInt srvMsgSize = 0;
			lex.Val(srvMsgSize);
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspCapServerMessageSize, RHTTPSession::GetTable()), srvMsgSize);
			}
		if (itemPtr.CompareF(KItemWspUseAcknowledgements)==0)
			{
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspCapUseAcknowledgements, RHTTPSession::GetTable()), value);
			}
		if (itemPtr.CompareF(KItemWspSupportSuspendResume)==0)
			{
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspCapSupportSuspendResume, RHTTPSession::GetTable()), value);
			}
		if (itemPtr.CompareF(KItemWspMaxOutstandingRequests)==0)
			{
			TLex lex(valuePtr);
			TInt maxOutReqs = 0;
			lex.Val(maxOutReqs);
			connInfo.SetPropertyL(iMyStrP.StringF(HTTP::EWspCapMaxOutstandingRequests, RHTTPSession::GetTable()), maxOutReqs);
			}
		if (itemPtr.CompareF(KItemExpectedError)==0)
			{
			//get the 'test_validation_param_expectederror' from the ini 
			//file...
			TPtrC expectedParamPtr = iIniFile->Section(iSectionNumber).ItemL(KItemExpectedError).Value();
			TLex paramLex(expectedParamPtr);
			TInt param;
			paramLex.Val(param);
			// ...and put it  into iExpectedError
			iExpectedError = param;
			}

		CleanupStack::PopAndDestroy(4, itemBuff8); // itemBuff8, item, 
												   // valueBuff8, value
		}
	}

void CWspTestTransactions::SetWspSecurityPolicyPropertiesL(TInt aStartItem)
	{
	// Create the security policy plugin ini file
	RFs fs;
	CleanupClosePushL(fs);
	User::LeaveIfError(fs.Connect());

	TParse fileName;
	User::LeaveIfError(iEngine->Utils().ResolveFile(KSecurityPolicy, KSecurityPolicyFileName, fileName));

	RFile file;
	CleanupClosePushL(file);
	User::LeaveIfError(file.Replace(fs, fileName.FullName(), 
		EFileWrite|EFileShareExclusive)); 

	// Write the section name
	file.Write(_L8("\r\n[Security Policy]\r\n"));

	TInt endItem = FindEndOfTransaction(aStartItem);
	for (TInt ii = aStartItem; ii < endItem; ++ii)
		{
		TPtrC itemPtr(iIniFile->Section(iSectionNumber).Item(ii).Item());
		HBufC8* itemBuff8 = HBufC8::NewLC(itemPtr.Length());
		TPtr8  itemPtr8 = itemBuff8->Des();
		itemPtr8.Copy(itemPtr);
		
		TPtrC valuePtr(iIniFile->Section(iSectionNumber).Item(ii).Value());
		HBufC8* valueBuff8 = HBufC8::NewLC(valuePtr.Length());
		TPtr8 valuePtr8 = valueBuff8->Des();
		valuePtr8.Copy(valuePtr);

		if( itemPtr.CompareF(KValidateUntrustedServerCert) == 0 ||
			itemPtr.CompareF(KWTLSCipherSuites) == 0||
			itemPtr.CompareF(KWTLSKeyExchangeSuites) == 0 ||
			itemPtr.CompareF(KTLSCipherSuites) == 0||
			itemPtr.Find(KPolicy) == 0) // matches 'Policy1', 'Policy2', etc.
			{
			// Write out entry to the securitypolicy.ini file
			file.Write(itemPtr8);
			file.Write(_L8("= "));
			file.Write(valuePtr8);
			file.Write(_L8("\r\n"));
			}

		CleanupStack::PopAndDestroy(2, itemBuff8); // itemBuff8, valueBuff8
		}
	// Write the end of script marked in security ini
	file.Write(_L8("\r\nendscript\r\n"));

	// Cleanup
	CleanupStack::PopAndDestroy(2, &fs); // fs, file
	}