#include    <e32std.h>

#include    <e32base.h>

#include    <f32file.h>

#include    <DRMCommon.h>

#include    <driveinfo.h>

#include    "FileScan.h"

// EXTERNAL FUNCTION PROTOTYPES

    if ( !aDir.CompareF( KIgnoreDir1 ) ||

    aFs.DriveToChar( driveNumber, driveLetter );

    TFileName ignore1;

    TFileName ignore2;

    ignore1.Format( KIgnoreDir1, (TUint)driveLetter );

    if ( !aDir.CompareF( ignore1 ) ||

CFileScan::CFileScan( RFs& aFs ) :

    {

    CleanInternal();

    TInt err = 0;

    if ( !iFs )

        {

        err = KErrArgument;

        }

    else

        {

        err = KErrNone;

        }

    User::LeaveIfError( err );

    }

    CFileScan* self = new( ELeave ) CFileScan( aFs );

    CleanupStack::PushL( self );

    CleanInternal();

    iServer = NULL;

    iFs = NULL;

    {

    CleanInternal();

    }

// Function returns whether the specific file is protected or not

    {

    TInt err = KErrNone;

    aIsDCF = EFalse;

    err = iServer->ProcessFile( aFileName , aIsDCF );

    if ( err && err != KErrNoMemory )

        {

        err = KErrNone;

        }

    return err;

    }

    {

    TInt err = KErrNone;

    if ( iSearching && iServer->State()!=EStateIdle )

        {

        err = SearchNext();

        if ( err == KErrCancel )

            {

            err = KErrNone;

            }

        if ( !err )

            {

            SetActive();

            TRequestStatus* status = &iStatus;

            User::RequestComplete( status , err );

            }

        }

    else

        {

        CleanInternal();

        iServer->CompleteScanning(err);

        Deque();

        }

    if ( err )

        {

        CleanInternal();

        iServer->CompleteScanning(err);

        Deque();

        }

    }

// Function starts the active objects to search protected file through whole file system

    {

    iServer = aServer;

    CActiveScheduler::Add( this );

    iSearching = ETrue;

    err = SearchNext();

    if ( err )

        {

        return err;

        }

    SetActive();

    TRequestStatus* status = &iStatus;

    User::RequestComplete( status , KErrNone );

    return err;

    }

    {

    _LIT( KDrive, "%c:");

    TDriveList drivelist;

    TChar driveLetter;

    TInt driveNumber = EDriveA - 1;

    TInt err = KErrNone;

    TInt ramDrive( -1 );

    err = iFs->DriveList( drivelist );

    if ( iLastPosition != KNullDesC )

        {

        driveLetter = iLastPosition[0];

        err = iFs->CharToDrive( driveLetter, driveNumber );

        }

    driveNumber++;

        if ( driveNumber == EDriveD || driveNumber == EDriveZ )

            {

            }

        if ( driveNumber == ramDrive || driveNumber == romDrive )

            {

            }

        else if ( drivelist[driveNumber] )

                {

                err = iFs->DriveToChar( driveNumber, driveLetter );

                iLastPosition.Format( KDrive, (TUint)driveLetter );

                iDeeper = ETrue;

                return err;

                }

    CleanInternal();

    return err;

    }

    {

    TInt err = KErrNone;

    TRAP( err , iCurrentLeaf->SetLeafL( aFolderList ) );

    return err;

    }

// This function searches for the DCF files under specific folder

    {

    TInt err = KErrNone;

    TInt i = 0;

    TBool isDCF;

    _LIT ( KFullFileName , "%S\\%S");

    TFileName fileName;

    for ( ; i < aFileList->Count() && !err ; i++ )

        {

        if (!( *aFileList )[i].IsDir())

            {

            TPtrC extension = (*aFileList)[i].iName.Right( 4 );

            if( !extension.CompareF( KWma ) ||

                !extension.CompareF( KWmv ) ||

                !extension.CompareF( KAsf ) )

                {

                }

            else

                {

                fileName.Format( KFullFileName , &iLastPosition , &( *aFileList )[i].iName );

                err = IsProtected( fileName , isDCF );

                }

            }

        }

    return err;

    }

    {

    TInt err = KErrNone;

    err = CheckDrive();

    if ( !err && iSearching )

        {

        if ( iDeeper )

            {

            err = CheckFolder();

            }

        if ( !err && iSearching )

            {

            err = ToNextLeaf();

            }

        }

    return err;

    }

    {

    // check if current drive is end of searching

    TInt err = KErrNone;

    if( !iCurrentLeaf )

        {

        err = SearchDrive();

        if ( err || !iSearching )

            {

            return err;

            }

        CSearchLeaf* root = NULL;

        TRAP( err , iCurrentLeaf = CSearchLeaf::NewL( root , iLastPosition ) );

        }

    return err;

    }

TInt CFileScan::CheckFolder()

    {

    // check current folder

    _LIT ( KSearchDir , "%S\\*");

    TInt err = KErrNone;

    CDir* fileList = NULL;

    CDir* dirList = NULL;

    TFileName temp;

    if ( IgnoreDir( *iFs, iLastPosition ) )

        temp.Format( KSearchDir , &iLastPosition );

        err = iFs->GetDir( temp

            , KEntryAttMaskSupported

            , ESortByName

            , fileList , dirList );

        if ( !err )

            {

            err = SearchFolder( dirList );

            if ( !err )

                {

                err = SearchFile( fileList );

                }

            }

        delete fileList;

        fileList = NULL;

        delete dirList;

        dirList = NULL;

    return err;

    }

    {

    _LIT ( KChildDir , "%S\\%S");

    TInt err = KErrNone;

    TFileName file;

    CSearchLeaf* temp = NULL;

    file.Format( iLastPosition );

    if ( iCurrentLeaf->LeafList().Count() > 0 )

        {

        iDeeper = ETrue;

        iCurrentLeaf = iCurrentLeaf->LeafList()[0];

        iLastPosition.Format( KChildDir , &file , &iCurrentLeaf->FolderName() );

        }

    else

        {

        iDeeper = EFalse;

        temp = iCurrentLeaf;

        iCurrentLeaf = iCurrentLeaf->Root();

        if ( iCurrentLeaf )

            {

            iCurrentLeaf->RemoveLeaf( temp );

            }

        delete temp;

        temp = NULL;

        err = UpFolder();

        }

    return err;

    }

    {

    TInt err = KErrNone;

    TParse file;

    if ( iLastPosition.Length()<3 )

        {

        return err;

        }

    err = file.Set( iLastPosition , NULL , NULL );

    iLastPosition.Format( file.DriveAndPath() );

    iLastPosition.SetLength( iLastPosition.Length() - 1 );

    return err;

    }

    {

    CSearchLeaf* root = iCurrentLeaf;

    iLastPosition = KNullDesC;