tests/hwsim/test_connect_cmd.py

   1 # cfg80211 connect command (SME in the driver/firmware)
   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 time
  10
  11 import hwsim_utils
  12 import hostapd
  13 from wpasupplicant import WpaSupplicant
  14 from p2p_utils import *
  15
  16 def test_connect_cmd_open(dev, apdev):
  17     """Open connection using cfg80211 connect command"""
  18     params = {"ssid": "sta-connect",
  19               "manage_p2p": "1",
  20               "allow_cross_connection": "1"}
  21     hostapd.add_ap(apdev[0], params)
  22
  23     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
  24     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
  25     wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412",
  26                  bg_scan_period="1")
  27     wpas.dump_monitor()
  28     wpas.request("DISCONNECT")
  29     wpas.wait_disconnected()
  30     wpas.dump_monitor()
  31
  32 def test_connect_cmd_wep(dev, apdev):
  33     """WEP Open System using cfg80211 connect command"""
  34     params = {"ssid": "sta-connect-wep", "wep_key0": '"hello"'}
  35     hapd = hostapd.add_ap(apdev[0], params)
  36
  37     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
  38     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
  39     wpas.connect("sta-connect-wep", key_mgmt="NONE", scan_freq="2412",
  40                  wep_key0='"hello"')
  41     wpas.dump_monitor()
  42     hwsim_utils.test_connectivity(wpas, hapd)
  43     wpas.request("DISCONNECT")
  44     wpas.wait_disconnected()
  45     wpas.dump_monitor()
  46
  47 def test_connect_cmd_wep_shared(dev, apdev):
  48     """WEP Shared key using cfg80211 connect command"""
  49     params = {"ssid": "sta-connect-wep", "wep_key0": '"hello"',
  50               "auth_algs": "2"}
  51     hapd = hostapd.add_ap(apdev[0], params)
  52
  53     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
  54     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
  55     id = wpas.connect("sta-connect-wep", key_mgmt="NONE", scan_freq="2412",
  56                       auth_alg="SHARED", wep_key0='"hello"')
  57     wpas.dump_monitor()
  58     hwsim_utils.test_connectivity(wpas, hapd)
  59     wpas.request("DISCONNECT")
  60     wpas.remove_network(id)
  61     wpas.connect("sta-connect-wep", key_mgmt="NONE", scan_freq="2412",
  62                  auth_alg="OPEN SHARED", wep_key0='"hello"')
  63     wpas.dump_monitor()
  64     hwsim_utils.test_connectivity(wpas, hapd)
  65     wpas.request("DISCONNECT")
  66     wpas.wait_disconnected()
  67     wpas.dump_monitor()
  68
  69 def test_connect_cmd_p2p_management(dev, apdev):
  70     """Open connection using cfg80211 connect command and AP using P2P management"""
  71     params = {"ssid": "sta-connect",
  72               "manage_p2p": "1",
  73               "allow_cross_connection": "0"}
  74     hostapd.add_ap(apdev[0], params)
  75
  76     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
  77     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
  78     wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412")
  79     wpas.dump_monitor()
  80     wpas.request("DISCONNECT")
  81     wpas.wait_disconnected()
  82     wpas.dump_monitor()
  83
  84 def test_connect_cmd_wpa2_psk(dev, apdev):
  85     """WPA2-PSK connection using cfg80211 connect command"""
  86     params = hostapd.wpa2_params(ssid="sta-connect", passphrase="12345678")
  87     hostapd.add_ap(apdev[0], params)
  88
  89     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
  90     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
  91     wpas.connect("sta-connect", psk="12345678", scan_freq="2412")
  92     wpas.dump_monitor()
  93     wpas.request("DISCONNECT")
  94     wpas.wait_disconnected()
  95     wpas.dump_monitor()
  96
  97 def test_connect_cmd_concurrent_grpform_while_connecting(dev, apdev):
  98     """Concurrent P2P group formation while connecting to an AP using cfg80211 connect command"""
  99     logger.info("Start connection to an infrastructure AP")
 100     hapd = hostapd.add_ap(apdev[0], {"ssid": "test-open"})
 101
 102     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 103     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
 104     wpas.connect("test-open", key_mgmt="NONE", wait_connect=False)
 105     wpas.dump_monitor()
 106
 107     logger.info("Form a P2P group while connecting to an AP")
 108     wpas.request("SET p2p_no_group_iface 0")
 109
 110     [i_res, r_res] = go_neg_pin_authorized(i_dev=dev[0], i_freq=2412,
 111                                            r_dev=wpas, r_freq=2412)
 112     check_grpform_results(i_res, r_res)
 113     remove_group(dev[0], wpas)
 114     wpas.dump_monitor()
 115
 116     logger.info("Confirm AP connection after P2P group removal")
 117     hwsim_utils.test_connectivity(wpas, hapd)
 118
 119     wpas.request("DISCONNECT")
 120     wpas.wait_disconnected()
 121     wpas.dump_monitor()
 122
 123 def test_connect_cmd_reject_assoc(dev, apdev):
 124     """Connection using cfg80211 connect command getting rejected"""
 125     params = {"ssid": "sta-connect",
 126               "require_ht": "1"}
 127     hostapd.add_ap(apdev[0], params)
 128
 129     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 130     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
 131     wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412",
 132                  disable_ht="1", wait_connect=False)
 133     ev = wpas.wait_event(["CTRL-EVENT-ASSOC-REJECT"], timeout=15)
 134     if ev is None:
 135         raise Exception("Association rejection timed out")
 136     if "status_code=27" not in ev:
 137         raise Exception("Unexpected rejection status code")
 138
 139     wpas.request("DISCONNECT")
 140     wpas.dump_monitor()
 141
 142 def test_connect_cmd_disconnect_event(dev, apdev):
 143     """Connection using cfg80211 connect command getting disconnected by the AP"""
 144     params = {"ssid": "sta-connect"}
 145     hapd = hostapd.add_ap(apdev[0], params)
 146
 147     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 148     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
 149     wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412")
 150
 151     if "OK" not in hapd.request("DEAUTHENTICATE " + wpas.p2p_interface_addr()):
 152         raise Exception("DEAUTHENTICATE command failed")
 153     ev = wpas.wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=5)
 154     if ev is None:
 155         raise Exception("Disconnection event timed out")
 156     # This event was actually based on deauthenticate event since we force
 157     # connect command to be used with a driver that supports auth+assoc for
 158     # testing purposes. Anyway, wait some time to allow the debug log to capture
 159     # the following NL80211_CMD_DISCONNECT event.
 160     time.sleep(0.1)
 161     wpas.dump_monitor()
 162
 163     # Clean up to avoid causing issue for following test cases
 164     wpas.request("REMOVE_NETWORK all")
 165     wpas.wait_event(["CTRL-EVENT-SCAN-RESULTS"], timeout=2)
 166     wpas.flush_scan_cache()
 167     wpas.dump_monitor()
 168     wpas.interface_remove("wlan5")
 169     del wpas
 170
 171 def test_connect_cmd_roam(dev, apdev):
 172     """cfg80211 connect command to trigger roam"""
 173     params = {"ssid": "sta-connect"}
 174     hostapd.add_ap(apdev[0], params)
 175
 176     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 177     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
 178     wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412")
 179     wpas.dump_monitor()
 180
 181     hostapd.add_ap(apdev[1], params)
 182     wpas.scan_for_bss(apdev[1]['bssid'], freq=2412, force_scan=True)
 183     wpas.roam(apdev[1]['bssid'])
 184     time.sleep(0.1)
 185     wpas.request("DISCONNECT")
 186     wpas.wait_disconnected()
 187     wpas.dump_monitor()
 188
 189 def test_connect_cmd_bssid_hint(dev, apdev):
 190     """cfg80211 connect command with bssid_hint"""
 191     params = {"ssid": "sta-connect"}
 192     hostapd.add_ap(apdev[0], params)
 193     hostapd.add_ap(apdev[1], params)
 194
 195     # This does not really give full coverage with mac80211_hwsim since the
 196     # driver does not end up claiming support for driver-based BSS selection.
 197     # Anyway, some test coverage can be achieved for setting the parameter and
 198     # checking that it does not prevent connection with another BSSID.
 199
 200     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 201     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
 202
 203     wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412",
 204                  bssid_hint=apdev[0]['bssid'])
 205     wpas.request("REMOVE_NETWORK all")
 206     wpas.wait_disconnected()
 207     wpas.dump_monitor()
 208
 209     wpas.request("BSS_FLUSH 0")
 210     wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412",
 211                  bssid_hint='22:33:44:55:66:77')
 212     wpas.request("REMOVE_NETWORK all")
 213     wpas.wait_disconnected()
 214     wpas.dump_monitor()
 215
 216     # Additional coverage using ap_scan=2 to prevent scan entry -based selection
 217     # within wpa_supplicant from overriding bssid_hint.
 218
 219     try:
 220         if "OK" not in wpas.request("AP_SCAN 2"):
 221             raise Exception("Failed to set AP_SCAN 2")
 222         wpas.request("BSS_FLUSH 0")
 223         wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412",
 224                      bssid_hint='22:33:44:55:66:77')
 225         wpas.request("REMOVE_NETWORK all")
 226         wpas.wait_disconnected()
 227         wpas.dump_monitor()
 228     finally:
 229         wpas.request("AP_SCAN 1")
 230     wpas.flush_scan_cache()