// Copyright (c) 2003-2010 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 "csocket.h"

#include "chttpsecuresocket.h"

#include "mcommsinfoprovider.h"

#include "thttptrlayerpanic.h"

#include  <http/rhttpconnectioninfo.h>

#include <httpsocketconstants.h>

#include <in6_opt.h> 

const TInt KDefaultFlags		= 0;

const TInt KTcpTriggeredKeepAlive	= 2;

const TInt KSocketRecvBufSize = 16 * 1024;

const TInt KSocketSendBufSize = 16 * 1024;

const TInt KSocketDefaultSendBufSize = 4 * 1024;

CSocket* CSocket::NewL(MCommsInfoProvider& aCommsInfoProvider, TSocketType aSocketType)

/**	

	The factory constructor.

	@param		aCommsInfoProvider	The comms info provider object.

	@param		aSocketType			The type of the socket.

	@return		A pointer to a fully constructed object.

*/

	{

	CSocket* self = new (ELeave) CSocket(aCommsInfoProvider);

	CleanupStack::PushL(self);

	self->ConstructL(aSocketType);

	CleanupStack::Pop(self);

	return self;

	}

CSocket* CSocket::New(MCommsInfoProvider& aCommsInfoProvider, TSocketType aSocketType)

    {

    CSocket* self = new CSocket(aCommsInfoProvider);

    if(self)

        {

        TInt error = self->Construct(aSocketType);

        if(error != KErrNone)

            {

            delete self;

            self = NULL;

            }

        }

    return self;

    }

CSocket::~CSocket()

/**

	Destructor.

*/

	{

	// Close socket and notify the socket status observer

	if( iSecureSocketController )

		delete iSecureSocketController;

	else

		iSocket.Close();

	}

CSocket::CSocket(MCommsInfoProvider& aCommsInfoProvider)

: CBase(), iCommsInfoProvider(aCommsInfoProvider)

/**	

	Constructor.

	@param		aCommsInfoProvider	The comms info provider object.

*/

	{

	}

void CSocket::ConstructL(TSocketType aSocketType)

/**	

	Second phase constructor.

	@param		aSocketType	The type of the socket.

*/

	{

    User::LeaveIfError(Construct(aSocketType));

	}

TInt CSocket::Construct(TSocketType aSocketType)

    {

    TInt error = KErrNone;

    switch( aSocketType )

        {

    case EProtocolSocket:

        {

        if ( iCommsInfoProvider.HasConnection() )

            {

            // Open a protocol socket with a RConnection

            error = iSocket.Open(iCommsInfoProvider.SocketServer(),

                                 iCommsInfoProvider.ProtocolDescription().iAddrFamily, 

                                 iCommsInfoProvider.ProtocolDescription().iSockType, 

                                 iCommsInfoProvider.ProtocolDescription().iProtocol,

                                 iCommsInfoProvider.Connection()

                                 );              

            }

        else

            {

            // Open a protocol socket without a RConnection, we don't need one ( Loopback address )

            error = iSocket.Open(iCommsInfoProvider.SocketServer(),

                                 iCommsInfoProvider.ProtocolDescription().iAddrFamily, 

                                 iCommsInfoProvider.ProtocolDescription().iSockType, 

                                 iCommsInfoProvider.ProtocolDescription().iProtocol

                                 );                                          

            }

        } break;

    case EBlankSocket:

        {

        // Open a blank socket

        error = iSocket.Open(iCommsInfoProvider.SocketServer());

        } break;

    default:

        User::Invariant();

        }

    if(error == KErrNone)

        {

        TInt id = iCommsInfoProvider.SessionId();

        if(id>=0) 

            {

            // set socket option

            iSocket.SetOpt(KSOHttpSessionId, KSOLHttpSessionInfo, id);

            iSocket.SetOpt(KSoTcpKeepAlive, KSolInetTcp, KTcpTriggeredKeepAlive);

            }

        if(aSocketType != EBlankSocket)

            {

            iSocket.SetOpt(KSoTcpNoDelay,KSolInetTcp,1);  // Disable the nagle.

            iSocket.SetOpt(KSORecvBuf, KSOLSocket, KSocketRecvBufSize); // Set the socket recv buf to be 16K

            }

        }

    return error;

    }

TInt CSocket::Listen(TUint aQSize, TUint16 aPort)

/**	

	Start the listen service. The socket is bound to the local port specified by

	aPort. The listen service is then started. The aQSize argument specifies the

	number of connections that can be received simultaneously, awaiting to be 

	accepted.

	@param		aQSize		The length of the listening queue.

	@param		aPort		The local port number on which to listen.

	@return		A value of KErrNone if the listen service was successfully 

				started.

*/

	{

	// Bind the socket to the port

	TSockAddr addr;

	addr.SetPort(aPort);

	TInt error = iSocket.Bind(addr);

	if( error == KErrNone )

		{

		// Start the listening service

		error = iSocket.Listen(aQSize);

		}

	return error;

	}

void CSocket::Accept(CSocket& aBlankSocket, TRequestStatus& aStatus)

/**	

	Start asynchronous accept service. The socket should have had the listening

	service started. When a connection has been received, the blank socket will

	be given the connection. The request status is completed either when a 

	connection is received or an error has occured - this is reflected in the 

	value of the request status.

	@param		aBlankSocket	The blank socket which will be given the 

								connection if one is received.

	@param		aStatus			The request status that is completed when the

								accept service completes.

*/

	{

	if( iSecureSocketController )

		{

		// Secure socket do not support secure listening

		aStatus=KRequestPending;

		TRequestStatus* pStat = &aStatus;

		User::RequestComplete(pStat, KErrNotSupported);

		}

	else

		iSocket.Accept(aBlankSocket.iSocket, aStatus);

	}

void CSocket::CancelAccept()

/**	

	Cancel the accept service.

*/

	{

	iSocket.CancelAccept();

	}

void CSocket::Connect(TInetAddr& aAddr, TRequestStatus& aStatus)

/**	

	Start asynchronous connect service. The address contains the IP address and

	port with which a tcp connection should be established with. The request 

	status is completed either when a connection has been established or an error

	has occurred - this is reflected in the value of the request status.

	@param		aAddr	The IP address and port of the remote host.

	@param		aStatus	The request status that is completed when the connect 

						service completes.

*/

	{

	iSocket.Connect(aAddr, aStatus);

	}

void CSocket::CancelConnect()

/**	

	Cancel the connect service.

*/

	{

	iSocket.CancelConnect();

	}

void CSocket::RecvOneOrMore(TDes8& aBuffer, TRequestStatus& aStatus)

/**	

	Receive data from socket asynchronously. Any data received by the socket is 

	placed in the buffer supplied by aBuffer. The request status is completed 

	either when data has been received or an error has occurred - this is 

	reflected in the value of the request status.

	@param		aBuffer	The buffer where any received data is placed.	

	@param		aStatus	The request status that is completed when the receive 

						service completes.

*/

	{

	if( iSecureSocketController )

		// aFlags not used for secure sockets

		iSecureSocketController->RecvOneOrMore(aBuffer, aStatus, iBytesReceived);

	else

		iSocket.RecvOneOrMore(aBuffer, KDefaultFlags, aStatus, iBytesReceived);

	}

void CSocket::CancelRecv()

/**	

	Cancel the receive service.

*/

	{

	if( iSecureSocketController )

		iSecureSocketController->CancelRecv();

	else

		iSocket.CancelRecv();

	}

void CSocket::Send(const TDesC8& aBuffer, TRequestStatus& aStatus)

/**	

	Send data to the socket asynchronously. The data in the supplied buffer is 

	sent to the socket. The request status is completed either when data has 

	been sent or an error has occurred - this is reflected in the value of the

	request status.

	@param		aBuffer	The buffer containing the data to be sent.

	@param		aStatus	The request status that is completed when the send 

						service completes.

*/

	{

	if( iSecureSocketController )

		iSecureSocketController->Send(aBuffer, aStatus);

	else

		iSocket.Write(aBuffer, aStatus);

	}

void CSocket::CancelSend()

/**	

	Cancel the send service.

*/

	{

	if( iSecureSocketController )

		iSecureSocketController->CancelSend();

	else

		iSocket.CancelWrite();

	}

void CSocket::Shutdown(TRequestStatus& aStatus)

/**	

	Shutdown the connection asynchronously. Any pending receive or send requests

	are allowed to complete before the connection is shutdown. The request 

	status is completed when either the connection has shutdown or an error has

	occurred - this is reflected in the value of the request status. This request

	cannot be cancelled. This method does not apply to secure connections. The

	request will complete with KErrNotSupported if used with a secure connection.

	@param		aStatus	The request status that is completed when the shutdown 

						service completes.

*/

	{

	if( iSecureSocketController )

		{

		// Not supported with secure connections

		aStatus=KRequestPending;

		TRequestStatus* pStat = &aStatus;

		User::RequestComplete(pStat, KErrNotSupported);

		}

	else

		iSocket.Shutdown(RSocket::ENormal, aStatus);

	}

void CSocket::ShutdownImmediate()

/**	

	Shutdown the connection synchronously. This method will shutdown the socket

	connection immediately using the abortive closing functionality. This method

	has no effect with secure connections.

*/

	{

	if( iSecureSocketController==NULL )

		{

		TRequestStatus shutdownStatus = KRequestPending;

		iSocket.Shutdown(RSocket::EImmediate, shutdownStatus);

		User::WaitForRequest(shutdownStatus);

		}

	}

void CSocket::RemoteName(TInetAddr& aAddr)

/**	

	Get the remote host name. The IP address and port of the remote host is set

	in the output argument.

	@param		aAddr	The output argument where the IP address and port of the

						remote host is placed.

*/

	{

	iSocket.RemoteName(aAddr);

	}

void CSocket::LocalName(TInetAddr& aAddr)

/**	

	Get the local socket name. The IP address and port of the local socket is 

	set in the output argument.

	@param		aAddr	The output argument where the IP address and port of the

						local socket is placed.

*/

	{

	iSocket.LocalName(aAddr);

	}

const CX509Certificate* CSocket::ServerCert()

 /**

 	Get the Server Certificate for this socket session.

	@return		A Certificate with the certificate information. This may be 

				NULL if the information is not available.

 */

 	{

 	if( iSecureSocketController )

		{

		return iSecureSocketController->ServerCert();

		}

	return NULL;

 	}

TInt CSocket::CipherSuite(TDes8& aCipherSuite)

/**

	Get the current Cipher Suite for this socket session.

	@param		aCiptherSuite	A descriptor which will be filled with the cipher suite. 

								This is a 2 digit code as defined by RFC 2246. 

	@return		An error code. KErrNone on sucess. KErrNotSupported if this socket session is not secure.

*/

	{

	if (iSecureSocketController)

		return iSecureSocketController->CipherSuite(aCipherSuite);

	//else

	return KErrNotSupported;

	}

void CSocket::UpgradeToSecureL(TRequestStatus& aStatus, const TDesC8& aHostName,const TDesC& aProtocol)

/**

	Asynchronous request to upgrade the socket to a secure socket.

	@param	aStatus The request status, this will complete with KErrNone on the successful

					upgrade to a secure connection.

	@param	aHostName The host name of the server used for domain name checking

*/

	{

	// Check that the socket is already a secure one.

	if( iSecureSocketController )

		{

		// The socket connection is already secure, simply complete the request

		aStatus=KRequestPending;

		TRequestStatus* pStat = &aStatus;

		User::RequestComplete(pStat, KErrNone);

		}

	else

		{

		// Create the secure socket controller and start the secure handshake

		TRAPD(error, iSecureSocketController = CHttpSecureSocket::NewL(iSocket, iCommsInfoProvider));

		if (error == KErrNotFound || error == KEComErrNoInterfaceIdentified)

			User::Leave(KErrNotSupported);

		else if (error != KErrNone)

			User::Leave(error);

		iSecureSocketController->StartSecureHandshakeL(aStatus, aHostName,aProtocol);

		}

	}

void CSocket::CancelUpgradeToSecure()

/**

	Cancel the upgrade to a secure connection.

*/

	{

	if( iSecureSocketController )

		iSecureSocketController->CancelHandshake();

	}

TInt CSocket::PendingBytesToRead ()

	{

    if ( iSecureSocketController )

		    return iSecureSocketController->PendingBytesToRead ();

	TInt bytesToRead;

	TInt err = iSocket.GetOpt ( KSOReadBytesPending, KSOLSocket, bytesToRead );

	if ( err == KErrNone )

		return bytesToRead;

	return err;

	}

void CSocket::SetTCPCorking(TBool aValue)

    {

    if(aValue)

    	{

    	iSocket.SetOpt(KSoTcpNoDelay,KSolInetTcp,0);  // Disable the nagle.

      iSocket.SetOpt(KSOSendBuf, KSOLSocket, KSocketSendBufSize);

      }

    else

    	{

    	iSocket.SetOpt(KSoTcpNoDelay,KSolInetTcp,1);  // Disable the nagle.

      iSocket.SetOpt(KSOSendBuf, KSOLSocket, KSocketDefaultSendBufSize);        

      }

    iSocket.SetOpt(KSoTcpCork, KSolInetTcp, aValue);       

    }