net/mac80211/tx.c

   1 /*
   2  * Copyright 2002-2005, Instant802 Networks, Inc.
   3  * Copyright 2005-2006, Devicescape Software, Inc.
   4  * Copyright 2006-2007  Jiri Benc <jbenc@suse.cz>
   5  * Copyright 2007       Johannes Berg <johannes@sipsolutions.net>
   6  * Copyright 2013-2014  Intel Mobile Communications GmbH
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 as
  10  * published by the Free Software Foundation.
  11  *
  12  *
  13  * Transmit and frame generation functions.
  14  */
  15
  16 #include <linux/kernel.h>
  17 #include <linux/slab.h>
  18 #include <linux/skbuff.h>
  19 #include <linux/if_vlan.h>
  20 #include <linux/etherdevice.h>
  21 #include <linux/bitmap.h>
  22 #include <linux/rcupdate.h>
  23 #include <linux/export.h>
  24 #include <net/net_namespace.h>
  25 #include <net/ieee80211_radiotap.h>
  26 #include <net/cfg80211.h>
  27 #include <net/mac80211.h>
  28 #include <net/codel.h>
  29 #include <net/codel_impl.h>
  30 #include <asm/unaligned.h>
  31 #include <net/fq_impl.h>
  32
  33 #include "ieee80211_i.h"
  34 #include "driver-ops.h"
  35 #include "led.h"
  36 #include "mesh.h"
  37 #include "wep.h"
  38 #include "wpa.h"
  39 #include "wme.h"
  40 #include "rate.h"
  41
  42 /* misc utils */
  43
  44 static inline void ieee80211_tx_stats(struct net_device *dev, u32 len)
  45 {
  46         struct pcpu_sw_netstats *tstats = this_cpu_ptr(dev->tstats);
  47
  48         u64_stats_update_begin(&tstats->syncp);
  49         tstats->tx_packets++;
  50         tstats->tx_bytes += len;
  51         u64_stats_update_end(&tstats->syncp);
  52 }
  53
  54 static __le16 ieee80211_duration(struct ieee80211_tx_data *tx,
  55                                  struct sk_buff *skb, int group_addr,
  56                                  int next_frag_len)
  57 {
  58         int rate, mrate, erp, dur, i, shift = 0;
  59         struct ieee80211_rate *txrate;
  60         struct ieee80211_local *local = tx->local;
  61         struct ieee80211_supported_band *sband;
  62         struct ieee80211_hdr *hdr;
  63         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
  64         struct ieee80211_chanctx_conf *chanctx_conf;
  65         u32 rate_flags = 0;
  66
  67         /* assume HW handles this */
  68         if (tx->rate.flags & (IEEE80211_TX_RC_MCS | IEEE80211_TX_RC_VHT_MCS))
  69                 return 0;
  70
  71         rcu_read_lock();
  72         chanctx_conf = rcu_dereference(tx->sdata->vif.chanctx_conf);
  73         if (chanctx_conf) {
  74                 shift = ieee80211_chandef_get_shift(&chanctx_conf->def);
  75                 rate_flags = ieee80211_chandef_rate_flags(&chanctx_conf->def);
  76         }
  77         rcu_read_unlock();
  78
  79         /* uh huh? */
  80         if (WARN_ON_ONCE(tx->rate.idx < 0))
  81                 return 0;
  82
  83         sband = local->hw.wiphy->bands[info->band];
  84         txrate = &sband->bitrates[tx->rate.idx];
  85
  86         erp = txrate->flags & IEEE80211_RATE_ERP_G;
  87
  88         /*
  89          * data and mgmt (except PS Poll):
  90          * - during CFP: 32768
  91          * - during contention period:
  92          *   if addr1 is group address: 0
  93          *   if more fragments = 0 and addr1 is individual address: time to
  94          *      transmit one ACK plus SIFS
  95          *   if more fragments = 1 and addr1 is individual address: time to
  96          *      transmit next fragment plus 2 x ACK plus 3 x SIFS
  97          *
  98          * IEEE 802.11, 9.6:
  99          * - control response frame (CTS or ACK) shall be transmitted using the
 100          *   same rate as the immediately previous frame in the frame exchange
 101          *   sequence, if this rate belongs to the PHY mandatory rates, or else
 102          *   at the highest possible rate belonging to the PHY rates in the
 103          *   BSSBasicRateSet
 104          */
 105         hdr = (struct ieee80211_hdr *)skb->data;
 106         if (ieee80211_is_ctl(hdr->frame_control)) {
 107                 /* TODO: These control frames are not currently sent by
 108                  * mac80211, but should they be implemented, this function
 109                  * needs to be updated to support duration field calculation.
 110                  *
 111                  * RTS: time needed to transmit pending data/mgmt frame plus
 112                  *    one CTS frame plus one ACK frame plus 3 x SIFS
 113                  * CTS: duration of immediately previous RTS minus time
 114                  *    required to transmit CTS and its SIFS
 115                  * ACK: 0 if immediately previous directed data/mgmt had
 116                  *    more=0, with more=1 duration in ACK frame is duration
 117                  *    from previous frame minus time needed to transmit ACK
 118                  *    and its SIFS
 119                  * PS Poll: BIT(15) | BIT(14) | aid
 120                  */
 121                 return 0;
 122         }
 123
 124         /* data/mgmt */
 125         if (0 /* FIX: data/mgmt during CFP */)
 126                 return cpu_to_le16(32768);
 127
 128         if (group_addr) /* Group address as the destination - no ACK */
 129                 return 0;
 130
 131         /* Individual destination address:
 132          * IEEE 802.11, Ch. 9.6 (after IEEE 802.11g changes)
 133          * CTS and ACK frames shall be transmitted using the highest rate in
 134          * basic rate set that is less than or equal to the rate of the
 135          * immediately previous frame and that is using the same modulation
 136          * (CCK or OFDM). If no basic rate set matches with these requirements,
 137          * the highest mandatory rate of the PHY that is less than or equal to
 138          * the rate of the previous frame is used.
 139          * Mandatory rates for IEEE 802.11g PHY: 1, 2, 5.5, 11, 6, 12, 24 Mbps
 140          */
 141         rate = -1;
 142         /* use lowest available if everything fails */
 143         mrate = sband->bitrates[0].bitrate;
 144         for (i = 0; i < sband->n_bitrates; i++) {
 145                 struct ieee80211_rate *r = &sband->bitrates[i];
 146
 147                 if (r->bitrate > txrate->bitrate)
 148                         break;
 149
 150                 if ((rate_flags & r->flags) != rate_flags)
 151                         continue;
 152
 153                 if (tx->sdata->vif.bss_conf.basic_rates & BIT(i))
 154                         rate = DIV_ROUND_UP(r->bitrate, 1 << shift);
 155
 156                 switch (sband->band) {
 157                 case NL80211_BAND_2GHZ: {
 158                         u32 flag;
 159                         if (tx->sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
 160                                 flag = IEEE80211_RATE_MANDATORY_G;
 161                         else
 162                                 flag = IEEE80211_RATE_MANDATORY_B;
 163                         if (r->flags & flag)
 164                                 mrate = r->bitrate;
 165                         break;
 166                 }
 167                 case NL80211_BAND_5GHZ:
 168                         if (r->flags & IEEE80211_RATE_MANDATORY_A)
 169                                 mrate = r->bitrate;
 170                         break;
 171                 case NL80211_BAND_60GHZ:
 172                         /* TODO, for now fall through */
 173                 case NUM_NL80211_BANDS:
 174                         WARN_ON(1);
 175                         break;
 176                 }
 177         }
 178         if (rate == -1) {
 179                 /* No matching basic rate found; use highest suitable mandatory
 180                  * PHY rate */
 181                 rate = DIV_ROUND_UP(mrate, 1 << shift);
 182         }
 183
 184         /* Don't calculate ACKs for QoS Frames with NoAck Policy set */
 185         if (ieee80211_is_data_qos(hdr->frame_control) &&
 186             *(ieee80211_get_qos_ctl(hdr)) & IEEE80211_QOS_CTL_ACK_POLICY_NOACK)
 187                 dur = 0;
 188         else
 189                 /* Time needed to transmit ACK
 190                  * (10 bytes + 4-byte FCS = 112 bits) plus SIFS; rounded up
 191                  * to closest integer */
 192                 dur = ieee80211_frame_duration(sband->band, 10, rate, erp,
 193                                 tx->sdata->vif.bss_conf.use_short_preamble,
 194                                 shift);
 195
 196         if (next_frag_len) {
 197                 /* Frame is fragmented: duration increases with time needed to
 198                  * transmit next fragment plus ACK and 2 x SIFS. */
 199                 dur *= 2; /* ACK + SIFS */
 200                 /* next fragment */
 201                 dur += ieee80211_frame_duration(sband->band, next_frag_len,
 202                                 txrate->bitrate, erp,
 203                                 tx->sdata->vif.bss_conf.use_short_preamble,
 204                                 shift);
 205         }
 206
 207         return cpu_to_le16(dur);
 208 }
 209
 210 /* tx handlers */
 211 static ieee80211_tx_result debug_noinline
 212 ieee80211_tx_h_dynamic_ps(struct ieee80211_tx_data *tx)
 213 {
 214         struct ieee80211_local *local = tx->local;
 215         struct ieee80211_if_managed *ifmgd;
 216
 217         /* driver doesn't support power save */
 218         if (!ieee80211_hw_check(&local->hw, SUPPORTS_PS))
 219                 return TX_CONTINUE;
 220
 221         /* hardware does dynamic power save */
 222         if (ieee80211_hw_check(&local->hw, SUPPORTS_DYNAMIC_PS))
 223                 return TX_CONTINUE;
 224
 225         /* dynamic power save disabled */
 226         if (local->hw.conf.dynamic_ps_timeout <= 0)
 227                 return TX_CONTINUE;
 228
 229         /* we are scanning, don't enable power save */
 230         if (local->scanning)
 231                 return TX_CONTINUE;
 232
 233         if (!local->ps_sdata)
 234                 return TX_CONTINUE;
 235
 236         /* No point if we're going to suspend */
 237         if (local->quiescing)
 238                 return TX_CONTINUE;
 239
 240         /* dynamic ps is supported only in managed mode */
 241         if (tx->sdata->vif.type != NL80211_IFTYPE_STATION)
 242                 return TX_CONTINUE;
 243
 244         ifmgd = &tx->sdata->u.mgd;
 245
 246         /*
 247          * Don't wakeup from power save if u-apsd is enabled, voip ac has
 248          * u-apsd enabled and the frame is in voip class. This effectively
 249          * means that even if all access categories have u-apsd enabled, in
 250          * practise u-apsd is only used with the voip ac. This is a
 251          * workaround for the case when received voip class packets do not
 252          * have correct qos tag for some reason, due the network or the
 253          * peer application.
 254          *
 255          * Note: ifmgd->uapsd_queues access is racy here. If the value is
 256          * changed via debugfs, user needs to reassociate manually to have
 257          * everything in sync.
 258          */
 259         if ((ifmgd->flags & IEEE80211_STA_UAPSD_ENABLED) &&
 260             (ifmgd->uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_VO) &&
 261             skb_get_queue_mapping(tx->skb) == IEEE80211_AC_VO)
 262                 return TX_CONTINUE;
 263
 264         if (local->hw.conf.flags & IEEE80211_CONF_PS) {
 265                 ieee80211_stop_queues_by_reason(&local->hw,
 266                                                 IEEE80211_MAX_QUEUE_MAP,
 267                                                 IEEE80211_QUEUE_STOP_REASON_PS,
 268                                                 false);
 269                 ifmgd->flags &= ~IEEE80211_STA_NULLFUNC_ACKED;
 270                 ieee80211_queue_work(&local->hw,
 271                                      &local->dynamic_ps_disable_work);
 272         }
 273
 274         /* Don't restart the timer if we're not disassociated */
 275         if (!ifmgd->associated)
 276                 return TX_CONTINUE;
 277
 278         mod_timer(&local->dynamic_ps_timer, jiffies +
 279                   msecs_to_jiffies(local->hw.conf.dynamic_ps_timeout));
 280
 281         return TX_CONTINUE;
 282 }
 283
 284 static ieee80211_tx_result debug_noinline
 285 ieee80211_tx_h_check_assoc(struct ieee80211_tx_data *tx)
 286 {
 287
 288         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
 289         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
 290         bool assoc = false;
 291
 292         if (unlikely(info->flags & IEEE80211_TX_CTL_INJECTED))
 293                 return TX_CONTINUE;
 294
 295         if (unlikely(test_bit(SCAN_SW_SCANNING, &tx->local->scanning)) &&
 296             test_bit(SDATA_STATE_OFFCHANNEL, &tx->sdata->state) &&
 297             !ieee80211_is_probe_req(hdr->frame_control) &&
 298             !ieee80211_is_nullfunc(hdr->frame_control))
 299                 /*
 300                  * When software scanning only nullfunc frames (to notify
 301                  * the sleep state to the AP) and probe requests (for the
 302                  * active scan) are allowed, all other frames should not be
 303                  * sent and we should not get here, but if we do
 304                  * nonetheless, drop them to avoid sending them
 305                  * off-channel. See the link below and
 306                  * ieee80211_start_scan() for more.
 307                  *
 308                  * http://article.gmane.org/gmane.linux.kernel.wireless.general/30089
 309                  */
 310                 return TX_DROP;
 311
 312         if (tx->sdata->vif.type == NL80211_IFTYPE_OCB)
 313                 return TX_CONTINUE;
 314
 315         if (tx->sdata->vif.type == NL80211_IFTYPE_WDS)
 316                 return TX_CONTINUE;
 317
 318         if (tx->flags & IEEE80211_TX_PS_BUFFERED)
 319                 return TX_CONTINUE;
 320
 321         if (tx->sta)
 322                 assoc = test_sta_flag(tx->sta, WLAN_STA_ASSOC);
 323
 324         if (likely(tx->flags & IEEE80211_TX_UNICAST)) {
 325                 if (unlikely(!assoc &&
 326                              ieee80211_is_data(hdr->frame_control))) {
 327 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
 328                         sdata_info(tx->sdata,
 329                                    "dropped data frame to not associated station %pM\n",
 330                                    hdr->addr1);
 331 #endif
 332                         I802_DEBUG_INC(tx->local->tx_handlers_drop_not_assoc);
 333                         return TX_DROP;
 334                 }
 335         } else if (unlikely(ieee80211_is_data(hdr->frame_control) &&
 336                             ieee80211_vif_get_num_mcast_if(tx->sdata) == 0)) {
 337                 /*
 338                  * No associated STAs - no need to send multicast
 339                  * frames.
 340                  */
 341                 return TX_DROP;
 342         }
 343
 344         return TX_CONTINUE;
 345 }
 346
 347 /* This function is called whenever the AP is about to exceed the maximum limit
 348  * of buffered frames for power saving STAs. This situation should not really
 349  * happen often during normal operation, so dropping the oldest buffered packet
 350  * from each queue should be OK to make some room for new frames. */
 351 static void purge_old_ps_buffers(struct ieee80211_local *local)
 352 {
 353         int total = 0, purged = 0;
 354         struct sk_buff *skb;
 355         struct ieee80211_sub_if_data *sdata;
 356         struct sta_info *sta;
 357
 358         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
 359                 struct ps_data *ps;
 360
 361                 if (sdata->vif.type == NL80211_IFTYPE_AP)
 362                         ps = &sdata->u.ap.ps;
 363                 else if (ieee80211_vif_is_mesh(&sdata->vif))
 364                         ps = &sdata->u.mesh.ps;
 365                 else
 366                         continue;
 367
 368                 skb = skb_dequeue(&ps->bc_buf);
 369                 if (skb) {
 370                         purged++;
 371                         ieee80211_free_txskb(&local->hw, skb);
 372                 }
 373                 total += skb_queue_len(&ps->bc_buf);
 374         }
 375
 376         /*
 377          * Drop one frame from each station from the lowest-priority
 378          * AC that has frames at all.
 379          */
 380         list_for_each_entry_rcu(sta, &local->sta_list, list) {
 381                 int ac;
 382
 383                 for (ac = IEEE80211_AC_BK; ac >= IEEE80211_AC_VO; ac--) {
 384                         skb = skb_dequeue(&sta->ps_tx_buf[ac]);
 385                         total += skb_queue_len(&sta->ps_tx_buf[ac]);
 386                         if (skb) {
 387                                 purged++;
 388                                 ieee80211_free_txskb(&local->hw, skb);
 389                                 break;
 390                         }
 391                 }
 392         }
 393
 394         local->total_ps_buffered = total;
 395         ps_dbg_hw(&local->hw, "PS buffers full - purged %d frames\n", purged);
 396 }
 397
 398 static ieee80211_tx_result
 399 ieee80211_tx_h_multicast_ps_buf(struct ieee80211_tx_data *tx)
 400 {
 401         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
 402         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
 403         struct ps_data *ps;
 404
 405         /*
 406          * broadcast/multicast frame
 407          *
 408          * If any of the associated/peer stations is in power save mode,
 409          * the frame is buffered to be sent after DTIM beacon frame.
 410          * This is done either by the hardware or us.
 411          */
 412
 413         /* powersaving STAs currently only in AP/VLAN/mesh mode */
 414         if (tx->sdata->vif.type == NL80211_IFTYPE_AP ||
 415             tx->sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
 416                 if (!tx->sdata->bss)
 417                         return TX_CONTINUE;
 418
 419                 ps = &tx->sdata->bss->ps;
 420         } else if (ieee80211_vif_is_mesh(&tx->sdata->vif)) {
 421                 ps = &tx->sdata->u.mesh.ps;
 422         } else {
 423                 return TX_CONTINUE;
 424         }
 425
 426
 427         /* no buffering for ordered frames */
 428         if (ieee80211_has_order(hdr->frame_control))
 429                 return TX_CONTINUE;
 430
 431         if (ieee80211_is_probe_req(hdr->frame_control))
 432                 return TX_CONTINUE;
 433
 434         if (ieee80211_hw_check(&tx->local->hw, QUEUE_CONTROL))
 435                 info->hw_queue = tx->sdata->vif.cab_queue;
 436
 437         /* no stations in PS mode */
 438         if (!atomic_read(&ps->num_sta_ps))
 439                 return TX_CONTINUE;
 440
 441         info->flags |= IEEE80211_TX_CTL_SEND_AFTER_DTIM;
 442
 443         /* device releases frame after DTIM beacon */
 444         if (!ieee80211_hw_check(&tx->local->hw, HOST_BROADCAST_PS_BUFFERING))
 445                 return TX_CONTINUE;
 446
 447         /* buffered in mac80211 */
 448         if (tx->local->total_ps_buffered >= TOTAL_MAX_TX_BUFFER)
 449                 purge_old_ps_buffers(tx->local);
 450
 451         if (skb_queue_len(&ps->bc_buf) >= AP_MAX_BC_BUFFER) {
 452                 ps_dbg(tx->sdata,
 453                        "BC TX buffer full - dropping the oldest frame\n");
 454                 ieee80211_free_txskb(&tx->local->hw, skb_dequeue(&ps->bc_buf));
 455         } else
 456                 tx->local->total_ps_buffered++;
 457
 458         skb_queue_tail(&ps->bc_buf, tx->skb);
 459
 460         return TX_QUEUED;
 461 }
 462
 463 static int ieee80211_use_mfp(__le16 fc, struct sta_info *sta,
 464                              struct sk_buff *skb)
 465 {
 466         if (!ieee80211_is_mgmt(fc))
 467                 return 0;
 468
 469         if (sta == NULL || !test_sta_flag(sta, WLAN_STA_MFP))
 470                 return 0;
 471
 472         if (!ieee80211_is_robust_mgmt_frame(skb))
 473                 return 0;
 474
 475         return 1;
 476 }
 477
 478 static ieee80211_tx_result
 479 ieee80211_tx_h_unicast_ps_buf(struct ieee80211_tx_data *tx)
 480 {
 481         struct sta_info *sta = tx->sta;
 482         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
 483         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
 484         struct ieee80211_local *local = tx->local;
 485
 486         if (unlikely(!sta))
 487                 return TX_CONTINUE;
 488
 489         if (unlikely((test_sta_flag(sta, WLAN_STA_PS_STA) ||
 490                       test_sta_flag(sta, WLAN_STA_PS_DRIVER) ||
 491                       test_sta_flag(sta, WLAN_STA_PS_DELIVER)) &&
 492                      !(info->flags & IEEE80211_TX_CTL_NO_PS_BUFFER))) {
 493                 int ac = skb_get_queue_mapping(tx->skb);
 494
 495                 if (ieee80211_is_mgmt(hdr->frame_control) &&
 496                     !ieee80211_is_bufferable_mmpdu(hdr->frame_control)) {
 497                         info->flags |= IEEE80211_TX_CTL_NO_PS_BUFFER;
 498                         return TX_CONTINUE;
 499                 }
 500
 501                 ps_dbg(sta->sdata, "STA %pM aid %d: PS buffer for AC %d\n",
 502                        sta->sta.addr, sta->sta.aid, ac);
 503                 if (tx->local->total_ps_buffered >= TOTAL_MAX_TX_BUFFER)
 504                         purge_old_ps_buffers(tx->local);
 505
 506                 /* sync with ieee80211_sta_ps_deliver_wakeup */
 507                 spin_lock(&sta->ps_lock);
 508                 /*
 509                  * STA woke up the meantime and all the frames on ps_tx_buf have
 510                  * been queued to pending queue. No reordering can happen, go
 511                  * ahead and Tx the packet.
 512                  */
 513                 if (!test_sta_flag(sta, WLAN_STA_PS_STA) &&
 514                     !test_sta_flag(sta, WLAN_STA_PS_DRIVER) &&
 515                     !test_sta_flag(sta, WLAN_STA_PS_DELIVER)) {
 516                         spin_unlock(&sta->ps_lock);
 517                         return TX_CONTINUE;
 518                 }
 519
 520                 if (skb_queue_len(&sta->ps_tx_buf[ac]) >= STA_MAX_TX_BUFFER) {
 521                         struct sk_buff *old = skb_dequeue(&sta->ps_tx_buf[ac]);
 522                         ps_dbg(tx->sdata,
 523                                "STA %pM TX buffer for AC %d full - dropping oldest frame\n",
 524                                sta->sta.addr, ac);
 525                         ieee80211_free_txskb(&local->hw, old);
 526                 } else
 527                         tx->local->total_ps_buffered++;
 528
 529                 info->control.jiffies = jiffies;
 530                 info->control.vif = &tx->sdata->vif;
 531                 info->flags |= IEEE80211_TX_INTFL_NEED_TXPROCESSING;
 532                 info->flags &= ~IEEE80211_TX_TEMPORARY_FLAGS;
 533                 skb_queue_tail(&sta->ps_tx_buf[ac], tx->skb);
 534                 spin_unlock(&sta->ps_lock);
 535
 536                 if (!timer_pending(&local->sta_cleanup))
 537                         mod_timer(&local->sta_cleanup,
 538                                   round_jiffies(jiffies +
 539                                                 STA_INFO_CLEANUP_INTERVAL));
 540
 541                 /*
 542                  * We queued up some frames, so the TIM bit might
 543                  * need to be set, recalculate it.
 544                  */
 545                 sta_info_recalc_tim(sta);
 546
 547                 return TX_QUEUED;
 548         } else if (unlikely(test_sta_flag(sta, WLAN_STA_PS_STA))) {
 549                 ps_dbg(tx->sdata,
 550                        "STA %pM in PS mode, but polling/in SP -> send frame\n",
 551                        sta->sta.addr);
 552         }
 553
 554         return TX_CONTINUE;
 555 }
 556
 557 static ieee80211_tx_result debug_noinline
 558 ieee80211_tx_h_ps_buf(struct ieee80211_tx_data *tx)
 559 {
 560         if (unlikely(tx->flags & IEEE80211_TX_PS_BUFFERED))
 561                 return TX_CONTINUE;
 562
 563         if (tx->flags & IEEE80211_TX_UNICAST)
 564                 return ieee80211_tx_h_unicast_ps_buf(tx);
 565         else
 566                 return ieee80211_tx_h_multicast_ps_buf(tx);
 567 }
 568
 569 static ieee80211_tx_result debug_noinline
 570 ieee80211_tx_h_check_control_port_protocol(struct ieee80211_tx_data *tx)
 571 {
 572         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
 573
 574         if (unlikely(tx->sdata->control_port_protocol == tx->skb->protocol)) {
 575                 if (tx->sdata->control_port_no_encrypt)
 576                         info->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
 577                 info->control.flags |= IEEE80211_TX_CTRL_PORT_CTRL_PROTO;
 578                 info->flags |= IEEE80211_TX_CTL_USE_MINRATE;
 579         }
 580
 581         return TX_CONTINUE;
 582 }
 583
 584 static ieee80211_tx_result debug_noinline
 585 ieee80211_tx_h_select_key(struct ieee80211_tx_data *tx)
 586 {
 587         struct ieee80211_key *key;
 588         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
 589         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
 590
 591         if (unlikely(info->flags & IEEE80211_TX_INTFL_DONT_ENCRYPT))
 592                 tx->key = NULL;
 593         else if (tx->sta &&
 594                  (key = rcu_dereference(tx->sta->ptk[tx->sta->ptk_idx])))
 595                 tx->key = key;
 596         else if (ieee80211_is_group_privacy_action(tx->skb) &&
 597                 (key = rcu_dereference(tx->sdata->default_multicast_key)))
 598                 tx->key = key;
 599         else if (ieee80211_is_mgmt(hdr->frame_control) &&
 600                  is_multicast_ether_addr(hdr->addr1) &&
 601                  ieee80211_is_robust_mgmt_frame(tx->skb) &&
 602                  (key = rcu_dereference(tx->sdata->default_mgmt_key)))
 603                 tx->key = key;
 604         else if (is_multicast_ether_addr(hdr->addr1) &&
 605                  (key = rcu_dereference(tx->sdata->default_multicast_key)))
 606                 tx->key = key;
 607         else if (!is_multicast_ether_addr(hdr->addr1) &&
 608                  (key = rcu_dereference(tx->sdata->default_unicast_key)))
 609                 tx->key = key;
 610         else
 611                 tx->key = NULL;
 612
 613         if (tx->key) {
 614                 bool skip_hw = false;
 615
 616                 /* TODO: add threshold stuff again */
 617
 618                 switch (tx->key->conf.cipher) {
 619                 case WLAN_CIPHER_SUITE_WEP40:
 620                 case WLAN_CIPHER_SUITE_WEP104:
 621                 case WLAN_CIPHER_SUITE_TKIP:
 622                         if (!ieee80211_is_data_present(hdr->frame_control))
 623                                 tx->key = NULL;
 624                         break;
 625                 case WLAN_CIPHER_SUITE_CCMP:
 626                 case WLAN_CIPHER_SUITE_CCMP_256:
 627                 case WLAN_CIPHER_SUITE_GCMP:
 628                 case WLAN_CIPHER_SUITE_GCMP_256:
 629                         if (!ieee80211_is_data_present(hdr->frame_control) &&
 630                             !ieee80211_use_mfp(hdr->frame_control, tx->sta,
 631                                                tx->skb) &&
 632                             !ieee80211_is_group_privacy_action(tx->skb))
 633                                 tx->key = NULL;
 634                         else
 635                                 skip_hw = (tx->key->conf.flags &
 636                                            IEEE80211_KEY_FLAG_SW_MGMT_TX) &&
 637                                         ieee80211_is_mgmt(hdr->frame_control);
 638                         break;
 639                 case WLAN_CIPHER_SUITE_AES_CMAC:
 640                 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
 641                 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
 642                 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
 643                         if (!ieee80211_is_mgmt(hdr->frame_control))
 644                                 tx->key = NULL;
 645                         break;
 646                 }
 647
 648                 if (unlikely(tx->key && tx->key->flags & KEY_FLAG_TAINTED &&
 649                              !ieee80211_is_deauth(hdr->frame_control)))
 650                         return TX_DROP;
 651
 652                 if (!skip_hw && tx->key &&
 653                     tx->key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE)
 654                         info->control.hw_key = &tx->key->conf;
 655         }
 656
 657         return TX_CONTINUE;
 658 }
 659
 660 static ieee80211_tx_result debug_noinline
 661 ieee80211_tx_h_rate_ctrl(struct ieee80211_tx_data *tx)
 662 {
 663         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
 664         struct ieee80211_hdr *hdr = (void *)tx->skb->data;
 665         struct ieee80211_supported_band *sband;
 666         u32 len;
 667         struct ieee80211_tx_rate_control txrc;
 668         struct ieee80211_sta_rates *ratetbl = NULL;
 669         bool assoc = false;
 670
 671         memset(&txrc, 0, sizeof(txrc));
 672
 673         sband = tx->local->hw.wiphy->bands[info->band];
 674
 675         len = min_t(u32, tx->skb->len + FCS_LEN,
 676                          tx->local->hw.wiphy->frag_threshold);
 677
 678         /* set up the tx rate control struct we give the RC algo */
 679         txrc.hw = &tx->local->hw;
 680         txrc.sband = sband;
 681         txrc.bss_conf = &tx->sdata->vif.bss_conf;
 682         txrc.skb = tx->skb;
 683         txrc.reported_rate.idx = -1;
 684         txrc.rate_idx_mask = tx->sdata->rc_rateidx_mask[info->band];
 685
 686         if (tx->sdata->rc_has_mcs_mask[info->band])
 687                 txrc.rate_idx_mcs_mask =
 688                         tx->sdata->rc_rateidx_mcs_mask[info->band];
 689
 690         txrc.bss = (tx->sdata->vif.type == NL80211_IFTYPE_AP ||
 691                     tx->sdata->vif.type == NL80211_IFTYPE_MESH_POINT ||
 692                     tx->sdata->vif.type == NL80211_IFTYPE_ADHOC ||
 693                     tx->sdata->vif.type == NL80211_IFTYPE_OCB);
 694
 695         /* set up RTS protection if desired */
 696         if (len > tx->local->hw.wiphy->rts_threshold) {
 697                 txrc.rts = true;
 698         }
 699
 700         info->control.use_rts = txrc.rts;
 701         info->control.use_cts_prot = tx->sdata->vif.bss_conf.use_cts_prot;
 702
 703         /*
 704          * Use short preamble if the BSS can handle it, but not for
 705          * management frames unless we know the receiver can handle
 706          * that -- the management frame might be to a station that
 707          * just wants a probe response.
 708          */
 709         if (tx->sdata->vif.bss_conf.use_short_preamble &&
 710             (ieee80211_is_data(hdr->frame_control) ||
 711              (tx->sta && test_sta_flag(tx->sta, WLAN_STA_SHORT_PREAMBLE))))
 712                 txrc.short_preamble = true;
 713
 714         info->control.short_preamble = txrc.short_preamble;
 715
 716         /* don't ask rate control when rate already injected via radiotap */
 717         if (info->control.flags & IEEE80211_TX_CTRL_RATE_INJECT)
 718                 return TX_CONTINUE;
 719
 720         if (tx->sta)
 721                 assoc = test_sta_flag(tx->sta, WLAN_STA_ASSOC);
 722
 723         /*
 724          * Lets not bother rate control if we're associated and cannot
 725          * talk to the sta. This should not happen.
 726          */
 727         if (WARN(test_bit(SCAN_SW_SCANNING, &tx->local->scanning) && assoc &&
 728                  !rate_usable_index_exists(sband, &tx->sta->sta),
 729                  "%s: Dropped data frame as no usable bitrate found while "
 730                  "scanning and associated. Target station: "
 731                  "%pM on %d GHz band\n",
 732                  tx->sdata->name, hdr->addr1,
 733                  info->band ? 5 : 2))
 734                 return TX_DROP;
 735
 736         /*
 737          * If we're associated with the sta at this point we know we can at
 738          * least send the frame at the lowest bit rate.
 739          */
 740         rate_control_get_rate(tx->sdata, tx->sta, &txrc);
 741
 742         if (tx->sta && !info->control.skip_table)
 743                 ratetbl = rcu_dereference(tx->sta->sta.rates);
 744
 745         if (unlikely(info->control.rates[0].idx < 0)) {
 746                 if (ratetbl) {
 747                         struct ieee80211_tx_rate rate = {
 748                                 .idx = ratetbl->rate[0].idx,
 749                                 .flags = ratetbl->rate[0].flags,
 750                                 .count = ratetbl->rate[0].count
 751                         };
 752
 753                         if (ratetbl->rate[0].idx < 0)
 754                                 return TX_DROP;
 755
 756                         tx->rate = rate;
 757                 } else {
 758                         return TX_DROP;
 759                 }
 760         } else {
 761                 tx->rate = info->control.rates[0];
 762         }
 763
 764         if (txrc.reported_rate.idx < 0) {
 765                 txrc.reported_rate = tx->rate;
 766                 if (tx->sta && ieee80211_is_data(hdr->frame_control))
 767                         tx->sta->tx_stats.last_rate = txrc.reported_rate;
 768         } else if (tx->sta)
 769                 tx->sta->tx_stats.last_rate = txrc.reported_rate;
 770
 771         if (ratetbl)
 772                 return TX_CONTINUE;
 773
 774         if (unlikely(!info->control.rates[0].count))
 775                 info->control.rates[0].count = 1;
 776
 777         if (WARN_ON_ONCE((info->control.rates[0].count > 1) &&
 778                          (info->flags & IEEE80211_TX_CTL_NO_ACK)))
 779                 info->control.rates[0].count = 1;
 780
 781         return TX_CONTINUE;
 782 }
 783
 784 static __le16 ieee80211_tx_next_seq(struct sta_info *sta, int tid)
 785 {
 786         u16 *seq = &sta->tid_seq[tid];
 787         __le16 ret = cpu_to_le16(*seq);
 788
 789         /* Increase the sequence number. */
 790         *seq = (*seq + 0x10) & IEEE80211_SCTL_SEQ;
 791
 792         return ret;
 793 }
 794
 795 static ieee80211_tx_result debug_noinline
 796 ieee80211_tx_h_sequence(struct ieee80211_tx_data *tx)
 797 {
 798         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
 799         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
 800         u8 *qc;
 801         int tid;
 802
 803         /*
 804          * Packet injection may want to control the sequence
 805          * number, if we have no matching interface then we
 806          * neither assign one ourselves nor ask the driver to.
 807          */
 808         if (unlikely(info->control.vif->type == NL80211_IFTYPE_MONITOR))
 809                 return TX_CONTINUE;
 810
 811         if (unlikely(ieee80211_is_ctl(hdr->frame_control)))
 812                 return TX_CONTINUE;
 813
 814         if (ieee80211_hdrlen(hdr->frame_control) < 24)
 815                 return TX_CONTINUE;
 816
 817         if (ieee80211_is_qos_nullfunc(hdr->frame_control))
 818                 return TX_CONTINUE;
 819
 820         /*
 821          * Anything but QoS data that has a sequence number field
 822          * (is long enough) gets a sequence number from the global
 823          * counter.  QoS data frames with a multicast destination
 824          * also use the global counter (802.11-2012 9.3.2.10).
 825          */
 826         if (!ieee80211_is_data_qos(hdr->frame_control) ||
 827             is_multicast_ether_addr(hdr->addr1)) {
 828                 /* driver should assign sequence number */
 829                 info->flags |= IEEE80211_TX_CTL_ASSIGN_SEQ;
 830                 /* for pure STA mode without beacons, we can do it */
 831                 hdr->seq_ctrl = cpu_to_le16(tx->sdata->sequence_number);
 832                 tx->sdata->sequence_number += 0x10;
 833                 if (tx->sta)
 834                         tx->sta->tx_stats.msdu[IEEE80211_NUM_TIDS]++;
 835                 return TX_CONTINUE;
 836         }
 837
 838         /*
 839          * This should be true for injected/management frames only, for
 840          * management frames we have set the IEEE80211_TX_CTL_ASSIGN_SEQ
 841          * above since they are not QoS-data frames.
 842          */
 843         if (!tx->sta)
 844                 return TX_CONTINUE;
 845
 846         /* include per-STA, per-TID sequence counter */
 847
 848         qc = ieee80211_get_qos_ctl(hdr);
 849         tid = *qc & IEEE80211_QOS_CTL_TID_MASK;
 850         tx->sta->tx_stats.msdu[tid]++;
 851
 852         hdr->seq_ctrl = ieee80211_tx_next_seq(tx->sta, tid);
 853
 854         return TX_CONTINUE;
 855 }
 856
 857 static int ieee80211_fragment(struct ieee80211_tx_data *tx,
 858                               struct sk_buff *skb, int hdrlen,
 859                               int frag_threshold)
 860 {
 861         struct ieee80211_local *local = tx->local;
 862         struct ieee80211_tx_info *info;
 863         struct sk_buff *tmp;
 864         int per_fragm = frag_threshold - hdrlen - FCS_LEN;
 865         int pos = hdrlen + per_fragm;
 866         int rem = skb->len - hdrlen - per_fragm;
 867
 868         if (WARN_ON(rem < 0))
 869                 return -EINVAL;
 870
 871         /* first fragment was already added to queue by caller */
 872
 873         while (rem) {
 874                 int fraglen = per_fragm;
 875
 876                 if (fraglen > rem)
 877                         fraglen = rem;
 878                 rem -= fraglen;
 879                 tmp = dev_alloc_skb(local->tx_headroom +
 880                                     frag_threshold +
 881                                     tx->sdata->encrypt_headroom +
 882                                     IEEE80211_ENCRYPT_TAILROOM);
 883                 if (!tmp)
 884                         return -ENOMEM;
 885
 886                 __skb_queue_tail(&tx->skbs, tmp);
 887
 888                 skb_reserve(tmp,
 889                             local->tx_headroom + tx->sdata->encrypt_headroom);
 890
 891                 /* copy control information */
 892                 memcpy(tmp->cb, skb->cb, sizeof(tmp->cb));
 893
 894                 info = IEEE80211_SKB_CB(tmp);
 895                 info->flags &= ~(IEEE80211_TX_CTL_CLEAR_PS_FILT |
 896                                  IEEE80211_TX_CTL_FIRST_FRAGMENT);
 897
 898                 if (rem)
 899                         info->flags |= IEEE80211_TX_CTL_MORE_FRAMES;
 900
 901                 skb_copy_queue_mapping(tmp, skb);
 902                 tmp->priority = skb->priority;
 903                 tmp->dev = skb->dev;
 904
 905                 /* copy header and data */
 906                 skb_put_data(tmp, skb->data, hdrlen);
 907                 skb_put_data(tmp, skb->data + pos, fraglen);
 908
 909                 pos += fraglen;
 910         }
 911
 912         /* adjust first fragment's length */
 913         skb_trim(skb, hdrlen + per_fragm);
 914         return 0;
 915 }
 916
 917 static ieee80211_tx_result debug_noinline
 918 ieee80211_tx_h_fragment(struct ieee80211_tx_data *tx)
 919 {
 920         struct sk_buff *skb = tx->skb;
 921         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
 922         struct ieee80211_hdr *hdr = (void *)skb->data;
 923         int frag_threshold = tx->local->hw.wiphy->frag_threshold;
 924         int hdrlen;
 925         int fragnum;
 926
 927         /* no matter what happens, tx->skb moves to tx->skbs */
 928         __skb_queue_tail(&tx->skbs, skb);
 929         tx->skb = NULL;
 930
 931         if (info->flags & IEEE80211_TX_CTL_DONTFRAG)
 932                 return TX_CONTINUE;
 933
 934         if (ieee80211_hw_check(&tx->local->hw, SUPPORTS_TX_FRAG))
 935                 return TX_CONTINUE;
 936
 937         /*
 938          * Warn when submitting a fragmented A-MPDU frame and drop it.
 939          * This scenario is handled in ieee80211_tx_prepare but extra
 940          * caution taken here as fragmented ampdu may cause Tx stop.
 941          */
 942         if (WARN_ON(info->flags & IEEE80211_TX_CTL_AMPDU))
 943                 return TX_DROP;
 944
 945         hdrlen = ieee80211_hdrlen(hdr->frame_control);
 946
 947         /* internal error, why isn't DONTFRAG set? */
 948         if (WARN_ON(skb->len + FCS_LEN <= frag_threshold))
 949                 return TX_DROP;
 950
 951         /*
 952          * Now fragment the frame. This will allocate all the fragments and
 953          * chain them (using skb as the first fragment) to skb->next.
 954          * During transmission, we will remove the successfully transmitted
 955          * fragments from this list. When the low-level driver rejects one
 956          * of the fragments then we will simply pretend to accept the skb
 957          * but store it away as pending.
 958          */
 959         if (ieee80211_fragment(tx, skb, hdrlen, frag_threshold))
 960                 return TX_DROP;
 961
 962         /* update duration/seq/flags of fragments */
 963         fragnum = 0;
 964
 965         skb_queue_walk(&tx->skbs, skb) {
 966                 const __le16 morefrags = cpu_to_le16(IEEE80211_FCTL_MOREFRAGS);
 967
 968                 hdr = (void *)skb->data;
 969                 info = IEEE80211_SKB_CB(skb);
 970
 971                 if (!skb_queue_is_last(&tx->skbs, skb)) {
 972                         hdr->frame_control |= morefrags;
 973                         /*
 974                          * No multi-rate retries for fragmented frames, that
 975                          * would completely throw off the NAV at other STAs.
 976                          */
 977                         info->control.rates[1].idx = -1;
 978                         info->control.rates[2].idx = -1;
 979                         info->control.rates[3].idx = -1;
 980                         BUILD_BUG_ON(IEEE80211_TX_MAX_RATES != 4);
 981                         info->flags &= ~IEEE80211_TX_CTL_RATE_CTRL_PROBE;
 982                 } else {
 983                         hdr->frame_control &= ~morefrags;
 984                 }
 985                 hdr->seq_ctrl |= cpu_to_le16(fragnum & IEEE80211_SCTL_FRAG);
 986                 fragnum++;
 987         }
 988
 989         return TX_CONTINUE;
 990 }
 991
 992 static ieee80211_tx_result debug_noinline
 993 ieee80211_tx_h_stats(struct ieee80211_tx_data *tx)
 994 {
 995         struct sk_buff *skb;
 996         int ac = -1;
 997
 998         if (!tx->sta)
 999                 return TX_CONTINUE;
1000
1001         skb_queue_walk(&tx->skbs, skb) {
1002                 ac = skb_get_queue_mapping(skb);
1003                 tx->sta->tx_stats.bytes[ac] += skb->len;
1004         }
1005         if (ac >= 0)
1006                 tx->sta->tx_stats.packets[ac]++;
1007
1008         return TX_CONTINUE;
1009 }
1010
1011 static ieee80211_tx_result debug_noinline
1012 ieee80211_tx_h_encrypt(struct ieee80211_tx_data *tx)
1013 {
1014         if (!tx->key)
1015                 return TX_CONTINUE;
1016
1017         switch (tx->key->conf.cipher) {
1018         case WLAN_CIPHER_SUITE_WEP40:
1019         case WLAN_CIPHER_SUITE_WEP104:
1020                 return ieee80211_crypto_wep_encrypt(tx);
1021         case WLAN_CIPHER_SUITE_TKIP:
1022                 return ieee80211_crypto_tkip_encrypt(tx);
1023         case WLAN_CIPHER_SUITE_CCMP:
1024                 return ieee80211_crypto_ccmp_encrypt(
1025                         tx, IEEE80211_CCMP_MIC_LEN);
1026         case WLAN_CIPHER_SUITE_CCMP_256:
1027                 return ieee80211_crypto_ccmp_encrypt(
1028                         tx, IEEE80211_CCMP_256_MIC_LEN);
1029         case WLAN_CIPHER_SUITE_AES_CMAC:
1030                 return ieee80211_crypto_aes_cmac_encrypt(tx);
1031         case WLAN_CIPHER_SUITE_BIP_CMAC_256:
1032                 return ieee80211_crypto_aes_cmac_256_encrypt(tx);
1033         case WLAN_CIPHER_SUITE_BIP_GMAC_128:
1034         case WLAN_CIPHER_SUITE_BIP_GMAC_256:
1035                 return ieee80211_crypto_aes_gmac_encrypt(tx);
1036         case WLAN_CIPHER_SUITE_GCMP:
1037         case WLAN_CIPHER_SUITE_GCMP_256:
1038                 return ieee80211_crypto_gcmp_encrypt(tx);
1039         default:
1040                 return ieee80211_crypto_hw_encrypt(tx);
1041         }
1042
1043         return TX_DROP;
1044 }
1045
1046 static ieee80211_tx_result debug_noinline
1047 ieee80211_tx_h_calculate_duration(struct ieee80211_tx_data *tx)
1048 {
1049         struct sk_buff *skb;
1050         struct ieee80211_hdr *hdr;
1051         int next_len;
1052         bool group_addr;
1053
1054         skb_queue_walk(&tx->skbs, skb) {
1055                 hdr = (void *) skb->data;
1056                 if (unlikely(ieee80211_is_pspoll(hdr->frame_control)))
1057                         break; /* must not overwrite AID */
1058                 if (!skb_queue_is_last(&tx->skbs, skb)) {
1059                         struct sk_buff *next = skb_queue_next(&tx->skbs, skb);
1060                         next_len = next->len;
1061                 } else
1062                         next_len = 0;
1063                 group_addr = is_multicast_ether_addr(hdr->addr1);
1064
1065                 hdr->duration_id =
1066                         ieee80211_duration(tx, skb, group_addr, next_len);
1067         }
1068
1069         return TX_CONTINUE;
1070 }
1071
1072 /* actual transmit path */
1073
1074 static bool ieee80211_tx_prep_agg(struct ieee80211_tx_data *tx,
1075                                   struct sk_buff *skb,
1076                                   struct ieee80211_tx_info *info,
1077                                   struct tid_ampdu_tx *tid_tx,
1078                                   int tid)
1079 {
1080         bool queued = false;
1081         bool reset_agg_timer = false;
1082         struct sk_buff *purge_skb = NULL;
1083
1084         if (test_bit(HT_AGG_STATE_OPERATIONAL, &tid_tx->state)) {
1085                 info->flags |= IEEE80211_TX_CTL_AMPDU;
1086                 reset_agg_timer = true;
1087         } else if (test_bit(HT_AGG_STATE_WANT_START, &tid_tx->state)) {
1088                 /*
1089                  * nothing -- this aggregation session is being started
1090                  * but that might still fail with the driver
1091                  */
1092         } else if (!tx->sta->sta.txq[tid]) {
1093                 spin_lock(&tx->sta->lock);
1094                 /*
1095                  * Need to re-check now, because we may get here
1096                  *
1097                  *  1) in the window during which the setup is actually
1098                  *     already done, but not marked yet because not all
1099                  *     packets are spliced over to the driver pending
1100                  *     queue yet -- if this happened we acquire the lock
1101                  *     either before or after the splice happens, but
1102                  *     need to recheck which of these cases happened.
1103                  *
1104                  *  2) during session teardown, if the OPERATIONAL bit
1105                  *     was cleared due to the teardown but the pointer
1106                  *     hasn't been assigned NULL yet (or we loaded it
1107                  *     before it was assigned) -- in this case it may
1108                  *     now be NULL which means we should just let the
1109                  *     packet pass through because splicing the frames
1110                  *     back is already done.
1111                  */
1112                 tid_tx = rcu_dereference_protected_tid_tx(tx->sta, tid);
1113
1114                 if (!tid_tx) {
1115                         /* do nothing, let packet pass through */
1116                 } else if (test_bit(HT_AGG_STATE_OPERATIONAL, &tid_tx->state)) {
1117                         info->flags |= IEEE80211_TX_CTL_AMPDU;
1118                         reset_agg_timer = true;
1119                 } else {
1120                         queued = true;
1121                         if (info->flags & IEEE80211_TX_CTL_NO_PS_BUFFER) {
1122                                 clear_sta_flag(tx->sta, WLAN_STA_SP);
1123                                 ps_dbg(tx->sta->sdata,
1124                                        "STA %pM aid %d: SP frame queued, close the SP w/o telling the peer\n",
1125                                        tx->sta->sta.addr, tx->sta->sta.aid);
1126                         }
1127                         info->control.vif = &tx->sdata->vif;
1128                         info->flags |= IEEE80211_TX_INTFL_NEED_TXPROCESSING;
1129                         info->flags &= ~IEEE80211_TX_TEMPORARY_FLAGS;
1130                         __skb_queue_tail(&tid_tx->pending, skb);
1131                         if (skb_queue_len(&tid_tx->pending) > STA_MAX_TX_BUFFER)
1132                                 purge_skb = __skb_dequeue(&tid_tx->pending);
1133                 }
1134                 spin_unlock(&tx->sta->lock);
1135
1136                 if (purge_skb)
1137                         ieee80211_free_txskb(&tx->local->hw, purge_skb);
1138         }
1139
1140         /* reset session timer */
1141         if (reset_agg_timer && tid_tx->timeout)
1142                 tid_tx->last_tx = jiffies;
1143
1144         return queued;
1145 }
1146
1147 /*
1148  * initialises @tx
1149  * pass %NULL for the station if unknown, a valid pointer if known
1150  * or an ERR_PTR() if the station is known not to exist
1151  */
1152 static ieee80211_tx_result
1153 ieee80211_tx_prepare(struct ieee80211_sub_if_data *sdata,
1154                      struct ieee80211_tx_data *tx,
1155                      struct sta_info *sta, struct sk_buff *skb)
1156 {
1157         struct ieee80211_local *local = sdata->local;
1158         struct ieee80211_hdr *hdr;
1159         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1160         int tid;
1161         u8 *qc;
1162
1163         memset(tx, 0, sizeof(*tx));
1164         tx->skb = skb;
1165         tx->local = local;
1166         tx->sdata = sdata;
1167         __skb_queue_head_init(&tx->skbs);
1168
1169         /*
1170          * If this flag is set to true anywhere, and we get here,
1171          * we are doing the needed processing, so remove the flag
1172          * now.
1173          */
1174         info->flags &= ~IEEE80211_TX_INTFL_NEED_TXPROCESSING;
1175
1176         hdr = (struct ieee80211_hdr *) skb->data;
1177
1178         if (likely(sta)) {
1179                 if (!IS_ERR(sta))
1180                         tx->sta = sta;
1181         } else {
1182                 if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
1183                         tx->sta = rcu_dereference(sdata->u.vlan.sta);
1184                         if (!tx->sta && sdata->wdev.use_4addr)
1185                                 return TX_DROP;
1186                 } else if (info->flags & (IEEE80211_TX_INTFL_NL80211_FRAME_TX |
1187                                           IEEE80211_TX_CTL_INJECTED) ||
1188                            tx->sdata->control_port_protocol == tx->skb->protocol) {
1189                         tx->sta = sta_info_get_bss(sdata, hdr->addr1);
1190                 }
1191                 if (!tx->sta && !is_multicast_ether_addr(hdr->addr1))
1192                         tx->sta = sta_info_get(sdata, hdr->addr1);
1193         }
1194
1195         if (tx->sta && ieee80211_is_data_qos(hdr->frame_control) &&
1196             !ieee80211_is_qos_nullfunc(hdr->frame_control) &&
1197             ieee80211_hw_check(&local->hw, AMPDU_AGGREGATION) &&
1198             !ieee80211_hw_check(&local->hw, TX_AMPDU_SETUP_IN_HW)) {
1199                 struct tid_ampdu_tx *tid_tx;
1200
1201                 qc = ieee80211_get_qos_ctl(hdr);
1202                 tid = *qc & IEEE80211_QOS_CTL_TID_MASK;
1203
1204                 tid_tx = rcu_dereference(tx->sta->ampdu_mlme.tid_tx[tid]);
1205                 if (tid_tx) {
1206                         bool queued;
1207
1208                         queued = ieee80211_tx_prep_agg(tx, skb, info,
1209                                                        tid_tx, tid);
1210
1211                         if (unlikely(queued))
1212                                 return TX_QUEUED;
1213                 }
1214         }
1215
1216         if (is_multicast_ether_addr(hdr->addr1)) {
1217                 tx->flags &= ~IEEE80211_TX_UNICAST;
1218                 info->flags |= IEEE80211_TX_CTL_NO_ACK;
1219         } else
1220                 tx->flags |= IEEE80211_TX_UNICAST;
1221
1222         if (!(info->flags & IEEE80211_TX_CTL_DONTFRAG)) {
1223                 if (!(tx->flags & IEEE80211_TX_UNICAST) ||
1224                     skb->len + FCS_LEN <= local->hw.wiphy->frag_threshold ||
1225                     info->flags & IEEE80211_TX_CTL_AMPDU)
1226                         info->flags |= IEEE80211_TX_CTL_DONTFRAG;
1227         }
1228
1229         if (!tx->sta)
1230                 info->flags |= IEEE80211_TX_CTL_CLEAR_PS_FILT;
1231         else if (test_and_clear_sta_flag(tx->sta, WLAN_STA_CLEAR_PS_FILT)) {
1232                 info->flags |= IEEE80211_TX_CTL_CLEAR_PS_FILT;
1233                 ieee80211_check_fast_xmit(tx->sta);
1234         }
1235
1236         info->flags |= IEEE80211_TX_CTL_FIRST_FRAGMENT;
1237
1238         return TX_CONTINUE;
1239 }
1240
1241 static struct txq_info *ieee80211_get_txq(struct ieee80211_local *local,
1242                                           struct ieee80211_vif *vif,
1243                                           struct sta_info *sta,
1244                                           struct sk_buff *skb)
1245 {
1246         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1247         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1248         struct ieee80211_txq *txq = NULL;
1249
1250         if ((info->flags & IEEE80211_TX_CTL_SEND_AFTER_DTIM) ||
1251             (info->control.flags & IEEE80211_TX_CTRL_PS_RESPONSE))
1252                 return NULL;
1253
1254         if (!ieee80211_is_data(hdr->frame_control))
1255                 return NULL;
1256
1257         if (sta) {
1258                 u8 tid = skb->priority & IEEE80211_QOS_CTL_TID_MASK;
1259
1260                 if (!sta->uploaded)
1261                         return NULL;
1262
1263                 txq = sta->sta.txq[tid];
1264         } else if (vif) {
1265                 txq = vif->txq;
1266         }
1267
1268         if (!txq)
1269                 return NULL;
1270
1271         return to_txq_info(txq);
1272 }
1273
1274 static void ieee80211_set_skb_enqueue_time(struct sk_buff *skb)
1275 {
1276         IEEE80211_SKB_CB(skb)->control.enqueue_time = codel_get_time();
1277 }
1278
1279 static void ieee80211_set_skb_vif(struct sk_buff *skb, struct txq_info *txqi)
1280 {
1281         IEEE80211_SKB_CB(skb)->control.vif = txqi->txq.vif;
1282 }
1283
1284 static u32 codel_skb_len_func(const struct sk_buff *skb)
1285 {
1286         return skb->len;
1287 }
1288
1289 static codel_time_t codel_skb_time_func(const struct sk_buff *skb)
1290 {
1291         const struct ieee80211_tx_info *info;
1292
1293         info = (const struct ieee80211_tx_info *)skb->cb;
1294         return info->control.enqueue_time;
1295 }
1296
1297 static struct sk_buff *codel_dequeue_func(struct codel_vars *cvars,
1298                                           void *ctx)
1299 {
1300         struct ieee80211_local *local;
1301         struct txq_info *txqi;
1302         struct fq *fq;
1303         struct fq_flow *flow;
1304
1305         txqi = ctx;
1306         local = vif_to_sdata(txqi->txq.vif)->local;
1307         fq = &local->fq;
1308
1309         if (cvars == &txqi->def_cvars)
1310                 flow = &txqi->def_flow;
1311         else
1312                 flow = &fq->flows[cvars - local->cvars];
1313
1314         return fq_flow_dequeue(fq, flow);
1315 }
1316
1317 static void codel_drop_func(struct sk_buff *skb,
1318                             void *ctx)
1319 {
1320         struct ieee80211_local *local;
1321         struct ieee80211_hw *hw;
1322         struct txq_info *txqi;
1323
1324         txqi = ctx;
1325         local = vif_to_sdata(txqi->txq.vif)->local;
1326         hw = &local->hw;
1327
1328         ieee80211_free_txskb(hw, skb);
1329 }
1330
1331 static struct sk_buff *fq_tin_dequeue_func(struct fq *fq,
1332                                            struct fq_tin *tin,
1333                                            struct fq_flow *flow)
1334 {
1335         struct ieee80211_local *local;
1336         struct txq_info *txqi;
1337         struct codel_vars *cvars;
1338         struct codel_params *cparams;
1339         struct codel_stats *cstats;
1340
1341         local = container_of(fq, struct ieee80211_local, fq);
1342         txqi = container_of(tin, struct txq_info, tin);
1343         cstats = &txqi->cstats;
1344
1345         if (txqi->txq.sta) {
1346                 struct sta_info *sta = container_of(txqi->txq.sta,
1347                                                     struct sta_info, sta);
1348                 cparams = &sta->cparams;
1349         } else {
1350                 cparams = &local->cparams;
1351         }
1352
1353         if (flow == &txqi->def_flow)
1354                 cvars = &txqi->def_cvars;
1355         else
1356                 cvars = &local->cvars[flow - fq->flows];
1357
1358         return codel_dequeue(txqi,
1359                              &flow->backlog,
1360                              cparams,
1361                              cvars,
1362                              cstats,
1363                              codel_skb_len_func,
1364                              codel_skb_time_func,
1365                              codel_drop_func,
1366                              codel_dequeue_func);
1367 }
1368
1369 static void fq_skb_free_func(struct fq *fq,
1370                              struct fq_tin *tin,
1371                              struct fq_flow *flow,
1372                              struct sk_buff *skb)
1373 {
1374         struct ieee80211_local *local;
1375
1376         local = container_of(fq, struct ieee80211_local, fq);
1377         ieee80211_free_txskb(&local->hw, skb);
1378 }
1379
1380 static struct fq_flow *fq_flow_get_default_func(struct fq *fq,
1381                                                 struct fq_tin *tin,
1382                                                 int idx,
1383                                                 struct sk_buff *skb)
1384 {
1385         struct txq_info *txqi;
1386
1387         txqi = container_of(tin, struct txq_info, tin);
1388         return &txqi->def_flow;
1389 }
1390
1391 static void ieee80211_txq_enqueue(struct ieee80211_local *local,
1392                                   struct txq_info *txqi,
1393                                   struct sk_buff *skb)
1394 {
1395         struct fq *fq = &local->fq;
1396         struct fq_tin *tin = &txqi->tin;
1397
1398         ieee80211_set_skb_enqueue_time(skb);
1399         fq_tin_enqueue(fq, tin, skb,
1400                        fq_skb_free_func,
1401                        fq_flow_get_default_func);
1402 }
1403
1404 void ieee80211_txq_init(struct ieee80211_sub_if_data *sdata,
1405                         struct sta_info *sta,
1406                         struct txq_info *txqi, int tid)
1407 {
1408         fq_tin_init(&txqi->tin);
1409         fq_flow_init(&txqi->def_flow);
1410         codel_vars_init(&txqi->def_cvars);
1411         codel_stats_init(&txqi->cstats);
1412         __skb_queue_head_init(&txqi->frags);
1413
1414         txqi->txq.vif = &sdata->vif;
1415
1416         if (sta) {
1417                 txqi->txq.sta = &sta->sta;
1418                 sta->sta.txq[tid] = &txqi->txq;
1419                 txqi->txq.tid = tid;
1420                 txqi->txq.ac = ieee80211_ac_from_tid(tid);
1421         } else {
1422                 sdata->vif.txq = &txqi->txq;
1423                 txqi->txq.tid = 0;
1424                 txqi->txq.ac = IEEE80211_AC_BE;
1425         }
1426 }
1427
1428 void ieee80211_txq_purge(struct ieee80211_local *local,
1429                          struct txq_info *txqi)
1430 {
1431         struct fq *fq = &local->fq;
1432         struct fq_tin *tin = &txqi->tin;
1433
1434         fq_tin_reset(fq, tin, fq_skb_free_func);
1435         ieee80211_purge_tx_queue(&local->hw, &txqi->frags);
1436 }
1437
1438 int ieee80211_txq_setup_flows(struct ieee80211_local *local)
1439 {
1440         struct fq *fq = &local->fq;
1441         int ret;
1442         int i;
1443         bool supp_vht = false;
1444         enum nl80211_band band;
1445
1446         if (!local->ops->wake_tx_queue)
1447                 return 0;
1448
1449         ret = fq_init(fq, 4096);
1450         if (ret)
1451                 return ret;
1452
1453         /*
1454          * If the hardware doesn't support VHT, it is safe to limit the maximum
1455          * queue size. 4 Mbytes is 64 max-size aggregates in 802.11n.
1456          */
1457         for (band = 0; band < NUM_NL80211_BANDS; band++) {
1458                 struct ieee80211_supported_band *sband;
1459
1460                 sband = local->hw.wiphy->bands[band];
1461                 if (!sband)
1462                         continue;
1463
1464                 supp_vht = supp_vht || sband->vht_cap.vht_supported;
1465         }
1466
1467         if (!supp_vht)
1468                 fq->memory_limit = 4 << 20; /* 4 Mbytes */
1469
1470         codel_params_init(&local->cparams);
1471         local->cparams.interval = MS2TIME(100);
1472         local->cparams.target = MS2TIME(20);
1473         local->cparams.ecn = true;
1474
1475         local->cvars = kcalloc(fq->flows_cnt, sizeof(local->cvars[0]),
1476                                GFP_KERNEL);
1477         if (!local->cvars) {
1478                 spin_lock_bh(&fq->lock);
1479                 fq_reset(fq, fq_skb_free_func);
1480                 spin_unlock_bh(&fq->lock);
1481                 return -ENOMEM;
1482         }
1483
1484         for (i = 0; i < fq->flows_cnt; i++)
1485                 codel_vars_init(&local->cvars[i]);
1486
1487         return 0;
1488 }
1489
1490 void ieee80211_txq_teardown_flows(struct ieee80211_local *local)
1491 {
1492         struct fq *fq = &local->fq;
1493
1494         if (!local->ops->wake_tx_queue)
1495                 return;
1496
1497         kfree(local->cvars);
1498         local->cvars = NULL;
1499
1500         spin_lock_bh(&fq->lock);
1501         fq_reset(fq, fq_skb_free_func);
1502         spin_unlock_bh(&fq->lock);
1503 }
1504
1505 static bool ieee80211_queue_skb(struct ieee80211_local *local,
1506                                 struct ieee80211_sub_if_data *sdata,
1507                                 struct sta_info *sta,
1508                                 struct sk_buff *skb)
1509 {
1510         struct fq *fq = &local->fq;
1511         struct ieee80211_vif *vif;
1512         struct txq_info *txqi;
1513
1514         if (!local->ops->wake_tx_queue ||
1515             sdata->vif.type == NL80211_IFTYPE_MONITOR)
1516                 return false;
1517
1518         if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
1519                 sdata = container_of(sdata->bss,
1520                                      struct ieee80211_sub_if_data, u.ap);
1521
1522         vif = &sdata->vif;
1523         txqi = ieee80211_get_txq(local, vif, sta, skb);
1524
1525         if (!txqi)
1526                 return false;
1527
1528         spin_lock_bh(&fq->lock);
1529         ieee80211_txq_enqueue(local, txqi, skb);
1530         spin_unlock_bh(&fq->lock);
1531
1532         drv_wake_tx_queue(local, txqi);
1533
1534         return true;
1535 }
1536
1537 static bool ieee80211_tx_frags(struct ieee80211_local *local,
1538                                struct ieee80211_vif *vif,
1539                                struct ieee80211_sta *sta,
1540                                struct sk_buff_head *skbs,
1541                                bool txpending)
1542 {
1543         struct ieee80211_tx_control control = {};
1544         struct sk_buff *skb, *tmp;
1545         unsigned long flags;
1546
1547         skb_queue_walk_safe(skbs, skb, tmp) {
1548                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1549                 int q = info->hw_queue;
1550
1551 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
1552                 if (WARN_ON_ONCE(q >= local->hw.queues)) {
1553                         __skb_unlink(skb, skbs);
1554                         ieee80211_free_txskb(&local->hw, skb);
1555                         continue;
1556                 }
1557 #endif
1558
1559                 spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
1560                 if (local->queue_stop_reasons[q] ||
1561                     (!txpending && !skb_queue_empty(&local->pending[q]))) {
1562                         if (unlikely(info->flags &
1563                                      IEEE80211_TX_INTFL_OFFCHAN_TX_OK)) {
1564                                 if (local->queue_stop_reasons[q] &
1565                                     ~BIT(IEEE80211_QUEUE_STOP_REASON_OFFCHANNEL)) {
1566                                         /*
1567                                          * Drop off-channel frames if queues
1568                                          * are stopped for any reason other
1569                                          * than off-channel operation. Never
1570                                          * queue them.
1571                                          */
1572                                         spin_unlock_irqrestore(
1573                                                 &local->queue_stop_reason_lock,
1574                                                 flags);
1575                                         ieee80211_purge_tx_queue(&local->hw,
1576                                                                  skbs);
1577                                         return true;
1578                                 }
1579                         } else {
1580
1581                                 /*
1582                                  * Since queue is stopped, queue up frames for
1583                                  * later transmission from the tx-pending
1584                                  * tasklet when the queue is woken again.
1585                                  */
1586                                 if (txpending)
1587                                         skb_queue_splice_init(skbs,
1588                                                               &local->pending[q]);
1589                                 else
1590                                         skb_queue_splice_tail_init(skbs,
1591                                                                    &local->pending[q]);
1592
1593                                 spin_unlock_irqrestore(&local->queue_stop_reason_lock,
1594                                                        flags);
1595                                 return false;
1596                         }
1597                 }
1598                 spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
1599
1600                 info->control.vif = vif;
1601                 control.sta = sta;
1602
1603                 __skb_unlink(skb, skbs);
1604                 drv_tx(local, &control, skb);
1605         }
1606
1607         return true;
1608 }
1609
1610 /*
1611  * Returns false if the frame couldn't be transmitted but was queued instead.
1612  */
1613 static bool __ieee80211_tx(struct ieee80211_local *local,
1614                            struct sk_buff_head *skbs, int led_len,
1615                            struct sta_info *sta, bool txpending)
1616 {
1617         struct ieee80211_tx_info *info;
1618         struct ieee80211_sub_if_data *sdata;
1619         struct ieee80211_vif *vif;
1620         struct ieee80211_sta *pubsta;
1621         struct sk_buff *skb;
1622         bool result = true;
1623         __le16 fc;
1624
1625         if (WARN_ON(skb_queue_empty(skbs)))
1626                 return true;
1627
1628         skb = skb_peek(skbs);
1629         fc = ((struct ieee80211_hdr *)skb->data)->frame_control;
1630         info = IEEE80211_SKB_CB(skb);
1631         sdata = vif_to_sdata(info->control.vif);
1632         if (sta && !sta->uploaded)
1633                 sta = NULL;
1634
1635         if (sta)
1636                 pubsta = &sta->sta;
1637         else
1638                 pubsta = NULL;
1639
1640         switch (sdata->vif.type) {
1641         case NL80211_IFTYPE_MONITOR:
1642                 if (sdata->u.mntr.flags & MONITOR_FLAG_ACTIVE) {
1643                         vif = &sdata->vif;
1644                         break;
1645                 }
1646                 sdata = rcu_dereference(local->monitor_sdata);
1647                 if (sdata) {
1648                         vif = &sdata->vif;
1649                         info->hw_queue =
1650                                 vif->hw_queue[skb_get_queue_mapping(skb)];
1651                 } else if (ieee80211_hw_check(&local->hw, QUEUE_CONTROL)) {
1652                         ieee80211_purge_tx_queue(&local->hw, skbs);
1653                         return true;
1654                 } else
1655                         vif = NULL;
1656                 break;
1657         case NL80211_IFTYPE_AP_VLAN:
1658                 sdata = container_of(sdata->bss,
1659                                      struct ieee80211_sub_if_data, u.ap);
1660                 /* fall through */
1661         default:
1662                 vif = &sdata->vif;
1663                 break;
1664         }
1665
1666         result = ieee80211_tx_frags(local, vif, pubsta, skbs,
1667                                     txpending);
1668
1669         ieee80211_tpt_led_trig_tx(local, fc, led_len);
1670
1671         WARN_ON_ONCE(!skb_queue_empty(skbs));
1672
1673         return result;
1674 }
1675
1676 /*
1677  * Invoke TX handlers, return 0 on success and non-zero if the
1678  * frame was dropped or queued.
1679  *
1680  * The handlers are split into an early and late part. The latter is everything
1681  * that can be sensitive to reordering, and will be deferred to after packets
1682  * are dequeued from the intermediate queues (when they are enabled).
1683  */
1684 static int invoke_tx_handlers_early(struct ieee80211_tx_data *tx)
1685 {
1686         ieee80211_tx_result res = TX_DROP;
1687
1688 #define CALL_TXH(txh) \
1689         do {                            \
1690                 res = txh(tx);          \
1691                 if (res != TX_CONTINUE) \
1692                         goto txh_done;  \
1693         } while (0)
1694
1695         CALL_TXH(ieee80211_tx_h_dynamic_ps);
1696         CALL_TXH(ieee80211_tx_h_check_assoc);
1697         CALL_TXH(ieee80211_tx_h_ps_buf);
1698         CALL_TXH(ieee80211_tx_h_check_control_port_protocol);
1699         CALL_TXH(ieee80211_tx_h_select_key);
1700         if (!ieee80211_hw_check(&tx->local->hw, HAS_RATE_CONTROL))
1701                 CALL_TXH(ieee80211_tx_h_rate_ctrl);
1702
1703  txh_done:
1704         if (unlikely(res == TX_DROP)) {
1705                 I802_DEBUG_INC(tx->local->tx_handlers_drop);
1706                 if (tx->skb)
1707                         ieee80211_free_txskb(&tx->local->hw, tx->skb);
1708                 else
1709                         ieee80211_purge_tx_queue(&tx->local->hw, &tx->skbs);
1710                 return -1;
1711         } else if (unlikely(res == TX_QUEUED)) {
1712                 I802_DEBUG_INC(tx->local->tx_handlers_queued);
1713                 return -1;
1714         }
1715
1716         return 0;
1717 }
1718
1719 /*
1720  * Late handlers can be called while the sta lock is held. Handlers that can
1721  * cause packets to be generated will cause deadlock!
1722  */
1723 static int invoke_tx_handlers_late(struct ieee80211_tx_data *tx)
1724 {
1725         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
1726         ieee80211_tx_result res = TX_CONTINUE;
1727
1728         if (unlikely(info->flags & IEEE80211_TX_INTFL_RETRANSMISSION)) {
1729                 __skb_queue_tail(&tx->skbs, tx->skb);
1730                 tx->skb = NULL;
1731                 goto txh_done;
1732         }
1733
1734         CALL_TXH(ieee80211_tx_h_michael_mic_add);
1735         CALL_TXH(ieee80211_tx_h_sequence);
1736         CALL_TXH(ieee80211_tx_h_fragment);
1737         /* handlers after fragment must be aware of tx info fragmentation! */
1738         CALL_TXH(ieee80211_tx_h_stats);
1739         CALL_TXH(ieee80211_tx_h_encrypt);
1740         if (!ieee80211_hw_check(&tx->local->hw, HAS_RATE_CONTROL))
1741                 CALL_TXH(ieee80211_tx_h_calculate_duration);
1742 #undef CALL_TXH
1743
1744  txh_done:
1745         if (unlikely(res == TX_DROP)) {
1746                 I802_DEBUG_INC(tx->local->tx_handlers_drop);
1747                 if (tx->skb)
1748                         ieee80211_free_txskb(&tx->local->hw, tx->skb);
1749                 else
1750                         ieee80211_purge_tx_queue(&tx->local->hw, &tx->skbs);
1751                 return -1;
1752         } else if (unlikely(res == TX_QUEUED)) {
1753                 I802_DEBUG_INC(tx->local->tx_handlers_queued);
1754                 return -1;
1755         }
1756
1757         return 0;
1758 }
1759
1760 static int invoke_tx_handlers(struct ieee80211_tx_data *tx)
1761 {
1762         int r = invoke_tx_handlers_early(tx);
1763
1764         if (r)
1765                 return r;
1766         return invoke_tx_handlers_late(tx);
1767 }
1768
1769 bool ieee80211_tx_prepare_skb(struct ieee80211_hw *hw,
1770                               struct ieee80211_vif *vif, struct sk_buff *skb,
1771                               int band, struct ieee80211_sta **sta)
1772 {
1773         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1774         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1775         struct ieee80211_tx_data tx;
1776         struct sk_buff *skb2;
1777
1778         if (ieee80211_tx_prepare(sdata, &tx, NULL, skb) == TX_DROP)
1779                 return false;
1780
1781         info->band = band;
1782         info->control.vif = vif;
1783         info->hw_queue = vif->hw_queue[skb_get_queue_mapping(skb)];
1784
1785         if (invoke_tx_handlers(&tx))
1786                 return false;
1787
1788         if (sta) {
1789                 if (tx.sta)
1790                         *sta = &tx.sta->sta;
1791                 else
1792                         *sta = NULL;
1793         }
1794
1795         /* this function isn't suitable for fragmented data frames */
1796         skb2 = __skb_dequeue(&tx.skbs);
1797         if (WARN_ON(skb2 != skb || !skb_queue_empty(&tx.skbs))) {
1798                 ieee80211_free_txskb(hw, skb2);
1799                 ieee80211_purge_tx_queue(hw, &tx.skbs);
1800                 return false;
1801         }
1802
1803         return true;
1804 }
1805 EXPORT_SYMBOL(ieee80211_tx_prepare_skb);
1806
1807 /*
1808  * Returns false if the frame couldn't be transmitted but was queued instead.
1809  */
1810 static bool ieee80211_tx(struct ieee80211_sub_if_data *sdata,
1811                          struct sta_info *sta, struct sk_buff *skb,
1812                          bool txpending)
1813 {
1814         struct ieee80211_local *local = sdata->local;
1815         struct ieee80211_tx_data tx;
1816         ieee80211_tx_result res_prepare;
1817         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1818         bool result = true;
1819         int led_len;
1820
1821         if (unlikely(skb->len < 10)) {
1822                 dev_kfree_skb(skb);
1823                 return true;
1824         }
1825
1826         /* initialises tx */
1827         led_len = skb->len;
1828         res_prepare = ieee80211_tx_prepare(sdata, &tx, sta, skb);
1829
1830         if (unlikely(res_prepare == TX_DROP)) {
1831                 ieee80211_free_txskb(&local->hw, skb);
1832                 return true;
1833         } else if (unlikely(res_prepare == TX_QUEUED)) {
1834                 return true;
1835         }
1836
1837         /* set up hw_queue value early */
1838         if (!(info->flags & IEEE80211_TX_CTL_TX_OFFCHAN) ||
1839             !ieee80211_hw_check(&local->hw, QUEUE_CONTROL))
1840                 info->hw_queue =
1841                         sdata->vif.hw_queue[skb_get_queue_mapping(skb)];
1842
1843         if (invoke_tx_handlers_early(&tx))
1844                 return false;
1845
1846         if (ieee80211_queue_skb(local, sdata, tx.sta, tx.skb))
1847                 return true;
1848
1849         if (!invoke_tx_handlers_late(&tx))
1850                 result = __ieee80211_tx(local, &tx.skbs, led_len,
1851                                         tx.sta, txpending);
1852
1853         return result;
1854 }
1855
1856 /* device xmit handlers */
1857
1858 static int ieee80211_skb_resize(struct ieee80211_sub_if_data *sdata,
1859                                 struct sk_buff *skb,
1860                                 int head_need, bool may_encrypt)
1861 {
1862         struct ieee80211_local *local = sdata->local;
1863         int tail_need = 0;
1864
1865         if (may_encrypt && sdata->crypto_tx_tailroom_needed_cnt) {
1866                 tail_need = IEEE80211_ENCRYPT_TAILROOM;
1867                 tail_need -= skb_tailroom(skb);
1868                 tail_need = max_t(int, tail_need, 0);
1869         }
1870
1871         if (skb_cloned(skb) &&
1872             (!ieee80211_hw_check(&local->hw, SUPPORTS_CLONED_SKBS) ||
1873              !skb_clone_writable(skb, ETH_HLEN) ||
1874              (may_encrypt && sdata->crypto_tx_tailroom_needed_cnt)))
1875                 I802_DEBUG_INC(local->tx_expand_skb_head_cloned);
1876         else if (head_need || tail_need)
1877                 I802_DEBUG_INC(local->tx_expand_skb_head);
1878         else
1879                 return 0;
1880
1881         if (pskb_expand_head(skb, head_need, tail_need, GFP_ATOMIC)) {
1882                 wiphy_debug(local->hw.wiphy,
1883                             "failed to reallocate TX buffer\n");
1884                 return -ENOMEM;
1885         }
1886
1887         return 0;
1888 }
1889
1890 void ieee80211_xmit(struct ieee80211_sub_if_data *sdata,
1891                     struct sta_info *sta, struct sk_buff *skb)
1892 {
1893         struct ieee80211_local *local = sdata->local;
1894         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1895         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1896         int headroom;
1897         bool may_encrypt;
1898
1899         may_encrypt = !(info->flags & IEEE80211_TX_INTFL_DONT_ENCRYPT);
1900
1901         headroom = local->tx_headroom;
1902         if (may_encrypt)
1903                 headroom += sdata->encrypt_headroom;
1904         headroom -= skb_headroom(skb);
1905         headroom = max_t(int, 0, headroom);
1906
1907         if (ieee80211_skb_resize(sdata, skb, headroom, may_encrypt)) {
1908                 ieee80211_free_txskb(&local->hw, skb);
1909                 return;
1910         }
1911
1912         hdr = (struct ieee80211_hdr *) skb->data;
1913         info->control.vif = &sdata->vif;
1914
1915         if (ieee80211_vif_is_mesh(&sdata->vif)) {
1916                 if (ieee80211_is_data(hdr->frame_control) &&
1917                     is_unicast_ether_addr(hdr->addr1)) {
1918                         if (mesh_nexthop_resolve(sdata, skb))
1919                                 return; /* skb queued: don't free */
1920                 } else {
1921                         ieee80211_mps_set_frame_flags(sdata, NULL, hdr);
1922                 }
1923         }
1924
1925         ieee80211_set_qos_hdr(sdata, skb);
1926         ieee80211_tx(sdata, sta, skb, false);
1927 }
1928
1929 static bool ieee80211_parse_tx_radiotap(struct ieee80211_local *local,
1930                                         struct sk_buff *skb)
1931 {
1932         struct ieee80211_radiotap_iterator iterator;
1933         struct ieee80211_radiotap_header *rthdr =
1934                 (struct ieee80211_radiotap_header *) skb->data;
1935         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1936         struct ieee80211_supported_band *sband =
1937                 local->hw.wiphy->bands[info->band];
1938         int ret = ieee80211_radiotap_iterator_init(&iterator, rthdr, skb->len,
1939                                                    NULL);
1940         u16 txflags;
1941         u16 rate = 0;
1942         bool rate_found = false;
1943         u8 rate_retries = 0;
1944         u16 rate_flags = 0;
1945         u8 mcs_known, mcs_flags, mcs_bw;
1946         u16 vht_known;
1947         u8 vht_mcs = 0, vht_nss = 0;
1948         int i;
1949
1950         info->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT |
1951                        IEEE80211_TX_CTL_DONTFRAG;
1952
1953         /*
1954          * for every radiotap entry that is present
1955          * (ieee80211_radiotap_iterator_next returns -ENOENT when no more
1956          * entries present, or -EINVAL on error)
1957          */
1958
1959         while (!ret) {
1960                 ret = ieee80211_radiotap_iterator_next(&iterator);
1961
1962                 if (ret)
1963                         continue;
1964
1965                 /* see if this argument is something we can use */
1966                 switch (iterator.this_arg_index) {
1967                 /*
1968                  * You must take care when dereferencing iterator.this_arg
1969                  * for multibyte types... the pointer is not aligned.  Use
1970                  * get_unaligned((type *)iterator.this_arg) to dereference
1971                  * iterator.this_arg for type "type" safely on all arches.
1972                 */
1973                 case IEEE80211_RADIOTAP_FLAGS:
1974                         if (*iterator.this_arg & IEEE80211_RADIOTAP_F_FCS) {
1975                                 /*
1976                                  * this indicates that the skb we have been
1977                                  * handed has the 32-bit FCS CRC at the end...
1978                                  * we should react to that by snipping it off
1979                                  * because it will be recomputed and added
1980                                  * on transmission
1981                                  */
1982                                 if (skb->len < (iterator._max_length + FCS_LEN))
1983                                         return false;
1984
1985                                 skb_trim(skb, skb->len - FCS_LEN);
1986                         }
1987                         if (*iterator.this_arg & IEEE80211_RADIOTAP_F_WEP)
1988                                 info->flags &= ~IEEE80211_TX_INTFL_DONT_ENCRYPT;
1989                         if (*iterator.this_arg & IEEE80211_RADIOTAP_F_FRAG)
1990                                 info->flags &= ~IEEE80211_TX_CTL_DONTFRAG;
1991                         break;
1992
1993                 case IEEE80211_RADIOTAP_TX_FLAGS:
1994                         txflags = get_unaligned_le16(iterator.this_arg);
1995                         if (txflags & IEEE80211_RADIOTAP_F_TX_NOACK)
1996                                 info->flags |= IEEE80211_TX_CTL_NO_ACK;
1997                         break;
1998
1999                 case IEEE80211_RADIOTAP_RATE:
2000                         rate = *iterator.this_arg;
2001                         rate_flags = 0;
2002                         rate_found = true;
2003                         break;
2004
2005                 case IEEE80211_RADIOTAP_DATA_RETRIES:
2006                         rate_retries = *iterator.this_arg;
2007                         break;
2008
2009                 case IEEE80211_RADIOTAP_MCS:
2010                         mcs_known = iterator.this_arg[0];
2011                         mcs_flags = iterator.this_arg[1];
2012                         if (!(mcs_known & IEEE80211_RADIOTAP_MCS_HAVE_MCS))
2013                                 break;
2014
2015                         rate_found = true;
2016                         rate = iterator.this_arg[2];
2017                         rate_flags = IEEE80211_TX_RC_MCS;
2018
2019                         if (mcs_known & IEEE80211_RADIOTAP_MCS_HAVE_GI &&
2020                             mcs_flags & IEEE80211_RADIOTAP_MCS_SGI)
2021                                 rate_flags |= IEEE80211_TX_RC_SHORT_GI;
2022
2023                         mcs_bw = mcs_flags & IEEE80211_RADIOTAP_MCS_BW_MASK;
2024                         if (mcs_known & IEEE80211_RADIOTAP_MCS_HAVE_BW &&
2025                             mcs_bw == IEEE80211_RADIOTAP_MCS_BW_40)
2026                                 rate_flags |= IEEE80211_TX_RC_40_MHZ_WIDTH;
2027                         break;
2028
2029                 case IEEE80211_RADIOTAP_VHT:
2030                         vht_known = get_unaligned_le16(iterator.this_arg);
2031                         rate_found = true;
2032
2033                         rate_flags = IEEE80211_TX_RC_VHT_MCS;
2034                         if ((vht_known & IEEE80211_RADIOTAP_VHT_KNOWN_GI) &&
2035                             (iterator.this_arg[2] &
2036                              IEEE80211_RADIOTAP_VHT_FLAG_SGI))
2037                                 rate_flags |= IEEE80211_TX_RC_SHORT_GI;
2038                         if (vht_known &
2039                             IEEE80211_RADIOTAP_VHT_KNOWN_BANDWIDTH) {
2040                                 if (iterator.this_arg[3] == 1)
2041                                         rate_flags |=
2042                                                 IEEE80211_TX_RC_40_MHZ_WIDTH;
2043                                 else if (iterator.this_arg[3] == 4)
2044                                         rate_flags |=
2045                                                 IEEE80211_TX_RC_80_MHZ_WIDTH;
2046                                 else if (iterator.this_arg[3] == 11)
2047                                         rate_flags |=
2048                                                 IEEE80211_TX_RC_160_MHZ_WIDTH;
2049                         }
2050
2051                         vht_mcs = iterator.this_arg[4] >> 4;
2052                         vht_nss = iterator.this_arg[4] & 0xF;
2053                         break;
2054
2055                 /*
2056                  * Please update the file
2057                  * Documentation/networking/mac80211-injection.txt
2058                  * when parsing new fields here.
2059                  */
2060
2061                 default:
2062                         break;
2063                 }
2064         }
2065
2066         if (ret != -ENOENT) /* ie, if we didn't simply run out of fields */
2067                 return false;
2068
2069         if (rate_found) {
2070                 info->control.flags |= IEEE80211_TX_CTRL_RATE_INJECT;
2071
2072                 for (i = 0; i < IEEE80211_TX_MAX_RATES; i++) {
2073                         info->control.rates[i].idx = -1;
2074                         info->control.rates[i].flags = 0;
2075                         info->control.rates[i].count = 0;
2076                 }
2077
2078                 if (rate_flags & IEEE80211_TX_RC_MCS) {
2079                         info->control.rates[0].idx = rate;
2080                 } else if (rate_flags & IEEE80211_TX_RC_VHT_MCS) {
2081                         ieee80211_rate_set_vht(info->control.rates, vht_mcs,
2082                                                vht_nss);
2083                 } else {
2084                         for (i = 0; i < sband->n_bitrates; i++) {
2085                                 if (rate * 5 != sband->bitrates[i].bitrate)
2086                                         continue;
2087
2088                                 info->control.rates[0].idx = i;
2089                                 break;
2090                         }
2091                 }
2092
2093                 if (info->control.rates[0].idx < 0)
2094                         info->control.flags &= ~IEEE80211_TX_CTRL_RATE_INJECT;
2095
2096                 info->control.rates[0].flags = rate_flags;
2097                 info->control.rates[0].count = min_t(u8, rate_retries + 1,
2098                                                      local->hw.max_rate_tries);
2099         }
2100
2101         /*
2102          * remove the radiotap header
2103          * iterator->_max_length was sanity-checked against
2104          * skb->len by iterator init
2105          */
2106         skb_pull(skb, iterator._max_length);
2107
2108         return true;
2109 }
2110
2111 netdev_tx_t ieee80211_monitor_start_xmit(struct sk_buff *skb,
2112                                          struct net_device *dev)
2113 {
2114         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
2115         struct ieee80211_chanctx_conf *chanctx_conf;
2116         struct ieee80211_radiotap_header *prthdr =
2117                 (struct ieee80211_radiotap_header *)skb->data;
2118         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
2119         struct ieee80211_hdr *hdr;
2120         struct ieee80211_sub_if_data *tmp_sdata, *sdata;
2121         struct cfg80211_chan_def *chandef;
2122         u16 len_rthdr;
2123         int hdrlen;
2124
2125         /* check for not even having the fixed radiotap header part */
2126         if (unlikely(skb->len < sizeof(struct ieee80211_radiotap_header)))
2127                 goto fail; /* too short to be possibly valid */
2128
2129         /* is it a header version we can trust to find length from? */
2130         if (unlikely(prthdr->it_version))
2131                 goto fail; /* only version 0 is supported */
2132
2133         /* then there must be a radiotap header with a length we can use */
2134         len_rthdr = ieee80211_get_radiotap_len(skb->data);
2135
2136         /* does the skb contain enough to deliver on the alleged length? */
2137         if (unlikely(skb->len < len_rthdr))
2138                 goto fail; /* skb too short for claimed rt header extent */
2139
2140         /*
2141          * fix up the pointers accounting for the radiotap
2142          * header still being in there.  We are being given
2143          * a precooked IEEE80211 header so no need for
2144          * normal processing
2145          */
2146         skb_set_mac_header(skb, len_rthdr);
2147         /*
2148          * these are just fixed to the end of the rt area since we
2149          * don't have any better information and at this point, nobody cares
2150          */
2151         skb_set_network_header(skb, len_rthdr);
2152         skb_set_transport_header(skb, len_rthdr);
2153
2154         if (skb->len < len_rthdr + 2)
2155                 goto fail;
2156
2157         hdr = (struct ieee80211_hdr *)(skb->data + len_rthdr);
2158         hdrlen = ieee80211_hdrlen(hdr->frame_control);
2159
2160         if (skb->len < len_rthdr + hdrlen)
2161                 goto fail;
2162
2163         /*
2164          * Initialize skb->protocol if the injected frame is a data frame
2165          * carrying a rfc1042 header
2166          */
2167         if (ieee80211_is_data(hdr->frame_control) &&
2168             skb->len >= len_rthdr + hdrlen + sizeof(rfc1042_header) + 2) {
2169                 u8 *payload = (u8 *)hdr + hdrlen;
2170
2171                 if (ether_addr_equal(payload, rfc1042_header))
2172                         skb->protocol = cpu_to_be16((payload[6] << 8) |
2173                                                     payload[7]);
2174         }
2175
2176         memset(info, 0, sizeof(*info));
2177
2178         info->flags = IEEE80211_TX_CTL_REQ_TX_STATUS |
2179                       IEEE80211_TX_CTL_INJECTED;
2180
2181         rcu_read_lock();
2182
2183         /*
2184          * We process outgoing injected frames that have a local address
2185          * we handle as though they are non-injected frames.
2186          * This code here isn't entirely correct, the local MAC address
2187          * isn't always enough to find the interface to use; for proper
2188          * VLAN/WDS support we will need a different mechanism (which
2189          * likely isn't going to be monitor interfaces).
2190          */
2191         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2192
2193         list_for_each_entry_rcu(tmp_sdata, &local->interfaces, list) {
2194                 if (!ieee80211_sdata_running(tmp_sdata))
2195                         continue;
2196                 if (tmp_sdata->vif.type == NL80211_IFTYPE_MONITOR ||
2197                     tmp_sdata->vif.type == NL80211_IFTYPE_AP_VLAN ||
2198                     tmp_sdata->vif.type == NL80211_IFTYPE_WDS)
2199                         continue;
2200                 if (ether_addr_equal(tmp_sdata->vif.addr, hdr->addr2)) {
2201                         sdata = tmp_sdata;
2202                         break;
2203                 }
2204         }
2205
2206         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2207         if (!chanctx_conf) {
2208                 tmp_sdata = rcu_dereference(local->monitor_sdata);
2209                 if (tmp_sdata)
2210                         chanctx_conf =
2211                                 rcu_dereference(tmp_sdata->vif.chanctx_conf);
2212         }
2213
2214         if (chanctx_conf)
2215                 chandef = &chanctx_conf->def;
2216         else if (!local->use_chanctx)
2217                 chandef = &local->_oper_chandef;
2218         else
2219                 goto fail_rcu;
2220
2221         /*
2222          * Frame injection is not allowed if beaconing is not allowed
2223          * or if we need radar detection. Beaconing is usually not allowed when
2224          * the mode or operation (Adhoc, AP, Mesh) does not support DFS.
2225          * Passive scan is also used in world regulatory domains where
2226          * your country is not known and as such it should be treated as
2227          * NO TX unless the channel is explicitly allowed in which case
2228          * your current regulatory domain would not have the passive scan
2229          * flag.
2230          *
2231          * Since AP mode uses monitor interfaces to inject/TX management
2232          * frames we can make AP mode the exception to this rule once it
2233          * supports radar detection as its implementation can deal with
2234          * radar detection by itself. We can do that later by adding a
2235          * monitor flag interfaces used for AP support.
2236          */
2237         if (!cfg80211_reg_can_beacon(local->hw.wiphy, chandef,
2238                                      sdata->vif.type))
2239                 goto fail_rcu;
2240
2241         info->band = chandef->chan->band;
2242
2243         /* process and remove the injection radiotap header */
2244         if (!ieee80211_parse_tx_radiotap(local, skb))
2245                 goto fail_rcu;
2246
2247         ieee80211_xmit(sdata, NULL, skb);
2248         rcu_read_unlock();
2249
2250         return NETDEV_TX_OK;
2251
2252 fail_rcu:
2253         rcu_read_unlock();
2254 fail:
2255         dev_kfree_skb(skb);
2256         return NETDEV_TX_OK; /* meaning, we dealt with the skb */
2257 }
2258
2259 static inline bool ieee80211_is_tdls_setup(struct sk_buff *skb)
2260 {
2261         u16 ethertype = (skb->data[12] << 8) | skb->data[13];
2262
2263         return ethertype == ETH_P_TDLS &&
2264                skb->len > 14 &&
2265                skb->data[14] == WLAN_TDLS_SNAP_RFTYPE;
2266 }
2267
2268 static int ieee80211_lookup_ra_sta(struct ieee80211_sub_if_data *sdata,
2269                                    struct sk_buff *skb,
2270                                    struct sta_info **sta_out)
2271 {
2272         struct sta_info *sta;
2273
2274         switch (sdata->vif.type) {
2275         case NL80211_IFTYPE_AP_VLAN:
2276                 sta = rcu_dereference(sdata->u.vlan.sta);
2277                 if (sta) {
2278                         *sta_out = sta;
2279                         return 0;
2280                 } else if (sdata->wdev.use_4addr) {
2281                         return -ENOLINK;
2282                 }
2283                 /* fall through */
2284         case NL80211_IFTYPE_AP:
2285         case NL80211_IFTYPE_OCB:
2286         case NL80211_IFTYPE_ADHOC:
2287                 if (is_multicast_ether_addr(skb->data)) {
2288                         *sta_out = ERR_PTR(-ENOENT);
2289                         return 0;
2290                 }
2291                 sta = sta_info_get_bss(sdata, skb->data);
2292                 break;
2293         case NL80211_IFTYPE_WDS:
2294                 sta = sta_info_get(sdata, sdata->u.wds.remote_addr);
2295                 break;
2296 #ifdef CONFIG_MAC80211_MESH
2297         case NL80211_IFTYPE_MESH_POINT:
2298                 /* determined much later */
2299                 *sta_out = NULL;
2300                 return 0;
2301 #endif
2302         case NL80211_IFTYPE_STATION:
2303                 if (sdata->wdev.wiphy->flags & WIPHY_FLAG_SUPPORTS_TDLS) {
2304                         sta = sta_info_get(sdata, skb->data);
2305                         if (sta && test_sta_flag(sta, WLAN_STA_TDLS_PEER)) {
2306                                 if (test_sta_flag(sta,
2307                                                   WLAN_STA_TDLS_PEER_AUTH)) {
2308                                         *sta_out = sta;
2309                                         return 0;
2310                                 }
2311
2312                                 /*
2313                                  * TDLS link during setup - throw out frames to
2314                                  * peer. Allow TDLS-setup frames to unauthorized
2315                                  * peers for the special case of a link teardown
2316                                  * after a TDLS sta is removed due to being
2317                                  * unreachable.
2318                                  */
2319                                 if (!ieee80211_is_tdls_setup(skb))
2320                                         return -EINVAL;
2321                         }
2322
2323                 }
2324
2325                 sta = sta_info_get(sdata, sdata->u.mgd.bssid);
2326                 if (!sta)
2327                         return -ENOLINK;
2328                 break;
2329         default:
2330                 return -EINVAL;
2331         }
2332
2333         *sta_out = sta ?: ERR_PTR(-ENOENT);
2334         return 0;
2335 }
2336
2337 /**
2338  * ieee80211_build_hdr - build 802.11 header in the given frame
2339  * @sdata: virtual interface to build the header for
2340  * @skb: the skb to build the header in
2341  * @info_flags: skb flags to set
2342  *
2343  * This function takes the skb with 802.3 header and reformats the header to
2344  * the appropriate IEEE 802.11 header based on which interface the packet is
2345  * being transmitted on.
2346  *
2347  * Note that this function also takes care of the TX status request and
2348  * potential unsharing of the SKB - this needs to be interleaved with the
2349  * header building.
2350  *
2351  * The function requires the read-side RCU lock held
2352  *
2353  * Returns: the (possibly reallocated) skb or an ERR_PTR() code
2354  */
2355 static struct sk_buff *ieee80211_build_hdr(struct ieee80211_sub_if_data *sdata,
2356                                            struct sk_buff *skb, u32 info_flags,
2357                                            struct sta_info *sta)
2358 {
2359         struct ieee80211_local *local = sdata->local;
2360         struct ieee80211_tx_info *info;
2361         int head_need;
2362         u16 ethertype, hdrlen,  meshhdrlen = 0;
2363         __le16 fc;
2364         struct ieee80211_hdr hdr;
2365         struct ieee80211s_hdr mesh_hdr __maybe_unused;
2366         struct mesh_path __maybe_unused *mppath = NULL, *mpath = NULL;
2367         const u8 *encaps_data;
2368         int encaps_len, skip_header_bytes;
2369         bool wme_sta = false, authorized = false;
2370         bool tdls_peer;
2371         bool multicast;
2372         u16 info_id = 0;
2373         struct ieee80211_chanctx_conf *chanctx_conf;
2374         struct ieee80211_sub_if_data *ap_sdata;
2375         enum nl80211_band band;
2376         int ret;
2377
2378         if (IS_ERR(sta))
2379                 sta = NULL;
2380
2381         /* convert Ethernet header to proper 802.11 header (based on
2382          * operation mode) */
2383         ethertype = (skb->data[12] << 8) | skb->data[13];
2384         fc = cpu_to_le16(IEEE80211_FTYPE_DATA | IEEE80211_STYPE_DATA);
2385
2386         switch (sdata->vif.type) {
2387         case NL80211_IFTYPE_AP_VLAN:
2388                 if (sdata->wdev.use_4addr) {
2389                         fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS);
2390                         /* RA TA DA SA */
2391                         memcpy(hdr.addr1, sta->sta.addr, ETH_ALEN);
2392                         memcpy(hdr.addr2, sdata->vif.addr, ETH_ALEN);
2393                         memcpy(hdr.addr3, skb->data, ETH_ALEN);
2394                         memcpy(hdr.addr4, skb->data + ETH_ALEN, ETH_ALEN);
2395                         hdrlen = 30;
2396                         authorized = test_sta_flag(sta, WLAN_STA_AUTHORIZED);
2397                         wme_sta = sta->sta.wme;
2398                 }
2399                 ap_sdata = container_of(sdata->bss, struct ieee80211_sub_if_data,
2400                                         u.ap);
2401                 chanctx_conf = rcu_dereference(ap_sdata->vif.chanctx_conf);
2402                 if (!chanctx_conf) {
2403                         ret = -ENOTCONN;
2404                         goto free;
2405                 }
2406                 band = chanctx_conf->def.chan->band;
2407                 if (sdata->wdev.use_4addr)
2408                         break;
2409                 /* fall through */
2410         case NL80211_IFTYPE_AP:
2411                 if (sdata->vif.type == NL80211_IFTYPE_AP)
2412                         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2413                 if (!chanctx_conf) {
2414                         ret = -ENOTCONN;
2415                         goto free;
2416                 }
2417                 fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS);
2418                 /* DA BSSID SA */
2419                 memcpy(hdr.addr1, skb->data, ETH_ALEN);
2420                 memcpy(hdr.addr2, sdata->vif.addr, ETH_ALEN);
2421                 memcpy(hdr.addr3, skb->data + ETH_ALEN, ETH_ALEN);
2422                 hdrlen = 24;
2423                 band = chanctx_conf->def.chan->band;
2424                 break;
2425         case NL80211_IFTYPE_WDS:
2426                 fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS);
2427                 /* RA TA DA SA */
2428                 memcpy(hdr.addr1, sdata->u.wds.remote_addr, ETH_ALEN);
2429                 memcpy(hdr.addr2, sdata->vif.addr, ETH_ALEN);
2430                 memcpy(hdr.addr3, skb->data, ETH_ALEN);
2431                 memcpy(hdr.addr4, skb->data + ETH_ALEN, ETH_ALEN);
2432                 hdrlen = 30;
2433                 /*
2434                  * This is the exception! WDS style interfaces are prohibited
2435                  * when channel contexts are in used so this must be valid
2436                  */
2437                 band = local->hw.conf.chandef.chan->band;
2438                 break;
2439 #ifdef CONFIG_MAC80211_MESH
2440         case NL80211_IFTYPE_MESH_POINT:
2441                 if (!is_multicast_ether_addr(skb->data)) {
2442                         struct sta_info *next_hop;
2443                         bool mpp_lookup = true;
2444
2445                         mpath = mesh_path_lookup(sdata, skb->data);
2446                         if (mpath) {
2447                                 mpp_lookup = false;
2448                                 next_hop = rcu_dereference(mpath->next_hop);
2449                                 if (!next_hop ||
2450                                     !(mpath->flags & (MESH_PATH_ACTIVE |
2451                                                       MESH_PATH_RESOLVING)))
2452                                         mpp_lookup = true;
2453                         }
2454
2455                         if (mpp_lookup) {
2456                                 mppath = mpp_path_lookup(sdata, skb->data);
2457                                 if (mppath)
2458                                         mppath->exp_time = jiffies;
2459                         }
2460
2461                         if (mppath && mpath)
2462                                 mesh_path_del(sdata, mpath->dst);
2463                 }
2464
2465                 /*
2466                  * Use address extension if it is a packet from
2467                  * another interface or if we know the destination
2468                  * is being proxied by a portal (i.e. portal address
2469                  * differs from proxied address)
2470                  */
2471                 if (ether_addr_equal(sdata->vif.addr, skb->data + ETH_ALEN) &&
2472                     !(mppath && !ether_addr_equal(mppath->mpp, skb->data))) {
2473                         hdrlen = ieee80211_fill_mesh_addresses(&hdr, &fc,
2474                                         skb->data, skb->data + ETH_ALEN);
2475                         meshhdrlen = ieee80211_new_mesh_header(sdata, &mesh_hdr,
2476                                                                NULL, NULL);
2477                 } else {
2478                         /* DS -> MBSS (802.11-2012 13.11.3.3).
2479                          * For unicast with unknown forwarding information,
2480                          * destination might be in the MBSS or if that fails
2481                          * forwarded to another mesh gate. In either case
2482                          * resolution will be handled in ieee80211_xmit(), so
2483                          * leave the original DA. This also works for mcast */
2484                         const u8 *mesh_da = skb->data;
2485
2486                         if (mppath)
2487                                 mesh_da = mppath->mpp;
2488                         else if (mpath)
2489                                 mesh_da = mpath->dst;
2490
2491                         hdrlen = ieee80211_fill_mesh_addresses(&hdr, &fc,
2492                                         mesh_da, sdata->vif.addr);
2493                         if (is_multicast_ether_addr(mesh_da))
2494                                 /* DA TA mSA AE:SA */
2495                                 meshhdrlen = ieee80211_new_mesh_header(
2496                                                 sdata, &mesh_hdr,
2497                                                 skb->data + ETH_ALEN, NULL);
2498                         else
2499                                 /* RA TA mDA mSA AE:DA SA */
2500                                 meshhdrlen = ieee80211_new_mesh_header(
2501                                                 sdata, &mesh_hdr, skb->data,
2502                                                 skb->data + ETH_ALEN);
2503
2504                 }
2505                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2506                 if (!chanctx_conf) {
2507                         ret = -ENOTCONN;
2508                         goto free;
2509                 }
2510                 band = chanctx_conf->def.chan->band;
2511                 break;
2512 #endif
2513         case NL80211_IFTYPE_STATION:
2514                 /* we already did checks when looking up the RA STA */
2515                 tdls_peer = test_sta_flag(sta, WLAN_STA_TDLS_PEER);
2516
2517                 if (tdls_peer) {
2518                         /* DA SA BSSID */
2519                         memcpy(hdr.addr1, skb->data, ETH_ALEN);
2520                         memcpy(hdr.addr2, skb->data + ETH_ALEN, ETH_ALEN);
2521                         memcpy(hdr.addr3, sdata->u.mgd.bssid, ETH_ALEN);
2522                         hdrlen = 24;
2523                 }  else if (sdata->u.mgd.use_4addr &&
2524                             cpu_to_be16(ethertype) != sdata->control_port_protocol) {
2525                         fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS |
2526                                           IEEE80211_FCTL_TODS);
2527                         /* RA TA DA SA */
2528                         memcpy(hdr.addr1, sdata->u.mgd.bssid, ETH_ALEN);
2529                         memcpy(hdr.addr2, sdata->vif.addr, ETH_ALEN);
2530                         memcpy(hdr.addr3, skb->data, ETH_ALEN);
2531                         memcpy(hdr.addr4, skb->data + ETH_ALEN, ETH_ALEN);
2532                         hdrlen = 30;
2533                 } else {
2534                         fc |= cpu_to_le16(IEEE80211_FCTL_TODS);
2535                         /* BSSID SA DA */
2536                         memcpy(hdr.addr1, sdata->u.mgd.bssid, ETH_ALEN);
2537                         memcpy(hdr.addr2, skb->data + ETH_ALEN, ETH_ALEN);
2538                         memcpy(hdr.addr3, skb->data, ETH_ALEN);
2539                         hdrlen = 24;
2540                 }
2541                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2542                 if (!chanctx_conf) {
2543                         ret = -ENOTCONN;
2544                         goto free;
2545                 }
2546                 band = chanctx_conf->def.chan->band;
2547                 break;
2548         case NL80211_IFTYPE_OCB:
2549                 /* DA SA BSSID */
2550                 memcpy(hdr.addr1, skb->data, ETH_ALEN);
2551                 memcpy(hdr.addr2, skb->data + ETH_ALEN, ETH_ALEN);
2552                 eth_broadcast_addr(hdr.addr3);
2553                 hdrlen = 24;
2554                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2555                 if (!chanctx_conf) {
2556                         ret = -ENOTCONN;
2557                         goto free;
2558                 }
2559                 band = chanctx_conf->def.chan->band;
2560                 break;
2561         case NL80211_IFTYPE_ADHOC:
2562                 /* DA SA BSSID */
2563                 memcpy(hdr.addr1, skb->data, ETH_ALEN);
2564                 memcpy(hdr.addr2, skb->data + ETH_ALEN, ETH_ALEN);
2565                 memcpy(hdr.addr3, sdata->u.ibss.bssid, ETH_ALEN);
2566                 hdrlen = 24;
2567                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2568                 if (!chanctx_conf) {
2569                         ret = -ENOTCONN;
2570                         goto free;
2571                 }
2572                 band = chanctx_conf->def.chan->band;
2573                 break;
2574         default:
2575                 ret = -EINVAL;
2576                 goto free;
2577         }
2578
2579         multicast = is_multicast_ether_addr(hdr.addr1);
2580
2581         /* sta is always NULL for mesh */
2582         if (sta) {
2583                 authorized = test_sta_flag(sta, WLAN_STA_AUTHORIZED);
2584                 wme_sta = sta->sta.wme;
2585         } else if (ieee80211_vif_is_mesh(&sdata->vif)) {
2586                 /* For mesh, the use of the QoS header is mandatory */
2587                 wme_sta = true;
2588         }
2589
2590         /* receiver does QoS (which also means we do) use it */
2591         if (wme_sta) {
2592                 fc |= cpu_to_le16(IEEE80211_STYPE_QOS_DATA);
2593                 hdrlen += 2;
2594         }
2595
2596         /*
2597          * Drop unicast frames to unauthorised stations unless they are
2598          * EAPOL frames from the local station.
2599          */
2600         if (unlikely(!ieee80211_vif_is_mesh(&sdata->vif) &&
2601                      (sdata->vif.type != NL80211_IFTYPE_OCB) &&
2602                      !multicast && !authorized &&
2603                      (cpu_to_be16(ethertype) != sdata->control_port_protocol ||
2604                       !ether_addr_equal(sdata->vif.addr, skb->data + ETH_ALEN)))) {
2605 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
2606                 net_info_ratelimited("%s: dropped frame to %pM (unauthorized port)\n",
2607                                     sdata->name, hdr.addr1);
2608 #endif
2609
2610                 I802_DEBUG_INC(local->tx_handlers_drop_unauth_port);
2611
2612                 ret = -EPERM;
2613                 goto free;
2614         }
2615
2616         if (unlikely(!multicast && skb->sk &&
2617                      skb_shinfo(skb)->tx_flags & SKBTX_WIFI_STATUS)) {
2618                 struct sk_buff *ack_skb = skb_clone_sk(skb);
2619
2620                 if (ack_skb) {
2621                         unsigned long flags;
2622                         int id;
2623
2624                         spin_lock_irqsave(&local->ack_status_lock, flags);
2625                         id = idr_alloc(&local->ack_status_frames, ack_skb,
2626                                        1, 0x10000, GFP_ATOMIC);
2627                         spin_unlock_irqrestore(&local->ack_status_lock, flags);
2628
2629                         if (id >= 0) {
2630                                 info_id = id;
2631                                 info_flags |= IEEE80211_TX_CTL_REQ_TX_STATUS;
2632                         } else {
2633                                 kfree_skb(ack_skb);
2634                         }
2635                 }
2636         }
2637
2638         /*
2639          * If the skb is shared we need to obtain our own copy.
2640          */
2641         if (skb_shared(skb)) {
2642                 struct sk_buff *tmp_skb = skb;
2643
2644                 /* can't happen -- skb is a clone if info_id != 0 */
2645                 WARN_ON(info_id);
2646
2647                 skb = skb_clone(skb, GFP_ATOMIC);
2648                 kfree_skb(tmp_skb);
2649
2650                 if (!skb) {
2651                         ret = -ENOMEM;
2652                         goto free;
2653                 }
2654         }
2655
2656         hdr.frame_control = fc;
2657         hdr.duration_id = 0;
2658         hdr.seq_ctrl = 0;
2659
2660         skip_header_bytes = ETH_HLEN;
2661         if (ethertype == ETH_P_AARP || ethertype == ETH_P_IPX) {
2662                 encaps_data = bridge_tunnel_header;
2663                 encaps_len = sizeof(bridge_tunnel_header);
2664                 skip_header_bytes -= 2;
2665         } else if (ethertype >= ETH_P_802_3_MIN) {
2666                 encaps_data = rfc1042_header;
2667                 encaps_len = sizeof(rfc1042_header);
2668                 skip_header_bytes -= 2;
2669         } else {
2670                 encaps_data = NULL;
2671                 encaps_len = 0;
2672         }
2673
2674         skb_pull(skb, skip_header_bytes);
2675         head_need = hdrlen + encaps_len + meshhdrlen - skb_headroom(skb);
2676
2677         /*
2678          * So we need to modify the skb header and hence need a copy of
2679          * that. The head_need variable above doesn't, so far, include
2680          * the needed header space that we don't need right away. If we
2681          * can, then we don't reallocate right now but only after the
2682          * frame arrives at the master device (if it does...)
2683          *
2684          * If we cannot, however, then we will reallocate to include all
2685          * the ever needed space. Also, if we need to reallocate it anyway,
2686          * make it big enough for everything we may ever need.
2687          */
2688
2689         if (head_need > 0 || skb_cloned(skb)) {
2690                 head_need += sdata->encrypt_headroom;
2691                 head_need += local->tx_headroom;
2692                 head_need = max_t(int, 0, head_need);
2693                 if (ieee80211_skb_resize(sdata, skb, head_need, true)) {
2694                         ieee80211_free_txskb(&local->hw, skb);
2695                         skb = NULL;
2696                         return ERR_PTR(-ENOMEM);
2697                 }
2698         }
2699
2700         if (encaps_data)
2701                 memcpy(skb_push(skb, encaps_len), encaps_data, encaps_len);
2702
2703 #ifdef CONFIG_MAC80211_MESH
2704         if (meshhdrlen > 0)
2705                 memcpy(skb_push(skb, meshhdrlen), &mesh_hdr, meshhdrlen);
2706 #endif
2707
2708         if (ieee80211_is_data_qos(fc)) {
2709                 __le16 *qos_control;
2710
2711                 qos_control = skb_push(skb, 2);
2712                 memcpy(skb_push(skb, hdrlen - 2), &hdr, hdrlen - 2);
2713                 /*
2714                  * Maybe we could actually set some fields here, for now just
2715                  * initialise to zero to indicate no special operation.
2716                  */
2717                 *qos_control = 0;
2718         } else
2719                 memcpy(skb_push(skb, hdrlen), &hdr, hdrlen);
2720
2721         skb_reset_mac_header(skb);
2722
2723         info = IEEE80211_SKB_CB(skb);
2724         memset(info, 0, sizeof(*info));
2725
2726         info->flags = info_flags;
2727         info->ack_frame_id = info_id;
2728         info->band = band;
2729
2730         return skb;
2731  free:
2732         kfree_skb(skb);
2733         return ERR_PTR(ret);
2734 }
2735
2736 /*
2737  * fast-xmit overview
2738  *
2739  * The core idea of this fast-xmit is to remove per-packet checks by checking
2740  * them out of band. ieee80211_check_fast_xmit() implements the out-of-band
2741  * checks that are needed to get the sta->fast_tx pointer assigned, after which
2742  * much less work can be done per packet. For example, fragmentation must be
2743  * disabled or the fast_tx pointer will not be set. All the conditions are seen
2744  * in the code here.
2745  *
2746  * Once assigned, the fast_tx data structure also caches the per-packet 802.11
2747  * header and other data to aid packet processing in ieee80211_xmit_fast().
2748  *
2749  * The most difficult part of this is that when any of these assumptions
2750  * change, an external trigger (i.e. a call to ieee80211_clear_fast_xmit(),
2751  * ieee80211_check_fast_xmit() or friends) is required to reset the data,
2752  * since the per-packet code no longer checks the conditions. This is reflected
2753  * by the calls to these functions throughout the rest of the code, and must be
2754  * maintained if any of the TX path checks change.
2755  */
2756
2757 void ieee80211_check_fast_xmit(struct sta_info *sta)
2758 {
2759         struct ieee80211_fast_tx build = {}, *fast_tx = NULL, *old;
2760         struct ieee80211_local *local = sta->local;
2761         struct ieee80211_sub_if_data *sdata = sta->sdata;
2762         struct ieee80211_hdr *hdr = (void *)build.hdr;
2763         struct ieee80211_chanctx_conf *chanctx_conf;
2764         __le16 fc;
2765
2766         if (!ieee80211_hw_check(&local->hw, SUPPORT_FAST_XMIT))
2767                 return;
2768
2769         /* Locking here protects both the pointer itself, and against concurrent
2770          * invocations winning data access races to, e.g., the key pointer that
2771          * is used.
2772          * Without it, the invocation of this function right after the key
2773          * pointer changes wouldn't be sufficient, as another CPU could access
2774          * the pointer, then stall, and then do the cache update after the CPU
2775          * that invalidated the key.
2776          * With the locking, such scenarios cannot happen as the check for the
2777          * key and the fast-tx assignment are done atomically, so the CPU that
2778          * modifies the key will either wait or other one will see the key
2779          * cleared/changed already.
2780          */
2781         spin_lock_bh(&sta->lock);
2782         if (ieee80211_hw_check(&local->hw, SUPPORTS_PS) &&
2783             !ieee80211_hw_check(&local->hw, SUPPORTS_DYNAMIC_PS) &&
2784             sdata->vif.type == NL80211_IFTYPE_STATION)
2785                 goto out;
2786
2787         if (!test_sta_flag(sta, WLAN_STA_AUTHORIZED))
2788                 goto out;
2789
2790         if (test_sta_flag(sta, WLAN_STA_PS_STA) ||
2791             test_sta_flag(sta, WLAN_STA_PS_DRIVER) ||
2792             test_sta_flag(sta, WLAN_STA_PS_DELIVER) ||
2793             test_sta_flag(sta, WLAN_STA_CLEAR_PS_FILT))
2794                 goto out;
2795
2796         if (sdata->noack_map)
2797                 goto out;
2798
2799         /* fast-xmit doesn't handle fragmentation at all */
2800         if (local->hw.wiphy->frag_threshold != (u32)-1 &&
2801             !ieee80211_hw_check(&local->hw, SUPPORTS_TX_FRAG))
2802                 goto out;
2803
2804         rcu_read_lock();
2805         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2806         if (!chanctx_conf) {
2807                 rcu_read_unlock();
2808                 goto out;
2809         }
2810         build.band = chanctx_conf->def.chan->band;
2811         rcu_read_unlock();
2812
2813         fc = cpu_to_le16(IEEE80211_FTYPE_DATA | IEEE80211_STYPE_DATA);
2814
2815         switch (sdata->vif.type) {
2816         case NL80211_IFTYPE_ADHOC:
2817                 /* DA SA BSSID */
2818                 build.da_offs = offsetof(struct ieee80211_hdr, addr1);
2819                 build.sa_offs = offsetof(struct ieee80211_hdr, addr2);
2820                 memcpy(hdr->addr3, sdata->u.ibss.bssid, ETH_ALEN);
2821                 build.hdr_len = 24;
2822                 break;
2823         case NL80211_IFTYPE_STATION:
2824                 if (test_sta_flag(sta, WLAN_STA_TDLS_PEER)) {
2825                         /* DA SA BSSID */
2826                         build.da_offs = offsetof(struct ieee80211_hdr, addr1);
2827                         build.sa_offs = offsetof(struct ieee80211_hdr, addr2);
2828                         memcpy(hdr->addr3, sdata->u.mgd.bssid, ETH_ALEN);
2829                         build.hdr_len = 24;
2830                         break;
2831                 }
2832
2833                 if (sdata->u.mgd.use_4addr) {
2834                         /* non-regular ethertype cannot use the fastpath */
2835                         fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS |
2836                                           IEEE80211_FCTL_TODS);
2837                         /* RA TA DA SA */
2838                         memcpy(hdr->addr1, sdata->u.mgd.bssid, ETH_ALEN);
2839                         memcpy(hdr->addr2, sdata->vif.addr, ETH_ALEN);
2840                         build.da_offs = offsetof(struct ieee80211_hdr, addr3);
2841                         build.sa_offs = offsetof(struct ieee80211_hdr, addr4);
2842                         build.hdr_len = 30;
2843                         break;
2844                 }
2845                 fc |= cpu_to_le16(IEEE80211_FCTL_TODS);
2846                 /* BSSID SA DA */
2847                 memcpy(hdr->addr1, sdata->u.mgd.bssid, ETH_ALEN);
2848                 build.da_offs = offsetof(struct ieee80211_hdr, addr3);
2849                 build.sa_offs = offsetof(struct ieee80211_hdr, addr2);
2850                 build.hdr_len = 24;
2851                 break;
2852         case NL80211_IFTYPE_AP_VLAN:
2853                 if (sdata->wdev.use_4addr) {
2854                         fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS |
2855                                           IEEE80211_FCTL_TODS);
2856                         /* RA TA DA SA */
2857                         memcpy(hdr->addr1, sta->sta.addr, ETH_ALEN);
2858                         memcpy(hdr->addr2, sdata->vif.addr, ETH_ALEN);
2859                         build.da_offs = offsetof(struct ieee80211_hdr, addr3);
2860                         build.sa_offs = offsetof(struct ieee80211_hdr, addr4);
2861                         build.hdr_len = 30;
2862                         break;
2863                 }
2864                 /* fall through */
2865         case NL80211_IFTYPE_AP:
2866                 fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS);
2867                 /* DA BSSID SA */
2868                 build.da_offs = offsetof(struct ieee80211_hdr, addr1);
2869                 memcpy(hdr->addr2, sdata->vif.addr, ETH_ALEN);
2870                 build.sa_offs = offsetof(struct ieee80211_hdr, addr3);
2871                 build.hdr_len = 24;
2872                 break;
2873         default:
2874                 /* not handled on fast-xmit */
2875                 goto out;
2876         }
2877
2878         if (sta->sta.wme) {
2879                 build.hdr_len += 2;
2880                 fc |= cpu_to_le16(IEEE80211_STYPE_QOS_DATA);
2881         }
2882
2883         /* We store the key here so there's no point in using rcu_dereference()
2884          * but that's fine because the code that changes the pointers will call
2885          * this function after doing so. For a single CPU that would be enough,
2886          * for multiple see the comment above.
2887          */
2888         build.key = rcu_access_pointer(sta->ptk[sta->ptk_idx]);
2889         if (!build.key)
2890                 build.key = rcu_access_pointer(sdata->default_unicast_key);
2891         if (build.key) {
2892                 bool gen_iv, iv_spc, mmic;
2893
2894                 gen_iv = build.key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV;
2895                 iv_spc = build.key->conf.flags & IEEE80211_KEY_FLAG_PUT_IV_SPACE;
2896                 mmic = build.key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_MMIC;
2897
2898                 /* don't handle software crypto */
2899                 if (!(build.key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE))
2900                         goto out;
2901
2902                 switch (build.key->conf.cipher) {
2903                 case WLAN_CIPHER_SUITE_CCMP:
2904                 case WLAN_CIPHER_SUITE_CCMP_256:
2905                         /* add fixed key ID */
2906                         if (gen_iv) {
2907                                 (build.hdr + build.hdr_len)[3] =
2908                                         0x20 | (build.key->conf.keyidx << 6);
2909                                 build.pn_offs = build.hdr_len;
2910                         }
2911                         if (gen_iv || iv_spc)
2912                                 build.hdr_len += IEEE80211_CCMP_HDR_LEN;
2913                         break;
2914                 case WLAN_CIPHER_SUITE_GCMP:
2915                 case WLAN_CIPHER_SUITE_GCMP_256:
2916                         /* add fixed key ID */
2917                         if (gen_iv) {
2918                                 (build.hdr + build.hdr_len)[3] =
2919                                         0x20 | (build.key->conf.keyidx << 6);
2920                                 build.pn_offs = build.hdr_len;
2921                         }
2922                         if (gen_iv || iv_spc)
2923                                 build.hdr_len += IEEE80211_GCMP_HDR_LEN;
2924                         break;
2925                 case WLAN_CIPHER_SUITE_TKIP:
2926                         /* cannot handle MMIC or IV generation in xmit-fast */
2927                         if (mmic || gen_iv)
2928                                 goto out;
2929                         if (iv_spc)
2930                                 build.hdr_len += IEEE80211_TKIP_IV_LEN;
2931                         break;
2932                 case WLAN_CIPHER_SUITE_WEP40:
2933                 case WLAN_CIPHER_SUITE_WEP104:
2934                         /* cannot handle IV generation in fast-xmit */
2935                         if (gen_iv)
2936                                 goto out;
2937                         if (iv_spc)
2938                                 build.hdr_len += IEEE80211_WEP_IV_LEN;
2939                         break;
2940                 case WLAN_CIPHER_SUITE_AES_CMAC:
2941                 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
2942                 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
2943                 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
2944                         WARN(1,
2945                              "management cipher suite 0x%x enabled for data\n",
2946                              build.key->conf.cipher);
2947                         goto out;
2948                 default:
2949                         /* we don't know how to generate IVs for this at all */
2950                         if (WARN_ON(gen_iv))
2951                                 goto out;
2952                         /* pure hardware keys are OK, of course */
2953                         if (!(build.key->flags & KEY_FLAG_CIPHER_SCHEME))
2954                                 break;
2955                         /* cipher scheme might require space allocation */
2956                         if (iv_spc &&
2957                             build.key->conf.iv_len > IEEE80211_FAST_XMIT_MAX_IV)
2958                                 goto out;
2959                         if (iv_spc)
2960                                 build.hdr_len += build.key->conf.iv_len;
2961                 }
2962
2963                 fc |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
2964         }
2965
2966         hdr->frame_control = fc;
2967
2968         memcpy(build.hdr + build.hdr_len,
2969                rfc1042_header,  sizeof(rfc1042_header));
2970         build.hdr_len += sizeof(rfc1042_header);
2971
2972         fast_tx = kmemdup(&build, sizeof(build), GFP_ATOMIC);
2973         /* if the kmemdup fails, continue w/o fast_tx */
2974         if (!fast_tx)
2975                 goto out;
2976
2977  out:
2978         /* we might have raced against another call to this function */
2979         old = rcu_dereference_protected(sta->fast_tx,
2980                                         lockdep_is_held(&sta->lock));
2981         rcu_assign_pointer(sta->fast_tx, fast_tx);
2982         if (old)
2983                 kfree_rcu(old, rcu_head);
2984         spin_unlock_bh(&sta->lock);
2985 }
2986
2987 void ieee80211_check_fast_xmit_all(struct ieee80211_local *local)
2988 {
2989         struct sta_info *sta;
2990
2991         rcu_read_lock();
2992         list_for_each_entry_rcu(sta, &local->sta_list, list)
2993                 ieee80211_check_fast_xmit(sta);
2994         rcu_read_unlock();
2995 }
2996
2997 void ieee80211_check_fast_xmit_iface(struct ieee80211_sub_if_data *sdata)
2998 {
2999         struct ieee80211_local *local = sdata->local;
3000         struct sta_info *sta;
3001
3002         rcu_read_lock();
3003
3004         list_for_each_entry_rcu(sta, &local->sta_list, list) {
3005                 if (sdata != sta->sdata &&
3006                     (!sta->sdata->bss || sta->sdata->bss != sdata->bss))
3007                         continue;
3008                 ieee80211_check_fast_xmit(sta);
3009         }
3010
3011         rcu_read_unlock();
3012 }
3013
3014 void ieee80211_clear_fast_xmit(struct sta_info *sta)
3015 {
3016         struct ieee80211_fast_tx *fast_tx;
3017
3018         spin_lock_bh(&sta->lock);
3019         fast_tx = rcu_dereference_protected(sta->fast_tx,
3020                                             lockdep_is_held(&sta->lock));
3021         RCU_INIT_POINTER(sta->fast_tx, NULL);
3022         spin_unlock_bh(&sta->lock);
3023
3024         if (fast_tx)
3025                 kfree_rcu(fast_tx, rcu_head);
3026 }
3027
3028 static bool ieee80211_amsdu_realloc_pad(struct ieee80211_local *local,
3029                                         struct sk_buff *skb, int headroom,
3030                                         int *subframe_len)
3031 {
3032         int amsdu_len = *subframe_len + sizeof(struct ethhdr);
3033         int padding = (4 - amsdu_len) & 3;
3034
3035         if (skb_headroom(skb) < headroom || skb_tailroom(skb) < padding) {
3036                 I802_DEBUG_INC(local->tx_expand_skb_head);
3037
3038                 if (pskb_expand_head(skb, headroom, padding, GFP_ATOMIC)) {
3039                         wiphy_debug(local->hw.wiphy,
3040                                     "failed to reallocate TX buffer\n");
3041                         return false;
3042                 }
3043         }
3044
3045         if (padding) {
3046                 *subframe_len += padding;
3047                 skb_put_zero(skb, padding);
3048         }
3049
3050         return true;
3051 }
3052
3053 static bool ieee80211_amsdu_prepare_head(struct ieee80211_sub_if_data *sdata,
3054                                          struct ieee80211_fast_tx *fast_tx,
3055                                          struct sk_buff *skb)
3056 {
3057         struct ieee80211_local *local = sdata->local;
3058         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
3059         struct ieee80211_hdr *hdr;
3060         struct ethhdr *amsdu_hdr;
3061         int hdr_len = fast_tx->hdr_len - sizeof(rfc1042_header);
3062         int subframe_len = skb->len - hdr_len;
3063         void *data;
3064         u8 *qc, *h_80211_src, *h_80211_dst;
3065         const u8 *bssid;
3066
3067         if (info->flags & IEEE80211_TX_CTL_RATE_CTRL_PROBE)
3068                 return false;
3069
3070         if (info->control.flags & IEEE80211_TX_CTRL_AMSDU)
3071                 return true;
3072
3073         if (!ieee80211_amsdu_realloc_pad(local, skb, sizeof(*amsdu_hdr),
3074                                          &subframe_len))
3075                 return false;
3076
3077         data = skb_push(skb, sizeof(*amsdu_hdr));
3078         memmove(data, data + sizeof(*amsdu_hdr), hdr_len);
3079         hdr = data;
3080         amsdu_hdr = data + hdr_len;
3081         /* h_80211_src/dst is addr* field within hdr */
3082         h_80211_src = data + fast_tx->sa_offs;
3083         h_80211_dst = data + fast_tx->da_offs;
3084
3085         amsdu_hdr->h_proto = cpu_to_be16(subframe_len);
3086         ether_addr_copy(amsdu_hdr->h_source, h_80211_src);
3087         ether_addr_copy(amsdu_hdr->h_dest, h_80211_dst);
3088
3089         /* according to IEEE 802.11-2012 8.3.2 table 8-19, the outer SA/DA
3090          * fields needs to be changed to BSSID for A-MSDU frames depending
3091          * on FromDS/ToDS values.
3092          */
3093         switch (sdata->vif.type) {
3094         case NL80211_IFTYPE_STATION:
3095                 bssid = sdata->u.mgd.bssid;
3096                 break;
3097         case NL80211_IFTYPE_AP:
3098         case NL80211_IFTYPE_AP_VLAN:
3099                 bssid = sdata->vif.addr;
3100                 break;
3101         default:
3102                 bssid = NULL;
3103         }
3104
3105         if (bssid && ieee80211_has_fromds(hdr->frame_control))
3106                 ether_addr_copy(h_80211_src, bssid);
3107
3108         if (bssid && ieee80211_has_tods(hdr->frame_control))
3109                 ether_addr_copy(h_80211_dst, bssid);
3110
3111         qc = ieee80211_get_qos_ctl(hdr);
3112         *qc |= IEEE80211_QOS_CTL_A_MSDU_PRESENT;
3113
3114         info->control.flags |= IEEE80211_TX_CTRL_AMSDU;
3115
3116         return true;
3117 }
3118
3119 static bool ieee80211_amsdu_aggregate(struct ieee80211_sub_if_data *sdata,
3120                                       struct sta_info *sta,
3121                                       struct ieee80211_fast_tx *fast_tx,
3122                                       struct sk_buff *skb)
3123 {
3124         struct ieee80211_local *local = sdata->local;
3125         struct fq *fq = &local->fq;
3126         struct fq_tin *tin;
3127         struct fq_flow *flow;
3128         u8 tid = skb->priority & IEEE80211_QOS_CTL_TAG1D_MASK;
3129         struct ieee80211_txq *txq = sta->sta.txq[tid];
3130         struct txq_info *txqi;
3131         struct sk_buff **frag_tail, *head;
3132         int subframe_len = skb->len - ETH_ALEN;
3133         u8 max_subframes = sta->sta.max_amsdu_subframes;
3134         int max_frags = local->hw.max_tx_fragments;
3135         int max_amsdu_len = sta->sta.max_amsdu_len;
3136         __be16 len;
3137         void *data;
3138         bool ret = false;
3139         unsigned int orig_len;
3140         int n = 1, nfrags;
3141
3142         if (!ieee80211_hw_check(&local->hw, TX_AMSDU))
3143                 return false;
3144
3145         if (!txq)
3146                 return false;
3147
3148         txqi = to_txq_info(txq);
3149         if (test_bit(IEEE80211_TXQ_NO_AMSDU, &txqi->flags))
3150                 return false;
3151
3152         if (sta->sta.max_rc_amsdu_len)
3153                 max_amsdu_len = min_t(int, max_amsdu_len,
3154                                       sta->sta.max_rc_amsdu_len);
3155
3156         spin_lock_bh(&fq->lock);
3157
3158         /* TODO: Ideally aggregation should be done on dequeue to remain
3159          * responsive to environment changes.
3160          */
3161
3162         tin = &txqi->tin;
3163         flow = fq_flow_classify(fq, tin, skb, fq_flow_get_default_func);
3164         head = skb_peek_tail(&flow->queue);
3165         if (!head)
3166                 goto out;
3167
3168         orig_len = head->len;
3169
3170         if (skb->len + head->len > max_amsdu_len)
3171                 goto out;
3172
3173         if (!ieee80211_amsdu_prepare_head(sdata, fast_tx, head))
3174                 goto out;
3175
3176         nfrags = 1 + skb_shinfo(skb)->nr_frags;
3177         nfrags += 1 + skb_shinfo(head)->nr_frags;
3178         frag_tail = &skb_shinfo(head)->frag_list;
3179         while (*frag_tail) {
3180                 nfrags += 1 + skb_shinfo(*frag_tail)->nr_frags;
3181                 frag_tail = &(*frag_tail)->next;
3182                 n++;
3183         }
3184
3185         if (max_subframes && n > max_subframes)
3186                 goto out;
3187
3188         if (max_frags && nfrags > max_frags)
3189                 goto out;
3190
3191         if (!ieee80211_amsdu_realloc_pad(local, skb, sizeof(rfc1042_header) + 2,
3192                                          &subframe_len))
3193                 goto out;
3194
3195         ret = true;
3196         data = skb_push(skb, ETH_ALEN + 2);
3197         memmove(data, data + ETH_ALEN + 2, 2 * ETH_ALEN);
3198
3199         data += 2 * ETH_ALEN;
3200         len = cpu_to_be16(subframe_len);
3201         memcpy(data, &len, 2);
3202         memcpy(data + 2, rfc1042_header, sizeof(rfc1042_header));
3203
3204         head->len += skb->len;
3205         head->data_len += skb->len;
3206         *frag_tail = skb;
3207
3208         flow->backlog += head->len - orig_len;
3209         tin->backlog_bytes += head->len - orig_len;
3210
3211         fq_recalc_backlog(fq, tin, flow);
3212
3213 out:
3214         spin_unlock_bh(&fq->lock);
3215
3216         return ret;
3217 }
3218
3219 /*
3220  * Can be called while the sta lock is held. Anything that can cause packets to
3221  * be generated will cause deadlock!
3222  */
3223 static void ieee80211_xmit_fast_finish(struct ieee80211_sub_if_data *sdata,
3224                                        struct sta_info *sta, u8 pn_offs,
3225                                        struct ieee80211_key *key,
3226                                        struct sk_buff *skb)
3227 {
3228         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
3229         struct ieee80211_hdr *hdr = (void *)skb->data;
3230         u8 tid = IEEE80211_NUM_TIDS;
3231
3232         if (key)
3233                 info->control.hw_key = &key->conf;
3234
3235         ieee80211_tx_stats(skb->dev, skb->len);
3236
3237         if (hdr->frame_control & cpu_to_le16(IEEE80211_STYPE_QOS_DATA)) {
3238                 tid = skb->priority & IEEE80211_QOS_CTL_TAG1D_MASK;
3239                 hdr->seq_ctrl = ieee80211_tx_next_seq(sta, tid);
3240         } else {
3241                 info->flags |= IEEE80211_TX_CTL_ASSIGN_SEQ;
3242                 hdr->seq_ctrl = cpu_to_le16(sdata->sequence_number);
3243                 sdata->sequence_number += 0x10;
3244         }
3245
3246         if (skb_shinfo(skb)->gso_size)
3247                 sta->tx_stats.msdu[tid] +=
3248                         DIV_ROUND_UP(skb->len, skb_shinfo(skb)->gso_size);
3249         else
3250                 sta->tx_stats.msdu[tid]++;
3251
3252         info->hw_queue = sdata->vif.hw_queue[skb_get_queue_mapping(skb)];
3253
3254         /* statistics normally done by ieee80211_tx_h_stats (but that
3255          * has to consider fragmentation, so is more complex)
3256          */
3257         sta->tx_stats.bytes[skb_get_queue_mapping(skb)] += skb->len;
3258         sta->tx_stats.packets[skb_get_queue_mapping(skb)]++;
3259
3260         if (pn_offs) {
3261                 u64 pn;
3262                 u8 *crypto_hdr = skb->data + pn_offs;
3263
3264                 switch (key->conf.cipher) {
3265                 case WLAN_CIPHER_SUITE_CCMP:
3266                 case WLAN_CIPHER_SUITE_CCMP_256:
3267                 case WLAN_CIPHER_SUITE_GCMP:
3268                 case WLAN_CIPHER_SUITE_GCMP_256:
3269                         pn = atomic64_inc_return(&key->conf.tx_pn);
3270                         crypto_hdr[0] = pn;
3271                         crypto_hdr[1] = pn >> 8;
3272                         crypto_hdr[4] = pn >> 16;
3273                         crypto_hdr[5] = pn >> 24;
3274                         crypto_hdr[6] = pn >> 32;
3275                         crypto_hdr[7] = pn >> 40;
3276                         break;
3277                 }
3278         }
3279 }
3280
3281 static bool ieee80211_xmit_fast(struct ieee80211_sub_if_data *sdata,
3282                                 struct sta_info *sta,
3283                                 struct ieee80211_fast_tx *fast_tx,
3284                                 struct sk_buff *skb)
3285 {
3286         struct ieee80211_local *local = sdata->local;
3287         u16 ethertype = (skb->data[12] << 8) | skb->data[13];
3288         int extra_head = fast_tx->hdr_len - (ETH_HLEN - 2);
3289         int hw_headroom = sdata->local->hw.extra_tx_headroom;
3290         struct ethhdr eth;
3291         struct ieee80211_tx_info *info;
3292         struct ieee80211_hdr *hdr = (void *)fast_tx->hdr;
3293         struct ieee80211_tx_data tx;
3294         ieee80211_tx_result r;
3295         struct tid_ampdu_tx *tid_tx = NULL;
3296         u8 tid = IEEE80211_NUM_TIDS;
3297
3298         /* control port protocol needs a lot of special handling */
3299         if (cpu_to_be16(ethertype) == sdata->control_port_protocol)
3300                 return false;
3301
3302         /* only RFC 1042 SNAP */
3303         if (ethertype < ETH_P_802_3_MIN)
3304                 return false;
3305
3306         /* don't handle TX status request here either */
3307         if (skb->sk && skb_shinfo(skb)->tx_flags & SKBTX_WIFI_STATUS)
3308                 return false;
3309
3310         if (hdr->frame_control & cpu_to_le16(IEEE80211_STYPE_QOS_DATA)) {
3311                 tid = skb->priority & IEEE80211_QOS_CTL_TAG1D_MASK;
3312                 tid_tx = rcu_dereference(sta->ampdu_mlme.tid_tx[tid]);
3313                 if (tid_tx) {
3314                         if (!test_bit(HT_AGG_STATE_OPERATIONAL, &tid_tx->state))
3315                                 return false;
3316                         if (tid_tx->timeout)
3317                                 tid_tx->last_tx = jiffies;
3318                 }
3319         }
3320
3321         /* after this point (skb is modified) we cannot return false */
3322
3323         if (skb_shared(skb)) {
3324                 struct sk_buff *tmp_skb = skb;
3325
3326                 skb = skb_clone(skb, GFP_ATOMIC);
3327                 kfree_skb(tmp_skb);
3328
3329                 if (!skb)
3330                         return true;
3331         }
3332
3333         if ((hdr->frame_control & cpu_to_le16(IEEE80211_STYPE_QOS_DATA)) &&
3334             ieee80211_amsdu_aggregate(sdata, sta, fast_tx, skb))
3335                 return true;
3336
3337         /* will not be crypto-handled beyond what we do here, so use false
3338          * as the may-encrypt argument for the resize to not account for
3339          * more room than we already have in 'extra_head'
3340          */
3341         if (unlikely(ieee80211_skb_resize(sdata, skb,
3342                                           max_t(int, extra_head + hw_headroom -
3343                                                      skb_headroom(skb), 0),
3344                                           false))) {
3345                 kfree_skb(skb);
3346                 return true;
3347         }
3348
3349         memcpy(&eth, skb->data, ETH_HLEN - 2);
3350         hdr = skb_push(skb, extra_head);
3351         memcpy(skb->data, fast_tx->hdr, fast_tx->hdr_len);
3352         memcpy(skb->data + fast_tx->da_offs, eth.h_dest, ETH_ALEN);
3353         memcpy(skb->data + fast_tx->sa_offs, eth.h_source, ETH_ALEN);
3354
3355         info = IEEE80211_SKB_CB(skb);
3356         memset(info, 0, sizeof(*info));
3357         info->band = fast_tx->band;
3358         info->control.vif = &sdata->vif;
3359         info->flags = IEEE80211_TX_CTL_FIRST_FRAGMENT |
3360                       IEEE80211_TX_CTL_DONTFRAG |
3361                       (tid_tx ? IEEE80211_TX_CTL_AMPDU : 0);
3362         info->control.flags = IEEE80211_TX_CTRL_FAST_XMIT;
3363
3364         if (hdr->frame_control & cpu_to_le16(IEEE80211_STYPE_QOS_DATA)) {
3365                 tid = skb->priority & IEEE80211_QOS_CTL_TAG1D_MASK;
3366                 *ieee80211_get_qos_ctl(hdr) = tid;
3367         }
3368
3369         __skb_queue_head_init(&tx.skbs);
3370
3371         tx.flags = IEEE80211_TX_UNICAST;
3372         tx.local = local;
3373         tx.sdata = sdata;
3374         tx.sta = sta;
3375         tx.key = fast_tx->key;
3376
3377         if (!ieee80211_hw_check(&local->hw, HAS_RATE_CONTROL)) {
3378                 tx.skb = skb;
3379                 r = ieee80211_tx_h_rate_ctrl(&tx);
3380                 skb = tx.skb;
3381                 tx.skb = NULL;
3382
3383                 if (r != TX_CONTINUE) {
3384                         if (r != TX_QUEUED)
3385                                 kfree_skb(skb);
3386                         return true;
3387                 }
3388         }
3389
3390         if (ieee80211_queue_skb(local, sdata, sta, skb))
3391                 return true;
3392
3393         ieee80211_xmit_fast_finish(sdata, sta, fast_tx->pn_offs,
3394                                    fast_tx->key, skb);
3395
3396         if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
3397                 sdata = container_of(sdata->bss,
3398                                      struct ieee80211_sub_if_data, u.ap);
3399
3400         __skb_queue_tail(&tx.skbs, skb);
3401         ieee80211_tx_frags(local, &sdata->vif, &sta->sta, &tx.skbs, false);
3402         return true;
3403 }
3404
3405 struct sk_buff *ieee80211_tx_dequeue(struct ieee80211_hw *hw,
3406                                      struct ieee80211_txq *txq)
3407 {
3408         struct ieee80211_local *local = hw_to_local(hw);
3409         struct txq_info *txqi = container_of(txq, struct txq_info, txq);
3410         struct ieee80211_hdr *hdr;
3411         struct sk_buff *skb = NULL;
3412         struct fq *fq = &local->fq;
3413         struct fq_tin *tin = &txqi->tin;
3414         struct ieee80211_tx_info *info;
3415         struct ieee80211_tx_data tx;
3416         ieee80211_tx_result r;
3417
3418         spin_lock_bh(&fq->lock);
3419
3420         if (test_bit(IEEE80211_TXQ_STOP, &txqi->flags))
3421                 goto out;
3422
3423         /* Make sure fragments stay together. */
3424         skb = __skb_dequeue(&txqi->frags);
3425         if (skb)
3426                 goto out;
3427
3428 begin:
3429         skb = fq_tin_dequeue(fq, tin, fq_tin_dequeue_func);
3430         if (!skb)
3431                 goto out;
3432
3433         ieee80211_set_skb_vif(skb, txqi);
3434
3435         hdr = (struct ieee80211_hdr *)skb->data;
3436         info = IEEE80211_SKB_CB(skb);
3437
3438         memset(&tx, 0, sizeof(tx));
3439         __skb_queue_head_init(&tx.skbs);
3440         tx.local = local;
3441         tx.skb = skb;
3442         tx.sdata = vif_to_sdata(info->control.vif);
3443
3444         if (txq->sta)
3445                 tx.sta = container_of(txq->sta, struct sta_info, sta);
3446
3447         /*
3448          * The key can be removed while the packet was queued, so need to call
3449          * this here to get the current key.
3450          */
3451         r = ieee80211_tx_h_select_key(&tx);
3452         if (r != TX_CONTINUE) {
3453                 ieee80211_free_txskb(&local->hw, skb);
3454                 goto begin;
3455         }
3456
3457         if (test_bit(IEEE80211_TXQ_AMPDU, &txqi->flags))
3458                 info->flags |= IEEE80211_TX_CTL_AMPDU;
3459         else
3460                 info->flags &= ~IEEE80211_TX_CTL_AMPDU;
3461
3462         if (info->control.flags & IEEE80211_TX_CTRL_FAST_XMIT) {
3463                 struct sta_info *sta = container_of(txq->sta, struct sta_info,
3464                                                     sta);
3465                 u8 pn_offs = 0;
3466
3467                 if (tx.key &&
3468                     (tx.key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV))
3469                         pn_offs = ieee80211_hdrlen(hdr->frame_control);
3470
3471                 ieee80211_xmit_fast_finish(sta->sdata, sta, pn_offs,
3472                                            tx.key, skb);
3473         } else {
3474                 if (invoke_tx_handlers_late(&tx))
3475                         goto begin;
3476
3477                 skb = __skb_dequeue(&tx.skbs);
3478
3479                 if (!skb_queue_empty(&tx.skbs))
3480                         skb_queue_splice_tail(&tx.skbs, &txqi->frags);
3481         }
3482
3483         if (skb && skb_has_frag_list(skb) &&
3484             !ieee80211_hw_check(&local->hw, TX_FRAG_LIST)) {
3485                 if (skb_linearize(skb)) {
3486                         ieee80211_free_txskb(&local->hw, skb);
3487                         goto begin;
3488                 }
3489         }
3490
3491 out:
3492         spin_unlock_bh(&fq->lock);
3493
3494         return skb;
3495 }
3496 EXPORT_SYMBOL(ieee80211_tx_dequeue);
3497
3498 void __ieee80211_subif_start_xmit(struct sk_buff *skb,
3499                                   struct net_device *dev,
3500                                   u32 info_flags)
3501 {
3502         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3503         struct sta_info *sta;
3504         struct sk_buff *next;
3505
3506         if (unlikely(skb->len < ETH_HLEN)) {
3507                 kfree_skb(skb);
3508                 return;
3509         }
3510
3511         rcu_read_lock();
3512
3513         if (ieee80211_lookup_ra_sta(sdata, skb, &sta))
3514                 goto out_free;
3515
3516         if (!IS_ERR_OR_NULL(sta)) {
3517                 struct ieee80211_fast_tx *fast_tx;
3518
3519                 fast_tx = rcu_dereference(sta->fast_tx);
3520
3521                 if (fast_tx &&
3522                     ieee80211_xmit_fast(sdata, sta, fast_tx, skb))
3523                         goto out;
3524         }
3525
3526         if (skb_is_gso(skb)) {
3527                 struct sk_buff *segs;
3528
3529                 segs = skb_gso_segment(skb, 0);
3530                 if (IS_ERR(segs)) {
3531                         goto out_free;
3532                 } else if (segs) {
3533                         consume_skb(skb);
3534                         skb = segs;
3535                 }
3536         } else {
3537                 /* we cannot process non-linear frames on this path */
3538                 if (skb_linearize(skb)) {
3539                         kfree_skb(skb);
3540                         goto out;
3541                 }
3542
3543                 /* the frame could be fragmented, software-encrypted, and other
3544                  * things so we cannot really handle checksum offload with it -
3545                  * fix it up in software before we handle anything else.
3546                  */
3547                 if (skb->ip_summed == CHECKSUM_PARTIAL) {
3548                         skb_set_transport_header(skb,
3549                                                  skb_checksum_start_offset(skb));
3550                         if (skb_checksum_help(skb))
3551                                 goto out_free;
3552                 }
3553         }
3554
3555         next = skb;
3556         while (next) {
3557                 skb = next;
3558                 next = skb->next;
3559
3560                 skb->prev = NULL;
3561                 skb->next = NULL;
3562
3563                 skb = ieee80211_build_hdr(sdata, skb, info_flags, sta);
3564                 if (IS_ERR(skb))
3565                         goto out;
3566
3567                 ieee80211_tx_stats(dev, skb->len);
3568
3569                 ieee80211_xmit(sdata, sta, skb);
3570         }
3571         goto out;
3572  out_free:
3573         kfree_skb(skb);
3574  out:
3575         rcu_read_unlock();
3576 }
3577
3578 static int ieee80211_change_da(struct sk_buff *skb, struct sta_info *sta)
3579 {
3580         struct ethhdr *eth;
3581         int err;
3582
3583         err = skb_ensure_writable(skb, ETH_HLEN);
3584         if (unlikely(err))
3585                 return err;
3586
3587         eth = (void *)skb->data;
3588         ether_addr_copy(eth->h_dest, sta->sta.addr);
3589
3590         return 0;
3591 }
3592
3593 static bool ieee80211_multicast_to_unicast(struct sk_buff *skb,
3594                                            struct net_device *dev)
3595 {
3596         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3597         const struct ethhdr *eth = (void *)skb->data;
3598         const struct vlan_ethhdr *ethvlan = (void *)skb->data;
3599         __be16 ethertype;
3600
3601         if (likely(!is_multicast_ether_addr(eth->h_dest)))
3602                 return false;
3603
3604         switch (sdata->vif.type) {
3605         case NL80211_IFTYPE_AP_VLAN:
3606                 if (sdata->u.vlan.sta)
3607                         return false;
3608                 if (sdata->wdev.use_4addr)
3609                         return false;
3610                 /* fall through */
3611         case NL80211_IFTYPE_AP:
3612                 /* check runtime toggle for this bss */
3613                 if (!sdata->bss->multicast_to_unicast)
3614                         return false;
3615                 break;
3616         default:
3617                 return false;
3618         }
3619
3620         /* multicast to unicast conversion only for some payload */
3621         ethertype = eth->h_proto;
3622         if (ethertype == htons(ETH_P_8021Q) && skb->len >= VLAN_ETH_HLEN)
3623                 ethertype = ethvlan->h_vlan_encapsulated_proto;
3624         switch (ethertype) {
3625         case htons(ETH_P_ARP):
3626         case htons(ETH_P_IP):
3627         case htons(ETH_P_IPV6):
3628                 break;
3629         default:
3630                 return false;
3631         }
3632
3633         return true;
3634 }
3635
3636 static void
3637 ieee80211_convert_to_unicast(struct sk_buff *skb, struct net_device *dev,
3638                              struct sk_buff_head *queue)
3639 {
3640         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3641         struct ieee80211_local *local = sdata->local;
3642         const struct ethhdr *eth = (struct ethhdr *)skb->data;
3643         struct sta_info *sta, *first = NULL;
3644         struct sk_buff *cloned_skb;
3645
3646         rcu_read_lock();
3647
3648         list_for_each_entry_rcu(sta, &local->sta_list, list) {
3649                 if (sdata != sta->sdata)
3650                         /* AP-VLAN mismatch */
3651                         continue;
3652                 if (unlikely(ether_addr_equal(eth->h_source, sta->sta.addr)))
3653                         /* do not send back to source */
3654                         continue;
3655                 if (!first) {
3656                         first = sta;
3657                         continue;
3658                 }
3659                 cloned_skb = skb_clone(skb, GFP_ATOMIC);
3660                 if (!cloned_skb)
3661                         goto multicast;
3662                 if (unlikely(ieee80211_change_da(cloned_skb, sta))) {
3663                         dev_kfree_skb(cloned_skb);
3664                         goto multicast;
3665                 }
3666                 __skb_queue_tail(queue, cloned_skb);
3667         }
3668
3669         if (likely(first)) {
3670                 if (unlikely(ieee80211_change_da(skb, first)))
3671                         goto multicast;
3672                 __skb_queue_tail(queue, skb);
3673         } else {
3674                 /* no STA connected, drop */
3675                 kfree_skb(skb);
3676                 skb = NULL;
3677         }
3678
3679         goto out;
3680 multicast:
3681         __skb_queue_purge(queue);
3682         __skb_queue_tail(queue, skb);
3683 out:
3684         rcu_read_unlock();
3685 }
3686
3687 /**
3688  * ieee80211_subif_start_xmit - netif start_xmit function for 802.3 vifs
3689  * @skb: packet to be sent
3690  * @dev: incoming interface
3691  *
3692  * On failure skb will be freed.
3693  */
3694 netdev_tx_t ieee80211_subif_start_xmit(struct sk_buff *skb,
3695                                        struct net_device *dev)
3696 {
3697         if (unlikely(ieee80211_multicast_to_unicast(skb, dev))) {
3698                 struct sk_buff_head queue;
3699
3700                 __skb_queue_head_init(&queue);
3701                 ieee80211_convert_to_unicast(skb, dev, &queue);
3702                 while ((skb = __skb_dequeue(&queue)))
3703                         __ieee80211_subif_start_xmit(skb, dev, 0);
3704         } else {
3705                 __ieee80211_subif_start_xmit(skb, dev, 0);
3706         }
3707
3708         return NETDEV_TX_OK;
3709 }
3710
3711 struct sk_buff *
3712 ieee80211_build_data_template(struct ieee80211_sub_if_data *sdata,
3713                               struct sk_buff *skb, u32 info_flags)
3714 {
3715         struct ieee80211_hdr *hdr;
3716         struct ieee80211_tx_data tx = {
3717                 .local = sdata->local,
3718                 .sdata = sdata,
3719         };
3720         struct sta_info *sta;
3721
3722         rcu_read_lock();
3723
3724         if (ieee80211_lookup_ra_sta(sdata, skb, &sta)) {
3725                 kfree_skb(skb);
3726                 skb = ERR_PTR(-EINVAL);
3727                 goto out;
3728         }
3729
3730         skb = ieee80211_build_hdr(sdata, skb, info_flags, sta);
3731         if (IS_ERR(skb))
3732                 goto out;
3733
3734         hdr = (void *)skb->data;
3735         tx.sta = sta_info_get(sdata, hdr->addr1);
3736         tx.skb = skb;
3737
3738         if (ieee80211_tx_h_select_key(&tx) != TX_CONTINUE) {
3739                 rcu_read_unlock();
3740                 kfree_skb(skb);
3741                 return ERR_PTR(-EINVAL);
3742         }
3743
3744 out:
3745         rcu_read_unlock();
3746         return skb;
3747 }
3748
3749 /*
3750  * ieee80211_clear_tx_pending may not be called in a context where
3751  * it is possible that it packets could come in again.
3752  */
3753 void ieee80211_clear_tx_pending(struct ieee80211_local *local)
3754 {
3755         struct sk_buff *skb;
3756         int i;
3757
3758         for (i = 0; i < local->hw.queues; i++) {
3759                 while ((skb = skb_dequeue(&local->pending[i])) != NULL)
3760                         ieee80211_free_txskb(&local->hw, skb);
3761         }
3762 }
3763
3764 /*
3765  * Returns false if the frame couldn't be transmitted but was queued instead,
3766  * which in this case means re-queued -- take as an indication to stop sending
3767  * more pending frames.
3768  */
3769 static bool ieee80211_tx_pending_skb(struct ieee80211_local *local,
3770                                      struct sk_buff *skb)
3771 {
3772         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
3773         struct ieee80211_sub_if_data *sdata;
3774         struct sta_info *sta;
3775         struct ieee80211_hdr *hdr;
3776         bool result;
3777         struct ieee80211_chanctx_conf *chanctx_conf;
3778
3779         sdata = vif_to_sdata(info->control.vif);
3780
3781         if (info->flags & IEEE80211_TX_INTFL_NEED_TXPROCESSING) {
3782                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
3783                 if (unlikely(!chanctx_conf)) {
3784                         dev_kfree_skb(skb);
3785                         return true;
3786                 }
3787                 info->band = chanctx_conf->def.chan->band;
3788                 result = ieee80211_tx(sdata, NULL, skb, true);
3789         } else {
3790                 struct sk_buff_head skbs;
3791
3792                 __skb_queue_head_init(&skbs);
3793                 __skb_queue_tail(&skbs, skb);
3794
3795                 hdr = (struct ieee80211_hdr *)skb->data;
3796                 sta = sta_info_get(sdata, hdr->addr1);
3797
3798                 result = __ieee80211_tx(local, &skbs, skb->len, sta, true);
3799         }
3800
3801         return result;
3802 }
3803
3804 /*
3805  * Transmit all pending packets. Called from tasklet.
3806  */
3807 void ieee80211_tx_pending(unsigned long data)
3808 {
3809         struct ieee80211_local *local = (struct ieee80211_local *)data;
3810         unsigned long flags;
3811         int i;
3812         bool txok;
3813
3814         rcu_read_lock();
3815
3816         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
3817         for (i = 0; i < local->hw.queues; i++) {
3818                 /*
3819                  * If queue is stopped by something other than due to pending
3820                  * frames, or we have no pending frames, proceed to next queue.
3821                  */
3822                 if (local->queue_stop_reasons[i] ||
3823                     skb_queue_empty(&local->pending[i]))
3824                         continue;
3825
3826                 while (!skb_queue_empty(&local->pending[i])) {
3827                         struct sk_buff *skb = __skb_dequeue(&local->pending[i]);
3828                         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
3829
3830                         if (WARN_ON(!info->control.vif)) {
3831                                 ieee80211_free_txskb(&local->hw, skb);
3832                                 continue;
3833                         }
3834
3835                         spin_unlock_irqrestore(&local->queue_stop_reason_lock,
3836                                                 flags);
3837
3838                         txok = ieee80211_tx_pending_skb(local, skb);
3839                         spin_lock_irqsave(&local->queue_stop_reason_lock,
3840                                           flags);
3841                         if (!txok)
3842                                 break;
3843                 }
3844
3845                 if (skb_queue_empty(&local->pending[i]))
3846                         ieee80211_propagate_queue_wake(local, i);
3847         }
3848         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
3849
3850         rcu_read_unlock();
3851 }
3852
3853 /* functions for drivers to get certain frames */
3854
3855 static void __ieee80211_beacon_add_tim(struct ieee80211_sub_if_data *sdata,
3856                                        struct ps_data *ps, struct sk_buff *skb,
3857                                        bool is_template)
3858 {
3859         u8 *pos, *tim;
3860         int aid0 = 0;
3861         int i, have_bits = 0, n1, n2;
3862
3863         /* Generate bitmap for TIM only if there are any STAs in power save
3864          * mode. */
3865         if (atomic_read(&ps->num_sta_ps) > 0)
3866                 /* in the hope that this is faster than
3867                  * checking byte-for-byte */
3868                 have_bits = !bitmap_empty((unsigned long *)ps->tim,
3869                                           IEEE80211_MAX_AID+1);
3870         if (!is_template) {
3871                 if (ps->dtim_count == 0)
3872                         ps->dtim_count = sdata->vif.bss_conf.dtim_period - 1;
3873                 else
3874                         ps->dtim_count--;
3875         }
3876
3877         tim = pos = skb_put(skb, 6);
3878         *pos++ = WLAN_EID_TIM;
3879         *pos++ = 4;
3880         *pos++ = ps->dtim_count;
3881         *pos++ = sdata->vif.bss_conf.dtim_period;
3882
3883         if (ps->dtim_count == 0 && !skb_queue_empty(&ps->bc_buf))
3884                 aid0 = 1;
3885
3886         ps->dtim_bc_mc = aid0 == 1;
3887
3888         if (have_bits) {
3889                 /* Find largest even number N1 so that bits numbered 1 through
3890                  * (N1 x 8) - 1 in the bitmap are 0 and number N2 so that bits
3891                  * (N2 + 1) x 8 through 2007 are 0. */
3892                 n1 = 0;
3893                 for (i = 0; i < IEEE80211_MAX_TIM_LEN; i++) {
3894                         if (ps->tim[i]) {
3895                                 n1 = i & 0xfe;
3896                                 break;
3897                         }
3898                 }
3899                 n2 = n1;
3900                 for (i = IEEE80211_MAX_TIM_LEN - 1; i >= n1; i--) {
3901                         if (ps->tim[i]) {
3902                                 n2 = i;
3903                                 break;
3904                         }
3905                 }
3906
3907                 /* Bitmap control */
3908                 *pos++ = n1 | aid0;
3909                 /* Part Virt Bitmap */
3910                 skb_put(skb, n2 - n1);
3911                 memcpy(pos, ps->tim + n1, n2 - n1 + 1);
3912
3913                 tim[1] = n2 - n1 + 4;
3914         } else {
3915                 *pos++ = aid0; /* Bitmap control */
3916                 *pos++ = 0; /* Part Virt Bitmap */
3917         }
3918 }
3919
3920 static int ieee80211_beacon_add_tim(struct ieee80211_sub_if_data *sdata,
3921                                     struct ps_data *ps, struct sk_buff *skb,
3922                                     bool is_template)
3923 {
3924         struct ieee80211_local *local = sdata->local;
3925
3926         /*
3927          * Not very nice, but we want to allow the driver to call
3928          * ieee80211_beacon_get() as a response to the set_tim()
3929          * callback. That, however, is already invoked under the
3930          * sta_lock to guarantee consistent and race-free update
3931          * of the tim bitmap in mac80211 and the driver.
3932          */
3933         if (local->tim_in_locked_section) {
3934                 __ieee80211_beacon_add_tim(sdata, ps, skb, is_template);
3935         } else {
3936                 spin_lock_bh(&local->tim_lock);
3937                 __ieee80211_beacon_add_tim(sdata, ps, skb, is_template);
3938                 spin_unlock_bh(&local->tim_lock);
3939         }
3940
3941         return 0;
3942 }
3943
3944 static void ieee80211_set_csa(struct ieee80211_sub_if_data *sdata,
3945                               struct beacon_data *beacon)
3946 {
3947         struct probe_resp *resp;
3948         u8 *beacon_data;
3949         size_t beacon_data_len;
3950         int i;
3951         u8 count = beacon->csa_current_counter;
3952
3953         switch (sdata->vif.type) {
3954         case NL80211_IFTYPE_AP:
3955                 beacon_data = beacon->tail;
3956                 beacon_data_len = beacon->tail_len;
3957                 break;
3958         case NL80211_IFTYPE_ADHOC:
3959                 beacon_data = beacon->head;
3960                 beacon_data_len = beacon->head_len;
3961                 break;
3962         case NL80211_IFTYPE_MESH_POINT:
3963                 beacon_data = beacon->head;
3964                 beacon_data_len = beacon->head_len;
3965                 break;
3966         default:
3967                 return;
3968         }
3969
3970         rcu_read_lock();
3971         for (i = 0; i < IEEE80211_MAX_CSA_COUNTERS_NUM; ++i) {
3972                 resp = rcu_dereference(sdata->u.ap.probe_resp);
3973
3974                 if (beacon->csa_counter_offsets[i]) {
3975                         if (WARN_ON_ONCE(beacon->csa_counter_offsets[i] >=
3976                                          beacon_data_len)) {
3977                                 rcu_read_unlock();
3978                                 return;
3979                         }
3980
3981                         beacon_data[beacon->csa_counter_offsets[i]] = count;
3982                 }
3983
3984                 if (sdata->vif.type == NL80211_IFTYPE_AP && resp)
3985                         resp->data[resp->csa_counter_offsets[i]] = count;
3986         }
3987         rcu_read_unlock();
3988 }
3989
3990 static u8 __ieee80211_csa_update_counter(struct beacon_data *beacon)
3991 {
3992         beacon->csa_current_counter--;
3993
3994         /* the counter should never reach 0 */
3995         WARN_ON_ONCE(!beacon->csa_current_counter);
3996
3997         return beacon->csa_current_counter;
3998 }
3999
4000 u8 ieee80211_csa_update_counter(struct ieee80211_vif *vif)
4001 {
4002         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
4003         struct beacon_data *beacon = NULL;
4004         u8 count = 0;
4005
4006         rcu_read_lock();
4007
4008         if (sdata->vif.type == NL80211_IFTYPE_AP)
4009                 beacon = rcu_dereference(sdata->u.ap.beacon);
4010         else if (sdata->vif.type == NL80211_IFTYPE_ADHOC)
4011                 beacon = rcu_dereference(sdata->u.ibss.presp);
4012         else if (ieee80211_vif_is_mesh(&sdata->vif))
4013                 beacon = rcu_dereference(sdata->u.mesh.beacon);
4014
4015         if (!beacon)
4016                 goto unlock;
4017
4018         count = __ieee80211_csa_update_counter(beacon);
4019
4020 unlock:
4021         rcu_read_unlock();
4022         return count;
4023 }
4024 EXPORT_SYMBOL(ieee80211_csa_update_counter);
4025
4026 bool ieee80211_csa_is_complete(struct ieee80211_vif *vif)
4027 {
4028         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
4029         struct beacon_data *beacon = NULL;
4030         u8 *beacon_data;
4031         size_t beacon_data_len;
4032         int ret = false;
4033
4034         if (!ieee80211_sdata_running(sdata))
4035                 return false;
4036
4037         rcu_read_lock();
4038         if (vif->type == NL80211_IFTYPE_AP) {
4039                 struct ieee80211_if_ap *ap = &sdata->u.ap;
4040
4041                 beacon = rcu_dereference(ap->beacon);
4042                 if (WARN_ON(!beacon || !beacon->tail))
4043                         goto out;
4044                 beacon_data = beacon->tail;
4045                 beacon_data_len = beacon->tail_len;
4046         } else if (vif->type == NL80211_IFTYPE_ADHOC) {
4047                 struct ieee80211_if_ibss *ifibss = &sdata->u.ibss;
4048
4049                 beacon = rcu_dereference(ifibss->presp);
4050                 if (!beacon)
4051                         goto out;
4052
4053                 beacon_data = beacon->head;
4054                 beacon_data_len = beacon->head_len;
4055         } else if (vif->type == NL80211_IFTYPE_MESH_POINT) {
4056                 struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
4057
4058                 beacon = rcu_dereference(ifmsh->beacon);
4059                 if (!beacon)
4060                         goto out;
4061
4062                 beacon_data = beacon->head;
4063                 beacon_data_len = beacon->head_len;
4064         } else {
4065                 WARN_ON(1);
4066                 goto out;
4067         }
4068
4069         if (!beacon->csa_counter_offsets[0])
4070                 goto out;
4071
4072         if (WARN_ON_ONCE(beacon->csa_counter_offsets[0] > beacon_data_len))
4073                 goto out;
4074
4075         if (beacon_data[beacon->csa_counter_offsets[0]] == 1)
4076                 ret = true;
4077  out:
4078         rcu_read_unlock();
4079
4080         return ret;
4081 }
4082 EXPORT_SYMBOL(ieee80211_csa_is_complete);
4083
4084 static struct sk_buff *
4085 __ieee80211_beacon_get(struct ieee80211_hw *hw,
4086                        struct ieee80211_vif *vif,
4087                        struct ieee80211_mutable_offsets *offs,
4088                        bool is_template)
4089 {
4090         struct ieee80211_local *local = hw_to_local(hw);
4091         struct beacon_data *beacon = NULL;
4092         struct sk_buff *skb = NULL;
4093         struct ieee80211_tx_info *info;
4094         struct ieee80211_sub_if_data *sdata = NULL;
4095         enum nl80211_band band;
4096         struct ieee80211_tx_rate_control txrc;
4097         struct ieee80211_chanctx_conf *chanctx_conf;
4098         int csa_off_base = 0;
4099
4100         rcu_read_lock();
4101
4102         sdata = vif_to_sdata(vif);
4103         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
4104
4105         if (!ieee80211_sdata_running(sdata) || !chanctx_conf)
4106                 goto out;
4107
4108         if (offs)
4109                 memset(offs, 0, sizeof(*offs));
4110
4111         if (sdata->vif.type == NL80211_IFTYPE_AP) {
4112                 struct ieee80211_if_ap *ap = &sdata->u.ap;
4113
4114                 beacon = rcu_dereference(ap->beacon);
4115                 if (beacon) {
4116                         if (beacon->csa_counter_offsets[0]) {
4117                                 if (!is_template)
4118                                         __ieee80211_csa_update_counter(beacon);
4119
4120                                 ieee80211_set_csa(sdata, beacon);
4121                         }
4122
4123                         /*
4124                          * headroom, head length,
4125                          * tail length and maximum TIM length
4126                          */
4127                         skb = dev_alloc_skb(local->tx_headroom +
4128                                             beacon->head_len +
4129                                             beacon->tail_len + 256 +
4130                                             local->hw.extra_beacon_tailroom);
4131                         if (!skb)
4132                                 goto out;
4133
4134                         skb_reserve(skb, local->tx_headroom);
4135                         skb_put_data(skb, beacon->head, beacon->head_len);
4136
4137                         ieee80211_beacon_add_tim(sdata, &ap->ps, skb,
4138                                                  is_template);
4139
4140                         if (offs) {
4141                                 offs->tim_offset = beacon->head_len;
4142                                 offs->tim_length = skb->len - beacon->head_len;
4143
4144                                 /* for AP the csa offsets are from tail */
4145                                 csa_off_base = skb->len;
4146                         }
4147
4148                         if (beacon->tail)
4149                                 skb_put_data(skb, beacon->tail,
4150                                              beacon->tail_len);
4151                 } else
4152                         goto out;
4153         } else if (sdata->vif.type == NL80211_IFTYPE_ADHOC) {
4154                 struct ieee80211_if_ibss *ifibss = &sdata->u.ibss;
4155                 struct ieee80211_hdr *hdr;
4156
4157                 beacon = rcu_dereference(ifibss->presp);
4158                 if (!beacon)
4159                         goto out;
4160
4161                 if (beacon->csa_counter_offsets[0]) {
4162                         if (!is_template)
4163                                 __ieee80211_csa_update_counter(beacon);
4164
4165                         ieee80211_set_csa(sdata, beacon);
4166                 }
4167
4168                 skb = dev_alloc_skb(local->tx_headroom + beacon->head_len +
4169                                     local->hw.extra_beacon_tailroom);
4170                 if (!skb)
4171                         goto out;
4172                 skb_reserve(skb, local->tx_headroom);
4173                 skb_put_data(skb, beacon->head, beacon->head_len);
4174
4175                 hdr = (struct ieee80211_hdr *) skb->data;
4176                 hdr->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
4177                                                  IEEE80211_STYPE_BEACON);
4178         } else if (ieee80211_vif_is_mesh(&sdata->vif)) {
4179                 struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
4180
4181                 beacon = rcu_dereference(ifmsh->beacon);
4182                 if (!beacon)
4183                         goto out;
4184
4185                 if (beacon->csa_counter_offsets[0]) {
4186                         if (!is_template)
4187                                 /* TODO: For mesh csa_counter is in TU, so
4188                                  * decrementing it by one isn't correct, but
4189                                  * for now we leave it consistent with overall
4190                                  * mac80211's behavior.
4191                                  */
4192                                 __ieee80211_csa_update_counter(beacon);
4193
4194                         ieee80211_set_csa(sdata, beacon);
4195                 }
4196
4197                 if (ifmsh->sync_ops)
4198                         ifmsh->sync_ops->adjust_tsf(sdata, beacon);
4199
4200                 skb = dev_alloc_skb(local->tx_headroom +
4201                                     beacon->head_len +
4202                                     256 + /* TIM IE */
4203                                     beacon->tail_len +
4204                                     local->hw.extra_beacon_tailroom);
4205                 if (!skb)
4206                         goto out;
4207                 skb_reserve(skb, local->tx_headroom);
4208                 skb_put_data(skb, beacon->head, beacon->head_len);
4209                 ieee80211_beacon_add_tim(sdata, &ifmsh->ps, skb, is_template);
4210
4211                 if (offs) {
4212                         offs->tim_offset = beacon->head_len;
4213                         offs->tim_length = skb->len - beacon->head_len;
4214                 }
4215
4216                 skb_put_data(skb, beacon->tail, beacon->tail_len);
4217         } else {
4218                 WARN_ON(1);
4219                 goto out;
4220         }
4221
4222         /* CSA offsets */
4223         if (offs && beacon) {
4224                 int i;
4225
4226                 for (i = 0; i < IEEE80211_MAX_CSA_COUNTERS_NUM; i++) {
4227                         u16 csa_off = beacon->csa_counter_offsets[i];
4228
4229                         if (!csa_off)
4230                                 continue;
4231
4232                         offs->csa_counter_offs[i] = csa_off_base + csa_off;
4233                 }
4234         }
4235
4236         band = chanctx_conf->def.chan->band;
4237
4238         info = IEEE80211_SKB_CB(skb);
4239
4240         info->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
4241         info->flags |= IEEE80211_TX_CTL_NO_ACK;
4242         info->band = band;
4243
4244         memset(&txrc, 0, sizeof(txrc));
4245         txrc.hw = hw;
4246         txrc.sband = local->hw.wiphy->bands[band];
4247         txrc.bss_conf = &sdata->vif.bss_conf;
4248         txrc.skb = skb;
4249         txrc.reported_rate.idx = -1;
4250         txrc.rate_idx_mask = sdata->rc_rateidx_mask[band];
4251         txrc.bss = true;
4252         rate_control_get_rate(sdata, NULL, &txrc);
4253
4254         info->control.vif = vif;
4255
4256         info->flags |= IEEE80211_TX_CTL_CLEAR_PS_FILT |
4257                         IEEE80211_TX_CTL_ASSIGN_SEQ |
4258                         IEEE80211_TX_CTL_FIRST_FRAGMENT;
4259  out:
4260         rcu_read_unlock();
4261         return skb;
4262
4263 }
4264
4265 struct sk_buff *
4266 ieee80211_beacon_get_template(struct ieee80211_hw *hw,
4267                               struct ieee80211_vif *vif,
4268                               struct ieee80211_mutable_offsets *offs)
4269 {
4270         return __ieee80211_beacon_get(hw, vif, offs, true);
4271 }
4272 EXPORT_SYMBOL(ieee80211_beacon_get_template);
4273
4274 struct sk_buff *ieee80211_beacon_get_tim(struct ieee80211_hw *hw,
4275                                          struct ieee80211_vif *vif,
4276                                          u16 *tim_offset, u16 *tim_length)
4277 {
4278         struct ieee80211_mutable_offsets offs = {};
4279         struct sk_buff *bcn = __ieee80211_beacon_get(hw, vif, &offs, false);
4280         struct sk_buff *copy;
4281         struct ieee80211_supported_band *sband;
4282         int shift;
4283
4284         if (!bcn)
4285                 return bcn;
4286
4287         if (tim_offset)
4288                 *tim_offset = offs.tim_offset;
4289
4290         if (tim_length)
4291                 *tim_length = offs.tim_length;
4292
4293         if (ieee80211_hw_check(hw, BEACON_TX_STATUS) ||
4294             !hw_to_local(hw)->monitors)
4295                 return bcn;
4296
4297         /* send a copy to monitor interfaces */
4298         copy = skb_copy(bcn, GFP_ATOMIC);
4299         if (!copy)
4300                 return bcn;
4301
4302         shift = ieee80211_vif_get_shift(vif);
4303         sband = ieee80211_get_sband(vif_to_sdata(vif));
4304         if (!sband)
4305                 return bcn;
4306
4307         ieee80211_tx_monitor(hw_to_local(hw), copy, sband, 1, shift, false);
4308
4309         return bcn;
4310 }
4311 EXPORT_SYMBOL(ieee80211_beacon_get_tim);
4312
4313 struct sk_buff *ieee80211_proberesp_get(struct ieee80211_hw *hw,
4314                                         struct ieee80211_vif *vif)
4315 {
4316         struct ieee80211_if_ap *ap = NULL;
4317         struct sk_buff *skb = NULL;
4318         struct probe_resp *presp = NULL;
4319         struct ieee80211_hdr *hdr;
4320         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
4321
4322         if (sdata->vif.type != NL80211_IFTYPE_AP)
4323                 return NULL;
4324
4325         rcu_read_lock();
4326
4327         ap = &sdata->u.ap;
4328         presp = rcu_dereference(ap->probe_resp);
4329         if (!presp)
4330                 goto out;
4331
4332         skb = dev_alloc_skb(presp->len);
4333         if (!skb)
4334                 goto out;
4335
4336         skb_put_data(skb, presp->data, presp->len);
4337
4338         hdr = (struct ieee80211_hdr *) skb->data;
4339         memset(hdr->addr1, 0, sizeof(hdr->addr1));
4340
4341 out:
4342         rcu_read_unlock();
4343         return skb;
4344 }
4345 EXPORT_SYMBOL(ieee80211_proberesp_get);
4346
4347 struct sk_buff *ieee80211_pspoll_get(struct ieee80211_hw *hw,
4348                                      struct ieee80211_vif *vif)
4349 {
4350         struct ieee80211_sub_if_data *sdata;
4351         struct ieee80211_if_managed *ifmgd;
4352         struct ieee80211_pspoll *pspoll;
4353         struct ieee80211_local *local;
4354         struct sk_buff *skb;
4355
4356         if (WARN_ON(vif->type != NL80211_IFTYPE_STATION))
4357                 return NULL;
4358
4359         sdata = vif_to_sdata(vif);
4360         ifmgd = &sdata->u.mgd;
4361         local = sdata->local;
4362
4363         skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*pspoll));
4364         if (!skb)
4365                 return NULL;
4366
4367         skb_reserve(skb, local->hw.extra_tx_headroom);
4368
4369         pspoll = skb_put_zero(skb, sizeof(*pspoll));
4370         pspoll->frame_control = cpu_to_le16(IEEE80211_FTYPE_CTL |
4371                                             IEEE80211_STYPE_PSPOLL);
4372         pspoll->aid = cpu_to_le16(ifmgd->aid);
4373
4374         /* aid in PS-Poll has its two MSBs each set to 1 */
4375         pspoll->aid |= cpu_to_le16(1 << 15 | 1 << 14);
4376
4377         memcpy(pspoll->bssid, ifmgd->bssid, ETH_ALEN);
4378         memcpy(pspoll->ta, vif->addr, ETH_ALEN);
4379
4380         return skb;
4381 }
4382 EXPORT_SYMBOL(ieee80211_pspoll_get);
4383
4384 struct sk_buff *ieee80211_nullfunc_get(struct ieee80211_hw *hw,
4385                                        struct ieee80211_vif *vif)
4386 {
4387         struct ieee80211_hdr_3addr *nullfunc;
4388         struct ieee80211_sub_if_data *sdata;
4389         struct ieee80211_if_managed *ifmgd;
4390         struct ieee80211_local *local;
4391         struct sk_buff *skb;
4392
4393         if (WARN_ON(vif->type != NL80211_IFTYPE_STATION))
4394                 return NULL;
4395
4396         sdata = vif_to_sdata(vif);
4397         ifmgd = &sdata->u.mgd;
4398         local = sdata->local;
4399
4400         skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*nullfunc));
4401         if (!skb)
4402                 return NULL;
4403
4404         skb_reserve(skb, local->hw.extra_tx_headroom);
4405
4406         nullfunc = skb_put_zero(skb, sizeof(*nullfunc));
4407         nullfunc->frame_control = cpu_to_le16(IEEE80211_FTYPE_DATA |
4408                                               IEEE80211_STYPE_NULLFUNC |
4409                                               IEEE80211_FCTL_TODS);
4410         memcpy(nullfunc->addr1, ifmgd->bssid, ETH_ALEN);
4411         memcpy(nullfunc->addr2, vif->addr, ETH_ALEN);
4412         memcpy(nullfunc->addr3, ifmgd->bssid, ETH_ALEN);
4413
4414         return skb;
4415 }
4416 EXPORT_SYMBOL(ieee80211_nullfunc_get);
4417
4418 struct sk_buff *ieee80211_probereq_get(struct ieee80211_hw *hw,
4419                                        const u8 *src_addr,
4420                                        const u8 *ssid, size_t ssid_len,
4421                                        size_t tailroom)
4422 {
4423         struct ieee80211_local *local = hw_to_local(hw);
4424         struct ieee80211_hdr_3addr *hdr;
4425         struct sk_buff *skb;
4426         size_t ie_ssid_len;
4427         u8 *pos;
4428
4429         ie_ssid_len = 2 + ssid_len;
4430
4431         skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*hdr) +
4432                             ie_ssid_len + tailroom);
4433         if (!skb)
4434                 return NULL;
4435
4436         skb_reserve(skb, local->hw.extra_tx_headroom);
4437
4438         hdr = skb_put_zero(skb, sizeof(*hdr));
4439         hdr->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
4440                                          IEEE80211_STYPE_PROBE_REQ);
4441         eth_broadcast_addr(hdr->addr1);
4442         memcpy(hdr->addr2, src_addr, ETH_ALEN);
4443         eth_broadcast_addr(hdr->addr3);
4444
4445         pos = skb_put(skb, ie_ssid_len);
4446         *pos++ = WLAN_EID_SSID;
4447         *pos++ = ssid_len;
4448         if (ssid_len)
4449                 memcpy(pos, ssid, ssid_len);
4450         pos += ssid_len;
4451
4452         return skb;
4453 }
4454 EXPORT_SYMBOL(ieee80211_probereq_get);
4455
4456 void ieee80211_rts_get(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
4457                        const void *frame, size_t frame_len,
4458                        const struct ieee80211_tx_info *frame_txctl,
4459                        struct ieee80211_rts *rts)
4460 {
4461         const struct ieee80211_hdr *hdr = frame;
4462
4463         rts->frame_control =
4464             cpu_to_le16(IEEE80211_FTYPE_CTL | IEEE80211_STYPE_RTS);
4465         rts->duration = ieee80211_rts_duration(hw, vif, frame_len,
4466                                                frame_txctl);
4467         memcpy(rts->ra, hdr->addr1, sizeof(rts->ra));
4468         memcpy(rts->ta, hdr->addr2, sizeof(rts->ta));
4469 }
4470 EXPORT_SYMBOL(ieee80211_rts_get);
4471
4472 void ieee80211_ctstoself_get(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
4473                              const void *frame, size_t frame_len,
4474                              const struct ieee80211_tx_info *frame_txctl,
4475                              struct ieee80211_cts *cts)
4476 {
4477         const struct ieee80211_hdr *hdr = frame;
4478
4479         cts->frame_control =
4480             cpu_to_le16(IEEE80211_FTYPE_CTL | IEEE80211_STYPE_CTS);
4481         cts->duration = ieee80211_ctstoself_duration(hw, vif,
4482                                                      frame_len, frame_txctl);
4483         memcpy(cts->ra, hdr->addr1, sizeof(cts->ra));
4484 }
4485 EXPORT_SYMBOL(ieee80211_ctstoself_get);
4486
4487 struct sk_buff *
4488 ieee80211_get_buffered_bc(struct ieee80211_hw *hw,
4489                           struct ieee80211_vif *vif)
4490 {
4491         struct ieee80211_local *local = hw_to_local(hw);
4492         struct sk_buff *skb = NULL;
4493         struct ieee80211_tx_data tx;
4494         struct ieee80211_sub_if_data *sdata;
4495         struct ps_data *ps;
4496         struct ieee80211_tx_info *info;
4497         struct ieee80211_chanctx_conf *chanctx_conf;
4498
4499         sdata = vif_to_sdata(vif);
4500
4501         rcu_read_lock();
4502         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
4503
4504         if (!chanctx_conf)
4505                 goto out;
4506
4507         if (sdata->vif.type == NL80211_IFTYPE_AP) {
4508                 struct beacon_data *beacon =
4509                                 rcu_dereference(sdata->u.ap.beacon);
4510
4511                 if (!beacon || !beacon->head)
4512                         goto out;
4513
4514                 ps = &sdata->u.ap.ps;
4515         } else if (ieee80211_vif_is_mesh(&sdata->vif)) {
4516                 ps = &sdata->u.mesh.ps;
4517         } else {
4518                 goto out;
4519         }
4520
4521         if (ps->dtim_count != 0 || !ps->dtim_bc_mc)
4522                 goto out; /* send buffered bc/mc only after DTIM beacon */
4523
4524         while (1) {
4525                 skb = skb_dequeue(&ps->bc_buf);
4526                 if (!skb)
4527                         goto out;
4528                 local->total_ps_buffered--;
4529
4530                 if (!skb_queue_empty(&ps->bc_buf) && skb->len >= 2) {
4531                         struct ieee80211_hdr *hdr =
4532                                 (struct ieee80211_hdr *) skb->data;
4533                         /* more buffered multicast/broadcast frames ==> set
4534                          * MoreData flag in IEEE 802.11 header to inform PS
4535                          * STAs */
4536                         hdr->frame_control |=
4537                                 cpu_to_le16(IEEE80211_FCTL_MOREDATA);
4538                 }
4539
4540                 if (sdata->vif.type == NL80211_IFTYPE_AP)
4541                         sdata = IEEE80211_DEV_TO_SUB_IF(skb->dev);
4542                 if (!ieee80211_tx_prepare(sdata, &tx, NULL, skb))
4543                         break;
4544                 ieee80211_free_txskb(hw, skb);
4545         }
4546
4547         info = IEEE80211_SKB_CB(skb);
4548
4549         tx.flags |= IEEE80211_TX_PS_BUFFERED;
4550         info->band = chanctx_conf->def.chan->band;
4551
4552         if (invoke_tx_handlers(&tx))
4553                 skb = NULL;
4554  out:
4555         rcu_read_unlock();
4556
4557         return skb;
4558 }
4559 EXPORT_SYMBOL(ieee80211_get_buffered_bc);
4560
4561 int ieee80211_reserve_tid(struct ieee80211_sta *pubsta, u8 tid)
4562 {
4563         struct sta_info *sta = container_of(pubsta, struct sta_info, sta);
4564         struct ieee80211_sub_if_data *sdata = sta->sdata;
4565         struct ieee80211_local *local = sdata->local;
4566         int ret;
4567         u32 queues;
4568
4569         lockdep_assert_held(&local->sta_mtx);
4570
4571         /* only some cases are supported right now */
4572         switch (sdata->vif.type) {
4573         case NL80211_IFTYPE_STATION:
4574         case NL80211_IFTYPE_AP:
4575         case NL80211_IFTYPE_AP_VLAN:
4576                 break;
4577         default:
4578                 WARN_ON(1);
4579                 return -EINVAL;
4580         }
4581
4582         if (WARN_ON(tid >= IEEE80211_NUM_UPS))
4583                 return -EINVAL;
4584
4585         if (sta->reserved_tid == tid) {
4586                 ret = 0;
4587                 goto out;
4588         }
4589
4590         if (sta->reserved_tid != IEEE80211_TID_UNRESERVED) {
4591                 sdata_err(sdata, "TID reservation already active\n");
4592                 ret = -EALREADY;
4593                 goto out;
4594         }
4595
4596         ieee80211_stop_vif_queues(sdata->local, sdata,
4597                                   IEEE80211_QUEUE_STOP_REASON_RESERVE_TID);
4598
4599         synchronize_net();
4600
4601         /* Tear down BA sessions so we stop aggregating on this TID */
4602         if (ieee80211_hw_check(&local->hw, AMPDU_AGGREGATION)) {
4603                 set_sta_flag(sta, WLAN_STA_BLOCK_BA);
4604                 __ieee80211_stop_tx_ba_session(sta, tid,
4605                                                AGG_STOP_LOCAL_REQUEST);
4606         }
4607
4608         queues = BIT(sdata->vif.hw_queue[ieee802_1d_to_ac[tid]]);
4609         __ieee80211_flush_queues(local, sdata, queues, false);
4610
4611         sta->reserved_tid = tid;
4612
4613         ieee80211_wake_vif_queues(local, sdata,
4614                                   IEEE80211_QUEUE_STOP_REASON_RESERVE_TID);
4615
4616         if (ieee80211_hw_check(&local->hw, AMPDU_AGGREGATION))
4617                 clear_sta_flag(sta, WLAN_STA_BLOCK_BA);
4618
4619         ret = 0;
4620  out:
4621         return ret;
4622 }
4623 EXPORT_SYMBOL(ieee80211_reserve_tid);
4624
4625 void ieee80211_unreserve_tid(struct ieee80211_sta *pubsta, u8 tid)
4626 {
4627         struct sta_info *sta = container_of(pubsta, struct sta_info, sta);
4628         struct ieee80211_sub_if_data *sdata = sta->sdata;
4629
4630         lockdep_assert_held(&sdata->local->sta_mtx);
4631
4632         /* only some cases are supported right now */
4633         switch (sdata->vif.type) {
4634         case NL80211_IFTYPE_STATION:
4635         case NL80211_IFTYPE_AP:
4636         case NL80211_IFTYPE_AP_VLAN:
4637                 break;
4638         default:
4639                 WARN_ON(1);
4640                 return;
4641         }
4642
4643         if (tid != sta->reserved_tid) {
4644                 sdata_err(sdata, "TID to unreserve (%d) isn't reserved\n", tid);
4645                 return;
4646         }
4647
4648         sta->reserved_tid = IEEE80211_TID_UNRESERVED;
4649 }
4650 EXPORT_SYMBOL(ieee80211_unreserve_tid);
4651
4652 void __ieee80211_tx_skb_tid_band(struct ieee80211_sub_if_data *sdata,
4653                                  struct sk_buff *skb, int tid,
4654                                  enum nl80211_band band)
4655 {
4656         int ac = ieee80211_ac_from_tid(tid);
4657
4658         skb_reset_mac_header(skb);
4659         skb_set_queue_mapping(skb, ac);
4660         skb->priority = tid;
4661
4662         skb->dev = sdata->dev;
4663
4664         /*
4665          * The other path calling ieee80211_xmit is from the tasklet,
4666          * and while we can handle concurrent transmissions locking
4667          * requirements are that we do not come into tx with bhs on.
4668          */
4669         local_bh_disable();
4670         IEEE80211_SKB_CB(skb)->band = band;
4671         ieee80211_xmit(sdata, NULL, skb);
4672         local_bh_enable();
4673 }