1 # Copyright (c) 2016 Cisco and/or its affiliates.
2 # Licensed under the Apache License, Version 2.0 (the "License");
3 # you may not use this file except in compliance with the License.
4 # You may obtain a copy of the License at:
6 # http://www.apache.org/licenses/LICENSE-2.0
8 # Unless required by applicable law or agreed to in writing, software
9 # distributed under the License is distributed on an "AS IS" BASIS,
10 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 # See the License for the specific language governing permissions and
12 # limitations under the License.
14 """Traffic script executor library."""
16 from robot.api import logger
18 from resources.libraries.python.constants import Constants
19 from resources.libraries.python.ssh import SSH
21 __all__ = ['TrafficScriptExecutor']
24 class TrafficScriptExecutor(object):
25 """Traffic script executor utilities."""
29 """Escape quotation mark and dollar mark for shell command.
31 :param string: String to escape.
33 :return: Escaped string.
36 return string.replace('"', '\\"').replace("$", "\\$")
39 def run_traffic_script_on_node(script_file_name, node, script_args,
41 """Run traffic script on the TG node.
43 :param script_file_name: Traffic script name.
44 :param node: Node to run traffic script on.
45 :param script_args: Traffic scripts arguments.
46 :param timeout: Timeout (optional).
47 :type script_file_name: str
49 :type script_args: str
52 logger.trace("{}".format(timeout))
55 cmd = ("cd {}; virtualenv env && " +
56 "export PYTHONPATH=${{PWD}}; " +
57 ". ${{PWD}}/env/bin/activate; " +
58 "resources/traffic_scripts/{} {}") \
59 .format(Constants.REMOTE_FW_DIR, script_file_name,
61 (ret_code, stdout, stderr) = ssh.exec_command_sudo(
62 'sh -c "{}"'.format(TrafficScriptExecutor._escape(cmd)),
64 logger.debug("stdout: {}".format(stdout))
65 logger.debug("stderr: {}".format(stderr))
66 logger.debug("ret_code: {}".format(ret_code))
68 raise Exception("Traffic script execution failed")
71 def traffic_script_gen_arg(rx_if, tx_if, src_mac, dst_mac, src_ip, dst_ip):
72 """Generate traffic script basic arguments string.
74 :param rx_if: Interface that receives traffic.
75 :param tx_if: Interface that sends traffic.
76 :param src_mac: Source MAC address.
77 :param dst_mac: Destination MAC address.
78 :param src_ip: Source IP address.
79 :param dst_ip: Destination IP address.
86 :return: Traffic script arguments string.
89 args = ('--rx_if {0} --tx_if {1} --src_mac {2} --dst_mac {3} --src_ip'
90 ' {4} --dst_ip {5}').format(rx_if, tx_if, src_mac, dst_mac,