tests/hwsim/test_ap_roam.py

   1 # Roaming tests
   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 time
   8 import logging
   9 logger = logging.getLogger()
  10
  11 import hwsim_utils
  12 import hostapd
  13
  14 def test_ap_roam_open(dev, apdev):
  15     """Roam between two open APs"""
  16     hapd0 = hostapd.add_ap(apdev[0], { "ssid": "test-open" })
  17     dev[0].connect("test-open", key_mgmt="NONE")
  18     hwsim_utils.test_connectivity(dev[0], hapd0)
  19     hapd1 = hostapd.add_ap(apdev[1], { "ssid": "test-open" })
  20     dev[0].scan(type="ONLY")
  21     dev[0].roam(apdev[1]['bssid'])
  22     hwsim_utils.test_connectivity(dev[0], hapd1)
  23     dev[0].roam(apdev[0]['bssid'])
  24     hwsim_utils.test_connectivity(dev[0], hapd0)
  25
  26 def test_ap_roam_open_failed(dev, apdev):
  27     """Roam failure due to rejected authentication"""
  28     hapd0 = hostapd.add_ap(apdev[0], { "ssid": "test-open" })
  29     dev[0].connect("test-open", key_mgmt="NONE", scan_freq="2412")
  30     hwsim_utils.test_connectivity(dev[0], hapd0)
  31     params = { "ssid": "test-open", "max_num_sta" : "0" }
  32     hapd1 = hostapd.add_ap(apdev[1], params)
  33     bssid = hapd1.own_addr()
  34
  35     dev[0].scan_for_bss(bssid, freq=2412)
  36     dev[0].dump_monitor()
  37     if "OK" not in dev[0].request("ROAM " + bssid):
  38         raise Exception("ROAM failed")
  39
  40     ev = dev[0].wait_event(["CTRL-EVENT-AUTH-REJECT"], 1)
  41     if not ev:
  42         raise Exception("CTRL-EVENT-AUTH-REJECT was not seen")
  43
  44     dev[0].wait_connected(timeout=5)
  45     hwsim_utils.test_connectivity(dev[0], hapd0)
  46
  47 def test_ap_roam_wpa2_psk(dev, apdev):
  48     """Roam between two WPA2-PSK APs"""
  49     params = hostapd.wpa2_params(ssid="test-wpa2-psk", passphrase="12345678")
  50     hapd0 = hostapd.add_ap(apdev[0], params)
  51     dev[0].connect("test-wpa2-psk", psk="12345678")
  52     hwsim_utils.test_connectivity(dev[0], hapd0)
  53     hapd1 = hostapd.add_ap(apdev[1], params)
  54     dev[0].scan(type="ONLY")
  55     dev[0].roam(apdev[1]['bssid'])
  56     hwsim_utils.test_connectivity(dev[0], hapd1)
  57     dev[0].roam(apdev[0]['bssid'])
  58     hwsim_utils.test_connectivity(dev[0], hapd0)
  59
  60 def test_ap_roam_wpa2_psk_failed(dev, apdev, params):
  61     """Roam failure with WPA2-PSK AP due to wrong passphrase"""
  62     params = hostapd.wpa2_params(ssid="test-wpa2-psk", passphrase="12345678")
  63     hapd0 = hostapd.add_ap(apdev[0], params)
  64     id = dev[0].connect("test-wpa2-psk", psk="12345678", scan_freq="2412")
  65     hwsim_utils.test_connectivity(dev[0], hapd0)
  66     params['wpa_passphrase'] = "22345678"
  67     hapd1 = hostapd.add_ap(apdev[1], params)
  68     bssid = hapd1.own_addr()
  69     dev[0].scan_for_bss(bssid, freq=2412)
  70
  71     dev[0].dump_monitor()
  72     if "OK" not in dev[0].request("ROAM " + bssid):
  73         raise Exception("ROAM failed")
  74
  75     ev = dev[0].wait_event(["CTRL-EVENT-SSID-TEMP-DISABLED",
  76                             "CTRL-EVENT-CONNECTED"], 5)
  77     if "CTRL-EVENT-CONNECTED" in ev:
  78         raise Exception("Got unexpected CTRL-EVENT-CONNECTED")
  79     if "CTRL-EVENT-SSID-TEMP-DISABLED" not in ev:
  80         raise Exception("CTRL-EVENT-SSID-TEMP-DISABLED not seen")
  81
  82     if "OK" not in dev[0].request("SELECT_NETWORK id=" + str(id)):
  83         raise Exception("SELECT_NETWORK failed")
  84
  85     ev = dev[0].wait_event(["CTRL-EVENT-SSID-REENABLED"], 3)
  86     if not ev:
  87         raise Exception("CTRL-EVENT-SSID-REENABLED not seen");
  88
  89     dev[0].wait_connected(timeout=5)
  90     hwsim_utils.test_connectivity(dev[0], hapd0)
  91
  92 def test_ap_reassociation_to_same_bss(dev, apdev):
  93     """Reassociate to the same BSS"""
  94     hapd = hostapd.add_ap(apdev[0], { "ssid": "test-open" })
  95     dev[0].connect("test-open", key_mgmt="NONE")
  96
  97     dev[0].request("REASSOCIATE")
  98     dev[0].wait_connected(timeout=10, error="Reassociation timed out")
  99     hwsim_utils.test_connectivity(dev[0], hapd)
 100
 101     dev[0].request("REATTACH")
 102     dev[0].wait_connected(timeout=10, error="Reattach timed out")
 103     hwsim_utils.test_connectivity(dev[0], hapd)
 104
 105     # Wait for previous scan results to expire to trigger new scan
 106     time.sleep(5)
 107     dev[0].request("REATTACH")
 108     dev[0].wait_connected(timeout=10, error="Reattach timed out")
 109     hwsim_utils.test_connectivity(dev[0], hapd)
 110
 111 def test_ap_roam_set_bssid(dev, apdev):
 112     """Roam control"""
 113     hostapd.add_ap(apdev[0], { "ssid": "test-open" })
 114     hostapd.add_ap(apdev[1], { "ssid": "test-open" })
 115     id = dev[0].connect("test-open", key_mgmt="NONE", bssid=apdev[1]['bssid'],
 116                         scan_freq="2412")
 117     if dev[0].get_status_field('bssid') != apdev[1]['bssid']:
 118         raise Exception("Unexpected BSS")
 119     # for now, these are just verifying that the code path to indicate
 120     # within-ESS roaming changes can be executed; the actual results of those
 121     # operations are not currently verified (that would require a test driver
 122     # that does BSS selection)
 123     dev[0].set_network(id, "bssid", "")
 124     dev[0].set_network(id, "bssid", apdev[0]['bssid'])
 125     dev[0].set_network(id, "bssid", apdev[1]['bssid'])
 126
 127 def test_ap_roam_wpa2_psk_race(dev, apdev):
 128     """Roam between two WPA2-PSK APs and try to hit a disconnection race"""
 129     params = hostapd.wpa2_params(ssid="test-wpa2-psk", passphrase="12345678")
 130     hapd0 = hostapd.add_ap(apdev[0], params)
 131     dev[0].connect("test-wpa2-psk", psk="12345678", scan_freq="2412")
 132     hwsim_utils.test_connectivity(dev[0], hapd0)
 133
 134     params['channel'] = '2'
 135     hapd1 = hostapd.add_ap(apdev[1], params)
 136     dev[0].scan_for_bss(apdev[1]['bssid'], freq=2417)
 137     dev[0].roam(apdev[1]['bssid'])
 138     hwsim_utils.test_connectivity(dev[0], hapd1)
 139     dev[0].roam(apdev[0]['bssid'])
 140     hwsim_utils.test_connectivity(dev[0], hapd0)
 141     # Wait at least two seconds to trigger the previous issue with the
 142     # disconnection callback.
 143     for i in range(3):
 144         time.sleep(0.8)
 145         hwsim_utils.test_connectivity(dev[0], hapd0)