/* 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 "atmisccmdpluginconsts.h"
#include "cusdreadwriteimp.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) :
    CATCmdAsyncBase(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
    User::LeaveIfError(iUssdMessaging.Open(iPhone));
    
    iUSSDSender = CCUSSDSendMessageImpl::NewL(this, iUssdMessaging);
    iUSSDReader = CCUSSDReadMessageImpl::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::DoCancel()
    {
    TRACE_FUNC_ENTRY
    // This would never get called as the AO is never activated 
    TRACE_FUNC_EXIT
    }

void CCUSDCommandHandler::HandleCommand( const TDesC8& aCmd,
                                         RBuf8& /*aReply*/,
                                         TBool /*aReplyNeeded*/ )
    {
    TRACE_FUNC_ENTRY
    TAtCommandParser::TCommandHandlerType cmdHandlerType = iATCmdParser.CommandHandlerType();
    
    switch (cmdHandlerType)
        {
        case (TAtCommandParser::ECmdHandlerTypeTest):
            {
            iReply.Create(KDefaultCmdBufLength);
            iReply.Append(KCRLF);
            iReply.Append(KCUSDSupportedCmdsList);
            iCallback->CreateReplyAndComplete( EReplyTypeOther, iReply );
            break;
            }
        case (TAtCommandParser::ECmdHandlerTypeRead):
            {
            iReply.Create(KDefaultCmdBufLength);
            iReply.Append(KCRLF);
            iReply.Append(KAtCUSD);
            iReply.AppendNum(iDisplayRetMessage);
            iReply.Append(KCRLF);

            iCallback->CreateReplyAndComplete( EReplyTypeOther, iReply );
            break;
            }
        case (TAtCommandParser::ECmdHandlerTypeSet):
            {         
            if(ParseCUSDCmd(aCmd) == KErrNone)
               {
               if(iUSSDCommand.Length() == 0)
                   {
                   Trace(KDebugPrintD, "No USSD command: ", iUSSDCommand.Length());

                   // the USSD command is not set then this is a set                   
                   iReply.Create(KDefaultCmdBufLength);
                   iReply.Append(KCRLF);
                   iReply.Append(KAtCUSD);
                   iReply.AppendNum(iDisplayRetMessage);
                   iReply.Append(KCRLF);

                   iCallback->CreateReplyAndComplete( EReplyTypeOther, iReply );                                     
                   }
                else
                    {
                    if(iUSSDSender->HandleSendUSSDCmd(iUSSDCommand, iSendAttributes) == KErrNone)
                        {
                        iSendCommandSent = ETrue;
                        }
                        else
                        {
                        iCallback->CreateReplyAndComplete(EReplyTypeError);
                        }                 
                    }
                }
            else
                {
                iCallback->CreateReplyAndComplete(EReplyTypeError);
                }
                 
            break;
            }
        default:
            {
            iCallback->CreateReplyAndComplete(EReplyTypeError);
            break;
            }
        }
    TRACE_FUNC_EXIT
    }

void CCUSDCommandHandler::RunL()
    {
    TRACE_FUNC_ENTRY
    // This would never get called as the AO is never activated 
    TRACE_FUNC_EXIT
    }


void CCUSDCommandHandler::HandleCommandCancel()
    {
    TRACE_FUNC_ENTRY   
    iUSSDSender->Cancel();
    TRACE_FUNC_EXIT
    }

TInt CCUSDCommandHandler::ParseCUSDCmd(const TDesC8& aCmd)
    {
    TRACE_FUNC_ENTRY
    
    iReply.Create(KDefaultCmdBufLength);
    TBool tempRetMessage = EFalse;
    TInt ret = iATCmdParser.NextIntParam(tempRetMessage);

    if(ret == KErrNone)
        {
        if(tempRetMessage < 0 || tempRetMessage > 2)
            {
            Trace(KDebugPrintD, "Display mode invalid: ", tempRetMessage);
            TRACE_FUNC_EXIT
            return KErrGeneral;       
            }
        else
            {
            iDisplayRetMessage = tempRetMessage;
            }
        }
    
    // now parse the USSD command
    TPtrC8 command = iATCmdParser.NextTextParam(ret);

    if (ret == KErrNone)
        {
        if(iUSSDCommand.Create(command.Length()) != KErrNone)
           {
            TRACE_FUNC_EXIT
            return KErrGeneral;
           }
        
        if(Pack(command, iUSSDCommand) < 0)
           {
           Trace(KDebugPrintS, "Pack() returned error, command: ", &command);
           TRACE_FUNC_EXIT
           return KErrGeneral;
           }             
        }
   
    TInt tempDcs = -1;    
    ret = iATCmdParser.NextIntParam(tempDcs); 

    if(ret == KErrNone)
        {
        if(!IsDcsValid(tempDcs))
            {
            Trace(KDebugPrintD, "Invalid coding standard: ", tempDcs);
            TRACE_FUNC_EXIT
            return KErrGeneral;
            }
        else
            {
            iSendAttributes.iDcs = tempDcs;
            }
        } 
    TRACE_FUNC_EXIT
    return KErrNone;
    }

void CCUSDCommandHandler::HandleSendMessageComplete(TInt aResult)
    {
    TRACE_FUNC_ENTRY
    if(aResult == KErrNone)
        {
        iReply.Create(KDefaultCmdBufLength);
        iReply.Append(KCRLF);
        iReply.Append(KAtCUSD);
        iReply.Append(KCRLF);
        iCallback->CreateReplyAndComplete( EReplyTypeOther, iReply );          
        }
    else
        {
        iCallback->CreateReplyAndComplete(EReplyTypeError);
        }
    
    iSendCommandSent = EFalse;
    iUSSDCommand.Close();
    TRACE_FUNC_EXIT
    }

void CCUSDCommandHandler::HandleReadMessageComplete(TInt aResult)
    {
    TRACE_FUNC_ENTRY
    if(iDisplayRetMessage)
        {
        FormatUSSDReceivedData();
        
        if(iSendCommandSent)
            {
            iCallback->CreateReplyAndComplete( EReplyTypeOther, iReply );
            }
         else
            {       
            iCallback->HandleUnsolicitedRequest(iReply);
            }
        }   
    // re-issue the read request
    iUSSDReader->ReadUSSDMessage(iReceivedData, iReceiveAttributes);
    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);
    if ( elementSizeInBits == 8 ) // 8 bit data
        {
        // Straight copy here
        aOut.Append( aIn );
        }
        else if ( elementSizeInBits == 7 ) // Need packing to 7-bit
            {
            // Get raw pointers and do packing
            TUint8* outPtr = ( TUint8* )aOut.Ptr() + aOut.Length();
            const TUint8* inPtr = aIn.Ptr();
    
            outPtr[0] = 0;
            for ( TInt i = 0; i < aIn.Length() ; ++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,...
             
                if ( i == aIn.Length() - 1 ) // if this is the last time
                    {
                    if ( ( to == 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 );
            }

    // 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;
             }
         }
     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.Create(KDefaultCmdBufLength);

    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
    }