TDLS: Copy peer's EHT capabilities

This is needed to be able to configure the STA entry into the driver
with the information for EHT.

Signed-off-by: Jouni Malinen <quic_klokere@quicinc.com>
This commit is contained in:
Kiran Kumar Lokere 2023-02-09 00:25:30 -08:00 committed by Jouni Malinen
parent c7561502f2
commit a41c8dbdd8

View file

@ -139,6 +139,8 @@ struct wpa_tdls_peer {
struct ieee80211_he_capabilities *he_capabilities;
size_t he_capab_len;
struct ieee80211_he_6ghz_band_cap *he_6ghz_band_capabilities;
struct ieee80211_eht_capabilities *eht_capabilities;
size_t eht_capab_len;
u8 qos_info;
@ -718,6 +720,8 @@ static void wpa_tdls_peer_clear(struct wpa_sm *sm, struct wpa_tdls_peer *peer)
peer->he_capabilities = NULL;
os_free(peer->he_6ghz_band_capabilities);
peer->he_6ghz_band_capabilities = NULL;
os_free(peer->eht_capabilities);
peer->eht_capabilities = NULL;
os_free(peer->ext_capab);
peer->ext_capab = NULL;
os_free(peer->supp_channels);
@ -1752,6 +1756,29 @@ static int copy_peer_ext_capab(const struct wpa_eapol_ie_parse *kde,
}
static int copy_peer_eht_capab(const struct wpa_eapol_ie_parse *kde,
struct wpa_tdls_peer *peer)
{
if (!kde->eht_capabilities) {
wpa_printf(MSG_DEBUG, "TDLS: No EHT capabilities received");
return 0;
}
os_free(peer->eht_capabilities);
peer->eht_capab_len = 0;
peer->eht_capabilities = os_memdup(kde->eht_capabilities,
kde->eht_capab_len);
if (!peer->eht_capabilities)
return -1;
peer->eht_capab_len = kde->eht_capab_len;
wpa_hexdump(MSG_DEBUG, "TDLS: Peer EHT capabilities",
peer->eht_capabilities, peer->eht_capab_len);
return 0;
}
static int copy_peer_wmm_capab(const struct wpa_eapol_ie_parse *kde,
struct wpa_tdls_peer *peer)
{
@ -1980,6 +2007,9 @@ static int wpa_tdls_process_tpk_m1(struct wpa_sm *sm, const u8 *src_addr,
copy_peer_he_6ghz_band_capab(&kde, peer) < 0)
goto error;
if (copy_peer_eht_capab(&kde, peer) < 0)
goto error;
if (copy_peer_ext_capab(&kde, peer) < 0)
goto error;
@ -2390,6 +2420,9 @@ static int wpa_tdls_process_tpk_m2(struct wpa_sm *sm, const u8 *src_addr,
copy_peer_he_6ghz_band_capab(&kde, peer) < 0)
goto error;
if (copy_peer_eht_capab(&kde, peer) < 0)
goto error;
if (copy_peer_ext_capab(&kde, peer) < 0)
goto error;