2ef0e19b06bb041c58d217e48e5e3d9ade30d7d2
[platform/kernel/linux-rpi.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  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License version 2 as
9  * published by the Free Software Foundation.
10  *
11  *
12  * Transmit and frame generation functions.
13  */
14
15 #include <linux/kernel.h>
16 #include <linux/slab.h>
17 #include <linux/skbuff.h>
18 #include <linux/etherdevice.h>
19 #include <linux/bitmap.h>
20 #include <linux/rcupdate.h>
21 #include <linux/export.h>
22 #include <net/net_namespace.h>
23 #include <net/ieee80211_radiotap.h>
24 #include <net/cfg80211.h>
25 #include <net/mac80211.h>
26 #include <asm/unaligned.h>
27
28 #include "ieee80211_i.h"
29 #include "driver-ops.h"
30 #include "led.h"
31 #include "mesh.h"
32 #include "wep.h"
33 #include "wpa.h"
34 #include "wme.h"
35 #include "rate.h"
36
37 /* misc utils */
38
39 static __le16 ieee80211_duration(struct ieee80211_tx_data *tx,
40                                  struct sk_buff *skb, int group_addr,
41                                  int next_frag_len)
42 {
43         int rate, mrate, erp, dur, i;
44         struct ieee80211_rate *txrate;
45         struct ieee80211_local *local = tx->local;
46         struct ieee80211_supported_band *sband;
47         struct ieee80211_hdr *hdr;
48         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
49
50         /* assume HW handles this */
51         if (info->control.rates[0].flags & IEEE80211_TX_RC_MCS)
52                 return 0;
53
54         /* uh huh? */
55         if (WARN_ON_ONCE(info->control.rates[0].idx < 0))
56                 return 0;
57
58         sband = local->hw.wiphy->bands[info->band];
59         txrate = &sband->bitrates[info->control.rates[0].idx];
60
61         erp = txrate->flags & IEEE80211_RATE_ERP_G;
62
63         /*
64          * data and mgmt (except PS Poll):
65          * - during CFP: 32768
66          * - during contention period:
67          *   if addr1 is group address: 0
68          *   if more fragments = 0 and addr1 is individual address: time to
69          *      transmit one ACK plus SIFS
70          *   if more fragments = 1 and addr1 is individual address: time to
71          *      transmit next fragment plus 2 x ACK plus 3 x SIFS
72          *
73          * IEEE 802.11, 9.6:
74          * - control response frame (CTS or ACK) shall be transmitted using the
75          *   same rate as the immediately previous frame in the frame exchange
76          *   sequence, if this rate belongs to the PHY mandatory rates, or else
77          *   at the highest possible rate belonging to the PHY rates in the
78          *   BSSBasicRateSet
79          */
80         hdr = (struct ieee80211_hdr *)skb->data;
81         if (ieee80211_is_ctl(hdr->frame_control)) {
82                 /* TODO: These control frames are not currently sent by
83                  * mac80211, but should they be implemented, this function
84                  * needs to be updated to support duration field calculation.
85                  *
86                  * RTS: time needed to transmit pending data/mgmt frame plus
87                  *    one CTS frame plus one ACK frame plus 3 x SIFS
88                  * CTS: duration of immediately previous RTS minus time
89                  *    required to transmit CTS and its SIFS
90                  * ACK: 0 if immediately previous directed data/mgmt had
91                  *    more=0, with more=1 duration in ACK frame is duration
92                  *    from previous frame minus time needed to transmit ACK
93                  *    and its SIFS
94                  * PS Poll: BIT(15) | BIT(14) | aid
95                  */
96                 return 0;
97         }
98
99         /* data/mgmt */
100         if (0 /* FIX: data/mgmt during CFP */)
101                 return cpu_to_le16(32768);
102
103         if (group_addr) /* Group address as the destination - no ACK */
104                 return 0;
105
106         /* Individual destination address:
107          * IEEE 802.11, Ch. 9.6 (after IEEE 802.11g changes)
108          * CTS and ACK frames shall be transmitted using the highest rate in
109          * basic rate set that is less than or equal to the rate of the
110          * immediately previous frame and that is using the same modulation
111          * (CCK or OFDM). If no basic rate set matches with these requirements,
112          * the highest mandatory rate of the PHY that is less than or equal to
113          * the rate of the previous frame is used.
114          * Mandatory rates for IEEE 802.11g PHY: 1, 2, 5.5, 11, 6, 12, 24 Mbps
115          */
116         rate = -1;
117         /* use lowest available if everything fails */
118         mrate = sband->bitrates[0].bitrate;
119         for (i = 0; i < sband->n_bitrates; i++) {
120                 struct ieee80211_rate *r = &sband->bitrates[i];
121
122                 if (r->bitrate > txrate->bitrate)
123                         break;
124
125                 if (tx->sdata->vif.bss_conf.basic_rates & BIT(i))
126                         rate = r->bitrate;
127
128                 switch (sband->band) {
129                 case IEEE80211_BAND_2GHZ: {
130                         u32 flag;
131                         if (tx->sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
132                                 flag = IEEE80211_RATE_MANDATORY_G;
133                         else
134                                 flag = IEEE80211_RATE_MANDATORY_B;
135                         if (r->flags & flag)
136                                 mrate = r->bitrate;
137                         break;
138                 }
139                 case IEEE80211_BAND_5GHZ:
140                         if (r->flags & IEEE80211_RATE_MANDATORY_A)
141                                 mrate = r->bitrate;
142                         break;
143                 case IEEE80211_BAND_60GHZ:
144                         /* TODO, for now fall through */
145                 case IEEE80211_NUM_BANDS:
146                         WARN_ON(1);
147                         break;
148                 }
149         }
150         if (rate == -1) {
151                 /* No matching basic rate found; use highest suitable mandatory
152                  * PHY rate */
153                 rate = mrate;
154         }
155
156         /* Don't calculate ACKs for QoS Frames with NoAck Policy set */
157         if (ieee80211_is_data_qos(hdr->frame_control) &&
158             *(ieee80211_get_qos_ctl(hdr)) & IEEE80211_QOS_CTL_ACK_POLICY_NOACK)
159                 dur = 0;
160         else
161                 /* Time needed to transmit ACK
162                  * (10 bytes + 4-byte FCS = 112 bits) plus SIFS; rounded up
163                  * to closest integer */
164                 dur = ieee80211_frame_duration(sband->band, 10, rate, erp,
165                                 tx->sdata->vif.bss_conf.use_short_preamble);
166
167         if (next_frag_len) {
168                 /* Frame is fragmented: duration increases with time needed to
169                  * transmit next fragment plus ACK and 2 x SIFS. */
170                 dur *= 2; /* ACK + SIFS */
171                 /* next fragment */
172                 dur += ieee80211_frame_duration(sband->band, next_frag_len,
173                                 txrate->bitrate, erp,
174                                 tx->sdata->vif.bss_conf.use_short_preamble);
175         }
176
177         return cpu_to_le16(dur);
178 }
179
180 /* tx handlers */
181 static ieee80211_tx_result debug_noinline
182 ieee80211_tx_h_dynamic_ps(struct ieee80211_tx_data *tx)
183 {
184         struct ieee80211_local *local = tx->local;
185         struct ieee80211_if_managed *ifmgd;
186
187         /* driver doesn't support power save */
188         if (!(local->hw.flags & IEEE80211_HW_SUPPORTS_PS))
189                 return TX_CONTINUE;
190
191         /* hardware does dynamic power save */
192         if (local->hw.flags & IEEE80211_HW_SUPPORTS_DYNAMIC_PS)
193                 return TX_CONTINUE;
194
195         /* dynamic power save disabled */
196         if (local->hw.conf.dynamic_ps_timeout <= 0)
197                 return TX_CONTINUE;
198
199         /* we are scanning, don't enable power save */
200         if (local->scanning)
201                 return TX_CONTINUE;
202
203         if (!local->ps_sdata)
204                 return TX_CONTINUE;
205
206         /* No point if we're going to suspend */
207         if (local->quiescing)
208                 return TX_CONTINUE;
209
210         /* dynamic ps is supported only in managed mode */
211         if (tx->sdata->vif.type != NL80211_IFTYPE_STATION)
212                 return TX_CONTINUE;
213
214         ifmgd = &tx->sdata->u.mgd;
215
216         /*
217          * Don't wakeup from power save if u-apsd is enabled, voip ac has
218          * u-apsd enabled and the frame is in voip class. This effectively
219          * means that even if all access categories have u-apsd enabled, in
220          * practise u-apsd is only used with the voip ac. This is a
221          * workaround for the case when received voip class packets do not
222          * have correct qos tag for some reason, due the network or the
223          * peer application.
224          *
225          * Note: ifmgd->uapsd_queues access is racy here. If the value is
226          * changed via debugfs, user needs to reassociate manually to have
227          * everything in sync.
228          */
229         if ((ifmgd->flags & IEEE80211_STA_UAPSD_ENABLED) &&
230             (ifmgd->uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_VO) &&
231             skb_get_queue_mapping(tx->skb) == IEEE80211_AC_VO)
232                 return TX_CONTINUE;
233
234         if (local->hw.conf.flags & IEEE80211_CONF_PS) {
235                 ieee80211_stop_queues_by_reason(&local->hw,
236                                                 IEEE80211_QUEUE_STOP_REASON_PS);
237                 ifmgd->flags &= ~IEEE80211_STA_NULLFUNC_ACKED;
238                 ieee80211_queue_work(&local->hw,
239                                      &local->dynamic_ps_disable_work);
240         }
241
242         /* Don't restart the timer if we're not disassociated */
243         if (!ifmgd->associated)
244                 return TX_CONTINUE;
245
246         mod_timer(&local->dynamic_ps_timer, jiffies +
247                   msecs_to_jiffies(local->hw.conf.dynamic_ps_timeout));
248
249         return TX_CONTINUE;
250 }
251
252 static ieee80211_tx_result debug_noinline
253 ieee80211_tx_h_check_assoc(struct ieee80211_tx_data *tx)
254 {
255
256         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
257         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
258         bool assoc = false;
259
260         if (unlikely(info->flags & IEEE80211_TX_CTL_INJECTED))
261                 return TX_CONTINUE;
262
263         if (unlikely(test_bit(SCAN_SW_SCANNING, &tx->local->scanning)) &&
264             test_bit(SDATA_STATE_OFFCHANNEL, &tx->sdata->state) &&
265             !ieee80211_is_probe_req(hdr->frame_control) &&
266             !ieee80211_is_nullfunc(hdr->frame_control))
267                 /*
268                  * When software scanning only nullfunc frames (to notify
269                  * the sleep state to the AP) and probe requests (for the
270                  * active scan) are allowed, all other frames should not be
271                  * sent and we should not get here, but if we do
272                  * nonetheless, drop them to avoid sending them
273                  * off-channel. See the link below and
274                  * ieee80211_start_scan() for more.
275                  *
276                  * http://article.gmane.org/gmane.linux.kernel.wireless.general/30089
277                  */
278                 return TX_DROP;
279
280         if (tx->sdata->vif.type == NL80211_IFTYPE_WDS)
281                 return TX_CONTINUE;
282
283         if (tx->sdata->vif.type == NL80211_IFTYPE_MESH_POINT)
284                 return TX_CONTINUE;
285
286         if (tx->flags & IEEE80211_TX_PS_BUFFERED)
287                 return TX_CONTINUE;
288
289         if (tx->sta)
290                 assoc = test_sta_flag(tx->sta, WLAN_STA_ASSOC);
291
292         if (likely(tx->flags & IEEE80211_TX_UNICAST)) {
293                 if (unlikely(!assoc &&
294                              ieee80211_is_data(hdr->frame_control))) {
295 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
296                         sdata_info(tx->sdata,
297                                    "dropped data frame to not associated station %pM\n",
298                                    hdr->addr1);
299 #endif
300                         I802_DEBUG_INC(tx->local->tx_handlers_drop_not_assoc);
301                         return TX_DROP;
302                 }
303         } else if (unlikely(tx->sdata->vif.type == NL80211_IFTYPE_AP &&
304                             ieee80211_is_data(hdr->frame_control) &&
305                             !atomic_read(&tx->sdata->u.ap.num_mcast_sta))) {
306                 /*
307                  * No associated STAs - no need to send multicast
308                  * frames.
309                  */
310                 return TX_DROP;
311         }
312
313         return TX_CONTINUE;
314 }
315
316 /* This function is called whenever the AP is about to exceed the maximum limit
317  * of buffered frames for power saving STAs. This situation should not really
318  * happen often during normal operation, so dropping the oldest buffered packet
319  * from each queue should be OK to make some room for new frames. */
320 static void purge_old_ps_buffers(struct ieee80211_local *local)
321 {
322         int total = 0, purged = 0;
323         struct sk_buff *skb;
324         struct ieee80211_sub_if_data *sdata;
325         struct sta_info *sta;
326
327         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
328                 struct ps_data *ps;
329
330                 if (sdata->vif.type == NL80211_IFTYPE_AP)
331                         ps = &sdata->u.ap.ps;
332                 else if (ieee80211_vif_is_mesh(&sdata->vif))
333                         ps = &sdata->u.mesh.ps;
334                 else
335                         continue;
336
337                 skb = skb_dequeue(&ps->bc_buf);
338                 if (skb) {
339                         purged++;
340                         dev_kfree_skb(skb);
341                 }
342                 total += skb_queue_len(&ps->bc_buf);
343         }
344
345         /*
346          * Drop one frame from each station from the lowest-priority
347          * AC that has frames at all.
348          */
349         list_for_each_entry_rcu(sta, &local->sta_list, list) {
350                 int ac;
351
352                 for (ac = IEEE80211_AC_BK; ac >= IEEE80211_AC_VO; ac--) {
353                         skb = skb_dequeue(&sta->ps_tx_buf[ac]);
354                         total += skb_queue_len(&sta->ps_tx_buf[ac]);
355                         if (skb) {
356                                 purged++;
357                                 ieee80211_free_txskb(&local->hw, skb);
358                                 break;
359                         }
360                 }
361         }
362
363         local->total_ps_buffered = total;
364         ps_dbg_hw(&local->hw, "PS buffers full - purged %d frames\n", purged);
365 }
366
367 static ieee80211_tx_result
368 ieee80211_tx_h_multicast_ps_buf(struct ieee80211_tx_data *tx)
369 {
370         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
371         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
372         struct ps_data *ps;
373
374         /*
375          * broadcast/multicast frame
376          *
377          * If any of the associated/peer stations is in power save mode,
378          * the frame is buffered to be sent after DTIM beacon frame.
379          * This is done either by the hardware or us.
380          */
381
382         /* powersaving STAs currently only in AP/VLAN/mesh mode */
383         if (tx->sdata->vif.type == NL80211_IFTYPE_AP ||
384             tx->sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
385                 if (!tx->sdata->bss)
386                         return TX_CONTINUE;
387
388                 ps = &tx->sdata->bss->ps;
389         } else if (ieee80211_vif_is_mesh(&tx->sdata->vif)) {
390                 ps = &tx->sdata->u.mesh.ps;
391         } else {
392                 return TX_CONTINUE;
393         }
394
395
396         /* no buffering for ordered frames */
397         if (ieee80211_has_order(hdr->frame_control))
398                 return TX_CONTINUE;
399
400         /* no stations in PS mode */
401         if (!atomic_read(&ps->num_sta_ps))
402                 return TX_CONTINUE;
403
404         info->flags |= IEEE80211_TX_CTL_SEND_AFTER_DTIM;
405         if (tx->local->hw.flags & IEEE80211_HW_QUEUE_CONTROL)
406                 info->hw_queue = tx->sdata->vif.cab_queue;
407
408         /* device releases frame after DTIM beacon */
409         if (!(tx->local->hw.flags & IEEE80211_HW_HOST_BROADCAST_PS_BUFFERING))
410                 return TX_CONTINUE;
411
412         /* buffered in mac80211 */
413         if (tx->local->total_ps_buffered >= TOTAL_MAX_TX_BUFFER)
414                 purge_old_ps_buffers(tx->local);
415
416         if (skb_queue_len(&ps->bc_buf) >= AP_MAX_BC_BUFFER) {
417                 ps_dbg(tx->sdata,
418                        "BC TX buffer full - dropping the oldest frame\n");
419                 dev_kfree_skb(skb_dequeue(&ps->bc_buf));
420         } else
421                 tx->local->total_ps_buffered++;
422
423         skb_queue_tail(&ps->bc_buf, tx->skb);
424
425         return TX_QUEUED;
426 }
427
428 static int ieee80211_use_mfp(__le16 fc, struct sta_info *sta,
429                              struct sk_buff *skb)
430 {
431         if (!ieee80211_is_mgmt(fc))
432                 return 0;
433
434         if (sta == NULL || !test_sta_flag(sta, WLAN_STA_MFP))
435                 return 0;
436
437         if (!ieee80211_is_robust_mgmt_frame((struct ieee80211_hdr *)
438                                             skb->data))
439                 return 0;
440
441         return 1;
442 }
443
444 static ieee80211_tx_result
445 ieee80211_tx_h_unicast_ps_buf(struct ieee80211_tx_data *tx)
446 {
447         struct sta_info *sta = tx->sta;
448         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
449         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
450         struct ieee80211_local *local = tx->local;
451
452         if (unlikely(!sta))
453                 return TX_CONTINUE;
454
455         if (unlikely((test_sta_flag(sta, WLAN_STA_PS_STA) ||
456                       test_sta_flag(sta, WLAN_STA_PS_DRIVER)) &&
457                      !(info->flags & IEEE80211_TX_CTL_NO_PS_BUFFER))) {
458                 int ac = skb_get_queue_mapping(tx->skb);
459
460                 /* only deauth, disassoc and action are bufferable MMPDUs */
461                 if (ieee80211_is_mgmt(hdr->frame_control) &&
462                     !ieee80211_is_deauth(hdr->frame_control) &&
463                     !ieee80211_is_disassoc(hdr->frame_control) &&
464                     !ieee80211_is_action(hdr->frame_control)) {
465                         info->flags |= IEEE80211_TX_CTL_NO_PS_BUFFER;
466                         return TX_CONTINUE;
467                 }
468
469                 ps_dbg(sta->sdata, "STA %pM aid %d: PS buffer for AC %d\n",
470                        sta->sta.addr, sta->sta.aid, ac);
471                 if (tx->local->total_ps_buffered >= TOTAL_MAX_TX_BUFFER)
472                         purge_old_ps_buffers(tx->local);
473                 if (skb_queue_len(&sta->ps_tx_buf[ac]) >= STA_MAX_TX_BUFFER) {
474                         struct sk_buff *old = skb_dequeue(&sta->ps_tx_buf[ac]);
475                         ps_dbg(tx->sdata,
476                                "STA %pM TX buffer for AC %d full - dropping oldest frame\n",
477                                sta->sta.addr, ac);
478                         ieee80211_free_txskb(&local->hw, old);
479                 } else
480                         tx->local->total_ps_buffered++;
481
482                 info->control.jiffies = jiffies;
483                 info->control.vif = &tx->sdata->vif;
484                 info->flags |= IEEE80211_TX_INTFL_NEED_TXPROCESSING;
485                 skb_queue_tail(&sta->ps_tx_buf[ac], tx->skb);
486
487                 if (!timer_pending(&local->sta_cleanup))
488                         mod_timer(&local->sta_cleanup,
489                                   round_jiffies(jiffies +
490                                                 STA_INFO_CLEANUP_INTERVAL));
491
492                 /*
493                  * We queued up some frames, so the TIM bit might
494                  * need to be set, recalculate it.
495                  */
496                 sta_info_recalc_tim(sta);
497
498                 return TX_QUEUED;
499         } else if (unlikely(test_sta_flag(sta, WLAN_STA_PS_STA))) {
500                 ps_dbg(tx->sdata,
501                        "STA %pM in PS mode, but polling/in SP -> send frame\n",
502                        sta->sta.addr);
503         }
504
505         return TX_CONTINUE;
506 }
507
508 static ieee80211_tx_result debug_noinline
509 ieee80211_tx_h_ps_buf(struct ieee80211_tx_data *tx)
510 {
511         if (unlikely(tx->flags & IEEE80211_TX_PS_BUFFERED))
512                 return TX_CONTINUE;
513
514         if (tx->flags & IEEE80211_TX_UNICAST)
515                 return ieee80211_tx_h_unicast_ps_buf(tx);
516         else
517                 return ieee80211_tx_h_multicast_ps_buf(tx);
518 }
519
520 static ieee80211_tx_result debug_noinline
521 ieee80211_tx_h_check_control_port_protocol(struct ieee80211_tx_data *tx)
522 {
523         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
524
525         if (unlikely(tx->sdata->control_port_protocol == tx->skb->protocol &&
526                      tx->sdata->control_port_no_encrypt))
527                 info->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
528
529         return TX_CONTINUE;
530 }
531
532 static ieee80211_tx_result debug_noinline
533 ieee80211_tx_h_select_key(struct ieee80211_tx_data *tx)
534 {
535         struct ieee80211_key *key;
536         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
537         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
538
539         if (unlikely(info->flags & IEEE80211_TX_INTFL_DONT_ENCRYPT))
540                 tx->key = NULL;
541         else if (tx->sta && (key = rcu_dereference(tx->sta->ptk)))
542                 tx->key = key;
543         else if (ieee80211_is_mgmt(hdr->frame_control) &&
544                  is_multicast_ether_addr(hdr->addr1) &&
545                  ieee80211_is_robust_mgmt_frame(hdr) &&
546                  (key = rcu_dereference(tx->sdata->default_mgmt_key)))
547                 tx->key = key;
548         else if (is_multicast_ether_addr(hdr->addr1) &&
549                  (key = rcu_dereference(tx->sdata->default_multicast_key)))
550                 tx->key = key;
551         else if (!is_multicast_ether_addr(hdr->addr1) &&
552                  (key = rcu_dereference(tx->sdata->default_unicast_key)))
553                 tx->key = key;
554         else if (info->flags & IEEE80211_TX_CTL_INJECTED)
555                 tx->key = NULL;
556         else if (!tx->sdata->drop_unencrypted)
557                 tx->key = NULL;
558         else if (tx->skb->protocol == tx->sdata->control_port_protocol)
559                 tx->key = NULL;
560         else if (ieee80211_is_robust_mgmt_frame(hdr) &&
561                  !(ieee80211_is_action(hdr->frame_control) &&
562                    tx->sta && test_sta_flag(tx->sta, WLAN_STA_MFP)))
563                 tx->key = NULL;
564         else if (ieee80211_is_mgmt(hdr->frame_control) &&
565                  !ieee80211_is_robust_mgmt_frame(hdr))
566                 tx->key = NULL;
567         else {
568                 I802_DEBUG_INC(tx->local->tx_handlers_drop_unencrypted);
569                 return TX_DROP;
570         }
571
572         if (tx->key) {
573                 bool skip_hw = false;
574
575                 tx->key->tx_rx_count++;
576                 /* TODO: add threshold stuff again */
577
578                 switch (tx->key->conf.cipher) {
579                 case WLAN_CIPHER_SUITE_WEP40:
580                 case WLAN_CIPHER_SUITE_WEP104:
581                 case WLAN_CIPHER_SUITE_TKIP:
582                         if (!ieee80211_is_data_present(hdr->frame_control))
583                                 tx->key = NULL;
584                         break;
585                 case WLAN_CIPHER_SUITE_CCMP:
586                         if (!ieee80211_is_data_present(hdr->frame_control) &&
587                             !ieee80211_use_mfp(hdr->frame_control, tx->sta,
588                                                tx->skb))
589                                 tx->key = NULL;
590                         else
591                                 skip_hw = (tx->key->conf.flags &
592                                            IEEE80211_KEY_FLAG_SW_MGMT_TX) &&
593                                         ieee80211_is_mgmt(hdr->frame_control);
594                         break;
595                 case WLAN_CIPHER_SUITE_AES_CMAC:
596                         if (!ieee80211_is_mgmt(hdr->frame_control))
597                                 tx->key = NULL;
598                         break;
599                 }
600
601                 if (unlikely(tx->key && tx->key->flags & KEY_FLAG_TAINTED &&
602                              !ieee80211_is_deauth(hdr->frame_control)))
603                         return TX_DROP;
604
605                 if (!skip_hw && tx->key &&
606                     tx->key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE)
607                         info->control.hw_key = &tx->key->conf;
608         }
609
610         return TX_CONTINUE;
611 }
612
613 static ieee80211_tx_result debug_noinline
614 ieee80211_tx_h_rate_ctrl(struct ieee80211_tx_data *tx)
615 {
616         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
617         struct ieee80211_hdr *hdr = (void *)tx->skb->data;
618         struct ieee80211_supported_band *sband;
619         struct ieee80211_rate *rate;
620         int i;
621         u32 len;
622         bool inval = false, rts = false, short_preamble = false;
623         struct ieee80211_tx_rate_control txrc;
624         bool assoc = false;
625
626         memset(&txrc, 0, sizeof(txrc));
627
628         sband = tx->local->hw.wiphy->bands[info->band];
629
630         len = min_t(u32, tx->skb->len + FCS_LEN,
631                          tx->local->hw.wiphy->frag_threshold);
632
633         /* set up the tx rate control struct we give the RC algo */
634         txrc.hw = &tx->local->hw;
635         txrc.sband = sband;
636         txrc.bss_conf = &tx->sdata->vif.bss_conf;
637         txrc.skb = tx->skb;
638         txrc.reported_rate.idx = -1;
639         txrc.rate_idx_mask = tx->sdata->rc_rateidx_mask[info->band];
640         if (txrc.rate_idx_mask == (1 << sband->n_bitrates) - 1)
641                 txrc.max_rate_idx = -1;
642         else
643                 txrc.max_rate_idx = fls(txrc.rate_idx_mask) - 1;
644         memcpy(txrc.rate_idx_mcs_mask,
645                tx->sdata->rc_rateidx_mcs_mask[info->band],
646                sizeof(txrc.rate_idx_mcs_mask));
647         txrc.bss = (tx->sdata->vif.type == NL80211_IFTYPE_AP ||
648                     tx->sdata->vif.type == NL80211_IFTYPE_MESH_POINT ||
649                     tx->sdata->vif.type == NL80211_IFTYPE_ADHOC);
650
651         /* set up RTS protection if desired */
652         if (len > tx->local->hw.wiphy->rts_threshold) {
653                 txrc.rts = rts = true;
654         }
655
656         /*
657          * Use short preamble if the BSS can handle it, but not for
658          * management frames unless we know the receiver can handle
659          * that -- the management frame might be to a station that
660          * just wants a probe response.
661          */
662         if (tx->sdata->vif.bss_conf.use_short_preamble &&
663             (ieee80211_is_data(hdr->frame_control) ||
664              (tx->sta && test_sta_flag(tx->sta, WLAN_STA_SHORT_PREAMBLE))))
665                 txrc.short_preamble = short_preamble = true;
666
667         if (tx->sta)
668                 assoc = test_sta_flag(tx->sta, WLAN_STA_ASSOC);
669
670         /*
671          * Lets not bother rate control if we're associated and cannot
672          * talk to the sta. This should not happen.
673          */
674         if (WARN(test_bit(SCAN_SW_SCANNING, &tx->local->scanning) && assoc &&
675                  !rate_usable_index_exists(sband, &tx->sta->sta),
676                  "%s: Dropped data frame as no usable bitrate found while "
677                  "scanning and associated. Target station: "
678                  "%pM on %d GHz band\n",
679                  tx->sdata->name, hdr->addr1,
680                  info->band ? 5 : 2))
681                 return TX_DROP;
682
683         /*
684          * If we're associated with the sta at this point we know we can at
685          * least send the frame at the lowest bit rate.
686          */
687         rate_control_get_rate(tx->sdata, tx->sta, &txrc);
688
689         if (unlikely(info->control.rates[0].idx < 0))
690                 return TX_DROP;
691
692         if (txrc.reported_rate.idx < 0) {
693                 txrc.reported_rate = info->control.rates[0];
694                 if (tx->sta && ieee80211_is_data(hdr->frame_control))
695                         tx->sta->last_tx_rate = txrc.reported_rate;
696         } else if (tx->sta)
697                 tx->sta->last_tx_rate = txrc.reported_rate;
698
699         if (unlikely(!info->control.rates[0].count))
700                 info->control.rates[0].count = 1;
701
702         if (WARN_ON_ONCE((info->control.rates[0].count > 1) &&
703                          (info->flags & IEEE80211_TX_CTL_NO_ACK)))
704                 info->control.rates[0].count = 1;
705
706         if (is_multicast_ether_addr(hdr->addr1)) {
707                 /*
708                  * XXX: verify the rate is in the basic rateset
709                  */
710                 return TX_CONTINUE;
711         }
712
713         /*
714          * set up the RTS/CTS rate as the fastest basic rate
715          * that is not faster than the data rate
716          *
717          * XXX: Should this check all retry rates?
718          */
719         if (!(info->control.rates[0].flags & IEEE80211_TX_RC_MCS)) {
720                 s8 baserate = 0;
721
722                 rate = &sband->bitrates[info->control.rates[0].idx];
723
724                 for (i = 0; i < sband->n_bitrates; i++) {
725                         /* must be a basic rate */
726                         if (!(tx->sdata->vif.bss_conf.basic_rates & BIT(i)))
727                                 continue;
728                         /* must not be faster than the data rate */
729                         if (sband->bitrates[i].bitrate > rate->bitrate)
730                                 continue;
731                         /* maximum */
732                         if (sband->bitrates[baserate].bitrate <
733                              sband->bitrates[i].bitrate)
734                                 baserate = i;
735                 }
736
737                 info->control.rts_cts_rate_idx = baserate;
738         }
739
740         for (i = 0; i < IEEE80211_TX_MAX_RATES; i++) {
741                 /*
742                  * make sure there's no valid rate following
743                  * an invalid one, just in case drivers don't
744                  * take the API seriously to stop at -1.
745                  */
746                 if (inval) {
747                         info->control.rates[i].idx = -1;
748                         continue;
749                 }
750                 if (info->control.rates[i].idx < 0) {
751                         inval = true;
752                         continue;
753                 }
754
755                 /*
756                  * For now assume MCS is already set up correctly, this
757                  * needs to be fixed.
758                  */
759                 if (info->control.rates[i].flags & IEEE80211_TX_RC_MCS) {
760                         WARN_ON(info->control.rates[i].idx > 76);
761                         continue;
762                 }
763
764                 /* set up RTS protection if desired */
765                 if (rts)
766                         info->control.rates[i].flags |=
767                                 IEEE80211_TX_RC_USE_RTS_CTS;
768
769                 /* RC is busted */
770                 if (WARN_ON_ONCE(info->control.rates[i].idx >=
771                                  sband->n_bitrates)) {
772                         info->control.rates[i].idx = -1;
773                         continue;
774                 }
775
776                 rate = &sband->bitrates[info->control.rates[i].idx];
777
778                 /* set up short preamble */
779                 if (short_preamble &&
780                     rate->flags & IEEE80211_RATE_SHORT_PREAMBLE)
781                         info->control.rates[i].flags |=
782                                 IEEE80211_TX_RC_USE_SHORT_PREAMBLE;
783
784                 /* set up G protection */
785                 if (!rts && tx->sdata->vif.bss_conf.use_cts_prot &&
786                     rate->flags & IEEE80211_RATE_ERP_G)
787                         info->control.rates[i].flags |=
788                                 IEEE80211_TX_RC_USE_CTS_PROTECT;
789         }
790
791         return TX_CONTINUE;
792 }
793
794 static ieee80211_tx_result debug_noinline
795 ieee80211_tx_h_sequence(struct ieee80211_tx_data *tx)
796 {
797         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
798         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
799         u16 *seq;
800         u8 *qc;
801         int tid;
802
803         /*
804          * Packet injection may want to control the sequence
805          * number, if we have no matching interface then we
806          * neither assign one ourselves nor ask the driver to.
807          */
808         if (unlikely(info->control.vif->type == NL80211_IFTYPE_MONITOR))
809                 return TX_CONTINUE;
810
811         if (unlikely(ieee80211_is_ctl(hdr->frame_control)))
812                 return TX_CONTINUE;
813
814         if (ieee80211_hdrlen(hdr->frame_control) < 24)
815                 return TX_CONTINUE;
816
817         if (ieee80211_is_qos_nullfunc(hdr->frame_control))
818                 return TX_CONTINUE;
819
820         /*
821          * Anything but QoS data that has a sequence number field
822          * (is long enough) gets a sequence number from the global
823          * counter.
824          */
825         if (!ieee80211_is_data_qos(hdr->frame_control)) {
826                 /* driver should assign sequence number */
827                 info->flags |= IEEE80211_TX_CTL_ASSIGN_SEQ;
828                 /* for pure STA mode without beacons, we can do it */
829                 hdr->seq_ctrl = cpu_to_le16(tx->sdata->sequence_number);
830                 tx->sdata->sequence_number += 0x10;
831                 return TX_CONTINUE;
832         }
833
834         /*
835          * This should be true for injected/management frames only, for
836          * management frames we have set the IEEE80211_TX_CTL_ASSIGN_SEQ
837          * above since they are not QoS-data frames.
838          */
839         if (!tx->sta)
840                 return TX_CONTINUE;
841
842         /* include per-STA, per-TID sequence counter */
843
844         qc = ieee80211_get_qos_ctl(hdr);
845         tid = *qc & IEEE80211_QOS_CTL_TID_MASK;
846         seq = &tx->sta->tid_seq[tid];
847
848         hdr->seq_ctrl = cpu_to_le16(*seq);
849
850         /* Increase the sequence number. */
851         *seq = (*seq + 0x10) & IEEE80211_SCTL_SEQ;
852
853         return TX_CONTINUE;
854 }
855
856 static int ieee80211_fragment(struct ieee80211_tx_data *tx,
857                               struct sk_buff *skb, int hdrlen,
858                               int frag_threshold)
859 {
860         struct ieee80211_local *local = tx->local;
861         struct ieee80211_tx_info *info;
862         struct sk_buff *tmp;
863         int per_fragm = frag_threshold - hdrlen - FCS_LEN;
864         int pos = hdrlen + per_fragm;
865         int rem = skb->len - hdrlen - per_fragm;
866
867         if (WARN_ON(rem < 0))
868                 return -EINVAL;
869
870         /* first fragment was already added to queue by caller */
871
872         while (rem) {
873                 int fraglen = per_fragm;
874
875                 if (fraglen > rem)
876                         fraglen = rem;
877                 rem -= fraglen;
878                 tmp = dev_alloc_skb(local->tx_headroom +
879                                     frag_threshold +
880                                     IEEE80211_ENCRYPT_HEADROOM +
881                                     IEEE80211_ENCRYPT_TAILROOM);
882                 if (!tmp)
883                         return -ENOMEM;
884
885                 __skb_queue_tail(&tx->skbs, tmp);
886
887                 skb_reserve(tmp, local->tx_headroom +
888                                  IEEE80211_ENCRYPT_HEADROOM);
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->len = 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
995         if (!tx->sta)
996                 return TX_CONTINUE;
997
998         tx->sta->tx_packets++;
999         skb_queue_walk(&tx->skbs, skb) {
1000                 tx->sta->tx_fragments++;
1001                 tx->sta->tx_bytes += skb->len;
1002         }
1003
1004         return TX_CONTINUE;
1005 }
1006
1007 static ieee80211_tx_result debug_noinline
1008 ieee80211_tx_h_encrypt(struct ieee80211_tx_data *tx)
1009 {
1010         if (!tx->key)
1011                 return TX_CONTINUE;
1012
1013         switch (tx->key->conf.cipher) {
1014         case WLAN_CIPHER_SUITE_WEP40:
1015         case WLAN_CIPHER_SUITE_WEP104:
1016                 return ieee80211_crypto_wep_encrypt(tx);
1017         case WLAN_CIPHER_SUITE_TKIP:
1018                 return ieee80211_crypto_tkip_encrypt(tx);
1019         case WLAN_CIPHER_SUITE_CCMP:
1020                 return ieee80211_crypto_ccmp_encrypt(tx);
1021         case WLAN_CIPHER_SUITE_AES_CMAC:
1022                 return ieee80211_crypto_aes_cmac_encrypt(tx);
1023         default:
1024                 return ieee80211_crypto_hw_encrypt(tx);
1025         }
1026
1027         return TX_DROP;
1028 }
1029
1030 static ieee80211_tx_result debug_noinline
1031 ieee80211_tx_h_calculate_duration(struct ieee80211_tx_data *tx)
1032 {
1033         struct sk_buff *skb;
1034         struct ieee80211_hdr *hdr;
1035         int next_len;
1036         bool group_addr;
1037
1038         skb_queue_walk(&tx->skbs, skb) {
1039                 hdr = (void *) skb->data;
1040                 if (unlikely(ieee80211_is_pspoll(hdr->frame_control)))
1041                         break; /* must not overwrite AID */
1042                 if (!skb_queue_is_last(&tx->skbs, skb)) {
1043                         struct sk_buff *next = skb_queue_next(&tx->skbs, skb);
1044                         next_len = next->len;
1045                 } else
1046                         next_len = 0;
1047                 group_addr = is_multicast_ether_addr(hdr->addr1);
1048
1049                 hdr->duration_id =
1050                         ieee80211_duration(tx, skb, group_addr, next_len);
1051         }
1052
1053         return TX_CONTINUE;
1054 }
1055
1056 /* actual transmit path */
1057
1058 static bool ieee80211_tx_prep_agg(struct ieee80211_tx_data *tx,
1059                                   struct sk_buff *skb,
1060                                   struct ieee80211_tx_info *info,
1061                                   struct tid_ampdu_tx *tid_tx,
1062                                   int tid)
1063 {
1064         bool queued = false;
1065         bool reset_agg_timer = false;
1066         struct sk_buff *purge_skb = NULL;
1067
1068         if (test_bit(HT_AGG_STATE_OPERATIONAL, &tid_tx->state)) {
1069                 info->flags |= IEEE80211_TX_CTL_AMPDU;
1070                 reset_agg_timer = true;
1071         } else if (test_bit(HT_AGG_STATE_WANT_START, &tid_tx->state)) {
1072                 /*
1073                  * nothing -- this aggregation session is being started
1074                  * but that might still fail with the driver
1075                  */
1076         } else {
1077                 spin_lock(&tx->sta->lock);
1078                 /*
1079                  * Need to re-check now, because we may get here
1080                  *
1081                  *  1) in the window during which the setup is actually
1082                  *     already done, but not marked yet because not all
1083                  *     packets are spliced over to the driver pending
1084                  *     queue yet -- if this happened we acquire the lock
1085                  *     either before or after the splice happens, but
1086                  *     need to recheck which of these cases happened.
1087                  *
1088                  *  2) during session teardown, if the OPERATIONAL bit
1089                  *     was cleared due to the teardown but the pointer
1090                  *     hasn't been assigned NULL yet (or we loaded it
1091                  *     before it was assigned) -- in this case it may
1092                  *     now be NULL which means we should just let the
1093                  *     packet pass through because splicing the frames
1094                  *     back is already done.
1095                  */
1096                 tid_tx = rcu_dereference_protected_tid_tx(tx->sta, tid);
1097
1098                 if (!tid_tx) {
1099                         /* do nothing, let packet pass through */
1100                 } else if (test_bit(HT_AGG_STATE_OPERATIONAL, &tid_tx->state)) {
1101                         info->flags |= IEEE80211_TX_CTL_AMPDU;
1102                         reset_agg_timer = true;
1103                 } else {
1104                         queued = true;
1105                         info->control.vif = &tx->sdata->vif;
1106                         info->flags |= IEEE80211_TX_INTFL_NEED_TXPROCESSING;
1107                         __skb_queue_tail(&tid_tx->pending, skb);
1108                         if (skb_queue_len(&tid_tx->pending) > STA_MAX_TX_BUFFER)
1109                                 purge_skb = __skb_dequeue(&tid_tx->pending);
1110                 }
1111                 spin_unlock(&tx->sta->lock);
1112
1113                 if (purge_skb)
1114                         ieee80211_free_txskb(&tx->local->hw, purge_skb);
1115         }
1116
1117         /* reset session timer */
1118         if (reset_agg_timer && tid_tx->timeout)
1119                 tid_tx->last_tx = jiffies;
1120
1121         return queued;
1122 }
1123
1124 /*
1125  * initialises @tx
1126  */
1127 static ieee80211_tx_result
1128 ieee80211_tx_prepare(struct ieee80211_sub_if_data *sdata,
1129                      struct ieee80211_tx_data *tx,
1130                      struct sk_buff *skb)
1131 {
1132         struct ieee80211_local *local = sdata->local;
1133         struct ieee80211_hdr *hdr;
1134         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1135         int tid;
1136         u8 *qc;
1137
1138         memset(tx, 0, sizeof(*tx));
1139         tx->skb = skb;
1140         tx->local = local;
1141         tx->sdata = sdata;
1142         __skb_queue_head_init(&tx->skbs);
1143
1144         /*
1145          * If this flag is set to true anywhere, and we get here,
1146          * we are doing the needed processing, so remove the flag
1147          * now.
1148          */
1149         info->flags &= ~IEEE80211_TX_INTFL_NEED_TXPROCESSING;
1150
1151         hdr = (struct ieee80211_hdr *) skb->data;
1152
1153         if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
1154                 tx->sta = rcu_dereference(sdata->u.vlan.sta);
1155                 if (!tx->sta && sdata->dev->ieee80211_ptr->use_4addr)
1156                         return TX_DROP;
1157         } else if (info->flags & IEEE80211_TX_CTL_INJECTED ||
1158                    tx->sdata->control_port_protocol == tx->skb->protocol) {
1159                 tx->sta = sta_info_get_bss(sdata, hdr->addr1);
1160         }
1161         if (!tx->sta)
1162                 tx->sta = sta_info_get(sdata, hdr->addr1);
1163
1164         if (tx->sta && ieee80211_is_data_qos(hdr->frame_control) &&
1165             !ieee80211_is_qos_nullfunc(hdr->frame_control) &&
1166             (local->hw.flags & IEEE80211_HW_AMPDU_AGGREGATION) &&
1167             !(local->hw.flags & IEEE80211_HW_TX_AMPDU_SETUP_IN_HW)) {
1168                 struct tid_ampdu_tx *tid_tx;
1169
1170                 qc = ieee80211_get_qos_ctl(hdr);
1171                 tid = *qc & IEEE80211_QOS_CTL_TID_MASK;
1172
1173                 tid_tx = rcu_dereference(tx->sta->ampdu_mlme.tid_tx[tid]);
1174                 if (tid_tx) {
1175                         bool queued;
1176
1177                         queued = ieee80211_tx_prep_agg(tx, skb, info,
1178                                                        tid_tx, tid);
1179
1180                         if (unlikely(queued))
1181                                 return TX_QUEUED;
1182                 }
1183         }
1184
1185         if (is_multicast_ether_addr(hdr->addr1)) {
1186                 tx->flags &= ~IEEE80211_TX_UNICAST;
1187                 info->flags |= IEEE80211_TX_CTL_NO_ACK;
1188         } else
1189                 tx->flags |= IEEE80211_TX_UNICAST;
1190
1191         if (!(info->flags & IEEE80211_TX_CTL_DONTFRAG)) {
1192                 if (!(tx->flags & IEEE80211_TX_UNICAST) ||
1193                     skb->len + FCS_LEN <= local->hw.wiphy->frag_threshold ||
1194                     info->flags & IEEE80211_TX_CTL_AMPDU)
1195                         info->flags |= IEEE80211_TX_CTL_DONTFRAG;
1196         }
1197
1198         if (!tx->sta)
1199                 info->flags |= IEEE80211_TX_CTL_CLEAR_PS_FILT;
1200         else if (test_and_clear_sta_flag(tx->sta, WLAN_STA_CLEAR_PS_FILT))
1201                 info->flags |= IEEE80211_TX_CTL_CLEAR_PS_FILT;
1202
1203         info->flags |= IEEE80211_TX_CTL_FIRST_FRAGMENT;
1204
1205         return TX_CONTINUE;
1206 }
1207
1208 static bool ieee80211_tx_frags(struct ieee80211_local *local,
1209                                struct ieee80211_vif *vif,
1210                                struct ieee80211_sta *sta,
1211                                struct sk_buff_head *skbs,
1212                                bool txpending)
1213 {
1214         struct ieee80211_tx_control control;
1215         struct sk_buff *skb, *tmp;
1216         unsigned long flags;
1217
1218         skb_queue_walk_safe(skbs, skb, tmp) {
1219                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1220                 int q = info->hw_queue;
1221
1222 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
1223                 if (WARN_ON_ONCE(q >= local->hw.queues)) {
1224                         __skb_unlink(skb, skbs);
1225                         ieee80211_free_txskb(&local->hw, skb);
1226                         continue;
1227                 }
1228 #endif
1229
1230                 spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
1231                 if (local->queue_stop_reasons[q] ||
1232                     (!txpending && !skb_queue_empty(&local->pending[q]))) {
1233                         /*
1234                          * Since queue is stopped, queue up frames for later
1235                          * transmission from the tx-pending tasklet when the
1236                          * queue is woken again.
1237                          */
1238                         if (txpending)
1239                                 skb_queue_splice_init(skbs, &local->pending[q]);
1240                         else
1241                                 skb_queue_splice_tail_init(skbs,
1242                                                            &local->pending[q]);
1243
1244                         spin_unlock_irqrestore(&local->queue_stop_reason_lock,
1245                                                flags);
1246                         return false;
1247                 }
1248                 spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
1249
1250                 info->control.vif = vif;
1251                 control.sta = sta;
1252
1253                 __skb_unlink(skb, skbs);
1254                 drv_tx(local, &control, skb);
1255         }
1256
1257         return true;
1258 }
1259
1260 /*
1261  * Returns false if the frame couldn't be transmitted but was queued instead.
1262  */
1263 static bool __ieee80211_tx(struct ieee80211_local *local,
1264                            struct sk_buff_head *skbs, int led_len,
1265                            struct sta_info *sta, bool txpending)
1266 {
1267         struct ieee80211_tx_info *info;
1268         struct ieee80211_sub_if_data *sdata;
1269         struct ieee80211_vif *vif;
1270         struct ieee80211_sta *pubsta;
1271         struct sk_buff *skb;
1272         bool result = true;
1273         __le16 fc;
1274
1275         if (WARN_ON(skb_queue_empty(skbs)))
1276                 return true;
1277
1278         skb = skb_peek(skbs);
1279         fc = ((struct ieee80211_hdr *)skb->data)->frame_control;
1280         info = IEEE80211_SKB_CB(skb);
1281         sdata = vif_to_sdata(info->control.vif);
1282         if (sta && !sta->uploaded)
1283                 sta = NULL;
1284
1285         if (sta)
1286                 pubsta = &sta->sta;
1287         else
1288                 pubsta = NULL;
1289
1290         switch (sdata->vif.type) {
1291         case NL80211_IFTYPE_MONITOR:
1292                 sdata = rcu_dereference(local->monitor_sdata);
1293                 if (sdata) {
1294                         vif = &sdata->vif;
1295                         info->hw_queue =
1296                                 vif->hw_queue[skb_get_queue_mapping(skb)];
1297                 } else if (local->hw.flags & IEEE80211_HW_QUEUE_CONTROL) {
1298                         dev_kfree_skb(skb);
1299                         return true;
1300                 } else
1301                         vif = NULL;
1302                 break;
1303         case NL80211_IFTYPE_AP_VLAN:
1304                 sdata = container_of(sdata->bss,
1305                                      struct ieee80211_sub_if_data, u.ap);
1306                 /* fall through */
1307         default:
1308                 vif = &sdata->vif;
1309                 break;
1310         }
1311
1312         result = ieee80211_tx_frags(local, vif, pubsta, skbs,
1313                                     txpending);
1314
1315         ieee80211_tpt_led_trig_tx(local, fc, led_len);
1316         ieee80211_led_tx(local, 1);
1317
1318         WARN_ON_ONCE(!skb_queue_empty(skbs));
1319
1320         return result;
1321 }
1322
1323 /*
1324  * Invoke TX handlers, return 0 on success and non-zero if the
1325  * frame was dropped or queued.
1326  */
1327 static int invoke_tx_handlers(struct ieee80211_tx_data *tx)
1328 {
1329         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
1330         ieee80211_tx_result res = TX_DROP;
1331
1332 #define CALL_TXH(txh) \
1333         do {                            \
1334                 res = txh(tx);          \
1335                 if (res != TX_CONTINUE) \
1336                         goto txh_done;  \
1337         } while (0)
1338
1339         CALL_TXH(ieee80211_tx_h_dynamic_ps);
1340         CALL_TXH(ieee80211_tx_h_check_assoc);
1341         CALL_TXH(ieee80211_tx_h_ps_buf);
1342         CALL_TXH(ieee80211_tx_h_check_control_port_protocol);
1343         CALL_TXH(ieee80211_tx_h_select_key);
1344         if (!(tx->local->hw.flags & IEEE80211_HW_HAS_RATE_CONTROL))
1345                 CALL_TXH(ieee80211_tx_h_rate_ctrl);
1346
1347         if (unlikely(info->flags & IEEE80211_TX_INTFL_RETRANSMISSION)) {
1348                 __skb_queue_tail(&tx->skbs, tx->skb);
1349                 tx->skb = NULL;
1350                 goto txh_done;
1351         }
1352
1353         CALL_TXH(ieee80211_tx_h_michael_mic_add);
1354         CALL_TXH(ieee80211_tx_h_sequence);
1355         CALL_TXH(ieee80211_tx_h_fragment);
1356         /* handlers after fragment must be aware of tx info fragmentation! */
1357         CALL_TXH(ieee80211_tx_h_stats);
1358         CALL_TXH(ieee80211_tx_h_encrypt);
1359         if (!(tx->local->hw.flags & IEEE80211_HW_HAS_RATE_CONTROL))
1360                 CALL_TXH(ieee80211_tx_h_calculate_duration);
1361 #undef CALL_TXH
1362
1363  txh_done:
1364         if (unlikely(res == TX_DROP)) {
1365                 I802_DEBUG_INC(tx->local->tx_handlers_drop);
1366                 if (tx->skb)
1367                         ieee80211_free_txskb(&tx->local->hw, tx->skb);
1368                 else
1369                         ieee80211_purge_tx_queue(&tx->local->hw, &tx->skbs);
1370                 return -1;
1371         } else if (unlikely(res == TX_QUEUED)) {
1372                 I802_DEBUG_INC(tx->local->tx_handlers_queued);
1373                 return -1;
1374         }
1375
1376         return 0;
1377 }
1378
1379 /*
1380  * Returns false if the frame couldn't be transmitted but was queued instead.
1381  */
1382 static bool ieee80211_tx(struct ieee80211_sub_if_data *sdata,
1383                          struct sk_buff *skb, bool txpending,
1384                          enum ieee80211_band band)
1385 {
1386         struct ieee80211_local *local = sdata->local;
1387         struct ieee80211_tx_data tx;
1388         ieee80211_tx_result res_prepare;
1389         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1390         bool result = true;
1391         int led_len;
1392
1393         if (unlikely(skb->len < 10)) {
1394                 dev_kfree_skb(skb);
1395                 return true;
1396         }
1397
1398         /* initialises tx */
1399         led_len = skb->len;
1400         res_prepare = ieee80211_tx_prepare(sdata, &tx, skb);
1401
1402         if (unlikely(res_prepare == TX_DROP)) {
1403                 ieee80211_free_txskb(&local->hw, skb);
1404                 return true;
1405         } else if (unlikely(res_prepare == TX_QUEUED)) {
1406                 return true;
1407         }
1408
1409         info->band = band;
1410
1411         /* set up hw_queue value early */
1412         if (!(info->flags & IEEE80211_TX_CTL_TX_OFFCHAN) ||
1413             !(local->hw.flags & IEEE80211_HW_QUEUE_CONTROL))
1414                 info->hw_queue =
1415                         sdata->vif.hw_queue[skb_get_queue_mapping(skb)];
1416
1417         if (!invoke_tx_handlers(&tx))
1418                 result = __ieee80211_tx(local, &tx.skbs, led_len,
1419                                         tx.sta, txpending);
1420
1421         return result;
1422 }
1423
1424 /* device xmit handlers */
1425
1426 static int ieee80211_skb_resize(struct ieee80211_sub_if_data *sdata,
1427                                 struct sk_buff *skb,
1428                                 int head_need, bool may_encrypt)
1429 {
1430         struct ieee80211_local *local = sdata->local;
1431         int tail_need = 0;
1432
1433         if (may_encrypt && sdata->crypto_tx_tailroom_needed_cnt) {
1434                 tail_need = IEEE80211_ENCRYPT_TAILROOM;
1435                 tail_need -= skb_tailroom(skb);
1436                 tail_need = max_t(int, tail_need, 0);
1437         }
1438
1439         if (skb_cloned(skb))
1440                 I802_DEBUG_INC(local->tx_expand_skb_head_cloned);
1441         else if (head_need || tail_need)
1442                 I802_DEBUG_INC(local->tx_expand_skb_head);
1443         else
1444                 return 0;
1445
1446         if (pskb_expand_head(skb, head_need, tail_need, GFP_ATOMIC)) {
1447                 wiphy_debug(local->hw.wiphy,
1448                             "failed to reallocate TX buffer\n");
1449                 return -ENOMEM;
1450         }
1451
1452         return 0;
1453 }
1454
1455 void ieee80211_xmit(struct ieee80211_sub_if_data *sdata, struct sk_buff *skb,
1456                     enum ieee80211_band band)
1457 {
1458         struct ieee80211_local *local = sdata->local;
1459         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1460         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1461         int headroom;
1462         bool may_encrypt;
1463
1464         may_encrypt = !(info->flags & IEEE80211_TX_INTFL_DONT_ENCRYPT);
1465
1466         headroom = local->tx_headroom;
1467         if (may_encrypt)
1468                 headroom += IEEE80211_ENCRYPT_HEADROOM;
1469         headroom -= skb_headroom(skb);
1470         headroom = max_t(int, 0, headroom);
1471
1472         if (ieee80211_skb_resize(sdata, skb, headroom, may_encrypt)) {
1473                 ieee80211_free_txskb(&local->hw, skb);
1474                 return;
1475         }
1476
1477         hdr = (struct ieee80211_hdr *) skb->data;
1478         info->control.vif = &sdata->vif;
1479
1480         if (ieee80211_vif_is_mesh(&sdata->vif)) {
1481                 if (ieee80211_is_data(hdr->frame_control) &&
1482                     is_unicast_ether_addr(hdr->addr1)) {
1483                         if (mesh_nexthop_resolve(skb, sdata))
1484                                 return; /* skb queued: don't free */
1485                 } else {
1486                         ieee80211_mps_set_frame_flags(sdata, NULL, hdr);
1487                 }
1488         }
1489
1490         ieee80211_set_qos_hdr(sdata, skb);
1491         ieee80211_tx(sdata, skb, false, band);
1492 }
1493
1494 static bool ieee80211_parse_tx_radiotap(struct sk_buff *skb)
1495 {
1496         struct ieee80211_radiotap_iterator iterator;
1497         struct ieee80211_radiotap_header *rthdr =
1498                 (struct ieee80211_radiotap_header *) skb->data;
1499         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1500         int ret = ieee80211_radiotap_iterator_init(&iterator, rthdr, skb->len,
1501                                                    NULL);
1502         u16 txflags;
1503
1504         info->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT |
1505                        IEEE80211_TX_CTL_DONTFRAG;
1506
1507         /*
1508          * for every radiotap entry that is present
1509          * (ieee80211_radiotap_iterator_next returns -ENOENT when no more
1510          * entries present, or -EINVAL on error)
1511          */
1512
1513         while (!ret) {
1514                 ret = ieee80211_radiotap_iterator_next(&iterator);
1515
1516                 if (ret)
1517                         continue;
1518
1519                 /* see if this argument is something we can use */
1520                 switch (iterator.this_arg_index) {
1521                 /*
1522                  * You must take care when dereferencing iterator.this_arg
1523                  * for multibyte types... the pointer is not aligned.  Use
1524                  * get_unaligned((type *)iterator.this_arg) to dereference
1525                  * iterator.this_arg for type "type" safely on all arches.
1526                 */
1527                 case IEEE80211_RADIOTAP_FLAGS:
1528                         if (*iterator.this_arg & IEEE80211_RADIOTAP_F_FCS) {
1529                                 /*
1530                                  * this indicates that the skb we have been
1531                                  * handed has the 32-bit FCS CRC at the end...
1532                                  * we should react to that by snipping it off
1533                                  * because it will be recomputed and added
1534                                  * on transmission
1535                                  */
1536                                 if (skb->len < (iterator._max_length + FCS_LEN))
1537                                         return false;
1538
1539                                 skb_trim(skb, skb->len - FCS_LEN);
1540                         }
1541                         if (*iterator.this_arg & IEEE80211_RADIOTAP_F_WEP)
1542                                 info->flags &= ~IEEE80211_TX_INTFL_DONT_ENCRYPT;
1543                         if (*iterator.this_arg & IEEE80211_RADIOTAP_F_FRAG)
1544                                 info->flags &= ~IEEE80211_TX_CTL_DONTFRAG;
1545                         break;
1546
1547                 case IEEE80211_RADIOTAP_TX_FLAGS:
1548                         txflags = get_unaligned_le16(iterator.this_arg);
1549                         if (txflags & IEEE80211_RADIOTAP_F_TX_NOACK)
1550                                 info->flags |= IEEE80211_TX_CTL_NO_ACK;
1551                         break;
1552
1553                 /*
1554                  * Please update the file
1555                  * Documentation/networking/mac80211-injection.txt
1556                  * when parsing new fields here.
1557                  */
1558
1559                 default:
1560                         break;
1561                 }
1562         }
1563
1564         if (ret != -ENOENT) /* ie, if we didn't simply run out of fields */
1565                 return false;
1566
1567         /*
1568          * remove the radiotap header
1569          * iterator->_max_length was sanity-checked against
1570          * skb->len by iterator init
1571          */
1572         skb_pull(skb, iterator._max_length);
1573
1574         return true;
1575 }
1576
1577 netdev_tx_t ieee80211_monitor_start_xmit(struct sk_buff *skb,
1578                                          struct net_device *dev)
1579 {
1580         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
1581         struct ieee80211_chanctx_conf *chanctx_conf;
1582         struct ieee80211_channel *chan;
1583         struct ieee80211_radiotap_header *prthdr =
1584                 (struct ieee80211_radiotap_header *)skb->data;
1585         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1586         struct ieee80211_hdr *hdr;
1587         struct ieee80211_sub_if_data *tmp_sdata, *sdata;
1588         u16 len_rthdr;
1589         int hdrlen;
1590
1591         /* check for not even having the fixed radiotap header part */
1592         if (unlikely(skb->len < sizeof(struct ieee80211_radiotap_header)))
1593                 goto fail; /* too short to be possibly valid */
1594
1595         /* is it a header version we can trust to find length from? */
1596         if (unlikely(prthdr->it_version))
1597                 goto fail; /* only version 0 is supported */
1598
1599         /* then there must be a radiotap header with a length we can use */
1600         len_rthdr = ieee80211_get_radiotap_len(skb->data);
1601
1602         /* does the skb contain enough to deliver on the alleged length? */
1603         if (unlikely(skb->len < len_rthdr))
1604                 goto fail; /* skb too short for claimed rt header extent */
1605
1606         /*
1607          * fix up the pointers accounting for the radiotap
1608          * header still being in there.  We are being given
1609          * a precooked IEEE80211 header so no need for
1610          * normal processing
1611          */
1612         skb_set_mac_header(skb, len_rthdr);
1613         /*
1614          * these are just fixed to the end of the rt area since we
1615          * don't have any better information and at this point, nobody cares
1616          */
1617         skb_set_network_header(skb, len_rthdr);
1618         skb_set_transport_header(skb, len_rthdr);
1619
1620         if (skb->len < len_rthdr + 2)
1621                 goto fail;
1622
1623         hdr = (struct ieee80211_hdr *)(skb->data + len_rthdr);
1624         hdrlen = ieee80211_hdrlen(hdr->frame_control);
1625
1626         if (skb->len < len_rthdr + hdrlen)
1627                 goto fail;
1628
1629         /*
1630          * Initialize skb->protocol if the injected frame is a data frame
1631          * carrying a rfc1042 header
1632          */
1633         if (ieee80211_is_data(hdr->frame_control) &&
1634             skb->len >= len_rthdr + hdrlen + sizeof(rfc1042_header) + 2) {
1635                 u8 *payload = (u8 *)hdr + hdrlen;
1636
1637                 if (ether_addr_equal(payload, rfc1042_header))
1638                         skb->protocol = cpu_to_be16((payload[6] << 8) |
1639                                                     payload[7]);
1640         }
1641
1642         memset(info, 0, sizeof(*info));
1643
1644         info->flags = IEEE80211_TX_CTL_REQ_TX_STATUS |
1645                       IEEE80211_TX_CTL_INJECTED;
1646
1647         /* process and remove the injection radiotap header */
1648         if (!ieee80211_parse_tx_radiotap(skb))
1649                 goto fail;
1650
1651         rcu_read_lock();
1652
1653         /*
1654          * We process outgoing injected frames that have a local address
1655          * we handle as though they are non-injected frames.
1656          * This code here isn't entirely correct, the local MAC address
1657          * isn't always enough to find the interface to use; for proper
1658          * VLAN/WDS support we will need a different mechanism (which
1659          * likely isn't going to be monitor interfaces).
1660          */
1661         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1662
1663         list_for_each_entry_rcu(tmp_sdata, &local->interfaces, list) {
1664                 if (!ieee80211_sdata_running(tmp_sdata))
1665                         continue;
1666                 if (tmp_sdata->vif.type == NL80211_IFTYPE_MONITOR ||
1667                     tmp_sdata->vif.type == NL80211_IFTYPE_AP_VLAN ||
1668                     tmp_sdata->vif.type == NL80211_IFTYPE_WDS)
1669                         continue;
1670                 if (ether_addr_equal(tmp_sdata->vif.addr, hdr->addr2)) {
1671                         sdata = tmp_sdata;
1672                         break;
1673                 }
1674         }
1675
1676         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1677         if (!chanctx_conf) {
1678                 tmp_sdata = rcu_dereference(local->monitor_sdata);
1679                 if (tmp_sdata)
1680                         chanctx_conf =
1681                                 rcu_dereference(tmp_sdata->vif.chanctx_conf);
1682         }
1683
1684         if (chanctx_conf)
1685                 chan = chanctx_conf->def.chan;
1686         else if (!local->use_chanctx)
1687                 chan = local->_oper_channel;
1688         else
1689                 goto fail_rcu;
1690
1691         /*
1692          * Frame injection is not allowed if beaconing is not allowed
1693          * or if we need radar detection. Beaconing is usually not allowed when
1694          * the mode or operation (Adhoc, AP, Mesh) does not support DFS.
1695          * Passive scan is also used in world regulatory domains where
1696          * your country is not known and as such it should be treated as
1697          * NO TX unless the channel is explicitly allowed in which case
1698          * your current regulatory domain would not have the passive scan
1699          * flag.
1700          *
1701          * Since AP mode uses monitor interfaces to inject/TX management
1702          * frames we can make AP mode the exception to this rule once it
1703          * supports radar detection as its implementation can deal with
1704          * radar detection by itself. We can do that later by adding a
1705          * monitor flag interfaces used for AP support.
1706          */
1707         if ((chan->flags & (IEEE80211_CHAN_NO_IBSS | IEEE80211_CHAN_RADAR |
1708                             IEEE80211_CHAN_PASSIVE_SCAN)))
1709                 goto fail_rcu;
1710
1711         ieee80211_xmit(sdata, skb, chan->band);
1712         rcu_read_unlock();
1713
1714         return NETDEV_TX_OK;
1715
1716 fail_rcu:
1717         rcu_read_unlock();
1718 fail:
1719         dev_kfree_skb(skb);
1720         return NETDEV_TX_OK; /* meaning, we dealt with the skb */
1721 }
1722
1723 /**
1724  * ieee80211_subif_start_xmit - netif start_xmit function for Ethernet-type
1725  * subinterfaces (wlan#, WDS, and VLAN interfaces)
1726  * @skb: packet to be sent
1727  * @dev: incoming interface
1728  *
1729  * Returns: 0 on success (and frees skb in this case) or 1 on failure (skb will
1730  * not be freed, and caller is responsible for either retrying later or freeing
1731  * skb).
1732  *
1733  * This function takes in an Ethernet header and encapsulates it with suitable
1734  * IEEE 802.11 header based on which interface the packet is coming in. The
1735  * encapsulated packet will then be passed to master interface, wlan#.11, for
1736  * transmission (through low-level driver).
1737  */
1738 netdev_tx_t ieee80211_subif_start_xmit(struct sk_buff *skb,
1739                                     struct net_device *dev)
1740 {
1741         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1742         struct ieee80211_local *local = sdata->local;
1743         struct ieee80211_tx_info *info;
1744         int head_need;
1745         u16 ethertype, hdrlen,  meshhdrlen = 0;
1746         __le16 fc;
1747         struct ieee80211_hdr hdr;
1748         struct ieee80211s_hdr mesh_hdr __maybe_unused;
1749         struct mesh_path __maybe_unused *mppath = NULL, *mpath = NULL;
1750         const u8 *encaps_data;
1751         int encaps_len, skip_header_bytes;
1752         int nh_pos, h_pos;
1753         struct sta_info *sta = NULL;
1754         bool wme_sta = false, authorized = false, tdls_auth = false;
1755         bool tdls_direct = false;
1756         bool multicast;
1757         u32 info_flags = 0;
1758         u16 info_id = 0;
1759         struct ieee80211_chanctx_conf *chanctx_conf;
1760         struct ieee80211_sub_if_data *ap_sdata;
1761         enum ieee80211_band band;
1762
1763         if (unlikely(skb->len < ETH_HLEN))
1764                 goto fail;
1765
1766         /* convert Ethernet header to proper 802.11 header (based on
1767          * operation mode) */
1768         ethertype = (skb->data[12] << 8) | skb->data[13];
1769         fc = cpu_to_le16(IEEE80211_FTYPE_DATA | IEEE80211_STYPE_DATA);
1770
1771         rcu_read_lock();
1772
1773         switch (sdata->vif.type) {
1774         case NL80211_IFTYPE_AP_VLAN:
1775                 sta = rcu_dereference(sdata->u.vlan.sta);
1776                 if (sta) {
1777                         fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS);
1778                         /* RA TA DA SA */
1779                         memcpy(hdr.addr1, sta->sta.addr, ETH_ALEN);
1780                         memcpy(hdr.addr2, sdata->vif.addr, ETH_ALEN);
1781                         memcpy(hdr.addr3, skb->data, ETH_ALEN);
1782                         memcpy(hdr.addr4, skb->data + ETH_ALEN, ETH_ALEN);
1783                         hdrlen = 30;
1784                         authorized = test_sta_flag(sta, WLAN_STA_AUTHORIZED);
1785                         wme_sta = test_sta_flag(sta, WLAN_STA_WME);
1786                 }
1787                 ap_sdata = container_of(sdata->bss, struct ieee80211_sub_if_data,
1788                                         u.ap);
1789                 chanctx_conf = rcu_dereference(ap_sdata->vif.chanctx_conf);
1790                 if (!chanctx_conf)
1791                         goto fail_rcu;
1792                 band = chanctx_conf->def.chan->band;
1793                 if (sta)
1794                         break;
1795                 /* fall through */
1796         case NL80211_IFTYPE_AP:
1797                 if (sdata->vif.type == NL80211_IFTYPE_AP)
1798                         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1799                 if (!chanctx_conf)
1800                         goto fail_rcu;
1801                 fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS);
1802                 /* DA BSSID SA */
1803                 memcpy(hdr.addr1, skb->data, ETH_ALEN);
1804                 memcpy(hdr.addr2, sdata->vif.addr, ETH_ALEN);
1805                 memcpy(hdr.addr3, skb->data + ETH_ALEN, ETH_ALEN);
1806                 hdrlen = 24;
1807                 band = chanctx_conf->def.chan->band;
1808                 break;
1809         case NL80211_IFTYPE_WDS:
1810                 fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS);
1811                 /* RA TA DA SA */
1812                 memcpy(hdr.addr1, sdata->u.wds.remote_addr, ETH_ALEN);
1813                 memcpy(hdr.addr2, sdata->vif.addr, ETH_ALEN);
1814                 memcpy(hdr.addr3, skb->data, ETH_ALEN);
1815                 memcpy(hdr.addr4, skb->data + ETH_ALEN, ETH_ALEN);
1816                 hdrlen = 30;
1817                 /*
1818                  * This is the exception! WDS style interfaces are prohibited
1819                  * when channel contexts are in used so this must be valid
1820                  */
1821                 band = local->hw.conf.channel->band;
1822                 break;
1823 #ifdef CONFIG_MAC80211_MESH
1824         case NL80211_IFTYPE_MESH_POINT:
1825                 if (!sdata->u.mesh.mshcfg.dot11MeshTTL) {
1826                         /* Do not send frames with mesh_ttl == 0 */
1827                         sdata->u.mesh.mshstats.dropped_frames_ttl++;
1828                         goto fail_rcu;
1829                 }
1830
1831                 if (!is_multicast_ether_addr(skb->data)) {
1832                         mpath = mesh_path_lookup(skb->data, sdata);
1833                         if (!mpath)
1834                                 mppath = mpp_path_lookup(skb->data, sdata);
1835                 }
1836
1837                 /*
1838                  * Use address extension if it is a packet from
1839                  * another interface or if we know the destination
1840                  * is being proxied by a portal (i.e. portal address
1841                  * differs from proxied address)
1842                  */
1843                 if (ether_addr_equal(sdata->vif.addr, skb->data + ETH_ALEN) &&
1844                     !(mppath && !ether_addr_equal(mppath->mpp, skb->data))) {
1845                         hdrlen = ieee80211_fill_mesh_addresses(&hdr, &fc,
1846                                         skb->data, skb->data + ETH_ALEN);
1847                         meshhdrlen = ieee80211_new_mesh_header(&mesh_hdr,
1848                                         sdata, NULL, NULL);
1849                 } else {
1850                         /* DS -> MBSS (802.11-2012 13.11.3.3).
1851                          * For unicast with unknown forwarding information,
1852                          * destination might be in the MBSS or if that fails
1853                          * forwarded to another mesh gate. In either case
1854                          * resolution will be handled in ieee80211_xmit(), so
1855                          * leave the original DA. This also works for mcast */
1856                         const u8 *mesh_da = skb->data;
1857
1858                         if (mppath)
1859                                 mesh_da = mppath->mpp;
1860                         else if (mpath)
1861                                 mesh_da = mpath->dst;
1862
1863                         hdrlen = ieee80211_fill_mesh_addresses(&hdr, &fc,
1864                                         mesh_da, sdata->vif.addr);
1865                         if (is_multicast_ether_addr(mesh_da))
1866                                 /* DA TA mSA AE:SA */
1867                                 meshhdrlen =
1868                                         ieee80211_new_mesh_header(&mesh_hdr,
1869                                                         sdata,
1870                                                         skb->data + ETH_ALEN,
1871                                                         NULL);
1872                         else
1873                                 /* RA TA mDA mSA AE:DA SA */
1874                                 meshhdrlen =
1875                                         ieee80211_new_mesh_header(&mesh_hdr,
1876                                                         sdata,
1877                                                         skb->data,
1878                                                         skb->data + ETH_ALEN);
1879
1880                 }
1881                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1882                 if (!chanctx_conf)
1883                         goto fail_rcu;
1884                 band = chanctx_conf->def.chan->band;
1885                 break;
1886 #endif
1887         case NL80211_IFTYPE_STATION:
1888                 if (sdata->wdev.wiphy->flags & WIPHY_FLAG_SUPPORTS_TDLS) {
1889                         bool tdls_peer = false;
1890
1891                         sta = sta_info_get(sdata, skb->data);
1892                         if (sta) {
1893                                 authorized = test_sta_flag(sta,
1894                                                         WLAN_STA_AUTHORIZED);
1895                                 wme_sta = test_sta_flag(sta, WLAN_STA_WME);
1896                                 tdls_peer = test_sta_flag(sta,
1897                                                          WLAN_STA_TDLS_PEER);
1898                                 tdls_auth = test_sta_flag(sta,
1899                                                 WLAN_STA_TDLS_PEER_AUTH);
1900                         }
1901
1902                         /*
1903                          * If the TDLS link is enabled, send everything
1904                          * directly. Otherwise, allow TDLS setup frames
1905                          * to be transmitted indirectly.
1906                          */
1907                         tdls_direct = tdls_peer && (tdls_auth ||
1908                                  !(ethertype == ETH_P_TDLS && skb->len > 14 &&
1909                                    skb->data[14] == WLAN_TDLS_SNAP_RFTYPE));
1910                 }
1911
1912                 if (tdls_direct) {
1913                         /* link during setup - throw out frames to peer */
1914                         if (!tdls_auth)
1915                                 goto fail_rcu;
1916
1917                         /* DA SA BSSID */
1918                         memcpy(hdr.addr1, skb->data, ETH_ALEN);
1919                         memcpy(hdr.addr2, skb->data + ETH_ALEN, ETH_ALEN);
1920                         memcpy(hdr.addr3, sdata->u.mgd.bssid, ETH_ALEN);
1921                         hdrlen = 24;
1922                 }  else if (sdata->u.mgd.use_4addr &&
1923                             cpu_to_be16(ethertype) != sdata->control_port_protocol) {
1924                         fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS |
1925                                           IEEE80211_FCTL_TODS);
1926                         /* RA TA DA SA */
1927                         memcpy(hdr.addr1, sdata->u.mgd.bssid, ETH_ALEN);
1928                         memcpy(hdr.addr2, sdata->vif.addr, ETH_ALEN);
1929                         memcpy(hdr.addr3, skb->data, ETH_ALEN);
1930                         memcpy(hdr.addr4, skb->data + ETH_ALEN, ETH_ALEN);
1931                         hdrlen = 30;
1932                 } else {
1933                         fc |= cpu_to_le16(IEEE80211_FCTL_TODS);
1934                         /* BSSID SA DA */
1935                         memcpy(hdr.addr1, sdata->u.mgd.bssid, ETH_ALEN);
1936                         memcpy(hdr.addr2, skb->data + ETH_ALEN, ETH_ALEN);
1937                         memcpy(hdr.addr3, skb->data, ETH_ALEN);
1938                         hdrlen = 24;
1939                 }
1940                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1941                 if (!chanctx_conf)
1942                         goto fail_rcu;
1943                 band = chanctx_conf->def.chan->band;
1944                 break;
1945         case NL80211_IFTYPE_ADHOC:
1946                 /* DA SA BSSID */
1947                 memcpy(hdr.addr1, skb->data, ETH_ALEN);
1948                 memcpy(hdr.addr2, skb->data + ETH_ALEN, ETH_ALEN);
1949                 memcpy(hdr.addr3, sdata->u.ibss.bssid, ETH_ALEN);
1950                 hdrlen = 24;
1951                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1952                 if (!chanctx_conf)
1953                         goto fail_rcu;
1954                 band = chanctx_conf->def.chan->band;
1955                 break;
1956         default:
1957                 goto fail_rcu;
1958         }
1959
1960         /*
1961          * There's no need to try to look up the destination
1962          * if it is a multicast address (which can only happen
1963          * in AP mode)
1964          */
1965         multicast = is_multicast_ether_addr(hdr.addr1);
1966         if (!multicast) {
1967                 sta = sta_info_get(sdata, hdr.addr1);
1968                 if (sta) {
1969                         authorized = test_sta_flag(sta, WLAN_STA_AUTHORIZED);
1970                         wme_sta = test_sta_flag(sta, WLAN_STA_WME);
1971                 }
1972         }
1973
1974         /* For mesh, the use of the QoS header is mandatory */
1975         if (ieee80211_vif_is_mesh(&sdata->vif))
1976                 wme_sta = true;
1977
1978         /* receiver and we are QoS enabled, use a QoS type frame */
1979         if (wme_sta && local->hw.queues >= IEEE80211_NUM_ACS) {
1980                 fc |= cpu_to_le16(IEEE80211_STYPE_QOS_DATA);
1981                 hdrlen += 2;
1982         }
1983
1984         /*
1985          * Drop unicast frames to unauthorised stations unless they are
1986          * EAPOL frames from the local station.
1987          */
1988         if (unlikely(!ieee80211_vif_is_mesh(&sdata->vif) &&
1989                      !is_multicast_ether_addr(hdr.addr1) && !authorized &&
1990                      (cpu_to_be16(ethertype) != sdata->control_port_protocol ||
1991                       !ether_addr_equal(sdata->vif.addr, skb->data + ETH_ALEN)))) {
1992 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
1993                 net_info_ratelimited("%s: dropped frame to %pM (unauthorized port)\n",
1994                                     dev->name, hdr.addr1);
1995 #endif
1996
1997                 I802_DEBUG_INC(local->tx_handlers_drop_unauth_port);
1998
1999                 goto fail_rcu;
2000         }
2001
2002         if (unlikely(!multicast && skb->sk &&
2003                      skb_shinfo(skb)->tx_flags & SKBTX_WIFI_STATUS)) {
2004                 struct sk_buff *orig_skb = skb;
2005
2006                 skb = skb_clone(skb, GFP_ATOMIC);
2007                 if (skb) {
2008                         unsigned long flags;
2009                         int id, r;
2010
2011                         spin_lock_irqsave(&local->ack_status_lock, flags);
2012                         r = idr_get_new_above(&local->ack_status_frames,
2013                                               orig_skb, 1, &id);
2014                         if (r == -EAGAIN) {
2015                                 idr_pre_get(&local->ack_status_frames,
2016                                             GFP_ATOMIC);
2017                                 r = idr_get_new_above(&local->ack_status_frames,
2018                                                       orig_skb, 1, &id);
2019                         }
2020                         if (WARN_ON(!id) || id > 0xffff) {
2021                                 idr_remove(&local->ack_status_frames, id);
2022                                 r = -ERANGE;
2023                         }
2024                         spin_unlock_irqrestore(&local->ack_status_lock, flags);
2025
2026                         if (!r) {
2027                                 info_id = id;
2028                                 info_flags |= IEEE80211_TX_CTL_REQ_TX_STATUS;
2029                         } else if (skb_shared(skb)) {
2030                                 kfree_skb(orig_skb);
2031                         } else {
2032                                 kfree_skb(skb);
2033                                 skb = orig_skb;
2034                         }
2035                 } else {
2036                         /* couldn't clone -- lose tx status ... */
2037                         skb = orig_skb;
2038                 }
2039         }
2040
2041         /*
2042          * If the skb is shared we need to obtain our own copy.
2043          */
2044         if (skb_shared(skb)) {
2045                 struct sk_buff *tmp_skb = skb;
2046
2047                 /* can't happen -- skb is a clone if info_id != 0 */
2048                 WARN_ON(info_id);
2049
2050                 skb = skb_clone(skb, GFP_ATOMIC);
2051                 kfree_skb(tmp_skb);
2052
2053                 if (!skb)
2054                         goto fail_rcu;
2055         }
2056
2057         hdr.frame_control = fc;
2058         hdr.duration_id = 0;
2059         hdr.seq_ctrl = 0;
2060
2061         skip_header_bytes = ETH_HLEN;
2062         if (ethertype == ETH_P_AARP || ethertype == ETH_P_IPX) {
2063                 encaps_data = bridge_tunnel_header;
2064                 encaps_len = sizeof(bridge_tunnel_header);
2065                 skip_header_bytes -= 2;
2066         } else if (ethertype >= 0x600) {
2067                 encaps_data = rfc1042_header;
2068                 encaps_len = sizeof(rfc1042_header);
2069                 skip_header_bytes -= 2;
2070         } else {
2071                 encaps_data = NULL;
2072                 encaps_len = 0;
2073         }
2074
2075         nh_pos = skb_network_header(skb) - skb->data;
2076         h_pos = skb_transport_header(skb) - skb->data;
2077
2078         skb_pull(skb, skip_header_bytes);
2079         nh_pos -= skip_header_bytes;
2080         h_pos -= skip_header_bytes;
2081
2082         head_need = hdrlen + encaps_len + meshhdrlen - skb_headroom(skb);
2083
2084         /*
2085          * So we need to modify the skb header and hence need a copy of
2086          * that. The head_need variable above doesn't, so far, include
2087          * the needed header space that we don't need right away. If we
2088          * can, then we don't reallocate right now but only after the
2089          * frame arrives at the master device (if it does...)
2090          *
2091          * If we cannot, however, then we will reallocate to include all
2092          * the ever needed space. Also, if we need to reallocate it anyway,
2093          * make it big enough for everything we may ever need.
2094          */
2095
2096         if (head_need > 0 || skb_cloned(skb)) {
2097                 head_need += IEEE80211_ENCRYPT_HEADROOM;
2098                 head_need += local->tx_headroom;
2099                 head_need = max_t(int, 0, head_need);
2100                 if (ieee80211_skb_resize(sdata, skb, head_need, true)) {
2101                         ieee80211_free_txskb(&local->hw, skb);
2102                         skb = NULL;
2103                         goto fail_rcu;
2104                 }
2105         }
2106
2107         if (encaps_data) {
2108                 memcpy(skb_push(skb, encaps_len), encaps_data, encaps_len);
2109                 nh_pos += encaps_len;
2110                 h_pos += encaps_len;
2111         }
2112
2113 #ifdef CONFIG_MAC80211_MESH
2114         if (meshhdrlen > 0) {
2115                 memcpy(skb_push(skb, meshhdrlen), &mesh_hdr, meshhdrlen);
2116                 nh_pos += meshhdrlen;
2117                 h_pos += meshhdrlen;
2118         }
2119 #endif
2120
2121         if (ieee80211_is_data_qos(fc)) {
2122                 __le16 *qos_control;
2123
2124                 qos_control = (__le16*) skb_push(skb, 2);
2125                 memcpy(skb_push(skb, hdrlen - 2), &hdr, hdrlen - 2);
2126                 /*
2127                  * Maybe we could actually set some fields here, for now just
2128                  * initialise to zero to indicate no special operation.
2129                  */
2130                 *qos_control = 0;
2131         } else
2132                 memcpy(skb_push(skb, hdrlen), &hdr, hdrlen);
2133
2134         nh_pos += hdrlen;
2135         h_pos += hdrlen;
2136
2137         dev->stats.tx_packets++;
2138         dev->stats.tx_bytes += skb->len;
2139
2140         /* Update skb pointers to various headers since this modified frame
2141          * is going to go through Linux networking code that may potentially
2142          * need things like pointer to IP header. */
2143         skb_set_mac_header(skb, 0);
2144         skb_set_network_header(skb, nh_pos);
2145         skb_set_transport_header(skb, h_pos);
2146
2147         info = IEEE80211_SKB_CB(skb);
2148         memset(info, 0, sizeof(*info));
2149
2150         dev->trans_start = jiffies;
2151
2152         info->flags = info_flags;
2153         info->ack_frame_id = info_id;
2154
2155         ieee80211_xmit(sdata, skb, band);
2156         rcu_read_unlock();
2157
2158         return NETDEV_TX_OK;
2159
2160  fail_rcu:
2161         rcu_read_unlock();
2162  fail:
2163         dev_kfree_skb(skb);
2164         return NETDEV_TX_OK;
2165 }
2166
2167
2168 /*
2169  * ieee80211_clear_tx_pending may not be called in a context where
2170  * it is possible that it packets could come in again.
2171  */
2172 void ieee80211_clear_tx_pending(struct ieee80211_local *local)
2173 {
2174         struct sk_buff *skb;
2175         int i;
2176
2177         for (i = 0; i < local->hw.queues; i++) {
2178                 while ((skb = skb_dequeue(&local->pending[i])) != NULL)
2179                         ieee80211_free_txskb(&local->hw, skb);
2180         }
2181 }
2182
2183 /*
2184  * Returns false if the frame couldn't be transmitted but was queued instead,
2185  * which in this case means re-queued -- take as an indication to stop sending
2186  * more pending frames.
2187  */
2188 static bool ieee80211_tx_pending_skb(struct ieee80211_local *local,
2189                                      struct sk_buff *skb)
2190 {
2191         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
2192         struct ieee80211_sub_if_data *sdata;
2193         struct sta_info *sta;
2194         struct ieee80211_hdr *hdr;
2195         bool result;
2196         struct ieee80211_chanctx_conf *chanctx_conf;
2197
2198         sdata = vif_to_sdata(info->control.vif);
2199
2200         if (info->flags & IEEE80211_TX_INTFL_NEED_TXPROCESSING) {
2201                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2202                 if (unlikely(!chanctx_conf)) {
2203                         dev_kfree_skb(skb);
2204                         return true;
2205                 }
2206                 result = ieee80211_tx(sdata, skb, true,
2207                                       chanctx_conf->def.chan->band);
2208         } else {
2209                 struct sk_buff_head skbs;
2210
2211                 __skb_queue_head_init(&skbs);
2212                 __skb_queue_tail(&skbs, skb);
2213
2214                 hdr = (struct ieee80211_hdr *)skb->data;
2215                 sta = sta_info_get(sdata, hdr->addr1);
2216
2217                 result = __ieee80211_tx(local, &skbs, skb->len, sta, true);
2218         }
2219
2220         return result;
2221 }
2222
2223 /*
2224  * Transmit all pending packets. Called from tasklet.
2225  */
2226 void ieee80211_tx_pending(unsigned long data)
2227 {
2228         struct ieee80211_local *local = (struct ieee80211_local *)data;
2229         unsigned long flags;
2230         int i;
2231         bool txok;
2232
2233         rcu_read_lock();
2234
2235         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
2236         for (i = 0; i < local->hw.queues; i++) {
2237                 /*
2238                  * If queue is stopped by something other than due to pending
2239                  * frames, or we have no pending frames, proceed to next queue.
2240                  */
2241                 if (local->queue_stop_reasons[i] ||
2242                     skb_queue_empty(&local->pending[i]))
2243                         continue;
2244
2245                 while (!skb_queue_empty(&local->pending[i])) {
2246                         struct sk_buff *skb = __skb_dequeue(&local->pending[i]);
2247                         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
2248
2249                         if (WARN_ON(!info->control.vif)) {
2250                                 ieee80211_free_txskb(&local->hw, skb);
2251                                 continue;
2252                         }
2253
2254                         spin_unlock_irqrestore(&local->queue_stop_reason_lock,
2255                                                 flags);
2256
2257                         txok = ieee80211_tx_pending_skb(local, skb);
2258                         spin_lock_irqsave(&local->queue_stop_reason_lock,
2259                                           flags);
2260                         if (!txok)
2261                                 break;
2262                 }
2263
2264                 if (skb_queue_empty(&local->pending[i]))
2265                         ieee80211_propagate_queue_wake(local, i);
2266         }
2267         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
2268
2269         rcu_read_unlock();
2270 }
2271
2272 /* functions for drivers to get certain frames */
2273
2274 static void __ieee80211_beacon_add_tim(struct ieee80211_sub_if_data *sdata,
2275                                        struct ps_data *ps, struct sk_buff *skb)
2276 {
2277         u8 *pos, *tim;
2278         int aid0 = 0;
2279         int i, have_bits = 0, n1, n2;
2280
2281         /* Generate bitmap for TIM only if there are any STAs in power save
2282          * mode. */
2283         if (atomic_read(&ps->num_sta_ps) > 0)
2284                 /* in the hope that this is faster than
2285                  * checking byte-for-byte */
2286                 have_bits = !bitmap_empty((unsigned long*)ps->tim,
2287                                           IEEE80211_MAX_AID+1);
2288
2289         if (ps->dtim_count == 0)
2290                 ps->dtim_count = sdata->vif.bss_conf.dtim_period - 1;
2291         else
2292                 ps->dtim_count--;
2293
2294         tim = pos = (u8 *) skb_put(skb, 6);
2295         *pos++ = WLAN_EID_TIM;
2296         *pos++ = 4;
2297         *pos++ = ps->dtim_count;
2298         *pos++ = sdata->vif.bss_conf.dtim_period;
2299
2300         if (ps->dtim_count == 0 && !skb_queue_empty(&ps->bc_buf))
2301                 aid0 = 1;
2302
2303         ps->dtim_bc_mc = aid0 == 1;
2304
2305         if (have_bits) {
2306                 /* Find largest even number N1 so that bits numbered 1 through
2307                  * (N1 x 8) - 1 in the bitmap are 0 and number N2 so that bits
2308                  * (N2 + 1) x 8 through 2007 are 0. */
2309                 n1 = 0;
2310                 for (i = 0; i < IEEE80211_MAX_TIM_LEN; i++) {
2311                         if (ps->tim[i]) {
2312                                 n1 = i & 0xfe;
2313                                 break;
2314                         }
2315                 }
2316                 n2 = n1;
2317                 for (i = IEEE80211_MAX_TIM_LEN - 1; i >= n1; i--) {
2318                         if (ps->tim[i]) {
2319                                 n2 = i;
2320                                 break;
2321                         }
2322                 }
2323
2324                 /* Bitmap control */
2325                 *pos++ = n1 | aid0;
2326                 /* Part Virt Bitmap */
2327                 skb_put(skb, n2 - n1);
2328                 memcpy(pos, ps->tim + n1, n2 - n1 + 1);
2329
2330                 tim[1] = n2 - n1 + 4;
2331         } else {
2332                 *pos++ = aid0; /* Bitmap control */
2333                 *pos++ = 0; /* Part Virt Bitmap */
2334         }
2335 }
2336
2337 static int ieee80211_beacon_add_tim(struct ieee80211_sub_if_data *sdata,
2338                                     struct ps_data *ps, struct sk_buff *skb)
2339 {
2340         struct ieee80211_local *local = sdata->local;
2341
2342         /*
2343          * Not very nice, but we want to allow the driver to call
2344          * ieee80211_beacon_get() as a response to the set_tim()
2345          * callback. That, however, is already invoked under the
2346          * sta_lock to guarantee consistent and race-free update
2347          * of the tim bitmap in mac80211 and the driver.
2348          */
2349         if (local->tim_in_locked_section) {
2350                 __ieee80211_beacon_add_tim(sdata, ps, skb);
2351         } else {
2352                 unsigned long flags;
2353
2354                 spin_lock_irqsave(&local->tim_lock, flags);
2355                 __ieee80211_beacon_add_tim(sdata, ps, skb);
2356                 spin_unlock_irqrestore(&local->tim_lock, flags);
2357         }
2358
2359         return 0;
2360 }
2361
2362 struct sk_buff *ieee80211_beacon_get_tim(struct ieee80211_hw *hw,
2363                                          struct ieee80211_vif *vif,
2364                                          u16 *tim_offset, u16 *tim_length)
2365 {
2366         struct ieee80211_local *local = hw_to_local(hw);
2367         struct sk_buff *skb = NULL;
2368         struct ieee80211_tx_info *info;
2369         struct ieee80211_sub_if_data *sdata = NULL;
2370         enum ieee80211_band band;
2371         struct ieee80211_tx_rate_control txrc;
2372         struct ieee80211_chanctx_conf *chanctx_conf;
2373
2374         rcu_read_lock();
2375
2376         sdata = vif_to_sdata(vif);
2377         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2378
2379         if (!ieee80211_sdata_running(sdata) || !chanctx_conf)
2380                 goto out;
2381
2382         if (tim_offset)
2383                 *tim_offset = 0;
2384         if (tim_length)
2385                 *tim_length = 0;
2386
2387         if (sdata->vif.type == NL80211_IFTYPE_AP) {
2388                 struct ieee80211_if_ap *ap = &sdata->u.ap;
2389                 struct beacon_data *beacon = rcu_dereference(ap->beacon);
2390
2391                 if (beacon) {
2392                         /*
2393                          * headroom, head length,
2394                          * tail length and maximum TIM length
2395                          */
2396                         skb = dev_alloc_skb(local->tx_headroom +
2397                                             beacon->head_len +
2398                                             beacon->tail_len + 256);
2399                         if (!skb)
2400                                 goto out;
2401
2402                         skb_reserve(skb, local->tx_headroom);
2403                         memcpy(skb_put(skb, beacon->head_len), beacon->head,
2404                                beacon->head_len);
2405
2406                         ieee80211_beacon_add_tim(sdata, &ap->ps, skb);
2407
2408                         if (tim_offset)
2409                                 *tim_offset = beacon->head_len;
2410                         if (tim_length)
2411                                 *tim_length = skb->len - beacon->head_len;
2412
2413                         if (beacon->tail)
2414                                 memcpy(skb_put(skb, beacon->tail_len),
2415                                        beacon->tail, beacon->tail_len);
2416                 } else
2417                         goto out;
2418         } else if (sdata->vif.type == NL80211_IFTYPE_ADHOC) {
2419                 struct ieee80211_if_ibss *ifibss = &sdata->u.ibss;
2420                 struct ieee80211_hdr *hdr;
2421                 struct sk_buff *presp = rcu_dereference(ifibss->presp);
2422
2423                 if (!presp)
2424                         goto out;
2425
2426                 skb = skb_copy(presp, GFP_ATOMIC);
2427                 if (!skb)
2428                         goto out;
2429
2430                 hdr = (struct ieee80211_hdr *) skb->data;
2431                 hdr->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
2432                                                  IEEE80211_STYPE_BEACON);
2433         } else if (ieee80211_vif_is_mesh(&sdata->vif)) {
2434                 struct ieee80211_mgmt *mgmt;
2435                 struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
2436                 u8 *pos;
2437                 int hdr_len = offsetof(struct ieee80211_mgmt, u.beacon) +
2438                               sizeof(mgmt->u.beacon);
2439
2440 #ifdef CONFIG_MAC80211_MESH
2441                 if (!sdata->u.mesh.mesh_id_len)
2442                         goto out;
2443 #endif
2444
2445                 if (ifmsh->sync_ops)
2446                         ifmsh->sync_ops->adjust_tbtt(
2447                                                 sdata);
2448
2449                 skb = dev_alloc_skb(local->tx_headroom +
2450                                     hdr_len +
2451                                     2 + /* NULL SSID */
2452                                     2 + 8 + /* supported rates */
2453                                     2 + 3 + /* DS params */
2454                                     256 + /* TIM IE */
2455                                     2 + (IEEE80211_MAX_SUPP_RATES - 8) +
2456                                     2 + sizeof(struct ieee80211_ht_cap) +
2457                                     2 + sizeof(struct ieee80211_ht_operation) +
2458                                     2 + sdata->u.mesh.mesh_id_len +
2459                                     2 + sizeof(struct ieee80211_meshconf_ie) +
2460                                     sdata->u.mesh.ie_len +
2461                                     2 + sizeof(__le16)); /* awake window */
2462                 if (!skb)
2463                         goto out;
2464
2465                 skb_reserve(skb, local->hw.extra_tx_headroom);
2466                 mgmt = (struct ieee80211_mgmt *) skb_put(skb, hdr_len);
2467                 memset(mgmt, 0, hdr_len);
2468                 mgmt->frame_control =
2469                     cpu_to_le16(IEEE80211_FTYPE_MGMT | IEEE80211_STYPE_BEACON);
2470                 eth_broadcast_addr(mgmt->da);
2471                 memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
2472                 memcpy(mgmt->bssid, sdata->vif.addr, ETH_ALEN);
2473                 ieee80211_mps_set_frame_flags(sdata, NULL, (void *) mgmt);
2474                 mgmt->u.beacon.beacon_int =
2475                         cpu_to_le16(sdata->vif.bss_conf.beacon_int);
2476                 mgmt->u.beacon.capab_info |= cpu_to_le16(
2477                         sdata->u.mesh.security ? WLAN_CAPABILITY_PRIVACY : 0);
2478
2479                 pos = skb_put(skb, 2);
2480                 *pos++ = WLAN_EID_SSID;
2481                 *pos++ = 0x0;
2482
2483                 band = chanctx_conf->def.chan->band;
2484
2485                 if (ieee80211_add_srates_ie(sdata, skb, true, band) ||
2486                     mesh_add_ds_params_ie(skb, sdata) ||
2487                     ieee80211_beacon_add_tim(sdata, &ifmsh->ps, skb) ||
2488                     ieee80211_add_ext_srates_ie(sdata, skb, true, band) ||
2489                     mesh_add_rsn_ie(skb, sdata) ||
2490                     mesh_add_ht_cap_ie(skb, sdata) ||
2491                     mesh_add_ht_oper_ie(skb, sdata) ||
2492                     mesh_add_meshid_ie(skb, sdata) ||
2493                     mesh_add_meshconf_ie(skb, sdata) ||
2494                     mesh_add_awake_window_ie(skb, sdata) ||
2495                     mesh_add_vendor_ies(skb, sdata)) {
2496                         pr_err("o11s: couldn't add ies!\n");
2497                         goto out;
2498                 }
2499         } else {
2500                 WARN_ON(1);
2501                 goto out;
2502         }
2503
2504         band = chanctx_conf->def.chan->band;
2505
2506         info = IEEE80211_SKB_CB(skb);
2507
2508         info->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
2509         info->flags |= IEEE80211_TX_CTL_NO_ACK;
2510         info->band = band;
2511
2512         memset(&txrc, 0, sizeof(txrc));
2513         txrc.hw = hw;
2514         txrc.sband = local->hw.wiphy->bands[band];
2515         txrc.bss_conf = &sdata->vif.bss_conf;
2516         txrc.skb = skb;
2517         txrc.reported_rate.idx = -1;
2518         txrc.rate_idx_mask = sdata->rc_rateidx_mask[band];
2519         if (txrc.rate_idx_mask == (1 << txrc.sband->n_bitrates) - 1)
2520                 txrc.max_rate_idx = -1;
2521         else
2522                 txrc.max_rate_idx = fls(txrc.rate_idx_mask) - 1;
2523         memcpy(txrc.rate_idx_mcs_mask, sdata->rc_rateidx_mcs_mask[band],
2524                sizeof(txrc.rate_idx_mcs_mask));
2525         txrc.bss = true;
2526         rate_control_get_rate(sdata, NULL, &txrc);
2527
2528         info->control.vif = vif;
2529
2530         info->flags |= IEEE80211_TX_CTL_CLEAR_PS_FILT |
2531                         IEEE80211_TX_CTL_ASSIGN_SEQ |
2532                         IEEE80211_TX_CTL_FIRST_FRAGMENT;
2533  out:
2534         rcu_read_unlock();
2535         return skb;
2536 }
2537 EXPORT_SYMBOL(ieee80211_beacon_get_tim);
2538
2539 struct sk_buff *ieee80211_proberesp_get(struct ieee80211_hw *hw,
2540                                         struct ieee80211_vif *vif)
2541 {
2542         struct ieee80211_if_ap *ap = NULL;
2543         struct sk_buff *skb = NULL;
2544         struct probe_resp *presp = NULL;
2545         struct ieee80211_hdr *hdr;
2546         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2547
2548         if (sdata->vif.type != NL80211_IFTYPE_AP)
2549                 return NULL;
2550
2551         rcu_read_lock();
2552
2553         ap = &sdata->u.ap;
2554         presp = rcu_dereference(ap->probe_resp);
2555         if (!presp)
2556                 goto out;
2557
2558         skb = dev_alloc_skb(presp->len);
2559         if (!skb)
2560                 goto out;
2561
2562         memcpy(skb_put(skb, presp->len), presp->data, presp->len);
2563
2564         hdr = (struct ieee80211_hdr *) skb->data;
2565         memset(hdr->addr1, 0, sizeof(hdr->addr1));
2566
2567 out:
2568         rcu_read_unlock();
2569         return skb;
2570 }
2571 EXPORT_SYMBOL(ieee80211_proberesp_get);
2572
2573 struct sk_buff *ieee80211_pspoll_get(struct ieee80211_hw *hw,
2574                                      struct ieee80211_vif *vif)
2575 {
2576         struct ieee80211_sub_if_data *sdata;
2577         struct ieee80211_if_managed *ifmgd;
2578         struct ieee80211_pspoll *pspoll;
2579         struct ieee80211_local *local;
2580         struct sk_buff *skb;
2581
2582         if (WARN_ON(vif->type != NL80211_IFTYPE_STATION))
2583                 return NULL;
2584
2585         sdata = vif_to_sdata(vif);
2586         ifmgd = &sdata->u.mgd;
2587         local = sdata->local;
2588
2589         skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*pspoll));
2590         if (!skb)
2591                 return NULL;
2592
2593         skb_reserve(skb, local->hw.extra_tx_headroom);
2594
2595         pspoll = (struct ieee80211_pspoll *) skb_put(skb, sizeof(*pspoll));
2596         memset(pspoll, 0, sizeof(*pspoll));
2597         pspoll->frame_control = cpu_to_le16(IEEE80211_FTYPE_CTL |
2598                                             IEEE80211_STYPE_PSPOLL);
2599         pspoll->aid = cpu_to_le16(ifmgd->aid);
2600
2601         /* aid in PS-Poll has its two MSBs each set to 1 */
2602         pspoll->aid |= cpu_to_le16(1 << 15 | 1 << 14);
2603
2604         memcpy(pspoll->bssid, ifmgd->bssid, ETH_ALEN);
2605         memcpy(pspoll->ta, vif->addr, ETH_ALEN);
2606
2607         return skb;
2608 }
2609 EXPORT_SYMBOL(ieee80211_pspoll_get);
2610
2611 struct sk_buff *ieee80211_nullfunc_get(struct ieee80211_hw *hw,
2612                                        struct ieee80211_vif *vif)
2613 {
2614         struct ieee80211_hdr_3addr *nullfunc;
2615         struct ieee80211_sub_if_data *sdata;
2616         struct ieee80211_if_managed *ifmgd;
2617         struct ieee80211_local *local;
2618         struct sk_buff *skb;
2619
2620         if (WARN_ON(vif->type != NL80211_IFTYPE_STATION))
2621                 return NULL;
2622
2623         sdata = vif_to_sdata(vif);
2624         ifmgd = &sdata->u.mgd;
2625         local = sdata->local;
2626
2627         skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*nullfunc));
2628         if (!skb)
2629                 return NULL;
2630
2631         skb_reserve(skb, local->hw.extra_tx_headroom);
2632
2633         nullfunc = (struct ieee80211_hdr_3addr *) skb_put(skb,
2634                                                           sizeof(*nullfunc));
2635         memset(nullfunc, 0, sizeof(*nullfunc));
2636         nullfunc->frame_control = cpu_to_le16(IEEE80211_FTYPE_DATA |
2637                                               IEEE80211_STYPE_NULLFUNC |
2638                                               IEEE80211_FCTL_TODS);
2639         memcpy(nullfunc->addr1, ifmgd->bssid, ETH_ALEN);
2640         memcpy(nullfunc->addr2, vif->addr, ETH_ALEN);
2641         memcpy(nullfunc->addr3, ifmgd->bssid, ETH_ALEN);
2642
2643         return skb;
2644 }
2645 EXPORT_SYMBOL(ieee80211_nullfunc_get);
2646
2647 struct sk_buff *ieee80211_probereq_get(struct ieee80211_hw *hw,
2648                                        struct ieee80211_vif *vif,
2649                                        const u8 *ssid, size_t ssid_len,
2650                                        size_t tailroom)
2651 {
2652         struct ieee80211_sub_if_data *sdata;
2653         struct ieee80211_local *local;
2654         struct ieee80211_hdr_3addr *hdr;
2655         struct sk_buff *skb;
2656         size_t ie_ssid_len;
2657         u8 *pos;
2658
2659         sdata = vif_to_sdata(vif);
2660         local = sdata->local;
2661         ie_ssid_len = 2 + ssid_len;
2662
2663         skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*hdr) +
2664                             ie_ssid_len + tailroom);
2665         if (!skb)
2666                 return NULL;
2667
2668         skb_reserve(skb, local->hw.extra_tx_headroom);
2669
2670         hdr = (struct ieee80211_hdr_3addr *) skb_put(skb, sizeof(*hdr));
2671         memset(hdr, 0, sizeof(*hdr));
2672         hdr->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
2673                                          IEEE80211_STYPE_PROBE_REQ);
2674         eth_broadcast_addr(hdr->addr1);
2675         memcpy(hdr->addr2, vif->addr, ETH_ALEN);
2676         eth_broadcast_addr(hdr->addr3);
2677
2678         pos = skb_put(skb, ie_ssid_len);
2679         *pos++ = WLAN_EID_SSID;
2680         *pos++ = ssid_len;
2681         if (ssid_len)
2682                 memcpy(pos, ssid, ssid_len);
2683         pos += ssid_len;
2684
2685         return skb;
2686 }
2687 EXPORT_SYMBOL(ieee80211_probereq_get);
2688
2689 void ieee80211_rts_get(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
2690                        const void *frame, size_t frame_len,
2691                        const struct ieee80211_tx_info *frame_txctl,
2692                        struct ieee80211_rts *rts)
2693 {
2694         const struct ieee80211_hdr *hdr = frame;
2695
2696         rts->frame_control =
2697             cpu_to_le16(IEEE80211_FTYPE_CTL | IEEE80211_STYPE_RTS);
2698         rts->duration = ieee80211_rts_duration(hw, vif, frame_len,
2699                                                frame_txctl);
2700         memcpy(rts->ra, hdr->addr1, sizeof(rts->ra));
2701         memcpy(rts->ta, hdr->addr2, sizeof(rts->ta));
2702 }
2703 EXPORT_SYMBOL(ieee80211_rts_get);
2704
2705 void ieee80211_ctstoself_get(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
2706                              const void *frame, size_t frame_len,
2707                              const struct ieee80211_tx_info *frame_txctl,
2708                              struct ieee80211_cts *cts)
2709 {
2710         const struct ieee80211_hdr *hdr = frame;
2711
2712         cts->frame_control =
2713             cpu_to_le16(IEEE80211_FTYPE_CTL | IEEE80211_STYPE_CTS);
2714         cts->duration = ieee80211_ctstoself_duration(hw, vif,
2715                                                      frame_len, frame_txctl);
2716         memcpy(cts->ra, hdr->addr1, sizeof(cts->ra));
2717 }
2718 EXPORT_SYMBOL(ieee80211_ctstoself_get);
2719
2720 struct sk_buff *
2721 ieee80211_get_buffered_bc(struct ieee80211_hw *hw,
2722                           struct ieee80211_vif *vif)
2723 {
2724         struct ieee80211_local *local = hw_to_local(hw);
2725         struct sk_buff *skb = NULL;
2726         struct ieee80211_tx_data tx;
2727         struct ieee80211_sub_if_data *sdata;
2728         struct ps_data *ps;
2729         struct ieee80211_tx_info *info;
2730         struct ieee80211_chanctx_conf *chanctx_conf;
2731
2732         sdata = vif_to_sdata(vif);
2733
2734         rcu_read_lock();
2735         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
2736
2737         if (!chanctx_conf)
2738                 goto out;
2739
2740         if (sdata->vif.type == NL80211_IFTYPE_AP) {
2741                 struct beacon_data *beacon =
2742                                 rcu_dereference(sdata->u.ap.beacon);
2743
2744                 if (!beacon || !beacon->head)
2745                         goto out;
2746
2747                 ps = &sdata->u.ap.ps;
2748         } else if (ieee80211_vif_is_mesh(&sdata->vif)) {
2749                 ps = &sdata->u.mesh.ps;
2750         } else {
2751                 goto out;
2752         }
2753
2754         if (ps->dtim_count != 0 || !ps->dtim_bc_mc)
2755                 goto out; /* send buffered bc/mc only after DTIM beacon */
2756
2757         while (1) {
2758                 skb = skb_dequeue(&ps->bc_buf);
2759                 if (!skb)
2760                         goto out;
2761                 local->total_ps_buffered--;
2762
2763                 if (!skb_queue_empty(&ps->bc_buf) && skb->len >= 2) {
2764                         struct ieee80211_hdr *hdr =
2765                                 (struct ieee80211_hdr *) skb->data;
2766                         /* more buffered multicast/broadcast frames ==> set
2767                          * MoreData flag in IEEE 802.11 header to inform PS
2768                          * STAs */
2769                         hdr->frame_control |=
2770                                 cpu_to_le16(IEEE80211_FCTL_MOREDATA);
2771                 }
2772
2773                 if (!ieee80211_tx_prepare(sdata, &tx, skb))
2774                         break;
2775                 dev_kfree_skb_any(skb);
2776         }
2777
2778         info = IEEE80211_SKB_CB(skb);
2779
2780         tx.flags |= IEEE80211_TX_PS_BUFFERED;
2781         info->band = chanctx_conf->def.chan->band;
2782
2783         if (invoke_tx_handlers(&tx))
2784                 skb = NULL;
2785  out:
2786         rcu_read_unlock();
2787
2788         return skb;
2789 }
2790 EXPORT_SYMBOL(ieee80211_get_buffered_bc);
2791
2792 void __ieee80211_tx_skb_tid_band(struct ieee80211_sub_if_data *sdata,
2793                                  struct sk_buff *skb, int tid,
2794                                  enum ieee80211_band band)
2795 {
2796         int ac = ieee802_1d_to_ac[tid & 7];
2797
2798         skb_set_mac_header(skb, 0);
2799         skb_set_network_header(skb, 0);
2800         skb_set_transport_header(skb, 0);
2801
2802         skb_set_queue_mapping(skb, ac);
2803         skb->priority = tid;
2804
2805         /*
2806          * The other path calling ieee80211_xmit is from the tasklet,
2807          * and while we can handle concurrent transmissions locking
2808          * requirements are that we do not come into tx with bhs on.
2809          */
2810         local_bh_disable();
2811         ieee80211_xmit(sdata, skb, band);
2812         local_bh_enable();
2813 }