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

//

#include <kernel/kern_priv.h>

#ifdef SYMBIAN_OLD_EXPORT_LOCATION

#include <comms-infras/legacy_loopback_driver.h>

#else

//this header is not exported, needs to be a user include

#include "legacy_loopback_driver.h"

#endif

#include "device.h"

_LIT(KESockLoopbackPanicCategory,"ESockLoopback");

/**

  Standard export function for LDDs. This creates a DLogicalDevice derived object,

  in this case, our DESockLoopbackFactory

*/

DECLARE_STANDARD_LDD()

	{

	return new DESockLoopbackFactory;

	}

/**

  Constructor

*/

DESockLoopbackFactory::DESockLoopbackFactory()

	{

	// Set version number for this device

	iVersion=RLegacyLoopbackDriver::VersionRequired();

	// Indicate that we work with a PDD

	iParseMask=KDeviceAllowPhysicalDevice;

	}

/**

  Second stage constructor for DESockLoopbackFactory.

  This must at least set a name for the driver object.

  @return KErrNone if successful, otherwise one of the other system wide error codes.

*/

TInt DESockLoopbackFactory::Install()

	{

	return SetName(&RLegacyLoopbackDriver::Name());

	}

DESockLoopbackFactory::~DESockLoopbackFactory()

   	{

   	}

/**

  Return the drivers capabilities.

  Called in the response to an RDevice::GetCaps() request.

  @param aDes User-side descriptor to write capabilities information into

*/

void DESockLoopbackFactory::GetCaps(TDes8& aDes) const

	{

	// Create a capabilities object

	RLegacyLoopbackDriver::TCaps caps;

	caps.iVersion = iVersion;

	// Write it back to user memory

	Kern::InfoCopy(aDes,(TUint8*)&caps,sizeof(caps));

	}

/**

  Called by the kernel's device driver framework to create a Logical Channel.

  This is called in the context of the user thread (client) which requested the creation of a Logical Channel

  (E.g. through a call to RBusLogicalChannel::DoCreate)

  The thread is in a critical section.

  @param aChannel Set to point to the created Logical Channel

  @return KErrNone if successful, otherwise one of the other system wide error codes.

*/

TInt DESockLoopbackFactory::Create(DLogicalChannelBase*& aChannel)

	{

	aChannel=new DESockLoopbackChannel;

	if(!aChannel)

		return KErrNoMemory;

	return KErrNone;

	}

DESockLoopbackChannel::DESockLoopbackChannel()

	:	iSendDataDfc(SendDataDfc, this, 1),        // DFC is priority '1'

		iReceiveDataDfc(ReceiveDataDfc, this, 1)   // DFC is priority '1'

	{

	// Get pointer to client threads DThread object

	iClient=&Kern::CurrentThread();

	// Open a reference on client thread so it's control block can't dissapear until

	// this driver has finished with it.

	// Note, this call to Open can't fail since its the thread we are currently running in

	iClient->Open();

	memclr(&iDebugProbes, sizeof(iDebugProbes));

	Kern::Printf("clearing probe points\n");

	}

/**

  Second stage constructor called by the kernel's device driver framework.

  This is called in the context of the user thread (client) which requested the creation of a Logical Channel

  (E.g. through a call to RBusLogicalChannel::DoCreate)

  The thread is in a critical section.

  @param aUnit The unit argument supplied by the client to RBusLogicalChannel::DoCreate

  @param aInfo The info argument supplied by the client to RBusLogicalChannel::DoCreate

  @param aVer The version argument supplied by the client to RBusLogicalChannel::DoCreate

  @return KErrNone if successful, otherwise one of the other system wide error codes.

*/

TInt DESockLoopbackChannel::DoCreate(TInt /*aUnit*/, const TDesC8* /*aInfo*/, const TVersion& aVer)

	{

	// Check Platform Security capabilities of client thread (if required).

	//

	// Here we handle the simple case where:

	// 1. The device driver can only have one client thread

	// 2. The security policy is the binary all-or-nothing policy.

	//    E.g. "If you have the right capability you can do anything with the driver

	//    and if you don't have the capability you can't do anything"

	// 

	// If only some functionality of the driver is restricted, then the security check should

	// go elsewhere. E.g. in DoRequest/DoControl. In that case Kern::CurrentThreadHasCapability

	// shouldn't be used because the 'current thread' isn't the client.

	//

	// In this example we do a check here for ECapability_None (which always passes)...

	if(!Kern::CurrentThreadHasCapability(ECapability_None,__PLATSEC_DIAGNOSTIC_STRING("Checked by ESockLoopback")))

		return KErrPermissionDenied;

	// Check version

	if (!Kern::QueryVersionSupported(RLegacyLoopbackDriver::VersionRequired(),aVer))

		return KErrNotSupported;

	// Setup LDD for receiving client messages

	SetDfcQ(((DESockLoopbackPddFactory*)iPhysicalDevice)->iDfcQ);

	iMsgQ.Receive();

	// Associate DFCs with the same queue we set above to receive client messages on

	iSendDataDfc.SetDfcQ(iDfcQ);

	iReceiveDataDfc.SetDfcQ(iDfcQ);

	// Give PDD a pointer to this channel

	Pdd()->iLdd=this;

	// Done

	return KErrNone;

	}

#define PRINT_PROBE_POINTS(name) \

	{\

	Kern::Printf("probe list:%s\n\t", name); \

	for(TInt i = 0; i < (sizeof(iDebugProbes) / sizeof(TInt)); i++)\

		{ \

		Kern::Printf("%d:%d, ", i, iDebugProbes[i]); \

		} \

	Kern::Printf("\n\n"); \

	}

#define PROBE_LDD(index) (iDebugProbes[index]++)

#define PROBE_LDD_STATIC(index, obj) (((obj)->iDebugProbes[index])++)

DESockLoopbackChannel::~DESockLoopbackChannel()

	{

	// Cancel all processing that we may be doing

	DoCancel(RLegacyLoopbackDriver::EAllRequests);

	// Close our reference on the client thread

	Kern::SafeClose((DObject*&)iClient,NULL);

	}

/**

  Called when a user thread requests a handle to this channel.

*/

TInt DESockLoopbackChannel::RequestUserHandle(DThread* aThread, TOwnerType aType)

	{

	// Make sure that only our client can get a handle

	if (aType!=EOwnerThread || aThread!=iClient)

		return KErrAccessDenied;

	return KErrNone;

	}

/**

  Process a message for this logical channel.

  This function is called in the context of a DFC thread.

  @param aMessage The message to process.

	              The iValue member of this distinguishes the message type:

	              iValue==ECloseMsg, channel close message

	              iValue==KMaxTInt, a 'DoCancel' message

	              iValue>=0, a 'DoControl' message with function number equal to iValue

	              iValue<0, a 'DoRequest' message with function number equal to ~iValue

*/

void DESockLoopbackChannel::HandleMsg(TMessageBase* aMsg)

	{

	TThreadMessage& m=*(TThreadMessage*)aMsg;

	// Get message type

	TInt id=m.iValue;

	// Decode the message type and dispatch it to the relevent handler function...

	if (id==(TInt)ECloseMsg)

		{

		// Channel Close

		PRINT_PROBE_POINTS("ldd");

		DoCancel(RLegacyLoopbackDriver::EAllRequests);

		m.Complete(KErrNone, EFalse);

		return;

		}

	if (id==KMaxTInt)

		{

		// DoCancel

		DoCancel(m.Int0());

		m.Complete(KErrNone,ETrue);

		return;

		}

	if (id<0)

		{

		// DoRequest

		TRequestStatus* pS=(TRequestStatus*)m.Ptr0();

		TInt r=DoRequest(~id,pS,m.Ptr1(),m.Ptr2());

		if (r!=KErrNone)

			{

			Kern::RequestComplete(iClient,pS,r);

			}

		m.Complete(KErrNone,ETrue);

		}

	else

		{

		// DoControl

		TInt r=DoControl(id,m.Ptr0(),m.Ptr1());

		m.Complete(r,ETrue);

		}

	}

/**

  Process synchronous 'control' requests

*/

TInt DESockLoopbackChannel::DoControl(TInt aFunction, TAny* a1, TAny* a2)

	{

	(void)a2;   // a2 not used in this example

	TInt r;

	switch (aFunction)

		{

		case RLegacyLoopbackDriver::EGetConfig:

			r = GetConfig((TDes8*)a1);

			break;

		case RLegacyLoopbackDriver::ESetConfig:

			r = SetConfig((const TDesC8*)a1);

			break;

		default:

			r = KErrNotSupported;

			break;

		}

	return r;

	}

/**

  Process asynchronous requests.

*/

TInt DESockLoopbackChannel::DoRequest(TInt aReqNo, TRequestStatus* aStatus, TAny* a1, TAny* a2)

	{

	(void)a2;   // a2 not used in this example

	TInt r;

	switch(aReqNo)

		{

		case RLegacyLoopbackDriver::ESendData:

			r=SendData(aStatus,(const TDesC8*)a1);

			break;

		case RLegacyLoopbackDriver::EReceiveData:

			// Example Platform Security capability check which tests the

			// client for ECapability_None (which always passes)...

			if(iClient->HasCapability(ECapability_None,__PLATSEC_DIAGNOSTIC_STRING("Checked by ESockLoopback")))

				r=ReceiveData(aStatus,(TDes8*)a1);

			else

				r=KErrPermissionDenied;

			break;

		default:

			r = KErrNotSupported;

			break;

		}

	return r;

	}

/**

  Process cancelling of asynchronous requests.

*/

void DESockLoopbackChannel::DoCancel(TUint aMask)

	{

	if(aMask&(1<<RLegacyLoopbackDriver::ESendData))

		SendDataCancel();

	if(aMask&(1<<RLegacyLoopbackDriver::EReceiveData))

		ReceiveDataCancel();

	}

/**

  Process a GetConfig control message. This writes the current driver configuration to a

  RLegacyLoopbackDriver::TConfigBuf supplied by the client.

*/

TInt DESockLoopbackChannel::GetConfig(TDes8* aConfigBuf)

	{

	// Create a structure giving the current configuration

	RLegacyLoopbackDriver::TConfig config;

	CurrentConfig(config);

	// Write the config to the client

	TPtrC8 ptr((const TUint8*)&config,sizeof(config));

	return Kern::ThreadDesWrite(iClient,aConfigBuf,ptr,0,KTruncateToMaxLength,NULL);

	}

/**

  Process a SetConfig control message. This sets the driver configuration using a

  RLegacyLoopbackDriver::TConfigBuf supplied by the client.

*/

TInt DESockLoopbackChannel::SetConfig(const TDesC8* aConfigBuf)

	{

	// Don't allow configuration changes whilst we're busy

	if(iSendDataStatus || iReceiveDataStatus)

		return KErrInUse;

	// Create a config structure.

	RLegacyLoopbackDriver::TConfig config;

	CurrentConfig(config);

	// Note: We have filled config with the current settings, this is to allow

	// backwards compatibility when a client gives us an old (and shorter) version

	// of the config structure.

	// Read the config structure from client

	TPtr8 ptr((TUint8*)&config,sizeof(config));

	TInt r=Kern::ThreadDesRead(iClient,aConfigBuf,ptr,0);

	if(r!=KErrNone)

		return r;

	// Use config data to setup the driver. Checking that parameters which aren't settable

	// either contain the correct values or are zero (meaning 'default')

	if(config.iPddBufferSize && config.iPddBufferSize!=Pdd()->BufferSize())

		return KErrArgument;

	if(config.iMaxSendDataSize && config.iMaxSendDataSize!=KLoopbackMTU)

		return KErrArgument;

	if(config.iMaxReceiveDataSize && config.iMaxReceiveDataSize!=KLoopbackMTU)

		return KErrArgument;

	r=Pdd()->SetSpeed(config.iSpeed);

	if(r!=KErrNone)

		return r;

	return r;

	}

/**

  Fill a TConfig with the drivers current configuration.

*/

void DESockLoopbackChannel::CurrentConfig(RLegacyLoopbackDriver::TConfig& aConfig)

	{

	aConfig.iSpeed = Pdd()->Speed();

	aConfig.iPddBufferSize = Pdd()->BufferSize();

	aConfig.iMaxSendDataSize = KLoopbackMTU;

	aConfig.iMaxReceiveDataSize = KLoopbackMTU;

	}

/**

  Start processing a SendData request.

*/

TInt DESockLoopbackChannel::SendData(TRequestStatus* aStatus,const TDesC8* aData)

	{

	// Check that a 'SendData' isn't already in progress

	PROBE_LDD(0);

	if(iSendDataStatus)

		{

		Kern::ThreadKill(iClient,EExitPanic,ERequestAlreadyPending,KESockLoopbackPanicCategory);

		return KErrInUse;

		}

	// Read data from client into our buffer

	TInt r=Kern::ThreadDesRead(iClient,aData,Pdd()->SendBuffer(),0);

	if(r!=KErrNone)

		return r;

	// Give data to PDD so that it can do the work

	r=Pdd()->RequestDataSend();

	if(r!=KErrNone)

		return r;

	// Save the client request status and return

	PROBE_LDD(1);

	iSendDataStatus = aStatus;

	return KErrNone;

	}

/**

  Cancel a SendData request.

*/

void DESockLoopbackChannel::SendDataCancel()

	{

	if(iSendDataStatus)

		{

		// Tell PDD to stop processing the request

		Pdd()->SendDataCancel();

		// Cancel DFC

		iSendDataDfc.Cancel();

		// Complete clients request

		Kern::RequestComplete(iClient,iSendDataStatus,KErrCancel);

		}

	}

/**

  Called by PDD from ISR to indicate that a SendData operation has completed.

*/

void DESockLoopbackChannel::SendDataComplete(TInt aResult)

	{

	PROBE_LDD(2);

	// Save result code

	iSendDataResult = aResult;

	// Queue DFC

	iSendDataDfc.Add();

	}

/**

  DFC callback which gets triggered after the PDD has signalled that SendData completed.

  This just casts aPtr and calls DoSendDataComplete().

*/

void DESockLoopbackChannel::SendDataDfc(TAny* aPtr)

	{

	PROBE_LDD_STATIC(3, ((DESockLoopbackChannel*)aPtr));

	((DESockLoopbackChannel*)aPtr)->DoSendDataComplete();

	}

/**

  Called from a DFC after the PDD has signalled that SendData completed.

*/

void DESockLoopbackChannel::DoSendDataComplete()

	{

	TInt result = iSendDataResult;

	// Complete clients request

	Kern::RequestComplete(iClient,iSendDataStatus,result);

	}

/**

  Start processing a ReceiveData request.

*/

TInt DESockLoopbackChannel::ReceiveData(TRequestStatus* aStatus,TDes8* aPtr)

	{

	PROBE_LDD(4);

	// Check that a 'ReceiveData' isn't already in progress

	if(iReceiveDataStatus)

		{

		Kern::ThreadKill(iClient,EExitPanic,ERequestAlreadyPending,KESockLoopbackPanicCategory);

		return KErrInUse;

		}

	// Save the client request status and descriptor for later completion

	iReceiveDataStatus = aStatus;

	iReceiveDataDescriptor = aPtr;

	if(Pdd()->ReceivedQueueLen() > 0)

		{

		// complete with waiting data

		DoReceiveDataComplete();

		}

	// Ask PDD for data

	NKern::Lock(); 

	TInt r = Pdd()->RequestDataReceipt();

	NKern::Unlock(); 

	if(r != KErrNone)

		{

		iReceiveDataDescriptor = NULL;

		return r;

		}

	PROBE_LDD(5);

	return KErrNone;

	}

/**

  Cancel a ReceiveData request.

*/

void DESockLoopbackChannel::ReceiveDataCancel()

	{

	if(iReceiveDataStatus)

		{

		// Tell PDD to stop processing the request

		Pdd()->ReceiveDataCancel();

		// Cancel DFC

		iReceiveDataDfc.Cancel();

		// Finished with client descriptor, so NULL it to help detect coding errors

		iReceiveDataDescriptor = NULL;

		// Complete clients request

		Kern::RequestComplete(iClient,iReceiveDataStatus,KErrCancel);

		}