[thirdparty/hostap.git] / wpa_supplicant / wifi_display.c

/*
 * wpa_supplicant - Wi-Fi Display
 * Copyright (c) 2011, Atheros Communications, Inc.
 * Copyright (c) 2011-2012, Qualcomm Atheros, Inc.
 *
 * This software may be distributed under the terms of the BSD license.
 * See README for more details.
 */

#include "includes.h"

#include "common.h"
#include "p2p/p2p.h"
#include "common/ieee802_11_defs.h"
#include "wpa_supplicant_i.h"
#include "wifi_display.h"


int wifi_display_init(struct wpa_global *global)
{
	global->wifi_display = 1;
	return 0;
}


void wifi_display_deinit(struct wpa_global *global)
{
	int i;
	for (i = 0; i < MAX_WFD_SUBELEMS; i++) {
		wpabuf_free(global->wfd_subelem[i]);
		global->wfd_subelem[i] = NULL;
	}
}


static int wifi_display_update_wfd_ie(struct wpa_global *global)
{
	struct wpabuf *ie, *buf;
	size_t len, plen;

	wpa_printf(MSG_DEBUG, "WFD: Update WFD IE");

	if (!global->wifi_display) {
		wpa_printf(MSG_DEBUG, "WFD: Wi-Fi Display disabled - do not "
			   "include WFD IE");
		p2p_set_wfd_ie_beacon(global->p2p, NULL);
		p2p_set_wfd_ie_probe_req(global->p2p, NULL);
		p2p_set_wfd_ie_probe_resp(global->p2p, NULL);
		p2p_set_wfd_ie_assoc_req(global->p2p, NULL);
		p2p_set_wfd_ie_invitation(global->p2p, NULL);
		p2p_set_wfd_ie_prov_disc_req(global->p2p, NULL);
		p2p_set_wfd_ie_prov_disc_resp(global->p2p, NULL);
		p2p_set_wfd_ie_go_neg(global->p2p, NULL);
		p2p_set_wfd_dev_info(global->p2p, NULL);
		p2p_set_wfd_assoc_bssid(global->p2p, NULL);
		p2p_set_wfd_coupled_sink_info(global->p2p, NULL);
		return 0;
	}

	p2p_set_wfd_dev_info(global->p2p,
			     global->wfd_subelem[WFD_SUBELEM_DEVICE_INFO]);
	p2p_set_wfd_assoc_bssid(
		global->p2p,
		global->wfd_subelem[WFD_SUBELEM_ASSOCIATED_BSSID]);
	p2p_set_wfd_coupled_sink_info(
		global->p2p, global->wfd_subelem[WFD_SUBELEM_COUPLED_SINK]);

	/*
	 * WFD IE is included in number of management frames. Two different
	 * sets of subelements are included depending on the frame:
	 *
	 * Beacon, (Re)Association Request, GO Negotiation Req/Resp/Conf,
	 * Provision Discovery Req:
	 * WFD Device Info
	 * [Associated BSSID]
	 * [Coupled Sink Info]
	 *
	 * Probe Request:
	 * WFD Device Info
	 * [Associated BSSID]
	 * [Coupled Sink Info]
	 * [WFD Extended Capability]
	 *
	 * Probe Response:
	 * WFD Device Info
	 * [Associated BSSID]
	 * [Coupled Sink Info]
	 * [WFD Extended Capability]
	 * [WFD Session Info]
	 *
	 * (Re)Association Response, P2P Invitation Req/Resp,
	 * Provision Discovery Resp:
	 * WFD Device Info
	 * [Associated BSSID]
	 * [Coupled Sink Info]
	 * [WFD Session Info]
	 */
	len = 0;
	if (global->wfd_subelem[WFD_SUBELEM_DEVICE_INFO])
		len += wpabuf_len(global->wfd_subelem[
					  WFD_SUBELEM_DEVICE_INFO]);
	if (global->wfd_subelem[WFD_SUBELEM_ASSOCIATED_BSSID])
		len += wpabuf_len(global->wfd_subelem[
					  WFD_SUBELEM_ASSOCIATED_BSSID]);
	if (global->wfd_subelem[WFD_SUBELEM_COUPLED_SINK])
		len += wpabuf_len(global->wfd_subelem[
					  WFD_SUBELEM_COUPLED_SINK]);
	if (global->wfd_subelem[WFD_SUBELEM_SESSION_INFO])
		len += wpabuf_len(global->wfd_subelem[
					  WFD_SUBELEM_SESSION_INFO]);
	if (global->wfd_subelem[WFD_SUBELEM_EXT_CAPAB])
		len += wpabuf_len(global->wfd_subelem[WFD_SUBELEM_EXT_CAPAB]);
	buf = wpabuf_alloc(len);
	if (buf == NULL)
		return -1;

	if (global->wfd_subelem[WFD_SUBELEM_DEVICE_INFO])
		wpabuf_put_buf(buf,
			       global->wfd_subelem[WFD_SUBELEM_DEVICE_INFO]);
	if (global->wfd_subelem[WFD_SUBELEM_ASSOCIATED_BSSID])
		wpabuf_put_buf(buf, global->wfd_subelem[
				       WFD_SUBELEM_ASSOCIATED_BSSID]);
	if (global->wfd_subelem[WFD_SUBELEM_COUPLED_SINK])
		wpabuf_put_buf(buf,
			       global->wfd_subelem[WFD_SUBELEM_COUPLED_SINK]);

	ie = wifi_display_encaps(buf);
	wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Beacon", ie);
	p2p_set_wfd_ie_beacon(global->p2p, ie);

	ie = wifi_display_encaps(buf);
	wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for (Re)Association Request",
			ie);
	p2p_set_wfd_ie_assoc_req(global->p2p, ie);

	ie = wifi_display_encaps(buf);
	wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for GO Negotiation", ie);
	p2p_set_wfd_ie_go_neg(global->p2p, ie);

	ie = wifi_display_encaps(buf);
	wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Provision Discovery "
			"Request", ie);
	p2p_set_wfd_ie_prov_disc_req(global->p2p, ie);

	plen = buf->used;
	if (global->wfd_subelem[WFD_SUBELEM_EXT_CAPAB])
		wpabuf_put_buf(buf,
			       global->wfd_subelem[WFD_SUBELEM_EXT_CAPAB]);

	ie = wifi_display_encaps(buf);
	wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Probe Request", ie);
	p2p_set_wfd_ie_probe_req(global->p2p, ie);

	if (global->wfd_subelem[WFD_SUBELEM_SESSION_INFO])
		wpabuf_put_buf(buf,
			       global->wfd_subelem[WFD_SUBELEM_SESSION_INFO]);
	ie = wifi_display_encaps(buf);
	wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Probe Response", ie);
	p2p_set_wfd_ie_probe_resp(global->p2p, ie);

	/* Remove WFD Extended Capability from buffer */
	buf->used = plen;
	if (global->wfd_subelem[WFD_SUBELEM_SESSION_INFO])
		wpabuf_put_buf(buf,
			       global->wfd_subelem[WFD_SUBELEM_SESSION_INFO]);

	ie = wifi_display_encaps(buf);
	wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for P2P Invitation", ie);
	p2p_set_wfd_ie_invitation(global->p2p, ie);

	ie = wifi_display_encaps(buf);
	wpa_hexdump_buf(MSG_DEBUG, "WFD: WFD IE for Provision Discovery "
			"Response", ie);
	p2p_set_wfd_ie_prov_disc_resp(global->p2p, ie);

	wpabuf_free(buf);

	return 0;
}


void wifi_display_enable(struct wpa_global *global, int enabled)
{
	wpa_printf(MSG_DEBUG, "WFD: Wi-Fi Display %s",
		   enabled ? "enabled" : "disabled");
	global->wifi_display = enabled;
	wifi_display_update_wfd_ie(global);
}


int wifi_display_subelem_set(struct wpa_global *global, char *cmd)
{
	char *pos;
	int subelem;
	size_t len;
	struct wpabuf *e;

	pos = os_strchr(cmd, ' ');
	if (pos == NULL)
		return -1;
	*pos++ = '\0';
	subelem = atoi(cmd);
	if (subelem < 0 || subelem >= MAX_WFD_SUBELEMS)
		return -1;

	len = os_strlen(pos);
	if (len & 1)
		return -1;
	len /= 2;

	if (len == 0) {
		/* Clear subelement */
		e = NULL;
		wpa_printf(MSG_DEBUG, "WFD: Clear subelement %d", subelem);
	} else {
		e = wpabuf_alloc(1 + len);
		if (e == NULL)
			return -1;
		wpabuf_put_u8(e, subelem);
		if (hexstr2bin(pos, wpabuf_put(e, len), len) < 0) {
			wpabuf_free(e);
			return -1;
		}
		wpa_printf(MSG_DEBUG, "WFD: Set subelement %d", subelem);
	}

	wpabuf_free(global->wfd_subelem[subelem]);
	global->wfd_subelem[subelem] = e;
	wifi_display_update_wfd_ie(global);

	return 0;
}


int wifi_display_subelem_get(struct wpa_global *global, char *cmd,
			     char *buf, size_t buflen)
{
	int subelem;

	subelem = atoi(cmd);
	if (subelem < 0 || subelem >= MAX_WFD_SUBELEMS)
		return -1;

	if (global->wfd_subelem[subelem] == NULL)
		return 0;

	return wpa_snprintf_hex(buf, buflen,
				wpabuf_head_u8(global->wfd_subelem[subelem]) +
				1,
				wpabuf_len(global->wfd_subelem[subelem]) - 1);
}
Commit	Line	Data
9675ce35 JM	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	}