Merge branch 'next' into for-linus
[platform/kernel/linux-starfive.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(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 u8 *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_ie(WLAN_EID_EXT_HE_OPERATION, ie, ie_len);
691         if (cap && cap[1] >= ieee80211_he_oper_size(&cap[3]))
692                 he_oper = (void *)(cap + 3);
693
694         if (he_oper)
695                 sdata->vif.bss_conf.he_oper.params =
696                         __le32_to_cpu(he_oper->he_oper_params);
697 }
698
699 /**
700  * ieee80211_fill_mesh_addresses - fill addresses of a locally originated mesh frame
701  * @hdr:        802.11 frame header
702  * @fc:         frame control field
703  * @meshda:     destination address in the mesh
704  * @meshsa:     source address in the mesh.  Same as TA, as frame is
705  *              locally originated.
706  *
707  * Return the length of the 802.11 (does not include a mesh control header)
708  */
709 int ieee80211_fill_mesh_addresses(struct ieee80211_hdr *hdr, __le16 *fc,
710                                   const u8 *meshda, const u8 *meshsa)
711 {
712         if (is_multicast_ether_addr(meshda)) {
713                 *fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS);
714                 /* DA TA SA */
715                 memcpy(hdr->addr1, meshda, ETH_ALEN);
716                 memcpy(hdr->addr2, meshsa, ETH_ALEN);
717                 memcpy(hdr->addr3, meshsa, ETH_ALEN);
718                 return 24;
719         } else {
720                 *fc |= cpu_to_le16(IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS);
721                 /* RA TA DA SA */
722                 eth_zero_addr(hdr->addr1);   /* RA is resolved later */
723                 memcpy(hdr->addr2, meshsa, ETH_ALEN);
724                 memcpy(hdr->addr3, meshda, ETH_ALEN);
725                 memcpy(hdr->addr4, meshsa, ETH_ALEN);
726                 return 30;
727         }
728 }
729
730 /**
731  * ieee80211_new_mesh_header - create a new mesh header
732  * @sdata:      mesh interface to be used
733  * @meshhdr:    uninitialized mesh header
734  * @addr4or5:   1st address in the ae header, which may correspond to address 4
735  *              (if addr6 is NULL) or address 5 (if addr6 is present). It may
736  *              be NULL.
737  * @addr6:      2nd address in the ae header, which corresponds to addr6 of the
738  *              mesh frame
739  *
740  * Return the header length.
741  */
742 unsigned int ieee80211_new_mesh_header(struct ieee80211_sub_if_data *sdata,
743                                        struct ieee80211s_hdr *meshhdr,
744                                        const char *addr4or5, const char *addr6)
745 {
746         if (WARN_ON(!addr4or5 && addr6))
747                 return 0;
748
749         memset(meshhdr, 0, sizeof(*meshhdr));
750
751         meshhdr->ttl = sdata->u.mesh.mshcfg.dot11MeshTTL;
752
753         /* FIXME: racy -- TX on multiple queues can be concurrent */
754         put_unaligned(cpu_to_le32(sdata->u.mesh.mesh_seqnum), &meshhdr->seqnum);
755         sdata->u.mesh.mesh_seqnum++;
756
757         if (addr4or5 && !addr6) {
758                 meshhdr->flags |= MESH_FLAGS_AE_A4;
759                 memcpy(meshhdr->eaddr1, addr4or5, ETH_ALEN);
760                 return 2 * ETH_ALEN;
761         } else if (addr4or5 && addr6) {
762                 meshhdr->flags |= MESH_FLAGS_AE_A5_A6;
763                 memcpy(meshhdr->eaddr1, addr4or5, ETH_ALEN);
764                 memcpy(meshhdr->eaddr2, addr6, ETH_ALEN);
765                 return 3 * ETH_ALEN;
766         }
767
768         return ETH_ALEN;
769 }
770
771 static void ieee80211_mesh_housekeeping(struct ieee80211_sub_if_data *sdata)
772 {
773         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
774         u32 changed;
775
776         if (ifmsh->mshcfg.plink_timeout > 0)
777                 ieee80211_sta_expire(sdata, ifmsh->mshcfg.plink_timeout * HZ);
778         mesh_path_expire(sdata);
779
780         changed = mesh_accept_plinks_update(sdata);
781         ieee80211_mbss_info_change_notify(sdata, changed);
782
783         mod_timer(&ifmsh->housekeeping_timer,
784                   round_jiffies(jiffies +
785                                 IEEE80211_MESH_HOUSEKEEPING_INTERVAL));
786 }
787
788 static void ieee80211_mesh_rootpath(struct ieee80211_sub_if_data *sdata)
789 {
790         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
791         u32 interval;
792
793         mesh_path_tx_root_frame(sdata);
794
795         if (ifmsh->mshcfg.dot11MeshHWMPRootMode == IEEE80211_PROACTIVE_RANN)
796                 interval = ifmsh->mshcfg.dot11MeshHWMPRannInterval;
797         else
798                 interval = ifmsh->mshcfg.dot11MeshHWMProotInterval;
799
800         mod_timer(&ifmsh->mesh_path_root_timer,
801                   round_jiffies(TU_TO_EXP_TIME(interval)));
802 }
803
804 static int
805 ieee80211_mesh_build_beacon(struct ieee80211_if_mesh *ifmsh)
806 {
807         struct beacon_data *bcn;
808         int head_len, tail_len;
809         struct sk_buff *skb;
810         struct ieee80211_mgmt *mgmt;
811         struct ieee80211_chanctx_conf *chanctx_conf;
812         struct mesh_csa_settings *csa;
813         enum nl80211_band band;
814         u8 ie_len_he_cap;
815         u8 *pos;
816         struct ieee80211_sub_if_data *sdata;
817         int hdr_len = offsetofend(struct ieee80211_mgmt, u.beacon);
818
819         sdata = container_of(ifmsh, struct ieee80211_sub_if_data, u.mesh);
820         rcu_read_lock();
821         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
822         band = chanctx_conf->def.chan->band;
823         rcu_read_unlock();
824
825         ie_len_he_cap = ieee80211_ie_len_he_cap(sdata,
826                                                 NL80211_IFTYPE_MESH_POINT);
827         head_len = hdr_len +
828                    2 + /* NULL SSID */
829                    /* Channel Switch Announcement */
830                    2 + sizeof(struct ieee80211_channel_sw_ie) +
831                    /* Mesh Channel Switch Parameters */
832                    2 + sizeof(struct ieee80211_mesh_chansw_params_ie) +
833                    /* Channel Switch Wrapper + Wide Bandwidth CSA IE */
834                    2 + 2 + sizeof(struct ieee80211_wide_bw_chansw_ie) +
835                    2 + sizeof(struct ieee80211_sec_chan_offs_ie) +
836                    2 + 8 + /* supported rates */
837                    2 + 3; /* DS params */
838         tail_len = 2 + (IEEE80211_MAX_SUPP_RATES - 8) +
839                    2 + sizeof(struct ieee80211_ht_cap) +
840                    2 + sizeof(struct ieee80211_ht_operation) +
841                    2 + ifmsh->mesh_id_len +
842                    2 + sizeof(struct ieee80211_meshconf_ie) +
843                    2 + sizeof(__le16) + /* awake window */
844                    2 + sizeof(struct ieee80211_vht_cap) +
845                    2 + sizeof(struct ieee80211_vht_operation) +
846                    ie_len_he_cap +
847                    2 + 1 + sizeof(struct ieee80211_he_operation) +
848                            sizeof(struct ieee80211_he_6ghz_oper) +
849                    2 + 1 + sizeof(struct ieee80211_he_6ghz_capa) +
850                    ifmsh->ie_len;
851
852         bcn = kzalloc(sizeof(*bcn) + head_len + tail_len, GFP_KERNEL);
853         /* need an skb for IE builders to operate on */
854         skb = dev_alloc_skb(max(head_len, tail_len));
855
856         if (!bcn || !skb)
857                 goto out_free;
858
859         /*
860          * pointers go into the block we allocated,
861          * memory is | beacon_data | head | tail |
862          */
863         bcn->head = ((u8 *) bcn) + sizeof(*bcn);
864
865         /* fill in the head */
866         mgmt = skb_put_zero(skb, hdr_len);
867         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
868                                           IEEE80211_STYPE_BEACON);
869         eth_broadcast_addr(mgmt->da);
870         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
871         memcpy(mgmt->bssid, sdata->vif.addr, ETH_ALEN);
872         ieee80211_mps_set_frame_flags(sdata, NULL, (void *) mgmt);
873         mgmt->u.beacon.beacon_int =
874                 cpu_to_le16(sdata->vif.bss_conf.beacon_int);
875         mgmt->u.beacon.capab_info |= cpu_to_le16(
876                 sdata->u.mesh.security ? WLAN_CAPABILITY_PRIVACY : 0);
877
878         pos = skb_put(skb, 2);
879         *pos++ = WLAN_EID_SSID;
880         *pos++ = 0x0;
881
882         rcu_read_lock();
883         csa = rcu_dereference(ifmsh->csa);
884         if (csa) {
885                 enum nl80211_channel_type ct;
886                 struct cfg80211_chan_def *chandef;
887                 int ie_len = 2 + sizeof(struct ieee80211_channel_sw_ie) +
888                              2 + sizeof(struct ieee80211_mesh_chansw_params_ie);
889
890                 pos = skb_put_zero(skb, ie_len);
891                 *pos++ = WLAN_EID_CHANNEL_SWITCH;
892                 *pos++ = 3;
893                 *pos++ = 0x0;
894                 *pos++ = ieee80211_frequency_to_channel(
895                                 csa->settings.chandef.chan->center_freq);
896                 bcn->cntdwn_current_counter = csa->settings.count;
897                 bcn->cntdwn_counter_offsets[0] = hdr_len + 6;
898                 *pos++ = csa->settings.count;
899                 *pos++ = WLAN_EID_CHAN_SWITCH_PARAM;
900                 *pos++ = 6;
901                 if (ifmsh->csa_role == IEEE80211_MESH_CSA_ROLE_INIT) {
902                         *pos++ = ifmsh->mshcfg.dot11MeshTTL;
903                         *pos |= WLAN_EID_CHAN_SWITCH_PARAM_INITIATOR;
904                 } else {
905                         *pos++ = ifmsh->chsw_ttl;
906                 }
907                 *pos++ |= csa->settings.block_tx ?
908                           WLAN_EID_CHAN_SWITCH_PARAM_TX_RESTRICT : 0x00;
909                 put_unaligned_le16(WLAN_REASON_MESH_CHAN, pos);
910                 pos += 2;
911                 put_unaligned_le16(ifmsh->pre_value, pos);
912                 pos += 2;
913
914                 switch (csa->settings.chandef.width) {
915                 case NL80211_CHAN_WIDTH_40:
916                         ie_len = 2 + sizeof(struct ieee80211_sec_chan_offs_ie);
917                         pos = skb_put_zero(skb, ie_len);
918
919                         *pos++ = WLAN_EID_SECONDARY_CHANNEL_OFFSET; /* EID */
920                         *pos++ = 1;                                 /* len */
921                         ct = cfg80211_get_chandef_type(&csa->settings.chandef);
922                         if (ct == NL80211_CHAN_HT40PLUS)
923                                 *pos++ = IEEE80211_HT_PARAM_CHA_SEC_ABOVE;
924                         else
925                                 *pos++ = IEEE80211_HT_PARAM_CHA_SEC_BELOW;
926                         break;
927                 case NL80211_CHAN_WIDTH_80:
928                 case NL80211_CHAN_WIDTH_80P80:
929                 case NL80211_CHAN_WIDTH_160:
930                         /* Channel Switch Wrapper + Wide Bandwidth CSA IE */
931                         ie_len = 2 + 2 +
932                                  sizeof(struct ieee80211_wide_bw_chansw_ie);
933                         pos = skb_put_zero(skb, ie_len);
934
935                         *pos++ = WLAN_EID_CHANNEL_SWITCH_WRAPPER; /* EID */
936                         *pos++ = 5;                               /* len */
937                         /* put sub IE */
938                         chandef = &csa->settings.chandef;
939                         ieee80211_ie_build_wide_bw_cs(pos, chandef);
940                         break;
941                 default:
942                         break;
943                 }
944         }
945         rcu_read_unlock();
946
947         if (ieee80211_add_srates_ie(sdata, skb, true, band) ||
948             mesh_add_ds_params_ie(sdata, skb))
949                 goto out_free;
950
951         bcn->head_len = skb->len;
952         memcpy(bcn->head, skb->data, bcn->head_len);
953
954         /* now the tail */
955         skb_trim(skb, 0);
956         bcn->tail = bcn->head + bcn->head_len;
957
958         if (ieee80211_add_ext_srates_ie(sdata, skb, true, band) ||
959             mesh_add_rsn_ie(sdata, skb) ||
960             mesh_add_ht_cap_ie(sdata, skb) ||
961             mesh_add_ht_oper_ie(sdata, skb) ||
962             mesh_add_meshid_ie(sdata, skb) ||
963             mesh_add_meshconf_ie(sdata, skb) ||
964             mesh_add_awake_window_ie(sdata, skb) ||
965             mesh_add_vht_cap_ie(sdata, skb) ||
966             mesh_add_vht_oper_ie(sdata, skb) ||
967             mesh_add_he_cap_ie(sdata, skb, ie_len_he_cap) ||
968             mesh_add_he_oper_ie(sdata, skb) ||
969             mesh_add_he_6ghz_cap_ie(sdata, skb) ||
970             mesh_add_vendor_ies(sdata, skb))
971                 goto out_free;
972
973         bcn->tail_len = skb->len;
974         memcpy(bcn->tail, skb->data, bcn->tail_len);
975         ieee80211_mesh_update_bss_params(sdata, bcn->tail, bcn->tail_len);
976         bcn->meshconf = (struct ieee80211_meshconf_ie *)
977                                         (bcn->tail + ifmsh->meshconf_offset);
978
979         dev_kfree_skb(skb);
980         rcu_assign_pointer(ifmsh->beacon, bcn);
981         return 0;
982 out_free:
983         kfree(bcn);
984         dev_kfree_skb(skb);
985         return -ENOMEM;
986 }
987
988 static int
989 ieee80211_mesh_rebuild_beacon(struct ieee80211_sub_if_data *sdata)
990 {
991         struct beacon_data *old_bcn;
992         int ret;
993
994         old_bcn = rcu_dereference_protected(sdata->u.mesh.beacon,
995                                             lockdep_is_held(&sdata->wdev.mtx));
996         ret = ieee80211_mesh_build_beacon(&sdata->u.mesh);
997         if (ret)
998                 /* just reuse old beacon */
999                 return ret;
1000
1001         if (old_bcn)
1002                 kfree_rcu(old_bcn, rcu_head);
1003         return 0;
1004 }
1005
1006 void ieee80211_mbss_info_change_notify(struct ieee80211_sub_if_data *sdata,
1007                                        u32 changed)
1008 {
1009         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1010         unsigned long bits = changed;
1011         u32 bit;
1012
1013         if (!bits)
1014                 return;
1015
1016         /* if we race with running work, worst case this work becomes a noop */
1017         for_each_set_bit(bit, &bits, sizeof(changed) * BITS_PER_BYTE)
1018                 set_bit(bit, &ifmsh->mbss_changed);
1019         set_bit(MESH_WORK_MBSS_CHANGED, &ifmsh->wrkq_flags);
1020         ieee80211_queue_work(&sdata->local->hw, &sdata->work);
1021 }
1022
1023 int ieee80211_start_mesh(struct ieee80211_sub_if_data *sdata)
1024 {
1025         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1026         struct ieee80211_local *local = sdata->local;
1027         u32 changed = BSS_CHANGED_BEACON |
1028                       BSS_CHANGED_BEACON_ENABLED |
1029                       BSS_CHANGED_HT |
1030                       BSS_CHANGED_BASIC_RATES |
1031                       BSS_CHANGED_BEACON_INT |
1032                       BSS_CHANGED_MCAST_RATE;
1033
1034         local->fif_other_bss++;
1035         /* mesh ifaces must set allmulti to forward mcast traffic */
1036         atomic_inc(&local->iff_allmultis);
1037         ieee80211_configure_filter(local);
1038
1039         ifmsh->mesh_cc_id = 0;  /* Disabled */
1040         /* register sync ops from extensible synchronization framework */
1041         ifmsh->sync_ops = ieee80211_mesh_sync_ops_get(ifmsh->mesh_sp_id);
1042         ifmsh->sync_offset_clockdrift_max = 0;
1043         set_bit(MESH_WORK_HOUSEKEEPING, &ifmsh->wrkq_flags);
1044         ieee80211_mesh_root_setup(ifmsh);
1045         ieee80211_queue_work(&local->hw, &sdata->work);
1046         sdata->vif.bss_conf.ht_operation_mode =
1047                                 ifmsh->mshcfg.ht_opmode;
1048         sdata->vif.bss_conf.enable_beacon = true;
1049
1050         changed |= ieee80211_mps_local_status_update(sdata);
1051
1052         if (ieee80211_mesh_build_beacon(ifmsh)) {
1053                 ieee80211_stop_mesh(sdata);
1054                 return -ENOMEM;
1055         }
1056
1057         ieee80211_recalc_dtim(local, sdata);
1058         ieee80211_bss_info_change_notify(sdata, changed);
1059
1060         netif_carrier_on(sdata->dev);
1061         return 0;
1062 }
1063
1064 void ieee80211_stop_mesh(struct ieee80211_sub_if_data *sdata)
1065 {
1066         struct ieee80211_local *local = sdata->local;
1067         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1068         struct beacon_data *bcn;
1069
1070         netif_carrier_off(sdata->dev);
1071
1072         /* flush STAs and mpaths on this iface */
1073         sta_info_flush(sdata);
1074         ieee80211_free_keys(sdata, true);
1075         mesh_path_flush_by_iface(sdata);
1076
1077         /* stop the beacon */
1078         ifmsh->mesh_id_len = 0;
1079         sdata->vif.bss_conf.enable_beacon = false;
1080         sdata->beacon_rate_set = false;
1081         clear_bit(SDATA_STATE_OFFCHANNEL_BEACON_STOPPED, &sdata->state);
1082         ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_BEACON_ENABLED);
1083
1084         /* remove beacon */
1085         bcn = rcu_dereference_protected(ifmsh->beacon,
1086                                         lockdep_is_held(&sdata->wdev.mtx));
1087         RCU_INIT_POINTER(ifmsh->beacon, NULL);
1088         kfree_rcu(bcn, rcu_head);
1089
1090         /* free all potentially still buffered group-addressed frames */
1091         local->total_ps_buffered -= skb_queue_len(&ifmsh->ps.bc_buf);
1092         skb_queue_purge(&ifmsh->ps.bc_buf);
1093
1094         del_timer_sync(&sdata->u.mesh.housekeeping_timer);
1095         del_timer_sync(&sdata->u.mesh.mesh_path_root_timer);
1096         del_timer_sync(&sdata->u.mesh.mesh_path_timer);
1097
1098         /* clear any mesh work (for next join) we may have accrued */
1099         ifmsh->wrkq_flags = 0;
1100         ifmsh->mbss_changed = 0;
1101
1102         local->fif_other_bss--;
1103         atomic_dec(&local->iff_allmultis);
1104         ieee80211_configure_filter(local);
1105 }
1106
1107 static void ieee80211_mesh_csa_mark_radar(struct ieee80211_sub_if_data *sdata)
1108 {
1109         int err;
1110
1111         /* if the current channel is a DFS channel, mark the channel as
1112          * unavailable.
1113          */
1114         err = cfg80211_chandef_dfs_required(sdata->local->hw.wiphy,
1115                                             &sdata->vif.bss_conf.chandef,
1116                                             NL80211_IFTYPE_MESH_POINT);
1117         if (err > 0)
1118                 cfg80211_radar_event(sdata->local->hw.wiphy,
1119                                      &sdata->vif.bss_conf.chandef, GFP_ATOMIC);
1120 }
1121
1122 static bool
1123 ieee80211_mesh_process_chnswitch(struct ieee80211_sub_if_data *sdata,
1124                                  struct ieee802_11_elems *elems, bool beacon)
1125 {
1126         struct cfg80211_csa_settings params;
1127         struct ieee80211_csa_ie csa_ie;
1128         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1129         struct ieee80211_supported_band *sband;
1130         int err;
1131         u32 sta_flags, vht_cap_info = 0;
1132
1133         sdata_assert_lock(sdata);
1134
1135         sband = ieee80211_get_sband(sdata);
1136         if (!sband)
1137                 return false;
1138
1139         sta_flags = 0;
1140         switch (sdata->vif.bss_conf.chandef.width) {
1141         case NL80211_CHAN_WIDTH_20_NOHT:
1142                 sta_flags |= IEEE80211_STA_DISABLE_HT;
1143                 fallthrough;
1144         case NL80211_CHAN_WIDTH_20:
1145                 sta_flags |= IEEE80211_STA_DISABLE_40MHZ;
1146                 fallthrough;
1147         case NL80211_CHAN_WIDTH_40:
1148                 sta_flags |= IEEE80211_STA_DISABLE_VHT;
1149                 break;
1150         default:
1151                 break;
1152         }
1153
1154         if (elems->vht_cap_elem)
1155                 vht_cap_info =
1156                         le32_to_cpu(elems->vht_cap_elem->vht_cap_info);
1157
1158         memset(&params, 0, sizeof(params));
1159         err = ieee80211_parse_ch_switch_ie(sdata, elems, sband->band,
1160                                            vht_cap_info,
1161                                            sta_flags, sdata->vif.addr,
1162                                            &csa_ie);
1163         if (err < 0)
1164                 return false;
1165         if (err)
1166                 return false;
1167
1168         /* Mark the channel unavailable if the reason for the switch is
1169          * regulatory.
1170          */
1171         if (csa_ie.reason_code == WLAN_REASON_MESH_CHAN_REGULATORY)
1172                 ieee80211_mesh_csa_mark_radar(sdata);
1173
1174         params.chandef = csa_ie.chandef;
1175         params.count = csa_ie.count;
1176
1177         if (!cfg80211_chandef_usable(sdata->local->hw.wiphy, &params.chandef,
1178                                      IEEE80211_CHAN_DISABLED) ||
1179             !cfg80211_reg_can_beacon(sdata->local->hw.wiphy, &params.chandef,
1180                                      NL80211_IFTYPE_MESH_POINT)) {
1181                 sdata_info(sdata,
1182                            "mesh STA %pM switches to unsupported channel (%d MHz, width:%d, CF1/2: %d/%d MHz), aborting\n",
1183                            sdata->vif.addr,
1184                            params.chandef.chan->center_freq,
1185                            params.chandef.width,
1186                            params.chandef.center_freq1,
1187                            params.chandef.center_freq2);
1188                 return false;
1189         }
1190
1191         err = cfg80211_chandef_dfs_required(sdata->local->hw.wiphy,
1192                                             &params.chandef,
1193                                             NL80211_IFTYPE_MESH_POINT);
1194         if (err < 0)
1195                 return false;
1196         if (err > 0 && !ifmsh->userspace_handles_dfs) {
1197                 sdata_info(sdata,
1198                            "mesh STA %pM switches to channel requiring DFS (%d MHz, width:%d, CF1/2: %d/%d MHz), aborting\n",
1199                            sdata->vif.addr,
1200                            params.chandef.chan->center_freq,
1201                            params.chandef.width,
1202                            params.chandef.center_freq1,
1203                            params.chandef.center_freq2);
1204                 return false;
1205         }
1206
1207         params.radar_required = err;
1208
1209         if (cfg80211_chandef_identical(&params.chandef,
1210                                        &sdata->vif.bss_conf.chandef)) {
1211                 mcsa_dbg(sdata,
1212                          "received csa with an identical chandef, ignoring\n");
1213                 return true;
1214         }
1215
1216         mcsa_dbg(sdata,
1217                  "received channel switch announcement to go to channel %d MHz\n",
1218                  params.chandef.chan->center_freq);
1219
1220         params.block_tx = csa_ie.mode & WLAN_EID_CHAN_SWITCH_PARAM_TX_RESTRICT;
1221         if (beacon) {
1222                 ifmsh->chsw_ttl = csa_ie.ttl - 1;
1223                 if (ifmsh->pre_value >= csa_ie.pre_value)
1224                         return false;
1225                 ifmsh->pre_value = csa_ie.pre_value;
1226         }
1227
1228         if (ifmsh->chsw_ttl >= ifmsh->mshcfg.dot11MeshTTL)
1229                 return false;
1230
1231         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_REPEATER;
1232
1233         if (ieee80211_channel_switch(sdata->local->hw.wiphy, sdata->dev,
1234                                      &params) < 0)
1235                 return false;
1236
1237         return true;
1238 }
1239
1240 static void
1241 ieee80211_mesh_rx_probe_req(struct ieee80211_sub_if_data *sdata,
1242                             struct ieee80211_mgmt *mgmt, size_t len)
1243 {
1244         struct ieee80211_local *local = sdata->local;
1245         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1246         struct sk_buff *presp;
1247         struct beacon_data *bcn;
1248         struct ieee80211_mgmt *hdr;
1249         struct ieee802_11_elems elems;
1250         size_t baselen;
1251         u8 *pos;
1252
1253         pos = mgmt->u.probe_req.variable;
1254         baselen = (u8 *) pos - (u8 *) mgmt;
1255         if (baselen > len)
1256                 return;
1257
1258         ieee802_11_parse_elems(pos, len - baselen, false, &elems, mgmt->bssid,
1259                                NULL);
1260
1261         if (!elems.mesh_id)
1262                 return;
1263
1264         /* 802.11-2012 10.1.4.3.2 */
1265         if ((!ether_addr_equal(mgmt->da, sdata->vif.addr) &&
1266              !is_broadcast_ether_addr(mgmt->da)) ||
1267             elems.ssid_len != 0)
1268                 return;
1269
1270         if (elems.mesh_id_len != 0 &&
1271             (elems.mesh_id_len != ifmsh->mesh_id_len ||
1272              memcmp(elems.mesh_id, ifmsh->mesh_id, ifmsh->mesh_id_len)))
1273                 return;
1274
1275         rcu_read_lock();
1276         bcn = rcu_dereference(ifmsh->beacon);
1277
1278         if (!bcn)
1279                 goto out;
1280
1281         presp = dev_alloc_skb(local->tx_headroom +
1282                               bcn->head_len + bcn->tail_len);
1283         if (!presp)
1284                 goto out;
1285
1286         skb_reserve(presp, local->tx_headroom);
1287         skb_put_data(presp, bcn->head, bcn->head_len);
1288         skb_put_data(presp, bcn->tail, bcn->tail_len);
1289         hdr = (struct ieee80211_mgmt *) presp->data;
1290         hdr->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
1291                                          IEEE80211_STYPE_PROBE_RESP);
1292         memcpy(hdr->da, mgmt->sa, ETH_ALEN);
1293         IEEE80211_SKB_CB(presp)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
1294         ieee80211_tx_skb(sdata, presp);
1295 out:
1296         rcu_read_unlock();
1297 }
1298
1299 static void ieee80211_mesh_rx_bcn_presp(struct ieee80211_sub_if_data *sdata,
1300                                         u16 stype,
1301                                         struct ieee80211_mgmt *mgmt,
1302                                         size_t len,
1303                                         struct ieee80211_rx_status *rx_status)
1304 {
1305         struct ieee80211_local *local = sdata->local;
1306         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1307         struct ieee802_11_elems elems;
1308         struct ieee80211_channel *channel;
1309         size_t baselen;
1310         int freq;
1311         enum nl80211_band band = rx_status->band;
1312
1313         /* ignore ProbeResp to foreign address */
1314         if (stype == IEEE80211_STYPE_PROBE_RESP &&
1315             !ether_addr_equal(mgmt->da, sdata->vif.addr))
1316                 return;
1317
1318         baselen = (u8 *) mgmt->u.probe_resp.variable - (u8 *) mgmt;
1319         if (baselen > len)
1320                 return;
1321
1322         ieee802_11_parse_elems(mgmt->u.probe_resp.variable, len - baselen,
1323                                false, &elems, mgmt->bssid, NULL);
1324
1325         /* ignore non-mesh or secure / unsecure mismatch */
1326         if ((!elems.mesh_id || !elems.mesh_config) ||
1327             (elems.rsn && sdata->u.mesh.security == IEEE80211_MESH_SEC_NONE) ||
1328             (!elems.rsn && sdata->u.mesh.security != IEEE80211_MESH_SEC_NONE))
1329                 return;
1330
1331         if (elems.ds_params)
1332                 freq = ieee80211_channel_to_frequency(elems.ds_params[0], band);
1333         else
1334                 freq = rx_status->freq;
1335
1336         channel = ieee80211_get_channel(local->hw.wiphy, freq);
1337
1338         if (!channel || channel->flags & IEEE80211_CHAN_DISABLED)
1339                 return;
1340
1341         if (mesh_matches_local(sdata, &elems)) {
1342                 mpl_dbg(sdata, "rssi_threshold=%d,rx_status->signal=%d\n",
1343                         sdata->u.mesh.mshcfg.rssi_threshold, rx_status->signal);
1344                 if (!sdata->u.mesh.user_mpm ||
1345                     sdata->u.mesh.mshcfg.rssi_threshold == 0 ||
1346                     sdata->u.mesh.mshcfg.rssi_threshold < rx_status->signal)
1347                         mesh_neighbour_update(sdata, mgmt->sa, &elems,
1348                                               rx_status);
1349
1350                 if (ifmsh->csa_role != IEEE80211_MESH_CSA_ROLE_INIT &&
1351                     !sdata->vif.csa_active)
1352                         ieee80211_mesh_process_chnswitch(sdata, &elems, true);
1353         }
1354
1355         if (ifmsh->sync_ops)
1356                 ifmsh->sync_ops->rx_bcn_presp(sdata,
1357                         stype, mgmt, &elems, rx_status);
1358 }
1359
1360 int ieee80211_mesh_finish_csa(struct ieee80211_sub_if_data *sdata)
1361 {
1362         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1363         struct mesh_csa_settings *tmp_csa_settings;
1364         int ret = 0;
1365         int changed = 0;
1366
1367         /* Reset the TTL value and Initiator flag */
1368         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_NONE;
1369         ifmsh->chsw_ttl = 0;
1370
1371         /* Remove the CSA and MCSP elements from the beacon */
1372         tmp_csa_settings = rcu_dereference_protected(ifmsh->csa,
1373                                             lockdep_is_held(&sdata->wdev.mtx));
1374         RCU_INIT_POINTER(ifmsh->csa, NULL);
1375         if (tmp_csa_settings)
1376                 kfree_rcu(tmp_csa_settings, rcu_head);
1377         ret = ieee80211_mesh_rebuild_beacon(sdata);
1378         if (ret)
1379                 return -EINVAL;
1380
1381         changed |= BSS_CHANGED_BEACON;
1382
1383         mcsa_dbg(sdata, "complete switching to center freq %d MHz",
1384                  sdata->vif.bss_conf.chandef.chan->center_freq);
1385         return changed;
1386 }
1387
1388 int ieee80211_mesh_csa_beacon(struct ieee80211_sub_if_data *sdata,
1389                               struct cfg80211_csa_settings *csa_settings)
1390 {
1391         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1392         struct mesh_csa_settings *tmp_csa_settings;
1393         int ret = 0;
1394
1395         lockdep_assert_held(&sdata->wdev.mtx);
1396
1397         tmp_csa_settings = kmalloc(sizeof(*tmp_csa_settings),
1398                                    GFP_ATOMIC);
1399         if (!tmp_csa_settings)
1400                 return -ENOMEM;
1401
1402         memcpy(&tmp_csa_settings->settings, csa_settings,
1403                sizeof(struct cfg80211_csa_settings));
1404
1405         rcu_assign_pointer(ifmsh->csa, tmp_csa_settings);
1406
1407         ret = ieee80211_mesh_rebuild_beacon(sdata);
1408         if (ret) {
1409                 tmp_csa_settings = rcu_dereference(ifmsh->csa);
1410                 RCU_INIT_POINTER(ifmsh->csa, NULL);
1411                 kfree_rcu(tmp_csa_settings, rcu_head);
1412                 return ret;
1413         }
1414
1415         return BSS_CHANGED_BEACON;
1416 }
1417
1418 static int mesh_fwd_csa_frame(struct ieee80211_sub_if_data *sdata,
1419                                struct ieee80211_mgmt *mgmt, size_t len,
1420                                struct ieee802_11_elems *elems)
1421 {
1422         struct ieee80211_mgmt *mgmt_fwd;
1423         struct sk_buff *skb;
1424         struct ieee80211_local *local = sdata->local;
1425
1426         skb = dev_alloc_skb(local->tx_headroom + len);
1427         if (!skb)
1428                 return -ENOMEM;
1429         skb_reserve(skb, local->tx_headroom);
1430         mgmt_fwd = skb_put(skb, len);
1431
1432         elems->mesh_chansw_params_ie->mesh_ttl--;
1433         elems->mesh_chansw_params_ie->mesh_flags &=
1434                 ~WLAN_EID_CHAN_SWITCH_PARAM_INITIATOR;
1435
1436         memcpy(mgmt_fwd, mgmt, len);
1437         eth_broadcast_addr(mgmt_fwd->da);
1438         memcpy(mgmt_fwd->sa, sdata->vif.addr, ETH_ALEN);
1439         memcpy(mgmt_fwd->bssid, sdata->vif.addr, ETH_ALEN);
1440
1441         ieee80211_tx_skb(sdata, skb);
1442         return 0;
1443 }
1444
1445 static void mesh_rx_csa_frame(struct ieee80211_sub_if_data *sdata,
1446                               struct ieee80211_mgmt *mgmt, size_t len)
1447 {
1448         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1449         struct ieee802_11_elems elems;
1450         u16 pre_value;
1451         bool fwd_csa = true;
1452         size_t baselen;
1453         u8 *pos;
1454
1455         if (mgmt->u.action.u.measurement.action_code !=
1456             WLAN_ACTION_SPCT_CHL_SWITCH)
1457                 return;
1458
1459         pos = mgmt->u.action.u.chan_switch.variable;
1460         baselen = offsetof(struct ieee80211_mgmt,
1461                            u.action.u.chan_switch.variable);
1462         ieee802_11_parse_elems(pos, len - baselen, true, &elems,
1463                                mgmt->bssid, NULL);
1464
1465         if (!mesh_matches_local(sdata, &elems))
1466                 return;
1467
1468         ifmsh->chsw_ttl = elems.mesh_chansw_params_ie->mesh_ttl;
1469         if (!--ifmsh->chsw_ttl)
1470                 fwd_csa = false;
1471
1472         pre_value = le16_to_cpu(elems.mesh_chansw_params_ie->mesh_pre_value);
1473         if (ifmsh->pre_value >= pre_value)
1474                 return;
1475
1476         ifmsh->pre_value = pre_value;
1477
1478         if (!sdata->vif.csa_active &&
1479             !ieee80211_mesh_process_chnswitch(sdata, &elems, false)) {
1480                 mcsa_dbg(sdata, "Failed to process CSA action frame");
1481                 return;
1482         }
1483
1484         /* forward or re-broadcast the CSA frame */
1485         if (fwd_csa) {
1486                 if (mesh_fwd_csa_frame(sdata, mgmt, len, &elems) < 0)
1487                         mcsa_dbg(sdata, "Failed to forward the CSA frame");
1488         }
1489 }
1490
1491 static void ieee80211_mesh_rx_mgmt_action(struct ieee80211_sub_if_data *sdata,
1492                                           struct ieee80211_mgmt *mgmt,
1493                                           size_t len,
1494                                           struct ieee80211_rx_status *rx_status)
1495 {
1496         switch (mgmt->u.action.category) {
1497         case WLAN_CATEGORY_SELF_PROTECTED:
1498                 switch (mgmt->u.action.u.self_prot.action_code) {
1499                 case WLAN_SP_MESH_PEERING_OPEN:
1500                 case WLAN_SP_MESH_PEERING_CLOSE:
1501                 case WLAN_SP_MESH_PEERING_CONFIRM:
1502                         mesh_rx_plink_frame(sdata, mgmt, len, rx_status);
1503                         break;
1504                 }
1505                 break;
1506         case WLAN_CATEGORY_MESH_ACTION:
1507                 if (mesh_action_is_path_sel(mgmt))
1508                         mesh_rx_path_sel_frame(sdata, mgmt, len);
1509                 break;
1510         case WLAN_CATEGORY_SPECTRUM_MGMT:
1511                 mesh_rx_csa_frame(sdata, mgmt, len);
1512                 break;
1513         }
1514 }
1515
1516 void ieee80211_mesh_rx_queued_mgmt(struct ieee80211_sub_if_data *sdata,
1517                                    struct sk_buff *skb)
1518 {
1519         struct ieee80211_rx_status *rx_status;
1520         struct ieee80211_mgmt *mgmt;
1521         u16 stype;
1522
1523         sdata_lock(sdata);
1524
1525         /* mesh already went down */
1526         if (!sdata->u.mesh.mesh_id_len)
1527                 goto out;
1528
1529         rx_status = IEEE80211_SKB_RXCB(skb);
1530         mgmt = (struct ieee80211_mgmt *) skb->data;
1531         stype = le16_to_cpu(mgmt->frame_control) & IEEE80211_FCTL_STYPE;
1532
1533         switch (stype) {
1534         case IEEE80211_STYPE_PROBE_RESP:
1535         case IEEE80211_STYPE_BEACON:
1536                 ieee80211_mesh_rx_bcn_presp(sdata, stype, mgmt, skb->len,
1537                                             rx_status);
1538                 break;
1539         case IEEE80211_STYPE_PROBE_REQ:
1540                 ieee80211_mesh_rx_probe_req(sdata, mgmt, skb->len);
1541                 break;
1542         case IEEE80211_STYPE_ACTION:
1543                 ieee80211_mesh_rx_mgmt_action(sdata, mgmt, skb->len, rx_status);
1544                 break;
1545         }
1546 out:
1547         sdata_unlock(sdata);
1548 }
1549
1550 static void mesh_bss_info_changed(struct ieee80211_sub_if_data *sdata)
1551 {
1552         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1553         u32 bit, changed = 0;
1554
1555         for_each_set_bit(bit, &ifmsh->mbss_changed,
1556                          sizeof(changed) * BITS_PER_BYTE) {
1557                 clear_bit(bit, &ifmsh->mbss_changed);
1558                 changed |= BIT(bit);
1559         }
1560
1561         if (sdata->vif.bss_conf.enable_beacon &&
1562             (changed & (BSS_CHANGED_BEACON |
1563                         BSS_CHANGED_HT |
1564                         BSS_CHANGED_BASIC_RATES |
1565                         BSS_CHANGED_BEACON_INT)))
1566                 if (ieee80211_mesh_rebuild_beacon(sdata))
1567                         return;
1568
1569         ieee80211_bss_info_change_notify(sdata, changed);
1570 }
1571
1572 void ieee80211_mesh_work(struct ieee80211_sub_if_data *sdata)
1573 {
1574         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1575
1576         sdata_lock(sdata);
1577
1578         /* mesh already went down */
1579         if (!sdata->u.mesh.mesh_id_len)
1580                 goto out;
1581
1582         if (ifmsh->preq_queue_len &&
1583             time_after(jiffies,
1584                        ifmsh->last_preq + msecs_to_jiffies(ifmsh->mshcfg.dot11MeshHWMPpreqMinInterval)))
1585                 mesh_path_start_discovery(sdata);
1586
1587         if (test_and_clear_bit(MESH_WORK_HOUSEKEEPING, &ifmsh->wrkq_flags))
1588                 ieee80211_mesh_housekeeping(sdata);
1589
1590         if (test_and_clear_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags))
1591                 ieee80211_mesh_rootpath(sdata);
1592
1593         if (test_and_clear_bit(MESH_WORK_DRIFT_ADJUST, &ifmsh->wrkq_flags))
1594                 mesh_sync_adjust_tsf(sdata);
1595
1596         if (test_and_clear_bit(MESH_WORK_MBSS_CHANGED, &ifmsh->wrkq_flags))
1597                 mesh_bss_info_changed(sdata);
1598 out:
1599         sdata_unlock(sdata);
1600 }
1601
1602
1603 void ieee80211_mesh_init_sdata(struct ieee80211_sub_if_data *sdata)
1604 {
1605         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1606         static u8 zero_addr[ETH_ALEN] = {};
1607
1608         timer_setup(&ifmsh->housekeeping_timer,
1609                     ieee80211_mesh_housekeeping_timer, 0);
1610
1611         ifmsh->accepting_plinks = true;
1612         atomic_set(&ifmsh->mpaths, 0);
1613         mesh_rmc_init(sdata);
1614         ifmsh->last_preq = jiffies;
1615         ifmsh->next_perr = jiffies;
1616         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_NONE;
1617         /* Allocate all mesh structures when creating the first mesh interface. */
1618         if (!mesh_allocated)
1619                 ieee80211s_init();
1620
1621         mesh_pathtbl_init(sdata);
1622
1623         timer_setup(&ifmsh->mesh_path_timer, ieee80211_mesh_path_timer, 0);
1624         timer_setup(&ifmsh->mesh_path_root_timer,
1625                     ieee80211_mesh_path_root_timer, 0);
1626         INIT_LIST_HEAD(&ifmsh->preq_queue.list);
1627         skb_queue_head_init(&ifmsh->ps.bc_buf);
1628         spin_lock_init(&ifmsh->mesh_preq_queue_lock);
1629         spin_lock_init(&ifmsh->sync_offset_lock);
1630         RCU_INIT_POINTER(ifmsh->beacon, NULL);
1631
1632         sdata->vif.bss_conf.bssid = zero_addr;
1633 }
1634
1635 void ieee80211_mesh_teardown_sdata(struct ieee80211_sub_if_data *sdata)
1636 {
1637         mesh_rmc_free(sdata);
1638         mesh_pathtbl_unregister(sdata);
1639 }