]> git.karo-electronics.de Git - karo-tx-linux.git/blob - net/mac80211/cfg.c
mac80211: optimise mixed AP/VLAN station removal
[karo-tx-linux.git] / net / mac80211 / cfg.c
1 /*
2  * mac80211 configuration hooks for cfg80211
3  *
4  * Copyright 2006-2010  Johannes Berg <johannes@sipsolutions.net>
5  *
6  * This file is GPLv2 as found in COPYING.
7  */
8
9 #include <linux/ieee80211.h>
10 #include <linux/nl80211.h>
11 #include <linux/rtnetlink.h>
12 #include <linux/slab.h>
13 #include <net/net_namespace.h>
14 #include <linux/rcupdate.h>
15 #include <linux/if_ether.h>
16 #include <net/cfg80211.h>
17 #include "ieee80211_i.h"
18 #include "driver-ops.h"
19 #include "cfg.h"
20 #include "rate.h"
21 #include "mesh.h"
22
23 static struct wireless_dev *ieee80211_add_iface(struct wiphy *wiphy,
24                                                 const char *name,
25                                                 enum nl80211_iftype type,
26                                                 u32 *flags,
27                                                 struct vif_params *params)
28 {
29         struct ieee80211_local *local = wiphy_priv(wiphy);
30         struct wireless_dev *wdev;
31         struct ieee80211_sub_if_data *sdata;
32         int err;
33
34         err = ieee80211_if_add(local, name, &wdev, type, params);
35         if (err)
36                 return ERR_PTR(err);
37
38         if (type == NL80211_IFTYPE_MONITOR && flags) {
39                 sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
40                 sdata->u.mntr_flags = *flags;
41         }
42
43         return wdev;
44 }
45
46 static int ieee80211_del_iface(struct wiphy *wiphy, struct wireless_dev *wdev)
47 {
48         ieee80211_if_remove(IEEE80211_WDEV_TO_SUB_IF(wdev));
49
50         return 0;
51 }
52
53 static int ieee80211_change_iface(struct wiphy *wiphy,
54                                   struct net_device *dev,
55                                   enum nl80211_iftype type, u32 *flags,
56                                   struct vif_params *params)
57 {
58         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
59         int ret;
60
61         ret = ieee80211_if_change_type(sdata, type);
62         if (ret)
63                 return ret;
64
65         if (type == NL80211_IFTYPE_AP_VLAN &&
66             params && params->use_4addr == 0)
67                 RCU_INIT_POINTER(sdata->u.vlan.sta, NULL);
68         else if (type == NL80211_IFTYPE_STATION &&
69                  params && params->use_4addr >= 0)
70                 sdata->u.mgd.use_4addr = params->use_4addr;
71
72         if (sdata->vif.type == NL80211_IFTYPE_MONITOR && flags) {
73                 struct ieee80211_local *local = sdata->local;
74
75                 if (ieee80211_sdata_running(sdata)) {
76                         u32 mask = MONITOR_FLAG_COOK_FRAMES |
77                                    MONITOR_FLAG_ACTIVE;
78
79                         /*
80                          * Prohibit MONITOR_FLAG_COOK_FRAMES and
81                          * MONITOR_FLAG_ACTIVE to be changed while the
82                          * interface is up.
83                          * Else we would need to add a lot of cruft
84                          * to update everything:
85                          *      cooked_mntrs, monitor and all fif_* counters
86                          *      reconfigure hardware
87                          */
88                         if ((*flags & mask) != (sdata->u.mntr_flags & mask))
89                                 return -EBUSY;
90
91                         ieee80211_adjust_monitor_flags(sdata, -1);
92                         sdata->u.mntr_flags = *flags;
93                         ieee80211_adjust_monitor_flags(sdata, 1);
94
95                         ieee80211_configure_filter(local);
96                 } else {
97                         /*
98                          * Because the interface is down, ieee80211_do_stop
99                          * and ieee80211_do_open take care of "everything"
100                          * mentioned in the comment above.
101                          */
102                         sdata->u.mntr_flags = *flags;
103                 }
104         }
105
106         return 0;
107 }
108
109 static int ieee80211_start_p2p_device(struct wiphy *wiphy,
110                                       struct wireless_dev *wdev)
111 {
112         return ieee80211_do_open(wdev, true);
113 }
114
115 static void ieee80211_stop_p2p_device(struct wiphy *wiphy,
116                                       struct wireless_dev *wdev)
117 {
118         ieee80211_sdata_stop(IEEE80211_WDEV_TO_SUB_IF(wdev));
119 }
120
121 static int ieee80211_set_noack_map(struct wiphy *wiphy,
122                                   struct net_device *dev,
123                                   u16 noack_map)
124 {
125         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
126
127         sdata->noack_map = noack_map;
128         return 0;
129 }
130
131 static int ieee80211_add_key(struct wiphy *wiphy, struct net_device *dev,
132                              u8 key_idx, bool pairwise, const u8 *mac_addr,
133                              struct key_params *params)
134 {
135         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
136         struct ieee80211_local *local = sdata->local;
137         struct sta_info *sta = NULL;
138         const struct ieee80211_cipher_scheme *cs = NULL;
139         struct ieee80211_key *key;
140         int err;
141
142         if (!ieee80211_sdata_running(sdata))
143                 return -ENETDOWN;
144
145         /* reject WEP and TKIP keys if WEP failed to initialize */
146         switch (params->cipher) {
147         case WLAN_CIPHER_SUITE_WEP40:
148         case WLAN_CIPHER_SUITE_TKIP:
149         case WLAN_CIPHER_SUITE_WEP104:
150                 if (IS_ERR(local->wep_tx_tfm))
151                         return -EINVAL;
152                 break;
153         case WLAN_CIPHER_SUITE_CCMP:
154         case WLAN_CIPHER_SUITE_AES_CMAC:
155         case WLAN_CIPHER_SUITE_GCMP:
156                 break;
157         default:
158                 cs = ieee80211_cs_get(local, params->cipher, sdata->vif.type);
159                 break;
160         }
161
162         key = ieee80211_key_alloc(params->cipher, key_idx, params->key_len,
163                                   params->key, params->seq_len, params->seq,
164                                   cs);
165         if (IS_ERR(key))
166                 return PTR_ERR(key);
167
168         if (pairwise)
169                 key->conf.flags |= IEEE80211_KEY_FLAG_PAIRWISE;
170
171         mutex_lock(&local->sta_mtx);
172
173         if (mac_addr) {
174                 if (ieee80211_vif_is_mesh(&sdata->vif))
175                         sta = sta_info_get(sdata, mac_addr);
176                 else
177                         sta = sta_info_get_bss(sdata, mac_addr);
178                 /*
179                  * The ASSOC test makes sure the driver is ready to
180                  * receive the key. When wpa_supplicant has roamed
181                  * using FT, it attempts to set the key before
182                  * association has completed, this rejects that attempt
183                  * so it will set the key again after assocation.
184                  *
185                  * TODO: accept the key if we have a station entry and
186                  *       add it to the device after the station.
187                  */
188                 if (!sta || !test_sta_flag(sta, WLAN_STA_ASSOC)) {
189                         ieee80211_key_free_unused(key);
190                         err = -ENOENT;
191                         goto out_unlock;
192                 }
193         }
194
195         switch (sdata->vif.type) {
196         case NL80211_IFTYPE_STATION:
197                 if (sdata->u.mgd.mfp != IEEE80211_MFP_DISABLED)
198                         key->conf.flags |= IEEE80211_KEY_FLAG_RX_MGMT;
199                 break;
200         case NL80211_IFTYPE_AP:
201         case NL80211_IFTYPE_AP_VLAN:
202                 /* Keys without a station are used for TX only */
203                 if (key->sta && test_sta_flag(key->sta, WLAN_STA_MFP))
204                         key->conf.flags |= IEEE80211_KEY_FLAG_RX_MGMT;
205                 break;
206         case NL80211_IFTYPE_ADHOC:
207                 /* no MFP (yet) */
208                 break;
209         case NL80211_IFTYPE_MESH_POINT:
210 #ifdef CONFIG_MAC80211_MESH
211                 if (sdata->u.mesh.security != IEEE80211_MESH_SEC_NONE)
212                         key->conf.flags |= IEEE80211_KEY_FLAG_RX_MGMT;
213                 break;
214 #endif
215         case NL80211_IFTYPE_WDS:
216         case NL80211_IFTYPE_MONITOR:
217         case NL80211_IFTYPE_P2P_DEVICE:
218         case NL80211_IFTYPE_UNSPECIFIED:
219         case NUM_NL80211_IFTYPES:
220         case NL80211_IFTYPE_P2P_CLIENT:
221         case NL80211_IFTYPE_P2P_GO:
222                 /* shouldn't happen */
223                 WARN_ON_ONCE(1);
224                 break;
225         }
226
227         if (sta)
228                 sta->cipher_scheme = cs;
229
230         err = ieee80211_key_link(key, sdata, sta);
231
232  out_unlock:
233         mutex_unlock(&local->sta_mtx);
234
235         return err;
236 }
237
238 static int ieee80211_del_key(struct wiphy *wiphy, struct net_device *dev,
239                              u8 key_idx, bool pairwise, const u8 *mac_addr)
240 {
241         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
242         struct ieee80211_local *local = sdata->local;
243         struct sta_info *sta;
244         struct ieee80211_key *key = NULL;
245         int ret;
246
247         mutex_lock(&local->sta_mtx);
248         mutex_lock(&local->key_mtx);
249
250         if (mac_addr) {
251                 ret = -ENOENT;
252
253                 sta = sta_info_get_bss(sdata, mac_addr);
254                 if (!sta)
255                         goto out_unlock;
256
257                 if (pairwise)
258                         key = key_mtx_dereference(local, sta->ptk[key_idx]);
259                 else
260                         key = key_mtx_dereference(local, sta->gtk[key_idx]);
261         } else
262                 key = key_mtx_dereference(local, sdata->keys[key_idx]);
263
264         if (!key) {
265                 ret = -ENOENT;
266                 goto out_unlock;
267         }
268
269         ieee80211_key_free(key, true);
270
271         ret = 0;
272  out_unlock:
273         mutex_unlock(&local->key_mtx);
274         mutex_unlock(&local->sta_mtx);
275
276         return ret;
277 }
278
279 static int ieee80211_get_key(struct wiphy *wiphy, struct net_device *dev,
280                              u8 key_idx, bool pairwise, const u8 *mac_addr,
281                              void *cookie,
282                              void (*callback)(void *cookie,
283                                               struct key_params *params))
284 {
285         struct ieee80211_sub_if_data *sdata;
286         struct sta_info *sta = NULL;
287         u8 seq[6] = {0};
288         struct key_params params;
289         struct ieee80211_key *key = NULL;
290         u64 pn64;
291         u32 iv32;
292         u16 iv16;
293         int err = -ENOENT;
294
295         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
296
297         rcu_read_lock();
298
299         if (mac_addr) {
300                 sta = sta_info_get_bss(sdata, mac_addr);
301                 if (!sta)
302                         goto out;
303
304                 if (pairwise)
305                         key = rcu_dereference(sta->ptk[key_idx]);
306                 else if (key_idx < NUM_DEFAULT_KEYS)
307                         key = rcu_dereference(sta->gtk[key_idx]);
308         } else
309                 key = rcu_dereference(sdata->keys[key_idx]);
310
311         if (!key)
312                 goto out;
313
314         memset(&params, 0, sizeof(params));
315
316         params.cipher = key->conf.cipher;
317
318         switch (key->conf.cipher) {
319         case WLAN_CIPHER_SUITE_TKIP:
320                 iv32 = key->u.tkip.tx.iv32;
321                 iv16 = key->u.tkip.tx.iv16;
322
323                 if (key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE)
324                         drv_get_tkip_seq(sdata->local,
325                                          key->conf.hw_key_idx,
326                                          &iv32, &iv16);
327
328                 seq[0] = iv16 & 0xff;
329                 seq[1] = (iv16 >> 8) & 0xff;
330                 seq[2] = iv32 & 0xff;
331                 seq[3] = (iv32 >> 8) & 0xff;
332                 seq[4] = (iv32 >> 16) & 0xff;
333                 seq[5] = (iv32 >> 24) & 0xff;
334                 params.seq = seq;
335                 params.seq_len = 6;
336                 break;
337         case WLAN_CIPHER_SUITE_CCMP:
338                 pn64 = atomic64_read(&key->u.ccmp.tx_pn);
339                 seq[0] = pn64;
340                 seq[1] = pn64 >> 8;
341                 seq[2] = pn64 >> 16;
342                 seq[3] = pn64 >> 24;
343                 seq[4] = pn64 >> 32;
344                 seq[5] = pn64 >> 40;
345                 params.seq = seq;
346                 params.seq_len = 6;
347                 break;
348         case WLAN_CIPHER_SUITE_AES_CMAC:
349                 pn64 = atomic64_read(&key->u.aes_cmac.tx_pn);
350                 seq[0] = pn64;
351                 seq[1] = pn64 >> 8;
352                 seq[2] = pn64 >> 16;
353                 seq[3] = pn64 >> 24;
354                 seq[4] = pn64 >> 32;
355                 seq[5] = pn64 >> 40;
356                 params.seq = seq;
357                 params.seq_len = 6;
358                 break;
359         }
360
361         params.key = key->conf.key;
362         params.key_len = key->conf.keylen;
363
364         callback(cookie, &params);
365         err = 0;
366
367  out:
368         rcu_read_unlock();
369         return err;
370 }
371
372 static int ieee80211_config_default_key(struct wiphy *wiphy,
373                                         struct net_device *dev,
374                                         u8 key_idx, bool uni,
375                                         bool multi)
376 {
377         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
378
379         ieee80211_set_default_key(sdata, key_idx, uni, multi);
380
381         return 0;
382 }
383
384 static int ieee80211_config_default_mgmt_key(struct wiphy *wiphy,
385                                              struct net_device *dev,
386                                              u8 key_idx)
387 {
388         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
389
390         ieee80211_set_default_mgmt_key(sdata, key_idx);
391
392         return 0;
393 }
394
395 void sta_set_rate_info_tx(struct sta_info *sta,
396                           const struct ieee80211_tx_rate *rate,
397                           struct rate_info *rinfo)
398 {
399         rinfo->flags = 0;
400         if (rate->flags & IEEE80211_TX_RC_MCS) {
401                 rinfo->flags |= RATE_INFO_FLAGS_MCS;
402                 rinfo->mcs = rate->idx;
403         } else if (rate->flags & IEEE80211_TX_RC_VHT_MCS) {
404                 rinfo->flags |= RATE_INFO_FLAGS_VHT_MCS;
405                 rinfo->mcs = ieee80211_rate_get_vht_mcs(rate);
406                 rinfo->nss = ieee80211_rate_get_vht_nss(rate);
407         } else {
408                 struct ieee80211_supported_band *sband;
409                 int shift = ieee80211_vif_get_shift(&sta->sdata->vif);
410                 u16 brate;
411
412                 sband = sta->local->hw.wiphy->bands[
413                                 ieee80211_get_sdata_band(sta->sdata)];
414                 brate = sband->bitrates[rate->idx].bitrate;
415                 rinfo->legacy = DIV_ROUND_UP(brate, 1 << shift);
416         }
417         if (rate->flags & IEEE80211_TX_RC_40_MHZ_WIDTH)
418                 rinfo->flags |= RATE_INFO_FLAGS_40_MHZ_WIDTH;
419         if (rate->flags & IEEE80211_TX_RC_80_MHZ_WIDTH)
420                 rinfo->flags |= RATE_INFO_FLAGS_80_MHZ_WIDTH;
421         if (rate->flags & IEEE80211_TX_RC_160_MHZ_WIDTH)
422                 rinfo->flags |= RATE_INFO_FLAGS_160_MHZ_WIDTH;
423         if (rate->flags & IEEE80211_TX_RC_SHORT_GI)
424                 rinfo->flags |= RATE_INFO_FLAGS_SHORT_GI;
425 }
426
427 void sta_set_rate_info_rx(struct sta_info *sta, struct rate_info *rinfo)
428 {
429         rinfo->flags = 0;
430
431         if (sta->last_rx_rate_flag & RX_FLAG_HT) {
432                 rinfo->flags |= RATE_INFO_FLAGS_MCS;
433                 rinfo->mcs = sta->last_rx_rate_idx;
434         } else if (sta->last_rx_rate_flag & RX_FLAG_VHT) {
435                 rinfo->flags |= RATE_INFO_FLAGS_VHT_MCS;
436                 rinfo->nss = sta->last_rx_rate_vht_nss;
437                 rinfo->mcs = sta->last_rx_rate_idx;
438         } else {
439                 struct ieee80211_supported_band *sband;
440                 int shift = ieee80211_vif_get_shift(&sta->sdata->vif);
441                 u16 brate;
442
443                 sband = sta->local->hw.wiphy->bands[
444                                 ieee80211_get_sdata_band(sta->sdata)];
445                 brate = sband->bitrates[sta->last_rx_rate_idx].bitrate;
446                 rinfo->legacy = DIV_ROUND_UP(brate, 1 << shift);
447         }
448
449         if (sta->last_rx_rate_flag & RX_FLAG_40MHZ)
450                 rinfo->flags |= RATE_INFO_FLAGS_40_MHZ_WIDTH;
451         if (sta->last_rx_rate_flag & RX_FLAG_SHORT_GI)
452                 rinfo->flags |= RATE_INFO_FLAGS_SHORT_GI;
453         if (sta->last_rx_rate_flag & RX_FLAG_80MHZ)
454                 rinfo->flags |= RATE_INFO_FLAGS_80_MHZ_WIDTH;
455         if (sta->last_rx_rate_flag & RX_FLAG_80P80MHZ)
456                 rinfo->flags |= RATE_INFO_FLAGS_80P80_MHZ_WIDTH;
457         if (sta->last_rx_rate_flag & RX_FLAG_160MHZ)
458                 rinfo->flags |= RATE_INFO_FLAGS_160_MHZ_WIDTH;
459 }
460
461 static void sta_set_sinfo(struct sta_info *sta, struct station_info *sinfo)
462 {
463         struct ieee80211_sub_if_data *sdata = sta->sdata;
464         struct ieee80211_local *local = sdata->local;
465         struct timespec uptime;
466         u64 packets = 0;
467         int i, ac;
468
469         sinfo->generation = sdata->local->sta_generation;
470
471         sinfo->filled = STATION_INFO_INACTIVE_TIME |
472                         STATION_INFO_RX_BYTES64 |
473                         STATION_INFO_TX_BYTES64 |
474                         STATION_INFO_RX_PACKETS |
475                         STATION_INFO_TX_PACKETS |
476                         STATION_INFO_TX_RETRIES |
477                         STATION_INFO_TX_FAILED |
478                         STATION_INFO_TX_BITRATE |
479                         STATION_INFO_RX_BITRATE |
480                         STATION_INFO_RX_DROP_MISC |
481                         STATION_INFO_BSS_PARAM |
482                         STATION_INFO_CONNECTED_TIME |
483                         STATION_INFO_STA_FLAGS |
484                         STATION_INFO_BEACON_LOSS_COUNT;
485
486         do_posix_clock_monotonic_gettime(&uptime);
487         sinfo->connected_time = uptime.tv_sec - sta->last_connected;
488
489         sinfo->inactive_time = jiffies_to_msecs(jiffies - sta->last_rx);
490         sinfo->tx_bytes = 0;
491         for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
492                 sinfo->tx_bytes += sta->tx_bytes[ac];
493                 packets += sta->tx_packets[ac];
494         }
495         sinfo->tx_packets = packets;
496         sinfo->rx_bytes = sta->rx_bytes;
497         sinfo->rx_packets = sta->rx_packets;
498         sinfo->tx_retries = sta->tx_retry_count;
499         sinfo->tx_failed = sta->tx_retry_failed;
500         sinfo->rx_dropped_misc = sta->rx_dropped;
501         sinfo->beacon_loss_count = sta->beacon_loss_count;
502
503         if ((sta->local->hw.flags & IEEE80211_HW_SIGNAL_DBM) ||
504             (sta->local->hw.flags & IEEE80211_HW_SIGNAL_UNSPEC)) {
505                 sinfo->filled |= STATION_INFO_SIGNAL | STATION_INFO_SIGNAL_AVG;
506                 if (!local->ops->get_rssi ||
507                     drv_get_rssi(local, sdata, &sta->sta, &sinfo->signal))
508                         sinfo->signal = (s8)sta->last_signal;
509                 sinfo->signal_avg = (s8) -ewma_read(&sta->avg_signal);
510         }
511         if (sta->chains) {
512                 sinfo->filled |= STATION_INFO_CHAIN_SIGNAL |
513                                  STATION_INFO_CHAIN_SIGNAL_AVG;
514
515                 sinfo->chains = sta->chains;
516                 for (i = 0; i < ARRAY_SIZE(sinfo->chain_signal); i++) {
517                         sinfo->chain_signal[i] = sta->chain_signal_last[i];
518                         sinfo->chain_signal_avg[i] =
519                                 (s8) -ewma_read(&sta->chain_signal_avg[i]);
520                 }
521         }
522
523         sta_set_rate_info_tx(sta, &sta->last_tx_rate, &sinfo->txrate);
524         sta_set_rate_info_rx(sta, &sinfo->rxrate);
525
526         if (ieee80211_vif_is_mesh(&sdata->vif)) {
527 #ifdef CONFIG_MAC80211_MESH
528                 sinfo->filled |= STATION_INFO_LLID |
529                                  STATION_INFO_PLID |
530                                  STATION_INFO_PLINK_STATE |
531                                  STATION_INFO_LOCAL_PM |
532                                  STATION_INFO_PEER_PM |
533                                  STATION_INFO_NONPEER_PM;
534
535                 sinfo->llid = sta->llid;
536                 sinfo->plid = sta->plid;
537                 sinfo->plink_state = sta->plink_state;
538                 if (test_sta_flag(sta, WLAN_STA_TOFFSET_KNOWN)) {
539                         sinfo->filled |= STATION_INFO_T_OFFSET;
540                         sinfo->t_offset = sta->t_offset;
541                 }
542                 sinfo->local_pm = sta->local_pm;
543                 sinfo->peer_pm = sta->peer_pm;
544                 sinfo->nonpeer_pm = sta->nonpeer_pm;
545 #endif
546         }
547
548         sinfo->bss_param.flags = 0;
549         if (sdata->vif.bss_conf.use_cts_prot)
550                 sinfo->bss_param.flags |= BSS_PARAM_FLAGS_CTS_PROT;
551         if (sdata->vif.bss_conf.use_short_preamble)
552                 sinfo->bss_param.flags |= BSS_PARAM_FLAGS_SHORT_PREAMBLE;
553         if (sdata->vif.bss_conf.use_short_slot)
554                 sinfo->bss_param.flags |= BSS_PARAM_FLAGS_SHORT_SLOT_TIME;
555         sinfo->bss_param.dtim_period = sdata->local->hw.conf.ps_dtim_period;
556         sinfo->bss_param.beacon_interval = sdata->vif.bss_conf.beacon_int;
557
558         sinfo->sta_flags.set = 0;
559         sinfo->sta_flags.mask = BIT(NL80211_STA_FLAG_AUTHORIZED) |
560                                 BIT(NL80211_STA_FLAG_SHORT_PREAMBLE) |
561                                 BIT(NL80211_STA_FLAG_WME) |
562                                 BIT(NL80211_STA_FLAG_MFP) |
563                                 BIT(NL80211_STA_FLAG_AUTHENTICATED) |
564                                 BIT(NL80211_STA_FLAG_ASSOCIATED) |
565                                 BIT(NL80211_STA_FLAG_TDLS_PEER);
566         if (test_sta_flag(sta, WLAN_STA_AUTHORIZED))
567                 sinfo->sta_flags.set |= BIT(NL80211_STA_FLAG_AUTHORIZED);
568         if (test_sta_flag(sta, WLAN_STA_SHORT_PREAMBLE))
569                 sinfo->sta_flags.set |= BIT(NL80211_STA_FLAG_SHORT_PREAMBLE);
570         if (test_sta_flag(sta, WLAN_STA_WME))
571                 sinfo->sta_flags.set |= BIT(NL80211_STA_FLAG_WME);
572         if (test_sta_flag(sta, WLAN_STA_MFP))
573                 sinfo->sta_flags.set |= BIT(NL80211_STA_FLAG_MFP);
574         if (test_sta_flag(sta, WLAN_STA_AUTH))
575                 sinfo->sta_flags.set |= BIT(NL80211_STA_FLAG_AUTHENTICATED);
576         if (test_sta_flag(sta, WLAN_STA_ASSOC))
577                 sinfo->sta_flags.set |= BIT(NL80211_STA_FLAG_ASSOCIATED);
578         if (test_sta_flag(sta, WLAN_STA_TDLS_PEER))
579                 sinfo->sta_flags.set |= BIT(NL80211_STA_FLAG_TDLS_PEER);
580 }
581
582 static const char ieee80211_gstrings_sta_stats[][ETH_GSTRING_LEN] = {
583         "rx_packets", "rx_bytes", "wep_weak_iv_count",
584         "rx_duplicates", "rx_fragments", "rx_dropped",
585         "tx_packets", "tx_bytes", "tx_fragments",
586         "tx_filtered", "tx_retry_failed", "tx_retries",
587         "beacon_loss", "sta_state", "txrate", "rxrate", "signal",
588         "channel", "noise", "ch_time", "ch_time_busy",
589         "ch_time_ext_busy", "ch_time_rx", "ch_time_tx"
590 };
591 #define STA_STATS_LEN   ARRAY_SIZE(ieee80211_gstrings_sta_stats)
592
593 static int ieee80211_get_et_sset_count(struct wiphy *wiphy,
594                                        struct net_device *dev,
595                                        int sset)
596 {
597         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
598         int rv = 0;
599
600         if (sset == ETH_SS_STATS)
601                 rv += STA_STATS_LEN;
602
603         rv += drv_get_et_sset_count(sdata, sset);
604
605         if (rv == 0)
606                 return -EOPNOTSUPP;
607         return rv;
608 }
609
610 static void ieee80211_get_et_stats(struct wiphy *wiphy,
611                                    struct net_device *dev,
612                                    struct ethtool_stats *stats,
613                                    u64 *data)
614 {
615         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
616         struct ieee80211_chanctx_conf *chanctx_conf;
617         struct ieee80211_channel *channel;
618         struct sta_info *sta;
619         struct ieee80211_local *local = sdata->local;
620         struct station_info sinfo;
621         struct survey_info survey;
622         int i, q;
623 #define STA_STATS_SURVEY_LEN 7
624
625         memset(data, 0, sizeof(u64) * STA_STATS_LEN);
626
627 #define ADD_STA_STATS(sta)                              \
628         do {                                            \
629                 data[i++] += sta->rx_packets;           \
630                 data[i++] += sta->rx_bytes;             \
631                 data[i++] += sta->wep_weak_iv_count;    \
632                 data[i++] += sta->num_duplicates;       \
633                 data[i++] += sta->rx_fragments;         \
634                 data[i++] += sta->rx_dropped;           \
635                                                         \
636                 data[i++] += sinfo.tx_packets;          \
637                 data[i++] += sinfo.tx_bytes;            \
638                 data[i++] += sta->tx_fragments;         \
639                 data[i++] += sta->tx_filtered_count;    \
640                 data[i++] += sta->tx_retry_failed;      \
641                 data[i++] += sta->tx_retry_count;       \
642                 data[i++] += sta->beacon_loss_count;    \
643         } while (0)
644
645         /* For Managed stations, find the single station based on BSSID
646          * and use that.  For interface types, iterate through all available
647          * stations and add stats for any station that is assigned to this
648          * network device.
649          */
650
651         mutex_lock(&local->sta_mtx);
652
653         if (sdata->vif.type == NL80211_IFTYPE_STATION) {
654                 sta = sta_info_get_bss(sdata, sdata->u.mgd.bssid);
655
656                 if (!(sta && !WARN_ON(sta->sdata->dev != dev)))
657                         goto do_survey;
658
659                 sinfo.filled = 0;
660                 sta_set_sinfo(sta, &sinfo);
661
662                 i = 0;
663                 ADD_STA_STATS(sta);
664
665                 data[i++] = sta->sta_state;
666
667
668                 if (sinfo.filled & STATION_INFO_TX_BITRATE)
669                         data[i] = 100000 *
670                                 cfg80211_calculate_bitrate(&sinfo.txrate);
671                 i++;
672                 if (sinfo.filled & STATION_INFO_RX_BITRATE)
673                         data[i] = 100000 *
674                                 cfg80211_calculate_bitrate(&sinfo.rxrate);
675                 i++;
676
677                 if (sinfo.filled & STATION_INFO_SIGNAL_AVG)
678                         data[i] = (u8)sinfo.signal_avg;
679                 i++;
680         } else {
681                 list_for_each_entry(sta, &local->sta_list, list) {
682                         /* Make sure this station belongs to the proper dev */
683                         if (sta->sdata->dev != dev)
684                                 continue;
685
686                         sinfo.filled = 0;
687                         sta_set_sinfo(sta, &sinfo);
688                         i = 0;
689                         ADD_STA_STATS(sta);
690                 }
691         }
692
693 do_survey:
694         i = STA_STATS_LEN - STA_STATS_SURVEY_LEN;
695         /* Get survey stats for current channel */
696         survey.filled = 0;
697
698         rcu_read_lock();
699         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
700         if (chanctx_conf)
701                 channel = chanctx_conf->def.chan;
702         else
703                 channel = NULL;
704         rcu_read_unlock();
705
706         if (channel) {
707                 q = 0;
708                 do {
709                         survey.filled = 0;
710                         if (drv_get_survey(local, q, &survey) != 0) {
711                                 survey.filled = 0;
712                                 break;
713                         }
714                         q++;
715                 } while (channel != survey.channel);
716         }
717
718         if (survey.filled)
719                 data[i++] = survey.channel->center_freq;
720         else
721                 data[i++] = 0;
722         if (survey.filled & SURVEY_INFO_NOISE_DBM)
723                 data[i++] = (u8)survey.noise;
724         else
725                 data[i++] = -1LL;
726         if (survey.filled & SURVEY_INFO_CHANNEL_TIME)
727                 data[i++] = survey.channel_time;
728         else
729                 data[i++] = -1LL;
730         if (survey.filled & SURVEY_INFO_CHANNEL_TIME_BUSY)
731                 data[i++] = survey.channel_time_busy;
732         else
733                 data[i++] = -1LL;
734         if (survey.filled & SURVEY_INFO_CHANNEL_TIME_EXT_BUSY)
735                 data[i++] = survey.channel_time_ext_busy;
736         else
737                 data[i++] = -1LL;
738         if (survey.filled & SURVEY_INFO_CHANNEL_TIME_RX)
739                 data[i++] = survey.channel_time_rx;
740         else
741                 data[i++] = -1LL;
742         if (survey.filled & SURVEY_INFO_CHANNEL_TIME_TX)
743                 data[i++] = survey.channel_time_tx;
744         else
745                 data[i++] = -1LL;
746
747         mutex_unlock(&local->sta_mtx);
748
749         if (WARN_ON(i != STA_STATS_LEN))
750                 return;
751
752         drv_get_et_stats(sdata, stats, &(data[STA_STATS_LEN]));
753 }
754
755 static void ieee80211_get_et_strings(struct wiphy *wiphy,
756                                      struct net_device *dev,
757                                      u32 sset, u8 *data)
758 {
759         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
760         int sz_sta_stats = 0;
761
762         if (sset == ETH_SS_STATS) {
763                 sz_sta_stats = sizeof(ieee80211_gstrings_sta_stats);
764                 memcpy(data, ieee80211_gstrings_sta_stats, sz_sta_stats);
765         }
766         drv_get_et_strings(sdata, sset, &(data[sz_sta_stats]));
767 }
768
769 static int ieee80211_dump_station(struct wiphy *wiphy, struct net_device *dev,
770                                  int idx, u8 *mac, struct station_info *sinfo)
771 {
772         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
773         struct ieee80211_local *local = sdata->local;
774         struct sta_info *sta;
775         int ret = -ENOENT;
776
777         mutex_lock(&local->sta_mtx);
778
779         sta = sta_info_get_by_idx(sdata, idx);
780         if (sta) {
781                 ret = 0;
782                 memcpy(mac, sta->sta.addr, ETH_ALEN);
783                 sta_set_sinfo(sta, sinfo);
784         }
785
786         mutex_unlock(&local->sta_mtx);
787
788         return ret;
789 }
790
791 static int ieee80211_dump_survey(struct wiphy *wiphy, struct net_device *dev,
792                                  int idx, struct survey_info *survey)
793 {
794         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
795
796         return drv_get_survey(local, idx, survey);
797 }
798
799 static int ieee80211_get_station(struct wiphy *wiphy, struct net_device *dev,
800                                  u8 *mac, struct station_info *sinfo)
801 {
802         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
803         struct ieee80211_local *local = sdata->local;
804         struct sta_info *sta;
805         int ret = -ENOENT;
806
807         mutex_lock(&local->sta_mtx);
808
809         sta = sta_info_get_bss(sdata, mac);
810         if (sta) {
811                 ret = 0;
812                 sta_set_sinfo(sta, sinfo);
813         }
814
815         mutex_unlock(&local->sta_mtx);
816
817         return ret;
818 }
819
820 static int ieee80211_set_monitor_channel(struct wiphy *wiphy,
821                                          struct cfg80211_chan_def *chandef)
822 {
823         struct ieee80211_local *local = wiphy_priv(wiphy);
824         struct ieee80211_sub_if_data *sdata;
825         int ret = 0;
826
827         if (cfg80211_chandef_identical(&local->monitor_chandef, chandef))
828                 return 0;
829
830         mutex_lock(&local->iflist_mtx);
831         if (local->use_chanctx) {
832                 sdata = rcu_dereference_protected(
833                                 local->monitor_sdata,
834                                 lockdep_is_held(&local->iflist_mtx));
835                 if (sdata) {
836                         ieee80211_vif_release_channel(sdata);
837                         ret = ieee80211_vif_use_channel(sdata, chandef,
838                                         IEEE80211_CHANCTX_EXCLUSIVE);
839                 }
840         } else if (local->open_count == local->monitors) {
841                 local->_oper_chandef = *chandef;
842                 ieee80211_hw_config(local, 0);
843         }
844
845         if (ret == 0)
846                 local->monitor_chandef = *chandef;
847         mutex_unlock(&local->iflist_mtx);
848
849         return ret;
850 }
851
852 static int ieee80211_set_probe_resp(struct ieee80211_sub_if_data *sdata,
853                                     const u8 *resp, size_t resp_len)
854 {
855         struct probe_resp *new, *old;
856
857         if (!resp || !resp_len)
858                 return 1;
859
860         old = sdata_dereference(sdata->u.ap.probe_resp, sdata);
861
862         new = kzalloc(sizeof(struct probe_resp) + resp_len, GFP_KERNEL);
863         if (!new)
864                 return -ENOMEM;
865
866         new->len = resp_len;
867         memcpy(new->data, resp, resp_len);
868
869         rcu_assign_pointer(sdata->u.ap.probe_resp, new);
870         if (old)
871                 kfree_rcu(old, rcu_head);
872
873         return 0;
874 }
875
876 int ieee80211_assign_beacon(struct ieee80211_sub_if_data *sdata,
877                             struct cfg80211_beacon_data *params)
878 {
879         struct beacon_data *new, *old;
880         int new_head_len, new_tail_len;
881         int size, err;
882         u32 changed = BSS_CHANGED_BEACON;
883
884         old = sdata_dereference(sdata->u.ap.beacon, sdata);
885
886
887         /* Need to have a beacon head if we don't have one yet */
888         if (!params->head && !old)
889                 return -EINVAL;
890
891         /* new or old head? */
892         if (params->head)
893                 new_head_len = params->head_len;
894         else
895                 new_head_len = old->head_len;
896
897         /* new or old tail? */
898         if (params->tail || !old)
899                 /* params->tail_len will be zero for !params->tail */
900                 new_tail_len = params->tail_len;
901         else
902                 new_tail_len = old->tail_len;
903
904         size = sizeof(*new) + new_head_len + new_tail_len;
905
906         new = kzalloc(size, GFP_KERNEL);
907         if (!new)
908                 return -ENOMEM;
909
910         /* start filling the new info now */
911
912         /*
913          * pointers go into the block we allocated,
914          * memory is | beacon_data | head | tail |
915          */
916         new->head = ((u8 *) new) + sizeof(*new);
917         new->tail = new->head + new_head_len;
918         new->head_len = new_head_len;
919         new->tail_len = new_tail_len;
920
921         /* copy in head */
922         if (params->head)
923                 memcpy(new->head, params->head, new_head_len);
924         else
925                 memcpy(new->head, old->head, new_head_len);
926
927         /* copy in optional tail */
928         if (params->tail)
929                 memcpy(new->tail, params->tail, new_tail_len);
930         else
931                 if (old)
932                         memcpy(new->tail, old->tail, new_tail_len);
933
934         err = ieee80211_set_probe_resp(sdata, params->probe_resp,
935                                        params->probe_resp_len);
936         if (err < 0)
937                 return err;
938         if (err == 0)
939                 changed |= BSS_CHANGED_AP_PROBE_RESP;
940
941         rcu_assign_pointer(sdata->u.ap.beacon, new);
942
943         if (old)
944                 kfree_rcu(old, rcu_head);
945
946         return changed;
947 }
948
949 static int ieee80211_start_ap(struct wiphy *wiphy, struct net_device *dev,
950                               struct cfg80211_ap_settings *params)
951 {
952         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
953         struct beacon_data *old;
954         struct ieee80211_sub_if_data *vlan;
955         u32 changed = BSS_CHANGED_BEACON_INT |
956                       BSS_CHANGED_BEACON_ENABLED |
957                       BSS_CHANGED_BEACON |
958                       BSS_CHANGED_SSID |
959                       BSS_CHANGED_P2P_PS;
960         int err;
961
962         old = sdata_dereference(sdata->u.ap.beacon, sdata);
963         if (old)
964                 return -EALREADY;
965
966         /* TODO: make hostapd tell us what it wants */
967         sdata->smps_mode = IEEE80211_SMPS_OFF;
968         sdata->needed_rx_chains = sdata->local->rx_chains;
969         sdata->radar_required = params->radar_required;
970
971         err = ieee80211_vif_use_channel(sdata, &params->chandef,
972                                         IEEE80211_CHANCTX_SHARED);
973         if (err)
974                 return err;
975         ieee80211_vif_copy_chanctx_to_vlans(sdata, false);
976
977         /*
978          * Apply control port protocol, this allows us to
979          * not encrypt dynamic WEP control frames.
980          */
981         sdata->control_port_protocol = params->crypto.control_port_ethertype;
982         sdata->control_port_no_encrypt = params->crypto.control_port_no_encrypt;
983         sdata->encrypt_headroom = ieee80211_cs_headroom(sdata->local,
984                                                         &params->crypto,
985                                                         sdata->vif.type);
986
987         list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list) {
988                 vlan->control_port_protocol =
989                         params->crypto.control_port_ethertype;
990                 vlan->control_port_no_encrypt =
991                         params->crypto.control_port_no_encrypt;
992                 vlan->encrypt_headroom =
993                         ieee80211_cs_headroom(sdata->local,
994                                               &params->crypto,
995                                               vlan->vif.type);
996         }
997
998         sdata->vif.bss_conf.beacon_int = params->beacon_interval;
999         sdata->vif.bss_conf.dtim_period = params->dtim_period;
1000         sdata->vif.bss_conf.enable_beacon = true;
1001
1002         sdata->vif.bss_conf.ssid_len = params->ssid_len;
1003         if (params->ssid_len)
1004                 memcpy(sdata->vif.bss_conf.ssid, params->ssid,
1005                        params->ssid_len);
1006         sdata->vif.bss_conf.hidden_ssid =
1007                 (params->hidden_ssid != NL80211_HIDDEN_SSID_NOT_IN_USE);
1008
1009         memset(&sdata->vif.bss_conf.p2p_noa_attr, 0,
1010                sizeof(sdata->vif.bss_conf.p2p_noa_attr));
1011         sdata->vif.bss_conf.p2p_noa_attr.oppps_ctwindow =
1012                 params->p2p_ctwindow & IEEE80211_P2P_OPPPS_CTWINDOW_MASK;
1013         if (params->p2p_opp_ps)
1014                 sdata->vif.bss_conf.p2p_noa_attr.oppps_ctwindow |=
1015                                         IEEE80211_P2P_OPPPS_ENABLE_BIT;
1016
1017         err = ieee80211_assign_beacon(sdata, &params->beacon);
1018         if (err < 0)
1019                 return err;
1020         changed |= err;
1021
1022         err = drv_start_ap(sdata->local, sdata);
1023         if (err) {
1024                 old = sdata_dereference(sdata->u.ap.beacon, sdata);
1025
1026                 if (old)
1027                         kfree_rcu(old, rcu_head);
1028                 RCU_INIT_POINTER(sdata->u.ap.beacon, NULL);
1029                 return err;
1030         }
1031
1032         ieee80211_bss_info_change_notify(sdata, changed);
1033
1034         netif_carrier_on(dev);
1035         list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
1036                 netif_carrier_on(vlan->dev);
1037
1038         return 0;
1039 }
1040
1041 static int ieee80211_change_beacon(struct wiphy *wiphy, struct net_device *dev,
1042                                    struct cfg80211_beacon_data *params)
1043 {
1044         struct ieee80211_sub_if_data *sdata;
1045         struct beacon_data *old;
1046         int err;
1047
1048         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1049
1050         /* don't allow changing the beacon while CSA is in place - offset
1051          * of channel switch counter may change
1052          */
1053         if (sdata->vif.csa_active)
1054                 return -EBUSY;
1055
1056         old = sdata_dereference(sdata->u.ap.beacon, sdata);
1057         if (!old)
1058                 return -ENOENT;
1059
1060         err = ieee80211_assign_beacon(sdata, params);
1061         if (err < 0)
1062                 return err;
1063         ieee80211_bss_info_change_notify(sdata, err);
1064         return 0;
1065 }
1066
1067 static int ieee80211_stop_ap(struct wiphy *wiphy, struct net_device *dev)
1068 {
1069         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1070         struct ieee80211_sub_if_data *vlan;
1071         struct ieee80211_local *local = sdata->local;
1072         struct beacon_data *old_beacon;
1073         struct probe_resp *old_probe_resp;
1074         struct cfg80211_chan_def chandef;
1075
1076         old_beacon = sdata_dereference(sdata->u.ap.beacon, sdata);
1077         if (!old_beacon)
1078                 return -ENOENT;
1079         old_probe_resp = sdata_dereference(sdata->u.ap.probe_resp, sdata);
1080
1081         /* abort any running channel switch */
1082         sdata->vif.csa_active = false;
1083         kfree(sdata->u.ap.next_beacon);
1084         sdata->u.ap.next_beacon = NULL;
1085
1086         cancel_work_sync(&sdata->u.ap.request_smps_work);
1087
1088         /* turn off carrier for this interface and dependent VLANs */
1089         list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
1090                 netif_carrier_off(vlan->dev);
1091         netif_carrier_off(dev);
1092
1093         /* remove beacon and probe response */
1094         RCU_INIT_POINTER(sdata->u.ap.beacon, NULL);
1095         RCU_INIT_POINTER(sdata->u.ap.probe_resp, NULL);
1096         kfree_rcu(old_beacon, rcu_head);
1097         if (old_probe_resp)
1098                 kfree_rcu(old_probe_resp, rcu_head);
1099
1100         __sta_info_flush(sdata, true);
1101         synchronize_net();
1102         list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
1103                 ieee80211_free_keys(vlan);
1104         ieee80211_free_keys(sdata);
1105
1106         sdata->vif.bss_conf.enable_beacon = false;
1107         sdata->vif.bss_conf.ssid_len = 0;
1108         clear_bit(SDATA_STATE_OFFCHANNEL_BEACON_STOPPED, &sdata->state);
1109         ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_BEACON_ENABLED);
1110
1111         if (sdata->wdev.cac_started) {
1112                 chandef = sdata->vif.bss_conf.chandef;
1113                 cancel_delayed_work_sync(&sdata->dfs_cac_timer_work);
1114                 cfg80211_cac_event(sdata->dev, &chandef,
1115                                    NL80211_RADAR_CAC_ABORTED,
1116                                    GFP_KERNEL);
1117         }
1118
1119         drv_stop_ap(sdata->local, sdata);
1120
1121         /* free all potentially still buffered bcast frames */
1122         local->total_ps_buffered -= skb_queue_len(&sdata->u.ap.ps.bc_buf);
1123         skb_queue_purge(&sdata->u.ap.ps.bc_buf);
1124
1125         ieee80211_vif_copy_chanctx_to_vlans(sdata, true);
1126         ieee80211_vif_release_channel(sdata);
1127
1128         return 0;
1129 }
1130
1131 /* Layer 2 Update frame (802.2 Type 1 LLC XID Update response) */
1132 struct iapp_layer2_update {
1133         u8 da[ETH_ALEN];        /* broadcast */
1134         u8 sa[ETH_ALEN];        /* STA addr */
1135         __be16 len;             /* 6 */
1136         u8 dsap;                /* 0 */
1137         u8 ssap;                /* 0 */
1138         u8 control;
1139         u8 xid_info[3];
1140 } __packed;
1141
1142 static void ieee80211_send_layer2_update(struct sta_info *sta)
1143 {
1144         struct iapp_layer2_update *msg;
1145         struct sk_buff *skb;
1146
1147         /* Send Level 2 Update Frame to update forwarding tables in layer 2
1148          * bridge devices */
1149
1150         skb = dev_alloc_skb(sizeof(*msg));
1151         if (!skb)
1152                 return;
1153         msg = (struct iapp_layer2_update *)skb_put(skb, sizeof(*msg));
1154
1155         /* 802.2 Type 1 Logical Link Control (LLC) Exchange Identifier (XID)
1156          * Update response frame; IEEE Std 802.2-1998, 5.4.1.2.1 */
1157
1158         eth_broadcast_addr(msg->da);
1159         memcpy(msg->sa, sta->sta.addr, ETH_ALEN);
1160         msg->len = htons(6);
1161         msg->dsap = 0;
1162         msg->ssap = 0x01;       /* NULL LSAP, CR Bit: Response */
1163         msg->control = 0xaf;    /* XID response lsb.1111F101.
1164                                  * F=0 (no poll command; unsolicited frame) */
1165         msg->xid_info[0] = 0x81;        /* XID format identifier */
1166         msg->xid_info[1] = 1;   /* LLC types/classes: Type 1 LLC */
1167         msg->xid_info[2] = 0;   /* XID sender's receive window size (RW) */
1168
1169         skb->dev = sta->sdata->dev;
1170         skb->protocol = eth_type_trans(skb, sta->sdata->dev);
1171         memset(skb->cb, 0, sizeof(skb->cb));
1172         netif_rx_ni(skb);
1173 }
1174
1175 static int sta_apply_auth_flags(struct ieee80211_local *local,
1176                                 struct sta_info *sta,
1177                                 u32 mask, u32 set)
1178 {
1179         int ret;
1180
1181         if (mask & BIT(NL80211_STA_FLAG_AUTHENTICATED) &&
1182             set & BIT(NL80211_STA_FLAG_AUTHENTICATED) &&
1183             !test_sta_flag(sta, WLAN_STA_AUTH)) {
1184                 ret = sta_info_move_state(sta, IEEE80211_STA_AUTH);
1185                 if (ret)
1186                         return ret;
1187         }
1188
1189         if (mask & BIT(NL80211_STA_FLAG_ASSOCIATED) &&
1190             set & BIT(NL80211_STA_FLAG_ASSOCIATED) &&
1191             !test_sta_flag(sta, WLAN_STA_ASSOC)) {
1192                 ret = sta_info_move_state(sta, IEEE80211_STA_ASSOC);
1193                 if (ret)
1194                         return ret;
1195         }
1196
1197         if (mask & BIT(NL80211_STA_FLAG_AUTHORIZED)) {
1198                 if (set & BIT(NL80211_STA_FLAG_AUTHORIZED))
1199                         ret = sta_info_move_state(sta, IEEE80211_STA_AUTHORIZED);
1200                 else if (test_sta_flag(sta, WLAN_STA_AUTHORIZED))
1201                         ret = sta_info_move_state(sta, IEEE80211_STA_ASSOC);
1202                 else
1203                         ret = 0;
1204                 if (ret)
1205                         return ret;
1206         }
1207
1208         if (mask & BIT(NL80211_STA_FLAG_ASSOCIATED) &&
1209             !(set & BIT(NL80211_STA_FLAG_ASSOCIATED)) &&
1210             test_sta_flag(sta, WLAN_STA_ASSOC)) {
1211                 ret = sta_info_move_state(sta, IEEE80211_STA_AUTH);
1212                 if (ret)
1213                         return ret;
1214         }
1215
1216         if (mask & BIT(NL80211_STA_FLAG_AUTHENTICATED) &&
1217             !(set & BIT(NL80211_STA_FLAG_AUTHENTICATED)) &&
1218             test_sta_flag(sta, WLAN_STA_AUTH)) {
1219                 ret = sta_info_move_state(sta, IEEE80211_STA_NONE);
1220                 if (ret)
1221                         return ret;
1222         }
1223
1224         return 0;
1225 }
1226
1227 static int sta_apply_parameters(struct ieee80211_local *local,
1228                                 struct sta_info *sta,
1229                                 struct station_parameters *params)
1230 {
1231         int ret = 0;
1232         struct ieee80211_supported_band *sband;
1233         struct ieee80211_sub_if_data *sdata = sta->sdata;
1234         enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
1235         u32 mask, set;
1236
1237         sband = local->hw.wiphy->bands[band];
1238
1239         mask = params->sta_flags_mask;
1240         set = params->sta_flags_set;
1241
1242         if (ieee80211_vif_is_mesh(&sdata->vif)) {
1243                 /*
1244                  * In mesh mode, ASSOCIATED isn't part of the nl80211
1245                  * API but must follow AUTHENTICATED for driver state.
1246                  */
1247                 if (mask & BIT(NL80211_STA_FLAG_AUTHENTICATED))
1248                         mask |= BIT(NL80211_STA_FLAG_ASSOCIATED);
1249                 if (set & BIT(NL80211_STA_FLAG_AUTHENTICATED))
1250                         set |= BIT(NL80211_STA_FLAG_ASSOCIATED);
1251         } else if (test_sta_flag(sta, WLAN_STA_TDLS_PEER)) {
1252                 /*
1253                  * TDLS -- everything follows authorized, but
1254                  * only becoming authorized is possible, not
1255                  * going back
1256                  */
1257                 if (set & BIT(NL80211_STA_FLAG_AUTHORIZED)) {
1258                         set |= BIT(NL80211_STA_FLAG_AUTHENTICATED) |
1259                                BIT(NL80211_STA_FLAG_ASSOCIATED);
1260                         mask |= BIT(NL80211_STA_FLAG_AUTHENTICATED) |
1261                                 BIT(NL80211_STA_FLAG_ASSOCIATED);
1262                 }
1263         }
1264
1265         ret = sta_apply_auth_flags(local, sta, mask, set);
1266         if (ret)
1267                 return ret;
1268
1269         if (mask & BIT(NL80211_STA_FLAG_SHORT_PREAMBLE)) {
1270                 if (set & BIT(NL80211_STA_FLAG_SHORT_PREAMBLE))
1271                         set_sta_flag(sta, WLAN_STA_SHORT_PREAMBLE);
1272                 else
1273                         clear_sta_flag(sta, WLAN_STA_SHORT_PREAMBLE);
1274         }
1275
1276         if (mask & BIT(NL80211_STA_FLAG_WME)) {
1277                 if (set & BIT(NL80211_STA_FLAG_WME)) {
1278                         set_sta_flag(sta, WLAN_STA_WME);
1279                         sta->sta.wme = true;
1280                 } else {
1281                         clear_sta_flag(sta, WLAN_STA_WME);
1282                         sta->sta.wme = false;
1283                 }
1284         }
1285
1286         if (mask & BIT(NL80211_STA_FLAG_MFP)) {
1287                 if (set & BIT(NL80211_STA_FLAG_MFP))
1288                         set_sta_flag(sta, WLAN_STA_MFP);
1289                 else
1290                         clear_sta_flag(sta, WLAN_STA_MFP);
1291         }
1292
1293         if (mask & BIT(NL80211_STA_FLAG_TDLS_PEER)) {
1294                 if (set & BIT(NL80211_STA_FLAG_TDLS_PEER))
1295                         set_sta_flag(sta, WLAN_STA_TDLS_PEER);
1296                 else
1297                         clear_sta_flag(sta, WLAN_STA_TDLS_PEER);
1298         }
1299
1300         if (params->sta_modify_mask & STATION_PARAM_APPLY_UAPSD) {
1301                 sta->sta.uapsd_queues = params->uapsd_queues;
1302                 sta->sta.max_sp = params->max_sp;
1303         }
1304
1305         /*
1306          * cfg80211 validates this (1-2007) and allows setting the AID
1307          * only when creating a new station entry
1308          */
1309         if (params->aid)
1310                 sta->sta.aid = params->aid;
1311
1312         /*
1313          * Some of the following updates would be racy if called on an
1314          * existing station, via ieee80211_change_station(). However,
1315          * all such changes are rejected by cfg80211 except for updates
1316          * changing the supported rates on an existing but not yet used
1317          * TDLS peer.
1318          */
1319
1320         if (params->listen_interval >= 0)
1321                 sta->listen_interval = params->listen_interval;
1322
1323         if (params->supported_rates) {
1324                 ieee80211_parse_bitrates(&sdata->vif.bss_conf.chandef,
1325                                          sband, params->supported_rates,
1326                                          params->supported_rates_len,
1327                                          &sta->sta.supp_rates[band]);
1328         }
1329
1330         if (params->ht_capa)
1331                 ieee80211_ht_cap_ie_to_sta_ht_cap(sdata, sband,
1332                                                   params->ht_capa, sta);
1333
1334         if (params->vht_capa)
1335                 ieee80211_vht_cap_ie_to_sta_vht_cap(sdata, sband,
1336                                                     params->vht_capa, sta);
1337
1338         if (ieee80211_vif_is_mesh(&sdata->vif)) {
1339 #ifdef CONFIG_MAC80211_MESH
1340                 u32 changed = 0;
1341
1342                 if (params->sta_modify_mask & STATION_PARAM_APPLY_PLINK_STATE) {
1343                         switch (params->plink_state) {
1344                         case NL80211_PLINK_ESTAB:
1345                                 if (sta->plink_state != NL80211_PLINK_ESTAB)
1346                                         changed = mesh_plink_inc_estab_count(
1347                                                         sdata);
1348                                 sta->plink_state = params->plink_state;
1349
1350                                 ieee80211_mps_sta_status_update(sta);
1351                                 changed |= ieee80211_mps_set_sta_local_pm(sta,
1352                                               sdata->u.mesh.mshcfg.power_mode);
1353                                 break;
1354                         case NL80211_PLINK_LISTEN:
1355                         case NL80211_PLINK_BLOCKED:
1356                         case NL80211_PLINK_OPN_SNT:
1357                         case NL80211_PLINK_OPN_RCVD:
1358                         case NL80211_PLINK_CNF_RCVD:
1359                         case NL80211_PLINK_HOLDING:
1360                                 if (sta->plink_state == NL80211_PLINK_ESTAB)
1361                                         changed = mesh_plink_dec_estab_count(
1362                                                         sdata);
1363                                 sta->plink_state = params->plink_state;
1364
1365                                 ieee80211_mps_sta_status_update(sta);
1366                                 changed |= ieee80211_mps_set_sta_local_pm(sta,
1367                                                 NL80211_MESH_POWER_UNKNOWN);
1368                                 break;
1369                         default:
1370                                 /*  nothing  */
1371                                 break;
1372                         }
1373                 }
1374
1375                 switch (params->plink_action) {
1376                 case NL80211_PLINK_ACTION_NO_ACTION:
1377                         /* nothing */
1378                         break;
1379                 case NL80211_PLINK_ACTION_OPEN:
1380                         changed |= mesh_plink_open(sta);
1381                         break;
1382                 case NL80211_PLINK_ACTION_BLOCK:
1383                         changed |= mesh_plink_block(sta);
1384                         break;
1385                 }
1386
1387                 if (params->local_pm)
1388                         changed |=
1389                               ieee80211_mps_set_sta_local_pm(sta,
1390                                                              params->local_pm);
1391                 ieee80211_mbss_info_change_notify(sdata, changed);
1392 #endif
1393         }
1394
1395         return 0;
1396 }
1397
1398 static int ieee80211_add_station(struct wiphy *wiphy, struct net_device *dev,
1399                                  u8 *mac, struct station_parameters *params)
1400 {
1401         struct ieee80211_local *local = wiphy_priv(wiphy);
1402         struct sta_info *sta;
1403         struct ieee80211_sub_if_data *sdata;
1404         int err;
1405         int layer2_update;
1406
1407         if (params->vlan) {
1408                 sdata = IEEE80211_DEV_TO_SUB_IF(params->vlan);
1409
1410                 if (sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
1411                     sdata->vif.type != NL80211_IFTYPE_AP)
1412                         return -EINVAL;
1413         } else
1414                 sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1415
1416         if (ether_addr_equal(mac, sdata->vif.addr))
1417                 return -EINVAL;
1418
1419         if (is_multicast_ether_addr(mac))
1420                 return -EINVAL;
1421
1422         sta = sta_info_alloc(sdata, mac, GFP_KERNEL);
1423         if (!sta)
1424                 return -ENOMEM;
1425
1426         /*
1427          * defaults -- if userspace wants something else we'll
1428          * change it accordingly in sta_apply_parameters()
1429          */
1430         if (!(params->sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER))) {
1431                 sta_info_pre_move_state(sta, IEEE80211_STA_AUTH);
1432                 sta_info_pre_move_state(sta, IEEE80211_STA_ASSOC);
1433         }
1434
1435         err = sta_apply_parameters(local, sta, params);
1436         if (err) {
1437                 sta_info_free(local, sta);
1438                 return err;
1439         }
1440
1441         /*
1442          * for TDLS, rate control should be initialized only when
1443          * rates are known and station is marked authorized
1444          */
1445         if (!test_sta_flag(sta, WLAN_STA_TDLS_PEER))
1446                 rate_control_rate_init(sta);
1447
1448         layer2_update = sdata->vif.type == NL80211_IFTYPE_AP_VLAN ||
1449                 sdata->vif.type == NL80211_IFTYPE_AP;
1450
1451         err = sta_info_insert_rcu(sta);
1452         if (err) {
1453                 rcu_read_unlock();
1454                 return err;
1455         }
1456
1457         if (layer2_update)
1458                 ieee80211_send_layer2_update(sta);
1459
1460         rcu_read_unlock();
1461
1462         return 0;
1463 }
1464
1465 static int ieee80211_del_station(struct wiphy *wiphy, struct net_device *dev,
1466                                  u8 *mac)
1467 {
1468         struct ieee80211_sub_if_data *sdata;
1469
1470         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1471
1472         if (mac)
1473                 return sta_info_destroy_addr_bss(sdata, mac);
1474
1475         sta_info_flush(sdata);
1476         return 0;
1477 }
1478
1479 static int ieee80211_change_station(struct wiphy *wiphy,
1480                                     struct net_device *dev, u8 *mac,
1481                                     struct station_parameters *params)
1482 {
1483         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1484         struct ieee80211_local *local = wiphy_priv(wiphy);
1485         struct sta_info *sta;
1486         struct ieee80211_sub_if_data *vlansdata;
1487         enum cfg80211_station_type statype;
1488         int err;
1489
1490         mutex_lock(&local->sta_mtx);
1491
1492         sta = sta_info_get_bss(sdata, mac);
1493         if (!sta) {
1494                 err = -ENOENT;
1495                 goto out_err;
1496         }
1497
1498         switch (sdata->vif.type) {
1499         case NL80211_IFTYPE_MESH_POINT:
1500                 if (sdata->u.mesh.user_mpm)
1501                         statype = CFG80211_STA_MESH_PEER_USER;
1502                 else
1503                         statype = CFG80211_STA_MESH_PEER_KERNEL;
1504                 break;
1505         case NL80211_IFTYPE_ADHOC:
1506                 statype = CFG80211_STA_IBSS;
1507                 break;
1508         case NL80211_IFTYPE_STATION:
1509                 if (!test_sta_flag(sta, WLAN_STA_TDLS_PEER)) {
1510                         statype = CFG80211_STA_AP_STA;
1511                         break;
1512                 }
1513                 if (test_sta_flag(sta, WLAN_STA_AUTHORIZED))
1514                         statype = CFG80211_STA_TDLS_PEER_ACTIVE;
1515                 else
1516                         statype = CFG80211_STA_TDLS_PEER_SETUP;
1517                 break;
1518         case NL80211_IFTYPE_AP:
1519         case NL80211_IFTYPE_AP_VLAN:
1520                 statype = CFG80211_STA_AP_CLIENT;
1521                 break;
1522         default:
1523                 err = -EOPNOTSUPP;
1524                 goto out_err;
1525         }
1526
1527         err = cfg80211_check_station_change(wiphy, params, statype);
1528         if (err)
1529                 goto out_err;
1530
1531         if (params->vlan && params->vlan != sta->sdata->dev) {
1532                 bool prev_4addr = false;
1533                 bool new_4addr = false;
1534
1535                 vlansdata = IEEE80211_DEV_TO_SUB_IF(params->vlan);
1536
1537                 if (params->vlan->ieee80211_ptr->use_4addr) {
1538                         if (vlansdata->u.vlan.sta) {
1539                                 err = -EBUSY;
1540                                 goto out_err;
1541                         }
1542
1543                         rcu_assign_pointer(vlansdata->u.vlan.sta, sta);
1544                         new_4addr = true;
1545                 }
1546
1547                 if (sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN &&
1548                     sta->sdata->u.vlan.sta) {
1549                         rcu_assign_pointer(sta->sdata->u.vlan.sta, NULL);
1550                         prev_4addr = true;
1551                 }
1552
1553                 sta->sdata = vlansdata;
1554
1555                 if (sta->sta_state == IEEE80211_STA_AUTHORIZED &&
1556                     prev_4addr != new_4addr) {
1557                         if (new_4addr)
1558                                 atomic_dec(&sta->sdata->bss->num_mcast_sta);
1559                         else
1560                                 atomic_inc(&sta->sdata->bss->num_mcast_sta);
1561                 }
1562
1563                 ieee80211_send_layer2_update(sta);
1564         }
1565
1566         err = sta_apply_parameters(local, sta, params);
1567         if (err)
1568                 goto out_err;
1569
1570         /* When peer becomes authorized, init rate control as well */
1571         if (test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
1572             test_sta_flag(sta, WLAN_STA_AUTHORIZED))
1573                 rate_control_rate_init(sta);
1574
1575         mutex_unlock(&local->sta_mtx);
1576
1577         if ((sdata->vif.type == NL80211_IFTYPE_AP ||
1578              sdata->vif.type == NL80211_IFTYPE_AP_VLAN) &&
1579             sta->known_smps_mode != sta->sdata->bss->req_smps &&
1580             test_sta_flag(sta, WLAN_STA_AUTHORIZED) &&
1581             sta_info_tx_streams(sta) != 1) {
1582                 ht_dbg(sta->sdata,
1583                        "%pM just authorized and MIMO capable - update SMPS\n",
1584                        sta->sta.addr);
1585                 ieee80211_send_smps_action(sta->sdata,
1586                         sta->sdata->bss->req_smps,
1587                         sta->sta.addr,
1588                         sta->sdata->vif.bss_conf.bssid);
1589         }
1590
1591         if (sdata->vif.type == NL80211_IFTYPE_STATION &&
1592             params->sta_flags_mask & BIT(NL80211_STA_FLAG_AUTHORIZED)) {
1593                 ieee80211_recalc_ps(local, -1);
1594                 ieee80211_recalc_ps_vif(sdata);
1595         }
1596
1597         return 0;
1598 out_err:
1599         mutex_unlock(&local->sta_mtx);
1600         return err;
1601 }
1602
1603 #ifdef CONFIG_MAC80211_MESH
1604 static int ieee80211_add_mpath(struct wiphy *wiphy, struct net_device *dev,
1605                                  u8 *dst, u8 *next_hop)
1606 {
1607         struct ieee80211_sub_if_data *sdata;
1608         struct mesh_path *mpath;
1609         struct sta_info *sta;
1610
1611         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1612
1613         rcu_read_lock();
1614         sta = sta_info_get(sdata, next_hop);
1615         if (!sta) {
1616                 rcu_read_unlock();
1617                 return -ENOENT;
1618         }
1619
1620         mpath = mesh_path_add(sdata, dst);
1621         if (IS_ERR(mpath)) {
1622                 rcu_read_unlock();
1623                 return PTR_ERR(mpath);
1624         }
1625
1626         mesh_path_fix_nexthop(mpath, sta);
1627
1628         rcu_read_unlock();
1629         return 0;
1630 }
1631
1632 static int ieee80211_del_mpath(struct wiphy *wiphy, struct net_device *dev,
1633                                u8 *dst)
1634 {
1635         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1636
1637         if (dst)
1638                 return mesh_path_del(sdata, dst);
1639
1640         mesh_path_flush_by_iface(sdata);
1641         return 0;
1642 }
1643
1644 static int ieee80211_change_mpath(struct wiphy *wiphy,
1645                                     struct net_device *dev,
1646                                     u8 *dst, u8 *next_hop)
1647 {
1648         struct ieee80211_sub_if_data *sdata;
1649         struct mesh_path *mpath;
1650         struct sta_info *sta;
1651
1652         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1653
1654         rcu_read_lock();
1655
1656         sta = sta_info_get(sdata, next_hop);
1657         if (!sta) {
1658                 rcu_read_unlock();
1659                 return -ENOENT;
1660         }
1661
1662         mpath = mesh_path_lookup(sdata, dst);
1663         if (!mpath) {
1664                 rcu_read_unlock();
1665                 return -ENOENT;
1666         }
1667
1668         mesh_path_fix_nexthop(mpath, sta);
1669
1670         rcu_read_unlock();
1671         return 0;
1672 }
1673
1674 static void mpath_set_pinfo(struct mesh_path *mpath, u8 *next_hop,
1675                             struct mpath_info *pinfo)
1676 {
1677         struct sta_info *next_hop_sta = rcu_dereference(mpath->next_hop);
1678
1679         if (next_hop_sta)
1680                 memcpy(next_hop, next_hop_sta->sta.addr, ETH_ALEN);
1681         else
1682                 memset(next_hop, 0, ETH_ALEN);
1683
1684         memset(pinfo, 0, sizeof(*pinfo));
1685
1686         pinfo->generation = mesh_paths_generation;
1687
1688         pinfo->filled = MPATH_INFO_FRAME_QLEN |
1689                         MPATH_INFO_SN |
1690                         MPATH_INFO_METRIC |
1691                         MPATH_INFO_EXPTIME |
1692                         MPATH_INFO_DISCOVERY_TIMEOUT |
1693                         MPATH_INFO_DISCOVERY_RETRIES |
1694                         MPATH_INFO_FLAGS;
1695
1696         pinfo->frame_qlen = mpath->frame_queue.qlen;
1697         pinfo->sn = mpath->sn;
1698         pinfo->metric = mpath->metric;
1699         if (time_before(jiffies, mpath->exp_time))
1700                 pinfo->exptime = jiffies_to_msecs(mpath->exp_time - jiffies);
1701         pinfo->discovery_timeout =
1702                         jiffies_to_msecs(mpath->discovery_timeout);
1703         pinfo->discovery_retries = mpath->discovery_retries;
1704         if (mpath->flags & MESH_PATH_ACTIVE)
1705                 pinfo->flags |= NL80211_MPATH_FLAG_ACTIVE;
1706         if (mpath->flags & MESH_PATH_RESOLVING)
1707                 pinfo->flags |= NL80211_MPATH_FLAG_RESOLVING;
1708         if (mpath->flags & MESH_PATH_SN_VALID)
1709                 pinfo->flags |= NL80211_MPATH_FLAG_SN_VALID;
1710         if (mpath->flags & MESH_PATH_FIXED)
1711                 pinfo->flags |= NL80211_MPATH_FLAG_FIXED;
1712         if (mpath->flags & MESH_PATH_RESOLVED)
1713                 pinfo->flags |= NL80211_MPATH_FLAG_RESOLVED;
1714 }
1715
1716 static int ieee80211_get_mpath(struct wiphy *wiphy, struct net_device *dev,
1717                                u8 *dst, u8 *next_hop, struct mpath_info *pinfo)
1718
1719 {
1720         struct ieee80211_sub_if_data *sdata;
1721         struct mesh_path *mpath;
1722
1723         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1724
1725         rcu_read_lock();
1726         mpath = mesh_path_lookup(sdata, dst);
1727         if (!mpath) {
1728                 rcu_read_unlock();
1729                 return -ENOENT;
1730         }
1731         memcpy(dst, mpath->dst, ETH_ALEN);
1732         mpath_set_pinfo(mpath, next_hop, pinfo);
1733         rcu_read_unlock();
1734         return 0;
1735 }
1736
1737 static int ieee80211_dump_mpath(struct wiphy *wiphy, struct net_device *dev,
1738                                  int idx, u8 *dst, u8 *next_hop,
1739                                  struct mpath_info *pinfo)
1740 {
1741         struct ieee80211_sub_if_data *sdata;
1742         struct mesh_path *mpath;
1743
1744         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1745
1746         rcu_read_lock();
1747         mpath = mesh_path_lookup_by_idx(sdata, idx);
1748         if (!mpath) {
1749                 rcu_read_unlock();
1750                 return -ENOENT;
1751         }
1752         memcpy(dst, mpath->dst, ETH_ALEN);
1753         mpath_set_pinfo(mpath, next_hop, pinfo);
1754         rcu_read_unlock();
1755         return 0;
1756 }
1757
1758 static int ieee80211_get_mesh_config(struct wiphy *wiphy,
1759                                 struct net_device *dev,
1760                                 struct mesh_config *conf)
1761 {
1762         struct ieee80211_sub_if_data *sdata;
1763         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1764
1765         memcpy(conf, &(sdata->u.mesh.mshcfg), sizeof(struct mesh_config));
1766         return 0;
1767 }
1768
1769 static inline bool _chg_mesh_attr(enum nl80211_meshconf_params parm, u32 mask)
1770 {
1771         return (mask >> (parm-1)) & 0x1;
1772 }
1773
1774 static int copy_mesh_setup(struct ieee80211_if_mesh *ifmsh,
1775                 const struct mesh_setup *setup)
1776 {
1777         u8 *new_ie;
1778         const u8 *old_ie;
1779         struct ieee80211_sub_if_data *sdata = container_of(ifmsh,
1780                                         struct ieee80211_sub_if_data, u.mesh);
1781
1782         /* allocate information elements */
1783         new_ie = NULL;
1784         old_ie = ifmsh->ie;
1785
1786         if (setup->ie_len) {
1787                 new_ie = kmemdup(setup->ie, setup->ie_len,
1788                                 GFP_KERNEL);
1789                 if (!new_ie)
1790                         return -ENOMEM;
1791         }
1792         ifmsh->ie_len = setup->ie_len;
1793         ifmsh->ie = new_ie;
1794         kfree(old_ie);
1795
1796         /* now copy the rest of the setup parameters */
1797         ifmsh->mesh_id_len = setup->mesh_id_len;
1798         memcpy(ifmsh->mesh_id, setup->mesh_id, ifmsh->mesh_id_len);
1799         ifmsh->mesh_sp_id = setup->sync_method;
1800         ifmsh->mesh_pp_id = setup->path_sel_proto;
1801         ifmsh->mesh_pm_id = setup->path_metric;
1802         ifmsh->user_mpm = setup->user_mpm;
1803         ifmsh->mesh_auth_id = setup->auth_id;
1804         ifmsh->security = IEEE80211_MESH_SEC_NONE;
1805         if (setup->is_authenticated)
1806                 ifmsh->security |= IEEE80211_MESH_SEC_AUTHED;
1807         if (setup->is_secure)
1808                 ifmsh->security |= IEEE80211_MESH_SEC_SECURED;
1809
1810         /* mcast rate setting in Mesh Node */
1811         memcpy(sdata->vif.bss_conf.mcast_rate, setup->mcast_rate,
1812                                                 sizeof(setup->mcast_rate));
1813         sdata->vif.bss_conf.basic_rates = setup->basic_rates;
1814
1815         sdata->vif.bss_conf.beacon_int = setup->beacon_interval;
1816         sdata->vif.bss_conf.dtim_period = setup->dtim_period;
1817
1818         return 0;
1819 }
1820
1821 static int ieee80211_update_mesh_config(struct wiphy *wiphy,
1822                                         struct net_device *dev, u32 mask,
1823                                         const struct mesh_config *nconf)
1824 {
1825         struct mesh_config *conf;
1826         struct ieee80211_sub_if_data *sdata;
1827         struct ieee80211_if_mesh *ifmsh;
1828
1829         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1830         ifmsh = &sdata->u.mesh;
1831
1832         /* Set the config options which we are interested in setting */
1833         conf = &(sdata->u.mesh.mshcfg);
1834         if (_chg_mesh_attr(NL80211_MESHCONF_RETRY_TIMEOUT, mask))
1835                 conf->dot11MeshRetryTimeout = nconf->dot11MeshRetryTimeout;
1836         if (_chg_mesh_attr(NL80211_MESHCONF_CONFIRM_TIMEOUT, mask))
1837                 conf->dot11MeshConfirmTimeout = nconf->dot11MeshConfirmTimeout;
1838         if (_chg_mesh_attr(NL80211_MESHCONF_HOLDING_TIMEOUT, mask))
1839                 conf->dot11MeshHoldingTimeout = nconf->dot11MeshHoldingTimeout;
1840         if (_chg_mesh_attr(NL80211_MESHCONF_MAX_PEER_LINKS, mask))
1841                 conf->dot11MeshMaxPeerLinks = nconf->dot11MeshMaxPeerLinks;
1842         if (_chg_mesh_attr(NL80211_MESHCONF_MAX_RETRIES, mask))
1843                 conf->dot11MeshMaxRetries = nconf->dot11MeshMaxRetries;
1844         if (_chg_mesh_attr(NL80211_MESHCONF_TTL, mask))
1845                 conf->dot11MeshTTL = nconf->dot11MeshTTL;
1846         if (_chg_mesh_attr(NL80211_MESHCONF_ELEMENT_TTL, mask))
1847                 conf->element_ttl = nconf->element_ttl;
1848         if (_chg_mesh_attr(NL80211_MESHCONF_AUTO_OPEN_PLINKS, mask)) {
1849                 if (ifmsh->user_mpm)
1850                         return -EBUSY;
1851                 conf->auto_open_plinks = nconf->auto_open_plinks;
1852         }
1853         if (_chg_mesh_attr(NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR, mask))
1854                 conf->dot11MeshNbrOffsetMaxNeighbor =
1855                         nconf->dot11MeshNbrOffsetMaxNeighbor;
1856         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES, mask))
1857                 conf->dot11MeshHWMPmaxPREQretries =
1858                         nconf->dot11MeshHWMPmaxPREQretries;
1859         if (_chg_mesh_attr(NL80211_MESHCONF_PATH_REFRESH_TIME, mask))
1860                 conf->path_refresh_time = nconf->path_refresh_time;
1861         if (_chg_mesh_attr(NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT, mask))
1862                 conf->min_discovery_timeout = nconf->min_discovery_timeout;
1863         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT, mask))
1864                 conf->dot11MeshHWMPactivePathTimeout =
1865                         nconf->dot11MeshHWMPactivePathTimeout;
1866         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL, mask))
1867                 conf->dot11MeshHWMPpreqMinInterval =
1868                         nconf->dot11MeshHWMPpreqMinInterval;
1869         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL, mask))
1870                 conf->dot11MeshHWMPperrMinInterval =
1871                         nconf->dot11MeshHWMPperrMinInterval;
1872         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
1873                            mask))
1874                 conf->dot11MeshHWMPnetDiameterTraversalTime =
1875                         nconf->dot11MeshHWMPnetDiameterTraversalTime;
1876         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_ROOTMODE, mask)) {
1877                 conf->dot11MeshHWMPRootMode = nconf->dot11MeshHWMPRootMode;
1878                 ieee80211_mesh_root_setup(ifmsh);
1879         }
1880         if (_chg_mesh_attr(NL80211_MESHCONF_GATE_ANNOUNCEMENTS, mask)) {
1881                 /* our current gate announcement implementation rides on root
1882                  * announcements, so require this ifmsh to also be a root node
1883                  * */
1884                 if (nconf->dot11MeshGateAnnouncementProtocol &&
1885                     !(conf->dot11MeshHWMPRootMode > IEEE80211_ROOTMODE_ROOT)) {
1886                         conf->dot11MeshHWMPRootMode = IEEE80211_PROACTIVE_RANN;
1887                         ieee80211_mesh_root_setup(ifmsh);
1888                 }
1889                 conf->dot11MeshGateAnnouncementProtocol =
1890                         nconf->dot11MeshGateAnnouncementProtocol;
1891         }
1892         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_RANN_INTERVAL, mask))
1893                 conf->dot11MeshHWMPRannInterval =
1894                         nconf->dot11MeshHWMPRannInterval;
1895         if (_chg_mesh_attr(NL80211_MESHCONF_FORWARDING, mask))
1896                 conf->dot11MeshForwarding = nconf->dot11MeshForwarding;
1897         if (_chg_mesh_attr(NL80211_MESHCONF_RSSI_THRESHOLD, mask)) {
1898                 /* our RSSI threshold implementation is supported only for
1899                  * devices that report signal in dBm.
1900                  */
1901                 if (!(sdata->local->hw.flags & IEEE80211_HW_SIGNAL_DBM))
1902                         return -ENOTSUPP;
1903                 conf->rssi_threshold = nconf->rssi_threshold;
1904         }
1905         if (_chg_mesh_attr(NL80211_MESHCONF_HT_OPMODE, mask)) {
1906                 conf->ht_opmode = nconf->ht_opmode;
1907                 sdata->vif.bss_conf.ht_operation_mode = nconf->ht_opmode;
1908                 ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_HT);
1909         }
1910         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT, mask))
1911                 conf->dot11MeshHWMPactivePathToRootTimeout =
1912                         nconf->dot11MeshHWMPactivePathToRootTimeout;
1913         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_ROOT_INTERVAL, mask))
1914                 conf->dot11MeshHWMProotInterval =
1915                         nconf->dot11MeshHWMProotInterval;
1916         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL, mask))
1917                 conf->dot11MeshHWMPconfirmationInterval =
1918                         nconf->dot11MeshHWMPconfirmationInterval;
1919         if (_chg_mesh_attr(NL80211_MESHCONF_POWER_MODE, mask)) {
1920                 conf->power_mode = nconf->power_mode;
1921                 ieee80211_mps_local_status_update(sdata);
1922         }
1923         if (_chg_mesh_attr(NL80211_MESHCONF_AWAKE_WINDOW, mask))
1924                 conf->dot11MeshAwakeWindowDuration =
1925                         nconf->dot11MeshAwakeWindowDuration;
1926         if (_chg_mesh_attr(NL80211_MESHCONF_PLINK_TIMEOUT, mask))
1927                 conf->plink_timeout = nconf->plink_timeout;
1928         ieee80211_mbss_info_change_notify(sdata, BSS_CHANGED_BEACON);
1929         return 0;
1930 }
1931
1932 static int ieee80211_join_mesh(struct wiphy *wiphy, struct net_device *dev,
1933                                const struct mesh_config *conf,
1934                                const struct mesh_setup *setup)
1935 {
1936         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1937         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1938         int err;
1939
1940         memcpy(&ifmsh->mshcfg, conf, sizeof(struct mesh_config));
1941         err = copy_mesh_setup(ifmsh, setup);
1942         if (err)
1943                 return err;
1944
1945         /* can mesh use other SMPS modes? */
1946         sdata->smps_mode = IEEE80211_SMPS_OFF;
1947         sdata->needed_rx_chains = sdata->local->rx_chains;
1948
1949         err = ieee80211_vif_use_channel(sdata, &setup->chandef,
1950                                         IEEE80211_CHANCTX_SHARED);
1951         if (err)
1952                 return err;
1953
1954         return ieee80211_start_mesh(sdata);
1955 }
1956
1957 static int ieee80211_leave_mesh(struct wiphy *wiphy, struct net_device *dev)
1958 {
1959         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1960
1961         ieee80211_stop_mesh(sdata);
1962         ieee80211_vif_release_channel(sdata);
1963
1964         return 0;
1965 }
1966 #endif
1967
1968 static int ieee80211_change_bss(struct wiphy *wiphy,
1969                                 struct net_device *dev,
1970                                 struct bss_parameters *params)
1971 {
1972         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1973         enum ieee80211_band band;
1974         u32 changed = 0;
1975
1976         if (!sdata_dereference(sdata->u.ap.beacon, sdata))
1977                 return -ENOENT;
1978
1979         band = ieee80211_get_sdata_band(sdata);
1980
1981         if (params->use_cts_prot >= 0) {
1982                 sdata->vif.bss_conf.use_cts_prot = params->use_cts_prot;
1983                 changed |= BSS_CHANGED_ERP_CTS_PROT;
1984         }
1985         if (params->use_short_preamble >= 0) {
1986                 sdata->vif.bss_conf.use_short_preamble =
1987                         params->use_short_preamble;
1988                 changed |= BSS_CHANGED_ERP_PREAMBLE;
1989         }
1990
1991         if (!sdata->vif.bss_conf.use_short_slot &&
1992             band == IEEE80211_BAND_5GHZ) {
1993                 sdata->vif.bss_conf.use_short_slot = true;
1994                 changed |= BSS_CHANGED_ERP_SLOT;
1995         }
1996
1997         if (params->use_short_slot_time >= 0) {
1998                 sdata->vif.bss_conf.use_short_slot =
1999                         params->use_short_slot_time;
2000                 changed |= BSS_CHANGED_ERP_SLOT;
2001         }
2002
2003         if (params->basic_rates) {
2004                 ieee80211_parse_bitrates(&sdata->vif.bss_conf.chandef,
2005                                          wiphy->bands[band],
2006                                          params->basic_rates,
2007                                          params->basic_rates_len,
2008                                          &sdata->vif.bss_conf.basic_rates);
2009                 changed |= BSS_CHANGED_BASIC_RATES;
2010         }
2011
2012         if (params->ap_isolate >= 0) {
2013                 if (params->ap_isolate)
2014                         sdata->flags |= IEEE80211_SDATA_DONT_BRIDGE_PACKETS;
2015                 else
2016                         sdata->flags &= ~IEEE80211_SDATA_DONT_BRIDGE_PACKETS;
2017         }
2018
2019         if (params->ht_opmode >= 0) {
2020                 sdata->vif.bss_conf.ht_operation_mode =
2021                         (u16) params->ht_opmode;
2022                 changed |= BSS_CHANGED_HT;
2023         }
2024
2025         if (params->p2p_ctwindow >= 0) {
2026                 sdata->vif.bss_conf.p2p_noa_attr.oppps_ctwindow &=
2027                                         ~IEEE80211_P2P_OPPPS_CTWINDOW_MASK;
2028                 sdata->vif.bss_conf.p2p_noa_attr.oppps_ctwindow |=
2029                         params->p2p_ctwindow & IEEE80211_P2P_OPPPS_CTWINDOW_MASK;
2030                 changed |= BSS_CHANGED_P2P_PS;
2031         }
2032
2033         if (params->p2p_opp_ps > 0) {
2034                 sdata->vif.bss_conf.p2p_noa_attr.oppps_ctwindow |=
2035                                         IEEE80211_P2P_OPPPS_ENABLE_BIT;
2036                 changed |= BSS_CHANGED_P2P_PS;
2037         } else if (params->p2p_opp_ps == 0) {
2038                 sdata->vif.bss_conf.p2p_noa_attr.oppps_ctwindow &=
2039                                         ~IEEE80211_P2P_OPPPS_ENABLE_BIT;
2040                 changed |= BSS_CHANGED_P2P_PS;
2041         }
2042
2043         ieee80211_bss_info_change_notify(sdata, changed);
2044
2045         return 0;
2046 }
2047
2048 static int ieee80211_set_txq_params(struct wiphy *wiphy,
2049                                     struct net_device *dev,
2050                                     struct ieee80211_txq_params *params)
2051 {
2052         struct ieee80211_local *local = wiphy_priv(wiphy);
2053         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2054         struct ieee80211_tx_queue_params p;
2055
2056         if (!local->ops->conf_tx)
2057                 return -EOPNOTSUPP;
2058
2059         if (local->hw.queues < IEEE80211_NUM_ACS)
2060                 return -EOPNOTSUPP;
2061
2062         memset(&p, 0, sizeof(p));
2063         p.aifs = params->aifs;
2064         p.cw_max = params->cwmax;
2065         p.cw_min = params->cwmin;
2066         p.txop = params->txop;
2067
2068         /*
2069          * Setting tx queue params disables u-apsd because it's only
2070          * called in master mode.
2071          */
2072         p.uapsd = false;
2073
2074         sdata->tx_conf[params->ac] = p;
2075         if (drv_conf_tx(local, sdata, params->ac, &p)) {
2076                 wiphy_debug(local->hw.wiphy,
2077                             "failed to set TX queue parameters for AC %d\n",
2078                             params->ac);
2079                 return -EINVAL;
2080         }
2081
2082         ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_QOS);
2083
2084         return 0;
2085 }
2086
2087 #ifdef CONFIG_PM
2088 static int ieee80211_suspend(struct wiphy *wiphy,
2089                              struct cfg80211_wowlan *wowlan)
2090 {
2091         return __ieee80211_suspend(wiphy_priv(wiphy), wowlan);
2092 }
2093
2094 static int ieee80211_resume(struct wiphy *wiphy)
2095 {
2096         return __ieee80211_resume(wiphy_priv(wiphy));
2097 }
2098 #else
2099 #define ieee80211_suspend NULL
2100 #define ieee80211_resume NULL
2101 #endif
2102
2103 static int ieee80211_scan(struct wiphy *wiphy,
2104                           struct cfg80211_scan_request *req)
2105 {
2106         struct ieee80211_sub_if_data *sdata;
2107
2108         sdata = IEEE80211_WDEV_TO_SUB_IF(req->wdev);
2109
2110         switch (ieee80211_vif_type_p2p(&sdata->vif)) {
2111         case NL80211_IFTYPE_STATION:
2112         case NL80211_IFTYPE_ADHOC:
2113         case NL80211_IFTYPE_MESH_POINT:
2114         case NL80211_IFTYPE_P2P_CLIENT:
2115         case NL80211_IFTYPE_P2P_DEVICE:
2116                 break;
2117         case NL80211_IFTYPE_P2P_GO:
2118                 if (sdata->local->ops->hw_scan)
2119                         break;
2120                 /*
2121                  * FIXME: implement NoA while scanning in software,
2122                  * for now fall through to allow scanning only when
2123                  * beaconing hasn't been configured yet
2124                  */
2125         case NL80211_IFTYPE_AP:
2126                 /*
2127                  * If the scan has been forced (and the driver supports
2128                  * forcing), don't care about being beaconing already.
2129                  * This will create problems to the attached stations (e.g. all
2130                  * the  frames sent while scanning on other channel will be
2131                  * lost)
2132                  */
2133                 if (sdata->u.ap.beacon &&
2134                     (!(wiphy->features & NL80211_FEATURE_AP_SCAN) ||
2135                      !(req->flags & NL80211_SCAN_FLAG_AP)))
2136                         return -EOPNOTSUPP;
2137                 break;
2138         default:
2139                 return -EOPNOTSUPP;
2140         }
2141
2142         return ieee80211_request_scan(sdata, req);
2143 }
2144
2145 static int
2146 ieee80211_sched_scan_start(struct wiphy *wiphy,
2147                            struct net_device *dev,
2148                            struct cfg80211_sched_scan_request *req)
2149 {
2150         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2151
2152         if (!sdata->local->ops->sched_scan_start)
2153                 return -EOPNOTSUPP;
2154
2155         return ieee80211_request_sched_scan_start(sdata, req);
2156 }
2157
2158 static int
2159 ieee80211_sched_scan_stop(struct wiphy *wiphy, struct net_device *dev)
2160 {
2161         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2162
2163         if (!sdata->local->ops->sched_scan_stop)
2164                 return -EOPNOTSUPP;
2165
2166         return ieee80211_request_sched_scan_stop(sdata);
2167 }
2168
2169 static int ieee80211_auth(struct wiphy *wiphy, struct net_device *dev,
2170                           struct cfg80211_auth_request *req)
2171 {
2172         return ieee80211_mgd_auth(IEEE80211_DEV_TO_SUB_IF(dev), req);
2173 }
2174
2175 static int ieee80211_assoc(struct wiphy *wiphy, struct net_device *dev,
2176                            struct cfg80211_assoc_request *req)
2177 {
2178         return ieee80211_mgd_assoc(IEEE80211_DEV_TO_SUB_IF(dev), req);
2179 }
2180
2181 static int ieee80211_deauth(struct wiphy *wiphy, struct net_device *dev,
2182                             struct cfg80211_deauth_request *req)
2183 {
2184         return ieee80211_mgd_deauth(IEEE80211_DEV_TO_SUB_IF(dev), req);
2185 }
2186
2187 static int ieee80211_disassoc(struct wiphy *wiphy, struct net_device *dev,
2188                               struct cfg80211_disassoc_request *req)
2189 {
2190         return ieee80211_mgd_disassoc(IEEE80211_DEV_TO_SUB_IF(dev), req);
2191 }
2192
2193 static int ieee80211_join_ibss(struct wiphy *wiphy, struct net_device *dev,
2194                                struct cfg80211_ibss_params *params)
2195 {
2196         return ieee80211_ibss_join(IEEE80211_DEV_TO_SUB_IF(dev), params);
2197 }
2198
2199 static int ieee80211_leave_ibss(struct wiphy *wiphy, struct net_device *dev)
2200 {
2201         return ieee80211_ibss_leave(IEEE80211_DEV_TO_SUB_IF(dev));
2202 }
2203
2204 static int ieee80211_set_mcast_rate(struct wiphy *wiphy, struct net_device *dev,
2205                                     int rate[IEEE80211_NUM_BANDS])
2206 {
2207         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2208
2209         memcpy(sdata->vif.bss_conf.mcast_rate, rate,
2210                sizeof(int) * IEEE80211_NUM_BANDS);
2211
2212         return 0;
2213 }
2214
2215 static int ieee80211_set_wiphy_params(struct wiphy *wiphy, u32 changed)
2216 {
2217         struct ieee80211_local *local = wiphy_priv(wiphy);
2218         int err;
2219
2220         if (changed & WIPHY_PARAM_FRAG_THRESHOLD) {
2221                 err = drv_set_frag_threshold(local, wiphy->frag_threshold);
2222
2223                 if (err)
2224                         return err;
2225         }
2226
2227         if (changed & WIPHY_PARAM_COVERAGE_CLASS) {
2228                 err = drv_set_coverage_class(local, wiphy->coverage_class);
2229
2230                 if (err)
2231                         return err;
2232         }
2233
2234         if (changed & WIPHY_PARAM_RTS_THRESHOLD) {
2235                 err = drv_set_rts_threshold(local, wiphy->rts_threshold);
2236
2237                 if (err)
2238                         return err;
2239         }
2240
2241         if (changed & WIPHY_PARAM_RETRY_SHORT) {
2242                 if (wiphy->retry_short > IEEE80211_MAX_TX_RETRY)
2243                         return -EINVAL;
2244                 local->hw.conf.short_frame_max_tx_count = wiphy->retry_short;
2245         }
2246         if (changed & WIPHY_PARAM_RETRY_LONG) {
2247                 if (wiphy->retry_long > IEEE80211_MAX_TX_RETRY)
2248                         return -EINVAL;
2249                 local->hw.conf.long_frame_max_tx_count = wiphy->retry_long;
2250         }
2251         if (changed &
2252             (WIPHY_PARAM_RETRY_SHORT | WIPHY_PARAM_RETRY_LONG))
2253                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_RETRY_LIMITS);
2254
2255         return 0;
2256 }
2257
2258 static int ieee80211_set_tx_power(struct wiphy *wiphy,
2259                                   struct wireless_dev *wdev,
2260                                   enum nl80211_tx_power_setting type, int mbm)
2261 {
2262         struct ieee80211_local *local = wiphy_priv(wiphy);
2263         struct ieee80211_sub_if_data *sdata;
2264
2265         if (wdev) {
2266                 sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
2267
2268                 switch (type) {
2269                 case NL80211_TX_POWER_AUTOMATIC:
2270                         sdata->user_power_level = IEEE80211_UNSET_POWER_LEVEL;
2271                         break;
2272                 case NL80211_TX_POWER_LIMITED:
2273                 case NL80211_TX_POWER_FIXED:
2274                         if (mbm < 0 || (mbm % 100))
2275                                 return -EOPNOTSUPP;
2276                         sdata->user_power_level = MBM_TO_DBM(mbm);
2277                         break;
2278                 }
2279
2280                 ieee80211_recalc_txpower(sdata);
2281
2282                 return 0;
2283         }
2284
2285         switch (type) {
2286         case NL80211_TX_POWER_AUTOMATIC:
2287                 local->user_power_level = IEEE80211_UNSET_POWER_LEVEL;
2288                 break;
2289         case NL80211_TX_POWER_LIMITED:
2290         case NL80211_TX_POWER_FIXED:
2291                 if (mbm < 0 || (mbm % 100))
2292                         return -EOPNOTSUPP;
2293                 local->user_power_level = MBM_TO_DBM(mbm);
2294                 break;
2295         }
2296
2297         mutex_lock(&local->iflist_mtx);
2298         list_for_each_entry(sdata, &local->interfaces, list)
2299                 sdata->user_power_level = local->user_power_level;
2300         list_for_each_entry(sdata, &local->interfaces, list)
2301                 ieee80211_recalc_txpower(sdata);
2302         mutex_unlock(&local->iflist_mtx);
2303
2304         return 0;
2305 }
2306
2307 static int ieee80211_get_tx_power(struct wiphy *wiphy,
2308                                   struct wireless_dev *wdev,
2309                                   int *dbm)
2310 {
2311         struct ieee80211_local *local = wiphy_priv(wiphy);
2312         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
2313
2314         if (!local->use_chanctx)
2315                 *dbm = local->hw.conf.power_level;
2316         else
2317                 *dbm = sdata->vif.bss_conf.txpower;
2318
2319         return 0;
2320 }
2321
2322 static int ieee80211_set_wds_peer(struct wiphy *wiphy, struct net_device *dev,
2323                                   const u8 *addr)
2324 {
2325         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2326
2327         memcpy(&sdata->u.wds.remote_addr, addr, ETH_ALEN);
2328
2329         return 0;
2330 }
2331
2332 static void ieee80211_rfkill_poll(struct wiphy *wiphy)
2333 {
2334         struct ieee80211_local *local = wiphy_priv(wiphy);
2335
2336         drv_rfkill_poll(local);
2337 }
2338
2339 #ifdef CONFIG_NL80211_TESTMODE
2340 static int ieee80211_testmode_cmd(struct wiphy *wiphy,
2341                                   struct wireless_dev *wdev,
2342                                   void *data, int len)
2343 {
2344         struct ieee80211_local *local = wiphy_priv(wiphy);
2345         struct ieee80211_vif *vif = NULL;
2346
2347         if (!local->ops->testmode_cmd)
2348                 return -EOPNOTSUPP;
2349
2350         if (wdev) {
2351                 struct ieee80211_sub_if_data *sdata;
2352
2353                 sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
2354                 if (sdata->flags & IEEE80211_SDATA_IN_DRIVER)
2355                         vif = &sdata->vif;
2356         }
2357
2358         return local->ops->testmode_cmd(&local->hw, vif, data, len);
2359 }
2360
2361 static int ieee80211_testmode_dump(struct wiphy *wiphy,
2362                                    struct sk_buff *skb,
2363                                    struct netlink_callback *cb,
2364                                    void *data, int len)
2365 {
2366         struct ieee80211_local *local = wiphy_priv(wiphy);
2367
2368         if (!local->ops->testmode_dump)
2369                 return -EOPNOTSUPP;
2370
2371         return local->ops->testmode_dump(&local->hw, skb, cb, data, len);
2372 }
2373 #endif
2374
2375 int __ieee80211_request_smps_ap(struct ieee80211_sub_if_data *sdata,
2376                                 enum ieee80211_smps_mode smps_mode)
2377 {
2378         struct sta_info *sta;
2379         enum ieee80211_smps_mode old_req;
2380         int i;
2381
2382         if (WARN_ON_ONCE(sdata->vif.type != NL80211_IFTYPE_AP))
2383                 return -EINVAL;
2384
2385         if (sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT)
2386                 return 0;
2387
2388         old_req = sdata->u.ap.req_smps;
2389         sdata->u.ap.req_smps = smps_mode;
2390
2391         /* AUTOMATIC doesn't mean much for AP - don't allow it */
2392         if (old_req == smps_mode ||
2393             smps_mode == IEEE80211_SMPS_AUTOMATIC)
2394                 return 0;
2395
2396          /* If no associated stations, there's no need to do anything */
2397         if (!atomic_read(&sdata->u.ap.num_mcast_sta)) {
2398                 sdata->smps_mode = smps_mode;
2399                 ieee80211_queue_work(&sdata->local->hw, &sdata->recalc_smps);
2400                 return 0;
2401         }
2402
2403         ht_dbg(sdata,
2404                "SMSP %d requested in AP mode, sending Action frame to %d stations\n",
2405                smps_mode, atomic_read(&sdata->u.ap.num_mcast_sta));
2406
2407         mutex_lock(&sdata->local->sta_mtx);
2408         for (i = 0; i < STA_HASH_SIZE; i++) {
2409                 for (sta = rcu_dereference_protected(sdata->local->sta_hash[i],
2410                                 lockdep_is_held(&sdata->local->sta_mtx));
2411                      sta;
2412                      sta = rcu_dereference_protected(sta->hnext,
2413                                 lockdep_is_held(&sdata->local->sta_mtx))) {
2414                         /*
2415                          * Only stations associated to our AP and
2416                          * associated VLANs
2417                          */
2418                         if (sta->sdata->bss != &sdata->u.ap)
2419                                 continue;
2420
2421                         /* This station doesn't support MIMO - skip it */
2422                         if (sta_info_tx_streams(sta) == 1)
2423                                 continue;
2424
2425                         /*
2426                          * Don't wake up a STA just to send the action frame
2427                          * unless we are getting more restrictive.
2428                          */
2429                         if (test_sta_flag(sta, WLAN_STA_PS_STA) &&
2430                             !ieee80211_smps_is_restrictive(sta->known_smps_mode,
2431                                                            smps_mode)) {
2432                                 ht_dbg(sdata,
2433                                        "Won't send SMPS to sleeping STA %pM\n",
2434                                        sta->sta.addr);
2435                                 continue;
2436                         }
2437
2438                         /*
2439                          * If the STA is not authorized, wait until it gets
2440                          * authorized and the action frame will be sent then.
2441                          */
2442                         if (!test_sta_flag(sta, WLAN_STA_AUTHORIZED))
2443                                 continue;
2444
2445                         ht_dbg(sdata, "Sending SMPS to %pM\n", sta->sta.addr);
2446                         ieee80211_send_smps_action(sdata, smps_mode,
2447                                                    sta->sta.addr,
2448                                                    sdata->vif.bss_conf.bssid);
2449                 }
2450         }
2451         mutex_unlock(&sdata->local->sta_mtx);
2452
2453         sdata->smps_mode = smps_mode;
2454         ieee80211_queue_work(&sdata->local->hw, &sdata->recalc_smps);
2455
2456         return 0;
2457 }
2458
2459 int __ieee80211_request_smps_mgd(struct ieee80211_sub_if_data *sdata,
2460                                  enum ieee80211_smps_mode smps_mode)
2461 {
2462         const u8 *ap;
2463         enum ieee80211_smps_mode old_req;
2464         int err;
2465
2466         lockdep_assert_held(&sdata->wdev.mtx);
2467
2468         if (WARN_ON_ONCE(sdata->vif.type != NL80211_IFTYPE_STATION))
2469                 return -EINVAL;
2470
2471         old_req = sdata->u.mgd.req_smps;
2472         sdata->u.mgd.req_smps = smps_mode;
2473
2474         if (old_req == smps_mode &&
2475             smps_mode != IEEE80211_SMPS_AUTOMATIC)
2476                 return 0;
2477
2478         /*
2479          * If not associated, or current association is not an HT
2480          * association, there's no need to do anything, just store
2481          * the new value until we associate.
2482          */
2483         if (!sdata->u.mgd.associated ||
2484             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT)
2485                 return 0;
2486
2487         ap = sdata->u.mgd.associated->bssid;
2488
2489         if (smps_mode == IEEE80211_SMPS_AUTOMATIC) {
2490                 if (sdata->u.mgd.powersave)
2491                         smps_mode = IEEE80211_SMPS_DYNAMIC;
2492                 else
2493                         smps_mode = IEEE80211_SMPS_OFF;
2494         }
2495
2496         /* send SM PS frame to AP */
2497         err = ieee80211_send_smps_action(sdata, smps_mode,
2498                                          ap, ap);
2499         if (err)
2500                 sdata->u.mgd.req_smps = old_req;
2501
2502         return err;
2503 }
2504
2505 static int ieee80211_set_power_mgmt(struct wiphy *wiphy, struct net_device *dev,
2506                                     bool enabled, int timeout)
2507 {
2508         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2509         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
2510
2511         if (sdata->vif.type != NL80211_IFTYPE_STATION)
2512                 return -EOPNOTSUPP;
2513
2514         if (!(local->hw.flags & IEEE80211_HW_SUPPORTS_PS))
2515                 return -EOPNOTSUPP;
2516
2517         if (enabled == sdata->u.mgd.powersave &&
2518             timeout == local->dynamic_ps_forced_timeout)
2519                 return 0;
2520
2521         sdata->u.mgd.powersave = enabled;
2522         local->dynamic_ps_forced_timeout = timeout;
2523
2524         /* no change, but if automatic follow powersave */
2525         sdata_lock(sdata);
2526         __ieee80211_request_smps_mgd(sdata, sdata->u.mgd.req_smps);
2527         sdata_unlock(sdata);
2528
2529         if (local->hw.flags & IEEE80211_HW_SUPPORTS_DYNAMIC_PS)
2530                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
2531
2532         ieee80211_recalc_ps(local, -1);
2533         ieee80211_recalc_ps_vif(sdata);
2534
2535         return 0;
2536 }
2537
2538 static int ieee80211_set_cqm_rssi_config(struct wiphy *wiphy,
2539                                          struct net_device *dev,
2540                                          s32 rssi_thold, u32 rssi_hyst)
2541 {
2542         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2543         struct ieee80211_vif *vif = &sdata->vif;
2544         struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
2545
2546         if (rssi_thold == bss_conf->cqm_rssi_thold &&
2547             rssi_hyst == bss_conf->cqm_rssi_hyst)
2548                 return 0;
2549
2550         bss_conf->cqm_rssi_thold = rssi_thold;
2551         bss_conf->cqm_rssi_hyst = rssi_hyst;
2552
2553         /* tell the driver upon association, unless already associated */
2554         if (sdata->u.mgd.associated &&
2555             sdata->vif.driver_flags & IEEE80211_VIF_SUPPORTS_CQM_RSSI)
2556                 ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_CQM);
2557
2558         return 0;
2559 }
2560
2561 static int ieee80211_set_bitrate_mask(struct wiphy *wiphy,
2562                                       struct net_device *dev,
2563                                       const u8 *addr,
2564                                       const struct cfg80211_bitrate_mask *mask)
2565 {
2566         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2567         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
2568         int i, ret;
2569
2570         if (!ieee80211_sdata_running(sdata))
2571                 return -ENETDOWN;
2572
2573         if (local->hw.flags & IEEE80211_HW_HAS_RATE_CONTROL) {
2574                 ret = drv_set_bitrate_mask(local, sdata, mask);
2575                 if (ret)
2576                         return ret;
2577         }
2578
2579         for (i = 0; i < IEEE80211_NUM_BANDS; i++) {
2580                 struct ieee80211_supported_band *sband = wiphy->bands[i];
2581                 int j;
2582
2583                 sdata->rc_rateidx_mask[i] = mask->control[i].legacy;
2584                 memcpy(sdata->rc_rateidx_mcs_mask[i], mask->control[i].ht_mcs,
2585                        sizeof(mask->control[i].ht_mcs));
2586
2587                 sdata->rc_has_mcs_mask[i] = false;
2588                 if (!sband)
2589                         continue;
2590
2591                 for (j = 0; j < IEEE80211_HT_MCS_MASK_LEN; j++)
2592                         if (~sdata->rc_rateidx_mcs_mask[i][j]) {
2593                                 sdata->rc_has_mcs_mask[i] = true;
2594                                 break;
2595                         }
2596         }
2597
2598         return 0;
2599 }
2600
2601 static int ieee80211_start_roc_work(struct ieee80211_local *local,
2602                                     struct ieee80211_sub_if_data *sdata,
2603                                     struct ieee80211_channel *channel,
2604                                     unsigned int duration, u64 *cookie,
2605                                     struct sk_buff *txskb,
2606                                     enum ieee80211_roc_type type)
2607 {
2608         struct ieee80211_roc_work *roc, *tmp;
2609         bool queued = false;
2610         int ret;
2611
2612         lockdep_assert_held(&local->mtx);
2613
2614         if (local->use_chanctx && !local->ops->remain_on_channel)
2615                 return -EOPNOTSUPP;
2616
2617         roc = kzalloc(sizeof(*roc), GFP_KERNEL);
2618         if (!roc)
2619                 return -ENOMEM;
2620
2621         roc->chan = channel;
2622         roc->duration = duration;
2623         roc->req_duration = duration;
2624         roc->frame = txskb;
2625         roc->type = type;
2626         roc->mgmt_tx_cookie = (unsigned long)txskb;
2627         roc->sdata = sdata;
2628         INIT_DELAYED_WORK(&roc->work, ieee80211_sw_roc_work);
2629         INIT_LIST_HEAD(&roc->dependents);
2630
2631         /* if there's one pending or we're scanning, queue this one */
2632         if (!list_empty(&local->roc_list) ||
2633             local->scanning || local->radar_detect_enabled)
2634                 goto out_check_combine;
2635
2636         /* if not HW assist, just queue & schedule work */
2637         if (!local->ops->remain_on_channel) {
2638                 ieee80211_queue_delayed_work(&local->hw, &roc->work, 0);
2639                 goto out_queue;
2640         }
2641
2642         /* otherwise actually kick it off here (for error handling) */
2643
2644         /*
2645          * If the duration is zero, then the driver
2646          * wouldn't actually do anything. Set it to
2647          * 10 for now.
2648          *
2649          * TODO: cancel the off-channel operation
2650          *       when we get the SKB's TX status and
2651          *       the wait time was zero before.
2652          */
2653         if (!duration)
2654                 duration = 10;
2655
2656         ret = drv_remain_on_channel(local, sdata, channel, duration, type);
2657         if (ret) {
2658                 kfree(roc);
2659                 return ret;
2660         }
2661
2662         roc->started = true;
2663         goto out_queue;
2664
2665  out_check_combine:
2666         list_for_each_entry(tmp, &local->roc_list, list) {
2667                 if (tmp->chan != channel || tmp->sdata != sdata)
2668                         continue;
2669
2670                 /*
2671                  * Extend this ROC if possible:
2672                  *
2673                  * If it hasn't started yet, just increase the duration
2674                  * and add the new one to the list of dependents.
2675                  * If the type of the new ROC has higher priority, modify the
2676                  * type of the previous one to match that of the new one.
2677                  */
2678                 if (!tmp->started) {
2679                         list_add_tail(&roc->list, &tmp->dependents);
2680                         tmp->duration = max(tmp->duration, roc->duration);
2681                         tmp->type = max(tmp->type, roc->type);
2682                         queued = true;
2683                         break;
2684                 }
2685
2686                 /* If it has already started, it's more difficult ... */
2687                 if (local->ops->remain_on_channel) {
2688                         unsigned long j = jiffies;
2689
2690                         /*
2691                          * In the offloaded ROC case, if it hasn't begun, add
2692                          * this new one to the dependent list to be handled
2693                          * when the master one begins. If it has begun,
2694                          * check that there's still a minimum time left and
2695                          * if so, start this one, transmitting the frame, but
2696                          * add it to the list directly after this one with
2697                          * a reduced time so we'll ask the driver to execute
2698                          * it right after finishing the previous one, in the
2699                          * hope that it'll also be executed right afterwards,
2700                          * effectively extending the old one.
2701                          * If there's no minimum time left, just add it to the
2702                          * normal list.
2703                          * TODO: the ROC type is ignored here, assuming that it
2704                          * is better to immediately use the current ROC.
2705                          */
2706                         if (!tmp->hw_begun) {
2707                                 list_add_tail(&roc->list, &tmp->dependents);
2708                                 queued = true;
2709                                 break;
2710                         }
2711
2712                         if (time_before(j + IEEE80211_ROC_MIN_LEFT,
2713                                         tmp->hw_start_time +
2714                                         msecs_to_jiffies(tmp->duration))) {
2715                                 int new_dur;
2716
2717                                 ieee80211_handle_roc_started(roc);
2718
2719                                 new_dur = roc->duration -
2720                                           jiffies_to_msecs(tmp->hw_start_time +
2721                                                            msecs_to_jiffies(
2722                                                                 tmp->duration) -
2723                                                            j);
2724
2725                                 if (new_dur > 0) {
2726                                         /* add right after tmp */
2727                                         list_add(&roc->list, &tmp->list);
2728                                 } else {
2729                                         list_add_tail(&roc->list,
2730                                                       &tmp->dependents);
2731                                 }
2732                                 queued = true;
2733                         }
2734                 } else if (del_timer_sync(&tmp->work.timer)) {
2735                         unsigned long new_end;
2736
2737                         /*
2738                          * In the software ROC case, cancel the timer, if
2739                          * that fails then the finish work is already
2740                          * queued/pending and thus we queue the new ROC
2741                          * normally, if that succeeds then we can extend
2742                          * the timer duration and TX the frame (if any.)
2743                          */
2744
2745                         list_add_tail(&roc->list, &tmp->dependents);
2746                         queued = true;
2747
2748                         new_end = jiffies + msecs_to_jiffies(roc->duration);
2749
2750                         /* ok, it was started & we canceled timer */
2751                         if (time_after(new_end, tmp->work.timer.expires))
2752                                 mod_timer(&tmp->work.timer, new_end);
2753                         else
2754                                 add_timer(&tmp->work.timer);
2755
2756                         ieee80211_handle_roc_started(roc);
2757                 }
2758                 break;
2759         }
2760
2761  out_queue:
2762         if (!queued)
2763                 list_add_tail(&roc->list, &local->roc_list);
2764
2765         /*
2766          * cookie is either the roc cookie (for normal roc)
2767          * or the SKB (for mgmt TX)
2768          */
2769         if (!txskb) {
2770                 /* local->mtx protects this */
2771                 local->roc_cookie_counter++;
2772                 roc->cookie = local->roc_cookie_counter;
2773                 /* wow, you wrapped 64 bits ... more likely a bug */
2774                 if (WARN_ON(roc->cookie == 0)) {
2775                         roc->cookie = 1;
2776                         local->roc_cookie_counter++;
2777                 }
2778                 *cookie = roc->cookie;
2779         } else {
2780                 *cookie = (unsigned long)txskb;
2781         }
2782
2783         return 0;
2784 }
2785
2786 static int ieee80211_remain_on_channel(struct wiphy *wiphy,
2787                                        struct wireless_dev *wdev,
2788                                        struct ieee80211_channel *chan,
2789                                        unsigned int duration,
2790                                        u64 *cookie)
2791 {
2792         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
2793         struct ieee80211_local *local = sdata->local;
2794         int ret;
2795
2796         mutex_lock(&local->mtx);
2797         ret = ieee80211_start_roc_work(local, sdata, chan,
2798                                        duration, cookie, NULL,
2799                                        IEEE80211_ROC_TYPE_NORMAL);
2800         mutex_unlock(&local->mtx);
2801
2802         return ret;
2803 }
2804
2805 static int ieee80211_cancel_roc(struct ieee80211_local *local,
2806                                 u64 cookie, bool mgmt_tx)
2807 {
2808         struct ieee80211_roc_work *roc, *tmp, *found = NULL;
2809         int ret;
2810
2811         mutex_lock(&local->mtx);
2812         list_for_each_entry_safe(roc, tmp, &local->roc_list, list) {
2813                 struct ieee80211_roc_work *dep, *tmp2;
2814
2815                 list_for_each_entry_safe(dep, tmp2, &roc->dependents, list) {
2816                         if (!mgmt_tx && dep->cookie != cookie)
2817                                 continue;
2818                         else if (mgmt_tx && dep->mgmt_tx_cookie != cookie)
2819                                 continue;
2820                         /* found dependent item -- just remove it */
2821                         list_del(&dep->list);
2822                         mutex_unlock(&local->mtx);
2823
2824                         ieee80211_roc_notify_destroy(dep, true);
2825                         return 0;
2826                 }
2827
2828                 if (!mgmt_tx && roc->cookie != cookie)
2829                         continue;
2830                 else if (mgmt_tx && roc->mgmt_tx_cookie != cookie)
2831                         continue;
2832
2833                 found = roc;
2834                 break;
2835         }
2836
2837         if (!found) {
2838                 mutex_unlock(&local->mtx);
2839                 return -ENOENT;
2840         }
2841
2842         /*
2843          * We found the item to cancel, so do that. Note that it
2844          * may have dependents, which we also cancel (and send
2845          * the expired signal for.) Not doing so would be quite
2846          * tricky here, but we may need to fix it later.
2847          */
2848
2849         if (local->ops->remain_on_channel) {
2850                 if (found->started) {
2851                         ret = drv_cancel_remain_on_channel(local);
2852                         if (WARN_ON_ONCE(ret)) {
2853                                 mutex_unlock(&local->mtx);
2854                                 return ret;
2855                         }
2856                 }
2857
2858                 list_del(&found->list);
2859
2860                 if (found->started)
2861                         ieee80211_start_next_roc(local);
2862                 mutex_unlock(&local->mtx);
2863
2864                 ieee80211_roc_notify_destroy(found, true);
2865         } else {
2866                 /* work may be pending so use it all the time */
2867                 found->abort = true;
2868                 ieee80211_queue_delayed_work(&local->hw, &found->work, 0);
2869
2870                 mutex_unlock(&local->mtx);
2871
2872                 /* work will clean up etc */
2873                 flush_delayed_work(&found->work);
2874                 WARN_ON(!found->to_be_freed);
2875                 kfree(found);
2876         }
2877
2878         return 0;
2879 }
2880
2881 static int ieee80211_cancel_remain_on_channel(struct wiphy *wiphy,
2882                                               struct wireless_dev *wdev,
2883                                               u64 cookie)
2884 {
2885         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
2886         struct ieee80211_local *local = sdata->local;
2887
2888         return ieee80211_cancel_roc(local, cookie, false);
2889 }
2890
2891 static int ieee80211_start_radar_detection(struct wiphy *wiphy,
2892                                            struct net_device *dev,
2893                                            struct cfg80211_chan_def *chandef)
2894 {
2895         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2896         struct ieee80211_local *local = sdata->local;
2897         unsigned long timeout;
2898         int err;
2899
2900         if (!list_empty(&local->roc_list) || local->scanning)
2901                 return -EBUSY;
2902
2903         /* whatever, but channel contexts should not complain about that one */
2904         sdata->smps_mode = IEEE80211_SMPS_OFF;
2905         sdata->needed_rx_chains = local->rx_chains;
2906         sdata->radar_required = true;
2907
2908         mutex_lock(&local->iflist_mtx);
2909         err = ieee80211_vif_use_channel(sdata, chandef,
2910                                         IEEE80211_CHANCTX_SHARED);
2911         mutex_unlock(&local->iflist_mtx);
2912         if (err)
2913                 return err;
2914
2915         timeout = msecs_to_jiffies(IEEE80211_DFS_MIN_CAC_TIME_MS);
2916         ieee80211_queue_delayed_work(&sdata->local->hw,
2917                                      &sdata->dfs_cac_timer_work, timeout);
2918
2919         return 0;
2920 }
2921
2922 static struct cfg80211_beacon_data *
2923 cfg80211_beacon_dup(struct cfg80211_beacon_data *beacon)
2924 {
2925         struct cfg80211_beacon_data *new_beacon;
2926         u8 *pos;
2927         int len;
2928
2929         len = beacon->head_len + beacon->tail_len + beacon->beacon_ies_len +
2930               beacon->proberesp_ies_len + beacon->assocresp_ies_len +
2931               beacon->probe_resp_len;
2932
2933         new_beacon = kzalloc(sizeof(*new_beacon) + len, GFP_KERNEL);
2934         if (!new_beacon)
2935                 return NULL;
2936
2937         pos = (u8 *)(new_beacon + 1);
2938         if (beacon->head_len) {
2939                 new_beacon->head_len = beacon->head_len;
2940                 new_beacon->head = pos;
2941                 memcpy(pos, beacon->head, beacon->head_len);
2942                 pos += beacon->head_len;
2943         }
2944         if (beacon->tail_len) {
2945                 new_beacon->tail_len = beacon->tail_len;
2946                 new_beacon->tail = pos;
2947                 memcpy(pos, beacon->tail, beacon->tail_len);
2948                 pos += beacon->tail_len;
2949         }
2950         if (beacon->beacon_ies_len) {
2951                 new_beacon->beacon_ies_len = beacon->beacon_ies_len;
2952                 new_beacon->beacon_ies = pos;
2953                 memcpy(pos, beacon->beacon_ies, beacon->beacon_ies_len);
2954                 pos += beacon->beacon_ies_len;
2955         }
2956         if (beacon->proberesp_ies_len) {
2957                 new_beacon->proberesp_ies_len = beacon->proberesp_ies_len;
2958                 new_beacon->proberesp_ies = pos;
2959                 memcpy(pos, beacon->proberesp_ies, beacon->proberesp_ies_len);
2960                 pos += beacon->proberesp_ies_len;
2961         }
2962         if (beacon->assocresp_ies_len) {
2963                 new_beacon->assocresp_ies_len = beacon->assocresp_ies_len;
2964                 new_beacon->assocresp_ies = pos;
2965                 memcpy(pos, beacon->assocresp_ies, beacon->assocresp_ies_len);
2966                 pos += beacon->assocresp_ies_len;
2967         }
2968         if (beacon->probe_resp_len) {
2969                 new_beacon->probe_resp_len = beacon->probe_resp_len;
2970                 beacon->probe_resp = pos;
2971                 memcpy(pos, beacon->probe_resp, beacon->probe_resp_len);
2972                 pos += beacon->probe_resp_len;
2973         }
2974
2975         return new_beacon;
2976 }
2977
2978 void ieee80211_csa_finalize_work(struct work_struct *work)
2979 {
2980         struct ieee80211_sub_if_data *sdata =
2981                 container_of(work, struct ieee80211_sub_if_data,
2982                              csa_finalize_work);
2983         struct ieee80211_local *local = sdata->local;
2984         int err, changed = 0;
2985
2986         sdata_lock(sdata);
2987         /* AP might have been stopped while waiting for the lock. */
2988         if (!sdata->vif.csa_active)
2989                 goto unlock;
2990
2991         if (!ieee80211_sdata_running(sdata))
2992                 goto unlock;
2993
2994         sdata->radar_required = sdata->csa_radar_required;
2995         err = ieee80211_vif_change_channel(sdata, &changed);
2996         if (WARN_ON(err < 0))
2997                 goto unlock;
2998
2999         if (!local->use_chanctx) {
3000                 local->_oper_chandef = sdata->csa_chandef;
3001                 ieee80211_hw_config(local, 0);
3002         }
3003
3004         ieee80211_bss_info_change_notify(sdata, changed);
3005
3006         sdata->vif.csa_active = false;
3007         switch (sdata->vif.type) {
3008         case NL80211_IFTYPE_AP:
3009                 err = ieee80211_assign_beacon(sdata, sdata->u.ap.next_beacon);
3010                 if (err < 0)
3011                         goto unlock;
3012
3013                 changed |= err;
3014                 kfree(sdata->u.ap.next_beacon);
3015                 sdata->u.ap.next_beacon = NULL;
3016
3017                 ieee80211_bss_info_change_notify(sdata, err);
3018                 break;
3019         case NL80211_IFTYPE_ADHOC:
3020                 ieee80211_ibss_finish_csa(sdata);
3021                 break;
3022 #ifdef CONFIG_MAC80211_MESH
3023         case NL80211_IFTYPE_MESH_POINT:
3024                 err = ieee80211_mesh_finish_csa(sdata);
3025                 if (err < 0)
3026                         goto unlock;
3027                 break;
3028 #endif
3029         default:
3030                 WARN_ON(1);
3031                 goto unlock;
3032         }
3033
3034         ieee80211_wake_queues_by_reason(&sdata->local->hw,
3035                                         IEEE80211_MAX_QUEUE_MAP,
3036                                         IEEE80211_QUEUE_STOP_REASON_CSA);
3037
3038         cfg80211_ch_switch_notify(sdata->dev, &sdata->csa_chandef);
3039
3040 unlock:
3041         sdata_unlock(sdata);
3042 }
3043
3044 static int ieee80211_channel_switch(struct wiphy *wiphy, struct net_device *dev,
3045                                     struct cfg80211_csa_settings *params)
3046 {
3047         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3048         struct ieee80211_local *local = sdata->local;
3049         struct ieee80211_chanctx_conf *chanctx_conf;
3050         struct ieee80211_chanctx *chanctx;
3051         struct ieee80211_if_mesh __maybe_unused *ifmsh;
3052         int err, num_chanctx;
3053
3054         lockdep_assert_held(&sdata->wdev.mtx);
3055
3056         if (!list_empty(&local->roc_list) || local->scanning)
3057                 return -EBUSY;
3058
3059         if (sdata->wdev.cac_started)
3060                 return -EBUSY;
3061
3062         if (cfg80211_chandef_identical(&params->chandef,
3063                                        &sdata->vif.bss_conf.chandef))
3064                 return -EINVAL;
3065
3066         rcu_read_lock();
3067         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
3068         if (!chanctx_conf) {
3069                 rcu_read_unlock();
3070                 return -EBUSY;
3071         }
3072
3073         /* don't handle for multi-VIF cases */
3074         chanctx = container_of(chanctx_conf, struct ieee80211_chanctx, conf);
3075         if (chanctx->refcount > 1) {
3076                 rcu_read_unlock();
3077                 return -EBUSY;
3078         }
3079         num_chanctx = 0;
3080         list_for_each_entry_rcu(chanctx, &local->chanctx_list, list)
3081                 num_chanctx++;
3082         rcu_read_unlock();
3083
3084         if (num_chanctx > 1)
3085                 return -EBUSY;
3086
3087         /* don't allow another channel switch if one is already active. */
3088         if (sdata->vif.csa_active)
3089                 return -EBUSY;
3090
3091         switch (sdata->vif.type) {
3092         case NL80211_IFTYPE_AP:
3093                 sdata->csa_counter_offset_beacon =
3094                         params->counter_offset_beacon;
3095                 sdata->csa_counter_offset_presp = params->counter_offset_presp;
3096                 sdata->u.ap.next_beacon =
3097                         cfg80211_beacon_dup(&params->beacon_after);
3098                 if (!sdata->u.ap.next_beacon)
3099                         return -ENOMEM;
3100
3101                 err = ieee80211_assign_beacon(sdata, &params->beacon_csa);
3102                 if (err < 0) {
3103                         kfree(sdata->u.ap.next_beacon);
3104                         return err;
3105                 }
3106                 break;
3107         case NL80211_IFTYPE_ADHOC:
3108                 if (!sdata->vif.bss_conf.ibss_joined)
3109                         return -EINVAL;
3110
3111                 if (params->chandef.width != sdata->u.ibss.chandef.width)
3112                         return -EINVAL;
3113
3114                 switch (params->chandef.width) {
3115                 case NL80211_CHAN_WIDTH_40:
3116                         if (cfg80211_get_chandef_type(&params->chandef) !=
3117                             cfg80211_get_chandef_type(&sdata->u.ibss.chandef))
3118                                 return -EINVAL;
3119                 case NL80211_CHAN_WIDTH_5:
3120                 case NL80211_CHAN_WIDTH_10:
3121                 case NL80211_CHAN_WIDTH_20_NOHT:
3122                 case NL80211_CHAN_WIDTH_20:
3123                         break;
3124                 default:
3125                         return -EINVAL;
3126                 }
3127
3128                 /* changes into another band are not supported */
3129                 if (sdata->u.ibss.chandef.chan->band !=
3130                     params->chandef.chan->band)
3131                         return -EINVAL;
3132
3133                 err = ieee80211_ibss_csa_beacon(sdata, params);
3134                 if (err < 0)
3135                         return err;
3136                 break;
3137 #ifdef CONFIG_MAC80211_MESH
3138         case NL80211_IFTYPE_MESH_POINT:
3139                 ifmsh = &sdata->u.mesh;
3140
3141                 if (!ifmsh->mesh_id)
3142                         return -EINVAL;
3143
3144                 if (params->chandef.width != sdata->vif.bss_conf.chandef.width)
3145                         return -EINVAL;
3146
3147                 /* changes into another band are not supported */
3148                 if (sdata->vif.bss_conf.chandef.chan->band !=
3149                     params->chandef.chan->band)
3150                         return -EINVAL;
3151
3152                 ifmsh->chsw_init = true;
3153                 if (!ifmsh->pre_value)
3154                         ifmsh->pre_value = 1;
3155                 else
3156                         ifmsh->pre_value++;
3157
3158                 err = ieee80211_mesh_csa_beacon(sdata, params, true);
3159                 if (err < 0) {
3160                         ifmsh->chsw_init = false;
3161                         return err;
3162                 }
3163                 break;
3164 #endif
3165         default:
3166                 return -EOPNOTSUPP;
3167         }
3168
3169         sdata->csa_radar_required = params->radar_required;
3170
3171         if (params->block_tx)
3172                 ieee80211_stop_queues_by_reason(&local->hw,
3173                                 IEEE80211_MAX_QUEUE_MAP,
3174                                 IEEE80211_QUEUE_STOP_REASON_CSA);
3175
3176         sdata->csa_chandef = params->chandef;
3177         sdata->vif.csa_active = true;
3178
3179         ieee80211_bss_info_change_notify(sdata, err);
3180         drv_channel_switch_beacon(sdata, &params->chandef);
3181
3182         return 0;
3183 }
3184
3185 static int ieee80211_mgmt_tx(struct wiphy *wiphy, struct wireless_dev *wdev,
3186                              struct cfg80211_mgmt_tx_params *params,
3187                              u64 *cookie)
3188 {
3189         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
3190         struct ieee80211_local *local = sdata->local;
3191         struct sk_buff *skb;
3192         struct sta_info *sta;
3193         const struct ieee80211_mgmt *mgmt = (void *)params->buf;
3194         bool need_offchan = false;
3195         u32 flags;
3196         int ret;
3197
3198         if (params->dont_wait_for_ack)
3199                 flags = IEEE80211_TX_CTL_NO_ACK;
3200         else
3201                 flags = IEEE80211_TX_INTFL_NL80211_FRAME_TX |
3202                         IEEE80211_TX_CTL_REQ_TX_STATUS;
3203
3204         if (params->no_cck)
3205                 flags |= IEEE80211_TX_CTL_NO_CCK_RATE;
3206
3207         switch (sdata->vif.type) {
3208         case NL80211_IFTYPE_ADHOC:
3209                 if (!sdata->vif.bss_conf.ibss_joined)
3210                         need_offchan = true;
3211                 /* fall through */
3212 #ifdef CONFIG_MAC80211_MESH
3213         case NL80211_IFTYPE_MESH_POINT:
3214                 if (ieee80211_vif_is_mesh(&sdata->vif) &&
3215                     !sdata->u.mesh.mesh_id_len)
3216                         need_offchan = true;
3217                 /* fall through */
3218 #endif
3219         case NL80211_IFTYPE_AP:
3220         case NL80211_IFTYPE_AP_VLAN:
3221         case NL80211_IFTYPE_P2P_GO:
3222                 if (sdata->vif.type != NL80211_IFTYPE_ADHOC &&
3223                     !ieee80211_vif_is_mesh(&sdata->vif) &&
3224                     !rcu_access_pointer(sdata->bss->beacon))
3225                         need_offchan = true;
3226                 if (!ieee80211_is_action(mgmt->frame_control) ||
3227                     mgmt->u.action.category == WLAN_CATEGORY_PUBLIC ||
3228                     mgmt->u.action.category == WLAN_CATEGORY_SELF_PROTECTED ||
3229                     mgmt->u.action.category == WLAN_CATEGORY_SPECTRUM_MGMT)
3230                         break;
3231                 rcu_read_lock();
3232                 sta = sta_info_get(sdata, mgmt->da);
3233                 rcu_read_unlock();
3234                 if (!sta)
3235                         return -ENOLINK;
3236                 break;
3237         case NL80211_IFTYPE_STATION:
3238         case NL80211_IFTYPE_P2P_CLIENT:
3239                 if (!sdata->u.mgd.associated)
3240                         need_offchan = true;
3241                 break;
3242         case NL80211_IFTYPE_P2P_DEVICE:
3243                 need_offchan = true;
3244                 break;
3245         default:
3246                 return -EOPNOTSUPP;
3247         }
3248
3249         /* configurations requiring offchan cannot work if no channel has been
3250          * specified
3251          */
3252         if (need_offchan && !params->chan)
3253                 return -EINVAL;
3254
3255         mutex_lock(&local->mtx);
3256
3257         /* Check if the operating channel is the requested channel */
3258         if (!need_offchan) {
3259                 struct ieee80211_chanctx_conf *chanctx_conf;
3260
3261                 rcu_read_lock();
3262                 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
3263
3264                 if (chanctx_conf) {
3265                         need_offchan = params->chan &&
3266                                        (params->chan !=
3267                                         chanctx_conf->def.chan);
3268                 } else if (!params->chan) {
3269                         ret = -EINVAL;
3270                         rcu_read_unlock();
3271                         goto out_unlock;
3272                 } else {
3273                         need_offchan = true;
3274                 }
3275                 rcu_read_unlock();
3276         }
3277
3278         if (need_offchan && !params->offchan) {
3279                 ret = -EBUSY;
3280                 goto out_unlock;
3281         }
3282
3283         skb = dev_alloc_skb(local->hw.extra_tx_headroom + params->len);
3284         if (!skb) {
3285                 ret = -ENOMEM;
3286                 goto out_unlock;
3287         }
3288         skb_reserve(skb, local->hw.extra_tx_headroom);
3289
3290         memcpy(skb_put(skb, params->len), params->buf, params->len);
3291
3292         IEEE80211_SKB_CB(skb)->flags = flags;
3293
3294         skb->dev = sdata->dev;
3295
3296         if (!need_offchan) {
3297                 *cookie = (unsigned long) skb;
3298                 ieee80211_tx_skb(sdata, skb);
3299                 ret = 0;
3300                 goto out_unlock;
3301         }
3302
3303         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_CTL_TX_OFFCHAN |
3304                                         IEEE80211_TX_INTFL_OFFCHAN_TX_OK;
3305         if (local->hw.flags & IEEE80211_HW_QUEUE_CONTROL)
3306                 IEEE80211_SKB_CB(skb)->hw_queue =
3307                         local->hw.offchannel_tx_hw_queue;
3308
3309         /* This will handle all kinds of coalescing and immediate TX */
3310         ret = ieee80211_start_roc_work(local, sdata, params->chan,
3311                                        params->wait, cookie, skb,
3312                                        IEEE80211_ROC_TYPE_MGMT_TX);
3313         if (ret)
3314                 kfree_skb(skb);
3315  out_unlock:
3316         mutex_unlock(&local->mtx);
3317         return ret;
3318 }
3319
3320 static int ieee80211_mgmt_tx_cancel_wait(struct wiphy *wiphy,
3321                                          struct wireless_dev *wdev,
3322                                          u64 cookie)
3323 {
3324         struct ieee80211_local *local = wiphy_priv(wiphy);
3325
3326         return ieee80211_cancel_roc(local, cookie, true);
3327 }
3328
3329 static void ieee80211_mgmt_frame_register(struct wiphy *wiphy,
3330                                           struct wireless_dev *wdev,
3331                                           u16 frame_type, bool reg)
3332 {
3333         struct ieee80211_local *local = wiphy_priv(wiphy);
3334
3335         switch (frame_type) {
3336         case IEEE80211_FTYPE_MGMT | IEEE80211_STYPE_PROBE_REQ:
3337                 if (reg)
3338                         local->probe_req_reg++;
3339                 else
3340                         local->probe_req_reg--;
3341
3342                 if (!local->open_count)
3343                         break;
3344
3345                 ieee80211_queue_work(&local->hw, &local->reconfig_filter);
3346                 break;
3347         default:
3348                 break;
3349         }
3350 }
3351
3352 static int ieee80211_set_antenna(struct wiphy *wiphy, u32 tx_ant, u32 rx_ant)
3353 {
3354         struct ieee80211_local *local = wiphy_priv(wiphy);
3355
3356         if (local->started)
3357                 return -EOPNOTSUPP;
3358
3359         return drv_set_antenna(local, tx_ant, rx_ant);
3360 }
3361
3362 static int ieee80211_get_antenna(struct wiphy *wiphy, u32 *tx_ant, u32 *rx_ant)
3363 {
3364         struct ieee80211_local *local = wiphy_priv(wiphy);
3365
3366         return drv_get_antenna(local, tx_ant, rx_ant);
3367 }
3368
3369 static int ieee80211_set_ringparam(struct wiphy *wiphy, u32 tx, u32 rx)
3370 {
3371         struct ieee80211_local *local = wiphy_priv(wiphy);
3372
3373         return drv_set_ringparam(local, tx, rx);
3374 }
3375
3376 static void ieee80211_get_ringparam(struct wiphy *wiphy,
3377                                     u32 *tx, u32 *tx_max, u32 *rx, u32 *rx_max)
3378 {
3379         struct ieee80211_local *local = wiphy_priv(wiphy);
3380
3381         drv_get_ringparam(local, tx, tx_max, rx, rx_max);
3382 }
3383
3384 static int ieee80211_set_rekey_data(struct wiphy *wiphy,
3385                                     struct net_device *dev,
3386                                     struct cfg80211_gtk_rekey_data *data)
3387 {
3388         struct ieee80211_local *local = wiphy_priv(wiphy);
3389         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3390
3391         if (!local->ops->set_rekey_data)
3392                 return -EOPNOTSUPP;
3393
3394         drv_set_rekey_data(local, sdata, data);
3395
3396         return 0;
3397 }
3398
3399 static void ieee80211_tdls_add_ext_capab(struct sk_buff *skb)
3400 {
3401         u8 *pos = (void *)skb_put(skb, 7);
3402
3403         *pos++ = WLAN_EID_EXT_CAPABILITY;
3404         *pos++ = 5; /* len */
3405         *pos++ = 0x0;
3406         *pos++ = 0x0;
3407         *pos++ = 0x0;
3408         *pos++ = 0x0;
3409         *pos++ = WLAN_EXT_CAPA5_TDLS_ENABLED;
3410 }
3411
3412 static u16 ieee80211_get_tdls_sta_capab(struct ieee80211_sub_if_data *sdata)
3413 {
3414         struct ieee80211_local *local = sdata->local;
3415         u16 capab;
3416
3417         capab = 0;
3418         if (ieee80211_get_sdata_band(sdata) != IEEE80211_BAND_2GHZ)
3419                 return capab;
3420
3421         if (!(local->hw.flags & IEEE80211_HW_2GHZ_SHORT_SLOT_INCAPABLE))
3422                 capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME;
3423         if (!(local->hw.flags & IEEE80211_HW_2GHZ_SHORT_PREAMBLE_INCAPABLE))
3424                 capab |= WLAN_CAPABILITY_SHORT_PREAMBLE;
3425
3426         return capab;
3427 }
3428
3429 static void ieee80211_tdls_add_link_ie(struct sk_buff *skb, u8 *src_addr,
3430                                        u8 *peer, u8 *bssid)
3431 {
3432         struct ieee80211_tdls_lnkie *lnkid;
3433
3434         lnkid = (void *)skb_put(skb, sizeof(struct ieee80211_tdls_lnkie));
3435
3436         lnkid->ie_type = WLAN_EID_LINK_ID;
3437         lnkid->ie_len = sizeof(struct ieee80211_tdls_lnkie) - 2;
3438
3439         memcpy(lnkid->bssid, bssid, ETH_ALEN);
3440         memcpy(lnkid->init_sta, src_addr, ETH_ALEN);
3441         memcpy(lnkid->resp_sta, peer, ETH_ALEN);
3442 }
3443
3444 static int
3445 ieee80211_prep_tdls_encap_data(struct wiphy *wiphy, struct net_device *dev,
3446                                u8 *peer, u8 action_code, u8 dialog_token,
3447                                u16 status_code, struct sk_buff *skb)
3448 {
3449         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3450         enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
3451         struct ieee80211_tdls_data *tf;
3452
3453         tf = (void *)skb_put(skb, offsetof(struct ieee80211_tdls_data, u));
3454
3455         memcpy(tf->da, peer, ETH_ALEN);
3456         memcpy(tf->sa, sdata->vif.addr, ETH_ALEN);
3457         tf->ether_type = cpu_to_be16(ETH_P_TDLS);
3458         tf->payload_type = WLAN_TDLS_SNAP_RFTYPE;
3459
3460         switch (action_code) {
3461         case WLAN_TDLS_SETUP_REQUEST:
3462                 tf->category = WLAN_CATEGORY_TDLS;
3463                 tf->action_code = WLAN_TDLS_SETUP_REQUEST;
3464
3465                 skb_put(skb, sizeof(tf->u.setup_req));
3466                 tf->u.setup_req.dialog_token = dialog_token;
3467                 tf->u.setup_req.capability =
3468                         cpu_to_le16(ieee80211_get_tdls_sta_capab(sdata));
3469
3470                 ieee80211_add_srates_ie(sdata, skb, false, band);
3471                 ieee80211_add_ext_srates_ie(sdata, skb, false, band);
3472                 ieee80211_tdls_add_ext_capab(skb);
3473                 break;
3474         case WLAN_TDLS_SETUP_RESPONSE:
3475                 tf->category = WLAN_CATEGORY_TDLS;
3476                 tf->action_code = WLAN_TDLS_SETUP_RESPONSE;
3477
3478                 skb_put(skb, sizeof(tf->u.setup_resp));
3479                 tf->u.setup_resp.status_code = cpu_to_le16(status_code);
3480                 tf->u.setup_resp.dialog_token = dialog_token;
3481                 tf->u.setup_resp.capability =
3482                         cpu_to_le16(ieee80211_get_tdls_sta_capab(sdata));
3483
3484                 ieee80211_add_srates_ie(sdata, skb, false, band);
3485                 ieee80211_add_ext_srates_ie(sdata, skb, false, band);
3486                 ieee80211_tdls_add_ext_capab(skb);
3487                 break;
3488         case WLAN_TDLS_SETUP_CONFIRM:
3489                 tf->category = WLAN_CATEGORY_TDLS;
3490                 tf->action_code = WLAN_TDLS_SETUP_CONFIRM;
3491
3492                 skb_put(skb, sizeof(tf->u.setup_cfm));
3493                 tf->u.setup_cfm.status_code = cpu_to_le16(status_code);
3494                 tf->u.setup_cfm.dialog_token = dialog_token;
3495                 break;
3496         case WLAN_TDLS_TEARDOWN:
3497                 tf->category = WLAN_CATEGORY_TDLS;
3498                 tf->action_code = WLAN_TDLS_TEARDOWN;
3499
3500                 skb_put(skb, sizeof(tf->u.teardown));
3501                 tf->u.teardown.reason_code = cpu_to_le16(status_code);
3502                 break;
3503         case WLAN_TDLS_DISCOVERY_REQUEST:
3504                 tf->category = WLAN_CATEGORY_TDLS;
3505                 tf->action_code = WLAN_TDLS_DISCOVERY_REQUEST;
3506
3507                 skb_put(skb, sizeof(tf->u.discover_req));
3508                 tf->u.discover_req.dialog_token = dialog_token;
3509                 break;
3510         default:
3511                 return -EINVAL;
3512         }
3513
3514         return 0;
3515 }
3516
3517 static int
3518 ieee80211_prep_tdls_direct(struct wiphy *wiphy, struct net_device *dev,
3519                            u8 *peer, u8 action_code, u8 dialog_token,
3520                            u16 status_code, struct sk_buff *skb)
3521 {
3522         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3523         enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
3524         struct ieee80211_mgmt *mgmt;
3525
3526         mgmt = (void *)skb_put(skb, 24);
3527         memset(mgmt, 0, 24);
3528         memcpy(mgmt->da, peer, ETH_ALEN);
3529         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
3530         memcpy(mgmt->bssid, sdata->u.mgd.bssid, ETH_ALEN);
3531
3532         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
3533                                           IEEE80211_STYPE_ACTION);
3534
3535         switch (action_code) {
3536         case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
3537                 skb_put(skb, 1 + sizeof(mgmt->u.action.u.tdls_discover_resp));
3538                 mgmt->u.action.category = WLAN_CATEGORY_PUBLIC;
3539                 mgmt->u.action.u.tdls_discover_resp.action_code =
3540                         WLAN_PUB_ACTION_TDLS_DISCOVER_RES;
3541                 mgmt->u.action.u.tdls_discover_resp.dialog_token =
3542                         dialog_token;
3543                 mgmt->u.action.u.tdls_discover_resp.capability =
3544                         cpu_to_le16(ieee80211_get_tdls_sta_capab(sdata));
3545
3546                 ieee80211_add_srates_ie(sdata, skb, false, band);
3547                 ieee80211_add_ext_srates_ie(sdata, skb, false, band);
3548                 ieee80211_tdls_add_ext_capab(skb);
3549                 break;
3550         default:
3551                 return -EINVAL;
3552         }
3553
3554         return 0;
3555 }
3556
3557 static int ieee80211_tdls_mgmt(struct wiphy *wiphy, struct net_device *dev,
3558                                u8 *peer, u8 action_code, u8 dialog_token,
3559                                u16 status_code, const u8 *extra_ies,
3560                                size_t extra_ies_len)
3561 {
3562         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3563         struct ieee80211_local *local = sdata->local;
3564         struct sk_buff *skb = NULL;
3565         bool send_direct;
3566         int ret;
3567
3568         if (!(wiphy->flags & WIPHY_FLAG_SUPPORTS_TDLS))
3569                 return -ENOTSUPP;
3570
3571         /* make sure we are in managed mode, and associated */
3572         if (sdata->vif.type != NL80211_IFTYPE_STATION ||
3573             !sdata->u.mgd.associated)
3574                 return -EINVAL;
3575
3576         tdls_dbg(sdata, "TDLS mgmt action %d peer %pM\n",
3577                  action_code, peer);
3578
3579         skb = dev_alloc_skb(local->hw.extra_tx_headroom +
3580                             max(sizeof(struct ieee80211_mgmt),
3581                                 sizeof(struct ieee80211_tdls_data)) +
3582                             50 + /* supported rates */
3583                             7 + /* ext capab */
3584                             extra_ies_len +
3585                             sizeof(struct ieee80211_tdls_lnkie));
3586         if (!skb)
3587                 return -ENOMEM;
3588
3589         skb_reserve(skb, local->hw.extra_tx_headroom);
3590
3591         switch (action_code) {
3592         case WLAN_TDLS_SETUP_REQUEST:
3593         case WLAN_TDLS_SETUP_RESPONSE:
3594         case WLAN_TDLS_SETUP_CONFIRM:
3595         case WLAN_TDLS_TEARDOWN:
3596         case WLAN_TDLS_DISCOVERY_REQUEST:
3597                 ret = ieee80211_prep_tdls_encap_data(wiphy, dev, peer,
3598                                                      action_code, dialog_token,
3599                                                      status_code, skb);
3600                 send_direct = false;
3601                 break;
3602         case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
3603                 ret = ieee80211_prep_tdls_direct(wiphy, dev, peer, action_code,
3604                                                  dialog_token, status_code,
3605                                                  skb);
3606                 send_direct = true;
3607                 break;
3608         default:
3609                 ret = -ENOTSUPP;
3610                 break;
3611         }
3612
3613         if (ret < 0)
3614                 goto fail;
3615
3616         if (extra_ies_len)
3617                 memcpy(skb_put(skb, extra_ies_len), extra_ies, extra_ies_len);
3618
3619         /* the TDLS link IE is always added last */
3620         switch (action_code) {
3621         case WLAN_TDLS_SETUP_REQUEST:
3622         case WLAN_TDLS_SETUP_CONFIRM:
3623         case WLAN_TDLS_TEARDOWN:
3624         case WLAN_TDLS_DISCOVERY_REQUEST:
3625                 /* we are the initiator */
3626                 ieee80211_tdls_add_link_ie(skb, sdata->vif.addr, peer,
3627                                            sdata->u.mgd.bssid);
3628                 break;
3629         case WLAN_TDLS_SETUP_RESPONSE:
3630         case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
3631                 /* we are the responder */
3632                 ieee80211_tdls_add_link_ie(skb, peer, sdata->vif.addr,
3633                                            sdata->u.mgd.bssid);
3634                 break;
3635         default:
3636                 ret = -ENOTSUPP;
3637                 goto fail;
3638         }
3639
3640         if (send_direct) {
3641                 ieee80211_tx_skb(sdata, skb);
3642                 return 0;
3643         }
3644
3645         /*
3646          * According to 802.11z: Setup req/resp are sent in AC_BK, otherwise
3647          * we should default to AC_VI.
3648          */
3649         switch (action_code) {
3650         case WLAN_TDLS_SETUP_REQUEST:
3651         case WLAN_TDLS_SETUP_RESPONSE:
3652                 skb_set_queue_mapping(skb, IEEE80211_AC_BK);
3653                 skb->priority = 2;
3654                 break;
3655         default:
3656                 skb_set_queue_mapping(skb, IEEE80211_AC_VI);
3657                 skb->priority = 5;
3658                 break;
3659         }
3660
3661         /* disable bottom halves when entering the Tx path */
3662         local_bh_disable();
3663         ret = ieee80211_subif_start_xmit(skb, dev);
3664         local_bh_enable();
3665
3666         return ret;
3667
3668 fail:
3669         dev_kfree_skb(skb);
3670         return ret;
3671 }
3672
3673 static int ieee80211_tdls_oper(struct wiphy *wiphy, struct net_device *dev,
3674                                u8 *peer, enum nl80211_tdls_operation oper)
3675 {
3676         struct sta_info *sta;
3677         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3678
3679         if (!(wiphy->flags & WIPHY_FLAG_SUPPORTS_TDLS))
3680                 return -ENOTSUPP;
3681
3682         if (sdata->vif.type != NL80211_IFTYPE_STATION)
3683                 return -EINVAL;
3684
3685         tdls_dbg(sdata, "TDLS oper %d peer %pM\n", oper, peer);
3686
3687         switch (oper) {
3688         case NL80211_TDLS_ENABLE_LINK:
3689                 rcu_read_lock();
3690                 sta = sta_info_get(sdata, peer);
3691                 if (!sta) {
3692                         rcu_read_unlock();
3693                         return -ENOLINK;
3694                 }
3695
3696                 set_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH);
3697                 rcu_read_unlock();
3698                 break;
3699         case NL80211_TDLS_DISABLE_LINK:
3700                 return sta_info_destroy_addr(sdata, peer);
3701         case NL80211_TDLS_TEARDOWN:
3702         case NL80211_TDLS_SETUP:
3703         case NL80211_TDLS_DISCOVERY_REQ:
3704                 /* We don't support in-driver setup/teardown/discovery */
3705                 return -ENOTSUPP;
3706         default:
3707                 return -ENOTSUPP;
3708         }
3709
3710         return 0;
3711 }
3712
3713 static int ieee80211_probe_client(struct wiphy *wiphy, struct net_device *dev,
3714                                   const u8 *peer, u64 *cookie)
3715 {
3716         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3717         struct ieee80211_local *local = sdata->local;
3718         struct ieee80211_qos_hdr *nullfunc;
3719         struct sk_buff *skb;
3720         int size = sizeof(*nullfunc);
3721         __le16 fc;
3722         bool qos;
3723         struct ieee80211_tx_info *info;
3724         struct sta_info *sta;
3725         struct ieee80211_chanctx_conf *chanctx_conf;
3726         enum ieee80211_band band;
3727
3728         rcu_read_lock();
3729         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
3730         if (WARN_ON(!chanctx_conf)) {
3731                 rcu_read_unlock();
3732                 return -EINVAL;
3733         }
3734         band = chanctx_conf->def.chan->band;
3735         sta = sta_info_get_bss(sdata, peer);
3736         if (sta) {
3737                 qos = test_sta_flag(sta, WLAN_STA_WME);
3738         } else {
3739                 rcu_read_unlock();
3740                 return -ENOLINK;
3741         }
3742
3743         if (qos) {
3744                 fc = cpu_to_le16(IEEE80211_FTYPE_DATA |
3745                                  IEEE80211_STYPE_QOS_NULLFUNC |
3746                                  IEEE80211_FCTL_FROMDS);
3747         } else {
3748                 size -= 2;
3749                 fc = cpu_to_le16(IEEE80211_FTYPE_DATA |
3750                                  IEEE80211_STYPE_NULLFUNC |
3751                                  IEEE80211_FCTL_FROMDS);
3752         }
3753
3754         skb = dev_alloc_skb(local->hw.extra_tx_headroom + size);
3755         if (!skb) {
3756                 rcu_read_unlock();
3757                 return -ENOMEM;
3758         }
3759
3760         skb->dev = dev;
3761
3762         skb_reserve(skb, local->hw.extra_tx_headroom);
3763
3764         nullfunc = (void *) skb_put(skb, size);
3765         nullfunc->frame_control = fc;
3766         nullfunc->duration_id = 0;
3767         memcpy(nullfunc->addr1, sta->sta.addr, ETH_ALEN);
3768         memcpy(nullfunc->addr2, sdata->vif.addr, ETH_ALEN);
3769         memcpy(nullfunc->addr3, sdata->vif.addr, ETH_ALEN);
3770         nullfunc->seq_ctrl = 0;
3771
3772         info = IEEE80211_SKB_CB(skb);
3773
3774         info->flags |= IEEE80211_TX_CTL_REQ_TX_STATUS |
3775                        IEEE80211_TX_INTFL_NL80211_FRAME_TX;
3776
3777         skb_set_queue_mapping(skb, IEEE80211_AC_VO);
3778         skb->priority = 7;
3779         if (qos)
3780                 nullfunc->qos_ctrl = cpu_to_le16(7);
3781
3782         local_bh_disable();
3783         ieee80211_xmit(sdata, skb, band);
3784         local_bh_enable();
3785         rcu_read_unlock();
3786
3787         *cookie = (unsigned long) skb;
3788         return 0;
3789 }
3790
3791 static int ieee80211_cfg_get_channel(struct wiphy *wiphy,
3792                                      struct wireless_dev *wdev,
3793                                      struct cfg80211_chan_def *chandef)
3794 {
3795         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
3796         struct ieee80211_local *local = wiphy_priv(wiphy);
3797         struct ieee80211_chanctx_conf *chanctx_conf;
3798         int ret = -ENODATA;
3799
3800         rcu_read_lock();
3801         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
3802         if (chanctx_conf) {
3803                 *chandef = chanctx_conf->def;
3804                 ret = 0;
3805         } else if (local->open_count > 0 &&
3806                    local->open_count == local->monitors &&
3807                    sdata->vif.type == NL80211_IFTYPE_MONITOR) {
3808                 if (local->use_chanctx)
3809                         *chandef = local->monitor_chandef;
3810                 else
3811                         *chandef = local->_oper_chandef;
3812                 ret = 0;
3813         }
3814         rcu_read_unlock();
3815
3816         return ret;
3817 }
3818
3819 #ifdef CONFIG_PM
3820 static void ieee80211_set_wakeup(struct wiphy *wiphy, bool enabled)
3821 {
3822         drv_set_wakeup(wiphy_priv(wiphy), enabled);
3823 }
3824 #endif
3825
3826 struct cfg80211_ops mac80211_config_ops = {
3827         .add_virtual_intf = ieee80211_add_iface,
3828         .del_virtual_intf = ieee80211_del_iface,
3829         .change_virtual_intf = ieee80211_change_iface,
3830         .start_p2p_device = ieee80211_start_p2p_device,
3831         .stop_p2p_device = ieee80211_stop_p2p_device,
3832         .add_key = ieee80211_add_key,
3833         .del_key = ieee80211_del_key,
3834         .get_key = ieee80211_get_key,
3835         .set_default_key = ieee80211_config_default_key,
3836         .set_default_mgmt_key = ieee80211_config_default_mgmt_key,
3837         .start_ap = ieee80211_start_ap,
3838         .change_beacon = ieee80211_change_beacon,
3839         .stop_ap = ieee80211_stop_ap,
3840         .add_station = ieee80211_add_station,
3841         .del_station = ieee80211_del_station,
3842         .change_station = ieee80211_change_station,
3843         .get_station = ieee80211_get_station,
3844         .dump_station = ieee80211_dump_station,
3845         .dump_survey = ieee80211_dump_survey,
3846 #ifdef CONFIG_MAC80211_MESH
3847         .add_mpath = ieee80211_add_mpath,
3848         .del_mpath = ieee80211_del_mpath,
3849         .change_mpath = ieee80211_change_mpath,
3850         .get_mpath = ieee80211_get_mpath,
3851         .dump_mpath = ieee80211_dump_mpath,
3852         .update_mesh_config = ieee80211_update_mesh_config,
3853         .get_mesh_config = ieee80211_get_mesh_config,
3854         .join_mesh = ieee80211_join_mesh,
3855         .leave_mesh = ieee80211_leave_mesh,
3856 #endif
3857         .change_bss = ieee80211_change_bss,
3858         .set_txq_params = ieee80211_set_txq_params,
3859         .set_monitor_channel = ieee80211_set_monitor_channel,
3860         .suspend = ieee80211_suspend,
3861         .resume = ieee80211_resume,
3862         .scan = ieee80211_scan,
3863         .sched_scan_start = ieee80211_sched_scan_start,
3864         .sched_scan_stop = ieee80211_sched_scan_stop,
3865         .auth = ieee80211_auth,
3866         .assoc = ieee80211_assoc,
3867         .deauth = ieee80211_deauth,
3868         .disassoc = ieee80211_disassoc,
3869         .join_ibss = ieee80211_join_ibss,
3870         .leave_ibss = ieee80211_leave_ibss,
3871         .set_mcast_rate = ieee80211_set_mcast_rate,
3872         .set_wiphy_params = ieee80211_set_wiphy_params,
3873         .set_tx_power = ieee80211_set_tx_power,
3874         .get_tx_power = ieee80211_get_tx_power,
3875         .set_wds_peer = ieee80211_set_wds_peer,
3876         .rfkill_poll = ieee80211_rfkill_poll,
3877         CFG80211_TESTMODE_CMD(ieee80211_testmode_cmd)
3878         CFG80211_TESTMODE_DUMP(ieee80211_testmode_dump)
3879         .set_power_mgmt = ieee80211_set_power_mgmt,
3880         .set_bitrate_mask = ieee80211_set_bitrate_mask,
3881         .remain_on_channel = ieee80211_remain_on_channel,
3882         .cancel_remain_on_channel = ieee80211_cancel_remain_on_channel,
3883         .mgmt_tx = ieee80211_mgmt_tx,
3884         .mgmt_tx_cancel_wait = ieee80211_mgmt_tx_cancel_wait,
3885         .set_cqm_rssi_config = ieee80211_set_cqm_rssi_config,
3886         .mgmt_frame_register = ieee80211_mgmt_frame_register,
3887         .set_antenna = ieee80211_set_antenna,
3888         .get_antenna = ieee80211_get_antenna,
3889         .set_ringparam = ieee80211_set_ringparam,
3890         .get_ringparam = ieee80211_get_ringparam,
3891         .set_rekey_data = ieee80211_set_rekey_data,
3892         .tdls_oper = ieee80211_tdls_oper,
3893         .tdls_mgmt = ieee80211_tdls_mgmt,
3894         .probe_client = ieee80211_probe_client,
3895         .set_noack_map = ieee80211_set_noack_map,
3896 #ifdef CONFIG_PM
3897         .set_wakeup = ieee80211_set_wakeup,
3898 #endif
3899         .get_et_sset_count = ieee80211_get_et_sset_count,
3900         .get_et_stats = ieee80211_get_et_stats,
3901         .get_et_strings = ieee80211_get_et_strings,
3902         .get_channel = ieee80211_cfg_get_channel,
3903         .start_radar_detection = ieee80211_start_radar_detection,
3904         .channel_switch = ieee80211_channel_switch,
3905 };