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