// Copyright (c) 2006-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:

//

/**

 @file Sockets.TestSteps.cpp

*/

#include "Sockets.TestSteps.h"

#include "SocketServer.TestSteps.h"

#include "Connections.TestSteps.h"

#include "SubConnections.TestSteps.h"

#include <utf.h>

// Open Socket

//------------

COpenRSocketStep::COpenRSocketStep(CCEsockTestBase*& aEsockTest)

:   CTe_EsockStepBase(aEsockTest)

	{

	SetTestStepName(KOpenRSocketStep);

	}

TInt COpenRSocketStep::ConfigureFromIni()

	{

	iParams.Reset();

	// Read in appropriate fields

	if((GetStringFromConfig(iSection, KTe_SocketName, iParams.iSocketName) != 1)

		|| (iParams.iSocketName.Length() == 0))

		{

		INFO_PRINTF1(_L("Couldn't find appropriate field in config file"));

		return KErrNotFound;

		}

    if (GetStringFromConfig(iSection,KTe_SocketServName,iParams.iSockServName)!=1)

        {

        INFO_PRINTF2(_L("%S: Socket server name missing."),&iParams.iSocketName);

        return KErrNotFound;

        }

    if ((GetStringFromConfig(iSection,KTe_ConnectionName,iParams.iConnectionName)==1)

    	&& iParams.iConnectionName.Length()>0)

        {

        INFO_PRINTF2(_L("%S: connection name found - opening with RConnection."),&iParams.iSocketName);

        }

    else if ((GetStringFromConfig(iSection,KTe_SubConnectionName,iParams.iSubconnectionName)==1)

    	&& iParams.iSubconnectionName.Length()>0)

        {

        INFO_PRINTF2(_L("%S: subconnection name found - opening with RSubConnection."),&iParams.iSocketName);

        }

    else

        {

        INFO_PRINTF2(_L("%S: connection or subconnection names missing - opening implicit."),&iParams.iSocketName);

        }

	if (GetStringFromConfig(iSection,KTe_ProtocolName,iParams.iProtocolName)==1)

        {

        INFO_PRINTF2(_L("%S: Protocol name found."),&iParams.iProtocolName);

        }

	else

		{

	    TPtrC protocol;

	    if (GetStringFromConfig(iSection,KTe_Protocol,protocol)!=1)

        {

	        INFO_PRINTF2(_L("%S: Protocol missing."),&iParams.iSocketName);

        	return KErrNotFound;

        }

	    if (protocol.Compare(KTe_TCPName)==0)

		    { iParams.iProtocol = KProtocolInetTcp; iParams.iSocketType = KSockStream; }

	    else if (protocol.Compare(KTe_UDPName)==0)

		    { iParams.iProtocol = KProtocolInetUdp; iParams.iSocketType = KSockDatagram; }

	    else if (protocol.Compare(KDummyProtocolName)==0)

		    {

			iParams.iProtocol = KProtocolInetDummy;

			iParams.iSocketType = KSockDatagram;

			}

	    else

	        {

	        INFO_PRINTF3(_L("%S: Protocol (%S) not recognised."),&iParams.iSocketName,&protocol);

	        return KErrNotFound;

	        }

        }

    // All ok if we got this far

    return KErrNone;

	}

TVerdict COpenRSocketStep::doSingleTestStep()

	{

    TInt error = iEsockTest->OpenSocket(iParams);

    if (error!=KErrNone)

        {

        INFO_PRINTF2(_L("Could not open socket (%S)."),&iParams.iSocketName);

        INFO_PRINTF2(_L("Error: %d."),error);

		SetTestStepError(error); 

		return EFail;

        }

	return EPass;

	}

// Close Socket

//-------------

CCloseRSocketStep::CCloseRSocketStep(CCEsockTestBase*& aEsockTest)

:   CTe_EsockStepBase(aEsockTest)

	{

	SetTestStepName(KCloseRSocketStep);

	}

TInt CCloseRSocketStep::ConfigureFromIni()

	{

	iSocketName.Set(KNullDesC);

	if((GetStringFromConfig(iSection, KTe_SocketName, iSocketName) != 1)

		|| (iSocketName.Length() == 0))

		{

		INFO_PRINTF1(_L("Couldn't find appropriate field in config file"));

		return KErrNotFound;

		}

    // All ok if we got this far

    return KErrNone;

	}

TVerdict CCloseRSocketStep::doSingleTestStep()

	{

    TInt error = iEsockTest->CloseSocket(iSocketName);

    if (error!=KErrNone)

        {

        INFO_PRINTF2(_L("Could not close socket (%S)."),&iSocketName);

        INFO_PRINTF2(_L("Error: %d."),error);

        return EFail;

        }

	return EPass;

	}

// Create Socket

//--------------

CCreateRSocketStep::CCreateRSocketStep(CCEsockTestBase*& aEsockTest)

:   CTe_EsockStepBase(aEsockTest)

	{

	SetTestStepName(KCreateRSocketStep);

	}

TVerdict CCreateRSocketStep::doTestStepPreambleL()

	{

	SetTestStepResult(EFail);

	if (iEsockTest==NULL)

	    iEsockTest = new (ELeave) CCEsockTestBase;

	SetTestStepResult(EPass);

	return TestStepResult();

	}

TInt CCreateRSocketStep::ConfigureFromIni()

	{

	iParams.Reset();

	// Read in appropriate fields

	if((GetStringFromConfig(iSection, KTe_SocketName, iParams.iSocketName) != 1)

		|| (iParams.iSocketName.Length() == 0))

		{

		INFO_PRINTF1(_L("Couldn't find appropriate field in config file"));

		return KErrNotFound;

		}

    // All ok if we got this far

    return KErrNone;

	}

TVerdict CCreateRSocketStep::doSingleTestStep()

	{

    TInt error = iEsockTest->CreateSocket(iParams.iSocketName);

    if (error!=KErrNone)

        {

        INFO_PRINTF2(_L("Could not create socket (%S)."),&iParams.iSocketName);

        INFO_PRINTF2(_L("Error: %d."),error);

        return EFail;

        }

	return EPass;

    }

// Connect Socket

//---------------

CConnectRSocketStep::CConnectRSocketStep(CCEsockTestBase*& aEsockTest)

:   CTe_EsockStepBase(aEsockTest)

	{

	SetTestStepName(KConnectRSocketStep);

	}

TInt CConnectRSocketStep::ConfigureFromIni()

	{

	iParams.Reset();

	// Read in appropriate fields

	if((GetStringFromConfig(iSection, KTe_SocketName, iParams.iSocketName) != 1)

		|| (iParams.iSocketName.Length() == 0))

		{

		INFO_PRINTF1(_L("Couldn't find appropriate field in config file"));

		return KErrNotFound;

		}

    TPtrC protocol;

	if (GetStringFromConfig(iSection,KTe_Protocol,protocol)!=1)

        {

        INFO_PRINTF2(_L("%S: Protocol missing."),&iParams.iSocketName);

        return KErrNotFound;

        }

    if (protocol.Compare(KTe_TCPName)==0)

	    { iParams.iProtocol = KProtocolInetTcp; }

    else if (protocol.Compare(KTe_UDPName)==0)

	    { iParams.iProtocol = KProtocolInetUdp; }

	else if (protocol.Compare(KDummyProtocolName)==0)

		{

		iParams.iProtocol = KProtocolInetDummy;

		}

    else

        {

        INFO_PRINTF3(_L("%S: Protocol (%S) not recognised."),&iParams.iSocketName,&protocol);

        return KErrNotFound;

        }

	// IP Address Local

	if (GetIpAddressFromConfig(iSection,KTe_SourceAddressName,iParams.iLocalIP)!=1)

        {

        INFO_PRINTF2(_L("%S: Local address missing."),&iParams.iSocketName);

        //return KErrNotFound;  // Source Address is not mandatory always.

        }

	// IP Address Remote

	if (GetIpAddressFromConfig(iSection,KTe_DestinationAddressName,iParams.iRemoteIP)!=1)

        {

        INFO_PRINTF2(_L("%S: Remote address missing."),&iParams.iSocketName);

        return KErrNotFound;

        }

	// Port Number Local

    TInt portLocal;

	if (GetIntFromConfig(iSection,KTe_SourcePortName,portLocal)!=1)

        {

        portLocal = 0;

        INFO_PRINTF2(_L("%S: Source port missing. Defaulting to unassigned (port=0)."), &iParams.iSocketName);

        }

    iParams.iLocalIP.SetPort(portLocal);

	// Port Number

	// Port Number Remote

    TInt port;

	if (GetIntFromConfig(iSection,KTe_DestinationPortName,port)!=1)

        {

        INFO_PRINTF2(_L("%S: Remote port missing."),&iParams.iSocketName);

        return KErrNotFound;

        }

	// Attach the port number to remote IP

	iParams.iRemoteIP.SetPort(port);

    iCancelFlag = EFalse;

    GetBoolFromConfig(iSection,KTe_ImmediateCancelFlagName,iCancelFlag);

    // All ok if we got this far

    return KErrNone;

	}

TVerdict CConnectRSocketStep::doSingleTestStep()

	{

    TRequestStatus reqStat;

    TInt error = iEsockTest->ConnectSocket(iParams,reqStat);

    if (error!=KErrNone)

        {

        INFO_PRINTF2(_L("Could not connect socket (%S)."),&iParams.iSocketName);

        INFO_PRINTF2(_L("Error: %d."),error);

        SetTestStepResult(EFail);

        }

	else

		{

		if (iCancelFlag)

			{

	        error = iEsockTest->CancelConnectSocket(iParams);

	        if (error!=KErrNone)

		        {

                INFO_PRINTF2(_L("Could not cancel connecting socket (%S)."),&iParams.iSocketName);

                INFO_PRINTF2(_L("Error: %d."),error);

				SetTestStepResult(EFail);

				}

			}

		User::WaitForRequest(reqStat);

	    if (reqStat.Int()!=KErrNone)

		    {

            INFO_PRINTF2(_L("Could not connect socket (%S)."),&iParams.iSocketName);

            INFO_PRINTF2(_L("Error: %d."),error);

	        SetTestStepResult(EFail);

	        }

		}

	return TestStepResult();

	}

// Send Receive on Socket

//-----------------------

_LIT(KPayloadColour, "PayloadColour");

CSendReceiveRSocketStep::CSendReceiveRSocketStep(CCEsockTestBase*& aEsockTest)

:   CTe_EsockStepBase(aEsockTest)

	{

	SetTestStepName(KSendReceiveRSocketStep);

	}

TInt CSendReceiveRSocketStep::ConfigureFromIni()

	{

	iParams.Reset();

	// Read in appropriate fields

	if((GetStringFromConfig(iSection, KTe_SocketName, iParams.iSocketName) != 1)

		|| (iParams.iSocketName.Length() == 0))

		{

		INFO_PRINTF1(_L("Couldn't find appropriate field in config file"));

		return KErrNotFound;

		}

	if (GetIntFromConfig(iSection,KTe_PacketSizeName,iParams.iPacketSize)!=1)

        {

        INFO_PRINTF2(_L("%S: Packet size missing."),&iParams.iSocketName);

        return KErrNotFound;

        }

    if (GetIntFromConfig(iSection,KTe_NrOfPacketsName,iParams.iNrOfPackets)!=1)

        {

        INFO_PRINTF2(_L("%S: Nr of packets missing."),&iParams.iSocketName);

        return KErrNotFound;

        }

    TPtrC protocol;

	if (GetStringFromConfig(iSection,KTe_Protocol,protocol)!=1)

        {

        INFO_PRINTF2(_L("%S: Protocol missing."),&iParams.iSocketName);

        return KErrNotFound;

        }

    if (protocol.Compare(KTe_TCPName)==0)

	    { iParams.iProtocol = KProtocolInetTcp; }

    else if (protocol.Compare(KTe_UDPName)==0)

	    { iParams.iProtocol = KProtocolInetUdp; }

	else if (protocol.Compare(KDummyProtocolName)==0)

		{

		iParams.iProtocol = KProtocolInetDummy;

		}

    else

        {

        INFO_PRINTF3(_L("%S: Protocol (%S) not recognised."),&iParams.iSocketName,&protocol);

        return KErrNotFound;

        }

	// OPTIONAL: colouring data by access point Id

	iParams.iPayloadColouringIsEnabled = EFalse;

	if (GetIntFromConfig(iSection, KPayloadColour, iParams.iPayloadColour) == 1)

		{

		if((iParams.iPayloadColour < 0) || (iParams.iPayloadColour > 255))

			{

			INFO_PRINTF2(_L("Payload colour out of range. Must be (0 <= x <= 255)."), iParams.iPayloadColour);

	        return KErrArgument;

			}

		else if (iParams.iProtocol != KProtocolInetUdp)

			{

			INFO_PRINTF2(_L("Payload colouring only available on UDP sockets."), iParams.iPayloadColour);

	        return KErrArgument;

			}

		else

			{

			// All good

			iParams.iPayloadColouringIsEnabled = ETrue;

			INFO_PRINTF2(_L("Payload colouring enabled. payloadColour:%d"), iParams.iPayloadColour);

			}

		}

    // All ok if we got this far

    return KErrNone;

	}

TVerdict CSendReceiveRSocketStep::doSingleTestStep()

	{

    TRAPD(error,iEsockTest->SendAndReceiveDataL(iParams));

    SetTestStepError(error);

    if (error!=KErrNone)

        {

        INFO_PRINTF2(_L("Could not send/receive on socket (%S)."),&iParams.iSocketName);

        INFO_PRINTF2(_L("Error: %d."),error);

        return EFail;

        }

	return EPass;

	}

// Add Socket to SubConnection

//----------------------------

CAddRSocketToRSubConnectionStep::CAddRSocketToRSubConnectionStep(CCEsockTestBase*& aEsockTest)

:   CTe_EsockStepBase(aEsockTest)

	{

	SetTestStepName(KAddRSocketToRSubConnectionStep);

	}

TInt CAddRSocketToRSubConnectionStep::ConfigureFromIni()

	{

	iParams.Reset();

	// Read in appropriate fields

	if((GetStringFromConfig(iSection, KTe_SocketName, iParams.iSocketName) != 1)

		|| (iParams.iSocketName.Length() == 0))

		{

		INFO_PRINTF1(_L("Couldn't find appropriate field in config file"));

		return KErrNotFound;

		}

    if (GetStringFromConfig(iSection,KTe_SubConnectionName,iParams.iSubConnectionName)!=1

        || iParams.iSubConnectionName.Length()==0)

            {

            INFO_PRINTF2(_L("%S: SubConnection name missing."),&iParams.iSocketName);

            return KErrNotFound;

            }

    // All ok if we got this far

    return KErrNone;

	}

TVerdict CAddRSocketToRSubConnectionStep::doSingleTestStep()

	{

    TRequestStatus reqStat;

    TInt error = iEsockTest->AddSocketToSubConnection(iParams,reqStat);

    if (error!=KErrNone)

        {

        INFO_PRINTF3(_L("Could not add socket (%S) to subconnection (%S)."),&iParams.iSocketName,&iParams.iSubConnectionName);

        INFO_PRINTF2(_L("Error: %d."),error);

        return EFail;

        }

	User::WaitForRequest(reqStat);

	error = reqStat.Int();

	if (error!=KErrNone)

		{

		INFO_PRINTF3(_L("Could not add socket (%S) to subconnection (%S)."),&iParams.iSocketName,&iParams.iSubConnectionName);

        INFO_PRINTF2(_L("Error: %d."),error);

	    return EFail;

		}

	return EPass;

	}

// Add Socket to SubConnection without calling User::WaitForRequest

// Tests PDEF134595

//----------------------------

CAddRSocketToRSubConNoWaitStep::CAddRSocketToRSubConNoWaitStep(CCEsockTestBase*& aEsockTest)

:   CTe_EsockStepBase(aEsockTest)

    {

    SetTestStepName(KAddRSocketToRSubConNoWaitStep);

    }

TInt CAddRSocketToRSubConNoWaitStep::ConfigureFromIni()

    {

    //clean parameters from previous runs

    iParams.Reset();

    if (GetStringFromConfig(iSection, KTe_SocketName, iParams.iSocketName) != 1

        || iParams.iSocketName.Length()==0)

            return KErrNotFound;

    if (GetStringFromConfig(iSection,KTe_SubConnectionName,iParams.iSubConnectionName)!=1

        || iParams.iSubConnectionName.Length()==0)

            {

            INFO_PRINTF2(_L("%S: SubConnection name missing."),&iParams.iSocketName);

            return KErrNotFound;

            }

    return KErrNone;

    }

TVerdict CAddRSocketToRSubConNoWaitStep::doSingleTestStep()

    {

        TRequestStatus reqStat,reqStat2;

        TInt error;

        error = iEsockTest->AddSocketToSubConnection(iParams, reqStat);

        if (error != KErrNone)

            {

            INFO_PRINTF2(_L("Could not add socket (%d)!"), error);

            SetTestStepResult(EFail);

            }

        return TestStepResult();

    }

// Add Socket to SubConnection (OOM)

//----------------------------------

CAddRSocketToRSubConnectionOOMStep::CAddRSocketToRSubConnectionOOMStep(CCEsockTestBase*& aEsockTest)

:   CTe_EsockStepBase(aEsockTest)

	{

	SetTestStepName(KAddRSocketToRSubConnectionOOMStep);

	}

TInt CAddRSocketToRSubConnectionOOMStep::ConfigureFromIni()

	{

	iParams.Reset();