// Copyright (c) 2001-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:
// Brief description of test harness:
// Tests SSL/TLS functionality in IMPS
// Input files required to run test harness:
// None
// Output files produced by running test harness:
// <DRIVE>:\msglogs\T_IMPSSECURESOCKET.<PLATFORM>.<VARIANT>.LOG.txt
// Description of how to build test harness:
// cd \messaging\msg\imps\
// bldmake bldfiles
// abld test build T_IMPSSECURESOCKET
// Description of how to run test harness:
// The following instructions are the same for all platforms:
// 2. Build the test utilities:
// cd \messaging\msg\testutils\group\
// bldmake bldfiles
// abld build
// WINS running instructions:
// 1. \epoc32\release\wins\<VARIANT>\T_IMPSSECURESOCKET.exe can be used at the command prompt
// or executable can be run from Windows Explorer.
// All other platform running instructions:
// 1. Copy \epoc32\release\<PLATFORM>\<VARIANT>\T_IMPSSECURESOCKET.exe onto the other platform
// 2. Copy \epoc32\release\<PLATFORM>\<VARIANT>\MSVTESTUTILS.DLL into 
// <DRIVE>:\system\libs on the other platform
// 3. Copy \epoc32\release\<PLATFORM>\<VARIANT>\EMAILTESTUTILS.DLL into 
// <DRIVE>:\system\libs on the other platform
// 5. Run T_IMPSSECURESOCKET.exe on the other platform with the CF card
// 
//

#include "EmailTestUtils.h"
#include <miutmsg.h>
#include "impsmtm.h"
#include <securesocket.h>
#include <bacline.h>
#include <cemailaccounts.h>

// For loading the serial comms device drivers 
#if defined (__WINS__)
	#define PDD_NAME		_L("ECDRV")
	#define LDD_NAME		_L("ECOMM")
#else
	#define PDD_NAME		_L("EUART1")
	#define LDD_NAME		_L("ECOMM")
#endif


LOCAL_D CTrapCleanup* theCleanup;
LOCAL_D CEmailTestUtils* testUtils; 
LOCAL_D CActiveScheduler* scheduler;
LOCAL_D RFs theFs;	
LOCAL_D TMsvId	imapService;


_LIT(KImpsTest, "T_IMPSSECURESOCKET - IMPS SecureSockets Functionality Testing");

RTest test(KImpsTest);

_LIT(KTestParams,"-testparams");
enum TCmdLineArgs
	{
	EArgProgramName,
	EArgTestParams,
	EArgServerAddress,
	EArgLogin,
	EArgPassword,
	EArgEnd
	};


LOCAL_C TInt readConsoleLineBuf(const TDesC& aPrompt, TDes& aBuf)
	{
	test.Printf(_L("\n "));
	TBool quit = EFalse;
	_LIT(KDeleteChar, "\b \b");

	test.Printf(aPrompt);
	while (!quit)
		{
		TKeyCode k = test.Getch();
		TChar key = k;
		switch (key)
			{
		case EKeyBackspace:
			if (aBuf.Length() > 0)
				{
				test.Printf(KDeleteChar);
				aBuf = aBuf.Left(aBuf.Length() - 1);
				}
			break;
		case EKeyEnter:
			if (aBuf.Length())
				quit = ETrue;
			else
				test.Printf(_L("\nLogin, password and server address cannot be zero length "));
			break;
		default:
			test.Printf(_L("%c"), key);
			aBuf.Append(key);
			break;
			}
		}
	return(KErrNone);
	}



class CImpsTestHarness : public CActive
	{
public:
	static CImpsTestHarness* NewLC();
	void SetUpImpsServiceL();
	~CImpsTestHarness();
	void StartTestL();
	void EndTestL();
	void RunL();
	void DoCancel();
private:
	CImpsTestHarness();
	void ConstructL();
private:
	TInt				iTestError;
	TInt				iTestNumber;
	CImap4ServerMtm*	imapServerMtm;
	CMsvEntrySelection* iSel;
	TImap4Cmds iCommand;
	};


//

CImpsTestHarness  *CImpsTestHarness::NewLC()
	{
	CImpsTestHarness* self=new (ELeave) CImpsTestHarness();
	CleanupStack::PushL(self);
	self->ConstructL();
	return self;
	}

CImpsTestHarness::~CImpsTestHarness()
	{
	delete iSel;
	}

CImpsTestHarness::CImpsTestHarness()
	: CActive(-1)
	{
	}

void CImpsTestHarness::ConstructL()
	{
	CActiveScheduler::Add(this); // add to active scheduler
	testUtils->CreateAllTestDirectories();
	testUtils->FileSession().SetSessionPath(_L("C:\\"));
	testUtils->CleanMessageFolderL();
	testUtils->GoServerSideL();
	// Get server MTM.
	testUtils->Reset();
	testUtils->CleanMessageFolderL();
	testUtils->GoServerSideL();
	SetUpImpsServiceL();
	testUtils->InstantiateImapServerMtmL();
	imapServerMtm=(CImap4ServerMtm*)testUtils->iImapServerMtm;
	}

void CImpsTestHarness::SetUpImpsServiceL()
	{
	// Handle command line arguments
	CCommandLineArguments* cmdLineArg=CCommandLineArguments::NewLC();
	TInt count = cmdLineArg->Count();
	TBool isCmdLine=FALSE;
	if (count>2)	// Command line arguments?
		{
		if ((!(cmdLineArg->Arg(EArgTestParams).Compare(KTestParams))) && count==EArgEnd)
			isCmdLine=TRUE;
		}	

	imapService=testUtils->CreateImapServiceL();
	testUtils->iServerEntry->SetEntry(imapService);

	CImImap4Settings* settings = new(ELeave) CImImap4Settings();
	CleanupStack::PushL(settings);
	settings->Reset();


	TBuf<100> buf;
	if (isCmdLine)
		{
		TLex lex(cmdLineArg->Arg(EArgServerAddress));
		buf=lex.NextToken();
		test.Printf(_L("Server address: %S\n"),&buf);
		}
	else
		{
		readConsoleLineBuf(_L("Server address: "),buf);
		}
	settings->SetServerAddressL(buf);
	buf.Zero();
	if (isCmdLine)
		{
		TLex lex(cmdLineArg->Arg(EArgLogin));
		buf=lex.NextToken();
		test.Printf(_L("Login: %S\n"),&buf);
		}
	else
		{
		readConsoleLineBuf(_L("login: "),buf);
		}
	TBuf8<64> data;
	data.Copy(buf);
	settings->SetLoginNameL(testUtils->MachineName());
	data.Zero();
	buf.Zero();
	if (isCmdLine)
		{
		TLex lex(cmdLineArg->Arg(EArgPassword));
		buf=lex.NextToken();
		test.Printf(_L("Password: %S\n"),&buf);
		}
	else
		{
		readConsoleLineBuf(_L("password: "),buf);
		}
	data.Copy(buf);
	settings->SetPasswordL(testUtils->MachineName());

	settings->SetFolderPathL(_L8(""));
	settings->SetPort(143);
	settings->SetPathSeparator('/');
	settings->SetDisconnectedUserMode(ETrue);
	settings->SetDeleteEmailsWhenDisconnecting(ETrue);
	settings->SetGetMailOptions(EGetImap4EmailBodyTextAndAttachments);
	settings->SetSecureSockets(ETrue); 
	settings->SetSubscribe(EUpdateNeither);
	settings->SetSynchronise(EUseCombination);
	
	CEmailAccounts* accounts = CEmailAccounts::NewLC();
 	TImapAccount imapAccount;
 	accounts->GetImapAccountL(imapService, imapAccount);						
 	accounts->SaveImapSettingsL(imapAccount, *settings);
	CleanupStack::PopAndDestroy(3, settings); // accounts, settings, cmdLineArg
	}

void CImpsTestHarness::StartTestL()
	{
	testUtils->TestStart(1);
	iSel = new (ELeave) CMsvEntrySelection;
	iSel->AppendL(imapService);
	TBuf8<128> parameter;
	testUtils->WriteComment(_L("Calling KIMAP4MTMConnect ")); 
	iCommand=KIMAP4MTMConnect;
	imapServerMtm->StartCommandL(*iSel, iCommand, parameter, iStatus);
	SetActive();
	}

void CImpsTestHarness::EndTestL()
	{
	testUtils->TestFinish(1, iTestError);
	}


void  CImpsTestHarness::DoCancel()
	{
	imapServerMtm->Cancel();
	}

void CImpsTestHarness::RunL()
	{
	TImap4CompoundProgress temp;	
	TPckgC<TImap4CompoundProgress> paramPack(temp);

	const TDesC8& progBuf = imapServerMtm->Progress();	
	paramPack.Set(progBuf);
	TImap4CompoundProgress progress=paramPack();	
	TInt err=progress.iGenericProgress.iErrorCode;

	if (iStatus.Int()!=KErrNone || iCommand==KIMAP4MTMDisconnect || err)
		{
		testUtils->TestFinish(1,err);
		CActiveScheduler::Stop();
		return;
		}
	if (iCommand==KIMAP4MTMConnect)
		{
		testUtils->WriteComment(_L("Calling KIMAP4MTMInboxNewSync ")); 
		iCommand=KIMAP4MTMInboxNewSync;
		TBuf8<128> parameter;
		imapServerMtm->StartCommandL(*iSel, iCommand, parameter, iStatus);
		SetActive();
		return;
		}

	if (iCommand==KIMAP4MTMInboxNewSync)
		{
		iCommand=KIMAP4MTMDisconnect;
		TBuf8<128> parameter;
		imapServerMtm->StartCommandL(*iSel, iCommand, parameter, iStatus);
		SetActive();
		return;

		}
	}

//

LOCAL_C void Init()
	{
	scheduler = new (ELeave) CActiveScheduler;
	CActiveScheduler::Install( scheduler );

	User::LeaveIfError(theFs.Connect());
	theFs.SetSessionPath(_L("C:\\"));
	}
	
LOCAL_C void Closedown()
	{
	theFs.Close();
	CleanupStack::PopAndDestroy(testUtils);
	delete scheduler;
	CSecureSocketLibraryLoader::Unload();
	}


LOCAL_C void doMainL()
	{
	Init();


	// Load the serial comms device drivers.  If this is not done,
	// connecting via NT-RAS returns KErrNotFound (-1).
	TInt driverErr;
	driverErr=User::LoadPhysicalDevice(PDD_NAME);
	if (driverErr!=KErrNone && driverErr!=KErrAlreadyExists)
		User::Leave(driverErr);
	driverErr=User::LoadLogicalDevice(LDD_NAME);
	if (driverErr!=KErrNone && driverErr!=KErrAlreadyExists)
		User::Leave(driverErr);


	theFs.SetSessionPath(_L("c:\\"));
	theFs.MkDir(_L("c:\\logs\\"));
	theFs.MkDir(_L("c:\\logs\\email\\"));
	testUtils = CEmailTestUtils::NewLC(test);
	testUtils->WriteComment(_L("T_IMPSECURESOCKETS  - Testing IMPS secure sockets"));
	testUtils->WriteComment(_L("Before you run this test harness, please create commdb dat file ")); 
	testUtils->WriteComment(_L("with server details which supports TLS."));
	CImpsTestHarness* impsTestHarness = CImpsTestHarness::NewLC();
	impsTestHarness->StartTestL();
 	CActiveScheduler::Start();
	testUtils->TestHarnessCompleted();
	CleanupStack::PopAndDestroy(impsTestHarness); 
	Closedown();
	}


GLDEF_C TInt E32Main()
	{
	test.Title();
	test.Start(_L("Testing IMPS Secure Socket connection"));
	__UHEAP_MARK;

	theCleanup=CTrapCleanup::New();
	test (theCleanup!=NULL);
	TRAPD(ret,doMainL());		
	test (ret==KErrNone);
	delete theCleanup;

	__UHEAP_MARKEND;
	test.End();
	return 0;
	}