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

   1 /*
   2  * Copyright (c) 2008-2009 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 "ath9k.h"
  18
  19 static struct ieee80211_hw * ath_get_virt_hw(struct ath_softc *sc,
  20                                              struct ieee80211_hdr *hdr)
  21 {
  22         struct ieee80211_hw *hw = sc->pri_wiphy->hw;
  23         int i;
  24
  25         spin_lock_bh(&sc->wiphy_lock);
  26         for (i = 0; i < sc->num_sec_wiphy; i++) {
  27                 struct ath_wiphy *aphy = sc->sec_wiphy[i];
  28                 if (aphy == NULL)
  29                         continue;
  30                 if (compare_ether_addr(hdr->addr1, aphy->hw->wiphy->perm_addr)
  31                     == 0) {
  32                         hw = aphy->hw;
  33                         break;
  34                 }
  35         }
  36         spin_unlock_bh(&sc->wiphy_lock);
  37         return hw;
  38 }
  39
  40 /*
  41  * Setup and link descriptors.
  42  *
  43  * 11N: we can no longer afford to self link the last descriptor.
  44  * MAC acknowledges BA status as long as it copies frames to host
  45  * buffer (or rx fifo). This can incorrectly acknowledge packets
  46  * to a sender if last desc is self-linked.
  47  */
  48 static void ath_rx_buf_link(struct ath_softc *sc, struct ath_buf *bf)
  49 {
  50         struct ath_hw *ah = sc->sc_ah;
  51         struct ath_common *common = ath9k_hw_common(ah);
  52         struct ath_desc *ds;
  53         struct sk_buff *skb;
  54
  55         ATH_RXBUF_RESET(bf);
  56
  57         ds = bf->bf_desc;
  58         ds->ds_link = 0; /* link to null */
  59         ds->ds_data = bf->bf_buf_addr;
  60
  61         /* virtual addr of the beginning of the buffer. */
  62         skb = bf->bf_mpdu;
  63         BUG_ON(skb == NULL);
  64         ds->ds_vdata = skb->data;
  65
  66         /*
  67          * setup rx descriptors. The rx_bufsize here tells the hardware
  68          * how much data it can DMA to us and that we are prepared
  69          * to process
  70          */
  71         ath9k_hw_setuprxdesc(ah, ds,
  72                              common->rx_bufsize,
  73                              0);
  74
  75         if (sc->rx.rxlink == NULL)
  76                 ath9k_hw_putrxbuf(ah, bf->bf_daddr);
  77         else
  78                 *sc->rx.rxlink = bf->bf_daddr;
  79
  80         sc->rx.rxlink = &ds->ds_link;
  81         ath9k_hw_rxena(ah);
  82 }
  83
  84 static void ath_setdefantenna(struct ath_softc *sc, u32 antenna)
  85 {
  86         /* XXX block beacon interrupts */
  87         ath9k_hw_setantenna(sc->sc_ah, antenna);
  88         sc->rx.defant = antenna;
  89         sc->rx.rxotherant = 0;
  90 }
  91
  92 /* Assumes you've already done the endian to CPU conversion */
  93 static bool ath9k_rx_accept(struct ath_common *common,
  94                             struct sk_buff *skb,
  95                             struct ieee80211_rx_status *rxs,
  96                             struct ath_rx_status *rx_stats,
  97                             bool *decrypt_error)
  98 {
  99         struct ath_hw *ah = common->ah;
 100         struct ieee80211_hdr *hdr;
 101         __le16 fc;
 102
 103         hdr = (struct ieee80211_hdr *) skb->data;
 104         fc = hdr->frame_control;
 105
 106         if (!rx_stats->rs_datalen)
 107                 return false;
 108         /*
 109          * rs_status follows rs_datalen so if rs_datalen is too large
 110          * we can take a hint that hardware corrupted it, so ignore
 111          * those frames.
 112          */
 113         if (rx_stats->rs_datalen > common->rx_bufsize)
 114                 return false;
 115
 116         if (rx_stats->rs_more) {
 117                 /*
 118                  * Frame spans multiple descriptors; this cannot happen yet
 119                  * as we don't support jumbograms. If not in monitor mode,
 120                  * discard the frame. Enable this if you want to see
 121                  * error frames in Monitor mode.
 122                  */
 123                 if (ah->opmode != NL80211_IFTYPE_MONITOR)
 124                         return false;
 125         } else if (rx_stats->rs_status != 0) {
 126                 if (rx_stats->rs_status & ATH9K_RXERR_CRC)
 127                         rxs->flag |= RX_FLAG_FAILED_FCS_CRC;
 128                 if (rx_stats->rs_status & ATH9K_RXERR_PHY)
 129                         return false;
 130
 131                 if (rx_stats->rs_status & ATH9K_RXERR_DECRYPT) {
 132                         *decrypt_error = true;
 133                 } else if (rx_stats->rs_status & ATH9K_RXERR_MIC) {
 134                         if (ieee80211_is_ctl(fc))
 135                                 /*
 136                                  * Sometimes, we get invalid
 137                                  * MIC failures on valid control frames.
 138                                  * Remove these mic errors.
 139                                  */
 140                                 rx_stats->rs_status &= ~ATH9K_RXERR_MIC;
 141                         else
 142                                 rxs->flag |= RX_FLAG_MMIC_ERROR;
 143                 }
 144                 /*
 145                  * Reject error frames with the exception of
 146                  * decryption and MIC failures. For monitor mode,
 147                  * we also ignore the CRC error.
 148                  */
 149                 if (ah->opmode == NL80211_IFTYPE_MONITOR) {
 150                         if (rx_stats->rs_status &
 151                             ~(ATH9K_RXERR_DECRYPT | ATH9K_RXERR_MIC |
 152                               ATH9K_RXERR_CRC))
 153                                 return false;
 154                 } else {
 155                         if (rx_stats->rs_status &
 156                             ~(ATH9K_RXERR_DECRYPT | ATH9K_RXERR_MIC)) {
 157                                 return false;
 158                         }
 159                 }
 160         }
 161         return true;
 162 }
 163
 164 static u8 ath9k_process_rate(struct ath_common *common,
 165                              struct ieee80211_hw *hw,
 166                              struct ath_rx_status *rx_stats,
 167                              struct ieee80211_rx_status *rxs,
 168                              struct sk_buff *skb)
 169 {
 170         struct ieee80211_supported_band *sband;
 171         enum ieee80211_band band;
 172         unsigned int i = 0;
 173
 174         band = hw->conf.channel->band;
 175         sband = hw->wiphy->bands[band];
 176
 177         if (rx_stats->rs_rate & 0x80) {
 178                 /* HT rate */
 179                 rxs->flag |= RX_FLAG_HT;
 180                 if (rx_stats->rs_flags & ATH9K_RX_2040)
 181                         rxs->flag |= RX_FLAG_40MHZ;
 182                 if (rx_stats->rs_flags & ATH9K_RX_GI)
 183                         rxs->flag |= RX_FLAG_SHORT_GI;
 184                 return rx_stats->rs_rate & 0x7f;
 185         }
 186
 187         for (i = 0; i < sband->n_bitrates; i++) {
 188                 if (sband->bitrates[i].hw_value == rx_stats->rs_rate)
 189                         return i;
 190                 if (sband->bitrates[i].hw_value_short == rx_stats->rs_rate) {
 191                         rxs->flag |= RX_FLAG_SHORTPRE;
 192                         return i;
 193                 }
 194         }
 195
 196         /* No valid hardware bitrate found -- we should not get here */
 197         ath_print(common, ATH_DBG_XMIT, "unsupported hw bitrate detected "
 198                   "0x%02x using 1 Mbit\n", rx_stats->rs_rate);
 199         if ((common->debug_mask & ATH_DBG_XMIT))
 200                 print_hex_dump_bytes("", DUMP_PREFIX_NONE, skb->data, skb->len);
 201
 202         return 0;
 203 }
 204
 205 /*
 206  * Theory for reporting quality:
 207  *
 208  * At a hardware RSSI of 45 you will be able to use MCS 7  reliably.
 209  * At a hardware RSSI of 45 you will be able to use MCS 15 reliably.
 210  * At a hardware RSSI of 35 you should be able use 54 Mbps reliably.
 211  *
 212  * MCS 7  is the highets MCS index usable by a 1-stream device.
 213  * MCS 15 is the highest MCS index usable by a 2-stream device.
 214  *
 215  * All ath9k devices are either 1-stream or 2-stream.
 216  *
 217  * How many bars you see is derived from the qual reporting.
 218  *
 219  * A more elaborate scheme can be used here but it requires tables
 220  * of SNR/throughput for each possible mode used. For the MCS table
 221  * you can refer to the wireless wiki:
 222  *
 223  * http://wireless.kernel.org/en/developers/Documentation/ieee80211/802.11n
 224  *
 225  */
 226 static int ath9k_compute_qual(struct ieee80211_hw *hw,
 227                               struct ath_rx_status *rx_stats)
 228 {
 229         int qual;
 230
 231         if (conf_is_ht(&hw->conf))
 232                 qual =  rx_stats->rs_rssi * 100 / 45;
 233         else
 234                 qual =  rx_stats->rs_rssi * 100 / 35;
 235
 236         /*
 237          * rssi can be more than 45 though, anything above that
 238          * should be considered at 100%
 239          */
 240         if (qual > 100)
 241                 qual = 100;
 242
 243         return qual;
 244 }
 245
 246 static void ath9k_process_rssi(struct ath_common *common,
 247                                struct ieee80211_hw *hw,
 248                                struct sk_buff *skb,
 249                                struct ath_rx_status *rx_stats)
 250 {
 251         struct ath_hw *ah = common->ah;
 252         struct ieee80211_sta *sta;
 253         struct ieee80211_hdr *hdr;
 254         struct ath_node *an;
 255         int last_rssi = ATH_RSSI_DUMMY_MARKER;
 256         __le16 fc;
 257
 258         hdr = (struct ieee80211_hdr *)skb->data;
 259         fc = hdr->frame_control;
 260
 261         rcu_read_lock();
 262         /* XXX: use ieee80211_find_sta! */
 263         sta = ieee80211_find_sta_by_hw(hw, hdr->addr2);
 264         if (sta) {
 265                 an = (struct ath_node *) sta->drv_priv;
 266                 if (rx_stats->rs_rssi != ATH9K_RSSI_BAD &&
 267                    !rx_stats->rs_moreaggr)
 268                         ATH_RSSI_LPF(an->last_rssi, rx_stats->rs_rssi);
 269                 last_rssi = an->last_rssi;
 270         }
 271         rcu_read_unlock();
 272
 273         if (likely(last_rssi != ATH_RSSI_DUMMY_MARKER))
 274                 rx_stats->rs_rssi = ATH_EP_RND(last_rssi,
 275                                               ATH_RSSI_EP_MULTIPLIER);
 276         if (rx_stats->rs_rssi < 0)
 277                 rx_stats->rs_rssi = 0;
 278         else if (rx_stats->rs_rssi > 127)
 279                 rx_stats->rs_rssi = 127;
 280
 281         /* Update Beacon RSSI, this is used by ANI. */
 282         if (ieee80211_is_beacon(fc))
 283                 ah->stats.avgbrssi = rx_stats->rs_rssi;
 284 }
 285
 286 /*
 287  * For Decrypt or Demic errors, we only mark packet status here and always push
 288  * up the frame up to let mac80211 handle the actual error case, be it no
 289  * decryption key or real decryption error. This let us keep statistics there.
 290  */
 291 static int ath_rx_prepare(struct ath_common *common,
 292                           struct ieee80211_hw *hw,
 293                           struct sk_buff *skb, struct ath_rx_status *rx_stats,
 294                           struct ieee80211_rx_status *rx_status,
 295                           bool *decrypt_error)
 296 {
 297         struct ath_hw *ah = common->ah;
 298
 299         if (!ath9k_rx_accept(common, skb, rx_status, rx_stats, decrypt_error))
 300                 goto rx_next;
 301
 302         ath9k_process_rssi(common, hw, skb, rx_stats);
 303
 304         rx_status->rate_idx = ath9k_process_rate(common, hw,
 305                                                  rx_stats, rx_status, skb);
 306         rx_status->mactime = ath9k_hw_extend_tsf(ah, rx_stats->rs_tstamp);
 307         rx_status->band = hw->conf.channel->band;
 308         rx_status->freq = hw->conf.channel->center_freq;
 309         rx_status->noise = common->ani.noise_floor;
 310         rx_status->signal = ATH_DEFAULT_NOISE_FLOOR + rx_stats->rs_rssi;
 311         rx_status->antenna = rx_stats->rs_antenna;
 312         rx_status->qual = ath9k_compute_qual(hw, rx_stats);
 313         rx_status->flag |= RX_FLAG_TSFT;
 314
 315         return 1;
 316 rx_next:
 317         return 0;
 318 }
 319
 320 static void ath_opmode_init(struct ath_softc *sc)
 321 {
 322         struct ath_hw *ah = sc->sc_ah;
 323         struct ath_common *common = ath9k_hw_common(ah);
 324
 325         u32 rfilt, mfilt[2];
 326
 327         /* configure rx filter */
 328         rfilt = ath_calcrxfilter(sc);
 329         ath9k_hw_setrxfilter(ah, rfilt);
 330
 331         /* configure bssid mask */
 332         if (ah->caps.hw_caps & ATH9K_HW_CAP_BSSIDMASK)
 333                 ath_hw_setbssidmask(common);
 334
 335         /* configure operational mode */
 336         ath9k_hw_setopmode(ah);
 337
 338         /* Handle any link-level address change. */
 339         ath9k_hw_setmac(ah, common->macaddr);
 340
 341         /* calculate and install multicast filter */
 342         mfilt[0] = mfilt[1] = ~0;
 343         ath9k_hw_setmcastfilter(ah, mfilt[0], mfilt[1]);
 344 }
 345
 346 int ath_rx_init(struct ath_softc *sc, int nbufs)
 347 {
 348         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
 349         struct sk_buff *skb;
 350         struct ath_buf *bf;
 351         int error = 0;
 352
 353         spin_lock_init(&sc->rx.rxflushlock);
 354         sc->sc_flags &= ~SC_OP_RXFLUSH;
 355         spin_lock_init(&sc->rx.rxbuflock);
 356
 357         common->rx_bufsize = roundup(IEEE80211_MAX_MPDU_LEN,
 358                                      min(common->cachelsz, (u16)64));
 359
 360         ath_print(common, ATH_DBG_CONFIG, "cachelsz %u rxbufsize %u\n",
 361                   common->cachelsz, common->rx_bufsize);
 362
 363         /* Initialize rx descriptors */
 364
 365         error = ath_descdma_setup(sc, &sc->rx.rxdma, &sc->rx.rxbuf,
 366                                   "rx", nbufs, 1);
 367         if (error != 0) {
 368                 ath_print(common, ATH_DBG_FATAL,
 369                           "failed to allocate rx descriptors: %d\n", error);
 370                 goto err;
 371         }
 372
 373         list_for_each_entry(bf, &sc->rx.rxbuf, list) {
 374                 skb = ath_rxbuf_alloc(common, common->rx_bufsize, GFP_KERNEL);
 375                 if (skb == NULL) {
 376                         error = -ENOMEM;
 377                         goto err;
 378                 }
 379
 380                 bf->bf_mpdu = skb;
 381                 bf->bf_buf_addr = dma_map_single(sc->dev, skb->data,
 382                                                  common->rx_bufsize,
 383                                                  DMA_FROM_DEVICE);
 384                 if (unlikely(dma_mapping_error(sc->dev,
 385                                                bf->bf_buf_addr))) {
 386                         dev_kfree_skb_any(skb);
 387                         bf->bf_mpdu = NULL;
 388                         ath_print(common, ATH_DBG_FATAL,
 389                                   "dma_mapping_error() on RX init\n");
 390                         error = -ENOMEM;
 391                         goto err;
 392                 }
 393                 bf->bf_dmacontext = bf->bf_buf_addr;
 394         }
 395         sc->rx.rxlink = NULL;
 396
 397 err:
 398         if (error)
 399                 ath_rx_cleanup(sc);
 400
 401         return error;
 402 }
 403
 404 void ath_rx_cleanup(struct ath_softc *sc)
 405 {
 406         struct ath_hw *ah = sc->sc_ah;
 407         struct ath_common *common = ath9k_hw_common(ah);
 408         struct sk_buff *skb;
 409         struct ath_buf *bf;
 410
 411         list_for_each_entry(bf, &sc->rx.rxbuf, list) {
 412                 skb = bf->bf_mpdu;
 413                 if (skb) {
 414                         dma_unmap_single(sc->dev, bf->bf_buf_addr,
 415                                          common->rx_bufsize, DMA_FROM_DEVICE);
 416                         dev_kfree_skb(skb);
 417                 }
 418         }
 419
 420         if (sc->rx.rxdma.dd_desc_len != 0)
 421                 ath_descdma_cleanup(sc, &sc->rx.rxdma, &sc->rx.rxbuf);
 422 }
 423
 424 /*
 425  * Calculate the receive filter according to the
 426  * operating mode and state:
 427  *
 428  * o always accept unicast, broadcast, and multicast traffic
 429  * o maintain current state of phy error reception (the hal
 430  *   may enable phy error frames for noise immunity work)
 431  * o probe request frames are accepted only when operating in
 432  *   hostap, adhoc, or monitor modes
 433  * o enable promiscuous mode according to the interface state
 434  * o accept beacons:
 435  *   - when operating in adhoc mode so the 802.11 layer creates
 436  *     node table entries for peers,
 437  *   - when operating in station mode for collecting rssi data when
 438  *     the station is otherwise quiet, or
 439  *   - when operating as a repeater so we see repeater-sta beacons
 440  *   - when scanning
 441  */
 442
 443 u32 ath_calcrxfilter(struct ath_softc *sc)
 444 {
 445 #define RX_FILTER_PRESERVE (ATH9K_RX_FILTER_PHYERR | ATH9K_RX_FILTER_PHYRADAR)
 446
 447         u32 rfilt;
 448
 449         rfilt = (ath9k_hw_getrxfilter(sc->sc_ah) & RX_FILTER_PRESERVE)
 450                 | ATH9K_RX_FILTER_UCAST | ATH9K_RX_FILTER_BCAST
 451                 | ATH9K_RX_FILTER_MCAST;
 452
 453         /* If not a STA, enable processing of Probe Requests */
 454         if (sc->sc_ah->opmode != NL80211_IFTYPE_STATION)
 455                 rfilt |= ATH9K_RX_FILTER_PROBEREQ;
 456
 457         /*
 458          * Set promiscuous mode when FIF_PROMISC_IN_BSS is enabled for station
 459          * mode interface or when in monitor mode. AP mode does not need this
 460          * since it receives all in-BSS frames anyway.
 461          */
 462         if (((sc->sc_ah->opmode != NL80211_IFTYPE_AP) &&
 463              (sc->rx.rxfilter & FIF_PROMISC_IN_BSS)) ||
 464             (sc->sc_ah->opmode == NL80211_IFTYPE_MONITOR))
 465                 rfilt |= ATH9K_RX_FILTER_PROM;
 466
 467         if (sc->rx.rxfilter & FIF_CONTROL)
 468                 rfilt |= ATH9K_RX_FILTER_CONTROL;
 469
 470         if ((sc->sc_ah->opmode == NL80211_IFTYPE_STATION) &&
 471             !(sc->rx.rxfilter & FIF_BCN_PRBRESP_PROMISC))
 472                 rfilt |= ATH9K_RX_FILTER_MYBEACON;
 473         else
 474                 rfilt |= ATH9K_RX_FILTER_BEACON;
 475
 476         if ((AR_SREV_9280_10_OR_LATER(sc->sc_ah) ||
 477             AR_SREV_9285_10_OR_LATER(sc->sc_ah)) &&
 478             (sc->sc_ah->opmode == NL80211_IFTYPE_AP) &&
 479             (sc->rx.rxfilter & FIF_PSPOLL))
 480                 rfilt |= ATH9K_RX_FILTER_PSPOLL;
 481
 482         if (conf_is_ht(&sc->hw->conf))
 483                 rfilt |= ATH9K_RX_FILTER_COMP_BAR;
 484
 485         if (sc->sec_wiphy || (sc->rx.rxfilter & FIF_OTHER_BSS)) {
 486                 /* TODO: only needed if more than one BSSID is in use in
 487                  * station/adhoc mode */
 488                 /* The following may also be needed for other older chips */
 489                 if (sc->sc_ah->hw_version.macVersion == AR_SREV_VERSION_9160)
 490                         rfilt |= ATH9K_RX_FILTER_PROM;
 491                 rfilt |= ATH9K_RX_FILTER_MCAST_BCAST_ALL;
 492         }
 493
 494         return rfilt;
 495
 496 #undef RX_FILTER_PRESERVE
 497 }
 498
 499 int ath_startrecv(struct ath_softc *sc)
 500 {
 501         struct ath_hw *ah = sc->sc_ah;
 502         struct ath_buf *bf, *tbf;
 503
 504         spin_lock_bh(&sc->rx.rxbuflock);
 505         if (list_empty(&sc->rx.rxbuf))
 506                 goto start_recv;
 507
 508         sc->rx.rxlink = NULL;
 509         list_for_each_entry_safe(bf, tbf, &sc->rx.rxbuf, list) {
 510                 ath_rx_buf_link(sc, bf);
 511         }
 512
 513         /* We could have deleted elements so the list may be empty now */
 514         if (list_empty(&sc->rx.rxbuf))
 515                 goto start_recv;
 516
 517         bf = list_first_entry(&sc->rx.rxbuf, struct ath_buf, list);
 518         ath9k_hw_putrxbuf(ah, bf->bf_daddr);
 519         ath9k_hw_rxena(ah);
 520
 521 start_recv:
 522         spin_unlock_bh(&sc->rx.rxbuflock);
 523         ath_opmode_init(sc);
 524         ath9k_hw_startpcureceive(ah);
 525
 526         return 0;
 527 }
 528
 529 bool ath_stoprecv(struct ath_softc *sc)
 530 {
 531         struct ath_hw *ah = sc->sc_ah;
 532         bool stopped;
 533
 534         ath9k_hw_stoppcurecv(ah);
 535         ath9k_hw_setrxfilter(ah, 0);
 536         stopped = ath9k_hw_stopdmarecv(ah);
 537         sc->rx.rxlink = NULL;
 538
 539         return stopped;
 540 }
 541
 542 void ath_flushrecv(struct ath_softc *sc)
 543 {
 544         spin_lock_bh(&sc->rx.rxflushlock);
 545         sc->sc_flags |= SC_OP_RXFLUSH;
 546         ath_rx_tasklet(sc, 1);
 547         sc->sc_flags &= ~SC_OP_RXFLUSH;
 548         spin_unlock_bh(&sc->rx.rxflushlock);
 549 }
 550
 551 static bool ath_beacon_dtim_pending_cab(struct sk_buff *skb)
 552 {
 553         /* Check whether the Beacon frame has DTIM indicating buffered bc/mc */
 554         struct ieee80211_mgmt *mgmt;
 555         u8 *pos, *end, id, elen;
 556         struct ieee80211_tim_ie *tim;
 557
 558         mgmt = (struct ieee80211_mgmt *)skb->data;
 559         pos = mgmt->u.beacon.variable;
 560         end = skb->data + skb->len;
 561
 562         while (pos + 2 < end) {
 563                 id = *pos++;
 564                 elen = *pos++;
 565                 if (pos + elen > end)
 566                         break;
 567
 568                 if (id == WLAN_EID_TIM) {
 569                         if (elen < sizeof(*tim))
 570                                 break;
 571                         tim = (struct ieee80211_tim_ie *) pos;
 572                         if (tim->dtim_count != 0)
 573                                 break;
 574                         return tim->bitmap_ctrl & 0x01;
 575                 }
 576
 577                 pos += elen;
 578         }
 579
 580         return false;
 581 }
 582
 583 static void ath_rx_ps_beacon(struct ath_softc *sc, struct sk_buff *skb)
 584 {
 585         struct ieee80211_mgmt *mgmt;
 586         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
 587
 588         if (skb->len < 24 + 8 + 2 + 2)
 589                 return;
 590
 591         mgmt = (struct ieee80211_mgmt *)skb->data;
 592         if (memcmp(common->curbssid, mgmt->bssid, ETH_ALEN) != 0)
 593                 return; /* not from our current AP */
 594
 595         sc->sc_flags &= ~SC_OP_WAIT_FOR_BEACON;
 596
 597         if (sc->sc_flags & SC_OP_BEACON_SYNC) {
 598                 sc->sc_flags &= ~SC_OP_BEACON_SYNC;
 599                 ath_print(common, ATH_DBG_PS,
 600                           "Reconfigure Beacon timers based on "
 601                           "timestamp from the AP\n");
 602                 ath_beacon_config(sc, NULL);
 603         }
 604
 605         if (ath_beacon_dtim_pending_cab(skb)) {
 606                 /*
 607                  * Remain awake waiting for buffered broadcast/multicast
 608                  * frames. If the last broadcast/multicast frame is not
 609                  * received properly, the next beacon frame will work as
 610                  * a backup trigger for returning into NETWORK SLEEP state,
 611                  * so we are waiting for it as well.
 612                  */
 613                 ath_print(common, ATH_DBG_PS, "Received DTIM beacon indicating "
 614                           "buffered broadcast/multicast frame(s)\n");
 615                 sc->sc_flags |= SC_OP_WAIT_FOR_CAB | SC_OP_WAIT_FOR_BEACON;
 616                 return;
 617         }
 618
 619         if (sc->sc_flags & SC_OP_WAIT_FOR_CAB) {
 620                 /*
 621                  * This can happen if a broadcast frame is dropped or the AP
 622                  * fails to send a frame indicating that all CAB frames have
 623                  * been delivered.
 624                  */
 625                 sc->sc_flags &= ~SC_OP_WAIT_FOR_CAB;
 626                 ath_print(common, ATH_DBG_PS,
 627                           "PS wait for CAB frames timed out\n");
 628         }
 629 }
 630
 631 static void ath_rx_ps(struct ath_softc *sc, struct sk_buff *skb)
 632 {
 633         struct ieee80211_hdr *hdr;
 634         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
 635
 636         hdr = (struct ieee80211_hdr *)skb->data;
 637
 638         /* Process Beacon and CAB receive in PS state */
 639         if ((sc->sc_flags & SC_OP_WAIT_FOR_BEACON) &&
 640             ieee80211_is_beacon(hdr->frame_control))
 641                 ath_rx_ps_beacon(sc, skb);
 642         else if ((sc->sc_flags & SC_OP_WAIT_FOR_CAB) &&
 643                  (ieee80211_is_data(hdr->frame_control) ||
 644                   ieee80211_is_action(hdr->frame_control)) &&
 645                  is_multicast_ether_addr(hdr->addr1) &&
 646                  !ieee80211_has_moredata(hdr->frame_control)) {
 647                 /*
 648                  * No more broadcast/multicast frames to be received at this
 649                  * point.
 650                  */
 651                 sc->sc_flags &= ~SC_OP_WAIT_FOR_CAB;
 652                 ath_print(common, ATH_DBG_PS,
 653                           "All PS CAB frames received, back to sleep\n");
 654         } else if ((sc->sc_flags & SC_OP_WAIT_FOR_PSPOLL_DATA) &&
 655                    !is_multicast_ether_addr(hdr->addr1) &&
 656                    !ieee80211_has_morefrags(hdr->frame_control)) {
 657                 sc->sc_flags &= ~SC_OP_WAIT_FOR_PSPOLL_DATA;
 658                 ath_print(common, ATH_DBG_PS,
 659                           "Going back to sleep after having received "
 660                           "PS-Poll data (0x%x)\n",
 661                         sc->sc_flags & (SC_OP_WAIT_FOR_BEACON |
 662                                         SC_OP_WAIT_FOR_CAB |
 663                                         SC_OP_WAIT_FOR_PSPOLL_DATA |
 664                                         SC_OP_WAIT_FOR_TX_ACK));
 665         }
 666 }
 667
 668 static void ath_rx_send_to_mac80211(struct ieee80211_hw *hw,
 669                                     struct ath_softc *sc, struct sk_buff *skb,
 670                                     struct ieee80211_rx_status *rxs)
 671 {
 672         struct ieee80211_hdr *hdr;
 673
 674         hdr = (struct ieee80211_hdr *)skb->data;
 675
 676         /* Send the frame to mac80211 */
 677         if (is_multicast_ether_addr(hdr->addr1)) {
 678                 int i;
 679                 /*
 680                  * Deliver broadcast/multicast frames to all suitable
 681                  * virtual wiphys.
 682                  */
 683                 /* TODO: filter based on channel configuration */
 684                 for (i = 0; i < sc->num_sec_wiphy; i++) {
 685                         struct ath_wiphy *aphy = sc->sec_wiphy[i];
 686                         struct sk_buff *nskb;
 687                         if (aphy == NULL)
 688                                 continue;
 689                         nskb = skb_copy(skb, GFP_ATOMIC);
 690                         if (!nskb)
 691                                 continue;
 692                         ieee80211_rx(aphy->hw, nskb);
 693                 }
 694                 ieee80211_rx(sc->hw, skb);
 695         } else
 696                 /* Deliver unicast frames based on receiver address */
 697                 ieee80211_rx(hw, skb);
 698 }
 699
 700 int ath_rx_tasklet(struct ath_softc *sc, int flush)
 701 {
 702 #define PA2DESC(_sc, _pa)                                               \
 703         ((struct ath_desc *)((caddr_t)(_sc)->rx.rxdma.dd_desc +         \
 704                              ((_pa) - (_sc)->rx.rxdma.dd_desc_paddr)))
 705
 706         struct ath_buf *bf;
 707         struct ath_desc *ds;
 708         struct ath_rx_status *rx_stats;
 709         struct sk_buff *skb = NULL, *requeue_skb;
 710         struct ieee80211_rx_status *rxs;
 711         struct ath_hw *ah = sc->sc_ah;
 712         struct ath_common *common = ath9k_hw_common(ah);
 713         /*
 714          * The hw can techncically differ from common->hw when using ath9k
 715          * virtual wiphy so to account for that we iterate over the active
 716          * wiphys and find the appropriate wiphy and therefore hw.
 717          */
 718         struct ieee80211_hw *hw = NULL;
 719         struct ieee80211_hdr *hdr;
 720         int hdrlen, padsize, retval;
 721         bool decrypt_error = false;
 722         u8 keyix;
 723         __le16 fc;
 724
 725         spin_lock_bh(&sc->rx.rxbuflock);
 726
 727         do {
 728                 /* If handling rx interrupt and flush is in progress => exit */
 729                 if ((sc->sc_flags & SC_OP_RXFLUSH) && (flush == 0))
 730                         break;
 731
 732                 if (list_empty(&sc->rx.rxbuf)) {
 733                         sc->rx.rxlink = NULL;
 734                         break;
 735                 }
 736
 737                 bf = list_first_entry(&sc->rx.rxbuf, struct ath_buf, list);
 738                 ds = bf->bf_desc;
 739
 740                 /*
 741                  * Must provide the virtual address of the current
 742                  * descriptor, the physical address, and the virtual
 743                  * address of the next descriptor in the h/w chain.
 744                  * This allows the HAL to look ahead to see if the
 745                  * hardware is done with a descriptor by checking the
 746                  * done bit in the following descriptor and the address
 747                  * of the current descriptor the DMA engine is working
 748                  * on.  All this is necessary because of our use of
 749                  * a self-linked list to avoid rx overruns.
 750                  */
 751                 retval = ath9k_hw_rxprocdesc(ah, ds,
 752                                              bf->bf_daddr,
 753                                              PA2DESC(sc, ds->ds_link),
 754                                              0);
 755                 if (retval == -EINPROGRESS) {
 756                         struct ath_buf *tbf;
 757                         struct ath_desc *tds;
 758
 759                         if (list_is_last(&bf->list, &sc->rx.rxbuf)) {
 760                                 sc->rx.rxlink = NULL;
 761                                 break;
 762                         }
 763
 764                         tbf = list_entry(bf->list.next, struct ath_buf, list);
 765
 766                         /*
 767                          * On some hardware the descriptor status words could
 768                          * get corrupted, including the done bit. Because of
 769                          * this, check if the next descriptor's done bit is
 770                          * set or not.
 771                          *
 772                          * If the next descriptor's done bit is set, the current
 773                          * descriptor has been corrupted. Force s/w to discard
 774                          * this descriptor and continue...
 775                          */
 776
 777                         tds = tbf->bf_desc;
 778                         retval = ath9k_hw_rxprocdesc(ah, tds, tbf->bf_daddr,
 779                                              PA2DESC(sc, tds->ds_link), 0);
 780                         if (retval == -EINPROGRESS) {
 781                                 break;
 782                         }
 783                 }
 784
 785                 skb = bf->bf_mpdu;
 786                 if (!skb)
 787                         continue;
 788
 789                 /*
 790                  * Synchronize the DMA transfer with CPU before
 791                  * 1. accessing the frame
 792                  * 2. requeueing the same buffer to h/w
 793                  */
 794                 dma_sync_single_for_cpu(sc->dev, bf->bf_buf_addr,
 795                                 common->rx_bufsize,
 796                                 DMA_FROM_DEVICE);
 797
 798                 hdr = (struct ieee80211_hdr *) skb->data;
 799                 rxs =  IEEE80211_SKB_RXCB(skb);
 800
 801                 hw = ath_get_virt_hw(sc, hdr);
 802                 rx_stats = &ds->ds_rxstat;
 803
 804                 /*
 805                  * If we're asked to flush receive queue, directly
 806                  * chain it back at the queue without processing it.
 807                  */
 808                 if (flush)
 809                         goto requeue;
 810
 811                 if (!ath_rx_prepare(common, hw, skb, rx_stats,
 812                                     rxs, &decrypt_error))
 813                         goto requeue;
 814
 815                 /* Ensure we always have an skb to requeue once we are done
 816                  * processing the current buffer's skb */
 817                 requeue_skb = ath_rxbuf_alloc(common, common->rx_bufsize, GFP_ATOMIC);
 818
 819                 /* If there is no memory we ignore the current RX'd frame,
 820                  * tell hardware it can give us a new frame using the old
 821                  * skb and put it at the tail of the sc->rx.rxbuf list for
 822                  * processing. */
 823                 if (!requeue_skb)
 824                         goto requeue;
 825
 826                 /* Unmap the frame */
 827                 dma_unmap_single(sc->dev, bf->bf_buf_addr,
 828                                  common->rx_bufsize,
 829                                  DMA_FROM_DEVICE);
 830
 831                 skb_put(skb, rx_stats->rs_datalen);
 832
 833                 /* see if any padding is done by the hw and remove it */
 834                 hdrlen = ieee80211_get_hdrlen_from_skb(skb);
 835                 fc = hdr->frame_control;
 836
 837                 /* The MAC header is padded to have 32-bit boundary if the
 838                  * packet payload is non-zero. The general calculation for
 839                  * padsize would take into account odd header lengths:
 840                  * padsize = (4 - hdrlen % 4) % 4; However, since only
 841                  * even-length headers are used, padding can only be 0 or 2
 842                  * bytes and we can optimize this a bit. In addition, we must
 843                  * not try to remove padding from short control frames that do
 844                  * not have payload. */
 845                 padsize = hdrlen & 3;
 846                 if (padsize && hdrlen >= 24) {
 847                         memmove(skb->data + padsize, skb->data, hdrlen);
 848                         skb_pull(skb, padsize);
 849                 }
 850
 851                 keyix = rx_stats->rs_keyix;
 852
 853                 if (!(keyix == ATH9K_RXKEYIX_INVALID) && !decrypt_error) {
 854                         rxs->flag |= RX_FLAG_DECRYPTED;
 855                 } else if (ieee80211_has_protected(fc)
 856                            && !decrypt_error && skb->len >= hdrlen + 4) {
 857                         keyix = skb->data[hdrlen + 3] >> 6;
 858
 859                         if (test_bit(keyix, sc->keymap))
 860                                 rxs->flag |= RX_FLAG_DECRYPTED;
 861                 }
 862                 if (ah->sw_mgmt_crypto &&
 863                     (rxs->flag & RX_FLAG_DECRYPTED) &&
 864                     ieee80211_is_mgmt(fc))
 865                         /* Use software decrypt for management frames. */
 866                         rxs->flag &= ~RX_FLAG_DECRYPTED;
 867
 868                 /* We will now give hardware our shiny new allocated skb */
 869                 bf->bf_mpdu = requeue_skb;
 870                 bf->bf_buf_addr = dma_map_single(sc->dev, requeue_skb->data,
 871                                                  common->rx_bufsize,
 872                                                  DMA_FROM_DEVICE);
 873                 if (unlikely(dma_mapping_error(sc->dev,
 874                           bf->bf_buf_addr))) {
 875                         dev_kfree_skb_any(requeue_skb);
 876                         bf->bf_mpdu = NULL;
 877                         ath_print(common, ATH_DBG_FATAL,
 878                                   "dma_mapping_error() on RX\n");
 879                         ath_rx_send_to_mac80211(hw, sc, skb, rxs);
 880                         break;
 881                 }
 882                 bf->bf_dmacontext = bf->bf_buf_addr;
 883
 884                 /*
 885                  * change the default rx antenna if rx diversity chooses the
 886                  * other antenna 3 times in a row.
 887                  */
 888                 if (sc->rx.defant != ds->ds_rxstat.rs_antenna) {
 889                         if (++sc->rx.rxotherant >= 3)
 890                                 ath_setdefantenna(sc, rx_stats->rs_antenna);
 891                 } else {
 892                         sc->rx.rxotherant = 0;
 893                 }
 894
 895                 if (unlikely(sc->sc_flags & (SC_OP_WAIT_FOR_BEACON |
 896                                              SC_OP_WAIT_FOR_CAB |
 897                                              SC_OP_WAIT_FOR_PSPOLL_DATA)))
 898                         ath_rx_ps(sc, skb);
 899
 900                 ath_rx_send_to_mac80211(hw, sc, skb, rxs);
 901
 902 requeue:
 903                 list_move_tail(&bf->list, &sc->rx.rxbuf);
 904                 ath_rx_buf_link(sc, bf);
 905         } while (1);
 906
 907         spin_unlock_bh(&sc->rx.rxbuflock);
 908
 909         return 0;
 910 #undef PA2DESC
 911 }