/*

* 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_CSRTPStreamIn.h"

#include "srtputils.h"

// CONSTANTS

const TInt KRtpVersion = 2;    // Current protocol version          

const TUint KMinRtpHeaderLen( 12 );

const TUint16 KTestSeqNr17 = 17171;

const TUint16 KTestSeqNr18 = 17172;

const TInt KLength = 50;

// 80 bit key, test 1

_LIT8(KTestPayload160Bits,  "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA");

// 80 bit key, test 1

//_LIT8(KTestMKI128Bits,      "BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB");

_LIT8(KTestMKI128Bits,      "ABCDEF1234567890ABCDEF1234567890");

// 80 bit key, test 1

_LIT8(KTestAuthTag80Bits,   "CCCCCCCCCCCCCCCCCCCC");

_LIT8(KRFC3711_TestMasterKey128bits,    "E1F97A0D3E018BE0D64FA32C06DE4139");

_LIT8(KRFC3711_TestMasterSalt112bits,   "0EC675AD498AFEEBB6960B3AABE6");	

_LIT8(KRFC3711_SessionEncrKey128bits,   "C61E7A93744F39EE10734AFE3FF7A087");  

_LIT8(KRFC3711_SessionSaltKey128bits,   "30CBBC08863D8C85D49DB34A9AE1");  

_LIT8(KRFC3711_SessionAuthKey128bits,   "CEBE321F6FF7716B6FD4AB49AF256A156D38BAA4");  

_LIT8(KRFC3711_RtcpEncrKey128bits,   	"4C1AA45A81F73D61C800BBB00FBB1EAA");  

_LIT8(KRFC3711_RtcpSaltKey128bits,   	"9581C7AD87B3E530BF3E4454A8B3");  

_LIT8(KRFC3711_RtcpAuthKey128bits,   	"8D54534FEB49AE8E7993A6BD0B844FC323A93DFD");  

_LIT8(KDecryptedPayload,                "2EDE3286C6E33E68C6BB93350620A1245EC0004F");  

_LIT8(KTestM_without_ROC,               "123456789ABCDE");  

_LIT8(KRefM_with_ROC,                   "123456789ABCDEFFFFFFFF");  

_LIT8(KTestAuthTag,                     "D07919ABC9F235CB4516");  

_LIT8(KDummyRTCPPacket, "dummyrtcp");

_LIT8(KRTCPPacket, "80C90001DEADBEEF9E30159059A23363F0FBDB62DC85AF52875C944495C5B00280000001ABCDEF1234567890ABCDEF1234567890DFC98F01DE16DA30F20D");

_LIT8(KSRTCPPacket2, "80C90001DEADBEEF2BC095F51495D87D35B816BE569A6A1BA6989D5A20B541D680000002ABCDEF1234567890ABCDEF12345678905140BA93E8CDAC0C89D5");

//  EXTERNAL INCLUDES

#include <digia/eunit/eunitmacros.h>

#include "srtpcryptohandlersrtp.h"

#include "srtpcryptohandlersrtcp.h"

#include "srtpmastersalt.h"

#include "srtpmasterkey.h"

#include "srtpstreamin.h"

#include "srtpsession.h"

#include "srtputils.h"

//  INTERNAL INCLUDES

#include "srtpstreamin.h"

// CONSTRUCTION

UT_CSRTPStreamIn* UT_CSRTPStreamIn::NewL()

    {

    UT_CSRTPStreamIn* self = UT_CSRTPStreamIn::NewLC(); 

    CleanupStack::Pop();

    return self;

    }

UT_CSRTPStreamIn* UT_CSRTPStreamIn::NewLC()

    {

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

    CleanupStack::PushL( self );

	self->ConstructL(); 

    return self;

    }

// Destructor (virtual by CBase)

UT_CSRTPStreamIn::~UT_CSRTPStreamIn()

    {

    }

// Default constructor

UT_CSRTPStreamIn::UT_CSRTPStreamIn() 

    {

    }

UT_CSRTPStreamIn::TPayloadDetails::TPayloadDetails() :

        seqNr(),

//      ssrc(),

        padding(0),

        numCSRC(0),

        fHeaderExtension(EFalse),

        payloadLen(0),

        payloadData(NULL),

        MKILen(0),

        MKIData(NULL),

        authTagLen(0),

        authTagData(NULL)

            {            

            }

void UT_CSRTPStreamIn::SRTPMasterKeyStaleEvent(const CSRTPStream& /*aStream*/ )

    {

    }

void UT_CSRTPStreamIn::SRTPMasterKeyStaleEvent(const CSRTPSession& /*aSession*/ )

    {

    }

// Second phase construct

void UT_CSRTPStreamIn::ConstructL()

    {

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

    // It generates the test case table.

    CEUnitTestSuiteClass::ConstructL();

    }

//  METHODS

void UT_CSRTPStreamIn::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);

}

// ---------------------------------------------------------------------------

// UT_CSRTPStreamIn::BuildSrtpPacketL()

// 

// ---------------------------------------------------------------------------

//

HBufC8* UT_CSRTPStreamIn::BuildPacketL(TBool aSrtp, TPayloadDetails& details)

    {

    TUint packetSize = KMinRtpHeaderLen + details.payloadLen;

    TUint payloadStartOffset = KMinRtpHeaderLen;

    TUint8 padCount = 0;

    packetSize += details.numCSRC*4;

    payloadStartOffset += details.numCSRC*4;          

    if (details.fHeaderExtension) 

        {

        packetSize += 4;        

        payloadStartOffset += 4;

        }

    if (details.padding) 

        {

        padCount = 3;

        }

    packetSize += padCount;            

    if (aSrtp)

        {

        packetSize += details.MKILen;                    

        packetSize += details.authTagLen;                        

        }

    HBufC8* packet = HBufC8::NewMaxL(packetSize);

    TUint8* dataP = const_cast<TUint8*>( packet->Des().Ptr() );    

    TUint8* ptr = dataP;

    Mem::FillZ( ptr, packetSize);

    // padding (1 bit)

    ptr[0] |= static_cast<TUint8>( details.padding << 5 );

    // header extension (1 bit)

    if ( details.fHeaderExtension )

        {

        ptr[0] |= ( 1 << 4 );

        }

    if (details.numCSRC)

        {

        // CC

        ptr[0] |= details.numCSRC;

        }

    //sequence number

    ptr+=2;

    TSRTPUtils::Write16( ptr, details.seqNr );    

    //ssrc   

    /*

    ptr+=6;

    if (details.ssrc)

        {

        Mem::Copy( ptr, details.ssrc, 4 );        

        }

   */

    //payload

    ptr = dataP + payloadStartOffset;

    if (details.payloadData)

        {

        Mem::Copy( ptr, details.payloadData, details.payloadLen );        

        }

    if (aSrtp)

        {

        //MKI

        ptr = dataP + payloadStartOffset + details.payloadLen + padCount;

        if (details.MKILen > 0)

            {

            Mem::Copy( ptr, details.MKIData, details.MKILen );        

            }

        ptr = dataP + payloadStartOffset + details.payloadLen + padCount + details.MKILen;    

    	if (details.authTagLen==14)

    		{

    		details.authTagLen=10;

    		TSRTPUtils::Write32( ptr,0);

  			ptr+=4;  	

    		}

        //auth tag

        if (details.authTagLen > 0)

            {

            Mem::Copy( ptr, details.authTagData, details.authTagLen );        

            }            

        }    

    return packet;

    }

void UT_CSRTPStreamIn::SetupL(  )

    {

    iDecryptedPayload = HBufC8::NewL(KDecryptedPayload().Length());

	*iDecryptedPayload = KDecryptedPayload;

	Hex(*iDecryptedPayload);

	iDecryptedPayload->Des().SetLength(20);

    iRFC3711_SessionEncrKey128bits = HBufC8::NewL(KRFC3711_SessionEncrKey128bits().Length());

	*iRFC3711_SessionEncrKey128bits = KRFC3711_SessionEncrKey128bits;

	Hex(*iRFC3711_SessionEncrKey128bits);

    iRFC3711_SessionSaltKey128bits = HBufC8::NewL(KRFC3711_SessionSaltKey128bits().Length());

	*iRFC3711_SessionSaltKey128bits = KRFC3711_SessionSaltKey128bits;

	Hex(*iRFC3711_SessionSaltKey128bits);

    iRFC3711_SessionAuthKey128bits = HBufC8::NewL(KRFC3711_SessionAuthKey128bits().Length());

	*iRFC3711_SessionAuthKey128bits = KRFC3711_SessionAuthKey128bits;

	Hex(*iRFC3711_SessionAuthKey128bits);

    iRFC3711_RtcpEncrKey128bits = HBufC8::NewL(KRFC3711_RtcpEncrKey128bits().Length());

	*iRFC3711_RtcpEncrKey128bits = KRFC3711_RtcpEncrKey128bits;

	Hex(*iRFC3711_RtcpEncrKey128bits);

    iRFC3711_RtcpSaltKey128bits = HBufC8::NewL(KRFC3711_RtcpSaltKey128bits().Length());

	*iRFC3711_RtcpSaltKey128bits = KRFC3711_RtcpSaltKey128bits;

	Hex(*iRFC3711_RtcpSaltKey128bits);

    iRFC3711_RtcpAuthKey128bits = HBufC8::NewL(KRFC3711_RtcpAuthKey128bits().Length());

	*iRFC3711_RtcpAuthKey128bits = KRFC3711_RtcpAuthKey128bits;

	Hex(*iRFC3711_RtcpAuthKey128bits);

	iTestPayload160Bits = HBufC8::NewL(KTestPayload160Bits().Length());

	*iTestPayload160Bits = KTestPayload160Bits;	

	Hex(*iTestPayload160Bits);	

    // our test MKI is 16 bit long:

    // ABCDEF1234567890ABCDEF1234567890

    HBufC8* masterKey = HBufC8::NewL(KRFC3711_TestMasterKey128bits().Length());

    CleanupStack::PushL( masterKey );

    *masterKey = KRFC3711_TestMasterKey128bits;

    HBufC8* masterSalt =HBufC8::NewL(KRFC3711_TestMasterSalt112bits().Length());

    CleanupStack::PushL( masterSalt);

	*masterSalt = KRFC3711_TestMasterSalt112bits;

	HBufC8* mki =HBufC8::NewL(KTestMKI128Bits().Length());

    CleanupStack::PushL( mki);

	*mki = KTestMKI128Bits;

	Hex(*mki);

	iMasterKey = CSRTPMasterKey::NewL( *masterKey, *mki );

    iMasterSalt = CSRTPMasterSalt::NewL( *masterSalt );

    TSrtpCryptoParams params;

    iContext = CSRTPCryptoContext::NewL(iMasterKey, iMasterSalt, params );

    CleanupStack::PopAndDestroy( mki );

    CleanupStack::PopAndDestroy( masterSalt );

    CleanupStack::PopAndDestroy( masterKey ); 

    iDestination.Input(_L("127.0.0.1"));

    iSRTPSession = CSRTPSession::NewL( iDestination, iContext, *this ); 

    iStreamIn = CSRTPStreamIn::NewL(*iSRTPSession, (TUint)1);

    iStreamInLateBD = CSRTPStreamIn::NewL(*iSRTPSession);

	iCryptoHandlerSRTP = iStreamIn->iHandlerRTP;

	iCryptoHandlerSRTCP =iStreamIn->iHandlerRTCP;

	iCryptoHandlerSRTP2 = iStreamInLateBD->iHandlerRTP;

	iCryptoHandlerSRTCP2 =iStreamInLateBD->iHandlerRTCP;

    } 

void UT_CSRTPStreamIn::Teardown(  )

    {

    delete iDecryptedPayload;

    delete iTestPayload160Bits;

    delete iStreamIn;

    delete iStreamInLateBD;

    delete iSRTPSession;    

    delete iRFC3711_SessionEncrKey128bits;

    delete iRFC3711_SessionSaltKey128bits;

    delete iRFC3711_SessionAuthKey128bits;    

    delete iRFC3711_RtcpEncrKey128bits;

	delete iRFC3711_RtcpSaltKey128bits;

	delete iRFC3711_RtcpAuthKey128bits;

    }

    {        

    CSRTPStreamIn* tempStreamIn = CSRTPStreamIn::NewL(*iSRTPSession, (TUint)1);

    EUNIT_ASSERT(tempStreamIn->SSRC()== 1);

    delete tempStreamIn;

    }

    {

    HBufC8* masterKey = HBufC8::NewL(KRFC3711_TestMasterKey128bits().Length());

    CleanupStack::PushL( masterKey );

    *masterKey = KRFC3711_TestMasterKey128bits;

    HBufC8* masterSalt =HBufC8::NewL(KRFC3711_TestMasterSalt112bits().Length());

    CleanupStack::PushL( masterSalt);

	*masterSalt = KRFC3711_TestMasterSalt112bits;

	HBufC8* mki =HBufC8::NewL(KTestMKI128Bits().Length());

    CleanupStack::PushL( mki);

	*mki = KRFC3711_TestMasterSalt112bits;

	Hex(*mki);

	TSrtpCryptoParams params;

	CSRTPMasterKey* masterkey = CSRTPMasterKey::NewLC( *masterKey, *mki );

    CSRTPMasterSalt* mastersalt = CSRTPMasterSalt::NewLC( *masterSalt );

    CSRTPCryptoContext* context = CSRTPCryptoContext::NewL(masterkey, mastersalt, params );

    CleanupStack::Pop( mastersalt );

    CleanupStack::Pop( masterkey );

    CleanupStack::PopAndDestroy( mki );

    CleanupStack::PopAndDestroy( masterSalt );

    CleanupStack::PopAndDestroy( masterKey ); 

    CleanupStack::PushL( context );

    CSRTPSession* srtpSession = CSRTPSession::NewL( iDestination, context, *this );     

    CleanupStack::Pop( context ); 

    CleanupStack::PushL( srtpSession );                      

    CSRTPStreamIn* tempStreamIn =CSRTPStreamIn::NewL( *srtpSession,

                                      (TUint)1,

                                      *this );

    EUNIT_ASSERT(tempStreamIn->SSRC()== 1);

    delete tempStreamIn;    

     CleanupStack::PopAndDestroy( srtpSession );                   

    }

    {

    HBufC8* masterKey = HBufC8::NewL(KRFC3711_TestMasterKey128bits().Length());

    CleanupStack::PushL( masterKey );

    *masterKey = KRFC3711_TestMasterKey128bits;

    HBufC8* masterSalt =HBufC8::NewL(KRFC3711_TestMasterSalt112bits().Length());

    CleanupStack::PushL( masterSalt);

	*masterSalt = KRFC3711_TestMasterSalt112bits;

	HBufC8* mki =HBufC8::NewL(KTestMKI128Bits().Length());

    CleanupStack::PushL( mki);

	*mki = KRFC3711_TestMasterSalt112bits;

	Hex(*mki);

	TSrtpCryptoParams params;

	CSRTPMasterKey* masterkey = CSRTPMasterKey::NewLC( *masterKey, *mki );

    CSRTPMasterSalt* mastersalt = CSRTPMasterSalt::NewLC( *masterSalt );

    CSRTPCryptoContext* context = CSRTPCryptoContext::NewL(masterkey, mastersalt, params );

    CleanupStack::Pop( mastersalt );

    CleanupStack::Pop( masterkey );

    CleanupStack::PopAndDestroy( mki );

    CleanupStack::PopAndDestroy( masterSalt );

    CleanupStack::PopAndDestroy( masterKey ); 

    CleanupStack::PushL( context );

    CSRTPSession* srtpSession = CSRTPSession::NewL( iDestination, context, *this ); 

    CleanupStack::Pop( context );

    CleanupStack::PushL( srtpSession );

    CSRTPStreamIn* tempStreamIn =CSRTPStreamIn::NewL( *srtpSession);

    EUNIT_ASSERT(tempStreamIn->GetCryptoContext().Valid()==ETrue); 

    delete tempStreamIn;   

    CleanupStack::PopAndDestroy( srtpSession );                   

    }

    {

     HBufC8* masterKey = HBufC8::NewL(KRFC3711_TestMasterKey128bits().Length());

    CleanupStack::PushL( masterKey );

    *masterKey = KRFC3711_TestMasterKey128bits;

    HBufC8* masterSalt =HBufC8::NewL(KRFC3711_TestMasterSalt112bits().Length());

    CleanupStack::PushL( masterSalt);

	*masterSalt = KRFC3711_TestMasterSalt112bits;

	HBufC8* mki =HBufC8::NewL(KTestMKI128Bits().Length());

    CleanupStack::PushL( mki);

	*mki = KRFC3711_TestMasterSalt112bits;

	Hex(*mki);

	TSrtpCryptoParams params;

	CSRTPMasterKey* masterkey = CSRTPMasterKey::NewLC( *masterKey, *mki );

    CSRTPMasterSalt* mastersalt = CSRTPMasterSalt::NewLC( *masterSalt );

    CSRTPCryptoContext* context = CSRTPCryptoContext::NewL(masterkey, mastersalt, params );

    CleanupStack::Pop( mastersalt );

    CleanupStack::Pop( masterkey );

    CleanupStack::PopAndDestroy( mki );

    CleanupStack::PopAndDestroy( masterSalt );

    CleanupStack::PopAndDestroy( masterKey ); 

    CleanupStack::PushL( context );

    CSRTPSession* srtpSession = CSRTPSession::NewL( iDestination, context, *this ); 

    CleanupStack::Pop( context );

    CleanupStack::PushL( srtpSession );

    CSRTPStreamIn* tempStreamIn =CSRTPStreamIn::NewL( *srtpSession, *this );

    EUNIT_ASSERT(tempStreamIn->GetCryptoContext().Valid()==ETrue); 

    delete tempStreamIn;  

    CleanupStack::PopAndDestroy( srtpSession );

    }

    {

     HBufC8* masterKey = HBufC8::NewL(KRFC3711_TestMasterKey128bits().Length());

    CleanupStack::PushL( masterKey );

    *masterKey = KRFC3711_TestMasterKey128bits;

    HBufC8* masterSalt =HBufC8::NewL(KRFC3711_TestMasterSalt112bits().Length());

    CleanupStack::PushL( masterSalt);

	*masterSalt = KRFC3711_TestMasterSalt112bits;

	HBufC8* mki =HBufC8::NewL(KTestMKI128Bits().Length());

    CleanupStack::PushL( mki);

	*mki = KRFC3711_TestMasterSalt112bits;

	Hex(*mki);

	TSrtpCryptoParams params;

	CSRTPMasterKey* masterkey = CSRTPMasterKey::NewLC( *masterKey, *mki );

    CSRTPMasterSalt* mastersalt = CSRTPMasterSalt::NewLC( *masterSalt );

    CSRTPCryptoContext* context = CSRTPCryptoContext::NewL(masterkey, mastersalt, params );

    CleanupStack::Pop( mastersalt );

    CleanupStack::Pop( masterkey );

    CleanupStack::PopAndDestroy( mki );

    CleanupStack::PopAndDestroy( masterSalt );

    CleanupStack::PopAndDestroy( masterKey ); 

    CleanupStack::PushL( context );

    CSRTPSession* srtpSession = CSRTPSession::NewL( iDestination); 

    CleanupStack::PushL( srtpSession );

    CSRTPStreamIn* tempStreamIn =CSRTPStreamIn::NewL( *srtpSession, context, *this );