/*

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

*

*/

#ifndef __UT_CSRTPSTREAMOUT_H__

#define __UT_CSRTPSTREAMOUT_H__

//  INTERNAL INCLUDES

#include <digia/eunit/ceunittestsuiteclass.h>

//  EXTERNAL INCLUDES

#include "srtpmastersalt.h"

#include "srtpmasterkey.h"

#include "srtpcryptocontext.h"

#include <in_sock.h>

#include "msrtprekeyingobserver.h"

//  FORWARD DECLARATIONS

class CSRTPCryptoHandlerSRTP;

class CSRTPCryptoHandlerSRTCP;

class CSRTPSession;

class CSRTPStreamOut;

//  CLASS DEFINITION

/**

 *

 * EUnitWizard generated test class. 

 *

 */

class UT_CSRTPStreamOut

     : public CEUnitTestSuiteClass,

     public MSRTPReKeyingObserver     

    {

    class TPayloadDetails

        {

        public:    

            TPayloadDetails();

        public:

            TUint16 seqNr;

          TUint16 ssrc;

            TUint8  padding;

            TUint8  numCSRC;

            TBool   fHeaderExtension;

            TUint8  payloadLen;

            const TUint8* payloadData;

            TUint8  MKILen;

            const TUint8* MKIData;

            TUint8  authTagLen;

            const TUint8* authTagData;    

        };

    public:     // Constructors and destructors

        /**

         * Two phase construction

         */

        static UT_CSRTPStreamOut* NewL();

        static UT_CSRTPStreamOut* NewLC();

        /**

         * Destructor

         */

        ~UT_CSRTPStreamOut();

    private:    // Constructors and destructors

        UT_CSRTPStreamOut();

        void ConstructL();

	public: 	// From observer interface

        void SRTPMasterKeyStaleEvent(const CSRTPStream& aStream); 

        void SRTPMasterKeyStaleEvent(const CSRTPSession& aSession); 	

    private:    // New methods

         void SetupL();

         void Teardown();

         void UT_CSRTPStreamOut_NewL_1L();

         void UT_CSRTPStreamOut_NewL_2L();

         void UT_CSRTPStreamOut_NewL_3L(  );

         void UT_CSRTPStreamOut_SetCryptoOutLL(  );

         void UT_CSRTPStreamOut_ProtectRtpLL();

         void UT_CSRTPStreamOut_ProtectRtcpLL();

         void UT_CSRTPStreamOut_StateTestsL();

         HBufC8* BuildPacketL(TBool aSrtp, TPayloadDetails& details);    

         void Hex(HBufC8& aString);            

         void UT_CSRTPStreamOut_ProtectRtpL2L(); 

    private:    // Data

        CSRTPMasterSalt* iMasterSalt; 

        CSRTPMasterKey*  iMasterKey;         

        CSRTPCryptoContext* iContext;

        HBufC8* iTestPayload160Bits;

        HBufC8* iTestMKI128Bits;

        HBufC8* iTestAuthTag80Bits;

        CSRTPSession* iSRTPSession;

        CSRTPStreamOut* iStreamOut;

        CSRTPCryptoHandlerSRTP *iCryptoHandlerSRTP;

        CSRTPCryptoHandlerSRTCP *iCryptoHandlerSRTCP;

        TInetAddr iDestination;

	    HBufC8* iRFC3711_SessionEncrKey128bits;

	    HBufC8* iRFC3711_SessionSaltKey128bits;

	    HBufC8* iRFC3711_SessionAuthKey128bits;

        HBufC8* iDecryptedPayload; 

	    HBufC8* iRFC3711_RtcpEncrKey128bits;

	    HBufC8* iRFC3711_RtcpSaltKey128bits;

	    HBufC8* iRFC3711_RtcpAuthKey128bits;

		EUNIT_DECLARE_TEST_TABLE; 

    };

#endif      //  __UT_CSRTPSTREAMOUT_H__

// End of file