drivers/net/wireless/ath/ath10k/htt_rx.c

   1 // SPDX-License-Identifier: ISC
   2 /*
   3  * Copyright (c) 2005-2011 Atheros Communications Inc.
   4  * Copyright (c) 2011-2017 Qualcomm Atheros, Inc.
   5  * Copyright (c) 2018, The Linux Foundation. All rights reserved.
   6  */
   7
   8 #include "core.h"
   9 #include "htc.h"
  10 #include "htt.h"
  11 #include "txrx.h"
  12 #include "debug.h"
  13 #include "trace.h"
  14 #include "mac.h"
  15
  16 #include <linux/log2.h>
  17 #include <linux/bitfield.h>
  18
  19 /* when under memory pressure rx ring refill may fail and needs a retry */
  20 #define HTT_RX_RING_REFILL_RETRY_MS 50
  21
  22 #define HTT_RX_RING_REFILL_RESCHED_MS 5
  23
  24 /* shortcut to interpret a raw memory buffer as a rx descriptor */
  25 #define HTT_RX_BUF_TO_RX_DESC(hw, buf) ath10k_htt_rx_desc_from_raw_buffer(hw, buf)
  26
  27 static int ath10k_htt_rx_get_csum_state(struct ath10k_hw_params *hw, struct sk_buff *skb);
  28
  29 static struct sk_buff *
  30 ath10k_htt_rx_find_skb_paddr(struct ath10k *ar, u64 paddr)
  31 {
  32         struct ath10k_skb_rxcb *rxcb;
  33
  34         hash_for_each_possible(ar->htt.rx_ring.skb_table, rxcb, hlist, paddr)
  35                 if (rxcb->paddr == paddr)
  36                         return ATH10K_RXCB_SKB(rxcb);
  37
  38         WARN_ON_ONCE(1);
  39         return NULL;
  40 }
  41
  42 static void ath10k_htt_rx_ring_free(struct ath10k_htt *htt)
  43 {
  44         struct sk_buff *skb;
  45         struct ath10k_skb_rxcb *rxcb;
  46         struct hlist_node *n;
  47         int i;
  48
  49         if (htt->rx_ring.in_ord_rx) {
  50                 hash_for_each_safe(htt->rx_ring.skb_table, i, n, rxcb, hlist) {
  51                         skb = ATH10K_RXCB_SKB(rxcb);
  52                         dma_unmap_single(htt->ar->dev, rxcb->paddr,
  53                                          skb->len + skb_tailroom(skb),
  54                                          DMA_FROM_DEVICE);
  55                         hash_del(&rxcb->hlist);
  56                         dev_kfree_skb_any(skb);
  57                 }
  58         } else {
  59                 for (i = 0; i < htt->rx_ring.size; i++) {
  60                         skb = htt->rx_ring.netbufs_ring[i];
  61                         if (!skb)
  62                                 continue;
  63
  64                         rxcb = ATH10K_SKB_RXCB(skb);
  65                         dma_unmap_single(htt->ar->dev, rxcb->paddr,
  66                                          skb->len + skb_tailroom(skb),
  67                                          DMA_FROM_DEVICE);
  68                         dev_kfree_skb_any(skb);
  69                 }
  70         }
  71
  72         htt->rx_ring.fill_cnt = 0;
  73         hash_init(htt->rx_ring.skb_table);
  74         memset(htt->rx_ring.netbufs_ring, 0,
  75                htt->rx_ring.size * sizeof(htt->rx_ring.netbufs_ring[0]));
  76 }
  77
  78 static size_t ath10k_htt_get_rx_ring_size_32(struct ath10k_htt *htt)
  79 {
  80         return htt->rx_ring.size * sizeof(htt->rx_ring.paddrs_ring_32);
  81 }
  82
  83 static size_t ath10k_htt_get_rx_ring_size_64(struct ath10k_htt *htt)
  84 {
  85         return htt->rx_ring.size * sizeof(htt->rx_ring.paddrs_ring_64);
  86 }
  87
  88 static void ath10k_htt_config_paddrs_ring_32(struct ath10k_htt *htt,
  89                                              void *vaddr)
  90 {
  91         htt->rx_ring.paddrs_ring_32 = vaddr;
  92 }
  93
  94 static void ath10k_htt_config_paddrs_ring_64(struct ath10k_htt *htt,
  95                                              void *vaddr)
  96 {
  97         htt->rx_ring.paddrs_ring_64 = vaddr;
  98 }
  99
 100 static void ath10k_htt_set_paddrs_ring_32(struct ath10k_htt *htt,
 101                                           dma_addr_t paddr, int idx)
 102 {
 103         htt->rx_ring.paddrs_ring_32[idx] = __cpu_to_le32(paddr);
 104 }
 105
 106 static void ath10k_htt_set_paddrs_ring_64(struct ath10k_htt *htt,
 107                                           dma_addr_t paddr, int idx)
 108 {
 109         htt->rx_ring.paddrs_ring_64[idx] = __cpu_to_le64(paddr);
 110 }
 111
 112 static void ath10k_htt_reset_paddrs_ring_32(struct ath10k_htt *htt, int idx)
 113 {
 114         htt->rx_ring.paddrs_ring_32[idx] = 0;
 115 }
 116
 117 static void ath10k_htt_reset_paddrs_ring_64(struct ath10k_htt *htt, int idx)
 118 {
 119         htt->rx_ring.paddrs_ring_64[idx] = 0;
 120 }
 121
 122 static void *ath10k_htt_get_vaddr_ring_32(struct ath10k_htt *htt)
 123 {
 124         return (void *)htt->rx_ring.paddrs_ring_32;
 125 }
 126
 127 static void *ath10k_htt_get_vaddr_ring_64(struct ath10k_htt *htt)
 128 {
 129         return (void *)htt->rx_ring.paddrs_ring_64;
 130 }
 131
 132 static int __ath10k_htt_rx_ring_fill_n(struct ath10k_htt *htt, int num)
 133 {
 134         struct ath10k_hw_params *hw = &htt->ar->hw_params;
 135         struct htt_rx_desc *rx_desc;
 136         struct ath10k_skb_rxcb *rxcb;
 137         struct sk_buff *skb;
 138         dma_addr_t paddr;
 139         int ret = 0, idx;
 140
 141         /* The Full Rx Reorder firmware has no way of telling the host
 142          * implicitly when it copied HTT Rx Ring buffers to MAC Rx Ring.
 143          * To keep things simple make sure ring is always half empty. This
 144          * guarantees there'll be no replenishment overruns possible.
 145          */
 146         BUILD_BUG_ON(HTT_RX_RING_FILL_LEVEL >= HTT_RX_RING_SIZE / 2);
 147
 148         idx = __le32_to_cpu(*htt->rx_ring.alloc_idx.vaddr);
 149
 150         if (idx < 0 || idx >= htt->rx_ring.size) {
 151                 ath10k_err(htt->ar, "rx ring index is not valid, firmware malfunctioning?\n");
 152                 idx &= htt->rx_ring.size_mask;
 153                 ret = -ENOMEM;
 154                 goto fail;
 155         }
 156
 157         while (num > 0) {
 158                 skb = dev_alloc_skb(HTT_RX_BUF_SIZE + HTT_RX_DESC_ALIGN);
 159                 if (!skb) {
 160                         ret = -ENOMEM;
 161                         goto fail;
 162                 }
 163
 164                 if (!IS_ALIGNED((unsigned long)skb->data, HTT_RX_DESC_ALIGN))
 165                         skb_pull(skb,
 166                                  PTR_ALIGN(skb->data, HTT_RX_DESC_ALIGN) -
 167                                  skb->data);
 168
 169                 /* Clear rx_desc attention word before posting to Rx ring */
 170                 rx_desc = HTT_RX_BUF_TO_RX_DESC(hw, skb->data);
 171                 ath10k_htt_rx_desc_get_attention(hw, rx_desc)->flags = __cpu_to_le32(0);
 172
 173                 paddr = dma_map_single(htt->ar->dev, skb->data,
 174                                        skb->len + skb_tailroom(skb),
 175                                        DMA_FROM_DEVICE);
 176
 177                 if (unlikely(dma_mapping_error(htt->ar->dev, paddr))) {
 178                         dev_kfree_skb_any(skb);
 179                         ret = -ENOMEM;
 180                         goto fail;
 181                 }
 182
 183                 rxcb = ATH10K_SKB_RXCB(skb);
 184                 rxcb->paddr = paddr;
 185                 htt->rx_ring.netbufs_ring[idx] = skb;
 186                 ath10k_htt_set_paddrs_ring(htt, paddr, idx);
 187                 htt->rx_ring.fill_cnt++;
 188
 189                 if (htt->rx_ring.in_ord_rx) {
 190                         hash_add(htt->rx_ring.skb_table,
 191                                  &ATH10K_SKB_RXCB(skb)->hlist,
 192                                  paddr);
 193                 }
 194
 195                 num--;
 196                 idx++;
 197                 idx &= htt->rx_ring.size_mask;
 198         }
 199
 200 fail:
 201         /*
 202          * Make sure the rx buffer is updated before available buffer
 203          * index to avoid any potential rx ring corruption.
 204          */
 205         mb();
 206         *htt->rx_ring.alloc_idx.vaddr = __cpu_to_le32(idx);
 207         return ret;
 208 }
 209
 210 static int ath10k_htt_rx_ring_fill_n(struct ath10k_htt *htt, int num)
 211 {
 212         lockdep_assert_held(&htt->rx_ring.lock);
 213         return __ath10k_htt_rx_ring_fill_n(htt, num);
 214 }
 215
 216 static void ath10k_htt_rx_msdu_buff_replenish(struct ath10k_htt *htt)
 217 {
 218         int ret, num_deficit, num_to_fill;
 219
 220         /* Refilling the whole RX ring buffer proves to be a bad idea. The
 221          * reason is RX may take up significant amount of CPU cycles and starve
 222          * other tasks, e.g. TX on an ethernet device while acting as a bridge
 223          * with ath10k wlan interface. This ended up with very poor performance
 224          * once CPU the host system was overwhelmed with RX on ath10k.
 225          *
 226          * By limiting the number of refills the replenishing occurs
 227          * progressively. This in turns makes use of the fact tasklets are
 228          * processed in FIFO order. This means actual RX processing can starve
 229          * out refilling. If there's not enough buffers on RX ring FW will not
 230          * report RX until it is refilled with enough buffers. This
 231          * automatically balances load wrt to CPU power.
 232          *
 233          * This probably comes at a cost of lower maximum throughput but
 234          * improves the average and stability.
 235          */
 236         spin_lock_bh(&htt->rx_ring.lock);
 237         num_deficit = htt->rx_ring.fill_level - htt->rx_ring.fill_cnt;
 238         num_to_fill = min(ATH10K_HTT_MAX_NUM_REFILL, num_deficit);
 239         num_deficit -= num_to_fill;
 240         ret = ath10k_htt_rx_ring_fill_n(htt, num_to_fill);
 241         if (ret == -ENOMEM) {
 242                 /*
 243                  * Failed to fill it to the desired level -
 244                  * we'll start a timer and try again next time.
 245                  * As long as enough buffers are left in the ring for
 246                  * another A-MPDU rx, no special recovery is needed.
 247                  */
 248                 mod_timer(&htt->rx_ring.refill_retry_timer, jiffies +
 249                           msecs_to_jiffies(HTT_RX_RING_REFILL_RETRY_MS));
 250         } else if (num_deficit > 0) {
 251                 mod_timer(&htt->rx_ring.refill_retry_timer, jiffies +
 252                           msecs_to_jiffies(HTT_RX_RING_REFILL_RESCHED_MS));
 253         }
 254         spin_unlock_bh(&htt->rx_ring.lock);
 255 }
 256
 257 static void ath10k_htt_rx_ring_refill_retry(struct timer_list *t)
 258 {
 259         struct ath10k_htt *htt = from_timer(htt, t, rx_ring.refill_retry_timer);
 260
 261         ath10k_htt_rx_msdu_buff_replenish(htt);
 262 }
 263
 264 int ath10k_htt_rx_ring_refill(struct ath10k *ar)
 265 {
 266         struct ath10k_htt *htt = &ar->htt;
 267         int ret;
 268
 269         if (ar->bus_param.dev_type == ATH10K_DEV_TYPE_HL)
 270                 return 0;
 271
 272         spin_lock_bh(&htt->rx_ring.lock);
 273         ret = ath10k_htt_rx_ring_fill_n(htt, (htt->rx_ring.fill_level -
 274                                               htt->rx_ring.fill_cnt));
 275
 276         if (ret)
 277                 ath10k_htt_rx_ring_free(htt);
 278
 279         spin_unlock_bh(&htt->rx_ring.lock);
 280
 281         return ret;
 282 }
 283
 284 void ath10k_htt_rx_free(struct ath10k_htt *htt)
 285 {
 286         if (htt->ar->bus_param.dev_type == ATH10K_DEV_TYPE_HL)
 287                 return;
 288
 289         del_timer_sync(&htt->rx_ring.refill_retry_timer);
 290
 291         skb_queue_purge(&htt->rx_msdus_q);
 292         skb_queue_purge(&htt->rx_in_ord_compl_q);
 293         skb_queue_purge(&htt->tx_fetch_ind_q);
 294
 295         spin_lock_bh(&htt->rx_ring.lock);
 296         ath10k_htt_rx_ring_free(htt);
 297         spin_unlock_bh(&htt->rx_ring.lock);
 298
 299         dma_free_coherent(htt->ar->dev,
 300                           ath10k_htt_get_rx_ring_size(htt),
 301                           ath10k_htt_get_vaddr_ring(htt),
 302                           htt->rx_ring.base_paddr);
 303
 304         dma_free_coherent(htt->ar->dev,
 305                           sizeof(*htt->rx_ring.alloc_idx.vaddr),
 306                           htt->rx_ring.alloc_idx.vaddr,
 307                           htt->rx_ring.alloc_idx.paddr);
 308
 309         kfree(htt->rx_ring.netbufs_ring);
 310 }
 311
 312 static inline struct sk_buff *ath10k_htt_rx_netbuf_pop(struct ath10k_htt *htt)
 313 {
 314         struct ath10k *ar = htt->ar;
 315         int idx;
 316         struct sk_buff *msdu;
 317
 318         lockdep_assert_held(&htt->rx_ring.lock);
 319
 320         if (htt->rx_ring.fill_cnt == 0) {
 321                 ath10k_warn(ar, "tried to pop sk_buff from an empty rx ring\n");
 322                 return NULL;
 323         }
 324
 325         idx = htt->rx_ring.sw_rd_idx.msdu_payld;
 326         msdu = htt->rx_ring.netbufs_ring[idx];
 327         htt->rx_ring.netbufs_ring[idx] = NULL;
 328         ath10k_htt_reset_paddrs_ring(htt, idx);
 329
 330         idx++;
 331         idx &= htt->rx_ring.size_mask;
 332         htt->rx_ring.sw_rd_idx.msdu_payld = idx;
 333         htt->rx_ring.fill_cnt--;
 334
 335         dma_unmap_single(htt->ar->dev,
 336                          ATH10K_SKB_RXCB(msdu)->paddr,
 337                          msdu->len + skb_tailroom(msdu),
 338                          DMA_FROM_DEVICE);
 339         ath10k_dbg_dump(ar, ATH10K_DBG_HTT_DUMP, NULL, "htt rx netbuf pop: ",
 340                         msdu->data, msdu->len + skb_tailroom(msdu));
 341
 342         return msdu;
 343 }
 344
 345 /* return: < 0 fatal error, 0 - non chained msdu, 1 chained msdu */
 346 static int ath10k_htt_rx_amsdu_pop(struct ath10k_htt *htt,
 347                                    struct sk_buff_head *amsdu)
 348 {
 349         struct ath10k *ar = htt->ar;
 350         struct ath10k_hw_params *hw = &ar->hw_params;
 351         int msdu_len, msdu_chaining = 0;
 352         struct sk_buff *msdu;
 353         struct htt_rx_desc *rx_desc;
 354         struct rx_attention *rx_desc_attention;
 355         struct rx_frag_info_common *rx_desc_frag_info_common;
 356         struct rx_msdu_start_common *rx_desc_msdu_start_common;
 357         struct rx_msdu_end_common *rx_desc_msdu_end_common;
 358
 359         lockdep_assert_held(&htt->rx_ring.lock);
 360
 361         for (;;) {
 362                 int last_msdu, msdu_len_invalid, msdu_chained;
 363
 364                 msdu = ath10k_htt_rx_netbuf_pop(htt);
 365                 if (!msdu) {
 366                         __skb_queue_purge(amsdu);
 367                         return -ENOENT;
 368                 }
 369
 370                 __skb_queue_tail(amsdu, msdu);
 371
 372                 rx_desc = HTT_RX_BUF_TO_RX_DESC(hw, msdu->data);
 373                 rx_desc_attention = ath10k_htt_rx_desc_get_attention(hw, rx_desc);
 374                 rx_desc_msdu_start_common = ath10k_htt_rx_desc_get_msdu_start(hw,
 375                                                                               rx_desc);
 376                 rx_desc_msdu_end_common = ath10k_htt_rx_desc_get_msdu_end(hw, rx_desc);
 377                 rx_desc_frag_info_common = ath10k_htt_rx_desc_get_frag_info(hw, rx_desc);
 378
 379                 /* FIXME: we must report msdu payload since this is what caller
 380                  * expects now
 381                  */
 382                 skb_put(msdu, hw->rx_desc_ops->rx_desc_msdu_payload_offset);
 383                 skb_pull(msdu, hw->rx_desc_ops->rx_desc_msdu_payload_offset);
 384
 385                 /*
 386                  * Sanity check - confirm the HW is finished filling in the
 387                  * rx data.
 388                  * If the HW and SW are working correctly, then it's guaranteed
 389                  * that the HW's MAC DMA is done before this point in the SW.
 390                  * To prevent the case that we handle a stale Rx descriptor,
 391                  * just assert for now until we have a way to recover.
 392                  */
 393                 if (!(__le32_to_cpu(rx_desc_attention->flags)
 394                                 & RX_ATTENTION_FLAGS_MSDU_DONE)) {
 395                         __skb_queue_purge(amsdu);
 396                         return -EIO;
 397                 }
 398
 399                 msdu_len_invalid = !!(__le32_to_cpu(rx_desc_attention->flags)
 400                                         & (RX_ATTENTION_FLAGS_MPDU_LENGTH_ERR |
 401                                            RX_ATTENTION_FLAGS_MSDU_LENGTH_ERR));
 402                 msdu_len = MS(__le32_to_cpu(rx_desc_msdu_start_common->info0),
 403                               RX_MSDU_START_INFO0_MSDU_LENGTH);
 404                 msdu_chained = rx_desc_frag_info_common->ring2_more_count;
 405
 406                 if (msdu_len_invalid)
 407                         msdu_len = 0;
 408
 409                 skb_trim(msdu, 0);
 410                 skb_put(msdu, min(msdu_len, ath10k_htt_rx_msdu_size(hw)));
 411                 msdu_len -= msdu->len;
 412
 413                 /* Note: Chained buffers do not contain rx descriptor */
 414                 while (msdu_chained--) {
 415                         msdu = ath10k_htt_rx_netbuf_pop(htt);
 416                         if (!msdu) {
 417                                 __skb_queue_purge(amsdu);
 418                                 return -ENOENT;
 419                         }
 420
 421                         __skb_queue_tail(amsdu, msdu);
 422                         skb_trim(msdu, 0);
 423                         skb_put(msdu, min(msdu_len, HTT_RX_BUF_SIZE));
 424                         msdu_len -= msdu->len;
 425                         msdu_chaining = 1;
 426                 }
 427
 428                 last_msdu = __le32_to_cpu(rx_desc_msdu_end_common->info0) &
 429                                 RX_MSDU_END_INFO0_LAST_MSDU;
 430
 431                 /* FIXME: why are we skipping the first part of the rx_desc? */
 432                 trace_ath10k_htt_rx_desc(ar, (void *)rx_desc + sizeof(u32),
 433                                          hw->rx_desc_ops->rx_desc_size - sizeof(u32));
 434
 435                 if (last_msdu)
 436                         break;
 437         }
 438
 439         if (skb_queue_empty(amsdu))
 440                 msdu_chaining = -1;
 441
 442         /*
 443          * Don't refill the ring yet.
 444          *
 445          * First, the elements popped here are still in use - it is not
 446          * safe to overwrite them until the matching call to
 447          * mpdu_desc_list_next. Second, for efficiency it is preferable to
 448          * refill the rx ring with 1 PPDU's worth of rx buffers (something
 449          * like 32 x 3 buffers), rather than one MPDU's worth of rx buffers
 450          * (something like 3 buffers). Consequently, we'll rely on the txrx
 451          * SW to tell us when it is done pulling all the PPDU's rx buffers
 452          * out of the rx ring, and then refill it just once.
 453          */
 454
 455         return msdu_chaining;
 456 }
 457
 458 static struct sk_buff *ath10k_htt_rx_pop_paddr(struct ath10k_htt *htt,
 459                                                u64 paddr)
 460 {
 461         struct ath10k *ar = htt->ar;
 462         struct ath10k_skb_rxcb *rxcb;
 463         struct sk_buff *msdu;
 464
 465         lockdep_assert_held(&htt->rx_ring.lock);
 466
 467         msdu = ath10k_htt_rx_find_skb_paddr(ar, paddr);
 468         if (!msdu)
 469                 return NULL;
 470
 471         rxcb = ATH10K_SKB_RXCB(msdu);
 472         hash_del(&rxcb->hlist);
 473         htt->rx_ring.fill_cnt--;
 474
 475         dma_unmap_single(htt->ar->dev, rxcb->paddr,
 476                          msdu->len + skb_tailroom(msdu),
 477                          DMA_FROM_DEVICE);
 478         ath10k_dbg_dump(ar, ATH10K_DBG_HTT_DUMP, NULL, "htt rx netbuf pop: ",
 479                         msdu->data, msdu->len + skb_tailroom(msdu));
 480
 481         return msdu;
 482 }
 483
 484 static inline void ath10k_htt_append_frag_list(struct sk_buff *skb_head,
 485                                                struct sk_buff *frag_list,
 486                                                unsigned int frag_len)
 487 {
 488         skb_shinfo(skb_head)->frag_list = frag_list;
 489         skb_head->data_len = frag_len;
 490         skb_head->len += skb_head->data_len;
 491 }
 492
 493 static int ath10k_htt_rx_handle_amsdu_mon_32(struct ath10k_htt *htt,
 494                                              struct sk_buff *msdu,
 495                                              struct htt_rx_in_ord_msdu_desc **msdu_desc)
 496 {
 497         struct ath10k *ar = htt->ar;
 498         struct ath10k_hw_params *hw = &ar->hw_params;
 499         u32 paddr;
 500         struct sk_buff *frag_buf;
 501         struct sk_buff *prev_frag_buf;
 502         u8 last_frag;
 503         struct htt_rx_in_ord_msdu_desc *ind_desc = *msdu_desc;
 504         struct htt_rx_desc *rxd;
 505         int amsdu_len = __le16_to_cpu(ind_desc->msdu_len);
 506
 507         rxd = HTT_RX_BUF_TO_RX_DESC(hw, msdu->data);
 508         trace_ath10k_htt_rx_desc(ar, rxd, hw->rx_desc_ops->rx_desc_size);
 509
 510         skb_put(msdu, hw->rx_desc_ops->rx_desc_size);
 511         skb_pull(msdu, hw->rx_desc_ops->rx_desc_size);
 512         skb_put(msdu, min(amsdu_len, ath10k_htt_rx_msdu_size(hw)));
 513         amsdu_len -= msdu->len;
 514
 515         last_frag = ind_desc->reserved;
 516         if (last_frag) {
 517                 if (amsdu_len) {
 518                         ath10k_warn(ar, "invalid amsdu len %u, left %d",
 519                                     __le16_to_cpu(ind_desc->msdu_len),
 520                                     amsdu_len);
 521                 }
 522                 return 0;
 523         }
 524
 525         ind_desc++;
 526         paddr = __le32_to_cpu(ind_desc->msdu_paddr);
 527         frag_buf = ath10k_htt_rx_pop_paddr(htt, paddr);
 528         if (!frag_buf) {
 529                 ath10k_warn(ar, "failed to pop frag-1 paddr: 0x%x", paddr);
 530                 return -ENOENT;
 531         }
 532
 533         skb_put(frag_buf, min(amsdu_len, HTT_RX_BUF_SIZE));
 534         ath10k_htt_append_frag_list(msdu, frag_buf, amsdu_len);
 535
 536         amsdu_len -= frag_buf->len;
 537         prev_frag_buf = frag_buf;
 538         last_frag = ind_desc->reserved;
 539         while (!last_frag) {
 540                 ind_desc++;
 541                 paddr = __le32_to_cpu(ind_desc->msdu_paddr);
 542                 frag_buf = ath10k_htt_rx_pop_paddr(htt, paddr);
 543                 if (!frag_buf) {
 544                         ath10k_warn(ar, "failed to pop frag-n paddr: 0x%x",
 545                                     paddr);
 546                         prev_frag_buf->next = NULL;
 547                         return -ENOENT;
 548                 }
 549
 550                 skb_put(frag_buf, min(amsdu_len, HTT_RX_BUF_SIZE));
 551                 last_frag = ind_desc->reserved;
 552                 amsdu_len -= frag_buf->len;
 553
 554                 prev_frag_buf->next = frag_buf;
 555                 prev_frag_buf = frag_buf;
 556         }
 557
 558         if (amsdu_len) {
 559                 ath10k_warn(ar, "invalid amsdu len %u, left %d",
 560                             __le16_to_cpu(ind_desc->msdu_len), amsdu_len);
 561         }
 562
 563         *msdu_desc = ind_desc;
 564
 565         prev_frag_buf->next = NULL;
 566         return 0;
 567 }
 568
 569 static int
 570 ath10k_htt_rx_handle_amsdu_mon_64(struct ath10k_htt *htt,
 571                                   struct sk_buff *msdu,
 572                                   struct htt_rx_in_ord_msdu_desc_ext **msdu_desc)
 573 {
 574         struct ath10k *ar = htt->ar;
 575         struct ath10k_hw_params *hw = &ar->hw_params;
 576         u64 paddr;
 577         struct sk_buff *frag_buf;
 578         struct sk_buff *prev_frag_buf;
 579         u8 last_frag;
 580         struct htt_rx_in_ord_msdu_desc_ext *ind_desc = *msdu_desc;
 581         struct htt_rx_desc *rxd;
 582         int amsdu_len = __le16_to_cpu(ind_desc->msdu_len);
 583
 584         rxd = HTT_RX_BUF_TO_RX_DESC(hw, msdu->data);
 585         trace_ath10k_htt_rx_desc(ar, rxd, hw->rx_desc_ops->rx_desc_size);
 586
 587         skb_put(msdu, hw->rx_desc_ops->rx_desc_size);
 588         skb_pull(msdu, hw->rx_desc_ops->rx_desc_size);
 589         skb_put(msdu, min(amsdu_len, ath10k_htt_rx_msdu_size(hw)));
 590         amsdu_len -= msdu->len;
 591
 592         last_frag = ind_desc->reserved;
 593         if (last_frag) {
 594                 if (amsdu_len) {
 595                         ath10k_warn(ar, "invalid amsdu len %u, left %d",
 596                                     __le16_to_cpu(ind_desc->msdu_len),
 597                                     amsdu_len);
 598                 }
 599                 return 0;
 600         }
 601
 602         ind_desc++;
 603         paddr = __le64_to_cpu(ind_desc->msdu_paddr);
 604         frag_buf = ath10k_htt_rx_pop_paddr(htt, paddr);
 605         if (!frag_buf) {
 606                 ath10k_warn(ar, "failed to pop frag-1 paddr: 0x%llx", paddr);
 607                 return -ENOENT;
 608         }
 609
 610         skb_put(frag_buf, min(amsdu_len, HTT_RX_BUF_SIZE));
 611         ath10k_htt_append_frag_list(msdu, frag_buf, amsdu_len);
 612
 613         amsdu_len -= frag_buf->len;
 614         prev_frag_buf = frag_buf;
 615         last_frag = ind_desc->reserved;
 616         while (!last_frag) {
 617                 ind_desc++;
 618                 paddr = __le64_to_cpu(ind_desc->msdu_paddr);
 619                 frag_buf = ath10k_htt_rx_pop_paddr(htt, paddr);
 620                 if (!frag_buf) {
 621                         ath10k_warn(ar, "failed to pop frag-n paddr: 0x%llx",
 622                                     paddr);
 623                         prev_frag_buf->next = NULL;
 624                         return -ENOENT;
 625                 }
 626
 627                 skb_put(frag_buf, min(amsdu_len, HTT_RX_BUF_SIZE));
 628                 last_frag = ind_desc->reserved;
 629                 amsdu_len -= frag_buf->len;
 630
 631                 prev_frag_buf->next = frag_buf;
 632                 prev_frag_buf = frag_buf;
 633         }
 634
 635         if (amsdu_len) {
 636                 ath10k_warn(ar, "invalid amsdu len %u, left %d",
 637                             __le16_to_cpu(ind_desc->msdu_len), amsdu_len);
 638         }
 639
 640         *msdu_desc = ind_desc;
 641
 642         prev_frag_buf->next = NULL;
 643         return 0;
 644 }
 645
 646 static int ath10k_htt_rx_pop_paddr32_list(struct ath10k_htt *htt,
 647                                           struct htt_rx_in_ord_ind *ev,
 648                                           struct sk_buff_head *list)
 649 {
 650         struct ath10k *ar = htt->ar;
 651         struct ath10k_hw_params *hw = &ar->hw_params;
 652         struct htt_rx_in_ord_msdu_desc *msdu_desc = ev->msdu_descs32;
 653         struct htt_rx_desc *rxd;
 654         struct rx_attention *rxd_attention;
 655         struct sk_buff *msdu;
 656         int msdu_count, ret;
 657         bool is_offload;
 658         u32 paddr;
 659
 660         lockdep_assert_held(&htt->rx_ring.lock);
 661
 662         msdu_count = __le16_to_cpu(ev->msdu_count);
 663         is_offload = !!(ev->info & HTT_RX_IN_ORD_IND_INFO_OFFLOAD_MASK);
 664
 665         while (msdu_count--) {
 666                 paddr = __le32_to_cpu(msdu_desc->msdu_paddr);
 667
 668                 msdu = ath10k_htt_rx_pop_paddr(htt, paddr);
 669                 if (!msdu) {
 670                         __skb_queue_purge(list);
 671                         return -ENOENT;
 672                 }
 673
 674                 if (!is_offload && ar->monitor_arvif) {
 675                         ret = ath10k_htt_rx_handle_amsdu_mon_32(htt, msdu,
 676                                                                 &msdu_desc);
 677                         if (ret) {
 678                                 __skb_queue_purge(list);
 679                                 return ret;
 680                         }
 681                         __skb_queue_tail(list, msdu);
 682                         msdu_desc++;
 683                         continue;
 684                 }
 685
 686                 __skb_queue_tail(list, msdu);
 687
 688                 if (!is_offload) {
 689                         rxd = HTT_RX_BUF_TO_RX_DESC(hw, msdu->data);
 690                         rxd_attention = ath10k_htt_rx_desc_get_attention(hw, rxd);
 691
 692                         trace_ath10k_htt_rx_desc(ar, rxd, hw->rx_desc_ops->rx_desc_size);
 693
 694                         skb_put(msdu, hw->rx_desc_ops->rx_desc_size);
 695                         skb_pull(msdu, hw->rx_desc_ops->rx_desc_size);
 696                         skb_put(msdu, __le16_to_cpu(msdu_desc->msdu_len));
 697
 698                         if (!(__le32_to_cpu(rxd_attention->flags) &
 699                               RX_ATTENTION_FLAGS_MSDU_DONE)) {
 700                                 ath10k_warn(htt->ar, "tried to pop an incomplete frame, oops!\n");
 701                                 return -EIO;
 702                         }
 703                 }
 704
 705                 msdu_desc++;
 706         }
 707
 708         return 0;
 709 }
 710
 711 static int ath10k_htt_rx_pop_paddr64_list(struct ath10k_htt *htt,
 712                                           struct htt_rx_in_ord_ind *ev,
 713                                           struct sk_buff_head *list)
 714 {
 715         struct ath10k *ar = htt->ar;
 716         struct ath10k_hw_params *hw = &ar->hw_params;
 717         struct htt_rx_in_ord_msdu_desc_ext *msdu_desc = ev->msdu_descs64;
 718         struct htt_rx_desc *rxd;
 719         struct rx_attention *rxd_attention;
 720         struct sk_buff *msdu;
 721         int msdu_count, ret;
 722         bool is_offload;
 723         u64 paddr;
 724
 725         lockdep_assert_held(&htt->rx_ring.lock);
 726
 727         msdu_count = __le16_to_cpu(ev->msdu_count);
 728         is_offload = !!(ev->info & HTT_RX_IN_ORD_IND_INFO_OFFLOAD_MASK);
 729
 730         while (msdu_count--) {
 731                 paddr = __le64_to_cpu(msdu_desc->msdu_paddr);
 732                 msdu = ath10k_htt_rx_pop_paddr(htt, paddr);
 733                 if (!msdu) {
 734                         __skb_queue_purge(list);
 735                         return -ENOENT;
 736                 }
 737
 738                 if (!is_offload && ar->monitor_arvif) {
 739                         ret = ath10k_htt_rx_handle_amsdu_mon_64(htt, msdu,
 740                                                                 &msdu_desc);
 741                         if (ret) {
 742                                 __skb_queue_purge(list);
 743                                 return ret;
 744                         }
 745                         __skb_queue_tail(list, msdu);
 746                         msdu_desc++;
 747                         continue;
 748                 }
 749
 750                 __skb_queue_tail(list, msdu);
 751
 752                 if (!is_offload) {
 753                         rxd = HTT_RX_BUF_TO_RX_DESC(hw, msdu->data);
 754                         rxd_attention = ath10k_htt_rx_desc_get_attention(hw, rxd);
 755
 756                         trace_ath10k_htt_rx_desc(ar, rxd, hw->rx_desc_ops->rx_desc_size);
 757
 758                         skb_put(msdu, hw->rx_desc_ops->rx_desc_size);
 759                         skb_pull(msdu, hw->rx_desc_ops->rx_desc_size);
 760                         skb_put(msdu, __le16_to_cpu(msdu_desc->msdu_len));
 761
 762                         if (!(__le32_to_cpu(rxd_attention->flags) &
 763                               RX_ATTENTION_FLAGS_MSDU_DONE)) {
 764                                 ath10k_warn(htt->ar, "tried to pop an incomplete frame, oops!\n");
 765                                 return -EIO;
 766                         }
 767                 }
 768
 769                 msdu_desc++;
 770         }
 771
 772         return 0;
 773 }
 774
 775 int ath10k_htt_rx_alloc(struct ath10k_htt *htt)
 776 {
 777         struct ath10k *ar = htt->ar;
 778         dma_addr_t paddr;
 779         void *vaddr, *vaddr_ring;
 780         size_t size;
 781         struct timer_list *timer = &htt->rx_ring.refill_retry_timer;
 782
 783         if (ar->bus_param.dev_type == ATH10K_DEV_TYPE_HL)
 784                 return 0;
 785
 786         htt->rx_confused = false;
 787
 788         /* XXX: The fill level could be changed during runtime in response to
 789          * the host processing latency. Is this really worth it?
 790          */
 791         htt->rx_ring.size = HTT_RX_RING_SIZE;
 792         htt->rx_ring.size_mask = htt->rx_ring.size - 1;
 793         htt->rx_ring.fill_level = ar->hw_params.rx_ring_fill_level;
 794
 795         if (!is_power_of_2(htt->rx_ring.size)) {
 796                 ath10k_warn(ar, "htt rx ring size is not power of 2\n");
 797                 return -EINVAL;
 798         }
 799
 800         htt->rx_ring.netbufs_ring =
 801                 kcalloc(htt->rx_ring.size, sizeof(struct sk_buff *),
 802                         GFP_KERNEL);
 803         if (!htt->rx_ring.netbufs_ring)
 804                 goto err_netbuf;
 805
 806         size = ath10k_htt_get_rx_ring_size(htt);
 807
 808         vaddr_ring = dma_alloc_coherent(htt->ar->dev, size, &paddr, GFP_KERNEL);
 809         if (!vaddr_ring)
 810                 goto err_dma_ring;
 811
 812         ath10k_htt_config_paddrs_ring(htt, vaddr_ring);
 813         htt->rx_ring.base_paddr = paddr;
 814
 815         vaddr = dma_alloc_coherent(htt->ar->dev,
 816                                    sizeof(*htt->rx_ring.alloc_idx.vaddr),
 817                                    &paddr, GFP_KERNEL);
 818         if (!vaddr)
 819                 goto err_dma_idx;
 820
 821         htt->rx_ring.alloc_idx.vaddr = vaddr;
 822         htt->rx_ring.alloc_idx.paddr = paddr;
 823         htt->rx_ring.sw_rd_idx.msdu_payld = htt->rx_ring.size_mask;
 824         *htt->rx_ring.alloc_idx.vaddr = 0;
 825
 826         /* Initialize the Rx refill retry timer */
 827         timer_setup(timer, ath10k_htt_rx_ring_refill_retry, 0);
 828
 829         spin_lock_init(&htt->rx_ring.lock);
 830
 831         htt->rx_ring.fill_cnt = 0;
 832         htt->rx_ring.sw_rd_idx.msdu_payld = 0;
 833         hash_init(htt->rx_ring.skb_table);
 834
 835         skb_queue_head_init(&htt->rx_msdus_q);
 836         skb_queue_head_init(&htt->rx_in_ord_compl_q);
 837         skb_queue_head_init(&htt->tx_fetch_ind_q);
 838         atomic_set(&htt->num_mpdus_ready, 0);
 839
 840         ath10k_dbg(ar, ATH10K_DBG_BOOT, "htt rx ring size %d fill_level %d\n",
 841                    htt->rx_ring.size, htt->rx_ring.fill_level);
 842         return 0;
 843
 844 err_dma_idx:
 845         dma_free_coherent(htt->ar->dev,
 846                           ath10k_htt_get_rx_ring_size(htt),
 847                           vaddr_ring,
 848                           htt->rx_ring.base_paddr);
 849 err_dma_ring:
 850         kfree(htt->rx_ring.netbufs_ring);
 851 err_netbuf:
 852         return -ENOMEM;
 853 }
 854
 855 static int ath10k_htt_rx_crypto_param_len(struct ath10k *ar,
 856                                           enum htt_rx_mpdu_encrypt_type type)
 857 {
 858         switch (type) {
 859         case HTT_RX_MPDU_ENCRYPT_NONE:
 860                 return 0;
 861         case HTT_RX_MPDU_ENCRYPT_WEP40:
 862         case HTT_RX_MPDU_ENCRYPT_WEP104:
 863                 return IEEE80211_WEP_IV_LEN;
 864         case HTT_RX_MPDU_ENCRYPT_TKIP_WITHOUT_MIC:
 865         case HTT_RX_MPDU_ENCRYPT_TKIP_WPA:
 866                 return IEEE80211_TKIP_IV_LEN;
 867         case HTT_RX_MPDU_ENCRYPT_AES_CCM_WPA2:
 868                 return IEEE80211_CCMP_HDR_LEN;
 869         case HTT_RX_MPDU_ENCRYPT_AES_CCM256_WPA2:
 870                 return IEEE80211_CCMP_256_HDR_LEN;
 871         case HTT_RX_MPDU_ENCRYPT_AES_GCMP_WPA2:
 872         case HTT_RX_MPDU_ENCRYPT_AES_GCMP256_WPA2:
 873                 return IEEE80211_GCMP_HDR_LEN;
 874         case HTT_RX_MPDU_ENCRYPT_WEP128:
 875         case HTT_RX_MPDU_ENCRYPT_WAPI:
 876                 break;
 877         }
 878
 879         ath10k_warn(ar, "unsupported encryption type %d\n", type);
 880         return 0;
 881 }
 882
 883 #define MICHAEL_MIC_LEN 8
 884
 885 static int ath10k_htt_rx_crypto_mic_len(struct ath10k *ar,
 886                                         enum htt_rx_mpdu_encrypt_type type)
 887 {
 888         switch (type) {
 889         case HTT_RX_MPDU_ENCRYPT_NONE:
 890         case HTT_RX_MPDU_ENCRYPT_WEP40:
 891         case HTT_RX_MPDU_ENCRYPT_WEP104:
 892         case HTT_RX_MPDU_ENCRYPT_TKIP_WITHOUT_MIC:
 893         case HTT_RX_MPDU_ENCRYPT_TKIP_WPA:
 894                 return 0;
 895         case HTT_RX_MPDU_ENCRYPT_AES_CCM_WPA2:
 896                 return IEEE80211_CCMP_MIC_LEN;
 897         case HTT_RX_MPDU_ENCRYPT_AES_CCM256_WPA2:
 898                 return IEEE80211_CCMP_256_MIC_LEN;
 899         case HTT_RX_MPDU_ENCRYPT_AES_GCMP_WPA2:
 900         case HTT_RX_MPDU_ENCRYPT_AES_GCMP256_WPA2:
 901                 return IEEE80211_GCMP_MIC_LEN;
 902         case HTT_RX_MPDU_ENCRYPT_WEP128:
 903         case HTT_RX_MPDU_ENCRYPT_WAPI:
 904                 break;
 905         }
 906
 907         ath10k_warn(ar, "unsupported encryption type %d\n", type);
 908         return 0;
 909 }
 910
 911 static int ath10k_htt_rx_crypto_icv_len(struct ath10k *ar,
 912                                         enum htt_rx_mpdu_encrypt_type type)
 913 {
 914         switch (type) {
 915         case HTT_RX_MPDU_ENCRYPT_NONE:
 916         case HTT_RX_MPDU_ENCRYPT_AES_CCM_WPA2:
 917         case HTT_RX_MPDU_ENCRYPT_AES_CCM256_WPA2:
 918         case HTT_RX_MPDU_ENCRYPT_AES_GCMP_WPA2:
 919         case HTT_RX_MPDU_ENCRYPT_AES_GCMP256_WPA2:
 920                 return 0;
 921         case HTT_RX_MPDU_ENCRYPT_WEP40:
 922         case HTT_RX_MPDU_ENCRYPT_WEP104:
 923                 return IEEE80211_WEP_ICV_LEN;
 924         case HTT_RX_MPDU_ENCRYPT_TKIP_WITHOUT_MIC:
 925         case HTT_RX_MPDU_ENCRYPT_TKIP_WPA:
 926                 return IEEE80211_TKIP_ICV_LEN;
 927         case HTT_RX_MPDU_ENCRYPT_WEP128:
 928         case HTT_RX_MPDU_ENCRYPT_WAPI:
 929                 break;
 930         }
 931
 932         ath10k_warn(ar, "unsupported encryption type %d\n", type);
 933         return 0;
 934 }
 935
 936 struct amsdu_subframe_hdr {
 937         u8 dst[ETH_ALEN];
 938         u8 src[ETH_ALEN];
 939         __be16 len;
 940 } __packed;
 941
 942 #define GROUP_ID_IS_SU_MIMO(x) ((x) == 0 || (x) == 63)
 943
 944 static inline u8 ath10k_bw_to_mac80211_bw(u8 bw)
 945 {
 946         u8 ret = 0;
 947
 948         switch (bw) {
 949         case 0:
 950                 ret = RATE_INFO_BW_20;
 951                 break;
 952         case 1:
 953                 ret = RATE_INFO_BW_40;
 954                 break;
 955         case 2:
 956                 ret = RATE_INFO_BW_80;
 957                 break;
 958         case 3:
 959                 ret = RATE_INFO_BW_160;
 960                 break;
 961         }
 962
 963         return ret;
 964 }
 965
 966 static void ath10k_htt_rx_h_rates(struct ath10k *ar,
 967                                   struct ieee80211_rx_status *status,
 968                                   struct htt_rx_desc *rxd)
 969 {
 970         struct ath10k_hw_params *hw = &ar->hw_params;
 971         struct rx_attention *rxd_attention;
 972         struct rx_mpdu_start *rxd_mpdu_start;
 973         struct rx_mpdu_end *rxd_mpdu_end;
 974         struct rx_msdu_start_common *rxd_msdu_start_common;
 975         struct rx_msdu_end_common *rxd_msdu_end_common;
 976         struct rx_ppdu_start *rxd_ppdu_start;
 977         struct ieee80211_supported_band *sband;
 978         u8 cck, rate, bw, sgi, mcs, nss;
 979         u8 *rxd_msdu_payload;
 980         u8 preamble = 0;
 981         u8 group_id;
 982         u32 info1, info2, info3;
 983         u32 stbc, nsts_su;
 984
 985         rxd_attention = ath10k_htt_rx_desc_get_attention(hw, rxd);
 986         rxd_mpdu_start = ath10k_htt_rx_desc_get_mpdu_start(hw, rxd);
 987         rxd_mpdu_end = ath10k_htt_rx_desc_get_mpdu_end(hw, rxd);
 988         rxd_msdu_start_common = ath10k_htt_rx_desc_get_msdu_start(hw, rxd);
 989         rxd_msdu_end_common = ath10k_htt_rx_desc_get_msdu_end(hw, rxd);
 990         rxd_ppdu_start = ath10k_htt_rx_desc_get_ppdu_start(hw, rxd);
 991         rxd_msdu_payload = ath10k_htt_rx_desc_get_msdu_payload(hw, rxd);
 992
 993         info1 = __le32_to_cpu(rxd_ppdu_start->info1);
 994         info2 = __le32_to_cpu(rxd_ppdu_start->info2);
 995         info3 = __le32_to_cpu(rxd_ppdu_start->info3);
 996
 997         preamble = MS(info1, RX_PPDU_START_INFO1_PREAMBLE_TYPE);
 998
 999         switch (preamble) {
1000         case HTT_RX_LEGACY:
1001                 /* To get legacy rate index band is required. Since band can't
1002                  * be undefined check if freq is non-zero.
1003                  */
1004                 if (!status->freq)
1005                         return;
1006
1007                 cck = info1 & RX_PPDU_START_INFO1_L_SIG_RATE_SELECT;
1008                 rate = MS(info1, RX_PPDU_START_INFO1_L_SIG_RATE);
1009                 rate &= ~RX_PPDU_START_RATE_FLAG;
1010
1011                 sband = &ar->mac.sbands[status->band];
1012                 status->rate_idx = ath10k_mac_hw_rate_to_idx(sband, rate, cck);
1013                 break;
1014         case HTT_RX_HT:
1015         case HTT_RX_HT_WITH_TXBF:
1016                 /* HT-SIG - Table 20-11 in info2 and info3 */
1017                 mcs = info2 & 0x1F;
1018                 nss = mcs >> 3;
1019                 bw = (info2 >> 7) & 1;
1020                 sgi = (info3 >> 7) & 1;
1021
1022                 status->rate_idx = mcs;
1023                 status->encoding = RX_ENC_HT;
1024                 if (sgi)
1025                         status->enc_flags |= RX_ENC_FLAG_SHORT_GI;
1026                 if (bw)
1027                         status->bw = RATE_INFO_BW_40;
1028                 break;
1029         case HTT_RX_VHT:
1030         case HTT_RX_VHT_WITH_TXBF:
1031                 /* VHT-SIG-A1 in info2, VHT-SIG-A2 in info3
1032                  * TODO check this
1033                  */
1034                 bw = info2 & 3;
1035                 sgi = info3 & 1;
1036                 stbc = (info2 >> 3) & 1;
1037                 group_id = (info2 >> 4) & 0x3F;
1038
1039                 if (GROUP_ID_IS_SU_MIMO(group_id)) {
1040                         mcs = (info3 >> 4) & 0x0F;
1041                         nsts_su = ((info2 >> 10) & 0x07);
1042                         if (stbc)
1043                                 nss = (nsts_su >> 2) + 1;
1044                         else
1045                                 nss = (nsts_su + 1);
1046                 } else {
1047                         /* Hardware doesn't decode VHT-SIG-B into Rx descriptor
1048                          * so it's impossible to decode MCS. Also since
1049                          * firmware consumes Group Id Management frames host
1050                          * has no knowledge regarding group/user position
1051                          * mapping so it's impossible to pick the correct Nsts
1052                          * from VHT-SIG-A1.
1053                          *
1054                          * Bandwidth and SGI are valid so report the rateinfo
1055                          * on best-effort basis.
1056                          */
1057                         mcs = 0;
1058                         nss = 1;
1059                 }
1060
1061                 if (mcs > 0x09) {
1062                         ath10k_warn(ar, "invalid MCS received %u\n", mcs);
1063                         ath10k_warn(ar, "rxd %08x mpdu start %08x %08x msdu start %08x %08x ppdu start %08x %08x %08x %08x %08x\n",
1064                                     __le32_to_cpu(rxd_attention->flags),
1065                                     __le32_to_cpu(rxd_mpdu_start->info0),
1066                                     __le32_to_cpu(rxd_mpdu_start->info1),
1067                                     __le32_to_cpu(rxd_msdu_start_common->info0),
1068                                     __le32_to_cpu(rxd_msdu_start_common->info1),
1069                                     rxd_ppdu_start->info0,
1070                                     __le32_to_cpu(rxd_ppdu_start->info1),
1071                                     __le32_to_cpu(rxd_ppdu_start->info2),
1072                                     __le32_to_cpu(rxd_ppdu_start->info3),
1073                                     __le32_to_cpu(rxd_ppdu_start->info4));
1074
1075                         ath10k_warn(ar, "msdu end %08x mpdu end %08x\n",
1076                                     __le32_to_cpu(rxd_msdu_end_common->info0),
1077                                     __le32_to_cpu(rxd_mpdu_end->info0));
1078
1079                         ath10k_dbg_dump(ar, ATH10K_DBG_HTT_DUMP, NULL,
1080                                         "rx desc msdu payload: ",
1081                                         rxd_msdu_payload, 50);
1082                 }
1083
1084                 status->rate_idx = mcs;
1085                 status->nss = nss;
1086
1087                 if (sgi)
1088                         status->enc_flags |= RX_ENC_FLAG_SHORT_GI;
1089
1090                 status->bw = ath10k_bw_to_mac80211_bw(bw);
1091                 status->encoding = RX_ENC_VHT;
1092                 break;
1093         default:
1094                 break;
1095         }
1096 }
1097
1098 static struct ieee80211_channel *
1099 ath10k_htt_rx_h_peer_channel(struct ath10k *ar, struct htt_rx_desc *rxd)
1100 {
1101         struct ath10k_hw_params *hw = &ar->hw_params;
1102         struct rx_attention *rxd_attention;
1103         struct rx_msdu_end_common *rxd_msdu_end_common;
1104         struct rx_mpdu_start *rxd_mpdu_start;
1105         struct ath10k_peer *peer;
1106         struct ath10k_vif *arvif;
1107         struct cfg80211_chan_def def;
1108         u16 peer_id;
1109
1110         lockdep_assert_held(&ar->data_lock);
1111
1112         if (!rxd)
1113                 return NULL;
1114
1115         rxd_attention = ath10k_htt_rx_desc_get_attention(hw, rxd);
1116         rxd_msdu_end_common = ath10k_htt_rx_desc_get_msdu_end(hw, rxd);
1117         rxd_mpdu_start = ath10k_htt_rx_desc_get_mpdu_start(hw, rxd);
1118
1119         if (rxd_attention->flags &
1120             __cpu_to_le32(RX_ATTENTION_FLAGS_PEER_IDX_INVALID))
1121                 return NULL;
1122
1123         if (!(rxd_msdu_end_common->info0 &
1124               __cpu_to_le32(RX_MSDU_END_INFO0_FIRST_MSDU)))
1125                 return NULL;
1126
1127         peer_id = MS(__le32_to_cpu(rxd_mpdu_start->info0),
1128                      RX_MPDU_START_INFO0_PEER_IDX);
1129
1130         peer = ath10k_peer_find_by_id(ar, peer_id);
1131         if (!peer)
1132                 return NULL;
1133
1134         arvif = ath10k_get_arvif(ar, peer->vdev_id);
1135         if (WARN_ON_ONCE(!arvif))
1136                 return NULL;
1137
1138         if (ath10k_mac_vif_chan(arvif->vif, &def))
1139                 return NULL;
1140
1141         return def.chan;
1142 }
1143
1144 static struct ieee80211_channel *
1145 ath10k_htt_rx_h_vdev_channel(struct ath10k *ar, u32 vdev_id)
1146 {
1147         struct ath10k_vif *arvif;
1148         struct cfg80211_chan_def def;
1149
1150         lockdep_assert_held(&ar->data_lock);
1151
1152         list_for_each_entry(arvif, &ar->arvifs, list) {
1153                 if (arvif->vdev_id == vdev_id &&
1154                     ath10k_mac_vif_chan(arvif->vif, &def) == 0)
1155                         return def.chan;
1156         }
1157
1158         return NULL;
1159 }
1160
1161 static void
1162 ath10k_htt_rx_h_any_chan_iter(struct ieee80211_hw *hw,
1163                               struct ieee80211_chanctx_conf *conf,
1164                               void *data)
1165 {
1166         struct cfg80211_chan_def *def = data;
1167
1168         *def = conf->def;
1169 }
1170
1171 static struct ieee80211_channel *
1172 ath10k_htt_rx_h_any_channel(struct ath10k *ar)
1173 {
1174         struct cfg80211_chan_def def = {};
1175
1176         ieee80211_iter_chan_contexts_atomic(ar->hw,
1177                                             ath10k_htt_rx_h_any_chan_iter,
1178                                             &def);
1179
1180         return def.chan;
1181 }
1182
1183 static bool ath10k_htt_rx_h_channel(struct ath10k *ar,
1184                                     struct ieee80211_rx_status *status,
1185                                     struct htt_rx_desc *rxd,
1186                                     u32 vdev_id)
1187 {
1188         struct ieee80211_channel *ch;
1189
1190         spin_lock_bh(&ar->data_lock);
1191         ch = ar->scan_channel;
1192         if (!ch)
1193                 ch = ar->rx_channel;
1194         if (!ch)
1195                 ch = ath10k_htt_rx_h_peer_channel(ar, rxd);
1196         if (!ch)
1197                 ch = ath10k_htt_rx_h_vdev_channel(ar, vdev_id);
1198         if (!ch)
1199                 ch = ath10k_htt_rx_h_any_channel(ar);
1200         if (!ch)
1201                 ch = ar->tgt_oper_chan;
1202         spin_unlock_bh(&ar->data_lock);
1203
1204         if (!ch)
1205                 return false;
1206
1207         status->band = ch->band;
1208         status->freq = ch->center_freq;
1209
1210         return true;
1211 }
1212
1213 static void ath10k_htt_rx_h_signal(struct ath10k *ar,
1214                                    struct ieee80211_rx_status *status,
1215                                    struct htt_rx_desc *rxd)
1216 {
1217         struct ath10k_hw_params *hw = &ar->hw_params;
1218         struct rx_ppdu_start *rxd_ppdu_start = ath10k_htt_rx_desc_get_ppdu_start(hw, rxd);
1219         int i;
1220
1221         for (i = 0; i < IEEE80211_MAX_CHAINS ; i++) {
1222                 status->chains &= ~BIT(i);
1223
1224                 if (rxd_ppdu_start->rssi_chains[i].pri20_mhz != 0x80) {
1225                         status->chain_signal[i] = ATH10K_DEFAULT_NOISE_FLOOR +
1226                                 rxd_ppdu_start->rssi_chains[i].pri20_mhz;
1227
1228                         status->chains |= BIT(i);
1229                 }
1230         }
1231
1232         /* FIXME: Get real NF */
1233         status->signal = ATH10K_DEFAULT_NOISE_FLOOR +
1234                          rxd_ppdu_start->rssi_comb;
1235         status->flag &= ~RX_FLAG_NO_SIGNAL_VAL;
1236 }
1237
1238 static void ath10k_htt_rx_h_mactime(struct ath10k *ar,
1239                                     struct ieee80211_rx_status *status,
1240                                     struct htt_rx_desc *rxd)
1241 {
1242         struct ath10k_hw_params *hw = &ar->hw_params;
1243         struct rx_ppdu_end_common *rxd_ppdu_end_common;
1244
1245         rxd_ppdu_end_common = ath10k_htt_rx_desc_get_ppdu_end(hw, rxd);
1246
1247         /* FIXME: TSF is known only at the end of PPDU, in the last MPDU. This
1248          * means all prior MSDUs in a PPDU are reported to mac80211 without the
1249          * TSF. Is it worth holding frames until end of PPDU is known?
1250          *
1251          * FIXME: Can we get/compute 64bit TSF?
1252          */
1253         status->mactime = __le32_to_cpu(rxd_ppdu_end_common->tsf_timestamp);
1254         status->flag |= RX_FLAG_MACTIME_END;
1255 }
1256
1257 static void ath10k_htt_rx_h_ppdu(struct ath10k *ar,
1258                                  struct sk_buff_head *amsdu,
1259                                  struct ieee80211_rx_status *status,
1260                                  u32 vdev_id)
1261 {
1262         struct sk_buff *first;
1263         struct ath10k_hw_params *hw = &ar->hw_params;
1264         struct htt_rx_desc *rxd;
1265         struct rx_attention *rxd_attention;
1266         bool is_first_ppdu;
1267         bool is_last_ppdu;
1268
1269         if (skb_queue_empty(amsdu))
1270                 return;
1271
1272         first = skb_peek(amsdu);
1273         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
1274                                     (void *)first->data - hw->rx_desc_ops->rx_desc_size);
1275
1276         rxd_attention = ath10k_htt_rx_desc_get_attention(hw, rxd);
1277
1278         is_first_ppdu = !!(rxd_attention->flags &
1279                            __cpu_to_le32(RX_ATTENTION_FLAGS_FIRST_MPDU));
1280         is_last_ppdu = !!(rxd_attention->flags &
1281                           __cpu_to_le32(RX_ATTENTION_FLAGS_LAST_MPDU));
1282
1283         if (is_first_ppdu) {
1284                 /* New PPDU starts so clear out the old per-PPDU status. */
1285                 status->freq = 0;
1286                 status->rate_idx = 0;
1287                 status->nss = 0;
1288                 status->encoding = RX_ENC_LEGACY;
1289                 status->bw = RATE_INFO_BW_20;
1290
1291                 status->flag &= ~RX_FLAG_MACTIME_END;
1292                 status->flag |= RX_FLAG_NO_SIGNAL_VAL;
1293
1294                 status->flag &= ~(RX_FLAG_AMPDU_IS_LAST);
1295                 status->flag |= RX_FLAG_AMPDU_DETAILS | RX_FLAG_AMPDU_LAST_KNOWN;
1296                 status->ampdu_reference = ar->ampdu_reference;
1297
1298                 ath10k_htt_rx_h_signal(ar, status, rxd);
1299                 ath10k_htt_rx_h_channel(ar, status, rxd, vdev_id);
1300                 ath10k_htt_rx_h_rates(ar, status, rxd);
1301         }
1302
1303         if (is_last_ppdu) {
1304                 ath10k_htt_rx_h_mactime(ar, status, rxd);
1305
1306                 /* set ampdu last segment flag */
1307                 status->flag |= RX_FLAG_AMPDU_IS_LAST;
1308                 ar->ampdu_reference++;
1309         }
1310 }
1311
1312 static const char * const tid_to_ac[] = {
1313         "BE",
1314         "BK",
1315         "BK",
1316         "BE",
1317         "VI",
1318         "VI",
1319         "VO",
1320         "VO",
1321 };
1322
1323 static char *ath10k_get_tid(struct ieee80211_hdr *hdr, char *out, size_t size)
1324 {
1325         u8 *qc;
1326         int tid;
1327
1328         if (!ieee80211_is_data_qos(hdr->frame_control))
1329                 return "";
1330
1331         qc = ieee80211_get_qos_ctl(hdr);
1332         tid = *qc & IEEE80211_QOS_CTL_TID_MASK;
1333         if (tid < 8)
1334                 snprintf(out, size, "tid %d (%s)", tid, tid_to_ac[tid]);
1335         else
1336                 snprintf(out, size, "tid %d", tid);
1337
1338         return out;
1339 }
1340
1341 static void ath10k_htt_rx_h_queue_msdu(struct ath10k *ar,
1342                                        struct ieee80211_rx_status *rx_status,
1343                                        struct sk_buff *skb)
1344 {
1345         struct ieee80211_rx_status *status;
1346
1347         status = IEEE80211_SKB_RXCB(skb);
1348         *status = *rx_status;
1349
1350         skb_queue_tail(&ar->htt.rx_msdus_q, skb);
1351 }
1352
1353 static void ath10k_process_rx(struct ath10k *ar, struct sk_buff *skb)
1354 {
1355         struct ieee80211_rx_status *status;
1356         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
1357         char tid[32];
1358
1359         status = IEEE80211_SKB_RXCB(skb);
1360
1361         if (!(ar->filter_flags & FIF_FCSFAIL) &&
1362             status->flag & RX_FLAG_FAILED_FCS_CRC) {
1363                 ar->stats.rx_crc_err_drop++;
1364                 dev_kfree_skb_any(skb);
1365                 return;
1366         }
1367
1368         ath10k_dbg(ar, ATH10K_DBG_DATA,
1369                    "rx skb %pK len %u peer %pM %s %s sn %u %s%s%s%s%s%s %srate_idx %u vht_nss %u freq %u band %u flag 0x%x fcs-err %i mic-err %i amsdu-more %i\n",
1370                    skb,
1371                    skb->len,
1372                    ieee80211_get_SA(hdr),
1373                    ath10k_get_tid(hdr, tid, sizeof(tid)),
1374                    is_multicast_ether_addr(ieee80211_get_DA(hdr)) ?
1375                                                         "mcast" : "ucast",
1376                    (__le16_to_cpu(hdr->seq_ctrl) & IEEE80211_SCTL_SEQ) >> 4,
1377                    (status->encoding == RX_ENC_LEGACY) ? "legacy" : "",
1378                    (status->encoding == RX_ENC_HT) ? "ht" : "",
1379                    (status->encoding == RX_ENC_VHT) ? "vht" : "",
1380                    (status->bw == RATE_INFO_BW_40) ? "40" : "",
1381                    (status->bw == RATE_INFO_BW_80) ? "80" : "",
1382                    (status->bw == RATE_INFO_BW_160) ? "160" : "",
1383                    status->enc_flags & RX_ENC_FLAG_SHORT_GI ? "sgi " : "",
1384                    status->rate_idx,
1385                    status->nss,
1386                    status->freq,
1387                    status->band, status->flag,
1388                    !!(status->flag & RX_FLAG_FAILED_FCS_CRC),
1389                    !!(status->flag & RX_FLAG_MMIC_ERROR),
1390                    !!(status->flag & RX_FLAG_AMSDU_MORE));
1391         ath10k_dbg_dump(ar, ATH10K_DBG_HTT_DUMP, NULL, "rx skb: ",
1392                         skb->data, skb->len);
1393         trace_ath10k_rx_hdr(ar, skb->data, skb->len);
1394         trace_ath10k_rx_payload(ar, skb->data, skb->len);
1395
1396         ieee80211_rx_napi(ar->hw, NULL, skb, &ar->napi);
1397 }
1398
1399 static int ath10k_htt_rx_nwifi_hdrlen(struct ath10k *ar,
1400                                       struct ieee80211_hdr *hdr)
1401 {
1402         int len = ieee80211_hdrlen(hdr->frame_control);
1403
1404         if (!test_bit(ATH10K_FW_FEATURE_NO_NWIFI_DECAP_4ADDR_PADDING,
1405                       ar->running_fw->fw_file.fw_features))
1406                 len = round_up(len, 4);
1407
1408         return len;
1409 }
1410
1411 static void ath10k_htt_rx_h_undecap_raw(struct ath10k *ar,
1412                                         struct sk_buff *msdu,
1413                                         struct ieee80211_rx_status *status,
1414                                         enum htt_rx_mpdu_encrypt_type enctype,
1415                                         bool is_decrypted,
1416                                         const u8 first_hdr[64])
1417 {
1418         struct ieee80211_hdr *hdr;
1419         struct ath10k_hw_params *hw = &ar->hw_params;
1420         struct htt_rx_desc *rxd;
1421         struct rx_msdu_end_common *rxd_msdu_end_common;
1422         size_t hdr_len;
1423         size_t crypto_len;
1424         bool is_first;
1425         bool is_last;
1426         bool msdu_limit_err;
1427         int bytes_aligned = ar->hw_params.decap_align_bytes;
1428         u8 *qos;
1429
1430         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
1431                                     (void *)msdu->data - hw->rx_desc_ops->rx_desc_size);
1432
1433         rxd_msdu_end_common = ath10k_htt_rx_desc_get_msdu_end(hw, rxd);
1434         is_first = !!(rxd_msdu_end_common->info0 &
1435                       __cpu_to_le32(RX_MSDU_END_INFO0_FIRST_MSDU));
1436         is_last = !!(rxd_msdu_end_common->info0 &
1437                      __cpu_to_le32(RX_MSDU_END_INFO0_LAST_MSDU));
1438
1439         /* Delivered decapped frame:
1440          * [802.11 header]
1441          * [crypto param] <-- can be trimmed if !fcs_err &&
1442          *                    !decrypt_err && !peer_idx_invalid
1443          * [amsdu header] <-- only if A-MSDU
1444          * [rfc1042/llc]
1445          * [payload]
1446          * [FCS] <-- at end, needs to be trimmed
1447          */
1448
1449         /* Some hardwares(QCA99x0 variants) limit number of msdus in a-msdu when
1450          * deaggregate, so that unwanted MSDU-deaggregation is avoided for
1451          * error packets. If limit exceeds, hw sends all remaining MSDUs as
1452          * a single last MSDU with this msdu limit error set.
1453          */
1454         msdu_limit_err = ath10k_htt_rx_desc_msdu_limit_error(hw, rxd);
1455
1456         /* If MSDU limit error happens, then don't warn on, the partial raw MSDU
1457          * without first MSDU is expected in that case, and handled later here.
1458          */
1459         /* This probably shouldn't happen but warn just in case */
1460         if (WARN_ON_ONCE(!is_first && !msdu_limit_err))
1461                 return;
1462
1463         /* This probably shouldn't happen but warn just in case */
1464         if (WARN_ON_ONCE(!(is_first && is_last) && !msdu_limit_err))
1465                 return;
1466
1467         skb_trim(msdu, msdu->len - FCS_LEN);
1468
1469         /* Push original 80211 header */
1470         if (unlikely(msdu_limit_err)) {
1471                 hdr = (struct ieee80211_hdr *)first_hdr;
1472                 hdr_len = ieee80211_hdrlen(hdr->frame_control);
1473                 crypto_len = ath10k_htt_rx_crypto_param_len(ar, enctype);
1474
1475                 if (ieee80211_is_data_qos(hdr->frame_control)) {
1476                         qos = ieee80211_get_qos_ctl(hdr);
1477                         qos[0] |= IEEE80211_QOS_CTL_A_MSDU_PRESENT;
1478                 }
1479
1480                 if (crypto_len)
1481                         memcpy(skb_push(msdu, crypto_len),
1482                                (void *)hdr + round_up(hdr_len, bytes_aligned),
1483                                crypto_len);
1484
1485                 memcpy(skb_push(msdu, hdr_len), hdr, hdr_len);
1486         }
1487
1488         /* In most cases this will be true for sniffed frames. It makes sense
1489          * to deliver them as-is without stripping the crypto param. This is
1490          * necessary for software based decryption.
1491          *
1492          * If there's no error then the frame is decrypted. At least that is
1493          * the case for frames that come in via fragmented rx indication.
1494          */
1495         if (!is_decrypted)
1496                 return;
1497
1498         /* The payload is decrypted so strip crypto params. Start from tail
1499          * since hdr is used to compute some stuff.
1500          */
1501
1502         hdr = (void *)msdu->data;
1503
1504         /* Tail */
1505         if (status->flag & RX_FLAG_IV_STRIPPED) {
1506                 skb_trim(msdu, msdu->len -
1507                          ath10k_htt_rx_crypto_mic_len(ar, enctype));
1508
1509                 skb_trim(msdu, msdu->len -
1510                          ath10k_htt_rx_crypto_icv_len(ar, enctype));
1511         } else {
1512                 /* MIC */
1513                 if (status->flag & RX_FLAG_MIC_STRIPPED)
1514                         skb_trim(msdu, msdu->len -
1515                                  ath10k_htt_rx_crypto_mic_len(ar, enctype));
1516
1517                 /* ICV */
1518                 if (status->flag & RX_FLAG_ICV_STRIPPED)
1519                         skb_trim(msdu, msdu->len -
1520                                  ath10k_htt_rx_crypto_icv_len(ar, enctype));
1521         }
1522
1523         /* MMIC */
1524         if ((status->flag & RX_FLAG_MMIC_STRIPPED) &&
1525             !ieee80211_has_morefrags(hdr->frame_control) &&
1526             enctype == HTT_RX_MPDU_ENCRYPT_TKIP_WPA)
1527                 skb_trim(msdu, msdu->len - MICHAEL_MIC_LEN);
1528
1529         /* Head */
1530         if (status->flag & RX_FLAG_IV_STRIPPED) {
1531                 hdr_len = ieee80211_hdrlen(hdr->frame_control);
1532                 crypto_len = ath10k_htt_rx_crypto_param_len(ar, enctype);
1533
1534                 memmove((void *)msdu->data + crypto_len,
1535                         (void *)msdu->data, hdr_len);
1536                 skb_pull(msdu, crypto_len);
1537         }
1538 }
1539
1540 static void ath10k_htt_rx_h_undecap_nwifi(struct ath10k *ar,
1541                                           struct sk_buff *msdu,
1542                                           struct ieee80211_rx_status *status,
1543                                           const u8 first_hdr[64],
1544                                           enum htt_rx_mpdu_encrypt_type enctype)
1545 {
1546         struct ath10k_hw_params *hw = &ar->hw_params;
1547         struct ieee80211_hdr *hdr;
1548         struct htt_rx_desc *rxd;
1549         size_t hdr_len;
1550         u8 da[ETH_ALEN];
1551         u8 sa[ETH_ALEN];
1552         int l3_pad_bytes;
1553         int bytes_aligned = ar->hw_params.decap_align_bytes;
1554
1555         /* Delivered decapped frame:
1556          * [nwifi 802.11 header] <-- replaced with 802.11 hdr
1557          * [rfc1042/llc]
1558          *
1559          * Note: The nwifi header doesn't have QoS Control and is
1560          * (always?) a 3addr frame.
1561          *
1562          * Note2: There's no A-MSDU subframe header. Even if it's part
1563          * of an A-MSDU.
1564          */
1565
1566         /* pull decapped header and copy SA & DA */
1567         rxd = HTT_RX_BUF_TO_RX_DESC(hw, (void *)msdu->data -
1568                                     hw->rx_desc_ops->rx_desc_size);
1569
1570         l3_pad_bytes = ath10k_htt_rx_desc_get_l3_pad_bytes(&ar->hw_params, rxd);
1571         skb_put(msdu, l3_pad_bytes);
1572
1573         hdr = (struct ieee80211_hdr *)(msdu->data + l3_pad_bytes);
1574
1575         hdr_len = ath10k_htt_rx_nwifi_hdrlen(ar, hdr);
1576         ether_addr_copy(da, ieee80211_get_DA(hdr));
1577         ether_addr_copy(sa, ieee80211_get_SA(hdr));
1578         skb_pull(msdu, hdr_len);
1579
1580         /* push original 802.11 header */
1581         hdr = (struct ieee80211_hdr *)first_hdr;
1582         hdr_len = ieee80211_hdrlen(hdr->frame_control);
1583
1584         if (!(status->flag & RX_FLAG_IV_STRIPPED)) {
1585                 memcpy(skb_push(msdu,
1586                                 ath10k_htt_rx_crypto_param_len(ar, enctype)),
1587                        (void *)hdr + round_up(hdr_len, bytes_aligned),
1588                         ath10k_htt_rx_crypto_param_len(ar, enctype));
1589         }
1590
1591         memcpy(skb_push(msdu, hdr_len), hdr, hdr_len);
1592
1593         /* original 802.11 header has a different DA and in
1594          * case of 4addr it may also have different SA
1595          */
1596         hdr = (struct ieee80211_hdr *)msdu->data;
1597         ether_addr_copy(ieee80211_get_DA(hdr), da);
1598         ether_addr_copy(ieee80211_get_SA(hdr), sa);
1599 }
1600
1601 static void *ath10k_htt_rx_h_find_rfc1042(struct ath10k *ar,
1602                                           struct sk_buff *msdu,
1603                                           enum htt_rx_mpdu_encrypt_type enctype)
1604 {
1605         struct ieee80211_hdr *hdr;
1606         struct ath10k_hw_params *hw = &ar->hw_params;
1607         struct htt_rx_desc *rxd;
1608         struct rx_msdu_end_common *rxd_msdu_end_common;
1609         u8 *rxd_rx_hdr_status;
1610         size_t hdr_len, crypto_len;
1611         void *rfc1042;
1612         bool is_first, is_last, is_amsdu;
1613         int bytes_aligned = ar->hw_params.decap_align_bytes;
1614
1615         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
1616                                     (void *)msdu->data - hw->rx_desc_ops->rx_desc_size);
1617
1618         rxd_msdu_end_common = ath10k_htt_rx_desc_get_msdu_end(hw, rxd);
1619         rxd_rx_hdr_status = ath10k_htt_rx_desc_get_rx_hdr_status(hw, rxd);
1620         hdr = (void *)rxd_rx_hdr_status;
1621
1622         is_first = !!(rxd_msdu_end_common->info0 &
1623                       __cpu_to_le32(RX_MSDU_END_INFO0_FIRST_MSDU));
1624         is_last = !!(rxd_msdu_end_common->info0 &
1625                      __cpu_to_le32(RX_MSDU_END_INFO0_LAST_MSDU));
1626         is_amsdu = !(is_first && is_last);
1627
1628         rfc1042 = hdr;
1629
1630         if (is_first) {
1631                 hdr_len = ieee80211_hdrlen(hdr->frame_control);
1632                 crypto_len = ath10k_htt_rx_crypto_param_len(ar, enctype);
1633
1634                 rfc1042 += round_up(hdr_len, bytes_aligned) +
1635                            round_up(crypto_len, bytes_aligned);
1636         }
1637
1638         if (is_amsdu)
1639                 rfc1042 += sizeof(struct amsdu_subframe_hdr);
1640
1641         return rfc1042;
1642 }
1643
1644 static void ath10k_htt_rx_h_undecap_eth(struct ath10k *ar,
1645                                         struct sk_buff *msdu,
1646                                         struct ieee80211_rx_status *status,
1647                                         const u8 first_hdr[64],
1648                                         enum htt_rx_mpdu_encrypt_type enctype)
1649 {
1650         struct ath10k_hw_params *hw = &ar->hw_params;
1651         struct ieee80211_hdr *hdr;
1652         struct ethhdr *eth;
1653         size_t hdr_len;
1654         void *rfc1042;
1655         u8 da[ETH_ALEN];
1656         u8 sa[ETH_ALEN];
1657         int l3_pad_bytes;
1658         struct htt_rx_desc *rxd;
1659         int bytes_aligned = ar->hw_params.decap_align_bytes;
1660
1661         /* Delivered decapped frame:
1662          * [eth header] <-- replaced with 802.11 hdr & rfc1042/llc
1663          * [payload]
1664          */
1665
1666         rfc1042 = ath10k_htt_rx_h_find_rfc1042(ar, msdu, enctype);
1667         if (WARN_ON_ONCE(!rfc1042))
1668                 return;
1669
1670         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
1671                                     (void *)msdu->data - hw->rx_desc_ops->rx_desc_size);
1672
1673         l3_pad_bytes = ath10k_htt_rx_desc_get_l3_pad_bytes(&ar->hw_params, rxd);
1674         skb_put(msdu, l3_pad_bytes);
1675         skb_pull(msdu, l3_pad_bytes);
1676
1677         /* pull decapped header and copy SA & DA */
1678         eth = (struct ethhdr *)msdu->data;
1679         ether_addr_copy(da, eth->h_dest);
1680         ether_addr_copy(sa, eth->h_source);
1681         skb_pull(msdu, sizeof(struct ethhdr));
1682
1683         /* push rfc1042/llc/snap */
1684         memcpy(skb_push(msdu, sizeof(struct rfc1042_hdr)), rfc1042,
1685                sizeof(struct rfc1042_hdr));
1686
1687         /* push original 802.11 header */
1688         hdr = (struct ieee80211_hdr *)first_hdr;
1689         hdr_len = ieee80211_hdrlen(hdr->frame_control);
1690
1691         if (!(status->flag & RX_FLAG_IV_STRIPPED)) {
1692                 memcpy(skb_push(msdu,
1693                                 ath10k_htt_rx_crypto_param_len(ar, enctype)),
1694                        (void *)hdr + round_up(hdr_len, bytes_aligned),
1695                         ath10k_htt_rx_crypto_param_len(ar, enctype));
1696         }
1697
1698         memcpy(skb_push(msdu, hdr_len), hdr, hdr_len);
1699
1700         /* original 802.11 header has a different DA and in
1701          * case of 4addr it may also have different SA
1702          */
1703         hdr = (struct ieee80211_hdr *)msdu->data;
1704         ether_addr_copy(ieee80211_get_DA(hdr), da);
1705         ether_addr_copy(ieee80211_get_SA(hdr), sa);
1706 }
1707
1708 static void ath10k_htt_rx_h_undecap_snap(struct ath10k *ar,
1709                                          struct sk_buff *msdu,
1710                                          struct ieee80211_rx_status *status,
1711                                          const u8 first_hdr[64],
1712                                          enum htt_rx_mpdu_encrypt_type enctype)
1713 {
1714         struct ath10k_hw_params *hw = &ar->hw_params;
1715         struct ieee80211_hdr *hdr;
1716         size_t hdr_len;
1717         int l3_pad_bytes;
1718         struct htt_rx_desc *rxd;
1719         int bytes_aligned = ar->hw_params.decap_align_bytes;
1720
1721         /* Delivered decapped frame:
1722          * [amsdu header] <-- replaced with 802.11 hdr
1723          * [rfc1042/llc]
1724          * [payload]
1725          */
1726
1727         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
1728                                     (void *)msdu->data - hw->rx_desc_ops->rx_desc_size);
1729
1730         l3_pad_bytes = ath10k_htt_rx_desc_get_l3_pad_bytes(&ar->hw_params, rxd);
1731
1732         skb_put(msdu, l3_pad_bytes);
1733         skb_pull(msdu, sizeof(struct amsdu_subframe_hdr) + l3_pad_bytes);
1734
1735         hdr = (struct ieee80211_hdr *)first_hdr;
1736         hdr_len = ieee80211_hdrlen(hdr->frame_control);
1737
1738         if (!(status->flag & RX_FLAG_IV_STRIPPED)) {
1739                 memcpy(skb_push(msdu,
1740                                 ath10k_htt_rx_crypto_param_len(ar, enctype)),
1741                        (void *)hdr + round_up(hdr_len, bytes_aligned),
1742                         ath10k_htt_rx_crypto_param_len(ar, enctype));
1743         }
1744
1745         memcpy(skb_push(msdu, hdr_len), hdr, hdr_len);
1746 }
1747
1748 static void ath10k_htt_rx_h_undecap(struct ath10k *ar,
1749                                     struct sk_buff *msdu,
1750                                     struct ieee80211_rx_status *status,
1751                                     u8 first_hdr[64],
1752                                     enum htt_rx_mpdu_encrypt_type enctype,
1753                                     bool is_decrypted)
1754 {
1755         struct ath10k_hw_params *hw = &ar->hw_params;
1756         struct htt_rx_desc *rxd;
1757         struct rx_msdu_start_common *rxd_msdu_start_common;
1758         enum rx_msdu_decap_format decap;
1759
1760         /* First msdu's decapped header:
1761          * [802.11 header] <-- padded to 4 bytes long
1762          * [crypto param] <-- padded to 4 bytes long
1763          * [amsdu header] <-- only if A-MSDU
1764          * [rfc1042/llc]
1765          *
1766          * Other (2nd, 3rd, ..) msdu's decapped header:
1767          * [amsdu header] <-- only if A-MSDU
1768          * [rfc1042/llc]
1769          */
1770
1771         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
1772                                     (void *)msdu->data - hw->rx_desc_ops->rx_desc_size);
1773
1774         rxd_msdu_start_common = ath10k_htt_rx_desc_get_msdu_start(hw, rxd);
1775         decap = MS(__le32_to_cpu(rxd_msdu_start_common->info1),
1776                    RX_MSDU_START_INFO1_DECAP_FORMAT);
1777
1778         switch (decap) {
1779         case RX_MSDU_DECAP_RAW:
1780                 ath10k_htt_rx_h_undecap_raw(ar, msdu, status, enctype,
1781                                             is_decrypted, first_hdr);
1782                 break;
1783         case RX_MSDU_DECAP_NATIVE_WIFI:
1784                 ath10k_htt_rx_h_undecap_nwifi(ar, msdu, status, first_hdr,
1785                                               enctype);
1786                 break;
1787         case RX_MSDU_DECAP_ETHERNET2_DIX:
1788                 ath10k_htt_rx_h_undecap_eth(ar, msdu, status, first_hdr, enctype);
1789                 break;
1790         case RX_MSDU_DECAP_8023_SNAP_LLC:
1791                 ath10k_htt_rx_h_undecap_snap(ar, msdu, status, first_hdr,
1792                                              enctype);
1793                 break;
1794         }
1795 }
1796
1797 static int ath10k_htt_rx_get_csum_state(struct ath10k_hw_params *hw, struct sk_buff *skb)
1798 {
1799         struct htt_rx_desc *rxd;
1800         struct rx_attention *rxd_attention;
1801         struct rx_msdu_start_common *rxd_msdu_start_common;
1802         u32 flags, info;
1803         bool is_ip4, is_ip6;
1804         bool is_tcp, is_udp;
1805         bool ip_csum_ok, tcpudp_csum_ok;
1806
1807         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
1808                                     (void *)skb->data - hw->rx_desc_ops->rx_desc_size);
1809
1810         rxd_attention = ath10k_htt_rx_desc_get_attention(hw, rxd);
1811         rxd_msdu_start_common = ath10k_htt_rx_desc_get_msdu_start(hw, rxd);
1812         flags = __le32_to_cpu(rxd_attention->flags);
1813         info = __le32_to_cpu(rxd_msdu_start_common->info1);
1814
1815         is_ip4 = !!(info & RX_MSDU_START_INFO1_IPV4_PROTO);
1816         is_ip6 = !!(info & RX_MSDU_START_INFO1_IPV6_PROTO);
1817         is_tcp = !!(info & RX_MSDU_START_INFO1_TCP_PROTO);
1818         is_udp = !!(info & RX_MSDU_START_INFO1_UDP_PROTO);
1819         ip_csum_ok = !(flags & RX_ATTENTION_FLAGS_IP_CHKSUM_FAIL);
1820         tcpudp_csum_ok = !(flags & RX_ATTENTION_FLAGS_TCP_UDP_CHKSUM_FAIL);
1821
1822         if (!is_ip4 && !is_ip6)
1823                 return CHECKSUM_NONE;
1824         if (!is_tcp && !is_udp)
1825                 return CHECKSUM_NONE;
1826         if (!ip_csum_ok)
1827                 return CHECKSUM_NONE;
1828         if (!tcpudp_csum_ok)
1829                 return CHECKSUM_NONE;
1830
1831         return CHECKSUM_UNNECESSARY;
1832 }
1833
1834 static void ath10k_htt_rx_h_csum_offload(struct ath10k_hw_params *hw,
1835                                          struct sk_buff *msdu)
1836 {
1837         msdu->ip_summed = ath10k_htt_rx_get_csum_state(hw, msdu);
1838 }
1839
1840 static u64 ath10k_htt_rx_h_get_pn(struct ath10k *ar, struct sk_buff *skb,
1841                                   u16 offset,
1842                                   enum htt_rx_mpdu_encrypt_type enctype)
1843 {
1844         struct ieee80211_hdr *hdr;
1845         u64 pn = 0;
1846         u8 *ehdr;
1847
1848         hdr = (struct ieee80211_hdr *)(skb->data + offset);
1849         ehdr = skb->data + offset + ieee80211_hdrlen(hdr->frame_control);
1850
1851         if (enctype == HTT_RX_MPDU_ENCRYPT_AES_CCM_WPA2) {
1852                 pn = ehdr[0];
1853                 pn |= (u64)ehdr[1] << 8;
1854                 pn |= (u64)ehdr[4] << 16;
1855                 pn |= (u64)ehdr[5] << 24;
1856                 pn |= (u64)ehdr[6] << 32;
1857                 pn |= (u64)ehdr[7] << 40;
1858         }
1859         return pn;
1860 }
1861
1862 static bool ath10k_htt_rx_h_frag_multicast_check(struct ath10k *ar,
1863                                                  struct sk_buff *skb,
1864                                                  u16 offset)
1865 {
1866         struct ieee80211_hdr *hdr;
1867
1868         hdr = (struct ieee80211_hdr *)(skb->data + offset);
1869         return !is_multicast_ether_addr(hdr->addr1);
1870 }
1871
1872 static bool ath10k_htt_rx_h_frag_pn_check(struct ath10k *ar,
1873                                           struct sk_buff *skb,
1874                                           u16 peer_id,
1875                                           u16 offset,
1876                                           enum htt_rx_mpdu_encrypt_type enctype)
1877 {
1878         struct ath10k_peer *peer;
1879         union htt_rx_pn_t *last_pn, new_pn = {0};
1880         struct ieee80211_hdr *hdr;
1881         u8 tid, frag_number;
1882         u32 seq;
1883
1884         peer = ath10k_peer_find_by_id(ar, peer_id);
1885         if (!peer) {
1886                 ath10k_dbg(ar, ATH10K_DBG_HTT, "invalid peer for frag pn check\n");
1887                 return false;
1888         }
1889
1890         hdr = (struct ieee80211_hdr *)(skb->data + offset);
1891         if (ieee80211_is_data_qos(hdr->frame_control))
1892                 tid = ieee80211_get_tid(hdr);
1893         else
1894                 tid = ATH10K_TXRX_NON_QOS_TID;
1895
1896         last_pn = &peer->frag_tids_last_pn[tid];
1897         new_pn.pn48 = ath10k_htt_rx_h_get_pn(ar, skb, offset, enctype);
1898         frag_number = le16_to_cpu(hdr->seq_ctrl) & IEEE80211_SCTL_FRAG;
1899         seq = (__le16_to_cpu(hdr->seq_ctrl) & IEEE80211_SCTL_SEQ) >> 4;
1900
1901         if (frag_number == 0) {
1902                 last_pn->pn48 = new_pn.pn48;
1903                 peer->frag_tids_seq[tid] = seq;
1904         } else {
1905                 if (seq != peer->frag_tids_seq[tid])
1906                         return false;
1907
1908                 if (new_pn.pn48 != last_pn->pn48 + 1)
1909                         return false;
1910
1911                 last_pn->pn48 = new_pn.pn48;
1912         }
1913
1914         return true;
1915 }
1916
1917 static void ath10k_htt_rx_h_mpdu(struct ath10k *ar,
1918                                  struct sk_buff_head *amsdu,
1919                                  struct ieee80211_rx_status *status,
1920                                  bool fill_crypt_header,
1921                                  u8 *rx_hdr,
1922                                  enum ath10k_pkt_rx_err *err,
1923                                  u16 peer_id,
1924                                  bool frag)
1925 {
1926         struct sk_buff *first;
1927         struct sk_buff *last;
1928         struct sk_buff *msdu, *temp;
1929         struct ath10k_hw_params *hw = &ar->hw_params;
1930         struct htt_rx_desc *rxd;
1931         struct rx_attention *rxd_attention;
1932         struct rx_mpdu_start *rxd_mpdu_start;
1933
1934         struct ieee80211_hdr *hdr;
1935         enum htt_rx_mpdu_encrypt_type enctype;
1936         u8 first_hdr[64];
1937         u8 *qos;
1938         bool has_fcs_err;
1939         bool has_crypto_err;
1940         bool has_tkip_err;
1941         bool has_peer_idx_invalid;
1942         bool is_decrypted;
1943         bool is_mgmt;
1944         u32 attention;
1945         bool frag_pn_check = true, multicast_check = true;
1946
1947         if (skb_queue_empty(amsdu))
1948                 return;
1949
1950         first = skb_peek(amsdu);
1951         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
1952                                     (void *)first->data - hw->rx_desc_ops->rx_desc_size);
1953
1954         rxd_attention = ath10k_htt_rx_desc_get_attention(hw, rxd);
1955         rxd_mpdu_start = ath10k_htt_rx_desc_get_mpdu_start(hw, rxd);
1956
1957         is_mgmt = !!(rxd_attention->flags &
1958                      __cpu_to_le32(RX_ATTENTION_FLAGS_MGMT_TYPE));
1959
1960         enctype = MS(__le32_to_cpu(rxd_mpdu_start->info0),
1961                      RX_MPDU_START_INFO0_ENCRYPT_TYPE);
1962
1963         /* First MSDU's Rx descriptor in an A-MSDU contains full 802.11
1964          * decapped header. It'll be used for undecapping of each MSDU.
1965          */
1966         hdr = (void *)ath10k_htt_rx_desc_get_rx_hdr_status(hw, rxd);
1967         memcpy(first_hdr, hdr, RX_HTT_HDR_STATUS_LEN);
1968
1969         if (rx_hdr)
1970                 memcpy(rx_hdr, hdr, RX_HTT_HDR_STATUS_LEN);
1971
1972         /* Each A-MSDU subframe will use the original header as the base and be
1973          * reported as a separate MSDU so strip the A-MSDU bit from QoS Ctl.
1974          */
1975         hdr = (void *)first_hdr;
1976
1977         if (ieee80211_is_data_qos(hdr->frame_control)) {
1978                 qos = ieee80211_get_qos_ctl(hdr);
1979                 qos[0] &= ~IEEE80211_QOS_CTL_A_MSDU_PRESENT;
1980         }
1981
1982         /* Some attention flags are valid only in the last MSDU. */
1983         last = skb_peek_tail(amsdu);
1984         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
1985                                     (void *)last->data - hw->rx_desc_ops->rx_desc_size);
1986
1987         rxd_attention = ath10k_htt_rx_desc_get_attention(hw, rxd);
1988         attention = __le32_to_cpu(rxd_attention->flags);
1989
1990         has_fcs_err = !!(attention & RX_ATTENTION_FLAGS_FCS_ERR);
1991         has_crypto_err = !!(attention & RX_ATTENTION_FLAGS_DECRYPT_ERR);
1992         has_tkip_err = !!(attention & RX_ATTENTION_FLAGS_TKIP_MIC_ERR);
1993         has_peer_idx_invalid = !!(attention & RX_ATTENTION_FLAGS_PEER_IDX_INVALID);
1994
1995         /* Note: If hardware captures an encrypted frame that it can't decrypt,
1996          * e.g. due to fcs error, missing peer or invalid key data it will
1997          * report the frame as raw.
1998          */
1999         is_decrypted = (enctype != HTT_RX_MPDU_ENCRYPT_NONE &&
2000                         !has_fcs_err &&
2001                         !has_crypto_err &&
2002                         !has_peer_idx_invalid);
2003
2004         /* Clear per-MPDU flags while leaving per-PPDU flags intact. */
2005         status->flag &= ~(RX_FLAG_FAILED_FCS_CRC |
2006                           RX_FLAG_MMIC_ERROR |
2007                           RX_FLAG_DECRYPTED |
2008                           RX_FLAG_IV_STRIPPED |
2009                           RX_FLAG_ONLY_MONITOR |
2010                           RX_FLAG_MMIC_STRIPPED);
2011
2012         if (has_fcs_err)
2013                 status->flag |= RX_FLAG_FAILED_FCS_CRC;
2014
2015         if (has_tkip_err)
2016                 status->flag |= RX_FLAG_MMIC_ERROR;
2017
2018         if (err) {
2019                 if (has_fcs_err)
2020                         *err = ATH10K_PKT_RX_ERR_FCS;
2021                 else if (has_tkip_err)
2022                         *err = ATH10K_PKT_RX_ERR_TKIP;
2023                 else if (has_crypto_err)
2024                         *err = ATH10K_PKT_RX_ERR_CRYPT;
2025                 else if (has_peer_idx_invalid)
2026                         *err = ATH10K_PKT_RX_ERR_PEER_IDX_INVAL;
2027         }
2028
2029         /* Firmware reports all necessary management frames via WMI already.
2030          * They are not reported to monitor interfaces at all so pass the ones
2031          * coming via HTT to monitor interfaces instead. This simplifies
2032          * matters a lot.
2033          */
2034         if (is_mgmt)
2035                 status->flag |= RX_FLAG_ONLY_MONITOR;
2036
2037         if (is_decrypted) {
2038                 status->flag |= RX_FLAG_DECRYPTED;
2039
2040                 if (likely(!is_mgmt))
2041                         status->flag |= RX_FLAG_MMIC_STRIPPED;
2042
2043                 if (fill_crypt_header)
2044                         status->flag |= RX_FLAG_MIC_STRIPPED |
2045                                         RX_FLAG_ICV_STRIPPED;
2046                 else
2047                         status->flag |= RX_FLAG_IV_STRIPPED;
2048         }
2049
2050         skb_queue_walk(amsdu, msdu) {
2051                 if (frag && !fill_crypt_header && is_decrypted &&
2052                     enctype == HTT_RX_MPDU_ENCRYPT_AES_CCM_WPA2)
2053                         frag_pn_check = ath10k_htt_rx_h_frag_pn_check(ar,
2054                                                                       msdu,
2055                                                                       peer_id,
2056                                                                       0,
2057                                                                       enctype);
2058
2059                 if (frag)
2060                         multicast_check = ath10k_htt_rx_h_frag_multicast_check(ar,
2061                                                                                msdu,
2062                                                                                0);
2063
2064                 if (!frag_pn_check || !multicast_check) {
2065                         /* Discard the fragment with invalid PN or multicast DA
2066                          */
2067                         temp = msdu->prev;
2068                         __skb_unlink(msdu, amsdu);
2069                         dev_kfree_skb_any(msdu);
2070                         msdu = temp;
2071                         frag_pn_check = true;
2072                         multicast_check = true;
2073                         continue;
2074                 }
2075
2076                 ath10k_htt_rx_h_csum_offload(&ar->hw_params, msdu);
2077
2078                 if (frag && !fill_crypt_header &&
2079                     enctype == HTT_RX_MPDU_ENCRYPT_TKIP_WPA)
2080                         status->flag &= ~RX_FLAG_MMIC_STRIPPED;
2081
2082                 ath10k_htt_rx_h_undecap(ar, msdu, status, first_hdr, enctype,
2083                                         is_decrypted);
2084
2085                 /* Undecapping involves copying the original 802.11 header back
2086                  * to sk_buff. If frame is protected and hardware has decrypted
2087                  * it then remove the protected bit.
2088                  */
2089                 if (!is_decrypted)
2090                         continue;
2091                 if (is_mgmt)
2092                         continue;
2093
2094                 if (fill_crypt_header)
2095                         continue;
2096
2097                 hdr = (void *)msdu->data;
2098                 hdr->frame_control &= ~__cpu_to_le16(IEEE80211_FCTL_PROTECTED);
2099
2100                 if (frag && !fill_crypt_header &&
2101                     enctype == HTT_RX_MPDU_ENCRYPT_TKIP_WPA)
2102                         status->flag &= ~RX_FLAG_IV_STRIPPED &
2103                                         ~RX_FLAG_MMIC_STRIPPED;
2104         }
2105 }
2106
2107 static void ath10k_htt_rx_h_enqueue(struct ath10k *ar,
2108                                     struct sk_buff_head *amsdu,
2109                                     struct ieee80211_rx_status *status)
2110 {
2111         struct sk_buff *msdu;
2112         struct sk_buff *first_subframe;
2113
2114         first_subframe = skb_peek(amsdu);
2115
2116         while ((msdu = __skb_dequeue(amsdu))) {
2117                 /* Setup per-MSDU flags */
2118                 if (skb_queue_empty(amsdu))
2119                         status->flag &= ~RX_FLAG_AMSDU_MORE;
2120                 else
2121                         status->flag |= RX_FLAG_AMSDU_MORE;
2122
2123                 if (msdu == first_subframe) {
2124                         first_subframe = NULL;
2125                         status->flag &= ~RX_FLAG_ALLOW_SAME_PN;
2126                 } else {
2127                         status->flag |= RX_FLAG_ALLOW_SAME_PN;
2128                 }
2129
2130                 ath10k_htt_rx_h_queue_msdu(ar, status, msdu);
2131         }
2132 }
2133
2134 static int ath10k_unchain_msdu(struct sk_buff_head *amsdu,
2135                                unsigned long *unchain_cnt)
2136 {
2137         struct sk_buff *skb, *first;
2138         int space;
2139         int total_len = 0;
2140         int amsdu_len = skb_queue_len(amsdu);
2141
2142         /* TODO:  Might could optimize this by using
2143          * skb_try_coalesce or similar method to
2144          * decrease copying, or maybe get mac80211 to
2145          * provide a way to just receive a list of
2146          * skb?
2147          */
2148
2149         first = __skb_dequeue(amsdu);
2150
2151         /* Allocate total length all at once. */
2152         skb_queue_walk(amsdu, skb)
2153                 total_len += skb->len;
2154
2155         space = total_len - skb_tailroom(first);
2156         if ((space > 0) &&
2157             (pskb_expand_head(first, 0, space, GFP_ATOMIC) < 0)) {
2158                 /* TODO:  bump some rx-oom error stat */
2159                 /* put it back together so we can free the
2160                  * whole list at once.
2161                  */
2162                 __skb_queue_head(amsdu, first);
2163                 return -1;
2164         }
2165
2166         /* Walk list again, copying contents into
2167          * msdu_head
2168          */
2169         while ((skb = __skb_dequeue(amsdu))) {
2170                 skb_copy_from_linear_data(skb, skb_put(first, skb->len),
2171                                           skb->len);
2172                 dev_kfree_skb_any(skb);
2173         }
2174
2175         __skb_queue_head(amsdu, first);
2176
2177         *unchain_cnt += amsdu_len - 1;
2178
2179         return 0;
2180 }
2181
2182 static void ath10k_htt_rx_h_unchain(struct ath10k *ar,
2183                                     struct sk_buff_head *amsdu,
2184                                     unsigned long *drop_cnt,
2185                                     unsigned long *unchain_cnt)
2186 {
2187         struct sk_buff *first;
2188         struct ath10k_hw_params *hw = &ar->hw_params;
2189         struct htt_rx_desc *rxd;
2190         struct rx_msdu_start_common *rxd_msdu_start_common;
2191         struct rx_frag_info_common *rxd_frag_info;
2192         enum rx_msdu_decap_format decap;
2193
2194         first = skb_peek(amsdu);
2195         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
2196                                     (void *)first->data - hw->rx_desc_ops->rx_desc_size);
2197
2198         rxd_msdu_start_common = ath10k_htt_rx_desc_get_msdu_start(hw, rxd);
2199         rxd_frag_info = ath10k_htt_rx_desc_get_frag_info(hw, rxd);
2200         decap = MS(__le32_to_cpu(rxd_msdu_start_common->info1),
2201                    RX_MSDU_START_INFO1_DECAP_FORMAT);
2202
2203         /* FIXME: Current unchaining logic can only handle simple case of raw
2204          * msdu chaining. If decapping is other than raw the chaining may be
2205          * more complex and this isn't handled by the current code. Don't even
2206          * try re-constructing such frames - it'll be pretty much garbage.
2207          */
2208         if (decap != RX_MSDU_DECAP_RAW ||
2209             skb_queue_len(amsdu) != 1 + rxd_frag_info->ring2_more_count) {
2210                 *drop_cnt += skb_queue_len(amsdu);
2211                 __skb_queue_purge(amsdu);
2212                 return;
2213         }
2214
2215         ath10k_unchain_msdu(amsdu, unchain_cnt);
2216 }
2217
2218 static bool ath10k_htt_rx_validate_amsdu(struct ath10k *ar,
2219                                          struct sk_buff_head *amsdu)
2220 {
2221         u8 *subframe_hdr;
2222         struct sk_buff *first;
2223         bool is_first, is_last;
2224         struct ath10k_hw_params *hw = &ar->hw_params;
2225         struct htt_rx_desc *rxd;
2226         struct rx_msdu_end_common *rxd_msdu_end_common;
2227         struct rx_mpdu_start *rxd_mpdu_start;
2228         struct ieee80211_hdr *hdr;
2229         size_t hdr_len, crypto_len;
2230         enum htt_rx_mpdu_encrypt_type enctype;
2231         int bytes_aligned = ar->hw_params.decap_align_bytes;
2232
2233         first = skb_peek(amsdu);
2234
2235         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
2236                                     (void *)first->data - hw->rx_desc_ops->rx_desc_size);
2237
2238         rxd_msdu_end_common = ath10k_htt_rx_desc_get_msdu_end(hw, rxd);
2239         rxd_mpdu_start = ath10k_htt_rx_desc_get_mpdu_start(hw, rxd);
2240         hdr = (void *)ath10k_htt_rx_desc_get_rx_hdr_status(hw, rxd);
2241
2242         is_first = !!(rxd_msdu_end_common->info0 &
2243                       __cpu_to_le32(RX_MSDU_END_INFO0_FIRST_MSDU));
2244         is_last = !!(rxd_msdu_end_common->info0 &
2245                      __cpu_to_le32(RX_MSDU_END_INFO0_LAST_MSDU));
2246
2247         /* Return in case of non-aggregated msdu */
2248         if (is_first && is_last)
2249                 return true;
2250
2251         /* First msdu flag is not set for the first msdu of the list */
2252         if (!is_first)
2253                 return false;
2254
2255         enctype = MS(__le32_to_cpu(rxd_mpdu_start->info0),
2256                      RX_MPDU_START_INFO0_ENCRYPT_TYPE);
2257
2258         hdr_len = ieee80211_hdrlen(hdr->frame_control);
2259         crypto_len = ath10k_htt_rx_crypto_param_len(ar, enctype);
2260
2261         subframe_hdr = (u8 *)hdr + round_up(hdr_len, bytes_aligned) +
2262                        crypto_len;
2263
2264         /* Validate if the amsdu has a proper first subframe.
2265          * There are chances a single msdu can be received as amsdu when
2266          * the unauthenticated amsdu flag of a QoS header
2267          * gets flipped in non-SPP AMSDU's, in such cases the first
2268          * subframe has llc/snap header in place of a valid da.
2269          * return false if the da matches rfc1042 pattern
2270          */
2271         if (ether_addr_equal(subframe_hdr, rfc1042_header))
2272                 return false;
2273
2274         return true;
2275 }
2276
2277 static bool ath10k_htt_rx_amsdu_allowed(struct ath10k *ar,
2278                                         struct sk_buff_head *amsdu,
2279                                         struct ieee80211_rx_status *rx_status)
2280 {
2281         if (!rx_status->freq) {
2282                 ath10k_dbg(ar, ATH10K_DBG_HTT, "no channel configured; ignoring frame(s)!\n");
2283                 return false;
2284         }
2285
2286         if (test_bit(ATH10K_CAC_RUNNING, &ar->dev_flags)) {
2287                 ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx cac running\n");
2288                 return false;
2289         }
2290
2291         if (!ath10k_htt_rx_validate_amsdu(ar, amsdu)) {
2292                 ath10k_dbg(ar, ATH10K_DBG_HTT, "invalid amsdu received\n");
2293                 return false;
2294         }
2295
2296         return true;
2297 }
2298
2299 static void ath10k_htt_rx_h_filter(struct ath10k *ar,
2300                                    struct sk_buff_head *amsdu,
2301                                    struct ieee80211_rx_status *rx_status,
2302                                    unsigned long *drop_cnt)
2303 {
2304         if (skb_queue_empty(amsdu))
2305                 return;
2306
2307         if (ath10k_htt_rx_amsdu_allowed(ar, amsdu, rx_status))
2308                 return;
2309
2310         if (drop_cnt)
2311                 *drop_cnt += skb_queue_len(amsdu);
2312
2313         __skb_queue_purge(amsdu);
2314 }
2315
2316 static int ath10k_htt_rx_handle_amsdu(struct ath10k_htt *htt)
2317 {
2318         struct ath10k *ar = htt->ar;
2319         struct ieee80211_rx_status *rx_status = &htt->rx_status;
2320         struct sk_buff_head amsdu;
2321         int ret;
2322         unsigned long drop_cnt = 0;
2323         unsigned long unchain_cnt = 0;
2324         unsigned long drop_cnt_filter = 0;
2325         unsigned long msdus_to_queue, num_msdus;
2326         enum ath10k_pkt_rx_err err = ATH10K_PKT_RX_ERR_MAX;
2327         u8 first_hdr[RX_HTT_HDR_STATUS_LEN];
2328
2329         __skb_queue_head_init(&amsdu);
2330
2331         spin_lock_bh(&htt->rx_ring.lock);
2332         if (htt->rx_confused) {
2333                 spin_unlock_bh(&htt->rx_ring.lock);
2334                 return -EIO;
2335         }
2336         ret = ath10k_htt_rx_amsdu_pop(htt, &amsdu);
2337         spin_unlock_bh(&htt->rx_ring.lock);
2338
2339         if (ret < 0) {
2340                 ath10k_warn(ar, "rx ring became corrupted: %d\n", ret);
2341                 __skb_queue_purge(&amsdu);
2342                 /* FIXME: It's probably a good idea to reboot the
2343                  * device instead of leaving it inoperable.
2344                  */
2345                 htt->rx_confused = true;
2346                 return ret;
2347         }
2348
2349         num_msdus = skb_queue_len(&amsdu);
2350
2351         ath10k_htt_rx_h_ppdu(ar, &amsdu, rx_status, 0xffff);
2352
2353         /* only for ret = 1 indicates chained msdus */
2354         if (ret > 0)
2355                 ath10k_htt_rx_h_unchain(ar, &amsdu, &drop_cnt, &unchain_cnt);
2356
2357         ath10k_htt_rx_h_filter(ar, &amsdu, rx_status, &drop_cnt_filter);
2358         ath10k_htt_rx_h_mpdu(ar, &amsdu, rx_status, true, first_hdr, &err, 0,
2359                              false);
2360         msdus_to_queue = skb_queue_len(&amsdu);
2361         ath10k_htt_rx_h_enqueue(ar, &amsdu, rx_status);
2362
2363         ath10k_sta_update_rx_tid_stats(ar, first_hdr, num_msdus, err,
2364                                        unchain_cnt, drop_cnt, drop_cnt_filter,
2365                                        msdus_to_queue);
2366
2367         return 0;
2368 }
2369
2370 static void ath10k_htt_rx_mpdu_desc_pn_hl(struct htt_hl_rx_desc *rx_desc,
2371                                           union htt_rx_pn_t *pn,
2372                                           int pn_len_bits)
2373 {
2374         switch (pn_len_bits) {
2375         case 48:
2376                 pn->pn48 = __le32_to_cpu(rx_desc->pn_31_0) +
2377                            ((u64)(__le32_to_cpu(rx_desc->u0.pn_63_32) & 0xFFFF) << 32);
2378                 break;
2379         case 24:
2380                 pn->pn24 = __le32_to_cpu(rx_desc->pn_31_0);
2381                 break;
2382         }
2383 }
2384
2385 static bool ath10k_htt_rx_pn_cmp48(union htt_rx_pn_t *new_pn,
2386                                    union htt_rx_pn_t *old_pn)
2387 {
2388         return ((new_pn->pn48 & 0xffffffffffffULL) <=
2389                 (old_pn->pn48 & 0xffffffffffffULL));
2390 }
2391
2392 static bool ath10k_htt_rx_pn_check_replay_hl(struct ath10k *ar,
2393                                              struct ath10k_peer *peer,
2394                                              struct htt_rx_indication_hl *rx)
2395 {
2396         bool last_pn_valid, pn_invalid = false;
2397         enum htt_txrx_sec_cast_type sec_index;
2398         enum htt_security_types sec_type;
2399         union htt_rx_pn_t new_pn = {0};
2400         struct htt_hl_rx_desc *rx_desc;
2401         union htt_rx_pn_t *last_pn;
2402         u32 rx_desc_info, tid;
2403         int num_mpdu_ranges;
2404
2405         lockdep_assert_held(&ar->data_lock);
2406
2407         if (!peer)
2408                 return false;
2409
2410         if (!(rx->fw_desc.flags & FW_RX_DESC_FLAGS_FIRST_MSDU))
2411                 return false;
2412
2413         num_mpdu_ranges = MS(__le32_to_cpu(rx->hdr.info1),
2414                              HTT_RX_INDICATION_INFO1_NUM_MPDU_RANGES);
2415
2416         rx_desc = (struct htt_hl_rx_desc *)&rx->mpdu_ranges[num_mpdu_ranges];
2417         rx_desc_info = __le32_to_cpu(rx_desc->info);
2418
2419         if (!MS(rx_desc_info, HTT_RX_DESC_HL_INFO_ENCRYPTED))
2420                 return false;
2421
2422         tid = MS(rx->hdr.info0, HTT_RX_INDICATION_INFO0_EXT_TID);
2423         last_pn_valid = peer->tids_last_pn_valid[tid];
2424         last_pn = &peer->tids_last_pn[tid];
2425
2426         if (MS(rx_desc_info, HTT_RX_DESC_HL_INFO_MCAST_BCAST))
2427                 sec_index = HTT_TXRX_SEC_MCAST;
2428         else
2429                 sec_index = HTT_TXRX_SEC_UCAST;
2430
2431         sec_type = peer->rx_pn[sec_index].sec_type;
2432         ath10k_htt_rx_mpdu_desc_pn_hl(rx_desc, &new_pn, peer->rx_pn[sec_index].pn_len);
2433
2434         if (sec_type != HTT_SECURITY_AES_CCMP &&
2435             sec_type != HTT_SECURITY_TKIP &&
2436             sec_type != HTT_SECURITY_TKIP_NOMIC)
2437                 return false;
2438
2439         if (last_pn_valid)
2440                 pn_invalid = ath10k_htt_rx_pn_cmp48(&new_pn, last_pn);
2441         else
2442                 peer->tids_last_pn_valid[tid] = true;
2443
2444         if (!pn_invalid)
2445                 last_pn->pn48 = new_pn.pn48;
2446
2447         return pn_invalid;
2448 }
2449
2450 static bool ath10k_htt_rx_proc_rx_ind_hl(struct ath10k_htt *htt,
2451                                          struct htt_rx_indication_hl *rx,
2452                                          struct sk_buff *skb,
2453                                          enum htt_rx_pn_check_type check_pn_type,
2454                                          enum htt_rx_tkip_demic_type tkip_mic_type)
2455 {
2456         struct ath10k *ar = htt->ar;
2457         struct ath10k_peer *peer;
2458         struct htt_rx_indication_mpdu_range *mpdu_ranges;
2459         struct fw_rx_desc_hl *fw_desc;
2460         enum htt_txrx_sec_cast_type sec_index;
2461         enum htt_security_types sec_type;
2462         union htt_rx_pn_t new_pn = {0};
2463         struct htt_hl_rx_desc *rx_desc;
2464         struct ieee80211_hdr *hdr;
2465         struct ieee80211_rx_status *rx_status;
2466         u16 peer_id;
2467         u8 rx_desc_len;
2468         int num_mpdu_ranges;
2469         size_t tot_hdr_len;
2470         struct ieee80211_channel *ch;
2471         bool pn_invalid, qos, first_msdu;
2472         u32 tid, rx_desc_info;
2473
2474         peer_id = __le16_to_cpu(rx->hdr.peer_id);
2475         tid = MS(rx->hdr.info0, HTT_RX_INDICATION_INFO0_EXT_TID);
2476
2477         spin_lock_bh(&ar->data_lock);
2478         peer = ath10k_peer_find_by_id(ar, peer_id);
2479         spin_unlock_bh(&ar->data_lock);
2480         if (!peer && peer_id != HTT_INVALID_PEERID)
2481                 ath10k_warn(ar, "Got RX ind from invalid peer: %u\n", peer_id);
2482
2483         if (!peer)
2484                 return true;
2485
2486         num_mpdu_ranges = MS(__le32_to_cpu(rx->hdr.info1),
2487                              HTT_RX_INDICATION_INFO1_NUM_MPDU_RANGES);
2488         mpdu_ranges = htt_rx_ind_get_mpdu_ranges_hl(rx);
2489         fw_desc = &rx->fw_desc;
2490         rx_desc_len = fw_desc->len;
2491
2492         if (fw_desc->u.bits.discard) {
2493                 ath10k_dbg(ar, ATH10K_DBG_HTT, "htt discard mpdu\n");
2494                 goto err;
2495         }
2496
2497         /* I have not yet seen any case where num_mpdu_ranges > 1.
2498          * qcacld does not seem handle that case either, so we introduce the
2499          * same limitiation here as well.
2500          */
2501         if (num_mpdu_ranges > 1)
2502                 ath10k_warn(ar,
2503                             "Unsupported number of MPDU ranges: %d, ignoring all but the first\n",
2504                             num_mpdu_ranges);
2505
2506         if (mpdu_ranges->mpdu_range_status !=
2507             HTT_RX_IND_MPDU_STATUS_OK &&
2508             mpdu_ranges->mpdu_range_status !=
2509             HTT_RX_IND_MPDU_STATUS_TKIP_MIC_ERR) {
2510                 ath10k_dbg(ar, ATH10K_DBG_HTT, "htt mpdu_range_status %d\n",
2511                            mpdu_ranges->mpdu_range_status);
2512                 goto err;
2513         }
2514
2515         rx_desc = (struct htt_hl_rx_desc *)&rx->mpdu_ranges[num_mpdu_ranges];
2516         rx_desc_info = __le32_to_cpu(rx_desc->info);
2517
2518         if (MS(rx_desc_info, HTT_RX_DESC_HL_INFO_MCAST_BCAST))
2519                 sec_index = HTT_TXRX_SEC_MCAST;
2520         else
2521                 sec_index = HTT_TXRX_SEC_UCAST;
2522
2523         sec_type = peer->rx_pn[sec_index].sec_type;
2524         first_msdu = rx->fw_desc.flags & FW_RX_DESC_FLAGS_FIRST_MSDU;
2525
2526         ath10k_htt_rx_mpdu_desc_pn_hl(rx_desc, &new_pn, peer->rx_pn[sec_index].pn_len);
2527
2528         if (check_pn_type == HTT_RX_PN_CHECK && tid >= IEEE80211_NUM_TIDS) {
2529                 spin_lock_bh(&ar->data_lock);
2530                 pn_invalid = ath10k_htt_rx_pn_check_replay_hl(ar, peer, rx);
2531                 spin_unlock_bh(&ar->data_lock);
2532
2533                 if (pn_invalid)
2534                         goto err;
2535         }
2536
2537         /* Strip off all headers before the MAC header before delivery to
2538          * mac80211
2539          */
2540         tot_hdr_len = sizeof(struct htt_resp_hdr) + sizeof(rx->hdr) +
2541                       sizeof(rx->ppdu) + sizeof(rx->prefix) +
2542                       sizeof(rx->fw_desc) +
2543                       sizeof(*mpdu_ranges) * num_mpdu_ranges + rx_desc_len;
2544
2545         skb_pull(skb, tot_hdr_len);
2546
2547         hdr = (struct ieee80211_hdr *)skb->data;
2548         qos = ieee80211_is_data_qos(hdr->frame_control);
2549
2550         rx_status = IEEE80211_SKB_RXCB(skb);
2551         memset(rx_status, 0, sizeof(*rx_status));
2552
2553         if (rx->ppdu.combined_rssi == 0) {
2554                 /* SDIO firmware does not provide signal */
2555                 rx_status->signal = 0;
2556                 rx_status->flag |= RX_FLAG_NO_SIGNAL_VAL;
2557         } else {
2558                 rx_status->signal = ATH10K_DEFAULT_NOISE_FLOOR +
2559                         rx->ppdu.combined_rssi;
2560                 rx_status->flag &= ~RX_FLAG_NO_SIGNAL_VAL;
2561         }
2562
2563         spin_lock_bh(&ar->data_lock);
2564         ch = ar->scan_channel;
2565         if (!ch)
2566                 ch = ar->rx_channel;
2567         if (!ch)
2568                 ch = ath10k_htt_rx_h_any_channel(ar);
2569         if (!ch)
2570                 ch = ar->tgt_oper_chan;
2571         spin_unlock_bh(&ar->data_lock);
2572
2573         if (ch) {
2574                 rx_status->band = ch->band;
2575                 rx_status->freq = ch->center_freq;
2576         }
2577         if (rx->fw_desc.flags & FW_RX_DESC_FLAGS_LAST_MSDU)
2578                 rx_status->flag &= ~RX_FLAG_AMSDU_MORE;
2579         else
2580                 rx_status->flag |= RX_FLAG_AMSDU_MORE;
2581
2582         /* Not entirely sure about this, but all frames from the chipset has
2583          * the protected flag set even though they have already been decrypted.
2584          * Unmasking this flag is necessary in order for mac80211 not to drop
2585          * the frame.
2586          * TODO: Verify this is always the case or find out a way to check
2587          * if there has been hw decryption.
2588          */
2589         if (ieee80211_has_protected(hdr->frame_control)) {
2590                 hdr->frame_control &= ~__cpu_to_le16(IEEE80211_FCTL_PROTECTED);
2591                 rx_status->flag |= RX_FLAG_DECRYPTED |
2592                                    RX_FLAG_IV_STRIPPED |
2593                                    RX_FLAG_MMIC_STRIPPED;
2594
2595                 if (tid < IEEE80211_NUM_TIDS &&
2596                     first_msdu &&
2597                     check_pn_type == HTT_RX_PN_CHECK &&
2598                    (sec_type == HTT_SECURITY_AES_CCMP ||
2599                     sec_type == HTT_SECURITY_TKIP ||
2600                     sec_type == HTT_SECURITY_TKIP_NOMIC)) {
2601                         u8 offset, *ivp, i;
2602                         s8 keyidx = 0;
2603                         __le64 pn48 = cpu_to_le64(new_pn.pn48);
2604
2605                         hdr = (struct ieee80211_hdr *)skb->data;
2606                         offset = ieee80211_hdrlen(hdr->frame_control);
2607                         hdr->frame_control |= __cpu_to_le16(IEEE80211_FCTL_PROTECTED);
2608                         rx_status->flag &= ~RX_FLAG_IV_STRIPPED;
2609
2610                         memmove(skb->data - IEEE80211_CCMP_HDR_LEN,
2611                                 skb->data, offset);
2612                         skb_push(skb, IEEE80211_CCMP_HDR_LEN);
2613                         ivp = skb->data + offset;
2614                         memset(skb->data + offset, 0, IEEE80211_CCMP_HDR_LEN);
2615                         /* Ext IV */
2616                         ivp[IEEE80211_WEP_IV_LEN - 1] |= ATH10K_IEEE80211_EXTIV;
2617
2618                         for (i = 0; i < ARRAY_SIZE(peer->keys); i++) {
2619                                 if (peer->keys[i] &&
2620                                     peer->keys[i]->flags & IEEE80211_KEY_FLAG_PAIRWISE)
2621                                         keyidx = peer->keys[i]->keyidx;
2622                         }
2623
2624                         /* Key ID */
2625                         ivp[IEEE80211_WEP_IV_LEN - 1] |= keyidx << 6;
2626
2627                         if (sec_type == HTT_SECURITY_AES_CCMP) {
2628                                 rx_status->flag |= RX_FLAG_MIC_STRIPPED;
2629                                 /* pn 0, pn 1 */
2630                                 memcpy(skb->data + offset, &pn48, 2);
2631                                 /* pn 1, pn 3 , pn 34 , pn 5 */
2632                                 memcpy(skb->data + offset + 4, ((u8 *)&pn48) + 2, 4);
2633                         } else {
2634                                 rx_status->flag |= RX_FLAG_ICV_STRIPPED;
2635                                 /* TSC 0 */
2636                                 memcpy(skb->data + offset + 2, &pn48, 1);
2637                                 /* TSC 1 */
2638                                 memcpy(skb->data + offset, ((u8 *)&pn48) + 1, 1);
2639                                 /* TSC 2 , TSC 3 , TSC 4 , TSC 5*/
2640                                 memcpy(skb->data + offset + 4, ((u8 *)&pn48) + 2, 4);
2641                         }
2642                 }
2643         }
2644
2645         if (tkip_mic_type == HTT_RX_TKIP_MIC)
2646                 rx_status->flag &= ~RX_FLAG_IV_STRIPPED &
2647                                    ~RX_FLAG_MMIC_STRIPPED;
2648
2649         if (mpdu_ranges->mpdu_range_status == HTT_RX_IND_MPDU_STATUS_TKIP_MIC_ERR)
2650                 rx_status->flag |= RX_FLAG_MMIC_ERROR;
2651
2652         if (!qos && tid < IEEE80211_NUM_TIDS) {
2653                 u8 offset;
2654                 __le16 qos_ctrl = 0;
2655
2656                 hdr = (struct ieee80211_hdr *)skb->data;
2657                 offset = ieee80211_hdrlen(hdr->frame_control);
2658
2659                 hdr->frame_control |= cpu_to_le16(IEEE80211_STYPE_QOS_DATA);
2660                 memmove(skb->data - IEEE80211_QOS_CTL_LEN, skb->data, offset);
2661                 skb_push(skb, IEEE80211_QOS_CTL_LEN);
2662                 qos_ctrl = cpu_to_le16(tid);
2663                 memcpy(skb->data + offset, &qos_ctrl, IEEE80211_QOS_CTL_LEN);
2664         }
2665
2666         if (ar->napi.dev)
2667                 ieee80211_rx_napi(ar->hw, NULL, skb, &ar->napi);
2668         else
2669                 ieee80211_rx_ni(ar->hw, skb);
2670
2671         /* We have delivered the skb to the upper layers (mac80211) so we
2672          * must not free it.
2673          */
2674         return false;
2675 err:
2676         /* Tell the caller that it must free the skb since we have not
2677          * consumed it
2678          */
2679         return true;
2680 }
2681
2682 static int ath10k_htt_rx_frag_tkip_decap_nomic(struct sk_buff *skb,
2683                                                u16 head_len,
2684                                                u16 hdr_len)
2685 {
2686         u8 *ivp, *orig_hdr;
2687
2688         orig_hdr = skb->data;
2689         ivp = orig_hdr + hdr_len + head_len;
2690
2691         /* the ExtIV bit is always set to 1 for TKIP */
2692         if (!(ivp[IEEE80211_WEP_IV_LEN - 1] & ATH10K_IEEE80211_EXTIV))
2693                 return -EINVAL;
2694
2695         memmove(orig_hdr + IEEE80211_TKIP_IV_LEN, orig_hdr, head_len + hdr_len);
2696         skb_pull(skb, IEEE80211_TKIP_IV_LEN);
2697         skb_trim(skb, skb->len - ATH10K_IEEE80211_TKIP_MICLEN);
2698         return 0;
2699 }
2700
2701 static int ath10k_htt_rx_frag_tkip_decap_withmic(struct sk_buff *skb,
2702                                                  u16 head_len,
2703                                                  u16 hdr_len)
2704 {
2705         u8 *ivp, *orig_hdr;
2706
2707         orig_hdr = skb->data;
2708         ivp = orig_hdr + hdr_len + head_len;
2709
2710         /* the ExtIV bit is always set to 1 for TKIP */
2711         if (!(ivp[IEEE80211_WEP_IV_LEN - 1] & ATH10K_IEEE80211_EXTIV))
2712                 return -EINVAL;
2713
2714         memmove(orig_hdr + IEEE80211_TKIP_IV_LEN, orig_hdr, head_len + hdr_len);
2715         skb_pull(skb, IEEE80211_TKIP_IV_LEN);
2716         skb_trim(skb, skb->len - IEEE80211_TKIP_ICV_LEN);
2717         return 0;
2718 }
2719
2720 static int ath10k_htt_rx_frag_ccmp_decap(struct sk_buff *skb,
2721                                          u16 head_len,
2722                                          u16 hdr_len)
2723 {
2724         u8 *ivp, *orig_hdr;
2725
2726         orig_hdr = skb->data;
2727         ivp = orig_hdr + hdr_len + head_len;
2728
2729         /* the ExtIV bit is always set to 1 for CCMP */
2730         if (!(ivp[IEEE80211_WEP_IV_LEN - 1] & ATH10K_IEEE80211_EXTIV))
2731                 return -EINVAL;
2732
2733         skb_trim(skb, skb->len - IEEE80211_CCMP_MIC_LEN);
2734         memmove(orig_hdr + IEEE80211_CCMP_HDR_LEN, orig_hdr, head_len + hdr_len);
2735         skb_pull(skb, IEEE80211_CCMP_HDR_LEN);
2736         return 0;
2737 }
2738
2739 static int ath10k_htt_rx_frag_wep_decap(struct sk_buff *skb,
2740                                         u16 head_len,
2741                                         u16 hdr_len)
2742 {
2743         u8 *orig_hdr;
2744
2745         orig_hdr = skb->data;
2746
2747         memmove(orig_hdr + IEEE80211_WEP_IV_LEN,
2748                 orig_hdr, head_len + hdr_len);
2749         skb_pull(skb, IEEE80211_WEP_IV_LEN);
2750         skb_trim(skb, skb->len - IEEE80211_WEP_ICV_LEN);
2751         return 0;
2752 }
2753
2754 static bool ath10k_htt_rx_proc_rx_frag_ind_hl(struct ath10k_htt *htt,
2755                                               struct htt_rx_fragment_indication *rx,
2756                                               struct sk_buff *skb)
2757 {
2758         struct ath10k *ar = htt->ar;
2759         enum htt_rx_tkip_demic_type tkip_mic = HTT_RX_NON_TKIP_MIC;
2760         enum htt_txrx_sec_cast_type sec_index;
2761         struct htt_rx_indication_hl *rx_hl;
2762         enum htt_security_types sec_type;
2763         u32 tid, frag, seq, rx_desc_info;
2764         union htt_rx_pn_t new_pn = {0};
2765         struct htt_hl_rx_desc *rx_desc;
2766         u16 peer_id, sc, hdr_space;
2767         union htt_rx_pn_t *last_pn;
2768         struct ieee80211_hdr *hdr;
2769         int ret, num_mpdu_ranges;
2770         struct ath10k_peer *peer;
2771         struct htt_resp *resp;
2772         size_t tot_hdr_len;
2773
2774         resp = (struct htt_resp *)(skb->data + HTT_RX_FRAG_IND_INFO0_HEADER_LEN);
2775         skb_pull(skb, HTT_RX_FRAG_IND_INFO0_HEADER_LEN);
2776         skb_trim(skb, skb->len - FCS_LEN);
2777
2778         peer_id = __le16_to_cpu(rx->peer_id);
2779         rx_hl = (struct htt_rx_indication_hl *)(&resp->rx_ind_hl);
2780
2781         spin_lock_bh(&ar->data_lock);
2782         peer = ath10k_peer_find_by_id(ar, peer_id);
2783         if (!peer) {
2784                 ath10k_dbg(ar, ATH10K_DBG_HTT, "invalid peer: %u\n", peer_id);
2785                 goto err;
2786         }
2787
2788         num_mpdu_ranges = MS(__le32_to_cpu(rx_hl->hdr.info1),
2789                              HTT_RX_INDICATION_INFO1_NUM_MPDU_RANGES);
2790
2791         tot_hdr_len = sizeof(struct htt_resp_hdr) +
2792                       sizeof(rx_hl->hdr) +
2793                       sizeof(rx_hl->ppdu) +
2794                       sizeof(rx_hl->prefix) +
2795                       sizeof(rx_hl->fw_desc) +
2796                       sizeof(struct htt_rx_indication_mpdu_range) * num_mpdu_ranges;
2797
2798         tid =  MS(rx_hl->hdr.info0, HTT_RX_INDICATION_INFO0_EXT_TID);
2799         rx_desc = (struct htt_hl_rx_desc *)(skb->data + tot_hdr_len);
2800         rx_desc_info = __le32_to_cpu(rx_desc->info);
2801
2802         hdr = (struct ieee80211_hdr *)((u8 *)rx_desc + rx_hl->fw_desc.len);
2803
2804         if (is_multicast_ether_addr(hdr->addr1)) {
2805                 /* Discard the fragment with multicast DA */
2806                 goto err;
2807         }
2808
2809         if (!MS(rx_desc_info, HTT_RX_DESC_HL_INFO_ENCRYPTED)) {
2810                 spin_unlock_bh(&ar->data_lock);
2811                 return ath10k_htt_rx_proc_rx_ind_hl(htt, &resp->rx_ind_hl, skb,
2812                                                     HTT_RX_NON_PN_CHECK,
2813                                                     HTT_RX_NON_TKIP_MIC);
2814         }
2815
2816         if (ieee80211_has_retry(hdr->frame_control))
2817                 goto err;
2818
2819         hdr_space = ieee80211_hdrlen(hdr->frame_control);
2820         sc = __le16_to_cpu(hdr->seq_ctrl);
2821         seq = (sc & IEEE80211_SCTL_SEQ) >> 4;
2822         frag = sc & IEEE80211_SCTL_FRAG;
2823
2824         sec_index = MS(rx_desc_info, HTT_RX_DESC_HL_INFO_MCAST_BCAST) ?
2825                     HTT_TXRX_SEC_MCAST : HTT_TXRX_SEC_UCAST;
2826         sec_type = peer->rx_pn[sec_index].sec_type;
2827         ath10k_htt_rx_mpdu_desc_pn_hl(rx_desc, &new_pn, peer->rx_pn[sec_index].pn_len);
2828
2829         switch (sec_type) {
2830         case HTT_SECURITY_TKIP:
2831                 tkip_mic = HTT_RX_TKIP_MIC;
2832                 ret = ath10k_htt_rx_frag_tkip_decap_withmic(skb,
2833                                                             tot_hdr_len +
2834                                                             rx_hl->fw_desc.len,
2835                                                             hdr_space);
2836                 if (ret)
2837                         goto err;
2838                 break;
2839         case HTT_SECURITY_TKIP_NOMIC:
2840                 ret = ath10k_htt_rx_frag_tkip_decap_nomic(skb,
2841                                                           tot_hdr_len +
2842                                                           rx_hl->fw_desc.len,
2843                                                           hdr_space);
2844                 if (ret)
2845                         goto err;
2846                 break;
2847         case HTT_SECURITY_AES_CCMP:
2848                 ret = ath10k_htt_rx_frag_ccmp_decap(skb,
2849                                                     tot_hdr_len + rx_hl->fw_desc.len,
2850                                                     hdr_space);
2851                 if (ret)
2852                         goto err;
2853                 break;
2854         case HTT_SECURITY_WEP128:
2855         case HTT_SECURITY_WEP104:
2856         case HTT_SECURITY_WEP40:
2857                 ret = ath10k_htt_rx_frag_wep_decap(skb,
2858                                                    tot_hdr_len + rx_hl->fw_desc.len,
2859                                                    hdr_space);
2860                 if (ret)
2861                         goto err;
2862                 break;
2863         default:
2864                 break;
2865         }
2866
2867         resp = (struct htt_resp *)(skb->data);
2868
2869         if (sec_type != HTT_SECURITY_AES_CCMP &&
2870             sec_type != HTT_SECURITY_TKIP &&
2871             sec_type != HTT_SECURITY_TKIP_NOMIC) {
2872                 spin_unlock_bh(&ar->data_lock);
2873                 return ath10k_htt_rx_proc_rx_ind_hl(htt, &resp->rx_ind_hl, skb,
2874                                                     HTT_RX_NON_PN_CHECK,
2875                                                     HTT_RX_NON_TKIP_MIC);
2876         }
2877
2878         last_pn = &peer->frag_tids_last_pn[tid];
2879
2880         if (frag == 0) {
2881                 if (ath10k_htt_rx_pn_check_replay_hl(ar, peer, &resp->rx_ind_hl))
2882                         goto err;
2883
2884                 last_pn->pn48 = new_pn.pn48;
2885                 peer->frag_tids_seq[tid] = seq;
2886         } else if (sec_type == HTT_SECURITY_AES_CCMP) {
2887                 if (seq != peer->frag_tids_seq[tid])
2888                         goto err;
2889
2890                 if (new_pn.pn48 != last_pn->pn48 + 1)
2891                         goto err;
2892
2893                 last_pn->pn48 = new_pn.pn48;
2894                 last_pn = &peer->tids_last_pn[tid];
2895                 last_pn->pn48 = new_pn.pn48;
2896         }
2897
2898         spin_unlock_bh(&ar->data_lock);
2899
2900         return ath10k_htt_rx_proc_rx_ind_hl(htt, &resp->rx_ind_hl, skb,
2901                                             HTT_RX_NON_PN_CHECK, tkip_mic);
2902
2903 err:
2904         spin_unlock_bh(&ar->data_lock);
2905
2906         /* Tell the caller that it must free the skb since we have not
2907          * consumed it
2908          */
2909         return true;
2910 }
2911
2912 static void ath10k_htt_rx_proc_rx_ind_ll(struct ath10k_htt *htt,
2913                                          struct htt_rx_indication *rx)
2914 {
2915         struct ath10k *ar = htt->ar;
2916         struct htt_rx_indication_mpdu_range *mpdu_ranges;
2917         int num_mpdu_ranges;
2918         int i, mpdu_count = 0;
2919         u16 peer_id;
2920         u8 tid;
2921
2922         num_mpdu_ranges = MS(__le32_to_cpu(rx->hdr.info1),
2923                              HTT_RX_INDICATION_INFO1_NUM_MPDU_RANGES);
2924         peer_id = __le16_to_cpu(rx->hdr.peer_id);
2925         tid =  MS(rx->hdr.info0, HTT_RX_INDICATION_INFO0_EXT_TID);
2926
2927         mpdu_ranges = htt_rx_ind_get_mpdu_ranges(rx);
2928
2929         ath10k_dbg_dump(ar, ATH10K_DBG_HTT_DUMP, NULL, "htt rx ind: ",
2930                         rx, struct_size(rx, mpdu_ranges, num_mpdu_ranges));
2931
2932         for (i = 0; i < num_mpdu_ranges; i++)
2933                 mpdu_count += mpdu_ranges[i].mpdu_count;
2934
2935         atomic_add(mpdu_count, &htt->num_mpdus_ready);
2936
2937         ath10k_sta_update_rx_tid_stats_ampdu(ar, peer_id, tid, mpdu_ranges,
2938                                              num_mpdu_ranges);
2939 }
2940
2941 static void ath10k_htt_rx_tx_compl_ind(struct ath10k *ar,
2942                                        struct sk_buff *skb)
2943 {
2944         struct ath10k_htt *htt = &ar->htt;
2945         struct htt_resp *resp = (struct htt_resp *)skb->data;
2946         struct htt_tx_done tx_done = {};
2947         int status = MS(resp->data_tx_completion.flags, HTT_DATA_TX_STATUS);
2948         __le16 msdu_id, *msdus;
2949         bool rssi_enabled = false;
2950         u8 msdu_count = 0, num_airtime_records, tid;
2951         int i, htt_pad = 0;
2952         struct htt_data_tx_compl_ppdu_dur *ppdu_info;
2953         struct ath10k_peer *peer;
2954         u16 ppdu_info_offset = 0, peer_id;
2955         u32 tx_duration;
2956
2957         switch (status) {
2958         case HTT_DATA_TX_STATUS_NO_ACK:
2959                 tx_done.status = HTT_TX_COMPL_STATE_NOACK;
2960                 break;
2961         case HTT_DATA_TX_STATUS_OK:
2962                 tx_done.status = HTT_TX_COMPL_STATE_ACK;
2963                 break;
2964         case HTT_DATA_TX_STATUS_DISCARD:
2965         case HTT_DATA_TX_STATUS_POSTPONE:
2966         case HTT_DATA_TX_STATUS_DOWNLOAD_FAIL:
2967                 tx_done.status = HTT_TX_COMPL_STATE_DISCARD;
2968                 break;
2969         default:
2970                 ath10k_warn(ar, "unhandled tx completion status %d\n", status);
2971                 tx_done.status = HTT_TX_COMPL_STATE_DISCARD;
2972                 break;
2973         }
2974
2975         ath10k_dbg(ar, ATH10K_DBG_HTT, "htt tx completion num_msdus %d\n",
2976                    resp->data_tx_completion.num_msdus);
2977
2978         msdu_count = resp->data_tx_completion.num_msdus;
2979         msdus = resp->data_tx_completion.msdus;
2980         rssi_enabled = ath10k_is_rssi_enable(&ar->hw_params, resp);
2981
2982         if (rssi_enabled)
2983                 htt_pad = ath10k_tx_data_rssi_get_pad_bytes(&ar->hw_params,
2984                                                             resp);
2985
2986         for (i = 0; i < msdu_count; i++) {
2987                 msdu_id = msdus[i];
2988                 tx_done.msdu_id = __le16_to_cpu(msdu_id);
2989
2990                 if (rssi_enabled) {
2991                         /* Total no of MSDUs should be even,
2992                          * if odd MSDUs are sent firmware fills
2993                          * last msdu id with 0xffff
2994                          */
2995                         if (msdu_count & 0x01) {
2996                                 msdu_id = msdus[msdu_count +  i + 1 + htt_pad];
2997                                 tx_done.ack_rssi = __le16_to_cpu(msdu_id);
2998                         } else {
2999                                 msdu_id = msdus[msdu_count +  i + htt_pad];
3000                                 tx_done.ack_rssi = __le16_to_cpu(msdu_id);
3001                         }
3002                 }
3003
3004                 /* kfifo_put: In practice firmware shouldn't fire off per-CE
3005                  * interrupt and main interrupt (MSI/-X range case) for the same
3006                  * HTC service so it should be safe to use kfifo_put w/o lock.
3007                  *
3008                  * From kfifo_put() documentation:
3009                  *  Note that with only one concurrent reader and one concurrent
3010                  *  writer, you don't need extra locking to use these macro.
3011                  */
3012                 if (ar->bus_param.dev_type == ATH10K_DEV_TYPE_HL) {
3013                         ath10k_txrx_tx_unref(htt, &tx_done);
3014                 } else if (!kfifo_put(&htt->txdone_fifo, tx_done)) {
3015                         ath10k_warn(ar, "txdone fifo overrun, msdu_id %d status %d\n",
3016                                     tx_done.msdu_id, tx_done.status);
3017                         ath10k_txrx_tx_unref(htt, &tx_done);
3018                 }
3019         }
3020
3021         if (!(resp->data_tx_completion.flags2 & HTT_TX_CMPL_FLAG_PPDU_DURATION_PRESENT))
3022                 return;
3023
3024         ppdu_info_offset = (msdu_count & 0x01) ? msdu_count + 1 : msdu_count;
3025
3026         if (rssi_enabled)
3027                 ppdu_info_offset += ppdu_info_offset;
3028
3029         if (resp->data_tx_completion.flags2 &
3030             (HTT_TX_CMPL_FLAG_PPID_PRESENT | HTT_TX_CMPL_FLAG_PA_PRESENT))
3031                 ppdu_info_offset += 2;
3032
3033         ppdu_info = (struct htt_data_tx_compl_ppdu_dur *)&msdus[ppdu_info_offset];
3034         num_airtime_records = FIELD_GET(HTT_TX_COMPL_PPDU_DUR_INFO0_NUM_ENTRIES_MASK,
3035                                         __le32_to_cpu(ppdu_info->info0));
3036
3037         for (i = 0; i < num_airtime_records; i++) {
3038                 struct htt_data_tx_ppdu_dur *ppdu_dur;
3039                 u32 info0;
3040
3041                 ppdu_dur = &ppdu_info->ppdu_dur[i];
3042                 info0 = __le32_to_cpu(ppdu_dur->info0);
3043
3044                 peer_id = FIELD_GET(HTT_TX_PPDU_DUR_INFO0_PEER_ID_MASK,
3045                                     info0);
3046                 rcu_read_lock();
3047                 spin_lock_bh(&ar->data_lock);
3048
3049                 peer = ath10k_peer_find_by_id(ar, peer_id);
3050                 if (!peer || !peer->sta) {
3051                         spin_unlock_bh(&ar->data_lock);
3052                         rcu_read_unlock();
3053                         continue;
3054                 }
3055
3056                 tid = FIELD_GET(HTT_TX_PPDU_DUR_INFO0_TID_MASK, info0) &
3057                                                 IEEE80211_QOS_CTL_TID_MASK;
3058                 tx_duration = __le32_to_cpu(ppdu_dur->tx_duration);
3059
3060                 ieee80211_sta_register_airtime(peer->sta, tid, tx_duration, 0);
3061
3062                 spin_unlock_bh(&ar->data_lock);
3063                 rcu_read_unlock();
3064         }
3065 }
3066
3067 static void ath10k_htt_rx_addba(struct ath10k *ar, struct htt_resp *resp)
3068 {
3069         struct htt_rx_addba *ev = &resp->rx_addba;
3070         struct ath10k_peer *peer;
3071         struct ath10k_vif *arvif;
3072         u16 info0, tid, peer_id;
3073
3074         info0 = __le16_to_cpu(ev->info0);
3075         tid = MS(info0, HTT_RX_BA_INFO0_TID);
3076         peer_id = MS(info0, HTT_RX_BA_INFO0_PEER_ID);
3077
3078         ath10k_dbg(ar, ATH10K_DBG_HTT,
3079                    "htt rx addba tid %u peer_id %u size %u\n",
3080                    tid, peer_id, ev->window_size);
3081
3082         spin_lock_bh(&ar->data_lock);
3083         peer = ath10k_peer_find_by_id(ar, peer_id);
3084         if (!peer) {
3085                 ath10k_warn(ar, "received addba event for invalid peer_id: %u\n",
3086                             peer_id);
3087                 spin_unlock_bh(&ar->data_lock);
3088                 return;
3089         }
3090
3091         arvif = ath10k_get_arvif(ar, peer->vdev_id);
3092         if (!arvif) {
3093                 ath10k_warn(ar, "received addba event for invalid vdev_id: %u\n",
3094                             peer->vdev_id);
3095                 spin_unlock_bh(&ar->data_lock);
3096                 return;
3097         }
3098
3099         ath10k_dbg(ar, ATH10K_DBG_HTT,
3100                    "htt rx start rx ba session sta %pM tid %u size %u\n",
3101                    peer->addr, tid, ev->window_size);
3102
3103         ieee80211_start_rx_ba_session_offl(arvif->vif, peer->addr, tid);
3104         spin_unlock_bh(&ar->data_lock);
3105 }
3106
3107 static void ath10k_htt_rx_delba(struct ath10k *ar, struct htt_resp *resp)
3108 {
3109         struct htt_rx_delba *ev = &resp->rx_delba;
3110         struct ath10k_peer *peer;
3111         struct ath10k_vif *arvif;
3112         u16 info0, tid, peer_id;
3113
3114         info0 = __le16_to_cpu(ev->info0);
3115         tid = MS(info0, HTT_RX_BA_INFO0_TID);
3116         peer_id = MS(info0, HTT_RX_BA_INFO0_PEER_ID);
3117
3118         ath10k_dbg(ar, ATH10K_DBG_HTT,
3119                    "htt rx delba tid %u peer_id %u\n",
3120                    tid, peer_id);
3121
3122         spin_lock_bh(&ar->data_lock);
3123         peer = ath10k_peer_find_by_id(ar, peer_id);
3124         if (!peer) {
3125                 ath10k_warn(ar, "received addba event for invalid peer_id: %u\n",
3126                             peer_id);
3127                 spin_unlock_bh(&ar->data_lock);
3128                 return;
3129         }
3130
3131         arvif = ath10k_get_arvif(ar, peer->vdev_id);
3132         if (!arvif) {
3133                 ath10k_warn(ar, "received addba event for invalid vdev_id: %u\n",
3134                             peer->vdev_id);
3135                 spin_unlock_bh(&ar->data_lock);
3136                 return;
3137         }
3138
3139         ath10k_dbg(ar, ATH10K_DBG_HTT,
3140                    "htt rx stop rx ba session sta %pM tid %u\n",
3141                    peer->addr, tid);
3142
3143         ieee80211_stop_rx_ba_session_offl(arvif->vif, peer->addr, tid);
3144         spin_unlock_bh(&ar->data_lock);
3145 }
3146
3147 static int ath10k_htt_rx_extract_amsdu(struct ath10k_hw_params *hw,
3148                                        struct sk_buff_head *list,
3149                                        struct sk_buff_head *amsdu)
3150 {
3151         struct sk_buff *msdu;
3152         struct htt_rx_desc *rxd;
3153         struct rx_msdu_end_common *rxd_msdu_end_common;
3154
3155         if (skb_queue_empty(list))
3156                 return -ENOBUFS;
3157
3158         if (WARN_ON(!skb_queue_empty(amsdu)))
3159                 return -EINVAL;
3160
3161         while ((msdu = __skb_dequeue(list))) {
3162                 __skb_queue_tail(amsdu, msdu);
3163
3164                 rxd = HTT_RX_BUF_TO_RX_DESC(hw,
3165                                             (void *)msdu->data -
3166                                             hw->rx_desc_ops->rx_desc_size);
3167
3168                 rxd_msdu_end_common = ath10k_htt_rx_desc_get_msdu_end(hw, rxd);
3169                 if (rxd_msdu_end_common->info0 &
3170                     __cpu_to_le32(RX_MSDU_END_INFO0_LAST_MSDU))
3171                         break;
3172         }
3173
3174         msdu = skb_peek_tail(amsdu);
3175         rxd = HTT_RX_BUF_TO_RX_DESC(hw,
3176                                     (void *)msdu->data - hw->rx_desc_ops->rx_desc_size);
3177
3178         rxd_msdu_end_common = ath10k_htt_rx_desc_get_msdu_end(hw, rxd);
3179         if (!(rxd_msdu_end_common->info0 &
3180               __cpu_to_le32(RX_MSDU_END_INFO0_LAST_MSDU))) {
3181                 skb_queue_splice_init(amsdu, list);
3182                 return -EAGAIN;
3183         }
3184
3185         return 0;
3186 }
3187
3188 static void ath10k_htt_rx_h_rx_offload_prot(struct ieee80211_rx_status *status,
3189                                             struct sk_buff *skb)
3190 {
3191         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
3192
3193         if (!ieee80211_has_protected(hdr->frame_control))
3194                 return;
3195
3196         /* Offloaded frames are already decrypted but firmware insists they are
3197          * protected in the 802.11 header. Strip the flag.  Otherwise mac80211
3198          * will drop the frame.
3199          */
3200
3201         hdr->frame_control &= ~__cpu_to_le16(IEEE80211_FCTL_PROTECTED);
3202         status->flag |= RX_FLAG_DECRYPTED |
3203                         RX_FLAG_IV_STRIPPED |
3204                         RX_FLAG_MMIC_STRIPPED;
3205 }
3206
3207 static void ath10k_htt_rx_h_rx_offload(struct ath10k *ar,
3208                                        struct sk_buff_head *list)
3209 {
3210         struct ath10k_htt *htt = &ar->htt;
3211         struct ieee80211_rx_status *status = &htt->rx_status;
3212         struct htt_rx_offload_msdu *rx;
3213         struct sk_buff *msdu;
3214         size_t offset;
3215
3216         while ((msdu = __skb_dequeue(list))) {
3217                 /* Offloaded frames don't have Rx descriptor. Instead they have
3218                  * a short meta information header.
3219                  */
3220
3221                 rx = (void *)msdu->data;
3222
3223                 skb_put(msdu, sizeof(*rx));
3224                 skb_pull(msdu, sizeof(*rx));
3225
3226                 if (skb_tailroom(msdu) < __le16_to_cpu(rx->msdu_len)) {
3227                         ath10k_warn(ar, "dropping frame: offloaded rx msdu is too long!\n");
3228                         dev_kfree_skb_any(msdu);
3229                         continue;
3230                 }
3231
3232                 skb_put(msdu, __le16_to_cpu(rx->msdu_len));
3233
3234                 /* Offloaded rx header length isn't multiple of 2 nor 4 so the
3235                  * actual payload is unaligned. Align the frame.  Otherwise
3236                  * mac80211 complains.  This shouldn't reduce performance much
3237                  * because these offloaded frames are rare.
3238                  */
3239                 offset = 4 - ((unsigned long)msdu->data & 3);
3240                 skb_put(msdu, offset);
3241                 memmove(msdu->data + offset, msdu->data, msdu->len);
3242                 skb_pull(msdu, offset);
3243
3244                 /* FIXME: The frame is NWifi. Re-construct QoS Control
3245                  * if possible later.
3246                  */
3247
3248                 memset(status, 0, sizeof(*status));
3249                 status->flag |= RX_FLAG_NO_SIGNAL_VAL;
3250
3251                 ath10k_htt_rx_h_rx_offload_prot(status, msdu);
3252                 ath10k_htt_rx_h_channel(ar, status, NULL, rx->vdev_id);
3253                 ath10k_htt_rx_h_queue_msdu(ar, status, msdu);
3254         }
3255 }
3256
3257 static int ath10k_htt_rx_in_ord_ind(struct ath10k *ar, struct sk_buff *skb)
3258 {
3259         struct ath10k_htt *htt = &ar->htt;
3260         struct htt_resp *resp = (void *)skb->data;
3261         struct ieee80211_rx_status *status = &htt->rx_status;
3262         struct sk_buff_head list;
3263         struct sk_buff_head amsdu;
3264         u16 peer_id;
3265         u16 msdu_count;
3266         u8 vdev_id;
3267         u8 tid;
3268         bool offload;
3269         bool frag;
3270         int ret;
3271
3272         lockdep_assert_held(&htt->rx_ring.lock);
3273
3274         if (htt->rx_confused)
3275                 return -EIO;
3276
3277         skb_pull(skb, sizeof(resp->hdr));
3278         skb_pull(skb, sizeof(resp->rx_in_ord_ind));
3279
3280         peer_id = __le16_to_cpu(resp->rx_in_ord_ind.peer_id);
3281         msdu_count = __le16_to_cpu(resp->rx_in_ord_ind.msdu_count);
3282         vdev_id = resp->rx_in_ord_ind.vdev_id;
3283         tid = SM(resp->rx_in_ord_ind.info, HTT_RX_IN_ORD_IND_INFO_TID);
3284         offload = !!(resp->rx_in_ord_ind.info &
3285                         HTT_RX_IN_ORD_IND_INFO_OFFLOAD_MASK);
3286         frag = !!(resp->rx_in_ord_ind.info & HTT_RX_IN_ORD_IND_INFO_FRAG_MASK);
3287
3288         ath10k_dbg(ar, ATH10K_DBG_HTT,
3289                    "htt rx in ord vdev %i peer %i tid %i offload %i frag %i msdu count %i\n",
3290                    vdev_id, peer_id, tid, offload, frag, msdu_count);
3291
3292         if (skb->len < msdu_count * sizeof(*resp->rx_in_ord_ind.msdu_descs32)) {
3293                 ath10k_warn(ar, "dropping invalid in order rx indication\n");
3294                 return -EINVAL;
3295         }
3296
3297         /* The event can deliver more than 1 A-MSDU. Each A-MSDU is later
3298          * extracted and processed.
3299          */
3300         __skb_queue_head_init(&list);
3301         if (ar->hw_params.target_64bit)
3302                 ret = ath10k_htt_rx_pop_paddr64_list(htt, &resp->rx_in_ord_ind,
3303                                                      &list);
3304         else
3305                 ret = ath10k_htt_rx_pop_paddr32_list(htt, &resp->rx_in_ord_ind,
3306                                                      &list);
3307
3308         if (ret < 0) {
3309                 ath10k_warn(ar, "failed to pop paddr list: %d\n", ret);
3310                 htt->rx_confused = true;
3311                 return -EIO;
3312         }
3313
3314         /* Offloaded frames are very different and need to be handled
3315          * separately.
3316          */
3317         if (offload)
3318                 ath10k_htt_rx_h_rx_offload(ar, &list);
3319
3320         while (!skb_queue_empty(&list)) {
3321                 __skb_queue_head_init(&amsdu);
3322                 ret = ath10k_htt_rx_extract_amsdu(&ar->hw_params, &list, &amsdu);
3323                 switch (ret) {
3324                 case 0:
3325                         /* Note: The in-order indication may report interleaved
3326                          * frames from different PPDUs meaning reported rx rate
3327                          * to mac80211 isn't accurate/reliable. It's still
3328                          * better to report something than nothing though. This
3329                          * should still give an idea about rx rate to the user.
3330                          */
3331                         ath10k_htt_rx_h_ppdu(ar, &amsdu, status, vdev_id);
3332                         ath10k_htt_rx_h_filter(ar, &amsdu, status, NULL);
3333                         ath10k_htt_rx_h_mpdu(ar, &amsdu, status, false, NULL,
3334                                              NULL, peer_id, frag);
3335                         ath10k_htt_rx_h_enqueue(ar, &amsdu, status);
3336                         break;
3337                 case -EAGAIN:
3338                         fallthrough;
3339                 default:
3340                         /* Should not happen. */
3341                         ath10k_warn(ar, "failed to extract amsdu: %d\n", ret);
3342                         htt->rx_confused = true;
3343                         __skb_queue_purge(&list);
3344                         return -EIO;
3345                 }
3346         }
3347         return ret;
3348 }
3349
3350 static void ath10k_htt_rx_tx_fetch_resp_id_confirm(struct ath10k *ar,
3351                                                    const __le32 *resp_ids,
3352                                                    int num_resp_ids)
3353 {
3354         int i;
3355         u32 resp_id;
3356
3357         ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx tx fetch confirm num_resp_ids %d\n",
3358                    num_resp_ids);
3359
3360         for (i = 0; i < num_resp_ids; i++) {
3361                 resp_id = le32_to_cpu(resp_ids[i]);
3362
3363                 ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx tx fetch confirm resp_id %u\n",
3364                            resp_id);
3365
3366                 /* TODO: free resp_id */
3367         }
3368 }
3369
3370 static void ath10k_htt_rx_tx_fetch_ind(struct ath10k *ar, struct sk_buff *skb)
3371 {
3372         struct ieee80211_hw *hw = ar->hw;
3373         struct ieee80211_txq *txq;
3374         struct htt_resp *resp = (struct htt_resp *)skb->data;
3375         struct htt_tx_fetch_record *record;
3376         size_t len;
3377         size_t max_num_bytes;
3378         size_t max_num_msdus;
3379         size_t num_bytes;
3380         size_t num_msdus;
3381         const __le32 *resp_ids;
3382         u16 num_records;
3383         u16 num_resp_ids;
3384         u16 peer_id;
3385         u8 tid;
3386         int ret;
3387         int i;
3388         bool may_tx;
3389
3390         ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx tx fetch ind\n");
3391
3392         len = sizeof(resp->hdr) + sizeof(resp->tx_fetch_ind);
3393         if (unlikely(skb->len < len)) {
3394                 ath10k_warn(ar, "received corrupted tx_fetch_ind event: buffer too short\n");
3395                 return;
3396         }
3397
3398         num_records = le16_to_cpu(resp->tx_fetch_ind.num_records);
3399         num_resp_ids = le16_to_cpu(resp->tx_fetch_ind.num_resp_ids);
3400
3401         len += sizeof(resp->tx_fetch_ind.records[0]) * num_records;
3402         len += sizeof(resp->tx_fetch_ind.resp_ids[0]) * num_resp_ids;
3403
3404         if (unlikely(skb->len < len)) {
3405                 ath10k_warn(ar, "received corrupted tx_fetch_ind event: too many records/resp_ids\n");
3406                 return;
3407         }
3408
3409         ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx tx fetch ind num records %u num resps %u seq %u\n",
3410                    num_records, num_resp_ids,
3411                    le16_to_cpu(resp->tx_fetch_ind.fetch_seq_num));
3412
3413         if (!ar->htt.tx_q_state.enabled) {
3414                 ath10k_warn(ar, "received unexpected tx_fetch_ind event: not enabled\n");
3415                 return;
3416         }
3417
3418         if (ar->htt.tx_q_state.mode == HTT_TX_MODE_SWITCH_PUSH) {
3419                 ath10k_warn(ar, "received unexpected tx_fetch_ind event: in push mode\n");
3420                 return;
3421         }
3422
3423         rcu_read_lock();
3424
3425         for (i = 0; i < num_records; i++) {
3426                 record = &resp->tx_fetch_ind.records[i];
3427                 peer_id = MS(le16_to_cpu(record->info),
3428                              HTT_TX_FETCH_RECORD_INFO_PEER_ID);
3429                 tid = MS(le16_to_cpu(record->info),
3430                          HTT_TX_FETCH_RECORD_INFO_TID);
3431                 max_num_msdus = le16_to_cpu(record->num_msdus);
3432                 max_num_bytes = le32_to_cpu(record->num_bytes);
3433
3434                 ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx tx fetch record %i peer_id %u tid %u msdus %zu bytes %zu\n",
3435                            i, peer_id, tid, max_num_msdus, max_num_bytes);
3436
3437                 if (unlikely(peer_id >= ar->htt.tx_q_state.num_peers) ||
3438                     unlikely(tid >= ar->htt.tx_q_state.num_tids)) {
3439                         ath10k_warn(ar, "received out of range peer_id %u tid %u\n",
3440                                     peer_id, tid);
3441                         continue;
3442                 }
3443
3444                 spin_lock_bh(&ar->data_lock);
3445                 txq = ath10k_mac_txq_lookup(ar, peer_id, tid);
3446                 spin_unlock_bh(&ar->data_lock);
3447
3448                 /* It is okay to release the lock and use txq because RCU read
3449                  * lock is held.
3450                  */
3451
3452                 if (unlikely(!txq)) {
3453                         ath10k_warn(ar, "failed to lookup txq for peer_id %u tid %u\n",
3454                                     peer_id, tid);
3455                         continue;
3456                 }
3457
3458                 num_msdus = 0;
3459                 num_bytes = 0;
3460
3461                 ieee80211_txq_schedule_start(hw, txq->ac);
3462                 may_tx = ieee80211_txq_may_transmit(hw, txq);
3463                 while (num_msdus < max_num_msdus &&
3464                        num_bytes < max_num_bytes) {
3465                         if (!may_tx)
3466                                 break;
3467
3468                         ret = ath10k_mac_tx_push_txq(hw, txq);
3469                         if (ret < 0)
3470                                 break;
3471
3472                         num_msdus++;
3473                         num_bytes += ret;
3474                 }
3475                 ieee80211_return_txq(hw, txq, false);
3476                 ieee80211_txq_schedule_end(hw, txq->ac);
3477
3478                 record->num_msdus = cpu_to_le16(num_msdus);
3479                 record->num_bytes = cpu_to_le32(num_bytes);
3480
3481                 ath10k_htt_tx_txq_recalc(hw, txq);
3482         }
3483
3484         rcu_read_unlock();
3485
3486         resp_ids = ath10k_htt_get_tx_fetch_ind_resp_ids(&resp->tx_fetch_ind);
3487         ath10k_htt_rx_tx_fetch_resp_id_confirm(ar, resp_ids, num_resp_ids);
3488
3489         ret = ath10k_htt_tx_fetch_resp(ar,
3490                                        resp->tx_fetch_ind.token,
3491                                        resp->tx_fetch_ind.fetch_seq_num,
3492                                        resp->tx_fetch_ind.records,
3493                                        num_records);
3494         if (unlikely(ret)) {
3495                 ath10k_warn(ar, "failed to submit tx fetch resp for token 0x%08x: %d\n",
3496                             le32_to_cpu(resp->tx_fetch_ind.token), ret);
3497                 /* FIXME: request fw restart */
3498         }
3499
3500         ath10k_htt_tx_txq_sync(ar);
3501 }
3502
3503 static void ath10k_htt_rx_tx_fetch_confirm(struct ath10k *ar,
3504                                            struct sk_buff *skb)
3505 {
3506         const struct htt_resp *resp = (void *)skb->data;
3507         size_t len;
3508         int num_resp_ids;
3509
3510         ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx tx fetch confirm\n");
3511
3512         len = sizeof(resp->hdr) + sizeof(resp->tx_fetch_confirm);
3513         if (unlikely(skb->len < len)) {
3514                 ath10k_warn(ar, "received corrupted tx_fetch_confirm event: buffer too short\n");
3515                 return;
3516         }
3517
3518         num_resp_ids = le16_to_cpu(resp->tx_fetch_confirm.num_resp_ids);
3519         len += sizeof(resp->tx_fetch_confirm.resp_ids[0]) * num_resp_ids;
3520
3521         if (unlikely(skb->len < len)) {
3522                 ath10k_warn(ar, "received corrupted tx_fetch_confirm event: resp_ids buffer overflow\n");
3523                 return;
3524         }
3525
3526         ath10k_htt_rx_tx_fetch_resp_id_confirm(ar,
3527                                                resp->tx_fetch_confirm.resp_ids,
3528                                                num_resp_ids);
3529 }
3530
3531 static void ath10k_htt_rx_tx_mode_switch_ind(struct ath10k *ar,
3532                                              struct sk_buff *skb)
3533 {
3534         const struct htt_resp *resp = (void *)skb->data;
3535         const struct htt_tx_mode_switch_record *record;
3536         struct ieee80211_txq *txq;
3537         struct ath10k_txq *artxq;
3538         size_t len;
3539         size_t num_records;
3540         enum htt_tx_mode_switch_mode mode;
3541         bool enable;
3542         u16 info0;
3543         u16 info1;
3544         u16 threshold;
3545         u16 peer_id;
3546         u8 tid;
3547         int i;
3548
3549         ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx tx mode switch ind\n");
3550
3551         len = sizeof(resp->hdr) + sizeof(resp->tx_mode_switch_ind);
3552         if (unlikely(skb->len < len)) {
3553                 ath10k_warn(ar, "received corrupted tx_mode_switch_ind event: buffer too short\n");
3554                 return;
3555         }
3556
3557         info0 = le16_to_cpu(resp->tx_mode_switch_ind.info0);
3558         info1 = le16_to_cpu(resp->tx_mode_switch_ind.info1);
3559
3560         enable = !!(info0 & HTT_TX_MODE_SWITCH_IND_INFO0_ENABLE);
3561         num_records = MS(info0, HTT_TX_MODE_SWITCH_IND_INFO1_THRESHOLD);
3562         mode = MS(info1, HTT_TX_MODE_SWITCH_IND_INFO1_MODE);
3563         threshold = MS(info1, HTT_TX_MODE_SWITCH_IND_INFO1_THRESHOLD);
3564
3565         ath10k_dbg(ar, ATH10K_DBG_HTT,
3566                    "htt rx tx mode switch ind info0 0x%04hx info1 0x%04x enable %d num records %zd mode %d threshold %u\n",
3567                    info0, info1, enable, num_records, mode, threshold);
3568
3569         len += sizeof(resp->tx_mode_switch_ind.records[0]) * num_records;
3570
3571         if (unlikely(skb->len < len)) {
3572                 ath10k_warn(ar, "received corrupted tx_mode_switch_mode_ind event: too many records\n");
3573                 return;
3574         }
3575
3576         switch (mode) {
3577         case HTT_TX_MODE_SWITCH_PUSH:
3578         case HTT_TX_MODE_SWITCH_PUSH_PULL:
3579                 break;
3580         default:
3581                 ath10k_warn(ar, "received invalid tx_mode_switch_mode_ind mode %d, ignoring\n",
3582                             mode);
3583                 return;
3584         }
3585
3586         if (!enable)
3587                 return;
3588
3589         ar->htt.tx_q_state.enabled = enable;
3590         ar->htt.tx_q_state.mode = mode;
3591         ar->htt.tx_q_state.num_push_allowed = threshold;
3592
3593         rcu_read_lock();
3594
3595         for (i = 0; i < num_records; i++) {
3596                 record = &resp->tx_mode_switch_ind.records[i];
3597                 info0 = le16_to_cpu(record->info0);
3598                 peer_id = MS(info0, HTT_TX_MODE_SWITCH_RECORD_INFO0_PEER_ID);
3599                 tid = MS(info0, HTT_TX_MODE_SWITCH_RECORD_INFO0_TID);
3600
3601                 if (unlikely(peer_id >= ar->htt.tx_q_state.num_peers) ||
3602                     unlikely(tid >= ar->htt.tx_q_state.num_tids)) {
3603                         ath10k_warn(ar, "received out of range peer_id %u tid %u\n",
3604                                     peer_id, tid);
3605                         continue;
3606                 }
3607
3608                 spin_lock_bh(&ar->data_lock);
3609                 txq = ath10k_mac_txq_lookup(ar, peer_id, tid);
3610                 spin_unlock_bh(&ar->data_lock);
3611
3612                 /* It is okay to release the lock and use txq because RCU read
3613                  * lock is held.
3614                  */
3615
3616                 if (unlikely(!txq)) {
3617                         ath10k_warn(ar, "failed to lookup txq for peer_id %u tid %u\n",
3618                                     peer_id, tid);
3619                         continue;
3620                 }
3621
3622                 spin_lock_bh(&ar->htt.tx_lock);
3623                 artxq = (void *)txq->drv_priv;
3624                 artxq->num_push_allowed = le16_to_cpu(record->num_max_msdus);
3625                 spin_unlock_bh(&ar->htt.tx_lock);
3626         }
3627
3628         rcu_read_unlock();
3629
3630         ath10k_mac_tx_push_pending(ar);
3631 }
3632
3633 void ath10k_htt_htc_t2h_msg_handler(struct ath10k *ar, struct sk_buff *skb)
3634 {
3635         bool release;
3636
3637         release = ath10k_htt_t2h_msg_handler(ar, skb);
3638
3639         /* Free the indication buffer */
3640         if (release)
3641                 dev_kfree_skb_any(skb);
3642 }
3643
3644 static inline s8 ath10k_get_legacy_rate_idx(struct ath10k *ar, u8 rate)
3645 {
3646         static const u8 legacy_rates[] = {1, 2, 5, 11, 6, 9, 12,
3647                                           18, 24, 36, 48, 54};
3648         int i;
3649
3650         for (i = 0; i < ARRAY_SIZE(legacy_rates); i++) {
3651                 if (rate == legacy_rates[i])
3652                         return i;
3653         }
3654
3655         ath10k_warn(ar, "Invalid legacy rate %d peer stats", rate);
3656         return -EINVAL;
3657 }
3658
3659 static void
3660 ath10k_accumulate_per_peer_tx_stats(struct ath10k *ar,
3661                                     struct ath10k_sta *arsta,
3662                                     struct ath10k_per_peer_tx_stats *pstats,
3663                                     s8 legacy_rate_idx)
3664 {
3665         struct rate_info *txrate = &arsta->txrate;
3666         struct ath10k_htt_tx_stats *tx_stats;
3667         int idx, ht_idx, gi, mcs, bw, nss;
3668         unsigned long flags;
3669
3670         if (!arsta->tx_stats)
3671                 return;
3672
3673         tx_stats = arsta->tx_stats;
3674         flags = txrate->flags;
3675         gi = test_bit(ATH10K_RATE_INFO_FLAGS_SGI_BIT, &flags);
3676         mcs = ATH10K_HW_MCS_RATE(pstats->ratecode);
3677         bw = txrate->bw;
3678         nss = txrate->nss;
3679         ht_idx = mcs + (nss - 1) * 8;
3680         idx = mcs * 8 + 8 * 10 * (nss - 1);
3681         idx += bw * 2 + gi;
3682
3683 #define STATS_OP_FMT(name) tx_stats->stats[ATH10K_STATS_TYPE_##name]
3684
3685         if (txrate->flags & RATE_INFO_FLAGS_VHT_MCS) {
3686                 STATS_OP_FMT(SUCC).vht[0][mcs] += pstats->succ_bytes;
3687                 STATS_OP_FMT(SUCC).vht[1][mcs] += pstats->succ_pkts;
3688                 STATS_OP_FMT(FAIL).vht[0][mcs] += pstats->failed_bytes;
3689                 STATS_OP_FMT(FAIL).vht[1][mcs] += pstats->failed_pkts;
3690                 STATS_OP_FMT(RETRY).vht[0][mcs] += pstats->retry_bytes;
3691                 STATS_OP_FMT(RETRY).vht[1][mcs] += pstats->retry_pkts;
3692         } else if (txrate->flags & RATE_INFO_FLAGS_MCS) {
3693                 STATS_OP_FMT(SUCC).ht[0][ht_idx] += pstats->succ_bytes;
3694                 STATS_OP_FMT(SUCC).ht[1][ht_idx] += pstats->succ_pkts;
3695                 STATS_OP_FMT(FAIL).ht[0][ht_idx] += pstats->failed_bytes;
3696                 STATS_OP_FMT(FAIL).ht[1][ht_idx] += pstats->failed_pkts;
3697                 STATS_OP_FMT(RETRY).ht[0][ht_idx] += pstats->retry_bytes;
3698                 STATS_OP_FMT(RETRY).ht[1][ht_idx] += pstats->retry_pkts;
3699         } else {
3700                 mcs = legacy_rate_idx;
3701
3702                 STATS_OP_FMT(SUCC).legacy[0][mcs] += pstats->succ_bytes;
3703                 STATS_OP_FMT(SUCC).legacy[1][mcs] += pstats->succ_pkts;
3704                 STATS_OP_FMT(FAIL).legacy[0][mcs] += pstats->failed_bytes;
3705                 STATS_OP_FMT(FAIL).legacy[1][mcs] += pstats->failed_pkts;
3706                 STATS_OP_FMT(RETRY).legacy[0][mcs] += pstats->retry_bytes;
3707                 STATS_OP_FMT(RETRY).legacy[1][mcs] += pstats->retry_pkts;
3708         }
3709
3710         if (ATH10K_HW_AMPDU(pstats->flags)) {
3711                 tx_stats->ba_fails += ATH10K_HW_BA_FAIL(pstats->flags);
3712
3713                 if (txrate->flags & RATE_INFO_FLAGS_MCS) {
3714                         STATS_OP_FMT(AMPDU).ht[0][ht_idx] +=
3715                                 pstats->succ_bytes + pstats->retry_bytes;
3716                         STATS_OP_FMT(AMPDU).ht[1][ht_idx] +=
3717                                 pstats->succ_pkts + pstats->retry_pkts;
3718                 } else {
3719                         STATS_OP_FMT(AMPDU).vht[0][mcs] +=
3720                                 pstats->succ_bytes + pstats->retry_bytes;
3721                         STATS_OP_FMT(AMPDU).vht[1][mcs] +=
3722                                 pstats->succ_pkts + pstats->retry_pkts;
3723                 }
3724                 STATS_OP_FMT(AMPDU).bw[0][bw] +=
3725                         pstats->succ_bytes + pstats->retry_bytes;
3726                 STATS_OP_FMT(AMPDU).nss[0][nss - 1] +=
3727                         pstats->succ_bytes + pstats->retry_bytes;
3728                 STATS_OP_FMT(AMPDU).gi[0][gi] +=
3729                         pstats->succ_bytes + pstats->retry_bytes;
3730                 STATS_OP_FMT(AMPDU).rate_table[0][idx] +=
3731                         pstats->succ_bytes + pstats->retry_bytes;
3732                 STATS_OP_FMT(AMPDU).bw[1][bw] +=
3733                         pstats->succ_pkts + pstats->retry_pkts;
3734                 STATS_OP_FMT(AMPDU).nss[1][nss - 1] +=
3735                         pstats->succ_pkts + pstats->retry_pkts;
3736                 STATS_OP_FMT(AMPDU).gi[1][gi] +=
3737                         pstats->succ_pkts + pstats->retry_pkts;
3738                 STATS_OP_FMT(AMPDU).rate_table[1][idx] +=
3739                         pstats->succ_pkts + pstats->retry_pkts;
3740         } else {
3741                 tx_stats->ack_fails +=
3742                                 ATH10K_HW_BA_FAIL(pstats->flags);
3743         }
3744
3745         STATS_OP_FMT(SUCC).bw[0][bw] += pstats->succ_bytes;
3746         STATS_OP_FMT(SUCC).nss[0][nss - 1] += pstats->succ_bytes;
3747         STATS_OP_FMT(SUCC).gi[0][gi] += pstats->succ_bytes;
3748
3749         STATS_OP_FMT(SUCC).bw[1][bw] += pstats->succ_pkts;
3750         STATS_OP_FMT(SUCC).nss[1][nss - 1] += pstats->succ_pkts;
3751         STATS_OP_FMT(SUCC).gi[1][gi] += pstats->succ_pkts;
3752
3753         STATS_OP_FMT(FAIL).bw[0][bw] += pstats->failed_bytes;
3754         STATS_OP_FMT(FAIL).nss[0][nss - 1] += pstats->failed_bytes;
3755         STATS_OP_FMT(FAIL).gi[0][gi] += pstats->failed_bytes;
3756
3757         STATS_OP_FMT(FAIL).bw[1][bw] += pstats->failed_pkts;
3758         STATS_OP_FMT(FAIL).nss[1][nss - 1] += pstats->failed_pkts;
3759         STATS_OP_FMT(FAIL).gi[1][gi] += pstats->failed_pkts;
3760
3761         STATS_OP_FMT(RETRY).bw[0][bw] += pstats->retry_bytes;
3762         STATS_OP_FMT(RETRY).nss[0][nss - 1] += pstats->retry_bytes;
3763         STATS_OP_FMT(RETRY).gi[0][gi] += pstats->retry_bytes;
3764
3765         STATS_OP_FMT(RETRY).bw[1][bw] += pstats->retry_pkts;
3766         STATS_OP_FMT(RETRY).nss[1][nss - 1] += pstats->retry_pkts;
3767         STATS_OP_FMT(RETRY).gi[1][gi] += pstats->retry_pkts;
3768
3769         if (txrate->flags >= RATE_INFO_FLAGS_MCS) {
3770                 STATS_OP_FMT(SUCC).rate_table[0][idx] += pstats->succ_bytes;
3771                 STATS_OP_FMT(SUCC).rate_table[1][idx] += pstats->succ_pkts;
3772                 STATS_OP_FMT(FAIL).rate_table[0][idx] += pstats->failed_bytes;
3773                 STATS_OP_FMT(FAIL).rate_table[1][idx] += pstats->failed_pkts;
3774                 STATS_OP_FMT(RETRY).rate_table[0][idx] += pstats->retry_bytes;
3775                 STATS_OP_FMT(RETRY).rate_table[1][idx] += pstats->retry_pkts;
3776         }
3777
3778         tx_stats->tx_duration += pstats->duration;
3779 }
3780
3781 static void
3782 ath10k_update_per_peer_tx_stats(struct ath10k *ar,
3783                                 struct ieee80211_sta *sta,
3784                                 struct ath10k_per_peer_tx_stats *peer_stats)
3785 {
3786         struct ath10k_sta *arsta = (struct ath10k_sta *)sta->drv_priv;
3787         struct ieee80211_chanctx_conf *conf = NULL;
3788         u8 rate = 0, sgi;
3789         s8 rate_idx = 0;
3790         bool skip_auto_rate;
3791         struct rate_info txrate;
3792
3793         lockdep_assert_held(&ar->data_lock);
3794
3795         txrate.flags = ATH10K_HW_PREAMBLE(peer_stats->ratecode);
3796         txrate.bw = ATH10K_HW_BW(peer_stats->flags);
3797         txrate.nss = ATH10K_HW_NSS(peer_stats->ratecode);
3798         txrate.mcs = ATH10K_HW_MCS_RATE(peer_stats->ratecode);
3799         sgi = ATH10K_HW_GI(peer_stats->flags);
3800         skip_auto_rate = ATH10K_FW_SKIPPED_RATE_CTRL(peer_stats->flags);
3801
3802         /* Firmware's rate control skips broadcast/management frames,
3803          * if host has configure fixed rates and in some other special cases.
3804          */
3805         if (skip_auto_rate)
3806                 return;
3807
3808         if (txrate.flags == WMI_RATE_PREAMBLE_VHT && txrate.mcs > 9) {
3809                 ath10k_warn(ar, "Invalid VHT mcs %d peer stats",  txrate.mcs);
3810                 return;
3811         }
3812
3813         if (txrate.flags == WMI_RATE_PREAMBLE_HT &&
3814             (txrate.mcs > 7 || txrate.nss < 1)) {
3815                 ath10k_warn(ar, "Invalid HT mcs %d nss %d peer stats",
3816                             txrate.mcs, txrate.nss);
3817                 return;
3818         }
3819
3820         memset(&arsta->txrate, 0, sizeof(arsta->txrate));
3821         memset(&arsta->tx_info.status, 0, sizeof(arsta->tx_info.status));
3822         if (txrate.flags == WMI_RATE_PREAMBLE_CCK ||
3823             txrate.flags == WMI_RATE_PREAMBLE_OFDM) {
3824                 rate = ATH10K_HW_LEGACY_RATE(peer_stats->ratecode);
3825                 /* This is hacky, FW sends CCK rate 5.5Mbps as 6 */
3826                 if (rate == 6 && txrate.flags == WMI_RATE_PREAMBLE_CCK)
3827                         rate = 5;
3828                 rate_idx = ath10k_get_legacy_rate_idx(ar, rate);
3829                 if (rate_idx < 0)
3830                         return;
3831                 arsta->txrate.legacy = rate;
3832         } else if (txrate.flags == WMI_RATE_PREAMBLE_HT) {
3833                 arsta->txrate.flags = RATE_INFO_FLAGS_MCS;
3834                 arsta->txrate.mcs = txrate.mcs + 8 * (txrate.nss - 1);
3835         } else {
3836                 arsta->txrate.flags = RATE_INFO_FLAGS_VHT_MCS;
3837                 arsta->txrate.mcs = txrate.mcs;
3838         }
3839
3840         switch (txrate.flags) {
3841         case WMI_RATE_PREAMBLE_OFDM:
3842                 if (arsta->arvif && arsta->arvif->vif)
3843                         conf = rcu_dereference(arsta->arvif->vif->chanctx_conf);
3844                 if (conf && conf->def.chan->band == NL80211_BAND_5GHZ)
3845                         arsta->tx_info.status.rates[0].idx = rate_idx - 4;
3846                 break;
3847         case WMI_RATE_PREAMBLE_CCK:
3848                 arsta->tx_info.status.rates[0].idx = rate_idx;
3849                 if (sgi)
3850                         arsta->tx_info.status.rates[0].flags |=
3851                                 (IEEE80211_TX_RC_USE_SHORT_PREAMBLE |
3852                                  IEEE80211_TX_RC_SHORT_GI);
3853                 break;
3854         case WMI_RATE_PREAMBLE_HT:
3855                 arsta->tx_info.status.rates[0].idx =
3856                                 txrate.mcs + ((txrate.nss - 1) * 8);
3857                 if (sgi)
3858                         arsta->tx_info.status.rates[0].flags |=
3859                                         IEEE80211_TX_RC_SHORT_GI;
3860                 arsta->tx_info.status.rates[0].flags |= IEEE80211_TX_RC_MCS;
3861                 break;
3862         case WMI_RATE_PREAMBLE_VHT:
3863                 ieee80211_rate_set_vht(&arsta->tx_info.status.rates[0],
3864                                        txrate.mcs, txrate.nss);
3865                 if (sgi)
3866                         arsta->tx_info.status.rates[0].flags |=
3867                                                 IEEE80211_TX_RC_SHORT_GI;
3868                 arsta->tx_info.status.rates[0].flags |= IEEE80211_TX_RC_VHT_MCS;
3869                 break;
3870         }
3871
3872         arsta->txrate.nss = txrate.nss;
3873         arsta->txrate.bw = ath10k_bw_to_mac80211_bw(txrate.bw);
3874         arsta->last_tx_bitrate = cfg80211_calculate_bitrate(&arsta->txrate);
3875         if (sgi)
3876                 arsta->txrate.flags |= RATE_INFO_FLAGS_SHORT_GI;
3877
3878         switch (arsta->txrate.bw) {
3879         case RATE_INFO_BW_40:
3880                 arsta->tx_info.status.rates[0].flags |=
3881                                 IEEE80211_TX_RC_40_MHZ_WIDTH;
3882                 break;
3883         case RATE_INFO_BW_80:
3884                 arsta->tx_info.status.rates[0].flags |=
3885                                 IEEE80211_TX_RC_80_MHZ_WIDTH;
3886                 break;
3887         }
3888
3889         if (peer_stats->succ_pkts) {
3890                 arsta->tx_info.flags = IEEE80211_TX_STAT_ACK;
3891                 arsta->tx_info.status.rates[0].count = 1;
3892                 ieee80211_tx_rate_update(ar->hw, sta, &arsta->tx_info);
3893         }
3894
3895         if (ar->htt.disable_tx_comp) {
3896                 arsta->tx_failed += peer_stats->failed_pkts;
3897                 ath10k_dbg(ar, ATH10K_DBG_HTT, "tx failed %d\n",
3898                            arsta->tx_failed);
3899         }
3900
3901         arsta->tx_retries += peer_stats->retry_pkts;
3902         ath10k_dbg(ar, ATH10K_DBG_HTT, "htt tx retries %d", arsta->tx_retries);
3903
3904         if (ath10k_debug_is_extd_tx_stats_enabled(ar))
3905                 ath10k_accumulate_per_peer_tx_stats(ar, arsta, peer_stats,
3906                                                     rate_idx);
3907 }
3908
3909 static void ath10k_htt_fetch_peer_stats(struct ath10k *ar,
3910                                         struct sk_buff *skb)
3911 {
3912         struct htt_resp *resp = (struct htt_resp *)skb->data;
3913         struct ath10k_per_peer_tx_stats *p_tx_stats = &ar->peer_tx_stats;
3914         struct htt_per_peer_tx_stats_ind *tx_stats;
3915         struct ieee80211_sta *sta;
3916         struct ath10k_peer *peer;
3917         int peer_id, i;
3918         u8 ppdu_len, num_ppdu;
3919
3920         num_ppdu = resp->peer_tx_stats.num_ppdu;
3921         ppdu_len = resp->peer_tx_stats.ppdu_len * sizeof(__le32);
3922
3923         if (skb->len < sizeof(struct htt_resp_hdr) + num_ppdu * ppdu_len) {
3924                 ath10k_warn(ar, "Invalid peer stats buf length %d\n", skb->len);
3925                 return;
3926         }
3927
3928         tx_stats = (struct htt_per_peer_tx_stats_ind *)
3929                         (resp->peer_tx_stats.payload);
3930         peer_id = __le16_to_cpu(tx_stats->peer_id);
3931
3932         rcu_read_lock();
3933         spin_lock_bh(&ar->data_lock);
3934         peer = ath10k_peer_find_by_id(ar, peer_id);
3935         if (!peer || !peer->sta) {
3936                 ath10k_warn(ar, "Invalid peer id %d peer stats buffer\n",
3937                             peer_id);
3938                 goto out;
3939         }
3940
3941         sta = peer->sta;
3942         for (i = 0; i < num_ppdu; i++) {
3943                 tx_stats = (struct htt_per_peer_tx_stats_ind *)
3944                            (resp->peer_tx_stats.payload + i * ppdu_len);
3945
3946                 p_tx_stats->succ_bytes = __le32_to_cpu(tx_stats->succ_bytes);
3947                 p_tx_stats->retry_bytes = __le32_to_cpu(tx_stats->retry_bytes);
3948                 p_tx_stats->failed_bytes =
3949                                 __le32_to_cpu(tx_stats->failed_bytes);
3950                 p_tx_stats->ratecode = tx_stats->ratecode;
3951                 p_tx_stats->flags = tx_stats->flags;
3952                 p_tx_stats->succ_pkts = __le16_to_cpu(tx_stats->succ_pkts);
3953                 p_tx_stats->retry_pkts = __le16_to_cpu(tx_stats->retry_pkts);
3954                 p_tx_stats->failed_pkts = __le16_to_cpu(tx_stats->failed_pkts);
3955                 p_tx_stats->duration = __le16_to_cpu(tx_stats->tx_duration);
3956
3957                 ath10k_update_per_peer_tx_stats(ar, sta, p_tx_stats);
3958         }
3959
3960 out:
3961         spin_unlock_bh(&ar->data_lock);
3962         rcu_read_unlock();
3963 }
3964
3965 static void ath10k_fetch_10_2_tx_stats(struct ath10k *ar, u8 *data)
3966 {
3967         struct ath10k_pktlog_hdr *hdr = (struct ath10k_pktlog_hdr *)data;
3968         struct ath10k_per_peer_tx_stats *p_tx_stats = &ar->peer_tx_stats;
3969         struct ath10k_10_2_peer_tx_stats *tx_stats;
3970         struct ieee80211_sta *sta;
3971         struct ath10k_peer *peer;
3972         u16 log_type = __le16_to_cpu(hdr->log_type);
3973         u32 peer_id = 0, i;
3974
3975         if (log_type != ATH_PKTLOG_TYPE_TX_STAT)
3976                 return;
3977
3978         tx_stats = (struct ath10k_10_2_peer_tx_stats *)((hdr->payload) +
3979                     ATH10K_10_2_TX_STATS_OFFSET);
3980
3981         if (!tx_stats->tx_ppdu_cnt)
3982                 return;
3983
3984         peer_id = tx_stats->peer_id;
3985
3986         rcu_read_lock();
3987         spin_lock_bh(&ar->data_lock);
3988         peer = ath10k_peer_find_by_id(ar, peer_id);
3989         if (!peer || !peer->sta) {
3990                 ath10k_warn(ar, "Invalid peer id %d in peer stats buffer\n",
3991                             peer_id);
3992                 goto out;
3993         }
3994
3995         sta = peer->sta;
3996         for (i = 0; i < tx_stats->tx_ppdu_cnt; i++) {
3997                 p_tx_stats->succ_bytes =
3998                         __le16_to_cpu(tx_stats->success_bytes[i]);
3999                 p_tx_stats->retry_bytes =
4000                         __le16_to_cpu(tx_stats->retry_bytes[i]);
4001                 p_tx_stats->failed_bytes =
4002                         __le16_to_cpu(tx_stats->failed_bytes[i]);
4003                 p_tx_stats->ratecode = tx_stats->ratecode[i];
4004                 p_tx_stats->flags = tx_stats->flags[i];
4005                 p_tx_stats->succ_pkts = tx_stats->success_pkts[i];
4006                 p_tx_stats->retry_pkts = tx_stats->retry_pkts[i];
4007                 p_tx_stats->failed_pkts = tx_stats->failed_pkts[i];
4008
4009                 ath10k_update_per_peer_tx_stats(ar, sta, p_tx_stats);
4010         }
4011         spin_unlock_bh(&ar->data_lock);
4012         rcu_read_unlock();
4013
4014         return;
4015
4016 out:
4017         spin_unlock_bh(&ar->data_lock);
4018         rcu_read_unlock();
4019 }
4020
4021 static int ath10k_htt_rx_pn_len(enum htt_security_types sec_type)
4022 {
4023         switch (sec_type) {
4024         case HTT_SECURITY_TKIP:
4025         case HTT_SECURITY_TKIP_NOMIC:
4026         case HTT_SECURITY_AES_CCMP:
4027                 return 48;
4028         default:
4029                 return 0;
4030         }
4031 }
4032
4033 static void ath10k_htt_rx_sec_ind_handler(struct ath10k *ar,
4034                                           struct htt_security_indication *ev)
4035 {
4036         enum htt_txrx_sec_cast_type sec_index;
4037         enum htt_security_types sec_type;
4038         struct ath10k_peer *peer;
4039
4040         spin_lock_bh(&ar->data_lock);
4041
4042         peer = ath10k_peer_find_by_id(ar, __le16_to_cpu(ev->peer_id));
4043         if (!peer) {
4044                 ath10k_warn(ar, "failed to find peer id %d for security indication",
4045                             __le16_to_cpu(ev->peer_id));
4046                 goto out;
4047         }
4048
4049         sec_type = MS(ev->flags, HTT_SECURITY_TYPE);
4050
4051         if (ev->flags & HTT_SECURITY_IS_UNICAST)
4052                 sec_index = HTT_TXRX_SEC_UCAST;
4053         else
4054                 sec_index = HTT_TXRX_SEC_MCAST;
4055
4056         peer->rx_pn[sec_index].sec_type = sec_type;
4057         peer->rx_pn[sec_index].pn_len = ath10k_htt_rx_pn_len(sec_type);
4058
4059         memset(peer->tids_last_pn_valid, 0, sizeof(peer->tids_last_pn_valid));
4060         memset(peer->tids_last_pn, 0, sizeof(peer->tids_last_pn));
4061
4062 out:
4063         spin_unlock_bh(&ar->data_lock);
4064 }
4065
4066 bool ath10k_htt_t2h_msg_handler(struct ath10k *ar, struct sk_buff *skb)
4067 {
4068         struct ath10k_htt *htt = &ar->htt;
4069         struct htt_resp *resp = (struct htt_resp *)skb->data;
4070         enum htt_t2h_msg_type type;
4071
4072         /* confirm alignment */
4073         if (!IS_ALIGNED((unsigned long)skb->data, 4))
4074                 ath10k_warn(ar, "unaligned htt message, expect trouble\n");
4075
4076         ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx, msg_type: 0x%0X\n",
4077                    resp->hdr.msg_type);
4078
4079         if (resp->hdr.msg_type >= ar->htt.t2h_msg_types_max) {
4080                 ath10k_dbg(ar, ATH10K_DBG_HTT, "htt rx, unsupported msg_type: 0x%0X\n max: 0x%0X",
4081                            resp->hdr.msg_type, ar->htt.t2h_msg_types_max);
4082                 return true;
4083         }
4084         type = ar->htt.t2h_msg_types[resp->hdr.msg_type];
4085
4086         switch (type) {
4087         case HTT_T2H_MSG_TYPE_VERSION_CONF: {
4088                 htt->target_version_major = resp->ver_resp.major;
4089                 htt->target_version_minor = resp->ver_resp.minor;
4090                 complete(&htt->target_version_received);
4091                 break;
4092         }
4093         case HTT_T2H_MSG_TYPE_RX_IND:
4094                 if (ar->bus_param.dev_type != ATH10K_DEV_TYPE_HL) {
4095                         ath10k_htt_rx_proc_rx_ind_ll(htt, &resp->rx_ind);
4096                 } else {
4097                         skb_queue_tail(&htt->rx_indication_head, skb);
4098                         return false;
4099                 }
4100                 break;
4101         case HTT_T2H_MSG_TYPE_PEER_MAP: {
4102                 struct htt_peer_map_event ev = {
4103                         .vdev_id = resp->peer_map.vdev_id,
4104                         .peer_id = __le16_to_cpu(resp->peer_map.peer_id),
4105                 };
4106                 memcpy(ev.addr, resp->peer_map.addr, sizeof(ev.addr));
4107                 ath10k_peer_map_event(htt, &ev);
4108                 break;
4109         }
4110         case HTT_T2H_MSG_TYPE_PEER_UNMAP: {
4111                 struct htt_peer_unmap_event ev = {
4112                         .peer_id = __le16_to_cpu(resp->peer_unmap.peer_id),
4113                 };
4114                 ath10k_peer_unmap_event(htt, &ev);
4115                 break;
4116         }
4117         case HTT_T2H_MSG_TYPE_MGMT_TX_COMPLETION: {
4118                 struct htt_tx_done tx_done = {};
4119                 struct ath10k_htt *htt = &ar->htt;
4120                 struct ath10k_htc *htc = &ar->htc;
4121                 struct ath10k_htc_ep *ep = &ar->htc.endpoint[htt->eid];
4122                 int status = __le32_to_cpu(resp->mgmt_tx_completion.status);
4123                 int info = __le32_to_cpu(resp->mgmt_tx_completion.info);
4124
4125                 tx_done.msdu_id = __le32_to_cpu(resp->mgmt_tx_completion.desc_id);
4126
4127                 switch (status) {
4128                 case HTT_MGMT_TX_STATUS_OK:
4129                         tx_done.status = HTT_TX_COMPL_STATE_ACK;
4130                         if (test_bit(WMI_SERVICE_HTT_MGMT_TX_COMP_VALID_FLAGS,
4131                                      ar->wmi.svc_map) &&
4132                             (resp->mgmt_tx_completion.flags &
4133                              HTT_MGMT_TX_CMPL_FLAG_ACK_RSSI)) {
4134                                 tx_done.ack_rssi =
4135                                 FIELD_GET(HTT_MGMT_TX_CMPL_INFO_ACK_RSSI_MASK,
4136                                           info);
4137                         }
4138                         break;
4139                 case HTT_MGMT_TX_STATUS_RETRY:
4140                         tx_done.status = HTT_TX_COMPL_STATE_NOACK;
4141                         break;
4142                 case HTT_MGMT_TX_STATUS_DROP:
4143                         tx_done.status = HTT_TX_COMPL_STATE_DISCARD;
4144                         break;
4145                 }
4146
4147                 if (htt->disable_tx_comp) {
4148                         spin_lock_bh(&htc->tx_lock);
4149                         ep->tx_credits++;
4150                         spin_unlock_bh(&htc->tx_lock);
4151                 }
4152
4153                 status = ath10k_txrx_tx_unref(htt, &tx_done);
4154                 if (!status) {
4155                         spin_lock_bh(&htt->tx_lock);
4156                         ath10k_htt_tx_mgmt_dec_pending(htt);
4157                         spin_unlock_bh(&htt->tx_lock);
4158                 }
4159                 break;
4160         }
4161         case HTT_T2H_MSG_TYPE_TX_COMPL_IND:
4162                 ath10k_htt_rx_tx_compl_ind(htt->ar, skb);
4163                 break;
4164         case HTT_T2H_MSG_TYPE_SEC_IND: {
4165                 struct ath10k *ar = htt->ar;
4166                 struct htt_security_indication *ev = &resp->security_indication;
4167
4168                 ath10k_htt_rx_sec_ind_handler(ar, ev);
4169                 ath10k_dbg(ar, ATH10K_DBG_HTT,
4170                            "sec ind peer_id %d unicast %d type %d\n",
4171                           __le16_to_cpu(ev->peer_id),
4172                           !!(ev->flags & HTT_SECURITY_IS_UNICAST),
4173                           MS(ev->flags, HTT_SECURITY_TYPE));
4174                 complete(&ar->install_key_done);
4175                 break;
4176         }
4177         case HTT_T2H_MSG_TYPE_RX_FRAG_IND: {
4178                 ath10k_dbg_dump(ar, ATH10K_DBG_HTT_DUMP, NULL, "htt event: ",
4179                                 skb->data, skb->len);
4180                 atomic_inc(&htt->num_mpdus_ready);
4181
4182                 return ath10k_htt_rx_proc_rx_frag_ind(htt,
4183                                                       &resp->rx_frag_ind,
4184                                                       skb);
4185         }
4186         case HTT_T2H_MSG_TYPE_TEST:
4187                 break;
4188         case HTT_T2H_MSG_TYPE_STATS_CONF:
4189                 trace_ath10k_htt_stats(ar, skb->data, skb->len);
4190                 break;
4191         case HTT_T2H_MSG_TYPE_TX_INSPECT_IND:
4192                 /* Firmware can return tx frames if it's unable to fully
4193                  * process them and suspects host may be able to fix it. ath10k
4194                  * sends all tx frames as already inspected so this shouldn't
4195                  * happen unless fw has a bug.
4196                  */
4197                 ath10k_warn(ar, "received an unexpected htt tx inspect event\n");
4198                 break;
4199         case HTT_T2H_MSG_TYPE_RX_ADDBA:
4200                 ath10k_htt_rx_addba(ar, resp);
4201                 break;
4202         case HTT_T2H_MSG_TYPE_RX_DELBA:
4203                 ath10k_htt_rx_delba(ar, resp);
4204                 break;
4205         case HTT_T2H_MSG_TYPE_PKTLOG: {
4206                 trace_ath10k_htt_pktlog(ar, resp->pktlog_msg.payload,
4207                                         skb->len -
4208                                         offsetof(struct htt_resp,
4209                                                  pktlog_msg.payload));
4210
4211                 if (ath10k_peer_stats_enabled(ar))
4212                         ath10k_fetch_10_2_tx_stats(ar,
4213                                                    resp->pktlog_msg.payload);
4214                 break;
4215         }
4216         case HTT_T2H_MSG_TYPE_RX_FLUSH: {
4217                 /* Ignore this event because mac80211 takes care of Rx
4218                  * aggregation reordering.
4219                  */
4220                 break;
4221         }
4222         case HTT_T2H_MSG_TYPE_RX_IN_ORD_PADDR_IND: {
4223                 skb_queue_tail(&htt->rx_in_ord_compl_q, skb);
4224                 return false;
4225         }
4226         case HTT_T2H_MSG_TYPE_TX_CREDIT_UPDATE_IND: {
4227                 struct ath10k_htt *htt = &ar->htt;
4228                 struct ath10k_htc *htc = &ar->htc;
4229                 struct ath10k_htc_ep *ep = &ar->htc.endpoint[htt->eid];
4230                 u32 msg_word = __le32_to_cpu(*(__le32 *)resp);
4231                 int htt_credit_delta;
4232
4233                 htt_credit_delta = HTT_TX_CREDIT_DELTA_ABS_GET(msg_word);
4234                 if (HTT_TX_CREDIT_SIGN_BIT_GET(msg_word))
4235                         htt_credit_delta = -htt_credit_delta;
4236
4237                 ath10k_dbg(ar, ATH10K_DBG_HTT,
4238                            "htt credit update delta %d\n",
4239                            htt_credit_delta);
4240
4241                 if (htt->disable_tx_comp) {
4242                         spin_lock_bh(&htc->tx_lock);
4243                         ep->tx_credits += htt_credit_delta;
4244                         spin_unlock_bh(&htc->tx_lock);
4245                         ath10k_dbg(ar, ATH10K_DBG_HTT,
4246                                    "htt credit total %d\n",
4247                                    ep->tx_credits);
4248                         ep->ep_ops.ep_tx_credits(htc->ar);
4249                 }
4250                 break;
4251         }
4252         case HTT_T2H_MSG_TYPE_CHAN_CHANGE: {
4253                 u32 phymode = __le32_to_cpu(resp->chan_change.phymode);
4254                 u32 freq = __le32_to_cpu(resp->chan_change.freq);
4255
4256                 ar->tgt_oper_chan = ieee80211_get_channel(ar->hw->wiphy, freq);
4257                 ath10k_dbg(ar, ATH10K_DBG_HTT,
4258                            "htt chan change freq %u phymode %s\n",
4259                            freq, ath10k_wmi_phymode_str(phymode));
4260                 break;
4261         }
4262         case HTT_T2H_MSG_TYPE_AGGR_CONF:
4263                 break;
4264         case HTT_T2H_MSG_TYPE_TX_FETCH_IND: {
4265                 struct sk_buff *tx_fetch_ind = skb_copy(skb, GFP_ATOMIC);
4266
4267                 if (!tx_fetch_ind) {
4268                         ath10k_warn(ar, "failed to copy htt tx fetch ind\n");
4269                         break;
4270                 }
4271                 skb_queue_tail(&htt->tx_fetch_ind_q, tx_fetch_ind);
4272                 break;
4273         }
4274         case HTT_T2H_MSG_TYPE_TX_FETCH_CONFIRM:
4275                 ath10k_htt_rx_tx_fetch_confirm(ar, skb);
4276                 break;
4277         case HTT_T2H_MSG_TYPE_TX_MODE_SWITCH_IND:
4278                 ath10k_htt_rx_tx_mode_switch_ind(ar, skb);
4279                 break;
4280         case HTT_T2H_MSG_TYPE_PEER_STATS:
4281                 ath10k_htt_fetch_peer_stats(ar, skb);
4282                 break;
4283         case HTT_T2H_MSG_TYPE_EN_STATS:
4284         default:
4285                 ath10k_warn(ar, "htt event (%d) not handled\n",
4286                             resp->hdr.msg_type);
4287                 ath10k_dbg_dump(ar, ATH10K_DBG_HTT_DUMP, NULL, "htt event: ",
4288                                 skb->data, skb->len);
4289                 break;
4290         }
4291         return true;
4292 }
4293 EXPORT_SYMBOL(ath10k_htt_t2h_msg_handler);
4294
4295 void ath10k_htt_rx_pktlog_completion_handler(struct ath10k *ar,
4296                                              struct sk_buff *skb)
4297 {
4298         trace_ath10k_htt_pktlog(ar, skb->data, skb->len);
4299         dev_kfree_skb_any(skb);
4300 }
4301 EXPORT_SYMBOL(ath10k_htt_rx_pktlog_completion_handler);
4302
4303 static int ath10k_htt_rx_deliver_msdu(struct ath10k *ar, int quota, int budget)
4304 {
4305         struct sk_buff *skb;
4306
4307         while (quota < budget) {
4308                 if (skb_queue_empty(&ar->htt.rx_msdus_q))
4309                         break;
4310
4311                 skb = skb_dequeue(&ar->htt.rx_msdus_q);
4312                 if (!skb)
4313                         break;
4314                 ath10k_process_rx(ar, skb);
4315                 quota++;
4316         }
4317
4318         return quota;
4319 }
4320
4321 int ath10k_htt_rx_hl_indication(struct ath10k *ar, int budget)
4322 {
4323         struct htt_resp *resp;
4324         struct ath10k_htt *htt = &ar->htt;
4325         struct sk_buff *skb;
4326         bool release;
4327         int quota;
4328
4329         for (quota = 0; quota < budget; quota++) {
4330                 skb = skb_dequeue(&htt->rx_indication_head);
4331                 if (!skb)
4332                         break;
4333
4334                 resp = (struct htt_resp *)skb->data;
4335
4336                 release = ath10k_htt_rx_proc_rx_ind_hl(htt,
4337                                                        &resp->rx_ind_hl,
4338                                                        skb,
4339                                                        HTT_RX_PN_CHECK,
4340                                                        HTT_RX_NON_TKIP_MIC);
4341
4342                 if (release)
4343                         dev_kfree_skb_any(skb);
4344
4345                 ath10k_dbg(ar, ATH10K_DBG_HTT, "rx indication poll pending count:%d\n",
4346                            skb_queue_len(&htt->rx_indication_head));
4347         }
4348         return quota;
4349 }
4350 EXPORT_SYMBOL(ath10k_htt_rx_hl_indication);
4351
4352 int ath10k_htt_txrx_compl_task(struct ath10k *ar, int budget)
4353 {
4354         struct ath10k_htt *htt = &ar->htt;
4355         struct htt_tx_done tx_done = {};
4356         struct sk_buff_head tx_ind_q;
4357         struct sk_buff *skb;
4358         unsigned long flags;
4359         int quota = 0, done, ret;
4360         bool resched_napi = false;
4361
4362         __skb_queue_head_init(&tx_ind_q);
4363
4364         /* Process pending frames before dequeuing more data
4365          * from hardware.
4366          */
4367         quota = ath10k_htt_rx_deliver_msdu(ar, quota, budget);
4368         if (quota == budget) {
4369                 resched_napi = true;
4370                 goto exit;
4371         }
4372
4373         while ((skb = skb_dequeue(&htt->rx_in_ord_compl_q))) {
4374                 spin_lock_bh(&htt->rx_ring.lock);
4375                 ret = ath10k_htt_rx_in_ord_ind(ar, skb);
4376                 spin_unlock_bh(&htt->rx_ring.lock);
4377
4378                 dev_kfree_skb_any(skb);
4379                 if (ret == -EIO) {
4380                         resched_napi = true;
4381                         goto exit;
4382                 }
4383         }
4384
4385         while (atomic_read(&htt->num_mpdus_ready)) {
4386                 ret = ath10k_htt_rx_handle_amsdu(htt);
4387                 if (ret == -EIO) {
4388                         resched_napi = true;
4389                         goto exit;
4390                 }
4391                 atomic_dec(&htt->num_mpdus_ready);
4392         }
4393
4394         /* Deliver received data after processing data from hardware */
4395         quota = ath10k_htt_rx_deliver_msdu(ar, quota, budget);
4396
4397         /* From NAPI documentation:
4398          *  The napi poll() function may also process TX completions, in which
4399          *  case if it processes the entire TX ring then it should count that
4400          *  work as the rest of the budget.
4401          */
4402         if ((quota < budget) && !kfifo_is_empty(&htt->txdone_fifo))
4403                 quota = budget;
4404
4405         /* kfifo_get: called only within txrx_tasklet so it's neatly serialized.
4406          * From kfifo_get() documentation:
4407          *  Note that with only one concurrent reader and one concurrent writer,
4408          *  you don't need extra locking to use these macro.
4409          */
4410         while (kfifo_get(&htt->txdone_fifo, &tx_done))
4411                 ath10k_txrx_tx_unref(htt, &tx_done);
4412
4413         ath10k_mac_tx_push_pending(ar);
4414
4415         spin_lock_irqsave(&htt->tx_fetch_ind_q.lock, flags);
4416         skb_queue_splice_init(&htt->tx_fetch_ind_q, &tx_ind_q);
4417         spin_unlock_irqrestore(&htt->tx_fetch_ind_q.lock, flags);
4418
4419         while ((skb = __skb_dequeue(&tx_ind_q))) {
4420                 ath10k_htt_rx_tx_fetch_ind(ar, skb);
4421                 dev_kfree_skb_any(skb);
4422         }
4423
4424 exit:
4425         ath10k_htt_rx_msdu_buff_replenish(htt);
4426         /* In case of rx failure or more data to read, report budget
4427          * to reschedule NAPI poll
4428          */
4429         done = resched_napi ? budget : quota;
4430
4431         return done;
4432 }
4433 EXPORT_SYMBOL(ath10k_htt_txrx_compl_task);
4434
4435 static const struct ath10k_htt_rx_ops htt_rx_ops_32 = {
4436         .htt_get_rx_ring_size = ath10k_htt_get_rx_ring_size_32,
4437         .htt_config_paddrs_ring = ath10k_htt_config_paddrs_ring_32,
4438         .htt_set_paddrs_ring = ath10k_htt_set_paddrs_ring_32,
4439         .htt_get_vaddr_ring = ath10k_htt_get_vaddr_ring_32,
4440         .htt_reset_paddrs_ring = ath10k_htt_reset_paddrs_ring_32,
4441 };
4442
4443 static const struct ath10k_htt_rx_ops htt_rx_ops_64 = {
4444         .htt_get_rx_ring_size = ath10k_htt_get_rx_ring_size_64,
4445         .htt_config_paddrs_ring = ath10k_htt_config_paddrs_ring_64,
4446         .htt_set_paddrs_ring = ath10k_htt_set_paddrs_ring_64,
4447         .htt_get_vaddr_ring = ath10k_htt_get_vaddr_ring_64,
4448         .htt_reset_paddrs_ring = ath10k_htt_reset_paddrs_ring_64,
4449 };
4450
4451 static const struct ath10k_htt_rx_ops htt_rx_ops_hl = {
4452         .htt_rx_proc_rx_frag_ind = ath10k_htt_rx_proc_rx_frag_ind_hl,
4453 };
4454
4455 void ath10k_htt_set_rx_ops(struct ath10k_htt *htt)
4456 {
4457         struct ath10k *ar = htt->ar;
4458
4459         if (ar->bus_param.dev_type == ATH10K_DEV_TYPE_HL)
4460                 htt->rx_ops = &htt_rx_ops_hl;
4461         else if (ar->hw_params.target_64bit)
4462                 htt->rx_ops = &htt_rx_ops_64;
4463         else
4464                 htt->rx_ops = &htt_rx_ops_32;
4465 }