WNM: Scan for BSSID if there are forbidden neighbors

The test to scan for a single BSSID assumed that there is only a single
neighbor in the candidate list. Also do this optimization if there are
multiple neighbors but only one of them is valid.

Signed-off-by: Benjamin Berg <benjamin.berg@intel.com>
This commit is contained in:
Benjamin Berg 2024-04-29 13:51:57 +02:00 committed by Jouni Malinen
parent 3cf4b4ad88
commit 9e50c12b31

View file

@ -1371,8 +1371,11 @@ static void wnm_set_scan_freqs(struct wpa_supplicant *wpa_s)
static int wnm_parse_candidate_list(struct wpa_supplicant *wpa_s, static int wnm_parse_candidate_list(struct wpa_supplicant *wpa_s,
const u8 *pos, const u8 *end) const u8 *pos, const u8 *end,
int *num_valid_candidates)
{ {
*num_valid_candidates = 0;
while (end - pos >= 2 && while (end - pos >= 2 &&
wpa_s->wnm_num_neighbor_report < WNM_MAX_NEIGHBOR_REPORT) { wpa_s->wnm_num_neighbor_report < WNM_MAX_NEIGHBOR_REPORT) {
u8 tag = *pos++; u8 tag = *pos++;
@ -1402,6 +1405,9 @@ static int wnm_parse_candidate_list(struct wpa_supplicant *wpa_s,
ether_addr_equal(rep->bssid, wpa_s->bssid)) ether_addr_equal(rep->bssid, wpa_s->bssid))
rep->disassoc_imminent = 1; rep->disassoc_imminent = 1;
if (rep->preference_present && rep->preference)
*num_valid_candidates += 1;
wpa_s->wnm_num_neighbor_report++; wpa_s->wnm_num_neighbor_report++;
#ifdef CONFIG_MBO #ifdef CONFIG_MBO
if (wpa_s->wnm_mbo_trans_reason_present && if (wpa_s->wnm_mbo_trans_reason_present &&
@ -1431,6 +1437,7 @@ static void ieee802_11_rx_bss_trans_mgmt_req(struct wpa_supplicant *wpa_s,
const u8 *vendor; const u8 *vendor;
#endif /* CONFIG_MBO */ #endif /* CONFIG_MBO */
bool disassoc_imminent; bool disassoc_imminent;
int num_valid_candidates;
if (wpa_s->disable_mbo_oce || wpa_s->conf->disable_btm) if (wpa_s->disable_mbo_oce || wpa_s->conf->disable_btm)
return; return;
@ -1564,7 +1571,8 @@ static void ieee802_11_rx_bss_trans_mgmt_req(struct wpa_supplicant *wpa_s,
wpas_mbo_ie_trans_req(wpa_s, vendor + 2, vendor[1]); wpas_mbo_ie_trans_req(wpa_s, vendor + 2, vendor[1]);
#endif /* CONFIG_MBO */ #endif /* CONFIG_MBO */
if (wnm_parse_candidate_list(wpa_s, pos, end) < 0) if (wnm_parse_candidate_list(wpa_s, pos, end,
&num_valid_candidates) < 0)
goto reset; goto reset;
if (wpa_s->wnm_mode & WNM_BSS_TM_REQ_PREF_CAND_LIST_INCLUDED) { if (wpa_s->wnm_mode & WNM_BSS_TM_REQ_PREF_CAND_LIST_INCLUDED) {
@ -1645,7 +1653,8 @@ static void ieee802_11_rx_bss_trans_mgmt_req(struct wpa_supplicant *wpa_s,
} }
wnm_set_scan_freqs(wpa_s); wnm_set_scan_freqs(wpa_s);
if (wpa_s->wnm_num_neighbor_report == 1) { if (num_valid_candidates == 1) {
/* Any invalid candidate was sorted to the end */
os_memcpy(wpa_s->next_scan_bssid, os_memcpy(wpa_s->next_scan_bssid,
wpa_s->wnm_neighbor_report_elements[0].bssid, wpa_s->wnm_neighbor_report_elements[0].bssid,
ETH_ALEN); ETH_ALEN);