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