1252190Srpaulo/* 2252190Srpaulo * Wi-Fi Direct - P2P group operations 3252190Srpaulo * Copyright (c) 2009-2010, Atheros Communications 4252190Srpaulo * 5252190Srpaulo * This software may be distributed under the terms of the BSD license. 6252190Srpaulo * See README for more details. 7252190Srpaulo */ 8252190Srpaulo 9252190Srpaulo#include "includes.h" 10252190Srpaulo 11252190Srpaulo#include "common.h" 12252190Srpaulo#include "common/ieee802_11_defs.h" 13252190Srpaulo#include "common/ieee802_11_common.h" 14252190Srpaulo#include "wps/wps_defs.h" 15252190Srpaulo#include "wps/wps_i.h" 16252190Srpaulo#include "p2p_i.h" 17252190Srpaulo#include "p2p.h" 18252190Srpaulo 19252190Srpaulo 20252190Srpaulostruct p2p_group_member { 21252190Srpaulo struct p2p_group_member *next; 22252190Srpaulo u8 addr[ETH_ALEN]; /* P2P Interface Address */ 23252190Srpaulo u8 dev_addr[ETH_ALEN]; /* P2P Device Address */ 24252190Srpaulo struct wpabuf *p2p_ie; 25252190Srpaulo struct wpabuf *wfd_ie; 26252190Srpaulo struct wpabuf *client_info; 27252190Srpaulo u8 dev_capab; 28252190Srpaulo}; 29252190Srpaulo 30252190Srpaulo/** 31252190Srpaulo * struct p2p_group - Internal P2P module per-group data 32252190Srpaulo */ 33252190Srpaulostruct p2p_group { 34252190Srpaulo struct p2p_data *p2p; 35252190Srpaulo struct p2p_group_config *cfg; 36252190Srpaulo struct p2p_group_member *members; 37252190Srpaulo unsigned int num_members; 38252190Srpaulo int group_formation; 39252190Srpaulo int beacon_update; 40252190Srpaulo struct wpabuf *noa; 41252190Srpaulo struct wpabuf *wfd_ie; 42252190Srpaulo}; 43252190Srpaulo 44252190Srpaulo 45252190Srpaulostruct p2p_group * p2p_group_init(struct p2p_data *p2p, 46252190Srpaulo struct p2p_group_config *config) 47252190Srpaulo{ 48252190Srpaulo struct p2p_group *group, **groups; 49252190Srpaulo 50252190Srpaulo group = os_zalloc(sizeof(*group)); 51252190Srpaulo if (group == NULL) 52252190Srpaulo return NULL; 53252190Srpaulo 54252190Srpaulo groups = os_realloc_array(p2p->groups, p2p->num_groups + 1, 55252190Srpaulo sizeof(struct p2p_group *)); 56252190Srpaulo if (groups == NULL) { 57252190Srpaulo os_free(group); 58252190Srpaulo return NULL; 59252190Srpaulo } 60252190Srpaulo groups[p2p->num_groups++] = group; 61252190Srpaulo p2p->groups = groups; 62252190Srpaulo 63252190Srpaulo group->p2p = p2p; 64252190Srpaulo group->cfg = config; 65252190Srpaulo group->group_formation = 1; 66252190Srpaulo group->beacon_update = 1; 67252190Srpaulo p2p_group_update_ies(group); 68252190Srpaulo group->cfg->idle_update(group->cfg->cb_ctx, 1); 69252190Srpaulo 70252190Srpaulo return group; 71252190Srpaulo} 72252190Srpaulo 73252190Srpaulo 74252190Srpaulostatic void p2p_group_free_member(struct p2p_group_member *m) 75252190Srpaulo{ 76252190Srpaulo wpabuf_free(m->wfd_ie); 77252190Srpaulo wpabuf_free(m->p2p_ie); 78252190Srpaulo wpabuf_free(m->client_info); 79252190Srpaulo os_free(m); 80252190Srpaulo} 81252190Srpaulo 82252190Srpaulo 83252190Srpaulostatic void p2p_group_free_members(struct p2p_group *group) 84252190Srpaulo{ 85252190Srpaulo struct p2p_group_member *m, *prev; 86252190Srpaulo m = group->members; 87252190Srpaulo group->members = NULL; 88252190Srpaulo group->num_members = 0; 89252190Srpaulo while (m) { 90252190Srpaulo prev = m; 91252190Srpaulo m = m->next; 92252190Srpaulo p2p_group_free_member(prev); 93252190Srpaulo } 94252190Srpaulo} 95252190Srpaulo 96252190Srpaulo 97252190Srpaulovoid p2p_group_deinit(struct p2p_group *group) 98252190Srpaulo{ 99252190Srpaulo size_t g; 100252190Srpaulo struct p2p_data *p2p; 101252190Srpaulo 102252190Srpaulo if (group == NULL) 103252190Srpaulo return; 104252190Srpaulo 105252190Srpaulo p2p = group->p2p; 106252190Srpaulo 107252190Srpaulo for (g = 0; g < p2p->num_groups; g++) { 108252190Srpaulo if (p2p->groups[g] == group) { 109252190Srpaulo while (g + 1 < p2p->num_groups) { 110252190Srpaulo p2p->groups[g] = p2p->groups[g + 1]; 111252190Srpaulo g++; 112252190Srpaulo } 113252190Srpaulo p2p->num_groups--; 114252190Srpaulo break; 115252190Srpaulo } 116252190Srpaulo } 117252190Srpaulo 118252190Srpaulo p2p_group_free_members(group); 119252190Srpaulo os_free(group->cfg); 120252190Srpaulo wpabuf_free(group->noa); 121252190Srpaulo wpabuf_free(group->wfd_ie); 122252190Srpaulo os_free(group); 123252190Srpaulo} 124252190Srpaulo 125252190Srpaulo 126252190Srpaulostatic void p2p_client_info(struct wpabuf *ie, struct p2p_group_member *m) 127252190Srpaulo{ 128252190Srpaulo if (m->client_info == NULL) 129252190Srpaulo return; 130252190Srpaulo if (wpabuf_tailroom(ie) < wpabuf_len(m->client_info) + 1) 131252190Srpaulo return; 132252190Srpaulo wpabuf_put_buf(ie, m->client_info); 133252190Srpaulo} 134252190Srpaulo 135252190Srpaulo 136252190Srpaulostatic void p2p_group_add_common_ies(struct p2p_group *group, 137252190Srpaulo struct wpabuf *ie) 138252190Srpaulo{ 139252190Srpaulo u8 dev_capab = group->p2p->dev_capab, group_capab = 0; 140252190Srpaulo 141252190Srpaulo /* P2P Capability */ 142252190Srpaulo dev_capab &= ~P2P_DEV_CAPAB_CLIENT_DISCOVERABILITY; 143252190Srpaulo group_capab |= P2P_GROUP_CAPAB_GROUP_OWNER; 144252190Srpaulo if (group->cfg->persistent_group) { 145252190Srpaulo group_capab |= P2P_GROUP_CAPAB_PERSISTENT_GROUP; 146252190Srpaulo if (group->cfg->persistent_group == 2) 147252190Srpaulo group_capab |= P2P_GROUP_CAPAB_PERSISTENT_RECONN; 148252190Srpaulo } 149252190Srpaulo if (group->p2p->cfg->p2p_intra_bss) 150252190Srpaulo group_capab |= P2P_GROUP_CAPAB_INTRA_BSS_DIST; 151252190Srpaulo if (group->group_formation) 152252190Srpaulo group_capab |= P2P_GROUP_CAPAB_GROUP_FORMATION; 153252190Srpaulo if (group->p2p->cross_connect) 154252190Srpaulo group_capab |= P2P_GROUP_CAPAB_CROSS_CONN; 155252190Srpaulo if (group->num_members >= group->cfg->max_clients) 156252190Srpaulo group_capab |= P2P_GROUP_CAPAB_GROUP_LIMIT; 157252190Srpaulo p2p_buf_add_capability(ie, dev_capab, group_capab); 158252190Srpaulo} 159252190Srpaulo 160252190Srpaulo 161252190Srpaulostatic void p2p_group_add_noa(struct wpabuf *ie, struct wpabuf *noa) 162252190Srpaulo{ 163252190Srpaulo if (noa == NULL) 164252190Srpaulo return; 165252190Srpaulo /* Notice of Absence */ 166252190Srpaulo wpabuf_put_u8(ie, P2P_ATTR_NOTICE_OF_ABSENCE); 167252190Srpaulo wpabuf_put_le16(ie, wpabuf_len(noa)); 168252190Srpaulo wpabuf_put_buf(ie, noa); 169252190Srpaulo} 170252190Srpaulo 171252190Srpaulo 172252190Srpaulostatic struct wpabuf * p2p_group_build_beacon_ie(struct p2p_group *group) 173252190Srpaulo{ 174252190Srpaulo struct wpabuf *ie; 175252190Srpaulo u8 *len; 176252190Srpaulo size_t extra = 0; 177252190Srpaulo 178252190Srpaulo#ifdef CONFIG_WIFI_DISPLAY 179252190Srpaulo if (group->p2p->wfd_ie_beacon) 180252190Srpaulo extra = wpabuf_len(group->p2p->wfd_ie_beacon); 181252190Srpaulo#endif /* CONFIG_WIFI_DISPLAY */ 182252190Srpaulo 183252190Srpaulo ie = wpabuf_alloc(257 + extra); 184252190Srpaulo if (ie == NULL) 185252190Srpaulo return NULL; 186252190Srpaulo 187252190Srpaulo#ifdef CONFIG_WIFI_DISPLAY 188252190Srpaulo if (group->p2p->wfd_ie_beacon) 189252190Srpaulo wpabuf_put_buf(ie, group->p2p->wfd_ie_beacon); 190252190Srpaulo#endif /* CONFIG_WIFI_DISPLAY */ 191252190Srpaulo 192252190Srpaulo len = p2p_buf_add_ie_hdr(ie); 193252190Srpaulo p2p_group_add_common_ies(group, ie); 194252190Srpaulo p2p_buf_add_device_id(ie, group->p2p->cfg->dev_addr); 195252190Srpaulo p2p_group_add_noa(ie, group->noa); 196252190Srpaulo p2p_buf_update_ie_hdr(ie, len); 197252190Srpaulo 198252190Srpaulo return ie; 199252190Srpaulo} 200252190Srpaulo 201252190Srpaulo 202252190Srpaulo#ifdef CONFIG_WIFI_DISPLAY 203252190Srpaulo 204252190Srpaulostruct wpabuf * p2p_group_get_wfd_ie(struct p2p_group *g) 205252190Srpaulo{ 206252190Srpaulo return g->wfd_ie; 207252190Srpaulo} 208252190Srpaulo 209252190Srpaulo 210252190Srpaulostruct wpabuf * wifi_display_encaps(struct wpabuf *subelems) 211252190Srpaulo{ 212252190Srpaulo struct wpabuf *ie; 213252190Srpaulo const u8 *pos, *end; 214252190Srpaulo 215252190Srpaulo if (subelems == NULL) 216252190Srpaulo return NULL; 217252190Srpaulo 218252190Srpaulo ie = wpabuf_alloc(wpabuf_len(subelems) + 100); 219252190Srpaulo if (ie == NULL) 220252190Srpaulo return NULL; 221252190Srpaulo 222252190Srpaulo pos = wpabuf_head(subelems); 223252190Srpaulo end = pos + wpabuf_len(subelems); 224252190Srpaulo 225252190Srpaulo while (end > pos) { 226252190Srpaulo size_t frag_len = end - pos; 227252190Srpaulo if (frag_len > 251) 228252190Srpaulo frag_len = 251; 229252190Srpaulo wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC); 230252190Srpaulo wpabuf_put_u8(ie, 4 + frag_len); 231252190Srpaulo wpabuf_put_be32(ie, WFD_IE_VENDOR_TYPE); 232252190Srpaulo wpabuf_put_data(ie, pos, frag_len); 233252190Srpaulo pos += frag_len; 234252190Srpaulo } 235252190Srpaulo 236252190Srpaulo return ie; 237252190Srpaulo} 238252190Srpaulo 239252190Srpaulo 240252190Srpaulostatic int wifi_display_add_dev_info_descr(struct wpabuf *buf, 241252190Srpaulo struct p2p_group_member *m) 242252190Srpaulo{ 243252190Srpaulo const u8 *pos, *end; 244252190Srpaulo const u8 *dev_info = NULL; 245252190Srpaulo const u8 *assoc_bssid = NULL; 246252190Srpaulo const u8 *coupled_sink = NULL; 247252190Srpaulo u8 zero_addr[ETH_ALEN]; 248252190Srpaulo 249252190Srpaulo if (m->wfd_ie == NULL) 250252190Srpaulo return 0; 251252190Srpaulo 252252190Srpaulo os_memset(zero_addr, 0, ETH_ALEN); 253252190Srpaulo pos = wpabuf_head_u8(m->wfd_ie); 254252190Srpaulo end = pos + wpabuf_len(m->wfd_ie); 255252190Srpaulo while (pos + 1 < end) { 256252190Srpaulo u8 id; 257252190Srpaulo u16 len; 258252190Srpaulo 259252190Srpaulo id = *pos++; 260252190Srpaulo len = WPA_GET_BE16(pos); 261252190Srpaulo pos += 2; 262252190Srpaulo if (pos + len > end) 263252190Srpaulo break; 264252190Srpaulo 265252190Srpaulo switch (id) { 266252190Srpaulo case WFD_SUBELEM_DEVICE_INFO: 267252190Srpaulo if (len < 6) 268252190Srpaulo break; 269252190Srpaulo dev_info = pos; 270252190Srpaulo break; 271252190Srpaulo case WFD_SUBELEM_ASSOCIATED_BSSID: 272252190Srpaulo if (len < ETH_ALEN) 273252190Srpaulo break; 274252190Srpaulo assoc_bssid = pos; 275252190Srpaulo break; 276252190Srpaulo case WFD_SUBELEM_COUPLED_SINK: 277252190Srpaulo if (len < 1 + ETH_ALEN) 278252190Srpaulo break; 279252190Srpaulo coupled_sink = pos; 280252190Srpaulo break; 281252190Srpaulo } 282252190Srpaulo 283252190Srpaulo pos += len; 284252190Srpaulo } 285252190Srpaulo 286252190Srpaulo if (dev_info == NULL) 287252190Srpaulo return 0; 288252190Srpaulo 289252190Srpaulo wpabuf_put_u8(buf, 23); 290252190Srpaulo wpabuf_put_data(buf, m->dev_addr, ETH_ALEN); 291252190Srpaulo if (assoc_bssid) 292252190Srpaulo wpabuf_put_data(buf, assoc_bssid, ETH_ALEN); 293252190Srpaulo else 294252190Srpaulo wpabuf_put_data(buf, zero_addr, ETH_ALEN); 295252190Srpaulo wpabuf_put_data(buf, dev_info, 2); /* WFD Device Info */ 296252190Srpaulo wpabuf_put_data(buf, dev_info + 4, 2); /* WFD Device Max Throughput */ 297252190Srpaulo if (coupled_sink) { 298252190Srpaulo wpabuf_put_data(buf, coupled_sink, 1 + ETH_ALEN); 299252190Srpaulo } else { 300252190Srpaulo wpabuf_put_u8(buf, 0); 301252190Srpaulo wpabuf_put_data(buf, zero_addr, ETH_ALEN); 302252190Srpaulo } 303252190Srpaulo 304252190Srpaulo return 1; 305252190Srpaulo} 306252190Srpaulo 307252190Srpaulo 308252190Srpaulostatic struct wpabuf * 309252190Srpaulowifi_display_build_go_ie(struct p2p_group *group) 310252190Srpaulo{ 311252190Srpaulo struct wpabuf *wfd_subelems, *wfd_ie; 312252190Srpaulo struct p2p_group_member *m; 313252190Srpaulo u8 *len; 314252190Srpaulo unsigned int count = 0; 315252190Srpaulo 316252190Srpaulo if (!group->p2p->wfd_ie_probe_resp) 317252190Srpaulo return NULL; 318252190Srpaulo 319252190Srpaulo wfd_subelems = wpabuf_alloc(wpabuf_len(group->p2p->wfd_ie_probe_resp) + 320252190Srpaulo group->num_members * 24 + 100); 321252190Srpaulo if (wfd_subelems == NULL) 322252190Srpaulo return NULL; 323252190Srpaulo if (group->p2p->wfd_dev_info) 324252190Srpaulo wpabuf_put_buf(wfd_subelems, group->p2p->wfd_dev_info); 325252190Srpaulo if (group->p2p->wfd_assoc_bssid) 326252190Srpaulo wpabuf_put_buf(wfd_subelems, 327252190Srpaulo group->p2p->wfd_assoc_bssid); 328252190Srpaulo if (group->p2p->wfd_coupled_sink_info) 329252190Srpaulo wpabuf_put_buf(wfd_subelems, 330252190Srpaulo group->p2p->wfd_coupled_sink_info); 331252190Srpaulo 332252190Srpaulo /* Build WFD Session Info */ 333252190Srpaulo wpabuf_put_u8(wfd_subelems, WFD_SUBELEM_SESSION_INFO); 334252190Srpaulo len = wpabuf_put(wfd_subelems, 2); 335252190Srpaulo m = group->members; 336252190Srpaulo while (m) { 337252190Srpaulo if (wifi_display_add_dev_info_descr(wfd_subelems, m)) 338252190Srpaulo count++; 339252190Srpaulo m = m->next; 340252190Srpaulo } 341252190Srpaulo 342252190Srpaulo if (count == 0) { 343252190Srpaulo /* No Wi-Fi Display clients - do not include subelement */ 344252190Srpaulo wfd_subelems->used -= 3; 345252190Srpaulo } else { 346252190Srpaulo WPA_PUT_BE16(len, (u8 *) wpabuf_put(wfd_subelems, 0) - len - 347252190Srpaulo 2); 348252190Srpaulo wpa_printf(MSG_DEBUG, "WFD: WFD Session Info: %u descriptors", 349252190Srpaulo count); 350252190Srpaulo } 351252190Srpaulo 352252190Srpaulo wfd_ie = wifi_display_encaps(wfd_subelems); 353252190Srpaulo wpabuf_free(wfd_subelems); 354252190Srpaulo 355252190Srpaulo return wfd_ie; 356252190Srpaulo} 357252190Srpaulo 358252190Srpaulostatic void wifi_display_group_update(struct p2p_group *group) 359252190Srpaulo{ 360252190Srpaulo wpabuf_free(group->wfd_ie); 361252190Srpaulo group->wfd_ie = wifi_display_build_go_ie(group); 362252190Srpaulo} 363252190Srpaulo 364252190Srpaulo#endif /* CONFIG_WIFI_DISPLAY */ 365252190Srpaulo 366252190Srpaulo 367252190Srpaulostatic struct wpabuf * p2p_group_build_probe_resp_ie(struct p2p_group *group) 368252190Srpaulo{ 369252190Srpaulo u8 *group_info; 370252190Srpaulo struct wpabuf *ie; 371252190Srpaulo struct p2p_group_member *m; 372252190Srpaulo u8 *len; 373252190Srpaulo size_t extra = 0; 374252190Srpaulo 375252190Srpaulo#ifdef CONFIG_WIFI_DISPLAY 376252190Srpaulo if (group->wfd_ie) 377252190Srpaulo extra += wpabuf_len(group->wfd_ie); 378252190Srpaulo#endif /* CONFIG_WIFI_DISPLAY */ 379252190Srpaulo 380252190Srpaulo ie = wpabuf_alloc(257 + extra); 381252190Srpaulo if (ie == NULL) 382252190Srpaulo return NULL; 383252190Srpaulo 384252190Srpaulo#ifdef CONFIG_WIFI_DISPLAY 385252190Srpaulo if (group->wfd_ie) 386252190Srpaulo wpabuf_put_buf(ie, group->wfd_ie); 387252190Srpaulo#endif /* CONFIG_WIFI_DISPLAY */ 388252190Srpaulo 389252190Srpaulo len = p2p_buf_add_ie_hdr(ie); 390252190Srpaulo 391252190Srpaulo p2p_group_add_common_ies(group, ie); 392252190Srpaulo p2p_group_add_noa(ie, group->noa); 393252190Srpaulo 394252190Srpaulo /* P2P Device Info */ 395252190Srpaulo p2p_buf_add_device_info(ie, group->p2p, NULL); 396252190Srpaulo 397252190Srpaulo /* P2P Group Info */ 398252190Srpaulo group_info = wpabuf_put(ie, 0); 399252190Srpaulo wpabuf_put_u8(ie, P2P_ATTR_GROUP_INFO); 400252190Srpaulo wpabuf_put_le16(ie, 0); /* Length to be filled */ 401252190Srpaulo for (m = group->members; m; m = m->next) 402252190Srpaulo p2p_client_info(ie, m); 403252190Srpaulo WPA_PUT_LE16(group_info + 1, 404252190Srpaulo (u8 *) wpabuf_put(ie, 0) - group_info - 3); 405252190Srpaulo 406252190Srpaulo p2p_buf_update_ie_hdr(ie, len); 407252190Srpaulo 408252190Srpaulo return ie; 409252190Srpaulo} 410252190Srpaulo 411252190Srpaulo 412252190Srpaulovoid p2p_group_update_ies(struct p2p_group *group) 413252190Srpaulo{ 414252190Srpaulo struct wpabuf *beacon_ie; 415252190Srpaulo struct wpabuf *probe_resp_ie; 416252190Srpaulo 417252190Srpaulo#ifdef CONFIG_WIFI_DISPLAY 418252190Srpaulo wifi_display_group_update(group); 419252190Srpaulo#endif /* CONFIG_WIFI_DISPLAY */ 420252190Srpaulo 421252190Srpaulo probe_resp_ie = p2p_group_build_probe_resp_ie(group); 422252190Srpaulo if (probe_resp_ie == NULL) 423252190Srpaulo return; 424252190Srpaulo wpa_hexdump_buf(MSG_MSGDUMP, "P2P: Update GO Probe Response P2P IE", 425252190Srpaulo probe_resp_ie); 426252190Srpaulo 427252190Srpaulo if (group->beacon_update) { 428252190Srpaulo beacon_ie = p2p_group_build_beacon_ie(group); 429252190Srpaulo if (beacon_ie) 430252190Srpaulo group->beacon_update = 0; 431252190Srpaulo wpa_hexdump_buf(MSG_MSGDUMP, "P2P: Update GO Beacon P2P IE", 432252190Srpaulo beacon_ie); 433252190Srpaulo } else 434252190Srpaulo beacon_ie = NULL; 435252190Srpaulo 436252190Srpaulo group->cfg->ie_update(group->cfg->cb_ctx, beacon_ie, probe_resp_ie); 437252190Srpaulo} 438252190Srpaulo 439252190Srpaulo 440252190Srpaulo/** 441252190Srpaulo * p2p_build_client_info - Build P2P Client Info Descriptor 442252190Srpaulo * @addr: MAC address of the peer device 443252190Srpaulo * @p2p_ie: P2P IE from (Re)Association Request 444252190Srpaulo * @dev_capab: Buffer for returning Device Capability 445252190Srpaulo * @dev_addr: Buffer for returning P2P Device Address 446252190Srpaulo * Returns: P2P Client Info Descriptor or %NULL on failure 447252190Srpaulo * 448252190Srpaulo * This function builds P2P Client Info Descriptor based on the information 449252190Srpaulo * available from (Re)Association Request frame. Group owner can use this to 450252190Srpaulo * build the P2P Group Info attribute for Probe Response frames. 451252190Srpaulo */ 452252190Srpaulostatic struct wpabuf * p2p_build_client_info(const u8 *addr, 453252190Srpaulo struct wpabuf *p2p_ie, 454252190Srpaulo u8 *dev_capab, u8 *dev_addr) 455252190Srpaulo{ 456252190Srpaulo const u8 *spos; 457252190Srpaulo struct p2p_message msg; 458252190Srpaulo u8 *len_pos; 459252190Srpaulo struct wpabuf *buf; 460252190Srpaulo 461252190Srpaulo if (p2p_ie == NULL) 462252190Srpaulo return NULL; 463252190Srpaulo 464252190Srpaulo os_memset(&msg, 0, sizeof(msg)); 465252190Srpaulo if (p2p_parse_p2p_ie(p2p_ie, &msg) || 466252190Srpaulo msg.capability == NULL || msg.p2p_device_info == NULL) 467252190Srpaulo return NULL; 468252190Srpaulo 469252190Srpaulo buf = wpabuf_alloc(ETH_ALEN + 1 + 1 + msg.p2p_device_info_len); 470252190Srpaulo if (buf == NULL) 471252190Srpaulo return NULL; 472252190Srpaulo 473252190Srpaulo *dev_capab = msg.capability[0]; 474252190Srpaulo os_memcpy(dev_addr, msg.p2p_device_addr, ETH_ALEN); 475252190Srpaulo 476252190Srpaulo spos = msg.p2p_device_info; /* P2P Device address */ 477252190Srpaulo 478252190Srpaulo /* P2P Client Info Descriptor */ 479252190Srpaulo /* Length to be set */ 480252190Srpaulo len_pos = wpabuf_put(buf, 1); 481252190Srpaulo /* P2P Device address */ 482252190Srpaulo wpabuf_put_data(buf, spos, ETH_ALEN); 483252190Srpaulo /* P2P Interface address */ 484252190Srpaulo wpabuf_put_data(buf, addr, ETH_ALEN); 485252190Srpaulo /* Device Capability Bitmap */ 486252190Srpaulo wpabuf_put_u8(buf, msg.capability[0]); 487252190Srpaulo /* 488252190Srpaulo * Config Methods, Primary Device Type, Number of Secondary Device 489252190Srpaulo * Types, Secondary Device Type List, Device Name copied from 490252190Srpaulo * Device Info 491252190Srpaulo */ 492252190Srpaulo wpabuf_put_data(buf, spos + ETH_ALEN, 493252190Srpaulo msg.p2p_device_info_len - ETH_ALEN); 494252190Srpaulo 495252190Srpaulo *len_pos = wpabuf_len(buf) - 1; 496252190Srpaulo 497252190Srpaulo 498252190Srpaulo return buf; 499252190Srpaulo} 500252190Srpaulo 501252190Srpaulo 502252190Srpaulostatic int p2p_group_remove_member(struct p2p_group *group, const u8 *addr) 503252190Srpaulo{ 504252190Srpaulo struct p2p_group_member *m, *prev; 505252190Srpaulo 506252190Srpaulo if (group == NULL) 507252190Srpaulo return 0; 508252190Srpaulo 509252190Srpaulo m = group->members; 510252190Srpaulo prev = NULL; 511252190Srpaulo while (m) { 512252190Srpaulo if (os_memcmp(m->addr, addr, ETH_ALEN) == 0) 513252190Srpaulo break; 514252190Srpaulo prev = m; 515252190Srpaulo m = m->next; 516252190Srpaulo } 517252190Srpaulo 518252190Srpaulo if (m == NULL) 519252190Srpaulo return 0; 520252190Srpaulo 521252190Srpaulo if (prev) 522252190Srpaulo prev->next = m->next; 523252190Srpaulo else 524252190Srpaulo group->members = m->next; 525252190Srpaulo p2p_group_free_member(m); 526252190Srpaulo group->num_members--; 527252190Srpaulo 528252190Srpaulo return 1; 529252190Srpaulo} 530252190Srpaulo 531252190Srpaulo 532252190Srpauloint p2p_group_notif_assoc(struct p2p_group *group, const u8 *addr, 533252190Srpaulo const u8 *ie, size_t len) 534252190Srpaulo{ 535252190Srpaulo struct p2p_group_member *m; 536252190Srpaulo 537252190Srpaulo if (group == NULL) 538252190Srpaulo return -1; 539252190Srpaulo 540252190Srpaulo m = os_zalloc(sizeof(*m)); 541252190Srpaulo if (m == NULL) 542252190Srpaulo return -1; 543252190Srpaulo os_memcpy(m->addr, addr, ETH_ALEN); 544252190Srpaulo m->p2p_ie = ieee802_11_vendor_ie_concat(ie, len, P2P_IE_VENDOR_TYPE); 545252190Srpaulo if (m->p2p_ie) { 546252190Srpaulo m->client_info = p2p_build_client_info(addr, m->p2p_ie, 547252190Srpaulo &m->dev_capab, 548252190Srpaulo m->dev_addr); 549252190Srpaulo } 550252190Srpaulo#ifdef CONFIG_WIFI_DISPLAY 551252190Srpaulo m->wfd_ie = ieee802_11_vendor_ie_concat(ie, len, WFD_IE_VENDOR_TYPE); 552252190Srpaulo#endif /* CONFIG_WIFI_DISPLAY */ 553252190Srpaulo 554252190Srpaulo p2p_group_remove_member(group, addr); 555252190Srpaulo 556252190Srpaulo m->next = group->members; 557252190Srpaulo group->members = m; 558252190Srpaulo group->num_members++; 559252190Srpaulo wpa_msg(group->p2p->cfg->msg_ctx, MSG_DEBUG, "P2P: Add client " MACSTR 560252190Srpaulo " to group (p2p=%d wfd=%d client_info=%d); num_members=%u/%u", 561252190Srpaulo MAC2STR(addr), m->p2p_ie ? 1 : 0, m->wfd_ie ? 1 : 0, 562252190Srpaulo m->client_info ? 1 : 0, 563252190Srpaulo group->num_members, group->cfg->max_clients); 564252190Srpaulo if (group->num_members == group->cfg->max_clients) 565252190Srpaulo group->beacon_update = 1; 566252190Srpaulo p2p_group_update_ies(group); 567252190Srpaulo if (group->num_members == 1) 568252190Srpaulo group->cfg->idle_update(group->cfg->cb_ctx, 0); 569252190Srpaulo 570252190Srpaulo return 0; 571252190Srpaulo} 572252190Srpaulo 573252190Srpaulo 574252190Srpaulostruct wpabuf * p2p_group_assoc_resp_ie(struct p2p_group *group, u8 status) 575252190Srpaulo{ 576252190Srpaulo struct wpabuf *resp; 577252190Srpaulo u8 *rlen; 578252190Srpaulo size_t extra = 0; 579252190Srpaulo 580252190Srpaulo#ifdef CONFIG_WIFI_DISPLAY 581252190Srpaulo if (group->wfd_ie) 582252190Srpaulo extra = wpabuf_len(group->wfd_ie); 583252190Srpaulo#endif /* CONFIG_WIFI_DISPLAY */ 584252190Srpaulo 585252190Srpaulo /* 586252190Srpaulo * (Re)Association Response - P2P IE 587252190Srpaulo * Status attribute (shall be present when association request is 588252190Srpaulo * denied) 589252190Srpaulo * Extended Listen Timing (may be present) 590252190Srpaulo */ 591252190Srpaulo resp = wpabuf_alloc(20 + extra); 592252190Srpaulo if (resp == NULL) 593252190Srpaulo return NULL; 594252190Srpaulo 595252190Srpaulo#ifdef CONFIG_WIFI_DISPLAY 596252190Srpaulo if (group->wfd_ie) 597252190Srpaulo wpabuf_put_buf(resp, group->wfd_ie); 598252190Srpaulo#endif /* CONFIG_WIFI_DISPLAY */ 599252190Srpaulo 600252190Srpaulo rlen = p2p_buf_add_ie_hdr(resp); 601252190Srpaulo if (status != P2P_SC_SUCCESS) 602252190Srpaulo p2p_buf_add_status(resp, status); 603252190Srpaulo p2p_buf_update_ie_hdr(resp, rlen); 604252190Srpaulo 605252190Srpaulo return resp; 606252190Srpaulo} 607252190Srpaulo 608252190Srpaulo 609252190Srpaulovoid p2p_group_notif_disassoc(struct p2p_group *group, const u8 *addr) 610252190Srpaulo{ 611252190Srpaulo if (p2p_group_remove_member(group, addr)) { 612252190Srpaulo wpa_msg(group->p2p->cfg->msg_ctx, MSG_DEBUG, "P2P: Remove " 613252190Srpaulo "client " MACSTR " from group; num_members=%u/%u", 614252190Srpaulo MAC2STR(addr), group->num_members, 615252190Srpaulo group->cfg->max_clients); 616252190Srpaulo if (group->num_members == group->cfg->max_clients - 1) 617252190Srpaulo group->beacon_update = 1; 618252190Srpaulo p2p_group_update_ies(group); 619252190Srpaulo if (group->num_members == 0) 620252190Srpaulo group->cfg->idle_update(group->cfg->cb_ctx, 1); 621252190Srpaulo } 622252190Srpaulo} 623252190Srpaulo 624252190Srpaulo 625252190Srpaulo/** 626252190Srpaulo * p2p_match_dev_type_member - Match client device type with requested type 627252190Srpaulo * @m: Group member 628252190Srpaulo * @wps: WPS TLVs from Probe Request frame (concatenated WPS IEs) 629252190Srpaulo * Returns: 1 on match, 0 on mismatch 630252190Srpaulo * 631252190Srpaulo * This function can be used to match the Requested Device Type attribute in 632252190Srpaulo * WPS IE with the device types of a group member for deciding whether a GO 633252190Srpaulo * should reply to a Probe Request frame. 634252190Srpaulo */ 635252190Srpaulostatic int p2p_match_dev_type_member(struct p2p_group_member *m, 636252190Srpaulo struct wpabuf *wps) 637252190Srpaulo{ 638252190Srpaulo const u8 *pos, *end; 639252190Srpaulo struct wps_parse_attr attr; 640252190Srpaulo u8 num_sec; 641252190Srpaulo 642252190Srpaulo if (m->client_info == NULL || wps == NULL) 643252190Srpaulo return 0; 644252190Srpaulo 645252190Srpaulo pos = wpabuf_head(m->client_info); 646252190Srpaulo end = pos + wpabuf_len(m->client_info); 647252190Srpaulo 648252190Srpaulo pos += 1 + 2 * ETH_ALEN + 1 + 2; 649252190Srpaulo if (end - pos < WPS_DEV_TYPE_LEN + 1) 650252190Srpaulo return 0; 651252190Srpaulo 652252190Srpaulo if (wps_parse_msg(wps, &attr)) 653252190Srpaulo return 1; /* assume no Requested Device Type attributes */ 654252190Srpaulo 655252190Srpaulo if (attr.num_req_dev_type == 0) 656252190Srpaulo return 1; /* no Requested Device Type attributes -> match */ 657252190Srpaulo 658252190Srpaulo if (dev_type_list_match(pos, attr.req_dev_type, attr.num_req_dev_type)) 659252190Srpaulo return 1; /* Match with client Primary Device Type */ 660252190Srpaulo 661252190Srpaulo pos += WPS_DEV_TYPE_LEN; 662252190Srpaulo num_sec = *pos++; 663252190Srpaulo if (end - pos < num_sec * WPS_DEV_TYPE_LEN) 664252190Srpaulo return 0; 665252190Srpaulo while (num_sec > 0) { 666252190Srpaulo num_sec--; 667252190Srpaulo if (dev_type_list_match(pos, attr.req_dev_type, 668252190Srpaulo attr.num_req_dev_type)) 669252190Srpaulo return 1; /* Match with client Secondary Device Type */ 670252190Srpaulo pos += WPS_DEV_TYPE_LEN; 671252190Srpaulo } 672252190Srpaulo 673252190Srpaulo /* No matching device type found */ 674252190Srpaulo return 0; 675252190Srpaulo} 676252190Srpaulo 677252190Srpaulo 678252190Srpauloint p2p_group_match_dev_type(struct p2p_group *group, struct wpabuf *wps) 679252190Srpaulo{ 680252190Srpaulo struct p2p_group_member *m; 681252190Srpaulo 682252190Srpaulo if (p2p_match_dev_type(group->p2p, wps)) 683252190Srpaulo return 1; /* Match with own device type */ 684252190Srpaulo 685252190Srpaulo for (m = group->members; m; m = m->next) { 686252190Srpaulo if (p2p_match_dev_type_member(m, wps)) 687252190Srpaulo return 1; /* Match with group client device type */ 688252190Srpaulo } 689252190Srpaulo 690252190Srpaulo /* No match with Requested Device Type */ 691252190Srpaulo return 0; 692252190Srpaulo} 693252190Srpaulo 694252190Srpaulo 695252190Srpauloint p2p_group_match_dev_id(struct p2p_group *group, struct wpabuf *p2p) 696252190Srpaulo{ 697252190Srpaulo struct p2p_group_member *m; 698252190Srpaulo struct p2p_message msg; 699252190Srpaulo 700252190Srpaulo os_memset(&msg, 0, sizeof(msg)); 701252190Srpaulo if (p2p_parse_p2p_ie(p2p, &msg)) 702252190Srpaulo return 1; /* Failed to parse - assume no filter on Device ID */ 703252190Srpaulo 704252190Srpaulo if (!msg.device_id) 705252190Srpaulo return 1; /* No filter on Device ID */ 706252190Srpaulo 707252190Srpaulo if (os_memcmp(msg.device_id, group->p2p->cfg->dev_addr, ETH_ALEN) == 0) 708252190Srpaulo return 1; /* Match with our P2P Device Address */ 709252190Srpaulo 710252190Srpaulo for (m = group->members; m; m = m->next) { 711252190Srpaulo if (os_memcmp(msg.device_id, m->dev_addr, ETH_ALEN) == 0) 712252190Srpaulo return 1; /* Match with group client P2P Device Address */ 713252190Srpaulo } 714252190Srpaulo 715252190Srpaulo /* No match with Device ID */ 716252190Srpaulo return 0; 717252190Srpaulo} 718252190Srpaulo 719252190Srpaulo 720252190Srpaulovoid p2p_group_notif_formation_done(struct p2p_group *group) 721252190Srpaulo{ 722252190Srpaulo if (group == NULL) 723252190Srpaulo return; 724252190Srpaulo group->group_formation = 0; 725252190Srpaulo group->beacon_update = 1; 726252190Srpaulo p2p_group_update_ies(group); 727252190Srpaulo} 728252190Srpaulo 729252190Srpaulo 730252190Srpauloint p2p_group_notif_noa(struct p2p_group *group, const u8 *noa, 731252190Srpaulo size_t noa_len) 732252190Srpaulo{ 733252190Srpaulo if (noa == NULL) { 734252190Srpaulo wpabuf_free(group->noa); 735252190Srpaulo group->noa = NULL; 736252190Srpaulo } else { 737252190Srpaulo if (group->noa) { 738252190Srpaulo if (wpabuf_size(group->noa) >= noa_len) { 739252190Srpaulo group->noa->used = 0; 740252190Srpaulo wpabuf_put_data(group->noa, noa, noa_len); 741252190Srpaulo } else { 742252190Srpaulo wpabuf_free(group->noa); 743252190Srpaulo group->noa = NULL; 744252190Srpaulo } 745252190Srpaulo } 746252190Srpaulo 747252190Srpaulo if (!group->noa) { 748252190Srpaulo group->noa = wpabuf_alloc_copy(noa, noa_len); 749252190Srpaulo if (group->noa == NULL) 750252190Srpaulo return -1; 751252190Srpaulo } 752252190Srpaulo } 753252190Srpaulo 754252190Srpaulo group->beacon_update = 1; 755252190Srpaulo p2p_group_update_ies(group); 756252190Srpaulo return 0; 757252190Srpaulo} 758252190Srpaulo 759252190Srpaulo 760252190Srpaulostatic struct p2p_group_member * p2p_group_get_client(struct p2p_group *group, 761252190Srpaulo const u8 *dev_id) 762252190Srpaulo{ 763252190Srpaulo struct p2p_group_member *m; 764252190Srpaulo 765252190Srpaulo for (m = group->members; m; m = m->next) { 766252190Srpaulo if (os_memcmp(dev_id, m->dev_addr, ETH_ALEN) == 0) 767252190Srpaulo return m; 768252190Srpaulo } 769252190Srpaulo 770252190Srpaulo return NULL; 771252190Srpaulo} 772252190Srpaulo 773252190Srpaulo 774252190Srpaulostatic struct p2p_group_member * p2p_group_get_client_iface( 775252190Srpaulo struct p2p_group *group, const u8 *interface_addr) 776252190Srpaulo{ 777252190Srpaulo struct p2p_group_member *m; 778252190Srpaulo 779252190Srpaulo for (m = group->members; m; m = m->next) { 780252190Srpaulo if (os_memcmp(interface_addr, m->addr, ETH_ALEN) == 0) 781252190Srpaulo return m; 782252190Srpaulo } 783252190Srpaulo 784252190Srpaulo return NULL; 785252190Srpaulo} 786252190Srpaulo 787252190Srpaulo 788252190Srpauloconst u8 * p2p_group_get_dev_addr(struct p2p_group *group, const u8 *addr) 789252190Srpaulo{ 790252190Srpaulo struct p2p_group_member *m; 791252190Srpaulo 792252190Srpaulo if (group == NULL) 793252190Srpaulo return NULL; 794252190Srpaulo m = p2p_group_get_client_iface(group, addr); 795252190Srpaulo if (m && !is_zero_ether_addr(m->dev_addr)) 796252190Srpaulo return m->dev_addr; 797252190Srpaulo return NULL; 798252190Srpaulo} 799252190Srpaulo 800252190Srpaulo 801252190Srpaulostatic struct wpabuf * p2p_build_go_disc_req(void) 802252190Srpaulo{ 803252190Srpaulo struct wpabuf *buf; 804252190Srpaulo 805252190Srpaulo buf = wpabuf_alloc(100); 806252190Srpaulo if (buf == NULL) 807252190Srpaulo return NULL; 808252190Srpaulo 809252190Srpaulo p2p_buf_add_action_hdr(buf, P2P_GO_DISC_REQ, 0); 810252190Srpaulo 811252190Srpaulo return buf; 812252190Srpaulo} 813252190Srpaulo 814252190Srpaulo 815252190Srpauloint p2p_group_go_discover(struct p2p_group *group, const u8 *dev_id, 816252190Srpaulo const u8 *searching_dev, int rx_freq) 817252190Srpaulo{ 818252190Srpaulo struct p2p_group_member *m; 819252190Srpaulo struct wpabuf *req; 820252190Srpaulo struct p2p_data *p2p = group->p2p; 821252190Srpaulo int freq; 822252190Srpaulo 823252190Srpaulo m = p2p_group_get_client(group, dev_id); 824252190Srpaulo if (m == NULL || m->client_info == NULL) { 825252190Srpaulo wpa_printf(MSG_DEBUG, "P2P: Requested client was not in this " 826252190Srpaulo "group " MACSTR, 827252190Srpaulo MAC2STR(group->cfg->interface_addr)); 828252190Srpaulo return -1; 829252190Srpaulo } 830252190Srpaulo 831252190Srpaulo if (!(m->dev_capab & P2P_DEV_CAPAB_CLIENT_DISCOVERABILITY)) { 832252190Srpaulo wpa_printf(MSG_DEBUG, "P2P: Requested client does not support " 833252190Srpaulo "client discoverability"); 834252190Srpaulo return -1; 835252190Srpaulo } 836252190Srpaulo 837252190Srpaulo wpa_printf(MSG_DEBUG, "P2P: Schedule GO Discoverability Request to be " 838252190Srpaulo "sent to " MACSTR, MAC2STR(dev_id)); 839252190Srpaulo 840252190Srpaulo req = p2p_build_go_disc_req(); 841252190Srpaulo if (req == NULL) 842252190Srpaulo return -1; 843252190Srpaulo 844252190Srpaulo /* TODO: Should really use group operating frequency here */ 845252190Srpaulo freq = rx_freq; 846252190Srpaulo 847252190Srpaulo p2p->pending_action_state = P2P_PENDING_GO_DISC_REQ; 848252190Srpaulo if (p2p->cfg->send_action(p2p->cfg->cb_ctx, freq, m->addr, 849252190Srpaulo group->cfg->interface_addr, 850252190Srpaulo group->cfg->interface_addr, 851252190Srpaulo wpabuf_head(req), wpabuf_len(req), 200) < 0) 852252190Srpaulo { 853252190Srpaulo wpa_msg(p2p->cfg->msg_ctx, MSG_DEBUG, 854252190Srpaulo "P2P: Failed to send Action frame"); 855252190Srpaulo } 856252190Srpaulo 857252190Srpaulo wpabuf_free(req); 858252190Srpaulo 859252190Srpaulo return 0; 860252190Srpaulo} 861252190Srpaulo 862252190Srpaulo 863252190Srpauloconst u8 * p2p_group_get_interface_addr(struct p2p_group *group) 864252190Srpaulo{ 865252190Srpaulo return group->cfg->interface_addr; 866252190Srpaulo} 867252190Srpaulo 868252190Srpaulo 869252190Srpaulou8 p2p_group_presence_req(struct p2p_group *group, 870252190Srpaulo const u8 *client_interface_addr, 871252190Srpaulo const u8 *noa, size_t noa_len) 872252190Srpaulo{ 873252190Srpaulo struct p2p_group_member *m; 874252190Srpaulo u8 curr_noa[50]; 875252190Srpaulo int curr_noa_len; 876252190Srpaulo 877252190Srpaulo m = p2p_group_get_client_iface(group, client_interface_addr); 878252190Srpaulo if (m == NULL || m->client_info == NULL) { 879252190Srpaulo wpa_printf(MSG_DEBUG, "P2P: Client was not in this group"); 880252190Srpaulo return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE; 881252190Srpaulo } 882252190Srpaulo 883252190Srpaulo wpa_hexdump(MSG_DEBUG, "P2P: Presence Request NoA", noa, noa_len); 884252190Srpaulo 885252190Srpaulo if (group->p2p->cfg->get_noa) 886252190Srpaulo curr_noa_len = group->p2p->cfg->get_noa( 887252190Srpaulo group->p2p->cfg->cb_ctx, group->cfg->interface_addr, 888252190Srpaulo curr_noa, sizeof(curr_noa)); 889252190Srpaulo else 890252190Srpaulo curr_noa_len = -1; 891252190Srpaulo if (curr_noa_len < 0) 892252190Srpaulo wpa_printf(MSG_DEBUG, "P2P: Failed to fetch current NoA"); 893252190Srpaulo else if (curr_noa_len == 0) 894252190Srpaulo wpa_printf(MSG_DEBUG, "P2P: No NoA being advertized"); 895252190Srpaulo else 896252190Srpaulo wpa_hexdump(MSG_DEBUG, "P2P: Current NoA", curr_noa, 897252190Srpaulo curr_noa_len); 898252190Srpaulo 899252190Srpaulo /* TODO: properly process request and store copy */ 900252190Srpaulo if (curr_noa_len > 0 || curr_noa_len == -1) 901252190Srpaulo return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE; 902252190Srpaulo 903252190Srpaulo return P2P_SC_SUCCESS; 904252190Srpaulo} 905252190Srpaulo 906252190Srpaulo 907252190Srpaulounsigned int p2p_get_group_num_members(struct p2p_group *group) 908252190Srpaulo{ 909252190Srpaulo return group->num_members; 910252190Srpaulo} 911252190Srpaulo 912252190Srpaulo 913252190Srpauloconst u8 * p2p_iterate_group_members(struct p2p_group *group, void **next) 914252190Srpaulo{ 915252190Srpaulo struct p2p_group_member *iter = *next; 916252190Srpaulo 917252190Srpaulo if (!iter) 918252190Srpaulo iter = group->members; 919252190Srpaulo else 920252190Srpaulo iter = iter->next; 921252190Srpaulo 922252190Srpaulo *next = iter; 923252190Srpaulo 924252190Srpaulo if (!iter) 925252190Srpaulo return NULL; 926252190Srpaulo 927252190Srpaulo return iter->addr; 928252190Srpaulo} 929252190Srpaulo 930252190Srpaulo 931252190Srpauloint p2p_group_is_client_connected(struct p2p_group *group, const u8 *dev_addr) 932252190Srpaulo{ 933252190Srpaulo struct p2p_group_member *m; 934252190Srpaulo 935252190Srpaulo for (m = group->members; m; m = m->next) { 936252190Srpaulo if (os_memcmp(m->dev_addr, dev_addr, ETH_ALEN) == 0) 937252190Srpaulo return 1; 938252190Srpaulo } 939252190Srpaulo 940252190Srpaulo return 0; 941252190Srpaulo} 942252190Srpaulo 943252190Srpaulo 944252190Srpauloint p2p_group_is_group_id_match(struct p2p_group *group, const u8 *group_id, 945252190Srpaulo size_t group_id_len) 946252190Srpaulo{ 947252190Srpaulo if (group_id_len != ETH_ALEN + group->cfg->ssid_len) 948252190Srpaulo return 0; 949252190Srpaulo if (os_memcmp(group_id, group->p2p->cfg->dev_addr, ETH_ALEN) != 0) 950252190Srpaulo return 0; 951252190Srpaulo return os_memcmp(group_id + ETH_ALEN, group->cfg->ssid, 952252190Srpaulo group->cfg->ssid_len) == 0; 953252190Srpaulo} 954