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

 *

 */

#include "cusdcommandhandler.h"

#include <exterror.h> 

#include "ussdreadmessageimpl.h"

#include "ussdsendmessageimpl.h"

#include "atmisccmdpluginconsts.h"

#include "debug.h"

const TUint8 KPhCltUssdCarriageReturn = 0x0d;

const TUint8 KPhCltUssdDcsNotSet  = 0x00; // not set

const TUint8 KQuarterShift( 2 );

const TUint8 KHighNibbleShift( 4 );

const TUint8 KDcsCharacterSet7Bit( 0x00 );

const TUint8 KDcsCharacterSet8Bit( 0x01 );

const TUint8 KDcsCharacterSet16Bit( 0x02 );

const TUint8 KDcsCharacterSet7Bit2( 0x00 );

const TUint8 KDcsCharacterSet16Bit2( 0x01 );

CCUSDCommandHandler* CCUSDCommandHandler::NewL(MATMiscCmdPlugin* aCallback, TAtCommandParser& aATCmdParser, RMobilePhone& aPhone)

    {

    TRACE_FUNC_ENTRY

    CCUSDCommandHandler* self = new (ELeave) CCUSDCommandHandler(aCallback, aATCmdParser, aPhone);

    CleanupStack::PushL(self);

    self->ConstructL();

    CleanupStack::Pop(self);

    TRACE_FUNC_EXIT

    return self;

    }

CCUSDCommandHandler::CCUSDCommandHandler(MATMiscCmdPlugin* aCallback, TAtCommandParser& aATCmdParser, RMobilePhone& aPhone) :

    CATCmdSyncBase(aCallback, aATCmdParser, aPhone)

    {

    TRACE_FUNC_ENTRY

    iSendAttributes.iDcs = KPhCltUssdDcsNotSet;  

    iSendAttributes.iType = RMobileUssdMessaging::EUssdMORequest;

    iSendAttributes.iFlags = RMobileUssdMessaging::KUssdMessageType | RMobileUssdMessaging::KUssdMessageDcs | RMobileUssdMessaging::KUssdDataFormat;

    TRACE_FUNC_EXIT

    }

void CCUSDCommandHandler::ConstructL()

    {    

    TRACE_FUNC_ENTRY

    iReply.CreateL(KDefaultCmdBufLength);

    User::LeaveIfError(iUssdMessaging.Open(iPhone));

    iUSSDSender = CUSSDSendMessageImpl::NewL(this, iUssdMessaging);

    iUSSDReader = CUSSDReadMessageImpl::NewL(this, iUssdMessaging);

    //start a read session for network based ussd requests

    User::LeaveIfError(iUSSDReader->ReadUSSDMessage(iReceivedData, iReceiveAttributes));

    TRACE_FUNC_EXIT

    }

CCUSDCommandHandler::~CCUSDCommandHandler()

    {

    TRACE_FUNC_ENTRY

    delete iUSSDSender;

    delete iUSSDReader;

    iReply.Close();

    iUssdMessaging.Close();

    TRACE_FUNC_EXIT

    }

void CCUSDCommandHandler::HandleCommand( const TDesC8& /*aCmd*/,

                                         RBuf8& /*aReply*/,

                                         TBool /*aReplyNeeded*/ )

    {

    TRACE_FUNC_ENTRY

    TAtCommandParser::TCommandHandlerType cmdHandlerType = iATCmdParser.CommandHandlerType();

    iReply.Zero();

    switch (cmdHandlerType)

        {

        case (TAtCommandParser::ECmdHandlerTypeTest):

            {

            iReply.Append(KCRLF);

            iReply.Append(KCUSDSupportedCmdsList);

            iCallback->CreateReplyAndComplete( EReplyTypeOther, iReply );

            break;

            }

        case (TAtCommandParser::ECmdHandlerTypeRead):

            {

            iReply.Append(KCRLF);

            iReply.Append(KAtCUSD);

            iReply.AppendNum(iDisplayRetMessage);

            iReply.Append(KCRLF);

            iCallback->CreateReplyAndComplete( EReplyTypeOther, iReply );

            break;

            }

        case (TAtCommandParser::ECmdHandlerTypeSet):

            {         

            if(ParseCUSDCmd() == KErrNone)

               {

               if(iUSSDCommand.Length() == 0)

                   {

                   Trace(KDebugPrintD, "No USSD command: ", iUSSDCommand.Length());

                   // the USSD command is not set then this is a set                   

                   iReply.Append( KOKCRLF );

                   iCallback->CreateReplyAndComplete( EReplyTypeOther, iReply);

                   }

                else

                    {

                    if(iUSSDSender->SendUSSDCmd(iUSSDCommand, iSendAttributes) == KErrNone)

                        {

                        iSendCommandSent = ETrue;

                        }

                    else

                        {

                        iCallback->CreateReplyAndComplete(EReplyTypeError);

                        }                 

                    }

                }

            else

                {

                iCallback->CreateReplyAndComplete(EReplyTypeError);

                }

            break;

            }

        default:

            {

            iCallback->CreateReplyAndComplete(EReplyTypeError);

            break;

            }

        }

    TRACE_FUNC_EXIT

    }

void CCUSDCommandHandler::HandleCommandCancel()

    {

    TRACE_FUNC_ENTRY   

    iUSSDSender->Cancel();

    iSendCommandSent = EFalse;

    TRACE_FUNC_EXIT

    }

TInt CCUSDCommandHandler::ParseCUSDCmd()

    {

    TRACE_FUNC_ENTRY

    TInt tempDispValue;

    TInt ret = iATCmdParser.NextIntParam(tempDispValue);

    if (ret == KErrNone && tempDispValue >= 0 && tempDispValue < 3)

        {

        Trace(KDebugPrintD, "tempDispValue: ", tempDispValue);

        iDisplayRetMessage = tempDispValue;

        }    

    else if (ret != KErrNotFound)

        {

        Trace(KDebugPrintD, "First NextIntParam failed: ", ret);

        TRACE_FUNC_EXIT

        return KErrGeneral;       

        }

    // now parse the USSD command

    TPtrC8 command;

    ret = iATCmdParser.NextTextParam(command);

    if (ret == KErrNone )

        {

        if(iUSSDCommand.Create(command.Length()) != KErrNone)

            {

            Trace(KDebugPrintD, "iUSSDCommand.Create() failed ", ret);

            TRACE_FUNC_EXIT

            return KErrGeneral;

            }

        Trace(KDebugPrintS, "command: ", &command);

        if(Pack(command, iUSSDCommand) < 0)

            {

            Trace(KDebugPrintS, "Pack() returned error, command: ", &command);

            TRACE_FUNC_EXIT

            return KErrGeneral;

            }    

        }

    else if (ret != KErrNotFound)

        {

        TRACE_FUNC_EXIT

        return KErrGeneral;       

        }

    TInt dcs = 0; // 3GPP TS 23.038 [25] Cell Broadcast Data Coding Scheme in integer format (default 0)

    ret = iATCmdParser.NextIntParam(dcs); 

    Trace(KDebugPrintD, "ret: ", ret);

    Trace(KDebugPrintD, "IsDcsValid(dcs): ", IsDcsValid(dcs));

    Trace(KDebugPrintD, "iATCmdParser.NextParam().Length(): ", iATCmdParser.NextParam().Length());

    if(ret != KErrNone && ret != KErrNotFound || !IsDcsValid(dcs) || 

            iATCmdParser.NextParam().Length() != 0)

        {

        Trace(KDebugPrintD, "Second NextIntParam failed: ", ret);

        TRACE_FUNC_EXIT

        return KErrGeneral;       

        }

    else

        {

        // dcs will have the default value if no parameter is given in the command line

        Trace(KDebugPrintD, "dcs ", dcs);   

        iSendAttributes.iDcs = dcs;

        }

    TRACE_FUNC_EXIT

    return KErrNone;

    }

void CCUSDCommandHandler::HandleSendMessageComplete(TInt aError)

    {

    TRACE_FUNC_ENTRY

    if (aError != KErrNone)

        {

        // Only respond from a Send USSD request if there has been an error

        iCallback->CreateReplyAndComplete(EReplyTypeError);

        }

    else if (iDisplayRetMessage == 0)

        {

        // if display mode is 0 then no response apart from OK should be displayed

        iCallback->CreateReplyAndComplete( EReplyTypeOther, KOKCRLF );

        }

    iUSSDCommand.Close();

    TRACE_FUNC_EXIT

    }

void CCUSDCommandHandler::HandleReadMessageComplete(TInt aError)

    {

    TRACE_FUNC_ENTRY

    if(aError == KErrNone)

        {

        if(iDisplayRetMessage)

            {

            FormatUSSDReceivedData();

            if(iSendCommandSent)

                {

                iCallback->CreateReplyAndComplete( EReplyTypeOther, iReply );

                }

             else

                {       

                iCallback->HandleUnsolicitedRequest(iReply);

                }

            }

        // as we have received the data initiated by the send USSD request we set

        // this parameter to EFalse

        iSendCommandSent = EFalse;

        // re-issue the read request

        iUSSDReader->ReadUSSDMessage(iReceivedData, iReceiveAttributes);

        }

    else

        {

        iCallback->CreateReplyAndComplete(EReplyTypeError);

        }

    TRACE_FUNC_EXIT

    }

TInt CCUSDCommandHandler::Pack(const TDesC8& aIn, TDes8& aOut)

    {

    TRACE_FUNC_ENTRY

    TInt startBit = 0; 

    // Ensure we've got the right length

    TInt packedOctetsRequired = PackedOctetsRequired( aIn.Length() );

    if (packedOctetsRequired > ( aOut.MaxLength() - aOut.Length() ) )

        {

        Trace(KDebugPrintD, "packedOctetsRequired: ", packedOctetsRequired);

        TRACE_FUNC_EXIT

        return KErrOverflow;

        }

    // Do the conversion

    TInt elementSizeInBits = ElementSizeInBits(TSmsDataCodingScheme::ESmsAlphabet7Bit);

    Trace(KDebugPrintD, "elementSizeInBits = : ", elementSizeInBits);

    if (elementSizeInBits == 8) // 8 bit data

        {

        // Straight copy here

        aOut.Append( aIn );

        }

    else if (elementSizeInBits == 7) // Need packing to 7-bit

        {

        Trace(KDebugPrintS, "aOut: ", &aOut);

        Trace(KDebugPrintS, "aIn: ", &aIn);

        // Get raw pointers and do packing

        TUint8* outPtr = ( TUint8* )aOut.Ptr() + aOut.Length();

        const TUint8* inPtr = aIn.Ptr();

        TInt inLength = aIn.Length();

        outPtr[0] = 0;

        Trace(KDebugPrintD, "inLength = : ", inLength);

        for (TInt i = 0; i < inLength; ++i)

           {

           TUint8 to = inPtr[i];

           *outPtr |= ( to << startBit );

           if (startBit)

               {

               ++outPtr;

               *outPtr = ( TUint8 ) ( to >> ( 8 - startBit ) );

               }

           startBit = (startBit + 7) % 8; // roll 0,1,2,3,4,5,6,7,0,1,2,...

           }

        Trace(KDebugPrintD, "startBit = : ", startBit);

        if ((inPtr[inLength - 1] == KPhCltUssdCarriageReturn && startBit == 0) || startBit == 1 )

            {

            //We have to add one CR more

            *outPtr |= ( KPhCltUssdCarriageReturn << startBit );

            if (startBit)

                {

                ++outPtr;

                *outPtr = (TUint8) (KPhCltUssdCarriageReturn >> (8 - startBit));

                }

            else

                {

                ++packedOctetsRequired;

                }

            startBit = (startBit + 7)%8;

            }

        // Increment the length for the packed data

        aOut.SetLength(aOut.Length() + packedOctetsRequired);

        Trace(KDebugPrintD, "aOut.Length() = : ", aOut.Length());

        }

    // Return number of bytes used

    TRACE_FUNC_EXIT

    return packedOctetsRequired;

    }

 TInt CCUSDCommandHandler::PackedOctetsRequired(TInt aNumUDUnits) const

    {

    TRACE_FUNC_ENTRY

    TInt startBit = 0; 

    TInt octetsRequired = 0;

    TInt elementSizeInBits = ElementSizeInBits(TSmsDataCodingScheme::ESmsAlphabet7Bit);

    if (elementSizeInBits == 8)

         {

        octetsRequired=aNumUDUnits;

         }

    else

        {

        octetsRequired =  (startBit + aNumUDUnits*elementSizeInBits + 7 )/8;    // Rounds up

        }

    TRACE_FUNC_EXIT

    return octetsRequired;

    }

 TBool CCUSDCommandHandler::IsDcsValid( const TUint8 aDcs ) const

     {

     TRACE_FUNC_ENTRY

     TBool isDcsValid(EFalse);

                                                                //      76543210

     TUint8 codingGroup  = ( aDcs & 0xF0 ) >> KHighNibbleShift; // bits XXXX____

     TUint8 characterSet = ( aDcs & 0x0C ) >> KQuarterShift;    // bits ____XX__

     TUint8 lowQuartet   = ( aDcs & 0x0F );                     // bits ____XXXX

     switch (codingGroup)

         {

         case 0x00:

         case 0x02:

         case 0x03:

         case 0x0F:

             {

             isDcsValid = ETrue;

             break;

             }

         case 0x01:

             {

             if ( ( KDcsCharacterSet7Bit2 == lowQuartet ) ||

                  ( KDcsCharacterSet16Bit2 == lowQuartet ) )

                 {

                 isDcsValid = ETrue;

                 }

             break;

             }

         case 0x04:

         case 0x05:

         case 0x06:

         case 0x07:

         case 0x09:

             {

             if ( ( KDcsCharacterSet7Bit == characterSet ) ||

                  ( KDcsCharacterSet8Bit == characterSet ) ||

                  ( KDcsCharacterSet16Bit == characterSet ) )

                 {

                 isDcsValid = ETrue;

                 }

             break;

             }

         default:

             {

             break;

             }

         }

     TRACE_FUNC_EXIT

     return isDcsValid;

     }

TInt CCUSDCommandHandler::ElementSizeInBits(TSmsDataCodingScheme::TSmsAlphabet aAlphabet) const

    {

    TRACE_FUNC_ENTRY

    TInt ret;

    switch (aAlphabet)

       {

       case TSmsDataCodingScheme::ESmsAlphabet7Bit:

           {

           ret = 7;

           break;

           }

       case TSmsDataCodingScheme::ESmsAlphabet8Bit:

       case TSmsDataCodingScheme::ESmsAlphabetUCS2:

           {

           ret = 8;

           break;

           }

       default:

           {

           ret = KErrGsmSMSDataCodingSchemeNotSupported;

           }

        }

    TRACE_FUNC_EXIT

    return ret;

    }

void CCUSDCommandHandler::FormatUSSDReceivedData()

    {

    TRACE_FUNC_ENTRY

    iReply.Zero();

    iReply.Append(KCRLF);

    iReply.Append(KAtCUSD);

    iReply.AppendNum(iDisplayRetMessage);

    iReply.Append(',');

    iReply.Append('"');

    TInt length = iReceivedData.Length();

    for(TInt i = 0; i < length; ++i)

       {       

       iReply.AppendNum(iReceivedData[i], EHex);

       }

    iReply.Append('"');

    iReply.Append(',');

    iReply.AppendNum(iReceiveAttributes.iDcs);

    iReply.Append(KCRLF);

    TRACE_FUNC_EXIT

    }