u32 cap);
 u8 *ieee80211_ie_build_vht_oper(u8 *pos, struct ieee80211_sta_vht_cap *vht_cap,
                                const struct cfg80211_chan_def *chandef);
-u8 ieee80211_ie_len_he_cap(struct ieee80211_sub_if_data *sdata, u8 iftype);
+u8 ieee80211_ie_len_he_cap(struct ieee80211_sub_if_data *sdata);
 u8 *ieee80211_ie_build_he_cap(const struct ieee80211_conn_settings *conn,
                              const struct ieee80211_sta_he_cap *he_cap,
                              u8 *pos, u8 *end);
 void ieee80211_init_frag_cache(struct ieee80211_fragment_cache *cache);
 void ieee80211_destroy_frag_cache(struct ieee80211_fragment_cache *cache);
 
-u8 ieee80211_ie_len_eht_cap(struct ieee80211_sub_if_data *sdata, u8 iftype);
+u8 ieee80211_ie_len_eht_cap(struct ieee80211_sub_if_data *sdata);
 u8 *ieee80211_ie_build_eht_cap(u8 *pos,
                               const struct ieee80211_sta_he_cap *he_cap,
                               const struct ieee80211_sta_eht_cap *eht_cap,
 
 // SPDX-License-Identifier: GPL-2.0-only
 /*
  * Copyright (c) 2008, 2009 open80211s Ltd.
- * Copyright (C) 2018 - 2023 Intel Corporation
+ * Copyright (C) 2018 - 2024 Intel Corporation
  * Authors:    Luis Carlos Cobo <luisca@cozybit.com>
  *            Javier Cardona <javier@cozybit.com>
  */
        band = chanctx_conf->def.chan->band;
        rcu_read_unlock();
 
-       ie_len_he_cap = ieee80211_ie_len_he_cap(sdata,
-                                               NL80211_IFTYPE_MESH_POINT);
-       ie_len_eht_cap = ieee80211_ie_len_eht_cap(sdata,
-                                                 NL80211_IFTYPE_MESH_POINT);
+       ie_len_he_cap = ieee80211_ie_len_he_cap(sdata);
+       ie_len_eht_cap = ieee80211_ie_len_eht_cap(sdata);
        head_len = hdr_len +
                   2 + /* NULL SSID */
                   /* Channel Switch Announcement */
 
 // SPDX-License-Identifier: GPL-2.0-only
 /*
  * Copyright (c) 2008, 2009 open80211s Ltd.
- * Copyright (C) 2019, 2021-2023 Intel Corporation
+ * Copyright (C) 2019, 2021-2024 Intel Corporation
  * Author:     Luis Carlos Cobo <luisca@cozybit.com>
  */
 #include <linux/gfp.h>
        int hdr_len = offsetofend(struct ieee80211_mgmt, u.action.u.self_prot);
        int err = -ENOMEM;
 
-       ie_len_he_cap = ieee80211_ie_len_he_cap(sdata,
-                                               NL80211_IFTYPE_MESH_POINT);
-       ie_len_eht_cap = ieee80211_ie_len_eht_cap(sdata,
-                                                 NL80211_IFTYPE_MESH_POINT);
+       ie_len_he_cap = ieee80211_ie_len_he_cap(sdata);
+       ie_len_eht_cap = ieee80211_ie_len_eht_cap(sdata);
        skb = dev_alloc_skb(local->tx_headroom +
                            hdr_len +
                            2 + /* capability info */
 
        return pos;
 }
 
-u8 ieee80211_ie_len_he_cap(struct ieee80211_sub_if_data *sdata, u8 iftype)
+/* this may return more than ieee80211_ie_build_he_cap() will need */
+u8 ieee80211_ie_len_he_cap(struct ieee80211_sub_if_data *sdata)
 {
        const struct ieee80211_sta_he_cap *he_cap;
        struct ieee80211_supported_band *sband;
        if (!sband)
                return 0;
 
-       he_cap = ieee80211_get_he_iftype_cap(sband, iftype);
+       he_cap = ieee80211_get_he_iftype_cap_vif(sband, &sdata->vif);
        if (!he_cap)
                return 0;
 
        return (u16) listen_interval;
 }
 
-u8 ieee80211_ie_len_eht_cap(struct ieee80211_sub_if_data *sdata, u8 iftype)
+/* this may return more than ieee80211_ie_build_eht_cap() will need */
+u8 ieee80211_ie_len_eht_cap(struct ieee80211_sub_if_data *sdata)
 {
        const struct ieee80211_sta_he_cap *he_cap;
        const struct ieee80211_sta_eht_cap *eht_cap;
        if (!sband)
                return 0;
 
-       he_cap = ieee80211_get_he_iftype_cap(sband, iftype);
-       eht_cap = ieee80211_get_eht_iftype_cap(sband, iftype);
+       he_cap = ieee80211_get_he_iftype_cap_vif(sband, &sdata->vif);
+       eht_cap = ieee80211_get_eht_iftype_cap_vif(sband, &sdata->vif);
        if (!he_cap || !eht_cap)
                return 0;
 
-       is_ap = iftype == NL80211_IFTYPE_AP ||
-               iftype == NL80211_IFTYPE_P2P_GO;
+       is_ap = sdata->vif.type == NL80211_IFTYPE_AP;
 
        n = ieee80211_eht_mcs_nss_size(&he_cap->he_cap_elem,
                                       &eht_cap->eht_cap_elem,