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

#!/usr/bin/python
#
# 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(__name__)

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):
    logger.info("Connect to first AP")
    dev.connect(ssid, psk=passphrase, key_mgmt="FT-PSK", 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")
    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")
    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)
Commit	Line	Data
cd7f1b9a JM	1	#!/usr/bin/python
	2	#
	3	# Fast BSS Transition tests
	4	# Copyright (c) 2013, Jouni Malinen <j@w1.fi>
	5	#
	6	# This software may be distributed under the terms of the BSD license.
	7	# See README for more details.
	8
	9	import time
	10	import subprocess
	11	import logging
	12	logger = logging.getLogger(__name__)
	13
	14	import hwsim_utils
	15	import hostapd
	16	from wlantest import Wlantest
	17
	18	def ft_base_rsn():
	19	params = { "wpa": "2",
	20	"wpa_key_mgmt": "FT-PSK",
	21	"rsn_pairwise": "CCMP" }
	22	return params
	23
	24	def ft_base_mixed():
	25	params = { "wpa": "3",
	26	"wpa_key_mgmt": "WPA-PSK FT-PSK",
	27	"wpa_pairwise": "TKIP",
	28	"rsn_pairwise": "CCMP" }
	29	return params
	30
	31	def ft_params(rsn=True, ssid=None, passphrase=None):
	32	if rsn:
	33	params = ft_base_rsn()
	34	else:
	35	params = ft_base_mixed()
	36	if ssid:
	37	params["ssid"] = ssid
	38	if passphrase:
	39	params["wpa_passphrase"] = passphrase
	40
	41	params["mobility_domain"] = "a1b2"
	42	params["r0_key_lifetime"] = "10000"
	43	params["pmk_r1_push"] = "1"
	44	params["reassociation_deadline"] = "1000"
	45	return params
	46
	47	def ft_params1(rsn=True, ssid=None, passphrase=None):
	48	params = ft_params(rsn, ssid, passphrase)
	49	params['nas_identifier'] = "nas1.w1.fi"
	50	params['r1_key_holder'] = "000102030405"
	51	params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 100102030405060708090a0b0c0d0e0f",
	52	"02:00:00:00:04:00 nas2.w1.fi 300102030405060708090a0b0c0d0e0f" ]
	53	params['r1kh'] = "02:00:00:00:04:00 00:01:02:03:04:06 200102030405060708090a0b0c0d0e0f"
	54	return params
	55
	56	def ft_params2(rsn=True, ssid=None, passphrase=None):
	57	params = ft_params(rsn, ssid, passphrase)
	58	params['nas_identifier'] = "nas2.w1.fi"
	59	params['r1_key_holder'] = "000102030406"
	60	params['r0kh'] = [ "02:00:00:00:03:00 nas1.w1.fi 200102030405060708090a0b0c0d0e0f",
	61	"02:00:00:00:04:00 nas2.w1.fi 000102030405060708090a0b0c0d0e0f" ]
	62	params['r1kh'] = "02:00:00:00:03:00 00:01:02:03:04:05 300102030405060708090a0b0c0d0e0f"
	63	return params
	64
65	def run_roams(dev, apdev, ssid, passphrase):
66	logger.info("Connect to first AP")
67	dev.connect(ssid, psk=passphrase, key_mgmt="FT-PSK", proto="WPA2",
68	ieee80211w="1")
69	if dev.get_status_field('bssid') == apdev[0]['bssid']:
70	ap1 = apdev[0]
71	ap2 = apdev[1]
72	else:
73	ap1 = apdev[1]
74	ap2 = apdev[0]
75	hwsim_utils.test_connectivity(dev.ifname, ap1['ifname'])
76
77	logger.info("Roam to the second AP")
78	dev.roam(ap2['bssid'])
79	if dev.get_status_field('bssid') != ap2['bssid']:
80	raise Exception("Did not connect to correct AP")
81	hwsim_utils.test_connectivity(dev.ifname, ap2['ifname'])
82
83	logger.info("Roam back to the first AP")
84	dev.roam(ap1['bssid'])
85	if dev.get_status_field('bssid') != ap1['bssid']:
86	raise Exception("Did not connect to correct AP")
87	hwsim_utils.test_connectivity(dev.ifname, ap1['ifname'])
88
89	def test_ap_ft(dev, apdev):
90	"""WPA2-PSK-FT AP"""
91	ssid = "test-ft"
92	passphrase="12345678"
93
94	params = ft_params1(ssid=ssid, passphrase=passphrase)
95	hostapd.add_ap(apdev[0]['ifname'], params)
96	params = ft_params2(ssid=ssid, passphrase=passphrase)
97	hostapd.add_ap(apdev[1]['ifname'], params)
98
99	run_roams(dev[0], apdev, ssid, passphrase)
100
101	def test_ap_ft_mixed(dev, apdev):
102	"""WPA2-PSK-FT mixed-mode AP"""
103	ssid = "test-ft-mixed"
104	passphrase="12345678"
105
106	params = ft_params1(rsn=False, ssid=ssid, passphrase=passphrase)
107	hostapd.add_ap(apdev[0]['ifname'], params)
108	params = ft_params2(rsn=False, 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_pmf(dev, apdev):
114	"""WPA2-PSK-FT AP with PMF"""
115	ssid = "test-ft"
116	passphrase="12345678"
117
118	params = ft_params1(ssid=ssid, passphrase=passphrase)
119	params["ieee80211w"] = "2";
120	hostapd.add_ap(apdev[0]['ifname'], params)
121	params = ft_params2(ssid=ssid, passphrase=passphrase)
122	params["ieee80211w"] = "2";
123	hostapd.add_ap(apdev[1]['ifname'], params)
124
125	run_roams(dev[0], apdev, ssid, passphrase)