xref: /linux/net/mac80211/util.c (revision ac84bac4062e7fc24f5e2c61c6a414b2a00a29ad)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright 2002-2005, Instant802 Networks, Inc.
4  * Copyright 2005-2006, Devicescape Software, Inc.
5  * Copyright 2006-2007	Jiri Benc <jbenc@suse.cz>
6  * Copyright 2007	Johannes Berg <johannes@sipsolutions.net>
7  * Copyright 2013-2014  Intel Mobile Communications GmbH
8  * Copyright (C) 2015-2017	Intel Deutschland GmbH
9  * Copyright (C) 2018-2020 Intel Corporation
10  *
11  * utilities for mac80211
12  */
13 
14 #include <net/mac80211.h>
15 #include <linux/netdevice.h>
16 #include <linux/export.h>
17 #include <linux/types.h>
18 #include <linux/slab.h>
19 #include <linux/skbuff.h>
20 #include <linux/etherdevice.h>
21 #include <linux/if_arp.h>
22 #include <linux/bitmap.h>
23 #include <linux/crc32.h>
24 #include <net/net_namespace.h>
25 #include <net/cfg80211.h>
26 #include <net/rtnetlink.h>
27 
28 #include "ieee80211_i.h"
29 #include "driver-ops.h"
30 #include "rate.h"
31 #include "mesh.h"
32 #include "wme.h"
33 #include "led.h"
34 #include "wep.h"
35 
36 /* privid for wiphys to determine whether they belong to us or not */
37 const void *const mac80211_wiphy_privid = &mac80211_wiphy_privid;
38 
39 struct ieee80211_hw *wiphy_to_ieee80211_hw(struct wiphy *wiphy)
40 {
41 	struct ieee80211_local *local;
42 
43 	local = wiphy_priv(wiphy);
44 	return &local->hw;
45 }
46 EXPORT_SYMBOL(wiphy_to_ieee80211_hw);
47 
48 void ieee80211_tx_set_protected(struct ieee80211_tx_data *tx)
49 {
50 	struct sk_buff *skb;
51 	struct ieee80211_hdr *hdr;
52 
53 	skb_queue_walk(&tx->skbs, skb) {
54 		hdr = (struct ieee80211_hdr *) skb->data;
55 		hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
56 	}
57 }
58 
59 int ieee80211_frame_duration(enum nl80211_band band, size_t len,
60 			     int rate, int erp, int short_preamble,
61 			     int shift)
62 {
63 	int dur;
64 
65 	/* calculate duration (in microseconds, rounded up to next higher
66 	 * integer if it includes a fractional microsecond) to send frame of
67 	 * len bytes (does not include FCS) at the given rate. Duration will
68 	 * also include SIFS.
69 	 *
70 	 * rate is in 100 kbps, so divident is multiplied by 10 in the
71 	 * DIV_ROUND_UP() operations.
72 	 *
73 	 * shift may be 2 for 5 MHz channels or 1 for 10 MHz channels, and
74 	 * is assumed to be 0 otherwise.
75 	 */
76 
77 	if (band == NL80211_BAND_5GHZ || erp) {
78 		/*
79 		 * OFDM:
80 		 *
81 		 * N_DBPS = DATARATE x 4
82 		 * N_SYM = Ceiling((16+8xLENGTH+6) / N_DBPS)
83 		 *	(16 = SIGNAL time, 6 = tail bits)
84 		 * TXTIME = T_PREAMBLE + T_SIGNAL + T_SYM x N_SYM + Signal Ext
85 		 *
86 		 * T_SYM = 4 usec
87 		 * 802.11a - 18.5.2: aSIFSTime = 16 usec
88 		 * 802.11g - 19.8.4: aSIFSTime = 10 usec +
89 		 *	signal ext = 6 usec
90 		 */
91 		dur = 16; /* SIFS + signal ext */
92 		dur += 16; /* IEEE 802.11-2012 18.3.2.4: T_PREAMBLE = 16 usec */
93 		dur += 4; /* IEEE 802.11-2012 18.3.2.4: T_SIGNAL = 4 usec */
94 
95 		/* IEEE 802.11-2012 18.3.2.4: all values above are:
96 		 *  * times 4 for 5 MHz
97 		 *  * times 2 for 10 MHz
98 		 */
99 		dur *= 1 << shift;
100 
101 		/* rates should already consider the channel bandwidth,
102 		 * don't apply divisor again.
103 		 */
104 		dur += 4 * DIV_ROUND_UP((16 + 8 * (len + 4) + 6) * 10,
105 					4 * rate); /* T_SYM x N_SYM */
106 	} else {
107 		/*
108 		 * 802.11b or 802.11g with 802.11b compatibility:
109 		 * 18.3.4: TXTIME = PreambleLength + PLCPHeaderTime +
110 		 * Ceiling(((LENGTH+PBCC)x8)/DATARATE). PBCC=0.
111 		 *
112 		 * 802.11 (DS): 15.3.3, 802.11b: 18.3.4
113 		 * aSIFSTime = 10 usec
114 		 * aPreambleLength = 144 usec or 72 usec with short preamble
115 		 * aPLCPHeaderLength = 48 usec or 24 usec with short preamble
116 		 */
117 		dur = 10; /* aSIFSTime = 10 usec */
118 		dur += short_preamble ? (72 + 24) : (144 + 48);
119 
120 		dur += DIV_ROUND_UP(8 * (len + 4) * 10, rate);
121 	}
122 
123 	return dur;
124 }
125 
126 /* Exported duration function for driver use */
127 __le16 ieee80211_generic_frame_duration(struct ieee80211_hw *hw,
128 					struct ieee80211_vif *vif,
129 					enum nl80211_band band,
130 					size_t frame_len,
131 					struct ieee80211_rate *rate)
132 {
133 	struct ieee80211_sub_if_data *sdata;
134 	u16 dur;
135 	int erp, shift = 0;
136 	bool short_preamble = false;
137 
138 	erp = 0;
139 	if (vif) {
140 		sdata = vif_to_sdata(vif);
141 		short_preamble = sdata->vif.bss_conf.use_short_preamble;
142 		if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
143 			erp = rate->flags & IEEE80211_RATE_ERP_G;
144 		shift = ieee80211_vif_get_shift(vif);
145 	}
146 
147 	dur = ieee80211_frame_duration(band, frame_len, rate->bitrate, erp,
148 				       short_preamble, shift);
149 
150 	return cpu_to_le16(dur);
151 }
152 EXPORT_SYMBOL(ieee80211_generic_frame_duration);
153 
154 __le16 ieee80211_rts_duration(struct ieee80211_hw *hw,
155 			      struct ieee80211_vif *vif, size_t frame_len,
156 			      const struct ieee80211_tx_info *frame_txctl)
157 {
158 	struct ieee80211_local *local = hw_to_local(hw);
159 	struct ieee80211_rate *rate;
160 	struct ieee80211_sub_if_data *sdata;
161 	bool short_preamble;
162 	int erp, shift = 0, bitrate;
163 	u16 dur;
164 	struct ieee80211_supported_band *sband;
165 
166 	sband = local->hw.wiphy->bands[frame_txctl->band];
167 
168 	short_preamble = false;
169 
170 	rate = &sband->bitrates[frame_txctl->control.rts_cts_rate_idx];
171 
172 	erp = 0;
173 	if (vif) {
174 		sdata = vif_to_sdata(vif);
175 		short_preamble = sdata->vif.bss_conf.use_short_preamble;
176 		if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
177 			erp = rate->flags & IEEE80211_RATE_ERP_G;
178 		shift = ieee80211_vif_get_shift(vif);
179 	}
180 
181 	bitrate = DIV_ROUND_UP(rate->bitrate, 1 << shift);
182 
183 	/* CTS duration */
184 	dur = ieee80211_frame_duration(sband->band, 10, bitrate,
185 				       erp, short_preamble, shift);
186 	/* Data frame duration */
187 	dur += ieee80211_frame_duration(sband->band, frame_len, bitrate,
188 					erp, short_preamble, shift);
189 	/* ACK duration */
190 	dur += ieee80211_frame_duration(sband->band, 10, bitrate,
191 					erp, short_preamble, shift);
192 
193 	return cpu_to_le16(dur);
194 }
195 EXPORT_SYMBOL(ieee80211_rts_duration);
196 
197 __le16 ieee80211_ctstoself_duration(struct ieee80211_hw *hw,
198 				    struct ieee80211_vif *vif,
199 				    size_t frame_len,
200 				    const struct ieee80211_tx_info *frame_txctl)
201 {
202 	struct ieee80211_local *local = hw_to_local(hw);
203 	struct ieee80211_rate *rate;
204 	struct ieee80211_sub_if_data *sdata;
205 	bool short_preamble;
206 	int erp, shift = 0, bitrate;
207 	u16 dur;
208 	struct ieee80211_supported_band *sband;
209 
210 	sband = local->hw.wiphy->bands[frame_txctl->band];
211 
212 	short_preamble = false;
213 
214 	rate = &sband->bitrates[frame_txctl->control.rts_cts_rate_idx];
215 	erp = 0;
216 	if (vif) {
217 		sdata = vif_to_sdata(vif);
218 		short_preamble = sdata->vif.bss_conf.use_short_preamble;
219 		if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
220 			erp = rate->flags & IEEE80211_RATE_ERP_G;
221 		shift = ieee80211_vif_get_shift(vif);
222 	}
223 
224 	bitrate = DIV_ROUND_UP(rate->bitrate, 1 << shift);
225 
226 	/* Data frame duration */
227 	dur = ieee80211_frame_duration(sband->band, frame_len, bitrate,
228 				       erp, short_preamble, shift);
229 	if (!(frame_txctl->flags & IEEE80211_TX_CTL_NO_ACK)) {
230 		/* ACK duration */
231 		dur += ieee80211_frame_duration(sband->band, 10, bitrate,
232 						erp, short_preamble, shift);
233 	}
234 
235 	return cpu_to_le16(dur);
236 }
237 EXPORT_SYMBOL(ieee80211_ctstoself_duration);
238 
239 static void __ieee80211_wake_txqs(struct ieee80211_sub_if_data *sdata, int ac)
240 {
241 	struct ieee80211_local *local = sdata->local;
242 	struct ieee80211_vif *vif = &sdata->vif;
243 	struct fq *fq = &local->fq;
244 	struct ps_data *ps = NULL;
245 	struct txq_info *txqi;
246 	struct sta_info *sta;
247 	int i;
248 
249 	local_bh_disable();
250 	spin_lock(&fq->lock);
251 
252 	if (sdata->vif.type == NL80211_IFTYPE_AP)
253 		ps = &sdata->bss->ps;
254 
255 	sdata->vif.txqs_stopped[ac] = false;
256 
257 	list_for_each_entry_rcu(sta, &local->sta_list, list) {
258 		if (sdata != sta->sdata)
259 			continue;
260 
261 		for (i = 0; i < ARRAY_SIZE(sta->sta.txq); i++) {
262 			struct ieee80211_txq *txq = sta->sta.txq[i];
263 
264 			if (!txq)
265 				continue;
266 
267 			txqi = to_txq_info(txq);
268 
269 			if (ac != txq->ac)
270 				continue;
271 
272 			if (!test_and_clear_bit(IEEE80211_TXQ_STOP_NETIF_TX,
273 						&txqi->flags))
274 				continue;
275 
276 			spin_unlock(&fq->lock);
277 			drv_wake_tx_queue(local, txqi);
278 			spin_lock(&fq->lock);
279 		}
280 	}
281 
282 	if (!vif->txq)
283 		goto out;
284 
285 	txqi = to_txq_info(vif->txq);
286 
287 	if (!test_and_clear_bit(IEEE80211_TXQ_STOP_NETIF_TX, &txqi->flags) ||
288 	    (ps && atomic_read(&ps->num_sta_ps)) || ac != vif->txq->ac)
289 		goto out;
290 
291 	spin_unlock(&fq->lock);
292 
293 	drv_wake_tx_queue(local, txqi);
294 	local_bh_enable();
295 	return;
296 out:
297 	spin_unlock(&fq->lock);
298 	local_bh_enable();
299 }
300 
301 static void
302 __releases(&local->queue_stop_reason_lock)
303 __acquires(&local->queue_stop_reason_lock)
304 _ieee80211_wake_txqs(struct ieee80211_local *local, unsigned long *flags)
305 {
306 	struct ieee80211_sub_if_data *sdata;
307 	int n_acs = IEEE80211_NUM_ACS;
308 	int i;
309 
310 	rcu_read_lock();
311 
312 	if (local->hw.queues < IEEE80211_NUM_ACS)
313 		n_acs = 1;
314 
315 	for (i = 0; i < local->hw.queues; i++) {
316 		if (local->queue_stop_reasons[i])
317 			continue;
318 
319 		spin_unlock_irqrestore(&local->queue_stop_reason_lock, *flags);
320 		list_for_each_entry_rcu(sdata, &local->interfaces, list) {
321 			int ac;
322 
323 			for (ac = 0; ac < n_acs; ac++) {
324 				int ac_queue = sdata->vif.hw_queue[ac];
325 
326 				if (ac_queue == i ||
327 				    sdata->vif.cab_queue == i)
328 					__ieee80211_wake_txqs(sdata, ac);
329 			}
330 		}
331 		spin_lock_irqsave(&local->queue_stop_reason_lock, *flags);
332 	}
333 
334 	rcu_read_unlock();
335 }
336 
337 void ieee80211_wake_txqs(unsigned long data)
338 {
339 	struct ieee80211_local *local = (struct ieee80211_local *)data;
340 	unsigned long flags;
341 
342 	spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
343 	_ieee80211_wake_txqs(local, &flags);
344 	spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
345 }
346 
347 void ieee80211_propagate_queue_wake(struct ieee80211_local *local, int queue)
348 {
349 	struct ieee80211_sub_if_data *sdata;
350 	int n_acs = IEEE80211_NUM_ACS;
351 
352 	if (local->ops->wake_tx_queue)
353 		return;
354 
355 	if (local->hw.queues < IEEE80211_NUM_ACS)
356 		n_acs = 1;
357 
358 	list_for_each_entry_rcu(sdata, &local->interfaces, list) {
359 		int ac;
360 
361 		if (!sdata->dev)
362 			continue;
363 
364 		if (sdata->vif.cab_queue != IEEE80211_INVAL_HW_QUEUE &&
365 		    local->queue_stop_reasons[sdata->vif.cab_queue] != 0)
366 			continue;
367 
368 		for (ac = 0; ac < n_acs; ac++) {
369 			int ac_queue = sdata->vif.hw_queue[ac];
370 
371 			if (ac_queue == queue ||
372 			    (sdata->vif.cab_queue == queue &&
373 			     local->queue_stop_reasons[ac_queue] == 0 &&
374 			     skb_queue_empty(&local->pending[ac_queue])))
375 				netif_wake_subqueue(sdata->dev, ac);
376 		}
377 	}
378 }
379 
380 static void __ieee80211_wake_queue(struct ieee80211_hw *hw, int queue,
381 				   enum queue_stop_reason reason,
382 				   bool refcounted,
383 				   unsigned long *flags)
384 {
385 	struct ieee80211_local *local = hw_to_local(hw);
386 
387 	trace_wake_queue(local, queue, reason);
388 
389 	if (WARN_ON(queue >= hw->queues))
390 		return;
391 
392 	if (!test_bit(reason, &local->queue_stop_reasons[queue]))
393 		return;
394 
395 	if (!refcounted) {
396 		local->q_stop_reasons[queue][reason] = 0;
397 	} else {
398 		local->q_stop_reasons[queue][reason]--;
399 		if (WARN_ON(local->q_stop_reasons[queue][reason] < 0))
400 			local->q_stop_reasons[queue][reason] = 0;
401 	}
402 
403 	if (local->q_stop_reasons[queue][reason] == 0)
404 		__clear_bit(reason, &local->queue_stop_reasons[queue]);
405 
406 	if (local->queue_stop_reasons[queue] != 0)
407 		/* someone still has this queue stopped */
408 		return;
409 
410 	if (skb_queue_empty(&local->pending[queue])) {
411 		rcu_read_lock();
412 		ieee80211_propagate_queue_wake(local, queue);
413 		rcu_read_unlock();
414 	} else
415 		tasklet_schedule(&local->tx_pending_tasklet);
416 
417 	/*
418 	 * Calling _ieee80211_wake_txqs here can be a problem because it may
419 	 * release queue_stop_reason_lock which has been taken by
420 	 * __ieee80211_wake_queue's caller. It is certainly not very nice to
421 	 * release someone's lock, but it is fine because all the callers of
422 	 * __ieee80211_wake_queue call it right before releasing the lock.
423 	 */
424 	if (local->ops->wake_tx_queue) {
425 		if (reason == IEEE80211_QUEUE_STOP_REASON_DRIVER)
426 			tasklet_schedule(&local->wake_txqs_tasklet);
427 		else
428 			_ieee80211_wake_txqs(local, flags);
429 	}
430 }
431 
432 void ieee80211_wake_queue_by_reason(struct ieee80211_hw *hw, int queue,
433 				    enum queue_stop_reason reason,
434 				    bool refcounted)
435 {
436 	struct ieee80211_local *local = hw_to_local(hw);
437 	unsigned long flags;
438 
439 	spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
440 	__ieee80211_wake_queue(hw, queue, reason, refcounted, &flags);
441 	spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
442 }
443 
444 void ieee80211_wake_queue(struct ieee80211_hw *hw, int queue)
445 {
446 	ieee80211_wake_queue_by_reason(hw, queue,
447 				       IEEE80211_QUEUE_STOP_REASON_DRIVER,
448 				       false);
449 }
450 EXPORT_SYMBOL(ieee80211_wake_queue);
451 
452 static void __ieee80211_stop_queue(struct ieee80211_hw *hw, int queue,
453 				   enum queue_stop_reason reason,
454 				   bool refcounted)
455 {
456 	struct ieee80211_local *local = hw_to_local(hw);
457 	struct ieee80211_sub_if_data *sdata;
458 	int n_acs = IEEE80211_NUM_ACS;
459 
460 	trace_stop_queue(local, queue, reason);
461 
462 	if (WARN_ON(queue >= hw->queues))
463 		return;
464 
465 	if (!refcounted)
466 		local->q_stop_reasons[queue][reason] = 1;
467 	else
468 		local->q_stop_reasons[queue][reason]++;
469 
470 	if (__test_and_set_bit(reason, &local->queue_stop_reasons[queue]))
471 		return;
472 
473 	if (local->hw.queues < IEEE80211_NUM_ACS)
474 		n_acs = 1;
475 
476 	rcu_read_lock();
477 	list_for_each_entry_rcu(sdata, &local->interfaces, list) {
478 		int ac;
479 
480 		if (!sdata->dev)
481 			continue;
482 
483 		for (ac = 0; ac < n_acs; ac++) {
484 			if (sdata->vif.hw_queue[ac] == queue ||
485 			    sdata->vif.cab_queue == queue) {
486 				if (!local->ops->wake_tx_queue) {
487 					netif_stop_subqueue(sdata->dev, ac);
488 					continue;
489 				}
490 				spin_lock(&local->fq.lock);
491 				sdata->vif.txqs_stopped[ac] = true;
492 				spin_unlock(&local->fq.lock);
493 			}
494 		}
495 	}
496 	rcu_read_unlock();
497 }
498 
499 void ieee80211_stop_queue_by_reason(struct ieee80211_hw *hw, int queue,
500 				    enum queue_stop_reason reason,
501 				    bool refcounted)
502 {
503 	struct ieee80211_local *local = hw_to_local(hw);
504 	unsigned long flags;
505 
506 	spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
507 	__ieee80211_stop_queue(hw, queue, reason, refcounted);
508 	spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
509 }
510 
511 void ieee80211_stop_queue(struct ieee80211_hw *hw, int queue)
512 {
513 	ieee80211_stop_queue_by_reason(hw, queue,
514 				       IEEE80211_QUEUE_STOP_REASON_DRIVER,
515 				       false);
516 }
517 EXPORT_SYMBOL(ieee80211_stop_queue);
518 
519 void ieee80211_add_pending_skb(struct ieee80211_local *local,
520 			       struct sk_buff *skb)
521 {
522 	struct ieee80211_hw *hw = &local->hw;
523 	unsigned long flags;
524 	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
525 	int queue = info->hw_queue;
526 
527 	if (WARN_ON(!info->control.vif)) {
528 		ieee80211_free_txskb(&local->hw, skb);
529 		return;
530 	}
531 
532 	spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
533 	__ieee80211_stop_queue(hw, queue, IEEE80211_QUEUE_STOP_REASON_SKB_ADD,
534 			       false);
535 	__skb_queue_tail(&local->pending[queue], skb);
536 	__ieee80211_wake_queue(hw, queue, IEEE80211_QUEUE_STOP_REASON_SKB_ADD,
537 			       false, &flags);
538 	spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
539 }
540 
541 void ieee80211_add_pending_skbs(struct ieee80211_local *local,
542 				struct sk_buff_head *skbs)
543 {
544 	struct ieee80211_hw *hw = &local->hw;
545 	struct sk_buff *skb;
546 	unsigned long flags;
547 	int queue, i;
548 
549 	spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
550 	while ((skb = skb_dequeue(skbs))) {
551 		struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
552 
553 		if (WARN_ON(!info->control.vif)) {
554 			ieee80211_free_txskb(&local->hw, skb);
555 			continue;
556 		}
557 
558 		queue = info->hw_queue;
559 
560 		__ieee80211_stop_queue(hw, queue,
561 				IEEE80211_QUEUE_STOP_REASON_SKB_ADD,
562 				false);
563 
564 		__skb_queue_tail(&local->pending[queue], skb);
565 	}
566 
567 	for (i = 0; i < hw->queues; i++)
568 		__ieee80211_wake_queue(hw, i,
569 			IEEE80211_QUEUE_STOP_REASON_SKB_ADD,
570 			false, &flags);
571 	spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
572 }
573 
574 void ieee80211_stop_queues_by_reason(struct ieee80211_hw *hw,
575 				     unsigned long queues,
576 				     enum queue_stop_reason reason,
577 				     bool refcounted)
578 {
579 	struct ieee80211_local *local = hw_to_local(hw);
580 	unsigned long flags;
581 	int i;
582 
583 	spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
584 
585 	for_each_set_bit(i, &queues, hw->queues)
586 		__ieee80211_stop_queue(hw, i, reason, refcounted);
587 
588 	spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
589 }
590 
591 void ieee80211_stop_queues(struct ieee80211_hw *hw)
592 {
593 	ieee80211_stop_queues_by_reason(hw, IEEE80211_MAX_QUEUE_MAP,
594 					IEEE80211_QUEUE_STOP_REASON_DRIVER,
595 					false);
596 }
597 EXPORT_SYMBOL(ieee80211_stop_queues);
598 
599 int ieee80211_queue_stopped(struct ieee80211_hw *hw, int queue)
600 {
601 	struct ieee80211_local *local = hw_to_local(hw);
602 	unsigned long flags;
603 	int ret;
604 
605 	if (WARN_ON(queue >= hw->queues))
606 		return true;
607 
608 	spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
609 	ret = test_bit(IEEE80211_QUEUE_STOP_REASON_DRIVER,
610 		       &local->queue_stop_reasons[queue]);
611 	spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
612 	return ret;
613 }
614 EXPORT_SYMBOL(ieee80211_queue_stopped);
615 
616 void ieee80211_wake_queues_by_reason(struct ieee80211_hw *hw,
617 				     unsigned long queues,
618 				     enum queue_stop_reason reason,
619 				     bool refcounted)
620 {
621 	struct ieee80211_local *local = hw_to_local(hw);
622 	unsigned long flags;
623 	int i;
624 
625 	spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
626 
627 	for_each_set_bit(i, &queues, hw->queues)
628 		__ieee80211_wake_queue(hw, i, reason, refcounted, &flags);
629 
630 	spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
631 }
632 
633 void ieee80211_wake_queues(struct ieee80211_hw *hw)
634 {
635 	ieee80211_wake_queues_by_reason(hw, IEEE80211_MAX_QUEUE_MAP,
636 					IEEE80211_QUEUE_STOP_REASON_DRIVER,
637 					false);
638 }
639 EXPORT_SYMBOL(ieee80211_wake_queues);
640 
641 static unsigned int
642 ieee80211_get_vif_queues(struct ieee80211_local *local,
643 			 struct ieee80211_sub_if_data *sdata)
644 {
645 	unsigned int queues;
646 
647 	if (sdata && ieee80211_hw_check(&local->hw, QUEUE_CONTROL)) {
648 		int ac;
649 
650 		queues = 0;
651 
652 		for (ac = 0; ac < IEEE80211_NUM_ACS; ac++)
653 			queues |= BIT(sdata->vif.hw_queue[ac]);
654 		if (sdata->vif.cab_queue != IEEE80211_INVAL_HW_QUEUE)
655 			queues |= BIT(sdata->vif.cab_queue);
656 	} else {
657 		/* all queues */
658 		queues = BIT(local->hw.queues) - 1;
659 	}
660 
661 	return queues;
662 }
663 
664 void __ieee80211_flush_queues(struct ieee80211_local *local,
665 			      struct ieee80211_sub_if_data *sdata,
666 			      unsigned int queues, bool drop)
667 {
668 	if (!local->ops->flush)
669 		return;
670 
671 	/*
672 	 * If no queue was set, or if the HW doesn't support
673 	 * IEEE80211_HW_QUEUE_CONTROL - flush all queues
674 	 */
675 	if (!queues || !ieee80211_hw_check(&local->hw, QUEUE_CONTROL))
676 		queues = ieee80211_get_vif_queues(local, sdata);
677 
678 	ieee80211_stop_queues_by_reason(&local->hw, queues,
679 					IEEE80211_QUEUE_STOP_REASON_FLUSH,
680 					false);
681 
682 	drv_flush(local, sdata, queues, drop);
683 
684 	ieee80211_wake_queues_by_reason(&local->hw, queues,
685 					IEEE80211_QUEUE_STOP_REASON_FLUSH,
686 					false);
687 }
688 
689 void ieee80211_flush_queues(struct ieee80211_local *local,
690 			    struct ieee80211_sub_if_data *sdata, bool drop)
691 {
692 	__ieee80211_flush_queues(local, sdata, 0, drop);
693 }
694 
695 void ieee80211_stop_vif_queues(struct ieee80211_local *local,
696 			       struct ieee80211_sub_if_data *sdata,
697 			       enum queue_stop_reason reason)
698 {
699 	ieee80211_stop_queues_by_reason(&local->hw,
700 					ieee80211_get_vif_queues(local, sdata),
701 					reason, true);
702 }
703 
704 void ieee80211_wake_vif_queues(struct ieee80211_local *local,
705 			       struct ieee80211_sub_if_data *sdata,
706 			       enum queue_stop_reason reason)
707 {
708 	ieee80211_wake_queues_by_reason(&local->hw,
709 					ieee80211_get_vif_queues(local, sdata),
710 					reason, true);
711 }
712 
713 static void __iterate_interfaces(struct ieee80211_local *local,
714 				 u32 iter_flags,
715 				 void (*iterator)(void *data, u8 *mac,
716 						  struct ieee80211_vif *vif),
717 				 void *data)
718 {
719 	struct ieee80211_sub_if_data *sdata;
720 	bool active_only = iter_flags & IEEE80211_IFACE_ITER_ACTIVE;
721 
722 	list_for_each_entry_rcu(sdata, &local->interfaces, list) {
723 		switch (sdata->vif.type) {
724 		case NL80211_IFTYPE_MONITOR:
725 			if (!(sdata->u.mntr.flags & MONITOR_FLAG_ACTIVE))
726 				continue;
727 			break;
728 		case NL80211_IFTYPE_AP_VLAN:
729 			continue;
730 		default:
731 			break;
732 		}
733 		if (!(iter_flags & IEEE80211_IFACE_ITER_RESUME_ALL) &&
734 		    active_only && !(sdata->flags & IEEE80211_SDATA_IN_DRIVER))
735 			continue;
736 		if (ieee80211_sdata_running(sdata) || !active_only)
737 			iterator(data, sdata->vif.addr,
738 				 &sdata->vif);
739 	}
740 
741 	sdata = rcu_dereference_check(local->monitor_sdata,
742 				      lockdep_is_held(&local->iflist_mtx) ||
743 				      lockdep_rtnl_is_held());
744 	if (sdata &&
745 	    (iter_flags & IEEE80211_IFACE_ITER_RESUME_ALL || !active_only ||
746 	     sdata->flags & IEEE80211_SDATA_IN_DRIVER))
747 		iterator(data, sdata->vif.addr, &sdata->vif);
748 }
749 
750 void ieee80211_iterate_interfaces(
751 	struct ieee80211_hw *hw, u32 iter_flags,
752 	void (*iterator)(void *data, u8 *mac,
753 			 struct ieee80211_vif *vif),
754 	void *data)
755 {
756 	struct ieee80211_local *local = hw_to_local(hw);
757 
758 	mutex_lock(&local->iflist_mtx);
759 	__iterate_interfaces(local, iter_flags, iterator, data);
760 	mutex_unlock(&local->iflist_mtx);
761 }
762 EXPORT_SYMBOL_GPL(ieee80211_iterate_interfaces);
763 
764 void ieee80211_iterate_active_interfaces_atomic(
765 	struct ieee80211_hw *hw, u32 iter_flags,
766 	void (*iterator)(void *data, u8 *mac,
767 			 struct ieee80211_vif *vif),
768 	void *data)
769 {
770 	struct ieee80211_local *local = hw_to_local(hw);
771 
772 	rcu_read_lock();
773 	__iterate_interfaces(local, iter_flags | IEEE80211_IFACE_ITER_ACTIVE,
774 			     iterator, data);
775 	rcu_read_unlock();
776 }
777 EXPORT_SYMBOL_GPL(ieee80211_iterate_active_interfaces_atomic);
778 
779 void ieee80211_iterate_active_interfaces_rtnl(
780 	struct ieee80211_hw *hw, u32 iter_flags,
781 	void (*iterator)(void *data, u8 *mac,
782 			 struct ieee80211_vif *vif),
783 	void *data)
784 {
785 	struct ieee80211_local *local = hw_to_local(hw);
786 
787 	ASSERT_RTNL();
788 
789 	__iterate_interfaces(local, iter_flags | IEEE80211_IFACE_ITER_ACTIVE,
790 			     iterator, data);
791 }
792 EXPORT_SYMBOL_GPL(ieee80211_iterate_active_interfaces_rtnl);
793 
794 static void __iterate_stations(struct ieee80211_local *local,
795 			       void (*iterator)(void *data,
796 						struct ieee80211_sta *sta),
797 			       void *data)
798 {
799 	struct sta_info *sta;
800 
801 	list_for_each_entry_rcu(sta, &local->sta_list, list) {
802 		if (!sta->uploaded)
803 			continue;
804 
805 		iterator(data, &sta->sta);
806 	}
807 }
808 
809 void ieee80211_iterate_stations_atomic(struct ieee80211_hw *hw,
810 			void (*iterator)(void *data,
811 					 struct ieee80211_sta *sta),
812 			void *data)
813 {
814 	struct ieee80211_local *local = hw_to_local(hw);
815 
816 	rcu_read_lock();
817 	__iterate_stations(local, iterator, data);
818 	rcu_read_unlock();
819 }
820 EXPORT_SYMBOL_GPL(ieee80211_iterate_stations_atomic);
821 
822 struct ieee80211_vif *wdev_to_ieee80211_vif(struct wireless_dev *wdev)
823 {
824 	struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
825 
826 	if (!ieee80211_sdata_running(sdata) ||
827 	    !(sdata->flags & IEEE80211_SDATA_IN_DRIVER))
828 		return NULL;
829 	return &sdata->vif;
830 }
831 EXPORT_SYMBOL_GPL(wdev_to_ieee80211_vif);
832 
833 struct wireless_dev *ieee80211_vif_to_wdev(struct ieee80211_vif *vif)
834 {
835 	struct ieee80211_sub_if_data *sdata;
836 
837 	if (!vif)
838 		return NULL;
839 
840 	sdata = vif_to_sdata(vif);
841 
842 	if (!ieee80211_sdata_running(sdata) ||
843 	    !(sdata->flags & IEEE80211_SDATA_IN_DRIVER))
844 		return NULL;
845 
846 	return &sdata->wdev;
847 }
848 EXPORT_SYMBOL_GPL(ieee80211_vif_to_wdev);
849 
850 /*
851  * Nothing should have been stuffed into the workqueue during
852  * the suspend->resume cycle. Since we can't check each caller
853  * of this function if we are already quiescing / suspended,
854  * check here and don't WARN since this can actually happen when
855  * the rx path (for example) is racing against __ieee80211_suspend
856  * and suspending / quiescing was set after the rx path checked
857  * them.
858  */
859 static bool ieee80211_can_queue_work(struct ieee80211_local *local)
860 {
861 	if (local->quiescing || (local->suspended && !local->resuming)) {
862 		pr_warn("queueing ieee80211 work while going to suspend\n");
863 		return false;
864 	}
865 
866 	return true;
867 }
868 
869 void ieee80211_queue_work(struct ieee80211_hw *hw, struct work_struct *work)
870 {
871 	struct ieee80211_local *local = hw_to_local(hw);
872 
873 	if (!ieee80211_can_queue_work(local))
874 		return;
875 
876 	queue_work(local->workqueue, work);
877 }
878 EXPORT_SYMBOL(ieee80211_queue_work);
879 
880 void ieee80211_queue_delayed_work(struct ieee80211_hw *hw,
881 				  struct delayed_work *dwork,
882 				  unsigned long delay)
883 {
884 	struct ieee80211_local *local = hw_to_local(hw);
885 
886 	if (!ieee80211_can_queue_work(local))
887 		return;
888 
889 	queue_delayed_work(local->workqueue, dwork, delay);
890 }
891 EXPORT_SYMBOL(ieee80211_queue_delayed_work);
892 
893 static void ieee80211_parse_extension_element(u32 *crc,
894 					      const struct element *elem,
895 					      struct ieee802_11_elems *elems)
896 {
897 	const void *data = elem->data + 1;
898 	u8 len = elem->datalen - 1;
899 
900 	switch (elem->data[0]) {
901 	case WLAN_EID_EXT_HE_MU_EDCA:
902 		if (len == sizeof(*elems->mu_edca_param_set)) {
903 			elems->mu_edca_param_set = data;
904 			if (crc)
905 				*crc = crc32_be(*crc, (void *)elem,
906 						elem->datalen + 2);
907 		}
908 		break;
909 	case WLAN_EID_EXT_HE_CAPABILITY:
910 		elems->he_cap = data;
911 		elems->he_cap_len = len;
912 		break;
913 	case WLAN_EID_EXT_HE_OPERATION:
914 		if (len >= sizeof(*elems->he_operation) &&
915 		    len == ieee80211_he_oper_size(data) - 1) {
916 			if (crc)
917 				*crc = crc32_be(*crc, (void *)elem,
918 						elem->datalen + 2);
919 			elems->he_operation = data;
920 		}
921 		break;
922 	case WLAN_EID_EXT_UORA:
923 		if (len == 1)
924 			elems->uora_element = data;
925 		break;
926 	case WLAN_EID_EXT_MAX_CHANNEL_SWITCH_TIME:
927 		if (len == 3)
928 			elems->max_channel_switch_time = data;
929 		break;
930 	case WLAN_EID_EXT_MULTIPLE_BSSID_CONFIGURATION:
931 		if (len == sizeof(*elems->mbssid_config_ie))
932 			elems->mbssid_config_ie = data;
933 		break;
934 	case WLAN_EID_EXT_HE_SPR:
935 		if (len >= sizeof(*elems->he_spr) &&
936 		    len >= ieee80211_he_spr_size(data))
937 			elems->he_spr = data;
938 		break;
939 	}
940 }
941 
942 static u32
943 _ieee802_11_parse_elems_crc(const u8 *start, size_t len, bool action,
944 			    struct ieee802_11_elems *elems,
945 			    u64 filter, u32 crc,
946 			    const struct element *check_inherit)
947 {
948 	const struct element *elem;
949 	bool calc_crc = filter != 0;
950 	DECLARE_BITMAP(seen_elems, 256);
951 	const u8 *ie;
952 
953 	bitmap_zero(seen_elems, 256);
954 
955 	for_each_element(elem, start, len) {
956 		bool elem_parse_failed;
957 		u8 id = elem->id;
958 		u8 elen = elem->datalen;
959 		const u8 *pos = elem->data;
960 
961 		if (check_inherit &&
962 		    !cfg80211_is_element_inherited(elem,
963 						   check_inherit))
964 			continue;
965 
966 		switch (id) {
967 		case WLAN_EID_SSID:
968 		case WLAN_EID_SUPP_RATES:
969 		case WLAN_EID_FH_PARAMS:
970 		case WLAN_EID_DS_PARAMS:
971 		case WLAN_EID_CF_PARAMS:
972 		case WLAN_EID_TIM:
973 		case WLAN_EID_IBSS_PARAMS:
974 		case WLAN_EID_CHALLENGE:
975 		case WLAN_EID_RSN:
976 		case WLAN_EID_ERP_INFO:
977 		case WLAN_EID_EXT_SUPP_RATES:
978 		case WLAN_EID_HT_CAPABILITY:
979 		case WLAN_EID_HT_OPERATION:
980 		case WLAN_EID_VHT_CAPABILITY:
981 		case WLAN_EID_VHT_OPERATION:
982 		case WLAN_EID_MESH_ID:
983 		case WLAN_EID_MESH_CONFIG:
984 		case WLAN_EID_PEER_MGMT:
985 		case WLAN_EID_PREQ:
986 		case WLAN_EID_PREP:
987 		case WLAN_EID_PERR:
988 		case WLAN_EID_RANN:
989 		case WLAN_EID_CHANNEL_SWITCH:
990 		case WLAN_EID_EXT_CHANSWITCH_ANN:
991 		case WLAN_EID_COUNTRY:
992 		case WLAN_EID_PWR_CONSTRAINT:
993 		case WLAN_EID_TIMEOUT_INTERVAL:
994 		case WLAN_EID_SECONDARY_CHANNEL_OFFSET:
995 		case WLAN_EID_WIDE_BW_CHANNEL_SWITCH:
996 		case WLAN_EID_CHAN_SWITCH_PARAM:
997 		case WLAN_EID_EXT_CAPABILITY:
998 		case WLAN_EID_CHAN_SWITCH_TIMING:
999 		case WLAN_EID_LINK_ID:
1000 		case WLAN_EID_BSS_MAX_IDLE_PERIOD:
1001 		case WLAN_EID_RSNX:
1002 		/*
1003 		 * not listing WLAN_EID_CHANNEL_SWITCH_WRAPPER -- it seems possible
1004 		 * that if the content gets bigger it might be needed more than once
1005 		 */
1006 			if (test_bit(id, seen_elems)) {
1007 				elems->parse_error = true;
1008 				continue;
1009 			}
1010 			break;
1011 		}
1012 
1013 		if (calc_crc && id < 64 && (filter & (1ULL << id)))
1014 			crc = crc32_be(crc, pos - 2, elen + 2);
1015 
1016 		elem_parse_failed = false;
1017 
1018 		switch (id) {
1019 		case WLAN_EID_LINK_ID:
1020 			if (elen + 2 != sizeof(struct ieee80211_tdls_lnkie)) {
1021 				elem_parse_failed = true;
1022 				break;
1023 			}
1024 			elems->lnk_id = (void *)(pos - 2);
1025 			break;
1026 		case WLAN_EID_CHAN_SWITCH_TIMING:
1027 			if (elen != sizeof(struct ieee80211_ch_switch_timing)) {
1028 				elem_parse_failed = true;
1029 				break;
1030 			}
1031 			elems->ch_sw_timing = (void *)pos;
1032 			break;
1033 		case WLAN_EID_EXT_CAPABILITY:
1034 			elems->ext_capab = pos;
1035 			elems->ext_capab_len = elen;
1036 			break;
1037 		case WLAN_EID_SSID:
1038 			elems->ssid = pos;
1039 			elems->ssid_len = elen;
1040 			break;
1041 		case WLAN_EID_SUPP_RATES:
1042 			elems->supp_rates = pos;
1043 			elems->supp_rates_len = elen;
1044 			break;
1045 		case WLAN_EID_DS_PARAMS:
1046 			if (elen >= 1)
1047 				elems->ds_params = pos;
1048 			else
1049 				elem_parse_failed = true;
1050 			break;
1051 		case WLAN_EID_TIM:
1052 			if (elen >= sizeof(struct ieee80211_tim_ie)) {
1053 				elems->tim = (void *)pos;
1054 				elems->tim_len = elen;
1055 			} else
1056 				elem_parse_failed = true;
1057 			break;
1058 		case WLAN_EID_CHALLENGE:
1059 			elems->challenge = pos;
1060 			elems->challenge_len = elen;
1061 			break;
1062 		case WLAN_EID_VENDOR_SPECIFIC:
1063 			if (elen >= 4 && pos[0] == 0x00 && pos[1] == 0x50 &&
1064 			    pos[2] == 0xf2) {
1065 				/* Microsoft OUI (00:50:F2) */
1066 
1067 				if (calc_crc)
1068 					crc = crc32_be(crc, pos - 2, elen + 2);
1069 
1070 				if (elen >= 5 && pos[3] == 2) {
1071 					/* OUI Type 2 - WMM IE */
1072 					if (pos[4] == 0) {
1073 						elems->wmm_info = pos;
1074 						elems->wmm_info_len = elen;
1075 					} else if (pos[4] == 1) {
1076 						elems->wmm_param = pos;
1077 						elems->wmm_param_len = elen;
1078 					}
1079 				}
1080 			}
1081 			break;
1082 		case WLAN_EID_RSN:
1083 			elems->rsn = pos;
1084 			elems->rsn_len = elen;
1085 			break;
1086 		case WLAN_EID_ERP_INFO:
1087 			if (elen >= 1)
1088 				elems->erp_info = pos;
1089 			else
1090 				elem_parse_failed = true;
1091 			break;
1092 		case WLAN_EID_EXT_SUPP_RATES:
1093 			elems->ext_supp_rates = pos;
1094 			elems->ext_supp_rates_len = elen;
1095 			break;
1096 		case WLAN_EID_HT_CAPABILITY:
1097 			if (elen >= sizeof(struct ieee80211_ht_cap))
1098 				elems->ht_cap_elem = (void *)pos;
1099 			else
1100 				elem_parse_failed = true;
1101 			break;
1102 		case WLAN_EID_HT_OPERATION:
1103 			if (elen >= sizeof(struct ieee80211_ht_operation))
1104 				elems->ht_operation = (void *)pos;
1105 			else
1106 				elem_parse_failed = true;
1107 			break;
1108 		case WLAN_EID_VHT_CAPABILITY:
1109 			if (elen >= sizeof(struct ieee80211_vht_cap))
1110 				elems->vht_cap_elem = (void *)pos;
1111 			else
1112 				elem_parse_failed = true;
1113 			break;
1114 		case WLAN_EID_VHT_OPERATION:
1115 			if (elen >= sizeof(struct ieee80211_vht_operation)) {
1116 				elems->vht_operation = (void *)pos;
1117 				if (calc_crc)
1118 					crc = crc32_be(crc, pos - 2, elen + 2);
1119 				break;
1120 			}
1121 			elem_parse_failed = true;
1122 			break;
1123 		case WLAN_EID_OPMODE_NOTIF:
1124 			if (elen > 0) {
1125 				elems->opmode_notif = pos;
1126 				if (calc_crc)
1127 					crc = crc32_be(crc, pos - 2, elen + 2);
1128 				break;
1129 			}
1130 			elem_parse_failed = true;
1131 			break;
1132 		case WLAN_EID_MESH_ID:
1133 			elems->mesh_id = pos;
1134 			elems->mesh_id_len = elen;
1135 			break;
1136 		case WLAN_EID_MESH_CONFIG:
1137 			if (elen >= sizeof(struct ieee80211_meshconf_ie))
1138 				elems->mesh_config = (void *)pos;
1139 			else
1140 				elem_parse_failed = true;
1141 			break;
1142 		case WLAN_EID_PEER_MGMT:
1143 			elems->peering = pos;
1144 			elems->peering_len = elen;
1145 			break;
1146 		case WLAN_EID_MESH_AWAKE_WINDOW:
1147 			if (elen >= 2)
1148 				elems->awake_window = (void *)pos;
1149 			break;
1150 		case WLAN_EID_PREQ:
1151 			elems->preq = pos;
1152 			elems->preq_len = elen;
1153 			break;
1154 		case WLAN_EID_PREP:
1155 			elems->prep = pos;
1156 			elems->prep_len = elen;
1157 			break;
1158 		case WLAN_EID_PERR:
1159 			elems->perr = pos;
1160 			elems->perr_len = elen;
1161 			break;
1162 		case WLAN_EID_RANN:
1163 			if (elen >= sizeof(struct ieee80211_rann_ie))
1164 				elems->rann = (void *)pos;
1165 			else
1166 				elem_parse_failed = true;
1167 			break;
1168 		case WLAN_EID_CHANNEL_SWITCH:
1169 			if (elen != sizeof(struct ieee80211_channel_sw_ie)) {
1170 				elem_parse_failed = true;
1171 				break;
1172 			}
1173 			elems->ch_switch_ie = (void *)pos;
1174 			break;
1175 		case WLAN_EID_EXT_CHANSWITCH_ANN:
1176 			if (elen != sizeof(struct ieee80211_ext_chansw_ie)) {
1177 				elem_parse_failed = true;
1178 				break;
1179 			}
1180 			elems->ext_chansw_ie = (void *)pos;
1181 			break;
1182 		case WLAN_EID_SECONDARY_CHANNEL_OFFSET:
1183 			if (elen != sizeof(struct ieee80211_sec_chan_offs_ie)) {
1184 				elem_parse_failed = true;
1185 				break;
1186 			}
1187 			elems->sec_chan_offs = (void *)pos;
1188 			break;
1189 		case WLAN_EID_CHAN_SWITCH_PARAM:
1190 			if (elen !=
1191 			    sizeof(*elems->mesh_chansw_params_ie)) {
1192 				elem_parse_failed = true;
1193 				break;
1194 			}
1195 			elems->mesh_chansw_params_ie = (void *)pos;
1196 			break;
1197 		case WLAN_EID_WIDE_BW_CHANNEL_SWITCH:
1198 			if (!action ||
1199 			    elen != sizeof(*elems->wide_bw_chansw_ie)) {
1200 				elem_parse_failed = true;
1201 				break;
1202 			}
1203 			elems->wide_bw_chansw_ie = (void *)pos;
1204 			break;
1205 		case WLAN_EID_CHANNEL_SWITCH_WRAPPER:
1206 			if (action) {
1207 				elem_parse_failed = true;
1208 				break;
1209 			}
1210 			/*
1211 			 * This is a bit tricky, but as we only care about
1212 			 * the wide bandwidth channel switch element, so
1213 			 * just parse it out manually.
1214 			 */
1215 			ie = cfg80211_find_ie(WLAN_EID_WIDE_BW_CHANNEL_SWITCH,
1216 					      pos, elen);
1217 			if (ie) {
1218 				if (ie[1] == sizeof(*elems->wide_bw_chansw_ie))
1219 					elems->wide_bw_chansw_ie =
1220 						(void *)(ie + 2);
1221 				else
1222 					elem_parse_failed = true;
1223 			}
1224 			break;
1225 		case WLAN_EID_COUNTRY:
1226 			elems->country_elem = pos;
1227 			elems->country_elem_len = elen;
1228 			break;
1229 		case WLAN_EID_PWR_CONSTRAINT:
1230 			if (elen != 1) {
1231 				elem_parse_failed = true;
1232 				break;
1233 			}
1234 			elems->pwr_constr_elem = pos;
1235 			break;
1236 		case WLAN_EID_CISCO_VENDOR_SPECIFIC:
1237 			/* Lots of different options exist, but we only care
1238 			 * about the Dynamic Transmit Power Control element.
1239 			 * First check for the Cisco OUI, then for the DTPC
1240 			 * tag (0x00).
1241 			 */
1242 			if (elen < 4) {
1243 				elem_parse_failed = true;
1244 				break;
1245 			}
1246 
1247 			if (pos[0] != 0x00 || pos[1] != 0x40 ||
1248 			    pos[2] != 0x96 || pos[3] != 0x00)
1249 				break;
1250 
1251 			if (elen != 6) {
1252 				elem_parse_failed = true;
1253 				break;
1254 			}
1255 
1256 			if (calc_crc)
1257 				crc = crc32_be(crc, pos - 2, elen + 2);
1258 
1259 			elems->cisco_dtpc_elem = pos;
1260 			break;
1261 		case WLAN_EID_ADDBA_EXT:
1262 			if (elen != sizeof(struct ieee80211_addba_ext_ie)) {
1263 				elem_parse_failed = true;
1264 				break;
1265 			}
1266 			elems->addba_ext_ie = (void *)pos;
1267 			break;
1268 		case WLAN_EID_TIMEOUT_INTERVAL:
1269 			if (elen >= sizeof(struct ieee80211_timeout_interval_ie))
1270 				elems->timeout_int = (void *)pos;
1271 			else
1272 				elem_parse_failed = true;
1273 			break;
1274 		case WLAN_EID_BSS_MAX_IDLE_PERIOD:
1275 			if (elen >= sizeof(*elems->max_idle_period_ie))
1276 				elems->max_idle_period_ie = (void *)pos;
1277 			break;
1278 		case WLAN_EID_RSNX:
1279 			elems->rsnx = pos;
1280 			elems->rsnx_len = elen;
1281 			break;
1282 		case WLAN_EID_EXTENSION:
1283 			ieee80211_parse_extension_element(calc_crc ?
1284 								&crc : NULL,
1285 							  elem, elems);
1286 			break;
1287 		default:
1288 			break;
1289 		}
1290 
1291 		if (elem_parse_failed)
1292 			elems->parse_error = true;
1293 		else
1294 			__set_bit(id, seen_elems);
1295 	}
1296 
1297 	if (!for_each_element_completed(elem, start, len))
1298 		elems->parse_error = true;
1299 
1300 	return crc;
1301 }
1302 
1303 static size_t ieee802_11_find_bssid_profile(const u8 *start, size_t len,
1304 					    struct ieee802_11_elems *elems,
1305 					    u8 *transmitter_bssid,
1306 					    u8 *bss_bssid,
1307 					    u8 *nontransmitted_profile)
1308 {
1309 	const struct element *elem, *sub;
1310 	size_t profile_len = 0;
1311 	bool found = false;
1312 
1313 	if (!bss_bssid || !transmitter_bssid)
1314 		return profile_len;
1315 
1316 	for_each_element_id(elem, WLAN_EID_MULTIPLE_BSSID, start, len) {
1317 		if (elem->datalen < 2)
1318 			continue;
1319 
1320 		for_each_element(sub, elem->data + 1, elem->datalen - 1) {
1321 			u8 new_bssid[ETH_ALEN];
1322 			const u8 *index;
1323 
1324 			if (sub->id != 0 || sub->datalen < 4) {
1325 				/* not a valid BSS profile */
1326 				continue;
1327 			}
1328 
1329 			if (sub->data[0] != WLAN_EID_NON_TX_BSSID_CAP ||
1330 			    sub->data[1] != 2) {
1331 				/* The first element of the
1332 				 * Nontransmitted BSSID Profile is not
1333 				 * the Nontransmitted BSSID Capability
1334 				 * element.
1335 				 */
1336 				continue;
1337 			}
1338 
1339 			memset(nontransmitted_profile, 0, len);
1340 			profile_len = cfg80211_merge_profile(start, len,
1341 							     elem,
1342 							     sub,
1343 							     nontransmitted_profile,
1344 							     len);
1345 
1346 			/* found a Nontransmitted BSSID Profile */
1347 			index = cfg80211_find_ie(WLAN_EID_MULTI_BSSID_IDX,
1348 						 nontransmitted_profile,
1349 						 profile_len);
1350 			if (!index || index[1] < 1 || index[2] == 0) {
1351 				/* Invalid MBSSID Index element */
1352 				continue;
1353 			}
1354 
1355 			cfg80211_gen_new_bssid(transmitter_bssid,
1356 					       elem->data[0],
1357 					       index[2],
1358 					       new_bssid);
1359 			if (ether_addr_equal(new_bssid, bss_bssid)) {
1360 				found = true;
1361 				elems->bssid_index_len = index[1];
1362 				elems->bssid_index = (void *)&index[2];
1363 				break;
1364 			}
1365 		}
1366 	}
1367 
1368 	return found ? profile_len : 0;
1369 }
1370 
1371 u32 ieee802_11_parse_elems_crc(const u8 *start, size_t len, bool action,
1372 			       struct ieee802_11_elems *elems,
1373 			       u64 filter, u32 crc, u8 *transmitter_bssid,
1374 			       u8 *bss_bssid)
1375 {
1376 	const struct element *non_inherit = NULL;
1377 	u8 *nontransmitted_profile;
1378 	int nontransmitted_profile_len = 0;
1379 
1380 	memset(elems, 0, sizeof(*elems));
1381 	elems->ie_start = start;
1382 	elems->total_len = len;
1383 
1384 	nontransmitted_profile = kmalloc(len, GFP_ATOMIC);
1385 	if (nontransmitted_profile) {
1386 		nontransmitted_profile_len =
1387 			ieee802_11_find_bssid_profile(start, len, elems,
1388 						      transmitter_bssid,
1389 						      bss_bssid,
1390 						      nontransmitted_profile);
1391 		non_inherit =
1392 			cfg80211_find_ext_elem(WLAN_EID_EXT_NON_INHERITANCE,
1393 					       nontransmitted_profile,
1394 					       nontransmitted_profile_len);
1395 	}
1396 
1397 	crc = _ieee802_11_parse_elems_crc(start, len, action, elems, filter,
1398 					  crc, non_inherit);
1399 
1400 	/* Override with nontransmitted profile, if found */
1401 	if (nontransmitted_profile_len)
1402 		_ieee802_11_parse_elems_crc(nontransmitted_profile,
1403 					    nontransmitted_profile_len,
1404 					    action, elems, 0, 0, NULL);
1405 
1406 	if (elems->tim && !elems->parse_error) {
1407 		const struct ieee80211_tim_ie *tim_ie = elems->tim;
1408 
1409 		elems->dtim_period = tim_ie->dtim_period;
1410 		elems->dtim_count = tim_ie->dtim_count;
1411 	}
1412 
1413 	/* Override DTIM period and count if needed */
1414 	if (elems->bssid_index &&
1415 	    elems->bssid_index_len >=
1416 	    offsetofend(struct ieee80211_bssid_index, dtim_period))
1417 		elems->dtim_period = elems->bssid_index->dtim_period;
1418 
1419 	if (elems->bssid_index &&
1420 	    elems->bssid_index_len >=
1421 	    offsetofend(struct ieee80211_bssid_index, dtim_count))
1422 		elems->dtim_count = elems->bssid_index->dtim_count;
1423 
1424 	kfree(nontransmitted_profile);
1425 
1426 	return crc;
1427 }
1428 
1429 void ieee80211_regulatory_limit_wmm_params(struct ieee80211_sub_if_data *sdata,
1430 					   struct ieee80211_tx_queue_params
1431 					   *qparam, int ac)
1432 {
1433 	struct ieee80211_chanctx_conf *chanctx_conf;
1434 	const struct ieee80211_reg_rule *rrule;
1435 	const struct ieee80211_wmm_ac *wmm_ac;
1436 	u16 center_freq = 0;
1437 
1438 	if (sdata->vif.type != NL80211_IFTYPE_AP &&
1439 	    sdata->vif.type != NL80211_IFTYPE_STATION)
1440 		return;
1441 
1442 	rcu_read_lock();
1443 	chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1444 	if (chanctx_conf)
1445 		center_freq = chanctx_conf->def.chan->center_freq;
1446 
1447 	if (!center_freq) {
1448 		rcu_read_unlock();
1449 		return;
1450 	}
1451 
1452 	rrule = freq_reg_info(sdata->wdev.wiphy, MHZ_TO_KHZ(center_freq));
1453 
1454 	if (IS_ERR_OR_NULL(rrule) || !rrule->has_wmm) {
1455 		rcu_read_unlock();
1456 		return;
1457 	}
1458 
1459 	if (sdata->vif.type == NL80211_IFTYPE_AP)
1460 		wmm_ac = &rrule->wmm_rule.ap[ac];
1461 	else
1462 		wmm_ac = &rrule->wmm_rule.client[ac];
1463 	qparam->cw_min = max_t(u16, qparam->cw_min, wmm_ac->cw_min);
1464 	qparam->cw_max = max_t(u16, qparam->cw_max, wmm_ac->cw_max);
1465 	qparam->aifs = max_t(u8, qparam->aifs, wmm_ac->aifsn);
1466 	qparam->txop = min_t(u16, qparam->txop, wmm_ac->cot / 32);
1467 	rcu_read_unlock();
1468 }
1469 
1470 void ieee80211_set_wmm_default(struct ieee80211_sub_if_data *sdata,
1471 			       bool bss_notify, bool enable_qos)
1472 {
1473 	struct ieee80211_local *local = sdata->local;
1474 	struct ieee80211_tx_queue_params qparam;
1475 	struct ieee80211_chanctx_conf *chanctx_conf;
1476 	int ac;
1477 	bool use_11b;
1478 	bool is_ocb; /* Use another EDCA parameters if dot11OCBActivated=true */
1479 	int aCWmin, aCWmax;
1480 
1481 	if (!local->ops->conf_tx)
1482 		return;
1483 
1484 	if (local->hw.queues < IEEE80211_NUM_ACS)
1485 		return;
1486 
1487 	memset(&qparam, 0, sizeof(qparam));
1488 
1489 	rcu_read_lock();
1490 	chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1491 	use_11b = (chanctx_conf &&
1492 		   chanctx_conf->def.chan->band == NL80211_BAND_2GHZ) &&
1493 		 !(sdata->flags & IEEE80211_SDATA_OPERATING_GMODE);
1494 	rcu_read_unlock();
1495 
1496 	is_ocb = (sdata->vif.type == NL80211_IFTYPE_OCB);
1497 
1498 	/* Set defaults according to 802.11-2007 Table 7-37 */
1499 	aCWmax = 1023;
1500 	if (use_11b)
1501 		aCWmin = 31;
1502 	else
1503 		aCWmin = 15;
1504 
1505 	/* Confiure old 802.11b/g medium access rules. */
1506 	qparam.cw_max = aCWmax;
1507 	qparam.cw_min = aCWmin;
1508 	qparam.txop = 0;
1509 	qparam.aifs = 2;
1510 
1511 	for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
1512 		/* Update if QoS is enabled. */
1513 		if (enable_qos) {
1514 			switch (ac) {
1515 			case IEEE80211_AC_BK:
1516 				qparam.cw_max = aCWmax;
1517 				qparam.cw_min = aCWmin;
1518 				qparam.txop = 0;
1519 				if (is_ocb)
1520 					qparam.aifs = 9;
1521 				else
1522 					qparam.aifs = 7;
1523 				break;
1524 			/* never happens but let's not leave undefined */
1525 			default:
1526 			case IEEE80211_AC_BE:
1527 				qparam.cw_max = aCWmax;
1528 				qparam.cw_min = aCWmin;
1529 				qparam.txop = 0;
1530 				if (is_ocb)
1531 					qparam.aifs = 6;
1532 				else
1533 					qparam.aifs = 3;
1534 				break;
1535 			case IEEE80211_AC_VI:
1536 				qparam.cw_max = aCWmin;
1537 				qparam.cw_min = (aCWmin + 1) / 2 - 1;
1538 				if (is_ocb)
1539 					qparam.txop = 0;
1540 				else if (use_11b)
1541 					qparam.txop = 6016/32;
1542 				else
1543 					qparam.txop = 3008/32;
1544 
1545 				if (is_ocb)
1546 					qparam.aifs = 3;
1547 				else
1548 					qparam.aifs = 2;
1549 				break;
1550 			case IEEE80211_AC_VO:
1551 				qparam.cw_max = (aCWmin + 1) / 2 - 1;
1552 				qparam.cw_min = (aCWmin + 1) / 4 - 1;
1553 				if (is_ocb)
1554 					qparam.txop = 0;
1555 				else if (use_11b)
1556 					qparam.txop = 3264/32;
1557 				else
1558 					qparam.txop = 1504/32;
1559 				qparam.aifs = 2;
1560 				break;
1561 			}
1562 		}
1563 		ieee80211_regulatory_limit_wmm_params(sdata, &qparam, ac);
1564 
1565 		qparam.uapsd = false;
1566 
1567 		sdata->tx_conf[ac] = qparam;
1568 		drv_conf_tx(local, sdata, ac, &qparam);
1569 	}
1570 
1571 	if (sdata->vif.type != NL80211_IFTYPE_MONITOR &&
1572 	    sdata->vif.type != NL80211_IFTYPE_P2P_DEVICE &&
1573 	    sdata->vif.type != NL80211_IFTYPE_NAN) {
1574 		sdata->vif.bss_conf.qos = enable_qos;
1575 		if (bss_notify)
1576 			ieee80211_bss_info_change_notify(sdata,
1577 							 BSS_CHANGED_QOS);
1578 	}
1579 }
1580 
1581 void ieee80211_send_auth(struct ieee80211_sub_if_data *sdata,
1582 			 u16 transaction, u16 auth_alg, u16 status,
1583 			 const u8 *extra, size_t extra_len, const u8 *da,
1584 			 const u8 *bssid, const u8 *key, u8 key_len, u8 key_idx,
1585 			 u32 tx_flags)
1586 {
1587 	struct ieee80211_local *local = sdata->local;
1588 	struct sk_buff *skb;
1589 	struct ieee80211_mgmt *mgmt;
1590 	int err;
1591 
1592 	/* 24 + 6 = header + auth_algo + auth_transaction + status_code */
1593 	skb = dev_alloc_skb(local->hw.extra_tx_headroom + IEEE80211_WEP_IV_LEN +
1594 			    24 + 6 + extra_len + IEEE80211_WEP_ICV_LEN);
1595 	if (!skb)
1596 		return;
1597 
1598 	skb_reserve(skb, local->hw.extra_tx_headroom + IEEE80211_WEP_IV_LEN);
1599 
1600 	mgmt = skb_put_zero(skb, 24 + 6);
1601 	mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
1602 					  IEEE80211_STYPE_AUTH);
1603 	memcpy(mgmt->da, da, ETH_ALEN);
1604 	memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
1605 	memcpy(mgmt->bssid, bssid, ETH_ALEN);
1606 	mgmt->u.auth.auth_alg = cpu_to_le16(auth_alg);
1607 	mgmt->u.auth.auth_transaction = cpu_to_le16(transaction);
1608 	mgmt->u.auth.status_code = cpu_to_le16(status);
1609 	if (extra)
1610 		skb_put_data(skb, extra, extra_len);
1611 
1612 	if (auth_alg == WLAN_AUTH_SHARED_KEY && transaction == 3) {
1613 		mgmt->frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
1614 		err = ieee80211_wep_encrypt(local, skb, key, key_len, key_idx);
1615 		WARN_ON(err);
1616 	}
1617 
1618 	IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT |
1619 					tx_flags;
1620 	ieee80211_tx_skb(sdata, skb);
1621 }
1622 
1623 void ieee80211_send_deauth_disassoc(struct ieee80211_sub_if_data *sdata,
1624 				    const u8 *da, const u8 *bssid,
1625 				    u16 stype, u16 reason,
1626 				    bool send_frame, u8 *frame_buf)
1627 {
1628 	struct ieee80211_local *local = sdata->local;
1629 	struct sk_buff *skb;
1630 	struct ieee80211_mgmt *mgmt = (void *)frame_buf;
1631 
1632 	/* build frame */
1633 	mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT | stype);
1634 	mgmt->duration = 0; /* initialize only */
1635 	mgmt->seq_ctrl = 0; /* initialize only */
1636 	memcpy(mgmt->da, da, ETH_ALEN);
1637 	memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
1638 	memcpy(mgmt->bssid, bssid, ETH_ALEN);
1639 	/* u.deauth.reason_code == u.disassoc.reason_code */
1640 	mgmt->u.deauth.reason_code = cpu_to_le16(reason);
1641 
1642 	if (send_frame) {
1643 		skb = dev_alloc_skb(local->hw.extra_tx_headroom +
1644 				    IEEE80211_DEAUTH_FRAME_LEN);
1645 		if (!skb)
1646 			return;
1647 
1648 		skb_reserve(skb, local->hw.extra_tx_headroom);
1649 
1650 		/* copy in frame */
1651 		skb_put_data(skb, mgmt, IEEE80211_DEAUTH_FRAME_LEN);
1652 
1653 		if (sdata->vif.type != NL80211_IFTYPE_STATION ||
1654 		    !(sdata->u.mgd.flags & IEEE80211_STA_MFP_ENABLED))
1655 			IEEE80211_SKB_CB(skb)->flags |=
1656 				IEEE80211_TX_INTFL_DONT_ENCRYPT;
1657 
1658 		ieee80211_tx_skb(sdata, skb);
1659 	}
1660 }
1661 
1662 static int ieee80211_build_preq_ies_band(struct ieee80211_local *local,
1663 					 u8 *buffer, size_t buffer_len,
1664 					 const u8 *ie, size_t ie_len,
1665 					 enum nl80211_band band,
1666 					 u32 rate_mask,
1667 					 struct cfg80211_chan_def *chandef,
1668 					 size_t *offset, u32 flags)
1669 {
1670 	struct ieee80211_supported_band *sband;
1671 	const struct ieee80211_sta_he_cap *he_cap;
1672 	u8 *pos = buffer, *end = buffer + buffer_len;
1673 	size_t noffset;
1674 	int supp_rates_len, i;
1675 	u8 rates[32];
1676 	int num_rates;
1677 	int ext_rates_len;
1678 	int shift;
1679 	u32 rate_flags;
1680 	bool have_80mhz = false;
1681 
1682 	*offset = 0;
1683 
1684 	sband = local->hw.wiphy->bands[band];
1685 	if (WARN_ON_ONCE(!sband))
1686 		return 0;
1687 
1688 	rate_flags = ieee80211_chandef_rate_flags(chandef);
1689 	shift = ieee80211_chandef_get_shift(chandef);
1690 
1691 	num_rates = 0;
1692 	for (i = 0; i < sband->n_bitrates; i++) {
1693 		if ((BIT(i) & rate_mask) == 0)
1694 			continue; /* skip rate */
1695 		if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
1696 			continue;
1697 
1698 		rates[num_rates++] =
1699 			(u8) DIV_ROUND_UP(sband->bitrates[i].bitrate,
1700 					  (1 << shift) * 5);
1701 	}
1702 
1703 	supp_rates_len = min_t(int, num_rates, 8);
1704 
1705 	if (end - pos < 2 + supp_rates_len)
1706 		goto out_err;
1707 	*pos++ = WLAN_EID_SUPP_RATES;
1708 	*pos++ = supp_rates_len;
1709 	memcpy(pos, rates, supp_rates_len);
1710 	pos += supp_rates_len;
1711 
1712 	/* insert "request information" if in custom IEs */
1713 	if (ie && ie_len) {
1714 		static const u8 before_extrates[] = {
1715 			WLAN_EID_SSID,
1716 			WLAN_EID_SUPP_RATES,
1717 			WLAN_EID_REQUEST,
1718 		};
1719 		noffset = ieee80211_ie_split(ie, ie_len,
1720 					     before_extrates,
1721 					     ARRAY_SIZE(before_extrates),
1722 					     *offset);
1723 		if (end - pos < noffset - *offset)
1724 			goto out_err;
1725 		memcpy(pos, ie + *offset, noffset - *offset);
1726 		pos += noffset - *offset;
1727 		*offset = noffset;
1728 	}
1729 
1730 	ext_rates_len = num_rates - supp_rates_len;
1731 	if (ext_rates_len > 0) {
1732 		if (end - pos < 2 + ext_rates_len)
1733 			goto out_err;
1734 		*pos++ = WLAN_EID_EXT_SUPP_RATES;
1735 		*pos++ = ext_rates_len;
1736 		memcpy(pos, rates + supp_rates_len, ext_rates_len);
1737 		pos += ext_rates_len;
1738 	}
1739 
1740 	if (chandef->chan && sband->band == NL80211_BAND_2GHZ) {
1741 		if (end - pos < 3)
1742 			goto out_err;
1743 		*pos++ = WLAN_EID_DS_PARAMS;
1744 		*pos++ = 1;
1745 		*pos++ = ieee80211_frequency_to_channel(
1746 				chandef->chan->center_freq);
1747 	}
1748 
1749 	if (flags & IEEE80211_PROBE_FLAG_MIN_CONTENT)
1750 		goto done;
1751 
1752 	/* insert custom IEs that go before HT */
1753 	if (ie && ie_len) {
1754 		static const u8 before_ht[] = {
1755 			/*
1756 			 * no need to list the ones split off already
1757 			 * (or generated here)
1758 			 */
1759 			WLAN_EID_DS_PARAMS,
1760 			WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
1761 		};
1762 		noffset = ieee80211_ie_split(ie, ie_len,
1763 					     before_ht, ARRAY_SIZE(before_ht),
1764 					     *offset);
1765 		if (end - pos < noffset - *offset)
1766 			goto out_err;
1767 		memcpy(pos, ie + *offset, noffset - *offset);
1768 		pos += noffset - *offset;
1769 		*offset = noffset;
1770 	}
1771 
1772 	if (sband->ht_cap.ht_supported) {
1773 		if (end - pos < 2 + sizeof(struct ieee80211_ht_cap))
1774 			goto out_err;
1775 		pos = ieee80211_ie_build_ht_cap(pos, &sband->ht_cap,
1776 						sband->ht_cap.cap);
1777 	}
1778 
1779 	/* insert custom IEs that go before VHT */
1780 	if (ie && ie_len) {
1781 		static const u8 before_vht[] = {
1782 			/*
1783 			 * no need to list the ones split off already
1784 			 * (or generated here)
1785 			 */
1786 			WLAN_EID_BSS_COEX_2040,
1787 			WLAN_EID_EXT_CAPABILITY,
1788 			WLAN_EID_SSID_LIST,
1789 			WLAN_EID_CHANNEL_USAGE,
1790 			WLAN_EID_INTERWORKING,
1791 			WLAN_EID_MESH_ID,
1792 			/* 60 GHz (Multi-band, DMG, MMS) can't happen */
1793 		};
1794 		noffset = ieee80211_ie_split(ie, ie_len,
1795 					     before_vht, ARRAY_SIZE(before_vht),
1796 					     *offset);
1797 		if (end - pos < noffset - *offset)
1798 			goto out_err;
1799 		memcpy(pos, ie + *offset, noffset - *offset);
1800 		pos += noffset - *offset;
1801 		*offset = noffset;
1802 	}
1803 
1804 	/* Check if any channel in this sband supports at least 80 MHz */
1805 	for (i = 0; i < sband->n_channels; i++) {
1806 		if (sband->channels[i].flags & (IEEE80211_CHAN_DISABLED |
1807 						IEEE80211_CHAN_NO_80MHZ))
1808 			continue;
1809 
1810 		have_80mhz = true;
1811 		break;
1812 	}
1813 
1814 	if (sband->vht_cap.vht_supported && have_80mhz) {
1815 		if (end - pos < 2 + sizeof(struct ieee80211_vht_cap))
1816 			goto out_err;
1817 		pos = ieee80211_ie_build_vht_cap(pos, &sband->vht_cap,
1818 						 sband->vht_cap.cap);
1819 	}
1820 
1821 	/* insert custom IEs that go before HE */
1822 	if (ie && ie_len) {
1823 		static const u8 before_he[] = {
1824 			/*
1825 			 * no need to list the ones split off before VHT
1826 			 * or generated here
1827 			 */
1828 			WLAN_EID_EXTENSION, WLAN_EID_EXT_FILS_REQ_PARAMS,
1829 			WLAN_EID_AP_CSN,
1830 			/* TODO: add 11ah/11aj/11ak elements */
1831 		};
1832 		noffset = ieee80211_ie_split(ie, ie_len,
1833 					     before_he, ARRAY_SIZE(before_he),
1834 					     *offset);
1835 		if (end - pos < noffset - *offset)
1836 			goto out_err;
1837 		memcpy(pos, ie + *offset, noffset - *offset);
1838 		pos += noffset - *offset;
1839 		*offset = noffset;
1840 	}
1841 
1842 	he_cap = ieee80211_get_he_sta_cap(sband);
1843 	if (he_cap) {
1844 		pos = ieee80211_ie_build_he_cap(pos, he_cap, end);
1845 		if (!pos)
1846 			goto out_err;
1847 	}
1848 
1849 	/*
1850 	 * If adding more here, adjust code in main.c
1851 	 * that calculates local->scan_ies_len.
1852 	 */
1853 
1854 	return pos - buffer;
1855  out_err:
1856 	WARN_ONCE(1, "not enough space for preq IEs\n");
1857  done:
1858 	return pos - buffer;
1859 }
1860 
1861 int ieee80211_build_preq_ies(struct ieee80211_local *local, u8 *buffer,
1862 			     size_t buffer_len,
1863 			     struct ieee80211_scan_ies *ie_desc,
1864 			     const u8 *ie, size_t ie_len,
1865 			     u8 bands_used, u32 *rate_masks,
1866 			     struct cfg80211_chan_def *chandef,
1867 			     u32 flags)
1868 {
1869 	size_t pos = 0, old_pos = 0, custom_ie_offset = 0;
1870 	int i;
1871 
1872 	memset(ie_desc, 0, sizeof(*ie_desc));
1873 
1874 	for (i = 0; i < NUM_NL80211_BANDS; i++) {
1875 		if (bands_used & BIT(i)) {
1876 			pos += ieee80211_build_preq_ies_band(local,
1877 							     buffer + pos,
1878 							     buffer_len - pos,
1879 							     ie, ie_len, i,
1880 							     rate_masks[i],
1881 							     chandef,
1882 							     &custom_ie_offset,
1883 							     flags);
1884 			ie_desc->ies[i] = buffer + old_pos;
1885 			ie_desc->len[i] = pos - old_pos;
1886 			old_pos = pos;
1887 		}
1888 	}
1889 
1890 	/* add any remaining custom IEs */
1891 	if (ie && ie_len) {
1892 		if (WARN_ONCE(buffer_len - pos < ie_len - custom_ie_offset,
1893 			      "not enough space for preq custom IEs\n"))
1894 			return pos;
1895 		memcpy(buffer + pos, ie + custom_ie_offset,
1896 		       ie_len - custom_ie_offset);
1897 		ie_desc->common_ies = buffer + pos;
1898 		ie_desc->common_ie_len = ie_len - custom_ie_offset;
1899 		pos += ie_len - custom_ie_offset;
1900 	}
1901 
1902 	return pos;
1903 };
1904 
1905 struct sk_buff *ieee80211_build_probe_req(struct ieee80211_sub_if_data *sdata,
1906 					  const u8 *src, const u8 *dst,
1907 					  u32 ratemask,
1908 					  struct ieee80211_channel *chan,
1909 					  const u8 *ssid, size_t ssid_len,
1910 					  const u8 *ie, size_t ie_len,
1911 					  u32 flags)
1912 {
1913 	struct ieee80211_local *local = sdata->local;
1914 	struct cfg80211_chan_def chandef;
1915 	struct sk_buff *skb;
1916 	struct ieee80211_mgmt *mgmt;
1917 	int ies_len;
1918 	u32 rate_masks[NUM_NL80211_BANDS] = {};
1919 	struct ieee80211_scan_ies dummy_ie_desc;
1920 
1921 	/*
1922 	 * Do not send DS Channel parameter for directed probe requests
1923 	 * in order to maximize the chance that we get a response.  Some
1924 	 * badly-behaved APs don't respond when this parameter is included.
1925 	 */
1926 	chandef.width = sdata->vif.bss_conf.chandef.width;
1927 	if (flags & IEEE80211_PROBE_FLAG_DIRECTED)
1928 		chandef.chan = NULL;
1929 	else
1930 		chandef.chan = chan;
1931 
1932 	skb = ieee80211_probereq_get(&local->hw, src, ssid, ssid_len,
1933 				     100 + ie_len);
1934 	if (!skb)
1935 		return NULL;
1936 
1937 	rate_masks[chan->band] = ratemask;
1938 	ies_len = ieee80211_build_preq_ies(local, skb_tail_pointer(skb),
1939 					   skb_tailroom(skb), &dummy_ie_desc,
1940 					   ie, ie_len, BIT(chan->band),
1941 					   rate_masks, &chandef, flags);
1942 	skb_put(skb, ies_len);
1943 
1944 	if (dst) {
1945 		mgmt = (struct ieee80211_mgmt *) skb->data;
1946 		memcpy(mgmt->da, dst, ETH_ALEN);
1947 		memcpy(mgmt->bssid, dst, ETH_ALEN);
1948 	}
1949 
1950 	IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
1951 
1952 	return skb;
1953 }
1954 
1955 u32 ieee80211_sta_get_rates(struct ieee80211_sub_if_data *sdata,
1956 			    struct ieee802_11_elems *elems,
1957 			    enum nl80211_band band, u32 *basic_rates)
1958 {
1959 	struct ieee80211_supported_band *sband;
1960 	size_t num_rates;
1961 	u32 supp_rates, rate_flags;
1962 	int i, j, shift;
1963 
1964 	sband = sdata->local->hw.wiphy->bands[band];
1965 	if (WARN_ON(!sband))
1966 		return 1;
1967 
1968 	rate_flags = ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef);
1969 	shift = ieee80211_vif_get_shift(&sdata->vif);
1970 
1971 	num_rates = sband->n_bitrates;
1972 	supp_rates = 0;
1973 	for (i = 0; i < elems->supp_rates_len +
1974 		     elems->ext_supp_rates_len; i++) {
1975 		u8 rate = 0;
1976 		int own_rate;
1977 		bool is_basic;
1978 		if (i < elems->supp_rates_len)
1979 			rate = elems->supp_rates[i];
1980 		else if (elems->ext_supp_rates)
1981 			rate = elems->ext_supp_rates
1982 				[i - elems->supp_rates_len];
1983 		own_rate = 5 * (rate & 0x7f);
1984 		is_basic = !!(rate & 0x80);
1985 
1986 		if (is_basic && (rate & 0x7f) == BSS_MEMBERSHIP_SELECTOR_HT_PHY)
1987 			continue;
1988 
1989 		for (j = 0; j < num_rates; j++) {
1990 			int brate;
1991 			if ((rate_flags & sband->bitrates[j].flags)
1992 			    != rate_flags)
1993 				continue;
1994 
1995 			brate = DIV_ROUND_UP(sband->bitrates[j].bitrate,
1996 					     1 << shift);
1997 
1998 			if (brate == own_rate) {
1999 				supp_rates |= BIT(j);
2000 				if (basic_rates && is_basic)
2001 					*basic_rates |= BIT(j);
2002 			}
2003 		}
2004 	}
2005 	return supp_rates;
2006 }
2007 
2008 void ieee80211_stop_device(struct ieee80211_local *local)
2009 {
2010 	ieee80211_led_radio(local, false);
2011 	ieee80211_mod_tpt_led_trig(local, 0, IEEE80211_TPT_LEDTRIG_FL_RADIO);
2012 
2013 	cancel_work_sync(&local->reconfig_filter);
2014 
2015 	flush_workqueue(local->workqueue);
2016 	drv_stop(local);
2017 }
2018 
2019 static void ieee80211_flush_completed_scan(struct ieee80211_local *local,
2020 					   bool aborted)
2021 {
2022 	/* It's possible that we don't handle the scan completion in
2023 	 * time during suspend, so if it's still marked as completed
2024 	 * here, queue the work and flush it to clean things up.
2025 	 * Instead of calling the worker function directly here, we
2026 	 * really queue it to avoid potential races with other flows
2027 	 * scheduling the same work.
2028 	 */
2029 	if (test_bit(SCAN_COMPLETED, &local->scanning)) {
2030 		/* If coming from reconfiguration failure, abort the scan so
2031 		 * we don't attempt to continue a partial HW scan - which is
2032 		 * possible otherwise if (e.g.) the 2.4 GHz portion was the
2033 		 * completed scan, and a 5 GHz portion is still pending.
2034 		 */
2035 		if (aborted)
2036 			set_bit(SCAN_ABORTED, &local->scanning);
2037 		ieee80211_queue_delayed_work(&local->hw, &local->scan_work, 0);
2038 		flush_delayed_work(&local->scan_work);
2039 	}
2040 }
2041 
2042 static void ieee80211_handle_reconfig_failure(struct ieee80211_local *local)
2043 {
2044 	struct ieee80211_sub_if_data *sdata;
2045 	struct ieee80211_chanctx *ctx;
2046 
2047 	/*
2048 	 * We get here if during resume the device can't be restarted properly.
2049 	 * We might also get here if this happens during HW reset, which is a
2050 	 * slightly different situation and we need to drop all connections in
2051 	 * the latter case.
2052 	 *
2053 	 * Ask cfg80211 to turn off all interfaces, this will result in more
2054 	 * warnings but at least we'll then get into a clean stopped state.
2055 	 */
2056 
2057 	local->resuming = false;
2058 	local->suspended = false;
2059 	local->in_reconfig = false;
2060 
2061 	ieee80211_flush_completed_scan(local, true);
2062 
2063 	/* scheduled scan clearly can't be running any more, but tell
2064 	 * cfg80211 and clear local state
2065 	 */
2066 	ieee80211_sched_scan_end(local);
2067 
2068 	list_for_each_entry(sdata, &local->interfaces, list)
2069 		sdata->flags &= ~IEEE80211_SDATA_IN_DRIVER;
2070 
2071 	/* Mark channel contexts as not being in the driver any more to avoid
2072 	 * removing them from the driver during the shutdown process...
2073 	 */
2074 	mutex_lock(&local->chanctx_mtx);
2075 	list_for_each_entry(ctx, &local->chanctx_list, list)
2076 		ctx->driver_present = false;
2077 	mutex_unlock(&local->chanctx_mtx);
2078 
2079 	cfg80211_shutdown_all_interfaces(local->hw.wiphy);
2080 }
2081 
2082 static void ieee80211_assign_chanctx(struct ieee80211_local *local,
2083 				     struct ieee80211_sub_if_data *sdata)
2084 {
2085 	struct ieee80211_chanctx_conf *conf;
2086 	struct ieee80211_chanctx *ctx;
2087 
2088 	if (!local->use_chanctx)
2089 		return;
2090 
2091 	mutex_lock(&local->chanctx_mtx);
2092 	conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
2093 					 lockdep_is_held(&local->chanctx_mtx));
2094 	if (conf) {
2095 		ctx = container_of(conf, struct ieee80211_chanctx, conf);
2096 		drv_assign_vif_chanctx(local, sdata, ctx);
2097 	}
2098 	mutex_unlock(&local->chanctx_mtx);
2099 }
2100 
2101 static void ieee80211_reconfig_stations(struct ieee80211_sub_if_data *sdata)
2102 {
2103 	struct ieee80211_local *local = sdata->local;
2104 	struct sta_info *sta;
2105 
2106 	/* add STAs back */
2107 	mutex_lock(&local->sta_mtx);
2108 	list_for_each_entry(sta, &local->sta_list, list) {
2109 		enum ieee80211_sta_state state;
2110 
2111 		if (!sta->uploaded || sta->sdata != sdata)
2112 			continue;
2113 
2114 		for (state = IEEE80211_STA_NOTEXIST;
2115 		     state < sta->sta_state; state++)
2116 			WARN_ON(drv_sta_state(local, sta->sdata, sta, state,
2117 					      state + 1));
2118 	}
2119 	mutex_unlock(&local->sta_mtx);
2120 }
2121 
2122 static int ieee80211_reconfig_nan(struct ieee80211_sub_if_data *sdata)
2123 {
2124 	struct cfg80211_nan_func *func, **funcs;
2125 	int res, id, i = 0;
2126 
2127 	res = drv_start_nan(sdata->local, sdata,
2128 			    &sdata->u.nan.conf);
2129 	if (WARN_ON(res))
2130 		return res;
2131 
2132 	funcs = kcalloc(sdata->local->hw.max_nan_de_entries + 1,
2133 			sizeof(*funcs),
2134 			GFP_KERNEL);
2135 	if (!funcs)
2136 		return -ENOMEM;
2137 
2138 	/* Add all the functions:
2139 	 * This is a little bit ugly. We need to call a potentially sleeping
2140 	 * callback for each NAN function, so we can't hold the spinlock.
2141 	 */
2142 	spin_lock_bh(&sdata->u.nan.func_lock);
2143 
2144 	idr_for_each_entry(&sdata->u.nan.function_inst_ids, func, id)
2145 		funcs[i++] = func;
2146 
2147 	spin_unlock_bh(&sdata->u.nan.func_lock);
2148 
2149 	for (i = 0; funcs[i]; i++) {
2150 		res = drv_add_nan_func(sdata->local, sdata, funcs[i]);
2151 		if (WARN_ON(res))
2152 			ieee80211_nan_func_terminated(&sdata->vif,
2153 						      funcs[i]->instance_id,
2154 						      NL80211_NAN_FUNC_TERM_REASON_ERROR,
2155 						      GFP_KERNEL);
2156 	}
2157 
2158 	kfree(funcs);
2159 
2160 	return 0;
2161 }
2162 
2163 int ieee80211_reconfig(struct ieee80211_local *local)
2164 {
2165 	struct ieee80211_hw *hw = &local->hw;
2166 	struct ieee80211_sub_if_data *sdata;
2167 	struct ieee80211_chanctx *ctx;
2168 	struct sta_info *sta;
2169 	int res, i;
2170 	bool reconfig_due_to_wowlan = false;
2171 	struct ieee80211_sub_if_data *sched_scan_sdata;
2172 	struct cfg80211_sched_scan_request *sched_scan_req;
2173 	bool sched_scan_stopped = false;
2174 	bool suspended = local->suspended;
2175 
2176 	/* nothing to do if HW shouldn't run */
2177 	if (!local->open_count)
2178 		goto wake_up;
2179 
2180 #ifdef CONFIG_PM
2181 	if (suspended)
2182 		local->resuming = true;
2183 
2184 	if (local->wowlan) {
2185 		/*
2186 		 * In the wowlan case, both mac80211 and the device
2187 		 * are functional when the resume op is called, so
2188 		 * clear local->suspended so the device could operate
2189 		 * normally (e.g. pass rx frames).
2190 		 */
2191 		local->suspended = false;
2192 		res = drv_resume(local);
2193 		local->wowlan = false;
2194 		if (res < 0) {
2195 			local->resuming = false;
2196 			return res;
2197 		}
2198 		if (res == 0)
2199 			goto wake_up;
2200 		WARN_ON(res > 1);
2201 		/*
2202 		 * res is 1, which means the driver requested
2203 		 * to go through a regular reset on wakeup.
2204 		 * restore local->suspended in this case.
2205 		 */
2206 		reconfig_due_to_wowlan = true;
2207 		local->suspended = true;
2208 	}
2209 #endif
2210 
2211 	/*
2212 	 * In case of hw_restart during suspend (without wowlan),
2213 	 * cancel restart work, as we are reconfiguring the device
2214 	 * anyway.
2215 	 * Note that restart_work is scheduled on a frozen workqueue,
2216 	 * so we can't deadlock in this case.
2217 	 */
2218 	if (suspended && local->in_reconfig && !reconfig_due_to_wowlan)
2219 		cancel_work_sync(&local->restart_work);
2220 
2221 	local->started = false;
2222 
2223 	/*
2224 	 * Upon resume hardware can sometimes be goofy due to
2225 	 * various platform / driver / bus issues, so restarting
2226 	 * the device may at times not work immediately. Propagate
2227 	 * the error.
2228 	 */
2229 	res = drv_start(local);
2230 	if (res) {
2231 		if (suspended)
2232 			WARN(1, "Hardware became unavailable upon resume. This could be a software issue prior to suspend or a hardware issue.\n");
2233 		else
2234 			WARN(1, "Hardware became unavailable during restart.\n");
2235 		ieee80211_handle_reconfig_failure(local);
2236 		return res;
2237 	}
2238 
2239 	/* setup fragmentation threshold */
2240 	drv_set_frag_threshold(local, hw->wiphy->frag_threshold);
2241 
2242 	/* setup RTS threshold */
2243 	drv_set_rts_threshold(local, hw->wiphy->rts_threshold);
2244 
2245 	/* reset coverage class */
2246 	drv_set_coverage_class(local, hw->wiphy->coverage_class);
2247 
2248 	ieee80211_led_radio(local, true);
2249 	ieee80211_mod_tpt_led_trig(local,
2250 				   IEEE80211_TPT_LEDTRIG_FL_RADIO, 0);
2251 
2252 	/* add interfaces */
2253 	sdata = rtnl_dereference(local->monitor_sdata);
2254 	if (sdata) {
2255 		/* in HW restart it exists already */
2256 		WARN_ON(local->resuming);
2257 		res = drv_add_interface(local, sdata);
2258 		if (WARN_ON(res)) {
2259 			RCU_INIT_POINTER(local->monitor_sdata, NULL);
2260 			synchronize_net();
2261 			kfree(sdata);
2262 		}
2263 	}
2264 
2265 	list_for_each_entry(sdata, &local->interfaces, list) {
2266 		if (sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
2267 		    sdata->vif.type != NL80211_IFTYPE_MONITOR &&
2268 		    ieee80211_sdata_running(sdata)) {
2269 			res = drv_add_interface(local, sdata);
2270 			if (WARN_ON(res))
2271 				break;
2272 		}
2273 	}
2274 
2275 	/* If adding any of the interfaces failed above, roll back and
2276 	 * report failure.
2277 	 */
2278 	if (res) {
2279 		list_for_each_entry_continue_reverse(sdata, &local->interfaces,
2280 						     list)
2281 			if (sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
2282 			    sdata->vif.type != NL80211_IFTYPE_MONITOR &&
2283 			    ieee80211_sdata_running(sdata))
2284 				drv_remove_interface(local, sdata);
2285 		ieee80211_handle_reconfig_failure(local);
2286 		return res;
2287 	}
2288 
2289 	/* add channel contexts */
2290 	if (local->use_chanctx) {
2291 		mutex_lock(&local->chanctx_mtx);
2292 		list_for_each_entry(ctx, &local->chanctx_list, list)
2293 			if (ctx->replace_state !=
2294 			    IEEE80211_CHANCTX_REPLACES_OTHER)
2295 				WARN_ON(drv_add_chanctx(local, ctx));
2296 		mutex_unlock(&local->chanctx_mtx);
2297 
2298 		sdata = rtnl_dereference(local->monitor_sdata);
2299 		if (sdata && ieee80211_sdata_running(sdata))
2300 			ieee80211_assign_chanctx(local, sdata);
2301 	}
2302 
2303 	/* reconfigure hardware */
2304 	ieee80211_hw_config(local, ~0);
2305 
2306 	ieee80211_configure_filter(local);
2307 
2308 	/* Finally also reconfigure all the BSS information */
2309 	list_for_each_entry(sdata, &local->interfaces, list) {
2310 		u32 changed;
2311 
2312 		if (!ieee80211_sdata_running(sdata))
2313 			continue;
2314 
2315 		ieee80211_assign_chanctx(local, sdata);
2316 
2317 		switch (sdata->vif.type) {
2318 		case NL80211_IFTYPE_AP_VLAN:
2319 		case NL80211_IFTYPE_MONITOR:
2320 			break;
2321 		case NL80211_IFTYPE_ADHOC:
2322 			if (sdata->vif.bss_conf.ibss_joined)
2323 				WARN_ON(drv_join_ibss(local, sdata));
2324 			/* fall through */
2325 		default:
2326 			ieee80211_reconfig_stations(sdata);
2327 			/* fall through */
2328 		case NL80211_IFTYPE_AP: /* AP stations are handled later */
2329 			for (i = 0; i < IEEE80211_NUM_ACS; i++)
2330 				drv_conf_tx(local, sdata, i,
2331 					    &sdata->tx_conf[i]);
2332 			break;
2333 		}
2334 
2335 		/* common change flags for all interface types */
2336 		changed = BSS_CHANGED_ERP_CTS_PROT |
2337 			  BSS_CHANGED_ERP_PREAMBLE |
2338 			  BSS_CHANGED_ERP_SLOT |
2339 			  BSS_CHANGED_HT |
2340 			  BSS_CHANGED_BASIC_RATES |
2341 			  BSS_CHANGED_BEACON_INT |
2342 			  BSS_CHANGED_BSSID |
2343 			  BSS_CHANGED_CQM |
2344 			  BSS_CHANGED_QOS |
2345 			  BSS_CHANGED_IDLE |
2346 			  BSS_CHANGED_TXPOWER |
2347 			  BSS_CHANGED_MCAST_RATE;
2348 
2349 		if (sdata->vif.mu_mimo_owner)
2350 			changed |= BSS_CHANGED_MU_GROUPS;
2351 
2352 		switch (sdata->vif.type) {
2353 		case NL80211_IFTYPE_STATION:
2354 			changed |= BSS_CHANGED_ASSOC |
2355 				   BSS_CHANGED_ARP_FILTER |
2356 				   BSS_CHANGED_PS;
2357 
2358 			/* Re-send beacon info report to the driver */
2359 			if (sdata->u.mgd.have_beacon)
2360 				changed |= BSS_CHANGED_BEACON_INFO;
2361 
2362 			if (sdata->vif.bss_conf.max_idle_period ||
2363 			    sdata->vif.bss_conf.protected_keep_alive)
2364 				changed |= BSS_CHANGED_KEEP_ALIVE;
2365 
2366 			sdata_lock(sdata);
2367 			ieee80211_bss_info_change_notify(sdata, changed);
2368 			sdata_unlock(sdata);
2369 			break;
2370 		case NL80211_IFTYPE_OCB:
2371 			changed |= BSS_CHANGED_OCB;
2372 			ieee80211_bss_info_change_notify(sdata, changed);
2373 			break;
2374 		case NL80211_IFTYPE_ADHOC:
2375 			changed |= BSS_CHANGED_IBSS;
2376 			/* fall through */
2377 		case NL80211_IFTYPE_AP:
2378 			changed |= BSS_CHANGED_SSID | BSS_CHANGED_P2P_PS;
2379 
2380 			if (sdata->vif.bss_conf.ftm_responder == 1 &&
2381 			    wiphy_ext_feature_isset(sdata->local->hw.wiphy,
2382 					NL80211_EXT_FEATURE_ENABLE_FTM_RESPONDER))
2383 				changed |= BSS_CHANGED_FTM_RESPONDER;
2384 
2385 			if (sdata->vif.type == NL80211_IFTYPE_AP) {
2386 				changed |= BSS_CHANGED_AP_PROBE_RESP;
2387 
2388 				if (rcu_access_pointer(sdata->u.ap.beacon))
2389 					drv_start_ap(local, sdata);
2390 			}
2391 
2392 			/* fall through */
2393 		case NL80211_IFTYPE_MESH_POINT:
2394 			if (sdata->vif.bss_conf.enable_beacon) {
2395 				changed |= BSS_CHANGED_BEACON |
2396 					   BSS_CHANGED_BEACON_ENABLED;
2397 				ieee80211_bss_info_change_notify(sdata, changed);
2398 			}
2399 			break;
2400 		case NL80211_IFTYPE_NAN:
2401 			res = ieee80211_reconfig_nan(sdata);
2402 			if (res < 0) {
2403 				ieee80211_handle_reconfig_failure(local);
2404 				return res;
2405 			}
2406 			break;
2407 		case NL80211_IFTYPE_WDS:
2408 		case NL80211_IFTYPE_AP_VLAN:
2409 		case NL80211_IFTYPE_MONITOR:
2410 		case NL80211_IFTYPE_P2P_DEVICE:
2411 			/* nothing to do */
2412 			break;
2413 		case NL80211_IFTYPE_UNSPECIFIED:
2414 		case NUM_NL80211_IFTYPES:
2415 		case NL80211_IFTYPE_P2P_CLIENT:
2416 		case NL80211_IFTYPE_P2P_GO:
2417 			WARN_ON(1);
2418 			break;
2419 		}
2420 	}
2421 
2422 	ieee80211_recalc_ps(local);
2423 
2424 	/*
2425 	 * The sta might be in psm against the ap (e.g. because
2426 	 * this was the state before a hw restart), so we
2427 	 * explicitly send a null packet in order to make sure
2428 	 * it'll sync against the ap (and get out of psm).
2429 	 */
2430 	if (!(local->hw.conf.flags & IEEE80211_CONF_PS)) {
2431 		list_for_each_entry(sdata, &local->interfaces, list) {
2432 			if (sdata->vif.type != NL80211_IFTYPE_STATION)
2433 				continue;
2434 			if (!sdata->u.mgd.associated)
2435 				continue;
2436 
2437 			ieee80211_send_nullfunc(local, sdata, false);
2438 		}
2439 	}
2440 
2441 	/* APs are now beaconing, add back stations */
2442 	mutex_lock(&local->sta_mtx);
2443 	list_for_each_entry(sta, &local->sta_list, list) {
2444 		enum ieee80211_sta_state state;
2445 
2446 		if (!sta->uploaded)
2447 			continue;
2448 
2449 		if (sta->sdata->vif.type != NL80211_IFTYPE_AP &&
2450 		    sta->sdata->vif.type != NL80211_IFTYPE_AP_VLAN)
2451 			continue;
2452 
2453 		for (state = IEEE80211_STA_NOTEXIST;
2454 		     state < sta->sta_state; state++)
2455 			WARN_ON(drv_sta_state(local, sta->sdata, sta, state,
2456 					      state + 1));
2457 	}
2458 	mutex_unlock(&local->sta_mtx);
2459 
2460 	/* add back keys */
2461 	list_for_each_entry(sdata, &local->interfaces, list)
2462 		ieee80211_reenable_keys(sdata);
2463 
2464 	/* Reconfigure sched scan if it was interrupted by FW restart */
2465 	mutex_lock(&local->mtx);
2466 	sched_scan_sdata = rcu_dereference_protected(local->sched_scan_sdata,
2467 						lockdep_is_held(&local->mtx));
2468 	sched_scan_req = rcu_dereference_protected(local->sched_scan_req,
2469 						lockdep_is_held(&local->mtx));
2470 	if (sched_scan_sdata && sched_scan_req)
2471 		/*
2472 		 * Sched scan stopped, but we don't want to report it. Instead,
2473 		 * we're trying to reschedule. However, if more than one scan
2474 		 * plan was set, we cannot reschedule since we don't know which
2475 		 * scan plan was currently running (and some scan plans may have
2476 		 * already finished).
2477 		 */
2478 		if (sched_scan_req->n_scan_plans > 1 ||
2479 		    __ieee80211_request_sched_scan_start(sched_scan_sdata,
2480 							 sched_scan_req)) {
2481 			RCU_INIT_POINTER(local->sched_scan_sdata, NULL);
2482 			RCU_INIT_POINTER(local->sched_scan_req, NULL);
2483 			sched_scan_stopped = true;
2484 		}
2485 	mutex_unlock(&local->mtx);
2486 
2487 	if (sched_scan_stopped)
2488 		cfg80211_sched_scan_stopped_rtnl(local->hw.wiphy, 0);
2489 
2490  wake_up:
2491 
2492 	if (local->monitors == local->open_count && local->monitors > 0)
2493 		ieee80211_add_virtual_monitor(local);
2494 
2495 	/*
2496 	 * Clear the WLAN_STA_BLOCK_BA flag so new aggregation
2497 	 * sessions can be established after a resume.
2498 	 *
2499 	 * Also tear down aggregation sessions since reconfiguring
2500 	 * them in a hardware restart scenario is not easily done
2501 	 * right now, and the hardware will have lost information
2502 	 * about the sessions, but we and the AP still think they
2503 	 * are active. This is really a workaround though.
2504 	 */
2505 	if (ieee80211_hw_check(hw, AMPDU_AGGREGATION)) {
2506 		mutex_lock(&local->sta_mtx);
2507 
2508 		list_for_each_entry(sta, &local->sta_list, list) {
2509 			if (!local->resuming)
2510 				ieee80211_sta_tear_down_BA_sessions(
2511 						sta, AGG_STOP_LOCAL_REQUEST);
2512 			clear_sta_flag(sta, WLAN_STA_BLOCK_BA);
2513 		}
2514 
2515 		mutex_unlock(&local->sta_mtx);
2516 	}
2517 
2518 	if (local->in_reconfig) {
2519 		local->in_reconfig = false;
2520 		barrier();
2521 
2522 		/* Restart deferred ROCs */
2523 		mutex_lock(&local->mtx);
2524 		ieee80211_start_next_roc(local);
2525 		mutex_unlock(&local->mtx);
2526 
2527 		/* Requeue all works */
2528 		list_for_each_entry(sdata, &local->interfaces, list)
2529 			ieee80211_queue_work(&local->hw, &sdata->work);
2530 	}
2531 
2532 	ieee80211_wake_queues_by_reason(hw, IEEE80211_MAX_QUEUE_MAP,
2533 					IEEE80211_QUEUE_STOP_REASON_SUSPEND,
2534 					false);
2535 
2536 	/*
2537 	 * If this is for hw restart things are still running.
2538 	 * We may want to change that later, however.
2539 	 */
2540 	if (local->open_count && (!suspended || reconfig_due_to_wowlan))
2541 		drv_reconfig_complete(local, IEEE80211_RECONFIG_TYPE_RESTART);
2542 
2543 	if (!suspended)
2544 		return 0;
2545 
2546 #ifdef CONFIG_PM
2547 	/* first set suspended false, then resuming */
2548 	local->suspended = false;
2549 	mb();
2550 	local->resuming = false;
2551 
2552 	ieee80211_flush_completed_scan(local, false);
2553 
2554 	if (local->open_count && !reconfig_due_to_wowlan)
2555 		drv_reconfig_complete(local, IEEE80211_RECONFIG_TYPE_SUSPEND);
2556 
2557 	list_for_each_entry(sdata, &local->interfaces, list) {
2558 		if (!ieee80211_sdata_running(sdata))
2559 			continue;
2560 		if (sdata->vif.type == NL80211_IFTYPE_STATION)
2561 			ieee80211_sta_restart(sdata);
2562 	}
2563 
2564 	mod_timer(&local->sta_cleanup, jiffies + 1);
2565 #else
2566 	WARN_ON(1);
2567 #endif
2568 
2569 	return 0;
2570 }
2571 
2572 void ieee80211_resume_disconnect(struct ieee80211_vif *vif)
2573 {
2574 	struct ieee80211_sub_if_data *sdata;
2575 	struct ieee80211_local *local;
2576 	struct ieee80211_key *key;
2577 
2578 	if (WARN_ON(!vif))
2579 		return;
2580 
2581 	sdata = vif_to_sdata(vif);
2582 	local = sdata->local;
2583 
2584 	if (WARN_ON(!local->resuming))
2585 		return;
2586 
2587 	if (WARN_ON(vif->type != NL80211_IFTYPE_STATION))
2588 		return;
2589 
2590 	sdata->flags |= IEEE80211_SDATA_DISCONNECT_RESUME;
2591 
2592 	mutex_lock(&local->key_mtx);
2593 	list_for_each_entry(key, &sdata->key_list, list)
2594 		key->flags |= KEY_FLAG_TAINTED;
2595 	mutex_unlock(&local->key_mtx);
2596 }
2597 EXPORT_SYMBOL_GPL(ieee80211_resume_disconnect);
2598 
2599 void ieee80211_recalc_smps(struct ieee80211_sub_if_data *sdata)
2600 {
2601 	struct ieee80211_local *local = sdata->local;
2602 	struct ieee80211_chanctx_conf *chanctx_conf;
2603 	struct ieee80211_chanctx *chanctx;
2604 
2605 	mutex_lock(&local->chanctx_mtx);
2606 
2607 	chanctx_conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
2608 					lockdep_is_held(&local->chanctx_mtx));
2609 
2610 	/*
2611 	 * This function can be called from a work, thus it may be possible
2612 	 * that the chanctx_conf is removed (due to a disconnection, for
2613 	 * example).
2614 	 * So nothing should be done in such case.
2615 	 */
2616 	if (!chanctx_conf)
2617 		goto unlock;
2618 
2619 	chanctx = container_of(chanctx_conf, struct ieee80211_chanctx, conf);
2620 	ieee80211_recalc_smps_chanctx(local, chanctx);
2621  unlock:
2622 	mutex_unlock(&local->chanctx_mtx);
2623 }
2624 
2625 void ieee80211_recalc_min_chandef(struct ieee80211_sub_if_data *sdata)
2626 {
2627 	struct ieee80211_local *local = sdata->local;
2628 	struct ieee80211_chanctx_conf *chanctx_conf;
2629 	struct ieee80211_chanctx *chanctx;
2630 
2631 	mutex_lock(&local->chanctx_mtx);
2632 
2633 	chanctx_conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
2634 					lockdep_is_held(&local->chanctx_mtx));
2635 
2636 	if (WARN_ON_ONCE(!chanctx_conf))
2637 		goto unlock;
2638 
2639 	chanctx = container_of(chanctx_conf, struct ieee80211_chanctx, conf);
2640 	ieee80211_recalc_chanctx_min_def(local, chanctx);
2641  unlock:
2642 	mutex_unlock(&local->chanctx_mtx);
2643 }
2644 
2645 size_t ieee80211_ie_split_vendor(const u8 *ies, size_t ielen, size_t offset)
2646 {
2647 	size_t pos = offset;
2648 
2649 	while (pos < ielen && ies[pos] != WLAN_EID_VENDOR_SPECIFIC)
2650 		pos += 2 + ies[pos + 1];
2651 
2652 	return pos;
2653 }
2654 
2655 static void _ieee80211_enable_rssi_reports(struct ieee80211_sub_if_data *sdata,
2656 					    int rssi_min_thold,
2657 					    int rssi_max_thold)
2658 {
2659 	trace_api_enable_rssi_reports(sdata, rssi_min_thold, rssi_max_thold);
2660 
2661 	if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
2662 		return;
2663 
2664 	/*
2665 	 * Scale up threshold values before storing it, as the RSSI averaging
2666 	 * algorithm uses a scaled up value as well. Change this scaling
2667 	 * factor if the RSSI averaging algorithm changes.
2668 	 */
2669 	sdata->u.mgd.rssi_min_thold = rssi_min_thold*16;
2670 	sdata->u.mgd.rssi_max_thold = rssi_max_thold*16;
2671 }
2672 
2673 void ieee80211_enable_rssi_reports(struct ieee80211_vif *vif,
2674 				    int rssi_min_thold,
2675 				    int rssi_max_thold)
2676 {
2677 	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2678 
2679 	WARN_ON(rssi_min_thold == rssi_max_thold ||
2680 		rssi_min_thold > rssi_max_thold);
2681 
2682 	_ieee80211_enable_rssi_reports(sdata, rssi_min_thold,
2683 				       rssi_max_thold);
2684 }
2685 EXPORT_SYMBOL(ieee80211_enable_rssi_reports);
2686 
2687 void ieee80211_disable_rssi_reports(struct ieee80211_vif *vif)
2688 {
2689 	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2690 
2691 	_ieee80211_enable_rssi_reports(sdata, 0, 0);
2692 }
2693 EXPORT_SYMBOL(ieee80211_disable_rssi_reports);
2694 
2695 u8 *ieee80211_ie_build_ht_cap(u8 *pos, struct ieee80211_sta_ht_cap *ht_cap,
2696 			      u16 cap)
2697 {
2698 	__le16 tmp;
2699 
2700 	*pos++ = WLAN_EID_HT_CAPABILITY;
2701 	*pos++ = sizeof(struct ieee80211_ht_cap);
2702 	memset(pos, 0, sizeof(struct ieee80211_ht_cap));
2703 
2704 	/* capability flags */
2705 	tmp = cpu_to_le16(cap);
2706 	memcpy(pos, &tmp, sizeof(u16));
2707 	pos += sizeof(u16);
2708 
2709 	/* AMPDU parameters */
2710 	*pos++ = ht_cap->ampdu_factor |
2711 		 (ht_cap->ampdu_density <<
2712 			IEEE80211_HT_AMPDU_PARM_DENSITY_SHIFT);
2713 
2714 	/* MCS set */
2715 	memcpy(pos, &ht_cap->mcs, sizeof(ht_cap->mcs));
2716 	pos += sizeof(ht_cap->mcs);
2717 
2718 	/* extended capabilities */
2719 	pos += sizeof(__le16);
2720 
2721 	/* BF capabilities */
2722 	pos += sizeof(__le32);
2723 
2724 	/* antenna selection */
2725 	pos += sizeof(u8);
2726 
2727 	return pos;
2728 }
2729 
2730 u8 *ieee80211_ie_build_vht_cap(u8 *pos, struct ieee80211_sta_vht_cap *vht_cap,
2731 			       u32 cap)
2732 {
2733 	__le32 tmp;
2734 
2735 	*pos++ = WLAN_EID_VHT_CAPABILITY;
2736 	*pos++ = sizeof(struct ieee80211_vht_cap);
2737 	memset(pos, 0, sizeof(struct ieee80211_vht_cap));
2738 
2739 	/* capability flags */
2740 	tmp = cpu_to_le32(cap);
2741 	memcpy(pos, &tmp, sizeof(u32));
2742 	pos += sizeof(u32);
2743 
2744 	/* VHT MCS set */
2745 	memcpy(pos, &vht_cap->vht_mcs, sizeof(vht_cap->vht_mcs));
2746 	pos += sizeof(vht_cap->vht_mcs);
2747 
2748 	return pos;
2749 }
2750 
2751 u8 ieee80211_ie_len_he_cap(struct ieee80211_sub_if_data *sdata, u8 iftype)
2752 {
2753 	const struct ieee80211_sta_he_cap *he_cap;
2754 	struct ieee80211_supported_band *sband;
2755 	u8 n;
2756 
2757 	sband = ieee80211_get_sband(sdata);
2758 	if (!sband)
2759 		return 0;
2760 
2761 	he_cap = ieee80211_get_he_iftype_cap(sband, iftype);
2762 	if (!he_cap)
2763 		return 0;
2764 
2765 	n = ieee80211_he_mcs_nss_size(&he_cap->he_cap_elem);
2766 	return 2 + 1 +
2767 	       sizeof(he_cap->he_cap_elem) + n +
2768 	       ieee80211_he_ppe_size(he_cap->ppe_thres[0],
2769 				     he_cap->he_cap_elem.phy_cap_info);
2770 }
2771 
2772 u8 *ieee80211_ie_build_he_cap(u8 *pos,
2773 			      const struct ieee80211_sta_he_cap *he_cap,
2774 			      u8 *end)
2775 {
2776 	u8 n;
2777 	u8 ie_len;
2778 	u8 *orig_pos = pos;
2779 
2780 	/* Make sure we have place for the IE */
2781 	/*
2782 	 * TODO: the 1 added is because this temporarily is under the EXTENSION
2783 	 * IE. Get rid of it when it moves.
2784 	 */
2785 	if (!he_cap)
2786 		return orig_pos;
2787 
2788 	n = ieee80211_he_mcs_nss_size(&he_cap->he_cap_elem);
2789 	ie_len = 2 + 1 +
2790 		 sizeof(he_cap->he_cap_elem) + n +
2791 		 ieee80211_he_ppe_size(he_cap->ppe_thres[0],
2792 				       he_cap->he_cap_elem.phy_cap_info);
2793 
2794 	if ((end - pos) < ie_len)
2795 		return orig_pos;
2796 
2797 	*pos++ = WLAN_EID_EXTENSION;
2798 	pos++; /* We'll set the size later below */
2799 	*pos++ = WLAN_EID_EXT_HE_CAPABILITY;
2800 
2801 	/* Fixed data */
2802 	memcpy(pos, &he_cap->he_cap_elem, sizeof(he_cap->he_cap_elem));
2803 	pos += sizeof(he_cap->he_cap_elem);
2804 
2805 	memcpy(pos, &he_cap->he_mcs_nss_supp, n);
2806 	pos += n;
2807 
2808 	/* Check if PPE Threshold should be present */
2809 	if ((he_cap->he_cap_elem.phy_cap_info[6] &
2810 	     IEEE80211_HE_PHY_CAP6_PPE_THRESHOLD_PRESENT) == 0)
2811 		goto end;
2812 
2813 	/*
2814 	 * Calculate how many PPET16/PPET8 pairs are to come. Algorithm:
2815 	 * (NSS_M1 + 1) x (num of 1 bits in RU_INDEX_BITMASK)
2816 	 */
2817 	n = hweight8(he_cap->ppe_thres[0] &
2818 		     IEEE80211_PPE_THRES_RU_INDEX_BITMASK_MASK);
2819 	n *= (1 + ((he_cap->ppe_thres[0] & IEEE80211_PPE_THRES_NSS_MASK) >>
2820 		   IEEE80211_PPE_THRES_NSS_POS));
2821 
2822 	/*
2823 	 * Each pair is 6 bits, and we need to add the 7 "header" bits to the
2824 	 * total size.
2825 	 */
2826 	n = (n * IEEE80211_PPE_THRES_INFO_PPET_SIZE * 2) + 7;
2827 	n = DIV_ROUND_UP(n, 8);
2828 
2829 	/* Copy PPE Thresholds */
2830 	memcpy(pos, &he_cap->ppe_thres, n);
2831 	pos += n;
2832 
2833 end:
2834 	orig_pos[1] = (pos - orig_pos) - 2;
2835 	return pos;
2836 }
2837 
2838 u8 *ieee80211_ie_build_ht_oper(u8 *pos, struct ieee80211_sta_ht_cap *ht_cap,
2839 			       const struct cfg80211_chan_def *chandef,
2840 			       u16 prot_mode, bool rifs_mode)
2841 {
2842 	struct ieee80211_ht_operation *ht_oper;
2843 	/* Build HT Information */
2844 	*pos++ = WLAN_EID_HT_OPERATION;
2845 	*pos++ = sizeof(struct ieee80211_ht_operation);
2846 	ht_oper = (struct ieee80211_ht_operation *)pos;
2847 	ht_oper->primary_chan = ieee80211_frequency_to_channel(
2848 					chandef->chan->center_freq);
2849 	switch (chandef->width) {
2850 	case NL80211_CHAN_WIDTH_160:
2851 	case NL80211_CHAN_WIDTH_80P80:
2852 	case NL80211_CHAN_WIDTH_80:
2853 	case NL80211_CHAN_WIDTH_40:
2854 		if (chandef->center_freq1 > chandef->chan->center_freq)
2855 			ht_oper->ht_param = IEEE80211_HT_PARAM_CHA_SEC_ABOVE;
2856 		else
2857 			ht_oper->ht_param = IEEE80211_HT_PARAM_CHA_SEC_BELOW;
2858 		break;
2859 	default:
2860 		ht_oper->ht_param = IEEE80211_HT_PARAM_CHA_SEC_NONE;
2861 		break;
2862 	}
2863 	if (ht_cap->cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 &&
2864 	    chandef->width != NL80211_CHAN_WIDTH_20_NOHT &&
2865 	    chandef->width != NL80211_CHAN_WIDTH_20)
2866 		ht_oper->ht_param |= IEEE80211_HT_PARAM_CHAN_WIDTH_ANY;
2867 
2868 	if (rifs_mode)
2869 		ht_oper->ht_param |= IEEE80211_HT_PARAM_RIFS_MODE;
2870 
2871 	ht_oper->operation_mode = cpu_to_le16(prot_mode);
2872 	ht_oper->stbc_param = 0x0000;
2873 
2874 	/* It seems that Basic MCS set and Supported MCS set
2875 	   are identical for the first 10 bytes */
2876 	memset(&ht_oper->basic_set, 0, 16);
2877 	memcpy(&ht_oper->basic_set, &ht_cap->mcs, 10);
2878 
2879 	return pos + sizeof(struct ieee80211_ht_operation);
2880 }
2881 
2882 void ieee80211_ie_build_wide_bw_cs(u8 *pos,
2883 				   const struct cfg80211_chan_def *chandef)
2884 {
2885 	*pos++ = WLAN_EID_WIDE_BW_CHANNEL_SWITCH;	/* EID */
2886 	*pos++ = 3;					/* IE length */
2887 	/* New channel width */
2888 	switch (chandef->width) {
2889 	case NL80211_CHAN_WIDTH_80:
2890 		*pos++ = IEEE80211_VHT_CHANWIDTH_80MHZ;
2891 		break;
2892 	case NL80211_CHAN_WIDTH_160:
2893 		*pos++ = IEEE80211_VHT_CHANWIDTH_160MHZ;
2894 		break;
2895 	case NL80211_CHAN_WIDTH_80P80:
2896 		*pos++ = IEEE80211_VHT_CHANWIDTH_80P80MHZ;
2897 		break;
2898 	default:
2899 		*pos++ = IEEE80211_VHT_CHANWIDTH_USE_HT;
2900 	}
2901 
2902 	/* new center frequency segment 0 */
2903 	*pos++ = ieee80211_frequency_to_channel(chandef->center_freq1);
2904 	/* new center frequency segment 1 */
2905 	if (chandef->center_freq2)
2906 		*pos++ = ieee80211_frequency_to_channel(chandef->center_freq2);
2907 	else
2908 		*pos++ = 0;
2909 }
2910 
2911 u8 *ieee80211_ie_build_vht_oper(u8 *pos, struct ieee80211_sta_vht_cap *vht_cap,
2912 				const struct cfg80211_chan_def *chandef)
2913 {
2914 	struct ieee80211_vht_operation *vht_oper;
2915 
2916 	*pos++ = WLAN_EID_VHT_OPERATION;
2917 	*pos++ = sizeof(struct ieee80211_vht_operation);
2918 	vht_oper = (struct ieee80211_vht_operation *)pos;
2919 	vht_oper->center_freq_seg0_idx = ieee80211_frequency_to_channel(
2920 							chandef->center_freq1);
2921 	if (chandef->center_freq2)
2922 		vht_oper->center_freq_seg1_idx =
2923 			ieee80211_frequency_to_channel(chandef->center_freq2);
2924 	else
2925 		vht_oper->center_freq_seg1_idx = 0x00;
2926 
2927 	switch (chandef->width) {
2928 	case NL80211_CHAN_WIDTH_160:
2929 		/*
2930 		 * Convert 160 MHz channel width to new style as interop
2931 		 * workaround.
2932 		 */
2933 		vht_oper->chan_width = IEEE80211_VHT_CHANWIDTH_80MHZ;
2934 		vht_oper->center_freq_seg1_idx = vht_oper->center_freq_seg0_idx;
2935 		if (chandef->chan->center_freq < chandef->center_freq1)
2936 			vht_oper->center_freq_seg0_idx -= 8;
2937 		else
2938 			vht_oper->center_freq_seg0_idx += 8;
2939 		break;
2940 	case NL80211_CHAN_WIDTH_80P80:
2941 		/*
2942 		 * Convert 80+80 MHz channel width to new style as interop
2943 		 * workaround.
2944 		 */
2945 		vht_oper->chan_width = IEEE80211_VHT_CHANWIDTH_80MHZ;
2946 		break;
2947 	case NL80211_CHAN_WIDTH_80:
2948 		vht_oper->chan_width = IEEE80211_VHT_CHANWIDTH_80MHZ;
2949 		break;
2950 	default:
2951 		vht_oper->chan_width = IEEE80211_VHT_CHANWIDTH_USE_HT;
2952 		break;
2953 	}
2954 
2955 	/* don't require special VHT peer rates */
2956 	vht_oper->basic_mcs_set = cpu_to_le16(0xffff);
2957 
2958 	return pos + sizeof(struct ieee80211_vht_operation);
2959 }
2960 
2961 u8 *ieee80211_ie_build_he_oper(u8 *pos)
2962 {
2963 	struct ieee80211_he_operation *he_oper;
2964 	u32 he_oper_params;
2965 
2966 	*pos++ = WLAN_EID_EXTENSION;
2967 	*pos++ = 1 + sizeof(struct ieee80211_he_operation);
2968 	*pos++ = WLAN_EID_EXT_HE_OPERATION;
2969 
2970 	he_oper_params = 0;
2971 	he_oper_params |= u32_encode_bits(1023, /* disabled */
2972 				IEEE80211_HE_OPERATION_RTS_THRESHOLD_MASK);
2973 	he_oper_params |= u32_encode_bits(1,
2974 				IEEE80211_HE_OPERATION_ER_SU_DISABLE);
2975 	he_oper_params |= u32_encode_bits(1,
2976 				IEEE80211_HE_OPERATION_BSS_COLOR_DISABLED);
2977 
2978 	he_oper = (struct ieee80211_he_operation *)pos;
2979 	he_oper->he_oper_params = cpu_to_le32(he_oper_params);
2980 
2981 	/* don't require special HE peer rates */
2982 	he_oper->he_mcs_nss_set = cpu_to_le16(0xffff);
2983 
2984 	/* TODO add VHT operational and 6GHz operational subelement? */
2985 
2986 	return pos + sizeof(struct ieee80211_vht_operation);
2987 }
2988 
2989 bool ieee80211_chandef_ht_oper(const struct ieee80211_ht_operation *ht_oper,
2990 			       struct cfg80211_chan_def *chandef)
2991 {
2992 	enum nl80211_channel_type channel_type;
2993 
2994 	if (!ht_oper)
2995 		return false;
2996 
2997 	switch (ht_oper->ht_param & IEEE80211_HT_PARAM_CHA_SEC_OFFSET) {
2998 	case IEEE80211_HT_PARAM_CHA_SEC_NONE:
2999 		channel_type = NL80211_CHAN_HT20;
3000 		break;
3001 	case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
3002 		channel_type = NL80211_CHAN_HT40PLUS;
3003 		break;
3004 	case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
3005 		channel_type = NL80211_CHAN_HT40MINUS;
3006 		break;
3007 	default:
3008 		channel_type = NL80211_CHAN_NO_HT;
3009 		return false;
3010 	}
3011 
3012 	cfg80211_chandef_create(chandef, chandef->chan, channel_type);
3013 	return true;
3014 }
3015 
3016 bool ieee80211_chandef_vht_oper(struct ieee80211_hw *hw,
3017 				const struct ieee80211_vht_operation *oper,
3018 				const struct ieee80211_ht_operation *htop,
3019 				struct cfg80211_chan_def *chandef)
3020 {
3021 	struct cfg80211_chan_def new = *chandef;
3022 	int cf0, cf1;
3023 	int ccfs0, ccfs1, ccfs2;
3024 	int ccf0, ccf1;
3025 	u32 vht_cap;
3026 	bool support_80_80 = false;
3027 	bool support_160 = false;
3028 
3029 	if (!oper || !htop)
3030 		return false;
3031 
3032 	vht_cap = hw->wiphy->bands[chandef->chan->band]->vht_cap.cap;
3033 	support_160 = (vht_cap & (IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK |
3034 				  IEEE80211_VHT_CAP_EXT_NSS_BW_MASK));
3035 	support_80_80 = ((vht_cap &
3036 			 IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ) ||
3037 			(vht_cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ &&
3038 			 vht_cap & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK) ||
3039 			((vht_cap & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK) >>
3040 				    IEEE80211_VHT_CAP_EXT_NSS_BW_SHIFT > 1));
3041 	ccfs0 = oper->center_freq_seg0_idx;
3042 	ccfs1 = oper->center_freq_seg1_idx;
3043 	ccfs2 = (le16_to_cpu(htop->operation_mode) &
3044 				IEEE80211_HT_OP_MODE_CCFS2_MASK)
3045 			>> IEEE80211_HT_OP_MODE_CCFS2_SHIFT;
3046 
3047 	/* when parsing (and we know how to) CCFS1 and CCFS2 are equivalent */
3048 	ccf0 = ccfs0;
3049 	ccf1 = ccfs1;
3050 	if (!ccfs1 && ieee80211_hw_check(hw, SUPPORTS_VHT_EXT_NSS_BW))
3051 		ccf1 = ccfs2;
3052 
3053 	cf0 = ieee80211_channel_to_frequency(ccf0, chandef->chan->band);
3054 	cf1 = ieee80211_channel_to_frequency(ccf1, chandef->chan->band);
3055 
3056 	switch (oper->chan_width) {
3057 	case IEEE80211_VHT_CHANWIDTH_USE_HT:
3058 		/* just use HT information directly */
3059 		break;
3060 	case IEEE80211_VHT_CHANWIDTH_80MHZ:
3061 		new.width = NL80211_CHAN_WIDTH_80;
3062 		new.center_freq1 = cf0;
3063 		/* If needed, adjust based on the newer interop workaround. */
3064 		if (ccf1) {
3065 			unsigned int diff;
3066 
3067 			diff = abs(ccf1 - ccf0);
3068 			if ((diff == 8) && support_160) {
3069 				new.width = NL80211_CHAN_WIDTH_160;
3070 				new.center_freq1 = cf1;
3071 			} else if ((diff > 8) && support_80_80) {
3072 				new.width = NL80211_CHAN_WIDTH_80P80;
3073 				new.center_freq2 = cf1;
3074 			}
3075 		}
3076 		break;
3077 	case IEEE80211_VHT_CHANWIDTH_160MHZ:
3078 		/* deprecated encoding */
3079 		new.width = NL80211_CHAN_WIDTH_160;
3080 		new.center_freq1 = cf0;
3081 		break;
3082 	case IEEE80211_VHT_CHANWIDTH_80P80MHZ:
3083 		/* deprecated encoding */
3084 		new.width = NL80211_CHAN_WIDTH_80P80;
3085 		new.center_freq1 = cf0;
3086 		new.center_freq2 = cf1;
3087 		break;
3088 	default:
3089 		return false;
3090 	}
3091 
3092 	if (!cfg80211_chandef_valid(&new))
3093 		return false;
3094 
3095 	*chandef = new;
3096 	return true;
3097 }
3098 
3099 int ieee80211_parse_bitrates(struct cfg80211_chan_def *chandef,
3100 			     const struct ieee80211_supported_band *sband,
3101 			     const u8 *srates, int srates_len, u32 *rates)
3102 {
3103 	u32 rate_flags = ieee80211_chandef_rate_flags(chandef);
3104 	int shift = ieee80211_chandef_get_shift(chandef);
3105 	struct ieee80211_rate *br;
3106 	int brate, rate, i, j, count = 0;
3107 
3108 	*rates = 0;
3109 
3110 	for (i = 0; i < srates_len; i++) {
3111 		rate = srates[i] & 0x7f;
3112 
3113 		for (j = 0; j < sband->n_bitrates; j++) {
3114 			br = &sband->bitrates[j];
3115 			if ((rate_flags & br->flags) != rate_flags)
3116 				continue;
3117 
3118 			brate = DIV_ROUND_UP(br->bitrate, (1 << shift) * 5);
3119 			if (brate == rate) {
3120 				*rates |= BIT(j);
3121 				count++;
3122 				break;
3123 			}
3124 		}
3125 	}
3126 	return count;
3127 }
3128 
3129 int ieee80211_add_srates_ie(struct ieee80211_sub_if_data *sdata,
3130 			    struct sk_buff *skb, bool need_basic,
3131 			    enum nl80211_band band)
3132 {
3133 	struct ieee80211_local *local = sdata->local;
3134 	struct ieee80211_supported_band *sband;
3135 	int rate, shift;
3136 	u8 i, rates, *pos;
3137 	u32 basic_rates = sdata->vif.bss_conf.basic_rates;
3138 	u32 rate_flags;
3139 
3140 	shift = ieee80211_vif_get_shift(&sdata->vif);
3141 	rate_flags = ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef);
3142 	sband = local->hw.wiphy->bands[band];
3143 	rates = 0;
3144 	for (i = 0; i < sband->n_bitrates; i++) {
3145 		if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
3146 			continue;
3147 		rates++;
3148 	}
3149 	if (rates > 8)
3150 		rates = 8;
3151 
3152 	if (skb_tailroom(skb) < rates + 2)
3153 		return -ENOMEM;
3154 
3155 	pos = skb_put(skb, rates + 2);
3156 	*pos++ = WLAN_EID_SUPP_RATES;
3157 	*pos++ = rates;
3158 	for (i = 0; i < rates; i++) {
3159 		u8 basic = 0;
3160 		if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
3161 			continue;
3162 
3163 		if (need_basic && basic_rates & BIT(i))
3164 			basic = 0x80;
3165 		rate = DIV_ROUND_UP(sband->bitrates[i].bitrate,
3166 				    5 * (1 << shift));
3167 		*pos++ = basic | (u8) rate;
3168 	}
3169 
3170 	return 0;
3171 }
3172 
3173 int ieee80211_add_ext_srates_ie(struct ieee80211_sub_if_data *sdata,
3174 				struct sk_buff *skb, bool need_basic,
3175 				enum nl80211_band band)
3176 {
3177 	struct ieee80211_local *local = sdata->local;
3178 	struct ieee80211_supported_band *sband;
3179 	int rate, shift;
3180 	u8 i, exrates, *pos;
3181 	u32 basic_rates = sdata->vif.bss_conf.basic_rates;
3182 	u32 rate_flags;
3183 
3184 	rate_flags = ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef);
3185 	shift = ieee80211_vif_get_shift(&sdata->vif);
3186 
3187 	sband = local->hw.wiphy->bands[band];
3188 	exrates = 0;
3189 	for (i = 0; i < sband->n_bitrates; i++) {
3190 		if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
3191 			continue;
3192 		exrates++;
3193 	}
3194 
3195 	if (exrates > 8)
3196 		exrates -= 8;
3197 	else
3198 		exrates = 0;
3199 
3200 	if (skb_tailroom(skb) < exrates + 2)
3201 		return -ENOMEM;
3202 
3203 	if (exrates) {
3204 		pos = skb_put(skb, exrates + 2);
3205 		*pos++ = WLAN_EID_EXT_SUPP_RATES;
3206 		*pos++ = exrates;
3207 		for (i = 8; i < sband->n_bitrates; i++) {
3208 			u8 basic = 0;
3209 			if ((rate_flags & sband->bitrates[i].flags)
3210 			    != rate_flags)
3211 				continue;
3212 			if (need_basic && basic_rates & BIT(i))
3213 				basic = 0x80;
3214 			rate = DIV_ROUND_UP(sband->bitrates[i].bitrate,
3215 					    5 * (1 << shift));
3216 			*pos++ = basic | (u8) rate;
3217 		}
3218 	}
3219 	return 0;
3220 }
3221 
3222 int ieee80211_ave_rssi(struct ieee80211_vif *vif)
3223 {
3224 	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
3225 	struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
3226 
3227 	if (WARN_ON_ONCE(sdata->vif.type != NL80211_IFTYPE_STATION)) {
3228 		/* non-managed type inferfaces */
3229 		return 0;
3230 	}
3231 	return -ewma_beacon_signal_read(&ifmgd->ave_beacon_signal);
3232 }
3233 EXPORT_SYMBOL_GPL(ieee80211_ave_rssi);
3234 
3235 u8 ieee80211_mcs_to_chains(const struct ieee80211_mcs_info *mcs)
3236 {
3237 	if (!mcs)
3238 		return 1;
3239 
3240 	/* TODO: consider rx_highest */
3241 
3242 	if (mcs->rx_mask[3])
3243 		return 4;
3244 	if (mcs->rx_mask[2])
3245 		return 3;
3246 	if (mcs->rx_mask[1])
3247 		return 2;
3248 	return 1;
3249 }
3250 
3251 /**
3252  * ieee80211_calculate_rx_timestamp - calculate timestamp in frame
3253  * @local: mac80211 hw info struct
3254  * @status: RX status
3255  * @mpdu_len: total MPDU length (including FCS)
3256  * @mpdu_offset: offset into MPDU to calculate timestamp at
3257  *
3258  * This function calculates the RX timestamp at the given MPDU offset, taking
3259  * into account what the RX timestamp was. An offset of 0 will just normalize
3260  * the timestamp to TSF at beginning of MPDU reception.
3261  */
3262 u64 ieee80211_calculate_rx_timestamp(struct ieee80211_local *local,
3263 				     struct ieee80211_rx_status *status,
3264 				     unsigned int mpdu_len,
3265 				     unsigned int mpdu_offset)
3266 {
3267 	u64 ts = status->mactime;
3268 	struct rate_info ri;
3269 	u16 rate;
3270 
3271 	if (WARN_ON(!ieee80211_have_rx_timestamp(status)))
3272 		return 0;
3273 
3274 	memset(&ri, 0, sizeof(ri));
3275 
3276 	ri.bw = status->bw;
3277 
3278 	/* Fill cfg80211 rate info */
3279 	switch (status->encoding) {
3280 	case RX_ENC_HT:
3281 		ri.mcs = status->rate_idx;
3282 		ri.flags |= RATE_INFO_FLAGS_MCS;
3283 		if (status->enc_flags & RX_ENC_FLAG_SHORT_GI)
3284 			ri.flags |= RATE_INFO_FLAGS_SHORT_GI;
3285 		break;
3286 	case RX_ENC_VHT:
3287 		ri.flags |= RATE_INFO_FLAGS_VHT_MCS;
3288 		ri.mcs = status->rate_idx;
3289 		ri.nss = status->nss;
3290 		if (status->enc_flags & RX_ENC_FLAG_SHORT_GI)
3291 			ri.flags |= RATE_INFO_FLAGS_SHORT_GI;
3292 		break;
3293 	default:
3294 		WARN_ON(1);
3295 		/* fall through */
3296 	case RX_ENC_LEGACY: {
3297 		struct ieee80211_supported_band *sband;
3298 		int shift = 0;
3299 		int bitrate;
3300 
3301 		switch (status->bw) {
3302 		case RATE_INFO_BW_10:
3303 			shift = 1;
3304 			break;
3305 		case RATE_INFO_BW_5:
3306 			shift = 2;
3307 			break;
3308 		}
3309 
3310 		sband = local->hw.wiphy->bands[status->band];
3311 		bitrate = sband->bitrates[status->rate_idx].bitrate;
3312 		ri.legacy = DIV_ROUND_UP(bitrate, (1 << shift));
3313 
3314 		if (status->flag & RX_FLAG_MACTIME_PLCP_START) {
3315 			/* TODO: handle HT/VHT preambles */
3316 			if (status->band == NL80211_BAND_5GHZ) {
3317 				ts += 20 << shift;
3318 				mpdu_offset += 2;
3319 			} else if (status->enc_flags & RX_ENC_FLAG_SHORTPRE) {
3320 				ts += 96;
3321 			} else {
3322 				ts += 192;
3323 			}
3324 		}
3325 		break;
3326 		}
3327 	}
3328 
3329 	rate = cfg80211_calculate_bitrate(&ri);
3330 	if (WARN_ONCE(!rate,
3331 		      "Invalid bitrate: flags=0x%llx, idx=%d, vht_nss=%d\n",
3332 		      (unsigned long long)status->flag, status->rate_idx,
3333 		      status->nss))
3334 		return 0;
3335 
3336 	/* rewind from end of MPDU */
3337 	if (status->flag & RX_FLAG_MACTIME_END)
3338 		ts -= mpdu_len * 8 * 10 / rate;
3339 
3340 	ts += mpdu_offset * 8 * 10 / rate;
3341 
3342 	return ts;
3343 }
3344 
3345 void ieee80211_dfs_cac_cancel(struct ieee80211_local *local)
3346 {
3347 	struct ieee80211_sub_if_data *sdata;
3348 	struct cfg80211_chan_def chandef;
3349 
3350 	/* for interface list, to avoid linking iflist_mtx and chanctx_mtx */
3351 	ASSERT_RTNL();
3352 
3353 	mutex_lock(&local->mtx);
3354 	list_for_each_entry(sdata, &local->interfaces, list) {
3355 		/* it might be waiting for the local->mtx, but then
3356 		 * by the time it gets it, sdata->wdev.cac_started
3357 		 * will no longer be true
3358 		 */
3359 		cancel_delayed_work(&sdata->dfs_cac_timer_work);
3360 
3361 		if (sdata->wdev.cac_started) {
3362 			chandef = sdata->vif.bss_conf.chandef;
3363 			ieee80211_vif_release_channel(sdata);
3364 			cfg80211_cac_event(sdata->dev,
3365 					   &chandef,
3366 					   NL80211_RADAR_CAC_ABORTED,
3367 					   GFP_KERNEL);
3368 		}
3369 	}
3370 	mutex_unlock(&local->mtx);
3371 }
3372 
3373 void ieee80211_dfs_radar_detected_work(struct work_struct *work)
3374 {
3375 	struct ieee80211_local *local =
3376 		container_of(work, struct ieee80211_local, radar_detected_work);
3377 	struct cfg80211_chan_def chandef = local->hw.conf.chandef;
3378 	struct ieee80211_chanctx *ctx;
3379 	int num_chanctx = 0;
3380 
3381 	mutex_lock(&local->chanctx_mtx);
3382 	list_for_each_entry(ctx, &local->chanctx_list, list) {
3383 		if (ctx->replace_state == IEEE80211_CHANCTX_REPLACES_OTHER)
3384 			continue;
3385 
3386 		num_chanctx++;
3387 		chandef = ctx->conf.def;
3388 	}
3389 	mutex_unlock(&local->chanctx_mtx);
3390 
3391 	rtnl_lock();
3392 	ieee80211_dfs_cac_cancel(local);
3393 	rtnl_unlock();
3394 
3395 	if (num_chanctx > 1)
3396 		/* XXX: multi-channel is not supported yet */
3397 		WARN_ON(1);
3398 	else
3399 		cfg80211_radar_event(local->hw.wiphy, &chandef, GFP_KERNEL);
3400 }
3401 
3402 void ieee80211_radar_detected(struct ieee80211_hw *hw)
3403 {
3404 	struct ieee80211_local *local = hw_to_local(hw);
3405 
3406 	trace_api_radar_detected(local);
3407 
3408 	schedule_work(&local->radar_detected_work);
3409 }
3410 EXPORT_SYMBOL(ieee80211_radar_detected);
3411 
3412 u32 ieee80211_chandef_downgrade(struct cfg80211_chan_def *c)
3413 {
3414 	u32 ret;
3415 	int tmp;
3416 
3417 	switch (c->width) {
3418 	case NL80211_CHAN_WIDTH_20:
3419 		c->width = NL80211_CHAN_WIDTH_20_NOHT;
3420 		ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
3421 		break;
3422 	case NL80211_CHAN_WIDTH_40:
3423 		c->width = NL80211_CHAN_WIDTH_20;
3424 		c->center_freq1 = c->chan->center_freq;
3425 		ret = IEEE80211_STA_DISABLE_40MHZ |
3426 		      IEEE80211_STA_DISABLE_VHT;
3427 		break;
3428 	case NL80211_CHAN_WIDTH_80:
3429 		tmp = (30 + c->chan->center_freq - c->center_freq1)/20;
3430 		/* n_P40 */
3431 		tmp /= 2;
3432 		/* freq_P40 */
3433 		c->center_freq1 = c->center_freq1 - 20 + 40 * tmp;
3434 		c->width = NL80211_CHAN_WIDTH_40;
3435 		ret = IEEE80211_STA_DISABLE_VHT;
3436 		break;
3437 	case NL80211_CHAN_WIDTH_80P80:
3438 		c->center_freq2 = 0;
3439 		c->width = NL80211_CHAN_WIDTH_80;
3440 		ret = IEEE80211_STA_DISABLE_80P80MHZ |
3441 		      IEEE80211_STA_DISABLE_160MHZ;
3442 		break;
3443 	case NL80211_CHAN_WIDTH_160:
3444 		/* n_P20 */
3445 		tmp = (70 + c->chan->center_freq - c->center_freq1)/20;
3446 		/* n_P80 */
3447 		tmp /= 4;
3448 		c->center_freq1 = c->center_freq1 - 40 + 80 * tmp;
3449 		c->width = NL80211_CHAN_WIDTH_80;
3450 		ret = IEEE80211_STA_DISABLE_80P80MHZ |
3451 		      IEEE80211_STA_DISABLE_160MHZ;
3452 		break;
3453 	default:
3454 	case NL80211_CHAN_WIDTH_20_NOHT:
3455 		WARN_ON_ONCE(1);
3456 		c->width = NL80211_CHAN_WIDTH_20_NOHT;
3457 		ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
3458 		break;
3459 	case NL80211_CHAN_WIDTH_5:
3460 	case NL80211_CHAN_WIDTH_10:
3461 		WARN_ON_ONCE(1);
3462 		/* keep c->width */
3463 		ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
3464 		break;
3465 	}
3466 
3467 	WARN_ON_ONCE(!cfg80211_chandef_valid(c));
3468 
3469 	return ret;
3470 }
3471 
3472 /*
3473  * Returns true if smps_mode_new is strictly more restrictive than
3474  * smps_mode_old.
3475  */
3476 bool ieee80211_smps_is_restrictive(enum ieee80211_smps_mode smps_mode_old,
3477 				   enum ieee80211_smps_mode smps_mode_new)
3478 {
3479 	if (WARN_ON_ONCE(smps_mode_old == IEEE80211_SMPS_AUTOMATIC ||
3480 			 smps_mode_new == IEEE80211_SMPS_AUTOMATIC))
3481 		return false;
3482 
3483 	switch (smps_mode_old) {
3484 	case IEEE80211_SMPS_STATIC:
3485 		return false;
3486 	case IEEE80211_SMPS_DYNAMIC:
3487 		return smps_mode_new == IEEE80211_SMPS_STATIC;
3488 	case IEEE80211_SMPS_OFF:
3489 		return smps_mode_new != IEEE80211_SMPS_OFF;
3490 	default:
3491 		WARN_ON(1);
3492 	}
3493 
3494 	return false;
3495 }
3496 
3497 int ieee80211_send_action_csa(struct ieee80211_sub_if_data *sdata,
3498 			      struct cfg80211_csa_settings *csa_settings)
3499 {
3500 	struct sk_buff *skb;
3501 	struct ieee80211_mgmt *mgmt;
3502 	struct ieee80211_local *local = sdata->local;
3503 	int freq;
3504 	int hdr_len = offsetofend(struct ieee80211_mgmt,
3505 				  u.action.u.chan_switch);
3506 	u8 *pos;
3507 
3508 	if (sdata->vif.type != NL80211_IFTYPE_ADHOC &&
3509 	    sdata->vif.type != NL80211_IFTYPE_MESH_POINT)
3510 		return -EOPNOTSUPP;
3511 
3512 	skb = dev_alloc_skb(local->tx_headroom + hdr_len +
3513 			    5 + /* channel switch announcement element */
3514 			    3 + /* secondary channel offset element */
3515 			    5 + /* wide bandwidth channel switch announcement */
3516 			    8); /* mesh channel switch parameters element */
3517 	if (!skb)
3518 		return -ENOMEM;
3519 
3520 	skb_reserve(skb, local->tx_headroom);
3521 	mgmt = skb_put_zero(skb, hdr_len);
3522 	mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
3523 					  IEEE80211_STYPE_ACTION);
3524 
3525 	eth_broadcast_addr(mgmt->da);
3526 	memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
3527 	if (ieee80211_vif_is_mesh(&sdata->vif)) {
3528 		memcpy(mgmt->bssid, sdata->vif.addr, ETH_ALEN);
3529 	} else {
3530 		struct ieee80211_if_ibss *ifibss = &sdata->u.ibss;
3531 		memcpy(mgmt->bssid, ifibss->bssid, ETH_ALEN);
3532 	}
3533 	mgmt->u.action.category = WLAN_CATEGORY_SPECTRUM_MGMT;
3534 	mgmt->u.action.u.chan_switch.action_code = WLAN_ACTION_SPCT_CHL_SWITCH;
3535 	pos = skb_put(skb, 5);
3536 	*pos++ = WLAN_EID_CHANNEL_SWITCH;			/* EID */
3537 	*pos++ = 3;						/* IE length */
3538 	*pos++ = csa_settings->block_tx ? 1 : 0;		/* CSA mode */
3539 	freq = csa_settings->chandef.chan->center_freq;
3540 	*pos++ = ieee80211_frequency_to_channel(freq);		/* channel */
3541 	*pos++ = csa_settings->count;				/* count */
3542 
3543 	if (csa_settings->chandef.width == NL80211_CHAN_WIDTH_40) {
3544 		enum nl80211_channel_type ch_type;
3545 
3546 		skb_put(skb, 3);
3547 		*pos++ = WLAN_EID_SECONDARY_CHANNEL_OFFSET;	/* EID */
3548 		*pos++ = 1;					/* IE length */
3549 		ch_type = cfg80211_get_chandef_type(&csa_settings->chandef);
3550 		if (ch_type == NL80211_CHAN_HT40PLUS)
3551 			*pos++ = IEEE80211_HT_PARAM_CHA_SEC_ABOVE;
3552 		else
3553 			*pos++ = IEEE80211_HT_PARAM_CHA_SEC_BELOW;
3554 	}
3555 
3556 	if (ieee80211_vif_is_mesh(&sdata->vif)) {
3557 		struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
3558 
3559 		skb_put(skb, 8);
3560 		*pos++ = WLAN_EID_CHAN_SWITCH_PARAM;		/* EID */
3561 		*pos++ = 6;					/* IE length */
3562 		*pos++ = sdata->u.mesh.mshcfg.dot11MeshTTL;	/* Mesh TTL */
3563 		*pos = 0x00;	/* Mesh Flag: Tx Restrict, Initiator, Reason */
3564 		*pos |= WLAN_EID_CHAN_SWITCH_PARAM_INITIATOR;
3565 		*pos++ |= csa_settings->block_tx ?
3566 			  WLAN_EID_CHAN_SWITCH_PARAM_TX_RESTRICT : 0x00;
3567 		put_unaligned_le16(WLAN_REASON_MESH_CHAN, pos); /* Reason Cd */
3568 		pos += 2;
3569 		put_unaligned_le16(ifmsh->pre_value, pos);/* Precedence Value */
3570 		pos += 2;
3571 	}
3572 
3573 	if (csa_settings->chandef.width == NL80211_CHAN_WIDTH_80 ||
3574 	    csa_settings->chandef.width == NL80211_CHAN_WIDTH_80P80 ||
3575 	    csa_settings->chandef.width == NL80211_CHAN_WIDTH_160) {
3576 		skb_put(skb, 5);
3577 		ieee80211_ie_build_wide_bw_cs(pos, &csa_settings->chandef);
3578 	}
3579 
3580 	ieee80211_tx_skb(sdata, skb);
3581 	return 0;
3582 }
3583 
3584 bool ieee80211_cs_valid(const struct ieee80211_cipher_scheme *cs)
3585 {
3586 	return !(cs == NULL || cs->cipher == 0 ||
3587 		 cs->hdr_len < cs->pn_len + cs->pn_off ||
3588 		 cs->hdr_len <= cs->key_idx_off ||
3589 		 cs->key_idx_shift > 7 ||
3590 		 cs->key_idx_mask == 0);
3591 }
3592 
3593 bool ieee80211_cs_list_valid(const struct ieee80211_cipher_scheme *cs, int n)
3594 {
3595 	int i;
3596 
3597 	/* Ensure we have enough iftype bitmap space for all iftype values */
3598 	WARN_ON((NUM_NL80211_IFTYPES / 8 + 1) > sizeof(cs[0].iftype));
3599 
3600 	for (i = 0; i < n; i++)
3601 		if (!ieee80211_cs_valid(&cs[i]))
3602 			return false;
3603 
3604 	return true;
3605 }
3606 
3607 const struct ieee80211_cipher_scheme *
3608 ieee80211_cs_get(struct ieee80211_local *local, u32 cipher,
3609 		 enum nl80211_iftype iftype)
3610 {
3611 	const struct ieee80211_cipher_scheme *l = local->hw.cipher_schemes;
3612 	int n = local->hw.n_cipher_schemes;
3613 	int i;
3614 	const struct ieee80211_cipher_scheme *cs = NULL;
3615 
3616 	for (i = 0; i < n; i++) {
3617 		if (l[i].cipher == cipher) {
3618 			cs = &l[i];
3619 			break;
3620 		}
3621 	}
3622 
3623 	if (!cs || !(cs->iftype & BIT(iftype)))
3624 		return NULL;
3625 
3626 	return cs;
3627 }
3628 
3629 int ieee80211_cs_headroom(struct ieee80211_local *local,
3630 			  struct cfg80211_crypto_settings *crypto,
3631 			  enum nl80211_iftype iftype)
3632 {
3633 	const struct ieee80211_cipher_scheme *cs;
3634 	int headroom = IEEE80211_ENCRYPT_HEADROOM;
3635 	int i;
3636 
3637 	for (i = 0; i < crypto->n_ciphers_pairwise; i++) {
3638 		cs = ieee80211_cs_get(local, crypto->ciphers_pairwise[i],
3639 				      iftype);
3640 
3641 		if (cs && headroom < cs->hdr_len)
3642 			headroom = cs->hdr_len;
3643 	}
3644 
3645 	cs = ieee80211_cs_get(local, crypto->cipher_group, iftype);
3646 	if (cs && headroom < cs->hdr_len)
3647 		headroom = cs->hdr_len;
3648 
3649 	return headroom;
3650 }
3651 
3652 static bool
3653 ieee80211_extend_noa_desc(struct ieee80211_noa_data *data, u32 tsf, int i)
3654 {
3655 	s32 end = data->desc[i].start + data->desc[i].duration - (tsf + 1);
3656 	int skip;
3657 
3658 	if (end > 0)
3659 		return false;
3660 
3661 	/* One shot NOA  */
3662 	if (data->count[i] == 1)
3663 		return false;
3664 
3665 	if (data->desc[i].interval == 0)
3666 		return false;
3667 
3668 	/* End time is in the past, check for repetitions */
3669 	skip = DIV_ROUND_UP(-end, data->desc[i].interval);
3670 	if (data->count[i] < 255) {
3671 		if (data->count[i] <= skip) {
3672 			data->count[i] = 0;
3673 			return false;
3674 		}
3675 
3676 		data->count[i] -= skip;
3677 	}
3678 
3679 	data->desc[i].start += skip * data->desc[i].interval;
3680 
3681 	return true;
3682 }
3683 
3684 static bool
3685 ieee80211_extend_absent_time(struct ieee80211_noa_data *data, u32 tsf,
3686 			     s32 *offset)
3687 {
3688 	bool ret = false;
3689 	int i;
3690 
3691 	for (i = 0; i < IEEE80211_P2P_NOA_DESC_MAX; i++) {
3692 		s32 cur;
3693 
3694 		if (!data->count[i])
3695 			continue;
3696 
3697 		if (ieee80211_extend_noa_desc(data, tsf + *offset, i))
3698 			ret = true;
3699 
3700 		cur = data->desc[i].start - tsf;
3701 		if (cur > *offset)
3702 			continue;
3703 
3704 		cur = data->desc[i].start + data->desc[i].duration - tsf;
3705 		if (cur > *offset)
3706 			*offset = cur;
3707 	}
3708 
3709 	return ret;
3710 }
3711 
3712 static u32
3713 ieee80211_get_noa_absent_time(struct ieee80211_noa_data *data, u32 tsf)
3714 {
3715 	s32 offset = 0;
3716 	int tries = 0;
3717 	/*
3718 	 * arbitrary limit, used to avoid infinite loops when combined NoA
3719 	 * descriptors cover the full time period.
3720 	 */
3721 	int max_tries = 5;
3722 
3723 	ieee80211_extend_absent_time(data, tsf, &offset);
3724 	do {
3725 		if (!ieee80211_extend_absent_time(data, tsf, &offset))
3726 			break;
3727 
3728 		tries++;
3729 	} while (tries < max_tries);
3730 
3731 	return offset;
3732 }
3733 
3734 void ieee80211_update_p2p_noa(struct ieee80211_noa_data *data, u32 tsf)
3735 {
3736 	u32 next_offset = BIT(31) - 1;
3737 	int i;
3738 
3739 	data->absent = 0;
3740 	data->has_next_tsf = false;
3741 	for (i = 0; i < IEEE80211_P2P_NOA_DESC_MAX; i++) {
3742 		s32 start;
3743 
3744 		if (!data->count[i])
3745 			continue;
3746 
3747 		ieee80211_extend_noa_desc(data, tsf, i);
3748 		start = data->desc[i].start - tsf;
3749 		if (start <= 0)
3750 			data->absent |= BIT(i);
3751 
3752 		if (next_offset > start)
3753 			next_offset = start;
3754 
3755 		data->has_next_tsf = true;
3756 	}
3757 
3758 	if (data->absent)
3759 		next_offset = ieee80211_get_noa_absent_time(data, tsf);
3760 
3761 	data->next_tsf = tsf + next_offset;
3762 }
3763 EXPORT_SYMBOL(ieee80211_update_p2p_noa);
3764 
3765 int ieee80211_parse_p2p_noa(const struct ieee80211_p2p_noa_attr *attr,
3766 			    struct ieee80211_noa_data *data, u32 tsf)
3767 {
3768 	int ret = 0;
3769 	int i;
3770 
3771 	memset(data, 0, sizeof(*data));
3772 
3773 	for (i = 0; i < IEEE80211_P2P_NOA_DESC_MAX; i++) {
3774 		const struct ieee80211_p2p_noa_desc *desc = &attr->desc[i];
3775 
3776 		if (!desc->count || !desc->duration)
3777 			continue;
3778 
3779 		data->count[i] = desc->count;
3780 		data->desc[i].start = le32_to_cpu(desc->start_time);
3781 		data->desc[i].duration = le32_to_cpu(desc->duration);
3782 		data->desc[i].interval = le32_to_cpu(desc->interval);
3783 
3784 		if (data->count[i] > 1 &&
3785 		    data->desc[i].interval < data->desc[i].duration)
3786 			continue;
3787 
3788 		ieee80211_extend_noa_desc(data, tsf, i);
3789 		ret++;
3790 	}
3791 
3792 	if (ret)
3793 		ieee80211_update_p2p_noa(data, tsf);
3794 
3795 	return ret;
3796 }
3797 EXPORT_SYMBOL(ieee80211_parse_p2p_noa);
3798 
3799 void ieee80211_recalc_dtim(struct ieee80211_local *local,
3800 			   struct ieee80211_sub_if_data *sdata)
3801 {
3802 	u64 tsf = drv_get_tsf(local, sdata);
3803 	u64 dtim_count = 0;
3804 	u16 beacon_int = sdata->vif.bss_conf.beacon_int * 1024;
3805 	u8 dtim_period = sdata->vif.bss_conf.dtim_period;
3806 	struct ps_data *ps;
3807 	u8 bcns_from_dtim;
3808 
3809 	if (tsf == -1ULL || !beacon_int || !dtim_period)
3810 		return;
3811 
3812 	if (sdata->vif.type == NL80211_IFTYPE_AP ||
3813 	    sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
3814 		if (!sdata->bss)
3815 			return;
3816 
3817 		ps = &sdata->bss->ps;
3818 	} else if (ieee80211_vif_is_mesh(&sdata->vif)) {
3819 		ps = &sdata->u.mesh.ps;
3820 	} else {
3821 		return;
3822 	}
3823 
3824 	/*
3825 	 * actually finds last dtim_count, mac80211 will update in
3826 	 * __beacon_add_tim().
3827 	 * dtim_count = dtim_period - (tsf / bcn_int) % dtim_period
3828 	 */
3829 	do_div(tsf, beacon_int);
3830 	bcns_from_dtim = do_div(tsf, dtim_period);
3831 	/* just had a DTIM */
3832 	if (!bcns_from_dtim)
3833 		dtim_count = 0;
3834 	else
3835 		dtim_count = dtim_period - bcns_from_dtim;
3836 
3837 	ps->dtim_count = dtim_count;
3838 }
3839 
3840 static u8 ieee80211_chanctx_radar_detect(struct ieee80211_local *local,
3841 					 struct ieee80211_chanctx *ctx)
3842 {
3843 	struct ieee80211_sub_if_data *sdata;
3844 	u8 radar_detect = 0;
3845 
3846 	lockdep_assert_held(&local->chanctx_mtx);
3847 
3848 	if (WARN_ON(ctx->replace_state == IEEE80211_CHANCTX_WILL_BE_REPLACED))
3849 		return 0;
3850 
3851 	list_for_each_entry(sdata, &ctx->reserved_vifs, reserved_chanctx_list)
3852 		if (sdata->reserved_radar_required)
3853 			radar_detect |= BIT(sdata->reserved_chandef.width);
3854 
3855 	/*
3856 	 * An in-place reservation context should not have any assigned vifs
3857 	 * until it replaces the other context.
3858 	 */
3859 	WARN_ON(ctx->replace_state == IEEE80211_CHANCTX_REPLACES_OTHER &&
3860 		!list_empty(&ctx->assigned_vifs));
3861 
3862 	list_for_each_entry(sdata, &ctx->assigned_vifs, assigned_chanctx_list)
3863 		if (sdata->radar_required)
3864 			radar_detect |= BIT(sdata->vif.bss_conf.chandef.width);
3865 
3866 	return radar_detect;
3867 }
3868 
3869 int ieee80211_check_combinations(struct ieee80211_sub_if_data *sdata,
3870 				 const struct cfg80211_chan_def *chandef,
3871 				 enum ieee80211_chanctx_mode chanmode,
3872 				 u8 radar_detect)
3873 {
3874 	struct ieee80211_local *local = sdata->local;
3875 	struct ieee80211_sub_if_data *sdata_iter;
3876 	enum nl80211_iftype iftype = sdata->wdev.iftype;
3877 	struct ieee80211_chanctx *ctx;
3878 	int total = 1;
3879 	struct iface_combination_params params = {
3880 		.radar_detect = radar_detect,
3881 	};
3882 
3883 	lockdep_assert_held(&local->chanctx_mtx);
3884 
3885 	if (WARN_ON(hweight32(radar_detect) > 1))
3886 		return -EINVAL;
3887 
3888 	if (WARN_ON(chandef && chanmode == IEEE80211_CHANCTX_SHARED &&
3889 		    !chandef->chan))
3890 		return -EINVAL;
3891 
3892 	if (WARN_ON(iftype >= NUM_NL80211_IFTYPES))
3893 		return -EINVAL;
3894 
3895 	if (sdata->vif.type == NL80211_IFTYPE_AP ||
3896 	    sdata->vif.type == NL80211_IFTYPE_MESH_POINT) {
3897 		/*
3898 		 * always passing this is harmless, since it'll be the
3899 		 * same value that cfg80211 finds if it finds the same
3900 		 * interface ... and that's always allowed
3901 		 */
3902 		params.new_beacon_int = sdata->vif.bss_conf.beacon_int;
3903 	}
3904 
3905 	/* Always allow software iftypes */
3906 	if (cfg80211_iftype_allowed(local->hw.wiphy, iftype, 0, 1)) {
3907 		if (radar_detect)
3908 			return -EINVAL;
3909 		return 0;
3910 	}
3911 
3912 	if (chandef)
3913 		params.num_different_channels = 1;
3914 
3915 	if (iftype != NL80211_IFTYPE_UNSPECIFIED)
3916 		params.iftype_num[iftype] = 1;
3917 
3918 	list_for_each_entry(ctx, &local->chanctx_list, list) {
3919 		if (ctx->replace_state == IEEE80211_CHANCTX_WILL_BE_REPLACED)
3920 			continue;
3921 		params.radar_detect |=
3922 			ieee80211_chanctx_radar_detect(local, ctx);
3923 		if (ctx->mode == IEEE80211_CHANCTX_EXCLUSIVE) {
3924 			params.num_different_channels++;
3925 			continue;
3926 		}
3927 		if (chandef && chanmode == IEEE80211_CHANCTX_SHARED &&
3928 		    cfg80211_chandef_compatible(chandef,
3929 						&ctx->conf.def))
3930 			continue;
3931 		params.num_different_channels++;
3932 	}
3933 
3934 	list_for_each_entry_rcu(sdata_iter, &local->interfaces, list) {
3935 		struct wireless_dev *wdev_iter;
3936 
3937 		wdev_iter = &sdata_iter->wdev;
3938 
3939 		if (sdata_iter == sdata ||
3940 		    !ieee80211_sdata_running(sdata_iter) ||
3941 		    cfg80211_iftype_allowed(local->hw.wiphy,
3942 					    wdev_iter->iftype, 0, 1))
3943 			continue;
3944 
3945 		params.iftype_num[wdev_iter->iftype]++;
3946 		total++;
3947 	}
3948 
3949 	if (total == 1 && !params.radar_detect)
3950 		return 0;
3951 
3952 	return cfg80211_check_combinations(local->hw.wiphy, &params);
3953 }
3954 
3955 static void
3956 ieee80211_iter_max_chans(const struct ieee80211_iface_combination *c,
3957 			 void *data)
3958 {
3959 	u32 *max_num_different_channels = data;
3960 
3961 	*max_num_different_channels = max(*max_num_different_channels,
3962 					  c->num_different_channels);
3963 }
3964 
3965 int ieee80211_max_num_channels(struct ieee80211_local *local)
3966 {
3967 	struct ieee80211_sub_if_data *sdata;
3968 	struct ieee80211_chanctx *ctx;
3969 	u32 max_num_different_channels = 1;
3970 	int err;
3971 	struct iface_combination_params params = {0};
3972 
3973 	lockdep_assert_held(&local->chanctx_mtx);
3974 
3975 	list_for_each_entry(ctx, &local->chanctx_list, list) {
3976 		if (ctx->replace_state == IEEE80211_CHANCTX_WILL_BE_REPLACED)
3977 			continue;
3978 
3979 		params.num_different_channels++;
3980 
3981 		params.radar_detect |=
3982 			ieee80211_chanctx_radar_detect(local, ctx);
3983 	}
3984 
3985 	list_for_each_entry_rcu(sdata, &local->interfaces, list)
3986 		params.iftype_num[sdata->wdev.iftype]++;
3987 
3988 	err = cfg80211_iter_combinations(local->hw.wiphy, &params,
3989 					 ieee80211_iter_max_chans,
3990 					 &max_num_different_channels);
3991 	if (err < 0)
3992 		return err;
3993 
3994 	return max_num_different_channels;
3995 }
3996 
3997 u8 *ieee80211_add_wmm_info_ie(u8 *buf, u8 qosinfo)
3998 {
3999 	*buf++ = WLAN_EID_VENDOR_SPECIFIC;
4000 	*buf++ = 7; /* len */
4001 	*buf++ = 0x00; /* Microsoft OUI 00:50:F2 */
4002 	*buf++ = 0x50;
4003 	*buf++ = 0xf2;
4004 	*buf++ = 2; /* WME */
4005 	*buf++ = 0; /* WME info */
4006 	*buf++ = 1; /* WME ver */
4007 	*buf++ = qosinfo; /* U-APSD no in use */
4008 
4009 	return buf;
4010 }
4011 
4012 void ieee80211_txq_get_depth(struct ieee80211_txq *txq,
4013 			     unsigned long *frame_cnt,
4014 			     unsigned long *byte_cnt)
4015 {
4016 	struct txq_info *txqi = to_txq_info(txq);
4017 	u32 frag_cnt = 0, frag_bytes = 0;
4018 	struct sk_buff *skb;
4019 
4020 	skb_queue_walk(&txqi->frags, skb) {
4021 		frag_cnt++;
4022 		frag_bytes += skb->len;
4023 	}
4024 
4025 	if (frame_cnt)
4026 		*frame_cnt = txqi->tin.backlog_packets + frag_cnt;
4027 
4028 	if (byte_cnt)
4029 		*byte_cnt = txqi->tin.backlog_bytes + frag_bytes;
4030 }
4031 EXPORT_SYMBOL(ieee80211_txq_get_depth);
4032 
4033 const u8 ieee80211_ac_to_qos_mask[IEEE80211_NUM_ACS] = {
4034 	IEEE80211_WMM_IE_STA_QOSINFO_AC_VO,
4035 	IEEE80211_WMM_IE_STA_QOSINFO_AC_VI,
4036 	IEEE80211_WMM_IE_STA_QOSINFO_AC_BE,
4037 	IEEE80211_WMM_IE_STA_QOSINFO_AC_BK
4038 };
4039