################################################################################
# Version 1.0    $Revision: 1 $
#
#    Copyright © 1997 - 2007 by IXIA
#    All Rights Reserved.
#
#    Revision Log:
#    08-oct-2007  RAntonescu
#
# Description:
#
################################################################################

################################################################################
#                                                                              #
#                                LEGAL  NOTICE:                                #
#                                ==============                                #
# The following code and documentation (hereinafter "the script") is an        #
# example script for demonstration purposes only.                              #
# The script is not a standard commercial product offered by Ixia and have     #
# been developed and is being provided for use only as indicated herein. The   #
# script [and all modifications, enhancements and updates thereto (whether     #
# made by Ixia and/or by the user and/or by a third party)] shall at all times #
# remain the property of Ixia.                                                 #
#                                                                              #
# Ixia does not warrant (i) that the functions contained in the script will    #
# meet the user's requirements or (ii) that the script will be without         #
# omissions or error-free.                                                     #
# THE SCRIPT IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND, AND IXIA        #
# DISCLAIMS ALL WARRANTIES, EXPRESS, IMPLIED, STATUTORY OR OTHERWISE,          #
# INCLUDING BUT NOT LIMITED TO ANY WARRANTY OF MERCHANTABILITY AND FITNESS FOR #
# A PARTICULAR PURPOSE OR OF NON-INFRINGEMENT.                                 #
# THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE SCRIPT  IS WITH THE #
# USER.                                                                        #
# IN NO EVENT SHALL IXIA BE LIABLE FOR ANY DAMAGES RESULTING FROM OR ARISING   #
# OUT OF THE USE OF, OR THE INABILITY TO USE THE SCRIPT OR ANY PART THEREOF,   #
# INCLUDING BUT NOT LIMITED TO ANY LOST PROFITS, LOST BUSINESS, LOST OR        #
# DAMAGED DATA OR SOFTWARE OR ANY INDIRECT, INCIDENTAL, PUNITIVE OR            #
# CONSEQUENTIAL DAMAGES, EVEN IF IXIA HAS BEEN ADVISED OF THE POSSIBILITY OF   #
# SUCH DAMAGES IN ADVANCE.                                                     #
# Ixia will not be required to provide any software maintenance or support     #
# services of any kind (e.g., any error corrections) in connection with the    #
# script or any part thereof. The user acknowledges that although Ixia may     #
# from time to time and in its sole discretion provide maintenance or support  #
# services for the script, any such services are subject to the warranty and   #
# damages limitations set forth herein and will not obligate Ixia to provide   #
# any additional maintenance or support services.                              #
#                                                                              #
################################################################################

################################################################################
#
# Description:
#    This sample creates the following MVPN setup.
#
#    +-----+     +-----+   +----------+      +-----+      +-------+            
#    | CE1 |-----| DUT |---| Provider |------| PE1 |-+ +--| MVRF1 |   MVPN     
#    +-----+     +-----+   +----------+      +-----+      +-------+            
#                                                                              
# Protocols used:                                                              
#    Provider Side EGP                - BGP                                    
#    Provider Side IGP                - ISIS                                   
#    Provider Side MPLS Protocol      - LDP                                    
#    Provider Side Multicast Protocol - PIM/SM                                 
#    Customer Side Multicast Protocol - PIM/SM                                 
#                                                                              
# DUT configuration:                                                           
# ip cef distributed
# ip vrf mvpn910
# rd 1:910
# route-target both 1:910
# mdt default 239.1.1.1
# exit
# 
# ip multicast-routing
# ip multicast-routing vrf mvpn910
# 
# interface Loopback20
# ip address 21.1.1.1 255.255.255.255
# ip pim sparse-mode
# exit
# 
# ip pim rp-address 21.1.1.1
# ip pim vrf mvpn910 rp-address 20.31.31.1
# 
# interface POS 4/1
# description Provider
# no shutdown
# ip address 20.30.30.1 255.255.255.0
# ip router isis mvpn910
# ip pim sparse-mode
# mpls label protocol ldp
# tag-switching ip
# no keepalive
# encapsulation hdlc
# clock source internal
# pos flag c2 9
# exit
# 
# interface gigabitEthernet7/36
# no shutdown
# 
# interface gigabitEthernet7/36.1
# description Customer
# encapsulation dot1Q 910
# ip vrf forwarding mvpn910
# ip address 20.31.31.1 255.255.255.0
# ip pim sparse-mode
# exit
# 
# router bgp 1
# neighbor 23.34.35.1 remote-as 1
# neighbor 23.34.35.1 update-source Loopback20
# address-family vpnv4
# neighbor 33.34.35.1 activate
# neighbor 33.34.35.1 send-community extended
# exit-address-family
# address-family ipv4 vrf mvpn910
# no auto-summary
# no synchronization
# redistribute connected
# exit-address-family
# exit
# 
# router isis mvpn910
# net 26.0001.0000.0000.072C.00
# is-type level-1-2
# metric-style transition
# exit
#     
# Module:                                                                      
#    The sample was tested on a MSM2.5 POS module (Provider Side) and a        
#    LM1000STXS4 module (Customer Side).                                       
#                                                                              
################################################################################


proc script_increment_ipv4_address {ip_addr ip_addr_step} {
    set addr_words [split $ip_addr .]
    set step_words [split $ip_addr_step .]
    set index 3
    set result [list]
    set carry 0
    while {$index >= 0} {
        scan [lindex $addr_words $index] "%u" addr_word
        scan [lindex $step_words $index] "%u" step_word
        set value [expr $addr_word + $step_word + $carry]
        set carry [expr $value / 0xFF]
        set value [expr $value % 0xFF]
        lappend result $value
        incr index -1
    }
    set new_addr [format "%u" [lindex $result 3]]
    for {set i 2} {$i >= 0} {incr i -1} {
        append new_addr ".[format "%u" [lindex $result $i]]"
    }
    return $new_addr
}

proc script_increment_ipv4_net {ipAddress {netmask 24} {amount 1}} {
    set ipVal   0
    if {[scan $ipAddress "%d.%d.%d.%d" a b c d] == 4} {
        set ipVal   [format %u [expr ($a<<24)|($b<<16)|($c<<8)|$d]]
    }
    set ipVal [mpexpr {($ipVal >> (32 - $netmask)) + $amount}]
    set ipVal [mpexpr {($ipVal << (32 - $netmask)) & 0xFFFFFFFF}]
    if [catch {set address [format "%s.%s.%s.%s" \
                [expr {(($ipVal >> 24) & 0xff)}] \
                [expr {(($ipVal >> 16) & 0xff)}] \
                [expr {(($ipVal >> 8 ) & 0xff)}] \
                [expr {$ipVal & 0xff}]]} address] {
        set address 0.0.0.0
    }
    return $address
}

package require Ixia

set test_name [info script]

set chassisIP sylvester
set port_list [list 3/2 3/1]

# Provider Side parameters
set p_ip_addr_start 20.30.30.2
set p_ip_addr_step  0.0.0.0
set p_gw_addr       20.30.30.1
set p_ip_prefix     24
set p_rp_addr_start 20.10.10.10
set p_rp_addr_step  0.0.0.0

# Set PE Routers parameters
set pe_count             1
set pe_ip_addr_start     23.34.35.1
set pe_ip_addr_step      0.0.0.1
set pe_dut_ip_addr_start 21.1.1.1
set pe_dut_ip_addr_step  0.0.0.0
set pe_ip_prefix         32
set pe_as_number         1

# Set MVPN/MVRF parameters
set mvrf_count                1
set mvrf_unique               0
set rd_type                   as
set rd_number_start           $pe_as_number
set rd_number_step            0
set rd_assigned_number_start  910
set rd_assigned_number_step   1

set mvpn_source_addr_start    28.0.0.1
set mvpn_source_addr_step     0.0.1.0
set mvpn_source_prefix        32
set mvpn_num_sources          1

set mvpn_group_addr_start     228.0.0.1
set mvpn_group_addr_step      1.0.0.0
set mvpn_group_prefix         32
set mvpn_num_groups           1

set mvpn_rp_addr_start        20.31.31.1
set mvpn_rp_addr_step         0.0.1.0

set mdt_group_addr_start      239.1.1.1
set mdt_group_addr_step       0.0.0.1
set mdt_group_prefix          32


# Set Customer CE Parameters
set ce_ip_addr_start  20.31.31.2
set ce_ip_addr_step   0.0.1.0
set ce_ip_prefix      24
set ce_gw_addr_start  20.31.31.1
set ce_gw_addr_step   0.0.1.0
set ce_vlan_id_start  910
set ce_vlan_id_step   1
set ce_rp_addr_start  29.1.1.1
set ce_rp_addr_step   0.0.1.0

if {$mvrf_unique} {
    set ce_count [mpexpr $pe_count * $mvrf_count]
} else  {
    set ce_count $mvrf_count
}
set total_mvrf_count $ce_count

################################################################################
# Connect to the chassis, reset to factory defaults and take ownership
################################################################################
# When using P2NO HLTSET, for loading the IxTclNetwork package please 
# provide –ixnetwork_tcl_server parameter to ::ixia::connect
set connect_status [::ixia::connect             \
        -reset                                  \
        -ixnetwork_tcl_server   localhost       \
        -device                 $chassisIP      \
        -port_list              $port_list      \
        -username               ixiaApiUser     ]
if {[keylget connect_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget connect_status log]"
}

set port_handle [list]
foreach port $port_list {
    if {![catch {keylget connect_status port_handle.$chassisIP.$port} \
                temp_port]} {
        lappend port_handle $temp_port
    }
}

set ce_port_handle [lindex $port_handle 0]
set pe_port_handle [lindex $port_handle 1]

################################################################################
# Configure interfaces in the test
################################################################################

set interface_status [::ixia::interface_config     \
        -port_handle     $ce_port_handle           \
        -intf_mode       ethernet                  \
        -autonegotiation 1                         \
        -duplex          full                      \
        -speed           ether100                  ]

if {[keylget interface_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget interface_status log]"
}

set interface_status [::ixia::interface_config     \
        -port_handle     $pe_port_handle           \
        -intf_mode       pos_hdlc                  \
        -speed           oc48                       \
        -clocksource     internal                  \
        -framing         sonet                     \
        -rx_fcs          16                        \
        -tx_fcs          16                        \
        -rx_scrambling   0                         \
        -tx_scrambling   0                         \
        -rx_c2           09                        \
        -tx_c2           09                        \
        -op_mode         normal                    ]

if {[keylget interface_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget interface_status log]"
}

################################################################################
# Configure BGP on PE Routers side
################################################################################
set bgp_routers_status [::ixia::emulation_bgp_config                  \
        -mode                            reset                      \
        -port_handle                     $pe_port_handle            \
        -count                           $pe_count                  \
        -local_ip_addr                   $p_ip_addr_start           \
        -remote_ip_addr                  $p_gw_addr                 \
        -local_addr_step                 0.0.0.0                    \
        -remote_addr_step                0.0.0.0                    \
        -local_loopback_ip_addr          $pe_ip_addr_start          \
        -local_loopback_ip_addr_step     $pe_ip_addr_step           \
        -remote_loopback_ip_addr         $pe_dut_ip_addr_start      \
        -remote_loopback_ip_addr_step    $pe_dut_ip_addr_step       \
        -local_router_id                 $pe_ip_addr_start          \
        -neighbor_type                   internal                   \
        -ip_version                      4                          \
        -local_as                        $pe_as_number              \
        -active_connect_enable                                      \
        -ipv4_unicast_nlri                                          \
        -ipv4_multicast_nlri                                        \
        -ipv4_mpls_nlri                                             \
        -ipv4_mpls_vpn_nlri                                         \
        -ipv6_unicast_nlri                                          \
        -ipv6_multicast_nlri                                        \
        -ipv6_mpls_nlri                                             \
        -ipv6_mpls_vpn_nlri                                         ]

if {[keylget bgp_routers_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget bgp_routers_status log]"
}
set pe_bgp_handles [keylget bgp_routers_status handles]

################################################################################
# Configure BGP mVRFs
################################################################################
set mvpn_source_address   $mvpn_source_addr_start
set mvpn_group_address    $mvpn_group_addr_start
set mdt_group_address     $mdt_group_addr_start
set rd_number             $rd_number_start
set rd_assigned_number    $rd_assigned_number_start

foreach {bgp_neighbor} $pe_bgp_handles {
    set mvpn_source_address_start $mvpn_source_address
    for {set i 1} {$i <= $mvrf_count} {incr i} {
        set bgp_route_range_status [::ixia::emulation_bgp_route_config \
                -mode                      add                       \
                -handle                    $bgp_neighbor             \
                -num_sites                 1                         \
                -num_routes                $mvpn_num_sources         \
                -max_route_ranges          1                         \
                -ip_version                4                         \
                -prefix                    $mvpn_source_address      \
                -prefix_step               1                         \
                -prefix_from               $mvpn_source_prefix       \
                -prefix_to                 $mvpn_source_prefix       \
                -route_ip_addr_step        0.0.0.0                   \
                -netmask                   255.255.255.255           \
                -default_mdt_ip            $mdt_group_address        \
                -default_mdt_ip_incr       $mdt_group_addr_step      \
                -label_value               16                        \
                -label_step                0                         \
                -rd_admin_value            $rd_number                \
                -rd_assign_value           $rd_assigned_number       \
                -rd_type                   0                         \
                -target_type               as                        \
                -target                    $rd_number                \
                -target_assign             $rd_assigned_number       \
                -import_target_type        as                        \
                -import_target             $rd_number                \
                -import_target_assign      $rd_assigned_number       \
                -ipv4_unicast_nlri                                   \
                -ipv4_multicast_nlri                                 \
                -ipv4_mpls_nlri                                      \
                -ipv4_mpls_vpn_nlri                                  \
                -origin_route_enable                                 \
                -next_hop_set_mode         same                      \
                -next_hop_enable           1                         ]
        
        if {[keylget bgp_route_range_status status] != $::SUCCESS} {
            return "FAIL - $test_name - [keylget bgp_route_range_status log]"
        }
        
        set mvpn_source_address [script_increment_ipv4_address \
                $mvpn_source_address $mvpn_source_addr_step]
        
        set mvpn_group_address  [script_increment_ipv4_address \
                $mvpn_group_address  $mvpn_group_addr_step]
        
        set mdt_group_address   [script_increment_ipv4_address \
                $mdt_group_address  $mdt_group_addr_step]
        
        set rd_number           [mpexpr $rd_number + $rd_number_step]
        set rd_assigned_number  [mpexpr $rd_assigned_number +  \
                $rd_assigned_number_step]
    }
    if {$mvrf_unique == 0} {
        set mvpn_source_address   $mvpn_source_address_start
        set mvpn_group_address    $mvpn_group_addr_start
        set mdt_group_address     $mdt_group_addr_start
        set rd_number             $rd_number_start
        set rd_assigned_number    $rd_assigned_number_start
        
        for {set i 0} {$i < $mvpn_num_sources} {incr i} {
            set mvpn_source_address   [script_increment_ipv4_net \
                    $mvpn_source_address $mvpn_source_prefix]
        }
    }
}

################################################################################
# Create ISIS Provider
################################################################################
set isis_router_status [::ixia::emulation_isis_config    \
        -mode                           create           \
        -reset                                           \
        -port_handle                    $pe_port_handle  \
        -intf_ip_addr                   $p_ip_addr_start \
        -gateway_ip_addr                $p_gw_addr       \
        -intf_ip_prefix_length          $p_ip_prefix     \
        -count                          1                \
        -intf_metric                    0                \
        -routing_level                  L1L2             \
        -te_enable                      0                \
        -te_router_id                   $p_ip_addr_start ]
if {[keylget isis_router_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget isis_router_status log]"
}
set provider_isis_handle [keylget isis_router_status handle]

################################################################################
# Create ISIS PE routes
################################################################################
set route_config_status [::ixia::emulation_isis_topology_route_config \
        -mode                   create                  \
        -handle                 $provider_isis_handle   \
        -type                   stub                    \
        -ip_version             4                       \
        -stub_ip_start          $pe_ip_addr_start       \
        -stub_ip_step           $pe_ip_addr_step        \
        -stub_ip_pfx_len        $pe_ip_prefix           \
        -stub_count             $pe_count               \
        -stub_metric            1                       \
        -stub_up_down_bit       0                       ]
if {[keylget route_config_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget route_config_status log]"
}

################################################################################
# Create LDP Provider
################################################################################
set ldp_routers_status [::ixia::emulation_ldp_config        \
        -reset                                              \
        -mode                           create              \
        -port_handle                    $pe_port_handle     \
        -count                          1                   \
        -intf_ip_addr                   $p_ip_addr_start    \
        -intf_prefix_length             24                  \
        -gateway_ip_addr                $p_gw_addr          \
        -lsr_id                         $p_ip_addr_start    \
        -label_space                    0                   \
        -label_adv                      unsolicited         \
        -peer_discovery                 link                \
        -hello_interval                 5                   \
        -hello_hold_time                15                  \
        -keepalive_interval             10                  \
        -keepalive_holdtime             30                  \
        -discard_self_adv_fecs          0                   \
        -enable_l2vpn_vc_fecs           1                   \
        -enable_explicit_include_ip_fec 0                   \
        -enable_remote_connect          1                   \
        -enable_vc_group_matching       0                   \
        -targeted_hello_hold_time       45                  \
        -targeted_hello_interval        15                  ]

if {[keylget ldp_routers_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget ldp_routers_status log]"
}
set provider_ldp_handle [keylget ldp_routers_status handle]

################################################################################
# Create LDP PE Routers
################################################################################
set pe_ip_address $pe_ip_addr_start
for {set i 0} {$i < $pe_count} {incr i} {
    set route_config_status [::ixia::emulation_ldp_route_config \
            -mode                   create                   \
            -handle                 $provider_ldp_handle     \
            -fec_type               ipv4_prefix              \
            -label_msg_type         mapping                  \
            -egress_label_mode      nextlabel                \
            -num_lsps               1                        \
            -fec_ip_prefix_start    $pe_ip_address           \
            -fec_ip_prefix_length   32                       \
            -packing_enable         0                        \
            -label_value_start      3                        \
            ]
    
    if {[keylget route_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget route_config_status log]"
    }
    
    set pe_ip_address   [script_increment_ipv4_address \
            $pe_ip_address $pe_ip_addr_step]
}

##############################################################################
#  Configure PIM, P and PE neighbors
##############################################################################
set pim_config_status [::ixia::emulation_pim_config             \
        -mode                       create                      \
        -reset                                                  \
        -port_handle                $pe_port_handle             \
        -count                      1                           \
        -ip_version                 4                           \
        -intf_ip_addr               $p_ip_addr_start            \
        -intf_ip_addr_step          $p_ip_addr_step             \
        -intf_ip_prefix_length      $p_ip_prefix                \
        -dr_priority                0                           \
        -bidir_capable              0                           \
        -hello_interval             30                          \
        -hello_holdtime             105                         \
        -join_prune_interval        60                          \
        -join_prune_holdtime        180                         \
        -prune_delay_enable         1                           \
        -prune_delay                500                         \
        -override_interval          2500                        \
        -gateway_intf_ip_addr       $p_gw_addr                  \
        -gateway_intf_ip_addr_step  0.0.0.0                     \
        -prune_delay_tbit           0                           \
        -send_generation_id         1                           \
        -generation_id_mode         constant                    \
        -mvpn_enable                1                           \
        -mvpn_pe_ip                 $pe_ip_addr_start           \
        -mvpn_pe_ip_incr            $pe_ip_addr_step            \
        -mvpn_pe_count              $pe_count                   \
        -mvrf_count                 $mvrf_count                 \
        -mvrf_unique                0                           \
        -default_mdt_ip             $mdt_group_addr_start       \
        -default_mdt_ip_incr        $mdt_group_addr_step        ]
if {[keylget pim_config_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget pim_config_status log]"
}
set provider_pim_handle [lindex [keylget pim_config_status handle] 0]
set pe_pim_handles      [lrange [keylget pim_config_status handle] 1 end]

################################################################################
# Create PIM MVPN sources on each PE
################################################################################
set mvpn_rp_address      $mvpn_rp_addr_start
set mvpn_source_address  $mvpn_source_addr_start
set mvpn_group_address   $mvpn_group_addr_start
foreach {pe_pim_handle}  $pe_pim_handles {
    # Create multicast source pool
    set pim_config_status [::ixia::emulation_multicast_source_config \
            -mode               create                   \
            -num_sources        $mvpn_num_sources        \
            -ip_addr_start      $mvpn_source_address     \
            -ip_addr_step       $mvpn_source_addr_step   \
            -ip_prefix_len      $mvpn_source_prefix      \
            ]
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    set mvpn_source_pool_handle [keylget pim_config_status handle]
    
    # Create multicast group pool
    set pim_config_status [::ixia::emulation_multicast_group_config \
            -mode               create                   \
            -num_groups         $mvpn_num_groups         \
            -ip_addr_start      $mvpn_group_address      \
            -ip_addr_step       $mvpn_group_addr_step    \
            -ip_prefix_len      $mvpn_group_prefix       \
            ]
    
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    set mvpn_group_pool_handle [keylget pim_config_status handle]
    
    # Create PIM Source
    set pim_config_status [::ixia::emulation_pim_group_config   \
            -mode                   create                      \
            -session_handle         $pe_pim_handle              \
            -group_pool_handle      $mvpn_group_pool_handle     \
            -source_pool_handle     $mvpn_source_pool_handle    \
            -rp_ip_addr             $mvpn_rp_addr_start         \
            -rp_ip_addr_step        $mvpn_rp_addr_step          \
            -group_pool_mode        register                    \
            -join_prune_aggregation_factor 10                   \
            -wildcard_group                 1                   \
            -s_g_rpt_group                  0                   \
            -rate_control                   1                   \
            -interval                       100                 \
            -join_prune_per_interval        99                  \
            -register_per_interval          101                 \
            -register_stop_per_interval     102                 \
            -flap_interval                  60                  \
            -spt_switchover                 0                   \
            -source_group_mapping           fully_meshed        \
            -switch_over_interval           5                   \
            ]
    
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    lappend pe_pim_source_handles [keylget pim_config_status handle]
    if {$mvrf_unique == 0} {
        set mvpn_group_address      $mvpn_group_addr_start
        for {set i 0} {$i < $mvpn_num_sources} {incr i} {
            set mvpn_source_address   [script_increment_ipv4_net \
                    $mvpn_source_address $mvpn_source_prefix]
        }
    } else {
        for {set i 0} {$i < $mvrf_count} {incr i} {
            set mvpn_group_address    [script_increment_ipv4_address \
                    $mvpn_group_address $mvpn_group_addr_step]
            
            set mvpn_source_address   [script_increment_ipv4_address \
                    $mvpn_source_address $mvpn_source_addr_step]
        }
    }
    set mvpn_rp_address   [script_increment_ipv4_address \
            $mvpn_rp_address $mvpn_rp_addr_step]
}

################################################################################
# Create PIM MDT sources on each PE
################################################################################
set mdt_source_address     $pe_ip_addr_start
set mdt_source_addr_step   $pe_ip_addr_step
set mdt_source_prefix      $pe_ip_prefix
set mdt_num_sources        1
set mdt_group_address      $mdt_group_addr_start
foreach {pe_pim_handle} $pe_pim_handles {
    # Create multicat source
    set pim_config_status [::ixia::emulation_multicast_source_config \
            -mode               create                   \
            -num_sources        $mdt_num_sources         \
            -ip_addr_start      $mdt_source_address      \
            -ip_addr_step       $mdt_source_addr_step    \
            -ip_prefix_len      $mdt_source_prefix       \
            ]
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    set mdt_source_pool_handle [keylget pim_config_status handle]
    
    # Create multicast group
    set pim_config_status [::ixia::emulation_multicast_group_config \
            -mode               create              \
            -num_groups         $mvrf_count         \
            -ip_addr_start      $mdt_group_address  \
            -ip_addr_step       0.0.0.0             \
            -ip_prefix_len      32                  \
            ]
    
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    set mdt_group_pool_handle [keylget pim_config_status handle]
    
    # Create PIM Source
    set pim_config_status [::ixia::emulation_pim_group_config   \
            -mode                   create                      \
            -session_handle         $pe_pim_handle              \
            -group_pool_handle      $mdt_group_pool_handle      \
            -source_pool_handle     $mdt_source_pool_handle     \
            -rp_ip_addr             $p_rp_addr_start            \
            -rp_ip_addr_step        0.0.0.0                     \
            -group_pool_mode        register                    \
            -join_prune_aggregation_factor 10                   \
            -wildcard_group                 1                   \
            -s_g_rpt_group                  0                   \
            -rate_control                   1                   \
            -interval                       100                 \
            -join_prune_per_interval        99                  \
            -register_per_interval          101                 \
            -register_stop_per_interval     102                 \
            -register_tx_iteration_gap      6000                \
            -flap_interval                  60                  \
            -spt_switchover                 0                   \
            -source_group_mapping           fully_meshed        \
            -switch_over_interval           5                   \
            -send_null_register             1                   \
            ]
    
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    
    if {$mvrf_unique == 0} {
        set mdt_group_address      $mdt_group_addr_start
        for {set i 0} {$i < $mdt_num_sources} {incr i} {
            set mdt_source_address   [script_increment_ipv4_net \
                    $mdt_source_address $mdt_source_prefix]
        }
    } else {
        for {set i 0} {$i < $mvrf_count} {incr i} {
            set mdt_group_address     [script_increment_ipv4_address \
                    $mdt_group_address $mdt_group_addr_step]
            
            set mvpn_source_address   [script_increment_ipv4_address \
                    $mdt_source_address $mdt_source_addr_step]
        }
    }
}

################################################################################
# Create PIM Provider groups
################################################################################
set mdt_group_address $mdt_group_addr_start
for {set i 0} {$i < $total_mvrf_count} {incr i} {
    set pim_config_status [::ixia::emulation_multicast_group_config \
            -mode               create              \
            -num_groups         1                   \
            -ip_addr_start      $mdt_group_address  \
            -ip_addr_step       0.0.0.0             \
            -ip_prefix_len      32                  \
            ]
    
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    set mdt_group_pool_handle [keylget pim_config_status handle]
    
    set pim_config_status [::ixia::emulation_pim_group_config   \
            -mode                   create                      \
            -session_handle         $provider_pim_handle        \
            -group_pool_handle      $mdt_group_pool_handle      \
            -rp_ip_addr             $p_rp_addr_start            \
            -rp_ip_addr_step        0.0.0.0                     \
            -join_prune_aggregation_factor 10                   \
            -s_g_rpt_group                  0                   \
            -rate_control                   1                   \
            -interval                       100                 \
            -join_prune_per_interval        99                  \
            -register_per_interval          101                 \
            -register_stop_per_interval     102                 \
            -flap_interval                  60                  \
            -spt_switchover                 0                   \
            -source_group_mapping           fully_meshed        \
            -switch_over_interval           5                   \
            ]
    
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    
    set mdt_group_address [script_increment_ipv4_address \
            $mdt_group_address $mdt_group_addr_step]
}

################################################################################
# Configure PIM CE Routers
################################################################################
set pim_config_status [::ixia::emulation_pim_config             \
        -mode                       create                      \
        -reset                                                  \
        -port_handle                $ce_port_handle             \
        -count                      $ce_count                   \
        -ip_version                 4                           \
        -intf_ip_addr               $ce_ip_addr_start           \
        -intf_ip_addr_step          $ce_ip_addr_step            \
        -vlan_id                    $ce_vlan_id_start           \
        -vlan_id_step               $ce_vlan_id_step            \
        -dr_priority                0                           \
        -bidir_capable              0                           \
        -hello_interval             30                          \
        -hello_holdtime             105                         \
        -join_prune_interval        60                          \
        -join_prune_holdtime        180                         \
        -prune_delay_enable         1                           \
        -prune_delay                500                         \
        -override_interval          2500                        \
        -gateway_intf_ip_addr       $ce_gw_addr_start           \
        -gateway_intf_ip_addr_step  $ce_gw_addr_step            \
        -prune_delay_tbit           0                           \
        -send_generation_id         1                           \
        -generation_id_mode         constant                    ]

if {[keylget pim_config_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget pim_config_status log]"
}
set ce_pim_handles    [keylget pim_config_status handle]


################################################################################
# Create CE PIM groups
################################################################################
set mvpn_source_address $mvpn_source_addr_start
set mvpn_group_address  $mvpn_group_addr_start
set mvpn_rp_address     $mvpn_rp_addr_start
foreach {ce_pim_handle} $ce_pim_handles {
    set pim_config_status [::ixia::emulation_multicast_group_config \
            -mode               create                      \
            -num_groups         1                           \
            -ip_addr_start      $mvpn_group_address         \
            -ip_addr_step       $mvpn_group_addr_step       \
            -ip_prefix_len      $mvpn_group_prefix          \
            ]
    
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    set mvpn_group_pool_handle [keylget pim_config_status handle]
    set pim_config_status [::ixia::emulation_pim_group_config   \
            -mode                   create                      \
            -session_handle         $ce_pim_handle              \
            -group_pool_handle      $mvpn_group_pool_handle     \
            -rp_ip_addr             $mvpn_rp_address            \
            -rp_ip_addr_step        0.0.0.0                     \
            -join_prune_aggregation_factor 10                   \
            -s_g_rpt_group                  0                   \
            -rate_control                   1                   \
            -interval                       100                 \
            -join_prune_per_interval        99                  \
            -register_per_interval          101                 \
            -register_stop_per_interval     102                 \
            -flap_interval                  60                  \
            -spt_switchover                 0                   \
            -source_group_mapping           fully_meshed        \
            -switch_over_interval           5                   \
            ]
    
    if {[keylget pim_config_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_config_status log]"
    }
    lappend ce_pim_join_handles [keylget pim_config_status handle]
    set mvpn_rp_address       [script_increment_ipv4_address \
            $mvpn_rp_address $mvpn_rp_addr_step]
    
    set mvpn_source_address   [script_increment_ipv4_address \
            $mvpn_source_address $mvpn_source_addr_step]
    
    set mvpn_group_address    [script_increment_ipv4_address \
            $mvpn_group_address $mvpn_group_addr_step]
}

# Create streams
foreach pe_pim_source_handle_list $pe_pim_source_handles {
    foreach ce_pim_join_handle $ce_pim_join_handles \
            pe_pim_source_handle $pe_pim_source_handle_list {
        set traffic_status [::ixia::traffic_config        \
            -mode                   create                \
            -traffic_generator      ixnetwork             \
            -emulation_src_handle   $pe_pim_source_handle \
            -emulation_dst_handle   $ce_pim_join_handle   \
            -frame_size             512                   \
            -l3_protocol            ipv4                  \
            -pkts_per_burst         2                     \
            -rate_percent           5                     \
            -enforce_min_gap        9                     \
            -tx_delay               10                    \
            -track_by               endpoint_pair         \
            -inter_frame_gap        8                     \
            -inter_burst_gap        11                    \
            -inter_stream_gap       12                    \
            ]
        if {[keylget traffic_status status] != $::SUCCESS} {
            return "FAIL - $test_name - [keylget traffic_status log]"
        }
    }
}

# Start BGP protocol
set bgp_control_status [::ixia::emulation_bgp_control  \
        -port_handle $pe_port_handle \
        -mode        start           ]
if {[keylget bgp_control_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget bgp_control_status log]"
}

# Start ISIS protocol
set isis_emulation_status [::ixia::emulation_isis_control \
        -handle      $provider_isis_handle              \
        -mode        start                              ]

if {[keylget isis_emulation_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget isis_emulation_status log]"
}

# Start LDP protocol
set ldp_control_status  [::ixia::emulation_ldp_control  \
        -port_handle                $pe_port_handle   \
        -mode                       start             ]

if {[keylget ldp_control_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget ldp_control_status log]"
}

# Start PIM protocol
set pim_control_status [::ixia::emulation_pim_control  \
        -handle                $provider_pim_handle  \
        -mode                  start                 ]

if {[keylget pim_control_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget pim_control_status log]"
}

foreach {pim_neighbor} $pe_pim_handles {
    set pim_control_status [::ixia::emulation_pim_control \
            -handle                $pim_neighbor        \
            -mode                  start                ]
    
    if {[keylget pim_control_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_control_status log]"
    }
}

foreach {pim_neighbor} $ce_pim_handles {
    set pim_control_status [::ixia::emulation_pim_control \
            -handle                $pim_neighbor        \
            -mode                  start                ]
    
    if {[keylget pim_control_status status] != $::SUCCESS} {
        return "FAIL - $test_name - [keylget pim_control_status log]"
    }
}

################################################################################
# Verifying that all BGP sessions are established
################################################################################
set continueFlag 0
set retries      10
while {$retries > 0} {
    set bgp_info [::ixia::emulation_bgp_info        \
            -handle      [lindex $pe_bgp_handles 0] \
            -mode        stats                      \
            ]
    if {[keylget bgp_info status] == $::FAILURE} {
        return "FAIL - $test_name - [keylget bgp_info log]."
    }
    if {[keylget bgp_info sessions_established] < $pe_count} {
        incr retries -1
        puts "Verifying the number of BGP sessions established. Retries left $retries ..."
        update idletasks
    } else {
        set continueFlag 1
        break
    }
}
if {$continueFlag != 1} {
    return "FAIL - $test_name - Not all BGP sessions have been established."
}
################################################################################
# Retrieving BGP learned routes
################################################################################
set retries      10
while {$retries > 0} {
    set continueFlag 1
    foreach pe_bgp_handle $pe_bgp_handles {
        set bgp_info [::ixia::emulation_bgp_info\
                -handle $pe_bgp_handle          \
                -mode   labels                  \
                ]
        if {[keylget bgp_info status] == $::FAILURE} {
            return "FAIL - $test_name - [keylget bgp_info log]"
        }
        for {set i 1} {$i <= $mvrf_count} {incr i} {
            if {[catch {keylget bgp_info $i}] || ([keylget bgp_info $i.label] < 0)} {
                set continueFlag 0
                break
            }
        }
    }
    if {$continueFlag == 1} {
        break
    } else {
        puts "Retrieving BGP learned VPN routes. Retries left [expr $retries - 1] ..."
        update idletasks
    }
    incr retries -1
    after 5000
}
if {$continueFlag != 1} {
    return "FAIL - $test_name - There are no BGP VPN learned routes."
}
# Wait for the DUT to learn the routes
after 10000

set port_handle [concat $pe_port_handle $ce_port_handle]

# Start traffic on TX ports
set traffic_status [::ixia::traffic_control  \
        -action                 run        \
        -traffic_generator      ixnetwork  ]
if {[keylget traffic_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget traffic_status log]"
}

# Procedure to print stats
proc post_stats {port_handle label key_list stat_key {stream ""}} {
    puts -nonewline [format "%-30s" $label]
    
    foreach port $port_handle {
        if {$stream != ""} {
            set key $port.stream.$stream.$stat_key
        } else {
            set key $port.$stat_key
        }
        
        if {[llength [keylget key_list $key]] > 1} {
            puts -nonewline "[format "%-16s" N/A]"
        } else  {
            puts -nonewline "[format "%-16s" [keylget key_list $key]]"
        }
    }
    puts ""
}

# Wait for TX port to send traffic
after 10000

# Stop traffic on the TX port
set traffic_stop_status [::ixia::traffic_control \
        -traffic_generator      ixnetwork      \
        -action      stop                      ]
if {[keylget traffic_stop_status status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget traffic_stop_status log]"
}

###############################################################################
#   Retrieve stats
###############################################################################
# Get aggregrate stats for all ports
set aggregate_stats [::ixia::traffic_stats \
        -mode                   aggregate            \
        -traffic_generator      ixnetwork            \
        ]
if {[keylget aggregate_stats status] != $::SUCCESS} {
    return "FAIL - $test_name - [keylget aggregate_stats log]"
}

puts "\n************************* STATISTICS **************************"
puts -nonewline "[format "%-30s" " "]"
foreach port $port_handle {
    puts -nonewline "[format "%-16s" $port]"
}
puts ""
puts -nonewline "[format "%-30s" " "]"
foreach port $port_handle {
    puts -nonewline "[format "%-16s" "-----"]"
}
puts ""


post_stats $port_handle "Packets Tx"     $aggregate_stats \
        aggregate.tx.pkt_count

post_stats $port_handle "Bytes Tx"       $aggregate_stats \
        aggregate.tx.pkt_byte_count

post_stats $port_handle "Packets Rx"     $aggregate_stats \
        aggregate.rx.pkt_count

post_stats $port_handle "Collisions"     $aggregate_stats \
        aggregate.rx.collisions_count

post_stats $port_handle "Data Integrity Frames" $aggregate_stats \
        aggregate.rx.data_int_frames_count

post_stats $port_handle "Data Integrity Error"  $aggregate_stats \
        aggregate.rx.data_int_errors_count

puts "***************************************************************\n"

return "SUCCESS - $test_name - [clock format [clock seconds]]"