| b.liu | e958203 | 2025-04-17 19:18:16 +0800 | [diff] [blame^] | 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 | "ipv4_test_net" : "192.168.12.0", |
| 28 | "trace_start" : "./tests/trace_start.sh", |
| 29 | "trace_stop" : "./tests/trace_stop.sh", |
| 30 | "perf_start" : "./tests/perf_start.sh", |
| 31 | "perf_stop" : "./tests/perf_stop.sh"} |
| 32 | |
| 33 | # |
| 34 | #devices = [{"hostname": "192.168.254.58", "ifname" : "wlan0", "port": "9877", "name" : "t2-ath9k", "flags" : "AP_HT40 STA_HT40"}, |
| 35 | # {"hostname": "192.168.254.58", "ifname" : "wlan1", "port": "9877", "name" : "t2-ath10k", "flags" : "AP_VHT80"}, |
| 36 | # {"hostname": "192.168.254.58", "ifname" : "wlan3", "port": "9877", "name" : "t2-intel7260", "flags" : "STA_VHT80"}, |
| 37 | # {"hostname": "192.168.254.55", "ifname" : "wlan0, wlan1, wlan2", "port": "", "name" : "t3-monitor"}, |
| 38 | # {"hostname": "192.168.254.50", "ifname" : "wlan0", "port": "9877", "name" : "t1-ath9k"}, |
| 39 | # {"hostname": "192.168.254.50", "ifname" : "wlan1", "port": "9877", "name" : "t1-ath10k"}] |
| 40 | |
| 41 | # |
| 42 | # HWSIM - ifaces available after modprobe mac80211_hwsim |
| 43 | # |
| 44 | devices = [{"hostname": "localhost", "ifname": "wlan0", "port": "9868", "name": "hwsim0", "flags": "AP_VHT80 STA_VHT80"}, |
| 45 | {"hostname": "localhost", "ifname": "wlan1", "port": "9878", "name": "hwsim1", "flags": "AP_VHT80 STA_VHT80"}, |
| 46 | {"hostname": "localhost", "ifname": "wlan2", "port": "9888", "name": "hwsim2", "flags": "AP_VHT80 STA_VHT80"}, |
| 47 | {"hostname": "localhost", "ifname": "wlan3", "port": "9898", "name": "hwsim3", "flags": "AP_VHT80 STA_VHT80"}, |
| 48 | {"hostname": "localhost", "ifname": "wlan4", "port": "9908", "name": "hwsim4", "flags": "AP_VHT80 STA_VHT80"}] |
| 49 | |
| 50 | |
| 51 | def get_setup_params(filename="cfg.py"): |
| 52 | try: |
| 53 | mod = __import__(filename.split(".")[0]) |
| 54 | return mod.setup_params |
| 55 | except: |
| 56 | logger.debug("__import__(" + filename + ") failed, using static settings") |
| 57 | pass |
| 58 | return setup_params |
| 59 | |
| 60 | def get_devices(filename="cfg.py"): |
| 61 | try: |
| 62 | mod = __import__(filename.split(".")[0]) |
| 63 | return mod.devices |
| 64 | except: |
| 65 | logger.debug("__import__(" + filename + ") failed, using static settings") |
| 66 | pass |
| 67 | return devices |
| 68 | |
| 69 | def get_device(devices, name=None, flags=None, lock=False): |
| 70 | if name is None and flags is None: |
| 71 | raise Exception("Failed to get device") |
| 72 | word = name.split(":") |
| 73 | name = word[0] |
| 74 | for device in devices: |
| 75 | if device['name'] == name: |
| 76 | return device |
| 77 | for device in devices: |
| 78 | try: |
| 79 | device_flags = device['flags'] |
| 80 | if device_flags.find(flags) != -1: |
| 81 | return device |
| 82 | except: |
| 83 | pass |
| 84 | raise Exception("Failed to get device " + name) |
| 85 | |
| 86 | def put_device(devices, name): |
| 87 | pass |