CSIT-183: Vpn baseline routed forwarding (VRF)

[csit.git] / resources / libraries / python / InterfaceUtil.py

diff --git a/resources/libraries/python/InterfaceUtil.py b/resources/libraries/python/InterfaceUtil.py
index 2eea8e6..a16a02f 100644 (file)
--- a/resources/libraries/python/InterfaceUtil.py
+++ b/resources/libraries/python/InterfaceUtil.py
@@ -780,3 +780,37 @@ class InterfaceUtil(object):
             vat.vat_terminal_exec_cmd_from_template(
                 "proxy_arp_intfc_enable.vat",
                 sw_if_index=sw_if_index)
+
+    @staticmethod
+    def vpp_ip_source_check_setup(node, interface):
+        """Setup Reverse Path Forwarding source check on interface.
+
+        :param node: Node to setup RPF source check.
+        :param interface: Interface name to setup RPF source check.
+        :type node: dict
+        :type interface: str
+        """
+        with VatTerminal(node) as vat:
+            vat.vat_terminal_exec_cmd_from_template("ip_source_check.vat",
+                                                    interface_name=interface)
+
+    @staticmethod
+    def assign_interface_to_fib_table(node, interface, table_id):
+        """Assign VPP interface to specific VRF/FIB table.
+
+        :param node: VPP node where the FIB and interface are located.
+        :param interface: Interface to be assigned to FIB.
+        :param table_id: VRF table ID.
+        :type node: dict
+        :type interface: str or int
+        :type table_id: int
+        """
+        if isinstance(interface, basestring):
+            sw_if_index = Topology.get_interface_sw_index(node, interface)
+        else:
+            sw_if_index = interface
+
+        with VatTerminal(node) as vat:
+            vat.vat_terminal_exec_cmd_from_template("set_fib_to_interface.vat",
+                                                    sw_index=sw_if_index,
+                                                    vrf=table_id)