/*

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

*     Zhiqiang Yang <zhiqiang.yang@nokia.com>

* 

* Description:

*     The source file of the file manager utilities on Symbian

*/

#include "fmutils.h"

#include "fmcommon.h"

#include "fms60utils.h"

#include <apgcli.h>

#include <sysutil.h>

#include <f32file.h>

#include <coemain.h>

#include <pathinfo.h>

#include <driveinfo.h>

#include <e32property.h>

#include <coreapplicationuisdomainpskeys.h>

#include <QDir>

#include <QFile>

#include <QRegExp>

#include <QStringList>

#include <QFileInfoList>

#include <xqappmgr.h>

#include <XQConversions>

#include <xqaiwrequest.h>

#include <hbglobal.h>

#include <hbdirectorynamelocalizer.h>

#define BURCONFIGFILE  "z:/private/2002BCC0/burconfig.xml"

/*!

    query drive info and status for \a driveName

    return \a FmDriverInfo

*/

FmDriverInfo FmUtils::queryDriverInfo( const QString &driveName )

{

    if( driveName.isEmpty() ) {

        return FmDriverInfo( 0, 0, driveName, QString(), FmDriverInfo::EDriveNotPresent );

    }

    TVolumeInfo volumeInfo;

    TInt drive = 0;

    drive = driveName[0].toUpper().toAscii() - 'A' + EDriveA;

    quint32 state( 0 );

    int volumeInfoErr( KErrNone );

    int driveInfoErr( KErrNone );

    int errorCode( KErrNone );

    volumeInfoErr = fs.Volume( volumeInfo, drive );

    errorCode = volumeInfoErr;

    QString volumeName( (QChar*)( volumeInfo.iName.Des().Ptr() ), volumeInfo.iName.Length() );

    TDriveInfo driveInfo;

    if( volumeInfoErr == KErrNone ) {

        driveInfo = volumeInfo.iDrive;

    } else {

        driveInfoErr = fs.Drive( driveInfo, drive );

        if( driveInfoErr != KErrNone ) {

            errorCode = driveInfoErr;

        }

    }

    if( volumeInfoErr == KErrNone || driveInfoErr == KErrNone ) {

        //TDriveInfo driveInfo = volumeInfo.iDrive;

        quint32 drvStatus( 0 );

        int err = DriveInfo::GetDriveStatus( fs, drive, drvStatus );

        if( err == KErrNone ) {

            if ( ( drvStatus & DriveInfo::EDriveInternal ) &&

                 ( drvStatus & DriveInfo::EDriveExternallyMountable ) ){

                // Handle mass storage bits here

                state |= FmDriverInfo::EDriveMassStorage | FmDriverInfo::EDriveRemovable;

            }

            if ( drvStatus & DriveInfo::EDriveUsbMemory )

                {

                state |= FmDriverInfo::EDriveUsbMemory;

                }

            if ( drvStatus & DriveInfo::EDriveRemote )

                {

                state |= FmDriverInfo::EDriveRemote;

                }

            if ( drvStatus & DriveInfo::EDriveRom ){

                state |= FmDriverInfo::EDriveRom;  

            }

            if ( drvStatus & DriveInfo::EDriveRam ){

                state |= FmDriverInfo::EDriveRam;  

            }

            if ( driveInfo.iMediaAtt & KMediaAttFormattable ){

                state |= FmDriverInfo::EDriveFormattable;

            }

            if ( driveInfo.iMediaAtt & KMediaAttWriteProtected ){

                state |= FmDriverInfo::EDriveWriteProtected;

            }

            if ( driveInfo.iMediaAtt & KMediaAttHasPassword ){

                state |= FmDriverInfo::EDrivePasswordProtected;

            }    

            if ( driveInfo.iMediaAtt & KMediaAttLocked ){

                state |= FmDriverInfo::EDriveLocked;

            }

            if ( driveInfo.iDriveAtt & KDriveAttRemovable ){

                state |= FmDriverInfo::EDriveRemovable;

                if ( drvStatus & DriveInfo::EDriveSwEjectable ){

                    state |= FmDriverInfo::EDriveEjectable;

                }

            }

            if( driveInfo.iType == EMediaNotPresent ){

                state |= FmDriverInfo::EDriveNotPresent;    

            }

        }

        // If memory card is not ready but type is present,

        // then check if it is reserved.

        if( err == KErrNone && volumeInfoErr == KErrNotReady &&

            driveInfo.iType != EMediaNotPresent )

            {

            // Check USB file transfer state

            TInt prop( ECoreAppUIsUSBFileTransferUninitialized );

            RProperty::Get(

                KPSUidCoreApplicationUIs,

                KCoreAppUIsUSBFileTransfer, prop );

            if ( prop == ECoreAppUIsUSBFileTransferActive )

                {

                errorCode = KErrInUse; // Reserved for file transfer

                }

            }

        if( err!= KErrNone )

            {

            errorCode = err;

            }

    }

    // handle error code

    // volumeInfoErr will occur while drive is lock,corrupted...

    // driveInfoErr can not be promoted for locked, corrupted drive.

    // so we can not use driveInfoErr to justify EDriveAvailable

    switch( errorCode )

    {

    case KErrNone:

        // this drive could be used as it is not be locked, or corrupt.

        state |= FmDriverInfo::EDriveAvailable; 

        break;

    case KErrLocked:

        state |= FmDriverInfo::EDriveLocked;

        break;

    case KErrCorrupt:

        state |= FmDriverInfo::EDriveCorrupted;

        break;

    case KErrInUse:

        state |= FmDriverInfo::EDriveInUse;

        break;

    default: // other errors

        state |= FmDriverInfo::EDriveNotPresent;

        break;

    }

    QString logString ( "FmUtils::queryDriverInfo_" + driveName + 

            "_volumeInfoErr:" + QString::number( volumeInfoErr ) +

            "_driveInfoErr:" + QString::number( driveInfoErr ) +

            "_errorCode:" + QString::number( errorCode ) + 

            "_driveSatus:" + QString::number( state ) );

    FM_LOG( logString );

    return FmDriverInfo( volumeInfo.iSize, volumeInfo.iFree, driveName, volumeName, state );

}

/*!

    remove drive password for \a driveName

    \a Pwd is original password.

*/

int FmUtils::removeDrivePwd( const QString &driveName,  const QString &Pwd )

{

    if( driveName.isEmpty() || Pwd.length() > FmMaxLengthofDrivePassword ) {

        return FmErrWrongParam;

    }

    QString logString = "Drive name:" + driveName;

    FM_LOG( logString );

    logString = "Password:" + Pwd;

    FM_LOG( logString );

    CCoeEnv *env = CCoeEnv::Static();

	RFs& fs = env->FsSession();

    TInt drive = 0;

    drive = driveName[0].toUpper().toAscii() - 'A' + EDriveA;

    HBufC* password16 = XQConversions::qStringToS60Desc( Pwd );

    TMediaPassword password;   

    TPtr ptrPassword16( password16->Des() );  

    FmS60Utils::ConvertCharsToPwd( ptrPassword16, password );

    int err( fs.ClearPassword( drive, password ) );

    logString = "Drive:" + QString::number( drive );

    FM_LOG( logString );

    logString = "Clear password error:" + QString::number( err );

    FM_LOG( logString );

    delete password16;

    if( err == KErrNone ){

        return FmErrNone;   

    }

    else if( err == KErrAccessDenied ){

        return FmErrAccessDenied;

    }

    else{

        return FmErrUnKnown;

    }

}

/*!

    Unlock drive \a driveName with provided password \a Pwd

*/

int FmUtils::unlockDrive( const QString &driveName,  const QString &Pwd )

{

    if( driveName.isEmpty() || Pwd.length() > FmMaxLengthofDrivePassword ) {

        return FmErrWrongParam;

    }

    QString logString = "Drive name:" + driveName;

    FM_LOG( logString );

    logString = "Password:" + Pwd;

    FM_LOG( logString );

    CCoeEnv *env = CCoeEnv::Static();

	RFs& fs = env->FsSession();

    TInt drive = 0;

	drive = driveName[0].toUpper().toAscii() - 'A' + EDriveA;

    HBufC* password16 = XQConversions::qStringToS60Desc( Pwd );

    TMediaPassword password;   

    TPtr ptrPassword16( password16->Des() );  

    FmS60Utils::ConvertCharsToPwd( ptrPassword16, password );

    int err( fs.UnlockDrive( drive, password, ETrue) );

    logString = "Drive:" + QString::number( drive );

    FM_LOG( logString );

    logString = "Unlock drive error:" + QString::number( err );

    FM_LOG( logString );

    delete password16;

    if( err == KErrNone ){

        return FmErrNone;   

    }

    else if( err == KErrAccessDenied ){

        return FmErrAccessDenied;

    }

    else if( err == KErrAlreadyExists ){

        return FmErrAlreadyExists;

    }

    else if( err == KErrNotSupported ){

        return FmErrNotSupported;

    }

    else{

        return FmErrUnKnown;

    }

}

/*!

    Check if \a pwd is the right password for drive \a driveName

*/

int FmUtils::checkDrivePwd( const QString &driveName, const QString &pwd )

{

    if( driveName.isEmpty() || pwd.length() > FmMaxLengthofDrivePassword ) {

        return FmErrWrongParam;

    }

    QString logString = "checkDrivePwd Drive name:" + driveName;

    logString += " password:" + pwd;

    FM_LOG( logString );

    return setDrivePwd( driveName, pwd, pwd );

}

/*!

    Set new password \a newPwd for drive \a driveName. \a oldPwd is old password

*/

int FmUtils::setDrivePwd( const QString &driveName, const QString &oldPwd, const QString &newPwd)

{

    if( driveName.isEmpty() || 

        oldPwd.length() > FmMaxLengthofDrivePassword || newPwd.length() > FmMaxLengthofDrivePassword  ) {

        return FmErrWrongParam;

    }

    QString logString = "setDrivePwd Drive name:" + driveName ;

    logString += " Old password:" + oldPwd;

    logString += " New password:" + newPwd;

    FM_LOG( logString );

    CCoeEnv *env = CCoeEnv::Static();

	RFs& fs = env->FsSession();

    TInt drive = 0;

	drive = driveName[0].toUpper().toAscii() - 'A' + EDriveA;

    HBufC* newPassword16 = XQConversions::qStringToS60Desc( newPwd);

    HBufC* oldPassword16 = XQConversions::qStringToS60Desc( oldPwd );

    TMediaPassword oldPassword;

    TMediaPassword newPassword;

    TPtr ptrNewPassword16( newPassword16->Des() );

    TPtr ptrOldPassword16( oldPassword16->Des() );

    FmS60Utils::ConvertCharsToPwd( ptrNewPassword16, newPassword );

    FmS60Utils::ConvertCharsToPwd( ptrOldPassword16, oldPassword );

    int err( fs.LockDrive( drive, oldPassword, newPassword, ETrue ) );

    logString = "Drive:" + QString::number( drive );

    FM_LOG( logString );

    logString = "Password set error:" + QString::number( err );

    FM_LOG( logString );

    delete newPassword16;

    delete oldPassword16;

    if( err == KErrNone ){

        return FmErrNone;   

    }

    else if( err == KErrNotSupported ){

        return FmErrNotSupported;

    }

    else{

        return FmErrUnKnown;

    }

}

/*!

    Set \a pwd as empty password

*/

void FmUtils::emptyPwd( QString &pwd )

{

    TBuf< FmMaxLengthofDrivePassword > nullPwd;

    nullPwd.FillZ( nullPwd.MaxLength() );

    nullPwd.Zero();

    pwd = XQConversions::s60DescToQString( nullPwd );

}

/*!

    Set drive volume for drive \a driveName

*/

int FmUtils::renameDrive( const QString &driveName, const QString &newVolumeName)

{

    if( driveName.isEmpty() ) {

        return FmErrWrongParam;

    }

    foreach( const QChar &ch, newVolumeName )

    {

        bool a = ch.isSpace();

        bool b = ch.isLetterOrNumber();

        // If not alphadigit or space, return error

        if( !ch.isLetterOrNumber() && !ch.isSpace() )

        {

            return FmErrBadName;

        }   

    }

    TInt drive = 0;

	drive = driveName[0].toUpper().toAscii() - 'A' + EDriveA;

    TPtr newName ( ( XQConversions::qStringToS60Desc( newVolumeName ) )->Des() );

    int err( fs.SetVolumeLabel( newName, drive ));

    QString logString = "Rename error:" + QString::number( err );

    FM_LOG( logString );

    if( err == KErrNone ){

        return FmErrNone;   

    }

    else if( err == KErrNotReady ){

        return FmErrNotReady;

    }

    else{

        return FmErrUnKnown;

    }

}

/*!

    Eject drive \a driveName

*/

int FmUtils::ejectDrive( const QString &driveName )

{

    if( driveName.isEmpty() ) {

        return FmErrWrongParam;

    }

    QString logString = "FmUtils::ejectDrive start";

    FM_LOG( logString );

    TInt drive = 0;

	drive = driveName[0].toUpper().toAscii() - 'A' + EDriveA;

    const int KDriveShift = 16;

    // Let SysAp handle eject

    RProperty::Set(

        KPSUidCoreApplicationUIs,

        KCoreAppUIsMmcRemovedWithoutEject,

        ECoreAppUIsEjectCommandUsedToDrive | ( drive << KDriveShift )

        );

    return FmErrNone;

}

/*!

    Check if drive \a driveName is accessable for user

*/

bool FmUtils::checkDriveAccessFilter( const QString &driveName )

{

    if( driveName.isEmpty() ) {

        return false;

    }

    FmDriverInfo driveInfo = queryDriverInfo( driveName );

    if( ( driveInfo.driveState()& FmDriverInfo::EDriveRam ) ||

        ( driveInfo.driveState()& FmDriverInfo::EDriveRom ) ) {

        return false;

    }

    return true;

}

/*!

    This function should be called to adjust path if user goto a drive.

    data folder will be append to C:\ becuase user could only view C:\data instead C:\ 

*/

QString FmUtils::checkDriveToFolderFilter( const QString &path )

{

    QString checkedPath = fillPathWithSplash( path );

    if( checkedPath.compare( Drive_C, Qt::CaseInsensitive ) == 0 ) {

        checkedPath += QString( "data" ) + QDir::separator();

        return checkedPath;

    }

    return path;

}

/*!

    This function should be called to adjust path if user back to up level path.

    If user is at C:\data then path level should be returned as C:\

    Becuase C:\data is root path for C drive 

*/

QString FmUtils::checkFolderToDriveFilter( const QString &path )

{

    QString logString;

    logString = QString( "checkFolderToDriveFilter: " ) + path;

    FM_LOG( logString );

    QString checkedPath = fillPathWithSplash( path );

    logString = QString( "checkFolderToDriveFilter_fillPathWithSplash: " ) + checkedPath;

    FM_LOG( logString );

    if( checkedPath.compare( Folder_C_Data, Qt::CaseInsensitive ) == 0 ) {

        FM_LOG( QString( " change from c:/data/ to C:/" ) );

        return Drive_C;

    }

    return path;