// SPDX-License-Identifier: GPL-2.0-only
/*
 * HT handling
 *
 * Copyright 2003, Jouni Malinen <jkmaline@cc.hut.fi>
 * Copyright 2002-2005, Instant802 Networks, Inc.
 * Copyright 2005-2006, Devicescape Software, Inc.
 * Copyright 2006-2007 Jiri Benc <jbenc@suse.cz>
 * Copyright 2007, Michael Wu <flamingice@sourmilk.net>
 * Copyright 2007-2010, Intel Corporation
 * Copyright 2017 Intel Deutschland GmbH
 * Copyright(c) 2020-2025 Intel Corporation
 */

#include <linux/ieee80211.h>
#include <linux/export.h>
#include <net/mac80211.h>
#include "ieee80211_i.h"
#include "rate.h"

static void __check_htcap_disable(struct ieee80211_ht_cap *ht_capa,
      struct ieee80211_ht_cap *ht_capa_mask,
      struct ieee80211_sta_ht_cap *ht_cap,
      u16 flag)
{
 __le16 le_flag = cpu_to_le16(flag);
 if (ht_capa_mask->cap_info & le_flag) {
  if (!(ht_capa->cap_info & le_flag))
   ht_cap->cap &= ~flag;
 }
}

static void __check_htcap_enable(struct ieee80211_ht_cap *ht_capa,
      struct ieee80211_ht_cap *ht_capa_mask,
      struct ieee80211_sta_ht_cap *ht_cap,
      u16 flag)
{
 __le16 le_flag = cpu_to_le16(flag);

 if ((ht_capa_mask->cap_info & le_flag) &&
     (ht_capa->cap_info & le_flag))
  ht_cap->cap |= flag;
}

void ieee80211_apply_htcap_overrides(struct ieee80211_sub_if_data *sdata,
         struct ieee80211_sta_ht_cap *ht_cap)
{
 struct ieee80211_ht_cap *ht_capa, *ht_capa_mask;
 u8 *scaps, *smask;
 int i;

 if (!ht_cap->ht_supported)
  return;

 switch (sdata->vif.type) {
 case NL80211_IFTYPE_STATION:
  ht_capa = &sdata->u.mgd.ht_capa;
  ht_capa_mask = &sdata->u.mgd.ht_capa_mask;
  break;
 case NL80211_IFTYPE_ADHOC:
  ht_capa = &sdata->u.ibss.ht_capa;
  ht_capa_mask = &sdata->u.ibss.ht_capa_mask;
  break;
 default:
  WARN_ON_ONCE(1);
  return;
 }

 scaps = (u8 *)(&ht_capa->mcs.rx_mask);
 smask = (u8 *)(&ht_capa_mask->mcs.rx_mask);

 /* NOTE:  If you add more over-rides here, update register_hw
  * ht_capa_mod_mask logic in main.c as well.
  * And, if this method can ever change ht_cap.ht_supported, fix
  * the check in ieee80211_add_ht_ie.
 */

 /* check for HT over-rides, MCS rates first. */
 for (i = 0; i < IEEE80211_HT_MCS_MASK_LEN; i++) {
  u8 m = smask[i];
  ht_cap->mcs.rx_mask[i] &= ~m; /* turn off all masked bits */
  /* Add back rates that are supported */
  ht_cap->mcs.rx_mask[i] |= (m & scaps[i]);
 }

 /* Force removal of HT-40 capabilities? */
 __check_htcap_disable(ht_capa, ht_capa_mask, ht_cap,
         IEEE80211_HT_CAP_SUP_WIDTH_20_40);
 __check_htcap_disable(ht_capa, ht_capa_mask, ht_cap,
         IEEE80211_HT_CAP_SGI_40);

 /* Allow user to disable SGI-20 (SGI-40 is handled above) */
 __check_htcap_disable(ht_capa, ht_capa_mask, ht_cap,
         IEEE80211_HT_CAP_SGI_20);

 /* Allow user to disable the max-AMSDU bit. */
 __check_htcap_disable(ht_capa, ht_capa_mask, ht_cap,
         IEEE80211_HT_CAP_MAX_AMSDU);

 /* Allow user to disable LDPC */
 __check_htcap_disable(ht_capa, ht_capa_mask, ht_cap,
         IEEE80211_HT_CAP_LDPC_CODING);

 /* Allow user to enable 40 MHz intolerant bit. */
 __check_htcap_enable(ht_capa, ht_capa_mask, ht_cap,
        IEEE80211_HT_CAP_40MHZ_INTOLERANT);

 /* Allow user to enable TX STBC bit  */
 __check_htcap_enable(ht_capa, ht_capa_mask, ht_cap,
        IEEE80211_HT_CAP_TX_STBC);

 /* Allow user to configure RX STBC bits */
 if (ht_capa_mask->cap_info & cpu_to_le16(IEEE80211_HT_CAP_RX_STBC))
  ht_cap->cap |= le16_to_cpu(ht_capa->cap_info) &
     IEEE80211_HT_CAP_RX_STBC;

 /* Allow user to decrease AMPDU factor */
 if (ht_capa_mask->ampdu_params_info &
     IEEE80211_HT_AMPDU_PARM_FACTOR) {
  u8 n = ht_capa->ampdu_params_info &
         IEEE80211_HT_AMPDU_PARM_FACTOR;
  if (n < ht_cap->ampdu_factor)
   ht_cap->ampdu_factor = n;
 }

 /* Allow the user to increase AMPDU density. */
 if (ht_capa_mask->ampdu_params_info &
     IEEE80211_HT_AMPDU_PARM_DENSITY) {
  u8 n = (ht_capa->ampdu_params_info &
   IEEE80211_HT_AMPDU_PARM_DENSITY)
   >> IEEE80211_HT_AMPDU_PARM_DENSITY_SHIFT;
  if (n > ht_cap->ampdu_density)
   ht_cap->ampdu_density = n;
 }
}


bool ieee80211_ht_cap_ie_to_sta_ht_cap(struct ieee80211_sub_if_data *sdata,
           struct ieee80211_supported_band *sband,
           const struct ieee80211_ht_cap *ht_cap_ie,
           struct link_sta_info *link_sta)
{
 struct ieee80211_bss_conf *link_conf;
 struct sta_info *sta = link_sta->sta;
 struct ieee80211_sta_ht_cap ht_cap, own_cap;
 u8 ampdu_info, tx_mcs_set_cap;
 int i, max_tx_streams;
 bool changed;
 enum ieee80211_sta_rx_bandwidth bw;
 enum nl80211_chan_width width;

 memset(&ht_cap, 0, sizeof(ht_cap));

 if (!ht_cap_ie || !sband->ht_cap.ht_supported)
  goto apply;

 ht_cap.ht_supported = true;

 own_cap = sband->ht_cap;

 /*
  * If user has specified capability over-rides, take care
  * of that if the station we're setting up is the AP or TDLS peer that
  * we advertised a restricted capability set to. Override
  * our own capabilities and then use those below.
 */
 if (sdata->vif.type == NL80211_IFTYPE_STATION ||
     sdata->vif.type == NL80211_IFTYPE_ADHOC)
  ieee80211_apply_htcap_overrides(sdata, &own_cap);

 /*
  * The bits listed in this expression should be
  * the same for the peer and us, if the station
  * advertises more then we can't use those thus
  * we mask them out.
 */
 ht_cap.cap = le16_to_cpu(ht_cap_ie->cap_info) &
  (own_cap.cap | ~(IEEE80211_HT_CAP_LDPC_CODING |
     IEEE80211_HT_CAP_SUP_WIDTH_20_40 |
     IEEE80211_HT_CAP_GRN_FLD |
     IEEE80211_HT_CAP_SGI_20 |
     IEEE80211_HT_CAP_SGI_40 |
     IEEE80211_HT_CAP_DSSSCCK40));

 /*
  * The STBC bits are asymmetric -- if we don't have
  * TX then mask out the peer's RX and vice versa.
 */
 if (!(own_cap.cap & IEEE80211_HT_CAP_TX_STBC))
  ht_cap.cap &= ~IEEE80211_HT_CAP_RX_STBC;
 if (!(own_cap.cap & IEEE80211_HT_CAP_RX_STBC))
  ht_cap.cap &= ~IEEE80211_HT_CAP_TX_STBC;

 ampdu_info = ht_cap_ie->ampdu_params_info;
 ht_cap.ampdu_factor =
  ampdu_info & IEEE80211_HT_AMPDU_PARM_FACTOR;
 ht_cap.ampdu_density =
  (ampdu_info & IEEE80211_HT_AMPDU_PARM_DENSITY) >> 2;

 /* own MCS TX capabilities */
 tx_mcs_set_cap = own_cap.mcs.tx_params;

 /* Copy peer MCS TX capabilities, the driver might need them. */
 ht_cap.mcs.tx_params = ht_cap_ie->mcs.tx_params;

 /* can we TX with MCS rates? */
 if (!(tx_mcs_set_cap & IEEE80211_HT_MCS_TX_DEFINED))
  goto apply;

 /* Counting from 0, therefore +1 */
 if (tx_mcs_set_cap & IEEE80211_HT_MCS_TX_RX_DIFF)
  max_tx_streams =
   ((tx_mcs_set_cap & IEEE80211_HT_MCS_TX_MAX_STREAMS_MASK)
    >> IEEE80211_HT_MCS_TX_MAX_STREAMS_SHIFT) + 1;
 else
  max_tx_streams = IEEE80211_HT_MCS_TX_MAX_STREAMS;

 /*
  * 802.11n-2009 20.3.5 / 20.6 says:
  * - indices 0 to 7 and 32 are single spatial stream
  * - 8 to 31 are multiple spatial streams using equal modulation
  *   [8..15 for two streams, 16..23 for three and 24..31 for four]
  * - remainder are multiple spatial streams using unequal modulation
 */
 for (i = 0; i < max_tx_streams; i++)
  ht_cap.mcs.rx_mask[i] =
   own_cap.mcs.rx_mask[i] & ht_cap_ie->mcs.rx_mask[i];

 if (tx_mcs_set_cap & IEEE80211_HT_MCS_TX_UNEQUAL_MODULATION)
  for (i = IEEE80211_HT_MCS_UNEQUAL_MODULATION_START_BYTE;
       i < IEEE80211_HT_MCS_MASK_LEN; i++)
   ht_cap.mcs.rx_mask[i] =
    own_cap.mcs.rx_mask[i] &
     ht_cap_ie->mcs.rx_mask[i];

 /* handle MCS rate 32 too */
 if (own_cap.mcs.rx_mask[32/8] & ht_cap_ie->mcs.rx_mask[32/8] & 1)
  ht_cap.mcs.rx_mask[32/8] |= 1;

 /* set Rx highest rate */
 ht_cap.mcs.rx_highest = ht_cap_ie->mcs.rx_highest;

 if (ht_cap.cap & IEEE80211_HT_CAP_MAX_AMSDU)
  link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_HT_7935;
 else
  link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_HT_3839;

 ieee80211_sta_recalc_aggregates(&sta->sta);

 apply:
 changed = memcmp(&link_sta->pub->ht_cap, &ht_cap, sizeof(ht_cap));

 memcpy(&link_sta->pub->ht_cap, &ht_cap, sizeof(ht_cap));

 rcu_read_lock();
 link_conf = rcu_dereference(sdata->vif.link_conf[link_sta->link_id]);
 if (WARN_ON(!link_conf))
  width = NL80211_CHAN_WIDTH_20_NOHT;
 else
  width = link_conf->chanreq.oper.width;

 switch (width) {
 default:
  WARN_ON_ONCE(1);
  fallthrough;
 case NL80211_CHAN_WIDTH_20_NOHT:
 case NL80211_CHAN_WIDTH_20:
  bw = IEEE80211_STA_RX_BW_20;
  break;
 case NL80211_CHAN_WIDTH_40:
 case NL80211_CHAN_WIDTH_80:
 case NL80211_CHAN_WIDTH_80P80:
 case NL80211_CHAN_WIDTH_160:
 case NL80211_CHAN_WIDTH_320:
  bw = ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
    IEEE80211_STA_RX_BW_40 : IEEE80211_STA_RX_BW_20;
  break;
 }
 rcu_read_unlock();

 link_sta->pub->bandwidth = bw;

 link_sta->cur_max_bandwidth =
  ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
    IEEE80211_STA_RX_BW_40 : IEEE80211_STA_RX_BW_20;

 if (sta->sdata->vif.type == NL80211_IFTYPE_AP ||
     sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
  enum ieee80211_smps_mode smps_mode;

  switch ((ht_cap.cap & IEEE80211_HT_CAP_SM_PS)
    >> IEEE80211_HT_CAP_SM_PS_SHIFT) {
  case WLAN_HT_CAP_SM_PS_INVALID:
  case WLAN_HT_CAP_SM_PS_STATIC:
   smps_mode = IEEE80211_SMPS_STATIC;
   break;
  case WLAN_HT_CAP_SM_PS_DYNAMIC:
   smps_mode = IEEE80211_SMPS_DYNAMIC;
   break;
  case WLAN_HT_CAP_SM_PS_DISABLED:
   smps_mode = IEEE80211_SMPS_OFF;
   break;
  }

  if (smps_mode != link_sta->pub->smps_mode)
   changed = true;
  link_sta->pub->smps_mode = smps_mode;
 } else {
  link_sta->pub->smps_mode = IEEE80211_SMPS_OFF;
 }

 return changed;
}

void ieee80211_sta_tear_down_BA_sessions(struct sta_info *sta,
      enum ieee80211_agg_stop_reason reason)
{
 int i;

 lockdep_assert_wiphy(sta->local->hw.wiphy);

 for (i = 0; i <  IEEE80211_NUM_TIDS; i++)
  __ieee80211_stop_rx_ba_session(sta, i, WLAN_BACK_RECIPIENT,
            WLAN_REASON_QSTA_LEAVE_QBSS,
            reason != AGG_STOP_DESTROY_STA &&
            reason != AGG_STOP_PEER_REQUEST);

 for (i = 0; i <  IEEE80211_NUM_TIDS; i++)
  __ieee80211_stop_tx_ba_session(sta, i, reason);

 /*
  * In case the tear down is part of a reconfigure due to HW restart
  * request, it is possible that the low level driver requested to stop
  * the BA session, so handle it to properly clean tid_tx data.
 */
 if(reason == AGG_STOP_DESTROY_STA) {
  wiphy_work_cancel(sta->local->hw.wiphy, &sta->ampdu_mlme.work);

  for (i = 0; i < IEEE80211_NUM_TIDS; i++) {
   struct tid_ampdu_tx *tid_tx =
    rcu_dereference_protected_tid_tx(sta, i);

   if (!tid_tx)
    continue;

   if (test_and_clear_bit(HT_AGG_STATE_STOP_CB, &tid_tx->state))
    ieee80211_stop_tx_ba_cb(sta, i, tid_tx);
  }
 }
}

void ieee80211_ba_session_work(struct wiphy *wiphy, struct wiphy_work *work)
{
 struct sta_info *sta =
  container_of(work, struct sta_info, ampdu_mlme.work);
 struct tid_ampdu_tx *tid_tx;
 bool blocked;
 int tid;

 lockdep_assert_wiphy(sta->local->hw.wiphy);

 /* When this flag is set, new sessions should be blocked. */
 blocked = test_sta_flag(sta, WLAN_STA_BLOCK_BA);

 for (tid = 0; tid < IEEE80211_NUM_TIDS; tid++) {
  if (test_and_clear_bit(tid, sta->ampdu_mlme.tid_rx_timer_expired))
   __ieee80211_stop_rx_ba_session(
    sta, tid, WLAN_BACK_RECIPIENT,
    WLAN_REASON_QSTA_TIMEOUT, true);

  if (test_and_clear_bit(tid,
           sta->ampdu_mlme.tid_rx_stop_requested))
   __ieee80211_stop_rx_ba_session(
    sta, tid, WLAN_BACK_RECIPIENT,
    WLAN_REASON_UNSPECIFIED, true);

  if (!blocked &&
      test_and_clear_bit(tid,
           sta->ampdu_mlme.tid_rx_manage_offl))
   __ieee80211_start_rx_ba_session(sta, 0, 0, 0, 1, tid,
       IEEE80211_MAX_AMPDU_BUF_HT,
       false, true, 0);

  if (test_and_clear_bit(tid + IEEE80211_NUM_TIDS,
           sta->ampdu_mlme.tid_rx_manage_offl))
   __ieee80211_stop_rx_ba_session(
    sta, tid, WLAN_BACK_RECIPIENT,
    0, false);

  spin_lock_bh(&sta->lock);

  tid_tx = sta->ampdu_mlme.tid_start_tx[tid];
  if (!blocked && tid_tx) {
   struct txq_info *txqi = to_txq_info(sta->sta.txq[tid]);
   struct ieee80211_sub_if_data *sdata =
    vif_to_sdata(txqi->txq.vif);
   struct fq *fq = &sdata->local->fq;

   spin_lock_bh(&fq->lock);

   /* Allow only frags to be dequeued */
   set_bit(IEEE80211_TXQ_STOP, &txqi->flags);

   if (!skb_queue_empty(&txqi->frags)) {
    /* Fragmented Tx is ongoing, wait for it to
     * finish. Reschedule worker to retry later.
 */

    spin_unlock_bh(&fq->lock);
    spin_unlock_bh(&sta->lock);

    /* Give the task working on the txq a chance
     * to send out the queued frags
 */
    synchronize_net();

    wiphy_work_queue(sdata->local->hw.wiphy, work);
    return;
   }

   spin_unlock_bh(&fq->lock);

   /*
    * Assign it over to the normal tid_tx array
    * where it "goes live".
 */

   sta->ampdu_mlme.tid_start_tx[tid] = NULL;
   /* could there be a race? */
   if (sta->ampdu_mlme.tid_tx[tid])
    kfree(tid_tx);
   else
    ieee80211_assign_tid_tx(sta, tid, tid_tx);
   spin_unlock_bh(&sta->lock);

   ieee80211_tx_ba_session_handle_start(sta, tid);
   continue;
  }
  spin_unlock_bh(&sta->lock);

  tid_tx = rcu_dereference_protected_tid_tx(sta, tid);
  if (!tid_tx)
   continue;

  if (!blocked &&
      test_and_clear_bit(HT_AGG_STATE_START_CB, &tid_tx->state))
   ieee80211_start_tx_ba_cb(sta, tid, tid_tx);
  if (test_and_clear_bit(HT_AGG_STATE_WANT_STOP, &tid_tx->state))
   __ieee80211_stop_tx_ba_session(sta, tid,
             AGG_STOP_LOCAL_REQUEST);
  if (test_and_clear_bit(HT_AGG_STATE_STOP_CB, &tid_tx->state))
   ieee80211_stop_tx_ba_cb(sta, tid, tid_tx);
 }
}

void ieee80211_send_delba(struct ieee80211_sub_if_data *sdata,
     const u8 *da, u16 tid,
     u16 initiator, u16 reason_code)
{
 struct ieee80211_local *local = sdata->local;
 struct sk_buff *skb;
 struct ieee80211_mgmt *mgmt;
 u16 params;

 skb = dev_alloc_skb(sizeof(*mgmt) + local->hw.extra_tx_headroom);
 if (!skb)
  return;

 skb_reserve(skb, local->hw.extra_tx_headroom);
 mgmt = ieee80211_mgmt_ba(skb, da, sdata);

 skb_put(skb, 1 + sizeof(mgmt->u.action.u.delba));

 mgmt->u.action.category = WLAN_CATEGORY_BACK;
 mgmt->u.action.u.delba.action_code = WLAN_ACTION_DELBA;
 params = (u16)(initiator << 11);  /* bit 11 initiator */
 params |= (u16)(tid << 12);   /* bit 15:12 TID number */

 mgmt->u.action.u.delba.params = cpu_to_le16(params);
 mgmt->u.action.u.delba.reason_code = cpu_to_le16(reason_code);

 ieee80211_tx_skb(sdata, skb);
}

void ieee80211_process_delba(struct ieee80211_sub_if_data *sdata,
        struct sta_info *sta,
        struct ieee80211_mgmt *mgmt, size_t len)
{
 u16 tid, params;
 u16 initiator;

 params = le16_to_cpu(mgmt->u.action.u.delba.params);
 tid = (params & IEEE80211_DELBA_PARAM_TID_MASK) >> 12;
 initiator = (params & IEEE80211_DELBA_PARAM_INITIATOR_MASK) >> 11;

 ht_dbg_ratelimited(sdata, "delba from %pM (%s) tid %d reason code %d\n",
      mgmt->sa, initiator ? "initiator" : "recipient",
      tid,
      le16_to_cpu(mgmt->u.action.u.delba.reason_code));

 if (initiator == WLAN_BACK_INITIATOR)
  __ieee80211_stop_rx_ba_session(sta, tid, WLAN_BACK_INITIATOR, 0,
            true);
 else
  __ieee80211_stop_tx_ba_session(sta, tid, AGG_STOP_PEER_REQUEST);
}

enum nl80211_smps_mode
ieee80211_smps_mode_to_smps_mode(enum ieee80211_smps_mode smps)
{
 switch (smps) {
 case IEEE80211_SMPS_OFF:
  return NL80211_SMPS_OFF;
 case IEEE80211_SMPS_STATIC:
  return NL80211_SMPS_STATIC;
 case IEEE80211_SMPS_DYNAMIC:
  return NL80211_SMPS_DYNAMIC;
 default:
  return NL80211_SMPS_OFF;
 }
}

int ieee80211_send_smps_action(struct ieee80211_sub_if_data *sdata,
          enum ieee80211_smps_mode smps, const u8 *da,
          const u8 *bssid, int link_id)
{
 struct ieee80211_local *local = sdata->local;
 struct sk_buff *skb;
 struct ieee80211_mgmt *action_frame;
 struct ieee80211_tx_info *info;
 u8 status_link_id = link_id < 0 ? 0 : link_id;

 /* 27 = header + category + action + smps mode */
 skb = dev_alloc_skb(27 + local->hw.extra_tx_headroom);
 if (!skb)
  return -ENOMEM;

 skb_reserve(skb, local->hw.extra_tx_headroom);
 action_frame = skb_put(skb, 27);
 memcpy(action_frame->da, da, ETH_ALEN);
 memcpy(action_frame->sa, sdata->dev->dev_addr, ETH_ALEN);
 memcpy(action_frame->bssid, bssid, ETH_ALEN);
 action_frame->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
        IEEE80211_STYPE_ACTION);
 action_frame->u.action.category = WLAN_CATEGORY_HT;
 action_frame->u.action.u.ht_smps.action = WLAN_HT_ACTION_SMPS;
 switch (smps) {
 case IEEE80211_SMPS_AUTOMATIC:
 case IEEE80211_SMPS_NUM_MODES:
  WARN_ON(1);
  smps = IEEE80211_SMPS_OFF;
  fallthrough;
 case IEEE80211_SMPS_OFF:
  action_frame->u.action.u.ht_smps.smps_control =
    WLAN_HT_SMPS_CONTROL_DISABLED;
  break;
 case IEEE80211_SMPS_STATIC:
  action_frame->u.action.u.ht_smps.smps_control =
    WLAN_HT_SMPS_CONTROL_STATIC;
  break;
 case IEEE80211_SMPS_DYNAMIC:
  action_frame->u.action.u.ht_smps.smps_control =
    WLAN_HT_SMPS_CONTROL_DYNAMIC;
  break;
 }

 /* we'll do more on status of this frame */
 info = IEEE80211_SKB_CB(skb);
 info->flags |= IEEE80211_TX_CTL_REQ_TX_STATUS;
 /* we have 13 bits, and need 6: link_id 4, smps 2 */
 info->status_data = IEEE80211_STATUS_TYPE_SMPS |
       u16_encode_bits(status_link_id << 2 | smps,
         IEEE80211_STATUS_SUBDATA_MASK);
 ieee80211_tx_skb_tid(sdata, skb, 7, link_id);

 return 0;
}

void ieee80211_request_smps(struct ieee80211_vif *vif, unsigned int link_id,
       enum ieee80211_smps_mode smps_mode)
{
 struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
 struct ieee80211_link_data *link;

 if (WARN_ON_ONCE(vif->type != NL80211_IFTYPE_STATION))
  return;

 rcu_read_lock();
 link = rcu_dereference(sdata->link[link_id]);
 if (WARN_ON(!link))
  goto out;

 trace_api_request_smps(sdata->local, sdata, link, smps_mode);

 if (link->u.mgd.driver_smps_mode == smps_mode)
  goto out;

 link->u.mgd.driver_smps_mode = smps_mode;
 wiphy_work_queue(sdata->local->hw.wiphy,
    &link->u.mgd.request_smps_work);
out:
 rcu_read_unlock();
}
/* this might change ... don't want non-open drivers using it */
EXPORT_SYMBOL_GPL(ieee80211_request_smps);

void ieee80211_ht_handle_chanwidth_notif(struct ieee80211_local *local,
      struct ieee80211_sub_if_data *sdata,
      struct sta_info *sta,
      struct link_sta_info *link_sta,
      u8 chanwidth, enum nl80211_band band)
{
 enum ieee80211_sta_rx_bandwidth max_bw, new_bw;
 struct ieee80211_supported_band *sband;
 struct sta_opmode_info sta_opmode = {};

 lockdep_assert_wiphy(local->hw.wiphy);

 if (chanwidth == IEEE80211_HT_CHANWIDTH_20MHZ)
  max_bw = IEEE80211_STA_RX_BW_20;
 else
  max_bw = ieee80211_sta_cap_rx_bw(link_sta);

 /* set cur_max_bandwidth and recalc sta bw */
 link_sta->cur_max_bandwidth = max_bw;
 new_bw = ieee80211_sta_cur_vht_bw(link_sta);

 if (link_sta->pub->bandwidth == new_bw)
  return;

 link_sta->pub->bandwidth = new_bw;
 sband = local->hw.wiphy->bands[band];
 sta_opmode.bw =
  ieee80211_sta_rx_bw_to_chan_width(link_sta);
 sta_opmode.changed = STA_OPMODE_MAX_BW_CHANGED;

 rate_control_rate_update(local, sband, link_sta,
     IEEE80211_RC_BW_CHANGED);
 cfg80211_sta_opmode_change_notify(sdata->dev,
       sta->addr,
       &sta_opmode,
       GFP_KERNEL);
}