X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FDPDK%2FDPDKTools.py;h=b403c3a5f9279ab277e68bbd4410e491e3896b87;hp=5435dffe51c5ce775fb8993d8a5569f292e0bfe2;hb=acab4b148ae7ebe6060bd8df0e3a46d8291fc12a;hpb=c7eb2002bcd007520309feb3e11a26ff847a4e05 diff --git a/resources/libraries/python/DPDK/DPDKTools.py b/resources/libraries/python/DPDK/DPDKTools.py index 5435dffe51..b403c3a5f9 100644 --- a/resources/libraries/python/DPDK/DPDKTools.py +++ b/resources/libraries/python/DPDK/DPDKTools.py @@ -1,4 +1,4 @@ -# Copyright (c) 2016 Cisco and/or its affiliates. +# Copyright (c) 2021 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: @@ -12,80 +12,101 @@ # limitations under the License. -"""This module implements initialization and cleanup of DPDK environment.""" +"""This module implements initialization and cleanup of DPDK framework.""" -from resources.libraries.python.ssh import SSH -from resources.libraries.python.constants import Constants as con -from resources.libraries.python.topology import Topology +from robot.api import logger +from resources.libraries.python.Constants import Constants +from resources.libraries.python.ssh import exec_cmd_no_error +from resources.libraries.python.topology import NodeType, Topology -class DPDKTools(object): + +class DPDKTools: """This class implements: - Initialization of DPDK environment, - Cleanup of DPDK environment. """ @staticmethod - def initialize_dpdk_environment(dut_node, dut_if1, dut_if2): + def initialize_dpdk_framework(node, if1, if2, nic_driver): """ - Initialize the DPDK test environment on the dut_node. - Load the module uio and igb_uio, then bind the test NIC to the igb_uio. - - :param dut_node: Will init the DPDK on this node. - :param dut_if1: DUT interface name. - :param dut_if2: DUT interface name. - :type dut_node: dict - :type dut_if1: str - :type dut_if2: str - :returns: none - :raises RuntimeError: If it fails to bind the interfaces to igb_uio. + Initialize the DPDK framework on the DUT node. Bind interfaces to + driver. + + :param node: DUT node. + :param if1: DUT first interface name. + :param if2: DUT second interface name. + :param nic_driver: Interface driver. + :type node: dict + :type if1: str + :type if2: str + :type nic_driver: str + :raises RuntimeError: If it fails to bind the interfaces to driver. """ - pci_address1 = Topology.get_interface_pci_addr(dut_node, dut_if1) - pci_address2 = Topology.get_interface_pci_addr(dut_node, dut_if2) + if node[u"type"] == NodeType.DUT: + pci_address1 = Topology.get_interface_pci_addr(node, if1) + pci_address2 = Topology.get_interface_pci_addr(node, if2) - ssh = SSH() - ssh.connect(dut_node) + command = f"{Constants.REMOTE_FW_DIR}/{Constants.RESOURCES_LIB_SH}"\ + f"/entry/init_dpdk.sh " \ + f"{nic_driver} {pci_address1} {pci_address2}" + message = u"Initialize the DPDK failed!" + exec_cmd_no_error(node, command, timeout=600, message=message) + + @staticmethod + def cleanup_dpdk_framework(node, if1, if2): + """ + Cleanup the DPDK framework on the DUT node. Bind interfaces to + default driver specified in topology. - arch = Topology.get_node_arch(dut_node) - cmd = 'cd {0}/tests/dpdk/dpdk_scripts/ &&'\ - 'sudo ./init_dpdk.sh {1} {2} {3}' .format(con.REMOTE_FW_DIR, - pci_address1, - pci_address2, - arch) + :param node: Will cleanup the DPDK on this node. + :param if1: DUT first interface name. + :param if2: DUT second interface name. + :type node: dict + :type if1: str + :type if2: str + :raises RuntimeError: If it fails to cleanup the dpdk. + """ + if node[u"type"] == NodeType.DUT: + pci_address1 = Topology.get_interface_pci_addr(node, if1) + pci_address2 = Topology.get_interface_pci_addr(node, if2) + # We are not supporting more than one driver yet. + nic_driver = Topology.get_interface_driver(node, if1) - (ret_code, _, _) = ssh.exec_command(cmd, timeout=600) - if ret_code != 0: - raise RuntimeError('Failed to bind the interfaces to igb_uio at ' - 'node {0}'.format(dut_node['host'])) + command = f"{Constants.REMOTE_FW_DIR}/{Constants.RESOURCES_LIB_SH}"\ + f"/entry/cleanup_dpdk.sh " \ + f"{nic_driver} {pci_address1} {pci_address2}" + message = u"Cleanup the DPDK failed!" + exec_cmd_no_error(node, command, timeout=1200, message=message) @staticmethod - def cleanup_dpdk_environment(dut_node, dut_if1, dut_if2): + def install_dpdk_framework(node): """ - Cleanup the DPDK test environment on the DUT node. - Unbind the NIC from the igb_uio and bind them to the kernel driver. - - :param dut_node: Will cleanup the DPDK on this node. - :param dut_if1: DUT interface name. - :param dut_if2: DUT interface name. - :type dut_node: dict - :type dut_if1: str - :type dut_if2: str - :returns: none - :raises RuntimeError: If it fails to cleanup the dpdk. + Prepare the DPDK framework on the DUT node. + + :param node: Node from topology file. + :type node: dict + :raises RuntimeError: If command returns nonzero return code. + """ + command = f"{Constants.REMOTE_FW_DIR}/{Constants.RESOURCES_LIB_SH}" \ + f"/entry/install_dpdk.sh" + message = u"Install the DPDK failed!" + exec_cmd_no_error(node, command, timeout=3600, message=message) + + command = f"cat {Constants.REMOTE_FW_DIR}/dpdk*/VERSION" + message = u"Get DPDK version failed!" + stdout, _ = exec_cmd_no_error(node, command, message=message) + + logger.info(f"DPDK Version: {stdout}") + + @staticmethod + def install_dpdk_framework_on_all_duts(nodes): + """ + Prepare the DPDK framework on all DUTs. + + :param nodes: Nodes from topology file. + :type nodes: dict """ - pci_address1 = Topology.get_interface_pci_addr(dut_node, dut_if1) - if1_driver = Topology.get_interface_driver(dut_node, dut_if1) - pci_address2 = Topology.get_interface_pci_addr(dut_node, dut_if2) - if2_driver = Topology.get_interface_driver(dut_node, dut_if2) - - ssh = SSH() - ssh.connect(dut_node) - - cmd = 'cd {0}/tests/dpdk/dpdk_scripts/ && sudo ./cleanup_dpdk.sh ' \ - '{1} {2} {3} {4}'.format(con.REMOTE_FW_DIR, if1_driver, - pci_address1, if2_driver, pci_address2) - - (ret_code, _, _) = ssh.exec_command(cmd, timeout=600) - if ret_code != 0: - raise RuntimeError('Failed to cleanup the dpdk at node {0}' - .format(dut_node['host'])) + for node in list(nodes.values()): + if node[u"type"] == NodeType.DUT: + DPDKTools.install_dpdk_framework(node)