Bluetooth: Cancel the Sniff timer
[platform/kernel/linux-rpi.git] / net / mac80211 / mesh.c
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (c) 2008, 2009 open80211s Ltd.
4  * Copyright (C) 2018 - 2020 Intel Corporation
5  * Authors:    Luis Carlos Cobo <luisca@cozybit.com>
6  *             Javier Cardona <javier@cozybit.com>
7  */
8
9 #include <linux/slab.h>
10 #include <asm/unaligned.h>
11 #include "ieee80211_i.h"
12 #include "mesh.h"
13 #include "driver-ops.h"
14
15 static int mesh_allocated;
16 static struct kmem_cache *rm_cache;
17
18 bool mesh_action_is_path_sel(struct ieee80211_mgmt *mgmt)
19 {
20         return (mgmt->u.action.u.mesh_action.action_code ==
21                         WLAN_MESH_ACTION_HWMP_PATH_SELECTION);
22 }
23
24 void ieee80211s_init(void)
25 {
26         mesh_allocated = 1;
27         rm_cache = kmem_cache_create("mesh_rmc", sizeof(struct rmc_entry),
28                                      0, 0, NULL);
29 }
30
31 void ieee80211s_stop(void)
32 {
33         if (!mesh_allocated)
34                 return;
35         kmem_cache_destroy(rm_cache);
36 }
37
38 static void ieee80211_mesh_housekeeping_timer(struct timer_list *t)
39 {
40         struct ieee80211_sub_if_data *sdata =
41                 from_timer(sdata, t, u.mesh.housekeeping_timer);
42         struct ieee80211_local *local = sdata->local;
43         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
44
45         set_bit(MESH_WORK_HOUSEKEEPING, &ifmsh->wrkq_flags);
46
47         ieee80211_queue_work(&local->hw, &sdata->work);
48 }
49
50 /**
51  * mesh_matches_local - check if the config of a mesh point matches ours
52  *
53  * @sdata: local mesh subif
54  * @ie: information elements of a management frame from the mesh peer
55  *
56  * This function checks if the mesh configuration of a mesh point matches the
57  * local mesh configuration, i.e. if both nodes belong to the same mesh network.
58  */
59 bool mesh_matches_local(struct ieee80211_sub_if_data *sdata,
60                         struct ieee802_11_elems *ie)
61 {
62         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
63         u32 basic_rates = 0;
64         struct cfg80211_chan_def sta_chan_def;
65         struct ieee80211_supported_band *sband;
66         u32 vht_cap_info = 0;
67
68         /*
69          * As support for each feature is added, check for matching
70          * - On mesh config capabilities
71          *   - Power Save Support En
72          *   - Sync support enabled
73          *   - Sync support active
74          *   - Sync support required from peer
75          *   - MDA enabled
76          * - Power management control on fc
77          */
78         if (!(ifmsh->mesh_id_len == ie->mesh_id_len &&
79              memcmp(ifmsh->mesh_id, ie->mesh_id, ie->mesh_id_len) == 0 &&
80              (ifmsh->mesh_pp_id == ie->mesh_config->meshconf_psel) &&
81              (ifmsh->mesh_pm_id == ie->mesh_config->meshconf_pmetric) &&
82              (ifmsh->mesh_cc_id == ie->mesh_config->meshconf_congest) &&
83              (ifmsh->mesh_sp_id == ie->mesh_config->meshconf_synch) &&
84              (ifmsh->mesh_auth_id == ie->mesh_config->meshconf_auth)))
85                 return false;
86
87         sband = ieee80211_get_sband(sdata);
88         if (!sband)
89                 return false;
90
91         ieee80211_sta_get_rates(sdata, ie, sband->band,
92                                 &basic_rates);
93
94         if (sdata->vif.bss_conf.basic_rates != basic_rates)
95                 return false;
96
97         cfg80211_chandef_create(&sta_chan_def, sdata->vif.bss_conf.chandef.chan,
98                                 NL80211_CHAN_NO_HT);
99         ieee80211_chandef_ht_oper(ie->ht_operation, &sta_chan_def);
100
101         if (ie->vht_cap_elem)
102                 vht_cap_info = le32_to_cpu(ie->vht_cap_elem->vht_cap_info);
103
104         ieee80211_chandef_vht_oper(&sdata->local->hw, vht_cap_info,
105                                    ie->vht_operation, ie->ht_operation,
106                                    &sta_chan_def);
107         ieee80211_chandef_he_6ghz_oper(sdata, ie->he_operation, &sta_chan_def);
108
109         if (!cfg80211_chandef_compatible(&sdata->vif.bss_conf.chandef,
110                                          &sta_chan_def))
111                 return false;
112
113         return true;
114 }
115
116 /**
117  * mesh_peer_accepts_plinks - check if an mp is willing to establish peer links
118  *
119  * @ie: information elements of a management frame from the mesh peer
120  */
121 bool mesh_peer_accepts_plinks(struct ieee802_11_elems *ie)
122 {
123         return (ie->mesh_config->meshconf_cap &
124                         IEEE80211_MESHCONF_CAPAB_ACCEPT_PLINKS) != 0;
125 }
126
127 /**
128  * mesh_accept_plinks_update - update accepting_plink in local mesh beacons
129  *
130  * @sdata: mesh interface in which mesh beacons are going to be updated
131  *
132  * Returns: beacon changed flag if the beacon content changed.
133  */
134 u32 mesh_accept_plinks_update(struct ieee80211_sub_if_data *sdata)
135 {
136         bool free_plinks;
137         u32 changed = 0;
138
139         /* In case mesh_plink_free_count > 0 and mesh_plinktbl_capacity == 0,
140          * the mesh interface might be able to establish plinks with peers that
141          * are already on the table but are not on PLINK_ESTAB state. However,
142          * in general the mesh interface is not accepting peer link requests
143          * from new peers, and that must be reflected in the beacon
144          */
145         free_plinks = mesh_plink_availables(sdata);
146
147         if (free_plinks != sdata->u.mesh.accepting_plinks) {
148                 sdata->u.mesh.accepting_plinks = free_plinks;
149                 changed = BSS_CHANGED_BEACON;
150         }
151
152         return changed;
153 }
154
155 /*
156  * mesh_sta_cleanup - clean up any mesh sta state
157  *
158  * @sta: mesh sta to clean up.
159  */
160 void mesh_sta_cleanup(struct sta_info *sta)
161 {
162         struct ieee80211_sub_if_data *sdata = sta->sdata;
163         u32 changed = mesh_plink_deactivate(sta);
164
165         if (changed)
166                 ieee80211_mbss_info_change_notify(sdata, changed);
167 }
168
169 int mesh_rmc_init(struct ieee80211_sub_if_data *sdata)
170 {
171         int i;
172
173         sdata->u.mesh.rmc = kmalloc(sizeof(struct mesh_rmc), GFP_KERNEL);
174         if (!sdata->u.mesh.rmc)
175                 return -ENOMEM;
176         sdata->u.mesh.rmc->idx_mask = RMC_BUCKETS - 1;
177         for (i = 0; i < RMC_BUCKETS; i++)
178                 INIT_HLIST_HEAD(&sdata->u.mesh.rmc->bucket[i]);
179         return 0;
180 }
181
182 void mesh_rmc_free(struct ieee80211_sub_if_data *sdata)
183 {
184         struct mesh_rmc *rmc = sdata->u.mesh.rmc;
185         struct rmc_entry *p;
186         struct hlist_node *n;
187         int i;
188
189         if (!sdata->u.mesh.rmc)
190                 return;
191
192         for (i = 0; i < RMC_BUCKETS; i++) {
193                 hlist_for_each_entry_safe(p, n, &rmc->bucket[i], list) {
194                         hlist_del(&p->list);
195                         kmem_cache_free(rm_cache, p);
196                 }
197         }
198
199         kfree(rmc);
200         sdata->u.mesh.rmc = NULL;
201 }
202
203 /**
204  * mesh_rmc_check - Check frame in recent multicast cache and add if absent.
205  *
206  * @sdata:      interface
207  * @sa:         source address
208  * @mesh_hdr:   mesh_header
209  *
210  * Returns: 0 if the frame is not in the cache, nonzero otherwise.
211  *
212  * Checks using the source address and the mesh sequence number if we have
213  * received this frame lately. If the frame is not in the cache, it is added to
214  * it.
215  */
216 int mesh_rmc_check(struct ieee80211_sub_if_data *sdata,
217                    const u8 *sa, struct ieee80211s_hdr *mesh_hdr)
218 {
219         struct mesh_rmc *rmc = sdata->u.mesh.rmc;
220         u32 seqnum = 0;
221         int entries = 0;
222         u8 idx;
223         struct rmc_entry *p;
224         struct hlist_node *n;
225
226         if (!rmc)
227                 return -1;
228
229         /* Don't care about endianness since only match matters */
230         memcpy(&seqnum, &mesh_hdr->seqnum, sizeof(mesh_hdr->seqnum));
231         idx = le32_to_cpu(mesh_hdr->seqnum) & rmc->idx_mask;
232         hlist_for_each_entry_safe(p, n, &rmc->bucket[idx], list) {
233                 ++entries;
234                 if (time_after(jiffies, p->exp_time) ||
235                     entries == RMC_QUEUE_MAX_LEN) {
236                         hlist_del(&p->list);
237                         kmem_cache_free(rm_cache, p);
238                         --entries;
239                 } else if ((seqnum == p->seqnum) && ether_addr_equal(sa, p->sa))
240                         return -1;
241         }
242
243         p = kmem_cache_alloc(rm_cache, GFP_ATOMIC);
244         if (!p)
245                 return 0;
246
247         p->seqnum = seqnum;
248         p->exp_time = jiffies + RMC_TIMEOUT;
249         memcpy(p->sa, sa, ETH_ALEN);
250         hlist_add_head(&p->list, &rmc->bucket[idx]);
251         return 0;
252 }
253
254 int mesh_add_meshconf_ie(struct ieee80211_sub_if_data *sdata,
255                          struct sk_buff *skb)
256 {
257         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
258         u8 *pos, neighbors;
259         u8 meshconf_len = sizeof(struct ieee80211_meshconf_ie);
260         bool is_connected_to_gate = ifmsh->num_gates > 0 ||
261                 ifmsh->mshcfg.dot11MeshGateAnnouncementProtocol ||
262                 ifmsh->mshcfg.dot11MeshConnectedToMeshGate;
263         bool is_connected_to_as = ifmsh->mshcfg.dot11MeshConnectedToAuthServer;
264
265         if (skb_tailroom(skb) < 2 + meshconf_len)
266                 return -ENOMEM;
267
268         pos = skb_put(skb, 2 + meshconf_len);
269         *pos++ = WLAN_EID_MESH_CONFIG;
270         *pos++ = meshconf_len;
271
272         /* save a pointer for quick updates in pre-tbtt */
273         ifmsh->meshconf_offset = pos - skb->data;
274
275         /* Active path selection protocol ID */
276         *pos++ = ifmsh->mesh_pp_id;
277         /* Active path selection metric ID   */
278         *pos++ = ifmsh->mesh_pm_id;
279         /* Congestion control mode identifier */
280         *pos++ = ifmsh->mesh_cc_id;
281         /* Synchronization protocol identifier */
282         *pos++ = ifmsh->mesh_sp_id;
283         /* Authentication Protocol identifier */
284         *pos++ = ifmsh->mesh_auth_id;
285         /* Mesh Formation Info - number of neighbors */
286         neighbors = atomic_read(&ifmsh->estab_plinks);
287         neighbors = min_t(int, neighbors, IEEE80211_MAX_MESH_PEERINGS);
288         *pos++ = (is_connected_to_as << 7) |
289                  (neighbors << 1) |
290                  is_connected_to_gate;
291         /* Mesh capability */
292         *pos = 0x00;
293         *pos |= ifmsh->mshcfg.dot11MeshForwarding ?
294                         IEEE80211_MESHCONF_CAPAB_FORWARDING : 0x00;
295         *pos |= ifmsh->accepting_plinks ?
296                         IEEE80211_MESHCONF_CAPAB_ACCEPT_PLINKS : 0x00;
297         /* Mesh PS mode. See IEEE802.11-2012 8.4.2.100.8 */
298         *pos |= ifmsh->ps_peers_deep_sleep ?
299                         IEEE80211_MESHCONF_CAPAB_POWER_SAVE_LEVEL : 0x00;
300         return 0;
301 }
302
303 int mesh_add_meshid_ie(struct ieee80211_sub_if_data *sdata, struct sk_buff *skb)
304 {
305         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
306         u8 *pos;
307
308         if (skb_tailroom(skb) < 2 + ifmsh->mesh_id_len)
309                 return -ENOMEM;
310
311         pos = skb_put(skb, 2 + ifmsh->mesh_id_len);
312         *pos++ = WLAN_EID_MESH_ID;
313         *pos++ = ifmsh->mesh_id_len;
314         if (ifmsh->mesh_id_len)
315                 memcpy(pos, ifmsh->mesh_id, ifmsh->mesh_id_len);
316
317         return 0;
318 }
319
320 static int mesh_add_awake_window_ie(struct ieee80211_sub_if_data *sdata,
321                                     struct sk_buff *skb)
322 {
323         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
324         u8 *pos;
325
326         /* see IEEE802.11-2012 13.14.6 */
327         if (ifmsh->ps_peers_light_sleep == 0 &&
328             ifmsh->ps_peers_deep_sleep == 0 &&
329             ifmsh->nonpeer_pm == NL80211_MESH_POWER_ACTIVE)
330                 return 0;
331
332         if (skb_tailroom(skb) < 4)
333                 return -ENOMEM;
334
335         pos = skb_put(skb, 2 + 2);
336         *pos++ = WLAN_EID_MESH_AWAKE_WINDOW;
337         *pos++ = 2;
338         put_unaligned_le16(ifmsh->mshcfg.dot11MeshAwakeWindowDuration, pos);
339
340         return 0;
341 }
342
343 int mesh_add_vendor_ies(struct ieee80211_sub_if_data *sdata,
344                         struct sk_buff *skb)
345 {
346         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
347         u8 offset, len;
348         const u8 *data;
349
350         if (!ifmsh->ie || !ifmsh->ie_len)
351                 return 0;
352
353         /* fast-forward to vendor IEs */
354         offset = ieee80211_ie_split_vendor(ifmsh->ie, ifmsh->ie_len, 0);
355
356         if (offset < ifmsh->ie_len) {
357                 len = ifmsh->ie_len - offset;
358                 data = ifmsh->ie + offset;
359                 if (skb_tailroom(skb) < len)
360                         return -ENOMEM;
361                 skb_put_data(skb, data, len);
362         }
363
364         return 0;
365 }
366
367 int mesh_add_rsn_ie(struct ieee80211_sub_if_data *sdata, struct sk_buff *skb)
368 {
369         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
370         u8 len = 0;
371         const u8 *data;
372
373         if (!ifmsh->ie || !ifmsh->ie_len)
374                 return 0;
375
376         /* find RSN IE */
377         data = cfg80211_find_ie(WLAN_EID_RSN, ifmsh->ie, ifmsh->ie_len);
378         if (!data)
379                 return 0;
380
381         len = data[1] + 2;
382
383         if (skb_tailroom(skb) < len)
384                 return -ENOMEM;
385         skb_put_data(skb, data, len);
386
387         return 0;
388 }
389
390 static int mesh_add_ds_params_ie(struct ieee80211_sub_if_data *sdata,
391                                  struct sk_buff *skb)
392 {
393         struct ieee80211_chanctx_conf *chanctx_conf;
394         struct ieee80211_channel *chan;
395         u8 *pos;
396
397         if (skb_tailroom(skb) < 3)
398                 return -ENOMEM;
399
400         rcu_read_lock();
401         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
402         if (WARN_ON(!chanctx_conf)) {
403                 rcu_read_unlock();
404                 return -EINVAL;
405         }
406         chan = chanctx_conf->def.chan;
407         rcu_read_unlock();
408
409         pos = skb_put(skb, 2 + 1);
410         *pos++ = WLAN_EID_DS_PARAMS;
411         *pos++ = 1;
412         *pos++ = ieee80211_frequency_to_channel(chan->center_freq);
413
414         return 0;
415 }
416
417 int mesh_add_ht_cap_ie(struct ieee80211_sub_if_data *sdata,
418                        struct sk_buff *skb)
419 {
420         struct ieee80211_supported_band *sband;
421         u8 *pos;
422
423         sband = ieee80211_get_sband(sdata);
424         if (!sband)
425                 return -EINVAL;
426
427         /* HT not allowed in 6 GHz */
428         if (sband->band == NL80211_BAND_6GHZ)
429                 return 0;
430
431         if (!sband->ht_cap.ht_supported ||
432             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
433             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
434             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
435                 return 0;
436
437         if (skb_tailroom(skb) < 2 + sizeof(struct ieee80211_ht_cap))
438                 return -ENOMEM;
439
440         pos = skb_put(skb, 2 + sizeof(struct ieee80211_ht_cap));
441         ieee80211_ie_build_ht_cap(pos, &sband->ht_cap, sband->ht_cap.cap);
442
443         return 0;
444 }
445
446 int mesh_add_ht_oper_ie(struct ieee80211_sub_if_data *sdata,
447                         struct sk_buff *skb)
448 {
449         struct ieee80211_local *local = sdata->local;
450         struct ieee80211_chanctx_conf *chanctx_conf;
451         struct ieee80211_channel *channel;
452         struct ieee80211_supported_band *sband;
453         struct ieee80211_sta_ht_cap *ht_cap;
454         u8 *pos;
455
456         rcu_read_lock();
457         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
458         if (WARN_ON(!chanctx_conf)) {
459                 rcu_read_unlock();
460                 return -EINVAL;
461         }
462         channel = chanctx_conf->def.chan;
463         rcu_read_unlock();
464
465         sband = local->hw.wiphy->bands[channel->band];
466         ht_cap = &sband->ht_cap;
467
468         /* HT not allowed in 6 GHz */
469         if (sband->band == NL80211_BAND_6GHZ)
470                 return 0;
471
472         if (!ht_cap->ht_supported ||
473             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
474             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
475             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
476                 return 0;
477
478         if (skb_tailroom(skb) < 2 + sizeof(struct ieee80211_ht_operation))
479                 return -ENOMEM;
480
481         pos = skb_put(skb, 2 + sizeof(struct ieee80211_ht_operation));
482         ieee80211_ie_build_ht_oper(pos, ht_cap, &sdata->vif.bss_conf.chandef,
483                                    sdata->vif.bss_conf.ht_operation_mode,
484                                    false);
485
486         return 0;
487 }
488
489 int mesh_add_vht_cap_ie(struct ieee80211_sub_if_data *sdata,
490                         struct sk_buff *skb)
491 {
492         struct ieee80211_supported_band *sband;
493         u8 *pos;
494
495         sband = ieee80211_get_sband(sdata);
496         if (!sband)
497                 return -EINVAL;
498
499         /* VHT not allowed in 6 GHz */
500         if (sband->band == NL80211_BAND_6GHZ)
501                 return 0;
502
503         if (!sband->vht_cap.vht_supported ||
504             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
505             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
506             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
507                 return 0;
508
509         if (skb_tailroom(skb) < 2 + sizeof(struct ieee80211_vht_cap))
510                 return -ENOMEM;
511
512         pos = skb_put(skb, 2 + sizeof(struct ieee80211_vht_cap));
513         ieee80211_ie_build_vht_cap(pos, &sband->vht_cap, sband->vht_cap.cap);
514
515         return 0;
516 }
517
518 int mesh_add_vht_oper_ie(struct ieee80211_sub_if_data *sdata,
519                          struct sk_buff *skb)
520 {
521         struct ieee80211_local *local = sdata->local;
522         struct ieee80211_chanctx_conf *chanctx_conf;
523         struct ieee80211_channel *channel;
524         struct ieee80211_supported_band *sband;
525         struct ieee80211_sta_vht_cap *vht_cap;
526         u8 *pos;
527
528         rcu_read_lock();
529         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
530         if (WARN_ON(!chanctx_conf)) {
531                 rcu_read_unlock();
532                 return -EINVAL;
533         }
534         channel = chanctx_conf->def.chan;
535         rcu_read_unlock();
536
537         sband = local->hw.wiphy->bands[channel->band];
538         vht_cap = &sband->vht_cap;
539
540         /* VHT not allowed in 6 GHz */
541         if (sband->band == NL80211_BAND_6GHZ)
542                 return 0;
543
544         if (!vht_cap->vht_supported ||
545             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
546             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
547             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
548                 return 0;
549
550         if (skb_tailroom(skb) < 2 + sizeof(struct ieee80211_vht_operation))
551                 return -ENOMEM;
552
553         pos = skb_put(skb, 2 + sizeof(struct ieee80211_vht_operation));
554         ieee80211_ie_build_vht_oper(pos, vht_cap,
555                                     &sdata->vif.bss_conf.chandef);
556
557         return 0;
558 }
559
560 int mesh_add_he_cap_ie(struct ieee80211_sub_if_data *sdata,
561                        struct sk_buff *skb, u8 ie_len)
562 {
563         const struct ieee80211_sta_he_cap *he_cap;
564         struct ieee80211_supported_band *sband;
565         u8 *pos;
566
567         sband = ieee80211_get_sband(sdata);
568         if (!sband)
569                 return -EINVAL;
570
571         he_cap = ieee80211_get_he_iftype_cap(sband, NL80211_IFTYPE_MESH_POINT);
572
573         if (!he_cap ||
574             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
575             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
576             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
577                 return 0;
578
579         if (skb_tailroom(skb) < ie_len)
580                 return -ENOMEM;
581
582         pos = skb_put(skb, ie_len);
583         ieee80211_ie_build_he_cap(0, pos, he_cap, pos + ie_len);
584
585         return 0;
586 }
587
588 int mesh_add_he_oper_ie(struct ieee80211_sub_if_data *sdata,
589                         struct sk_buff *skb)
590 {
591         const struct ieee80211_sta_he_cap *he_cap;
592         struct ieee80211_supported_band *sband;
593         u32 len;
594         u8 *pos;
595
596         sband = ieee80211_get_sband(sdata);
597         if (!sband)
598                 return -EINVAL;
599
600         he_cap = ieee80211_get_he_iftype_cap(sband, NL80211_IFTYPE_MESH_POINT);
601         if (!he_cap ||
602             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
603             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
604             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
605                 return 0;
606
607         len = 2 + 1 + sizeof(struct ieee80211_he_operation);
608         if (sdata->vif.bss_conf.chandef.chan->band == NL80211_BAND_6GHZ)
609                 len += sizeof(struct ieee80211_he_6ghz_oper);
610
611         if (skb_tailroom(skb) < len)
612                 return -ENOMEM;
613
614         pos = skb_put(skb, len);
615         ieee80211_ie_build_he_oper(pos, &sdata->vif.bss_conf.chandef);
616
617         return 0;
618 }
619
620 int mesh_add_he_6ghz_cap_ie(struct ieee80211_sub_if_data *sdata,
621                             struct sk_buff *skb)
622 {
623         struct ieee80211_supported_band *sband;
624         const struct ieee80211_sband_iftype_data *iftd;
625
626         sband = ieee80211_get_sband(sdata);
627         if (!sband)
628                 return -EINVAL;
629
630         iftd = ieee80211_get_sband_iftype_data(sband,
631                                                NL80211_IFTYPE_MESH_POINT);
632         /* The device doesn't support HE in mesh mode or at all */
633         if (!iftd)
634                 return 0;
635
636         ieee80211_ie_build_he_6ghz_cap(sdata, skb);
637         return 0;
638 }
639
640 static void ieee80211_mesh_path_timer(struct timer_list *t)
641 {
642         struct ieee80211_sub_if_data *sdata =
643                 from_timer(sdata, t, u.mesh.mesh_path_timer);
644
645         ieee80211_queue_work(&sdata->local->hw, &sdata->work);
646 }
647
648 static void ieee80211_mesh_path_root_timer(struct timer_list *t)
649 {
650         struct ieee80211_sub_if_data *sdata =
651                 from_timer(sdata, t, u.mesh.mesh_path_root_timer);
652         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
653
654         set_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags);
655
656         ieee80211_queue_work(&sdata->local->hw, &sdata->work);
657 }
658
659 void ieee80211_mesh_root_setup(struct ieee80211_if_mesh *ifmsh)
660 {
661         if (ifmsh->mshcfg.dot11MeshHWMPRootMode > IEEE80211_ROOTMODE_ROOT)
662                 set_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags);
663         else {
664                 clear_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags);
665                 /* stop running timer */
666                 del_timer_sync(&ifmsh->mesh_path_root_timer);
667         }
668 }
669
670 static void
671 ieee80211_mesh_update_bss_params(struct ieee80211_sub_if_data *sdata,
672                                  u8 *ie, u8 ie_len)
673 {
674         struct ieee80211_supported_band *sband;
675         const struct element *cap;
676         const struct ieee80211_he_operation *he_oper = NULL;
677
678         sband = ieee80211_get_sband(sdata);
679         if (!sband)
680                 return;
681
682         if (!ieee80211_get_he_iftype_cap(sband, NL80211_IFTYPE_MESH_POINT) ||
683             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
684             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
685             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
686                 return;
687
688         sdata->vif.bss_conf.he_support = true;
689
690         cap = cfg80211_find_ext_elem(WLAN_EID_EXT_HE_OPERATION, ie, ie_len);
691         if (cap && cap->datalen >= 1 + sizeof(*he_oper) &&
692             cap->datalen >= 1 + ieee80211_he_oper_size(cap->data + 1))
693                 he_oper = (void *)(cap->data + 1);
694
695         if (he_oper)
696                 sdata->vif.bss_conf.he_oper.params =
697                         __le32_to_cpu(he_oper->he_oper_params);
698 }
699
700 /**
701  * ieee80211_fill_mesh_addresses - fill addresses of a locally originated mesh frame
702  * @hdr:        802.11 frame header
703  * @fc:         frame control field
704  * @meshda:     destination address in the mesh
705  * @meshsa:     source address in the mesh.  Same as TA, as frame is
706  *              locally originated.
707  *
708  * Return the length of the 802.11 (does not include a mesh control header)
709  */
710 int ieee80211_fill_mesh_addresses(struct ieee80211_hdr *hdr, __le16 *fc,
711                                   const u8 *meshda, const u8 *meshsa)
712 {
713         if (is_multicast_ether_addr(meshda)) {
714                 *fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS);
715                 /* DA TA SA */
716                 memcpy(hdr->addr1, meshda, ETH_ALEN);
717                 memcpy(hdr->addr2, meshsa, ETH_ALEN);
718                 memcpy(hdr->addr3, meshsa, ETH_ALEN);
719                 return 24;
720         } else {
721                 *fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS);
722                 /* RA TA DA SA */
723                 eth_zero_addr(hdr->addr1);   /* RA is resolved later */
724                 memcpy(hdr->addr2, meshsa, ETH_ALEN);
725                 memcpy(hdr->addr3, meshda, ETH_ALEN);
726                 memcpy(hdr->addr4, meshsa, ETH_ALEN);
727                 return 30;
728         }
729 }
730
731 /**
732  * ieee80211_new_mesh_header - create a new mesh header
733  * @sdata:      mesh interface to be used
734  * @meshhdr:    uninitialized mesh header
735  * @addr4or5:   1st address in the ae header, which may correspond to address 4
736  *              (if addr6 is NULL) or address 5 (if addr6 is present). It may
737  *              be NULL.
738  * @addr6:      2nd address in the ae header, which corresponds to addr6 of the
739  *              mesh frame
740  *
741  * Return the header length.
742  */
743 unsigned int ieee80211_new_mesh_header(struct ieee80211_sub_if_data *sdata,
744                                        struct ieee80211s_hdr *meshhdr,
745                                        const char *addr4or5, const char *addr6)
746 {
747         if (WARN_ON(!addr4or5 && addr6))
748                 return 0;
749
750         memset(meshhdr, 0, sizeof(*meshhdr));
751
752         meshhdr->ttl = sdata->u.mesh.mshcfg.dot11MeshTTL;
753
754         /* FIXME: racy -- TX on multiple queues can be concurrent */
755         put_unaligned(cpu_to_le32(sdata->u.mesh.mesh_seqnum), &meshhdr->seqnum);
756         sdata->u.mesh.mesh_seqnum++;
757
758         if (addr4or5 && !addr6) {
759                 meshhdr->flags |= MESH_FLAGS_AE_A4;
760                 memcpy(meshhdr->eaddr1, addr4or5, ETH_ALEN);
761                 return 2 * ETH_ALEN;
762         } else if (addr4or5 && addr6) {
763                 meshhdr->flags |= MESH_FLAGS_AE_A5_A6;
764                 memcpy(meshhdr->eaddr1, addr4or5, ETH_ALEN);
765                 memcpy(meshhdr->eaddr2, addr6, ETH_ALEN);
766                 return 3 * ETH_ALEN;
767         }
768
769         return ETH_ALEN;
770 }
771
772 static void ieee80211_mesh_housekeeping(struct ieee80211_sub_if_data *sdata)
773 {
774         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
775         u32 changed;
776
777         if (ifmsh->mshcfg.plink_timeout > 0)
778                 ieee80211_sta_expire(sdata, ifmsh->mshcfg.plink_timeout * HZ);
779         mesh_path_expire(sdata);
780
781         changed = mesh_accept_plinks_update(sdata);
782         ieee80211_mbss_info_change_notify(sdata, changed);
783
784         mod_timer(&ifmsh->housekeeping_timer,
785                   round_jiffies(jiffies +
786                                 IEEE80211_MESH_HOUSEKEEPING_INTERVAL));
787 }
788
789 static void ieee80211_mesh_rootpath(struct ieee80211_sub_if_data *sdata)
790 {
791         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
792         u32 interval;
793
794         mesh_path_tx_root_frame(sdata);
795
796         if (ifmsh->mshcfg.dot11MeshHWMPRootMode == IEEE80211_PROACTIVE_RANN)
797                 interval = ifmsh->mshcfg.dot11MeshHWMPRannInterval;
798         else
799                 interval = ifmsh->mshcfg.dot11MeshHWMProotInterval;
800
801         mod_timer(&ifmsh->mesh_path_root_timer,
802                   round_jiffies(TU_TO_EXP_TIME(interval)));
803 }
804
805 static int
806 ieee80211_mesh_build_beacon(struct ieee80211_if_mesh *ifmsh)
807 {
808         struct beacon_data *bcn;
809         int head_len, tail_len;
810         struct sk_buff *skb;
811         struct ieee80211_mgmt *mgmt;
812         struct ieee80211_chanctx_conf *chanctx_conf;
813         struct mesh_csa_settings *csa;
814         enum nl80211_band band;
815         u8 ie_len_he_cap;
816         u8 *pos;
817         struct ieee80211_sub_if_data *sdata;
818         int hdr_len = offsetofend(struct ieee80211_mgmt, u.beacon);
819
820         sdata = container_of(ifmsh, struct ieee80211_sub_if_data, u.mesh);
821         rcu_read_lock();
822         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
823         band = chanctx_conf->def.chan->band;
824         rcu_read_unlock();
825
826         ie_len_he_cap = ieee80211_ie_len_he_cap(sdata,
827                                                 NL80211_IFTYPE_MESH_POINT);
828         head_len = hdr_len +
829                    2 + /* NULL SSID */
830                    /* Channel Switch Announcement */
831                    2 + sizeof(struct ieee80211_channel_sw_ie) +
832                    /* Mesh Channel Switch Parameters */
833                    2 + sizeof(struct ieee80211_mesh_chansw_params_ie) +
834                    /* Channel Switch Wrapper + Wide Bandwidth CSA IE */
835                    2 + 2 + sizeof(struct ieee80211_wide_bw_chansw_ie) +
836                    2 + sizeof(struct ieee80211_sec_chan_offs_ie) +
837                    2 + 8 + /* supported rates */
838                    2 + 3; /* DS params */
839         tail_len = 2 + (IEEE80211_MAX_SUPP_RATES - 8) +
840                    2 + sizeof(struct ieee80211_ht_cap) +
841                    2 + sizeof(struct ieee80211_ht_operation) +
842                    2 + ifmsh->mesh_id_len +
843                    2 + sizeof(struct ieee80211_meshconf_ie) +
844                    2 + sizeof(__le16) + /* awake window */
845                    2 + sizeof(struct ieee80211_vht_cap) +
846                    2 + sizeof(struct ieee80211_vht_operation) +
847                    ie_len_he_cap +
848                    2 + 1 + sizeof(struct ieee80211_he_operation) +
849                            sizeof(struct ieee80211_he_6ghz_oper) +
850                    2 + 1 + sizeof(struct ieee80211_he_6ghz_capa) +
851                    ifmsh->ie_len;
852
853         bcn = kzalloc(sizeof(*bcn) + head_len + tail_len, GFP_KERNEL);
854         /* need an skb for IE builders to operate on */
855         skb = dev_alloc_skb(max(head_len, tail_len));
856
857         if (!bcn || !skb)
858                 goto out_free;
859
860         /*
861          * pointers go into the block we allocated,
862          * memory is | beacon_data | head | tail |
863          */
864         bcn->head = ((u8 *) bcn) + sizeof(*bcn);
865
866         /* fill in the head */
867         mgmt = skb_put_zero(skb, hdr_len);
868         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
869                                           IEEE80211_STYPE_BEACON);
870         eth_broadcast_addr(mgmt->da);
871         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
872         memcpy(mgmt->bssid, sdata->vif.addr, ETH_ALEN);
873         ieee80211_mps_set_frame_flags(sdata, NULL, (void *) mgmt);
874         mgmt->u.beacon.beacon_int =
875                 cpu_to_le16(sdata->vif.bss_conf.beacon_int);
876         mgmt->u.beacon.capab_info |= cpu_to_le16(
877                 sdata->u.mesh.security ? WLAN_CAPABILITY_PRIVACY : 0);
878
879         pos = skb_put(skb, 2);
880         *pos++ = WLAN_EID_SSID;
881         *pos++ = 0x0;
882
883         rcu_read_lock();
884         csa = rcu_dereference(ifmsh->csa);
885         if (csa) {
886                 enum nl80211_channel_type ct;
887                 struct cfg80211_chan_def *chandef;
888                 int ie_len = 2 + sizeof(struct ieee80211_channel_sw_ie) +
889                              2 + sizeof(struct ieee80211_mesh_chansw_params_ie);
890
891                 pos = skb_put_zero(skb, ie_len);
892                 *pos++ = WLAN_EID_CHANNEL_SWITCH;
893                 *pos++ = 3;
894                 *pos++ = 0x0;
895                 *pos++ = ieee80211_frequency_to_channel(
896                                 csa->settings.chandef.chan->center_freq);
897                 bcn->cntdwn_current_counter = csa->settings.count;
898                 bcn->cntdwn_counter_offsets[0] = hdr_len + 6;
899                 *pos++ = csa->settings.count;
900                 *pos++ = WLAN_EID_CHAN_SWITCH_PARAM;
901                 *pos++ = 6;
902                 if (ifmsh->csa_role == IEEE80211_MESH_CSA_ROLE_INIT) {
903                         *pos++ = ifmsh->mshcfg.dot11MeshTTL;
904                         *pos |= WLAN_EID_CHAN_SWITCH_PARAM_INITIATOR;
905                 } else {
906                         *pos++ = ifmsh->chsw_ttl;
907                 }
908                 *pos++ |= csa->settings.block_tx ?
909                           WLAN_EID_CHAN_SWITCH_PARAM_TX_RESTRICT : 0x00;
910                 put_unaligned_le16(WLAN_REASON_MESH_CHAN, pos);
911                 pos += 2;
912                 put_unaligned_le16(ifmsh->pre_value, pos);
913                 pos += 2;
914
915                 switch (csa->settings.chandef.width) {
916                 case NL80211_CHAN_WIDTH_40:
917                         ie_len = 2 + sizeof(struct ieee80211_sec_chan_offs_ie);
918                         pos = skb_put_zero(skb, ie_len);
919
920                         *pos++ = WLAN_EID_SECONDARY_CHANNEL_OFFSET; /* EID */
921                         *pos++ = 1;                                 /* len */
922                         ct = cfg80211_get_chandef_type(&csa->settings.chandef);
923                         if (ct == NL80211_CHAN_HT40PLUS)
924                                 *pos++ = IEEE80211_HT_PARAM_CHA_SEC_ABOVE;
925                         else
926                                 *pos++ = IEEE80211_HT_PARAM_CHA_SEC_BELOW;
927                         break;
928                 case NL80211_CHAN_WIDTH_80:
929                 case NL80211_CHAN_WIDTH_80P80:
930                 case NL80211_CHAN_WIDTH_160:
931                         /* Channel Switch Wrapper + Wide Bandwidth CSA IE */
932                         ie_len = 2 + 2 +
933                                  sizeof(struct ieee80211_wide_bw_chansw_ie);
934                         pos = skb_put_zero(skb, ie_len);
935
936                         *pos++ = WLAN_EID_CHANNEL_SWITCH_WRAPPER; /* EID */
937                         *pos++ = 5;                               /* len */
938                         /* put sub IE */
939                         chandef = &csa->settings.chandef;
940                         ieee80211_ie_build_wide_bw_cs(pos, chandef);
941                         break;
942                 default:
943                         break;
944                 }
945         }
946         rcu_read_unlock();
947
948         if (ieee80211_add_srates_ie(sdata, skb, true, band) ||
949             mesh_add_ds_params_ie(sdata, skb))
950                 goto out_free;
951
952         bcn->head_len = skb->len;
953         memcpy(bcn->head, skb->data, bcn->head_len);
954
955         /* now the tail */
956         skb_trim(skb, 0);
957         bcn->tail = bcn->head + bcn->head_len;
958
959         if (ieee80211_add_ext_srates_ie(sdata, skb, true, band) ||
960             mesh_add_rsn_ie(sdata, skb) ||
961             mesh_add_ht_cap_ie(sdata, skb) ||
962             mesh_add_ht_oper_ie(sdata, skb) ||
963             mesh_add_meshid_ie(sdata, skb) ||
964             mesh_add_meshconf_ie(sdata, skb) ||
965             mesh_add_awake_window_ie(sdata, skb) ||
966             mesh_add_vht_cap_ie(sdata, skb) ||
967             mesh_add_vht_oper_ie(sdata, skb) ||
968             mesh_add_he_cap_ie(sdata, skb, ie_len_he_cap) ||
969             mesh_add_he_oper_ie(sdata, skb) ||
970             mesh_add_he_6ghz_cap_ie(sdata, skb) ||
971             mesh_add_vendor_ies(sdata, skb))
972                 goto out_free;
973
974         bcn->tail_len = skb->len;
975         memcpy(bcn->tail, skb->data, bcn->tail_len);
976         ieee80211_mesh_update_bss_params(sdata, bcn->tail, bcn->tail_len);
977         bcn->meshconf = (struct ieee80211_meshconf_ie *)
978                                         (bcn->tail + ifmsh->meshconf_offset);
979
980         dev_kfree_skb(skb);
981         rcu_assign_pointer(ifmsh->beacon, bcn);
982         return 0;
983 out_free:
984         kfree(bcn);
985         dev_kfree_skb(skb);
986         return -ENOMEM;
987 }
988
989 static int
990 ieee80211_mesh_rebuild_beacon(struct ieee80211_sub_if_data *sdata)
991 {
992         struct beacon_data *old_bcn;
993         int ret;
994
995         old_bcn = rcu_dereference_protected(sdata->u.mesh.beacon,
996                                             lockdep_is_held(&sdata->wdev.mtx));
997         ret = ieee80211_mesh_build_beacon(&sdata->u.mesh);
998         if (ret)
999                 /* just reuse old beacon */
1000                 return ret;
1001
1002         if (old_bcn)
1003                 kfree_rcu(old_bcn, rcu_head);
1004         return 0;
1005 }
1006
1007 void ieee80211_mbss_info_change_notify(struct ieee80211_sub_if_data *sdata,
1008                                        u32 changed)
1009 {
1010         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1011         unsigned long bits = changed;
1012         u32 bit;
1013
1014         if (!bits)
1015                 return;
1016
1017         /* if we race with running work, worst case this work becomes a noop */
1018         for_each_set_bit(bit, &bits, sizeof(changed) * BITS_PER_BYTE)
1019                 set_bit(bit, &ifmsh->mbss_changed);
1020         set_bit(MESH_WORK_MBSS_CHANGED, &ifmsh->wrkq_flags);
1021         ieee80211_queue_work(&sdata->local->hw, &sdata->work);
1022 }
1023
1024 int ieee80211_start_mesh(struct ieee80211_sub_if_data *sdata)
1025 {
1026         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1027         struct ieee80211_local *local = sdata->local;
1028         u32 changed = BSS_CHANGED_BEACON |
1029                       BSS_CHANGED_BEACON_ENABLED |
1030                       BSS_CHANGED_HT |
1031                       BSS_CHANGED_BASIC_RATES |
1032                       BSS_CHANGED_BEACON_INT |
1033                       BSS_CHANGED_MCAST_RATE;
1034
1035         local->fif_other_bss++;
1036         /* mesh ifaces must set allmulti to forward mcast traffic */
1037         atomic_inc(&local->iff_allmultis);
1038         ieee80211_configure_filter(local);
1039
1040         ifmsh->mesh_cc_id = 0;  /* Disabled */
1041         /* register sync ops from extensible synchronization framework */
1042         ifmsh->sync_ops = ieee80211_mesh_sync_ops_get(ifmsh->mesh_sp_id);
1043         ifmsh->sync_offset_clockdrift_max = 0;
1044         set_bit(MESH_WORK_HOUSEKEEPING, &ifmsh->wrkq_flags);
1045         ieee80211_mesh_root_setup(ifmsh);
1046         ieee80211_queue_work(&local->hw, &sdata->work);
1047         sdata->vif.bss_conf.ht_operation_mode =
1048                                 ifmsh->mshcfg.ht_opmode;
1049         sdata->vif.bss_conf.enable_beacon = true;
1050
1051         changed |= ieee80211_mps_local_status_update(sdata);
1052
1053         if (ieee80211_mesh_build_beacon(ifmsh)) {
1054                 ieee80211_stop_mesh(sdata);
1055                 return -ENOMEM;
1056         }
1057
1058         ieee80211_recalc_dtim(local, sdata);
1059         ieee80211_bss_info_change_notify(sdata, changed);
1060
1061         netif_carrier_on(sdata->dev);
1062         return 0;
1063 }
1064
1065 void ieee80211_stop_mesh(struct ieee80211_sub_if_data *sdata)
1066 {
1067         struct ieee80211_local *local = sdata->local;
1068         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1069         struct beacon_data *bcn;
1070
1071         netif_carrier_off(sdata->dev);
1072
1073         /* flush STAs and mpaths on this iface */
1074         sta_info_flush(sdata);
1075         ieee80211_free_keys(sdata, true);
1076         mesh_path_flush_by_iface(sdata);
1077
1078         /* stop the beacon */
1079         ifmsh->mesh_id_len = 0;
1080         sdata->vif.bss_conf.enable_beacon = false;
1081         sdata->beacon_rate_set = false;
1082         clear_bit(SDATA_STATE_OFFCHANNEL_BEACON_STOPPED, &sdata->state);
1083         ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_BEACON_ENABLED);
1084
1085         /* remove beacon */
1086         bcn = rcu_dereference_protected(ifmsh->beacon,
1087                                         lockdep_is_held(&sdata->wdev.mtx));
1088         RCU_INIT_POINTER(ifmsh->beacon, NULL);
1089         kfree_rcu(bcn, rcu_head);
1090
1091         /* free all potentially still buffered group-addressed frames */
1092         local->total_ps_buffered -= skb_queue_len(&ifmsh->ps.bc_buf);
1093         skb_queue_purge(&ifmsh->ps.bc_buf);
1094
1095         del_timer_sync(&sdata->u.mesh.housekeeping_timer);
1096         del_timer_sync(&sdata->u.mesh.mesh_path_root_timer);
1097         del_timer_sync(&sdata->u.mesh.mesh_path_timer);
1098
1099         /* clear any mesh work (for next join) we may have accrued */
1100         ifmsh->wrkq_flags = 0;
1101         ifmsh->mbss_changed = 0;
1102
1103         local->fif_other_bss--;
1104         atomic_dec(&local->iff_allmultis);
1105         ieee80211_configure_filter(local);
1106 }
1107
1108 static void ieee80211_mesh_csa_mark_radar(struct ieee80211_sub_if_data *sdata)
1109 {
1110         int err;
1111
1112         /* if the current channel is a DFS channel, mark the channel as
1113          * unavailable.
1114          */
1115         err = cfg80211_chandef_dfs_required(sdata->local->hw.wiphy,
1116                                             &sdata->vif.bss_conf.chandef,
1117                                             NL80211_IFTYPE_MESH_POINT);
1118         if (err > 0)
1119                 cfg80211_radar_event(sdata->local->hw.wiphy,
1120                                      &sdata->vif.bss_conf.chandef, GFP_ATOMIC);
1121 }
1122
1123 static bool
1124 ieee80211_mesh_process_chnswitch(struct ieee80211_sub_if_data *sdata,
1125                                  struct ieee802_11_elems *elems, bool beacon)
1126 {
1127         struct cfg80211_csa_settings params;
1128         struct ieee80211_csa_ie csa_ie;
1129         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1130         struct ieee80211_supported_band *sband;
1131         int err;
1132         u32 sta_flags, vht_cap_info = 0;
1133
1134         sdata_assert_lock(sdata);
1135
1136         sband = ieee80211_get_sband(sdata);
1137         if (!sband)
1138                 return false;
1139
1140         sta_flags = 0;
1141         switch (sdata->vif.bss_conf.chandef.width) {
1142         case NL80211_CHAN_WIDTH_20_NOHT:
1143                 sta_flags |= IEEE80211_STA_DISABLE_HT;
1144                 fallthrough;
1145         case NL80211_CHAN_WIDTH_20:
1146                 sta_flags |= IEEE80211_STA_DISABLE_40MHZ;
1147                 fallthrough;
1148         case NL80211_CHAN_WIDTH_40:
1149                 sta_flags |= IEEE80211_STA_DISABLE_VHT;
1150                 break;
1151         default:
1152                 break;
1153         }
1154
1155         if (elems->vht_cap_elem)
1156                 vht_cap_info =
1157                         le32_to_cpu(elems->vht_cap_elem->vht_cap_info);
1158
1159         memset(&params, 0, sizeof(params));
1160         err = ieee80211_parse_ch_switch_ie(sdata, elems, sband->band,
1161                                            vht_cap_info,
1162                                            sta_flags, sdata->vif.addr,
1163                                            &csa_ie);
1164         if (err < 0)
1165                 return false;
1166         if (err)
1167                 return false;
1168
1169         /* Mark the channel unavailable if the reason for the switch is
1170          * regulatory.
1171          */
1172         if (csa_ie.reason_code == WLAN_REASON_MESH_CHAN_REGULATORY)
1173                 ieee80211_mesh_csa_mark_radar(sdata);
1174
1175         params.chandef = csa_ie.chandef;
1176         params.count = csa_ie.count;
1177
1178         if (!cfg80211_chandef_usable(sdata->local->hw.wiphy, &params.chandef,
1179                                      IEEE80211_CHAN_DISABLED) ||
1180             !cfg80211_reg_can_beacon(sdata->local->hw.wiphy, &params.chandef,
1181                                      NL80211_IFTYPE_MESH_POINT)) {
1182                 sdata_info(sdata,
1183                            "mesh STA %pM switches to unsupported channel (%d MHz, width:%d, CF1/2: %d/%d MHz), aborting\n",
1184                            sdata->vif.addr,
1185                            params.chandef.chan->center_freq,
1186                            params.chandef.width,
1187                            params.chandef.center_freq1,
1188                            params.chandef.center_freq2);
1189                 return false;
1190         }
1191
1192         err = cfg80211_chandef_dfs_required(sdata->local->hw.wiphy,
1193                                             &params.chandef,
1194                                             NL80211_IFTYPE_MESH_POINT);
1195         if (err < 0)
1196                 return false;
1197         if (err > 0 && !ifmsh->userspace_handles_dfs) {
1198                 sdata_info(sdata,
1199                            "mesh STA %pM switches to channel requiring DFS (%d MHz, width:%d, CF1/2: %d/%d MHz), aborting\n",
1200                            sdata->vif.addr,
1201                            params.chandef.chan->center_freq,
1202                            params.chandef.width,
1203                            params.chandef.center_freq1,
1204                            params.chandef.center_freq2);
1205                 return false;
1206         }
1207
1208         params.radar_required = err;
1209
1210         if (cfg80211_chandef_identical(&params.chandef,
1211                                        &sdata->vif.bss_conf.chandef)) {
1212                 mcsa_dbg(sdata,
1213                          "received csa with an identical chandef, ignoring\n");
1214                 return true;
1215         }
1216
1217         mcsa_dbg(sdata,
1218                  "received channel switch announcement to go to channel %d MHz\n",
1219                  params.chandef.chan->center_freq);
1220
1221         params.block_tx = csa_ie.mode & WLAN_EID_CHAN_SWITCH_PARAM_TX_RESTRICT;
1222         if (beacon) {
1223                 ifmsh->chsw_ttl = csa_ie.ttl - 1;
1224                 if (ifmsh->pre_value >= csa_ie.pre_value)
1225                         return false;
1226                 ifmsh->pre_value = csa_ie.pre_value;
1227         }
1228
1229         if (ifmsh->chsw_ttl >= ifmsh->mshcfg.dot11MeshTTL)
1230                 return false;
1231
1232         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_REPEATER;
1233
1234         if (ieee80211_channel_switch(sdata->local->hw.wiphy, sdata->dev,
1235                                      &params) < 0)
1236                 return false;
1237
1238         return true;
1239 }
1240
1241 static void
1242 ieee80211_mesh_rx_probe_req(struct ieee80211_sub_if_data *sdata,
1243                             struct ieee80211_mgmt *mgmt, size_t len)
1244 {
1245         struct ieee80211_local *local = sdata->local;
1246         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1247         struct sk_buff *presp;
1248         struct beacon_data *bcn;
1249         struct ieee80211_mgmt *hdr;
1250         struct ieee802_11_elems *elems;
1251         size_t baselen;
1252         u8 *pos;
1253
1254         pos = mgmt->u.probe_req.variable;
1255         baselen = (u8 *) pos - (u8 *) mgmt;
1256         if (baselen > len)
1257                 return;
1258
1259         elems = ieee802_11_parse_elems(pos, len - baselen, false, mgmt->bssid,
1260                                        NULL);
1261         if (!elems)
1262                 return;
1263
1264         if (!elems->mesh_id)
1265                 goto free;
1266
1267         /* 802.11-2012 10.1.4.3.2 */
1268         if ((!ether_addr_equal(mgmt->da, sdata->vif.addr) &&
1269              !is_broadcast_ether_addr(mgmt->da)) ||
1270             elems->ssid_len != 0)
1271                 goto free;
1272
1273         if (elems->mesh_id_len != 0 &&
1274             (elems->mesh_id_len != ifmsh->mesh_id_len ||
1275              memcmp(elems->mesh_id, ifmsh->mesh_id, ifmsh->mesh_id_len)))
1276                 goto free;
1277
1278         rcu_read_lock();
1279         bcn = rcu_dereference(ifmsh->beacon);
1280
1281         if (!bcn)
1282                 goto out;
1283
1284         presp = dev_alloc_skb(local->tx_headroom +
1285                               bcn->head_len + bcn->tail_len);
1286         if (!presp)
1287                 goto out;
1288
1289         skb_reserve(presp, local->tx_headroom);
1290         skb_put_data(presp, bcn->head, bcn->head_len);
1291         skb_put_data(presp, bcn->tail, bcn->tail_len);
1292         hdr = (struct ieee80211_mgmt *) presp->data;
1293         hdr->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
1294                                          IEEE80211_STYPE_PROBE_RESP);
1295         memcpy(hdr->da, mgmt->sa, ETH_ALEN);
1296         IEEE80211_SKB_CB(presp)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
1297         ieee80211_tx_skb(sdata, presp);
1298 out:
1299         rcu_read_unlock();
1300 free:
1301         kfree(elems);
1302 }
1303
1304 static void ieee80211_mesh_rx_bcn_presp(struct ieee80211_sub_if_data *sdata,
1305                                         u16 stype,
1306                                         struct ieee80211_mgmt *mgmt,
1307                                         size_t len,
1308                                         struct ieee80211_rx_status *rx_status)
1309 {
1310         struct ieee80211_local *local = sdata->local;
1311         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1312         struct ieee802_11_elems *elems;
1313         struct ieee80211_channel *channel;
1314         size_t baselen;
1315         int freq;
1316         enum nl80211_band band = rx_status->band;
1317
1318         /* ignore ProbeResp to foreign address */
1319         if (stype == IEEE80211_STYPE_PROBE_RESP &&
1320             !ether_addr_equal(mgmt->da, sdata->vif.addr))
1321                 return;
1322
1323         baselen = (u8 *) mgmt->u.probe_resp.variable - (u8 *) mgmt;
1324         if (baselen > len)
1325                 return;
1326
1327         elems = ieee802_11_parse_elems(mgmt->u.probe_resp.variable,
1328                                        len - baselen,
1329                                        false, mgmt->bssid, NULL);
1330         if (!elems)
1331                 return;
1332
1333         /* ignore non-mesh or secure / unsecure mismatch */
1334         if ((!elems->mesh_id || !elems->mesh_config) ||
1335             (elems->rsn && sdata->u.mesh.security == IEEE80211_MESH_SEC_NONE) ||
1336             (!elems->rsn && sdata->u.mesh.security != IEEE80211_MESH_SEC_NONE))
1337                 goto free;
1338
1339         if (elems->ds_params)
1340                 freq = ieee80211_channel_to_frequency(elems->ds_params[0], band);
1341         else
1342                 freq = rx_status->freq;
1343
1344         channel = ieee80211_get_channel(local->hw.wiphy, freq);
1345
1346         if (!channel || channel->flags & IEEE80211_CHAN_DISABLED)
1347                 goto free;
1348
1349         if (mesh_matches_local(sdata, elems)) {
1350                 mpl_dbg(sdata, "rssi_threshold=%d,rx_status->signal=%d\n",
1351                         sdata->u.mesh.mshcfg.rssi_threshold, rx_status->signal);
1352                 if (!sdata->u.mesh.user_mpm ||
1353                     sdata->u.mesh.mshcfg.rssi_threshold == 0 ||
1354                     sdata->u.mesh.mshcfg.rssi_threshold < rx_status->signal)
1355                         mesh_neighbour_update(sdata, mgmt->sa, elems,
1356                                               rx_status);
1357
1358                 if (ifmsh->csa_role != IEEE80211_MESH_CSA_ROLE_INIT &&
1359                     !sdata->vif.csa_active)
1360                         ieee80211_mesh_process_chnswitch(sdata, elems, true);
1361         }
1362
1363         if (ifmsh->sync_ops)
1364                 ifmsh->sync_ops->rx_bcn_presp(sdata, stype, mgmt, len,
1365                                               elems->mesh_config, rx_status);
1366 free:
1367         kfree(elems);
1368 }
1369
1370 int ieee80211_mesh_finish_csa(struct ieee80211_sub_if_data *sdata)
1371 {
1372         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1373         struct mesh_csa_settings *tmp_csa_settings;
1374         int ret = 0;
1375         int changed = 0;
1376
1377         /* Reset the TTL value and Initiator flag */
1378         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_NONE;
1379         ifmsh->chsw_ttl = 0;
1380
1381         /* Remove the CSA and MCSP elements from the beacon */
1382         tmp_csa_settings = rcu_dereference_protected(ifmsh->csa,
1383                                             lockdep_is_held(&sdata->wdev.mtx));
1384         RCU_INIT_POINTER(ifmsh->csa, NULL);
1385         if (tmp_csa_settings)
1386                 kfree_rcu(tmp_csa_settings, rcu_head);
1387         ret = ieee80211_mesh_rebuild_beacon(sdata);
1388         if (ret)
1389                 return -EINVAL;
1390
1391         changed |= BSS_CHANGED_BEACON;
1392
1393         mcsa_dbg(sdata, "complete switching to center freq %d MHz",
1394                  sdata->vif.bss_conf.chandef.chan->center_freq);
1395         return changed;
1396 }
1397
1398 int ieee80211_mesh_csa_beacon(struct ieee80211_sub_if_data *sdata,
1399                               struct cfg80211_csa_settings *csa_settings)
1400 {
1401         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1402         struct mesh_csa_settings *tmp_csa_settings;
1403         int ret = 0;
1404
1405         lockdep_assert_held(&sdata->wdev.mtx);
1406
1407         tmp_csa_settings = kmalloc(sizeof(*tmp_csa_settings),
1408                                    GFP_ATOMIC);
1409         if (!tmp_csa_settings)
1410                 return -ENOMEM;
1411
1412         memcpy(&tmp_csa_settings->settings, csa_settings,
1413                sizeof(struct cfg80211_csa_settings));
1414
1415         rcu_assign_pointer(ifmsh->csa, tmp_csa_settings);
1416
1417         ret = ieee80211_mesh_rebuild_beacon(sdata);
1418         if (ret) {
1419                 tmp_csa_settings = rcu_dereference(ifmsh->csa);
1420                 RCU_INIT_POINTER(ifmsh->csa, NULL);
1421                 kfree_rcu(tmp_csa_settings, rcu_head);
1422                 return ret;
1423         }
1424
1425         return BSS_CHANGED_BEACON;
1426 }
1427
1428 static int mesh_fwd_csa_frame(struct ieee80211_sub_if_data *sdata,
1429                                struct ieee80211_mgmt *mgmt, size_t len,
1430                                struct ieee802_11_elems *elems)
1431 {
1432         struct ieee80211_mgmt *mgmt_fwd;
1433         struct sk_buff *skb;
1434         struct ieee80211_local *local = sdata->local;
1435
1436         skb = dev_alloc_skb(local->tx_headroom + len);
1437         if (!skb)
1438                 return -ENOMEM;
1439         skb_reserve(skb, local->tx_headroom);
1440         mgmt_fwd = skb_put(skb, len);
1441
1442         elems->mesh_chansw_params_ie->mesh_ttl--;
1443         elems->mesh_chansw_params_ie->mesh_flags &=
1444                 ~WLAN_EID_CHAN_SWITCH_PARAM_INITIATOR;
1445
1446         memcpy(mgmt_fwd, mgmt, len);
1447         eth_broadcast_addr(mgmt_fwd->da);
1448         memcpy(mgmt_fwd->sa, sdata->vif.addr, ETH_ALEN);
1449         memcpy(mgmt_fwd->bssid, sdata->vif.addr, ETH_ALEN);
1450
1451         ieee80211_tx_skb(sdata, skb);
1452         return 0;
1453 }
1454
1455 static void mesh_rx_csa_frame(struct ieee80211_sub_if_data *sdata,
1456                               struct ieee80211_mgmt *mgmt, size_t len)
1457 {
1458         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1459         struct ieee802_11_elems *elems;
1460         u16 pre_value;
1461         bool fwd_csa = true;
1462         size_t baselen;
1463         u8 *pos;
1464
1465         if (mgmt->u.action.u.measurement.action_code !=
1466             WLAN_ACTION_SPCT_CHL_SWITCH)
1467                 return;
1468
1469         pos = mgmt->u.action.u.chan_switch.variable;
1470         baselen = offsetof(struct ieee80211_mgmt,
1471                            u.action.u.chan_switch.variable);
1472         elems = ieee802_11_parse_elems(pos, len - baselen, true,
1473                                        mgmt->bssid, NULL);
1474         if (!elems)
1475                 return;
1476
1477         if (!mesh_matches_local(sdata, elems))
1478                 goto free;
1479
1480         ifmsh->chsw_ttl = elems->mesh_chansw_params_ie->mesh_ttl;
1481         if (!--ifmsh->chsw_ttl)
1482                 fwd_csa = false;
1483
1484         pre_value = le16_to_cpu(elems->mesh_chansw_params_ie->mesh_pre_value);
1485         if (ifmsh->pre_value >= pre_value)
1486                 goto free;
1487
1488         ifmsh->pre_value = pre_value;
1489
1490         if (!sdata->vif.csa_active &&
1491             !ieee80211_mesh_process_chnswitch(sdata, elems, false)) {
1492                 mcsa_dbg(sdata, "Failed to process CSA action frame");
1493                 goto free;
1494         }
1495
1496         /* forward or re-broadcast the CSA frame */
1497         if (fwd_csa) {
1498                 if (mesh_fwd_csa_frame(sdata, mgmt, len, elems) < 0)
1499                         mcsa_dbg(sdata, "Failed to forward the CSA frame");
1500         }
1501 free:
1502         kfree(elems);
1503 }
1504
1505 static void ieee80211_mesh_rx_mgmt_action(struct ieee80211_sub_if_data *sdata,
1506                                           struct ieee80211_mgmt *mgmt,
1507                                           size_t len,
1508                                           struct ieee80211_rx_status *rx_status)
1509 {
1510         switch (mgmt->u.action.category) {
1511         case WLAN_CATEGORY_SELF_PROTECTED:
1512                 switch (mgmt->u.action.u.self_prot.action_code) {
1513                 case WLAN_SP_MESH_PEERING_OPEN:
1514                 case WLAN_SP_MESH_PEERING_CLOSE:
1515                 case WLAN_SP_MESH_PEERING_CONFIRM:
1516                         mesh_rx_plink_frame(sdata, mgmt, len, rx_status);
1517                         break;
1518                 }
1519                 break;
1520         case WLAN_CATEGORY_MESH_ACTION:
1521                 if (mesh_action_is_path_sel(mgmt))
1522                         mesh_rx_path_sel_frame(sdata, mgmt, len);
1523                 break;
1524         case WLAN_CATEGORY_SPECTRUM_MGMT:
1525                 mesh_rx_csa_frame(sdata, mgmt, len);
1526                 break;
1527         }
1528 }
1529
1530 void ieee80211_mesh_rx_queued_mgmt(struct ieee80211_sub_if_data *sdata,
1531                                    struct sk_buff *skb)
1532 {
1533         struct ieee80211_rx_status *rx_status;
1534         struct ieee80211_mgmt *mgmt;
1535         u16 stype;
1536
1537         sdata_lock(sdata);
1538
1539         /* mesh already went down */
1540         if (!sdata->u.mesh.mesh_id_len)
1541                 goto out;
1542
1543         rx_status = IEEE80211_SKB_RXCB(skb);
1544         mgmt = (struct ieee80211_mgmt *) skb->data;
1545         stype = le16_to_cpu(mgmt->frame_control) & IEEE80211_FCTL_STYPE;
1546
1547         switch (stype) {
1548         case IEEE80211_STYPE_PROBE_RESP:
1549         case IEEE80211_STYPE_BEACON:
1550                 ieee80211_mesh_rx_bcn_presp(sdata, stype, mgmt, skb->len,
1551                                             rx_status);
1552                 break;
1553         case IEEE80211_STYPE_PROBE_REQ:
1554                 ieee80211_mesh_rx_probe_req(sdata, mgmt, skb->len);
1555                 break;
1556         case IEEE80211_STYPE_ACTION:
1557                 ieee80211_mesh_rx_mgmt_action(sdata, mgmt, skb->len, rx_status);
1558                 break;
1559         }
1560 out:
1561         sdata_unlock(sdata);
1562 }
1563
1564 static void mesh_bss_info_changed(struct ieee80211_sub_if_data *sdata)
1565 {
1566         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1567         u32 bit, changed = 0;
1568
1569         for_each_set_bit(bit, &ifmsh->mbss_changed,
1570                          sizeof(changed) * BITS_PER_BYTE) {
1571                 clear_bit(bit, &ifmsh->mbss_changed);
1572                 changed |= BIT(bit);
1573         }
1574
1575         if (sdata->vif.bss_conf.enable_beacon &&
1576             (changed & (BSS_CHANGED_BEACON |
1577                         BSS_CHANGED_HT |
1578                         BSS_CHANGED_BASIC_RATES |
1579                         BSS_CHANGED_BEACON_INT)))
1580                 if (ieee80211_mesh_rebuild_beacon(sdata))
1581                         return;
1582
1583         ieee80211_bss_info_change_notify(sdata, changed);
1584 }
1585
1586 void ieee80211_mesh_work(struct ieee80211_sub_if_data *sdata)
1587 {
1588         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1589
1590         sdata_lock(sdata);
1591
1592         /* mesh already went down */
1593         if (!sdata->u.mesh.mesh_id_len)
1594                 goto out;
1595
1596         if (ifmsh->preq_queue_len &&
1597             time_after(jiffies,
1598                        ifmsh->last_preq + msecs_to_jiffies(ifmsh->mshcfg.dot11MeshHWMPpreqMinInterval)))
1599                 mesh_path_start_discovery(sdata);
1600
1601         if (test_and_clear_bit(MESH_WORK_HOUSEKEEPING, &ifmsh->wrkq_flags))
1602                 ieee80211_mesh_housekeeping(sdata);
1603
1604         if (test_and_clear_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags))
1605                 ieee80211_mesh_rootpath(sdata);
1606
1607         if (test_and_clear_bit(MESH_WORK_DRIFT_ADJUST, &ifmsh->wrkq_flags))
1608                 mesh_sync_adjust_tsf(sdata);
1609
1610         if (test_and_clear_bit(MESH_WORK_MBSS_CHANGED, &ifmsh->wrkq_flags))
1611                 mesh_bss_info_changed(sdata);
1612 out:
1613         sdata_unlock(sdata);
1614 }
1615
1616
1617 void ieee80211_mesh_init_sdata(struct ieee80211_sub_if_data *sdata)
1618 {
1619         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1620         static u8 zero_addr[ETH_ALEN] = {};
1621
1622         timer_setup(&ifmsh->housekeeping_timer,
1623                     ieee80211_mesh_housekeeping_timer, 0);
1624
1625         ifmsh->accepting_plinks = true;
1626         atomic_set(&ifmsh->mpaths, 0);
1627         mesh_rmc_init(sdata);
1628         ifmsh->last_preq = jiffies;
1629         ifmsh->next_perr = jiffies;
1630         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_NONE;
1631         /* Allocate all mesh structures when creating the first mesh interface. */
1632         if (!mesh_allocated)
1633                 ieee80211s_init();
1634
1635         mesh_pathtbl_init(sdata);
1636
1637         timer_setup(&ifmsh->mesh_path_timer, ieee80211_mesh_path_timer, 0);
1638         timer_setup(&ifmsh->mesh_path_root_timer,
1639                     ieee80211_mesh_path_root_timer, 0);
1640         INIT_LIST_HEAD(&ifmsh->preq_queue.list);
1641         skb_queue_head_init(&ifmsh->ps.bc_buf);
1642         spin_lock_init(&ifmsh->mesh_preq_queue_lock);
1643         spin_lock_init(&ifmsh->sync_offset_lock);
1644         RCU_INIT_POINTER(ifmsh->beacon, NULL);
1645
1646         sdata->vif.bss_conf.bssid = zero_addr;
1647 }
1648
1649 void ieee80211_mesh_teardown_sdata(struct ieee80211_sub_if_data *sdata)
1650 {
1651         mesh_rmc_free(sdata);
1652         mesh_pathtbl_unregister(sdata);
1653 }