tests/hwsim/test_ap_ft.py

   1 # Fast BSS Transition tests
   2 # Copyright (c) 2013-2014, 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 time
   8 import subprocess
   9 import logging
  10 logger = logging.getLogger()
  11
  12 import hwsim_utils
  13 import hostapd
  14 from wlantest import Wlantest
  15 from test_ap_psk import check_mib
  16
  17 def ft_base_rsn():
  18     params = { "wpa": "2",
  19                "wpa_key_mgmt": "FT-PSK",
  20                "rsn_pairwise": "CCMP" }
  21     return params
  22
  23 def ft_base_mixed():
  24     params = { "wpa": "3",
  25                "wpa_key_mgmt": "WPA-PSK FT-PSK",
  26                "wpa_pairwise": "TKIP",
  27                "rsn_pairwise": "CCMP" }
  28     return params
  29
  30 def ft_params(rsn=True, ssid=None, passphrase=None):
  31     if rsn:
  32         params = ft_base_rsn()
  33     else:
  34         params = ft_base_mixed()
  35     if ssid:
  36         params["ssid"] = ssid
  37     if passphrase:
  38         params["wpa_passphrase"] = passphrase
  39
  40     params["mobility_domain"] = "a1b2"
  41     params["r0_key_lifetime"] = "10000"
  42     params["pmk_r1_push"] = "1"
  43     params["reassociation_deadline"] = "1000"
  44     return params
  45
  46 def ft_params1(rsn=True, ssid=None, passphrase=None):
  47     params = ft_params(rsn, ssid, passphrase)
  48     params['nas_identifier'] = "nas1.w1.fi"
  49     params['r1_key_holder'] = "000102030405"
  50     params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 100102030405060708090a0b0c0d0e0f",
  51                        "02:00:00:00:04:00 nas2.w1.fi 300102030405060708090a0b0c0d0e0f" ]
  52     params['r1kh'] = "02:00:00:00:04:00 00:01:02:03:04:06 200102030405060708090a0b0c0d0e0f"
  53     return params
  54
  55 def ft_params2(rsn=True, ssid=None, passphrase=None):
  56     params = ft_params(rsn, ssid, passphrase)
  57     params['nas_identifier'] = "nas2.w1.fi"
  58     params['r1_key_holder'] = "000102030406"
  59     params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0e0f",
  60                        "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0e0f" ]
  61     params['r1kh'] = "02:00:00:00:03:00 00:01:02:03:04:05 300102030405060708090a0b0c0d0e0f"
  62     return params
  63
  64 def ft_params1_r0kh_mismatch(rsn=True, ssid=None, passphrase=None):
  65     params = ft_params(rsn, ssid, passphrase)
  66     params['nas_identifier'] = "nas1.w1.fi"
  67     params['r1_key_holder'] = "000102030405"
  68     params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 100102030405060708090a0b0c0d0e0f",
  69                        "12:00:00:00:04:00 nas2.w1.fi 300102030405060708090a0b0c0d0e0f" ]
  70     params['r1kh'] = "12:00:00:00:04:00 10:01:02:03:04:06 200102030405060708090a0b0c0d0e0f"
  71     return params
  72
  73 def ft_params2_incorrect_rrb_key(rsn=True, ssid=None, passphrase=None):
  74     params = ft_params(rsn, ssid, passphrase)
  75     params['nas_identifier'] = "nas2.w1.fi"
  76     params['r1_key_holder'] = "000102030406"
  77     params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0ef1",
  78                        "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0ef2" ]
  79     params['r1kh'] = "02:00:00:00:03:00 00:01:02:03:04:05 300102030405060708090a0b0c0d0ef3"
  80     return params
  81
  82 def ft_params2_r0kh_mismatch(rsn=True, ssid=None, passphrase=None):
  83     params = ft_params(rsn, ssid, passphrase)
  84     params['nas_identifier'] = "nas2.w1.fi"
  85     params['r1_key_holder'] = "000102030406"
  86     params['r0kh'] = [ "12:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0e0f",
  87                        "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0e0f" ]
  88     params['r1kh'] = "12:00:00:00:03:00 10:01:02:03:04:05 300102030405060708090a0b0c0d0e0f"
  89     return params
  90
  91 def run_roams(dev, apdev, hapd0, hapd1, ssid, passphrase, over_ds=False, sae=False, eap=False, fail_test=False, roams=1):
  92     logger.info("Connect to first AP")
  93     if eap:
  94         dev.connect(ssid, key_mgmt="FT-EAP", proto="WPA2", ieee80211w="1",
  95                     eap="EKE", identity="eke user", password="hello")
  96     else:
  97         if sae:
  98             key_mgmt="FT-SAE"
  99         else:
 100             key_mgmt="FT-PSK"
 101         dev.connect(ssid, psk=passphrase, key_mgmt=key_mgmt, proto="WPA2",
 102                     ieee80211w="1")
 103     if dev.get_status_field('bssid') == apdev[0]['bssid']:
 104         ap1 = apdev[0]
 105         ap2 = apdev[1]
 106         hapd1ap = hapd0
 107         hapd2ap = hapd1
 108     else:
 109         ap1 = apdev[1]
 110         ap2 = apdev[0]
 111         hapd1ap = hapd1
 112         hapd2ap = hapd0
 113     hwsim_utils.test_connectivity(dev, hapd1ap)
 114
 115     dev.scan_for_bss(ap2['bssid'], freq="2412")
 116
 117     for i in range(0, roams):
 118         logger.info("Roam to the second AP")
 119         if over_ds:
 120             dev.roam_over_ds(ap2['bssid'], fail_test=fail_test)
 121         else:
 122             dev.roam(ap2['bssid'], fail_test=fail_test)
 123         if fail_test:
 124             return
 125         if dev.get_status_field('bssid') != ap2['bssid']:
 126             raise Exception("Did not connect to correct AP")
 127         if i == 0 or i == roams - 1:
 128             hwsim_utils.test_connectivity(dev, hapd2ap)
 129
 130         logger.info("Roam back to the first AP")
 131         if over_ds:
 132             dev.roam_over_ds(ap1['bssid'])
 133         else:
 134             dev.roam(ap1['bssid'])
 135         if dev.get_status_field('bssid') != ap1['bssid']:
 136             raise Exception("Did not connect to correct AP")
 137         if i == 0 or i == roams - 1:
 138             hwsim_utils.test_connectivity(dev, hapd1ap)
 139
 140 def test_ap_ft(dev, apdev):
 141     """WPA2-PSK-FT AP"""
 142     ssid = "test-ft"
 143     passphrase="12345678"
 144
 145     params = ft_params1(ssid=ssid, passphrase=passphrase)
 146     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 147     params = ft_params2(ssid=ssid, passphrase=passphrase)
 148     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 149
 150     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase)
 151     if "[WPA2-FT/PSK-CCMP]" not in dev[0].request("SCAN_RESULTS"):
 152         raise Exception("Scan results missing RSN element info")
 153
 154 def test_ap_ft_many(dev, apdev):
 155     """WPA2-PSK-FT AP multiple times"""
 156     ssid = "test-ft"
 157     passphrase="12345678"
 158
 159     params = ft_params1(ssid=ssid, passphrase=passphrase)
 160     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 161     params = ft_params2(ssid=ssid, passphrase=passphrase)
 162     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 163
 164     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, roams=50)
 165
 166 def test_ap_ft_mixed(dev, apdev):
 167     """WPA2-PSK-FT mixed-mode AP"""
 168     ssid = "test-ft-mixed"
 169     passphrase="12345678"
 170
 171     params = ft_params1(rsn=False, ssid=ssid, passphrase=passphrase)
 172     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
 173     key_mgmt = hapd.get_config()['key_mgmt']
 174     vals = key_mgmt.split(' ')
 175     if vals[0] != "WPA-PSK" or vals[1] != "FT-PSK":
 176         raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
 177     params = ft_params2(rsn=False, ssid=ssid, passphrase=passphrase)
 178     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 179
 180     run_roams(dev[0], apdev, hapd, hapd1, ssid, passphrase)
 181
 182 def test_ap_ft_pmf(dev, apdev):
 183     """WPA2-PSK-FT AP with PMF"""
 184     ssid = "test-ft"
 185     passphrase="12345678"
 186
 187     params = ft_params1(ssid=ssid, passphrase=passphrase)
 188     params["ieee80211w"] = "2";
 189     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 190     params = ft_params2(ssid=ssid, passphrase=passphrase)
 191     params["ieee80211w"] = "2";
 192     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 193
 194     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase)
 195
 196 def test_ap_ft_over_ds(dev, apdev):
 197     """WPA2-PSK-FT AP over DS"""
 198     ssid = "test-ft"
 199     passphrase="12345678"
 200
 201     params = ft_params1(ssid=ssid, passphrase=passphrase)
 202     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 203     params = ft_params2(ssid=ssid, passphrase=passphrase)
 204     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 205
 206     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True)
 207     check_mib(dev[0], [ ("dot11RSNAAuthenticationSuiteRequested", "00-0f-ac-4"),
 208                         ("dot11RSNAAuthenticationSuiteSelected", "00-0f-ac-4") ])
 209
 210 def test_ap_ft_over_ds_many(dev, apdev):
 211     """WPA2-PSK-FT AP over DS multiple times"""
 212     ssid = "test-ft"
 213     passphrase="12345678"
 214
 215     params = ft_params1(ssid=ssid, passphrase=passphrase)
 216     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 217     params = ft_params2(ssid=ssid, passphrase=passphrase)
 218     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 219
 220     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
 221               roams=50)
 222
 223 def test_ap_ft_over_ds_unknown_target(dev, apdev):
 224     """WPA2-PSK-FT AP"""
 225     ssid = "test-ft"
 226     passphrase="12345678"
 227
 228     params = ft_params1(ssid=ssid, passphrase=passphrase)
 229     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 230
 231     dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
 232                    scan_freq="2412")
 233     dev[0].roam_over_ds("02:11:22:33:44:55", fail_test=True)
 234
 235 def test_ap_ft_pmf_over_ds(dev, apdev):
 236     """WPA2-PSK-FT AP over DS with PMF"""
 237     ssid = "test-ft"
 238     passphrase="12345678"
 239
 240     params = ft_params1(ssid=ssid, passphrase=passphrase)
 241     params["ieee80211w"] = "2";
 242     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 243     params = ft_params2(ssid=ssid, passphrase=passphrase)
 244     params["ieee80211w"] = "2";
 245     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 246
 247     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True)
 248
 249 def test_ap_ft_over_ds_pull(dev, apdev):
 250     """WPA2-PSK-FT AP over DS (pull PMK)"""
 251     ssid = "test-ft"
 252     passphrase="12345678"
 253
 254     params = ft_params1(ssid=ssid, passphrase=passphrase)
 255     params["pmk_r1_push"] = "0"
 256     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 257     params = ft_params2(ssid=ssid, passphrase=passphrase)
 258     params["pmk_r1_push"] = "0"
 259     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 260
 261     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True)
 262
 263 def test_ap_ft_sae(dev, apdev):
 264     """WPA2-PSK-FT-SAE AP"""
 265     ssid = "test-ft"
 266     passphrase="12345678"
 267
 268     params = ft_params1(ssid=ssid, passphrase=passphrase)
 269     params['wpa_key_mgmt'] = "FT-SAE"
 270     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 271     params = ft_params2(ssid=ssid, passphrase=passphrase)
 272     params['wpa_key_mgmt'] = "FT-SAE"
 273     hapd = hostapd.add_ap(apdev[1]['ifname'], params)
 274     key_mgmt = hapd.get_config()['key_mgmt']
 275     if key_mgmt.split(' ')[0] != "FT-SAE":
 276         raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
 277
 278     dev[0].request("SET sae_groups ")
 279     run_roams(dev[0], apdev, hapd0, hapd, ssid, passphrase, sae=True)
 280
 281 def test_ap_ft_sae_over_ds(dev, apdev):
 282     """WPA2-PSK-FT-SAE AP over DS"""
 283     ssid = "test-ft"
 284     passphrase="12345678"
 285
 286     params = ft_params1(ssid=ssid, passphrase=passphrase)
 287     params['wpa_key_mgmt'] = "FT-SAE"
 288     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 289     params = ft_params2(ssid=ssid, passphrase=passphrase)
 290     params['wpa_key_mgmt'] = "FT-SAE"
 291     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 292
 293     dev[0].request("SET sae_groups ")
 294     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, sae=True,
 295               over_ds=True)
 296
 297 def test_ap_ft_eap(dev, apdev):
 298     """WPA2-EAP-FT AP"""
 299     ssid = "test-ft"
 300     passphrase="12345678"
 301
 302     radius = hostapd.radius_params()
 303     params = ft_params1(ssid=ssid, passphrase=passphrase)
 304     params['wpa_key_mgmt'] = "FT-EAP"
 305     params["ieee8021x"] = "1"
 306     params = dict(radius.items() + params.items())
 307     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
 308     key_mgmt = hapd.get_config()['key_mgmt']
 309     if key_mgmt.split(' ')[0] != "FT-EAP":
 310         raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
 311     params = ft_params2(ssid=ssid, passphrase=passphrase)
 312     params['wpa_key_mgmt'] = "FT-EAP"
 313     params["ieee8021x"] = "1"
 314     params = dict(radius.items() + params.items())
 315     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 316
 317     run_roams(dev[0], apdev, hapd, hapd1, ssid, passphrase, eap=True)
 318     if "[WPA2-FT/EAP-CCMP]" not in dev[0].request("SCAN_RESULTS"):
 319         raise Exception("Scan results missing RSN element info")
 320     check_mib(dev[0], [ ("dot11RSNAAuthenticationSuiteRequested", "00-0f-ac-3"),
 321                         ("dot11RSNAAuthenticationSuiteSelected", "00-0f-ac-3") ])
 322
 323 def test_ap_ft_eap_pull(dev, apdev):
 324     """WPA2-EAP-FT AP (pull PMK)"""
 325     ssid = "test-ft"
 326     passphrase="12345678"
 327
 328     radius = hostapd.radius_params()
 329     params = ft_params1(ssid=ssid, passphrase=passphrase)
 330     params['wpa_key_mgmt'] = "FT-EAP"
 331     params["ieee8021x"] = "1"
 332     params["pmk_r1_push"] = "0"
 333     params = dict(radius.items() + params.items())
 334     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
 335     key_mgmt = hapd.get_config()['key_mgmt']
 336     if key_mgmt.split(' ')[0] != "FT-EAP":
 337         raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
 338     params = ft_params2(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     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 344
 345     run_roams(dev[0], apdev, hapd, hapd1, ssid, passphrase, eap=True)
 346
 347 def test_ap_ft_mismatching_rrb_key_push(dev, apdev):
 348     """WPA2-PSK-FT AP over DS with mismatching RRB key (push)"""
 349     ssid = "test-ft"
 350     passphrase="12345678"
 351
 352     params = ft_params1(ssid=ssid, passphrase=passphrase)
 353     params["ieee80211w"] = "2";
 354     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 355     params = ft_params2_incorrect_rrb_key(ssid=ssid, passphrase=passphrase)
 356     params["ieee80211w"] = "2";
 357     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 358
 359     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
 360               fail_test=True)
 361
 362 def test_ap_ft_mismatching_rrb_key_pull(dev, apdev):
 363     """WPA2-PSK-FT AP over DS with mismatching RRB key (pull)"""
 364     ssid = "test-ft"
 365     passphrase="12345678"
 366
 367     params = ft_params1(ssid=ssid, passphrase=passphrase)
 368     params["pmk_r1_push"] = "0"
 369     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 370     params = ft_params2_incorrect_rrb_key(ssid=ssid, passphrase=passphrase)
 371     params["pmk_r1_push"] = "0"
 372     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 373
 374     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
 375               fail_test=True)
 376
 377 def test_ap_ft_mismatching_r0kh_id_pull(dev, apdev):
 378     """WPA2-PSK-FT AP over DS with mismatching R0KH-ID (pull)"""
 379     ssid = "test-ft"
 380     passphrase="12345678"
 381
 382     params = ft_params1(ssid=ssid, passphrase=passphrase)
 383     params["pmk_r1_push"] = "0"
 384     params["nas_identifier"] = "nas0.w1.fi"
 385     hostapd.add_ap(apdev[0]['ifname'], params)
 386     dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2")
 387
 388     params = ft_params2(ssid=ssid, passphrase=passphrase)
 389     params["pmk_r1_push"] = "0"
 390     hostapd.add_ap(apdev[1]['ifname'], params)
 391
 392     dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
 393     dev[0].roam_over_ds(apdev[1]['bssid'], fail_test=True)
 394
 395 def test_ap_ft_mismatching_rrb_r0kh_push(dev, apdev):
 396     """WPA2-PSK-FT AP over DS with mismatching R0KH key (push)"""
 397     ssid = "test-ft"
 398     passphrase="12345678"
 399
 400     params = ft_params1(ssid=ssid, passphrase=passphrase)
 401     params["ieee80211w"] = "2";
 402     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 403     params = ft_params2_r0kh_mismatch(ssid=ssid, passphrase=passphrase)
 404     params["ieee80211w"] = "2";
 405     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 406
 407     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
 408               fail_test=True)
 409
 410 def test_ap_ft_mismatching_rrb_r0kh_pull(dev, apdev):
 411     """WPA2-PSK-FT AP over DS with mismatching R0KH key (pull)"""
 412     ssid = "test-ft"
 413     passphrase="12345678"
 414
 415     params = ft_params1_r0kh_mismatch(ssid=ssid, passphrase=passphrase)
 416     params["pmk_r1_push"] = "0"
 417     hapd0 = hostapd.add_ap(apdev[0]['ifname'], params)
 418     params = ft_params2(ssid=ssid, passphrase=passphrase)
 419     params["pmk_r1_push"] = "0"
 420     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 421
 422     run_roams(dev[0], apdev, hapd0, hapd1, ssid, passphrase, over_ds=True,
 423               fail_test=True)
 424
 425 def test_ap_ft_gtk_rekey(dev, apdev):
 426     """WPA2-PSK-FT AP and GTK rekey"""
 427     ssid = "test-ft"
 428     passphrase="12345678"
 429
 430     params = ft_params1(ssid=ssid, passphrase=passphrase)
 431     params['wpa_group_rekey'] = '1'
 432     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
 433
 434     dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
 435                    ieee80211w="1")
 436
 437     ev = dev[0].wait_event(["WPA: Group rekeying completed"], timeout=2)
 438     if ev is None:
 439         raise Exception("GTK rekey timed out after initial association")
 440     hwsim_utils.test_connectivity(dev[0], hapd)
 441
 442     params = ft_params2(ssid=ssid, passphrase=passphrase)
 443     params['wpa_group_rekey'] = '1'
 444     hapd1 = hostapd.add_ap(apdev[1]['ifname'], params)
 445
 446     dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
 447     dev[0].roam(apdev[1]['bssid'])
 448     if dev[0].get_status_field('bssid') != apdev[1]['bssid']:
 449         raise Exception("Did not connect to correct AP")
 450     hwsim_utils.test_connectivity(dev[0], hapd1)
 451
 452     ev = dev[0].wait_event(["WPA: Group rekeying completed"], timeout=2)
 453     if ev is None:
 454         raise Exception("GTK rekey timed out after FT protocol")
 455     hwsim_utils.test_connectivity(dev[0], hapd1)