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

//

#include <uri8.h>

#include <e32base.h>

#include <http.h>

#include <chttpformencoder.h>

#include <ssl.h>

#include <signed.h>

#include "httpexampleclient.h"

#include "httpexampleutils.h"

// format for output of data/time values

_LIT(KDateFormat,"%D%M%Y%/0%1%/1%2%/2%3%/3 %:0%H%:1%T%:2%S.%C%:3");

// Format for hook file name and location for hooks

_LIT(KHookFormat, "Z:\\httptest\\%S.esk");

_LIT(KHookDir, "C:\\System\\Data\\");

_LIT(KEnterProx, "Enter Proxy authority");

_LIT(KEnterHook, "Enter Hook name (preface with '-' to delete)");

_LIT(KEnterSessId, "Enter Session ID (positive integer or 'none')");

/* 	If the file httpexampleclient.txt exists in the root directory of a drive

	it will be used to provide default initial values for when setting the URL, 

	proxy, etc. The file format is:

		[var] = [value]

	for example:

		PROXY = my.proxy.co.uk:5473

		URL = http://www.example.org/dir/file.html

	whitespace is ignored.

	Note: these are the the values set at run time, they are merely the default

	options presented to the user. In the above example the proxy is not turned

	on by default, but the string "my.proxy.co.uk:5473" is presetned as the default

	option when the use hits "p", saving them the need to have to type it in. 

	*/

// Defaults

_LIT(KDefaultFileFormat,"%c:\\httpexampleclient.txt");

_LIT(KHookString, "HOOK");

_LIT(KProxyString, "PROXY");

_LIT(KURLString, "URL");

_LIT(KSessionIdString, "SESSIONID");

_LIT(KNone, "none");

_LIT(KSchemeHttps, "https");

// File system root

_LIT(KFileSystemRoot,"C:\\");

// Standard headers used by default

_LIT8(KUserAgent, "HTTPExampleClient (1.0)");

_LIT8(KAccept, "*/*");

_LIT(KHttpExampleClientPanic, "HTTP-EC");

enum THttpExampleClientPanics

	{

	EReqBodySumitBufferNotAllocated,

	KBodyWithInvalidSize,

	KCouldntNotifyBodyDataPart

	};

// Size of buffer used when submitting request bodies

const TInt KMaxSubmitSize = 1024;

const TInt KMaxHeaderNameLen = 32;

const TInt KMaxHeaderValueLen = 128;

//

// Implementation of CHttpClient

//

// Supplied as the name of the test program to CHttpExampleUtils

_LIT(KHttpExampleClientTestName, "HttpExample");

CHttpClient::CHttpClient()

	: iReqBodySubmitBufferPtr(0,0)

	{

	// Initial timestamp is time now

	ResetTimeElapsed();

	}

CHttpClient::~CHttpClient()

	{

	iSess.Close();

	iFileServ.Close();

	delete iReqBodySubmitBuffer;

	delete iTransObs;

	delete iFormEncoder;

	delete iUtils;

	}

CHttpClient* CHttpClient::NewLC()

	{

	CHttpClient* me = new(ELeave) CHttpClient;

	CleanupStack::PushL(me);

	me->ConstructL();

	return me;

	}

CHttpClient* CHttpClient::NewL()

	{

	CHttpClient* me = NewLC();

	CleanupStack::Pop(me);

	return me;

	}

void CHttpClient::ConstructL()

	{

	iUtils = CHttpExampleUtils::NewL(KHttpExampleClientTestName);

	// Open the RHTTPSession

	iSess.OpenL();

	// Install this class as the callback for authentication requests

	InstallAuthenticationL(iSess);

	User::LeaveIfError(iFileServ.Connect());

	iTransObs = CHttpEventHandler::NewL(*iUtils);

	}

TBool CHttpClient::ClearHook(TPtrC aHook)

	{

	TInt err = KErrNone;

	if(aHook.Length()!=0)

		{

		err=iFileServ.Delete(aHook);

		}

	else if(iHookFileName.Length()>0) 

		{

		err=iFileServ.Delete(iHookFileName);

		if(err==KErrNone) 

			{

			iHookFileName.Zero();

			}

		}

	if(err == KErrNone)

		{

		User::InfoPrint(_L("Hook revmoved. Need to restart "));

		return ETrue;

		}

	return EFalse;

	}

// return true if need to quit

void CHttpClient::SetHookL(TDesC& aHook)

	{

	iHookFileName.Format(KHookFormat, &aHook);

	CFileMan* fileMan = CFileMan::NewL(iFileServ);

	CleanupStack::PushL(fileMan);

	TInt err = fileMan->Copy( iHookFileName, KHookDir, CFileMan::EOverWrite );

	if( err != KErrNone && err != KErrNotFound) 

		{

		User::Leave(err);

		}

	CleanupStack::PopAndDestroy(fileMan);

	User::InfoPrint(_L("Hook installed. Need to restart "));

	}

// to avoid code bloat

#define CHECKVAL(name)  if(var==K ##name##String)	{a##name.Copy(val);}

void CHttpClient::SetDefaults(TDes& aURL, TDes& aProxy, TDes& aHook, TDes& aSessionId)

	{

	TFileName filename;

	RFile file;

	filename.Format(KDefaultFileFormat, 'C');

	TInt err = file.Open(iFileServ, filename, EFileRead);

	if(err != KErrNone) 

		{

		filename.Format(KDefaultFileFormat, 'Z');

		err = file.Open(iFileServ, filename, EFileRead);

		}

	if(err == KErrNone) 

		{

		TFileText line;

		line.Set(file);

		TInt err = line.Read(filename);

		while(err == KErrNone || err ==KErrTooBig) 

			{

			filename.Trim();

			TInt div = filename.Locate('=');

			if(div>0) 

				{

				TInt i;

				for (i=div-1;i>0 && filename[i]==' ';i--);	// ibid

				TPtrC var = filename.Left(i+1);

				for (i=div+1;i<filename.Length() && filename[i]==' ';i++);	//ibid

				TPtrC val = filename.Right(filename.Length()-i);

				CHECKVAL(Hook)

				else CHECKVAL(Proxy)

				else CHECKVAL(URL)

				else CHECKVAL(SessionId);	

				}

			err = line.Read(filename);

			}

		}

		file.Close();

	}

void CHttpClient::StartClientL()

	{

	ResetTimeElapsed();

	TBuf<256> url;

	TBuf<256> prox;

	TBuf<256> hook;

	TBuf<16> sessid;

	SetDefaults(url, prox, hook, sessid);

	RStringPool strP = iSess.StringPool();

	// Repeat until user selects quit

	TBool done = EFalse;

	while (!done)

		{

		// Get user command from menu

		iUtils->Test().Console()->ClearScreen();

		_LIT(KSelectOption, " Select an option \n\n");

		iUtils->Test().Printf(KSelectOption);

		_LIT(KPossibleSelectionsText, " 1 Get \n 2 Post \n 3 Head \n 4 Trace \n 5 Toggle Verbose \n p Set Proxy\n s Set Session ID\n h Set Hook\n 6 Quit \n");

		_LIT(KPossibleSelections,"123456psh");

		TInt selection = iUtils->GetSelection(KPossibleSelectionsText, KPossibleSelections);

		RStringF method;

		iHasARequestBody = EFalse;

		// Set the method to use and ask for a url

		switch (selection)

			{

		case EGet: 

			method = strP.StringF(HTTP::EGET,RHTTPSession::GetTable());

			break;

		case EPost:

			method = strP.StringF(HTTP::EPOST,RHTTPSession::GetTable());

			iHasARequestBody = ETrue; 

			break;

		case EHead: 

			method = strP.StringF(HTTP::EHEAD,RHTTPSession::GetTable()); 

			break;

		case ETrace:

			method = strP.StringF(HTTP::ETRACE,RHTTPSession::GetTable()); 

			break;

		case EToggleVerbosity: 

			{

			TBool verbose = iTransObs->Verbose();

			iTransObs->SetVerbose(!verbose);

			if (!verbose)

				User::InfoPrint(_L("Verbosity is ON "));

			else

				User::InfoPrint(_L("Verbosity is OFF"));

			}

			break;

		case EQuit:

			done = ETrue;

			break;

		case EHook:

			iUtils->GetAnEntry(KEnterHook, hook);

			if(hook.Length() > 0) 

				{		

				if(hook[0]=='-')

					{

					hook.Replace(0,1,KHookDir);

					hook.Append(_L(".esk"));

					done = ClearHook(hook);

					}

				else 

					{

					SetHookL(hook);

					done = ETrue;

					}

				}

			break;

		case ESession:

			iUtils->GetAnEntry(KEnterSessId, sessid);

			if(sessid.Length() > 0) 

				{

				TLex number(sessid);

				TInt val;

				if(number.Val(val)==KErrNone)

					{

					iSess.ConnectionInfo().SetPropertyL(strP.StringF(HTTP::ESessionId,RHTTPSession::GetTable()), val);

					}

				else if (sessid==KNone) 

					{				

					iSess.ConnectionInfo().RemoveProperty(strP.StringF(HTTP::ESessionId,RHTTPSession::GetTable()));

					}

				}

			break;

		case EProxy:

			iUtils->GetAnEntry(KEnterProx, prox);

			if(prox.Length() > 0) 

				{

			  	  iSess.ConnectionInfo().SetPropertyL(iSess.StringPool().StringF(HTTP::EProxyUsage,RHTTPSession::GetTable()), iSess.StringPool().StringF(HTTP::EUseProxy,RHTTPSession::GetTable()));

				TBuf8<256> prox8;

				prox8.Copy(prox);

				 RStringF proxy = iSess.StringPool().OpenFStringL(prox8);

				CleanupClosePushL(proxy);

				 iSess.ConnectionInfo().SetPropertyL(iSess.StringPool().StringF(HTTP::EProxyAddress,RHTTPSession::GetTable()), proxy);

				CleanupStack::PopAndDestroy(&proxy);

				}

			else

				{

			 	iSess.ConnectionInfo().SetPropertyL(iSess.StringPool().StringF(HTTP::EProxyUsage,RHTTPSession::GetTable()), iSess.StringPool().StringF(HTTP::EDoNotUseProxy,RHTTPSession::GetTable()));

				}

			break;

		default: 

			 break;

			} 

		if (!done && (selection >= 0) && (selection <= 3))

			{

			// Get the URL

			_LIT(KEnterUrl, "Enter Url");

			iUtils->GetAnEntry(KEnterUrl, url);

			if (url.Length() > 0) // back to main menu if we still have an empty URL

				{

				// Get a filename to submit as our request body, for methods that need one

				if (iHasARequestBody)

					{

					GetRequestBodyL(method);

					}

				// Start the method off

				TBuf8<256> url8;

				url8.Copy(url);

				InvokeHttpMethodL(url8, method);

				}

			else

				{

				_LIT(KEmptyUrlErrorText, "You must supply a non empty url");

				iUtils->Test().Printf(KEmptyUrlErrorText);

				User::After(1000000);

				}

			}

		} // while

	// End of tests

	DisplayTimeElapsed();

	iUtils->PressAnyKey();

}

/** Get the post body data from the user. 

The user supplies name and value pairs. These are then encoded as added by CHTTPFormEncoder which acts as a data supplier

for this request. This is typically how a form body for a html form submission that uses POST might be constructed.

*/

void CHttpClient::GetPostBodyManuallyL()

	{

	if (iFormEncoder)

		{

		delete iFormEncoder;

		iFormEncoder = NULL;

		}

	iFormEncoder = CHTTPFormEncoder::NewL(); 

	TBuf<256> name;

	TBuf<256> value;

	TBuf8<256> name8;

	TBuf8<256> value8;

	_LIT(KGetPostName, "Enter Name (END to finish)");

	_LIT(KGetPostValue, "Enter Value ");

	_LIT(KEnd, "END");

	do

		{

		iUtils->GetAnEntry(KGetPostName, name);

		if (name.CompareF(KEnd) != 0)

			{

			iUtils->GetAnEntry(KGetPostValue, value);

			name8.Copy(name);

			value8.Copy(value);

			iFormEncoder->AddFieldL(name8, value8);

			}

		}

		while (name.CompareF(KEnd) != 0);

	}

/** Gets the body that you wish to submit when using a POST Method

This can be supplied as a file or can be input manually in name,value pairs

*/

void CHttpClient::GetRequestBodyL(RStringF& aMethod)

	{

	if (aMethod== iSess.StringPool().StringF(HTTP::EPOST,RHTTPSession::GetTable()))

		{

		// get a post body by file or contruct manually?

		_LIT(KConstructPostManually, "\n> Construct Post Manually?  Yes | No ");

		_LIT(KYesNo, "YyNn");

		TInt selection = iUtils->GetSelection(KConstructPostManually,KYesNo);

		if (selection < 2) // 2 is NO

			{

			iManualPost = ETrue;

			GetPostBodyManuallyL();

			return;

			}

		// else carry on as usual and get post data from a file

		}

	iManualPost = EFalse;

	_LIT(KRequestPathPrompt, "Enter path to request body file: ");

	iUtils->GetAnEntry(KRequestPathPrompt, iReqBodyFileName);

	_LIT(KRequestBodyContentType, "Enter request body content-type: ");

	iUtils->GetAnEntry(KRequestBodyContentType, iReqBodyContentType);

	iParsedFileName.Set(KFileSystemRoot,&iReqBodyFileName,NULL);

	// Check it exists and open a file handle

	if (iFileServ.IsValidName(iReqBodyFileName))

		{

		TInt err= iReqBodyFile.Open(iFileServ, iParsedFileName.FullName(), EFileRead);

		if(err!=KErrNone)

		   {

		   _LIT(KFileNotOpen, "Unable to open a file.\n");

		   iUtils->Test().Printf(KFileNotOpen);

		   User::Leave(KErrNotFound);			

		   } 

		delete iReqBodySubmitBuffer;

		iReqBodySubmitBuffer = NULL;

		iReqBodySubmitBuffer = HBufC8::NewMaxL(KMaxSubmitSize);

		iReqBodySubmitBufferPtr.Set(iReqBodySubmitBuffer->Des());

		}

	else

		{

		_LIT(KFileNameNotValid, "The specified filename is not valid!.\n");

		iUtils->Test().Printf(KFileNameNotValid);

		User::Leave(KErrBadName);

		}

	}

TBool CHttpClient::GetNextDataPart(TPtrC8& aDataPart)

	{

	__ASSERT_DEBUG(iReqBodySubmitBuffer, User::Panic(KHttpExampleClientPanic, EReqBodySumitBufferNotAllocated));

	// Read from the request body file

	iNoMoreDate = EFalse;

	TInt err = iReqBodyFile.Read(iReqBodySubmitBufferPtr);

	if (err == KErrNone)

		{

		aDataPart.Set(iReqBodySubmitBufferPtr);

		++iDataChunkCount;

		iNoMoreDate = (iReqBodySubmitBufferPtr.Length() == 0);

		}

	return iNoMoreDate;

	}

void CHttpClient::ReleaseData()

	{

	// Clear out the submit buffer

	TPtr8 buff = iReqBodySubmitBuffer->Des();

	buff.Zero();

	if (iNoMoreDate==EFalse)

		{

		// Notify HTTP of more data available immediately, since it's being read from file

		TRAPD(err, iTrans.NotifyNewRequestBodyPartL());

		if (err != KErrNone)

			User::Panic(KHttpExampleClientPanic, KCouldntNotifyBodyDataPart);

		}

	}

TInt CHttpClient::OverallDataSize()

	{

	TInt size = 0;

	TInt err = iReqBodyFile.Size(size);

	if (err < 0)

		User::Panic(KHttpExampleClientPanic,KBodyWithInvalidSize);

	return size;

	}

TInt CHttpClient::Reset()

  {

  if (iHasARequestBody)

    {

    // Reset to beginning of file

	TInt pos = 0;

    iReqBodyFile.Seek(ESeekStart, pos);

    return KErrNone;

    }

  else

    {

    return KErrNotSupported;

    }

 } 

/** Invoke the http method

This actually creates the transaction, sets the headers and body and then starts the transaction 

*/

void CHttpClient::InvokeHttpMethodL(const TDesC8& aUri, RStringF aMethod)

	{

	// Set the protocol, before the first transaction gets started

/*	if (iUseWspProtocol)

		{

		RHTTPConnectionInfo ci = iSess.ConnectionInfo();

		ci.SetPropertyL(iSess.StringPool().StringF(HTTP::EProtocol,RHTTPSession::GetTable()),

						THTTPHdrVal(iSess.StringPool().StringF(HTTP::EWSP,RHTTPSession::GetTable())));

		}*/

	iDataChunkCount = 0;

	TUriParser8 uri; 

	uri.Parse( aUri );

	HBufC* scheme = uri.DisplayFormL( EUriScheme );

	iTransObs->SetSecuredHttp( !scheme->CompareF( KSchemeHttps ) );

	delete scheme;

	iTrans = iSess.OpenTransactionL(uri, *iTransObs, aMethod);

	RHTTPHeaders hdr = iTrans.Request().GetHeaderCollection();

	// Add headers appropriate to all methods