1 # Environment configuration
2 # Copyright (c) 2016, Tieto Corporation
3 #
4 # This software may be distributed under the terms of the BSD license.
5 # See README for more details.
6 
7 #
8 # Currently static definition, in the future this could be a config file,
9 # or even common database with host management.
10 #
11 
12 import logging
13 logger = logging.getLogger()
14 
15 #
16 # You can put your settings in cfg.py file with setup_params, devices
17 # definitions in the format as below. In other case HWSIM cfg will be used.
18 #
19 setup_params = {"setup_hw" : "./tests/setup_hw.sh",
20                 "hostapd" : "./tests/hostapd-rt",
21                 "wpa_supplicant" : "./tests/wpa_supplicant-rt",
22                 "iperf" : "iperf",
23                 "wlantest" : "./tests/wlantest",
24                 "wlantest_cli" : "./tests/wlantest_cli",
25                 "country" : "US",
26                 "log_dir" : "/tmp/",
27                 "remote_cli" : True,
28                 "ipv4_test_net" : "192.168.12.0",
29                 "trace_start" : "./tests/trace_start.sh",
30                 "trace_stop" : "./tests/trace_stop.sh",
31                 "perf_start" : "./tests/perf_start.sh",
32                 "perf_stop" : "./tests/perf_stop.sh"}
33 
34 #
35 #devices = [{"hostname": "192.168.254.58", "ifname" : "wlan0", "port": "9877", "name" : "t2-ath9k", "flags" : "AP_HT40 STA_HT40"},
36 #           {"hostname": "192.168.254.58", "ifname" : "wlan1", "port": "9877", "name" : "t2-ath10k", "flags" : "AP_VHT80"},
37 #           {"hostname": "192.168.254.58", "ifname" : "wlan3", "port": "9877", "name" : "t2-intel7260", "flags" : "STA_VHT80"},
38 #           {"hostname": "192.168.254.55", "ifname" : "wlan0, wlan1, wlan2", "port": "", "name" : "t3-monitor"},
39 #           {"hostname": "192.168.254.50", "ifname" : "wlan0", "port": "9877", "name" : "t1-ath9k"},
40 #           {"hostname": "192.168.254.50", "ifname" : "wlan1", "port": "9877", "name" : "t1-ath10k"}]
41 
42 #
43 # HWSIM - ifaces available after modprobe mac80211_hwsim
44 #
45 devices = [{"hostname": "localhost", "ifname": "wlan0", "port": "9868", "name": "hwsim0", "flags": "AP_VHT80 STA_VHT80"},
46            {"hostname": "localhost", "ifname": "wlan1", "port": "9878", "name": "hwsim1", "flags": "AP_VHT80 STA_VHT80"},
47            {"hostname": "localhost", "ifname": "wlan2", "port": "9888", "name": "hwsim2", "flags": "AP_VHT80 STA_VHT80"},
48            {"hostname": "localhost", "ifname": "wlan3", "port": "9898", "name": "hwsim3", "flags": "AP_VHT80 STA_VHT80"},
49            {"hostname": "localhost", "ifname": "wlan4", "port": "9908", "name": "hwsim4", "flags": "AP_VHT80 STA_VHT80"}]
50 
51 
52 def get_setup_params(filename="cfg.py"):
53     try:
54        mod = __import__(filename.split(".")[0])
55        return mod.setup_params
56     except:
57        logger.debug("__import__(" + filename + ") failed, using static settings")
58        pass
59     return setup_params
60 
61 def get_devices(filename="cfg.py"):
62     try:
63        mod = __import__(filename.split(".")[0])
64        return mod.devices
65     except:
66        logger.debug("__import__(" + filename + ") failed, using static settings")
67        pass
68     return devices
69 
70 def get_device(devices, name=None, flags=None, lock=False):
71     if name is None and flags is None:
72         raise Exception("Failed to get device")
73     word = name.split(":")
74     name = word[0]
75     for device in devices:
76         if device['name'] == name:
77             return device
78     for device in devices:
79         try:
80             device_flags = device['flags']
81             if device_flags.find(flags) != -1:
82                 return device
83         except:
84             pass
85     raise Exception("Failed to get device " + name)
86 
87 def put_device(devices, name):
88     pass
89