/*
* Copyright (c) 2015-2016 Cisco and/or its affiliates.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
option version = "2.0.0";
import "vnet/interface_types.api";
imp# Copyright (c) 2024 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""IP Topology Library."""
from robot.libraries.BuiltIn import BuiltIn
from resources.libraries.python.IPUtil import IPUtil
class IPTopology:
"""IP Topology Library."""
@staticmethod
def initialize_ipv4_forwarding(count=1, pfs=2, route_prefix=32):
"""
Custom setup of IPv4 forwarding with scalability of IP routes on all
DUT nodes in 2-node / 3-node circular topology.
:param count: Number of routes to configure.
:param pfs: Number of physical interfaces to configure.
:param route_prefix: Route prefix to configure.
:type count: int
:type pfs: int
:type route_prefix: int
"""
topology = BuiltIn().get_variable_value("&{topology_info}")
dut = topology["duts"][-1]
ifl = BuiltIn().get_variable_value("${int}")
for l, i in zip(range(pfs // 2), range(1, pfs, 2)):
dut1_int1 = BuiltIn().get_variable_value(f"${{DUT1_{ifl}{i}}}[0]")
dut1_int2 = BuiltIn().get_variable_value(f"${{DUT1_{ifl}{i+1}}}[0]")
dut_int1 = BuiltIn().get_variable_value(f"${{{dut}_{ifl}{i}}}[0]")
dut_int2 = BuiltIn().get_variable_value(f"${{{dut}_{ifl}{i+1}}}[0]")
IPUtil.vpp_add_ip_neighbor(
topology["DUT1"], dut1_int1, f"1.{l}.1.1",
topology[f"TG_pf{i}_mac"][0]
)
if dut == "DUT2":
dut_mac1 = BuiltIn().get_variable_value(
f"${{{dut}_{ifl}{i}_mac}}[0]"
)
IPUtil.vpp_add_ip_neighbor(
topology["DUT1"], dut1_int2, f"3.{l}.3.2", dut_mac1
)
dut_mac2 = BuiltIn().get_variable_value(
f"${{DUT1_{ifl}{i+1}_mac}}[0]"
)
IPUtil.vpp_add_ip_neighbor(
topology["DUT2"], dut_int1, f"3.{l}.3.1", dut_mac2
)
IPUtil.vpp_add_ip_neighbor(
topology[dut], dut_int2, f"2.{l}.2.1",
topology[f"TG_pf{i+1}_mac"][0]
)
IPUtil.vpp_interface_set_ip_address(
topology["DUT1"], dut1_int1, f"1.{l}.1.2", 30
)
if dut == "DUT2":
IPUtil.vpp_interface_set_ip_address(
topology["DUT1"], dut1_int2, f"3.{l}.3.1", 30
)
IPUtil.vpp_interface_set_ip_address(
topology["DUT2"], dut_int1, f"3.{l}.3.2", 30
)
IPUtil.vpp_interface_set_ip_address(
topology[dut], dut_int2, f"2.{l}.2.2", 30
)
IPUtil.vpp_route_add(
topology["DUT1"], f"{i}0.0.0.0", route_prefix,
gateway=f"1.{l}.1.1", interface=dut1_int1, count=count
)
if dut == "DUT2":
IPUtil.vpp_route_add(
topology["DUT1"], f"{i+1}0.0.0.0", route_prefix,
gateway=f"3.{l}.3.2", interface=dut1_int2, count=count
)
IPUtil.vpp_route_add(
topology["DUT2"], f"{i}0.0.0.0", route_prefix,
gateway=f"3.{l}.3.1", interface=dut_int1, count=count
)
IPUtil.vpp_route_add(
topology[dut], f"{i+1}0.0.0.0", route_prefix,
gateway=f"2.{l}.2.1", interface=dut_int2, count=count
)
@staticmethod
def initialize_ipv6_forwarding(count=1, pfs=2, route_prefix=128):
"""
Custom setup of IPv6 forwarding with scalability of IP routes on all
DUT nodes in 2-node / 3-node circular topology.
:param count: Number of routes to configure.
:param pfs: Number of physical interfaces to configure.
:param route_prefix: Route prefix to configure.
:type count: int
:type pfs: int
:type route_prefix: int
"""
topology = BuiltIn().get_variable_value("&{topology_info}")
dut = topology["duts"][-1]
ifl = BuiltIn().get_variable_value("${int}")
for l, i in zip(range(pfs // 2), range(1, pfs, 2)):
dut1_int1 = BuiltIn().get_variable_value(f"${{DUT1_{ifl}{i}}}[0]")
dut1_int2 = BuiltIn().get_variable_value(f"${{DUT1_{ifl}{i+1}}}[0]")
dut_int1 = BuiltIn().get_variable_value(f"${{{dut}_{ifl}{i}}}[0]")
dut_int2 = BuiltIn().get_variable_value(f"${{{dut}_{ifl}{i+1}}}[0]")
IPUtil.vpp_add_ip_neighbor(
topology["DUT1"], dut1_int1, f"2001:{l}::1",
topology[f"TG_pf{i}_mac"][0]
)
if dut == "DUT2":
dut_mac1 = BuiltIn().get_variable_value(
f"${{{dut}_{ifl}{i}_mac}}[0]"
)
IPUtil.vpp_add_ip_neighbor(
topology["DUT1"], dut1_int2, f"2003:{l}::2", dut_mac1
)
dut_mac2 = BuiltIn().get_variable_value(
f"${{DUT1_{ifl}{i+1}_mac}}[0]"
)
IPUtil.vpp_add_ip_neighbor(
topology["DUT2"], dut_int1, f"2003:{l}::1", dut_mac2
)
IPUtil.vpp_add_ip_neighbor(
topology[dut], dut_int2, f"2002:{l}::1",
topology[f"TG_pf{i+1}_mac"][0]
)
IPUtil.vpp_interface_set_ip_address(
topology["DUT1"], dut1_int1, f"2001:{l}::2", 64
)
if dut == "DUT2":
IPUtil.vpp_interface_set_ip_address(
topology["DUT1"], dut1_int2, f"2003:{l}::1", 64
)
IPUtil.vpp_interface_set_ip_address(
topology["DUT2"], dut_int1, f"2003:{l}::2", 64
)
IPUtil.vpp_interface_set_ip_address(
topology[dut], dut_int2, f"2002:{l}::2", 64
)
IPUtil.vpp_route_add(
topology["DUT1"], f"2{i}00::0", route_prefix,
gateway=f"2001:{l}::1", interface=dut1_int1, count=count
)
if dut == "DUT2":
IPUtil.vpp_route_add(
topology["DUT1"], f"2{i+1}00::0", route_prefix,
gateway=f"2003:{l}::2", interface=dut1_int2, count=count
)
IPUtil.vpp_route_add(
topology["DUT2"], f"2{i}00::0", route_prefix,
gateway=f"2003:{l}::1", interface=dut_int1, count=count
)
IPUtil.vpp_route_add(
topology[dut], f"2{i+1}00::0", route_prefix,
gateway=f"2002:{l}::1", interface=dut_int2, count=count
)
