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

// @file

// 

//

#include "TE_wapdgrmbase.h"

#include "smsstackutils.h"

// WapProt code needs it

GLDEF_C void Panic(TWAPSMSPanic aPanic)

{

	_LIT(KWapProtPanic, "WAPPROT");

	User::Panic(KWapProtPanic, aPanic);

}

// implementation of helper functions

void CWapDgrmTestStep::_Print(TInt aNum)

/**

 *  print a number to the console

 */

{

	TBuf<16>  NumStr;

	// Ordinal

	NumStr.Num(aNum);

	INFO_PRINTF1(NumStr);

}

void CWapDgrmTestStep::_Print(const TPtrC8 aStr)

/**

 *  print an 8-bit descriptor to the console

 */

{

	TBuf<1024> PrintBuffer;

	PrintBuffer.Copy(aStr);

	INFO_PRINTF1(PrintBuffer);

}

void CWapDgrmTestStep::_Print(const TDesC& aStr)

/**

 *  print a descriptor to the console

 */

{

	INFO_PRINTF1(aStr);

}

void CWapDgrmTestStep::_PrintBool(TBool aTruth)

/**

 *  print a boolean number to the console

 */

{

	_Print((aTruth ? _L8("true") : _L8("false")));

}

void CWapDgrmTestStep::_Print(CWapDatagram& aWap)

/**

 *  print a WAP datagram to the console

 */

{

	TBool                IsComplete = EFalse;

	const TPtrC&         ToAddress = aWap.ToAddress();

	const TPtrC&         FromAddress = aWap.FromAddress();

	TInt                 ToPort = 0;

	TInt                 FromPort = 0;

	TInt                 Is16BitPorts = 0;

	TBool                IsPorts = ETrue;

	TSmsDataCodingScheme::TSmsAlphabet Alphabet;

	IsComplete = aWap.IsComplete();

	Alphabet = aWap.Alphabet();

	_Print(_L8("CWapDatagram:: IsComplete: FromAddress: ToAddress: Alphabet\n"));

	_PrintBool(IsComplete);

	_Print(_L8(": "));

	_Print(FromAddress);

	_Print(_L8(": "));

	_Print(ToAddress);

	_Print(_L8(": "));

	_Print(Alphabet);

	_Print(_L8("\n"));

	aWap.Ports(FromPort,ToPort,&Is16BitPorts);

	_Print(_L8("IsPorts: FromPort: ToPort: Is16BitPorts\n"));

	_PrintBool(IsPorts);

	_Print(_L8(": "));

	_Print(FromPort);

	_Print(_L8(": "));

	_Print(ToPort);

	_Print(_L8(": "));

	_PrintBool(Is16BitPorts);

	_Print(_L8("\n"));

   	if (aWap.IsComplete())

   		{

   		// allocate buffer for data

   		HBufC8* DataBuffer = HBufC8::New(aWap.WapDatagramLength());

   		TPtr8 DataBufferPtr(DataBuffer->Des());

 		RMBufChain dgramChain;

 		aWap.WapDatagram(dgramChain, aWap.WapDatagramLength());

 		dgramChain.CopyOut(DataBufferPtr);

   		// Length

   		_Print(DataBuffer->Length());

   		_Print(_L8(": <"));

   		// Data

   		_Print(*DataBuffer);

   		_Print(_L8(">\n"));

   		delete DataBuffer;

   		}

}

void CWapDgrmTestStep::_Print(CSmsMessage& aSms)

/**

 *  print a SMS message to the console

 */

{

	TSmsUserDataSettings UserDataSettings;

	TBool                IsComplete = EFalse;

	const TPtrC&         Address = aSms.ToFromAddress();

	TInt                 ToPort = 0;

	TInt                 FromPort = 0;

	TInt                 Is16BitPorts = 0;

	TBool                IsPorts = EFalse;

	TSmsDataCodingScheme::TSmsAlphabet Alphabet;

	// Print CSmsMessage settings

	CSmsPDU& Pdu = aSms.SmsPDU();

	IsComplete = aSms.IsComplete();

	aSms.UserDataSettings(UserDataSettings);

	Alphabet = UserDataSettings.Alphabet();

	_Print(_L8("CSmsMessage:: IsComplete: ToFromAddress: Alphabet\n"));

	_PrintBool(IsComplete);

	_Print(_L8(": "));

	_Print(Address);

	_Print(_L8(": "));

	_Print(Alphabet);

	_Print(_L8("\n"));

	IsPorts = Pdu.ApplicationPortAddressing(ToPort,FromPort,&Is16BitPorts);

	_Print(_L8("IsPorts: FromPort: ToPort: Is16BitPorts\n"));

	_PrintBool(IsPorts);

	_Print(_L8(": "));

	_Print(FromPort);

	_Print(_L8(": "));

	_Print(ToPort);

	_Print(_L8(": "));

	_PrintBool(Is16BitPorts);

	_Print(_L8("\n"));

}

void CWapDgrmTestStep::_PrintL(CSmsBufferBase& aSmsBuffer)

/**

 *  print a SMS buffer to the console

 */

{

	HBufC16* WideBuffer = NULL;

	TInt Length = 0;

	// copy the data to the descriptor from aSmsBuffer

	Length = aSmsBuffer.Length();

	WideBuffer = HBufC16::NewL(Length);

	TPtr16 WideChars = WideBuffer->Des();

	aSmsBuffer.Extract(WideChars,0,Length);

	// Length

	_Print(WideChars.Length());

	_Print(_L8(": <"));

	// Data

	_Print(WideChars);

	_Print(_L8(">\n"));

	delete WideBuffer;

}

void CWapDgrmTestStep::_PrintL(CArrayPtr<CSmsMessage>& aSmsArray)

/**

 *  print an array of SMS messages to the console

 */

{

	TInt Count = aSmsArray.Count();

	TInt j = 1;

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

	{

		// Ordinal

		_Print(i);

		_Print(_L8(": \n"));

		CSmsMessage* Sms = aSmsArray.At(i);

		_Print(*Sms);

		// Print data

		CSmsBufferBase& SmsBuffer = Sms->Buffer();

		_PrintL(SmsBuffer);

	}

}

void CWapDgrmTestStep::_FlushReserveArrayL(CWapReassemblyStore* aWapStore)

/**

 *  Method should be called only once during a test program run !

 */

{

	TInt Count = iReserveArray->Count();

	// In reverse oder

	for (TInt i=Count-1;i>=0;i--)

	{

		CWapDatagram* Wap=iReserveArray->At(i);

		if (!Wap->IsComplete())

		{

			TInt Index = 0;

			TBool IsComplete = EFalse;

			IsComplete = aWapStore->AddMessageL(Index,*Wap);

			if (IsComplete)

			{

				aWapStore->GetDatagramL(Index,*Wap);

				aWapStore->BeginTransactionLC();

				aWapStore->DeleteEntryL(Index);

				aWapStore->CommitTransactionL();

				_Print(*Wap);

			}

		}

		else

			_Print(*Wap);

	}

	iReserveArray->ResetAndDestroy();

	delete iReserveArray;

	iReserveArray = NULL;

}

void CWapDgrmTestStep::_Parse2L(CArrayPtrFlat<CSmsMessage>& aSmsArray,

						CWapReassemblyStore* aWapStore)

/**

 *  This method tests mainly CWapDatagram::DecodeConcatenatedMessagesL

 */

{

    TInt Count = aSmsArray.Count();

    // In reverse order

    // Note ! You can test additional features by changing value of

    // i or a global variable (insertSms). For example:

    // incrementing i by 1: a duplicate test

    // decrementing i by 1: not all datagrams are pushed to the store

    // ncreasing insertSms causes datagrams to be appended to the reserve array

    // Remember to set flush the reserve array on a later phase

    // by setting iIsFlushReserveArray true

    for (TInt i=Count-1;i>=iInsertSms;i--)

    {

        CWapDatagram* Wap=CWapDatagram::NewL(*(aSmsArray.At(i)));

        if (!Wap->IsComplete())

        {

            TInt Index = 0;

            TBool IsComplete = EFalse;

            IsComplete = aWapStore->AddMessageL(Index,*Wap);

            if (IsComplete)

            {

                aWapStore->GetDatagramL(Index,*Wap);

				aWapStore->BeginTransactionLC();

                aWapStore->DeleteEntryL(Index);

				aWapStore->CommitTransactionL();

                _Print(*Wap);

            }

        }

        else

            _Print(*Wap);

        delete Wap;

    }

	// append rest of the datagrams to the ReserveArray

	if (iInsertSms>0 && Count>1)

	{

		// reserve order here too

        for (TInt i=iInsertSms-1;i>=0;i--)

        {

            CWapDatagram* Wap=CWapDatagram::NewL(*(aSmsArray.At(i)));

            iReserveArray->AppendL(Wap);

        }

	}

    if (iIsFlushReserveArray)

    {

        _FlushReserveArrayL(aWapStore);

        iInsertSms = 0;

        iIsFlushReserveArray = EFalse;

    }

}

void CWapDgrmTestStep::ConstructWap(CWapDatagram& aWap,

						TDesC& aFromAddress, TDesC& aToAddress,

						TInt aFromPort, TInt aToPort,

						TSmsDataCodingScheme::TSmsAlphabet aAlphabet,

						TBool aIs16BitPorts = EFalse)

/**

 *  create a WAP datagram, fill in data like to/from adress, ports, etc...

 */

{

	static TInt Ref = 0;

	Ref++;

	aWap.SetFromAddress(aFromAddress);

	aWap.SetToAddress(aToAddress);

	aWap.SetPorts(aFromPort,aToPort,aIs16BitPorts);

	aWap.SetAlphabet(aAlphabet);

	aWap.SetConcatenatedMessageReference(Ref);

}

void CWapDgrmTestStep::DoTestWapProtDatagramL(const TDesC8& aBuffer,

                        TDesC& aToAddress, TInt aToPort, TDesC& aFromAddress,

                        TInt aFromPort, TSmsDataCodingScheme::TSmsAlphabet aAlphabet,

                        TBool aIs16BitPorts)

/**

 *  @test a short 7/8-bit datagram with explicit 16-bit port number

 */

    {

        INFO_PRINTF1(_L("a 7/8-bit datagram with explicit 16-bit port number"));

        CArrayPtrFlat<CSmsMessage>* SmsArray = 0;

        CWapDatagram* Wap = NULL;

        RFs FileServerSession;

        CWapReassemblyStore* wapStore = NULL;

        if(iIsRunParsing)

        {

            TInt ret = FileServerSession.Connect();

            TESTL(ret == KErrNone);

            wapStore = CWapReassemblyStore::NewL(FileServerSession);

            CleanupStack::PushL(wapStore);

        }

        _Print((aBuffer).Length());_Print(_L8(": <"));_Print(aBuffer);_Print(_L8(">\n"));

        Wap=CWapDatagram::NewL(aBuffer);

        CleanupStack::PushL(Wap);        

        SmsArray = new (ELeave) CArrayPtrFlat<CSmsMessage>(8);

        CleanupStack::PushL(SmsArray);

        CleanupResetAndDestroyPushL(*SmsArray);        

        ConstructWap(*Wap, aFromAddress, aToAddress, aFromPort, aToPort, aAlphabet, aIs16BitPorts);

        Wap->EncodeConcatenatedMessagesL(iFs, *SmsArray);

        if (iIsPrintEncoded)

            _PrintL(*SmsArray);

        if (iIsRunParsing)

            _Parse2L(*SmsArray, wapStore);

        CleanupStack::PopAndDestroy(3, Wap);  // SmsArray elements (Reset and Destroy), SmsArray, Wap       

        if (iIsRunParsing)

        {

            CleanupStack::PopAndDestroy(wapStore);

            FileServerSession.Close();

        }

    }

TVerdict CWapDgrmTestStep::doTestStepPreambleL()

	{

	const TInt KMBuf_MaxAvail = 393216;

	const TInt KMBuf_MBufSize = 128;

	const TInt KMBuf_MBufSizeBig = 1600;

	const TInt KMBuf_InitialAllocation = 128;

	const TInt KMBuf_MinGrowth = 64;

	const TInt KMBuf_GrowthThreshold = 40;

	__UHEAP_MARK;

#ifdef __CFLOG_ACTIVE

	__CFLOG_CREATEL;

	__CFLOG_OPEN;

#endif

	iIsFlushReserveArray = EFalse;

	iReserveArray = NULL;

	iInsertSms = 0;

	iIsRunParsing = ETrue;

	iIsPrintEncoded = ETrue;

    // Create and install the active scheduler

    iScheduler=new (ELeave) CActiveScheduler;

    CActiveScheduler::Install(iScheduler);

	// The WAP protocol unpacking code uses MBufs, so we need a local MBufManager

#ifdef SYMBIAN_ZERO_COPY_NETWORKING	

    // Initialize the commsbuf pond

    RArray<TCommsBufPoolCreateInfo> createInfos;

    TCommsBufPoolCreateInfo createInfo;

    createInfo.iBufSize = KMBuf_MBufSize;

    createInfo.iInitialBufs = KMBuf_InitialAllocation;

    createInfo.iGrowByBufs = KMBuf_MinGrowth;

    createInfo.iMinFreeBufs = KMBuf_GrowthThreshold;

    createInfo.iCeiling = (KMBuf_MaxAvail/2) / KMBuf_MBufSize;

    createInfos.AppendL(createInfo);

    TCommsBufPoolCreateInfo createInfo2;

    createInfo2.iBufSize = KMBuf_MBufSizeBig;

    createInfo2.iInitialBufs = KMBuf_InitialAllocation;

    createInfo2.iGrowByBufs = KMBuf_MinGrowth;

    createInfo2.iMinFreeBufs = KMBuf_GrowthThreshold;

    createInfo2.iCeiling = (KMBuf_MaxAvail/2) / KMBuf_MBufSize;    

    createInfos.AppendL(createInfo2);

	User::LeaveIfError(iCommsBufPond.Open(createInfos));

	createInfos.Close();

#else

	MMBufSizeAllocator*  mBufSizeAllocator;

	iBufMgr = CMBufManager::NewL(KMBuf_MaxAvail, mBufSizeAllocator);

	if (iBufMgr != NULL  &&  mBufSizeAllocator != NULL)

		{

		mBufSizeAllocator->AddL(KMBuf_MBufSize,    KMBuf_InitialAllocation, KMBuf_MinGrowth, KMBuf_GrowthThreshold);

		mBufSizeAllocator->AddL(KMBuf_MBufSizeBig, KMBuf_InitialAllocation, KMBuf_MinGrowth, KMBuf_GrowthThreshold);

		}

	iBufMgr->SetContext();

#endif

	iWapDgrmTable[0] = &KWapDgrm1;

	iWapDgrmTable[1] = &KWapDgrm2;

	iWapDgrmTable[2] = &KWapDgrm3;

	iWapDgrmTable[3] = &KWapDgrm4;

	iWapDgrmTable[4] = &KWapDgrm5;

	iWapDgrmTable[5] = &KWapDgrm6;

	iWapDgrmTable[6] = NULL;

	TInt ret = iFs.Connect();

    TESTL(ret == KErrNone);

	return TestStepResult() ;

	}

TVerdict CWapDgrmTestStep::doTestStepPostambleL()

	{

    iFs.Close();

#ifdef SYMBIAN_ZERO_COPY_NETWORKING	

	iCommsBufPond.Close();

#else

   	delete iBufMgr;

   	iBufMgr = NULL;

#endif 	// SYMBIAN_ZERO_COPY_NETWORKING	

	delete iScheduler;

	iScheduler = NULL;

#ifdef __CFLOG_ACTIVE

	__CFLOG_CLOSE;

	__CFLOG_DELETE;

#endif

	__UHEAP_MARKEND;

	return TestStepResult() ;

	}