]> git.ipfire.org Git - thirdparty/hostap.git/blame - tests/hwsim/test_ap_ft.py
tests: wpa_supplicant and station interface in a bridge
[thirdparty/hostap.git] / tests / hwsim / test_ap_ft.py
CommitLineData
cd7f1b9a 1# Fast BSS Transition tests
ae14a2e2 2# Copyright (c) 2013-2014, Jouni Malinen <j@w1.fi>
cd7f1b9a
JM
3#
4# This software may be distributed under the terms of the BSD license.
5# See README for more details.
6
5b3c40a6
JM
7import binascii
8import os
cd7f1b9a
JM
9import time
10import subprocess
11import logging
c9aa4308 12logger = logging.getLogger()
cd7f1b9a
JM
13
14import hwsim_utils
15import hostapd
81e787b7 16from utils import HwsimSkip
cd7f1b9a 17from wlantest import Wlantest
5b3c40a6 18from test_ap_psk import check_mib, find_wpas_process, read_process_memory, verify_not_present, get_key_locations
cd7f1b9a
JM
19
20def ft_base_rsn():
21 params = { "wpa": "2",
22 "wpa_key_mgmt": "FT-PSK",
23 "rsn_pairwise": "CCMP" }
24 return params
25
26def ft_base_mixed():
27 params = { "wpa": "3",
28 "wpa_key_mgmt": "WPA-PSK FT-PSK",
29 "wpa_pairwise": "TKIP",
30 "rsn_pairwise": "CCMP" }
31 return params
32
33def ft_params(rsn=True, ssid=None, passphrase=None):
34 if rsn:
35 params = ft_base_rsn()
36 else:
37 params = ft_base_mixed()
38 if ssid:
39 params["ssid"] = ssid
40 if passphrase:
41 params["wpa_passphrase"] = passphrase
42
43 params["mobility_domain"] = "a1b2"
44 params["r0_key_lifetime"] = "10000"
45 params["pmk_r1_push"] = "1"
46 params["reassociation_deadline"] = "1000"
47 return params
48
49def ft_params1(rsn=True, ssid=None, passphrase=None):
50 params = ft_params(rsn, ssid, passphrase)
51 params['nas_identifier'] = "nas1.w1.fi"
52 params['r1_key_holder'] = "000102030405"
53 params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 100102030405060708090a0b0c0d0e0f",
54 "02:00:00:00:04:00 nas2.w1.fi 300102030405060708090a0b0c0d0e0f" ]
55 params['r1kh'] = "02:00:00:00:04:00 00:01:02:03:04:06 200102030405060708090a0b0c0d0e0f"
56 return params
57
58def ft_params2(rsn=True, ssid=None, passphrase=None):
59 params = ft_params(rsn, ssid, passphrase)
60 params['nas_identifier'] = "nas2.w1.fi"
61 params['r1_key_holder'] = "000102030406"
62 params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0e0f",
63 "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0e0f" ]
64 params['r1kh'] = "02:00:00:00:03:00 00:01:02:03:04:05 300102030405060708090a0b0c0d0e0f"
65 return params
66
3b808945
JM
67def ft_params1_r0kh_mismatch(rsn=True, ssid=None, passphrase=None):
68 params = ft_params(rsn, ssid, passphrase)
69 params['nas_identifier'] = "nas1.w1.fi"
70 params['r1_key_holder'] = "000102030405"
71 params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 100102030405060708090a0b0c0d0e0f",
72 "12:00:00:00:04:00 nas2.w1.fi 300102030405060708090a0b0c0d0e0f" ]
73 params['r1kh'] = "12:00:00:00:04:00 10:01:02:03:04:06 200102030405060708090a0b0c0d0e0f"
74 return params
75
76def ft_params2_incorrect_rrb_key(rsn=True, ssid=None, passphrase=None):
77 params = ft_params(rsn, ssid, passphrase)
78 params['nas_identifier'] = "nas2.w1.fi"
79 params['r1_key_holder'] = "000102030406"
80 params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0ef1",
81 "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0ef2" ]
82 params['r1kh'] = "02:00:00:00:03:00 00:01:02:03:04:05 300102030405060708090a0b0c0d0ef3"
83 return params
84
85def ft_params2_r0kh_mismatch(rsn=True, ssid=None, passphrase=None):
86 params = ft_params(rsn, ssid, passphrase)
87 params['nas_identifier'] = "nas2.w1.fi"
88 params['r1_key_holder'] = "000102030406"
89 params['r0kh'] = [ "12:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0e0f",
90 "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0e0f" ]
91 params['r1kh'] = "12:00:00:00:03:00 10:01:02:03:04:05 300102030405060708090a0b0c0d0e0f"
92 return params
93
a8375c94 94def run_roams(dev, apdev, hapd0, hapd1, ssid, passphrase, over_ds=False, sae=False, eap=False, fail_test=False, roams=1):
cd7f1b9a 95 logger.info("Connect to first AP")
6f62809b
JM
96 if eap:
97 dev.connect(ssid, key_mgmt="FT-EAP", proto="WPA2", ieee80211w="1",
2f816c21
JM
98 eap="GPSK", identity="gpsk user",
99 password="abcdefghijklmnop0123456789abcdef",
100 scan_freq="2412")
6e658cc4 101 else:
6f62809b
JM
102 if sae:
103 key_mgmt="FT-SAE"
104 else:
105 key_mgmt="FT-PSK"
106 dev.connect(ssid, psk=passphrase, key_mgmt=key_mgmt, proto="WPA2",
2f816c21 107 ieee80211w="1", scan_freq="2412")
cd7f1b9a
JM
108 if dev.get_status_field('bssid') == apdev[0]['bssid']:
109 ap1 = apdev[0]
110 ap2 = apdev[1]
a8375c94
JM
111 hapd1ap = hapd0
112 hapd2ap = hapd1
cd7f1b9a
JM
113 else:
114 ap1 = apdev[1]
115 ap2 = apdev[0]
a8375c94
JM
116 hapd1ap = hapd1
117 hapd2ap = hapd0
118 hwsim_utils.test_connectivity(dev, hapd1ap)
cd7f1b9a 119
655bc8bf 120 dev.scan_for_bss(ap2['bssid'], freq="2412")
40602101
JM
121
122 for i in range(0, roams):
123 logger.info("Roam to the second AP")
124 if over_ds:
125 dev.roam_over_ds(ap2['bssid'], fail_test=fail_test)
126 else:
127 dev.roam(ap2['bssid'], fail_test=fail_test)
128 if fail_test:
129 return
130 if dev.get_status_field('bssid') != ap2['bssid']:
131 raise Exception("Did not connect to correct AP")
132 if i == 0 or i == roams - 1:
a8375c94 133 hwsim_utils.test_connectivity(dev, hapd2ap)
40602101
JM
134
135 logger.info("Roam back to the first AP")
136 if over_ds:
137 dev.roam_over_ds(ap1['bssid'])
138 else:
139 dev.roam(ap1['bssid'])
140 if dev.get_status_field('bssid') != ap1['bssid']:
141 raise Exception("Did not connect to correct AP")
142 if i == 0 or i == roams - 1:
a8375c94 143 hwsim_utils.test_connectivity(dev, hapd1ap)
cd7f1b9a
JM
144
145def test_ap_ft(dev, apdev):
146 """WPA2-PSK-FT AP"""
147 ssid = "test-ft"
148 passphrase="12345678"
149
150 params = ft_params1(ssid=ssid, passphrase=passphrase)
a8375c94 151 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
cd7f1b9a 152 params = ft_params2(ssid=ssid, passphrase=passphrase)
a8375c94 153 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
cd7f1b9a 154
a8375c94 155 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase)
91bc6c36
JM
156 if "[WPA2-FT/PSK-CCMP]" not in dev[0].request("SCAN_RESULTS"):
157 raise Exception("Scan results missing RSN element info")
cd7f1b9a 158
40602101
JM
159def test_ap_ft_many(dev, apdev):
160 """WPA2-PSK-FT AP multiple times"""
161 ssid = "test-ft"
162 passphrase="12345678"
163
164 params = ft_params1(ssid=ssid, passphrase=passphrase)
a8375c94 165 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
40602101 166 params = ft_params2(ssid=ssid, passphrase=passphrase)
a8375c94 167 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
40602101 168
a8375c94 169 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, roams=50)
40602101 170
cd7f1b9a
JM
171def test_ap_ft_mixed(dev, apdev):
172 """WPA2-PSK-FT mixed-mode AP"""
173 ssid = "test-ft-mixed"
174 passphrase="12345678"
175
176 params = ft_params1(rsn=False, ssid=ssid, passphrase=passphrase)
65038313
JM
177 hapd = hostapd.add_ap(apdev[0]['ifname'], params)
178 key_mgmt = hapd.get_config()['key_mgmt']
179 vals = key_mgmt.split(' ')
180 if vals[0] != "WPA-PSK" or vals[1] != "FT-PSK":
181 raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
cd7f1b9a 182 params = ft_params2(rsn=False, ssid=ssid, passphrase=passphrase)
a8375c94 183 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
cd7f1b9a 184
a8375c94 185 run_roams(dev[0], apdev, hapd, hapd1, ssid, passphrase)
cd7f1b9a
JM
186
187def test_ap_ft_pmf(dev, apdev):
188 """WPA2-PSK-FT AP with PMF"""
189 ssid = "test-ft"
190 passphrase="12345678"
191
192 params = ft_params1(ssid=ssid, passphrase=passphrase)
193 params["ieee80211w"] = "2";
a8375c94 194 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
cd7f1b9a
JM
195 params = ft_params2(ssid=ssid, passphrase=passphrase)
196 params["ieee80211w"] = "2";
a8375c94 197 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
cd7f1b9a 198
a8375c94 199 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase)
b553eab1
JM
200
201def test_ap_ft_over_ds(dev, apdev):
202 """WPA2-PSK-FT AP over DS"""
203 ssid = "test-ft"
204 passphrase="12345678"
205
206 params = ft_params1(ssid=ssid, passphrase=passphrase)
a8375c94 207 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
b553eab1 208 params = ft_params2(ssid=ssid, passphrase=passphrase)
a8375c94 209 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
b553eab1 210
a8375c94 211 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True)
eaf3f9b1
JM
212 check_mib(dev[0], [ ("dot11RSNAAuthenticationSuiteRequested", "00-0f-ac-4"),
213 ("dot11RSNAAuthenticationSuiteSelected", "00-0f-ac-4") ])
b553eab1 214
40602101
JM
215def test_ap_ft_over_ds_many(dev, apdev):
216 """WPA2-PSK-FT AP over DS multiple times"""
217 ssid = "test-ft"
218 passphrase="12345678"
219
220 params = ft_params1(ssid=ssid, passphrase=passphrase)
a8375c94 221 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
40602101 222 params = ft_params2(ssid=ssid, passphrase=passphrase)
a8375c94 223 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
40602101 224
a8375c94
JM
225 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
226 roams=50)
40602101 227
c337d07a
JM
228def test_ap_ft_over_ds_unknown_target(dev, apdev):
229 """WPA2-PSK-FT AP"""
230 ssid = "test-ft"
231 passphrase="12345678"
232
233 params = ft_params1(ssid=ssid, passphrase=passphrase)
234 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
235
236 dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
237 scan_freq="2412")
238 dev[0].roam_over_ds("02:11:22:33:44:55", fail_test=True)
239
b553eab1
JM
240def test_ap_ft_pmf_over_ds(dev, apdev):
241 """WPA2-PSK-FT AP over DS with PMF"""
242 ssid = "test-ft"
243 passphrase="12345678"
244
245 params = ft_params1(ssid=ssid, passphrase=passphrase)
246 params["ieee80211w"] = "2";
a8375c94 247 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
b553eab1
JM
248 params = ft_params2(ssid=ssid, passphrase=passphrase)
249 params["ieee80211w"] = "2";
a8375c94 250 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
b553eab1 251
a8375c94 252 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True)
6e658cc4 253
aaba98d3
JM
254def test_ap_ft_over_ds_pull(dev, apdev):
255 """WPA2-PSK-FT AP over DS (pull PMK)"""
256 ssid = "test-ft"
257 passphrase="12345678"
258
259 params = ft_params1(ssid=ssid, passphrase=passphrase)
260 params["pmk_r1_push"] = "0"
a8375c94 261 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
aaba98d3
JM
262 params = ft_params2(ssid=ssid, passphrase=passphrase)
263 params["pmk_r1_push"] = "0"
a8375c94 264 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
aaba98d3 265
a8375c94 266 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True)
aaba98d3 267
6e658cc4
JM
268def test_ap_ft_sae(dev, apdev):
269 """WPA2-PSK-FT-SAE AP"""
b9749b6a
JM
270 if "SAE" not in dev[0].get_capability("auth_alg"):
271 raise HwsimSkip("SAE not supported")
6e658cc4
JM
272 ssid = "test-ft"
273 passphrase="12345678"
274
275 params = ft_params1(ssid=ssid, passphrase=passphrase)
276 params['wpa_key_mgmt'] = "FT-SAE"
a8375c94 277 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
6e658cc4
JM
278 params = ft_params2(ssid=ssid, passphrase=passphrase)
279 params['wpa_key_mgmt'] = "FT-SAE"
65038313
JM
280 hapd = hostapd.add_ap(apdev[1]['ifname'], params)
281 key_mgmt = hapd.get_config()['key_mgmt']
282 if key_mgmt.split(' ')[0] != "FT-SAE":
283 raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
6e658cc4 284
17ffdf39 285 dev[0].request("SET sae_groups ")
a8375c94 286 run_roams(dev[0], apdev, hapd0, hapd, ssid, passphrase, sae=True)
6e658cc4
JM
287
288def test_ap_ft_sae_over_ds(dev, apdev):
289 """WPA2-PSK-FT-SAE AP over DS"""
b9749b6a
JM
290 if "SAE" not in dev[0].get_capability("auth_alg"):
291 raise HwsimSkip("SAE not supported")
6e658cc4
JM
292 ssid = "test-ft"
293 passphrase="12345678"
294
295 params = ft_params1(ssid=ssid, passphrase=passphrase)
296 params['wpa_key_mgmt'] = "FT-SAE"
a8375c94 297 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
6e658cc4
JM
298 params = ft_params2(ssid=ssid, passphrase=passphrase)
299 params['wpa_key_mgmt'] = "FT-SAE"
a8375c94 300 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
6e658cc4 301
17ffdf39 302 dev[0].request("SET sae_groups ")
a8375c94
JM
303 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, sae=True,
304 over_ds=True)
6f62809b
JM
305
306def test_ap_ft_eap(dev, apdev):
307 """WPA2-EAP-FT AP"""
308 ssid = "test-ft"
309 passphrase="12345678"
310
311 radius = hostapd.radius_params()
312 params = ft_params1(ssid=ssid, passphrase=passphrase)
313 params['wpa_key_mgmt'] = "FT-EAP"
314 params["ieee8021x"] = "1"
315 params = dict(radius.items() + params.items())
65038313
JM
316 hapd = hostapd.add_ap(apdev[0]['ifname'], params)
317 key_mgmt = hapd.get_config()['key_mgmt']
318 if key_mgmt.split(' ')[0] != "FT-EAP":
319 raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
6f62809b
JM
320 params = ft_params2(ssid=ssid, passphrase=passphrase)
321 params['wpa_key_mgmt'] = "FT-EAP"
322 params["ieee8021x"] = "1"
323 params = dict(radius.items() + params.items())
a8375c94 324 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
6f62809b 325
a8375c94 326 run_roams(dev[0], apdev, hapd, hapd1, ssid, passphrase, eap=True)
91bc6c36
JM
327 if "[WPA2-FT/EAP-CCMP]" not in dev[0].request("SCAN_RESULTS"):
328 raise Exception("Scan results missing RSN element info")
eaf3f9b1
JM
329 check_mib(dev[0], [ ("dot11RSNAAuthenticationSuiteRequested", "00-0f-ac-3"),
330 ("dot11RSNAAuthenticationSuiteSelected", "00-0f-ac-3") ])
aaba98d3
JM
331
332def test_ap_ft_eap_pull(dev, apdev):
333 """WPA2-EAP-FT AP (pull PMK)"""
334 ssid = "test-ft"
335 passphrase="12345678"
336
337 radius = hostapd.radius_params()
338 params = ft_params1(ssid=ssid, passphrase=passphrase)
339 params['wpa_key_mgmt'] = "FT-EAP"
340 params["ieee8021x"] = "1"
341 params["pmk_r1_push"] = "0"
342 params = dict(radius.items() + params.items())
343 hapd = hostapd.add_ap(apdev[0]['ifname'], params)
344 key_mgmt = hapd.get_config()['key_mgmt']
345 if key_mgmt.split(' ')[0] != "FT-EAP":
346 raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
347 params = ft_params2(ssid=ssid, passphrase=passphrase)
348 params['wpa_key_mgmt'] = "FT-EAP"
349 params["ieee8021x"] = "1"
350 params["pmk_r1_push"] = "0"
351 params = dict(radius.items() + params.items())
a8375c94 352 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
aaba98d3 353
a8375c94 354 run_roams(dev[0], apdev, hapd, hapd1, ssid, passphrase, eap=True)
3b808945
JM
355
356def test_ap_ft_mismatching_rrb_key_push(dev, apdev):
357 """WPA2-PSK-FT AP over DS with mismatching RRB key (push)"""
358 ssid = "test-ft"
359 passphrase="12345678"
360
361 params = ft_params1(ssid=ssid, passphrase=passphrase)
362 params["ieee80211w"] = "2";
a8375c94 363 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
3b808945
JM
364 params = ft_params2_incorrect_rrb_key(ssid=ssid, passphrase=passphrase)
365 params["ieee80211w"] = "2";
a8375c94 366 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
3b808945 367
a8375c94
JM
368 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
369 fail_test=True)
3b808945
JM
370
371def test_ap_ft_mismatching_rrb_key_pull(dev, apdev):
372 """WPA2-PSK-FT AP over DS with mismatching RRB key (pull)"""
373 ssid = "test-ft"
374 passphrase="12345678"
375
376 params = ft_params1(ssid=ssid, passphrase=passphrase)
377 params["pmk_r1_push"] = "0"
a8375c94 378 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
3b808945
JM
379 params = ft_params2_incorrect_rrb_key(ssid=ssid, passphrase=passphrase)
380 params["pmk_r1_push"] = "0"
a8375c94 381 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
3b808945 382
a8375c94
JM
383 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
384 fail_test=True)
3b808945 385
ae14a2e2
JM
386def test_ap_ft_mismatching_r0kh_id_pull(dev, apdev):
387 """WPA2-PSK-FT AP over DS with mismatching R0KH-ID (pull)"""
388 ssid = "test-ft"
389 passphrase="12345678"
390
391 params = ft_params1(ssid=ssid, passphrase=passphrase)
392 params["pmk_r1_push"] = "0"
393 params["nas_identifier"] = "nas0.w1.fi"
394 hostapd.add_ap(apdev[0]['ifname'], params)
2f816c21
JM
395 dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
396 scan_freq="2412")
ae14a2e2
JM
397
398 params = ft_params2(ssid=ssid, passphrase=passphrase)
399 params["pmk_r1_push"] = "0"
400 hostapd.add_ap(apdev[1]['ifname'], params)
401
402 dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
403 dev[0].roam_over_ds(apdev[1]['bssid'], fail_test=True)
404
3b808945
JM
405def test_ap_ft_mismatching_rrb_r0kh_push(dev, apdev):
406 """WPA2-PSK-FT AP over DS with mismatching R0KH key (push)"""
407 ssid = "test-ft"
408 passphrase="12345678"
409
410 params = ft_params1(ssid=ssid, passphrase=passphrase)
411 params["ieee80211w"] = "2";
a8375c94 412 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
3b808945
JM
413 params = ft_params2_r0kh_mismatch(ssid=ssid, passphrase=passphrase)
414 params["ieee80211w"] = "2";
a8375c94 415 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
3b808945 416
a8375c94
JM
417 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
418 fail_test=True)
3b808945
JM
419
420def test_ap_ft_mismatching_rrb_r0kh_pull(dev, apdev):
421 """WPA2-PSK-FT AP over DS with mismatching R0KH key (pull)"""
422 ssid = "test-ft"
423 passphrase="12345678"
424
425 params = ft_params1_r0kh_mismatch(ssid=ssid, passphrase=passphrase)
426 params["pmk_r1_push"] = "0"
a8375c94 427 hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
3b808945
JM
428 params = ft_params2(ssid=ssid, passphrase=passphrase)
429 params["pmk_r1_push"] = "0"
a8375c94 430 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
3b808945 431
a8375c94
JM
432 run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
433 fail_test=True)
c6b6e105
JM
434
435def test_ap_ft_gtk_rekey(dev, apdev):
436 """WPA2-PSK-FT AP and GTK rekey"""
437 ssid = "test-ft"
438 passphrase="12345678"
439
440 params = ft_params1(ssid=ssid, passphrase=passphrase)
441 params['wpa_group_rekey'] = '1'
a8375c94 442 hapd = hostapd.add_ap(apdev[0]['ifname'], params)
c6b6e105
JM
443
444 dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
2f816c21 445 ieee80211w="1", scan_freq="2412")
c6b6e105
JM
446
447 ev = dev[0].wait_event(["WPA: Group rekeying completed"], timeout=2)
448 if ev is None:
449 raise Exception("GTK rekey timed out after initial association")
a8375c94 450 hwsim_utils.test_connectivity(dev[0], hapd)
c6b6e105
JM
451
452 params = ft_params2(ssid=ssid, passphrase=passphrase)
453 params['wpa_group_rekey'] = '1'
a8375c94 454 hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
c6b6e105
JM
455
456 dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
457 dev[0].roam(apdev[1]['bssid'])
458 if dev[0].get_status_field('bssid') != apdev[1]['bssid']:
459 raise Exception("Did not connect to correct AP")
a8375c94 460 hwsim_utils.test_connectivity(dev[0], hapd1)
c6b6e105
JM
461
462 ev = dev[0].wait_event(["WPA: Group rekeying completed"], timeout=2)
463 if ev is None:
464 raise Exception("GTK rekey timed out after FT protocol")
a8375c94 465 hwsim_utils.test_connectivity(dev[0], hapd1)
5b3c40a6
JM
466
467def test_ft_psk_key_lifetime_in_memory(dev, apdev, params):
468 """WPA2-PSK-FT and key lifetime in memory"""
469 ssid = "test-ft"
470 passphrase="04c2726b4b8d5f1b4db9c07aa4d9e9d8f765cb5d25ec817e6cc4fcdd5255db0"
471 psk = '93c90846ff67af9037ed83fb72b63dbeddaa81d47f926c20909b5886f1d9358d'
472 pmk = binascii.unhexlify(psk)
473 p = ft_params1(ssid=ssid, passphrase=passphrase)
474 hapd0 = hostapd.add_ap(apdev[0]['ifname'], p)
475 p = ft_params2(ssid=ssid, passphrase=passphrase)
476 hapd1 = hostapd.add_ap(apdev[1]['ifname'], p)
477
478 pid = find_wpas_process(dev[0])
479
480 dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
481 scan_freq="2412")
482 time.sleep(0.1)
483
484 buf = read_process_memory(pid, pmk)
485
486 dev[0].request("DISCONNECT")
487 dev[0].wait_disconnected()
488
489 dev[0].relog()
490 pmkr0 = None
491 pmkr1 = None
492 ptk = None
493 gtk = None
494 with open(os.path.join(params['logdir'], 'log0'), 'r') as f:
495 for l in f.readlines():
496 if "FT: PMK-R0 - hexdump" in l:
497 val = l.strip().split(':')[3].replace(' ', '')
498 pmkr0 = binascii.unhexlify(val)
499 if "FT: PMK-R1 - hexdump" in l:
500 val = l.strip().split(':')[3].replace(' ', '')
501 pmkr1 = binascii.unhexlify(val)
f918b95b 502 if "FT: KCK - hexdump" in l:
5b3c40a6 503 val = l.strip().split(':')[3].replace(' ', '')
f918b95b
JM
504 kck = binascii.unhexlify(val)
505 if "FT: KEK - hexdump" in l:
506 val = l.strip().split(':')[3].replace(' ', '')
507 kek = binascii.unhexlify(val)
508 if "FT: TK - hexdump" in l:
509 val = l.strip().split(':')[3].replace(' ', '')
510 tk = binascii.unhexlify(val)
5b3c40a6
JM
511 if "WPA: Group Key - hexdump" in l:
512 val = l.strip().split(':')[3].replace(' ', '')
513 gtk = binascii.unhexlify(val)
f918b95b 514 if not pmkr0 or not pmkr1 or not kck or not kek or not tk or not gtk:
5b3c40a6
JM
515 raise Exception("Could not find keys from debug log")
516 if len(gtk) != 16:
517 raise Exception("Unexpected GTK length")
518
5b3c40a6
JM
519 logger.info("Checking keys in memory while associated")
520 get_key_locations(buf, pmk, "PMK")
521 get_key_locations(buf, pmkr0, "PMK-R0")
522 get_key_locations(buf, pmkr1, "PMK-R1")
523 if pmk not in buf:
81e787b7 524 raise HwsimSkip("PMK not found while associated")
5b3c40a6 525 if pmkr0 not in buf:
81e787b7 526 raise HwsimSkip("PMK-R0 not found while associated")
5b3c40a6 527 if pmkr1 not in buf:
81e787b7 528 raise HwsimSkip("PMK-R1 not found while associated")
5b3c40a6
JM
529 if kck not in buf:
530 raise Exception("KCK not found while associated")
531 if kek not in buf:
532 raise Exception("KEK not found while associated")
533 if tk in buf:
534 raise Exception("TK found from memory")
535 if gtk in buf:
536 raise Exception("GTK found from memory")
537
538 logger.info("Checking keys in memory after disassociation")
539 buf = read_process_memory(pid, pmk)
540 get_key_locations(buf, pmk, "PMK")
541 get_key_locations(buf, pmkr0, "PMK-R0")
542 get_key_locations(buf, pmkr1, "PMK-R1")
543
544 # Note: PMK/PSK is still present in network configuration
545
546 fname = os.path.join(params['logdir'],
547 'ft_psk_key_lifetime_in_memory.memctx-')
548 verify_not_present(buf, pmkr0, fname, "PMK-R0")
549 verify_not_present(buf, pmkr1, fname, "PMK-R1")
550 verify_not_present(buf, kck, fname, "KCK")
551 verify_not_present(buf, kek, fname, "KEK")
552 verify_not_present(buf, tk, fname, "TK")
553 verify_not_present(buf, gtk, fname, "GTK")
554
555 dev[0].request("REMOVE_NETWORK all")
556
557 logger.info("Checking keys in memory after network profile removal")
558 buf = read_process_memory(pid, pmk)
559 get_key_locations(buf, pmk, "PMK")
560 get_key_locations(buf, pmkr0, "PMK-R0")
561 get_key_locations(buf, pmkr1, "PMK-R1")
562
563 verify_not_present(buf, pmk, fname, "PMK")
564 verify_not_present(buf, pmkr0, fname, "PMK-R0")
565 verify_not_present(buf, pmkr1, fname, "PMK-R1")
566 verify_not_present(buf, kck, fname, "KCK")
567 verify_not_present(buf, kek, fname, "KEK")
568 verify_not_present(buf, tk, fname, "TK")
569 verify_not_present(buf, gtk, fname, "GTK")