From e6cb3709bd670738471fff40ce13dcb5aff8692f Mon Sep 17 00:00:00 2001 From: Stefan Kobza Date: Fri, 12 Feb 2016 15:28:57 +0100 Subject: Last bulk update of CSIT. Change-Id: I815e4d54e74a49fd19a9927554ce5c37a2719f7e Signed-off-by: Stefan Kobza --- resources/libraries/python/IPv6Setup.py | 61 +++++------ resources/libraries/python/Routing.py | 45 ++++++++ resources/libraries/python/TrafficGenerator.py | 142 +++++++++++++++++++++++-- resources/libraries/python/topology.py | 63 +++++++++++ 4 files changed, 267 insertions(+), 44 deletions(-) create mode 100644 resources/libraries/python/Routing.py (limited to 'resources/libraries/python') diff --git a/resources/libraries/python/IPv6Setup.py b/resources/libraries/python/IPv6Setup.py index 45a8eba58d..db497757bc 100644 --- a/resources/libraries/python/IPv6Setup.py +++ b/resources/libraries/python/IPv6Setup.py @@ -248,42 +248,33 @@ class IPv6Setup(object): 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 + def get_link_address(link, nodes_addr): + """Get link IPv6 address. + + :param link: Link name. + :param nodes_addr: Available nodes IPv6 adresses. + :type link: str + :type nodes_addr: dict + :return: Link IPv6 address. + :rtype: str """ - 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']) + raise ValueError('Link "{0}" address not found'.format(link)) + return net.get('net_addr') - # 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'])) + @staticmethod + def get_link_prefix(link, nodes_addr): + """Get link IPv6 address prefix. + + :param link: Link name. + :param nodes_addr: Available nodes IPv6 adresses. + :type link: str + :type nodes_addr: dict + :return: Link IPv6 address prefix. + :rtype: int + """ + net = nodes_addr.get(link) + if net is None: + raise ValueError('Link "{0}" address not found'.format(link)) + return net.get('prefix') diff --git a/resources/libraries/python/Routing.py b/resources/libraries/python/Routing.py new file mode 100644 index 0000000000..b22516a7c4 --- /dev/null +++ b/resources/libraries/python/Routing.py @@ -0,0 +1,45 @@ +# 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. + +"""Routing utilities library.""" + +from VatExecutor import VatTerminal +from topology import Topology + + +class Routing(object): + """Routing utilities.""" + + @staticmethod + def vpp_route_add(node, network, prefix_len, gateway, interface): + """Add route to the VPP node. + + :param node: Node to add route on. + :param network: Route destination network address. + :param prefix_len: Route destination network prefix length. + :param gateway: Route gateway address. + :param interface: Route interface. + :type node: str + :type network: str + :type prefix_len: int + :type gateway: str + :type interface: str + """ + sw_if_index = Topology.get_interface_sw_index(node, interface) + vat = VatTerminal(node) + vat.vat_terminal_exec_cmd_from_template('add_route.vat', + network=network, + prefix_length=prefix_len, + gateway=gateway, + sw_if_index=sw_if_index) + vat.vat_terminal_close() diff --git a/resources/libraries/python/TrafficGenerator.py b/resources/libraries/python/TrafficGenerator.py index d86917a181..24bec0af6c 100644 --- a/resources/libraries/python/TrafficGenerator.py +++ b/resources/libraries/python/TrafficGenerator.py @@ -10,12 +10,20 @@ # 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. -from ssh import SSH + +"""Traffic generator library.""" + from robot.api import logger +from resources.libraries.python.ssh import SSH +from resources.libraries.python.topology import NodeType +from resources.libraries.python.topology import NodeSubTypeTG +from resources.libraries.python.topology import Topology + __all__ = ['TrafficGenerator'] class TrafficGenerator(object): + """Traffic Generator""" def __init__(self): self._result = None @@ -23,18 +31,130 @@ class TrafficGenerator(object): self._sent = None self._received = None + @staticmethod + def initialize_traffic_generator(node, interface1, interface2): + """TG initialization + :param node: Traffic generator node + :param interface1: PCI address of first interface + :param interface2: PCI address of second interface + :type node: dict + :type interface1: str + :type interface2: str + :return: nothing + """ + + if node['type'] != NodeType.TG: + raise Exception('Node type is not a TG') + if node['subtype'] == NodeSubTypeTG.TREX: + ssh = SSH() + ssh.connect(node) + + (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 &'") + + @staticmethod + def teardown_traffic_generator(node): + """TG teardown + :param node: Traffic generator node + :type node: dict + :return: nothing + """ + + if node['type'] != NodeType.TG: + raise Exception('Node type is not a TG') + if node['subtype'] == NodeSubTypeTG.TREX: + ssh = SSH() + ssh.connect(node) + (ret, stdout, stderr) = ssh.exec_command( + "sh -c 'sudo pkill t-rex'") + + def send_traffic_on(self, nodes_info, duration, rate, + framesize, traffic_type): + """Send traffic from all configured interfaces on TG + :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 framesize: Frame size (L2) in Bytes + :param traffic_type: Traffic profile + :type nodes_info: dict + :type duration: str + :type rate: str + :type framesize: str + :type traffic_type: str + :return: TG output + :rtype: str + """ + + node = nodes_info["TG"] + + if node['type'] != NodeType.TG: + raise Exception('Node type is not a TG') + + if node['subtype'] is None: + raise Exception('TG subtype not defined') - def send_traffic_on(self, node, tx_port, rx_port, duration, rate, - framesize): ssh = SSH() ssh.connect(node) - (ret, stdout, stderr) = ssh.exec_command( - "sh -c 'cd MoonGen && sudo -S build/MoonGen " - "rfc2544/benchmarks/vpp-frameloss.lua --txport 0 --rxport 1 " - "--duration {0} --rate {1} --framesize {2}'".format( - duration, rate, framesize), - timeout=int(duration)+60) + 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 " + "--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'".\ + 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} " + "--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) + else: + raise NotImplementedError('Unsupported traffic type') + + else: + raise NotImplementedError("TG subtype not supported") logger.trace(ret) logger.trace(stdout) @@ -51,6 +171,10 @@ class TrafficGenerator(object): return self._result def no_traffic_loss_occured(self): + """Fail is loss occured in traffic run + :return: nothing + """ + if self._loss is None: raise Exception('The traffic generation has not been issued') if self._loss != '0': diff --git a/resources/libraries/python/topology.py b/resources/libraries/python/topology.py index 2b202e5b4a..3ced69d7fe 100644 --- a/resources/libraries/python/topology.py +++ b/resources/libraries/python/topology.py @@ -43,6 +43,14 @@ class NodeType(object): # Traffic Generator (this node has traffic generator on it) TG = 'TG' +class NodeSubTypeTG(object): + #T-Rex traffic generator + TREX = 'TREX' + # Moongen + MOONGEN = 'MOONGEN' + #IxNetwork + IXNET = 'IXNET' + DICT__nodes = load_topo_from_yaml() @@ -345,6 +353,22 @@ class Topology(object): return None + @staticmethod + def get_interface_mac_by_port_key(node, port_key): + """Get MAC address for the interface based on port key. + + :param node: Node to get interface mac on. + :param port_key: Dictionary key name of interface. + :type node: dict + :type port_key: str + :return: Return MAC or None if not found. + """ + for port_name, port_data in node['interfaces'].iteritems(): + if port_name == port_key: + return port_data['mac_address'] + + return None + @staticmethod def get_interface_mac(node, interface): """Get MAC address for the interface. @@ -364,6 +388,45 @@ class Topology(object): return None + @staticmethod + def get_adjacent_node_and_interface_by_key(nodes_info, node, port_key): + """Get node and interface adjacent to specified interface + on local network. + + :param nodes_info: Dictionary containing information on all nodes + in topology. + :param node: Node that contains specified interface. + :param port_key: Interface port key. + :type nodes_info: dict + :type node: dict + :type port_key: str + :return: Return (node, interface info) tuple or None if not found. + :rtype: (dict, dict) + """ + link_name = None + # get link name where the interface belongs to + for port_name, port_data in node['interfaces'].iteritems(): + if port_name == 'mgmt': + continue + if port_name == port_key: + link_name = port_data['link'] + break + + if link_name is None: + return None + + # find link + for node_data in nodes_info.values(): + # skip self + if node_data['host'] == node['host']: + continue + for interface, interface_data \ + in node_data['interfaces'].iteritems(): + if 'link' not in interface_data: + continue + if interface_data['link'] == link_name: + return node_data, node_data['interfaces'][interface] + @staticmethod def get_adjacent_node_and_interface(nodes_info, node, interface_name): """Get node and interface adjacent to specified interface -- cgit 1.2.3-korg