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

// Implementation file for the NIF Protocol2

// 

//

/**

 @file DummyNifBinder.cpp

*/

#include <e32std.h>

#include <eui_addr.h>

#include <ip4_hdr.h>

#include <udp_hdr.h>

#include <in_chk.h>

#include <in_iface.h>

#include <comms-infras/nifif.h>

#include <in_sock.h>

#include "dummynifvar.h"

#include "Dummynifbinder.h"

#include "DummyProvision.h"

using namespace ESock;

_LIT8(KNif,"DummyNif");

_LIT8(KBinder4,"Binder4");

_LIT8(KBinder6,"Binder6");

//

// CDummyNifBinder4 //

//

CDummyNifBinder4::CDummyNifBinder4(CDummyNifSubConnectionFlow& aDummyNifSubConnectionFlow) : iDummyNifSubConnectionFlow(aDummyNifSubConnectionFlow)

	{

    __FLOG_OPEN(KNif, KBinder4);

   	// generate my local ip address (ip4) - vals potentially will be overwritten by any derived classes

	iLocalAddressBase = KDummyNifLocalAddressBase; // also used later in control method

	TUint32 id = ((TUint32)this) % 255;

	// Avoid the reserved address (least significant byte KDummyHungryNifReservedHostId) that

	// is never to be allocated as a local address.

	if (id == KDummyHungryNifReservedHostId)

		{

		++id;

		}

	iLocalAddress = iLocalAddressBase + id;

    __FLOG_3(_L8("CDummyNifBinder4 %08x:\tCDummyNifBinder4(CDummyNifSubConnectionFlow& %08x): iLocalAddress %08x"), this, &aDummyNifSubConnectionFlow, iLocalAddress);

}

CDummyNifBinder4::~CDummyNifBinder4()

/** 

Destroys 'this'

*/

    {

    if (iTestSubscriber)

    	{

    	iTestSubscriber->Cancel();

    	}

    delete iTestSubscriber;

    __FLOG_1(_L8("CDummyNifBinder4 %08x:\t~CDummyNifBinder4()"), this);

    ASSERT(iUpperControl == NULL);

    ASSERT(iUpperReceiver == NULL);

	ASSERT(!iErrorOneShot.IsActive());

    __FLOG_CLOSE;

    }

CDummyNifBinder4* CDummyNifBinder4::NewL(CDummyNifSubConnectionFlow& aDummyNifSubConnectionFlow)

	{

	CDummyNifBinder4* self = new (ELeave) CDummyNifBinder4(aDummyNifSubConnectionFlow);

	CleanupStack::PushL(self);

	const TProviderInfo& providerInfo = static_cast<const TProviderInfoExt&>(

		aDummyNifSubConnectionFlow.AccessPointConfig().FindExtensionL(

			STypeId::CreateSTypeId(TProviderInfoExt::EUid, TProviderInfoExt::ETypeId))).iProviderInfo;

	TInt ap = providerInfo.APId();

	if (CDummyNifFlowTestingSubscriber::ShouldRun(ap))

		{

		ASSERT(self->iTestSubscriber==NULL);

		self->iTestSubscriber = CDummyNifFlowTestingSubscriber::NewL(aDummyNifSubConnectionFlow, ap);

		}

	CleanupStack::Pop(self);

	return self;

	}

void CDummyNifBinder4::UpdateHeaders(TInet6HeaderIP4* aIp4, TInet6HeaderUDP* aUdp)

/**

Update the IPv4 and UDP headers to allow the packet to be looped back.

*/

{

	// swap over the destination and source addresses

	TUint32 temp;

	temp = aIp4->SrcAddr();

	aIp4->SetSrcAddr(aIp4->DstAddr());

	aIp4->SetDstAddr(temp);

   	// swap over the destination and source ports

	if (aUdp)

		{

		TUint tempPort;

		tempPort = aUdp->DstPort();

		aUdp->SetDstPort(aUdp->SrcPort());

		aUdp->SetSrcPort(tempPort);

		}

	// we've changed the ip hdr so need to recalculate the ip hdr checksum

	aIp4->SetChecksum(0); 

	aIp4->SetChecksum(TChecksum::ComplementedFold(TChecksum::Calculate((TUint16*)aIp4, aIp4->HeaderLength())));

	// also want to set the udp checksum to zero cos it will be wrong now that we have 

	// changed the ip hdr - just set to zero and it is ignored

	if (aUdp)

		{

		aUdp->SetChecksum(0);

		}

}

MLowerDataSender::TSendResult CDummyNifBinder4::Send(RMBufChain& aData)

/**

Entry point for receiving IPv4 outgoing data

@param aData MBuf chain containing data to send

@return an indication of whether the upper layer should block.

*/

	{

    __FLOG_1(_L8("CDummyNifBinder4 %08x:\tSend()"), this);

	// Loop the data straight back into the TCP/IP stack

	ProcessPacket(aData);

	return ESendAccepted;

	}

MLowerDataSender* CDummyNifBinder4::Bind(MUpperDataReceiver& aUpperReceiver, MUpperControl& aUpperControl)

/**

Return the MLowerDataSender instance (CDummyNifBinder4) that we

previously allocated.

*/

	{

    __FLOG_3(_L8("CDummyNifBinder4 %08x:\tBind(aUpperReceiver %08x, aUpperControl %08x)"), this, &aUpperReceiver, &aUpperControl);

	iUpperReceiver = &aUpperReceiver;

	iUpperControl = &aUpperControl;

	// Signal upper layer that we are ready

	BinderReady();

	return this;

	}

void CDummyNifBinder4::Unbind(MUpperDataReceiver& aUpperReceiver, MUpperControl& aUpperControl)

/**

Forget our association with upper layer.

*/

	{

    static_cast<void>(aUpperReceiver);

    static_cast<void>(aUpperControl);

    __FLOG_1(_L8("CDummyNifBinder4 %08x:\tUnbind()"), this);

    ASSERT(&aUpperReceiver == iUpperReceiver);

    ASSERT(&aUpperControl == iUpperControl);

    iUpperReceiver = NULL;

    iUpperControl = NULL;

	}

void CDummyNifBinder4::BinderReady()

/**

Signal upper layer that we are ready

*/

	{

    __FLOG_1(_L8("CDummyNifBinder4 %08x:\tBinderReady()"), this);

	iUpperControl->StartSending();

	}

void CDummyNifBinder4::ProcessPacket(RMBufChain& aPdu)

/**

Process incoming data

*/

    {

	// this received data has already been looped back...

	// get the ip header from the RMBufChain

	TInet6HeaderIP4* ip4 = (TInet6HeaderIP4*) aPdu.First()->Next()->Ptr();

	if (ip4->Protocol() != KProtocolInetUdp)

		{

		//Non UDP traffic goes here!

		__FLOG_3(_L("CDummyNifBinder4 %08x:\tProcessPacket(): IPv4 length %d, protocol %d"), this, ip4->TotalLength(), ip4->Protocol());

		UpdateHeaders(ip4, NULL);

		// now process it (pass up the stack)

		iUpperReceiver->Process(aPdu);

		return;

		}

	// get the udp header as well - assume only udp traffic here

	TInet6HeaderUDP* udp = (TInet6HeaderUDP*) ip4->EndPtr();

	__FLOG_4(_L("CDummyNifBinder4 %08x:\tProcessPacket(): UDP length %d, src port %d, dst port %d"), this, udp->Length(), udp->SrcPort(), udp->DstPort());

	// depending on the contents, pass it on up thru the stack 

	// or maybe do something else

	// use the destination port number to decide whether or not the payload is a command

	TUint dstPort = udp->DstPort();

	if (KDummyNifCmdPort == dstPort)

    	{

		// let's use the first payload byte as the command byte

		switch (*(udp->EndPtr()))

    		{

		    case KForceDisconnect:

			    __FLOG(_L("KForceDisconnect command"));

			    // do some action

	            Flow()->Progress(KLinkLayerClosed, KErrCommsLineFail);

	            Flow()->FlowDown(KErrCommsLineFail, MNifIfNotify::EDisconnect);

			    // no return code so all we can do is respond with what we got

			    UpdateHeaders(ip4, udp);

			    iUpperReceiver->Process(aPdu);

			    break;

		    case KForceReconnect:

			    __FLOG(_L("KForceReconnect command"));

			    // do some action

	            Flow()->Progress(KLinkLayerClosed, KErrCommsLineFail);

	            Flow()->FlowDown(KErrCommsLineFail, MNifIfNotify::EReconnect);

			    // no return code so all we can do is respond with what we got

			    UpdateHeaders(ip4, udp);

			    iUpperReceiver->Process(aPdu);

			    break;

		    case KSendNotification:

			    __FLOG(_L("KSendNotification command"));

			    //let's write the result in the next byte of the reply

			    if (Flow()->AgentProvision()->IsDialIn() == KErrNotSupported)

				    udp->EndPtr()[1] = (unsigned char) KErrNone;

			    else

				    udp->EndPtr()[1] = (unsigned char) KErrGeneral; // this will lose it's sign :-(

			    UpdateHeaders(ip4, udp);

                iUpperReceiver->Process(aPdu);

			    break;

			case KForceFinishedSelection:

				__FLOG(_L("KForceFinishedSelection command"));

				// force subConn into KFinishedSelection State

	            Flow()->Progress(KFinishedSelection, KErrNone);

				UpdateHeaders(ip4, udp);

				aPdu.Free();

				break;

    		case KForceBinderError:

			    __FLOG(_L("KForceBinderError command"));

				// We cannot signal an error whilst in the middle of a send in the TCP/IP stack,

				// as the act of signalling the error will eventually result in the CNifIfBase binder

				// being destructed by the TCP/IP stack whilst we're in the middle of using it.

				// Consequently, we would panic on exit from this routine.  So make the call

				// via an asynchronous callback.

				if (!iErrorOneShot.IsActive())

					{

					iErrorOneShot.Schedule(iUpperControl);

					}

    			aPdu.Free();

    			break;

    		case KColourDataByLinkTierAccessPointId:

				{

				const TProviderInfoExt* providerInfo = static_cast<const TProviderInfoExt*>(

					iDummyNifSubConnectionFlow.AccessPointConfig().FindExtension(

						STypeId::CreateSTypeId(TProviderInfoExt::EUid, TProviderInfoExt::ETypeId)));

				ASSERT(providerInfo); // Should always be present

				// We are going to simply add the access point id to the command byte

				// A test client can then validate that the socket is connected on the expected access point

			    __FLOG(_L("KColourDataByAccessPointId command"));

				*(udp->EndPtr()) += static_cast<TUint8>(providerInfo->iProviderInfo.APId());

				// Update the udp headers and forward on

				UpdateHeaders(ip4, udp);

				iUpperReceiver->Process(aPdu);

				}

				break;

		    default:

			    __FLOG(_L("Unknown command - ignoring it"));

			    aPdu.Free();

			    // unknown command, just ignore this packet???

		    }

	    }

	else

	    {

        __FLOG(_L("Standard echo packet"));

        if (iTestSubscriber && !iTestSubscriber->IsEnabled())

        	{

        	__FLOG(_L("Bearer not available. Packet dropped."));

        	aPdu.Free();

        	return;

        	}

        // just echo the packet back to the original sender

		// update the headers (addresses, checksums etc)

		UpdateHeaders(ip4, udp);

		// now process it (pass up the stack)

		iUpperReceiver->Process(aPdu);

		}

    }

TInt CDummyNifBinder4::GetConfig(TBinderConfig& aConfig)

/**

Return IPv4 configuration information.

Called from upper layer.

@param aConfig structure to populate with IPv4 configuration

*/

	{

    TBinderConfig4* config = TBinderConfig::Cast<TBinderConfig4>(aConfig);

	if(config == NULL)

		{

		return KErrNotSupported;

		}

	// Setup config

	config->iInfo.iFeatures = KIfCanBroadcast | KIfCanMulticast;

	config->iInfo.iMtu = 1500;

	config->iInfo.iRMtu = 1500;

	config->iInfo.iSpeedMetric = 0;

	config->iFamily = KAfInet6;

    TUint32 address;

	const TInt KPort = 65;

    config->iFamily = KAfInet;

	__FLOG_2(_L8("CDummyNifBinder4 %08x:\tGetConfig(): iLocalAddress %08x"), this, iLocalAddress);        

	config->iAddress.SetAddress(iLocalAddress);

	config->iAddress.SetPort(KPort);

	// network mask

	config->iNetMask.SetAddress(KInetAddrNetMaskC);	// 255.255.255.0

	config->iNetMask.SetPort(KPort);

	// broadcast address

	address = iLocalAddressBase + KBroadcastAddressSuffix;

	config->iBrdAddr.SetAddress(address);

	config->iBrdAddr.SetPort(KPort);

	// default gateway

	address = iLocalAddressBase + KDefaultGatewayAddressSuffix;

	config->iDefGate.SetAddress(address);

	config->iDefGate.SetPort(KPort);

	// primary DNS, just make same as default gateway

	config->iNameSer1.SetAddress(address);

	config->iNameSer1.SetPort(KPort);

	// secondary DNS

	address = iLocalAddressBase + KSecondaryDnsAddressSuffix;

	config->iNameSer2.SetAddress(address);

	config->iNameSer2.SetPort(KPort);

	return KErrNone;

	}

TInt CDummyNifBinder4::Control(TUint aLevel, TUint aName, TDes8& aOption)

/**

Called from upper layer for special control functionality.

*/

	{

    static_cast<void> (aOption);

	__FLOG_3(_L("CDummyNifBinder4 %08x:\tControl(aLevel %x, aName %x)"), this, aLevel, aName);

	return KErrNotSupported;

	}

TInt CDummyNifBinder4::GetName(TDes& aName)

/**

Called from upper layer to retrieve the binder name.

@param aName populated with name

@return KErrNone on success, else a system wide error code.

*/

	{

	// This name matches the NIF-based DummyNif to match any potential

	// test code expectations on the name.

	aName.Format(_L("dummynif[0x%08x]"), this);

	__FLOG_2(_L("CDummyNifBinder4 %08x:\tGetName(): %S"), this, &aName);

	return KErrNone;

	}

TInt CDummyNifBinder4::BlockFlow(MLowerControl::TBlockOption /*aOption*/)

	{

    __FLOG_1(_L8("CDummyNifBinder4 %08x:\tBlockFlow()"), this);

	return KErrNotSupported;

	}

TBool CDummyNifBinder4::MatchesUpperControl(ESock::MUpperControl* aUpperControl) const

/**

Utility function that returns whether this binder is associated with the

MUpperControl object passed as argument.

@param aUpperControl upper layer to match against

@return ETrue on a match else EFalse.

*/

	{

	return aUpperControl == iUpperControl;

	}

// =================================================================================

//

// CDummyNifBinder6

CDummyNifBinder6::CDummyNifBinder6(CDummyNifSubConnectionFlow& aDummyNifSubConnectionFlow) : iDummyNifSubConnectionFlow(aDummyNifSubConnectionFlow)

	{

    __FLOG_OPEN(KNif, KBinder6);

    __FLOG_2(_L8("CDummyNifBinder6 %08x:\tCDummyNifBinder6(CDummyNifSubConnectionFlow& %08x)"), this, &aDummyNifSubConnectionFlow);

	}

CDummyNifBinder6::~CDummyNifBinder6()

/** 

Destroys 'this'

*/

    {

    __FLOG(_L8("CDummyNifBinder6:\t~CDummyNifBinder6()"));

    __FLOG_CLOSE;

    }

CDummyNifBinder6* CDummyNifBinder6::NewL(CDummyNifSubConnectionFlow& aDummyNifSubConnectionFlow)

	{

	return new (ELeave) CDummyNifBinder6(aDummyNifSubConnectionFlow);

	}

void CDummyNifBinder6::UpdateHeaders(TInet6HeaderIP* aIp6, TInet6HeaderUDP* /*aUdp*/)

    {

	// swap over the destination and source addresses

	TIp6Addr temp;

	temp = aIp6->SrcAddr();

	aIp6->SetSrcAddr(aIp6->DstAddr());

	aIp6->SetDstAddr(temp);

    }

MLowerDataSender::TSendResult CDummyNifBinder6::Send(RMBufChain& aData)

/**

Send IPv6 data

Note: not clear that this is properly supported or used.

@param aData data to send

*/

	{

    __FLOG(_L8("CDummyNifBinder6:\tSend()"));

   	// Loop the data straight back into the TCP/IP stack

	ProcessPacket(aData);

	return ESendAccepted;

	}

MLowerDataSender* CDummyNifBinder6::Bind(MUpperDataReceiver& aUpperReceiver, MUpperControl& aUpperControl)

	{

    __FLOG_2(_L8("CDummyNifBinder6:\tBind(MUpperDataReceiver %08x, MUpperControl %08x)"), &aUpperReceiver, &aUpperControl);

	iUpperReceiver = &aUpperReceiver;

	iUpperControl = &aUpperControl;

	// Signal upper layer that we are ready

	BinderReady();

	return this;

	}

void CDummyNifBinder6::BinderReady()

/**

Signal to upper layer that we are ready

*/

    {

    __FLOG(_L8("CDummyNifBinder6:\tBinderReady()"));

	iUpperControl->StartSending();

    }

void CDummyNifBinder6::ProcessPacket(RMBufChain& aPdu)

/**

Process incoming IPv6 packets.

Note: not clear that this is properly supported or used.

@param aPdu incoming data packet

*/

    {

    __FLOG(_L8("CDummyNifBinder6:\tProcessPacket()"));

	// this received data has already been looped back...

	// get the ip header from the RMBufChain

	TInet6HeaderIP* ip6 = (TInet6HeaderIP*) aPdu.First()->Next()->Ptr();

	TInet6HeaderUDP* udp = NULL;

	if ((TUint)ip6->NextHeader() == KProtocolInetUdp)

		{

		// get the udp header as well - assume only udp traffic here

		udp = (TInet6HeaderUDP*) ip6->EndPtr();

		__FLOG_3(_L("CDummyNifBinder6:\tProcessPacket(...): UDP length %d, src port %d, dst port %d"),

			udp->Length(), udp->SrcPort(), udp->DstPort());

		// depending on the contents, pass it on up thru the stack 

		// or maybe do something else

		// use the destination port number to decide whether or not the payload is a command

		TUint dstPort = udp->DstPort();

		if (KDummyNifCmdPort == dstPort)

            {