Merge tag 'drm-misc-next-fixes-2023-06-29' of git://anongit.freedesktop.org/drm/drm...
[linux-2.6-microblaze.git] / net / mac80211 / cfg.c
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * mac80211 configuration hooks for cfg80211
4  *
5  * Copyright 2006-2010  Johannes Berg <johannes@sipsolutions.net>
6  * Copyright 2013-2015  Intel Mobile Communications GmbH
7  * Copyright (C) 2015-2017 Intel Deutschland GmbH
8  * Copyright (C) 2018-2022 Intel Corporation
9  */
10
11 #include <linux/ieee80211.h>
12 #include <linux/nl80211.h>
13 #include <linux/rtnetlink.h>
14 #include <linux/slab.h>
15 #include <net/net_namespace.h>
16 #include <linux/rcupdate.h>
17 #include <linux/fips.h>
18 #include <linux/if_ether.h>
19 #include <net/cfg80211.h>
20 #include "ieee80211_i.h"
21 #include "driver-ops.h"
22 #include "rate.h"
23 #include "mesh.h"
24 #include "wme.h"
25
26 static struct ieee80211_link_data *
27 ieee80211_link_or_deflink(struct ieee80211_sub_if_data *sdata, int link_id,
28                           bool require_valid)
29 {
30         struct ieee80211_link_data *link;
31
32         if (link_id < 0) {
33                 /*
34                  * For keys, if sdata is not an MLD, we might not use
35                  * the return value at all (if it's not a pairwise key),
36                  * so in that case (require_valid==false) don't error.
37                  */
38                 if (require_valid && sdata->vif.valid_links)
39                         return ERR_PTR(-EINVAL);
40
41                 return &sdata->deflink;
42         }
43
44         link = sdata_dereference(sdata->link[link_id], sdata);
45         if (!link)
46                 return ERR_PTR(-ENOLINK);
47         return link;
48 }
49
50 static void ieee80211_set_mu_mimo_follow(struct ieee80211_sub_if_data *sdata,
51                                          struct vif_params *params)
52 {
53         bool mu_mimo_groups = false;
54         bool mu_mimo_follow = false;
55
56         if (params->vht_mumimo_groups) {
57                 u64 membership;
58
59                 BUILD_BUG_ON(sizeof(membership) != WLAN_MEMBERSHIP_LEN);
60
61                 memcpy(sdata->vif.bss_conf.mu_group.membership,
62                        params->vht_mumimo_groups, WLAN_MEMBERSHIP_LEN);
63                 memcpy(sdata->vif.bss_conf.mu_group.position,
64                        params->vht_mumimo_groups + WLAN_MEMBERSHIP_LEN,
65                        WLAN_USER_POSITION_LEN);
66                 ieee80211_link_info_change_notify(sdata, &sdata->deflink,
67                                                   BSS_CHANGED_MU_GROUPS);
68                 /* don't care about endianness - just check for 0 */
69                 memcpy(&membership, params->vht_mumimo_groups,
70                        WLAN_MEMBERSHIP_LEN);
71                 mu_mimo_groups = membership != 0;
72         }
73
74         if (params->vht_mumimo_follow_addr) {
75                 mu_mimo_follow =
76                         is_valid_ether_addr(params->vht_mumimo_follow_addr);
77                 ether_addr_copy(sdata->u.mntr.mu_follow_addr,
78                                 params->vht_mumimo_follow_addr);
79         }
80
81         sdata->vif.bss_conf.mu_mimo_owner = mu_mimo_groups || mu_mimo_follow;
82 }
83
84 static int ieee80211_set_mon_options(struct ieee80211_sub_if_data *sdata,
85                                      struct vif_params *params)
86 {
87         struct ieee80211_local *local = sdata->local;
88         struct ieee80211_sub_if_data *monitor_sdata;
89
90         /* check flags first */
91         if (params->flags && ieee80211_sdata_running(sdata)) {
92                 u32 mask = MONITOR_FLAG_COOK_FRAMES | MONITOR_FLAG_ACTIVE;
93
94                 /*
95                  * Prohibit MONITOR_FLAG_COOK_FRAMES and
96                  * MONITOR_FLAG_ACTIVE to be changed while the
97                  * interface is up.
98                  * Else we would need to add a lot of cruft
99                  * to update everything:
100                  *      cooked_mntrs, monitor and all fif_* counters
101                  *      reconfigure hardware
102                  */
103                 if ((params->flags & mask) != (sdata->u.mntr.flags & mask))
104                         return -EBUSY;
105         }
106
107         /* also validate MU-MIMO change */
108         monitor_sdata = wiphy_dereference(local->hw.wiphy,
109                                           local->monitor_sdata);
110
111         if (!monitor_sdata &&
112             (params->vht_mumimo_groups || params->vht_mumimo_follow_addr))
113                 return -EOPNOTSUPP;
114
115         /* apply all changes now - no failures allowed */
116
117         if (monitor_sdata)
118                 ieee80211_set_mu_mimo_follow(monitor_sdata, params);
119
120         if (params->flags) {
121                 if (ieee80211_sdata_running(sdata)) {
122                         ieee80211_adjust_monitor_flags(sdata, -1);
123                         sdata->u.mntr.flags = params->flags;
124                         ieee80211_adjust_monitor_flags(sdata, 1);
125
126                         ieee80211_configure_filter(local);
127                 } else {
128                         /*
129                          * Because the interface is down, ieee80211_do_stop
130                          * and ieee80211_do_open take care of "everything"
131                          * mentioned in the comment above.
132                          */
133                         sdata->u.mntr.flags = params->flags;
134                 }
135         }
136
137         return 0;
138 }
139
140 static int ieee80211_set_ap_mbssid_options(struct ieee80211_sub_if_data *sdata,
141                                            struct cfg80211_mbssid_config params,
142                                            struct ieee80211_bss_conf *link_conf)
143 {
144         struct ieee80211_sub_if_data *tx_sdata;
145
146         sdata->vif.mbssid_tx_vif = NULL;
147         link_conf->bssid_index = 0;
148         link_conf->nontransmitted = false;
149         link_conf->ema_ap = false;
150         link_conf->bssid_indicator = 0;
151
152         if (sdata->vif.type != NL80211_IFTYPE_AP || !params.tx_wdev)
153                 return -EINVAL;
154
155         tx_sdata = IEEE80211_WDEV_TO_SUB_IF(params.tx_wdev);
156         if (!tx_sdata)
157                 return -EINVAL;
158
159         if (tx_sdata == sdata) {
160                 sdata->vif.mbssid_tx_vif = &sdata->vif;
161         } else {
162                 sdata->vif.mbssid_tx_vif = &tx_sdata->vif;
163                 link_conf->nontransmitted = true;
164                 link_conf->bssid_index = params.index;
165         }
166         if (params.ema)
167                 link_conf->ema_ap = true;
168
169         return 0;
170 }
171
172 static struct wireless_dev *ieee80211_add_iface(struct wiphy *wiphy,
173                                                 const char *name,
174                                                 unsigned char name_assign_type,
175                                                 enum nl80211_iftype type,
176                                                 struct vif_params *params)
177 {
178         struct ieee80211_local *local = wiphy_priv(wiphy);
179         struct wireless_dev *wdev;
180         struct ieee80211_sub_if_data *sdata;
181         int err;
182
183         err = ieee80211_if_add(local, name, name_assign_type, &wdev, type, params);
184         if (err)
185                 return ERR_PTR(err);
186
187         sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
188
189         if (type == NL80211_IFTYPE_MONITOR) {
190                 err = ieee80211_set_mon_options(sdata, params);
191                 if (err) {
192                         ieee80211_if_remove(sdata);
193                         return NULL;
194                 }
195         }
196
197         return wdev;
198 }
199
200 static int ieee80211_del_iface(struct wiphy *wiphy, struct wireless_dev *wdev)
201 {
202         ieee80211_if_remove(IEEE80211_WDEV_TO_SUB_IF(wdev));
203
204         return 0;
205 }
206
207 static int ieee80211_change_iface(struct wiphy *wiphy,
208                                   struct net_device *dev,
209                                   enum nl80211_iftype type,
210                                   struct vif_params *params)
211 {
212         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
213         struct ieee80211_local *local = sdata->local;
214         struct sta_info *sta;
215         int ret;
216
217         ret = ieee80211_if_change_type(sdata, type);
218         if (ret)
219                 return ret;
220
221         if (type == NL80211_IFTYPE_AP_VLAN && params->use_4addr == 0) {
222                 RCU_INIT_POINTER(sdata->u.vlan.sta, NULL);
223                 ieee80211_check_fast_rx_iface(sdata);
224         } else if (type == NL80211_IFTYPE_STATION && params->use_4addr >= 0) {
225                 struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
226
227                 if (params->use_4addr == ifmgd->use_4addr)
228                         return 0;
229
230                 /* FIXME: no support for 4-addr MLO yet */
231                 if (sdata->vif.valid_links)
232                         return -EOPNOTSUPP;
233
234                 sdata->u.mgd.use_4addr = params->use_4addr;
235                 if (!ifmgd->associated)
236                         return 0;
237
238                 mutex_lock(&local->sta_mtx);
239                 sta = sta_info_get(sdata, sdata->deflink.u.mgd.bssid);
240                 if (sta)
241                         drv_sta_set_4addr(local, sdata, &sta->sta,
242                                           params->use_4addr);
243                 mutex_unlock(&local->sta_mtx);
244
245                 if (params->use_4addr)
246                         ieee80211_send_4addr_nullfunc(local, sdata);
247         }
248
249         if (sdata->vif.type == NL80211_IFTYPE_MONITOR) {
250                 ret = ieee80211_set_mon_options(sdata, params);
251                 if (ret)
252                         return ret;
253         }
254
255         return 0;
256 }
257
258 static int ieee80211_start_p2p_device(struct wiphy *wiphy,
259                                       struct wireless_dev *wdev)
260 {
261         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
262         int ret;
263
264         mutex_lock(&sdata->local->chanctx_mtx);
265         ret = ieee80211_check_combinations(sdata, NULL, 0, 0);
266         mutex_unlock(&sdata->local->chanctx_mtx);
267         if (ret < 0)
268                 return ret;
269
270         return ieee80211_do_open(wdev, true);
271 }
272
273 static void ieee80211_stop_p2p_device(struct wiphy *wiphy,
274                                       struct wireless_dev *wdev)
275 {
276         ieee80211_sdata_stop(IEEE80211_WDEV_TO_SUB_IF(wdev));
277 }
278
279 static int ieee80211_start_nan(struct wiphy *wiphy,
280                                struct wireless_dev *wdev,
281                                struct cfg80211_nan_conf *conf)
282 {
283         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
284         int ret;
285
286         mutex_lock(&sdata->local->chanctx_mtx);
287         ret = ieee80211_check_combinations(sdata, NULL, 0, 0);
288         mutex_unlock(&sdata->local->chanctx_mtx);
289         if (ret < 0)
290                 return ret;
291
292         ret = ieee80211_do_open(wdev, true);
293         if (ret)
294                 return ret;
295
296         ret = drv_start_nan(sdata->local, sdata, conf);
297         if (ret)
298                 ieee80211_sdata_stop(sdata);
299
300         sdata->u.nan.conf = *conf;
301
302         return ret;
303 }
304
305 static void ieee80211_stop_nan(struct wiphy *wiphy,
306                                struct wireless_dev *wdev)
307 {
308         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
309
310         drv_stop_nan(sdata->local, sdata);
311         ieee80211_sdata_stop(sdata);
312 }
313
314 static int ieee80211_nan_change_conf(struct wiphy *wiphy,
315                                      struct wireless_dev *wdev,
316                                      struct cfg80211_nan_conf *conf,
317                                      u32 changes)
318 {
319         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
320         struct cfg80211_nan_conf new_conf;
321         int ret = 0;
322
323         if (sdata->vif.type != NL80211_IFTYPE_NAN)
324                 return -EOPNOTSUPP;
325
326         if (!ieee80211_sdata_running(sdata))
327                 return -ENETDOWN;
328
329         new_conf = sdata->u.nan.conf;
330
331         if (changes & CFG80211_NAN_CONF_CHANGED_PREF)
332                 new_conf.master_pref = conf->master_pref;
333
334         if (changes & CFG80211_NAN_CONF_CHANGED_BANDS)
335                 new_conf.bands = conf->bands;
336
337         ret = drv_nan_change_conf(sdata->local, sdata, &new_conf, changes);
338         if (!ret)
339                 sdata->u.nan.conf = new_conf;
340
341         return ret;
342 }
343
344 static int ieee80211_add_nan_func(struct wiphy *wiphy,
345                                   struct wireless_dev *wdev,
346                                   struct cfg80211_nan_func *nan_func)
347 {
348         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
349         int ret;
350
351         if (sdata->vif.type != NL80211_IFTYPE_NAN)
352                 return -EOPNOTSUPP;
353
354         if (!ieee80211_sdata_running(sdata))
355                 return -ENETDOWN;
356
357         spin_lock_bh(&sdata->u.nan.func_lock);
358
359         ret = idr_alloc(&sdata->u.nan.function_inst_ids,
360                         nan_func, 1, sdata->local->hw.max_nan_de_entries + 1,
361                         GFP_ATOMIC);
362         spin_unlock_bh(&sdata->u.nan.func_lock);
363
364         if (ret < 0)
365                 return ret;
366
367         nan_func->instance_id = ret;
368
369         WARN_ON(nan_func->instance_id == 0);
370
371         ret = drv_add_nan_func(sdata->local, sdata, nan_func);
372         if (ret) {
373                 spin_lock_bh(&sdata->u.nan.func_lock);
374                 idr_remove(&sdata->u.nan.function_inst_ids,
375                            nan_func->instance_id);
376                 spin_unlock_bh(&sdata->u.nan.func_lock);
377         }
378
379         return ret;
380 }
381
382 static struct cfg80211_nan_func *
383 ieee80211_find_nan_func_by_cookie(struct ieee80211_sub_if_data *sdata,
384                                   u64 cookie)
385 {
386         struct cfg80211_nan_func *func;
387         int id;
388
389         lockdep_assert_held(&sdata->u.nan.func_lock);
390
391         idr_for_each_entry(&sdata->u.nan.function_inst_ids, func, id) {
392                 if (func->cookie == cookie)
393                         return func;
394         }
395
396         return NULL;
397 }
398
399 static void ieee80211_del_nan_func(struct wiphy *wiphy,
400                                   struct wireless_dev *wdev, u64 cookie)
401 {
402         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
403         struct cfg80211_nan_func *func;
404         u8 instance_id = 0;
405
406         if (sdata->vif.type != NL80211_IFTYPE_NAN ||
407             !ieee80211_sdata_running(sdata))
408                 return;
409
410         spin_lock_bh(&sdata->u.nan.func_lock);
411
412         func = ieee80211_find_nan_func_by_cookie(sdata, cookie);
413         if (func)
414                 instance_id = func->instance_id;
415
416         spin_unlock_bh(&sdata->u.nan.func_lock);
417
418         if (instance_id)
419                 drv_del_nan_func(sdata->local, sdata, instance_id);
420 }
421
422 static int ieee80211_set_noack_map(struct wiphy *wiphy,
423                                   struct net_device *dev,
424                                   u16 noack_map)
425 {
426         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
427
428         sdata->noack_map = noack_map;
429
430         ieee80211_check_fast_xmit_iface(sdata);
431
432         return 0;
433 }
434
435 static int ieee80211_set_tx(struct ieee80211_sub_if_data *sdata,
436                             const u8 *mac_addr, u8 key_idx)
437 {
438         struct ieee80211_local *local = sdata->local;
439         struct ieee80211_key *key;
440         struct sta_info *sta;
441         int ret = -EINVAL;
442
443         if (!wiphy_ext_feature_isset(local->hw.wiphy,
444                                      NL80211_EXT_FEATURE_EXT_KEY_ID))
445                 return -EINVAL;
446
447         sta = sta_info_get_bss(sdata, mac_addr);
448
449         if (!sta)
450                 return -EINVAL;
451
452         if (sta->ptk_idx == key_idx)
453                 return 0;
454
455         mutex_lock(&local->key_mtx);
456         key = key_mtx_dereference(local, sta->ptk[key_idx]);
457
458         if (key && key->conf.flags & IEEE80211_KEY_FLAG_NO_AUTO_TX)
459                 ret = ieee80211_set_tx_key(key);
460
461         mutex_unlock(&local->key_mtx);
462         return ret;
463 }
464
465 static int ieee80211_add_key(struct wiphy *wiphy, struct net_device *dev,
466                              int link_id, u8 key_idx, bool pairwise,
467                              const u8 *mac_addr, struct key_params *params)
468 {
469         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
470         struct ieee80211_link_data *link =
471                 ieee80211_link_or_deflink(sdata, link_id, false);
472         struct ieee80211_local *local = sdata->local;
473         struct sta_info *sta = NULL;
474         struct ieee80211_key *key;
475         int err;
476
477         if (!ieee80211_sdata_running(sdata))
478                 return -ENETDOWN;
479
480         if (IS_ERR(link))
481                 return PTR_ERR(link);
482
483         if (pairwise && params->mode == NL80211_KEY_SET_TX)
484                 return ieee80211_set_tx(sdata, mac_addr, key_idx);
485
486         /* reject WEP and TKIP keys if WEP failed to initialize */
487         switch (params->cipher) {
488         case WLAN_CIPHER_SUITE_WEP40:
489         case WLAN_CIPHER_SUITE_TKIP:
490         case WLAN_CIPHER_SUITE_WEP104:
491                 if (link_id >= 0)
492                         return -EINVAL;
493                 if (WARN_ON_ONCE(fips_enabled))
494                         return -EINVAL;
495                 break;
496         default:
497                 break;
498         }
499
500         key = ieee80211_key_alloc(params->cipher, key_idx, params->key_len,
501                                   params->key, params->seq_len, params->seq);
502         if (IS_ERR(key))
503                 return PTR_ERR(key);
504
505         key->conf.link_id = link_id;
506
507         if (pairwise)
508                 key->conf.flags |= IEEE80211_KEY_FLAG_PAIRWISE;
509
510         if (params->mode == NL80211_KEY_NO_TX)
511                 key->conf.flags |= IEEE80211_KEY_FLAG_NO_AUTO_TX;
512
513         mutex_lock(&local->sta_mtx);
514
515         if (mac_addr) {
516                 sta = sta_info_get_bss(sdata, mac_addr);
517                 /*
518                  * The ASSOC test makes sure the driver is ready to
519                  * receive the key. When wpa_supplicant has roamed
520                  * using FT, it attempts to set the key before
521                  * association has completed, this rejects that attempt
522                  * so it will set the key again after association.
523                  *
524                  * TODO: accept the key if we have a station entry and
525                  *       add it to the device after the station.
526                  */
527                 if (!sta || !test_sta_flag(sta, WLAN_STA_ASSOC)) {
528                         ieee80211_key_free_unused(key);
529                         err = -ENOENT;
530                         goto out_unlock;
531                 }
532         }
533
534         switch (sdata->vif.type) {
535         case NL80211_IFTYPE_STATION:
536                 if (sdata->u.mgd.mfp != IEEE80211_MFP_DISABLED)
537                         key->conf.flags |= IEEE80211_KEY_FLAG_RX_MGMT;
538                 break;
539         case NL80211_IFTYPE_AP:
540         case NL80211_IFTYPE_AP_VLAN:
541                 /* Keys without a station are used for TX only */
542                 if (sta && test_sta_flag(sta, WLAN_STA_MFP))
543                         key->conf.flags |= IEEE80211_KEY_FLAG_RX_MGMT;
544                 break;
545         case NL80211_IFTYPE_ADHOC:
546                 /* no MFP (yet) */
547                 break;
548         case NL80211_IFTYPE_MESH_POINT:
549 #ifdef CONFIG_MAC80211_MESH
550                 if (sdata->u.mesh.security != IEEE80211_MESH_SEC_NONE)
551                         key->conf.flags |= IEEE80211_KEY_FLAG_RX_MGMT;
552                 break;
553 #endif
554         case NL80211_IFTYPE_WDS:
555         case NL80211_IFTYPE_MONITOR:
556         case NL80211_IFTYPE_P2P_DEVICE:
557         case NL80211_IFTYPE_NAN:
558         case NL80211_IFTYPE_UNSPECIFIED:
559         case NUM_NL80211_IFTYPES:
560         case NL80211_IFTYPE_P2P_CLIENT:
561         case NL80211_IFTYPE_P2P_GO:
562         case NL80211_IFTYPE_OCB:
563                 /* shouldn't happen */
564                 WARN_ON_ONCE(1);
565                 break;
566         }
567
568         err = ieee80211_key_link(key, link, sta);
569
570  out_unlock:
571         mutex_unlock(&local->sta_mtx);
572
573         return err;
574 }
575
576 static struct ieee80211_key *
577 ieee80211_lookup_key(struct ieee80211_sub_if_data *sdata, int link_id,
578                      u8 key_idx, bool pairwise, const u8 *mac_addr)
579 {
580         struct ieee80211_local *local __maybe_unused = sdata->local;
581         struct ieee80211_link_data *link = &sdata->deflink;
582         struct ieee80211_key *key;
583
584         if (link_id >= 0) {
585                 link = rcu_dereference_check(sdata->link[link_id],
586                                              lockdep_is_held(&sdata->wdev.mtx));
587                 if (!link)
588                         return NULL;
589         }
590
591         if (mac_addr) {
592                 struct sta_info *sta;
593                 struct link_sta_info *link_sta;
594
595                 sta = sta_info_get_bss(sdata, mac_addr);
596                 if (!sta)
597                         return NULL;
598
599                 if (link_id >= 0) {
600                         link_sta = rcu_dereference_check(sta->link[link_id],
601                                                          lockdep_is_held(&local->sta_mtx));
602                         if (!link_sta)
603                                 return NULL;
604                 } else {
605                         link_sta = &sta->deflink;
606                 }
607
608                 if (pairwise && key_idx < NUM_DEFAULT_KEYS)
609                         return rcu_dereference_check_key_mtx(local,
610                                                              sta->ptk[key_idx]);
611
612                 if (!pairwise &&
613                     key_idx < NUM_DEFAULT_KEYS +
614                               NUM_DEFAULT_MGMT_KEYS +
615                               NUM_DEFAULT_BEACON_KEYS)
616                         return rcu_dereference_check_key_mtx(local,
617                                                              link_sta->gtk[key_idx]);
618
619                 return NULL;
620         }
621
622         if (pairwise && key_idx < NUM_DEFAULT_KEYS)
623                 return rcu_dereference_check_key_mtx(local,
624                                                      sdata->keys[key_idx]);
625
626         key = rcu_dereference_check_key_mtx(local, link->gtk[key_idx]);
627         if (key)
628                 return key;
629
630         /* or maybe it was a WEP key */
631         if (key_idx < NUM_DEFAULT_KEYS)
632                 return rcu_dereference_check_key_mtx(local, sdata->keys[key_idx]);
633
634         return NULL;
635 }
636
637 static int ieee80211_del_key(struct wiphy *wiphy, struct net_device *dev,
638                              int link_id, u8 key_idx, bool pairwise,
639                              const u8 *mac_addr)
640 {
641         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
642         struct ieee80211_local *local = sdata->local;
643         struct ieee80211_key *key;
644         int ret;
645
646         mutex_lock(&local->sta_mtx);
647         mutex_lock(&local->key_mtx);
648
649         key = ieee80211_lookup_key(sdata, link_id, key_idx, pairwise, mac_addr);
650         if (!key) {
651                 ret = -ENOENT;
652                 goto out_unlock;
653         }
654
655         ieee80211_key_free(key, sdata->vif.type == NL80211_IFTYPE_STATION);
656
657         ret = 0;
658  out_unlock:
659         mutex_unlock(&local->key_mtx);
660         mutex_unlock(&local->sta_mtx);
661
662         return ret;
663 }
664
665 static int ieee80211_get_key(struct wiphy *wiphy, struct net_device *dev,
666                              int link_id, u8 key_idx, bool pairwise,
667                              const u8 *mac_addr, void *cookie,
668                              void (*callback)(void *cookie,
669                                               struct key_params *params))
670 {
671         struct ieee80211_sub_if_data *sdata;
672         u8 seq[6] = {0};
673         struct key_params params;
674         struct ieee80211_key *key;
675         u64 pn64;
676         u32 iv32;
677         u16 iv16;
678         int err = -ENOENT;
679         struct ieee80211_key_seq kseq = {};
680
681         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
682
683         rcu_read_lock();
684
685         key = ieee80211_lookup_key(sdata, link_id, key_idx, pairwise, mac_addr);
686         if (!key)
687                 goto out;
688
689         memset(&params, 0, sizeof(params));
690
691         params.cipher = key->conf.cipher;
692
693         switch (key->conf.cipher) {
694         case WLAN_CIPHER_SUITE_TKIP:
695                 pn64 = atomic64_read(&key->conf.tx_pn);
696                 iv32 = TKIP_PN_TO_IV32(pn64);
697                 iv16 = TKIP_PN_TO_IV16(pn64);
698
699                 if (key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE &&
700                     !(key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV)) {
701                         drv_get_key_seq(sdata->local, key, &kseq);
702                         iv32 = kseq.tkip.iv32;
703                         iv16 = kseq.tkip.iv16;
704                 }
705
706                 seq[0] = iv16 & 0xff;
707                 seq[1] = (iv16 >> 8) & 0xff;
708                 seq[2] = iv32 & 0xff;
709                 seq[3] = (iv32 >> 8) & 0xff;
710                 seq[4] = (iv32 >> 16) & 0xff;
711                 seq[5] = (iv32 >> 24) & 0xff;
712                 params.seq = seq;
713                 params.seq_len = 6;
714                 break;
715         case WLAN_CIPHER_SUITE_CCMP:
716         case WLAN_CIPHER_SUITE_CCMP_256:
717         case WLAN_CIPHER_SUITE_AES_CMAC:
718         case WLAN_CIPHER_SUITE_BIP_CMAC_256:
719                 BUILD_BUG_ON(offsetof(typeof(kseq), ccmp) !=
720                              offsetof(typeof(kseq), aes_cmac));
721                 fallthrough;
722         case WLAN_CIPHER_SUITE_BIP_GMAC_128:
723         case WLAN_CIPHER_SUITE_BIP_GMAC_256:
724                 BUILD_BUG_ON(offsetof(typeof(kseq), ccmp) !=
725                              offsetof(typeof(kseq), aes_gmac));
726                 fallthrough;
727         case WLAN_CIPHER_SUITE_GCMP:
728         case WLAN_CIPHER_SUITE_GCMP_256:
729                 BUILD_BUG_ON(offsetof(typeof(kseq), ccmp) !=
730                              offsetof(typeof(kseq), gcmp));
731
732                 if (key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE &&
733                     !(key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV)) {
734                         drv_get_key_seq(sdata->local, key, &kseq);
735                         memcpy(seq, kseq.ccmp.pn, 6);
736                 } else {
737                         pn64 = atomic64_read(&key->conf.tx_pn);
738                         seq[0] = pn64;
739                         seq[1] = pn64 >> 8;
740                         seq[2] = pn64 >> 16;
741                         seq[3] = pn64 >> 24;
742                         seq[4] = pn64 >> 32;
743                         seq[5] = pn64 >> 40;
744                 }
745                 params.seq = seq;
746                 params.seq_len = 6;
747                 break;
748         default:
749                 if (!(key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE))
750                         break;
751                 if (WARN_ON(key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV))
752                         break;
753                 drv_get_key_seq(sdata->local, key, &kseq);
754                 params.seq = kseq.hw.seq;
755                 params.seq_len = kseq.hw.seq_len;
756                 break;
757         }
758
759         params.key = key->conf.key;
760         params.key_len = key->conf.keylen;
761
762         callback(cookie, &params);
763         err = 0;
764
765  out:
766         rcu_read_unlock();
767         return err;
768 }
769
770 static int ieee80211_config_default_key(struct wiphy *wiphy,
771                                         struct net_device *dev,
772                                         int link_id, u8 key_idx, bool uni,
773                                         bool multi)
774 {
775         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
776         struct ieee80211_link_data *link =
777                 ieee80211_link_or_deflink(sdata, link_id, false);
778
779         if (IS_ERR(link))
780                 return PTR_ERR(link);
781
782         ieee80211_set_default_key(link, key_idx, uni, multi);
783
784         return 0;
785 }
786
787 static int ieee80211_config_default_mgmt_key(struct wiphy *wiphy,
788                                              struct net_device *dev,
789                                              int link_id, u8 key_idx)
790 {
791         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
792         struct ieee80211_link_data *link =
793                 ieee80211_link_or_deflink(sdata, link_id, true);
794
795         if (IS_ERR(link))
796                 return PTR_ERR(link);
797
798         ieee80211_set_default_mgmt_key(link, key_idx);
799
800         return 0;
801 }
802
803 static int ieee80211_config_default_beacon_key(struct wiphy *wiphy,
804                                                struct net_device *dev,
805                                                int link_id, u8 key_idx)
806 {
807         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
808         struct ieee80211_link_data *link =
809                 ieee80211_link_or_deflink(sdata, link_id, true);
810
811         if (IS_ERR(link))
812                 return PTR_ERR(link);
813
814         ieee80211_set_default_beacon_key(link, key_idx);
815
816         return 0;
817 }
818
819 void sta_set_rate_info_tx(struct sta_info *sta,
820                           const struct ieee80211_tx_rate *rate,
821                           struct rate_info *rinfo)
822 {
823         rinfo->flags = 0;
824         if (rate->flags & IEEE80211_TX_RC_MCS) {
825                 rinfo->flags |= RATE_INFO_FLAGS_MCS;
826                 rinfo->mcs = rate->idx;
827         } else if (rate->flags & IEEE80211_TX_RC_VHT_MCS) {
828                 rinfo->flags |= RATE_INFO_FLAGS_VHT_MCS;
829                 rinfo->mcs = ieee80211_rate_get_vht_mcs(rate);
830                 rinfo->nss = ieee80211_rate_get_vht_nss(rate);
831         } else {
832                 struct ieee80211_supported_band *sband;
833                 int shift = ieee80211_vif_get_shift(&sta->sdata->vif);
834                 u16 brate;
835
836                 sband = ieee80211_get_sband(sta->sdata);
837                 WARN_ON_ONCE(sband && !sband->bitrates);
838                 if (sband && sband->bitrates) {
839                         brate = sband->bitrates[rate->idx].bitrate;
840                         rinfo->legacy = DIV_ROUND_UP(brate, 1 << shift);
841                 }
842         }
843         if (rate->flags & IEEE80211_TX_RC_40_MHZ_WIDTH)
844                 rinfo->bw = RATE_INFO_BW_40;
845         else if (rate->flags & IEEE80211_TX_RC_80_MHZ_WIDTH)
846                 rinfo->bw = RATE_INFO_BW_80;
847         else if (rate->flags & IEEE80211_TX_RC_160_MHZ_WIDTH)
848                 rinfo->bw = RATE_INFO_BW_160;
849         else
850                 rinfo->bw = RATE_INFO_BW_20;
851         if (rate->flags & IEEE80211_TX_RC_SHORT_GI)
852                 rinfo->flags |= RATE_INFO_FLAGS_SHORT_GI;
853 }
854
855 static int ieee80211_dump_station(struct wiphy *wiphy, struct net_device *dev,
856                                   int idx, u8 *mac, struct station_info *sinfo)
857 {
858         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
859         struct ieee80211_local *local = sdata->local;
860         struct sta_info *sta;
861         int ret = -ENOENT;
862
863         mutex_lock(&local->sta_mtx);
864
865         sta = sta_info_get_by_idx(sdata, idx);
866         if (sta) {
867                 ret = 0;
868                 memcpy(mac, sta->sta.addr, ETH_ALEN);
869                 sta_set_sinfo(sta, sinfo, true);
870         }
871
872         mutex_unlock(&local->sta_mtx);
873
874         return ret;
875 }
876
877 static int ieee80211_dump_survey(struct wiphy *wiphy, struct net_device *dev,
878                                  int idx, struct survey_info *survey)
879 {
880         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
881
882         return drv_get_survey(local, idx, survey);
883 }
884
885 static int ieee80211_get_station(struct wiphy *wiphy, struct net_device *dev,
886                                  const u8 *mac, struct station_info *sinfo)
887 {
888         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
889         struct ieee80211_local *local = sdata->local;
890         struct sta_info *sta;
891         int ret = -ENOENT;
892
893         mutex_lock(&local->sta_mtx);
894
895         sta = sta_info_get_bss(sdata, mac);
896         if (sta) {
897                 ret = 0;
898                 sta_set_sinfo(sta, sinfo, true);
899         }
900
901         mutex_unlock(&local->sta_mtx);
902
903         return ret;
904 }
905
906 static int ieee80211_set_monitor_channel(struct wiphy *wiphy,
907                                          struct cfg80211_chan_def *chandef)
908 {
909         struct ieee80211_local *local = wiphy_priv(wiphy);
910         struct ieee80211_sub_if_data *sdata;
911         int ret = 0;
912
913         if (cfg80211_chandef_identical(&local->monitor_chandef, chandef))
914                 return 0;
915
916         mutex_lock(&local->mtx);
917         if (local->use_chanctx) {
918                 sdata = wiphy_dereference(local->hw.wiphy,
919                                           local->monitor_sdata);
920                 if (sdata) {
921                         ieee80211_link_release_channel(&sdata->deflink);
922                         ret = ieee80211_link_use_channel(&sdata->deflink,
923                                                          chandef,
924                                                          IEEE80211_CHANCTX_EXCLUSIVE);
925                 }
926         } else if (local->open_count == local->monitors) {
927                 local->_oper_chandef = *chandef;
928                 ieee80211_hw_config(local, 0);
929         }
930
931         if (ret == 0)
932                 local->monitor_chandef = *chandef;
933         mutex_unlock(&local->mtx);
934
935         return ret;
936 }
937
938 static int
939 ieee80211_set_probe_resp(struct ieee80211_sub_if_data *sdata,
940                          const u8 *resp, size_t resp_len,
941                          const struct ieee80211_csa_settings *csa,
942                          const struct ieee80211_color_change_settings *cca,
943                          struct ieee80211_link_data *link)
944 {
945         struct probe_resp *new, *old;
946
947         if (!resp || !resp_len)
948                 return 1;
949
950         old = sdata_dereference(link->u.ap.probe_resp, sdata);
951
952         new = kzalloc(sizeof(struct probe_resp) + resp_len, GFP_KERNEL);
953         if (!new)
954                 return -ENOMEM;
955
956         new->len = resp_len;
957         memcpy(new->data, resp, resp_len);
958
959         if (csa)
960                 memcpy(new->cntdwn_counter_offsets, csa->counter_offsets_presp,
961                        csa->n_counter_offsets_presp *
962                        sizeof(new->cntdwn_counter_offsets[0]));
963         else if (cca)
964                 new->cntdwn_counter_offsets[0] = cca->counter_offset_presp;
965
966         rcu_assign_pointer(link->u.ap.probe_resp, new);
967         if (old)
968                 kfree_rcu(old, rcu_head);
969
970         return 0;
971 }
972
973 static int ieee80211_set_fils_discovery(struct ieee80211_sub_if_data *sdata,
974                                         struct cfg80211_fils_discovery *params,
975                                         struct ieee80211_link_data *link,
976                                         struct ieee80211_bss_conf *link_conf)
977 {
978         struct fils_discovery_data *new, *old = NULL;
979         struct ieee80211_fils_discovery *fd;
980
981         if (!params->tmpl || !params->tmpl_len)
982                 return -EINVAL;
983
984         fd = &link_conf->fils_discovery;
985         fd->min_interval = params->min_interval;
986         fd->max_interval = params->max_interval;
987
988         old = sdata_dereference(link->u.ap.fils_discovery, sdata);
989         new = kzalloc(sizeof(*new) + params->tmpl_len, GFP_KERNEL);
990         if (!new)
991                 return -ENOMEM;
992         new->len = params->tmpl_len;
993         memcpy(new->data, params->tmpl, params->tmpl_len);
994         rcu_assign_pointer(link->u.ap.fils_discovery, new);
995
996         if (old)
997                 kfree_rcu(old, rcu_head);
998
999         return 0;
1000 }
1001
1002 static int
1003 ieee80211_set_unsol_bcast_probe_resp(struct ieee80211_sub_if_data *sdata,
1004                                      struct cfg80211_unsol_bcast_probe_resp *params,
1005                                      struct ieee80211_link_data *link,
1006                                      struct ieee80211_bss_conf *link_conf)
1007 {
1008         struct unsol_bcast_probe_resp_data *new, *old = NULL;
1009
1010         if (!params->tmpl || !params->tmpl_len)
1011                 return -EINVAL;
1012
1013         old = sdata_dereference(link->u.ap.unsol_bcast_probe_resp, sdata);
1014         new = kzalloc(sizeof(*new) + params->tmpl_len, GFP_KERNEL);
1015         if (!new)
1016                 return -ENOMEM;
1017         new->len = params->tmpl_len;
1018         memcpy(new->data, params->tmpl, params->tmpl_len);
1019         rcu_assign_pointer(link->u.ap.unsol_bcast_probe_resp, new);
1020
1021         if (old)
1022                 kfree_rcu(old, rcu_head);
1023
1024         link_conf->unsol_bcast_probe_resp_interval = params->interval;
1025
1026         return 0;
1027 }
1028
1029 static int ieee80211_set_ftm_responder_params(
1030                                 struct ieee80211_sub_if_data *sdata,
1031                                 const u8 *lci, size_t lci_len,
1032                                 const u8 *civicloc, size_t civicloc_len,
1033                                 struct ieee80211_bss_conf *link_conf)
1034 {
1035         struct ieee80211_ftm_responder_params *new, *old;
1036         u8 *pos;
1037         int len;
1038
1039         if (!lci_len && !civicloc_len)
1040                 return 0;
1041
1042         old = link_conf->ftmr_params;
1043         len = lci_len + civicloc_len;
1044
1045         new = kzalloc(sizeof(*new) + len, GFP_KERNEL);
1046         if (!new)
1047                 return -ENOMEM;
1048
1049         pos = (u8 *)(new + 1);
1050         if (lci_len) {
1051                 new->lci_len = lci_len;
1052                 new->lci = pos;
1053                 memcpy(pos, lci, lci_len);
1054                 pos += lci_len;
1055         }
1056
1057         if (civicloc_len) {
1058                 new->civicloc_len = civicloc_len;
1059                 new->civicloc = pos;
1060                 memcpy(pos, civicloc, civicloc_len);
1061                 pos += civicloc_len;
1062         }
1063
1064         link_conf->ftmr_params = new;
1065         kfree(old);
1066
1067         return 0;
1068 }
1069
1070 static int
1071 ieee80211_copy_mbssid_beacon(u8 *pos, struct cfg80211_mbssid_elems *dst,
1072                              struct cfg80211_mbssid_elems *src)
1073 {
1074         int i, offset = 0;
1075
1076         for (i = 0; i < src->cnt; i++) {
1077                 memcpy(pos + offset, src->elem[i].data, src->elem[i].len);
1078                 dst->elem[i].len = src->elem[i].len;
1079                 dst->elem[i].data = pos + offset;
1080                 offset += dst->elem[i].len;
1081         }
1082         dst->cnt = src->cnt;
1083
1084         return offset;
1085 }
1086
1087 static int
1088 ieee80211_copy_rnr_beacon(u8 *pos, struct cfg80211_rnr_elems *dst,
1089                           struct cfg80211_rnr_elems *src)
1090 {
1091         int i, offset = 0;
1092
1093         for (i = 0; i < src->cnt; i++) {
1094                 memcpy(pos + offset, src->elem[i].data, src->elem[i].len);
1095                 dst->elem[i].len = src->elem[i].len;
1096                 dst->elem[i].data = pos + offset;
1097                 offset += dst->elem[i].len;
1098         }
1099         dst->cnt = src->cnt;
1100
1101         return offset;
1102 }
1103
1104 static int ieee80211_assign_beacon(struct ieee80211_sub_if_data *sdata,
1105                                    struct ieee80211_link_data *link,
1106                                    struct cfg80211_beacon_data *params,
1107                                    const struct ieee80211_csa_settings *csa,
1108                                    const struct ieee80211_color_change_settings *cca)
1109 {
1110         struct cfg80211_mbssid_elems *mbssid = NULL;
1111         struct cfg80211_rnr_elems *rnr = NULL;
1112         struct beacon_data *new, *old;
1113         int new_head_len, new_tail_len;
1114         int size, err;
1115         u32 changed = BSS_CHANGED_BEACON;
1116         struct ieee80211_bss_conf *link_conf = link->conf;
1117
1118         old = sdata_dereference(link->u.ap.beacon, sdata);
1119
1120         /* Need to have a beacon head if we don't have one yet */
1121         if (!params->head && !old)
1122                 return -EINVAL;
1123
1124         /* new or old head? */
1125         if (params->head)
1126                 new_head_len = params->head_len;
1127         else
1128                 new_head_len = old->head_len;
1129
1130         /* new or old tail? */
1131         if (params->tail || !old)
1132                 /* params->tail_len will be zero for !params->tail */
1133                 new_tail_len = params->tail_len;
1134         else
1135                 new_tail_len = old->tail_len;
1136
1137         size = sizeof(*new) + new_head_len + new_tail_len;
1138
1139         /* new or old multiple BSSID elements? */
1140         if (params->mbssid_ies) {
1141                 mbssid = params->mbssid_ies;
1142                 size += struct_size(new->mbssid_ies, elem, mbssid->cnt);
1143                 if (params->rnr_ies) {
1144                         rnr = params->rnr_ies;
1145                         size += struct_size(new->rnr_ies, elem, rnr->cnt);
1146                 }
1147                 size += ieee80211_get_mbssid_beacon_len(mbssid, rnr,
1148                                                         mbssid->cnt);
1149         } else if (old && old->mbssid_ies) {
1150                 mbssid = old->mbssid_ies;
1151                 size += struct_size(new->mbssid_ies, elem, mbssid->cnt);
1152                 if (old && old->rnr_ies) {
1153                         rnr = old->rnr_ies;
1154                         size += struct_size(new->rnr_ies, elem, rnr->cnt);
1155                 }
1156                 size += ieee80211_get_mbssid_beacon_len(mbssid, rnr,
1157                                                         mbssid->cnt);
1158         }
1159
1160         new = kzalloc(size, GFP_KERNEL);
1161         if (!new)
1162                 return -ENOMEM;
1163
1164         /* start filling the new info now */
1165
1166         /*
1167          * pointers go into the block we allocated,
1168          * memory is | beacon_data | head | tail | mbssid_ies | rnr_ies
1169          */
1170         new->head = ((u8 *) new) + sizeof(*new);
1171         new->tail = new->head + new_head_len;
1172         new->head_len = new_head_len;
1173         new->tail_len = new_tail_len;
1174         /* copy in optional mbssid_ies */
1175         if (mbssid) {
1176                 u8 *pos = new->tail + new->tail_len;
1177
1178                 new->mbssid_ies = (void *)pos;
1179                 pos += struct_size(new->mbssid_ies, elem, mbssid->cnt);
1180                 pos += ieee80211_copy_mbssid_beacon(pos, new->mbssid_ies,
1181                                                     mbssid);
1182                 if (rnr) {
1183                         new->rnr_ies = (void *)pos;
1184                         pos += struct_size(new->rnr_ies, elem, rnr->cnt);
1185                         ieee80211_copy_rnr_beacon(pos, new->rnr_ies, rnr);
1186                 }
1187                 /* update bssid_indicator */
1188                 link_conf->bssid_indicator =
1189                         ilog2(__roundup_pow_of_two(mbssid->cnt + 1));
1190         }
1191
1192         if (csa) {
1193                 new->cntdwn_current_counter = csa->count;
1194                 memcpy(new->cntdwn_counter_offsets, csa->counter_offsets_beacon,
1195                        csa->n_counter_offsets_beacon *
1196                        sizeof(new->cntdwn_counter_offsets[0]));
1197         } else if (cca) {
1198                 new->cntdwn_current_counter = cca->count;
1199                 new->cntdwn_counter_offsets[0] = cca->counter_offset_beacon;
1200         }
1201
1202         /* copy in head */
1203         if (params->head)
1204                 memcpy(new->head, params->head, new_head_len);
1205         else
1206                 memcpy(new->head, old->head, new_head_len);
1207
1208         /* copy in optional tail */
1209         if (params->tail)
1210                 memcpy(new->tail, params->tail, new_tail_len);
1211         else
1212                 if (old)
1213                         memcpy(new->tail, old->tail, new_tail_len);
1214
1215         err = ieee80211_set_probe_resp(sdata, params->probe_resp,
1216                                        params->probe_resp_len, csa, cca, link);
1217         if (err < 0) {
1218                 kfree(new);
1219                 return err;
1220         }
1221         if (err == 0)
1222                 changed |= BSS_CHANGED_AP_PROBE_RESP;
1223
1224         if (params->ftm_responder != -1) {
1225                 link_conf->ftm_responder = params->ftm_responder;
1226                 err = ieee80211_set_ftm_responder_params(sdata,
1227                                                          params->lci,
1228                                                          params->lci_len,
1229                                                          params->civicloc,
1230                                                          params->civicloc_len,
1231                                                          link_conf);
1232
1233                 if (err < 0) {
1234                         kfree(new);
1235                         return err;
1236                 }
1237
1238                 changed |= BSS_CHANGED_FTM_RESPONDER;
1239         }
1240
1241         rcu_assign_pointer(link->u.ap.beacon, new);
1242         sdata->u.ap.active = true;
1243
1244         if (old)
1245                 kfree_rcu(old, rcu_head);
1246
1247         return changed;
1248 }
1249
1250 static int ieee80211_start_ap(struct wiphy *wiphy, struct net_device *dev,
1251                               struct cfg80211_ap_settings *params)
1252 {
1253         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1254         struct ieee80211_local *local = sdata->local;
1255         struct beacon_data *old;
1256         struct ieee80211_sub_if_data *vlan;
1257         u64 changed = BSS_CHANGED_BEACON_INT |
1258                       BSS_CHANGED_BEACON_ENABLED |
1259                       BSS_CHANGED_BEACON |
1260                       BSS_CHANGED_P2P_PS |
1261                       BSS_CHANGED_TXPOWER |
1262                       BSS_CHANGED_TWT;
1263         int i, err;
1264         int prev_beacon_int;
1265         unsigned int link_id = params->beacon.link_id;
1266         struct ieee80211_link_data *link;
1267         struct ieee80211_bss_conf *link_conf;
1268
1269         link = sdata_dereference(sdata->link[link_id], sdata);
1270         if (!link)
1271                 return -ENOLINK;
1272
1273         link_conf = link->conf;
1274
1275         old = sdata_dereference(link->u.ap.beacon, sdata);
1276         if (old)
1277                 return -EALREADY;
1278
1279         if (params->smps_mode != NL80211_SMPS_OFF)
1280                 return -ENOTSUPP;
1281
1282         link->smps_mode = IEEE80211_SMPS_OFF;
1283
1284         link->needed_rx_chains = sdata->local->rx_chains;
1285
1286         prev_beacon_int = link_conf->beacon_int;
1287         link_conf->beacon_int = params->beacon_interval;
1288
1289         if (params->ht_cap)
1290                 link_conf->ht_ldpc =
1291                         params->ht_cap->cap_info &
1292                                 cpu_to_le16(IEEE80211_HT_CAP_LDPC_CODING);
1293
1294         if (params->vht_cap) {
1295                 link_conf->vht_ldpc =
1296                         params->vht_cap->vht_cap_info &
1297                                 cpu_to_le32(IEEE80211_VHT_CAP_RXLDPC);
1298                 link_conf->vht_su_beamformer =
1299                         params->vht_cap->vht_cap_info &
1300                                 cpu_to_le32(IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE);
1301                 link_conf->vht_su_beamformee =
1302                         params->vht_cap->vht_cap_info &
1303                                 cpu_to_le32(IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE);
1304                 link_conf->vht_mu_beamformer =
1305                         params->vht_cap->vht_cap_info &
1306                                 cpu_to_le32(IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE);
1307                 link_conf->vht_mu_beamformee =
1308                         params->vht_cap->vht_cap_info &
1309                                 cpu_to_le32(IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE);
1310         }
1311
1312         if (params->he_cap && params->he_oper) {
1313                 link_conf->he_support = true;
1314                 link_conf->htc_trig_based_pkt_ext =
1315                         le32_get_bits(params->he_oper->he_oper_params,
1316                               IEEE80211_HE_OPERATION_DFLT_PE_DURATION_MASK);
1317                 link_conf->frame_time_rts_th =
1318                         le32_get_bits(params->he_oper->he_oper_params,
1319                               IEEE80211_HE_OPERATION_RTS_THRESHOLD_MASK);
1320                 changed |= BSS_CHANGED_HE_OBSS_PD;
1321
1322                 if (params->beacon.he_bss_color.enabled)
1323                         changed |= BSS_CHANGED_HE_BSS_COLOR;
1324         }
1325
1326         if (params->he_cap) {
1327                 link_conf->he_ldpc =
1328                         params->he_cap->phy_cap_info[1] &
1329                                 IEEE80211_HE_PHY_CAP1_LDPC_CODING_IN_PAYLOAD;
1330                 link_conf->he_su_beamformer =
1331                         params->he_cap->phy_cap_info[3] &
1332                                 IEEE80211_HE_PHY_CAP3_SU_BEAMFORMER;
1333                 link_conf->he_su_beamformee =
1334                         params->he_cap->phy_cap_info[4] &
1335                                 IEEE80211_HE_PHY_CAP4_SU_BEAMFORMEE;
1336                 link_conf->he_mu_beamformer =
1337                         params->he_cap->phy_cap_info[4] &
1338                                 IEEE80211_HE_PHY_CAP4_MU_BEAMFORMER;
1339                 link_conf->he_full_ul_mumimo =
1340                         params->he_cap->phy_cap_info[2] &
1341                                 IEEE80211_HE_PHY_CAP2_UL_MU_FULL_MU_MIMO;
1342         }
1343
1344         if (params->eht_cap) {
1345                 if (!link_conf->he_support)
1346                         return -EOPNOTSUPP;
1347
1348                 link_conf->eht_support = true;
1349                 link_conf->eht_puncturing = params->punct_bitmap;
1350                 changed |= BSS_CHANGED_EHT_PUNCTURING;
1351
1352                 link_conf->eht_su_beamformer =
1353                         params->eht_cap->fixed.phy_cap_info[0] &
1354                                 IEEE80211_EHT_PHY_CAP0_SU_BEAMFORMER;
1355                 link_conf->eht_su_beamformee =
1356                         params->eht_cap->fixed.phy_cap_info[0] &
1357                                 IEEE80211_EHT_PHY_CAP0_SU_BEAMFORMEE;
1358                 link_conf->eht_mu_beamformer =
1359                         params->eht_cap->fixed.phy_cap_info[7] &
1360                                 (IEEE80211_EHT_PHY_CAP7_MU_BEAMFORMER_80MHZ |
1361                                  IEEE80211_EHT_PHY_CAP7_MU_BEAMFORMER_160MHZ |
1362                                  IEEE80211_EHT_PHY_CAP7_MU_BEAMFORMER_320MHZ);
1363         } else {
1364                 link_conf->eht_su_beamformer = false;
1365                 link_conf->eht_su_beamformee = false;
1366                 link_conf->eht_mu_beamformer = false;
1367         }
1368
1369         if (sdata->vif.type == NL80211_IFTYPE_AP &&
1370             params->mbssid_config.tx_wdev) {
1371                 err = ieee80211_set_ap_mbssid_options(sdata,
1372                                                       params->mbssid_config,
1373                                                       link_conf);
1374                 if (err)
1375                         return err;
1376         }
1377
1378         mutex_lock(&local->mtx);
1379         err = ieee80211_link_use_channel(link, &params->chandef,
1380                                          IEEE80211_CHANCTX_SHARED);
1381         if (!err)
1382                 ieee80211_link_copy_chanctx_to_vlans(link, false);
1383         mutex_unlock(&local->mtx);
1384         if (err) {
1385                 link_conf->beacon_int = prev_beacon_int;
1386                 return err;
1387         }
1388
1389         /*
1390          * Apply control port protocol, this allows us to
1391          * not encrypt dynamic WEP control frames.
1392          */
1393         sdata->control_port_protocol = params->crypto.control_port_ethertype;
1394         sdata->control_port_no_encrypt = params->crypto.control_port_no_encrypt;
1395         sdata->control_port_over_nl80211 =
1396                                 params->crypto.control_port_over_nl80211;
1397         sdata->control_port_no_preauth =
1398                                 params->crypto.control_port_no_preauth;
1399
1400         list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list) {
1401                 vlan->control_port_protocol =
1402                         params->crypto.control_port_ethertype;
1403                 vlan->control_port_no_encrypt =
1404                         params->crypto.control_port_no_encrypt;
1405                 vlan->control_port_over_nl80211 =
1406                         params->crypto.control_port_over_nl80211;
1407                 vlan->control_port_no_preauth =
1408                         params->crypto.control_port_no_preauth;
1409         }
1410
1411         link_conf->dtim_period = params->dtim_period;
1412         link_conf->enable_beacon = true;
1413         link_conf->allow_p2p_go_ps = sdata->vif.p2p;
1414         link_conf->twt_responder = params->twt_responder;
1415         link_conf->he_obss_pd = params->he_obss_pd;
1416         link_conf->he_bss_color = params->beacon.he_bss_color;
1417         sdata->vif.cfg.s1g = params->chandef.chan->band ==
1418                                   NL80211_BAND_S1GHZ;
1419
1420         sdata->vif.cfg.ssid_len = params->ssid_len;
1421         if (params->ssid_len)
1422                 memcpy(sdata->vif.cfg.ssid, params->ssid,
1423                        params->ssid_len);
1424         link_conf->hidden_ssid =
1425                 (params->hidden_ssid != NL80211_HIDDEN_SSID_NOT_IN_USE);
1426
1427         memset(&link_conf->p2p_noa_attr, 0,
1428                sizeof(link_conf->p2p_noa_attr));
1429         link_conf->p2p_noa_attr.oppps_ctwindow =
1430                 params->p2p_ctwindow & IEEE80211_P2P_OPPPS_CTWINDOW_MASK;
1431         if (params->p2p_opp_ps)
1432                 link_conf->p2p_noa_attr.oppps_ctwindow |=
1433                                         IEEE80211_P2P_OPPPS_ENABLE_BIT;
1434
1435         sdata->beacon_rate_set = false;
1436         if (wiphy_ext_feature_isset(local->hw.wiphy,
1437                                     NL80211_EXT_FEATURE_BEACON_RATE_LEGACY)) {
1438                 for (i = 0; i < NUM_NL80211_BANDS; i++) {
1439                         sdata->beacon_rateidx_mask[i] =
1440                                 params->beacon_rate.control[i].legacy;
1441                         if (sdata->beacon_rateidx_mask[i])
1442                                 sdata->beacon_rate_set = true;
1443                 }
1444         }
1445
1446         if (ieee80211_hw_check(&local->hw, HAS_RATE_CONTROL))
1447                 link_conf->beacon_tx_rate = params->beacon_rate;
1448
1449         err = ieee80211_assign_beacon(sdata, link, &params->beacon, NULL, NULL);
1450         if (err < 0)
1451                 goto error;
1452         changed |= err;
1453
1454         if (params->fils_discovery.max_interval) {
1455                 err = ieee80211_set_fils_discovery(sdata,
1456                                                    &params->fils_discovery,
1457                                                    link, link_conf);
1458                 if (err < 0)
1459                         goto error;
1460                 changed |= BSS_CHANGED_FILS_DISCOVERY;
1461         }
1462
1463         if (params->unsol_bcast_probe_resp.interval) {
1464                 err = ieee80211_set_unsol_bcast_probe_resp(sdata,
1465                                                            &params->unsol_bcast_probe_resp,
1466                                                            link, link_conf);
1467                 if (err < 0)
1468                         goto error;
1469                 changed |= BSS_CHANGED_UNSOL_BCAST_PROBE_RESP;
1470         }
1471
1472         err = drv_start_ap(sdata->local, sdata, link_conf);
1473         if (err) {
1474                 old = sdata_dereference(link->u.ap.beacon, sdata);
1475
1476                 if (old)
1477                         kfree_rcu(old, rcu_head);
1478                 RCU_INIT_POINTER(link->u.ap.beacon, NULL);
1479                 sdata->u.ap.active = false;
1480                 goto error;
1481         }
1482
1483         ieee80211_recalc_dtim(local, sdata);
1484         ieee80211_vif_cfg_change_notify(sdata, BSS_CHANGED_SSID);
1485         ieee80211_link_info_change_notify(sdata, link, changed);
1486
1487         netif_carrier_on(dev);
1488         list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
1489                 netif_carrier_on(vlan->dev);
1490
1491         return 0;
1492
1493 error:
1494         mutex_lock(&local->mtx);
1495         ieee80211_link_release_channel(link);
1496         mutex_unlock(&local->mtx);
1497
1498         return err;
1499 }
1500
1501 static int ieee80211_change_beacon(struct wiphy *wiphy, struct net_device *dev,
1502                                    struct cfg80211_beacon_data *params)
1503 {
1504         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1505         struct ieee80211_link_data *link;
1506         struct beacon_data *old;
1507         int err;
1508         struct ieee80211_bss_conf *link_conf;
1509
1510         sdata_assert_lock(sdata);
1511
1512         link = sdata_dereference(sdata->link[params->link_id], sdata);
1513         if (!link)
1514                 return -ENOLINK;
1515
1516         link_conf = link->conf;
1517
1518         /* don't allow changing the beacon while a countdown is in place - offset
1519          * of channel switch counter may change
1520          */
1521         if (link_conf->csa_active || link_conf->color_change_active)
1522                 return -EBUSY;
1523
1524         old = sdata_dereference(link->u.ap.beacon, sdata);
1525         if (!old)
1526                 return -ENOENT;
1527
1528         err = ieee80211_assign_beacon(sdata, link, params, NULL, NULL);
1529         if (err < 0)
1530                 return err;
1531
1532         if (params->he_bss_color_valid &&
1533             params->he_bss_color.enabled != link_conf->he_bss_color.enabled) {
1534                 link_conf->he_bss_color.enabled = params->he_bss_color.enabled;
1535                 err |= BSS_CHANGED_HE_BSS_COLOR;
1536         }
1537
1538         ieee80211_link_info_change_notify(sdata, link, err);
1539         return 0;
1540 }
1541
1542 static void ieee80211_free_next_beacon(struct ieee80211_link_data *link)
1543 {
1544         if (!link->u.ap.next_beacon)
1545                 return;
1546
1547         kfree(link->u.ap.next_beacon->mbssid_ies);
1548         kfree(link->u.ap.next_beacon->rnr_ies);
1549         kfree(link->u.ap.next_beacon);
1550         link->u.ap.next_beacon = NULL;
1551 }
1552
1553 static int ieee80211_stop_ap(struct wiphy *wiphy, struct net_device *dev,
1554                              unsigned int link_id)
1555 {
1556         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1557         struct ieee80211_sub_if_data *vlan;
1558         struct ieee80211_local *local = sdata->local;
1559         struct beacon_data *old_beacon;
1560         struct probe_resp *old_probe_resp;
1561         struct fils_discovery_data *old_fils_discovery;
1562         struct unsol_bcast_probe_resp_data *old_unsol_bcast_probe_resp;
1563         struct cfg80211_chan_def chandef;
1564         struct ieee80211_link_data *link =
1565                 sdata_dereference(sdata->link[link_id], sdata);
1566         struct ieee80211_bss_conf *link_conf = link->conf;
1567
1568         sdata_assert_lock(sdata);
1569
1570         old_beacon = sdata_dereference(link->u.ap.beacon, sdata);
1571         if (!old_beacon)
1572                 return -ENOENT;
1573         old_probe_resp = sdata_dereference(link->u.ap.probe_resp,
1574                                            sdata);
1575         old_fils_discovery = sdata_dereference(link->u.ap.fils_discovery,
1576                                                sdata);
1577         old_unsol_bcast_probe_resp =
1578                 sdata_dereference(link->u.ap.unsol_bcast_probe_resp,
1579                                   sdata);
1580
1581         /* abort any running channel switch or color change */
1582         mutex_lock(&local->mtx);
1583         link_conf->csa_active = false;
1584         link_conf->color_change_active = false;
1585         if (link->csa_block_tx) {
1586                 ieee80211_wake_vif_queues(local, sdata,
1587                                           IEEE80211_QUEUE_STOP_REASON_CSA);
1588                 link->csa_block_tx = false;
1589         }
1590
1591         mutex_unlock(&local->mtx);
1592
1593         ieee80211_free_next_beacon(link);
1594
1595         /* turn off carrier for this interface and dependent VLANs */
1596         list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
1597                 netif_carrier_off(vlan->dev);
1598         netif_carrier_off(dev);
1599
1600         /* remove beacon and probe response */
1601         sdata->u.ap.active = false;
1602         RCU_INIT_POINTER(link->u.ap.beacon, NULL);
1603         RCU_INIT_POINTER(link->u.ap.probe_resp, NULL);
1604         RCU_INIT_POINTER(link->u.ap.fils_discovery, NULL);
1605         RCU_INIT_POINTER(link->u.ap.unsol_bcast_probe_resp, NULL);
1606         kfree_rcu(old_beacon, rcu_head);
1607         if (old_probe_resp)
1608                 kfree_rcu(old_probe_resp, rcu_head);
1609         if (old_fils_discovery)
1610                 kfree_rcu(old_fils_discovery, rcu_head);
1611         if (old_unsol_bcast_probe_resp)
1612                 kfree_rcu(old_unsol_bcast_probe_resp, rcu_head);
1613
1614         kfree(link_conf->ftmr_params);
1615         link_conf->ftmr_params = NULL;
1616
1617         sdata->vif.mbssid_tx_vif = NULL;
1618         link_conf->bssid_index = 0;
1619         link_conf->nontransmitted = false;
1620         link_conf->ema_ap = false;
1621         link_conf->bssid_indicator = 0;
1622
1623         __sta_info_flush(sdata, true);
1624         ieee80211_free_keys(sdata, true);
1625
1626         link_conf->enable_beacon = false;
1627         sdata->beacon_rate_set = false;
1628         sdata->vif.cfg.ssid_len = 0;
1629         clear_bit(SDATA_STATE_OFFCHANNEL_BEACON_STOPPED, &sdata->state);
1630         ieee80211_link_info_change_notify(sdata, link,
1631                                           BSS_CHANGED_BEACON_ENABLED);
1632
1633         if (sdata->wdev.cac_started) {
1634                 chandef = link_conf->chandef;
1635                 cancel_delayed_work_sync(&link->dfs_cac_timer_work);
1636                 cfg80211_cac_event(sdata->dev, &chandef,
1637                                    NL80211_RADAR_CAC_ABORTED,
1638                                    GFP_KERNEL);
1639         }
1640
1641         drv_stop_ap(sdata->local, sdata, link_conf);
1642
1643         /* free all potentially still buffered bcast frames */
1644         local->total_ps_buffered -= skb_queue_len(&sdata->u.ap.ps.bc_buf);
1645         ieee80211_purge_tx_queue(&local->hw, &sdata->u.ap.ps.bc_buf);
1646
1647         mutex_lock(&local->mtx);
1648         ieee80211_link_copy_chanctx_to_vlans(link, true);
1649         ieee80211_link_release_channel(link);
1650         mutex_unlock(&local->mtx);
1651
1652         return 0;
1653 }
1654
1655 static int sta_apply_auth_flags(struct ieee80211_local *local,
1656                                 struct sta_info *sta,
1657                                 u32 mask, u32 set)
1658 {
1659         int ret;
1660
1661         if (mask & BIT(NL80211_STA_FLAG_AUTHENTICATED) &&
1662             set & BIT(NL80211_STA_FLAG_AUTHENTICATED) &&
1663             !test_sta_flag(sta, WLAN_STA_AUTH)) {
1664                 ret = sta_info_move_state(sta, IEEE80211_STA_AUTH);
1665                 if (ret)
1666                         return ret;
1667         }
1668
1669         if (mask & BIT(NL80211_STA_FLAG_ASSOCIATED) &&
1670             set & BIT(NL80211_STA_FLAG_ASSOCIATED) &&
1671             !test_sta_flag(sta, WLAN_STA_ASSOC)) {
1672                 /*
1673                  * When peer becomes associated, init rate control as
1674                  * well. Some drivers require rate control initialized
1675                  * before drv_sta_state() is called.
1676                  */
1677                 if (!test_sta_flag(sta, WLAN_STA_RATE_CONTROL))
1678                         rate_control_rate_init(sta);
1679
1680                 ret = sta_info_move_state(sta, IEEE80211_STA_ASSOC);
1681                 if (ret)
1682                         return ret;
1683         }
1684
1685         if (mask & BIT(NL80211_STA_FLAG_AUTHORIZED)) {
1686                 if (set & BIT(NL80211_STA_FLAG_AUTHORIZED))
1687                         ret = sta_info_move_state(sta, IEEE80211_STA_AUTHORIZED);
1688                 else if (test_sta_flag(sta, WLAN_STA_AUTHORIZED))
1689                         ret = sta_info_move_state(sta, IEEE80211_STA_ASSOC);
1690                 else
1691                         ret = 0;
1692                 if (ret)
1693                         return ret;
1694         }
1695
1696         if (mask & BIT(NL80211_STA_FLAG_ASSOCIATED) &&
1697             !(set & BIT(NL80211_STA_FLAG_ASSOCIATED)) &&
1698             test_sta_flag(sta, WLAN_STA_ASSOC)) {
1699                 ret = sta_info_move_state(sta, IEEE80211_STA_AUTH);
1700                 if (ret)
1701                         return ret;
1702         }
1703
1704         if (mask & BIT(NL80211_STA_FLAG_AUTHENTICATED) &&
1705             !(set & BIT(NL80211_STA_FLAG_AUTHENTICATED)) &&
1706             test_sta_flag(sta, WLAN_STA_AUTH)) {
1707                 ret = sta_info_move_state(sta, IEEE80211_STA_NONE);
1708                 if (ret)
1709                         return ret;
1710         }
1711
1712         return 0;
1713 }
1714
1715 static void sta_apply_mesh_params(struct ieee80211_local *local,
1716                                   struct sta_info *sta,
1717                                   struct station_parameters *params)
1718 {
1719 #ifdef CONFIG_MAC80211_MESH
1720         struct ieee80211_sub_if_data *sdata = sta->sdata;
1721         u32 changed = 0;
1722
1723         if (params->sta_modify_mask & STATION_PARAM_APPLY_PLINK_STATE) {
1724                 switch (params->plink_state) {
1725                 case NL80211_PLINK_ESTAB:
1726                         if (sta->mesh->plink_state != NL80211_PLINK_ESTAB)
1727                                 changed = mesh_plink_inc_estab_count(sdata);
1728                         sta->mesh->plink_state = params->plink_state;
1729                         sta->mesh->aid = params->peer_aid;
1730
1731                         ieee80211_mps_sta_status_update(sta);
1732                         changed |= ieee80211_mps_set_sta_local_pm(sta,
1733                                       sdata->u.mesh.mshcfg.power_mode);
1734
1735                         ewma_mesh_tx_rate_avg_init(&sta->mesh->tx_rate_avg);
1736                         /* init at low value */
1737                         ewma_mesh_tx_rate_avg_add(&sta->mesh->tx_rate_avg, 10);
1738
1739                         break;
1740                 case NL80211_PLINK_LISTEN:
1741                 case NL80211_PLINK_BLOCKED:
1742                 case NL80211_PLINK_OPN_SNT:
1743                 case NL80211_PLINK_OPN_RCVD:
1744                 case NL80211_PLINK_CNF_RCVD:
1745                 case NL80211_PLINK_HOLDING:
1746                         if (sta->mesh->plink_state == NL80211_PLINK_ESTAB)
1747                                 changed = mesh_plink_dec_estab_count(sdata);
1748                         sta->mesh->plink_state = params->plink_state;
1749
1750                         ieee80211_mps_sta_status_update(sta);
1751                         changed |= ieee80211_mps_set_sta_local_pm(sta,
1752                                         NL80211_MESH_POWER_UNKNOWN);
1753                         break;
1754                 default:
1755                         /*  nothing  */
1756                         break;
1757                 }
1758         }
1759
1760         switch (params->plink_action) {
1761         case NL80211_PLINK_ACTION_NO_ACTION:
1762                 /* nothing */
1763                 break;
1764         case NL80211_PLINK_ACTION_OPEN:
1765                 changed |= mesh_plink_open(sta);
1766                 break;
1767         case NL80211_PLINK_ACTION_BLOCK:
1768                 changed |= mesh_plink_block(sta);
1769                 break;
1770         }
1771
1772         if (params->local_pm)
1773                 changed |= ieee80211_mps_set_sta_local_pm(sta,
1774                                                           params->local_pm);
1775
1776         ieee80211_mbss_info_change_notify(sdata, changed);
1777 #endif
1778 }
1779
1780 static int sta_link_apply_parameters(struct ieee80211_local *local,
1781                                      struct sta_info *sta, bool new_link,
1782                                      struct link_station_parameters *params)
1783 {
1784         int ret = 0;
1785         struct ieee80211_supported_band *sband;
1786         struct ieee80211_sub_if_data *sdata = sta->sdata;
1787         u32 link_id = params->link_id < 0 ? 0 : params->link_id;
1788         struct ieee80211_link_data *link =
1789                 sdata_dereference(sdata->link[link_id], sdata);
1790         struct link_sta_info *link_sta =
1791                 rcu_dereference_protected(sta->link[link_id],
1792                                           lockdep_is_held(&local->sta_mtx));
1793
1794         /*
1795          * If there are no changes, then accept a link that doesn't exist,
1796          * unless it's a new link.
1797          */
1798         if (params->link_id < 0 && !new_link &&
1799             !params->link_mac && !params->txpwr_set &&
1800             !params->supported_rates_len &&
1801             !params->ht_capa && !params->vht_capa &&
1802             !params->he_capa && !params->eht_capa &&
1803             !params->opmode_notif_used)
1804                 return 0;
1805
1806         if (!link || !link_sta)
1807                 return -EINVAL;
1808
1809         sband = ieee80211_get_link_sband(link);
1810         if (!sband)
1811                 return -EINVAL;
1812
1813         if (params->link_mac) {
1814                 if (new_link) {
1815                         memcpy(link_sta->addr, params->link_mac, ETH_ALEN);
1816                         memcpy(link_sta->pub->addr, params->link_mac, ETH_ALEN);
1817                 } else if (!ether_addr_equal(link_sta->addr,
1818                                              params->link_mac)) {
1819                         return -EINVAL;
1820                 }
1821         } else if (new_link) {
1822                 return -EINVAL;
1823         }
1824
1825         if (params->txpwr_set) {
1826                 link_sta->pub->txpwr.type = params->txpwr.type;
1827                 if (params->txpwr.type == NL80211_TX_POWER_LIMITED)
1828                         link_sta->pub->txpwr.power = params->txpwr.power;
1829                 ret = drv_sta_set_txpwr(local, sdata, sta);
1830                 if (ret)
1831                         return ret;
1832         }
1833
1834         if (params->supported_rates &&
1835             params->supported_rates_len) {
1836                 ieee80211_parse_bitrates(link->conf->chandef.width,
1837                                          sband, params->supported_rates,
1838                                          params->supported_rates_len,
1839                                          &link_sta->pub->supp_rates[sband->band]);
1840         }
1841
1842         if (params->ht_capa)
1843                 ieee80211_ht_cap_ie_to_sta_ht_cap(sdata, sband,
1844                                                   params->ht_capa, link_sta);
1845
1846         /* VHT can override some HT caps such as the A-MSDU max length */
1847         if (params->vht_capa)
1848                 ieee80211_vht_cap_ie_to_sta_vht_cap(sdata, sband,
1849                                                     params->vht_capa, link_sta);
1850
1851         if (params->he_capa)
1852                 ieee80211_he_cap_ie_to_sta_he_cap(sdata, sband,
1853                                                   (void *)params->he_capa,
1854                                                   params->he_capa_len,
1855                                                   (void *)params->he_6ghz_capa,
1856                                                   link_sta);
1857
1858         if (params->he_capa && params->eht_capa)
1859                 ieee80211_eht_cap_ie_to_sta_eht_cap(sdata, sband,
1860                                                     (u8 *)params->he_capa,
1861                                                     params->he_capa_len,
1862                                                     params->eht_capa,
1863                                                     params->eht_capa_len,
1864                                                     link_sta);
1865
1866         if (params->opmode_notif_used) {
1867                 /* returned value is only needed for rc update, but the
1868                  * rc isn't initialized here yet, so ignore it
1869                  */
1870                 __ieee80211_vht_handle_opmode(sdata, link_sta,
1871                                               params->opmode_notif,
1872                                               sband->band);
1873         }
1874
1875         return ret;
1876 }
1877
1878 static int sta_apply_parameters(struct ieee80211_local *local,
1879                                 struct sta_info *sta,
1880                                 struct station_parameters *params)
1881 {
1882         struct ieee80211_sub_if_data *sdata = sta->sdata;
1883         u32 mask, set;
1884         int ret = 0;
1885
1886         mask = params->sta_flags_mask;
1887         set = params->sta_flags_set;
1888
1889         if (ieee80211_vif_is_mesh(&sdata->vif)) {
1890                 /*
1891                  * In mesh mode, ASSOCIATED isn't part of the nl80211
1892                  * API but must follow AUTHENTICATED for driver state.
1893                  */
1894                 if (mask & BIT(NL80211_STA_FLAG_AUTHENTICATED))
1895                         mask |= BIT(NL80211_STA_FLAG_ASSOCIATED);
1896                 if (set & BIT(NL80211_STA_FLAG_AUTHENTICATED))
1897                         set |= BIT(NL80211_STA_FLAG_ASSOCIATED);
1898         } else if (test_sta_flag(sta, WLAN_STA_TDLS_PEER)) {
1899                 /*
1900                  * TDLS -- everything follows authorized, but
1901                  * only becoming authorized is possible, not
1902                  * going back
1903                  */
1904                 if (set & BIT(NL80211_STA_FLAG_AUTHORIZED)) {
1905                         set |= BIT(NL80211_STA_FLAG_AUTHENTICATED) |
1906                                BIT(NL80211_STA_FLAG_ASSOCIATED);
1907                         mask |= BIT(NL80211_STA_FLAG_AUTHENTICATED) |
1908                                 BIT(NL80211_STA_FLAG_ASSOCIATED);
1909                 }
1910         }
1911
1912         if (mask & BIT(NL80211_STA_FLAG_WME) &&
1913             local->hw.queues >= IEEE80211_NUM_ACS)
1914                 sta->sta.wme = set & BIT(NL80211_STA_FLAG_WME);
1915
1916         /* auth flags will be set later for TDLS,
1917          * and for unassociated stations that move to associated */
1918         if (!test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
1919             !((mask & BIT(NL80211_STA_FLAG_ASSOCIATED)) &&
1920               (set & BIT(NL80211_STA_FLAG_ASSOCIATED)))) {
1921                 ret = sta_apply_auth_flags(local, sta, mask, set);
1922                 if (ret)
1923                         return ret;
1924         }
1925
1926         if (mask & BIT(NL80211_STA_FLAG_SHORT_PREAMBLE)) {
1927                 if (set & BIT(NL80211_STA_FLAG_SHORT_PREAMBLE))
1928                         set_sta_flag(sta, WLAN_STA_SHORT_PREAMBLE);
1929                 else
1930                         clear_sta_flag(sta, WLAN_STA_SHORT_PREAMBLE);
1931         }
1932
1933         if (mask & BIT(NL80211_STA_FLAG_MFP)) {
1934                 sta->sta.mfp = !!(set & BIT(NL80211_STA_FLAG_MFP));
1935                 if (set & BIT(NL80211_STA_FLAG_MFP))
1936                         set_sta_flag(sta, WLAN_STA_MFP);
1937                 else
1938                         clear_sta_flag(sta, WLAN_STA_MFP);
1939         }
1940
1941         if (mask & BIT(NL80211_STA_FLAG_TDLS_PEER)) {
1942                 if (set & BIT(NL80211_STA_FLAG_TDLS_PEER))
1943                         set_sta_flag(sta, WLAN_STA_TDLS_PEER);
1944                 else
1945                         clear_sta_flag(sta, WLAN_STA_TDLS_PEER);
1946         }
1947
1948         /* mark TDLS channel switch support, if the AP allows it */
1949         if (test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
1950             !sdata->deflink.u.mgd.tdls_chan_switch_prohibited &&
1951             params->ext_capab_len >= 4 &&
1952             params->ext_capab[3] & WLAN_EXT_CAPA4_TDLS_CHAN_SWITCH)
1953                 set_sta_flag(sta, WLAN_STA_TDLS_CHAN_SWITCH);
1954
1955         if (test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
1956             !sdata->u.mgd.tdls_wider_bw_prohibited &&
1957             ieee80211_hw_check(&local->hw, TDLS_WIDER_BW) &&
1958             params->ext_capab_len >= 8 &&
1959             params->ext_capab[7] & WLAN_EXT_CAPA8_TDLS_WIDE_BW_ENABLED)
1960                 set_sta_flag(sta, WLAN_STA_TDLS_WIDER_BW);
1961
1962         if (params->sta_modify_mask & STATION_PARAM_APPLY_UAPSD) {
1963                 sta->sta.uapsd_queues = params->uapsd_queues;
1964                 sta->sta.max_sp = params->max_sp;
1965         }
1966
1967         ieee80211_sta_set_max_amsdu_subframes(sta, params->ext_capab,
1968                                               params->ext_capab_len);
1969
1970         /*
1971          * cfg80211 validates this (1-2007) and allows setting the AID
1972          * only when creating a new station entry
1973          */
1974         if (params->aid)
1975                 sta->sta.aid = params->aid;
1976
1977         /*
1978          * Some of the following updates would be racy if called on an
1979          * existing station, via ieee80211_change_station(). However,
1980          * all such changes are rejected by cfg80211 except for updates
1981          * changing the supported rates on an existing but not yet used
1982          * TDLS peer.
1983          */
1984
1985         if (params->listen_interval >= 0)
1986                 sta->listen_interval = params->listen_interval;
1987
1988         ret = sta_link_apply_parameters(local, sta, false,
1989                                         &params->link_sta_params);
1990         if (ret)
1991                 return ret;
1992
1993         if (params->support_p2p_ps >= 0)
1994                 sta->sta.support_p2p_ps = params->support_p2p_ps;
1995
1996         if (ieee80211_vif_is_mesh(&sdata->vif))
1997                 sta_apply_mesh_params(local, sta, params);
1998
1999         if (params->airtime_weight)
2000                 sta->airtime_weight = params->airtime_weight;
2001
2002         /* set the STA state after all sta info from usermode has been set */
2003         if (test_sta_flag(sta, WLAN_STA_TDLS_PEER) ||
2004             set & BIT(NL80211_STA_FLAG_ASSOCIATED)) {
2005                 ret = sta_apply_auth_flags(local, sta, mask, set);
2006                 if (ret)
2007                         return ret;
2008         }
2009
2010         /* Mark the STA as MLO if MLD MAC address is available */
2011         if (params->link_sta_params.mld_mac)
2012                 sta->sta.mlo = true;
2013
2014         return 0;
2015 }
2016
2017 static int ieee80211_add_station(struct wiphy *wiphy, struct net_device *dev,
2018                                  const u8 *mac,
2019                                  struct station_parameters *params)
2020 {
2021         struct ieee80211_local *local = wiphy_priv(wiphy);
2022         struct sta_info *sta;
2023         struct ieee80211_sub_if_data *sdata;
2024         int err;
2025
2026         if (params->vlan) {
2027                 sdata = IEEE80211_DEV_TO_SUB_IF(params->vlan);
2028
2029                 if (sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
2030                     sdata->vif.type != NL80211_IFTYPE_AP)
2031                         return -EINVAL;
2032         } else
2033                 sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2034
2035         if (ether_addr_equal(mac, sdata->vif.addr))
2036                 return -EINVAL;
2037
2038         if (!is_valid_ether_addr(mac))
2039                 return -EINVAL;
2040
2041         if (params->sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER) &&
2042             sdata->vif.type == NL80211_IFTYPE_STATION &&
2043             !sdata->u.mgd.associated)
2044                 return -EINVAL;
2045
2046         /*
2047          * If we have a link ID, it can be a non-MLO station on an AP MLD,
2048          * but we need to have a link_mac in that case as well, so use the
2049          * STA's MAC address in that case.
2050          */
2051         if (params->link_sta_params.link_id >= 0)
2052                 sta = sta_info_alloc_with_link(sdata, mac,
2053                                                params->link_sta_params.link_id,
2054                                                params->link_sta_params.link_mac ?: mac,
2055                                                GFP_KERNEL);
2056         else
2057                 sta = sta_info_alloc(sdata, mac, GFP_KERNEL);
2058
2059         if (!sta)
2060                 return -ENOMEM;
2061
2062         if (params->sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER))
2063                 sta->sta.tdls = true;
2064
2065         /* Though the mutex is not needed here (since the station is not
2066          * visible yet), sta_apply_parameters (and inner functions) require
2067          * the mutex due to other paths.
2068          */
2069         mutex_lock(&local->sta_mtx);
2070         err = sta_apply_parameters(local, sta, params);
2071         mutex_unlock(&local->sta_mtx);
2072         if (err) {
2073                 sta_info_free(local, sta);
2074                 return err;
2075         }
2076
2077         /*
2078          * for TDLS and for unassociated station, rate control should be
2079          * initialized only when rates are known and station is marked
2080          * authorized/associated
2081          */
2082         if (!test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
2083             test_sta_flag(sta, WLAN_STA_ASSOC))
2084                 rate_control_rate_init(sta);
2085
2086         return sta_info_insert(sta);
2087 }
2088
2089 static int ieee80211_del_station(struct wiphy *wiphy, struct net_device *dev,
2090                                  struct station_del_parameters *params)
2091 {
2092         struct ieee80211_sub_if_data *sdata;
2093
2094         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2095
2096         if (params->mac)
2097                 return sta_info_destroy_addr_bss(sdata, params->mac);
2098
2099         sta_info_flush(sdata);
2100         return 0;
2101 }
2102
2103 static int ieee80211_change_station(struct wiphy *wiphy,
2104                                     struct net_device *dev, const u8 *mac,
2105                                     struct station_parameters *params)
2106 {
2107         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2108         struct ieee80211_local *local = wiphy_priv(wiphy);
2109         struct sta_info *sta;
2110         struct ieee80211_sub_if_data *vlansdata;
2111         enum cfg80211_station_type statype;
2112         int err;
2113
2114         mutex_lock(&local->sta_mtx);
2115
2116         sta = sta_info_get_bss(sdata, mac);
2117         if (!sta) {
2118                 err = -ENOENT;
2119                 goto out_err;
2120         }
2121
2122         switch (sdata->vif.type) {
2123         case NL80211_IFTYPE_MESH_POINT:
2124                 if (sdata->u.mesh.user_mpm)
2125                         statype = CFG80211_STA_MESH_PEER_USER;
2126                 else
2127                         statype = CFG80211_STA_MESH_PEER_KERNEL;
2128                 break;
2129         case NL80211_IFTYPE_ADHOC:
2130                 statype = CFG80211_STA_IBSS;
2131                 break;
2132         case NL80211_IFTYPE_STATION:
2133                 if (!test_sta_flag(sta, WLAN_STA_TDLS_PEER)) {
2134                         statype = CFG80211_STA_AP_STA;
2135                         break;
2136                 }
2137                 if (test_sta_flag(sta, WLAN_STA_AUTHORIZED))
2138                         statype = CFG80211_STA_TDLS_PEER_ACTIVE;
2139                 else
2140                         statype = CFG80211_STA_TDLS_PEER_SETUP;
2141                 break;
2142         case NL80211_IFTYPE_AP:
2143         case NL80211_IFTYPE_AP_VLAN:
2144                 if (test_sta_flag(sta, WLAN_STA_ASSOC))
2145                         statype = CFG80211_STA_AP_CLIENT;
2146                 else
2147                         statype = CFG80211_STA_AP_CLIENT_UNASSOC;
2148                 break;
2149         default:
2150                 err = -EOPNOTSUPP;
2151                 goto out_err;
2152         }
2153
2154         err = cfg80211_check_station_change(wiphy, params, statype);
2155         if (err)
2156                 goto out_err;
2157
2158         if (params->vlan && params->vlan != sta->sdata->dev) {
2159                 vlansdata = IEEE80211_DEV_TO_SUB_IF(params->vlan);
2160
2161                 if (params->vlan->ieee80211_ptr->use_4addr) {
2162                         if (vlansdata->u.vlan.sta) {
2163                                 err = -EBUSY;
2164                                 goto out_err;
2165                         }
2166
2167                         rcu_assign_pointer(vlansdata->u.vlan.sta, sta);
2168                         __ieee80211_check_fast_rx_iface(vlansdata);
2169                         drv_sta_set_4addr(local, sta->sdata, &sta->sta, true);
2170                 }
2171
2172                 if (sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN &&
2173                     sta->sdata->u.vlan.sta) {
2174                         ieee80211_clear_fast_rx(sta);
2175                         RCU_INIT_POINTER(sta->sdata->u.vlan.sta, NULL);
2176                 }
2177
2178                 if (test_sta_flag(sta, WLAN_STA_AUTHORIZED))
2179                         ieee80211_vif_dec_num_mcast(sta->sdata);
2180
2181                 sta->sdata = vlansdata;
2182                 ieee80211_check_fast_xmit(sta);
2183
2184                 if (test_sta_flag(sta, WLAN_STA_AUTHORIZED)) {
2185                         ieee80211_vif_inc_num_mcast(sta->sdata);
2186                         cfg80211_send_layer2_update(sta->sdata->dev,
2187                                                     sta->sta.addr);
2188                 }
2189         }
2190
2191         /* we use sta_info_get_bss() so this might be different */
2192         if (sdata != sta->sdata) {
2193                 mutex_lock_nested(&sta->sdata->wdev.mtx, 1);
2194                 err = sta_apply_parameters(local, sta, params);
2195                 mutex_unlock(&sta->sdata->wdev.mtx);
2196         } else {
2197                 err = sta_apply_parameters(local, sta, params);
2198         }
2199         if (err)
2200                 goto out_err;
2201
2202         mutex_unlock(&local->sta_mtx);
2203
2204         if (sdata->vif.type == NL80211_IFTYPE_STATION &&
2205             params->sta_flags_mask & BIT(NL80211_STA_FLAG_AUTHORIZED)) {
2206                 ieee80211_recalc_ps(local);
2207                 ieee80211_recalc_ps_vif(sdata);
2208         }
2209
2210         return 0;
2211 out_err:
2212         mutex_unlock(&local->sta_mtx);
2213         return err;
2214 }
2215
2216 #ifdef CONFIG_MAC80211_MESH
2217 static int ieee80211_add_mpath(struct wiphy *wiphy, struct net_device *dev,
2218                                const u8 *dst, const u8 *next_hop)
2219 {
2220         struct ieee80211_sub_if_data *sdata;
2221         struct mesh_path *mpath;
2222         struct sta_info *sta;
2223
2224         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2225
2226         rcu_read_lock();
2227         sta = sta_info_get(sdata, next_hop);
2228         if (!sta) {
2229                 rcu_read_unlock();
2230                 return -ENOENT;
2231         }
2232
2233         mpath = mesh_path_add(sdata, dst);
2234         if (IS_ERR(mpath)) {
2235                 rcu_read_unlock();
2236                 return PTR_ERR(mpath);
2237         }
2238
2239         mesh_path_fix_nexthop(mpath, sta);
2240
2241         rcu_read_unlock();
2242         return 0;
2243 }
2244
2245 static int ieee80211_del_mpath(struct wiphy *wiphy, struct net_device *dev,
2246                                const u8 *dst)
2247 {
2248         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2249
2250         if (dst)
2251                 return mesh_path_del(sdata, dst);
2252
2253         mesh_path_flush_by_iface(sdata);
2254         return 0;
2255 }
2256
2257 static int ieee80211_change_mpath(struct wiphy *wiphy, struct net_device *dev,
2258                                   const u8 *dst, const u8 *next_hop)
2259 {
2260         struct ieee80211_sub_if_data *sdata;
2261         struct mesh_path *mpath;
2262         struct sta_info *sta;
2263
2264         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2265
2266         rcu_read_lock();
2267
2268         sta = sta_info_get(sdata, next_hop);
2269         if (!sta) {
2270                 rcu_read_unlock();
2271                 return -ENOENT;
2272         }
2273
2274         mpath = mesh_path_lookup(sdata, dst);
2275         if (!mpath) {
2276                 rcu_read_unlock();
2277                 return -ENOENT;
2278         }
2279
2280         mesh_path_fix_nexthop(mpath, sta);
2281
2282         rcu_read_unlock();
2283         return 0;
2284 }
2285
2286 static void mpath_set_pinfo(struct mesh_path *mpath, u8 *next_hop,
2287                             struct mpath_info *pinfo)
2288 {
2289         struct sta_info *next_hop_sta = rcu_dereference(mpath->next_hop);
2290
2291         if (next_hop_sta)
2292                 memcpy(next_hop, next_hop_sta->sta.addr, ETH_ALEN);
2293         else
2294                 eth_zero_addr(next_hop);
2295
2296         memset(pinfo, 0, sizeof(*pinfo));
2297
2298         pinfo->generation = mpath->sdata->u.mesh.mesh_paths_generation;
2299
2300         pinfo->filled = MPATH_INFO_FRAME_QLEN |
2301                         MPATH_INFO_SN |
2302                         MPATH_INFO_METRIC |
2303                         MPATH_INFO_EXPTIME |
2304                         MPATH_INFO_DISCOVERY_TIMEOUT |
2305                         MPATH_INFO_DISCOVERY_RETRIES |
2306                         MPATH_INFO_FLAGS |
2307                         MPATH_INFO_HOP_COUNT |
2308                         MPATH_INFO_PATH_CHANGE;
2309
2310         pinfo->frame_qlen = mpath->frame_queue.qlen;
2311         pinfo->sn = mpath->sn;
2312         pinfo->metric = mpath->metric;
2313         if (time_before(jiffies, mpath->exp_time))
2314                 pinfo->exptime = jiffies_to_msecs(mpath->exp_time - jiffies);
2315         pinfo->discovery_timeout =
2316                         jiffies_to_msecs(mpath->discovery_timeout);
2317         pinfo->discovery_retries = mpath->discovery_retries;
2318         if (mpath->flags & MESH_PATH_ACTIVE)
2319                 pinfo->flags |= NL80211_MPATH_FLAG_ACTIVE;
2320         if (mpath->flags & MESH_PATH_RESOLVING)
2321                 pinfo->flags |= NL80211_MPATH_FLAG_RESOLVING;
2322         if (mpath->flags & MESH_PATH_SN_VALID)
2323                 pinfo->flags |= NL80211_MPATH_FLAG_SN_VALID;
2324         if (mpath->flags & MESH_PATH_FIXED)
2325                 pinfo->flags |= NL80211_MPATH_FLAG_FIXED;
2326         if (mpath->flags & MESH_PATH_RESOLVED)
2327                 pinfo->flags |= NL80211_MPATH_FLAG_RESOLVED;
2328         pinfo->hop_count = mpath->hop_count;
2329         pinfo->path_change_count = mpath->path_change_count;
2330 }
2331
2332 static int ieee80211_get_mpath(struct wiphy *wiphy, struct net_device *dev,
2333                                u8 *dst, u8 *next_hop, struct mpath_info *pinfo)
2334
2335 {
2336         struct ieee80211_sub_if_data *sdata;
2337         struct mesh_path *mpath;
2338
2339         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2340
2341         rcu_read_lock();
2342         mpath = mesh_path_lookup(sdata, dst);
2343         if (!mpath) {
2344                 rcu_read_unlock();
2345                 return -ENOENT;
2346         }
2347         memcpy(dst, mpath->dst, ETH_ALEN);
2348         mpath_set_pinfo(mpath, next_hop, pinfo);
2349         rcu_read_unlock();
2350         return 0;
2351 }
2352
2353 static int ieee80211_dump_mpath(struct wiphy *wiphy, struct net_device *dev,
2354                                 int idx, u8 *dst, u8 *next_hop,
2355                                 struct mpath_info *pinfo)
2356 {
2357         struct ieee80211_sub_if_data *sdata;
2358         struct mesh_path *mpath;
2359
2360         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2361
2362         rcu_read_lock();
2363         mpath = mesh_path_lookup_by_idx(sdata, idx);
2364         if (!mpath) {
2365                 rcu_read_unlock();
2366                 return -ENOENT;
2367         }
2368         memcpy(dst, mpath->dst, ETH_ALEN);
2369         mpath_set_pinfo(mpath, next_hop, pinfo);
2370         rcu_read_unlock();
2371         return 0;
2372 }
2373
2374 static void mpp_set_pinfo(struct mesh_path *mpath, u8 *mpp,
2375                           struct mpath_info *pinfo)
2376 {
2377         memset(pinfo, 0, sizeof(*pinfo));
2378         memcpy(mpp, mpath->mpp, ETH_ALEN);
2379
2380         pinfo->generation = mpath->sdata->u.mesh.mpp_paths_generation;
2381 }
2382
2383 static int ieee80211_get_mpp(struct wiphy *wiphy, struct net_device *dev,
2384                              u8 *dst, u8 *mpp, struct mpath_info *pinfo)
2385
2386 {
2387         struct ieee80211_sub_if_data *sdata;
2388         struct mesh_path *mpath;
2389
2390         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2391
2392         rcu_read_lock();
2393         mpath = mpp_path_lookup(sdata, dst);
2394         if (!mpath) {
2395                 rcu_read_unlock();
2396                 return -ENOENT;
2397         }
2398         memcpy(dst, mpath->dst, ETH_ALEN);
2399         mpp_set_pinfo(mpath, mpp, pinfo);
2400         rcu_read_unlock();
2401         return 0;
2402 }
2403
2404 static int ieee80211_dump_mpp(struct wiphy *wiphy, struct net_device *dev,
2405                               int idx, u8 *dst, u8 *mpp,
2406                               struct mpath_info *pinfo)
2407 {
2408         struct ieee80211_sub_if_data *sdata;
2409         struct mesh_path *mpath;
2410
2411         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2412
2413         rcu_read_lock();
2414         mpath = mpp_path_lookup_by_idx(sdata, idx);
2415         if (!mpath) {
2416                 rcu_read_unlock();
2417                 return -ENOENT;
2418         }
2419         memcpy(dst, mpath->dst, ETH_ALEN);
2420         mpp_set_pinfo(mpath, mpp, pinfo);
2421         rcu_read_unlock();
2422         return 0;
2423 }
2424
2425 static int ieee80211_get_mesh_config(struct wiphy *wiphy,
2426                                 struct net_device *dev,
2427                                 struct mesh_config *conf)
2428 {
2429         struct ieee80211_sub_if_data *sdata;
2430         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2431
2432         memcpy(conf, &(sdata->u.mesh.mshcfg), sizeof(struct mesh_config));
2433         return 0;
2434 }
2435
2436 static inline bool _chg_mesh_attr(enum nl80211_meshconf_params parm, u32 mask)
2437 {
2438         return (mask >> (parm-1)) & 0x1;
2439 }
2440
2441 static int copy_mesh_setup(struct ieee80211_if_mesh *ifmsh,
2442                 const struct mesh_setup *setup)
2443 {
2444         u8 *new_ie;
2445         struct ieee80211_sub_if_data *sdata = container_of(ifmsh,
2446                                         struct ieee80211_sub_if_data, u.mesh);
2447         int i;
2448
2449         /* allocate information elements */
2450         new_ie = NULL;
2451
2452         if (setup->ie_len) {
2453                 new_ie = kmemdup(setup->ie, setup->ie_len,
2454                                 GFP_KERNEL);
2455                 if (!new_ie)
2456                         return -ENOMEM;
2457         }
2458         ifmsh->ie_len = setup->ie_len;
2459         ifmsh->ie = new_ie;
2460
2461         /* now copy the rest of the setup parameters */
2462         ifmsh->mesh_id_len = setup->mesh_id_len;
2463         memcpy(ifmsh->mesh_id, setup->mesh_id, ifmsh->mesh_id_len);
2464         ifmsh->mesh_sp_id = setup->sync_method;
2465         ifmsh->mesh_pp_id = setup->path_sel_proto;
2466         ifmsh->mesh_pm_id = setup->path_metric;
2467         ifmsh->user_mpm = setup->user_mpm;
2468         ifmsh->mesh_auth_id = setup->auth_id;
2469         ifmsh->security = IEEE80211_MESH_SEC_NONE;
2470         ifmsh->userspace_handles_dfs = setup->userspace_handles_dfs;
2471         if (setup->is_authenticated)
2472                 ifmsh->security |= IEEE80211_MESH_SEC_AUTHED;
2473         if (setup->is_secure)
2474                 ifmsh->security |= IEEE80211_MESH_SEC_SECURED;
2475
2476         /* mcast rate setting in Mesh Node */
2477         memcpy(sdata->vif.bss_conf.mcast_rate, setup->mcast_rate,
2478                                                 sizeof(setup->mcast_rate));
2479         sdata->vif.bss_conf.basic_rates = setup->basic_rates;
2480
2481         sdata->vif.bss_conf.beacon_int = setup->beacon_interval;
2482         sdata->vif.bss_conf.dtim_period = setup->dtim_period;
2483
2484         sdata->beacon_rate_set = false;
2485         if (wiphy_ext_feature_isset(sdata->local->hw.wiphy,
2486                                     NL80211_EXT_FEATURE_BEACON_RATE_LEGACY)) {
2487                 for (i = 0; i < NUM_NL80211_BANDS; i++) {
2488                         sdata->beacon_rateidx_mask[i] =
2489                                 setup->beacon_rate.control[i].legacy;
2490                         if (sdata->beacon_rateidx_mask[i])
2491                                 sdata->beacon_rate_set = true;
2492                 }
2493         }
2494
2495         return 0;
2496 }
2497
2498 static int ieee80211_update_mesh_config(struct wiphy *wiphy,
2499                                         struct net_device *dev, u32 mask,
2500                                         const struct mesh_config *nconf)
2501 {
2502         struct mesh_config *conf;
2503         struct ieee80211_sub_if_data *sdata;
2504         struct ieee80211_if_mesh *ifmsh;
2505
2506         sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2507         ifmsh = &sdata->u.mesh;
2508
2509         /* Set the config options which we are interested in setting */
2510         conf = &(sdata->u.mesh.mshcfg);
2511         if (_chg_mesh_attr(NL80211_MESHCONF_RETRY_TIMEOUT, mask))
2512                 conf->dot11MeshRetryTimeout = nconf->dot11MeshRetryTimeout;
2513         if (_chg_mesh_attr(NL80211_MESHCONF_CONFIRM_TIMEOUT, mask))
2514                 conf->dot11MeshConfirmTimeout = nconf->dot11MeshConfirmTimeout;
2515         if (_chg_mesh_attr(NL80211_MESHCONF_HOLDING_TIMEOUT, mask))
2516                 conf->dot11MeshHoldingTimeout = nconf->dot11MeshHoldingTimeout;
2517         if (_chg_mesh_attr(NL80211_MESHCONF_MAX_PEER_LINKS, mask))
2518                 conf->dot11MeshMaxPeerLinks = nconf->dot11MeshMaxPeerLinks;
2519         if (_chg_mesh_attr(NL80211_MESHCONF_MAX_RETRIES, mask))
2520                 conf->dot11MeshMaxRetries = nconf->dot11MeshMaxRetries;
2521         if (_chg_mesh_attr(NL80211_MESHCONF_TTL, mask))
2522                 conf->dot11MeshTTL = nconf->dot11MeshTTL;
2523         if (_chg_mesh_attr(NL80211_MESHCONF_ELEMENT_TTL, mask))
2524                 conf->element_ttl = nconf->element_ttl;
2525         if (_chg_mesh_attr(NL80211_MESHCONF_AUTO_OPEN_PLINKS, mask)) {
2526                 if (ifmsh->user_mpm)
2527                         return -EBUSY;
2528                 conf->auto_open_plinks = nconf->auto_open_plinks;
2529         }
2530         if (_chg_mesh_attr(NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR, mask))
2531                 conf->dot11MeshNbrOffsetMaxNeighbor =
2532                         nconf->dot11MeshNbrOffsetMaxNeighbor;
2533         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES, mask))
2534                 conf->dot11MeshHWMPmaxPREQretries =
2535                         nconf->dot11MeshHWMPmaxPREQretries;
2536         if (_chg_mesh_attr(NL80211_MESHCONF_PATH_REFRESH_TIME, mask))
2537                 conf->path_refresh_time = nconf->path_refresh_time;
2538         if (_chg_mesh_attr(NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT, mask))
2539                 conf->min_discovery_timeout = nconf->min_discovery_timeout;
2540         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT, mask))
2541                 conf->dot11MeshHWMPactivePathTimeout =
2542                         nconf->dot11MeshHWMPactivePathTimeout;
2543         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL, mask))
2544                 conf->dot11MeshHWMPpreqMinInterval =
2545                         nconf->dot11MeshHWMPpreqMinInterval;
2546         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL, mask))
2547                 conf->dot11MeshHWMPperrMinInterval =
2548                         nconf->dot11MeshHWMPperrMinInterval;
2549         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
2550                            mask))
2551                 conf->dot11MeshHWMPnetDiameterTraversalTime =
2552                         nconf->dot11MeshHWMPnetDiameterTraversalTime;
2553         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_ROOTMODE, mask)) {
2554                 conf->dot11MeshHWMPRootMode = nconf->dot11MeshHWMPRootMode;
2555                 ieee80211_mesh_root_setup(ifmsh);
2556         }
2557         if (_chg_mesh_attr(NL80211_MESHCONF_GATE_ANNOUNCEMENTS, mask)) {
2558                 /* our current gate announcement implementation rides on root
2559                  * announcements, so require this ifmsh to also be a root node
2560                  * */
2561                 if (nconf->dot11MeshGateAnnouncementProtocol &&
2562                     !(conf->dot11MeshHWMPRootMode > IEEE80211_ROOTMODE_ROOT)) {
2563                         conf->dot11MeshHWMPRootMode = IEEE80211_PROACTIVE_RANN;
2564                         ieee80211_mesh_root_setup(ifmsh);
2565                 }
2566                 conf->dot11MeshGateAnnouncementProtocol =
2567                         nconf->dot11MeshGateAnnouncementProtocol;
2568         }
2569         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_RANN_INTERVAL, mask))
2570                 conf->dot11MeshHWMPRannInterval =
2571                         nconf->dot11MeshHWMPRannInterval;
2572         if (_chg_mesh_attr(NL80211_MESHCONF_FORWARDING, mask))
2573                 conf->dot11MeshForwarding = nconf->dot11MeshForwarding;
2574         if (_chg_mesh_attr(NL80211_MESHCONF_RSSI_THRESHOLD, mask)) {
2575                 /* our RSSI threshold implementation is supported only for
2576                  * devices that report signal in dBm.
2577                  */
2578                 if (!ieee80211_hw_check(&sdata->local->hw, SIGNAL_DBM))
2579                         return -ENOTSUPP;
2580                 conf->rssi_threshold = nconf->rssi_threshold;
2581         }
2582         if (_chg_mesh_attr(NL80211_MESHCONF_HT_OPMODE, mask)) {
2583                 conf->ht_opmode = nconf->ht_opmode;
2584                 sdata->vif.bss_conf.ht_operation_mode = nconf->ht_opmode;
2585                 ieee80211_link_info_change_notify(sdata, &sdata->deflink,
2586                                                   BSS_CHANGED_HT);
2587         }
2588         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT, mask))
2589                 conf->dot11MeshHWMPactivePathToRootTimeout =
2590                         nconf->dot11MeshHWMPactivePathToRootTimeout;
2591         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_ROOT_INTERVAL, mask))
2592                 conf->dot11MeshHWMProotInterval =
2593                         nconf->dot11MeshHWMProotInterval;
2594         if (_chg_mesh_attr(NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL, mask))
2595                 conf->dot11MeshHWMPconfirmationInterval =
2596                         nconf->dot11MeshHWMPconfirmationInterval;
2597         if (_chg_mesh_attr(NL80211_MESHCONF_POWER_MODE, mask)) {
2598                 conf->power_mode = nconf->power_mode;
2599                 ieee80211_mps_local_status_update(sdata);
2600         }
2601         if (_chg_mesh_attr(NL80211_MESHCONF_AWAKE_WINDOW, mask))
2602                 conf->dot11MeshAwakeWindowDuration =
2603                         nconf->dot11MeshAwakeWindowDuration;
2604         if (_chg_mesh_attr(NL80211_MESHCONF_PLINK_TIMEOUT, mask))
2605                 conf->plink_timeout = nconf->plink_timeout;
2606         if (_chg_mesh_attr(NL80211_MESHCONF_CONNECTED_TO_GATE, mask))
2607                 conf->dot11MeshConnectedToMeshGate =
2608                         nconf->dot11MeshConnectedToMeshGate;
2609         if (_chg_mesh_attr(NL80211_MESHCONF_NOLEARN, mask))
2610                 conf->dot11MeshNolearn = nconf->dot11MeshNolearn;
2611         if (_chg_mesh_attr(NL80211_MESHCONF_CONNECTED_TO_AS, mask))
2612                 conf->dot11MeshConnectedToAuthServer =
2613                         nconf->dot11MeshConnectedToAuthServer;
2614         ieee80211_mbss_info_change_notify(sdata, BSS_CHANGED_BEACON);
2615         return 0;
2616 }
2617
2618 static int ieee80211_join_mesh(struct wiphy *wiphy, struct net_device *dev,
2619                                const struct mesh_config *conf,
2620                                const struct mesh_setup *setup)
2621 {
2622         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2623         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
2624         int err;
2625
2626         memcpy(&ifmsh->mshcfg, conf, sizeof(struct mesh_config));
2627         err = copy_mesh_setup(ifmsh, setup);
2628         if (err)
2629                 return err;
2630
2631         sdata->control_port_over_nl80211 = setup->control_port_over_nl80211;
2632
2633         /* can mesh use other SMPS modes? */
2634         sdata->deflink.smps_mode = IEEE80211_SMPS_OFF;
2635         sdata->deflink.needed_rx_chains = sdata->local->rx_chains;
2636
2637         mutex_lock(&sdata->local->mtx);
2638         err = ieee80211_link_use_channel(&sdata->deflink, &setup->chandef,
2639                                          IEEE80211_CHANCTX_SHARED);
2640         mutex_unlock(&sdata->local->mtx);
2641         if (err)
2642                 return err;
2643
2644         return ieee80211_start_mesh(sdata);
2645 }
2646
2647 static int ieee80211_leave_mesh(struct wiphy *wiphy, struct net_device *dev)
2648 {
2649         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2650
2651         ieee80211_stop_mesh(sdata);
2652         mutex_lock(&sdata->local->mtx);
2653         ieee80211_link_release_channel(&sdata->deflink);
2654         kfree(sdata->u.mesh.ie);
2655         mutex_unlock(&sdata->local->mtx);
2656
2657         return 0;
2658 }
2659 #endif
2660
2661 static int ieee80211_change_bss(struct wiphy *wiphy,
2662                                 struct net_device *dev,
2663                                 struct bss_parameters *params)
2664 {
2665         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2666         struct ieee80211_link_data *link;
2667         struct ieee80211_supported_band *sband;
2668         u32 changed = 0;
2669
2670         link = ieee80211_link_or_deflink(sdata, params->link_id, true);
2671         if (IS_ERR(link))
2672                 return PTR_ERR(link);
2673
2674         if (!sdata_dereference(link->u.ap.beacon, sdata))
2675                 return -ENOENT;
2676
2677         sband = ieee80211_get_link_sband(link);
2678         if (!sband)
2679                 return -EINVAL;
2680
2681         if (params->basic_rates) {
2682                 if (!ieee80211_parse_bitrates(link->conf->chandef.width,
2683                                               wiphy->bands[sband->band],
2684                                               params->basic_rates,
2685                                               params->basic_rates_len,
2686                                               &link->conf->basic_rates))
2687                         return -EINVAL;
2688                 changed |= BSS_CHANGED_BASIC_RATES;
2689                 ieee80211_check_rate_mask(link);
2690         }
2691
2692         if (params->use_cts_prot >= 0) {
2693                 link->conf->use_cts_prot = params->use_cts_prot;
2694                 changed |= BSS_CHANGED_ERP_CTS_PROT;
2695         }
2696         if (params->use_short_preamble >= 0) {
2697                 link->conf->use_short_preamble = params->use_short_preamble;
2698                 changed |= BSS_CHANGED_ERP_PREAMBLE;
2699         }
2700
2701         if (!link->conf->use_short_slot &&
2702             (sband->band == NL80211_BAND_5GHZ ||
2703              sband->band == NL80211_BAND_6GHZ)) {
2704                 link->conf->use_short_slot = true;
2705                 changed |= BSS_CHANGED_ERP_SLOT;
2706         }
2707
2708         if (params->use_short_slot_time >= 0) {
2709                 link->conf->use_short_slot = params->use_short_slot_time;
2710                 changed |= BSS_CHANGED_ERP_SLOT;
2711         }
2712
2713         if (params->ap_isolate >= 0) {
2714                 if (params->ap_isolate)
2715                         sdata->flags |= IEEE80211_SDATA_DONT_BRIDGE_PACKETS;
2716                 else
2717                         sdata->flags &= ~IEEE80211_SDATA_DONT_BRIDGE_PACKETS;
2718                 ieee80211_check_fast_rx_iface(sdata);
2719         }
2720
2721         if (params->ht_opmode >= 0) {
2722                 link->conf->ht_operation_mode = (u16)params->ht_opmode;
2723                 changed |= BSS_CHANGED_HT;
2724         }
2725
2726         if (params->p2p_ctwindow >= 0) {
2727                 link->conf->p2p_noa_attr.oppps_ctwindow &=
2728                                         ~IEEE80211_P2P_OPPPS_CTWINDOW_MASK;
2729                 link->conf->p2p_noa_attr.oppps_ctwindow |=
2730                         params->p2p_ctwindow & IEEE80211_P2P_OPPPS_CTWINDOW_MASK;
2731                 changed |= BSS_CHANGED_P2P_PS;
2732         }
2733
2734         if (params->p2p_opp_ps > 0) {
2735                 link->conf->p2p_noa_attr.oppps_ctwindow |=
2736                                         IEEE80211_P2P_OPPPS_ENABLE_BIT;
2737                 changed |= BSS_CHANGED_P2P_PS;
2738         } else if (params->p2p_opp_ps == 0) {
2739                 link->conf->p2p_noa_attr.oppps_ctwindow &=
2740                                         ~IEEE80211_P2P_OPPPS_ENABLE_BIT;
2741                 changed |= BSS_CHANGED_P2P_PS;
2742         }
2743
2744         ieee80211_link_info_change_notify(sdata, link, changed);
2745
2746         return 0;
2747 }
2748
2749 static int ieee80211_set_txq_params(struct wiphy *wiphy,
2750                                     struct net_device *dev,
2751                                     struct ieee80211_txq_params *params)
2752 {
2753         struct ieee80211_local *local = wiphy_priv(wiphy);
2754         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2755         struct ieee80211_link_data *link =
2756                 ieee80211_link_or_deflink(sdata, params->link_id, true);
2757         struct ieee80211_tx_queue_params p;
2758
2759         if (!local->ops->conf_tx)
2760                 return -EOPNOTSUPP;
2761
2762         if (local->hw.queues < IEEE80211_NUM_ACS)
2763                 return -EOPNOTSUPP;
2764
2765         if (IS_ERR(link))
2766                 return PTR_ERR(link);
2767
2768         memset(&p, 0, sizeof(p));
2769         p.aifs = params->aifs;
2770         p.cw_max = params->cwmax;
2771         p.cw_min = params->cwmin;
2772         p.txop = params->txop;
2773
2774         /*
2775          * Setting tx queue params disables u-apsd because it's only
2776          * called in master mode.
2777          */
2778         p.uapsd = false;
2779
2780         ieee80211_regulatory_limit_wmm_params(sdata, &p, params->ac);
2781
2782         link->tx_conf[params->ac] = p;
2783         if (drv_conf_tx(local, link, params->ac, &p)) {
2784                 wiphy_debug(local->hw.wiphy,
2785                             "failed to set TX queue parameters for AC %d\n",
2786                             params->ac);
2787                 return -EINVAL;
2788         }
2789
2790         ieee80211_link_info_change_notify(sdata, link,
2791                                           BSS_CHANGED_QOS);
2792
2793         return 0;
2794 }
2795
2796 #ifdef CONFIG_PM
2797 static int ieee80211_suspend(struct wiphy *wiphy,
2798                              struct cfg80211_wowlan *wowlan)
2799 {
2800         return __ieee80211_suspend(wiphy_priv(wiphy), wowlan);
2801 }
2802
2803 static int ieee80211_resume(struct wiphy *wiphy)
2804 {
2805         return __ieee80211_resume(wiphy_priv(wiphy));
2806 }
2807 #else
2808 #define ieee80211_suspend NULL
2809 #define ieee80211_resume NULL
2810 #endif
2811
2812 static int ieee80211_scan(struct wiphy *wiphy,
2813                           struct cfg80211_scan_request *req)
2814 {
2815         struct ieee80211_sub_if_data *sdata;
2816
2817         sdata = IEEE80211_WDEV_TO_SUB_IF(req->wdev);
2818
2819         switch (ieee80211_vif_type_p2p(&sdata->vif)) {
2820         case NL80211_IFTYPE_STATION:
2821         case NL80211_IFTYPE_ADHOC:
2822         case NL80211_IFTYPE_MESH_POINT:
2823         case NL80211_IFTYPE_P2P_CLIENT:
2824         case NL80211_IFTYPE_P2P_DEVICE:
2825                 break;
2826         case NL80211_IFTYPE_P2P_GO:
2827                 if (sdata->local->ops->hw_scan)
2828                         break;
2829                 /*
2830                  * FIXME: implement NoA while scanning in software,
2831                  * for now fall through to allow scanning only when
2832                  * beaconing hasn't been configured yet
2833                  */
2834                 fallthrough;
2835         case NL80211_IFTYPE_AP:
2836                 /*
2837                  * If the scan has been forced (and the driver supports
2838                  * forcing), don't care about being beaconing already.
2839                  * This will create problems to the attached stations (e.g. all
2840                  * the frames sent while scanning on other channel will be
2841                  * lost)
2842                  */
2843                 if (sdata->deflink.u.ap.beacon &&
2844                     (!(wiphy->features & NL80211_FEATURE_AP_SCAN) ||
2845                      !(req->flags & NL80211_SCAN_FLAG_AP)))
2846                         return -EOPNOTSUPP;
2847                 break;
2848         case NL80211_IFTYPE_NAN:
2849         default:
2850                 return -EOPNOTSUPP;
2851         }
2852
2853         return ieee80211_request_scan(sdata, req);
2854 }
2855
2856 static void ieee80211_abort_scan(struct wiphy *wiphy, struct wireless_dev *wdev)
2857 {
2858         ieee80211_scan_cancel(wiphy_priv(wiphy));
2859 }
2860
2861 static int
2862 ieee80211_sched_scan_start(struct wiphy *wiphy,
2863                            struct net_device *dev,
2864                            struct cfg80211_sched_scan_request *req)
2865 {
2866         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2867
2868         if (!sdata->local->ops->sched_scan_start)
2869                 return -EOPNOTSUPP;
2870
2871         return ieee80211_request_sched_scan_start(sdata, req);
2872 }
2873
2874 static int
2875 ieee80211_sched_scan_stop(struct wiphy *wiphy, struct net_device *dev,
2876                           u64 reqid)
2877 {
2878         struct ieee80211_local *local = wiphy_priv(wiphy);
2879
2880         if (!local->ops->sched_scan_stop)
2881                 return -EOPNOTSUPP;
2882
2883         return ieee80211_request_sched_scan_stop(local);
2884 }
2885
2886 static int ieee80211_auth(struct wiphy *wiphy, struct net_device *dev,
2887                           struct cfg80211_auth_request *req)
2888 {
2889         return ieee80211_mgd_auth(IEEE80211_DEV_TO_SUB_IF(dev), req);
2890 }
2891
2892 static int ieee80211_assoc(struct wiphy *wiphy, struct net_device *dev,
2893                            struct cfg80211_assoc_request *req)
2894 {
2895         return ieee80211_mgd_assoc(IEEE80211_DEV_TO_SUB_IF(dev), req);
2896 }
2897
2898 static int ieee80211_deauth(struct wiphy *wiphy, struct net_device *dev,
2899                             struct cfg80211_deauth_request *req)
2900 {
2901         return ieee80211_mgd_deauth(IEEE80211_DEV_TO_SUB_IF(dev), req);
2902 }
2903
2904 static int ieee80211_disassoc(struct wiphy *wiphy, struct net_device *dev,
2905                               struct cfg80211_disassoc_request *req)
2906 {
2907         return ieee80211_mgd_disassoc(IEEE80211_DEV_TO_SUB_IF(dev), req);
2908 }
2909
2910 static int ieee80211_join_ibss(struct wiphy *wiphy, struct net_device *dev,
2911                                struct cfg80211_ibss_params *params)
2912 {
2913         return ieee80211_ibss_join(IEEE80211_DEV_TO_SUB_IF(dev), params);
2914 }
2915
2916 static int ieee80211_leave_ibss(struct wiphy *wiphy, struct net_device *dev)
2917 {
2918         return ieee80211_ibss_leave(IEEE80211_DEV_TO_SUB_IF(dev));
2919 }
2920
2921 static int ieee80211_join_ocb(struct wiphy *wiphy, struct net_device *dev,
2922                               struct ocb_setup *setup)
2923 {
2924         return ieee80211_ocb_join(IEEE80211_DEV_TO_SUB_IF(dev), setup);
2925 }
2926
2927 static int ieee80211_leave_ocb(struct wiphy *wiphy, struct net_device *dev)
2928 {
2929         return ieee80211_ocb_leave(IEEE80211_DEV_TO_SUB_IF(dev));
2930 }
2931
2932 static int ieee80211_set_mcast_rate(struct wiphy *wiphy, struct net_device *dev,
2933                                     int rate[NUM_NL80211_BANDS])
2934 {
2935         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
2936
2937         memcpy(sdata->vif.bss_conf.mcast_rate, rate,
2938                sizeof(int) * NUM_NL80211_BANDS);
2939
2940         ieee80211_link_info_change_notify(sdata, &sdata->deflink,
2941                                           BSS_CHANGED_MCAST_RATE);
2942
2943         return 0;
2944 }
2945
2946 static int ieee80211_set_wiphy_params(struct wiphy *wiphy, u32 changed)
2947 {
2948         struct ieee80211_local *local = wiphy_priv(wiphy);
2949         int err;
2950
2951         if (changed & WIPHY_PARAM_FRAG_THRESHOLD) {
2952                 ieee80211_check_fast_xmit_all(local);
2953
2954                 err = drv_set_frag_threshold(local, wiphy->frag_threshold);
2955
2956                 if (err) {
2957                         ieee80211_check_fast_xmit_all(local);
2958                         return err;
2959                 }
2960         }
2961
2962         if ((changed & WIPHY_PARAM_COVERAGE_CLASS) ||
2963             (changed & WIPHY_PARAM_DYN_ACK)) {
2964                 s16 coverage_class;
2965
2966                 coverage_class = changed & WIPHY_PARAM_COVERAGE_CLASS ?
2967                                         wiphy->coverage_class : -1;
2968                 err = drv_set_coverage_class(local, coverage_class);
2969
2970                 if (err)
2971                         return err;
2972         }
2973
2974         if (changed & WIPHY_PARAM_RTS_THRESHOLD) {
2975                 err = drv_set_rts_threshold(local, wiphy->rts_threshold);
2976
2977                 if (err)
2978                         return err;
2979         }
2980
2981         if (changed & WIPHY_PARAM_RETRY_SHORT) {
2982                 if (wiphy->retry_short > IEEE80211_MAX_TX_RETRY)
2983                         return -EINVAL;
2984                 local->hw.conf.short_frame_max_tx_count = wiphy->retry_short;
2985         }
2986         if (changed & WIPHY_PARAM_RETRY_LONG) {
2987                 if (wiphy->retry_long > IEEE80211_MAX_TX_RETRY)
2988                         return -EINVAL;
2989                 local->hw.conf.long_frame_max_tx_count = wiphy->retry_long;
2990         }
2991         if (changed &
2992             (WIPHY_PARAM_RETRY_SHORT | WIPHY_PARAM_RETRY_LONG))
2993                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_RETRY_LIMITS);
2994
2995         if (changed & (WIPHY_PARAM_TXQ_LIMIT |
2996                        WIPHY_PARAM_TXQ_MEMORY_LIMIT |
2997                        WIPHY_PARAM_TXQ_QUANTUM))
2998                 ieee80211_txq_set_params(local);
2999
3000         return 0;
3001 }
3002
3003 static int ieee80211_set_tx_power(struct wiphy *wiphy,
3004                                   struct wireless_dev *wdev,
3005                                   enum nl80211_tx_power_setting type, int mbm)
3006 {
3007         struct ieee80211_local *local = wiphy_priv(wiphy);
3008         struct ieee80211_sub_if_data *sdata;
3009         enum nl80211_tx_power_setting txp_type = type;
3010         bool update_txp_type = false;
3011         bool has_monitor = false;
3012
3013         if (wdev) {
3014                 sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
3015
3016                 if (sdata->vif.type == NL80211_IFTYPE_MONITOR) {
3017                         sdata = wiphy_dereference(local->hw.wiphy,
3018                                                   local->monitor_sdata);
3019                         if (!sdata)
3020                                 return -EOPNOTSUPP;
3021                 }
3022
3023                 switch (type) {
3024                 case NL80211_TX_POWER_AUTOMATIC:
3025                         sdata->deflink.user_power_level =
3026                                 IEEE80211_UNSET_POWER_LEVEL;
3027                         txp_type = NL80211_TX_POWER_LIMITED;
3028                         break;
3029                 case NL80211_TX_POWER_LIMITED:
3030                 case NL80211_TX_POWER_FIXED:
3031                         if (mbm < 0 || (mbm % 100))
3032                                 return -EOPNOTSUPP;
3033                         sdata->deflink.user_power_level = MBM_TO_DBM(mbm);
3034                         break;
3035                 }
3036
3037                 if (txp_type != sdata->vif.bss_conf.txpower_type) {
3038                         update_txp_type = true;
3039                         sdata->vif.bss_conf.txpower_type = txp_type;
3040                 }
3041
3042                 ieee80211_recalc_txpower(sdata, update_txp_type);
3043
3044                 return 0;
3045         }
3046
3047         switch (type) {
3048         case NL80211_TX_POWER_AUTOMATIC:
3049                 local->user_power_level = IEEE80211_UNSET_POWER_LEVEL;
3050                 txp_type = NL80211_TX_POWER_LIMITED;
3051                 break;
3052         case NL80211_TX_POWER_LIMITED:
3053         case NL80211_TX_POWER_FIXED:
3054                 if (mbm < 0 || (mbm % 100))
3055                         return -EOPNOTSUPP;
3056                 local->user_power_level = MBM_TO_DBM(mbm);
3057                 break;
3058         }
3059
3060         mutex_lock(&local->iflist_mtx);
3061         list_for_each_entry(sdata, &local->interfaces, list) {
3062                 if (sdata->vif.type == NL80211_IFTYPE_MONITOR) {
3063                         has_monitor = true;
3064                         continue;
3065                 }
3066                 sdata->deflink.user_power_level = local->user_power_level;
3067                 if (txp_type != sdata->vif.bss_conf.txpower_type)
3068                         update_txp_type = true;
3069                 sdata->vif.bss_conf.txpower_type = txp_type;
3070         }
3071         list_for_each_entry(sdata, &local->interfaces, list) {
3072                 if (sdata->vif.type == NL80211_IFTYPE_MONITOR)
3073                         continue;
3074                 ieee80211_recalc_txpower(sdata, update_txp_type);
3075         }
3076         mutex_unlock(&local->iflist_mtx);
3077
3078         if (has_monitor) {
3079                 sdata = wiphy_dereference(local->hw.wiphy,
3080                                           local->monitor_sdata);
3081                 if (sdata) {
3082                         sdata->deflink.user_power_level = local->user_power_level;
3083                         if (txp_type != sdata->vif.bss_conf.txpower_type)
3084                                 update_txp_type = true;
3085                         sdata->vif.bss_conf.txpower_type = txp_type;
3086
3087                         ieee80211_recalc_txpower(sdata, update_txp_type);
3088                 }
3089         }
3090
3091         return 0;
3092 }
3093
3094 static int ieee80211_get_tx_power(struct wiphy *wiphy,
3095                                   struct wireless_dev *wdev,
3096                                   int *dbm)
3097 {
3098         struct ieee80211_local *local = wiphy_priv(wiphy);
3099         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
3100
3101         if (local->ops->get_txpower)
3102                 return drv_get_txpower(local, sdata, dbm);
3103
3104         if (!local->use_chanctx)
3105                 *dbm = local->hw.conf.power_level;
3106         else
3107                 *dbm = sdata->vif.bss_conf.txpower;
3108
3109         return 0;
3110 }
3111
3112 static void ieee80211_rfkill_poll(struct wiphy *wiphy)
3113 {
3114         struct ieee80211_local *local = wiphy_priv(wiphy);
3115
3116         drv_rfkill_poll(local);
3117 }
3118
3119 #ifdef CONFIG_NL80211_TESTMODE
3120 static int ieee80211_testmode_cmd(struct wiphy *wiphy,
3121                                   struct wireless_dev *wdev,
3122                                   void *data, int len)
3123 {
3124         struct ieee80211_local *local = wiphy_priv(wiphy);
3125         struct ieee80211_vif *vif = NULL;
3126
3127         if (!local->ops->testmode_cmd)
3128                 return -EOPNOTSUPP;
3129
3130         if (wdev) {
3131                 struct ieee80211_sub_if_data *sdata;
3132
3133                 sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
3134                 if (sdata->flags & IEEE80211_SDATA_IN_DRIVER)
3135                         vif = &sdata->vif;
3136         }
3137
3138         return local->ops->testmode_cmd(&local->hw, vif, data, len);
3139 }
3140
3141 static int ieee80211_testmode_dump(struct wiphy *wiphy,
3142                                    struct sk_buff *skb,
3143                                    struct netlink_callback *cb,
3144                                    void *data, int len)
3145 {
3146         struct ieee80211_local *local = wiphy_priv(wiphy);
3147
3148         if (!local->ops->testmode_dump)
3149                 return -EOPNOTSUPP;
3150
3151         return local->ops->testmode_dump(&local->hw, skb, cb, data, len);
3152 }
3153 #endif
3154
3155 int __ieee80211_request_smps_mgd(struct ieee80211_sub_if_data *sdata,
3156                                  struct ieee80211_link_data *link,
3157                                  enum ieee80211_smps_mode smps_mode)
3158 {
3159         const u8 *ap;
3160         enum ieee80211_smps_mode old_req;
3161         int err;
3162         struct sta_info *sta;
3163         bool tdls_peer_found = false;
3164
3165         lockdep_assert_held(&sdata->wdev.mtx);
3166
3167         if (WARN_ON_ONCE(sdata->vif.type != NL80211_IFTYPE_STATION))
3168                 return -EINVAL;
3169
3170         old_req = link->u.mgd.req_smps;
3171         link->u.mgd.req_smps = smps_mode;
3172
3173         if (old_req == smps_mode &&
3174             smps_mode != IEEE80211_SMPS_AUTOMATIC)
3175                 return 0;
3176
3177         /*
3178          * If not associated, or current association is not an HT
3179          * association, there's no need to do anything, just store
3180          * the new value until we associate.
3181          */
3182         if (!sdata->u.mgd.associated ||
3183             link->conf->chandef.width == NL80211_CHAN_WIDTH_20_NOHT)
3184                 return 0;
3185
3186         ap = link->u.mgd.bssid;
3187
3188         rcu_read_lock();
3189         list_for_each_entry_rcu(sta, &sdata->local->sta_list, list) {
3190                 if (!sta->sta.tdls || sta->sdata != sdata || !sta->uploaded ||
3191                     !test_sta_flag(sta, WLAN_STA_AUTHORIZED))
3192                         continue;
3193
3194                 tdls_peer_found = true;
3195                 break;
3196         }
3197         rcu_read_unlock();
3198
3199         if (smps_mode == IEEE80211_SMPS_AUTOMATIC) {
3200                 if (tdls_peer_found || !sdata->u.mgd.powersave)
3201                         smps_mode = IEEE80211_SMPS_OFF;
3202                 else
3203                         smps_mode = IEEE80211_SMPS_DYNAMIC;
3204         }
3205
3206         /* send SM PS frame to AP */
3207         err = ieee80211_send_smps_action(sdata, smps_mode,
3208                                          ap, ap);
3209         if (err)
3210                 link->u.mgd.req_smps = old_req;
3211         else if (smps_mode != IEEE80211_SMPS_OFF && tdls_peer_found)
3212                 ieee80211_teardown_tdls_peers(sdata);
3213
3214         return err;
3215 }
3216
3217 static int ieee80211_set_power_mgmt(struct wiphy *wiphy, struct net_device *dev,
3218                                     bool enabled, int timeout)
3219 {
3220         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3221         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
3222         unsigned int link_id;
3223
3224         if (sdata->vif.type != NL80211_IFTYPE_STATION)
3225                 return -EOPNOTSUPP;
3226
3227         if (!ieee80211_hw_check(&local->hw, SUPPORTS_PS))
3228                 return -EOPNOTSUPP;
3229
3230         if (enabled == sdata->u.mgd.powersave &&
3231             timeout == local->dynamic_ps_forced_timeout)
3232                 return 0;
3233
3234         sdata->u.mgd.powersave = enabled;
3235         local->dynamic_ps_forced_timeout = timeout;
3236
3237         /* no change, but if automatic follow powersave */
3238         sdata_lock(sdata);
3239         for (link_id = 0; link_id < ARRAY_SIZE(sdata->link); link_id++) {
3240                 struct ieee80211_link_data *link;
3241
3242                 link = sdata_dereference(sdata->link[link_id], sdata);
3243
3244                 if (!link)
3245                         continue;
3246                 __ieee80211_request_smps_mgd(sdata, link,
3247                                              link->u.mgd.req_smps);
3248         }
3249         sdata_unlock(sdata);
3250
3251         if (ieee80211_hw_check(&local->hw, SUPPORTS_DYNAMIC_PS))
3252                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
3253
3254         ieee80211_recalc_ps(local);
3255         ieee80211_recalc_ps_vif(sdata);
3256         ieee80211_check_fast_rx_iface(sdata);
3257
3258         return 0;
3259 }
3260
3261 static int ieee80211_set_cqm_rssi_config(struct wiphy *wiphy,
3262                                          struct net_device *dev,
3263                                          s32 rssi_thold, u32 rssi_hyst)
3264 {
3265         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3266         struct ieee80211_vif *vif = &sdata->vif;
3267         struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
3268
3269         if (rssi_thold == bss_conf->cqm_rssi_thold &&
3270             rssi_hyst == bss_conf->cqm_rssi_hyst)
3271                 return 0;
3272
3273         if (sdata->vif.driver_flags & IEEE80211_VIF_BEACON_FILTER &&
3274             !(sdata->vif.driver_flags & IEEE80211_VIF_SUPPORTS_CQM_RSSI))
3275                 return -EOPNOTSUPP;
3276
3277         bss_conf->cqm_rssi_thold = rssi_thold;
3278         bss_conf->cqm_rssi_hyst = rssi_hyst;
3279         bss_conf->cqm_rssi_low = 0;
3280         bss_conf->cqm_rssi_high = 0;
3281         sdata->deflink.u.mgd.last_cqm_event_signal = 0;
3282
3283         /* tell the driver upon association, unless already associated */
3284         if (sdata->u.mgd.associated &&
3285             sdata->vif.driver_flags & IEEE80211_VIF_SUPPORTS_CQM_RSSI)
3286                 ieee80211_link_info_change_notify(sdata, &sdata->deflink,
3287                                                   BSS_CHANGED_CQM);
3288
3289         return 0;
3290 }
3291
3292 static int ieee80211_set_cqm_rssi_range_config(struct wiphy *wiphy,
3293                                                struct net_device *dev,
3294                                                s32 rssi_low, s32 rssi_high)
3295 {
3296         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3297         struct ieee80211_vif *vif = &sdata->vif;
3298         struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
3299
3300         if (sdata->vif.driver_flags & IEEE80211_VIF_BEACON_FILTER)
3301                 return -EOPNOTSUPP;
3302
3303         bss_conf->cqm_rssi_low = rssi_low;
3304         bss_conf->cqm_rssi_high = rssi_high;
3305         bss_conf->cqm_rssi_thold = 0;
3306         bss_conf->cqm_rssi_hyst = 0;
3307         sdata->deflink.u.mgd.last_cqm_event_signal = 0;
3308
3309         /* tell the driver upon association, unless already associated */
3310         if (sdata->u.mgd.associated &&
3311             sdata->vif.driver_flags & IEEE80211_VIF_SUPPORTS_CQM_RSSI)
3312                 ieee80211_link_info_change_notify(sdata, &sdata->deflink,
3313                                                   BSS_CHANGED_CQM);
3314
3315         return 0;
3316 }
3317
3318 static int ieee80211_set_bitrate_mask(struct wiphy *wiphy,
3319                                       struct net_device *dev,
3320                                       unsigned int link_id,
3321                                       const u8 *addr,
3322                                       const struct cfg80211_bitrate_mask *mask)
3323 {
3324         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3325         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
3326         int i, ret;
3327
3328         if (!ieee80211_sdata_running(sdata))
3329                 return -ENETDOWN;
3330
3331         /*
3332          * If active validate the setting and reject it if it doesn't leave
3333          * at least one basic rate usable, since we really have to be able
3334          * to send something, and if we're an AP we have to be able to do
3335          * so at a basic rate so that all clients can receive it.
3336          */
3337         if (rcu_access_pointer(sdata->vif.bss_conf.chanctx_conf) &&
3338             sdata->vif.bss_conf.chandef.chan) {
3339                 u32 basic_rates = sdata->vif.bss_conf.basic_rates;
3340                 enum nl80211_band band = sdata->vif.bss_conf.chandef.chan->band;
3341
3342                 if (!(mask->control[band].legacy & basic_rates))
3343                         return -EINVAL;
3344         }
3345
3346         if (ieee80211_hw_check(&local->hw, HAS_RATE_CONTROL)) {
3347                 ret = drv_set_bitrate_mask(local, sdata, mask);
3348                 if (ret)
3349                         return ret;
3350         }
3351
3352         for (i = 0; i < NUM_NL80211_BANDS; i++) {
3353                 struct ieee80211_supported_band *sband = wiphy->bands[i];
3354                 int j;
3355
3356                 sdata->rc_rateidx_mask[i] = mask->control[i].legacy;
3357                 memcpy(sdata->rc_rateidx_mcs_mask[i], mask->control[i].ht_mcs,
3358                        sizeof(mask->control[i].ht_mcs));
3359                 memcpy(sdata->rc_rateidx_vht_mcs_mask[i],
3360                        mask->control[i].vht_mcs,
3361                        sizeof(mask->control[i].vht_mcs));
3362
3363                 sdata->rc_has_mcs_mask[i] = false;
3364                 sdata->rc_has_vht_mcs_mask[i] = false;
3365                 if (!sband)
3366                         continue;
3367
3368                 for (j = 0; j < IEEE80211_HT_MCS_MASK_LEN; j++) {
3369                         if (sdata->rc_rateidx_mcs_mask[i][j] != 0xff) {
3370                                 sdata->rc_has_mcs_mask[i] = true;
3371                                 break;
3372                         }
3373                 }
3374
3375                 for (j = 0; j < NL80211_VHT_NSS_MAX; j++) {
3376                         if (sdata->rc_rateidx_vht_mcs_mask[i][j] != 0xffff) {
3377                                 sdata->rc_has_vht_mcs_mask[i] = true;
3378                                 break;
3379                         }
3380                 }
3381         }
3382
3383         return 0;
3384 }
3385
3386 static int ieee80211_start_radar_detection(struct wiphy *wiphy,
3387                                            struct net_device *dev,
3388                                            struct cfg80211_chan_def *chandef,
3389                                            u32 cac_time_ms)
3390 {
3391         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3392         struct ieee80211_local *local = sdata->local;
3393         int err;
3394
3395         mutex_lock(&local->mtx);
3396         if (!list_empty(&local->roc_list) || local->scanning) {
3397                 err = -EBUSY;
3398                 goto out_unlock;
3399         }
3400
3401         /* whatever, but channel contexts should not complain about that one */
3402         sdata->deflink.smps_mode = IEEE80211_SMPS_OFF;
3403         sdata->deflink.needed_rx_chains = local->rx_chains;
3404
3405         err = ieee80211_link_use_channel(&sdata->deflink, chandef,
3406                                          IEEE80211_CHANCTX_SHARED);
3407         if (err)
3408                 goto out_unlock;
3409
3410         ieee80211_queue_delayed_work(&sdata->local->hw,
3411                                      &sdata->deflink.dfs_cac_timer_work,
3412                                      msecs_to_jiffies(cac_time_ms));
3413
3414  out_unlock:
3415         mutex_unlock(&local->mtx);
3416         return err;
3417 }
3418
3419 static void ieee80211_end_cac(struct wiphy *wiphy,
3420                               struct net_device *dev)
3421 {
3422         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3423         struct ieee80211_local *local = sdata->local;
3424
3425         mutex_lock(&local->mtx);
3426         list_for_each_entry(sdata, &local->interfaces, list) {
3427                 /* it might be waiting for the local->mtx, but then
3428                  * by the time it gets it, sdata->wdev.cac_started
3429                  * will no longer be true
3430                  */
3431                 cancel_delayed_work(&sdata->deflink.dfs_cac_timer_work);
3432
3433                 if (sdata->wdev.cac_started) {
3434                         ieee80211_link_release_channel(&sdata->deflink);
3435                         sdata->wdev.cac_started = false;
3436                 }
3437         }
3438         mutex_unlock(&local->mtx);
3439 }
3440
3441 static struct cfg80211_beacon_data *
3442 cfg80211_beacon_dup(struct cfg80211_beacon_data *beacon)
3443 {
3444         struct cfg80211_beacon_data *new_beacon;
3445         u8 *pos;
3446         int len;
3447
3448         len = beacon->head_len + beacon->tail_len + beacon->beacon_ies_len +
3449               beacon->proberesp_ies_len + beacon->assocresp_ies_len +
3450               beacon->probe_resp_len + beacon->lci_len + beacon->civicloc_len;
3451
3452         if (beacon->mbssid_ies)
3453                 len += ieee80211_get_mbssid_beacon_len(beacon->mbssid_ies,
3454                                                        beacon->rnr_ies,
3455                                                        beacon->mbssid_ies->cnt);
3456
3457         new_beacon = kzalloc(sizeof(*new_beacon) + len, GFP_KERNEL);
3458         if (!new_beacon)
3459                 return NULL;
3460
3461         if (beacon->mbssid_ies && beacon->mbssid_ies->cnt) {
3462                 new_beacon->mbssid_ies =
3463                         kzalloc(struct_size(new_beacon->mbssid_ies,
3464                                             elem, beacon->mbssid_ies->cnt),
3465                                 GFP_KERNEL);
3466                 if (!new_beacon->mbssid_ies) {
3467                         kfree(new_beacon);
3468                         return NULL;
3469                 }
3470
3471                 if (beacon->rnr_ies && beacon->rnr_ies->cnt) {
3472                         new_beacon->rnr_ies =
3473                                 kzalloc(struct_size(new_beacon->rnr_ies,
3474                                                     elem, beacon->rnr_ies->cnt),
3475                                         GFP_KERNEL);
3476                         if (!new_beacon->rnr_ies) {
3477                                 kfree(new_beacon->mbssid_ies);
3478                                 kfree(new_beacon);
3479                                 return NULL;
3480                         }
3481                 }
3482         }
3483
3484         pos = (u8 *)(new_beacon + 1);
3485         if (beacon->head_len) {
3486                 new_beacon->head_len = beacon->head_len;
3487                 new_beacon->head = pos;
3488                 memcpy(pos, beacon->head, beacon->head_len);
3489                 pos += beacon->head_len;
3490         }
3491         if (beacon->tail_len) {
3492                 new_beacon->tail_len = beacon->tail_len;
3493                 new_beacon->tail = pos;
3494                 memcpy(pos, beacon->tail, beacon->tail_len);
3495                 pos += beacon->tail_len;
3496         }
3497         if (beacon->beacon_ies_len) {
3498                 new_beacon->beacon_ies_len = beacon->beacon_ies_len;
3499                 new_beacon->beacon_ies = pos;
3500                 memcpy(pos, beacon->beacon_ies, beacon->beacon_ies_len);
3501                 pos += beacon->beacon_ies_len;
3502         }
3503         if (beacon->proberesp_ies_len) {
3504                 new_beacon->proberesp_ies_len = beacon->proberesp_ies_len;
3505                 new_beacon->proberesp_ies = pos;
3506                 memcpy(pos, beacon->proberesp_ies, beacon->proberesp_ies_len);
3507                 pos += beacon->proberesp_ies_len;
3508         }
3509         if (beacon->assocresp_ies_len) {
3510                 new_beacon->assocresp_ies_len = beacon->assocresp_ies_len;
3511                 new_beacon->assocresp_ies = pos;
3512                 memcpy(pos, beacon->assocresp_ies, beacon->assocresp_ies_len);
3513                 pos += beacon->assocresp_ies_len;
3514         }
3515         if (beacon->probe_resp_len) {
3516                 new_beacon->probe_resp_len = beacon->probe_resp_len;
3517                 new_beacon->probe_resp = pos;
3518                 memcpy(pos, beacon->probe_resp, beacon->probe_resp_len);
3519                 pos += beacon->probe_resp_len;
3520         }
3521         if (beacon->mbssid_ies && beacon->mbssid_ies->cnt) {
3522                 pos += ieee80211_copy_mbssid_beacon(pos,
3523                                                     new_beacon->mbssid_ies,
3524                                                     beacon->mbssid_ies);
3525                 if (beacon->rnr_ies && beacon->rnr_ies->cnt)
3526                         pos += ieee80211_copy_rnr_beacon(pos,
3527                                                          new_beacon->rnr_ies,
3528                                                          beacon->rnr_ies);
3529         }
3530
3531         /* might copy -1, meaning no changes requested */
3532         new_beacon->ftm_responder = beacon->ftm_responder;
3533         if (beacon->lci) {
3534                 new_beacon->lci_len = beacon->lci_len;
3535                 new_beacon->lci = pos;
3536                 memcpy(pos, beacon->lci, beacon->lci_len);
3537                 pos += beacon->lci_len;
3538         }
3539         if (beacon->civicloc) {
3540                 new_beacon->civicloc_len = beacon->civicloc_len;
3541                 new_beacon->civicloc = pos;
3542                 memcpy(pos, beacon->civicloc, beacon->civicloc_len);
3543                 pos += beacon->civicloc_len;
3544         }
3545
3546         return new_beacon;
3547 }
3548
3549 void ieee80211_csa_finish(struct ieee80211_vif *vif)
3550 {
3551         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
3552         struct ieee80211_local *local = sdata->local;
3553
3554         rcu_read_lock();
3555
3556         if (vif->mbssid_tx_vif == vif) {
3557                 /* Trigger ieee80211_csa_finish() on the non-transmitting
3558                  * interfaces when channel switch is received on
3559                  * transmitting interface
3560                  */
3561                 struct ieee80211_sub_if_data *iter;
3562
3563                 list_for_each_entry_rcu(iter, &local->interfaces, list) {
3564                         if (!ieee80211_sdata_running(iter))
3565                                 continue;
3566
3567                         if (iter == sdata || iter->vif.mbssid_tx_vif != vif)
3568                                 continue;
3569
3570                         ieee80211_queue_work(&iter->local->hw,
3571                                              &iter->deflink.csa_finalize_work);
3572                 }
3573         }
3574         ieee80211_queue_work(&local->hw, &sdata->deflink.csa_finalize_work);
3575
3576         rcu_read_unlock();
3577 }
3578 EXPORT_SYMBOL(ieee80211_csa_finish);
3579
3580 void ieee80211_channel_switch_disconnect(struct ieee80211_vif *vif, bool block_tx)
3581 {
3582         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
3583         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
3584         struct ieee80211_local *local = sdata->local;
3585
3586         sdata->deflink.csa_block_tx = block_tx;
3587         sdata_info(sdata, "channel switch failed, disconnecting\n");
3588         ieee80211_queue_work(&local->hw, &ifmgd->csa_connection_drop_work);
3589 }
3590 EXPORT_SYMBOL(ieee80211_channel_switch_disconnect);
3591
3592 static int ieee80211_set_after_csa_beacon(struct ieee80211_sub_if_data *sdata,
3593                                           u64 *changed)
3594 {
3595         int err;
3596
3597         switch (sdata->vif.type) {
3598         case NL80211_IFTYPE_AP:
3599                 if (!sdata->deflink.u.ap.next_beacon)
3600                         return -EINVAL;
3601
3602                 err = ieee80211_assign_beacon(sdata, &sdata->deflink,
3603                                               sdata->deflink.u.ap.next_beacon,
3604                                               NULL, NULL);
3605                 ieee80211_free_next_beacon(&sdata->deflink);
3606
3607                 if (err < 0)
3608                         return err;
3609                 *changed |= err;
3610                 break;
3611         case NL80211_IFTYPE_ADHOC:
3612                 err = ieee80211_ibss_finish_csa(sdata);
3613                 if (err < 0)
3614                         return err;
3615                 *changed |= err;
3616                 break;
3617 #ifdef CONFIG_MAC80211_MESH
3618         case NL80211_IFTYPE_MESH_POINT:
3619                 err = ieee80211_mesh_finish_csa(sdata);
3620                 if (err < 0)
3621                         return err;
3622                 *changed |= err;
3623                 break;
3624 #endif
3625         default:
3626                 WARN_ON(1);
3627                 return -EINVAL;
3628         }
3629
3630         return 0;
3631 }
3632
3633 static int __ieee80211_csa_finalize(struct ieee80211_sub_if_data *sdata)
3634 {
3635         struct ieee80211_local *local = sdata->local;
3636         u64 changed = 0;
3637         int err;
3638
3639         sdata_assert_lock(sdata);
3640         lockdep_assert_held(&local->mtx);
3641         lockdep_assert_held(&local->chanctx_mtx);
3642
3643         if (sdata->vif.bss_conf.eht_puncturing != sdata->vif.bss_conf.csa_punct_bitmap) {
3644                 sdata->vif.bss_conf.eht_puncturing =
3645                                         sdata->vif.bss_conf.csa_punct_bitmap;
3646                 changed |= BSS_CHANGED_EHT_PUNCTURING;
3647         }
3648
3649         /*
3650          * using reservation isn't immediate as it may be deferred until later
3651          * with multi-vif. once reservation is complete it will re-schedule the
3652          * work with no reserved_chanctx so verify chandef to check if it
3653          * completed successfully
3654          */
3655
3656         if (sdata->deflink.reserved_chanctx) {
3657                 /*
3658                  * with multi-vif csa driver may call ieee80211_csa_finish()
3659                  * many times while waiting for other interfaces to use their
3660                  * reservations
3661                  */
3662                 if (sdata->deflink.reserved_ready)
3663                         return 0;
3664
3665                 return ieee80211_link_use_reserved_context(&sdata->deflink);
3666         }
3667
3668         if (!cfg80211_chandef_identical(&sdata->vif.bss_conf.chandef,
3669                                         &sdata->deflink.csa_chandef))
3670                 return -EINVAL;
3671
3672         sdata->vif.bss_conf.csa_active = false;
3673
3674         err = ieee80211_set_after_csa_beacon(sdata, &changed);
3675         if (err)
3676                 return err;
3677
3678         ieee80211_link_info_change_notify(sdata, &sdata->deflink, changed);
3679
3680         if (sdata->deflink.csa_block_tx) {
3681                 ieee80211_wake_vif_queues(local, sdata,
3682                                           IEEE80211_QUEUE_STOP_REASON_CSA);
3683                 sdata->deflink.csa_block_tx = false;
3684         }
3685
3686         err = drv_post_channel_switch(sdata);
3687         if (err)
3688                 return err;
3689
3690         cfg80211_ch_switch_notify(sdata->dev, &sdata->deflink.csa_chandef, 0,
3691                                   sdata->vif.bss_conf.eht_puncturing);
3692
3693         return 0;
3694 }
3695
3696 static void ieee80211_csa_finalize(struct ieee80211_sub_if_data *sdata)
3697 {
3698         if (__ieee80211_csa_finalize(sdata)) {
3699                 sdata_info(sdata, "failed to finalize CSA, disconnecting\n");
3700                 cfg80211_stop_iface(sdata->local->hw.wiphy, &sdata->wdev,
3701                                     GFP_KERNEL);
3702         }
3703 }
3704
3705 void ieee80211_csa_finalize_work(struct work_struct *work)
3706 {
3707         struct ieee80211_sub_if_data *sdata =
3708                 container_of(work, struct ieee80211_sub_if_data,
3709                              deflink.csa_finalize_work);
3710         struct ieee80211_local *local = sdata->local;
3711
3712         sdata_lock(sdata);
3713         mutex_lock(&local->mtx);
3714         mutex_lock(&local->chanctx_mtx);
3715
3716         /* AP might have been stopped while waiting for the lock. */
3717         if (!sdata->vif.bss_conf.csa_active)
3718                 goto unlock;
3719
3720         if (!ieee80211_sdata_running(sdata))
3721                 goto unlock;
3722
3723         ieee80211_csa_finalize(sdata);
3724
3725 unlock:
3726         mutex_unlock(&local->chanctx_mtx);
3727         mutex_unlock(&local->mtx);
3728         sdata_unlock(sdata);
3729 }
3730
3731 static int ieee80211_set_csa_beacon(struct ieee80211_sub_if_data *sdata,
3732                                     struct cfg80211_csa_settings *params,
3733                                     u32 *changed)
3734 {
3735         struct ieee80211_csa_settings csa = {};
3736         int err;
3737
3738         switch (sdata->vif.type) {
3739         case NL80211_IFTYPE_AP:
3740                 sdata->deflink.u.ap.next_beacon =
3741                         cfg80211_beacon_dup(&params->beacon_after);
3742                 if (!sdata->deflink.u.ap.next_beacon)
3743                         return -ENOMEM;
3744
3745                 /*
3746                  * With a count of 0, we don't have to wait for any
3747                  * TBTT before switching, so complete the CSA
3748                  * immediately.  In theory, with a count == 1 we
3749                  * should delay the switch until just before the next
3750                  * TBTT, but that would complicate things so we switch
3751                  * immediately too.  If we would delay the switch
3752                  * until the next TBTT, we would have to set the probe
3753                  * response here.
3754                  *
3755                  * TODO: A channel switch with count <= 1 without
3756                  * sending a CSA action frame is kind of useless,
3757                  * because the clients won't know we're changing
3758                  * channels.  The action frame must be implemented
3759                  * either here or in the userspace.
3760                  */
3761                 if (params->count <= 1)
3762                         break;
3763
3764                 if ((params->n_counter_offsets_beacon >
3765                      IEEE80211_MAX_CNTDWN_COUNTERS_NUM) ||
3766                     (params->n_counter_offsets_presp >
3767                      IEEE80211_MAX_CNTDWN_COUNTERS_NUM)) {
3768                         ieee80211_free_next_beacon(&sdata->deflink);
3769                         return -EINVAL;
3770                 }
3771
3772                 csa.counter_offsets_beacon = params->counter_offsets_beacon;
3773                 csa.counter_offsets_presp = params->counter_offsets_presp;
3774                 csa.n_counter_offsets_beacon = params->n_counter_offsets_beacon;
3775                 csa.n_counter_offsets_presp = params->n_counter_offsets_presp;
3776                 csa.count = params->count;
3777
3778                 err = ieee80211_assign_beacon(sdata, &sdata->deflink,
3779                                               &params->beacon_csa, &csa,
3780                                               NULL);
3781                 if (err < 0) {
3782                         ieee80211_free_next_beacon(&sdata->deflink);
3783                         return err;
3784                 }
3785                 *changed |= err;
3786
3787                 break;
3788         case NL80211_IFTYPE_ADHOC:
3789                 if (!sdata->vif.cfg.ibss_joined)
3790                         return -EINVAL;
3791
3792                 if (params->chandef.width != sdata->u.ibss.chandef.width)
3793                         return -EINVAL;
3794
3795                 switch (params->chandef.width) {
3796                 case NL80211_CHAN_WIDTH_40:
3797                         if (cfg80211_get_chandef_type(&params->chandef) !=
3798                             cfg80211_get_chandef_type(&sdata->u.ibss.chandef))
3799                                 return -EINVAL;
3800                         break;
3801                 case NL80211_CHAN_WIDTH_5:
3802                 case NL80211_CHAN_WIDTH_10:
3803                 case NL80211_CHAN_WIDTH_20_NOHT:
3804                 case NL80211_CHAN_WIDTH_20:
3805                         break;
3806                 default:
3807                         return -EINVAL;
3808                 }
3809
3810                 /* changes into another band are not supported */
3811                 if (sdata->u.ibss.chandef.chan->band !=
3812                     params->chandef.chan->band)
3813                         return -EINVAL;
3814
3815                 /* see comments in the NL80211_IFTYPE_AP block */
3816                 if (params->count > 1) {
3817                         err = ieee80211_ibss_csa_beacon(sdata, params);
3818                         if (err < 0)
3819                                 return err;
3820                         *changed |= err;
3821                 }
3822
3823                 ieee80211_send_action_csa(sdata, params);
3824
3825                 break;
3826 #ifdef CONFIG_MAC80211_MESH
3827         case NL80211_IFTYPE_MESH_POINT: {
3828                 struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
3829
3830                 /* changes into another band are not supported */
3831                 if (sdata->vif.bss_conf.chandef.chan->band !=
3832                     params->chandef.chan->band)
3833                         return -EINVAL;
3834
3835                 if (ifmsh->csa_role == IEEE80211_MESH_CSA_ROLE_NONE) {
3836                         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_INIT;
3837                         if (!ifmsh->pre_value)
3838                                 ifmsh->pre_value = 1;
3839                         else
3840                                 ifmsh->pre_value++;
3841                 }
3842
3843                 /* see comments in the NL80211_IFTYPE_AP block */
3844                 if (params->count > 1) {
3845                         err = ieee80211_mesh_csa_beacon(sdata, params);
3846                         if (err < 0) {
3847                                 ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_NONE;
3848                                 return err;
3849                         }
3850                         *changed |= err;
3851                 }
3852
3853                 if (ifmsh->csa_role == IEEE80211_MESH_CSA_ROLE_INIT)
3854                         ieee80211_send_action_csa(sdata, params);
3855
3856                 break;
3857                 }
3858 #endif
3859         default:
3860                 return -EOPNOTSUPP;
3861         }
3862
3863         return 0;
3864 }
3865
3866 static void ieee80211_color_change_abort(struct ieee80211_sub_if_data  *sdata)
3867 {
3868         sdata->vif.bss_conf.color_change_active = false;
3869
3870         ieee80211_free_next_beacon(&sdata->deflink);
3871
3872         cfg80211_color_change_aborted_notify(sdata->dev);
3873 }
3874
3875 static int
3876 __ieee80211_channel_switch(struct wiphy *wiphy, struct net_device *dev,
3877                            struct cfg80211_csa_settings *params)
3878 {
3879         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3880         struct ieee80211_local *local = sdata->local;
3881         struct ieee80211_channel_switch ch_switch;
3882         struct ieee80211_chanctx_conf *conf;
3883         struct ieee80211_chanctx *chanctx;
3884         u32 changed = 0;
3885         int err;
3886
3887         sdata_assert_lock(sdata);
3888         lockdep_assert_held(&local->mtx);
3889
3890         if (!list_empty(&local->roc_list) || local->scanning)
3891                 return -EBUSY;
3892
3893         if (sdata->wdev.cac_started)
3894                 return -EBUSY;
3895
3896         if (cfg80211_chandef_identical(&params->chandef,
3897                                        &sdata->vif.bss_conf.chandef))
3898                 return -EINVAL;
3899
3900         /* don't allow another channel switch if one is already active. */
3901         if (sdata->vif.bss_conf.csa_active)
3902                 return -EBUSY;
3903
3904         mutex_lock(&local->chanctx_mtx);
3905         conf = rcu_dereference_protected(sdata->vif.bss_conf.chanctx_conf,
3906                                          lockdep_is_held(&local->chanctx_mtx));
3907         if (!conf) {
3908                 err = -EBUSY;
3909                 goto out;
3910         }
3911
3912         if (params->chandef.chan->freq_offset) {
3913                 /* this may work, but is untested */
3914                 err = -EOPNOTSUPP;
3915                 goto out;
3916         }
3917
3918         chanctx = container_of(conf, struct ieee80211_chanctx, conf);
3919
3920         ch_switch.timestamp = 0;
3921         ch_switch.device_timestamp = 0;
3922         ch_switch.block_tx = params->block_tx;
3923         ch_switch.chandef = params->chandef;
3924         ch_switch.count = params->count;
3925
3926         err = drv_pre_channel_switch(sdata, &ch_switch);
3927         if (err)
3928                 goto out;
3929
3930         err = ieee80211_link_reserve_chanctx(&sdata->deflink, &params->chandef,
3931                                              chanctx->mode,
3932                                              params->radar_required);
3933         if (err)
3934                 goto out;
3935
3936         /* if reservation is invalid then this will fail */
3937         err = ieee80211_check_combinations(sdata, NULL, chanctx->mode, 0);
3938         if (err) {
3939                 ieee80211_link_unreserve_chanctx(&sdata->deflink);
3940                 goto out;
3941         }
3942
3943         /* if there is a color change in progress, abort it */
3944         if (sdata->vif.bss_conf.color_change_active)
3945                 ieee80211_color_change_abort(sdata);
3946
3947         err = ieee80211_set_csa_beacon(sdata, params, &changed);
3948         if (err) {
3949                 ieee80211_link_unreserve_chanctx(&sdata->deflink);
3950                 goto out;
3951         }
3952
3953         if (params->punct_bitmap && !sdata->vif.bss_conf.eht_support)
3954                 goto out;
3955
3956         sdata->deflink.csa_chandef = params->chandef;
3957         sdata->deflink.csa_block_tx = params->block_tx;
3958         sdata->vif.bss_conf.csa_active = true;
3959         sdata->vif.bss_conf.csa_punct_bitmap = params->punct_bitmap;
3960
3961         if (sdata->deflink.csa_block_tx)
3962                 ieee80211_stop_vif_queues(local, sdata,
3963                                           IEEE80211_QUEUE_STOP_REASON_CSA);
3964
3965         cfg80211_ch_switch_started_notify(sdata->dev,
3966                                           &sdata->deflink.csa_chandef, 0,
3967                                           params->count, params->block_tx,
3968                                           sdata->vif.bss_conf.csa_punct_bitmap);
3969
3970         if (changed) {
3971                 ieee80211_link_info_change_notify(sdata, &sdata->deflink,
3972                                                   changed);
3973                 drv_channel_switch_beacon(sdata, &params->chandef);
3974         } else {
3975                 /* if the beacon didn't change, we can finalize immediately */
3976                 ieee80211_csa_finalize(sdata);
3977         }
3978
3979 out:
3980         mutex_unlock(&local->chanctx_mtx);
3981         return err;
3982 }
3983
3984 int ieee80211_channel_switch(struct wiphy *wiphy, struct net_device *dev,
3985                              struct cfg80211_csa_settings *params)
3986 {
3987         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
3988         struct ieee80211_local *local = sdata->local;
3989         int err;
3990
3991         mutex_lock(&local->mtx);
3992         err = __ieee80211_channel_switch(wiphy, dev, params);
3993         mutex_unlock(&local->mtx);
3994
3995         return err;
3996 }
3997
3998 u64 ieee80211_mgmt_tx_cookie(struct ieee80211_local *local)
3999 {
4000         lockdep_assert_held(&local->mtx);
4001
4002         local->roc_cookie_counter++;
4003
4004         /* wow, you wrapped 64 bits ... more likely a bug */
4005         if (WARN_ON(local->roc_cookie_counter == 0))
4006                 local->roc_cookie_counter++;
4007
4008         return local->roc_cookie_counter;
4009 }
4010
4011 int ieee80211_attach_ack_skb(struct ieee80211_local *local, struct sk_buff *skb,
4012                              u64 *cookie, gfp_t gfp)
4013 {
4014         unsigned long spin_flags;
4015         struct sk_buff *ack_skb;
4016         int id;
4017
4018         ack_skb = skb_copy(skb, gfp);
4019         if (!ack_skb)
4020                 return -ENOMEM;
4021
4022         spin_lock_irqsave(&local->ack_status_lock, spin_flags);
4023         id = idr_alloc(&local->ack_status_frames, ack_skb,
4024                        1, 0x2000, GFP_ATOMIC);
4025         spin_unlock_irqrestore(&local->ack_status_lock, spin_flags);
4026
4027         if (id < 0) {
4028                 kfree_skb(ack_skb);
4029                 return -ENOMEM;
4030         }
4031
4032         IEEE80211_SKB_CB(skb)->ack_frame_id = id;
4033
4034         *cookie = ieee80211_mgmt_tx_cookie(local);
4035         IEEE80211_SKB_CB(ack_skb)->ack.cookie = *cookie;
4036
4037         return 0;
4038 }
4039
4040 static void
4041 ieee80211_update_mgmt_frame_registrations(struct wiphy *wiphy,
4042                                           struct wireless_dev *wdev,
4043                                           struct mgmt_frame_regs *upd)
4044 {
4045         struct ieee80211_local *local = wiphy_priv(wiphy);
4046         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
4047         u32 preq_mask = BIT(IEEE80211_STYPE_PROBE_REQ >> 4);
4048         u32 action_mask = BIT(IEEE80211_STYPE_ACTION >> 4);
4049         bool global_change, intf_change;
4050
4051         global_change =
4052                 (local->probe_req_reg != !!(upd->global_stypes & preq_mask)) ||
4053                 (local->rx_mcast_action_reg !=
4054                  !!(upd->global_mcast_stypes & action_mask));
4055         local->probe_req_reg = upd->global_stypes & preq_mask;
4056         local->rx_mcast_action_reg = upd->global_mcast_stypes & action_mask;
4057
4058         intf_change = (sdata->vif.probe_req_reg !=
4059                        !!(upd->interface_stypes & preq_mask)) ||
4060                 (sdata->vif.rx_mcast_action_reg !=
4061                  !!(upd->interface_mcast_stypes & action_mask));
4062         sdata->vif.probe_req_reg = upd->interface_stypes & preq_mask;
4063         sdata->vif.rx_mcast_action_reg =
4064                 upd->interface_mcast_stypes & action_mask;
4065
4066         if (!local->open_count)
4067                 return;
4068
4069         if (intf_change && ieee80211_sdata_running(sdata))
4070                 drv_config_iface_filter(local, sdata,
4071                                         sdata->vif.probe_req_reg ?
4072                                                 FIF_PROBE_REQ : 0,
4073                                         FIF_PROBE_REQ);
4074
4075         if (global_change)
4076                 ieee80211_configure_filter(local);
4077 }
4078
4079 static int ieee80211_set_antenna(struct wiphy *wiphy, u32 tx_ant, u32 rx_ant)
4080 {
4081         struct ieee80211_local *local = wiphy_priv(wiphy);
4082
4083         if (local->started)
4084                 return -EOPNOTSUPP;
4085
4086         return drv_set_antenna(local, tx_ant, rx_ant);
4087 }
4088
4089 static int ieee80211_get_antenna(struct wiphy *wiphy, u32 *tx_ant, u32 *rx_ant)
4090 {
4091         struct ieee80211_local *local = wiphy_priv(wiphy);
4092
4093         return drv_get_antenna(local, tx_ant, rx_ant);
4094 }
4095
4096 static int ieee80211_set_rekey_data(struct wiphy *wiphy,
4097                                     struct net_device *dev,
4098                                     struct cfg80211_gtk_rekey_data *data)
4099 {
4100         struct ieee80211_local *local = wiphy_priv(wiphy);
4101         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4102
4103         if (!local->ops->set_rekey_data)
4104                 return -EOPNOTSUPP;
4105
4106         drv_set_rekey_data(local, sdata, data);
4107
4108         return 0;
4109 }
4110
4111 static int ieee80211_probe_client(struct wiphy *wiphy, struct net_device *dev,
4112                                   const u8 *peer, u64 *cookie)
4113 {
4114         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4115         struct ieee80211_local *local = sdata->local;
4116         struct ieee80211_qos_hdr *nullfunc;
4117         struct sk_buff *skb;
4118         int size = sizeof(*nullfunc);
4119         __le16 fc;
4120         bool qos;
4121         struct ieee80211_tx_info *info;
4122         struct sta_info *sta;
4123         struct ieee80211_chanctx_conf *chanctx_conf;
4124         enum nl80211_band band;
4125         int ret;
4126
4127         /* the lock is needed to assign the cookie later */
4128         mutex_lock(&local->mtx);
4129
4130         rcu_read_lock();
4131         chanctx_conf = rcu_dereference(sdata->vif.bss_conf.chanctx_conf);
4132         if (WARN_ON(!chanctx_conf)) {
4133                 ret = -EINVAL;
4134                 goto unlock;
4135         }
4136         band = chanctx_conf->def.chan->band;
4137         sta = sta_info_get_bss(sdata, peer);
4138         if (sta) {
4139                 qos = sta->sta.wme;
4140         } else {
4141                 ret = -ENOLINK;
4142                 goto unlock;
4143         }
4144
4145         if (qos) {
4146                 fc = cpu_to_le16(IEEE80211_FTYPE_DATA |
4147                                  IEEE80211_STYPE_QOS_NULLFUNC |
4148                                  IEEE80211_FCTL_FROMDS);
4149         } else {
4150                 size -= 2;
4151                 fc = cpu_to_le16(IEEE80211_FTYPE_DATA |
4152                                  IEEE80211_STYPE_NULLFUNC |
4153                                  IEEE80211_FCTL_FROMDS);
4154         }
4155
4156         skb = dev_alloc_skb(local->hw.extra_tx_headroom + size);
4157         if (!skb) {
4158                 ret = -ENOMEM;
4159                 goto unlock;
4160         }
4161
4162         skb->dev = dev;
4163
4164         skb_reserve(skb, local->hw.extra_tx_headroom);
4165
4166         nullfunc = skb_put(skb, size);
4167         nullfunc->frame_control = fc;
4168         nullfunc->duration_id = 0;
4169         memcpy(nullfunc->addr1, sta->sta.addr, ETH_ALEN);
4170         memcpy(nullfunc->addr2, sdata->vif.addr, ETH_ALEN);
4171         memcpy(nullfunc->addr3, sdata->vif.addr, ETH_ALEN);
4172         nullfunc->seq_ctrl = 0;
4173
4174         info = IEEE80211_SKB_CB(skb);
4175
4176         info->flags |= IEEE80211_TX_CTL_REQ_TX_STATUS |
4177                        IEEE80211_TX_INTFL_NL80211_FRAME_TX;
4178         info->band = band;
4179
4180         skb_set_queue_mapping(skb, IEEE80211_AC_VO);
4181         skb->priority = 7;
4182         if (qos)
4183                 nullfunc->qos_ctrl = cpu_to_le16(7);
4184
4185         ret = ieee80211_attach_ack_skb(local, skb, cookie, GFP_ATOMIC);
4186         if (ret) {
4187                 kfree_skb(skb);
4188                 goto unlock;
4189         }
4190
4191         local_bh_disable();
4192         ieee80211_xmit(sdata, sta, skb);
4193         local_bh_enable();
4194
4195         ret = 0;
4196 unlock:
4197         rcu_read_unlock();
4198         mutex_unlock(&local->mtx);
4199
4200         return ret;
4201 }
4202
4203 static int ieee80211_cfg_get_channel(struct wiphy *wiphy,
4204                                      struct wireless_dev *wdev,
4205                                      unsigned int link_id,
4206                                      struct cfg80211_chan_def *chandef)
4207 {
4208         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
4209         struct ieee80211_local *local = wiphy_priv(wiphy);
4210         struct ieee80211_chanctx_conf *chanctx_conf;
4211         struct ieee80211_link_data *link;
4212         int ret = -ENODATA;
4213
4214         rcu_read_lock();
4215         link = rcu_dereference(sdata->link[link_id]);
4216         if (!link) {
4217                 ret = -ENOLINK;
4218                 goto out;
4219         }
4220
4221         chanctx_conf = rcu_dereference(link->conf->chanctx_conf);
4222         if (chanctx_conf) {
4223                 *chandef = link->conf->chandef;
4224                 ret = 0;
4225         } else if (local->open_count > 0 &&
4226                    local->open_count == local->monitors &&
4227                    sdata->vif.type == NL80211_IFTYPE_MONITOR) {
4228                 if (local->use_chanctx)
4229                         *chandef = local->monitor_chandef;
4230                 else
4231                         *chandef = local->_oper_chandef;
4232                 ret = 0;
4233         }
4234 out:
4235         rcu_read_unlock();
4236
4237         return ret;
4238 }
4239
4240 #ifdef CONFIG_PM
4241 static void ieee80211_set_wakeup(struct wiphy *wiphy, bool enabled)
4242 {
4243         drv_set_wakeup(wiphy_priv(wiphy), enabled);
4244 }
4245 #endif
4246
4247 static int ieee80211_set_qos_map(struct wiphy *wiphy,
4248                                  struct net_device *dev,
4249                                  struct cfg80211_qos_map *qos_map)
4250 {
4251         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4252         struct mac80211_qos_map *new_qos_map, *old_qos_map;
4253
4254         if (qos_map) {
4255                 new_qos_map = kzalloc(sizeof(*new_qos_map), GFP_KERNEL);
4256                 if (!new_qos_map)
4257                         return -ENOMEM;
4258                 memcpy(&new_qos_map->qos_map, qos_map, sizeof(*qos_map));
4259         } else {
4260                 /* A NULL qos_map was passed to disable QoS mapping */
4261                 new_qos_map = NULL;
4262         }
4263
4264         old_qos_map = sdata_dereference(sdata->qos_map, sdata);
4265         rcu_assign_pointer(sdata->qos_map, new_qos_map);
4266         if (old_qos_map)
4267                 kfree_rcu(old_qos_map, rcu_head);
4268
4269         return 0;
4270 }
4271
4272 static int ieee80211_set_ap_chanwidth(struct wiphy *wiphy,
4273                                       struct net_device *dev,
4274                                       unsigned int link_id,
4275                                       struct cfg80211_chan_def *chandef)
4276 {
4277         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4278         struct ieee80211_link_data *link;
4279         int ret;
4280         u64 changed = 0;
4281
4282         link = sdata_dereference(sdata->link[link_id], sdata);
4283
4284         ret = ieee80211_link_change_bandwidth(link, chandef, &changed);
4285         if (ret == 0)
4286                 ieee80211_link_info_change_notify(sdata, link, changed);
4287
4288         return ret;
4289 }
4290
4291 static int ieee80211_add_tx_ts(struct wiphy *wiphy, struct net_device *dev,
4292                                u8 tsid, const u8 *peer, u8 up,
4293                                u16 admitted_time)
4294 {
4295         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4296         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4297         int ac = ieee802_1d_to_ac[up];
4298
4299         if (sdata->vif.type != NL80211_IFTYPE_STATION)
4300                 return -EOPNOTSUPP;
4301
4302         if (!(sdata->wmm_acm & BIT(up)))
4303                 return -EINVAL;
4304
4305         if (ifmgd->tx_tspec[ac].admitted_time)
4306                 return -EBUSY;
4307
4308         if (admitted_time) {
4309                 ifmgd->tx_tspec[ac].admitted_time = 32 * admitted_time;
4310                 ifmgd->tx_tspec[ac].tsid = tsid;
4311                 ifmgd->tx_tspec[ac].up = up;
4312         }
4313
4314         return 0;
4315 }
4316
4317 static int ieee80211_del_tx_ts(struct wiphy *wiphy, struct net_device *dev,
4318                                u8 tsid, const u8 *peer)
4319 {
4320         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4321         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4322         struct ieee80211_local *local = wiphy_priv(wiphy);
4323         int ac;
4324
4325         for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
4326                 struct ieee80211_sta_tx_tspec *tx_tspec = &ifmgd->tx_tspec[ac];
4327
4328                 /* skip unused entries */
4329                 if (!tx_tspec->admitted_time)
4330                         continue;
4331
4332                 if (tx_tspec->tsid != tsid)
4333                         continue;
4334
4335                 /* due to this new packets will be reassigned to non-ACM ACs */
4336                 tx_tspec->up = -1;
4337
4338                 /* Make sure that all packets have been sent to avoid to
4339                  * restore the QoS params on packets that are still on the
4340                  * queues.
4341                  */
4342                 synchronize_net();
4343                 ieee80211_flush_queues(local, sdata, false);
4344
4345                 /* restore the normal QoS parameters
4346                  * (unconditionally to avoid races)
4347                  */
4348                 tx_tspec->action = TX_TSPEC_ACTION_STOP_DOWNGRADE;
4349                 tx_tspec->downgraded = false;
4350                 ieee80211_sta_handle_tspec_ac_params(sdata);
4351
4352                 /* finally clear all the data */
4353                 memset(tx_tspec, 0, sizeof(*tx_tspec));
4354
4355                 return 0;
4356         }
4357
4358         return -ENOENT;
4359 }
4360
4361 void ieee80211_nan_func_terminated(struct ieee80211_vif *vif,
4362                                    u8 inst_id,
4363                                    enum nl80211_nan_func_term_reason reason,
4364                                    gfp_t gfp)
4365 {
4366         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
4367         struct cfg80211_nan_func *func;
4368         u64 cookie;
4369
4370         if (WARN_ON(vif->type != NL80211_IFTYPE_NAN))
4371                 return;
4372
4373         spin_lock_bh(&sdata->u.nan.func_lock);
4374
4375         func = idr_find(&sdata->u.nan.function_inst_ids, inst_id);
4376         if (WARN_ON(!func)) {
4377                 spin_unlock_bh(&sdata->u.nan.func_lock);
4378                 return;
4379         }
4380
4381         cookie = func->cookie;
4382         idr_remove(&sdata->u.nan.function_inst_ids, inst_id);
4383
4384         spin_unlock_bh(&sdata->u.nan.func_lock);
4385
4386         cfg80211_free_nan_func(func);
4387
4388         cfg80211_nan_func_terminated(ieee80211_vif_to_wdev(vif), inst_id,
4389                                      reason, cookie, gfp);
4390 }
4391 EXPORT_SYMBOL(ieee80211_nan_func_terminated);
4392
4393 void ieee80211_nan_func_match(struct ieee80211_vif *vif,
4394                               struct cfg80211_nan_match_params *match,
4395                               gfp_t gfp)
4396 {
4397         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
4398         struct cfg80211_nan_func *func;
4399
4400         if (WARN_ON(vif->type != NL80211_IFTYPE_NAN))
4401                 return;
4402
4403         spin_lock_bh(&sdata->u.nan.func_lock);
4404
4405         func = idr_find(&sdata->u.nan.function_inst_ids,  match->inst_id);
4406         if (WARN_ON(!func)) {
4407                 spin_unlock_bh(&sdata->u.nan.func_lock);
4408                 return;
4409         }
4410         match->cookie = func->cookie;
4411
4412         spin_unlock_bh(&sdata->u.nan.func_lock);
4413
4414         cfg80211_nan_match(ieee80211_vif_to_wdev(vif), match, gfp);
4415 }
4416 EXPORT_SYMBOL(ieee80211_nan_func_match);
4417
4418 static int ieee80211_set_multicast_to_unicast(struct wiphy *wiphy,
4419                                               struct net_device *dev,
4420                                               const bool enabled)
4421 {
4422         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4423
4424         sdata->u.ap.multicast_to_unicast = enabled;
4425
4426         return 0;
4427 }
4428
4429 void ieee80211_fill_txq_stats(struct cfg80211_txq_stats *txqstats,
4430                               struct txq_info *txqi)
4431 {
4432         if (!(txqstats->filled & BIT(NL80211_TXQ_STATS_BACKLOG_BYTES))) {
4433                 txqstats->filled |= BIT(NL80211_TXQ_STATS_BACKLOG_BYTES);
4434                 txqstats->backlog_bytes = txqi->tin.backlog_bytes;
4435         }
4436
4437         if (!(txqstats->filled & BIT(NL80211_TXQ_STATS_BACKLOG_PACKETS))) {
4438                 txqstats->filled |= BIT(NL80211_TXQ_STATS_BACKLOG_PACKETS);
4439                 txqstats->backlog_packets = txqi->tin.backlog_packets;
4440         }
4441
4442         if (!(txqstats->filled & BIT(NL80211_TXQ_STATS_FLOWS))) {
4443                 txqstats->filled |= BIT(NL80211_TXQ_STATS_FLOWS);
4444                 txqstats->flows = txqi->tin.flows;
4445         }
4446
4447         if (!(txqstats->filled & BIT(NL80211_TXQ_STATS_DROPS))) {
4448                 txqstats->filled |= BIT(NL80211_TXQ_STATS_DROPS);
4449                 txqstats->drops = txqi->cstats.drop_count;
4450         }
4451
4452         if (!(txqstats->filled & BIT(NL80211_TXQ_STATS_ECN_MARKS))) {
4453                 txqstats->filled |= BIT(NL80211_TXQ_STATS_ECN_MARKS);
4454                 txqstats->ecn_marks = txqi->cstats.ecn_mark;
4455         }
4456
4457         if (!(txqstats->filled & BIT(NL80211_TXQ_STATS_OVERLIMIT))) {
4458                 txqstats->filled |= BIT(NL80211_TXQ_STATS_OVERLIMIT);
4459                 txqstats->overlimit = txqi->tin.overlimit;
4460         }
4461
4462         if (!(txqstats->filled & BIT(NL80211_TXQ_STATS_COLLISIONS))) {
4463                 txqstats->filled |= BIT(NL80211_TXQ_STATS_COLLISIONS);
4464                 txqstats->collisions = txqi->tin.collisions;
4465         }
4466
4467         if (!(txqstats->filled & BIT(NL80211_TXQ_STATS_TX_BYTES))) {
4468                 txqstats->filled |= BIT(NL80211_TXQ_STATS_TX_BYTES);
4469                 txqstats->tx_bytes = txqi->tin.tx_bytes;
4470         }
4471
4472         if (!(txqstats->filled & BIT(NL80211_TXQ_STATS_TX_PACKETS))) {
4473                 txqstats->filled |= BIT(NL80211_TXQ_STATS_TX_PACKETS);
4474                 txqstats->tx_packets = txqi->tin.tx_packets;
4475         }
4476 }
4477
4478 static int ieee80211_get_txq_stats(struct wiphy *wiphy,
4479                                    struct wireless_dev *wdev,
4480                                    struct cfg80211_txq_stats *txqstats)
4481 {
4482         struct ieee80211_local *local = wiphy_priv(wiphy);
4483         struct ieee80211_sub_if_data *sdata;
4484         int ret = 0;
4485
4486         spin_lock_bh(&local->fq.lock);
4487         rcu_read_lock();
4488
4489         if (wdev) {
4490                 sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
4491                 if (!sdata->vif.txq) {
4492                         ret = 1;
4493                         goto out;
4494                 }
4495                 ieee80211_fill_txq_stats(txqstats, to_txq_info(sdata->vif.txq));
4496         } else {
4497                 /* phy stats */
4498                 txqstats->filled |= BIT(NL80211_TXQ_STATS_BACKLOG_PACKETS) |
4499                                     BIT(NL80211_TXQ_STATS_BACKLOG_BYTES) |
4500                                     BIT(NL80211_TXQ_STATS_OVERLIMIT) |
4501                                     BIT(NL80211_TXQ_STATS_OVERMEMORY) |
4502                                     BIT(NL80211_TXQ_STATS_COLLISIONS) |
4503                                     BIT(NL80211_TXQ_STATS_MAX_FLOWS);
4504                 txqstats->backlog_packets = local->fq.backlog;
4505                 txqstats->backlog_bytes = local->fq.memory_usage;
4506                 txqstats->overlimit = local->fq.overlimit;
4507                 txqstats->overmemory = local->fq.overmemory;
4508                 txqstats->collisions = local->fq.collisions;
4509                 txqstats->max_flows = local->fq.flows_cnt;
4510         }
4511
4512 out:
4513         rcu_read_unlock();
4514         spin_unlock_bh(&local->fq.lock);
4515
4516         return ret;
4517 }
4518
4519 static int
4520 ieee80211_get_ftm_responder_stats(struct wiphy *wiphy,
4521                                   struct net_device *dev,
4522                                   struct cfg80211_ftm_responder_stats *ftm_stats)
4523 {
4524         struct ieee80211_local *local = wiphy_priv(wiphy);
4525         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4526
4527         return drv_get_ftm_responder_stats(local, sdata, ftm_stats);
4528 }
4529
4530 static int
4531 ieee80211_start_pmsr(struct wiphy *wiphy, struct wireless_dev *dev,
4532                      struct cfg80211_pmsr_request *request)
4533 {
4534         struct ieee80211_local *local = wiphy_priv(wiphy);
4535         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(dev);
4536
4537         return drv_start_pmsr(local, sdata, request);
4538 }
4539
4540 static void
4541 ieee80211_abort_pmsr(struct wiphy *wiphy, struct wireless_dev *dev,
4542                      struct cfg80211_pmsr_request *request)
4543 {
4544         struct ieee80211_local *local = wiphy_priv(wiphy);
4545         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(dev);
4546
4547         return drv_abort_pmsr(local, sdata, request);
4548 }
4549
4550 static int ieee80211_set_tid_config(struct wiphy *wiphy,
4551                                     struct net_device *dev,
4552                                     struct cfg80211_tid_config *tid_conf)
4553 {
4554         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4555         struct sta_info *sta;
4556         int ret;
4557
4558         if (!sdata->local->ops->set_tid_config)
4559                 return -EOPNOTSUPP;
4560
4561         if (!tid_conf->peer)
4562                 return drv_set_tid_config(sdata->local, sdata, NULL, tid_conf);
4563
4564         mutex_lock(&sdata->local->sta_mtx);
4565         sta = sta_info_get_bss(sdata, tid_conf->peer);
4566         if (!sta) {
4567                 mutex_unlock(&sdata->local->sta_mtx);
4568                 return -ENOENT;
4569         }
4570
4571         ret = drv_set_tid_config(sdata->local, sdata, &sta->sta, tid_conf);
4572         mutex_unlock(&sdata->local->sta_mtx);
4573
4574         return ret;
4575 }
4576
4577 static int ieee80211_reset_tid_config(struct wiphy *wiphy,
4578                                       struct net_device *dev,
4579                                       const u8 *peer, u8 tids)
4580 {
4581         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4582         struct sta_info *sta;
4583         int ret;
4584
4585         if (!sdata->local->ops->reset_tid_config)
4586                 return -EOPNOTSUPP;
4587
4588         if (!peer)
4589                 return drv_reset_tid_config(sdata->local, sdata, NULL, tids);
4590
4591         mutex_lock(&sdata->local->sta_mtx);
4592         sta = sta_info_get_bss(sdata, peer);
4593         if (!sta) {
4594                 mutex_unlock(&sdata->local->sta_mtx);
4595                 return -ENOENT;
4596         }
4597
4598         ret = drv_reset_tid_config(sdata->local, sdata, &sta->sta, tids);
4599         mutex_unlock(&sdata->local->sta_mtx);
4600
4601         return ret;
4602 }
4603
4604 static int ieee80211_set_sar_specs(struct wiphy *wiphy,
4605                                    struct cfg80211_sar_specs *sar)
4606 {
4607         struct ieee80211_local *local = wiphy_priv(wiphy);
4608
4609         if (!local->ops->set_sar_specs)
4610                 return -EOPNOTSUPP;
4611
4612         return local->ops->set_sar_specs(&local->hw, sar);
4613 }
4614
4615 static int
4616 ieee80211_set_after_color_change_beacon(struct ieee80211_sub_if_data *sdata,
4617                                         u32 *changed)
4618 {
4619         switch (sdata->vif.type) {
4620         case NL80211_IFTYPE_AP: {
4621                 int ret;
4622
4623                 if (!sdata->deflink.u.ap.next_beacon)
4624                         return -EINVAL;
4625
4626                 ret = ieee80211_assign_beacon(sdata, &sdata->deflink,
4627                                               sdata->deflink.u.ap.next_beacon,
4628                                               NULL, NULL);
4629                 ieee80211_free_next_beacon(&sdata->deflink);
4630
4631                 if (ret < 0)
4632                         return ret;
4633
4634                 *changed |= ret;
4635                 break;
4636         }
4637         default:
4638                 WARN_ON_ONCE(1);
4639                 return -EINVAL;
4640         }
4641
4642         return 0;
4643 }
4644
4645 static int
4646 ieee80211_set_color_change_beacon(struct ieee80211_sub_if_data *sdata,
4647                                   struct cfg80211_color_change_settings *params,
4648                                   u32 *changed)
4649 {
4650         struct ieee80211_color_change_settings color_change = {};
4651         int err;
4652
4653         switch (sdata->vif.type) {
4654         case NL80211_IFTYPE_AP:
4655                 sdata->deflink.u.ap.next_beacon =
4656                         cfg80211_beacon_dup(&params->beacon_next);
4657                 if (!sdata->deflink.u.ap.next_beacon)
4658                         return -ENOMEM;
4659
4660                 if (params->count <= 1)
4661                         break;
4662
4663                 color_change.counter_offset_beacon =
4664                         params->counter_offset_beacon;
4665                 color_change.counter_offset_presp =
4666                         params->counter_offset_presp;
4667                 color_change.count = params->count;
4668
4669                 err = ieee80211_assign_beacon(sdata, &sdata->deflink,
4670                                               &params->beacon_color_change,
4671                                               NULL, &color_change);
4672                 if (err < 0) {
4673                         ieee80211_free_next_beacon(&sdata->deflink);
4674                         return err;
4675                 }
4676                 *changed |= err;
4677                 break;
4678         default:
4679                 return -EOPNOTSUPP;
4680         }
4681
4682         return 0;
4683 }
4684
4685 static void
4686 ieee80211_color_change_bss_config_notify(struct ieee80211_sub_if_data *sdata,
4687                                          u8 color, int enable, u32 changed)
4688 {
4689         sdata->vif.bss_conf.he_bss_color.color = color;
4690         sdata->vif.bss_conf.he_bss_color.enabled = enable;
4691         changed |= BSS_CHANGED_HE_BSS_COLOR;
4692
4693         ieee80211_link_info_change_notify(sdata, &sdata->deflink, changed);
4694
4695         if (!sdata->vif.bss_conf.nontransmitted && sdata->vif.mbssid_tx_vif) {
4696                 struct ieee80211_sub_if_data *child;
4697
4698                 mutex_lock(&sdata->local->iflist_mtx);
4699                 list_for_each_entry(child, &sdata->local->interfaces, list) {
4700                         if (child != sdata && child->vif.mbssid_tx_vif == &sdata->vif) {
4701                                 child->vif.bss_conf.he_bss_color.color = color;
4702                                 child->vif.bss_conf.he_bss_color.enabled = enable;
4703                                 ieee80211_link_info_change_notify(child,
4704                                                                   &child->deflink,
4705                                                                   BSS_CHANGED_HE_BSS_COLOR);
4706                         }
4707                 }
4708                 mutex_unlock(&sdata->local->iflist_mtx);
4709         }
4710 }
4711
4712 static int ieee80211_color_change_finalize(struct ieee80211_sub_if_data *sdata)
4713 {
4714         struct ieee80211_local *local = sdata->local;
4715         u32 changed = 0;
4716         int err;
4717
4718         sdata_assert_lock(sdata);
4719         lockdep_assert_held(&local->mtx);
4720
4721         sdata->vif.bss_conf.color_change_active = false;
4722
4723         err = ieee80211_set_after_color_change_beacon(sdata, &changed);
4724         if (err) {
4725                 cfg80211_color_change_aborted_notify(sdata->dev);
4726                 return err;
4727         }
4728
4729         ieee80211_color_change_bss_config_notify(sdata,
4730                                                  sdata->vif.bss_conf.color_change_color,
4731                                                  1, changed);
4732         cfg80211_color_change_notify(sdata->dev);
4733
4734         return 0;
4735 }
4736
4737 void ieee80211_color_change_finalize_work(struct work_struct *work)
4738 {
4739         struct ieee80211_sub_if_data *sdata =
4740                 container_of(work, struct ieee80211_sub_if_data,
4741                              deflink.color_change_finalize_work);
4742         struct ieee80211_local *local = sdata->local;
4743
4744         sdata_lock(sdata);
4745         mutex_lock(&local->mtx);
4746
4747         /* AP might have been stopped while waiting for the lock. */
4748         if (!sdata->vif.bss_conf.color_change_active)
4749                 goto unlock;
4750
4751         if (!ieee80211_sdata_running(sdata))
4752                 goto unlock;
4753
4754         ieee80211_color_change_finalize(sdata);
4755
4756 unlock:
4757         mutex_unlock(&local->mtx);
4758         sdata_unlock(sdata);
4759 }
4760
4761 void ieee80211_color_collision_detection_work(struct work_struct *work)
4762 {
4763         struct delayed_work *delayed_work = to_delayed_work(work);
4764         struct ieee80211_link_data *link =
4765                 container_of(delayed_work, struct ieee80211_link_data,
4766                              color_collision_detect_work);
4767         struct ieee80211_sub_if_data *sdata = link->sdata;
4768
4769         sdata_lock(sdata);
4770         cfg80211_obss_color_collision_notify(sdata->dev, link->color_bitmap);
4771         sdata_unlock(sdata);
4772 }
4773
4774 void ieee80211_color_change_finish(struct ieee80211_vif *vif)
4775 {
4776         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
4777
4778         ieee80211_queue_work(&sdata->local->hw,
4779                              &sdata->deflink.color_change_finalize_work);
4780 }
4781 EXPORT_SYMBOL_GPL(ieee80211_color_change_finish);
4782
4783 void
4784 ieee80211_obss_color_collision_notify(struct ieee80211_vif *vif,
4785                                        u64 color_bitmap, gfp_t gfp)
4786 {
4787         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
4788         struct ieee80211_link_data *link = &sdata->deflink;
4789
4790         if (sdata->vif.bss_conf.color_change_active || sdata->vif.bss_conf.csa_active)
4791                 return;
4792
4793         if (delayed_work_pending(&link->color_collision_detect_work))
4794                 return;
4795
4796         link->color_bitmap = color_bitmap;
4797         /* queue the color collision detection event every 500 ms in order to
4798          * avoid sending too much netlink messages to userspace.
4799          */
4800         ieee80211_queue_delayed_work(&sdata->local->hw,
4801                                      &link->color_collision_detect_work,
4802                                      msecs_to_jiffies(500));
4803 }
4804 EXPORT_SYMBOL_GPL(ieee80211_obss_color_collision_notify);
4805
4806 static int
4807 ieee80211_color_change(struct wiphy *wiphy, struct net_device *dev,
4808                        struct cfg80211_color_change_settings *params)
4809 {
4810         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4811         struct ieee80211_local *local = sdata->local;
4812         u32 changed = 0;
4813         int err;
4814
4815         sdata_assert_lock(sdata);
4816
4817         if (sdata->vif.bss_conf.nontransmitted)
4818                 return -EINVAL;
4819
4820         mutex_lock(&local->mtx);
4821
4822         /* don't allow another color change if one is already active or if csa
4823          * is active
4824          */
4825         if (sdata->vif.bss_conf.color_change_active || sdata->vif.bss_conf.csa_active) {
4826                 err = -EBUSY;
4827                 goto out;
4828         }
4829
4830         err = ieee80211_set_color_change_beacon(sdata, params, &changed);
4831         if (err)
4832                 goto out;
4833
4834         sdata->vif.bss_conf.color_change_active = true;
4835         sdata->vif.bss_conf.color_change_color = params->color;
4836
4837         cfg80211_color_change_started_notify(sdata->dev, params->count);
4838
4839         if (changed)
4840                 ieee80211_color_change_bss_config_notify(sdata, 0, 0, changed);
4841         else
4842                 /* if the beacon didn't change, we can finalize immediately */
4843                 ieee80211_color_change_finalize(sdata);
4844
4845 out:
4846         mutex_unlock(&local->mtx);
4847
4848         return err;
4849 }
4850
4851 static int
4852 ieee80211_set_radar_background(struct wiphy *wiphy,
4853                                struct cfg80211_chan_def *chandef)
4854 {
4855         struct ieee80211_local *local = wiphy_priv(wiphy);
4856
4857         if (!local->ops->set_radar_background)
4858                 return -EOPNOTSUPP;
4859
4860         return local->ops->set_radar_background(&local->hw, chandef);
4861 }
4862
4863 static int ieee80211_add_intf_link(struct wiphy *wiphy,
4864                                    struct wireless_dev *wdev,
4865                                    unsigned int link_id)
4866 {
4867         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
4868         int res;
4869
4870         if (wdev->use_4addr)
4871                 return -EOPNOTSUPP;
4872
4873         mutex_lock(&sdata->local->mtx);
4874         res = ieee80211_vif_set_links(sdata, wdev->valid_links);
4875         mutex_unlock(&sdata->local->mtx);
4876
4877         return res;
4878 }
4879
4880 static void ieee80211_del_intf_link(struct wiphy *wiphy,
4881                                     struct wireless_dev *wdev,
4882                                     unsigned int link_id)
4883 {
4884         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
4885
4886         mutex_lock(&sdata->local->mtx);
4887         ieee80211_vif_set_links(sdata, wdev->valid_links);
4888         mutex_unlock(&sdata->local->mtx);
4889 }
4890
4891 static int sta_add_link_station(struct ieee80211_local *local,
4892                                 struct ieee80211_sub_if_data *sdata,
4893                                 struct link_station_parameters *params)
4894 {
4895         struct sta_info *sta;
4896         int ret;
4897
4898         sta = sta_info_get_bss(sdata, params->mld_mac);
4899         if (!sta)
4900                 return -ENOENT;
4901
4902         if (!sta->sta.valid_links)
4903                 return -EINVAL;
4904
4905         if (sta->sta.valid_links & BIT(params->link_id))
4906                 return -EALREADY;
4907
4908         ret = ieee80211_sta_allocate_link(sta, params->link_id);
4909         if (ret)
4910                 return ret;
4911
4912         ret = sta_link_apply_parameters(local, sta, true, params);
4913         if (ret) {
4914                 ieee80211_sta_free_link(sta, params->link_id);
4915                 return ret;
4916         }
4917
4918         /* ieee80211_sta_activate_link frees the link upon failure */
4919         return ieee80211_sta_activate_link(sta, params->link_id);
4920 }
4921
4922 static int
4923 ieee80211_add_link_station(struct wiphy *wiphy, struct net_device *dev,
4924                            struct link_station_parameters *params)
4925 {
4926         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4927         struct ieee80211_local *local = wiphy_priv(wiphy);
4928         int ret;
4929
4930         mutex_lock(&sdata->local->sta_mtx);
4931         ret = sta_add_link_station(local, sdata, params);
4932         mutex_unlock(&sdata->local->sta_mtx);
4933
4934         return ret;
4935 }
4936
4937 static int sta_mod_link_station(struct ieee80211_local *local,
4938                                 struct ieee80211_sub_if_data *sdata,
4939                                 struct link_station_parameters *params)
4940 {
4941         struct sta_info *sta;
4942
4943         sta = sta_info_get_bss(sdata, params->mld_mac);
4944         if (!sta)
4945                 return -ENOENT;
4946
4947         if (!(sta->sta.valid_links & BIT(params->link_id)))
4948                 return -EINVAL;
4949
4950         return sta_link_apply_parameters(local, sta, false, params);
4951 }
4952
4953 static int
4954 ieee80211_mod_link_station(struct wiphy *wiphy, struct net_device *dev,
4955                            struct link_station_parameters *params)
4956 {
4957         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4958         struct ieee80211_local *local = wiphy_priv(wiphy);
4959         int ret;
4960
4961         mutex_lock(&sdata->local->sta_mtx);
4962         ret = sta_mod_link_station(local, sdata, params);
4963         mutex_unlock(&sdata->local->sta_mtx);
4964
4965         return ret;
4966 }
4967
4968 static int sta_del_link_station(struct ieee80211_sub_if_data *sdata,
4969                                 struct link_station_del_parameters *params)
4970 {
4971         struct sta_info *sta;
4972
4973         sta = sta_info_get_bss(sdata, params->mld_mac);
4974         if (!sta)
4975                 return -ENOENT;
4976
4977         if (!(sta->sta.valid_links & BIT(params->link_id)))
4978                 return -EINVAL;
4979
4980         /* must not create a STA without links */
4981         if (sta->sta.valid_links == BIT(params->link_id))
4982                 return -EINVAL;
4983
4984         ieee80211_sta_remove_link(sta, params->link_id);
4985
4986         return 0;
4987 }
4988
4989 static int
4990 ieee80211_del_link_station(struct wiphy *wiphy, struct net_device *dev,
4991                            struct link_station_del_parameters *params)
4992 {
4993         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
4994         int ret;
4995
4996         mutex_lock(&sdata->local->sta_mtx);
4997         ret = sta_del_link_station(sdata, params);
4998         mutex_unlock(&sdata->local->sta_mtx);
4999
5000         return ret;
5001 }
5002
5003 static int ieee80211_set_hw_timestamp(struct wiphy *wiphy,
5004                                       struct net_device *dev,
5005                                       struct cfg80211_set_hw_timestamp *hwts)
5006 {
5007         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
5008         struct ieee80211_local *local = sdata->local;
5009
5010         if (!local->ops->set_hw_timestamp)
5011                 return -EOPNOTSUPP;
5012
5013         if (!check_sdata_in_driver(sdata))
5014                 return -EIO;
5015
5016         return local->ops->set_hw_timestamp(&local->hw, &sdata->vif, hwts);
5017 }
5018
5019 const struct cfg80211_ops mac80211_config_ops = {
5020         .add_virtual_intf = ieee80211_add_iface,
5021         .del_virtual_intf = ieee80211_del_iface,
5022         .change_virtual_intf = ieee80211_change_iface,
5023         .start_p2p_device = ieee80211_start_p2p_device,
5024         .stop_p2p_device = ieee80211_stop_p2p_device,
5025         .add_key = ieee80211_add_key,
5026         .del_key = ieee80211_del_key,
5027         .get_key = ieee80211_get_key,
5028         .set_default_key = ieee80211_config_default_key,
5029         .set_default_mgmt_key = ieee80211_config_default_mgmt_key,
5030         .set_default_beacon_key = ieee80211_config_default_beacon_key,
5031         .start_ap = ieee80211_start_ap,
5032         .change_beacon = ieee80211_change_beacon,
5033         .stop_ap = ieee80211_stop_ap,
5034         .add_station = ieee80211_add_station,
5035         .del_station = ieee80211_del_station,
5036         .change_station = ieee80211_change_station,
5037         .get_station = ieee80211_get_station,
5038         .dump_station = ieee80211_dump_station,
5039         .dump_survey = ieee80211_dump_survey,
5040 #ifdef CONFIG_MAC80211_MESH
5041         .add_mpath = ieee80211_add_mpath,
5042         .del_mpath = ieee80211_del_mpath,
5043         .change_mpath = ieee80211_change_mpath,
5044         .get_mpath = ieee80211_get_mpath,
5045         .dump_mpath = ieee80211_dump_mpath,
5046         .get_mpp = ieee80211_get_mpp,
5047         .dump_mpp = ieee80211_dump_mpp,
5048         .update_mesh_config = ieee80211_update_mesh_config,
5049         .get_mesh_config = ieee80211_get_mesh_config,
5050         .join_mesh = ieee80211_join_mesh,
5051         .leave_mesh = ieee80211_leave_mesh,
5052 #endif
5053         .join_ocb = ieee80211_join_ocb,
5054         .leave_ocb = ieee80211_leave_ocb,
5055         .change_bss = ieee80211_change_bss,
5056         .set_txq_params = ieee80211_set_txq_params,
5057         .set_monitor_channel = ieee80211_set_monitor_channel,
5058         .suspend = ieee80211_suspend,
5059         .resume = ieee80211_resume,
5060         .scan = ieee80211_scan,
5061         .abort_scan = ieee80211_abort_scan,
5062         .sched_scan_start = ieee80211_sched_scan_start,
5063         .sched_scan_stop = ieee80211_sched_scan_stop,
5064         .auth = ieee80211_auth,
5065         .assoc = ieee80211_assoc,
5066         .deauth = ieee80211_deauth,
5067         .disassoc = ieee80211_disassoc,
5068         .join_ibss = ieee80211_join_ibss,
5069         .leave_ibss = ieee80211_leave_ibss,
5070         .set_mcast_rate = ieee80211_set_mcast_rate,
5071         .set_wiphy_params = ieee80211_set_wiphy_params,
5072         .set_tx_power = ieee80211_set_tx_power,
5073         .get_tx_power = ieee80211_get_tx_power,
5074         .rfkill_poll = ieee80211_rfkill_poll,
5075         CFG80211_TESTMODE_CMD(ieee80211_testmode_cmd)
5076         CFG80211_TESTMODE_DUMP(ieee80211_testmode_dump)
5077         .set_power_mgmt = ieee80211_set_power_mgmt,
5078         .set_bitrate_mask = ieee80211_set_bitrate_mask,
5079         .remain_on_channel = ieee80211_remain_on_channel,
5080         .cancel_remain_on_channel = ieee80211_cancel_remain_on_channel,
5081         .mgmt_tx = ieee80211_mgmt_tx,
5082         .mgmt_tx_cancel_wait = ieee80211_mgmt_tx_cancel_wait,
5083         .set_cqm_rssi_config = ieee80211_set_cqm_rssi_config,
5084         .set_cqm_rssi_range_config = ieee80211_set_cqm_rssi_range_config,
5085         .update_mgmt_frame_registrations =
5086                 ieee80211_update_mgmt_frame_registrations,
5087         .set_antenna = ieee80211_set_antenna,
5088         .get_antenna = ieee80211_get_antenna,
5089         .set_rekey_data = ieee80211_set_rekey_data,
5090         .tdls_oper = ieee80211_tdls_oper,
5091         .tdls_mgmt = ieee80211_tdls_mgmt,
5092         .tdls_channel_switch = ieee80211_tdls_channel_switch,
5093         .tdls_cancel_channel_switch = ieee80211_tdls_cancel_channel_switch,
5094         .probe_client = ieee80211_probe_client,
5095         .set_noack_map = ieee80211_set_noack_map,
5096 #ifdef CONFIG_PM
5097         .set_wakeup = ieee80211_set_wakeup,
5098 #endif
5099         .get_channel = ieee80211_cfg_get_channel,
5100         .start_radar_detection = ieee80211_start_radar_detection,
5101         .end_cac = ieee80211_end_cac,
5102         .channel_switch = ieee80211_channel_switch,
5103         .set_qos_map = ieee80211_set_qos_map,
5104         .set_ap_chanwidth = ieee80211_set_ap_chanwidth,
5105         .add_tx_ts = ieee80211_add_tx_ts,
5106         .del_tx_ts = ieee80211_del_tx_ts,
5107         .start_nan = ieee80211_start_nan,
5108         .stop_nan = ieee80211_stop_nan,
5109         .nan_change_conf = ieee80211_nan_change_conf,
5110         .add_nan_func = ieee80211_add_nan_func,
5111         .del_nan_func = ieee80211_del_nan_func,
5112         .set_multicast_to_unicast = ieee80211_set_multicast_to_unicast,
5113         .tx_control_port = ieee80211_tx_control_port,
5114         .get_txq_stats = ieee80211_get_txq_stats,
5115         .get_ftm_responder_stats = ieee80211_get_ftm_responder_stats,
5116         .start_pmsr = ieee80211_start_pmsr,
5117         .abort_pmsr = ieee80211_abort_pmsr,
5118         .probe_mesh_link = ieee80211_probe_mesh_link,
5119         .set_tid_config = ieee80211_set_tid_config,
5120         .reset_tid_config = ieee80211_reset_tid_config,
5121         .set_sar_specs = ieee80211_set_sar_specs,
5122         .color_change = ieee80211_color_change,
5123         .set_radar_background = ieee80211_set_radar_background,
5124         .add_intf_link = ieee80211_add_intf_link,
5125         .del_intf_link = ieee80211_del_intf_link,
5126         .add_link_station = ieee80211_add_link_station,
5127         .mod_link_station = ieee80211_mod_link_station,
5128         .del_link_station = ieee80211_del_link_station,
5129         .set_hw_timestamp = ieee80211_set_hw_timestamp,
5130 };