X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FIPv6Setup.py;fp=resources%2Flibraries%2Fpython%2FIPv6Setup.py;h=45a8eba58dab6fff3de93f9ef43179dc271794a4;hp=0000000000000000000000000000000000000000;hb=33499c81c94c2d3baef9d3e9f061cd76ef86fa74;hpb=5cbeca02602061d32212e14f289d65cf648920e4

diff --git a/resources/libraries/python/IPv6Setup.py b/resources/libraries/python/IPv6Setup.py
new file mode 100644
index 0000000000..45a8eba58d
--- /dev/null
+++ b/resources/libraries/python/IPv6Setup.py
@@ -0,0 +1,289 @@
+# Copyright (c) 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.
+
+"""Library to set up IPv6 in topology."""
+
+from ssh import SSH
+from ipaddress import IPv6Network
+from topology import NodeType
+from topology import Topology
+from constants import Constants
+
+
+class IPv6Networks(object):
+    """IPv6 network iterator.
+
+       :param networks: List of the available IPv6 networks.
+       :type networks: list
+    """
+    def __init__(self, networks):
+        self._networks = list()
+        for network in networks:
+            net = IPv6Network(unicode(network))
+            self._networks.append(net)
+        num = len(self._networks)
+        if num == 0:
+            raise Exception('No IPv6 networks')
+
+    def next_network(self):
+        """Get the next elemnt of the iterator.
+
+           :return: IPv6 network.
+           :rtype: IPv6Network object
+           :raises: StopIteration if there is no more elements.
+        """
+        if len(self._networks):
+            return self._networks.pop()
+        else:
+            raise StopIteration()
+
+
+class IPv6Setup(object):
+    """IPv6 setup in topology."""
+
+    def __init__(self):
+        pass
+
+    def nodes_setup_ipv6_addresses(self, nodes, nodes_addr):
+        """Setup IPv6 addresses on all VPP nodes in topology.
+
+           :param nodes: Nodes of the test topology.
+           :param nodes_addr: Available nodes IPv6 adresses.
+           :type nodes: dict
+           :type nodes_addr: dict
+        """
+        for net in nodes_addr.values():
+            for port in net['ports'].values():
+                host = port.get('node')
+                if host is None:
+                    continue
+                topo = Topology()
+                node = topo.get_node_by_hostname(nodes, host)
+                if node is None:
+                    continue
+                if node['type'] == NodeType.DUT:
+                    self.vpp_set_if_ipv6_addr(node, port['if'], port['addr'],
+                                              net['prefix'])
+
+    def nodes_clear_ipv6_addresses(self, nodes, nodes_addr):
+        """Remove IPv6 addresses from all VPP nodes in topology.
+
+           :param nodes: Nodes of the test topology.
+           :param nodes_addr: Available nodes IPv6 adresses.
+           :type nodes: dict
+           :type nodes_addr: dict
+         """
+        for net in nodes_addr.values():
+            for port in net['ports'].values():
+                host = port.get('node')
+                if host is None:
+                    continue
+                topo = Topology()
+                node = topo.get_node_by_hostname(nodes, host)
+                if node is None:
+                    continue
+                if node['type'] == NodeType.DUT:
+                    self.vpp_del_if_ipv6_addr(node, port['if'], port['addr'],
+                                              net['prefix'])
+
+    @staticmethod
+    def linux_set_if_ipv6_addr(node, interface, addr, prefix):
+        """Set IPv6 address on linux host.
+
+           :param node: Linux node.
+           :param interface: Node interface.
+           :param addr: IPv6 address.
+           :param prefix: IPv6 address prefix.
+           :type node: dict
+           :type interface: str
+           :type addr: str
+           :type prefix: str
+        """
+        ssh = SSH()
+        ssh.connect(node)
+
+        cmd = "ifconfig {dev} inet6 add {ip}/{p} up".format(dev=interface,
+                                                            ip=addr, p=prefix)
+        (ret_code, _, _) = ssh.exec_command_sudo(cmd)
+        if int(ret_code) != 0:
+            raise Exception('TG ifconfig failed')
+
+    @staticmethod
+    def linux_del_if_ipv6_addr(node, interface, addr, prefix):
+        """Delete IPv6 address on linux host.
+
+           :param node: Linux node.
+           :param interface: Node interface.
+           :param addr: IPv6 address.
+           :param prefix: IPv6 address prefix.
+           :type node: dict
+           :type interface: str
+           :type addr: str
+           :type prefix: str
+        """
+        ssh = SSH()
+        ssh.connect(node)
+
+        cmd = "ifconfig {dev} inet6 del {ip}/{p}".format(dev=interface,
+                                                         ip=addr,
+                                                         p=prefix)
+        (ret_code, _, _) = ssh.exec_command_sudo(cmd)
+        if int(ret_code) != 0:
+            raise Exception('TG ifconfig failed')
+
+        cmd = "ifconfig {dev} down".format(dev=interface)
+        (ret_code, _, _) = ssh.exec_command_sudo(cmd)
+        if int(ret_code) != 0:
+            raise Exception('TG ifconfig failed')
+
+    @staticmethod
+    def vpp_set_if_ipv6_addr(node, interface, addr, prefix):
+        """Set IPv6 address on VPP.
+
+           :param node: VPP node.
+           :param interface: Node interface.
+           :param addr: IPv6 address.
+           :param prefix: IPv6 address prefix.
+           :type node: dict
+           :type interface: str
+           :type addr: str
+           :type prefix: str
+        """
+        ssh = SSH()
+        ssh.connect(node)
+
+        cmd = '{c}'.format(c=Constants.VAT_BIN_NAME)
+        cmd_input = 'sw_interface_add_del_address {dev} {ip}/{p}'.format(
+            dev=interface, ip=addr, p=prefix)
+        (ret_code, _, _) = ssh.exec_command_sudo(cmd, cmd_input)
+        if int(ret_code) != 0:
+            raise Exception('VPP sw_interface_add_del_address failed on {h}'
+                            .format(h=node['host']))
+
+        cmd_input = 'sw_interface_set_flags {dev} admin-up'.format(
+            dev=interface)
+        (ret_code, _, _) = ssh.exec_command_sudo(cmd, cmd_input)
+        if int(ret_code) != 0:
+            raise Exception('VPP sw_interface_set_flags failed on {h}'.format(
+                h=node['host']))
+
+    @staticmethod
+    def vpp_del_if_ipv6_addr(node, interface, addr, prefix):
+        """Delete IPv6 address on VPP.
+
+           :param node: VPP node.
+           :param interface: Node interface.
+           :param addr: IPv6 address.
+           :param prefix: IPv6 address prefix.
+           :type node: dict
+           :type interface: str
+           :type addr: str
+           :type prefix: str
+        """
+        ssh = SSH()
+        ssh.connect(node)
+
+        cmd = '{c}'.format(c=Constants.VAT_BIN_NAME)
+        cmd_input = 'sw_interface_add_del_address {dev} {ip}/{p} del'.format(
+            dev=interface, ip=addr, p=prefix)
+        (ret_code, _, _) = ssh.exec_command_sudo(cmd, cmd_input)
+        if int(ret_code) != 0:
+            raise Exception(
+                'sw_interface_add_del_address failed on {h}'.
+                format(h=node['host']))
+
+        cmd_input = 'sw_interface_set_flags {dev} admin-down'.format(
+            dev=interface)
+        (ret_code, _, _) = ssh.exec_command_sudo(cmd, cmd_input)
+        if int(ret_code) != 0:
+            raise Exception('VPP sw_interface_set_flags failed on {h}'.format(
+                h=node['host']))
+
+    @staticmethod
+    def vpp_ra_supress_link_layer(node, interface):
+        """Supress ICMPv6 router advertisement message for link scope address
+
+           :param node: VPP node.
+           :param interface: Interface name.
+           :type node: dict
+           :type interface: str
+        """
+        ssh = SSH()
+        ssh.connect(node)
+
+        cmd = '{c}'.format(c=Constants.VAT_BIN_NAME)
+        cmd_input = 'exec ip6 nd {0} ra-surpress-link-layer'.format(
+            interface)
+        (ret_code, _, _) = ssh.exec_command_sudo(cmd, cmd_input)
+        if int(ret_code) != 0:
+            raise Exception("'{0}' failed on {1}".format(cmd_input,
+                                                         node['host']))
+
+    def vpp_all_ra_supress_link_layer(self, nodes):
+        """Supress ICMPv6 router advertisement message for link scope address
+           on all VPP nodes in the topology
+
+           :param nodes: Nodes of the test topology.
+           :type nodes: dict
+        """
+        for node in nodes.values():
+            if node['type'] == NodeType.TG:
+                continue
+            for port_k, port_v in node['interfaces'].items():
+                if port_k == 'mgmt':
+                    continue
+                if_name = port_v.get('name')
+                if if_name is None:
+                    continue
+                self.vpp_ra_supress_link_layer(node, if_name)
+
+    @staticmethod
+    def vpp_ipv6_route_add(node, link, interface, nodes_addr):
+        """Setup IPv6 route on the VPP node.
+
+           :param node: Node to add route on.
+           :param link: Route to following link.
+           :param interface: Route output interface.
+           :param nodes_addr: Available nodes IPv6 adresses.
+           :type node: dict
+           :type link: str
+           :type interface: str
+           :type nodes_addr: dict
+        """
+        ssh = SSH()
+        ssh.connect(node)
+
+        # Get route destination address from link name
+        net = nodes_addr.get(link)
+        if net is None:
+            raise ValueError('No network for link "{0}"'.format(link))
+        dst_net = '{0}/{1}'.format(net['net_addr'], net['prefix'])
+
+        # Get next-hop address
+        nh_addr = None
+        for net in nodes_addr.values():
+            for port in net['ports'].values():
+                if port['if'] == interface and port['node'] == node['host']:
+                    for nh in net['ports'].values():
+                        if nh['if'] != interface and nh['node'] != node['host']:
+                            nh_addr = nh['addr']
+        if nh_addr is None:
+            raise Exception('next-hop not found')
+
+        cmd_input = 'ip_add_del_route {0} via {1} {2} resolve-attempts 10'. \
+            format(dst_net, nh_addr, interface)
+        (ret_code, _, _) = ssh.exec_command_sudo(Constants.VAT_BIN_NAME,
+                                                 cmd_input)
+        if int(ret_code) != 0:
+            raise Exception("'{0}' failed on {1}".format(cmd_input,
+                                                         node['host']))