summaryrefslogtreecommitdiff
path: root/tests/remote/test_devices.py
blob: ccd9984a25f4a1c4dc4dfb96bc1e09b098e2eacc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#!/usr/bin/env python2
#
# Show/check devices
# 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 traceback
import config
import os
import sys
import getopt
import re

import logging
logger = logging.getLogger()

import rutils
from remotehost import Host
from wpasupplicant import WpaSupplicant
import hostapd

def show_devices(devices, setup_params):
    """Show/check available devices"""
    print("Devices:")
    for device in devices:
        host = rutils.get_host(devices, device['name'])
        # simple check if authorized_keys works correctly
        status, buf = host.execute(["id"])
        if status != 0:
            print("[" + host.name + "] - ssh communication:  FAILED")
            continue
        else:
            print("[" + host.name + "] - ssh communication: OK")
        # check setup_hw works correctly
        rutils.setup_hw_host(host, setup_params)

        # show uname
        status, buf = host.execute(["uname", "-s", "-n", "-r", "-m", "-o"])
        print("\t" + buf)
        # show ifconfig
        ifaces = re.split('; | |, ', host.ifname)
        for iface in ifaces:
            status, buf = host.execute(["ifconfig", iface])
            if status != 0:
                print("\t" + iface + " failed\n")
                continue
            lines = buf.splitlines()
            for line in lines:
                print("\t" + line)
        # check hostapd, wpa_supplicant, iperf exist
        status, buf = host.execute([setup_params['wpa_supplicant'], "-v"])
        if status != 0:
            print("\t" + setup_params['wpa_supplicant'] + " not find\n")
            continue
        lines = buf.splitlines()
        for line in lines:
            print("\t" + line)
        print("")
        status, buf = host.execute([setup_params['hostapd'], "-v"])
        if status != 1:
            print("\t" + setup_params['hostapd'] + " not find\n")
            continue
        lines = buf.splitlines()
        for line in lines:
            print("\t" + line)
        print("")
        status, buf = host.execute([setup_params['iperf'], "-v"])
        if status != 0 and status != 1:
            print("\t" + setup_params['iperf'] + " not find\n")
            continue
        lines = buf.splitlines()
        for line in lines:
            print("\t" + line)
        print("")

def check_device(devices, setup_params, dev_name, monitor=False):
    host = rutils.get_host(devices, dev_name)
    # simple check if authorized_keys works correctly
    status, buf = host.execute(["id"])
    if status != 0:
        raise Exception(dev_name + " - ssh communication FAILED: " + buf)

    rutils.setup_hw_host(host, setup_params)

    ifaces = re.split('; | |, ', host.ifname)
    # check interfaces (multi for monitor)
    for iface in ifaces:
        status, buf = host.execute(["ifconfig", iface])
        if status != 0:
            raise Exception(dev_name + " ifconfig " + iface + " failed: " + buf)

    # monitor doesn't need wpa_supplicant/hostapd ...
    if monitor == True:
        return

    status, buf = host.execute(["ls", "-l", setup_params['wpa_supplicant']])
    if status != 0:
        raise Exception(dev_name + " - wpa_supplicant: " + buf)

    status, buf = host.execute(["ls", "-l", setup_params['hostapd']])
    if status != 0:
        raise Exception(dev_name + " - hostapd: " + buf)

    status, buf = host.execute(["which", setup_params['iperf']])
    if status != 0:
        raise Exception(dev_name + " - iperf: " + buf)

    status, buf = host.execute(["which", "tshark"])
    if status != 0:
        logger.debug(dev_name + " - tshark: " + buf)

def check_devices(devices, setup_params, refs, duts, monitors):
    """Check duts/refs/monitors devices"""
    for dut in duts:
        check_device(devices, setup_params, dut)
    for ref in refs:
        check_device(devices, setup_params, ref)
    for monitor in monitors:
        if monitor == "all":
            continue
        check_device(devices, setup_params, monitor, monitor=True)