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

// EPOC includes
#include <e32base.h>
#include <in_sock.h>

// Test system includes
#include "EsockTestSection3.h"

static const TInt KMicrosecondsPerSecond = 1000000;

// Test step 3.1
const TDesC& CEsockTest3_1::GetTestName()
	{
	// store the name of this test case
	_LIT(ret,"Test3.1");
	
	return ret;
	}

CEsockTest3_1::~CEsockTest3_1()
	{
	}

enum TVerdict CEsockTest3_1::easyTestStepPreambleL()
	{
	if (KErrNone != OpenMinSockets(_L("Test_3.1"),KAfInet, KSockStream, KProtocolInetTcp))
		{
		return EFail;
		}

	return EPass;
	}

enum TVerdict CEsockTest3_1::easyTestStepL()
	{
	// Test 1.x Open a socket
	
	// Set the address to loopback and the port to 0, IPv4 or IPv6 address
	TInetAddr addr;
	TESTL(GetIpAddressFromConfig(SectionName(_L("Test_3.1")), _L("ipAddress"), addr));
	
	addr.SetPort(0);
	
	TRequestStatus stat;
	for (TInt i = 1; i <= iEsockSuite->GetSocketListCount(); i++)
		{
		// Connect to the loopback address on port 0
		iEsockSuite->GetSocketHandle(i).Connect(addr, stat);
		User::WaitForRequest(stat);
		TESTEL(stat==KErrGeneral, stat.Int());
		}
	
	// Test 1.15 Close socket
	
	return EPass;
	}


// Test step 3.2
const TDesC& CEsockTest3_2::GetTestName()
	{
	// store the name of this test case
	_LIT(ret,"Test3.2");
	
	return ret;
	}

CEsockTest3_2::~CEsockTest3_2()
	{
	}

enum TVerdict CEsockTest3_2::easyTestStepPreambleL()
	{
	if (KErrNone != OpenMinSockets(_L("Test_3.2"),KAfInet, KSockStream, KProtocolInetTcp))
		{
		return EFail;
		}

	return EPass;
	}

enum TVerdict CEsockTest3_2::easyTestStepL()
	{
	// Test 1.x Open a socket
	
	// Set the address to loopback IPv4 or IPv6 address
	TInetAddr addr;
	TESTL(GetIpAddressFromConfig(SectionName(_L("Test_3.2")), _L("ipAddress"), addr));
	
	// set the port numnber
	addr.SetPort(65537);
	
	TRequestStatus stat;
	for (TInt i = 1; i <= iEsockSuite->GetSocketListCount(); i++)
		{
		// Connect to the loopback address on port 65537
		iEsockSuite->GetSocketHandle(i).Connect(addr, stat);
		User::WaitForRequest(stat);
		TESTEL(stat==KErrGeneral, stat.Int());
		}
	
	// Test 1.15 Close socket
	
	return EPass;
	}


// Test step 3.3
const TDesC& CEsockTest3_3::GetTestName()
	{
	// store the name of this test case
	_LIT(ret,"Test3.3");
	
	return ret;
	}

CEsockTest3_3::~CEsockTest3_3()
	{
	}

enum TVerdict CEsockTest3_3::easyTestStepPreambleL()
	{
	if (KErrNone != OpenMinSockets(_L("Test_3.3"),KAfInet, KSockStream, KProtocolInetTcp))
		{
		return EFail;
		}

	return EPass;
	}

enum TVerdict CEsockTest3_3::easyTestStepL()
	{
	// Test 1.x Open a socket
	
	// Set the address to loopback, IPv4 or IPv6 address
	TInetAddr addr;
	TESTL(GetIpAddressFromConfig(SectionName(_L("Test_3.3")), _L("ipAddress"), addr));
	
	// Set to a non listening port
	TInt port;
	TESTL(GetIntFromConfig(_L("Test_3.3"), _L("port"), port));
	
	addr.SetPort(port);
	
	TRequestStatus stat;
	for (TInt i = 1; i <= iEsockSuite->GetSocketListCount(); i++)
		{
		// Connect to the loopback address on the non listening port n 
		iEsockSuite->GetSocketHandle(i).Connect(addr, stat);
		User::WaitForRequest(stat);
		TESTEL(stat==KErrCouldNotConnect, stat.Int());
		}
	
	// Test 1.15 Close socket
	
	return EPass;
	}

// Test step 3.4
const TDesC& CEsockTest3_4::GetTestName()
	{
	// store the name of this test case
	_LIT(ret,"Test3.4");
	
	return ret;
	}

CEsockTest3_4::~CEsockTest3_4()
	{
	}

enum TVerdict CEsockTest3_4::easyTestStepL()
	{
	// Test 1.x Open a socket
	
	// Set the address to Remote host, IPv4 or IPv6 address
	TInetAddr addr;
	TESTL(GetIpAddressFromConfig(SectionName(_L("Test_3.4")), _L("ipAddress"), addr));
	
	// Set to non listening port
	TInt port;
	TESTL(GetIntFromConfig(_L("Test_3.4"), _L("port"), port));
	addr.SetPort(port);
	
	TRequestStatus stat;
	for (TInt i = 1; i <= iEsockSuite->GetSocketListCount(); i++)
		{
		// Connect to the remote host address on the non listening port n 
		iEsockSuite->GetSocketHandle(i).Connect(addr, stat);
		User::WaitForRequest(stat);
		TESTEL(stat==KErrCouldNotConnect, stat.Int());
		}
	
	// Test 1.15 Close socket
	
	return EPass;
	}

// Test step 3.5
const TDesC& CEsockTest3_5::GetTestName()
	{
	// store the name of this test case
	_LIT(ret,"Test3.5");
	
	return ret;
	}

CEsockTest3_5::~CEsockTest3_5()
	{
	}

enum TVerdict CEsockTest3_5::easyTestStepPreambleL()
	{
	CloseSockets();
	if (KErrNone != OpenMinSockets(_L("Test_3.5"),KAfInet, KSockStream, KProtocolInetTcp))
		{
		return EFail;
		}

	return EPass;
	}

enum TVerdict CEsockTest3_5::easyTestStepL()
	{
	TRequestStatus stat;
	TInetAddr addr;
	TInt port;
	
	// Test 1.x Open a socket
	
	// Set the address to Remote host, IPv4 or IPv6 address
	TESTL(GetIpAddressFromConfig(_L("Test_Common"), _L("ipAddress"), addr));
	
	// Set to echo port
	TESTL(GetIntFromConfig(SectionName(_L("Test_3.5")), _L("port"), port));
	addr.SetPort(port);
	
	// Connect to the remote host address on the echo port 
	iEsockSuite->GetSocketHandle(1).Connect(addr, stat);
	iEsockSuite->GetSocketHandle(1).CancelConnect();
	User::WaitForRequest(stat);
	TESTEL(stat==KErrCancel, stat.Int());
	
	// Test 1.15 Close socket
	
	return EPass;
	}


// Test step 3.6
const TDesC& CEsockTest3_6::GetTestName()
	{
	// store the name of this test case
	_LIT(ret,"Test3.6");
	
	return ret;
	}

CEsockTest3_6::~CEsockTest3_6()
	{
	}


enum TVerdict CEsockTest3_6::subTest2(RTimer	&aWatchdog,
									  RSocket	&aSocket,
									  TDes8		&aBuffer,
									  TInt		aExpectedStatus)
	{
	TBool finished = EFalse;
	TInt i = 1;
	TRequestStatus sendStatus;
	TRequestStatus recvStatus;
	TRequestStatus timerStatus;
	while ((i <= iNumTries) && !finished)
		{
		aWatchdog.After(timerStatus, 7 * KMicrosecondsPerSecond);
		Logger().Write(_L("Issuing recv..."));
		aSocket.Recv(aBuffer, 0, recvStatus);
		Logger().Write(_L("Sending..."));
        aSocket.SendTo(aBuffer, iAddr, 0, sendStatus);

    	User::WaitForRequest(timerStatus, recvStatus);
        aWatchdog.Cancel();
        aSocket.CancelRecv();
    	User::WaitForAnyRequest();
    	
    	finished = recvStatus == aExpectedStatus && 
        (timerStatus == KRequestPending || timerStatus == KErrNone || timerStatus == KErrCancel);

    	if (!finished)
			{
			User::After(1 * KMicrosecondsPerSecond);
			i++;
			}
		}

	if (!finished)
		{
		return EFail;
		}
	return EPass;
	}


enum TVerdict CEsockTest3_6::easyTestStepPreambleL()
	{
    // Get the ini file data.
	_LIT(KIniTestKey,    "Test_3.6");

	// Set the address to Remote host, IPv4 or IPv6 address
	_LIT(KIniAddressKey, "ipAddress");
	TESTL(GetIpAddressFromConfig(KIniTestKey(), KIniAddressKey(), iAddr));
	
	// Set to non-existent port
	_LIT(KIniErrPortKey, "errorPort");
	TESTL(GetIntFromConfig(KIniTestKey(), KIniErrPortKey, iErrorPort));
	iAddr.SetPort(iErrorPort);

	// Get the valid port
	_LIT(KIniValPortKey, "validPort");
	TESTL(GetIntFromConfig(KIniTestKey(), KIniValPortKey, iValidPort));
	
	// Get the number of times to try the sub-test before giving up.
	_LIT(KIniValTries,   "validTries");
	TESTL(GetIntFromConfig(KIniTestKey(), KIniValTries, iNumTries));

	if (KErrNone != OpenMinSockets(KIniTestKey(), KAfInet, KSockDatagram, KProtocolInetUdp))
		{
		return EFail;
		}
	
   	// Enable notification of ICMP errors.
	RSocket &sock = iEsockSuite->GetSocketHandle(1);
	TInt ret = sock.SetOpt(KSoUdpReceiveICMPError, KSolInetUdp, 1);
	if (KErrNone != ret)
		{
		Logger().WriteFormat(_L("SetOpt to receive ICMP errors returns %d"), ret);
		return EFail;
		}

	return EPass;
	}


enum TVerdict CEsockTest3_6::easyTestStepL()
	{
	TESTL(EPass == TestStepResult());
	TVerdict verdict = EPass;

    const TInt KStringMaxLen = 40;
    TBuf<KStringMaxLen> addressBuf;
	iAddr.Output(addressBuf);
	Logger().Write(_L(""));
	Logger().WriteFormat(_L("Test3.6 - Testing address = %S, errorPort = %d, validPort = %d"), &addressBuf, iErrorPort, iValidPort);

	RTimer         watchdog;
	CleanupClosePushL(watchdog);
	watchdog.CreateLocal();
	
    const TInt KBufferLength = 10;  
	TBuf8<KBufferLength> buffer;
	buffer.SetMax();
	buffer.FillZ();

	RSocket &sock = iEsockSuite->GetSocketHandle(1);

    // Part 1. Invalid port#, no pending receive
    //
	// First Recv returns -34, second one will pause.
    
	Logger().WriteFormat(_L("Test3.6 Part 1 - Address = %S, errorPort = %d"), &addressBuf, iErrorPort);
	
	TBool finished = EFalse;
    TInt i = 1;
	TRequestStatus status;
	TRequestStatus recvStatus;
	TRequestStatus timerStatus;
	TInt ret;
    while ((i <= iNumTries)  && !finished)
		{
    	status = recvStatus = timerStatus = KRequestPending;

    	sock.SendTo(buffer, iAddr, 0, status);
    	User::WaitForRequest(status);
    	ret = status.Int();
    	Logger().WriteFormat(_L("Send returned %d"), ret);
    	TESTL(KErrNone == ret);
    	
    	// Delay to allow the connection to come up.
    	User::After(5 * KMicrosecondsPerSecond);

		watchdog.After(timerStatus, 5 * KMicrosecondsPerSecond);
		sock.Recv(buffer, 0, recvStatus);
		User::WaitForRequest(timerStatus, recvStatus);
		watchdog.Cancel();
		sock.CancelRecv();
		User::WaitForAnyRequest();
		ret = recvStatus.Int();
		Logger().WriteFormat(_L("1st recv returned %d"), ret);
		
        // This Recv should complete with -34.
    	finished = recvStatus == KErrCouldNotConnect && 
    		(timerStatus == KRequestPending || timerStatus == KErrNone || timerStatus == KErrCancel);

        if (finished)
        	{
        	recvStatus = timerStatus = KRequestPending;

    		watchdog.After(timerStatus, 5 * KMicrosecondsPerSecond);
    		sock.Recv(buffer, 0, recvStatus);
    		User::WaitForRequest(timerStatus, recvStatus);
    		watchdog.Cancel();
    		sock.CancelRecv();
    		User::WaitForAnyRequest();
    		ret = recvStatus.Int();
    		Logger().WriteFormat(_L("2nd recv returned %d"), ret);
    		
    		// This time we expect the watchdog to fire, an immediate Recv completion here means the
    		// test has failed (error not cleared down).
    		
    		finished = timerStatus == KErrNone && 
    			(recvStatus == KRequestPending || recvStatus == KErrNone || recvStatus == KErrCancel);
        	}

        if (!finished)
        	{
        	User::After(1 * KMicrosecondsPerSecond);
        	i++;
        	}
		}

    if (!finished)
    	{
    	verdict = EFail;
    	}
    TESTL(EPass == verdict);

    // Part 2. Invalid port#, with pending receive; should complete with -34
	Logger().WriteFormat(_L("Test3.6 Part 2 - Address = %S, errorPort = %d"), &addressBuf, iErrorPort);
	TVerdict subTestVerdict = subTest2(watchdog, sock, buffer, KErrCouldNotConnect);
	if (verdict == EPass)
		{
		verdict = subTestVerdict;
		}
    TESTL(EPass == verdict);

    // Part 3. Valid port#, with pending receive; should complete with KErrNone
	Logger().WriteFormat(_L("Test3.6 Part 3 - Address = %S, validPort = %d"), &addressBuf, iValidPort);
	iAddr.SetPort(iValidPort);
	subTestVerdict = subTest2(watchdog, sock, buffer, KErrNone);
	if (verdict == EPass)
		{
		verdict = subTestVerdict;
		}
    TESTL(EPass == verdict);

	// Test 1.15 Close socket
	CleanupStack::PopAndDestroy(1, &watchdog);
    return EPass;
	}