1 # Protected management frames tests
2 # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
4 # This software may be distributed under the terms of the BSD license.
5 # See README for more details.
7 from remotehost import remote_compatible
10 logger = logging.getLogger()
14 from wlantest import Wlantest
15 from wpasupplicant import WpaSupplicant
18 def test_ap_pmf_required(dev, apdev):
19 """WPA2-PSK AP with PMF required"""
20 ssid = "test-pmf-required"
21 params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
22 params["wpa_key_mgmt"] = "WPA-PSK-SHA256";
23 params["ieee80211w"] = "2";
24 hapd = hostapd.add_ap(apdev[0], params)
28 wt.add_passphrase("12345678")
29 key_mgmt = hapd.get_config()['key_mgmt']
30 if key_mgmt.split(' ')[0] != "WPA-PSK-SHA256":
31 raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
32 dev[0].connect(ssid, psk="12345678", ieee80211w="1",
33 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
35 if "[WPA2-PSK-SHA256-CCMP]" not in dev[0].request("SCAN_RESULTS"):
36 raise Exception("Scan results missing RSN element info")
37 hwsim_utils.test_connectivity(dev[0], hapd)
38 dev[1].connect(ssid, psk="12345678", ieee80211w="2",
39 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
41 hwsim_utils.test_connectivity(dev[1], hapd)
42 hapd.request("SA_QUERY " + dev[0].p2p_interface_addr())
43 hapd.request("SA_QUERY " + dev[1].p2p_interface_addr())
44 wt.require_ap_pmf_mandatory(apdev[0]['bssid'])
45 wt.require_sta_pmf(apdev[0]['bssid'], dev[0].p2p_interface_addr())
46 wt.require_sta_pmf_mandatory(apdev[0]['bssid'], dev[1].p2p_interface_addr())
48 if wt.get_sta_counter("valid_saqueryresp_tx", apdev[0]['bssid'],
49 dev[0].p2p_interface_addr()) < 1:
50 raise Exception("STA did not reply to SA Query")
51 if wt.get_sta_counter("valid_saqueryresp_tx", apdev[0]['bssid'],
52 dev[1].p2p_interface_addr()) < 1:
53 raise Exception("STA did not reply to SA Query")
56 def test_ap_pmf_optional(dev, apdev):
57 """WPA2-PSK AP with PMF optional"""
58 ssid = "test-pmf-optional"
59 params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
60 params["wpa_key_mgmt"] = "WPA-PSK";
61 params["ieee80211w"] = "1";
62 hapd = hostapd.add_ap(apdev[0], params)
66 wt.add_passphrase("12345678")
67 dev[0].connect(ssid, psk="12345678", ieee80211w="1",
68 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
70 hwsim_utils.test_connectivity(dev[0], hapd)
71 dev[1].connect(ssid, psk="12345678", ieee80211w="2",
72 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
74 hwsim_utils.test_connectivity(dev[1], hapd)
75 wt.require_ap_pmf_optional(apdev[0]['bssid'])
76 wt.require_sta_pmf(apdev[0]['bssid'], dev[0].p2p_interface_addr())
77 wt.require_sta_pmf_mandatory(apdev[0]['bssid'], dev[1].p2p_interface_addr())
80 def test_ap_pmf_optional_2akm(dev, apdev):
81 """WPA2-PSK AP with PMF optional (2 AKMs)"""
82 ssid = "test-pmf-optional-2akm"
83 params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
84 params["wpa_key_mgmt"] = "WPA-PSK WPA-PSK-SHA256";
85 params["ieee80211w"] = "1";
86 hapd = hostapd.add_ap(apdev[0], params)
90 wt.add_passphrase("12345678")
91 dev[0].connect(ssid, psk="12345678", ieee80211w="1",
92 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
94 hwsim_utils.test_connectivity(dev[0], hapd)
95 dev[1].connect(ssid, psk="12345678", ieee80211w="2",
96 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
98 hwsim_utils.test_connectivity(dev[1], hapd)
99 wt.require_ap_pmf_optional(apdev[0]['bssid'])
100 wt.require_sta_pmf(apdev[0]['bssid'], dev[0].p2p_interface_addr())
101 wt.require_sta_key_mgmt(apdev[0]['bssid'], dev[0].p2p_interface_addr(),
103 wt.require_sta_pmf_mandatory(apdev[0]['bssid'], dev[1].p2p_interface_addr())
104 wt.require_sta_key_mgmt(apdev[0]['bssid'], dev[1].p2p_interface_addr(),
108 def test_ap_pmf_negative(dev, apdev):
109 """WPA2-PSK AP without PMF (negative test)"""
110 ssid = "test-pmf-negative"
111 params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
112 hapd = hostapd.add_ap(apdev[0], params)
116 wt.add_passphrase("12345678")
117 dev[0].connect(ssid, psk="12345678", ieee80211w="1",
118 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
120 hwsim_utils.test_connectivity(dev[0], hapd)
122 dev[1].connect(ssid, psk="12345678", ieee80211w="2",
123 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
125 hwsim_utils.test_connectivity(dev[1], hapd)
126 raise Exception("PMF required STA connected to no PMF AP")
128 logger.debug("Ignore expected exception: " + str(e))
129 wt.require_ap_no_pmf(apdev[0]['bssid'])
132 def test_ap_pmf_assoc_comeback(dev, apdev):
133 """WPA2-PSK AP with PMF association comeback"""
134 ssid = "assoc-comeback"
135 params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
136 params["wpa_key_mgmt"] = "WPA-PSK-SHA256";
137 params["ieee80211w"] = "2";
138 hapd = hostapd.add_ap(apdev[0], params)
142 wt.add_passphrase("12345678")
143 dev[0].connect(ssid, psk="12345678", ieee80211w="1",
144 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
146 hapd.set("ext_mgmt_frame_handling", "1")
147 dev[0].request("DISCONNECT")
148 dev[0].wait_disconnected(timeout=10)
149 hapd.set("ext_mgmt_frame_handling", "0")
150 dev[0].request("REASSOCIATE")
151 dev[0].wait_connected(timeout=10, error="Timeout on re-connection")
152 if wt.get_sta_counter("assocresp_comeback", apdev[0]['bssid'],
153 dev[0].p2p_interface_addr()) < 1:
154 raise Exception("AP did not use association comeback request")
157 def test_ap_pmf_assoc_comeback2(dev, apdev):
158 """WPA2-PSK AP with PMF association comeback (using DROP_SA)"""
159 ssid = "assoc-comeback"
160 params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
161 params["wpa_key_mgmt"] = "WPA-PSK";
162 params["ieee80211w"] = "1";
163 hapd = hostapd.add_ap(apdev[0], params)
167 wt.add_passphrase("12345678")
168 dev[0].connect(ssid, psk="12345678", ieee80211w="2",
169 key_mgmt="WPA-PSK", proto="WPA2", scan_freq="2412")
170 if "OK" not in dev[0].request("DROP_SA"):
171 raise Exception("DROP_SA failed")
172 dev[0].request("REASSOCIATE")
173 dev[0].wait_connected(timeout=10, error="Timeout on re-connection")
174 if wt.get_sta_counter("reassocresp_comeback", apdev[0]['bssid'],
175 dev[0].p2p_interface_addr()) < 1:
176 raise Exception("AP did not use reassociation comeback request")
178 def test_ap_pmf_sta_sa_query(dev, apdev):
179 """WPA2-PSK AP with station using SA Query"""
180 ssid = "assoc-comeback"
181 addr = dev[0].own_addr()
183 wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
184 wpas.interface_add("wlan5", drv_params="use_monitor=1")
185 id = wpas.add_network()
186 wpas.set_network(id, "mode", "2")
187 wpas.set_network_quoted(id, "ssid", ssid)
188 wpas.set_network(id, "proto", "WPA2")
189 wpas.set_network(id, "key_mgmt", "WPA-PSK-SHA256")
190 wpas.set_network(id, "ieee80211w", "2")
191 wpas.set_network_quoted(id, "psk", "12345678")
192 wpas.set_network(id, "pairwise", "CCMP")
193 wpas.set_network(id, "group", "CCMP")
194 wpas.set_network(id, "frequency", "2412")
195 wpas.connect_network(id)
196 bssid = wpas.own_addr()
202 wt.add_passphrase("12345678")
204 dev[0].connect(ssid, psk="12345678", ieee80211w="1",
205 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
208 wpas.request("DEAUTHENTICATE " + addr + " test=0")
210 wpas.request("DISASSOCIATE " + addr + " test=0")
212 ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=1)
214 raise Exception("Unexpected disconnection")
216 wpas.request("DEAUTHENTICATE " + addr + " reason=6 test=0")
218 wpas.request("DISASSOCIATE " + addr + " reason=7 test=0")
220 ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=1)
222 raise Exception("Unexpected disconnection")
223 if wt.get_sta_counter("valid_saqueryreq_tx", bssid, addr) < 1:
224 raise Exception("STA did not send SA Query")
225 if wt.get_sta_counter("valid_saqueryresp_rx", bssid, addr) < 1:
226 raise Exception("AP did not reply to SA Query")
229 def test_ap_pmf_sta_sa_query_no_response(dev, apdev):
230 """WPA2-PSK AP with station using SA Query and getting no response"""
231 ssid = "assoc-comeback"
232 addr = dev[0].own_addr()
234 wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
235 wpas.interface_add("wlan5", drv_params="use_monitor=1")
236 id = wpas.add_network()
237 wpas.set_network(id, "mode", "2")
238 wpas.set_network_quoted(id, "ssid", ssid)
239 wpas.set_network(id, "proto", "WPA2")
240 wpas.set_network(id, "key_mgmt", "WPA-PSK-SHA256")
241 wpas.set_network(id, "ieee80211w", "2")
242 wpas.set_network_quoted(id, "psk", "12345678")
243 wpas.set_network(id, "pairwise", "CCMP")
244 wpas.set_network(id, "group", "CCMP")
245 wpas.set_network(id, "frequency", "2412")
246 wpas.connect_network(id)
247 bssid = wpas.own_addr()
250 dev[0].connect(ssid, psk="12345678", ieee80211w="1",
251 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
254 wpas.request("DEAUTHENTICATE " + addr + " test=0")
256 wpas.request("DISASSOCIATE " + addr + " test=0")
258 ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=1)
260 raise Exception("Unexpected disconnection")
262 wpas.request("SET ext_mgmt_frame_handling 1")
263 wpas.request("DEAUTHENTICATE " + addr + " reason=6 test=0")
265 wpas.request("DISASSOCIATE " + addr + " reason=7 test=0")
267 dev[0].wait_disconnected()
269 wpas.request("SET ext_mgmt_frame_handling 0")
270 dev[0].wait_connected()
273 def test_ap_pmf_sta_unprot_deauth_burst(dev, apdev):
274 """WPA2-PSK AP with station receiving burst of unprotected Deauthentication frames"""
275 ssid = "deauth-attack"
276 addr = dev[0].own_addr()
278 wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
279 wpas.interface_add("wlan5", drv_params="use_monitor=1")
280 id = wpas.add_network()
281 wpas.set_network(id, "mode", "2")
282 wpas.set_network_quoted(id, "ssid", ssid)
283 wpas.set_network(id, "proto", "WPA2")
284 wpas.set_network(id, "key_mgmt", "WPA-PSK-SHA256")
285 wpas.set_network(id, "ieee80211w", "2")
286 wpas.set_network_quoted(id, "psk", "12345678")
287 wpas.set_network(id, "pairwise", "CCMP")
288 wpas.set_network(id, "group", "CCMP")
289 wpas.set_network(id, "frequency", "2412")
290 wpas.connect_network(id)
291 bssid = wpas.own_addr()
296 wt.add_passphrase("12345678")
298 dev[0].connect(ssid, psk="12345678", ieee80211w="1",
299 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
302 for i in range(0, 10):
303 wpas.request("DEAUTHENTICATE " + addr + " reason=6 test=0")
304 wpas.request("DISASSOCIATE " + addr + " reason=7 test=0")
305 ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=1)
307 raise Exception("Unexpected disconnection")
308 num_req = wt.get_sta_counter("valid_saqueryreq_tx", bssid, addr)
309 num_resp = wt.get_sta_counter("valid_saqueryresp_rx", bssid, addr)
311 raise Exception("STA did not send SA Query")
313 raise Exception("AP did not reply to SA Query")
315 raise Exception("STA initiated too many SA Query procedures (%d)" % num_req)
318 for i in range(0, 5):
319 wpas.request("DEAUTHENTICATE " + addr + " reason=6 test=0")
320 wpas.request("DISASSOCIATE " + addr + " reason=7 test=0")
321 ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=1)
323 raise Exception("Unexpected disconnection")
324 num_req = wt.get_sta_counter("valid_saqueryreq_tx", bssid, addr)
325 num_resp = wt.get_sta_counter("valid_saqueryresp_rx", bssid, addr)
326 if num_req != 2 or num_resp != 2:
327 raise Exception("Unexpected number of SA Query procedures (req=%d resp=%d)" % (num_req, num_resp))
329 def test_ap_pmf_required_eap(dev, apdev):
330 """WPA2-EAP AP with PMF required"""
331 ssid = "test-pmf-required-eap"
332 params = hostapd.wpa2_eap_params(ssid=ssid)
333 params["wpa_key_mgmt"] = "WPA-EAP-SHA256";
334 params["ieee80211w"] = "2";
335 hapd = hostapd.add_ap(apdev[0], params)
336 key_mgmt = hapd.get_config()['key_mgmt']
337 if key_mgmt.split(' ')[0] != "WPA-EAP-SHA256":
338 raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
339 dev[0].connect("test-pmf-required-eap", key_mgmt="WPA-EAP-SHA256",
340 ieee80211w="2", eap="PSK", identity="psk.user@example.com",
341 password_hex="0123456789abcdef0123456789abcdef",
343 dev[1].connect("test-pmf-required-eap", key_mgmt="WPA-EAP WPA-EAP-SHA256",
344 ieee80211w="1", eap="PSK", identity="psk.user@example.com",
345 password_hex="0123456789abcdef0123456789abcdef",
348 def test_ap_pmf_optional_eap(dev, apdev):
349 """WPA2EAP AP with PMF optional"""
350 params = hostapd.wpa2_eap_params(ssid="test-wpa2-eap")
351 params["ieee80211w"] = "1";
352 hapd = hostapd.add_ap(apdev[0], params)
353 dev[0].connect("test-wpa2-eap", key_mgmt="WPA-EAP", eap="TTLS",
354 identity="pap user", anonymous_identity="ttls",
356 ca_cert="auth_serv/ca.pem", phase2="auth=PAP",
357 ieee80211w="1", scan_freq="2412")
358 dev[1].connect("test-wpa2-eap", key_mgmt="WPA-EAP WPA-EAP-SHA256",
359 eap="TTLS", identity="pap user", anonymous_identity="ttls",
361 ca_cert="auth_serv/ca.pem", phase2="auth=PAP",
362 ieee80211w="2", scan_freq="2412")
365 def test_ap_pmf_required_sha1(dev, apdev):
366 """WPA2-PSK AP with PMF required with SHA1 AKM"""
367 ssid = "test-pmf-required-sha1"
368 params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
369 params["wpa_key_mgmt"] = "WPA-PSK";
370 params["ieee80211w"] = "2";
371 hapd = hostapd.add_ap(apdev[0], params)
375 wt.add_passphrase("12345678")
376 key_mgmt = hapd.get_config()['key_mgmt']
377 if key_mgmt.split(' ')[0] != "WPA-PSK":
378 raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
379 dev[0].connect(ssid, psk="12345678", ieee80211w="2",
380 key_mgmt="WPA-PSK", proto="WPA2", scan_freq="2412")
381 if "[WPA2-PSK-CCMP]" not in dev[0].request("SCAN_RESULTS"):
382 raise Exception("Scan results missing RSN element info")
383 hwsim_utils.test_connectivity(dev[0], hapd)
386 def test_ap_pmf_toggle(dev, apdev):
387 """WPA2-PSK AP with PMF optional and changing PMF on reassociation"""
389 _test_ap_pmf_toggle(dev, apdev)
391 dev[0].request("SET reassoc_same_bss_optim 0")
393 def _test_ap_pmf_toggle(dev, apdev):
394 ssid = "test-pmf-optional"
395 params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
396 params["wpa_key_mgmt"] = "WPA-PSK";
397 params["ieee80211w"] = "1";
398 params["assoc_sa_query_max_timeout"] = "1"
399 params["assoc_sa_query_retry_timeout"] = "1"
400 hapd = hostapd.add_ap(apdev[0], params)
404 wt.add_passphrase("12345678")
405 bssid = apdev[0]['bssid']
406 addr = dev[0].own_addr()
407 dev[0].request("SET reassoc_same_bss_optim 1")
408 id = dev[0].connect(ssid, psk="12345678", ieee80211w="1",
409 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
411 wt.require_ap_pmf_optional(bssid)
412 wt.require_sta_pmf(bssid, addr)
413 sta = hapd.get_sta(addr)
414 if '[MFP]' not in sta['flags']:
415 raise Exception("MFP flag not present for STA")
417 dev[0].set_network(id, "ieee80211w", "0")
418 dev[0].request("REASSOCIATE")
419 dev[0].wait_connected()
420 wt.require_sta_no_pmf(bssid, addr)
421 sta = hapd.get_sta(addr)
422 if '[MFP]' in sta['flags']:
423 raise Exception("MFP flag unexpectedly present for STA")
424 err, data = hapd.cmd_execute(['iw', 'dev', apdev[0]['ifname'], 'station',
426 if "yes" in [l for l in data.splitlines() if "MFP" in l][0]:
427 raise Exception("Kernel STA entry had MFP enabled")
429 dev[0].set_network(id, "ieee80211w", "1")
430 dev[0].request("REASSOCIATE")
431 dev[0].wait_connected()
432 wt.require_sta_pmf(bssid, addr)
433 sta = hapd.get_sta(addr)
434 if '[MFP]' not in sta['flags']:
435 raise Exception("MFP flag not present for STA")
436 err, data = hapd.cmd_execute(['iw', 'dev', apdev[0]['ifname'], 'station',
438 if "yes" not in [l for l in data.splitlines() if "MFP" in l][0]:
439 raise Exception("Kernel STA entry did not have MFP enabled")
442 def test_ap_pmf_required_sta_no_pmf(dev, apdev):
443 """WPA2-PSK AP with PMF required and PMF disabled on STA"""
444 ssid = "test-pmf-required"
445 params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
446 params["wpa_key_mgmt"] = "WPA-PSK-SHA256";
447 params["ieee80211w"] = "2";
448 hapd = hostapd.add_ap(apdev[0], params)
450 # Disable PMF on the station and try to connect
451 dev[0].connect(ssid, psk="12345678", ieee80211w="0",
452 key_mgmt="WPA-PSK WPA-PSK-SHA256", proto="WPA2",
453 scan_freq="2412", wait_connect=False)
454 ev = dev[0].wait_event(["CTRL-EVENT-NETWORK-NOT-FOUND",
455 "CTRL-EVENT-ASSOC-REJECT"], timeout=2)
457 raise Exception("No connection result")
458 if "CTRL-EVENT-ASSOC-REJECT" in ev:
459 raise Exception("Tried to connect to PMF required AP without PMF enabled")
460 dev[0].request("REMOVE_NETWORK all")