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

 // CMassStorageFileSystem implementation.

 // 

 //

 /**

  @file

  @internalTechnology

 */

 #include <f32file.h>

 #include "cmassstoragefilesystem.h"

 #include "cmassstoragemountcb.h"

 #include "massstoragedebug.h"

 #include "massstorage.h"

 _LIT(KMsFsyName, "MassStorageFileSystem");

 _LIT(KMsThreadName, "MassStorageThread");

 _LIT(KMsDeadThreadName, "MassStorageDeadThread");

 _LIT(KMsFsysSemName, "MassStorageSemaphore");

 LOCAL_D const TInt KMsFsyMajorVersionNumber=1;

 LOCAL_D const TInt KMsFsyMinorVersionNumber=0;

 CMassStorageFileSystem::CMassStorageFileSystem()

 	{

 	}

 CMassStorageFileSystem::~CMassStorageFileSystem()

 	{

 	//Kill the controller thread if it exists

 	delete iMassStorageController;

 	delete iMediaChanged;

 	RThread thread;

 	TInt err = thread.Open(KMsThreadName);

 	if (err == KErrNone)

 		{

 		thread.Kill(1); //Parameter is irrelevant

 		}

 	thread.Close();

 	iMsDrives.Close();

 	}

 CMassStorageFileSystem* CMassStorageFileSystem::NewL()

 	{

 	CMassStorageFileSystem*  self = new (ELeave) CMassStorageFileSystem();

 	CleanupStack::PushL(self);

 	self->ConstructL();

 	CleanupStack::Pop(self);

 	return self;

 	}

 void CMassStorageFileSystem::ConstructL()

 	{

 	iMediaChanged = new(ELeave) CArrayFixFlat<TBool>(KMaxDrives);

 	}

 /**

 Set the file system version and name

 @return Any of the standard Symbian error codes.

 */

 TInt CMassStorageFileSystem::Install()

 	{

 	__PRINT(_L("CMassStorageFileSystem::Install In\n"));

 	iVersion=TVersion(KMsFsyMajorVersionNumber, KMsFsyMinorVersionNumber, KF32BuildVersionNumber);

 	TInt err = SetName(&KMsFsyName);

 	__PRINT(_L("CMassStorageFileSystem::Install Out\n"));

 	return err;

 	}

 TInt CMassStorageFileSystem::Remove()

 	{

 	__PRINT(_L("CMassStorageFileSystem::Remove In\n"));

 	TInt err = KErrNone;

 	if (iInstalled)

 		{

 		// Try connecting to the server to send a shutdown message.

 		// - If the class controller has a session in use, this will return KErrInUse

 		RUsbMassStorage usbMs;

 		err = usbMs.Connect();

 		if(err == KErrNone)

 			{

 			err = usbMs.Shutdown();

 			usbMs.Close();

 			if(err == KErrNone)

 				{

 				User::WaitForRequest(iThreadStat);

 				err = iThreadStat.Int();

 				}

 			else

 				{

 				__PRINT1(_L("CMassStorageFileSystem::Remove Shutdown Error %d\n"),err);

 				}

 			}

 		else

 			{

 			__PRINT1(_L("CMassStorageFileSystem::Remove Connect Error %d\n"),err);

 			}

 		}

 	__PRINT(_L("CMassStorageFileSystem::Remove Out\n"));

 	return(err);

 	}

 /**

 Creates a new Mass Storage mount object.

 @return A new CMassStorageMountCB

 @leave KErrNotReady if the Mass Storage controller is not running.

 */

 CMountCB* CMassStorageFileSystem::NewMountL() const

 	{

 	if (!iRunning)

 		{

 		User::Leave(KErrNotReady);

 		}

 	return CMassStorageMountCB::NewL(iMsDrives);

 	}

 /**

 Sets the media attributes and type in the aInfo parameter to those of the specified drive.

 @param anInfo TDriveInfo object to store the drive information.

 @param aDriveNumber The number of the drive to get the information from.

 */

 void CMassStorageFileSystem::DriveInfo(TDriveInfo& aInfo, TInt aDriveNumber) const

 	{

 	__PRINT(_L("CMassStorageFileSystem::DriveInfo In\n"));

 	TLocalDriveCapsV2Buf caps;

 	if (!IsValidLocalDriveMapping(aDriveNumber))

 		{

 		return;

 		}

 	(void)GetLocalDrive(aDriveNumber).Caps(caps);   

     // error ignored as Caps always returns  valid Media and Drive attributes

 	aInfo.iMediaAtt=caps().iMediaAtt;

     aInfo.iType = ::EMediaNotPresent;  // Media is not available to the file system

 	aInfo.iDriveAtt=caps().iDriveAtt;

 	__PRINT(_L("CMassStorageFileSystem::DriveInfo Out\n"));

 	}

 /**

 Returns a reference to the Mass Storage controller.

 @return Reference to the Mass Storage controller.

 */

 CUsbMassStorageController& CMassStorageFileSystem::Controller()

 	{

 	return *iMassStorageController;

 	}

 /**

 Fill iMsDrives with a mapping of lun->drive number for supported mass storage drives

 */

 TInt CMassStorageFileSystem::EnumerateMsDrivesL()

 	{

 	__PRINT(_L("CMassStorageFileSystem::EnumerateMsDrives In\n"));

 	iMsDrives.Reset();

 	TInt driveCount = 0;

 	TLocalDriveCapsV2Buf caps;

 	for (TInt i = EDriveC; i < KMaxDrives; i++)

 		{

 		caps.FillZ();

 		if (IsValidLocalDriveMapping(i))

 			{

 			TInt err = GetLocalDrive(i).Caps(caps);

 			TInt locDrvNum = DriveNumberToLocalDriveNumber(i);

 			__PRINT2(_L("Caps: err=%d, att=%d\n"), err, caps().iDriveAtt);

 			TBool isRemovable = err==KErrNotReady || (caps().iDriveAtt & KDriveAttRemovable);

 			__PRINT2(_L("EnumerateMsDrives: Drive %c: is %sremovable\n"),

 							'A'+i-EDriveA, 

 							isRemovable?_S(""):_S("NOT "));

 			if (isRemovable)

 				{

 				//

 				// STF: Connect to the local drive here.  This gives us the media changed flag, and

 				//		our own TBusLocalDrive object for use by the proxy drive and controller.

 				//

 				TBool& mediaChanged = (*iMediaChanged).ExtendL();

 				mediaChanged = EFalse;

 				TBusLocalDrive* localDrive = new(ELeave) TBusLocalDrive;

 				iLocalDriveForMediaFlag.Append(*localDrive);

 				TInt err=iLocalDriveForMediaFlag[driveCount].Connect(locDrvNum, mediaChanged);

 				if(err == KErrNone)

 					{

 					iMsDrives.Append(i);

 					}

 				driveCount++;

 				}

 			}

 		}

 	__PRINT1(_L("CMassStorageFileSystem::EnumerateMsDrives Out, %d MS drives found\n"), driveCount);

 	return driveCount;

 	}

 TBool CMassStorageFileSystem::IsExtensionSupported() const

 	{

 	return ETrue;

 	}

 /**

 Creates a TrapCleanup and ActiveScheduler and initializes the Mass Storage controller.

 Start the ActiveScheduler.

 @return Any of the standard Symbian error codes.

 */

 TInt CMassStorageFileSystem::InitThread()

 	{

 	__PRINT(_L("CMassStorageFileSystem::InitThread In\n"));

 	//Give the thread a name so we can kill it later

 	User::RenameThread(KMsThreadName);

 	CTrapCleanup* cleanup = CTrapCleanup::New();

 	if (cleanup == NULL)

 		{

 		return KErrNoMemory;

 		}

 	TRAPD(err, InitThreadL());

 	delete cleanup;

 	__PRINT1(_L("CMassStorageFileSystem::InitThread Out, error=%d\n"), err);

 	return err;

 	}

 TInt CMassStorageFileSystem::InitThreadL()

 	{

 	__PRINT(_L("CMassStorageFileSystem::InitThreadL In\n"));

 	RSemaphore gSemThreadReady;

 	TInt ret = gSemThreadReady.OpenGlobal(KMsFsysSemName);

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

 		{

 		User::Leave(ret);

 		}

 	// Determine which drives are available for Mass Storage.

 	// (this also creates a local TBusLocalDrive for use by the drive controller)

 	EnumerateMsDrivesL();

 	CActiveScheduler* sched = new CActiveScheduler;

 	if (sched == NULL)

 		{

 		gSemThreadReady.Signal();

 		User::Leave(KErrNoMemory);

 		}

 	CleanupStack::PushL(sched);

 	CActiveScheduler::Install(sched);

 	iMassStorageController = new CUsbMassStorageController;

 	if (iMassStorageController == NULL)

 		{

 		gSemThreadReady.Signal();

 		User::Leave(KErrNoMemory);

 		}

 	__PRINT(_L("CMassStorageFileSystem::InitThread: Creating Mass Storage Controller\n"));

 	TRAPD(err, iMassStorageController->CreateL(iMsDrives));

 	if (err != KErrNone)

 		{

 		gSemThreadReady.Signal();

 		CActiveScheduler::Install(NULL);

 		User::Leave(err);

 		}

 	CleanupStack::Pop(sched);

 	iRunning = ETrue;

 	gSemThreadReady.Signal();

 	gSemThreadReady.Close();

 	CActiveScheduler::Start();

 //========= stop thread ================

 	delete iMassStorageController;

 	iMassStorageController = NULL;

 	TInt i=0;

 	for (;i<iLocalDriveForMediaFlag.Count();i++)

 		{

 		iLocalDriveForMediaFlag[i].Disconnect();

 		}

 	iLocalDriveForMediaFlag.Reset();

 	(*iMediaChanged).Reset();	

 	delete sched;

 	iRunning = EFalse;

 	__PRINT(_L("CMassStorageFileSystem::InitThread Out\n"));

 	return KErrNone;

 	}

 /**

 Not supported in Mass Storage file system.

 @leave KErrNotReady

 */

 CFileCB* CMassStorageFileSystem::NewFileL() const

 	{

 	__PRINT(_L("CMassStorageFileSystem::NewFileL In\n"));

 	User::Leave(KErrNotReady);

 	return NULL;

 	}

 /**

 Not supported in Mass Storage file system.

 @leave KErrNotReady

 */

 CDirCB* CMassStorageFileSystem::NewDirL() const

 	{

 	User::Leave(KErrNotReady);

 	return NULL;

 	}

 /**

 Not supported in Mass Storage file system.

 @leave KErrNotReady

 */

 CFormatCB* CMassStorageFileSystem::NewFormatL() const

 	{

 	User::Leave(KErrNotReady);

 	return NULL;

 	}

 /**

 Not supported in Mass Storage file system.

 @return KErrNotSupported

 */

 TInt CMassStorageFileSystem::DefaultPath(TDes& /*aPath*/) const

 	{

 	return KErrNotSupported;

 	}

 /**

 Not supported in Mass Storage file system.

 @return KErrNotSupported

 */

 TInt CMassStorageFileSystem::DriveList(TDriveList& /*aList*/) const

 	{

 	return KErrNotSupported;

 	}

 /**

 Thread entry point.

 */

 LOCAL_C TInt MsInitThreadFn(TAny* aPtr)

 	{

 	User::SetCritical(User::ESystemCritical);

 	((CMassStorageFileSystem*)aPtr)->InitThread();

 	//Rename the thread so we can create a new one with the same original name later

 	User::RenameThread(KMsDeadThreadName);

 	return KErrNone;

 	}

 /**

 Standard entry point for file systems.

 Creates a new file system object and starts a new thread for the Mass Storage controller.

 */

 extern "C" EXPORT_C CFileSystem* CreateFileSystem()

 	{

 	__PRINT(_L("CMassStorageFileSystem::CreateFileSystem In\n"));

 	RSemaphore gSemThreadReady;

 	TInt err = gSemThreadReady.CreateGlobal(KMsFsysSemName, 0);

 	if (err != KErrNone)

 		{

 		__PRINT1(_L("CMassStorageFileSystem::CreateFileSystem Out Semaphore Error %d\n"),err);

 		return NULL;

 		}

 	CFileSystem* msFsys = NULL;

 	TRAP(err,  msFsys = CMassStorageFileSystem::NewL());

 	if (err != KErrNone)

 		{

 		__PRINT1(_L("CMassStorageFileSystem::CreateFileSystem Out MSFS Error %d\n"),err);

 		gSemThreadReady.Close();

 		return NULL;

 		}

 	RThread msThread;

 	__PRINT(_L("CMassStorageFileSystem::CreateFileSystem: Creating Mass Storage thread\n"));

 	err = msThread.Create(KMsThreadName, MsInitThreadFn, KDefaultStackSize, NULL, msFsys);

 	if (err != KErrNone)

 		{

 		__PRINT1(_L("CMassStorageFileSystem::CreateFileSystem Out Thread Error %d\n"),err);

 		gSemThreadReady.Close();

 		return msFsys;

 		}

 	((CMassStorageFileSystem*)msFsys)->iInstalled=ETrue;

 	msThread.Logon(((CMassStorageFileSystem*)msFsys)->iThreadStat);

 	msThread.Resume();

 	gSemThreadReady.Wait();

 	gSemThreadReady.Close();

 	msThread.Close();

 	__PRINT(_L("CMassStorageFileSystem::CreateFileSystem Out Clean\n"));

 	return msFsys;

 	}