Merge git://git.kernel.org/pub/scm/linux/kernel/git/davem/net
[cascardo/linux.git] / drivers / net / wireless / intel / iwlwifi / mvm / tx.c
1 /******************************************************************************
2  *
3  * This file is provided under a dual BSD/GPLv2 license.  When using or
4  * redistributing this file, you may do so under either license.
5  *
6  * GPL LICENSE SUMMARY
7  *
8  * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
9  * Copyright(c) 2013 - 2015 Intel Mobile Communications GmbH
10  * Copyright(c) 2016        Intel Deutschland GmbH
11  *
12  * This program is free software; you can redistribute it and/or modify
13  * it under the terms of version 2 of the GNU General Public License as
14  * published by the Free Software Foundation.
15  *
16  * This program is distributed in the hope that it will be useful, but
17  * WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19  * General Public License for more details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program; if not, write to the Free Software
23  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110,
24  * USA
25  *
26  * The full GNU General Public License is included in this distribution
27  * in the file called COPYING.
28  *
29  * Contact Information:
30  *  Intel Linux Wireless <linuxwifi@intel.com>
31  * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
32  *
33  * BSD LICENSE
34  *
35  * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
36  * Copyright(c) 2013 - 2015 Intel Mobile Communications GmbH
37  * All rights reserved.
38  *
39  * Redistribution and use in source and binary forms, with or without
40  * modification, are permitted provided that the following conditions
41  * are met:
42  *
43  *  * Redistributions of source code must retain the above copyright
44  *    notice, this list of conditions and the following disclaimer.
45  *  * Redistributions in binary form must reproduce the above copyright
46  *    notice, this list of conditions and the following disclaimer in
47  *    the documentation and/or other materials provided with the
48  *    distribution.
49  *  * Neither the name Intel Corporation nor the names of its
50  *    contributors may be used to endorse or promote products derived
51  *    from this software without specific prior written permission.
52  *
53  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
54  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
55  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
56  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
57  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
58  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
59  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
60  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
61  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
62  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
63  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
64  *
65  *****************************************************************************/
66 #include <linux/ieee80211.h>
67 #include <linux/etherdevice.h>
68 #include <linux/tcp.h>
69 #include <net/ip.h>
70 #include <net/ipv6.h>
71
72 #include "iwl-trans.h"
73 #include "iwl-eeprom-parse.h"
74 #include "mvm.h"
75 #include "sta.h"
76 #include "fw-dbg.h"
77
78 static void
79 iwl_mvm_bar_check_trigger(struct iwl_mvm *mvm, const u8 *addr,
80                           u16 tid, u16 ssn)
81 {
82         struct iwl_fw_dbg_trigger_tlv *trig;
83         struct iwl_fw_dbg_trigger_ba *ba_trig;
84
85         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_BA))
86                 return;
87
88         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_BA);
89         ba_trig = (void *)trig->data;
90
91         if (!iwl_fw_dbg_trigger_check_stop(mvm, NULL, trig))
92                 return;
93
94         if (!(le16_to_cpu(ba_trig->tx_bar) & BIT(tid)))
95                 return;
96
97         iwl_mvm_fw_dbg_collect_trig(mvm, trig,
98                                     "BAR sent to %pM, tid %d, ssn %d",
99                                     addr, tid, ssn);
100 }
101
102 #define OPT_HDR(type, skb, off) \
103         (type *)(skb_network_header(skb) + (off))
104
105 static void iwl_mvm_tx_csum(struct iwl_mvm *mvm, struct sk_buff *skb,
106                             struct ieee80211_hdr *hdr,
107                             struct ieee80211_tx_info *info,
108                             struct iwl_tx_cmd *tx_cmd)
109 {
110 #if IS_ENABLED(CONFIG_INET)
111         u16 mh_len = ieee80211_hdrlen(hdr->frame_control);
112         u16 offload_assist = le16_to_cpu(tx_cmd->offload_assist);
113         u8 protocol = 0;
114
115         /*
116          * Do not compute checksum if already computed or if transport will
117          * compute it
118          */
119         if (skb->ip_summed != CHECKSUM_PARTIAL || IWL_MVM_SW_TX_CSUM_OFFLOAD)
120                 return;
121
122         /* We do not expect to be requested to csum stuff we do not support */
123         if (WARN_ONCE(!(mvm->hw->netdev_features & IWL_TX_CSUM_NETIF_FLAGS) ||
124                       (skb->protocol != htons(ETH_P_IP) &&
125                        skb->protocol != htons(ETH_P_IPV6)),
126                       "No support for requested checksum\n")) {
127                 skb_checksum_help(skb);
128                 return;
129         }
130
131         if (skb->protocol == htons(ETH_P_IP)) {
132                 protocol = ip_hdr(skb)->protocol;
133         } else {
134 #if IS_ENABLED(CONFIG_IPV6)
135                 struct ipv6hdr *ipv6h =
136                         (struct ipv6hdr *)skb_network_header(skb);
137                 unsigned int off = sizeof(*ipv6h);
138
139                 protocol = ipv6h->nexthdr;
140                 while (protocol != NEXTHDR_NONE && ipv6_ext_hdr(protocol)) {
141                         struct ipv6_opt_hdr *hp;
142
143                         /* only supported extension headers */
144                         if (protocol != NEXTHDR_ROUTING &&
145                             protocol != NEXTHDR_HOP &&
146                             protocol != NEXTHDR_DEST) {
147                                 skb_checksum_help(skb);
148                                 return;
149                         }
150
151                         hp = OPT_HDR(struct ipv6_opt_hdr, skb, off);
152                         protocol = hp->nexthdr;
153                         off += ipv6_optlen(hp);
154                 }
155                 /* if we get here - protocol now should be TCP/UDP */
156 #endif
157         }
158
159         if (protocol != IPPROTO_TCP && protocol != IPPROTO_UDP) {
160                 WARN_ON_ONCE(1);
161                 skb_checksum_help(skb);
162                 return;
163         }
164
165         /* enable L4 csum */
166         offload_assist |= BIT(TX_CMD_OFFLD_L4_EN);
167
168         /*
169          * Set offset to IP header (snap).
170          * We don't support tunneling so no need to take care of inner header.
171          * Size is in words.
172          */
173         offload_assist |= (4 << TX_CMD_OFFLD_IP_HDR);
174
175         /* Do IPv4 csum for AMSDU only (no IP csum for Ipv6) */
176         if (skb->protocol == htons(ETH_P_IP) &&
177             (offload_assist & BIT(TX_CMD_OFFLD_AMSDU))) {
178                 ip_hdr(skb)->check = 0;
179                 offload_assist |= BIT(TX_CMD_OFFLD_L3_EN);
180         }
181
182         /* reset UDP/TCP header csum */
183         if (protocol == IPPROTO_TCP)
184                 tcp_hdr(skb)->check = 0;
185         else
186                 udp_hdr(skb)->check = 0;
187
188         /* mac header len should include IV, size is in words */
189         if (info->control.hw_key)
190                 mh_len += info->control.hw_key->iv_len;
191         mh_len /= 2;
192         offload_assist |= mh_len << TX_CMD_OFFLD_MH_SIZE;
193
194         tx_cmd->offload_assist = cpu_to_le16(offload_assist);
195 #endif
196 }
197
198 /*
199  * Sets most of the Tx cmd's fields
200  */
201 void iwl_mvm_set_tx_cmd(struct iwl_mvm *mvm, struct sk_buff *skb,
202                         struct iwl_tx_cmd *tx_cmd,
203                         struct ieee80211_tx_info *info, u8 sta_id)
204 {
205         struct ieee80211_tx_info *skb_info = IEEE80211_SKB_CB(skb);
206         struct ieee80211_hdr *hdr = (void *)skb->data;
207         __le16 fc = hdr->frame_control;
208         u32 tx_flags = le32_to_cpu(tx_cmd->tx_flags);
209         u32 len = skb->len + FCS_LEN;
210         u8 ac;
211
212         if (!(info->flags & IEEE80211_TX_CTL_NO_ACK))
213                 tx_flags |= TX_CMD_FLG_ACK;
214         else
215                 tx_flags &= ~TX_CMD_FLG_ACK;
216
217         if (ieee80211_is_probe_resp(fc))
218                 tx_flags |= TX_CMD_FLG_TSF;
219
220         if (ieee80211_has_morefrags(fc))
221                 tx_flags |= TX_CMD_FLG_MORE_FRAG;
222
223         if (ieee80211_is_data_qos(fc)) {
224                 u8 *qc = ieee80211_get_qos_ctl(hdr);
225                 tx_cmd->tid_tspec = qc[0] & 0xf;
226                 tx_flags &= ~TX_CMD_FLG_SEQ_CTL;
227                 if (*qc & IEEE80211_QOS_CTL_A_MSDU_PRESENT)
228                         tx_cmd->offload_assist |=
229                                 cpu_to_le16(BIT(TX_CMD_OFFLD_AMSDU));
230         } else if (ieee80211_is_back_req(fc)) {
231                 struct ieee80211_bar *bar = (void *)skb->data;
232                 u16 control = le16_to_cpu(bar->control);
233                 u16 ssn = le16_to_cpu(bar->start_seq_num);
234
235                 tx_flags |= TX_CMD_FLG_ACK | TX_CMD_FLG_BAR;
236                 tx_cmd->tid_tspec = (control &
237                                      IEEE80211_BAR_CTRL_TID_INFO_MASK) >>
238                         IEEE80211_BAR_CTRL_TID_INFO_SHIFT;
239                 WARN_ON_ONCE(tx_cmd->tid_tspec >= IWL_MAX_TID_COUNT);
240                 iwl_mvm_bar_check_trigger(mvm, bar->ra, tx_cmd->tid_tspec,
241                                           ssn);
242         } else {
243                 tx_cmd->tid_tspec = IWL_TID_NON_QOS;
244                 if (info->flags & IEEE80211_TX_CTL_ASSIGN_SEQ)
245                         tx_flags |= TX_CMD_FLG_SEQ_CTL;
246                 else
247                         tx_flags &= ~TX_CMD_FLG_SEQ_CTL;
248         }
249
250         /* Default to 0 (BE) when tid_spec is set to IWL_TID_NON_QOS */
251         if (tx_cmd->tid_tspec < IWL_MAX_TID_COUNT)
252                 ac = tid_to_mac80211_ac[tx_cmd->tid_tspec];
253         else
254                 ac = tid_to_mac80211_ac[0];
255
256         tx_flags |= iwl_mvm_bt_coex_tx_prio(mvm, hdr, info, ac) <<
257                         TX_CMD_FLG_BT_PRIO_POS;
258
259         if (ieee80211_is_mgmt(fc)) {
260                 if (ieee80211_is_assoc_req(fc) || ieee80211_is_reassoc_req(fc))
261                         tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_ASSOC);
262                 else if (ieee80211_is_action(fc))
263                         tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_NONE);
264                 else
265                         tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_MGMT);
266
267                 /* The spec allows Action frames in A-MPDU, we don't support
268                  * it
269                  */
270                 WARN_ON_ONCE(info->flags & IEEE80211_TX_CTL_AMPDU);
271         } else if (info->control.flags & IEEE80211_TX_CTRL_PORT_CTRL_PROTO) {
272                 tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_MGMT);
273         } else {
274                 tx_cmd->pm_frame_timeout = cpu_to_le16(PM_FRAME_NONE);
275         }
276
277         if (ieee80211_is_data(fc) && len > mvm->rts_threshold &&
278             !is_multicast_ether_addr(ieee80211_get_DA(hdr)))
279                 tx_flags |= TX_CMD_FLG_PROT_REQUIRE;
280
281         if (fw_has_capa(&mvm->fw->ucode_capa,
282                         IWL_UCODE_TLV_CAPA_TXPOWER_INSERTION_SUPPORT) &&
283             ieee80211_action_contains_tpc(skb))
284                 tx_flags |= TX_CMD_FLG_WRITE_TX_POWER;
285
286         tx_cmd->tx_flags = cpu_to_le32(tx_flags);
287         /* Total # bytes to be transmitted */
288         tx_cmd->len = cpu_to_le16((u16)skb->len +
289                 (uintptr_t)skb_info->driver_data[0]);
290         tx_cmd->life_time = cpu_to_le32(TX_CMD_LIFE_TIME_INFINITE);
291         tx_cmd->sta_id = sta_id;
292
293         /* padding is inserted later in transport */
294         if (ieee80211_hdrlen(fc) % 4 &&
295             !(tx_cmd->offload_assist & cpu_to_le16(BIT(TX_CMD_OFFLD_AMSDU))))
296                 tx_cmd->offload_assist |= cpu_to_le16(BIT(TX_CMD_OFFLD_PAD));
297
298         iwl_mvm_tx_csum(mvm, skb, hdr, info, tx_cmd);
299 }
300
301 /*
302  * Sets the fields in the Tx cmd that are rate related
303  */
304 void iwl_mvm_set_tx_cmd_rate(struct iwl_mvm *mvm, struct iwl_tx_cmd *tx_cmd,
305                             struct ieee80211_tx_info *info,
306                             struct ieee80211_sta *sta, __le16 fc)
307 {
308         u32 rate_flags;
309         int rate_idx;
310         u8 rate_plcp;
311
312         /* Set retry limit on RTS packets */
313         tx_cmd->rts_retry_limit = IWL_RTS_DFAULT_RETRY_LIMIT;
314
315         /* Set retry limit on DATA packets and Probe Responses*/
316         if (ieee80211_is_probe_resp(fc)) {
317                 tx_cmd->data_retry_limit = IWL_MGMT_DFAULT_RETRY_LIMIT;
318                 tx_cmd->rts_retry_limit =
319                         min(tx_cmd->data_retry_limit, tx_cmd->rts_retry_limit);
320         } else if (ieee80211_is_back_req(fc)) {
321                 tx_cmd->data_retry_limit = IWL_BAR_DFAULT_RETRY_LIMIT;
322         } else {
323                 tx_cmd->data_retry_limit = IWL_DEFAULT_TX_RETRY;
324         }
325
326         /*
327          * for data packets, rate info comes from the table inside the fw. This
328          * table is controlled by LINK_QUALITY commands
329          */
330
331         if (ieee80211_is_data(fc) && sta) {
332                 tx_cmd->initial_rate_index = 0;
333                 tx_cmd->tx_flags |= cpu_to_le32(TX_CMD_FLG_STA_RATE);
334                 return;
335         } else if (ieee80211_is_back_req(fc)) {
336                 tx_cmd->tx_flags |=
337                         cpu_to_le32(TX_CMD_FLG_ACK | TX_CMD_FLG_BAR);
338         }
339
340         /* HT rate doesn't make sense for a non data frame */
341         WARN_ONCE(info->control.rates[0].flags & IEEE80211_TX_RC_MCS,
342                   "Got an HT rate (flags:0x%x/mcs:%d) for a non data frame (fc:0x%x)\n",
343                   info->control.rates[0].flags,
344                   info->control.rates[0].idx,
345                   le16_to_cpu(fc));
346
347         rate_idx = info->control.rates[0].idx;
348         /* if the rate isn't a well known legacy rate, take the lowest one */
349         if (rate_idx < 0 || rate_idx > IWL_RATE_COUNT_LEGACY)
350                 rate_idx = rate_lowest_index(
351                                 &mvm->nvm_data->bands[info->band], sta);
352
353         /* For 5 GHZ band, remap mac80211 rate indices into driver indices */
354         if (info->band == NL80211_BAND_5GHZ)
355                 rate_idx += IWL_FIRST_OFDM_RATE;
356
357         /* For 2.4 GHZ band, check that there is no need to remap */
358         BUILD_BUG_ON(IWL_FIRST_CCK_RATE != 0);
359
360         /* Get PLCP rate for tx_cmd->rate_n_flags */
361         rate_plcp = iwl_mvm_mac80211_idx_to_hwrate(rate_idx);
362
363         mvm->mgmt_last_antenna_idx =
364                 iwl_mvm_next_antenna(mvm, iwl_mvm_get_valid_tx_ant(mvm),
365                                      mvm->mgmt_last_antenna_idx);
366
367         if (info->band == NL80211_BAND_2GHZ &&
368             !iwl_mvm_bt_coex_is_shared_ant_avail(mvm))
369                 rate_flags = mvm->cfg->non_shared_ant << RATE_MCS_ANT_POS;
370         else
371                 rate_flags =
372                         BIT(mvm->mgmt_last_antenna_idx) << RATE_MCS_ANT_POS;
373
374         /* Set CCK flag as needed */
375         if ((rate_idx >= IWL_FIRST_CCK_RATE) && (rate_idx <= IWL_LAST_CCK_RATE))
376                 rate_flags |= RATE_MCS_CCK_MSK;
377
378         /* Set the rate in the TX cmd */
379         tx_cmd->rate_n_flags = cpu_to_le32((u32)rate_plcp | rate_flags);
380 }
381
382 static inline void iwl_mvm_set_tx_cmd_pn(struct ieee80211_tx_info *info,
383                                          u8 *crypto_hdr)
384 {
385         struct ieee80211_key_conf *keyconf = info->control.hw_key;
386         u64 pn;
387
388         pn = atomic64_inc_return(&keyconf->tx_pn);
389         crypto_hdr[0] = pn;
390         crypto_hdr[2] = 0;
391         crypto_hdr[3] = 0x20 | (keyconf->keyidx << 6);
392         crypto_hdr[1] = pn >> 8;
393         crypto_hdr[4] = pn >> 16;
394         crypto_hdr[5] = pn >> 24;
395         crypto_hdr[6] = pn >> 32;
396         crypto_hdr[7] = pn >> 40;
397 }
398
399 /*
400  * Sets the fields in the Tx cmd that are crypto related
401  */
402 static void iwl_mvm_set_tx_cmd_crypto(struct iwl_mvm *mvm,
403                                       struct ieee80211_tx_info *info,
404                                       struct iwl_tx_cmd *tx_cmd,
405                                       struct sk_buff *skb_frag,
406                                       int hdrlen)
407 {
408         struct ieee80211_key_conf *keyconf = info->control.hw_key;
409         u8 *crypto_hdr = skb_frag->data + hdrlen;
410         u64 pn;
411
412         switch (keyconf->cipher) {
413         case WLAN_CIPHER_SUITE_CCMP:
414         case WLAN_CIPHER_SUITE_CCMP_256:
415                 iwl_mvm_set_tx_cmd_ccmp(info, tx_cmd);
416                 iwl_mvm_set_tx_cmd_pn(info, crypto_hdr);
417                 break;
418
419         case WLAN_CIPHER_SUITE_TKIP:
420                 tx_cmd->sec_ctl = TX_CMD_SEC_TKIP;
421                 pn = atomic64_inc_return(&keyconf->tx_pn);
422                 ieee80211_tkip_add_iv(crypto_hdr, keyconf, pn);
423                 ieee80211_get_tkip_p2k(keyconf, skb_frag, tx_cmd->key);
424                 break;
425
426         case WLAN_CIPHER_SUITE_WEP104:
427                 tx_cmd->sec_ctl |= TX_CMD_SEC_KEY128;
428                 /* fall through */
429         case WLAN_CIPHER_SUITE_WEP40:
430                 tx_cmd->sec_ctl |= TX_CMD_SEC_WEP |
431                         ((keyconf->keyidx << TX_CMD_SEC_WEP_KEY_IDX_POS) &
432                           TX_CMD_SEC_WEP_KEY_IDX_MSK);
433
434                 memcpy(&tx_cmd->key[3], keyconf->key, keyconf->keylen);
435                 break;
436         case WLAN_CIPHER_SUITE_GCMP:
437         case WLAN_CIPHER_SUITE_GCMP_256:
438                 /* TODO: Taking the key from the table might introduce a race
439                  * when PTK rekeying is done, having an old packets with a PN
440                  * based on the old key but the message encrypted with a new
441                  * one.
442                  * Need to handle this.
443                  */
444                 tx_cmd->sec_ctl |= TX_CMD_SEC_GCMP | TC_CMD_SEC_KEY_FROM_TABLE;
445                 tx_cmd->key[0] = keyconf->hw_key_idx;
446                 iwl_mvm_set_tx_cmd_pn(info, crypto_hdr);
447                 break;
448         default:
449                 tx_cmd->sec_ctl |= TX_CMD_SEC_EXT;
450         }
451 }
452
453 /*
454  * Allocates and sets the Tx cmd the driver data pointers in the skb
455  */
456 static struct iwl_device_cmd *
457 iwl_mvm_set_tx_params(struct iwl_mvm *mvm, struct sk_buff *skb,
458                       struct ieee80211_tx_info *info, int hdrlen,
459                       struct ieee80211_sta *sta, u8 sta_id)
460 {
461         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
462         struct ieee80211_tx_info *skb_info = IEEE80211_SKB_CB(skb);
463         struct iwl_device_cmd *dev_cmd;
464         struct iwl_tx_cmd *tx_cmd;
465
466         dev_cmd = iwl_trans_alloc_tx_cmd(mvm->trans);
467
468         if (unlikely(!dev_cmd))
469                 return NULL;
470
471         memset(dev_cmd, 0, sizeof(*dev_cmd));
472         dev_cmd->hdr.cmd = TX_CMD;
473         tx_cmd = (struct iwl_tx_cmd *)dev_cmd->payload;
474
475         if (info->control.hw_key)
476                 iwl_mvm_set_tx_cmd_crypto(mvm, info, tx_cmd, skb, hdrlen);
477
478         iwl_mvm_set_tx_cmd(mvm, skb, tx_cmd, info, sta_id);
479
480         iwl_mvm_set_tx_cmd_rate(mvm, tx_cmd, info, sta, hdr->frame_control);
481
482         memset(&skb_info->status, 0, sizeof(skb_info->status));
483         memset(skb_info->driver_data, 0, sizeof(skb_info->driver_data));
484
485         skb_info->driver_data[1] = dev_cmd;
486
487         return dev_cmd;
488 }
489
490 static int iwl_mvm_get_ctrl_vif_queue(struct iwl_mvm *mvm,
491                                       struct ieee80211_tx_info *info, __le16 fc)
492 {
493         if (iwl_mvm_is_dqa_supported(mvm)) {
494                 if (info->control.vif->type == NL80211_IFTYPE_AP &&
495                     ieee80211_is_probe_resp(fc))
496                         return IWL_MVM_DQA_AP_PROBE_RESP_QUEUE;
497                 else if (ieee80211_is_mgmt(fc) &&
498                          info->control.vif->type == NL80211_IFTYPE_P2P_DEVICE)
499                         return IWL_MVM_DQA_P2P_DEVICE_QUEUE;
500         }
501
502         return info->hw_queue;
503 }
504
505 int iwl_mvm_tx_skb_non_sta(struct iwl_mvm *mvm, struct sk_buff *skb)
506 {
507         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
508         struct ieee80211_tx_info *skb_info = IEEE80211_SKB_CB(skb);
509         struct ieee80211_tx_info info;
510         struct iwl_device_cmd *dev_cmd;
511         struct iwl_tx_cmd *tx_cmd;
512         u8 sta_id;
513         int hdrlen = ieee80211_hdrlen(hdr->frame_control);
514         int queue;
515
516         /* IWL_MVM_OFFCHANNEL_QUEUE is used for ROC packets that can be used
517          * in 2 different types of vifs, P2P & STATION. P2P uses the offchannel
518          * queue. STATION (HS2.0) uses the auxiliary context of the FW,
519          * and hence needs to be sent on the aux queue
520          */
521         if (IEEE80211_SKB_CB(skb)->hw_queue == IWL_MVM_OFFCHANNEL_QUEUE &&
522             skb_info->control.vif->type == NL80211_IFTYPE_STATION)
523                 IEEE80211_SKB_CB(skb)->hw_queue = mvm->aux_queue;
524
525         memcpy(&info, skb->cb, sizeof(info));
526
527         if (WARN_ON_ONCE(info.flags & IEEE80211_TX_CTL_AMPDU))
528                 return -1;
529
530         if (WARN_ON_ONCE(info.flags & IEEE80211_TX_CTL_SEND_AFTER_DTIM &&
531                          (!info.control.vif ||
532                           info.hw_queue != info.control.vif->cab_queue)))
533                 return -1;
534
535         /* This holds the amsdu headers length */
536         skb_info->driver_data[0] = (void *)(uintptr_t)0;
537
538         queue = info.hw_queue;
539
540         /*
541          * If the interface on which the frame is sent is the P2P_DEVICE
542          * or an AP/GO interface use the broadcast station associated
543          * with it; otherwise if the interface is a managed interface
544          * use the AP station associated with it for multicast traffic
545          * (this is not possible for unicast packets as a TLDS discovery
546          * response are sent without a station entry); otherwise use the
547          * AUX station.
548          * In DQA mode, if vif is of type STATION and frames are not multicast,
549          * they should be sent from the BSS queue. For example, TDLS setup
550          * frames should be sent on this queue, as they go through the AP.
551          */
552         sta_id = mvm->aux_sta.sta_id;
553         if (info.control.vif) {
554                 struct iwl_mvm_vif *mvmvif =
555                         iwl_mvm_vif_from_mac80211(info.control.vif);
556
557                 if (info.control.vif->type == NL80211_IFTYPE_P2P_DEVICE ||
558                     info.control.vif->type == NL80211_IFTYPE_AP) {
559                         sta_id = mvmvif->bcast_sta.sta_id;
560                         queue = iwl_mvm_get_ctrl_vif_queue(mvm, &info,
561                                                            hdr->frame_control);
562                 } else if (info.control.vif->type == NL80211_IFTYPE_STATION &&
563                            is_multicast_ether_addr(hdr->addr1)) {
564                         u8 ap_sta_id = ACCESS_ONCE(mvmvif->ap_sta_id);
565
566                         if (ap_sta_id != IWL_MVM_STATION_COUNT)
567                                 sta_id = ap_sta_id;
568                 } else if (iwl_mvm_is_dqa_supported(mvm) &&
569                            info.control.vif->type == NL80211_IFTYPE_STATION) {
570                         queue = IWL_MVM_DQA_BSS_CLIENT_QUEUE;
571                 }
572         }
573
574         IWL_DEBUG_TX(mvm, "station Id %d, queue=%d\n", sta_id, queue);
575
576         dev_cmd = iwl_mvm_set_tx_params(mvm, skb, &info, hdrlen, NULL, sta_id);
577         if (!dev_cmd)
578                 return -1;
579
580         tx_cmd = (struct iwl_tx_cmd *)dev_cmd->payload;
581
582         /* Copy MAC header from skb into command buffer */
583         memcpy(tx_cmd->hdr, hdr, hdrlen);
584
585         if (iwl_trans_tx(mvm->trans, skb, dev_cmd, queue)) {
586                 iwl_trans_free_tx_cmd(mvm->trans, dev_cmd);
587                 return -1;
588         }
589
590         /*
591          * Increase the pending frames counter, so that later when a reply comes
592          * in and the counter is decreased - we don't start getting negative
593          * values.
594          * Note that we don't need to make sure it isn't agg'd, since we're
595          * TXing non-sta
596          */
597         atomic_inc(&mvm->pending_frames[sta_id]);
598
599         return 0;
600 }
601
602 #ifdef CONFIG_INET
603 static int iwl_mvm_tx_tso(struct iwl_mvm *mvm, struct sk_buff *skb,
604                           struct ieee80211_tx_info *info,
605                           struct ieee80211_sta *sta,
606                           struct sk_buff_head *mpdus_skb)
607 {
608         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
609         struct ieee80211_hdr *hdr = (void *)skb->data;
610         unsigned int mss = skb_shinfo(skb)->gso_size;
611         struct sk_buff *tmp, *next;
612         char cb[sizeof(skb->cb)];
613         unsigned int num_subframes, tcp_payload_len, subf_len, max_amsdu_len;
614         bool ipv4 = (skb->protocol == htons(ETH_P_IP));
615         u16 ip_base_id = ipv4 ? ntohs(ip_hdr(skb)->id) : 0;
616         u16 amsdu_add, snap_ip_tcp, pad, i = 0;
617         unsigned int dbg_max_amsdu_len;
618         netdev_features_t netdev_features = NETIF_F_CSUM_MASK | NETIF_F_SG;
619         u8 *qc, tid, txf;
620
621         snap_ip_tcp = 8 + skb_transport_header(skb) - skb_network_header(skb) +
622                 tcp_hdrlen(skb);
623
624         qc = ieee80211_get_qos_ctl(hdr);
625         tid = *qc & IEEE80211_QOS_CTL_TID_MASK;
626         if (WARN_ON_ONCE(tid >= IWL_MAX_TID_COUNT))
627                 return -EINVAL;
628
629         dbg_max_amsdu_len = ACCESS_ONCE(mvm->max_amsdu_len);
630
631         if (!sta->max_amsdu_len ||
632             !ieee80211_is_data_qos(hdr->frame_control) ||
633             (!mvmsta->tlc_amsdu && !dbg_max_amsdu_len)) {
634                 num_subframes = 1;
635                 pad = 0;
636                 goto segment;
637         }
638
639         /*
640          * Do not build AMSDU for IPv6 with extension headers.
641          * ask stack to segment and checkum the generated MPDUs for us.
642          */
643         if (skb->protocol == htons(ETH_P_IPV6) &&
644             ((struct ipv6hdr *)skb_network_header(skb))->nexthdr !=
645             IPPROTO_TCP) {
646                 num_subframes = 1;
647                 pad = 0;
648                 netdev_features &= ~NETIF_F_CSUM_MASK;
649                 goto segment;
650         }
651
652         /*
653          * No need to lock amsdu_in_ampdu_allowed since it can't be modified
654          * during an BA session.
655          */
656         if (info->flags & IEEE80211_TX_CTL_AMPDU &&
657             !mvmsta->tid_data[tid].amsdu_in_ampdu_allowed) {
658                 num_subframes = 1;
659                 pad = 0;
660                 goto segment;
661         }
662
663         max_amsdu_len = sta->max_amsdu_len;
664
665         /* the Tx FIFO to which this A-MSDU will be routed */
666         txf = iwl_mvm_ac_to_tx_fifo[tid_to_mac80211_ac[tid]];
667
668         /*
669          * Don't send an AMSDU that will be longer than the TXF.
670          * Add a security margin of 256 for the TX command + headers.
671          * We also want to have the start of the next packet inside the
672          * fifo to be able to send bursts.
673          */
674         max_amsdu_len = min_t(unsigned int, max_amsdu_len,
675                               mvm->shared_mem_cfg.txfifo_size[txf] - 256);
676
677         if (unlikely(dbg_max_amsdu_len))
678                 max_amsdu_len = min_t(unsigned int, max_amsdu_len,
679                                       dbg_max_amsdu_len);
680
681         /*
682          * Limit A-MSDU in A-MPDU to 4095 bytes when VHT is not
683          * supported. This is a spec requirement (IEEE 802.11-2015
684          * section 8.7.3 NOTE 3).
685          */
686         if (info->flags & IEEE80211_TX_CTL_AMPDU &&
687             !sta->vht_cap.vht_supported)
688                 max_amsdu_len = min_t(unsigned int, max_amsdu_len, 4095);
689
690         /* Sub frame header + SNAP + IP header + TCP header + MSS */
691         subf_len = sizeof(struct ethhdr) + snap_ip_tcp + mss;
692         pad = (4 - subf_len) & 0x3;
693
694         /*
695          * If we have N subframes in the A-MSDU, then the A-MSDU's size is
696          * N * subf_len + (N - 1) * pad.
697          */
698         num_subframes = (max_amsdu_len + pad) / (subf_len + pad);
699         if (num_subframes > 1)
700                 *qc |= IEEE80211_QOS_CTL_A_MSDU_PRESENT;
701
702         tcp_payload_len = skb_tail_pointer(skb) - skb_transport_header(skb) -
703                 tcp_hdrlen(skb) + skb->data_len;
704
705         /*
706          * Make sure we have enough TBs for the A-MSDU:
707          *      2 for each subframe
708          *      1 more for each fragment
709          *      1 more for the potential data in the header
710          */
711         num_subframes =
712                 min_t(unsigned int, num_subframes,
713                       (mvm->trans->max_skb_frags - 1 -
714                        skb_shinfo(skb)->nr_frags) / 2);
715
716         /* This skb fits in one single A-MSDU */
717         if (num_subframes * mss >= tcp_payload_len) {
718                 struct ieee80211_tx_info *skb_info = IEEE80211_SKB_CB(skb);
719
720                 /*
721                  * Compute the length of all the data added for the A-MSDU.
722                  * This will be used to compute the length to write in the TX
723                  * command. We have: SNAP + IP + TCP for n -1 subframes and
724                  * ETH header for n subframes. Note that the original skb
725                  * already had one set of SNAP / IP / TCP headers.
726                  */
727                 num_subframes = DIV_ROUND_UP(tcp_payload_len, mss);
728                 amsdu_add = num_subframes * sizeof(struct ethhdr) +
729                         (num_subframes - 1) * (snap_ip_tcp + pad);
730                 /* This holds the amsdu headers length */
731                 skb_info->driver_data[0] = (void *)(uintptr_t)amsdu_add;
732
733                 __skb_queue_tail(mpdus_skb, skb);
734                 return 0;
735         }
736
737         /*
738          * Trick the segmentation function to make it
739          * create SKBs that can fit into one A-MSDU.
740          */
741 segment:
742         skb_shinfo(skb)->gso_size = num_subframes * mss;
743         memcpy(cb, skb->cb, sizeof(cb));
744
745         next = skb_gso_segment(skb, netdev_features);
746         skb_shinfo(skb)->gso_size = mss;
747         if (WARN_ON_ONCE(IS_ERR(next)))
748                 return -EINVAL;
749         else if (next)
750                 consume_skb(skb);
751
752         while (next) {
753                 tmp = next;
754                 next = tmp->next;
755
756                 memcpy(tmp->cb, cb, sizeof(tmp->cb));
757                 /*
758                  * Compute the length of all the data added for the A-MSDU.
759                  * This will be used to compute the length to write in the TX
760                  * command. We have: SNAP + IP + TCP for n -1 subframes and
761                  * ETH header for n subframes.
762                  */
763                 tcp_payload_len = skb_tail_pointer(tmp) -
764                         skb_transport_header(tmp) -
765                         tcp_hdrlen(tmp) + tmp->data_len;
766
767                 if (ipv4)
768                         ip_hdr(tmp)->id = htons(ip_base_id + i * num_subframes);
769
770                 if (tcp_payload_len > mss) {
771                         struct ieee80211_tx_info *skb_info =
772                                 IEEE80211_SKB_CB(tmp);
773
774                         num_subframes = DIV_ROUND_UP(tcp_payload_len, mss);
775                         amsdu_add = num_subframes * sizeof(struct ethhdr) +
776                                 (num_subframes - 1) * (snap_ip_tcp + pad);
777                         skb_info->driver_data[0] =
778                                 (void *)(uintptr_t)amsdu_add;
779                         skb_shinfo(tmp)->gso_size = mss;
780                 } else {
781                         qc = ieee80211_get_qos_ctl((void *)tmp->data);
782
783                         if (ipv4)
784                                 ip_send_check(ip_hdr(tmp));
785                         *qc &= ~IEEE80211_QOS_CTL_A_MSDU_PRESENT;
786                         skb_shinfo(tmp)->gso_size = 0;
787                 }
788
789                 tmp->prev = NULL;
790                 tmp->next = NULL;
791
792                 __skb_queue_tail(mpdus_skb, tmp);
793                 i++;
794         }
795
796         return 0;
797 }
798 #else /* CONFIG_INET */
799 static int iwl_mvm_tx_tso(struct iwl_mvm *mvm, struct sk_buff *skb,
800                           struct ieee80211_tx_info *info,
801                           struct ieee80211_sta *sta,
802                           struct sk_buff_head *mpdus_skb)
803 {
804         /* Impossible to get TSO with CONFIG_INET */
805         WARN_ON(1);
806
807         return -1;
808 }
809 #endif
810
811 static void iwl_mvm_tx_add_stream(struct iwl_mvm *mvm,
812                                   struct iwl_mvm_sta *mvm_sta, u8 tid,
813                                   struct sk_buff *skb)
814 {
815         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
816         u8 mac_queue = info->hw_queue;
817         struct sk_buff_head *deferred_tx_frames;
818
819         lockdep_assert_held(&mvm_sta->lock);
820
821         mvm_sta->deferred_traffic_tid_map |= BIT(tid);
822         set_bit(mvm_sta->sta_id, mvm->sta_deferred_frames);
823
824         deferred_tx_frames = &mvm_sta->tid_data[tid].deferred_tx_frames;
825
826         skb_queue_tail(deferred_tx_frames, skb);
827
828         /*
829          * The first deferred frame should've stopped the MAC queues, so we
830          * should never get a second deferred frame for the RA/TID.
831          */
832         if (!WARN(skb_queue_len(deferred_tx_frames) != 1,
833                   "RATID %d/%d has %d deferred frames\n", mvm_sta->sta_id, tid,
834                   skb_queue_len(deferred_tx_frames))) {
835                 iwl_mvm_stop_mac_queues(mvm, BIT(mac_queue));
836                 schedule_work(&mvm->add_stream_wk);
837         }
838 }
839
840 /* Check if there are any timed-out TIDs on a given shared TXQ */
841 static bool iwl_mvm_txq_should_update(struct iwl_mvm *mvm, int txq_id)
842 {
843         unsigned long queue_tid_bitmap = mvm->queue_info[txq_id].tid_bitmap;
844         unsigned long now = jiffies;
845         int tid;
846
847         for_each_set_bit(tid, &queue_tid_bitmap, IWL_MAX_TID_COUNT + 1) {
848                 if (time_before(mvm->queue_info[txq_id].last_frame_time[tid] +
849                                 IWL_MVM_DQA_QUEUE_TIMEOUT, now))
850                         return true;
851         }
852
853         return false;
854 }
855
856 /*
857  * Sets the fields in the Tx cmd that are crypto related
858  */
859 static int iwl_mvm_tx_mpdu(struct iwl_mvm *mvm, struct sk_buff *skb,
860                            struct ieee80211_tx_info *info,
861                            struct ieee80211_sta *sta)
862 {
863         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
864         struct iwl_mvm_sta *mvmsta;
865         struct iwl_device_cmd *dev_cmd;
866         struct iwl_tx_cmd *tx_cmd;
867         __le16 fc;
868         u16 seq_number = 0;
869         u8 tid = IWL_MAX_TID_COUNT;
870         u8 txq_id = info->hw_queue;
871         bool is_ampdu = false;
872         int hdrlen;
873
874         mvmsta = iwl_mvm_sta_from_mac80211(sta);
875         fc = hdr->frame_control;
876         hdrlen = ieee80211_hdrlen(fc);
877
878         if (WARN_ON_ONCE(!mvmsta))
879                 return -1;
880
881         if (WARN_ON_ONCE(mvmsta->sta_id == IWL_MVM_STATION_COUNT))
882                 return -1;
883
884         dev_cmd = iwl_mvm_set_tx_params(mvm, skb, info, hdrlen,
885                                         sta, mvmsta->sta_id);
886         if (!dev_cmd)
887                 goto drop;
888
889         tx_cmd = (struct iwl_tx_cmd *)dev_cmd->payload;
890         /* From now on, we cannot access info->control */
891
892         /*
893          * we handle that entirely ourselves -- for uAPSD the firmware
894          * will always send a notification, and for PS-Poll responses
895          * we'll notify mac80211 when getting frame status
896          */
897         info->flags &= ~IEEE80211_TX_STATUS_EOSP;
898
899         spin_lock(&mvmsta->lock);
900
901         if (ieee80211_is_data_qos(fc) && !ieee80211_is_qos_nullfunc(fc)) {
902                 u8 *qc = NULL;
903                 qc = ieee80211_get_qos_ctl(hdr);
904                 tid = qc[0] & IEEE80211_QOS_CTL_TID_MASK;
905                 if (WARN_ON_ONCE(tid >= IWL_MAX_TID_COUNT))
906                         goto drop_unlock_sta;
907
908                 seq_number = mvmsta->tid_data[tid].seq_number;
909                 seq_number &= IEEE80211_SCTL_SEQ;
910                 hdr->seq_ctrl &= cpu_to_le16(IEEE80211_SCTL_FRAG);
911                 hdr->seq_ctrl |= cpu_to_le16(seq_number);
912                 is_ampdu = info->flags & IEEE80211_TX_CTL_AMPDU;
913         } else if (iwl_mvm_is_dqa_supported(mvm) &&
914                    (ieee80211_is_qos_nullfunc(fc) ||
915                     ieee80211_is_nullfunc(fc))) {
916                 /*
917                  * nullfunc frames should go to the MGMT queue regardless of QOS
918                  */
919                 tid = IWL_MAX_TID_COUNT;
920         }
921
922         if (iwl_mvm_is_dqa_supported(mvm))
923                 txq_id = mvmsta->tid_data[tid].txq_id;
924
925         /* Copy MAC header from skb into command buffer */
926         memcpy(tx_cmd->hdr, hdr, hdrlen);
927
928         WARN_ON_ONCE(info->flags & IEEE80211_TX_CTL_SEND_AFTER_DTIM);
929
930         if (sta->tdls && !iwl_mvm_is_dqa_supported(mvm)) {
931                 /* default to TID 0 for non-QoS packets */
932                 u8 tdls_tid = tid == IWL_MAX_TID_COUNT ? 0 : tid;
933
934                 txq_id = mvmsta->hw_queue[tid_to_mac80211_ac[tdls_tid]];
935         }
936
937         if (is_ampdu) {
938                 if (WARN_ON_ONCE(mvmsta->tid_data[tid].state != IWL_AGG_ON))
939                         goto drop_unlock_sta;
940                 txq_id = mvmsta->tid_data[tid].txq_id;
941         }
942
943         /* Check if TXQ needs to be allocated or re-activated */
944         if (unlikely(txq_id == IEEE80211_INVAL_HW_QUEUE ||
945                      !mvmsta->tid_data[tid].is_tid_active) &&
946             iwl_mvm_is_dqa_supported(mvm)) {
947                 /* If TXQ needs to be allocated... */
948                 if (txq_id == IEEE80211_INVAL_HW_QUEUE) {
949                         iwl_mvm_tx_add_stream(mvm, mvmsta, tid, skb);
950
951                         /*
952                          * The frame is now deferred, and the worker scheduled
953                          * will re-allocate it, so we can free it for now.
954                          */
955                         iwl_trans_free_tx_cmd(mvm->trans, dev_cmd);
956                         spin_unlock(&mvmsta->lock);
957                         return 0;
958                 }
959
960                 /* If we are here - TXQ exists and needs to be re-activated */
961                 spin_lock(&mvm->queue_info_lock);
962                 mvm->queue_info[txq_id].status = IWL_MVM_QUEUE_READY;
963                 mvmsta->tid_data[tid].is_tid_active = true;
964                 spin_unlock(&mvm->queue_info_lock);
965
966                 IWL_DEBUG_TX_QUEUES(mvm, "Re-activating queue %d for TX\n",
967                                     txq_id);
968         }
969
970         if (iwl_mvm_is_dqa_supported(mvm)) {
971                 /* Keep track of the time of the last frame for this RA/TID */
972                 mvm->queue_info[txq_id].last_frame_time[tid] = jiffies;
973
974                 /*
975                  * If we have timed-out TIDs - schedule the worker that will
976                  * reconfig the queues and update them
977                  *
978                  * Note that the mvm->queue_info_lock isn't being taken here in
979                  * order to not serialize the TX flow. This isn't dangerous
980                  * because scheduling mvm->add_stream_wk can't ruin the state,
981                  * and if we DON'T schedule it due to some race condition then
982                  * next TX we get here we will.
983                  */
984                 if (unlikely(mvm->queue_info[txq_id].status ==
985                              IWL_MVM_QUEUE_SHARED &&
986                              iwl_mvm_txq_should_update(mvm, txq_id)))
987                         schedule_work(&mvm->add_stream_wk);
988         }
989
990         IWL_DEBUG_TX(mvm, "TX to [%d|%d] Q:%d - seq: 0x%x\n", mvmsta->sta_id,
991                      tid, txq_id, IEEE80211_SEQ_TO_SN(seq_number));
992
993         if (iwl_trans_tx(mvm->trans, skb, dev_cmd, txq_id))
994                 goto drop_unlock_sta;
995
996         if (tid < IWL_MAX_TID_COUNT && !ieee80211_has_morefrags(fc))
997                 mvmsta->tid_data[tid].seq_number = seq_number + 0x10;
998
999         spin_unlock(&mvmsta->lock);
1000
1001         /* Increase pending frames count if this isn't AMPDU */
1002         if (!is_ampdu)
1003                 atomic_inc(&mvm->pending_frames[mvmsta->sta_id]);
1004
1005         return 0;
1006
1007 drop_unlock_sta:
1008         iwl_trans_free_tx_cmd(mvm->trans, dev_cmd);
1009         spin_unlock(&mvmsta->lock);
1010 drop:
1011         return -1;
1012 }
1013
1014 int iwl_mvm_tx_skb(struct iwl_mvm *mvm, struct sk_buff *skb,
1015                    struct ieee80211_sta *sta)
1016 {
1017         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1018         struct ieee80211_tx_info *skb_info = IEEE80211_SKB_CB(skb);
1019         struct ieee80211_tx_info info;
1020         struct sk_buff_head mpdus_skbs;
1021         unsigned int payload_len;
1022         int ret;
1023
1024         if (WARN_ON_ONCE(!mvmsta))
1025                 return -1;
1026
1027         if (WARN_ON_ONCE(mvmsta->sta_id == IWL_MVM_STATION_COUNT))
1028                 return -1;
1029
1030         memcpy(&info, skb->cb, sizeof(info));
1031
1032         /* This holds the amsdu headers length */
1033         skb_info->driver_data[0] = (void *)(uintptr_t)0;
1034
1035         if (!skb_is_gso(skb))
1036                 return iwl_mvm_tx_mpdu(mvm, skb, &info, sta);
1037
1038         payload_len = skb_tail_pointer(skb) - skb_transport_header(skb) -
1039                 tcp_hdrlen(skb) + skb->data_len;
1040
1041         if (payload_len <= skb_shinfo(skb)->gso_size)
1042                 return iwl_mvm_tx_mpdu(mvm, skb, &info, sta);
1043
1044         __skb_queue_head_init(&mpdus_skbs);
1045
1046         ret = iwl_mvm_tx_tso(mvm, skb, &info, sta, &mpdus_skbs);
1047         if (ret)
1048                 return ret;
1049
1050         if (WARN_ON(skb_queue_empty(&mpdus_skbs)))
1051                 return ret;
1052
1053         while (!skb_queue_empty(&mpdus_skbs)) {
1054                 skb = __skb_dequeue(&mpdus_skbs);
1055
1056                 ret = iwl_mvm_tx_mpdu(mvm, skb, &info, sta);
1057                 if (ret) {
1058                         __skb_queue_purge(&mpdus_skbs);
1059                         return ret;
1060                 }
1061         }
1062
1063         return 0;
1064 }
1065
1066 static void iwl_mvm_check_ratid_empty(struct iwl_mvm *mvm,
1067                                       struct ieee80211_sta *sta, u8 tid)
1068 {
1069         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1070         struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
1071         struct ieee80211_vif *vif = mvmsta->vif;
1072
1073         lockdep_assert_held(&mvmsta->lock);
1074
1075         if ((tid_data->state == IWL_AGG_ON ||
1076              tid_data->state == IWL_EMPTYING_HW_QUEUE_DELBA) &&
1077             iwl_mvm_tid_queued(tid_data) == 0) {
1078                 /*
1079                  * Now that this aggregation queue is empty tell mac80211 so it
1080                  * knows we no longer have frames buffered for the station on
1081                  * this TID (for the TIM bitmap calculation.)
1082                  */
1083                 ieee80211_sta_set_buffered(sta, tid, false);
1084         }
1085
1086         if (tid_data->ssn != tid_data->next_reclaimed)
1087                 return;
1088
1089         switch (tid_data->state) {
1090         case IWL_EMPTYING_HW_QUEUE_ADDBA:
1091                 IWL_DEBUG_TX_QUEUES(mvm,
1092                                     "Can continue addBA flow ssn = next_recl = %d\n",
1093                                     tid_data->next_reclaimed);
1094                 tid_data->state = IWL_AGG_STARTING;
1095                 ieee80211_start_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1096                 break;
1097
1098         case IWL_EMPTYING_HW_QUEUE_DELBA:
1099                 IWL_DEBUG_TX_QUEUES(mvm,
1100                                     "Can continue DELBA flow ssn = next_recl = %d\n",
1101                                     tid_data->next_reclaimed);
1102                 iwl_mvm_disable_txq(mvm, tid_data->txq_id,
1103                                     vif->hw_queue[tid_to_mac80211_ac[tid]], tid,
1104                                     CMD_ASYNC);
1105                 tid_data->state = IWL_AGG_OFF;
1106                 ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1107                 break;
1108
1109         default:
1110                 break;
1111         }
1112 }
1113
1114 #ifdef CONFIG_IWLWIFI_DEBUG
1115 const char *iwl_mvm_get_tx_fail_reason(u32 status)
1116 {
1117 #define TX_STATUS_FAIL(x) case TX_STATUS_FAIL_ ## x: return #x
1118 #define TX_STATUS_POSTPONE(x) case TX_STATUS_POSTPONE_ ## x: return #x
1119
1120         switch (status & TX_STATUS_MSK) {
1121         case TX_STATUS_SUCCESS:
1122                 return "SUCCESS";
1123         TX_STATUS_POSTPONE(DELAY);
1124         TX_STATUS_POSTPONE(FEW_BYTES);
1125         TX_STATUS_POSTPONE(BT_PRIO);
1126         TX_STATUS_POSTPONE(QUIET_PERIOD);
1127         TX_STATUS_POSTPONE(CALC_TTAK);
1128         TX_STATUS_FAIL(INTERNAL_CROSSED_RETRY);
1129         TX_STATUS_FAIL(SHORT_LIMIT);
1130         TX_STATUS_FAIL(LONG_LIMIT);
1131         TX_STATUS_FAIL(UNDERRUN);
1132         TX_STATUS_FAIL(DRAIN_FLOW);
1133         TX_STATUS_FAIL(RFKILL_FLUSH);
1134         TX_STATUS_FAIL(LIFE_EXPIRE);
1135         TX_STATUS_FAIL(DEST_PS);
1136         TX_STATUS_FAIL(HOST_ABORTED);
1137         TX_STATUS_FAIL(BT_RETRY);
1138         TX_STATUS_FAIL(STA_INVALID);
1139         TX_STATUS_FAIL(FRAG_DROPPED);
1140         TX_STATUS_FAIL(TID_DISABLE);
1141         TX_STATUS_FAIL(FIFO_FLUSHED);
1142         TX_STATUS_FAIL(SMALL_CF_POLL);
1143         TX_STATUS_FAIL(FW_DROP);
1144         TX_STATUS_FAIL(STA_COLOR_MISMATCH);
1145         }
1146
1147         return "UNKNOWN";
1148
1149 #undef TX_STATUS_FAIL
1150 #undef TX_STATUS_POSTPONE
1151 }
1152 #endif /* CONFIG_IWLWIFI_DEBUG */
1153
1154 void iwl_mvm_hwrate_to_tx_rate(u32 rate_n_flags,
1155                                enum nl80211_band band,
1156                                struct ieee80211_tx_rate *r)
1157 {
1158         if (rate_n_flags & RATE_HT_MCS_GF_MSK)
1159                 r->flags |= IEEE80211_TX_RC_GREEN_FIELD;
1160         switch (rate_n_flags & RATE_MCS_CHAN_WIDTH_MSK) {
1161         case RATE_MCS_CHAN_WIDTH_20:
1162                 break;
1163         case RATE_MCS_CHAN_WIDTH_40:
1164                 r->flags |= IEEE80211_TX_RC_40_MHZ_WIDTH;
1165                 break;
1166         case RATE_MCS_CHAN_WIDTH_80:
1167                 r->flags |= IEEE80211_TX_RC_80_MHZ_WIDTH;
1168                 break;
1169         case RATE_MCS_CHAN_WIDTH_160:
1170                 r->flags |= IEEE80211_TX_RC_160_MHZ_WIDTH;
1171                 break;
1172         }
1173         if (rate_n_flags & RATE_MCS_SGI_MSK)
1174                 r->flags |= IEEE80211_TX_RC_SHORT_GI;
1175         if (rate_n_flags & RATE_MCS_HT_MSK) {
1176                 r->flags |= IEEE80211_TX_RC_MCS;
1177                 r->idx = rate_n_flags & RATE_HT_MCS_INDEX_MSK;
1178         } else if (rate_n_flags & RATE_MCS_VHT_MSK) {
1179                 ieee80211_rate_set_vht(
1180                         r, rate_n_flags & RATE_VHT_MCS_RATE_CODE_MSK,
1181                         ((rate_n_flags & RATE_VHT_MCS_NSS_MSK) >>
1182                                                 RATE_VHT_MCS_NSS_POS) + 1);
1183                 r->flags |= IEEE80211_TX_RC_VHT_MCS;
1184         } else {
1185                 r->idx = iwl_mvm_legacy_rate_to_mac80211_idx(rate_n_flags,
1186                                                              band);
1187         }
1188 }
1189
1190 /**
1191  * translate ucode response to mac80211 tx status control values
1192  */
1193 static void iwl_mvm_hwrate_to_tx_status(u32 rate_n_flags,
1194                                         struct ieee80211_tx_info *info)
1195 {
1196         struct ieee80211_tx_rate *r = &info->status.rates[0];
1197
1198         info->status.antenna =
1199                 ((rate_n_flags & RATE_MCS_ANT_ABC_MSK) >> RATE_MCS_ANT_POS);
1200         iwl_mvm_hwrate_to_tx_rate(rate_n_flags, info->band, r);
1201 }
1202
1203 static void iwl_mvm_tx_status_check_trigger(struct iwl_mvm *mvm,
1204                                             u32 status)
1205 {
1206         struct iwl_fw_dbg_trigger_tlv *trig;
1207         struct iwl_fw_dbg_trigger_tx_status *status_trig;
1208         int i;
1209
1210         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_TX_STATUS))
1211                 return;
1212
1213         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_TX_STATUS);
1214         status_trig = (void *)trig->data;
1215
1216         if (!iwl_fw_dbg_trigger_check_stop(mvm, NULL, trig))
1217                 return;
1218
1219         for (i = 0; i < ARRAY_SIZE(status_trig->statuses); i++) {
1220                 /* don't collect on status 0 */
1221                 if (!status_trig->statuses[i].status)
1222                         break;
1223
1224                 if (status_trig->statuses[i].status != (status & TX_STATUS_MSK))
1225                         continue;
1226
1227                 iwl_mvm_fw_dbg_collect_trig(mvm, trig,
1228                                             "Tx status %d was received",
1229                                             status & TX_STATUS_MSK);
1230                 break;
1231         }
1232 }
1233
1234 static void iwl_mvm_rx_tx_cmd_single(struct iwl_mvm *mvm,
1235                                      struct iwl_rx_packet *pkt)
1236 {
1237         struct ieee80211_sta *sta;
1238         u16 sequence = le16_to_cpu(pkt->hdr.sequence);
1239         int txq_id = SEQ_TO_QUEUE(sequence);
1240         struct iwl_mvm_tx_resp *tx_resp = (void *)pkt->data;
1241         int sta_id = IWL_MVM_TX_RES_GET_RA(tx_resp->ra_tid);
1242         int tid = IWL_MVM_TX_RES_GET_TID(tx_resp->ra_tid);
1243         u32 status = le16_to_cpu(tx_resp->status.status);
1244         u16 ssn = iwl_mvm_get_scd_ssn(tx_resp);
1245         struct iwl_mvm_sta *mvmsta;
1246         struct sk_buff_head skbs;
1247         u8 skb_freed = 0;
1248         u16 next_reclaimed, seq_ctl;
1249         bool is_ndp = false;
1250         bool txq_agg = false; /* Is this TXQ aggregated */
1251
1252         __skb_queue_head_init(&skbs);
1253
1254         seq_ctl = le16_to_cpu(tx_resp->seq_ctl);
1255
1256         /* we can free until ssn % q.n_bd not inclusive */
1257         iwl_trans_reclaim(mvm->trans, txq_id, ssn, &skbs);
1258
1259         while (!skb_queue_empty(&skbs)) {
1260                 struct sk_buff *skb = __skb_dequeue(&skbs);
1261                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1262
1263                 skb_freed++;
1264
1265                 iwl_trans_free_tx_cmd(mvm->trans, info->driver_data[1]);
1266
1267                 memset(&info->status, 0, sizeof(info->status));
1268
1269                 info->flags &= ~IEEE80211_TX_CTL_AMPDU;
1270
1271                 /* inform mac80211 about what happened with the frame */
1272                 switch (status & TX_STATUS_MSK) {
1273                 case TX_STATUS_SUCCESS:
1274                 case TX_STATUS_DIRECT_DONE:
1275                         info->flags |= IEEE80211_TX_STAT_ACK;
1276                         break;
1277                 case TX_STATUS_FAIL_DEST_PS:
1278                         info->flags |= IEEE80211_TX_STAT_TX_FILTERED;
1279                         break;
1280                 default:
1281                         break;
1282                 }
1283
1284                 iwl_mvm_tx_status_check_trigger(mvm, status);
1285
1286                 info->status.rates[0].count = tx_resp->failure_frame + 1;
1287                 iwl_mvm_hwrate_to_tx_status(le32_to_cpu(tx_resp->initial_rate),
1288                                             info);
1289                 info->status.status_driver_data[1] =
1290                         (void *)(uintptr_t)le32_to_cpu(tx_resp->initial_rate);
1291
1292                 /* Single frame failure in an AMPDU queue => send BAR */
1293                 if (txq_id >= mvm->first_agg_queue &&
1294                     !(info->flags & IEEE80211_TX_STAT_ACK) &&
1295                     !(info->flags & IEEE80211_TX_STAT_TX_FILTERED))
1296                         info->flags |= IEEE80211_TX_STAT_AMPDU_NO_BACK;
1297
1298                 /* W/A FW bug: seq_ctl is wrong when the status isn't success */
1299                 if (status != TX_STATUS_SUCCESS) {
1300                         struct ieee80211_hdr *hdr = (void *)skb->data;
1301                         seq_ctl = le16_to_cpu(hdr->seq_ctrl);
1302                 }
1303
1304                 if (unlikely(!seq_ctl)) {
1305                         struct ieee80211_hdr *hdr = (void *)skb->data;
1306
1307                         /*
1308                          * If it is an NDP, we can't update next_reclaim since
1309                          * its sequence control is 0. Note that for that same
1310                          * reason, NDPs are never sent to A-MPDU'able queues
1311                          * so that we can never have more than one freed frame
1312                          * for a single Tx resonse (see WARN_ON below).
1313                          */
1314                         if (ieee80211_is_qos_nullfunc(hdr->frame_control))
1315                                 is_ndp = true;
1316                 }
1317
1318                 /*
1319                  * TODO: this is not accurate if we are freeing more than one
1320                  * packet.
1321                  */
1322                 info->status.tx_time =
1323                         le16_to_cpu(tx_resp->wireless_media_time);
1324                 BUILD_BUG_ON(ARRAY_SIZE(info->status.status_driver_data) < 1);
1325                 info->status.status_driver_data[0] =
1326                                 (void *)(uintptr_t)tx_resp->reduced_tpc;
1327
1328                 ieee80211_tx_status(mvm->hw, skb);
1329         }
1330
1331         if (txq_id >= mvm->first_agg_queue) {
1332                 /* If this is an aggregation queue, we use the ssn since:
1333                  * ssn = wifi seq_num % 256.
1334                  * The seq_ctl is the sequence control of the packet to which
1335                  * this Tx response relates. But if there is a hole in the
1336                  * bitmap of the BA we received, this Tx response may allow to
1337                  * reclaim the hole and all the subsequent packets that were
1338                  * already acked. In that case, seq_ctl != ssn, and the next
1339                  * packet to be reclaimed will be ssn and not seq_ctl. In that
1340                  * case, several packets will be reclaimed even if
1341                  * frame_count = 1.
1342                  *
1343                  * The ssn is the index (% 256) of the latest packet that has
1344                  * treated (acked / dropped) + 1.
1345                  */
1346                 next_reclaimed = ssn;
1347         } else {
1348                 /* The next packet to be reclaimed is the one after this one */
1349                 next_reclaimed = IEEE80211_SEQ_TO_SN(seq_ctl + 0x10);
1350         }
1351
1352         IWL_DEBUG_TX_REPLY(mvm,
1353                            "TXQ %d status %s (0x%08x)\n",
1354                            txq_id, iwl_mvm_get_tx_fail_reason(status), status);
1355
1356         IWL_DEBUG_TX_REPLY(mvm,
1357                            "\t\t\t\tinitial_rate 0x%x retries %d, idx=%d ssn=%d next_reclaimed=0x%x seq_ctl=0x%x\n",
1358                            le32_to_cpu(tx_resp->initial_rate),
1359                            tx_resp->failure_frame, SEQ_TO_INDEX(sequence),
1360                            ssn, next_reclaimed, seq_ctl);
1361
1362         rcu_read_lock();
1363
1364         sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
1365         /*
1366          * sta can't be NULL otherwise it'd mean that the sta has been freed in
1367          * the firmware while we still have packets for it in the Tx queues.
1368          */
1369         if (WARN_ON_ONCE(!sta))
1370                 goto out;
1371
1372         if (!IS_ERR(sta)) {
1373                 mvmsta = iwl_mvm_sta_from_mac80211(sta);
1374
1375                 if (tid != IWL_TID_NON_QOS) {
1376                         struct iwl_mvm_tid_data *tid_data =
1377                                 &mvmsta->tid_data[tid];
1378                         bool send_eosp_ndp = false;
1379
1380                         spin_lock_bh(&mvmsta->lock);
1381                         if (iwl_mvm_is_dqa_supported(mvm)) {
1382                                 enum iwl_mvm_agg_state state;
1383
1384                                 state = mvmsta->tid_data[tid].state;
1385                                 txq_agg = (state == IWL_AGG_ON ||
1386                                         state == IWL_EMPTYING_HW_QUEUE_DELBA);
1387                         } else {
1388                                 txq_agg = txq_id >= mvm->first_agg_queue;
1389                         }
1390
1391                         if (!is_ndp) {
1392                                 tid_data->next_reclaimed = next_reclaimed;
1393                                 IWL_DEBUG_TX_REPLY(mvm,
1394                                                    "Next reclaimed packet:%d\n",
1395                                                    next_reclaimed);
1396                         } else {
1397                                 IWL_DEBUG_TX_REPLY(mvm,
1398                                                    "NDP - don't update next_reclaimed\n");
1399                         }
1400
1401                         iwl_mvm_check_ratid_empty(mvm, sta, tid);
1402
1403                         if (mvmsta->sleep_tx_count) {
1404                                 mvmsta->sleep_tx_count--;
1405                                 if (mvmsta->sleep_tx_count &&
1406                                     !iwl_mvm_tid_queued(tid_data)) {
1407                                         /*
1408                                          * The number of frames in the queue
1409                                          * dropped to 0 even if we sent less
1410                                          * frames than we thought we had on the
1411                                          * Tx queue.
1412                                          * This means we had holes in the BA
1413                                          * window that we just filled, ask
1414                                          * mac80211 to send EOSP since the
1415                                          * firmware won't know how to do that.
1416                                          * Send NDP and the firmware will send
1417                                          * EOSP notification that will trigger
1418                                          * a call to ieee80211_sta_eosp().
1419                                          */
1420                                         send_eosp_ndp = true;
1421                                 }
1422                         }
1423
1424                         spin_unlock_bh(&mvmsta->lock);
1425                         if (send_eosp_ndp) {
1426                                 iwl_mvm_sta_modify_sleep_tx_count(mvm, sta,
1427                                         IEEE80211_FRAME_RELEASE_UAPSD,
1428                                         1, tid, false, false);
1429                                 mvmsta->sleep_tx_count = 0;
1430                                 ieee80211_send_eosp_nullfunc(sta, tid);
1431                         }
1432                 }
1433
1434                 if (mvmsta->next_status_eosp) {
1435                         mvmsta->next_status_eosp = false;
1436                         ieee80211_sta_eosp(sta);
1437                 }
1438         } else {
1439                 mvmsta = NULL;
1440         }
1441
1442         /*
1443          * If the txq is not an AMPDU queue, there is no chance we freed
1444          * several skbs. Check that out...
1445          */
1446         if (txq_agg)
1447                 goto out;
1448
1449         /* We can't free more than one frame at once on a shared queue */
1450         WARN_ON(!iwl_mvm_is_dqa_supported(mvm) && (skb_freed > 1));
1451
1452         /* If we have still frames for this STA nothing to do here */
1453         if (!atomic_sub_and_test(skb_freed, &mvm->pending_frames[sta_id]))
1454                 goto out;
1455
1456         if (mvmsta && mvmsta->vif->type == NL80211_IFTYPE_AP) {
1457
1458                 /*
1459                  * If there are no pending frames for this STA and
1460                  * the tx to this station is not disabled, notify
1461                  * mac80211 that this station can now wake up in its
1462                  * STA table.
1463                  * If mvmsta is not NULL, sta is valid.
1464                  */
1465
1466                 spin_lock_bh(&mvmsta->lock);
1467
1468                 if (!mvmsta->disable_tx)
1469                         ieee80211_sta_block_awake(mvm->hw, sta, false);
1470
1471                 spin_unlock_bh(&mvmsta->lock);
1472         }
1473
1474         if (PTR_ERR(sta) == -EBUSY || PTR_ERR(sta) == -ENOENT) {
1475                 /*
1476                  * We are draining and this was the last packet - pre_rcu_remove
1477                  * has been called already. We might be after the
1478                  * synchronize_net already.
1479                  * Don't rely on iwl_mvm_rm_sta to see the empty Tx queues.
1480                  */
1481                 set_bit(sta_id, mvm->sta_drained);
1482                 schedule_work(&mvm->sta_drained_wk);
1483         }
1484
1485 out:
1486         rcu_read_unlock();
1487 }
1488
1489 #ifdef CONFIG_IWLWIFI_DEBUG
1490 #define AGG_TX_STATE_(x) case AGG_TX_STATE_ ## x: return #x
1491 static const char *iwl_get_agg_tx_status(u16 status)
1492 {
1493         switch (status & AGG_TX_STATE_STATUS_MSK) {
1494         AGG_TX_STATE_(TRANSMITTED);
1495         AGG_TX_STATE_(UNDERRUN);
1496         AGG_TX_STATE_(BT_PRIO);
1497         AGG_TX_STATE_(FEW_BYTES);
1498         AGG_TX_STATE_(ABORT);
1499         AGG_TX_STATE_(LAST_SENT_TTL);
1500         AGG_TX_STATE_(LAST_SENT_TRY_CNT);
1501         AGG_TX_STATE_(LAST_SENT_BT_KILL);
1502         AGG_TX_STATE_(SCD_QUERY);
1503         AGG_TX_STATE_(TEST_BAD_CRC32);
1504         AGG_TX_STATE_(RESPONSE);
1505         AGG_TX_STATE_(DUMP_TX);
1506         AGG_TX_STATE_(DELAY_TX);
1507         }
1508
1509         return "UNKNOWN";
1510 }
1511
1512 static void iwl_mvm_rx_tx_cmd_agg_dbg(struct iwl_mvm *mvm,
1513                                       struct iwl_rx_packet *pkt)
1514 {
1515         struct iwl_mvm_tx_resp *tx_resp = (void *)pkt->data;
1516         struct agg_tx_status *frame_status = &tx_resp->status;
1517         int i;
1518
1519         for (i = 0; i < tx_resp->frame_count; i++) {
1520                 u16 fstatus = le16_to_cpu(frame_status[i].status);
1521
1522                 IWL_DEBUG_TX_REPLY(mvm,
1523                                    "status %s (0x%04x), try-count (%d) seq (0x%x)\n",
1524                                    iwl_get_agg_tx_status(fstatus),
1525                                    fstatus & AGG_TX_STATE_STATUS_MSK,
1526                                    (fstatus & AGG_TX_STATE_TRY_CNT_MSK) >>
1527                                         AGG_TX_STATE_TRY_CNT_POS,
1528                                    le16_to_cpu(frame_status[i].sequence));
1529         }
1530 }
1531 #else
1532 static void iwl_mvm_rx_tx_cmd_agg_dbg(struct iwl_mvm *mvm,
1533                                       struct iwl_rx_packet *pkt)
1534 {}
1535 #endif /* CONFIG_IWLWIFI_DEBUG */
1536
1537 static void iwl_mvm_rx_tx_cmd_agg(struct iwl_mvm *mvm,
1538                                   struct iwl_rx_packet *pkt)
1539 {
1540         struct iwl_mvm_tx_resp *tx_resp = (void *)pkt->data;
1541         int sta_id = IWL_MVM_TX_RES_GET_RA(tx_resp->ra_tid);
1542         int tid = IWL_MVM_TX_RES_GET_TID(tx_resp->ra_tid);
1543         u16 sequence = le16_to_cpu(pkt->hdr.sequence);
1544         struct iwl_mvm_sta *mvmsta;
1545         int queue = SEQ_TO_QUEUE(sequence);
1546
1547         if (WARN_ON_ONCE(queue < mvm->first_agg_queue &&
1548                          (!iwl_mvm_is_dqa_supported(mvm) ||
1549                           (queue != IWL_MVM_DQA_BSS_CLIENT_QUEUE))))
1550                 return;
1551
1552         if (WARN_ON_ONCE(tid == IWL_TID_NON_QOS))
1553                 return;
1554
1555         iwl_mvm_rx_tx_cmd_agg_dbg(mvm, pkt);
1556
1557         rcu_read_lock();
1558
1559         mvmsta = iwl_mvm_sta_from_staid_rcu(mvm, sta_id);
1560
1561         if (!WARN_ON_ONCE(!mvmsta)) {
1562                 mvmsta->tid_data[tid].rate_n_flags =
1563                         le32_to_cpu(tx_resp->initial_rate);
1564                 mvmsta->tid_data[tid].tx_time =
1565                         le16_to_cpu(tx_resp->wireless_media_time);
1566         }
1567
1568         rcu_read_unlock();
1569 }
1570
1571 void iwl_mvm_rx_tx_cmd(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb)
1572 {
1573         struct iwl_rx_packet *pkt = rxb_addr(rxb);
1574         struct iwl_mvm_tx_resp *tx_resp = (void *)pkt->data;
1575
1576         if (tx_resp->frame_count == 1)
1577                 iwl_mvm_rx_tx_cmd_single(mvm, pkt);
1578         else
1579                 iwl_mvm_rx_tx_cmd_agg(mvm, pkt);
1580 }
1581
1582 static void iwl_mvm_tx_info_from_ba_notif(struct ieee80211_tx_info *info,
1583                                           struct iwl_mvm_ba_notif *ba_notif,
1584                                           struct iwl_mvm_tid_data *tid_data)
1585 {
1586         info->flags |= IEEE80211_TX_STAT_AMPDU;
1587         info->status.ampdu_ack_len = ba_notif->txed_2_done;
1588         info->status.ampdu_len = ba_notif->txed;
1589         iwl_mvm_hwrate_to_tx_status(tid_data->rate_n_flags,
1590                                     info);
1591         /* TODO: not accounted if the whole A-MPDU failed */
1592         info->status.tx_time = tid_data->tx_time;
1593         info->status.status_driver_data[0] =
1594                 (void *)(uintptr_t)ba_notif->reduced_txp;
1595         info->status.status_driver_data[1] =
1596                 (void *)(uintptr_t)tid_data->rate_n_flags;
1597 }
1598
1599 void iwl_mvm_rx_ba_notif(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb)
1600 {
1601         struct iwl_rx_packet *pkt = rxb_addr(rxb);
1602         struct iwl_mvm_ba_notif *ba_notif = (void *)pkt->data;
1603         struct sk_buff_head reclaimed_skbs;
1604         struct iwl_mvm_tid_data *tid_data;
1605         struct ieee80211_sta *sta;
1606         struct iwl_mvm_sta *mvmsta;
1607         struct sk_buff *skb;
1608         int sta_id, tid, freed;
1609         /* "flow" corresponds to Tx queue */
1610         u16 scd_flow = le16_to_cpu(ba_notif->scd_flow);
1611         /* "ssn" is start of block-ack Tx window, corresponds to index
1612          * (in Tx queue's circular buffer) of first TFD/frame in window */
1613         u16 ba_resp_scd_ssn = le16_to_cpu(ba_notif->scd_ssn);
1614
1615         sta_id = ba_notif->sta_id;
1616         tid = ba_notif->tid;
1617
1618         if (WARN_ONCE(sta_id >= IWL_MVM_STATION_COUNT ||
1619                       tid >= IWL_MAX_TID_COUNT,
1620                       "sta_id %d tid %d", sta_id, tid))
1621                 return;
1622
1623         rcu_read_lock();
1624
1625         sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
1626
1627         /* Reclaiming frames for a station that has been deleted ? */
1628         if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta))) {
1629                 rcu_read_unlock();
1630                 return;
1631         }
1632
1633         mvmsta = iwl_mvm_sta_from_mac80211(sta);
1634         tid_data = &mvmsta->tid_data[tid];
1635
1636         if (tid_data->txq_id != scd_flow) {
1637                 IWL_ERR(mvm,
1638                         "invalid BA notification: Q %d, tid %d, flow %d\n",
1639                         tid_data->txq_id, tid, scd_flow);
1640                 rcu_read_unlock();
1641                 return;
1642         }
1643
1644         spin_lock_bh(&mvmsta->lock);
1645
1646         __skb_queue_head_init(&reclaimed_skbs);
1647
1648         /*
1649          * Release all TFDs before the SSN, i.e. all TFDs in front of
1650          * block-ack window (we assume that they've been successfully
1651          * transmitted ... if not, it's too late anyway).
1652          */
1653         iwl_trans_reclaim(mvm->trans, scd_flow, ba_resp_scd_ssn,
1654                           &reclaimed_skbs);
1655
1656         IWL_DEBUG_TX_REPLY(mvm,
1657                            "BA_NOTIFICATION Received from %pM, sta_id = %d\n",
1658                            (u8 *)&ba_notif->sta_addr_lo32,
1659                            ba_notif->sta_id);
1660         IWL_DEBUG_TX_REPLY(mvm,
1661                            "TID = %d, SeqCtl = %d, bitmap = 0x%llx, scd_flow = %d, scd_ssn = %d sent:%d, acked:%d\n",
1662                            ba_notif->tid, le16_to_cpu(ba_notif->seq_ctl),
1663                            (unsigned long long)le64_to_cpu(ba_notif->bitmap),
1664                            scd_flow, ba_resp_scd_ssn, ba_notif->txed,
1665                            ba_notif->txed_2_done);
1666
1667         IWL_DEBUG_TX_REPLY(mvm, "reduced txp from ba notif %d\n",
1668                            ba_notif->reduced_txp);
1669         tid_data->next_reclaimed = ba_resp_scd_ssn;
1670
1671         iwl_mvm_check_ratid_empty(mvm, sta, tid);
1672
1673         freed = 0;
1674
1675         skb_queue_walk(&reclaimed_skbs, skb) {
1676                 struct ieee80211_hdr *hdr = (void *)skb->data;
1677                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1678
1679                 if (ieee80211_is_data_qos(hdr->frame_control))
1680                         freed++;
1681                 else
1682                         WARN_ON_ONCE(1);
1683
1684                 iwl_trans_free_tx_cmd(mvm->trans, info->driver_data[1]);
1685
1686                 memset(&info->status, 0, sizeof(info->status));
1687                 /* Packet was transmitted successfully, failures come as single
1688                  * frames because before failing a frame the firmware transmits
1689                  * it without aggregation at least once.
1690                  */
1691                 info->flags |= IEEE80211_TX_STAT_ACK;
1692
1693                 /* this is the first skb we deliver in this batch */
1694                 /* put the rate scaling data there */
1695                 if (freed == 1)
1696                         iwl_mvm_tx_info_from_ba_notif(info, ba_notif, tid_data);
1697         }
1698
1699         spin_unlock_bh(&mvmsta->lock);
1700
1701         /* We got a BA notif with 0 acked or scd_ssn didn't progress which is
1702          * possible (i.e. first MPDU in the aggregation wasn't acked)
1703          * Still it's important to update RS about sent vs. acked.
1704          */
1705         if (skb_queue_empty(&reclaimed_skbs)) {
1706                 struct ieee80211_tx_info ba_info = {};
1707                 struct ieee80211_chanctx_conf *chanctx_conf = NULL;
1708
1709                 if (mvmsta->vif)
1710                         chanctx_conf =
1711                                 rcu_dereference(mvmsta->vif->chanctx_conf);
1712
1713                 if (WARN_ON_ONCE(!chanctx_conf))
1714                         goto out;
1715
1716                 ba_info.band = chanctx_conf->def.chan->band;
1717                 iwl_mvm_tx_info_from_ba_notif(&ba_info, ba_notif, tid_data);
1718
1719                 IWL_DEBUG_TX_REPLY(mvm, "No reclaim. Update rs directly\n");
1720                 iwl_mvm_rs_tx_status(mvm, sta, tid, &ba_info, false);
1721         }
1722
1723 out:
1724         rcu_read_unlock();
1725
1726         while (!skb_queue_empty(&reclaimed_skbs)) {
1727                 skb = __skb_dequeue(&reclaimed_skbs);
1728                 ieee80211_tx_status(mvm->hw, skb);
1729         }
1730 }
1731
1732 /*
1733  * Note that there are transports that buffer frames before they reach
1734  * the firmware. This means that after flush_tx_path is called, the
1735  * queue might not be empty. The race-free way to handle this is to:
1736  * 1) set the station as draining
1737  * 2) flush the Tx path
1738  * 3) wait for the transport queues to be empty
1739  */
1740 int iwl_mvm_flush_tx_path(struct iwl_mvm *mvm, u32 tfd_msk, u32 flags)
1741 {
1742         int ret;
1743         struct iwl_tx_path_flush_cmd flush_cmd = {
1744                 .queues_ctl = cpu_to_le32(tfd_msk),
1745                 .flush_ctl = cpu_to_le16(DUMP_TX_FIFO_FLUSH),
1746         };
1747
1748         ret = iwl_mvm_send_cmd_pdu(mvm, TXPATH_FLUSH, flags,
1749                                    sizeof(flush_cmd), &flush_cmd);
1750         if (ret)
1751                 IWL_ERR(mvm, "Failed to send flush command (%d)\n", ret);
1752         return ret;
1753 }