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