Line data Source code
1 : /*
2 : * Wi-Fi Direct - P2P group operations
3 : * Copyright (c) 2009-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/ieee802_11_common.h"
14 : #include "common/wpa_ctrl.h"
15 : #include "wps/wps_defs.h"
16 : #include "wps/wps_i.h"
17 : #include "p2p_i.h"
18 : #include "p2p.h"
19 :
20 :
21 : struct p2p_group_member {
22 : struct p2p_group_member *next;
23 : u8 addr[ETH_ALEN]; /* P2P Interface Address */
24 : u8 dev_addr[ETH_ALEN]; /* P2P Device Address */
25 : struct wpabuf *p2p_ie;
26 : struct wpabuf *wfd_ie;
27 : struct wpabuf *client_info;
28 : u8 dev_capab;
29 : };
30 :
31 : /**
32 : * struct p2p_group - Internal P2P module per-group data
33 : */
34 : struct p2p_group {
35 : struct p2p_data *p2p;
36 : struct p2p_group_config *cfg;
37 : struct p2p_group_member *members;
38 : unsigned int num_members;
39 : int group_formation;
40 : int beacon_update;
41 : struct wpabuf *noa;
42 : struct wpabuf *wfd_ie;
43 : };
44 :
45 :
46 338 : struct p2p_group * p2p_group_init(struct p2p_data *p2p,
47 : struct p2p_group_config *config)
48 : {
49 : struct p2p_group *group, **groups;
50 :
51 338 : group = os_zalloc(sizeof(*group));
52 338 : if (group == NULL)
53 0 : return NULL;
54 :
55 338 : groups = os_realloc_array(p2p->groups, p2p->num_groups + 1,
56 : sizeof(struct p2p_group *));
57 338 : if (groups == NULL) {
58 0 : os_free(group);
59 0 : return NULL;
60 : }
61 338 : groups[p2p->num_groups++] = group;
62 338 : p2p->groups = groups;
63 :
64 338 : group->p2p = p2p;
65 338 : group->cfg = config;
66 338 : group->group_formation = 1;
67 338 : group->beacon_update = 1;
68 338 : p2p_group_update_ies(group);
69 338 : group->cfg->idle_update(group->cfg->cb_ctx, 1);
70 :
71 338 : return group;
72 : }
73 :
74 :
75 435 : static void p2p_group_free_member(struct p2p_group_member *m)
76 : {
77 435 : wpabuf_free(m->wfd_ie);
78 435 : wpabuf_free(m->p2p_ie);
79 435 : wpabuf_free(m->client_info);
80 435 : os_free(m);
81 435 : }
82 :
83 :
84 338 : static void p2p_group_free_members(struct p2p_group *group)
85 : {
86 : struct p2p_group_member *m, *prev;
87 338 : m = group->members;
88 338 : group->members = NULL;
89 338 : group->num_members = 0;
90 830 : while (m) {
91 154 : prev = m;
92 154 : m = m->next;
93 154 : p2p_group_free_member(prev);
94 : }
95 338 : }
96 :
97 :
98 338 : void p2p_group_deinit(struct p2p_group *group)
99 : {
100 : size_t g;
101 : struct p2p_data *p2p;
102 :
103 338 : if (group == NULL)
104 338 : return;
105 :
106 338 : p2p = group->p2p;
107 :
108 2291 : for (g = 0; g < p2p->num_groups; g++) {
109 2291 : if (p2p->groups[g] == group) {
110 676 : while (g + 1 < p2p->num_groups) {
111 0 : p2p->groups[g] = p2p->groups[g + 1];
112 0 : g++;
113 : }
114 338 : p2p->num_groups--;
115 338 : break;
116 : }
117 : }
118 :
119 338 : p2p_group_free_members(group);
120 338 : os_free(group->cfg);
121 338 : wpabuf_free(group->noa);
122 338 : wpabuf_free(group->wfd_ie);
123 338 : os_free(group);
124 : }
125 :
126 :
127 633 : static void p2p_client_info(struct wpabuf *ie, struct p2p_group_member *m)
128 : {
129 633 : if (m->client_info == NULL)
130 53 : return;
131 580 : if (wpabuf_tailroom(ie) < wpabuf_len(m->client_info) + 1)
132 0 : return;
133 580 : wpabuf_put_buf(ie, m->client_info);
134 : }
135 :
136 :
137 2054 : static void p2p_group_add_common_ies(struct p2p_group *group,
138 : struct wpabuf *ie)
139 : {
140 2054 : u8 dev_capab = group->p2p->dev_capab, group_capab = 0;
141 :
142 : /* P2P Capability */
143 2054 : dev_capab &= ~P2P_DEV_CAPAB_CLIENT_DISCOVERABILITY;
144 2054 : group_capab |= P2P_GROUP_CAPAB_GROUP_OWNER;
145 2054 : if (group->cfg->persistent_group) {
146 527 : group_capab |= P2P_GROUP_CAPAB_PERSISTENT_GROUP;
147 527 : if (group->cfg->persistent_group == 2)
148 379 : group_capab |= P2P_GROUP_CAPAB_PERSISTENT_RECONN;
149 : }
150 2054 : if (group->p2p->cfg->p2p_intra_bss)
151 2054 : group_capab |= P2P_GROUP_CAPAB_INTRA_BSS_DIST;
152 2054 : if (group->group_formation)
153 791 : group_capab |= P2P_GROUP_CAPAB_GROUP_FORMATION;
154 2054 : if (group->p2p->cross_connect)
155 8 : group_capab |= P2P_GROUP_CAPAB_CROSS_CONN;
156 2054 : if (group->num_members >= group->cfg->max_clients)
157 0 : group_capab |= P2P_GROUP_CAPAB_GROUP_LIMIT;
158 2054 : group_capab |= P2P_GROUP_CAPAB_IP_ADDR_ALLOCATION;
159 2054 : p2p_buf_add_capability(ie, dev_capab, group_capab);
160 2054 : }
161 :
162 :
163 2054 : static void p2p_group_add_noa(struct wpabuf *ie, struct wpabuf *noa)
164 : {
165 2054 : if (noa == NULL)
166 4108 : return;
167 : /* Notice of Absence */
168 0 : wpabuf_put_u8(ie, P2P_ATTR_NOTICE_OF_ABSENCE);
169 0 : wpabuf_put_le16(ie, wpabuf_len(noa));
170 0 : wpabuf_put_buf(ie, noa);
171 : }
172 :
173 :
174 1380 : static struct wpabuf * p2p_group_encaps_probe_resp(struct wpabuf *subelems)
175 : {
176 : struct wpabuf *ie;
177 : const u8 *pos, *end;
178 : size_t len;
179 :
180 1380 : if (subelems == NULL)
181 0 : return NULL;
182 :
183 1380 : len = wpabuf_len(subelems) + 100;
184 :
185 1380 : ie = wpabuf_alloc(len);
186 1380 : if (ie == NULL)
187 0 : return NULL;
188 :
189 1380 : pos = wpabuf_head(subelems);
190 1380 : end = pos + wpabuf_len(subelems);
191 :
192 4142 : while (end > pos) {
193 1382 : size_t frag_len = end - pos;
194 1382 : if (frag_len > 251)
195 2 : frag_len = 251;
196 1382 : wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC);
197 1382 : wpabuf_put_u8(ie, 4 + frag_len);
198 1382 : wpabuf_put_be32(ie, P2P_IE_VENDOR_TYPE);
199 1382 : wpabuf_put_data(ie, pos, frag_len);
200 1382 : pos += frag_len;
201 : }
202 :
203 1380 : return ie;
204 : }
205 :
206 :
207 674 : static struct wpabuf * p2p_group_build_beacon_ie(struct p2p_group *group)
208 : {
209 : struct wpabuf *ie;
210 : u8 *len;
211 674 : size_t extra = 0;
212 :
213 : #ifdef CONFIG_WIFI_DISPLAY
214 674 : if (group->p2p->wfd_ie_beacon)
215 14 : extra = wpabuf_len(group->p2p->wfd_ie_beacon);
216 : #endif /* CONFIG_WIFI_DISPLAY */
217 :
218 839 : if (group->p2p->vendor_elem &&
219 165 : group->p2p->vendor_elem[VENDOR_ELEM_BEACON_P2P_GO])
220 2 : extra += wpabuf_len(group->p2p->vendor_elem[VENDOR_ELEM_BEACON_P2P_GO]);
221 :
222 674 : ie = wpabuf_alloc(257 + extra);
223 674 : if (ie == NULL)
224 0 : return NULL;
225 :
226 : #ifdef CONFIG_WIFI_DISPLAY
227 674 : if (group->p2p->wfd_ie_beacon)
228 14 : wpabuf_put_buf(ie, group->p2p->wfd_ie_beacon);
229 : #endif /* CONFIG_WIFI_DISPLAY */
230 :
231 839 : if (group->p2p->vendor_elem &&
232 165 : group->p2p->vendor_elem[VENDOR_ELEM_BEACON_P2P_GO])
233 2 : wpabuf_put_buf(ie,
234 2 : group->p2p->vendor_elem[VENDOR_ELEM_BEACON_P2P_GO]);
235 :
236 674 : len = p2p_buf_add_ie_hdr(ie);
237 674 : p2p_group_add_common_ies(group, ie);
238 674 : p2p_buf_add_device_id(ie, group->p2p->cfg->dev_addr);
239 674 : p2p_group_add_noa(ie, group->noa);
240 674 : p2p_buf_update_ie_hdr(ie, len);
241 :
242 674 : return ie;
243 : }
244 :
245 :
246 : #ifdef CONFIG_WIFI_DISPLAY
247 :
248 5 : struct wpabuf * p2p_group_get_wfd_ie(struct p2p_group *g)
249 : {
250 5 : return g->wfd_ie;
251 : }
252 :
253 :
254 394 : struct wpabuf * wifi_display_encaps(struct wpabuf *subelems)
255 : {
256 : struct wpabuf *ie;
257 : const u8 *pos, *end;
258 :
259 394 : if (subelems == NULL)
260 0 : return NULL;
261 :
262 394 : ie = wpabuf_alloc(wpabuf_len(subelems) + 100);
263 394 : if (ie == NULL)
264 0 : return NULL;
265 :
266 394 : pos = wpabuf_head(subelems);
267 394 : end = pos + wpabuf_len(subelems);
268 :
269 1070 : while (end > pos) {
270 282 : size_t frag_len = end - pos;
271 282 : if (frag_len > 251)
272 0 : frag_len = 251;
273 282 : wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC);
274 282 : wpabuf_put_u8(ie, 4 + frag_len);
275 282 : wpabuf_put_be32(ie, WFD_IE_VENDOR_TYPE);
276 282 : wpabuf_put_data(ie, pos, frag_len);
277 282 : pos += frag_len;
278 : }
279 :
280 394 : return ie;
281 : }
282 :
283 :
284 22 : static int wifi_display_add_dev_info_descr(struct wpabuf *buf,
285 : struct p2p_group_member *m)
286 : {
287 : const u8 *pos, *end;
288 22 : const u8 *dev_info = NULL;
289 22 : const u8 *assoc_bssid = NULL;
290 22 : const u8 *coupled_sink = NULL;
291 : u8 zero_addr[ETH_ALEN];
292 :
293 22 : if (m->wfd_ie == NULL)
294 0 : return 0;
295 :
296 22 : os_memset(zero_addr, 0, ETH_ALEN);
297 22 : pos = wpabuf_head_u8(m->wfd_ie);
298 22 : end = pos + wpabuf_len(m->wfd_ie);
299 74 : while (pos + 1 < end) {
300 : u8 id;
301 : u16 len;
302 :
303 30 : id = *pos++;
304 30 : len = WPA_GET_BE16(pos);
305 30 : pos += 2;
306 30 : if (pos + len > end)
307 0 : break;
308 :
309 30 : switch (id) {
310 : case WFD_SUBELEM_DEVICE_INFO:
311 22 : if (len < 6)
312 0 : break;
313 22 : dev_info = pos;
314 22 : break;
315 : case WFD_SUBELEM_ASSOCIATED_BSSID:
316 4 : if (len < ETH_ALEN)
317 0 : break;
318 4 : assoc_bssid = pos;
319 4 : break;
320 : case WFD_SUBELEM_COUPLED_SINK:
321 4 : if (len < 1 + ETH_ALEN)
322 0 : break;
323 4 : coupled_sink = pos;
324 4 : break;
325 : }
326 :
327 30 : pos += len;
328 : }
329 :
330 22 : if (dev_info == NULL)
331 0 : return 0;
332 :
333 22 : wpabuf_put_u8(buf, 23);
334 22 : wpabuf_put_data(buf, m->dev_addr, ETH_ALEN);
335 22 : if (assoc_bssid)
336 4 : wpabuf_put_data(buf, assoc_bssid, ETH_ALEN);
337 : else
338 18 : wpabuf_put_data(buf, zero_addr, ETH_ALEN);
339 22 : wpabuf_put_data(buf, dev_info, 2); /* WFD Device Info */
340 22 : wpabuf_put_data(buf, dev_info + 4, 2); /* WFD Device Max Throughput */
341 22 : if (coupled_sink) {
342 4 : wpabuf_put_data(buf, coupled_sink, 1 + ETH_ALEN);
343 : } else {
344 18 : wpabuf_put_u8(buf, 0);
345 18 : wpabuf_put_data(buf, zero_addr, ETH_ALEN);
346 : }
347 :
348 22 : return 1;
349 : }
350 :
351 :
352 : static struct wpabuf *
353 1380 : wifi_display_build_go_ie(struct p2p_group *group)
354 : {
355 : struct wpabuf *wfd_subelems, *wfd_ie;
356 : struct p2p_group_member *m;
357 : u8 *len;
358 1380 : unsigned int count = 0;
359 :
360 1380 : if (!group->p2p->wfd_ie_probe_resp)
361 1346 : return NULL;
362 :
363 68 : wfd_subelems = wpabuf_alloc(wpabuf_len(group->p2p->wfd_ie_probe_resp) +
364 34 : group->num_members * 24 + 100);
365 34 : if (wfd_subelems == NULL)
366 0 : return NULL;
367 34 : if (group->p2p->wfd_dev_info)
368 34 : wpabuf_put_buf(wfd_subelems, group->p2p->wfd_dev_info);
369 34 : if (group->p2p->wfd_assoc_bssid)
370 0 : wpabuf_put_buf(wfd_subelems,
371 0 : group->p2p->wfd_assoc_bssid);
372 34 : if (group->p2p->wfd_coupled_sink_info)
373 0 : wpabuf_put_buf(wfd_subelems,
374 0 : group->p2p->wfd_coupled_sink_info);
375 :
376 : /* Build WFD Session Info */
377 34 : wpabuf_put_u8(wfd_subelems, WFD_SUBELEM_SESSION_INFO);
378 34 : len = wpabuf_put(wfd_subelems, 2);
379 34 : m = group->members;
380 90 : while (m) {
381 22 : if (wifi_display_add_dev_info_descr(wfd_subelems, m))
382 22 : count++;
383 22 : m = m->next;
384 : }
385 :
386 34 : if (count == 0) {
387 : /* No Wi-Fi Display clients - do not include subelement */
388 16 : wfd_subelems->used -= 3;
389 : } else {
390 18 : WPA_PUT_BE16(len, (u8 *) wpabuf_put(wfd_subelems, 0) - len -
391 : 2);
392 18 : p2p_dbg(group->p2p, "WFD: WFD Session Info: %u descriptors",
393 : count);
394 : }
395 :
396 34 : wfd_ie = wifi_display_encaps(wfd_subelems);
397 34 : wpabuf_free(wfd_subelems);
398 :
399 34 : return wfd_ie;
400 : }
401 :
402 1380 : static void wifi_display_group_update(struct p2p_group *group)
403 : {
404 1380 : wpabuf_free(group->wfd_ie);
405 1380 : group->wfd_ie = wifi_display_build_go_ie(group);
406 1380 : }
407 :
408 : #endif /* CONFIG_WIFI_DISPLAY */
409 :
410 :
411 583 : void p2p_buf_add_group_info(struct p2p_group *group, struct wpabuf *buf,
412 : int max_clients)
413 : {
414 : u8 *group_info;
415 583 : int count = 0;
416 : struct p2p_group_member *m;
417 :
418 583 : p2p_dbg(group->p2p, "* P2P Group Info");
419 583 : group_info = wpabuf_put(buf, 0);
420 583 : wpabuf_put_u8(buf, P2P_ATTR_GROUP_INFO);
421 583 : wpabuf_put_le16(buf, 0); /* Length to be filled */
422 1216 : for (m = group->members; m; m = m->next) {
423 633 : p2p_client_info(buf, m);
424 633 : count++;
425 633 : if (max_clients >= 0 && count >= max_clients)
426 0 : break;
427 : }
428 583 : WPA_PUT_LE16(group_info + 1,
429 583 : (u8 *) wpabuf_put(buf, 0) - group_info - 3);
430 583 : }
431 :
432 :
433 5 : void p2p_group_buf_add_id(struct p2p_group *group, struct wpabuf *buf)
434 : {
435 5 : p2p_buf_add_group_id(buf, group->p2p->cfg->dev_addr, group->cfg->ssid,
436 5 : group->cfg->ssid_len);
437 5 : }
438 :
439 :
440 1380 : static struct wpabuf * p2p_group_build_probe_resp_ie(struct p2p_group *group)
441 : {
442 : struct wpabuf *p2p_subelems, *ie;
443 :
444 1380 : p2p_subelems = wpabuf_alloc(500);
445 1380 : if (p2p_subelems == NULL)
446 0 : return NULL;
447 :
448 1380 : p2p_group_add_common_ies(group, p2p_subelems);
449 1380 : p2p_group_add_noa(p2p_subelems, group->noa);
450 :
451 : /* P2P Device Info */
452 1380 : p2p_buf_add_device_info(p2p_subelems, group->p2p, NULL);
453 :
454 : /* P2P Group Info: Only when at least one P2P Client is connected */
455 1380 : if (group->members)
456 578 : p2p_buf_add_group_info(group, p2p_subelems, -1);
457 :
458 1380 : ie = p2p_group_encaps_probe_resp(p2p_subelems);
459 1380 : wpabuf_free(p2p_subelems);
460 :
461 1737 : if (group->p2p->vendor_elem &&
462 357 : group->p2p->vendor_elem[VENDOR_ELEM_PROBE_RESP_P2P_GO]) {
463 : struct wpabuf *extra;
464 2 : extra = wpabuf_dup(group->p2p->vendor_elem[VENDOR_ELEM_PROBE_RESP_P2P_GO]);
465 2 : ie = wpabuf_concat(extra, ie);
466 : }
467 :
468 : #ifdef CONFIG_WIFI_DISPLAY
469 1380 : if (group->wfd_ie) {
470 34 : struct wpabuf *wfd = wpabuf_dup(group->wfd_ie);
471 34 : ie = wpabuf_concat(wfd, ie);
472 : }
473 : #endif /* CONFIG_WIFI_DISPLAY */
474 :
475 1380 : return ie;
476 : }
477 :
478 :
479 1380 : void p2p_group_update_ies(struct p2p_group *group)
480 : {
481 : struct wpabuf *beacon_ie;
482 : struct wpabuf *probe_resp_ie;
483 :
484 : #ifdef CONFIG_WIFI_DISPLAY
485 1380 : wifi_display_group_update(group);
486 : #endif /* CONFIG_WIFI_DISPLAY */
487 :
488 1380 : probe_resp_ie = p2p_group_build_probe_resp_ie(group);
489 1380 : if (probe_resp_ie == NULL)
490 1380 : return;
491 1380 : wpa_hexdump_buf(MSG_MSGDUMP, "P2P: Update GO Probe Response P2P IE",
492 : probe_resp_ie);
493 :
494 1380 : if (group->beacon_update) {
495 674 : beacon_ie = p2p_group_build_beacon_ie(group);
496 674 : if (beacon_ie)
497 674 : group->beacon_update = 0;
498 674 : wpa_hexdump_buf(MSG_MSGDUMP, "P2P: Update GO Beacon P2P IE",
499 : beacon_ie);
500 : } else
501 706 : beacon_ie = NULL;
502 :
503 1380 : group->cfg->ie_update(group->cfg->cb_ctx, beacon_ie, probe_resp_ie);
504 : }
505 :
506 :
507 : /**
508 : * p2p_build_client_info - Build P2P Client Info Descriptor
509 : * @addr: MAC address of the peer device
510 : * @p2p_ie: P2P IE from (Re)Association Request
511 : * @dev_capab: Buffer for returning Device Capability
512 : * @dev_addr: Buffer for returning P2P Device Address
513 : * Returns: P2P Client Info Descriptor or %NULL on failure
514 : *
515 : * This function builds P2P Client Info Descriptor based on the information
516 : * available from (Re)Association Request frame. Group owner can use this to
517 : * build the P2P Group Info attribute for Probe Response frames.
518 : */
519 392 : static struct wpabuf * p2p_build_client_info(const u8 *addr,
520 : struct wpabuf *p2p_ie,
521 : u8 *dev_capab, u8 *dev_addr)
522 : {
523 : const u8 *spos;
524 : struct p2p_message msg;
525 : u8 *len_pos;
526 : struct wpabuf *buf;
527 :
528 392 : if (p2p_ie == NULL)
529 0 : return NULL;
530 :
531 392 : os_memset(&msg, 0, sizeof(msg));
532 784 : if (p2p_parse_p2p_ie(p2p_ie, &msg) ||
533 784 : msg.capability == NULL || msg.p2p_device_info == NULL)
534 0 : return NULL;
535 :
536 392 : buf = wpabuf_alloc(ETH_ALEN + 1 + 1 + msg.p2p_device_info_len);
537 392 : if (buf == NULL)
538 0 : return NULL;
539 :
540 392 : *dev_capab = msg.capability[0];
541 392 : os_memcpy(dev_addr, msg.p2p_device_addr, ETH_ALEN);
542 :
543 392 : spos = msg.p2p_device_info; /* P2P Device address */
544 :
545 : /* P2P Client Info Descriptor */
546 : /* Length to be set */
547 392 : len_pos = wpabuf_put(buf, 1);
548 : /* P2P Device address */
549 392 : wpabuf_put_data(buf, spos, ETH_ALEN);
550 : /* P2P Interface address */
551 392 : wpabuf_put_data(buf, addr, ETH_ALEN);
552 : /* Device Capability Bitmap */
553 392 : wpabuf_put_u8(buf, msg.capability[0]);
554 : /*
555 : * Config Methods, Primary Device Type, Number of Secondary Device
556 : * Types, Secondary Device Type List, Device Name copied from
557 : * Device Info
558 : */
559 392 : wpabuf_put_data(buf, spos + ETH_ALEN,
560 392 : msg.p2p_device_info_len - ETH_ALEN);
561 :
562 392 : *len_pos = wpabuf_len(buf) - 1;
563 :
564 :
565 392 : return buf;
566 : }
567 :
568 :
569 902 : static int p2p_group_remove_member(struct p2p_group *group, const u8 *addr)
570 : {
571 : struct p2p_group_member *m, *prev;
572 :
573 902 : if (group == NULL)
574 196 : return 0;
575 :
576 706 : m = group->members;
577 706 : prev = NULL;
578 1471 : while (m) {
579 340 : if (os_memcmp(m->addr, addr, ETH_ALEN) == 0)
580 281 : break;
581 59 : prev = m;
582 59 : m = m->next;
583 : }
584 :
585 706 : if (m == NULL)
586 425 : return 0;
587 :
588 281 : if (prev)
589 11 : prev->next = m->next;
590 : else
591 270 : group->members = m->next;
592 281 : p2p_group_free_member(m);
593 281 : group->num_members--;
594 :
595 281 : return 1;
596 : }
597 :
598 :
599 435 : int p2p_group_notif_assoc(struct p2p_group *group, const u8 *addr,
600 : const u8 *ie, size_t len)
601 : {
602 : struct p2p_group_member *m;
603 :
604 435 : if (group == NULL)
605 0 : return -1;
606 :
607 435 : p2p_add_device(group->p2p, addr, 0, NULL, 0, ie, len, 0);
608 :
609 435 : m = os_zalloc(sizeof(*m));
610 435 : if (m == NULL)
611 0 : return -1;
612 435 : os_memcpy(m->addr, addr, ETH_ALEN);
613 435 : m->p2p_ie = ieee802_11_vendor_ie_concat(ie, len, P2P_IE_VENDOR_TYPE);
614 435 : if (m->p2p_ie) {
615 392 : m->client_info = p2p_build_client_info(addr, m->p2p_ie,
616 : &m->dev_capab,
617 392 : m->dev_addr);
618 : }
619 : #ifdef CONFIG_WIFI_DISPLAY
620 435 : m->wfd_ie = ieee802_11_vendor_ie_concat(ie, len, WFD_IE_VENDOR_TYPE);
621 : #endif /* CONFIG_WIFI_DISPLAY */
622 :
623 435 : p2p_group_remove_member(group, addr);
624 :
625 435 : m->next = group->members;
626 435 : group->members = m;
627 435 : group->num_members++;
628 4350 : p2p_dbg(group->p2p, "Add client " MACSTR
629 : " to group (p2p=%d wfd=%d client_info=%d); num_members=%u/%u",
630 3480 : MAC2STR(addr), m->p2p_ie ? 1 : 0, m->wfd_ie ? 1 : 0,
631 435 : m->client_info ? 1 : 0,
632 435 : group->num_members, group->cfg->max_clients);
633 435 : if (group->num_members == group->cfg->max_clients)
634 0 : group->beacon_update = 1;
635 435 : p2p_group_update_ies(group);
636 435 : if (group->num_members == 1)
637 385 : group->cfg->idle_update(group->cfg->cb_ctx, 0);
638 :
639 435 : return 0;
640 : }
641 :
642 :
643 2026 : struct wpabuf * p2p_group_assoc_resp_ie(struct p2p_group *group, u8 status)
644 : {
645 : struct wpabuf *resp;
646 : u8 *rlen;
647 2026 : size_t extra = 0;
648 :
649 : #ifdef CONFIG_WIFI_DISPLAY
650 2026 : if (group->wfd_ie)
651 57 : extra = wpabuf_len(group->wfd_ie);
652 : #endif /* CONFIG_WIFI_DISPLAY */
653 :
654 2586 : if (group->p2p->vendor_elem &&
655 560 : group->p2p->vendor_elem[VENDOR_ELEM_P2P_ASSOC_RESP])
656 11 : extra += wpabuf_len(group->p2p->vendor_elem[VENDOR_ELEM_P2P_ASSOC_RESP]);
657 :
658 : /*
659 : * (Re)Association Response - P2P IE
660 : * Status attribute (shall be present when association request is
661 : * denied)
662 : * Extended Listen Timing (may be present)
663 : */
664 2026 : resp = wpabuf_alloc(20 + extra);
665 2026 : if (resp == NULL)
666 0 : return NULL;
667 :
668 : #ifdef CONFIG_WIFI_DISPLAY
669 2026 : if (group->wfd_ie)
670 57 : wpabuf_put_buf(resp, group->wfd_ie);
671 : #endif /* CONFIG_WIFI_DISPLAY */
672 :
673 2586 : if (group->p2p->vendor_elem &&
674 560 : group->p2p->vendor_elem[VENDOR_ELEM_P2P_ASSOC_RESP])
675 11 : wpabuf_put_buf(resp,
676 11 : group->p2p->vendor_elem[VENDOR_ELEM_P2P_ASSOC_RESP]);
677 :
678 2026 : rlen = p2p_buf_add_ie_hdr(resp);
679 2026 : if (status != P2P_SC_SUCCESS)
680 0 : p2p_buf_add_status(resp, status);
681 2026 : p2p_buf_update_ie_hdr(resp, rlen);
682 :
683 2026 : return resp;
684 : }
685 :
686 :
687 467 : void p2p_group_notif_disassoc(struct p2p_group *group, const u8 *addr)
688 : {
689 467 : if (p2p_group_remove_member(group, addr)) {
690 1897 : p2p_dbg(group->p2p, "Remove client " MACSTR
691 : " from group; num_members=%u/%u",
692 1626 : MAC2STR(addr), group->num_members,
693 271 : group->cfg->max_clients);
694 271 : if (group->num_members == group->cfg->max_clients - 1)
695 0 : group->beacon_update = 1;
696 271 : p2p_group_update_ies(group);
697 271 : if (group->num_members == 0)
698 237 : group->cfg->idle_update(group->cfg->cb_ctx, 1);
699 : }
700 467 : }
701 :
702 :
703 : /**
704 : * p2p_match_dev_type_member - Match client device type with requested type
705 : * @m: Group member
706 : * @wps: WPS TLVs from Probe Request frame (concatenated WPS IEs)
707 : * Returns: 1 on match, 0 on mismatch
708 : *
709 : * This function can be used to match the Requested Device Type attribute in
710 : * WPS IE with the device types of a group member for deciding whether a GO
711 : * should reply to a Probe Request frame.
712 : */
713 4 : static int p2p_match_dev_type_member(struct p2p_group_member *m,
714 : struct wpabuf *wps)
715 : {
716 : const u8 *pos, *end;
717 : struct wps_parse_attr attr;
718 : u8 num_sec;
719 :
720 4 : if (m->client_info == NULL || wps == NULL)
721 0 : return 0;
722 :
723 4 : pos = wpabuf_head(m->client_info);
724 4 : end = pos + wpabuf_len(m->client_info);
725 :
726 4 : pos += 1 + 2 * ETH_ALEN + 1 + 2;
727 4 : if (end - pos < WPS_DEV_TYPE_LEN + 1)
728 0 : return 0;
729 :
730 4 : if (wps_parse_msg(wps, &attr))
731 0 : return 1; /* assume no Requested Device Type attributes */
732 :
733 4 : if (attr.num_req_dev_type == 0)
734 0 : return 1; /* no Requested Device Type attributes -> match */
735 :
736 4 : if (dev_type_list_match(pos, attr.req_dev_type, attr.num_req_dev_type))
737 0 : return 1; /* Match with client Primary Device Type */
738 :
739 4 : pos += WPS_DEV_TYPE_LEN;
740 4 : num_sec = *pos++;
741 4 : if (end - pos < num_sec * WPS_DEV_TYPE_LEN)
742 0 : return 0;
743 11 : while (num_sec > 0) {
744 4 : num_sec--;
745 4 : if (dev_type_list_match(pos, attr.req_dev_type,
746 4 : attr.num_req_dev_type))
747 1 : return 1; /* Match with client Secondary Device Type */
748 3 : pos += WPS_DEV_TYPE_LEN;
749 : }
750 :
751 : /* No matching device type found */
752 3 : return 0;
753 : }
754 :
755 :
756 425 : int p2p_group_match_dev_type(struct p2p_group *group, struct wpabuf *wps)
757 : {
758 : struct p2p_group_member *m;
759 :
760 425 : if (p2p_match_dev_type(group->p2p, wps))
761 421 : return 1; /* Match with own device type */
762 :
763 7 : for (m = group->members; m; m = m->next) {
764 4 : if (p2p_match_dev_type_member(m, wps))
765 1 : return 1; /* Match with group client device type */
766 : }
767 :
768 : /* No match with Requested Device Type */
769 3 : return 0;
770 : }
771 :
772 :
773 420 : int p2p_group_match_dev_id(struct p2p_group *group, struct wpabuf *p2p)
774 : {
775 : struct p2p_group_member *m;
776 : struct p2p_message msg;
777 :
778 420 : os_memset(&msg, 0, sizeof(msg));
779 420 : if (p2p_parse_p2p_ie(p2p, &msg))
780 0 : return 1; /* Failed to parse - assume no filter on Device ID */
781 :
782 420 : if (!msg.device_id)
783 416 : return 1; /* No filter on Device ID */
784 :
785 4 : if (os_memcmp(msg.device_id, group->p2p->cfg->dev_addr, ETH_ALEN) == 0)
786 0 : return 1; /* Match with our P2P Device Address */
787 :
788 7 : for (m = group->members; m; m = m->next) {
789 4 : if (os_memcmp(msg.device_id, m->dev_addr, ETH_ALEN) == 0)
790 1 : return 1; /* Match with group client P2P Device Address */
791 : }
792 :
793 : /* No match with Device ID */
794 3 : return 0;
795 : }
796 :
797 :
798 332 : void p2p_group_notif_formation_done(struct p2p_group *group)
799 : {
800 332 : if (group == NULL)
801 332 : return;
802 332 : group->group_formation = 0;
803 332 : group->beacon_update = 1;
804 332 : p2p_group_update_ies(group);
805 : }
806 :
807 :
808 0 : int p2p_group_notif_noa(struct p2p_group *group, const u8 *noa,
809 : size_t noa_len)
810 : {
811 0 : if (noa == NULL) {
812 0 : wpabuf_free(group->noa);
813 0 : group->noa = NULL;
814 : } else {
815 0 : if (group->noa) {
816 0 : if (wpabuf_size(group->noa) >= noa_len) {
817 0 : group->noa->used = 0;
818 0 : wpabuf_put_data(group->noa, noa, noa_len);
819 : } else {
820 0 : wpabuf_free(group->noa);
821 0 : group->noa = NULL;
822 : }
823 : }
824 :
825 0 : if (!group->noa) {
826 0 : group->noa = wpabuf_alloc_copy(noa, noa_len);
827 0 : if (group->noa == NULL)
828 0 : return -1;
829 : }
830 : }
831 :
832 0 : group->beacon_update = 1;
833 0 : p2p_group_update_ies(group);
834 0 : return 0;
835 : }
836 :
837 :
838 8 : static struct p2p_group_member * p2p_group_get_client(struct p2p_group *group,
839 : const u8 *dev_id)
840 : {
841 : struct p2p_group_member *m;
842 :
843 8 : for (m = group->members; m; m = m->next) {
844 0 : if (os_memcmp(dev_id, m->dev_addr, ETH_ALEN) == 0)
845 0 : return m;
846 : }
847 :
848 8 : return NULL;
849 : }
850 :
851 :
852 320 : static struct p2p_group_member * p2p_group_get_client_iface(
853 : struct p2p_group *group, const u8 *interface_addr)
854 : {
855 : struct p2p_group_member *m;
856 :
857 329 : for (m = group->members; m; m = m->next) {
858 329 : if (os_memcmp(interface_addr, m->addr, ETH_ALEN) == 0)
859 320 : return m;
860 : }
861 :
862 0 : return NULL;
863 : }
864 :
865 :
866 315 : const u8 * p2p_group_get_dev_addr(struct p2p_group *group, const u8 *addr)
867 : {
868 : struct p2p_group_member *m;
869 :
870 315 : if (group == NULL)
871 0 : return NULL;
872 315 : m = p2p_group_get_client_iface(group, addr);
873 315 : if (m && !is_zero_ether_addr(m->dev_addr))
874 264 : return m->dev_addr;
875 51 : return NULL;
876 : }
877 :
878 :
879 0 : static struct wpabuf * p2p_build_go_disc_req(void)
880 : {
881 : struct wpabuf *buf;
882 :
883 0 : buf = wpabuf_alloc(100);
884 0 : if (buf == NULL)
885 0 : return NULL;
886 :
887 0 : p2p_buf_add_action_hdr(buf, P2P_GO_DISC_REQ, 0);
888 :
889 0 : return buf;
890 : }
891 :
892 :
893 8 : int p2p_group_go_discover(struct p2p_group *group, const u8 *dev_id,
894 : const u8 *searching_dev, int rx_freq)
895 : {
896 : struct p2p_group_member *m;
897 : struct wpabuf *req;
898 8 : struct p2p_data *p2p = group->p2p;
899 : int freq;
900 :
901 8 : m = p2p_group_get_client(group, dev_id);
902 8 : if (m == NULL || m->client_info == NULL) {
903 48 : p2p_dbg(group->p2p, "Requested client was not in this group "
904 48 : MACSTR, MAC2STR(group->cfg->interface_addr));
905 8 : return -1;
906 : }
907 :
908 0 : if (!(m->dev_capab & P2P_DEV_CAPAB_CLIENT_DISCOVERABILITY)) {
909 0 : p2p_dbg(group->p2p, "Requested client does not support client discoverability");
910 0 : return -1;
911 : }
912 :
913 0 : p2p_dbg(group->p2p, "Schedule GO Discoverability Request to be sent to "
914 0 : MACSTR, MAC2STR(dev_id));
915 :
916 0 : req = p2p_build_go_disc_req();
917 0 : if (req == NULL)
918 0 : return -1;
919 :
920 : /* TODO: Should really use group operating frequency here */
921 0 : freq = rx_freq;
922 :
923 0 : p2p->pending_action_state = P2P_PENDING_GO_DISC_REQ;
924 0 : if (p2p->cfg->send_action(p2p->cfg->cb_ctx, freq, m->addr,
925 0 : group->cfg->interface_addr,
926 0 : group->cfg->interface_addr,
927 0 : wpabuf_head(req), wpabuf_len(req), 200) < 0)
928 : {
929 0 : p2p_dbg(p2p, "Failed to send Action frame");
930 : }
931 :
932 0 : wpabuf_free(req);
933 :
934 0 : return 0;
935 : }
936 :
937 :
938 7 : const u8 * p2p_group_get_interface_addr(struct p2p_group *group)
939 : {
940 7 : return group->cfg->interface_addr;
941 : }
942 :
943 :
944 5 : u8 p2p_group_presence_req(struct p2p_group *group,
945 : const u8 *client_interface_addr,
946 : const u8 *noa, size_t noa_len)
947 : {
948 : struct p2p_group_member *m;
949 : u8 curr_noa[50];
950 : int curr_noa_len;
951 :
952 5 : m = p2p_group_get_client_iface(group, client_interface_addr);
953 5 : if (m == NULL || m->client_info == NULL) {
954 0 : p2p_dbg(group->p2p, "Client was not in this group");
955 0 : return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE;
956 : }
957 :
958 5 : wpa_hexdump(MSG_DEBUG, "P2P: Presence Request NoA", noa, noa_len);
959 :
960 5 : if (group->p2p->cfg->get_noa)
961 10 : curr_noa_len = group->p2p->cfg->get_noa(
962 5 : group->p2p->cfg->cb_ctx, group->cfg->interface_addr,
963 : curr_noa, sizeof(curr_noa));
964 : else
965 0 : curr_noa_len = -1;
966 5 : if (curr_noa_len < 0)
967 5 : p2p_dbg(group->p2p, "Failed to fetch current NoA");
968 0 : else if (curr_noa_len == 0)
969 0 : p2p_dbg(group->p2p, "No NoA being advertized");
970 : else
971 0 : wpa_hexdump(MSG_DEBUG, "P2P: Current NoA", curr_noa,
972 : curr_noa_len);
973 :
974 : /* TODO: properly process request and store copy */
975 5 : if (curr_noa_len > 0 || curr_noa_len == -1)
976 5 : return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE;
977 :
978 0 : return P2P_SC_SUCCESS;
979 : }
980 :
981 :
982 5 : unsigned int p2p_get_group_num_members(struct p2p_group *group)
983 : {
984 5 : if (!group)
985 0 : return 0;
986 :
987 5 : return group->num_members;
988 : }
989 :
990 :
991 0 : int p2p_client_limit_reached(struct p2p_group *group)
992 : {
993 0 : if (!group || !group->cfg)
994 0 : return 1;
995 :
996 0 : return group->num_members >= group->cfg->max_clients;
997 : }
998 :
999 :
1000 7 : const u8 * p2p_iterate_group_members(struct p2p_group *group, void **next)
1001 : {
1002 7 : struct p2p_group_member *iter = *next;
1003 :
1004 7 : if (!iter)
1005 5 : iter = group->members;
1006 : else
1007 2 : iter = iter->next;
1008 :
1009 7 : *next = iter;
1010 :
1011 7 : if (!iter)
1012 5 : return NULL;
1013 :
1014 2 : return iter->dev_addr;
1015 : }
1016 :
1017 :
1018 491 : int p2p_group_is_client_connected(struct p2p_group *group, const u8 *dev_addr)
1019 : {
1020 : struct p2p_group_member *m;
1021 :
1022 515 : for (m = group->members; m; m = m->next) {
1023 458 : if (os_memcmp(m->dev_addr, dev_addr, ETH_ALEN) == 0)
1024 434 : return 1;
1025 : }
1026 :
1027 57 : return 0;
1028 : }
1029 :
1030 :
1031 61 : int p2p_group_is_group_id_match(struct p2p_group *group, const u8 *group_id,
1032 : size_t group_id_len)
1033 : {
1034 61 : if (group_id_len != ETH_ALEN + group->cfg->ssid_len)
1035 0 : return 0;
1036 61 : if (os_memcmp(group_id, group->p2p->cfg->dev_addr, ETH_ALEN) != 0)
1037 0 : return 0;
1038 122 : return os_memcmp(group_id + ETH_ALEN, group->cfg->ssid,
1039 61 : group->cfg->ssid_len) == 0;
1040 : }
1041 :
1042 :
1043 4 : void p2p_group_force_beacon_update_ies(struct p2p_group *group)
1044 : {
1045 4 : group->beacon_update = 1;
1046 4 : p2p_group_update_ies(group);
1047 4 : }
1048 :
1049 :
1050 5 : int p2p_group_get_freq(struct p2p_group *group)
1051 : {
1052 5 : return group->cfg->freq;
1053 : }
1054 :
1055 :
1056 434 : const struct p2p_group_config * p2p_group_get_config(struct p2p_group *group)
1057 : {
1058 434 : return group->cfg;
1059 : }
1060 :
1061 :
1062 880 : void p2p_loop_on_all_groups(struct p2p_data *p2p,
1063 : int (*group_callback)(struct p2p_group *group,
1064 : void *user_data),
1065 : void *user_data)
1066 : {
1067 : unsigned int i;
1068 :
1069 1258 : for (i = 0; i < p2p->num_groups; i++) {
1070 491 : if (!group_callback(p2p->groups[i], user_data))
1071 113 : break;
1072 : }
1073 880 : }
1074 :
1075 :
1076 8004 : int p2p_group_get_common_freqs(struct p2p_group *group, int *common_freqs,
1077 : unsigned int *num)
1078 :
1079 : {
1080 : struct p2p_channels intersect, res;
1081 : struct p2p_group_member *m;
1082 :
1083 8004 : if (!group || !common_freqs || !num)
1084 0 : return -1;
1085 :
1086 8004 : os_memset(&intersect, 0, sizeof(intersect));
1087 8004 : os_memset(&res, 0, sizeof(res));
1088 :
1089 8004 : p2p_channels_union(&intersect, &group->p2p->cfg->channels,
1090 : &intersect);
1091 :
1092 8004 : p2p_channels_dump(group->p2p,
1093 : "Group common freqs before iterating members",
1094 : &intersect);
1095 :
1096 8076 : for (m = group->members; m; m = m->next) {
1097 : struct p2p_device *dev;
1098 :
1099 72 : dev = p2p_get_device(group->p2p, m->dev_addr);
1100 72 : if (!dev)
1101 0 : continue;
1102 :
1103 72 : p2p_channels_intersect(&intersect, &dev->channels, &res);
1104 72 : intersect = res;
1105 : }
1106 :
1107 8004 : p2p_channels_dump(group->p2p, "Group common channels", &intersect);
1108 :
1109 8004 : os_memset(common_freqs, 0, *num * sizeof(int));
1110 8004 : *num = p2p_channels_to_freqs(&intersect, common_freqs, *num);
1111 :
1112 8004 : return 0;
1113 : }
|
