/*

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

//  EXTERNAL INCLUDES

#include <digia/eunit/eunitmacros.h>

#include <digia/eunit/EUnitDecorators.h>

#include <commdbconnpref.h>

#include <e32debug.h>

//  INTERNAL INCLUDES

#include "natfwstunrelaybinding.h"

#include "cstunrelaybindingimplementation.h"

#include "mstunbindingobserver.h"

#include "natfwstunclientdefs.h"

#include "natfwunsafunknownattributesattribute.h"

#include "natfwunsafbindingrequest.h"

#include "natfwunsafbindingresponse.h"

#include "natfwunsafbindingerrorresponse.h"

#include "natfwunsafallocaterequest.h"

#include "natfwunsafallocateerrorresponse.h"

#include "natfwunsafallocateresponse.h"

#include "natfwunsafconnectresponse.h"

#include "natfwunsafconnecterrorresponse.h"

#include "natfwunsafconnectrequest.h"

#include "natfwunsafsetactivedestinationrequest.h"

#include "natfwunsafsetactivedestinationresponse.h"

#include "natfwunsafsetactivedestinationerrorresponse.h"

#include "cstunbindinggetaddress.h"

#include "natfwunsaficecontrolledattribute.h"

#include "natfwbindingimplementationbase.h"

#include "natfwunsafdataindication.h"

#include "natfwunsafdataattribute.h"

#include "natfwunsafremoteaddressattribute.h"

#include "natfwunsafalternateserverattribute.h"

#include "cstunbindinggetsharedsecret.h"

#include "cstunbindinggetaddress.h"

#include "cstunbindingactive.h"

#include "cstunbindingwaittoretry.h"

#include "natfwunsafrealmattribute.h"

#include "natfwunsafnonceattribute.h"

#include "cunsafmessageencoder.h"

#include "cstunclientimplementation.h"

#include "natfwunsaftcprelaypacket.h"

#include "cnatfwunsafencodeditem.h"

#include "natfwunittestmacros.h"

_LIT8( KServerAddress, "10.32.194.251" );

_LIT8( KUsername, "userAA" );

_LIT8( KPassword,"salasana");

_LIT8(KRealmVal,"\"myrealm\"");

_LIT8(KRealmVal2,"\"myREALMI\"");

const TDesC8& dataDesc5(KRealmVal);

const TDesC8& dataDesc6(KRealmVal2);

const TUint KDefaultRtoValue = 0;

// CONSTRUCTION

ut_cstunrelaybindingimplementation* ut_cstunrelaybindingimplementation::NewL()

    {

    ut_cstunrelaybindingimplementation* self = ut_cstunrelaybindingimplementation::NewLC();

    CleanupStack::Pop();

    return self;

    }

ut_cstunrelaybindingimplementation* ut_cstunrelaybindingimplementation::NewLC()

    {

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

    CleanupStack::PushL( self );

    self->ConstructL();

    return self;

    }

// Destructor (virtual by CBase)

ut_cstunrelaybindingimplementation::~ut_cstunrelaybindingimplementation()

    {

    }

// Default constructor

ut_cstunrelaybindingimplementation::ut_cstunrelaybindingimplementation()

    {

    }

// Second phase construct

void ut_cstunrelaybindingimplementation::ConstructL()

    {

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

    // It generates the test case table.

    CEUnitTestSuiteClass::ConstructL();

    }

//  METHODS

// from mstunclientbserver

void ut_cstunrelaybindingimplementation::STUNClientInitCompleted( const CSTUNClient& /*aClient*/,

                              TInt aCompletionCode )

    {

    RDebug::Print( _L("**NATFW STUNClientInitCompleted, code: %d"), aCompletionCode );

    CActiveScheduler::Stop();

    }

void ut_cstunrelaybindingimplementation::STUNBindingEventOccurredL( TSTUNBindingEvent aEvent, 

                                const CBinding& /*aBinding*/ )

    {

    RDebug::Print( _L("**NATFW STUNBindingEventOccurredL, code: %d"), aEvent );

    CActiveScheduler::Stop();

    }

void ut_cstunrelaybindingimplementation::STUNBindingErrorOccurred( const CBinding& /*aBinding*/, 

                               TInt aError )

    {

    RDebug::Print( _L("**NATFW STUNBindingErrorOccurred, code: %d"), aError );

    CActiveScheduler::Stop();

    }

void ut_cstunrelaybindingimplementation::Notify( TUint /*aSessionId*/, TUint /*aStreamId*/, 

    TNotifyType /*aType*/, TInt /*aError*/ )

    {

    RDebug::Print( _L("**NATFW Notify") );

    }

const CSTUNClient& ut_cstunrelaybindingimplementation::STUNClient() const

    {

    return *iStunclient;

    }

CDeltaTimer& ut_cstunrelaybindingimplementation::TimerProvider()

    {  

    return *iDeltatimer;

    }

TInt ut_cstunrelaybindingimplementation::RetransmitInterval() const

    {

    TInt dummy(0);

    return dummy;

    }

TTransportProtocol ut_cstunrelaybindingimplementation::TransportProtocol() const

    {

    iStunclient->Implementation().TransportProtocol();

    }

void ut_cstunrelaybindingimplementation::AddressResolvedL( const CBinding& /*aBinding*/ ) const

    {    

    }

void ut_cstunrelaybindingimplementation::ObtainSharedSecretL( CBinding& /*aBinding*/ )

    {

    TBufC8<16> password( KPassword );

    TBufC8<6> usernameA( KUsername );    

    iImpl->CreateBindingRequestL( usernameA, password );

    iImpl->SendBindingRequestL();

    }

TBool ut_cstunrelaybindingimplementation::SharedSecretRejectedL( 

    const CBinding& /*aBinding*/, const TDesC8& /*aUsername*/, const TDesC8& /*aPassword*/ )

    {

    TBool dummy( EFalse );

    return dummy;    

    }

TBool ut_cstunrelaybindingimplementation::ObtainServerAddress( TInetAddr& aAddress )

    {

    #ifdef _DEBUG_EUNIT

        TInetAddr addr;

        addr.SetAddress( INET_ADDR( 10,32,194,251 ) );

        aAddress = addr;

        return ETrue;

    #endif

    return EFalse;

    }

void ut_cstunrelaybindingimplementation::ObtainTransactionIDL( TNATFWUNSAFTransactionID& aTransactionID )

    {

    iStunclient->ObtainTransactionIDL( aTransactionID );    

    }

void ut_cstunrelaybindingimplementation::AttachBindingL( const CBinding& /*aBinding*/ )

    {

    }

void ut_cstunrelaybindingimplementation::DetachBinding( const CBinding& /*aBinding*/ )

    {

    }

void ut_cstunrelaybindingimplementation::BindingErrorL( const CBinding& /*aBinding*/, TInt aError, TBool /*aIsFatal*/ ) 

    {

    RDebug::Print( _L("**NATFW BindingErrorL, error: %d"), aError );

    }

void ut_cstunrelaybindingimplementation::BindingEventOccurred( const CBinding& /*aBinding*/, 

                           TSTUNCallbackInfo::TFunction /*aEvent*/ )

    {    

    }

const TDesC8& ut_cstunrelaybindingimplementation::UsernameForIndication()

    {

    TBufC8<16> password( KPassword );

    return password;    

    }

const TDesC8& ut_cstunrelaybindingimplementation::PasswordForIndication()

    {

    TBufC8<16> password( KPassword );

    return password;

    }

void ut_cstunrelaybindingimplementation::SetupL(  )

    {

    TUint iapID( 6 );

    TInt retransmitInterval( 10 );

    TBufC8<13> serveraddress( KServerAddress );

    TUint serverport( 2000 );

    TBufC8<10> service( KStunRelay );

    TBool obtainsharedsecret( ETrue );

    TBool failifnoSRVrecordfound( EFalse );

    iDeltatimer = CDeltaTimer::NewL( 1 );

    User::LeaveIfError( iSocketServ.Connect() );

    iStreamId = 1;

    iSubstreamId = 1;

    TBool icmpUsed( EFalse );

    iMux = CNcmConnectionMultiplexer::NewL( *this );

    TCommDbConnPref pref;

    pref.SetDialogPreference( ECommDbDialogPrefDoNotPrompt );

    pref.SetIapId( iapID );

    User::LeaveIfError( iConnection.Open( iSocketServ ) );

    iConnection.Start( pref, iStatus );

    User::WaitForRequest( iStatus );

    User::LeaveIfError( iStatus.Int() );    

    iStunclient = CSTUNClient::NewL( retransmitInterval,

                                     serveraddress,

                                     serverport,

                                     service,

                                     iSocketServ,

                                     iConnection,

                                     *iDeltatimer,

                                     *this,

                                     obtainsharedsecret,

                                     failifnoSRVrecordfound,

                                     icmpUsed,

                                     iMux );

    CActiveScheduler::Start();  

    User::LeaveIfError( iSocket.Open( iSocketServ, KAfInet, KSockDatagram,

                                     KProtocolInetUdp ));

    iSocket.SetLocalPort( 5000 );                                 

    iBinding = CSTUNRelayBinding::NewL( *iStunclient, iStreamId, iSubstreamId );

    iImpl = CSTUNRelayBindingImplementation::NewL( *iBinding, *this,

                                                   iStreamId, iSubstreamId, iMux );

    }

void ut_cstunrelaybindingimplementation::Teardown(  )

    {

    delete iImpl;

    delete iBinding;    

    delete iStunclient;    

    iSocket.Close();

    iConnection.Stop();

    iConnection.Close();

    iSocketServ.Close();

    delete iServeraddress;

    delete iResponse;

    delete iRespDataIndication;    

    delete iDeltatimer;

    delete iMux;   

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_NewLL(  )

    {

    CSTUNRelayBindingImplementation* impl = 

        CSTUNRelayBindingImplementation::NewL( *iBinding, *this, iSocket );

    delete impl;

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_NewL_1L(  )

    {

    CSTUNRelayBindingImplementation* impl = 

        CSTUNRelayBindingImplementation::NewL( *iBinding, *this, iStreamId, iSubstreamId, iMux );

    delete impl;

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_TimerExpiredLL(  )

    {

    CSTUNBindingGetAddress* getAddress = new ( ELeave ) CSTUNBindingGetAddress();

    CleanupStack::PushL( getAddress );

    CSTUNBindingGetSharedSecret* getSharedSecret =

        new ( ELeave ) CSTUNBindingGetSharedSecret( *getAddress );

    CleanupStack::PushL( getSharedSecret );

    CSTUNBindingWaitToRetry* waitToRetry = new ( ELeave ) CSTUNBindingWaitToRetry( *getSharedSecret );

    CleanupStack::PushL( waitToRetry );

    iImpl->ChangeState( *waitToRetry );

    TInetAddr addr( KProtocolInetUdp );

    _LIT( KServer, "10.32.194.251" );

    addr.Input( KServer );

    iImpl->iServerAddress = addr;

    iImpl->iRequestType = iImpl->ESendRequest;     

    iImpl->TimerExpiredL();

    CleanupStack::PopAndDestroy( waitToRetry );

    CleanupStack::PopAndDestroy( getSharedSecret );

    CleanupStack::PopAndDestroy( getAddress );

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_LeaveFromTimerExpiredL(  )

    {

    //EUNIT_ASSERT_PANIC( iImpl->LeaveFromTimerExpired( KErrNone), "", KErrArgument );    

    iImpl->LeaveFromTimerExpired( KErrNotFound );

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_PublicAddressObtainedLL(  )

    {

    TInetAddr addr;

    CSTUNBindingGetAddress* getAddress = new ( ELeave ) CSTUNBindingGetAddress();

    iImpl->ChangeState( *getAddress );

    iImpl->PublicAddressObtainedL( addr );

    delete getAddress;

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_PublicAddressObtainedL_1L(  )

    {

    TInetAddr reflexsiveAddr;

    TInetAddr relayAddr;

    CSTUNBindingGetAddress* getAddress = new ( ELeave ) CSTUNBindingGetAddress();

    iImpl->ChangeState( *getAddress );

    iImpl->PublicAddressObtainedL( reflexsiveAddr, relayAddr );

    delete getAddress;

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_TransactionErrorL(  )

    {

    EUNIT_ASSERT_PANIC( iImpl->TransactionError( 

        KErrNone, CNATFWUNSAFUnknownAttributesAttribute::NewLC() ), "", KErrArgument );     

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_TransactionEventOccurredLL(  )

    {

    iImpl->TransactionEventOccurredL( TSTUNCallbackInfo::EInitCompleted );

    TBufC8<16> password( KPassword );

    TBufC8<6> usernameA( KUsername );

    iImpl->iRequestType = iImpl->ESendRequest;

    iImpl->CreateBindingRequestL( usernameA, password );

    TInetAddr addr( KProtocolInetUdp );

    _LIT( KServer, "10.32.194.251" );

    addr.Input( KServer );

    iImpl->iServerAddress = addr; 

    iImpl->SendBindingRequestL();    

    //iImpl->iRequestType = iImpl->ESetActiveDestinationRequest;

    iImpl->Terminate(401);    

    TUint32 timerValue(10);

    iImpl->SetActiveDestinationRequestL( addr, timerValue );

    iImpl->TransactionEventOccurredL( TSTUNCallbackInfo::EInitCompleted );

    //iImpl->iRequestType = iImpl->ESendIndicationToNetwork;

    iImpl->TransactionEventOccurredL( TSTUNCallbackInfo::EInitCompleted );    

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_AllocateRequestLL(  )

    {

    iImpl->AllocateRequestL( KDefaultRtoValue );

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_CancelRequestL(  )

    {

    iImpl->CancelRequest();

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_SocketL(  )

    {

    CSTUNRelayBindingImplementation* impl = 

        CSTUNRelayBindingImplementation::NewL( *iBinding, *this, iSocket );       

    RSocket socket = impl->Socket();    

    delete impl;

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_AddressResolvedL(  )

    {

    TBool isResolved = iImpl->AddressResolved();

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_PublicAddrL(  )

    {

    TInetAddr addr = iImpl->PublicAddr();

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_AlternateServerAddrL(  )

    {

    TInetAddr addr = iImpl->AlternateServerAddr();

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_HandleDataLL(  )

    {

    EUNIT_DISABLE_ALLOC_DECORATOR;

    CUnsafMessageEncoder* encoder = CUnsafMessageEncoder::NewLC();

    _LIT8( KPassword,"aaaabbbbccccdddd");

    TBufC8<16> password( KPassword );

    TNATFWUNSAFTransactionID transactionID;

    iStunclient->ObtainTransactionIDL( transactionID );

    CNATFWUNSAFDataIndication* indication = CNATFWUNSAFDataIndication::NewLC( transactionID );

    CNATFWUNSAFBindingResponse* response = CNATFWUNSAFBindingResponse::NewLC( transactionID );

    HBufC8* respmsg = encoder->EncodeMessageL( *response );

    CleanupStack::PopAndDestroy( response );

    CleanupStack::PushL( respmsg );

    TInetAddr remoteAddr;

    TBool consumed = EFalse;

    EUNIT_ENABLE_ALLOC_DECORATOR;

    HBufC8* ptr = iImpl->HandleDataL( *respmsg, consumed, remoteAddr );

    EUNIT_DISABLE_ALLOC_DECORATOR;

    EUNIT_ASSERT( NULL == ptr );

    EUNIT_ASSERT( EFalse == consumed );

    CleanupStack::PopAndDestroy( respmsg );

// non-valid stun req inside indication

    _LIT8( KDummypacket, "oysescatechnologiesabFIN" );

    TBufC8<24> buffi( KDummypacket );

    indication->AddAttributeL( CNATFWUNSAFDataAttribute::NewLC( buffi ) );

    CleanupStack::Pop(); // CNATFWUNSAFDataAttribute

    indication->AddAttributeL( CNATFWUNSAFRemoteAddressAttribute::NewLC( remoteAddr ) );

    CleanupStack::Pop(); // CNATFWUNSAFRemoteAddressAttribute

    HBufC8* msg = encoder->EncodeMessageL( *indication );

    CleanupStack::PushL( msg );

    EUNIT_ENABLE_ALLOC_DECORATOR;

    ptr = iImpl->HandleDataL( *msg, consumed, remoteAddr );

    EUNIT_ASSERT( NULL != ptr );

    EUNIT_ASSERT( EFalse == consumed );

    delete ptr;

    CleanupStack::PopAndDestroy( msg );

    CleanupStack::PopAndDestroy( indication );  // test

    CleanupStack::PopAndDestroy( encoder );

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_HandleDataL2L(  )  

    {

    EUNIT_DISABLE_ALLOC_DECORATOR;

    CUnsafMessageEncoder* encoder = CUnsafMessageEncoder::NewLC();

    TInetAddr remoteAddr;

    TBool consumed = EFalse;

    _LIT8( KPassword,"aaaabbbbccccdddd");

    TBufC8<16> password( KPassword );

    TNATFWUNSAFTransactionID transactionID;

    iStunclient->ObtainTransactionIDL( transactionID );    

    CNATFWUNSAFDataIndication* indication = CNATFWUNSAFDataIndication::NewLC( transactionID );

    CNATFWUNSAFBindingResponse* response = CNATFWUNSAFBindingResponse::NewLC( transactionID );

    HBufC8* respmsg = encoder->EncodeMessageL( *response );

    CleanupStack::PopAndDestroy( response );

    CleanupStack::PushL( respmsg );

// valid stun request inside indication

    indication->DeleteAttribute( CNATFWUNSAFAttribute::EData );

    indication->AddAttributeL( CNATFWUNSAFDataAttribute::NewLC( *respmsg ) );

    CleanupStack::Pop(); // CNATFWUNSAFDataAttribute

    CleanupStack::PopAndDestroy( respmsg );

    HBufC8* msg = encoder->EncodeMessageL( *indication );

    CleanupStack::PushL( msg );

    EUNIT_ENABLE_ALLOC_DECORATOR;

    HBufC8* ptr = iImpl->HandleDataL( *msg, consumed, remoteAddr );

    EUNIT_DISABLE_ALLOC_DECORATOR;

    EUNIT_ASSERT( NULL != ptr );             

    EUNIT_ASSERT( EFalse == consumed );

    delete ptr;

// Tcp framing not valid 

    iStunclient->Implementation().iTransportProtocol = ETcpProtocol;

    EUNIT_ENABLE_ALLOC_DECORATOR;

    ptr = iImpl->HandleDataL( *msg, consumed, remoteAddr );

    EUNIT_ASSERT( NULL == ptr );

    EUNIT_ASSERT( EFalse == consumed );

    CleanupStack::PopAndDestroy( msg );

    CleanupStack::PopAndDestroy( indication ); 

    CleanupStack::PopAndDestroy( encoder );

    }

void ut_cstunrelaybindingimplementation::UT_CSTUNRelayBindingImplementation_HandleDataL3L(  )

    {

    EUNIT_DISABLE_ALLOC_DECORATOR;

    CUnsafMessageEncoder* encoder = CUnsafMessageEncoder::NewLC();