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, 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     else:
 107         ap1 = apdev[1]
 108         ap2 = apdev[0]
 109     hwsim_utils.test_connectivity(dev.ifname, ap1['ifname'])
 110
 111     dev.scan_for_bss(ap2['bssid'], freq="2412")
 112
 113     for i in range(0, roams):
 114         logger.info("Roam to the second AP")
 115         if over_ds:
 116             dev.roam_over_ds(ap2['bssid'], fail_test=fail_test)
 117         else:
 118             dev.roam(ap2['bssid'], fail_test=fail_test)
 119         if fail_test:
 120             return
 121         if dev.get_status_field('bssid') != ap2['bssid']:
 122             raise Exception("Did not connect to correct AP")
 123         if i == 0 or i == roams - 1:
 124             hwsim_utils.test_connectivity(dev.ifname, ap2['ifname'])
 125
 126         logger.info("Roam back to the first AP")
 127         if over_ds:
 128             dev.roam_over_ds(ap1['bssid'])
 129         else:
 130             dev.roam(ap1['bssid'])
 131         if dev.get_status_field('bssid') != ap1['bssid']:
 132             raise Exception("Did not connect to correct AP")
 133         if i == 0 or i == roams - 1:
 134             hwsim_utils.test_connectivity(dev.ifname, ap1['ifname'])
 135
 136 def test_ap_ft(dev, apdev):
 137     """WPA2-PSK-FT AP"""
 138     ssid = "test-ft"
 139     passphrase="12345678"
 140
 141     params = ft_params1(ssid=ssid, passphrase=passphrase)
 142     hostapd.add_ap(apdev[0]['ifname'], params)
 143     params = ft_params2(ssid=ssid, passphrase=passphrase)
 144     hostapd.add_ap(apdev[1]['ifname'], params)
 145
 146     run_roams(dev[0], apdev, ssid, passphrase)
 147     if "[WPA2-FT/PSK-CCMP]" not in dev[0].request("SCAN_RESULTS"):
 148         raise Exception("Scan results missing RSN element info")
 149
 150 def test_ap_ft_many(dev, apdev):
 151     """WPA2-PSK-FT AP multiple times"""
 152     ssid = "test-ft"
 153     passphrase="12345678"
 154
 155     params = ft_params1(ssid=ssid, passphrase=passphrase)
 156     hostapd.add_ap(apdev[0]['ifname'], params)
 157     params = ft_params2(ssid=ssid, passphrase=passphrase)
 158     hostapd.add_ap(apdev[1]['ifname'], params)
 159
 160     run_roams(dev[0], apdev, ssid, passphrase, roams=50)
 161
 162 def test_ap_ft_mixed(dev, apdev):
 163     """WPA2-PSK-FT mixed-mode AP"""
 164     ssid = "test-ft-mixed"
 165     passphrase="12345678"
 166
 167     params = ft_params1(rsn=False, ssid=ssid, passphrase=passphrase)
 168     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
 169     key_mgmt = hapd.get_config()['key_mgmt']
 170     vals = key_mgmt.split(' ')
 171     if vals[0] != "WPA-PSK" or vals[1] != "FT-PSK":
 172         raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
 173     params = ft_params2(rsn=False, ssid=ssid, passphrase=passphrase)
 174     hostapd.add_ap(apdev[1]['ifname'], params)
 175
 176     run_roams(dev[0], apdev, ssid, passphrase)
 177
 178 def test_ap_ft_pmf(dev, apdev):
 179     """WPA2-PSK-FT AP with PMF"""
 180     ssid = "test-ft"
 181     passphrase="12345678"
 182
 183     params = ft_params1(ssid=ssid, passphrase=passphrase)
 184     params["ieee80211w"] = "2";
 185     hostapd.add_ap(apdev[0]['ifname'], params)
 186     params = ft_params2(ssid=ssid, passphrase=passphrase)
 187     params["ieee80211w"] = "2";
 188     hostapd.add_ap(apdev[1]['ifname'], params)
 189
 190     run_roams(dev[0], apdev, ssid, passphrase)
 191
 192 def test_ap_ft_over_ds(dev, apdev):
 193     """WPA2-PSK-FT AP over DS"""
 194     ssid = "test-ft"
 195     passphrase="12345678"
 196
 197     params = ft_params1(ssid=ssid, passphrase=passphrase)
 198     hostapd.add_ap(apdev[0]['ifname'], params)
 199     params = ft_params2(ssid=ssid, passphrase=passphrase)
 200     hostapd.add_ap(apdev[1]['ifname'], params)
 201
 202     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True)
 203     check_mib(dev[0], [ ("dot11RSNAAuthenticationSuiteRequested", "00-0f-ac-4"),
 204                         ("dot11RSNAAuthenticationSuiteSelected", "00-0f-ac-4") ])
 205
 206 def test_ap_ft_over_ds_many(dev, apdev):
 207     """WPA2-PSK-FT AP over DS multiple times"""
 208     ssid = "test-ft"
 209     passphrase="12345678"
 210
 211     params = ft_params1(ssid=ssid, passphrase=passphrase)
 212     hostapd.add_ap(apdev[0]['ifname'], params)
 213     params = ft_params2(ssid=ssid, passphrase=passphrase)
 214     hostapd.add_ap(apdev[1]['ifname'], params)
 215
 216     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True, roams=50)
 217
 218 def test_ap_ft_pmf_over_ds(dev, apdev):
 219     """WPA2-PSK-FT AP over DS with PMF"""
 220     ssid = "test-ft"
 221     passphrase="12345678"
 222
 223     params = ft_params1(ssid=ssid, passphrase=passphrase)
 224     params["ieee80211w"] = "2";
 225     hostapd.add_ap(apdev[0]['ifname'], params)
 226     params = ft_params2(ssid=ssid, passphrase=passphrase)
 227     params["ieee80211w"] = "2";
 228     hostapd.add_ap(apdev[1]['ifname'], params)
 229
 230     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True)
 231
 232 def test_ap_ft_over_ds_pull(dev, apdev):
 233     """WPA2-PSK-FT AP over DS (pull PMK)"""
 234     ssid = "test-ft"
 235     passphrase="12345678"
 236
 237     params = ft_params1(ssid=ssid, passphrase=passphrase)
 238     params["pmk_r1_push"] = "0"
 239     hostapd.add_ap(apdev[0]['ifname'], params)
 240     params = ft_params2(ssid=ssid, passphrase=passphrase)
 241     params["pmk_r1_push"] = "0"
 242     hostapd.add_ap(apdev[1]['ifname'], params)
 243
 244     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True)
 245
 246 def test_ap_ft_sae(dev, apdev):
 247     """WPA2-PSK-FT-SAE AP"""
 248     ssid = "test-ft"
 249     passphrase="12345678"
 250
 251     params = ft_params1(ssid=ssid, passphrase=passphrase)
 252     params['wpa_key_mgmt'] = "FT-SAE"
 253     hostapd.add_ap(apdev[0]['ifname'], params)
 254     params = ft_params2(ssid=ssid, passphrase=passphrase)
 255     params['wpa_key_mgmt'] = "FT-SAE"
 256     hapd = hostapd.add_ap(apdev[1]['ifname'], params)
 257     key_mgmt = hapd.get_config()['key_mgmt']
 258     if key_mgmt.split(' ')[0] != "FT-SAE":
 259         raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
 260
 261     run_roams(dev[0], apdev, ssid, passphrase, sae=True)
 262
 263 def test_ap_ft_sae_over_ds(dev, apdev):
 264     """WPA2-PSK-FT-SAE AP over DS"""
 265     ssid = "test-ft"
 266     passphrase="12345678"
 267
 268     params = ft_params1(ssid=ssid, passphrase=passphrase)
 269     params['wpa_key_mgmt'] = "FT-SAE"
 270     hostapd.add_ap(apdev[0]['ifname'], params)
 271     params = ft_params2(ssid=ssid, passphrase=passphrase)
 272     params['wpa_key_mgmt'] = "FT-SAE"
 273     hostapd.add_ap(apdev[1]['ifname'], params)
 274
 275     run_roams(dev[0], apdev, ssid, passphrase, sae=True, over_ds=True)
 276
 277 def test_ap_ft_eap(dev, apdev):
 278     """WPA2-EAP-FT AP"""
 279     ssid = "test-ft"
 280     passphrase="12345678"
 281
 282     radius = hostapd.radius_params()
 283     params = ft_params1(ssid=ssid, passphrase=passphrase)
 284     params['wpa_key_mgmt'] = "FT-EAP"
 285     params["ieee8021x"] = "1"
 286     params = dict(radius.items() + params.items())
 287     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
 288     key_mgmt = hapd.get_config()['key_mgmt']
 289     if key_mgmt.split(' ')[0] != "FT-EAP":
 290         raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
 291     params = ft_params2(ssid=ssid, passphrase=passphrase)
 292     params['wpa_key_mgmt'] = "FT-EAP"
 293     params["ieee8021x"] = "1"
 294     params = dict(radius.items() + params.items())
 295     hostapd.add_ap(apdev[1]['ifname'], params)
 296
 297     run_roams(dev[0], apdev, ssid, passphrase, eap=True)
 298     if "[WPA2-FT/EAP-CCMP]" not in dev[0].request("SCAN_RESULTS"):
 299         raise Exception("Scan results missing RSN element info")
 300     check_mib(dev[0], [ ("dot11RSNAAuthenticationSuiteRequested", "00-0f-ac-3"),
 301                         ("dot11RSNAAuthenticationSuiteSelected", "00-0f-ac-3") ])
 302
 303 def test_ap_ft_eap_pull(dev, apdev):
 304     """WPA2-EAP-FT AP (pull PMK)"""
 305     ssid = "test-ft"
 306     passphrase="12345678"
 307
 308     radius = hostapd.radius_params()
 309     params = ft_params1(ssid=ssid, passphrase=passphrase)
 310     params['wpa_key_mgmt'] = "FT-EAP"
 311     params["ieee8021x"] = "1"
 312     params["pmk_r1_push"] = "0"
 313     params = dict(radius.items() + params.items())
 314     hapd = hostapd.add_ap(apdev[0]['ifname'], params)
 315     key_mgmt = hapd.get_config()['key_mgmt']
 316     if key_mgmt.split(' ')[0] != "FT-EAP":
 317         raise Exception("Unexpected GET_CONFIG(key_mgmt): " + key_mgmt)
 318     params = ft_params2(ssid=ssid, passphrase=passphrase)
 319     params['wpa_key_mgmt'] = "FT-EAP"
 320     params["ieee8021x"] = "1"
 321     params["pmk_r1_push"] = "0"
 322     params = dict(radius.items() + params.items())
 323     hostapd.add_ap(apdev[1]['ifname'], params)
 324
 325     run_roams(dev[0], apdev, ssid, passphrase, eap=True)
 326
 327 def test_ap_ft_mismatching_rrb_key_push(dev, apdev):
 328     """WPA2-PSK-FT AP over DS with mismatching RRB key (push)"""
 329     ssid = "test-ft"
 330     passphrase="12345678"
 331
 332     params = ft_params1(ssid=ssid, passphrase=passphrase)
 333     params["ieee80211w"] = "2";
 334     hostapd.add_ap(apdev[0]['ifname'], params)
 335     params = ft_params2_incorrect_rrb_key(ssid=ssid, passphrase=passphrase)
 336     params["ieee80211w"] = "2";
 337     hostapd.add_ap(apdev[1]['ifname'], params)
 338
 339     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True, fail_test=True)
 340
 341 def test_ap_ft_mismatching_rrb_key_pull(dev, apdev):
 342     """WPA2-PSK-FT AP over DS with mismatching RRB key (pull)"""
 343     ssid = "test-ft"
 344     passphrase="12345678"
 345
 346     params = ft_params1(ssid=ssid, passphrase=passphrase)
 347     params["pmk_r1_push"] = "0"
 348     hostapd.add_ap(apdev[0]['ifname'], params)
 349     params = ft_params2_incorrect_rrb_key(ssid=ssid, passphrase=passphrase)
 350     params["pmk_r1_push"] = "0"
 351     hostapd.add_ap(apdev[1]['ifname'], params)
 352
 353     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True, fail_test=True)
 354
 355 def test_ap_ft_mismatching_r0kh_id_pull(dev, apdev):
 356     """WPA2-PSK-FT AP over DS with mismatching R0KH-ID (pull)"""
 357     ssid = "test-ft"
 358     passphrase="12345678"
 359
 360     params = ft_params1(ssid=ssid, passphrase=passphrase)
 361     params["pmk_r1_push"] = "0"
 362     params["nas_identifier"] = "nas0.w1.fi"
 363     hostapd.add_ap(apdev[0]['ifname'], params)
 364     dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2")
 365
 366     params = ft_params2(ssid=ssid, passphrase=passphrase)
 367     params["pmk_r1_push"] = "0"
 368     hostapd.add_ap(apdev[1]['ifname'], params)
 369
 370     dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
 371     dev[0].roam_over_ds(apdev[1]['bssid'], fail_test=True)
 372
 373 def test_ap_ft_mismatching_rrb_r0kh_push(dev, apdev):
 374     """WPA2-PSK-FT AP over DS with mismatching R0KH key (push)"""
 375     ssid = "test-ft"
 376     passphrase="12345678"
 377
 378     params = ft_params1(ssid=ssid, passphrase=passphrase)
 379     params["ieee80211w"] = "2";
 380     hostapd.add_ap(apdev[0]['ifname'], params)
 381     params = ft_params2_r0kh_mismatch(ssid=ssid, passphrase=passphrase)
 382     params["ieee80211w"] = "2";
 383     hostapd.add_ap(apdev[1]['ifname'], params)
 384
 385     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True, fail_test=True)
 386
 387 def test_ap_ft_mismatching_rrb_r0kh_pull(dev, apdev):
 388     """WPA2-PSK-FT AP over DS with mismatching R0KH key (pull)"""
 389     ssid = "test-ft"
 390     passphrase="12345678"
 391
 392     params = ft_params1_r0kh_mismatch(ssid=ssid, passphrase=passphrase)
 393     params["pmk_r1_push"] = "0"
 394     hostapd.add_ap(apdev[0]['ifname'], params)
 395     params = ft_params2(ssid=ssid, passphrase=passphrase)
 396     params["pmk_r1_push"] = "0"
 397     hostapd.add_ap(apdev[1]['ifname'], params)
 398
 399     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True, fail_test=True)
 400
 401 def test_ap_ft_gtk_rekey(dev, apdev):
 402     """WPA2-PSK-FT AP and GTK rekey"""
 403     ssid = "test-ft"
 404     passphrase="12345678"
 405
 406     params = ft_params1(ssid=ssid, passphrase=passphrase)
 407     params['wpa_group_rekey'] = '1'
 408     hostapd.add_ap(apdev[0]['ifname'], params)
 409
 410     dev[0].connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
 411                    ieee80211w="1")
 412
 413     ev = dev[0].wait_event(["WPA: Group rekeying completed"], timeout=2)
 414     if ev is None:
 415         raise Exception("GTK rekey timed out after initial association")
 416     hwsim_utils.test_connectivity(dev[0].ifname, apdev[0]['ifname'])
 417
 418     params = ft_params2(ssid=ssid, passphrase=passphrase)
 419     params['wpa_group_rekey'] = '1'
 420     hostapd.add_ap(apdev[1]['ifname'], params)
 421
 422     dev[0].scan_for_bss(apdev[1]['bssid'], freq="2412")
 423     dev[0].roam(apdev[1]['bssid'])
 424     if dev[0].get_status_field('bssid') != apdev[1]['bssid']:
 425         raise Exception("Did not connect to correct AP")
 426     hwsim_utils.test_connectivity(dev[0].ifname, apdev[1]['ifname'])
 427
 428     ev = dev[0].wait_event(["WPA: Group rekeying completed"], timeout=2)
 429     if ev is None:
 430         raise Exception("GTK rekey timed out after FT protocol")
 431     hwsim_utils.test_connectivity(dev[0].ifname, apdev[1]['ifname'])