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

// Enables USB mass storage mode. Ends when disconnected by user.

// 

//

#include <massstorage.h>

#include <d32usbc.h>

#include "sdusb.h"

/*

Class constructor

@param None

@return None

*/

CBaseTestSDUsb::CBaseTestSDUsb()

	{

	SetTestStepName(KTestStepUsb);

	}

/*

Test Step Preamble

 - Initialise attribute iDrive

 - Connect to the File Server

@param None

@return EPass if successful or EFail if not

@see TVerdict

*/

TVerdict CBaseTestSDUsb::doTestStepPreambleL()

	{

	SetTestStepResult(EFail);

	if (!InitDriveLetter())

		return TestStepResult();

	if (!InitFileServer())

		return TestStepResult();

	SetTestStepResult(EPass);

	return TestStepResult();

	}

/*

Test step

@param None

@return EPass if successful or EFail if not

@see TVerdict

*/

TVerdict CBaseTestSDUsb::doTestStepL()

	{

	TInt r;

	_LIT(KMsFsy, "MSFS.FSY");

	_LIT(KMsFs, "MassStorageFileSystem");

	// Add MS file system	

	r = iFs.AddFileSystem(KMsFsy);

	if (r != KErrNone && r != KErrAlreadyExists)

		{

		ERR_PRINTF2(_L("AddFileSystem failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	RDevUsbcClient usb;

	// Load the logical device

	_LIT(KDriverFileName,"EUSBC.LDD");

	r = User::LoadLogicalDevice(KDriverFileName);

	if (r != KErrNone && r != KErrAlreadyExists)

		{

		ERR_PRINTF2(_L("LoadLogicalDevice failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	r = usb.Open(0);

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RDevUsbcClient::Open failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	RUsbMassStorage UsbMs;

	TBuf<8>  t_vendorId(_L("vendor"));

	TBuf<16> t_productId(_L("product"));

	TBuf<4>  t_productRev(_L("1.00"));

	TMassStorageConfig msConfig;

	msConfig.iVendorId.Copy(t_vendorId);

	msConfig.iProductId.Copy(t_productId);

	msConfig.iProductRev.Copy(t_productRev);

	// Connect to Mass Storage

	r = UsbMs.Connect();

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RUsbMassStorage::Connect failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	// Start Mass Storage

	r = UsbMs.Start(msConfig);

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RUsbMassStorage::Start failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	TBuf8<KUsbDescSize_Device> deviceDescriptor;

	r = usb.GetDeviceDescriptor(deviceDescriptor);

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RDevUsbcClient::GetDeviceDescriptor failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	const TInt KUsbSpecOffset = 2;

	const TInt KUsbDeviceClassOffset = 4;

	const TInt KUsbVendorIdOffset = 8;

	const TInt KUsbProductIdOffset = 10;

	const TInt KUsbDevReleaseOffset = 12;

	//Change the USB spec number to 2.00

	deviceDescriptor[KUsbSpecOffset]   = 0x00;

	deviceDescriptor[KUsbSpecOffset+1] = 0x02;

	//Change the Device Class, Device SubClass and Device Protocol 

	deviceDescriptor[KUsbDeviceClassOffset] = 0x00;

	deviceDescriptor[KUsbDeviceClassOffset+1] = 0x00;

	deviceDescriptor[KUsbDeviceClassOffset+2] = 0x00;

	//Change the device vendor ID (VID) to 0x0E22 (Symbian)

	deviceDescriptor[KUsbVendorIdOffset]   = 0x22;   // little endian

	deviceDescriptor[KUsbVendorIdOffset+1] = 0x0E;

	//Change the device product ID (PID) to 0x1111

	deviceDescriptor[KUsbProductIdOffset]   = 0x12;

	deviceDescriptor[KUsbProductIdOffset+1] = 0x11;

	//Change the device release number to 3.05

	deviceDescriptor[KUsbDevReleaseOffset]   = 0x05;

	deviceDescriptor[KUsbDevReleaseOffset+1] = 0x03;

	r = usb.SetDeviceDescriptor(deviceDescriptor);

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RDevUsbcClient::SetDeviceDescriptor failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	_LIT16(productID_L, "Symbian USB Mass Storage Device");

	TBuf16<KUsbStringDescStringMaxSize / 2> productID(productID_L);

	// Set product string descriptor

	r = usb.SetProductStringDescriptor(productID);

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RDevUsbcClient::SetProductStringDescriptor failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	TRequestStatus enum_status;

	INFO_PRINTF1(_L("Re-enumerating..."));

	usb.ReEnumerate(enum_status);

	User::WaitForRequest(enum_status);

   	INFO_PRINTF1(_L("Re-enumerating done"));

   	// Mount Mass Storage FS

    r = iFs.DismountFileSystem(_L("fat"), iDrive);

    if (r != KErrNone)

   		{

		ERR_PRINTF2(_L("RFs::DismountFileSystem (FAT) failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	r = iFs.MountFileSystem(KMsFs, iDrive);

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RFs::MountFileSystem (MSFS) failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	// Wait until the USB cable is removed or device is suspended

	TUsbcDeviceState initialStatus;

	r = usb.DeviceStatus(initialStatus);

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RDevUsbcClient::DeviceStatus failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	if (initialStatus == EUsbcDeviceStateUndefined)

		{

		ERR_PRINTF1(_L("USB device status is undefined"));

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	TUint deviceState = initialStatus;

	do {

		TRequestStatus rs;

		usb.AlternateDeviceStatusNotify(rs, deviceState);

		User::WaitForRequest(rs);

		} while (deviceState != EUsbcDeviceStateUndefined && deviceState != EUsbcDeviceStateSuspended);

	// Dismount Mass Storage FS

    r = iFs.DismountFileSystem(KMsFs, iDrive);

    if (r != KErrNone)

   		{

		ERR_PRINTF2(_L("RFs::DismountFileSystem (MSFS) failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	r = iFs.MountFileSystem(_L("fat"), iDrive);

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RFs::MountFileSystem (FAT) failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	// Stop USB Mass Storage

	r = UsbMs.Stop();

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RUsbMassStorage::Stop failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	UsbMs.Close();

	usb.Close();

	r = iFs.RemoveFileSystem(KMsFs);

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("RFs::RemoveFileSystem failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	r = User::FreeLogicalDevice(_L("USBC"));

	if (r != KErrNone)

		{

		ERR_PRINTF2(_L("FreeLogicalDevice failed: %d"), r);

		SetTestStepResult(EFail);

		return TestStepResult();

		}

	SetTestStepResult(EPass);

	return TestStepResult();

	}