libeap/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     # Reject event gets reported twice since we force connect command to be used
 134     # with a driver that supports auth+assoc for testing purposes.
 135     for i in range(0, 2):
 136         ev = wpas.wait_event(["CTRL-EVENT-ASSOC-REJECT"], timeout=15)
 137         if ev is None:
 138             raise Exception("Association rejection timed out")
 139         if "status_code=27" not in ev:
 140             raise Exception("Unexpected rejection status code")
 141
 142     wpas.request("DISCONNECT")
 143     wpas.dump_monitor()
 144
 145 def test_connect_cmd_disconnect_event(dev, apdev):
 146     """Connection using cfg80211 connect command getting disconnected by the AP"""
 147     params = { "ssid": "sta-connect" }
 148     hapd = hostapd.add_ap(apdev[0], params)
 149
 150     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 151     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
 152     wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412")
 153
 154     if "OK" not in hapd.request("DEAUTHENTICATE " + wpas.p2p_interface_addr()):
 155         raise Exception("DEAUTHENTICATE command failed")
 156     ev = wpas.wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=5)
 157     if ev is None:
 158         raise Exception("Disconnection event timed out")
 159     # This event was actually based on deauthenticate event since we force
 160     # connect command to be used with a driver that supports auth+assoc for
 161     # testing purposes. Anyway, wait some time to allow the debug log to capture
 162     # the following NL80211_CMD_DISCONNECT event.
 163     time.sleep(0.1)
 164     wpas.dump_monitor()
 165
 166     # Clean up to avoid causing issue for following test cases
 167     wpas.request("REMOVE_NETWORK all")
 168     wpas.wait_event(["CTRL-EVENT-SCAN-RESULTS"], timeout=2)
 169     wpas.flush_scan_cache()
 170     wpas.dump_monitor()
 171     wpas.interface_remove("wlan5")
 172     del wpas
 173
 174 def test_connect_cmd_roam(dev, apdev):
 175     """cfg80211 connect command to trigger roam"""
 176     params = { "ssid": "sta-connect" }
 177     hostapd.add_ap(apdev[0], params)
 178
 179     wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
 180     wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
 181     wpas.connect("sta-connect", key_mgmt="NONE", scan_freq="2412")
 182     wpas.dump_monitor()
 183
 184     hostapd.add_ap(apdev[1], params)
 185     wpas.scan_for_bss(apdev[1]['bssid'], freq=2412, force_scan=True)
 186     wpas.roam(apdev[1]['bssid'])
 187     time.sleep(0.1)
 188     wpas.request("DISCONNECT")
 189     wpas.wait_disconnected()
 190     wpas.dump_monitor()