]>
Commit | Line | Data |
---|---|---|
fdae3c40 JM |
1 | # IBSS test cases |
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 logging | |
c9aa4308 | 8 | logger = logging.getLogger() |
fdae3c40 | 9 | import time |
dcfb4d65 | 10 | import re |
fdae3c40 JM |
11 | |
12 | import hwsim_utils | |
13 | ||
14 | def connect_ibss_cmd(dev, id): | |
15 | dev.dump_monitor() | |
ca4fd182 | 16 | dev.select_network(id, freq="2412") |
fdae3c40 JM |
17 | |
18 | def wait_ibss_connection(dev): | |
e5a39838 | 19 | logger.info(dev.ifname + " waiting for IBSS start/join to complete") |
fdae3c40 JM |
20 | ev = dev.wait_event(["CTRL-EVENT-CONNECTED"], timeout=20) |
21 | if ev is None: | |
22 | raise Exception("Connection to the IBSS timed out") | |
dcfb4d65 JM |
23 | exp = r'<.>(CTRL-EVENT-CONNECTED) - Connection to ([0-9a-f:]*) completed.*' |
24 | s = re.split(exp, ev) | |
25 | if len(s) < 3: | |
26 | return None | |
27 | return s[2] | |
e5a39838 JM |
28 | |
29 | def wait_4way_handshake(dev1, dev2): | |
30 | logger.info(dev1.ifname + " waiting for 4-way handshake completion with " + dev2.ifname + " " + dev2.p2p_interface_addr()) | |
31 | ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr()], | |
32 | timeout=20) | |
33 | if ev is None: | |
34 | raise Exception("4-way handshake in IBSS timed out") | |
fdae3c40 | 35 | |
3d5b356a JM |
36 | def wait_4way_handshake2(dev1, dev2, dev3): |
37 | logger.info(dev1.ifname + " waiting for 4-way handshake completion with " + dev2.ifname + " " + dev2.p2p_interface_addr() + " and " + dev3.p2p_interface_addr()) | |
38 | ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr(), | |
39 | "IBSS-RSN-COMPLETED " + dev3.p2p_interface_addr()], | |
40 | timeout=20) | |
41 | if ev is None: | |
42 | raise Exception("4-way handshake in IBSS timed out") | |
43 | ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr(), | |
44 | "IBSS-RSN-COMPLETED " + dev3.p2p_interface_addr()], | |
45 | timeout=20) | |
46 | if ev is None: | |
47 | raise Exception("4-way handshake in IBSS timed out") | |
48 | ||
13adbe23 | 49 | def add_ibss(dev, ssid, psk=None, proto=None, key_mgmt=None, pairwise=None, group=None, beacon_int=None): |
fdae3c40 JM |
50 | id = dev.add_network() |
51 | dev.set_network(id, "mode", "1") | |
52 | dev.set_network(id, "frequency", "2412") | |
53 | dev.set_network_quoted(id, "ssid", ssid) | |
54 | if psk: | |
55 | dev.set_network_quoted(id, "psk", psk) | |
56 | if proto: | |
57 | dev.set_network(id, "proto", proto) | |
58 | if key_mgmt: | |
59 | dev.set_network(id, "key_mgmt", key_mgmt) | |
60 | if pairwise: | |
61 | dev.set_network(id, "pairwise", pairwise) | |
62 | if group: | |
63 | dev.set_network(id, "group", group) | |
13adbe23 JM |
64 | if beacon_int: |
65 | dev.set_network(id, "beacon_int", beacon_int) | |
fdae3c40 JM |
66 | return id |
67 | ||
68 | def add_ibss_rsn(dev, ssid): | |
69 | return add_ibss(dev, ssid, "12345678", "RSN", "WPA-PSK", "CCMP", "CCMP") | |
70 | ||
910d2ca7 JM |
71 | def add_ibss_wpa_none(dev, ssid): |
72 | return add_ibss(dev, ssid, "12345678", "WPA", "WPA-NONE", "TKIP", "TKIP") | |
73 | ||
fdae3c40 JM |
74 | def test_ibss_rsn(dev): |
75 | """IBSS RSN""" | |
76 | ssid="ibss-rsn" | |
77 | ||
78 | logger.info("Start IBSS on the first STA") | |
79 | id = add_ibss_rsn(dev[0], ssid) | |
80 | connect_ibss_cmd(dev[0], id) | |
dcfb4d65 | 81 | bssid0 = wait_ibss_connection(dev[0]) |
fdae3c40 JM |
82 | |
83 | logger.info("Join two STAs to the IBSS") | |
84 | ||
85 | id = add_ibss_rsn(dev[1], ssid) | |
86 | connect_ibss_cmd(dev[1], id) | |
dcfb4d65 JM |
87 | bssid1 = wait_ibss_connection(dev[1]) |
88 | if bssid0 != bssid1: | |
89 | logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1) | |
90 | # try to merge with a scan | |
91 | dev[1].scan() | |
e5a39838 JM |
92 | wait_4way_handshake(dev[0], dev[1]) |
93 | wait_4way_handshake(dev[1], dev[0]) | |
fdae3c40 JM |
94 | |
95 | id = add_ibss_rsn(dev[2], ssid) | |
96 | connect_ibss_cmd(dev[2], id) | |
dcfb4d65 JM |
97 | bssid2 = wait_ibss_connection(dev[2]) |
98 | if bssid0 != bssid2: | |
99 | logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2) | |
100 | # try to merge with a scan | |
101 | dev[2].scan() | |
e5a39838 | 102 | wait_4way_handshake(dev[0], dev[2]) |
3d5b356a | 103 | wait_4way_handshake2(dev[2], dev[0], dev[1]) |
fdae3c40 JM |
104 | |
105 | # Allow some time for all peers to complete key setup | |
106 | time.sleep(3) | |
107 | hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname) | |
108 | hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname) | |
fdae3c40 | 109 | hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname) |
910d2ca7 JM |
110 | |
111 | def test_ibss_wpa_none(dev): | |
112 | """IBSS WPA-None""" | |
113 | ssid="ibss-wpa-none" | |
114 | ||
115 | logger.info("Start IBSS on the first STA") | |
116 | id = add_ibss_wpa_none(dev[0], ssid) | |
117 | connect_ibss_cmd(dev[0], id) | |
118 | bssid0 = wait_ibss_connection(dev[0]) | |
119 | ||
120 | logger.info("Join two STAs to the IBSS") | |
121 | ||
122 | id = add_ibss_wpa_none(dev[1], ssid) | |
123 | connect_ibss_cmd(dev[1], id) | |
124 | id = add_ibss_wpa_none(dev[2], ssid) | |
125 | connect_ibss_cmd(dev[2], id) | |
126 | ||
127 | # This is a bit ugly, but no one really cares about WPA-None, so there may | |
128 | # not be enough justification to clean this up.. For now, wpa_supplicant | |
129 | # will show two connection events with mac80211_hwsim where the first one | |
130 | # comes with all zeros address. | |
131 | if bssid0 == "00:00:00:00:00:00": | |
132 | logger.info("Waiting for real BSSID on the first STA") | |
133 | bssid0 = wait_ibss_connection(dev[0]) | |
134 | ||
135 | bssid1 = wait_ibss_connection(dev[1]) | |
136 | if bssid0 != bssid1: | |
137 | logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1) | |
138 | bssid1 = wait_ibss_connection(dev[1]) | |
139 | ||
140 | bssid2 = wait_ibss_connection(dev[2]) | |
141 | if bssid0 != bssid2: | |
142 | logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2) | |
143 | bssid2 = wait_ibss_connection(dev[2]) | |
144 | ||
145 | print bssid0 | |
146 | print bssid1 | |
147 | print bssid2 | |
148 | ||
149 | # Allow some time for all peers to complete key setup | |
150 | time.sleep(1) | |
151 | ||
152 | # This is supposed to work, but looks like WPA-None does not work with | |
153 | # mac80211 currently.. | |
154 | try: | |
155 | hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname) | |
156 | except Exception, e: | |
157 | logger.info("Ignoring known connectivity failure: " + str(e)) | |
158 | try: | |
159 | hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname) | |
160 | except Exception, e: | |
161 | logger.info("Ignoring known connectivity failure: " + str(e)) | |
162 | try: | |
163 | hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname) | |
164 | except Exception, e: | |
165 | logger.info("Ignoring known connectivity failure: " + str(e)) | |
13adbe23 JM |
166 | |
167 | def test_ibss_open(dev): | |
168 | """IBSS open (no security)""" | |
169 | ssid="ibss" | |
170 | id = add_ibss(dev[0], ssid, key_mgmt="NONE", beacon_int="150") | |
171 | connect_ibss_cmd(dev[0], id) | |
172 | bssid0 = wait_ibss_connection(dev[0]) | |
173 | ||
174 | id = add_ibss(dev[1], ssid, key_mgmt="NONE", beacon_int="200") | |
175 | connect_ibss_cmd(dev[1], id) | |
176 | bssid1 = wait_ibss_connection(dev[1]) | |
177 | if bssid0 != bssid1: | |
178 | logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1) |