drivers/net/wireless/ath/ath9k/recv.c

   1 /*
   2  * Copyright (c) 2008-2011 Atheros Communications Inc.
   3  *
   4  * Permission to use, copy, modify, and/or distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include <linux/dma-mapping.h>
  18 #include <linux/relay.h>
  19 #include "ath9k.h"
  20 #include "ar9003_mac.h"
  21
  22 #define SKB_CB_ATHBUF(__skb)    (*((struct ath_buf **)__skb->cb))
  23
  24 static inline bool ath9k_check_auto_sleep(struct ath_softc *sc)
  25 {
  26         return sc->ps_enabled &&
  27                (sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_AUTOSLEEP);
  28 }
  29
  30 /*
  31  * Setup and link descriptors.
  32  *
  33  * 11N: we can no longer afford to self link the last descriptor.
  34  * MAC acknowledges BA status as long as it copies frames to host
  35  * buffer (or rx fifo). This can incorrectly acknowledge packets
  36  * to a sender if last desc is self-linked.
  37  */
  38 static void ath_rx_buf_link(struct ath_softc *sc, struct ath_buf *bf)
  39 {
  40         struct ath_hw *ah = sc->sc_ah;
  41         struct ath_common *common = ath9k_hw_common(ah);
  42         struct ath_desc *ds;
  43         struct sk_buff *skb;
  44
  45         ds = bf->bf_desc;
  46         ds->ds_link = 0; /* link to null */
  47         ds->ds_data = bf->bf_buf_addr;
  48
  49         /* virtual addr of the beginning of the buffer. */
  50         skb = bf->bf_mpdu;
  51         BUG_ON(skb == NULL);
  52         ds->ds_vdata = skb->data;
  53
  54         /*
  55          * setup rx descriptors. The rx_bufsize here tells the hardware
  56          * how much data it can DMA to us and that we are prepared
  57          * to process
  58          */
  59         ath9k_hw_setuprxdesc(ah, ds,
  60                              common->rx_bufsize,
  61                              0);
  62
  63         if (sc->rx.rxlink == NULL)
  64                 ath9k_hw_putrxbuf(ah, bf->bf_daddr);
  65         else
  66                 *sc->rx.rxlink = bf->bf_daddr;
  67
  68         sc->rx.rxlink = &ds->ds_link;
  69 }
  70
  71 static void ath_rx_buf_relink(struct ath_softc *sc, struct ath_buf *bf)
  72 {
  73         if (sc->rx.buf_hold)
  74                 ath_rx_buf_link(sc, sc->rx.buf_hold);
  75
  76         sc->rx.buf_hold = bf;
  77 }
  78
  79 static void ath_setdefantenna(struct ath_softc *sc, u32 antenna)
  80 {
  81         /* XXX block beacon interrupts */
  82         ath9k_hw_setantenna(sc->sc_ah, antenna);
  83         sc->rx.defant = antenna;
  84         sc->rx.rxotherant = 0;
  85 }
  86
  87 static void ath_opmode_init(struct ath_softc *sc)
  88 {
  89         struct ath_hw *ah = sc->sc_ah;
  90         struct ath_common *common = ath9k_hw_common(ah);
  91
  92         u32 rfilt, mfilt[2];
  93
  94         /* configure rx filter */
  95         rfilt = ath_calcrxfilter(sc);
  96         ath9k_hw_setrxfilter(ah, rfilt);
  97
  98         /* configure bssid mask */
  99         ath_hw_setbssidmask(common);
 100
 101         /* configure operational mode */
 102         ath9k_hw_setopmode(ah);
 103
 104         /* calculate and install multicast filter */
 105         mfilt[0] = mfilt[1] = ~0;
 106         ath9k_hw_setmcastfilter(ah, mfilt[0], mfilt[1]);
 107 }
 108
 109 static bool ath_rx_edma_buf_link(struct ath_softc *sc,
 110                                  enum ath9k_rx_qtype qtype)
 111 {
 112         struct ath_hw *ah = sc->sc_ah;
 113         struct ath_rx_edma *rx_edma;
 114         struct sk_buff *skb;
 115         struct ath_buf *bf;
 116
 117         rx_edma = &sc->rx.rx_edma[qtype];
 118         if (skb_queue_len(&rx_edma->rx_fifo) >= rx_edma->rx_fifo_hwsize)
 119                 return false;
 120
 121         bf = list_first_entry(&sc->rx.rxbuf, struct ath_buf, list);
 122         list_del_init(&bf->list);
 123
 124         skb = bf->bf_mpdu;
 125
 126         memset(skb->data, 0, ah->caps.rx_status_len);
 127         dma_sync_single_for_device(sc->dev, bf->bf_buf_addr,
 128                                 ah->caps.rx_status_len, DMA_TO_DEVICE);
 129
 130         SKB_CB_ATHBUF(skb) = bf;
 131         ath9k_hw_addrxbuf_edma(ah, bf->bf_buf_addr, qtype);
 132         __skb_queue_tail(&rx_edma->rx_fifo, skb);
 133
 134         return true;
 135 }
 136
 137 static void ath_rx_addbuffer_edma(struct ath_softc *sc,
 138                                   enum ath9k_rx_qtype qtype)
 139 {
 140         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
 141         struct ath_buf *bf, *tbf;
 142
 143         if (list_empty(&sc->rx.rxbuf)) {
 144                 ath_dbg(common, QUEUE, "No free rx buf available\n");
 145                 return;
 146         }
 147
 148         list_for_each_entry_safe(bf, tbf, &sc->rx.rxbuf, list)
 149                 if (!ath_rx_edma_buf_link(sc, qtype))
 150                         break;
 151
 152 }
 153
 154 static void ath_rx_remove_buffer(struct ath_softc *sc,
 155                                  enum ath9k_rx_qtype qtype)
 156 {
 157         struct ath_buf *bf;
 158         struct ath_rx_edma *rx_edma;
 159         struct sk_buff *skb;
 160
 161         rx_edma = &sc->rx.rx_edma[qtype];
 162
 163         while ((skb = __skb_dequeue(&rx_edma->rx_fifo)) != NULL) {
 164                 bf = SKB_CB_ATHBUF(skb);
 165                 BUG_ON(!bf);
 166                 list_add_tail(&bf->list, &sc->rx.rxbuf);
 167         }
 168 }
 169
 170 static void ath_rx_edma_cleanup(struct ath_softc *sc)
 171 {
 172         struct ath_hw *ah = sc->sc_ah;
 173         struct ath_common *common = ath9k_hw_common(ah);
 174         struct ath_buf *bf;
 175
 176         ath_rx_remove_buffer(sc, ATH9K_RX_QUEUE_LP);
 177         ath_rx_remove_buffer(sc, ATH9K_RX_QUEUE_HP);
 178
 179         list_for_each_entry(bf, &sc->rx.rxbuf, list) {
 180                 if (bf->bf_mpdu) {
 181                         dma_unmap_single(sc->dev, bf->bf_buf_addr,
 182                                         common->rx_bufsize,
 183                                         DMA_BIDIRECTIONAL);
 184                         dev_kfree_skb_any(bf->bf_mpdu);
 185                         bf->bf_buf_addr = 0;
 186                         bf->bf_mpdu = NULL;
 187                 }
 188         }
 189 }
 190
 191 static void ath_rx_edma_init_queue(struct ath_rx_edma *rx_edma, int size)
 192 {
 193         __skb_queue_head_init(&rx_edma->rx_fifo);
 194         rx_edma->rx_fifo_hwsize = size;
 195 }
 196
 197 static int ath_rx_edma_init(struct ath_softc *sc, int nbufs)
 198 {
 199         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
 200         struct ath_hw *ah = sc->sc_ah;
 201         struct sk_buff *skb;
 202         struct ath_buf *bf;
 203         int error = 0, i;
 204         u32 size;
 205
 206         ath9k_hw_set_rx_bufsize(ah, common->rx_bufsize -
 207                                     ah->caps.rx_status_len);
 208
 209         ath_rx_edma_init_queue(&sc->rx.rx_edma[ATH9K_RX_QUEUE_LP],
 210                                ah->caps.rx_lp_qdepth);
 211         ath_rx_edma_init_queue(&sc->rx.rx_edma[ATH9K_RX_QUEUE_HP],
 212                                ah->caps.rx_hp_qdepth);
 213
 214         size = sizeof(struct ath_buf) * nbufs;
 215         bf = devm_kzalloc(sc->dev, size, GFP_KERNEL);
 216         if (!bf)
 217                 return -ENOMEM;
 218
 219         INIT_LIST_HEAD(&sc->rx.rxbuf);
 220
 221         for (i = 0; i < nbufs; i++, bf++) {
 222                 skb = ath_rxbuf_alloc(common, common->rx_bufsize, GFP_KERNEL);
 223                 if (!skb) {
 224                         error = -ENOMEM;
 225                         goto rx_init_fail;
 226                 }
 227
 228                 memset(skb->data, 0, common->rx_bufsize);
 229                 bf->bf_mpdu = skb;
 230
 231                 bf->bf_buf_addr = dma_map_single(sc->dev, skb->data,
 232                                                  common->rx_bufsize,
 233                                                  DMA_BIDIRECTIONAL);
 234                 if (unlikely(dma_mapping_error(sc->dev,
 235                                                 bf->bf_buf_addr))) {
 236                                 dev_kfree_skb_any(skb);
 237                                 bf->bf_mpdu = NULL;
 238                                 bf->bf_buf_addr = 0;
 239                                 ath_err(common,
 240                                         "dma_mapping_error() on RX init\n");
 241                                 error = -ENOMEM;
 242                                 goto rx_init_fail;
 243                 }
 244
 245                 list_add_tail(&bf->list, &sc->rx.rxbuf);
 246         }
 247
 248         return 0;
 249
 250 rx_init_fail:
 251         ath_rx_edma_cleanup(sc);
 252         return error;
 253 }
 254
 255 static void ath_edma_start_recv(struct ath_softc *sc)
 256 {
 257         ath9k_hw_rxena(sc->sc_ah);
 258         ath_rx_addbuffer_edma(sc, ATH9K_RX_QUEUE_HP);
 259         ath_rx_addbuffer_edma(sc, ATH9K_RX_QUEUE_LP);
 260         ath_opmode_init(sc);
 261         ath9k_hw_startpcureceive(sc->sc_ah, !!(sc->hw->conf.flags & IEEE80211_CONF_OFFCHANNEL));
 262 }
 263
 264 static void ath_edma_stop_recv(struct ath_softc *sc)
 265 {
 266         ath_rx_remove_buffer(sc, ATH9K_RX_QUEUE_HP);
 267         ath_rx_remove_buffer(sc, ATH9K_RX_QUEUE_LP);
 268 }
 269
 270 int ath_rx_init(struct ath_softc *sc, int nbufs)
 271 {
 272         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
 273         struct sk_buff *skb;
 274         struct ath_buf *bf;
 275         int error = 0;
 276
 277         spin_lock_init(&sc->sc_pcu_lock);
 278
 279         common->rx_bufsize = IEEE80211_MAX_MPDU_LEN / 2 +
 280                              sc->sc_ah->caps.rx_status_len;
 281
 282         if (sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_EDMA)
 283                 return ath_rx_edma_init(sc, nbufs);
 284
 285         ath_dbg(common, CONFIG, "cachelsz %u rxbufsize %u\n",
 286                 common->cachelsz, common->rx_bufsize);
 287
 288         /* Initialize rx descriptors */
 289
 290         error = ath_descdma_setup(sc, &sc->rx.rxdma, &sc->rx.rxbuf,
 291                                   "rx", nbufs, 1, 0);
 292         if (error != 0) {
 293                 ath_err(common,
 294                         "failed to allocate rx descriptors: %d\n",
 295                         error);
 296                 goto err;
 297         }
 298
 299         list_for_each_entry(bf, &sc->rx.rxbuf, list) {
 300                 skb = ath_rxbuf_alloc(common, common->rx_bufsize,
 301                                       GFP_KERNEL);
 302                 if (skb == NULL) {
 303                         error = -ENOMEM;
 304                         goto err;
 305                 }
 306
 307                 bf->bf_mpdu = skb;
 308                 bf->bf_buf_addr = dma_map_single(sc->dev, skb->data,
 309                                                  common->rx_bufsize,
 310                                                  DMA_FROM_DEVICE);
 311                 if (unlikely(dma_mapping_error(sc->dev,
 312                                                bf->bf_buf_addr))) {
 313                         dev_kfree_skb_any(skb);
 314                         bf->bf_mpdu = NULL;
 315                         bf->bf_buf_addr = 0;
 316                         ath_err(common,
 317                                 "dma_mapping_error() on RX init\n");
 318                         error = -ENOMEM;
 319                         goto err;
 320                 }
 321         }
 322         sc->rx.rxlink = NULL;
 323 err:
 324         if (error)
 325                 ath_rx_cleanup(sc);
 326
 327         return error;
 328 }
 329
 330 void ath_rx_cleanup(struct ath_softc *sc)
 331 {
 332         struct ath_hw *ah = sc->sc_ah;
 333         struct ath_common *common = ath9k_hw_common(ah);
 334         struct sk_buff *skb;
 335         struct ath_buf *bf;
 336
 337         if (sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_EDMA) {
 338                 ath_rx_edma_cleanup(sc);
 339                 return;
 340         }
 341
 342         list_for_each_entry(bf, &sc->rx.rxbuf, list) {
 343                 skb = bf->bf_mpdu;
 344                 if (skb) {
 345                         dma_unmap_single(sc->dev, bf->bf_buf_addr,
 346                                          common->rx_bufsize,
 347                                          DMA_FROM_DEVICE);
 348                         dev_kfree_skb(skb);
 349                         bf->bf_buf_addr = 0;
 350                         bf->bf_mpdu = NULL;
 351                 }
 352         }
 353 }
 354
 355 /*
 356  * Calculate the receive filter according to the
 357  * operating mode and state:
 358  *
 359  * o always accept unicast, broadcast, and multicast traffic
 360  * o maintain current state of phy error reception (the hal
 361  *   may enable phy error frames for noise immunity work)
 362  * o probe request frames are accepted only when operating in
 363  *   hostap, adhoc, or monitor modes
 364  * o enable promiscuous mode according to the interface state
 365  * o accept beacons:
 366  *   - when operating in adhoc mode so the 802.11 layer creates
 367  *     node table entries for peers,
 368  *   - when operating in station mode for collecting rssi data when
 369  *     the station is otherwise quiet, or
 370  *   - when operating as a repeater so we see repeater-sta beacons
 371  *   - when scanning
 372  */
 373
 374 u32 ath_calcrxfilter(struct ath_softc *sc)
 375 {
 376         u32 rfilt;
 377
 378         rfilt = ATH9K_RX_FILTER_UCAST | ATH9K_RX_FILTER_BCAST
 379                 | ATH9K_RX_FILTER_MCAST;
 380
 381         /* if operating on a DFS channel, enable radar pulse detection */
 382         if (sc->hw->conf.radar_enabled)
 383                 rfilt |= ATH9K_RX_FILTER_PHYRADAR | ATH9K_RX_FILTER_PHYERR;
 384
 385         if (sc->rx.rxfilter & FIF_PROBE_REQ)
 386                 rfilt |= ATH9K_RX_FILTER_PROBEREQ;
 387
 388         /*
 389          * Set promiscuous mode when FIF_PROMISC_IN_BSS is enabled for station
 390          * mode interface or when in monitor mode. AP mode does not need this
 391          * since it receives all in-BSS frames anyway.
 392          */
 393         if (sc->sc_ah->is_monitoring)
 394                 rfilt |= ATH9K_RX_FILTER_PROM;
 395
 396         if (sc->rx.rxfilter & FIF_CONTROL)
 397                 rfilt |= ATH9K_RX_FILTER_CONTROL;
 398
 399         if ((sc->sc_ah->opmode == NL80211_IFTYPE_STATION) &&
 400             (sc->nvifs <= 1) &&
 401             !(sc->rx.rxfilter & FIF_BCN_PRBRESP_PROMISC))
 402                 rfilt |= ATH9K_RX_FILTER_MYBEACON;
 403         else
 404                 rfilt |= ATH9K_RX_FILTER_BEACON;
 405
 406         if ((sc->sc_ah->opmode == NL80211_IFTYPE_AP) ||
 407             (sc->rx.rxfilter & FIF_PSPOLL))
 408                 rfilt |= ATH9K_RX_FILTER_PSPOLL;
 409
 410         if (conf_is_ht(&sc->hw->conf))
 411                 rfilt |= ATH9K_RX_FILTER_COMP_BAR;
 412
 413         if (sc->nvifs > 1 || (sc->rx.rxfilter & FIF_OTHER_BSS)) {
 414                 /* This is needed for older chips */
 415                 if (sc->sc_ah->hw_version.macVersion <= AR_SREV_VERSION_9160)
 416                         rfilt |= ATH9K_RX_FILTER_PROM;
 417                 rfilt |= ATH9K_RX_FILTER_MCAST_BCAST_ALL;
 418         }
 419
 420         if (AR_SREV_9550(sc->sc_ah))
 421                 rfilt |= ATH9K_RX_FILTER_4ADDRESS;
 422
 423         return rfilt;
 424
 425 }
 426
 427 int ath_startrecv(struct ath_softc *sc)
 428 {
 429         struct ath_hw *ah = sc->sc_ah;
 430         struct ath_buf *bf, *tbf;
 431
 432         if (ah->caps.hw_caps & ATH9K_HW_CAP_EDMA) {
 433                 ath_edma_start_recv(sc);
 434                 return 0;
 435         }
 436
 437         if (list_empty(&sc->rx.rxbuf))
 438                 goto start_recv;
 439
 440         sc->rx.buf_hold = NULL;
 441         sc->rx.rxlink = NULL;
 442         list_for_each_entry_safe(bf, tbf, &sc->rx.rxbuf, list) {
 443                 ath_rx_buf_link(sc, bf);
 444         }
 445
 446         /* We could have deleted elements so the list may be empty now */
 447         if (list_empty(&sc->rx.rxbuf))
 448                 goto start_recv;
 449
 450         bf = list_first_entry(&sc->rx.rxbuf, struct ath_buf, list);
 451         ath9k_hw_putrxbuf(ah, bf->bf_daddr);
 452         ath9k_hw_rxena(ah);
 453
 454 start_recv:
 455         ath_opmode_init(sc);
 456         ath9k_hw_startpcureceive(ah, !!(sc->hw->conf.flags & IEEE80211_CONF_OFFCHANNEL));
 457
 458         return 0;
 459 }
 460
 461 static void ath_flushrecv(struct ath_softc *sc)
 462 {
 463         if (sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_EDMA)
 464                 ath_rx_tasklet(sc, 1, true);
 465         ath_rx_tasklet(sc, 1, false);
 466 }
 467
 468 bool ath_stoprecv(struct ath_softc *sc)
 469 {
 470         struct ath_hw *ah = sc->sc_ah;
 471         bool stopped, reset = false;
 472
 473         ath9k_hw_abortpcurecv(ah);
 474         ath9k_hw_setrxfilter(ah, 0);
 475         stopped = ath9k_hw_stopdmarecv(ah, &reset);
 476
 477         ath_flushrecv(sc);
 478
 479         if (sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_EDMA)
 480                 ath_edma_stop_recv(sc);
 481         else
 482                 sc->rx.rxlink = NULL;
 483
 484         if (!(ah->ah_flags & AH_UNPLUGGED) &&
 485             unlikely(!stopped)) {
 486                 ath_err(ath9k_hw_common(sc->sc_ah),
 487                         "Could not stop RX, we could be "
 488                         "confusing the DMA engine when we start RX up\n");
 489                 ATH_DBG_WARN_ON_ONCE(!stopped);
 490         }
 491         return stopped && !reset;
 492 }
 493
 494 static bool ath_beacon_dtim_pending_cab(struct sk_buff *skb)
 495 {
 496         /* Check whether the Beacon frame has DTIM indicating buffered bc/mc */
 497         struct ieee80211_mgmt *mgmt;
 498         u8 *pos, *end, id, elen;
 499         struct ieee80211_tim_ie *tim;
 500
 501         mgmt = (struct ieee80211_mgmt *)skb->data;
 502         pos = mgmt->u.beacon.variable;
 503         end = skb->data + skb->len;
 504
 505         while (pos + 2 < end) {
 506                 id = *pos++;
 507                 elen = *pos++;
 508                 if (pos + elen > end)
 509                         break;
 510
 511                 if (id == WLAN_EID_TIM) {
 512                         if (elen < sizeof(*tim))
 513                                 break;
 514                         tim = (struct ieee80211_tim_ie *) pos;
 515                         if (tim->dtim_count != 0)
 516                                 break;
 517                         return tim->bitmap_ctrl & 0x01;
 518                 }
 519
 520                 pos += elen;
 521         }
 522
 523         return false;
 524 }
 525
 526 static void ath_rx_ps_beacon(struct ath_softc *sc, struct sk_buff *skb)
 527 {
 528         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
 529
 530         if (skb->len < 24 + 8 + 2 + 2)
 531                 return;
 532
 533         sc->ps_flags &= ~PS_WAIT_FOR_BEACON;
 534
 535         if (sc->ps_flags & PS_BEACON_SYNC) {
 536                 sc->ps_flags &= ~PS_BEACON_SYNC;
 537                 ath_dbg(common, PS,
 538                         "Reconfigure beacon timers based on synchronized timestamp\n");
 539                 ath9k_set_beacon(sc);
 540         }
 541
 542         if (ath_beacon_dtim_pending_cab(skb)) {
 543                 /*
 544                  * Remain awake waiting for buffered broadcast/multicast
 545                  * frames. If the last broadcast/multicast frame is not
 546                  * received properly, the next beacon frame will work as
 547                  * a backup trigger for returning into NETWORK SLEEP state,
 548                  * so we are waiting for it as well.
 549                  */
 550                 ath_dbg(common, PS,
 551                         "Received DTIM beacon indicating buffered broadcast/multicast frame(s)\n");
 552                 sc->ps_flags |= PS_WAIT_FOR_CAB | PS_WAIT_FOR_BEACON;
 553                 return;
 554         }
 555
 556         if (sc->ps_flags & PS_WAIT_FOR_CAB) {
 557                 /*
 558                  * This can happen if a broadcast frame is dropped or the AP
 559                  * fails to send a frame indicating that all CAB frames have
 560                  * been delivered.
 561                  */
 562                 sc->ps_flags &= ~PS_WAIT_FOR_CAB;
 563                 ath_dbg(common, PS, "PS wait for CAB frames timed out\n");
 564         }
 565 }
 566
 567 static void ath_rx_ps(struct ath_softc *sc, struct sk_buff *skb, bool mybeacon)
 568 {
 569         struct ieee80211_hdr *hdr;
 570         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
 571
 572         hdr = (struct ieee80211_hdr *)skb->data;
 573
 574         /* Process Beacon and CAB receive in PS state */
 575         if (((sc->ps_flags & PS_WAIT_FOR_BEACON) || ath9k_check_auto_sleep(sc))
 576             && mybeacon) {
 577                 ath_rx_ps_beacon(sc, skb);
 578         } else if ((sc->ps_flags & PS_WAIT_FOR_CAB) &&
 579                    (ieee80211_is_data(hdr->frame_control) ||
 580                     ieee80211_is_action(hdr->frame_control)) &&
 581                    is_multicast_ether_addr(hdr->addr1) &&
 582                    !ieee80211_has_moredata(hdr->frame_control)) {
 583                 /*
 584                  * No more broadcast/multicast frames to be received at this
 585                  * point.
 586                  */
 587                 sc->ps_flags &= ~(PS_WAIT_FOR_CAB | PS_WAIT_FOR_BEACON);
 588                 ath_dbg(common, PS,
 589                         "All PS CAB frames received, back to sleep\n");
 590         } else if ((sc->ps_flags & PS_WAIT_FOR_PSPOLL_DATA) &&
 591                    !is_multicast_ether_addr(hdr->addr1) &&
 592                    !ieee80211_has_morefrags(hdr->frame_control)) {
 593                 sc->ps_flags &= ~PS_WAIT_FOR_PSPOLL_DATA;
 594                 ath_dbg(common, PS,
 595                         "Going back to sleep after having received PS-Poll data (0x%lx)\n",
 596                         sc->ps_flags & (PS_WAIT_FOR_BEACON |
 597                                         PS_WAIT_FOR_CAB |
 598                                         PS_WAIT_FOR_PSPOLL_DATA |
 599                                         PS_WAIT_FOR_TX_ACK));
 600         }
 601 }
 602
 603 static bool ath_edma_get_buffers(struct ath_softc *sc,
 604                                  enum ath9k_rx_qtype qtype,
 605                                  struct ath_rx_status *rs,
 606                                  struct ath_buf **dest)
 607 {
 608         struct ath_rx_edma *rx_edma = &sc->rx.rx_edma[qtype];
 609         struct ath_hw *ah = sc->sc_ah;
 610         struct ath_common *common = ath9k_hw_common(ah);
 611         struct sk_buff *skb;
 612         struct ath_buf *bf;
 613         int ret;
 614
 615         skb = skb_peek(&rx_edma->rx_fifo);
 616         if (!skb)
 617                 return false;
 618
 619         bf = SKB_CB_ATHBUF(skb);
 620         BUG_ON(!bf);
 621
 622         dma_sync_single_for_cpu(sc->dev, bf->bf_buf_addr,
 623                                 common->rx_bufsize, DMA_FROM_DEVICE);
 624
 625         ret = ath9k_hw_process_rxdesc_edma(ah, rs, skb->data);
 626         if (ret == -EINPROGRESS) {
 627                 /*let device gain the buffer again*/
 628                 dma_sync_single_for_device(sc->dev, bf->bf_buf_addr,
 629                                 common->rx_bufsize, DMA_FROM_DEVICE);
 630                 return false;
 631         }
 632
 633         __skb_unlink(skb, &rx_edma->rx_fifo);
 634         if (ret == -EINVAL) {
 635                 /* corrupt descriptor, skip this one and the following one */
 636                 list_add_tail(&bf->list, &sc->rx.rxbuf);
 637                 ath_rx_edma_buf_link(sc, qtype);
 638
 639                 skb = skb_peek(&rx_edma->rx_fifo);
 640                 if (skb) {
 641                         bf = SKB_CB_ATHBUF(skb);
 642                         BUG_ON(!bf);
 643
 644                         __skb_unlink(skb, &rx_edma->rx_fifo);
 645                         list_add_tail(&bf->list, &sc->rx.rxbuf);
 646                         ath_rx_edma_buf_link(sc, qtype);
 647                 }
 648
 649                 bf = NULL;
 650         }
 651
 652         *dest = bf;
 653         return true;
 654 }
 655
 656 static struct ath_buf *ath_edma_get_next_rx_buf(struct ath_softc *sc,
 657                                                 struct ath_rx_status *rs,
 658                                                 enum ath9k_rx_qtype qtype)
 659 {
 660         struct ath_buf *bf = NULL;
 661
 662         while (ath_edma_get_buffers(sc, qtype, rs, &bf)) {
 663                 if (!bf)
 664                         continue;
 665
 666                 return bf;
 667         }
 668         return NULL;
 669 }
 670
 671 static struct ath_buf *ath_get_next_rx_buf(struct ath_softc *sc,
 672                                            struct ath_rx_status *rs)
 673 {
 674         struct ath_hw *ah = sc->sc_ah;
 675         struct ath_common *common = ath9k_hw_common(ah);
 676         struct ath_desc *ds;
 677         struct ath_buf *bf;
 678         int ret;
 679
 680         if (list_empty(&sc->rx.rxbuf)) {
 681                 sc->rx.rxlink = NULL;
 682                 return NULL;
 683         }
 684
 685         bf = list_first_entry(&sc->rx.rxbuf, struct ath_buf, list);
 686         if (bf == sc->rx.buf_hold)
 687                 return NULL;
 688
 689         ds = bf->bf_desc;
 690
 691         /*
 692          * Must provide the virtual address of the current
 693          * descriptor, the physical address, and the virtual
 694          * address of the next descriptor in the h/w chain.
 695          * This allows the HAL to look ahead to see if the
 696          * hardware is done with a descriptor by checking the
 697          * done bit in the following descriptor and the address
 698          * of the current descriptor the DMA engine is working
 699          * on.  All this is necessary because of our use of
 700          * a self-linked list to avoid rx overruns.
 701          */
 702         ret = ath9k_hw_rxprocdesc(ah, ds, rs);
 703         if (ret == -EINPROGRESS) {
 704                 struct ath_rx_status trs;
 705                 struct ath_buf *tbf;
 706                 struct ath_desc *tds;
 707
 708                 memset(&trs, 0, sizeof(trs));
 709                 if (list_is_last(&bf->list, &sc->rx.rxbuf)) {
 710                         sc->rx.rxlink = NULL;
 711                         return NULL;
 712                 }
 713
 714                 tbf = list_entry(bf->list.next, struct ath_buf, list);
 715
 716                 /*
 717                  * On some hardware the descriptor status words could
 718                  * get corrupted, including the done bit. Because of
 719                  * this, check if the next descriptor's done bit is
 720                  * set or not.
 721                  *
 722                  * If the next descriptor's done bit is set, the current
 723                  * descriptor has been corrupted. Force s/w to discard
 724                  * this descriptor and continue...
 725                  */
 726
 727                 tds = tbf->bf_desc;
 728                 ret = ath9k_hw_rxprocdesc(ah, tds, &trs);
 729                 if (ret == -EINPROGRESS)
 730                         return NULL;
 731
 732                 /*
 733                  * mark descriptor as zero-length and set the 'more'
 734                  * flag to ensure that both buffers get discarded
 735                  */
 736                 rs->rs_datalen = 0;
 737                 rs->rs_more = true;
 738         }
 739
 740         list_del(&bf->list);
 741         if (!bf->bf_mpdu)
 742                 return bf;
 743
 744         /*
 745          * Synchronize the DMA transfer with CPU before
 746          * 1. accessing the frame
 747          * 2. requeueing the same buffer to h/w
 748          */
 749         dma_sync_single_for_cpu(sc->dev, bf->bf_buf_addr,
 750                         common->rx_bufsize,
 751                         DMA_FROM_DEVICE);
 752
 753         return bf;
 754 }
 755
 756 /* Assumes you've already done the endian to CPU conversion */
 757 static bool ath9k_rx_accept(struct ath_common *common,
 758                             struct ieee80211_hdr *hdr,
 759                             struct ieee80211_rx_status *rxs,
 760                             struct ath_rx_status *rx_stats,
 761                             bool *decrypt_error)
 762 {
 763         struct ath_softc *sc = (struct ath_softc *) common->priv;
 764         bool is_mc, is_valid_tkip, strip_mic, mic_error;
 765         struct ath_hw *ah = common->ah;
 766         __le16 fc;
 767
 768         fc = hdr->frame_control;
 769
 770         is_mc = !!is_multicast_ether_addr(hdr->addr1);
 771         is_valid_tkip = rx_stats->rs_keyix != ATH9K_RXKEYIX_INVALID &&
 772                 test_bit(rx_stats->rs_keyix, common->tkip_keymap);
 773         strip_mic = is_valid_tkip && ieee80211_is_data(fc) &&
 774                 ieee80211_has_protected(fc) &&
 775                 !(rx_stats->rs_status &
 776                 (ATH9K_RXERR_DECRYPT | ATH9K_RXERR_CRC | ATH9K_RXERR_MIC |
 777                  ATH9K_RXERR_KEYMISS));
 778
 779         /*
 780          * Key miss events are only relevant for pairwise keys where the
 781          * descriptor does contain a valid key index. This has been observed
 782          * mostly with CCMP encryption.
 783          */
 784         if (rx_stats->rs_keyix == ATH9K_RXKEYIX_INVALID ||
 785             !test_bit(rx_stats->rs_keyix, common->ccmp_keymap))
 786                 rx_stats->rs_status &= ~ATH9K_RXERR_KEYMISS;
 787
 788         mic_error = is_valid_tkip && !ieee80211_is_ctl(fc) &&
 789                 !ieee80211_has_morefrags(fc) &&
 790                 !(le16_to_cpu(hdr->seq_ctrl) & IEEE80211_SCTL_FRAG) &&
 791                 (rx_stats->rs_status & ATH9K_RXERR_MIC);
 792
 793         /*
 794          * The rx_stats->rs_status will not be set until the end of the
 795          * chained descriptors so it can be ignored if rs_more is set. The
 796          * rs_more will be false at the last element of the chained
 797          * descriptors.
 798          */
 799         if (rx_stats->rs_status != 0) {
 800                 u8 status_mask;
 801
 802                 if (rx_stats->rs_status & ATH9K_RXERR_CRC) {
 803                         rxs->flag |= RX_FLAG_FAILED_FCS_CRC;
 804                         mic_error = false;
 805                 }
 806
 807                 if ((rx_stats->rs_status & ATH9K_RXERR_DECRYPT) ||
 808                     (!is_mc && (rx_stats->rs_status & ATH9K_RXERR_KEYMISS))) {
 809                         *decrypt_error = true;
 810                         mic_error = false;
 811                 }
 812
 813                 /*
 814                  * Reject error frames with the exception of
 815                  * decryption and MIC failures. For monitor mode,
 816                  * we also ignore the CRC error.
 817                  */
 818                 status_mask = ATH9K_RXERR_DECRYPT | ATH9K_RXERR_MIC |
 819                               ATH9K_RXERR_KEYMISS;
 820
 821                 if (ah->is_monitoring && (sc->rx.rxfilter & FIF_FCSFAIL))
 822                         status_mask |= ATH9K_RXERR_CRC;
 823
 824                 if (rx_stats->rs_status & ~status_mask)
 825                         return false;
 826         }
 827
 828         /*
 829          * For unicast frames the MIC error bit can have false positives,
 830          * so all MIC error reports need to be validated in software.
 831          * False negatives are not common, so skip software verification
 832          * if the hardware considers the MIC valid.
 833          */
 834         if (strip_mic)
 835                 rxs->flag |= RX_FLAG_MMIC_STRIPPED;
 836         else if (is_mc && mic_error)
 837                 rxs->flag |= RX_FLAG_MMIC_ERROR;
 838
 839         return true;
 840 }
 841
 842 static int ath9k_process_rate(struct ath_common *common,
 843                               struct ieee80211_hw *hw,
 844                               struct ath_rx_status *rx_stats,
 845                               struct ieee80211_rx_status *rxs)
 846 {
 847         struct ieee80211_supported_band *sband;
 848         enum ieee80211_band band;
 849         unsigned int i = 0;
 850         struct ath_softc __maybe_unused *sc = common->priv;
 851
 852         band = hw->conf.chandef.chan->band;
 853         sband = hw->wiphy->bands[band];
 854
 855         if (rx_stats->rs_rate & 0x80) {
 856                 /* HT rate */
 857                 rxs->flag |= RX_FLAG_HT;
 858                 rxs->flag |= rx_stats->flag;
 859                 rxs->rate_idx = rx_stats->rs_rate & 0x7f;
 860                 return 0;
 861         }
 862
 863         for (i = 0; i < sband->n_bitrates; i++) {
 864                 if (sband->bitrates[i].hw_value == rx_stats->rs_rate) {
 865                         rxs->rate_idx = i;
 866                         return 0;
 867                 }
 868                 if (sband->bitrates[i].hw_value_short == rx_stats->rs_rate) {
 869                         rxs->flag |= RX_FLAG_SHORTPRE;
 870                         rxs->rate_idx = i;
 871                         return 0;
 872                 }
 873         }
 874
 875         /*
 876          * No valid hardware bitrate found -- we should not get here
 877          * because hardware has already validated this frame as OK.
 878          */
 879         ath_dbg(common, ANY,
 880                 "unsupported hw bitrate detected 0x%02x using 1 Mbit\n",
 881                 rx_stats->rs_rate);
 882         RX_STAT_INC(rx_rate_err);
 883         return -EINVAL;
 884 }
 885
 886 static void ath9k_process_rssi(struct ath_common *common,
 887                                struct ieee80211_hw *hw,
 888                                struct ath_rx_status *rx_stats,
 889                                struct ieee80211_rx_status *rxs)
 890 {
 891         struct ath_softc *sc = hw->priv;
 892         struct ath_hw *ah = common->ah;
 893         int last_rssi;
 894         int rssi = rx_stats->rs_rssi;
 895
 896         /*
 897          * RSSI is not available for subframes in an A-MPDU.
 898          */
 899         if (rx_stats->rs_moreaggr) {
 900                 rxs->flag |= RX_FLAG_NO_SIGNAL_VAL;
 901                 return;
 902         }
 903
 904         /*
 905          * Check if the RSSI for the last subframe in an A-MPDU
 906          * or an unaggregated frame is valid.
 907          */
 908         if (rx_stats->rs_rssi == ATH9K_RSSI_BAD) {
 909                 rxs->flag |= RX_FLAG_NO_SIGNAL_VAL;
 910                 return;
 911         }
 912
 913         /*
 914          * Update Beacon RSSI, this is used by ANI.
 915          */
 916         if (rx_stats->is_mybeacon &&
 917             ((ah->opmode == NL80211_IFTYPE_STATION) ||
 918              (ah->opmode == NL80211_IFTYPE_ADHOC))) {
 919                 ATH_RSSI_LPF(sc->last_rssi, rx_stats->rs_rssi);
 920                 last_rssi = sc->last_rssi;
 921
 922                 if (likely(last_rssi != ATH_RSSI_DUMMY_MARKER))
 923                         rssi = ATH_EP_RND(last_rssi, ATH_RSSI_EP_MULTIPLIER);
 924                 if (rssi < 0)
 925                         rssi = 0;
 926
 927                 ah->stats.avgbrssi = rssi;
 928         }
 929
 930         rxs->signal = ah->noise + rx_stats->rs_rssi;
 931 }
 932
 933 static void ath9k_process_tsf(struct ath_rx_status *rs,
 934                               struct ieee80211_rx_status *rxs,
 935                               u64 tsf)
 936 {
 937         u32 tsf_lower = tsf & 0xffffffff;
 938
 939         rxs->mactime = (tsf & ~0xffffffffULL) | rs->rs_tstamp;
 940         if (rs->rs_tstamp > tsf_lower &&
 941             unlikely(rs->rs_tstamp - tsf_lower > 0x10000000))
 942                 rxs->mactime -= 0x100000000ULL;
 943
 944         if (rs->rs_tstamp < tsf_lower &&
 945             unlikely(tsf_lower - rs->rs_tstamp > 0x10000000))
 946                 rxs->mactime += 0x100000000ULL;
 947 }
 948
 949 #ifdef CONFIG_ATH9K_DEBUGFS
 950 static s8 fix_rssi_inv_only(u8 rssi_val)
 951 {
 952         if (rssi_val == 128)
 953                 rssi_val = 0;
 954         return (s8) rssi_val;
 955 }
 956 #endif
 957
 958 /* returns 1 if this was a spectral frame, even if not handled. */
 959 static int ath_process_fft(struct ath_softc *sc, struct ieee80211_hdr *hdr,
 960                            struct ath_rx_status *rs, u64 tsf)
 961 {
 962 #ifdef CONFIG_ATH9K_DEBUGFS
 963         struct ath_hw *ah = sc->sc_ah;
 964         u8 bins[SPECTRAL_HT20_NUM_BINS];
 965         u8 *vdata = (u8 *)hdr;
 966         struct fft_sample_ht20 fft_sample;
 967         struct ath_radar_info *radar_info;
 968         struct ath_ht20_mag_info *mag_info;
 969         int len = rs->rs_datalen;
 970         int dc_pos;
 971         u16 length, max_magnitude;
 972
 973         /* AR9280 and before report via ATH9K_PHYERR_RADAR, AR93xx and newer
 974          * via ATH9K_PHYERR_SPECTRAL. Haven't seen ATH9K_PHYERR_FALSE_RADAR_EXT
 975          * yet, but this is supposed to be possible as well.
 976          */
 977         if (rs->rs_phyerr != ATH9K_PHYERR_RADAR &&
 978             rs->rs_phyerr != ATH9K_PHYERR_FALSE_RADAR_EXT &&
 979             rs->rs_phyerr != ATH9K_PHYERR_SPECTRAL)
 980                 return 0;
 981
 982         /* check if spectral scan bit is set. This does not have to be checked
 983          * if received through a SPECTRAL phy error, but shouldn't hurt.
 984          */
 985         radar_info = ((struct ath_radar_info *)&vdata[len]) - 1;
 986         if (!(radar_info->pulse_bw_info & SPECTRAL_SCAN_BITMASK))
 987                 return 0;
 988
 989         /* Variation in the data length is possible and will be fixed later.
 990          * Note that we only support HT20 for now.
 991          *
 992          * TODO: add HT20_40 support as well.
 993          */
 994         if ((len > SPECTRAL_HT20_TOTAL_DATA_LEN + 2) ||
 995             (len < SPECTRAL_HT20_TOTAL_DATA_LEN - 1))
 996                 return 1;
 997
 998         fft_sample.tlv.type = ATH_FFT_SAMPLE_HT20;
 999         length = sizeof(fft_sample) - sizeof(fft_sample.tlv);
1000         fft_sample.tlv.length = __cpu_to_be16(length);
1001
1002         fft_sample.freq = __cpu_to_be16(ah->curchan->chan->center_freq);
1003         fft_sample.rssi = fix_rssi_inv_only(rs->rs_rssi_ctl0);
1004         fft_sample.noise = ah->noise;
1005
1006         switch (len - SPECTRAL_HT20_TOTAL_DATA_LEN) {
1007         case 0:
1008                 /* length correct, nothing to do. */
1009                 memcpy(bins, vdata, SPECTRAL_HT20_NUM_BINS);
1010                 break;
1011         case -1:
1012                 /* first byte missing, duplicate it. */
1013                 memcpy(&bins[1], vdata, SPECTRAL_HT20_NUM_BINS - 1);
1014                 bins[0] = vdata[0];
1015                 break;
1016         case 2:
1017                 /* MAC added 2 extra bytes at bin 30 and 32, remove them. */
1018                 memcpy(bins, vdata, 30);
1019                 bins[30] = vdata[31];
1020                 memcpy(&bins[31], &vdata[33], SPECTRAL_HT20_NUM_BINS - 31);
1021                 break;
1022         case 1:
1023                 /* MAC added 2 extra bytes AND first byte is missing. */
1024                 bins[0] = vdata[0];
1025                 memcpy(&bins[0], vdata, 30);
1026                 bins[31] = vdata[31];
1027                 memcpy(&bins[32], &vdata[33], SPECTRAL_HT20_NUM_BINS - 32);
1028                 break;
1029         default:
1030                 return 1;
1031         }
1032
1033         /* DC value (value in the middle) is the blind spot of the spectral
1034          * sample and invalid, interpolate it.
1035          */
1036         dc_pos = SPECTRAL_HT20_NUM_BINS / 2;
1037         bins[dc_pos] = (bins[dc_pos + 1] + bins[dc_pos - 1]) / 2;
1038
1039         /* mag data is at the end of the frame, in front of radar_info */
1040         mag_info = ((struct ath_ht20_mag_info *)radar_info) - 1;
1041
1042         /* copy raw bins without scaling them */
1043         memcpy(fft_sample.data, bins, SPECTRAL_HT20_NUM_BINS);
1044         fft_sample.max_exp = mag_info->max_exp & 0xf;
1045
1046         max_magnitude = spectral_max_magnitude(mag_info->all_bins);
1047         fft_sample.max_magnitude = __cpu_to_be16(max_magnitude);
1048         fft_sample.max_index = spectral_max_index(mag_info->all_bins);
1049         fft_sample.bitmap_weight = spectral_bitmap_weight(mag_info->all_bins);
1050         fft_sample.tsf = __cpu_to_be64(tsf);
1051
1052         ath_debug_send_fft_sample(sc, &fft_sample.tlv);
1053         return 1;
1054 #else
1055         return 0;
1056 #endif
1057 }
1058
1059 static bool ath9k_is_mybeacon(struct ath_softc *sc, struct ieee80211_hdr *hdr)
1060 {
1061         struct ath_hw *ah = sc->sc_ah;
1062         struct ath_common *common = ath9k_hw_common(ah);
1063
1064         if (ieee80211_is_beacon(hdr->frame_control)) {
1065                 RX_STAT_INC(rx_beacons);
1066                 if (!is_zero_ether_addr(common->curbssid) &&
1067                     ether_addr_equal(hdr->addr3, common->curbssid))
1068                         return true;
1069         }
1070
1071         return false;
1072 }
1073
1074 /*
1075  * For Decrypt or Demic errors, we only mark packet status here and always push
1076  * up the frame up to let mac80211 handle the actual error case, be it no
1077  * decryption key or real decryption error. This let us keep statistics there.
1078  */
1079 static int ath9k_rx_skb_preprocess(struct ath_softc *sc,
1080                                    struct sk_buff *skb,
1081                                    struct ath_rx_status *rx_stats,
1082                                    struct ieee80211_rx_status *rx_status,
1083                                    bool *decrypt_error, u64 tsf)
1084 {
1085         struct ieee80211_hw *hw = sc->hw;
1086         struct ath_hw *ah = sc->sc_ah;
1087         struct ath_common *common = ath9k_hw_common(ah);
1088         struct ieee80211_hdr *hdr;
1089         bool discard_current = sc->rx.discard_next;
1090         int ret = 0;
1091
1092         /*
1093          * Discard corrupt descriptors which are marked in
1094          * ath_get_next_rx_buf().
1095          */
1096         sc->rx.discard_next = rx_stats->rs_more;
1097         if (discard_current)
1098                 return -EINVAL;
1099
1100         /*
1101          * Discard zero-length packets.
1102          */
1103         if (!rx_stats->rs_datalen) {
1104                 RX_STAT_INC(rx_len_err);
1105                 return -EINVAL;
1106         }
1107
1108         /*
1109          * rs_status follows rs_datalen so if rs_datalen is too large
1110          * we can take a hint that hardware corrupted it, so ignore
1111          * those frames.
1112          */
1113         if (rx_stats->rs_datalen > (common->rx_bufsize - ah->caps.rx_status_len)) {
1114                 RX_STAT_INC(rx_len_err);
1115                 return -EINVAL;
1116         }
1117
1118         /* Only use status info from the last fragment */
1119         if (rx_stats->rs_more)
1120                 return 0;
1121
1122         /*
1123          * Return immediately if the RX descriptor has been marked
1124          * as corrupt based on the various error bits.
1125          *
1126          * This is different from the other corrupt descriptor
1127          * condition handled above.
1128          */
1129         if (rx_stats->rs_status & ATH9K_RXERR_CORRUPT_DESC) {
1130                 ret = -EINVAL;
1131                 goto exit;
1132         }
1133
1134         hdr = (struct ieee80211_hdr *) (skb->data + ah->caps.rx_status_len);
1135
1136         ath9k_process_tsf(rx_stats, rx_status, tsf);
1137         ath_debug_stat_rx(sc, rx_stats);
1138
1139         /*
1140          * Process PHY errors and return so that the packet
1141          * can be dropped.
1142          */
1143         if (rx_stats->rs_status & ATH9K_RXERR_PHY) {
1144                 ath9k_dfs_process_phyerr(sc, hdr, rx_stats, rx_status->mactime);
1145                 if (ath_process_fft(sc, hdr, rx_stats, rx_status->mactime))
1146                         RX_STAT_INC(rx_spectral);
1147
1148                 ret = -EINVAL;
1149                 goto exit;
1150         }
1151
1152         /*
1153          * everything but the rate is checked here, the rate check is done
1154          * separately to avoid doing two lookups for a rate for each frame.
1155          */
1156         if (!ath9k_rx_accept(common, hdr, rx_status, rx_stats, decrypt_error)) {
1157                 ret = -EINVAL;
1158                 goto exit;
1159         }
1160
1161         rx_stats->is_mybeacon = ath9k_is_mybeacon(sc, hdr);
1162         if (rx_stats->is_mybeacon) {
1163                 sc->hw_busy_count = 0;
1164                 ath_start_rx_poll(sc, 3);
1165         }
1166
1167         if (ath9k_process_rate(common, hw, rx_stats, rx_status)) {
1168                 ret =-EINVAL;
1169                 goto exit;
1170         }
1171
1172         ath9k_process_rssi(common, hw, rx_stats, rx_status);
1173
1174         rx_status->band = hw->conf.chandef.chan->band;
1175         rx_status->freq = hw->conf.chandef.chan->center_freq;
1176         rx_status->antenna = rx_stats->rs_antenna;
1177         rx_status->flag |= RX_FLAG_MACTIME_END;
1178
1179 #ifdef CONFIG_ATH9K_BTCOEX_SUPPORT
1180         if (ieee80211_is_data_present(hdr->frame_control) &&
1181             !ieee80211_is_qos_nullfunc(hdr->frame_control))
1182                 sc->rx.num_pkts++;
1183 #endif
1184
1185 exit:
1186         sc->rx.discard_next = false;
1187         return ret;
1188 }
1189
1190 static void ath9k_rx_skb_postprocess(struct ath_common *common,
1191                                      struct sk_buff *skb,
1192                                      struct ath_rx_status *rx_stats,
1193                                      struct ieee80211_rx_status *rxs,
1194                                      bool decrypt_error)
1195 {
1196         struct ath_hw *ah = common->ah;
1197         struct ieee80211_hdr *hdr;
1198         int hdrlen, padpos, padsize;
1199         u8 keyix;
1200         __le16 fc;
1201
1202         /* see if any padding is done by the hw and remove it */
1203         hdr = (struct ieee80211_hdr *) skb->data;
1204         hdrlen = ieee80211_get_hdrlen_from_skb(skb);
1205         fc = hdr->frame_control;
1206         padpos = ieee80211_hdrlen(fc);
1207
1208         /* The MAC header is padded to have 32-bit boundary if the
1209          * packet payload is non-zero. The general calculation for
1210          * padsize would take into account odd header lengths:
1211          * padsize = (4 - padpos % 4) % 4; However, since only
1212          * even-length headers are used, padding can only be 0 or 2
1213          * bytes and we can optimize this a bit. In addition, we must
1214          * not try to remove padding from short control frames that do
1215          * not have payload. */
1216         padsize = padpos & 3;
1217         if (padsize && skb->len>=padpos+padsize+FCS_LEN) {
1218                 memmove(skb->data + padsize, skb->data, padpos);
1219                 skb_pull(skb, padsize);
1220         }
1221
1222         keyix = rx_stats->rs_keyix;
1223
1224         if (!(keyix == ATH9K_RXKEYIX_INVALID) && !decrypt_error &&
1225             ieee80211_has_protected(fc)) {
1226                 rxs->flag |= RX_FLAG_DECRYPTED;
1227         } else if (ieee80211_has_protected(fc)
1228                    && !decrypt_error && skb->len >= hdrlen + 4) {
1229                 keyix = skb->data[hdrlen + 3] >> 6;
1230
1231                 if (test_bit(keyix, common->keymap))
1232                         rxs->flag |= RX_FLAG_DECRYPTED;
1233         }
1234         if (ah->sw_mgmt_crypto &&
1235             (rxs->flag & RX_FLAG_DECRYPTED) &&
1236             ieee80211_is_mgmt(fc))
1237                 /* Use software decrypt for management frames. */
1238                 rxs->flag &= ~RX_FLAG_DECRYPTED;
1239 }
1240
1241 /*
1242  * Run the LNA combining algorithm only in these cases:
1243  *
1244  * Standalone WLAN cards with both LNA/Antenna diversity
1245  * enabled in the EEPROM.
1246  *
1247  * WLAN+BT cards which are in the supported card list
1248  * in ath_pci_id_table and the user has loaded the
1249  * driver with "bt_ant_diversity" set to true.
1250  */
1251 static void ath9k_antenna_check(struct ath_softc *sc,
1252                                 struct ath_rx_status *rs)
1253 {
1254         struct ath_hw *ah = sc->sc_ah;
1255         struct ath9k_hw_capabilities *pCap = &ah->caps;
1256         struct ath_common *common = ath9k_hw_common(ah);
1257
1258         if (!(ah->caps.hw_caps & ATH9K_HW_CAP_ANT_DIV_COMB))
1259                 return;
1260
1261         /*
1262          * All MPDUs in an aggregate will use the same LNA
1263          * as the first MPDU.
1264          */
1265         if (rs->rs_isaggr && !rs->rs_firstaggr)
1266                 return;
1267
1268         /*
1269          * Change the default rx antenna if rx diversity
1270          * chooses the other antenna 3 times in a row.
1271          */
1272         if (sc->rx.defant != rs->rs_antenna) {
1273                 if (++sc->rx.rxotherant >= 3)
1274                         ath_setdefantenna(sc, rs->rs_antenna);
1275         } else {
1276                 sc->rx.rxotherant = 0;
1277         }
1278
1279         if (pCap->hw_caps & ATH9K_HW_CAP_BT_ANT_DIV) {
1280                 if (common->bt_ant_diversity)
1281                         ath_ant_comb_scan(sc, rs);
1282         } else {
1283                 ath_ant_comb_scan(sc, rs);
1284         }
1285 }
1286
1287 static void ath9k_apply_ampdu_details(struct ath_softc *sc,
1288         struct ath_rx_status *rs, struct ieee80211_rx_status *rxs)
1289 {
1290         if (rs->rs_isaggr) {
1291                 rxs->flag |= RX_FLAG_AMPDU_DETAILS | RX_FLAG_AMPDU_LAST_KNOWN;
1292
1293                 rxs->ampdu_reference = sc->rx.ampdu_ref;
1294
1295                 if (!rs->rs_moreaggr) {
1296                         rxs->flag |= RX_FLAG_AMPDU_IS_LAST;
1297                         sc->rx.ampdu_ref++;
1298                 }
1299
1300                 if (rs->rs_flags & ATH9K_RX_DELIM_CRC_PRE)
1301                         rxs->flag |= RX_FLAG_AMPDU_DELIM_CRC_ERROR;
1302         }
1303 }
1304
1305 int ath_rx_tasklet(struct ath_softc *sc, int flush, bool hp)
1306 {
1307         struct ath_buf *bf;
1308         struct sk_buff *skb = NULL, *requeue_skb, *hdr_skb;
1309         struct ieee80211_rx_status *rxs;
1310         struct ath_hw *ah = sc->sc_ah;
1311         struct ath_common *common = ath9k_hw_common(ah);
1312         struct ieee80211_hw *hw = sc->hw;
1313         int retval;
1314         struct ath_rx_status rs;
1315         enum ath9k_rx_qtype qtype;
1316         bool edma = !!(ah->caps.hw_caps & ATH9K_HW_CAP_EDMA);
1317         int dma_type;
1318         u64 tsf = 0;
1319         unsigned long flags;
1320         dma_addr_t new_buf_addr;
1321
1322         if (edma)
1323                 dma_type = DMA_BIDIRECTIONAL;
1324         else
1325                 dma_type = DMA_FROM_DEVICE;
1326
1327         qtype = hp ? ATH9K_RX_QUEUE_HP : ATH9K_RX_QUEUE_LP;
1328
1329         tsf = ath9k_hw_gettsf64(ah);
1330
1331         do {
1332                 bool decrypt_error = false;
1333
1334                 memset(&rs, 0, sizeof(rs));
1335                 if (edma)
1336                         bf = ath_edma_get_next_rx_buf(sc, &rs, qtype);
1337                 else
1338                         bf = ath_get_next_rx_buf(sc, &rs);
1339
1340                 if (!bf)
1341                         break;
1342
1343                 skb = bf->bf_mpdu;
1344                 if (!skb)
1345                         continue;
1346
1347                 /*
1348                  * Take frame header from the first fragment and RX status from
1349                  * the last one.
1350                  */
1351                 if (sc->rx.frag)
1352                         hdr_skb = sc->rx.frag;
1353                 else
1354                         hdr_skb = skb;
1355
1356                 rxs = IEEE80211_SKB_RXCB(hdr_skb);
1357                 memset(rxs, 0, sizeof(struct ieee80211_rx_status));
1358
1359                 retval = ath9k_rx_skb_preprocess(sc, hdr_skb, &rs, rxs,
1360                                                  &decrypt_error, tsf);
1361                 if (retval)
1362                         goto requeue_drop_frag;
1363
1364                 /* Ensure we always have an skb to requeue once we are done
1365                  * processing the current buffer's skb */
1366                 requeue_skb = ath_rxbuf_alloc(common, common->rx_bufsize, GFP_ATOMIC);
1367
1368                 /* If there is no memory we ignore the current RX'd frame,
1369                  * tell hardware it can give us a new frame using the old
1370                  * skb and put it at the tail of the sc->rx.rxbuf list for
1371                  * processing. */
1372                 if (!requeue_skb) {
1373                         RX_STAT_INC(rx_oom_err);
1374                         goto requeue_drop_frag;
1375                 }
1376
1377                 /* We will now give hardware our shiny new allocated skb */
1378                 new_buf_addr = dma_map_single(sc->dev, requeue_skb->data,
1379                                               common->rx_bufsize, dma_type);
1380                 if (unlikely(dma_mapping_error(sc->dev, new_buf_addr))) {
1381                         dev_kfree_skb_any(requeue_skb);
1382                         goto requeue_drop_frag;
1383                 }
1384
1385                 /* Unmap the frame */
1386                 dma_unmap_single(sc->dev, bf->bf_buf_addr,
1387                                  common->rx_bufsize, dma_type);
1388
1389                 bf->bf_mpdu = requeue_skb;
1390                 bf->bf_buf_addr = new_buf_addr;
1391
1392                 skb_put(skb, rs.rs_datalen + ah->caps.rx_status_len);
1393                 if (ah->caps.rx_status_len)
1394                         skb_pull(skb, ah->caps.rx_status_len);
1395
1396                 if (!rs.rs_more)
1397                         ath9k_rx_skb_postprocess(common, hdr_skb, &rs,
1398                                                  rxs, decrypt_error);
1399
1400                 if (rs.rs_more) {
1401                         RX_STAT_INC(rx_frags);
1402                         /*
1403                          * rs_more indicates chained descriptors which can be
1404                          * used to link buffers together for a sort of
1405                          * scatter-gather operation.
1406                          */
1407                         if (sc->rx.frag) {
1408                                 /* too many fragments - cannot handle frame */
1409                                 dev_kfree_skb_any(sc->rx.frag);
1410                                 dev_kfree_skb_any(skb);
1411                                 RX_STAT_INC(rx_too_many_frags_err);
1412                                 skb = NULL;
1413                         }
1414                         sc->rx.frag = skb;
1415                         goto requeue;
1416                 }
1417
1418                 if (sc->rx.frag) {
1419                         int space = skb->len - skb_tailroom(hdr_skb);
1420
1421                         if (pskb_expand_head(hdr_skb, 0, space, GFP_ATOMIC) < 0) {
1422                                 dev_kfree_skb(skb);
1423                                 RX_STAT_INC(rx_oom_err);
1424                                 goto requeue_drop_frag;
1425                         }
1426
1427                         sc->rx.frag = NULL;
1428
1429                         skb_copy_from_linear_data(skb, skb_put(hdr_skb, skb->len),
1430                                                   skb->len);
1431                         dev_kfree_skb_any(skb);
1432                         skb = hdr_skb;
1433                 }
1434
1435                 if (rxs->flag & RX_FLAG_MMIC_STRIPPED)
1436                         skb_trim(skb, skb->len - 8);
1437
1438                 spin_lock_irqsave(&sc->sc_pm_lock, flags);
1439                 if ((sc->ps_flags & (PS_WAIT_FOR_BEACON |
1440                                      PS_WAIT_FOR_CAB |
1441                                      PS_WAIT_FOR_PSPOLL_DATA)) ||
1442                     ath9k_check_auto_sleep(sc))
1443                         ath_rx_ps(sc, skb, rs.is_mybeacon);
1444                 spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
1445
1446                 ath9k_antenna_check(sc, &rs);
1447
1448                 ath9k_apply_ampdu_details(sc, &rs, rxs);
1449
1450                 ieee80211_rx(hw, skb);
1451
1452 requeue_drop_frag:
1453                 if (sc->rx.frag) {
1454                         dev_kfree_skb_any(sc->rx.frag);
1455                         sc->rx.frag = NULL;
1456                 }
1457 requeue:
1458                 list_add_tail(&bf->list, &sc->rx.rxbuf);
1459                 if (flush)
1460                         continue;
1461
1462                 if (edma) {
1463                         ath_rx_edma_buf_link(sc, qtype);
1464                 } else {
1465                         ath_rx_buf_relink(sc, bf);
1466                         ath9k_hw_rxena(ah);
1467                 }
1468         } while (1);
1469
1470         if (!(ah->imask & ATH9K_INT_RXEOL)) {
1471                 ah->imask |= (ATH9K_INT_RXEOL | ATH9K_INT_RXORN);
1472                 ath9k_hw_set_interrupts(ah);
1473         }
1474
1475         return 0;
1476 }