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

// Performs actions related to writing a file

// 

//

#include "basetestfat32writefile.h"

static RRawDisk TheDisk;

static RFile TheFile;

/**

Class Constructor

*/		    

CBaseTestFat32WriteFile::CBaseTestFat32WriteFile() 

	{

	SetTestStepName(KTestStepWriteFile);

	}

/**

Class Destructor

*/

CBaseTestFat32WriteFile::~CBaseTestFat32WriteFile() 

	{

	}

/** 

The function performs the following actions

1. If there is a file specified in the ini file then 

	-> create a file with that filename 

	-> Else create a file name ReadTestFile.txt

2. Get the file action required from the ini file and carry out that action

@return EPass if test passes and EFail if test fails

*/ 

TVerdict CBaseTestFat32WriteFile::doTestStepL()

	{

	TInt r = 0;

	TInt errcode;

	SetTestStepResult(EFail);

	_LIT(KAction,"Action");

	TBuf<10> actionAfterCreate;

	TPtrC16 action = actionAfterCreate;

	_LIT(KName,"SetFileName");

	_LIT(KPath,"%c:\\");

	TBuf<10> setFileName;

	TPtrC16 name = setFileName;

	TBuf<255> fileName;

	_LIT(KCreatedFileName,"%c:\\ReadTestFile.txt");

	TBool alright = GetStringFromConfig(ConfigSection(), KName, name);

	if (alright)

		{

		fileName.Format(KPath, (TUint)iDriveToTest);

		fileName.Append(name);

		}

	else

		{

		fileName.Format(KCreatedFileName, (TUint)iDriveToTest);

		}

	TInt res = TheFile.Replace(iTheFs,fileName,EFileWrite);

	TheFile.Close();

	TBool alright2 = GetStringFromConfig(ConfigSection(), KAction, action);

	if (alright2)

		{

		errcode = SetAttribs(fileName);

		if (action == _L("Write"))

			{

			errcode = WriteFile(fileName);

			}

		if (action == _L("DirList"))

			{

			errcode = DirList(fileName);

			}

		if (action == _L("CheckAtt"))

			{

			errcode = SetAttribs(fileName);

			}

		if (action == _L("CheckCode"))

			{

			errcode = CheckErrCode(res);

			}

		if (action == _L("MakeDirectory"))

			{

			errcode = MakeDirectory(_L("TestDirectory\\"));

			}

		if (action == _L("MakeTwoDirectory"))

			{

			errcode = MakeDirectory(_L("TestDirectory\\"));

			if (errcode == KErrNone)

				{

				errcode = MakeDirectory(_L("TestDirectory\\TestDirectory2\\"));

				}

			}

		if (action == _L("SetLabel"))

			{

			errcode = SetLabel(_L("TestLabel"));

			}

		if (action == _L("SetTwoLabels"))

			{

			errcode = SetLabel(_L("TestLabel"));

			if (errcode == KErrNone)

				{

				errcode = SetLabel(_L("TestSecondLabel"));

				}

			}

		if (action == _L("SetFileSize"))

			{

			errcode = SetFileSize(fileName);

			}

		if (action == _L("DeleteFile"))

			{

			errcode = DeleteFile(fileName);

			}

		r = CheckErrCode(errcode);

		if(r != KErrNone)

			{

			_LIT(KErrorWriteFile, "Error writing the file, Error Code = %d");

			INFO_PRINTF2(KErrorWriteFile, r);

			SetTestStepResult(EFail);

			}

		else

			{

			SetTestStepResult(EPass);

			_LIT(KReadPass, "Read Passed");

			INFO_PRINTF1(KReadPass);

			}

		}

	else

		{

		r = CheckErrCode(res);

		if(r != KErrNone)

			{

			_LIT(KErrorWriteFile, "Error writing the file, Error Code = %d");

			INFO_PRINTF2(KErrorWriteFile, r);

			SetTestStepResult(EFail);

			}

		else

			{

			_LIT(KReadPass, "Read Passed");

			INFO_PRINTF1(KReadPass);

			SetTestStepResult(EPass);

			}					

		}

	return TestStepResult(); 

	}

/** 

Open the file

@param aFile The name of the file to write 

@return KErrNone if successful

*/

TInt CBaseTestFat32WriteFile::WriteFile(const TDesC16& aFile)

	{

	TBuf8<255> temp;

	temp.Copy(aFile);

	TInt r = 0; 

	r = TheFile.Open(iTheFs,aFile,EFileWrite);

	if(r != KErrNone)

		{

		_LIT(KErrorOpenFile, "Unable to open the file, Error Code = %d");

		INFO_PRINTF2(KErrorOpenFile, r);

		return r; 

		}

	TheFile.Close();

	return r; 

	}

/** 

Get the attribute to set from the ini file and set the attributes of the file

@param aFile The name of the file whos attributes are to be set 

@return KErrNone if successful

*/

TInt CBaseTestFat32WriteFile::SetAttribs(const TDesC16& aFile)

	{		

	TBuf8<255> temp;

	temp.Copy(aFile);

	TInt r = 0; 

	TUint setMask = KEntryAttNormal;

	_LIT(KAtt,"Attributes");

	TBuf<10> fileAttributes;

	TPtrC16 attributes = fileAttributes;

	TBool alright3 = GetStringFromConfig(ConfigSection(), KAtt, attributes);

	if (alright3)

		{

		if (attributes == _L("Normal"))

			{

			setMask = KEntryAttNormal;

			}	

		else if (attributes == _L("ReadOnly"))

			{

			setMask = KEntryAttReadOnly;

			}

		else if (attributes == _L("Hidden"))

			{

			setMask = KEntryAttHidden;

			}

		else if (attributes == _L("System"))

			{

			setMask = KEntryAttSystem;

			}

		else if (attributes == _L("Dir"))

			{

			setMask = KEntryAttDir;

			}

		r = TheFile.Open(iTheFs,aFile,EFileWrite);

		if (r != KErrNone)

			{

			_LIT(KOpenFail, "Cannot open the file, error = %d");

			INFO_PRINTF2(KOpenFail, r);

			return r;

			}

		r = TheFile.SetAtt(setMask, 0);

		if (r != KErrNone)

			{

			_LIT(KSetAttFail, "Cannot set the attributes, error = %d");

			INFO_PRINTF2(KSetAttFail, r);

			return r;

			}

		TheFile.Close();

		r = CheckAtt(aFile, setMask);	

		if (r != KErrNone)

			{

			_LIT(KAttFail, "Cannot check the attributes, error = %d");

			INFO_PRINTF2(KAttFail, r);

			return r;

			}

		}

	else

		{

		_LIT(KNoAttribs, "No attributes specified in the ini file");

		INFO_PRINTF1(KNoAttribs);

		return KErrNone;	

		}

	return r; 

	}

/** 

Check the attributes of the file and compare them to the attributes expected

@param aFile The name of the file whos attributes are to be checked 

@return KErrNone if successful

*/

TInt CBaseTestFat32WriteFile::CheckAtt(const TDesC16& aFile, TUint setMask)

	{

	TInt r = 0;

	TUint fileAttributes;

	r = iTheFs.Att(aFile, fileAttributes);

	if (r != KErrNone)

		{

		_LIT(KGetAttFail, "Failed to get the file attributes");

		INFO_PRINTF1(KGetAttFail);

		return KErrGeneral;

		}

	else

		{

		if (fileAttributes&setMask)

			{

			_LIT(KAttChanged, "Attributes changed correctly");

			INFO_PRINTF1(KAttChanged);

			return r;

			}

		else

			{

			_LIT(KAttNotChanged, "Attributes have not been changed");

			INFO_PRINTF1(KAttNotChanged);

			return KErrGeneral;

			}

		}

	}

/** 

Searching the directory to see if it contains the file who's attributes 

were set as hidden. 

@param aFile The name of the file to search for 

@return KErrNone if successful

*/

TInt CBaseTestFat32WriteFile::DirList(const TDesC16& aFile)

	{

	TInt i; 	

	CDir* dirPtr;	

	TBuf<10> dir1;

	dir1 = iSessionPath;

	dir1.Append(_L("*.*"));

	TBuf<20> buf;

	buf.Append(aFile);

	buf.Delete(0,3);

	TInt match = 0;

	TInt r = iTheFs.GetDir(dir1,KEntryAttMatchExclude|(KEntryAttHidden|0x24),ESortNone,dirPtr);

	TInt num = dirPtr->Count();

	TEntry entry;

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

		{

		entry=(*dirPtr)[i];

		if (entry.iName == _L("ReadTestFile.txt"))

			{

			match = match + 1;

			}

		}

	if (match > 0)

		{	

		_LIT(KFileFound, "File found");

		INFO_PRINTF1(KFileFound);

		return KErrNone;

		}

	else

		{

		_LIT(KFileNotFound, "File not found");

		INFO_PRINTF1(KFileNotFound);

		return KErrNotFound;

		}

	}

/** 

Compare the error code that is expected to the one returned from the action

@param aReturnCode The return value that is to be compared 

@return KErrNone if successful

*/

TInt CBaseTestFat32WriteFile::CheckErrCode(TInt aReturnCode)

	{

	_LIT(KCheckCode,"CheckCode");

	TInt checkcode;

	TBool alright4 = GetIntFromConfig(ConfigSection(), KCheckCode, checkcode);

	if (alright4)

		{

		if (aReturnCode == checkcode)

			{

			_LIT(KCorrectCode, "Correct error code = %d");

			INFO_PRINTF2(KCorrectCode, aReturnCode);

			return KErrNone;

			}

		else

			{	

			_LIT(KBadCode, "Incorrect error code = %d");

			INFO_PRINTF2(KBadCode, aReturnCode);

			return KErrGeneral;

			}

		}

	else 

		{

		_LIT(KNoIniCode, "CheckCode not specified in ini file");

		INFO_PRINTF1(KNoIniCode);

		return KErrGeneral;

		}		

	}

/** 

Creating a directory

@param aDir The name of the directory to create 

@return KErrNone if successful

*/	

TInt CBaseTestFat32WriteFile::MakeDirectory(const TDesC16& aDir)

	{

	TInt r = 0;

	TBuf<255> dirName;

	dirName.Append(iSessionPath);

	dirName.Append(aDir);

	TUint setMask = KEntryAttDir;

	r = iTheFs.MkDir(dirName);

	if (r != KErrNone)

		{

		_LIT(KMkDirFail, "Failed make the directory");

		INFO_PRINTF1(KMkDirFail);

		return r;

		}	

	r = CheckAtt(dirName, setMask);	

	if (r != KErrNone)

		{

		_LIT(KChkAttFail, "Failed to check the attributes");

		INFO_PRINTF1(KChkAttFail);

		return r;

		}

	return r; 

	}

/** 

Setting a volume label

@param aLabel The name of the lable to set

@return KErrNone if successful

*/		

TInt CBaseTestFat32WriteFile::SetLabel(const TDesC16& aLabel)

	{

	TInt r = 0;

	r = iTheFs.SetVolumeLabel(aLabel, CurrentDrive());

	if (r != KErrNone)

		{

		_LIT(KSetLabelFail, "Failed to set volume label");

		INFO_PRINTF1(KSetLabelFail);

		return r;

		}

	return r; 

	}

/** 

Setting the size of the file

@param aFile The name of the file whos size to set

@return KErrNone if successful

*/			

TInt CBaseTestFat32WriteFile::SetFileSize(const TDesC16& aFile)

	{

	TInt r = 0;

	r = TheFile.Open(iTheFs,aFile,EFileWrite);

	if (r != KErrNone)

		{

		_LIT(KOpenFail, "Failed to open the file");

		INFO_PRINTF1(KOpenFail);

		return r;

		}

	TInt fileSize = (iBPB_SecPerClus/2) * 3 * 1024; // File accomodates 3 clusters

	r = TheFile.SetSize(fileSize);

	if (r != KErrNone)

		{

		_LIT(KSetSizeFail, "Failed set the size of the file");

		INFO_PRINTF1(KSetSizeFail);

		return r;

		}

	TheFile.Close();

	r = iTheFs.FinaliseDrive(CurrentDrive(), RFs::EFinal_RW);

	return r; 

	}

/** 

Delete the file

@param aFile The name of the file to delete

@return KErrNone if successful

*/				

TInt CBaseTestFat32WriteFile::DeleteFile(const TDesC16& aFile)

	{

	TInt r = 0;

	r = iTheFs.Delete(aFile);

	if (r != KErrNone)

		{

		_LIT(KDeleteFail, "Failed delete the file");

		INFO_PRINTF1(KDeleteFail);

		return r;

		}

	return r; 

	}