2 * Wi-Fi Direct - P2P group operations
3 * Copyright (c) 2009-2010, Atheros Communications
5 * This software may be distributed under the terms of the BSD license.
6 * See README for more details.
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"
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 *client_info;
30 * struct p2p_group - Internal P2P module per-group data
34 struct p2p_group_config *cfg;
35 struct p2p_group_member *members;
36 unsigned int num_members;
43 static void p2p_group_update_ies(struct p2p_group *group);
46 struct p2p_group * p2p_group_init(struct p2p_data *p2p,
47 struct p2p_group_config *config)
49 struct p2p_group *group, **groups;
51 group = os_zalloc(sizeof(*group));
55 groups = os_realloc(p2p->groups, (p2p->num_groups + 1) *
56 sizeof(struct p2p_group *));
61 groups[p2p->num_groups++] = group;
66 group->group_formation = 1;
67 group->beacon_update = 1;
68 p2p_group_update_ies(group);
69 group->cfg->idle_update(group->cfg->cb_ctx, 1);
75 static void p2p_group_free_member(struct p2p_group_member *m)
77 wpabuf_free(m->p2p_ie);
78 wpabuf_free(m->client_info);
83 static void p2p_group_free_members(struct p2p_group *group)
85 struct p2p_group_member *m, *prev;
87 group->members = NULL;
88 group->num_members = 0;
92 p2p_group_free_member(prev);
97 void p2p_group_deinit(struct p2p_group *group)
100 struct p2p_data *p2p;
107 for (g = 0; g < p2p->num_groups; g++) {
108 if (p2p->groups[g] == group) {
109 while (g + 1 < p2p->num_groups) {
110 p2p->groups[g] = p2p->groups[g + 1];
118 p2p_group_free_members(group);
120 wpabuf_free(group->noa);
125 static void p2p_client_info(struct wpabuf *ie, struct p2p_group_member *m)
127 if (m->client_info == NULL)
129 if (wpabuf_tailroom(ie) < wpabuf_len(m->client_info) + 1)
131 wpabuf_put_buf(ie, m->client_info);
135 static void p2p_group_add_common_ies(struct p2p_group *group,
138 u8 dev_capab = group->p2p->dev_capab, group_capab = 0;
141 dev_capab &= ~P2P_DEV_CAPAB_CLIENT_DISCOVERABILITY;
142 group_capab |= P2P_GROUP_CAPAB_GROUP_OWNER;
143 if (group->cfg->persistent_group) {
144 group_capab |= P2P_GROUP_CAPAB_PERSISTENT_GROUP;
145 if (group->cfg->persistent_group == 2)
146 group_capab |= P2P_GROUP_CAPAB_PERSISTENT_RECONN;
148 if (group->p2p->cfg->p2p_intra_bss)
149 group_capab |= P2P_GROUP_CAPAB_INTRA_BSS_DIST;
150 if (group->group_formation)
151 group_capab |= P2P_GROUP_CAPAB_GROUP_FORMATION;
152 if (group->p2p->cross_connect)
153 group_capab |= P2P_GROUP_CAPAB_CROSS_CONN;
154 if (group->num_members >= group->cfg->max_clients)
155 group_capab |= P2P_GROUP_CAPAB_GROUP_LIMIT;
156 p2p_buf_add_capability(ie, dev_capab, group_capab);
160 static void p2p_group_add_noa(struct wpabuf *ie, struct wpabuf *noa)
164 /* Notice of Absence */
165 wpabuf_put_u8(ie, P2P_ATTR_NOTICE_OF_ABSENCE);
166 wpabuf_put_le16(ie, wpabuf_len(noa));
167 wpabuf_put_buf(ie, noa);
171 static struct wpabuf * p2p_group_build_beacon_ie(struct p2p_group *group)
176 ie = wpabuf_alloc(257);
180 len = p2p_buf_add_ie_hdr(ie);
181 p2p_group_add_common_ies(group, ie);
182 p2p_buf_add_device_id(ie, group->p2p->cfg->dev_addr);
183 p2p_group_add_noa(ie, group->noa);
184 p2p_buf_update_ie_hdr(ie, len);
190 static struct wpabuf * p2p_group_build_probe_resp_ie(struct p2p_group *group)
194 struct p2p_group_member *m;
197 ie = wpabuf_alloc(257);
201 len = p2p_buf_add_ie_hdr(ie);
203 p2p_group_add_common_ies(group, ie);
204 p2p_group_add_noa(ie, group->noa);
206 /* P2P Device Info */
207 p2p_buf_add_device_info(ie, group->p2p, NULL);
210 group_info = wpabuf_put(ie, 0);
211 wpabuf_put_u8(ie, P2P_ATTR_GROUP_INFO);
212 wpabuf_put_le16(ie, 0); /* Length to be filled */
213 for (m = group->members; m; m = m->next)
214 p2p_client_info(ie, m);
215 WPA_PUT_LE16(group_info + 1,
216 (u8 *) wpabuf_put(ie, 0) - group_info - 3);
218 p2p_buf_update_ie_hdr(ie, len);
223 static void p2p_group_update_ies(struct p2p_group *group)
225 struct wpabuf *beacon_ie;
226 struct wpabuf *probe_resp_ie;
228 probe_resp_ie = p2p_group_build_probe_resp_ie(group);
229 if (probe_resp_ie == NULL)
231 wpa_hexdump_buf(MSG_MSGDUMP, "P2P: Update GO Probe Response P2P IE",
234 if (group->beacon_update) {
235 beacon_ie = p2p_group_build_beacon_ie(group);
237 group->beacon_update = 0;
238 wpa_hexdump_buf(MSG_MSGDUMP, "P2P: Update GO Beacon P2P IE",
243 group->cfg->ie_update(group->cfg->cb_ctx, beacon_ie, probe_resp_ie);
248 * p2p_build_client_info - Build P2P Client Info Descriptor
249 * @addr: MAC address of the peer device
250 * @p2p_ie: P2P IE from (Re)Association Request
251 * @dev_capab: Buffer for returning Device Capability
252 * @dev_addr: Buffer for returning P2P Device Address
253 * Returns: P2P Client Info Descriptor or %NULL on failure
255 * This function builds P2P Client Info Descriptor based on the information
256 * available from (Re)Association Request frame. Group owner can use this to
257 * build the P2P Group Info attribute for Probe Response frames.
259 static struct wpabuf * p2p_build_client_info(const u8 *addr,
260 struct wpabuf *p2p_ie,
261 u8 *dev_capab, u8 *dev_addr)
264 struct p2p_message msg;
271 os_memset(&msg, 0, sizeof(msg));
272 if (p2p_parse_p2p_ie(p2p_ie, &msg) ||
273 msg.capability == NULL || msg.p2p_device_info == NULL)
276 buf = wpabuf_alloc(ETH_ALEN + 1 + 1 + msg.p2p_device_info_len);
280 *dev_capab = msg.capability[0];
281 os_memcpy(dev_addr, msg.p2p_device_addr, ETH_ALEN);
283 spos = msg.p2p_device_info; /* P2P Device address */
285 /* P2P Client Info Descriptor */
286 /* Length to be set */
287 len_pos = wpabuf_put(buf, 1);
288 /* P2P Device address */
289 wpabuf_put_data(buf, spos, ETH_ALEN);
290 /* P2P Interface address */
291 wpabuf_put_data(buf, addr, ETH_ALEN);
292 /* Device Capability Bitmap */
293 wpabuf_put_u8(buf, msg.capability[0]);
295 * Config Methods, Primary Device Type, Number of Secondary Device
296 * Types, Secondary Device Type List, Device Name copied from
299 wpabuf_put_data(buf, spos + ETH_ALEN,
300 msg.p2p_device_info_len - ETH_ALEN);
302 *len_pos = wpabuf_len(buf) - 1;
309 static int p2p_group_remove_member(struct p2p_group *group, const u8 *addr)
311 struct p2p_group_member *m, *prev;
319 if (os_memcmp(m->addr, addr, ETH_ALEN) == 0)
329 prev->next = m->next;
331 group->members = m->next;
332 p2p_group_free_member(m);
333 group->num_members--;
339 int p2p_group_notif_assoc(struct p2p_group *group, const u8 *addr,
340 const u8 *ie, size_t len)
342 struct p2p_group_member *m;
347 m = os_zalloc(sizeof(*m));
350 os_memcpy(m->addr, addr, ETH_ALEN);
351 m->p2p_ie = ieee802_11_vendor_ie_concat(ie, len, P2P_IE_VENDOR_TYPE);
353 m->client_info = p2p_build_client_info(addr, m->p2p_ie,
358 p2p_group_remove_member(group, addr);
360 m->next = group->members;
362 group->num_members++;
363 wpa_msg(group->p2p->cfg->msg_ctx, MSG_DEBUG, "P2P: Add client " MACSTR
364 " to group (p2p=%d client_info=%d); num_members=%u/%u",
365 MAC2STR(addr), m->p2p_ie ? 1 : 0, m->client_info ? 1 : 0,
366 group->num_members, group->cfg->max_clients);
367 if (group->num_members == group->cfg->max_clients)
368 group->beacon_update = 1;
369 p2p_group_update_ies(group);
370 if (group->num_members == 1)
371 group->cfg->idle_update(group->cfg->cb_ctx, 0);
377 struct wpabuf * p2p_group_assoc_resp_ie(struct p2p_group *group, u8 status)
383 * (Re)Association Response - P2P IE
384 * Status attribute (shall be present when association request is
386 * Extended Listen Timing (may be present)
388 resp = wpabuf_alloc(20);
391 rlen = p2p_buf_add_ie_hdr(resp);
392 if (status != P2P_SC_SUCCESS)
393 p2p_buf_add_status(resp, status);
394 p2p_buf_update_ie_hdr(resp, rlen);
400 void p2p_group_notif_disassoc(struct p2p_group *group, const u8 *addr)
402 if (p2p_group_remove_member(group, addr)) {
403 wpa_msg(group->p2p->cfg->msg_ctx, MSG_DEBUG, "P2P: Remove "
404 "client " MACSTR " from group; num_members=%u/%u",
405 MAC2STR(addr), group->num_members,
406 group->cfg->max_clients);
407 if (group->num_members == group->cfg->max_clients - 1)
408 group->beacon_update = 1;
409 p2p_group_update_ies(group);
410 if (group->num_members == 0)
411 group->cfg->idle_update(group->cfg->cb_ctx, 1);
417 * p2p_match_dev_type_member - Match client device type with requested type
419 * @wps: WPS TLVs from Probe Request frame (concatenated WPS IEs)
420 * Returns: 1 on match, 0 on mismatch
422 * This function can be used to match the Requested Device Type attribute in
423 * WPS IE with the device types of a group member for deciding whether a GO
424 * should reply to a Probe Request frame.
426 static int p2p_match_dev_type_member(struct p2p_group_member *m,
430 struct wps_parse_attr attr;
433 if (m->client_info == NULL || wps == NULL)
436 pos = wpabuf_head(m->client_info);
437 end = pos + wpabuf_len(m->client_info);
439 pos += 1 + 2 * ETH_ALEN + 1 + 2;
440 if (end - pos < WPS_DEV_TYPE_LEN + 1)
443 if (wps_parse_msg(wps, &attr))
444 return 1; /* assume no Requested Device Type attributes */
446 if (attr.num_req_dev_type == 0)
447 return 1; /* no Requested Device Type attributes -> match */
449 if (dev_type_list_match(pos, attr.req_dev_type, attr.num_req_dev_type))
450 return 1; /* Match with client Primary Device Type */
452 pos += WPS_DEV_TYPE_LEN;
454 if (end - pos < num_sec * WPS_DEV_TYPE_LEN)
456 while (num_sec > 0) {
458 if (dev_type_list_match(pos, attr.req_dev_type,
459 attr.num_req_dev_type))
460 return 1; /* Match with client Secondary Device Type */
461 pos += WPS_DEV_TYPE_LEN;
464 /* No matching device type found */
469 int p2p_group_match_dev_type(struct p2p_group *group, struct wpabuf *wps)
471 struct p2p_group_member *m;
473 if (p2p_match_dev_type(group->p2p, wps))
474 return 1; /* Match with own device type */
476 for (m = group->members; m; m = m->next) {
477 if (p2p_match_dev_type_member(m, wps))
478 return 1; /* Match with group client device type */
481 /* No match with Requested Device Type */
486 int p2p_group_match_dev_id(struct p2p_group *group, struct wpabuf *p2p)
488 struct p2p_group_member *m;
489 struct p2p_message msg;
491 os_memset(&msg, 0, sizeof(msg));
492 if (p2p_parse_p2p_ie(p2p, &msg))
493 return 1; /* Failed to parse - assume no filter on Device ID */
496 return 1; /* No filter on Device ID */
498 if (os_memcmp(msg.device_id, group->p2p->cfg->dev_addr, ETH_ALEN) == 0)
499 return 1; /* Match with our P2P Device Address */
501 for (m = group->members; m; m = m->next) {
502 if (os_memcmp(msg.device_id, m->dev_addr, ETH_ALEN) == 0)
503 return 1; /* Match with group client P2P Device Address */
506 /* No match with Device ID */
511 void p2p_group_notif_formation_done(struct p2p_group *group)
515 group->group_formation = 0;
516 group->beacon_update = 1;
517 p2p_group_update_ies(group);
521 int p2p_group_notif_noa(struct p2p_group *group, const u8 *noa,
525 wpabuf_free(group->noa);
529 if (wpabuf_size(group->noa) >= noa_len) {
530 group->noa->used = 0;
531 wpabuf_put_data(group->noa, noa, noa_len);
533 wpabuf_free(group->noa);
539 group->noa = wpabuf_alloc_copy(noa, noa_len);
540 if (group->noa == NULL)
545 group->beacon_update = 1;
546 p2p_group_update_ies(group);
551 static struct p2p_group_member * p2p_group_get_client(struct p2p_group *group,
554 struct p2p_group_member *m;
556 for (m = group->members; m; m = m->next) {
557 if (os_memcmp(dev_id, m->dev_addr, ETH_ALEN) == 0)
565 static struct p2p_group_member * p2p_group_get_client_iface(
566 struct p2p_group *group, const u8 *interface_addr)
568 struct p2p_group_member *m;
570 for (m = group->members; m; m = m->next) {
571 if (os_memcmp(interface_addr, m->addr, ETH_ALEN) == 0)
579 const u8 * p2p_group_get_dev_addr(struct p2p_group *group, const u8 *addr)
581 struct p2p_group_member *m;
585 m = p2p_group_get_client_iface(group, addr);
586 if (m && !is_zero_ether_addr(m->dev_addr))
592 static struct wpabuf * p2p_build_go_disc_req(void)
596 buf = wpabuf_alloc(100);
600 p2p_buf_add_action_hdr(buf, P2P_GO_DISC_REQ, 0);
606 int p2p_group_go_discover(struct p2p_group *group, const u8 *dev_id,
607 const u8 *searching_dev, int rx_freq)
609 struct p2p_group_member *m;
611 struct p2p_data *p2p = group->p2p;
614 m = p2p_group_get_client(group, dev_id);
615 if (m == NULL || m->client_info == NULL) {
616 wpa_printf(MSG_DEBUG, "P2P: Requested client was not in this "
618 MAC2STR(group->cfg->interface_addr));
622 if (!(m->dev_capab & P2P_DEV_CAPAB_CLIENT_DISCOVERABILITY)) {
623 wpa_printf(MSG_DEBUG, "P2P: Requested client does not support "
624 "client discoverability");
628 wpa_printf(MSG_DEBUG, "P2P: Schedule GO Discoverability Request to be "
629 "sent to " MACSTR, MAC2STR(dev_id));
631 req = p2p_build_go_disc_req();
635 /* TODO: Should really use group operating frequency here */
638 p2p->pending_action_state = P2P_PENDING_GO_DISC_REQ;
639 if (p2p->cfg->send_action(p2p->cfg->cb_ctx, freq, m->addr,
640 group->cfg->interface_addr,
641 group->cfg->interface_addr,
642 wpabuf_head(req), wpabuf_len(req), 200) < 0)
644 wpa_msg(p2p->cfg->msg_ctx, MSG_DEBUG,
645 "P2P: Failed to send Action frame");
654 const u8 * p2p_group_get_interface_addr(struct p2p_group *group)
656 return group->cfg->interface_addr;
660 u8 p2p_group_presence_req(struct p2p_group *group,
661 const u8 *client_interface_addr,
662 const u8 *noa, size_t noa_len)
664 struct p2p_group_member *m;
668 m = p2p_group_get_client_iface(group, client_interface_addr);
669 if (m == NULL || m->client_info == NULL) {
670 wpa_printf(MSG_DEBUG, "P2P: Client was not in this group");
671 return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE;
674 wpa_hexdump(MSG_DEBUG, "P2P: Presence Request NoA", noa, noa_len);
676 if (group->p2p->cfg->get_noa)
677 curr_noa_len = group->p2p->cfg->get_noa(
678 group->p2p->cfg->cb_ctx, group->cfg->interface_addr,
679 curr_noa, sizeof(curr_noa));
682 if (curr_noa_len < 0)
683 wpa_printf(MSG_DEBUG, "P2P: Failed to fetch current NoA");
684 else if (curr_noa_len == 0)
685 wpa_printf(MSG_DEBUG, "P2P: No NoA being advertized");
687 wpa_hexdump(MSG_DEBUG, "P2P: Current NoA", curr_noa,
690 /* TODO: properly process request and store copy */
691 if (curr_noa_len > 0 || curr_noa_len == -1)
692 return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE;
694 return P2P_SC_SUCCESS;
698 unsigned int p2p_get_group_num_members(struct p2p_group *group)
700 return group->num_members;
704 const u8 * p2p_iterate_group_members(struct p2p_group *group, void **next)
706 struct p2p_group_member *iter = *next;
709 iter = group->members;
722 int p2p_group_is_client_connected(struct p2p_group *group, const u8 *dev_addr)
724 struct p2p_group_member *m;
726 for (m = group->members; m; m = m->next) {
727 if (os_memcmp(m->dev_addr, dev_addr, ETH_ALEN) == 0)
735 int p2p_group_is_group_id_match(struct p2p_group *group, const u8 *group_id,
738 if (group_id_len != ETH_ALEN + group->cfg->ssid_len)
740 if (os_memcmp(group_id, group->p2p->cfg->dev_addr, ETH_ALEN) != 0)
742 return os_memcmp(group_id + ETH_ALEN, group->cfg->ssid,
743 group->cfg->ssid_len) == 0;