Merge tag '6.6-rc4-ksmbd-server-fixes' of git://git.samba.org/ksmbd
[platform/kernel/linux-starfive.git] / net / wireless / mlme.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * cfg80211 MLME SAP interface
4  *
5  * Copyright (c) 2009, Jouni Malinen <j@w1.fi>
6  * Copyright (c) 2015           Intel Deutschland GmbH
7  * Copyright (C) 2019-2020, 2022 Intel Corporation
8  */
9
10 #include <linux/kernel.h>
11 #include <linux/module.h>
12 #include <linux/etherdevice.h>
13 #include <linux/netdevice.h>
14 #include <linux/nl80211.h>
15 #include <linux/slab.h>
16 #include <linux/wireless.h>
17 #include <net/cfg80211.h>
18 #include <net/iw_handler.h>
19 #include "core.h"
20 #include "nl80211.h"
21 #include "rdev-ops.h"
22
23
24 void cfg80211_rx_assoc_resp(struct net_device *dev,
25                             struct cfg80211_rx_assoc_resp *data)
26 {
27         struct wireless_dev *wdev = dev->ieee80211_ptr;
28         struct wiphy *wiphy = wdev->wiphy;
29         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
30         struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)data->buf;
31         struct cfg80211_connect_resp_params cr = {
32                 .timeout_reason = NL80211_TIMEOUT_UNSPECIFIED,
33                 .req_ie = data->req_ies,
34                 .req_ie_len = data->req_ies_len,
35                 .resp_ie = mgmt->u.assoc_resp.variable,
36                 .resp_ie_len = data->len -
37                                offsetof(struct ieee80211_mgmt,
38                                         u.assoc_resp.variable),
39                 .status = le16_to_cpu(mgmt->u.assoc_resp.status_code),
40                 .ap_mld_addr = data->ap_mld_addr,
41         };
42         unsigned int link_id;
43
44         for (link_id = 0; link_id < ARRAY_SIZE(data->links); link_id++) {
45                 cr.links[link_id].status = data->links[link_id].status;
46                 WARN_ON_ONCE(cr.links[link_id].status != WLAN_STATUS_SUCCESS &&
47                              (!cr.ap_mld_addr || !cr.links[link_id].bss));
48
49                 cr.links[link_id].bss = data->links[link_id].bss;
50                 if (!cr.links[link_id].bss)
51                         continue;
52                 cr.links[link_id].bssid = data->links[link_id].bss->bssid;
53                 cr.links[link_id].addr = data->links[link_id].addr;
54                 /* need to have local link addresses for MLO connections */
55                 WARN_ON(cr.ap_mld_addr &&
56                         !is_valid_ether_addr(cr.links[link_id].addr));
57
58                 BUG_ON(!cr.links[link_id].bss->channel);
59
60                 if (cr.links[link_id].bss->channel->band == NL80211_BAND_S1GHZ) {
61                         WARN_ON(link_id);
62                         cr.resp_ie = (u8 *)&mgmt->u.s1g_assoc_resp.variable;
63                         cr.resp_ie_len = data->len -
64                                          offsetof(struct ieee80211_mgmt,
65                                                   u.s1g_assoc_resp.variable);
66                 }
67
68                 if (cr.ap_mld_addr)
69                         cr.valid_links |= BIT(link_id);
70         }
71
72         trace_cfg80211_send_rx_assoc(dev, data);
73
74         /*
75          * This is a bit of a hack, we don't notify userspace of
76          * a (re-)association reply if we tried to send a reassoc
77          * and got a reject -- we only try again with an assoc
78          * frame instead of reassoc.
79          */
80         if (cfg80211_sme_rx_assoc_resp(wdev, cr.status)) {
81                 for (link_id = 0; link_id < ARRAY_SIZE(data->links); link_id++) {
82                         struct cfg80211_bss *bss = data->links[link_id].bss;
83
84                         if (!bss)
85                                 continue;
86
87                         cfg80211_unhold_bss(bss_from_pub(bss));
88                         cfg80211_put_bss(wiphy, bss);
89                 }
90                 return;
91         }
92
93         nl80211_send_rx_assoc(rdev, dev, data);
94         /* update current_bss etc., consumes the bss reference */
95         __cfg80211_connect_result(dev, &cr, cr.status == WLAN_STATUS_SUCCESS);
96 }
97 EXPORT_SYMBOL(cfg80211_rx_assoc_resp);
98
99 static void cfg80211_process_auth(struct wireless_dev *wdev,
100                                   const u8 *buf, size_t len)
101 {
102         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
103
104         nl80211_send_rx_auth(rdev, wdev->netdev, buf, len, GFP_KERNEL);
105         cfg80211_sme_rx_auth(wdev, buf, len);
106 }
107
108 static void cfg80211_process_deauth(struct wireless_dev *wdev,
109                                     const u8 *buf, size_t len,
110                                     bool reconnect)
111 {
112         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
113         struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)buf;
114         const u8 *bssid = mgmt->bssid;
115         u16 reason_code = le16_to_cpu(mgmt->u.deauth.reason_code);
116         bool from_ap = !ether_addr_equal(mgmt->sa, wdev->netdev->dev_addr);
117
118         nl80211_send_deauth(rdev, wdev->netdev, buf, len, reconnect, GFP_KERNEL);
119
120         if (!wdev->connected || !ether_addr_equal(wdev->u.client.connected_addr, bssid))
121                 return;
122
123         __cfg80211_disconnected(wdev->netdev, NULL, 0, reason_code, from_ap);
124         cfg80211_sme_deauth(wdev);
125 }
126
127 static void cfg80211_process_disassoc(struct wireless_dev *wdev,
128                                       const u8 *buf, size_t len,
129                                       bool reconnect)
130 {
131         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
132         struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)buf;
133         const u8 *bssid = mgmt->bssid;
134         u16 reason_code = le16_to_cpu(mgmt->u.disassoc.reason_code);
135         bool from_ap = !ether_addr_equal(mgmt->sa, wdev->netdev->dev_addr);
136
137         nl80211_send_disassoc(rdev, wdev->netdev, buf, len, reconnect,
138                               GFP_KERNEL);
139
140         if (WARN_ON(!wdev->connected ||
141                     !ether_addr_equal(wdev->u.client.connected_addr, bssid)))
142                 return;
143
144         __cfg80211_disconnected(wdev->netdev, NULL, 0, reason_code, from_ap);
145         cfg80211_sme_disassoc(wdev);
146 }
147
148 void cfg80211_rx_mlme_mgmt(struct net_device *dev, const u8 *buf, size_t len)
149 {
150         struct wireless_dev *wdev = dev->ieee80211_ptr;
151         struct ieee80211_mgmt *mgmt = (void *)buf;
152
153         ASSERT_WDEV_LOCK(wdev);
154
155         trace_cfg80211_rx_mlme_mgmt(dev, buf, len);
156
157         if (WARN_ON(len < 2))
158                 return;
159
160         if (ieee80211_is_auth(mgmt->frame_control))
161                 cfg80211_process_auth(wdev, buf, len);
162         else if (ieee80211_is_deauth(mgmt->frame_control))
163                 cfg80211_process_deauth(wdev, buf, len, false);
164         else if (ieee80211_is_disassoc(mgmt->frame_control))
165                 cfg80211_process_disassoc(wdev, buf, len, false);
166 }
167 EXPORT_SYMBOL(cfg80211_rx_mlme_mgmt);
168
169 void cfg80211_auth_timeout(struct net_device *dev, const u8 *addr)
170 {
171         struct wireless_dev *wdev = dev->ieee80211_ptr;
172         struct wiphy *wiphy = wdev->wiphy;
173         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
174
175         trace_cfg80211_send_auth_timeout(dev, addr);
176
177         nl80211_send_auth_timeout(rdev, dev, addr, GFP_KERNEL);
178         cfg80211_sme_auth_timeout(wdev);
179 }
180 EXPORT_SYMBOL(cfg80211_auth_timeout);
181
182 void cfg80211_assoc_failure(struct net_device *dev,
183                             struct cfg80211_assoc_failure *data)
184 {
185         struct wireless_dev *wdev = dev->ieee80211_ptr;
186         struct wiphy *wiphy = wdev->wiphy;
187         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
188         const u8 *addr = data->ap_mld_addr ?: data->bss[0]->bssid;
189         int i;
190
191         trace_cfg80211_send_assoc_failure(dev, data);
192
193         if (data->timeout) {
194                 nl80211_send_assoc_timeout(rdev, dev, addr, GFP_KERNEL);
195                 cfg80211_sme_assoc_timeout(wdev);
196         } else {
197                 cfg80211_sme_abandon_assoc(wdev);
198         }
199
200         for (i = 0; i < ARRAY_SIZE(data->bss); i++) {
201                 struct cfg80211_bss *bss = data->bss[i];
202
203                 if (!bss)
204                         continue;
205
206                 cfg80211_unhold_bss(bss_from_pub(bss));
207                 cfg80211_put_bss(wiphy, bss);
208         }
209 }
210 EXPORT_SYMBOL(cfg80211_assoc_failure);
211
212 void cfg80211_tx_mlme_mgmt(struct net_device *dev, const u8 *buf, size_t len,
213                            bool reconnect)
214 {
215         struct wireless_dev *wdev = dev->ieee80211_ptr;
216         struct ieee80211_mgmt *mgmt = (void *)buf;
217
218         ASSERT_WDEV_LOCK(wdev);
219
220         trace_cfg80211_tx_mlme_mgmt(dev, buf, len, reconnect);
221
222         if (WARN_ON(len < 2))
223                 return;
224
225         if (ieee80211_is_deauth(mgmt->frame_control))
226                 cfg80211_process_deauth(wdev, buf, len, reconnect);
227         else
228                 cfg80211_process_disassoc(wdev, buf, len, reconnect);
229 }
230 EXPORT_SYMBOL(cfg80211_tx_mlme_mgmt);
231
232 void cfg80211_michael_mic_failure(struct net_device *dev, const u8 *addr,
233                                   enum nl80211_key_type key_type, int key_id,
234                                   const u8 *tsc, gfp_t gfp)
235 {
236         struct wiphy *wiphy = dev->ieee80211_ptr->wiphy;
237         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
238 #ifdef CONFIG_CFG80211_WEXT
239         union iwreq_data wrqu;
240         char *buf = kmalloc(128, gfp);
241
242         if (buf) {
243                 sprintf(buf, "MLME-MICHAELMICFAILURE.indication("
244                         "keyid=%d %scast addr=%pM)", key_id,
245                         key_type == NL80211_KEYTYPE_GROUP ? "broad" : "uni",
246                         addr);
247                 memset(&wrqu, 0, sizeof(wrqu));
248                 wrqu.data.length = strlen(buf);
249                 wireless_send_event(dev, IWEVCUSTOM, &wrqu, buf);
250                 kfree(buf);
251         }
252 #endif
253
254         trace_cfg80211_michael_mic_failure(dev, addr, key_type, key_id, tsc);
255         nl80211_michael_mic_failure(rdev, dev, addr, key_type, key_id, tsc, gfp);
256 }
257 EXPORT_SYMBOL(cfg80211_michael_mic_failure);
258
259 /* some MLME handling for userspace SME */
260 int cfg80211_mlme_auth(struct cfg80211_registered_device *rdev,
261                        struct net_device *dev,
262                        struct cfg80211_auth_request *req)
263 {
264         struct wireless_dev *wdev = dev->ieee80211_ptr;
265
266         ASSERT_WDEV_LOCK(wdev);
267
268         if (!req->bss)
269                 return -ENOENT;
270
271         if (req->link_id >= 0 &&
272             !(wdev->wiphy->flags & WIPHY_FLAG_SUPPORTS_MLO))
273                 return -EINVAL;
274
275         if (req->auth_type == NL80211_AUTHTYPE_SHARED_KEY) {
276                 if (!req->key || !req->key_len ||
277                     req->key_idx < 0 || req->key_idx > 3)
278                         return -EINVAL;
279         }
280
281         if (wdev->connected &&
282             ether_addr_equal(req->bss->bssid, wdev->u.client.connected_addr))
283                 return -EALREADY;
284
285         if (ether_addr_equal(req->bss->bssid, dev->dev_addr) ||
286             (req->link_id >= 0 &&
287              ether_addr_equal(req->ap_mld_addr, dev->dev_addr)))
288                 return -EINVAL;
289
290         return rdev_auth(rdev, dev, req);
291 }
292
293 /*  Do a logical ht_capa &= ht_capa_mask.  */
294 void cfg80211_oper_and_ht_capa(struct ieee80211_ht_cap *ht_capa,
295                                const struct ieee80211_ht_cap *ht_capa_mask)
296 {
297         int i;
298         u8 *p1, *p2;
299         if (!ht_capa_mask) {
300                 memset(ht_capa, 0, sizeof(*ht_capa));
301                 return;
302         }
303
304         p1 = (u8*)(ht_capa);
305         p2 = (u8*)(ht_capa_mask);
306         for (i = 0; i < sizeof(*ht_capa); i++)
307                 p1[i] &= p2[i];
308 }
309
310 /*  Do a logical vht_capa &= vht_capa_mask.  */
311 void cfg80211_oper_and_vht_capa(struct ieee80211_vht_cap *vht_capa,
312                                 const struct ieee80211_vht_cap *vht_capa_mask)
313 {
314         int i;
315         u8 *p1, *p2;
316         if (!vht_capa_mask) {
317                 memset(vht_capa, 0, sizeof(*vht_capa));
318                 return;
319         }
320
321         p1 = (u8*)(vht_capa);
322         p2 = (u8*)(vht_capa_mask);
323         for (i = 0; i < sizeof(*vht_capa); i++)
324                 p1[i] &= p2[i];
325 }
326
327 /* Note: caller must cfg80211_put_bss() regardless of result */
328 int cfg80211_mlme_assoc(struct cfg80211_registered_device *rdev,
329                         struct net_device *dev,
330                         struct cfg80211_assoc_request *req)
331 {
332         struct wireless_dev *wdev = dev->ieee80211_ptr;
333         int err, i, j;
334
335         ASSERT_WDEV_LOCK(wdev);
336
337         for (i = 1; i < ARRAY_SIZE(req->links); i++) {
338                 if (!req->links[i].bss)
339                         continue;
340                 for (j = 0; j < i; j++) {
341                         if (req->links[i].bss == req->links[j].bss)
342                                 return -EINVAL;
343                 }
344
345                 if (ether_addr_equal(req->links[i].bss->bssid, dev->dev_addr))
346                         return -EINVAL;
347         }
348
349         if (wdev->connected &&
350             (!req->prev_bssid ||
351              !ether_addr_equal(wdev->u.client.connected_addr, req->prev_bssid)))
352                 return -EALREADY;
353
354         if ((req->bss && ether_addr_equal(req->bss->bssid, dev->dev_addr)) ||
355             (req->link_id >= 0 &&
356              ether_addr_equal(req->ap_mld_addr, dev->dev_addr)))
357                 return -EINVAL;
358
359         cfg80211_oper_and_ht_capa(&req->ht_capa_mask,
360                                   rdev->wiphy.ht_capa_mod_mask);
361         cfg80211_oper_and_vht_capa(&req->vht_capa_mask,
362                                    rdev->wiphy.vht_capa_mod_mask);
363
364         err = rdev_assoc(rdev, dev, req);
365         if (!err) {
366                 int link_id;
367
368                 if (req->bss) {
369                         cfg80211_ref_bss(&rdev->wiphy, req->bss);
370                         cfg80211_hold_bss(bss_from_pub(req->bss));
371                 }
372
373                 for (link_id = 0; link_id < ARRAY_SIZE(req->links); link_id++) {
374                         if (!req->links[link_id].bss)
375                                 continue;
376                         cfg80211_ref_bss(&rdev->wiphy, req->links[link_id].bss);
377                         cfg80211_hold_bss(bss_from_pub(req->links[link_id].bss));
378                 }
379         }
380         return err;
381 }
382
383 int cfg80211_mlme_deauth(struct cfg80211_registered_device *rdev,
384                          struct net_device *dev, const u8 *bssid,
385                          const u8 *ie, int ie_len, u16 reason,
386                          bool local_state_change)
387 {
388         struct wireless_dev *wdev = dev->ieee80211_ptr;
389         struct cfg80211_deauth_request req = {
390                 .bssid = bssid,
391                 .reason_code = reason,
392                 .ie = ie,
393                 .ie_len = ie_len,
394                 .local_state_change = local_state_change,
395         };
396
397         ASSERT_WDEV_LOCK(wdev);
398
399         if (local_state_change &&
400             (!wdev->connected ||
401              !ether_addr_equal(wdev->u.client.connected_addr, bssid)))
402                 return 0;
403
404         if (ether_addr_equal(wdev->disconnect_bssid, bssid) ||
405             (wdev->connected &&
406              ether_addr_equal(wdev->u.client.connected_addr, bssid)))
407                 wdev->conn_owner_nlportid = 0;
408
409         return rdev_deauth(rdev, dev, &req);
410 }
411
412 int cfg80211_mlme_disassoc(struct cfg80211_registered_device *rdev,
413                            struct net_device *dev, const u8 *ap_addr,
414                            const u8 *ie, int ie_len, u16 reason,
415                            bool local_state_change)
416 {
417         struct wireless_dev *wdev = dev->ieee80211_ptr;
418         struct cfg80211_disassoc_request req = {
419                 .reason_code = reason,
420                 .local_state_change = local_state_change,
421                 .ie = ie,
422                 .ie_len = ie_len,
423                 .ap_addr = ap_addr,
424         };
425         int err;
426
427         ASSERT_WDEV_LOCK(wdev);
428
429         if (!wdev->connected)
430                 return -ENOTCONN;
431
432         if (memcmp(wdev->u.client.connected_addr, ap_addr, ETH_ALEN))
433                 return -ENOTCONN;
434
435         err = rdev_disassoc(rdev, dev, &req);
436         if (err)
437                 return err;
438
439         /* driver should have reported the disassoc */
440         WARN_ON(wdev->connected);
441         return 0;
442 }
443
444 void cfg80211_mlme_down(struct cfg80211_registered_device *rdev,
445                         struct net_device *dev)
446 {
447         struct wireless_dev *wdev = dev->ieee80211_ptr;
448         u8 bssid[ETH_ALEN];
449
450         ASSERT_WDEV_LOCK(wdev);
451
452         if (!rdev->ops->deauth)
453                 return;
454
455         if (!wdev->connected)
456                 return;
457
458         memcpy(bssid, wdev->u.client.connected_addr, ETH_ALEN);
459         cfg80211_mlme_deauth(rdev, dev, bssid, NULL, 0,
460                              WLAN_REASON_DEAUTH_LEAVING, false);
461 }
462
463 struct cfg80211_mgmt_registration {
464         struct list_head list;
465         struct wireless_dev *wdev;
466
467         u32 nlportid;
468
469         int match_len;
470
471         __le16 frame_type;
472
473         bool multicast_rx;
474
475         u8 match[];
476 };
477
478 static void cfg80211_mgmt_registrations_update(struct wireless_dev *wdev)
479 {
480         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
481         struct wireless_dev *tmp;
482         struct cfg80211_mgmt_registration *reg;
483         struct mgmt_frame_regs upd = {};
484
485         lockdep_assert_held(&rdev->wiphy.mtx);
486
487         spin_lock_bh(&rdev->mgmt_registrations_lock);
488         if (!wdev->mgmt_registrations_need_update) {
489                 spin_unlock_bh(&rdev->mgmt_registrations_lock);
490                 return;
491         }
492
493         rcu_read_lock();
494         list_for_each_entry_rcu(tmp, &rdev->wiphy.wdev_list, list) {
495                 list_for_each_entry(reg, &tmp->mgmt_registrations, list) {
496                         u32 mask = BIT(le16_to_cpu(reg->frame_type) >> 4);
497                         u32 mcast_mask = 0;
498
499                         if (reg->multicast_rx)
500                                 mcast_mask = mask;
501
502                         upd.global_stypes |= mask;
503                         upd.global_mcast_stypes |= mcast_mask;
504
505                         if (tmp == wdev) {
506                                 upd.interface_stypes |= mask;
507                                 upd.interface_mcast_stypes |= mcast_mask;
508                         }
509                 }
510         }
511         rcu_read_unlock();
512
513         wdev->mgmt_registrations_need_update = 0;
514         spin_unlock_bh(&rdev->mgmt_registrations_lock);
515
516         rdev_update_mgmt_frame_registrations(rdev, wdev, &upd);
517 }
518
519 void cfg80211_mgmt_registrations_update_wk(struct work_struct *wk)
520 {
521         struct cfg80211_registered_device *rdev;
522         struct wireless_dev *wdev;
523
524         rdev = container_of(wk, struct cfg80211_registered_device,
525                             mgmt_registrations_update_wk);
526
527         wiphy_lock(&rdev->wiphy);
528         list_for_each_entry(wdev, &rdev->wiphy.wdev_list, list)
529                 cfg80211_mgmt_registrations_update(wdev);
530         wiphy_unlock(&rdev->wiphy);
531 }
532
533 int cfg80211_mlme_register_mgmt(struct wireless_dev *wdev, u32 snd_portid,
534                                 u16 frame_type, const u8 *match_data,
535                                 int match_len, bool multicast_rx,
536                                 struct netlink_ext_ack *extack)
537 {
538         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
539         struct cfg80211_mgmt_registration *reg, *nreg;
540         int err = 0;
541         u16 mgmt_type;
542         bool update_multicast = false;
543
544         if (!wdev->wiphy->mgmt_stypes)
545                 return -EOPNOTSUPP;
546
547         if ((frame_type & IEEE80211_FCTL_FTYPE) != IEEE80211_FTYPE_MGMT) {
548                 NL_SET_ERR_MSG(extack, "frame type not management");
549                 return -EINVAL;
550         }
551
552         if (frame_type & ~(IEEE80211_FCTL_FTYPE | IEEE80211_FCTL_STYPE)) {
553                 NL_SET_ERR_MSG(extack, "Invalid frame type");
554                 return -EINVAL;
555         }
556
557         mgmt_type = (frame_type & IEEE80211_FCTL_STYPE) >> 4;
558         if (!(wdev->wiphy->mgmt_stypes[wdev->iftype].rx & BIT(mgmt_type))) {
559                 NL_SET_ERR_MSG(extack,
560                                "Registration to specific type not supported");
561                 return -EINVAL;
562         }
563
564         /*
565          * To support Pre Association Security Negotiation (PASN), registration
566          * for authentication frames should be supported. However, as some
567          * versions of the user space daemons wrongly register to all types of
568          * authentication frames (which might result in unexpected behavior)
569          * allow such registration if the request is for a specific
570          * authentication algorithm number.
571          */
572         if (wdev->iftype == NL80211_IFTYPE_STATION &&
573             (frame_type & IEEE80211_FCTL_STYPE) == IEEE80211_STYPE_AUTH &&
574             !(match_data && match_len >= 2)) {
575                 NL_SET_ERR_MSG(extack,
576                                "Authentication algorithm number required");
577                 return -EINVAL;
578         }
579
580         nreg = kzalloc(sizeof(*reg) + match_len, GFP_KERNEL);
581         if (!nreg)
582                 return -ENOMEM;
583
584         spin_lock_bh(&rdev->mgmt_registrations_lock);
585
586         list_for_each_entry(reg, &wdev->mgmt_registrations, list) {
587                 int mlen = min(match_len, reg->match_len);
588
589                 if (frame_type != le16_to_cpu(reg->frame_type))
590                         continue;
591
592                 if (memcmp(reg->match, match_data, mlen) == 0) {
593                         if (reg->multicast_rx != multicast_rx) {
594                                 update_multicast = true;
595                                 reg->multicast_rx = multicast_rx;
596                                 break;
597                         }
598                         NL_SET_ERR_MSG(extack, "Match already configured");
599                         err = -EALREADY;
600                         break;
601                 }
602         }
603
604         if (err)
605                 goto out;
606
607         if (update_multicast) {
608                 kfree(nreg);
609         } else {
610                 memcpy(nreg->match, match_data, match_len);
611                 nreg->match_len = match_len;
612                 nreg->nlportid = snd_portid;
613                 nreg->frame_type = cpu_to_le16(frame_type);
614                 nreg->wdev = wdev;
615                 nreg->multicast_rx = multicast_rx;
616                 list_add(&nreg->list, &wdev->mgmt_registrations);
617         }
618         wdev->mgmt_registrations_need_update = 1;
619         spin_unlock_bh(&rdev->mgmt_registrations_lock);
620
621         cfg80211_mgmt_registrations_update(wdev);
622
623         return 0;
624
625  out:
626         kfree(nreg);
627         spin_unlock_bh(&rdev->mgmt_registrations_lock);
628
629         return err;
630 }
631
632 void cfg80211_mlme_unregister_socket(struct wireless_dev *wdev, u32 nlportid)
633 {
634         struct wiphy *wiphy = wdev->wiphy;
635         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
636         struct cfg80211_mgmt_registration *reg, *tmp;
637
638         spin_lock_bh(&rdev->mgmt_registrations_lock);
639
640         list_for_each_entry_safe(reg, tmp, &wdev->mgmt_registrations, list) {
641                 if (reg->nlportid != nlportid)
642                         continue;
643
644                 list_del(&reg->list);
645                 kfree(reg);
646
647                 wdev->mgmt_registrations_need_update = 1;
648                 schedule_work(&rdev->mgmt_registrations_update_wk);
649         }
650
651         spin_unlock_bh(&rdev->mgmt_registrations_lock);
652
653         if (nlportid && rdev->crit_proto_nlportid == nlportid) {
654                 rdev->crit_proto_nlportid = 0;
655                 rdev_crit_proto_stop(rdev, wdev);
656         }
657
658         if (nlportid == wdev->ap_unexpected_nlportid)
659                 wdev->ap_unexpected_nlportid = 0;
660 }
661
662 void cfg80211_mlme_purge_registrations(struct wireless_dev *wdev)
663 {
664         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
665         struct cfg80211_mgmt_registration *reg, *tmp;
666
667         spin_lock_bh(&rdev->mgmt_registrations_lock);
668         list_for_each_entry_safe(reg, tmp, &wdev->mgmt_registrations, list) {
669                 list_del(&reg->list);
670                 kfree(reg);
671         }
672         wdev->mgmt_registrations_need_update = 1;
673         spin_unlock_bh(&rdev->mgmt_registrations_lock);
674
675         cfg80211_mgmt_registrations_update(wdev);
676 }
677
678 static bool cfg80211_allowed_address(struct wireless_dev *wdev, const u8 *addr)
679 {
680         int i;
681
682         for_each_valid_link(wdev, i) {
683                 if (ether_addr_equal(addr, wdev->links[i].addr))
684                         return true;
685         }
686
687         return ether_addr_equal(addr, wdev_address(wdev));
688 }
689
690 static bool cfg80211_allowed_random_address(struct wireless_dev *wdev,
691                                             const struct ieee80211_mgmt *mgmt)
692 {
693         if (ieee80211_is_auth(mgmt->frame_control) ||
694             ieee80211_is_deauth(mgmt->frame_control)) {
695                 /* Allow random TA to be used with authentication and
696                  * deauthentication frames if the driver has indicated support.
697                  */
698                 if (wiphy_ext_feature_isset(
699                             wdev->wiphy,
700                             NL80211_EXT_FEATURE_AUTH_AND_DEAUTH_RANDOM_TA))
701                         return true;
702         } else if (ieee80211_is_action(mgmt->frame_control) &&
703                    mgmt->u.action.category == WLAN_CATEGORY_PUBLIC) {
704                 /* Allow random TA to be used with Public Action frames if the
705                  * driver has indicated support.
706                  */
707                 if (!wdev->connected &&
708                     wiphy_ext_feature_isset(
709                             wdev->wiphy,
710                             NL80211_EXT_FEATURE_MGMT_TX_RANDOM_TA))
711                         return true;
712
713                 if (wdev->connected &&
714                     wiphy_ext_feature_isset(
715                             wdev->wiphy,
716                             NL80211_EXT_FEATURE_MGMT_TX_RANDOM_TA_CONNECTED))
717                         return true;
718         }
719
720         return false;
721 }
722
723 int cfg80211_mlme_mgmt_tx(struct cfg80211_registered_device *rdev,
724                           struct wireless_dev *wdev,
725                           struct cfg80211_mgmt_tx_params *params, u64 *cookie)
726 {
727         const struct ieee80211_mgmt *mgmt;
728         u16 stype;
729
730         if (!wdev->wiphy->mgmt_stypes)
731                 return -EOPNOTSUPP;
732
733         if (!rdev->ops->mgmt_tx)
734                 return -EOPNOTSUPP;
735
736         if (params->len < 24 + 1)
737                 return -EINVAL;
738
739         mgmt = (const struct ieee80211_mgmt *)params->buf;
740
741         if (!ieee80211_is_mgmt(mgmt->frame_control))
742                 return -EINVAL;
743
744         stype = le16_to_cpu(mgmt->frame_control) & IEEE80211_FCTL_STYPE;
745         if (!(wdev->wiphy->mgmt_stypes[wdev->iftype].tx & BIT(stype >> 4)))
746                 return -EINVAL;
747
748         if (ieee80211_is_action(mgmt->frame_control) &&
749             mgmt->u.action.category != WLAN_CATEGORY_PUBLIC) {
750                 int err = 0;
751
752                 wdev_lock(wdev);
753
754                 switch (wdev->iftype) {
755                 case NL80211_IFTYPE_ADHOC:
756                         /*
757                          * check for IBSS DA must be done by driver as
758                          * cfg80211 doesn't track the stations
759                          */
760                         if (!wdev->u.ibss.current_bss ||
761                             !ether_addr_equal(wdev->u.ibss.current_bss->pub.bssid,
762                                               mgmt->bssid)) {
763                                 err = -ENOTCONN;
764                                 break;
765                         }
766                         break;
767                 case NL80211_IFTYPE_STATION:
768                 case NL80211_IFTYPE_P2P_CLIENT:
769                         if (!wdev->connected) {
770                                 err = -ENOTCONN;
771                                 break;
772                         }
773
774                         /* FIXME: MLD may address this differently */
775
776                         if (!ether_addr_equal(wdev->u.client.connected_addr,
777                                               mgmt->bssid)) {
778                                 err = -ENOTCONN;
779                                 break;
780                         }
781
782                         /* for station, check that DA is the AP */
783                         if (!ether_addr_equal(wdev->u.client.connected_addr,
784                                               mgmt->da)) {
785                                 err = -ENOTCONN;
786                                 break;
787                         }
788                         break;
789                 case NL80211_IFTYPE_AP:
790                 case NL80211_IFTYPE_P2P_GO:
791                 case NL80211_IFTYPE_AP_VLAN:
792                         if (!ether_addr_equal(mgmt->bssid, wdev_address(wdev)) &&
793                             (params->link_id < 0 ||
794                              !ether_addr_equal(mgmt->bssid,
795                                                wdev->links[params->link_id].addr)))
796                                 err = -EINVAL;
797                         break;
798                 case NL80211_IFTYPE_MESH_POINT:
799                         if (!ether_addr_equal(mgmt->sa, mgmt->bssid)) {
800                                 err = -EINVAL;
801                                 break;
802                         }
803                         /*
804                          * check for mesh DA must be done by driver as
805                          * cfg80211 doesn't track the stations
806                          */
807                         break;
808                 case NL80211_IFTYPE_P2P_DEVICE:
809                         /*
810                          * fall through, P2P device only supports
811                          * public action frames
812                          */
813                 case NL80211_IFTYPE_NAN:
814                 default:
815                         err = -EOPNOTSUPP;
816                         break;
817                 }
818                 wdev_unlock(wdev);
819
820                 if (err)
821                         return err;
822         }
823
824         if (!cfg80211_allowed_address(wdev, mgmt->sa) &&
825             !cfg80211_allowed_random_address(wdev, mgmt))
826                 return -EINVAL;
827
828         /* Transmit the management frame as requested by user space */
829         return rdev_mgmt_tx(rdev, wdev, params, cookie);
830 }
831
832 bool cfg80211_rx_mgmt_ext(struct wireless_dev *wdev,
833                           struct cfg80211_rx_info *info)
834 {
835         struct wiphy *wiphy = wdev->wiphy;
836         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
837         struct cfg80211_mgmt_registration *reg;
838         const struct ieee80211_txrx_stypes *stypes =
839                 &wiphy->mgmt_stypes[wdev->iftype];
840         struct ieee80211_mgmt *mgmt = (void *)info->buf;
841         const u8 *data;
842         int data_len;
843         bool result = false;
844         __le16 ftype = mgmt->frame_control &
845                 cpu_to_le16(IEEE80211_FCTL_FTYPE | IEEE80211_FCTL_STYPE);
846         u16 stype;
847
848         trace_cfg80211_rx_mgmt(wdev, info);
849         stype = (le16_to_cpu(mgmt->frame_control) & IEEE80211_FCTL_STYPE) >> 4;
850
851         if (!(stypes->rx & BIT(stype))) {
852                 trace_cfg80211_return_bool(false);
853                 return false;
854         }
855
856         data = info->buf + ieee80211_hdrlen(mgmt->frame_control);
857         data_len = info->len - ieee80211_hdrlen(mgmt->frame_control);
858
859         spin_lock_bh(&rdev->mgmt_registrations_lock);
860
861         list_for_each_entry(reg, &wdev->mgmt_registrations, list) {
862                 if (reg->frame_type != ftype)
863                         continue;
864
865                 if (reg->match_len > data_len)
866                         continue;
867
868                 if (memcmp(reg->match, data, reg->match_len))
869                         continue;
870
871                 /* found match! */
872
873                 /* Indicate the received Action frame to user space */
874                 if (nl80211_send_mgmt(rdev, wdev, reg->nlportid, info,
875                                       GFP_ATOMIC))
876                         continue;
877
878                 result = true;
879                 break;
880         }
881
882         spin_unlock_bh(&rdev->mgmt_registrations_lock);
883
884         trace_cfg80211_return_bool(result);
885         return result;
886 }
887 EXPORT_SYMBOL(cfg80211_rx_mgmt_ext);
888
889 void cfg80211_sched_dfs_chan_update(struct cfg80211_registered_device *rdev)
890 {
891         cancel_delayed_work(&rdev->dfs_update_channels_wk);
892         queue_delayed_work(cfg80211_wq, &rdev->dfs_update_channels_wk, 0);
893 }
894
895 void cfg80211_dfs_channels_update_work(struct work_struct *work)
896 {
897         struct delayed_work *delayed_work = to_delayed_work(work);
898         struct cfg80211_registered_device *rdev;
899         struct cfg80211_chan_def chandef;
900         struct ieee80211_supported_band *sband;
901         struct ieee80211_channel *c;
902         struct wiphy *wiphy;
903         bool check_again = false;
904         unsigned long timeout, next_time = 0;
905         unsigned long time_dfs_update;
906         enum nl80211_radar_event radar_event;
907         int bandid, i;
908
909         rdev = container_of(delayed_work, struct cfg80211_registered_device,
910                             dfs_update_channels_wk);
911         wiphy = &rdev->wiphy;
912
913         rtnl_lock();
914         for (bandid = 0; bandid < NUM_NL80211_BANDS; bandid++) {
915                 sband = wiphy->bands[bandid];
916                 if (!sband)
917                         continue;
918
919                 for (i = 0; i < sband->n_channels; i++) {
920                         c = &sband->channels[i];
921
922                         if (!(c->flags & IEEE80211_CHAN_RADAR))
923                                 continue;
924
925                         if (c->dfs_state != NL80211_DFS_UNAVAILABLE &&
926                             c->dfs_state != NL80211_DFS_AVAILABLE)
927                                 continue;
928
929                         if (c->dfs_state == NL80211_DFS_UNAVAILABLE) {
930                                 time_dfs_update = IEEE80211_DFS_MIN_NOP_TIME_MS;
931                                 radar_event = NL80211_RADAR_NOP_FINISHED;
932                         } else {
933                                 if (regulatory_pre_cac_allowed(wiphy) ||
934                                     cfg80211_any_wiphy_oper_chan(wiphy, c))
935                                         continue;
936
937                                 time_dfs_update = REG_PRE_CAC_EXPIRY_GRACE_MS;
938                                 radar_event = NL80211_RADAR_PRE_CAC_EXPIRED;
939                         }
940
941                         timeout = c->dfs_state_entered +
942                                   msecs_to_jiffies(time_dfs_update);
943
944                         if (time_after_eq(jiffies, timeout)) {
945                                 c->dfs_state = NL80211_DFS_USABLE;
946                                 c->dfs_state_entered = jiffies;
947
948                                 cfg80211_chandef_create(&chandef, c,
949                                                         NL80211_CHAN_NO_HT);
950
951                                 nl80211_radar_notify(rdev, &chandef,
952                                                      radar_event, NULL,
953                                                      GFP_ATOMIC);
954
955                                 regulatory_propagate_dfs_state(wiphy, &chandef,
956                                                                c->dfs_state,
957                                                                radar_event);
958                                 continue;
959                         }
960
961                         if (!check_again)
962                                 next_time = timeout - jiffies;
963                         else
964                                 next_time = min(next_time, timeout - jiffies);
965                         check_again = true;
966                 }
967         }
968         rtnl_unlock();
969
970         /* reschedule if there are other channels waiting to be cleared again */
971         if (check_again)
972                 queue_delayed_work(cfg80211_wq, &rdev->dfs_update_channels_wk,
973                                    next_time);
974 }
975
976
977 void __cfg80211_radar_event(struct wiphy *wiphy,
978                             struct cfg80211_chan_def *chandef,
979                             bool offchan, gfp_t gfp)
980 {
981         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
982
983         trace_cfg80211_radar_event(wiphy, chandef, offchan);
984
985         /* only set the chandef supplied channel to unavailable, in
986          * case the radar is detected on only one of multiple channels
987          * spanned by the chandef.
988          */
989         cfg80211_set_dfs_state(wiphy, chandef, NL80211_DFS_UNAVAILABLE);
990
991         if (offchan)
992                 queue_work(cfg80211_wq, &rdev->background_cac_abort_wk);
993
994         cfg80211_sched_dfs_chan_update(rdev);
995
996         nl80211_radar_notify(rdev, chandef, NL80211_RADAR_DETECTED, NULL, gfp);
997
998         memcpy(&rdev->radar_chandef, chandef, sizeof(struct cfg80211_chan_def));
999         queue_work(cfg80211_wq, &rdev->propagate_radar_detect_wk);
1000 }
1001 EXPORT_SYMBOL(__cfg80211_radar_event);
1002
1003 void cfg80211_cac_event(struct net_device *netdev,
1004                         const struct cfg80211_chan_def *chandef,
1005                         enum nl80211_radar_event event, gfp_t gfp)
1006 {
1007         struct wireless_dev *wdev = netdev->ieee80211_ptr;
1008         struct wiphy *wiphy = wdev->wiphy;
1009         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
1010         unsigned long timeout;
1011
1012         /* not yet supported */
1013         if (wdev->valid_links)
1014                 return;
1015
1016         trace_cfg80211_cac_event(netdev, event);
1017
1018         if (WARN_ON(!wdev->cac_started && event != NL80211_RADAR_CAC_STARTED))
1019                 return;
1020
1021         switch (event) {
1022         case NL80211_RADAR_CAC_FINISHED:
1023                 timeout = wdev->cac_start_time +
1024                           msecs_to_jiffies(wdev->cac_time_ms);
1025                 WARN_ON(!time_after_eq(jiffies, timeout));
1026                 cfg80211_set_dfs_state(wiphy, chandef, NL80211_DFS_AVAILABLE);
1027                 memcpy(&rdev->cac_done_chandef, chandef,
1028                        sizeof(struct cfg80211_chan_def));
1029                 queue_work(cfg80211_wq, &rdev->propagate_cac_done_wk);
1030                 cfg80211_sched_dfs_chan_update(rdev);
1031                 fallthrough;
1032         case NL80211_RADAR_CAC_ABORTED:
1033                 wdev->cac_started = false;
1034                 break;
1035         case NL80211_RADAR_CAC_STARTED:
1036                 wdev->cac_started = true;
1037                 break;
1038         default:
1039                 WARN_ON(1);
1040                 return;
1041         }
1042
1043         nl80211_radar_notify(rdev, chandef, event, netdev, gfp);
1044 }
1045 EXPORT_SYMBOL(cfg80211_cac_event);
1046
1047 static void
1048 __cfg80211_background_cac_event(struct cfg80211_registered_device *rdev,
1049                                 struct wireless_dev *wdev,
1050                                 const struct cfg80211_chan_def *chandef,
1051                                 enum nl80211_radar_event event)
1052 {
1053         struct wiphy *wiphy = &rdev->wiphy;
1054         struct net_device *netdev;
1055
1056         lockdep_assert_wiphy(&rdev->wiphy);
1057
1058         if (!cfg80211_chandef_valid(chandef))
1059                 return;
1060
1061         if (!rdev->background_radar_wdev)
1062                 return;
1063
1064         switch (event) {
1065         case NL80211_RADAR_CAC_FINISHED:
1066                 cfg80211_set_dfs_state(wiphy, chandef, NL80211_DFS_AVAILABLE);
1067                 memcpy(&rdev->cac_done_chandef, chandef, sizeof(*chandef));
1068                 queue_work(cfg80211_wq, &rdev->propagate_cac_done_wk);
1069                 cfg80211_sched_dfs_chan_update(rdev);
1070                 wdev = rdev->background_radar_wdev;
1071                 break;
1072         case NL80211_RADAR_CAC_ABORTED:
1073                 if (!cancel_delayed_work(&rdev->background_cac_done_wk))
1074                         return;
1075                 wdev = rdev->background_radar_wdev;
1076                 break;
1077         case NL80211_RADAR_CAC_STARTED:
1078                 break;
1079         default:
1080                 return;
1081         }
1082
1083         netdev = wdev ? wdev->netdev : NULL;
1084         nl80211_radar_notify(rdev, chandef, event, netdev, GFP_KERNEL);
1085 }
1086
1087 static void
1088 cfg80211_background_cac_event(struct cfg80211_registered_device *rdev,
1089                               const struct cfg80211_chan_def *chandef,
1090                               enum nl80211_radar_event event)
1091 {
1092         wiphy_lock(&rdev->wiphy);
1093         __cfg80211_background_cac_event(rdev, rdev->background_radar_wdev,
1094                                         chandef, event);
1095         wiphy_unlock(&rdev->wiphy);
1096 }
1097
1098 void cfg80211_background_cac_done_wk(struct work_struct *work)
1099 {
1100         struct delayed_work *delayed_work = to_delayed_work(work);
1101         struct cfg80211_registered_device *rdev;
1102
1103         rdev = container_of(delayed_work, struct cfg80211_registered_device,
1104                             background_cac_done_wk);
1105         cfg80211_background_cac_event(rdev, &rdev->background_radar_chandef,
1106                                       NL80211_RADAR_CAC_FINISHED);
1107 }
1108
1109 void cfg80211_background_cac_abort_wk(struct work_struct *work)
1110 {
1111         struct cfg80211_registered_device *rdev;
1112
1113         rdev = container_of(work, struct cfg80211_registered_device,
1114                             background_cac_abort_wk);
1115         cfg80211_background_cac_event(rdev, &rdev->background_radar_chandef,
1116                                       NL80211_RADAR_CAC_ABORTED);
1117 }
1118
1119 void cfg80211_background_cac_abort(struct wiphy *wiphy)
1120 {
1121         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
1122
1123         queue_work(cfg80211_wq, &rdev->background_cac_abort_wk);
1124 }
1125 EXPORT_SYMBOL(cfg80211_background_cac_abort);
1126
1127 int
1128 cfg80211_start_background_radar_detection(struct cfg80211_registered_device *rdev,
1129                                           struct wireless_dev *wdev,
1130                                           struct cfg80211_chan_def *chandef)
1131 {
1132         unsigned int cac_time_ms;
1133         int err;
1134
1135         lockdep_assert_wiphy(&rdev->wiphy);
1136
1137         if (!wiphy_ext_feature_isset(&rdev->wiphy,
1138                                      NL80211_EXT_FEATURE_RADAR_BACKGROUND))
1139                 return -EOPNOTSUPP;
1140
1141         /* Offchannel chain already locked by another wdev */
1142         if (rdev->background_radar_wdev && rdev->background_radar_wdev != wdev)
1143                 return -EBUSY;
1144
1145         /* CAC already in progress on the offchannel chain */
1146         if (rdev->background_radar_wdev == wdev &&
1147             delayed_work_pending(&rdev->background_cac_done_wk))
1148                 return -EBUSY;
1149
1150         err = rdev_set_radar_background(rdev, chandef);
1151         if (err)
1152                 return err;
1153
1154         cac_time_ms = cfg80211_chandef_dfs_cac_time(&rdev->wiphy, chandef);
1155         if (!cac_time_ms)
1156                 cac_time_ms = IEEE80211_DFS_MIN_CAC_TIME_MS;
1157
1158         rdev->background_radar_chandef = *chandef;
1159         rdev->background_radar_wdev = wdev; /* Get offchain ownership */
1160
1161         __cfg80211_background_cac_event(rdev, wdev, chandef,
1162                                         NL80211_RADAR_CAC_STARTED);
1163         queue_delayed_work(cfg80211_wq, &rdev->background_cac_done_wk,
1164                            msecs_to_jiffies(cac_time_ms));
1165
1166         return 0;
1167 }
1168
1169 void cfg80211_stop_background_radar_detection(struct wireless_dev *wdev)
1170 {
1171         struct wiphy *wiphy = wdev->wiphy;
1172         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
1173
1174         lockdep_assert_wiphy(wiphy);
1175
1176         if (wdev != rdev->background_radar_wdev)
1177                 return;
1178
1179         rdev_set_radar_background(rdev, NULL);
1180         rdev->background_radar_wdev = NULL; /* Release offchain ownership */
1181
1182         __cfg80211_background_cac_event(rdev, wdev,
1183                                         &rdev->background_radar_chandef,
1184                                         NL80211_RADAR_CAC_ABORTED);
1185 }