From b4e85e57cf825684acd24ff1bd54f3512f4aad6f Mon Sep 17 00:00:00 2001 From: Yaroslav Brustinov Date: Wed, 15 Feb 2017 11:19:59 +0200 Subject: update sock.py script: adapt to Python2/3, use Scapy lib instead of removed dpkt, works with IPv4/6, TCP/UDP, in case of TCP checks for out of order and retransmissions. Change-Id: Iab0b7473d915950c834397e365457ecb6f6a8a24 Signed-off-by: Yaroslav Brustinov --- .../unit_tests/outer_packages.py | 48 ++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 scripts/automation/trex_control_plane/unit_tests/outer_packages.py (limited to 'scripts/automation/trex_control_plane/unit_tests/outer_packages.py') diff --git a/scripts/automation/trex_control_plane/unit_tests/outer_packages.py b/scripts/automation/trex_control_plane/unit_tests/outer_packages.py new file mode 100644 index 00000000..77bfd11b --- /dev/null +++ b/scripts/automation/trex_control_plane/unit_tests/outer_packages.py @@ -0,0 +1,48 @@ +import sys +import os +import platform + +EXT_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, os.pardir, 'external_libs')) +if not os.path.exists(EXT_PATH): + raise Exception('Wrong path to external libs: %s' % EXT_PATH) + +CLIENT_UTILS_MODULES = [ + {'name': 'scapy-2.3.1', 'py-dep': True}, + {'name': 'texttable-0.8.4'}, + ] + + +def generate_module_path (module, is_python3, is_64bit, is_ucs2): + platform_path = [module['name']] + + if module.get('py-dep'): + platform_path.append('python3' if is_python3 else 'python2') + + if module.get('arch-dep'): + platform_path.append('ucs2' if is_ucs2 else 'ucs4') + platform_path.append('64bit' if is_64bit else '32bit') + + return os.path.normcase(os.path.join(EXT_PATH, *platform_path)) + + +def import_module_list(modules_list): + + # platform data + is_64bit = platform.architecture()[0] == '64bit' + is_python3 = (sys.version_info >= (3, 0)) + is_ucs2 = (sys.maxunicode == 65535) + + # regular modules + for p in modules_list: + full_path = generate_module_path(p, is_python3, is_64bit, is_ucs2) + + if not os.path.exists(full_path): + print("Unable to find required module library: '{0}'".format(p['name'])) + print("Please provide the correct path using TREX_STL_EXT_PATH variable") + print("current path used: '{0}'".format(full_path)) + exit(1) + if full_path not in sys.path: + sys.path.insert(1, full_path) + + +import_module_list(CLIENT_UTILS_MODULES) -- cgit 1.2.3-korg