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