/*

* Copyright (c) 2004 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 HEADER

#include "UT_CRtpSession.h"

//  EXTERNAL INCLUDES

#include <digia/eunit/eunitmacros.h>

//  INTERNAL INCLUDES

#include "rtpsession.h"

#include "rtppacketext.h"

#include "rtpcommsend.h"

const TUint KDefaultIap = 6;

//_LIT( KDefaultPath, "c:\\system\\data\\RtpStpPacket.dll" );

const TUint8 KRtpPayloadClockConversions[KRtpNumPayloadTypes] =

    {

    125, 125, 125, 125, 125, 125, 63, 125, 125, 63, // 0-9 

    23, 23, 125, 125, 11, 125, 91, 45, 125, 0,      // 10-19

    0, 0, 0, 0, 0, 11, 11, 0, 11, 0,                // 20-29

    0, 11, 11, 11, 11, 0, 0, 0, 0, 0,               // 30-39

    0, 0, 0, 0, 0, 0, 0, 0, 0, 0,                   // 40-49

    0, 0, 0, 0, 0, 0, 0, 0, 0, 0,                   // 50-59

    0, 0, 0, 0, 0, 0, 0, 0, 0, 0,                   // 60-69

    0, 0, 0, 0, 0, 0, 0, 0, 0, 0,                   // 70-79

    0, 0, 0, 0, 0, 0, 0, 0, 0, 0,                   // 80-89

    0, 0, 0, 0, 0, 0, 0, 0, 0, 0,                   // 90-99

    0, 0, 0, 0, 0, 0, 0, 0, 0, 0,                   // 100-109

    0, 0, 0, 0, 0, 0, 0, 0, 0, 0,                   // 110-119

    0, 0, 0, 0, 0, 0, 0, 0                          // 120-127

    };

const TUint KUTRtpMaxPayloadTypes = 128; 

_LIT8(KRTCPPacket1, "80C90001DEADBEEF9E30159059A23363F0FBDB62DC85AF52875C944495C5B00280000001ABCDEF1234567890ABCDEF1234567890DFC98F01DE16DA30F20D");

_LIT8(KRTCPPacket2, "80C80006DEADBEEFC7B1ACD423FE000000E12F18000000010000020081CA0015DEADBEEF010A52747055492075736572020455736572030D7573657240686F6D652E6E6574040B2B313233343536373839300504486F6D65060B527470206578616D706C65070641637469766508010000000000");

_LIT8(KRTCPPacket3, "80C90001000000029E30159059A23363F0FBDB62DC85AF52875C944495C5B00280000001ABCDEF1234567890ABCDEF1234567890DFC98F01DE16DA30F20D");

_LIT8(KRTCPPacket4, "80CB0001000000029E30159059A23363F0FBDB62DC85AF52875C944495C5B00280000001");

_LIT8(KRTPPacket1, "8001CDDD00000001DEADBEEF9206D8823BACDD");   

_LIT8(KRTPPacket2, "8001CDE000000004DEADBEEF3958B1FB3F0E");

#define RTP_EUNIT_ASSERT_EQUALS( func, val2 ) {\

TInt unittestRetVal = func;\

if ( val2 != KErrNoMemory && unittestRetVal == KErrNoMemory ){\

	User::Leave( KErrNoMemory );\

	}\

else{\

	EUNIT_ASSERT_EQUALS( unittestRetVal, val2 );\

	}}

#define RTP_EUNIT_ASSERT_RTP_ID_ON

#ifdef RTP_EUNIT_ASSERT_RTP_ID_ON

#define RTP_EUNIT_ASSERT_RTP_ID( val1 ) \

if ( val1 == KNullId )\

    {\

    User::Leave( KErrNoMemory );\

    }

#else

#define RTP_EUNIT_ASSERT_RTP_ID( val1 )

EUNIT_ASSERT( val1 != KNullId );

#endif

// CONSTRUCTION

UT_CRtpSession* UT_CRtpSession::NewL()

    {

    UT_CRtpSession* self = UT_CRtpSession::NewLC();

    CleanupStack::Pop();

    return self;

    }

UT_CRtpSession* UT_CRtpSession::NewLC()

    {

    UT_CRtpSession* self = new( ELeave ) UT_CRtpSession();

    CleanupStack::PushL( self );

    self->ConstructL();

    return self;

    }

// Destructor (virtual by CBase)

UT_CRtpSession::~UT_CRtpSession()

    {

    }

// Default constructor

UT_CRtpSession::UT_CRtpSession()

    {

    }

// Second phase construct

void UT_CRtpSession::ConstructL()

    {

    // The ConstructL from the base class CEUnitTestSuiteClass must be called.

    // It generates the test case table.

    CEUnitTestSuiteClass::ConstructL();

    }

//  METHODS

void UT_CRtpSession::SetupL(  )

    {

    iPacketsExpected = 0;

    TCreateSessionParams sessionParams;

    TUint port( 5000 );

    TRtcpParams rtcpParams;

    rtcpParams.iRtcpFraction = 0.10;

    rtcpParams.iSessionBWidth = 65000;

    TInt err( KErrNone );

    for ( TUint k = 0; k < KRtpNumPayloadTypes; k++ )

        {

        iProfileRTPTimeRates[k] = ( TUint32 ) KRtpPayloadClockConversions[k];

        }

    TBuf8<3> cName; 

    TBuf8<3> userName;

    TBuf8<3> email;

    TBuf8<3> phoneNumber;

    TBuf8<3> location;

    TBuf8<3> swToolName;

    TBuf8<3> noticeStatus;

    TBuf8<3> privateStr;

    // Create SDES

    cName.Format( _L8( "ab" ) );

    cName.ZeroTerminate();

    userName.Format( _L8( "bc" ) );

    userName.ZeroTerminate();

    email.Format( _L8( "cd" ) );

    email.ZeroTerminate();

    phoneNumber.Format( _L8( "de" ) );

    phoneNumber.ZeroTerminate();

    location.Format( _L8( "ef" ) );

    location.ZeroTerminate();

    swToolName.Format( _L8( "fg" ) );

    swToolName.ZeroTerminate();

    noticeStatus.Format( _L8( "gh" ) );

    noticeStatus.ZeroTerminate();

    privateStr.Format( _L8( "hi" ) );

    privateStr.ZeroTerminate();

    TRtpSdesParams sdes;

    sdes.iCName.Set( cName.Ptr() );

    sdes.iUserName.Set( userName.Ptr() );

    sdes.iEmail.Set( email.Ptr() );

    sdes.iPhoneNumber.Set( phoneNumber.Ptr() );

    sdes.iLocation.Set( location.Ptr() );

    sdes.iSwToolName.Set( swToolName.Ptr() );

    sdes.iNoticeStatus.Set( noticeStatus.Ptr() );

    sdes.iPrivate.Set( privateStr.Ptr() );

    // Set it

    iSdes = CRtpSDES::NewL( sdes );

    // Opens a session to the socket server

    // i.e. creates an IPC communication channel to the socket server

    if ( ( err = iSocketServ.Connect() ) != KErrNone )

        {

        //"CRtpUtRtpCommSend::ConstructL()") );

        //"RSocketServ - Open Connection Error

        User::Leave( KErrGeneral );

        }

    TCommDbConnPref prefs;

    prefs.SetDirection( ECommDbConnectionDirectionOutgoing );

    prefs.SetDialogPreference( ECommDbDialogPrefDoNotPrompt );    

    prefs.SetIapId ( KDefaultIap );//6

    if ( ( err = iConn.Open( iSocketServ ) ) != KErrNone )

        {

        //"CRtpUtRtpCommSend::ConstructL()") );

        //"RConnection - Start error %d"), err );

        User::Leave( KErrGeneral );

        }

    if ( ( err = iConn.Start( prefs ) ) != KErrNone )

        {

        ////"CRtpUtRtpCommSend::ConstructL()") );

        ////"RConnection - Start error %d"), err );

        User::Leave( KErrGeneral );

        }

    rtcpParams.iSessionBWidth = 6000; // less than the minimum of 8000

    rtcpParams.iRtcpFraction = 1.5; // more than the maximum of 1.0

    // Create a session with RTCP and bad parameters -> default values in use

    iSession = CRtpSession::NewL( sessionParams,

                                  port,

                                  ETrue, // enable RTCP

                                  &rtcpParams, // Rtcp params

                                  iSocketServ,

                                  iConn,

                                  1,

                                  iSdes,

                                  iProfileRTPTimeRates,

                                  ETrue,

                                  iLibrary,

                                  *this,

                                  *this

                                );

    if ( iSession->iNumOfRxStreams != 0 ||

         iSession->iNumOfTxStreams != 0 ||

         iSession->iSendSdesCounter != KSDES_SEND_PACKET ||

         iSession->iAverageRtcpSize != KAverageRtcpPacketLength ||

         iSession->iTotalParticipantsSession != 1 ||

         iSession->iRtcpTimeInterval == 0 ||

         iSession->iBandWidth != KRtpDefaultBandWidth ||

         iSession->iFraction != KRtpDefaultRtcpFrac ||

         iSession->iDefaultSSRC == 0 ||

         iSession->iRtpObserver != 0 ||

         iSession->iRtcpObserver != 0 

        )

        {

        //"Parameters wrongly updated during construction(2)" ) );

        //"CRtpSession::NewL() - FAILED" ) );

        User::Leave( KErrGeneral );

        }

    // Test the random SSRC generation, now that we have a few SSRC values

    // An SSRC collision here is improbable, though possible

    iTranstreamId=1 ;

    iRecvstreamId=2;

    TRtpSSRC ssrcOut( 1234 );

    TRtpSSRC ssrcOut2(5555);

    TRtpSSRC ssrcIn( 4321 );

    TRtpPayloadType payloadType( 0 );

    TInt error(KErrNone);

    error =iSession->CreateTransmitStreamL( iTranstreamId, payloadType, ssrcOut );

	EUNIT_ASSERT(error==KErrNone);

	error = iSession->CreateReceiveStreamL( iRecvstreamId, payloadType );   

	EUNIT_ASSERT(error==KErrNone);

	error = iSession->CreateTransmitStreamExtL(iTranstreamId, payloadType, ssrcOut2);

	TInetAddr remoteAddr( INET_ADDR( 127, 0, 0, 1 ), 5000 );

    error = iSession->SetRemoteAddress( remoteAddr );    

    }

void UT_CRtpSession::Setup2L(  )

    {

    TPtrC16 fileName( _L16( "rtpstppacket.dll" ) );

  	User::LeaveIfError( iLibrary.Load( fileName ) );

  	SetupL();

    }

void UT_CRtpSession::Teardown(  )

    {

    if ( iSession  )

        {

        if ( iRollBackRtcp )

            {

            iSession->iRtcpEnabled = iRollBackRtcpValue;

            }

        if (iSession->iPktExtSnd )

            {

            iSession->iPktExtSnd->Close();

            iSession->iPktExtSnd = NULL;

            }

        }

    delete iSession;

    iSession = NULL;

    delete iSdes;

    iSdes = NULL;

    iConn.Close();

    iSocketServ.Close();

    iLibrary.Close();

    iRollBackRtcp = EFalse;

    }

void UT_CRtpSession::Hex(HBufC8& aString)

{

    TPtr8 ptr=aString.Des();

    TInt length = aString.Length();

    if (aString.Length()%2)

        {

        ptr.SetLength(0);

        return;

        }

    TInt i;

    for (i=0;i<aString.Length();i+=2)

        {

        TUint8 tmp;

        tmp=(TUint8)(aString[i]-(aString[i]>'9'?('A'-10):'0'));

        tmp*=16;

        tmp|=(TUint8)(aString[i+1]-(aString[i+1]>'9'?('A'-10):'0'));

        ptr[i/2]=tmp;

        }

    ptr.SetLength(aString.Length()/2);

}

void UT_CRtpSession::UT_CRtpSession_NewLL(  )

    {

    TCreateSessionParams sessionParams;

    TUint port( 5000 );

    TRtcpParams rtcpParams;

    rtcpParams.iRtcpFraction = 0.10;

    rtcpParams.iSessionBWidth = 65000;

    TRtpSSRC ssrc1;

    TRtpSSRC ssrc2;

    TRtpSSRC ssrc3;

 	ssrc3 = iSession->SSRC(); // store the random SSRC for the test below

	delete iSession;

	iSession = NULL;

    // Create a session without RTCP

    iSession = CRtpSession::NewL( sessionParams,

                                  port,

                                  EFalse, // disable RTCP

                                  NULL, // Rtcp params

                                  iSocketServ,

                                  iConn,

                                  1,

                                  iSdes,

                                  iProfileRTPTimeRates,

                                  ETrue,

                                  iLibrary,

                                  *this,

                                  *this);

    ssrc1 = iSession->SSRC(); // store the random SSRC for the test below

    delete iSession;

    iSession = NULL;

    // Create a session with RTCP and good parameters

    iSession = CRtpSession::NewL( sessionParams,

                                  port,

                                  ETrue, // enable RTCP

                                  &rtcpParams, // Rtcp params

                                  iSocketServ,

                                  iConn,

                                  1,

                                  iSdes,

                                  iProfileRTPTimeRates,

                                  ETrue,

                                  iLibrary,

                                  *this,

                                  *this

                                );

    if ( iSession->iNumOfRxStreams != 0 ||

         iSession->iNumOfTxStreams != 0 ||

         iSession->iSendSdesCounter != KSDES_SEND_PACKET ||

         iSession->iAverageRtcpSize != KAverageRtcpPacketLength ||

         iSession->iTotalParticipantsSession != 1 ||

         iSession->iRtcpTimeInterval == 0 ||

         iSession->iBandWidth != rtcpParams.iSessionBWidth ||

         iSession->iFraction != rtcpParams.iRtcpFraction ||

         iSession->iDefaultSSRC == 0 ||

         iSession->iRtpObserver != 0 ||

         iSession->iRtcpObserver != 0 

        )

        {

        //"Parameters wrongly updated during construction (1) " ) );

        //"CRtpSession::NewL() - FAILED" ) );

        User::Leave( KErrGeneral );

        }

    ssrc2 = iSession->SSRC(); // store the random SSRC for the test below

    iSession->iSdesArray->ResetAndDestroy();

    if ( ssrc1 == ssrc2 ||

         ssrc1 == ssrc3 ||

         ssrc2 == ssrc3 )

        {

        // _L( "SSRC collision" ) );

        EUNIT_ASSERT(EFalse); 

        }

    // Create a session with non-default socket buf size

    delete iSession;

    iSession = NULL;

    sessionParams.iSocketBufSize = 10000;

    iSession = CRtpSession::NewL( sessionParams,

                                  port,

                                  ETrue, // enable RTCP

                                  &rtcpParams, // Rtcp params

                                  iSocketServ,

                                  iConn,

                                  1,

                                  iSdes,

                                  iProfileRTPTimeRates,

                                  ETrue,

                                  iLibrary,

                                  *this,

                                  *this );

    EUNIT_ASSERT_EQUALS( iSession->iCommNet->MaxSocketSize(), 10000 );                           

    }

void UT_CRtpSession::UT_CRtpSession_SSRCL(  )

    {

    EUNIT_ASSERT( ETrue );

    //Tested in SetUp

    }

void UT_CRtpSession::UT_CRtpSession_StartSessionL(  )

    {

    TInt error(KErrNone);

    TInt tempNumOfTxStreams( iSession->iNumOfTxStreams );

    TInt tempNumOfRxStreams( iSession->iNumOfRxStreams );

    iSession->iNumOfTxStreams = 0;

    iSession->iNumOfRxStreams = 0;

    // With the stream amounts set to 0, the function call should do nothing

    error = iSession->StartSession();

    if ( error != KErrNotReady )

        {

        EUNIT_ASSERT(EFalse);

        }

    // Restore the previous Rx value

    iSession->iNumOfRxStreams = tempNumOfRxStreams;

    // This should start the RTP & RTCP reception

    error = iSession->StartSession();

    if ( error != KErrNone || !iSession->iSessionStarted )

        {

        EUNIT_ASSERT(EFalse);

        }

    // Restore the previous Tx value

    iSession->iNumOfTxStreams = tempNumOfTxStreams;

    // This should start RTCP transmission

    error = iSession->StartSession();

    if ( error != KErrNone || !iSession->iSessionStarted )

        {

        EUNIT_ASSERT(EFalse);

        }

    iSession->iRtcpEnabled = EFalse; // temporarily

    // This should cover the rest of the branches

    error = iSession->StartSession();

    if ( error != KErrNone || !iSession->iSessionStarted )

        {

        EUNIT_ASSERT(EFalse);

        }

    iSession->iRtcpEnabled = ETrue; // restore

    }

void UT_CRtpSession::UT_CRtpSession_SetRemoteAddressL(  )

    {

    TInt error( KErrNone );

    TInetAddr remoteAddr( INET_ADDR( 127, 0, 0, 1 ), 5000 );

    // Try setting the address when the communications are down

    CRtpComm* tempComm = iSession->iCommNet;

    iSession->iCommNet = NULL;

    // Run the function

    error = iSession->SetRemoteAddress( remoteAddr );

    if ( error != KErrGeneral )

        {     

        EUNIT_ASSERT(EFalse);

        }

    // Restore communications

    iSession->iCommNet = tempComm;

    // Try again. This should work

    error = iSession->SetRemoteAddress( remoteAddr );

    if ( error != KErrNone )

        {

        EUNIT_ASSERT(EFalse);

        }

    // Check that the address and the session parameters are really updated

    TInetAddr tempAddr;    

    if (! iSession->iRemoteAddr[ERTPPort].Match(remoteAddr))

        {

        EUNIT_ASSERT(EFalse);

        } 

    // Flip the session started status to reach more branches

    iSession->iSessionStarted = !iSession->iSessionStarted;

    error = iSession->SetRemoteAddress( remoteAddr );