// Copyright (c) 2002-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 "chttpclientheaderreader.h"

#include <http/rhttpsession.h>

#include <httpstringconstants.h>

#include <httperr.h>

#include <inetprottextutils.h>

_LIT8(KFieldSeparator, "\n");

_LIT8(KCommaNewline, ",\n");

CHttpClientHeaderReader* CHttpClientHeaderReader::NewL(RStringPool aStringPool)

	{

	return new (ELeave) CHttpClientHeaderReader(aStringPool);

	}

CHttpClientHeaderReader::~CHttpClientHeaderReader()

	{

	// The reader and writer classes are owned and destroyed by the base class CHeaderCodec

	}

CHttpClientHeaderReader::CHttpClientHeaderReader(RStringPool aStringPool)

: CHttpHeaderReader(aStringPool)

	{

	}

void CHttpClientHeaderReader::DecodeSetCookieL(RHeaderField& aHeader) const

	{

	// The generic header will add each cookie on a separate line. Each cookie forms 1 part.

	// Each cookie will be made up of a number of parameters with name being the first and value the second, following that 

	// will be just the actual parameters that form part of a cookie.

	TPtrC8 buffer;

	aHeader.RawDataL(buffer);	

	TInt remainingSize = buffer.Length();

	TPtrC8 token;

	TInt cookieCount = 0;

	while (remainingSize > 0)

		{

		// Extract tokens seperated by header field separator defined as KFieldSeparator in textmodestd.h

		remainingSize -= InetProtTextUtils::ExtractNextTokenFromList(buffer, token, KFieldSeparator);

		DecodeOneCookieL(aHeader, token, cookieCount);

		++cookieCount;

		}

	}

void CHttpClientHeaderReader::DecodeOneCookieL(RHeaderField& aHeader, const TDesC8& aCookieData, TInt aCookieNumber) const

	{

	// NETSCAPE Persistent Client State HTTP Cookies (http://www.netscape.com/newsref/std/cookie_spec.html)

	// SetCookie: NAME=VALUE [; expires=DATE] [; path=PATH] [; domain= DOMAIN_NAME] [; secure]

	// RFC 2109

	// set-cookie = "Set-Cookie:" 1#cookie

	// cookie = NAME "=" VALUE *(";" cookie-av)

	// cookie-av = "Comment" "=" value | "Domain" "=" value | "Max-Age" "=" value | "Path" "=" value | "Secure" |

	//				| "Version" "=" 1*DIGIT

	// RFC 2965 

	// set-cookie = "Set-Cookie2:" 1#cookie

	// cookie = NAME "=" VALUE *(";" set-cookie-av)

	// set-cookie-av = "Comment" "=" value | CommentURIL "=" <"> http_URL <">| "Discard" | "Domain" "=" value 

	//					| "Max-Age" "=" value | "Path" "=" value | "Port" ["=" <"> portlist <">] |

	//					"Secure" | "Version" "=" 1*DIGIT

	// portlist = 1#portnum

	// portnum = 1*DIGIT

	CHeaderFieldPart* cookie = SetNewFStringPartL(aHeader, aCookieNumber, iStrPool.StringF(HTTP::ECookie,iStringTable).DesC());

	TPtrC8 buffer(aCookieData);

	TInt remainingSize = buffer.Length();

	TPtrC8 token;

	TBool setCookieNameAndValue = EFalse;

	while (remainingSize > 0)

		{

		remainingSize -= InetProtTextUtils::ExtractNextTokenFromList(buffer, token, ';');

		if (setCookieNameAndValue)

			{

			// first attribute is NAME = VALUE which needs to be stored as 2 attributes

			SetParamNameAndValueL(*cookie, token, EFalse);

			}

		else

			{

			SetCookieNameAndValueL(*cookie, token);

			setCookieNameAndValue = ETrue;

			}

		}

	}

void CHttpClientHeaderReader::SetCookieNameAndValueL(CHeaderFieldPart& aCookie, const TDesC8& aNameValue) const

	{

	TInt equalPos = aNameValue.Locate('=');

	if (equalPos <= 0)

		User::Leave(KErrHttpDecodeCookie);

	TPtrC8 nameVal(aNameValue.Left(equalPos));

	InetProtTextUtils::RemoveWhiteSpace(nameVal, InetProtTextUtils::ERemoveBoth);

	TPtrC8 name(iStrPool.StringF(HTTP::ECookieName,iStringTable).DesC());

	SetNewStringParamL(aCookie, name ,nameVal);

	TPtrC8 valueVal(aNameValue.Right(aNameValue.Length() - (equalPos+1)));

	InetProtTextUtils::RemoveWhiteSpace(valueVal, InetProtTextUtils::ERemoveBoth);

	TPtrC8 value(iStrPool.StringF(HTTP::ECookieValue,iStringTable).DesC());

	SetNewStringParamL(aCookie, value ,valueVal);

	}

// Convert the WWW-Authenticate header field from OTA to generic form.

void CHttpClientHeaderReader::DecodeWWWAuthenticateL( RHeaderField& aHeader ) const

	{

	// RFC2616, section 14.47 WWW-Authenticate

	// RFC2617, 'HTTP Authentication: Basic and Digest Access Authentication'

	//

	// WWW-Authenticate  = "WWW-Authenticate" ":" 1#challenge

	// challenge   = auth-scheme 1*SP 1#auth-param

	// auth-scheme    = token

	// auth-param     = token "=" ( token | quoted-string )

	// There may be one or more challenge, in a comma-separated list.

	TPtrC8 buffer;

	aHeader.RawDataL( buffer );

	TInt totalBytesConsumed = 0;

	TInt numChallenges = 0;

	CHeaderFieldPart* part = NULL;

	TBool done = EFalse;

	while ( !done )

		{

		_LIT8( commaSpaceNewline, ", \n" );

		TPtrC8 token;

		TInt bytesConsumed = InetProtTextUtils::ExtractNextTokenFromList( buffer, token, commaSpaceNewline );

		done = (bytesConsumed == 0);

		if ( done && ( numChallenges == 0 ) ) // if we didn't find _anything_ at all...

			{

			User::Leave( KErrHttpDecodeWWWAuthenticate );

			}

		if ( !done && ( token.Length() > 0 ) )

			{

			totalBytesConsumed += bytesConsumed;

			TBool equalsPresent = ( token.Locate( '=' ) != KErrNotFound );

			if ( ( totalBytesConsumed == bytesConsumed ) && equalsPresent )

				{

				// The first token has an equals sign in it. That

				// can't be as it has to be an authentication scheme

				User::Leave( KErrHttpDecodeWWWAuthenticate );

				}

			if ( !equalsPresent )

				{

				// Got a new part. Add it.

				++numChallenges;

				part = SetNewFStringPartL( aHeader, numChallenges - 1, token );

			if( token.Compare( iStrPool.StringF(HTTP::ENTLM, iStringTable).DesC() ) == 0 )

				{

 					TInt consumed = InetProtTextUtils::ExtractNextTokenFromList( buffer, token, commaSpaceNewline );

 					if( consumed > 0 )

 						{

 						++numChallenges;

 						part = SetNewFStringPartL( aHeader, numChallenges -1, token );

						}

					}

				}

			else

				{

				// Got a param & parameter value.

				TPtrC8 paramName;

				TInt paramBytesConsumed = InetProtTextUtils::ExtractNextTokenFromList( token, paramName, '=' );

				if ( paramBytesConsumed == 0 )

					{

					User::Leave( KErrHttpDecodeBasicAuth );

					}

				// Obtain the parameter value. It is a string which

				// may or may not be quoted. 

				TPtrC8 paramVal;

				if ( token.Length() > 0 && token[0] == '"' )

					{

					bytesConsumed += InetProtTextUtils::ExtractQuotedStringL( token, paramVal );

					}

				else

					{

					paramVal.Set( token );

					}

				SetNewStringParamL( *part, paramName, paramVal );

				}

			}

		}

	}

/*

 *	Methods from CHeaderReader

 */

void CHttpClientHeaderReader::DecodeHeaderL(RHeaderField& aHeader)

	{

	RStringF fieldStr = iStrPool.StringF(aHeader.Name()); // this doesn't have to be closed

	switch(fieldStr.Index(iStringTable))

		{

		case HTTP::EWWWAuthenticate:

			{

			DecodeWWWAuthenticateL(aHeader);

			} break;

		case HTTP::ESetCookie:

		case HTTP::ESetCookie2:

			DecodeSetCookieL(aHeader);

			break;

		case HTTP::EAge:

			DecodeGenericNumberL(aHeader);

			break;

		case HTTP::EVary:

		case HTTP::EContentLanguage:

		case HTTP::EUpgrade:

			DecodeGenericL(aHeader, KCommaNewline);

			break;

		case HTTP::ELastModified:

			DecodeDateL(aHeader);

			break;

		default:

			User::Leave(KErrNotSupported);

		}

	}