/*

* Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "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:  Monitors file creations, modifications and deletions.

*

*/

#include <f32pluginutils.h>

#include "fastfindfileserverplugin.h"

#include "harvesterlog.h"

_LIT( KFastFindFileServerPlugin, "CPixFileServerPlugin" );

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

// CFastFindFileServerPlugin implementation

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

//

CFastFindFileServerPlugin::CFastFindFileServerPlugin()

: iFormatDriveNumber( -1 )

	{

	WRITELOG( "CFastFindFileServerPlugin::CFastFindFileServerPlugin()" );

	}

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

// 

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

//

CFastFindFileServerPlugin::~CFastFindFileServerPlugin()

    {

    WRITELOG( "CFastFindFileServerPlugin::~CFastFindFileServerPlugin()" );

    TRAP_IGNORE( DisableL() );

    iFsSession.Close();

    iCreatedFiles.ResetAndDestroy();

    iCreatedFiles.Close();

    iPaths.ResetAndDestroy();

    iPaths.Close();

    iIgnorePaths.ResetAndDestroy();

    iIgnorePaths.Close();

    iQueue.ResetAndDestroy();

    iQueue.Close();

    }

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

// 

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

//

CFastFindFileServerPlugin* CFastFindFileServerPlugin::NewL()

    {

    WRITELOG( "CFastFindFileServerPlugin::NewL()" );

    return new (ELeave) CFastFindFileServerPlugin;

    }

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

// 

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

//

void CFastFindFileServerPlugin::InitialiseL()

    {

    WRITELOG( "CFastFindFileServerPlugin::InitializeL()" );

    User::LeaveIfError( iFsSession.Connect() );

    }

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

// 

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

//

void CFastFindFileServerPlugin::EnableL()

    {

    WRITELOG( "CFastFindFileServerPlugin::EnableL()" );

    User::LeaveIfError( RegisterIntercept(EFsFileCreate, EPostIntercept) );

    User::LeaveIfError( RegisterIntercept(EFsFileRename, EPostIntercept) );

    User::LeaveIfError( RegisterIntercept(EFsRename, EPostIntercept) );

    User::LeaveIfError( RegisterIntercept(EFsDelete, EPostIntercept) );

//    User::LeaveIfError( RegisterIntercept(EFsFileReplace, EPostIntercept) );

    User::LeaveIfError( RegisterIntercept(EFsReplace, EPostIntercept) );

//    User::LeaveIfError( RegisterIntercept(EFsFileModified, EPostIntercept) );

    User::LeaveIfError( RegisterIntercept(EFsFileSetModified, EPostIntercept) );

    User::LeaveIfError( RegisterIntercept(EFsFileSubClose, EPostIntercept) );

	// format events

	User::LeaveIfError( RegisterIntercept(EFsFormatSubClose, EPostIntercept) );

	User::LeaveIfError( RegisterIntercept(EFsFormatOpen, EPostIntercept) );

#ifdef _DEBUG_EVENTS

    RegisterDebugEventsL();

#endif

    }

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

// 

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

//

void CFastFindFileServerPlugin::DisableL()

    {

    WRITELOG( "CFastFindFileServerPlugin::DisableL()" );

    User::LeaveIfError( UnregisterIntercept(EFsFileCreate, EPrePostIntercept) );

    User::LeaveIfError( UnregisterIntercept(EFsFileRename, EPrePostIntercept) );

    User::LeaveIfError( UnregisterIntercept(EFsRename, EPrePostIntercept) );

    User::LeaveIfError( UnregisterIntercept(EFsDelete, EPrePostIntercept) );

//    User::LeaveIfError( UnregisterIntercept(EFsFileReplace, EPrePostIntercept) );

    User::LeaveIfError( UnregisterIntercept(EFsReplace, EPrePostIntercept) );

//    User::LeaveIfError( UnregisterIntercept(EFsFileModified, EPrePostIntercept) );

    User::LeaveIfError( UnregisterIntercept(EFsFileSetModified, EPrePostIntercept) );

    User::LeaveIfError( UnregisterIntercept(EFsFileSubClose, EPrePostIntercept) );

	// format events

	User::LeaveIfError( UnregisterIntercept(EFsFormatSubClose, EPostIntercept) );

	User::LeaveIfError( UnregisterIntercept(EFsFormatOpen, EPostIntercept) );

#ifdef _DEBUG_EVENTS

    UnregisterDebugEventsL();

#endif

    }

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

// AddConnection

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

//

void CFastFindFileServerPlugin::AddConnection()

    {

    WRITELOG( "CFastFindFileServerPlugin::AddConnection()" );

    ++iConnectionCount;

    }

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

// RemoveConnection

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

//

void CFastFindFileServerPlugin::RemoveConnection()

    {

    WRITELOG( "CFastFindFileServerPlugin::RemoveConnection()" );

    --iConnectionCount;

    // remove notification request if this was last connection

    if ( iConnectionCount <= 0 )

        {

        WRITELOG( "CFastFindFileServerPlugin::RemoveConnection() last connection" );

        iNotification = NULL;

        }

    }

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

// 

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

//

TInt CFastFindFileServerPlugin::DoRequestL( TFsPluginRequest& aRequest )

	{

	TInt err = KErrNone;

	TInt function = aRequest.Function();

	const TBool formatFunction = function == EFsFormatOpen || function == EFsFormatSubClose;

#ifdef _DEBUG_EVENTS

    PrintDebugEvents( function );

#endif

    if ( function == EFsFileSubClose && iCreatedFiles.Count() == 0 )

        {

        return KErrNone;

        }

    WRITELOG1( "----- CFastFindFileServerPlugin::DoRequestL() - plugin function: %d -----", function );

    TFileName fileName;

    fileName.Zero();

    TFileName newFileName;

    newFileName.Zero();

	if ( !formatFunction )

		{

		GetName( &aRequest, fileName );

		WRITELOG1( "CFastFindFileServerPlugin::DoRequestL() - fileName: %S", &fileName );

		}

    // get process id

	TUid processId = { 0 };

    /*RThread clientThread;

    RProcess clientProcess;

    err = aRequest.Message().Client( clientThread, EOwnerThread );

    if ( err == KErrNone )

        {

        err = clientThread.Process( clientProcess );

        if ( err == KErrNone )

            {

            processId = clientProcess.Identity();

            }

        }

    clientProcess.Close();

    clientThread.Close();*/

	processId = aRequest.Message().Identity();

    // if rename, check destination path

    if ( function == EFsRename || function == EFsFileRename || function == EFsReplace )

        {

        GetNewName( &aRequest, newFileName );

        WRITELOG1( "CFastFindFileServerPlugin::DoRequestL() - newFileName: %S", &newFileName );

        if ( !CheckPath(newFileName) )

            {

            WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - path not supported" );

            return KErrNone;

            }

        if ( !CheckAttribs(newFileName) )

            {

            WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - attribute check failed" );

            return KErrNone;

            }

        }

    else if ( !formatFunction )

        {

        if ( !CheckPath(fileName) )

            {

            WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - path not supported" );

            return KErrNone;

            }

        if ( !CheckAttribs(fileName) )

            {

            WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - attribute check failed" );

            return KErrNone;

            }

        }

    TInt fileEventType = EFastFindFileUnknown;

    TInt drvNumber = aRequest.DriveNumber();

    WRITELOG1( "CFastFindFileServerPlugin::DoRequestL() - drive number: %d", drvNumber );

    switch( function )

        {

        case EFsFileCreate:

            {

            WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - EFsFileCreate" );

            TFileName* fn = new (ELeave) TFileName;

            fn->Zero();

            fn->Copy( fileName );

            iCreatedFiles.Append( fn );

            User::LeaveIfError( UnregisterIntercept(EFsFileSetModified, EPrePostIntercept) );

            return KErrNone;

            }

        case EFsFileSubClose:

            {

            WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - EFsFileSubClose" );

            TBool found = EFalse;

            for ( TInt i = 0; i < iCreatedFiles.Count(); i++ )

                {

                TFileName* fn = iCreatedFiles[i];

                if ( fn->CompareC(fileName,1,NULL) == 0 )

                    {

                    fileEventType = EFastFindFileCreated;

                    delete fn;

                    fn = NULL;

                    iCreatedFiles.Remove( i );

                    found = ETrue;

                    User::LeaveIfError( RegisterIntercept(EFsFileSetModified, EPostIntercept) );

                    }

                }

            if ( !found )

                {

                return KErrNone;

                }

            }

            break;

        case EFsRename:

            {

            WRITELOG1( "CFastFindFileServerPlugin::DoRequestL() - EFsRenamed, new file: %S", &newFileName );

            fileEventType = EFastFindFileRenamed;

            }

            break;

        case EFsFileRename:

            WRITELOG1( "CFastFindFileServerPlugin::DoRequestL() - EFsFileRenamed, new file: %S", &newFileName );

            fileEventType = EFastFindFileRenamed;

            break;

        case EFsFileSetModified:

            WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - EFsFileModified" );

            fileEventType = EFastFindFileModified;

            break;

        case EFsReplace:

            WRITELOG1( "CFastFindFileServerPlugin::DoRequestL() - EFsReplace, new file: %S", &newFileName );

            fileEventType = EFastFindFileReplaced;

            break;

        case EFsDelete:

            WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - EFsDelete" );

            fileEventType = EFastFindFileDeleted;

            break;

		case EFsFormatOpen:

			WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - EFsFormatOpen" );

			// get the drive letter

			iFormatDriveNumber = drvNumber;

			fileEventType = EFastFindDriveFormatted;

			processId.iUid = 0;

			//return KErrNone;

			break;

		case EFsFormatSubClose:

			WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - EFsFormatSubClose" );

			if ( iFormatDriveNumber < 0 )

				{

				return KErrNone;

				}

			drvNumber = iFormatDriveNumber;

			iFormatDriveNumber = -1;

			fileEventType = EFastFindDriveFormatted;

			if ( processId.iUid == 0 )

				{

				processId.iUid = 1;

				}

			break;

        default:

            WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - Unknown function" );

            return KErrNone;

        }

    if ( iNotification )

        {

        WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - iNotification found" );

        TFastFindFSPStatusPckg clientStatusBuf;

        TFastFindFSPStatus& clientStatus = clientStatusBuf();

        clientStatus.iDriveNumber = drvNumber;

        clientStatus.iFileName.Copy( fileName );

        if ( newFileName.Length() > 0 )

            {

            clientStatus.iNewFileName.Copy( newFileName );

            }

        clientStatus.iFileEventType = fileEventType;

        clientStatus.iProcessId = processId;

        TRAP( err, iNotification->WriteParam1L(clientStatusBuf) );

        iNotification->Complete( err );

        iNotification = NULL;

        WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - iNotification complete" );

        }

    else // no notification ready, put in the queue

        {

        WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - iNotification not found. Put in the queue" );

        TFastFindFSPStatus* clientStatus = new (ELeave) TFastFindFSPStatus;

        clientStatus->iDriveNumber = drvNumber;

        clientStatus->iFileName.Copy( fileName );

        if ( newFileName.Length() > 0 )

            {

            clientStatus->iNewFileName.Copy( newFileName );

            }

        clientStatus->iFileEventType = fileEventType;

        clientStatus->iProcessId = processId;

        iQueue.Append( clientStatus ); // owenership is transferred

        err = KErrNone;

        WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - added to queue" );

        }

    WRITELOG( "CFastFindFileServerPlugin::DoRequestL() - return" );

    return err;

    }

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

// CFastFindFileServerPluginConn implementation

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

class CFastFindFileServerPluginConn : public CFsPluginConn

    {

    public:

        static CFastFindFileServerPluginConn* NewL( CFastFindFileServerPlugin& aPlugin );

        virtual ~CFastFindFileServerPluginConn();

        virtual TInt DoControl( CFsPluginConnRequest& aRequest );

        virtual void DoRequest( CFsPluginConnRequest& aRequest );

        virtual void DoCancel( TInt aReqMask );

    private:

        CFastFindFileServerPluginConn( CFastFindFileServerPlugin& aPlugin );

        CFastFindFileServerPlugin& iPlugin;

        RMessage2* iMessage;

    };

/**

* Leaving New function for the plugin

* @internalComponent

*/

CFastFindFileServerPluginConn* CFastFindFileServerPluginConn::NewL( 

		CFastFindFileServerPlugin& aPlugin )

    {

    WRITELOG( "CFastFindFileServerPluginConn::NewL()" );

    return new (ELeave) CFastFindFileServerPluginConn( aPlugin );

    }

/**

* Constructor for the plugin

* @internalComponent

*/

CFastFindFileServerPluginConn::CFastFindFileServerPluginConn(

               CFastFindFileServerPlugin& aPlugin )

  : iPlugin( aPlugin )

    {

    WRITELOG( "CFastFindFileServerPluginConn::CFastFindFileServerPluginConn()" );

    iPlugin.AddConnection();

    }

/**

* The destructor for the test virus scanner hook.

* @internalComponent

*/