/*

* Copyright (c) 2005 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:    implementation

*

*/

#include "tsrtpstreamoutstateuninit.h"

#include "msrtpstreamoutcontext.h"

#include "msrtpcryptohandlercontext.h"

#include "msrtpcryptohandlercontextrtp.h"

#include "msrtpcryptohandlercontextrtcp.h"

#include "srtputils.h"

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

// TSRTPStreamOutStateUninit::TSRTPStreamOutStateUninit

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

//

TSRTPStreamOutStateUninit::TSRTPStreamOutStateUninit(

    MSRTPStreamOutContext& aStreamContext,

    MSRTPCryptoHandlerContextRTP& aCryptoHandlerRTPContext,

    MSRTPCryptoHandlerContextRTCP& aCryptoHandlerRTCPContext)

    : TSRTPStreamOutStateBase(aStreamContext,

                            aCryptoHandlerRTPContext, 

                            aCryptoHandlerRTCPContext)

    {

    }

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

// TSRTPStreamOutStateUninit::DoProtectRtpL

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

//		

HBufC8* TSRTPStreamOutStateUninit::DoProtectRtpL(const TDesC8& aPacket)

    {

    SRTP_DEBUG_DETAIL( "TSRTPStreamOutStateUninit::DoProtectRtpL entry" );

    SRTP_DEBUG_TINT_VALUE( "PacketLength is", aPacket.Length());

    // first initialize current RTP packet..

    iCryptoHandlerRTPContext.InitializePlainPacketL(aPacket);    

    // Step 2 (in RFC 3711, section 3.3):

    // count the packet index i

     iCryptoHandlerRTPContext.CountSenderPacketIndexL();

     SRTP_DEBUG_DETAIL( "StreamOutStateUninit, Protect, DeriveSessionKeysL" );

     // derive session keys..

     iCryptoHandlerRTPContext.DeriveSessionKeysL();

     SRTP_DEBUG_DETAIL( "StreamOutStateUninit, Protect, Encrypt" );

     // encrypt..

     HBufC8* protectedPacket = iCryptoHandlerRTPContext.EncryptL();

     // finally, conditional update of roll-over counter..

     iCryptoHandlerRTPContext.UpdateROC();

    // first RTP packet successfully encrypted..

    // session keys derived etc..

    // we can go to normal state

    SRTP_DEBUG_DETAIL( "ESRTPStreamOutNormal" );

    iStreamContext.ChangeRTPState(MSRTPStreamOutContext::ESRTPStreamOutNormal);

    SRTP_DEBUG_TINT_VALUE("length of protectPacket", protectedPacket->Length() );

    SRTP_DEBUG_DETAIL( "TSRTPStreamOutStateUninit::DoProtectRtpL exit" );

    // return encrypted RTP packet

    return protectedPacket;    

    }

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

// TSRTPStreamOutStateUninit::DoProtectRtcpL

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

//		

HBufC8* TSRTPStreamOutStateUninit::DoProtectRtcpL(const TDesC8& aPacket)

    {

     // first initialize current RTP packet..

    iCryptoHandlerRTCPContext.InitializePlainPacketL(aPacket);    

    // Step 2 (in RFC 3711, section 3.4):

    //Srtcp packet index must be set to zero before the first srtcp packet is sent

    iCryptoHandlerRTCPContext.InitialPacketIndex();

     // derive session keys..

     iCryptoHandlerRTCPContext.DeriveSessionKeysL();

     // encrypt..

     HBufC8* protectedPacket = iCryptoHandlerRTCPContext.EncryptL();

    // first RTP packet successfully encrypted..

    // session keys derived etc..

    // we can go to normal state

    iStreamContext.ChangeRTCPState(MSRTPStreamOutContext::ESRTPStreamOutNormal);

    SRTP_DEBUG_TINT_VALUE("length of protect RTCP Packet", protectedPacket->Length() );

    // return encrypted RTP packet

    return protectedPacket;    

    }