// All rights reserved.

// This component and the accompanying materials are made available

// under the terms of the License "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:

//

#include <e32base.h>

#include <f32file.h>

#include <e32property.h>

#include "mstypes.h"

#include "msctypes.h"

#include "usbmsshared.h"

#include "drivemanager.h"

#include "drivepublisher.h"

#include "tscsiserverreq.h"

#include "tscsiservercmds.h"

#include "mserverprotocol.h"

#include "mdevicetransport.h"

#include "testman.h"

#include "cscsiserverprotocol.h"

#include "debug.h"

TMediaWriteMan::TMediaWriteMan()

:   iActive(EFalse),

    iOffset(0),

    {

    }

void TMediaWriteMan::ReportHighSpeedDevice()

TInt64 TMediaWriteMan::Start(TUint32 aLba, TUint32 aLength, TUint32 aBlockSize)

    {

    iActive = ETrue;

    iOffset = static_cast<TInt64>(aLba) * aBlockSize;

    iBytesRemain = aLength * aBlockSize;

    return theEnd;

    }

TUint32 TMediaWriteMan::NextPacket()

    {

    iActive = ETrue;

    return (iBytesRemain < iMediaWriteSize) ? iBytesRemain : iMediaWriteSize;

    }

void TMediaWriteMan::Reset()

    {

    iActive = EFalse;

    iOffset = 0;

    }

void TMediaWriteMan::SetOffset(const TInt64& aOffset, TUint aLength)

    {

    iOffset = aOffset + aLength;

    iBytesRemain -= aLength;

    }

TUint32 TMediaWriteMan::GetPacketLength() const

    {

    // KMaxBufSize or the MediaWriteSize, whichever is smallest.

    return thisLength;

    }

/**

Creates the CScsiProtocol object.  Called during controller initialisation.

@param aDriveManager reference to the drive manager object

*/

CScsiServerProtocol* CScsiServerProtocol::NewL(CDriveManager& aDriveManager)

#ifdef MSDC_TESTMODE

CScsiServerProtocol* CScsiServerProtocol::NewL(CDriveManager& aDriveManager, TTestParser* aTestParser)

#endif

/**

c'tor

@param aDriveManager reference to the drive manager object

*/

CScsiServerProtocol::CScsiServerProtocol(CDriveManager& aDriveManager)

:   iDriveManager(aDriveManager)

#ifdef MSDC_TESTMODE

CScsiServerProtocol::CScsiServerProtocol(CDriveManager& aDriveManager, TTestParser* aTestParser)

:   iDriveManager(aDriveManager),

    iTestParser(aTestParser)

#endif

CScsiServerProtocol::~CScsiServerProtocol()

    iDataBuf.Close();

void CScsiServerProtocol::ConstructL()

/**

Associates the transport with the protocol. Called during initialisation of the controller.

@param aTransport pointer to the transport object

*/

void CScsiServerProtocol::RegisterTransport(MDeviceTransport* aTransport)

/**

Called by the Transport when it detects that the USB device is either running

at High Speed or is at least capable of HS operation. The Protocol can use this

information (for instance) to select the optimal write block size to use.

This function is preferably called before actual MS data transfer operation

starts, and usually only once.

*/

void CScsiServerProtocol::ReportHighSpeedDevice()

    iMediaWriteMan.ReportHighSpeedDevice();

void CScsiServerProtocol::SetParameters(const TMassStorageConfig& aConfig)

/**

Called by the transport layer when a packet is available for decoding.

If an error occurs, the sense code is updated and EFalse is returned.

@param aData

@return  ETrue if command was decoded and executed successfully

*/

TBool CScsiServerProtocol::DecodePacket(TPtrC8& aData, TUint8 aLun)

    TScsiServerReq* cdb = NULL;

    TRAPD(err, cdb = cdb->CreateL(static_cast<TScsiServerReq::TOperationCode>(aData[0]), aData));

    TBool decodeGood = EFalse;

    if (err == KErrNotSupported)

    else if (cdb->iNaca) // Check the CONTROL byte

    else if (cdb->iLink)

        iSenseInfo.SetSense(TSenseInfo::EIllegalRequest, TSenseInfo::EInvalidFieldInCdb);

    else

        {

        TScsiServerReq::TOperationCode operationCode = cdb->iOperationCode;

        if (aLun > iDriveManager.MaxLun())

            {

            __PRINT(_L("No drive available\n"));

            iSenseInfo.SetSense(TSenseInfo::EIllegalRequest, TSenseInfo::ELuNotSupported);

            }

        else

            {

            iLun = aLun;

            if (operationCode != TScsiServerReq::ERequestSense)

                {

                iSenseInfo.SetSense(TSenseInfo::ENoSense);

                }

            switch (operationCode)

                {

            case TScsiServerReq::ETestUnitReady:

                HandleUnitReady();

                break;

            case TScsiServerReq::ERequestSense:

                HandleRequestSense(*cdb);

                break;

            case TScsiServerReq::EInquiry:

                HandleInquiry(*cdb);

                break;

            case TScsiServerReq::EModeSense6:

                HandleModeSense6(*cdb);

                break;

            case TScsiServerReq::EStartStopUnit:

                HandleStartStopUnit(*cdb);

                break;

            case TScsiServerReq::EPreventMediaRemoval:

                HandlePreventMediaRemoval(*cdb);

                break;

            case TScsiServerReq::EReadCapacity10:

                HandleReadCapacity10(*cdb);

                break;

            case TScsiServerReq::ERead10:

                HandleRead10(*cdb);

                break;

            case TScsiServerReq::EWrite10:

                HandleWrite10(*cdb);

                break;

            case TScsiServerReq::EReadFormatCapacities:

                HandleReadFormatCapacities(*cdb);

                break;

            default:

                iSenseInfo.SetSense(TSenseInfo::EIllegalRequest, TSenseInfo::EInvalidCmdCode);

                break;

                }

            }

        __PRINT1(_L("DecodePacket result = %d"), iSenseInfo.SenseOk());

        }

    delete cdb;

    return decodeGood;

/**

Checks if drive ready

@param aLun Logic unit number

@return pointer to drive correspondent to LUN if drive mounted and ready, NULL otherwise

*/

CMassStorageDrive* CScsiServerProtocol::GetCheckDrive()

#ifdef MSDC_TESTMODE

    if (iTestParser && iTestParser->SenseError() != TTestParser::ETestSenseErrorNoSense)

        {

        switch (iTestParser->SenseError())

            {

            case TTestParser::ETestSenseErrorMediaNotPresent:

                __TESTMODEPRINT("Set SENSE ERROR(ENotReady, EMediaNotPresent)");

                iSenseInfo.SetSense(TSenseInfo::ENotReady,

                                    TSenseInfo::EMediaNotPresent);

                break;

            case TTestParser::ETestSenseErrorUnitAttention:

            default:

                __TESTMODEPRINT("Set SENSE ERROR(EUnitAttention, ENotReadyToReadyChange)");

                iSenseInfo.SetSense(TSenseInfo::EUnitAttention,

                                    TSenseInfo::ENotReadyToReadyChange);

                break;

            }

        iTestParser->ClrSenseError();

        return NULL;

        }

#endif

/**

Command Parser for the UNIT READY command (0x00)

@param aLun Logic unit number

@return ETrue if successful,

*/

TBool CScsiServerProtocol::HandleUnitReady()

/**

Command Parser for the REQUEST SENSE command (0x03)

@return ETrue if successful,

*/

TBool CScsiServerProtocol::HandleRequestSense(const TScsiServerReq& aRequest)

    const TScsiServerRequestSenseReq request = static_cast<const TScsiServerRequestSenseReq&>(aRequest);

    TScsiServerRequestSenseResp requestSense;

    requestSense.iAllocationLength = request.iAllocationLength;

    requestSense.SetResponseCode(TScsiServerRequestSenseResp::ECurrentErrors);

    requestSense.iSensePtr = &iSenseInfo;

    requestSense.Encode(iCommandBuf);

/**

Command Parser for the INQUIRY command (0x12)

@param aLun Logic unit number

@return ETrue if successful,

*/

TBool CScsiServerProtocol::HandleInquiry(const TScsiServerReq& aRequest)

    const TScsiServerInquiryReq request = static_cast<const TScsiServerInquiryReq&>(aRequest);

    TScsiServerInquiryResp inquiry(iConfig);

    inquiry.SetAllocationLength(request.iAllocationLength);

#ifdef MSDC_TESTMODE

    if (iTestParser && !iTestParser->Removable())

        {

        __TESTMODEPRINT("RMB Cleared");

        inquiry.SetNotRemovable();

        }

#endif

    inquiry.Encode(iCommandBuf);

    TUint length = inquiry.Length();

/**

 Command Parser for the START STOP UNIT command (0x1B)

 @param aData command data (started form position 1)

 @param aLun Logic unit number

 @return ETrue if successful, TFalse otherwise

 */

TBool CScsiServerProtocol::HandleStartStopUnit(const TScsiServerReq& aRequest)

            // rd/wr publisher

               mountState == CMassStorageDrive::EConnected)))

/**

Command Parser for the PREVENT/ALLOW MEDIA REMOVAL command (0x1E)

@param aData command data (started form position 1)