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

// class for the rtp core unit test cases

// 

//

/**

 @file

 @internalComponent

*/

#include <test/testexecutelog.h>

#include <in_sock.h>

#include <bt_sock.h>

#include <sys/time.h>

#include <badesca.h>

#include <random.h>

#include "trtpcorestep.h"

#include "trtpcoreserver.h"

#include "rtpsock.h"

#include "rtpcollisionmng.h"

#include "rtpbthandler.h"

// CF Rtp Related..

#ifdef SYMBIAN_NON_SEAMLESS_NETWORK_BEARER_MOBILITY 

#include <rtp_uid.h>

#include "trtpsock_utils.h"

#endif

TVerdict CTRtpCoreStep::doTestStepL()

	{

	iServer->iCurrentStep = this;

	iServer->iStopping = EFalse;

	iServer->iStopCount = 1;

	if (!iServer->iOom)

		return doRtpTestStepL();

	else

		{

		TVerdict verdict = EFail;

		for (TInt oomCount = 0; ; oomCount++)

			{

			__UHEAP_RESET;

			__UHEAP_SETFAIL(RHeap::EDeterministic, oomCount);

			TInt countBefore = User::CountAllocCells();

			INFO_PRINTF2(_L("countbefore at %d"), countBefore);

			iServer->iStopping = EFalse;

			iServer->iStopCount = 1;

			iIgnoreIncrease = EFalse;

			iError = KErrNone;

			TRAPD(error, verdict = doRtpTestStepL());

			TInt countAfter = User::CountAllocCells();

			INFO_PRINTF2(_L("countafter at %d"), countAfter);

			__UHEAP_RESET;

			if (error != KErrNoMemory && iError != KErrNoMemory)

				{

				break;

				}

			else

				{

				if (countBefore != countAfter && !iIgnoreIncrease)

					{

					INFO_PRINTF2(_L("OOM Failed at %d"), oomCount);

					verdict = EFail;

					break;

					}

				}

			}

		SetTestStepResult(verdict);

		return verdict;

		}

	}

TVerdict COom::doRtpTestStepL()

	{

	SetTestStepResult(EPass);

	TBool oom;

	GetBoolFromConfig(ConfigSection(),_L("Oom"),oom);

	INFO_PRINTF2(_L("Setting OOM mode to %d"), oom);

	iServer->iOom = oom;

	return TestStepResult();

	}

TVerdict COpenRtpSocket::doTestStepPreambleL()

/**

 * @return - TVerdict code

 * Override of base class virtual

 */

	{

	SetTestStepResult(EPass);

	User::LeaveIfError(iServer->iSocketServer.Connect());

	return TestStepResult();

	}

TVerdict COpenRtpSocket::doRtpTestStepL()

/**

 * @return - TVerdict code

 * Override of base class pure virtual

 * Demonstrates reading configuration parameters fom an ini file section

 */

	{

	SetTestStepResult(EPass);

	TInetAddr localAddress(KInetAddrLoop, 9000);

	// family KAfInet

	// protocol should be KProtocolInetRtp rather then UDP

	iServer->iController->RtpSocket().Open(iServer->iSocketServer, localAddress.Family(), 

											KSockDatagram,KProtocolInetUdp);

	return TestStepResult();

	}

TVerdict CCreateRtpController::doTestStepPreambleL()

	{

	SetTestStepResult(EPass);

		if(iServer->iScheduler)

		{

		SetTestStepResult(ETestSuiteError);	

		return TestStepResult();

		}

	iServer->iScheduler = new (ELeave) CActiveScheduler;

	CActiveScheduler::Install(iServer->iScheduler);

	//

	TInt err = iServer->iSocketServer.Connect();

	INFO_PRINTF2(_L("iSocketServer.Connect()  err = %d"), err);

	User::LeaveIfError(err);

	User::LeaveIfError(iServer->iConnection.Open(iServer->iSocketServer));

	// open a socket just in case we need it...

	TInt addrFamily; // either KAfInet or BT

	GetIntFromConfig(ConfigSection(),_L("AddressFamily"),addrFamily);

	if(KBTAddrFamily == static_cast<TUint>(addrFamily))

		{

		User::LeaveIfError(iServer->iRtpSocket.Open(iServer->iSocketServer,_L("L2CAP")));

		User::LeaveIfError(iServer->iRtcpSocket.Open(iServer->iSocketServer,_L("L2CAP")));

		//TInt error = iServer->iRtpSocket.Open(iServer->iSocketServer,_L("L2CAP"));

		}

		else

	if(KAfInet == static_cast<TUint>(addrFamily))

		{

		//User::LeaveIfError(iServer->iRtpSocket.Open(iServer->iSocketServer,_L("udp")));

		User::LeaveIfError(iServer->iRtpSocket.Open(iServer->iSocketServer,

													KAfInet, KSockDatagram, KProtocolInetUdp));

		User::LeaveIfError(iServer->iRtcpSocket.Open(iServer->iSocketServer,

													KAfInet, KSockDatagram, KProtocolInetUdp));

		}

	return TestStepResult();

	}

TVerdict CCreateRtpController::doRtpTestStepL()

/**

 * @return - TVerdict code

 * Override of base class pure virtual

 * Demonstrates reading configuration parameters fom an ini file section

 */

	{

	iIgnoreIncrease = 1;

	INFO_PRINTF1(_L("In Step:"));

	INFO_PRINTF1(KCreateRtpController);

	SetTestStepResult(EFail);

	TPtrC controllerType;

	GetStringFromConfig(ConfigSection(),_L("Type"),controllerType);

	TInt addrFamily; // either KAfInet or BT

	GetIntFromConfig(ConfigSection(),_L("AddressFamily"),addrFamily);

	TSockAddr localAddr;

	localAddr.SetFamily(addrFamily);

	TInt rtpPort;

	GetIntFromConfig(ConfigSection(),_L("RtpPortNumber"),rtpPort);

	localAddr.SetPort(rtpPort);

	TPtrC cname16;

	//

	GetStringFromConfig(ConfigSection(),_L("CNAME"),cname16);	

	HBufC8* cname = HBufC8::NewMaxLC(cname16.Length());

	cname->Des().Copy(cname16);

	//

	TInt address;

	GetIntFromConfig(ConfigSection(),_L("LocalAddress"),address);

	switch(addrFamily)

		{

		case KAfInet:

			{

			TInetAddr bearerAddr(localAddr);

			bearerAddr.SetAddress(address);

			//bearerAddr.SetAddress(KInetAddrAny);

			localAddr = bearerAddr;

			break;

			}

		case KBTAddrFamily:

			{

			TBTSockAddr bearerAddr(localAddr);

			TBTDevAddr btDevAddr(address);

			bearerAddr.SetBTAddr(btDevAddr);

			localAddr = bearerAddr;

			break;

			}

		default:

			SetTestStepResult(ETestSuiteError);

			CleanupStack::PopAndDestroy(cname);

			return TestStepResult();

		}

	//

	TBool setRtpStreamParameters = EFalse;

	TInt maxDropout = 0;

	TInt maxMisorder = 0;

	TInt minSequential = 0;

	if(GetIntFromConfig(ConfigSection(),_L("MaxDropout"),maxDropout))

		setRtpStreamParameters = ETrue;

	if(GetIntFromConfig(ConfigSection(),_L("MaxMisorder"),maxMisorder))

		setRtpStreamParameters = ETrue;

	if(GetIntFromConfig(ConfigSection(),_L("MinSequential"),minSequential))

		setRtpStreamParameters = ETrue;

	//

	if (controllerType == _L("ControllerFromSockServ"))

		{

		SetTestStepResult(EPass);

		iServer->iUsingConnectedSocket = EFalse;

		// create RTP socket only from TSockAddr

		CRtpControllerBase* controller = CRtpControllerBase::NewL(iServer->iSocketServer,localAddr,KNullDesC8);

		CleanupStack::PushL(controller);

		// lets check a few things...

		CRtpHandlerBase* rtpHandler = controller->RtpHandler();

		if(ERtp != rtpHandler->Type())

			SetTestStepResult(EFail);

		// lets test address in handler

		localAddr.Family();

		TSockAddr tmp = rtpHandler->LocalAddress();

		tmp.Family();

		if(!rtpHandler->CompareAddresses(localAddr,rtpHandler->LocalAddress()))

			SetTestStepResult(EFail);

		iServer->iController = (CRtpControllerBase *)controller;

		CleanupStack::Pop(controller);

		}

	if (controllerType == _L("ControllerFromSockServWithCNAME"))

		{

		SetTestStepResult(EPass);

		iServer->iUsingConnectedSocket = EFalse;

		CRtpControllerBase* controller = CRtpControllerBase::NewL(iServer->iSocketServer,localAddr, *cname);

		CleanupStack::PushL(controller);

		// lets check a few things...

		CRtpHandlerBase* rtpHandler = controller->RtpHandler();

		CRtpHandlerBase* rtcpHandler = controller->RtcpHandler();

		TInt rtpPortTest = rtpHandler->LocalAddress().Port();

		TInt rtcpPortTest = 0;

		if(0 != rtcpHandler)

			{

			rtcpPortTest = rtcpHandler->LocalAddress().Port();

			}

		if(ERtp != rtpHandler->Type())

			SetTestStepResult(EFail);

		if(!rtpHandler->CompareAddresses(localAddr,rtpHandler->LocalAddress()))

			SetTestStepResult(EFail);

		if(0 != rtcpHandler)

			{

			if(ERtcp != rtcpHandler->Type())

				SetTestStepResult(EFail);

			if(rtpPortTest != (rtcpPortTest-1))

				SetTestStepResult(EFail);

			}

		else

			SetTestStepResult(EFail);

		iServer->iController = controller;

		CleanupStack::Pop(controller);

		}

	if (controllerType == _L("ControllerFromSockServWithRtcp"))

		{

		SetTestStepResult(EPass);

		iServer->iUsingConnectedSocket = EFalse;

		TInt rtcpPort;

		GetIntFromConfig(ConfigSection(),_L("RtcpPortNumber"),rtcpPort);

		// create RTP and RTCP sockets from TSockAddr and rtcp port number

		CRtpControllerBase* controller = CRtpControllerBase::NewL(iServer->iSocketServer,localAddr,rtcpPort,*cname);

		CleanupStack::PushL(controller);

		// lets check a few things...

		CRtpHandlerBase* rtpHandler = controller->RtpHandler();

		CRtpHandlerBase* rtcpHandler = controller->RtcpHandler();

//		TInt rtpPortTest = rtpHandler->LocalAddress().Port();

		TInt rtcpPortTest = 0;

		if(0 != rtcpHandler)

			{

			rtcpPortTest = rtcpHandler->LocalAddress().Port();

			}

		if(ERtp != rtpHandler->Type())

			SetTestStepResult(EFail);

		if(!rtpHandler->CompareAddresses(localAddr,rtpHandler->LocalAddress()))

			SetTestStepResult(EFail);

		if(0 != rtcpHandler)

			{

			if(ERtcp != rtcpHandler->Type())

				SetTestStepResult(EFail);

			if(rtcpPort != rtcpPortTest)

				SetTestStepResult(EFail);

			}

		else

			SetTestStepResult(EFail);

		iServer->iController = controller;

		CleanupStack::Pop(controller);

		}

	if (controllerType == _L("ControllerFromSocket"))

		{

		SetTestStepResult(EPass);

		iServer->iUsingConnectedSocket = ETrue;

		User::LeaveIfError(iServer->iRtpSocket.Bind(localAddr));

		// create RTP socket only from TSockAddr

		iServer->iController = CRtpControllerBase::NewL(iServer->iRtpSocket);

		// lets check a few things...

		CRtpHandlerBase* rtpHandler = iServer->iController->RtpHandler();

		if(ERtp != rtpHandler->Type())

			SetTestStepResult(EFail);

		// lets test address in handler

		if(!rtpHandler->CompareAddresses(localAddr,rtpHandler->LocalAddress()))

			SetTestStepResult(EFail);

		// and address in socket itself

		TSockAddr testAddr;

		rtpHandler->RtpSocket().LocalName(testAddr);

		if(!rtpHandler->CompareAddresses(localAddr,testAddr))

			SetTestStepResult(EFail);

		}

	if (controllerType == _L("ControllerFromSocketWithRtcp"))

		{

		SetTestStepResult(EPass);

		iServer->iUsingConnectedSocket = ETrue;

		}

	if(setRtpStreamParameters)

		{

		if(KErrNone != iServer->iController->SetRtpStreamParameters(maxDropout,maxMisorder,minSequential))

			SetTestStepResult(EFail);

		}

	CleanupStack::PopAndDestroy(cname);

	return TestStepResult();

	}

TVerdict CActivateRtpController::doRtpTestStepL()

	{

	TInt type; // the type of RTP connections

	if(!GetIntFromConfig(ConfigSection(),_L("ConnectionType"),type))

		{

		SetTestStepResult(ETestSuiteError);

		return TestStepResult();

		}

	if(iServer->iController)

		SetTestStepResult(EPass);

	else

		{

		SetTestStepResult(ETestSuiteError);

		return TestStepResult();

		}

	iServer->iController->SetRTPTimeConversion(100, 100);

	TRAPD(error,iServer->iController->ActivateL(TRtpConnectionType(type)));

	//const TInt KErrInUse=(-14);

	//const TInt KErrServerTerminated=(-15);

	if(KErrNone != error)

		{

		/* We are interested in this particular error since it's returned from socket 

		when trying to activate already active handler/socket. */

		if((KErrInUse == error)||(KErrArgument == error))

			{

			SetTestStepResult(EFail);

			}

		else if(KErrNotSupported == error)

			{

			// returned for COInConnection over UDP

			SetTestStepResult(EFail);

			}

		else

			User::Leave(error);

		}

	else

		{

		// lets test RTP socket now 

		CRtpHandlerBase* rtpHandler = iServer->iController->RtpHandler();

		TSockAddr testAddr;

		rtpHandler->RtpSocket().LocalName(testAddr);

		if(!rtpHandler->CompareAddresses(rtpHandler->LocalAddress(),testAddr))

			SetTestStepResult(EFail);

		}

	return TestStepResult();

	}

TVerdict CDeleteRtpController::doRtpTestStepL()

	{

	if(iServer->iController)

		SetTestStepResult(EPass);

	else

		{

		SetTestStepResult(ETestSuiteError);

		return TestStepResult();

		}

	TSockAddr rtpAddr;

	if(iServer->iUsingConnectedSocket)

		{

		iServer->iController->RtpHandler()->RtpSocket().LocalName(rtpAddr);

		}

	delete iServer->iController;

	iServer->iController = 0;

	if(iServer->iUsingConnectedSocket)

		{

		// lets check we did not take down the sockets with controller

		// buffers

		TBuf8<1024> sendBuf;

		sendBuf.SetLength(10);

		TBuf8<1024> recvBuf;

		recvBuf.SetLength(10);

		// prepare reciver (RTP socket used previsously by controller)

		CWaiter* waiter = CWaiter::NewLC(iServer);

		TSockAddr dummyAddr;

		iServer->iRtpSocket.RecvFrom(recvBuf,dummyAddr, 0,waiter->iStatus);

		// sender

		User::LeaveIfError(iServer->iSocket.Open(iServer->iSocketServer, KAfInet, 

									   KSockDatagram, KProtocolInetUdp));

		CleanupClosePushL(iServer->iSocket);

		CWaiter* sender = CWaiter::NewLC(iServer);

		iServer->iSocket.SendTo(sendBuf,rtpAddr,0,sender->iStatus);

		User::WaitForRequest(sender->iStatus);

		//

		User::WaitForRequest(waiter->iStatus);

		//

		CleanupStack::PopAndDestroy(3); // socket, waiter, sender

		}

	iServer->iSocket.Close();

	iServer->iRtpSocket.Close();

	iServer->iRtcpSocket.Close();

	iServer->iUsingConnectedSocket = EFalse;

	// NOTE: This wil panic if already closed!!! Defect in ESock?!?

	iServer->iSocketServer.Close();

	//

	CActiveScheduler::Install(NULL);

	delete iServer->iScheduler;

	iServer->iScheduler = 0;

	return TestStepResult();

	}

// Close everything except the session - Used to tidy up after close

// tests to check evrything is closed.

TVerdict CTidyUp::doRtpTestStepL()

	{

	SetTestStepResult(EPass);

	iServer->iSocketServer.Close();

	CActiveScheduler::Install(NULL);

	delete iServer->iScheduler;

	iServer->iScheduler = 0;

	return TestStepResult();

	}

TVerdict CBasicSend::doTestStepPreambleL()

	{

	SetTestStepResult(EPass);

	return TestStepResult();

	}

TVerdict CBasicSend::doRtpTestStepL()

	{

	if(0 == CActiveScheduler::Current())

		{

		SetTestStepResult(ETestSuiteError);	

		return TestStepResult();

		}

	if(0 == iServer->iController)

		{

		SetTestStepResult(ETestSuiteError);	

		return TestStepResult();

		}

	SetTestStepResult(EPass);

	TBuf8<1024> sendBuf;

	sendBuf.SetLength(50);

	TInt i;

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

		sendBuf[i]=0;

	TBuf8<1024> recvBuf; // To put the packet in

	recvBuf.SetLength(50);

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

		recvBuf[i]=55;

	TInt stopCount;

	if (!GetIntFromConfig(ConfigSection(), _L("StopCount"), stopCount))

		stopCount = 2;

	TInt port;