[PATCH v4 27/29] tests: remote add utils

[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]

 



Add remote tests utils.

Signed-off-by: Janusz Dziedzic <janusz.dziedzic@xxxxxxxxx>
---
 tests/remote/utils.py | 340 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 340 insertions(+)
 create mode 100644 tests/remote/utils.py

diff --git a/tests/remote/utils.py b/tests/remote/utils.py
new file mode 100644
index 0000000..1521e1d
--- /dev/null
+++ b/tests/remote/utils.py
@@ -0,0 +1,340 @@
+#
+# Utils
+# Copyright (c) 2016, Tieto Corporation
+#
+# This software may be distributed under the terms of the BSD license.
+# See README for more details.
+import time
+import wpaspy
+
+def run_hostapd(dev, setup_params):
+    log_file = None
+    host = wpaspy.Host(host = dev['hostname'],
+                       ifname = dev['ifname'],
+                       port = dev['port'],
+                       name = dev['name'])
+    try:
+        setup_hw = setup_params['setup_hw']
+        try:
+            restart_device = setup_params['restart_device']
+        except:
+            restart_device = "0"
+        host.execute(setup_hw + " -I " + dev['ifname'] + " -R " + restart_device)
+    except:
+        pass
+
+    try:
+        tc_name = setup_params['tc_name']
+        log_dir = setup_params['log_dir']
+        log_file = log_dir + tc_name + "_hostapd_" + host.name + "_" + host.ifname + ".log"
+        host.execute("rm " + log_file)
+        log = " -f " + log_file
+    except:
+        log = ""
+
+    status, buf = host.execute(setup_params['hostapd'] + " -B -ddt -g udp:" + host.port + log)
+    if status != 0:
+        raise Exception("Could not run hostapd: " + buf)
+
+    return host, log_file
+
+def run_wpasupplicant(dev, setup_params):
+    log_file = None
+    host = wpaspy.Host(host = dev['hostname'],
+                       ifname = dev['ifname'],
+                       port = dev['port'],
+                       name = dev['name'])
+    try:
+        setup_hw = setup_params['setup_hw']
+        try:
+            restart_device = setup_params['restart_device']
+        except:
+            restart_device = "0"
+        host.execute(setup_hw + " -I " + dev['ifname'] + " -R " + restart_device)
+    except:
+        pass
+
+    try:
+        tc_name = setup_params['tc_name']
+        log_dir = setup_params['log_dir']
+        log_file = log_dir + tc_name + "_wpa_supplicant_" + host.name + "_" + host.ifname + ".log"
+        host.execute("rm " + log_file)
+        log = " -f " + log_file
+    except:
+        log = ""
+
+    status, buf = host.execute(setup_params['wpa_supplicant'] + " -B -g udp:" + host.port + log)
+    if status != 0:
+        raise Exception("Could not run wpa_supplicant: " + buf)
+
+    return host, log_file
+
+def get_ping_packet_loss(ping_res):
+    loss_line = ""
+    lines = ping_res.splitlines()
+    for line in lines:
+        if line.find("packet loss") != -1:
+            loss_line = line
+            break;
+
+    if loss_line == "":
+        return "100%"
+
+    sections = loss_line.split(",")
+
+    for section in sections:
+        if section.find("packet loss") != -1:
+            words = section.split()
+            return words[0]
+
+    return "100%"
+
+
+def get_ipv4(client, ifname=None):
+    if ifname is None:
+        ifname = client.ifname
+    status, buf = client.execute("ifconfig " + ifname)
+    lines = buf.splitlines()
+
+    for line in lines:
+        res = line.find("inet addr:")
+        if res != -1:
+            break
+
+    if res != -1:
+        words = line.split()
+        addr = words[1].split(":")
+        return addr[1]
+
+    return "unknown"
+
+def get_ipv6(client, ifname=None):
+    res = -1
+    if ifname is None:
+        ifname = client.ifname
+    status, buf = client.execute("ifconfig " + ifname)
+    lines = buf.splitlines()
+
+    for line in lines:
+        res = line.find("Scope:Link")
+        if res != -1:
+            break
+
+    if res != -1:
+        words = line.split()
+        if words[0] == "inet6" and words[1] == "addr:":
+            addr_mask = words[2]
+            addr = addr_mask.split("/")
+            return addr[0]
+
+    return "unknown"
+
+def get_ip(client, addr_type="ipv6", iface=None):
+    if addr_type == "ipv6":
+        return get_ipv6(client, iface)
+    elif addr_type == "ipv4":
+        return get_ipv4(client, iface)
+    else:
+        return "unknown addr_type: " + addr_type
+
+def ac_to_ping_ac(qos):
+    if qos == "be":
+        qos_param = "-Q 0x00"
+    elif qos == "bk":
+        qos_param = "-Q 0x20"
+    elif qos == "vi":
+        qos_param = "-Q 0xA0"
+    elif qos == "vo":
+        qos_param = "-Q 0xE0"
+    else:
+        qos_param = "-Q 0x00"
+    return qos_param
+
+def ping_run(host, ip, result, ifname=None, addr_type="ipv4", deadline="5", qos=None):
+    ping = "ping"
+    if ifname is None:
+       ifname = host.ifname
+    if addr_type == "ipv6":
+        ping = "ping6"
+    if qos:
+        qos = " " + ac_to_ping_ac(qos) + " "
+    else:
+        qos = ""
+    ping = ping + " -w " + deadline + " -I " + ifname + qos + " " + ip
+
+    host.execute("ip -s -s neigh flush all")
+
+    thread = host.execute_run(ping, result)
+    return thread
+
+def ping_wait(host, thread, timeout=None):
+    host.wait_execute_complete(thread, timeout)
+    if thread.isAlive():
+        raise Exception("ping thread still alive")
+
+def check_connectivity(a, b, addr_type = "ipv4", deadline="5", qos=None):
+    addr_a = get_ip(a, addr_type)
+    addr_b = get_ip(b, addr_type)
+
+    if addr_type == "ipv4":
+        ping = "ping"
+    else:
+        ping = "ping6"
+
+    if qos:
+        qos = " " + ac_to_ping_ac(qos) + " "
+    else:
+        qos = ""
+
+    # Clear arp cache
+    a.execute("ip -s -s neigh flush all")
+    b.execute("ip -s -s neigh flush all")
+
+    status, buf = a.execute(ping + " -w " + deadline + qos + " -I " + a.ifname + " " + addr_b)
+    if status == 2 and ping == "ping6":
+        # tentative possible for a while, try again
+        time.sleep(3)
+        status, buf = a.execute(ping + " -w " + deadline + qos + " -I " + a.ifname + " " + addr_b)
+    if status != 0:
+        raise Exception("ping " + a.ifname + " >> " + b.ifname)
+
+    a_b = get_ping_packet_loss(buf)
+
+    status, buf = b.execute(ping + " -w " + deadline + qos + " -I " + b.ifname + " " + addr_a)
+    if status != 0:
+        raise Exception("ping " + b.ifname + " >> " + a.ifname)
+
+    b_a = get_ping_packet_loss(buf)
+
+    if int(a_b[:-1]) > 40:
+        raise Exception("Too high packet lost: " + a_b)
+
+    if int(b_a[:-1]) > 40:
+        raise Exception("Too high packet lost: " + b_a)
+
+    return a_b, b_a
+
+def get_iperf_speed(iperf_res, pattern="Mbits/sec"):
+    lines = iperf_res.splitlines()
+    sum_line = ""
+    last_line = ""
+    count = 0
+    res = -1
+
+    # first find last SUM line
+    for line in lines:
+        res  = line.find("[SUM]")
+        if res != -1:
+            sum_line = line
+
+    # next check SUM status
+    if sum_line != "":
+        words = sum_line.split()
+        for word in words:
+            res = word.find(pattern)
+            if res != -1:
+                return words[count - 1] + " " + pattern
+            count = count + 1
+
+    # no SUM - one thread - find last line
+    for line in lines:
+        res = line.find(pattern)
+        if res != -1:
+            last_line = line
+
+    if last_line == "":
+        return "0 " + pattern
+
+    count = 0
+    words = last_line.split()
+    for word in words:
+        res = word.find(pattern)
+        if res != -1:
+            return words[count - 1] + " " + pattern
+            break;
+        count = count + 1
+    return "0 " + pattern
+
+def ac_to_iperf_ac(qos):
+    if qos == "be":
+        qos_param = "-S 0x00"
+    elif qos == "bk":
+        qos_param = "-S 0x20"
+    elif qos == "vi":
+        qos_param = "-S 0xA0"
+    elif qos == "vo":
+        qos_param = "-S 0xE0"
+    else:
+        qos_param = "-S 0x00"
+    return qos_param
+
+def iperf_run(server, client, server_ip, client_res, server_res,
+              l4="udp", bw="30M", test_time="30", parallel="5",
+              qos="be", param=" -i 5 ", ifname=None, l3="ipv4",
+              port="5001", iperf="iperf"):
+    if ifname == None:
+        ifname = client.ifname
+
+    if iperf == "iperf":
+        iperf_server = iperf
+    elif iperf == "iperf3":
+        iperf_server = iperf + " -1"
+
+    if l3 == "ipv4":
+        iperf_client = iperf + " -c " + server_ip + " -p " + port
+        iperf_server = iperf_server + " -p " + port
+    elif l3 == "ipv6":
+        iperf_client = iperf + " -V -c " + server_ip + "%" + ifname + " -p " + port
+        iperf_server = iperf_server + " -V -p " + port
+    else:
+        return -1, -1
+
+    iperf_server = iperf_server + " -s -f m " + param
+    iperf_client = iperf_client + " -f m -t " + test_time
+
+    if parallel != "1":
+        iperf_client = iperf_client + " -P" + parallel
+
+    if l4 == "udp":
+        if iperf != "iperf3":
+            iperf_server = iperf_server + " -u"
+        iperf_client = iperf_client + " -u -b " + bw
+
+    qos_param = ac_to_iperf_ac(qos)
+    iperf_client = iperf_client + " " + qos_param
+
+    server.execute("ip -s -s neigh flush all")
+    client.execute("ip -s -s neigh flush all")
+
+    server_thread = server.execute_run(iperf_server, server_res)
+    time.sleep(1)
+    client_thread = client.execute_run(iperf_client, client_res)
+
+    return server_thread, client_thread
+
+def iperf_wait(server, client, server_thread, client_thread, timeout=None, iperf="iperf"):
+    client.wait_execute_complete(client_thread, timeout)
+    if client_thread.isAlive():
+        raise Exception("iperf client thread still alive")
+
+    server.wait_execute_complete(server_thread, 5)
+    if server_thread.isAlive():
+        server.execute("killall -s INT " + iperf)
+        time.sleep(1)
+
+    server.wait_execute_complete(server_thread, 5)
+    if server_thread.isAlive():
+        raise Execption("iperf server thread still alive")
+
+    return
+
+def get_ipv4_addr(setup_params, number):
+    try:
+        ipv4_base = setup_params['ipv4_test_net']
+    except:
+        ipv4_base = "172.16.12.0"
+
+    parts = ipv4_base.split('.')
+    ipv4 = parts[0] + "." + parts[1] + "." + parts[2] + "." + str(number)
+
+    return ipv4
-- 
1.9.1


_______________________________________________
Hostap mailing list
Hostap@xxxxxxxxxxxxxxxxxxx
http://lists.infradead.org/mailman/listinfo/hostap



[Index of Archives]     [Linux Wireless]     [Linux Kernel]     [ATH6KL]     [Linux Bluetooth]     [Linux Netdev]     [Kernel Newbies]     [IDE]     [Security]     [Git]     [Netfilter]     [Bugtraq]     [Yosemite News]     [MIPS Linux]     [ARM Linux]     [Linux Security]     [Linux RAID]     [Linux ATA RAID]     [Samba]     [Device Mapper]

  Powered by Linux