// Copyright (c) 1997-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:

//

#include "gsmubuf.h"

#include "gsmuset.h"

#include "Gsmuelem.h"

#include "Gsmumsg.h"

#include "gsmusar.h"

#include "gsmupdu.h"

#include "WAPDGRM.H"

#include "ws_main.h"

#include "smsstackutils.h"

//

// For incoming short message

//

CWapDatagram* CWapDatagram::NewL(const CSmsMessage& aSms)

    {

    LOGWAPPROT2("CWapDatagram::NewL(): aSms=0x%08x", (TInt) &aSms);

    CWapDatagram* datagram = new (ELeave)CWapDatagram();

    CleanupStack::PushL(datagram);

    datagram->ConstructL(aSms);

    CleanupStack::Pop();

    LOGWAPPROT2("CWapDatagram::NewL(): iFromPort: %d", datagram->iFromPort);

    LOGWAPPROT2("CWapDatagram::NewL(): iToPort: %d", datagram->iToPort);

    LOGWAPPROT2("CWapDatagram::NewL(): iReference: %d", datagram->iReference);

    LOGWAPPROT2("CWapDatagram::NewL(): iTotalSegments: %d", datagram->iTotalSegments);

    LOGWAPPROT2("CWapDatagram::NewL(): iSegmentNumber: %d", datagram->iSegmentNumber);

    LOGWAPPROT2("CWapDatagram::NewL(): iIsComplete: %d", datagram->iIsComplete);

    LOGWAPPROT2("CWapDatagram::NewL(): iReference: %d", datagram->iReference);

    LOGWAPPROT2("CWapDatagram::NewL(): iIsTextHeader: %d", datagram->iIsTextHeader);

    LOGWAPPROT2("CWapDatagram::NewL(): iLogServerId: %d", datagram->iLogServerId);

    LOGWAPPROT2("CWapDatagram::NewL(): i16BitPorts: %d", datagram->i16BitPorts);

    // assert destination port

	if (datagram->i16BitPorts)

		{

		if (datagram->iToPort>=0  &&  datagram->iToPort<=65535)

			{

			LOGWAPPROT1("iToPort OK");

			}

		else

			{

			LOGWAPPROT1("iToPort FAILED");

			}

		}

    else

		{

        if (datagram->iToPort>=0  &&  datagram->iToPort<=255)

			{

		    LOGWAPPROT1("iToPort OK");

			}

		else

			{

			LOGWAPPROT1("iToPort FAILED");

			}

		}

	if (datagram->i16BitPorts)

		{

		if (datagram->iFromPort>=0  &&  datagram->iFromPort<=65535)

			{

			LOGWAPPROT1("iFromPort OK");

			}

		else

			{

			LOGWAPPROT1("iFromPort FAILED");

			}

		}

    else

		{

        if (datagram->iFromPort>=0  &&  datagram->iFromPort<=255)

			{

		    LOGWAPPROT1("iFromPort OK");

			}

		else

			{

			LOGWAPPROT1("iFromPort FAILED");

			}

		}

    return datagram;

    } // CWapDatagram::NewL

//

// For outgoing short messages

//

CWapDatagram* CWapDatagram::NewL(const TDesC8& aSendBuffer)

    {

    LOGWAPPROT2("CWapDatagram::NewL(): aSendBuffer=0x%08x", (TInt) &aSendBuffer);

    CWapDatagram* datagram = new (ELeave)CWapDatagram();

    CleanupStack::PushL(datagram);

    datagram->Construct(aSendBuffer);

    CleanupStack::Pop();

    return datagram;

    } // CWapDatagram::NewL

CWapDatagram::~CWapDatagram()

    {

    LOGWAPPROT1("CWapDatagram::~CWapDatagram()");

	delete iRecvbuf;

    delete iBuffer;

    delete iSegment;

    } // CWapDatagram::~CWapDatagram

//

// What about service centre address ?

// aSmsMessageArray contains CSmsMessage objects

//

void CWapDatagram::EncodeConcatenatedMessagesL(RFs& aFs, CArrayPtr<CSmsMessage>& aSmsMessageArray)

    {

    LOGWAPPROT2("CWapDatagram::EncodeConcatenatedMessagesL(): %d messages", aSmsMessageArray.Count());

    // Couple of checkings makes sense

    __ASSERT_DEBUG(iToPort >=0 && iToAddress.Length()>=0

            && (iUserDataSettings.Alphabet()==TSmsDataCodingScheme::ESmsAlphabet8Bit

                 || iUserDataSettings.Alphabet()==TSmsDataCodingScheme::ESmsAlphabet7Bit),

            Panic(KPanicUsageError));

    // Determine CSmsMessage encoding by character width

    if (iUserDataSettings.Alphabet() == TSmsDataCodingScheme::ESmsAlphabet8Bit)

        {

        CSmsBufferBase* SmsBuffer = CSmsBuffer::NewL();

        CleanupStack::PushL(SmsBuffer);

        ConvertL(iSendBuffer,*SmsBuffer);

        CleanupStack::Pop(); // SmsBuffer, popped here since it is pushed again in the following NewL call

        CSmsMessage* SmsMessage = CSmsMessage::NewL(aFs, CSmsPDU::ESmsSubmit,SmsBuffer,EFalse);

        CleanupStack::PushL(SmsMessage);

        // one and only object

        aSmsMessageArray.AppendL(SmsMessage);

        CleanupStack::Pop(); // SmsMessage

        SetSmsMessageSettingsL(*SmsMessage,ETrue);

        }

    else

        {

        // contruct TWapTextMessage object

        TWapTextMessage* Segment = new (ELeave) TWapTextMessage(KNullDesC8);

        CleanupStack::PushL(Segment);

        CArrayPtrFlat<HBufC8>* SegmentArray = new (ELeave) CArrayPtrFlat<HBufC8> (8);

        CleanupStack::PushL(SegmentArray);

		// coverity[double_push]

        CleanupResetAndDestroyPushL(*SegmentArray);

        Segment->SetDestinationPort(iToPort,i16BitPorts);

        Segment->SetSourcePort(iFromPort,i16BitPorts);

        Segment->SetReferenceNumber(iReference);

        Segment->SetUserData(iSendBuffer);

        Segment->SetOtherHeader(iOtherHeader);

        Segment->EncodeSegmentsL(*SegmentArray);

        // contruct array of CSmsMessages

        TInt Count = SegmentArray->Count();

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

            {

            CSmsBufferBase* SmsBuffer = CSmsBuffer::NewL();

            CleanupStack::PushL(SmsBuffer);

            ConvertL(*(SegmentArray->At(i)),*SmsBuffer);

            CleanupStack::Pop(); // SmsBuffer, popped here since it is pushed again in the following NewL call

            CSmsMessage* SmsMessage=CSmsMessage::NewL(aFs, CSmsPDU::ESmsSubmit,SmsBuffer,EFalse);

            CleanupStack::PushL(SmsMessage);

            aSmsMessageArray.AppendL(SmsMessage);

            CleanupStack::Pop(); // SmsMessage

            SetSmsMessageSettingsL(*SmsMessage,EFalse);

            }

        CleanupStack::PopAndDestroy(3, Segment);  // SegmentArray elements (Reset and Destroy), SegmentArray, Segment

        }

    } // CWapDatagram::EncodeConcatenatedMessagesL

void CWapDatagram::DecodeConcatenatedMessagesL(CArrayPtr<TSegmentData>& aSmsMessageArray)

    {

    LOGWAPPROT1("CWapDatagram::DecodeConcatenatedMessagesL()");

    // The TSegmentData elements are in the random order in the array

    TInt Count = aSmsMessageArray.Count();

    TInt i=0;

    if (Count > 0)

        {

        TInt DataLength = 0;  // total length of data

        TInt TempLength = 0;

        // Every TSegmentData should be equal length except last segment

        // Use that 'constant' segment length as indexes

        // Count the actual data length

        for(i=0; i<Count; i++)

            {

            TempLength = aSmsMessageArray.At(i)->iData.Length();

            DataLength += TempLength;

            }

        TSegmentData* Segment = NULL;

        if (iBuffer)

            {

            delete iBuffer;

            iBuffer = NULL;

            }

        iBuffer = HBufC8::NewL(DataLength );

        TPtr8 BufferPtr(iBuffer->Des());

        BufferPtr.SetLength(DataLength );

        TInt segmentStartPosition=0;

        for(i=0; i<Count; i++)

            {

              if (segmentStartPosition >= DataLength)  

                  {

                  // Once the start position is out of range,

                  // there is no point in continuing reconstructing the

                  // wapdatagram.

                  break;

				 }			

            Segment = aSmsMessageArray.At(i);

			TPtr8 CopyBufferPtr(&(BufferPtr[segmentStartPosition]),0,160);

			CopyBufferPtr.Copy(Segment->iData);

 			segmentStartPosition+=Segment->iData.Length();

            }

        }

    iIsComplete = ETrue;

    } // CWapDatagram::DecodeConcatenatedMessagesL

void CWapDatagram::InternalizeL(RReadStream& aStream)

    {

    aStream >> iUserDataSettings;

    aStream >> iToAddress;

    iToPort = aStream.ReadInt32L();

    aStream >> iFromAddress;

    iFromPort = aStream.ReadInt32L();

    TInt64 time;

    aStream >> time;

    iTime = time;

    iUTCOffset = aStream.ReadInt32L();

    iIsTextHeader = aStream.ReadInt32L();

    iReference = aStream.ReadInt32L();

    iTotalSegments = aStream.ReadInt32L();

    iSegmentNumber = aStream.ReadInt32L();

    iLogServerId = aStream.ReadInt32L();

    iVersionNumber = aStream.ReadInt32L();

    iSpare1 =  aStream.ReadInt32L();

    iSpare2 =  aStream.ReadInt32L();

    iSpare3 =  aStream.ReadInt32L();

    // Required for version 1, which is reading and internalizing extra SMS parameters from stream

    if(iVersionNumber == EFirstVersion)

    	{

    	TInt length = aStream.ReadInt32L();

    	if(length>0)

    		{

    		HBufC8* smsBuffer=HBufC8::NewMaxLC(length);

			TPtr8 recvbuftmp = smsBuffer->Des();

			aStream >> recvbuftmp;

			if(!iRecvbuf)

				{

				iRecvbuf = CBufFlat::NewL(KSmsBufferExpansion);

				}

			iRecvbuf->ResizeL(recvbuftmp.Size());

			iRecvbuf->Write(0, recvbuftmp);

			CleanupStack::PopAndDestroy(smsBuffer);

			}

    	}

    } // CWapDatagram::InternalizeL

//

//	Reads iBuffer from stream (for 8Bit messages)

//

void CWapDatagram::InternalizeBufferL(RReadStream& aStream)

	{

	TInt length;

	length=aStream.ReadInt32L();

	iBuffer=HBufC8::NewMaxL(length);

	TPtr8 tmp = iBuffer->Des();

	aStream >> tmp;

	iIsComplete = ETrue;

	}

void CWapDatagram::ExternalizeL(RWriteStream& aStream) const

    {

    aStream << iUserDataSettings;

    aStream << iToAddress;

    aStream.WriteInt32L(iToPort);

    aStream << iFromAddress;

    aStream.WriteInt32L(iFromPort);

    aStream << iTime.Int64();

    aStream.WriteInt32L(iUTCOffset.Int());

    aStream.WriteInt32L(iIsTextHeader);

    aStream.WriteInt32L(iReference);

    aStream.WriteInt32L(iTotalSegments);

    aStream.WriteInt32L(iSegmentNumber);

    aStream.WriteInt32L(iLogServerId);

    aStream.WriteInt32L(iVersionNumber);

    aStream.WriteInt32L(iSpare1);

    aStream.WriteInt32L(iSpare2);

    aStream.WriteInt32L(iSpare3);

	// Externalizing the SMS params if exists

	if(iRecvbuf)

		{

		TPtr8 recvbuftmp = iRecvbuf->Ptr(0);

		aStream.WriteInt32L(recvbuftmp.Length());

		aStream << recvbuftmp;

		}

	else

		{

		aStream.WriteInt32L(0);

		}

    }

//

// writes iBuffer to writeStream for 8-bit incomming messages

//

void CWapDatagram::ExternalizeBufferL(RWriteStream& aStream) const

	{

	TInt length= WapDatagramLength();

	aStream.WriteInt32L(length);

	TPtr8 tmp= iBuffer->Des();

	aStream << tmp;

	} // CWapDatagram::ExternalizeBufferL

//

// Outgoing

//

void CWapDatagram::Construct(const TDesC8& aSendBuffer)

    {

    LOGWAPPROT1("CWapDatagram::Construct()");

	// Set version number to 1, as we have had to make 

	// changes to CWapDatagram for CR0929

	iVersionNumber = EFirstVersion;

    iSendBuffer.Set(aSendBuffer);

    } // CWapDatagram::Construct

//

// Incoming

// code actualy supports 7-bit characters and port numbers

// encoded in information elements (IE) although it is not

// a requirement

//

void CWapDatagram::ConstructL(const CSmsMessage& aSms)

    {

    LOGWAPPROT1("CWapDatagram::ConstructL()");

	// Set version number to 1, as we have had to make 

	// changes to CWapDatagram for CR0929

	iVersionNumber = EFirstVersion;

	iRecvbuf = CBufFlat::NewL(KSmsBufferExpansion);

	RBufWriteStream writestream(*iRecvbuf);

	writestream.Open(*iRecvbuf);

	CleanupClosePushL(writestream);

	// Externalizing everything within CSMSMessage except buffer

	aSms.ExternalizeWithoutBufferL(writestream);

	CleanupStack::PopAndDestroy();  //  writestream

	iRecvbuf->Compress();

    const CSmsBufferBase& SmsBufferBase = aSms.Buffer();

    GetDatagramSettings(aSms);

    // Get the data from CSmsMessage and convert

    // SmsBuffer store Unicode short message from CSmsMessage

    ConvertL(SmsBufferBase,&iBuffer);

	GetDatagramSettingsL();

    iIsComplete = ETrue;

    if (iUserDataSettings.Alphabet() == TSmsDataCodingScheme::ESmsAlphabet7Bit)

        {

        // this is a good candidate to include a text header.

        if (iIsTextHeader)

            {

            // there is still a minor chance that message is complete:

            // one fragment long datagram

            if (iTotalSegments == 1)

                {

                // Get the segment data

                TPtr8 PtrNarrow = iBuffer->Des();

                PtrNarrow.Zero();

                iSegment->UserData(PtrNarrow);

                }

            else

                {

                iIsComplete = EFalse;

                // delete incomplete message from iBuffer;

                delete iBuffer;

                iBuffer = 0;

                }

            }

        else

            // This means that IEs with 7-bits are used

            // Happily we support it

            iIsComplete = ETrue;

        }

    } // CWapDatagram::ConstructL

//

// Set WAP datagram private members from a text based concatenated message

// Copies contents of iBuffer to iSmsBuffer

// Allocates iSegment object

//

void CWapDatagram::GetDatagramSettingsL()

    {

    LOGWAPPROT1("CWapDatagram::GetDatagramSettingsL()");

	/* The WAP stack always receives 8 bit WAP datagrams from the SMS Stack as a single CSmsMessage.

	This is because 8 bit WAP messages which cannot be encoded into a single PDU are sent in a segmented SMS

	message which is reassembled inside the SMS Stack; the SMS Stack always passes the WAP stack

	a single CSmsMessage containing a single WAP message regardless of the message length.

	The WAP stack can receive 7 Bit WAP datagrams either as a single CSmsMessage or as a number of CSmsMessages

	which need to be reassembled into a single WAP message by the WAP stack.  The latter method is provided to maintain backward

	compatibility with early versions of the WAP stack which were implemented before the concatenation function was

	available in the SMS stack.*/

	/* 160 is the maximum number of character that can be fitted in a single PDU using 7 bit encoding */

	if((iUserDataSettings.Alphabet()== TSmsDataCodingScheme::ESmsAlphabet8Bit) ||

	(iUserDataSettings.Alphabet()== TSmsDataCodingScheme::ESmsAlphabet7Bit && iBuffer->Length()>160 ))

		{

		iSegment = new (ELeave)TWapTextMessage(*iBuffer);

		}

	else if (iBuffer->Length()<=160)

        {

        iSmsBuffer = *iBuffer;

		iSegment = new (ELeave)TWapTextMessage(iSmsBuffer);

        iIsTextHeader = iSegment->Parse();

        if (iIsTextHeader)

            {

            iFromPort = iSegment->SourcePort(&i16BitPorts);

            iToPort = iSegment->DestinationPort(&i16BitPorts);

            iReference = iSegment->ReferenceNumber();

            iTotalSegments = iSegment->TotalSegments();

            iSegmentNumber = iSegment->SegmentNumber();

            }

        }

	else

		{

        delete iSegment;

        iSegment = 0;

		}

     } // CWapDatagram::GetDatagramSettingsL

//

// Set WAP datagram private members from information element structures

//

void CWapDatagram::GetDatagramSettings(const CSmsMessage& aSms)

    {

    LOGWAPPROT1("CWapDatagram::GetDatagramSettings()");