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

// Implementation of CHttpHdrCodecTest

// 

//

#include <inetprottextutils.h>

#include "t_hdrcodec.h"

_LIT8(KFieldSeparator, "\n");

void CHttpHdrCodecTest::DoRunL()

	{

	iExpectedStatusCode = 0;

	// insert test code here, allow it to leave if the test fails

	RunTestsL();

	}

TInt CHttpHdrCodecTest::RunError(TInt aErr)

	{

	iEngine->Utils().LogIt(_L("\nTest failed with error code %d\n"), aErr);

	return KErrNone;

	}

void CHttpHdrCodecTest::DoCancel()

	{

	}

const TDesC& CHttpHdrCodecTest::TestName()

	{

	_LIT(KHeaderTestName,"CHttpHdrCodecTest");

	return KHeaderTestName;

	}

CHttpHdrCodecTest::CHttpHdrCodecTest()

// Constructor

	{

	// Initial timestamp is time now

	ResetTimeElapsed();

	}

CHttpHdrCodecTest::~CHttpHdrCodecTest()

// Destructor (virtual)

	{

	}

CHttpHdrCodecTest* CHttpHdrCodecTest::NewLC()

// Create a new CHttpHdrCodecTest and place it on the clean-up stack then return it

	{

	CHttpHdrCodecTest* me = new(ELeave) CHttpHdrCodecTest;

	CleanupStack::PushL(me);

	return me;

	}

CHttpHdrCodecTest* CHttpHdrCodecTest::NewL()

// Create a new CHttpHdrCodecTest and return it

	{

	CHttpHdrCodecTest* me = new(ELeave) CHttpHdrCodecTest;

	return me;

	}

void CHttpHdrCodecTest::RunTestsL()

	{

	// start the session

	iSession.OpenL();

	CleanupClosePushL(iSession);

	// Locate the string pool

	iStrP = iSession.StringPool();

	_LIT8(KTxtHTTPProt, "HTTP/TCP");

	iProtHnd = CProtocolHandler::NewL(KTxtHTTPProt(), iSession);

	CleanupStack::PushL(iProtHnd);

	iEngine->Console().Printf(_L("\nStart of header codec tests"));

	ResetTimeElapsed();

	// Run tests on parsed -> OTA data conversions, for each header implemented

	TestEncodeAcceptL();

	TestEncodeAcceptCharsetL();

	TestEncodeAuthorizationL();

	TestEncodeConnectionL();

	TestEncodeContentLengthL();

	TestEncodeContentTypeL();

	TestEncodeHostL();

	TestEncodeTransferEncodingL();

	TestEncodeUserAgentL();

	TestEncodeDateL();

	TestEncodeLocaleIndependentDateL();

	TestEncodeUpgradeL();

	TestEncodeCookieL();

	TestEncodeCacheControlL();

	// Run tests on OTA data -> parsed conversions, for each header implemented

	TestDecodeConnectionL();

	TestDecodeContentLengthL();

	TestDecodeContentTypeL();

	TestDecodeContentTypeWithMultipleParametersL();

	TestDecodeContentTypeMissingParameterL();

	TestDecodeServerL();

	TestDecodeDateL();

	TestDecodeTransferEncodingL();

	TestDecodeWWWAuthenticateL();

	TestDecodeUpgradeL();

	TestDecodeSetCookieL();

	TestDecodeSetCookie2L();

	TestDecodeSetCookie3L();

	TestDecodeSetCookie4L();

	//test for DEF080697: Cookies: When sending a request to an origin server, Browser is not sending the

	TestEncodeCookiePortL();

	TestDecodeCacheControlL();

	// Do some round trip encode->decode and check the original input is the same as the final output

	TestRoundTripConnectionL();

	TestRoundTripContentLengthL();

	TestRoundTripContentLength2L();

	TestRoundTripContentTypeL();

	TestRoundTripTransferEncodingL();

	TestDuplicateHeaderResponsesL();

	//test for DEF078684: Authorization header nonce count is sent incorrectly

	TestEncodeAuthorization1L();

	//test for INC118854:Invalid parsing of HTTP headers when value contains a ";" character 

	TestDecodeContentDispositionL();

	TestGetCustomFieldValueL();

	TestGetCustomFieldNullValueL();

	TestGetCustomFieldNoValueL();

	// test for PDEF138861: Trying to access header field part at Index < 0

	TestInvalidHeaderFieldPartL ();

	// End of tests

	DisplayTimeElapsed();

	iEngine->Console().Printf(_L("\nEnd of header codec tests"));

	CleanupStack::PopAndDestroy(2,&iSession);

	}

void CHttpHdrCodecTest::TestDuplicateHeaderResponsesL()

	{

	// Test Headers that are handled by textmode codec

	// Response with 2 connection type headers

	TInt numParts=2; 

	RStringF header = iStrP.StringF(HTTP::EConnection, RHTTPSession::GetTable());

	_LIT8(KConnectionRawHeader1, "close\nkeep-alive");

	TestDuplicateHeaderDecodingL(header,KConnectionRawHeader1, numParts);

	_LIT8(KConnectionRawHeader2, "close, keep-alive");

	TestDuplicateHeaderDecodingL(header,KConnectionRawHeader2, numParts);

	// Response with 2 content type headers

	numParts=1;

	header = iStrP.StringF(HTTP::EContentType, RHTTPSession::GetTable());

	_LIT8(KContentTypeRawHeader1, "text/html\ntext/html");

	TestDuplicateHeaderDecodingL(header, KContentTypeRawHeader1,numParts);

	// Response with 2 different date headers 

	// This also checks Expires and Last-Modified as they use the same generic date parsing in

	// the codec

	header = iStrP.StringF(HTTP::EDate, RHTTPSession::GetTable());

	_LIT8(KDateRawHeader1, "Tue, 15 Nov 1994 08:12:31 GMT\n Wed, 16 Nov 1994 09:00:30 GMT");

	TestDuplicateHeaderDecodingL(header, KDateRawHeader1, numParts);					

    // Response with 3 different content length headers

	// This also checks Age as that uses the same generic number parsing

	header = iStrP.StringF(HTTP::EContentLength, RHTTPSession::GetTable());

	_LIT8(KContentLengthRawHeader1, "10\n200\n3000");

	TestDuplicateHeaderDecodingL(header, KContentLengthRawHeader1, numParts);					

    // Response with 2 different transfer-encodings

    numParts=2;

    header = iStrP.StringF(HTTP::ETransferEncoding, RHTTPSession::GetTable());

    _LIT8(KTransferEncodingRawHeader1, "chunked\nchunked2");

    TestDuplicateHeaderDecodingL(header, KTransferEncodingRawHeader1, numParts);

    // Response with 2 different WWWAuthenticate headers

    header = iStrP.StringF(HTTP::EWWWAuthenticate, RHTTPSession::GetTable());

    _LIT8(KWWWAuthenticateRawHeader1, "basic realm=castle\nbasic realm=magic");

    TestDuplicateHeaderDecodingL(header, KWWWAuthenticateRawHeader1, numParts);

	// Response with 2 different SetCookie headers

	// Also tests SetCookie2 as that goes through same codec

	header = iStrP.StringF(HTTP::ESetCookie, RHTTPSession::GetTable());

    _LIT8(KSetCookieRawHeader1, "cookie=chocolate\ncookie=ginger");

    TestDuplicateHeaderDecodingL(header, KSetCookieRawHeader1, numParts);

    // Response with 2 cache control headers

    header = iStrP.StringF(HTTP::ECacheControl, RHTTPSession::GetTable());

    _LIT8(KCacheControlRawHeader1, "max-age=100\nno-transform");

    TestDuplicateHeaderDecodingL(header, KCacheControlRawHeader1, numParts);

    header = iStrP.StringF(HTTP::ECacheControl, RHTTPSession::GetTable());

    _LIT8(KCacheControlRawHeader2, "max-age=100, no-transform");

    TestDuplicateHeaderDecodingL(header, KCacheControlRawHeader2, numParts);

	// Response with 2 pragma headers

	// This also tests vary and content-langugage which both use same generic parser

	header = iStrP.StringF(HTTP::EPragma, RHTTPSession::GetTable());

    _LIT8(KPragmaRawHeader1, "pragma1\npragma2");

    TestDuplicateHeaderDecodingL(header, KPragmaRawHeader1, numParts);

    // Test some headers that will use the default codec

    numParts=2;

    // Response with 2 ETag headers

    header = iStrP.StringF(HTTP::EETag, RHTTPSession::GetTable());

    _LIT8(KETagRawHeader1, "\"xxxyz\"\n\"xyyz\"");

    TestDuplicateHeaderDecodingL(header, KETagRawHeader1, numParts);

	header = iStrP.StringF(HTTP::EETag, RHTTPSession::GetTable());

    _LIT8(KETagRawHeader2, "\"xxxyz\"\n\"xyyz\"   ");

    TestDuplicateHeaderDecodingL(header, KETagRawHeader2, numParts);

	// Default codec with one header

	numParts=1;

    header = iStrP.StringF(HTTP::EETag, RHTTPSession::GetTable());

    _LIT8(KETagRawHeader3, "\"xxxyz\"");

    TestDuplicateHeaderDecodingL(header, KETagRawHeader3, numParts);

	iEngine->Utils().LogIt ( _L ("\nTestDuplicateHeaderResponsesL test success."));

	}

void CHttpHdrCodecTest::TestDuplicateHeaderDecodingL(RStringF aHeader,

												 const TDesC8& aRawResponseHeader,

												 TInt aExpectedNumberParts)

	/* This is to test decoding of multiple response headers

	   HTTP Framework deals with these by joining them together using \n as a seperator

	*/

	{

	RHTTPHeaders headers = GetHeadersLC();

	headers.SetRawFieldL(aHeader, aRawResponseHeader, KFieldSeparator);

	// Check parsing of raw header. Framework should only return first header if there are duplicates

	// Don't bother checking multiple values on one line.

	TPtrC8 rawHeader(KNullDesC8);

	User::LeaveIfError(headers.GetRawField(aHeader,rawHeader));

	TInt firstNewLinePosition = aRawResponseHeader.Locate('\n');

	if (firstNewLinePosition != KErrNotFound)

		{

		TPtrC8 expectedRawResponse(rawHeader.Left(firstNewLinePosition));

		if (rawHeader.CompareF(expectedRawResponse) != 0)

			{

			TBuf < 1024 > rawHeader16;

			TBuf < 1024 > expectedRawResponse16;

			rawHeader16.Copy ( rawHeader );

			expectedRawResponse16.Copy ( expectedRawResponse );

			iEngine->Utils().LogIt ( _L("Raw header comparison failed -> %S != %S"), &rawHeader16, &expectedRawResponse16 );

			User::Leave(KErrArgument);

			}

		}

	// Check parsing when converting through codec

	TInt numberParts = headers.FieldPartsL(aHeader);

	if (numberParts != aExpectedNumberParts)

		{

		iEngine->Utils().LogIt ( _L("\nNumber parts differ from expected. -> %d != %d"), numberParts, aExpectedNumberParts );

		User::Leave(KErrArgument);

		}

	ReleaseHeaders();

	iEngine->Utils().LogIt ( _L ("\nTestDuplicateHeaderDecodingL test success."));

	}

void CHttpHdrCodecTest::TestEncodeAcceptL()

	{

	// the result we expect and that we will validate

	_LIT8(KAcceptOTAVal, "text/html; q=0.8, text/vnd.wap.wml; q=0.2; extended=value, text/*");

	// Open strings used in this test

	RStringF textHtml = iStrP.StringF(HTTP::ETextHtml,RHTTPSession::GetTable());

	RStringF textWml  = iStrP.StringF(HTTP::ETextVndWapWml,RHTTPSession::GetTable());

	RStringF textAny  = iStrP.StringF(HTTP::ETextAny,RHTTPSession::GetTable());

	RStringF extended  = iStrP.OpenFStringL(_L8("extended"));

	CleanupClosePushL(extended);

	RStringF extendVal = iStrP.OpenFStringL(_L8("value"));

	CleanupClosePushL(extendVal);

	//

	iEngine->Console().Printf(_L("\nTest encoding 'Accept' header"));

	RHTTPHeaders hdr = 	GetHeadersLC();

	// Set up the parsed header

	RStringF accept = iStrP.StringF(HTTP::EAccept,RHTTPSession::GetTable());

	THTTPHdrVal accVal(textHtml);

	THTTPHdrVal q(THTTPHdrVal::TQConv(0.8));

	hdr.SetFieldL(accept, accVal, iStrP.StringF(HTTP::EQ,RHTTPSession::GetTable()), q);

	accVal.SetStrF(textWml);

	q.SetInt(THTTPHdrVal::TQConv(0.2));

	hdr.SetFieldL(accept, accVal, iStrP.StringF(HTTP::EQ,RHTTPSession::GetTable()), q);

	q.SetStrF(extendVal);

	hdr.SetFieldL(accept, accVal, extended, q);

	accVal.SetStrF(textAny);

	hdr.SetFieldL(accept, accVal);

	// now get the OTA data

	TPtrC8 ota;

	hdr.GetRawField(accept, ota);

	if (ota.Compare(KAcceptOTAVal()) != 0)

		{		

		TBuf<512> ota16;

		ota16.Copy (ota);		

		iEngine->Utils().LogIt ( _L("\nAccept OTA comparison failed -> %S"), &ota16 );

		User::Leave(KErrArgument);		

		}

	// Close strings used in this test

	ReleaseHeaders();

	CleanupStack::PopAndDestroy(2, &extended);

	iEngine->Utils().LogIt ( _L ("\nTestEncodeAcceptL test success."));

	}

void CHttpHdrCodecTest::TestEncodeAcceptCharsetL()

	{

	// the result we expect and that we will validate

	_LIT8(KAccChSetOTAVal, "iso-8859-5, unicode-1-1; q=0.8");

	// Open strings used in this test

	RStringF iso8859_5 = iStrP.OpenFStringL(_L8("iso-8859-5"));

	CleanupClosePushL(iso8859_5);

	RStringF unicode1_1  = iStrP.OpenFStringL(_L8("unicode-1-1"));

	CleanupClosePushL(unicode1_1);

	iEngine->Console().Printf(_L("\nTest encoding 'Accept-Charset' header"));

	RHTTPHeaders hdr = GetHeadersLC();

	// Set up the parsed header

	RStringF accChSet= iStrP.StringF(HTTP::EAcceptCharset,RHTTPSession::GetTable());

	THTTPHdrVal accChSetVal(iso8859_5);

	hdr.SetFieldL(accChSet, accChSetVal);

	//

	accChSetVal.SetStrF(unicode1_1);

	THTTPHdrVal q(THTTPHdrVal::TQConv(0.8));

	hdr.SetFieldL(accChSet, accChSetVal, iStrP.StringF(HTTP::EQ,RHTTPSession::GetTable()), q);

	// now get the OTA data

	TPtrC8 ota;

	hdr.GetRawField(accChSet, ota);

	if (ota.Compare(KAccChSetOTAVal()) != 0)

		{

		TBuf<512> ota16;

		ota16.Copy (ota);		

		iEngine->Utils().LogIt ( _L("\nComparison of charset field failed -> %S"), &ota16 );	

		User::Leave(KErrArgument);

		}

	ReleaseHeaders();

	CleanupStack::PopAndDestroy(2, &iso8859_5);

	iEngine->Utils().LogIt ( _L ("\nTestEncodeAcceptCharsetL test success."));

	}

//test for DEF078684: Authorization header nonce count is sent incorrectly

void CHttpHdrCodecTest::TestEncodeAuthorization1L()

{

	// the result we expect and that we will validate

	_LIT8(KAuthorizationOTAVal, "Digest nc=00000001, algorithm=MD5, nonce=\"999999\"");

	// Open strings used in this test

	RString ncVal = iStrP.OpenStringL(_L8("00000001"));

	CleanupClosePushL(ncVal);

	RString algVal = iStrP.OpenStringL(_L8("MD5"));

	CleanupClosePushL(algVal);

	RString nonceVal = iStrP.OpenStringL(_L8("999999"));

	CleanupClosePushL(nonceVal);

	//

	iEngine->Console().Printf(_L("\nTest encoding 'Authorization' header"));

	RHTTPHeaders hdr = GetHeadersLC();

	RStringF authStr = iStrP.StringF(HTTP::EAuthorization,RHTTPSession::GetTable());

	THTTPHdrVal authVal(iStrP.StringF(HTTP::EDigest,RHTTPSession::GetTable()));

	THTTPHdrVal hdrNcVal(ncVal);

	RStringF ncStr = iStrP.StringF(HTTP::ENc,RHTTPSession::GetTable());

	hdr.SetFieldL(authStr, authVal,ncStr,hdrNcVal);

	THTTPHdrVal hdralgVal(algVal);

	RStringF algStr = iStrP.StringF(HTTP::EAlgorithm,RHTTPSession::GetTable());

	hdr.SetFieldL(authStr, authVal,algStr,hdralgVal);

	THTTPHdrVal hdrnonceVal(nonceVal);

	RStringF nonceStr = iStrP.StringF(HTTP::ENonce,RHTTPSession::GetTable());

	hdr.SetFieldL(authStr, authVal,nonceStr,hdrnonceVal);

	// now get the OTA data

	TPtrC8 ota;

	hdr.GetRawField(authStr, ota);

	if (ota.Compare(KAuthorizationOTAVal()) != 0)

		{

		TBuf<512> ota16;

		ota16.Copy (ota);

		iEngine->Utils().LogIt ( _L("\nComparison of Authorization field failed -> %S"), &ota16 );	

		User::Leave(KErrArgument);

		}

	ReleaseHeaders();

	CleanupStack::PopAndDestroy(&nonceVal);

	CleanupStack::PopAndDestroy(&algVal);

	CleanupStack::PopAndDestroy(&ncVal);

	iEngine->Utils().LogIt ( _L ("\nTestEncodeAuthorizationL test success."));

}

void CHttpHdrCodecTest::TestEncodeAuthorizationL()

	{

	// the result we expect and that we will validate

	_LIT8(KAuthorizationOTAVal, "Basic c3ltYmlhbjpmMXN5bmNtbA==");

	// Open strings used in this test

	RString basicCred = iStrP.OpenStringL(_L8("c3ltYmlhbjpmMXN5bmNtbA=="));

	CleanupClosePushL(basicCred);

	//

	iEngine->Console().Printf(_L("\nTest encoding 'Authorization' header"));

	RHTTPHeaders hdr = GetHeadersLC();

	RStringF authStr = iStrP.StringF(HTTP::EAuthorization,RHTTPSession::GetTable());

	THTTPHdrVal authVal(iStrP.StringF(HTTP::EBasic,RHTTPSession::GetTable()));

	hdr.SetFieldL(authStr, authVal); // sets part 1

	THTTPHdrVal credVal(basicCred);

	hdr.SetFieldL(authStr, credVal); // sets part 2

	// now get the OTA data

	TPtrC8 ota;

	hdr.GetRawField(authStr, ota);

	if (ota.Compare(KAuthorizationOTAVal()) != 0)

		{

		TBuf<512> ota16;

		ota16.Copy (ota);

		iEngine->Utils().LogIt ( _L("\nComparison of Authorization field failed -> %S"), &ota16 );	

		User::Leave(KErrArgument);

		}

	ReleaseHeaders();

	CleanupStack::PopAndDestroy(&basicCred);

	iEngine->Utils().LogIt ( _L ("\nTestEncodeAuthorizationL test success."));

	}

void CHttpHdrCodecTest::TestEncodeConnectionL()

	{

	// the result we expect and that we will validate

	_LIT8(KConnectionOTAVal, "close, keep-alive");

	// Open strings used in this test

	RStringF otherToken = iStrP.StringF(HTTP::EKeepAlive,RHTTPSession::GetTable());

	//

	iEngine->Console().Printf(_L("\nTest encoding 'Connection' header"));

	RHTTPHeaders hdr = GetHeadersLC();

	RStringF connStr = iStrP.StringF(HTTP::EConnection,RHTTPSession::GetTable());

	THTTPHdrVal connVal(iStrP.StringF(HTTP::EClose,RHTTPSession::GetTable()));

	hdr.SetFieldL(connStr, connVal); // sets part 1

	THTTPHdrVal keepAlive(otherToken);

	hdr.SetFieldL(connStr, keepAlive); // sets part 2

	// now get the OTA data

	TPtrC8 ota;

	hdr.GetRawField(connStr, ota);

	if (ota.CompareF(KConnectionOTAVal()) != 0)

		{

		TBuf<512> ota16;

		ota16.Copy (ota);

		iEngine->Utils().LogIt ( _L("\nComparison of connection field failed -> %S"), &ota16 );	

		User::Leave(KErrArgument);

		}

	ReleaseHeaders();

	iEngine->Utils().LogIt ( _L ("\nTestEncodeConnectionL test success.\n"));

	}

void CHttpHdrCodecTest::TestEncodeContentLengthL()

	{

	// the result we expect and that we will validate

	_LIT8(KContentLengthOTAVal, "12345");

	//

	iEngine->Console().Printf(_L("\nTest encoding 'Content-Length' header"));

	RHTTPHeaders hdr = GetHeadersLC();

	RStringF lengthStr = iStrP.StringF(HTTP::EContentLength,RHTTPSession::GetTable());

	THTTPHdrVal lengthVal(12345);

	hdr.SetFieldL(lengthStr, lengthVal); // sets part 1

	// now get the OTA data

	TPtrC8 ota;

	hdr.GetRawField(lengthStr, ota);

	if (ota.Compare(KContentLengthOTAVal()) != 0)

		{

		TBuf<512> ota16;

		ota16.Copy (ota);

		iEngine->Utils().LogIt ( _L("\nComparison of content length field value failed. -> %S"), &ota16 );	

		User::Leave(KErrArgument);

		}