1 # WPA2-Enterprise PMKSA caching tests
2 # Copyright (c) 2013-2014, Jouni Malinen <j@w1.fi>
4 # This software may be distributed under the terms of the BSD license.
5 # See README for more details.
8 logger = logging.getLogger()
13 from wpasupplicant import WpaSupplicant
14 from test_ap_eap import eap_connect
16 def test_pmksa_cache_on_roam_back(dev, apdev):
17 """PMKSA cache to skip EAP on reassociation back to same AP"""
18 params = hostapd.wpa2_eap_params(ssid="test-pmksa-cache")
19 hostapd.add_ap(apdev[0]['ifname'], params)
20 bssid = apdev[0]['bssid']
21 dev[0].connect("test-pmksa-cache", proto="RSN", key_mgmt="WPA-EAP",
22 eap="GPSK", identity="gpsk user",
23 password="abcdefghijklmnop0123456789abcdef",
25 pmksa = dev[0].get_pmksa(bssid)
27 raise Exception("No PMKSA cache entry created")
28 if pmksa['opportunistic'] != '0':
29 raise Exception("Unexpected opportunistic PMKSA cache entry")
31 hostapd.add_ap(apdev[1]['ifname'], params)
32 bssid2 = apdev[1]['bssid']
35 logger.info("Roam to AP2")
36 dev[0].scan(freq="2412")
37 dev[0].request("ROAM " + bssid2)
38 ev = dev[0].wait_event(["CTRL-EVENT-EAP-SUCCESS"], timeout=10)
40 raise Exception("EAP success timed out")
41 ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"], timeout=10)
43 raise Exception("Roaming with the AP timed out")
44 pmksa2 = dev[0].get_pmksa(bssid2)
46 raise Exception("No PMKSA cache entry found")
47 if pmksa2['opportunistic'] != '0':
48 raise Exception("Unexpected opportunistic PMKSA cache entry")
51 logger.info("Roam back to AP1")
52 dev[0].scan(freq="2412")
53 dev[0].request("ROAM " + bssid)
54 ev = dev[0].wait_event(["CTRL-EVENT-EAP-STARTED",
55 "CTRL-EVENT-CONNECTED"], timeout=10)
57 raise Exception("Roaming with the AP timed out")
58 if "CTRL-EVENT-EAP-STARTED" in ev:
59 raise Exception("Unexpected EAP exchange")
60 pmksa1b = dev[0].get_pmksa(bssid)
62 raise Exception("No PMKSA cache entry found")
63 if pmksa['pmkid'] != pmksa1b['pmkid']:
64 raise Exception("Unexpected PMKID change for AP1")
66 def test_pmksa_cache_opportunistic_only_on_sta(dev, apdev):
67 """Opportunistic PMKSA caching enabled only on station"""
68 params = hostapd.wpa2_eap_params(ssid="test-pmksa-cache")
69 hostapd.add_ap(apdev[0]['ifname'], params)
70 bssid = apdev[0]['bssid']
71 dev[0].connect("test-pmksa-cache", proto="RSN", key_mgmt="WPA-EAP",
72 eap="GPSK", identity="gpsk user",
73 password="abcdefghijklmnop0123456789abcdef", okc=True,
75 pmksa = dev[0].get_pmksa(bssid)
77 raise Exception("No PMKSA cache entry created")
78 if pmksa['opportunistic'] != '0':
79 raise Exception("Unexpected opportunistic PMKSA cache entry")
81 hostapd.add_ap(apdev[1]['ifname'], params)
82 bssid2 = apdev[1]['bssid']
85 logger.info("Roam to AP2")
86 dev[0].scan(freq="2412")
87 dev[0].request("ROAM " + bssid2)
88 ev = dev[0].wait_event(["CTRL-EVENT-EAP-SUCCESS"], timeout=10)
90 raise Exception("EAP success timed out")
91 ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"], timeout=10)
93 raise Exception("Roaming with the AP timed out")
94 pmksa2 = dev[0].get_pmksa(bssid2)
96 raise Exception("No PMKSA cache entry found")
97 if pmksa2['opportunistic'] != '0':
98 raise Exception("Unexpected opportunistic PMKSA cache entry")
100 dev[0].dump_monitor()
101 logger.info("Roam back to AP1")
102 dev[0].scan(freq="2412")
103 dev[0].request("ROAM " + bssid)
104 ev = dev[0].wait_event(["CTRL-EVENT-EAP-STARTED",
105 "CTRL-EVENT-CONNECTED"], timeout=10)
107 raise Exception("Roaming with the AP timed out")
108 if "CTRL-EVENT-EAP-STARTED" in ev:
109 raise Exception("Unexpected EAP exchange")
110 pmksa1b = dev[0].get_pmksa(bssid)
112 raise Exception("No PMKSA cache entry found")
113 if pmksa['pmkid'] != pmksa1b['pmkid']:
114 raise Exception("Unexpected PMKID change for AP1")
116 def test_pmksa_cache_opportunistic(dev, apdev):
117 """Opportunistic PMKSA caching"""
118 params = hostapd.wpa2_eap_params(ssid="test-pmksa-cache")
120 hostapd.add_ap(apdev[0]['ifname'], params)
121 bssid = apdev[0]['bssid']
122 dev[0].connect("test-pmksa-cache", proto="RSN", key_mgmt="WPA-EAP",
123 eap="GPSK", identity="gpsk user",
124 password="abcdefghijklmnop0123456789abcdef", okc=True,
126 pmksa = dev[0].get_pmksa(bssid)
128 raise Exception("No PMKSA cache entry created")
129 if pmksa['opportunistic'] != '0':
130 raise Exception("Unexpected opportunistic PMKSA cache entry")
132 hostapd.add_ap(apdev[1]['ifname'], params)
133 bssid2 = apdev[1]['bssid']
135 dev[0].dump_monitor()
136 logger.info("Roam to AP2")
137 dev[0].scan(freq="2412")
138 dev[0].request("ROAM " + bssid2)
139 ev = dev[0].wait_event(["CTRL-EVENT-EAP-STARTED",
140 "CTRL-EVENT-CONNECTED"], timeout=10)
142 raise Exception("Roaming with the AP timed out")
143 if "CTRL-EVENT-EAP-STARTED" in ev:
144 raise Exception("Unexpected EAP exchange")
145 pmksa2 = dev[0].get_pmksa(bssid2)
147 raise Exception("No PMKSA cache entry created")
149 dev[0].dump_monitor()
150 logger.info("Roam back to AP1")
151 dev[0].scan(freq="2412")
152 dev[0].request("ROAM " + bssid)
153 ev = dev[0].wait_event(["CTRL-EVENT-EAP-STARTED",
154 "CTRL-EVENT-CONNECTED"], timeout=10)
156 raise Exception("Roaming with the AP timed out")
157 if "CTRL-EVENT-EAP-STARTED" in ev:
158 raise Exception("Unexpected EAP exchange")
160 pmksa1b = dev[0].get_pmksa(bssid)
162 raise Exception("No PMKSA cache entry found")
163 if pmksa['pmkid'] != pmksa1b['pmkid']:
164 raise Exception("Unexpected PMKID change for AP1")
166 def test_pmksa_cache_opportunistic_connect(dev, apdev):
167 """Opportunistic PMKSA caching with connect API"""
168 params = hostapd.wpa2_eap_params(ssid="test-pmksa-cache")
170 hostapd.add_ap(apdev[0]['ifname'], params)
171 bssid = apdev[0]['bssid']
172 wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
173 wpas.interface_add("wlan5", drv_params="force_connect_cmd=1")
174 wpas.connect("test-pmksa-cache", proto="RSN", key_mgmt="WPA-EAP",
175 eap="GPSK", identity="gpsk user",
176 password="abcdefghijklmnop0123456789abcdef", okc=True,
178 pmksa = wpas.get_pmksa(bssid)
180 raise Exception("No PMKSA cache entry created")
181 if pmksa['opportunistic'] != '0':
182 raise Exception("Unexpected opportunistic PMKSA cache entry")
184 hostapd.add_ap(apdev[1]['ifname'], params)
185 bssid2 = apdev[1]['bssid']
188 logger.info("Roam to AP2")
189 wpas.scan(freq="2412")
190 wpas.request("ROAM " + bssid2)
191 ev = wpas.wait_event(["CTRL-EVENT-EAP-STARTED",
192 "CTRL-EVENT-CONNECTED"], timeout=10)
194 raise Exception("Roaming with the AP timed out")
195 if "CTRL-EVENT-EAP-STARTED" in ev:
196 raise Exception("Unexpected EAP exchange")
197 pmksa2 = wpas.get_pmksa(bssid2)
199 raise Exception("No PMKSA cache entry created")
202 logger.info("Roam back to AP1")
203 wpas.scan(freq="2412")
204 wpas.request("ROAM " + bssid)
205 ev = wpas.wait_event(["CTRL-EVENT-EAP-STARTED",
206 "CTRL-EVENT-CONNECTED"], timeout=10)
208 raise Exception("Roaming with the AP timed out")
209 if "CTRL-EVENT-EAP-STARTED" in ev:
210 raise Exception("Unexpected EAP exchange")
212 pmksa1b = wpas.get_pmksa(bssid)
214 raise Exception("No PMKSA cache entry found")
215 if pmksa['pmkid'] != pmksa1b['pmkid']:
216 raise Exception("Unexpected PMKID change for AP1")
218 def test_pmksa_cache_expiration(dev, apdev):
219 """PMKSA cache entry expiration"""
220 params = hostapd.wpa2_eap_params(ssid="test-pmksa-cache")
221 hostapd.add_ap(apdev[0]['ifname'], params)
222 bssid = apdev[0]['bssid']
223 dev[0].request("SET dot11RSNAConfigPMKLifetime 10")
224 dev[0].connect("test-pmksa-cache", proto="RSN", key_mgmt="WPA-EAP",
225 eap="GPSK", identity="gpsk user",
226 password="abcdefghijklmnop0123456789abcdef",
228 pmksa = dev[0].get_pmksa(bssid)
230 raise Exception("No PMKSA cache entry created")
231 logger.info("Wait for PMKSA cache entry to expire")
232 ev = dev[0].wait_event(["WPA: Key negotiation completed",
233 "CTRL-EVENT-DISCONNECTED"], timeout=15)
235 raise Exception("No EAP reauthentication seen")
236 if "CTRL-EVENT-DISCONNECTED" in ev:
237 raise Exception("Unexpected disconnection")
238 pmksa2 = dev[0].get_pmksa(bssid)
239 if pmksa['pmkid'] == pmksa2['pmkid']:
240 raise Exception("PMKID did not change")
242 def test_pmksa_cache_expiration_disconnect(dev, apdev):
243 """PMKSA cache entry expiration (disconnect)"""
244 params = hostapd.wpa2_eap_params(ssid="test-pmksa-cache")
245 hapd = hostapd.add_ap(apdev[0]['ifname'], params)
246 bssid = apdev[0]['bssid']
247 dev[0].request("SET dot11RSNAConfigPMKLifetime 2")
248 dev[0].request("SET dot11RSNAConfigPMKReauthThreshold 100")
249 dev[0].connect("test-pmksa-cache", proto="RSN", key_mgmt="WPA-EAP",
250 eap="GPSK", identity="gpsk user",
251 password="abcdefghijklmnop0123456789abcdef",
253 pmksa = dev[0].get_pmksa(bssid)
255 raise Exception("No PMKSA cache entry created")
256 hapd.request("SET auth_server_shared_secret incorrect")
257 logger.info("Wait for PMKSA cache entry to expire")
258 ev = dev[0].wait_event(["WPA: Key negotiation completed",
259 "CTRL-EVENT-DISCONNECTED"], timeout=15)
261 raise Exception("No EAP reauthentication seen")
262 if "CTRL-EVENT-DISCONNECTED" not in ev:
263 raise Exception("Missing disconnection")
264 hapd.request("SET auth_server_shared_secret radius")
265 ev = dev[0].wait_event(["WPA: Key negotiation completed"], timeout=15)
267 raise Exception("No EAP reauthentication seen")
268 pmksa2 = dev[0].get_pmksa(bssid)
269 if pmksa['pmkid'] == pmksa2['pmkid']:
270 raise Exception("PMKID did not change")
272 def test_pmksa_cache_and_cui(dev, apdev):
273 """PMKSA cache and Chargeable-User-Identity"""
274 params = hostapd.wpa2_eap_params(ssid="cui")
275 params['radius_request_cui'] = '1'
276 params['acct_server_addr'] = "127.0.0.1"
277 params['acct_server_port'] = "1813"
278 params['acct_server_shared_secret'] = "radius"
279 hostapd.add_ap(apdev[0]['ifname'], params)
280 bssid = apdev[0]['bssid']
281 dev[0].connect("cui", proto="RSN", key_mgmt="WPA-EAP",
282 eap="GPSK", identity="gpsk-cui",
283 password="abcdefghijklmnop0123456789abcdef",
285 pmksa = dev[0].get_pmksa(bssid)
287 raise Exception("No PMKSA cache entry created")
289 dev[0].dump_monitor()
290 logger.info("Disconnect and reconnect to the same AP")
291 dev[0].request("DISCONNECT")
292 dev[0].request("RECONNECT")
293 ev = dev[0].wait_event(["CTRL-EVENT-EAP-STARTED",
294 "CTRL-EVENT-CONNECTED"], timeout=10)
296 raise Exception("Reconnect timed out")
297 if "CTRL-EVENT-EAP-STARTED" in ev:
298 raise Exception("Unexpected EAP exchange")
299 pmksa1b = dev[0].get_pmksa(bssid)
301 raise Exception("No PMKSA cache entry found")
302 if pmksa['pmkid'] != pmksa1b['pmkid']:
303 raise Exception("Unexpected PMKID change for AP1")
305 dev[0].request("REAUTHENTICATE")
306 ev = dev[0].wait_event(["CTRL-EVENT-EAP-SUCCESS"], timeout=10)
308 raise Exception("EAP success timed out")
309 for i in range(0, 20):
310 state = dev[0].get_status_field("wpa_state")
311 if state == "COMPLETED":
314 if state != "COMPLETED":
315 raise Exception("Reauthentication did not complete")
317 def test_pmksa_cache_preauth(dev, apdev):
318 """RSN pre-authentication to generate PMKSA cache entry"""
320 params = hostapd.wpa2_eap_params(ssid="test-wpa2-eap")
321 params['bridge'] = 'ap-br0'
322 hostapd.add_ap(apdev[0]['ifname'], params)
323 subprocess.call(['sudo', 'brctl', 'setfd', 'ap-br0', '0'])
324 subprocess.call(['sudo', 'ip', 'link', 'set', 'dev', 'ap-br0', 'up'])
325 eap_connect(dev[0], apdev[0], "PAX", "pax.user@example.com",
326 password_hex="0123456789abcdef0123456789abcdef")
328 params = hostapd.wpa2_eap_params(ssid="test-wpa2-eap")
329 params['bridge'] = 'ap-br0'
330 params['rsn_preauth'] = '1'
331 params['rsn_preauth_interfaces'] = 'ap-br0'
332 hostapd.add_ap(apdev[1]['ifname'], params)
333 bssid1 = apdev[1]['bssid']
334 dev[0].scan(freq="2412")
337 for i in range(0, 50):
339 status = dev[0].request("STATUS")
340 if "Pre-authentication EAPOL state machines:" in status:
343 pmksa = dev[0].get_pmksa(bssid1)
348 raise Exception("No PMKSA cache entry created from pre-authentication")
350 raise Exception("Pre-authentication EAPOL status was not available")
352 dev[0].scan(freq="2412")
353 dev[0].request("ROAM " + bssid1)
354 ev = dev[0].wait_event(["CTRL-EVENT-EAP-STARTED",
355 "CTRL-EVENT-CONNECTED"], timeout=10)
357 raise Exception("Roaming with the AP timed out")
358 if "CTRL-EVENT-EAP-STARTED" in ev:
359 raise Exception("Unexpected EAP exchange")
360 pmksa2 = dev[0].get_pmksa(bssid1)
362 raise Exception("No PMKSA cache entry")
363 if pmksa['pmkid'] != pmksa2['pmkid']:
364 raise Exception("Unexpected PMKID change")
367 subprocess.call(['sudo', 'ip', 'link', 'set', 'dev', 'ap-br0', 'down'])
368 subprocess.call(['sudo', 'brctl', 'delbr', 'ap-br0'])