tests/hwsim/test_sta_dynamic.py

   1 # Dynamic wpa_supplicant interface
   2 # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
   3 #
   4 # This software may be distributed under the terms of the BSD license.
   5 # See README for more details.
   6
   7 import logging
   8 logger = logging.getLogger()
   9 import subprocess
  10 import time
  11
  12 import hwsim_utils
  13 import hostapd
  14 from wpasupplicant import WpaSupplicant
  15
  16 def test_sta_dynamic(dev, apdev):
  17     """Dynamically added wpa_supplicant interface"""
  18     params = hostapd.wpa2_params(ssid="sta-dynamic", passphrase="12345678")
  19     hostapd.add_ap(apdev[0]['ifname'], params)
  20
  21     logger.info("Create a dynamic wpa_supplicant interface and connect")
  22     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
  23     wpas.interface_add("wlan5")
  24
  25     wpas.connect("sta-dynamic", psk="12345678", scan_freq="2412")
  26
  27 def test_sta_ap_scan_0(dev, apdev):
  28     """Dynamically added wpa_supplicant interface with AP_SCAN 0 connection"""
  29     hostapd.add_ap(apdev[0]['ifname'], { "ssid": "test" })
  30     bssid = apdev[0]['bssid']
  31
  32     logger.info("Create a dynamic wpa_supplicant interface and connect")
  33     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
  34     wpas.interface_add("wlan5")
  35
  36     if "OK" not in wpas.request("AP_SCAN 0"):
  37         raise Exception("Failed to set AP_SCAN 2")
  38
  39     id = wpas.connect("", key_mgmt="NONE", bssid=bssid,
  40                       only_add_network=True)
  41     wpas.request("ENABLE_NETWORK " + str(id) + " no-connect")
  42     wpas.request("SCAN")
  43     time.sleep(0.5)
  44     subprocess.call(['sudo', 'iw', wpas.ifname, 'connect', 'test', '2412'])
  45     ev = wpas.wait_event(["CTRL-EVENT-CONNECTED"], timeout=10)
  46     if ev is None:
  47         raise Exception("Connection not reported")
  48     wpas.request("SCAN")
  49     ev = wpas.wait_event(["CTRL-EVENT-CONNECTED"], timeout=5)
  50     if ev is None:
  51         raise Exception("Connection not reported")
  52
  53 def test_sta_ap_scan_2(dev, apdev):
  54     """Dynamically added wpa_supplicant interface with AP_SCAN 2 connection"""
  55     hostapd.add_ap(apdev[0]['ifname'], { "ssid": "test" })
  56     bssid = apdev[0]['bssid']
  57
  58     logger.info("Create a dynamic wpa_supplicant interface and connect")
  59     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
  60     wpas.interface_add("wlan5")
  61
  62     if "FAIL" not in wpas.request("AP_SCAN -1"):
  63         raise Exception("Invalid AP_SCAN -1 accepted")
  64     if "FAIL" not in wpas.request("AP_SCAN 3"):
  65         raise Exception("Invalid AP_SCAN 3 accepted")
  66     if "OK" not in wpas.request("AP_SCAN 2"):
  67         raise Exception("Failed to set AP_SCAN 2")
  68
  69     id = wpas.connect("", key_mgmt="NONE", bssid=bssid,
  70                       only_add_network=True)
  71     wpas.request("ENABLE_NETWORK " + str(id) + " no-connect")
  72     subprocess.call(['sudo', 'iw', wpas.ifname, 'scan', 'trigger',
  73                      'freq', '2412'])
  74     time.sleep(1)
  75     subprocess.call(['sudo', 'iw', wpas.ifname, 'connect', 'test', '2412'])
  76     ev = wpas.wait_event(["CTRL-EVENT-CONNECTED"], timeout=10)
  77     if ev is None:
  78         raise Exception("Connection not reported")
  79
  80     wpas.request("SET disallow_aps bssid " + bssid)
  81     ev = wpas.wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=10)
  82     if ev is None:
  83         raise Exception("Disconnection not reported")
  84
  85     subprocess.call(['sudo', 'iw', wpas.ifname, 'connect', 'test', '2412'])
  86     ev = wpas.wait_event(["CTRL-EVENT-CONNECTED"], timeout=1)
  87     if ev is not None:
  88         raise Exception("Unexpected connection reported")
  89
  90 def test_sta_dynamic_down_up(dev, apdev):
  91     """Dynamically added wpa_supplicant interface down/up"""
  92     params = hostapd.wpa2_params(ssid="sta-dynamic", passphrase="12345678")
  93     hostapd.add_ap(apdev[0]['ifname'], params)
  94
  95     logger.info("Create a dynamic wpa_supplicant interface and connect")
  96     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
  97     wpas.interface_add("wlan5")
  98     wpas.connect("sta-dynamic", psk="12345678", scan_freq="2412")
  99     hwsim_utils.test_connectivity(wpas.ifname, apdev[0]['ifname'])
 100     subprocess.call(['sudo', 'ifconfig', wpas.ifname, 'down'])
 101     ev = wpas.wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=10)
 102     if ev is None:
 103         raise Exception("Disconnection not reported")
 104     if wpas.get_status_field("wpa_state") != "INTERFACE_DISABLED":
 105         raise Exception("Unexpected wpa_state")
 106     subprocess.call(['sudo', 'ifconfig', wpas.ifname, 'up'])
 107     ev = wpas.wait_event(["CTRL-EVENT-CONNECTED"], timeout=15)
 108     if ev is None:
 109         raise Exception("Reconnection not reported")
 110     hwsim_utils.test_connectivity(wpas.ifname, apdev[0]['ifname'])
 111
 112 def test_sta_dynamic_random_mac_addr(dev, apdev):
 113     """Dynamically added wpa_supplicant interface and random MAC address"""
 114     params = hostapd.wpa2_params(ssid="sta-dynamic", passphrase="12345678")
 115     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
 116
 117     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 118     wpas.interface_add("wlan5")
 119     addr0 = wpas.get_driver_status_field("addr")
 120     wpas.request("SET preassoc_mac_addr 1")
 121     wpas.request("SET rand_addr_lifetime 0")
 122
 123     id = wpas.connect("sta-dynamic", psk="12345678", mac_addr="1",
 124                       scan_freq="2412")
 125     addr1 = wpas.get_driver_status_field("addr")
 126
 127     if addr0 == addr1:
 128         raise Exception("Random MAC address not used")
 129
 130     sta = hapd.get_sta(addr0)
 131     if sta['addr'] != "FAIL":
 132         raise Exception("Unexpected STA association with permanent address")
 133     sta = hapd.get_sta(addr1)
 134     if sta['addr'] != addr1:
 135         raise Exception("STA association with random address not found")
 136
 137     wpas.request("DISCONNECT")
 138     wpas.connect_network(id)
 139     addr2 = wpas.get_driver_status_field("addr")
 140     if addr1 != addr2:
 141         raise Exception("Random MAC address changed unexpectedly")
 142
 143     wpas.remove_network(id)
 144     id = wpas.connect("sta-dynamic", psk="12345678", mac_addr="1",
 145                       scan_freq="2412")
 146     addr2 = wpas.get_driver_status_field("addr")
 147     if addr1 == addr2:
 148         raise Exception("Random MAC address did not change")
 149
 150 def test_sta_dynamic_random_mac_addr_scan(dev, apdev):
 151     """Dynamically added wpa_supplicant interface and random MAC address for scan"""
 152     params = hostapd.wpa2_params(ssid="sta-dynamic", passphrase="12345678")
 153     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
 154
 155     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 156     wpas.interface_add("wlan5")
 157     addr0 = wpas.get_driver_status_field("addr")
 158     wpas.request("SET preassoc_mac_addr 1")
 159     wpas.request("SET rand_addr_lifetime 0")
 160
 161     id = wpas.connect("sta-dynamic", psk="12345678", scan_freq="2412")
 162     addr1 = wpas.get_driver_status_field("addr")
 163
 164     if addr0 != addr1:
 165         raise Exception("Random MAC address used unexpectedly")