]>
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 | |
c9aa4308 | 12 | logger = logging.getLogger() |
cd7f1b9a JM |
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) |