// 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 <e32hal.h>

#include <e32math.h>

#include <f32dbg.h>

#include "t_server.h"

#include "t_chlffs.h"

GLDEF_D	RFs TheFs;

GLDEF_D TFileName gSessionPath;

GLDEF_D TFileName gExeFileName(RProcess().FileName());

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);

		}

	}

_LIT(KFatName, "Fat");

_LIT(KFat32Name, "Fat32");

static

TBool isFAT(RFs &aFsSession, TInt aDrive)

{

	TFileName f;

	TInt r = aFsSession.FileSystemName(f, aDrive);

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

	return (f.CompareF(KFatName) == 0 || f.CompareF(KFat32Name) == 0);

}	

static 

TUint16 getRootEntCnt(RFs &aFsSession, TInt aDrive)

	{

	RRawDisk	rdisk;

	TUint16		rootEntCnt;

	TPtr8		reader((TUint8*)&rootEntCnt, sizeof(rootEntCnt));

	test_KErrNone(rdisk.Open(aFsSession, aDrive));

	test_KErrNone(rdisk.Read(17, reader)); // "BPB_RootEntCnt" field at offset 17

	rdisk.Close();

	return rootEntCnt;

	}

GLDEF_C TBool IsFileSystemFAT(RFs &aFsSession,TInt aDrive)

//

// return true if FAT on aDrive

//

	{

	return (isFAT(aFsSession, aDrive) && getRootEntCnt(aFsSession, aDrive) != 0);

	}

GLDEF_C TBool IsFileSystemFAT32(RFs &aFsSession,TInt aDrive)

//

// return true if FAT32 on aDrive

//

	{

	return (isFAT(aFsSession, aDrive) && getRootEntCnt(aFsSession, aDrive) == 0);

	}

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 CheckEntry(const TDesC& aName,TUint anAttributes,const TTime& aModified)

//

// Checks the values associated with an entry

//

	{

	TEntry entry;

	TInt r=TheFs.Entry(aName,entry);

	test_KErrNone(r);

	test(entry.iAtt==anAttributes);

	if (aModified!=TTime(0))

		test(entry.iModified==aModified);

	}

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 TurnAllocFailureOff()

//

// Switch off all allocFailure

//

	{

	test.Printf(_L("Disable Alloc Failure\n"));

	TheFs.SetAllocFailure(gAllocFailOff);

	gAllocFailOn=KAllocFailureOff; // Disable gAllocFailOn

	}

GLDEF_C void TurnAllocFailureOn()

//

// Switch off all allocFailure

//

	{

	test.Printf(_L("Enable Alloc Failure\n"));

	gAllocFailOn=KAllocFailureOn; // Enable gAllocFailOn

	TheFs.SetAllocFailure(gAllocFailOn);

	}

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 SetSessionPath(const TDesC& aPathName)

//

// Set the session path and update gSessionPath

//

	{

	TInt r=TheFs.SetSessionPath(aPathName);

	test_KErrNone(r);

	r=TheFs.SessionPath(gSessionPath);

	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 TInt CheckFileExists(const TDesC& aName,TInt aResult,TBool aCompRes/*=ETrue*/)

//

// Check aName exists

//

	{

	TEntry entry;

	TInt r=TheFs.Entry(aName,entry);

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

	if (aResult!=KErrNone)

		return(0);

	TParsePtrC nameParse(aName);

	TParsePtrC entryParse(entry.iName);

	TBool nameMatch=(entryParse.Name()==nameParse.Name());

	TBool extMatch=(entryParse.Ext()==nameParse.Ext()) || (entryParse.Ext().Length()<=1 && nameParse.Ext().Length()<=1);

	test((nameMatch && extMatch)==aCompRes);

	return(entry.iSize);

	}

GLDEF_C void CheckFileContents(const TDesC& aName,const TDesC8& aContents)

//

// Check contents of file

//

	{

	RFile f;

	TInt r=f.Open(TheFs,aName,EFileRead);

	test_KErrNone(r);

	HBufC8* testBuf=HBufC8::NewL(aContents.Length());

	test(testBuf!=NULL);

	TPtr8 bufPtr(testBuf->Des());

	r=f.Read(bufPtr);

	test_KErrNone(r);

	test(bufPtr==aContents);

	r=f.Read(bufPtr);

	test_KErrNone(r);

	test(bufPtr.Length()==0);

	f.Close();

	User::Free(testBuf);

	}

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);

// !!! Disable platform security tests until we get the new APIs

//	if(User::Capability() & KCapabilityRoot)

		CheckMountLFFS(TheFs,driveLetter);

	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)