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