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 : 86 : 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 : 86 : group = os_zalloc(sizeof(*group));
51 [ - + ]: 86 : if (group == NULL)
52 : 0 : return NULL;
53 : :
54 : 86 : groups = os_realloc_array(p2p->groups, p2p->num_groups + 1,
55 : : sizeof(struct p2p_group *));
56 [ - + ]: 86 : if (groups == NULL) {
57 : 0 : os_free(group);
58 : 0 : return NULL;
59 : : }
60 : 86 : groups[p2p->num_groups++] = group;
61 : 86 : p2p->groups = groups;
62 : :
63 : 86 : group->p2p = p2p;
64 : 86 : group->cfg = config;
65 : 86 : group->group_formation = 1;
66 : 86 : group->beacon_update = 1;
67 : 86 : p2p_group_update_ies(group);
68 : 86 : group->cfg->idle_update(group->cfg->cb_ctx, 1);
69 : :
70 : 86 : return group;
71 : : }
72 : :
73 : :
74 : 193 : static void p2p_group_free_member(struct p2p_group_member *m)
75 : : {
76 : 193 : wpabuf_free(m->wfd_ie);
77 : 193 : wpabuf_free(m->p2p_ie);
78 : 193 : wpabuf_free(m->client_info);
79 : 193 : os_free(m);
80 : 193 : }
81 : :
82 : :
83 : 86 : static void p2p_group_free_members(struct p2p_group *group)
84 : : {
85 : : struct p2p_group_member *m, *prev;
86 : 86 : m = group->members;
87 : 86 : group->members = NULL;
88 : 86 : group->num_members = 0;
89 [ + + ]: 149 : while (m) {
90 : 63 : prev = m;
91 : 63 : m = m->next;
92 : 63 : p2p_group_free_member(prev);
93 : : }
94 : 86 : }
95 : :
96 : :
97 : 86 : void p2p_group_deinit(struct p2p_group *group)
98 : : {
99 : : size_t g;
100 : : struct p2p_data *p2p;
101 : :
102 [ - + ]: 86 : if (group == NULL)
103 : 86 : return;
104 : :
105 : 86 : p2p = group->p2p;
106 : :
107 [ + - ]: 86 : for (g = 0; g < p2p->num_groups; g++) {
108 [ + - ]: 86 : if (p2p->groups[g] == group) {
109 [ - + ]: 86 : while (g + 1 < p2p->num_groups) {
110 : 0 : p2p->groups[g] = p2p->groups[g + 1];
111 : 0 : g++;
112 : : }
113 : 86 : p2p->num_groups--;
114 : 86 : break;
115 : : }
116 : : }
117 : :
118 : 86 : p2p_group_free_members(group);
119 : 86 : os_free(group->cfg);
120 : 86 : wpabuf_free(group->noa);
121 : 86 : wpabuf_free(group->wfd_ie);
122 : 86 : os_free(group);
123 : : }
124 : :
125 : :
126 : 294 : static void p2p_client_info(struct wpabuf *ie, struct p2p_group_member *m)
127 : : {
128 [ + + ]: 294 : if (m->client_info == NULL)
129 : 33 : return;
130 [ - + ]: 261 : if (wpabuf_tailroom(ie) < wpabuf_len(m->client_info) + 1)
131 : 0 : return;
132 : 294 : wpabuf_put_buf(ie, m->client_info);
133 : : }
134 : :
135 : :
136 : 659 : static void p2p_group_add_common_ies(struct p2p_group *group,
137 : : struct wpabuf *ie)
138 : : {
139 : 659 : u8 dev_capab = group->p2p->dev_capab, group_capab = 0;
140 : :
141 : : /* P2P Capability */
142 : 659 : dev_capab &= ~P2P_DEV_CAPAB_CLIENT_DISCOVERABILITY;
143 : 659 : group_capab |= P2P_GROUP_CAPAB_GROUP_OWNER;
144 [ + + ]: 659 : if (group->cfg->persistent_group) {
145 : 108 : group_capab |= P2P_GROUP_CAPAB_PERSISTENT_GROUP;
146 [ + + ]: 108 : if (group->cfg->persistent_group == 2)
147 : 87 : group_capab |= P2P_GROUP_CAPAB_PERSISTENT_RECONN;
148 : : }
149 [ + - ]: 659 : if (group->p2p->cfg->p2p_intra_bss)
150 : 659 : group_capab |= P2P_GROUP_CAPAB_INTRA_BSS_DIST;
151 [ + + ]: 659 : if (group->group_formation)
152 : 218 : group_capab |= P2P_GROUP_CAPAB_GROUP_FORMATION;
153 [ - + ]: 659 : if (group->p2p->cross_connect)
154 : 0 : group_capab |= P2P_GROUP_CAPAB_CROSS_CONN;
155 [ - + ]: 659 : if (group->num_members >= group->cfg->max_clients)
156 : 0 : group_capab |= P2P_GROUP_CAPAB_GROUP_LIMIT;
157 : 659 : group_capab |= P2P_GROUP_CAPAB_IP_ADDR_ALLOCATION;
158 : 659 : p2p_buf_add_capability(ie, dev_capab, group_capab);
159 : 659 : }
160 : :
161 : :
162 : 659 : static void p2p_group_add_noa(struct wpabuf *ie, struct wpabuf *noa)
163 : : {
164 [ + - ]: 659 : if (noa == NULL)
165 : 659 : 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 : 488 : 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 [ - + ]: 488 : if (subelems == NULL)
180 : 0 : return NULL;
181 : :
182 : 488 : len = wpabuf_len(subelems) + 100;
183 : :
184 : 488 : ie = wpabuf_alloc(len);
185 [ - + ]: 488 : if (ie == NULL)
186 : 0 : return NULL;
187 : :
188 : 488 : pos = wpabuf_head(subelems);
189 : 488 : end = pos + wpabuf_len(subelems);
190 : :
191 [ + + ]: 976 : while (end > pos) {
192 : 488 : size_t frag_len = end - pos;
193 [ - + ]: 488 : if (frag_len > 251)
194 : 0 : frag_len = 251;
195 : 488 : wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC);
196 : 488 : wpabuf_put_u8(ie, 4 + frag_len);
197 : 488 : wpabuf_put_be32(ie, P2P_IE_VENDOR_TYPE);
198 : 488 : wpabuf_put_data(ie, pos, frag_len);
199 : 488 : pos += frag_len;
200 : : }
201 : :
202 : 488 : return ie;
203 : : }
204 : :
205 : :
206 : 171 : static struct wpabuf * p2p_group_build_beacon_ie(struct p2p_group *group)
207 : : {
208 : : struct wpabuf *ie;
209 : : u8 *len;
210 : 171 : size_t extra = 0;
211 : :
212 : : #ifdef CONFIG_WIFI_DISPLAY
213 [ + + ]: 171 : if (group->p2p->wfd_ie_beacon)
214 : 139 : extra = wpabuf_len(group->p2p->wfd_ie_beacon);
215 : : #endif /* CONFIG_WIFI_DISPLAY */
216 : :
217 : 171 : ie = wpabuf_alloc(257 + extra);
218 [ - + ]: 171 : if (ie == NULL)
219 : 0 : return NULL;
220 : :
221 : : #ifdef CONFIG_WIFI_DISPLAY
222 [ + + ]: 171 : if (group->p2p->wfd_ie_beacon)
223 : 139 : wpabuf_put_buf(ie, group->p2p->wfd_ie_beacon);
224 : : #endif /* CONFIG_WIFI_DISPLAY */
225 : :
226 : 171 : len = p2p_buf_add_ie_hdr(ie);
227 : 171 : p2p_group_add_common_ies(group, ie);
228 : 171 : p2p_buf_add_device_id(ie, group->p2p->cfg->dev_addr);
229 : 171 : p2p_group_add_noa(ie, group->noa);
230 : 171 : p2p_buf_update_ie_hdr(ie, len);
231 : :
232 : 171 : return ie;
233 : : }
234 : :
235 : :
236 : : #ifdef CONFIG_WIFI_DISPLAY
237 : :
238 : 31 : struct wpabuf * p2p_group_get_wfd_ie(struct p2p_group *g)
239 : : {
240 : 31 : return g->wfd_ie;
241 : : }
242 : :
243 : :
244 : 440 : struct wpabuf * wifi_display_encaps(struct wpabuf *subelems)
245 : : {
246 : : struct wpabuf *ie;
247 : : const u8 *pos, *end;
248 : :
249 [ - + ]: 440 : if (subelems == NULL)
250 : 0 : return NULL;
251 : :
252 : 440 : ie = wpabuf_alloc(wpabuf_len(subelems) + 100);
253 [ - + ]: 440 : if (ie == NULL)
254 : 0 : return NULL;
255 : :
256 : 440 : pos = wpabuf_head(subelems);
257 : 440 : end = pos + wpabuf_len(subelems);
258 : :
259 [ + + ]: 856 : while (end > pos) {
260 : 416 : size_t frag_len = end - pos;
261 [ - + ]: 416 : if (frag_len > 251)
262 : 0 : frag_len = 251;
263 : 416 : wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC);
264 : 416 : wpabuf_put_u8(ie, 4 + frag_len);
265 : 416 : wpabuf_put_be32(ie, WFD_IE_VENDOR_TYPE);
266 : 416 : wpabuf_put_data(ie, pos, frag_len);
267 : 416 : pos += frag_len;
268 : : }
269 : :
270 : 440 : return ie;
271 : : }
272 : :
273 : :
274 : 241 : static int wifi_display_add_dev_info_descr(struct wpabuf *buf,
275 : : struct p2p_group_member *m)
276 : : {
277 : : const u8 *pos, *end;
278 : 241 : const u8 *dev_info = NULL;
279 : 241 : const u8 *assoc_bssid = NULL;
280 : 241 : const u8 *coupled_sink = NULL;
281 : : u8 zero_addr[ETH_ALEN];
282 : :
283 [ + + ]: 241 : if (m->wfd_ie == NULL)
284 : 14 : return 0;
285 : :
286 : 227 : os_memset(zero_addr, 0, ETH_ALEN);
287 : 227 : pos = wpabuf_head_u8(m->wfd_ie);
288 : 227 : end = pos + wpabuf_len(m->wfd_ie);
289 [ + + ]: 454 : while (pos + 1 < end) {
290 : : u8 id;
291 : : u16 len;
292 : :
293 : 227 : id = *pos++;
294 : 227 : len = WPA_GET_BE16(pos);
295 : 227 : pos += 2;
296 [ - + ]: 227 : if (pos + len > end)
297 : 0 : break;
298 : :
299 [ + - - - ]: 227 : switch (id) {
300 : : case WFD_SUBELEM_DEVICE_INFO:
301 [ - + ]: 227 : if (len < 6)
302 : 0 : break;
303 : 227 : dev_info = pos;
304 : 227 : break;
305 : : case WFD_SUBELEM_ASSOCIATED_BSSID:
306 [ # # ]: 0 : if (len < ETH_ALEN)
307 : 0 : break;
308 : 0 : assoc_bssid = pos;
309 : 0 : break;
310 : : case WFD_SUBELEM_COUPLED_SINK:
311 [ # # ]: 0 : if (len < 1 + ETH_ALEN)
312 : 0 : break;
313 : 0 : coupled_sink = pos;
314 : 0 : break;
315 : : }
316 : :
317 : 227 : pos += len;
318 : : }
319 : :
320 [ - + ]: 227 : if (dev_info == NULL)
321 : 0 : return 0;
322 : :
323 : 227 : wpabuf_put_u8(buf, 23);
324 : 227 : wpabuf_put_data(buf, m->dev_addr, ETH_ALEN);
325 [ - + ]: 227 : if (assoc_bssid)
326 : 0 : wpabuf_put_data(buf, assoc_bssid, ETH_ALEN);
327 : : else
328 : 227 : wpabuf_put_data(buf, zero_addr, ETH_ALEN);
329 : 227 : wpabuf_put_data(buf, dev_info, 2); /* WFD Device Info */
330 : 227 : wpabuf_put_data(buf, dev_info + 4, 2); /* WFD Device Max Throughput */
331 [ - + ]: 227 : if (coupled_sink) {
332 : 0 : wpabuf_put_data(buf, coupled_sink, 1 + ETH_ALEN);
333 : : } else {
334 : 227 : wpabuf_put_u8(buf, 0);
335 : 227 : wpabuf_put_data(buf, zero_addr, ETH_ALEN);
336 : : }
337 : :
338 : 241 : return 1;
339 : : }
340 : :
341 : :
342 : : static struct wpabuf *
343 : 488 : 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 : 488 : unsigned int count = 0;
349 : :
350 [ + + ]: 488 : if (!group->p2p->wfd_ie_probe_resp)
351 : 96 : return NULL;
352 : :
353 : 392 : wfd_subelems = wpabuf_alloc(wpabuf_len(group->p2p->wfd_ie_probe_resp) +
354 : 392 : group->num_members * 24 + 100);
355 [ - + ]: 392 : if (wfd_subelems == NULL)
356 : 0 : return NULL;
357 [ + - ]: 392 : if (group->p2p->wfd_dev_info)
358 : 392 : wpabuf_put_buf(wfd_subelems, group->p2p->wfd_dev_info);
359 [ - + ]: 392 : if (group->p2p->wfd_assoc_bssid)
360 : 0 : wpabuf_put_buf(wfd_subelems,
361 : 0 : group->p2p->wfd_assoc_bssid);
362 [ - + ]: 392 : 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 : 392 : wpabuf_put_u8(wfd_subelems, WFD_SUBELEM_SESSION_INFO);
368 : 392 : len = wpabuf_put(wfd_subelems, 2);
369 : 392 : m = group->members;
370 [ + + ]: 633 : while (m) {
371 [ + + ]: 241 : if (wifi_display_add_dev_info_descr(wfd_subelems, m))
372 : 227 : count++;
373 : 241 : m = m->next;
374 : : }
375 : :
376 [ + + ]: 392 : if (count == 0) {
377 : : /* No Wi-Fi Display clients - do not include subelement */
378 : 189 : wfd_subelems->used -= 3;
379 : : } else {
380 : 203 : WPA_PUT_BE16(len, (u8 *) wpabuf_put(wfd_subelems, 0) - len -
381 : : 2);
382 : 203 : p2p_dbg(group->p2p, "WFD: WFD Session Info: %u descriptors",
383 : : count);
384 : : }
385 : :
386 : 392 : wfd_ie = wifi_display_encaps(wfd_subelems);
387 : 392 : wpabuf_free(wfd_subelems);
388 : :
389 : 488 : return wfd_ie;
390 : : }
391 : :
392 : 488 : static void wifi_display_group_update(struct p2p_group *group)
393 : : {
394 : 488 : wpabuf_free(group->wfd_ie);
395 : 488 : group->wfd_ie = wifi_display_build_go_ie(group);
396 : 488 : }
397 : :
398 : : #endif /* CONFIG_WIFI_DISPLAY */
399 : :
400 : :
401 : 265 : void p2p_buf_add_group_info(struct p2p_group *group, struct wpabuf *buf,
402 : : int max_clients)
403 : : {
404 : : u8 *group_info;
405 : 265 : int count = 0;
406 : : struct p2p_group_member *m;
407 : :
408 : 265 : p2p_dbg(group->p2p, "* P2P Group Info");
409 : 265 : group_info = wpabuf_put(buf, 0);
410 : 265 : wpabuf_put_u8(buf, P2P_ATTR_GROUP_INFO);
411 : 265 : wpabuf_put_le16(buf, 0); /* Length to be filled */
412 [ + + ]: 559 : for (m = group->members; m; m = m->next) {
413 : 294 : p2p_client_info(buf, m);
414 : 294 : count++;
415 [ - + ][ # # ]: 294 : if (max_clients >= 0 && count >= max_clients)
416 : 0 : break;
417 : : }
418 : 265 : WPA_PUT_LE16(group_info + 1,
419 : 265 : (u8 *) wpabuf_put(buf, 0) - group_info - 3);
420 : 265 : }
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 : 488 : static struct wpabuf * p2p_group_build_probe_resp_ie(struct p2p_group *group)
431 : : {
432 : : struct wpabuf *p2p_subelems, *ie;
433 : :
434 : 488 : p2p_subelems = wpabuf_alloc(500);
435 [ - + ]: 488 : if (p2p_subelems == NULL)
436 : 0 : return NULL;
437 : :
438 : 488 : p2p_group_add_common_ies(group, p2p_subelems);
439 : 488 : p2p_group_add_noa(p2p_subelems, group->noa);
440 : :
441 : : /* P2P Device Info */
442 : 488 : 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 [ + + ]: 488 : if (group->members)
446 : 260 : p2p_buf_add_group_info(group, p2p_subelems, -1);
447 : :
448 : 488 : ie = p2p_group_encaps_probe_resp(p2p_subelems);
449 : 488 : wpabuf_free(p2p_subelems);
450 : :
451 : : #ifdef CONFIG_WIFI_DISPLAY
452 [ + + ]: 488 : if (group->wfd_ie) {
453 : 392 : struct wpabuf *wfd = wpabuf_dup(group->wfd_ie);
454 : 392 : ie = wpabuf_concat(wfd, ie);
455 : : }
456 : : #endif /* CONFIG_WIFI_DISPLAY */
457 : :
458 : 488 : return ie;
459 : : }
460 : :
461 : :
462 : 488 : 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 : 488 : wifi_display_group_update(group);
469 : : #endif /* CONFIG_WIFI_DISPLAY */
470 : :
471 : 488 : probe_resp_ie = p2p_group_build_probe_resp_ie(group);
472 [ - + ]: 488 : if (probe_resp_ie == NULL)
473 : 488 : return;
474 : 488 : wpa_hexdump_buf(MSG_MSGDUMP, "P2P: Update GO Probe Response P2P IE",
475 : : probe_resp_ie);
476 : :
477 [ + + ]: 488 : if (group->beacon_update) {
478 : 171 : beacon_ie = p2p_group_build_beacon_ie(group);
479 [ + - ]: 171 : if (beacon_ie)
480 : 171 : group->beacon_update = 0;
481 : 171 : wpa_hexdump_buf(MSG_MSGDUMP, "P2P: Update GO Beacon P2P IE",
482 : : beacon_ie);
483 : : } else
484 : 317 : beacon_ie = NULL;
485 : :
486 : 488 : 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 : 172 : 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 [ - + ]: 172 : if (p2p_ie == NULL)
512 : 0 : return NULL;
513 : :
514 : 172 : os_memset(&msg, 0, sizeof(msg));
515 [ + - ][ + - ]: 172 : if (p2p_parse_p2p_ie(p2p_ie, &msg) ||
516 [ + + ]: 172 : msg.capability == NULL || msg.p2p_device_info == NULL)
517 : 5 : return NULL;
518 : :
519 : 167 : buf = wpabuf_alloc(ETH_ALEN + 1 + 1 + msg.p2p_device_info_len);
520 [ - + ]: 167 : if (buf == NULL)
521 : 0 : return NULL;
522 : :
523 : 167 : *dev_capab = msg.capability[0];
524 : 167 : os_memcpy(dev_addr, msg.p2p_device_addr, ETH_ALEN);
525 : :
526 : 167 : spos = msg.p2p_device_info; /* P2P Device address */
527 : :
528 : : /* P2P Client Info Descriptor */
529 : : /* Length to be set */
530 : 167 : len_pos = wpabuf_put(buf, 1);
531 : : /* P2P Device address */
532 : 167 : wpabuf_put_data(buf, spos, ETH_ALEN);
533 : : /* P2P Interface address */
534 : 167 : wpabuf_put_data(buf, addr, ETH_ALEN);
535 : : /* Device Capability Bitmap */
536 : 167 : 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 : 167 : wpabuf_put_data(buf, spos + ETH_ALEN,
543 : 167 : msg.p2p_device_info_len - ETH_ALEN);
544 : :
545 : 167 : *len_pos = wpabuf_len(buf) - 1;
546 : :
547 : :
548 : 172 : return buf;
549 : : }
550 : :
551 : :
552 : 380 : static int p2p_group_remove_member(struct p2p_group *group, const u8 *addr)
553 : : {
554 : : struct p2p_group_member *m, *prev;
555 : :
556 [ + + ]: 380 : if (group == NULL)
557 : 63 : return 0;
558 : :
559 : 317 : m = group->members;
560 : 317 : prev = NULL;
561 [ + + ]: 355 : while (m) {
562 [ + + ]: 168 : if (os_memcmp(m->addr, addr, ETH_ALEN) == 0)
563 : 130 : break;
564 : 38 : prev = m;
565 : 38 : m = m->next;
566 : : }
567 : :
568 [ + + ]: 317 : if (m == NULL)
569 : 187 : return 0;
570 : :
571 [ + + ]: 130 : if (prev)
572 : 7 : prev->next = m->next;
573 : : else
574 : 123 : group->members = m->next;
575 : 130 : p2p_group_free_member(m);
576 : 130 : group->num_members--;
577 : :
578 : 380 : return 1;
579 : : }
580 : :
581 : :
582 : 193 : 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 [ - + ]: 193 : if (group == NULL)
588 : 0 : return -1;
589 : :
590 : 193 : p2p_add_device(group->p2p, addr, 0, NULL, 0, ie, len, 0);
591 : :
592 : 193 : m = os_zalloc(sizeof(*m));
593 [ - + ]: 193 : if (m == NULL)
594 : 0 : return -1;
595 : 193 : os_memcpy(m->addr, addr, ETH_ALEN);
596 : 193 : m->p2p_ie = ieee802_11_vendor_ie_concat(ie, len, P2P_IE_VENDOR_TYPE);
597 [ + + ]: 193 : if (m->p2p_ie) {
598 : 172 : m->client_info = p2p_build_client_info(addr, m->p2p_ie,
599 : : &m->dev_capab,
600 : 172 : m->dev_addr);
601 : : }
602 : : #ifdef CONFIG_WIFI_DISPLAY
603 : 193 : m->wfd_ie = ieee802_11_vendor_ie_concat(ie, len, WFD_IE_VENDOR_TYPE);
604 : : #endif /* CONFIG_WIFI_DISPLAY */
605 : :
606 : 193 : p2p_group_remove_member(group, addr);
607 : :
608 : 193 : m->next = group->members;
609 : 193 : group->members = m;
610 : 193 : group->num_members++;
611 : 193 : p2p_dbg(group->p2p, "Add client " MACSTR
612 : : " to group (p2p=%d wfd=%d client_info=%d); num_members=%u/%u",
613 : 1544 : MAC2STR(addr), m->p2p_ie ? 1 : 0, m->wfd_ie ? 1 : 0,
614 : 193 : m->client_info ? 1 : 0,
615 : 193 : group->num_members, group->cfg->max_clients);
616 [ - + ]: 193 : if (group->num_members == group->cfg->max_clients)
617 : 0 : group->beacon_update = 1;
618 : 193 : p2p_group_update_ies(group);
619 [ + + ]: 193 : if (group->num_members == 1)
620 : 159 : group->cfg->idle_update(group->cfg->cb_ctx, 0);
621 : :
622 : 193 : return 0;
623 : : }
624 : :
625 : :
626 : 830 : struct wpabuf * p2p_group_assoc_resp_ie(struct p2p_group *group, u8 status)
627 : : {
628 : : struct wpabuf *resp;
629 : : u8 *rlen;
630 : 830 : size_t extra = 0;
631 : :
632 : : #ifdef CONFIG_WIFI_DISPLAY
633 [ + + ]: 830 : if (group->wfd_ie)
634 : 715 : 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 : 830 : resp = wpabuf_alloc(20 + extra);
644 [ - + ]: 830 : if (resp == NULL)
645 : 0 : return NULL;
646 : :
647 : : #ifdef CONFIG_WIFI_DISPLAY
648 [ + + ]: 830 : if (group->wfd_ie)
649 : 715 : wpabuf_put_buf(resp, group->wfd_ie);
650 : : #endif /* CONFIG_WIFI_DISPLAY */
651 : :
652 : 830 : rlen = p2p_buf_add_ie_hdr(resp);
653 [ - + ]: 830 : if (status != P2P_SC_SUCCESS)
654 : 0 : p2p_buf_add_status(resp, status);
655 : 830 : p2p_buf_update_ie_hdr(resp, rlen);
656 : :
657 : 830 : return resp;
658 : : }
659 : :
660 : :
661 : 187 : void p2p_group_notif_disassoc(struct p2p_group *group, const u8 *addr)
662 : : {
663 [ + + ]: 187 : if (p2p_group_remove_member(group, addr)) {
664 : 124 : p2p_dbg(group->p2p, "Remove client " MACSTR
665 : : " from group; num_members=%u/%u",
666 : 744 : MAC2STR(addr), group->num_members,
667 : 124 : group->cfg->max_clients);
668 [ - + ]: 124 : if (group->num_members == group->cfg->max_clients - 1)
669 : 0 : group->beacon_update = 1;
670 : 124 : p2p_group_update_ies(group);
671 [ + + ]: 124 : if (group->num_members == 0)
672 : 101 : group->cfg->idle_update(group->cfg->cb_ctx, 1);
673 : : }
674 : 187 : }
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 [ + + ]: 7 : 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 : 4 : return 0;
727 : : }
728 : :
729 : :
730 : 175 : int p2p_group_match_dev_type(struct p2p_group *group, struct wpabuf *wps)
731 : : {
732 : : struct p2p_group_member *m;
733 : :
734 [ + + ]: 175 : if (p2p_match_dev_type(group->p2p, wps))
735 : 171 : 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 : 175 : return 0;
744 : : }
745 : :
746 : :
747 : 171 : 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 : 171 : os_memset(&msg, 0, sizeof(msg));
753 [ - + ]: 171 : if (p2p_parse_p2p_ie(p2p, &msg))
754 : 0 : return 1; /* Failed to parse - assume no filter on Device ID */
755 : :
756 [ + + ]: 171 : if (!msg.device_id)
757 : 167 : 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 : 171 : return 0;
769 : : }
770 : :
771 : :
772 : 85 : void p2p_group_notif_formation_done(struct p2p_group *group)
773 : : {
774 [ - + ]: 85 : if (group == NULL)
775 : 85 : return;
776 : 85 : group->group_formation = 0;
777 : 85 : group->beacon_update = 1;
778 : 85 : 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 : 145 : 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 [ + - ]: 153 : for (m = group->members; m; m = m->next) {
832 [ + + ]: 153 : if (os_memcmp(interface_addr, m->addr, ETH_ALEN) == 0)
833 : 145 : return m;
834 : : }
835 : :
836 : 145 : return NULL;
837 : : }
838 : :
839 : :
840 : 144 : const u8 * p2p_group_get_dev_addr(struct p2p_group *group, const u8 *addr)
841 : : {
842 : : struct p2p_group_member *m;
843 : :
844 [ - + ]: 144 : if (group == NULL)
845 : 0 : return NULL;
846 : 144 : m = p2p_group_get_client_iface(group, addr);
847 [ + - ][ + + ]: 144 : if (m && !is_zero_ether_addr(m->dev_addr))
848 : 117 : return m->dev_addr;
849 : 144 : 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 : 3 : 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 : 3 : return 0;
909 : : }
910 : :
911 : :
912 : 8 : const u8 * p2p_group_get_interface_addr(struct p2p_group *group)
913 : : {
914 : 8 : 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 : 1 : 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 : 1 : 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 : 50 : int p2p_group_is_group_id_match(struct p2p_group *group, const u8 *group_id,
994 : : size_t group_id_len)
995 : : {
996 [ - + ]: 50 : if (group_id_len != ETH_ALEN + group->cfg->ssid_len)
997 : 0 : return 0;
998 [ - + ]: 50 : if (os_memcmp(group_id, group->p2p->cfg->dev_addr, ETH_ALEN) != 0)
999 : 0 : return 0;
1000 : 50 : return os_memcmp(group_id + ETH_ALEN, group->cfg->ssid,
1001 : : group->cfg->ssid_len) == 0;
1002 : : }
1003 : :
1004 : :
1005 : 0 : void p2p_group_force_beacon_update_ies(struct p2p_group *group)
1006 : : {
1007 : 0 : group->beacon_update = 1;
1008 : 0 : p2p_group_update_ies(group);
1009 : 0 : }
1010 : :
1011 : :
1012 : 5 : int p2p_group_get_freq(struct p2p_group *group)
1013 : : {
1014 : 5 : return group->cfg->freq;
1015 : : }
|
