net/mac80211/tx.c

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