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