// Copyright (c) 2007-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 "chttpdownload.h"
#include "cctsytestlogging.h"
#include <cs_subconparams.h>
#include <networking/qos3gpp_subconparams.h>

// Declare a static, single, server socket session to be used with all active downloads
RSocketServ CHTTPDownload::iSocketServer;
// Declare a static, counter fot active downloads, to know when to open and close the server socket session
TInt CHTTPDownload::iActiveDownload = 0;

_LIT(KDefaultHost, "www.smartphoneshow.com");
_LIT(KDefaultPage, "/newt/l/downloads/symbian/presentations/Day1Pres1_NigelClifford_Symbian_keynote.pdf");
const TInt KDefaultPort = 80;
const TInt KBufferSize = 4096;

CHTTPDownload::CHTTPDownload(CCTSYIntegrationTestSuiteStepBase *aTestStep): iOpenSocketServer(EFalse),
																			iTestStep(aTestStep),
                                                                            iSocketOpen(EFalse),
                                                                            iHostResolverOpen(EFalse),
                                                                            iConnectionOpen(EFalse),
                                                                            iSubConnectionOpen(EFalse),
                                                                            iWebPageOpen(EFalse)
/*
 * Constructor
 * 
 * @param aTestStep a pointer to the test step object
 */
                               
    {
    }

CHTTPDownload::~CHTTPDownload()
/*
 * Destructor
 * 
 */
    {
    StopDownload();
    if(iWebPageOpen)
    	{
    	iWebPage.Close();
    	iWebPageOpen=EFalse;
    	}
    if(iOpenSocketServer)
    	{
    	iActiveDownload--;
    	TEST_FRAMEWORK_LOG2(_L("iActiveDownload changed  iActiveDownload = %d"),iActiveDownload);    
    	if(iActiveDownload==0)
    		{
    		iSocketServer.Close();
    		TEST_FRAMEWORK_LOG2(_L("iSocketServer Closed iActiveDownload = %d"),iActiveDownload);    
    		}
    	}
    }


TBool CHTTPDownload::StartDownloadL()
/*
 * Start download an http page with default parameters
 * 
 *
 * @return ETrue if the download started
 */
    {
    iHost = &KDefaultHost;
    iPage = &KDefaultPage;
    iPort = KDefaultPort;
    iUseSubConnection = EFalse;
    if(InitSocket()!=KErrNone)
    	{
    	Close();
    	return EFalse;
    	}
	return ETrue;
    }

TBool CHTTPDownload::StartDownloadL(const TDesC &aHost, const TDesC &aPage)
/*
 * Start download a requested http page with default port 
 * 
 * @param aHost a descriptor of the web host (i.e. www.symbian.com)
 * @param aPage The actual page (i.e. /symbianos/index.html)
 *
 * @return ETrue if the download started
 */
    {
    iHost = &aHost;
    iPage = &aPage;
    iPort = KDefaultPort;
    iUseSubConnection = EFalse;
    if(InitSocket()!=KErrNone)
    	{
    	Close();
    	return EFalse;
    	}
	return ETrue;
    }

TBool CHTTPDownload::StartDownloadL(const TDesC &aHost, const TDesC &aPage,TInt aPort)
/*
 * Start download a requested http page with a requested port
 * 
 * @param aHost a descriptor of the web host (i.e. www.symbian.com)
 * @param aPage The actual page (i.e. /symbianos/index.html)
 * @param aPort the protocol port (80 for HTTP)
 *
 * @return ETrue if the download started
 */
    {
    iHost = &aHost;
    iPage = &aPage;
    iPort = aPort;
    iUseSubConnection = EFalse;
    if(InitSocket()!=KErrNone)
    	{
    	Close();
    	return EFalse;
    	}
	return ETrue;
    }


TBool CHTTPDownload::StartDownloadWithSecondaryContextL(const TDesC &aHost, const TDesC &aPage, TInt aPort)
/*
 * Start download a requested http page with a requested port using a sub connection
 * 
 * @param aHost a descriptor of the web host (i.e. www.symbian.com)
 * @param aPage The actual page (i.e. /symbianos/index.html)
 * @param aPort the protocol port (80 for HTTP)
 *
 * @return ETrue if the download started
 */
	{
    iHost = &aHost;
    iPage = &aPage;
    iPort = aPort;
    iUseSubConnection = ETrue;
    if(InitSocket()!=KErrNone)
    	{
    	Close();
    	return EFalse;
    	}
	return ETrue;
	}

TInt CHTTPDownload::GetByteCount()
/*
 * Return the number of bytes downloaded so far
 * 
 * @return The number of bytes downloaded so far
 */
    {
    VerifyDownloading();
    return iByteCount;
    }

void CHTTPDownload::StopDownload()
/*
 * Stop a active download; waits for the current socket read to complete, rather than force an immediate terminate.
 * 
 */
    {
    if(iSocketOpen)
    	{
    	iSocket.CancelRecv();
    	}
    Close();
    }

TBool CHTTPDownload::VerifyDownloading()
/*
 * Return ETrue if the download is still active
 * 
 * @return ETrue if the download is still active
 */
    {
    if(iSocketOpen)
    	{
    	if(iStatus.Int()==KRequestPending)
    		{
    		return ETrue;
    		}
    	else if (iStatus.Int()==KErrNone)
    		{
    		iByteCount += iWebPage.Length();

            // Logging the first 180 characters of the text...
            TBuf<200> temp;
            temp.Copy(iWebPage.Left(180));
            TEST_FRAMEWORK_LOG3(_L("Got %S... in %d chars"),&temp,iWebPage.Length());
    		
    		iSocket.RecvOneOrMore(iWebPage,0,iStatus,iHttpHeaderLen);
    		return  ETrue;
    		}
    	else 
    		{
    		iByteCount += iWebPage.Length();
    		Close();
    		return EFalse;
    		}
    		
    	}
    else
    	{
    	return EFalse;
    	}
    }




TInt CHTTPDownload::InitSocket()
/*
 * Initilize a socket 
 * 
 * @return KErrNone if it had no problem, or the specific system error code that cause the problem.
 */
	{
	TInt status;
    if(!iWebPageOpen)
    	{
    	status = iWebPage.Create(KBufferSize);
    	if(status !=KErrNone)
    		{
    		TEST_FRAMEWORK_LOG2(_L("RRBuf8::Create Could not create a buffer, got %d"),status);
    		return status;
    		}
    	iWebPageOpen = ETrue;
    	}
    if(!iOpenSocketServer)
    	{
    	if(iActiveDownload==0)
    		{
    		status = iSocketServer.Connect();
    		TEST_FRAMEWORK_LOG3(_L("iSocketServer.Connected, status = %d, iActiveDownload = %d"),status, iActiveDownload);
    		if(status !=KErrNone)
    			{
    			TEST_FRAMEWORK_LOG2(_L("RConnection::Open Could not open a socket servet session, got %d"),status);
    			return status;
    			}
    		}
    	iActiveDownload++;
    	iOpenSocketServer = ETrue;
    	}
    TEST_FRAMEWORK_LOG2(_L("iActiveDownload changed  iActiveDownload = %d"),iActiveDownload);
    if(iUseSubConnection)
    	{
    	if(!iConnectionOpen)
    		{
    		status = iConnection.Open(iSocketServer);
    		if(status != KErrNone) 
    			{
    			TEST_FRAMEWORK_LOG2(_L("RConnection::Open Could not open a connection, got %d"),status);
    			return status;
    			}
    		iConnectionOpen=ETrue;
    		}
        status=iConnection.Start();
        if(status != KErrNone)
            {
            TEST_FRAMEWORK_LOG2(_L("RConnection::Start Could not start a connection, got %d"),status);
            return status;
            }
        if(!iSubConnectionOpen)
            {
            status = iSubConnection.Open(iSocketServer,RSubConnection::ECreateNew,iConnection);
            if(status != KErrNone)
                {
                TEST_FRAMEWORK_LOG2(_L("RSubConnection::Open Could not open a sub connection, got %d"),status);
                return status;
                }
            iSubConnectionOpen=ETrue;
            }
        }
    if(!iSocketOpen)
    	{
    	if(iSubConnectionOpen)
    		{
        	status = iSocket.Open(iSocketServer, KAfInet, KSockStream, KProtocolInetTcp,iSubConnection);
        	}
        else
        	{
        	status = iSocket.Open(iSocketServer, KAfInet, KSockStream, KProtocolInetTcp);
        	}
        if(status != KErrNone)
            {
            TEST_FRAMEWORK_LOG2(_L("RSocket::Open Could not open a socket, got %d"),status);
            return status;
            }
        iSocketOpen=ETrue;
        }
    if(!iHostResolverOpen)
        {
        status = iHostResolver.Open(iSocketServer, KAfInet, KProtocolInetUdp);
        if(status != KErrNone)
            {
            TEST_FRAMEWORK_LOG2(_L("RHostResolver::Open Could not open an host resolver, got %d"),status);
            return status;
            }
        iHostResolverOpen=ETrue;
        }
	iHostResolver.GetByName(*iHost, iHostNameEntry, iStatus);
	User::WaitForRequest(iStatus);
	if(iStatus.Int()!=KErrNone)
		{
        TEST_FRAMEWORK_LOG2(_L("RHostResolver::GetByName Could get host by name, got %d"),iStatus.Int());
        return iStatus.Int(); 
		}
	return Connect();
	}

TInt CHTTPDownload::Connect()
/*
 * Connect to the host 
 * 
 * @return KErrNone, if it had no problem, or the specific system error code that cause the problem.
 */
	{
	TInetAddr addr(iHostNameEntry().iAddr);
	addr.SetPort(iPort);
	
	TCoreEtelRequestStatus<RSocket> connectStatus(iSocket, &RSocket::CancelConnect);
	iSocket.Connect(addr, connectStatus);
	iTestStep->WaitForRequestWithTimeOut(connectStatus,ETimeLong);
	if(connectStatus.Int()!=KErrNone)
		{
        TEST_FRAMEWORK_LOG2(_L("RSocket::Connect Could not connect to the host, got %d"),connectStatus.Int());
        return connectStatus.Int();
		}
	return Get();
	}

TInt CHTTPDownload::Get()
/*
 * sends the GET request to the server 
 * 
 * @return KErrNone, if it had no problem, or the specific system error code that cause the problem.
 */
	{
    TBuf8<200> httpGetLine;
    TBuf8<200> host8;
    TBuf8<200> page8;
    host8.Copy(*iHost);
    page8.Copy(*iPage);
    httpGetLine.Format(_L8("GET %S HTTP/1.0\r\nHost: %S\r\n\r\n"),&page8,&host8);

    TCoreEtelRequestStatus<RSocket> getStatus(iSocket, &RSocket::CancelWrite);
    iSocket.Write(httpGetLine, getStatus);
	iTestStep->WaitForRequestWithTimeOut(getStatus,ETimeLong);
	if(getStatus.Int()!=KErrNone)
		{
        TEST_FRAMEWORK_LOG2(_L("RSocket::Write Could not write to the host, got %d"),getStatus.Int());
        return getStatus.Int();
		}
	
    iByteCount = 0;
	iSocket.RecvOneOrMore(iWebPage,0,iStatus,iHttpHeaderLen);
	if(iStatus==KRequestPending)
		{
		return KErrNone;
		}
    return iStatus.Int();
	}



void CHTTPDownload::Close()
/*
 * Close everything... 
 * 
 */
	{
    if(iHostResolverOpen)
        {
        iHostResolver.Close();
        iHostResolverOpen=EFalse;
        }
    if(iSocketOpen)
    	{
        iSocket.Close();
        iSocketOpen=EFalse;
    	}
    if(iSubConnectionOpen)
    	{
    	iSubConnection.Close();
    	iSubConnectionOpen = EFalse;
    	}
    if(iConnectionOpen)
    	{
    	iConnection.Close();
    	iConnectionOpen = EFalse;
    	}
	}

