]>
git.ipfire.org Git - thirdparty/hostap.git/blob - tests/hwsim/test_p2p_persistent.py
3 # P2P persistent group test cases
4 # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
6 # This software may be distributed under the terms of the BSD license.
7 # See README for more details.
10 logger
= logging
.getLogger()
13 from test_p2p_autogo
import connect_cli
15 def go_neg_pin_authorized_persistent(i_dev
, r_dev
, i_intent
=None, r_intent
=None, i_method
='enter', r_method
='display', test_data
=True):
18 pin
= r_dev
.wps_read_pin()
19 logger
.info("Start GO negotiation " + i_dev
.ifname
+ " -> " + r_dev
.ifname
)
20 r_dev
.p2p_go_neg_auth(i_dev
.p2p_dev_addr(), pin
, r_method
,
21 go_intent
=r_intent
, persistent
=True)
22 i_res
= i_dev
.p2p_go_neg_init(r_dev
.p2p_dev_addr(), pin
, i_method
,
23 timeout
=20, go_intent
=i_intent
,
25 r_res
= r_dev
.p2p_go_neg_auth_result()
26 logger
.debug("i_res: " + str(i_res
))
27 logger
.debug("r_res: " + str(r_res
))
30 logger
.info("Group formed")
32 hwsim_utils
.test_connectivity_p2p(r_dev
, i_dev
)
35 def terminate_group(go
, cli
):
36 logger
.info("Terminate persistent group")
38 cli
.wait_go_ending_session()
40 def invite(inv
, resp
, extra
=None):
41 addr
= resp
.p2p_dev_addr()
42 resp
.request("SET persistent_reconnect 1")
44 if not inv
.discover_peer(addr
, social
=True):
45 raise Exception("Peer " + addr
+ " not found")
47 peer
= inv
.get_peer(addr
)
48 cmd
= "P2P_INVITE persistent=" + peer
['persistent'] + " peer=" + addr
50 cmd
= cmd
+ " " + extra
;
51 inv
.global_request(cmd
)
53 def check_result(go
, cli
):
54 ev
= go
.wait_global_event(["P2P-GROUP-STARTED"], timeout
=30)
56 raise Exception("Timeout on group re-invocation (on GO)")
57 if "[PERSISTENT]" not in ev
:
58 raise Exception("Re-invoked group not marked persistent")
59 go_res
= go
.group_form_result(ev
)
60 if go_res
['role'] != 'GO':
61 raise Exception("Persistent group GO did not become GO")
62 if not go_res
['persistent']:
63 raise Exception("Persistent group not re-invoked as persistent (GO)")
64 ev
= cli
.wait_global_event(["P2P-GROUP-STARTED"], timeout
=30)
66 raise Exception("Timeout on group re-invocation (on client)")
67 if "[PERSISTENT]" not in ev
:
68 raise Exception("Re-invoked group not marked persistent")
69 cli_res
= cli
.group_form_result(ev
)
70 if cli_res
['role'] != 'client':
71 raise Exception("Persistent group client did not become client")
72 if not cli_res
['persistent']:
73 raise Exception("Persistent group not re-invoked as persistent (cli)")
74 return [go_res
, cli_res
]
76 def form(go
, cli
, test_data
=True):
77 logger
.info("Form a persistent group")
78 [i_res
, r_res
] = go_neg_pin_authorized_persistent(i_dev
=go
, i_intent
=15,
79 r_dev
=cli
, r_intent
=0,
81 if not i_res
['persistent'] or not r_res
['persistent']:
82 raise Exception("Formed group was not persistent")
83 terminate_group(go
, cli
)
85 def invite_from_cli(go
, cli
):
86 logger
.info("Re-invoke persistent group from client")
89 hwsim_utils
.test_connectivity_p2p(go
, cli
)
90 terminate_group(go
, cli
)
92 def invite_from_go(go
, cli
):
93 logger
.info("Re-invoke persistent group from GO")
96 hwsim_utils
.test_connectivity_p2p(go
, cli
)
97 terminate_group(go
, cli
)
99 def test_persistent_group(dev
):
100 """P2P persistent group formation and re-invocation"""
102 invite_from_cli(dev
[0], dev
[1])
103 invite_from_go(dev
[0], dev
[1])
105 def test_persistent_group_per_sta_psk(dev
):
106 """P2P persistent group formation and re-invocation using per-client PSK"""
107 addr0
= dev
[0].p2p_dev_addr()
108 addr1
= dev
[1].p2p_dev_addr()
109 addr2
= dev
[2].p2p_dev_addr()
110 dev
[0].request("P2P_SET per_sta_psk 1")
111 logger
.info("Form a persistent group")
112 [i_res
, r_res
] = go_neg_pin_authorized_persistent(i_dev
=dev
[0], i_intent
=15,
113 r_dev
=dev
[1], r_intent
=0)
114 if not i_res
['persistent'] or not r_res
['persistent']:
115 raise Exception("Formed group was not persistent")
117 logger
.info("Join another client to the group")
118 pin
= dev
[2].wps_read_pin()
119 dev
[0].p2p_go_authorize_client(pin
)
120 c_res
= dev
[2].p2p_connect_group(addr0
, pin
, timeout
=60)
121 if not c_res
['persistent']:
122 raise Exception("Joining client did not recognize persistent group")
123 if r_res
['psk'] == c_res
['psk']:
124 raise Exception("Same PSK assigned for both clients")
125 hwsim_utils
.test_connectivity_p2p_sta(dev
[1], dev
[2])
127 logger
.info("Leave persistent group and rejoin it")
128 dev
[2].remove_group()
129 ev
= dev
[2].wait_event(["P2P-GROUP-REMOVED"], timeout
=3)
131 raise Exception("Group removal event timed out")
132 if not dev
[2].discover_peer(addr0
, social
=True):
133 raise Exception("Peer " + peer
+ " not found")
134 dev
[2].dump_monitor()
135 peer
= dev
[2].get_peer(addr0
)
136 dev
[2].global_request("P2P_GROUP_ADD persistent=" + peer
['persistent'])
137 ev
= dev
[2].wait_global_event(["P2P-GROUP-STARTED"], timeout
=30)
139 raise Exception("Timeout on group restart (on client)")
140 cli_res
= dev
[2].group_form_result(ev
)
141 if not cli_res
['persistent']:
142 raise Exception("Persistent group not restarted as persistent (cli)")
143 hwsim_utils
.test_connectivity_p2p(dev
[1], dev
[2])
145 logger
.info("Remove one of the clients from the group")
146 dev
[0].global_request("P2P_REMOVE_CLIENT " + addr2
)
147 dev
[2].wait_go_ending_session()
149 logger
.info("Try to reconnect after having been removed from group")
150 if not dev
[2].discover_peer(addr0
, social
=True):
151 raise Exception("Peer " + peer
+ " not found")
152 dev
[2].dump_monitor()
153 peer
= dev
[2].get_peer(addr0
)
154 dev
[2].global_request("P2P_GROUP_ADD persistent=" + peer
['persistent'])
155 ev
= dev
[2].wait_global_event(["P2P-GROUP-STARTED","WPA: 4-Way Handshake failed"], timeout
=30)
157 raise Exception("Timeout on group restart (on client)")
158 if "P2P-GROUP-STARTED" in ev
:
159 raise Exception("Client managed to connect after being removed")
161 logger
.info("Remove the remaining client from the group")
162 dev
[0].global_request("P2P_REMOVE_CLIENT " + addr1
)
163 dev
[1].wait_go_ending_session()
165 logger
.info("Terminate persistent group")
166 dev
[0].remove_group()
167 dev
[0].dump_monitor()
169 logger
.info("Try to re-invoke persistent group from client")
170 dev
[0].request("SET persistent_reconnect 1")
172 if not dev
[1].discover_peer(addr0
, social
=True):
173 raise Exception("Peer " + peer
+ " not found")
174 dev
[1].dump_monitor()
175 peer
= dev
[1].get_peer(addr0
)
176 dev
[1].global_request("P2P_INVITE persistent=" + peer
['persistent'] + " peer=" + addr0
)
177 ev
= dev
[1].wait_global_event(["P2P-GROUP-STARTED","WPA: 4-Way Handshake failed"], timeout
=30)
179 raise Exception("Timeout on group restart (on client)")
180 if "P2P-GROUP-STARTED" in ev
:
181 raise Exception("Client managed to re-invoke after being removed")
182 dev
[0].dump_monitor()
184 logger
.info("Terminate persistent group")
185 dev
[0].remove_group()
186 dev
[0].dump_monitor()
188 def test_persistent_group_invite_removed_client(dev
):
189 """P2P persistent group client removal and re-invitation"""
190 addr0
= dev
[0].p2p_dev_addr()
191 addr1
= dev
[1].p2p_dev_addr()
192 dev
[0].request("P2P_SET per_sta_psk 1")
193 logger
.info("Form a persistent group")
194 [i_res
, r_res
] = go_neg_pin_authorized_persistent(i_dev
=dev
[0], i_intent
=15,
195 r_dev
=dev
[1], r_intent
=0)
196 if not i_res
['persistent'] or not r_res
['persistent']:
197 raise Exception("Formed group was not persistent")
199 logger
.info("Remove client from the group")
200 dev
[0].global_request("P2P_REMOVE_CLIENT " + addr1
)
201 dev
[1].wait_go_ending_session()
203 logger
.info("Re-invite the removed client to join the group")
205 if not dev
[0].discover_peer(addr1
, social
=True):
206 raise Exception("Peer " + peer
+ " not found")
207 dev
[0].global_request("P2P_INVITE group=" + dev
[0].group_ifname
+ " peer=" + addr1
)
208 ev
= dev
[1].wait_global_event(["P2P-INVITATION-RECEIVED"], timeout
=10)
210 raise Exception("Timeout on invitation")
211 if "sa=" + addr0
+ " persistent=" not in ev
:
212 raise Exception("Unexpected invitation event")
213 [event
,addr
,persistent
] = ev
.split(' ', 2)
214 dev
[1].global_request("P2P_GROUP_ADD " + persistent
)
215 ev
= dev
[1].wait_global_event(["P2P-PERSISTENT-PSK-FAIL"], timeout
=30)
217 raise Exception("Did not receive PSK failure report")
218 [tmp
,id] = ev
.split('=', 1)
219 ev
= dev
[1].wait_global_event(["P2P-GROUP-REMOVED"], timeout
=10)
221 raise Exception("Group removal event timed out")
222 if "reason=PSK_FAILURE" not in ev
:
223 raise Exception("Unexpected group removal reason")
224 dev
[1].request("REMOVE_NETWORK " + id)
226 logger
.info("Re-invite after client removed persistent group info")
228 if not dev
[0].discover_peer(addr1
, social
=True):
229 raise Exception("Peer " + peer
+ " not found")
230 dev
[0].global_request("P2P_INVITE group=" + dev
[0].group_ifname
+ " peer=" + addr1
)
231 ev
= dev
[1].wait_global_event(["P2P-INVITATION-RECEIVED"], timeout
=10)
233 raise Exception("Timeout on invitation")
234 if " persistent=" in ev
:
235 raise Exception("Unexpected invitation event")
236 pin
= dev
[1].wps_read_pin()
237 dev
[0].p2p_go_authorize_client(pin
)
238 c_res
= dev
[1].p2p_connect_group(addr0
, pin
, timeout
=60)
239 if not c_res
['persistent']:
240 raise Exception("Joining client did not recognize persistent group")
241 if r_res
['psk'] == c_res
['psk']:
242 raise Exception("Same PSK assigned on both times")
243 hwsim_utils
.test_connectivity_p2p(dev
[0], dev
[1])
245 terminate_group(dev
[0], dev
[1])
247 def test_persistent_group_channel(dev
):
248 """P2P persistent group re-invocation with channel selection"""
249 form(dev
[0], dev
[1], test_data
=False)
251 logger
.info("Re-invoke persistent group from client with forced channel")
252 invite(dev
[1], dev
[0], "freq=2427")
253 [go_res
, cli_res
] = check_result(dev
[0], dev
[1])
254 if go_res
['freq'] != "2427":
255 raise Exception("Persistent group client forced channel not followed")
256 terminate_group(dev
[0], dev
[1])
258 logger
.info("Re-invoke persistent group from GO with forced channel")
259 invite(dev
[0], dev
[1], "freq=2432")
260 [go_res
, cli_res
] = check_result(dev
[0], dev
[1])
261 if go_res
['freq'] != "2432":
262 raise Exception("Persistent group GO channel preference not followed")
263 terminate_group(dev
[0], dev
[1])
265 logger
.info("Re-invoke persistent group from client with channel preference")
266 invite(dev
[1], dev
[0], "pref=2417")
267 [go_res
, cli_res
] = check_result(dev
[0], dev
[1])
268 if go_res
['freq'] != "2417":
269 raise Exception("Persistent group client channel preference not followed")
270 terminate_group(dev
[0], dev
[1])
272 def test_persistent_group_and_role_change(dev
):
273 """P2P persistent group, auto GO in another role, and re-invocation"""
276 logger
.info("Start and stop autonomous GO on previous P2P client device")
277 dev
[1].p2p_start_go()
278 dev
[1].remove_group()
279 dev
[1].dump_monitor()
281 logger
.info("Re-invoke the persistent group")
282 invite_from_go(dev
[0], dev
[1])
284 def test_persistent_go_client_list(dev
):
285 """P2P GO and list of clients in persistent group"""
286 addr0
= dev
[0].p2p_dev_addr()
287 addr1
= dev
[1].p2p_dev_addr()
288 addr2
= dev
[2].p2p_dev_addr()
290 res
= dev
[0].p2p_start_go(persistent
=True)
292 for n
in dev
[0].list_networks():
293 if "[P2P-PERSISTENT]" in n
['flags']:
297 raise Exception("Could not find persistent group entry")
299 connect_cli(dev
[0], dev
[1])
300 clients
= dev
[0].request("GET_NETWORK " + id + " p2p_client_list").rstrip()
302 raise Exception("Unexpected p2p_client_list entry(2): " + clients
)
303 connect_cli(dev
[0], dev
[2])
304 clients
= dev
[0].request("GET_NETWORK " + id + " p2p_client_list").rstrip()
305 if clients
!= addr2
+ " " + addr1
:
306 raise Exception("Unexpected p2p_client_list entry(3): " + clients
)
308 peer
= dev
[1].get_peer(res
['go_dev_addr'])
309 dev
[1].remove_group()
310 dev
[1].request("P2P_GROUP_ADD persistent=" + peer
['persistent'])
311 ev
= dev
[1].wait_global_event(["P2P-GROUP-STARTED"], timeout
=30)
313 raise Exception("Timeout on group restart (on client)")
314 clients
= dev
[0].request("GET_NETWORK " + id + " p2p_client_list").rstrip()
315 if clients
!= addr1
+ " " + addr2
:
316 raise Exception("Unexpected p2p_client_list entry(4): " + clients
)
318 dev
[2].remove_group()
319 dev
[1].remove_group()
320 dev
[0].remove_group()
322 clients
= dev
[0].request("GET_NETWORK " + id + " p2p_client_list").rstrip()
323 if clients
!= addr1
+ " " + addr2
:
324 raise Exception("Unexpected p2p_client_list entry(5): " + clients
)
328 dev
[0].request("P2P_FLUSH")
329 dev
[0].discover_peer(addr1
, social
=True)
330 peer
= dev
[0].get_peer(addr1
)
331 if 'persistent' not in peer
or peer
['persistent'] != id:
332 raise Exception("Persistent group client not recognized(1)")
334 dev
[0].discover_peer(addr2
, social
=True)
335 peer
= dev
[0].get_peer(addr2
)
336 if 'persistent' not in peer
or peer
['persistent'] != id:
337 raise Exception("Persistent group client not recognized(2)")