• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * hostapd - WNM
3  * Copyright (c) 2011-2014, 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 "utils/eloop.h"
13 #include "common/ieee802_11_defs.h"
14 #include "common/wpa_ctrl.h"
15 #include "common/ocv.h"
16 #include "ap/hostapd.h"
17 #include "ap/sta_info.h"
18 #include "ap/ap_config.h"
19 #include "ap/ap_drv_ops.h"
20 #include "ap/wpa_auth.h"
21 #include "mbo_ap.h"
22 #include "wnm_ap.h"
23 
24 #define MAX_TFS_IE_LEN  1024
25 
26 
27 /* get the TFS IE from driver */
ieee80211_11_get_tfs_ie(struct hostapd_data * hapd,const u8 * addr,u8 * buf,u16 * buf_len,enum wnm_oper oper)28 static int ieee80211_11_get_tfs_ie(struct hostapd_data *hapd, const u8 *addr,
29 				   u8 *buf, u16 *buf_len, enum wnm_oper oper)
30 {
31 	wpa_printf(MSG_DEBUG, "%s: TFS get operation %d", __func__, oper);
32 
33 	return hostapd_drv_wnm_oper(hapd, oper, addr, buf, buf_len);
34 }
35 
36 
37 /* set the TFS IE to driver */
ieee80211_11_set_tfs_ie(struct hostapd_data * hapd,const u8 * addr,u8 * buf,u16 * buf_len,enum wnm_oper oper)38 static int ieee80211_11_set_tfs_ie(struct hostapd_data *hapd, const u8 *addr,
39 				   u8 *buf, u16 *buf_len, enum wnm_oper oper)
40 {
41 	wpa_printf(MSG_DEBUG, "%s: TFS set operation %d", __func__, oper);
42 
43 	return hostapd_drv_wnm_oper(hapd, oper, addr, buf, buf_len);
44 }
45 
46 
wnm_ap_get_own_addr(struct hostapd_data * hapd,struct sta_info * sta)47 static const u8 * wnm_ap_get_own_addr(struct hostapd_data *hapd,
48 				      struct sta_info *sta)
49 {
50 	const u8 *own_addr = hapd->own_addr;
51 
52 #ifdef CONFIG_IEEE80211BE
53 	if (hapd->conf->mld_ap && (!sta || ap_sta_is_mld(hapd, sta)))
54 		own_addr = hapd->mld->mld_addr;
55 #endif /* CONFIG_IEEE80211BE */
56 
57 	return own_addr;
58 }
59 
60 
61 /* MLME-SLEEPMODE.response */
ieee802_11_send_wnmsleep_resp(struct hostapd_data * hapd,const u8 * addr,u8 dialog_token,u8 action_type,u16 intval)62 static int ieee802_11_send_wnmsleep_resp(struct hostapd_data *hapd,
63 					 const u8 *addr, u8 dialog_token,
64 					 u8 action_type, u16 intval)
65 {
66 	struct ieee80211_mgmt *mgmt;
67 	int res;
68 	size_t len;
69 	size_t gtk_elem_len = 0;
70 	size_t igtk_elem_len = 0;
71 	size_t bigtk_elem_len = 0;
72 	struct wnm_sleep_element wnmsleep_ie;
73 	u8 *wnmtfs_ie, *oci_ie;
74 	u8 wnmsleep_ie_len, oci_ie_len;
75 	u16 wnmtfs_ie_len;
76 	u8 *pos;
77 	struct sta_info *sta;
78 	enum wnm_oper tfs_oper = action_type == WNM_SLEEP_MODE_ENTER ?
79 		WNM_SLEEP_TFS_RESP_IE_ADD : WNM_SLEEP_TFS_RESP_IE_NONE;
80 	const u8 *own_addr;
81 
82 	sta = ap_get_sta(hapd, addr);
83 	if (sta == NULL) {
84 		wpa_printf(MSG_DEBUG, "%s: station not found", __func__);
85 		return -EINVAL;
86 	}
87 
88 	/* WNM-Sleep Mode IE */
89 	os_memset(&wnmsleep_ie, 0, sizeof(struct wnm_sleep_element));
90 	wnmsleep_ie_len = sizeof(struct wnm_sleep_element);
91 	wnmsleep_ie.eid = WLAN_EID_WNMSLEEP;
92 	wnmsleep_ie.len = wnmsleep_ie_len - 2;
93 	wnmsleep_ie.action_type = action_type;
94 	wnmsleep_ie.status = WNM_STATUS_SLEEP_ACCEPT;
95 	wnmsleep_ie.intval = host_to_le16(intval);
96 
97 	/* TFS IE(s) */
98 	wnmtfs_ie = os_zalloc(MAX_TFS_IE_LEN);
99 	if (wnmtfs_ie == NULL)
100 		return -1;
101 	if (ieee80211_11_get_tfs_ie(hapd, addr, wnmtfs_ie, &wnmtfs_ie_len,
102 				    tfs_oper)) {
103 		wnmtfs_ie_len = 0;
104 		os_free(wnmtfs_ie);
105 		wnmtfs_ie = NULL;
106 	}
107 
108 	oci_ie = NULL;
109 	oci_ie_len = 0;
110 #ifdef CONFIG_OCV
111 	if (action_type == WNM_SLEEP_MODE_EXIT &&
112 	    wpa_auth_uses_ocv(sta->wpa_sm)) {
113 		struct wpa_channel_info ci;
114 
115 		if (hostapd_drv_channel_info(hapd, &ci) != 0) {
116 			wpa_printf(MSG_WARNING,
117 				   "Failed to get channel info for OCI element in WNM-Sleep Mode frame");
118 			os_free(wnmtfs_ie);
119 			return -1;
120 		}
121 #ifdef CONFIG_TESTING_OPTIONS
122 		if (hapd->conf->oci_freq_override_wnm_sleep) {
123 			wpa_printf(MSG_INFO,
124 				   "TEST: Override OCI frequency %d -> %u MHz",
125 				   ci.frequency,
126 				   hapd->conf->oci_freq_override_wnm_sleep);
127 			ci.frequency = hapd->conf->oci_freq_override_wnm_sleep;
128 		}
129 #endif /* CONFIG_TESTING_OPTIONS */
130 
131 		oci_ie_len = OCV_OCI_EXTENDED_LEN;
132 		oci_ie = os_zalloc(oci_ie_len);
133 		if (!oci_ie) {
134 			wpa_printf(MSG_WARNING,
135 				   "Failed to allocate buffer for OCI element in WNM-Sleep Mode frame");
136 			os_free(wnmtfs_ie);
137 			return -1;
138 		}
139 
140 		if (ocv_insert_extended_oci(&ci, oci_ie) < 0) {
141 			os_free(wnmtfs_ie);
142 			os_free(oci_ie);
143 			return -1;
144 		}
145 	}
146 #endif /* CONFIG_OCV */
147 
148 #define MAX_GTK_SUBELEM_LEN 45
149 #define MAX_IGTK_SUBELEM_LEN 26
150 #define MAX_BIGTK_SUBELEM_LEN 26
151 	mgmt = os_zalloc(sizeof(*mgmt) + wnmsleep_ie_len +
152 			 MAX_GTK_SUBELEM_LEN + MAX_IGTK_SUBELEM_LEN +
153 			 MAX_BIGTK_SUBELEM_LEN +
154 			 oci_ie_len);
155 	if (mgmt == NULL) {
156 		wpa_printf(MSG_DEBUG, "MLME: Failed to allocate buffer for "
157 			   "WNM-Sleep Response action frame");
158 		res = -1;
159 		goto fail;
160 	}
161 
162 	own_addr = wnm_ap_get_own_addr(hapd, sta);
163 
164 	os_memcpy(mgmt->da, addr, ETH_ALEN);
165 	os_memcpy(mgmt->sa, own_addr, ETH_ALEN);
166 	os_memcpy(mgmt->bssid, own_addr, ETH_ALEN);
167 	mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
168 					   WLAN_FC_STYPE_ACTION);
169 	mgmt->u.action.category = WLAN_ACTION_WNM;
170 	mgmt->u.action.u.wnm_sleep_resp.action = WNM_SLEEP_MODE_RESP;
171 	mgmt->u.action.u.wnm_sleep_resp.dialogtoken = dialog_token;
172 	pos = (u8 *)mgmt->u.action.u.wnm_sleep_resp.variable;
173 	/* add key data if MFP is enabled */
174 	if (!wpa_auth_uses_mfp(sta->wpa_sm) ||
175 	    hapd->conf->wnm_sleep_mode_no_keys ||
176 	    action_type != WNM_SLEEP_MODE_EXIT) {
177 		mgmt->u.action.u.wnm_sleep_resp.keydata_len = 0;
178 	} else {
179 		gtk_elem_len = wpa_wnmsleep_gtk_subelem(sta->wpa_sm, pos);
180 		pos += gtk_elem_len;
181 		wpa_printf(MSG_DEBUG, "Pass 4, gtk_len = %d",
182 			   (int) gtk_elem_len);
183 		res = wpa_wnmsleep_igtk_subelem(sta->wpa_sm, pos);
184 		if (res < 0)
185 			goto fail;
186 		igtk_elem_len = res;
187 		pos += igtk_elem_len;
188 		wpa_printf(MSG_DEBUG, "Pass 4 igtk_len = %d",
189 			   (int) igtk_elem_len);
190 		if (hapd->conf->beacon_prot &&
191 		    (hapd->iface->drv_flags &
192 		     WPA_DRIVER_FLAGS_BEACON_PROTECTION)) {
193 			res = wpa_wnmsleep_bigtk_subelem(sta->wpa_sm, pos);
194 			if (res < 0)
195 				goto fail;
196 			bigtk_elem_len = res;
197 			pos += bigtk_elem_len;
198 			wpa_printf(MSG_DEBUG, "Pass 4 bigtk_len = %d",
199 				   (int) bigtk_elem_len);
200 		}
201 
202 		WPA_PUT_LE16((u8 *)
203 			     &mgmt->u.action.u.wnm_sleep_resp.keydata_len,
204 			     gtk_elem_len + igtk_elem_len + bigtk_elem_len);
205 	}
206 	os_memcpy(pos, &wnmsleep_ie, wnmsleep_ie_len);
207 	/* copy TFS IE here */
208 	pos += wnmsleep_ie_len;
209 	if (wnmtfs_ie) {
210 		os_memcpy(pos, wnmtfs_ie, wnmtfs_ie_len);
211 		pos += wnmtfs_ie_len;
212 	}
213 #ifdef CONFIG_OCV
214 	/* copy OCV OCI here */
215 	if (oci_ie_len > 0)
216 		os_memcpy(pos, oci_ie, oci_ie_len);
217 #endif /* CONFIG_OCV */
218 
219 	len = 1 + sizeof(mgmt->u.action.u.wnm_sleep_resp) + gtk_elem_len +
220 		igtk_elem_len + bigtk_elem_len +
221 		wnmsleep_ie_len + wnmtfs_ie_len + oci_ie_len;
222 
223 	/* In driver, response frame should be forced to sent when STA is in
224 	 * PS mode */
225 	res = hostapd_drv_send_action(hapd, hapd->iface->freq, 0,
226 				      mgmt->da, &mgmt->u.action.category, len);
227 
228 	if (!res) {
229 		wpa_printf(MSG_DEBUG, "Successfully send WNM-Sleep Response "
230 			   "frame");
231 
232 		/* when entering wnmsleep
233 		 * 1. pause the node in driver
234 		 * 2. mark the node so that AP won't update GTK/IGTK/BIGTK
235 		 * during WNM Sleep
236 		 */
237 		if (wnmsleep_ie.status == WNM_STATUS_SLEEP_ACCEPT &&
238 		    wnmsleep_ie.action_type == WNM_SLEEP_MODE_ENTER) {
239 			sta->flags |= WLAN_STA_WNM_SLEEP_MODE;
240 			hostapd_drv_wnm_oper(hapd, WNM_SLEEP_ENTER_CONFIRM,
241 					     addr, NULL, NULL);
242 			wpa_set_wnmsleep(sta->wpa_sm, 1);
243 		}
244 		/* when exiting wnmsleep
245 		 * 1. unmark the node
246 		 * 2. start GTK/IGTK/BIGTK update if MFP is not used
247 		 * 3. unpause the node in driver
248 		 */
249 		if ((wnmsleep_ie.status == WNM_STATUS_SLEEP_ACCEPT ||
250 		     wnmsleep_ie.status ==
251 		     WNM_STATUS_SLEEP_EXIT_ACCEPT_GTK_UPDATE) &&
252 		    wnmsleep_ie.action_type == WNM_SLEEP_MODE_EXIT) {
253 			sta->flags &= ~WLAN_STA_WNM_SLEEP_MODE;
254 			wpa_set_wnmsleep(sta->wpa_sm, 0);
255 			hostapd_drv_wnm_oper(hapd, WNM_SLEEP_EXIT_CONFIRM,
256 					     addr, NULL, NULL);
257 			if (!wpa_auth_uses_mfp(sta->wpa_sm) ||
258 			    hapd->conf->wnm_sleep_mode_no_keys)
259 				wpa_wnmsleep_rekey_gtk(sta->wpa_sm);
260 		}
261 	} else
262 		wpa_printf(MSG_DEBUG, "Fail to send WNM-Sleep Response frame");
263 
264 #undef MAX_GTK_SUBELEM_LEN
265 #undef MAX_IGTK_SUBELEM_LEN
266 #undef MAX_BIGTK_SUBELEM_LEN
267 fail:
268 	os_free(wnmtfs_ie);
269 	os_free(oci_ie);
270 	os_free(mgmt);
271 	return res;
272 }
273 
274 
ieee802_11_rx_wnmsleep_req(struct hostapd_data * hapd,const u8 * addr,const u8 * frm,int len)275 static void ieee802_11_rx_wnmsleep_req(struct hostapd_data *hapd,
276 				       const u8 *addr, const u8 *frm, int len)
277 {
278 	/* Dialog Token [1] | WNM-Sleep Mode IE | TFS Response IE */
279 	const u8 *pos = frm;
280 	u8 dialog_token;
281 	struct wnm_sleep_element *wnmsleep_ie = NULL;
282 	/* multiple TFS Req IE (assuming consecutive) */
283 	u8 *tfsreq_ie_start = NULL;
284 	u8 *tfsreq_ie_end = NULL;
285 	u16 tfsreq_ie_len = 0;
286 #ifdef CONFIG_OCV
287 	struct sta_info *sta;
288 	const u8 *oci_ie = NULL;
289 	u8 oci_ie_len = 0;
290 #endif /* CONFIG_OCV */
291 
292 	if (!hapd->conf->wnm_sleep_mode) {
293 		wpa_printf(MSG_DEBUG, "Ignore WNM-Sleep Mode Request from "
294 			   MACSTR_SEC " since WNM-Sleep Mode is disabled",
295 			   MAC2STR_SEC(addr));
296 		return;
297 	}
298 
299 	if (len < 1) {
300 		wpa_printf(MSG_DEBUG,
301 			   "WNM: Ignore too short WNM-Sleep Mode Request from "
302 			   MACSTR_SEC, MAC2STR_SEC(addr));
303 		return;
304 	}
305 
306 	dialog_token = *pos++;
307 	while (pos + 1 < frm + len) {
308 		u8 ie_len = pos[1];
309 		if (pos + 2 + ie_len > frm + len)
310 			break;
311 		if (*pos == WLAN_EID_WNMSLEEP &&
312 		    ie_len >= (int) sizeof(*wnmsleep_ie) - 2)
313 			wnmsleep_ie = (struct wnm_sleep_element *) pos;
314 		else if (*pos == WLAN_EID_TFS_REQ) {
315 			if (!tfsreq_ie_start)
316 				tfsreq_ie_start = (u8 *) pos;
317 			tfsreq_ie_end = (u8 *) pos;
318 #ifdef CONFIG_OCV
319 		} else if (*pos == WLAN_EID_EXTENSION && ie_len >= 1 &&
320 			   pos[2] == WLAN_EID_EXT_OCV_OCI) {
321 			oci_ie = pos + 3;
322 			oci_ie_len = ie_len - 1;
323 #endif /* CONFIG_OCV */
324 		} else
325 			wpa_printf(MSG_DEBUG, "WNM: EID %d not recognized",
326 				   *pos);
327 		pos += ie_len + 2;
328 	}
329 
330 	if (!wnmsleep_ie) {
331 		wpa_printf(MSG_DEBUG, "No WNM-Sleep IE found");
332 		return;
333 	}
334 
335 #ifdef CONFIG_OCV
336 	sta = ap_get_sta(hapd, addr);
337 	if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_EXIT &&
338 	    sta && wpa_auth_uses_ocv(sta->wpa_sm)) {
339 		struct wpa_channel_info ci;
340 
341 		if (hostapd_drv_channel_info(hapd, &ci) != 0) {
342 			wpa_printf(MSG_WARNING,
343 				   "Failed to get channel info to validate received OCI in WNM-Sleep Mode frame");
344 			return;
345 		}
346 
347 		if (ocv_verify_tx_params(oci_ie, oci_ie_len, &ci,
348 					 channel_width_to_int(ci.chanwidth),
349 					 ci.seg1_idx) != OCI_SUCCESS) {
350 			wpa_msg(hapd, MSG_WARNING, "WNM: OCV failed: %s",
351 				ocv_errorstr);
352 			return;
353 		}
354 	}
355 #endif /* CONFIG_OCV */
356 
357 	if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_ENTER &&
358 	    tfsreq_ie_start && tfsreq_ie_end &&
359 	    tfsreq_ie_end - tfsreq_ie_start >= 0) {
360 		tfsreq_ie_len = (tfsreq_ie_end + tfsreq_ie_end[1] + 2) -
361 			tfsreq_ie_start;
362 		wpa_printf(MSG_DEBUG, "TFS Req IE(s) found");
363 		/* pass the TFS Req IE(s) to driver for processing */
364 		if (ieee80211_11_set_tfs_ie(hapd, addr, tfsreq_ie_start,
365 					    &tfsreq_ie_len,
366 					    WNM_SLEEP_TFS_REQ_IE_SET))
367 			wpa_printf(MSG_DEBUG, "Fail to set TFS Req IE");
368 	}
369 
370 	ieee802_11_send_wnmsleep_resp(hapd, addr, dialog_token,
371 				      wnmsleep_ie->action_type,
372 				      le_to_host16(wnmsleep_ie->intval));
373 
374 	if (wnmsleep_ie->action_type == WNM_SLEEP_MODE_EXIT) {
375 		/* clear the tfs after sending the resp frame */
376 		ieee80211_11_set_tfs_ie(hapd, addr, tfsreq_ie_start,
377 					&tfsreq_ie_len, WNM_SLEEP_TFS_IE_DEL);
378 	}
379 }
380 
381 
ieee802_11_send_bss_trans_mgmt_request(struct hostapd_data * hapd,const u8 * addr,u8 dialog_token)382 static int ieee802_11_send_bss_trans_mgmt_request(struct hostapd_data *hapd,
383 						  const u8 *addr,
384 						  u8 dialog_token)
385 {
386 	struct ieee80211_mgmt *mgmt;
387 	const u8 *own_addr;
388 	struct sta_info *sta;
389 	size_t len;
390 	u8 *pos;
391 	int res;
392 
393 	mgmt = os_zalloc(sizeof(*mgmt));
394 	if (mgmt == NULL)
395 		return -1;
396 
397 	sta = ap_get_sta(hapd, addr);
398 	own_addr = wnm_ap_get_own_addr(hapd, sta);
399 
400 	os_memcpy(mgmt->da, addr, ETH_ALEN);
401 	os_memcpy(mgmt->sa, own_addr, ETH_ALEN);
402 	os_memcpy(mgmt->bssid, own_addr, ETH_ALEN);
403 	mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
404 					   WLAN_FC_STYPE_ACTION);
405 	mgmt->u.action.category = WLAN_ACTION_WNM;
406 	mgmt->u.action.u.bss_tm_req.action = WNM_BSS_TRANS_MGMT_REQ;
407 	mgmt->u.action.u.bss_tm_req.dialog_token = dialog_token;
408 	mgmt->u.action.u.bss_tm_req.req_mode = 0;
409 	mgmt->u.action.u.bss_tm_req.disassoc_timer = host_to_le16(0);
410 	mgmt->u.action.u.bss_tm_req.validity_interval = 1;
411 	pos = mgmt->u.action.u.bss_tm_req.variable;
412 
413 	wpa_printf(MSG_DEBUG, "WNM: Send BSS Transition Management Request to "
414 		   MACSTR_SEC " dialog_token=%u req_mode=0x%x disassoc_timer=%u "
415 		   "validity_interval=%u",
416 		   MAC2STR_SEC(addr), dialog_token,
417 		   mgmt->u.action.u.bss_tm_req.req_mode,
418 		   le_to_host16(mgmt->u.action.u.bss_tm_req.disassoc_timer),
419 		   mgmt->u.action.u.bss_tm_req.validity_interval);
420 
421 	len = pos - &mgmt->u.action.category;
422 	res = hostapd_drv_send_action(hapd, hapd->iface->freq, 0,
423 				      mgmt->da, &mgmt->u.action.category, len);
424 	os_free(mgmt);
425 	return res;
426 }
427 
428 
ieee802_11_rx_bss_trans_mgmt_query(struct hostapd_data * hapd,const u8 * addr,const u8 * frm,size_t len)429 static void ieee802_11_rx_bss_trans_mgmt_query(struct hostapd_data *hapd,
430 					       const u8 *addr, const u8 *frm,
431 					       size_t len)
432 {
433 	u8 dialog_token, reason;
434 	const u8 *pos, *end;
435 	int enabled = hapd->conf->bss_transition;
436 	char *hex = NULL;
437 	size_t hex_len;
438 
439 #ifdef CONFIG_MBO
440 	if (hapd->conf->mbo_enabled)
441 		enabled = 1;
442 #endif /* CONFIG_MBO */
443 	if (!enabled) {
444 		wpa_printf(MSG_DEBUG,
445 			   "Ignore BSS Transition Management Query from "
446 			   MACSTR_SEC
447 			   " since BSS Transition Management is disabled",
448 			   MAC2STR_SEC(addr));
449 		return;
450 	}
451 
452 	if (len < 2) {
453 		wpa_printf(MSG_DEBUG, "WNM: Ignore too short BSS Transition Management Query from "
454 			   MACSTR_SEC, MAC2STR_SEC(addr));
455 		return;
456 	}
457 
458 	pos = frm;
459 	end = pos + len;
460 	dialog_token = *pos++;
461 	reason = *pos++;
462 
463 	wpa_printf(MSG_DEBUG, "WNM: BSS Transition Management Query from "
464 		   MACSTR_SEC " dialog_token=%u reason=%u",
465 		   MAC2STR_SEC(addr), dialog_token, reason);
466 
467 	wpa_hexdump(MSG_DEBUG, "WNM: BSS Transition Candidate List Entries",
468 		    pos, end - pos);
469 
470 	hex_len = 2 * (end - pos) + 1;
471 	if (hex_len > 1) {
472 		hex = os_malloc(hex_len);
473 		if (hex)
474 			wpa_snprintf_hex(hex, hex_len, pos, end - pos);
475 	}
476 	wpa_msg(hapd->msg_ctx, MSG_INFO,
477 		BSS_TM_QUERY MACSTR " reason=%u%s%s",
478 		MAC2STR(addr), reason, hex ? " neighbor=" : "", hex);
479 	os_free(hex);
480 
481 	ieee802_11_send_bss_trans_mgmt_request(hapd, addr, dialog_token);
482 }
483 
484 
ap_sta_reset_steer_flag_timer(void * eloop_ctx,void * timeout_ctx)485 void ap_sta_reset_steer_flag_timer(void *eloop_ctx, void *timeout_ctx)
486 {
487 	struct hostapd_data *hapd = eloop_ctx;
488 	struct sta_info *sta = timeout_ctx;
489 
490 	if (sta->agreed_to_steer) {
491 		wpa_printf(MSG_DEBUG, "%s: Reset steering flag for STA " MACSTR_SEC,
492 			   hapd->conf->iface, MAC2STR_SEC(sta->addr));
493 		sta->agreed_to_steer = 0;
494 	}
495 }
496 
497 
ieee802_11_rx_bss_trans_mgmt_resp(struct hostapd_data * hapd,const u8 * addr,const u8 * frm,size_t len)498 static void ieee802_11_rx_bss_trans_mgmt_resp(struct hostapd_data *hapd,
499 					      const u8 *addr, const u8 *frm,
500 					      size_t len)
501 {
502 	u8 dialog_token, status_code, bss_termination_delay;
503 	const u8 *pos, *end;
504 	int enabled = hapd->conf->bss_transition;
505 	struct sta_info *sta;
506 
507 #ifdef CONFIG_MBO
508 	if (hapd->conf->mbo_enabled)
509 		enabled = 1;
510 #endif /* CONFIG_MBO */
511 	if (!enabled) {
512 		wpa_printf(MSG_DEBUG,
513 			   "Ignore BSS Transition Management Response from "
514 			   MACSTR_SEC
515 			   " since BSS Transition Management is disabled",
516 			   MAC2STR_SEC(addr));
517 		return;
518 	}
519 
520 	if (len < 3) {
521 		wpa_printf(MSG_DEBUG, "WNM: Ignore too short BSS Transition Management Response from "
522 			   MACSTR_SEC, MAC2STR_SEC(addr));
523 		return;
524 	}
525 
526 	pos = frm;
527 	end = pos + len;
528 	dialog_token = *pos++;
529 	status_code = *pos++;
530 	bss_termination_delay = *pos++;
531 
532 	wpa_printf(MSG_DEBUG, "WNM: BSS Transition Management Response from "
533 		   MACSTR_SEC " dialog_token=%u status_code=%u "
534 		   "bss_termination_delay=%u", MAC2STR_SEC(addr), dialog_token,
535 		   status_code, bss_termination_delay);
536 
537 	sta = ap_get_sta(hapd, addr);
538 	if (!sta) {
539 		wpa_printf(MSG_DEBUG, "Station " MACSTR_SEC
540 			   " not found for received BSS TM Response",
541 			   MAC2STR_SEC(addr));
542 		return;
543 	}
544 
545 	if (status_code == WNM_BSS_TM_ACCEPT) {
546 		if (end - pos < ETH_ALEN) {
547 			wpa_printf(MSG_DEBUG, "WNM: not enough room for Target BSSID field");
548 			return;
549 		}
550 		sta->agreed_to_steer = 1;
551 		eloop_cancel_timeout(ap_sta_reset_steer_flag_timer, hapd, sta);
552 		eloop_register_timeout(2, 0, ap_sta_reset_steer_flag_timer,
553 				       hapd, sta);
554 		wpa_printf(MSG_DEBUG, "WNM: Target BSSID: " MACSTR_SEC,
555 			   MAC2STR_SEC(pos));
556 		wpa_msg_only_for_cb(hapd->msg_ctx, MSG_INFO, BSS_TM_RESP MACSTR
557 			" status_code=%u bss_termination_delay=%u target_bssid="
558 			MACSTR,
559 			MAC2STR(addr), status_code, bss_termination_delay,
560 			MAC2STR(pos));
561 		wpa_printf(MSG_INFO, BSS_TM_RESP MACSTR_SEC
562 			" status_code=%u bss_termination_delay=%u target_bssid="
563 			MACSTR_SEC,
564 			MAC2STR_SEC(addr), status_code, bss_termination_delay,
565 			MAC2STR_SEC(pos));
566 		pos += ETH_ALEN;
567 	} else {
568 		sta->agreed_to_steer = 0;
569 		wpa_msg(hapd->msg_ctx, MSG_INFO, BSS_TM_RESP MACSTR
570 			" status_code=%u bss_termination_delay=%u",
571 			MAC2STR(addr), status_code, bss_termination_delay);
572 	}
573 
574 	wpa_hexdump(MSG_DEBUG, "WNM: BSS Transition Candidate List Entries",
575 		    pos, end - pos);
576 }
577 
578 
wnm_beacon_protection_failure(struct hostapd_data * hapd,const u8 * addr)579 static void wnm_beacon_protection_failure(struct hostapd_data *hapd,
580 					  const u8 *addr)
581 {
582 	struct sta_info *sta;
583 
584 	if (!hapd->conf->beacon_prot ||
585 	    !(hapd->iface->drv_flags & WPA_DRIVER_FLAGS_BEACON_PROTECTION))
586 		return;
587 
588 	sta = ap_get_sta(hapd, addr);
589 	if (!sta || !(sta->flags & WLAN_STA_AUTHORIZED)) {
590 		wpa_printf(MSG_DEBUG, "Station " MACSTR_SEC
591 			   " not found for received WNM-Notification Request",
592 			   MAC2STR_SEC(addr));
593 		return;
594 	}
595 
596 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
597 		       HOSTAPD_LEVEL_INFO,
598 		       "Beacon protection failure reported");
599 	wpa_msg(hapd->msg_ctx, MSG_INFO, WPA_EVENT_UNPROT_BEACON "reporter="
600 		MACSTR, MAC2STR(addr));
601 }
602 
603 
ieee802_11_rx_wnm_notification_req(struct hostapd_data * hapd,const u8 * addr,const u8 * buf,size_t len)604 static void ieee802_11_rx_wnm_notification_req(struct hostapd_data *hapd,
605 					       const u8 *addr, const u8 *buf,
606 					       size_t len)
607 {
608 	u8 dialog_token, type;
609 
610 	if (len < 2)
611 		return;
612 	dialog_token = *buf++;
613 	type = *buf++;
614 	len -= 2;
615 
616 	wpa_printf(MSG_DEBUG,
617 		   "WNM: Received WNM Notification Request frame from "
618 		   MACSTR_SEC " (dialog_token=%u type=%u)",
619 		   MAC2STR_SEC(addr), dialog_token, type);
620 	wpa_hexdump(MSG_MSGDUMP, "WNM: Notification Request subelements",
621 		    buf, len);
622 	switch (type) {
623 	case WNM_NOTIF_TYPE_BEACON_PROTECTION_FAILURE:
624 		wnm_beacon_protection_failure(hapd, addr);
625 		break;
626 	case WNM_NOTIF_TYPE_VENDOR_SPECIFIC:
627 		mbo_ap_wnm_notification_req(hapd, addr, buf, len);
628 		break;
629 	}
630 }
631 
632 
ieee802_11_rx_wnm_coloc_intf_report(struct hostapd_data * hapd,const u8 * addr,const u8 * buf,size_t len)633 static void ieee802_11_rx_wnm_coloc_intf_report(struct hostapd_data *hapd,
634 						const u8 *addr, const u8 *buf,
635 						size_t len)
636 {
637 	u8 dialog_token;
638 	char *hex;
639 	size_t hex_len;
640 
641 	if (!hapd->conf->coloc_intf_reporting) {
642 		wpa_printf(MSG_DEBUG,
643 			   "WNM: Ignore unexpected Collocated Interference Report from "
644 			   MACSTR_SEC, MAC2STR_SEC(addr));
645 		return;
646 	}
647 
648 	if (len < 1) {
649 		wpa_printf(MSG_DEBUG,
650 			   "WNM: Ignore too short Collocated Interference Report from "
651 			   MACSTR_SEC, MAC2STR_SEC(addr));
652 		return;
653 	}
654 	dialog_token = *buf++;
655 	len--;
656 
657 	wpa_printf(MSG_DEBUG,
658 		   "WNM: Received Collocated Interference Report frame from "
659 		   MACSTR_SEC " (dialog_token=%u)",
660 		   MAC2STR_SEC(addr), dialog_token);
661 	wpa_hexdump(MSG_MSGDUMP, "WNM: Collocated Interference Report Elements",
662 		    buf, len);
663 
664 	hex_len = 2 * len + 1;
665 	hex = os_malloc(hex_len);
666 	if (!hex)
667 		return;
668 	wpa_snprintf_hex(hex, hex_len, buf, len);
669 	wpa_msg_ctrl(hapd->msg_ctx, MSG_INFO, COLOC_INTF_REPORT MACSTR " %d %s",
670 		MAC2STR(addr), dialog_token, hex);
671 	os_free(hex);
672 }
673 
674 
675 
wnm_event_type2str(enum wnm_event_report_type wtype)676 static const char * wnm_event_type2str(enum wnm_event_report_type wtype)
677 {
678 #define W2S(wtype) case WNM_EVENT_TYPE_ ## wtype: return #wtype;
679 	switch (wtype) {
680 	W2S(TRANSITION)
681 	W2S(RSNA)
682 	W2S(P2P_LINK)
683 	W2S(WNM_LOG)
684 	W2S(BSS_COLOR_COLLISION)
685 	W2S(BSS_COLOR_IN_USE)
686 	}
687 	return "UNKNOWN";
688 #undef W2S
689 }
690 
691 
ieee802_11_rx_wnm_event_report(struct hostapd_data * hapd,const u8 * addr,const u8 * buf,size_t len)692 static void ieee802_11_rx_wnm_event_report(struct hostapd_data *hapd,
693 					   const u8 *addr, const u8 *buf,
694 					   size_t len)
695 {
696 	struct sta_info *sta;
697 	u8 dialog_token;
698 	struct wnm_event_report_element *report_ie;
699 	const u8 *pos = buf, *end = buf + len;
700 	const size_t fixed_field_len = 3; /* Event Token/Type/Report Status */
701 #ifdef CONFIG_IEEE80211AX
702 	const size_t tsf_len = 8;
703 	u8 color;
704 	u64 bitmap;
705 #endif /* CONFIG_IEEE80211AX */
706 
707 	if (end - pos < 1 + 2) {
708 		wpa_printf(MSG_DEBUG,
709 			   "WNM: Ignore too short WNM Event Report frame from "
710 			   MACSTR, MAC2STR(addr));
711 		return;
712 	}
713 
714 	dialog_token = *pos++;
715 	report_ie = (struct wnm_event_report_element *) pos;
716 
717 	if (end - pos < 2 + report_ie->len ||
718 	    report_ie->len < fixed_field_len) {
719 		wpa_printf(MSG_DEBUG,
720 			   "WNM: Ignore truncated WNM Event Report frame from "
721 			   MACSTR, MAC2STR(addr));
722 		return;
723 	}
724 
725 	if (report_ie->eid != WLAN_EID_EVENT_REPORT ||
726 	    report_ie->status != WNM_STATUS_SUCCESSFUL)
727 		return;
728 
729 	wpa_printf(MSG_DEBUG, "WNM: Received WNM Event Report frame from "
730 		   MACSTR " dialog_token=%u event_token=%u type=%d (%s)",
731 		   MAC2STR(addr), dialog_token, report_ie->token,
732 		   report_ie->type, wnm_event_type2str(report_ie->type));
733 
734 	pos += 2 + fixed_field_len;
735 	wpa_hexdump(MSG_MSGDUMP, "WNM: Event Report", pos, end - pos);
736 
737 	sta = ap_get_sta(hapd, addr);
738 	if (!sta || !(sta->flags & WLAN_STA_ASSOC)) {
739 		wpa_printf(MSG_DEBUG, "Station " MACSTR
740 			   " not found for received WNM Event Report",
741 			   MAC2STR(addr));
742 		return;
743 	}
744 
745 	switch (report_ie->type) {
746 #ifdef CONFIG_IEEE80211AX
747 	case WNM_EVENT_TYPE_BSS_COLOR_COLLISION:
748 		if (!hapd->iconf->ieee80211ax || hapd->conf->disable_11ax)
749 			return;
750 		if (report_ie->len <
751 		    fixed_field_len + tsf_len + 8) {
752 			wpa_printf(MSG_DEBUG,
753 				   "WNM: Too short BSS color collision event report from "
754 				   MACSTR, MAC2STR(addr));
755 			return;
756 		}
757 		bitmap = WPA_GET_LE64(report_ie->u.bss_color_collision.color_bitmap);
758 		wpa_printf(MSG_DEBUG,
759 			   "WNM: BSS color collision bitmap 0x%llx reported by "
760 			   MACSTR, (unsigned long long) bitmap, MAC2STR(addr));
761 		hostapd_switch_color(hapd->iface->bss[0], bitmap);
762 		break;
763 	case WNM_EVENT_TYPE_BSS_COLOR_IN_USE:
764 		if (!hapd->iconf->ieee80211ax || hapd->conf->disable_11ax)
765 			return;
766 		if (report_ie->len < fixed_field_len + tsf_len + 1) {
767 			wpa_printf(MSG_DEBUG,
768 				   "WNM: Too short BSS color in use event report from "
769 				   MACSTR, MAC2STR(addr));
770 			return;
771 		}
772 		color = report_ie->u.bss_color_in_use.color;
773 		if (color > 63) {
774 			wpa_printf(MSG_DEBUG,
775 				   "WNM: Invalid BSS color %u report from "
776 				   MACSTR, color, MAC2STR(addr));
777 			return;
778 		}
779 		if (color == 0) {
780 			wpa_printf(MSG_DEBUG,
781 				   "WNM: BSS color use report canceled by "
782 				   MACSTR, MAC2STR(addr));
783 			/* TODO: Clear stored color from the collision bitmap
784 			 * if there are no other users for it. */
785 			return;
786 		}
787 		wpa_printf(MSG_DEBUG, "WNM: BSS color %u use report by "
788 			   MACSTR, color, MAC2STR(addr));
789 		hapd->color_collision_bitmap |= 1ULL << color;
790 		break;
791 #endif /* CONFIG_IEEE80211AX */
792 	default:
793 		wpa_printf(MSG_DEBUG,
794 			   "WNM Event Report type=%d (%s) not supported",
795 			   report_ie->type,
796 			   wnm_event_type2str(report_ie->type));
797 		break;
798 	}
799 }
800 
801 
ieee802_11_rx_wnm_action_ap(struct hostapd_data * hapd,const struct ieee80211_mgmt * mgmt,size_t len)802 int ieee802_11_rx_wnm_action_ap(struct hostapd_data *hapd,
803 				const struct ieee80211_mgmt *mgmt, size_t len)
804 {
805 	u8 action;
806 	const u8 *payload;
807 	size_t plen;
808 
809 	if (len < IEEE80211_HDRLEN + 2)
810 		return -1;
811 
812 	payload = ((const u8 *) mgmt) + IEEE80211_HDRLEN + 1;
813 	action = *payload++;
814 	plen = len - IEEE80211_HDRLEN - 2;
815 
816 	switch (action) {
817 	case WNM_EVENT_REPORT:
818 		ieee802_11_rx_wnm_event_report(hapd, mgmt->sa, payload,
819 					       plen);
820 		return 0;
821 	case WNM_BSS_TRANS_MGMT_QUERY:
822 		ieee802_11_rx_bss_trans_mgmt_query(hapd, mgmt->sa, payload,
823 						   plen);
824 		return 0;
825 	case WNM_BSS_TRANS_MGMT_RESP:
826 		ieee802_11_rx_bss_trans_mgmt_resp(hapd, mgmt->sa, payload,
827 						  plen);
828 		return 0;
829 	case WNM_SLEEP_MODE_REQ:
830 		ieee802_11_rx_wnmsleep_req(hapd, mgmt->sa, payload, plen);
831 		return 0;
832 	case WNM_NOTIFICATION_REQ:
833 		ieee802_11_rx_wnm_notification_req(hapd, mgmt->sa, payload,
834 						   plen);
835 		return 0;
836 	case WNM_COLLOCATED_INTERFERENCE_REPORT:
837 		ieee802_11_rx_wnm_coloc_intf_report(hapd, mgmt->sa, payload,
838 						    plen);
839 		return 0;
840 	}
841 
842 	wpa_printf(MSG_DEBUG, "WNM: Unsupported WNM Action %u from " MACSTR_SEC,
843 		   action, MAC2STR_SEC(mgmt->sa));
844 	return -1;
845 }
846 
847 
wnm_send_disassoc_imminent(struct hostapd_data * hapd,struct sta_info * sta,int disassoc_timer)848 int wnm_send_disassoc_imminent(struct hostapd_data *hapd,
849 			       struct sta_info *sta, int disassoc_timer)
850 {
851 	u8 buf[1000], *pos;
852 	struct ieee80211_mgmt *mgmt;
853 	const u8 *own_addr = wnm_ap_get_own_addr(hapd, sta);
854 
855 	os_memset(buf, 0, sizeof(buf));
856 	mgmt = (struct ieee80211_mgmt *) buf;
857 	mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
858 					   WLAN_FC_STYPE_ACTION);
859 	os_memcpy(mgmt->da, sta->addr, ETH_ALEN);
860 	os_memcpy(mgmt->sa, own_addr, ETH_ALEN);
861 	os_memcpy(mgmt->bssid, own_addr, ETH_ALEN);
862 	mgmt->u.action.category = WLAN_ACTION_WNM;
863 	mgmt->u.action.u.bss_tm_req.action = WNM_BSS_TRANS_MGMT_REQ;
864 	mgmt->u.action.u.bss_tm_req.dialog_token = 1;
865 	mgmt->u.action.u.bss_tm_req.req_mode =
866 		WNM_BSS_TM_REQ_DISASSOC_IMMINENT;
867 	mgmt->u.action.u.bss_tm_req.disassoc_timer =
868 		host_to_le16(disassoc_timer);
869 	mgmt->u.action.u.bss_tm_req.validity_interval = 0;
870 
871 	pos = mgmt->u.action.u.bss_tm_req.variable;
872 
873 	wpa_printf(MSG_DEBUG, "WNM: Send BSS Transition Management Request frame to indicate imminent disassociation (disassoc_timer=%d) to "
874 		   MACSTR_SEC, disassoc_timer, MAC2STR_SEC(sta->addr));
875 	if (hostapd_drv_send_mlme(hapd, buf, pos - buf, 0, NULL, 0, 0) < 0) {
876 		wpa_printf(MSG_DEBUG, "Failed to send BSS Transition "
877 			   "Management Request frame");
878 		return -1;
879 	}
880 
881 	return 0;
882 }
883 
884 
set_disassoc_timer(struct hostapd_data * hapd,struct sta_info * sta,int disassoc_timer)885 static void set_disassoc_timer(struct hostapd_data *hapd, struct sta_info *sta,
886 			       int disassoc_timer)
887 {
888 	int timeout, beacon_int;
889 
890 	/*
891 	 * Prevent STA from reconnecting using cached PMKSA to force
892 	 * full authentication with the authentication server (which may
893 	 * decide to reject the connection),
894 	 */
895 	wpa_auth_pmksa_remove(hapd->wpa_auth, sta->addr);
896 
897 	beacon_int = hapd->iconf->beacon_int;
898 	if (beacon_int < 1)
899 		beacon_int = 100; /* best guess */
900 	/* Calculate timeout in ms based on beacon_int in TU */
901 	timeout = disassoc_timer * beacon_int * 128 / 125;
902 	wpa_printf(MSG_DEBUG, "Disassociation timer for " MACSTR_SEC
903 		   " set to %d ms", MAC2STR_SEC(sta->addr), timeout);
904 
905 	sta->timeout_next = STA_DISASSOC_FROM_CLI;
906 	eloop_cancel_timeout(ap_handle_timer, hapd, sta);
907 	eloop_register_timeout(timeout / 1000,
908 			       timeout % 1000 * 1000,
909 			       ap_handle_timer, hapd, sta);
910 }
911 
912 
wnm_send_ess_disassoc_imminent(struct hostapd_data * hapd,struct sta_info * sta,const char * url,int disassoc_timer)913 int wnm_send_ess_disassoc_imminent(struct hostapd_data *hapd,
914 				   struct sta_info *sta, const char *url,
915 				   int disassoc_timer)
916 {
917 	u8 buf[1000], *pos;
918 	struct ieee80211_mgmt *mgmt;
919 	size_t url_len;
920 	const u8 *own_addr = wnm_ap_get_own_addr(hapd, sta);
921 
922 	os_memset(buf, 0, sizeof(buf));
923 	mgmt = (struct ieee80211_mgmt *) buf;
924 	mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
925 					   WLAN_FC_STYPE_ACTION);
926 	os_memcpy(mgmt->da, sta->addr, ETH_ALEN);
927 	os_memcpy(mgmt->sa, own_addr, ETH_ALEN);
928 	os_memcpy(mgmt->bssid, own_addr, ETH_ALEN);
929 	mgmt->u.action.category = WLAN_ACTION_WNM;
930 	mgmt->u.action.u.bss_tm_req.action = WNM_BSS_TRANS_MGMT_REQ;
931 	mgmt->u.action.u.bss_tm_req.dialog_token = 1;
932 	mgmt->u.action.u.bss_tm_req.req_mode =
933 		WNM_BSS_TM_REQ_DISASSOC_IMMINENT |
934 		WNM_BSS_TM_REQ_ESS_DISASSOC_IMMINENT;
935 	mgmt->u.action.u.bss_tm_req.disassoc_timer =
936 		host_to_le16(disassoc_timer);
937 	mgmt->u.action.u.bss_tm_req.validity_interval = 0x01;
938 
939 	pos = mgmt->u.action.u.bss_tm_req.variable;
940 
941 	/* Session Information URL */
942 	url_len = os_strlen(url);
943 	if (url_len > 255)
944 		return -1;
945 	*pos++ = url_len;
946 	os_memcpy(pos, url, url_len);
947 	pos += url_len;
948 
949 	if (hostapd_drv_send_mlme(hapd, buf, pos - buf, 0, NULL, 0, 0) < 0) {
950 		wpa_printf(MSG_DEBUG, "Failed to send BSS Transition "
951 			   "Management Request frame");
952 		return -1;
953 	}
954 
955 	if (disassoc_timer) {
956 		/* send disassociation frame after time-out */
957 		set_disassoc_timer(hapd, sta, disassoc_timer);
958 	}
959 
960 	return 0;
961 }
962 
963 
wnm_send_bss_tm_req(struct hostapd_data * hapd,struct sta_info * sta,u8 req_mode,int disassoc_timer,u8 valid_int,const u8 * bss_term_dur,u8 dialog_token,const char * url,const u8 * nei_rep,size_t nei_rep_len,const u8 * mbo_attrs,size_t mbo_len)964 int wnm_send_bss_tm_req(struct hostapd_data *hapd, struct sta_info *sta,
965 			u8 req_mode, int disassoc_timer, u8 valid_int,
966 			const u8 *bss_term_dur, u8 dialog_token,
967 			const char *url, const u8 *nei_rep, size_t nei_rep_len,
968 			const u8 *mbo_attrs, size_t mbo_len)
969 {
970 	u8 *buf, *pos;
971 	struct ieee80211_mgmt *mgmt;
972 	size_t url_len;
973 	const u8 *own_addr = wnm_ap_get_own_addr(hapd, sta);
974 
975 	wpa_printf(MSG_DEBUG, "WNM: Send BSS Transition Management Request to "
976 		   MACSTR_SEC
977 		   " req_mode=0x%x disassoc_timer=%d valid_int=0x%x dialog_token=%u",
978 		   MAC2STR_SEC(sta->addr), req_mode, disassoc_timer, valid_int,
979 		   dialog_token);
980 	buf = os_zalloc(1000 + nei_rep_len + mbo_len);
981 	if (buf == NULL)
982 		return -1;
983 	mgmt = (struct ieee80211_mgmt *) buf;
984 	mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
985 					   WLAN_FC_STYPE_ACTION);
986 	os_memcpy(mgmt->da, sta->addr, ETH_ALEN);
987 	os_memcpy(mgmt->sa, own_addr, ETH_ALEN);
988 	os_memcpy(mgmt->bssid, own_addr, ETH_ALEN);
989 	mgmt->u.action.category = WLAN_ACTION_WNM;
990 	mgmt->u.action.u.bss_tm_req.action = WNM_BSS_TRANS_MGMT_REQ;
991 	mgmt->u.action.u.bss_tm_req.dialog_token = dialog_token;
992 	mgmt->u.action.u.bss_tm_req.req_mode = req_mode;
993 	mgmt->u.action.u.bss_tm_req.disassoc_timer =
994 		host_to_le16(disassoc_timer);
995 	mgmt->u.action.u.bss_tm_req.validity_interval = valid_int;
996 
997 	pos = mgmt->u.action.u.bss_tm_req.variable;
998 
999 	if ((req_mode & WNM_BSS_TM_REQ_BSS_TERMINATION_INCLUDED) &&
1000 	    bss_term_dur) {
1001 		os_memcpy(pos, bss_term_dur, 12);
1002 		pos += 12;
1003 	}
1004 
1005 	if (url) {
1006 		/* Session Information URL */
1007 		url_len = os_strlen(url);
1008 		if (url_len > 255) {
1009 			os_free(buf);
1010 			return -1;
1011 		}
1012 
1013 		*pos++ = url_len;
1014 		os_memcpy(pos, url, url_len);
1015 		pos += url_len;
1016 	}
1017 
1018 	if (nei_rep) {
1019 		os_memcpy(pos, nei_rep, nei_rep_len);
1020 		pos += nei_rep_len;
1021 	}
1022 
1023 	if (mbo_len > 0) {
1024 		pos += mbo_add_ie(pos, buf + sizeof(buf) - pos, mbo_attrs,
1025 				  mbo_len);
1026 	}
1027 
1028 	if (hostapd_drv_send_mlme(hapd, buf, pos - buf, 0, NULL, 0, 0) < 0) {
1029 		wpa_printf(MSG_DEBUG,
1030 			   "Failed to send BSS Transition Management Request frame");
1031 		os_free(buf);
1032 		return -1;
1033 	}
1034 	os_free(buf);
1035 
1036 	if (disassoc_timer) {
1037 #ifdef CONFIG_IEEE80211BE
1038 		if (ap_sta_is_mld(hapd, sta)) {
1039 			int i;
1040 			unsigned int links = 0;
1041 
1042 			for (i = 0; i < MAX_NUM_MLD_LINKS; i++) {
1043 				if (sta->mld_info.links[i].valid)
1044 					links++;
1045 			}
1046 
1047 			if (links > 1) {
1048 				wpa_printf(MSG_DEBUG,
1049 					   "WNM: Only terminating one link - other links remains associated for "
1050 					   MACSTR,
1051 					   MAC2STR(sta->mld_info.common_info.mld_addr));
1052 				return 0;
1053 			}
1054 		}
1055 #endif /* CONFIG_IEEE80211BE */
1056 
1057 		/* send disassociation frame after time-out */
1058 		set_disassoc_timer(hapd, sta, disassoc_timer);
1059 	}
1060 
1061 	return 0;
1062 }
1063 
1064 
wnm_send_coloc_intf_req(struct hostapd_data * hapd,struct sta_info * sta,unsigned int auto_report,unsigned int timeout)1065 int wnm_send_coloc_intf_req(struct hostapd_data *hapd, struct sta_info *sta,
1066 			    unsigned int auto_report, unsigned int timeout)
1067 {
1068 	u8 buf[100], *pos;
1069 	struct ieee80211_mgmt *mgmt;
1070 	u8 dialog_token = 1;
1071 	const u8 *own_addr = wnm_ap_get_own_addr(hapd, sta);
1072 
1073 	if (auto_report > 3 || timeout > 63)
1074 		return -1;
1075 	os_memset(buf, 0, sizeof(buf));
1076 	mgmt = (struct ieee80211_mgmt *) buf;
1077 	mgmt->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1078 					   WLAN_FC_STYPE_ACTION);
1079 	os_memcpy(mgmt->da, sta->addr, ETH_ALEN);
1080 	os_memcpy(mgmt->sa, own_addr, ETH_ALEN);
1081 	os_memcpy(mgmt->bssid, own_addr, ETH_ALEN);
1082 	mgmt->u.action.category = WLAN_ACTION_WNM;
1083 	mgmt->u.action.u.coloc_intf_req.action =
1084 		WNM_COLLOCATED_INTERFERENCE_REQ;
1085 	mgmt->u.action.u.coloc_intf_req.dialog_token = dialog_token;
1086 	mgmt->u.action.u.coloc_intf_req.req_info = auto_report | (timeout << 2);
1087 	pos = &mgmt->u.action.u.coloc_intf_req.req_info;
1088 	pos++;
1089 
1090 	wpa_printf(MSG_DEBUG, "WNM: Sending Collocated Interference Request to "
1091 		   MACSTR_SEC " (dialog_token=%u auto_report=%u timeout=%u)",
1092 		   MAC2STR_SEC(sta->addr), dialog_token, auto_report, timeout);
1093 	if (hostapd_drv_send_mlme(hapd, buf, pos - buf, 0, NULL, 0, 0) < 0) {
1094 		wpa_printf(MSG_DEBUG,
1095 			   "WNM: Failed to send Collocated Interference Request frame");
1096 		return -1;
1097 	}
1098 
1099 	return 0;
1100 }
1101