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

//

#include <f32file.h>

#include <d32usbdi_hubdriver.h>

#include <d32usbdi.h>

#include <d32otgdi.h>

#include <d32usbdescriptors.h>

#include <d32usbtransfers.h>

#include <shared.h>

#include "rusbhostmsdevice.h"

#include "rextfilesystem.h"

#include "usbtypes.h"

#include "cusbmsmountmanager.h"

#include "tmslog.h"

#include "debug.h"

CDevice* CDevice::NewL()

    {

    __MSFNSLOG

	CDevice* r = new (ELeave) CDevice();

	CleanupStack::PushL(r);

	r->ConstructL();

	CleanupStack::Pop();

	return r;

    }

void CDevice::ConstructL()

    {

    __MSFNLOG

    }

CDevice::CDevice()

    {

    __MSFNLOG

    }

CDevice::~CDevice()

    {

    __MSFNLOG

    __USBHOSTPRINT1(_L("~CDevice Token=%d"), iDeviceToken);

    }

TToken CDevice::OpenDeviceL(TUint aDeviceHandle, RUsbHubDriver& aHub)

    {

    __MSFNLOG

    __USBHOSTPRINT1(_L("CDevice::OpenDeviceL Handle=%d"), aDeviceHandle);

    TInt err = iUsbDevice.Open(aHub, aDeviceHandle);

    __USBHOSTPRINT1(_L(" - returned %d\n"), err);

    User::LeaveIfError(err);

    /* Retrieve the device descriptor */

    TUsbDeviceDescriptor devDescriptor;

    User::LeaveIfError(iUsbDevice.GetDeviceDescriptor(devDescriptor));

    iUsbPrint.PrintDescriptor(devDescriptor, 0, &iUsbDevice);

    iUsbPrint.PrintTree(devDescriptor);

    /* Retrieve the configuration descriptor */

    TUsbConfigurationDescriptor configDescriptor;

    User::LeaveIfError(iUsbDevice.GetConfigurationDescriptor(configDescriptor));

    iUsbPrint.PrintDescriptor(configDescriptor, 0, &iUsbDevice);

    /* Get the token for interface 0 */

    TUint32 token;

    err = iUsbDevice.GetTokenForInterface(0, token);

    __USBHOSTPRINT2(_L("RUsbDevice::GetTokenForInterface returned error %d, token %08x"), err, token);

    User::LeaveIfError(err);

    /* open the interface */

    RUsbInterface interface_ep0;

    err = interface_ep0.Open(token);

    __USBHOSTPRINT1(_L("RUsbInterface::Open returned error %d"), err);

    User::LeaveIfError(err);

    /* Retrieve the interface and device descriptors */

    TUsbInterfaceDescriptor ifDescriptor;

    User::LeaveIfError(interface_ep0.GetInterfaceDescriptor(ifDescriptor));

    iUsbPrint.PrintDescriptor(ifDescriptor);

    if (!IsDeviceMassStorage(ifDescriptor, devDescriptor))

        {

        RDebug::Print(_L("ATTACHED DEVICE IS NOT A MASS STORAGE DEVICE!\n"));

        User::Leave(KErrGeneral);

        }

	TUint8 iProtocolId = ifDescriptor.InterfaceSubClass();

	TUint8 iTransportId = ifDescriptor.InterfaceProtocol();

    interface_ep0.Close();

    THostMassStorageConfig msConfig;

    msConfig.iInterfaceToken = token;

	msConfig.iProtocolId =iProtocolId;

	msConfig.iTransportId = iTransportId;

	msConfig.iStatusPollingInterval = 10; // 10 secs

    TUint32 numLun;

    TRequestStatus status;

    iUsbHostMsDevice.Add(msConfig, status);

	User::WaitForRequest(status);

	if (status.Int() != KErrNone)

        {

		__USBHOSTPRINT(_L("Add device failed"));

		User::Leave(status.Int());

        }

	TInt r = iUsbHostMsDevice.GetNumLun(numLun);

	if (r != KErrNone)

        {

		__USBHOSTPRINT(_L("GetNumLun failed"));

		User::Leave(r);

        }

    if (numLun > KMaxLun)

        {

        __USBHOSTPRINT1(_L("Device MaxLun = %d. Error MaxLun > MAXLUN !"), numLun);

        User::Leave(KErrGeneral);

        }

    __USBHOSTPRINT1(_L("MSC registered with %d Luns"), numLun);

    iDeviceToken = token;

    iNumLuns = numLun;

    iDeviceHandle = aDeviceHandle;

    return token;

    }

void CDevice::CloseDeviceL()

    {

    __MSFNLOG

    THostMassStorageConfig msConfig;

    msConfig.iInterfaceToken = iDeviceToken;

    iUsbHostMsDevice.Remove();

    iUsbDevice.Close();

    }

void CDevice::MountLogicalUnitsL()

    {

    __MSFNLOG

    iExt.OpenL();

    for (TInt lun = 0; lun < iNumLuns; lun++)

        {

        TDriveNumber driveNumber = iExt.GetDriveL();

        __PRINT2(_L("Mounting drive=%d lun=%d..."), driveNumber, lun);

        RDebug::Print(_L("Mounting drive=%d lun=%d..."), driveNumber, lun);

        TRAPD(err, iExt.MountL(iUsbHostMsDevice, driveNumber, iDeviceToken, lun));

        if (err == KErrNone || err == KErrNotReady || err == KErrCorrupt)

            {

            iLuList.Append(driveNumber);

            }

        __PRINT1(_L("%d"), err);

        RDebug::Print(_L("err=%d"), err);

        }

    }

void CDevice::DismountLogicalUnitsL()

    {

    __MSFNLOG

    for (TInt lun = 0; lun < iLuList.Count(); lun++)

        {

        TDriveNumber driveNumber = iLuList[lun];

        iExt.DismountL(iUsbHostMsDevice, driveNumber);

        }

    iLuList.Reset();

    iExt.CloseL();

    }

TInt CDevice::GetEndpointAddress(RUsbInterface& aUsbInterface,

                                 TInt aInterfaceSetting,

                                 TUint8 aTransferType,

                                 TUint8 aDirection,

                                 TInt& aEndpointAddress) const

	{

    __MSFNSLOG

	// Get the interface descriptor

	RDebug::Print(_L("GetEndpointAddress : Getting the interface descriptor for this alternate setting"));

	TUsbInterfaceDescriptor alternateInterfaceDescriptor;

	TInt err = aUsbInterface.GetAlternateInterfaceDescriptor(aInterfaceSetting, alternateInterfaceDescriptor);

	if (err)

		{

		RDebug::Print(_L("GetEndpointAddress : <Error %d> Unable to get alternate interface (%d) descriptor"),err,aInterfaceSetting);

		return err;

		}

	// Parse the descriptor tree from the interface

	RDebug::Print(_L("Search the child descriptors for matching endpoint attributes"));

	TUsbGenericDescriptor* descriptor = alternateInterfaceDescriptor.iFirstChild;

	while (descriptor)

		{

		RDebug::Print(_L("GetEndpointAddress : Check descriptor type for endpoint"));

		// Cast the descriptor to an endpoint descriptor

		TUsbEndpointDescriptor* endpoint = TUsbEndpointDescriptor::Cast(descriptor);

		if (endpoint)

			{

			RDebug::Print(_L("GetEndpointAddress : Match attributes for transfer type"));

			if ( (endpoint->Attributes() & aTransferType) == aTransferType)

				{

				RDebug::Print(_L("GetEndpointAddress : Match attributes for endpoint direction"));

				if ( (endpoint->EndpointAddress() & 0x80) == aDirection)

					{

					aEndpointAddress = endpoint->EndpointAddress();

					RDebug::Print(_L("GetEndpointAddress : Endpoint address found"));

					return KErrNone;

					}

				}

			}

		descriptor = descriptor->iNextPeer;

		}

	// Unable to find the endpoint address

	RDebug::Print(_L("GetEndpointAddress : Unable to find endpoint address matching the specified attributes"));

	return KErrNotFound;

	}

TBool CDevice::IsDeviceMassStorage(const TUsbInterfaceDescriptor& aInterfaceDesc,

                                   const TUsbDeviceDescriptor& aDeviceDesc) const

    {

    __MSFNSLOG

	/* check the interface descriptor */

	if(aInterfaceDesc.InterfaceClass() == 0x08 &&

		aInterfaceDesc.InterfaceSubClass() == 0x06 &&

		aInterfaceDesc.InterfaceProtocol() == 0x50)

        {

		if(aDeviceDesc.DeviceClass() == 0x00 &&

			aDeviceDesc.DeviceSubClass() == 0x00 &&

			aDeviceDesc.DeviceProtocol() == 0x00)

			return ETrue;

        }

	return EFalse;

    }

TLun CDevice::DriveMap(TDriveMap& aDriveMap) const

    {

    __MSFNSLOG

    TDriveNumber driveNumber;

    RDebug::Printf("LuList.Count=%d", iLuList.Count());

    for (TInt i = 0; i < iLuList.Count(); i++)

        {

        driveNumber = iLuList[i];

        aDriveMap[driveNumber] = iDeviceToken;

        RDebug::Printf("Device %d token=%d driveNumber=%d", i, iDeviceToken, driveNumber);

        }

    return iNumLuns;

    }

TLun CDevice::DeviceMap(TDeviceMap& aDeviceMap) const

    {

    __MSFNSLOG

    TDriveNumber driveNumber;

    RDebug::Printf("LuList.Count=%d", iLuList.Count());

    for (TInt i = 0; i < iLuList.Count(); i++)

        {

        driveNumber = iLuList[i];

        aDeviceMap[i] = driveNumber;

        RDebug::Printf("CDevice LUN=%d driveNumber=%d", i, driveNumber);

        }

    return iNumLuns;

    }

CUsbMsMountManager* CUsbMsMountManager::NewL()

    {

    __MSFNSLOG

	CUsbMsMountManager* r = new (ELeave) CUsbMsMountManager();

	CleanupStack::PushL(r);

	r->ConstructL();

	CleanupStack::Pop();

	return r;

    }

void CUsbMsMountManager::ConstructL()

    {

    __MSFNLOG

    }

CUsbMsMountManager::CUsbMsMountManager()

    {

    __MSFNLOG

    }

CUsbMsMountManager::~CUsbMsMountManager()

    {

    __MSFNLOG

    iDeviceList.ResetAndDestroy();

    }

// adds new entry for this device

void CUsbMsMountManager::AddDeviceL(CDevice* aDevice)

    {

    __MSFNLOG

    iDeviceList.Append(aDevice);

    }

CDevice* CUsbMsMountManager::RemoveDeviceL(TUint aDeviceHandle)

    {

    __MSFNLOG

	TInt index = GetHandleIndexL(aDeviceHandle);

	CDevice* device = iDeviceList[index];

    iDeviceList.Remove(index);

    return device;

    }

void CUsbMsMountManager::CloseAllDevicesL()

    {

    __MSFNLOG

    for (TInt i = 0; i < iDeviceList.Count(); i++)

        {

        iDeviceList[i]->CloseDeviceL();

        }

    }

TInt CUsbMsMountManager::GetDeviceIndexL(TToken aDeviceToken) const

    {

    __MSFNSLOG

    TInt index;

    for (index = 0; index < iDeviceList.Count(); index++)

        {

        if (aDeviceToken == iDeviceList[index]->DeviceToken())

            {

            break;

            }

        }

    if (index == iDeviceList.Count())

        {

        User::Leave(KErrNotFound);

        }

    return index;

    }

TInt CUsbMsMountManager::GetHandleIndexL(TUint aDeviceHandle) const

    {

    __MSFNSLOG

    TInt index;

    for (index = 0; index < iDeviceList.Count(); index++)

        {

        if (aDeviceHandle == iDeviceList[index]->DeviceHandle())

            {

            break;

            }

        }

    if (index == iDeviceList.Count())

        {

        User::Leave(KErrNotFound);

        }

    return index;

    }

// mounts all LUNs for the device

void CUsbMsMountManager::MountDeviceL(TUint aDeviceHandle)

    {

    __MSFNLOG

    TInt index = GetHandleIndexL(aDeviceHandle);

    iDeviceList[index]->MountLogicalUnitsL();

    }

// dismount all LUNs for this device

void CUsbMsMountManager::DismountDeviceL(TUint aDeviceHandle)

    {

    __MSFNLOG

    TInt index = GetHandleIndexL(aDeviceHandle);

    iDeviceList[index]->DismountLogicalUnitsL();

    }

// dismount all LUNs

void CUsbMsMountManager::DismountL()

    {

    __MSFNLOG

    for (TInt i = 0; i < iDeviceList.Count(); i++)

        {

        iDeviceList[i]->DismountLogicalUnitsL();

        }

    }

void CUsbMsMountManager::DriveMap(TDriveMap& aDriveMap) const

    {

    __MSFNSLOG

    TInt maxLun = 0;

    RDebug::Printf("DeviceList.Count=%d", iDeviceList.Count());

    for (TInt i = 0; i < iDeviceList.Count(); i++)

        {

        maxLun = iDeviceList[i]->DriveMap(aDriveMap);

        RDebug::Printf("%d %d", i, maxLun);

        }

    }

void CUsbMsMountManager::DeviceMap(TInt aDeviceIndex, TDeviceMap& aDeviceMap) const

    {

    __MSFNSLOG

    RDebug::Printf("Device=%d", aDeviceIndex);

    __ASSERT_DEBUG(aDeviceIndex < iDeviceList.Count(), User::Invariant());

    iDeviceList[aDeviceIndex]->DeviceMap(aDeviceMap);

    }

TUsbPrint::TUsbPrint()

:   iDebug(EFalse)

    {

    }

void TUsbPrint::PrintTree(const TUsbGenericDescriptor& aDesc, TInt aDepth)

	{

    if (!iDebug)

        {

        return;

        }

	TBuf<20> buf;

	for(TInt depth=aDepth;depth>=0;--depth)

		{

		buf.Append(_L("  "));

		}

	if(aDesc.iRecognisedAndParsed == TUsbGenericDescriptor::ERecognised)

		{

		RDebug::Print(_L("%S+0x%08x - %d 0x%02x"), &buf, &aDesc, aDesc.ibLength, aDesc.ibDescriptorType);

		}

	else

		{

		RDebug::Print(_L("%S-0x%08x - %d 0x%02x"), &buf, &aDesc, aDesc.ibLength, aDesc.ibDescriptorType);

		}

	HBufC* blob = HBufC::New(5*aDesc.iBlob.Length()); // 5* for " 0x" + 2*bytes for hex representation

	if(blob)

		{

		for(TInt i=0;i<aDesc.iBlob.Length();++i)

			{

			blob->Des().AppendFormat(_L("0x%02x "), aDesc.iBlob[i]);

			}

		RDebug::Print(_L("%S >%S"), &buf, blob);

		delete blob;

		}

	if(aDesc.iFirstChild)

		{

		RDebug::Print(_L("%S \\ "), &buf);

		PrintTree(*(aDesc.iFirstChild), aDepth+1);

		RDebug::Print(_L("%S / "), &buf);

		}

	if(aDesc.iNextPeer)

		{

		PrintTree(*(aDesc.iNextPeer), aDepth);

		}

	}

static TUint16 gLangId = 0x0000;

void TUsbPrint::SetLanguageToPrintL(RUsbDevice& aDevice)

	{

    if (!iDebug) return;

	// Try to set language to US Eng, otherwise take the first one listed.

	if(gLangId == 0x0000) // Only make the request if not been made before.

		{

		// Get string descriptor 0.

		TBuf8<256> stringBuf;

		TUsbStringDescriptor* stringDesc = NULL;

		User::LeaveIfError(aDevice.GetStringDescriptor(stringDesc, stringBuf, 0));

		CleanupStack::PushL(*stringDesc);

		// Search for US English

		TBool usEngLang = EFalse;