X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FInterfaceUtil.py;h=25503c08df66600046f0c16c8716e057b7708bd5;hp=bc8bf94a9b2b68af2308d923a8362007aa494303;hb=e300d155302d493e0f4cf36b20081a1653909521;hpb=da15035461569ea175aabbac1df735cd5598b0b3 diff --git a/resources/libraries/python/InterfaceUtil.py b/resources/libraries/python/InterfaceUtil.py index bc8bf94a9b..25503c08df 100644 --- a/resources/libraries/python/InterfaceUtil.py +++ b/resources/libraries/python/InterfaceUtil.py @@ -13,9 +13,11 @@ """Interface util library""" +from time import time, sleep +from robot.api import logger from resources.libraries.python.ssh import exec_cmd_no_error from resources.libraries.python.topology import NodeType, Topology -from resources.libraries.python.VatExecutor import VatExecutor +from resources.libraries.python.VatExecutor import VatExecutor, VatTerminal class InterfaceUtil(object): @@ -28,27 +30,164 @@ class InterfaceUtil(object): Function can be used for DUTs as well as for TGs. :param node: node where the interface is - :param interface: interface name + :param interface: interface name or sw_if_index :param state: one of 'up' or 'down' :type node: dict - :type interface: str + :type interface: str or int :type state: str :return: nothing """ if node['type'] == NodeType.DUT: if state == 'up': - state = 'admin-up link-up' + state = 'admin-up' elif state == 'down': - state = 'admin-down link-down' + state = 'admin-down' + else: + raise ValueError('Unexpected interface state: {}'.format(state)) + + if isinstance(interface, basestring): + sw_if_index = Topology.get_interface_sw_index(node, interface) else: - raise Exception('Unexpected interface state: {}'.format(state)) + sw_if_index = interface - sw_if_index = Topology.get_interface_sw_index(node, interface) VatExecutor.cmd_from_template(node, 'set_if_state.vat', sw_if_index=sw_if_index, state=state) - elif node['type'] == NodeType.TG: + elif node['type'] == NodeType.TG or node['type'] == NodeType.VM: cmd = 'ip link set {} {}'.format(interface, state) exec_cmd_no_error(node, cmd, sudo=True) else: - raise Exception('Unknown NodeType: "{}"'.format(node['type'])) + raise Exception('Node {} has unknown NodeType: "{}"'. + format(node['host'], node['type'])) + + @staticmethod + def set_interface_ethernet_mtu(node, interface, mtu): + """Set Ethernet MTU for specified interface. + + Function can be used only for TGs. + + :param node: node where the interface is + :param interface: interface name + :param mtu: MTU to set + :type node: dict + :type interface: str + :type mtu: int + :return: nothing + """ + if node['type'] == NodeType.DUT: + ValueError('Node {}: Setting Ethernet MTU for interface ' + 'on DUT nodes not supported', node['host']) + elif node['type'] == NodeType.TG: + cmd = 'ip link set {} mtu {}'.format(interface, mtu) + exec_cmd_no_error(node, cmd, sudo=True) + else: + raise ValueError('Node {} has unknown NodeType: "{}"'. + format(node['host'], node['type'])) + + @staticmethod + def set_default_ethernet_mtu_on_all_interfaces_on_node(node): + """Set default Ethernet MTU on all interfaces on node. + + Function can be used only for TGs. + + :param node: node where to set default MTU + :type node: dict + :return: nothing + """ + for ifc in node['interfaces'].values(): + InterfaceUtil.set_interface_ethernet_mtu(node, ifc['name'], 1500) + + @staticmethod + def vpp_node_interfaces_ready_wait(node, timeout=10): + """Wait until all interfaces with admin-up are in link-up state. + + :param node: Node to wait on. + :param timeout: Waiting timeout in seconds (optional, default 10s) + :type node: dict + :type timeout: int + :raises: RuntimeError if the timeout period value has elapsed. + """ + if_ready = False + not_ready = [] + start = time() + while not if_ready: + out = InterfaceUtil.vpp_get_interface_data(node) + if time() - start > timeout: + for interface in out: + if interface.get('admin_up_down') == 1: + if interface.get('link_up_down') != 1: + logger.debug('{0} link-down'.format( + interface.get('interface_name'))) + raise RuntimeError('timeout, not up {0}'.format(not_ready)) + not_ready = [] + for interface in out: + if interface.get('admin_up_down') == 1: + if interface.get('link_up_down') != 1: + not_ready.append(interface.get('interface_name')) + if not not_ready: + if_ready = True + else: + logger.debug('Interfaces still in link-down state: {0}, ' + 'waiting...'.format(not_ready)) + sleep(1) + + @staticmethod + def vpp_nodes_interfaces_ready_wait(nodes, timeout=10): + """Wait until all interfaces with admin-up are in link-up state for + listed nodes. + + :param nodes: List of nodes to wait on. + :param timeout: Seconds to wait per node for all interfaces to come up. + :type nodes: list + :type timeout: int + :raises: RuntimeError if the timeout period value has elapsed. + """ + for node in nodes: + InterfaceUtil.vpp_node_interfaces_ready_wait(node, timeout) + + @staticmethod + def all_vpp_interfaces_ready_wait(nodes, timeout=10): + """Wait until all interfaces with admin-up are in link-up state for all + nodes in the topology. + + :param nodes: Nodes in the topology. + :param timeout: Seconds to wait per node for all interfaces to come up. + :type nodes: dict + :type timeout: int + :raises: RuntimeError if the timeout period value has elapsed. + """ + for node in nodes.values(): + if node['type'] == NodeType.DUT: + InterfaceUtil.vpp_node_interfaces_ready_wait(node, timeout) + + @staticmethod + def vpp_get_interface_data(node, interface=None): + """Get all interface data from a VPP node. If a name or + sw_interface_index is provided, return only data for the matching + interface. + :param node: VPP node to get interface data from. + :param interface: Numeric index or name string of a specific interface. + :type node: dict + :type interface: int or str + :return: List of dictionaries containing data for each interface, or a + single dictionary for the specified interface. + :rtype: list or dict + """ + with VatTerminal(node) as vat: + response = vat.vat_terminal_exec_cmd_from_template( + "interface_dump.vat") + + data = response[0] + + if interface is not None: + if isinstance(interface, basestring): + sw_if_index = Topology.get_interface_sw_index(node, interface) + else: + sw_if_index = interface + + for data_if in data: + if data_if["sw_if_index"] == sw_if_index: + + return data_if + + return data