Merge tag 'mm-stable-2022-08-03' of git://git.kernel.org/pub/scm/linux/kernel/git...
[linux-2.6-microblaze.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 - 2022 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, NULL,
108                                        &sta_chan_def);
109
110         if (!cfg80211_chandef_compatible(&sdata->vif.bss_conf.chandef,
111                                          &sta_chan_def))
112                 return false;
113
114         return true;
115 }
116
117 /**
118  * mesh_peer_accepts_plinks - check if an mp is willing to establish peer links
119  *
120  * @ie: information elements of a management frame from the mesh peer
121  */
122 bool mesh_peer_accepts_plinks(struct ieee802_11_elems *ie)
123 {
124         return (ie->mesh_config->meshconf_cap &
125                         IEEE80211_MESHCONF_CAPAB_ACCEPT_PLINKS) != 0;
126 }
127
128 /**
129  * mesh_accept_plinks_update - update accepting_plink in local mesh beacons
130  *
131  * @sdata: mesh interface in which mesh beacons are going to be updated
132  *
133  * Returns: beacon changed flag if the beacon content changed.
134  */
135 u32 mesh_accept_plinks_update(struct ieee80211_sub_if_data *sdata)
136 {
137         bool free_plinks;
138         u32 changed = 0;
139
140         /* In case mesh_plink_free_count > 0 and mesh_plinktbl_capacity == 0,
141          * the mesh interface might be able to establish plinks with peers that
142          * are already on the table but are not on PLINK_ESTAB state. However,
143          * in general the mesh interface is not accepting peer link requests
144          * from new peers, and that must be reflected in the beacon
145          */
146         free_plinks = mesh_plink_availables(sdata);
147
148         if (free_plinks != sdata->u.mesh.accepting_plinks) {
149                 sdata->u.mesh.accepting_plinks = free_plinks;
150                 changed = BSS_CHANGED_BEACON;
151         }
152
153         return changed;
154 }
155
156 /*
157  * mesh_sta_cleanup - clean up any mesh sta state
158  *
159  * @sta: mesh sta to clean up.
160  */
161 void mesh_sta_cleanup(struct sta_info *sta)
162 {
163         struct ieee80211_sub_if_data *sdata = sta->sdata;
164         u32 changed = mesh_plink_deactivate(sta);
165
166         if (changed)
167                 ieee80211_mbss_info_change_notify(sdata, changed);
168 }
169
170 int mesh_rmc_init(struct ieee80211_sub_if_data *sdata)
171 {
172         int i;
173
174         sdata->u.mesh.rmc = kmalloc(sizeof(struct mesh_rmc), GFP_KERNEL);
175         if (!sdata->u.mesh.rmc)
176                 return -ENOMEM;
177         sdata->u.mesh.rmc->idx_mask = RMC_BUCKETS - 1;
178         for (i = 0; i < RMC_BUCKETS; i++)
179                 INIT_HLIST_HEAD(&sdata->u.mesh.rmc->bucket[i]);
180         return 0;
181 }
182
183 void mesh_rmc_free(struct ieee80211_sub_if_data *sdata)
184 {
185         struct mesh_rmc *rmc = sdata->u.mesh.rmc;
186         struct rmc_entry *p;
187         struct hlist_node *n;
188         int i;
189
190         if (!sdata->u.mesh.rmc)
191                 return;
192
193         for (i = 0; i < RMC_BUCKETS; i++) {
194                 hlist_for_each_entry_safe(p, n, &rmc->bucket[i], list) {
195                         hlist_del(&p->list);
196                         kmem_cache_free(rm_cache, p);
197                 }
198         }
199
200         kfree(rmc);
201         sdata->u.mesh.rmc = NULL;
202 }
203
204 /**
205  * mesh_rmc_check - Check frame in recent multicast cache and add if absent.
206  *
207  * @sdata:      interface
208  * @sa:         source address
209  * @mesh_hdr:   mesh_header
210  *
211  * Returns: 0 if the frame is not in the cache, nonzero otherwise.
212  *
213  * Checks using the source address and the mesh sequence number if we have
214  * received this frame lately. If the frame is not in the cache, it is added to
215  * it.
216  */
217 int mesh_rmc_check(struct ieee80211_sub_if_data *sdata,
218                    const u8 *sa, struct ieee80211s_hdr *mesh_hdr)
219 {
220         struct mesh_rmc *rmc = sdata->u.mesh.rmc;
221         u32 seqnum = 0;
222         int entries = 0;
223         u8 idx;
224         struct rmc_entry *p;
225         struct hlist_node *n;
226
227         if (!rmc)
228                 return -1;
229
230         /* Don't care about endianness since only match matters */
231         memcpy(&seqnum, &mesh_hdr->seqnum, sizeof(mesh_hdr->seqnum));
232         idx = le32_to_cpu(mesh_hdr->seqnum) & rmc->idx_mask;
233         hlist_for_each_entry_safe(p, n, &rmc->bucket[idx], list) {
234                 ++entries;
235                 if (time_after(jiffies, p->exp_time) ||
236                     entries == RMC_QUEUE_MAX_LEN) {
237                         hlist_del(&p->list);
238                         kmem_cache_free(rm_cache, p);
239                         --entries;
240                 } else if ((seqnum == p->seqnum) && ether_addr_equal(sa, p->sa))
241                         return -1;
242         }
243
244         p = kmem_cache_alloc(rm_cache, GFP_ATOMIC);
245         if (!p)
246                 return 0;
247
248         p->seqnum = seqnum;
249         p->exp_time = jiffies + RMC_TIMEOUT;
250         memcpy(p->sa, sa, ETH_ALEN);
251         hlist_add_head(&p->list, &rmc->bucket[idx]);
252         return 0;
253 }
254
255 int mesh_add_meshconf_ie(struct ieee80211_sub_if_data *sdata,
256                          struct sk_buff *skb)
257 {
258         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
259         u8 *pos, neighbors;
260         u8 meshconf_len = sizeof(struct ieee80211_meshconf_ie);
261         bool is_connected_to_gate = ifmsh->num_gates > 0 ||
262                 ifmsh->mshcfg.dot11MeshGateAnnouncementProtocol ||
263                 ifmsh->mshcfg.dot11MeshConnectedToMeshGate;
264         bool is_connected_to_as = ifmsh->mshcfg.dot11MeshConnectedToAuthServer;
265
266         if (skb_tailroom(skb) < 2 + meshconf_len)
267                 return -ENOMEM;
268
269         pos = skb_put(skb, 2 + meshconf_len);
270         *pos++ = WLAN_EID_MESH_CONFIG;
271         *pos++ = meshconf_len;
272
273         /* save a pointer for quick updates in pre-tbtt */
274         ifmsh->meshconf_offset = pos - skb->data;
275
276         /* Active path selection protocol ID */
277         *pos++ = ifmsh->mesh_pp_id;
278         /* Active path selection metric ID   */
279         *pos++ = ifmsh->mesh_pm_id;
280         /* Congestion control mode identifier */
281         *pos++ = ifmsh->mesh_cc_id;
282         /* Synchronization protocol identifier */
283         *pos++ = ifmsh->mesh_sp_id;
284         /* Authentication Protocol identifier */
285         *pos++ = ifmsh->mesh_auth_id;
286         /* Mesh Formation Info - number of neighbors */
287         neighbors = atomic_read(&ifmsh->estab_plinks);
288         neighbors = min_t(int, neighbors, IEEE80211_MAX_MESH_PEERINGS);
289         *pos++ = (is_connected_to_as << 7) |
290                  (neighbors << 1) |
291                  is_connected_to_gate;
292         /* Mesh capability */
293         *pos = 0x00;
294         *pos |= ifmsh->mshcfg.dot11MeshForwarding ?
295                         IEEE80211_MESHCONF_CAPAB_FORWARDING : 0x00;
296         *pos |= ifmsh->accepting_plinks ?
297                         IEEE80211_MESHCONF_CAPAB_ACCEPT_PLINKS : 0x00;
298         /* Mesh PS mode. See IEEE802.11-2012 8.4.2.100.8 */
299         *pos |= ifmsh->ps_peers_deep_sleep ?
300                         IEEE80211_MESHCONF_CAPAB_POWER_SAVE_LEVEL : 0x00;
301         return 0;
302 }
303
304 int mesh_add_meshid_ie(struct ieee80211_sub_if_data *sdata, struct sk_buff *skb)
305 {
306         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
307         u8 *pos;
308
309         if (skb_tailroom(skb) < 2 + ifmsh->mesh_id_len)
310                 return -ENOMEM;
311
312         pos = skb_put(skb, 2 + ifmsh->mesh_id_len);
313         *pos++ = WLAN_EID_MESH_ID;
314         *pos++ = ifmsh->mesh_id_len;
315         if (ifmsh->mesh_id_len)
316                 memcpy(pos, ifmsh->mesh_id, ifmsh->mesh_id_len);
317
318         return 0;
319 }
320
321 static int mesh_add_awake_window_ie(struct ieee80211_sub_if_data *sdata,
322                                     struct sk_buff *skb)
323 {
324         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
325         u8 *pos;
326
327         /* see IEEE802.11-2012 13.14.6 */
328         if (ifmsh->ps_peers_light_sleep == 0 &&
329             ifmsh->ps_peers_deep_sleep == 0 &&
330             ifmsh->nonpeer_pm == NL80211_MESH_POWER_ACTIVE)
331                 return 0;
332
333         if (skb_tailroom(skb) < 4)
334                 return -ENOMEM;
335
336         pos = skb_put(skb, 2 + 2);
337         *pos++ = WLAN_EID_MESH_AWAKE_WINDOW;
338         *pos++ = 2;
339         put_unaligned_le16(ifmsh->mshcfg.dot11MeshAwakeWindowDuration, pos);
340
341         return 0;
342 }
343
344 int mesh_add_vendor_ies(struct ieee80211_sub_if_data *sdata,
345                         struct sk_buff *skb)
346 {
347         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
348         u8 offset, len;
349         const u8 *data;
350
351         if (!ifmsh->ie || !ifmsh->ie_len)
352                 return 0;
353
354         /* fast-forward to vendor IEs */
355         offset = ieee80211_ie_split_vendor(ifmsh->ie, ifmsh->ie_len, 0);
356
357         if (offset < ifmsh->ie_len) {
358                 len = ifmsh->ie_len - offset;
359                 data = ifmsh->ie + offset;
360                 if (skb_tailroom(skb) < len)
361                         return -ENOMEM;
362                 skb_put_data(skb, data, len);
363         }
364
365         return 0;
366 }
367
368 int mesh_add_rsn_ie(struct ieee80211_sub_if_data *sdata, struct sk_buff *skb)
369 {
370         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
371         u8 len = 0;
372         const u8 *data;
373
374         if (!ifmsh->ie || !ifmsh->ie_len)
375                 return 0;
376
377         /* find RSN IE */
378         data = cfg80211_find_ie(WLAN_EID_RSN, ifmsh->ie, ifmsh->ie_len);
379         if (!data)
380                 return 0;
381
382         len = data[1] + 2;
383
384         if (skb_tailroom(skb) < len)
385                 return -ENOMEM;
386         skb_put_data(skb, data, len);
387
388         return 0;
389 }
390
391 static int mesh_add_ds_params_ie(struct ieee80211_sub_if_data *sdata,
392                                  struct sk_buff *skb)
393 {
394         struct ieee80211_chanctx_conf *chanctx_conf;
395         struct ieee80211_channel *chan;
396         u8 *pos;
397
398         if (skb_tailroom(skb) < 3)
399                 return -ENOMEM;
400
401         rcu_read_lock();
402         chanctx_conf = rcu_dereference(sdata->vif.bss_conf.chanctx_conf);
403         if (WARN_ON(!chanctx_conf)) {
404                 rcu_read_unlock();
405                 return -EINVAL;
406         }
407         chan = chanctx_conf->def.chan;
408         rcu_read_unlock();
409
410         pos = skb_put(skb, 2 + 1);
411         *pos++ = WLAN_EID_DS_PARAMS;
412         *pos++ = 1;
413         *pos++ = ieee80211_frequency_to_channel(chan->center_freq);
414
415         return 0;
416 }
417
418 int mesh_add_ht_cap_ie(struct ieee80211_sub_if_data *sdata,
419                        struct sk_buff *skb)
420 {
421         struct ieee80211_supported_band *sband;
422         u8 *pos;
423
424         sband = ieee80211_get_sband(sdata);
425         if (!sband)
426                 return -EINVAL;
427
428         /* HT not allowed in 6 GHz */
429         if (sband->band == NL80211_BAND_6GHZ)
430                 return 0;
431
432         if (!sband->ht_cap.ht_supported ||
433             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
434             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
435             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
436                 return 0;
437
438         if (skb_tailroom(skb) < 2 + sizeof(struct ieee80211_ht_cap))
439                 return -ENOMEM;
440
441         pos = skb_put(skb, 2 + sizeof(struct ieee80211_ht_cap));
442         ieee80211_ie_build_ht_cap(pos, &sband->ht_cap, sband->ht_cap.cap);
443
444         return 0;
445 }
446
447 int mesh_add_ht_oper_ie(struct ieee80211_sub_if_data *sdata,
448                         struct sk_buff *skb)
449 {
450         struct ieee80211_local *local = sdata->local;
451         struct ieee80211_chanctx_conf *chanctx_conf;
452         struct ieee80211_channel *channel;
453         struct ieee80211_supported_band *sband;
454         struct ieee80211_sta_ht_cap *ht_cap;
455         u8 *pos;
456
457         rcu_read_lock();
458         chanctx_conf = rcu_dereference(sdata->vif.bss_conf.chanctx_conf);
459         if (WARN_ON(!chanctx_conf)) {
460                 rcu_read_unlock();
461                 return -EINVAL;
462         }
463         channel = chanctx_conf->def.chan;
464         rcu_read_unlock();
465
466         sband = local->hw.wiphy->bands[channel->band];
467         ht_cap = &sband->ht_cap;
468
469         /* HT not allowed in 6 GHz */
470         if (sband->band == NL80211_BAND_6GHZ)
471                 return 0;
472
473         if (!ht_cap->ht_supported ||
474             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
475             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
476             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
477                 return 0;
478
479         if (skb_tailroom(skb) < 2 + sizeof(struct ieee80211_ht_operation))
480                 return -ENOMEM;
481
482         pos = skb_put(skb, 2 + sizeof(struct ieee80211_ht_operation));
483         ieee80211_ie_build_ht_oper(pos, ht_cap, &sdata->vif.bss_conf.chandef,
484                                    sdata->vif.bss_conf.ht_operation_mode,
485                                    false);
486
487         return 0;
488 }
489
490 int mesh_add_vht_cap_ie(struct ieee80211_sub_if_data *sdata,
491                         struct sk_buff *skb)
492 {
493         struct ieee80211_supported_band *sband;
494         u8 *pos;
495
496         sband = ieee80211_get_sband(sdata);
497         if (!sband)
498                 return -EINVAL;
499
500         /* VHT not allowed in 6 GHz */
501         if (sband->band == NL80211_BAND_6GHZ)
502                 return 0;
503
504         if (!sband->vht_cap.vht_supported ||
505             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
506             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
507             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
508                 return 0;
509
510         if (skb_tailroom(skb) < 2 + sizeof(struct ieee80211_vht_cap))
511                 return -ENOMEM;
512
513         pos = skb_put(skb, 2 + sizeof(struct ieee80211_vht_cap));
514         ieee80211_ie_build_vht_cap(pos, &sband->vht_cap, sband->vht_cap.cap);
515
516         return 0;
517 }
518
519 int mesh_add_vht_oper_ie(struct ieee80211_sub_if_data *sdata,
520                          struct sk_buff *skb)
521 {
522         struct ieee80211_local *local = sdata->local;
523         struct ieee80211_chanctx_conf *chanctx_conf;
524         struct ieee80211_channel *channel;
525         struct ieee80211_supported_band *sband;
526         struct ieee80211_sta_vht_cap *vht_cap;
527         u8 *pos;
528
529         rcu_read_lock();
530         chanctx_conf = rcu_dereference(sdata->vif.bss_conf.chanctx_conf);
531         if (WARN_ON(!chanctx_conf)) {
532                 rcu_read_unlock();
533                 return -EINVAL;
534         }
535         channel = chanctx_conf->def.chan;
536         rcu_read_unlock();
537
538         sband = local->hw.wiphy->bands[channel->band];
539         vht_cap = &sband->vht_cap;
540
541         /* VHT not allowed in 6 GHz */
542         if (sband->band == NL80211_BAND_6GHZ)
543                 return 0;
544
545         if (!vht_cap->vht_supported ||
546             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
547             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
548             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
549                 return 0;
550
551         if (skb_tailroom(skb) < 2 + sizeof(struct ieee80211_vht_operation))
552                 return -ENOMEM;
553
554         pos = skb_put(skb, 2 + sizeof(struct ieee80211_vht_operation));
555         ieee80211_ie_build_vht_oper(pos, vht_cap,
556                                     &sdata->vif.bss_conf.chandef);
557
558         return 0;
559 }
560
561 int mesh_add_he_cap_ie(struct ieee80211_sub_if_data *sdata,
562                        struct sk_buff *skb, u8 ie_len)
563 {
564         const struct ieee80211_sta_he_cap *he_cap;
565         struct ieee80211_supported_band *sband;
566         u8 *pos;
567
568         sband = ieee80211_get_sband(sdata);
569         if (!sband)
570                 return -EINVAL;
571
572         he_cap = ieee80211_get_he_iftype_cap(sband, NL80211_IFTYPE_MESH_POINT);
573
574         if (!he_cap ||
575             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
576             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
577             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
578                 return 0;
579
580         if (skb_tailroom(skb) < ie_len)
581                 return -ENOMEM;
582
583         pos = skb_put(skb, ie_len);
584         ieee80211_ie_build_he_cap(0, pos, he_cap, pos + ie_len);
585
586         return 0;
587 }
588
589 int mesh_add_he_oper_ie(struct ieee80211_sub_if_data *sdata,
590                         struct sk_buff *skb)
591 {
592         const struct ieee80211_sta_he_cap *he_cap;
593         struct ieee80211_supported_band *sband;
594         u32 len;
595         u8 *pos;
596
597         sband = ieee80211_get_sband(sdata);
598         if (!sband)
599                 return -EINVAL;
600
601         he_cap = ieee80211_get_he_iftype_cap(sband, NL80211_IFTYPE_MESH_POINT);
602         if (!he_cap ||
603             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
604             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
605             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
606                 return 0;
607
608         len = 2 + 1 + sizeof(struct ieee80211_he_operation);
609         if (sdata->vif.bss_conf.chandef.chan->band == NL80211_BAND_6GHZ)
610                 len += sizeof(struct ieee80211_he_6ghz_oper);
611
612         if (skb_tailroom(skb) < len)
613                 return -ENOMEM;
614
615         pos = skb_put(skb, len);
616         ieee80211_ie_build_he_oper(pos, &sdata->vif.bss_conf.chandef);
617
618         return 0;
619 }
620
621 int mesh_add_he_6ghz_cap_ie(struct ieee80211_sub_if_data *sdata,
622                             struct sk_buff *skb)
623 {
624         struct ieee80211_supported_band *sband;
625         const struct ieee80211_sband_iftype_data *iftd;
626
627         sband = ieee80211_get_sband(sdata);
628         if (!sband)
629                 return -EINVAL;
630
631         iftd = ieee80211_get_sband_iftype_data(sband,
632                                                NL80211_IFTYPE_MESH_POINT);
633         /* The device doesn't support HE in mesh mode or at all */
634         if (!iftd)
635                 return 0;
636
637         ieee80211_ie_build_he_6ghz_cap(sdata, skb);
638         return 0;
639 }
640
641 static void ieee80211_mesh_path_timer(struct timer_list *t)
642 {
643         struct ieee80211_sub_if_data *sdata =
644                 from_timer(sdata, t, u.mesh.mesh_path_timer);
645
646         ieee80211_queue_work(&sdata->local->hw, &sdata->work);
647 }
648
649 static void ieee80211_mesh_path_root_timer(struct timer_list *t)
650 {
651         struct ieee80211_sub_if_data *sdata =
652                 from_timer(sdata, t, u.mesh.mesh_path_root_timer);
653         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
654
655         set_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags);
656
657         ieee80211_queue_work(&sdata->local->hw, &sdata->work);
658 }
659
660 void ieee80211_mesh_root_setup(struct ieee80211_if_mesh *ifmsh)
661 {
662         if (ifmsh->mshcfg.dot11MeshHWMPRootMode > IEEE80211_ROOTMODE_ROOT)
663                 set_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags);
664         else {
665                 clear_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags);
666                 /* stop running timer */
667                 del_timer_sync(&ifmsh->mesh_path_root_timer);
668         }
669 }
670
671 static void
672 ieee80211_mesh_update_bss_params(struct ieee80211_sub_if_data *sdata,
673                                  u8 *ie, u8 ie_len)
674 {
675         struct ieee80211_supported_band *sband;
676         const struct element *cap;
677         const struct ieee80211_he_operation *he_oper = NULL;
678
679         sband = ieee80211_get_sband(sdata);
680         if (!sband)
681                 return;
682
683         if (!ieee80211_get_he_iftype_cap(sband, NL80211_IFTYPE_MESH_POINT) ||
684             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT ||
685             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_5 ||
686             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_10)
687                 return;
688
689         sdata->vif.bss_conf.he_support = true;
690
691         cap = cfg80211_find_ext_elem(WLAN_EID_EXT_HE_OPERATION, ie, ie_len);
692         if (cap && cap->datalen >= 1 + sizeof(*he_oper) &&
693             cap->datalen >= 1 + ieee80211_he_oper_size(cap->data + 1))
694                 he_oper = (void *)(cap->data + 1);
695
696         if (he_oper)
697                 sdata->vif.bss_conf.he_oper.params =
698                         __le32_to_cpu(he_oper->he_oper_params);
699 }
700
701 /**
702  * ieee80211_fill_mesh_addresses - fill addresses of a locally originated mesh frame
703  * @hdr:        802.11 frame header
704  * @fc:         frame control field
705  * @meshda:     destination address in the mesh
706  * @meshsa:     source address in the mesh.  Same as TA, as frame is
707  *              locally originated.
708  *
709  * Return the length of the 802.11 (does not include a mesh control header)
710  */
711 int ieee80211_fill_mesh_addresses(struct ieee80211_hdr *hdr, __le16 *fc,
712                                   const u8 *meshda, const u8 *meshsa)
713 {
714         if (is_multicast_ether_addr(meshda)) {
715                 *fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS);
716                 /* DA TA SA */
717                 memcpy(hdr->addr1, meshda, ETH_ALEN);
718                 memcpy(hdr->addr2, meshsa, ETH_ALEN);
719                 memcpy(hdr->addr3, meshsa, ETH_ALEN);
720                 return 24;
721         } else {
722                 *fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS);
723                 /* RA TA DA SA */
724                 eth_zero_addr(hdr->addr1);   /* RA is resolved later */
725                 memcpy(hdr->addr2, meshsa, ETH_ALEN);
726                 memcpy(hdr->addr3, meshda, ETH_ALEN);
727                 memcpy(hdr->addr4, meshsa, ETH_ALEN);
728                 return 30;
729         }
730 }
731
732 /**
733  * ieee80211_new_mesh_header - create a new mesh header
734  * @sdata:      mesh interface to be used
735  * @meshhdr:    uninitialized mesh header
736  * @addr4or5:   1st address in the ae header, which may correspond to address 4
737  *              (if addr6 is NULL) or address 5 (if addr6 is present). It may
738  *              be NULL.
739  * @addr6:      2nd address in the ae header, which corresponds to addr6 of the
740  *              mesh frame
741  *
742  * Return the header length.
743  */
744 unsigned int ieee80211_new_mesh_header(struct ieee80211_sub_if_data *sdata,
745                                        struct ieee80211s_hdr *meshhdr,
746                                        const char *addr4or5, const char *addr6)
747 {
748         if (WARN_ON(!addr4or5 && addr6))
749                 return 0;
750
751         memset(meshhdr, 0, sizeof(*meshhdr));
752
753         meshhdr->ttl = sdata->u.mesh.mshcfg.dot11MeshTTL;
754
755         /* FIXME: racy -- TX on multiple queues can be concurrent */
756         put_unaligned(cpu_to_le32(sdata->u.mesh.mesh_seqnum), &meshhdr->seqnum);
757         sdata->u.mesh.mesh_seqnum++;
758
759         if (addr4or5 && !addr6) {
760                 meshhdr->flags |= MESH_FLAGS_AE_A4;
761                 memcpy(meshhdr->eaddr1, addr4or5, ETH_ALEN);
762                 return 2 * ETH_ALEN;
763         } else if (addr4or5 && addr6) {
764                 meshhdr->flags |= MESH_FLAGS_AE_A5_A6;
765                 memcpy(meshhdr->eaddr1, addr4or5, ETH_ALEN);
766                 memcpy(meshhdr->eaddr2, addr6, ETH_ALEN);
767                 return 3 * ETH_ALEN;
768         }
769
770         return ETH_ALEN;
771 }
772
773 static void ieee80211_mesh_housekeeping(struct ieee80211_sub_if_data *sdata)
774 {
775         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
776         u32 changed;
777
778         if (ifmsh->mshcfg.plink_timeout > 0)
779                 ieee80211_sta_expire(sdata, ifmsh->mshcfg.plink_timeout * HZ);
780         mesh_path_expire(sdata);
781
782         changed = mesh_accept_plinks_update(sdata);
783         ieee80211_mbss_info_change_notify(sdata, changed);
784
785         mod_timer(&ifmsh->housekeeping_timer,
786                   round_jiffies(jiffies +
787                                 IEEE80211_MESH_HOUSEKEEPING_INTERVAL));
788 }
789
790 static void ieee80211_mesh_rootpath(struct ieee80211_sub_if_data *sdata)
791 {
792         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
793         u32 interval;
794
795         mesh_path_tx_root_frame(sdata);
796
797         if (ifmsh->mshcfg.dot11MeshHWMPRootMode == IEEE80211_PROACTIVE_RANN)
798                 interval = ifmsh->mshcfg.dot11MeshHWMPRannInterval;
799         else
800                 interval = ifmsh->mshcfg.dot11MeshHWMProotInterval;
801
802         mod_timer(&ifmsh->mesh_path_root_timer,
803                   round_jiffies(TU_TO_EXP_TIME(interval)));
804 }
805
806 static int
807 ieee80211_mesh_build_beacon(struct ieee80211_if_mesh *ifmsh)
808 {
809         struct beacon_data *bcn;
810         int head_len, tail_len;
811         struct sk_buff *skb;
812         struct ieee80211_mgmt *mgmt;
813         struct ieee80211_chanctx_conf *chanctx_conf;
814         struct mesh_csa_settings *csa;
815         enum nl80211_band band;
816         u8 ie_len_he_cap;
817         u8 *pos;
818         struct ieee80211_sub_if_data *sdata;
819         int hdr_len = offsetofend(struct ieee80211_mgmt, u.beacon);
820
821         sdata = container_of(ifmsh, struct ieee80211_sub_if_data, u.mesh);
822         rcu_read_lock();
823         chanctx_conf = rcu_dereference(sdata->vif.bss_conf.chanctx_conf);
824         band = chanctx_conf->def.chan->band;
825         rcu_read_unlock();
826
827         ie_len_he_cap = ieee80211_ie_len_he_cap(sdata,
828                                                 NL80211_IFTYPE_MESH_POINT);
829         head_len = hdr_len +
830                    2 + /* NULL SSID */
831                    /* Channel Switch Announcement */
832                    2 + sizeof(struct ieee80211_channel_sw_ie) +
833                    /* Mesh Channel Switch Parameters */
834                    2 + sizeof(struct ieee80211_mesh_chansw_params_ie) +
835                    /* Channel Switch Wrapper + Wide Bandwidth CSA IE */
836                    2 + 2 + sizeof(struct ieee80211_wide_bw_chansw_ie) +
837                    2 + sizeof(struct ieee80211_sec_chan_offs_ie) +
838                    2 + 8 + /* supported rates */
839                    2 + 3; /* DS params */
840         tail_len = 2 + (IEEE80211_MAX_SUPP_RATES - 8) +
841                    2 + sizeof(struct ieee80211_ht_cap) +
842                    2 + sizeof(struct ieee80211_ht_operation) +
843                    2 + ifmsh->mesh_id_len +
844                    2 + sizeof(struct ieee80211_meshconf_ie) +
845                    2 + sizeof(__le16) + /* awake window */
846                    2 + sizeof(struct ieee80211_vht_cap) +
847                    2 + sizeof(struct ieee80211_vht_operation) +
848                    ie_len_he_cap +
849                    2 + 1 + sizeof(struct ieee80211_he_operation) +
850                            sizeof(struct ieee80211_he_6ghz_oper) +
851                    2 + 1 + sizeof(struct ieee80211_he_6ghz_capa) +
852                    ifmsh->ie_len;
853
854         bcn = kzalloc(sizeof(*bcn) + head_len + tail_len, GFP_KERNEL);
855         /* need an skb for IE builders to operate on */
856         skb = __dev_alloc_skb(max(head_len, tail_len), GFP_KERNEL);
857
858         if (!bcn || !skb)
859                 goto out_free;
860
861         /*
862          * pointers go into the block we allocated,
863          * memory is | beacon_data | head | tail |
864          */
865         bcn->head = ((u8 *) bcn) + sizeof(*bcn);
866
867         /* fill in the head */
868         mgmt = skb_put_zero(skb, hdr_len);
869         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
870                                           IEEE80211_STYPE_BEACON);
871         eth_broadcast_addr(mgmt->da);
872         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
873         memcpy(mgmt->bssid, sdata->vif.addr, ETH_ALEN);
874         ieee80211_mps_set_frame_flags(sdata, NULL, (void *) mgmt);
875         mgmt->u.beacon.beacon_int =
876                 cpu_to_le16(sdata->vif.bss_conf.beacon_int);
877         mgmt->u.beacon.capab_info |= cpu_to_le16(
878                 sdata->u.mesh.security ? WLAN_CAPABILITY_PRIVACY : 0);
879
880         pos = skb_put(skb, 2);
881         *pos++ = WLAN_EID_SSID;
882         *pos++ = 0x0;
883
884         rcu_read_lock();
885         csa = rcu_dereference(ifmsh->csa);
886         if (csa) {
887                 enum nl80211_channel_type ct;
888                 struct cfg80211_chan_def *chandef;
889                 int ie_len = 2 + sizeof(struct ieee80211_channel_sw_ie) +
890                              2 + sizeof(struct ieee80211_mesh_chansw_params_ie);
891
892                 pos = skb_put_zero(skb, ie_len);
893                 *pos++ = WLAN_EID_CHANNEL_SWITCH;
894                 *pos++ = 3;
895                 *pos++ = 0x0;
896                 *pos++ = ieee80211_frequency_to_channel(
897                                 csa->settings.chandef.chan->center_freq);
898                 bcn->cntdwn_current_counter = csa->settings.count;
899                 bcn->cntdwn_counter_offsets[0] = hdr_len + 6;
900                 *pos++ = csa->settings.count;
901                 *pos++ = WLAN_EID_CHAN_SWITCH_PARAM;
902                 *pos++ = 6;
903                 if (ifmsh->csa_role == IEEE80211_MESH_CSA_ROLE_INIT) {
904                         *pos++ = ifmsh->mshcfg.dot11MeshTTL;
905                         *pos |= WLAN_EID_CHAN_SWITCH_PARAM_INITIATOR;
906                 } else {
907                         *pos++ = ifmsh->chsw_ttl;
908                 }
909                 *pos++ |= csa->settings.block_tx ?
910                           WLAN_EID_CHAN_SWITCH_PARAM_TX_RESTRICT : 0x00;
911                 put_unaligned_le16(WLAN_REASON_MESH_CHAN, pos);
912                 pos += 2;
913                 put_unaligned_le16(ifmsh->pre_value, pos);
914                 pos += 2;
915
916                 switch (csa->settings.chandef.width) {
917                 case NL80211_CHAN_WIDTH_40:
918                         ie_len = 2 + sizeof(struct ieee80211_sec_chan_offs_ie);
919                         pos = skb_put_zero(skb, ie_len);
920
921                         *pos++ = WLAN_EID_SECONDARY_CHANNEL_OFFSET; /* EID */
922                         *pos++ = 1;                                 /* len */
923                         ct = cfg80211_get_chandef_type(&csa->settings.chandef);
924                         if (ct == NL80211_CHAN_HT40PLUS)
925                                 *pos++ = IEEE80211_HT_PARAM_CHA_SEC_ABOVE;
926                         else
927                                 *pos++ = IEEE80211_HT_PARAM_CHA_SEC_BELOW;
928                         break;
929                 case NL80211_CHAN_WIDTH_80:
930                 case NL80211_CHAN_WIDTH_80P80:
931                 case NL80211_CHAN_WIDTH_160:
932                         /* Channel Switch Wrapper + Wide Bandwidth CSA IE */
933                         ie_len = 2 + 2 +
934                                  sizeof(struct ieee80211_wide_bw_chansw_ie);
935                         pos = skb_put_zero(skb, ie_len);
936
937                         *pos++ = WLAN_EID_CHANNEL_SWITCH_WRAPPER; /* EID */
938                         *pos++ = 5;                               /* len */
939                         /* put sub IE */
940                         chandef = &csa->settings.chandef;
941                         ieee80211_ie_build_wide_bw_cs(pos, chandef);
942                         break;
943                 default:
944                         break;
945                 }
946         }
947         rcu_read_unlock();
948
949         if (ieee80211_add_srates_ie(sdata, skb, true, band) ||
950             mesh_add_ds_params_ie(sdata, skb))
951                 goto out_free;
952
953         bcn->head_len = skb->len;
954         memcpy(bcn->head, skb->data, bcn->head_len);
955
956         /* now the tail */
957         skb_trim(skb, 0);
958         bcn->tail = bcn->head + bcn->head_len;
959
960         if (ieee80211_add_ext_srates_ie(sdata, skb, true, band) ||
961             mesh_add_rsn_ie(sdata, skb) ||
962             mesh_add_ht_cap_ie(sdata, skb) ||
963             mesh_add_ht_oper_ie(sdata, skb) ||
964             mesh_add_meshid_ie(sdata, skb) ||
965             mesh_add_meshconf_ie(sdata, skb) ||
966             mesh_add_awake_window_ie(sdata, skb) ||
967             mesh_add_vht_cap_ie(sdata, skb) ||
968             mesh_add_vht_oper_ie(sdata, skb) ||
969             mesh_add_he_cap_ie(sdata, skb, ie_len_he_cap) ||
970             mesh_add_he_oper_ie(sdata, skb) ||
971             mesh_add_he_6ghz_cap_ie(sdata, skb) ||
972             mesh_add_vendor_ies(sdata, skb))
973                 goto out_free;
974
975         bcn->tail_len = skb->len;
976         memcpy(bcn->tail, skb->data, bcn->tail_len);
977         ieee80211_mesh_update_bss_params(sdata, bcn->tail, bcn->tail_len);
978         bcn->meshconf = (struct ieee80211_meshconf_ie *)
979                                         (bcn->tail + ifmsh->meshconf_offset);
980
981         dev_kfree_skb(skb);
982         rcu_assign_pointer(ifmsh->beacon, bcn);
983         return 0;
984 out_free:
985         kfree(bcn);
986         dev_kfree_skb(skb);
987         return -ENOMEM;
988 }
989
990 static int
991 ieee80211_mesh_rebuild_beacon(struct ieee80211_sub_if_data *sdata)
992 {
993         struct beacon_data *old_bcn;
994         int ret;
995
996         old_bcn = sdata_dereference(sdata->u.mesh.beacon, sdata);
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_link_info_change_notify(sdata, &sdata->deflink, 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_link_info_change_notify(sdata, &sdata->deflink,
1084                                           BSS_CHANGED_BEACON_ENABLED);
1085
1086         /* remove beacon */
1087         bcn = sdata_dereference(ifmsh->beacon, sdata);
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         ieee80211_conn_flags_t conn_flags = 0;
1133         u32 vht_cap_info = 0;
1134
1135         sdata_assert_lock(sdata);
1136
1137         sband = ieee80211_get_sband(sdata);
1138         if (!sband)
1139                 return false;
1140
1141         switch (sdata->vif.bss_conf.chandef.width) {
1142         case NL80211_CHAN_WIDTH_20_NOHT:
1143                 conn_flags |= IEEE80211_CONN_DISABLE_HT;
1144                 fallthrough;
1145         case NL80211_CHAN_WIDTH_20:
1146                 conn_flags |= IEEE80211_CONN_DISABLE_40MHZ;
1147                 fallthrough;
1148         case NL80211_CHAN_WIDTH_40:
1149                 conn_flags |= IEEE80211_CONN_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                                            conn_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, NULL);
1260         if (!elems)
1261                 return;
1262
1263         if (!elems->mesh_id)
1264                 goto free;
1265
1266         /* 802.11-2012 10.1.4.3.2 */
1267         if ((!ether_addr_equal(mgmt->da, sdata->vif.addr) &&
1268              !is_broadcast_ether_addr(mgmt->da)) ||
1269             elems->ssid_len != 0)
1270                 goto free;
1271
1272         if (elems->mesh_id_len != 0 &&
1273             (elems->mesh_id_len != ifmsh->mesh_id_len ||
1274              memcmp(elems->mesh_id, ifmsh->mesh_id, ifmsh->mesh_id_len)))
1275                 goto free;
1276
1277         rcu_read_lock();
1278         bcn = rcu_dereference(ifmsh->beacon);
1279
1280         if (!bcn)
1281                 goto out;
1282
1283         presp = dev_alloc_skb(local->tx_headroom +
1284                               bcn->head_len + bcn->tail_len);
1285         if (!presp)
1286                 goto out;
1287
1288         skb_reserve(presp, local->tx_headroom);
1289         skb_put_data(presp, bcn->head, bcn->head_len);
1290         skb_put_data(presp, bcn->tail, bcn->tail_len);
1291         hdr = (struct ieee80211_mgmt *) presp->data;
1292         hdr->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
1293                                          IEEE80211_STYPE_PROBE_RESP);
1294         memcpy(hdr->da, mgmt->sa, ETH_ALEN);
1295         IEEE80211_SKB_CB(presp)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
1296         ieee80211_tx_skb(sdata, presp);
1297 out:
1298         rcu_read_unlock();
1299 free:
1300         kfree(elems);
1301 }
1302
1303 static void ieee80211_mesh_rx_bcn_presp(struct ieee80211_sub_if_data *sdata,
1304                                         u16 stype,
1305                                         struct ieee80211_mgmt *mgmt,
1306                                         size_t len,
1307                                         struct ieee80211_rx_status *rx_status)
1308 {
1309         struct ieee80211_local *local = sdata->local;
1310         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1311         struct ieee802_11_elems *elems;
1312         struct ieee80211_channel *channel;
1313         size_t baselen;
1314         int freq;
1315         enum nl80211_band band = rx_status->band;
1316
1317         /* ignore ProbeResp to foreign address */
1318         if (stype == IEEE80211_STYPE_PROBE_RESP &&
1319             !ether_addr_equal(mgmt->da, sdata->vif.addr))
1320                 return;
1321
1322         baselen = (u8 *) mgmt->u.probe_resp.variable - (u8 *) mgmt;
1323         if (baselen > len)
1324                 return;
1325
1326         elems = ieee802_11_parse_elems(mgmt->u.probe_resp.variable,
1327                                        len - baselen,
1328                                        false, NULL);
1329         if (!elems)
1330                 return;
1331
1332         /* ignore non-mesh or secure / unsecure mismatch */
1333         if ((!elems->mesh_id || !elems->mesh_config) ||
1334             (elems->rsn && sdata->u.mesh.security == IEEE80211_MESH_SEC_NONE) ||
1335             (!elems->rsn && sdata->u.mesh.security != IEEE80211_MESH_SEC_NONE))
1336                 goto free;
1337
1338         if (elems->ds_params)
1339                 freq = ieee80211_channel_to_frequency(elems->ds_params[0], band);
1340         else
1341                 freq = rx_status->freq;
1342
1343         channel = ieee80211_get_channel(local->hw.wiphy, freq);
1344
1345         if (!channel || channel->flags & IEEE80211_CHAN_DISABLED)
1346                 goto free;
1347
1348         if (mesh_matches_local(sdata, elems)) {
1349                 mpl_dbg(sdata, "rssi_threshold=%d,rx_status->signal=%d\n",
1350                         sdata->u.mesh.mshcfg.rssi_threshold, rx_status->signal);
1351                 if (!sdata->u.mesh.user_mpm ||
1352                     sdata->u.mesh.mshcfg.rssi_threshold == 0 ||
1353                     sdata->u.mesh.mshcfg.rssi_threshold < rx_status->signal)
1354                         mesh_neighbour_update(sdata, mgmt->sa, elems,
1355                                               rx_status);
1356
1357                 if (ifmsh->csa_role != IEEE80211_MESH_CSA_ROLE_INIT &&
1358                     !sdata->vif.bss_conf.csa_active)
1359                         ieee80211_mesh_process_chnswitch(sdata, elems, true);
1360         }
1361
1362         if (ifmsh->sync_ops)
1363                 ifmsh->sync_ops->rx_bcn_presp(sdata, stype, mgmt, len,
1364                                               elems->mesh_config, rx_status);
1365 free:
1366         kfree(elems);
1367 }
1368
1369 int ieee80211_mesh_finish_csa(struct ieee80211_sub_if_data *sdata)
1370 {
1371         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1372         struct mesh_csa_settings *tmp_csa_settings;
1373         int ret = 0;
1374         int changed = 0;
1375
1376         /* Reset the TTL value and Initiator flag */
1377         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_NONE;
1378         ifmsh->chsw_ttl = 0;
1379
1380         /* Remove the CSA and MCSP elements from the beacon */
1381         tmp_csa_settings = sdata_dereference(ifmsh->csa, sdata);
1382         RCU_INIT_POINTER(ifmsh->csa, NULL);
1383         if (tmp_csa_settings)
1384                 kfree_rcu(tmp_csa_settings, rcu_head);
1385         ret = ieee80211_mesh_rebuild_beacon(sdata);
1386         if (ret)
1387                 return -EINVAL;
1388
1389         changed |= BSS_CHANGED_BEACON;
1390
1391         mcsa_dbg(sdata, "complete switching to center freq %d MHz",
1392                  sdata->vif.bss_conf.chandef.chan->center_freq);
1393         return changed;
1394 }
1395
1396 int ieee80211_mesh_csa_beacon(struct ieee80211_sub_if_data *sdata,
1397                               struct cfg80211_csa_settings *csa_settings)
1398 {
1399         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1400         struct mesh_csa_settings *tmp_csa_settings;
1401         int ret = 0;
1402
1403         lockdep_assert_held(&sdata->wdev.mtx);
1404
1405         tmp_csa_settings = kmalloc(sizeof(*tmp_csa_settings),
1406                                    GFP_ATOMIC);
1407         if (!tmp_csa_settings)
1408                 return -ENOMEM;
1409
1410         memcpy(&tmp_csa_settings->settings, csa_settings,
1411                sizeof(struct cfg80211_csa_settings));
1412
1413         rcu_assign_pointer(ifmsh->csa, tmp_csa_settings);
1414
1415         ret = ieee80211_mesh_rebuild_beacon(sdata);
1416         if (ret) {
1417                 tmp_csa_settings = rcu_dereference(ifmsh->csa);
1418                 RCU_INIT_POINTER(ifmsh->csa, NULL);
1419                 kfree_rcu(tmp_csa_settings, rcu_head);
1420                 return ret;
1421         }
1422
1423         return BSS_CHANGED_BEACON;
1424 }
1425
1426 static int mesh_fwd_csa_frame(struct ieee80211_sub_if_data *sdata,
1427                                struct ieee80211_mgmt *mgmt, size_t len,
1428                                struct ieee802_11_elems *elems)
1429 {
1430         struct ieee80211_mgmt *mgmt_fwd;
1431         struct sk_buff *skb;
1432         struct ieee80211_local *local = sdata->local;
1433
1434         skb = dev_alloc_skb(local->tx_headroom + len);
1435         if (!skb)
1436                 return -ENOMEM;
1437         skb_reserve(skb, local->tx_headroom);
1438         mgmt_fwd = skb_put(skb, len);
1439
1440         elems->mesh_chansw_params_ie->mesh_ttl--;
1441         elems->mesh_chansw_params_ie->mesh_flags &=
1442                 ~WLAN_EID_CHAN_SWITCH_PARAM_INITIATOR;
1443
1444         memcpy(mgmt_fwd, mgmt, len);
1445         eth_broadcast_addr(mgmt_fwd->da);
1446         memcpy(mgmt_fwd->sa, sdata->vif.addr, ETH_ALEN);
1447         memcpy(mgmt_fwd->bssid, sdata->vif.addr, ETH_ALEN);
1448
1449         ieee80211_tx_skb(sdata, skb);
1450         return 0;
1451 }
1452
1453 static void mesh_rx_csa_frame(struct ieee80211_sub_if_data *sdata,
1454                               struct ieee80211_mgmt *mgmt, size_t len)
1455 {
1456         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1457         struct ieee802_11_elems *elems;
1458         u16 pre_value;
1459         bool fwd_csa = true;
1460         size_t baselen;
1461         u8 *pos;
1462
1463         if (mgmt->u.action.u.measurement.action_code !=
1464             WLAN_ACTION_SPCT_CHL_SWITCH)
1465                 return;
1466
1467         pos = mgmt->u.action.u.chan_switch.variable;
1468         baselen = offsetof(struct ieee80211_mgmt,
1469                            u.action.u.chan_switch.variable);
1470         elems = ieee802_11_parse_elems(pos, len - baselen, true, NULL);
1471         if (!elems)
1472                 return;
1473
1474         if (!mesh_matches_local(sdata, elems))
1475                 goto free;
1476
1477         ifmsh->chsw_ttl = elems->mesh_chansw_params_ie->mesh_ttl;
1478         if (!--ifmsh->chsw_ttl)
1479                 fwd_csa = false;
1480
1481         pre_value = le16_to_cpu(elems->mesh_chansw_params_ie->mesh_pre_value);
1482         if (ifmsh->pre_value >= pre_value)
1483                 goto free;
1484
1485         ifmsh->pre_value = pre_value;
1486
1487         if (!sdata->vif.bss_conf.csa_active &&
1488             !ieee80211_mesh_process_chnswitch(sdata, elems, false)) {
1489                 mcsa_dbg(sdata, "Failed to process CSA action frame");
1490                 goto free;
1491         }
1492
1493         /* forward or re-broadcast the CSA frame */
1494         if (fwd_csa) {
1495                 if (mesh_fwd_csa_frame(sdata, mgmt, len, elems) < 0)
1496                         mcsa_dbg(sdata, "Failed to forward the CSA frame");
1497         }
1498 free:
1499         kfree(elems);
1500 }
1501
1502 static void ieee80211_mesh_rx_mgmt_action(struct ieee80211_sub_if_data *sdata,
1503                                           struct ieee80211_mgmt *mgmt,
1504                                           size_t len,
1505                                           struct ieee80211_rx_status *rx_status)
1506 {
1507         switch (mgmt->u.action.category) {
1508         case WLAN_CATEGORY_SELF_PROTECTED:
1509                 switch (mgmt->u.action.u.self_prot.action_code) {
1510                 case WLAN_SP_MESH_PEERING_OPEN:
1511                 case WLAN_SP_MESH_PEERING_CLOSE:
1512                 case WLAN_SP_MESH_PEERING_CONFIRM:
1513                         mesh_rx_plink_frame(sdata, mgmt, len, rx_status);
1514                         break;
1515                 }
1516                 break;
1517         case WLAN_CATEGORY_MESH_ACTION:
1518                 if (mesh_action_is_path_sel(mgmt))
1519                         mesh_rx_path_sel_frame(sdata, mgmt, len);
1520                 break;
1521         case WLAN_CATEGORY_SPECTRUM_MGMT:
1522                 mesh_rx_csa_frame(sdata, mgmt, len);
1523                 break;
1524         }
1525 }
1526
1527 void ieee80211_mesh_rx_queued_mgmt(struct ieee80211_sub_if_data *sdata,
1528                                    struct sk_buff *skb)
1529 {
1530         struct ieee80211_rx_status *rx_status;
1531         struct ieee80211_mgmt *mgmt;
1532         u16 stype;
1533
1534         sdata_lock(sdata);
1535
1536         /* mesh already went down */
1537         if (!sdata->u.mesh.mesh_id_len)
1538                 goto out;
1539
1540         rx_status = IEEE80211_SKB_RXCB(skb);
1541         mgmt = (struct ieee80211_mgmt *) skb->data;
1542         stype = le16_to_cpu(mgmt->frame_control) & IEEE80211_FCTL_STYPE;
1543
1544         switch (stype) {
1545         case IEEE80211_STYPE_PROBE_RESP:
1546         case IEEE80211_STYPE_BEACON:
1547                 ieee80211_mesh_rx_bcn_presp(sdata, stype, mgmt, skb->len,
1548                                             rx_status);
1549                 break;
1550         case IEEE80211_STYPE_PROBE_REQ:
1551                 ieee80211_mesh_rx_probe_req(sdata, mgmt, skb->len);
1552                 break;
1553         case IEEE80211_STYPE_ACTION:
1554                 ieee80211_mesh_rx_mgmt_action(sdata, mgmt, skb->len, rx_status);
1555                 break;
1556         }
1557 out:
1558         sdata_unlock(sdata);
1559 }
1560
1561 static void mesh_bss_info_changed(struct ieee80211_sub_if_data *sdata)
1562 {
1563         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1564         u32 bit, changed = 0;
1565
1566         for_each_set_bit(bit, &ifmsh->mbss_changed,
1567                          sizeof(changed) * BITS_PER_BYTE) {
1568                 clear_bit(bit, &ifmsh->mbss_changed);
1569                 changed |= BIT(bit);
1570         }
1571
1572         if (sdata->vif.bss_conf.enable_beacon &&
1573             (changed & (BSS_CHANGED_BEACON |
1574                         BSS_CHANGED_HT |
1575                         BSS_CHANGED_BASIC_RATES |
1576                         BSS_CHANGED_BEACON_INT)))
1577                 if (ieee80211_mesh_rebuild_beacon(sdata))
1578                         return;
1579
1580         ieee80211_link_info_change_notify(sdata, &sdata->deflink, changed);
1581 }
1582
1583 void ieee80211_mesh_work(struct ieee80211_sub_if_data *sdata)
1584 {
1585         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1586
1587         sdata_lock(sdata);
1588
1589         /* mesh already went down */
1590         if (!sdata->u.mesh.mesh_id_len)
1591                 goto out;
1592
1593         if (ifmsh->preq_queue_len &&
1594             time_after(jiffies,
1595                        ifmsh->last_preq + msecs_to_jiffies(ifmsh->mshcfg.dot11MeshHWMPpreqMinInterval)))
1596                 mesh_path_start_discovery(sdata);
1597
1598         if (test_and_clear_bit(MESH_WORK_HOUSEKEEPING, &ifmsh->wrkq_flags))
1599                 ieee80211_mesh_housekeeping(sdata);
1600
1601         if (test_and_clear_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags))
1602                 ieee80211_mesh_rootpath(sdata);
1603
1604         if (test_and_clear_bit(MESH_WORK_DRIFT_ADJUST, &ifmsh->wrkq_flags))
1605                 mesh_sync_adjust_tsf(sdata);
1606
1607         if (test_and_clear_bit(MESH_WORK_MBSS_CHANGED, &ifmsh->wrkq_flags))
1608                 mesh_bss_info_changed(sdata);
1609 out:
1610         sdata_unlock(sdata);
1611 }
1612
1613
1614 void ieee80211_mesh_init_sdata(struct ieee80211_sub_if_data *sdata)
1615 {
1616         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1617         static u8 zero_addr[ETH_ALEN] = {};
1618
1619         timer_setup(&ifmsh->housekeeping_timer,
1620                     ieee80211_mesh_housekeeping_timer, 0);
1621
1622         ifmsh->accepting_plinks = true;
1623         atomic_set(&ifmsh->mpaths, 0);
1624         mesh_rmc_init(sdata);
1625         ifmsh->last_preq = jiffies;
1626         ifmsh->next_perr = jiffies;
1627         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_NONE;
1628         /* Allocate all mesh structures when creating the first mesh interface. */
1629         if (!mesh_allocated)
1630                 ieee80211s_init();
1631
1632         mesh_pathtbl_init(sdata);
1633
1634         timer_setup(&ifmsh->mesh_path_timer, ieee80211_mesh_path_timer, 0);
1635         timer_setup(&ifmsh->mesh_path_root_timer,
1636                     ieee80211_mesh_path_root_timer, 0);
1637         INIT_LIST_HEAD(&ifmsh->preq_queue.list);
1638         skb_queue_head_init(&ifmsh->ps.bc_buf);
1639         spin_lock_init(&ifmsh->mesh_preq_queue_lock);
1640         spin_lock_init(&ifmsh->sync_offset_lock);
1641         RCU_INIT_POINTER(ifmsh->beacon, NULL);
1642
1643         sdata->vif.bss_conf.bssid = zero_addr;
1644 }
1645
1646 void ieee80211_mesh_teardown_sdata(struct ieee80211_sub_if_data *sdata)
1647 {
1648         mesh_rmc_free(sdata);
1649         mesh_pathtbl_unregister(sdata);
1650 }