tests/hwsim/test_ap_ft.py

   1 # Fast BSS Transition tests
   2 # Copyright (c) 2013, 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
  16 def ft_base_rsn():
  17     params = { "wpa": "2",
  18                "wpa_key_mgmt": "FT-PSK",
  19                "rsn_pairwise": "CCMP" }
  20     return params
  21
  22 def ft_base_mixed():
  23     params = { "wpa": "3",
  24                "wpa_key_mgmt": "WPA-PSK FT-PSK",
  25                "wpa_pairwise": "TKIP",
  26                "rsn_pairwise": "CCMP" }
  27     return params
  28
  29 def ft_params(rsn=True, ssid=None, passphrase=None):
  30     if rsn:
  31         params = ft_base_rsn()
  32     else:
  33         params = ft_base_mixed()
  34     if ssid:
  35         params["ssid"] = ssid
  36     if passphrase:
  37         params["wpa_passphrase"] = passphrase
  38
  39     params["mobility_domain"] = "a1b2"
  40     params["r0_key_lifetime"] = "10000"
  41     params["pmk_r1_push"] = "1"
  42     params["reassociation_deadline"] = "1000"
  43     return params
  44
  45 def ft_params1(rsn=True, ssid=None, passphrase=None):
  46     params = ft_params(rsn, ssid, passphrase)
  47     params['nas_identifier'] = "nas1.w1.fi"
  48     params['r1_key_holder'] = "000102030405"
  49     params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 100102030405060708090a0b0c0d0e0f",
  50                        "02:00:00:00:04:00 nas2.w1.fi 300102030405060708090a0b0c0d0e0f" ]
  51     params['r1kh'] = "02:00:00:00:04:00 00:01:02:03:04:06 200102030405060708090a0b0c0d0e0f"
  52     return params
  53
  54 def ft_params2(rsn=True, ssid=None, passphrase=None):
  55     params = ft_params(rsn, ssid, passphrase)
  56     params['nas_identifier'] = "nas2.w1.fi"
  57     params['r1_key_holder'] = "000102030406"
  58     params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0e0f",
  59                        "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0e0f" ]
  60     params['r1kh'] = "02:00:00:00:03:00 00:01:02:03:04:05 300102030405060708090a0b0c0d0e0f"
  61     return params
  62
  63 def run_roams(dev, apdev, ssid, passphrase, over_ds=False, sae=False, eap=False):
  64     logger.info("Connect to first AP")
  65     if eap:
  66         dev.connect(ssid, key_mgmt="FT-EAP", proto="WPA2", ieee80211w="1",
  67                     eap="EKE", identity="eke user", password="hello")
  68     else:
  69         if sae:
  70             key_mgmt="FT-SAE"
  71         else:
  72             key_mgmt="FT-PSK"
  73         dev.connect(ssid, psk=passphrase, key_mgmt=key_mgmt, proto="WPA2",
  74                     ieee80211w="1")
  75     if dev.get_status_field('bssid') == apdev[0]['bssid']:
  76         ap1 = apdev[0]
  77         ap2 = apdev[1]
  78     else:
  79         ap1 = apdev[1]
  80         ap2 = apdev[0]
  81     hwsim_utils.test_connectivity(dev.ifname, ap1['ifname'])
  82
  83     logger.info("Roam to the second AP")
  84     if over_ds:
  85         dev.roam_over_ds(ap2['bssid'])
  86     else:
  87         dev.roam(ap2['bssid'])
  88     if dev.get_status_field('bssid') != ap2['bssid']:
  89         raise Exception("Did not connect to correct AP")
  90     hwsim_utils.test_connectivity(dev.ifname, ap2['ifname'])
  91
  92     logger.info("Roam back to the first AP")
  93     if over_ds:
  94         dev.roam_over_ds(ap1['bssid'])
  95     else:
  96         dev.roam(ap1['bssid'])
  97     if dev.get_status_field('bssid') != ap1['bssid']:
  98         raise Exception("Did not connect to correct AP")
  99     hwsim_utils.test_connectivity(dev.ifname, ap1['ifname'])
 100
 101 def test_ap_ft(dev, apdev):
 102     """WPA2-PSK-FT AP"""
 103     ssid = "test-ft"
 104     passphrase="12345678"
 105
 106     params = ft_params1(ssid=ssid, passphrase=passphrase)
 107     hostapd.add_ap(apdev[0]['ifname'], params)
 108     params = ft_params2(ssid=ssid, passphrase=passphrase)
 109     hostapd.add_ap(apdev[1]['ifname'], params)
 110
 111     run_roams(dev[0], apdev, ssid, passphrase)
 112
 113 def test_ap_ft_mixed(dev, apdev):
 114     """WPA2-PSK-FT mixed-mode AP"""
 115     ssid = "test-ft-mixed"
 116     passphrase="12345678"
 117
 118     params = ft_params1(rsn=False, ssid=ssid, passphrase=passphrase)
 119     hostapd.add_ap(apdev[0]['ifname'], params)
 120     params = ft_params2(rsn=False, ssid=ssid, passphrase=passphrase)
 121     hostapd.add_ap(apdev[1]['ifname'], params)
 122
 123     run_roams(dev[0], apdev, ssid, passphrase)
 124
 125 def test_ap_ft_pmf(dev, apdev):
 126     """WPA2-PSK-FT AP with PMF"""
 127     ssid = "test-ft"
 128     passphrase="12345678"
 129
 130     params = ft_params1(ssid=ssid, passphrase=passphrase)
 131     params["ieee80211w"] = "2";
 132     hostapd.add_ap(apdev[0]['ifname'], params)
 133     params = ft_params2(ssid=ssid, passphrase=passphrase)
 134     params["ieee80211w"] = "2";
 135     hostapd.add_ap(apdev[1]['ifname'], params)
 136
 137     run_roams(dev[0], apdev, ssid, passphrase)
 138
 139 def test_ap_ft_over_ds(dev, apdev):
 140     """WPA2-PSK-FT AP over DS"""
 141     ssid = "test-ft"
 142     passphrase="12345678"
 143
 144     params = ft_params1(ssid=ssid, passphrase=passphrase)
 145     hostapd.add_ap(apdev[0]['ifname'], params)
 146     params = ft_params2(ssid=ssid, passphrase=passphrase)
 147     hostapd.add_ap(apdev[1]['ifname'], params)
 148
 149     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True)
 150
 151 def test_ap_ft_pmf_over_ds(dev, apdev):
 152     """WPA2-PSK-FT AP over DS with PMF"""
 153     ssid = "test-ft"
 154     passphrase="12345678"
 155
 156     params = ft_params1(ssid=ssid, passphrase=passphrase)
 157     params["ieee80211w"] = "2";
 158     hostapd.add_ap(apdev[0]['ifname'], params)
 159     params = ft_params2(ssid=ssid, passphrase=passphrase)
 160     params["ieee80211w"] = "2";
 161     hostapd.add_ap(apdev[1]['ifname'], params)
 162
 163     run_roams(dev[0], apdev, ssid, passphrase, over_ds=True)
 164
 165 def test_ap_ft_sae(dev, apdev):
 166     """WPA2-PSK-FT-SAE AP"""
 167     ssid = "test-ft"
 168     passphrase="12345678"
 169
 170     params = ft_params1(ssid=ssid, passphrase=passphrase)
 171     params['wpa_key_mgmt'] = "FT-SAE"
 172     hostapd.add_ap(apdev[0]['ifname'], params)
 173     params = ft_params2(ssid=ssid, passphrase=passphrase)
 174     params['wpa_key_mgmt'] = "FT-SAE"
 175     hostapd.add_ap(apdev[1]['ifname'], params)
 176
 177     run_roams(dev[0], apdev, ssid, passphrase, sae=True)
 178
 179 def test_ap_ft_sae_over_ds(dev, apdev):
 180     """WPA2-PSK-FT-SAE AP over DS"""
 181     ssid = "test-ft"
 182     passphrase="12345678"
 183
 184     params = ft_params1(ssid=ssid, passphrase=passphrase)
 185     params['wpa_key_mgmt'] = "FT-SAE"
 186     hostapd.add_ap(apdev[0]['ifname'], params)
 187     params = ft_params2(ssid=ssid, passphrase=passphrase)
 188     params['wpa_key_mgmt'] = "FT-SAE"
 189     hostapd.add_ap(apdev[1]['ifname'], params)
 190
 191     run_roams(dev[0], apdev, ssid, passphrase, sae=True, over_ds=True)
 192
 193 def test_ap_ft_eap(dev, apdev):
 194     """WPA2-EAP-FT AP"""
 195     ssid = "test-ft"
 196     passphrase="12345678"
 197
 198     radius = hostapd.radius_params()
 199     params = ft_params1(ssid=ssid, passphrase=passphrase)
 200     params['wpa_key_mgmt'] = "FT-EAP"
 201     params["ieee8021x"] = "1"
 202     params = dict(radius.items() + params.items())
 203     hostapd.add_ap(apdev[0]['ifname'], params)
 204     params = ft_params2(ssid=ssid, passphrase=passphrase)
 205     params['wpa_key_mgmt'] = "FT-EAP"
 206     params["ieee8021x"] = "1"
 207     params = dict(radius.items() + params.items())
 208     hostapd.add_ap(apdev[1]['ifname'], params)
 209
 210     run_roams(dev[0], apdev, ssid, passphrase, eap=True)