/*

* 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:  HtiFtpServicePlugin implementation

*

*/

// INCLUDE FILES

#include "HtiFtpServicePlugin.h"

#include <HtiFtpBackupFakeBase.h>

#include <HtiDispatcherInterface.h>

#include <HtiLogging.h>

#include <e32property.h>

#include <hash.h>

// CONSTANTS

static const TUid KFtpServiceUid = { 0x1020DEC5 };

_LIT( KBackslash, "\\" );

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

_LIT( KHtiFtpBackupFakeDllName, "HtiFtpBackupFake.dll" );

const static TInt KMinBufferSize = 1024;

const static TUint8 KUnicodeMask = 0x1;

const static TInt KFileSizeMsgSize = 5;

//error description

_LIT8( KErrDescrUnknownCmd, "unknown command" );

_LIT8( KErrDescrNoSpace, "no disk space" );

_LIT8( KErrDescrFailedRead, "failed read file" );

_LIT8( KErrDescrFailedWrite, "failed write file" );

_LIT8( KErrDescrFailedGetDir, "failed to read directory" );

_LIT8( KErrDescrEmptyDirname, "directory name empty" );

_LIT8( KErrDescrInvalidDirnameLength, "invalid directory name length" );

_LIT8( KErrDescrEmptyFilename, "file name empty" );

_LIT8( KErrDescrInvalidFilenameLength, "invalid file name length" );

_LIT8( KErrDescrFailedMkDir, "failed create directory" );

_LIT8( KErrDescrFailedRmDir, "failed remove directory" );

_LIT8( KErrDescrFailedDeleFile, "failed delete file" );

_LIT8( KErrDescrFailedRenameFile, "failed rename file" );

_LIT8( KErrDescrFailedCopyFile, "failed copy file" );

_LIT8( KErrDescrFailedMoveFile, "failed move file" );

_LIT8( KErrDescrInvalidStorArgument, "invalid arguments" );

_LIT8( KErrDescrFailedCreateFile, "invalid create file" );

_LIT8( KErrDescrFailedOpenFile, "failed open file" );

_LIT8( KErrDescrFailedCloseFile, "failed close file" );

_LIT8( KErrDescrInvalidDataMessage, "invalid data message" );

_LIT8( KErrDescrNoMemory, "no memory to send file" );

_LIT8( KErrDescrNoCancel, "nothing to cancel" );

_LIT8( KErrDescrBusy, "Busy" );

_LIT8( KErrDescrFailedCopyTcb, "Failed to copy to Tcb directories" );

_LIT8( KErrDescrInvalidForceArgs, "invalid arguments for setforce" );

_LIT8( KErrDescrInvalidChecksumArgs, "invalid arguments for checksum" );

_LIT8( KErrDescrInvalidFormatArgs, "invalid arguments for format" );

_LIT8( KErrDescrInvalidDriveListArgs, "invalid arguments for drive list" );

_LIT8( KErrDescrFailedFormat, "failed to format" );

_LIT8( KErrDescrNotSupported, "command not supported" );

// MACROS

// LOCAL CONSTANTS AND MACROS

// MODULE DATA STRUCTURES

// LOCAL FUNCTION PROTOTYPES

// FORWARD DECLARATIONS

// ============================ MEMBER FUNCTIONS ===============================

/**

* CFtpHandlerAO implementation

*/

CFtpHandlerAO::CFtpHandlerAO( MFtpObserverAO* anObserver ):

    CActive( EPriorityStandard ),

    iObserver( anObserver ),

    iCancelFileMan( EFalse )

    {

    CActiveScheduler::Add( this );

    }

CFtpHandlerAO::~CFtpHandlerAO()

    {

    }

void CFtpHandlerAO::Start()

    {

    SetActive();

    }

MFileManObserver::TControl CFtpHandlerAO::NotifyFileManStarted()

    {

    return iCancelFileMan?

                MFileManObserver::EAbort :

                MFileManObserver::EContinue;

    }

MFileManObserver::TControl CFtpHandlerAO::NotifyFileManOperation()

    {

    return iCancelFileMan?

                MFileManObserver::ECancel :

                MFileManObserver::EContinue;

    }

MFileManObserver::TControl CFtpHandlerAO::NotifyFileManEnded()

    {

    return iCancelFileMan?

                MFileManObserver::EAbort :

                MFileManObserver::EContinue;

    }

void CFtpHandlerAO::RunL()

    {

    iObserver->FtpComplete( iStatus.Int() );

    }

void CFtpHandlerAO::DoCancel()

    {

    iCancelFileMan = ETrue;

    }

CProcessLogonAO::CProcessLogonAO( MFtpObserverAO* anObserver ):

    CActive( EPriorityStandard ),

    iObserver( anObserver )

    {

    CActiveScheduler::Add( this );

    }

CProcessLogonAO::~CProcessLogonAO()

    {

    Cancel();

    iProcess.Close();

    }

void CProcessLogonAO::Start( const TDesC& aCmdLine )

    {

    TInt err = iProcess.Create( KHtiFileHlp, aCmdLine );

    if ( err == KErrNone )

        {

        iProcess.Logon( iStatus );

        //start HtiFileHlp.exe

        iProcess.Resume();

        }

    else

        {

        TRequestStatus* pS = &iStatus;

        User::RequestComplete( pS, err );

        }

    SetActive();

    }

void CProcessLogonAO::RunL()

    {

    iObserver->FtpComplete( iStatus.Int() );

    }

void CProcessLogonAO::DoCancel()

    {

    iProcess.LogonCancel( iStatus );

    }

// Create instance of concrete ECOM interface implementation

CHtiFtpServicePlugin* CHtiFtpServicePlugin::NewL()

    {

    CHtiFtpServicePlugin* self = new (ELeave) CHtiFtpServicePlugin;

    CleanupStack::PushL (self);

    self->ConstructL();

    CleanupStack::Pop();

    return self;

    }

// Constructor

CHtiFtpServicePlugin::CHtiFtpServicePlugin():

    iState( EIdle ),

    iSendBuffer( NULL ),

    iSendBufferDes( NULL, 0 ),

    iHandlerAO( NULL ),

    iProcessLogonAO( NULL )

    {

    }

CHtiFtpServicePlugin::~CHtiFtpServicePlugin()

    {

    HTI_LOG_FUNC_IN("~CHtiFtpServicePlugin");

    delete iSendBuffer;

    delete iHandlerAO;

    delete iProcessLogonAO;

    delete iFileMan;

    delete iBackupFake;

    iBackupFakeLib.Close();

    iFs.Close();

    HTI_LOG_FUNC_OUT("~CHtiFtpServicePlugin");

    }

// Second phase construction.

void CHtiFtpServicePlugin::ConstructL()

    {

    HTI_LOG_FUNC_IN("CHtiFtpServicePlugin::ConstructL");

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

    iFileMan = CFileMan::NewL( iFs );

    HTI_LOG_FUNC_OUT("CHtiFtpServicePlugin::ConstructL");

    }

void CHtiFtpServicePlugin::InitL()

    {

    if ( iDispatcher->GetFreeMemory() < KMinBufferSize )

        {

        User::Leave( KErrNoMemory );

        }

    // just guess

    iBufferSize = iDispatcher->GetFreeMemory()/10;

    iBufferSize = iBufferSize < KMinBufferSize ?

                    KMinBufferSize :

                    iBufferSize;

    //test

    iBufferSize = 3*KMinBufferSize;

    }

TInt CHtiFtpServicePlugin::SendControlMsg( TFtpCommand aCmd,

                                           const TDesC8& aMsg  )

    {

    TInt err = KErrNone;

    HBufC8* temp = NULL;

    TRAP( err, temp = HBufC8::NewL( 1 + aMsg.Size() ) );

    if ( err == KErrNone )

        {

        temp->Des().Append( aCmd );

        temp->Des().Append( aMsg );

        err = iDispatcher->DispatchOutgoingMessage( temp,

                        KFtpServiceUid,

                        EFalse,

                        EHtiPriorityControl );

        if ( err != KErrNone )

            {

            delete temp;

            }

        }

    return err;

    }

inline TInt CHtiFtpServicePlugin::SendErrorMsg( TInt anError,

                                                const TDesC8& aMsg )

    {

    SetBURFakeState( EFalse ); // errors ignored

    return iDispatcher->DispatchOutgoingErrorMessage( anError,

                                               aMsg,

                                               KFtpServiceUid);

    }

TBool CHtiFtpServicePlugin::IsBusy()

    {

    return !( iState == EIdle || iState == EStorWait || iState == ERetrWait );

    }

void CHtiFtpServicePlugin::ProcessMessageL(const TDesC8& aMessage,

                THtiMessagePriority aPriority)

    {

    HTI_LOG_FUNC_IN("CHtiFtpServicePlugin::ProcessMessage");

    if ( IsBusy() )

        {

        //should not happend - service busy,deny request

        User::Leave( KErrInUse );

        }

    switch ( aPriority )

        {

        case EHtiPriorityData:

            {

            HandleDataMessageL( aMessage );

            }

            break;

        case EHtiPriorityControl:

            {

            if ( iState == EStorWait || iState == ERetrWait )

                {

                HandleCancelL( aMessage );

                }

            else

                {

                HandleControlMessageL( aMessage );

                }

            }

            break;

        default:

            HTI_LOG_TEXT("Unknown priority");

        }

    HTI_LOG_FUNC_OUT("CHtiFtpServicePlugin::ProcessMessage");

    }

void CHtiFtpServicePlugin::HandleControlMessageL(const TDesC8& aMessage)

    {

    TInt err = KErrNone;

    if ( aMessage.Length() > 0 )

        {

        TBool unicode = aMessage[0]&KUnicodeMask;

        HTI_LOG_FORMAT("cmd %d", aMessage[0] );

        switch ( aMessage[0] )

            {

            case EFtpSTOR:

            case EFtpSTOR_u:

                {

                //receive file

                //get filesize

                if ( aMessage.Length() > 5 )

                    {

                    iFileSize = aMessage[1] +

                                ( aMessage[2] << 8  ) +

                                ( aMessage[3] << 16 ) +

                                ( aMessage[4] << 24 );

                    HTI_LOG_FORMAT( "Filesize %d", iFileSize );

                    //get fileName

                    if ( GetFileNameL( aMessage.Mid( 5 ),

                                      unicode ) )

                        {

                        HandleReceiveFileL();

                        }

                    }

                else

                    {

                    HTI_LOG_TEXT("no file size or file name");

                    User::LeaveIfError( SendErrorMsg( KErrArgument,

                                            KErrDescrInvalidStorArgument) );

                    }

                }

                break;

            case EFtpRETR:

            case EFtpRETR_u:

                {

                //send file

                //get fileName and size

                if ( GetFileNameL( aMessage.Mid(1),

                                   unicode ) )

                    {

                    HandleSendFileL();

                    }

                }

                break;

            case EFtpLIST:

            case EFtpLIST_u:

            case EFtpLISTSIZES:

            case EFtpLISTSIZES_u:

                {

                if ( GetDirectoryL( aMessage.Mid( 1 ),

                                   unicode ) )

                    {

                    HandleListL( unicode,

                        ( KEntryAttHidden| KEntryAttSystem|KEntryAttNormal ),

                        ( aMessage[0] == EFtpLISTSIZES ||

                          aMessage[0] == EFtpLISTSIZES_u ) );

                    }

                }

                break;

            case EFtpLISTDIR:

            case EFtpLISTDIR_u:

                {

                if ( GetDirectoryL( aMessage.Mid( 1 ),

                                   unicode ) )

                    {

                    HandleListL( unicode,

                                (KEntryAttMatchExclusive|

                                KEntryAttHidden|KEntryAttSystem|

                                KEntryAttDir), EFalse );

                    }

                }

                break;

            case EFtpMKD:

            case EFtpMKD_u:

                {

                if ( GetDirectoryL( aMessage.Mid( 1 ),

                                   unicode ) )

                    {

                    if ( IsFileTcb( iFileName ) )

                        {

                        HandleTcbMkdL( iFileName );

                        }

                    else

                        {

                        err = iFs.MkDirAll( iFileName );

                        if ( err == KErrNone || err == KErrAlreadyExists )

                            {

                            User::LeaveIfError( SendControlMsg( EFtpOK,

                                                                KNullDesC8) );

                            }

                        else

                            {

                            User::LeaveIfError( SendErrorMsg( err,

                                                        KErrDescrFailedMkDir ) );

                            }

                        }

                    }

                }

                break;

            case EFtpRMD:

            case EFtpRMD_u:

                {

                if ( GetDirectoryL( aMessage.Mid( 1 ),

                                   unicode ) )

                    {

                    if ( IsFileTcb( iFileName ) )

                        {

                        HandleTcbRmdL( iFileName );

                        }

                    else

                        {

                        delete iHandlerAO;

                        iHandlerAO = NULL;

                        iHandlerAO = new(ELeave) CFtpHandlerAO( this );

                        err = iFileMan->RmDir( iFileName, iHandlerAO->iStatus );

                        if ( err == KErrNone)

                            {

                            iState = ERmdBusy;

                            iHandlerAO->Start();

                            }

                        else

                            {

                            delete iHandlerAO;

                            iHandlerAO = NULL;

                            User::LeaveIfError( SendErrorMsg( err,

                                                        KErrDescrFailedRmDir ) );

                            }

                        }

                    }

                }

                break;

            case EFtpDELE:

            case EFtpDELE_u:

                {

                if ( GetFileNameL( aMessage.Mid( 1 ),

                                   unicode ) )

                    {

                    if ( IsFileTcb( iFileName ) )

                        {

                        HandleTcbDeleteL( iFileName );

                        }

                    else

                        {

                        HandleDeleteL( iFileName );

                        }

                    }

                }

                break;

            case EFtpCANCEL:

                {

                User::LeaveIfError( SendErrorMsg( KErrArgument,

                                                KErrDescrNoCancel ) );

                }

                break;

            case EFtpSETFORCE:

                {

                if ( aMessage.Length() != 2 )

                    {

                    User::LeaveIfError( SendErrorMsg( KErrArgument,

                                                KErrDescrInvalidForceArgs ) );

                    }

                else

                    {

                    if ( aMessage[1] )

                        {

                        // Setting forced operations on

                        err = iBackupFakeLib.Load( KHtiFtpBackupFakeDllName );

                        HTI_LOG_FORMAT( "BackupFake DLL load returned %d", err );

                        if ( err == KErrNone && iBackupFakeLib.Type()[1] ==

                                KHtiFtpBackupFakeInterfaceUid )

                            {

                            HTI_LOG_TEXT( "BackupFake DLL found" );

                            TLibraryFunction entry = iBackupFakeLib.Lookup( 1 );

                            if ( entry != NULL )

                                {

                                iBackupFake = ( CHtiFtpBackupFakeBase* ) entry();

                                TRAP( err, iBackupFake->ConstructL( &iFs ) );

                                }

                            }

                        if ( err == KErrNone )

                            {

                            SendControlMsg( EFtpOK, KNullDesC8 );

                            }

                        else

                            {

                            User::LeaveIfError( SendErrorMsg( KErrNotSupported,

                                KErrDescrNotSupported ) );

                            }

                        }

                    else

                        {

                        // Setting forced operations off

                        delete iBackupFake;

                        iBackupFake = NULL;

                        iBackupFakeLib.Close();

                        SendControlMsg( EFtpOK, KNullDesC8 );

                        }

                    }

                }

                break;

            case EFtpCHECKSUM:

            case EFtpCHECKSUM_u:

                {

                if ( aMessage.Length() < 3 )

                    {

                    User::LeaveIfError(

                        SendErrorMsg( KErrArgument,

                                      KErrDescrInvalidChecksumArgs ) );

                    }

                else if ( GetFileNameL( aMessage.Mid( 2 ), unicode ) )

                    {

                    HandleCheckSumCalcL( (TAlgorithm)aMessage[1], iFileName );

                    }