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

// @file testinterfacebase.cpp

// @internalComponent

// 

//

#include "testdevicebase.h"

#include "testinterfacebase.h"

#include "testinterfacesettingbase.h"

#include "testdebug.h"

#include "controltransferrequests.h"

#include "endpointwriter.h"

namespace NUnitTesting_USBDI

	{

CInterfaceBase::CInterfaceBase(RUsbTestDevice& aTestDevice,const TDesC16& aName)

:	iDevice(aTestDevice),

	iInterfaceName(aName),

	iCurrentAlternateInterfaceSetting(0) // The default alternate interface setting will be zero when opened

	{

	}

CInterfaceBase::~CInterfaceBase()

	{

	delete iAuxBuffer;

	delete iStallWatcher;

	delete iSelectionWatcher;

	// Release all interfaces

	delete iEp0Writer;

	// Destroy the endpoint 0 reader

	delete iEp0Reader;

	// Destroy interface settings

	iAlternateSettings.ResetAndDestroy();

	// Close the channel to the driver

	iClientDriver.Close();

	}

void CInterfaceBase::BaseConstructL()

	{

	// Open channel to driver

	TInt err(iClientDriver.Open(0));

	if(err != KErrNone)

		{

		User::Leave(err);

		}

	// Start the watcher for host selecting alternate interface settings

	iSelectionWatcher = CAlternateInterfaceSelectionWatcher::NewL(iClientDriver,*this);

	//

	iStallWatcher = new (ELeave) CEndpointStallWatcher(iClientDriver);

	// Create the Ep0 reader/writer

	iEp0Reader = new (ELeave) CControlEndpointReader(iClientDriver,*this);

	iEp0Writer = new (ELeave) CEndpointWriter(iClientDriver,EEndpoint0);

	// Hide bus from host while interfaces are being set up

	iClientDriver.DeviceDisconnectFromHost();

	}

void CInterfaceBase::AddInterfaceSettingL(CInterfaceSettingBase* aInterfaceSetting)

	{

	// Append to the container

	TInt err(iAlternateSettings.Append(aInterfaceSetting));

	if(err != KErrNone)

		{

		User::Leave(err);

		}

	// Get the current number of alternate interface settings	

	TInt alternateSettingNumber(iAlternateSettings.Count()-1);

	TUint endpointSettingCount(aInterfaceSetting->iInterfaceInfo().iTotalEndpointsUsed);

	if(endpointSettingCount > 0)

		{

		// Device capabilities

		TUsbDeviceCaps devCaps;

		err = iClientDriver.DeviceCaps(devCaps);

		if(err != KErrNone)

			{

			User::Leave(err);

			}

		// Endpoint capabilities

		TUsbcEndpointData endpointCaps[KUsbcMaxEndpoints];

		TPtr8 dataptr(reinterpret_cast<TUint8*>(endpointCaps), sizeof(endpointCaps), sizeof(endpointCaps));

		err = iClientDriver.EndpointCaps(dataptr);

		if(err != KErrNone)

			{

			User::Leave(err);

			}		

		TInt totalEndpoints(devCaps().iTotalEndpoints);

		// Loop through available hardware endpoints to find suitable one

		// i.e. endpoints that can be configured

		TUint epIndex(0);

		TUint epCount(0);

		for(; epIndex<totalEndpoints; epIndex++)

			{

			const TUsbcEndpointData ep = endpointCaps[epIndex];

			// Check the endpoint index to see if already claimed

			if(!ep.iInUse)

				{			

				const TUsbcEndpointCaps caps(ep.iCaps);

				// Information about the endpoint we are looking for	

				TUsbcEndpointInfo& endpointSpec = aInterfaceSetting->iInterfaceInfo().iEndpointData[epCount];

				if( (caps.iTypesAndDir & (endpointSpec.iDir | endpointSpec.iType)) == 

					(endpointSpec.iDir | endpointSpec.iType) )

					{

					// Found suitable endpoint

					// Create the reader/writer for this endpoint					

					if(endpointSpec.iDir == KUsbEpDirIn)

						{

						// Create an endpoint writer for this endpoint

						aInterfaceSetting->CreateEndpointWriterL(iClientDriver,(epCount+1));

						}

					else if(endpointSpec.iDir == KUsbEpDirOut)

						{

						// Create an endpoint reader for this endpoint

						aInterfaceSetting->CreateEndpointReaderL(iClientDriver,epCount+1);

						}					

					epCount++; // Increment to next endpoint spec

					endpointSpec.iSize = caps.MaxPacketSize();

					if(epCount >= endpointSettingCount)

						{

						// Found all desired endpoints

						break;

						}

					}

				}

			else

				{

				}

			}

		if(epCount < endpointSettingCount)

			{

			User::Leave(KErrNotFound);

			}			

		}

	else

		{

		}

	// Add the new setting to the device

	err = iClientDriver.SetInterface(alternateSettingNumber,aInterfaceSetting->iInterfaceInfo);

	if(err != KErrNone)

		{

		User::Leave(err);

		}

	}

TInt CInterfaceBase::StallEndpoint(TUint16 aEndpointNumber)

	{

	return iClientDriver.HaltEndpoint(static_cast<TEndpointNumber>(aEndpointNumber));

	}

CInterfaceSettingBase& CInterfaceBase::AlternateSetting(TInt aSettingNumber) const

	{

	return *iAlternateSettings[aSettingNumber];

	}

TInt CInterfaceBase::InterfaceSettingCount() const

	{

	return iAlternateSettings.Count();

	}

TUint32 CInterfaceBase::ExtractNumberL(const TDesC8& aPayload)

	{

	// Read the number of repeats and the data supplied by the host, on the specified endpoint

	TLex8 lex(aPayload.Left(KNumberStringLength));

	TUint32 numBytes;

	User::LeaveIfError(lex.Val(numBytes, EDecimal));

	return numBytes;

	}

void CInterfaceBase::ExtractTwoNumbersL(const TDesC8& aPayload, TUint32& aFirstNum, TUint32& aSecondNum)

	{

	// Read the number of repeats and the data supplied by the host, on the specified endpoint

	TLex8 lex1(aPayload.Left(KNumberStringLength));

	User::LeaveIfError(lex1.Val(aFirstNum, EDecimal));

	TLex8 lex2(aPayload.Mid(KNumberStringLength, KNumberStringLength));

	User::LeaveIfError(lex2.Val(aSecondNum, EDecimal));

	return;

	}

void CInterfaceBase::AlternateInterfaceSelectedL(TInt aAlternateInterfaceSetting)

	{

	iCurrentAlternateInterfaceSetting = aAlternateInterfaceSetting;

	}

TInt CInterfaceBase::ProcessRequestL(TUint8 aRequest,TUint16 aValue,TUint16 aIndex,

	TUint16 aDataReqLength,const TDesC8& aPayload)

	{

	switch(aRequest)

		{

		case KVendorEmptyRequest:

			// Acknowledge the request and do nothing

			iEp0Reader->Acknowledge();

			break;

		case KVendorPutPayloadRequest:

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			if(aPayload.Compare(_L8("DEADBEEF")) != 0)

				{

				iDevice.ReportError(KErrCorrupt);

				}

			break;

		case KVendorGetPayloadRequest:

			{

			__ASSERT_DEBUG(iAuxBuffer, User::Panic(_L("Trying to write non-allocated buffer"), KErrGeneral));

			//Perform synchronous write to EP0

			//This allows the subsequent 'Read' request to

			//take place

			TInt ret = iEp0Writer->WriteSynchronous(*iAuxBuffer, ETrue);

			}

			break;

		case KVendorGetRecordedNumBytesReadInPayloadRequest:

			{

			delete iAuxBuffer;

			iAuxBuffer = HBufC8::NewL(KNumberStringLength);

			TPtr8 ptr(iAuxBuffer->Des());

			TInt retValue = 0;

			retValue = AlternateSetting(iCurrentAlternateInterfaceSetting).NumBytesReadSoFarL(aValue);

			ptr.Zero();

			ptr.Format(KNumberFormatString, retValue);

			//Perform synchronous write to EP0

			//This allows the subsequent 'Read' request to

			//take place

			TInt ret = iEp0Writer->WriteSynchronous(*iAuxBuffer, ETrue);

			}

			break;

		case KVendorGetRecordedNumBytesWrittenInPayloadRequest:

			{

			delete iAuxBuffer;

			iAuxBuffer = HBufC8::NewL(KNumberStringLength);

			TPtr8 ptr(iAuxBuffer->Des());

			TInt retValue = 0;

			retValue = AlternateSetting(iCurrentAlternateInterfaceSetting).NumBytesWrittenSoFarL(aValue);

			ptr.Zero();

			ptr.Format(KNumberFormatString, retValue);

			//Perform synchronous write to EP0

			//This allows the subsequent 'Read' request to

			//take place

			TInt ret = iEp0Writer->WriteSynchronous(*iAuxBuffer, ETrue);

			}

			break;

		case KVendorWriteToEndpointRequest:

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			// Write the data supplied by the host, back to the host though the specified endpoint

			AlternateSetting(iCurrentAlternateInterfaceSetting).WriteSpecifiedDataToEndpointL(aPayload,aValue);

			break;

		case KVendorCancelWriteToEndpointRequest:

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			// CANCEL writing the data supplied by the host, back to the host though the specified endpoint

			AlternateSetting(iCurrentAlternateInterfaceSetting).CancelWriteDataToEndpointL(aValue);

			break;

		case KVendorPatternWriteToEndpointRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			// Read the number of repeats and the data supplied by the host, on the specified endpoint

			TUint32 numBytes = ExtractNumberL(aPayload);

			// Write the data supplied by the host, back to the host though the specified endpoint

			AlternateSetting(iCurrentAlternateInterfaceSetting).WriteSpecifiedDataToEndpointL(aPayload.Mid(KNumberStringLength),numBytes,aValue);

			}

			break;

		case KVendorPatternWriteSynchronousToEndpointRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			// Read the number of repeats and the data supplied by the host, on the specified endpoint

			TUint32 numBytes = ExtractNumberL(aPayload);

			// Write the data supplied by the host, back to the host though the specified endpoint

			AlternateSetting(iCurrentAlternateInterfaceSetting).WriteSynchronousSpecifiedDataToEndpointL(aPayload.Mid(KNumberStringLength),numBytes,aValue);

			}

			break;

		case KVendorPatternWriteSynchronousToAndHaltEndpointRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			// Read the number of repeats and the data supplied by the host, on the specified endpoint

			TUint32 numBytes = ExtractNumberL(aPayload);

			// Write the data supplied by the host, back to the host though the specified endpoint

			AlternateSetting(iCurrentAlternateInterfaceSetting).WriteSynchronousSpecifiedDataToAndHaltEndpointL(aPayload.Mid(KNumberStringLength),numBytes,aValue);

			}

			break;

		case KVendorRepeatedReadAndValidateDataRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			// Read the number of bytes to read in total and per individual 'Read' together with the data supplied by the host, on the specified endpoint

			TUint32 numBytesPerRead = 0;

			TUint32 totalNumBytes = 0;

			ExtractTwoNumbersL(aPayload, numBytesPerRead, totalNumBytes);

			// Write the data supplied by the host, back to the host though the specified endpoint

			AlternateSetting(iCurrentAlternateInterfaceSetting).RepeatedReadAndValidateFromEndpointL(aPayload.Mid(KTwoNumberStringLength),numBytesPerRead,totalNumBytes,aValue);

			}

			break;

		case KVendorRepeatedPatternWriteDataRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			// Read the number of bytes to write in total and per individual 'Write' together with the data supplied by the host, on the specified endpoint

			TUint32 numBytesPerWrite = 0;

			TUint32 totalNumBytes = 0;

			ExtractTwoNumbersL(aPayload, numBytesPerWrite, totalNumBytes);

			// Write the data supplied by the host, back to the host though the specified endpoint

			AlternateSetting(iCurrentAlternateInterfaceSetting).RepeatedWriteSpecifiedDataToEndpointL(aPayload.Mid(KTwoNumberStringLength),numBytesPerWrite,totalNumBytes,aValue);

			}

			break;

		case KVendorWriteCachedReadRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			TUint16 readEndpoint = aValue >> 8; //HI 8 buts

			TUint16 writeEndpoint = aValue & 0x00ff; //LO 8 bits

			// Write the data supplied by the host, back to the host though the specified endpoint

			AlternateSetting(iCurrentAlternateInterfaceSetting).WriteCachedEndpointDataToEndpointL(readEndpoint,writeEndpoint);

			}

			break;

		case KVendorWriteSynchronousCachedReadRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			TUint16 readEndpoint = aValue >> 8; //HI 8 buts

			TUint16 writeEndpoint = aValue & 0x00ff; //LO 8 bits

			// Write the data supplied by the host, back to the host though the specified endpoint

			AlternateSetting(iCurrentAlternateInterfaceSetting).WriteSynchronousCachedEndpointDataToEndpointL(readEndpoint,writeEndpoint);

			}

			break;

		case KVendorSplitWriteSynchronousCachedReadRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			TUint16 readEndpoint = aValue >> 8; //HI 8 buts

			TUint16 writeEndpoint = aValue & 0x00ff; //LO 8 bits

			// Read the number of bytes to use for each Write

			TUint numBytes[KNumSplitWriteSections];

			TUint numBytesWritten = 0;

			for(TUint i=0; i<KNumSplitWriteSections; ++i)

				{

				TLex8 lex(aPayload.Mid(i*KNumberStringLength, KNumberStringLength));

				User::LeaveIfError(lex.Val(numBytes[i], EDecimal));

				// Write the data supplied by the host, back to the host though the specified endpoint

				AlternateSetting(iCurrentAlternateInterfaceSetting).WriteSynchronousCachedEndpointDataToEndpointL(readEndpoint,writeEndpoint,numBytesWritten,numBytes[i]);

				// Updates bytes written for next round of 'for'loop

				numBytesWritten += numBytes[i];

				}

			}

			break;

		case KVendorReadFromEndpointRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			// Read the amount of data supplied by the host, on the specified endpoint

			TLex8 lex(aPayload);

			TUint32 numBytes;

			User::LeaveIfError(lex.Val(numBytes, EDecimal));

			AlternateSetting(iCurrentAlternateInterfaceSetting).ReadDataFromEndpointL(numBytes,aValue);

			}

			break;

		case KVendorReadFromAndHaltEndpointRequest:

			{

			// Acknowledge the request

			iEp0Reader->Acknowledge();

			// Read the amount of data supplied by the host, on the specified endpoint

			TLex8 lex(aPayload);