/*

* Copyright (c) 2009 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:  Implementation of ECOM plug-in service interface providing

*               the FTP service.

*

*/

#ifndef HTIFTPSERVICEPLUGIN_H__

#define HTIFTPSERVICEPLUGIN_H__

// INCLUDES

#include <e32std.h>

#include <f32file.h>

#include <HtiServicePluginInterface.h>

// CONSTANTS

enum TFtpCommand

    {

    EFtpSTOR         = 0x02,

    EFtpSTOR_u       = 0x03,

    EFtpRETR         = 0x04,

    EFtpRETR_u       = 0x05,

    EFtpLIST         = 0x06,

    EFtpLIST_u       = 0x07,

    EFtpMKD          = 0x08,

    EFtpMKD_u        = 0x09,

    EFtpRMD          = 0x0A,

    EFtpRMD_u        = 0x0B,

    EFtpDELE         = 0x0C,

    EFtpDELE_u       = 0x0D,

    EFtpCANCEL       = 0x0E,

    EFtpFILESIZE     = 0x0F,

    EFtpLISTDIR      = 0x10,

    EFtpLISTDIR_u    = 0x11,

    EFtpLISTSIZES    = 0x12,

    EFtpLISTSIZES_u  = 0x13,

    EFtpLISTDRIVES   = 0x14,

    EFtpLISTDRIVES_u = 0x15,

    EFtpRENAME       = 0x16,

    EFtpRENAME_u     = 0x17,

    EFtpCOPY         = 0x18,

    EFtpCOPY_u       = 0x19,

    EFtpMOVE         = 0x1A,

    EFtpMOVE_u       = 0x1B,

    EFtpSETFORCE     = 0x20,

    EFtpCHECKSUM     = 0x30,

    EFtpCHECKSUM_u   = 0x31,

    EFtpFORMAT       = 0x40,

    EFtpOK           = 0xF0,

    };

enum TAlgorithm

    {

    EMD5 = 0x01,

    };

_LIT(KHtiFileHlp, "HtiFileHlp.exe");

//temporary file used if file is uploaded to Tcb

_LIT(KTmpFileName, "C:\\htitemp.bin");

//tokens used for command line for HtiFileHlp.exe

_LIT( KHtiFileHlpDeleteCmd, "d" );

_LIT( KHtiFileHlpCopyCmd,   "c" );

_LIT( KHtiFileHlpMkdCmd,    "m" );

_LIT( KHtiFileHlpRmdCmd,    "r" );

_LIT( KHtiFileHlpMoveCmd,   "mv" );

_LIT( KHtiFileHlpRenameCmd, "re" );

_LIT( KHtiFileHlpDelim,     "|" );

//tokens used to id Tcb folders

_LIT(KHtiTcbSys, "\\sys\\");

_LIT(KHtiTcbResource, "\\resource\\");

//offset where first folder's backslash starts in absolute filename

                                   // 012

const static TInt KPathOffset = 2; //"C:\"

// MACROS

// DATA TYPES

// FUNCTION PROTOTYPES

// FORWARD DECLARATIONS

class CHtiFtpBackupFakeBase;

// CLASS DECLARATION

/**

* Callback interface for AO

*/

class MFtpObserverAO

    {

public:

    virtual void FtpComplete( TInt anError) = 0;

    };

// CLASS DECLARATION

/**

* AO for async request to CFileMan and RFile

*/

class CFtpHandlerAO : public CActive, public MFileManObserver

    {

public:

    CFtpHandlerAO(MFtpObserverAO* anObserver);

    ~CFtpHandlerAO();

    /** Set AO active */

    void Start();

public: //MFileManObserver

    virtual MFileManObserver::TControl NotifyFileManStarted();

    virtual MFileManObserver::TControl NotifyFileManOperation();

    virtual MFileManObserver::TControl NotifyFileManEnded();

protected: //from CActive

    void RunL();

    /**

    * no direct cancel for RFile and CFileMan

    */

    void DoCancel();

    //TInt RunError(TInt aError);

protected:

    MFtpObserverAO* iObserver;

    /** set to ERtue if CFileMan operation should be canceled */

    TBool iCancelFileMan;

    };

class CProcessLogonAO : public CActive

    {

public:

    CProcessLogonAO(MFtpObserverAO* anObserver);

    ~CProcessLogonAO();

    /** Set AO active */

    void Start(const TDesC& aCmdLine);

protected: //from CActive

    void RunL();

    void DoCancel();

protected:

    MFtpObserverAO* iObserver;

    RProcess iProcess;

    };

// CLASS DECLARATION

/**

* FTP plugin implementation

*/

class CHtiFtpServicePlugin : public CHTIServicePluginInterface,

                             public MFtpObserverAO

    {

protected:

    /**

    * Major states of FPT plug-in

    */

    enum TFtpServiceState

        {

        EIdle,          /** waits for a new command                 */

        EListBusy,      /** waits for memory to send LIST response  */

        ERmdBusy,       /** outstanding request CFileMan::RmDir()   */

        EDeleBusy,      /** outstanding request CFileMan::Delete()  */

        EStorWait,      /** waits for data packages                 */

        EStorBusy,      /** outstanding request to RFile::Write()   */

        ERetrBusy,      /** outstanding request to RFile::Read()    */

        ERetrWait,      /** waits for memory to send data package   */

        EStorTcbBusy,   /** outstandig request to HtiFileHlp.exe    */

        EDeleTcbBusy,   /** outstandig request to HtiFileHlp.exe    */

        EMkdTcbBusy,    /** outstandig request to HtiFileHlp.exe    */

        ERmdTcbBusy,    /** outstandig request to HtiFileHlp.exe    */

        ERenameBusy,    /** outstanding request CFileMan::Rename()  */

        ERenameTcbBusy, /** outstandig request to HtiFileHlp.exe   */

        ECopyBusy,      /** outstanding request CFileMan::Copy()    */

        ECopyTcbBusy,   /** outstandig request to HtiFileHlp.exe   */

        EMoveBusy,      /** outstanding request CFileMan::Move()    */

        EMoveTcbBusy    /** outstandig request to HtiFileHlp.exe   */

        };

public:

    static CHtiFtpServicePlugin* NewL();

    // Interface implementation

    /**

    * Calculates iBufferSize based on iDispatcher->FreeMemory() value

    */

    void InitL();

    /**

    * @return ETrue when in any EXXXBusy state

    */

    TBool IsBusy();

    /**

    * Process either control or data message depending on aPriority

    *

    * @param aMessage message with commands or data

    * @param aPriority indicates type of aMessage

    */

    void ProcessMessageL( const TDesC8& aMessage,

                         THtiMessagePriority aPriority );

    void NotifyMemoryChange( TInt aAvailableMemory );

public: // MFtpObserverAO

    void FtpComplete( TInt anError );

protected:

    CHtiFtpServicePlugin();

    void ConstructL();

    virtual ~CHtiFtpServicePlugin();

    /**

    * Handle FTP control messages

    *

    * @param aMessage message with command

    */

    void HandleControlMessageL( const TDesC8& aMessage );

    /**

    * Handle expected CANCEL control message

    * received in EStorWait or ERetrWait state

    * If aMessage does not contain CANCEL command,

    * BUSY message is sent, otherwise current operation is canceled

    * and plugin goes to EIdle state and sends OK message.

    *

    * @param aMessage message with command

    */

    void HandleCancelL( const TDesC8& aMessage );

    /**

    * Accepts data messages for STOR command

    *

    * @param aMessage message with file data

    */

    void HandleDataMessageL( const TDesC8& aMessage );

    //command handlers

    /**

    * Handle STOR command request

    */

    void HandleReceiveFileL();

    /**

    * Handle RETR command request

    */

    void HandleSendFileL();

    /**

    * Reads one portion of file data during RETR command handeling.

    */

    void ReadToBuffer();

    /**

    * Sends one portion of file data during RETR command handeling.

    */

    void SendBuffer();

    /**

    * Handle LIST command

    *

    * @param aUnicodText if ETrue then response in unicode

    * @param aReadingAtt specifies what entries to read from a dir

    * @param aSizes if ETrue filesizes are included in the response

    */

    void HandleListL( TBool aUnicodText, TUint aReadingAtt, TBool aSizes );

    /**

    * Extracts and validate file name to iFileName

    * If error sends err msg

    * Return ETrue when filename is valid

    * @param aFilename descriptor with filename

    * @param aToUnicode flag indicates that data in aFilename

    *                   should be treated as an unicode string

    */

    TBool GetFileNameL( const TDesC8& aFilename, TBool aToUnicode );

    /**

    * Extracts and validates path to iFileName

    * If error sends err msg

    * Return ETrue when directory is valid

    * @param aDirname descriptor with directory name

    * @param aToUnicode flag indicates that data in aFilename

    *                   should be treated as an unicode string

    */

    TBool GetDirectoryL( const TDesC8& aDirname, TBool aToUnicode );

    /**

    * Helper to send short control messages

    * (e.g. OK, BUSY, FILESIZE)

    * @param aCmd command code

    * @param aMsg additional command parameters

    * @return KErrNone or Symbian error code

    */

    TInt SendControlMsg( TFtpCommand aCmd, const TDesC8& aMsg );

    /**

    * Helper to send error message

    * @return KErrNone or some system-wide error code

    */

    inline TInt SendErrorMsg( TInt anError, const TDesC8& aMsg );

    /**

    * Handle delete command

    * @param aFilename the file to delete

    */

    void HandleDeleteL( const TDesC& aFilename );

    /**

    * Handle rename command

    * @param aMessage message that contains

    * old and new file/directory names.

    * @param aToUnicode flag indicates that data in aFilename

    *  should be treated as an unicode string

    */

    void HandleRenameL( const TDesC8& aMessage, TBool aUnicode );

    /**

    * Handle copy command

    * @param aMessage message that contains the file/directory to be copied

    * and the location and name of the copy.

    * @param aToUnicode flag indicates that data in aFilename

    *  should be treated as an unicode string

    */

    void HandleCopyL( const TDesC8& aMessage, TBool aUnicode );

    /**

    * Handle move command

    * @param aMessage message that contains the file/directory to be moved

    * and the new location

    * @param aToUnicode flag indicates that data in aFilename

    *  should be treated as an unicode string

    */

    void HandleMoveL( const TDesC8& aMessage, TBool aUnicode );

    /**

    * Safe unicode copying from 8bit descr to 16bit buffer

    * @param aTo destination descriptor

    * @param aFrom source descriptor

    */

    void CopyUnicode( TDes& aTo, const TDesC8& aFrom );

    /**

    * Check either aFilename point in a TCB directories (sys, private, resource)

    * @param aFilename the file to check

    */

    TBool IsFileTcb( const TDesC& aFilename );

    /**

    * Uses HtiFileHlp.exe to delete file from TCB directories

    * @param aFilename the file to delete

    */

    void HandleTcbDeleteL( const TDesC& aFilename );

    /**

    * Uses HtiFileHlp.exe to copy files/folders to TCB directories

    * @param aFromFilename the source path

    * @param aToFilename the destination path

    */

    void HandleTcbCopyL( const TDesC& aFromFilename, const TDesC& aToFilename );

    /**

    * Uses HtiFileHlp.exe to makedir in TCB directories

    * @param aDirname the directory to create

    */

    void HandleTcbMkdL( const TDesC& aDirname );

    /**

    * Uses HtiFileHlp.exe to delete dir in TCB directories

    * @param aDirname the directory to delete

    */

    void HandleTcbRmdL( const TDesC& aDirname );

    /**

    * Uses HtiFileHlp.exe to do rename in TCB folders

    * @param aTargetName the path to rename

    * @param aDestName the new path name

    */

    void HandleTcbRenameL( const TDesC& aTargetName, const TDesC& aDestName );

    /**

    * Uses HtiFileHlp.exe to do move to TCB folders

    * @param aTargetName the path to move

    * @param aDestName the destination path

    */

    void HandleTcbMoveL( const TDesC& aTargetName, const TDesC& aDestName );

    /**

    * Sets the faking of backup/restore operation on/off.

    * If aOn is ETrue sets backup/restore on, otherwise sets it off.

    * @return KErrNone on success, otherwise system error code

    */

    TInt SetBURFakeState( TBool aOn );

    /**

    * Calculates a checksum for the given file with the given algorithm.

    */

    void HandleCheckSumCalcL( TAlgorithm aAlgorithm,

                              const TDesC& aFilename );

    /**

    * Formats a drive.

    */

    void HandleFormat( const TUint8 aDrive, const TUint8 aMode );

    /**

    * Creates a list of drives.

    */

    void HandleListDrivesL( TBool aUnicode );

private:

    void RemoveEndBackslash( TFileName& aFileName );

protected:

    /** plugin state */

    TFtpServiceState iState;

    /** File server */

    RFs iFs;

    /** File manager used for RMD and DELE */

    CFileMan* iFileMan;

    /** current file for STOR and RETR*/

    RFile iFile;

    /** file name of iFile */

    TFileName iFileName;

    /** current offset in file during STOR or RETR */

    TInt iCurrentOffset; //mean received bytes or sent bytes

    /** the final size of iFile */

    TInt iFileSize;

    /** the size of iSendBuffer, calculated in InitL() */

    TInt iBufferSize;

    /** send buffer for RETR and LIST commands */

    HBufC8* iSendBuffer;

    /** des of iSendBuffer, used for RETR */

    TPtr8 iSendBufferDes;

    /** AO to handle current asyn request */

    CFtpHandlerAO* iHandlerAO;

    /** used for HtiFileHlp.exe */

    CProcessLogonAO* iProcessLogonAO;

    /**

    * Class for activating/deactivating a backup session.

    * This is used for forced operations to locked files. Activating backup

    * should release locks on files that are in use by some other application.

    */

    CHtiFtpBackupFakeBase* iBackupFake;

    /** Handle to the DLL for activating/deactivating backup */

    RLibrary iBackupFakeLib;

    };

#endif

// End of File