[thirdparty/hostap.git] / tests / hwsim / test_ap_ft.py

# Fast BSS Transition tests
# Copyright (c) 2013, Jouni Malinen <j@w1.fi>
#
# This software may be distributed under the terms of the BSD license.
# See README for more details.

import time
import subprocess
import logging
logger = logging.getLogger()

import hwsim_utils
import hostapd
from wlantest import Wlantest

def ft_base_rsn():
    params = { "wpa": "2",
               "wpa_key_mgmt": "FT-PSK",
               "rsn_pairwise": "CCMP" }
    return params

def ft_base_mixed():
    params = { "wpa": "3",
               "wpa_key_mgmt": "WPA-PSK FT-PSK",
               "wpa_pairwise": "TKIP",
               "rsn_pairwise": "CCMP" }
    return params

def ft_params(rsn=True, ssid=None, passphrase=None):
    if rsn:
        params = ft_base_rsn()
    else:
        params = ft_base_mixed()
    if ssid:
        params["ssid"] = ssid
    if passphrase:
        params["wpa_passphrase"] = passphrase

    params["mobility_domain"] = "a1b2"
    params["r0_key_lifetime"] = "10000"
    params["pmk_r1_push"] = "1"
    params["reassociation_deadline"] = "1000"
    return params

def ft_params1(rsn=True, ssid=None, passphrase=None):
    params = ft_params(rsn, ssid, passphrase)
    params['nas_identifier'] = "nas1.w1.fi"
    params['r1_key_holder'] = "000102030405"
    params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 100102030405060708090a0b0c0d0e0f",
                       "02:00:00:00:04:00 nas2.w1.fi 300102030405060708090a0b0c0d0e0f" ]
    params['r1kh'] = "02:00:00:00:04:00 00:01:02:03:04:06 200102030405060708090a0b0c0d0e0f"
    return params

def ft_params2(rsn=True, ssid=None, passphrase=None):
    params = ft_params(rsn, ssid, passphrase)
    params['nas_identifier'] = "nas2.w1.fi"
    params['r1_key_holder'] = "000102030406"
    params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0e0f",
                       "02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0e0f" ]
    params['r1kh'] = "02:00:00:00:03:00 00:01:02:03:04:05 300102030405060708090a0b0c0d0e0f"
    return params

def run_roams(dev, apdev, ssid, passphrase, over_ds=False, sae=False, eap=False):
    logger.info("Connect to first AP")
    if eap:
        dev.connect(ssid, key_mgmt="FT-EAP", proto="WPA2", ieee80211w="1",
                    eap="EKE", identity="eke user", password="hello")
    else:
        if sae:
            key_mgmt="FT-SAE"
        else:
            key_mgmt="FT-PSK"
        dev.connect(ssid, psk=passphrase, key_mgmt=key_mgmt, proto="WPA2",
                    ieee80211w="1")
    if dev.get_status_field('bssid') == apdev[0]['bssid']:
        ap1 = apdev[0]
        ap2 = apdev[1]
    else:
        ap1 = apdev[1]
        ap2 = apdev[0]
    hwsim_utils.test_connectivity(dev.ifname, ap1['ifname'])

    logger.info("Roam to the second AP")
    if over_ds:
        dev.roam_over_ds(ap2['bssid'])
    else:
        dev.roam(ap2['bssid'])
    if dev.get_status_field('bssid') != ap2['bssid']:
        raise Exception("Did not connect to correct AP")
    hwsim_utils.test_connectivity(dev.ifname, ap2['ifname'])

    logger.info("Roam back to the first AP")
    if over_ds:
        dev.roam_over_ds(ap1['bssid'])
    else:
        dev.roam(ap1['bssid'])
    if dev.get_status_field('bssid') != ap1['bssid']:
        raise Exception("Did not connect to correct AP")
    hwsim_utils.test_connectivity(dev.ifname, ap1['ifname'])

def test_ap_ft(dev, apdev):
    """WPA2-PSK-FT AP"""
    ssid = "test-ft"
    passphrase="12345678"

    params = ft_params1(ssid=ssid, passphrase=passphrase)
    hostapd.add_ap(apdev[0]['ifname'], params)
    params = ft_params2(ssid=ssid, passphrase=passphrase)
    hostapd.add_ap(apdev[1]['ifname'], params)

    run_roams(dev[0], apdev, ssid, passphrase)

def test_ap_ft_mixed(dev, apdev):
    """WPA2-PSK-FT mixed-mode AP"""
    ssid = "test-ft-mixed"
    passphrase="12345678"

    params = ft_params1(rsn=False, ssid=ssid, passphrase=passphrase)
    hostapd.add_ap(apdev[0]['ifname'], params)
    params = ft_params2(rsn=False, ssid=ssid, passphrase=passphrase)
    hostapd.add_ap(apdev[1]['ifname'], params)

    run_roams(dev[0], apdev, ssid, passphrase)

def test_ap_ft_pmf(dev, apdev):
    """WPA2-PSK-FT AP with PMF"""
    ssid = "test-ft"
    passphrase="12345678"

    params = ft_params1(ssid=ssid, passphrase=passphrase)
    params["ieee80211w"] = "2";
    hostapd.add_ap(apdev[0]['ifname'], params)
    params = ft_params2(ssid=ssid, passphrase=passphrase)
    params["ieee80211w"] = "2";
    hostapd.add_ap(apdev[1]['ifname'], params)

    run_roams(dev[0], apdev, ssid, passphrase)

def test_ap_ft_over_ds(dev, apdev):
    """WPA2-PSK-FT AP over DS"""
    ssid = "test-ft"
    passphrase="12345678"

    params = ft_params1(ssid=ssid, passphrase=passphrase)
    hostapd.add_ap(apdev[0]['ifname'], params)
    params = ft_params2(ssid=ssid, passphrase=passphrase)
    hostapd.add_ap(apdev[1]['ifname'], params)

    run_roams(dev[0], apdev, ssid, passphrase, over_ds=True)

def test_ap_ft_pmf_over_ds(dev, apdev):
    """WPA2-PSK-FT AP over DS with PMF"""
    ssid = "test-ft"
    passphrase="12345678"

    params = ft_params1(ssid=ssid, passphrase=passphrase)
    params["ieee80211w"] = "2";
    hostapd.add_ap(apdev[0]['ifname'], params)
    params = ft_params2(ssid=ssid, passphrase=passphrase)
    params["ieee80211w"] = "2";
    hostapd.add_ap(apdev[1]['ifname'], params)

    run_roams(dev[0], apdev, ssid, passphrase, over_ds=True)

def test_ap_ft_sae(dev, apdev):
    """WPA2-PSK-FT-SAE AP"""
    ssid = "test-ft"
    passphrase="12345678"

    params = ft_params1(ssid=ssid, passphrase=passphrase)
    params['wpa_key_mgmt'] = "FT-SAE"
    hostapd.add_ap(apdev[0]['ifname'], params)
    params = ft_params2(ssid=ssid, passphrase=passphrase)
    params['wpa_key_mgmt'] = "FT-SAE"
    hostapd.add_ap(apdev[1]['ifname'], params)

    run_roams(dev[0], apdev, ssid, passphrase, sae=True)

def test_ap_ft_sae_over_ds(dev, apdev):
    """WPA2-PSK-FT-SAE AP over DS"""
    ssid = "test-ft"
    passphrase="12345678"

    params = ft_params1(ssid=ssid, passphrase=passphrase)
    params['wpa_key_mgmt'] = "FT-SAE"
    hostapd.add_ap(apdev[0]['ifname'], params)
    params = ft_params2(ssid=ssid, passphrase=passphrase)
    params['wpa_key_mgmt'] = "FT-SAE"
    hostapd.add_ap(apdev[1]['ifname'], params)

    run_roams(dev[0], apdev, ssid, passphrase, sae=True, over_ds=True)

def test_ap_ft_eap(dev, apdev):
    """WPA2-EAP-FT AP"""
    ssid = "test-ft"
    passphrase="12345678"

    radius = hostapd.radius_params()
    params = ft_params1(ssid=ssid, passphrase=passphrase)
    params['wpa_key_mgmt'] = "FT-EAP"
    params["ieee8021x"] = "1"
    params = dict(radius.items() + params.items())
    hostapd.add_ap(apdev[0]['ifname'], params)
    params = ft_params2(ssid=ssid, passphrase=passphrase)
    params['wpa_key_mgmt'] = "FT-EAP"
    params["ieee8021x"] = "1"
    params = dict(radius.items() + params.items())
    hostapd.add_ap(apdev[1]['ifname'], params)

    run_roams(dev[0], apdev, ssid, passphrase, eap=True)
Commit	Line	Data
cd7f1b9a JM	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
c9aa4308	10	logger = logging.getLogger()
cd7f1b9a JM	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
6f62809b	63	def run_roams(dev, apdev, ssid, passphrase, over_ds=False, sae=False, eap=False):
cd7f1b9a	64	logger.info("Connect to first AP")
6f62809b JM	65	if eap:
	66	dev.connect(ssid, key_mgmt="FT-EAP", proto="WPA2", ieee80211w="1",
	67	eap="EKE", identity="eke user", password="hello")
6e658cc4	68	else:
6f62809b JM	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")
cd7f1b9a JM	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")
b553eab1 JM	84	if over_ds:
	85	dev.roam_over_ds(ap2['bssid'])
	86	else:
	87	dev.roam(ap2['bssid'])
cd7f1b9a JM	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")
b553eab1 JM	93	if over_ds:
	94	dev.roam_over_ds(ap1['bssid'])
	95	else:
	96	dev.roam(ap1['bssid'])
cd7f1b9a JM	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)
b553eab1 JM	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)
6e658cc4 JM	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)
6f62809b JM	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)