Back to home page

OSCL-LXR

 
 

    


0001 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
0002 /*
0003  * Copyright (C) 2012-2014, 2018-2021 Intel Corporation
0004  * Copyright (C) 2013-2015 Intel Mobile Communications GmbH
0005  * Copyright (C) 2016-2017 Intel Deutschland GmbH
0006  */
0007 #include <linux/ieee80211.h>
0008 #include <linux/etherdevice.h>
0009 #include <linux/tcp.h>
0010 #include <net/ip.h>
0011 #include <net/ipv6.h>
0012 
0013 #include "iwl-trans.h"
0014 #include "iwl-eeprom-parse.h"
0015 #include "mvm.h"
0016 #include "sta.h"
0017 
0018 static void
0019 iwl_mvm_bar_check_trigger(struct iwl_mvm *mvm, const u8 *addr,
0020               u16 tid, u16 ssn)
0021 {
0022     struct iwl_fw_dbg_trigger_tlv *trig;
0023     struct iwl_fw_dbg_trigger_ba *ba_trig;
0024 
0025     trig = iwl_fw_dbg_trigger_on(&mvm->fwrt, NULL, FW_DBG_TRIGGER_BA);
0026     if (!trig)
0027         return;
0028 
0029     ba_trig = (void *)trig->data;
0030 
0031     if (!(le16_to_cpu(ba_trig->tx_bar) & BIT(tid)))
0032         return;
0033 
0034     iwl_fw_dbg_collect_trig(&mvm->fwrt, trig,
0035                 "BAR sent to %pM, tid %d, ssn %d",
0036                 addr, tid, ssn);
0037 }
0038 
0039 #define OPT_HDR(type, skb, off) \
0040     (type *)(skb_network_header(skb) + (off))
0041 
0042 static u16 iwl_mvm_tx_csum_pre_bz(struct iwl_mvm *mvm, struct sk_buff *skb,
0043                   struct ieee80211_tx_info *info, bool amsdu)
0044 {
0045     struct ieee80211_hdr *hdr = (void *)skb->data;
0046     u16 offload_assist = 0;
0047 #if IS_ENABLED(CONFIG_INET)
0048     u16 mh_len = ieee80211_hdrlen(hdr->frame_control);
0049     u8 protocol = 0;
0050 
0051     /* Do not compute checksum if already computed */
0052     if (skb->ip_summed != CHECKSUM_PARTIAL)
0053         goto out;
0054 
0055     /* We do not expect to be requested to csum stuff we do not support */
0056     if (WARN_ONCE(!(mvm->hw->netdev_features & IWL_TX_CSUM_NETIF_FLAGS) ||
0057               (skb->protocol != htons(ETH_P_IP) &&
0058                skb->protocol != htons(ETH_P_IPV6)),
0059               "No support for requested checksum\n")) {
0060         skb_checksum_help(skb);
0061         goto out;
0062     }
0063 
0064     if (skb->protocol == htons(ETH_P_IP)) {
0065         protocol = ip_hdr(skb)->protocol;
0066     } else {
0067 #if IS_ENABLED(CONFIG_IPV6)
0068         struct ipv6hdr *ipv6h =
0069             (struct ipv6hdr *)skb_network_header(skb);
0070         unsigned int off = sizeof(*ipv6h);
0071 
0072         protocol = ipv6h->nexthdr;
0073         while (protocol != NEXTHDR_NONE && ipv6_ext_hdr(protocol)) {
0074             struct ipv6_opt_hdr *hp;
0075 
0076             /* only supported extension headers */
0077             if (protocol != NEXTHDR_ROUTING &&
0078                 protocol != NEXTHDR_HOP &&
0079                 protocol != NEXTHDR_DEST) {
0080                 skb_checksum_help(skb);
0081                 goto out;
0082             }
0083 
0084             hp = OPT_HDR(struct ipv6_opt_hdr, skb, off);
0085             protocol = hp->nexthdr;
0086             off += ipv6_optlen(hp);
0087         }
0088         /* if we get here - protocol now should be TCP/UDP */
0089 #endif
0090     }
0091 
0092     if (protocol != IPPROTO_TCP && protocol != IPPROTO_UDP) {
0093         WARN_ON_ONCE(1);
0094         skb_checksum_help(skb);
0095         goto out;
0096     }
0097 
0098     /* enable L4 csum */
0099     offload_assist |= BIT(TX_CMD_OFFLD_L4_EN);
0100 
0101     /*
0102      * Set offset to IP header (snap).
0103      * We don't support tunneling so no need to take care of inner header.
0104      * Size is in words.
0105      */
0106     offload_assist |= (4 << TX_CMD_OFFLD_IP_HDR);
0107 
0108     /* Do IPv4 csum for AMSDU only (no IP csum for Ipv6) */
0109     if (skb->protocol == htons(ETH_P_IP) && amsdu) {
0110         ip_hdr(skb)->check = 0;
0111         offload_assist |= BIT(TX_CMD_OFFLD_L3_EN);
0112     }
0113 
0114     /* reset UDP/TCP header csum */
0115     if (protocol == IPPROTO_TCP)
0116         tcp_hdr(skb)->check = 0;
0117     else
0118         udp_hdr(skb)->check = 0;
0119 
0120     /*
0121      * mac header len should include IV, size is in words unless
0122      * the IV is added by the firmware like in WEP.
0123      * In new Tx API, the IV is always added by the firmware.
0124      */
0125     if (!iwl_mvm_has_new_tx_api(mvm) && info->control.hw_key &&
0126         info->control.hw_key->cipher != WLAN_CIPHER_SUITE_WEP40 &&
0127         info->control.hw_key->cipher != WLAN_CIPHER_SUITE_WEP104)
0128         mh_len += info->control.hw_key->iv_len;
0129     mh_len /= 2;
0130     offload_assist |= mh_len << TX_CMD_OFFLD_MH_SIZE;
0131 
0132 out:
0133 #endif
0134     if (amsdu)
0135         offload_assist |= BIT(TX_CMD_OFFLD_AMSDU);
0136     else if (ieee80211_hdrlen(hdr->frame_control) % 4)
0137         /* padding is inserted later in transport */
0138         offload_assist |= BIT(TX_CMD_OFFLD_PAD);
0139 
0140     return offload_assist;
0141 }
0142 
0143 u32 iwl_mvm_tx_csum_bz(struct iwl_mvm *mvm, struct sk_buff *skb, bool amsdu)
0144 {
0145     struct ieee80211_hdr *hdr = (void *)skb->data;
0146     u32 offload_assist = IWL_TX_CMD_OFFLD_BZ_PARTIAL_CSUM;
0147     unsigned int hdrlen = ieee80211_hdrlen(hdr->frame_control);
0148     unsigned int csum_start = skb_checksum_start_offset(skb);
0149 
0150     offload_assist |= u32_encode_bits(hdrlen / 2,
0151                       IWL_TX_CMD_OFFLD_BZ_MH_LEN);
0152     if (amsdu)
0153         offload_assist |= IWL_TX_CMD_OFFLD_BZ_AMSDU;
0154     else if (hdrlen % 4)
0155         /* padding is inserted later in transport */
0156         offload_assist |= IWL_TX_CMD_OFFLD_BZ_MH_PAD;
0157 
0158     if (skb->ip_summed != CHECKSUM_PARTIAL)
0159         return offload_assist;
0160 
0161     offload_assist |= IWL_TX_CMD_OFFLD_BZ_ENABLE_CSUM |
0162               IWL_TX_CMD_OFFLD_BZ_ZERO2ONES;
0163 
0164     /*
0165      * mac80211 will always calculate checksum in software for
0166      * non-fast-xmit, and so we can only do offloaded checksum
0167      * for fast-xmit frames. In this case, we always have the
0168      * RFC 1042 header present. skb_checksum_start_offset()
0169      * returns the offset from the beginning, but the hardware
0170      * needs it from after the header & SNAP header.
0171      */
0172     csum_start -= hdrlen + 8;
0173 
0174     offload_assist |= u32_encode_bits(csum_start,
0175                       IWL_TX_CMD_OFFLD_BZ_START_OFFS);
0176     offload_assist |= u32_encode_bits(csum_start + skb->csum_offset,
0177                       IWL_TX_CMD_OFFLD_BZ_RESULT_OFFS);
0178 
0179     return offload_assist;
0180 }
0181 
0182 static u32 iwl_mvm_tx_csum(struct iwl_mvm *mvm, struct sk_buff *skb,
0183                struct ieee80211_tx_info *info,
0184                bool amsdu)
0185 {
0186     if (mvm->trans->trans_cfg->device_family < IWL_DEVICE_FAMILY_BZ)
0187         return iwl_mvm_tx_csum_pre_bz(mvm, skb, info, amsdu);
0188     return iwl_mvm_tx_csum_bz(mvm, skb, amsdu);
0189 }
0190 
0191 /*
0192  * Sets most of the Tx cmd's fields
0193  */
0194 void iwl_mvm_set_tx_cmd(struct iwl_mvm *mvm, struct sk_buff *skb,
0195             struct iwl_tx_cmd *tx_cmd,
0196             struct ieee80211_tx_info *info, u8 sta_id)
0197 {
0198     struct ieee80211_hdr *hdr = (void *)skb->data;
0199     __le16 fc = hdr->frame_control;
0200     u32 tx_flags = le32_to_cpu(tx_cmd->tx_flags);
0201     u32 len = skb->len + FCS_LEN;
0202     bool amsdu = false;
0203     u8 ac;
0204 
0205     if (!(info->flags & IEEE80211_TX_CTL_NO_ACK) ||
0206         (ieee80211_is_probe_resp(fc) &&
0207          !is_multicast_ether_addr(hdr->addr1)))
0208         tx_flags |= TX_CMD_FLG_ACK;
0209     else
0210         tx_flags &= ~TX_CMD_FLG_ACK;
0211 
0212     if (ieee80211_is_probe_resp(fc))
0213         tx_flags |= TX_CMD_FLG_TSF;
0214 
0215     if (ieee80211_has_morefrags(fc))
0216         tx_flags |= TX_CMD_FLG_MORE_FRAG;
0217 
0218     if (ieee80211_is_data_qos(fc)) {
0219         u8 *qc = ieee80211_get_qos_ctl(hdr);
0220         tx_cmd->tid_tspec = qc[0] & 0xf;
0221         tx_flags &= ~TX_CMD_FLG_SEQ_CTL;
0222         amsdu = *qc & IEEE80211_QOS_CTL_A_MSDU_PRESENT;
0223     } else if (ieee80211_is_back_req(fc)) {
0224         struct ieee80211_bar *bar = (void *)skb->data;
0225         u16 control = le16_to_cpu(bar->control);
0226         u16 ssn = le16_to_cpu(bar->start_seq_num);
0227 
0228         tx_flags |= TX_CMD_FLG_ACK | TX_CMD_FLG_BAR;
0229         tx_cmd->tid_tspec = (control &
0230                      IEEE80211_BAR_CTRL_TID_INFO_MASK) >>
0231             IEEE80211_BAR_CTRL_TID_INFO_SHIFT;
0232         WARN_ON_ONCE(tx_cmd->tid_tspec >= IWL_MAX_TID_COUNT);
0233         iwl_mvm_bar_check_trigger(mvm, bar->ra, tx_cmd->tid_tspec,
0234                       ssn);
0235     } else {
0236         if (ieee80211_is_data(fc))
0237             tx_cmd->tid_tspec = IWL_TID_NON_QOS;
0238         else
0239             tx_cmd->tid_tspec = IWL_MAX_TID_COUNT;
0240 
0241         if (info->flags & IEEE80211_TX_CTL_ASSIGN_SEQ)
0242             tx_flags |= TX_CMD_FLG_SEQ_CTL;
0243         else
0244             tx_flags &= ~TX_CMD_FLG_SEQ_CTL;
0245     }
0246 
0247     /* Default to 0 (BE) when tid_spec is set to IWL_MAX_TID_COUNT */
0248     if (tx_cmd->tid_tspec < IWL_MAX_TID_COUNT)
0249         ac = tid_to_mac80211_ac[tx_cmd->tid_tspec];
0250     else
0251         ac = tid_to_mac80211_ac[0];
0252 
0253     tx_flags |= iwl_mvm_bt_coex_tx_prio(mvm, hdr, info, ac) <<
0254             TX_CMD_FLG_BT_PRIO_POS;
0255 
0256     if (ieee80211_is_mgmt(fc)) {
0257         if (ieee80211_is_assoc_req(fc) || ieee80211_is_reassoc_req(fc))
0258             tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_ASSOC);
0259         else if (ieee80211_is_action(fc))
0260             tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_NONE);
0261         else
0262             tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_MGMT);
0263 
0264         /* The spec allows Action frames in A-MPDU, we don't support
0265          * it
0266          */
0267         WARN_ON_ONCE(info->flags & IEEE80211_TX_CTL_AMPDU);
0268     } else if (info->control.flags & IEEE80211_TX_CTRL_PORT_CTRL_PROTO) {
0269         tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_MGMT);
0270     } else {
0271         tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_NONE);
0272     }
0273 
0274     if (ieee80211_is_data(fc) && len > mvm->rts_threshold &&
0275         !is_multicast_ether_addr(hdr->addr1))
0276         tx_flags |= TX_CMD_FLG_PROT_REQUIRE;
0277 
0278     if (fw_has_capa(&mvm->fw->ucode_capa,
0279             IWL_UCODE_TLV_CAPA_TXPOWER_INSERTION_SUPPORT) &&
0280         ieee80211_action_contains_tpc(skb))
0281         tx_flags |= TX_CMD_FLG_WRITE_TX_POWER;
0282 
0283     tx_cmd->tx_flags = cpu_to_le32(tx_flags);
0284     /* Total # bytes to be transmitted - PCIe code will adjust for A-MSDU */
0285     tx_cmd->len = cpu_to_le16((u16)skb->len);
0286     tx_cmd->life_time = cpu_to_le32(TX_CMD_LIFE_TIME_INFINITE);
0287     tx_cmd->sta_id = sta_id;
0288 
0289     tx_cmd->offload_assist =
0290         cpu_to_le16(iwl_mvm_tx_csum_pre_bz(mvm, skb, info, amsdu));
0291 }
0292 
0293 static u32 iwl_mvm_get_tx_ant(struct iwl_mvm *mvm,
0294                   struct ieee80211_tx_info *info,
0295                   struct ieee80211_sta *sta, __le16 fc)
0296 {
0297     if (info->band == NL80211_BAND_2GHZ &&
0298         !iwl_mvm_bt_coex_is_shared_ant_avail(mvm))
0299         return mvm->cfg->non_shared_ant << RATE_MCS_ANT_POS;
0300 
0301     if (sta && ieee80211_is_data(fc)) {
0302         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
0303 
0304         return BIT(mvmsta->tx_ant) << RATE_MCS_ANT_POS;
0305     }
0306 
0307     return BIT(mvm->mgmt_last_antenna_idx) << RATE_MCS_ANT_POS;
0308 }
0309 
0310 static u32 iwl_mvm_get_tx_rate(struct iwl_mvm *mvm,
0311                    struct ieee80211_tx_info *info,
0312                    struct ieee80211_sta *sta, __le16 fc)
0313 {
0314     int rate_idx = -1;
0315     u8 rate_plcp;
0316     u32 rate_flags = 0;
0317     bool is_cck;
0318 
0319     /* info->control is only relevant for non HW rate control */
0320     if (!ieee80211_hw_check(mvm->hw, HAS_RATE_CONTROL)) {
0321         /* HT rate doesn't make sense for a non data frame */
0322         WARN_ONCE(info->control.rates[0].flags & IEEE80211_TX_RC_MCS &&
0323               !ieee80211_is_data(fc),
0324               "Got a HT rate (flags:0x%x/mcs:%d/fc:0x%x/state:%d) for a non data frame\n",
0325               info->control.rates[0].flags,
0326               info->control.rates[0].idx,
0327               le16_to_cpu(fc),
0328               sta ? iwl_mvm_sta_from_mac80211(sta)->sta_state : -1);
0329 
0330         rate_idx = info->control.rates[0].idx;
0331     }
0332 
0333     /* if the rate isn't a well known legacy rate, take the lowest one */
0334     if (rate_idx < 0 || rate_idx >= IWL_RATE_COUNT_LEGACY)
0335         rate_idx = rate_lowest_index(
0336                 &mvm->nvm_data->bands[info->band], sta);
0337 
0338     /*
0339      * For non 2 GHZ band, remap mac80211 rate
0340      * indices into driver indices
0341      */
0342     if (info->band != NL80211_BAND_2GHZ)
0343         rate_idx += IWL_FIRST_OFDM_RATE;
0344 
0345     /* For 2.4 GHZ band, check that there is no need to remap */
0346     BUILD_BUG_ON(IWL_FIRST_CCK_RATE != 0);
0347 
0348     /* Get PLCP rate for tx_cmd->rate_n_flags */
0349     rate_plcp = iwl_mvm_mac80211_idx_to_hwrate(mvm->fw, rate_idx);
0350     is_cck = (rate_idx >= IWL_FIRST_CCK_RATE) && (rate_idx <= IWL_LAST_CCK_RATE);
0351 
0352     /* Set CCK or OFDM flag */
0353     if (iwl_fw_lookup_cmd_ver(mvm->fw, TX_CMD, 0) > 8) {
0354         if (!is_cck)
0355             rate_flags |= RATE_MCS_LEGACY_OFDM_MSK;
0356         else
0357             rate_flags |= RATE_MCS_CCK_MSK;
0358     } else if (is_cck) {
0359         rate_flags |= RATE_MCS_CCK_MSK_V1;
0360     }
0361 
0362     return (u32)rate_plcp | rate_flags;
0363 }
0364 
0365 static u32 iwl_mvm_get_tx_rate_n_flags(struct iwl_mvm *mvm,
0366                        struct ieee80211_tx_info *info,
0367                        struct ieee80211_sta *sta, __le16 fc)
0368 {
0369     return iwl_mvm_get_tx_rate(mvm, info, sta, fc) |
0370         iwl_mvm_get_tx_ant(mvm, info, sta, fc);
0371 }
0372 
0373 /*
0374  * Sets the fields in the Tx cmd that are rate related
0375  */
0376 void iwl_mvm_set_tx_cmd_rate(struct iwl_mvm *mvm, struct iwl_tx_cmd *tx_cmd,
0377                 struct ieee80211_tx_info *info,
0378                 struct ieee80211_sta *sta, __le16 fc)
0379 {
0380     /* Set retry limit on RTS packets */
0381     tx_cmd->rts_retry_limit = IWL_RTS_DFAULT_RETRY_LIMIT;
0382 
0383     /* Set retry limit on DATA packets and Probe Responses*/
0384     if (ieee80211_is_probe_resp(fc)) {
0385         tx_cmd->data_retry_limit = IWL_MGMT_DFAULT_RETRY_LIMIT;
0386         tx_cmd->rts_retry_limit =
0387             min(tx_cmd->data_retry_limit, tx_cmd->rts_retry_limit);
0388     } else if (ieee80211_is_back_req(fc)) {
0389         tx_cmd->data_retry_limit = IWL_BAR_DFAULT_RETRY_LIMIT;
0390     } else {
0391         tx_cmd->data_retry_limit = IWL_DEFAULT_TX_RETRY;
0392     }
0393 
0394     /*
0395      * for data packets, rate info comes from the table inside the fw. This
0396      * table is controlled by LINK_QUALITY commands
0397      */
0398 
0399     if (ieee80211_is_data(fc) && sta) {
0400         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
0401 
0402         if (mvmsta->sta_state >= IEEE80211_STA_AUTHORIZED) {
0403             tx_cmd->initial_rate_index = 0;
0404             tx_cmd->tx_flags |= cpu_to_le32(TX_CMD_FLG_STA_RATE);
0405             return;
0406         }
0407     } else if (ieee80211_is_back_req(fc)) {
0408         tx_cmd->tx_flags |=
0409             cpu_to_le32(TX_CMD_FLG_ACK | TX_CMD_FLG_BAR);
0410     }
0411 
0412     /* Set the rate in the TX cmd */
0413     tx_cmd->rate_n_flags =
0414         cpu_to_le32(iwl_mvm_get_tx_rate_n_flags(mvm, info, sta, fc));
0415 }
0416 
0417 static inline void iwl_mvm_set_tx_cmd_pn(struct ieee80211_tx_info *info,
0418                      u8 *crypto_hdr)
0419 {
0420     struct ieee80211_key_conf *keyconf = info->control.hw_key;
0421     u64 pn;
0422 
0423     pn = atomic64_inc_return(&keyconf->tx_pn);
0424     crypto_hdr[0] = pn;
0425     crypto_hdr[2] = 0;
0426     crypto_hdr[3] = 0x20 | (keyconf->keyidx << 6);
0427     crypto_hdr[1] = pn >> 8;
0428     crypto_hdr[4] = pn >> 16;
0429     crypto_hdr[5] = pn >> 24;
0430     crypto_hdr[6] = pn >> 32;
0431     crypto_hdr[7] = pn >> 40;
0432 }
0433 
0434 /*
0435  * Sets the fields in the Tx cmd that are crypto related
0436  */
0437 static void iwl_mvm_set_tx_cmd_crypto(struct iwl_mvm *mvm,
0438                       struct ieee80211_tx_info *info,
0439                       struct iwl_tx_cmd *tx_cmd,
0440                       struct sk_buff *skb_frag,
0441                       int hdrlen)
0442 {
0443     struct ieee80211_key_conf *keyconf = info->control.hw_key;
0444     u8 *crypto_hdr = skb_frag->data + hdrlen;
0445     enum iwl_tx_cmd_sec_ctrl type = TX_CMD_SEC_CCM;
0446     u64 pn;
0447 
0448     switch (keyconf->cipher) {
0449     case WLAN_CIPHER_SUITE_CCMP:
0450         iwl_mvm_set_tx_cmd_ccmp(info, tx_cmd);
0451         iwl_mvm_set_tx_cmd_pn(info, crypto_hdr);
0452         break;
0453 
0454     case WLAN_CIPHER_SUITE_TKIP:
0455         tx_cmd->sec_ctl = TX_CMD_SEC_TKIP;
0456         pn = atomic64_inc_return(&keyconf->tx_pn);
0457         ieee80211_tkip_add_iv(crypto_hdr, keyconf, pn);
0458         ieee80211_get_tkip_p2k(keyconf, skb_frag, tx_cmd->key);
0459         break;
0460 
0461     case WLAN_CIPHER_SUITE_WEP104:
0462         tx_cmd->sec_ctl |= TX_CMD_SEC_KEY128;
0463         fallthrough;
0464     case WLAN_CIPHER_SUITE_WEP40:
0465         tx_cmd->sec_ctl |= TX_CMD_SEC_WEP |
0466             ((keyconf->keyidx << TX_CMD_SEC_WEP_KEY_IDX_POS) &
0467               TX_CMD_SEC_WEP_KEY_IDX_MSK);
0468 
0469         memcpy(&tx_cmd->key[3], keyconf->key, keyconf->keylen);
0470         break;
0471     case WLAN_CIPHER_SUITE_GCMP:
0472     case WLAN_CIPHER_SUITE_GCMP_256:
0473         type = TX_CMD_SEC_GCMP;
0474         fallthrough;
0475     case WLAN_CIPHER_SUITE_CCMP_256:
0476         /* TODO: Taking the key from the table might introduce a race
0477          * when PTK rekeying is done, having an old packets with a PN
0478          * based on the old key but the message encrypted with a new
0479          * one.
0480          * Need to handle this.
0481          */
0482         tx_cmd->sec_ctl |= type | TX_CMD_SEC_KEY_FROM_TABLE;
0483         tx_cmd->key[0] = keyconf->hw_key_idx;
0484         iwl_mvm_set_tx_cmd_pn(info, crypto_hdr);
0485         break;
0486     default:
0487         tx_cmd->sec_ctl |= TX_CMD_SEC_EXT;
0488     }
0489 }
0490 
0491 /*
0492  * Allocates and sets the Tx cmd the driver data pointers in the skb
0493  */
0494 static struct iwl_device_tx_cmd *
0495 iwl_mvm_set_tx_params(struct iwl_mvm *mvm, struct sk_buff *skb,
0496               struct ieee80211_tx_info *info, int hdrlen,
0497               struct ieee80211_sta *sta, u8 sta_id)
0498 {
0499     struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
0500     struct iwl_device_tx_cmd *dev_cmd;
0501     struct iwl_tx_cmd *tx_cmd;
0502 
0503     dev_cmd = iwl_trans_alloc_tx_cmd(mvm->trans);
0504 
0505     if (unlikely(!dev_cmd))
0506         return NULL;
0507 
0508     dev_cmd->hdr.cmd = TX_CMD;
0509 
0510     if (iwl_mvm_has_new_tx_api(mvm)) {
0511         u32 rate_n_flags = 0;
0512         u16 flags = 0;
0513         struct iwl_mvm_sta *mvmsta = sta ?
0514             iwl_mvm_sta_from_mac80211(sta) : NULL;
0515         bool amsdu = false;
0516 
0517         if (ieee80211_is_data_qos(hdr->frame_control)) {
0518             u8 *qc = ieee80211_get_qos_ctl(hdr);
0519 
0520             amsdu = *qc & IEEE80211_QOS_CTL_A_MSDU_PRESENT;
0521         }
0522 
0523         if (!info->control.hw_key)
0524             flags |= IWL_TX_FLAGS_ENCRYPT_DIS;
0525 
0526         /*
0527          * For data packets rate info comes from the fw. Only
0528          * set rate/antenna during connection establishment or in case
0529          * no station is given.
0530          */
0531         if (!sta || !ieee80211_is_data(hdr->frame_control) ||
0532             mvmsta->sta_state < IEEE80211_STA_AUTHORIZED) {
0533             flags |= IWL_TX_FLAGS_CMD_RATE;
0534             rate_n_flags =
0535                 iwl_mvm_get_tx_rate_n_flags(mvm, info, sta,
0536                                 hdr->frame_control);
0537         }
0538 
0539         if (mvm->trans->trans_cfg->device_family >=
0540             IWL_DEVICE_FAMILY_AX210) {
0541             struct iwl_tx_cmd_gen3 *cmd = (void *)dev_cmd->payload;
0542             u32 offload_assist = iwl_mvm_tx_csum(mvm, skb,
0543                                  info, amsdu);
0544 
0545             cmd->offload_assist = cpu_to_le32(offload_assist);
0546 
0547             /* Total # bytes to be transmitted */
0548             cmd->len = cpu_to_le16((u16)skb->len);
0549 
0550             /* Copy MAC header from skb into command buffer */
0551             memcpy(cmd->hdr, hdr, hdrlen);
0552 
0553             cmd->flags = cpu_to_le16(flags);
0554             cmd->rate_n_flags = cpu_to_le32(rate_n_flags);
0555         } else {
0556             struct iwl_tx_cmd_gen2 *cmd = (void *)dev_cmd->payload;
0557             u16 offload_assist = iwl_mvm_tx_csum_pre_bz(mvm, skb,
0558                                     info,
0559                                     amsdu);
0560 
0561             cmd->offload_assist = cpu_to_le16(offload_assist);
0562 
0563             /* Total # bytes to be transmitted */
0564             cmd->len = cpu_to_le16((u16)skb->len);
0565 
0566             /* Copy MAC header from skb into command buffer */
0567             memcpy(cmd->hdr, hdr, hdrlen);
0568 
0569             cmd->flags = cpu_to_le32(flags);
0570             cmd->rate_n_flags = cpu_to_le32(rate_n_flags);
0571         }
0572         goto out;
0573     }
0574 
0575     tx_cmd = (struct iwl_tx_cmd *)dev_cmd->payload;
0576 
0577     if (info->control.hw_key)
0578         iwl_mvm_set_tx_cmd_crypto(mvm, info, tx_cmd, skb, hdrlen);
0579 
0580     iwl_mvm_set_tx_cmd(mvm, skb, tx_cmd, info, sta_id);
0581 
0582     iwl_mvm_set_tx_cmd_rate(mvm, tx_cmd, info, sta, hdr->frame_control);
0583 
0584     /* Copy MAC header from skb into command buffer */
0585     memcpy(tx_cmd->hdr, hdr, hdrlen);
0586 
0587 out:
0588     return dev_cmd;
0589 }
0590 
0591 static void iwl_mvm_skb_prepare_status(struct sk_buff *skb,
0592                        struct iwl_device_tx_cmd *cmd)
0593 {
0594     struct ieee80211_tx_info *skb_info = IEEE80211_SKB_CB(skb);
0595 
0596     memset(&skb_info->status, 0, sizeof(skb_info->status));
0597     memset(skb_info->driver_data, 0, sizeof(skb_info->driver_data));
0598 
0599     skb_info->driver_data[1] = cmd;
0600 }
0601 
0602 static int iwl_mvm_get_ctrl_vif_queue(struct iwl_mvm *mvm,
0603                       struct ieee80211_tx_info *info,
0604                       struct ieee80211_hdr *hdr)
0605 {
0606     struct iwl_mvm_vif *mvmvif =
0607         iwl_mvm_vif_from_mac80211(info->control.vif);
0608     __le16 fc = hdr->frame_control;
0609 
0610     switch (info->control.vif->type) {
0611     case NL80211_IFTYPE_AP:
0612     case NL80211_IFTYPE_ADHOC:
0613         /*
0614          * Non-bufferable frames use the broadcast station, thus they
0615          * use the probe queue.
0616          * Also take care of the case where we send a deauth to a
0617          * station that we don't have, or similarly an association
0618          * response (with non-success status) for a station we can't
0619          * accept.
0620          * Also, disassociate frames might happen, particular with
0621          * reason 7 ("Class 3 frame received from nonassociated STA").
0622          */
0623         if (ieee80211_is_mgmt(fc) &&
0624             (!ieee80211_is_bufferable_mmpdu(fc) ||
0625              ieee80211_is_deauth(fc) || ieee80211_is_disassoc(fc)))
0626             return mvm->probe_queue;
0627 
0628         if (!ieee80211_has_order(fc) && !ieee80211_is_probe_req(fc) &&
0629             is_multicast_ether_addr(hdr->addr1))
0630             return mvmvif->cab_queue;
0631 
0632         WARN_ONCE(info->control.vif->type != NL80211_IFTYPE_ADHOC,
0633               "fc=0x%02x", le16_to_cpu(fc));
0634         return mvm->probe_queue;
0635     case NL80211_IFTYPE_P2P_DEVICE:
0636         if (ieee80211_is_mgmt(fc))
0637             return mvm->p2p_dev_queue;
0638 
0639         WARN_ON_ONCE(1);
0640         return mvm->p2p_dev_queue;
0641     default:
0642         WARN_ONCE(1, "Not a ctrl vif, no available queue\n");
0643         return -1;
0644     }
0645 }
0646 
0647 static void iwl_mvm_probe_resp_set_noa(struct iwl_mvm *mvm,
0648                        struct sk_buff *skb)
0649 {
0650     struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
0651     struct iwl_mvm_vif *mvmvif =
0652         iwl_mvm_vif_from_mac80211(info->control.vif);
0653     struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)skb->data;
0654     int base_len = (u8 *)mgmt->u.probe_resp.variable - (u8 *)mgmt;
0655     struct iwl_probe_resp_data *resp_data;
0656     const u8 *ie;
0657     u8 *pos;
0658     u8 match[] = {
0659         (WLAN_OUI_WFA >> 16) & 0xff,
0660         (WLAN_OUI_WFA >> 8) & 0xff,
0661         WLAN_OUI_WFA & 0xff,
0662         WLAN_OUI_TYPE_WFA_P2P,
0663     };
0664 
0665     rcu_read_lock();
0666 
0667     resp_data = rcu_dereference(mvmvif->probe_resp_data);
0668     if (!resp_data)
0669         goto out;
0670 
0671     if (!resp_data->notif.noa_active)
0672         goto out;
0673 
0674     ie = cfg80211_find_ie_match(WLAN_EID_VENDOR_SPECIFIC,
0675                     mgmt->u.probe_resp.variable,
0676                     skb->len - base_len,
0677                     match, 4, 2);
0678     if (!ie) {
0679         IWL_DEBUG_TX(mvm, "probe resp doesn't have P2P IE\n");
0680         goto out;
0681     }
0682 
0683     if (skb_tailroom(skb) < resp_data->noa_len) {
0684         if (pskb_expand_head(skb, 0, resp_data->noa_len, GFP_ATOMIC)) {
0685             IWL_ERR(mvm,
0686                 "Failed to reallocate probe resp\n");
0687             goto out;
0688         }
0689     }
0690 
0691     pos = skb_put(skb, resp_data->noa_len);
0692 
0693     *pos++ = WLAN_EID_VENDOR_SPECIFIC;
0694     /* Set length of IE body (not including ID and length itself) */
0695     *pos++ = resp_data->noa_len - 2;
0696     *pos++ = (WLAN_OUI_WFA >> 16) & 0xff;
0697     *pos++ = (WLAN_OUI_WFA >> 8) & 0xff;
0698     *pos++ = WLAN_OUI_WFA & 0xff;
0699     *pos++ = WLAN_OUI_TYPE_WFA_P2P;
0700 
0701     memcpy(pos, &resp_data->notif.noa_attr,
0702            resp_data->noa_len - sizeof(struct ieee80211_vendor_ie));
0703 
0704 out:
0705     rcu_read_unlock();
0706 }
0707 
0708 int iwl_mvm_tx_skb_non_sta(struct iwl_mvm *mvm, struct sk_buff *skb)
0709 {
0710     struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
0711     struct ieee80211_tx_info info;
0712     struct iwl_device_tx_cmd *dev_cmd;
0713     u8 sta_id;
0714     int hdrlen = ieee80211_hdrlen(hdr->frame_control);
0715     __le16 fc = hdr->frame_control;
0716     bool offchannel = IEEE80211_SKB_CB(skb)->flags &
0717         IEEE80211_TX_CTL_TX_OFFCHAN;
0718     int queue = -1;
0719 
0720     if (IWL_MVM_NON_TRANSMITTING_AP && ieee80211_is_probe_resp(fc))
0721         return -1;
0722 
0723     memcpy(&info, skb->cb, sizeof(info));
0724 
0725     if (WARN_ON_ONCE(skb->len > IEEE80211_MAX_DATA_LEN + hdrlen))
0726         return -1;
0727 
0728     if (WARN_ON_ONCE(info.flags & IEEE80211_TX_CTL_AMPDU))
0729         return -1;
0730 
0731     if (info.control.vif) {
0732         struct iwl_mvm_vif *mvmvif =
0733             iwl_mvm_vif_from_mac80211(info.control.vif);
0734 
0735         if (info.control.vif->type == NL80211_IFTYPE_P2P_DEVICE ||
0736             info.control.vif->type == NL80211_IFTYPE_AP ||
0737             info.control.vif->type == NL80211_IFTYPE_ADHOC) {
0738             if (!ieee80211_is_data(hdr->frame_control))
0739                 sta_id = mvmvif->bcast_sta.sta_id;
0740             else
0741                 sta_id = mvmvif->mcast_sta.sta_id;
0742 
0743             queue = iwl_mvm_get_ctrl_vif_queue(mvm, &info, hdr);
0744         } else if (info.control.vif->type == NL80211_IFTYPE_MONITOR) {
0745             queue = mvm->snif_queue;
0746             sta_id = mvm->snif_sta.sta_id;
0747         } else if (info.control.vif->type == NL80211_IFTYPE_STATION &&
0748                offchannel) {
0749             /*
0750              * IWL_MVM_OFFCHANNEL_QUEUE is used for ROC packets
0751              * that can be used in 2 different types of vifs, P2P &
0752              * STATION.
0753              * P2P uses the offchannel queue.
0754              * STATION (HS2.0) uses the auxiliary context of the FW,
0755              * and hence needs to be sent on the aux queue.
0756              */
0757             sta_id = mvm->aux_sta.sta_id;
0758             queue = mvm->aux_queue;
0759         }
0760     }
0761 
0762     if (queue < 0) {
0763         IWL_ERR(mvm, "No queue was found. Dropping TX\n");
0764         return -1;
0765     }
0766 
0767     if (unlikely(ieee80211_is_probe_resp(fc)))
0768         iwl_mvm_probe_resp_set_noa(mvm, skb);
0769 
0770     IWL_DEBUG_TX(mvm, "station Id %d, queue=%d\n", sta_id, queue);
0771 
0772     dev_cmd = iwl_mvm_set_tx_params(mvm, skb, &info, hdrlen, NULL, sta_id);
0773     if (!dev_cmd)
0774         return -1;
0775 
0776     /* From now on, we cannot access info->control */
0777     iwl_mvm_skb_prepare_status(skb, dev_cmd);
0778 
0779     if (iwl_trans_tx(mvm->trans, skb, dev_cmd, queue)) {
0780         iwl_trans_free_tx_cmd(mvm->trans, dev_cmd);
0781         return -1;
0782     }
0783 
0784     return 0;
0785 }
0786 
0787 unsigned int iwl_mvm_max_amsdu_size(struct iwl_mvm *mvm,
0788                     struct ieee80211_sta *sta, unsigned int tid)
0789 {
0790     struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
0791     enum nl80211_band band = mvmsta->vif->bss_conf.chandef.chan->band;
0792     u8 ac = tid_to_mac80211_ac[tid];
0793     unsigned int txf;
0794     int lmac = iwl_mvm_get_lmac_id(mvm->fw, band);
0795 
0796     /* For HE redirect to trigger based fifos */
0797     if (sta->deflink.he_cap.has_he && !WARN_ON(!iwl_mvm_has_new_tx_api(mvm)))
0798         ac += 4;
0799 
0800     txf = iwl_mvm_mac_ac_to_tx_fifo(mvm, ac);
0801 
0802     /*
0803      * Don't send an AMSDU that will be longer than the TXF.
0804      * Add a security margin of 256 for the TX command + headers.
0805      * We also want to have the start of the next packet inside the
0806      * fifo to be able to send bursts.
0807      */
0808     return min_t(unsigned int, mvmsta->max_amsdu_len,
0809              mvm->fwrt.smem_cfg.lmac[lmac].txfifo_size[txf] - 256);
0810 }
0811 
0812 #ifdef CONFIG_INET
0813 
0814 static int
0815 iwl_mvm_tx_tso_segment(struct sk_buff *skb, unsigned int num_subframes,
0816                netdev_features_t netdev_flags,
0817                struct sk_buff_head *mpdus_skb)
0818 {
0819     struct sk_buff *tmp, *next;
0820     struct ieee80211_hdr *hdr = (void *)skb->data;
0821     char cb[sizeof(skb->cb)];
0822     u16 i = 0;
0823     unsigned int tcp_payload_len;
0824     unsigned int mss = skb_shinfo(skb)->gso_size;
0825     bool ipv4 = (skb->protocol == htons(ETH_P_IP));
0826     bool qos = ieee80211_is_data_qos(hdr->frame_control);
0827     u16 ip_base_id = ipv4 ? ntohs(ip_hdr(skb)->id) : 0;
0828 
0829     skb_shinfo(skb)->gso_size = num_subframes * mss;
0830     memcpy(cb, skb->cb, sizeof(cb));
0831 
0832     next = skb_gso_segment(skb, netdev_flags);
0833     skb_shinfo(skb)->gso_size = mss;
0834     skb_shinfo(skb)->gso_type = ipv4 ? SKB_GSO_TCPV4 : SKB_GSO_TCPV6;
0835     if (WARN_ON_ONCE(IS_ERR(next)))
0836         return -EINVAL;
0837     else if (next)
0838         consume_skb(skb);
0839 
0840     skb_list_walk_safe(next, tmp, next) {
0841         memcpy(tmp->cb, cb, sizeof(tmp->cb));
0842         /*
0843          * Compute the length of all the data added for the A-MSDU.
0844          * This will be used to compute the length to write in the TX
0845          * command. We have: SNAP + IP + TCP for n -1 subframes and
0846          * ETH header for n subframes.
0847          */
0848         tcp_payload_len = skb_tail_pointer(tmp) -
0849             skb_transport_header(tmp) -
0850             tcp_hdrlen(tmp) + tmp->data_len;
0851 
0852         if (ipv4)
0853             ip_hdr(tmp)->id = htons(ip_base_id + i * num_subframes);
0854 
0855         if (tcp_payload_len > mss) {
0856             skb_shinfo(tmp)->gso_size = mss;
0857             skb_shinfo(tmp)->gso_type = ipv4 ? SKB_GSO_TCPV4 :
0858                                SKB_GSO_TCPV6;
0859         } else {
0860             if (qos) {
0861                 u8 *qc;
0862 
0863                 if (ipv4)
0864                     ip_send_check(ip_hdr(tmp));
0865 
0866                 qc = ieee80211_get_qos_ctl((void *)tmp->data);
0867                 *qc &= ~IEEE80211_QOS_CTL_A_MSDU_PRESENT;
0868             }
0869             skb_shinfo(tmp)->gso_size = 0;
0870         }
0871 
0872         skb_mark_not_on_list(tmp);
0873         __skb_queue_tail(mpdus_skb, tmp);
0874         i++;
0875     }
0876 
0877     return 0;
0878 }
0879 
0880 static int iwl_mvm_tx_tso(struct iwl_mvm *mvm, struct sk_buff *skb,
0881               struct ieee80211_tx_info *info,
0882               struct ieee80211_sta *sta,
0883               struct sk_buff_head *mpdus_skb)
0884 {
0885     struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
0886     struct ieee80211_hdr *hdr = (void *)skb->data;
0887     unsigned int mss = skb_shinfo(skb)->gso_size;
0888     unsigned int num_subframes, tcp_payload_len, subf_len, max_amsdu_len;
0889     u16 snap_ip_tcp, pad;
0890     netdev_features_t netdev_flags = NETIF_F_CSUM_MASK | NETIF_F_SG;
0891     u8 tid;
0892 
0893     snap_ip_tcp = 8 + skb_transport_header(skb) - skb_network_header(skb) +
0894         tcp_hdrlen(skb);
0895 
0896     if (!mvmsta->max_amsdu_len ||
0897         !ieee80211_is_data_qos(hdr->frame_control) ||
0898         !mvmsta->amsdu_enabled)
0899         return iwl_mvm_tx_tso_segment(skb, 1, netdev_flags, mpdus_skb);
0900 
0901     /*
0902      * Do not build AMSDU for IPv6 with extension headers.
0903      * ask stack to segment and checkum the generated MPDUs for us.
0904      */
0905     if (skb->protocol == htons(ETH_P_IPV6) &&
0906         ((struct ipv6hdr *)skb_network_header(skb))->nexthdr !=
0907         IPPROTO_TCP) {
0908         netdev_flags &= ~NETIF_F_CSUM_MASK;
0909         return iwl_mvm_tx_tso_segment(skb, 1, netdev_flags, mpdus_skb);
0910     }
0911 
0912     tid = ieee80211_get_tid(hdr);
0913     if (WARN_ON_ONCE(tid >= IWL_MAX_TID_COUNT))
0914         return -EINVAL;
0915 
0916     /*
0917      * No need to lock amsdu_in_ampdu_allowed since it can't be modified
0918      * during an BA session.
0919      */
0920     if ((info->flags & IEEE80211_TX_CTL_AMPDU &&
0921          !mvmsta->tid_data[tid].amsdu_in_ampdu_allowed) ||
0922         !(mvmsta->amsdu_enabled & BIT(tid)))
0923         return iwl_mvm_tx_tso_segment(skb, 1, netdev_flags, mpdus_skb);
0924 
0925     /*
0926      * Take the min of ieee80211 station and mvm station
0927      */
0928     max_amsdu_len =
0929         min_t(unsigned int, sta->max_amsdu_len,
0930               iwl_mvm_max_amsdu_size(mvm, sta, tid));
0931 
0932     /*
0933      * Limit A-MSDU in A-MPDU to 4095 bytes when VHT is not
0934      * supported. This is a spec requirement (IEEE 802.11-2015
0935      * section 8.7.3 NOTE 3).
0936      */
0937     if (info->flags & IEEE80211_TX_CTL_AMPDU &&
0938         !sta->deflink.vht_cap.vht_supported)
0939         max_amsdu_len = min_t(unsigned int, max_amsdu_len, 4095);
0940 
0941     /* Sub frame header + SNAP + IP header + TCP header + MSS */
0942     subf_len = sizeof(struct ethhdr) + snap_ip_tcp + mss;
0943     pad = (4 - subf_len) & 0x3;
0944 
0945     /*
0946      * If we have N subframes in the A-MSDU, then the A-MSDU's size is
0947      * N * subf_len + (N - 1) * pad.
0948      */
0949     num_subframes = (max_amsdu_len + pad) / (subf_len + pad);
0950 
0951     if (sta->max_amsdu_subframes &&
0952         num_subframes > sta->max_amsdu_subframes)
0953         num_subframes = sta->max_amsdu_subframes;
0954 
0955     tcp_payload_len = skb_tail_pointer(skb) - skb_transport_header(skb) -
0956         tcp_hdrlen(skb) + skb->data_len;
0957 
0958     /*
0959      * Make sure we have enough TBs for the A-MSDU:
0960      *  2 for each subframe
0961      *  1 more for each fragment
0962      *  1 more for the potential data in the header
0963      */
0964     if ((num_subframes * 2 + skb_shinfo(skb)->nr_frags + 1) >
0965         mvm->trans->max_skb_frags)
0966         num_subframes = 1;
0967 
0968     if (num_subframes > 1)
0969         *ieee80211_get_qos_ctl(hdr) |= IEEE80211_QOS_CTL_A_MSDU_PRESENT;
0970 
0971     /* This skb fits in one single A-MSDU */
0972     if (num_subframes * mss >= tcp_payload_len) {
0973         __skb_queue_tail(mpdus_skb, skb);
0974         return 0;
0975     }
0976 
0977     /*
0978      * Trick the segmentation function to make it
0979      * create SKBs that can fit into one A-MSDU.
0980      */
0981     return iwl_mvm_tx_tso_segment(skb, num_subframes, netdev_flags,
0982                       mpdus_skb);
0983 }
0984 #else /* CONFIG_INET */
0985 static int iwl_mvm_tx_tso(struct iwl_mvm *mvm, struct sk_buff *skb,
0986               struct ieee80211_tx_info *info,
0987               struct ieee80211_sta *sta,
0988               struct sk_buff_head *mpdus_skb)
0989 {
0990     /* Impossible to get TSO with CONFIG_INET */
0991     WARN_ON(1);
0992 
0993     return -1;
0994 }
0995 #endif
0996 
0997 /* Check if there are any timed-out TIDs on a given shared TXQ */
0998 static bool iwl_mvm_txq_should_update(struct iwl_mvm *mvm, int txq_id)
0999 {
1000     unsigned long queue_tid_bitmap = mvm->queue_info[txq_id].tid_bitmap;
1001     unsigned long now = jiffies;
1002     int tid;
1003 
1004     if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
1005         return false;
1006 
1007     for_each_set_bit(tid, &queue_tid_bitmap, IWL_MAX_TID_COUNT + 1) {
1008         if (time_before(mvm->queue_info[txq_id].last_frame_time[tid] +
1009                 IWL_MVM_DQA_QUEUE_TIMEOUT, now))
1010             return true;
1011     }
1012 
1013     return false;
1014 }
1015 
1016 static void iwl_mvm_tx_airtime(struct iwl_mvm *mvm,
1017                    struct iwl_mvm_sta *mvmsta,
1018                    int airtime)
1019 {
1020     int mac = mvmsta->mac_id_n_color & FW_CTXT_ID_MSK;
1021     struct iwl_mvm_tcm_mac *mdata;
1022 
1023     if (mac >= NUM_MAC_INDEX_DRIVER)
1024         return;
1025 
1026     mdata = &mvm->tcm.data[mac];
1027 
1028     if (mvm->tcm.paused)
1029         return;
1030 
1031     if (time_after(jiffies, mvm->tcm.ts + MVM_TCM_PERIOD))
1032         schedule_delayed_work(&mvm->tcm.work, 0);
1033 
1034     mdata->tx.airtime += airtime;
1035 }
1036 
1037 static int iwl_mvm_tx_pkt_queued(struct iwl_mvm *mvm,
1038                  struct iwl_mvm_sta *mvmsta, int tid)
1039 {
1040     u32 ac = tid_to_mac80211_ac[tid];
1041     int mac = mvmsta->mac_id_n_color & FW_CTXT_ID_MSK;
1042     struct iwl_mvm_tcm_mac *mdata;
1043 
1044     if (mac >= NUM_MAC_INDEX_DRIVER)
1045         return -EINVAL;
1046 
1047     mdata = &mvm->tcm.data[mac];
1048 
1049     mdata->tx.pkts[ac]++;
1050 
1051     return 0;
1052 }
1053 
1054 /*
1055  * Sets the fields in the Tx cmd that are crypto related.
1056  *
1057  * This function must be called with BHs disabled.
1058  */
1059 static int iwl_mvm_tx_mpdu(struct iwl_mvm *mvm, struct sk_buff *skb,
1060                struct ieee80211_tx_info *info,
1061                struct ieee80211_sta *sta)
1062 {
1063     struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
1064     struct iwl_mvm_sta *mvmsta;
1065     struct iwl_device_tx_cmd *dev_cmd;
1066     __le16 fc;
1067     u16 seq_number = 0;
1068     u8 tid = IWL_MAX_TID_COUNT;
1069     u16 txq_id;
1070     bool is_ampdu = false;
1071     int hdrlen;
1072 
1073     mvmsta = iwl_mvm_sta_from_mac80211(sta);
1074     fc = hdr->frame_control;
1075     hdrlen = ieee80211_hdrlen(fc);
1076 
1077     if (IWL_MVM_NON_TRANSMITTING_AP && ieee80211_is_probe_resp(fc))
1078         return -1;
1079 
1080     if (WARN_ON_ONCE(!mvmsta))
1081         return -1;
1082 
1083     if (WARN_ON_ONCE(mvmsta->sta_id == IWL_MVM_INVALID_STA))
1084         return -1;
1085 
1086     if (unlikely(ieee80211_is_any_nullfunc(fc)) && sta->deflink.he_cap.has_he)
1087         return -1;
1088 
1089     if (unlikely(ieee80211_is_probe_resp(fc)))
1090         iwl_mvm_probe_resp_set_noa(mvm, skb);
1091 
1092     dev_cmd = iwl_mvm_set_tx_params(mvm, skb, info, hdrlen,
1093                     sta, mvmsta->sta_id);
1094     if (!dev_cmd)
1095         goto drop;
1096 
1097     /*
1098      * we handle that entirely ourselves -- for uAPSD the firmware
1099      * will always send a notification, and for PS-Poll responses
1100      * we'll notify mac80211 when getting frame status
1101      */
1102     info->flags &= ~IEEE80211_TX_STATUS_EOSP;
1103 
1104     spin_lock(&mvmsta->lock);
1105 
1106     /* nullfunc frames should go to the MGMT queue regardless of QOS,
1107      * the condition of !ieee80211_is_qos_nullfunc(fc) keeps the default
1108      * assignment of MGMT TID
1109      */
1110     if (ieee80211_is_data_qos(fc) && !ieee80211_is_qos_nullfunc(fc)) {
1111         tid = ieee80211_get_tid(hdr);
1112         if (WARN_ONCE(tid >= IWL_MAX_TID_COUNT, "Invalid TID %d", tid))
1113             goto drop_unlock_sta;
1114 
1115         is_ampdu = info->flags & IEEE80211_TX_CTL_AMPDU;
1116         if (WARN_ONCE(is_ampdu &&
1117                   mvmsta->tid_data[tid].state != IWL_AGG_ON,
1118                   "Invalid internal agg state %d for TID %d",
1119                    mvmsta->tid_data[tid].state, tid))
1120             goto drop_unlock_sta;
1121 
1122         seq_number = mvmsta->tid_data[tid].seq_number;
1123         seq_number &= IEEE80211_SCTL_SEQ;
1124 
1125         if (!iwl_mvm_has_new_tx_api(mvm)) {
1126             struct iwl_tx_cmd *tx_cmd = (void *)dev_cmd->payload;
1127 
1128             hdr->seq_ctrl &= cpu_to_le16(IEEE80211_SCTL_FRAG);
1129             hdr->seq_ctrl |= cpu_to_le16(seq_number);
1130             /* update the tx_cmd hdr as it was already copied */
1131             tx_cmd->hdr->seq_ctrl = hdr->seq_ctrl;
1132         }
1133     } else if (ieee80211_is_data(fc) && !ieee80211_is_data_qos(fc)) {
1134         tid = IWL_TID_NON_QOS;
1135     }
1136 
1137     txq_id = mvmsta->tid_data[tid].txq_id;
1138 
1139     WARN_ON_ONCE(info->flags & IEEE80211_TX_CTL_SEND_AFTER_DTIM);
1140 
1141     if (WARN_ONCE(txq_id == IWL_MVM_INVALID_QUEUE, "Invalid TXQ id")) {
1142         iwl_trans_free_tx_cmd(mvm->trans, dev_cmd);
1143         spin_unlock(&mvmsta->lock);
1144         return -1;
1145     }
1146 
1147     if (!iwl_mvm_has_new_tx_api(mvm)) {
1148         /* Keep track of the time of the last frame for this RA/TID */
1149         mvm->queue_info[txq_id].last_frame_time[tid] = jiffies;
1150 
1151         /*
1152          * If we have timed-out TIDs - schedule the worker that will
1153          * reconfig the queues and update them
1154          *
1155          * Note that the no lock is taken here in order to not serialize
1156          * the TX flow. This isn't dangerous because scheduling
1157          * mvm->add_stream_wk can't ruin the state, and if we DON'T
1158          * schedule it due to some race condition then next TX we get
1159          * here we will.
1160          */
1161         if (unlikely(mvm->queue_info[txq_id].status ==
1162                  IWL_MVM_QUEUE_SHARED &&
1163                  iwl_mvm_txq_should_update(mvm, txq_id)))
1164             schedule_work(&mvm->add_stream_wk);
1165     }
1166 
1167     IWL_DEBUG_TX(mvm, "TX to [%d|%d] Q:%d - seq: 0x%x len %d\n",
1168              mvmsta->sta_id, tid, txq_id,
1169              IEEE80211_SEQ_TO_SN(seq_number), skb->len);
1170 
1171     /* From now on, we cannot access info->control */
1172     iwl_mvm_skb_prepare_status(skb, dev_cmd);
1173 
1174     if (ieee80211_is_data(fc))
1175         iwl_mvm_mei_tx_copy_to_csme(mvm, skb,
1176                         info->control.hw_key ?
1177                         info->control.hw_key->iv_len : 0);
1178 
1179     if (iwl_trans_tx(mvm->trans, skb, dev_cmd, txq_id))
1180         goto drop_unlock_sta;
1181 
1182     if (tid < IWL_MAX_TID_COUNT && !ieee80211_has_morefrags(fc))
1183         mvmsta->tid_data[tid].seq_number = seq_number + 0x10;
1184 
1185     spin_unlock(&mvmsta->lock);
1186 
1187     if (iwl_mvm_tx_pkt_queued(mvm, mvmsta,
1188                   tid == IWL_MAX_TID_COUNT ? 0 : tid))
1189         goto drop;
1190 
1191     return 0;
1192 
1193 drop_unlock_sta:
1194     iwl_trans_free_tx_cmd(mvm->trans, dev_cmd);
1195     spin_unlock(&mvmsta->lock);
1196 drop:
1197     IWL_DEBUG_TX(mvm, "TX to [%d|%d] dropped\n", mvmsta->sta_id, tid);
1198     return -1;
1199 }
1200 
1201 int iwl_mvm_tx_skb_sta(struct iwl_mvm *mvm, struct sk_buff *skb,
1202                struct ieee80211_sta *sta)
1203 {
1204     struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1205     struct ieee80211_tx_info info;
1206     struct sk_buff_head mpdus_skbs;
1207     unsigned int payload_len;
1208     int ret;
1209 
1210     if (WARN_ON_ONCE(!mvmsta))
1211         return -1;
1212 
1213     if (WARN_ON_ONCE(mvmsta->sta_id == IWL_MVM_INVALID_STA))
1214         return -1;
1215 
1216     memcpy(&info, skb->cb, sizeof(info));
1217 
1218     if (!skb_is_gso(skb))
1219         return iwl_mvm_tx_mpdu(mvm, skb, &info, sta);
1220 
1221     payload_len = skb_tail_pointer(skb) - skb_transport_header(skb) -
1222         tcp_hdrlen(skb) + skb->data_len;
1223 
1224     if (payload_len <= skb_shinfo(skb)->gso_size)
1225         return iwl_mvm_tx_mpdu(mvm, skb, &info, sta);
1226 
1227     __skb_queue_head_init(&mpdus_skbs);
1228 
1229     ret = iwl_mvm_tx_tso(mvm, skb, &info, sta, &mpdus_skbs);
1230     if (ret)
1231         return ret;
1232 
1233     if (WARN_ON(skb_queue_empty(&mpdus_skbs)))
1234         return ret;
1235 
1236     while (!skb_queue_empty(&mpdus_skbs)) {
1237         skb = __skb_dequeue(&mpdus_skbs);
1238 
1239         ret = iwl_mvm_tx_mpdu(mvm, skb, &info, sta);
1240         if (ret) {
1241             __skb_queue_purge(&mpdus_skbs);
1242             return ret;
1243         }
1244     }
1245 
1246     return 0;
1247 }
1248 
1249 static void iwl_mvm_check_ratid_empty(struct iwl_mvm *mvm,
1250                       struct ieee80211_sta *sta, u8 tid)
1251 {
1252     struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1253     struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
1254     struct ieee80211_vif *vif = mvmsta->vif;
1255     u16 normalized_ssn;
1256 
1257     lockdep_assert_held(&mvmsta->lock);
1258 
1259     if ((tid_data->state == IWL_AGG_ON ||
1260          tid_data->state == IWL_EMPTYING_HW_QUEUE_DELBA) &&
1261         iwl_mvm_tid_queued(mvm, tid_data) == 0) {
1262         /*
1263          * Now that this aggregation or DQA queue is empty tell
1264          * mac80211 so it knows we no longer have frames buffered for
1265          * the station on this TID (for the TIM bitmap calculation.)
1266          */
1267         ieee80211_sta_set_buffered(sta, tid, false);
1268     }
1269 
1270     /*
1271      * In 22000 HW, the next_reclaimed index is only 8 bit, so we'll need
1272      * to align the wrap around of ssn so we compare relevant values.
1273      */
1274     normalized_ssn = tid_data->ssn;
1275     if (mvm->trans->trans_cfg->gen2)
1276         normalized_ssn &= 0xff;
1277 
1278     if (normalized_ssn != tid_data->next_reclaimed)
1279         return;
1280 
1281     switch (tid_data->state) {
1282     case IWL_EMPTYING_HW_QUEUE_ADDBA:
1283         IWL_DEBUG_TX_QUEUES(mvm,
1284                     "Can continue addBA flow ssn = next_recl = %d\n",
1285                     tid_data->next_reclaimed);
1286         tid_data->state = IWL_AGG_STARTING;
1287         ieee80211_start_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1288         break;
1289 
1290     case IWL_EMPTYING_HW_QUEUE_DELBA:
1291         IWL_DEBUG_TX_QUEUES(mvm,
1292                     "Can continue DELBA flow ssn = next_recl = %d\n",
1293                     tid_data->next_reclaimed);
1294         tid_data->state = IWL_AGG_OFF;
1295         ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1296         break;
1297 
1298     default:
1299         break;
1300     }
1301 }
1302 
1303 #ifdef CONFIG_IWLWIFI_DEBUG
1304 const char *iwl_mvm_get_tx_fail_reason(u32 status)
1305 {
1306 #define TX_STATUS_FAIL(x) case TX_STATUS_FAIL_ ## x: return #x
1307 #define TX_STATUS_POSTPONE(x) case TX_STATUS_POSTPONE_ ## x: return #x
1308 
1309     switch (status & TX_STATUS_MSK) {
1310     case TX_STATUS_SUCCESS:
1311         return "SUCCESS";
1312     TX_STATUS_POSTPONE(DELAY);
1313     TX_STATUS_POSTPONE(FEW_BYTES);
1314     TX_STATUS_POSTPONE(BT_PRIO);
1315     TX_STATUS_POSTPONE(QUIET_PERIOD);
1316     TX_STATUS_POSTPONE(CALC_TTAK);
1317     TX_STATUS_FAIL(INTERNAL_CROSSED_RETRY);
1318     TX_STATUS_FAIL(SHORT_LIMIT);
1319     TX_STATUS_FAIL(LONG_LIMIT);
1320     TX_STATUS_FAIL(UNDERRUN);
1321     TX_STATUS_FAIL(DRAIN_FLOW);
1322     TX_STATUS_FAIL(RFKILL_FLUSH);
1323     TX_STATUS_FAIL(LIFE_EXPIRE);
1324     TX_STATUS_FAIL(DEST_PS);
1325     TX_STATUS_FAIL(HOST_ABORTED);
1326     TX_STATUS_FAIL(BT_RETRY);
1327     TX_STATUS_FAIL(STA_INVALID);
1328     TX_STATUS_FAIL(FRAG_DROPPED);
1329     TX_STATUS_FAIL(TID_DISABLE);
1330     TX_STATUS_FAIL(FIFO_FLUSHED);
1331     TX_STATUS_FAIL(SMALL_CF_POLL);
1332     TX_STATUS_FAIL(FW_DROP);
1333     TX_STATUS_FAIL(STA_COLOR_MISMATCH);
1334     }
1335 
1336     return "UNKNOWN";
1337 
1338 #undef TX_STATUS_FAIL
1339 #undef TX_STATUS_POSTPONE
1340 }
1341 #endif /* CONFIG_IWLWIFI_DEBUG */
1342 
1343 static int iwl_mvm_get_hwrate_chan_width(u32 chan_width)
1344 {
1345     switch (chan_width) {
1346     case RATE_MCS_CHAN_WIDTH_20:
1347         return 0;
1348     case RATE_MCS_CHAN_WIDTH_40:
1349         return IEEE80211_TX_RC_40_MHZ_WIDTH;
1350     case RATE_MCS_CHAN_WIDTH_80:
1351         return IEEE80211_TX_RC_80_MHZ_WIDTH;
1352     case RATE_MCS_CHAN_WIDTH_160:
1353         return IEEE80211_TX_RC_160_MHZ_WIDTH;
1354     default:
1355         return 0;
1356     }
1357 }
1358 
1359 void iwl_mvm_hwrate_to_tx_rate(u32 rate_n_flags,
1360                    enum nl80211_band band,
1361                    struct ieee80211_tx_rate *r)
1362 {
1363     u32 format = rate_n_flags & RATE_MCS_MOD_TYPE_MSK;
1364     u32 rate = format == RATE_MCS_HT_MSK ?
1365         RATE_HT_MCS_INDEX(rate_n_flags) :
1366         rate_n_flags & RATE_MCS_CODE_MSK;
1367 
1368     r->flags |=
1369         iwl_mvm_get_hwrate_chan_width(rate_n_flags &
1370                           RATE_MCS_CHAN_WIDTH_MSK);
1371 
1372     if (rate_n_flags & RATE_MCS_SGI_MSK)
1373         r->flags |= IEEE80211_TX_RC_SHORT_GI;
1374     if (format ==  RATE_MCS_HT_MSK) {
1375         r->flags |= IEEE80211_TX_RC_MCS;
1376         r->idx = rate;
1377     } else if (format ==  RATE_MCS_VHT_MSK) {
1378         ieee80211_rate_set_vht(r, rate,
1379                        ((rate_n_flags & RATE_MCS_NSS_MSK) >>
1380                     RATE_MCS_NSS_POS) + 1);
1381         r->flags |= IEEE80211_TX_RC_VHT_MCS;
1382     } else if (format == RATE_MCS_HE_MSK) {
1383         /* mac80211 cannot do this without ieee80211_tx_status_ext()
1384          * but it only matters for radiotap */
1385         r->idx = 0;
1386     } else {
1387         r->idx = iwl_mvm_legacy_hw_idx_to_mac80211_idx(rate_n_flags,
1388                                    band);
1389     }
1390 }
1391 
1392 void iwl_mvm_hwrate_to_tx_rate_v1(u32 rate_n_flags,
1393                   enum nl80211_band band,
1394                   struct ieee80211_tx_rate *r)
1395 {
1396     if (rate_n_flags & RATE_HT_MCS_GF_MSK)
1397         r->flags |= IEEE80211_TX_RC_GREEN_FIELD;
1398 
1399     r->flags |=
1400         iwl_mvm_get_hwrate_chan_width(rate_n_flags &
1401                           RATE_MCS_CHAN_WIDTH_MSK_V1);
1402 
1403     if (rate_n_flags & RATE_MCS_SGI_MSK_V1)
1404         r->flags |= IEEE80211_TX_RC_SHORT_GI;
1405     if (rate_n_flags & RATE_MCS_HT_MSK_V1) {
1406         r->flags |= IEEE80211_TX_RC_MCS;
1407         r->idx = rate_n_flags & RATE_HT_MCS_INDEX_MSK_V1;
1408     } else if (rate_n_flags & RATE_MCS_VHT_MSK_V1) {
1409         ieee80211_rate_set_vht(
1410             r, rate_n_flags & RATE_VHT_MCS_RATE_CODE_MSK,
1411             ((rate_n_flags & RATE_VHT_MCS_NSS_MSK) >>
1412                         RATE_VHT_MCS_NSS_POS) + 1);
1413         r->flags |= IEEE80211_TX_RC_VHT_MCS;
1414     } else {
1415         r->idx = iwl_mvm_legacy_rate_to_mac80211_idx(rate_n_flags,
1416                                  band);
1417     }
1418 }
1419 
1420 /*
1421  * translate ucode response to mac80211 tx status control values
1422  */
1423 static void iwl_mvm_hwrate_to_tx_status(const struct iwl_fw *fw,
1424                     u32 rate_n_flags,
1425                     struct ieee80211_tx_info *info)
1426 {
1427     struct ieee80211_tx_rate *r = &info->status.rates[0];
1428 
1429     if (iwl_fw_lookup_notif_ver(fw, LONG_GROUP,
1430                     TX_CMD, 0) <= 6)
1431         rate_n_flags = iwl_new_rate_from_v1(rate_n_flags);
1432 
1433     info->status.antenna =
1434         ((rate_n_flags & RATE_MCS_ANT_AB_MSK) >> RATE_MCS_ANT_POS);
1435     iwl_mvm_hwrate_to_tx_rate(rate_n_flags,
1436                   info->band, r);
1437 }
1438 
1439 static void iwl_mvm_tx_status_check_trigger(struct iwl_mvm *mvm,
1440                         u32 status, __le16 frame_control)
1441 {
1442     struct iwl_fw_dbg_trigger_tlv *trig;
1443     struct iwl_fw_dbg_trigger_tx_status *status_trig;
1444     int i;
1445 
1446     if ((status & TX_STATUS_MSK) != TX_STATUS_SUCCESS) {
1447         enum iwl_fw_ini_time_point tp =
1448             IWL_FW_INI_TIME_POINT_TX_FAILED;
1449 
1450         if (ieee80211_is_action(frame_control))
1451             tp = IWL_FW_INI_TIME_POINT_TX_WFD_ACTION_FRAME_FAILED;
1452 
1453         iwl_dbg_tlv_time_point(&mvm->fwrt,
1454                        tp, NULL);
1455         return;
1456     }
1457 
1458     trig = iwl_fw_dbg_trigger_on(&mvm->fwrt, NULL,
1459                      FW_DBG_TRIGGER_TX_STATUS);
1460     if (!trig)
1461         return;
1462 
1463     status_trig = (void *)trig->data;
1464 
1465     for (i = 0; i < ARRAY_SIZE(status_trig->statuses); i++) {
1466         /* don't collect on status 0 */
1467         if (!status_trig->statuses[i].status)
1468             break;
1469 
1470         if (status_trig->statuses[i].status != (status & TX_STATUS_MSK))
1471             continue;
1472 
1473         iwl_fw_dbg_collect_trig(&mvm->fwrt, trig,
1474                     "Tx status %d was received",
1475                     status & TX_STATUS_MSK);
1476         break;
1477     }
1478 }
1479 
1480 /*
1481  * iwl_mvm_get_scd_ssn - returns the SSN of the SCD
1482  * @tx_resp: the Tx response from the fw (agg or non-agg)
1483  *
1484  * When the fw sends an AMPDU, it fetches the MPDUs one after the other. Since
1485  * it can't know that everything will go well until the end of the AMPDU, it
1486  * can't know in advance the number of MPDUs that will be sent in the current
1487  * batch. This is why it writes the agg Tx response while it fetches the MPDUs.
1488  * Hence, it can't know in advance what the SSN of the SCD will be at the end
1489  * of the batch. This is why the SSN of the SCD is written at the end of the
1490  * whole struct at a variable offset. This function knows how to cope with the
1491  * variable offset and returns the SSN of the SCD.
1492  */
1493 static inline u32 iwl_mvm_get_scd_ssn(struct iwl_mvm *mvm,
1494                       struct iwl_mvm_tx_resp *tx_resp)
1495 {
1496     return le32_to_cpup((__le32 *)iwl_mvm_get_agg_status(mvm, tx_resp) +
1497                 tx_resp->frame_count) & 0xfff;
1498 }
1499 
1500 static void iwl_mvm_rx_tx_cmd_single(struct iwl_mvm *mvm,
1501                      struct iwl_rx_packet *pkt)
1502 {
1503     struct ieee80211_sta *sta;
1504     u16 sequence = le16_to_cpu(pkt->hdr.sequence);
1505     int txq_id = SEQ_TO_QUEUE(sequence);
1506     /* struct iwl_mvm_tx_resp_v3 is almost the same */
1507     struct iwl_mvm_tx_resp *tx_resp = (void *)pkt->data;
1508     int sta_id = IWL_MVM_TX_RES_GET_RA(tx_resp->ra_tid);
1509     int tid = IWL_MVM_TX_RES_GET_TID(tx_resp->ra_tid);
1510     struct agg_tx_status *agg_status =
1511         iwl_mvm_get_agg_status(mvm, tx_resp);
1512     u32 status = le16_to_cpu(agg_status->status);
1513     u16 ssn = iwl_mvm_get_scd_ssn(mvm, tx_resp);
1514     struct sk_buff_head skbs;
1515     u8 skb_freed = 0;
1516     u8 lq_color;
1517     u16 next_reclaimed, seq_ctl;
1518     bool is_ndp = false;
1519 
1520     __skb_queue_head_init(&skbs);
1521 
1522     if (iwl_mvm_has_new_tx_api(mvm))
1523         txq_id = le16_to_cpu(tx_resp->tx_queue);
1524 
1525     seq_ctl = le16_to_cpu(tx_resp->seq_ctl);
1526 
1527     /* we can free until ssn % q.n_bd not inclusive */
1528     iwl_trans_reclaim(mvm->trans, txq_id, ssn, &skbs);
1529 
1530     while (!skb_queue_empty(&skbs)) {
1531         struct sk_buff *skb = __skb_dequeue(&skbs);
1532         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1533         struct ieee80211_hdr *hdr = (void *)skb->data;
1534         bool flushed = false;
1535 
1536         skb_freed++;
1537 
1538         iwl_trans_free_tx_cmd(mvm->trans, info->driver_data[1]);
1539 
1540         memset(&info->status, 0, sizeof(info->status));
1541 
1542         /* inform mac80211 about what happened with the frame */
1543         switch (status & TX_STATUS_MSK) {
1544         case TX_STATUS_SUCCESS:
1545         case TX_STATUS_DIRECT_DONE:
1546             info->flags |= IEEE80211_TX_STAT_ACK;
1547             break;
1548         case TX_STATUS_FAIL_FIFO_FLUSHED:
1549         case TX_STATUS_FAIL_DRAIN_FLOW:
1550             flushed = true;
1551             break;
1552         case TX_STATUS_FAIL_DEST_PS:
1553             /* the FW should have stopped the queue and not
1554              * return this status
1555              */
1556             IWL_ERR_LIMIT(mvm,
1557                       "FW reported TX filtered, status=0x%x, FC=0x%x\n",
1558                       status, le16_to_cpu(hdr->frame_control));
1559             info->flags |= IEEE80211_TX_STAT_TX_FILTERED;
1560             break;
1561         default:
1562             break;
1563         }
1564 
1565         if ((status & TX_STATUS_MSK) != TX_STATUS_SUCCESS &&
1566             ieee80211_is_mgmt(hdr->frame_control))
1567             iwl_mvm_toggle_tx_ant(mvm, &mvm->mgmt_last_antenna_idx);
1568 
1569         /*
1570          * If we are freeing multiple frames, mark all the frames
1571          * but the first one as acked, since they were acknowledged
1572          * before
1573          * */
1574         if (skb_freed > 1)
1575             info->flags |= IEEE80211_TX_STAT_ACK;
1576 
1577         iwl_mvm_tx_status_check_trigger(mvm, status, hdr->frame_control);
1578 
1579         info->status.rates[0].count = tx_resp->failure_frame + 1;
1580 
1581         iwl_mvm_hwrate_to_tx_status(mvm->fw,
1582                         le32_to_cpu(tx_resp->initial_rate),
1583                         info);
1584 
1585         /* Don't assign the converted initial_rate, because driver
1586          * TLC uses this and doesn't support the new FW rate
1587          */
1588         info->status.status_driver_data[1] =
1589             (void *)(uintptr_t)le32_to_cpu(tx_resp->initial_rate);
1590 
1591         /* Single frame failure in an AMPDU queue => send BAR */
1592         if (info->flags & IEEE80211_TX_CTL_AMPDU &&
1593             !(info->flags & IEEE80211_TX_STAT_ACK) &&
1594             !(info->flags & IEEE80211_TX_STAT_TX_FILTERED) && !flushed)
1595             info->flags |= IEEE80211_TX_STAT_AMPDU_NO_BACK;
1596         info->flags &= ~IEEE80211_TX_CTL_AMPDU;
1597 
1598         /* W/A FW bug: seq_ctl is wrong upon failure / BAR frame */
1599         if (ieee80211_is_back_req(hdr->frame_control))
1600             seq_ctl = 0;
1601         else if (status != TX_STATUS_SUCCESS)
1602             seq_ctl = le16_to_cpu(hdr->seq_ctrl);
1603 
1604         if (unlikely(!seq_ctl)) {
1605             /*
1606              * If it is an NDP, we can't update next_reclaim since
1607              * its sequence control is 0. Note that for that same
1608              * reason, NDPs are never sent to A-MPDU'able queues
1609              * so that we can never have more than one freed frame
1610              * for a single Tx resonse (see WARN_ON below).
1611              */
1612             if (ieee80211_is_qos_nullfunc(hdr->frame_control))
1613                 is_ndp = true;
1614         }
1615 
1616         /*
1617          * TODO: this is not accurate if we are freeing more than one
1618          * packet.
1619          */
1620         info->status.tx_time =
1621             le16_to_cpu(tx_resp->wireless_media_time);
1622         BUILD_BUG_ON(ARRAY_SIZE(info->status.status_driver_data) < 1);
1623         lq_color = TX_RES_RATE_TABLE_COL_GET(tx_resp->tlc_info);
1624         info->status.status_driver_data[0] =
1625             RS_DRV_DATA_PACK(lq_color, tx_resp->reduced_tpc);
1626 
1627         ieee80211_tx_status(mvm->hw, skb);
1628     }
1629 
1630     /* This is an aggregation queue or might become one, so we use
1631      * the ssn since: ssn = wifi seq_num % 256.
1632      * The seq_ctl is the sequence control of the packet to which
1633      * this Tx response relates. But if there is a hole in the
1634      * bitmap of the BA we received, this Tx response may allow to
1635      * reclaim the hole and all the subsequent packets that were
1636      * already acked. In that case, seq_ctl != ssn, and the next
1637      * packet to be reclaimed will be ssn and not seq_ctl. In that
1638      * case, several packets will be reclaimed even if
1639      * frame_count = 1.
1640      *
1641      * The ssn is the index (% 256) of the latest packet that has
1642      * treated (acked / dropped) + 1.
1643      */
1644     next_reclaimed = ssn;
1645 
1646     IWL_DEBUG_TX_REPLY(mvm,
1647                "TXQ %d status %s (0x%08x)\n",
1648                txq_id, iwl_mvm_get_tx_fail_reason(status), status);
1649 
1650     IWL_DEBUG_TX_REPLY(mvm,
1651                "\t\t\t\tinitial_rate 0x%x retries %d, idx=%d ssn=%d next_reclaimed=0x%x seq_ctl=0x%x\n",
1652                le32_to_cpu(tx_resp->initial_rate),
1653                tx_resp->failure_frame, SEQ_TO_INDEX(sequence),
1654                ssn, next_reclaimed, seq_ctl);
1655 
1656     rcu_read_lock();
1657 
1658     sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
1659     /*
1660      * sta can't be NULL otherwise it'd mean that the sta has been freed in
1661      * the firmware while we still have packets for it in the Tx queues.
1662      */
1663     if (WARN_ON_ONCE(!sta))
1664         goto out;
1665 
1666     if (!IS_ERR(sta)) {
1667         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1668 
1669         iwl_mvm_tx_airtime(mvm, mvmsta,
1670                    le16_to_cpu(tx_resp->wireless_media_time));
1671 
1672         if ((status & TX_STATUS_MSK) != TX_STATUS_SUCCESS &&
1673             mvmsta->sta_state < IEEE80211_STA_AUTHORIZED)
1674             iwl_mvm_toggle_tx_ant(mvm, &mvmsta->tx_ant);
1675 
1676         if (sta->wme && tid != IWL_MGMT_TID) {
1677             struct iwl_mvm_tid_data *tid_data =
1678                 &mvmsta->tid_data[tid];
1679             bool send_eosp_ndp = false;
1680 
1681             spin_lock_bh(&mvmsta->lock);
1682 
1683             if (!is_ndp) {
1684                 tid_data->next_reclaimed = next_reclaimed;
1685                 IWL_DEBUG_TX_REPLY(mvm,
1686                            "Next reclaimed packet:%d\n",
1687                            next_reclaimed);
1688             } else {
1689                 IWL_DEBUG_TX_REPLY(mvm,
1690                            "NDP - don't update next_reclaimed\n");
1691             }
1692 
1693             iwl_mvm_check_ratid_empty(mvm, sta, tid);
1694 
1695             if (mvmsta->sleep_tx_count) {
1696                 mvmsta->sleep_tx_count--;
1697                 if (mvmsta->sleep_tx_count &&
1698                     !iwl_mvm_tid_queued(mvm, tid_data)) {
1699                     /*
1700                      * The number of frames in the queue
1701                      * dropped to 0 even if we sent less
1702                      * frames than we thought we had on the
1703                      * Tx queue.
1704                      * This means we had holes in the BA
1705                      * window that we just filled, ask
1706                      * mac80211 to send EOSP since the
1707                      * firmware won't know how to do that.
1708                      * Send NDP and the firmware will send
1709                      * EOSP notification that will trigger
1710                      * a call to ieee80211_sta_eosp().
1711                      */
1712                     send_eosp_ndp = true;
1713                 }
1714             }
1715 
1716             spin_unlock_bh(&mvmsta->lock);
1717             if (send_eosp_ndp) {
1718                 iwl_mvm_sta_modify_sleep_tx_count(mvm, sta,
1719                     IEEE80211_FRAME_RELEASE_UAPSD,
1720                     1, tid, false, false);
1721                 mvmsta->sleep_tx_count = 0;
1722                 ieee80211_send_eosp_nullfunc(sta, tid);
1723             }
1724         }
1725 
1726         if (mvmsta->next_status_eosp) {
1727             mvmsta->next_status_eosp = false;
1728             ieee80211_sta_eosp(sta);
1729         }
1730     }
1731 out:
1732     rcu_read_unlock();
1733 }
1734 
1735 #ifdef CONFIG_IWLWIFI_DEBUG
1736 #define AGG_TX_STATE_(x) case AGG_TX_STATE_ ## x: return #x
1737 static const char *iwl_get_agg_tx_status(u16 status)
1738 {
1739     switch (status & AGG_TX_STATE_STATUS_MSK) {
1740     AGG_TX_STATE_(TRANSMITTED);
1741     AGG_TX_STATE_(UNDERRUN);
1742     AGG_TX_STATE_(BT_PRIO);
1743     AGG_TX_STATE_(FEW_BYTES);
1744     AGG_TX_STATE_(ABORT);
1745     AGG_TX_STATE_(TX_ON_AIR_DROP);
1746     AGG_TX_STATE_(LAST_SENT_TRY_CNT);
1747     AGG_TX_STATE_(LAST_SENT_BT_KILL);
1748     AGG_TX_STATE_(SCD_QUERY);
1749     AGG_TX_STATE_(TEST_BAD_CRC32);
1750     AGG_TX_STATE_(RESPONSE);
1751     AGG_TX_STATE_(DUMP_TX);
1752     AGG_TX_STATE_(DELAY_TX);
1753     }
1754 
1755     return "UNKNOWN";
1756 }
1757 
1758 static void iwl_mvm_rx_tx_cmd_agg_dbg(struct iwl_mvm *mvm,
1759                       struct iwl_rx_packet *pkt)
1760 {
1761     struct iwl_mvm_tx_resp *tx_resp = (void *)pkt->data;
1762     struct agg_tx_status *frame_status =
1763         iwl_mvm_get_agg_status(mvm, tx_resp);
1764     int i;
1765     bool tirgger_timepoint = false;
1766 
1767     for (i = 0; i < tx_resp->frame_count; i++) {
1768         u16 fstatus = le16_to_cpu(frame_status[i].status);
1769         /* In case one frame wasn't transmitted trigger time point */
1770         tirgger_timepoint |= ((fstatus & AGG_TX_STATE_STATUS_MSK) !=
1771                       AGG_TX_STATE_TRANSMITTED);
1772         IWL_DEBUG_TX_REPLY(mvm,
1773                    "status %s (0x%04x), try-count (%d) seq (0x%x)\n",
1774                    iwl_get_agg_tx_status(fstatus),
1775                    fstatus & AGG_TX_STATE_STATUS_MSK,
1776                    (fstatus & AGG_TX_STATE_TRY_CNT_MSK) >>
1777                     AGG_TX_STATE_TRY_CNT_POS,
1778                    le16_to_cpu(frame_status[i].sequence));
1779     }
1780 
1781     if (tirgger_timepoint)
1782         iwl_dbg_tlv_time_point(&mvm->fwrt,
1783                        IWL_FW_INI_TIME_POINT_TX_FAILED, NULL);
1784 
1785 }
1786 #else
1787 static void iwl_mvm_rx_tx_cmd_agg_dbg(struct iwl_mvm *mvm,
1788                       struct iwl_rx_packet *pkt)
1789 {}
1790 #endif /* CONFIG_IWLWIFI_DEBUG */
1791 
1792 static void iwl_mvm_rx_tx_cmd_agg(struct iwl_mvm *mvm,
1793                   struct iwl_rx_packet *pkt)
1794 {
1795     struct iwl_mvm_tx_resp *tx_resp = (void *)pkt->data;
1796     int sta_id = IWL_MVM_TX_RES_GET_RA(tx_resp->ra_tid);
1797     int tid = IWL_MVM_TX_RES_GET_TID(tx_resp->ra_tid);
1798     u16 sequence = le16_to_cpu(pkt->hdr.sequence);
1799     struct iwl_mvm_sta *mvmsta;
1800     int queue = SEQ_TO_QUEUE(sequence);
1801     struct ieee80211_sta *sta;
1802 
1803     if (WARN_ON_ONCE(queue < IWL_MVM_DQA_MIN_DATA_QUEUE &&
1804              (queue != IWL_MVM_DQA_BSS_CLIENT_QUEUE)))
1805         return;
1806 
1807     iwl_mvm_rx_tx_cmd_agg_dbg(mvm, pkt);
1808 
1809     rcu_read_lock();
1810 
1811     mvmsta = iwl_mvm_sta_from_staid_rcu(mvm, sta_id);
1812 
1813     sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
1814     if (WARN_ON_ONCE(!sta || !sta->wme)) {
1815         rcu_read_unlock();
1816         return;
1817     }
1818 
1819     if (!WARN_ON_ONCE(!mvmsta)) {
1820         mvmsta->tid_data[tid].rate_n_flags =
1821             le32_to_cpu(tx_resp->initial_rate);
1822         mvmsta->tid_data[tid].tx_time =
1823             le16_to_cpu(tx_resp->wireless_media_time);
1824         mvmsta->tid_data[tid].lq_color =
1825             TX_RES_RATE_TABLE_COL_GET(tx_resp->tlc_info);
1826         iwl_mvm_tx_airtime(mvm, mvmsta,
1827                    le16_to_cpu(tx_resp->wireless_media_time));
1828     }
1829 
1830     rcu_read_unlock();
1831 }
1832 
1833 void iwl_mvm_rx_tx_cmd(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb)
1834 {
1835     struct iwl_rx_packet *pkt = rxb_addr(rxb);
1836     struct iwl_mvm_tx_resp *tx_resp = (void *)pkt->data;
1837 
1838     if (tx_resp->frame_count == 1)
1839         iwl_mvm_rx_tx_cmd_single(mvm, pkt);
1840     else
1841         iwl_mvm_rx_tx_cmd_agg(mvm, pkt);
1842 }
1843 
1844 static void iwl_mvm_tx_reclaim(struct iwl_mvm *mvm, int sta_id, int tid,
1845                    int txq, int index,
1846                    struct ieee80211_tx_info *tx_info, u32 rate,
1847                    bool is_flush)
1848 {
1849     struct sk_buff_head reclaimed_skbs;
1850     struct iwl_mvm_tid_data *tid_data = NULL;
1851     struct ieee80211_sta *sta;
1852     struct iwl_mvm_sta *mvmsta = NULL;
1853     struct sk_buff *skb;
1854     int freed;
1855 
1856     if (WARN_ONCE(sta_id >= mvm->fw->ucode_capa.num_stations ||
1857               tid > IWL_MAX_TID_COUNT,
1858               "sta_id %d tid %d", sta_id, tid))
1859         return;
1860 
1861     rcu_read_lock();
1862 
1863     sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
1864 
1865     /* Reclaiming frames for a station that has been deleted ? */
1866     if (WARN_ON_ONCE(!sta)) {
1867         rcu_read_unlock();
1868         return;
1869     }
1870 
1871     __skb_queue_head_init(&reclaimed_skbs);
1872 
1873     /*
1874      * Release all TFDs before the SSN, i.e. all TFDs in front of
1875      * block-ack window (we assume that they've been successfully
1876      * transmitted ... if not, it's too late anyway).
1877      */
1878     iwl_trans_reclaim(mvm->trans, txq, index, &reclaimed_skbs);
1879 
1880     skb_queue_walk(&reclaimed_skbs, skb) {
1881         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1882 
1883         iwl_trans_free_tx_cmd(mvm->trans, info->driver_data[1]);
1884 
1885         memset(&info->status, 0, sizeof(info->status));
1886         /* Packet was transmitted successfully, failures come as single
1887          * frames because before failing a frame the firmware transmits
1888          * it without aggregation at least once.
1889          */
1890         if (!is_flush)
1891             info->flags |= IEEE80211_TX_STAT_ACK;
1892     }
1893 
1894     /*
1895      * It's possible to get a BA response after invalidating the rcu (rcu is
1896      * invalidated in order to prevent new Tx from being sent, but there may
1897      * be some frames already in-flight).
1898      * In this case we just want to reclaim, and could skip all the
1899      * sta-dependent stuff since it's in the middle of being removed
1900      * anyways.
1901      */
1902     if (IS_ERR(sta))
1903         goto out;
1904 
1905     mvmsta = iwl_mvm_sta_from_mac80211(sta);
1906     tid_data = &mvmsta->tid_data[tid];
1907 
1908     if (tid_data->txq_id != txq) {
1909         IWL_ERR(mvm,
1910             "invalid reclaim request: Q %d, tid %d\n",
1911             tid_data->txq_id, tid);
1912         rcu_read_unlock();
1913         return;
1914     }
1915 
1916     spin_lock_bh(&mvmsta->lock);
1917 
1918     tid_data->next_reclaimed = index;
1919 
1920     iwl_mvm_check_ratid_empty(mvm, sta, tid);
1921 
1922     freed = 0;
1923 
1924     /* pack lq color from tid_data along the reduced txp */
1925     tx_info->status.status_driver_data[0] =
1926         RS_DRV_DATA_PACK(tid_data->lq_color,
1927                  tx_info->status.status_driver_data[0]);
1928     tx_info->status.status_driver_data[1] = (void *)(uintptr_t)rate;
1929 
1930     skb_queue_walk(&reclaimed_skbs, skb) {
1931         struct ieee80211_hdr *hdr = (void *)skb->data;
1932         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1933 
1934         if (!is_flush) {
1935             if (ieee80211_is_data_qos(hdr->frame_control))
1936                 freed++;
1937             else
1938                 WARN_ON_ONCE(tid != IWL_MAX_TID_COUNT);
1939         }
1940 
1941         /* this is the first skb we deliver in this batch */
1942         /* put the rate scaling data there */
1943         if (freed == 1) {
1944             info->flags |= IEEE80211_TX_STAT_AMPDU;
1945             memcpy(&info->status, &tx_info->status,
1946                    sizeof(tx_info->status));
1947             iwl_mvm_hwrate_to_tx_status(mvm->fw, rate, info);
1948         }
1949     }
1950 
1951     spin_unlock_bh(&mvmsta->lock);
1952 
1953     /* We got a BA notif with 0 acked or scd_ssn didn't progress which is
1954      * possible (i.e. first MPDU in the aggregation wasn't acked)
1955      * Still it's important to update RS about sent vs. acked.
1956      */
1957     if (!is_flush && skb_queue_empty(&reclaimed_skbs)) {
1958         struct ieee80211_chanctx_conf *chanctx_conf = NULL;
1959 
1960         if (mvmsta->vif)
1961             chanctx_conf =
1962                 rcu_dereference(mvmsta->vif->bss_conf.chanctx_conf);
1963 
1964         if (WARN_ON_ONCE(!chanctx_conf))
1965             goto out;
1966 
1967         tx_info->band = chanctx_conf->def.chan->band;
1968         iwl_mvm_hwrate_to_tx_status(mvm->fw, rate, tx_info);
1969 
1970         if (!iwl_mvm_has_tlc_offload(mvm)) {
1971             IWL_DEBUG_TX_REPLY(mvm,
1972                        "No reclaim. Update rs directly\n");
1973             iwl_mvm_rs_tx_status(mvm, sta, tid, tx_info, false);
1974         }
1975     }
1976 
1977 out:
1978     rcu_read_unlock();
1979 
1980     while (!skb_queue_empty(&reclaimed_skbs)) {
1981         skb = __skb_dequeue(&reclaimed_skbs);
1982         ieee80211_tx_status(mvm->hw, skb);
1983     }
1984 }
1985 
1986 void iwl_mvm_rx_ba_notif(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb)
1987 {
1988     struct iwl_rx_packet *pkt = rxb_addr(rxb);
1989     unsigned int pkt_len = iwl_rx_packet_payload_len(pkt);
1990     int sta_id, tid, txq, index;
1991     struct ieee80211_tx_info ba_info = {};
1992     struct iwl_mvm_ba_notif *ba_notif;
1993     struct iwl_mvm_tid_data *tid_data;
1994     struct iwl_mvm_sta *mvmsta;
1995 
1996     ba_info.flags = IEEE80211_TX_STAT_AMPDU;
1997 
1998     if (iwl_mvm_has_new_tx_api(mvm)) {
1999         struct iwl_mvm_compressed_ba_notif *ba_res =
2000             (void *)pkt->data;
2001         u8 lq_color = TX_RES_RATE_TABLE_COL_GET(ba_res->tlc_rate_info);
2002         u16 tfd_cnt;
2003         int i;
2004 
2005         if (unlikely(sizeof(*ba_res) > pkt_len))
2006             return;
2007 
2008         sta_id = ba_res->sta_id;
2009         ba_info.status.ampdu_ack_len = (u8)le16_to_cpu(ba_res->done);
2010         ba_info.status.ampdu_len = (u8)le16_to_cpu(ba_res->txed);
2011         ba_info.status.tx_time =
2012             (u16)le32_to_cpu(ba_res->wireless_time);
2013         ba_info.status.status_driver_data[0] =
2014             (void *)(uintptr_t)ba_res->reduced_txp;
2015 
2016         tfd_cnt = le16_to_cpu(ba_res->tfd_cnt);
2017         if (!tfd_cnt || struct_size(ba_res, tfd, tfd_cnt) > pkt_len)
2018             return;
2019 
2020         rcu_read_lock();
2021 
2022         mvmsta = iwl_mvm_sta_from_staid_rcu(mvm, sta_id);
2023         /*
2024          * It's possible to get a BA response after invalidating the rcu
2025          * (rcu is invalidated in order to prevent new Tx from being
2026          * sent, but there may be some frames already in-flight).
2027          * In this case we just want to reclaim, and could skip all the
2028          * sta-dependent stuff since it's in the middle of being removed
2029          * anyways.
2030          */
2031 
2032         /* Free per TID */
2033         for (i = 0; i < tfd_cnt; i++) {
2034             struct iwl_mvm_compressed_ba_tfd *ba_tfd =
2035                 &ba_res->tfd[i];
2036 
2037             tid = ba_tfd->tid;
2038             if (tid == IWL_MGMT_TID)
2039                 tid = IWL_MAX_TID_COUNT;
2040 
2041             if (mvmsta)
2042                 mvmsta->tid_data[i].lq_color = lq_color;
2043 
2044             iwl_mvm_tx_reclaim(mvm, sta_id, tid,
2045                        (int)(le16_to_cpu(ba_tfd->q_num)),
2046                        le16_to_cpu(ba_tfd->tfd_index),
2047                        &ba_info,
2048                        le32_to_cpu(ba_res->tx_rate), false);
2049         }
2050 
2051         if (mvmsta)
2052             iwl_mvm_tx_airtime(mvm, mvmsta,
2053                        le32_to_cpu(ba_res->wireless_time));
2054         rcu_read_unlock();
2055 
2056         IWL_DEBUG_TX_REPLY(mvm,
2057                    "BA_NOTIFICATION Received from sta_id = %d, flags %x, sent:%d, acked:%d\n",
2058                    sta_id, le32_to_cpu(ba_res->flags),
2059                    le16_to_cpu(ba_res->txed),
2060                    le16_to_cpu(ba_res->done));
2061         return;
2062     }
2063 
2064     ba_notif = (void *)pkt->data;
2065     sta_id = ba_notif->sta_id;
2066     tid = ba_notif->tid;
2067     /* "flow" corresponds to Tx queue */
2068     txq = le16_to_cpu(ba_notif->scd_flow);
2069     /* "ssn" is start of block-ack Tx window, corresponds to index
2070      * (in Tx queue's circular buffer) of first TFD/frame in window */
2071     index = le16_to_cpu(ba_notif->scd_ssn);
2072 
2073     rcu_read_lock();
2074     mvmsta = iwl_mvm_sta_from_staid_rcu(mvm, sta_id);
2075     if (WARN_ON_ONCE(!mvmsta)) {
2076         rcu_read_unlock();
2077         return;
2078     }
2079 
2080     tid_data = &mvmsta->tid_data[tid];
2081 
2082     ba_info.status.ampdu_ack_len = ba_notif->txed_2_done;
2083     ba_info.status.ampdu_len = ba_notif->txed;
2084     ba_info.status.tx_time = tid_data->tx_time;
2085     ba_info.status.status_driver_data[0] =
2086         (void *)(uintptr_t)ba_notif->reduced_txp;
2087 
2088     rcu_read_unlock();
2089 
2090     iwl_mvm_tx_reclaim(mvm, sta_id, tid, txq, index, &ba_info,
2091                tid_data->rate_n_flags, false);
2092 
2093     IWL_DEBUG_TX_REPLY(mvm,
2094                "BA_NOTIFICATION Received from %pM, sta_id = %d\n",
2095                ba_notif->sta_addr, ba_notif->sta_id);
2096 
2097     IWL_DEBUG_TX_REPLY(mvm,
2098                "TID = %d, SeqCtl = %d, bitmap = 0x%llx, scd_flow = %d, scd_ssn = %d sent:%d, acked:%d\n",
2099                ba_notif->tid, le16_to_cpu(ba_notif->seq_ctl),
2100                le64_to_cpu(ba_notif->bitmap), txq, index,
2101                ba_notif->txed, ba_notif->txed_2_done);
2102 
2103     IWL_DEBUG_TX_REPLY(mvm, "reduced txp from ba notif %d\n",
2104                ba_notif->reduced_txp);
2105 }
2106 
2107 /*
2108  * Note that there are transports that buffer frames before they reach
2109  * the firmware. This means that after flush_tx_path is called, the
2110  * queue might not be empty. The race-free way to handle this is to:
2111  * 1) set the station as draining
2112  * 2) flush the Tx path
2113  * 3) wait for the transport queues to be empty
2114  */
2115 int iwl_mvm_flush_tx_path(struct iwl_mvm *mvm, u32 tfd_msk)
2116 {
2117     int ret;
2118     struct iwl_tx_path_flush_cmd_v1 flush_cmd = {
2119         .queues_ctl = cpu_to_le32(tfd_msk),
2120         .flush_ctl = cpu_to_le16(DUMP_TX_FIFO_FLUSH),
2121     };
2122 
2123     WARN_ON(iwl_mvm_has_new_tx_api(mvm));
2124     ret = iwl_mvm_send_cmd_pdu(mvm, TXPATH_FLUSH, 0,
2125                    sizeof(flush_cmd), &flush_cmd);
2126     if (ret)
2127         IWL_ERR(mvm, "Failed to send flush command (%d)\n", ret);
2128     return ret;
2129 }
2130 
2131 int iwl_mvm_flush_sta_tids(struct iwl_mvm *mvm, u32 sta_id, u16 tids)
2132 {
2133     int ret;
2134     struct iwl_tx_path_flush_cmd_rsp *rsp;
2135     struct iwl_tx_path_flush_cmd flush_cmd = {
2136         .sta_id = cpu_to_le32(sta_id),
2137         .tid_mask = cpu_to_le16(tids),
2138     };
2139 
2140     struct iwl_host_cmd cmd = {
2141         .id = TXPATH_FLUSH,
2142         .len = { sizeof(flush_cmd), },
2143         .data = { &flush_cmd, },
2144     };
2145 
2146     WARN_ON(!iwl_mvm_has_new_tx_api(mvm));
2147 
2148     if (iwl_fw_lookup_notif_ver(mvm->fw, LONG_GROUP, TXPATH_FLUSH, 0) > 0)
2149         cmd.flags |= CMD_WANT_SKB;
2150 
2151     IWL_DEBUG_TX_QUEUES(mvm, "flush for sta id %d tid mask 0x%x\n",
2152                 sta_id, tids);
2153 
2154     ret = iwl_mvm_send_cmd(mvm, &cmd);
2155 
2156     if (ret) {
2157         IWL_ERR(mvm, "Failed to send flush command (%d)\n", ret);
2158         return ret;
2159     }
2160 
2161     if (cmd.flags & CMD_WANT_SKB) {
2162         int i;
2163         int num_flushed_queues;
2164 
2165         if (WARN_ON_ONCE(iwl_rx_packet_payload_len(cmd.resp_pkt) != sizeof(*rsp))) {
2166             ret = -EIO;
2167             goto free_rsp;
2168         }
2169 
2170         rsp = (void *)cmd.resp_pkt->data;
2171 
2172         if (WARN_ONCE(le16_to_cpu(rsp->sta_id) != sta_id,
2173                   "sta_id %d != rsp_sta_id %d",
2174                   sta_id, le16_to_cpu(rsp->sta_id))) {
2175             ret = -EIO;
2176             goto free_rsp;
2177         }
2178 
2179         num_flushed_queues = le16_to_cpu(rsp->num_flushed_queues);
2180         if (WARN_ONCE(num_flushed_queues > IWL_TX_FLUSH_QUEUE_RSP,
2181                   "num_flushed_queues %d", num_flushed_queues)) {
2182             ret = -EIO;
2183             goto free_rsp;
2184         }
2185 
2186         for (i = 0; i < num_flushed_queues; i++) {
2187             struct ieee80211_tx_info tx_info = {};
2188             struct iwl_flush_queue_info *queue_info = &rsp->queues[i];
2189             int tid = le16_to_cpu(queue_info->tid);
2190             int read_before = le16_to_cpu(queue_info->read_before_flush);
2191             int read_after = le16_to_cpu(queue_info->read_after_flush);
2192             int queue_num = le16_to_cpu(queue_info->queue_num);
2193 
2194             if (tid == IWL_MGMT_TID)
2195                 tid = IWL_MAX_TID_COUNT;
2196 
2197             IWL_DEBUG_TX_QUEUES(mvm,
2198                         "tid %d queue_id %d read-before %d read-after %d\n",
2199                         tid, queue_num, read_before, read_after);
2200 
2201             iwl_mvm_tx_reclaim(mvm, sta_id, tid, queue_num, read_after,
2202                        &tx_info, 0, true);
2203         }
2204 free_rsp:
2205         iwl_free_resp(&cmd);
2206     }
2207     return ret;
2208 }
2209 
2210 int iwl_mvm_flush_sta(struct iwl_mvm *mvm, void *sta, bool internal)
2211 {
2212     struct iwl_mvm_int_sta *int_sta = sta;
2213     struct iwl_mvm_sta *mvm_sta = sta;
2214 
2215     BUILD_BUG_ON(offsetof(struct iwl_mvm_int_sta, sta_id) !=
2216              offsetof(struct iwl_mvm_sta, sta_id));
2217 
2218     if (iwl_mvm_has_new_tx_api(mvm))
2219         return iwl_mvm_flush_sta_tids(mvm, mvm_sta->sta_id, 0xffff);
2220 
2221     if (internal)
2222         return iwl_mvm_flush_tx_path(mvm, int_sta->tfd_queue_msk);
2223 
2224     return iwl_mvm_flush_tx_path(mvm, mvm_sta->tfd_queue_msk);
2225 }