cfg80211: Allow use_mfp to be specified with the connect command
[firefly-linux-kernel-4.4.55.git] / net / wireless / nl80211.c
1 /*
2  * This is the new netlink-based wireless configuration interface.
3  *
4  * Copyright 2006-2010  Johannes Berg <johannes@sipsolutions.net>
5  */
6
7 #include <linux/if.h>
8 #include <linux/module.h>
9 #include <linux/err.h>
10 #include <linux/slab.h>
11 #include <linux/list.h>
12 #include <linux/if_ether.h>
13 #include <linux/ieee80211.h>
14 #include <linux/nl80211.h>
15 #include <linux/rtnetlink.h>
16 #include <linux/netlink.h>
17 #include <linux/etherdevice.h>
18 #include <net/net_namespace.h>
19 #include <net/genetlink.h>
20 #include <net/cfg80211.h>
21 #include <net/sock.h>
22 #include "core.h"
23 #include "nl80211.h"
24 #include "reg.h"
25 #include "rdev-ops.h"
26
27 static int nl80211_crypto_settings(struct cfg80211_registered_device *rdev,
28                                    struct genl_info *info,
29                                    struct cfg80211_crypto_settings *settings,
30                                    int cipher_limit);
31
32 static int nl80211_pre_doit(struct genl_ops *ops, struct sk_buff *skb,
33                             struct genl_info *info);
34 static void nl80211_post_doit(struct genl_ops *ops, struct sk_buff *skb,
35                               struct genl_info *info);
36
37 /* the netlink family */
38 static struct genl_family nl80211_fam = {
39         .id = GENL_ID_GENERATE, /* don't bother with a hardcoded ID */
40         .name = "nl80211",      /* have users key off the name instead */
41         .hdrsize = 0,           /* no private header */
42         .version = 1,           /* no particular meaning now */
43         .maxattr = NL80211_ATTR_MAX,
44         .netnsok = true,
45         .pre_doit = nl80211_pre_doit,
46         .post_doit = nl80211_post_doit,
47 };
48
49 /* returns ERR_PTR values */
50 static struct wireless_dev *
51 __cfg80211_wdev_from_attrs(struct net *netns, struct nlattr **attrs)
52 {
53         struct cfg80211_registered_device *rdev;
54         struct wireless_dev *result = NULL;
55         bool have_ifidx = attrs[NL80211_ATTR_IFINDEX];
56         bool have_wdev_id = attrs[NL80211_ATTR_WDEV];
57         u64 wdev_id;
58         int wiphy_idx = -1;
59         int ifidx = -1;
60
61         assert_cfg80211_lock();
62
63         if (!have_ifidx && !have_wdev_id)
64                 return ERR_PTR(-EINVAL);
65
66         if (have_ifidx)
67                 ifidx = nla_get_u32(attrs[NL80211_ATTR_IFINDEX]);
68         if (have_wdev_id) {
69                 wdev_id = nla_get_u64(attrs[NL80211_ATTR_WDEV]);
70                 wiphy_idx = wdev_id >> 32;
71         }
72
73         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
74                 struct wireless_dev *wdev;
75
76                 if (wiphy_net(&rdev->wiphy) != netns)
77                         continue;
78
79                 if (have_wdev_id && rdev->wiphy_idx != wiphy_idx)
80                         continue;
81
82                 mutex_lock(&rdev->devlist_mtx);
83                 list_for_each_entry(wdev, &rdev->wdev_list, list) {
84                         if (have_ifidx && wdev->netdev &&
85                             wdev->netdev->ifindex == ifidx) {
86                                 result = wdev;
87                                 break;
88                         }
89                         if (have_wdev_id && wdev->identifier == (u32)wdev_id) {
90                                 result = wdev;
91                                 break;
92                         }
93                 }
94                 mutex_unlock(&rdev->devlist_mtx);
95
96                 if (result)
97                         break;
98         }
99
100         if (result)
101                 return result;
102         return ERR_PTR(-ENODEV);
103 }
104
105 static struct cfg80211_registered_device *
106 __cfg80211_rdev_from_attrs(struct net *netns, struct nlattr **attrs)
107 {
108         struct cfg80211_registered_device *rdev = NULL, *tmp;
109         struct net_device *netdev;
110
111         assert_cfg80211_lock();
112
113         if (!attrs[NL80211_ATTR_WIPHY] &&
114             !attrs[NL80211_ATTR_IFINDEX] &&
115             !attrs[NL80211_ATTR_WDEV])
116                 return ERR_PTR(-EINVAL);
117
118         if (attrs[NL80211_ATTR_WIPHY])
119                 rdev = cfg80211_rdev_by_wiphy_idx(
120                                 nla_get_u32(attrs[NL80211_ATTR_WIPHY]));
121
122         if (attrs[NL80211_ATTR_WDEV]) {
123                 u64 wdev_id = nla_get_u64(attrs[NL80211_ATTR_WDEV]);
124                 struct wireless_dev *wdev;
125                 bool found = false;
126
127                 tmp = cfg80211_rdev_by_wiphy_idx(wdev_id >> 32);
128                 if (tmp) {
129                         /* make sure wdev exists */
130                         mutex_lock(&tmp->devlist_mtx);
131                         list_for_each_entry(wdev, &tmp->wdev_list, list) {
132                                 if (wdev->identifier != (u32)wdev_id)
133                                         continue;
134                                 found = true;
135                                 break;
136                         }
137                         mutex_unlock(&tmp->devlist_mtx);
138
139                         if (!found)
140                                 tmp = NULL;
141
142                         if (rdev && tmp != rdev)
143                                 return ERR_PTR(-EINVAL);
144                         rdev = tmp;
145                 }
146         }
147
148         if (attrs[NL80211_ATTR_IFINDEX]) {
149                 int ifindex = nla_get_u32(attrs[NL80211_ATTR_IFINDEX]);
150                 netdev = dev_get_by_index(netns, ifindex);
151                 if (netdev) {
152                         if (netdev->ieee80211_ptr)
153                                 tmp = wiphy_to_dev(
154                                                 netdev->ieee80211_ptr->wiphy);
155                         else
156                                 tmp = NULL;
157
158                         dev_put(netdev);
159
160                         /* not wireless device -- return error */
161                         if (!tmp)
162                                 return ERR_PTR(-EINVAL);
163
164                         /* mismatch -- return error */
165                         if (rdev && tmp != rdev)
166                                 return ERR_PTR(-EINVAL);
167
168                         rdev = tmp;
169                 }
170         }
171
172         if (!rdev)
173                 return ERR_PTR(-ENODEV);
174
175         if (netns != wiphy_net(&rdev->wiphy))
176                 return ERR_PTR(-ENODEV);
177
178         return rdev;
179 }
180
181 /*
182  * This function returns a pointer to the driver
183  * that the genl_info item that is passed refers to.
184  * If successful, it returns non-NULL and also locks
185  * the driver's mutex!
186  *
187  * This means that you need to call cfg80211_unlock_rdev()
188  * before being allowed to acquire &cfg80211_mutex!
189  *
190  * This is necessary because we need to lock the global
191  * mutex to get an item off the list safely, and then
192  * we lock the rdev mutex so it doesn't go away under us.
193  *
194  * We don't want to keep cfg80211_mutex locked
195  * for all the time in order to allow requests on
196  * other interfaces to go through at the same time.
197  *
198  * The result of this can be a PTR_ERR and hence must
199  * be checked with IS_ERR() for errors.
200  */
201 static struct cfg80211_registered_device *
202 cfg80211_get_dev_from_info(struct net *netns, struct genl_info *info)
203 {
204         struct cfg80211_registered_device *rdev;
205
206         mutex_lock(&cfg80211_mutex);
207         rdev = __cfg80211_rdev_from_attrs(netns, info->attrs);
208
209         /* if it is not an error we grab the lock on
210          * it to assure it won't be going away while
211          * we operate on it */
212         if (!IS_ERR(rdev))
213                 mutex_lock(&rdev->mtx);
214
215         mutex_unlock(&cfg80211_mutex);
216
217         return rdev;
218 }
219
220 /* policy for the attributes */
221 static const struct nla_policy nl80211_policy[NL80211_ATTR_MAX+1] = {
222         [NL80211_ATTR_WIPHY] = { .type = NLA_U32 },
223         [NL80211_ATTR_WIPHY_NAME] = { .type = NLA_NUL_STRING,
224                                       .len = 20-1 },
225         [NL80211_ATTR_WIPHY_TXQ_PARAMS] = { .type = NLA_NESTED },
226
227         [NL80211_ATTR_WIPHY_FREQ] = { .type = NLA_U32 },
228         [NL80211_ATTR_WIPHY_CHANNEL_TYPE] = { .type = NLA_U32 },
229         [NL80211_ATTR_CHANNEL_WIDTH] = { .type = NLA_U32 },
230         [NL80211_ATTR_CENTER_FREQ1] = { .type = NLA_U32 },
231         [NL80211_ATTR_CENTER_FREQ2] = { .type = NLA_U32 },
232
233         [NL80211_ATTR_WIPHY_RETRY_SHORT] = { .type = NLA_U8 },
234         [NL80211_ATTR_WIPHY_RETRY_LONG] = { .type = NLA_U8 },
235         [NL80211_ATTR_WIPHY_FRAG_THRESHOLD] = { .type = NLA_U32 },
236         [NL80211_ATTR_WIPHY_RTS_THRESHOLD] = { .type = NLA_U32 },
237         [NL80211_ATTR_WIPHY_COVERAGE_CLASS] = { .type = NLA_U8 },
238
239         [NL80211_ATTR_IFTYPE] = { .type = NLA_U32 },
240         [NL80211_ATTR_IFINDEX] = { .type = NLA_U32 },
241         [NL80211_ATTR_IFNAME] = { .type = NLA_NUL_STRING, .len = IFNAMSIZ-1 },
242
243         [NL80211_ATTR_MAC] = { .len = ETH_ALEN },
244         [NL80211_ATTR_PREV_BSSID] = { .len = ETH_ALEN },
245
246         [NL80211_ATTR_KEY] = { .type = NLA_NESTED, },
247         [NL80211_ATTR_KEY_DATA] = { .type = NLA_BINARY,
248                                     .len = WLAN_MAX_KEY_LEN },
249         [NL80211_ATTR_KEY_IDX] = { .type = NLA_U8 },
250         [NL80211_ATTR_KEY_CIPHER] = { .type = NLA_U32 },
251         [NL80211_ATTR_KEY_DEFAULT] = { .type = NLA_FLAG },
252         [NL80211_ATTR_KEY_SEQ] = { .type = NLA_BINARY, .len = 16 },
253         [NL80211_ATTR_KEY_TYPE] = { .type = NLA_U32 },
254
255         [NL80211_ATTR_BEACON_INTERVAL] = { .type = NLA_U32 },
256         [NL80211_ATTR_DTIM_PERIOD] = { .type = NLA_U32 },
257         [NL80211_ATTR_BEACON_HEAD] = { .type = NLA_BINARY,
258                                        .len = IEEE80211_MAX_DATA_LEN },
259         [NL80211_ATTR_BEACON_TAIL] = { .type = NLA_BINARY,
260                                        .len = IEEE80211_MAX_DATA_LEN },
261         [NL80211_ATTR_STA_AID] = { .type = NLA_U16 },
262         [NL80211_ATTR_STA_FLAGS] = { .type = NLA_NESTED },
263         [NL80211_ATTR_STA_LISTEN_INTERVAL] = { .type = NLA_U16 },
264         [NL80211_ATTR_STA_SUPPORTED_RATES] = { .type = NLA_BINARY,
265                                                .len = NL80211_MAX_SUPP_RATES },
266         [NL80211_ATTR_STA_PLINK_ACTION] = { .type = NLA_U8 },
267         [NL80211_ATTR_STA_VLAN] = { .type = NLA_U32 },
268         [NL80211_ATTR_MNTR_FLAGS] = { /* NLA_NESTED can't be empty */ },
269         [NL80211_ATTR_MESH_ID] = { .type = NLA_BINARY,
270                                    .len = IEEE80211_MAX_MESH_ID_LEN },
271         [NL80211_ATTR_MPATH_NEXT_HOP] = { .type = NLA_U32 },
272
273         [NL80211_ATTR_REG_ALPHA2] = { .type = NLA_STRING, .len = 2 },
274         [NL80211_ATTR_REG_RULES] = { .type = NLA_NESTED },
275
276         [NL80211_ATTR_BSS_CTS_PROT] = { .type = NLA_U8 },
277         [NL80211_ATTR_BSS_SHORT_PREAMBLE] = { .type = NLA_U8 },
278         [NL80211_ATTR_BSS_SHORT_SLOT_TIME] = { .type = NLA_U8 },
279         [NL80211_ATTR_BSS_BASIC_RATES] = { .type = NLA_BINARY,
280                                            .len = NL80211_MAX_SUPP_RATES },
281         [NL80211_ATTR_BSS_HT_OPMODE] = { .type = NLA_U16 },
282
283         [NL80211_ATTR_MESH_CONFIG] = { .type = NLA_NESTED },
284         [NL80211_ATTR_SUPPORT_MESH_AUTH] = { .type = NLA_FLAG },
285
286         [NL80211_ATTR_HT_CAPABILITY] = { .len = NL80211_HT_CAPABILITY_LEN },
287
288         [NL80211_ATTR_MGMT_SUBTYPE] = { .type = NLA_U8 },
289         [NL80211_ATTR_IE] = { .type = NLA_BINARY,
290                               .len = IEEE80211_MAX_DATA_LEN },
291         [NL80211_ATTR_SCAN_FREQUENCIES] = { .type = NLA_NESTED },
292         [NL80211_ATTR_SCAN_SSIDS] = { .type = NLA_NESTED },
293
294         [NL80211_ATTR_SSID] = { .type = NLA_BINARY,
295                                 .len = IEEE80211_MAX_SSID_LEN },
296         [NL80211_ATTR_AUTH_TYPE] = { .type = NLA_U32 },
297         [NL80211_ATTR_REASON_CODE] = { .type = NLA_U16 },
298         [NL80211_ATTR_FREQ_FIXED] = { .type = NLA_FLAG },
299         [NL80211_ATTR_TIMED_OUT] = { .type = NLA_FLAG },
300         [NL80211_ATTR_USE_MFP] = { .type = NLA_U32 },
301         [NL80211_ATTR_STA_FLAGS2] = {
302                 .len = sizeof(struct nl80211_sta_flag_update),
303         },
304         [NL80211_ATTR_CONTROL_PORT] = { .type = NLA_FLAG },
305         [NL80211_ATTR_CONTROL_PORT_ETHERTYPE] = { .type = NLA_U16 },
306         [NL80211_ATTR_CONTROL_PORT_NO_ENCRYPT] = { .type = NLA_FLAG },
307         [NL80211_ATTR_PRIVACY] = { .type = NLA_FLAG },
308         [NL80211_ATTR_CIPHER_SUITE_GROUP] = { .type = NLA_U32 },
309         [NL80211_ATTR_WPA_VERSIONS] = { .type = NLA_U32 },
310         [NL80211_ATTR_PID] = { .type = NLA_U32 },
311         [NL80211_ATTR_4ADDR] = { .type = NLA_U8 },
312         [NL80211_ATTR_PMKID] = { .type = NLA_BINARY,
313                                  .len = WLAN_PMKID_LEN },
314         [NL80211_ATTR_DURATION] = { .type = NLA_U32 },
315         [NL80211_ATTR_COOKIE] = { .type = NLA_U64 },
316         [NL80211_ATTR_TX_RATES] = { .type = NLA_NESTED },
317         [NL80211_ATTR_FRAME] = { .type = NLA_BINARY,
318                                  .len = IEEE80211_MAX_DATA_LEN },
319         [NL80211_ATTR_FRAME_MATCH] = { .type = NLA_BINARY, },
320         [NL80211_ATTR_PS_STATE] = { .type = NLA_U32 },
321         [NL80211_ATTR_CQM] = { .type = NLA_NESTED, },
322         [NL80211_ATTR_LOCAL_STATE_CHANGE] = { .type = NLA_FLAG },
323         [NL80211_ATTR_AP_ISOLATE] = { .type = NLA_U8 },
324         [NL80211_ATTR_WIPHY_TX_POWER_SETTING] = { .type = NLA_U32 },
325         [NL80211_ATTR_WIPHY_TX_POWER_LEVEL] = { .type = NLA_U32 },
326         [NL80211_ATTR_FRAME_TYPE] = { .type = NLA_U16 },
327         [NL80211_ATTR_WIPHY_ANTENNA_TX] = { .type = NLA_U32 },
328         [NL80211_ATTR_WIPHY_ANTENNA_RX] = { .type = NLA_U32 },
329         [NL80211_ATTR_MCAST_RATE] = { .type = NLA_U32 },
330         [NL80211_ATTR_OFFCHANNEL_TX_OK] = { .type = NLA_FLAG },
331         [NL80211_ATTR_KEY_DEFAULT_TYPES] = { .type = NLA_NESTED },
332         [NL80211_ATTR_WOWLAN_TRIGGERS] = { .type = NLA_NESTED },
333         [NL80211_ATTR_STA_PLINK_STATE] = { .type = NLA_U8 },
334         [NL80211_ATTR_SCHED_SCAN_INTERVAL] = { .type = NLA_U32 },
335         [NL80211_ATTR_REKEY_DATA] = { .type = NLA_NESTED },
336         [NL80211_ATTR_SCAN_SUPP_RATES] = { .type = NLA_NESTED },
337         [NL80211_ATTR_HIDDEN_SSID] = { .type = NLA_U32 },
338         [NL80211_ATTR_IE_PROBE_RESP] = { .type = NLA_BINARY,
339                                          .len = IEEE80211_MAX_DATA_LEN },
340         [NL80211_ATTR_IE_ASSOC_RESP] = { .type = NLA_BINARY,
341                                          .len = IEEE80211_MAX_DATA_LEN },
342         [NL80211_ATTR_ROAM_SUPPORT] = { .type = NLA_FLAG },
343         [NL80211_ATTR_SCHED_SCAN_MATCH] = { .type = NLA_NESTED },
344         [NL80211_ATTR_TX_NO_CCK_RATE] = { .type = NLA_FLAG },
345         [NL80211_ATTR_TDLS_ACTION] = { .type = NLA_U8 },
346         [NL80211_ATTR_TDLS_DIALOG_TOKEN] = { .type = NLA_U8 },
347         [NL80211_ATTR_TDLS_OPERATION] = { .type = NLA_U8 },
348         [NL80211_ATTR_TDLS_SUPPORT] = { .type = NLA_FLAG },
349         [NL80211_ATTR_TDLS_EXTERNAL_SETUP] = { .type = NLA_FLAG },
350         [NL80211_ATTR_DONT_WAIT_FOR_ACK] = { .type = NLA_FLAG },
351         [NL80211_ATTR_PROBE_RESP] = { .type = NLA_BINARY,
352                                       .len = IEEE80211_MAX_DATA_LEN },
353         [NL80211_ATTR_DFS_REGION] = { .type = NLA_U8 },
354         [NL80211_ATTR_DISABLE_HT] = { .type = NLA_FLAG },
355         [NL80211_ATTR_HT_CAPABILITY_MASK] = {
356                 .len = NL80211_HT_CAPABILITY_LEN
357         },
358         [NL80211_ATTR_NOACK_MAP] = { .type = NLA_U16 },
359         [NL80211_ATTR_INACTIVITY_TIMEOUT] = { .type = NLA_U16 },
360         [NL80211_ATTR_BG_SCAN_PERIOD] = { .type = NLA_U16 },
361         [NL80211_ATTR_WDEV] = { .type = NLA_U64 },
362         [NL80211_ATTR_USER_REG_HINT_TYPE] = { .type = NLA_U32 },
363         [NL80211_ATTR_SAE_DATA] = { .type = NLA_BINARY, },
364         [NL80211_ATTR_VHT_CAPABILITY] = { .len = NL80211_VHT_CAPABILITY_LEN },
365         [NL80211_ATTR_SCAN_FLAGS] = { .type = NLA_U32 },
366         [NL80211_ATTR_P2P_CTWINDOW] = { .type = NLA_U8 },
367         [NL80211_ATTR_P2P_OPPPS] = { .type = NLA_U8 },
368 };
369
370 /* policy for the key attributes */
371 static const struct nla_policy nl80211_key_policy[NL80211_KEY_MAX + 1] = {
372         [NL80211_KEY_DATA] = { .type = NLA_BINARY, .len = WLAN_MAX_KEY_LEN },
373         [NL80211_KEY_IDX] = { .type = NLA_U8 },
374         [NL80211_KEY_CIPHER] = { .type = NLA_U32 },
375         [NL80211_KEY_SEQ] = { .type = NLA_BINARY, .len = 16 },
376         [NL80211_KEY_DEFAULT] = { .type = NLA_FLAG },
377         [NL80211_KEY_DEFAULT_MGMT] = { .type = NLA_FLAG },
378         [NL80211_KEY_TYPE] = { .type = NLA_U32 },
379         [NL80211_KEY_DEFAULT_TYPES] = { .type = NLA_NESTED },
380 };
381
382 /* policy for the key default flags */
383 static const struct nla_policy
384 nl80211_key_default_policy[NUM_NL80211_KEY_DEFAULT_TYPES] = {
385         [NL80211_KEY_DEFAULT_TYPE_UNICAST] = { .type = NLA_FLAG },
386         [NL80211_KEY_DEFAULT_TYPE_MULTICAST] = { .type = NLA_FLAG },
387 };
388
389 /* policy for WoWLAN attributes */
390 static const struct nla_policy
391 nl80211_wowlan_policy[NUM_NL80211_WOWLAN_TRIG] = {
392         [NL80211_WOWLAN_TRIG_ANY] = { .type = NLA_FLAG },
393         [NL80211_WOWLAN_TRIG_DISCONNECT] = { .type = NLA_FLAG },
394         [NL80211_WOWLAN_TRIG_MAGIC_PKT] = { .type = NLA_FLAG },
395         [NL80211_WOWLAN_TRIG_PKT_PATTERN] = { .type = NLA_NESTED },
396         [NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE] = { .type = NLA_FLAG },
397         [NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST] = { .type = NLA_FLAG },
398         [NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE] = { .type = NLA_FLAG },
399         [NL80211_WOWLAN_TRIG_RFKILL_RELEASE] = { .type = NLA_FLAG },
400 };
401
402 /* policy for GTK rekey offload attributes */
403 static const struct nla_policy
404 nl80211_rekey_policy[NUM_NL80211_REKEY_DATA] = {
405         [NL80211_REKEY_DATA_KEK] = { .len = NL80211_KEK_LEN },
406         [NL80211_REKEY_DATA_KCK] = { .len = NL80211_KCK_LEN },
407         [NL80211_REKEY_DATA_REPLAY_CTR] = { .len = NL80211_REPLAY_CTR_LEN },
408 };
409
410 static const struct nla_policy
411 nl80211_match_policy[NL80211_SCHED_SCAN_MATCH_ATTR_MAX + 1] = {
412         [NL80211_SCHED_SCAN_MATCH_ATTR_SSID] = { .type = NLA_BINARY,
413                                                  .len = IEEE80211_MAX_SSID_LEN },
414         [NL80211_SCHED_SCAN_MATCH_ATTR_RSSI] = { .type = NLA_U32 },
415 };
416
417 /* ifidx get helper */
418 static int nl80211_get_ifidx(struct netlink_callback *cb)
419 {
420         int res;
421
422         res = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
423                           nl80211_fam.attrbuf, nl80211_fam.maxattr,
424                           nl80211_policy);
425         if (res)
426                 return res;
427
428         if (!nl80211_fam.attrbuf[NL80211_ATTR_IFINDEX])
429                 return -EINVAL;
430
431         res = nla_get_u32(nl80211_fam.attrbuf[NL80211_ATTR_IFINDEX]);
432         if (!res)
433                 return -EINVAL;
434         return res;
435 }
436
437 static int nl80211_prepare_netdev_dump(struct sk_buff *skb,
438                                        struct netlink_callback *cb,
439                                        struct cfg80211_registered_device **rdev,
440                                        struct net_device **dev)
441 {
442         int ifidx = cb->args[0];
443         int err;
444
445         if (!ifidx)
446                 ifidx = nl80211_get_ifidx(cb);
447         if (ifidx < 0)
448                 return ifidx;
449
450         cb->args[0] = ifidx;
451
452         rtnl_lock();
453
454         *dev = __dev_get_by_index(sock_net(skb->sk), ifidx);
455         if (!*dev) {
456                 err = -ENODEV;
457                 goto out_rtnl;
458         }
459
460         *rdev = cfg80211_get_dev_from_ifindex(sock_net(skb->sk), ifidx);
461         if (IS_ERR(*rdev)) {
462                 err = PTR_ERR(*rdev);
463                 goto out_rtnl;
464         }
465
466         return 0;
467  out_rtnl:
468         rtnl_unlock();
469         return err;
470 }
471
472 static void nl80211_finish_netdev_dump(struct cfg80211_registered_device *rdev)
473 {
474         cfg80211_unlock_rdev(rdev);
475         rtnl_unlock();
476 }
477
478 /* IE validation */
479 static bool is_valid_ie_attr(const struct nlattr *attr)
480 {
481         const u8 *pos;
482         int len;
483
484         if (!attr)
485                 return true;
486
487         pos = nla_data(attr);
488         len = nla_len(attr);
489
490         while (len) {
491                 u8 elemlen;
492
493                 if (len < 2)
494                         return false;
495                 len -= 2;
496
497                 elemlen = pos[1];
498                 if (elemlen > len)
499                         return false;
500
501                 len -= elemlen;
502                 pos += 2 + elemlen;
503         }
504
505         return true;
506 }
507
508 /* message building helper */
509 static inline void *nl80211hdr_put(struct sk_buff *skb, u32 portid, u32 seq,
510                                    int flags, u8 cmd)
511 {
512         /* since there is no private header just add the generic one */
513         return genlmsg_put(skb, portid, seq, &nl80211_fam, flags, cmd);
514 }
515
516 static int nl80211_msg_put_channel(struct sk_buff *msg,
517                                    struct ieee80211_channel *chan)
518 {
519         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_FREQ,
520                         chan->center_freq))
521                 goto nla_put_failure;
522
523         if ((chan->flags & IEEE80211_CHAN_DISABLED) &&
524             nla_put_flag(msg, NL80211_FREQUENCY_ATTR_DISABLED))
525                 goto nla_put_failure;
526         if ((chan->flags & IEEE80211_CHAN_PASSIVE_SCAN) &&
527             nla_put_flag(msg, NL80211_FREQUENCY_ATTR_PASSIVE_SCAN))
528                 goto nla_put_failure;
529         if ((chan->flags & IEEE80211_CHAN_NO_IBSS) &&
530             nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_IBSS))
531                 goto nla_put_failure;
532         if ((chan->flags & IEEE80211_CHAN_RADAR) &&
533             nla_put_flag(msg, NL80211_FREQUENCY_ATTR_RADAR))
534                 goto nla_put_failure;
535
536         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_MAX_TX_POWER,
537                         DBM_TO_MBM(chan->max_power)))
538                 goto nla_put_failure;
539
540         return 0;
541
542  nla_put_failure:
543         return -ENOBUFS;
544 }
545
546 /* netlink command implementations */
547
548 struct key_parse {
549         struct key_params p;
550         int idx;
551         int type;
552         bool def, defmgmt;
553         bool def_uni, def_multi;
554 };
555
556 static int nl80211_parse_key_new(struct nlattr *key, struct key_parse *k)
557 {
558         struct nlattr *tb[NL80211_KEY_MAX + 1];
559         int err = nla_parse_nested(tb, NL80211_KEY_MAX, key,
560                                    nl80211_key_policy);
561         if (err)
562                 return err;
563
564         k->def = !!tb[NL80211_KEY_DEFAULT];
565         k->defmgmt = !!tb[NL80211_KEY_DEFAULT_MGMT];
566
567         if (k->def) {
568                 k->def_uni = true;
569                 k->def_multi = true;
570         }
571         if (k->defmgmt)
572                 k->def_multi = true;
573
574         if (tb[NL80211_KEY_IDX])
575                 k->idx = nla_get_u8(tb[NL80211_KEY_IDX]);
576
577         if (tb[NL80211_KEY_DATA]) {
578                 k->p.key = nla_data(tb[NL80211_KEY_DATA]);
579                 k->p.key_len = nla_len(tb[NL80211_KEY_DATA]);
580         }
581
582         if (tb[NL80211_KEY_SEQ]) {
583                 k->p.seq = nla_data(tb[NL80211_KEY_SEQ]);
584                 k->p.seq_len = nla_len(tb[NL80211_KEY_SEQ]);
585         }
586
587         if (tb[NL80211_KEY_CIPHER])
588                 k->p.cipher = nla_get_u32(tb[NL80211_KEY_CIPHER]);
589
590         if (tb[NL80211_KEY_TYPE]) {
591                 k->type = nla_get_u32(tb[NL80211_KEY_TYPE]);
592                 if (k->type < 0 || k->type >= NUM_NL80211_KEYTYPES)
593                         return -EINVAL;
594         }
595
596         if (tb[NL80211_KEY_DEFAULT_TYPES]) {
597                 struct nlattr *kdt[NUM_NL80211_KEY_DEFAULT_TYPES];
598                 err = nla_parse_nested(kdt, NUM_NL80211_KEY_DEFAULT_TYPES - 1,
599                                        tb[NL80211_KEY_DEFAULT_TYPES],
600                                        nl80211_key_default_policy);
601                 if (err)
602                         return err;
603
604                 k->def_uni = kdt[NL80211_KEY_DEFAULT_TYPE_UNICAST];
605                 k->def_multi = kdt[NL80211_KEY_DEFAULT_TYPE_MULTICAST];
606         }
607
608         return 0;
609 }
610
611 static int nl80211_parse_key_old(struct genl_info *info, struct key_parse *k)
612 {
613         if (info->attrs[NL80211_ATTR_KEY_DATA]) {
614                 k->p.key = nla_data(info->attrs[NL80211_ATTR_KEY_DATA]);
615                 k->p.key_len = nla_len(info->attrs[NL80211_ATTR_KEY_DATA]);
616         }
617
618         if (info->attrs[NL80211_ATTR_KEY_SEQ]) {
619                 k->p.seq = nla_data(info->attrs[NL80211_ATTR_KEY_SEQ]);
620                 k->p.seq_len = nla_len(info->attrs[NL80211_ATTR_KEY_SEQ]);
621         }
622
623         if (info->attrs[NL80211_ATTR_KEY_IDX])
624                 k->idx = nla_get_u8(info->attrs[NL80211_ATTR_KEY_IDX]);
625
626         if (info->attrs[NL80211_ATTR_KEY_CIPHER])
627                 k->p.cipher = nla_get_u32(info->attrs[NL80211_ATTR_KEY_CIPHER]);
628
629         k->def = !!info->attrs[NL80211_ATTR_KEY_DEFAULT];
630         k->defmgmt = !!info->attrs[NL80211_ATTR_KEY_DEFAULT_MGMT];
631
632         if (k->def) {
633                 k->def_uni = true;
634                 k->def_multi = true;
635         }
636         if (k->defmgmt)
637                 k->def_multi = true;
638
639         if (info->attrs[NL80211_ATTR_KEY_TYPE]) {
640                 k->type = nla_get_u32(info->attrs[NL80211_ATTR_KEY_TYPE]);
641                 if (k->type < 0 || k->type >= NUM_NL80211_KEYTYPES)
642                         return -EINVAL;
643         }
644
645         if (info->attrs[NL80211_ATTR_KEY_DEFAULT_TYPES]) {
646                 struct nlattr *kdt[NUM_NL80211_KEY_DEFAULT_TYPES];
647                 int err = nla_parse_nested(
648                                 kdt, NUM_NL80211_KEY_DEFAULT_TYPES - 1,
649                                 info->attrs[NL80211_ATTR_KEY_DEFAULT_TYPES],
650                                 nl80211_key_default_policy);
651                 if (err)
652                         return err;
653
654                 k->def_uni = kdt[NL80211_KEY_DEFAULT_TYPE_UNICAST];
655                 k->def_multi = kdt[NL80211_KEY_DEFAULT_TYPE_MULTICAST];
656         }
657
658         return 0;
659 }
660
661 static int nl80211_parse_key(struct genl_info *info, struct key_parse *k)
662 {
663         int err;
664
665         memset(k, 0, sizeof(*k));
666         k->idx = -1;
667         k->type = -1;
668
669         if (info->attrs[NL80211_ATTR_KEY])
670                 err = nl80211_parse_key_new(info->attrs[NL80211_ATTR_KEY], k);
671         else
672                 err = nl80211_parse_key_old(info, k);
673
674         if (err)
675                 return err;
676
677         if (k->def && k->defmgmt)
678                 return -EINVAL;
679
680         if (k->defmgmt) {
681                 if (k->def_uni || !k->def_multi)
682                         return -EINVAL;
683         }
684
685         if (k->idx != -1) {
686                 if (k->defmgmt) {
687                         if (k->idx < 4 || k->idx > 5)
688                                 return -EINVAL;
689                 } else if (k->def) {
690                         if (k->idx < 0 || k->idx > 3)
691                                 return -EINVAL;
692                 } else {
693                         if (k->idx < 0 || k->idx > 5)
694                                 return -EINVAL;
695                 }
696         }
697
698         return 0;
699 }
700
701 static struct cfg80211_cached_keys *
702 nl80211_parse_connkeys(struct cfg80211_registered_device *rdev,
703                        struct nlattr *keys, bool *no_ht)
704 {
705         struct key_parse parse;
706         struct nlattr *key;
707         struct cfg80211_cached_keys *result;
708         int rem, err, def = 0;
709
710         result = kzalloc(sizeof(*result), GFP_KERNEL);
711         if (!result)
712                 return ERR_PTR(-ENOMEM);
713
714         result->def = -1;
715         result->defmgmt = -1;
716
717         nla_for_each_nested(key, keys, rem) {
718                 memset(&parse, 0, sizeof(parse));
719                 parse.idx = -1;
720
721                 err = nl80211_parse_key_new(key, &parse);
722                 if (err)
723                         goto error;
724                 err = -EINVAL;
725                 if (!parse.p.key)
726                         goto error;
727                 if (parse.idx < 0 || parse.idx > 4)
728                         goto error;
729                 if (parse.def) {
730                         if (def)
731                                 goto error;
732                         def = 1;
733                         result->def = parse.idx;
734                         if (!parse.def_uni || !parse.def_multi)
735                                 goto error;
736                 } else if (parse.defmgmt)
737                         goto error;
738                 err = cfg80211_validate_key_settings(rdev, &parse.p,
739                                                      parse.idx, false, NULL);
740                 if (err)
741                         goto error;
742                 result->params[parse.idx].cipher = parse.p.cipher;
743                 result->params[parse.idx].key_len = parse.p.key_len;
744                 result->params[parse.idx].key = result->data[parse.idx];
745                 memcpy(result->data[parse.idx], parse.p.key, parse.p.key_len);
746
747                 if (parse.p.cipher == WLAN_CIPHER_SUITE_WEP40 ||
748                     parse.p.cipher == WLAN_CIPHER_SUITE_WEP104) {
749                         if (no_ht)
750                                 *no_ht = true;
751                 }
752         }
753
754         return result;
755  error:
756         kfree(result);
757         return ERR_PTR(err);
758 }
759
760 static int nl80211_key_allowed(struct wireless_dev *wdev)
761 {
762         ASSERT_WDEV_LOCK(wdev);
763
764         switch (wdev->iftype) {
765         case NL80211_IFTYPE_AP:
766         case NL80211_IFTYPE_AP_VLAN:
767         case NL80211_IFTYPE_P2P_GO:
768         case NL80211_IFTYPE_MESH_POINT:
769                 break;
770         case NL80211_IFTYPE_ADHOC:
771                 if (!wdev->current_bss)
772                         return -ENOLINK;
773                 break;
774         case NL80211_IFTYPE_STATION:
775         case NL80211_IFTYPE_P2P_CLIENT:
776                 if (wdev->sme_state != CFG80211_SME_CONNECTED)
777                         return -ENOLINK;
778                 break;
779         default:
780                 return -EINVAL;
781         }
782
783         return 0;
784 }
785
786 static int nl80211_put_iftypes(struct sk_buff *msg, u32 attr, u16 ifmodes)
787 {
788         struct nlattr *nl_modes = nla_nest_start(msg, attr);
789         int i;
790
791         if (!nl_modes)
792                 goto nla_put_failure;
793
794         i = 0;
795         while (ifmodes) {
796                 if ((ifmodes & 1) && nla_put_flag(msg, i))
797                         goto nla_put_failure;
798                 ifmodes >>= 1;
799                 i++;
800         }
801
802         nla_nest_end(msg, nl_modes);
803         return 0;
804
805 nla_put_failure:
806         return -ENOBUFS;
807 }
808
809 static int nl80211_put_iface_combinations(struct wiphy *wiphy,
810                                           struct sk_buff *msg)
811 {
812         struct nlattr *nl_combis;
813         int i, j;
814
815         nl_combis = nla_nest_start(msg,
816                                 NL80211_ATTR_INTERFACE_COMBINATIONS);
817         if (!nl_combis)
818                 goto nla_put_failure;
819
820         for (i = 0; i < wiphy->n_iface_combinations; i++) {
821                 const struct ieee80211_iface_combination *c;
822                 struct nlattr *nl_combi, *nl_limits;
823
824                 c = &wiphy->iface_combinations[i];
825
826                 nl_combi = nla_nest_start(msg, i + 1);
827                 if (!nl_combi)
828                         goto nla_put_failure;
829
830                 nl_limits = nla_nest_start(msg, NL80211_IFACE_COMB_LIMITS);
831                 if (!nl_limits)
832                         goto nla_put_failure;
833
834                 for (j = 0; j < c->n_limits; j++) {
835                         struct nlattr *nl_limit;
836
837                         nl_limit = nla_nest_start(msg, j + 1);
838                         if (!nl_limit)
839                                 goto nla_put_failure;
840                         if (nla_put_u32(msg, NL80211_IFACE_LIMIT_MAX,
841                                         c->limits[j].max))
842                                 goto nla_put_failure;
843                         if (nl80211_put_iftypes(msg, NL80211_IFACE_LIMIT_TYPES,
844                                                 c->limits[j].types))
845                                 goto nla_put_failure;
846                         nla_nest_end(msg, nl_limit);
847                 }
848
849                 nla_nest_end(msg, nl_limits);
850
851                 if (c->beacon_int_infra_match &&
852                     nla_put_flag(msg, NL80211_IFACE_COMB_STA_AP_BI_MATCH))
853                         goto nla_put_failure;
854                 if (nla_put_u32(msg, NL80211_IFACE_COMB_NUM_CHANNELS,
855                                 c->num_different_channels) ||
856                     nla_put_u32(msg, NL80211_IFACE_COMB_MAXNUM,
857                                 c->max_interfaces))
858                         goto nla_put_failure;
859
860                 nla_nest_end(msg, nl_combi);
861         }
862
863         nla_nest_end(msg, nl_combis);
864
865         return 0;
866 nla_put_failure:
867         return -ENOBUFS;
868 }
869
870 static int nl80211_send_wiphy(struct sk_buff *msg, u32 portid, u32 seq, int flags,
871                               struct cfg80211_registered_device *dev)
872 {
873         void *hdr;
874         struct nlattr *nl_bands, *nl_band;
875         struct nlattr *nl_freqs, *nl_freq;
876         struct nlattr *nl_rates, *nl_rate;
877         struct nlattr *nl_cmds;
878         enum ieee80211_band band;
879         struct ieee80211_channel *chan;
880         struct ieee80211_rate *rate;
881         int i;
882         const struct ieee80211_txrx_stypes *mgmt_stypes =
883                                 dev->wiphy.mgmt_stypes;
884
885         hdr = nl80211hdr_put(msg, portid, seq, flags, NL80211_CMD_NEW_WIPHY);
886         if (!hdr)
887                 return -1;
888
889         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, dev->wiphy_idx) ||
890             nla_put_string(msg, NL80211_ATTR_WIPHY_NAME, wiphy_name(&dev->wiphy)) ||
891             nla_put_u32(msg, NL80211_ATTR_GENERATION,
892                         cfg80211_rdev_list_generation) ||
893             nla_put_u8(msg, NL80211_ATTR_WIPHY_RETRY_SHORT,
894                        dev->wiphy.retry_short) ||
895             nla_put_u8(msg, NL80211_ATTR_WIPHY_RETRY_LONG,
896                        dev->wiphy.retry_long) ||
897             nla_put_u32(msg, NL80211_ATTR_WIPHY_FRAG_THRESHOLD,
898                         dev->wiphy.frag_threshold) ||
899             nla_put_u32(msg, NL80211_ATTR_WIPHY_RTS_THRESHOLD,
900                         dev->wiphy.rts_threshold) ||
901             nla_put_u8(msg, NL80211_ATTR_WIPHY_COVERAGE_CLASS,
902                        dev->wiphy.coverage_class) ||
903             nla_put_u8(msg, NL80211_ATTR_MAX_NUM_SCAN_SSIDS,
904                        dev->wiphy.max_scan_ssids) ||
905             nla_put_u8(msg, NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS,
906                        dev->wiphy.max_sched_scan_ssids) ||
907             nla_put_u16(msg, NL80211_ATTR_MAX_SCAN_IE_LEN,
908                         dev->wiphy.max_scan_ie_len) ||
909             nla_put_u16(msg, NL80211_ATTR_MAX_SCHED_SCAN_IE_LEN,
910                         dev->wiphy.max_sched_scan_ie_len) ||
911             nla_put_u8(msg, NL80211_ATTR_MAX_MATCH_SETS,
912                        dev->wiphy.max_match_sets))
913                 goto nla_put_failure;
914
915         if ((dev->wiphy.flags & WIPHY_FLAG_IBSS_RSN) &&
916             nla_put_flag(msg, NL80211_ATTR_SUPPORT_IBSS_RSN))
917                 goto nla_put_failure;
918         if ((dev->wiphy.flags & WIPHY_FLAG_MESH_AUTH) &&
919             nla_put_flag(msg, NL80211_ATTR_SUPPORT_MESH_AUTH))
920                 goto nla_put_failure;
921         if ((dev->wiphy.flags & WIPHY_FLAG_AP_UAPSD) &&
922             nla_put_flag(msg, NL80211_ATTR_SUPPORT_AP_UAPSD))
923                 goto nla_put_failure;
924         if ((dev->wiphy.flags & WIPHY_FLAG_SUPPORTS_FW_ROAM) &&
925             nla_put_flag(msg, NL80211_ATTR_ROAM_SUPPORT))
926                 goto nla_put_failure;
927         if ((dev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) &&
928             nla_put_flag(msg, NL80211_ATTR_TDLS_SUPPORT))
929                 goto nla_put_failure;
930         if ((dev->wiphy.flags & WIPHY_FLAG_TDLS_EXTERNAL_SETUP) &&
931             nla_put_flag(msg, NL80211_ATTR_TDLS_EXTERNAL_SETUP))
932                 goto nla_put_failure;
933
934         if (nla_put(msg, NL80211_ATTR_CIPHER_SUITES,
935                     sizeof(u32) * dev->wiphy.n_cipher_suites,
936                     dev->wiphy.cipher_suites))
937                 goto nla_put_failure;
938
939         if (nla_put_u8(msg, NL80211_ATTR_MAX_NUM_PMKIDS,
940                        dev->wiphy.max_num_pmkids))
941                 goto nla_put_failure;
942
943         if ((dev->wiphy.flags & WIPHY_FLAG_CONTROL_PORT_PROTOCOL) &&
944             nla_put_flag(msg, NL80211_ATTR_CONTROL_PORT_ETHERTYPE))
945                 goto nla_put_failure;
946
947         if (nla_put_u32(msg, NL80211_ATTR_WIPHY_ANTENNA_AVAIL_TX,
948                         dev->wiphy.available_antennas_tx) ||
949             nla_put_u32(msg, NL80211_ATTR_WIPHY_ANTENNA_AVAIL_RX,
950                         dev->wiphy.available_antennas_rx))
951                 goto nla_put_failure;
952
953         if ((dev->wiphy.flags & WIPHY_FLAG_AP_PROBE_RESP_OFFLOAD) &&
954             nla_put_u32(msg, NL80211_ATTR_PROBE_RESP_OFFLOAD,
955                         dev->wiphy.probe_resp_offload))
956                 goto nla_put_failure;
957
958         if ((dev->wiphy.available_antennas_tx ||
959              dev->wiphy.available_antennas_rx) && dev->ops->get_antenna) {
960                 u32 tx_ant = 0, rx_ant = 0;
961                 int res;
962                 res = rdev_get_antenna(dev, &tx_ant, &rx_ant);
963                 if (!res) {
964                         if (nla_put_u32(msg, NL80211_ATTR_WIPHY_ANTENNA_TX,
965                                         tx_ant) ||
966                             nla_put_u32(msg, NL80211_ATTR_WIPHY_ANTENNA_RX,
967                                         rx_ant))
968                                 goto nla_put_failure;
969                 }
970         }
971
972         if (nl80211_put_iftypes(msg, NL80211_ATTR_SUPPORTED_IFTYPES,
973                                 dev->wiphy.interface_modes))
974                 goto nla_put_failure;
975
976         nl_bands = nla_nest_start(msg, NL80211_ATTR_WIPHY_BANDS);
977         if (!nl_bands)
978                 goto nla_put_failure;
979
980         for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
981                 if (!dev->wiphy.bands[band])
982                         continue;
983
984                 nl_band = nla_nest_start(msg, band);
985                 if (!nl_band)
986                         goto nla_put_failure;
987
988                 /* add HT info */
989                 if (dev->wiphy.bands[band]->ht_cap.ht_supported &&
990                     (nla_put(msg, NL80211_BAND_ATTR_HT_MCS_SET,
991                              sizeof(dev->wiphy.bands[band]->ht_cap.mcs),
992                              &dev->wiphy.bands[band]->ht_cap.mcs) ||
993                      nla_put_u16(msg, NL80211_BAND_ATTR_HT_CAPA,
994                                  dev->wiphy.bands[band]->ht_cap.cap) ||
995                      nla_put_u8(msg, NL80211_BAND_ATTR_HT_AMPDU_FACTOR,
996                                 dev->wiphy.bands[band]->ht_cap.ampdu_factor) ||
997                      nla_put_u8(msg, NL80211_BAND_ATTR_HT_AMPDU_DENSITY,
998                                 dev->wiphy.bands[band]->ht_cap.ampdu_density)))
999                         goto nla_put_failure;
1000
1001                 /* add VHT info */
1002                 if (dev->wiphy.bands[band]->vht_cap.vht_supported &&
1003                     (nla_put(msg, NL80211_BAND_ATTR_VHT_MCS_SET,
1004                              sizeof(dev->wiphy.bands[band]->vht_cap.vht_mcs),
1005                              &dev->wiphy.bands[band]->vht_cap.vht_mcs) ||
1006                      nla_put_u32(msg, NL80211_BAND_ATTR_VHT_CAPA,
1007                                  dev->wiphy.bands[band]->vht_cap.cap)))
1008                         goto nla_put_failure;
1009
1010                 /* add frequencies */
1011                 nl_freqs = nla_nest_start(msg, NL80211_BAND_ATTR_FREQS);
1012                 if (!nl_freqs)
1013                         goto nla_put_failure;
1014
1015                 for (i = 0; i < dev->wiphy.bands[band]->n_channels; i++) {
1016                         nl_freq = nla_nest_start(msg, i);
1017                         if (!nl_freq)
1018                                 goto nla_put_failure;
1019
1020                         chan = &dev->wiphy.bands[band]->channels[i];
1021
1022                         if (nl80211_msg_put_channel(msg, chan))
1023                                 goto nla_put_failure;
1024
1025                         nla_nest_end(msg, nl_freq);
1026                 }
1027
1028                 nla_nest_end(msg, nl_freqs);
1029
1030                 /* add bitrates */
1031                 nl_rates = nla_nest_start(msg, NL80211_BAND_ATTR_RATES);
1032                 if (!nl_rates)
1033                         goto nla_put_failure;
1034
1035                 for (i = 0; i < dev->wiphy.bands[band]->n_bitrates; i++) {
1036                         nl_rate = nla_nest_start(msg, i);
1037                         if (!nl_rate)
1038                                 goto nla_put_failure;
1039
1040                         rate = &dev->wiphy.bands[band]->bitrates[i];
1041                         if (nla_put_u32(msg, NL80211_BITRATE_ATTR_RATE,
1042                                         rate->bitrate))
1043                                 goto nla_put_failure;
1044                         if ((rate->flags & IEEE80211_RATE_SHORT_PREAMBLE) &&
1045                             nla_put_flag(msg,
1046                                          NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE))
1047                                 goto nla_put_failure;
1048
1049                         nla_nest_end(msg, nl_rate);
1050                 }
1051
1052                 nla_nest_end(msg, nl_rates);
1053
1054                 nla_nest_end(msg, nl_band);
1055         }
1056         nla_nest_end(msg, nl_bands);
1057
1058         nl_cmds = nla_nest_start(msg, NL80211_ATTR_SUPPORTED_COMMANDS);
1059         if (!nl_cmds)
1060                 goto nla_put_failure;
1061
1062         i = 0;
1063 #define CMD(op, n)                                              \
1064          do {                                                   \
1065                 if (dev->ops->op) {                             \
1066                         i++;                                    \
1067                         if (nla_put_u32(msg, i, NL80211_CMD_ ## n)) \
1068                                 goto nla_put_failure;           \
1069                 }                                               \
1070         } while (0)
1071
1072         CMD(add_virtual_intf, NEW_INTERFACE);
1073         CMD(change_virtual_intf, SET_INTERFACE);
1074         CMD(add_key, NEW_KEY);
1075         CMD(start_ap, START_AP);
1076         CMD(add_station, NEW_STATION);
1077         CMD(add_mpath, NEW_MPATH);
1078         CMD(update_mesh_config, SET_MESH_CONFIG);
1079         CMD(change_bss, SET_BSS);
1080         CMD(auth, AUTHENTICATE);
1081         CMD(assoc, ASSOCIATE);
1082         CMD(deauth, DEAUTHENTICATE);
1083         CMD(disassoc, DISASSOCIATE);
1084         CMD(join_ibss, JOIN_IBSS);
1085         CMD(join_mesh, JOIN_MESH);
1086         CMD(set_pmksa, SET_PMKSA);
1087         CMD(del_pmksa, DEL_PMKSA);
1088         CMD(flush_pmksa, FLUSH_PMKSA);
1089         if (dev->wiphy.flags & WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL)
1090                 CMD(remain_on_channel, REMAIN_ON_CHANNEL);
1091         CMD(set_bitrate_mask, SET_TX_BITRATE_MASK);
1092         CMD(mgmt_tx, FRAME);
1093         CMD(mgmt_tx_cancel_wait, FRAME_WAIT_CANCEL);
1094         if (dev->wiphy.flags & WIPHY_FLAG_NETNS_OK) {
1095                 i++;
1096                 if (nla_put_u32(msg, i, NL80211_CMD_SET_WIPHY_NETNS))
1097                         goto nla_put_failure;
1098         }
1099         if (dev->ops->set_monitor_channel || dev->ops->start_ap ||
1100             dev->ops->join_mesh) {
1101                 i++;
1102                 if (nla_put_u32(msg, i, NL80211_CMD_SET_CHANNEL))
1103                         goto nla_put_failure;
1104         }
1105         CMD(set_wds_peer, SET_WDS_PEER);
1106         if (dev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) {
1107                 CMD(tdls_mgmt, TDLS_MGMT);
1108                 CMD(tdls_oper, TDLS_OPER);
1109         }
1110         if (dev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN)
1111                 CMD(sched_scan_start, START_SCHED_SCAN);
1112         CMD(probe_client, PROBE_CLIENT);
1113         CMD(set_noack_map, SET_NOACK_MAP);
1114         if (dev->wiphy.flags & WIPHY_FLAG_REPORTS_OBSS) {
1115                 i++;
1116                 if (nla_put_u32(msg, i, NL80211_CMD_REGISTER_BEACONS))
1117                         goto nla_put_failure;
1118         }
1119         CMD(start_p2p_device, START_P2P_DEVICE);
1120         CMD(set_mcast_rate, SET_MCAST_RATE);
1121
1122 #ifdef CONFIG_NL80211_TESTMODE
1123         CMD(testmode_cmd, TESTMODE);
1124 #endif
1125
1126 #undef CMD
1127
1128         if (dev->ops->connect || dev->ops->auth) {
1129                 i++;
1130                 if (nla_put_u32(msg, i, NL80211_CMD_CONNECT))
1131                         goto nla_put_failure;
1132         }
1133
1134         if (dev->ops->disconnect || dev->ops->deauth) {
1135                 i++;
1136                 if (nla_put_u32(msg, i, NL80211_CMD_DISCONNECT))
1137                         goto nla_put_failure;
1138         }
1139
1140         nla_nest_end(msg, nl_cmds);
1141
1142         if (dev->ops->remain_on_channel &&
1143             (dev->wiphy.flags & WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL) &&
1144             nla_put_u32(msg, NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION,
1145                         dev->wiphy.max_remain_on_channel_duration))
1146                 goto nla_put_failure;
1147
1148         if ((dev->wiphy.flags & WIPHY_FLAG_OFFCHAN_TX) &&
1149             nla_put_flag(msg, NL80211_ATTR_OFFCHANNEL_TX_OK))
1150                 goto nla_put_failure;
1151
1152         if (mgmt_stypes) {
1153                 u16 stypes;
1154                 struct nlattr *nl_ftypes, *nl_ifs;
1155                 enum nl80211_iftype ift;
1156
1157                 nl_ifs = nla_nest_start(msg, NL80211_ATTR_TX_FRAME_TYPES);
1158                 if (!nl_ifs)
1159                         goto nla_put_failure;
1160
1161                 for (ift = 0; ift < NUM_NL80211_IFTYPES; ift++) {
1162                         nl_ftypes = nla_nest_start(msg, ift);
1163                         if (!nl_ftypes)
1164                                 goto nla_put_failure;
1165                         i = 0;
1166                         stypes = mgmt_stypes[ift].tx;
1167                         while (stypes) {
1168                                 if ((stypes & 1) &&
1169                                     nla_put_u16(msg, NL80211_ATTR_FRAME_TYPE,
1170                                                 (i << 4) | IEEE80211_FTYPE_MGMT))
1171                                         goto nla_put_failure;
1172                                 stypes >>= 1;
1173                                 i++;
1174                         }
1175                         nla_nest_end(msg, nl_ftypes);
1176                 }
1177
1178                 nla_nest_end(msg, nl_ifs);
1179
1180                 nl_ifs = nla_nest_start(msg, NL80211_ATTR_RX_FRAME_TYPES);
1181                 if (!nl_ifs)
1182                         goto nla_put_failure;
1183
1184                 for (ift = 0; ift < NUM_NL80211_IFTYPES; ift++) {
1185                         nl_ftypes = nla_nest_start(msg, ift);
1186                         if (!nl_ftypes)
1187                                 goto nla_put_failure;
1188                         i = 0;
1189                         stypes = mgmt_stypes[ift].rx;
1190                         while (stypes) {
1191                                 if ((stypes & 1) &&
1192                                     nla_put_u16(msg, NL80211_ATTR_FRAME_TYPE,
1193                                                 (i << 4) | IEEE80211_FTYPE_MGMT))
1194                                         goto nla_put_failure;
1195                                 stypes >>= 1;
1196                                 i++;
1197                         }
1198                         nla_nest_end(msg, nl_ftypes);
1199                 }
1200                 nla_nest_end(msg, nl_ifs);
1201         }
1202
1203 #ifdef CONFIG_PM
1204         if (dev->wiphy.wowlan.flags || dev->wiphy.wowlan.n_patterns) {
1205                 struct nlattr *nl_wowlan;
1206
1207                 nl_wowlan = nla_nest_start(msg,
1208                                 NL80211_ATTR_WOWLAN_TRIGGERS_SUPPORTED);
1209                 if (!nl_wowlan)
1210                         goto nla_put_failure;
1211
1212                 if (((dev->wiphy.wowlan.flags & WIPHY_WOWLAN_ANY) &&
1213                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_ANY)) ||
1214                     ((dev->wiphy.wowlan.flags & WIPHY_WOWLAN_DISCONNECT) &&
1215                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_DISCONNECT)) ||
1216                     ((dev->wiphy.wowlan.flags & WIPHY_WOWLAN_MAGIC_PKT) &&
1217                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_MAGIC_PKT)) ||
1218                     ((dev->wiphy.wowlan.flags & WIPHY_WOWLAN_SUPPORTS_GTK_REKEY) &&
1219                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_SUPPORTED)) ||
1220                     ((dev->wiphy.wowlan.flags & WIPHY_WOWLAN_GTK_REKEY_FAILURE) &&
1221                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE)) ||
1222                     ((dev->wiphy.wowlan.flags & WIPHY_WOWLAN_EAP_IDENTITY_REQ) &&
1223                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST)) ||
1224                     ((dev->wiphy.wowlan.flags & WIPHY_WOWLAN_4WAY_HANDSHAKE) &&
1225                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE)) ||
1226                     ((dev->wiphy.wowlan.flags & WIPHY_WOWLAN_RFKILL_RELEASE) &&
1227                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_RFKILL_RELEASE)))
1228                     goto nla_put_failure;
1229                 if (dev->wiphy.wowlan.n_patterns) {
1230                         struct nl80211_wowlan_pattern_support pat = {
1231                                 .max_patterns = dev->wiphy.wowlan.n_patterns,
1232                                 .min_pattern_len =
1233                                         dev->wiphy.wowlan.pattern_min_len,
1234                                 .max_pattern_len =
1235                                         dev->wiphy.wowlan.pattern_max_len,
1236                         };
1237                         if (nla_put(msg, NL80211_WOWLAN_TRIG_PKT_PATTERN,
1238                                     sizeof(pat), &pat))
1239                                 goto nla_put_failure;
1240                 }
1241
1242                 nla_nest_end(msg, nl_wowlan);
1243         }
1244 #endif
1245
1246         if (nl80211_put_iftypes(msg, NL80211_ATTR_SOFTWARE_IFTYPES,
1247                                 dev->wiphy.software_iftypes))
1248                 goto nla_put_failure;
1249
1250         if (nl80211_put_iface_combinations(&dev->wiphy, msg))
1251                 goto nla_put_failure;
1252
1253         if ((dev->wiphy.flags & WIPHY_FLAG_HAVE_AP_SME) &&
1254             nla_put_u32(msg, NL80211_ATTR_DEVICE_AP_SME,
1255                         dev->wiphy.ap_sme_capa))
1256                 goto nla_put_failure;
1257
1258         if (nla_put_u32(msg, NL80211_ATTR_FEATURE_FLAGS,
1259                         dev->wiphy.features))
1260                 goto nla_put_failure;
1261
1262         if (dev->wiphy.ht_capa_mod_mask &&
1263             nla_put(msg, NL80211_ATTR_HT_CAPABILITY_MASK,
1264                     sizeof(*dev->wiphy.ht_capa_mod_mask),
1265                     dev->wiphy.ht_capa_mod_mask))
1266                 goto nla_put_failure;
1267
1268         return genlmsg_end(msg, hdr);
1269
1270  nla_put_failure:
1271         genlmsg_cancel(msg, hdr);
1272         return -EMSGSIZE;
1273 }
1274
1275 static int nl80211_dump_wiphy(struct sk_buff *skb, struct netlink_callback *cb)
1276 {
1277         int idx = 0;
1278         int start = cb->args[0];
1279         struct cfg80211_registered_device *dev;
1280
1281         mutex_lock(&cfg80211_mutex);
1282         list_for_each_entry(dev, &cfg80211_rdev_list, list) {
1283                 if (!net_eq(wiphy_net(&dev->wiphy), sock_net(skb->sk)))
1284                         continue;
1285                 if (++idx <= start)
1286                         continue;
1287                 if (nl80211_send_wiphy(skb, NETLINK_CB(cb->skb).portid,
1288                                        cb->nlh->nlmsg_seq, NLM_F_MULTI,
1289                                        dev) < 0) {
1290                         idx--;
1291                         break;
1292                 }
1293         }
1294         mutex_unlock(&cfg80211_mutex);
1295
1296         cb->args[0] = idx;
1297
1298         return skb->len;
1299 }
1300
1301 static int nl80211_get_wiphy(struct sk_buff *skb, struct genl_info *info)
1302 {
1303         struct sk_buff *msg;
1304         struct cfg80211_registered_device *dev = info->user_ptr[0];
1305
1306         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1307         if (!msg)
1308                 return -ENOMEM;
1309
1310         if (nl80211_send_wiphy(msg, info->snd_portid, info->snd_seq, 0, dev) < 0) {
1311                 nlmsg_free(msg);
1312                 return -ENOBUFS;
1313         }
1314
1315         return genlmsg_reply(msg, info);
1316 }
1317
1318 static const struct nla_policy txq_params_policy[NL80211_TXQ_ATTR_MAX + 1] = {
1319         [NL80211_TXQ_ATTR_QUEUE]                = { .type = NLA_U8 },
1320         [NL80211_TXQ_ATTR_TXOP]                 = { .type = NLA_U16 },
1321         [NL80211_TXQ_ATTR_CWMIN]                = { .type = NLA_U16 },
1322         [NL80211_TXQ_ATTR_CWMAX]                = { .type = NLA_U16 },
1323         [NL80211_TXQ_ATTR_AIFS]                 = { .type = NLA_U8 },
1324 };
1325
1326 static int parse_txq_params(struct nlattr *tb[],
1327                             struct ieee80211_txq_params *txq_params)
1328 {
1329         if (!tb[NL80211_TXQ_ATTR_AC] || !tb[NL80211_TXQ_ATTR_TXOP] ||
1330             !tb[NL80211_TXQ_ATTR_CWMIN] || !tb[NL80211_TXQ_ATTR_CWMAX] ||
1331             !tb[NL80211_TXQ_ATTR_AIFS])
1332                 return -EINVAL;
1333
1334         txq_params->ac = nla_get_u8(tb[NL80211_TXQ_ATTR_AC]);
1335         txq_params->txop = nla_get_u16(tb[NL80211_TXQ_ATTR_TXOP]);
1336         txq_params->cwmin = nla_get_u16(tb[NL80211_TXQ_ATTR_CWMIN]);
1337         txq_params->cwmax = nla_get_u16(tb[NL80211_TXQ_ATTR_CWMAX]);
1338         txq_params->aifs = nla_get_u8(tb[NL80211_TXQ_ATTR_AIFS]);
1339
1340         if (txq_params->ac >= NL80211_NUM_ACS)
1341                 return -EINVAL;
1342
1343         return 0;
1344 }
1345
1346 static bool nl80211_can_set_dev_channel(struct wireless_dev *wdev)
1347 {
1348         /*
1349          * You can only set the channel explicitly for WDS interfaces,
1350          * all others have their channel managed via their respective
1351          * "establish a connection" command (connect, join, ...)
1352          *
1353          * For AP/GO and mesh mode, the channel can be set with the
1354          * channel userspace API, but is only stored and passed to the
1355          * low-level driver when the AP starts or the mesh is joined.
1356          * This is for backward compatibility, userspace can also give
1357          * the channel in the start-ap or join-mesh commands instead.
1358          *
1359          * Monitors are special as they are normally slaved to
1360          * whatever else is going on, so they have their own special
1361          * operation to set the monitor channel if possible.
1362          */
1363         return !wdev ||
1364                 wdev->iftype == NL80211_IFTYPE_AP ||
1365                 wdev->iftype == NL80211_IFTYPE_MESH_POINT ||
1366                 wdev->iftype == NL80211_IFTYPE_MONITOR ||
1367                 wdev->iftype == NL80211_IFTYPE_P2P_GO;
1368 }
1369
1370 static int nl80211_parse_chandef(struct cfg80211_registered_device *rdev,
1371                                  struct genl_info *info,
1372                                  struct cfg80211_chan_def *chandef)
1373 {
1374         u32 control_freq;
1375
1376         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ])
1377                 return -EINVAL;
1378
1379         control_freq = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]);
1380
1381         chandef->chan = ieee80211_get_channel(&rdev->wiphy, control_freq);
1382         chandef->width = NL80211_CHAN_WIDTH_20_NOHT;
1383         chandef->center_freq1 = control_freq;
1384         chandef->center_freq2 = 0;
1385
1386         /* Primary channel not allowed */
1387         if (!chandef->chan || chandef->chan->flags & IEEE80211_CHAN_DISABLED)
1388                 return -EINVAL;
1389
1390         if (info->attrs[NL80211_ATTR_WIPHY_CHANNEL_TYPE]) {
1391                 enum nl80211_channel_type chantype;
1392
1393                 chantype = nla_get_u32(
1394                                 info->attrs[NL80211_ATTR_WIPHY_CHANNEL_TYPE]);
1395
1396                 switch (chantype) {
1397                 case NL80211_CHAN_NO_HT:
1398                 case NL80211_CHAN_HT20:
1399                 case NL80211_CHAN_HT40PLUS:
1400                 case NL80211_CHAN_HT40MINUS:
1401                         cfg80211_chandef_create(chandef, chandef->chan,
1402                                                 chantype);
1403                         break;
1404                 default:
1405                         return -EINVAL;
1406                 }
1407         } else if (info->attrs[NL80211_ATTR_CHANNEL_WIDTH]) {
1408                 chandef->width =
1409                         nla_get_u32(info->attrs[NL80211_ATTR_CHANNEL_WIDTH]);
1410                 if (info->attrs[NL80211_ATTR_CENTER_FREQ1])
1411                         chandef->center_freq1 =
1412                                 nla_get_u32(
1413                                         info->attrs[NL80211_ATTR_CENTER_FREQ1]);
1414                 if (info->attrs[NL80211_ATTR_CENTER_FREQ2])
1415                         chandef->center_freq2 =
1416                                 nla_get_u32(
1417                                         info->attrs[NL80211_ATTR_CENTER_FREQ2]);
1418         }
1419
1420         if (!cfg80211_chandef_valid(chandef))
1421                 return -EINVAL;
1422
1423         if (!cfg80211_chandef_usable(&rdev->wiphy, chandef,
1424                                      IEEE80211_CHAN_DISABLED))
1425                 return -EINVAL;
1426
1427         return 0;
1428 }
1429
1430 static int __nl80211_set_channel(struct cfg80211_registered_device *rdev,
1431                                  struct wireless_dev *wdev,
1432                                  struct genl_info *info)
1433 {
1434         struct cfg80211_chan_def chandef;
1435         int result;
1436         enum nl80211_iftype iftype = NL80211_IFTYPE_MONITOR;
1437
1438         if (wdev)
1439                 iftype = wdev->iftype;
1440
1441         if (!nl80211_can_set_dev_channel(wdev))
1442                 return -EOPNOTSUPP;
1443
1444         result = nl80211_parse_chandef(rdev, info, &chandef);
1445         if (result)
1446                 return result;
1447
1448         mutex_lock(&rdev->devlist_mtx);
1449         switch (iftype) {
1450         case NL80211_IFTYPE_AP:
1451         case NL80211_IFTYPE_P2P_GO:
1452                 if (wdev->beacon_interval) {
1453                         result = -EBUSY;
1454                         break;
1455                 }
1456                 if (!cfg80211_reg_can_beacon(&rdev->wiphy, &chandef)) {
1457                         result = -EINVAL;
1458                         break;
1459                 }
1460                 wdev->preset_chandef = chandef;
1461                 result = 0;
1462                 break;
1463         case NL80211_IFTYPE_MESH_POINT:
1464                 result = cfg80211_set_mesh_channel(rdev, wdev, &chandef);
1465                 break;
1466         case NL80211_IFTYPE_MONITOR:
1467                 result = cfg80211_set_monitor_channel(rdev, &chandef);
1468                 break;
1469         default:
1470                 result = -EINVAL;
1471         }
1472         mutex_unlock(&rdev->devlist_mtx);
1473
1474         return result;
1475 }
1476
1477 static int nl80211_set_channel(struct sk_buff *skb, struct genl_info *info)
1478 {
1479         struct cfg80211_registered_device *rdev = info->user_ptr[0];
1480         struct net_device *netdev = info->user_ptr[1];
1481
1482         return __nl80211_set_channel(rdev, netdev->ieee80211_ptr, info);
1483 }
1484
1485 static int nl80211_set_wds_peer(struct sk_buff *skb, struct genl_info *info)
1486 {
1487         struct cfg80211_registered_device *rdev = info->user_ptr[0];
1488         struct net_device *dev = info->user_ptr[1];
1489         struct wireless_dev *wdev = dev->ieee80211_ptr;
1490         const u8 *bssid;
1491
1492         if (!info->attrs[NL80211_ATTR_MAC])
1493                 return -EINVAL;
1494
1495         if (netif_running(dev))
1496                 return -EBUSY;
1497
1498         if (!rdev->ops->set_wds_peer)
1499                 return -EOPNOTSUPP;
1500
1501         if (wdev->iftype != NL80211_IFTYPE_WDS)
1502                 return -EOPNOTSUPP;
1503
1504         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
1505         return rdev_set_wds_peer(rdev, dev, bssid);
1506 }
1507
1508
1509 static int nl80211_set_wiphy(struct sk_buff *skb, struct genl_info *info)
1510 {
1511         struct cfg80211_registered_device *rdev;
1512         struct net_device *netdev = NULL;
1513         struct wireless_dev *wdev;
1514         int result = 0, rem_txq_params = 0;
1515         struct nlattr *nl_txq_params;
1516         u32 changed;
1517         u8 retry_short = 0, retry_long = 0;
1518         u32 frag_threshold = 0, rts_threshold = 0;
1519         u8 coverage_class = 0;
1520
1521         /*
1522          * Try to find the wiphy and netdev. Normally this
1523          * function shouldn't need the netdev, but this is
1524          * done for backward compatibility -- previously
1525          * setting the channel was done per wiphy, but now
1526          * it is per netdev. Previous userland like hostapd
1527          * also passed a netdev to set_wiphy, so that it is
1528          * possible to let that go to the right netdev!
1529          */
1530         mutex_lock(&cfg80211_mutex);
1531
1532         if (info->attrs[NL80211_ATTR_IFINDEX]) {
1533                 int ifindex = nla_get_u32(info->attrs[NL80211_ATTR_IFINDEX]);
1534
1535                 netdev = dev_get_by_index(genl_info_net(info), ifindex);
1536                 if (netdev && netdev->ieee80211_ptr) {
1537                         rdev = wiphy_to_dev(netdev->ieee80211_ptr->wiphy);
1538                         mutex_lock(&rdev->mtx);
1539                 } else
1540                         netdev = NULL;
1541         }
1542
1543         if (!netdev) {
1544                 rdev = __cfg80211_rdev_from_attrs(genl_info_net(info),
1545                                                   info->attrs);
1546                 if (IS_ERR(rdev)) {
1547                         mutex_unlock(&cfg80211_mutex);
1548                         return PTR_ERR(rdev);
1549                 }
1550                 wdev = NULL;
1551                 netdev = NULL;
1552                 result = 0;
1553
1554                 mutex_lock(&rdev->mtx);
1555         } else
1556                 wdev = netdev->ieee80211_ptr;
1557
1558         /*
1559          * end workaround code, by now the rdev is available
1560          * and locked, and wdev may or may not be NULL.
1561          */
1562
1563         if (info->attrs[NL80211_ATTR_WIPHY_NAME])
1564                 result = cfg80211_dev_rename(
1565                         rdev, nla_data(info->attrs[NL80211_ATTR_WIPHY_NAME]));
1566
1567         mutex_unlock(&cfg80211_mutex);
1568
1569         if (result)
1570                 goto bad_res;
1571
1572         if (info->attrs[NL80211_ATTR_WIPHY_TXQ_PARAMS]) {
1573                 struct ieee80211_txq_params txq_params;
1574                 struct nlattr *tb[NL80211_TXQ_ATTR_MAX + 1];
1575
1576                 if (!rdev->ops->set_txq_params) {
1577                         result = -EOPNOTSUPP;
1578                         goto bad_res;
1579                 }
1580
1581                 if (!netdev) {
1582                         result = -EINVAL;
1583                         goto bad_res;
1584                 }
1585
1586                 if (netdev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
1587                     netdev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO) {
1588                         result = -EINVAL;
1589                         goto bad_res;
1590                 }
1591
1592                 if (!netif_running(netdev)) {
1593                         result = -ENETDOWN;
1594                         goto bad_res;
1595                 }
1596
1597                 nla_for_each_nested(nl_txq_params,
1598                                     info->attrs[NL80211_ATTR_WIPHY_TXQ_PARAMS],
1599                                     rem_txq_params) {
1600                         nla_parse(tb, NL80211_TXQ_ATTR_MAX,
1601                                   nla_data(nl_txq_params),
1602                                   nla_len(nl_txq_params),
1603                                   txq_params_policy);
1604                         result = parse_txq_params(tb, &txq_params);
1605                         if (result)
1606                                 goto bad_res;
1607
1608                         result = rdev_set_txq_params(rdev, netdev,
1609                                                      &txq_params);
1610                         if (result)
1611                                 goto bad_res;
1612                 }
1613         }
1614
1615         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
1616                 result = __nl80211_set_channel(rdev,
1617                                 nl80211_can_set_dev_channel(wdev) ? wdev : NULL,
1618                                 info);
1619                 if (result)
1620                         goto bad_res;
1621         }
1622
1623         if (info->attrs[NL80211_ATTR_WIPHY_TX_POWER_SETTING]) {
1624                 struct wireless_dev *txp_wdev = wdev;
1625                 enum nl80211_tx_power_setting type;
1626                 int idx, mbm = 0;
1627
1628                 if (!(rdev->wiphy.features & NL80211_FEATURE_VIF_TXPOWER))
1629                         txp_wdev = NULL;
1630
1631                 if (!rdev->ops->set_tx_power) {
1632                         result = -EOPNOTSUPP;
1633                         goto bad_res;
1634                 }
1635
1636                 idx = NL80211_ATTR_WIPHY_TX_POWER_SETTING;
1637                 type = nla_get_u32(info->attrs[idx]);
1638
1639                 if (!info->attrs[NL80211_ATTR_WIPHY_TX_POWER_LEVEL] &&
1640                     (type != NL80211_TX_POWER_AUTOMATIC)) {
1641                         result = -EINVAL;
1642                         goto bad_res;
1643                 }
1644
1645                 if (type != NL80211_TX_POWER_AUTOMATIC) {
1646                         idx = NL80211_ATTR_WIPHY_TX_POWER_LEVEL;
1647                         mbm = nla_get_u32(info->attrs[idx]);
1648                 }
1649
1650                 result = rdev_set_tx_power(rdev, txp_wdev, type, mbm);
1651                 if (result)
1652                         goto bad_res;
1653         }
1654
1655         if (info->attrs[NL80211_ATTR_WIPHY_ANTENNA_TX] &&
1656             info->attrs[NL80211_ATTR_WIPHY_ANTENNA_RX]) {
1657                 u32 tx_ant, rx_ant;
1658                 if ((!rdev->wiphy.available_antennas_tx &&
1659                      !rdev->wiphy.available_antennas_rx) ||
1660                     !rdev->ops->set_antenna) {
1661                         result = -EOPNOTSUPP;
1662                         goto bad_res;
1663                 }
1664
1665                 tx_ant = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_ANTENNA_TX]);
1666                 rx_ant = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_ANTENNA_RX]);
1667
1668                 /* reject antenna configurations which don't match the
1669                  * available antenna masks, except for the "all" mask */
1670                 if ((~tx_ant && (tx_ant & ~rdev->wiphy.available_antennas_tx)) ||
1671                     (~rx_ant && (rx_ant & ~rdev->wiphy.available_antennas_rx))) {
1672                         result = -EINVAL;
1673                         goto bad_res;
1674                 }
1675
1676                 tx_ant = tx_ant & rdev->wiphy.available_antennas_tx;
1677                 rx_ant = rx_ant & rdev->wiphy.available_antennas_rx;
1678
1679                 result = rdev_set_antenna(rdev, tx_ant, rx_ant);
1680                 if (result)
1681                         goto bad_res;
1682         }
1683
1684         changed = 0;
1685
1686         if (info->attrs[NL80211_ATTR_WIPHY_RETRY_SHORT]) {
1687                 retry_short = nla_get_u8(
1688                         info->attrs[NL80211_ATTR_WIPHY_RETRY_SHORT]);
1689                 if (retry_short == 0) {
1690                         result = -EINVAL;
1691                         goto bad_res;
1692                 }
1693                 changed |= WIPHY_PARAM_RETRY_SHORT;
1694         }
1695
1696         if (info->attrs[NL80211_ATTR_WIPHY_RETRY_LONG]) {
1697                 retry_long = nla_get_u8(
1698                         info->attrs[NL80211_ATTR_WIPHY_RETRY_LONG]);
1699                 if (retry_long == 0) {
1700                         result = -EINVAL;
1701                         goto bad_res;
1702                 }
1703                 changed |= WIPHY_PARAM_RETRY_LONG;
1704         }
1705
1706         if (info->attrs[NL80211_ATTR_WIPHY_FRAG_THRESHOLD]) {
1707                 frag_threshold = nla_get_u32(
1708                         info->attrs[NL80211_ATTR_WIPHY_FRAG_THRESHOLD]);
1709                 if (frag_threshold < 256) {
1710                         result = -EINVAL;
1711                         goto bad_res;
1712                 }
1713                 if (frag_threshold != (u32) -1) {
1714                         /*
1715                          * Fragments (apart from the last one) are required to
1716                          * have even length. Make the fragmentation code
1717                          * simpler by stripping LSB should someone try to use
1718                          * odd threshold value.
1719                          */
1720                         frag_threshold &= ~0x1;
1721                 }
1722                 changed |= WIPHY_PARAM_FRAG_THRESHOLD;
1723         }
1724
1725         if (info->attrs[NL80211_ATTR_WIPHY_RTS_THRESHOLD]) {
1726                 rts_threshold = nla_get_u32(
1727                         info->attrs[NL80211_ATTR_WIPHY_RTS_THRESHOLD]);
1728                 changed |= WIPHY_PARAM_RTS_THRESHOLD;
1729         }
1730
1731         if (info->attrs[NL80211_ATTR_WIPHY_COVERAGE_CLASS]) {
1732                 coverage_class = nla_get_u8(
1733                         info->attrs[NL80211_ATTR_WIPHY_COVERAGE_CLASS]);
1734                 changed |= WIPHY_PARAM_COVERAGE_CLASS;
1735         }
1736
1737         if (changed) {
1738                 u8 old_retry_short, old_retry_long;
1739                 u32 old_frag_threshold, old_rts_threshold;
1740                 u8 old_coverage_class;
1741
1742                 if (!rdev->ops->set_wiphy_params) {
1743                         result = -EOPNOTSUPP;
1744                         goto bad_res;
1745                 }
1746
1747                 old_retry_short = rdev->wiphy.retry_short;
1748                 old_retry_long = rdev->wiphy.retry_long;
1749                 old_frag_threshold = rdev->wiphy.frag_threshold;
1750                 old_rts_threshold = rdev->wiphy.rts_threshold;
1751                 old_coverage_class = rdev->wiphy.coverage_class;
1752
1753                 if (changed & WIPHY_PARAM_RETRY_SHORT)
1754                         rdev->wiphy.retry_short = retry_short;
1755                 if (changed & WIPHY_PARAM_RETRY_LONG)
1756                         rdev->wiphy.retry_long = retry_long;
1757                 if (changed & WIPHY_PARAM_FRAG_THRESHOLD)
1758                         rdev->wiphy.frag_threshold = frag_threshold;
1759                 if (changed & WIPHY_PARAM_RTS_THRESHOLD)
1760                         rdev->wiphy.rts_threshold = rts_threshold;
1761                 if (changed & WIPHY_PARAM_COVERAGE_CLASS)
1762                         rdev->wiphy.coverage_class = coverage_class;
1763
1764                 result = rdev_set_wiphy_params(rdev, changed);
1765                 if (result) {
1766                         rdev->wiphy.retry_short = old_retry_short;
1767                         rdev->wiphy.retry_long = old_retry_long;
1768                         rdev->wiphy.frag_threshold = old_frag_threshold;
1769                         rdev->wiphy.rts_threshold = old_rts_threshold;
1770                         rdev->wiphy.coverage_class = old_coverage_class;
1771                 }
1772         }
1773
1774  bad_res:
1775         mutex_unlock(&rdev->mtx);
1776         if (netdev)
1777                 dev_put(netdev);
1778         return result;
1779 }
1780
1781 static inline u64 wdev_id(struct wireless_dev *wdev)
1782 {
1783         return (u64)wdev->identifier |
1784                ((u64)wiphy_to_dev(wdev->wiphy)->wiphy_idx << 32);
1785 }
1786
1787 static int nl80211_send_chandef(struct sk_buff *msg,
1788                                  struct cfg80211_chan_def *chandef)
1789 {
1790         WARN_ON(!cfg80211_chandef_valid(chandef));
1791
1792         if (nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ,
1793                         chandef->chan->center_freq))
1794                 return -ENOBUFS;
1795         switch (chandef->width) {
1796         case NL80211_CHAN_WIDTH_20_NOHT:
1797         case NL80211_CHAN_WIDTH_20:
1798         case NL80211_CHAN_WIDTH_40:
1799                 if (nla_put_u32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
1800                                 cfg80211_get_chandef_type(chandef)))
1801                         return -ENOBUFS;
1802                 break;
1803         default:
1804                 break;
1805         }
1806         if (nla_put_u32(msg, NL80211_ATTR_CHANNEL_WIDTH, chandef->width))
1807                 return -ENOBUFS;
1808         if (nla_put_u32(msg, NL80211_ATTR_CENTER_FREQ1, chandef->center_freq1))
1809                 return -ENOBUFS;
1810         if (chandef->center_freq2 &&
1811             nla_put_u32(msg, NL80211_ATTR_CENTER_FREQ2, chandef->center_freq2))
1812                 return -ENOBUFS;
1813         return 0;
1814 }
1815
1816 static int nl80211_send_iface(struct sk_buff *msg, u32 portid, u32 seq, int flags,
1817                               struct cfg80211_registered_device *rdev,
1818                               struct wireless_dev *wdev)
1819 {
1820         struct net_device *dev = wdev->netdev;
1821         void *hdr;
1822
1823         hdr = nl80211hdr_put(msg, portid, seq, flags, NL80211_CMD_NEW_INTERFACE);
1824         if (!hdr)
1825                 return -1;
1826
1827         if (dev &&
1828             (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
1829              nla_put_string(msg, NL80211_ATTR_IFNAME, dev->name)))
1830                 goto nla_put_failure;
1831
1832         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
1833             nla_put_u32(msg, NL80211_ATTR_IFTYPE, wdev->iftype) ||
1834             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)) ||
1835             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, wdev_address(wdev)) ||
1836             nla_put_u32(msg, NL80211_ATTR_GENERATION,
1837                         rdev->devlist_generation ^
1838                         (cfg80211_rdev_list_generation << 2)))
1839                 goto nla_put_failure;
1840
1841         if (rdev->ops->get_channel) {
1842                 int ret;
1843                 struct cfg80211_chan_def chandef;
1844
1845                 ret = rdev_get_channel(rdev, wdev, &chandef);
1846                 if (ret == 0) {
1847                         if (nl80211_send_chandef(msg, &chandef))
1848                                 goto nla_put_failure;
1849                 }
1850         }
1851
1852         if (wdev->ssid_len) {
1853                 if (nla_put(msg, NL80211_ATTR_SSID, wdev->ssid_len, wdev->ssid))
1854                         goto nla_put_failure;
1855         }
1856
1857         return genlmsg_end(msg, hdr);
1858
1859  nla_put_failure:
1860         genlmsg_cancel(msg, hdr);
1861         return -EMSGSIZE;
1862 }
1863
1864 static int nl80211_dump_interface(struct sk_buff *skb, struct netlink_callback *cb)
1865 {
1866         int wp_idx = 0;
1867         int if_idx = 0;
1868         int wp_start = cb->args[0];
1869         int if_start = cb->args[1];
1870         struct cfg80211_registered_device *rdev;
1871         struct wireless_dev *wdev;
1872
1873         mutex_lock(&cfg80211_mutex);
1874         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
1875                 if (!net_eq(wiphy_net(&rdev->wiphy), sock_net(skb->sk)))
1876                         continue;
1877                 if (wp_idx < wp_start) {
1878                         wp_idx++;
1879                         continue;
1880                 }
1881                 if_idx = 0;
1882
1883                 mutex_lock(&rdev->devlist_mtx);
1884                 list_for_each_entry(wdev, &rdev->wdev_list, list) {
1885                         if (if_idx < if_start) {
1886                                 if_idx++;
1887                                 continue;
1888                         }
1889                         if (nl80211_send_iface(skb, NETLINK_CB(cb->skb).portid,
1890                                                cb->nlh->nlmsg_seq, NLM_F_MULTI,
1891                                                rdev, wdev) < 0) {
1892                                 mutex_unlock(&rdev->devlist_mtx);
1893                                 goto out;
1894                         }
1895                         if_idx++;
1896                 }
1897                 mutex_unlock(&rdev->devlist_mtx);
1898
1899                 wp_idx++;
1900         }
1901  out:
1902         mutex_unlock(&cfg80211_mutex);
1903
1904         cb->args[0] = wp_idx;
1905         cb->args[1] = if_idx;
1906
1907         return skb->len;
1908 }
1909
1910 static int nl80211_get_interface(struct sk_buff *skb, struct genl_info *info)
1911 {
1912         struct sk_buff *msg;
1913         struct cfg80211_registered_device *dev = info->user_ptr[0];
1914         struct wireless_dev *wdev = info->user_ptr[1];
1915
1916         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1917         if (!msg)
1918                 return -ENOMEM;
1919
1920         if (nl80211_send_iface(msg, info->snd_portid, info->snd_seq, 0,
1921                                dev, wdev) < 0) {
1922                 nlmsg_free(msg);
1923                 return -ENOBUFS;
1924         }
1925
1926         return genlmsg_reply(msg, info);
1927 }
1928
1929 static const struct nla_policy mntr_flags_policy[NL80211_MNTR_FLAG_MAX + 1] = {
1930         [NL80211_MNTR_FLAG_FCSFAIL] = { .type = NLA_FLAG },
1931         [NL80211_MNTR_FLAG_PLCPFAIL] = { .type = NLA_FLAG },
1932         [NL80211_MNTR_FLAG_CONTROL] = { .type = NLA_FLAG },
1933         [NL80211_MNTR_FLAG_OTHER_BSS] = { .type = NLA_FLAG },
1934         [NL80211_MNTR_FLAG_COOK_FRAMES] = { .type = NLA_FLAG },
1935 };
1936
1937 static int parse_monitor_flags(struct nlattr *nla, u32 *mntrflags)
1938 {
1939         struct nlattr *flags[NL80211_MNTR_FLAG_MAX + 1];
1940         int flag;
1941
1942         *mntrflags = 0;
1943
1944         if (!nla)
1945                 return -EINVAL;
1946
1947         if (nla_parse_nested(flags, NL80211_MNTR_FLAG_MAX,
1948                              nla, mntr_flags_policy))
1949                 return -EINVAL;
1950
1951         for (flag = 1; flag <= NL80211_MNTR_FLAG_MAX; flag++)
1952                 if (flags[flag])
1953                         *mntrflags |= (1<<flag);
1954
1955         return 0;
1956 }
1957
1958 static int nl80211_valid_4addr(struct cfg80211_registered_device *rdev,
1959                                struct net_device *netdev, u8 use_4addr,
1960                                enum nl80211_iftype iftype)
1961 {
1962         if (!use_4addr) {
1963                 if (netdev && (netdev->priv_flags & IFF_BRIDGE_PORT))
1964                         return -EBUSY;
1965                 return 0;
1966         }
1967
1968         switch (iftype) {
1969         case NL80211_IFTYPE_AP_VLAN:
1970                 if (rdev->wiphy.flags & WIPHY_FLAG_4ADDR_AP)
1971                         return 0;
1972                 break;
1973         case NL80211_IFTYPE_STATION:
1974                 if (rdev->wiphy.flags & WIPHY_FLAG_4ADDR_STATION)
1975                         return 0;
1976                 break;
1977         default:
1978                 break;
1979         }
1980
1981         return -EOPNOTSUPP;
1982 }
1983
1984 static int nl80211_set_interface(struct sk_buff *skb, struct genl_info *info)
1985 {
1986         struct cfg80211_registered_device *rdev = info->user_ptr[0];
1987         struct vif_params params;
1988         int err;
1989         enum nl80211_iftype otype, ntype;
1990         struct net_device *dev = info->user_ptr[1];
1991         u32 _flags, *flags = NULL;
1992         bool change = false;
1993
1994         memset(&params, 0, sizeof(params));
1995
1996         otype = ntype = dev->ieee80211_ptr->iftype;
1997
1998         if (info->attrs[NL80211_ATTR_IFTYPE]) {
1999                 ntype = nla_get_u32(info->attrs[NL80211_ATTR_IFTYPE]);
2000                 if (otype != ntype)
2001                         change = true;
2002                 if (ntype > NL80211_IFTYPE_MAX)
2003                         return -EINVAL;
2004         }
2005
2006         if (info->attrs[NL80211_ATTR_MESH_ID]) {
2007                 struct wireless_dev *wdev = dev->ieee80211_ptr;
2008
2009                 if (ntype != NL80211_IFTYPE_MESH_POINT)
2010                         return -EINVAL;
2011                 if (netif_running(dev))
2012                         return -EBUSY;
2013
2014                 wdev_lock(wdev);
2015                 BUILD_BUG_ON(IEEE80211_MAX_SSID_LEN !=
2016                              IEEE80211_MAX_MESH_ID_LEN);
2017                 wdev->mesh_id_up_len =
2018                         nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
2019                 memcpy(wdev->ssid, nla_data(info->attrs[NL80211_ATTR_MESH_ID]),
2020                        wdev->mesh_id_up_len);
2021                 wdev_unlock(wdev);
2022         }
2023
2024         if (info->attrs[NL80211_ATTR_4ADDR]) {
2025                 params.use_4addr = !!nla_get_u8(info->attrs[NL80211_ATTR_4ADDR]);
2026                 change = true;
2027                 err = nl80211_valid_4addr(rdev, dev, params.use_4addr, ntype);
2028                 if (err)
2029                         return err;
2030         } else {
2031                 params.use_4addr = -1;
2032         }
2033
2034         if (info->attrs[NL80211_ATTR_MNTR_FLAGS]) {
2035                 if (ntype != NL80211_IFTYPE_MONITOR)
2036                         return -EINVAL;
2037                 err = parse_monitor_flags(info->attrs[NL80211_ATTR_MNTR_FLAGS],
2038                                           &_flags);
2039                 if (err)
2040                         return err;
2041
2042                 flags = &_flags;
2043                 change = true;
2044         }
2045
2046         if (change)
2047                 err = cfg80211_change_iface(rdev, dev, ntype, flags, &params);
2048         else
2049                 err = 0;
2050
2051         if (!err && params.use_4addr != -1)
2052                 dev->ieee80211_ptr->use_4addr = params.use_4addr;
2053
2054         return err;
2055 }
2056
2057 static int nl80211_new_interface(struct sk_buff *skb, struct genl_info *info)
2058 {
2059         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2060         struct vif_params params;
2061         struct wireless_dev *wdev;
2062         struct sk_buff *msg;
2063         int err;
2064         enum nl80211_iftype type = NL80211_IFTYPE_UNSPECIFIED;
2065         u32 flags;
2066
2067         memset(&params, 0, sizeof(params));
2068
2069         if (!info->attrs[NL80211_ATTR_IFNAME])
2070                 return -EINVAL;
2071
2072         if (info->attrs[NL80211_ATTR_IFTYPE]) {
2073                 type = nla_get_u32(info->attrs[NL80211_ATTR_IFTYPE]);
2074                 if (type > NL80211_IFTYPE_MAX)
2075                         return -EINVAL;
2076         }
2077
2078         if (!rdev->ops->add_virtual_intf ||
2079             !(rdev->wiphy.interface_modes & (1 << type)))
2080                 return -EOPNOTSUPP;
2081
2082         if (type == NL80211_IFTYPE_P2P_DEVICE && info->attrs[NL80211_ATTR_MAC]) {
2083                 nla_memcpy(params.macaddr, info->attrs[NL80211_ATTR_MAC],
2084                            ETH_ALEN);
2085                 if (!is_valid_ether_addr(params.macaddr))
2086                         return -EADDRNOTAVAIL;
2087         }
2088
2089         if (info->attrs[NL80211_ATTR_4ADDR]) {
2090                 params.use_4addr = !!nla_get_u8(info->attrs[NL80211_ATTR_4ADDR]);
2091                 err = nl80211_valid_4addr(rdev, NULL, params.use_4addr, type);
2092                 if (err)
2093                         return err;
2094         }
2095
2096         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2097         if (!msg)
2098                 return -ENOMEM;
2099
2100         err = parse_monitor_flags(type == NL80211_IFTYPE_MONITOR ?
2101                                   info->attrs[NL80211_ATTR_MNTR_FLAGS] : NULL,
2102                                   &flags);
2103         wdev = rdev_add_virtual_intf(rdev,
2104                                 nla_data(info->attrs[NL80211_ATTR_IFNAME]),
2105                                 type, err ? NULL : &flags, &params);
2106         if (IS_ERR(wdev)) {
2107                 nlmsg_free(msg);
2108                 return PTR_ERR(wdev);
2109         }
2110
2111         switch (type) {
2112         case NL80211_IFTYPE_MESH_POINT:
2113                 if (!info->attrs[NL80211_ATTR_MESH_ID])
2114                         break;
2115                 wdev_lock(wdev);
2116                 BUILD_BUG_ON(IEEE80211_MAX_SSID_LEN !=
2117                              IEEE80211_MAX_MESH_ID_LEN);
2118                 wdev->mesh_id_up_len =
2119                         nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
2120                 memcpy(wdev->ssid, nla_data(info->attrs[NL80211_ATTR_MESH_ID]),
2121                        wdev->mesh_id_up_len);
2122                 wdev_unlock(wdev);
2123                 break;
2124         case NL80211_IFTYPE_P2P_DEVICE:
2125                 /*
2126                  * P2P Device doesn't have a netdev, so doesn't go
2127                  * through the netdev notifier and must be added here
2128                  */
2129                 mutex_init(&wdev->mtx);
2130                 INIT_LIST_HEAD(&wdev->event_list);
2131                 spin_lock_init(&wdev->event_lock);
2132                 INIT_LIST_HEAD(&wdev->mgmt_registrations);
2133                 spin_lock_init(&wdev->mgmt_registrations_lock);
2134
2135                 mutex_lock(&rdev->devlist_mtx);
2136                 wdev->identifier = ++rdev->wdev_id;
2137                 list_add_rcu(&wdev->list, &rdev->wdev_list);
2138                 rdev->devlist_generation++;
2139                 mutex_unlock(&rdev->devlist_mtx);
2140                 break;
2141         default:
2142                 break;
2143         }
2144
2145         if (nl80211_send_iface(msg, info->snd_portid, info->snd_seq, 0,
2146                                rdev, wdev) < 0) {
2147                 nlmsg_free(msg);
2148                 return -ENOBUFS;
2149         }
2150
2151         return genlmsg_reply(msg, info);
2152 }
2153
2154 static int nl80211_del_interface(struct sk_buff *skb, struct genl_info *info)
2155 {
2156         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2157         struct wireless_dev *wdev = info->user_ptr[1];
2158
2159         if (!rdev->ops->del_virtual_intf)
2160                 return -EOPNOTSUPP;
2161
2162         /*
2163          * If we remove a wireless device without a netdev then clear
2164          * user_ptr[1] so that nl80211_post_doit won't dereference it
2165          * to check if it needs to do dev_put(). Otherwise it crashes
2166          * since the wdev has been freed, unlike with a netdev where
2167          * we need the dev_put() for the netdev to really be freed.
2168          */
2169         if (!wdev->netdev)
2170                 info->user_ptr[1] = NULL;
2171
2172         return rdev_del_virtual_intf(rdev, wdev);
2173 }
2174
2175 static int nl80211_set_noack_map(struct sk_buff *skb, struct genl_info *info)
2176 {
2177         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2178         struct net_device *dev = info->user_ptr[1];
2179         u16 noack_map;
2180
2181         if (!info->attrs[NL80211_ATTR_NOACK_MAP])
2182                 return -EINVAL;
2183
2184         if (!rdev->ops->set_noack_map)
2185                 return -EOPNOTSUPP;
2186
2187         noack_map = nla_get_u16(info->attrs[NL80211_ATTR_NOACK_MAP]);
2188
2189         return rdev_set_noack_map(rdev, dev, noack_map);
2190 }
2191
2192 struct get_key_cookie {
2193         struct sk_buff *msg;
2194         int error;
2195         int idx;
2196 };
2197
2198 static void get_key_callback(void *c, struct key_params *params)
2199 {
2200         struct nlattr *key;
2201         struct get_key_cookie *cookie = c;
2202
2203         if ((params->key &&
2204              nla_put(cookie->msg, NL80211_ATTR_KEY_DATA,
2205                      params->key_len, params->key)) ||
2206             (params->seq &&
2207              nla_put(cookie->msg, NL80211_ATTR_KEY_SEQ,
2208                      params->seq_len, params->seq)) ||
2209             (params->cipher &&
2210              nla_put_u32(cookie->msg, NL80211_ATTR_KEY_CIPHER,
2211                          params->cipher)))
2212                 goto nla_put_failure;
2213
2214         key = nla_nest_start(cookie->msg, NL80211_ATTR_KEY);
2215         if (!key)
2216                 goto nla_put_failure;
2217
2218         if ((params->key &&
2219              nla_put(cookie->msg, NL80211_KEY_DATA,
2220                      params->key_len, params->key)) ||
2221             (params->seq &&
2222              nla_put(cookie->msg, NL80211_KEY_SEQ,
2223                      params->seq_len, params->seq)) ||
2224             (params->cipher &&
2225              nla_put_u32(cookie->msg, NL80211_KEY_CIPHER,
2226                          params->cipher)))
2227                 goto nla_put_failure;
2228
2229         if (nla_put_u8(cookie->msg, NL80211_ATTR_KEY_IDX, cookie->idx))
2230                 goto nla_put_failure;
2231
2232         nla_nest_end(cookie->msg, key);
2233
2234         return;
2235  nla_put_failure:
2236         cookie->error = 1;
2237 }
2238
2239 static int nl80211_get_key(struct sk_buff *skb, struct genl_info *info)
2240 {
2241         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2242         int err;
2243         struct net_device *dev = info->user_ptr[1];
2244         u8 key_idx = 0;
2245         const u8 *mac_addr = NULL;
2246         bool pairwise;
2247         struct get_key_cookie cookie = {
2248                 .error = 0,
2249         };
2250         void *hdr;
2251         struct sk_buff *msg;
2252
2253         if (info->attrs[NL80211_ATTR_KEY_IDX])
2254                 key_idx = nla_get_u8(info->attrs[NL80211_ATTR_KEY_IDX]);
2255
2256         if (key_idx > 5)
2257                 return -EINVAL;
2258
2259         if (info->attrs[NL80211_ATTR_MAC])
2260                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
2261
2262         pairwise = !!mac_addr;
2263         if (info->attrs[NL80211_ATTR_KEY_TYPE]) {
2264                 u32 kt = nla_get_u32(info->attrs[NL80211_ATTR_KEY_TYPE]);
2265                 if (kt >= NUM_NL80211_KEYTYPES)
2266                         return -EINVAL;
2267                 if (kt != NL80211_KEYTYPE_GROUP &&
2268                     kt != NL80211_KEYTYPE_PAIRWISE)
2269                         return -EINVAL;
2270                 pairwise = kt == NL80211_KEYTYPE_PAIRWISE;
2271         }
2272
2273         if (!rdev->ops->get_key)
2274                 return -EOPNOTSUPP;
2275
2276         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2277         if (!msg)
2278                 return -ENOMEM;
2279
2280         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
2281                              NL80211_CMD_NEW_KEY);
2282         if (IS_ERR(hdr))
2283                 return PTR_ERR(hdr);
2284
2285         cookie.msg = msg;
2286         cookie.idx = key_idx;
2287
2288         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
2289             nla_put_u8(msg, NL80211_ATTR_KEY_IDX, key_idx))
2290                 goto nla_put_failure;
2291         if (mac_addr &&
2292             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr))
2293                 goto nla_put_failure;
2294
2295         if (pairwise && mac_addr &&
2296             !(rdev->wiphy.flags & WIPHY_FLAG_IBSS_RSN))
2297                 return -ENOENT;
2298
2299         err = rdev_get_key(rdev, dev, key_idx, pairwise, mac_addr, &cookie,
2300                            get_key_callback);
2301
2302         if (err)
2303                 goto free_msg;
2304
2305         if (cookie.error)
2306                 goto nla_put_failure;
2307
2308         genlmsg_end(msg, hdr);
2309         return genlmsg_reply(msg, info);
2310
2311  nla_put_failure:
2312         err = -ENOBUFS;
2313  free_msg:
2314         nlmsg_free(msg);
2315         return err;
2316 }
2317
2318 static int nl80211_set_key(struct sk_buff *skb, struct genl_info *info)
2319 {
2320         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2321         struct key_parse key;
2322         int err;
2323         struct net_device *dev = info->user_ptr[1];
2324
2325         err = nl80211_parse_key(info, &key);
2326         if (err)
2327                 return err;
2328
2329         if (key.idx < 0)
2330                 return -EINVAL;
2331
2332         /* only support setting default key */
2333         if (!key.def && !key.defmgmt)
2334                 return -EINVAL;
2335
2336         wdev_lock(dev->ieee80211_ptr);
2337
2338         if (key.def) {
2339                 if (!rdev->ops->set_default_key) {
2340                         err = -EOPNOTSUPP;
2341                         goto out;
2342                 }
2343
2344                 err = nl80211_key_allowed(dev->ieee80211_ptr);
2345                 if (err)
2346                         goto out;
2347
2348                 err = rdev_set_default_key(rdev, dev, key.idx,
2349                                                  key.def_uni, key.def_multi);
2350
2351                 if (err)
2352                         goto out;
2353
2354 #ifdef CONFIG_CFG80211_WEXT
2355                 dev->ieee80211_ptr->wext.default_key = key.idx;
2356 #endif
2357         } else {
2358                 if (key.def_uni || !key.def_multi) {
2359                         err = -EINVAL;
2360                         goto out;
2361                 }
2362
2363                 if (!rdev->ops->set_default_mgmt_key) {
2364                         err = -EOPNOTSUPP;
2365                         goto out;
2366                 }
2367
2368                 err = nl80211_key_allowed(dev->ieee80211_ptr);
2369                 if (err)
2370                         goto out;
2371
2372                 err = rdev_set_default_mgmt_key(rdev, dev, key.idx);
2373                 if (err)
2374                         goto out;
2375
2376 #ifdef CONFIG_CFG80211_WEXT
2377                 dev->ieee80211_ptr->wext.default_mgmt_key = key.idx;
2378 #endif
2379         }
2380
2381  out:
2382         wdev_unlock(dev->ieee80211_ptr);
2383
2384         return err;
2385 }
2386
2387 static int nl80211_new_key(struct sk_buff *skb, struct genl_info *info)
2388 {
2389         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2390         int err;
2391         struct net_device *dev = info->user_ptr[1];
2392         struct key_parse key;
2393         const u8 *mac_addr = NULL;
2394
2395         err = nl80211_parse_key(info, &key);
2396         if (err)
2397                 return err;
2398
2399         if (!key.p.key)
2400                 return -EINVAL;
2401
2402         if (info->attrs[NL80211_ATTR_MAC])
2403                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
2404
2405         if (key.type == -1) {
2406                 if (mac_addr)
2407                         key.type = NL80211_KEYTYPE_PAIRWISE;
2408                 else
2409                         key.type = NL80211_KEYTYPE_GROUP;
2410         }
2411
2412         /* for now */
2413         if (key.type != NL80211_KEYTYPE_PAIRWISE &&
2414             key.type != NL80211_KEYTYPE_GROUP)
2415                 return -EINVAL;
2416
2417         if (!rdev->ops->add_key)
2418                 return -EOPNOTSUPP;
2419
2420         if (cfg80211_validate_key_settings(rdev, &key.p, key.idx,
2421                                            key.type == NL80211_KEYTYPE_PAIRWISE,
2422                                            mac_addr))
2423                 return -EINVAL;
2424
2425         wdev_lock(dev->ieee80211_ptr);
2426         err = nl80211_key_allowed(dev->ieee80211_ptr);
2427         if (!err)
2428                 err = rdev_add_key(rdev, dev, key.idx,
2429                                    key.type == NL80211_KEYTYPE_PAIRWISE,
2430                                     mac_addr, &key.p);
2431         wdev_unlock(dev->ieee80211_ptr);
2432
2433         return err;
2434 }
2435
2436 static int nl80211_del_key(struct sk_buff *skb, struct genl_info *info)
2437 {
2438         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2439         int err;
2440         struct net_device *dev = info->user_ptr[1];
2441         u8 *mac_addr = NULL;
2442         struct key_parse key;
2443
2444         err = nl80211_parse_key(info, &key);
2445         if (err)
2446                 return err;
2447
2448         if (info->attrs[NL80211_ATTR_MAC])
2449                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
2450
2451         if (key.type == -1) {
2452                 if (mac_addr)
2453                         key.type = NL80211_KEYTYPE_PAIRWISE;
2454                 else
2455                         key.type = NL80211_KEYTYPE_GROUP;
2456         }
2457
2458         /* for now */
2459         if (key.type != NL80211_KEYTYPE_PAIRWISE &&
2460             key.type != NL80211_KEYTYPE_GROUP)
2461                 return -EINVAL;
2462
2463         if (!rdev->ops->del_key)
2464                 return -EOPNOTSUPP;
2465
2466         wdev_lock(dev->ieee80211_ptr);
2467         err = nl80211_key_allowed(dev->ieee80211_ptr);
2468
2469         if (key.type == NL80211_KEYTYPE_PAIRWISE && mac_addr &&
2470             !(rdev->wiphy.flags & WIPHY_FLAG_IBSS_RSN))
2471                 err = -ENOENT;
2472
2473         if (!err)
2474                 err = rdev_del_key(rdev, dev, key.idx,
2475                                    key.type == NL80211_KEYTYPE_PAIRWISE,
2476                                    mac_addr);
2477
2478 #ifdef CONFIG_CFG80211_WEXT
2479         if (!err) {
2480                 if (key.idx == dev->ieee80211_ptr->wext.default_key)
2481                         dev->ieee80211_ptr->wext.default_key = -1;
2482                 else if (key.idx == dev->ieee80211_ptr->wext.default_mgmt_key)
2483                         dev->ieee80211_ptr->wext.default_mgmt_key = -1;
2484         }
2485 #endif
2486         wdev_unlock(dev->ieee80211_ptr);
2487
2488         return err;
2489 }
2490
2491 static int nl80211_parse_beacon(struct genl_info *info,
2492                                 struct cfg80211_beacon_data *bcn)
2493 {
2494         bool haveinfo = false;
2495
2496         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_BEACON_TAIL]) ||
2497             !is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]) ||
2498             !is_valid_ie_attr(info->attrs[NL80211_ATTR_IE_PROBE_RESP]) ||
2499             !is_valid_ie_attr(info->attrs[NL80211_ATTR_IE_ASSOC_RESP]))
2500                 return -EINVAL;
2501
2502         memset(bcn, 0, sizeof(*bcn));
2503
2504         if (info->attrs[NL80211_ATTR_BEACON_HEAD]) {
2505                 bcn->head = nla_data(info->attrs[NL80211_ATTR_BEACON_HEAD]);
2506                 bcn->head_len = nla_len(info->attrs[NL80211_ATTR_BEACON_HEAD]);
2507                 if (!bcn->head_len)
2508                         return -EINVAL;
2509                 haveinfo = true;
2510         }
2511
2512         if (info->attrs[NL80211_ATTR_BEACON_TAIL]) {
2513                 bcn->tail = nla_data(info->attrs[NL80211_ATTR_BEACON_TAIL]);
2514                 bcn->tail_len =
2515                     nla_len(info->attrs[NL80211_ATTR_BEACON_TAIL]);
2516                 haveinfo = true;
2517         }
2518
2519         if (!haveinfo)
2520                 return -EINVAL;
2521
2522         if (info->attrs[NL80211_ATTR_IE]) {
2523                 bcn->beacon_ies = nla_data(info->attrs[NL80211_ATTR_IE]);
2524                 bcn->beacon_ies_len = nla_len(info->attrs[NL80211_ATTR_IE]);
2525         }
2526
2527         if (info->attrs[NL80211_ATTR_IE_PROBE_RESP]) {
2528                 bcn->proberesp_ies =
2529                         nla_data(info->attrs[NL80211_ATTR_IE_PROBE_RESP]);
2530                 bcn->proberesp_ies_len =
2531                         nla_len(info->attrs[NL80211_ATTR_IE_PROBE_RESP]);
2532         }
2533
2534         if (info->attrs[NL80211_ATTR_IE_ASSOC_RESP]) {
2535                 bcn->assocresp_ies =
2536                         nla_data(info->attrs[NL80211_ATTR_IE_ASSOC_RESP]);
2537                 bcn->assocresp_ies_len =
2538                         nla_len(info->attrs[NL80211_ATTR_IE_ASSOC_RESP]);
2539         }
2540
2541         if (info->attrs[NL80211_ATTR_PROBE_RESP]) {
2542                 bcn->probe_resp =
2543                         nla_data(info->attrs[NL80211_ATTR_PROBE_RESP]);
2544                 bcn->probe_resp_len =
2545                         nla_len(info->attrs[NL80211_ATTR_PROBE_RESP]);
2546         }
2547
2548         return 0;
2549 }
2550
2551 static bool nl80211_get_ap_channel(struct cfg80211_registered_device *rdev,
2552                                    struct cfg80211_ap_settings *params)
2553 {
2554         struct wireless_dev *wdev;
2555         bool ret = false;
2556
2557         mutex_lock(&rdev->devlist_mtx);
2558
2559         list_for_each_entry(wdev, &rdev->wdev_list, list) {
2560                 if (wdev->iftype != NL80211_IFTYPE_AP &&
2561                     wdev->iftype != NL80211_IFTYPE_P2P_GO)
2562                         continue;
2563
2564                 if (!wdev->preset_chandef.chan)
2565                         continue;
2566
2567                 params->chandef = wdev->preset_chandef;
2568                 ret = true;
2569                 break;
2570         }
2571
2572         mutex_unlock(&rdev->devlist_mtx);
2573
2574         return ret;
2575 }
2576
2577 static bool nl80211_valid_auth_type(struct cfg80211_registered_device *rdev,
2578                                     enum nl80211_auth_type auth_type,
2579                                     enum nl80211_commands cmd)
2580 {
2581         if (auth_type > NL80211_AUTHTYPE_MAX)
2582                 return false;
2583
2584         switch (cmd) {
2585         case NL80211_CMD_AUTHENTICATE:
2586                 if (!(rdev->wiphy.features & NL80211_FEATURE_SAE) &&
2587                     auth_type == NL80211_AUTHTYPE_SAE)
2588                         return false;
2589                 return true;
2590         case NL80211_CMD_CONNECT:
2591         case NL80211_CMD_START_AP:
2592                 /* SAE not supported yet */
2593                 if (auth_type == NL80211_AUTHTYPE_SAE)
2594                         return false;
2595                 return true;
2596         default:
2597                 return false;
2598         }
2599 }
2600
2601 static int nl80211_start_ap(struct sk_buff *skb, struct genl_info *info)
2602 {
2603         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2604         struct net_device *dev = info->user_ptr[1];
2605         struct wireless_dev *wdev = dev->ieee80211_ptr;
2606         struct cfg80211_ap_settings params;
2607         int err;
2608
2609         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
2610             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
2611                 return -EOPNOTSUPP;
2612
2613         if (!rdev->ops->start_ap)
2614                 return -EOPNOTSUPP;
2615
2616         if (wdev->beacon_interval)
2617                 return -EALREADY;
2618
2619         memset(&params, 0, sizeof(params));
2620
2621         /* these are required for START_AP */
2622         if (!info->attrs[NL80211_ATTR_BEACON_INTERVAL] ||
2623             !info->attrs[NL80211_ATTR_DTIM_PERIOD] ||
2624             !info->attrs[NL80211_ATTR_BEACON_HEAD])
2625                 return -EINVAL;
2626
2627         err = nl80211_parse_beacon(info, &params.beacon);
2628         if (err)
2629                 return err;
2630
2631         params.beacon_interval =
2632                 nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
2633         params.dtim_period =
2634                 nla_get_u32(info->attrs[NL80211_ATTR_DTIM_PERIOD]);
2635
2636         err = cfg80211_validate_beacon_int(rdev, params.beacon_interval);
2637         if (err)
2638                 return err;
2639
2640         /*
2641          * In theory, some of these attributes should be required here
2642          * but since they were not used when the command was originally
2643          * added, keep them optional for old user space programs to let
2644          * them continue to work with drivers that do not need the
2645          * additional information -- drivers must check!
2646          */
2647         if (info->attrs[NL80211_ATTR_SSID]) {
2648                 params.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
2649                 params.ssid_len =
2650                         nla_len(info->attrs[NL80211_ATTR_SSID]);
2651                 if (params.ssid_len == 0 ||
2652                     params.ssid_len > IEEE80211_MAX_SSID_LEN)
2653                         return -EINVAL;
2654         }
2655
2656         if (info->attrs[NL80211_ATTR_HIDDEN_SSID]) {
2657                 params.hidden_ssid = nla_get_u32(
2658                         info->attrs[NL80211_ATTR_HIDDEN_SSID]);
2659                 if (params.hidden_ssid != NL80211_HIDDEN_SSID_NOT_IN_USE &&
2660                     params.hidden_ssid != NL80211_HIDDEN_SSID_ZERO_LEN &&
2661                     params.hidden_ssid != NL80211_HIDDEN_SSID_ZERO_CONTENTS)
2662                         return -EINVAL;
2663         }
2664
2665         params.privacy = !!info->attrs[NL80211_ATTR_PRIVACY];
2666
2667         if (info->attrs[NL80211_ATTR_AUTH_TYPE]) {
2668                 params.auth_type = nla_get_u32(
2669                         info->attrs[NL80211_ATTR_AUTH_TYPE]);
2670                 if (!nl80211_valid_auth_type(rdev, params.auth_type,
2671                                              NL80211_CMD_START_AP))
2672                         return -EINVAL;
2673         } else
2674                 params.auth_type = NL80211_AUTHTYPE_AUTOMATIC;
2675
2676         err = nl80211_crypto_settings(rdev, info, &params.crypto,
2677                                       NL80211_MAX_NR_CIPHER_SUITES);
2678         if (err)
2679                 return err;
2680
2681         if (info->attrs[NL80211_ATTR_INACTIVITY_TIMEOUT]) {
2682                 if (!(rdev->wiphy.features & NL80211_FEATURE_INACTIVITY_TIMER))
2683                         return -EOPNOTSUPP;
2684                 params.inactivity_timeout = nla_get_u16(
2685                         info->attrs[NL80211_ATTR_INACTIVITY_TIMEOUT]);
2686         }
2687
2688         if (info->attrs[NL80211_ATTR_P2P_CTWINDOW]) {
2689                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
2690                         return -EINVAL;
2691                 params.p2p_ctwindow =
2692                         nla_get_u8(info->attrs[NL80211_ATTR_P2P_CTWINDOW]);
2693                 if (params.p2p_ctwindow > 127)
2694                         return -EINVAL;
2695                 if (params.p2p_ctwindow != 0 &&
2696                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_CTWIN))
2697                         return -EINVAL;
2698         }
2699
2700         if (info->attrs[NL80211_ATTR_P2P_OPPPS]) {
2701                 u8 tmp;
2702
2703                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
2704                         return -EINVAL;
2705                 tmp = nla_get_u8(info->attrs[NL80211_ATTR_P2P_OPPPS]);
2706                 if (tmp > 1)
2707                         return -EINVAL;
2708                 params.p2p_opp_ps = tmp;
2709                 if (params.p2p_opp_ps != 0 &&
2710                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_OPPPS))
2711                         return -EINVAL;
2712         }
2713
2714         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
2715                 err = nl80211_parse_chandef(rdev, info, &params.chandef);
2716                 if (err)
2717                         return err;
2718         } else if (wdev->preset_chandef.chan) {
2719                 params.chandef = wdev->preset_chandef;
2720         } else if (!nl80211_get_ap_channel(rdev, &params))
2721                 return -EINVAL;
2722
2723         if (!cfg80211_reg_can_beacon(&rdev->wiphy, &params.chandef))
2724                 return -EINVAL;
2725
2726         mutex_lock(&rdev->devlist_mtx);
2727         err = cfg80211_can_use_chan(rdev, wdev, params.chandef.chan,
2728                                     CHAN_MODE_SHARED);
2729         mutex_unlock(&rdev->devlist_mtx);
2730
2731         if (err)
2732                 return err;
2733
2734         err = rdev_start_ap(rdev, dev, &params);
2735         if (!err) {
2736                 wdev->preset_chandef = params.chandef;
2737                 wdev->beacon_interval = params.beacon_interval;
2738                 wdev->channel = params.chandef.chan;
2739                 wdev->ssid_len = params.ssid_len;
2740                 memcpy(wdev->ssid, params.ssid, wdev->ssid_len);
2741         }
2742         return err;
2743 }
2744
2745 static int nl80211_set_beacon(struct sk_buff *skb, struct genl_info *info)
2746 {
2747         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2748         struct net_device *dev = info->user_ptr[1];
2749         struct wireless_dev *wdev = dev->ieee80211_ptr;
2750         struct cfg80211_beacon_data params;
2751         int err;
2752
2753         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
2754             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
2755                 return -EOPNOTSUPP;
2756
2757         if (!rdev->ops->change_beacon)
2758                 return -EOPNOTSUPP;
2759
2760         if (!wdev->beacon_interval)
2761                 return -EINVAL;
2762
2763         err = nl80211_parse_beacon(info, &params);
2764         if (err)
2765                 return err;
2766
2767         return rdev_change_beacon(rdev, dev, &params);
2768 }
2769
2770 static int nl80211_stop_ap(struct sk_buff *skb, struct genl_info *info)
2771 {
2772         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2773         struct net_device *dev = info->user_ptr[1];
2774
2775         return cfg80211_stop_ap(rdev, dev);
2776 }
2777
2778 static const struct nla_policy sta_flags_policy[NL80211_STA_FLAG_MAX + 1] = {
2779         [NL80211_STA_FLAG_AUTHORIZED] = { .type = NLA_FLAG },
2780         [NL80211_STA_FLAG_SHORT_PREAMBLE] = { .type = NLA_FLAG },
2781         [NL80211_STA_FLAG_WME] = { .type = NLA_FLAG },
2782         [NL80211_STA_FLAG_MFP] = { .type = NLA_FLAG },
2783         [NL80211_STA_FLAG_AUTHENTICATED] = { .type = NLA_FLAG },
2784         [NL80211_STA_FLAG_TDLS_PEER] = { .type = NLA_FLAG },
2785 };
2786
2787 static int parse_station_flags(struct genl_info *info,
2788                                enum nl80211_iftype iftype,
2789                                struct station_parameters *params)
2790 {
2791         struct nlattr *flags[NL80211_STA_FLAG_MAX + 1];
2792         struct nlattr *nla;
2793         int flag;
2794
2795         /*
2796          * Try parsing the new attribute first so userspace
2797          * can specify both for older kernels.
2798          */
2799         nla = info->attrs[NL80211_ATTR_STA_FLAGS2];
2800         if (nla) {
2801                 struct nl80211_sta_flag_update *sta_flags;
2802
2803                 sta_flags = nla_data(nla);
2804                 params->sta_flags_mask = sta_flags->mask;
2805                 params->sta_flags_set = sta_flags->set;
2806                 if ((params->sta_flags_mask |
2807                      params->sta_flags_set) & BIT(__NL80211_STA_FLAG_INVALID))
2808                         return -EINVAL;
2809                 return 0;
2810         }
2811
2812         /* if present, parse the old attribute */
2813
2814         nla = info->attrs[NL80211_ATTR_STA_FLAGS];
2815         if (!nla)
2816                 return 0;
2817
2818         if (nla_parse_nested(flags, NL80211_STA_FLAG_MAX,
2819                              nla, sta_flags_policy))
2820                 return -EINVAL;
2821
2822         /*
2823          * Only allow certain flags for interface types so that
2824          * other attributes are silently ignored. Remember that
2825          * this is backward compatibility code with old userspace
2826          * and shouldn't be hit in other cases anyway.
2827          */
2828         switch (iftype) {
2829         case NL80211_IFTYPE_AP:
2830         case NL80211_IFTYPE_AP_VLAN:
2831         case NL80211_IFTYPE_P2P_GO:
2832                 params->sta_flags_mask = BIT(NL80211_STA_FLAG_AUTHORIZED) |
2833                                          BIT(NL80211_STA_FLAG_SHORT_PREAMBLE) |
2834                                          BIT(NL80211_STA_FLAG_WME) |
2835                                          BIT(NL80211_STA_FLAG_MFP);
2836                 break;
2837         case NL80211_IFTYPE_P2P_CLIENT:
2838         case NL80211_IFTYPE_STATION:
2839                 params->sta_flags_mask = BIT(NL80211_STA_FLAG_AUTHORIZED) |
2840                                          BIT(NL80211_STA_FLAG_TDLS_PEER);
2841                 break;
2842         case NL80211_IFTYPE_MESH_POINT:
2843                 params->sta_flags_mask = BIT(NL80211_STA_FLAG_AUTHENTICATED) |
2844                                          BIT(NL80211_STA_FLAG_MFP) |
2845                                          BIT(NL80211_STA_FLAG_AUTHORIZED);
2846         default:
2847                 return -EINVAL;
2848         }
2849
2850         for (flag = 1; flag <= NL80211_STA_FLAG_MAX; flag++) {
2851                 if (flags[flag]) {
2852                         params->sta_flags_set |= (1<<flag);
2853
2854                         /* no longer support new API additions in old API */
2855                         if (flag > NL80211_STA_FLAG_MAX_OLD_API)
2856                                 return -EINVAL;
2857                 }
2858         }
2859
2860         return 0;
2861 }
2862
2863 static bool nl80211_put_sta_rate(struct sk_buff *msg, struct rate_info *info,
2864                                  int attr)
2865 {
2866         struct nlattr *rate;
2867         u32 bitrate;
2868         u16 bitrate_compat;
2869
2870         rate = nla_nest_start(msg, attr);
2871         if (!rate)
2872                 return false;
2873
2874         /* cfg80211_calculate_bitrate will return 0 for mcs >= 32 */
2875         bitrate = cfg80211_calculate_bitrate(info);
2876         /* report 16-bit bitrate only if we can */
2877         bitrate_compat = bitrate < (1UL << 16) ? bitrate : 0;
2878         if (bitrate > 0 &&
2879             nla_put_u32(msg, NL80211_RATE_INFO_BITRATE32, bitrate))
2880                 return false;
2881         if (bitrate_compat > 0 &&
2882             nla_put_u16(msg, NL80211_RATE_INFO_BITRATE, bitrate_compat))
2883                 return false;
2884
2885         if (info->flags & RATE_INFO_FLAGS_MCS) {
2886                 if (nla_put_u8(msg, NL80211_RATE_INFO_MCS, info->mcs))
2887                         return false;
2888                 if (info->flags & RATE_INFO_FLAGS_40_MHZ_WIDTH &&
2889                     nla_put_flag(msg, NL80211_RATE_INFO_40_MHZ_WIDTH))
2890                         return false;
2891                 if (info->flags & RATE_INFO_FLAGS_SHORT_GI &&
2892                     nla_put_flag(msg, NL80211_RATE_INFO_SHORT_GI))
2893                         return false;
2894         } else if (info->flags & RATE_INFO_FLAGS_VHT_MCS) {
2895                 if (nla_put_u8(msg, NL80211_RATE_INFO_VHT_MCS, info->mcs))
2896                         return false;
2897                 if (nla_put_u8(msg, NL80211_RATE_INFO_VHT_NSS, info->nss))
2898                         return false;
2899                 if (info->flags & RATE_INFO_FLAGS_40_MHZ_WIDTH &&
2900                     nla_put_flag(msg, NL80211_RATE_INFO_40_MHZ_WIDTH))
2901                         return false;
2902                 if (info->flags & RATE_INFO_FLAGS_80_MHZ_WIDTH &&
2903                     nla_put_flag(msg, NL80211_RATE_INFO_80_MHZ_WIDTH))
2904                         return false;
2905                 if (info->flags & RATE_INFO_FLAGS_80P80_MHZ_WIDTH &&
2906                     nla_put_flag(msg, NL80211_RATE_INFO_80P80_MHZ_WIDTH))
2907                         return false;
2908                 if (info->flags & RATE_INFO_FLAGS_160_MHZ_WIDTH &&
2909                     nla_put_flag(msg, NL80211_RATE_INFO_160_MHZ_WIDTH))
2910                         return false;
2911                 if (info->flags & RATE_INFO_FLAGS_SHORT_GI &&
2912                     nla_put_flag(msg, NL80211_RATE_INFO_SHORT_GI))
2913                         return false;
2914         }
2915
2916         nla_nest_end(msg, rate);
2917         return true;
2918 }
2919
2920 static int nl80211_send_station(struct sk_buff *msg, u32 portid, u32 seq,
2921                                 int flags,
2922                                 struct cfg80211_registered_device *rdev,
2923                                 struct net_device *dev,
2924                                 const u8 *mac_addr, struct station_info *sinfo)
2925 {
2926         void *hdr;
2927         struct nlattr *sinfoattr, *bss_param;
2928
2929         hdr = nl80211hdr_put(msg, portid, seq, flags, NL80211_CMD_NEW_STATION);
2930         if (!hdr)
2931                 return -1;
2932
2933         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
2934             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr) ||
2935             nla_put_u32(msg, NL80211_ATTR_GENERATION, sinfo->generation))
2936                 goto nla_put_failure;
2937
2938         sinfoattr = nla_nest_start(msg, NL80211_ATTR_STA_INFO);
2939         if (!sinfoattr)
2940                 goto nla_put_failure;
2941         if ((sinfo->filled & STATION_INFO_CONNECTED_TIME) &&
2942             nla_put_u32(msg, NL80211_STA_INFO_CONNECTED_TIME,
2943                         sinfo->connected_time))
2944                 goto nla_put_failure;
2945         if ((sinfo->filled & STATION_INFO_INACTIVE_TIME) &&
2946             nla_put_u32(msg, NL80211_STA_INFO_INACTIVE_TIME,
2947                         sinfo->inactive_time))
2948                 goto nla_put_failure;
2949         if ((sinfo->filled & STATION_INFO_RX_BYTES) &&
2950             nla_put_u32(msg, NL80211_STA_INFO_RX_BYTES,
2951                         sinfo->rx_bytes))
2952                 goto nla_put_failure;
2953         if ((sinfo->filled & STATION_INFO_TX_BYTES) &&
2954             nla_put_u32(msg, NL80211_STA_INFO_TX_BYTES,
2955                         sinfo->tx_bytes))
2956                 goto nla_put_failure;
2957         if ((sinfo->filled & STATION_INFO_LLID) &&
2958             nla_put_u16(msg, NL80211_STA_INFO_LLID, sinfo->llid))
2959                 goto nla_put_failure;
2960         if ((sinfo->filled & STATION_INFO_PLID) &&
2961             nla_put_u16(msg, NL80211_STA_INFO_PLID, sinfo->plid))
2962                 goto nla_put_failure;
2963         if ((sinfo->filled & STATION_INFO_PLINK_STATE) &&
2964             nla_put_u8(msg, NL80211_STA_INFO_PLINK_STATE,
2965                        sinfo->plink_state))
2966                 goto nla_put_failure;
2967         switch (rdev->wiphy.signal_type) {
2968         case CFG80211_SIGNAL_TYPE_MBM:
2969                 if ((sinfo->filled & STATION_INFO_SIGNAL) &&
2970                     nla_put_u8(msg, NL80211_STA_INFO_SIGNAL,
2971                                sinfo->signal))
2972                         goto nla_put_failure;
2973                 if ((sinfo->filled & STATION_INFO_SIGNAL_AVG) &&
2974                     nla_put_u8(msg, NL80211_STA_INFO_SIGNAL_AVG,
2975                                sinfo->signal_avg))
2976                         goto nla_put_failure;
2977                 break;
2978         default:
2979                 break;
2980         }
2981         if (sinfo->filled & STATION_INFO_TX_BITRATE) {
2982                 if (!nl80211_put_sta_rate(msg, &sinfo->txrate,
2983                                           NL80211_STA_INFO_TX_BITRATE))
2984                         goto nla_put_failure;
2985         }
2986         if (sinfo->filled & STATION_INFO_RX_BITRATE) {
2987                 if (!nl80211_put_sta_rate(msg, &sinfo->rxrate,
2988                                           NL80211_STA_INFO_RX_BITRATE))
2989                         goto nla_put_failure;
2990         }
2991         if ((sinfo->filled & STATION_INFO_RX_PACKETS) &&
2992             nla_put_u32(msg, NL80211_STA_INFO_RX_PACKETS,
2993                         sinfo->rx_packets))
2994                 goto nla_put_failure;
2995         if ((sinfo->filled & STATION_INFO_TX_PACKETS) &&
2996             nla_put_u32(msg, NL80211_STA_INFO_TX_PACKETS,
2997                         sinfo->tx_packets))
2998                 goto nla_put_failure;
2999         if ((sinfo->filled & STATION_INFO_TX_RETRIES) &&
3000             nla_put_u32(msg, NL80211_STA_INFO_TX_RETRIES,
3001                         sinfo->tx_retries))
3002                 goto nla_put_failure;
3003         if ((sinfo->filled & STATION_INFO_TX_FAILED) &&
3004             nla_put_u32(msg, NL80211_STA_INFO_TX_FAILED,
3005                         sinfo->tx_failed))
3006                 goto nla_put_failure;
3007         if ((sinfo->filled & STATION_INFO_BEACON_LOSS_COUNT) &&
3008             nla_put_u32(msg, NL80211_STA_INFO_BEACON_LOSS,
3009                         sinfo->beacon_loss_count))
3010                 goto nla_put_failure;
3011         if ((sinfo->filled & STATION_INFO_LOCAL_PM) &&
3012             nla_put_u32(msg, NL80211_STA_INFO_LOCAL_PM,
3013                         sinfo->local_pm))
3014                 goto nla_put_failure;
3015         if ((sinfo->filled & STATION_INFO_PEER_PM) &&
3016             nla_put_u32(msg, NL80211_STA_INFO_PEER_PM,
3017                         sinfo->peer_pm))
3018                 goto nla_put_failure;
3019         if ((sinfo->filled & STATION_INFO_NONPEER_PM) &&
3020             nla_put_u32(msg, NL80211_STA_INFO_NONPEER_PM,
3021                         sinfo->nonpeer_pm))
3022                 goto nla_put_failure;
3023         if (sinfo->filled & STATION_INFO_BSS_PARAM) {
3024                 bss_param = nla_nest_start(msg, NL80211_STA_INFO_BSS_PARAM);
3025                 if (!bss_param)
3026                         goto nla_put_failure;
3027
3028                 if (((sinfo->bss_param.flags & BSS_PARAM_FLAGS_CTS_PROT) &&
3029                      nla_put_flag(msg, NL80211_STA_BSS_PARAM_CTS_PROT)) ||
3030                     ((sinfo->bss_param.flags & BSS_PARAM_FLAGS_SHORT_PREAMBLE) &&
3031                      nla_put_flag(msg, NL80211_STA_BSS_PARAM_SHORT_PREAMBLE)) ||
3032                     ((sinfo->bss_param.flags & BSS_PARAM_FLAGS_SHORT_SLOT_TIME) &&
3033                      nla_put_flag(msg, NL80211_STA_BSS_PARAM_SHORT_SLOT_TIME)) ||
3034                     nla_put_u8(msg, NL80211_STA_BSS_PARAM_DTIM_PERIOD,
3035                                sinfo->bss_param.dtim_period) ||
3036                     nla_put_u16(msg, NL80211_STA_BSS_PARAM_BEACON_INTERVAL,
3037                                 sinfo->bss_param.beacon_interval))
3038                         goto nla_put_failure;
3039
3040                 nla_nest_end(msg, bss_param);
3041         }
3042         if ((sinfo->filled & STATION_INFO_STA_FLAGS) &&
3043             nla_put(msg, NL80211_STA_INFO_STA_FLAGS,
3044                     sizeof(struct nl80211_sta_flag_update),
3045                     &sinfo->sta_flags))
3046                 goto nla_put_failure;
3047         if ((sinfo->filled & STATION_INFO_T_OFFSET) &&
3048                 nla_put_u64(msg, NL80211_STA_INFO_T_OFFSET,
3049                             sinfo->t_offset))
3050                 goto nla_put_failure;
3051         nla_nest_end(msg, sinfoattr);
3052
3053         if ((sinfo->filled & STATION_INFO_ASSOC_REQ_IES) &&
3054             nla_put(msg, NL80211_ATTR_IE, sinfo->assoc_req_ies_len,
3055                     sinfo->assoc_req_ies))
3056                 goto nla_put_failure;
3057
3058         return genlmsg_end(msg, hdr);
3059
3060  nla_put_failure:
3061         genlmsg_cancel(msg, hdr);
3062         return -EMSGSIZE;
3063 }
3064
3065 static int nl80211_dump_station(struct sk_buff *skb,
3066                                 struct netlink_callback *cb)
3067 {
3068         struct station_info sinfo;
3069         struct cfg80211_registered_device *dev;
3070         struct net_device *netdev;
3071         u8 mac_addr[ETH_ALEN];
3072         int sta_idx = cb->args[1];
3073         int err;
3074
3075         err = nl80211_prepare_netdev_dump(skb, cb, &dev, &netdev);
3076         if (err)
3077                 return err;
3078
3079         if (!dev->ops->dump_station) {
3080                 err = -EOPNOTSUPP;
3081                 goto out_err;
3082         }
3083
3084         while (1) {
3085                 memset(&sinfo, 0, sizeof(sinfo));
3086                 err = rdev_dump_station(dev, netdev, sta_idx,
3087                                         mac_addr, &sinfo);
3088                 if (err == -ENOENT)
3089                         break;
3090                 if (err)
3091                         goto out_err;
3092
3093                 if (nl80211_send_station(skb,
3094                                 NETLINK_CB(cb->skb).portid,
3095                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
3096                                 dev, netdev, mac_addr,
3097                                 &sinfo) < 0)
3098                         goto out;
3099
3100                 sta_idx++;
3101         }
3102
3103
3104  out:
3105         cb->args[1] = sta_idx;
3106         err = skb->len;
3107  out_err:
3108         nl80211_finish_netdev_dump(dev);
3109
3110         return err;
3111 }
3112
3113 static int nl80211_get_station(struct sk_buff *skb, struct genl_info *info)
3114 {
3115         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3116         struct net_device *dev = info->user_ptr[1];
3117         struct station_info sinfo;
3118         struct sk_buff *msg;
3119         u8 *mac_addr = NULL;
3120         int err;
3121
3122         memset(&sinfo, 0, sizeof(sinfo));
3123
3124         if (!info->attrs[NL80211_ATTR_MAC])
3125                 return -EINVAL;
3126
3127         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
3128
3129         if (!rdev->ops->get_station)
3130                 return -EOPNOTSUPP;
3131
3132         err = rdev_get_station(rdev, dev, mac_addr, &sinfo);
3133         if (err)
3134                 return err;
3135
3136         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3137         if (!msg)
3138                 return -ENOMEM;
3139
3140         if (nl80211_send_station(msg, info->snd_portid, info->snd_seq, 0,
3141                                  rdev, dev, mac_addr, &sinfo) < 0) {
3142                 nlmsg_free(msg);
3143                 return -ENOBUFS;
3144         }
3145
3146         return genlmsg_reply(msg, info);
3147 }
3148
3149 /*
3150  * Get vlan interface making sure it is running and on the right wiphy.
3151  */
3152 static struct net_device *get_vlan(struct genl_info *info,
3153                                    struct cfg80211_registered_device *rdev)
3154 {
3155         struct nlattr *vlanattr = info->attrs[NL80211_ATTR_STA_VLAN];
3156         struct net_device *v;
3157         int ret;
3158
3159         if (!vlanattr)
3160                 return NULL;
3161
3162         v = dev_get_by_index(genl_info_net(info), nla_get_u32(vlanattr));
3163         if (!v)
3164                 return ERR_PTR(-ENODEV);
3165
3166         if (!v->ieee80211_ptr || v->ieee80211_ptr->wiphy != &rdev->wiphy) {
3167                 ret = -EINVAL;
3168                 goto error;
3169         }
3170
3171         if (!netif_running(v)) {
3172                 ret = -ENETDOWN;
3173                 goto error;
3174         }
3175
3176         return v;
3177  error:
3178         dev_put(v);
3179         return ERR_PTR(ret);
3180 }
3181
3182 static int nl80211_set_station(struct sk_buff *skb, struct genl_info *info)
3183 {
3184         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3185         int err;
3186         struct net_device *dev = info->user_ptr[1];
3187         struct station_parameters params;
3188         u8 *mac_addr = NULL;
3189
3190         memset(&params, 0, sizeof(params));
3191
3192         params.listen_interval = -1;
3193         params.plink_state = -1;
3194
3195         if (info->attrs[NL80211_ATTR_STA_AID])
3196                 return -EINVAL;
3197
3198         if (!info->attrs[NL80211_ATTR_MAC])
3199                 return -EINVAL;
3200
3201         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
3202
3203         if (info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]) {
3204                 params.supported_rates =
3205                         nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
3206                 params.supported_rates_len =
3207                         nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
3208         }
3209
3210         if (info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL] ||
3211             info->attrs[NL80211_ATTR_HT_CAPABILITY])
3212                 return -EINVAL;
3213
3214         if (!rdev->ops->change_station)
3215                 return -EOPNOTSUPP;
3216
3217         if (parse_station_flags(info, dev->ieee80211_ptr->iftype, &params))
3218                 return -EINVAL;
3219
3220         if (info->attrs[NL80211_ATTR_STA_PLINK_ACTION])
3221                 params.plink_action =
3222                     nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_ACTION]);
3223
3224         if (info->attrs[NL80211_ATTR_STA_PLINK_STATE])
3225                 params.plink_state =
3226                     nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_STATE]);
3227
3228         if (info->attrs[NL80211_ATTR_LOCAL_MESH_POWER_MODE]) {
3229                 enum nl80211_mesh_power_mode pm = nla_get_u32(
3230                         info->attrs[NL80211_ATTR_LOCAL_MESH_POWER_MODE]);
3231
3232                 if (pm <= NL80211_MESH_POWER_UNKNOWN ||
3233                     pm > NL80211_MESH_POWER_MAX)
3234                         return -EINVAL;
3235
3236                 params.local_pm = pm;
3237         }
3238
3239         switch (dev->ieee80211_ptr->iftype) {
3240         case NL80211_IFTYPE_AP:
3241         case NL80211_IFTYPE_AP_VLAN:
3242         case NL80211_IFTYPE_P2P_GO:
3243                 /* disallow mesh-specific things */
3244                 if (params.plink_action)
3245                         return -EINVAL;
3246                 if (params.local_pm)
3247                         return -EINVAL;
3248
3249                 /* TDLS can't be set, ... */
3250                 if (params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER))
3251                         return -EINVAL;
3252                 /*
3253                  * ... but don't bother the driver with it. This works around
3254                  * a hostapd/wpa_supplicant issue -- it always includes the
3255                  * TLDS_PEER flag in the mask even for AP mode.
3256                  */
3257                 params.sta_flags_mask &= ~BIT(NL80211_STA_FLAG_TDLS_PEER);
3258
3259                 /* accept only the listed bits */
3260                 if (params.sta_flags_mask &
3261                                 ~(BIT(NL80211_STA_FLAG_AUTHORIZED) |
3262                                   BIT(NL80211_STA_FLAG_AUTHENTICATED) |
3263                                   BIT(NL80211_STA_FLAG_ASSOCIATED) |
3264                                   BIT(NL80211_STA_FLAG_SHORT_PREAMBLE) |
3265                                   BIT(NL80211_STA_FLAG_WME) |
3266                                   BIT(NL80211_STA_FLAG_MFP)))
3267                         return -EINVAL;
3268
3269                 /* but authenticated/associated only if driver handles it */
3270                 if (!(rdev->wiphy.features &
3271                                 NL80211_FEATURE_FULL_AP_CLIENT_STATE) &&
3272                     params.sta_flags_mask &
3273                                 (BIT(NL80211_STA_FLAG_AUTHENTICATED) |
3274                                  BIT(NL80211_STA_FLAG_ASSOCIATED)))
3275                         return -EINVAL;
3276
3277                 /* reject other things that can't change */
3278                 if (params.supported_rates)
3279                         return -EINVAL;
3280
3281                 /* must be last in here for error handling */
3282                 params.vlan = get_vlan(info, rdev);
3283                 if (IS_ERR(params.vlan))
3284                         return PTR_ERR(params.vlan);
3285                 break;
3286         case NL80211_IFTYPE_P2P_CLIENT:
3287         case NL80211_IFTYPE_STATION:
3288                 /*
3289                  * Don't allow userspace to change the TDLS_PEER flag,
3290                  * but silently ignore attempts to change it since we
3291                  * don't have state here to verify that it doesn't try
3292                  * to change the flag.
3293                  */
3294                 params.sta_flags_mask &= ~BIT(NL80211_STA_FLAG_TDLS_PEER);
3295                 /* fall through */
3296         case NL80211_IFTYPE_ADHOC:
3297                 /* disallow things sta doesn't support */
3298                 if (params.plink_action)
3299                         return -EINVAL;
3300                 if (params.local_pm)
3301                         return -EINVAL;
3302                 /* reject any changes other than AUTHORIZED */
3303                 if (params.sta_flags_mask & ~BIT(NL80211_STA_FLAG_AUTHORIZED))
3304                         return -EINVAL;
3305                 break;
3306         case NL80211_IFTYPE_MESH_POINT:
3307                 /* disallow things mesh doesn't support */
3308                 if (params.vlan)
3309                         return -EINVAL;
3310                 if (params.supported_rates)
3311                         return -EINVAL;
3312                 /*
3313                  * No special handling for TDLS here -- the userspace
3314                  * mesh code doesn't have this bug.
3315                  */
3316                 if (params.sta_flags_mask &
3317                                 ~(BIT(NL80211_STA_FLAG_AUTHENTICATED) |
3318                                   BIT(NL80211_STA_FLAG_MFP) |
3319                                   BIT(NL80211_STA_FLAG_AUTHORIZED)))
3320                         return -EINVAL;
3321                 break;
3322         default:
3323                 return -EOPNOTSUPP;
3324         }
3325
3326         /* be aware of params.vlan when changing code here */
3327
3328         err = rdev_change_station(rdev, dev, mac_addr, &params);
3329
3330         if (params.vlan)
3331                 dev_put(params.vlan);
3332
3333         return err;
3334 }
3335
3336 static struct nla_policy
3337 nl80211_sta_wme_policy[NL80211_STA_WME_MAX + 1] __read_mostly = {
3338         [NL80211_STA_WME_UAPSD_QUEUES] = { .type = NLA_U8 },
3339         [NL80211_STA_WME_MAX_SP] = { .type = NLA_U8 },
3340 };
3341
3342 static int nl80211_new_station(struct sk_buff *skb, struct genl_info *info)
3343 {
3344         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3345         int err;
3346         struct net_device *dev = info->user_ptr[1];
3347         struct station_parameters params;
3348         u8 *mac_addr = NULL;
3349
3350         memset(&params, 0, sizeof(params));
3351
3352         if (!info->attrs[NL80211_ATTR_MAC])
3353                 return -EINVAL;
3354
3355         if (!info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL])
3356                 return -EINVAL;
3357
3358         if (!info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES])
3359                 return -EINVAL;
3360
3361         if (!info->attrs[NL80211_ATTR_STA_AID])
3362                 return -EINVAL;
3363
3364         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
3365         params.supported_rates =
3366                 nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
3367         params.supported_rates_len =
3368                 nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
3369         params.listen_interval =
3370                 nla_get_u16(info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL]);
3371
3372         params.aid = nla_get_u16(info->attrs[NL80211_ATTR_STA_AID]);
3373         if (!params.aid || params.aid > IEEE80211_MAX_AID)
3374                 return -EINVAL;
3375
3376         if (info->attrs[NL80211_ATTR_HT_CAPABILITY])
3377                 params.ht_capa =
3378                         nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]);
3379
3380         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY])
3381                 params.vht_capa =
3382                         nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY]);
3383
3384         if (info->attrs[NL80211_ATTR_STA_PLINK_ACTION])
3385                 params.plink_action =
3386                     nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_ACTION]);
3387
3388         if (!rdev->ops->add_station)
3389                 return -EOPNOTSUPP;
3390
3391         if (parse_station_flags(info, dev->ieee80211_ptr->iftype, &params))
3392                 return -EINVAL;
3393
3394         switch (dev->ieee80211_ptr->iftype) {
3395         case NL80211_IFTYPE_AP:
3396         case NL80211_IFTYPE_AP_VLAN:
3397         case NL80211_IFTYPE_P2P_GO:
3398                 /* parse WME attributes if sta is WME capable */
3399                 if ((rdev->wiphy.flags & WIPHY_FLAG_AP_UAPSD) &&
3400                     (params.sta_flags_set & BIT(NL80211_STA_FLAG_WME)) &&
3401                     info->attrs[NL80211_ATTR_STA_WME]) {
3402                         struct nlattr *tb[NL80211_STA_WME_MAX + 1];
3403                         struct nlattr *nla;
3404
3405                         nla = info->attrs[NL80211_ATTR_STA_WME];
3406                         err = nla_parse_nested(tb, NL80211_STA_WME_MAX, nla,
3407                                                nl80211_sta_wme_policy);
3408                         if (err)
3409                                 return err;
3410
3411                         if (tb[NL80211_STA_WME_UAPSD_QUEUES])
3412                                 params.uapsd_queues =
3413                                      nla_get_u8(tb[NL80211_STA_WME_UAPSD_QUEUES]);
3414                         if (params.uapsd_queues &
3415                                         ~IEEE80211_WMM_IE_STA_QOSINFO_AC_MASK)
3416                                 return -EINVAL;
3417
3418                         if (tb[NL80211_STA_WME_MAX_SP])
3419                                 params.max_sp =
3420                                      nla_get_u8(tb[NL80211_STA_WME_MAX_SP]);
3421
3422                         if (params.max_sp &
3423                                         ~IEEE80211_WMM_IE_STA_QOSINFO_SP_MASK)
3424                                 return -EINVAL;
3425
3426                         params.sta_modify_mask |= STATION_PARAM_APPLY_UAPSD;
3427                 }
3428                 /* TDLS peers cannot be added */
3429                 if (params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER))
3430                         return -EINVAL;
3431                 /* but don't bother the driver with it */
3432                 params.sta_flags_mask &= ~BIT(NL80211_STA_FLAG_TDLS_PEER);
3433
3434                 /* allow authenticated/associated only if driver handles it */
3435                 if (!(rdev->wiphy.features &
3436                                 NL80211_FEATURE_FULL_AP_CLIENT_STATE) &&
3437                     params.sta_flags_mask &
3438                                 (BIT(NL80211_STA_FLAG_AUTHENTICATED) |
3439                                  BIT(NL80211_STA_FLAG_ASSOCIATED)))
3440                         return -EINVAL;
3441
3442                 /* must be last in here for error handling */
3443                 params.vlan = get_vlan(info, rdev);
3444                 if (IS_ERR(params.vlan))
3445                         return PTR_ERR(params.vlan);
3446                 break;
3447         case NL80211_IFTYPE_MESH_POINT:
3448                 /* associated is disallowed */
3449                 if (params.sta_flags_mask & BIT(NL80211_STA_FLAG_ASSOCIATED))
3450                         return -EINVAL;
3451                 /* TDLS peers cannot be added */
3452                 if (params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER))
3453                         return -EINVAL;
3454                 break;
3455         case NL80211_IFTYPE_STATION:
3456                 /* associated is disallowed */
3457                 if (params.sta_flags_mask & BIT(NL80211_STA_FLAG_ASSOCIATED))
3458                         return -EINVAL;
3459                 /* Only TDLS peers can be added */
3460                 if (!(params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)))
3461                         return -EINVAL;
3462                 /* Can only add if TDLS ... */
3463                 if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS))
3464                         return -EOPNOTSUPP;
3465                 /* ... with external setup is supported */
3466                 if (!(rdev->wiphy.flags & WIPHY_FLAG_TDLS_EXTERNAL_SETUP))
3467                         return -EOPNOTSUPP;
3468                 break;
3469         default:
3470                 return -EOPNOTSUPP;
3471         }
3472
3473         /* be aware of params.vlan when changing code here */
3474
3475         err = rdev_add_station(rdev, dev, mac_addr, &params);
3476
3477         if (params.vlan)
3478                 dev_put(params.vlan);
3479         return err;
3480 }
3481
3482 static int nl80211_del_station(struct sk_buff *skb, struct genl_info *info)
3483 {
3484         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3485         struct net_device *dev = info->user_ptr[1];
3486         u8 *mac_addr = NULL;
3487
3488         if (info->attrs[NL80211_ATTR_MAC])
3489                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
3490
3491         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
3492             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP_VLAN &&
3493             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT &&
3494             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3495                 return -EINVAL;
3496
3497         if (!rdev->ops->del_station)
3498                 return -EOPNOTSUPP;
3499
3500         return rdev_del_station(rdev, dev, mac_addr);
3501 }
3502
3503 static int nl80211_send_mpath(struct sk_buff *msg, u32 portid, u32 seq,
3504                                 int flags, struct net_device *dev,
3505                                 u8 *dst, u8 *next_hop,
3506                                 struct mpath_info *pinfo)
3507 {
3508         void *hdr;
3509         struct nlattr *pinfoattr;
3510
3511         hdr = nl80211hdr_put(msg, portid, seq, flags, NL80211_CMD_NEW_STATION);
3512         if (!hdr)
3513                 return -1;
3514
3515         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
3516             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, dst) ||
3517             nla_put(msg, NL80211_ATTR_MPATH_NEXT_HOP, ETH_ALEN, next_hop) ||
3518             nla_put_u32(msg, NL80211_ATTR_GENERATION, pinfo->generation))
3519                 goto nla_put_failure;
3520
3521         pinfoattr = nla_nest_start(msg, NL80211_ATTR_MPATH_INFO);
3522         if (!pinfoattr)
3523                 goto nla_put_failure;
3524         if ((pinfo->filled & MPATH_INFO_FRAME_QLEN) &&
3525             nla_put_u32(msg, NL80211_MPATH_INFO_FRAME_QLEN,
3526                         pinfo->frame_qlen))
3527                 goto nla_put_failure;
3528         if (((pinfo->filled & MPATH_INFO_SN) &&
3529              nla_put_u32(msg, NL80211_MPATH_INFO_SN, pinfo->sn)) ||
3530             ((pinfo->filled & MPATH_INFO_METRIC) &&
3531              nla_put_u32(msg, NL80211_MPATH_INFO_METRIC,
3532                          pinfo->metric)) ||
3533             ((pinfo->filled & MPATH_INFO_EXPTIME) &&
3534              nla_put_u32(msg, NL80211_MPATH_INFO_EXPTIME,
3535                          pinfo->exptime)) ||
3536             ((pinfo->filled & MPATH_INFO_FLAGS) &&
3537              nla_put_u8(msg, NL80211_MPATH_INFO_FLAGS,
3538                         pinfo->flags)) ||
3539             ((pinfo->filled & MPATH_INFO_DISCOVERY_TIMEOUT) &&
3540              nla_put_u32(msg, NL80211_MPATH_INFO_DISCOVERY_TIMEOUT,
3541                          pinfo->discovery_timeout)) ||
3542             ((pinfo->filled & MPATH_INFO_DISCOVERY_RETRIES) &&
3543              nla_put_u8(msg, NL80211_MPATH_INFO_DISCOVERY_RETRIES,
3544                         pinfo->discovery_retries)))
3545                 goto nla_put_failure;
3546
3547         nla_nest_end(msg, pinfoattr);
3548
3549         return genlmsg_end(msg, hdr);
3550
3551  nla_put_failure:
3552         genlmsg_cancel(msg, hdr);
3553         return -EMSGSIZE;
3554 }
3555
3556 static int nl80211_dump_mpath(struct sk_buff *skb,
3557                               struct netlink_callback *cb)
3558 {
3559         struct mpath_info pinfo;
3560         struct cfg80211_registered_device *dev;
3561         struct net_device *netdev;
3562         u8 dst[ETH_ALEN];
3563         u8 next_hop[ETH_ALEN];
3564         int path_idx = cb->args[1];
3565         int err;
3566
3567         err = nl80211_prepare_netdev_dump(skb, cb, &dev, &netdev);
3568         if (err)
3569                 return err;
3570
3571         if (!dev->ops->dump_mpath) {
3572                 err = -EOPNOTSUPP;
3573                 goto out_err;
3574         }
3575
3576         if (netdev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT) {
3577                 err = -EOPNOTSUPP;
3578                 goto out_err;
3579         }
3580
3581         while (1) {
3582                 err = rdev_dump_mpath(dev, netdev, path_idx, dst, next_hop,
3583                                       &pinfo);
3584                 if (err == -ENOENT)
3585                         break;
3586                 if (err)
3587                         goto out_err;
3588
3589                 if (nl80211_send_mpath(skb, NETLINK_CB(cb->skb).portid,
3590                                        cb->nlh->nlmsg_seq, NLM_F_MULTI,
3591                                        netdev, dst, next_hop,
3592                                        &pinfo) < 0)
3593                         goto out;
3594
3595                 path_idx++;
3596         }
3597
3598
3599  out:
3600         cb->args[1] = path_idx;
3601         err = skb->len;
3602  out_err:
3603         nl80211_finish_netdev_dump(dev);
3604         return err;
3605 }
3606
3607 static int nl80211_get_mpath(struct sk_buff *skb, struct genl_info *info)
3608 {
3609         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3610         int err;
3611         struct net_device *dev = info->user_ptr[1];
3612         struct mpath_info pinfo;
3613         struct sk_buff *msg;
3614         u8 *dst = NULL;
3615         u8 next_hop[ETH_ALEN];
3616
3617         memset(&pinfo, 0, sizeof(pinfo));
3618
3619         if (!info->attrs[NL80211_ATTR_MAC])
3620                 return -EINVAL;
3621
3622         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
3623
3624         if (!rdev->ops->get_mpath)
3625                 return -EOPNOTSUPP;
3626
3627         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
3628                 return -EOPNOTSUPP;
3629
3630         err = rdev_get_mpath(rdev, dev, dst, next_hop, &pinfo);
3631         if (err)
3632                 return err;
3633
3634         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3635         if (!msg)
3636                 return -ENOMEM;
3637
3638         if (nl80211_send_mpath(msg, info->snd_portid, info->snd_seq, 0,
3639                                  dev, dst, next_hop, &pinfo) < 0) {
3640                 nlmsg_free(msg);
3641                 return -ENOBUFS;
3642         }
3643
3644         return genlmsg_reply(msg, info);
3645 }
3646
3647 static int nl80211_set_mpath(struct sk_buff *skb, struct genl_info *info)
3648 {
3649         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3650         struct net_device *dev = info->user_ptr[1];
3651         u8 *dst = NULL;
3652         u8 *next_hop = NULL;
3653
3654         if (!info->attrs[NL80211_ATTR_MAC])
3655                 return -EINVAL;
3656
3657         if (!info->attrs[NL80211_ATTR_MPATH_NEXT_HOP])
3658                 return -EINVAL;
3659
3660         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
3661         next_hop = nla_data(info->attrs[NL80211_ATTR_MPATH_NEXT_HOP]);
3662
3663         if (!rdev->ops->change_mpath)
3664                 return -EOPNOTSUPP;
3665
3666         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
3667                 return -EOPNOTSUPP;
3668
3669         return rdev_change_mpath(rdev, dev, dst, next_hop);
3670 }
3671
3672 static int nl80211_new_mpath(struct sk_buff *skb, struct genl_info *info)
3673 {
3674         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3675         struct net_device *dev = info->user_ptr[1];
3676         u8 *dst = NULL;
3677         u8 *next_hop = NULL;
3678
3679         if (!info->attrs[NL80211_ATTR_MAC])
3680                 return -EINVAL;
3681
3682         if (!info->attrs[NL80211_ATTR_MPATH_NEXT_HOP])
3683                 return -EINVAL;
3684
3685         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
3686         next_hop = nla_data(info->attrs[NL80211_ATTR_MPATH_NEXT_HOP]);
3687
3688         if (!rdev->ops->add_mpath)
3689                 return -EOPNOTSUPP;
3690
3691         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
3692                 return -EOPNOTSUPP;
3693
3694         return rdev_add_mpath(rdev, dev, dst, next_hop);
3695 }
3696
3697 static int nl80211_del_mpath(struct sk_buff *skb, struct genl_info *info)
3698 {
3699         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3700         struct net_device *dev = info->user_ptr[1];
3701         u8 *dst = NULL;
3702
3703         if (info->attrs[NL80211_ATTR_MAC])
3704                 dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
3705
3706         if (!rdev->ops->del_mpath)
3707                 return -EOPNOTSUPP;
3708
3709         return rdev_del_mpath(rdev, dev, dst);
3710 }
3711
3712 static int nl80211_set_bss(struct sk_buff *skb, struct genl_info *info)
3713 {
3714         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3715         struct net_device *dev = info->user_ptr[1];
3716         struct bss_parameters params;
3717
3718         memset(&params, 0, sizeof(params));
3719         /* default to not changing parameters */
3720         params.use_cts_prot = -1;
3721         params.use_short_preamble = -1;
3722         params.use_short_slot_time = -1;
3723         params.ap_isolate = -1;
3724         params.ht_opmode = -1;
3725         params.p2p_ctwindow = -1;
3726         params.p2p_opp_ps = -1;
3727
3728         if (info->attrs[NL80211_ATTR_BSS_CTS_PROT])
3729                 params.use_cts_prot =
3730                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_CTS_PROT]);
3731         if (info->attrs[NL80211_ATTR_BSS_SHORT_PREAMBLE])
3732                 params.use_short_preamble =
3733                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_SHORT_PREAMBLE]);
3734         if (info->attrs[NL80211_ATTR_BSS_SHORT_SLOT_TIME])
3735                 params.use_short_slot_time =
3736                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_SHORT_SLOT_TIME]);
3737         if (info->attrs[NL80211_ATTR_BSS_BASIC_RATES]) {
3738                 params.basic_rates =
3739                         nla_data(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
3740                 params.basic_rates_len =
3741                         nla_len(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
3742         }
3743         if (info->attrs[NL80211_ATTR_AP_ISOLATE])
3744                 params.ap_isolate = !!nla_get_u8(info->attrs[NL80211_ATTR_AP_ISOLATE]);
3745         if (info->attrs[NL80211_ATTR_BSS_HT_OPMODE])
3746                 params.ht_opmode =
3747                         nla_get_u16(info->attrs[NL80211_ATTR_BSS_HT_OPMODE]);
3748
3749         if (info->attrs[NL80211_ATTR_P2P_CTWINDOW]) {
3750                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3751                         return -EINVAL;
3752                 params.p2p_ctwindow =
3753                         nla_get_s8(info->attrs[NL80211_ATTR_P2P_CTWINDOW]);
3754                 if (params.p2p_ctwindow < 0)
3755                         return -EINVAL;
3756                 if (params.p2p_ctwindow != 0 &&
3757                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_CTWIN))
3758                         return -EINVAL;
3759         }
3760
3761         if (info->attrs[NL80211_ATTR_P2P_OPPPS]) {
3762                 u8 tmp;
3763
3764                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3765                         return -EINVAL;
3766                 tmp = nla_get_u8(info->attrs[NL80211_ATTR_P2P_OPPPS]);
3767                 if (tmp > 1)
3768                         return -EINVAL;
3769                 params.p2p_opp_ps = tmp;
3770                 if (params.p2p_opp_ps &&
3771                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_OPPPS))
3772                         return -EINVAL;
3773         }
3774
3775         if (!rdev->ops->change_bss)
3776                 return -EOPNOTSUPP;
3777
3778         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
3779             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3780                 return -EOPNOTSUPP;
3781
3782         return rdev_change_bss(rdev, dev, &params);
3783 }
3784
3785 static const struct nla_policy reg_rule_policy[NL80211_REG_RULE_ATTR_MAX + 1] = {
3786         [NL80211_ATTR_REG_RULE_FLAGS]           = { .type = NLA_U32 },
3787         [NL80211_ATTR_FREQ_RANGE_START]         = { .type = NLA_U32 },
3788         [NL80211_ATTR_FREQ_RANGE_END]           = { .type = NLA_U32 },
3789         [NL80211_ATTR_FREQ_RANGE_MAX_BW]        = { .type = NLA_U32 },
3790         [NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN]  = { .type = NLA_U32 },
3791         [NL80211_ATTR_POWER_RULE_MAX_EIRP]      = { .type = NLA_U32 },
3792 };
3793
3794 static int parse_reg_rule(struct nlattr *tb[],
3795         struct ieee80211_reg_rule *reg_rule)
3796 {
3797         struct ieee80211_freq_range *freq_range = &reg_rule->freq_range;
3798         struct ieee80211_power_rule *power_rule = &reg_rule->power_rule;
3799
3800         if (!tb[NL80211_ATTR_REG_RULE_FLAGS])
3801                 return -EINVAL;
3802         if (!tb[NL80211_ATTR_FREQ_RANGE_START])
3803                 return -EINVAL;
3804         if (!tb[NL80211_ATTR_FREQ_RANGE_END])
3805                 return -EINVAL;
3806         if (!tb[NL80211_ATTR_FREQ_RANGE_MAX_BW])
3807                 return -EINVAL;
3808         if (!tb[NL80211_ATTR_POWER_RULE_MAX_EIRP])
3809                 return -EINVAL;
3810
3811         reg_rule->flags = nla_get_u32(tb[NL80211_ATTR_REG_RULE_FLAGS]);
3812
3813         freq_range->start_freq_khz =
3814                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]);
3815         freq_range->end_freq_khz =
3816                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]);
3817         freq_range->max_bandwidth_khz =
3818                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]);
3819
3820         power_rule->max_eirp =
3821                 nla_get_u32(tb[NL80211_ATTR_POWER_RULE_MAX_EIRP]);
3822
3823         if (tb[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN])
3824                 power_rule->max_antenna_gain =
3825                         nla_get_u32(tb[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN]);
3826
3827         return 0;
3828 }
3829
3830 static int nl80211_req_set_reg(struct sk_buff *skb, struct genl_info *info)
3831 {
3832         int r;
3833         char *data = NULL;
3834         enum nl80211_user_reg_hint_type user_reg_hint_type;
3835
3836         /*
3837          * You should only get this when cfg80211 hasn't yet initialized
3838          * completely when built-in to the kernel right between the time
3839          * window between nl80211_init() and regulatory_init(), if that is
3840          * even possible.
3841          */
3842         if (unlikely(!rcu_access_pointer(cfg80211_regdomain)))
3843                 return -EINPROGRESS;
3844
3845         if (!info->attrs[NL80211_ATTR_REG_ALPHA2])
3846                 return -EINVAL;
3847
3848         data = nla_data(info->attrs[NL80211_ATTR_REG_ALPHA2]);
3849
3850         if (info->attrs[NL80211_ATTR_USER_REG_HINT_TYPE])
3851                 user_reg_hint_type =
3852                   nla_get_u32(info->attrs[NL80211_ATTR_USER_REG_HINT_TYPE]);
3853         else
3854                 user_reg_hint_type = NL80211_USER_REG_HINT_USER;
3855
3856         switch (user_reg_hint_type) {
3857         case NL80211_USER_REG_HINT_USER:
3858         case NL80211_USER_REG_HINT_CELL_BASE:
3859                 break;
3860         default:
3861                 return -EINVAL;
3862         }
3863
3864         r = regulatory_hint_user(data, user_reg_hint_type);
3865
3866         return r;
3867 }
3868
3869 static int nl80211_get_mesh_config(struct sk_buff *skb,
3870                                    struct genl_info *info)
3871 {
3872         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3873         struct net_device *dev = info->user_ptr[1];
3874         struct wireless_dev *wdev = dev->ieee80211_ptr;
3875         struct mesh_config cur_params;
3876         int err = 0;
3877         void *hdr;
3878         struct nlattr *pinfoattr;
3879         struct sk_buff *msg;
3880
3881         if (wdev->iftype != NL80211_IFTYPE_MESH_POINT)
3882                 return -EOPNOTSUPP;
3883
3884         if (!rdev->ops->get_mesh_config)
3885                 return -EOPNOTSUPP;
3886
3887         wdev_lock(wdev);
3888         /* If not connected, get default parameters */
3889         if (!wdev->mesh_id_len)
3890                 memcpy(&cur_params, &default_mesh_config, sizeof(cur_params));
3891         else
3892                 err = rdev_get_mesh_config(rdev, dev, &cur_params);
3893         wdev_unlock(wdev);
3894
3895         if (err)
3896                 return err;
3897
3898         /* Draw up a netlink message to send back */
3899         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3900         if (!msg)
3901                 return -ENOMEM;
3902         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
3903                              NL80211_CMD_GET_MESH_CONFIG);
3904         if (!hdr)
3905                 goto out;
3906         pinfoattr = nla_nest_start(msg, NL80211_ATTR_MESH_CONFIG);
3907         if (!pinfoattr)
3908                 goto nla_put_failure;
3909         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
3910             nla_put_u16(msg, NL80211_MESHCONF_RETRY_TIMEOUT,
3911                         cur_params.dot11MeshRetryTimeout) ||
3912             nla_put_u16(msg, NL80211_MESHCONF_CONFIRM_TIMEOUT,
3913                         cur_params.dot11MeshConfirmTimeout) ||
3914             nla_put_u16(msg, NL80211_MESHCONF_HOLDING_TIMEOUT,
3915                         cur_params.dot11MeshHoldingTimeout) ||
3916             nla_put_u16(msg, NL80211_MESHCONF_MAX_PEER_LINKS,
3917                         cur_params.dot11MeshMaxPeerLinks) ||
3918             nla_put_u8(msg, NL80211_MESHCONF_MAX_RETRIES,
3919                        cur_params.dot11MeshMaxRetries) ||
3920             nla_put_u8(msg, NL80211_MESHCONF_TTL,
3921                        cur_params.dot11MeshTTL) ||
3922             nla_put_u8(msg, NL80211_MESHCONF_ELEMENT_TTL,
3923                        cur_params.element_ttl) ||
3924             nla_put_u8(msg, NL80211_MESHCONF_AUTO_OPEN_PLINKS,
3925                        cur_params.auto_open_plinks) ||
3926             nla_put_u32(msg, NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR,
3927                         cur_params.dot11MeshNbrOffsetMaxNeighbor) ||
3928             nla_put_u8(msg, NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES,
3929                        cur_params.dot11MeshHWMPmaxPREQretries) ||
3930             nla_put_u32(msg, NL80211_MESHCONF_PATH_REFRESH_TIME,
3931                         cur_params.path_refresh_time) ||
3932             nla_put_u16(msg, NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT,
3933                         cur_params.min_discovery_timeout) ||
3934             nla_put_u32(msg, NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT,
3935                         cur_params.dot11MeshHWMPactivePathTimeout) ||
3936             nla_put_u16(msg, NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL,
3937                         cur_params.dot11MeshHWMPpreqMinInterval) ||
3938             nla_put_u16(msg, NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL,
3939                         cur_params.dot11MeshHWMPperrMinInterval) ||
3940             nla_put_u16(msg, NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
3941                         cur_params.dot11MeshHWMPnetDiameterTraversalTime) ||
3942             nla_put_u8(msg, NL80211_MESHCONF_HWMP_ROOTMODE,
3943                        cur_params.dot11MeshHWMPRootMode) ||
3944             nla_put_u16(msg, NL80211_MESHCONF_HWMP_RANN_INTERVAL,
3945                         cur_params.dot11MeshHWMPRannInterval) ||
3946             nla_put_u8(msg, NL80211_MESHCONF_GATE_ANNOUNCEMENTS,
3947                        cur_params.dot11MeshGateAnnouncementProtocol) ||
3948             nla_put_u8(msg, NL80211_MESHCONF_FORWARDING,
3949                        cur_params.dot11MeshForwarding) ||
3950             nla_put_u32(msg, NL80211_MESHCONF_RSSI_THRESHOLD,
3951                         cur_params.rssi_threshold) ||
3952             nla_put_u32(msg, NL80211_MESHCONF_HT_OPMODE,
3953                         cur_params.ht_opmode) ||
3954             nla_put_u32(msg, NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT,
3955                         cur_params.dot11MeshHWMPactivePathToRootTimeout) ||
3956             nla_put_u16(msg, NL80211_MESHCONF_HWMP_ROOT_INTERVAL,
3957                         cur_params.dot11MeshHWMProotInterval) ||
3958             nla_put_u16(msg, NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL,
3959                         cur_params.dot11MeshHWMPconfirmationInterval) ||
3960             nla_put_u32(msg, NL80211_MESHCONF_POWER_MODE,
3961                         cur_params.power_mode) ||
3962             nla_put_u16(msg, NL80211_MESHCONF_AWAKE_WINDOW,
3963                         cur_params.dot11MeshAwakeWindowDuration))
3964                 goto nla_put_failure;
3965         nla_nest_end(msg, pinfoattr);
3966         genlmsg_end(msg, hdr);
3967         return genlmsg_reply(msg, info);
3968
3969  nla_put_failure:
3970         genlmsg_cancel(msg, hdr);
3971  out:
3972         nlmsg_free(msg);
3973         return -ENOBUFS;
3974 }
3975
3976 static const struct nla_policy nl80211_meshconf_params_policy[NL80211_MESHCONF_ATTR_MAX+1] = {
3977         [NL80211_MESHCONF_RETRY_TIMEOUT] = { .type = NLA_U16 },
3978         [NL80211_MESHCONF_CONFIRM_TIMEOUT] = { .type = NLA_U16 },
3979         [NL80211_MESHCONF_HOLDING_TIMEOUT] = { .type = NLA_U16 },
3980         [NL80211_MESHCONF_MAX_PEER_LINKS] = { .type = NLA_U16 },
3981         [NL80211_MESHCONF_MAX_RETRIES] = { .type = NLA_U8 },
3982         [NL80211_MESHCONF_TTL] = { .type = NLA_U8 },
3983         [NL80211_MESHCONF_ELEMENT_TTL] = { .type = NLA_U8 },
3984         [NL80211_MESHCONF_AUTO_OPEN_PLINKS] = { .type = NLA_U8 },
3985         [NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR] = { .type = NLA_U32 },
3986         [NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES] = { .type = NLA_U8 },
3987         [NL80211_MESHCONF_PATH_REFRESH_TIME] = { .type = NLA_U32 },
3988         [NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT] = { .type = NLA_U16 },
3989         [NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT] = { .type = NLA_U32 },
3990         [NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL] = { .type = NLA_U16 },
3991         [NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL] = { .type = NLA_U16 },
3992         [NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME] = { .type = NLA_U16 },
3993         [NL80211_MESHCONF_HWMP_ROOTMODE] = { .type = NLA_U8 },
3994         [NL80211_MESHCONF_HWMP_RANN_INTERVAL] = { .type = NLA_U16 },
3995         [NL80211_MESHCONF_GATE_ANNOUNCEMENTS] = { .type = NLA_U8 },
3996         [NL80211_MESHCONF_FORWARDING] = { .type = NLA_U8 },
3997         [NL80211_MESHCONF_RSSI_THRESHOLD] = { .type = NLA_U32 },
3998         [NL80211_MESHCONF_HT_OPMODE] = { .type = NLA_U16 },
3999         [NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT] = { .type = NLA_U32 },
4000         [NL80211_MESHCONF_HWMP_ROOT_INTERVAL] = { .type = NLA_U16 },
4001         [NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL] = { .type = NLA_U16 },
4002         [NL80211_MESHCONF_POWER_MODE] = { .type = NLA_U32 },
4003         [NL80211_MESHCONF_AWAKE_WINDOW] = { .type = NLA_U16 },
4004 };
4005
4006 static const struct nla_policy
4007         nl80211_mesh_setup_params_policy[NL80211_MESH_SETUP_ATTR_MAX+1] = {
4008         [NL80211_MESH_SETUP_ENABLE_VENDOR_SYNC] = { .type = NLA_U8 },
4009         [NL80211_MESH_SETUP_ENABLE_VENDOR_PATH_SEL] = { .type = NLA_U8 },
4010         [NL80211_MESH_SETUP_ENABLE_VENDOR_METRIC] = { .type = NLA_U8 },
4011         [NL80211_MESH_SETUP_USERSPACE_AUTH] = { .type = NLA_FLAG },
4012         [NL80211_MESH_SETUP_IE] = { .type = NLA_BINARY,
4013                                     .len = IEEE80211_MAX_DATA_LEN },
4014         [NL80211_MESH_SETUP_USERSPACE_AMPE] = { .type = NLA_FLAG },
4015 };
4016
4017 static int nl80211_parse_mesh_config(struct genl_info *info,
4018                                      struct mesh_config *cfg,
4019                                      u32 *mask_out)
4020 {
4021         struct nlattr *tb[NL80211_MESHCONF_ATTR_MAX + 1];
4022         u32 mask = 0;
4023
4024 #define FILL_IN_MESH_PARAM_IF_SET(tb, cfg, param, min, max, mask, attr, fn) \
4025 do {                                                                        \
4026         if (tb[attr]) {                                                     \
4027                 if (fn(tb[attr]) < min || fn(tb[attr]) > max)               \
4028                         return -EINVAL;                                     \
4029                 cfg->param = fn(tb[attr]);                                  \
4030                 mask |= (1 << (attr - 1));                                  \
4031         }                                                                   \
4032 } while (0)
4033
4034
4035         if (!info->attrs[NL80211_ATTR_MESH_CONFIG])
4036                 return -EINVAL;
4037         if (nla_parse_nested(tb, NL80211_MESHCONF_ATTR_MAX,
4038                              info->attrs[NL80211_ATTR_MESH_CONFIG],
4039                              nl80211_meshconf_params_policy))
4040                 return -EINVAL;
4041
4042         /* This makes sure that there aren't more than 32 mesh config
4043          * parameters (otherwise our bitfield scheme would not work.) */
4044         BUILD_BUG_ON(NL80211_MESHCONF_ATTR_MAX > 32);
4045
4046         /* Fill in the params struct */
4047         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshRetryTimeout, 1, 255,
4048                                   mask, NL80211_MESHCONF_RETRY_TIMEOUT,
4049                                   nla_get_u16);
4050         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshConfirmTimeout, 1, 255,
4051                                   mask, NL80211_MESHCONF_CONFIRM_TIMEOUT,
4052                                   nla_get_u16);
4053         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHoldingTimeout, 1, 255,
4054                                   mask, NL80211_MESHCONF_HOLDING_TIMEOUT,
4055                                   nla_get_u16);
4056         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshMaxPeerLinks, 0, 255,
4057                                   mask, NL80211_MESHCONF_MAX_PEER_LINKS,
4058                                   nla_get_u16);
4059         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshMaxRetries, 0, 16,
4060                                   mask, NL80211_MESHCONF_MAX_RETRIES,
4061                                   nla_get_u8);
4062         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshTTL, 1, 255,
4063                                   mask, NL80211_MESHCONF_TTL, nla_get_u8);
4064         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, element_ttl, 1, 255,
4065                                   mask, NL80211_MESHCONF_ELEMENT_TTL,
4066                                   nla_get_u8);
4067         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, auto_open_plinks, 0, 1,
4068                                   mask, NL80211_MESHCONF_AUTO_OPEN_PLINKS,
4069                                   nla_get_u8);
4070         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshNbrOffsetMaxNeighbor,
4071                                   1, 255, mask,
4072                                   NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR,
4073                                   nla_get_u32);
4074         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPmaxPREQretries, 0, 255,
4075                                   mask, NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES,
4076                                   nla_get_u8);
4077         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, path_refresh_time, 1, 65535,
4078                                   mask, NL80211_MESHCONF_PATH_REFRESH_TIME,
4079                                   nla_get_u32);
4080         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, min_discovery_timeout, 1, 65535,
4081                                   mask, NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT,
4082                                   nla_get_u16);
4083         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPactivePathTimeout,
4084                                   1, 65535, mask,
4085                                   NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT,
4086                                   nla_get_u32);
4087         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPpreqMinInterval,
4088                                   1, 65535, mask,
4089                                   NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL,
4090                                   nla_get_u16);
4091         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPperrMinInterval,
4092                                   1, 65535, mask,
4093                                   NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL,
4094                                   nla_get_u16);
4095         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
4096                                   dot11MeshHWMPnetDiameterTraversalTime,
4097                                   1, 65535, mask,
4098                                   NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
4099                                   nla_get_u16);
4100         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPRootMode, 0, 4,
4101                                   mask, NL80211_MESHCONF_HWMP_ROOTMODE,
4102                                   nla_get_u8);
4103         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPRannInterval, 1, 65535,
4104                                   mask, NL80211_MESHCONF_HWMP_RANN_INTERVAL,
4105                                   nla_get_u16);
4106         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
4107                                   dot11MeshGateAnnouncementProtocol, 0, 1,
4108                                   mask, NL80211_MESHCONF_GATE_ANNOUNCEMENTS,
4109                                   nla_get_u8);
4110         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshForwarding, 0, 1,
4111                                   mask, NL80211_MESHCONF_FORWARDING,
4112                                   nla_get_u8);
4113         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, rssi_threshold, 1, 255,
4114                                   mask, NL80211_MESHCONF_RSSI_THRESHOLD,
4115                                   nla_get_u32);
4116         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, ht_opmode, 0, 16,
4117                                   mask, NL80211_MESHCONF_HT_OPMODE,
4118                                   nla_get_u16);
4119         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPactivePathToRootTimeout,
4120                                   1, 65535, mask,
4121                                   NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT,
4122                                   nla_get_u32);
4123         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMProotInterval, 1, 65535,
4124                                   mask, NL80211_MESHCONF_HWMP_ROOT_INTERVAL,
4125                                   nla_get_u16);
4126         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
4127                                   dot11MeshHWMPconfirmationInterval,
4128                                   1, 65535, mask,
4129                                   NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL,
4130                                   nla_get_u16);
4131         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, power_mode,
4132                                   NL80211_MESH_POWER_ACTIVE,
4133                                   NL80211_MESH_POWER_MAX,
4134                                   mask, NL80211_MESHCONF_POWER_MODE,
4135                                   nla_get_u32);
4136         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshAwakeWindowDuration,
4137                                   0, 65535, mask,
4138                                   NL80211_MESHCONF_AWAKE_WINDOW, nla_get_u16);
4139         if (mask_out)
4140                 *mask_out = mask;
4141
4142         return 0;
4143
4144 #undef FILL_IN_MESH_PARAM_IF_SET
4145 }
4146
4147 static int nl80211_parse_mesh_setup(struct genl_info *info,
4148                                      struct mesh_setup *setup)
4149 {
4150         struct nlattr *tb[NL80211_MESH_SETUP_ATTR_MAX + 1];
4151
4152         if (!info->attrs[NL80211_ATTR_MESH_SETUP])
4153                 return -EINVAL;
4154         if (nla_parse_nested(tb, NL80211_MESH_SETUP_ATTR_MAX,
4155                              info->attrs[NL80211_ATTR_MESH_SETUP],
4156                              nl80211_mesh_setup_params_policy))
4157                 return -EINVAL;
4158
4159         if (tb[NL80211_MESH_SETUP_ENABLE_VENDOR_SYNC])
4160                 setup->sync_method =
4161                 (nla_get_u8(tb[NL80211_MESH_SETUP_ENABLE_VENDOR_SYNC])) ?
4162                  IEEE80211_SYNC_METHOD_VENDOR :
4163                  IEEE80211_SYNC_METHOD_NEIGHBOR_OFFSET;
4164
4165         if (tb[NL80211_MESH_SETUP_ENABLE_VENDOR_PATH_SEL])
4166                 setup->path_sel_proto =
4167                 (nla_get_u8(tb[NL80211_MESH_SETUP_ENABLE_VENDOR_PATH_SEL])) ?
4168                  IEEE80211_PATH_PROTOCOL_VENDOR :
4169                  IEEE80211_PATH_PROTOCOL_HWMP;
4170
4171         if (tb[NL80211_MESH_SETUP_ENABLE_VENDOR_METRIC])
4172                 setup->path_metric =
4173                 (nla_get_u8(tb[NL80211_MESH_SETUP_ENABLE_VENDOR_METRIC])) ?
4174                  IEEE80211_PATH_METRIC_VENDOR :
4175                  IEEE80211_PATH_METRIC_AIRTIME;
4176
4177
4178         if (tb[NL80211_MESH_SETUP_IE]) {
4179                 struct nlattr *ieattr =
4180                         tb[NL80211_MESH_SETUP_IE];
4181                 if (!is_valid_ie_attr(ieattr))
4182                         return -EINVAL;
4183                 setup->ie = nla_data(ieattr);
4184                 setup->ie_len = nla_len(ieattr);
4185         }
4186         setup->is_authenticated = nla_get_flag(tb[NL80211_MESH_SETUP_USERSPACE_AUTH]);
4187         setup->is_secure = nla_get_flag(tb[NL80211_MESH_SETUP_USERSPACE_AMPE]);
4188
4189         return 0;
4190 }
4191
4192 static int nl80211_update_mesh_config(struct sk_buff *skb,
4193                                       struct genl_info *info)
4194 {
4195         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4196         struct net_device *dev = info->user_ptr[1];
4197         struct wireless_dev *wdev = dev->ieee80211_ptr;
4198         struct mesh_config cfg;
4199         u32 mask;
4200         int err;
4201
4202         if (wdev->iftype != NL80211_IFTYPE_MESH_POINT)
4203                 return -EOPNOTSUPP;
4204
4205         if (!rdev->ops->update_mesh_config)
4206                 return -EOPNOTSUPP;
4207
4208         err = nl80211_parse_mesh_config(info, &cfg, &mask);
4209         if (err)
4210                 return err;
4211
4212         wdev_lock(wdev);
4213         if (!wdev->mesh_id_len)
4214                 err = -ENOLINK;
4215
4216         if (!err)
4217                 err = rdev_update_mesh_config(rdev, dev, mask, &cfg);
4218
4219         wdev_unlock(wdev);
4220
4221         return err;
4222 }
4223
4224 static int nl80211_get_reg(struct sk_buff *skb, struct genl_info *info)
4225 {
4226         const struct ieee80211_regdomain *regdom;
4227         struct sk_buff *msg;
4228         void *hdr = NULL;
4229         struct nlattr *nl_reg_rules;
4230         unsigned int i;
4231         int err = -EINVAL;
4232
4233         mutex_lock(&cfg80211_mutex);
4234
4235         if (!cfg80211_regdomain)
4236                 goto out;
4237
4238         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4239         if (!msg) {
4240                 err = -ENOBUFS;
4241                 goto out;
4242         }
4243
4244         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
4245                              NL80211_CMD_GET_REG);
4246         if (!hdr)
4247                 goto put_failure;
4248
4249         if (reg_last_request_cell_base() &&
4250             nla_put_u32(msg, NL80211_ATTR_USER_REG_HINT_TYPE,
4251                         NL80211_USER_REG_HINT_CELL_BASE))
4252                 goto nla_put_failure;
4253
4254         rcu_read_lock();
4255         regdom = rcu_dereference(cfg80211_regdomain);
4256
4257         if (nla_put_string(msg, NL80211_ATTR_REG_ALPHA2, regdom->alpha2) ||
4258             (regdom->dfs_region &&
4259              nla_put_u8(msg, NL80211_ATTR_DFS_REGION, regdom->dfs_region)))
4260                 goto nla_put_failure_rcu;
4261
4262         nl_reg_rules = nla_nest_start(msg, NL80211_ATTR_REG_RULES);
4263         if (!nl_reg_rules)
4264                 goto nla_put_failure_rcu;
4265
4266         for (i = 0; i < regdom->n_reg_rules; i++) {
4267                 struct nlattr *nl_reg_rule;
4268                 const struct ieee80211_reg_rule *reg_rule;
4269                 const struct ieee80211_freq_range *freq_range;
4270                 const struct ieee80211_power_rule *power_rule;
4271
4272                 reg_rule = &regdom->reg_rules[i];
4273                 freq_range = &reg_rule->freq_range;
4274                 power_rule = &reg_rule->power_rule;
4275
4276                 nl_reg_rule = nla_nest_start(msg, i);
4277                 if (!nl_reg_rule)
4278                         goto nla_put_failure_rcu;
4279
4280                 if (nla_put_u32(msg, NL80211_ATTR_REG_RULE_FLAGS,
4281                                 reg_rule->flags) ||
4282                     nla_put_u32(msg, NL80211_ATTR_FREQ_RANGE_START,
4283                                 freq_range->start_freq_khz) ||
4284                     nla_put_u32(msg, NL80211_ATTR_FREQ_RANGE_END,
4285                                 freq_range->end_freq_khz) ||
4286                     nla_put_u32(msg, NL80211_ATTR_FREQ_RANGE_MAX_BW,
4287                                 freq_range->max_bandwidth_khz) ||
4288                     nla_put_u32(msg, NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN,
4289                                 power_rule->max_antenna_gain) ||
4290                     nla_put_u32(msg, NL80211_ATTR_POWER_RULE_MAX_EIRP,
4291                                 power_rule->max_eirp))
4292                         goto nla_put_failure_rcu;
4293
4294                 nla_nest_end(msg, nl_reg_rule);
4295         }
4296         rcu_read_unlock();
4297
4298         nla_nest_end(msg, nl_reg_rules);
4299
4300         genlmsg_end(msg, hdr);
4301         err = genlmsg_reply(msg, info);
4302         goto out;
4303
4304 nla_put_failure_rcu:
4305         rcu_read_unlock();
4306 nla_put_failure:
4307         genlmsg_cancel(msg, hdr);
4308 put_failure:
4309         nlmsg_free(msg);
4310         err = -EMSGSIZE;
4311 out:
4312         mutex_unlock(&cfg80211_mutex);
4313         return err;
4314 }
4315
4316 static int nl80211_set_reg(struct sk_buff *skb, struct genl_info *info)
4317 {
4318         struct nlattr *tb[NL80211_REG_RULE_ATTR_MAX + 1];
4319         struct nlattr *nl_reg_rule;
4320         char *alpha2 = NULL;
4321         int rem_reg_rules = 0, r = 0;
4322         u32 num_rules = 0, rule_idx = 0, size_of_regd;
4323         u8 dfs_region = 0;
4324         struct ieee80211_regdomain *rd = NULL;
4325
4326         if (!info->attrs[NL80211_ATTR_REG_ALPHA2])
4327                 return -EINVAL;
4328
4329         if (!info->attrs[NL80211_ATTR_REG_RULES])
4330                 return -EINVAL;
4331
4332         alpha2 = nla_data(info->attrs[NL80211_ATTR_REG_ALPHA2]);
4333
4334         if (info->attrs[NL80211_ATTR_DFS_REGION])
4335                 dfs_region = nla_get_u8(info->attrs[NL80211_ATTR_DFS_REGION]);
4336
4337         nla_for_each_nested(nl_reg_rule, info->attrs[NL80211_ATTR_REG_RULES],
4338                             rem_reg_rules) {
4339                 num_rules++;
4340                 if (num_rules > NL80211_MAX_SUPP_REG_RULES)
4341                         return -EINVAL;
4342         }
4343
4344         size_of_regd = sizeof(struct ieee80211_regdomain) +
4345                        num_rules * sizeof(struct ieee80211_reg_rule);
4346
4347         rd = kzalloc(size_of_regd, GFP_KERNEL);
4348         if (!rd)
4349                 return -ENOMEM;
4350
4351         rd->n_reg_rules = num_rules;
4352         rd->alpha2[0] = alpha2[0];
4353         rd->alpha2[1] = alpha2[1];
4354
4355         /*
4356          * Disable DFS master mode if the DFS region was
4357          * not supported or known on this kernel.
4358          */
4359         if (reg_supported_dfs_region(dfs_region))
4360                 rd->dfs_region = dfs_region;
4361
4362         nla_for_each_nested(nl_reg_rule, info->attrs[NL80211_ATTR_REG_RULES],
4363                             rem_reg_rules) {
4364                 nla_parse(tb, NL80211_REG_RULE_ATTR_MAX,
4365                           nla_data(nl_reg_rule), nla_len(nl_reg_rule),
4366                           reg_rule_policy);
4367                 r = parse_reg_rule(tb, &rd->reg_rules[rule_idx]);
4368                 if (r)
4369                         goto bad_reg;
4370
4371                 rule_idx++;
4372
4373                 if (rule_idx > NL80211_MAX_SUPP_REG_RULES) {
4374                         r = -EINVAL;
4375                         goto bad_reg;
4376                 }
4377         }
4378
4379         mutex_lock(&cfg80211_mutex);
4380
4381         r = set_regdom(rd);
4382         /* set_regdom took ownership */
4383         rd = NULL;
4384         mutex_unlock(&cfg80211_mutex);
4385
4386  bad_reg:
4387         kfree(rd);
4388         return r;
4389 }
4390
4391 static int validate_scan_freqs(struct nlattr *freqs)
4392 {
4393         struct nlattr *attr1, *attr2;
4394         int n_channels = 0, tmp1, tmp2;
4395
4396         nla_for_each_nested(attr1, freqs, tmp1) {
4397                 n_channels++;
4398                 /*
4399                  * Some hardware has a limited channel list for
4400                  * scanning, and it is pretty much nonsensical
4401                  * to scan for a channel twice, so disallow that
4402                  * and don't require drivers to check that the
4403                  * channel list they get isn't longer than what
4404                  * they can scan, as long as they can scan all
4405                  * the channels they registered at once.
4406                  */
4407                 nla_for_each_nested(attr2, freqs, tmp2)
4408                         if (attr1 != attr2 &&
4409                             nla_get_u32(attr1) == nla_get_u32(attr2))
4410                                 return 0;
4411         }
4412
4413         return n_channels;
4414 }
4415
4416 static int nl80211_trigger_scan(struct sk_buff *skb, struct genl_info *info)
4417 {
4418         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4419         struct wireless_dev *wdev = info->user_ptr[1];
4420         struct cfg80211_scan_request *request;
4421         struct nlattr *attr;
4422         struct wiphy *wiphy;
4423         int err, tmp, n_ssids = 0, n_channels, i;
4424         size_t ie_len;
4425
4426         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
4427                 return -EINVAL;
4428
4429         wiphy = &rdev->wiphy;
4430
4431         if (!rdev->ops->scan)
4432                 return -EOPNOTSUPP;
4433
4434         if (rdev->scan_req)
4435                 return -EBUSY;
4436
4437         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
4438                 n_channels = validate_scan_freqs(
4439                                 info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]);
4440                 if (!n_channels)
4441                         return -EINVAL;
4442         } else {
4443                 enum ieee80211_band band;
4444                 n_channels = 0;
4445
4446                 for (band = 0; band < IEEE80211_NUM_BANDS; band++)
4447                         if (wiphy->bands[band])
4448                                 n_channels += wiphy->bands[band]->n_channels;
4449         }
4450
4451         if (info->attrs[NL80211_ATTR_SCAN_SSIDS])
4452                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS], tmp)
4453                         n_ssids++;
4454
4455         if (n_ssids > wiphy->max_scan_ssids)
4456                 return -EINVAL;
4457
4458         if (info->attrs[NL80211_ATTR_IE])
4459                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
4460         else
4461                 ie_len = 0;
4462
4463         if (ie_len > wiphy->max_scan_ie_len)
4464                 return -EINVAL;
4465
4466         request = kzalloc(sizeof(*request)
4467                         + sizeof(*request->ssids) * n_ssids
4468                         + sizeof(*request->channels) * n_channels
4469                         + ie_len, GFP_KERNEL);
4470         if (!request)
4471                 return -ENOMEM;
4472
4473         if (n_ssids)
4474                 request->ssids = (void *)&request->channels[n_channels];
4475         request->n_ssids = n_ssids;
4476         if (ie_len) {
4477                 if (request->ssids)
4478                         request->ie = (void *)(request->ssids + n_ssids);
4479                 else
4480                         request->ie = (void *)(request->channels + n_channels);
4481         }
4482
4483         i = 0;
4484         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
4485                 /* user specified, bail out if channel not found */
4486                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_FREQUENCIES], tmp) {
4487                         struct ieee80211_channel *chan;
4488
4489                         chan = ieee80211_get_channel(wiphy, nla_get_u32(attr));
4490
4491                         if (!chan) {
4492                                 err = -EINVAL;
4493                                 goto out_free;
4494                         }
4495
4496                         /* ignore disabled channels */
4497                         if (chan->flags & IEEE80211_CHAN_DISABLED)
4498                                 continue;
4499
4500                         request->channels[i] = chan;
4501                         i++;
4502                 }
4503         } else {
4504                 enum ieee80211_band band;
4505
4506                 /* all channels */
4507                 for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
4508                         int j;
4509                         if (!wiphy->bands[band])
4510                                 continue;
4511                         for (j = 0; j < wiphy->bands[band]->n_channels; j++) {
4512                                 struct ieee80211_channel *chan;
4513
4514                                 chan = &wiphy->bands[band]->channels[j];
4515
4516                                 if (chan->flags & IEEE80211_CHAN_DISABLED)
4517                                         continue;
4518
4519                                 request->channels[i] = chan;
4520                                 i++;
4521                         }
4522                 }
4523         }
4524
4525         if (!i) {
4526                 err = -EINVAL;
4527                 goto out_free;
4528         }
4529
4530         request->n_channels = i;
4531
4532         i = 0;
4533         if (info->attrs[NL80211_ATTR_SCAN_SSIDS]) {
4534                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS], tmp) {
4535                         if (nla_len(attr) > IEEE80211_MAX_SSID_LEN) {
4536                                 err = -EINVAL;
4537                                 goto out_free;
4538                         }
4539                         request->ssids[i].ssid_len = nla_len(attr);
4540                         memcpy(request->ssids[i].ssid, nla_data(attr), nla_len(attr));
4541                         i++;
4542                 }
4543         }
4544
4545         if (info->attrs[NL80211_ATTR_IE]) {
4546                 request->ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
4547                 memcpy((void *)request->ie,
4548                        nla_data(info->attrs[NL80211_ATTR_IE]),
4549                        request->ie_len);
4550         }
4551
4552         for (i = 0; i < IEEE80211_NUM_BANDS; i++)
4553                 if (wiphy->bands[i])
4554                         request->rates[i] =
4555                                 (1 << wiphy->bands[i]->n_bitrates) - 1;
4556
4557         if (info->attrs[NL80211_ATTR_SCAN_SUPP_RATES]) {
4558                 nla_for_each_nested(attr,
4559                                     info->attrs[NL80211_ATTR_SCAN_SUPP_RATES],
4560                                     tmp) {
4561                         enum ieee80211_band band = nla_type(attr);
4562
4563                         if (band < 0 || band >= IEEE80211_NUM_BANDS) {
4564                                 err = -EINVAL;
4565                                 goto out_free;
4566                         }
4567                         err = ieee80211_get_ratemask(wiphy->bands[band],
4568                                                      nla_data(attr),
4569                                                      nla_len(attr),
4570                                                      &request->rates[band]);
4571                         if (err)
4572                                 goto out_free;
4573                 }
4574         }
4575
4576         if (info->attrs[NL80211_ATTR_SCAN_FLAGS]) {
4577                 request->flags = nla_get_u32(
4578                         info->attrs[NL80211_ATTR_SCAN_FLAGS]);
4579                 if (((request->flags & NL80211_SCAN_FLAG_LOW_PRIORITY) &&
4580                      !(wiphy->features & NL80211_FEATURE_LOW_PRIORITY_SCAN)) ||
4581                     ((request->flags & NL80211_SCAN_FLAG_FLUSH) &&
4582                      !(wiphy->features & NL80211_FEATURE_SCAN_FLUSH))) {
4583                         err = -EOPNOTSUPP;
4584                         goto out_free;
4585                 }
4586         }
4587
4588         request->no_cck =
4589                 nla_get_flag(info->attrs[NL80211_ATTR_TX_NO_CCK_RATE]);
4590
4591         request->wdev = wdev;
4592         request->wiphy = &rdev->wiphy;
4593         request->scan_start = jiffies;
4594
4595         rdev->scan_req = request;
4596         err = rdev_scan(rdev, request);
4597
4598         if (!err) {
4599                 nl80211_send_scan_start(rdev, wdev);
4600                 if (wdev->netdev)
4601                         dev_hold(wdev->netdev);
4602         } else {
4603  out_free:
4604                 rdev->scan_req = NULL;
4605                 kfree(request);
4606         }
4607
4608         return err;
4609 }
4610
4611 static int nl80211_start_sched_scan(struct sk_buff *skb,
4612                                     struct genl_info *info)
4613 {
4614         struct cfg80211_sched_scan_request *request;
4615         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4616         struct net_device *dev = info->user_ptr[1];
4617         struct nlattr *attr;
4618         struct wiphy *wiphy;
4619         int err, tmp, n_ssids = 0, n_match_sets = 0, n_channels, i;
4620         u32 interval;
4621         enum ieee80211_band band;
4622         size_t ie_len;
4623         struct nlattr *tb[NL80211_SCHED_SCAN_MATCH_ATTR_MAX + 1];
4624
4625         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN) ||
4626             !rdev->ops->sched_scan_start)
4627                 return -EOPNOTSUPP;
4628
4629         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
4630                 return -EINVAL;
4631
4632         if (!info->attrs[NL80211_ATTR_SCHED_SCAN_INTERVAL])
4633                 return -EINVAL;
4634
4635         interval = nla_get_u32(info->attrs[NL80211_ATTR_SCHED_SCAN_INTERVAL]);
4636         if (interval == 0)
4637                 return -EINVAL;
4638
4639         wiphy = &rdev->wiphy;
4640
4641         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
4642                 n_channels = validate_scan_freqs(
4643                                 info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]);
4644                 if (!n_channels)
4645                         return -EINVAL;
4646         } else {
4647                 n_channels = 0;
4648
4649                 for (band = 0; band < IEEE80211_NUM_BANDS; band++)
4650                         if (wiphy->bands[band])
4651                                 n_channels += wiphy->bands[band]->n_channels;
4652         }
4653
4654         if (info->attrs[NL80211_ATTR_SCAN_SSIDS])
4655                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS],
4656                                     tmp)
4657                         n_ssids++;
4658
4659         if (n_ssids > wiphy->max_sched_scan_ssids)
4660                 return -EINVAL;
4661
4662         if (info->attrs[NL80211_ATTR_SCHED_SCAN_MATCH])
4663                 nla_for_each_nested(attr,
4664                                     info->attrs[NL80211_ATTR_SCHED_SCAN_MATCH],
4665                                     tmp)
4666                         n_match_sets++;
4667
4668         if (n_match_sets > wiphy->max_match_sets)
4669                 return -EINVAL;
4670
4671         if (info->attrs[NL80211_ATTR_IE])
4672                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
4673         else
4674                 ie_len = 0;
4675
4676         if (ie_len > wiphy->max_sched_scan_ie_len)
4677                 return -EINVAL;
4678
4679         mutex_lock(&rdev->sched_scan_mtx);
4680
4681         if (rdev->sched_scan_req) {
4682                 err = -EINPROGRESS;
4683                 goto out;
4684         }
4685
4686         request = kzalloc(sizeof(*request)
4687                         + sizeof(*request->ssids) * n_ssids
4688                         + sizeof(*request->match_sets) * n_match_sets
4689                         + sizeof(*request->channels) * n_channels
4690                         + ie_len, GFP_KERNEL);
4691         if (!request) {
4692                 err = -ENOMEM;
4693                 goto out;
4694         }
4695
4696         if (n_ssids)
4697                 request->ssids = (void *)&request->channels[n_channels];
4698         request->n_ssids = n_ssids;
4699         if (ie_len) {
4700                 if (request->ssids)
4701                         request->ie = (void *)(request->ssids + n_ssids);
4702                 else
4703                         request->ie = (void *)(request->channels + n_channels);
4704         }
4705
4706         if (n_match_sets) {
4707                 if (request->ie)
4708                         request->match_sets = (void *)(request->ie + ie_len);
4709                 else if (request->ssids)
4710                         request->match_sets =
4711                                 (void *)(request->ssids + n_ssids);
4712                 else
4713                         request->match_sets =
4714                                 (void *)(request->channels + n_channels);
4715         }
4716         request->n_match_sets = n_match_sets;
4717
4718         i = 0;
4719         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
4720                 /* user specified, bail out if channel not found */
4721                 nla_for_each_nested(attr,
4722                                     info->attrs[NL80211_ATTR_SCAN_FREQUENCIES],
4723                                     tmp) {
4724                         struct ieee80211_channel *chan;
4725
4726                         chan = ieee80211_get_channel(wiphy, nla_get_u32(attr));
4727
4728                         if (!chan) {
4729                                 err = -EINVAL;
4730                                 goto out_free;
4731                         }
4732
4733                         /* ignore disabled channels */
4734                         if (chan->flags & IEEE80211_CHAN_DISABLED)
4735                                 continue;
4736
4737                         request->channels[i] = chan;
4738                         i++;
4739                 }
4740         } else {
4741                 /* all channels */
4742                 for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
4743                         int j;
4744                         if (!wiphy->bands[band])
4745                                 continue;
4746                         for (j = 0; j < wiphy->bands[band]->n_channels; j++) {
4747                                 struct ieee80211_channel *chan;
4748
4749                                 chan = &wiphy->bands[band]->channels[j];
4750
4751                                 if (chan->flags & IEEE80211_CHAN_DISABLED)
4752                                         continue;
4753
4754                                 request->channels[i] = chan;
4755                                 i++;
4756                         }
4757                 }
4758         }
4759
4760         if (!i) {
4761                 err = -EINVAL;
4762                 goto out_free;
4763         }
4764
4765         request->n_channels = i;
4766
4767         i = 0;
4768         if (info->attrs[NL80211_ATTR_SCAN_SSIDS]) {
4769                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS],
4770                                     tmp) {
4771                         if (nla_len(attr) > IEEE80211_MAX_SSID_LEN) {
4772                                 err = -EINVAL;
4773                                 goto out_free;
4774                         }
4775                         request->ssids[i].ssid_len = nla_len(attr);
4776                         memcpy(request->ssids[i].ssid, nla_data(attr),
4777                                nla_len(attr));
4778                         i++;
4779                 }
4780         }
4781
4782         i = 0;
4783         if (info->attrs[NL80211_ATTR_SCHED_SCAN_MATCH]) {
4784                 nla_for_each_nested(attr,
4785                                     info->attrs[NL80211_ATTR_SCHED_SCAN_MATCH],
4786                                     tmp) {
4787                         struct nlattr *ssid, *rssi;
4788
4789                         nla_parse(tb, NL80211_SCHED_SCAN_MATCH_ATTR_MAX,
4790                                   nla_data(attr), nla_len(attr),
4791                                   nl80211_match_policy);
4792                         ssid = tb[NL80211_SCHED_SCAN_MATCH_ATTR_SSID];
4793                         if (ssid) {
4794                                 if (nla_len(ssid) > IEEE80211_MAX_SSID_LEN) {
4795                                         err = -EINVAL;
4796                                         goto out_free;
4797                                 }
4798                                 memcpy(request->match_sets[i].ssid.ssid,
4799                                        nla_data(ssid), nla_len(ssid));
4800                                 request->match_sets[i].ssid.ssid_len =
4801                                         nla_len(ssid);
4802                         }
4803                         rssi = tb[NL80211_SCHED_SCAN_MATCH_ATTR_RSSI];
4804                         if (rssi)
4805                                 request->rssi_thold = nla_get_u32(rssi);
4806                         else
4807                                 request->rssi_thold =
4808                                                    NL80211_SCAN_RSSI_THOLD_OFF;
4809                         i++;
4810                 }
4811         }
4812
4813         if (info->attrs[NL80211_ATTR_IE]) {
4814                 request->ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
4815                 memcpy((void *)request->ie,
4816                        nla_data(info->attrs[NL80211_ATTR_IE]),
4817                        request->ie_len);
4818         }
4819
4820         if (info->attrs[NL80211_ATTR_SCAN_FLAGS]) {
4821                 request->flags = nla_get_u32(
4822                         info->attrs[NL80211_ATTR_SCAN_FLAGS]);
4823                 if (((request->flags & NL80211_SCAN_FLAG_LOW_PRIORITY) &&
4824                      !(wiphy->features & NL80211_FEATURE_LOW_PRIORITY_SCAN)) ||
4825                     ((request->flags & NL80211_SCAN_FLAG_FLUSH) &&
4826                      !(wiphy->features & NL80211_FEATURE_SCAN_FLUSH))) {
4827                         err = -EOPNOTSUPP;
4828                         goto out_free;
4829                 }
4830         }
4831
4832         request->dev = dev;
4833         request->wiphy = &rdev->wiphy;
4834         request->interval = interval;
4835         request->scan_start = jiffies;
4836
4837         err = rdev_sched_scan_start(rdev, dev, request);
4838         if (!err) {
4839                 rdev->sched_scan_req = request;
4840                 nl80211_send_sched_scan(rdev, dev,
4841                                         NL80211_CMD_START_SCHED_SCAN);
4842                 goto out;
4843         }
4844
4845 out_free:
4846         kfree(request);
4847 out:
4848         mutex_unlock(&rdev->sched_scan_mtx);
4849         return err;
4850 }
4851
4852 static int nl80211_stop_sched_scan(struct sk_buff *skb,
4853                                    struct genl_info *info)
4854 {
4855         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4856         int err;
4857
4858         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN) ||
4859             !rdev->ops->sched_scan_stop)
4860                 return -EOPNOTSUPP;
4861
4862         mutex_lock(&rdev->sched_scan_mtx);
4863         err = __cfg80211_stop_sched_scan(rdev, false);
4864         mutex_unlock(&rdev->sched_scan_mtx);
4865
4866         return err;
4867 }
4868
4869 static int nl80211_send_bss(struct sk_buff *msg, struct netlink_callback *cb,
4870                             u32 seq, int flags,
4871                             struct cfg80211_registered_device *rdev,
4872                             struct wireless_dev *wdev,
4873                             struct cfg80211_internal_bss *intbss)
4874 {
4875         struct cfg80211_bss *res = &intbss->pub;
4876         const struct cfg80211_bss_ies *ies;
4877         void *hdr;
4878         struct nlattr *bss;
4879
4880         ASSERT_WDEV_LOCK(wdev);
4881
4882         hdr = nl80211hdr_put(msg, NETLINK_CB(cb->skb).portid, seq, flags,
4883                              NL80211_CMD_NEW_SCAN_RESULTS);
4884         if (!hdr)
4885                 return -1;
4886
4887         genl_dump_check_consistent(cb, hdr, &nl80211_fam);
4888
4889         if (nla_put_u32(msg, NL80211_ATTR_GENERATION, rdev->bss_generation) ||
4890             nla_put_u32(msg, NL80211_ATTR_IFINDEX, wdev->netdev->ifindex))
4891                 goto nla_put_failure;
4892
4893         bss = nla_nest_start(msg, NL80211_ATTR_BSS);
4894         if (!bss)
4895                 goto nla_put_failure;
4896         if ((!is_zero_ether_addr(res->bssid) &&
4897              nla_put(msg, NL80211_BSS_BSSID, ETH_ALEN, res->bssid)))
4898                 goto nla_put_failure;
4899
4900         rcu_read_lock();
4901         ies = rcu_dereference(res->ies);
4902         if (ies && ies->len && nla_put(msg, NL80211_BSS_INFORMATION_ELEMENTS,
4903                                        ies->len, ies->data)) {
4904                 rcu_read_unlock();
4905                 goto nla_put_failure;
4906         }
4907         ies = rcu_dereference(res->beacon_ies);
4908         if (ies && ies->len && nla_put(msg, NL80211_BSS_BEACON_IES,
4909                                        ies->len, ies->data)) {
4910                 rcu_read_unlock();
4911                 goto nla_put_failure;
4912         }
4913         rcu_read_unlock();
4914
4915         if (res->tsf &&
4916             nla_put_u64(msg, NL80211_BSS_TSF, res->tsf))
4917                 goto nla_put_failure;
4918         if (res->beacon_interval &&
4919             nla_put_u16(msg, NL80211_BSS_BEACON_INTERVAL, res->beacon_interval))
4920                 goto nla_put_failure;
4921         if (nla_put_u16(msg, NL80211_BSS_CAPABILITY, res->capability) ||
4922             nla_put_u32(msg, NL80211_BSS_FREQUENCY, res->channel->center_freq) ||
4923             nla_put_u32(msg, NL80211_BSS_SEEN_MS_AGO,
4924                         jiffies_to_msecs(jiffies - intbss->ts)))
4925                 goto nla_put_failure;
4926
4927         switch (rdev->wiphy.signal_type) {
4928         case CFG80211_SIGNAL_TYPE_MBM:
4929                 if (nla_put_u32(msg, NL80211_BSS_SIGNAL_MBM, res->signal))
4930                         goto nla_put_failure;
4931                 break;
4932         case CFG80211_SIGNAL_TYPE_UNSPEC:
4933                 if (nla_put_u8(msg, NL80211_BSS_SIGNAL_UNSPEC, res->signal))
4934                         goto nla_put_failure;
4935                 break;
4936         default:
4937                 break;
4938         }
4939
4940         switch (wdev->iftype) {
4941         case NL80211_IFTYPE_P2P_CLIENT:
4942         case NL80211_IFTYPE_STATION:
4943                 if (intbss == wdev->current_bss &&
4944                     nla_put_u32(msg, NL80211_BSS_STATUS,
4945                                 NL80211_BSS_STATUS_ASSOCIATED))
4946                         goto nla_put_failure;
4947                 break;
4948         case NL80211_IFTYPE_ADHOC:
4949                 if (intbss == wdev->current_bss &&
4950                     nla_put_u32(msg, NL80211_BSS_STATUS,
4951                                 NL80211_BSS_STATUS_IBSS_JOINED))
4952                         goto nla_put_failure;
4953                 break;
4954         default:
4955                 break;
4956         }
4957
4958         nla_nest_end(msg, bss);
4959
4960         return genlmsg_end(msg, hdr);
4961
4962  nla_put_failure:
4963         genlmsg_cancel(msg, hdr);
4964         return -EMSGSIZE;
4965 }
4966
4967 static int nl80211_dump_scan(struct sk_buff *skb,
4968                              struct netlink_callback *cb)
4969 {
4970         struct cfg80211_registered_device *rdev;
4971         struct net_device *dev;
4972         struct cfg80211_internal_bss *scan;
4973         struct wireless_dev *wdev;
4974         int start = cb->args[1], idx = 0;
4975         int err;
4976
4977         err = nl80211_prepare_netdev_dump(skb, cb, &rdev, &dev);
4978         if (err)
4979                 return err;
4980
4981         wdev = dev->ieee80211_ptr;
4982
4983         wdev_lock(wdev);
4984         spin_lock_bh(&rdev->bss_lock);
4985         cfg80211_bss_expire(rdev);
4986
4987         cb->seq = rdev->bss_generation;
4988
4989         list_for_each_entry(scan, &rdev->bss_list, list) {
4990                 if (++idx <= start)
4991                         continue;
4992                 if (nl80211_send_bss(skb, cb,
4993                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
4994                                 rdev, wdev, scan) < 0) {
4995                         idx--;
4996                         break;
4997                 }
4998         }
4999
5000         spin_unlock_bh(&rdev->bss_lock);
5001         wdev_unlock(wdev);
5002
5003         cb->args[1] = idx;
5004         nl80211_finish_netdev_dump(rdev);
5005
5006         return skb->len;
5007 }
5008
5009 static int nl80211_send_survey(struct sk_buff *msg, u32 portid, u32 seq,
5010                                 int flags, struct net_device *dev,
5011                                 struct survey_info *survey)
5012 {
5013         void *hdr;
5014         struct nlattr *infoattr;
5015
5016         hdr = nl80211hdr_put(msg, portid, seq, flags,
5017                              NL80211_CMD_NEW_SURVEY_RESULTS);
5018         if (!hdr)
5019                 return -ENOMEM;
5020
5021         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex))
5022                 goto nla_put_failure;
5023
5024         infoattr = nla_nest_start(msg, NL80211_ATTR_SURVEY_INFO);
5025         if (!infoattr)
5026                 goto nla_put_failure;
5027
5028         if (nla_put_u32(msg, NL80211_SURVEY_INFO_FREQUENCY,
5029                         survey->channel->center_freq))
5030                 goto nla_put_failure;
5031
5032         if ((survey->filled & SURVEY_INFO_NOISE_DBM) &&
5033             nla_put_u8(msg, NL80211_SURVEY_INFO_NOISE, survey->noise))
5034                 goto nla_put_failure;
5035         if ((survey->filled & SURVEY_INFO_IN_USE) &&
5036             nla_put_flag(msg, NL80211_SURVEY_INFO_IN_USE))
5037                 goto nla_put_failure;
5038         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME) &&
5039             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME,
5040                         survey->channel_time))
5041                 goto nla_put_failure;
5042         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME_BUSY) &&
5043             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME_BUSY,
5044                         survey->channel_time_busy))
5045                 goto nla_put_failure;
5046         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME_EXT_BUSY) &&
5047             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME_EXT_BUSY,
5048                         survey->channel_time_ext_busy))
5049                 goto nla_put_failure;
5050         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME_RX) &&
5051             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME_RX,
5052                         survey->channel_time_rx))
5053                 goto nla_put_failure;
5054         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME_TX) &&
5055             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME_TX,
5056                         survey->channel_time_tx))
5057                 goto nla_put_failure;
5058
5059         nla_nest_end(msg, infoattr);
5060
5061         return genlmsg_end(msg, hdr);
5062
5063  nla_put_failure:
5064         genlmsg_cancel(msg, hdr);
5065         return -EMSGSIZE;
5066 }
5067
5068 static int nl80211_dump_survey(struct sk_buff *skb,
5069                         struct netlink_callback *cb)
5070 {
5071         struct survey_info survey;
5072         struct cfg80211_registered_device *dev;
5073         struct net_device *netdev;
5074         int survey_idx = cb->args[1];
5075         int res;
5076
5077         res = nl80211_prepare_netdev_dump(skb, cb, &dev, &netdev);
5078         if (res)
5079                 return res;
5080
5081         if (!dev->ops->dump_survey) {
5082                 res = -EOPNOTSUPP;
5083                 goto out_err;
5084         }
5085
5086         while (1) {
5087                 struct ieee80211_channel *chan;
5088
5089                 res = rdev_dump_survey(dev, netdev, survey_idx, &survey);
5090                 if (res == -ENOENT)
5091                         break;
5092                 if (res)
5093                         goto out_err;
5094
5095                 /* Survey without a channel doesn't make sense */
5096                 if (!survey.channel) {
5097                         res = -EINVAL;
5098                         goto out;
5099                 }
5100
5101                 chan = ieee80211_get_channel(&dev->wiphy,
5102                                              survey.channel->center_freq);
5103                 if (!chan || chan->flags & IEEE80211_CHAN_DISABLED) {
5104                         survey_idx++;
5105                         continue;
5106                 }
5107
5108                 if (nl80211_send_survey(skb,
5109                                 NETLINK_CB(cb->skb).portid,
5110                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
5111                                 netdev,
5112                                 &survey) < 0)
5113                         goto out;
5114                 survey_idx++;
5115         }
5116
5117  out:
5118         cb->args[1] = survey_idx;
5119         res = skb->len;
5120  out_err:
5121         nl80211_finish_netdev_dump(dev);
5122         return res;
5123 }
5124
5125 static bool nl80211_valid_wpa_versions(u32 wpa_versions)
5126 {
5127         return !(wpa_versions & ~(NL80211_WPA_VERSION_1 |
5128                                   NL80211_WPA_VERSION_2));
5129 }
5130
5131 static int nl80211_authenticate(struct sk_buff *skb, struct genl_info *info)
5132 {
5133         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5134         struct net_device *dev = info->user_ptr[1];
5135         struct ieee80211_channel *chan;
5136         const u8 *bssid, *ssid, *ie = NULL, *sae_data = NULL;
5137         int err, ssid_len, ie_len = 0, sae_data_len = 0;
5138         enum nl80211_auth_type auth_type;
5139         struct key_parse key;
5140         bool local_state_change;
5141
5142         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
5143                 return -EINVAL;
5144
5145         if (!info->attrs[NL80211_ATTR_MAC])
5146                 return -EINVAL;
5147
5148         if (!info->attrs[NL80211_ATTR_AUTH_TYPE])
5149                 return -EINVAL;
5150
5151         if (!info->attrs[NL80211_ATTR_SSID])
5152                 return -EINVAL;
5153
5154         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ])
5155                 return -EINVAL;
5156
5157         err = nl80211_parse_key(info, &key);
5158         if (err)
5159                 return err;
5160
5161         if (key.idx >= 0) {
5162                 if (key.type != -1 && key.type != NL80211_KEYTYPE_GROUP)
5163                         return -EINVAL;
5164                 if (!key.p.key || !key.p.key_len)
5165                         return -EINVAL;
5166                 if ((key.p.cipher != WLAN_CIPHER_SUITE_WEP40 ||
5167                      key.p.key_len != WLAN_KEY_LEN_WEP40) &&
5168                     (key.p.cipher != WLAN_CIPHER_SUITE_WEP104 ||
5169                      key.p.key_len != WLAN_KEY_LEN_WEP104))
5170                         return -EINVAL;
5171                 if (key.idx > 4)
5172                         return -EINVAL;
5173         } else {
5174                 key.p.key_len = 0;
5175                 key.p.key = NULL;
5176         }
5177
5178         if (key.idx >= 0) {
5179                 int i;
5180                 bool ok = false;
5181                 for (i = 0; i < rdev->wiphy.n_cipher_suites; i++) {
5182                         if (key.p.cipher == rdev->wiphy.cipher_suites[i]) {
5183                                 ok = true;
5184                                 break;
5185                         }
5186                 }
5187                 if (!ok)
5188                         return -EINVAL;
5189         }
5190
5191         if (!rdev->ops->auth)
5192                 return -EOPNOTSUPP;
5193
5194         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
5195             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
5196                 return -EOPNOTSUPP;
5197
5198         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
5199         chan = ieee80211_get_channel(&rdev->wiphy,
5200                 nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
5201         if (!chan || (chan->flags & IEEE80211_CHAN_DISABLED))
5202                 return -EINVAL;
5203
5204         ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
5205         ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
5206
5207         if (info->attrs[NL80211_ATTR_IE]) {
5208                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
5209                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5210         }
5211
5212         auth_type = nla_get_u32(info->attrs[NL80211_ATTR_AUTH_TYPE]);
5213         if (!nl80211_valid_auth_type(rdev, auth_type, NL80211_CMD_AUTHENTICATE))
5214                 return -EINVAL;
5215
5216         if (auth_type == NL80211_AUTHTYPE_SAE &&
5217             !info->attrs[NL80211_ATTR_SAE_DATA])
5218                 return -EINVAL;
5219
5220         if (info->attrs[NL80211_ATTR_SAE_DATA]) {
5221                 if (auth_type != NL80211_AUTHTYPE_SAE)
5222                         return -EINVAL;
5223                 sae_data = nla_data(info->attrs[NL80211_ATTR_SAE_DATA]);
5224                 sae_data_len = nla_len(info->attrs[NL80211_ATTR_SAE_DATA]);
5225                 /* need to include at least Auth Transaction and Status Code */
5226                 if (sae_data_len < 4)
5227                         return -EINVAL;
5228         }
5229
5230         local_state_change = !!info->attrs[NL80211_ATTR_LOCAL_STATE_CHANGE];
5231
5232         /*
5233          * Since we no longer track auth state, ignore
5234          * requests to only change local state.
5235          */
5236         if (local_state_change)
5237                 return 0;
5238
5239         return cfg80211_mlme_auth(rdev, dev, chan, auth_type, bssid,
5240                                   ssid, ssid_len, ie, ie_len,
5241                                   key.p.key, key.p.key_len, key.idx,
5242                                   sae_data, sae_data_len);
5243 }
5244
5245 static int nl80211_crypto_settings(struct cfg80211_registered_device *rdev,
5246                                    struct genl_info *info,
5247                                    struct cfg80211_crypto_settings *settings,
5248                                    int cipher_limit)
5249 {
5250         memset(settings, 0, sizeof(*settings));
5251
5252         settings->control_port = info->attrs[NL80211_ATTR_CONTROL_PORT];
5253
5254         if (info->attrs[NL80211_ATTR_CONTROL_PORT_ETHERTYPE]) {
5255                 u16 proto;
5256                 proto = nla_get_u16(
5257                         info->attrs[NL80211_ATTR_CONTROL_PORT_ETHERTYPE]);
5258                 settings->control_port_ethertype = cpu_to_be16(proto);
5259                 if (!(rdev->wiphy.flags & WIPHY_FLAG_CONTROL_PORT_PROTOCOL) &&
5260                     proto != ETH_P_PAE)
5261                         return -EINVAL;
5262                 if (info->attrs[NL80211_ATTR_CONTROL_PORT_NO_ENCRYPT])
5263                         settings->control_port_no_encrypt = true;
5264         } else
5265                 settings->control_port_ethertype = cpu_to_be16(ETH_P_PAE);
5266
5267         if (info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]) {
5268                 void *data;
5269                 int len, i;
5270
5271                 data = nla_data(info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]);
5272                 len = nla_len(info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]);
5273                 settings->n_ciphers_pairwise = len / sizeof(u32);
5274
5275                 if (len % sizeof(u32))
5276                         return -EINVAL;
5277
5278                 if (settings->n_ciphers_pairwise > cipher_limit)
5279                         return -EINVAL;
5280
5281                 memcpy(settings->ciphers_pairwise, data, len);
5282
5283                 for (i = 0; i < settings->n_ciphers_pairwise; i++)
5284                         if (!cfg80211_supported_cipher_suite(
5285                                         &rdev->wiphy,
5286                                         settings->ciphers_pairwise[i]))
5287                                 return -EINVAL;
5288         }
5289
5290         if (info->attrs[NL80211_ATTR_CIPHER_SUITE_GROUP]) {
5291                 settings->cipher_group =
5292                         nla_get_u32(info->attrs[NL80211_ATTR_CIPHER_SUITE_GROUP]);
5293                 if (!cfg80211_supported_cipher_suite(&rdev->wiphy,
5294                                                      settings->cipher_group))
5295                         return -EINVAL;
5296         }
5297
5298         if (info->attrs[NL80211_ATTR_WPA_VERSIONS]) {
5299                 settings->wpa_versions =
5300                         nla_get_u32(info->attrs[NL80211_ATTR_WPA_VERSIONS]);
5301                 if (!nl80211_valid_wpa_versions(settings->wpa_versions))
5302                         return -EINVAL;
5303         }
5304
5305         if (info->attrs[NL80211_ATTR_AKM_SUITES]) {
5306                 void *data;
5307                 int len;
5308
5309                 data = nla_data(info->attrs[NL80211_ATTR_AKM_SUITES]);
5310                 len = nla_len(info->attrs[NL80211_ATTR_AKM_SUITES]);
5311                 settings->n_akm_suites = len / sizeof(u32);
5312
5313                 if (len % sizeof(u32))
5314                         return -EINVAL;
5315
5316                 if (settings->n_akm_suites > NL80211_MAX_NR_AKM_SUITES)
5317                         return -EINVAL;
5318
5319                 memcpy(settings->akm_suites, data, len);
5320         }
5321
5322         return 0;
5323 }
5324
5325 static int nl80211_associate(struct sk_buff *skb, struct genl_info *info)
5326 {
5327         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5328         struct net_device *dev = info->user_ptr[1];
5329         struct cfg80211_crypto_settings crypto;
5330         struct ieee80211_channel *chan;
5331         const u8 *bssid, *ssid, *ie = NULL, *prev_bssid = NULL;
5332         int err, ssid_len, ie_len = 0;
5333         bool use_mfp = false;
5334         u32 flags = 0;
5335         struct ieee80211_ht_cap *ht_capa = NULL;
5336         struct ieee80211_ht_cap *ht_capa_mask = NULL;
5337
5338         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
5339                 return -EINVAL;
5340
5341         if (!info->attrs[NL80211_ATTR_MAC] ||
5342             !info->attrs[NL80211_ATTR_SSID] ||
5343             !info->attrs[NL80211_ATTR_WIPHY_FREQ])
5344                 return -EINVAL;
5345
5346         if (!rdev->ops->assoc)
5347                 return -EOPNOTSUPP;
5348
5349         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
5350             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
5351                 return -EOPNOTSUPP;
5352
5353         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
5354
5355         chan = ieee80211_get_channel(&rdev->wiphy,
5356                 nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
5357         if (!chan || (chan->flags & IEEE80211_CHAN_DISABLED))
5358                 return -EINVAL;
5359
5360         ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
5361         ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
5362
5363         if (info->attrs[NL80211_ATTR_IE]) {
5364                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
5365                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5366         }
5367
5368         if (info->attrs[NL80211_ATTR_USE_MFP]) {
5369                 enum nl80211_mfp mfp =
5370                         nla_get_u32(info->attrs[NL80211_ATTR_USE_MFP]);
5371                 if (mfp == NL80211_MFP_REQUIRED)
5372                         use_mfp = true;
5373                 else if (mfp != NL80211_MFP_NO)
5374                         return -EINVAL;
5375         }
5376
5377         if (info->attrs[NL80211_ATTR_PREV_BSSID])
5378                 prev_bssid = nla_data(info->attrs[NL80211_ATTR_PREV_BSSID]);
5379
5380         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_HT]))
5381                 flags |= ASSOC_REQ_DISABLE_HT;
5382
5383         if (info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
5384                 ht_capa_mask =
5385                         nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]);
5386
5387         if (info->attrs[NL80211_ATTR_HT_CAPABILITY]) {
5388                 if (!ht_capa_mask)
5389                         return -EINVAL;
5390                 ht_capa = nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]);
5391         }
5392
5393         err = nl80211_crypto_settings(rdev, info, &crypto, 1);
5394         if (!err)
5395                 err = cfg80211_mlme_assoc(rdev, dev, chan, bssid, prev_bssid,
5396                                           ssid, ssid_len, ie, ie_len, use_mfp,
5397                                           &crypto, flags, ht_capa,
5398                                           ht_capa_mask);
5399
5400         return err;
5401 }
5402
5403 static int nl80211_deauthenticate(struct sk_buff *skb, struct genl_info *info)
5404 {
5405         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5406         struct net_device *dev = info->user_ptr[1];
5407         const u8 *ie = NULL, *bssid;
5408         int ie_len = 0;
5409         u16 reason_code;
5410         bool local_state_change;
5411
5412         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
5413                 return -EINVAL;
5414
5415         if (!info->attrs[NL80211_ATTR_MAC])
5416                 return -EINVAL;
5417
5418         if (!info->attrs[NL80211_ATTR_REASON_CODE])
5419                 return -EINVAL;
5420
5421         if (!rdev->ops->deauth)
5422                 return -EOPNOTSUPP;
5423
5424         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
5425             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
5426                 return -EOPNOTSUPP;
5427
5428         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
5429
5430         reason_code = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
5431         if (reason_code == 0) {
5432                 /* Reason Code 0 is reserved */
5433                 return -EINVAL;
5434         }
5435
5436         if (info->attrs[NL80211_ATTR_IE]) {
5437                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
5438                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5439         }
5440
5441         local_state_change = !!info->attrs[NL80211_ATTR_LOCAL_STATE_CHANGE];
5442
5443         return cfg80211_mlme_deauth(rdev, dev, bssid, ie, ie_len, reason_code,
5444                                     local_state_change);
5445 }
5446
5447 static int nl80211_disassociate(struct sk_buff *skb, struct genl_info *info)
5448 {
5449         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5450         struct net_device *dev = info->user_ptr[1];
5451         const u8 *ie = NULL, *bssid;
5452         int ie_len = 0;
5453         u16 reason_code;
5454         bool local_state_change;
5455
5456         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
5457                 return -EINVAL;
5458
5459         if (!info->attrs[NL80211_ATTR_MAC])
5460                 return -EINVAL;
5461
5462         if (!info->attrs[NL80211_ATTR_REASON_CODE])
5463                 return -EINVAL;
5464
5465         if (!rdev->ops->disassoc)
5466                 return -EOPNOTSUPP;
5467
5468         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
5469             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
5470                 return -EOPNOTSUPP;
5471
5472         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
5473
5474         reason_code = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
5475         if (reason_code == 0) {
5476                 /* Reason Code 0 is reserved */
5477                 return -EINVAL;
5478         }
5479
5480         if (info->attrs[NL80211_ATTR_IE]) {
5481                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
5482                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5483         }
5484
5485         local_state_change = !!info->attrs[NL80211_ATTR_LOCAL_STATE_CHANGE];
5486
5487         return cfg80211_mlme_disassoc(rdev, dev, bssid, ie, ie_len, reason_code,
5488                                       local_state_change);
5489 }
5490
5491 static bool
5492 nl80211_parse_mcast_rate(struct cfg80211_registered_device *rdev,
5493                          int mcast_rate[IEEE80211_NUM_BANDS],
5494                          int rateval)
5495 {
5496         struct wiphy *wiphy = &rdev->wiphy;
5497         bool found = false;
5498         int band, i;
5499
5500         for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
5501                 struct ieee80211_supported_band *sband;
5502
5503                 sband = wiphy->bands[band];
5504                 if (!sband)
5505                         continue;
5506
5507                 for (i = 0; i < sband->n_bitrates; i++) {
5508                         if (sband->bitrates[i].bitrate == rateval) {
5509                                 mcast_rate[band] = i + 1;
5510                                 found = true;
5511                                 break;
5512                         }
5513                 }
5514         }
5515
5516         return found;
5517 }
5518
5519 static int nl80211_join_ibss(struct sk_buff *skb, struct genl_info *info)
5520 {
5521         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5522         struct net_device *dev = info->user_ptr[1];
5523         struct cfg80211_ibss_params ibss;
5524         struct wiphy *wiphy;
5525         struct cfg80211_cached_keys *connkeys = NULL;
5526         int err;
5527
5528         memset(&ibss, 0, sizeof(ibss));
5529
5530         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
5531                 return -EINVAL;
5532
5533         if (!info->attrs[NL80211_ATTR_SSID] ||
5534             !nla_len(info->attrs[NL80211_ATTR_SSID]))
5535                 return -EINVAL;
5536
5537         ibss.beacon_interval = 100;
5538
5539         if (info->attrs[NL80211_ATTR_BEACON_INTERVAL]) {
5540                 ibss.beacon_interval =
5541                         nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
5542                 if (ibss.beacon_interval < 1 || ibss.beacon_interval > 10000)
5543                         return -EINVAL;
5544         }
5545
5546         if (!rdev->ops->join_ibss)
5547                 return -EOPNOTSUPP;
5548
5549         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC)
5550                 return -EOPNOTSUPP;
5551
5552         wiphy = &rdev->wiphy;
5553
5554         if (info->attrs[NL80211_ATTR_MAC]) {
5555                 ibss.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
5556
5557                 if (!is_valid_ether_addr(ibss.bssid))
5558                         return -EINVAL;
5559         }
5560         ibss.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
5561         ibss.ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
5562
5563         if (info->attrs[NL80211_ATTR_IE]) {
5564                 ibss.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
5565                 ibss.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5566         }
5567
5568         err = nl80211_parse_chandef(rdev, info, &ibss.chandef);
5569         if (err)
5570                 return err;
5571
5572         if (!cfg80211_reg_can_beacon(&rdev->wiphy, &ibss.chandef))
5573                 return -EINVAL;
5574
5575         if (ibss.chandef.width > NL80211_CHAN_WIDTH_40)
5576                 return -EINVAL;
5577         if (ibss.chandef.width != NL80211_CHAN_WIDTH_20_NOHT &&
5578             !(rdev->wiphy.features & NL80211_FEATURE_HT_IBSS))
5579                 return -EINVAL;
5580
5581         ibss.channel_fixed = !!info->attrs[NL80211_ATTR_FREQ_FIXED];
5582         ibss.privacy = !!info->attrs[NL80211_ATTR_PRIVACY];
5583
5584         if (info->attrs[NL80211_ATTR_BSS_BASIC_RATES]) {
5585                 u8 *rates =
5586                         nla_data(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
5587                 int n_rates =
5588                         nla_len(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
5589                 struct ieee80211_supported_band *sband =
5590                         wiphy->bands[ibss.chandef.chan->band];
5591
5592                 err = ieee80211_get_ratemask(sband, rates, n_rates,
5593                                              &ibss.basic_rates);
5594                 if (err)
5595                         return err;
5596         }
5597
5598         if (info->attrs[NL80211_ATTR_MCAST_RATE] &&
5599             !nl80211_parse_mcast_rate(rdev, ibss.mcast_rate,
5600                         nla_get_u32(info->attrs[NL80211_ATTR_MCAST_RATE])))
5601                 return -EINVAL;
5602
5603         if (ibss.privacy && info->attrs[NL80211_ATTR_KEYS]) {
5604                 bool no_ht = false;
5605
5606                 connkeys = nl80211_parse_connkeys(rdev,
5607                                           info->attrs[NL80211_ATTR_KEYS],
5608                                           &no_ht);
5609                 if (IS_ERR(connkeys))
5610                         return PTR_ERR(connkeys);
5611
5612                 if ((ibss.chandef.width != NL80211_CHAN_WIDTH_20_NOHT) &&
5613                     no_ht) {
5614                         kfree(connkeys);
5615                         return -EINVAL;
5616                 }
5617         }
5618
5619         ibss.control_port =
5620                 nla_get_flag(info->attrs[NL80211_ATTR_CONTROL_PORT]);
5621
5622         err = cfg80211_join_ibss(rdev, dev, &ibss, connkeys);
5623         if (err)
5624                 kfree(connkeys);
5625         return err;
5626 }
5627
5628 static int nl80211_leave_ibss(struct sk_buff *skb, struct genl_info *info)
5629 {
5630         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5631         struct net_device *dev = info->user_ptr[1];
5632
5633         if (!rdev->ops->leave_ibss)
5634                 return -EOPNOTSUPP;
5635
5636         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC)
5637                 return -EOPNOTSUPP;
5638
5639         return cfg80211_leave_ibss(rdev, dev, false);
5640 }
5641
5642 static int nl80211_set_mcast_rate(struct sk_buff *skb, struct genl_info *info)
5643 {
5644         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5645         struct net_device *dev = info->user_ptr[1];
5646         int mcast_rate[IEEE80211_NUM_BANDS];
5647         u32 nla_rate;
5648         int err;
5649
5650         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC &&
5651             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
5652                 return -EOPNOTSUPP;
5653
5654         if (!rdev->ops->set_mcast_rate)
5655                 return -EOPNOTSUPP;
5656
5657         memset(mcast_rate, 0, sizeof(mcast_rate));
5658
5659         if (!info->attrs[NL80211_ATTR_MCAST_RATE])
5660                 return -EINVAL;
5661
5662         nla_rate = nla_get_u32(info->attrs[NL80211_ATTR_MCAST_RATE]);
5663         if (!nl80211_parse_mcast_rate(rdev, mcast_rate, nla_rate))
5664                 return -EINVAL;
5665
5666         err = rdev->ops->set_mcast_rate(&rdev->wiphy, dev, mcast_rate);
5667
5668         return err;
5669 }
5670
5671
5672 #ifdef CONFIG_NL80211_TESTMODE
5673 static struct genl_multicast_group nl80211_testmode_mcgrp = {
5674         .name = "testmode",
5675 };
5676
5677 static int nl80211_testmode_do(struct sk_buff *skb, struct genl_info *info)
5678 {
5679         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5680         int err;
5681
5682         if (!info->attrs[NL80211_ATTR_TESTDATA])
5683                 return -EINVAL;
5684
5685         err = -EOPNOTSUPP;
5686         if (rdev->ops->testmode_cmd) {
5687                 rdev->testmode_info = info;
5688                 err = rdev_testmode_cmd(rdev,
5689                                 nla_data(info->attrs[NL80211_ATTR_TESTDATA]),
5690                                 nla_len(info->attrs[NL80211_ATTR_TESTDATA]));
5691                 rdev->testmode_info = NULL;
5692         }
5693
5694         return err;
5695 }
5696
5697 static int nl80211_testmode_dump(struct sk_buff *skb,
5698                                  struct netlink_callback *cb)
5699 {
5700         struct cfg80211_registered_device *rdev;
5701         int err;
5702         long phy_idx;
5703         void *data = NULL;
5704         int data_len = 0;
5705
5706         if (cb->args[0]) {
5707                 /*
5708                  * 0 is a valid index, but not valid for args[0],
5709                  * so we need to offset by 1.
5710                  */
5711                 phy_idx = cb->args[0] - 1;
5712         } else {
5713                 err = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
5714                                   nl80211_fam.attrbuf, nl80211_fam.maxattr,
5715                                   nl80211_policy);
5716                 if (err)
5717                         return err;
5718
5719                 mutex_lock(&cfg80211_mutex);
5720                 rdev = __cfg80211_rdev_from_attrs(sock_net(skb->sk),
5721                                                   nl80211_fam.attrbuf);
5722                 if (IS_ERR(rdev)) {
5723                         mutex_unlock(&cfg80211_mutex);
5724                         return PTR_ERR(rdev);
5725                 }
5726                 phy_idx = rdev->wiphy_idx;
5727                 rdev = NULL;
5728                 mutex_unlock(&cfg80211_mutex);
5729
5730                 if (nl80211_fam.attrbuf[NL80211_ATTR_TESTDATA])
5731                         cb->args[1] =
5732                                 (long)nl80211_fam.attrbuf[NL80211_ATTR_TESTDATA];
5733         }
5734
5735         if (cb->args[1]) {
5736                 data = nla_data((void *)cb->args[1]);
5737                 data_len = nla_len((void *)cb->args[1]);
5738         }
5739
5740         mutex_lock(&cfg80211_mutex);
5741         rdev = cfg80211_rdev_by_wiphy_idx(phy_idx);
5742         if (!rdev) {
5743                 mutex_unlock(&cfg80211_mutex);
5744                 return -ENOENT;
5745         }
5746         cfg80211_lock_rdev(rdev);
5747         mutex_unlock(&cfg80211_mutex);
5748
5749         if (!rdev->ops->testmode_dump) {
5750                 err = -EOPNOTSUPP;
5751                 goto out_err;
5752         }
5753
5754         while (1) {
5755                 void *hdr = nl80211hdr_put(skb, NETLINK_CB(cb->skb).portid,
5756                                            cb->nlh->nlmsg_seq, NLM_F_MULTI,
5757                                            NL80211_CMD_TESTMODE);
5758                 struct nlattr *tmdata;
5759
5760                 if (nla_put_u32(skb, NL80211_ATTR_WIPHY, phy_idx)) {
5761                         genlmsg_cancel(skb, hdr);
5762                         break;
5763                 }
5764
5765                 tmdata = nla_nest_start(skb, NL80211_ATTR_TESTDATA);
5766                 if (!tmdata) {
5767                         genlmsg_cancel(skb, hdr);
5768                         break;
5769                 }
5770                 err = rdev_testmode_dump(rdev, skb, cb, data, data_len);
5771                 nla_nest_end(skb, tmdata);
5772
5773                 if (err == -ENOBUFS || err == -ENOENT) {
5774                         genlmsg_cancel(skb, hdr);
5775                         break;
5776                 } else if (err) {
5777                         genlmsg_cancel(skb, hdr);
5778                         goto out_err;
5779                 }
5780
5781                 genlmsg_end(skb, hdr);
5782         }
5783
5784         err = skb->len;
5785         /* see above */
5786         cb->args[0] = phy_idx + 1;
5787  out_err:
5788         cfg80211_unlock_rdev(rdev);
5789         return err;
5790 }
5791
5792 static struct sk_buff *
5793 __cfg80211_testmode_alloc_skb(struct cfg80211_registered_device *rdev,
5794                               int approxlen, u32 portid, u32 seq, gfp_t gfp)
5795 {
5796         struct sk_buff *skb;
5797         void *hdr;
5798         struct nlattr *data;
5799
5800         skb = nlmsg_new(approxlen + 100, gfp);
5801         if (!skb)
5802                 return NULL;
5803
5804         hdr = nl80211hdr_put(skb, portid, seq, 0, NL80211_CMD_TESTMODE);
5805         if (!hdr) {
5806                 kfree_skb(skb);
5807                 return NULL;
5808         }
5809
5810         if (nla_put_u32(skb, NL80211_ATTR_WIPHY, rdev->wiphy_idx))
5811                 goto nla_put_failure;
5812         data = nla_nest_start(skb, NL80211_ATTR_TESTDATA);
5813
5814         ((void **)skb->cb)[0] = rdev;
5815         ((void **)skb->cb)[1] = hdr;
5816         ((void **)skb->cb)[2] = data;
5817
5818         return skb;
5819
5820  nla_put_failure:
5821         kfree_skb(skb);
5822         return NULL;
5823 }
5824
5825 struct sk_buff *cfg80211_testmode_alloc_reply_skb(struct wiphy *wiphy,
5826                                                   int approxlen)
5827 {
5828         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
5829
5830         if (WARN_ON(!rdev->testmode_info))
5831                 return NULL;
5832
5833         return __cfg80211_testmode_alloc_skb(rdev, approxlen,
5834                                 rdev->testmode_info->snd_portid,
5835                                 rdev->testmode_info->snd_seq,
5836                                 GFP_KERNEL);
5837 }
5838 EXPORT_SYMBOL(cfg80211_testmode_alloc_reply_skb);
5839
5840 int cfg80211_testmode_reply(struct sk_buff *skb)
5841 {
5842         struct cfg80211_registered_device *rdev = ((void **)skb->cb)[0];
5843         void *hdr = ((void **)skb->cb)[1];
5844         struct nlattr *data = ((void **)skb->cb)[2];
5845
5846         if (WARN_ON(!rdev->testmode_info)) {
5847                 kfree_skb(skb);
5848                 return -EINVAL;
5849         }
5850
5851         nla_nest_end(skb, data);
5852         genlmsg_end(skb, hdr);
5853         return genlmsg_reply(skb, rdev->testmode_info);
5854 }
5855 EXPORT_SYMBOL(cfg80211_testmode_reply);
5856
5857 struct sk_buff *cfg80211_testmode_alloc_event_skb(struct wiphy *wiphy,
5858                                                   int approxlen, gfp_t gfp)
5859 {
5860         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
5861
5862         return __cfg80211_testmode_alloc_skb(rdev, approxlen, 0, 0, gfp);
5863 }
5864 EXPORT_SYMBOL(cfg80211_testmode_alloc_event_skb);
5865
5866 void cfg80211_testmode_event(struct sk_buff *skb, gfp_t gfp)
5867 {
5868         void *hdr = ((void **)skb->cb)[1];
5869         struct nlattr *data = ((void **)skb->cb)[2];
5870
5871         nla_nest_end(skb, data);
5872         genlmsg_end(skb, hdr);
5873         genlmsg_multicast(skb, 0, nl80211_testmode_mcgrp.id, gfp);
5874 }
5875 EXPORT_SYMBOL(cfg80211_testmode_event);
5876 #endif
5877
5878 static int nl80211_connect(struct sk_buff *skb, struct genl_info *info)
5879 {
5880         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5881         struct net_device *dev = info->user_ptr[1];
5882         struct cfg80211_connect_params connect;
5883         struct wiphy *wiphy;
5884         struct cfg80211_cached_keys *connkeys = NULL;
5885         int err;
5886
5887         memset(&connect, 0, sizeof(connect));
5888
5889         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
5890                 return -EINVAL;
5891
5892         if (!info->attrs[NL80211_ATTR_SSID] ||
5893             !nla_len(info->attrs[NL80211_ATTR_SSID]))
5894                 return -EINVAL;
5895
5896         if (info->attrs[NL80211_ATTR_AUTH_TYPE]) {
5897                 connect.auth_type =
5898                         nla_get_u32(info->attrs[NL80211_ATTR_AUTH_TYPE]);
5899                 if (!nl80211_valid_auth_type(rdev, connect.auth_type,
5900                                              NL80211_CMD_CONNECT))
5901                         return -EINVAL;
5902         } else
5903                 connect.auth_type = NL80211_AUTHTYPE_AUTOMATIC;
5904
5905         connect.privacy = info->attrs[NL80211_ATTR_PRIVACY];
5906
5907         err = nl80211_crypto_settings(rdev, info, &connect.crypto,
5908                                       NL80211_MAX_NR_CIPHER_SUITES);
5909         if (err)
5910                 return err;
5911
5912         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
5913             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
5914                 return -EOPNOTSUPP;
5915
5916         wiphy = &rdev->wiphy;
5917
5918         connect.bg_scan_period = -1;
5919         if (info->attrs[NL80211_ATTR_BG_SCAN_PERIOD] &&
5920                 (wiphy->flags & WIPHY_FLAG_SUPPORTS_FW_ROAM)) {
5921                 connect.bg_scan_period =
5922                         nla_get_u16(info->attrs[NL80211_ATTR_BG_SCAN_PERIOD]);
5923         }
5924
5925         if (info->attrs[NL80211_ATTR_MAC])
5926                 connect.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
5927         connect.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
5928         connect.ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
5929
5930         if (info->attrs[NL80211_ATTR_IE]) {
5931                 connect.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
5932                 connect.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5933         }
5934
5935         if (info->attrs[NL80211_ATTR_USE_MFP]) {
5936                 connect.mfp = nla_get_u32(info->attrs[NL80211_ATTR_USE_MFP]);
5937                 if (connect.mfp != NL80211_MFP_REQUIRED &&
5938                     connect.mfp != NL80211_MFP_NO)
5939                         return -EINVAL;
5940         } else {
5941                 connect.mfp = NL80211_MFP_NO;
5942         }
5943
5944         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
5945                 connect.channel =
5946                         ieee80211_get_channel(wiphy,
5947                             nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
5948                 if (!connect.channel ||
5949                     connect.channel->flags & IEEE80211_CHAN_DISABLED)
5950                         return -EINVAL;
5951         }
5952
5953         if (connect.privacy && info->attrs[NL80211_ATTR_KEYS]) {
5954                 connkeys = nl80211_parse_connkeys(rdev,
5955                                           info->attrs[NL80211_ATTR_KEYS], NULL);
5956                 if (IS_ERR(connkeys))
5957                         return PTR_ERR(connkeys);
5958         }
5959
5960         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_HT]))
5961                 connect.flags |= ASSOC_REQ_DISABLE_HT;
5962
5963         if (info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
5964                 memcpy(&connect.ht_capa_mask,
5965                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]),
5966                        sizeof(connect.ht_capa_mask));
5967
5968         if (info->attrs[NL80211_ATTR_HT_CAPABILITY]) {
5969                 if (!info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]) {
5970                         kfree(connkeys);
5971                         return -EINVAL;
5972                 }
5973                 memcpy(&connect.ht_capa,
5974                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]),
5975                        sizeof(connect.ht_capa));
5976         }
5977
5978         err = cfg80211_connect(rdev, dev, &connect, connkeys);
5979         if (err)
5980                 kfree(connkeys);
5981         return err;
5982 }
5983
5984 static int nl80211_disconnect(struct sk_buff *skb, struct genl_info *info)
5985 {
5986         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5987         struct net_device *dev = info->user_ptr[1];
5988         u16 reason;
5989
5990         if (!info->attrs[NL80211_ATTR_REASON_CODE])
5991                 reason = WLAN_REASON_DEAUTH_LEAVING;
5992         else
5993                 reason = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
5994
5995         if (reason == 0)
5996                 return -EINVAL;
5997
5998         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
5999             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
6000                 return -EOPNOTSUPP;
6001
6002         return cfg80211_disconnect(rdev, dev, reason, true);
6003 }
6004
6005 static int nl80211_wiphy_netns(struct sk_buff *skb, struct genl_info *info)
6006 {
6007         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6008         struct net *net;
6009         int err;
6010         u32 pid;
6011
6012         if (!info->attrs[NL80211_ATTR_PID])
6013                 return -EINVAL;
6014
6015         pid = nla_get_u32(info->attrs[NL80211_ATTR_PID]);
6016
6017         net = get_net_ns_by_pid(pid);
6018         if (IS_ERR(net))
6019                 return PTR_ERR(net);
6020
6021         err = 0;
6022
6023         /* check if anything to do */
6024         if (!net_eq(wiphy_net(&rdev->wiphy), net))
6025                 err = cfg80211_switch_netns(rdev, net);
6026
6027         put_net(net);
6028         return err;
6029 }
6030
6031 static int nl80211_setdel_pmksa(struct sk_buff *skb, struct genl_info *info)
6032 {
6033         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6034         int (*rdev_ops)(struct wiphy *wiphy, struct net_device *dev,
6035                         struct cfg80211_pmksa *pmksa) = NULL;
6036         struct net_device *dev = info->user_ptr[1];
6037         struct cfg80211_pmksa pmksa;
6038
6039         memset(&pmksa, 0, sizeof(struct cfg80211_pmksa));
6040
6041         if (!info->attrs[NL80211_ATTR_MAC])
6042                 return -EINVAL;
6043
6044         if (!info->attrs[NL80211_ATTR_PMKID])
6045                 return -EINVAL;
6046
6047         pmksa.pmkid = nla_data(info->attrs[NL80211_ATTR_PMKID]);
6048         pmksa.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
6049
6050         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
6051             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
6052                 return -EOPNOTSUPP;
6053
6054         switch (info->genlhdr->cmd) {
6055         case NL80211_CMD_SET_PMKSA:
6056                 rdev_ops = rdev->ops->set_pmksa;
6057                 break;
6058         case NL80211_CMD_DEL_PMKSA:
6059                 rdev_ops = rdev->ops->del_pmksa;
6060                 break;
6061         default:
6062                 WARN_ON(1);
6063                 break;
6064         }
6065
6066         if (!rdev_ops)
6067                 return -EOPNOTSUPP;
6068
6069         return rdev_ops(&rdev->wiphy, dev, &pmksa);
6070 }
6071
6072 static int nl80211_flush_pmksa(struct sk_buff *skb, struct genl_info *info)
6073 {
6074         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6075         struct net_device *dev = info->user_ptr[1];
6076
6077         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
6078             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
6079                 return -EOPNOTSUPP;
6080
6081         if (!rdev->ops->flush_pmksa)
6082                 return -EOPNOTSUPP;
6083
6084         return rdev_flush_pmksa(rdev, dev);
6085 }
6086
6087 static int nl80211_tdls_mgmt(struct sk_buff *skb, struct genl_info *info)
6088 {
6089         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6090         struct net_device *dev = info->user_ptr[1];
6091         u8 action_code, dialog_token;
6092         u16 status_code;
6093         u8 *peer;
6094
6095         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) ||
6096             !rdev->ops->tdls_mgmt)
6097                 return -EOPNOTSUPP;
6098
6099         if (!info->attrs[NL80211_ATTR_TDLS_ACTION] ||
6100             !info->attrs[NL80211_ATTR_STATUS_CODE] ||
6101             !info->attrs[NL80211_ATTR_TDLS_DIALOG_TOKEN] ||
6102             !info->attrs[NL80211_ATTR_IE] ||
6103             !info->attrs[NL80211_ATTR_MAC])
6104                 return -EINVAL;
6105
6106         peer = nla_data(info->attrs[NL80211_ATTR_MAC]);
6107         action_code = nla_get_u8(info->attrs[NL80211_ATTR_TDLS_ACTION]);
6108         status_code = nla_get_u16(info->attrs[NL80211_ATTR_STATUS_CODE]);
6109         dialog_token = nla_get_u8(info->attrs[NL80211_ATTR_TDLS_DIALOG_TOKEN]);
6110
6111         return rdev_tdls_mgmt(rdev, dev, peer, action_code,
6112                               dialog_token, status_code,
6113                               nla_data(info->attrs[NL80211_ATTR_IE]),
6114                               nla_len(info->attrs[NL80211_ATTR_IE]));
6115 }
6116
6117 static int nl80211_tdls_oper(struct sk_buff *skb, struct genl_info *info)
6118 {
6119         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6120         struct net_device *dev = info->user_ptr[1];
6121         enum nl80211_tdls_operation operation;
6122         u8 *peer;
6123
6124         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) ||
6125             !rdev->ops->tdls_oper)
6126                 return -EOPNOTSUPP;
6127
6128         if (!info->attrs[NL80211_ATTR_TDLS_OPERATION] ||
6129             !info->attrs[NL80211_ATTR_MAC])
6130                 return -EINVAL;
6131
6132         operation = nla_get_u8(info->attrs[NL80211_ATTR_TDLS_OPERATION]);
6133         peer = nla_data(info->attrs[NL80211_ATTR_MAC]);
6134
6135         return rdev_tdls_oper(rdev, dev, peer, operation);
6136 }
6137
6138 static int nl80211_remain_on_channel(struct sk_buff *skb,
6139                                      struct genl_info *info)
6140 {
6141         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6142         struct wireless_dev *wdev = info->user_ptr[1];
6143         struct cfg80211_chan_def chandef;
6144         struct sk_buff *msg;
6145         void *hdr;
6146         u64 cookie;
6147         u32 duration;
6148         int err;
6149
6150         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ] ||
6151             !info->attrs[NL80211_ATTR_DURATION])
6152                 return -EINVAL;
6153
6154         duration = nla_get_u32(info->attrs[NL80211_ATTR_DURATION]);
6155
6156         if (!rdev->ops->remain_on_channel ||
6157             !(rdev->wiphy.flags & WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL))
6158                 return -EOPNOTSUPP;
6159
6160         /*
6161          * We should be on that channel for at least a minimum amount of
6162          * time (10ms) but no longer than the driver supports.
6163          */
6164         if (duration < NL80211_MIN_REMAIN_ON_CHANNEL_TIME ||
6165             duration > rdev->wiphy.max_remain_on_channel_duration)
6166                 return -EINVAL;
6167
6168         err = nl80211_parse_chandef(rdev, info, &chandef);
6169         if (err)
6170                 return err;
6171
6172         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
6173         if (!msg)
6174                 return -ENOMEM;
6175
6176         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
6177                              NL80211_CMD_REMAIN_ON_CHANNEL);
6178
6179         if (IS_ERR(hdr)) {
6180                 err = PTR_ERR(hdr);
6181                 goto free_msg;
6182         }
6183
6184         err = rdev_remain_on_channel(rdev, wdev, chandef.chan,
6185                                      duration, &cookie);
6186
6187         if (err)
6188                 goto free_msg;
6189
6190         if (nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie))
6191                 goto nla_put_failure;
6192
6193         genlmsg_end(msg, hdr);
6194
6195         return genlmsg_reply(msg, info);
6196
6197  nla_put_failure:
6198         err = -ENOBUFS;
6199  free_msg:
6200         nlmsg_free(msg);
6201         return err;
6202 }
6203
6204 static int nl80211_cancel_remain_on_channel(struct sk_buff *skb,
6205                                             struct genl_info *info)
6206 {
6207         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6208         struct wireless_dev *wdev = info->user_ptr[1];
6209         u64 cookie;
6210
6211         if (!info->attrs[NL80211_ATTR_COOKIE])
6212                 return -EINVAL;
6213
6214         if (!rdev->ops->cancel_remain_on_channel)
6215                 return -EOPNOTSUPP;
6216
6217         cookie = nla_get_u64(info->attrs[NL80211_ATTR_COOKIE]);
6218
6219         return rdev_cancel_remain_on_channel(rdev, wdev, cookie);
6220 }
6221
6222 static u32 rateset_to_mask(struct ieee80211_supported_band *sband,
6223                            u8 *rates, u8 rates_len)
6224 {
6225         u8 i;
6226         u32 mask = 0;
6227
6228         for (i = 0; i < rates_len; i++) {
6229                 int rate = (rates[i] & 0x7f) * 5;
6230                 int ridx;
6231                 for (ridx = 0; ridx < sband->n_bitrates; ridx++) {
6232                         struct ieee80211_rate *srate =
6233                                 &sband->bitrates[ridx];
6234                         if (rate == srate->bitrate) {
6235                                 mask |= 1 << ridx;
6236                                 break;
6237                         }
6238                 }
6239                 if (ridx == sband->n_bitrates)
6240                         return 0; /* rate not found */
6241         }
6242
6243         return mask;
6244 }
6245
6246 static bool ht_rateset_to_mask(struct ieee80211_supported_band *sband,
6247                                u8 *rates, u8 rates_len,
6248                                u8 mcs[IEEE80211_HT_MCS_MASK_LEN])
6249 {
6250         u8 i;
6251
6252         memset(mcs, 0, IEEE80211_HT_MCS_MASK_LEN);
6253
6254         for (i = 0; i < rates_len; i++) {
6255                 int ridx, rbit;
6256
6257                 ridx = rates[i] / 8;
6258                 rbit = BIT(rates[i] % 8);
6259
6260                 /* check validity */
6261                 if ((ridx < 0) || (ridx >= IEEE80211_HT_MCS_MASK_LEN))
6262                         return false;
6263
6264                 /* check availability */
6265                 if (sband->ht_cap.mcs.rx_mask[ridx] & rbit)
6266                         mcs[ridx] |= rbit;
6267                 else
6268                         return false;
6269         }
6270
6271         return true;
6272 }
6273
6274 static const struct nla_policy nl80211_txattr_policy[NL80211_TXRATE_MAX + 1] = {
6275         [NL80211_TXRATE_LEGACY] = { .type = NLA_BINARY,
6276                                     .len = NL80211_MAX_SUPP_RATES },
6277         [NL80211_TXRATE_MCS] = { .type = NLA_BINARY,
6278                                  .len = NL80211_MAX_SUPP_HT_RATES },
6279 };
6280
6281 static int nl80211_set_tx_bitrate_mask(struct sk_buff *skb,
6282                                        struct genl_info *info)
6283 {
6284         struct nlattr *tb[NL80211_TXRATE_MAX + 1];
6285         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6286         struct cfg80211_bitrate_mask mask;
6287         int rem, i;
6288         struct net_device *dev = info->user_ptr[1];
6289         struct nlattr *tx_rates;
6290         struct ieee80211_supported_band *sband;
6291
6292         if (info->attrs[NL80211_ATTR_TX_RATES] == NULL)
6293                 return -EINVAL;
6294
6295         if (!rdev->ops->set_bitrate_mask)
6296                 return -EOPNOTSUPP;
6297
6298         memset(&mask, 0, sizeof(mask));
6299         /* Default to all rates enabled */
6300         for (i = 0; i < IEEE80211_NUM_BANDS; i++) {
6301                 sband = rdev->wiphy.bands[i];
6302                 mask.control[i].legacy =
6303                         sband ? (1 << sband->n_bitrates) - 1 : 0;
6304                 if (sband)
6305                         memcpy(mask.control[i].mcs,
6306                                sband->ht_cap.mcs.rx_mask,
6307                                sizeof(mask.control[i].mcs));
6308                 else
6309                         memset(mask.control[i].mcs, 0,
6310                                sizeof(mask.control[i].mcs));
6311         }
6312
6313         /*
6314          * The nested attribute uses enum nl80211_band as the index. This maps
6315          * directly to the enum ieee80211_band values used in cfg80211.
6316          */
6317         BUILD_BUG_ON(NL80211_MAX_SUPP_HT_RATES > IEEE80211_HT_MCS_MASK_LEN * 8);
6318         nla_for_each_nested(tx_rates, info->attrs[NL80211_ATTR_TX_RATES], rem)
6319         {
6320                 enum ieee80211_band band = nla_type(tx_rates);
6321                 if (band < 0 || band >= IEEE80211_NUM_BANDS)
6322                         return -EINVAL;
6323                 sband = rdev->wiphy.bands[band];
6324                 if (sband == NULL)
6325                         return -EINVAL;
6326                 nla_parse(tb, NL80211_TXRATE_MAX, nla_data(tx_rates),
6327                           nla_len(tx_rates), nl80211_txattr_policy);
6328                 if (tb[NL80211_TXRATE_LEGACY]) {
6329                         mask.control[band].legacy = rateset_to_mask(
6330                                 sband,
6331                                 nla_data(tb[NL80211_TXRATE_LEGACY]),
6332                                 nla_len(tb[NL80211_TXRATE_LEGACY]));
6333                         if ((mask.control[band].legacy == 0) &&
6334                             nla_len(tb[NL80211_TXRATE_LEGACY]))
6335                                 return -EINVAL;
6336                 }
6337                 if (tb[NL80211_TXRATE_MCS]) {
6338                         if (!ht_rateset_to_mask(
6339                                         sband,
6340                                         nla_data(tb[NL80211_TXRATE_MCS]),
6341                                         nla_len(tb[NL80211_TXRATE_MCS]),
6342                                         mask.control[band].mcs))
6343                                 return -EINVAL;
6344                 }
6345
6346                 if (mask.control[band].legacy == 0) {
6347                         /* don't allow empty legacy rates if HT
6348                          * is not even supported. */
6349                         if (!rdev->wiphy.bands[band]->ht_cap.ht_supported)
6350                                 return -EINVAL;
6351
6352                         for (i = 0; i < IEEE80211_HT_MCS_MASK_LEN; i++)
6353                                 if (mask.control[band].mcs[i])
6354                                         break;
6355
6356                         /* legacy and mcs rates may not be both empty */
6357                         if (i == IEEE80211_HT_MCS_MASK_LEN)
6358                                 return -EINVAL;
6359                 }
6360         }
6361
6362         return rdev_set_bitrate_mask(rdev, dev, NULL, &mask);
6363 }
6364
6365 static int nl80211_register_mgmt(struct sk_buff *skb, struct genl_info *info)
6366 {
6367         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6368         struct wireless_dev *wdev = info->user_ptr[1];
6369         u16 frame_type = IEEE80211_FTYPE_MGMT | IEEE80211_STYPE_ACTION;
6370
6371         if (!info->attrs[NL80211_ATTR_FRAME_MATCH])
6372                 return -EINVAL;
6373
6374         if (info->attrs[NL80211_ATTR_FRAME_TYPE])
6375                 frame_type = nla_get_u16(info->attrs[NL80211_ATTR_FRAME_TYPE]);
6376
6377         switch (wdev->iftype) {
6378         case NL80211_IFTYPE_STATION:
6379         case NL80211_IFTYPE_ADHOC:
6380         case NL80211_IFTYPE_P2P_CLIENT:
6381         case NL80211_IFTYPE_AP:
6382         case NL80211_IFTYPE_AP_VLAN:
6383         case NL80211_IFTYPE_MESH_POINT:
6384         case NL80211_IFTYPE_P2P_GO:
6385         case NL80211_IFTYPE_P2P_DEVICE:
6386                 break;
6387         default:
6388                 return -EOPNOTSUPP;
6389         }
6390
6391         /* not much point in registering if we can't reply */
6392         if (!rdev->ops->mgmt_tx)
6393                 return -EOPNOTSUPP;
6394
6395         return cfg80211_mlme_register_mgmt(wdev, info->snd_portid, frame_type,
6396                         nla_data(info->attrs[NL80211_ATTR_FRAME_MATCH]),
6397                         nla_len(info->attrs[NL80211_ATTR_FRAME_MATCH]));
6398 }
6399
6400 static int nl80211_tx_mgmt(struct sk_buff *skb, struct genl_info *info)
6401 {
6402         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6403         struct wireless_dev *wdev = info->user_ptr[1];
6404         struct cfg80211_chan_def chandef;
6405         int err;
6406         void *hdr = NULL;
6407         u64 cookie;
6408         struct sk_buff *msg = NULL;
6409         unsigned int wait = 0;
6410         bool offchan, no_cck, dont_wait_for_ack;
6411
6412         dont_wait_for_ack = info->attrs[NL80211_ATTR_DONT_WAIT_FOR_ACK];
6413
6414         if (!info->attrs[NL80211_ATTR_FRAME])
6415                 return -EINVAL;
6416
6417         if (!rdev->ops->mgmt_tx)
6418                 return -EOPNOTSUPP;
6419
6420         switch (wdev->iftype) {
6421         case NL80211_IFTYPE_STATION:
6422         case NL80211_IFTYPE_ADHOC:
6423         case NL80211_IFTYPE_P2P_CLIENT:
6424         case NL80211_IFTYPE_AP:
6425         case NL80211_IFTYPE_AP_VLAN:
6426         case NL80211_IFTYPE_MESH_POINT:
6427         case NL80211_IFTYPE_P2P_GO:
6428         case NL80211_IFTYPE_P2P_DEVICE:
6429                 break;
6430         default:
6431                 return -EOPNOTSUPP;
6432         }
6433
6434         if (info->attrs[NL80211_ATTR_DURATION]) {
6435                 if (!(rdev->wiphy.flags & WIPHY_FLAG_OFFCHAN_TX))
6436                         return -EINVAL;
6437                 wait = nla_get_u32(info->attrs[NL80211_ATTR_DURATION]);
6438
6439                 /*
6440                  * We should wait on the channel for at least a minimum amount
6441                  * of time (10ms) but no longer than the driver supports.
6442                  */
6443                 if (wait < NL80211_MIN_REMAIN_ON_CHANNEL_TIME ||
6444                     wait > rdev->wiphy.max_remain_on_channel_duration)
6445                         return -EINVAL;
6446
6447         }
6448
6449         offchan = info->attrs[NL80211_ATTR_OFFCHANNEL_TX_OK];
6450
6451         if (offchan && !(rdev->wiphy.flags & WIPHY_FLAG_OFFCHAN_TX))
6452                 return -EINVAL;
6453
6454         no_cck = nla_get_flag(info->attrs[NL80211_ATTR_TX_NO_CCK_RATE]);
6455
6456         err = nl80211_parse_chandef(rdev, info, &chandef);
6457         if (err)
6458                 return err;
6459
6460         if (!dont_wait_for_ack) {
6461                 msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
6462                 if (!msg)
6463                         return -ENOMEM;
6464
6465                 hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
6466                                      NL80211_CMD_FRAME);
6467
6468                 if (IS_ERR(hdr)) {
6469                         err = PTR_ERR(hdr);
6470                         goto free_msg;
6471                 }
6472         }
6473
6474         err = cfg80211_mlme_mgmt_tx(rdev, wdev, chandef.chan, offchan, wait,
6475                                     nla_data(info->attrs[NL80211_ATTR_FRAME]),
6476                                     nla_len(info->attrs[NL80211_ATTR_FRAME]),
6477                                     no_cck, dont_wait_for_ack, &cookie);
6478         if (err)
6479                 goto free_msg;
6480
6481         if (msg) {
6482                 if (nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie))
6483                         goto nla_put_failure;
6484
6485                 genlmsg_end(msg, hdr);
6486                 return genlmsg_reply(msg, info);
6487         }
6488
6489         return 0;
6490
6491  nla_put_failure:
6492         err = -ENOBUFS;
6493  free_msg:
6494         nlmsg_free(msg);
6495         return err;
6496 }
6497
6498 static int nl80211_tx_mgmt_cancel_wait(struct sk_buff *skb, struct genl_info *info)
6499 {
6500         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6501         struct wireless_dev *wdev = info->user_ptr[1];
6502         u64 cookie;
6503
6504         if (!info->attrs[NL80211_ATTR_COOKIE])
6505                 return -EINVAL;
6506
6507         if (!rdev->ops->mgmt_tx_cancel_wait)
6508                 return -EOPNOTSUPP;
6509
6510         switch (wdev->iftype) {
6511         case NL80211_IFTYPE_STATION:
6512         case NL80211_IFTYPE_ADHOC:
6513         case NL80211_IFTYPE_P2P_CLIENT:
6514         case NL80211_IFTYPE_AP:
6515         case NL80211_IFTYPE_AP_VLAN:
6516         case NL80211_IFTYPE_P2P_GO:
6517         case NL80211_IFTYPE_P2P_DEVICE:
6518                 break;
6519         default:
6520                 return -EOPNOTSUPP;
6521         }
6522
6523         cookie = nla_get_u64(info->attrs[NL80211_ATTR_COOKIE]);
6524
6525         return rdev_mgmt_tx_cancel_wait(rdev, wdev, cookie);
6526 }
6527
6528 static int nl80211_set_power_save(struct sk_buff *skb, struct genl_info *info)
6529 {
6530         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6531         struct wireless_dev *wdev;
6532         struct net_device *dev = info->user_ptr[1];
6533         u8 ps_state;
6534         bool state;
6535         int err;
6536
6537         if (!info->attrs[NL80211_ATTR_PS_STATE])
6538                 return -EINVAL;
6539
6540         ps_state = nla_get_u32(info->attrs[NL80211_ATTR_PS_STATE]);
6541
6542         if (ps_state != NL80211_PS_DISABLED && ps_state != NL80211_PS_ENABLED)
6543                 return -EINVAL;
6544
6545         wdev = dev->ieee80211_ptr;
6546
6547         if (!rdev->ops->set_power_mgmt)
6548                 return -EOPNOTSUPP;
6549
6550         state = (ps_state == NL80211_PS_ENABLED) ? true : false;
6551
6552         if (state == wdev->ps)
6553                 return 0;
6554
6555         err = rdev_set_power_mgmt(rdev, dev, state, wdev->ps_timeout);
6556         if (!err)
6557                 wdev->ps = state;
6558         return err;
6559 }
6560
6561 static int nl80211_get_power_save(struct sk_buff *skb, struct genl_info *info)
6562 {
6563         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6564         enum nl80211_ps_state ps_state;
6565         struct wireless_dev *wdev;
6566         struct net_device *dev = info->user_ptr[1];
6567         struct sk_buff *msg;
6568         void *hdr;
6569         int err;
6570
6571         wdev = dev->ieee80211_ptr;
6572
6573         if (!rdev->ops->set_power_mgmt)
6574                 return -EOPNOTSUPP;
6575
6576         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
6577         if (!msg)
6578                 return -ENOMEM;
6579
6580         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
6581                              NL80211_CMD_GET_POWER_SAVE);
6582         if (!hdr) {
6583                 err = -ENOBUFS;
6584                 goto free_msg;
6585         }
6586
6587         if (wdev->ps)
6588                 ps_state = NL80211_PS_ENABLED;
6589         else
6590                 ps_state = NL80211_PS_DISABLED;
6591
6592         if (nla_put_u32(msg, NL80211_ATTR_PS_STATE, ps_state))
6593                 goto nla_put_failure;
6594
6595         genlmsg_end(msg, hdr);
6596         return genlmsg_reply(msg, info);
6597
6598  nla_put_failure:
6599         err = -ENOBUFS;
6600  free_msg:
6601         nlmsg_free(msg);
6602         return err;
6603 }
6604
6605 static struct nla_policy
6606 nl80211_attr_cqm_policy[NL80211_ATTR_CQM_MAX + 1] __read_mostly = {
6607         [NL80211_ATTR_CQM_RSSI_THOLD] = { .type = NLA_U32 },
6608         [NL80211_ATTR_CQM_RSSI_HYST] = { .type = NLA_U32 },
6609         [NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT] = { .type = NLA_U32 },
6610         [NL80211_ATTR_CQM_TXE_RATE] = { .type = NLA_U32 },
6611         [NL80211_ATTR_CQM_TXE_PKTS] = { .type = NLA_U32 },
6612         [NL80211_ATTR_CQM_TXE_INTVL] = { .type = NLA_U32 },
6613 };
6614
6615 static int nl80211_set_cqm_txe(struct genl_info *info,
6616                                u32 rate, u32 pkts, u32 intvl)
6617 {
6618         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6619         struct wireless_dev *wdev;
6620         struct net_device *dev = info->user_ptr[1];
6621
6622         if (rate > 100 || intvl > NL80211_CQM_TXE_MAX_INTVL)
6623                 return -EINVAL;
6624
6625         wdev = dev->ieee80211_ptr;
6626
6627         if (!rdev->ops->set_cqm_txe_config)
6628                 return -EOPNOTSUPP;
6629
6630         if (wdev->iftype != NL80211_IFTYPE_STATION &&
6631             wdev->iftype != NL80211_IFTYPE_P2P_CLIENT)
6632                 return -EOPNOTSUPP;
6633
6634         return rdev_set_cqm_txe_config(rdev, dev, rate, pkts, intvl);
6635 }
6636
6637 static int nl80211_set_cqm_rssi(struct genl_info *info,
6638                                 s32 threshold, u32 hysteresis)
6639 {
6640         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6641         struct wireless_dev *wdev;
6642         struct net_device *dev = info->user_ptr[1];
6643
6644         if (threshold > 0)
6645                 return -EINVAL;
6646
6647         wdev = dev->ieee80211_ptr;
6648
6649         if (!rdev->ops->set_cqm_rssi_config)
6650                 return -EOPNOTSUPP;
6651
6652         if (wdev->iftype != NL80211_IFTYPE_STATION &&
6653             wdev->iftype != NL80211_IFTYPE_P2P_CLIENT)
6654                 return -EOPNOTSUPP;
6655
6656         return rdev_set_cqm_rssi_config(rdev, dev, threshold, hysteresis);
6657 }
6658
6659 static int nl80211_set_cqm(struct sk_buff *skb, struct genl_info *info)
6660 {
6661         struct nlattr *attrs[NL80211_ATTR_CQM_MAX + 1];
6662         struct nlattr *cqm;
6663         int err;
6664
6665         cqm = info->attrs[NL80211_ATTR_CQM];
6666         if (!cqm) {
6667                 err = -EINVAL;
6668                 goto out;
6669         }
6670
6671         err = nla_parse_nested(attrs, NL80211_ATTR_CQM_MAX, cqm,
6672                                nl80211_attr_cqm_policy);
6673         if (err)
6674                 goto out;
6675
6676         if (attrs[NL80211_ATTR_CQM_RSSI_THOLD] &&
6677             attrs[NL80211_ATTR_CQM_RSSI_HYST]) {
6678                 s32 threshold;
6679                 u32 hysteresis;
6680                 threshold = nla_get_u32(attrs[NL80211_ATTR_CQM_RSSI_THOLD]);
6681                 hysteresis = nla_get_u32(attrs[NL80211_ATTR_CQM_RSSI_HYST]);
6682                 err = nl80211_set_cqm_rssi(info, threshold, hysteresis);
6683         } else if (attrs[NL80211_ATTR_CQM_TXE_RATE] &&
6684                    attrs[NL80211_ATTR_CQM_TXE_PKTS] &&
6685                    attrs[NL80211_ATTR_CQM_TXE_INTVL]) {
6686                 u32 rate, pkts, intvl;
6687                 rate = nla_get_u32(attrs[NL80211_ATTR_CQM_TXE_RATE]);
6688                 pkts = nla_get_u32(attrs[NL80211_ATTR_CQM_TXE_PKTS]);
6689                 intvl = nla_get_u32(attrs[NL80211_ATTR_CQM_TXE_INTVL]);
6690                 err = nl80211_set_cqm_txe(info, rate, pkts, intvl);
6691         } else
6692                 err = -EINVAL;
6693
6694 out:
6695         return err;
6696 }
6697
6698 static int nl80211_join_mesh(struct sk_buff *skb, struct genl_info *info)
6699 {
6700         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6701         struct net_device *dev = info->user_ptr[1];
6702         struct mesh_config cfg;
6703         struct mesh_setup setup;
6704         int err;
6705
6706         /* start with default */
6707         memcpy(&cfg, &default_mesh_config, sizeof(cfg));
6708         memcpy(&setup, &default_mesh_setup, sizeof(setup));
6709
6710         if (info->attrs[NL80211_ATTR_MESH_CONFIG]) {
6711                 /* and parse parameters if given */
6712                 err = nl80211_parse_mesh_config(info, &cfg, NULL);
6713                 if (err)
6714                         return err;
6715         }
6716
6717         if (!info->attrs[NL80211_ATTR_MESH_ID] ||
6718             !nla_len(info->attrs[NL80211_ATTR_MESH_ID]))
6719                 return -EINVAL;
6720
6721         setup.mesh_id = nla_data(info->attrs[NL80211_ATTR_MESH_ID]);
6722         setup.mesh_id_len = nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
6723
6724         if (info->attrs[NL80211_ATTR_MCAST_RATE] &&
6725             !nl80211_parse_mcast_rate(rdev, setup.mcast_rate,
6726                             nla_get_u32(info->attrs[NL80211_ATTR_MCAST_RATE])))
6727                         return -EINVAL;
6728
6729         if (info->attrs[NL80211_ATTR_BEACON_INTERVAL]) {
6730                 setup.beacon_interval =
6731                         nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
6732                 if (setup.beacon_interval < 10 ||
6733                     setup.beacon_interval > 10000)
6734                         return -EINVAL;
6735         }
6736
6737         if (info->attrs[NL80211_ATTR_DTIM_PERIOD]) {
6738                 setup.dtim_period =
6739                         nla_get_u32(info->attrs[NL80211_ATTR_DTIM_PERIOD]);
6740                 if (setup.dtim_period < 1 || setup.dtim_period > 100)
6741                         return -EINVAL;
6742         }
6743
6744         if (info->attrs[NL80211_ATTR_MESH_SETUP]) {
6745                 /* parse additional setup parameters if given */
6746                 err = nl80211_parse_mesh_setup(info, &setup);
6747                 if (err)
6748                         return err;
6749         }
6750
6751         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
6752                 err = nl80211_parse_chandef(rdev, info, &setup.chandef);
6753                 if (err)
6754                         return err;
6755         } else {
6756                 /* cfg80211_join_mesh() will sort it out */
6757                 setup.chandef.chan = NULL;
6758         }
6759
6760         return cfg80211_join_mesh(rdev, dev, &setup, &cfg);
6761 }
6762
6763 static int nl80211_leave_mesh(struct sk_buff *skb, struct genl_info *info)
6764 {
6765         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6766         struct net_device *dev = info->user_ptr[1];
6767
6768         return cfg80211_leave_mesh(rdev, dev);
6769 }
6770
6771 #ifdef CONFIG_PM
6772 static int nl80211_get_wowlan(struct sk_buff *skb, struct genl_info *info)
6773 {
6774         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6775         struct sk_buff *msg;
6776         void *hdr;
6777
6778         if (!rdev->wiphy.wowlan.flags && !rdev->wiphy.wowlan.n_patterns)
6779                 return -EOPNOTSUPP;
6780
6781         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
6782         if (!msg)
6783                 return -ENOMEM;
6784
6785         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
6786                              NL80211_CMD_GET_WOWLAN);
6787         if (!hdr)
6788                 goto nla_put_failure;
6789
6790         if (rdev->wowlan) {
6791                 struct nlattr *nl_wowlan;
6792
6793                 nl_wowlan = nla_nest_start(msg, NL80211_ATTR_WOWLAN_TRIGGERS);
6794                 if (!nl_wowlan)
6795                         goto nla_put_failure;
6796
6797                 if ((rdev->wowlan->any &&
6798                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_ANY)) ||
6799                     (rdev->wowlan->disconnect &&
6800                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_DISCONNECT)) ||
6801                     (rdev->wowlan->magic_pkt &&
6802                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_MAGIC_PKT)) ||
6803                     (rdev->wowlan->gtk_rekey_failure &&
6804                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE)) ||
6805                     (rdev->wowlan->eap_identity_req &&
6806                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST)) ||
6807                     (rdev->wowlan->four_way_handshake &&
6808                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE)) ||
6809                     (rdev->wowlan->rfkill_release &&
6810                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_RFKILL_RELEASE)))
6811                         goto nla_put_failure;
6812                 if (rdev->wowlan->n_patterns) {
6813                         struct nlattr *nl_pats, *nl_pat;
6814                         int i, pat_len;
6815
6816                         nl_pats = nla_nest_start(msg,
6817                                         NL80211_WOWLAN_TRIG_PKT_PATTERN);
6818                         if (!nl_pats)
6819                                 goto nla_put_failure;
6820
6821                         for (i = 0; i < rdev->wowlan->n_patterns; i++) {
6822                                 nl_pat = nla_nest_start(msg, i + 1);
6823                                 if (!nl_pat)
6824                                         goto nla_put_failure;
6825                                 pat_len = rdev->wowlan->patterns[i].pattern_len;
6826                                 if (nla_put(msg, NL80211_WOWLAN_PKTPAT_MASK,
6827                                             DIV_ROUND_UP(pat_len, 8),
6828                                             rdev->wowlan->patterns[i].mask) ||
6829                                     nla_put(msg, NL80211_WOWLAN_PKTPAT_PATTERN,
6830                                             pat_len,
6831                                             rdev->wowlan->patterns[i].pattern))
6832                                         goto nla_put_failure;
6833                                 nla_nest_end(msg, nl_pat);
6834                         }
6835                         nla_nest_end(msg, nl_pats);
6836                 }
6837
6838                 nla_nest_end(msg, nl_wowlan);
6839         }
6840
6841         genlmsg_end(msg, hdr);
6842         return genlmsg_reply(msg, info);
6843
6844 nla_put_failure:
6845         nlmsg_free(msg);
6846         return -ENOBUFS;
6847 }
6848
6849 static int nl80211_set_wowlan(struct sk_buff *skb, struct genl_info *info)
6850 {
6851         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6852         struct nlattr *tb[NUM_NL80211_WOWLAN_TRIG];
6853         struct cfg80211_wowlan new_triggers = {};
6854         struct cfg80211_wowlan *ntrig;
6855         struct wiphy_wowlan_support *wowlan = &rdev->wiphy.wowlan;
6856         int err, i;
6857         bool prev_enabled = rdev->wowlan;
6858
6859         if (!rdev->wiphy.wowlan.flags && !rdev->wiphy.wowlan.n_patterns)
6860                 return -EOPNOTSUPP;
6861
6862         if (!info->attrs[NL80211_ATTR_WOWLAN_TRIGGERS]) {
6863                 cfg80211_rdev_free_wowlan(rdev);
6864                 rdev->wowlan = NULL;
6865                 goto set_wakeup;
6866         }
6867
6868         err = nla_parse(tb, MAX_NL80211_WOWLAN_TRIG,
6869                         nla_data(info->attrs[NL80211_ATTR_WOWLAN_TRIGGERS]),
6870                         nla_len(info->attrs[NL80211_ATTR_WOWLAN_TRIGGERS]),
6871                         nl80211_wowlan_policy);
6872         if (err)
6873                 return err;
6874
6875         if (tb[NL80211_WOWLAN_TRIG_ANY]) {
6876                 if (!(wowlan->flags & WIPHY_WOWLAN_ANY))
6877                         return -EINVAL;
6878                 new_triggers.any = true;
6879         }
6880
6881         if (tb[NL80211_WOWLAN_TRIG_DISCONNECT]) {
6882                 if (!(wowlan->flags & WIPHY_WOWLAN_DISCONNECT))
6883                         return -EINVAL;
6884                 new_triggers.disconnect = true;
6885         }
6886
6887         if (tb[NL80211_WOWLAN_TRIG_MAGIC_PKT]) {
6888                 if (!(wowlan->flags & WIPHY_WOWLAN_MAGIC_PKT))
6889                         return -EINVAL;
6890                 new_triggers.magic_pkt = true;
6891         }
6892
6893         if (tb[NL80211_WOWLAN_TRIG_GTK_REKEY_SUPPORTED])
6894                 return -EINVAL;
6895
6896         if (tb[NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE]) {
6897                 if (!(wowlan->flags & WIPHY_WOWLAN_GTK_REKEY_FAILURE))
6898                         return -EINVAL;
6899                 new_triggers.gtk_rekey_failure = true;
6900         }
6901
6902         if (tb[NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST]) {
6903                 if (!(wowlan->flags & WIPHY_WOWLAN_EAP_IDENTITY_REQ))
6904                         return -EINVAL;
6905                 new_triggers.eap_identity_req = true;
6906         }
6907
6908         if (tb[NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE]) {
6909                 if (!(wowlan->flags & WIPHY_WOWLAN_4WAY_HANDSHAKE))
6910                         return -EINVAL;
6911                 new_triggers.four_way_handshake = true;
6912         }
6913
6914         if (tb[NL80211_WOWLAN_TRIG_RFKILL_RELEASE]) {
6915                 if (!(wowlan->flags & WIPHY_WOWLAN_RFKILL_RELEASE))
6916                         return -EINVAL;
6917                 new_triggers.rfkill_release = true;
6918         }
6919
6920         if (tb[NL80211_WOWLAN_TRIG_PKT_PATTERN]) {
6921                 struct nlattr *pat;
6922                 int n_patterns = 0;
6923                 int rem, pat_len, mask_len;
6924                 struct nlattr *pat_tb[NUM_NL80211_WOWLAN_PKTPAT];
6925
6926                 nla_for_each_nested(pat, tb[NL80211_WOWLAN_TRIG_PKT_PATTERN],
6927                                     rem)
6928                         n_patterns++;
6929                 if (n_patterns > wowlan->n_patterns)
6930                         return -EINVAL;
6931
6932                 new_triggers.patterns = kcalloc(n_patterns,
6933                                                 sizeof(new_triggers.patterns[0]),
6934                                                 GFP_KERNEL);
6935                 if (!new_triggers.patterns)
6936                         return -ENOMEM;
6937
6938                 new_triggers.n_patterns = n_patterns;
6939                 i = 0;
6940
6941                 nla_for_each_nested(pat, tb[NL80211_WOWLAN_TRIG_PKT_PATTERN],
6942                                     rem) {
6943                         nla_parse(pat_tb, MAX_NL80211_WOWLAN_PKTPAT,
6944                                   nla_data(pat), nla_len(pat), NULL);
6945                         err = -EINVAL;
6946                         if (!pat_tb[NL80211_WOWLAN_PKTPAT_MASK] ||
6947                             !pat_tb[NL80211_WOWLAN_PKTPAT_PATTERN])
6948                                 goto error;
6949                         pat_len = nla_len(pat_tb[NL80211_WOWLAN_PKTPAT_PATTERN]);
6950                         mask_len = DIV_ROUND_UP(pat_len, 8);
6951                         if (nla_len(pat_tb[NL80211_WOWLAN_PKTPAT_MASK]) !=
6952                             mask_len)
6953                                 goto error;
6954                         if (pat_len > wowlan->pattern_max_len ||
6955                             pat_len < wowlan->pattern_min_len)
6956                                 goto error;
6957
6958                         new_triggers.patterns[i].mask =
6959                                 kmalloc(mask_len + pat_len, GFP_KERNEL);
6960                         if (!new_triggers.patterns[i].mask) {
6961                                 err = -ENOMEM;
6962                                 goto error;
6963                         }
6964                         new_triggers.patterns[i].pattern =
6965                                 new_triggers.patterns[i].mask + mask_len;
6966                         memcpy(new_triggers.patterns[i].mask,
6967                                nla_data(pat_tb[NL80211_WOWLAN_PKTPAT_MASK]),
6968                                mask_len);
6969                         new_triggers.patterns[i].pattern_len = pat_len;
6970                         memcpy(new_triggers.patterns[i].pattern,
6971                                nla_data(pat_tb[NL80211_WOWLAN_PKTPAT_PATTERN]),
6972                                pat_len);
6973                         i++;
6974                 }
6975         }
6976
6977         ntrig = kmemdup(&new_triggers, sizeof(new_triggers), GFP_KERNEL);
6978         if (!ntrig) {
6979                 err = -ENOMEM;
6980                 goto error;
6981         }
6982         cfg80211_rdev_free_wowlan(rdev);
6983         rdev->wowlan = ntrig;
6984
6985  set_wakeup:
6986         if (rdev->ops->set_wakeup && prev_enabled != !!rdev->wowlan)
6987                 rdev_set_wakeup(rdev, rdev->wowlan);
6988
6989         return 0;
6990  error:
6991         for (i = 0; i < new_triggers.n_patterns; i++)
6992                 kfree(new_triggers.patterns[i].mask);
6993         kfree(new_triggers.patterns);
6994         return err;
6995 }
6996 #endif
6997
6998 static int nl80211_set_rekey_data(struct sk_buff *skb, struct genl_info *info)
6999 {
7000         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7001         struct net_device *dev = info->user_ptr[1];
7002         struct wireless_dev *wdev = dev->ieee80211_ptr;
7003         struct nlattr *tb[NUM_NL80211_REKEY_DATA];
7004         struct cfg80211_gtk_rekey_data rekey_data;
7005         int err;
7006
7007         if (!info->attrs[NL80211_ATTR_REKEY_DATA])
7008                 return -EINVAL;
7009
7010         err = nla_parse(tb, MAX_NL80211_REKEY_DATA,
7011                         nla_data(info->attrs[NL80211_ATTR_REKEY_DATA]),
7012                         nla_len(info->attrs[NL80211_ATTR_REKEY_DATA]),
7013                         nl80211_rekey_policy);
7014         if (err)
7015                 return err;
7016
7017         if (nla_len(tb[NL80211_REKEY_DATA_REPLAY_CTR]) != NL80211_REPLAY_CTR_LEN)
7018                 return -ERANGE;
7019         if (nla_len(tb[NL80211_REKEY_DATA_KEK]) != NL80211_KEK_LEN)
7020                 return -ERANGE;
7021         if (nla_len(tb[NL80211_REKEY_DATA_KCK]) != NL80211_KCK_LEN)
7022                 return -ERANGE;
7023
7024         memcpy(rekey_data.kek, nla_data(tb[NL80211_REKEY_DATA_KEK]),
7025                NL80211_KEK_LEN);
7026         memcpy(rekey_data.kck, nla_data(tb[NL80211_REKEY_DATA_KCK]),
7027                NL80211_KCK_LEN);
7028         memcpy(rekey_data.replay_ctr,
7029                nla_data(tb[NL80211_REKEY_DATA_REPLAY_CTR]),
7030                NL80211_REPLAY_CTR_LEN);
7031
7032         wdev_lock(wdev);
7033         if (!wdev->current_bss) {
7034                 err = -ENOTCONN;
7035                 goto out;
7036         }
7037
7038         if (!rdev->ops->set_rekey_data) {
7039                 err = -EOPNOTSUPP;
7040                 goto out;
7041         }
7042
7043         err = rdev_set_rekey_data(rdev, dev, &rekey_data);
7044  out:
7045         wdev_unlock(wdev);
7046         return err;
7047 }
7048
7049 static int nl80211_register_unexpected_frame(struct sk_buff *skb,
7050                                              struct genl_info *info)
7051 {
7052         struct net_device *dev = info->user_ptr[1];
7053         struct wireless_dev *wdev = dev->ieee80211_ptr;
7054
7055         if (wdev->iftype != NL80211_IFTYPE_AP &&
7056             wdev->iftype != NL80211_IFTYPE_P2P_GO)
7057                 return -EINVAL;
7058
7059         if (wdev->ap_unexpected_nlportid)
7060                 return -EBUSY;
7061
7062         wdev->ap_unexpected_nlportid = info->snd_portid;
7063         return 0;
7064 }
7065
7066 static int nl80211_probe_client(struct sk_buff *skb,
7067                                 struct genl_info *info)
7068 {
7069         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7070         struct net_device *dev = info->user_ptr[1];
7071         struct wireless_dev *wdev = dev->ieee80211_ptr;
7072         struct sk_buff *msg;
7073         void *hdr;
7074         const u8 *addr;
7075         u64 cookie;
7076         int err;
7077
7078         if (wdev->iftype != NL80211_IFTYPE_AP &&
7079             wdev->iftype != NL80211_IFTYPE_P2P_GO)
7080                 return -EOPNOTSUPP;
7081
7082         if (!info->attrs[NL80211_ATTR_MAC])
7083                 return -EINVAL;
7084
7085         if (!rdev->ops->probe_client)
7086                 return -EOPNOTSUPP;
7087
7088         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
7089         if (!msg)
7090                 return -ENOMEM;
7091
7092         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
7093                              NL80211_CMD_PROBE_CLIENT);
7094
7095         if (IS_ERR(hdr)) {
7096                 err = PTR_ERR(hdr);
7097                 goto free_msg;
7098         }
7099
7100         addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
7101
7102         err = rdev_probe_client(rdev, dev, addr, &cookie);
7103         if (err)
7104                 goto free_msg;
7105
7106         if (nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie))
7107                 goto nla_put_failure;
7108
7109         genlmsg_end(msg, hdr);
7110
7111         return genlmsg_reply(msg, info);
7112
7113  nla_put_failure:
7114         err = -ENOBUFS;
7115  free_msg:
7116         nlmsg_free(msg);
7117         return err;
7118 }
7119
7120 static int nl80211_register_beacons(struct sk_buff *skb, struct genl_info *info)
7121 {
7122         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7123         struct cfg80211_beacon_registration *reg, *nreg;
7124         int rv;
7125
7126         if (!(rdev->wiphy.flags & WIPHY_FLAG_REPORTS_OBSS))
7127                 return -EOPNOTSUPP;
7128
7129         nreg = kzalloc(sizeof(*nreg), GFP_KERNEL);
7130         if (!nreg)
7131                 return -ENOMEM;
7132
7133         /* First, check if already registered. */
7134         spin_lock_bh(&rdev->beacon_registrations_lock);
7135         list_for_each_entry(reg, &rdev->beacon_registrations, list) {
7136                 if (reg->nlportid == info->snd_portid) {
7137                         rv = -EALREADY;
7138                         goto out_err;
7139                 }
7140         }
7141         /* Add it to the list */
7142         nreg->nlportid = info->snd_portid;
7143         list_add(&nreg->list, &rdev->beacon_registrations);
7144
7145         spin_unlock_bh(&rdev->beacon_registrations_lock);
7146
7147         return 0;
7148 out_err:
7149         spin_unlock_bh(&rdev->beacon_registrations_lock);
7150         kfree(nreg);
7151         return rv;
7152 }
7153
7154 static int nl80211_start_p2p_device(struct sk_buff *skb, struct genl_info *info)
7155 {
7156         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7157         struct wireless_dev *wdev = info->user_ptr[1];
7158         int err;
7159
7160         if (!rdev->ops->start_p2p_device)
7161                 return -EOPNOTSUPP;
7162
7163         if (wdev->iftype != NL80211_IFTYPE_P2P_DEVICE)
7164                 return -EOPNOTSUPP;
7165
7166         if (wdev->p2p_started)
7167                 return 0;
7168
7169         mutex_lock(&rdev->devlist_mtx);
7170         err = cfg80211_can_add_interface(rdev, wdev->iftype);
7171         mutex_unlock(&rdev->devlist_mtx);
7172         if (err)
7173                 return err;
7174
7175         err = rdev_start_p2p_device(rdev, wdev);
7176         if (err)
7177                 return err;
7178
7179         wdev->p2p_started = true;
7180         mutex_lock(&rdev->devlist_mtx);
7181         rdev->opencount++;
7182         mutex_unlock(&rdev->devlist_mtx);
7183
7184         return 0;
7185 }
7186
7187 static int nl80211_stop_p2p_device(struct sk_buff *skb, struct genl_info *info)
7188 {
7189         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7190         struct wireless_dev *wdev = info->user_ptr[1];
7191
7192         if (wdev->iftype != NL80211_IFTYPE_P2P_DEVICE)
7193                 return -EOPNOTSUPP;
7194
7195         if (!rdev->ops->stop_p2p_device)
7196                 return -EOPNOTSUPP;
7197
7198         if (!wdev->p2p_started)
7199                 return 0;
7200
7201         rdev_stop_p2p_device(rdev, wdev);
7202         wdev->p2p_started = false;
7203
7204         mutex_lock(&rdev->devlist_mtx);
7205         rdev->opencount--;
7206         mutex_unlock(&rdev->devlist_mtx);
7207
7208         if (WARN_ON(rdev->scan_req && rdev->scan_req->wdev == wdev)) {
7209                 rdev->scan_req->aborted = true;
7210                 ___cfg80211_scan_done(rdev, true);
7211         }
7212
7213         return 0;
7214 }
7215
7216 #define NL80211_FLAG_NEED_WIPHY         0x01
7217 #define NL80211_FLAG_NEED_NETDEV        0x02
7218 #define NL80211_FLAG_NEED_RTNL          0x04
7219 #define NL80211_FLAG_CHECK_NETDEV_UP    0x08
7220 #define NL80211_FLAG_NEED_NETDEV_UP     (NL80211_FLAG_NEED_NETDEV |\
7221                                          NL80211_FLAG_CHECK_NETDEV_UP)
7222 #define NL80211_FLAG_NEED_WDEV          0x10
7223 /* If a netdev is associated, it must be UP, P2P must be started */
7224 #define NL80211_FLAG_NEED_WDEV_UP       (NL80211_FLAG_NEED_WDEV |\
7225                                          NL80211_FLAG_CHECK_NETDEV_UP)
7226
7227 static int nl80211_pre_doit(struct genl_ops *ops, struct sk_buff *skb,
7228                             struct genl_info *info)
7229 {
7230         struct cfg80211_registered_device *rdev;
7231         struct wireless_dev *wdev;
7232         struct net_device *dev;
7233         bool rtnl = ops->internal_flags & NL80211_FLAG_NEED_RTNL;
7234
7235         if (rtnl)
7236                 rtnl_lock();
7237
7238         if (ops->internal_flags & NL80211_FLAG_NEED_WIPHY) {
7239                 rdev = cfg80211_get_dev_from_info(genl_info_net(info), info);
7240                 if (IS_ERR(rdev)) {
7241                         if (rtnl)
7242                                 rtnl_unlock();
7243                         return PTR_ERR(rdev);
7244                 }
7245                 info->user_ptr[0] = rdev;
7246         } else if (ops->internal_flags & NL80211_FLAG_NEED_NETDEV ||
7247                    ops->internal_flags & NL80211_FLAG_NEED_WDEV) {
7248                 mutex_lock(&cfg80211_mutex);
7249                 wdev = __cfg80211_wdev_from_attrs(genl_info_net(info),
7250                                                   info->attrs);
7251                 if (IS_ERR(wdev)) {
7252                         mutex_unlock(&cfg80211_mutex);
7253                         if (rtnl)
7254                                 rtnl_unlock();
7255                         return PTR_ERR(wdev);
7256                 }
7257
7258                 dev = wdev->netdev;
7259                 rdev = wiphy_to_dev(wdev->wiphy);
7260
7261                 if (ops->internal_flags & NL80211_FLAG_NEED_NETDEV) {
7262                         if (!dev) {
7263                                 mutex_unlock(&cfg80211_mutex);
7264                                 if (rtnl)
7265                                         rtnl_unlock();
7266                                 return -EINVAL;
7267                         }
7268
7269                         info->user_ptr[1] = dev;
7270                 } else {
7271                         info->user_ptr[1] = wdev;
7272                 }
7273
7274                 if (dev) {
7275                         if (ops->internal_flags & NL80211_FLAG_CHECK_NETDEV_UP &&
7276                             !netif_running(dev)) {
7277                                 mutex_unlock(&cfg80211_mutex);
7278                                 if (rtnl)
7279                                         rtnl_unlock();
7280                                 return -ENETDOWN;
7281                         }
7282
7283                         dev_hold(dev);
7284                 } else if (ops->internal_flags & NL80211_FLAG_CHECK_NETDEV_UP) {
7285                         if (!wdev->p2p_started) {
7286                                 mutex_unlock(&cfg80211_mutex);
7287                                 if (rtnl)
7288                                         rtnl_unlock();
7289                                 return -ENETDOWN;
7290                         }
7291                 }
7292
7293                 cfg80211_lock_rdev(rdev);
7294
7295                 mutex_unlock(&cfg80211_mutex);
7296
7297                 info->user_ptr[0] = rdev;
7298         }
7299
7300         return 0;
7301 }
7302
7303 static void nl80211_post_doit(struct genl_ops *ops, struct sk_buff *skb,
7304                               struct genl_info *info)
7305 {
7306         if (info->user_ptr[0])
7307                 cfg80211_unlock_rdev(info->user_ptr[0]);
7308         if (info->user_ptr[1]) {
7309                 if (ops->internal_flags & NL80211_FLAG_NEED_WDEV) {
7310                         struct wireless_dev *wdev = info->user_ptr[1];
7311
7312                         if (wdev->netdev)
7313                                 dev_put(wdev->netdev);
7314                 } else {
7315                         dev_put(info->user_ptr[1]);
7316                 }
7317         }
7318         if (ops->internal_flags & NL80211_FLAG_NEED_RTNL)
7319                 rtnl_unlock();
7320 }
7321
7322 static struct genl_ops nl80211_ops[] = {
7323         {
7324                 .cmd = NL80211_CMD_GET_WIPHY,
7325                 .doit = nl80211_get_wiphy,
7326                 .dumpit = nl80211_dump_wiphy,
7327                 .policy = nl80211_policy,
7328                 /* can be retrieved by unprivileged users */
7329                 .internal_flags = NL80211_FLAG_NEED_WIPHY,
7330         },
7331         {
7332                 .cmd = NL80211_CMD_SET_WIPHY,
7333                 .doit = nl80211_set_wiphy,
7334                 .policy = nl80211_policy,
7335                 .flags = GENL_ADMIN_PERM,
7336                 .internal_flags = NL80211_FLAG_NEED_RTNL,
7337         },
7338         {
7339                 .cmd = NL80211_CMD_GET_INTERFACE,
7340                 .doit = nl80211_get_interface,
7341                 .dumpit = nl80211_dump_interface,
7342                 .policy = nl80211_policy,
7343                 /* can be retrieved by unprivileged users */
7344                 .internal_flags = NL80211_FLAG_NEED_WDEV,
7345         },
7346         {
7347                 .cmd = NL80211_CMD_SET_INTERFACE,
7348                 .doit = nl80211_set_interface,
7349                 .policy = nl80211_policy,
7350                 .flags = GENL_ADMIN_PERM,
7351                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7352                                   NL80211_FLAG_NEED_RTNL,
7353         },
7354         {
7355                 .cmd = NL80211_CMD_NEW_INTERFACE,
7356                 .doit = nl80211_new_interface,
7357                 .policy = nl80211_policy,
7358                 .flags = GENL_ADMIN_PERM,
7359                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
7360                                   NL80211_FLAG_NEED_RTNL,
7361         },
7362         {
7363                 .cmd = NL80211_CMD_DEL_INTERFACE,
7364                 .doit = nl80211_del_interface,
7365                 .policy = nl80211_policy,
7366                 .flags = GENL_ADMIN_PERM,
7367                 .internal_flags = NL80211_FLAG_NEED_WDEV |
7368                                   NL80211_FLAG_NEED_RTNL,
7369         },
7370         {
7371                 .cmd = NL80211_CMD_GET_KEY,
7372                 .doit = nl80211_get_key,
7373                 .policy = nl80211_policy,
7374                 .flags = GENL_ADMIN_PERM,
7375                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7376                                   NL80211_FLAG_NEED_RTNL,
7377         },
7378         {
7379                 .cmd = NL80211_CMD_SET_KEY,
7380                 .doit = nl80211_set_key,
7381                 .policy = nl80211_policy,
7382                 .flags = GENL_ADMIN_PERM,
7383                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7384                                   NL80211_FLAG_NEED_RTNL,
7385         },
7386         {
7387                 .cmd = NL80211_CMD_NEW_KEY,
7388                 .doit = nl80211_new_key,
7389                 .policy = nl80211_policy,
7390                 .flags = GENL_ADMIN_PERM,
7391                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7392                                   NL80211_FLAG_NEED_RTNL,
7393         },
7394         {
7395                 .cmd = NL80211_CMD_DEL_KEY,
7396                 .doit = nl80211_del_key,
7397                 .policy = nl80211_policy,
7398                 .flags = GENL_ADMIN_PERM,
7399                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7400                                   NL80211_FLAG_NEED_RTNL,
7401         },
7402         {
7403                 .cmd = NL80211_CMD_SET_BEACON,
7404                 .policy = nl80211_policy,
7405                 .flags = GENL_ADMIN_PERM,
7406                 .doit = nl80211_set_beacon,
7407                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7408                                   NL80211_FLAG_NEED_RTNL,
7409         },
7410         {
7411                 .cmd = NL80211_CMD_START_AP,
7412                 .policy = nl80211_policy,
7413                 .flags = GENL_ADMIN_PERM,
7414                 .doit = nl80211_start_ap,
7415                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7416                                   NL80211_FLAG_NEED_RTNL,
7417         },
7418         {
7419                 .cmd = NL80211_CMD_STOP_AP,
7420                 .policy = nl80211_policy,
7421                 .flags = GENL_ADMIN_PERM,
7422                 .doit = nl80211_stop_ap,
7423                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7424                                   NL80211_FLAG_NEED_RTNL,
7425         },
7426         {
7427                 .cmd = NL80211_CMD_GET_STATION,
7428                 .doit = nl80211_get_station,
7429                 .dumpit = nl80211_dump_station,
7430                 .policy = nl80211_policy,
7431                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7432                                   NL80211_FLAG_NEED_RTNL,
7433         },
7434         {
7435                 .cmd = NL80211_CMD_SET_STATION,
7436                 .doit = nl80211_set_station,
7437                 .policy = nl80211_policy,
7438                 .flags = GENL_ADMIN_PERM,
7439                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7440                                   NL80211_FLAG_NEED_RTNL,
7441         },
7442         {
7443                 .cmd = NL80211_CMD_NEW_STATION,
7444                 .doit = nl80211_new_station,
7445                 .policy = nl80211_policy,
7446                 .flags = GENL_ADMIN_PERM,
7447                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7448                                   NL80211_FLAG_NEED_RTNL,
7449         },
7450         {
7451                 .cmd = NL80211_CMD_DEL_STATION,
7452                 .doit = nl80211_del_station,
7453                 .policy = nl80211_policy,
7454                 .flags = GENL_ADMIN_PERM,
7455                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7456                                   NL80211_FLAG_NEED_RTNL,
7457         },
7458         {
7459                 .cmd = NL80211_CMD_GET_MPATH,
7460                 .doit = nl80211_get_mpath,
7461                 .dumpit = nl80211_dump_mpath,
7462                 .policy = nl80211_policy,
7463                 .flags = GENL_ADMIN_PERM,
7464                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7465                                   NL80211_FLAG_NEED_RTNL,
7466         },
7467         {
7468                 .cmd = NL80211_CMD_SET_MPATH,
7469                 .doit = nl80211_set_mpath,
7470                 .policy = nl80211_policy,
7471                 .flags = GENL_ADMIN_PERM,
7472                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7473                                   NL80211_FLAG_NEED_RTNL,
7474         },
7475         {
7476                 .cmd = NL80211_CMD_NEW_MPATH,
7477                 .doit = nl80211_new_mpath,
7478                 .policy = nl80211_policy,
7479                 .flags = GENL_ADMIN_PERM,
7480                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7481                                   NL80211_FLAG_NEED_RTNL,
7482         },
7483         {
7484                 .cmd = NL80211_CMD_DEL_MPATH,
7485                 .doit = nl80211_del_mpath,
7486                 .policy = nl80211_policy,
7487                 .flags = GENL_ADMIN_PERM,
7488                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7489                                   NL80211_FLAG_NEED_RTNL,
7490         },
7491         {
7492                 .cmd = NL80211_CMD_SET_BSS,
7493                 .doit = nl80211_set_bss,
7494                 .policy = nl80211_policy,
7495                 .flags = GENL_ADMIN_PERM,
7496                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7497                                   NL80211_FLAG_NEED_RTNL,
7498         },
7499         {
7500                 .cmd = NL80211_CMD_GET_REG,
7501                 .doit = nl80211_get_reg,
7502                 .policy = nl80211_policy,
7503                 /* can be retrieved by unprivileged users */
7504         },
7505         {
7506                 .cmd = NL80211_CMD_SET_REG,
7507                 .doit = nl80211_set_reg,
7508                 .policy = nl80211_policy,
7509                 .flags = GENL_ADMIN_PERM,
7510         },
7511         {
7512                 .cmd = NL80211_CMD_REQ_SET_REG,
7513                 .doit = nl80211_req_set_reg,
7514                 .policy = nl80211_policy,
7515                 .flags = GENL_ADMIN_PERM,
7516         },
7517         {
7518                 .cmd = NL80211_CMD_GET_MESH_CONFIG,
7519                 .doit = nl80211_get_mesh_config,
7520                 .policy = nl80211_policy,
7521                 /* can be retrieved by unprivileged users */
7522                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7523                                   NL80211_FLAG_NEED_RTNL,
7524         },
7525         {
7526                 .cmd = NL80211_CMD_SET_MESH_CONFIG,
7527                 .doit = nl80211_update_mesh_config,
7528                 .policy = nl80211_policy,
7529                 .flags = GENL_ADMIN_PERM,
7530                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7531                                   NL80211_FLAG_NEED_RTNL,
7532         },
7533         {
7534                 .cmd = NL80211_CMD_TRIGGER_SCAN,
7535                 .doit = nl80211_trigger_scan,
7536                 .policy = nl80211_policy,
7537                 .flags = GENL_ADMIN_PERM,
7538                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
7539                                   NL80211_FLAG_NEED_RTNL,
7540         },
7541         {
7542                 .cmd = NL80211_CMD_GET_SCAN,
7543                 .policy = nl80211_policy,
7544                 .dumpit = nl80211_dump_scan,
7545         },
7546         {
7547                 .cmd = NL80211_CMD_START_SCHED_SCAN,
7548                 .doit = nl80211_start_sched_scan,
7549                 .policy = nl80211_policy,
7550                 .flags = GENL_ADMIN_PERM,
7551                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7552                                   NL80211_FLAG_NEED_RTNL,
7553         },
7554         {
7555                 .cmd = NL80211_CMD_STOP_SCHED_SCAN,
7556                 .doit = nl80211_stop_sched_scan,
7557                 .policy = nl80211_policy,
7558                 .flags = GENL_ADMIN_PERM,
7559                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7560                                   NL80211_FLAG_NEED_RTNL,
7561         },
7562         {
7563                 .cmd = NL80211_CMD_AUTHENTICATE,
7564                 .doit = nl80211_authenticate,
7565                 .policy = nl80211_policy,
7566                 .flags = GENL_ADMIN_PERM,
7567                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7568                                   NL80211_FLAG_NEED_RTNL,
7569         },
7570         {
7571                 .cmd = NL80211_CMD_ASSOCIATE,
7572                 .doit = nl80211_associate,
7573                 .policy = nl80211_policy,
7574                 .flags = GENL_ADMIN_PERM,
7575                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7576                                   NL80211_FLAG_NEED_RTNL,
7577         },
7578         {
7579                 .cmd = NL80211_CMD_DEAUTHENTICATE,
7580                 .doit = nl80211_deauthenticate,
7581                 .policy = nl80211_policy,
7582                 .flags = GENL_ADMIN_PERM,
7583                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7584                                   NL80211_FLAG_NEED_RTNL,
7585         },
7586         {
7587                 .cmd = NL80211_CMD_DISASSOCIATE,
7588                 .doit = nl80211_disassociate,
7589                 .policy = nl80211_policy,
7590                 .flags = GENL_ADMIN_PERM,
7591                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7592                                   NL80211_FLAG_NEED_RTNL,
7593         },
7594         {
7595                 .cmd = NL80211_CMD_JOIN_IBSS,
7596                 .doit = nl80211_join_ibss,
7597                 .policy = nl80211_policy,
7598                 .flags = GENL_ADMIN_PERM,
7599                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7600                                   NL80211_FLAG_NEED_RTNL,
7601         },
7602         {
7603                 .cmd = NL80211_CMD_LEAVE_IBSS,
7604                 .doit = nl80211_leave_ibss,
7605                 .policy = nl80211_policy,
7606                 .flags = GENL_ADMIN_PERM,
7607                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7608                                   NL80211_FLAG_NEED_RTNL,
7609         },
7610 #ifdef CONFIG_NL80211_TESTMODE
7611         {
7612                 .cmd = NL80211_CMD_TESTMODE,
7613                 .doit = nl80211_testmode_do,
7614                 .dumpit = nl80211_testmode_dump,
7615                 .policy = nl80211_policy,
7616                 .flags = GENL_ADMIN_PERM,
7617                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
7618                                   NL80211_FLAG_NEED_RTNL,
7619         },
7620 #endif
7621         {
7622                 .cmd = NL80211_CMD_CONNECT,
7623                 .doit = nl80211_connect,
7624                 .policy = nl80211_policy,
7625                 .flags = GENL_ADMIN_PERM,
7626                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7627                                   NL80211_FLAG_NEED_RTNL,
7628         },
7629         {
7630                 .cmd = NL80211_CMD_DISCONNECT,
7631                 .doit = nl80211_disconnect,
7632                 .policy = nl80211_policy,
7633                 .flags = GENL_ADMIN_PERM,
7634                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7635                                   NL80211_FLAG_NEED_RTNL,
7636         },
7637         {
7638                 .cmd = NL80211_CMD_SET_WIPHY_NETNS,
7639                 .doit = nl80211_wiphy_netns,
7640                 .policy = nl80211_policy,
7641                 .flags = GENL_ADMIN_PERM,
7642                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
7643                                   NL80211_FLAG_NEED_RTNL,
7644         },
7645         {
7646                 .cmd = NL80211_CMD_GET_SURVEY,
7647                 .policy = nl80211_policy,
7648                 .dumpit = nl80211_dump_survey,
7649         },
7650         {
7651                 .cmd = NL80211_CMD_SET_PMKSA,
7652                 .doit = nl80211_setdel_pmksa,
7653                 .policy = nl80211_policy,
7654                 .flags = GENL_ADMIN_PERM,
7655                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7656                                   NL80211_FLAG_NEED_RTNL,
7657         },
7658         {
7659                 .cmd = NL80211_CMD_DEL_PMKSA,
7660                 .doit = nl80211_setdel_pmksa,
7661                 .policy = nl80211_policy,
7662                 .flags = GENL_ADMIN_PERM,
7663                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7664                                   NL80211_FLAG_NEED_RTNL,
7665         },
7666         {
7667                 .cmd = NL80211_CMD_FLUSH_PMKSA,
7668                 .doit = nl80211_flush_pmksa,
7669                 .policy = nl80211_policy,
7670                 .flags = GENL_ADMIN_PERM,
7671                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7672                                   NL80211_FLAG_NEED_RTNL,
7673         },
7674         {
7675                 .cmd = NL80211_CMD_REMAIN_ON_CHANNEL,
7676                 .doit = nl80211_remain_on_channel,
7677                 .policy = nl80211_policy,
7678                 .flags = GENL_ADMIN_PERM,
7679                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
7680                                   NL80211_FLAG_NEED_RTNL,
7681         },
7682         {
7683                 .cmd = NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL,
7684                 .doit = nl80211_cancel_remain_on_channel,
7685                 .policy = nl80211_policy,
7686                 .flags = GENL_ADMIN_PERM,
7687                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
7688                                   NL80211_FLAG_NEED_RTNL,
7689         },
7690         {
7691                 .cmd = NL80211_CMD_SET_TX_BITRATE_MASK,
7692                 .doit = nl80211_set_tx_bitrate_mask,
7693                 .policy = nl80211_policy,
7694                 .flags = GENL_ADMIN_PERM,
7695                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7696                                   NL80211_FLAG_NEED_RTNL,
7697         },
7698         {
7699                 .cmd = NL80211_CMD_REGISTER_FRAME,
7700                 .doit = nl80211_register_mgmt,
7701                 .policy = nl80211_policy,
7702                 .flags = GENL_ADMIN_PERM,
7703                 .internal_flags = NL80211_FLAG_NEED_WDEV |
7704                                   NL80211_FLAG_NEED_RTNL,
7705         },
7706         {
7707                 .cmd = NL80211_CMD_FRAME,
7708                 .doit = nl80211_tx_mgmt,
7709                 .policy = nl80211_policy,
7710                 .flags = GENL_ADMIN_PERM,
7711                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
7712                                   NL80211_FLAG_NEED_RTNL,
7713         },
7714         {
7715                 .cmd = NL80211_CMD_FRAME_WAIT_CANCEL,
7716                 .doit = nl80211_tx_mgmt_cancel_wait,
7717                 .policy = nl80211_policy,
7718                 .flags = GENL_ADMIN_PERM,
7719                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
7720                                   NL80211_FLAG_NEED_RTNL,
7721         },
7722         {
7723                 .cmd = NL80211_CMD_SET_POWER_SAVE,
7724                 .doit = nl80211_set_power_save,
7725                 .policy = nl80211_policy,
7726                 .flags = GENL_ADMIN_PERM,
7727                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7728                                   NL80211_FLAG_NEED_RTNL,
7729         },
7730         {
7731                 .cmd = NL80211_CMD_GET_POWER_SAVE,
7732                 .doit = nl80211_get_power_save,
7733                 .policy = nl80211_policy,
7734                 /* can be retrieved by unprivileged users */
7735                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7736                                   NL80211_FLAG_NEED_RTNL,
7737         },
7738         {
7739                 .cmd = NL80211_CMD_SET_CQM,
7740                 .doit = nl80211_set_cqm,
7741                 .policy = nl80211_policy,
7742                 .flags = GENL_ADMIN_PERM,
7743                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7744                                   NL80211_FLAG_NEED_RTNL,
7745         },
7746         {
7747                 .cmd = NL80211_CMD_SET_CHANNEL,
7748                 .doit = nl80211_set_channel,
7749                 .policy = nl80211_policy,
7750                 .flags = GENL_ADMIN_PERM,
7751                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7752                                   NL80211_FLAG_NEED_RTNL,
7753         },
7754         {
7755                 .cmd = NL80211_CMD_SET_WDS_PEER,
7756                 .doit = nl80211_set_wds_peer,
7757                 .policy = nl80211_policy,
7758                 .flags = GENL_ADMIN_PERM,
7759                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7760                                   NL80211_FLAG_NEED_RTNL,
7761         },
7762         {
7763                 .cmd = NL80211_CMD_JOIN_MESH,
7764                 .doit = nl80211_join_mesh,
7765                 .policy = nl80211_policy,
7766                 .flags = GENL_ADMIN_PERM,
7767                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7768                                   NL80211_FLAG_NEED_RTNL,
7769         },
7770         {
7771                 .cmd = NL80211_CMD_LEAVE_MESH,
7772                 .doit = nl80211_leave_mesh,
7773                 .policy = nl80211_policy,
7774                 .flags = GENL_ADMIN_PERM,
7775                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7776                                   NL80211_FLAG_NEED_RTNL,
7777         },
7778 #ifdef CONFIG_PM
7779         {
7780                 .cmd = NL80211_CMD_GET_WOWLAN,
7781                 .doit = nl80211_get_wowlan,
7782                 .policy = nl80211_policy,
7783                 /* can be retrieved by unprivileged users */
7784                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
7785                                   NL80211_FLAG_NEED_RTNL,
7786         },
7787         {
7788                 .cmd = NL80211_CMD_SET_WOWLAN,
7789                 .doit = nl80211_set_wowlan,
7790                 .policy = nl80211_policy,
7791                 .flags = GENL_ADMIN_PERM,
7792                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
7793                                   NL80211_FLAG_NEED_RTNL,
7794         },
7795 #endif
7796         {
7797                 .cmd = NL80211_CMD_SET_REKEY_OFFLOAD,
7798                 .doit = nl80211_set_rekey_data,
7799                 .policy = nl80211_policy,
7800                 .flags = GENL_ADMIN_PERM,
7801                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7802                                   NL80211_FLAG_NEED_RTNL,
7803         },
7804         {
7805                 .cmd = NL80211_CMD_TDLS_MGMT,
7806                 .doit = nl80211_tdls_mgmt,
7807                 .policy = nl80211_policy,
7808                 .flags = GENL_ADMIN_PERM,
7809                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7810                                   NL80211_FLAG_NEED_RTNL,
7811         },
7812         {
7813                 .cmd = NL80211_CMD_TDLS_OPER,
7814                 .doit = nl80211_tdls_oper,
7815                 .policy = nl80211_policy,
7816                 .flags = GENL_ADMIN_PERM,
7817                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7818                                   NL80211_FLAG_NEED_RTNL,
7819         },
7820         {
7821                 .cmd = NL80211_CMD_UNEXPECTED_FRAME,
7822                 .doit = nl80211_register_unexpected_frame,
7823                 .policy = nl80211_policy,
7824                 .flags = GENL_ADMIN_PERM,
7825                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7826                                   NL80211_FLAG_NEED_RTNL,
7827         },
7828         {
7829                 .cmd = NL80211_CMD_PROBE_CLIENT,
7830                 .doit = nl80211_probe_client,
7831                 .policy = nl80211_policy,
7832                 .flags = GENL_ADMIN_PERM,
7833                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
7834                                   NL80211_FLAG_NEED_RTNL,
7835         },
7836         {
7837                 .cmd = NL80211_CMD_REGISTER_BEACONS,
7838                 .doit = nl80211_register_beacons,
7839                 .policy = nl80211_policy,
7840                 .flags = GENL_ADMIN_PERM,
7841                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
7842                                   NL80211_FLAG_NEED_RTNL,
7843         },
7844         {
7845                 .cmd = NL80211_CMD_SET_NOACK_MAP,
7846                 .doit = nl80211_set_noack_map,
7847                 .policy = nl80211_policy,
7848                 .flags = GENL_ADMIN_PERM,
7849                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7850                                   NL80211_FLAG_NEED_RTNL,
7851         },
7852         {
7853                 .cmd = NL80211_CMD_START_P2P_DEVICE,
7854                 .doit = nl80211_start_p2p_device,
7855                 .policy = nl80211_policy,
7856                 .flags = GENL_ADMIN_PERM,
7857                 .internal_flags = NL80211_FLAG_NEED_WDEV |
7858                                   NL80211_FLAG_NEED_RTNL,
7859         },
7860         {
7861                 .cmd = NL80211_CMD_STOP_P2P_DEVICE,
7862                 .doit = nl80211_stop_p2p_device,
7863                 .policy = nl80211_policy,
7864                 .flags = GENL_ADMIN_PERM,
7865                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
7866                                   NL80211_FLAG_NEED_RTNL,
7867         },
7868         {
7869                 .cmd = NL80211_CMD_SET_MCAST_RATE,
7870                 .doit = nl80211_set_mcast_rate,
7871                 .policy = nl80211_policy,
7872                 .flags = GENL_ADMIN_PERM,
7873                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
7874                                   NL80211_FLAG_NEED_RTNL,
7875         },
7876 };
7877
7878 static struct genl_multicast_group nl80211_mlme_mcgrp = {
7879         .name = "mlme",
7880 };
7881
7882 /* multicast groups */
7883 static struct genl_multicast_group nl80211_config_mcgrp = {
7884         .name = "config",
7885 };
7886 static struct genl_multicast_group nl80211_scan_mcgrp = {
7887         .name = "scan",
7888 };
7889 static struct genl_multicast_group nl80211_regulatory_mcgrp = {
7890         .name = "regulatory",
7891 };
7892
7893 /* notification functions */
7894
7895 void nl80211_notify_dev_rename(struct cfg80211_registered_device *rdev)
7896 {
7897         struct sk_buff *msg;
7898
7899         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
7900         if (!msg)
7901                 return;
7902
7903         if (nl80211_send_wiphy(msg, 0, 0, 0, rdev) < 0) {
7904                 nlmsg_free(msg);
7905                 return;
7906         }
7907
7908         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
7909                                 nl80211_config_mcgrp.id, GFP_KERNEL);
7910 }
7911
7912 static int nl80211_add_scan_req(struct sk_buff *msg,
7913                                 struct cfg80211_registered_device *rdev)
7914 {
7915         struct cfg80211_scan_request *req = rdev->scan_req;
7916         struct nlattr *nest;
7917         int i;
7918
7919         ASSERT_RDEV_LOCK(rdev);
7920
7921         if (WARN_ON(!req))
7922                 return 0;
7923
7924         nest = nla_nest_start(msg, NL80211_ATTR_SCAN_SSIDS);
7925         if (!nest)
7926                 goto nla_put_failure;
7927         for (i = 0; i < req->n_ssids; i++) {
7928                 if (nla_put(msg, i, req->ssids[i].ssid_len, req->ssids[i].ssid))
7929                         goto nla_put_failure;
7930         }
7931         nla_nest_end(msg, nest);
7932
7933         nest = nla_nest_start(msg, NL80211_ATTR_SCAN_FREQUENCIES);
7934         if (!nest)
7935                 goto nla_put_failure;
7936         for (i = 0; i < req->n_channels; i++) {
7937                 if (nla_put_u32(msg, i, req->channels[i]->center_freq))
7938                         goto nla_put_failure;
7939         }
7940         nla_nest_end(msg, nest);
7941
7942         if (req->ie &&
7943             nla_put(msg, NL80211_ATTR_IE, req->ie_len, req->ie))
7944                 goto nla_put_failure;
7945
7946         if (req->flags)
7947                 nla_put_u32(msg, NL80211_ATTR_SCAN_FLAGS, req->flags);
7948
7949         return 0;
7950  nla_put_failure:
7951         return -ENOBUFS;
7952 }
7953
7954 static int nl80211_send_scan_msg(struct sk_buff *msg,
7955                                  struct cfg80211_registered_device *rdev,
7956                                  struct wireless_dev *wdev,
7957                                  u32 portid, u32 seq, int flags,
7958                                  u32 cmd)
7959 {
7960         void *hdr;
7961
7962         hdr = nl80211hdr_put(msg, portid, seq, flags, cmd);
7963         if (!hdr)
7964                 return -1;
7965
7966         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
7967             (wdev->netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
7968                                          wdev->netdev->ifindex)) ||
7969             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)))
7970                 goto nla_put_failure;
7971
7972         /* ignore errors and send incomplete event anyway */
7973         nl80211_add_scan_req(msg, rdev);
7974
7975         return genlmsg_end(msg, hdr);
7976
7977  nla_put_failure:
7978         genlmsg_cancel(msg, hdr);
7979         return -EMSGSIZE;
7980 }
7981
7982 static int
7983 nl80211_send_sched_scan_msg(struct sk_buff *msg,
7984                             struct cfg80211_registered_device *rdev,
7985                             struct net_device *netdev,
7986                             u32 portid, u32 seq, int flags, u32 cmd)
7987 {
7988         void *hdr;
7989
7990         hdr = nl80211hdr_put(msg, portid, seq, flags, cmd);
7991         if (!hdr)
7992                 return -1;
7993
7994         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
7995             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
7996                 goto nla_put_failure;
7997
7998         return genlmsg_end(msg, hdr);
7999
8000  nla_put_failure:
8001         genlmsg_cancel(msg, hdr);
8002         return -EMSGSIZE;
8003 }
8004
8005 void nl80211_send_scan_start(struct cfg80211_registered_device *rdev,
8006                              struct wireless_dev *wdev)
8007 {
8008         struct sk_buff *msg;
8009
8010         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8011         if (!msg)
8012                 return;
8013
8014         if (nl80211_send_scan_msg(msg, rdev, wdev, 0, 0, 0,
8015                                   NL80211_CMD_TRIGGER_SCAN) < 0) {
8016                 nlmsg_free(msg);
8017                 return;
8018         }
8019
8020         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8021                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
8022 }
8023
8024 void nl80211_send_scan_done(struct cfg80211_registered_device *rdev,
8025                             struct wireless_dev *wdev)
8026 {
8027         struct sk_buff *msg;
8028
8029         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8030         if (!msg)
8031                 return;
8032
8033         if (nl80211_send_scan_msg(msg, rdev, wdev, 0, 0, 0,
8034                                   NL80211_CMD_NEW_SCAN_RESULTS) < 0) {
8035                 nlmsg_free(msg);
8036                 return;
8037         }
8038
8039         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8040                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
8041 }
8042
8043 void nl80211_send_scan_aborted(struct cfg80211_registered_device *rdev,
8044                                struct wireless_dev *wdev)
8045 {
8046         struct sk_buff *msg;
8047
8048         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8049         if (!msg)
8050                 return;
8051
8052         if (nl80211_send_scan_msg(msg, rdev, wdev, 0, 0, 0,
8053                                   NL80211_CMD_SCAN_ABORTED) < 0) {
8054                 nlmsg_free(msg);
8055                 return;
8056         }
8057
8058         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8059                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
8060 }
8061
8062 void nl80211_send_sched_scan_results(struct cfg80211_registered_device *rdev,
8063                                      struct net_device *netdev)
8064 {
8065         struct sk_buff *msg;
8066
8067         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8068         if (!msg)
8069                 return;
8070
8071         if (nl80211_send_sched_scan_msg(msg, rdev, netdev, 0, 0, 0,
8072                                         NL80211_CMD_SCHED_SCAN_RESULTS) < 0) {
8073                 nlmsg_free(msg);
8074                 return;
8075         }
8076
8077         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8078                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
8079 }
8080
8081 void nl80211_send_sched_scan(struct cfg80211_registered_device *rdev,
8082                              struct net_device *netdev, u32 cmd)
8083 {
8084         struct sk_buff *msg;
8085
8086         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8087         if (!msg)
8088                 return;
8089
8090         if (nl80211_send_sched_scan_msg(msg, rdev, netdev, 0, 0, 0, cmd) < 0) {
8091                 nlmsg_free(msg);
8092                 return;
8093         }
8094
8095         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8096                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
8097 }
8098
8099 /*
8100  * This can happen on global regulatory changes or device specific settings
8101  * based on custom world regulatory domains.
8102  */
8103 void nl80211_send_reg_change_event(struct regulatory_request *request)
8104 {
8105         struct sk_buff *msg;
8106         void *hdr;
8107
8108         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8109         if (!msg)
8110                 return;
8111
8112         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_REG_CHANGE);
8113         if (!hdr) {
8114                 nlmsg_free(msg);
8115                 return;
8116         }
8117
8118         /* Userspace can always count this one always being set */
8119         if (nla_put_u8(msg, NL80211_ATTR_REG_INITIATOR, request->initiator))
8120                 goto nla_put_failure;
8121
8122         if (request->alpha2[0] == '0' && request->alpha2[1] == '0') {
8123                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
8124                                NL80211_REGDOM_TYPE_WORLD))
8125                         goto nla_put_failure;
8126         } else if (request->alpha2[0] == '9' && request->alpha2[1] == '9') {
8127                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
8128                                NL80211_REGDOM_TYPE_CUSTOM_WORLD))
8129                         goto nla_put_failure;
8130         } else if ((request->alpha2[0] == '9' && request->alpha2[1] == '8') ||
8131                    request->intersect) {
8132                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
8133                                NL80211_REGDOM_TYPE_INTERSECTION))
8134                         goto nla_put_failure;
8135         } else {
8136                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
8137                                NL80211_REGDOM_TYPE_COUNTRY) ||
8138                     nla_put_string(msg, NL80211_ATTR_REG_ALPHA2,
8139                                    request->alpha2))
8140                         goto nla_put_failure;
8141         }
8142
8143         if (request->wiphy_idx != WIPHY_IDX_INVALID &&
8144             nla_put_u32(msg, NL80211_ATTR_WIPHY, request->wiphy_idx))
8145                 goto nla_put_failure;
8146
8147         genlmsg_end(msg, hdr);
8148
8149         rcu_read_lock();
8150         genlmsg_multicast_allns(msg, 0, nl80211_regulatory_mcgrp.id,
8151                                 GFP_ATOMIC);
8152         rcu_read_unlock();
8153
8154         return;
8155
8156 nla_put_failure:
8157         genlmsg_cancel(msg, hdr);
8158         nlmsg_free(msg);
8159 }
8160
8161 static void nl80211_send_mlme_event(struct cfg80211_registered_device *rdev,
8162                                     struct net_device *netdev,
8163                                     const u8 *buf, size_t len,
8164                                     enum nl80211_commands cmd, gfp_t gfp)
8165 {
8166         struct sk_buff *msg;
8167         void *hdr;
8168
8169         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8170         if (!msg)
8171                 return;
8172
8173         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
8174         if (!hdr) {
8175                 nlmsg_free(msg);
8176                 return;
8177         }
8178
8179         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8180             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
8181             nla_put(msg, NL80211_ATTR_FRAME, len, buf))
8182                 goto nla_put_failure;
8183
8184         genlmsg_end(msg, hdr);
8185
8186         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8187                                 nl80211_mlme_mcgrp.id, gfp);
8188         return;
8189
8190  nla_put_failure:
8191         genlmsg_cancel(msg, hdr);
8192         nlmsg_free(msg);
8193 }
8194
8195 void nl80211_send_rx_auth(struct cfg80211_registered_device *rdev,
8196                           struct net_device *netdev, const u8 *buf,
8197                           size_t len, gfp_t gfp)
8198 {
8199         nl80211_send_mlme_event(rdev, netdev, buf, len,
8200                                 NL80211_CMD_AUTHENTICATE, gfp);
8201 }
8202
8203 void nl80211_send_rx_assoc(struct cfg80211_registered_device *rdev,
8204                            struct net_device *netdev, const u8 *buf,
8205                            size_t len, gfp_t gfp)
8206 {
8207         nl80211_send_mlme_event(rdev, netdev, buf, len,
8208                                 NL80211_CMD_ASSOCIATE, gfp);
8209 }
8210
8211 void nl80211_send_deauth(struct cfg80211_registered_device *rdev,
8212                          struct net_device *netdev, const u8 *buf,
8213                          size_t len, gfp_t gfp)
8214 {
8215         nl80211_send_mlme_event(rdev, netdev, buf, len,
8216                                 NL80211_CMD_DEAUTHENTICATE, gfp);
8217 }
8218
8219 void nl80211_send_disassoc(struct cfg80211_registered_device *rdev,
8220                            struct net_device *netdev, const u8 *buf,
8221                            size_t len, gfp_t gfp)
8222 {
8223         nl80211_send_mlme_event(rdev, netdev, buf, len,
8224                                 NL80211_CMD_DISASSOCIATE, gfp);
8225 }
8226
8227 void nl80211_send_unprot_deauth(struct cfg80211_registered_device *rdev,
8228                                 struct net_device *netdev, const u8 *buf,
8229                                 size_t len, gfp_t gfp)
8230 {
8231         nl80211_send_mlme_event(rdev, netdev, buf, len,
8232                                 NL80211_CMD_UNPROT_DEAUTHENTICATE, gfp);
8233 }
8234
8235 void nl80211_send_unprot_disassoc(struct cfg80211_registered_device *rdev,
8236                                   struct net_device *netdev, const u8 *buf,
8237                                   size_t len, gfp_t gfp)
8238 {
8239         nl80211_send_mlme_event(rdev, netdev, buf, len,
8240                                 NL80211_CMD_UNPROT_DISASSOCIATE, gfp);
8241 }
8242
8243 static void nl80211_send_mlme_timeout(struct cfg80211_registered_device *rdev,
8244                                       struct net_device *netdev, int cmd,
8245                                       const u8 *addr, gfp_t gfp)
8246 {
8247         struct sk_buff *msg;
8248         void *hdr;
8249
8250         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8251         if (!msg)
8252                 return;
8253
8254         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
8255         if (!hdr) {
8256                 nlmsg_free(msg);
8257                 return;
8258         }
8259
8260         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8261             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
8262             nla_put_flag(msg, NL80211_ATTR_TIMED_OUT) ||
8263             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr))
8264                 goto nla_put_failure;
8265
8266         genlmsg_end(msg, hdr);
8267
8268         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8269                                 nl80211_mlme_mcgrp.id, gfp);
8270         return;
8271
8272  nla_put_failure:
8273         genlmsg_cancel(msg, hdr);
8274         nlmsg_free(msg);
8275 }
8276
8277 void nl80211_send_auth_timeout(struct cfg80211_registered_device *rdev,
8278                                struct net_device *netdev, const u8 *addr,
8279                                gfp_t gfp)
8280 {
8281         nl80211_send_mlme_timeout(rdev, netdev, NL80211_CMD_AUTHENTICATE,
8282                                   addr, gfp);
8283 }
8284
8285 void nl80211_send_assoc_timeout(struct cfg80211_registered_device *rdev,
8286                                 struct net_device *netdev, const u8 *addr,
8287                                 gfp_t gfp)
8288 {
8289         nl80211_send_mlme_timeout(rdev, netdev, NL80211_CMD_ASSOCIATE,
8290                                   addr, gfp);
8291 }
8292
8293 void nl80211_send_connect_result(struct cfg80211_registered_device *rdev,
8294                                  struct net_device *netdev, const u8 *bssid,
8295                                  const u8 *req_ie, size_t req_ie_len,
8296                                  const u8 *resp_ie, size_t resp_ie_len,
8297                                  u16 status, gfp_t gfp)
8298 {
8299         struct sk_buff *msg;
8300         void *hdr;
8301
8302         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8303         if (!msg)
8304                 return;
8305
8306         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CONNECT);
8307         if (!hdr) {
8308                 nlmsg_free(msg);
8309                 return;
8310         }
8311
8312         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8313             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
8314             (bssid && nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid)) ||
8315             nla_put_u16(msg, NL80211_ATTR_STATUS_CODE, status) ||
8316             (req_ie &&
8317              nla_put(msg, NL80211_ATTR_REQ_IE, req_ie_len, req_ie)) ||
8318             (resp_ie &&
8319              nla_put(msg, NL80211_ATTR_RESP_IE, resp_ie_len, resp_ie)))
8320                 goto nla_put_failure;
8321
8322         genlmsg_end(msg, hdr);
8323
8324         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8325                                 nl80211_mlme_mcgrp.id, gfp);
8326         return;
8327
8328  nla_put_failure:
8329         genlmsg_cancel(msg, hdr);
8330         nlmsg_free(msg);
8331
8332 }
8333
8334 void nl80211_send_roamed(struct cfg80211_registered_device *rdev,
8335                          struct net_device *netdev, const u8 *bssid,
8336                          const u8 *req_ie, size_t req_ie_len,
8337                          const u8 *resp_ie, size_t resp_ie_len, gfp_t gfp)
8338 {
8339         struct sk_buff *msg;
8340         void *hdr;
8341
8342         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8343         if (!msg)
8344                 return;
8345
8346         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_ROAM);
8347         if (!hdr) {
8348                 nlmsg_free(msg);
8349                 return;
8350         }
8351
8352         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8353             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
8354             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid) ||
8355             (req_ie &&
8356              nla_put(msg, NL80211_ATTR_REQ_IE, req_ie_len, req_ie)) ||
8357             (resp_ie &&
8358              nla_put(msg, NL80211_ATTR_RESP_IE, resp_ie_len, resp_ie)))
8359                 goto nla_put_failure;
8360
8361         genlmsg_end(msg, hdr);
8362
8363         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8364                                 nl80211_mlme_mcgrp.id, gfp);
8365         return;
8366
8367  nla_put_failure:
8368         genlmsg_cancel(msg, hdr);
8369         nlmsg_free(msg);
8370
8371 }
8372
8373 void nl80211_send_disconnected(struct cfg80211_registered_device *rdev,
8374                                struct net_device *netdev, u16 reason,
8375                                const u8 *ie, size_t ie_len, bool from_ap)
8376 {
8377         struct sk_buff *msg;
8378         void *hdr;
8379
8380         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8381         if (!msg)
8382                 return;
8383
8384         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_DISCONNECT);
8385         if (!hdr) {
8386                 nlmsg_free(msg);
8387                 return;
8388         }
8389
8390         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8391             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
8392             (from_ap && reason &&
8393              nla_put_u16(msg, NL80211_ATTR_REASON_CODE, reason)) ||
8394             (from_ap &&
8395              nla_put_flag(msg, NL80211_ATTR_DISCONNECTED_BY_AP)) ||
8396             (ie && nla_put(msg, NL80211_ATTR_IE, ie_len, ie)))
8397                 goto nla_put_failure;
8398
8399         genlmsg_end(msg, hdr);
8400
8401         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8402                                 nl80211_mlme_mcgrp.id, GFP_KERNEL);
8403         return;
8404
8405  nla_put_failure:
8406         genlmsg_cancel(msg, hdr);
8407         nlmsg_free(msg);
8408
8409 }
8410
8411 void nl80211_send_ibss_bssid(struct cfg80211_registered_device *rdev,
8412                              struct net_device *netdev, const u8 *bssid,
8413                              gfp_t gfp)
8414 {
8415         struct sk_buff *msg;
8416         void *hdr;
8417
8418         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8419         if (!msg)
8420                 return;
8421
8422         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_JOIN_IBSS);
8423         if (!hdr) {
8424                 nlmsg_free(msg);
8425                 return;
8426         }
8427
8428         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8429             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
8430             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid))
8431                 goto nla_put_failure;
8432
8433         genlmsg_end(msg, hdr);
8434
8435         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8436                                 nl80211_mlme_mcgrp.id, gfp);
8437         return;
8438
8439  nla_put_failure:
8440         genlmsg_cancel(msg, hdr);
8441         nlmsg_free(msg);
8442 }
8443
8444 void nl80211_send_new_peer_candidate(struct cfg80211_registered_device *rdev,
8445                 struct net_device *netdev,
8446                 const u8 *macaddr, const u8* ie, u8 ie_len,
8447                 gfp_t gfp)
8448 {
8449         struct sk_buff *msg;
8450         void *hdr;
8451
8452         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8453         if (!msg)
8454                 return;
8455
8456         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NEW_PEER_CANDIDATE);
8457         if (!hdr) {
8458                 nlmsg_free(msg);
8459                 return;
8460         }
8461
8462         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8463             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
8464             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, macaddr) ||
8465             (ie_len && ie &&
8466              nla_put(msg, NL80211_ATTR_IE, ie_len , ie)))
8467                 goto nla_put_failure;
8468
8469         genlmsg_end(msg, hdr);
8470
8471         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8472                                 nl80211_mlme_mcgrp.id, gfp);
8473         return;
8474
8475  nla_put_failure:
8476         genlmsg_cancel(msg, hdr);
8477         nlmsg_free(msg);
8478 }
8479
8480 void nl80211_michael_mic_failure(struct cfg80211_registered_device *rdev,
8481                                  struct net_device *netdev, const u8 *addr,
8482                                  enum nl80211_key_type key_type, int key_id,
8483                                  const u8 *tsc, gfp_t gfp)
8484 {
8485         struct sk_buff *msg;
8486         void *hdr;
8487
8488         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8489         if (!msg)
8490                 return;
8491
8492         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_MICHAEL_MIC_FAILURE);
8493         if (!hdr) {
8494                 nlmsg_free(msg);
8495                 return;
8496         }
8497
8498         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8499             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
8500             (addr && nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr)) ||
8501             nla_put_u32(msg, NL80211_ATTR_KEY_TYPE, key_type) ||
8502             (key_id != -1 &&
8503              nla_put_u8(msg, NL80211_ATTR_KEY_IDX, key_id)) ||
8504             (tsc && nla_put(msg, NL80211_ATTR_KEY_SEQ, 6, tsc)))
8505                 goto nla_put_failure;
8506
8507         genlmsg_end(msg, hdr);
8508
8509         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8510                                 nl80211_mlme_mcgrp.id, gfp);
8511         return;
8512
8513  nla_put_failure:
8514         genlmsg_cancel(msg, hdr);
8515         nlmsg_free(msg);
8516 }
8517
8518 void nl80211_send_beacon_hint_event(struct wiphy *wiphy,
8519                                     struct ieee80211_channel *channel_before,
8520                                     struct ieee80211_channel *channel_after)
8521 {
8522         struct sk_buff *msg;
8523         void *hdr;
8524         struct nlattr *nl_freq;
8525
8526         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_ATOMIC);
8527         if (!msg)
8528                 return;
8529
8530         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_REG_BEACON_HINT);
8531         if (!hdr) {
8532                 nlmsg_free(msg);
8533                 return;
8534         }
8535
8536         /*
8537          * Since we are applying the beacon hint to a wiphy we know its
8538          * wiphy_idx is valid
8539          */
8540         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, get_wiphy_idx(wiphy)))
8541                 goto nla_put_failure;
8542
8543         /* Before */
8544         nl_freq = nla_nest_start(msg, NL80211_ATTR_FREQ_BEFORE);
8545         if (!nl_freq)
8546                 goto nla_put_failure;
8547         if (nl80211_msg_put_channel(msg, channel_before))
8548                 goto nla_put_failure;
8549         nla_nest_end(msg, nl_freq);
8550
8551         /* After */
8552         nl_freq = nla_nest_start(msg, NL80211_ATTR_FREQ_AFTER);
8553         if (!nl_freq)
8554                 goto nla_put_failure;
8555         if (nl80211_msg_put_channel(msg, channel_after))
8556                 goto nla_put_failure;
8557         nla_nest_end(msg, nl_freq);
8558
8559         genlmsg_end(msg, hdr);
8560
8561         rcu_read_lock();
8562         genlmsg_multicast_allns(msg, 0, nl80211_regulatory_mcgrp.id,
8563                                 GFP_ATOMIC);
8564         rcu_read_unlock();
8565
8566         return;
8567
8568 nla_put_failure:
8569         genlmsg_cancel(msg, hdr);
8570         nlmsg_free(msg);
8571 }
8572
8573 static void nl80211_send_remain_on_chan_event(
8574         int cmd, struct cfg80211_registered_device *rdev,
8575         struct wireless_dev *wdev, u64 cookie,
8576         struct ieee80211_channel *chan,
8577         unsigned int duration, gfp_t gfp)
8578 {
8579         struct sk_buff *msg;
8580         void *hdr;
8581
8582         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8583         if (!msg)
8584                 return;
8585
8586         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
8587         if (!hdr) {
8588                 nlmsg_free(msg);
8589                 return;
8590         }
8591
8592         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8593             (wdev->netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
8594                                          wdev->netdev->ifindex)) ||
8595             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)) ||
8596             nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ, chan->center_freq) ||
8597             nla_put_u32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
8598                         NL80211_CHAN_NO_HT) ||
8599             nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie))
8600                 goto nla_put_failure;
8601
8602         if (cmd == NL80211_CMD_REMAIN_ON_CHANNEL &&
8603             nla_put_u32(msg, NL80211_ATTR_DURATION, duration))
8604                 goto nla_put_failure;
8605
8606         genlmsg_end(msg, hdr);
8607
8608         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8609                                 nl80211_mlme_mcgrp.id, gfp);
8610         return;
8611
8612  nla_put_failure:
8613         genlmsg_cancel(msg, hdr);
8614         nlmsg_free(msg);
8615 }
8616
8617 void nl80211_send_remain_on_channel(struct cfg80211_registered_device *rdev,
8618                                     struct wireless_dev *wdev, u64 cookie,
8619                                     struct ieee80211_channel *chan,
8620                                     unsigned int duration, gfp_t gfp)
8621 {
8622         nl80211_send_remain_on_chan_event(NL80211_CMD_REMAIN_ON_CHANNEL,
8623                                           rdev, wdev, cookie, chan,
8624                                           duration, gfp);
8625 }
8626
8627 void nl80211_send_remain_on_channel_cancel(
8628         struct cfg80211_registered_device *rdev,
8629         struct wireless_dev *wdev,
8630         u64 cookie, struct ieee80211_channel *chan, gfp_t gfp)
8631 {
8632         nl80211_send_remain_on_chan_event(NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL,
8633                                           rdev, wdev, cookie, chan, 0, gfp);
8634 }
8635
8636 void nl80211_send_sta_event(struct cfg80211_registered_device *rdev,
8637                             struct net_device *dev, const u8 *mac_addr,
8638                             struct station_info *sinfo, gfp_t gfp)
8639 {
8640         struct sk_buff *msg;
8641
8642         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8643         if (!msg)
8644                 return;
8645
8646         if (nl80211_send_station(msg, 0, 0, 0,
8647                                  rdev, dev, mac_addr, sinfo) < 0) {
8648                 nlmsg_free(msg);
8649                 return;
8650         }
8651
8652         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8653                                 nl80211_mlme_mcgrp.id, gfp);
8654 }
8655
8656 void nl80211_send_sta_del_event(struct cfg80211_registered_device *rdev,
8657                                 struct net_device *dev, const u8 *mac_addr,
8658                                 gfp_t gfp)
8659 {
8660         struct sk_buff *msg;
8661         void *hdr;
8662
8663         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8664         if (!msg)
8665                 return;
8666
8667         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_DEL_STATION);
8668         if (!hdr) {
8669                 nlmsg_free(msg);
8670                 return;
8671         }
8672
8673         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
8674             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr))
8675                 goto nla_put_failure;
8676
8677         genlmsg_end(msg, hdr);
8678
8679         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8680                                 nl80211_mlme_mcgrp.id, gfp);
8681         return;
8682
8683  nla_put_failure:
8684         genlmsg_cancel(msg, hdr);
8685         nlmsg_free(msg);
8686 }
8687
8688 void nl80211_send_conn_failed_event(struct cfg80211_registered_device *rdev,
8689                                     struct net_device *dev, const u8 *mac_addr,
8690                                     enum nl80211_connect_failed_reason reason,
8691                                     gfp_t gfp)
8692 {
8693         struct sk_buff *msg;
8694         void *hdr;
8695
8696         msg = nlmsg_new(NLMSG_GOODSIZE, gfp);
8697         if (!msg)
8698                 return;
8699
8700         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CONN_FAILED);
8701         if (!hdr) {
8702                 nlmsg_free(msg);
8703                 return;
8704         }
8705
8706         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
8707             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr) ||
8708             nla_put_u32(msg, NL80211_ATTR_CONN_FAILED_REASON, reason))
8709                 goto nla_put_failure;
8710
8711         genlmsg_end(msg, hdr);
8712
8713         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8714                                 nl80211_mlme_mcgrp.id, gfp);
8715         return;
8716
8717  nla_put_failure:
8718         genlmsg_cancel(msg, hdr);
8719         nlmsg_free(msg);
8720 }
8721
8722 static bool __nl80211_unexpected_frame(struct net_device *dev, u8 cmd,
8723                                        const u8 *addr, gfp_t gfp)
8724 {
8725         struct wireless_dev *wdev = dev->ieee80211_ptr;
8726         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
8727         struct sk_buff *msg;
8728         void *hdr;
8729         int err;
8730         u32 nlportid = ACCESS_ONCE(wdev->ap_unexpected_nlportid);
8731
8732         if (!nlportid)
8733                 return false;
8734
8735         msg = nlmsg_new(100, gfp);
8736         if (!msg)
8737                 return true;
8738
8739         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
8740         if (!hdr) {
8741                 nlmsg_free(msg);
8742                 return true;
8743         }
8744
8745         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8746             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
8747             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr))
8748                 goto nla_put_failure;
8749
8750         err = genlmsg_end(msg, hdr);
8751         if (err < 0) {
8752                 nlmsg_free(msg);
8753                 return true;
8754         }
8755
8756         genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, nlportid);
8757         return true;
8758
8759  nla_put_failure:
8760         genlmsg_cancel(msg, hdr);
8761         nlmsg_free(msg);
8762         return true;
8763 }
8764
8765 bool nl80211_unexpected_frame(struct net_device *dev, const u8 *addr, gfp_t gfp)
8766 {
8767         return __nl80211_unexpected_frame(dev, NL80211_CMD_UNEXPECTED_FRAME,
8768                                           addr, gfp);
8769 }
8770
8771 bool nl80211_unexpected_4addr_frame(struct net_device *dev,
8772                                     const u8 *addr, gfp_t gfp)
8773 {
8774         return __nl80211_unexpected_frame(dev,
8775                                           NL80211_CMD_UNEXPECTED_4ADDR_FRAME,
8776                                           addr, gfp);
8777 }
8778
8779 int nl80211_send_mgmt(struct cfg80211_registered_device *rdev,
8780                       struct wireless_dev *wdev, u32 nlportid,
8781                       int freq, int sig_dbm,
8782                       const u8 *buf, size_t len, gfp_t gfp)
8783 {
8784         struct net_device *netdev = wdev->netdev;
8785         struct sk_buff *msg;
8786         void *hdr;
8787
8788         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8789         if (!msg)
8790                 return -ENOMEM;
8791
8792         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FRAME);
8793         if (!hdr) {
8794                 nlmsg_free(msg);
8795                 return -ENOMEM;
8796         }
8797
8798         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8799             (netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
8800                                         netdev->ifindex)) ||
8801             nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ, freq) ||
8802             (sig_dbm &&
8803              nla_put_u32(msg, NL80211_ATTR_RX_SIGNAL_DBM, sig_dbm)) ||
8804             nla_put(msg, NL80211_ATTR_FRAME, len, buf))
8805                 goto nla_put_failure;
8806
8807         genlmsg_end(msg, hdr);
8808
8809         return genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, nlportid);
8810
8811  nla_put_failure:
8812         genlmsg_cancel(msg, hdr);
8813         nlmsg_free(msg);
8814         return -ENOBUFS;
8815 }
8816
8817 void nl80211_send_mgmt_tx_status(struct cfg80211_registered_device *rdev,
8818                                  struct wireless_dev *wdev, u64 cookie,
8819                                  const u8 *buf, size_t len, bool ack,
8820                                  gfp_t gfp)
8821 {
8822         struct net_device *netdev = wdev->netdev;
8823         struct sk_buff *msg;
8824         void *hdr;
8825
8826         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8827         if (!msg)
8828                 return;
8829
8830         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FRAME_TX_STATUS);
8831         if (!hdr) {
8832                 nlmsg_free(msg);
8833                 return;
8834         }
8835
8836         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8837             (netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
8838                                    netdev->ifindex)) ||
8839             nla_put(msg, NL80211_ATTR_FRAME, len, buf) ||
8840             nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie) ||
8841             (ack && nla_put_flag(msg, NL80211_ATTR_ACK)))
8842                 goto nla_put_failure;
8843
8844         genlmsg_end(msg, hdr);
8845
8846         genlmsg_multicast(msg, 0, nl80211_mlme_mcgrp.id, gfp);
8847         return;
8848
8849  nla_put_failure:
8850         genlmsg_cancel(msg, hdr);
8851         nlmsg_free(msg);
8852 }
8853
8854 void
8855 nl80211_send_cqm_rssi_notify(struct cfg80211_registered_device *rdev,
8856                              struct net_device *netdev,
8857                              enum nl80211_cqm_rssi_threshold_event rssi_event,
8858                              gfp_t gfp)
8859 {
8860         struct sk_buff *msg;
8861         struct nlattr *pinfoattr;
8862         void *hdr;
8863
8864         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8865         if (!msg)
8866                 return;
8867
8868         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NOTIFY_CQM);
8869         if (!hdr) {
8870                 nlmsg_free(msg);
8871                 return;
8872         }
8873
8874         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8875             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
8876                 goto nla_put_failure;
8877
8878         pinfoattr = nla_nest_start(msg, NL80211_ATTR_CQM);
8879         if (!pinfoattr)
8880                 goto nla_put_failure;
8881
8882         if (nla_put_u32(msg, NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT,
8883                         rssi_event))
8884                 goto nla_put_failure;
8885
8886         nla_nest_end(msg, pinfoattr);
8887
8888         genlmsg_end(msg, hdr);
8889
8890         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8891                                 nl80211_mlme_mcgrp.id, gfp);
8892         return;
8893
8894  nla_put_failure:
8895         genlmsg_cancel(msg, hdr);
8896         nlmsg_free(msg);
8897 }
8898
8899 void nl80211_gtk_rekey_notify(struct cfg80211_registered_device *rdev,
8900                               struct net_device *netdev, const u8 *bssid,
8901                               const u8 *replay_ctr, gfp_t gfp)
8902 {
8903         struct sk_buff *msg;
8904         struct nlattr *rekey_attr;
8905         void *hdr;
8906
8907         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8908         if (!msg)
8909                 return;
8910
8911         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_SET_REKEY_OFFLOAD);
8912         if (!hdr) {
8913                 nlmsg_free(msg);
8914                 return;
8915         }
8916
8917         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8918             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
8919             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid))
8920                 goto nla_put_failure;
8921
8922         rekey_attr = nla_nest_start(msg, NL80211_ATTR_REKEY_DATA);
8923         if (!rekey_attr)
8924                 goto nla_put_failure;
8925
8926         if (nla_put(msg, NL80211_REKEY_DATA_REPLAY_CTR,
8927                     NL80211_REPLAY_CTR_LEN, replay_ctr))
8928                 goto nla_put_failure;
8929
8930         nla_nest_end(msg, rekey_attr);
8931
8932         genlmsg_end(msg, hdr);
8933
8934         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8935                                 nl80211_mlme_mcgrp.id, gfp);
8936         return;
8937
8938  nla_put_failure:
8939         genlmsg_cancel(msg, hdr);
8940         nlmsg_free(msg);
8941 }
8942
8943 void nl80211_pmksa_candidate_notify(struct cfg80211_registered_device *rdev,
8944                                     struct net_device *netdev, int index,
8945                                     const u8 *bssid, bool preauth, gfp_t gfp)
8946 {
8947         struct sk_buff *msg;
8948         struct nlattr *attr;
8949         void *hdr;
8950
8951         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8952         if (!msg)
8953                 return;
8954
8955         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_PMKSA_CANDIDATE);
8956         if (!hdr) {
8957                 nlmsg_free(msg);
8958                 return;
8959         }
8960
8961         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
8962             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
8963                 goto nla_put_failure;
8964
8965         attr = nla_nest_start(msg, NL80211_ATTR_PMKSA_CANDIDATE);
8966         if (!attr)
8967                 goto nla_put_failure;
8968
8969         if (nla_put_u32(msg, NL80211_PMKSA_CANDIDATE_INDEX, index) ||
8970             nla_put(msg, NL80211_PMKSA_CANDIDATE_BSSID, ETH_ALEN, bssid) ||
8971             (preauth &&
8972              nla_put_flag(msg, NL80211_PMKSA_CANDIDATE_PREAUTH)))
8973                 goto nla_put_failure;
8974
8975         nla_nest_end(msg, attr);
8976
8977         genlmsg_end(msg, hdr);
8978
8979         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
8980                                 nl80211_mlme_mcgrp.id, gfp);
8981         return;
8982
8983  nla_put_failure:
8984         genlmsg_cancel(msg, hdr);
8985         nlmsg_free(msg);
8986 }
8987
8988 void nl80211_ch_switch_notify(struct cfg80211_registered_device *rdev,
8989                               struct net_device *netdev,
8990                               struct cfg80211_chan_def *chandef, gfp_t gfp)
8991 {
8992         struct sk_buff *msg;
8993         void *hdr;
8994
8995         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
8996         if (!msg)
8997                 return;
8998
8999         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CH_SWITCH_NOTIFY);
9000         if (!hdr) {
9001                 nlmsg_free(msg);
9002                 return;
9003         }
9004
9005         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
9006                 goto nla_put_failure;
9007
9008         if (nl80211_send_chandef(msg, chandef))
9009                 goto nla_put_failure;
9010
9011         genlmsg_end(msg, hdr);
9012
9013         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9014                                 nl80211_mlme_mcgrp.id, gfp);
9015         return;
9016
9017  nla_put_failure:
9018         genlmsg_cancel(msg, hdr);
9019         nlmsg_free(msg);
9020 }
9021
9022 void
9023 nl80211_send_cqm_txe_notify(struct cfg80211_registered_device *rdev,
9024                             struct net_device *netdev, const u8 *peer,
9025                             u32 num_packets, u32 rate, u32 intvl, gfp_t gfp)
9026 {
9027         struct sk_buff *msg;
9028         struct nlattr *pinfoattr;
9029         void *hdr;
9030
9031         msg = nlmsg_new(NLMSG_GOODSIZE, gfp);
9032         if (!msg)
9033                 return;
9034
9035         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NOTIFY_CQM);
9036         if (!hdr) {
9037                 nlmsg_free(msg);
9038                 return;
9039         }
9040
9041         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
9042             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
9043             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, peer))
9044                 goto nla_put_failure;
9045
9046         pinfoattr = nla_nest_start(msg, NL80211_ATTR_CQM);
9047         if (!pinfoattr)
9048                 goto nla_put_failure;
9049
9050         if (nla_put_u32(msg, NL80211_ATTR_CQM_TXE_PKTS, num_packets))
9051                 goto nla_put_failure;
9052
9053         if (nla_put_u32(msg, NL80211_ATTR_CQM_TXE_RATE, rate))
9054                 goto nla_put_failure;
9055
9056         if (nla_put_u32(msg, NL80211_ATTR_CQM_TXE_INTVL, intvl))
9057                 goto nla_put_failure;
9058
9059         nla_nest_end(msg, pinfoattr);
9060
9061         genlmsg_end(msg, hdr);
9062
9063         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9064                                 nl80211_mlme_mcgrp.id, gfp);
9065         return;
9066
9067  nla_put_failure:
9068         genlmsg_cancel(msg, hdr);
9069         nlmsg_free(msg);
9070 }
9071
9072 void
9073 nl80211_send_cqm_pktloss_notify(struct cfg80211_registered_device *rdev,
9074                                 struct net_device *netdev, const u8 *peer,
9075                                 u32 num_packets, gfp_t gfp)
9076 {
9077         struct sk_buff *msg;
9078         struct nlattr *pinfoattr;
9079         void *hdr;
9080
9081         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
9082         if (!msg)
9083                 return;
9084
9085         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NOTIFY_CQM);
9086         if (!hdr) {
9087                 nlmsg_free(msg);
9088                 return;
9089         }
9090
9091         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
9092             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
9093             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, peer))
9094                 goto nla_put_failure;
9095
9096         pinfoattr = nla_nest_start(msg, NL80211_ATTR_CQM);
9097         if (!pinfoattr)
9098                 goto nla_put_failure;
9099
9100         if (nla_put_u32(msg, NL80211_ATTR_CQM_PKT_LOSS_EVENT, num_packets))
9101                 goto nla_put_failure;
9102
9103         nla_nest_end(msg, pinfoattr);
9104
9105         genlmsg_end(msg, hdr);
9106
9107         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9108                                 nl80211_mlme_mcgrp.id, gfp);
9109         return;
9110
9111  nla_put_failure:
9112         genlmsg_cancel(msg, hdr);
9113         nlmsg_free(msg);
9114 }
9115
9116 void cfg80211_probe_status(struct net_device *dev, const u8 *addr,
9117                            u64 cookie, bool acked, gfp_t gfp)
9118 {
9119         struct wireless_dev *wdev = dev->ieee80211_ptr;
9120         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
9121         struct sk_buff *msg;
9122         void *hdr;
9123         int err;
9124
9125         trace_cfg80211_probe_status(dev, addr, cookie, acked);
9126
9127         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
9128
9129         if (!msg)
9130                 return;
9131
9132         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_PROBE_CLIENT);
9133         if (!hdr) {
9134                 nlmsg_free(msg);
9135                 return;
9136         }
9137
9138         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
9139             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
9140             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr) ||
9141             nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie) ||
9142             (acked && nla_put_flag(msg, NL80211_ATTR_ACK)))
9143                 goto nla_put_failure;
9144
9145         err = genlmsg_end(msg, hdr);
9146         if (err < 0) {
9147                 nlmsg_free(msg);
9148                 return;
9149         }
9150
9151         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9152                                 nl80211_mlme_mcgrp.id, gfp);
9153         return;
9154
9155  nla_put_failure:
9156         genlmsg_cancel(msg, hdr);
9157         nlmsg_free(msg);
9158 }
9159 EXPORT_SYMBOL(cfg80211_probe_status);
9160
9161 void cfg80211_report_obss_beacon(struct wiphy *wiphy,
9162                                  const u8 *frame, size_t len,
9163                                  int freq, int sig_dbm)
9164 {
9165         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
9166         struct sk_buff *msg;
9167         void *hdr;
9168         struct cfg80211_beacon_registration *reg;
9169
9170         trace_cfg80211_report_obss_beacon(wiphy, frame, len, freq, sig_dbm);
9171
9172         spin_lock_bh(&rdev->beacon_registrations_lock);
9173         list_for_each_entry(reg, &rdev->beacon_registrations, list) {
9174                 msg = nlmsg_new(len + 100, GFP_ATOMIC);
9175                 if (!msg) {
9176                         spin_unlock_bh(&rdev->beacon_registrations_lock);
9177                         return;
9178                 }
9179
9180                 hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FRAME);
9181                 if (!hdr)
9182                         goto nla_put_failure;
9183
9184                 if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
9185                     (freq &&
9186                      nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ, freq)) ||
9187                     (sig_dbm &&
9188                      nla_put_u32(msg, NL80211_ATTR_RX_SIGNAL_DBM, sig_dbm)) ||
9189                     nla_put(msg, NL80211_ATTR_FRAME, len, frame))
9190                         goto nla_put_failure;
9191
9192                 genlmsg_end(msg, hdr);
9193
9194                 genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, reg->nlportid);
9195         }
9196         spin_unlock_bh(&rdev->beacon_registrations_lock);
9197         return;
9198
9199  nla_put_failure:
9200         spin_unlock_bh(&rdev->beacon_registrations_lock);
9201         if (hdr)
9202                 genlmsg_cancel(msg, hdr);
9203         nlmsg_free(msg);
9204 }
9205 EXPORT_SYMBOL(cfg80211_report_obss_beacon);
9206
9207 void cfg80211_tdls_oper_request(struct net_device *dev, const u8 *peer,
9208                                 enum nl80211_tdls_operation oper,
9209                                 u16 reason_code, gfp_t gfp)
9210 {
9211         struct wireless_dev *wdev = dev->ieee80211_ptr;
9212         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
9213         struct sk_buff *msg;
9214         void *hdr;
9215         int err;
9216
9217         trace_cfg80211_tdls_oper_request(wdev->wiphy, dev, peer, oper,
9218                                          reason_code);
9219
9220         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
9221         if (!msg)
9222                 return;
9223
9224         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_TDLS_OPER);
9225         if (!hdr) {
9226                 nlmsg_free(msg);
9227                 return;
9228         }
9229
9230         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
9231             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
9232             nla_put_u8(msg, NL80211_ATTR_TDLS_OPERATION, oper) ||
9233             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, peer) ||
9234             (reason_code > 0 &&
9235              nla_put_u16(msg, NL80211_ATTR_REASON_CODE, reason_code)))
9236                 goto nla_put_failure;
9237
9238         err = genlmsg_end(msg, hdr);
9239         if (err < 0) {
9240                 nlmsg_free(msg);
9241                 return;
9242         }
9243
9244         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9245                                 nl80211_mlme_mcgrp.id, gfp);
9246         return;
9247
9248  nla_put_failure:
9249         genlmsg_cancel(msg, hdr);
9250         nlmsg_free(msg);
9251 }
9252 EXPORT_SYMBOL(cfg80211_tdls_oper_request);
9253
9254 static int nl80211_netlink_notify(struct notifier_block * nb,
9255                                   unsigned long state,
9256                                   void *_notify)
9257 {
9258         struct netlink_notify *notify = _notify;
9259         struct cfg80211_registered_device *rdev;
9260         struct wireless_dev *wdev;
9261         struct cfg80211_beacon_registration *reg, *tmp;
9262
9263         if (state != NETLINK_URELEASE)
9264                 return NOTIFY_DONE;
9265
9266         rcu_read_lock();
9267
9268         list_for_each_entry_rcu(rdev, &cfg80211_rdev_list, list) {
9269                 list_for_each_entry_rcu(wdev, &rdev->wdev_list, list)
9270                         cfg80211_mlme_unregister_socket(wdev, notify->portid);
9271
9272                 spin_lock_bh(&rdev->beacon_registrations_lock);
9273                 list_for_each_entry_safe(reg, tmp, &rdev->beacon_registrations,
9274                                          list) {
9275                         if (reg->nlportid == notify->portid) {
9276                                 list_del(&reg->list);
9277                                 kfree(reg);
9278                                 break;
9279                         }
9280                 }
9281                 spin_unlock_bh(&rdev->beacon_registrations_lock);
9282         }
9283
9284         rcu_read_unlock();
9285
9286         return NOTIFY_DONE;
9287 }
9288
9289 static struct notifier_block nl80211_netlink_notifier = {
9290         .notifier_call = nl80211_netlink_notify,
9291 };
9292
9293 /* initialisation/exit functions */
9294
9295 int nl80211_init(void)
9296 {
9297         int err;
9298
9299         err = genl_register_family_with_ops(&nl80211_fam,
9300                 nl80211_ops, ARRAY_SIZE(nl80211_ops));
9301         if (err)
9302                 return err;
9303
9304         err = genl_register_mc_group(&nl80211_fam, &nl80211_config_mcgrp);
9305         if (err)
9306                 goto err_out;
9307
9308         err = genl_register_mc_group(&nl80211_fam, &nl80211_scan_mcgrp);
9309         if (err)
9310                 goto err_out;
9311
9312         err = genl_register_mc_group(&nl80211_fam, &nl80211_regulatory_mcgrp);
9313         if (err)
9314                 goto err_out;
9315
9316         err = genl_register_mc_group(&nl80211_fam, &nl80211_mlme_mcgrp);
9317         if (err)
9318                 goto err_out;
9319
9320 #ifdef CONFIG_NL80211_TESTMODE
9321         err = genl_register_mc_group(&nl80211_fam, &nl80211_testmode_mcgrp);
9322         if (err)
9323                 goto err_out;
9324 #endif
9325
9326         err = netlink_register_notifier(&nl80211_netlink_notifier);
9327         if (err)
9328                 goto err_out;
9329
9330         return 0;
9331  err_out:
9332         genl_unregister_family(&nl80211_fam);
9333         return err;
9334 }
9335
9336 void nl80211_exit(void)
9337 {
9338         netlink_unregister_notifier(&nl80211_netlink_notifier);
9339         genl_unregister_family(&nl80211_fam);
9340 }