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

// This contains ESock Test cases from section 5

// EPOC includes

// 

//

#include <e32base.h>

#include <in_sock.h>

#include <e32math.h>

//anonymous namespace

namespace

    {

    //once the aRefBoolFlag set to false it never gets set again...

    void setResultFlagBool(const TBool aFlag, TBool& aRefBoolFlag)

        {

        if (aRefBoolFlag)

            {

            aRefBoolFlag = aFlag;

            }

        }

    //once the aRefBoolFlag set to false it never gets set again...

     void setResultFlagInt(const TInt aFlag, TBool& aRefBoolFlag)

        {

        if (aRefBoolFlag)

            {

            if (aFlag != KErrNone)

                {

                    aRefBoolFlag = EFalse;

                }

            }

        }

        const TUint KMaxTestStepCounter = 5;

    }

// Test system includes

#include "EsockTestSection5.h"

// Test step SendData

// this test is used by the integration test scripts

// and is based on esock test 5.3

// This test step sends TCP and UDP (if enabled in ini file) data

const TDesC& CEsockSendData::GetTestName()

	{

	// store the name of this test case

	_LIT(ret,"SendData");

	return ret;

	}

CEsockSendData::~CEsockSendData()

	{

	}

enum TVerdict CEsockSendData::easyTestStepL()

	{

	TInetAddr addrRemote, addr;

	TRequestStatus stat(KErrNone);

	TInt connectTimer, Port;

	TBool UdpEnable;

	TInt testStepCounter = 0;

	TBool testStepResult;

	TBool socketConnected = EFalse;

	TBool udpBindOK = EFalse;

	do

	    {

	    testStepResult = ETrue;

    	// get ip address to connect to (on echo port)

    	TESTL(GetIpAddressFromConfig(_L("Test_Common"), _L("ipAddress"), addrRemote));

    	// get the port number

    	TESTL(GetIntFromConfig(_L("SendData"), _L("Port"), Port));

    	addrRemote.SetPort(Port);

    	// format for display 

    	TBuf<39> AddrBuf;

    	addrRemote.OutputWithScope(AddrBuf);

    	// check if UDP test enabled

    	TBool res = GetBoolFromConfig(_L("SendData"), _L("UdpEnable"), UdpEnable );

    	if (( res == EFalse ) || ( UdpEnable == EFalse ))

    		{

    		Logger().WriteFormat( _L("UDP test disabled"));

    		UdpEnable = EFalse;

    		}

    	// get a value for the connection guard timer 

    	TESTL(GetIntFromConfig(_L("SendData"), _L("connectTimer"), connectTimer));

    	// create timer to use as a guard timer

    	RTimer GuardTimer;

    	GuardTimer.CreateLocal();			// for this thread

    	TRequestStatus TimerStatus;			// status to monitor timer

    	TInt RepeatCount = 0;				// repeat counter 

    	TInt WaitTimeout = 5000000;

        if (!socketConnected)

            {

        	// try 5 times (sometimes it fails because of test network)

        	for(RepeatCount=0; RepeatCount <=5;  RepeatCount++)

        		{

        		// connect

        		(iEsockSuite->GetSocketHandle(1)).Connect(addrRemote, stat);

        		// start timer

        		GuardTimer.After(TimerStatus,connectTimer * 1000000); 

        		// wait for either timer or connect

        		User::WaitForRequest(stat, TimerStatus);

        		GuardTimer.Cancel();

        		User::WaitForAnyRequest();

        		// check connect result

        		if (stat.Int() == KRequestPending)

        			{

        			(iEsockSuite->GetSocketHandle(1)).CancelConnect();

        			Logger().WriteFormat( _L("Socket connect to %S port %d has not completed within %d seconds (connectTimer from ini) "), 

        				&AddrBuf,

        				Port,

        				connectTimer ); 

        			User::After(WaitTimeout);

        			continue; // next attempt

        			}

        		else if(stat.Int() != KErrNone)

        			{

        			Logger().WriteFormat( _L("Socket connect failed with error %d"), stat.Int());

        			if(RepeatCount <= 5)

        				{

        				Logger().WriteFormat( _L("Try to reconnect. RepeatCount %d"), RepeatCount);

        				User::After(WaitTimeout);

        				continue; // next attempt

        				}

        		    else

        		        {

        		        setResultFlagInt(stat.Int(), testStepResult);

        		        }

        			}

        		else

        		    {

        		    socketConnected = ETrue;

        		    }

        		break; // exit "for"

        		}

        	TESTEL((stat.Int()==KErrNone),stat.Int());

            }

    	// get number of packets and size of packets 

    	TInt PacketSize = 0;

    	TESTL(GetIntFromConfig(_L("SendData"), _L("PacketSize"), PacketSize));

    	TInt NoPackets = 0;

    	TESTL(GetIntFromConfig(_L("SendData"), _L("NoPackets"), NoPackets));

    	Logger().WriteFormat( _L("Sending to %S port %d TCP data %d packets of %d bytes = %d"), 

    		&AddrBuf, Port, NoPackets, PacketSize, NoPackets * PacketSize);

    	// set up data buffers

    	HBufC8 * writebuf = HBufC8::NewMaxLC( PacketSize );

    	HBufC8 * readbuf  = HBufC8::NewMaxLC( PacketSize );

    	TPtr8 ptrWritebuf = writebuf->Des();

    	TPtr8 ptrReadbuf = readbuf->Des();

    	// small buffer for data pattern to be sent

    	TBuf8<50> Data;

    	// send data (NoPackets packets of PacketSize bytes)

    	TInt recvCount = 0;

    	for (TInt i = 0; i < NoPackets; i++)

    		{

    		// initialise data

    		Data.Format(_L8("TCP-packet:%d helloworld"),i);

    		ptrWritebuf.Repeat( Data );

    		// write data 

    		GuardTimer.After(TimerStatus,connectTimer * 1000000); 

    		(iEsockSuite->GetSocketHandle(1)).Write(ptrWritebuf, stat);

    		User::WaitForRequest(stat, TimerStatus);

    		GuardTimer.Cancel();

    		User::WaitForAnyRequest();

    		if (stat == KRequestPending)

    			{

    			(iEsockSuite->GetSocketHandle(1)).CancelWrite();

    			Logger().WriteFormat(_L("Timeout on Tcp Send packet %d after %d seconds"), i, connectTimer);

    			setResultFlagInt(stat.Int(), testStepResult);

    			}

    		else if (stat != KErrNone)

    			{

    			Logger().WriteFormat(_L("Tcp Send packet %d failed with error %d"), i, stat.Int());

    			setResultFlagInt(stat.Int(), testStepResult);

    			}

    		TESTEL( stat == KErrNone, stat.Int() );

    		// read data (from echo server)

    		GuardTimer.After(TimerStatus,connectTimer * 1000000); 

    		(iEsockSuite->GetSocketHandle(1)).Read(ptrReadbuf, stat);

    		User::WaitForRequest(stat, TimerStatus);

    		GuardTimer.Cancel();

    		User::WaitForAnyRequest();

    		TESTE( stat == KErrNone, stat.Int() );

    		if (stat == KRequestPending)

    			{

    			(iEsockSuite->GetSocketHandle(1)).CancelRead();

    			Logger().WriteFormat(_L("Timeout on Tcp Recv packet %d after %d seconds"), i, connectTimer);

    			setResultFlagInt(stat.Int(), testStepResult);

    			}

    		else if (stat != KErrNone)

    			{

    			Logger().WriteFormat(_L("Tcp Recv packet %d failed with error %d"), i, stat.Int());

    			setResultFlagInt(stat.Int(), testStepResult);

    			}

    		else

    			{

    			// compare the data

    			TEST( ptrWritebuf.Compare( ptrReadbuf ) == 0);

    			recvCount+=ptrReadbuf.Length();

    			}

    		}

    	// check the total received.

    	TEST(recvCount==(NoPackets*PacketSize));

    	//** UDP STARTS HERE

    	// test UDP if required

    	if ( UdpEnable )

    		{

    		TESTL(GetIntFromConfig(_L("SendData"), _L("UdpPacketSize"), PacketSize));

    		// get the port number

    		// Can't send back on Echo port as it would enter an infinite loop

    		if (!GetIntFromConfig(_L("SendData"), _L("UdpPort"), Port))

    			// use default

    			Port = 9200;

    		Logger().WriteFormat( _L("Sending to %S Port %d UDP data %d packets of %d bytes = %d"), 

    			&AddrBuf, Port, NoPackets, PacketSize, NoPackets * PacketSize);

    		// and make some new buffers the correct size

    		HBufC8 * UdpWritebuf = HBufC8::NewMaxLC( PacketSize );

    		HBufC8 * UdpReadbuf  = HBufC8::NewMaxLC( PacketSize );

    		TPtr8 ptrUdpWritebuf = UdpWritebuf->Des();

    		TPtr8 ptrUdpReadbuf = UdpReadbuf->Des();

    		// open a udp socket

    		RSocket UdpSock;

    		TInt nRet = UdpSock.Open(iEsockSuite->iSocketServer, KAfInet, KSockDatagram, KProtocolInetUdp);

    		if (nRet != KErrNone)

    			{

    			Logger().WriteFormat(_L("Failed to open socket: return value = <%d>"), nRet);

    			setResultFlagBool(EFail, testStepResult);

    			//return EFail;  

    			}

    		if (testStepResult)

    		    {

            		addr.SetPort(Port);

            		// Bind

            		TInt result = UdpSock.SetOpt(KSoUdpSynchronousSend, KSolInetUdp, 1 );

            		UdpSock.SetOpt(KSoReuseAddr, KSolInetIp, KMaxTestStepCounter);

            		TESTEL( KErrNone ==result,result);

            		if (!udpBindOK)

            		    {

                		result = UdpSock.Bind(addr);

                		TESTEL( KErrNone ==result,result);

                		udpBindOK = ETrue;

            		    }

            		// write & read back the data

            		recvCount = 0;

            		for (TInt j = 0; j < NoPackets; j++)

            			{

            			// initialise data

            			Data.Format(_L8("UDP-packet:%d helloworld"),j);

            			ptrUdpWritebuf.Repeat( Data );

            			// write data 

            			GuardTimer.After(TimerStatus,connectTimer * 1000000); 

            			UdpSock.SendTo(ptrUdpWritebuf, addrRemote, 0, stat);

            			User::WaitForRequest(stat, TimerStatus);

            			GuardTimer.Cancel();

            			User::WaitForAnyRequest();

            			if (stat == KRequestPending)

            				{

            				UdpSock.CancelSend();

            				Logger().WriteFormat(_L("Timeout on Udp Send packet %d after %d seconds"), j, connectTimer);

            				setResultFlagInt(stat.Int(), testStepResult);

            				}

            			else if (stat != KErrNone)

            				{

            				Logger().WriteFormat(_L("Udp Send packet %d failed with error %d"), j, stat.Int());

            				setResultFlagInt(stat.Int(), testStepResult);

            				}

            			// read data (from echo server)

            			GuardTimer.After(TimerStatus,connectTimer * 1000000); 

            			UdpSock.RecvFrom(ptrUdpReadbuf, addr, 0, stat);

            			User::WaitForRequest(stat, TimerStatus);

            			GuardTimer.Cancel();

            			User::WaitForAnyRequest();

            			if (stat == KRequestPending)

            				{

            				UdpSock.CancelRecv();

            				Logger().WriteFormat(_L("Timeout on Udp Recv packet %d after %d seconds"), j, connectTimer);

            				setResultFlagInt(stat.Int(), testStepResult);

            				}

            			else if (stat.Int() != KErrNone) 

            				{

            				Logger().WriteFormat(_L("Udp Recv packet %d failed with error %d"), j, stat.Int());

            				setResultFlagInt(stat.Int(), testStepResult);

            				}

            			//TEST( stat == KErrNone );  ** We could have packet loss with UDP in real life

            			else

            				{

            				// compare the data

            				TEST( ptrUdpWritebuf.Compare( ptrUdpReadbuf ) == 0);

            				recvCount += ptrUdpReadbuf.Length();

            				}

            			}

            		Logger().WriteFormat(_L("Udp data sent %d bytes, data received %d bytes"),(NoPackets*PacketSize),recvCount);

            		// check the total received (95 per cent is allowable for us)

            		//TESTE( recvCount >= (NoPackets*PacketSize), (NoPackets*PacketSize) );

            		// close the UDP socket

                    UdpSock.Close();

    		    }

    		// get rid of the old buffers 

    		CleanupStack::PopAndDestroy();	//	UdpWritebuf;

    		CleanupStack::PopAndDestroy();	//	UdpReadbuf;

    		}

    	CleanupStack::PopAndDestroy();	//	writebuf;

    	CleanupStack::PopAndDestroy();	//	readbuf;

    	if (!testStepResult)

    	    {

    	    Logger().WriteFormat(_L("TestStep failed %d time(s) from %d. Trying further..."), testStepCounter, KMaxTestStepCounter);

    	    ++testStepCounter;

    	    }

	    }

	    while ( (testStepCounter < KMaxTestStepCounter) &&

	          (!testStepResult) ); //big while...

    if ( testStepResult )

        {

	    return EPass;

        }

    else

        {

        return EFail;

        }

	}

// Test step SendTcpData

// this test is used by the integration test scripts

// and is based on esock test 5.3

// This test step sends TCP data only

const TDesC& CEsockSendTcpData::GetTestName()

	{

	// store the name of this test case

	_LIT(ret,"SendTcpData");

	return ret;

	}

CEsockSendTcpData::~CEsockSendTcpData()

	{

	}

enum TVerdict CEsockSendTcpData::easyTestStepL()

	{

	TInetAddr addrRemote;

	TRequestStatus stat;

	TInt connectTimer, Port;

	// get ip address to connect to (on echo port)

	TESTL(GetIpAddressFromConfig(_L("Test_Common"), _L("ipAddress"), addrRemote));

	// get the port number

	TESTL(GetIntFromConfig(_L("SendData"), _L("Port"), Port));

	addrRemote.SetPort(Port);

	// format for display 

	TBuf<39> AddrBuf;

	addrRemote.OutputWithScope(AddrBuf);

	// get a value for the connection guard timer 

	TESTL(GetIntFromConfig(_L("SendData"), _L("connectTimer"), connectTimer));

	// create timer to use as a guard timer

	RTimer GuardTimer;

	GuardTimer.CreateLocal();			// for this thread

	TRequestStatus TimerStatus;			// status to monitor timer

    // connect

    (iEsockSuite->GetSocketHandle(1)).Connect(addrRemote, stat);

	// start timer

	GuardTimer.After(TimerStatus,connectTimer * 1000000); 

	// wait for either timer or connect

	User::WaitForRequest(stat, TimerStatus);

	GuardTimer.Cancel();

	User::WaitForAnyRequest();

	// check connect result

	if (stat!=KErrNone)

		{

		// either connect failed or timer expired

		// because it may be the timer cancel connect

		(iEsockSuite->GetSocketHandle(1)).CancelConnect();

		Logger().WriteFormat( _L("Socket connect has not completed within %d seconds (connectTimer from ini) "), connectTimer ); 

		// report this failure and leave

		TESTL((stat==KErrNone));

		}

	// get number of packets and size of packets 

	TInt PacketSize = 0;

	TESTL(GetIntFromConfig(_L("SendData"), _L("PacketSize"), PacketSize));

	TInt NoPackets = 0;

	TESTL(GetIntFromConfig(_L("SendData"), _L("NoPackets"), NoPackets));

	Logger().WriteFormat( _L("Sending to %S TCP data %d packets of %d bytes = %d"), 

		&AddrBuf, NoPackets, PacketSize, NoPackets * PacketSize);

	// set up data buffers

	HBufC8 * writebuf = HBufC8::NewMaxLC( PacketSize );

	HBufC8 * readbuf  = HBufC8::NewMaxLC( PacketSize );

	TPtr8 ptrWritebuf = writebuf->Des();

	TPtr8 ptrReadbuf = readbuf->Des();

	// initialise data

	TBuf8<50> Data;

	// send data (NoPackets packets of PacketSize bytes)

	TInt recvCount = 0;

	for (TInt i = 0; i < NoPackets; i++)

		{

		// initialise data

		Data.Format(_L8("TCP-packet:%d helloworld"),i);

		ptrWritebuf.Repeat( Data );

		// write data 

		GuardTimer.After(TimerStatus,connectTimer * 1000000); 

		(iEsockSuite->GetSocketHandle(1)).Write(ptrWritebuf, stat);

		User::WaitForRequest(stat, TimerStatus);

		GuardTimer.Cancel();

		User::WaitForAnyRequest();

		if (stat != KErrNone) Logger().WriteFormat(_L("Timeout on Tcp Send after %d seconds"), connectTimer);

		TESTL( stat == KErrNone );

		// read data (from echo server)

		GuardTimer.After(TimerStatus,connectTimer * 1000000); 

		(iEsockSuite->GetSocketHandle(1)).Read(ptrReadbuf, stat);

		User::WaitForRequest(stat, TimerStatus);

		GuardTimer.Cancel();

		User::WaitForAnyRequest();

		if (stat != KErrNone) Logger().WriteFormat(_L("Timeout on Tcp Recv after %d seconds"), connectTimer);

		TESTL( stat == KErrNone );

		// compare the data

		TESTL( ptrWritebuf.Compare( ptrReadbuf ) == 0);

		recvCount+=ptrReadbuf.Length();

		}

	// check the total received.

	TESTL(recvCount==(NoPackets*PacketSize));

	CleanupStack::PopAndDestroy();	//	writebuf;

	CleanupStack::PopAndDestroy();	//	readbuf;

	return EPass;

	}

// Open Udp Socket

const TDesC& CEsockOpenUdpSocket::GetTestName()

	{

	// store the name of this test case

	_LIT(ret,"OpenUdpSocket");

	return ret;

	}

CEsockOpenUdpSocket::~CEsockOpenUdpSocket()

	{

	}

enum TVerdict CEsockOpenUdpSocket::easyTestStepL()

	{

	TInt nRet;

	TInt nSockets; // number of sockets to open