diff options
Diffstat (limited to 'resources/libraries/python/TrafficGenerator.py')
| -rw-r--r-- | resources/libraries/python/TrafficGenerator.py | 142 |
1 files changed, 133 insertions, 9 deletions
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': |