/*
* Copyright (c) 2007 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:    
*
*/




// INCLUDES
#include <e32debug.h>
#include "natfwconnectivityapidefs.h"
#include <mmccinterface.h>
#include <mmcccodecinformation.h>
#include "cncmconnectionmultiplexer.h"

#include "testdefinitions.hrh"
#include "ncmconnectionmultiplexerlogs.h"
#include "cncmportstore.h"
#include "natfwsocketmediaconnwrapper.h"
#include "muxtestconsolestubs.h"
#include "multiplexertcptests.h"


#ifndef __WINSCW__
const TInt KTestIapId = 1;
#else
const TInt KTestIapId = 6;
#endif

const TInt KAudioPrefVoipAudioUplink           = 0x05220001;
const TInt KAudioPrefVoipAudioUplinkNonSignal  = 0x05230001;
const TInt KAudioPrefVoipAudioDownlink         = 0x05210001;
const TInt KAudioDTMFString                    = 0x00150001;
const TInt KAudioPriorityDTMFString            = 45;

const TInt KDefaultQos = 46;
_LIT8(KUDPMessage,"UDP_Message"); 

const TUint KMediaConnection = 1;

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

// ---------------------------------------------------------------------------
// CMultiplexerTcpTestsInterface::CMultiplexerTcpTestsInterface
// C++ default constructor can NOT contain any code, that
// might leave.
// ---------------------------------------------------------------------------
//
CMultiplexerTcpTests::CMultiplexerTcpTests( CConsoleBase& aConsole,
    CNcmConnectionMultiplexer& aMux, CMuxTestConsoleStubs& aIFStub ) :
    iConsole( aConsole ), iMux( aMux ), iIfStub( aIFStub )
    {
    }
    
// ---------------------------------------------------------------------------
// CMultiplexerTcpTests::ConstructL
// Symbian 2nd phase constructor can leave.
// ---------------------------------------------------------------------------
//
void CMultiplexerTcpTests::ConstructL()
    {
    // Dummy 'cause first uheap marking
    TUint sessionId( 0 );
    TUint streamId( 0 );

    sessionId = iMux.CreateSessionL( KTestIapId, 5500, 10000 );
    CActiveScheduler::Start();
    
    streamId = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetUdp );
    
    iMux.RegisterIncomingConnectionObserverL( streamId, iIfStub ); 
    
    TUint sub = iMux.CreateConnectionL( streamId, KAfInet );
    iMux.UnregisterIncomingConnectionObserverL( streamId, iIfStub );
    iMux.RemoveSessionL( sessionId );
    }
    
// ---------------------------------------------------------------------------
// CMultiplexerTcpTests::NewL
// Static constructor.
// ---------------------------------------------------------------------------
//   
CMultiplexerTcpTests* CMultiplexerTcpTests::NewL( CConsoleBase& aConsole,
    CNcmConnectionMultiplexer& aMux, CMuxTestConsoleStubs& aIFStub )
    {
    CMultiplexerTcpTests* self 
        = new ( ELeave ) CMultiplexerTcpTests( aConsole, aMux, aIFStub );

    CleanupStack::PushL( self );
    self->ConstructL();
    CleanupStack::Pop( self );
    return self;
    }

// ---------------------------------------------------------------------------
// CMultiplexerTcpTests::~CMultiplexerTcpTests
// Destructor.
// ---------------------------------------------------------------------------
//
CMultiplexerTcpTests::~CMultiplexerTcpTests()
    {
    }

 void CMultiplexerTcpTests::DoRunTcpMuxTestsL()
    {
    iConsole.ClearScreen();
    iConsole.Printf( _L("\n*** MUX TESTS ***\n") );
    
    TInt err( KErrNone );

    iConsole.Printf( _L("Create session test\n") );
    RDebug::Print( _L("MUX_TEST_START: TestCreateSessionL") );
    //__UHEAP_MARK;
    TRAP( err, TestCreateSessionL() );
    //__UHEAP_MARKEND;
    RDebug::Print( _L("MUX_TEST_STOP: TestCreateSessionL, ERR: %d"), err );

    // Check again with memory leak check
    iConsole.Printf( _L("Create session test\n") );
    RDebug::Print( _L("MUX_TEST_START: TestCreateSessionL") );
    __UHEAP_MARK;
    TRAP( err, TestCreateSessionL() );
    __UHEAP_MARKEND;
    RDebug::Print( _L("MUX_TEST_STOP: TestCreateSessionL, ERR: %d"), err );
    
    err = KErrNone;
    RDebug::Print( _L("MUX_TEST_START: TestCreateStreamL") );
    iConsole.Printf( _L("Create stream test\n") );
    __UHEAP_MARK;
    TRAP( err, TestCreateStreamL() );
    __UHEAP_MARKEND;
    RDebug::Print( _L("MUX_TEST_STOP: TestCreateStreamL, ERR: %d"), err );

    err = KErrNone;
    RDebug::Print( _L("MUX_TEST_START: TestCreateConnectionL") );
    iConsole.Printf( _L("Create connection test\n") );
    __UHEAP_MARK;
    TRAP( err, TestCreateConnectionL() );
    __UHEAP_MARKEND;
    RDebug::Print( _L("MUX_TEST_STOP: TestCreateConnectionL, ERR: %d"), err );
 
    err = KErrNone;
    RDebug::Print( _L("MUX_TEST_START: TestConnectTcpConnectionsL") );
    iConsole.Printf( _L("Connect TCP connections test\n") );
    //__UHEAP_MARK;
    TRAP( err, TestConnectTcpConnectionsL() );
    //__UHEAP_MARKEND;
    RDebug::Print( _L("MUX_TEST_STOP: TestConnectTcpConnectionsL, ERR: %d"), err );

    // Check again with memory leak check
    err = KErrNone;
    RDebug::Print( _L("MUX_TEST_START: TestConnectTcpConnectionsL") );
    iConsole.Printf( _L("Connect TCP connections test\n") );
    __UHEAP_MARK;
    TRAP( err, TestConnectTcpConnectionsL() );
    __UHEAP_MARKEND;
    RDebug::Print( _L("MUX_TEST_STOP: TestConnectTcpConnectionsL, ERR: %d"), err );
    
    err = KErrNone;
    RDebug::Print( _L("MUX_TEST_START: TestActivateStunTcpMediaConnectionsL") );
    iConsole.Printf( _L("Activate STUN TCP MEDIA test\n") );
    __UHEAP_MARK;
    TRAP( err, TestActivateStunTcpMediaConnectionsL() );
    __UHEAP_MARKEND;
    RDebug::Print( _L("MUX_TEST_STOP: TestActivateStunTcpMediaConnectionsL, ERR: %d"), err );    
    }
      
// ---------------------------------------------------------------------------
// CMultiplexerTcpTests::TestCreateSessionL
// ---------------------------------------------------------------------------
//
void CMultiplexerTcpTests::TestCreateSessionL()
    {
    TUint sessionId( 0 );
    
    sessionId = iMux.CreateSessionL( KTestIapId, 5000, 8000 );
    CActiveScheduler::Start();
    
    iMux.RemoveSessionL( sessionId );
    
    sessionId = iMux.CreateSessionL( KTestIapId, 5000, 8000 );
    iMux.RemoveSessionL( sessionId );
    }

    
// ---------------------------------------------------------------------------
// CMultiplexerTcpTests::TestCreateSessionL
// ---------------------------------------------------------------------------
//
void CMultiplexerTcpTests::TestCreateStreamL()
    {
    TUint sessionId( 0 );
    TUint sessionId2( 0 );
    TUint streamId( 0 );
    TUint streamId2( 0 );  
    
    sessionId = iMux.CreateSessionL( KTestIapId, 5000, 8000 ); 
    CActiveScheduler::Start();

        
    streamId = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );
    streamId2 = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );
      
    iMux.RemoveStreamL( streamId ); 
    iMux.RemoveStreamL( streamId2 );

    streamId = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );
    
    iMux.RegisterIncomingConnectionObserverL( streamId, iIfStub );
    iMux.RegisterIncomingConnectionObserverL( streamId, iIfStub );
    
    streamId2 = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );
    
    iMux.RemoveStreamL( streamId );
    iMux.RemoveSessionL( sessionId );
    iMux.RemoveSessionL( sessionId2 );
    }

// ---------------------------------------------------------------------------
// CMultiplexerTcpTests::TestCreateConnectionL
// ---------------------------------------------------------------------------
//
void CMultiplexerTcpTests::TestCreateConnectionL()
    {
    TUint sessionId( 0 );
    
    TUint streamId( 0 );
    TUint streamId2( 0 );
      
    TUint connectionId( 0 );
    TUint connectionId2( 0 );
    TUint connectionId3( 0 );
    TUint connectionId4( 0 );
    TUint connectionId5( 0 );
    
    sessionId = iMux.CreateSessionL( KTestIapId, 5000, 8000 );
    CActiveScheduler::Start();
    
    streamId = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );
    streamId2 = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );
    
    iMux.RegisterIncomingConnectionObserverL(streamId, iIfStub );
  
    connectionId = iMux.CreateConnectionL( streamId, KAfInet );
    connectionId2 = iMux.CreateConnectionL( streamId, KAfInet );  
    connectionId3 = iMux.CreateConnectionL( streamId2, KAfInet );
    
    // New connection, Reuse local address
    TInetAddr connectionAddr = iMux.LocalIPAddressL( streamId, connectionId2 );
    connectionId4 = iMux.CreateConnectionL( streamId2, connectionAddr );  
    // Check success of address reuse
    TInetAddr connectionAddr2 = iMux.LocalIPAddressL( streamId, connectionId4 );
    ASSERT( connectionAddr.CmpAddr( connectionAddr2 ) );
    
    
    // New connection, Reuse local address
    TInetAddr connectionAddr3 = iMux.LocalIPAddressL( streamId, connectionId3 );
    connectionId5 = iMux.CreateConnectionL( streamId2, connectionAddr3 );
    // Check success of address reuse
    TInetAddr connectionAddr5 = iMux.LocalIPAddressL( streamId2, connectionId5 );    
    ASSERT( connectionAddr3.CmpAddr( connectionAddr5 ) );    
    
    // New connection, Reuse local address
    connectionAddr3 = iMux.LocalIPAddressL( streamId, connectionId3 );
    connectionId5 = iMux.CreateConnectionL( streamId2, connectionAddr3 );
    // Check success of address reuse
    connectionAddr5 = iMux.LocalIPAddressL( streamId2, connectionId5 );    
    ASSERT( connectionAddr3.CmpAddr( connectionAddr5 ) );  
        
    iMux.RemoveConnectionL( streamId2, connectionId5 );
    iMux.RemoveConnectionL( streamId, connectionId3 );
    iMux.RemoveStreamL( streamId );
    iMux.RemoveSessionL( sessionId ); 
    }

  
// ---------------------------------------------------------------------------
// CMultiplexerUdpTests::TestConnectTcpConnectionsL
// ---------------------------------------------------------------------------
//
void CMultiplexerTcpTests::TestConnectTcpConnectionsL()
    {
    TUint sessionId( 0 );
    
    TUint streamId( 0 );
    TUint streamId2( 0 );
      
    TUint connectionId( 0 );
    TUint connectionId2( 0 );
    TUint connectionId3( 0 );
    
 
    // Create 1 session, 1 stream, 1 connection
    sessionId = iMux.CreateSessionL( KTestIapId, 8000, 11000 );
    CActiveScheduler::Start();  
    
    streamId = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );
    streamId2 = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );     
    connectionId = iMux.CreateConnectionL( streamId, KAfInet ); 
    connectionId2 = iMux.CreateConnectionL( streamId2, KAfInet );
        
    // Address resolving
    TInetAddr connectionAddr( iMux.LocalIPAddressL( streamId, connectionId ) );
    TInetAddr connectionAddr2( iMux.LocalIPAddressL( streamId2, connectionId2 ) );    
    
    iMux.RegisterConnectionObserverL( streamId, iIfStub );
    iMux.RegisterConnectionObserverL( streamId2, iIfStub );    
  
    iMux.OpenTcpConnectionL( streamId, connectionId, ETcpSetupActive, connectionAddr2 ); 
    iMux.OpenTcpConnectionL( streamId2, connectionId2, ETcpSetupActive, connectionAddr );
    CActiveScheduler::Start(); 
    CActiveScheduler::Start();    

    iMux.CloseTcpConnection( streamId, connectionId );
    iMux.CloseTcpConnection( streamId2, connectionId2 );
    
    iMux.OpenTcpConnectionL( streamId, connectionId, ETcpSetupPassive, connectionAddr2 ); 
    iMux.OpenTcpConnectionL( streamId2, connectionId2, ETcpSetupActive, connectionAddr );
    CActiveScheduler::Start(); 
    CActiveScheduler::Start();
 
    iMux.CloseTcpConnection( streamId, connectionId );
    iMux.CloseTcpConnection( streamId2, connectionId2 );
    
    iMux.OpenTcpConnectionL( streamId2, connectionId2, ETcpSetupActive, connectionAddr );
    iMux.OpenTcpConnectionL( streamId, connectionId, ETcpSetupPassive, connectionAddr2 );
    CActiveScheduler::Start(); 
    CActiveScheduler::Start();
    
    iMux.CloseTcpConnection( streamId, connectionId );
    iMux.CloseTcpConnection( streamId2, connectionId2 ); 
       
    // Cancel the connection process
    iMux.OpenTcpConnectionL( streamId, connectionId, ETcpSetupPassive, connectionAddr2 );    
    iMux.CloseTcpConnection( streamId, connectionId );
    CActiveScheduler::Start();
    
    iMux.OpenTcpConnectionL( streamId, connectionId, ETcpSetupPassive, connectionAddr2 );    
    iMux.CloseTcpConnection( streamId, connectionId );
    CActiveScheduler::Start();  
    iMux.RemoveSessionL( sessionId );
    }



// ---------------------------------------------------------------------------
// CMultiplexerTcpTests::TestActivateStunTcpMediaConnectionsL
// ---------------------------------------------------------------------------
//
void CMultiplexerTcpTests::TestActivateStunTcpMediaConnectionsL()
    {
    TUint sessionId( 0 );
    
    TUint streamId( 0 );
    TUint streamId2( 0 );
      
    TUint connectionId( 0 );
    TUint connectionId2( 0 );
    TUint connectionId3( 0 );
    
    sessionId = iMux.CreateSessionL( KTestIapId, 5000, 8000 );
    CActiveScheduler::Start();
    
    streamId = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );
    streamId2 = iMux.CreateStreamL( sessionId, KDefaultQos, KProtocolInetTcp );
    
    // Simulate UDP fetch candidate    
    iMux.RegisterIncomingConnectionObserverL( streamId, iIfStub );
    iMux.RegisterConnectionObserverL( streamId, iIfStub );
    iMux.RegisterConnectionObserverL( streamId2, iIfStub );
    connectionId = iMux.CreateConnectionL( streamId, KAfInet );
    
    // Test connection
    connectionId2 = iMux.CreateConnectionL( streamId2, KAfInet );  
    TInetAddr connectionAddress2( iMux.LocalIPAddressL( streamId2, connectionId2 ) );

    // Test connection, SERVER
    connectionId3 = iMux.CreateConnectionL( streamId2, KAfInet );  
    TInetAddr connectionAddress3( iMux.LocalIPAddressL( streamId2, connectionId3 ) );
    
    /**** setup wrapper *****/
    CNATFWSocketMediaConnWrapper* mediaWrapper = CNATFWSocketMediaConnWrapper::NewL(
             iMux.PortStoreL( sessionId ).Port(), KProtocolInetUdp, streamId );
       
    iMux.RegisterMediaWrapperL( mediaWrapper );
            
    TInetAddr destinationAddress;    
    mediaWrapper->OutgoingAddr( destinationAddress );    
    __CONNECTIONMULTIPLEXER_ADDRLOG( "CMultiplexerUdpTests::TestUdpAudioL- Wrapper LOCAL ADDRESS: ", destinationAddress )

    
    mediaWrapper->SetIncomingAddrL( destinationAddress );
    //TODO connect connection 2 to wrapper and start wrapper receiving
    /**************************/

    // Dummy address
    TInetAddr stunServerAddress;
    stunServerAddress.SetAddress( INET_ADDR( 10,36,7,6 ) );
    stunServerAddress.SetPort( 3478 );
   
    // STUN-plugin connects connection to stun server
    iMux.OpenTcpConnectionL( streamId2, connectionId3, ETcpSetupPassive, stunServerAddress );  // Server
    iMux.OpenTcpConnectionL( streamId, connectionId, ETcpSetupActive, connectionAddress3 );
    CActiveScheduler::Start();   
    CActiveScheduler::Start(); 
    
    iMux.SetReceivingStateL( streamId, connectionId, EStreamingStateActive );
    iMux.SetSendingStateL( streamId, connectionId, stunServerAddress, EStreamingStateActive );
    iMux.SetReceivingStateL( streamId2, connectionId3, EStreamingStateActive );
    CActiveScheduler::Start();  
    CActiveScheduler::Start();    
    CActiveScheduler::Start();
    
    // Binding request
    _LIT8(KUDPMessage,"UDP_Binding_Request");
    iMux.SendL( streamId, connectionId, KUDPMessage, NULL );
    
   
    // STUN plugin, reuse connection.. connect to peer address
    iMux.CloseTcpConnection( streamId, connectionId );  // Connection to STUN server
    CActiveScheduler::Start(); 
        
    iMux.OpenTcpConnectionL( streamId2, connectionId2, ETcpSetupPassive, stunServerAddress );  // Remote end
    iMux.OpenTcpConnectionL( streamId, connectionId, ETcpSetupActive, connectionAddress2 );   
    CActiveScheduler::Start();     
    CActiveScheduler::Start();      
    
    // Remote end
         
    TInetAddr connectionAddress( iMux.LocalIPAddressL( streamId, connectionId ) );
    iMux.SetSendingStateL( streamId2, connectionId2, connectionAddress, EStreamingStateActive );
    CActiveScheduler::Start();
    
    
    // Setup media flow to peer
    // NATFW client
    iMux.SetReceivingStateL( streamId, connectionId, EStreamingStateActive );
    iMux.SetReceivingStateForMediaL( streamId, connectionId, EStreamingStateActive );
    CActiveScheduler::Start();
    CActiveScheduler::Start();
    
    // NATFW client
    iMux.SetSendingStateL( streamId, connectionId, connectionAddress2, EStreamingStateActive );;
    iMux.SetSendingStateForMediaL( streamId, connectionId, EStreamingStateActive );
    CActiveScheduler::Start(); 
    CActiveScheduler::Start(); 


    // Set sending again
    iMux.SetSendingStateL( streamId, connectionId, connectionAddress2, EStreamingStateActive );
    CActiveScheduler::Start(); 
    iMux.SetSendingStateL( streamId, connectionId, connectionAddress2, EStreamingStateActive );
    CActiveScheduler::Start(); 
           
    // Set receiving again
    iMux.SetReceivingStateL( streamId, connectionId, EStreamingStateActive );
    CActiveScheduler::Start();
    iMux.SetReceivingStateL( streamId, connectionId, EStreamingStateActive );
    CActiveScheduler::Start(); 
    
    iMux.SendL( streamId2, connectionId2, KUDPMessage, &iIfStub );
    iMux.SendL( streamId2, connectionId2, KUDPMessage, connectionAddress2, &iIfStub );
    CActiveScheduler::Start(); 
    CActiveScheduler::Start();
    
    iMux.CloseTcpConnection( streamId, connectionId );  // Connection to STUN server   
    iMux.RemoveSessionL( sessionId );  
    }