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