// 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;
	}

