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