drivers/net/wireless/ti/wlcore/tx.c

   1 /*
   2  * This file is part of wl1271
   3  *
   4  * Copyright (C) 2009 Nokia Corporation
   5  *
   6  * Contact: Luciano Coelho <luciano.coelho@nokia.com>
   7  *
   8  * This program is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU General Public License
  10  * version 2 as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  20  * 02110-1301 USA
  21  *
  22  */
  23
  24 #include <linux/kernel.h>
  25 #include <linux/module.h>
  26 #include <linux/etherdevice.h>
  27
  28 #include "wlcore.h"
  29 #include "debug.h"
  30 #include "io.h"
  31 #include "ps.h"
  32 #include "tx.h"
  33 #include "event.h"
  34 #include "hw_ops.h"
  35
  36 /*
  37  * TODO: this is here just for now, it must be removed when the data
  38  * operations are in place.
  39  */
  40 #include "../wl12xx/reg.h"
  41
  42 static int wl1271_set_default_wep_key(struct wl1271 *wl,
  43                                       struct wl12xx_vif *wlvif, u8 id)
  44 {
  45         int ret;
  46         bool is_ap = (wlvif->bss_type == BSS_TYPE_AP_BSS);
  47
  48         if (is_ap)
  49                 ret = wl12xx_cmd_set_default_wep_key(wl, id,
  50                                                      wlvif->ap.bcast_hlid);
  51         else
  52                 ret = wl12xx_cmd_set_default_wep_key(wl, id, wlvif->sta.hlid);
  53
  54         if (ret < 0)
  55                 return ret;
  56
  57         wl1271_debug(DEBUG_CRYPT, "default wep key idx: %d", (int)id);
  58         return 0;
  59 }
  60
  61 static int wl1271_alloc_tx_id(struct wl1271 *wl, struct sk_buff *skb)
  62 {
  63         int id;
  64
  65         id = find_first_zero_bit(wl->tx_frames_map, wl->num_tx_desc);
  66         if (id >= wl->num_tx_desc)
  67                 return -EBUSY;
  68
  69         __set_bit(id, wl->tx_frames_map);
  70         wl->tx_frames[id] = skb;
  71         wl->tx_frames_cnt++;
  72         return id;
  73 }
  74
  75 void wl1271_free_tx_id(struct wl1271 *wl, int id)
  76 {
  77         if (__test_and_clear_bit(id, wl->tx_frames_map)) {
  78                 if (unlikely(wl->tx_frames_cnt == wl->num_tx_desc))
  79                         clear_bit(WL1271_FLAG_FW_TX_BUSY, &wl->flags);
  80
  81                 wl->tx_frames[id] = NULL;
  82                 wl->tx_frames_cnt--;
  83         }
  84 }
  85 EXPORT_SYMBOL(wl1271_free_tx_id);
  86
  87 static void wl1271_tx_ap_update_inconnection_sta(struct wl1271 *wl,
  88                                                  struct sk_buff *skb)
  89 {
  90         struct ieee80211_hdr *hdr;
  91
  92         /*
  93          * add the station to the known list before transmitting the
  94          * authentication response. this way it won't get de-authed by FW
  95          * when transmitting too soon.
  96          */
  97         hdr = (struct ieee80211_hdr *)(skb->data +
  98                                        sizeof(struct wl1271_tx_hw_descr));
  99         if (ieee80211_is_auth(hdr->frame_control))
 100                 wl1271_acx_set_inconnection_sta(wl, hdr->addr1);
 101 }
 102
 103 static void wl1271_tx_regulate_link(struct wl1271 *wl,
 104                                     struct wl12xx_vif *wlvif,
 105                                     u8 hlid)
 106 {
 107         bool fw_ps;
 108         u8 tx_pkts;
 109
 110         if (WARN_ON(!test_bit(hlid, wlvif->links_map)))
 111                 return;
 112
 113         fw_ps = test_bit(hlid, (unsigned long *)&wl->ap_fw_ps_map);
 114         tx_pkts = wl->links[hlid].allocated_pkts;
 115
 116         /*
 117          * if in FW PS and there is enough data in FW we can put the link
 118          * into high-level PS and clean out its TX queues.
 119          * Make an exception if this is the only connected link. In this
 120          * case FW-memory congestion is less of a problem.
 121          * Note that a single connected STA means 3 active links, since we must
 122          * account for the global and broadcast AP links. The "fw_ps" check
 123          * assures us the third link is a STA connected to the AP. Otherwise
 124          * the FW would not set the PSM bit.
 125          */
 126         if (wl->active_link_count > 3 && fw_ps &&
 127             tx_pkts >= WL1271_PS_STA_MAX_PACKETS)
 128                 wl12xx_ps_link_start(wl, wlvif, hlid, true);
 129 }
 130
 131 bool wl12xx_is_dummy_packet(struct wl1271 *wl, struct sk_buff *skb)
 132 {
 133         return wl->dummy_packet == skb;
 134 }
 135 EXPORT_SYMBOL(wl12xx_is_dummy_packet);
 136
 137 static u8 wl12xx_tx_get_hlid_ap(struct wl1271 *wl, struct wl12xx_vif *wlvif,
 138                                 struct sk_buff *skb, struct ieee80211_sta *sta)
 139 {
 140         if (sta) {
 141                 struct wl1271_station *wl_sta;
 142
 143                 wl_sta = (struct wl1271_station *)sta->drv_priv;
 144                 return wl_sta->hlid;
 145         } else {
 146                 struct ieee80211_hdr *hdr;
 147
 148                 if (!test_bit(WLVIF_FLAG_AP_STARTED, &wlvif->flags))
 149                         return wl->system_hlid;
 150
 151                 hdr = (struct ieee80211_hdr *)skb->data;
 152                 if (is_multicast_ether_addr(ieee80211_get_DA(hdr)))
 153                         return wlvif->ap.bcast_hlid;
 154                 else
 155                         return wlvif->ap.global_hlid;
 156         }
 157 }
 158
 159 u8 wl12xx_tx_get_hlid(struct wl1271 *wl, struct wl12xx_vif *wlvif,
 160                       struct sk_buff *skb, struct ieee80211_sta *sta)
 161 {
 162         struct ieee80211_tx_info *control;
 163
 164         if (wlvif->bss_type == BSS_TYPE_AP_BSS)
 165                 return wl12xx_tx_get_hlid_ap(wl, wlvif, skb, sta);
 166
 167         control = IEEE80211_SKB_CB(skb);
 168         if (control->flags & IEEE80211_TX_CTL_TX_OFFCHAN) {
 169                 wl1271_debug(DEBUG_TX, "tx offchannel");
 170                 return wlvif->dev_hlid;
 171         }
 172
 173         return wlvif->sta.hlid;
 174 }
 175
 176 unsigned int wlcore_calc_packet_alignment(struct wl1271 *wl,
 177                                           unsigned int packet_length)
 178 {
 179         if ((wl->quirks & WLCORE_QUIRK_TX_PAD_LAST_FRAME) ||
 180             !(wl->quirks & WLCORE_QUIRK_TX_BLOCKSIZE_ALIGN))
 181                 return ALIGN(packet_length, WL1271_TX_ALIGN_TO);
 182         else
 183                 return ALIGN(packet_length, WL12XX_BUS_BLOCK_SIZE);
 184 }
 185 EXPORT_SYMBOL(wlcore_calc_packet_alignment);
 186
 187 static int wl1271_tx_allocate(struct wl1271 *wl, struct wl12xx_vif *wlvif,
 188                               struct sk_buff *skb, u32 extra, u32 buf_offset,
 189                               u8 hlid, bool is_gem)
 190 {
 191         struct wl1271_tx_hw_descr *desc;
 192         u32 total_len = skb->len + sizeof(struct wl1271_tx_hw_descr) + extra;
 193         u32 total_blocks;
 194         int id, ret = -EBUSY, ac;
 195         u32 spare_blocks;
 196
 197         if (buf_offset + total_len > wl->aggr_buf_size)
 198                 return -EAGAIN;
 199
 200         spare_blocks = wlcore_hw_get_spare_blocks(wl, is_gem);
 201
 202         /* allocate free identifier for the packet */
 203         id = wl1271_alloc_tx_id(wl, skb);
 204         if (id < 0)
 205                 return id;
 206
 207         total_blocks = wlcore_hw_calc_tx_blocks(wl, total_len, spare_blocks);
 208
 209         if (total_blocks <= wl->tx_blocks_available) {
 210                 desc = (struct wl1271_tx_hw_descr *)skb_push(
 211                         skb, total_len - skb->len);
 212
 213                 wlcore_hw_set_tx_desc_blocks(wl, desc, total_blocks,
 214                                              spare_blocks);
 215
 216                 desc->id = id;
 217
 218                 wl->tx_blocks_available -= total_blocks;
 219                 wl->tx_allocated_blocks += total_blocks;
 220
 221                 /* If the FW was empty before, arm the Tx watchdog */
 222                 if (wl->tx_allocated_blocks == total_blocks)
 223                         wl12xx_rearm_tx_watchdog_locked(wl);
 224
 225                 ac = wl1271_tx_get_queue(skb_get_queue_mapping(skb));
 226                 wl->tx_allocated_pkts[ac]++;
 227
 228                 if (test_bit(hlid, wl->links_map))
 229                         wl->links[hlid].allocated_pkts++;
 230
 231                 ret = 0;
 232
 233                 wl1271_debug(DEBUG_TX,
 234                              "tx_allocate: size: %d, blocks: %d, id: %d",
 235                              total_len, total_blocks, id);
 236         } else {
 237                 wl1271_free_tx_id(wl, id);
 238         }
 239
 240         return ret;
 241 }
 242
 243 static void wl1271_tx_fill_hdr(struct wl1271 *wl, struct wl12xx_vif *wlvif,
 244                                struct sk_buff *skb, u32 extra,
 245                                struct ieee80211_tx_info *control, u8 hlid)
 246 {
 247         struct timespec ts;
 248         struct wl1271_tx_hw_descr *desc;
 249         int ac, rate_idx;
 250         s64 hosttime;
 251         u16 tx_attr = 0;
 252         __le16 frame_control;
 253         struct ieee80211_hdr *hdr;
 254         u8 *frame_start;
 255         bool is_dummy;
 256
 257         desc = (struct wl1271_tx_hw_descr *) skb->data;
 258         frame_start = (u8 *)(desc + 1);
 259         hdr = (struct ieee80211_hdr *)(frame_start + extra);
 260         frame_control = hdr->frame_control;
 261
 262         /* relocate space for security header */
 263         if (extra) {
 264                 int hdrlen = ieee80211_hdrlen(frame_control);
 265                 memmove(frame_start, hdr, hdrlen);
 266                 skb_set_network_header(skb, skb_network_offset(skb) + extra);
 267         }
 268
 269         /* configure packet life time */
 270         getnstimeofday(&ts);
 271         hosttime = (timespec_to_ns(&ts) >> 10);
 272         desc->start_time = cpu_to_le32(hosttime - wl->time_offset);
 273
 274         is_dummy = wl12xx_is_dummy_packet(wl, skb);
 275         if (is_dummy || !wlvif || wlvif->bss_type != BSS_TYPE_AP_BSS)
 276                 desc->life_time = cpu_to_le16(TX_HW_MGMT_PKT_LIFETIME_TU);
 277         else
 278                 desc->life_time = cpu_to_le16(TX_HW_AP_MODE_PKT_LIFETIME_TU);
 279
 280         /* queue */
 281         ac = wl1271_tx_get_queue(skb_get_queue_mapping(skb));
 282         desc->tid = skb->priority;
 283
 284         if (is_dummy) {
 285                 /*
 286                  * FW expects the dummy packet to have an invalid session id -
 287                  * any session id that is different than the one set in the join
 288                  */
 289                 tx_attr = (SESSION_COUNTER_INVALID <<
 290                            TX_HW_ATTR_OFST_SESSION_COUNTER) &
 291                            TX_HW_ATTR_SESSION_COUNTER;
 292
 293                 tx_attr |= TX_HW_ATTR_TX_DUMMY_REQ;
 294         } else if (wlvif) {
 295                 u8 session_id = wl->session_ids[hlid];
 296
 297                 if ((wl->quirks & WLCORE_QUIRK_AP_ZERO_SESSION_ID) &&
 298                     (wlvif->bss_type == BSS_TYPE_AP_BSS))
 299                         session_id = 0;
 300
 301                 /* configure the tx attributes */
 302                 tx_attr = session_id << TX_HW_ATTR_OFST_SESSION_COUNTER;
 303         }
 304
 305         desc->hlid = hlid;
 306         if (is_dummy || !wlvif)
 307                 rate_idx = 0;
 308         else if (wlvif->bss_type != BSS_TYPE_AP_BSS) {
 309                 /*
 310                  * if the packets are data packets
 311                  * send them with AP rate policies (EAPOLs are an exception),
 312                  * otherwise use default basic rates
 313                  */
 314                 if (skb->protocol == cpu_to_be16(ETH_P_PAE))
 315                         rate_idx = wlvif->sta.basic_rate_idx;
 316                 else if (control->flags & IEEE80211_TX_CTL_NO_CCK_RATE)
 317                         rate_idx = wlvif->sta.p2p_rate_idx;
 318                 else if (ieee80211_is_data(frame_control))
 319                         rate_idx = wlvif->sta.ap_rate_idx;
 320                 else
 321                         rate_idx = wlvif->sta.basic_rate_idx;
 322         } else {
 323                 if (hlid == wlvif->ap.global_hlid)
 324                         rate_idx = wlvif->ap.mgmt_rate_idx;
 325                 else if (hlid == wlvif->ap.bcast_hlid ||
 326                          skb->protocol == cpu_to_be16(ETH_P_PAE) ||
 327                          !ieee80211_is_data(frame_control))
 328                         /*
 329                          * send non-data, bcast and EAPOLs using the
 330                          * min basic rate
 331                          */
 332                         rate_idx = wlvif->ap.bcast_rate_idx;
 333                 else
 334                         rate_idx = wlvif->ap.ucast_rate_idx[ac];
 335         }
 336
 337         tx_attr |= rate_idx << TX_HW_ATTR_OFST_RATE_POLICY;
 338
 339         /* for WEP shared auth - no fw encryption is needed */
 340         if (ieee80211_is_auth(frame_control) &&
 341             ieee80211_has_protected(frame_control))
 342                 tx_attr |= TX_HW_ATTR_HOST_ENCRYPT;
 343
 344         desc->tx_attr = cpu_to_le16(tx_attr);
 345
 346         wlcore_hw_set_tx_desc_csum(wl, desc, skb);
 347         wlcore_hw_set_tx_desc_data_len(wl, desc, skb);
 348 }
 349
 350 /* caller must hold wl->mutex */
 351 static int wl1271_prepare_tx_frame(struct wl1271 *wl, struct wl12xx_vif *wlvif,
 352                                    struct sk_buff *skb, u32 buf_offset, u8 hlid)
 353 {
 354         struct ieee80211_tx_info *info;
 355         u32 extra = 0;
 356         int ret = 0;
 357         u32 total_len;
 358         bool is_dummy;
 359         bool is_gem = false;
 360
 361         if (!skb) {
 362                 wl1271_error("discarding null skb");
 363                 return -EINVAL;
 364         }
 365
 366         if (hlid == WL12XX_INVALID_LINK_ID) {
 367                 wl1271_error("invalid hlid. dropping skb 0x%p", skb);
 368                 return -EINVAL;
 369         }
 370
 371         info = IEEE80211_SKB_CB(skb);
 372
 373         is_dummy = wl12xx_is_dummy_packet(wl, skb);
 374
 375         if ((wl->quirks & WLCORE_QUIRK_TKIP_HEADER_SPACE) &&
 376             info->control.hw_key &&
 377             info->control.hw_key->cipher == WLAN_CIPHER_SUITE_TKIP)
 378                 extra = WL1271_EXTRA_SPACE_TKIP;
 379
 380         if (info->control.hw_key) {
 381                 bool is_wep;
 382                 u8 idx = info->control.hw_key->hw_key_idx;
 383                 u32 cipher = info->control.hw_key->cipher;
 384
 385                 is_wep = (cipher == WLAN_CIPHER_SUITE_WEP40) ||
 386                          (cipher == WLAN_CIPHER_SUITE_WEP104);
 387
 388                 if (unlikely(is_wep && wlvif->default_key != idx)) {
 389                         ret = wl1271_set_default_wep_key(wl, wlvif, idx);
 390                         if (ret < 0)
 391                                 return ret;
 392                         wlvif->default_key = idx;
 393                 }
 394
 395                 is_gem = (cipher == WL1271_CIPHER_SUITE_GEM);
 396         }
 397
 398         ret = wl1271_tx_allocate(wl, wlvif, skb, extra, buf_offset, hlid,
 399                                  is_gem);
 400         if (ret < 0)
 401                 return ret;
 402
 403         wl1271_tx_fill_hdr(wl, wlvif, skb, extra, info, hlid);
 404
 405         if (!is_dummy && wlvif && wlvif->bss_type == BSS_TYPE_AP_BSS) {
 406                 wl1271_tx_ap_update_inconnection_sta(wl, skb);
 407                 wl1271_tx_regulate_link(wl, wlvif, hlid);
 408         }
 409
 410         /*
 411          * The length of each packet is stored in terms of
 412          * words. Thus, we must pad the skb data to make sure its
 413          * length is aligned.  The number of padding bytes is computed
 414          * and set in wl1271_tx_fill_hdr.
 415          * In special cases, we want to align to a specific block size
 416          * (eg. for wl128x with SDIO we align to 256).
 417          */
 418         total_len = wlcore_calc_packet_alignment(wl, skb->len);
 419
 420         memcpy(wl->aggr_buf + buf_offset, skb->data, skb->len);
 421         memset(wl->aggr_buf + buf_offset + skb->len, 0, total_len - skb->len);
 422
 423         /* Revert side effects in the dummy packet skb, so it can be reused */
 424         if (is_dummy)
 425                 skb_pull(skb, sizeof(struct wl1271_tx_hw_descr));
 426
 427         return total_len;
 428 }
 429
 430 u32 wl1271_tx_enabled_rates_get(struct wl1271 *wl, u32 rate_set,
 431                                 enum ieee80211_band rate_band)
 432 {
 433         struct ieee80211_supported_band *band;
 434         u32 enabled_rates = 0;
 435         int bit;
 436
 437         band = wl->hw->wiphy->bands[rate_band];
 438         for (bit = 0; bit < band->n_bitrates; bit++) {
 439                 if (rate_set & 0x1)
 440                         enabled_rates |= band->bitrates[bit].hw_value;
 441                 rate_set >>= 1;
 442         }
 443
 444         /* MCS rates indication are on bits 16 - 31 */
 445         rate_set >>= HW_HT_RATES_OFFSET - band->n_bitrates;
 446
 447         for (bit = 0; bit < 16; bit++) {
 448                 if (rate_set & 0x1)
 449                         enabled_rates |= (CONF_HW_BIT_RATE_MCS_0 << bit);
 450                 rate_set >>= 1;
 451         }
 452
 453         return enabled_rates;
 454 }
 455
 456 void wl1271_handle_tx_low_watermark(struct wl1271 *wl)
 457 {
 458         int i;
 459         struct wl12xx_vif *wlvif;
 460
 461         wl12xx_for_each_wlvif(wl, wlvif) {
 462                 for (i = 0; i < NUM_TX_QUEUES; i++) {
 463                         if (wlcore_is_queue_stopped_by_reason(wl, wlvif, i,
 464                                         WLCORE_QUEUE_STOP_REASON_WATERMARK) &&
 465                             wlvif->tx_queue_count[i] <=
 466                                         WL1271_TX_QUEUE_LOW_WATERMARK)
 467                                 /* firmware buffer has space, restart queues */
 468                                 wlcore_wake_queue(wl, wlvif, i,
 469                                         WLCORE_QUEUE_STOP_REASON_WATERMARK);
 470                 }
 471         }
 472 }
 473
 474 static int wlcore_select_ac(struct wl1271 *wl)
 475 {
 476         int i, q = -1, ac;
 477         u32 min_pkts = 0xffffffff;
 478
 479         /*
 480          * Find a non-empty ac where:
 481          * 1. There are packets to transmit
 482          * 2. The FW has the least allocated blocks
 483          *
 484          * We prioritize the ACs according to VO>VI>BE>BK
 485          */
 486         for (i = 0; i < NUM_TX_QUEUES; i++) {
 487                 ac = wl1271_tx_get_queue(i);
 488                 if (wl->tx_queue_count[ac] &&
 489                     wl->tx_allocated_pkts[ac] < min_pkts) {
 490                         q = ac;
 491                         min_pkts = wl->tx_allocated_pkts[q];
 492                 }
 493         }
 494
 495         return q;
 496 }
 497
 498 static struct sk_buff *wlcore_lnk_dequeue(struct wl1271 *wl,
 499                                           struct wl1271_link *lnk, u8 q)
 500 {
 501         struct sk_buff *skb;
 502         unsigned long flags;
 503
 504         skb = skb_dequeue(&lnk->tx_queue[q]);
 505         if (skb) {
 506                 spin_lock_irqsave(&wl->wl_lock, flags);
 507                 WARN_ON_ONCE(wl->tx_queue_count[q] <= 0);
 508                 wl->tx_queue_count[q]--;
 509                 if (lnk->wlvif) {
 510                         WARN_ON_ONCE(lnk->wlvif->tx_queue_count[q] <= 0);
 511                         lnk->wlvif->tx_queue_count[q]--;
 512                 }
 513                 spin_unlock_irqrestore(&wl->wl_lock, flags);
 514         }
 515
 516         return skb;
 517 }
 518
 519 static struct sk_buff *wlcore_lnk_dequeue_high_prio(struct wl1271 *wl,
 520                                                     u8 hlid, u8 ac,
 521                                                     u8 *low_prio_hlid)
 522 {
 523         struct wl1271_link *lnk = &wl->links[hlid];
 524
 525         if (!wlcore_hw_lnk_high_prio(wl, hlid, lnk)) {
 526                 if (*low_prio_hlid == WL12XX_INVALID_LINK_ID &&
 527                     !skb_queue_empty(&lnk->tx_queue[ac]) &&
 528                     wlcore_hw_lnk_low_prio(wl, hlid, lnk))
 529                         /* we found the first non-empty low priority queue */
 530                         *low_prio_hlid = hlid;
 531
 532                 return NULL;
 533         }
 534
 535         return wlcore_lnk_dequeue(wl, lnk, ac);
 536 }
 537
 538 static struct sk_buff *wlcore_vif_dequeue_high_prio(struct wl1271 *wl,
 539                                                     struct wl12xx_vif *wlvif,
 540                                                     u8 ac, u8 *hlid,
 541                                                     u8 *low_prio_hlid)
 542 {
 543         struct sk_buff *skb = NULL;
 544         int i, h, start_hlid;
 545
 546         /* start from the link after the last one */
 547         start_hlid = (wlvif->last_tx_hlid + 1) % WL12XX_MAX_LINKS;
 548
 549         /* dequeue according to AC, round robin on each link */
 550         for (i = 0; i < WL12XX_MAX_LINKS; i++) {
 551                 h = (start_hlid + i) % WL12XX_MAX_LINKS;
 552
 553                 /* only consider connected stations */
 554                 if (!test_bit(h, wlvif->links_map))
 555                         continue;
 556
 557                 skb = wlcore_lnk_dequeue_high_prio(wl, h, ac,
 558                                                    low_prio_hlid);
 559                 if (!skb)
 560                         continue;
 561
 562                 wlvif->last_tx_hlid = h;
 563                 break;
 564         }
 565
 566         if (!skb)
 567                 wlvif->last_tx_hlid = 0;
 568
 569         *hlid = wlvif->last_tx_hlid;
 570         return skb;
 571 }
 572
 573 static struct sk_buff *wl1271_skb_dequeue(struct wl1271 *wl, u8 *hlid)
 574 {
 575         unsigned long flags;
 576         struct wl12xx_vif *wlvif = wl->last_wlvif;
 577         struct sk_buff *skb = NULL;
 578         int ac;
 579         u8 low_prio_hlid = WL12XX_INVALID_LINK_ID;
 580
 581         ac = wlcore_select_ac(wl);
 582         if (ac < 0)
 583                 goto out;
 584
 585         /* continue from last wlvif (round robin) */
 586         if (wlvif) {
 587                 wl12xx_for_each_wlvif_continue(wl, wlvif) {
 588                         if (!wlvif->tx_queue_count[ac])
 589                                 continue;
 590
 591                         skb = wlcore_vif_dequeue_high_prio(wl, wlvif, ac, hlid,
 592                                                            &low_prio_hlid);
 593                         if (!skb)
 594                                 continue;
 595
 596                         wl->last_wlvif = wlvif;
 597                         break;
 598                 }
 599         }
 600
 601         /* dequeue from the system HLID before the restarting wlvif list */
 602         if (!skb) {
 603                 skb = wlcore_lnk_dequeue_high_prio(wl, wl->system_hlid,
 604                                                    ac, &low_prio_hlid);
 605                 if (skb) {
 606                         *hlid = wl->system_hlid;
 607                         wl->last_wlvif = NULL;
 608                 }
 609         }
 610
 611         /* Do a new pass over the wlvif list. But no need to continue
 612          * after last_wlvif. The previous pass should have found it. */
 613         if (!skb) {
 614                 wl12xx_for_each_wlvif(wl, wlvif) {
 615                         if (!wlvif->tx_queue_count[ac])
 616                                 goto next;
 617
 618                         skb = wlcore_vif_dequeue_high_prio(wl, wlvif, ac, hlid,
 619                                                            &low_prio_hlid);
 620                         if (skb) {
 621                                 wl->last_wlvif = wlvif;
 622                                 break;
 623                         }
 624
 625 next:
 626                         if (wlvif == wl->last_wlvif)
 627                                 break;
 628                 }
 629         }
 630
 631         /* no high priority skbs found - but maybe a low priority one? */
 632         if (!skb && low_prio_hlid != WL12XX_INVALID_LINK_ID) {
 633                 struct wl1271_link *lnk = &wl->links[low_prio_hlid];
 634                 skb = wlcore_lnk_dequeue(wl, lnk, ac);
 635
 636                 WARN_ON(!skb); /* we checked this before */
 637                 *hlid = low_prio_hlid;
 638
 639                 /* ensure proper round robin in the vif/link levels */
 640                 wl->last_wlvif = lnk->wlvif;
 641                 if (lnk->wlvif)
 642                         lnk->wlvif->last_tx_hlid = low_prio_hlid;
 643
 644         }
 645
 646         if (!skb &&
 647             test_and_clear_bit(WL1271_FLAG_DUMMY_PACKET_PENDING, &wl->flags)) {
 648                 int q;
 649
 650                 skb = wl->dummy_packet;
 651                 *hlid = wl->system_hlid;
 652                 q = wl1271_tx_get_queue(skb_get_queue_mapping(skb));
 653                 spin_lock_irqsave(&wl->wl_lock, flags);
 654                 WARN_ON_ONCE(wl->tx_queue_count[q] <= 0);
 655                 wl->tx_queue_count[q]--;
 656                 spin_unlock_irqrestore(&wl->wl_lock, flags);
 657         }
 658
 659 out:
 660         return skb;
 661 }
 662
 663 static void wl1271_skb_queue_head(struct wl1271 *wl, struct wl12xx_vif *wlvif,
 664                                   struct sk_buff *skb, u8 hlid)
 665 {
 666         unsigned long flags;
 667         int q = wl1271_tx_get_queue(skb_get_queue_mapping(skb));
 668
 669         if (wl12xx_is_dummy_packet(wl, skb)) {
 670                 set_bit(WL1271_FLAG_DUMMY_PACKET_PENDING, &wl->flags);
 671         } else {
 672                 skb_queue_head(&wl->links[hlid].tx_queue[q], skb);
 673
 674                 /* make sure we dequeue the same packet next time */
 675                 wlvif->last_tx_hlid = (hlid + WL12XX_MAX_LINKS - 1) %
 676                                       WL12XX_MAX_LINKS;
 677         }
 678
 679         spin_lock_irqsave(&wl->wl_lock, flags);
 680         wl->tx_queue_count[q]++;
 681         if (wlvif)
 682                 wlvif->tx_queue_count[q]++;
 683         spin_unlock_irqrestore(&wl->wl_lock, flags);
 684 }
 685
 686 static bool wl1271_tx_is_data_present(struct sk_buff *skb)
 687 {
 688         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)(skb->data);
 689
 690         return ieee80211_is_data_present(hdr->frame_control);
 691 }
 692
 693 void wl12xx_rearm_rx_streaming(struct wl1271 *wl, unsigned long *active_hlids)
 694 {
 695         struct wl12xx_vif *wlvif;
 696         u32 timeout;
 697         u8 hlid;
 698
 699         if (!wl->conf.rx_streaming.interval)
 700                 return;
 701
 702         if (!wl->conf.rx_streaming.always &&
 703             !test_bit(WL1271_FLAG_SOFT_GEMINI, &wl->flags))
 704                 return;
 705
 706         timeout = wl->conf.rx_streaming.duration;
 707         wl12xx_for_each_wlvif_sta(wl, wlvif) {
 708                 bool found = false;
 709                 for_each_set_bit(hlid, active_hlids, WL12XX_MAX_LINKS) {
 710                         if (test_bit(hlid, wlvif->links_map)) {
 711                                 found  = true;
 712                                 break;
 713                         }
 714                 }
 715
 716                 if (!found)
 717                         continue;
 718
 719                 /* enable rx streaming */
 720                 if (!test_bit(WLVIF_FLAG_RX_STREAMING_STARTED, &wlvif->flags))
 721                         ieee80211_queue_work(wl->hw,
 722                                              &wlvif->rx_streaming_enable_work);
 723
 724                 mod_timer(&wlvif->rx_streaming_timer,
 725                           jiffies + msecs_to_jiffies(timeout));
 726         }
 727 }
 728
 729 /*
 730  * Returns failure values only in case of failed bus ops within this function.
 731  * wl1271_prepare_tx_frame retvals won't be returned in order to avoid
 732  * triggering recovery by higher layers when not necessary.
 733  * In case a FW command fails within wl1271_prepare_tx_frame fails a recovery
 734  * will be queued in wl1271_cmd_send. -EAGAIN/-EBUSY from prepare_tx_frame
 735  * can occur and are legitimate so don't propagate. -EINVAL will emit a WARNING
 736  * within prepare_tx_frame code but there's nothing we should do about those
 737  * as well.
 738  */
 739 int wlcore_tx_work_locked(struct wl1271 *wl)
 740 {
 741         struct wl12xx_vif *wlvif;
 742         struct sk_buff *skb;
 743         struct wl1271_tx_hw_descr *desc;
 744         u32 buf_offset = 0, last_len = 0;
 745         bool sent_packets = false;
 746         unsigned long active_hlids[BITS_TO_LONGS(WL12XX_MAX_LINKS)] = {0};
 747         int ret = 0;
 748         int bus_ret = 0;
 749         u8 hlid;
 750
 751         if (unlikely(wl->state != WLCORE_STATE_ON))
 752                 return 0;
 753
 754         while ((skb = wl1271_skb_dequeue(wl, &hlid))) {
 755                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
 756                 bool has_data = false;
 757
 758                 wlvif = NULL;
 759                 if (!wl12xx_is_dummy_packet(wl, skb))
 760                         wlvif = wl12xx_vif_to_data(info->control.vif);
 761                 else
 762                         hlid = wl->system_hlid;
 763
 764                 has_data = wlvif && wl1271_tx_is_data_present(skb);
 765                 ret = wl1271_prepare_tx_frame(wl, wlvif, skb, buf_offset,
 766                                               hlid);
 767                 if (ret == -EAGAIN) {
 768                         /*
 769                          * Aggregation buffer is full.
 770                          * Flush buffer and try again.
 771                          */
 772                         wl1271_skb_queue_head(wl, wlvif, skb, hlid);
 773
 774                         buf_offset = wlcore_hw_pre_pkt_send(wl, buf_offset,
 775                                                             last_len);
 776                         bus_ret = wlcore_write_data(wl, REG_SLV_MEM_DATA,
 777                                              wl->aggr_buf, buf_offset, true);
 778                         if (bus_ret < 0)
 779                                 goto out;
 780
 781                         sent_packets = true;
 782                         buf_offset = 0;
 783                         continue;
 784                 } else if (ret == -EBUSY) {
 785                         /*
 786                          * Firmware buffer is full.
 787                          * Queue back last skb, and stop aggregating.
 788                          */
 789                         wl1271_skb_queue_head(wl, wlvif, skb, hlid);
 790                         /* No work left, avoid scheduling redundant tx work */
 791                         set_bit(WL1271_FLAG_FW_TX_BUSY, &wl->flags);
 792                         goto out_ack;
 793                 } else if (ret < 0) {
 794                         if (wl12xx_is_dummy_packet(wl, skb))
 795                                 /*
 796                                  * fw still expects dummy packet,
 797                                  * so re-enqueue it
 798                                  */
 799                                 wl1271_skb_queue_head(wl, wlvif, skb, hlid);
 800                         else
 801                                 ieee80211_free_txskb(wl->hw, skb);
 802                         goto out_ack;
 803                 }
 804                 last_len = ret;
 805                 buf_offset += last_len;
 806                 wl->tx_packets_count++;
 807                 if (has_data) {
 808                         desc = (struct wl1271_tx_hw_descr *) skb->data;
 809                         __set_bit(desc->hlid, active_hlids);
 810                 }
 811         }
 812
 813 out_ack:
 814         if (buf_offset) {
 815                 buf_offset = wlcore_hw_pre_pkt_send(wl, buf_offset, last_len);
 816                 bus_ret = wlcore_write_data(wl, REG_SLV_MEM_DATA, wl->aggr_buf,
 817                                              buf_offset, true);
 818                 if (bus_ret < 0)
 819                         goto out;
 820
 821                 sent_packets = true;
 822         }
 823         if (sent_packets) {
 824                 /*
 825                  * Interrupt the firmware with the new packets. This is only
 826                  * required for older hardware revisions
 827                  */
 828                 if (wl->quirks & WLCORE_QUIRK_END_OF_TRANSACTION) {
 829                         bus_ret = wlcore_write32(wl, WL12XX_HOST_WR_ACCESS,
 830                                              wl->tx_packets_count);
 831                         if (bus_ret < 0)
 832                                 goto out;
 833                 }
 834
 835                 wl1271_handle_tx_low_watermark(wl);
 836         }
 837         wl12xx_rearm_rx_streaming(wl, active_hlids);
 838
 839 out:
 840         return bus_ret;
 841 }
 842
 843 void wl1271_tx_work(struct work_struct *work)
 844 {
 845         struct wl1271 *wl = container_of(work, struct wl1271, tx_work);
 846         int ret;
 847
 848         mutex_lock(&wl->mutex);
 849         ret = wl1271_ps_elp_wakeup(wl);
 850         if (ret < 0)
 851                 goto out;
 852
 853         ret = wlcore_tx_work_locked(wl);
 854         if (ret < 0) {
 855                 wl12xx_queue_recovery_work(wl);
 856                 goto out;
 857         }
 858
 859         wl1271_ps_elp_sleep(wl);
 860 out:
 861         mutex_unlock(&wl->mutex);
 862 }
 863
 864 static u8 wl1271_tx_get_rate_flags(u8 rate_class_index)
 865 {
 866         u8 flags = 0;
 867
 868         /*
 869          * TODO: use wl12xx constants when this code is moved to wl12xx, as
 870          * only it uses Tx-completion.
 871          */
 872         if (rate_class_index <= 8)
 873                 flags |= IEEE80211_TX_RC_MCS;
 874
 875         /*
 876          * TODO: use wl12xx constants when this code is moved to wl12xx, as
 877          * only it uses Tx-completion.
 878          */
 879         if (rate_class_index == 0)
 880                 flags |= IEEE80211_TX_RC_SHORT_GI;
 881
 882         return flags;
 883 }
 884
 885 static void wl1271_tx_complete_packet(struct wl1271 *wl,
 886                                       struct wl1271_tx_hw_res_descr *result)
 887 {
 888         struct ieee80211_tx_info *info;
 889         struct ieee80211_vif *vif;
 890         struct wl12xx_vif *wlvif;
 891         struct sk_buff *skb;
 892         int id = result->id;
 893         int rate = -1;
 894         u8 rate_flags = 0;
 895         u8 retries = 0;
 896
 897         /* check for id legality */
 898         if (unlikely(id >= wl->num_tx_desc || wl->tx_frames[id] == NULL)) {
 899                 wl1271_warning("TX result illegal id: %d", id);
 900                 return;
 901         }
 902
 903         skb = wl->tx_frames[id];
 904         info = IEEE80211_SKB_CB(skb);
 905
 906         if (wl12xx_is_dummy_packet(wl, skb)) {
 907                 wl1271_free_tx_id(wl, id);
 908                 return;
 909         }
 910
 911         /* info->control is valid as long as we don't update info->status */
 912         vif = info->control.vif;
 913         wlvif = wl12xx_vif_to_data(vif);
 914
 915         /* update the TX status info */
 916         if (result->status == TX_SUCCESS) {
 917                 if (!(info->flags & IEEE80211_TX_CTL_NO_ACK))
 918                         info->flags |= IEEE80211_TX_STAT_ACK;
 919                 rate = wlcore_rate_to_idx(wl, result->rate_class_index,
 920                                           wlvif->band);
 921                 rate_flags = wl1271_tx_get_rate_flags(result->rate_class_index);
 922                 retries = result->ack_failures;
 923         } else if (result->status == TX_RETRY_EXCEEDED) {
 924                 wl->stats.excessive_retries++;
 925                 retries = result->ack_failures;
 926         }
 927
 928         info->status.rates[0].idx = rate;
 929         info->status.rates[0].count = retries;
 930         info->status.rates[0].flags = rate_flags;
 931         info->status.ack_signal = -1;
 932
 933         wl->stats.retry_count += result->ack_failures;
 934
 935         /* remove private header from packet */
 936         skb_pull(skb, sizeof(struct wl1271_tx_hw_descr));
 937
 938         /* remove TKIP header space if present */
 939         if ((wl->quirks & WLCORE_QUIRK_TKIP_HEADER_SPACE) &&
 940             info->control.hw_key &&
 941             info->control.hw_key->cipher == WLAN_CIPHER_SUITE_TKIP) {
 942                 int hdrlen = ieee80211_get_hdrlen_from_skb(skb);
 943                 memmove(skb->data + WL1271_EXTRA_SPACE_TKIP, skb->data,
 944                         hdrlen);
 945                 skb_pull(skb, WL1271_EXTRA_SPACE_TKIP);
 946         }
 947
 948         wl1271_debug(DEBUG_TX, "tx status id %u skb 0x%p failures %u rate 0x%x"
 949                      " status 0x%x",
 950                      result->id, skb, result->ack_failures,
 951                      result->rate_class_index, result->status);
 952
 953         /* return the packet to the stack */
 954         skb_queue_tail(&wl->deferred_tx_queue, skb);
 955         queue_work(wl->freezable_wq, &wl->netstack_work);
 956         wl1271_free_tx_id(wl, result->id);
 957 }
 958
 959 /* Called upon reception of a TX complete interrupt */
 960 int wlcore_tx_complete(struct wl1271 *wl)
 961 {
 962         struct wl1271_acx_mem_map *memmap = wl->target_mem_map;
 963         u32 count, fw_counter;
 964         u32 i;
 965         int ret;
 966
 967         /* read the tx results from the chipset */
 968         ret = wlcore_read(wl, le32_to_cpu(memmap->tx_result),
 969                           wl->tx_res_if, sizeof(*wl->tx_res_if), false);
 970         if (ret < 0)
 971                 goto out;
 972
 973         fw_counter = le32_to_cpu(wl->tx_res_if->tx_result_fw_counter);
 974
 975         /* write host counter to chipset (to ack) */
 976         ret = wlcore_write32(wl, le32_to_cpu(memmap->tx_result) +
 977                              offsetof(struct wl1271_tx_hw_res_if,
 978                                       tx_result_host_counter), fw_counter);
 979         if (ret < 0)
 980                 goto out;
 981
 982         count = fw_counter - wl->tx_results_count;
 983         wl1271_debug(DEBUG_TX, "tx_complete received, packets: %d", count);
 984
 985         /* verify that the result buffer is not getting overrun */
 986         if (unlikely(count > TX_HW_RESULT_QUEUE_LEN))
 987                 wl1271_warning("TX result overflow from chipset: %d", count);
 988
 989         /* process the results */
 990         for (i = 0; i < count; i++) {
 991                 struct wl1271_tx_hw_res_descr *result;
 992                 u8 offset = wl->tx_results_count & TX_HW_RESULT_QUEUE_LEN_MASK;
 993
 994                 /* process the packet */
 995                 result =  &(wl->tx_res_if->tx_results_queue[offset]);
 996                 wl1271_tx_complete_packet(wl, result);
 997
 998                 wl->tx_results_count++;
 999         }
1000
1001 out:
1002         return ret;
1003 }
1004 EXPORT_SYMBOL(wlcore_tx_complete);
1005
1006 void wl1271_tx_reset_link_queues(struct wl1271 *wl, u8 hlid)
1007 {
1008         struct sk_buff *skb;
1009         int i;
1010         unsigned long flags;
1011         struct ieee80211_tx_info *info;
1012         int total[NUM_TX_QUEUES];
1013         struct wl1271_link *lnk = &wl->links[hlid];
1014
1015         for (i = 0; i < NUM_TX_QUEUES; i++) {
1016                 total[i] = 0;
1017                 while ((skb = skb_dequeue(&lnk->tx_queue[i]))) {
1018                         wl1271_debug(DEBUG_TX, "link freeing skb 0x%p", skb);
1019
1020                         if (!wl12xx_is_dummy_packet(wl, skb)) {
1021                                 info = IEEE80211_SKB_CB(skb);
1022                                 info->status.rates[0].idx = -1;
1023                                 info->status.rates[0].count = 0;
1024                                 ieee80211_tx_status_ni(wl->hw, skb);
1025                         }
1026
1027                         total[i]++;
1028                 }
1029         }
1030
1031         spin_lock_irqsave(&wl->wl_lock, flags);
1032         for (i = 0; i < NUM_TX_QUEUES; i++) {
1033                 wl->tx_queue_count[i] -= total[i];
1034                 if (lnk->wlvif)
1035                         lnk->wlvif->tx_queue_count[i] -= total[i];
1036         }
1037         spin_unlock_irqrestore(&wl->wl_lock, flags);
1038
1039         wl1271_handle_tx_low_watermark(wl);
1040 }
1041
1042 /* caller must hold wl->mutex and TX must be stopped */
1043 void wl12xx_tx_reset_wlvif(struct wl1271 *wl, struct wl12xx_vif *wlvif)
1044 {
1045         int i;
1046
1047         /* TX failure */
1048         for_each_set_bit(i, wlvif->links_map, WL12XX_MAX_LINKS) {
1049                 if (wlvif->bss_type == BSS_TYPE_AP_BSS) {
1050                         /* this calls wl12xx_free_link */
1051                         wl1271_free_sta(wl, wlvif, i);
1052                 } else {
1053                         u8 hlid = i;
1054                         wl12xx_free_link(wl, wlvif, &hlid);
1055                 }
1056         }
1057         wlvif->last_tx_hlid = 0;
1058
1059         for (i = 0; i < NUM_TX_QUEUES; i++)
1060                 wlvif->tx_queue_count[i] = 0;
1061 }
1062 /* caller must hold wl->mutex and TX must be stopped */
1063 void wl12xx_tx_reset(struct wl1271 *wl)
1064 {
1065         int i;
1066         struct sk_buff *skb;
1067         struct ieee80211_tx_info *info;
1068
1069         /* only reset the queues if something bad happened */
1070         if (wl1271_tx_total_queue_count(wl) != 0) {
1071                 for (i = 0; i < WL12XX_MAX_LINKS; i++)
1072                         wl1271_tx_reset_link_queues(wl, i);
1073
1074                 for (i = 0; i < NUM_TX_QUEUES; i++)
1075                         wl->tx_queue_count[i] = 0;
1076         }
1077
1078         /*
1079          * Make sure the driver is at a consistent state, in case this
1080          * function is called from a context other than interface removal.
1081          * This call will always wake the TX queues.
1082          */
1083         wl1271_handle_tx_low_watermark(wl);
1084
1085         for (i = 0; i < wl->num_tx_desc; i++) {
1086                 if (wl->tx_frames[i] == NULL)
1087                         continue;
1088
1089                 skb = wl->tx_frames[i];
1090                 wl1271_free_tx_id(wl, i);
1091                 wl1271_debug(DEBUG_TX, "freeing skb 0x%p", skb);
1092
1093                 if (!wl12xx_is_dummy_packet(wl, skb)) {
1094                         /*
1095                          * Remove private headers before passing the skb to
1096                          * mac80211
1097                          */
1098                         info = IEEE80211_SKB_CB(skb);
1099                         skb_pull(skb, sizeof(struct wl1271_tx_hw_descr));
1100                         if ((wl->quirks & WLCORE_QUIRK_TKIP_HEADER_SPACE) &&
1101                             info->control.hw_key &&
1102                             info->control.hw_key->cipher ==
1103                             WLAN_CIPHER_SUITE_TKIP) {
1104                                 int hdrlen = ieee80211_get_hdrlen_from_skb(skb);
1105                                 memmove(skb->data + WL1271_EXTRA_SPACE_TKIP,
1106                                         skb->data, hdrlen);
1107                                 skb_pull(skb, WL1271_EXTRA_SPACE_TKIP);
1108                         }
1109
1110                         info->status.rates[0].idx = -1;
1111                         info->status.rates[0].count = 0;
1112
1113                         ieee80211_tx_status_ni(wl->hw, skb);
1114                 }
1115         }
1116 }
1117
1118 #define WL1271_TX_FLUSH_TIMEOUT 500000
1119
1120 /* caller must *NOT* hold wl->mutex */
1121 void wl1271_tx_flush(struct wl1271 *wl)
1122 {
1123         unsigned long timeout, start_time;
1124         int i;
1125         start_time = jiffies;
1126         timeout = start_time + usecs_to_jiffies(WL1271_TX_FLUSH_TIMEOUT);
1127
1128         /* only one flush should be in progress, for consistent queue state */
1129         mutex_lock(&wl->flush_mutex);
1130
1131         mutex_lock(&wl->mutex);
1132         if (wl->tx_frames_cnt == 0 && wl1271_tx_total_queue_count(wl) == 0) {
1133                 mutex_unlock(&wl->mutex);
1134                 goto out;
1135         }
1136
1137         wlcore_stop_queues(wl, WLCORE_QUEUE_STOP_REASON_FLUSH);
1138
1139         while (!time_after(jiffies, timeout)) {
1140                 wl1271_debug(DEBUG_MAC80211, "flushing tx buffer: %d %d",
1141                              wl->tx_frames_cnt,
1142                              wl1271_tx_total_queue_count(wl));
1143
1144                 /* force Tx and give the driver some time to flush data */
1145                 mutex_unlock(&wl->mutex);
1146                 if (wl1271_tx_total_queue_count(wl))
1147                         wl1271_tx_work(&wl->tx_work);
1148                 msleep(20);
1149                 mutex_lock(&wl->mutex);
1150
1151                 if ((wl->tx_frames_cnt == 0) &&
1152                     (wl1271_tx_total_queue_count(wl) == 0)) {
1153                         wl1271_debug(DEBUG_MAC80211, "tx flush took %d ms",
1154                                      jiffies_to_msecs(jiffies - start_time));
1155                         goto out_wake;
1156                 }
1157         }
1158
1159         wl1271_warning("Unable to flush all TX buffers, "
1160                        "timed out (timeout %d ms",
1161                        WL1271_TX_FLUSH_TIMEOUT / 1000);
1162
1163         /* forcibly flush all Tx buffers on our queues */
1164         for (i = 0; i < WL12XX_MAX_LINKS; i++)
1165                 wl1271_tx_reset_link_queues(wl, i);
1166
1167 out_wake:
1168         wlcore_wake_queues(wl, WLCORE_QUEUE_STOP_REASON_FLUSH);
1169         mutex_unlock(&wl->mutex);
1170 out:
1171         mutex_unlock(&wl->flush_mutex);
1172 }
1173 EXPORT_SYMBOL_GPL(wl1271_tx_flush);
1174
1175 u32 wl1271_tx_min_rate_get(struct wl1271 *wl, u32 rate_set)
1176 {
1177         if (WARN_ON(!rate_set))
1178                 return 0;
1179
1180         return BIT(__ffs(rate_set));
1181 }
1182 EXPORT_SYMBOL_GPL(wl1271_tx_min_rate_get);
1183
1184 void wlcore_stop_queue_locked(struct wl1271 *wl, struct wl12xx_vif *wlvif,
1185                               u8 queue, enum wlcore_queue_stop_reason reason)
1186 {
1187         int hwq = wlcore_tx_get_mac80211_queue(wlvif, queue);
1188         bool stopped = !!wl->queue_stop_reasons[hwq];
1189
1190         /* queue should not be stopped for this reason */
1191         WARN_ON_ONCE(test_and_set_bit(reason, &wl->queue_stop_reasons[hwq]));
1192
1193         if (stopped)
1194                 return;
1195
1196         ieee80211_stop_queue(wl->hw, hwq);
1197 }
1198
1199 void wlcore_stop_queue(struct wl1271 *wl, struct wl12xx_vif *wlvif, u8 queue,
1200                        enum wlcore_queue_stop_reason reason)
1201 {
1202         unsigned long flags;
1203
1204         spin_lock_irqsave(&wl->wl_lock, flags);
1205         wlcore_stop_queue_locked(wl, wlvif, queue, reason);
1206         spin_unlock_irqrestore(&wl->wl_lock, flags);
1207 }
1208
1209 void wlcore_wake_queue(struct wl1271 *wl, struct wl12xx_vif *wlvif, u8 queue,
1210                        enum wlcore_queue_stop_reason reason)
1211 {
1212         unsigned long flags;
1213         int hwq = wlcore_tx_get_mac80211_queue(wlvif, queue);
1214
1215         spin_lock_irqsave(&wl->wl_lock, flags);
1216
1217         /* queue should not be clear for this reason */
1218         WARN_ON_ONCE(!test_and_clear_bit(reason, &wl->queue_stop_reasons[hwq]));
1219
1220         if (wl->queue_stop_reasons[hwq])
1221                 goto out;
1222
1223         ieee80211_wake_queue(wl->hw, hwq);
1224
1225 out:
1226         spin_unlock_irqrestore(&wl->wl_lock, flags);
1227 }
1228
1229 void wlcore_stop_queues(struct wl1271 *wl,
1230                         enum wlcore_queue_stop_reason reason)
1231 {
1232         int i;
1233         unsigned long flags;
1234
1235         spin_lock_irqsave(&wl->wl_lock, flags);
1236
1237         /* mark all possible queues as stopped */
1238         for (i = 0; i < WLCORE_NUM_MAC_ADDRESSES * NUM_TX_QUEUES; i++)
1239                 WARN_ON_ONCE(test_and_set_bit(reason,
1240                                               &wl->queue_stop_reasons[i]));
1241
1242         /* use the global version to make sure all vifs in mac80211 we don't
1243          * know are stopped.
1244          */
1245         ieee80211_stop_queues(wl->hw);
1246
1247         spin_unlock_irqrestore(&wl->wl_lock, flags);
1248 }
1249
1250 void wlcore_wake_queues(struct wl1271 *wl,
1251                         enum wlcore_queue_stop_reason reason)
1252 {
1253         int i;
1254         unsigned long flags;
1255
1256         spin_lock_irqsave(&wl->wl_lock, flags);
1257
1258         /* mark all possible queues as awake */
1259         for (i = 0; i < WLCORE_NUM_MAC_ADDRESSES * NUM_TX_QUEUES; i++)
1260                 WARN_ON_ONCE(!test_and_clear_bit(reason,
1261                                                  &wl->queue_stop_reasons[i]));
1262
1263         /* use the global version to make sure all vifs in mac80211 we don't
1264          * know are woken up.
1265          */
1266         ieee80211_wake_queues(wl->hw);
1267
1268         spin_unlock_irqrestore(&wl->wl_lock, flags);
1269 }
1270
1271 bool wlcore_is_queue_stopped_by_reason(struct wl1271 *wl,
1272                                        struct wl12xx_vif *wlvif, u8 queue,
1273                                        enum wlcore_queue_stop_reason reason)
1274 {
1275         unsigned long flags;
1276         bool stopped;
1277
1278         spin_lock_irqsave(&wl->wl_lock, flags);
1279         stopped = wlcore_is_queue_stopped_by_reason_locked(wl, wlvif, queue,
1280                                                            reason);
1281         spin_unlock_irqrestore(&wl->wl_lock, flags);
1282
1283         return stopped;
1284 }
1285
1286 bool wlcore_is_queue_stopped_by_reason_locked(struct wl1271 *wl,
1287                                        struct wl12xx_vif *wlvif, u8 queue,
1288                                        enum wlcore_queue_stop_reason reason)
1289 {
1290         int hwq = wlcore_tx_get_mac80211_queue(wlvif, queue);
1291
1292         WARN_ON_ONCE(!spin_is_locked(&wl->wl_lock));
1293         return test_bit(reason, &wl->queue_stop_reasons[hwq]);
1294 }
1295
1296 bool wlcore_is_queue_stopped_locked(struct wl1271 *wl, struct wl12xx_vif *wlvif,
1297                                     u8 queue)
1298 {
1299         int hwq = wlcore_tx_get_mac80211_queue(wlvif, queue);
1300
1301         WARN_ON_ONCE(!spin_is_locked(&wl->wl_lock));
1302         return !!wl->queue_stop_reasons[hwq];
1303 }