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

// @file

// various FAT utilities 

// 

//

#include "f32_test_utils.h"

using namespace F32_Test_Utils;

#include "filesystem_fat.h"

//-------------------------------------------------------------------------------------------------------------------

//-- define this macro if it is necessarily to exclude all stuff that uses RFs, consoles etc.

//-- may be useful if this code is used for file server extensions. mmp file is a good place to define this macro.

#ifndef FAT_UTILS_LEAN_AND_MEAN

#include <e32cons.h>

#include <e32math.h>

//-- Note that the writable static data are not allowed in DLLs i.e. plugins or somethig else.

//-- Thus it needs to be thrown away by preprocessing in such a case.

static CConsoleBase* pConsole = NULL;   //-- pointer to the text console for printing out data

static TBool  bPrintOutEnabled = ETrue; //-- global flag, if EFalse, all printing out is disabled

/**

    Set the console where the ouput will go.

    @param  apConsole pointer to the console. if NULL, the print out will be debug port only.

*/

void F32_Test_Utils::SetConsole(CConsoleBase* apConsole)

{

    pConsole = apConsole;

}

/** 

    Enable or disable printing out. See DoPrintf()

    @param  bEnable If ETrue, print out is enabled

    @return previous value of the global bPrintOutEnabled flag

*/

TBool F32_Test_Utils::EnablePrintOutput(TBool bEnable)

{

    TBool bPrevVal = bPrintOutEnabled;

    bPrintOutEnabled = bEnable;

    return bPrevVal;

}

/**

    Print out the drive information.

    @param  aFs             reference to the FS session

    @param  aDrive          drive number

    @return system-wide error codes.

*/

TInt F32_Test_Utils::PrintDrvInfo(RFs &aFs, TInt aDrive)

{

    TInt        nRes;

    TDriveInfo  driveInfo;

    TVolumeInfo volInfo;

    TBuf<256>   Buf;

    //-- get drive info

    nRes = aFs.Drive(driveInfo, aDrive);

    if(nRes != KErrNone)

        return nRes;

    nRes = aFs.Volume(volInfo, aDrive);

    if(nRes != KErrNone)

        return nRes;

    DoPrintf(_L("Drive %c: #%d\n"), 'A'+aDrive, aDrive);

    //-- print the FS name

    nRes = aFs.FileSystemName(Buf, aDrive);

    if(nRes != KErrNone)

        return nRes;

    //-- to find out FS sub type

    TFSName fsName;

    nRes = aFs.FileSystemSubType(aDrive, fsName); 

    if(nRes == KErrNone && Buf.CompareF(fsName) !=KErrNone)

    {

        Buf.AppendFormat(_L(" (%S)"), &fsName);

    }

   DoPrintf(_L("FS name: %S\n"), &Buf);

   //-- print drive and media attributes

   DoPrintf(_L("MediaType: 0x%x\n"), driveInfo.iType);

   DoPrintf(_L("DriveAtt:0x%x\n"),driveInfo.iDriveAtt);

   DoPrintf(_L("MediaAtt:0x%x\n"),driveInfo.iMediaAtt);

   //-- volume information

   DoPrintf(_L("VolId:0x%x\n"),volInfo.iUniqueID);

   DoPrintf(_L("VolSz:%ld (%ldK)\n"),volInfo.iSize, volInfo.iSize/1024);

   DoPrintf(_L("Free:%ld (%ldK)\n"),volInfo.iFree, volInfo.iFree/1024);

   return KErrNone;

}

//-------------------------------------------------------------------------------------------------------------------

/**

    Fill a media region with a given byte pattern

    @param  aFs             reference to the FS session

    @param  aDrive          drive number

    @param  aMediaStartPos  media region start position  

    @param  aMediaEndPos    media region end position

    @param  aBytePattern    byte to fill the media region with

    @return system-wide error codes.

*/

TInt F32_Test_Utils::FillMedia(RFs &aFs, TInt aDrive, TInt64 aMediaStartPos, TInt64 aMediaEndPos, TUint8 aBytePattern/*=0*/)

{

    DoPrintf(_L("~ F32_Test_Utils::FillMedia() drv:%d, from:%u to:%u\n"),aDrive, (TUint32)aMediaStartPos, (TUint32)aMediaEndPos);

    ASSERT(aMediaStartPos<=aMediaEndPos && aMediaStartPos >=0  && aMediaEndPos >=0);

    TInt  nRes = KErrNone;

    RBuf8 buf;

    const TUint32 KBufSz=65536; //-- zero-buffer size, bytes

    //-- create a buffer, filled with given pattern

    nRes = buf.CreateMax(KBufSz);

    ASSERT(nRes == KErrNone);

    buf.Fill(aBytePattern);

    TUint32 rem = (TUint32)(aMediaEndPos - aMediaStartPos);

    while(rem)

    {

        const TUint32 bytesToWrite=Min(rem, KBufSz);

        TPtrC8 ptrData(buf.Ptr(), bytesToWrite);

        nRes = MediaRawWrite(aFs, aDrive, aMediaStartPos, ptrData); 

        if(nRes != KErrNone && nRes != KErrDiskFull)

            break;

        aMediaStartPos+=bytesToWrite;

        rem-=bytesToWrite;

    }

    buf.Close();

    return nRes;    

}

//-------------------------------------------------------------------------------------------------------------------

/**

    Raw read from the media.

    @param  aFs         reference to the FS session

    @param  aDrive      drive number

    @param  aMediaPos   media position 

    @param  aLen        how many bytes to read

    @param  aData       descriptor for the data

    @return system-wide error code.

*/

TInt F32_Test_Utils::MediaRawRead(RFs &aFs, TInt aDrive, TInt64 aMediaPos, TUint32 aLen, TDes8& aData)

{

    TInt nRes=KErrNone;

    TRAP(nRes, DoMediaRawReadL(aFs, aDrive, aMediaPos, aLen, aData));

    return nRes;

}

//-------------------------------------------------------------------------------------------------------------------

/**

    Raw write to the media.

    @param  aFs         reference to the FS session

    @param  aDrive      drive number

    @param  aMediaPos   media position 

    @param  aData       descriptor with the data to write

    @return system-wide error code.

*/

TInt F32_Test_Utils::MediaRawWrite(RFs &aFs, TInt aDrive, TInt64 aMediaPos, const TDesC8& aData)

{

    TInt nRes=KErrNone;

    TRAP(nRes, DoMediaRawWriteL(aFs, aDrive, aMediaPos, aData));

    return nRes;

}

//-------------------------------------------------------------------------------------------------------------------

void F32_Test_Utils::DoMediaRawReadL(RFs &aFs, TInt aDrive, TInt64 aMediaPos, TUint32 aLen, TDes8& aData)

{

    ASSERT(aMediaPos>=0);

    ASSERT(aDrive >= EDriveA && aDrive <= EDriveZ);

    if((TUint32)aData.MaxSize() < aLen)

        User::Leave(KErrArgument);

    if(aLen == 0)

        return;

    aData.SetLength(aLen);

    RRawDisk  rawDisk;

    CleanupClosePushL(rawDisk);

    User::LeaveIfError(rawDisk.Open(aFs, aDrive));

    User::LeaveIfError(rawDisk.Read(aMediaPos, aData));

    CleanupStack::PopAndDestroy(&rawDisk);

}

//-------------------------------------------------------------------------------------------------------------------

void F32_Test_Utils::DoMediaRawWriteL(RFs &aFs, TInt aDrive, TInt64 aMediaPos, const TDesC8& aData)

{

    ASSERT(aMediaPos>=0);

    ASSERT(aDrive >= EDriveA && aDrive <= EDriveZ);

    if(aData.Size() == 0)

        return;

    RRawDisk  rawDisk;

    CleanupClosePushL(rawDisk);

    User::LeaveIfError(rawDisk.Open(aFs, aDrive));

    User::LeaveIfError(rawDisk.Write(aMediaPos, (TDesC8&)aData));

    CleanupStack::PopAndDestroy(&rawDisk);

}

//-------------------------------------------------------------------------------------------------------------------

_LIT(KFsName_LFFS,  "lffs");

_LIT(KFsName_Win32, "Win32");

_LIT(KFsName_ExFAT, "ExFat");

_LIT(KFsName_AutoMonuter, "automounter");

/**  @return ETrue if "Automounter" FS is mounted on this drive */

TBool F32_Test_Utils::Is_Automounter(RFs &aFs, TInt aDrive)

{

	ASSERT(aDrive >= EDriveA && aDrive <= EDriveZ);

    TFSName f;

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

    __ASSERT_ALWAYS((r==KErrNone) && (f.Length()>0), User::Invariant());

    return (f.CompareF(KFsName_AutoMonuter) == 0 );

}

/** @return ETrue if "lffs" FS is mounted on this drive */

TBool F32_Test_Utils::Is_Lffs(RFs &aFs, TInt aDrive)

{

	ASSERT(aDrive >= EDriveA && aDrive <= EDriveZ);

    TFSName f;

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

    __ASSERT_ALWAYS((r==KErrNone) && (f.Length()>0), User::Invariant());

    return (f.CompareF(KFsName_LFFS) == 0 );

}

/** @return ETrue if "Win32" FS is mounted on this drive (i.e this is emulator's drive c:) */

TBool F32_Test_Utils::Is_Win32(RFs &aFs, TInt aDrive)   

{

	ASSERT(aDrive >= EDriveA && aDrive <= EDriveZ);

    TFSName f;

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

    __ASSERT_ALWAYS((r==KErrNone) && (f.Length()>0), User::Invariant());

    return (f.CompareF(KFsName_Win32) == 0 );

}

/** @return ETrue if the filesystem if FAT (fat12/16/32) */

TBool F32_Test_Utils::Is_Fat(RFs &aFs, TInt aDrive)

{

	ASSERT(aDrive >= EDriveA && aDrive <= EDriveZ);

    TFSName fsName;

	TInt nRes = aFs.FileSystemName(fsName, aDrive);

    __ASSERT_ALWAYS((nRes==KErrNone) && (fsName.Length()>0), User::Invariant());

    if(fsName.CompareF(KFileSystemName_FAT) == 0)

        return ETrue; //-- "FAT" FS is explicitly mounted on this drive

    //-- try analyse file system subtype for the case of automounter FS

    nRes = aFs.FileSystemSubType(aDrive,fsName); 

    if(nRes !=KErrNone)

        return EFalse;

    return (fsName.CompareF(KFSSubType_FAT16) == 0 || fsName.CompareF(KFSSubType_FAT32) == 0 || fsName.CompareF(KFSSubType_FAT12) == 0);

}

/** returns ETrue if "exFAT" FS is mounted on this drive */

TBool F32_Test_Utils::Is_ExFat(RFs &aFs, TInt aDrive)

{

	ASSERT(aDrive >= EDriveA && aDrive <= EDriveZ);

    TFSName fsName;

	TInt nRes = aFs.FileSystemName(fsName, aDrive);

    __ASSERT_ALWAYS((nRes==KErrNone) && (fsName.Length()>0), User::Invariant());

    if(fsName.CompareF(KFsName_ExFAT) == 0 )

        return ETrue; //-- "exFAT" FS is explicitly mounted on this drive

    //-- try analyse file system subtype for the case of automounter FS

    nRes = aFs.FileSystemSubType(aDrive,fsName); 

    if(nRes !=KErrNone)

        return EFalse;

    return (fsName.CompareF(KFsName_ExFAT) == 0);

}

/** @return ETrue if the filesystem if FAT32 */

TBool F32_Test_Utils::Is_Fat32(RFs &aFs, TInt aDrive)

{

    if(!Is_Fat(aFs, aDrive))

        return EFalse;

    TFSName fsName;

    TInt nRes = aFs.FileSystemSubType(aDrive,fsName); 

    if(nRes !=KErrNone)

        return EFalse;

    return (fsName.CompareF(KFSSubType_FAT32) == 0);

}

/** @return ETrue if the filesystem if FAT16 */

TBool F32_Test_Utils::Is_Fat16(RFs &aFs, TInt aDrive)

{

    if(!Is_Fat(aFs, aDrive))

        return EFalse;

    TFSName fsName;

    TInt nRes = aFs.FileSystemSubType(aDrive,fsName); 

    if(nRes !=KErrNone)

        return EFalse;

    return (fsName.CompareF(KFSSubType_FAT16) == 0);

}

/** @return ETrue if the filesystem if FAT12 */

TBool F32_Test_Utils::Is_Fat12(RFs &aFs, TInt aDrive)

{

    if(!Is_Fat(aFs, aDrive))

        return EFalse;

    TFSName fsName;

    TInt nRes = aFs.FileSystemSubType(aDrive,fsName); 

    if(nRes !=KErrNone)

        return EFalse;

    return (fsName.CompareF(KFSSubType_FAT12) == 0);

}

static void DoCreateCheckableFileL(RFile64& aFile, TUint64 aSize)

{

    ASSERT(aSize >= TMD5::HashSize);

    //-- 1. set file size

    User::LeaveIfError(aFile.SetSize((TInt)aSize));

    //-- 2. leave place for the 16-bytes MD5 hash in the file beginning

    TInt64 filePos = TMD5::HashSize;

    User::LeaveIfError(aFile.Seek(ESeekStart,filePos));

    aSize-=TMD5::HashSize;

    TMD5 md5Hash;

    //-- 3. fill the file with random bytes

    const TInt KBufSize=65536; //-- buffer size for writing data

    TInt64 rndSeed = aSize % 43283;

    if(!rndSeed)

        rndSeed = 33521; 

    RBuf8 buf;

    buf.CreateMaxL(KBufSize);

    TUint64 prevSz = aSize;

    while(aSize)

    {

        //-- initialize buffer with random rubbish

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

        {

            buf[i] = (TUint8)Math::Rand(rndSeed);

        }

        const TUint32 nBytes = (TUint32)Min((TUint64)KBufSize, aSize);

        TPtrC8 ptrData(buf.Ptr(), nBytes);

        User::LeaveIfError(aFile.Write(ptrData)); //-- write data to the file

        md5Hash.Update(ptrData); //-- update MD5 hash        

        aSize-=nBytes;

        if((prevSz - aSize) >= K1MegaByte)

        {

            prevSz = aSize;

            DoPrintf(_L("."));

        }

    }

    buf.Close();

    DoPrintf(_L("\n"));

    //-- 4. write MD5 hash to the beginning of the file

    filePos = 0;

    User::LeaveIfError(aFile.Seek(ESeekStart,filePos));

    User::LeaveIfError(aFile.Write(md5Hash.Final()));

}

static void DoVerifyCheckableFileL(RFile64& aFile)

{

    TInt64 fileSize;

    User::LeaveIfError(aFile.Size(fileSize));

    if(fileSize < TMD5::HashSize)

       User::Leave(KErrCorrupt); //-- MD5 hash is 16 bytes, it's the minimal file size

    //-- 1. read MD5 header from the file

    TBuf8<TMD5::HashSize> md5Header(TMD5::HashSize);

    User::LeaveIfError(aFile.Read(md5Header));

    fileSize -= TMD5::HashSize;

    //-- 2. read the rest of the data and calculate the checksum

    TMD5 md5Hash;

    RBuf8 buf;

    const TInt KBufSize=65536; //-- buffer size for writing data

    buf.CreateMaxL(KBufSize);

    TUint64 prevSz = fileSize;

    while(fileSize)

    {

        User::LeaveIfError(aFile.Read(buf)); //-- read data from the file

		if (buf.Length() == 0)

			User::Leave(KErrEof);

        md5Hash.Update(buf); //-- update MD5 hash        

        fileSize-=buf.Length();

        if((prevSz - fileSize) >= K1MegaByte)

        {

            prevSz = fileSize;

            DoPrintf(_L("."));

        }

    }

    buf.Close();

    DoPrintf(_L("\n"));

    if(md5Hash.Final() != md5Header)

        User::Leave(KErrCorrupt); //-- the file is corrupt

}

//-------------------------------------------------------------------------------------------------------------------

/**

    Creates a file filled with random data. The file has a 16-byte MD5 header at the start, so it is possible to

    verify the data validity later. Thus the minimal file size is 16 bytes. The file with the iven name may already

    exist; in this case it will be replaced.

    @param  aFs         reference to the FS session

    @param  aFileName   name of the file to be created / replaced

    @param  aSize       size of the file; 16 bytes min.

    @return Standard error code

*/

TInt  F32_Test_Utils::CreateCheckableStuffedFile(RFs& aFs, const TDesC& aFileName, TUint64 aSize)

{

    DoPrintf(_L("~ F32_Test_Utils::CreateCheckableStuffedFile() file:%S, Size:%LU\n"), &aFileName, aSize);

    if(aSize < TMD5::HashSize)

        return KErrArgument; //-- MD5 hash is 16 bytes, it's the minimal file size

    RFile64 file;

    //-- 1. create a file

    TInt nRes = file.Replace(aFs, aFileName, EFileWrite);

    if(nRes != KErrNone)

        return nRes;

    TRAP(nRes, DoCreateCheckableFileL(file, aSize));

    file.Close();

    return nRes;

}

/**

    Verify previously created file that has MD5 header at the beginning. See  CreateCheckableStuffedFile(...)

    @param  aFs         reference to the FS session

    @param  aFileName   name of the file to be verified

    @return Standard error code. KErrNone if the file contents matches the MD5 header.

*/

TInt  F32_Test_Utils::VerifyCheckableFile(RFs& aFs, const TDesC& aFileName)

{

    DoPrintf(_L("~ F32_Test_Utils::VerifyCheckableFile() file:%S\n"), &aFileName);

    RFile64 file;

    //-- 1. create a file

    TInt nRes = file.Open(aFs, aFileName, EFileRead);

    if(nRes != KErrNone)

        return nRes;

    TRAP(nRes, DoVerifyCheckableFileL(file));

    file.Close();

    return nRes;

}

/**

    Create an empty file (not filled with anything). The size of the file is set by using RFile::SetSize().

    For FAT this will result in allocating a cluster chain in FAT fable, for the file systems that support sparse files (like LFFS)

    this might not work as expected.

    @param  aFs         reference to the FS session

    @param  aFileName   name of the file to be created / replaced

    @param  aSize       size of the file

    @return Standard error code

*/

TInt  F32_Test_Utils::CreateEmptyFile(RFs& aFs, const TDesC& aFileName, TUint64 aSize)

{

    DoPrintf(_L("~ F32_Test_Utils::CreateEmptyFile() file:%S, sz:%LU\n"), &aFileName, aSize);

    RFile64 file;

    //-- 1. create a file

    TInt nRes = file.Replace(aFs, aFileName, EFileWrite);

    if(nRes != KErrNone)

        return nRes;

    //-- 2. set file size

    nRes = file.SetSize(aSize);

    file.Close();

    return nRes;

}

//-------------------------------------------------------------------------------------------------------------------

/**

    Dismount and mount the filesystem again, optionally taking time when the mount starts.

    The FS can have extensions added into it; this function will handle only the primary extension (if it is present) and

    will mont the FS with it. Non-primary extensions are not supported yet.

    @param  aFs         reference to the FS session

    @param  aDrive      drive number

    @param  apTimeMountStart pointer to the TTime object, that can be called TTime::UniversalTime() on mount start (this can be

                             used for measuring time taken to mount the FS).

                             if NULL, no action will be taken.

    @return error code from the RFs::MountFileSystem()

*/

TInt  F32_Test_Utils::RemountFS(RFs& aFs, TInt aDrive, TTime* apTimeMountStart/*=NULL*/)

{

    TInt nRes;

    DoPrintf(_L("~ F32_Test_Utils::RemountingFS at drive:%d\n"), aDrive);    

    TFSDescriptor fsDescriptor;

    //-- 1. get current FS information 

    nRes = GetFileSystemDescriptor(aFs, aDrive, fsDescriptor);

    if(nRes != KErrNone)

        return nRes;

    //-- 2. dismount the file system

    if(fsDescriptor.iPExtName.Length() > 0)

    {

        DoPrintf(_L("~ Dismounting FS:%S with ext:%S\n"), &fsDescriptor.iFsName, &fsDescriptor.iPExtName);

    }

    else

    {

        DoPrintf(_L("~ Dismounting FS:%S\n"), &fsDescriptor.iFsName);    

    }

    nRes = aFs.DismountFileSystem(fsDescriptor.iFsName, aDrive);

    if(nRes != KErrNone)

    {

        ASSERT(0);

        return nRes;

    }

    //-- 3. mount it again

    if(apTimeMountStart)

        apTimeMountStart->UniversalTime(); //-- take Mount start time

    nRes = MountFileSystem(aFs, aDrive, fsDescriptor);

    return nRes;

}

//-------------------------------------------------------------------------------------------------------------------

TFSDescriptor::TFSDescriptor()

{

    Init();

}

void TFSDescriptor::Init()

{

    iFsName.SetLength(0);

    iPExtName.SetLength(0);

    iDriveSynch = EFalse;

}

TBool TFSDescriptor::operator==(const TFSDescriptor& aRhs) const

{

    ASSERT(this != &aRhs);

    return (iFsName.CompareF(aRhs.iFsName) == 0 && iPExtName.CompareF(aRhs.iPExtName) == 0 && iDriveSynch == aRhs.iDriveSynch);

}

//-------------------------------------------------------------------------------------------------------------------

/**

    Gets the information about file system mounted on a drive. This information can be later used for mounting this FS back if it is going to be dismounted

    @param  aFs         reference to the FS session

    @param  aDrive      drive number

    @param  aFsDesc     file system descriptor

    @return standard error code

*/

TInt F32_Test_Utils::GetFileSystemDescriptor(RFs &aFs, TInt aDrive, TFSDescriptor& aFsDesc)

{

    TInt nRes;

    //-- 1. get file system name

    nRes = aFs.FileSystemName(aFsDesc.iFsName, aDrive);

    if(nRes != KErrNone)

    {

        ASSERT(0);

        return nRes;

    }

    //-- 2. find out if the drive sync/async

    TPckgBuf<TBool> drvSyncBuf;

    nRes = aFs.QueryVolumeInfoExt(aDrive, EIsDriveSync, drvSyncBuf);

    if(nRes != KErrNone)

    {//-- pretend that the drive is asynch. in the case of file system being corrupted. this is 99.9% true

       aFsDesc.iDriveSynch = EFalse;

    }

    else

    {

        aFsDesc.iDriveSynch = drvSyncBuf();

    }

    //-- 3. find out primary extension name if it is present; we will need to add it again when mounting the FS

    //-- other extensions (non-primary) are not supported yet

    nRes = aFs.ExtensionName(aFsDesc.iPExtName, aDrive, 0);

    if(nRes != KErrNone)

    {

        aFsDesc.iPExtName.SetLength(0);

    }

    //-- 3.1 check if the drive has non-primary extensions, fail in this case

    TBuf<40> extName;

    nRes = aFs.ExtensionName(extName, aDrive, 1);

    if(nRes == KErrNone)

    {   

        DoPrintf(_L("~ F32_Test_Utils::GetFileSystemDescriptor: Non-primary extensions are not supported!\n"));

        return KErrNotSupported;

    }

    return KErrNone;

}

//-------------------------------------------------------------------------------------------------------------------

/**

    Mount the file system by the information provided in the FS descriptor

    @param  aFs         reference to the FS session

    @param  aDrive      drive number

    @param  aFsDesc     file system descriptor containing all necessary information to mount the FS.

    @return standard error code

*/

TInt F32_Test_Utils::MountFileSystem(RFs &aFs, TInt aDrive, const TFSDescriptor& aFsDesc)

{

    DoPrintf(_L("~ F32_Test_Utils::MountFileSystem() drive:%d Name:%S\n"), aDrive, &aFsDesc.iFsName);  

    TInt nRes;

    if(aFsDesc.iFsName.Length() <=0 )

    {

        ASSERT(0);

        return KErrArgument;

    }  

    //-- mount File system 

    const TBool bPrimaryExt = (aFsDesc.iPExtName.Length() > 0);

    if(bPrimaryExt)

    {//-- we need to mount FS with the primary extension

        nRes = aFs.AddExtension(aFsDesc.iPExtName);

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

        {

            ASSERT(0);

            return nRes;

        }

        nRes = aFs.MountFileSystem(aFsDesc.iFsName, aFsDesc.iPExtName, aDrive, aFsDesc.iDriveSynch);

    }

    else

    {//-- the FS did not have primary extension

        nRes = aFs.MountFileSystem(aFsDesc.iFsName, aDrive, aFsDesc.iDriveSynch);

    }

    return nRes;

}

//-------------------------------------------------------------------------------------------------------------------

/**

    Format volume, regardless the file system installed.

    @param  aFs             reference to the FS session

    @param  aDrive          drive number

    @param  aQuickFormat    if True, a quick format will be performed. otherwise - full

    @return system-wide error codes.

*/

TInt F32_Test_Utils::FormatDrive(RFs &aFs, TInt aDrive, TBool aQuickFormat)

{

    TPtrC fmtTypeName = (aQuickFormat ? _L("Quick") : _L("Full"));

    DoPrintf(_L("~ F32_Test_Utils::FormatDrive() drv:%d, type:%S\n"),aDrive, &fmtTypeName);

    ASSERT(aDrive >= EDriveA && aDrive <= EDriveZ);

    RFormat format;

    TUint   fmtMode=0;

    TInt    fmtCnt=0;

    TInt    prevCnt;

    TInt    nRes;

    if(aQuickFormat) 

        fmtMode |= EQuickFormat;

    //if(aForceErase)

    //    fmtMode |= EForceErase;

    TBuf<10> drvName;

    drvName.Format(_L("%C:"),'A'+aDrive);

    nRes = format.Open(aFs, drvName, fmtMode, fmtCnt);

    if(nRes!=KErrNone)

        goto Fail;

    //-- do format steps

    prevCnt=fmtCnt;

    while(fmtCnt)

    {

        nRes = format.Next(fmtCnt);

        if(nRes!=KErrNone)

            goto Fail;

        if(fmtCnt != prevCnt)

        {

            DoPrintf(_L("."));

            prevCnt = fmtCnt;

        }

    }

    //-- formatting has finished

    DoPrintf(_L("\n"));

    format.Close();

    return KErrNone;

   Fail:

    format.Close();

    DoPrintf(_L("~ F32_Test_Utils::FormatFatDrive() failed! code:%d\n"), nRes);

    return nRes;

}

#endif //FAT_UTILS_LEAN_AND_MEAN

//-------------------------------------------------------------------------------------------------------------------

/**

    printing interface. Prints out to the console (if is set) and to the debug interface

    if pConsole is NULL will print to the debug port only.

*/ 

void F32_Test_Utils::DoPrintf(TRefByValue<const TDesC> aFmt,...)

{

#ifndef FAT_UTILS_LEAN_AND_MEAN

    if(!bPrintOutEnabled) 

        return; //-- disabled by global flag

#endif //FAT_UTILS_LEAN_AND_MEAN

    VA_LIST list;

    VA_START(list, aFmt);

    TBuf<0x100> buf;

    buf.FormatList(aFmt, list); //-- ignore overflows

#ifndef FAT_UTILS_LEAN_AND_MEAN

    if(pConsole)

    {

        pConsole->Write(buf);

    }

#endif //FAT_UTILS_LEAN_AND_MEAN

    const TInt bufLen = buf.Length();

    if(bufLen >0 && buf[bufLen-1] == '\n')

    {

        buf.Insert(bufLen-1, _L("\r"));

    }

    RDebug::RawPrint(buf);

}

//-------------------------------------------------------------------------------------------------------------------

TBool F32_Test_Utils::IsPowerOf2(TUint32 aVal)

    {

    if (aVal==0)

        return EFalse;

    return !(aVal & (aVal-1));

    }

//-------------------------------------------------------------------------------------------------------------------

TUint32 F32_Test_Utils::Log2(TUint32 aVal)

{

    __ASSERT_COMPILE(sizeof(TUint32) == 4);

    ASSERT(aVal);

    TUint32 bitPos=31;

    if(!(aVal >> 16)) {bitPos-=16; aVal<<=16;}

    if(!(aVal >> 24)) {bitPos-=8;  aVal<<=8 ;}

    if(!(aVal >> 28)) {bitPos-=4;  aVal<<=4 ;}

    if(!(aVal >> 30)) {bitPos-=2;  aVal<<=2 ;}

    if(!(aVal >> 31)) {bitPos-=1;}

    return bitPos;

}

//-------------------------------------------------------------------------------------------------------------------

//###################################################################################################################

//#     TMD5 class implementation 

//###################################################################################################################

#define T_MASK ((TUint32)~0)

#define T1 /* 0xd76aa478 */ (T_MASK ^ 0x28955b87)

#define T2 /* 0xe8c7b756 */ (T_MASK ^ 0x173848a9)

#define T3    0x242070db

#define T4 /* 0xc1bdceee */ (T_MASK ^ 0x3e423111)

#define T5 /* 0xf57c0faf */ (T_MASK ^ 0x0a83f050)

#define T6    0x4787c62a

#define T7 /* 0xa8304613 */ (T_MASK ^ 0x57cfb9ec)

#define T8 /* 0xfd469501 */ (T_MASK ^ 0x02b96afe)

#define T9    0x698098d8

#define T10 /* 0x8b44f7af */ (T_MASK ^ 0x74bb0850)

#define T11 /* 0xffff5bb1 */ (T_MASK ^ 0x0000a44e)

#define T12 /* 0x895cd7be */ (T_MASK ^ 0x76a32841)

#define T13    0x6b901122

#define T14 /* 0xfd987193 */ (T_MASK ^ 0x02678e6c)

#define T15 /* 0xa679438e */ (T_MASK ^ 0x5986bc71)

#define T16    0x49b40821

#define T17 /* 0xf61e2562 */ (T_MASK ^ 0x09e1da9d)

#define T18 /* 0xc040b340 */ (T_MASK ^ 0x3fbf4cbf)

#define T19    0x265e5a51

#define T20 /* 0xe9b6c7aa */ (T_MASK ^ 0x16493855)

#define T21 /* 0xd62f105d */ (T_MASK ^ 0x29d0efa2)

#define T22    0x02441453

#define T23 /* 0xd8a1e681 */ (T_MASK ^ 0x275e197e)

#define T24 /* 0xe7d3fbc8 */ (T_MASK ^ 0x182c0437)

#define T25    0x21e1cde6

#define T26 /* 0xc33707d6 */ (T_MASK ^ 0x3cc8f829)

#define T27 /* 0xf4d50d87 */ (T_MASK ^ 0x0b2af278)

#define T28    0x455a14ed

#define T29 /* 0xa9e3e905 */ (T_MASK ^ 0x561c16fa)

#define T30 /* 0xfcefa3f8 */ (T_MASK ^ 0x03105c07)

#define T31    0x676f02d9

#define T32 /* 0x8d2a4c8a */ (T_MASK ^ 0x72d5b375)

#define T33 /* 0xfffa3942 */ (T_MASK ^ 0x0005c6bd)

#define T34 /* 0x8771f681 */ (T_MASK ^ 0x788e097e)

#define T35    0x6d9d6122

#define T36 /* 0xfde5380c */ (T_MASK ^ 0x021ac7f3)

#define T37 /* 0xa4beea44 */ (T_MASK ^ 0x5b4115bb)

#define T38    0x4bdecfa9

#define T39 /* 0xf6bb4b60 */ (T_MASK ^ 0x0944b49f)

#define T40 /* 0xbebfbc70 */ (T_MASK ^ 0x4140438f)

#define T41    0x289b7ec6

#define T42 /* 0xeaa127fa */ (T_MASK ^ 0x155ed805)

#define T43 /* 0xd4ef3085 */ (T_MASK ^ 0x2b10cf7a)

#define T44    0x04881d05

#define T45 /* 0xd9d4d039 */ (T_MASK ^ 0x262b2fc6)

#define T46 /* 0xe6db99e5 */ (T_MASK ^ 0x1924661a)

#define T47    0x1fa27cf8

#define T48 /* 0xc4ac5665 */ (T_MASK ^ 0x3b53a99a)

#define T49 /* 0xf4292244 */ (T_MASK ^ 0x0bd6ddbb)

#define T50    0x432aff97

#define T51 /* 0xab9423a7 */ (T_MASK ^ 0x546bdc58)

#define T52 /* 0xfc93a039 */ (T_MASK ^ 0x036c5fc6)

#define T53    0x655b59c3

#define T54 /* 0x8f0ccc92 */ (T_MASK ^ 0x70f3336d)

#define T55 /* 0xffeff47d */ (T_MASK ^ 0x00100b82)

#define T56 /* 0x85845dd1 */ (T_MASK ^ 0x7a7ba22e)

#define T57    0x6fa87e4f

#define T58 /* 0xfe2ce6e0 */ (T_MASK ^ 0x01d3191f)

#define T59 /* 0xa3014314 */ (T_MASK ^ 0x5cfebceb)

#define T60    0x4e0811a1

#define T61 /* 0xf7537e82 */ (T_MASK ^ 0x08ac817d)

#define T62 /* 0xbd3af235 */ (T_MASK ^ 0x42c50dca)

#define T63    0x2ad7d2bb

#define T64 /* 0xeb86d391 */ (T_MASK ^ 0x14792c6e)

TMD5::TMD5()

{

    Reset();

}

//-------------------------------------------------------------------------------------------------------------------

void TMD5::Md5_process(const TUint8 *data /*[64]*/)

{

    TUint32

    a = iState.abcd[0], b = iState.abcd[1],

    c = iState.abcd[2], d = iState.abcd[3];

    TUint32 t;

    TUint32 xbuf[16];

    const TUint32 *X;

    {

        static const TInt w = 1;

        if (*((const TUint8 *)&w)) 

        {

            if (!((data - (const TUint8 *)0) & 3)) {

            X = (const TUint32 *)data;

            } else {

            memcpy(xbuf, data, 64);

            X = xbuf;

            }

        }

        else

        {

            const TUint8 *xp = data;

            TInt i;

            X = xbuf;       /* (dynamic only) */

            for (i = 0; i < 16; ++i, xp += 4)

            xbuf[i] = xp[0] + (xp[1] << 8) + (xp[2] << 16) + (xp[3] << 24);

        }

    }

#define ROTATE_LEFT(x, n) (((x) << (n)) | ((x) >> (32 - (n))))

#define F(x, y, z) (((x) & (y)) | (~(x) & (z)))

#define SET(a, b, c, d, k, s, Ti)\

  t = a + F(b,c,d) + X[k] + Ti;\

  a = ROTATE_LEFT(t, s) + b

    SET(a, b, c, d,  0,  7,  T1);

    SET(d, a, b, c,  1, 12,  T2);

    SET(c, d, a, b,  2, 17,  T3);