1 /* 2 * wpa_supplicant - Wi-Fi Display 3 * Copyright (c) 2011, Atheros Communications, Inc. 4 * Copyright (c) 2011-2012, Qualcomm Atheros, Inc. 5 * 6 * This software may be distributed under the terms of the BSD license. 7 * See README for more details. 8 */ 9 10 #include "includes.h" 11 12 #include "common.h" 13 #include "p2p/p2p.h" 14 #include "common/ieee802_11_defs.h" 15 #include "wpa_supplicant_i.h" 16 #include "wifi_display.h" 17 18 19 int wifi_display_init(struct wpa_global *global) 20 { 21 global->wifi_display = 1; 22 return 0; 23 } 24 25 26 void wifi_display_deinit(struct wpa_global *global) 27 { 28 int i; 29 for (i = 0; i < MAX_WFD_SUBELEMS; i++) { 30 wpabuf_free(global->wfd_subelem[i]); 31 global->wfd_subelem[i] = NULL; 32 } 33 } 34 35 36 static int wifi_display_update_wfd_ie(struct wpa_global *global) 37 { 38 struct wpabuf *ie, *buf; 39 size_t len, plen; 40 41 wpa_printf(MSG_DEBUG, "WFD: Update WFD IE"); 42 43 if (!global->wifi_display) { 44 wpa_printf(MSG_DEBUG, "WFD: Wi-Fi Display disabled - do not " 45 "include WFD IE"); 46 p2p_set_wfd_ie_beacon(global->p2p, NULL); 47 p2p_set_wfd_ie_probe_req(global->p2p, NULL); 48 p2p_set_wfd_ie_probe_resp(global->p2p, NULL); 49 p2p_set_wfd_ie_assoc_req(global->p2p, NULL); 50 p2p_set_wfd_ie_invitation(global->p2p, NULL); 51 p2p_set_wfd_ie_prov_disc_req(global->p2p, NULL); 52 p2p_set_wfd_ie_prov_disc_resp(global->p2p, NULL); 53 p2p_set_wfd_ie_go_neg(global->p2p, NULL); 54 p2p_set_wfd_dev_info(global->p2p, NULL); 55 p2p_set_wfd_assoc_bssid(global->p2p, NULL); 56 p2p_set_wfd_coupled_sink_info(global->p2p, NULL); 57 return 0; 58 } 59 60 p2p_set_wfd_dev_info(global->p2p, 61 global->wfd_subelem[WFD_SUBELEM_DEVICE_INFO]); 62 p2p_set_wfd_assoc_bssid( 63 global->p2p, 64 global->wfd_subelem[WFD_SUBELEM_ASSOCIATED_BSSID]); 65 p2p_set_wfd_coupled_sink_info( 66 global->p2p, global->wfd_subelem[WFD_SUBELEM_COUPLED_SINK]); 67 68 /* 69 * WFD IE is included in number of management frames. Two different 70 * sets of subelements are included depending on the frame: 71 * 72 * Beacon, (Re)Association Request, GO Negotiation Req/Resp/Conf, 73 * Provision Discovery Req: 74 * WFD Device Info 75 * [Associated BSSID] 76 * [Coupled Sink Info] 77 * 78 * Probe Request: 79 * WFD Device Info 80 * [Associated BSSID] 81 * [Coupled Sink Info] 82 * [WFD Extended Capability] 83 * 84 * Probe Response: 85 * WFD Device Info 86 * [Associated BSSID] 87 * [Coupled Sink Info] 88 * [WFD Extended Capability] 89 * [WFD Session Info] 90 * 91 * (Re)Association Response, P2P Invitation Req/Resp, 92 * Provision Discovery Resp: 93 * WFD Device Info 94 * [Associated BSSID] 95 * [Coupled Sink Info] 96 * [WFD Session Info] 97 */ 98 len = 0; 99 if (global->wfd_subelem[WFD_SUBELEM_DEVICE_INFO]) 100 len += wpabuf_len(global->wfd_subelem[ 101 WFD_SUBELEM_DEVICE_INFO]); 102 if (global->wfd_subelem[WFD_SUBELEM_ASSOCIATED_BSSID]) 103 len += wpabuf_len(global->wfd_subelem[ 104 WFD_SUBELEM_ASSOCIATED_BSSID]); 105 if (global->wfd_subelem[WFD_SUBELEM_COUPLED_SINK]) 106 len += wpabuf_len(global->wfd_subelem[ 107 WFD_SUBELEM_COUPLED_SINK]); 108 if (global->wfd_subelem[WFD_SUBELEM_SESSION_INFO]) 109 len += wpabuf_len(global->wfd_subelem[ 110 WFD_SUBELEM_SESSION_INFO]); 111 if (global->wfd_subelem[WFD_SUBELEM_EXT_CAPAB]) 112 len += wpabuf_len(global->wfd_subelem[WFD_SUBELEM_EXT_CAPAB]); 113 buf = wpabuf_alloc(len); 114 if (buf == NULL) 115 return -1; 116 117 if (global->wfd_subelem[WFD_SUBELEM_DEVICE_INFO]) 118 wpabuf_put_buf(buf, 119 global->wfd_subelem[WFD_SUBELEM_DEVICE_INFO]); 120 if (global->wfd_subelem[WFD_SUBELEM_ASSOCIATED_BSSID]) 121 wpabuf_put_buf(buf, global->wfd_subelem[ 122 WFD_SUBELEM_ASSOCIATED_BSSID]); 123 if (global->wfd_subelem[WFD_SUBELEM_COUPLED_SINK]) 124 wpabuf_put_buf(buf, 125 global->wfd_subelem[WFD_SUBELEM_COUPLED_SINK]); 126 127 ie = wifi_display_encaps(buf); 128 wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Beacon", ie); 129 p2p_set_wfd_ie_beacon(global->p2p, ie); 130 131 ie = wifi_display_encaps(buf); 132 wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for (Re)Association Request", 133 ie); 134 p2p_set_wfd_ie_assoc_req(global->p2p, ie); 135 136 ie = wifi_display_encaps(buf); 137 wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for GO Negotiation", ie); 138 p2p_set_wfd_ie_go_neg(global->p2p, ie); 139 140 ie = wifi_display_encaps(buf); 141 wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Provision Discovery " 142 "Request", ie); 143 p2p_set_wfd_ie_prov_disc_req(global->p2p, ie); 144 145 plen = buf->used; 146 if (global->wfd_subelem[WFD_SUBELEM_EXT_CAPAB]) 147 wpabuf_put_buf(buf, 148 global->wfd_subelem[WFD_SUBELEM_EXT_CAPAB]); 149 150 ie = wifi_display_encaps(buf); 151 wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Probe Request", ie); 152 p2p_set_wfd_ie_probe_req(global->p2p, ie); 153 154 if (global->wfd_subelem[WFD_SUBELEM_SESSION_INFO]) 155 wpabuf_put_buf(buf, 156 global->wfd_subelem[WFD_SUBELEM_SESSION_INFO]); 157 ie = wifi_display_encaps(buf); 158 wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Probe Response", ie); 159 p2p_set_wfd_ie_probe_resp(global->p2p, ie); 160 161 /* Remove WFD Extended Capability from buffer */ 162 buf->used = plen; 163 if (global->wfd_subelem[WFD_SUBELEM_SESSION_INFO]) 164 wpabuf_put_buf(buf, 165 global->wfd_subelem[WFD_SUBELEM_SESSION_INFO]); 166 167 ie = wifi_display_encaps(buf); 168 wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for P2P Invitation", ie); 169 p2p_set_wfd_ie_invitation(global->p2p, ie); 170 171 ie = wifi_display_encaps(buf); 172 wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Provision Discovery " 173 "Response", ie); 174 p2p_set_wfd_ie_prov_disc_resp(global->p2p, ie); 175 176 wpabuf_free(buf); 177 178 return 0; 179 } 180 181 182 void wifi_display_enable(struct wpa_global *global, int enabled) 183 { 184 wpa_printf(MSG_DEBUG, "WFD: Wi-Fi Display %s", 185 enabled ? "enabled" : "disabled"); 186 global->wifi_display = enabled; 187 wifi_display_update_wfd_ie(global); 188 } 189 190 191 int wifi_display_subelem_set(struct wpa_global *global, char *cmd) 192 { 193 char *pos; 194 int subelem; 195 size_t len; 196 struct wpabuf *e; 197 198 pos = os_strchr(cmd, ' '); 199 if (pos == NULL) 200 return -1; 201 *pos++ = '\0'; 202 subelem = atoi(cmd); 203 if (subelem < 0 || subelem >= MAX_WFD_SUBELEMS) 204 return -1; 205 206 len = os_strlen(pos); 207 if (len & 1) 208 return -1; 209 len /= 2; 210 211 if (len == 0) { 212 /* Clear subelement */ 213 e = NULL; 214 wpa_printf(MSG_DEBUG, "WFD: Clear subelement %d", subelem); 215 } else { 216 e = wpabuf_alloc(1 + len); 217 if (e == NULL) 218 return -1; 219 wpabuf_put_u8(e, subelem); 220 if (hexstr2bin(pos, wpabuf_put(e, len), len) < 0) { 221 wpabuf_free(e); 222 return -1; 223 } 224 wpa_printf(MSG_DEBUG, "WFD: Set subelement %d", subelem); 225 } 226 227 wpabuf_free(global->wfd_subelem[subelem]); 228 global->wfd_subelem[subelem] = e; 229 wifi_display_update_wfd_ie(global); 230 231 return 0; 232 } 233 234 235 int wifi_display_subelem_get(struct wpa_global *global, char *cmd, 236 char *buf, size_t buflen) 237 { 238 int subelem; 239 240 subelem = atoi(cmd); 241 if (subelem < 0 || subelem >= MAX_WFD_SUBELEMS) 242 return -1; 243 244 if (global->wfd_subelem[subelem] == NULL) 245 return 0; 246 247 return wpa_snprintf_hex(buf, buflen, 248 wpabuf_head_u8(global->wfd_subelem[subelem]) + 249 1, 250 wpabuf_len(global->wfd_subelem[subelem]) - 1); 251 } 252
