]>
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() | |
16 | dev.select_network(id) | |
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 | ||
fdae3c40 JM |
49 | def add_ibss(dev, ssid, psk=None, proto=None, key_mgmt=None, pairwise=None, group=None): |
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) | |
64 | return id | |
65 | ||
66 | def add_ibss_rsn(dev, ssid): | |
67 | return add_ibss(dev, ssid, "12345678", "RSN", "WPA-PSK", "CCMP", "CCMP") | |
68 | ||
910d2ca7 JM |
69 | def add_ibss_wpa_none(dev, ssid): |
70 | return add_ibss(dev, ssid, "12345678", "WPA", "WPA-NONE", "TKIP", "TKIP") | |
71 | ||
fdae3c40 JM |
72 | def test_ibss_rsn(dev): |
73 | """IBSS RSN""" | |
74 | ssid="ibss-rsn" | |
75 | ||
76 | logger.info("Start IBSS on the first STA") | |
77 | id = add_ibss_rsn(dev[0], ssid) | |
78 | connect_ibss_cmd(dev[0], id) | |
dcfb4d65 | 79 | bssid0 = wait_ibss_connection(dev[0]) |
fdae3c40 JM |
80 | |
81 | logger.info("Join two STAs to the IBSS") | |
82 | ||
83 | id = add_ibss_rsn(dev[1], ssid) | |
84 | connect_ibss_cmd(dev[1], id) | |
dcfb4d65 JM |
85 | bssid1 = wait_ibss_connection(dev[1]) |
86 | if bssid0 != bssid1: | |
87 | logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1) | |
88 | # try to merge with a scan | |
89 | dev[1].scan() | |
e5a39838 JM |
90 | wait_4way_handshake(dev[0], dev[1]) |
91 | wait_4way_handshake(dev[1], dev[0]) | |
fdae3c40 JM |
92 | |
93 | id = add_ibss_rsn(dev[2], ssid) | |
94 | connect_ibss_cmd(dev[2], id) | |
dcfb4d65 JM |
95 | bssid2 = wait_ibss_connection(dev[2]) |
96 | if bssid0 != bssid2: | |
97 | logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2) | |
98 | # try to merge with a scan | |
99 | dev[2].scan() | |
e5a39838 | 100 | wait_4way_handshake(dev[0], dev[2]) |
3d5b356a | 101 | wait_4way_handshake2(dev[2], dev[0], dev[1]) |
fdae3c40 JM |
102 | |
103 | # Allow some time for all peers to complete key setup | |
104 | time.sleep(3) | |
105 | hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname) | |
106 | hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname) | |
fdae3c40 | 107 | hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname) |
910d2ca7 JM |
108 | |
109 | def test_ibss_wpa_none(dev): | |
110 | """IBSS WPA-None""" | |
111 | ssid="ibss-wpa-none" | |
112 | ||
113 | logger.info("Start IBSS on the first STA") | |
114 | id = add_ibss_wpa_none(dev[0], ssid) | |
115 | connect_ibss_cmd(dev[0], id) | |
116 | bssid0 = wait_ibss_connection(dev[0]) | |
117 | ||
118 | logger.info("Join two STAs to the IBSS") | |
119 | ||
120 | id = add_ibss_wpa_none(dev[1], ssid) | |
121 | connect_ibss_cmd(dev[1], id) | |
122 | id = add_ibss_wpa_none(dev[2], ssid) | |
123 | connect_ibss_cmd(dev[2], id) | |
124 | ||
125 | # This is a bit ugly, but no one really cares about WPA-None, so there may | |
126 | # not be enough justification to clean this up.. For now, wpa_supplicant | |
127 | # will show two connection events with mac80211_hwsim where the first one | |
128 | # comes with all zeros address. | |
129 | if bssid0 == "00:00:00:00:00:00": | |
130 | logger.info("Waiting for real BSSID on the first STA") | |
131 | bssid0 = wait_ibss_connection(dev[0]) | |
132 | ||
133 | bssid1 = wait_ibss_connection(dev[1]) | |
134 | if bssid0 != bssid1: | |
135 | logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1) | |
136 | bssid1 = wait_ibss_connection(dev[1]) | |
137 | ||
138 | bssid2 = wait_ibss_connection(dev[2]) | |
139 | if bssid0 != bssid2: | |
140 | logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2) | |
141 | bssid2 = wait_ibss_connection(dev[2]) | |
142 | ||
143 | print bssid0 | |
144 | print bssid1 | |
145 | print bssid2 | |
146 | ||
147 | # Allow some time for all peers to complete key setup | |
148 | time.sleep(1) | |
149 | ||
150 | # This is supposed to work, but looks like WPA-None does not work with | |
151 | # mac80211 currently.. | |
152 | try: | |
153 | hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname) | |
154 | except Exception, e: | |
155 | logger.info("Ignoring known connectivity failure: " + str(e)) | |
156 | try: | |
157 | hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname) | |
158 | except Exception, e: | |
159 | logger.info("Ignoring known connectivity failure: " + str(e)) | |
160 | try: | |
161 | hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname) | |
162 | except Exception, e: | |
163 | logger.info("Ignoring known connectivity failure: " + str(e)) |