[thirdparty/hostap.git] / src / ap / wnm_ap.c

/*
 * hostapd - WNM
 * 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 "utils/includes.h"

#include "utils/common.h"
#include "common/ieee802_11_defs.h"
#include "ap/hostapd.h"
#include "ap/sta_info.h"
#include "ap/ap_config.h"
#include "ap/ap_drv_ops.h"
#include "ap/wpa_auth.h"
#include "wnm_ap.h"

#define MAX_TFS_IE_LEN  1024


/* get the TFS IE from driver */
static int ieee80211_11_get_tfs_ie(struct hostapd_data *hapd, const u8 *addr,
				   u8 *buf, u16 *buf_len, enum wnm_oper oper)
{
	wpa_printf(MSG_DEBUG, "%s: TFS get operation %d", __func__, oper);

	return hostapd_drv_wnm_oper(hapd, oper, addr, buf, buf_len);
}


/* set the TFS IE to driver */
static int ieee80211_11_set_tfs_ie(struct hostapd_data *hapd, const u8 *addr,
				   u8 *buf, u16 *buf_len, enum wnm_oper oper)
{
	wpa_printf(MSG_DEBUG, "%s: TFS set operation %d", __func__, oper);

	return hostapd_drv_wnm_oper(hapd, oper, addr, buf, buf_len);
}


/* MLME-SLEEPMODE.response */
static int ieee802_11_send_wnmsleep_resp(struct hostapd_data *hapd,
					 const u8 *addr, u8 dialog_token,
					 u8 action_type, u16 intval)
{
	struct ieee80211_mgmt *mgmt;
	int res;
	size_t len;
	size_t gtk_elem_len = 0;
	size_t igtk_elem_len = 0;
	struct wnm_sleep_element wnmsleep_ie;
	u8 *wnmtfs_ie;
	u8 wnmsleep_ie_len;
	u16 wnmtfs_ie_len;
	u8 *pos;
	struct sta_info *sta;
	enum wnm_oper tfs_oper = action_type == 0 ? WNM_SLEEP_TFS_RESP_IE_ADD :
		WNM_SLEEP_TFS_RESP_IE_NONE;

	sta = ap_get_sta(hapd, addr);
	if (sta == NULL) {
		wpa_printf(MSG_DEBUG, "%s: station not found", __func__);
		return -EINVAL;
	}

	/* WNM-Sleep Mode IE */
	os_memset(&wnmsleep_ie, 0, sizeof(struct wnm_sleep_element));
	wnmsleep_ie_len = sizeof(struct wnm_sleep_element);
	wnmsleep_ie.eid = WLAN_EID_WNMSLEEP;
	wnmsleep_ie.len = wnmsleep_ie_len - 2;
	wnmsleep_ie.action_type = action_type;
	wnmsleep_ie.status = WNM_STATUS_SLEEP_ACCEPT;
	wnmsleep_ie.intval = intval;

	/* TFS IE(s) */
	wnmtfs_ie = os_zalloc(MAX_TFS_IE_LEN);
	if (wnmtfs_ie == NULL)
		return -1;
	if (ieee80211_11_get_tfs_ie(hapd, addr, wnmtfs_ie, &wnmtfs_ie_len,
				    tfs_oper)) {
		wnmtfs_ie_len = 0;
		os_free(wnmtfs_ie);
		wnmtfs_ie = NULL;
	}

#define MAX_GTK_SUBELEM_LEN 45
#define MAX_IGTK_SUBELEM_LEN 26
	mgmt = os_zalloc(sizeof(*mgmt) + wnmsleep_ie_len +
			 MAX_GTK_SUBELEM_LEN + MAX_IGTK_SUBELEM_LEN);
	if (mgmt == NULL) {
		wpa_printf(MSG_DEBUG, "MLME: Failed to allocate buffer for "
			   "WNM-Sleep Response action frame");
		return -1;
	}
	os_memcpy(mgmt->da, addr, ETH_ALEN);
	os_memcpy(mgmt->sa, hapd->own_addr, ETH_ALEN);
	os_memcpy(mgmt->bssid, hapd->own_addr, ETH_ALEN);
	mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
					   WLAN_FC_STYPE_ACTION);
	mgmt->u.action.category = WLAN_ACTION_WNM;
	mgmt->u.action.u.wnm_sleep_resp.action = WNM_SLEEP_MODE_RESP;
	mgmt->u.action.u.wnm_sleep_resp.dialogtoken = dialog_token;
	pos = (u8 *)mgmt->u.action.u.wnm_sleep_resp.variable;
	/* add key data if MFP is enabled */
	if (wpa_auth_uses_mfp(sta->wpa_sm) || action_type != 1){
		mgmt->u.action.u.wnm_sleep_resp.keydata_len = 0;
	} else {
		gtk_elem_len = wpa_wnmsleep_gtk_subelem(sta->wpa_sm, pos);
		pos += gtk_elem_len;
		wpa_printf(MSG_DEBUG, "Pass 4, gtk_len = %d",
			   (int) gtk_elem_len);
#ifdef CONFIG_IEEE80211W
		res = wpa_wnmsleep_igtk_subelem(sta->wpa_sm, pos);
		if (res < 0) {
			os_free(wnmtfs_ie);
			os_free(mgmt);
			return -1;
		}
		igtk_elem_len = res;
		pos += igtk_elem_len;
		wpa_printf(MSG_DEBUG, "Pass 4 igtk_len = %d",
			   (int) igtk_elem_len);
#endif /* CONFIG_IEEE80211W */

		WPA_PUT_LE16((u8 *)
			     &mgmt->u.action.u.wnm_sleep_resp.keydata_len,
			     gtk_elem_len + igtk_elem_len);
	}
	os_memcpy(pos, &wnmsleep_ie, wnmsleep_ie_len);
	/* copy TFS IE here */
	pos += wnmsleep_ie_len;
	os_memcpy(pos, wnmtfs_ie, wnmtfs_ie_len);

	len = 1 + sizeof(mgmt->u.action.u.wnm_sleep_resp) + gtk_elem_len +
		igtk_elem_len + wnmsleep_ie_len + wnmtfs_ie_len;

	/* In driver, response frame should be forced to sent when STA is in
	 * PS mode */
	res = hostapd_drv_send_action(hapd, hapd->iface->freq, 0,
				      mgmt->da, &mgmt->u.action.category, len);

	if (!res) {
		wpa_printf(MSG_DEBUG, "Successfully send WNM-Sleep Response "
			   "frame");

		/* when entering wnmsleep
		 * 1. pause the node in driver
		 * 2. mark the node so that AP won't update GTK/IGTK during
		 * WNM Sleep
		 */
		if (wnmsleep_ie.status == WNM_STATUS_SLEEP_ACCEPT &&
		    wnmsleep_ie.action_type == 0) {
			hostapd_drv_wnm_oper(hapd, WNM_SLEEP_ENTER_CONFIRM,
					     addr, NULL, NULL);
			wpa_set_wnmsleep(sta->wpa_sm, 1);
		}
		/* when exiting wnmsleep
		 * 1. unmark the node
		 * 2. start GTK/IGTK update if MFP is not used
		 * 3. unpause the node in driver
		 */
		if (wnmsleep_ie.status == WNM_STATUS_SLEEP_ACCEPT &&
		    wnmsleep_ie.action_type == 1) {
			wpa_set_wnmsleep(sta->wpa_sm, 0);
			hostapd_drv_wnm_oper(hapd, WNM_SLEEP_EXIT_CONFIRM,
					     addr, NULL, NULL);
			if (wpa_auth_uses_mfp(sta->wpa_sm) && action_type == 1)
				wpa_wnmsleep_rekey_gtk(sta->wpa_sm);
		}
	} else
		wpa_printf(MSG_DEBUG, "Fail to send WNM-Sleep Response frame");

#undef MAX_GTK_SUBELEM_LEN
#undef MAX_IGTK_SUBELEM_LEN
	os_free(wnmtfs_ie);
	os_free(mgmt);
	return res;
}


static void ieee802_11_rx_wnmsleep_req(struct hostapd_data *hapd,
				       const u8 *addr, const u8 *frm, int len)
{
	/*
	 * Action [1] | Dialog Token [1] | WNM-Sleep Mode IE |
	 * TFS Response IE
	 */
	u8 *pos = (u8 *) frm; /* point to action field */
	u8 dialog_token = pos[1];
	struct wnm_sleep_element *wnmsleep_ie = NULL;
	/* multiple TFS Req IE (assuming consecutive) */
	u8 *tfsreq_ie_start = NULL;
	u8 *tfsreq_ie_end = NULL;
	u16 tfsreq_ie_len = 0;

	pos += 1 + 1;
	while (pos - frm < len - 1) {
		u8 ie_len = *(pos+1);
		if (*pos == WLAN_EID_WNMSLEEP)
			wnmsleep_ie = (struct wnm_sleep_element *)pos;
		else if (*pos == WLAN_EID_TFS_REQ) {
			if (!tfsreq_ie_start)
				tfsreq_ie_start = pos;
			tfsreq_ie_end = pos;
		} else
			wpa_printf(MSG_DEBUG, "EID %d not recognized", *pos);
		pos += ie_len + 2;
	}

	if (!wnmsleep_ie) {
		wpa_printf(MSG_DEBUG, "No WNM-Sleep IE found");
		return;
	}

	if (wnmsleep_ie->action_type == 0 && tfsreq_ie_start &&
	    tfsreq_ie_end && tfsreq_ie_end - tfsreq_ie_start >= 0) {
		tfsreq_ie_len = (tfsreq_ie_end + tfsreq_ie_end[1] + 2) -
			tfsreq_ie_start;
		wpa_printf(MSG_DEBUG, "TFS Req IE(s) found");
		/* pass the TFS Req IE(s) to driver for processing */
		if (ieee80211_11_set_tfs_ie(hapd, addr, tfsreq_ie_start,
					    &tfsreq_ie_len,
					    WNM_SLEEP_TFS_REQ_IE_SET))
			wpa_printf(MSG_DEBUG, "Fail to set TFS Req IE");
	}

	ieee802_11_send_wnmsleep_resp(hapd, addr, dialog_token,
				      wnmsleep_ie->action_type,
				      wnmsleep_ie->intval);

	if (wnmsleep_ie->action_type == 1) {
		/* clear the tfs after sending the resp frame */
		ieee80211_11_set_tfs_ie(hapd, addr, tfsreq_ie_start,
					&tfsreq_ie_len, WNM_SLEEP_TFS_IE_DEL);
	}
}


void ieee802_11_rx_wnm_action_ap(struct hostapd_data *hapd,
				 struct rx_action *action)
{
	u8 *pos = (u8 *) action->data + 1; /* point to the action field */
	u8 act = *pos;

	switch (act) {
	case WNM_SLEEP_MODE_REQ:
		ieee802_11_rx_wnmsleep_req(hapd, action->sa, action->data + 1,
					   action->len);
		break;
	default:
		break;
	}
}
Commit	Line	Data
d32d94db XC	1	/*
	2	* hostapd - WNM
	3	* Copyright (c) 2011-2012, Qualcomm Atheros, Inc.
	4	*
	5	* This software may be distributed under the terms of the BSD license.
	6	* See README for more details.
	7	*/
	8
	9	#include "utils/includes.h"
	10
	11	#include "utils/common.h"
	12	#include "common/ieee802_11_defs.h"
	13	#include "ap/hostapd.h"
	14	#include "ap/sta_info.h"
	15	#include "ap/ap_config.h"
	16	#include "ap/ap_drv_ops.h"
	17	#include "ap/wpa_auth.h"
	18	#include "wnm_ap.h"
	19
	20	#define MAX_TFS_IE_LEN 1024
	21
d32d94db XC	22
	23	/* get the TFS IE from driver */
	24	static int ieee80211_11_get_tfs_ie(struct hostapd_data hapd, const u8 addr,
	25	u8 buf, u16 buf_len, enum wnm_oper oper)
	26	{
	27	wpa_printf(MSG_DEBUG, "%s: TFS get operation %d", __func__, oper);
	28
	29	return hostapd_drv_wnm_oper(hapd, oper, addr, buf, buf_len);
	30	}
	31
	32
	33	/* set the TFS IE to driver */
	34	static int ieee80211_11_set_tfs_ie(struct hostapd_data hapd, const u8 addr,
	35	u8 buf, u16 buf_len, enum wnm_oper oper)
	36	{
	37	wpa_printf(MSG_DEBUG, "%s: TFS set operation %d", __func__, oper);
	38
	39	return hostapd_drv_wnm_oper(hapd, oper, addr, buf, buf_len);
	40	}
	41
	42
	43	/* MLME-SLEEPMODE.response */
	44	static int ieee802_11_send_wnmsleep_resp(struct hostapd_data *hapd,
	45	const u8 *addr, u8 dialog_token,
	46	u8 action_type, u16 intval)
	47	{
	48	struct ieee80211_mgmt *mgmt;
	49	int res;
	50	size_t len;
	51	size_t gtk_elem_len = 0;
	52	size_t igtk_elem_len = 0;
	53	struct wnm_sleep_element wnmsleep_ie;
	54	u8 *wnmtfs_ie;
	55	u8 wnmsleep_ie_len;
	56	u16 wnmtfs_ie_len;
	57	u8 *pos;
	58	struct sta_info *sta;
	59	enum wnm_oper tfs_oper = action_type == 0 ? WNM_SLEEP_TFS_RESP_IE_ADD :
	60	WNM_SLEEP_TFS_RESP_IE_NONE;
	61
	62	sta = ap_get_sta(hapd, addr);
	63	if (sta == NULL) {
	64	wpa_printf(MSG_DEBUG, "%s: station not found", __func__);
	65	return -EINVAL;
	66	}
	67
	68	/* WNM-Sleep Mode IE */
	69	os_memset(&wnmsleep_ie, 0, sizeof(struct wnm_sleep_element));
	70	wnmsleep_ie_len = sizeof(struct wnm_sleep_element);
	71	wnmsleep_ie.eid = WLAN_EID_WNMSLEEP;
	72	wnmsleep_ie.len = wnmsleep_ie_len - 2;
	73	wnmsleep_ie.action_type = action_type;
	74	wnmsleep_ie.status = WNM_STATUS_SLEEP_ACCEPT;
	75	wnmsleep_ie.intval = intval;
	76
	77	/* TFS IE(s) */
	78	wnmtfs_ie = os_zalloc(MAX_TFS_IE_LEN);
	79	if (wnmtfs_ie == NULL)
	80	return -1;
	81	if (ieee80211_11_get_tfs_ie(hapd, addr, wnmtfs_ie, &wnmtfs_ie_len,
	82	tfs_oper)) {
	83	wnmtfs_ie_len = 0;
	84	os_free(wnmtfs_ie);
	85	wnmtfs_ie = NULL;
86	}
87
88	#define MAX_GTK_SUBELEM_LEN 45
89	#define MAX_IGTK_SUBELEM_LEN 26
90	mgmt = os_zalloc(sizeof(*mgmt) + wnmsleep_ie_len +
91	MAX_GTK_SUBELEM_LEN + MAX_IGTK_SUBELEM_LEN);
92	if (mgmt == NULL) {
93	wpa_printf(MSG_DEBUG, "MLME: Failed to allocate buffer for "
94	"WNM-Sleep Response action frame");
95	return -1;
96	}
97	os_memcpy(mgmt->da, addr, ETH_ALEN);
98	os_memcpy(mgmt->sa, hapd->own_addr, ETH_ALEN);
99	os_memcpy(mgmt->bssid, hapd->own_addr, ETH_ALEN);
100	mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
101	WLAN_FC_STYPE_ACTION);
102	mgmt->u.action.category = WLAN_ACTION_WNM;
103	mgmt->u.action.u.wnm_sleep_resp.action = WNM_SLEEP_MODE_RESP;
104	mgmt->u.action.u.wnm_sleep_resp.dialogtoken = dialog_token;
105	pos = (u8 *)mgmt->u.action.u.wnm_sleep_resp.variable;
106	/* add key data if MFP is enabled */
107	if (wpa_auth_uses_mfp(sta->wpa_sm) \|\| action_type != 1){
108	mgmt->u.action.u.wnm_sleep_resp.keydata_len = 0;
109	} else {
110	gtk_elem_len = wpa_wnmsleep_gtk_subelem(sta->wpa_sm, pos);
111	pos += gtk_elem_len;
112	wpa_printf(MSG_DEBUG, "Pass 4, gtk_len = %d",
113	(int) gtk_elem_len);
114	#ifdef CONFIG_IEEE80211W
115	res = wpa_wnmsleep_igtk_subelem(sta->wpa_sm, pos);
116	if (res < 0) {
117	os_free(wnmtfs_ie);
118	os_free(mgmt);
119	return -1;
120	}
121	igtk_elem_len = res;
122	pos += igtk_elem_len;
123	wpa_printf(MSG_DEBUG, "Pass 4 igtk_len = %d",
124	(int) igtk_elem_len);
125	#endif /* CONFIG_IEEE80211W */
126
127	WPA_PUT_LE16((u8 *)
128	&mgmt->u.action.u.wnm_sleep_resp.keydata_len,
129	gtk_elem_len + igtk_elem_len);
130	}
131	os_memcpy(pos, &wnmsleep_ie, wnmsleep_ie_len);
132	/* copy TFS IE here */
133	pos += wnmsleep_ie_len;
134	os_memcpy(pos, wnmtfs_ie, wnmtfs_ie_len);
135
136	len = 1 + sizeof(mgmt->u.action.u.wnm_sleep_resp) + gtk_elem_len +
137	igtk_elem_len + wnmsleep_ie_len + wnmtfs_ie_len;
138
139	/* In driver, response frame should be forced to sent when STA is in
140	* PS mode */
141	res = hostapd_drv_send_action(hapd, hapd->iface->freq, 0,
142	mgmt->da, &mgmt->u.action.category, len);
143
144	if (!res) {
145	wpa_printf(MSG_DEBUG, "Successfully send WNM-Sleep Response "
146	"frame");
147
148	/* when entering wnmsleep
149	* 1. pause the node in driver
150	* 2. mark the node so that AP won't update GTK/IGTK during
151	* WNM Sleep
152	*/
153	if (wnmsleep_ie.status == WNM_STATUS_SLEEP_ACCEPT &&
154	wnmsleep_ie.action_type == 0) {
155	hostapd_drv_wnm_oper(hapd, WNM_SLEEP_ENTER_CONFIRM,
156	addr, NULL, NULL);
157	wpa_set_wnmsleep(sta->wpa_sm, 1);
158	}
159	/* when exiting wnmsleep
160	* 1. unmark the node
161	* 2. start GTK/IGTK update if MFP is not used
162	* 3. unpause the node in driver
163	*/
164	if (wnmsleep_ie.status == WNM_STATUS_SLEEP_ACCEPT &&
165	wnmsleep_ie.action_type == 1) {
166	wpa_set_wnmsleep(sta->wpa_sm, 0);
167	hostapd_drv_wnm_oper(hapd, WNM_SLEEP_EXIT_CONFIRM,
168	addr, NULL, NULL);
169	if (wpa_auth_uses_mfp(sta->wpa_sm) && action_type == 1)
170	wpa_wnmsleep_rekey_gtk(sta->wpa_sm);
171	}
172	} else
173	wpa_printf(MSG_DEBUG, "Fail to send WNM-Sleep Response frame");
174
175	#undef MAX_GTK_SUBELEM_LEN
176	#undef MAX_IGTK_SUBELEM_LEN
177	os_free(wnmtfs_ie);
178	os_free(mgmt);
179	return res;
180	}
181
182
183	static void ieee802_11_rx_wnmsleep_req(struct hostapd_data *hapd,
184	const u8 addr, const u8 frm, int len)
185	{
186	/*
187	* Action [1] \| Dialog Token [1] \| WNM-Sleep Mode IE \|
188	* TFS Response IE
189	*/
190	u8 pos = (u8 ) frm; /* point to action field */
191	u8 dialog_token = pos[1];
192	struct wnm_sleep_element *wnmsleep_ie = NULL;
193	/* multiple TFS Req IE (assuming consecutive) */
194	u8 *tfsreq_ie_start = NULL;
195	u8 *tfsreq_ie_end = NULL;
196	u16 tfsreq_ie_len = 0;
197
198	pos += 1 + 1;
199	while (pos - frm < len - 1) {
200	u8 ie_len = *(pos+1);
201	if (*pos == WLAN_EID_WNMSLEEP)
202	wnmsleep_ie = (struct wnm_sleep_element *)pos;
203	else if (*pos == WLAN_EID_TFS_REQ) {
204	if (!tfsreq_ie_start)
205	tfsreq_ie_start = pos;
206	tfsreq_ie_end = pos;
207	} else
208	wpa_printf(MSG_DEBUG, "EID %d not recognized", *pos);
209	pos += ie_len + 2;
210	}
211
212	if (!wnmsleep_ie) {
213	wpa_printf(MSG_DEBUG, "No WNM-Sleep IE found");
214	return;
215	}
216
217	if (wnmsleep_ie->action_type == 0 && tfsreq_ie_start &&
218	tfsreq_ie_end && tfsreq_ie_end - tfsreq_ie_start >= 0) {
219	tfsreq_ie_len = (tfsreq_ie_end + tfsreq_ie_end[1] + 2) -
220	tfsreq_ie_start;
221	wpa_printf(MSG_DEBUG, "TFS Req IE(s) found");
222	/* pass the TFS Req IE(s) to driver for processing */
223	if (ieee80211_11_set_tfs_ie(hapd, addr, tfsreq_ie_start,
224	&tfsreq_ie_len,
225	WNM_SLEEP_TFS_REQ_IE_SET))
226	wpa_printf(MSG_DEBUG, "Fail to set TFS Req IE");
227	}
228
229	ieee802_11_send_wnmsleep_resp(hapd, addr, dialog_token,
230	wnmsleep_ie->action_type,
231	wnmsleep_ie->intval);
232
233	if (wnmsleep_ie->action_type == 1) {
234	/* clear the tfs after sending the resp frame */
235	ieee80211_11_set_tfs_ie(hapd, addr, tfsreq_ie_start,
236	&tfsreq_ie_len, WNM_SLEEP_TFS_IE_DEL);
237	}
238	}
239
240
241	void ieee802_11_rx_wnm_action_ap(struct hostapd_data *hapd,
242	struct rx_action *action)
243	{
244	u8 pos = (u8 ) action->data + 1; /* point to the action field */
245	u8 act = *pos;
246
247	switch (act) {
248	case WNM_SLEEP_MODE_REQ:
249	ieee802_11_rx_wnmsleep_req(hapd, action->sa, action->data + 1,
250	action->len);
251	break;
252	default:
253	break;
254	}
255	}