// Copyright (c) 2008-2009 Nokia Corporation and/or its subsidiary(-ies).

// All rights reserved.

// This component and the accompanying materials are made available

// under the terms of the License "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:

// e32\e32test\demandpaging\d_pagingexample_2_post.cpp

// Demand paging migration example device driver d_pagingexample_2_post: a

// DLogicalChannelBase-dervied driver, post-migration

// 

//

#include "d_pagingexample.h"

#include <kernel/kernel.h>

#include <kernel/kern_priv.h>

const TInt KDfcQThreadPriority = 25;

const TInt KBufferSize = KMaxTransferSize;

//

// Logical channel

//

class DExampleChannel : public DLogicalChannelBase

	{

public:

	typedef RPagingExample::TConfigData TConfigData;

	typedef RPagingExample::TValueStruct TValueStruct;

public:

	DExampleChannel();

	~DExampleChannel();

	virtual TInt DoCreate(TInt aUnit, const TDesC8* anInfo, const TVersion& aVer);

	virtual TInt Request(TInt aReqNo, TAny* a1, TAny* a2);

	TInt DoControl(TInt aFunction, TAny* a1, TAny* a2);

	TInt DoRequest(TInt aFunction, TRequestStatus* aStatus, TAny* a1, TAny* a2);

	TInt DoCancel(TUint aMask);

private:

 	TDfcQue* DfcQ();

	void Shutdown();

	void GetConfig(TConfigData&);

	void SetConfig(const TConfigData&);

	TInt StartNotify(TRequestStatus* aStatus);

	TInt StartAsyncGetValue(TInt* aValue, TRequestStatus* aStatus);

	TInt StartAsyncGetValue2(TInt* aValue1, TInt* aValue2, TRequestStatus* aStatus);

	TInt StartRead(TAny* aBuffer, TInt aLength, TRequestStatus* aStatus);

	TInt StartReadDes(TDes8* aDesOut, TRequestStatus* aStatus);

	TInt StartWrite(TAny* aBuffer, TInt aLength, TRequestStatus* aStatus);

	TInt StartWriteDes(const TDesC8* aDesIn, TRequestStatus* aStatus);

	void ReceiveToReadBuffer();

	void SendFromWriteBuffer();

	static void CancelDfcFunc(TAny* aPtr);

	static void AsyncGetValueCompleteDfcFunc(TAny* aPtr);

	static void AsyncGetValue2CompleteDfcFunc(TAny* aPtr);

	static void ReceiveCompleteDfcFunc(TAny* aPtr);

	static void SendCompleteDfcFunc(TAny* aPtr);

	void CompleteNotify();

	void CompleteAsyncGetValue();

	void CompleteAsyncGetValue2();

	void CompleteRead();

	void CompleteWrite();

	void Cancel();

private:

	NFastMutex iLock;

	TDynamicDfcQue* iDynamicDfcQ;

	TConfigData iConfig;

	DThread* iClient;

	TDfc iCancelDfc;

	// Notify

	TClientRequest* iNotifyRequest;

	// Async get value

	TClientDataRequest<TValueStruct>* iAsyncGetValueRequest;

	NTimer iAsyncGetValueTimer;

	TDfc iAsyncGetValueDfc;

	// Async get value

	TClientDataRequest2<TInt,TInt>* iAsyncGetValue2Request;

	NTimer iAsyncGetValue2Timer;

	TDfc iAsyncGetValue2Dfc;

	// Read

	TClientBufferRequest* iReadRequest;

	NTimer iReadTimer;

	TClientBuffer* iClientReadBuffer;

	TDfc iCompleteReadDfc;

	// Write

	TClientBufferRequest* iWriteRequest;

	NTimer iWriteTimer;

	TClientBuffer* iClientWriteBuffer;

	TDfc iCompleteWriteDfc;

	TUint8 iBuffer[KBufferSize];

	};

DExampleChannel::DExampleChannel() :

	iCancelDfc(CancelDfcFunc, this, 0),

	iAsyncGetValueTimer(NULL, this),

	iAsyncGetValueDfc(AsyncGetValueCompleteDfcFunc, this, 0),

	iAsyncGetValue2Timer(NULL, this),

	iAsyncGetValue2Dfc(AsyncGetValue2CompleteDfcFunc, this, 0),

	iReadTimer(NULL, this),

	iCompleteReadDfc(ReceiveCompleteDfcFunc, this, 0),

	iWriteTimer(NULL, this),

	iCompleteWriteDfc(SendCompleteDfcFunc, this, 0)

	{

	iClient = &Kern::CurrentThread();

	iClient->Open();

	}

DExampleChannel::~DExampleChannel()

	{

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

	if (iDynamicDfcQ)

		iDynamicDfcQ->Destroy();

	Kern::DestroyClientRequest(iNotifyRequest);

	Kern::DestroyClientRequest(iAsyncGetValueRequest);

	Kern::DestroyClientRequest(iAsyncGetValue2Request);

	Kern::DestroyClientBufferRequest(iReadRequest);

	Kern::DestroyClientBufferRequest(iWriteRequest);

	}

TDfcQue* DExampleChannel::DfcQ()

	{

	return iDynamicDfcQ;

	}

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

	{

	TInt r;

	r = Kern::CreateClientRequest(iNotifyRequest);

	if (r != KErrNone)

		return r;

	r = Kern::CreateClientDataRequest(iAsyncGetValueRequest);

	if (r != KErrNone)

		return r;

	r = Kern::CreateClientDataRequest2(iAsyncGetValue2Request);

	if (r != KErrNone)

		return r;

	r = Kern::CreateClientBufferRequest(iWriteRequest, 1, TClientBufferRequest::EPinVirtual);

	if (r != KErrNone)

		return r;

	// create a dynamic DFC queue, which is used for handling client messages and for our own DFCs

	r = Kern::DynamicDfcQCreate(iDynamicDfcQ, KDfcQThreadPriority, KPagingExample1PostLdd);

	if (r != KErrNone)

		return r;

	// todo: this will be the default anyway

	iDynamicDfcQ->SetRealtimeState(ERealtimeStateOn);

	iCancelDfc.SetDfcQ(DfcQ());

	iAsyncGetValueDfc.SetDfcQ(DfcQ());

	iAsyncGetValue2Dfc.SetDfcQ(DfcQ());

	iCompleteReadDfc.SetDfcQ(DfcQ());

	iCompleteWriteDfc.SetDfcQ(DfcQ());

	return KErrNone;

	}

TInt DExampleChannel::Request(TInt aReqNo, TAny* a1, TAny* a2)

	{

	TInt r = KErrNone;

	if (&Kern::CurrentThread() != iClient)

		r = KErrAccessDenied;  // we only support one client

	else if (aReqNo==KMaxTInt)

		{

		// DoCancel

		r = DoCancel((TInt)a1);

		}

    else if (aReqNo<0)

		{

		// DoRequest

		TRequestStatus* pS=(TRequestStatus*)a1;

		TAny* array[2] = { NULL, NULL };

		kumemget32(array, a2, 2 * sizeof(TAny*));

		r = DoRequest(~aReqNo, pS, array[0], array[1]);

		if(r != KErrNone)

			Kern::RequestComplete(pS, r);

		r = KErrNone;

		}

    else

		{

		// DoControl

		r = DoControl(aReqNo, a1, a2);

		}

	return r;

	}

TInt DExampleChannel::DoControl(TInt aFunction,TAny* a1,TAny* /*a2*/)

	{

	TInt r = KErrNone;

	TConfigData configBuffer;

	switch (aFunction)

		{

		case RPagingExample::EGetConfig:

			GetConfig(configBuffer);

			umemput32(a1, (TAny*)&configBuffer, sizeof(TConfigData));

			break;

		case RPagingExample::ESetConfig:

			umemget32(&configBuffer, a1, sizeof(TConfigData));

			SetConfig(configBuffer);

			break;

		default:

			r = KErrNotSupported;

		}

	return r;

	}

TInt DExampleChannel::DoRequest(TInt aFunction, TRequestStatus* aStatus, TAny* a1, TAny* a2)

	{

	TInt r = KErrNotSupported;

	switch (aFunction)

		{

		case RPagingExample::ERequestNotify:

			r = StartNotify(aStatus);

			break;

		case RPagingExample::ERequestAsyncGetValue:

			r = StartAsyncGetValue((TInt*)a1, aStatus);

			break;

		case RPagingExample::ERequestAsyncGetValue2:

			r = StartAsyncGetValue2((TInt*)a1, (TInt*)a2, aStatus);

			break;

		case RPagingExample::ERequestRead:

			r = StartRead(a1, (TInt)a2, aStatus);

			break;

		case RPagingExample::ERequestReadDes:

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

			break;

		case RPagingExample::ERequestWrite:

			r = StartWrite(a1, (TInt)a2, aStatus);

			break;

		case RPagingExample::ERequestWriteDes:

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

			break;

		}

	return r;

	}

TInt DExampleChannel::DoCancel(TUint /*aMask*/)

	{

	iCancelDfc.Enque();

	return KErrNone;

	}

void DExampleChannel::CancelDfcFunc(TAny* aPtr)

	{

	DExampleChannel* self = (DExampleChannel*)aPtr;

	self->Cancel();

	}

void DExampleChannel::Cancel()

	{

	if (iAsyncGetValueRequest->IsReady())

		{

		iAsyncGetValueTimer.Cancel();

		iAsyncGetValueDfc.Cancel();

		Kern::QueueRequestComplete(iClient, iAsyncGetValueRequest, KErrCancel);

		}

	if (iAsyncGetValue2Request->IsReady())

		{

		iAsyncGetValue2Timer.Cancel();

		iAsyncGetValue2Dfc.Cancel();

		Kern::QueueRequestComplete(iClient, iAsyncGetValue2Request, KErrCancel);

		}

	if (iReadRequest && iReadRequest->IsReady())

		{

		iReadTimer.Cancel();

		iCompleteReadDfc.Cancel();

		Kern::QueueBufferRequestComplete(iClient, iReadRequest, KErrCancel);

		Kern::DestroyClientBufferRequest(iReadRequest);

		}

	if (iWriteRequest->IsReady())

		{

		iWriteTimer.Cancel();

		iCompleteWriteDfc.Cancel();

		Kern::QueueBufferRequestComplete(iClient, iWriteRequest, KErrCancel);

		}

	}

void DExampleChannel::Shutdown()

	{

	}

void DExampleChannel::GetConfig(TConfigData& aConfigOut)

	{

	NKern::FMWait(&iLock);

	aConfigOut = iConfig;

	NKern::FMSignal(&iLock);

	}

void DExampleChannel::SetConfig(const TConfigData& aNewConfig)

	{

	NKern::FMWait(&iLock);

	iConfig = aNewConfig;

	NKern::FMSignal(&iLock);

	}

TInt DExampleChannel::StartNotify(TRequestStatus* aStatus)

	{

	// example implementation completes the request immediately

	TInt r = iNotifyRequest->SetStatus(aStatus);

	if (r != KErrNone)

		return r;

	CompleteNotify(); // example implementation completes the request immediately

	return KErrNone;

	}

void DExampleChannel::CompleteNotify()

	{

	Kern::QueueRequestComplete(iClient, iNotifyRequest, KErrNone);

	}

TInt DExampleChannel::StartAsyncGetValue(TInt* aValue, TRequestStatus* aStatus)

	{

	// use of TClientDataRequest API protected by fast mutex

	NKern::FMWait(&iLock);

	TInt r = iAsyncGetValueRequest->SetStatus(aStatus);

	if (r == KErrNone)

		iAsyncGetValueRequest->SetDestPtr(aValue);

	NKern::FMSignal(&iLock);

	if (r != KErrNone)

		return r;

	// queue a timer to simulate an asynchronous operation

	iAsyncGetValueTimer.OneShot(KAsyncDelay, iAsyncGetValueDfc);

	return KErrNone;

	}

void DExampleChannel::AsyncGetValueCompleteDfcFunc(TAny* aPtr)

	{

	DExampleChannel* self = (DExampleChannel*)aPtr;

	self->CompleteAsyncGetValue();

	}

void DExampleChannel::CompleteAsyncGetValue()

	{

	// use of TClientDataRequest API protected by fast mutex

	NKern::FMWait(&iLock);

	if (iAsyncGetValueRequest->IsReady())

		{

		iAsyncGetValueRequest->Data().iValue1 = 1;

		iAsyncGetValueRequest->Data().iValue2 = _L8("shrt");

		Kern::QueueRequestComplete(iClient, iAsyncGetValueRequest, KErrNone);

		}

	NKern::FMSignal(&iLock);

	}

TInt DExampleChannel::StartAsyncGetValue2(TInt* aValue1, TInt* aValue2, TRequestStatus* aStatus)

	{

	// use of TClientDataRequest API protected by fast mutex

	NKern::FMWait(&iLock);

	TInt r = iAsyncGetValue2Request->SetStatus(aStatus);

	if (r == KErrNone)

		iAsyncGetValue2Request->SetDestPtr1(aValue1);

	if (r == KErrNone)

		iAsyncGetValue2Request->SetDestPtr2(aValue2);

	NKern::FMSignal(&iLock);

	if (r != KErrNone)

		return r;

	// queue a timer to simulate an asynchronous operation

	iAsyncGetValue2Timer.OneShot(KAsyncDelay, iAsyncGetValue2Dfc);

	return KErrNone;

	}

void DExampleChannel::AsyncGetValue2CompleteDfcFunc(TAny* aPtr)

	{

	DExampleChannel* self = (DExampleChannel*)aPtr;

	self->CompleteAsyncGetValue2();

	}

void DExampleChannel::CompleteAsyncGetValue2()

	{

	// use of TClientDataRequest API protected by fast mutex

	NKern::FMWait(&iLock);

	if (iAsyncGetValue2Request->IsReady())

		{

		iAsyncGetValue2Request->Data1() = 1;

		iAsyncGetValue2Request->Data2() = 2;

		Kern::QueueRequestComplete(iClient, iAsyncGetValue2Request, KErrNone);

		}

	NKern::FMSignal(&iLock);

	}

TInt DExampleChannel::StartRead(TAny* aBuffer, TInt aLength, TRequestStatus* aStatus)

	{

	// check length argument first

	if (aLength < 1 || aLength > KBufferSize)

		return KErrArgument;

	// normally drivers would pre-create a TClientBufferRequest where possible, but here we create

	// one on demand to test this possibilty

	NKern::ThreadEnterCS();

	TInt r = Kern::CreateClientBufferRequest(iReadRequest, 1, TClientBufferRequest::EPinVirtual);

	NKern::ThreadLeaveCS();  // iReadRequest is deleted by the destructor

	if (r != KErrNone)

		return r;

	// start request setup

	r = iReadRequest->StartSetup(aStatus);

	if (r != KErrNone)

		return r;

	// add client buffer, this does pinning in context of client thread

	r = iReadRequest->AddBuffer(iClientReadBuffer, (TLinAddr)aBuffer, aLength, ETrue);

	if (r != KErrNone)

		return KErrNoMemory;

	iReadRequest->EndSetup();

	ReceiveToReadBuffer();

	return KErrNone;

	}

TInt DExampleChannel::StartReadDes(TDes8* aDesOut, TRequestStatus* aStatus)

	{

	// normally drivers would pre-create a TClientBufferRequest where possible, but here we create

	// one on demand to test this possibilty

	NKern::ThreadEnterCS();

	TInt r = Kern::CreateClientBufferRequest(iReadRequest, 1, TClientBufferRequest::EPinVirtual);

	NKern::ThreadLeaveCS();  // iReadRequest is deleted by the destructor

	if (r != KErrNone)

		return r;

	// start request setup

	r = iReadRequest->StartSetup(aStatus);

	if (r != KErrNone)

		return r;

	// add client descriptor, this does pinning in context of client thread

	r = iReadRequest->AddBuffer(iClientReadBuffer, aDesOut);

	if (r != KErrNone)

		return r;

	// can check length argument here

	TInt length = iClientReadBuffer->MaxLength();

	if (length < 1 || length > KBufferSize)

		{

		// need to reset object so it can be reused

		iReadRequest->Reset();

		return KErrArgument;

		}

	iReadRequest->EndSetup();	

	ReceiveToReadBuffer();

	return KErrNone;

	}

TInt DExampleChannel::StartWrite(TAny* aBuffer, TInt aLength, TRequestStatus* aStatus)

	{

	// check length argument first

	if (aLength < 1 || aLength > KBufferSize)

		return KErrArgument;

	// demonstrate use the single-buffer version of Setup

	TInt r = iWriteRequest->Setup(iClientWriteBuffer, aStatus, (TLinAddr)aBuffer, aLength);

	if (r != KErrNone)

		return r;

	SendFromWriteBuffer();

	return KErrNone;

	}

TInt DExampleChannel::StartWriteDes(const TDesC8* aDesIn, TRequestStatus* aStatus)

	{

	// demonstrate use the single-buffer version of Setup

	TInt r = iWriteRequest->Setup(iClientWriteBuffer, aStatus, (TAny*)aDesIn);

	if (r != KErrNone)

		return r;

	// can check length argument here

	TInt length = iClientWriteBuffer->Length();

	if (length < 1 || length > KBufferSize)

		{

		// need to reset object so it can be reused

		iWriteRequest->Reset();

		return KErrArgument;

		}

	SendFromWriteBuffer();

	return KErrNone;

	}

void DExampleChannel::ReceiveToReadBuffer()

	{

	// just queue a timer to simulate an asynchronous receive operation

	// actually will return the previous contents of the buffer

	NKern::FMWait(&iLock);

	// The synchonisation is for illustrative purposes only - in a real driver we might make use of

	// the configution here, so this would need to be protected from concurrent modification.

	NKern::FMSignal(&iLock);

	iReadTimer.OneShot(KAsyncDelay, iCompleteReadDfc);

	}

void DExampleChannel::SendFromWriteBuffer()

	{

	// just queue a timer to simulate an asynchronous send operation

	NKern::FMWait(&iLock);

	// The synchonisation is for illustrative purposes only - in a real driver we might make use of

	// the configution here, so this would need to be protected from concurrent modification.

	NKern::FMSignal(&iLock);

	iWriteTimer.OneShot(KAsyncDelay, iCompleteWriteDfc);

	}

void DExampleChannel::ReceiveCompleteDfcFunc(TAny* aPtr)

	{

	DExampleChannel* self = (DExampleChannel*)aPtr;

	self->CompleteRead();