xref: /aosp_15_r20/external/wpa_supplicant_8/src/p2p/p2p_invitation.c (revision 03f9172ca588f91df233974f4258bab95191f931)
1 /*
2  * Wi-Fi Direct - P2P Invitation procedure
3  * Copyright (c) 2010, Atheros Communications
4  *
5  * This software may be distributed under the terms of the BSD license.
6  * See README for more details.
7  */
8 
9 #include "includes.h"
10 
11 #include "common.h"
12 #include "common/ieee802_11_defs.h"
13 #include "common/wpa_ctrl.h"
14 #include "p2p_i.h"
15 #include "p2p.h"
16 
17 
p2p_build_invitation_req(struct p2p_data * p2p,struct p2p_device * peer,const u8 * go_dev_addr,int dev_pw_id)18 static struct wpabuf * p2p_build_invitation_req(struct p2p_data *p2p,
19 						struct p2p_device *peer,
20 						const u8 *go_dev_addr,
21 						int dev_pw_id)
22 {
23 	struct wpabuf *buf;
24 	u8 *len;
25 	const u8 *dev_addr;
26 	size_t extra = 0;
27 	bool is_6ghz_capab;
28 
29 #ifdef CONFIG_WIFI_DISPLAY
30 	struct wpabuf *wfd_ie = p2p->wfd_ie_invitation;
31 	if (wfd_ie && p2p->inv_role == P2P_INVITE_ROLE_ACTIVE_GO) {
32 		size_t i;
33 		for (i = 0; i < p2p->num_groups; i++) {
34 			struct p2p_group *g = p2p->groups[i];
35 			struct wpabuf *ie;
36 			if (!ether_addr_equal(p2p_group_get_interface_addr(g),
37 					      p2p->inv_bssid))
38 				continue;
39 			ie = p2p_group_get_wfd_ie(g);
40 			if (ie) {
41 				wfd_ie = ie;
42 				break;
43 			}
44 		}
45 	}
46 	if (wfd_ie)
47 		extra = wpabuf_len(wfd_ie);
48 #endif /* CONFIG_WIFI_DISPLAY */
49 
50 	if (p2p->vendor_elem && p2p->vendor_elem[VENDOR_ELEM_P2P_INV_REQ])
51 		extra += wpabuf_len(p2p->vendor_elem[VENDOR_ELEM_P2P_INV_REQ]);
52 
53 	buf = wpabuf_alloc(1000 + extra);
54 	if (buf == NULL)
55 		return NULL;
56 
57 	peer->dialog_token++;
58 	if (peer->dialog_token == 0)
59 		peer->dialog_token = 1;
60 	p2p_buf_add_public_action_hdr(buf, P2P_INVITATION_REQ,
61 				      peer->dialog_token);
62 
63 	len = p2p_buf_add_ie_hdr(buf);
64 	if (p2p->inv_role == P2P_INVITE_ROLE_ACTIVE_GO || !p2p->inv_persistent)
65 		p2p_buf_add_config_timeout(buf, 0, 0);
66 	else
67 		p2p_buf_add_config_timeout(buf, p2p->go_timeout,
68 					   p2p->client_timeout);
69 	p2p_buf_add_invitation_flags(buf, p2p->inv_persistent ?
70 				     P2P_INVITATION_FLAGS_TYPE : 0);
71 	if (p2p->inv_role != P2P_INVITE_ROLE_CLIENT ||
72 	    !(peer->flags & P2P_DEV_NO_PREF_CHAN))
73 		p2p_buf_add_operating_channel(buf, p2p->cfg->country,
74 					      p2p->op_reg_class,
75 					      p2p->op_channel);
76 	if (p2p->inv_bssid_set)
77 		p2p_buf_add_group_bssid(buf, p2p->inv_bssid);
78 	is_6ghz_capab = is_p2p_6ghz_capable(p2p) &&
79 		p2p_is_peer_6ghz_capab(p2p, peer->info.p2p_device_addr);
80 	p2p_buf_add_channel_list(buf, p2p->cfg->country, &p2p->channels,
81 				 is_6ghz_capab);
82 	if (go_dev_addr)
83 		dev_addr = go_dev_addr;
84 	else if (p2p->inv_role == P2P_INVITE_ROLE_CLIENT)
85 		dev_addr = peer->info.p2p_device_addr;
86 	else
87 		dev_addr = p2p->cfg->dev_addr;
88 	p2p_buf_add_group_id(buf, dev_addr, p2p->inv_ssid, p2p->inv_ssid_len);
89 	p2p_buf_add_device_info(buf, p2p, peer);
90 	p2p_buf_update_ie_hdr(buf, len);
91 
92 	p2p_buf_add_pref_channel_list(buf, p2p->pref_freq_list,
93 				      p2p->num_pref_freq);
94 
95 #ifdef CONFIG_WIFI_DISPLAY
96 	if (wfd_ie)
97 		wpabuf_put_buf(buf, wfd_ie);
98 #endif /* CONFIG_WIFI_DISPLAY */
99 
100 	if (p2p->vendor_elem && p2p->vendor_elem[VENDOR_ELEM_P2P_INV_REQ])
101 		wpabuf_put_buf(buf, p2p->vendor_elem[VENDOR_ELEM_P2P_INV_REQ]);
102 
103 	if (dev_pw_id >= 0) {
104 		/* WSC IE in Invitation Request for NFC static handover */
105 		p2p_build_wps_ie(p2p, buf, dev_pw_id, 0);
106 	}
107 
108 	return buf;
109 }
110 
111 
p2p_build_invitation_resp(struct p2p_data * p2p,struct p2p_device * peer,u8 dialog_token,u8 status,const u8 * group_bssid,u8 reg_class,u8 channel,struct p2p_channels * channels)112 static struct wpabuf * p2p_build_invitation_resp(struct p2p_data *p2p,
113 						 struct p2p_device *peer,
114 						 u8 dialog_token, u8 status,
115 						 const u8 *group_bssid,
116 						 u8 reg_class, u8 channel,
117 						 struct p2p_channels *channels)
118 {
119 	struct wpabuf *buf;
120 	u8 *len;
121 	size_t extra = 0;
122 
123 #ifdef CONFIG_WIFI_DISPLAY
124 	struct wpabuf *wfd_ie = p2p->wfd_ie_invitation;
125 	if (wfd_ie && group_bssid) {
126 		size_t i;
127 		for (i = 0; i < p2p->num_groups; i++) {
128 			struct p2p_group *g = p2p->groups[i];
129 			struct wpabuf *ie;
130 			if (!ether_addr_equal(p2p_group_get_interface_addr(g),
131 					      group_bssid))
132 				continue;
133 			ie = p2p_group_get_wfd_ie(g);
134 			if (ie) {
135 				wfd_ie = ie;
136 				break;
137 			}
138 		}
139 	}
140 	if (wfd_ie)
141 		extra = wpabuf_len(wfd_ie);
142 #endif /* CONFIG_WIFI_DISPLAY */
143 
144 	if (p2p->vendor_elem && p2p->vendor_elem[VENDOR_ELEM_P2P_INV_RESP])
145 		extra += wpabuf_len(p2p->vendor_elem[VENDOR_ELEM_P2P_INV_RESP]);
146 
147 	buf = wpabuf_alloc(1000 + extra);
148 	if (buf == NULL)
149 		return NULL;
150 
151 	p2p_buf_add_public_action_hdr(buf, P2P_INVITATION_RESP,
152 				      dialog_token);
153 
154 	len = p2p_buf_add_ie_hdr(buf);
155 	p2p_buf_add_status(buf, status);
156 	p2p_buf_add_config_timeout(buf, 0, 0); /* FIX */
157 	if (reg_class && channel)
158 		p2p_buf_add_operating_channel(buf, p2p->cfg->country,
159 					      reg_class, channel);
160 	if (group_bssid)
161 		p2p_buf_add_group_bssid(buf, group_bssid);
162 	if (channels) {
163 		bool is_6ghz_capab;
164 
165 		is_6ghz_capab = is_p2p_6ghz_capable(p2p) &&
166 			p2p_is_peer_6ghz_capab(p2p, peer->info.p2p_device_addr);
167 		p2p_buf_add_channel_list(buf, p2p->cfg->country, channels,
168 					 is_6ghz_capab);
169 	}
170 	p2p_buf_update_ie_hdr(buf, len);
171 
172 #ifdef CONFIG_WIFI_DISPLAY
173 	if (wfd_ie)
174 		wpabuf_put_buf(buf, wfd_ie);
175 #endif /* CONFIG_WIFI_DISPLAY */
176 
177 	if (p2p->vendor_elem && p2p->vendor_elem[VENDOR_ELEM_P2P_INV_RESP])
178 		wpabuf_put_buf(buf, p2p->vendor_elem[VENDOR_ELEM_P2P_INV_RESP]);
179 
180 	return buf;
181 }
182 
183 
p2p_process_invitation_req(struct p2p_data * p2p,const u8 * sa,const u8 * data,size_t len,int rx_freq)184 struct wpabuf * p2p_process_invitation_req(struct p2p_data *p2p, const u8 *sa,
185 					   const u8 *data, size_t len,
186 					   int rx_freq)
187 {
188 	struct p2p_device *dev;
189 	struct p2p_message msg;
190 	struct wpabuf *resp = NULL;
191 	u8 status = P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE;
192 	int go = 0;
193 	u8 group_bssid[ETH_ALEN], *bssid;
194 	int op_freq = 0;
195 	u8 reg_class = 0, channel = 0;
196 	struct p2p_channels all_channels, intersection, *channels = NULL;
197 	int persistent;
198 
199 	os_memset(group_bssid, 0, sizeof(group_bssid));
200 
201 	p2p_dbg(p2p, "Received Invitation Request from " MACSTR " (freq=%d)",
202 		MAC2STR(sa), rx_freq);
203 
204 	if (p2p_parse(data, len, &msg))
205 		return NULL;
206 
207 	dev = p2p_get_device(p2p, sa);
208 	if (dev == NULL || (dev->flags & P2P_DEV_PROBE_REQ_ONLY)) {
209 		p2p_dbg(p2p, "Invitation Request from unknown peer " MACSTR,
210 			MAC2STR(sa));
211 
212 		if (p2p_add_device(p2p, sa, rx_freq, NULL, 0, data + 1, len - 1,
213 				   0)) {
214 			p2p_dbg(p2p, "Invitation Request add device failed "
215 				MACSTR, MAC2STR(sa));
216 			status = P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE;
217 			goto fail;
218 		}
219 
220 		dev = p2p_get_device(p2p, sa);
221 		if (dev == NULL) {
222 			p2p_dbg(p2p, "Reject Invitation Request from unknown peer "
223 				MACSTR, MAC2STR(sa));
224 			status = P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE;
225 			goto fail;
226 		}
227 	}
228 
229 	if (!msg.group_id || !msg.channel_list) {
230 		p2p_dbg(p2p, "Mandatory attribute missing in Invitation Request from "
231 			MACSTR, MAC2STR(sa));
232 		status = P2P_SC_FAIL_INVALID_PARAMS;
233 		goto fail;
234 	}
235 
236 	if (msg.invitation_flags)
237 		persistent = *msg.invitation_flags & P2P_INVITATION_FLAGS_TYPE;
238 	else {
239 		/* Invitation Flags is a mandatory attribute starting from P2P
240 		 * spec 1.06. As a backwards compatibility mechanism, assume
241 		 * the request was for a persistent group if the attribute is
242 		 * missing.
243 		 */
244 		p2p_dbg(p2p, "Mandatory Invitation Flags attribute missing from Invitation Request");
245 		persistent = 1;
246 	}
247 
248 	p2p_channels_union(&p2p->cfg->channels, &p2p->cfg->cli_channels,
249 			   &all_channels);
250 
251 	if (p2p_peer_channels_check(p2p, &all_channels, dev,
252 				    msg.channel_list, msg.channel_list_len) <
253 	    0) {
254 		p2p_dbg(p2p, "No common channels found");
255 		status = P2P_SC_FAIL_NO_COMMON_CHANNELS;
256 		goto fail;
257 	}
258 
259 	p2p_channels_dump(p2p, "own channels", &p2p->cfg->channels);
260 	p2p_channels_dump(p2p, "own client channels", &all_channels);
261 	p2p_channels_dump(p2p, "peer channels", &dev->channels);
262 	p2p_channels_intersect(&all_channels, &dev->channels,
263 			       &intersection);
264 	p2p_channels_dump(p2p, "intersection", &intersection);
265 
266 	if (p2p->cfg->invitation_process) {
267 		status = p2p->cfg->invitation_process(
268 			p2p->cfg->cb_ctx, sa, msg.group_bssid, msg.group_id,
269 			msg.group_id + ETH_ALEN, msg.group_id_len - ETH_ALEN,
270 			&go, group_bssid, &op_freq, persistent, &intersection,
271 			msg.dev_password_id_present ? msg.dev_password_id : -1);
272 	}
273 
274 	if (go) {
275 		p2p_channels_intersect(&p2p->cfg->channels, &dev->channels,
276 				       &intersection);
277 		p2p_channels_dump(p2p, "intersection(GO)", &intersection);
278 		if (intersection.reg_classes == 0) {
279 			p2p_dbg(p2p, "No common channels found (GO)");
280 			status = P2P_SC_FAIL_NO_COMMON_CHANNELS;
281 			goto fail;
282 		}
283 	}
284 
285 	if (op_freq) {
286 		p2p_dbg(p2p, "Invitation processing forced frequency %d MHz",
287 			op_freq);
288 		if (p2p_freq_to_channel(op_freq, &reg_class, &channel) < 0) {
289 			p2p_dbg(p2p, "Unknown forced freq %d MHz from invitation_process()",
290 				op_freq);
291 			status = P2P_SC_FAIL_NO_COMMON_CHANNELS;
292 			goto fail;
293 		}
294 
295 		if (!p2p_channels_includes(&intersection, reg_class, channel))
296 		{
297 			p2p_dbg(p2p, "forced freq %d MHz not in the supported channels intersection",
298 				op_freq);
299 			status = P2P_SC_FAIL_NO_COMMON_CHANNELS;
300 			goto fail;
301 		}
302 
303 		if (status == P2P_SC_SUCCESS)
304 			channels = &intersection;
305 	} else {
306 		p2p_dbg(p2p, "No forced channel from invitation processing - figure out best one to use");
307 
308 		/* Default to own configuration as a starting point */
309 		p2p->op_reg_class = p2p->cfg->op_reg_class;
310 		p2p->op_channel = p2p->cfg->op_channel;
311 		p2p_dbg(p2p, "Own default op_class %d channel %d",
312 			p2p->op_reg_class, p2p->op_channel);
313 
314 		/* Use peer preference if specified and compatible */
315 		if (msg.operating_channel) {
316 			int req_freq;
317 			req_freq = p2p_channel_to_freq(
318 				msg.operating_channel[3],
319 				msg.operating_channel[4]);
320 			p2p_dbg(p2p, "Peer operating channel preference: %d MHz",
321 				req_freq);
322 			if (req_freq > 0 &&
323 			    p2p_channels_includes(&intersection,
324 						  msg.operating_channel[3],
325 						  msg.operating_channel[4])) {
326 				p2p->op_reg_class = msg.operating_channel[3];
327 				p2p->op_channel = msg.operating_channel[4];
328 				p2p_dbg(p2p, "Use peer preference op_class %d channel %d",
329 					p2p->op_reg_class, p2p->op_channel);
330 			} else {
331 				p2p_dbg(p2p, "Cannot use peer channel preference");
332 			}
333 		}
334 
335 		/* Reselect the channel only for the case of the GO */
336 		if (go &&
337 		    !p2p_channels_includes(&intersection, p2p->op_reg_class,
338 					   p2p->op_channel)) {
339 			p2p_dbg(p2p, "Initially selected channel (op_class %d channel %d) not in channel intersection - try to reselect",
340 				p2p->op_reg_class, p2p->op_channel);
341 			p2p_reselect_channel(p2p, &intersection);
342 			p2p_dbg(p2p, "Re-selection result: op_class %d channel %d",
343 				p2p->op_reg_class, p2p->op_channel);
344 			if (!p2p_channels_includes(&intersection,
345 						   p2p->op_reg_class,
346 						   p2p->op_channel)) {
347 				p2p_dbg(p2p, "Peer does not support selected operating channel (reg_class=%u channel=%u)",
348 					p2p->op_reg_class, p2p->op_channel);
349 				status = P2P_SC_FAIL_NO_COMMON_CHANNELS;
350 				goto fail;
351 			}
352 		} else if (go && !(dev->flags & P2P_DEV_FORCE_FREQ) &&
353 			   !p2p->cfg->cfg_op_channel) {
354 			p2p_dbg(p2p, "Try to reselect channel selection with peer information received; previously selected op_class %u channel %u",
355 				p2p->op_reg_class, p2p->op_channel);
356 			p2p_reselect_channel(p2p, &intersection);
357 		}
358 
359 		/*
360 		 * Use the driver preferred frequency list extension if
361 		 * supported.
362 		 */
363 		p2p_check_pref_chan(p2p, go, dev, &msg);
364 
365 		op_freq = p2p_channel_to_freq(p2p->op_reg_class,
366 					      p2p->op_channel);
367 		if (op_freq < 0) {
368 			p2p_dbg(p2p, "Unknown operational channel (country=%c%c reg_class=%u channel=%u)",
369 				p2p->cfg->country[0], p2p->cfg->country[1],
370 				p2p->op_reg_class, p2p->op_channel);
371 			status = P2P_SC_FAIL_NO_COMMON_CHANNELS;
372 			goto fail;
373 		}
374 		p2p_dbg(p2p, "Selected operating channel - %d MHz", op_freq);
375 
376 		if (status == P2P_SC_SUCCESS) {
377 			reg_class = p2p->op_reg_class;
378 			channel = p2p->op_channel;
379 			channels = &intersection;
380 		}
381 	}
382 
383 fail:
384 	if (go && status == P2P_SC_SUCCESS && !is_zero_ether_addr(group_bssid))
385 		bssid = group_bssid;
386 	else
387 		bssid = NULL;
388 	resp = p2p_build_invitation_resp(p2p, dev, msg.dialog_token, status,
389 					 bssid, reg_class, channel, channels);
390 
391 	/*
392 	 * Store copy of invitation data to be used when processing TX status
393 	 * callback for the Acton frame.
394 	 */
395 	os_memcpy(p2p->inv_sa, sa, ETH_ALEN);
396 	if (msg.group_bssid) {
397 		os_memcpy(p2p->inv_group_bssid, msg.group_bssid, ETH_ALEN);
398 		p2p->inv_group_bssid_ptr = p2p->inv_group_bssid;
399 	} else
400 		p2p->inv_group_bssid_ptr = NULL;
401 	if (msg.group_id) {
402 		if (msg.group_id_len - ETH_ALEN <= SSID_MAX_LEN) {
403 			os_memcpy(p2p->inv_ssid, msg.group_id + ETH_ALEN,
404 				  msg.group_id_len - ETH_ALEN);
405 			p2p->inv_ssid_len = msg.group_id_len - ETH_ALEN;
406 		}
407 		os_memcpy(p2p->inv_go_dev_addr, msg.group_id, ETH_ALEN);
408 	} else {
409 		p2p->inv_ssid_len = 0;
410 		os_memset(p2p->inv_go_dev_addr, 0, ETH_ALEN);
411 	}
412 	p2p->inv_status = status;
413 	p2p->inv_op_freq = op_freq;
414 	p2p_parse_free(&msg);
415 	return resp;
416 }
417 
418 
p2p_handle_invitation_req(struct p2p_data * p2p,const u8 * sa,const u8 * data,size_t len,int rx_freq)419 void p2p_handle_invitation_req(struct p2p_data *p2p, const u8 *sa,
420 			       const u8 *data, size_t len, int rx_freq)
421 {
422 	int freq;
423 	struct wpabuf *resp;
424 
425 	resp = p2p_process_invitation_req(p2p, sa, data, len, rx_freq);
426 	if (!resp)
427 		return;
428 
429 	if (rx_freq > 0)
430 		freq = rx_freq;
431 	else
432 		freq = p2p_channel_to_freq(p2p->cfg->reg_class,
433 					   p2p->cfg->channel);
434 	if (freq < 0)
435 		p2p_dbg(p2p, "Unknown regulatory class/channel");
436 
437 	p2p->pending_action_state = P2P_PENDING_INVITATION_RESPONSE;
438 	if (p2p_send_action(p2p, freq, sa, p2p->cfg->dev_addr,
439 			    p2p->cfg->dev_addr,
440 			    wpabuf_head(resp), wpabuf_len(resp), 50) < 0) {
441 		p2p_dbg(p2p, "Failed to send Action frame");
442 	}
443 
444 	wpabuf_free(resp);
445 }
446 
447 
p2p_process_invitation_resp(struct p2p_data * p2p,const u8 * sa,const u8 * data,size_t len)448 void p2p_process_invitation_resp(struct p2p_data *p2p, const u8 *sa,
449 				 const u8 *data, size_t len)
450 {
451 	struct p2p_device *dev;
452 	struct p2p_message msg;
453 	struct p2p_channels intersection, *channels = NULL;
454 
455 	p2p_dbg(p2p, "Received Invitation Response from " MACSTR,
456 		MAC2STR(sa));
457 
458 	dev = p2p_get_device(p2p, sa);
459 	if (dev == NULL) {
460 		p2p_dbg(p2p, "Ignore Invitation Response from unknown peer "
461 			MACSTR, MAC2STR(sa));
462 		p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
463 		return;
464 	}
465 
466 	if (dev != p2p->invite_peer) {
467 		p2p_dbg(p2p, "Ignore unexpected Invitation Response from peer "
468 			MACSTR, MAC2STR(sa));
469 		p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
470 		return;
471 	}
472 
473 	if (p2p_parse(data, len, &msg)) {
474 		p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
475 		return;
476 	}
477 
478 	if (!msg.status) {
479 		p2p_dbg(p2p, "Mandatory Status attribute missing in Invitation Response from "
480 			MACSTR, MAC2STR(sa));
481 		p2p_parse_free(&msg);
482 		p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
483 		return;
484 	}
485 
486 	/*
487 	 * We should not really receive a replayed response twice since
488 	 * duplicate frames are supposed to be dropped. However, not all drivers
489 	 * do that for pre-association frames. We did not use to verify dialog
490 	 * token matches for invitation response frames, but that check can be
491 	 * safely used to drop a replayed response to the previous Invitation
492 	 * Request in case the suggested operating channel was changed. This
493 	 * allows a duplicated reject frame to be dropped with the assumption
494 	 * that the real response follows after it.
495 	 */
496 	if (*msg.status == P2P_SC_FAIL_NO_COMMON_CHANNELS &&
497 	    p2p->retry_invite_req_sent &&
498 	    msg.dialog_token != dev->dialog_token) {
499 		p2p_dbg(p2p, "Unexpected Dialog Token %u (expected %u)",
500 			msg.dialog_token, dev->dialog_token);
501 		p2p_parse_free(&msg);
502 		return;
503 	}
504 
505 	if (*msg.status == P2P_SC_FAIL_NO_COMMON_CHANNELS &&
506 	    p2p->retry_invite_req &&
507 	    p2p_channel_random_social(&p2p->cfg->channels, &p2p->op_reg_class,
508 				      &p2p->op_channel, NULL, NULL) == 0) {
509 		p2p->retry_invite_req = 0;
510 		p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
511 		p2p->cfg->stop_listen(p2p->cfg->cb_ctx);
512 		p2p_set_state(p2p, P2P_INVITE);
513 		p2p_dbg(p2p, "Resend Invitation Request setting op_class %u channel %u as operating channel",
514 			p2p->op_reg_class, p2p->op_channel);
515 		p2p->retry_invite_req_sent = 1;
516 		p2p_invite_send(p2p, p2p->invite_peer, p2p->invite_go_dev_addr,
517 				p2p->invite_dev_pw_id);
518 		p2p_parse_free(&msg);
519 		return;
520 	}
521 	p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
522 	p2p->retry_invite_req = 0;
523 
524 	if (!msg.channel_list && *msg.status == P2P_SC_SUCCESS) {
525 		p2p_dbg(p2p, "Mandatory Channel List attribute missing in Invitation Response from "
526 			MACSTR, MAC2STR(sa));
527 #ifdef CONFIG_P2P_STRICT
528 		p2p_parse_free(&msg);
529 		return;
530 #endif /* CONFIG_P2P_STRICT */
531 		/* Try to survive without peer channel list */
532 		channels = &p2p->channels;
533 	} else if (!msg.channel_list) {
534 		/* Non-success cases are not required to include Channel List */
535 		channels = &p2p->channels;
536 	} else if (p2p_peer_channels_check(p2p, &p2p->channels, dev,
537 					   msg.channel_list,
538 					   msg.channel_list_len) < 0) {
539 		p2p_dbg(p2p, "No common channels found");
540 		p2p_parse_free(&msg);
541 		return;
542 	} else {
543 		p2p_channels_intersect(&p2p->channels, &dev->channels,
544 				       &intersection);
545 		channels = &intersection;
546 	}
547 
548 	if (p2p->cfg->invitation_result) {
549 		int peer_oper_freq = 0;
550 		int freq = p2p_channel_to_freq(p2p->op_reg_class,
551 					       p2p->op_channel);
552 		if (freq < 0)
553 			freq = 0;
554 
555 		if (msg.operating_channel) {
556 			peer_oper_freq = p2p_channel_to_freq(
557 				msg.operating_channel[3],
558 				msg.operating_channel[4]);
559 			if (peer_oper_freq < 0)
560 				peer_oper_freq = 0;
561 		}
562 
563 		/*
564 		 * Use the driver preferred frequency list extension if
565 		 * supported.
566 		 */
567 		p2p_check_pref_chan(p2p, 0, dev, &msg);
568 
569 		p2p->cfg->invitation_result(p2p->cfg->cb_ctx, *msg.status,
570 					    msg.group_bssid, channels, sa,
571 					    freq, peer_oper_freq);
572 	}
573 
574 	p2p_parse_free(&msg);
575 
576 	p2p_clear_timeout(p2p);
577 	p2p_set_state(p2p, P2P_IDLE);
578 	p2p->invite_peer = NULL;
579 }
580 
581 
p2p_invite_send(struct p2p_data * p2p,struct p2p_device * dev,const u8 * go_dev_addr,int dev_pw_id)582 int p2p_invite_send(struct p2p_data *p2p, struct p2p_device *dev,
583 		    const u8 *go_dev_addr, int dev_pw_id)
584 {
585 	struct wpabuf *req;
586 	int freq;
587 
588 	freq = dev->listen_freq > 0 ? dev->listen_freq : dev->oper_freq;
589 	if (freq <= 0)
590 		freq = dev->oob_go_neg_freq;
591 	if (freq <= 0) {
592 		p2p_dbg(p2p, "No Listen/Operating frequency known for the peer "
593 			MACSTR " to send Invitation Request",
594 			MAC2STR(dev->info.p2p_device_addr));
595 		return -1;
596 	}
597 
598 	req = p2p_build_invitation_req(p2p, dev, go_dev_addr, dev_pw_id);
599 	if (req == NULL)
600 		return -1;
601 	if (p2p->state != P2P_IDLE)
602 		p2p_stop_listen_for_freq(p2p, freq);
603 	p2p_dbg(p2p, "Sending Invitation Request");
604 	p2p_set_state(p2p, P2P_INVITE);
605 	p2p->pending_action_state = P2P_PENDING_INVITATION_REQUEST;
606 	p2p->invite_peer = dev;
607 	dev->invitation_reqs++;
608 
609 	/* In case of an active P2P GO use a shorter wait time to avoid
610 	 * issues if not sending out multiple consecutive Beacon frames. */
611 	if (p2p_send_action(p2p, freq, dev->info.p2p_device_addr,
612 			    p2p->cfg->dev_addr, dev->info.p2p_device_addr,
613 			    wpabuf_head(req), wpabuf_len(req),
614 			    p2p->inv_role == P2P_INVITE_ROLE_ACTIVE_GO ?
615 			    150 : 500) < 0) {
616 		p2p_dbg(p2p, "Failed to send Action frame");
617 		/* Use P2P find to recover and retry */
618 		p2p_set_timeout(p2p, 0, 0);
619 	} else {
620 		dev->flags |= P2P_DEV_WAIT_INV_REQ_ACK;
621 	}
622 
623 	wpabuf_free(req);
624 
625 	return 0;
626 }
627 
628 
p2p_invitation_req_cb(struct p2p_data * p2p,int success)629 void p2p_invitation_req_cb(struct p2p_data *p2p, int success)
630 {
631 	p2p_dbg(p2p, "Invitation Request TX callback: success=%d", success);
632 
633 	if (p2p->invite_peer == NULL) {
634 		p2p_dbg(p2p, "No pending Invite");
635 		return;
636 	}
637 
638 	if (success)
639 		p2p->invite_peer->flags &= ~P2P_DEV_WAIT_INV_REQ_ACK;
640 
641 	/*
642 	 * Use P2P find, if needed, to find the other device from its listen
643 	 * channel.
644 	 */
645 	p2p_set_state(p2p, P2P_INVITE);
646 	p2p_set_timeout(p2p, 0, success ? 500000 : 100000);
647 }
648 
649 
p2p_invitation_resp_cb(struct p2p_data * p2p,int success)650 void p2p_invitation_resp_cb(struct p2p_data *p2p, int success)
651 {
652 	p2p_dbg(p2p, "Invitation Response TX callback: success=%d", success);
653 	p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
654 
655 	if (!success)
656 		p2p_dbg(p2p, "Assume Invitation Response was actually received by the peer even though Ack was not reported");
657 
658 	if (p2p->cfg->invitation_received) {
659 		p2p->cfg->invitation_received(p2p->cfg->cb_ctx,
660 					      p2p->inv_sa,
661 					      p2p->inv_group_bssid_ptr,
662 					      p2p->inv_ssid, p2p->inv_ssid_len,
663 					      p2p->inv_go_dev_addr,
664 					      p2p->inv_status,
665 					      p2p->inv_op_freq);
666 	}
667 }
668 
669 
p2p_invite(struct p2p_data * p2p,const u8 * peer,enum p2p_invite_role role,const u8 * bssid,const u8 * ssid,size_t ssid_len,unsigned int force_freq,const u8 * go_dev_addr,int persistent_group,unsigned int pref_freq,int dev_pw_id)670 int p2p_invite(struct p2p_data *p2p, const u8 *peer, enum p2p_invite_role role,
671 	       const u8 *bssid, const u8 *ssid, size_t ssid_len,
672 	       unsigned int force_freq, const u8 *go_dev_addr,
673 	       int persistent_group, unsigned int pref_freq, int dev_pw_id)
674 {
675 	struct p2p_device *dev;
676 
677 	p2p_dbg(p2p, "Request to invite peer " MACSTR " role=%d persistent=%d "
678 		"force_freq=%u allow_6ghz=%d",
679 		MAC2STR(peer), role, persistent_group, force_freq,
680 		p2p->allow_6ghz);
681 	if (bssid)
682 		p2p_dbg(p2p, "Invitation for BSSID " MACSTR, MAC2STR(bssid));
683 	if (go_dev_addr) {
684 		p2p_dbg(p2p, "Invitation for GO Device Address " MACSTR,
685 			MAC2STR(go_dev_addr));
686 		os_memcpy(p2p->invite_go_dev_addr_buf, go_dev_addr, ETH_ALEN);
687 		p2p->invite_go_dev_addr = p2p->invite_go_dev_addr_buf;
688 	} else
689 		p2p->invite_go_dev_addr = NULL;
690 	wpa_hexdump_ascii(MSG_DEBUG, "Invitation for SSID",
691 			  ssid, ssid_len);
692 	if (dev_pw_id >= 0) {
693 		p2p_dbg(p2p, "Invitation to use Device Password ID %d",
694 			dev_pw_id);
695 	}
696 	p2p->invite_dev_pw_id = dev_pw_id;
697 	p2p->retry_invite_req = role == P2P_INVITE_ROLE_GO &&
698 		persistent_group && !force_freq;
699 	p2p->retry_invite_req_sent = 0;
700 
701 	dev = p2p_get_device(p2p, peer);
702 	if (dev == NULL || (dev->listen_freq <= 0 && dev->oper_freq <= 0 &&
703 			    dev->oob_go_neg_freq <= 0)) {
704 		p2p_dbg(p2p, "Cannot invite unknown P2P Device " MACSTR,
705 			MAC2STR(peer));
706 		return -1;
707 	}
708 
709 	if (p2p_prepare_channel(p2p, dev, force_freq, pref_freq,
710 				role != P2P_INVITE_ROLE_CLIENT) < 0)
711 		return -1;
712 
713 	if (persistent_group && role == P2P_INVITE_ROLE_CLIENT && !force_freq &&
714 	    !pref_freq)
715 		dev->flags |= P2P_DEV_NO_PREF_CHAN;
716 	else
717 		dev->flags &= ~P2P_DEV_NO_PREF_CHAN;
718 
719 	if (dev->flags & P2P_DEV_GROUP_CLIENT_ONLY) {
720 		if (!(dev->info.dev_capab &
721 		      P2P_DEV_CAPAB_CLIENT_DISCOVERABILITY)) {
722 			p2p_dbg(p2p, "Cannot invite a P2P Device " MACSTR
723 				" that is in a group and is not discoverable",
724 				MAC2STR(peer));
725 		}
726 		/* TODO: use device discoverability request through GO */
727 	}
728 
729 	dev->invitation_reqs = 0;
730 
731 	if (p2p->state != P2P_IDLE)
732 		p2p_stop_find(p2p);
733 
734 	p2p->inv_role = role;
735 	p2p->inv_bssid_set = bssid != NULL;
736 	if (bssid)
737 		os_memcpy(p2p->inv_bssid, bssid, ETH_ALEN);
738 	os_memcpy(p2p->inv_ssid, ssid, ssid_len);
739 	p2p->inv_ssid_len = ssid_len;
740 	p2p->inv_persistent = persistent_group;
741 	return p2p_invite_send(p2p, dev, go_dev_addr, dev_pw_id);
742 }
743