X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FTrafficGenerator.py;h=9c66982bb2c9bd0c7f484b3b2ca29fcc2835284a;hb=0a8f41c2178c4b3b6de8ba93ee499d234a8b208c;hp=24bec0af6c75c6dcfc3ac7142d19c4208154ba73;hpb=e6cb3709bd670738471fff40ce13dcb5aff8692f;p=csit.git diff --git a/resources/libraries/python/TrafficGenerator.py b/resources/libraries/python/TrafficGenerator.py index 24bec0af6c..9c66982bb2 100644 --- a/resources/libraries/python/TrafficGenerator.py +++ b/resources/libraries/python/TrafficGenerator.py @@ -11,7 +11,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Traffic generator library.""" +"""Performance testing traffic generator library.""" from robot.api import logger @@ -30,31 +30,84 @@ class TrafficGenerator(object): self._loss = None self._sent = None self._received = None + #T-REX interface order mapping + self._ifaces_reordered = 0 - @staticmethod - def initialize_traffic_generator(node, interface1, interface2): + def initialize_traffic_generator(self, node, interface1, interface2, test_type): """TG initialization :param node: Traffic generator node - :param interface1: PCI address of first interface - :param interface2: PCI address of second interface + :param interface1: interface name of first interface + :param interface2: interface name of second interface + :test_type: 'L2' or 'L3' - src/dst MAC address :type node: dict :type interface1: str :type interface2: str + :type test_type: str :return: nothing """ + trex_path = "/opt/trex-core-1.91" + + topo = Topology() + if node['type'] != NodeType.TG: raise Exception('Node type is not a TG') if node['subtype'] == NodeSubTypeTG.TREX: ssh = SSH() ssh.connect(node) + iface1_pci = topo.get_interface_pci_addr(node, interface1) + iface2_pci = topo.get_interface_pci_addr(node, interface2) + iface1_mac = topo.get_interface_mac(node, interface1) + iface2_mac = topo.get_interface_mac(node, interface2) + if min(iface1_pci, iface2_pci) != iface1_pci: + iface1_mac, iface2_mac = iface2_mac, iface1_mac + iface1_pci, iface2_pci = iface2_pci, iface1_pci + self._ifaces_reordered = 1 + + iface1_mac_hex = "0x"+iface1_mac.replace(":", ",0x") + iface2_mac_hex = "0x"+iface2_mac.replace(":", ",0x") + + if test_type == 'L2': + (ret, stdout, stderr) = ssh.exec_command( + "sudo sh -c 'cat << EOF > /etc/trex_cfg.yaml\n" + "- port_limit : 2\n" + " version : 2\n" + " interfaces : [\"{}\",\"{}\"]\n" + " port_info :\n" + " - dest_mac : [{}]\n" + " src_mac : [{}]\n" + " - dest_mac : [{}]\n" + " src_mac : [{}]\n" + "EOF'"\ + .format(iface1_pci, iface2_pci, + iface2_mac_hex, iface1_mac_hex, + iface1_mac_hex, iface2_mac_hex)) + if int(ret) != 0: + logger.error("failed to create t-rex config: {}"\ + .format(stdout + stderr)) + raise RuntimeError('trex config generation error') + elif test_type == 'L3': + raise NotImplementedError("L3 test_type not supported") + else: + raise Exception("test_type unknown") + + (ret, stdout, stderr) = ssh.exec_command( - "sh -c 'cd /opt/trex-core-1.88/scripts/ && " - "--bind=igb_uio {0} {1}".format(interface1, interface2)) - (ret, stdout, stderr) = ssh.exec_command( - "sh -c 'cd /opt/trex-core-1.88/scripts/ && " - "sudo nohup ./t-rex-64 -i -c 4 --iom 0 > /dev/null 2>&1 &'") + "sh -c 'cd {0}/scripts/ && " + "sudo ./trex-cfg'"\ + .format(trex_path)) + if int(ret) != 0: + logger.error('trex-cfg failed: {0}'.format(stdout + stderr)) + raise RuntimeError('trex-cfg failed') + + (ret, _, _) = ssh.exec_command( + "sh -c 'cd {0}/scripts/ && " + "sudo nohup ./t-rex-64 -i -c 4 --iom 0 > /dev/null 2>&1 &'" + "> /dev/null"\ + .format(trex_path)) + if int(ret) != 0: + raise RuntimeError('t-rex-64 startup failed') @staticmethod def teardown_traffic_generator(node): @@ -78,7 +131,7 @@ class TrafficGenerator(object): :param nodes_info: Dictionary containing information on all nodes in topology. :param duration: Duration of test traffic generation in seconds - :param rate: Percentage of linerate + :param rate: Offered load per interface (e.g. 1%, 3gbps, 4mpps, ...) :param framesize: Frame size (L2) in Bytes :param traffic_type: Traffic profile :type nodes_info: dict @@ -101,55 +154,30 @@ class TrafficGenerator(object): ssh = SSH() ssh.connect(node) - tg_port3_src_mac = Topology.get_interface_mac_by_port_key(node, "port3") - _, adj_int = Topology.\ - get_adjacent_node_and_interface_by_key(nodes_info, node, "port3") - tg_port3_dst_mac = adj_int['mac_address'] - - tg_port5_src_mac = Topology.get_interface_mac_by_port_key(node, "port5") - _, adj_int = Topology.\ - get_adjacent_node_and_interface_by_key(nodes_info, node, "port5") - tg_port5_dst_mac = adj_int['mac_address'] - - if node['subtype'] == NodeSubTypeTG.TREX: if traffic_type in ["3-node-xconnect", "3-node-bridge"]: (ret, stdout, stderr) = ssh.exec_command( - "sh -c '/tmp/openvpp-testing/resources/tools/t-rex-stateless.py " - "-d {0} -r {1}% -s {2} " - "--p1_src_mac 52:00:00:00:00:01 " - "--p1_dst_mac 52:00:00:00:00:02 " + "sh -c '/tmp/openvpp-testing/resources/tools/t-rex/t-rex-stateless.py " + "-d {0} -r {1} -s {2} " "--p1_src_start_ip 10.10.10.1 " "--p1_src_end_ip 10.10.10.254 " "--p1_dst_start_ip 20.20.20.1 " - "--p1_dst_end_ip 20.20.20.254 " - "--p2_src_mac 52:00:00:00:00:02 " - "--p2_dst_mac 52:00:00:00:00:01 " "--p2_src_start_ip 20.20.20.1 " "--p2_src_end_ip 20.20.20.254 " - "--p2_dst_start_ip 10.10.10.1 " - "--p2_dst_end_ip 10.10.10.254'".\ + "--p2_dst_start_ip 10.10.10.1'".\ format(duration, rate, framesize), timeout=int(duration)+60) elif traffic_type in ["3-node-IPv4"]: (ret, stdout, stderr) = ssh.exec_command( - "sh -c '/tmp/openvpp-testing/resources/tools/t-rex-stateless.py " - "-d {0} -r {1}% -s {2} " - "--p1_src_mac {3} " - "--p1_dst_mac {4} " + "sh -c '/tmp/openvpp-testing/resources/tools/t-rex/t-rex-stateless.py " + "-d {0} -r {1} -s {2} " "--p1_src_start_ip 10.10.10.2 " "--p1_src_end_ip 10.10.10.254 " "--p1_dst_start_ip 20.20.20.2 " - "--p1_dst_end_ip 20.20.20.2 " - "--p2_src_mac {5} " - "--p2_dst_mac {6} " "--p2_src_start_ip 20.20.20.2 " "--p2_src_end_ip 20.20.20.254 " - "--p2_dst_start_ip 10.10.10.2 " - "--p2_dst_end_ip 10.10.10.2'".\ - format(duration, rate, framesize,\ - tg_port3_src_mac, tg_port3_dst_mac,\ - tg_port5_src_mac, tg_port5_dst_mac),\ - timeout=int(duration)+60) + "--p2_dst_start_ip 10.10.10.2'".\ + format(duration, rate, framesize),\ + timeout=int(duration)+60) else: raise NotImplementedError('Unsupported traffic type')