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

 // f32test\server\t_main.cpp

 //

 //

 #define __E32TEST_EXTENSION__

 #include <f32file.h>

 #include <e32test.h>

 #include <e32math.h>

 #include <f32dbg.h>

 #include "t_server.h"

 GLDEF_D	RFs TheFs;

 GLDEF_D TFileName gSessionPath;

 GLDEF_D TInt gAllocFailOff=KAllocFailureOff;

 GLDEF_D TInt gAllocFailOn=KAllocFailureOff;

 GLDEF_D TInt64 gSeed=51703;

 GLDEF_D TChar gDriveToTest;

 GLDEF_D TVolumeInfo gVolInfo;	// volume info for current drive

 GLDEF_D TFileCacheFlags gDriveCacheFlags;

 _LIT(KPrivate, "\\Private\\");

 ////////////////////////////////////////////////////////////

 // Template functions encapsulating ControlIo magic

 //

 GLDEF_D template <class C>

 GLDEF_C TInt controlIo(RFs &fs, TInt drv, TInt fkn, C &c)

 {

     TPtr8 ptrC((TUint8 *)&c, sizeof(C), sizeof(C));

     TInt r = fs.ControlIo(drv, fkn, ptrC);

     return r;

 }

 GLDEF_C void CreateShortName(TDes& aFileName,TInt64& aSeed)

 //

 // Create a random, dos legal 8.3 char name

 //

 	{

 	TInt length=Math::Rand(aSeed)%11;

 	if (length==0)

 		length=1;

 	else if (length==3)	// don't create three letter names like 'AUX' or 'PRN'

 		length++;

 	else if (length>8)	// end in '.' if no extension

 		length++;

 	aFileName.SetLength(length);

 	for(TInt i=0;i<length;i++)

 		{

 		if (i==9)

 			{

 			aFileName[i]='.';

 			continue;

 			}

 		TInt letter=Math::Rand(aSeed)%26;

 		aFileName[i]=(TText)('A'+letter);

 		}

 	}

 GLDEF_C void CreateLongName(TDes& aFileName,TInt64& aSeed,TInt aLength)

 //

 // Create a random, dos legal 8.3 char name

 //

 	{

 	TInt length;

 	if (aLength>0)

 		length=aLength;

 	else

 		{

 		length=Math::Rand(aSeed)%128;

 		length+=Math::Rand(aSeed)%128;

 		length+=Math::Rand(aSeed)%128;

 		length+=Math::Rand(aSeed)%128;

 		length-=256;

 			length=Abs(length);

 		if (length==0)

 			length=1;

 		if (length>220)

 			length=31;

 		}

 	if (length==3)	// don't create three letter names like 'AUX' or 'PRN'

 		length++;

 	aFileName.SetLength(length);

 	TInt spaceChar=-1;

 	TInt i;

 	for(i=0;i<length;i++)

 		{

 StartAgain:

 		TChar letter=0;

 		TBool illegalChar=ETrue;

 		while(illegalChar)

 			{

 #if defined(__WINS__)

 			if (gSessionPath[0]=='C')

 				letter=(TChar)('A'+Math::Rand(aSeed)%26);

 			else

 				letter=(TChar)Math::Rand(aSeed)%256;

 #else

 			letter=(TChar)Math::Rand(aSeed)%256;

 #endif

 			TBool space=letter.IsSpace();

 			if (space && spaceChar==-1)

 				spaceChar=i;

 			else if (!space && spaceChar!=-1)

 				spaceChar=-1;

 			switch(letter)

 				{

 			case '<':

 			case '>':

 			case ':':

 			case '"':

 			case '/':

 			case '|':

 			case '*':

 			case '?':

 			case '\\':

 			case '\0':

 				break;

 			default:

 				illegalChar=EFalse;

 				};

 			}

 		aFileName[i]=(TText)letter;

 		}

 	if (spaceChar!=-1)

 		{

 		i=spaceChar;

 		goto StartAgain;

 		}

 	}

 GLDEF_C void CheckDisk()

 //

 // Do a checkdisk and report failure

 //

 	{

 	test.Next(_L("Check Disk"));

 	TInt r=TheFs.CheckDisk(gSessionPath);

 	if (r!=KErrNone && r!=KErrNotSupported && r!=KErrPermissionDenied)

 		ReportCheckDiskFailure(r);

 	}

 GLDEF_C void ReportCheckDiskFailure(TInt aRet)

 //

 // Report the failure of checkdisk

 //

 	{

 	test.Printf(_L("CHECKDISK FAILED: "));

 	switch(aRet)

 		{

 	case 1:	test.Printf(_L("File cluster chain contains a bad value (<2 or >maxCluster)\n")); break;

 	case 2:	test.Printf(_L("Two files are linked to the same cluster\n")); break;

 	case 3:	test.Printf(_L("Unallocated cluster contains a value != 0\n"));	break;

 	case 4:	test.Printf(_L("Size of file != number of clusters in chain\n")); break;

 	default: test.Printf(_L("Undefined Error value %d\n"),aRet);

 		}

 	test(EFalse);

 	}

 GLDEF_C void MakeFile(const TDesC& aFileName,const TUidType& aUidType,const TDesC8& aFileContents)

 //

 // Make a file and write uid and data

 //

 	{

 	RFile file;

 	TInt r=file.Replace(TheFs,aFileName,0);

 	if (r==KErrPathNotFound)

 		{

 		r=TheFs.MkDirAll(aFileName);

 		test_KErrNone(r);

 		r=file.Replace(TheFs,aFileName,0);

 		}

 	test_KErrNone(r);

 	TCheckedUid checkedUid(aUidType);

 	TPtrC8 uidData((TUint8*)&checkedUid,sizeof(TCheckedUid));

 	r=file.Write(uidData);

 	test_KErrNone(r);

 	r=file.Write(aFileContents);

 	test_KErrNone(r);

 	file.Close();

 	}

 GLDEF_C void MakeFile(const TDesC& aFileName,const TDesC8& aFileContents)

 //

 // Make a file and write something in it

 //

 	{

 	RFile file;

 	TInt r=file.Replace(TheFs,aFileName,0);

 	if (r==KErrPathNotFound)

 		{

 		r=TheFs.MkDirAll(aFileName);

 		test_KErrNone(r);

 		r=file.Replace(TheFs,aFileName,0);

 		}

 	test_KErrNone(r);

 	r=file.Write(aFileContents);

 	test_KErrNone(r);

 	file.Close();

 	}

 GLDEF_C void MakeFile(const TDesC& aFileName,TInt anAttributes)

 //

 // Make a file and write something in it

 //

 	{

 	RFile file;

 	TInt r=file.Replace(TheFs,aFileName,0);

 	if (r==KErrPathNotFound)

 		{

 		r=TheFs.MkDirAll(aFileName);

 		test_KErrNone(r);

 		r=file.Replace(TheFs,aFileName,0);

 		}

 	test_KErrNone(r);

 	file.Close();

 	r=TheFs.SetAtt(aFileName,anAttributes,0);

 	test_KErrNone(r);

 	}

 GLDEF_C void MakeFile(const TDesC& aFileName)

 //

 // Make a file

 //

 	{

 	MakeFile(aFileName,_L8(""));

 	}

 GLDEF_C void MakeDir(const TDesC& aDirName)

 //

 // Make a directory

 //

 	{

 	TInt r=TheFs.MkDirAll(aDirName);

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

 		{

 		test.Printf(_L("%c: MakeDir Error %d\n"),aDirName[0],r);

 		test(0);

 		}

 	}

 GLDEF_C void DeleteTestDirectory()

 //

 // Delete the leaf session path directory

 //

 	{

 	TheFs.SetAtt(_L("\\F32-TST\\SCANTEST\\Left\\Dir3\\Dir4\\Hidden"), 0, KEntryAttHidden);

 	TheFs.SetAtt(_L("\\F32-TST\\SCANTEST\\Left\\Dir3\\Dir4\\Hidden\\HiddenFile"), 0, KEntryAttHidden);

 	TheFs.SetAtt(_L("\\F32-TST\\SCANTEST\\Left\\Dir3\\Dir4\\Hidden\\System"), 0, KEntryAttSystem);

 	test.Next(_L("Delete test directory"));

 	CFileMan* fMan=CFileMan::NewL(TheFs);

 	test(fMan!=NULL);

 	TInt r=TheFs.SessionPath(gSessionPath);

 	test_KErrNone(r);

 	r=TheFs.CheckDisk(gSessionPath);

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

 		ReportCheckDiskFailure(r);

 	r=fMan->RmDir(gSessionPath);

 	test_KErrNone(r);

 	delete fMan;

 	}

 GLDEF_C void CreateTestDirectory(const TDesC& aSessionPath)

 //

 // Create directory for test

 //

 	{

 	TParsePtrC path(aSessionPath);

 	test(path.DrivePresent()==EFalse);

 	TInt r=TheFs.SetSessionPath(aSessionPath);

 	test_KErrNone(r);

 	r=TheFs.SessionPath(gSessionPath);

 	test_KErrNone(r);

 	r=TheFs.MkDirAll(gSessionPath);

 	test_Value(r, r == KErrNone || r == KErrAlreadyExists);

 	}

 GLDEF_C TInt CurrentDrive()

 //

 // Return the current drive number

 //

 	{

 	TInt driveNum;

 	TInt r=TheFs.CharToDrive(gSessionPath[0],driveNum);

 	test_KErrNone(r);

 	return(driveNum);

 	}

 GLDEF_C void Format(TInt aDrive)

 //

 // Format current drive

 //

 	{

 	test.Next(_L("Format"));

 	TBuf<4> driveBuf=_L("?:\\");

 	driveBuf[0]=(TText)(aDrive+'A');

 	RFormat format;

 	TInt count;

 	TInt r=format.Open(TheFs,driveBuf,EQuickFormat,count);

 	test_KErrNone(r);

 	while(count)

 		{

 		TInt r=format.Next(count);

 		test_KErrNone(r);

 		}

 	format.Close();

 	}

 LOCAL_C void PushLotsL()

 //

 // Expand the cleanup stack

 //

 	{

 	TInt i;

 	for(i=0;i<1000;i++)

 		CleanupStack::PushL((CBase*)NULL);

 	CleanupStack::Pop(1000);

 	}

 LOCAL_C void DoTests(TInt aDrive)

 //

 // Do testing on aDrive

 //

 	{

 	gSessionPath=_L("?:\\F32-TST\\");

 	TChar driveLetter;

 	TInt r=TheFs.DriveToChar(aDrive,driveLetter);

 	test_KErrNone(r);

 	gSessionPath[0]=(TText)driveLetter;

 	r=TheFs.SetSessionPath(gSessionPath);

 	test_KErrNone(r);

 	User::After(1000000);

 //	Format(CurrentDrive());

 	test.Printf(_L("Creating session path"));

 	r=TheFs.MkDirAll(gSessionPath);

 	if(r == KErrCorrupt)

 		{

 		test.Printf(_L("Attempting to create directory \'%S\' failed, KErrCorrupt\n"), &gSessionPath);

 		test.Printf(_L("This could be caused by a previous failing test, or a test media defect\n"));

 		test.Printf(_L("Formatting drive, retrying MkDirall\nShould subsequent tests fail with KErrCorrupt (%d) as well, replace test medium !\n"),

 			r);

 		Format(aDrive);

 		r=TheFs.MkDirAll(gSessionPath);

 		test_KErrNone(r);

 		}

 	else if (r == KErrNotReady)

 		{

 		TDriveInfo d;

 		r=TheFs.Drive(d, aDrive);

 		test_KErrNone(r);

 		if (d.iType == EMediaNotPresent)

 			test.Printf(_L("%c: Medium not present - cannot perform test.\n"), (TUint)driveLetter);

 		else

 			test.Printf(_L("medium found (type %d) but drive %c: not ready\nPrevious test may have hung; else, check hardware.\n"), (TInt)d.iType, (TUint)driveLetter);

 		}

 	test_Value(r, r == KErrNone || r == KErrAlreadyExists);

 	TheFs.ResourceCountMarkStart();

 	test.Printf(_L("Calling main test sequence ...\n"));

 	TRAP(r,CallTestsL());

 	test_KErrNone(r);

 	test.Printf(_L("test sequence completed without error\n"));

 	TheFs.ResourceCountMarkEnd();

 	CheckDisk();

 	}

 void ParseCommandArguments()

 //

 //

 //

 	{

 	TBuf<0x100> cmd;

 	User::CommandLine(cmd);

 	TLex lex(cmd);

 	TPtrC token=lex.NextToken();

 	TFileName thisfile=RProcess().FileName();

 	if (token.MatchF(thisfile)==0)

 		{

 		token.Set(lex.NextToken());

 		}

 	test.Printf(_L("CLP=%S\n"),&token);

 	if(token.Length()!=0)

 		{

 		gDriveToTest=token[0];

 		gDriveToTest.UpperCase();

 		}

 	else

 		gDriveToTest='C';

 	}

 TFullName gExtName;

 TBool gPrimaryExtensionExists = EFalse;

 GLDEF_C TInt DismountFileSystem(RFs& aFs, const TDesC& aFileSystemName,TInt aDrive)

 	{

 	//Make note of the first extension if it exists, so that we remount

 	//it when the file system is remounted.

 	TInt r = aFs.ExtensionName(gExtName, aDrive, 0);

 	if (r == KErrNone)

 		{

 		gPrimaryExtensionExists = ETrue;

 		}

 	return aFs.DismountFileSystem(aFileSystemName, aDrive);

 	}

 GLDEF_C TInt MountFileSystem(RFs& aFs, const TDesC& aFileSystemName,TInt aDrive, TBool aIsSync)

 	{

 	TInt r;

 	if (gPrimaryExtensionExists)

 		{

 		r = aFs.MountFileSystem(aFileSystemName, gExtName, aDrive, aIsSync);

 		}

 	else

 		{

 		r = aFs. MountFileSystem(aFileSystemName, aDrive, aIsSync);

 		}

 	return r;

 	}

 GLDEF_C TInt E32Main()

 //

 // Test with drive nearly full

 //

     {

 	CTrapCleanup* cleanup;

 	cleanup=CTrapCleanup::New();

 	TRAPD(r,PushLotsL());

 	__UHEAP_MARK;

 	test.Title();

 	test.Start(_L("Starting tests..."));

 	ParseCommandArguments(); //need this for drive letter to test

 	r=TheFs.Connect();

 	test_KErrNone(r);

 	TheFs.SetAllocFailure(gAllocFailOn);

 	TTime timerC;

 	timerC.HomeTime();

 	TFileName sessionp;

 	TheFs.SessionPath(sessionp);

 	TBuf<30> privatedir;

 	privatedir = KPrivate;

 	TUid thisUID = RProcess().Identity();

 	privatedir.AppendFormat(_L("%08x"),thisUID.iUid);

 	privatedir.Append(_L("\\"));

 	test(privatedir == sessionp.Mid(2,sessionp.Length()-2));

 	test.Printf(_L("sp=%S\n"),&sessionp);

 	sessionp[0]=(TText)gDriveToTest;

 	test.Printf(_L("sp1=%S\n"),&sessionp);

 	TInt theDrive;

 	r=TheFs.CharToDrive(gDriveToTest,theDrive);

 	test_KErrNone(r);

 	// Get the TFileCacheFlags for this drive

 	r = TheFs.Volume(gVolInfo, theDrive);

 	if (r == KErrNotReady)

 		{

 		TDriveInfo info;

 		TInt err = TheFs.Drive(info,theDrive);

 		test_KErrNone(err);

 		if (info.iType == EMediaNotPresent)

 			test.Printf(_L("%c: Medium not present - cannot perform test.\n"), (TUint)gDriveToTest);

 		else

 			test.Printf(_L("%c: medium found (type %d) but drive not ready\nPrevious test may have hung; else, check hardware.\n"), (TUint)gDriveToTest, (TInt)info.iType);

 		}

 	else if (r == KErrCorrupt)

 		{

 		test.Printf(_L("%c: Media corruption; previous test may have aborted; else, check hardware\n"), (TUint)gDriveToTest);

 		}

 	test_KErrNone(r);

 	gDriveCacheFlags = gVolInfo.iFileCacheFlags;

 	test.Printf(_L("DriveCacheFlags = %08X\n"), gDriveCacheFlags);

 #if defined(_DEBUG) || defined(_DEBUG_RELEASE)

 	TPckgBuf<TIOCacheValues> pkgOrgValues;

 	TIOCacheValues& orgValues=pkgOrgValues();

 	r = controlIo(TheFs,theDrive, KControlIoCacheCount, orgValues);

 	test_KErrNone(r);

 	test.Printf(_L("\n"));

 	test.Printf(_L("Requests on close queue at start=%d\n"),orgValues.iCloseCount);

 	test.Printf(_L("Requests on free queue at start=%d\n"),orgValues.iFreeCount);

 	test.Printf(_L("Requests dynamically allocated at start=%d\n"),orgValues.iAllocated);

 	test.Printf(_L("Requests in total at start=%d\n"),orgValues.iTotalCount);

 	// File cache

 	// flush closed files queue

 	r = TheFs.ControlIo(theDrive, KControlIoFlushClosedFiles);

 	test_KErrNone(r);

 	// get number of items on File Cache

 	TFileCacheStats startFileCacheStats;

 	r = controlIo(TheFs,theDrive, KControlIoFileCacheStats, startFileCacheStats);

 	test_Value(r, r == KErrNone || r == KErrNotSupported);

 	test.Printf(_L("File cache: Cachelines (free %d, used %d), Segments(allocated %d locked %d). Closed files(%d)\n"),

 		startFileCacheStats.iFreeCount,

 		startFileCacheStats.iUsedCount,

 		startFileCacheStats.iAllocatedSegmentCount,

 		startFileCacheStats.iLockedSegmentCount,

 		startFileCacheStats.iFilesOnClosedQueue);

 #endif

 	DoTests(theDrive);

 	TTime endTimeC;

 	endTimeC.HomeTime();

 	TTimeIntervalSeconds timeTakenC;

 	r=endTimeC.SecondsFrom(timerC,timeTakenC);

 	test_KErrNone(r);

 	test.Printf(_L("Time taken for test = %d seconds\n"),timeTakenC.Int());

 	TheFs.SetAllocFailure(gAllocFailOff);

 #if defined(_DEBUG) || defined(_DEBUG_RELEASE)

 	TPckgBuf<TIOCacheValues> pkgValues;

 	TIOCacheValues& values=pkgValues();

 	r = controlIo(TheFs,theDrive, KControlIoCacheCount, values);

 	test_KErrNone(r);

 	test.Printf(_L("Requests on close queue at end=%d\n"),values.iCloseCount);

 	test.Printf(_L("Requests on free queue at end=%d\n"),values.iFreeCount);

 	test.Printf(_L("Requests dynamically allocated at end=%d\n"),values.iAllocated);

 	test.Printf(_L("Requests in total at end=%d\n"),values.iTotalCount);

 	test(orgValues.iCloseCount==values.iCloseCount);

 	test(orgValues.iAllocated == values.iAllocated);

 	// The free count can increase if the file server runs out of requests in the RequestAllocator

 	// free pool but this should never decrease - this implies a request leak

 	test(orgValues.iFreeCount <= values.iFreeCount);

 	// The total number of allocated requests should be equal to :

 	// requests on the close queue + requests on free queue

 	// + 1 (because we used one request to issue KControlIoCacheCount)

 	// If this doesn't equate then this implies a request leak

 	test(values.iTotalCount == values.iCloseCount + values.iFreeCount + 1);

 	// File cache

 	TFileCacheStats endFileCacheStats;

 	r = controlIo(TheFs,theDrive, KControlIoFileCacheStats, endFileCacheStats);

 	test_Value(r, r == KErrNone || r == KErrNotSupported);

 	test.Printf(_L("File cache: Cachelines (free %d, used %d), Segments(allocated %d locked %d). Closed files(%d)\n"),

 		endFileCacheStats.iFreeCount,

 		endFileCacheStats.iUsedCount,

 		endFileCacheStats.iAllocatedSegmentCount,

 		endFileCacheStats.iLockedSegmentCount,

 		endFileCacheStats.iFilesOnClosedQueue);

 	// flush closed files queue

 	test.Printf(_L("Flushing close queue..."));

 	r = TheFs.ControlIo(theDrive, KControlIoFlushClosedFiles);

 	test_KErrNone(r);

 	r = controlIo(TheFs,theDrive, KControlIoFileCacheStats, endFileCacheStats);

 	test_Value(r, r == KErrNone || r == KErrNotSupported);

 	test.Printf(_L("File cache: Cachelines (free %d, used %d), Segments(allocated %d locked %d). Closed files(%d)\n"),

 		endFileCacheStats.iFreeCount,

 		endFileCacheStats.iUsedCount,

 		endFileCacheStats.iAllocatedSegmentCount,

 		endFileCacheStats.iLockedSegmentCount,

 		endFileCacheStats.iFilesOnClosedQueue);

 	if (r == KErrNone)

 		{

 		test(startFileCacheStats.iFreeCount == endFileCacheStats.iFreeCount);

 		test(startFileCacheStats.iUsedCount == endFileCacheStats.iUsedCount);

 		test(startFileCacheStats.iAllocatedSegmentCount == endFileCacheStats.iAllocatedSegmentCount);

 		test(startFileCacheStats.iLockedSegmentCount == endFileCacheStats.iLockedSegmentCount);

 		test(startFileCacheStats.iFileCount == endFileCacheStats.iFileCount);

 		}

 #endif

 	TheFs.Close();

 	test.End();

 	test.Close();

 	__UHEAP_MARKEND;

 	delete cleanup;

 	return(KErrNone);

     }