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

		{

        TInt protocolInt = 0;

        //Try obtaining protocol type as int, failing that, go to predefined types

        if (GetIntFromConfig(iSection, KTe_Protocol, protocolInt)!=1)

            {

            TPtrC protocolStr;

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

                {

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

                return KErrNotFound;        

                }

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

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

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

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

            else if (protocolStr.Compare(KTe_DummyNetworkProtocolName)==0)

                { protocolInt = KDummyNetworkProtocol; iParams.iSocketType = KSockDatagram;} 

            else if (protocolStr.Compare(KTe_DummyProtocolName)==0)

                { protocolInt = KDummyProtocol; iParams.iSocketType = KSockDatagram;} 

            else

                {

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

                return KErrNotFound;

                }

            }

        iParams.iProtocol = protocolInt;

        }

	TPtrC socketTypeStr;	

    if (GetStringFromConfig(iSection, KTe_SocketType,socketTypeStr)!=1 )

        {

        INFO_PRINTF2(_L("%S: Socket type missing, defaulting based on specified protocol"),&iParams.iSocketName);

        }

    else

        {

        if (socketTypeStr.Compare(KTe_SocketDatagram)==0)

             { iParams.iSocketType = KSockDatagram;}

        else if (socketTypeStr.Compare(KTe_SocketStream)==0)

             { iParams.iSocketType = KSockStream; }

        else

            {

            INFO_PRINTF3(_L("%S: Socket type (%S) not recognised."),&iParams.iSocketName,&socketTypeStr);

            return KErrNotFound;

            }

        }

    TInt protocolInt = KAfInet;

    if (GetIntFromConfig(iSection, KTe_ConnectionType, protocolInt)!=1 )

        {

        INFO_PRINTF2(_L("%S: Address Family (ConnType) missing, defaulting to KAfInet"),&iParams.iSocketName);

        }

    iParams.iAddrFamily = protocolInt;

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

    }

// Bind Socket

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

CBindRSocketStep::CBindRSocketStep(CCEsockTestBase*& aEsockTest)

:   CTe_EsockStepBase(aEsockTest)

    {

    SetTestStepName(KBindRSocketStep);

    }

TInt CBindRSocketStep::ConfigureFromIni()

    {

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

        }

    TInt protocolInt = 0;

    if (GetIntFromConfig(iSection, KTe_Protocol, protocolInt)!=1)

        {

        TPtrC protocolStr;

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

            {

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

            return KErrNotFound;        

            }

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

            { protocolInt = KProtocolInetTcp; }

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

            { protocolInt = KProtocolInetUdp; }

        else if (protocolStr.Compare(KTe_DummyNetworkProtocolName)==0)

            { protocolInt = KDummyNetworkProtocol; }

        else if (protocolStr.Compare(KTe_DummyProtocolName)==0)

            { protocolInt = KDummyProtocol; }

        else

            {

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

            return KErrNotFound;

            }

        }

    iParams.iProtocol = protocolInt;

    // IP Address Local

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

        {

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

        }    

    // All ok if we got this far

    return KErrNone;

    }

TVerdict CBindRSocketStep::doSingleTestStep()

    {

    TInt error = iEsockTest->BindSocket(iParams/*,reqStat*/);

    if (error!=KErrNone)

        {

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

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

        SetTestStepResult(EFail);

        }

    return TestStepResult();

    }

// 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(KTe_DummyNetworkProtocolName)==0)

		{ iParams.iProtocol = KDummyNetworkProtocol; }

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

        { iParams.iProtocol = KDummyProtocol; }    

    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(KTe_DummyNetworkProtocolName)==0)

		{ iParams.iProtocol = KDummyNetworkProtocol; }

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

        { iParams.iProtocol = KDummyProtocol; }    

    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);