/*

* Copyright (c) 2008-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:  Receiver of UDP datagrams

*

*/

#include <in_sock.h>

#include "ikedatainterface.h"

#include "ikemsgheader.h"

// CLASS HEADER

#include "ikev2receiver.h"

// ======== MEMBER FUNCTIONS ========

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

// Two-phased constructor.

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

//

CIkev2Receiver* CIkev2Receiver::NewL( MIkeDataInterface& aDataInterface,

                                      MIkev2ReceiverCallback& aCallback )

    {

    CIkev2Receiver* self = new (ELeave) CIkev2Receiver( aDataInterface,

                                                        aCallback );

    CleanupStack::PushL( self );

    self->ConstructL();

    CleanupStack::Pop( self );

    return self;

    }

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

// Destructor.

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

//

CIkev2Receiver::~CIkev2Receiver()

    {

    StopReceive();

    delete iUdpData;    

    }

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

// Constructor.

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

//

CIkev2Receiver::CIkev2Receiver( MIkeDataInterface& aDataInterface,

                                MIkev2ReceiverCallback& aCallback )

 : CActive( EPriorityStandard ),

   iUdpData( NULL ),

   iDataInterface( aDataInterface ),

   iCallback( aCallback )

    {

    CActiveScheduler::Add( this );

    }

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

// Second phase construction.

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

//

void CIkev2Receiver::ConstructL()

    {

    StartReceive();

    }

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

// Starts receive.

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

//

void CIkev2Receiver::StartReceive()

    {

    iReceivingData = ETrue;

    DoReceive();

    }

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

// Stops receive.

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

//

void CIkev2Receiver::StopReceive()

    {

    iReceivingData = EFalse;

    Cancel();

    iDataInterface.StopReceive();

    }

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

// From class CActive

// Handles completion of receive. 

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

//

void CIkev2Receiver::RunL()

    {

    if ( iStatus.Int() == KErrNone )

        {

        __ASSERT_DEBUG( iUdpData != NULL,

                        User::Invariant() );

        const ThdrISAKMP* ikeHdr = ThdrISAKMP::Ptr( iUdpData->Des() );

        TInt msgLth = iUdpData->Length();

        // Ignore possible <non-ESP marker> in the beginning of IKE message.

        TUint32 ikeMsgHdrOctets = GET32( ikeHdr );

        if ( ikeMsgHdrOctets == NON_ESP_MARKER )

            {

            ikeHdr  = ikeHdr->GotoOffset( NON_ESP_MARKER_SIZE );

            msgLth -= NON_ESP_MARKER_SIZE;

            }

        iCallback.IkeMsgReceived( *ikeHdr, iSrcAddr, iLocalPort );                

        }

    delete iUdpData;

    iUdpData = NULL;

    if ( iStatus.Int() == KErrNone )

        {

        if ( iReceivingData )

            {

            // Continue receiving.

            DoReceive();

            }

        }

    else

        {

        iCallback.ReceiveError( iStatus.Int() );

        }

    }

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

// From class CActive

// Handles cancellation of receive. 

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

//

void CIkev2Receiver::DoCancel()

    {

    iDataInterface.CancelReceive();

    delete iUdpData;

    iUdpData = NULL;

    }

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

// Receives UDP data. 

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

//

void CIkev2Receiver::DoReceive()

    {

    iDataInterface.ReceiveUdpData( iUdpData, iSrcAddr, iLocalPort, iStatus );

    SetActive();

    }