Merge tag 'arm-dt-6.0' of git://git.kernel.org/pub/scm/linux/kernel/git/soc/soc
[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 - 2021 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.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.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.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.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 = rcu_dereference_protected(sdata->u.mesh.beacon,
997                                             lockdep_is_held(&sdata->wdev.mtx));
998         ret = ieee80211_mesh_build_beacon(&sdata->u.mesh);
999         if (ret)
1000                 /* just reuse old beacon */
1001                 return ret;
1002
1003         if (old_bcn)
1004                 kfree_rcu(old_bcn, rcu_head);
1005         return 0;
1006 }
1007
1008 void ieee80211_mbss_info_change_notify(struct ieee80211_sub_if_data *sdata,
1009                                        u32 changed)
1010 {
1011         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1012         unsigned long bits = changed;
1013         u32 bit;
1014
1015         if (!bits)
1016                 return;
1017
1018         /* if we race with running work, worst case this work becomes a noop */
1019         for_each_set_bit(bit, &bits, sizeof(changed) * BITS_PER_BYTE)
1020                 set_bit(bit, &ifmsh->mbss_changed);
1021         set_bit(MESH_WORK_MBSS_CHANGED, &ifmsh->wrkq_flags);
1022         ieee80211_queue_work(&sdata->local->hw, &sdata->work);
1023 }
1024
1025 int ieee80211_start_mesh(struct ieee80211_sub_if_data *sdata)
1026 {
1027         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1028         struct ieee80211_local *local = sdata->local;
1029         u32 changed = BSS_CHANGED_BEACON |
1030                       BSS_CHANGED_BEACON_ENABLED |
1031                       BSS_CHANGED_HT |
1032                       BSS_CHANGED_BASIC_RATES |
1033                       BSS_CHANGED_BEACON_INT |
1034                       BSS_CHANGED_MCAST_RATE;
1035
1036         local->fif_other_bss++;
1037         /* mesh ifaces must set allmulti to forward mcast traffic */
1038         atomic_inc(&local->iff_allmultis);
1039         ieee80211_configure_filter(local);
1040
1041         ifmsh->mesh_cc_id = 0;  /* Disabled */
1042         /* register sync ops from extensible synchronization framework */
1043         ifmsh->sync_ops = ieee80211_mesh_sync_ops_get(ifmsh->mesh_sp_id);
1044         ifmsh->sync_offset_clockdrift_max = 0;
1045         set_bit(MESH_WORK_HOUSEKEEPING, &ifmsh->wrkq_flags);
1046         ieee80211_mesh_root_setup(ifmsh);
1047         ieee80211_queue_work(&local->hw, &sdata->work);
1048         sdata->vif.bss_conf.ht_operation_mode =
1049                                 ifmsh->mshcfg.ht_opmode;
1050         sdata->vif.bss_conf.enable_beacon = true;
1051
1052         changed |= ieee80211_mps_local_status_update(sdata);
1053
1054         if (ieee80211_mesh_build_beacon(ifmsh)) {
1055                 ieee80211_stop_mesh(sdata);
1056                 return -ENOMEM;
1057         }
1058
1059         ieee80211_recalc_dtim(local, sdata);
1060         ieee80211_bss_info_change_notify(sdata, changed);
1061
1062         netif_carrier_on(sdata->dev);
1063         return 0;
1064 }
1065
1066 void ieee80211_stop_mesh(struct ieee80211_sub_if_data *sdata)
1067 {
1068         struct ieee80211_local *local = sdata->local;
1069         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1070         struct beacon_data *bcn;
1071
1072         netif_carrier_off(sdata->dev);
1073
1074         /* flush STAs and mpaths on this iface */
1075         sta_info_flush(sdata);
1076         ieee80211_free_keys(sdata, true);
1077         mesh_path_flush_by_iface(sdata);
1078
1079         /* stop the beacon */
1080         ifmsh->mesh_id_len = 0;
1081         sdata->vif.bss_conf.enable_beacon = false;
1082         sdata->beacon_rate_set = false;
1083         clear_bit(SDATA_STATE_OFFCHANNEL_BEACON_STOPPED, &sdata->state);
1084         ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_BEACON_ENABLED);
1085
1086         /* remove beacon */
1087         bcn = rcu_dereference_protected(ifmsh->beacon,
1088                                         lockdep_is_held(&sdata->wdev.mtx));
1089         RCU_INIT_POINTER(ifmsh->beacon, NULL);
1090         kfree_rcu(bcn, rcu_head);
1091
1092         /* free all potentially still buffered group-addressed frames */
1093         local->total_ps_buffered -= skb_queue_len(&ifmsh->ps.bc_buf);
1094         skb_queue_purge(&ifmsh->ps.bc_buf);
1095
1096         del_timer_sync(&sdata->u.mesh.housekeeping_timer);
1097         del_timer_sync(&sdata->u.mesh.mesh_path_root_timer);
1098         del_timer_sync(&sdata->u.mesh.mesh_path_timer);
1099
1100         /* clear any mesh work (for next join) we may have accrued */
1101         ifmsh->wrkq_flags = 0;
1102         ifmsh->mbss_changed = 0;
1103
1104         local->fif_other_bss--;
1105         atomic_dec(&local->iff_allmultis);
1106         ieee80211_configure_filter(local);
1107 }
1108
1109 static void ieee80211_mesh_csa_mark_radar(struct ieee80211_sub_if_data *sdata)
1110 {
1111         int err;
1112
1113         /* if the current channel is a DFS channel, mark the channel as
1114          * unavailable.
1115          */
1116         err = cfg80211_chandef_dfs_required(sdata->local->hw.wiphy,
1117                                             &sdata->vif.bss_conf.chandef,
1118                                             NL80211_IFTYPE_MESH_POINT);
1119         if (err > 0)
1120                 cfg80211_radar_event(sdata->local->hw.wiphy,
1121                                      &sdata->vif.bss_conf.chandef, GFP_ATOMIC);
1122 }
1123
1124 static bool
1125 ieee80211_mesh_process_chnswitch(struct ieee80211_sub_if_data *sdata,
1126                                  struct ieee802_11_elems *elems, bool beacon)
1127 {
1128         struct cfg80211_csa_settings params;
1129         struct ieee80211_csa_ie csa_ie;
1130         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1131         struct ieee80211_supported_band *sband;
1132         int err;
1133         u32 sta_flags, 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         sta_flags = 0;
1142         switch (sdata->vif.bss_conf.chandef.width) {
1143         case NL80211_CHAN_WIDTH_20_NOHT:
1144                 sta_flags |= IEEE80211_STA_DISABLE_HT;
1145                 fallthrough;
1146         case NL80211_CHAN_WIDTH_20:
1147                 sta_flags |= IEEE80211_STA_DISABLE_40MHZ;
1148                 fallthrough;
1149         case NL80211_CHAN_WIDTH_40:
1150                 sta_flags |= IEEE80211_STA_DISABLE_VHT;
1151                 break;
1152         default:
1153                 break;
1154         }
1155
1156         if (elems->vht_cap_elem)
1157                 vht_cap_info =
1158                         le32_to_cpu(elems->vht_cap_elem->vht_cap_info);
1159
1160         memset(&params, 0, sizeof(params));
1161         err = ieee80211_parse_ch_switch_ie(sdata, elems, sband->band,
1162                                            vht_cap_info,
1163                                            sta_flags, sdata->vif.addr,
1164                                            &csa_ie);
1165         if (err < 0)
1166                 return false;
1167         if (err)
1168                 return false;
1169
1170         /* Mark the channel unavailable if the reason for the switch is
1171          * regulatory.
1172          */
1173         if (csa_ie.reason_code == WLAN_REASON_MESH_CHAN_REGULATORY)
1174                 ieee80211_mesh_csa_mark_radar(sdata);
1175
1176         params.chandef = csa_ie.chandef;
1177         params.count = csa_ie.count;
1178
1179         if (!cfg80211_chandef_usable(sdata->local->hw.wiphy, &params.chandef,
1180                                      IEEE80211_CHAN_DISABLED) ||
1181             !cfg80211_reg_can_beacon(sdata->local->hw.wiphy, &params.chandef,
1182                                      NL80211_IFTYPE_MESH_POINT)) {
1183                 sdata_info(sdata,
1184                            "mesh STA %pM switches to unsupported channel (%d MHz, width:%d, CF1/2: %d/%d MHz), aborting\n",
1185                            sdata->vif.addr,
1186                            params.chandef.chan->center_freq,
1187                            params.chandef.width,
1188                            params.chandef.center_freq1,
1189                            params.chandef.center_freq2);
1190                 return false;
1191         }
1192
1193         err = cfg80211_chandef_dfs_required(sdata->local->hw.wiphy,
1194                                             &params.chandef,
1195                                             NL80211_IFTYPE_MESH_POINT);
1196         if (err < 0)
1197                 return false;
1198         if (err > 0 && !ifmsh->userspace_handles_dfs) {
1199                 sdata_info(sdata,
1200                            "mesh STA %pM switches to channel requiring DFS (%d MHz, width:%d, CF1/2: %d/%d MHz), aborting\n",
1201                            sdata->vif.addr,
1202                            params.chandef.chan->center_freq,
1203                            params.chandef.width,
1204                            params.chandef.center_freq1,
1205                            params.chandef.center_freq2);
1206                 return false;
1207         }
1208
1209         params.radar_required = err;
1210
1211         if (cfg80211_chandef_identical(&params.chandef,
1212                                        &sdata->vif.bss_conf.chandef)) {
1213                 mcsa_dbg(sdata,
1214                          "received csa with an identical chandef, ignoring\n");
1215                 return true;
1216         }
1217
1218         mcsa_dbg(sdata,
1219                  "received channel switch announcement to go to channel %d MHz\n",
1220                  params.chandef.chan->center_freq);
1221
1222         params.block_tx = csa_ie.mode & WLAN_EID_CHAN_SWITCH_PARAM_TX_RESTRICT;
1223         if (beacon) {
1224                 ifmsh->chsw_ttl = csa_ie.ttl - 1;
1225                 if (ifmsh->pre_value >= csa_ie.pre_value)
1226                         return false;
1227                 ifmsh->pre_value = csa_ie.pre_value;
1228         }
1229
1230         if (ifmsh->chsw_ttl >= ifmsh->mshcfg.dot11MeshTTL)
1231                 return false;
1232
1233         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_REPEATER;
1234
1235         if (ieee80211_channel_switch(sdata->local->hw.wiphy, sdata->dev,
1236                                      &params) < 0)
1237                 return false;
1238
1239         return true;
1240 }
1241
1242 static void
1243 ieee80211_mesh_rx_probe_req(struct ieee80211_sub_if_data *sdata,
1244                             struct ieee80211_mgmt *mgmt, size_t len)
1245 {
1246         struct ieee80211_local *local = sdata->local;
1247         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1248         struct sk_buff *presp;
1249         struct beacon_data *bcn;
1250         struct ieee80211_mgmt *hdr;
1251         struct ieee802_11_elems *elems;
1252         size_t baselen;
1253         u8 *pos;
1254
1255         pos = mgmt->u.probe_req.variable;
1256         baselen = (u8 *) pos - (u8 *) mgmt;
1257         if (baselen > len)
1258                 return;
1259
1260         elems = ieee802_11_parse_elems(pos, len - baselen, false, mgmt->bssid,
1261                                        NULL);
1262         if (!elems)
1263                 return;
1264
1265         if (!elems->mesh_id)
1266                 goto free;
1267
1268         /* 802.11-2012 10.1.4.3.2 */
1269         if ((!ether_addr_equal(mgmt->da, sdata->vif.addr) &&
1270              !is_broadcast_ether_addr(mgmt->da)) ||
1271             elems->ssid_len != 0)
1272                 goto free;
1273
1274         if (elems->mesh_id_len != 0 &&
1275             (elems->mesh_id_len != ifmsh->mesh_id_len ||
1276              memcmp(elems->mesh_id, ifmsh->mesh_id, ifmsh->mesh_id_len)))
1277                 goto free;
1278
1279         rcu_read_lock();
1280         bcn = rcu_dereference(ifmsh->beacon);
1281
1282         if (!bcn)
1283                 goto out;
1284
1285         presp = dev_alloc_skb(local->tx_headroom +
1286                               bcn->head_len + bcn->tail_len);
1287         if (!presp)
1288                 goto out;
1289
1290         skb_reserve(presp, local->tx_headroom);
1291         skb_put_data(presp, bcn->head, bcn->head_len);
1292         skb_put_data(presp, bcn->tail, bcn->tail_len);
1293         hdr = (struct ieee80211_mgmt *) presp->data;
1294         hdr->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
1295                                          IEEE80211_STYPE_PROBE_RESP);
1296         memcpy(hdr->da, mgmt->sa, ETH_ALEN);
1297         IEEE80211_SKB_CB(presp)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
1298         ieee80211_tx_skb(sdata, presp);
1299 out:
1300         rcu_read_unlock();
1301 free:
1302         kfree(elems);
1303 }
1304
1305 static void ieee80211_mesh_rx_bcn_presp(struct ieee80211_sub_if_data *sdata,
1306                                         u16 stype,
1307                                         struct ieee80211_mgmt *mgmt,
1308                                         size_t len,
1309                                         struct ieee80211_rx_status *rx_status)
1310 {
1311         struct ieee80211_local *local = sdata->local;
1312         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1313         struct ieee802_11_elems *elems;
1314         struct ieee80211_channel *channel;
1315         size_t baselen;
1316         int freq;
1317         enum nl80211_band band = rx_status->band;
1318
1319         /* ignore ProbeResp to foreign address */
1320         if (stype == IEEE80211_STYPE_PROBE_RESP &&
1321             !ether_addr_equal(mgmt->da, sdata->vif.addr))
1322                 return;
1323
1324         baselen = (u8 *) mgmt->u.probe_resp.variable - (u8 *) mgmt;
1325         if (baselen > len)
1326                 return;
1327
1328         elems = ieee802_11_parse_elems(mgmt->u.probe_resp.variable,
1329                                        len - baselen,
1330                                        false, mgmt->bssid, NULL);
1331         if (!elems)
1332                 return;
1333
1334         /* ignore non-mesh or secure / unsecure mismatch */
1335         if ((!elems->mesh_id || !elems->mesh_config) ||
1336             (elems->rsn && sdata->u.mesh.security == IEEE80211_MESH_SEC_NONE) ||
1337             (!elems->rsn && sdata->u.mesh.security != IEEE80211_MESH_SEC_NONE))
1338                 goto free;
1339
1340         if (elems->ds_params)
1341                 freq = ieee80211_channel_to_frequency(elems->ds_params[0], band);
1342         else
1343                 freq = rx_status->freq;
1344
1345         channel = ieee80211_get_channel(local->hw.wiphy, freq);
1346
1347         if (!channel || channel->flags & IEEE80211_CHAN_DISABLED)
1348                 goto free;
1349
1350         if (mesh_matches_local(sdata, elems)) {
1351                 mpl_dbg(sdata, "rssi_threshold=%d,rx_status->signal=%d\n",
1352                         sdata->u.mesh.mshcfg.rssi_threshold, rx_status->signal);
1353                 if (!sdata->u.mesh.user_mpm ||
1354                     sdata->u.mesh.mshcfg.rssi_threshold == 0 ||
1355                     sdata->u.mesh.mshcfg.rssi_threshold < rx_status->signal)
1356                         mesh_neighbour_update(sdata, mgmt->sa, elems,
1357                                               rx_status);
1358
1359                 if (ifmsh->csa_role != IEEE80211_MESH_CSA_ROLE_INIT &&
1360                     !sdata->vif.csa_active)
1361                         ieee80211_mesh_process_chnswitch(sdata, elems, true);
1362         }
1363
1364         if (ifmsh->sync_ops)
1365                 ifmsh->sync_ops->rx_bcn_presp(sdata, stype, mgmt, len,
1366                                               elems->mesh_config, rx_status);
1367 free:
1368         kfree(elems);
1369 }
1370
1371 int ieee80211_mesh_finish_csa(struct ieee80211_sub_if_data *sdata)
1372 {
1373         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1374         struct mesh_csa_settings *tmp_csa_settings;
1375         int ret = 0;
1376         int changed = 0;
1377
1378         /* Reset the TTL value and Initiator flag */
1379         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_NONE;
1380         ifmsh->chsw_ttl = 0;
1381
1382         /* Remove the CSA and MCSP elements from the beacon */
1383         tmp_csa_settings = rcu_dereference_protected(ifmsh->csa,
1384                                             lockdep_is_held(&sdata->wdev.mtx));
1385         RCU_INIT_POINTER(ifmsh->csa, NULL);
1386         if (tmp_csa_settings)
1387                 kfree_rcu(tmp_csa_settings, rcu_head);
1388         ret = ieee80211_mesh_rebuild_beacon(sdata);
1389         if (ret)
1390                 return -EINVAL;
1391
1392         changed |= BSS_CHANGED_BEACON;
1393
1394         mcsa_dbg(sdata, "complete switching to center freq %d MHz",
1395                  sdata->vif.bss_conf.chandef.chan->center_freq);
1396         return changed;
1397 }
1398
1399 int ieee80211_mesh_csa_beacon(struct ieee80211_sub_if_data *sdata,
1400                               struct cfg80211_csa_settings *csa_settings)
1401 {
1402         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1403         struct mesh_csa_settings *tmp_csa_settings;
1404         int ret = 0;
1405
1406         lockdep_assert_held(&sdata->wdev.mtx);
1407
1408         tmp_csa_settings = kmalloc(sizeof(*tmp_csa_settings),
1409                                    GFP_ATOMIC);
1410         if (!tmp_csa_settings)
1411                 return -ENOMEM;
1412
1413         memcpy(&tmp_csa_settings->settings, csa_settings,
1414                sizeof(struct cfg80211_csa_settings));
1415
1416         rcu_assign_pointer(ifmsh->csa, tmp_csa_settings);
1417
1418         ret = ieee80211_mesh_rebuild_beacon(sdata);
1419         if (ret) {
1420                 tmp_csa_settings = rcu_dereference(ifmsh->csa);
1421                 RCU_INIT_POINTER(ifmsh->csa, NULL);
1422                 kfree_rcu(tmp_csa_settings, rcu_head);
1423                 return ret;
1424         }
1425
1426         return BSS_CHANGED_BEACON;
1427 }
1428
1429 static int mesh_fwd_csa_frame(struct ieee80211_sub_if_data *sdata,
1430                                struct ieee80211_mgmt *mgmt, size_t len,
1431                                struct ieee802_11_elems *elems)
1432 {
1433         struct ieee80211_mgmt *mgmt_fwd;
1434         struct sk_buff *skb;
1435         struct ieee80211_local *local = sdata->local;
1436
1437         skb = dev_alloc_skb(local->tx_headroom + len);
1438         if (!skb)
1439                 return -ENOMEM;
1440         skb_reserve(skb, local->tx_headroom);
1441         mgmt_fwd = skb_put(skb, len);
1442
1443         elems->mesh_chansw_params_ie->mesh_ttl--;
1444         elems->mesh_chansw_params_ie->mesh_flags &=
1445                 ~WLAN_EID_CHAN_SWITCH_PARAM_INITIATOR;
1446
1447         memcpy(mgmt_fwd, mgmt, len);
1448         eth_broadcast_addr(mgmt_fwd->da);
1449         memcpy(mgmt_fwd->sa, sdata->vif.addr, ETH_ALEN);
1450         memcpy(mgmt_fwd->bssid, sdata->vif.addr, ETH_ALEN);
1451
1452         ieee80211_tx_skb(sdata, skb);
1453         return 0;
1454 }
1455
1456 static void mesh_rx_csa_frame(struct ieee80211_sub_if_data *sdata,
1457                               struct ieee80211_mgmt *mgmt, size_t len)
1458 {
1459         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1460         struct ieee802_11_elems *elems;
1461         u16 pre_value;
1462         bool fwd_csa = true;
1463         size_t baselen;
1464         u8 *pos;
1465
1466         if (mgmt->u.action.u.measurement.action_code !=
1467             WLAN_ACTION_SPCT_CHL_SWITCH)
1468                 return;
1469
1470         pos = mgmt->u.action.u.chan_switch.variable;
1471         baselen = offsetof(struct ieee80211_mgmt,
1472                            u.action.u.chan_switch.variable);
1473         elems = ieee802_11_parse_elems(pos, len - baselen, true,
1474                                        mgmt->bssid, NULL);
1475         if (!elems)
1476                 return;
1477
1478         if (!mesh_matches_local(sdata, elems))
1479                 goto free;
1480
1481         ifmsh->chsw_ttl = elems->mesh_chansw_params_ie->mesh_ttl;
1482         if (!--ifmsh->chsw_ttl)
1483                 fwd_csa = false;
1484
1485         pre_value = le16_to_cpu(elems->mesh_chansw_params_ie->mesh_pre_value);
1486         if (ifmsh->pre_value >= pre_value)
1487                 goto free;
1488
1489         ifmsh->pre_value = pre_value;
1490
1491         if (!sdata->vif.csa_active &&
1492             !ieee80211_mesh_process_chnswitch(sdata, elems, false)) {
1493                 mcsa_dbg(sdata, "Failed to process CSA action frame");
1494                 goto free;
1495         }
1496
1497         /* forward or re-broadcast the CSA frame */
1498         if (fwd_csa) {
1499                 if (mesh_fwd_csa_frame(sdata, mgmt, len, elems) < 0)
1500                         mcsa_dbg(sdata, "Failed to forward the CSA frame");
1501         }
1502 free:
1503         kfree(elems);
1504 }
1505
1506 static void ieee80211_mesh_rx_mgmt_action(struct ieee80211_sub_if_data *sdata,
1507                                           struct ieee80211_mgmt *mgmt,
1508                                           size_t len,
1509                                           struct ieee80211_rx_status *rx_status)
1510 {
1511         switch (mgmt->u.action.category) {
1512         case WLAN_CATEGORY_SELF_PROTECTED:
1513                 switch (mgmt->u.action.u.self_prot.action_code) {
1514                 case WLAN_SP_MESH_PEERING_OPEN:
1515                 case WLAN_SP_MESH_PEERING_CLOSE:
1516                 case WLAN_SP_MESH_PEERING_CONFIRM:
1517                         mesh_rx_plink_frame(sdata, mgmt, len, rx_status);
1518                         break;
1519                 }
1520                 break;
1521         case WLAN_CATEGORY_MESH_ACTION:
1522                 if (mesh_action_is_path_sel(mgmt))
1523                         mesh_rx_path_sel_frame(sdata, mgmt, len);
1524                 break;
1525         case WLAN_CATEGORY_SPECTRUM_MGMT:
1526                 mesh_rx_csa_frame(sdata, mgmt, len);
1527                 break;
1528         }
1529 }
1530
1531 void ieee80211_mesh_rx_queued_mgmt(struct ieee80211_sub_if_data *sdata,
1532                                    struct sk_buff *skb)
1533 {
1534         struct ieee80211_rx_status *rx_status;
1535         struct ieee80211_mgmt *mgmt;
1536         u16 stype;
1537
1538         sdata_lock(sdata);
1539
1540         /* mesh already went down */
1541         if (!sdata->u.mesh.mesh_id_len)
1542                 goto out;
1543
1544         rx_status = IEEE80211_SKB_RXCB(skb);
1545         mgmt = (struct ieee80211_mgmt *) skb->data;
1546         stype = le16_to_cpu(mgmt->frame_control) & IEEE80211_FCTL_STYPE;
1547
1548         switch (stype) {
1549         case IEEE80211_STYPE_PROBE_RESP:
1550         case IEEE80211_STYPE_BEACON:
1551                 ieee80211_mesh_rx_bcn_presp(sdata, stype, mgmt, skb->len,
1552                                             rx_status);
1553                 break;
1554         case IEEE80211_STYPE_PROBE_REQ:
1555                 ieee80211_mesh_rx_probe_req(sdata, mgmt, skb->len);
1556                 break;
1557         case IEEE80211_STYPE_ACTION:
1558                 ieee80211_mesh_rx_mgmt_action(sdata, mgmt, skb->len, rx_status);
1559                 break;
1560         }
1561 out:
1562         sdata_unlock(sdata);
1563 }
1564
1565 static void mesh_bss_info_changed(struct ieee80211_sub_if_data *sdata)
1566 {
1567         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1568         u32 bit, changed = 0;
1569
1570         for_each_set_bit(bit, &ifmsh->mbss_changed,
1571                          sizeof(changed) * BITS_PER_BYTE) {
1572                 clear_bit(bit, &ifmsh->mbss_changed);
1573                 changed |= BIT(bit);
1574         }
1575
1576         if (sdata->vif.bss_conf.enable_beacon &&
1577             (changed & (BSS_CHANGED_BEACON |
1578                         BSS_CHANGED_HT |
1579                         BSS_CHANGED_BASIC_RATES |
1580                         BSS_CHANGED_BEACON_INT)))
1581                 if (ieee80211_mesh_rebuild_beacon(sdata))
1582                         return;
1583
1584         ieee80211_bss_info_change_notify(sdata, changed);
1585 }
1586
1587 void ieee80211_mesh_work(struct ieee80211_sub_if_data *sdata)
1588 {
1589         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1590
1591         sdata_lock(sdata);
1592
1593         /* mesh already went down */
1594         if (!sdata->u.mesh.mesh_id_len)
1595                 goto out;
1596
1597         if (ifmsh->preq_queue_len &&
1598             time_after(jiffies,
1599                        ifmsh->last_preq + msecs_to_jiffies(ifmsh->mshcfg.dot11MeshHWMPpreqMinInterval)))
1600                 mesh_path_start_discovery(sdata);
1601
1602         if (test_and_clear_bit(MESH_WORK_HOUSEKEEPING, &ifmsh->wrkq_flags))
1603                 ieee80211_mesh_housekeeping(sdata);
1604
1605         if (test_and_clear_bit(MESH_WORK_ROOT, &ifmsh->wrkq_flags))
1606                 ieee80211_mesh_rootpath(sdata);
1607
1608         if (test_and_clear_bit(MESH_WORK_DRIFT_ADJUST, &ifmsh->wrkq_flags))
1609                 mesh_sync_adjust_tsf(sdata);
1610
1611         if (test_and_clear_bit(MESH_WORK_MBSS_CHANGED, &ifmsh->wrkq_flags))
1612                 mesh_bss_info_changed(sdata);
1613 out:
1614         sdata_unlock(sdata);
1615 }
1616
1617
1618 void ieee80211_mesh_init_sdata(struct ieee80211_sub_if_data *sdata)
1619 {
1620         struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
1621         static u8 zero_addr[ETH_ALEN] = {};
1622
1623         timer_setup(&ifmsh->housekeeping_timer,
1624                     ieee80211_mesh_housekeeping_timer, 0);
1625
1626         ifmsh->accepting_plinks = true;
1627         atomic_set(&ifmsh->mpaths, 0);
1628         mesh_rmc_init(sdata);
1629         ifmsh->last_preq = jiffies;
1630         ifmsh->next_perr = jiffies;
1631         ifmsh->csa_role = IEEE80211_MESH_CSA_ROLE_NONE;
1632         /* Allocate all mesh structures when creating the first mesh interface. */
1633         if (!mesh_allocated)
1634                 ieee80211s_init();
1635
1636         mesh_pathtbl_init(sdata);
1637
1638         timer_setup(&ifmsh->mesh_path_timer, ieee80211_mesh_path_timer, 0);
1639         timer_setup(&ifmsh->mesh_path_root_timer,
1640                     ieee80211_mesh_path_root_timer, 0);
1641         INIT_LIST_HEAD(&ifmsh->preq_queue.list);
1642         skb_queue_head_init(&ifmsh->ps.bc_buf);
1643         spin_lock_init(&ifmsh->mesh_preq_queue_lock);
1644         spin_lock_init(&ifmsh->sync_offset_lock);
1645         RCU_INIT_POINTER(ifmsh->beacon, NULL);
1646
1647         sdata->vif.bss_conf.bssid = zero_addr;
1648 }
1649
1650 void ieee80211_mesh_teardown_sdata(struct ieee80211_sub_if_data *sdata)
1651 {
1652         mesh_rmc_free(sdata);
1653         mesh_pathtbl_unregister(sdata);
1654 }