a8c062dbd51d31f6ef7279bc0e6fb134dc8c3a4e
[cascardo/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  * Copyright 2013-2014  Intel Mobile Communications GmbH
6  * Copyright 2015-2016  Intel Deutschland GmbH
7  */
8
9 #include <linux/if.h>
10 #include <linux/module.h>
11 #include <linux/err.h>
12 #include <linux/slab.h>
13 #include <linux/list.h>
14 #include <linux/if_ether.h>
15 #include <linux/ieee80211.h>
16 #include <linux/nl80211.h>
17 #include <linux/rtnetlink.h>
18 #include <linux/netlink.h>
19 #include <linux/etherdevice.h>
20 #include <net/net_namespace.h>
21 #include <net/genetlink.h>
22 #include <net/cfg80211.h>
23 #include <net/sock.h>
24 #include <net/inet_connection_sock.h>
25 #include "core.h"
26 #include "nl80211.h"
27 #include "reg.h"
28 #include "rdev-ops.h"
29
30 static int nl80211_crypto_settings(struct cfg80211_registered_device *rdev,
31                                    struct genl_info *info,
32                                    struct cfg80211_crypto_settings *settings,
33                                    int cipher_limit);
34
35 static int nl80211_pre_doit(const struct genl_ops *ops, struct sk_buff *skb,
36                             struct genl_info *info);
37 static void nl80211_post_doit(const struct genl_ops *ops, struct sk_buff *skb,
38                               struct genl_info *info);
39
40 /* the netlink family */
41 static struct genl_family nl80211_fam = {
42         .id = GENL_ID_GENERATE,         /* don't bother with a hardcoded ID */
43         .name = NL80211_GENL_NAME,      /* have users key off the name instead */
44         .hdrsize = 0,                   /* no private header */
45         .version = 1,                   /* no particular meaning now */
46         .maxattr = NL80211_ATTR_MAX,
47         .netnsok = true,
48         .pre_doit = nl80211_pre_doit,
49         .post_doit = nl80211_post_doit,
50 };
51
52 /* multicast groups */
53 enum nl80211_multicast_groups {
54         NL80211_MCGRP_CONFIG,
55         NL80211_MCGRP_SCAN,
56         NL80211_MCGRP_REGULATORY,
57         NL80211_MCGRP_MLME,
58         NL80211_MCGRP_VENDOR,
59         NL80211_MCGRP_TESTMODE /* keep last - ifdef! */
60 };
61
62 static const struct genl_multicast_group nl80211_mcgrps[] = {
63         [NL80211_MCGRP_CONFIG] = { .name = NL80211_MULTICAST_GROUP_CONFIG },
64         [NL80211_MCGRP_SCAN] = { .name = NL80211_MULTICAST_GROUP_SCAN },
65         [NL80211_MCGRP_REGULATORY] = { .name = NL80211_MULTICAST_GROUP_REG },
66         [NL80211_MCGRP_MLME] = { .name = NL80211_MULTICAST_GROUP_MLME },
67         [NL80211_MCGRP_VENDOR] = { .name = NL80211_MULTICAST_GROUP_VENDOR },
68 #ifdef CONFIG_NL80211_TESTMODE
69         [NL80211_MCGRP_TESTMODE] = { .name = NL80211_MULTICAST_GROUP_TESTMODE }
70 #endif
71 };
72
73 /* returns ERR_PTR values */
74 static struct wireless_dev *
75 __cfg80211_wdev_from_attrs(struct net *netns, struct nlattr **attrs)
76 {
77         struct cfg80211_registered_device *rdev;
78         struct wireless_dev *result = NULL;
79         bool have_ifidx = attrs[NL80211_ATTR_IFINDEX];
80         bool have_wdev_id = attrs[NL80211_ATTR_WDEV];
81         u64 wdev_id;
82         int wiphy_idx = -1;
83         int ifidx = -1;
84
85         ASSERT_RTNL();
86
87         if (!have_ifidx && !have_wdev_id)
88                 return ERR_PTR(-EINVAL);
89
90         if (have_ifidx)
91                 ifidx = nla_get_u32(attrs[NL80211_ATTR_IFINDEX]);
92         if (have_wdev_id) {
93                 wdev_id = nla_get_u64(attrs[NL80211_ATTR_WDEV]);
94                 wiphy_idx = wdev_id >> 32;
95         }
96
97         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
98                 struct wireless_dev *wdev;
99
100                 if (wiphy_net(&rdev->wiphy) != netns)
101                         continue;
102
103                 if (have_wdev_id && rdev->wiphy_idx != wiphy_idx)
104                         continue;
105
106                 list_for_each_entry(wdev, &rdev->wiphy.wdev_list, list) {
107                         if (have_ifidx && wdev->netdev &&
108                             wdev->netdev->ifindex == ifidx) {
109                                 result = wdev;
110                                 break;
111                         }
112                         if (have_wdev_id && wdev->identifier == (u32)wdev_id) {
113                                 result = wdev;
114                                 break;
115                         }
116                 }
117
118                 if (result)
119                         break;
120         }
121
122         if (result)
123                 return result;
124         return ERR_PTR(-ENODEV);
125 }
126
127 static struct cfg80211_registered_device *
128 __cfg80211_rdev_from_attrs(struct net *netns, struct nlattr **attrs)
129 {
130         struct cfg80211_registered_device *rdev = NULL, *tmp;
131         struct net_device *netdev;
132
133         ASSERT_RTNL();
134
135         if (!attrs[NL80211_ATTR_WIPHY] &&
136             !attrs[NL80211_ATTR_IFINDEX] &&
137             !attrs[NL80211_ATTR_WDEV])
138                 return ERR_PTR(-EINVAL);
139
140         if (attrs[NL80211_ATTR_WIPHY])
141                 rdev = cfg80211_rdev_by_wiphy_idx(
142                                 nla_get_u32(attrs[NL80211_ATTR_WIPHY]));
143
144         if (attrs[NL80211_ATTR_WDEV]) {
145                 u64 wdev_id = nla_get_u64(attrs[NL80211_ATTR_WDEV]);
146                 struct wireless_dev *wdev;
147                 bool found = false;
148
149                 tmp = cfg80211_rdev_by_wiphy_idx(wdev_id >> 32);
150                 if (tmp) {
151                         /* make sure wdev exists */
152                         list_for_each_entry(wdev, &tmp->wiphy.wdev_list, list) {
153                                 if (wdev->identifier != (u32)wdev_id)
154                                         continue;
155                                 found = true;
156                                 break;
157                         }
158
159                         if (!found)
160                                 tmp = NULL;
161
162                         if (rdev && tmp != rdev)
163                                 return ERR_PTR(-EINVAL);
164                         rdev = tmp;
165                 }
166         }
167
168         if (attrs[NL80211_ATTR_IFINDEX]) {
169                 int ifindex = nla_get_u32(attrs[NL80211_ATTR_IFINDEX]);
170
171                 netdev = __dev_get_by_index(netns, ifindex);
172                 if (netdev) {
173                         if (netdev->ieee80211_ptr)
174                                 tmp = wiphy_to_rdev(
175                                         netdev->ieee80211_ptr->wiphy);
176                         else
177                                 tmp = NULL;
178
179                         /* not wireless device -- return error */
180                         if (!tmp)
181                                 return ERR_PTR(-EINVAL);
182
183                         /* mismatch -- return error */
184                         if (rdev && tmp != rdev)
185                                 return ERR_PTR(-EINVAL);
186
187                         rdev = tmp;
188                 }
189         }
190
191         if (!rdev)
192                 return ERR_PTR(-ENODEV);
193
194         if (netns != wiphy_net(&rdev->wiphy))
195                 return ERR_PTR(-ENODEV);
196
197         return rdev;
198 }
199
200 /*
201  * This function returns a pointer to the driver
202  * that the genl_info item that is passed refers to.
203  *
204  * The result of this can be a PTR_ERR and hence must
205  * be checked with IS_ERR() for errors.
206  */
207 static struct cfg80211_registered_device *
208 cfg80211_get_dev_from_info(struct net *netns, struct genl_info *info)
209 {
210         return __cfg80211_rdev_from_attrs(netns, info->attrs);
211 }
212
213 /* policy for the attributes */
214 static const struct nla_policy nl80211_policy[NUM_NL80211_ATTR] = {
215         [NL80211_ATTR_WIPHY] = { .type = NLA_U32 },
216         [NL80211_ATTR_WIPHY_NAME] = { .type = NLA_NUL_STRING,
217                                       .len = 20-1 },
218         [NL80211_ATTR_WIPHY_TXQ_PARAMS] = { .type = NLA_NESTED },
219
220         [NL80211_ATTR_WIPHY_FREQ] = { .type = NLA_U32 },
221         [NL80211_ATTR_WIPHY_CHANNEL_TYPE] = { .type = NLA_U32 },
222         [NL80211_ATTR_CHANNEL_WIDTH] = { .type = NLA_U32 },
223         [NL80211_ATTR_CENTER_FREQ1] = { .type = NLA_U32 },
224         [NL80211_ATTR_CENTER_FREQ2] = { .type = NLA_U32 },
225
226         [NL80211_ATTR_WIPHY_RETRY_SHORT] = { .type = NLA_U8 },
227         [NL80211_ATTR_WIPHY_RETRY_LONG] = { .type = NLA_U8 },
228         [NL80211_ATTR_WIPHY_FRAG_THRESHOLD] = { .type = NLA_U32 },
229         [NL80211_ATTR_WIPHY_RTS_THRESHOLD] = { .type = NLA_U32 },
230         [NL80211_ATTR_WIPHY_COVERAGE_CLASS] = { .type = NLA_U8 },
231         [NL80211_ATTR_WIPHY_DYN_ACK] = { .type = NLA_FLAG },
232
233         [NL80211_ATTR_IFTYPE] = { .type = NLA_U32 },
234         [NL80211_ATTR_IFINDEX] = { .type = NLA_U32 },
235         [NL80211_ATTR_IFNAME] = { .type = NLA_NUL_STRING, .len = IFNAMSIZ-1 },
236
237         [NL80211_ATTR_MAC] = { .len = ETH_ALEN },
238         [NL80211_ATTR_PREV_BSSID] = { .len = ETH_ALEN },
239
240         [NL80211_ATTR_KEY] = { .type = NLA_NESTED, },
241         [NL80211_ATTR_KEY_DATA] = { .type = NLA_BINARY,
242                                     .len = WLAN_MAX_KEY_LEN },
243         [NL80211_ATTR_KEY_IDX] = { .type = NLA_U8 },
244         [NL80211_ATTR_KEY_CIPHER] = { .type = NLA_U32 },
245         [NL80211_ATTR_KEY_DEFAULT] = { .type = NLA_FLAG },
246         [NL80211_ATTR_KEY_SEQ] = { .type = NLA_BINARY, .len = 16 },
247         [NL80211_ATTR_KEY_TYPE] = { .type = NLA_U32 },
248
249         [NL80211_ATTR_BEACON_INTERVAL] = { .type = NLA_U32 },
250         [NL80211_ATTR_DTIM_PERIOD] = { .type = NLA_U32 },
251         [NL80211_ATTR_BEACON_HEAD] = { .type = NLA_BINARY,
252                                        .len = IEEE80211_MAX_DATA_LEN },
253         [NL80211_ATTR_BEACON_TAIL] = { .type = NLA_BINARY,
254                                        .len = IEEE80211_MAX_DATA_LEN },
255         [NL80211_ATTR_STA_AID] = { .type = NLA_U16 },
256         [NL80211_ATTR_STA_FLAGS] = { .type = NLA_NESTED },
257         [NL80211_ATTR_STA_LISTEN_INTERVAL] = { .type = NLA_U16 },
258         [NL80211_ATTR_STA_SUPPORTED_RATES] = { .type = NLA_BINARY,
259                                                .len = NL80211_MAX_SUPP_RATES },
260         [NL80211_ATTR_STA_PLINK_ACTION] = { .type = NLA_U8 },
261         [NL80211_ATTR_STA_VLAN] = { .type = NLA_U32 },
262         [NL80211_ATTR_MNTR_FLAGS] = { /* NLA_NESTED can't be empty */ },
263         [NL80211_ATTR_MESH_ID] = { .type = NLA_BINARY,
264                                    .len = IEEE80211_MAX_MESH_ID_LEN },
265         [NL80211_ATTR_MPATH_NEXT_HOP] = { .type = NLA_U32 },
266
267         [NL80211_ATTR_REG_ALPHA2] = { .type = NLA_STRING, .len = 2 },
268         [NL80211_ATTR_REG_RULES] = { .type = NLA_NESTED },
269
270         [NL80211_ATTR_BSS_CTS_PROT] = { .type = NLA_U8 },
271         [NL80211_ATTR_BSS_SHORT_PREAMBLE] = { .type = NLA_U8 },
272         [NL80211_ATTR_BSS_SHORT_SLOT_TIME] = { .type = NLA_U8 },
273         [NL80211_ATTR_BSS_BASIC_RATES] = { .type = NLA_BINARY,
274                                            .len = NL80211_MAX_SUPP_RATES },
275         [NL80211_ATTR_BSS_HT_OPMODE] = { .type = NLA_U16 },
276
277         [NL80211_ATTR_MESH_CONFIG] = { .type = NLA_NESTED },
278         [NL80211_ATTR_SUPPORT_MESH_AUTH] = { .type = NLA_FLAG },
279
280         [NL80211_ATTR_HT_CAPABILITY] = { .len = NL80211_HT_CAPABILITY_LEN },
281
282         [NL80211_ATTR_MGMT_SUBTYPE] = { .type = NLA_U8 },
283         [NL80211_ATTR_IE] = { .type = NLA_BINARY,
284                               .len = IEEE80211_MAX_DATA_LEN },
285         [NL80211_ATTR_SCAN_FREQUENCIES] = { .type = NLA_NESTED },
286         [NL80211_ATTR_SCAN_SSIDS] = { .type = NLA_NESTED },
287
288         [NL80211_ATTR_SSID] = { .type = NLA_BINARY,
289                                 .len = IEEE80211_MAX_SSID_LEN },
290         [NL80211_ATTR_AUTH_TYPE] = { .type = NLA_U32 },
291         [NL80211_ATTR_REASON_CODE] = { .type = NLA_U16 },
292         [NL80211_ATTR_FREQ_FIXED] = { .type = NLA_FLAG },
293         [NL80211_ATTR_TIMED_OUT] = { .type = NLA_FLAG },
294         [NL80211_ATTR_USE_MFP] = { .type = NLA_U32 },
295         [NL80211_ATTR_STA_FLAGS2] = {
296                 .len = sizeof(struct nl80211_sta_flag_update),
297         },
298         [NL80211_ATTR_CONTROL_PORT] = { .type = NLA_FLAG },
299         [NL80211_ATTR_CONTROL_PORT_ETHERTYPE] = { .type = NLA_U16 },
300         [NL80211_ATTR_CONTROL_PORT_NO_ENCRYPT] = { .type = NLA_FLAG },
301         [NL80211_ATTR_PRIVACY] = { .type = NLA_FLAG },
302         [NL80211_ATTR_CIPHER_SUITE_GROUP] = { .type = NLA_U32 },
303         [NL80211_ATTR_WPA_VERSIONS] = { .type = NLA_U32 },
304         [NL80211_ATTR_PID] = { .type = NLA_U32 },
305         [NL80211_ATTR_4ADDR] = { .type = NLA_U8 },
306         [NL80211_ATTR_PMKID] = { .type = NLA_BINARY,
307                                  .len = WLAN_PMKID_LEN },
308         [NL80211_ATTR_DURATION] = { .type = NLA_U32 },
309         [NL80211_ATTR_COOKIE] = { .type = NLA_U64 },
310         [NL80211_ATTR_TX_RATES] = { .type = NLA_NESTED },
311         [NL80211_ATTR_FRAME] = { .type = NLA_BINARY,
312                                  .len = IEEE80211_MAX_DATA_LEN },
313         [NL80211_ATTR_FRAME_MATCH] = { .type = NLA_BINARY, },
314         [NL80211_ATTR_PS_STATE] = { .type = NLA_U32 },
315         [NL80211_ATTR_CQM] = { .type = NLA_NESTED, },
316         [NL80211_ATTR_LOCAL_STATE_CHANGE] = { .type = NLA_FLAG },
317         [NL80211_ATTR_AP_ISOLATE] = { .type = NLA_U8 },
318         [NL80211_ATTR_WIPHY_TX_POWER_SETTING] = { .type = NLA_U32 },
319         [NL80211_ATTR_WIPHY_TX_POWER_LEVEL] = { .type = NLA_U32 },
320         [NL80211_ATTR_FRAME_TYPE] = { .type = NLA_U16 },
321         [NL80211_ATTR_WIPHY_ANTENNA_TX] = { .type = NLA_U32 },
322         [NL80211_ATTR_WIPHY_ANTENNA_RX] = { .type = NLA_U32 },
323         [NL80211_ATTR_MCAST_RATE] = { .type = NLA_U32 },
324         [NL80211_ATTR_OFFCHANNEL_TX_OK] = { .type = NLA_FLAG },
325         [NL80211_ATTR_KEY_DEFAULT_TYPES] = { .type = NLA_NESTED },
326         [NL80211_ATTR_WOWLAN_TRIGGERS] = { .type = NLA_NESTED },
327         [NL80211_ATTR_STA_PLINK_STATE] = { .type = NLA_U8 },
328         [NL80211_ATTR_SCHED_SCAN_INTERVAL] = { .type = NLA_U32 },
329         [NL80211_ATTR_REKEY_DATA] = { .type = NLA_NESTED },
330         [NL80211_ATTR_SCAN_SUPP_RATES] = { .type = NLA_NESTED },
331         [NL80211_ATTR_HIDDEN_SSID] = { .type = NLA_U32 },
332         [NL80211_ATTR_IE_PROBE_RESP] = { .type = NLA_BINARY,
333                                          .len = IEEE80211_MAX_DATA_LEN },
334         [NL80211_ATTR_IE_ASSOC_RESP] = { .type = NLA_BINARY,
335                                          .len = IEEE80211_MAX_DATA_LEN },
336         [NL80211_ATTR_ROAM_SUPPORT] = { .type = NLA_FLAG },
337         [NL80211_ATTR_SCHED_SCAN_MATCH] = { .type = NLA_NESTED },
338         [NL80211_ATTR_TX_NO_CCK_RATE] = { .type = NLA_FLAG },
339         [NL80211_ATTR_TDLS_ACTION] = { .type = NLA_U8 },
340         [NL80211_ATTR_TDLS_DIALOG_TOKEN] = { .type = NLA_U8 },
341         [NL80211_ATTR_TDLS_OPERATION] = { .type = NLA_U8 },
342         [NL80211_ATTR_TDLS_SUPPORT] = { .type = NLA_FLAG },
343         [NL80211_ATTR_TDLS_EXTERNAL_SETUP] = { .type = NLA_FLAG },
344         [NL80211_ATTR_TDLS_INITIATOR] = { .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         [NL80211_ATTR_USER_REG_HINT_TYPE] = { .type = NLA_U32 },
358         [NL80211_ATTR_SAE_DATA] = { .type = NLA_BINARY, },
359         [NL80211_ATTR_VHT_CAPABILITY] = { .len = NL80211_VHT_CAPABILITY_LEN },
360         [NL80211_ATTR_SCAN_FLAGS] = { .type = NLA_U32 },
361         [NL80211_ATTR_P2P_CTWINDOW] = { .type = NLA_U8 },
362         [NL80211_ATTR_P2P_OPPPS] = { .type = NLA_U8 },
363         [NL80211_ATTR_ACL_POLICY] = {. type = NLA_U32 },
364         [NL80211_ATTR_MAC_ADDRS] = { .type = NLA_NESTED },
365         [NL80211_ATTR_STA_CAPABILITY] = { .type = NLA_U16 },
366         [NL80211_ATTR_STA_EXT_CAPABILITY] = { .type = NLA_BINARY, },
367         [NL80211_ATTR_SPLIT_WIPHY_DUMP] = { .type = NLA_FLAG, },
368         [NL80211_ATTR_DISABLE_VHT] = { .type = NLA_FLAG },
369         [NL80211_ATTR_VHT_CAPABILITY_MASK] = {
370                 .len = NL80211_VHT_CAPABILITY_LEN,
371         },
372         [NL80211_ATTR_MDID] = { .type = NLA_U16 },
373         [NL80211_ATTR_IE_RIC] = { .type = NLA_BINARY,
374                                   .len = IEEE80211_MAX_DATA_LEN },
375         [NL80211_ATTR_PEER_AID] = { .type = NLA_U16 },
376         [NL80211_ATTR_CH_SWITCH_COUNT] = { .type = NLA_U32 },
377         [NL80211_ATTR_CH_SWITCH_BLOCK_TX] = { .type = NLA_FLAG },
378         [NL80211_ATTR_CSA_IES] = { .type = NLA_NESTED },
379         [NL80211_ATTR_CSA_C_OFF_BEACON] = { .type = NLA_BINARY },
380         [NL80211_ATTR_CSA_C_OFF_PRESP] = { .type = NLA_BINARY },
381         [NL80211_ATTR_STA_SUPPORTED_CHANNELS] = { .type = NLA_BINARY },
382         [NL80211_ATTR_STA_SUPPORTED_OPER_CLASSES] = { .type = NLA_BINARY },
383         [NL80211_ATTR_HANDLE_DFS] = { .type = NLA_FLAG },
384         [NL80211_ATTR_OPMODE_NOTIF] = { .type = NLA_U8 },
385         [NL80211_ATTR_VENDOR_ID] = { .type = NLA_U32 },
386         [NL80211_ATTR_VENDOR_SUBCMD] = { .type = NLA_U32 },
387         [NL80211_ATTR_VENDOR_DATA] = { .type = NLA_BINARY },
388         [NL80211_ATTR_QOS_MAP] = { .type = NLA_BINARY,
389                                    .len = IEEE80211_QOS_MAP_LEN_MAX },
390         [NL80211_ATTR_MAC_HINT] = { .len = ETH_ALEN },
391         [NL80211_ATTR_WIPHY_FREQ_HINT] = { .type = NLA_U32 },
392         [NL80211_ATTR_TDLS_PEER_CAPABILITY] = { .type = NLA_U32 },
393         [NL80211_ATTR_SOCKET_OWNER] = { .type = NLA_FLAG },
394         [NL80211_ATTR_CSA_C_OFFSETS_TX] = { .type = NLA_BINARY },
395         [NL80211_ATTR_USE_RRM] = { .type = NLA_FLAG },
396         [NL80211_ATTR_TSID] = { .type = NLA_U8 },
397         [NL80211_ATTR_USER_PRIO] = { .type = NLA_U8 },
398         [NL80211_ATTR_ADMITTED_TIME] = { .type = NLA_U16 },
399         [NL80211_ATTR_SMPS_MODE] = { .type = NLA_U8 },
400         [NL80211_ATTR_MAC_MASK] = { .len = ETH_ALEN },
401         [NL80211_ATTR_WIPHY_SELF_MANAGED_REG] = { .type = NLA_FLAG },
402         [NL80211_ATTR_NETNS_FD] = { .type = NLA_U32 },
403         [NL80211_ATTR_SCHED_SCAN_DELAY] = { .type = NLA_U32 },
404         [NL80211_ATTR_REG_INDOOR] = { .type = NLA_FLAG },
405         [NL80211_ATTR_PBSS] = { .type = NLA_FLAG },
406         [NL80211_ATTR_BSS_SELECT] = { .type = NLA_NESTED },
407         [NL80211_ATTR_STA_SUPPORT_P2P_PS] = { .type = NLA_U8 },
408         [NL80211_ATTR_MU_MIMO_GROUP_DATA] = {
409                 .len = VHT_MUMIMO_GROUPS_DATA_LEN
410         },
411         [NL80211_ATTR_MU_MIMO_FOLLOW_MAC_ADDR] = { .len = ETH_ALEN },
412 };
413
414 /* policy for the key attributes */
415 static const struct nla_policy nl80211_key_policy[NL80211_KEY_MAX + 1] = {
416         [NL80211_KEY_DATA] = { .type = NLA_BINARY, .len = WLAN_MAX_KEY_LEN },
417         [NL80211_KEY_IDX] = { .type = NLA_U8 },
418         [NL80211_KEY_CIPHER] = { .type = NLA_U32 },
419         [NL80211_KEY_SEQ] = { .type = NLA_BINARY, .len = 16 },
420         [NL80211_KEY_DEFAULT] = { .type = NLA_FLAG },
421         [NL80211_KEY_DEFAULT_MGMT] = { .type = NLA_FLAG },
422         [NL80211_KEY_TYPE] = { .type = NLA_U32 },
423         [NL80211_KEY_DEFAULT_TYPES] = { .type = NLA_NESTED },
424 };
425
426 /* policy for the key default flags */
427 static const struct nla_policy
428 nl80211_key_default_policy[NUM_NL80211_KEY_DEFAULT_TYPES] = {
429         [NL80211_KEY_DEFAULT_TYPE_UNICAST] = { .type = NLA_FLAG },
430         [NL80211_KEY_DEFAULT_TYPE_MULTICAST] = { .type = NLA_FLAG },
431 };
432
433 /* policy for WoWLAN attributes */
434 static const struct nla_policy
435 nl80211_wowlan_policy[NUM_NL80211_WOWLAN_TRIG] = {
436         [NL80211_WOWLAN_TRIG_ANY] = { .type = NLA_FLAG },
437         [NL80211_WOWLAN_TRIG_DISCONNECT] = { .type = NLA_FLAG },
438         [NL80211_WOWLAN_TRIG_MAGIC_PKT] = { .type = NLA_FLAG },
439         [NL80211_WOWLAN_TRIG_PKT_PATTERN] = { .type = NLA_NESTED },
440         [NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE] = { .type = NLA_FLAG },
441         [NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST] = { .type = NLA_FLAG },
442         [NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE] = { .type = NLA_FLAG },
443         [NL80211_WOWLAN_TRIG_RFKILL_RELEASE] = { .type = NLA_FLAG },
444         [NL80211_WOWLAN_TRIG_TCP_CONNECTION] = { .type = NLA_NESTED },
445         [NL80211_WOWLAN_TRIG_NET_DETECT] = { .type = NLA_NESTED },
446 };
447
448 static const struct nla_policy
449 nl80211_wowlan_tcp_policy[NUM_NL80211_WOWLAN_TCP] = {
450         [NL80211_WOWLAN_TCP_SRC_IPV4] = { .type = NLA_U32 },
451         [NL80211_WOWLAN_TCP_DST_IPV4] = { .type = NLA_U32 },
452         [NL80211_WOWLAN_TCP_DST_MAC] = { .len = ETH_ALEN },
453         [NL80211_WOWLAN_TCP_SRC_PORT] = { .type = NLA_U16 },
454         [NL80211_WOWLAN_TCP_DST_PORT] = { .type = NLA_U16 },
455         [NL80211_WOWLAN_TCP_DATA_PAYLOAD] = { .len = 1 },
456         [NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ] = {
457                 .len = sizeof(struct nl80211_wowlan_tcp_data_seq)
458         },
459         [NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN] = {
460                 .len = sizeof(struct nl80211_wowlan_tcp_data_token)
461         },
462         [NL80211_WOWLAN_TCP_DATA_INTERVAL] = { .type = NLA_U32 },
463         [NL80211_WOWLAN_TCP_WAKE_PAYLOAD] = { .len = 1 },
464         [NL80211_WOWLAN_TCP_WAKE_MASK] = { .len = 1 },
465 };
466
467 /* policy for coalesce rule attributes */
468 static const struct nla_policy
469 nl80211_coalesce_policy[NUM_NL80211_ATTR_COALESCE_RULE] = {
470         [NL80211_ATTR_COALESCE_RULE_DELAY] = { .type = NLA_U32 },
471         [NL80211_ATTR_COALESCE_RULE_CONDITION] = { .type = NLA_U32 },
472         [NL80211_ATTR_COALESCE_RULE_PKT_PATTERN] = { .type = NLA_NESTED },
473 };
474
475 /* policy for GTK rekey offload attributes */
476 static const struct nla_policy
477 nl80211_rekey_policy[NUM_NL80211_REKEY_DATA] = {
478         [NL80211_REKEY_DATA_KEK] = { .len = NL80211_KEK_LEN },
479         [NL80211_REKEY_DATA_KCK] = { .len = NL80211_KCK_LEN },
480         [NL80211_REKEY_DATA_REPLAY_CTR] = { .len = NL80211_REPLAY_CTR_LEN },
481 };
482
483 static const struct nla_policy
484 nl80211_match_policy[NL80211_SCHED_SCAN_MATCH_ATTR_MAX + 1] = {
485         [NL80211_SCHED_SCAN_MATCH_ATTR_SSID] = { .type = NLA_BINARY,
486                                                  .len = IEEE80211_MAX_SSID_LEN },
487         [NL80211_SCHED_SCAN_MATCH_ATTR_RSSI] = { .type = NLA_U32 },
488 };
489
490 static const struct nla_policy
491 nl80211_plan_policy[NL80211_SCHED_SCAN_PLAN_MAX + 1] = {
492         [NL80211_SCHED_SCAN_PLAN_INTERVAL] = { .type = NLA_U32 },
493         [NL80211_SCHED_SCAN_PLAN_ITERATIONS] = { .type = NLA_U32 },
494 };
495
496 static const struct nla_policy
497 nl80211_bss_select_policy[NL80211_BSS_SELECT_ATTR_MAX + 1] = {
498         [NL80211_BSS_SELECT_ATTR_RSSI] = { .type = NLA_FLAG },
499         [NL80211_BSS_SELECT_ATTR_BAND_PREF] = { .type = NLA_U32 },
500         [NL80211_BSS_SELECT_ATTR_RSSI_ADJUST] = {
501                 .len = sizeof(struct nl80211_bss_select_rssi_adjust)
502         },
503 };
504
505 static int nl80211_prepare_wdev_dump(struct sk_buff *skb,
506                                      struct netlink_callback *cb,
507                                      struct cfg80211_registered_device **rdev,
508                                      struct wireless_dev **wdev)
509 {
510         int err;
511
512         rtnl_lock();
513
514         if (!cb->args[0]) {
515                 err = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
516                                   nl80211_fam.attrbuf, nl80211_fam.maxattr,
517                                   nl80211_policy);
518                 if (err)
519                         goto out_unlock;
520
521                 *wdev = __cfg80211_wdev_from_attrs(sock_net(skb->sk),
522                                                    nl80211_fam.attrbuf);
523                 if (IS_ERR(*wdev)) {
524                         err = PTR_ERR(*wdev);
525                         goto out_unlock;
526                 }
527                 *rdev = wiphy_to_rdev((*wdev)->wiphy);
528                 /* 0 is the first index - add 1 to parse only once */
529                 cb->args[0] = (*rdev)->wiphy_idx + 1;
530                 cb->args[1] = (*wdev)->identifier;
531         } else {
532                 /* subtract the 1 again here */
533                 struct wiphy *wiphy = wiphy_idx_to_wiphy(cb->args[0] - 1);
534                 struct wireless_dev *tmp;
535
536                 if (!wiphy) {
537                         err = -ENODEV;
538                         goto out_unlock;
539                 }
540                 *rdev = wiphy_to_rdev(wiphy);
541                 *wdev = NULL;
542
543                 list_for_each_entry(tmp, &(*rdev)->wiphy.wdev_list, list) {
544                         if (tmp->identifier == cb->args[1]) {
545                                 *wdev = tmp;
546                                 break;
547                         }
548                 }
549
550                 if (!*wdev) {
551                         err = -ENODEV;
552                         goto out_unlock;
553                 }
554         }
555
556         return 0;
557  out_unlock:
558         rtnl_unlock();
559         return err;
560 }
561
562 static void nl80211_finish_wdev_dump(struct cfg80211_registered_device *rdev)
563 {
564         rtnl_unlock();
565 }
566
567 /* IE validation */
568 static bool is_valid_ie_attr(const struct nlattr *attr)
569 {
570         const u8 *pos;
571         int len;
572
573         if (!attr)
574                 return true;
575
576         pos = nla_data(attr);
577         len = nla_len(attr);
578
579         while (len) {
580                 u8 elemlen;
581
582                 if (len < 2)
583                         return false;
584                 len -= 2;
585
586                 elemlen = pos[1];
587                 if (elemlen > len)
588                         return false;
589
590                 len -= elemlen;
591                 pos += 2 + elemlen;
592         }
593
594         return true;
595 }
596
597 /* message building helper */
598 static inline void *nl80211hdr_put(struct sk_buff *skb, u32 portid, u32 seq,
599                                    int flags, u8 cmd)
600 {
601         /* since there is no private header just add the generic one */
602         return genlmsg_put(skb, portid, seq, &nl80211_fam, flags, cmd);
603 }
604
605 static int nl80211_msg_put_channel(struct sk_buff *msg,
606                                    struct ieee80211_channel *chan,
607                                    bool large)
608 {
609         /* Some channels must be completely excluded from the
610          * list to protect old user-space tools from breaking
611          */
612         if (!large && chan->flags &
613             (IEEE80211_CHAN_NO_10MHZ | IEEE80211_CHAN_NO_20MHZ))
614                 return 0;
615
616         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_FREQ,
617                         chan->center_freq))
618                 goto nla_put_failure;
619
620         if ((chan->flags & IEEE80211_CHAN_DISABLED) &&
621             nla_put_flag(msg, NL80211_FREQUENCY_ATTR_DISABLED))
622                 goto nla_put_failure;
623         if (chan->flags & IEEE80211_CHAN_NO_IR) {
624                 if (nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_IR))
625                         goto nla_put_failure;
626                 if (nla_put_flag(msg, __NL80211_FREQUENCY_ATTR_NO_IBSS))
627                         goto nla_put_failure;
628         }
629         if (chan->flags & IEEE80211_CHAN_RADAR) {
630                 if (nla_put_flag(msg, NL80211_FREQUENCY_ATTR_RADAR))
631                         goto nla_put_failure;
632                 if (large) {
633                         u32 time;
634
635                         time = elapsed_jiffies_msecs(chan->dfs_state_entered);
636
637                         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_DFS_STATE,
638                                         chan->dfs_state))
639                                 goto nla_put_failure;
640                         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_DFS_TIME,
641                                         time))
642                                 goto nla_put_failure;
643                         if (nla_put_u32(msg,
644                                         NL80211_FREQUENCY_ATTR_DFS_CAC_TIME,
645                                         chan->dfs_cac_ms))
646                                 goto nla_put_failure;
647                 }
648         }
649
650         if (large) {
651                 if ((chan->flags & IEEE80211_CHAN_NO_HT40MINUS) &&
652                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_HT40_MINUS))
653                         goto nla_put_failure;
654                 if ((chan->flags & IEEE80211_CHAN_NO_HT40PLUS) &&
655                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_HT40_PLUS))
656                         goto nla_put_failure;
657                 if ((chan->flags & IEEE80211_CHAN_NO_80MHZ) &&
658                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_80MHZ))
659                         goto nla_put_failure;
660                 if ((chan->flags & IEEE80211_CHAN_NO_160MHZ) &&
661                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_160MHZ))
662                         goto nla_put_failure;
663                 if ((chan->flags & IEEE80211_CHAN_INDOOR_ONLY) &&
664                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_INDOOR_ONLY))
665                         goto nla_put_failure;
666                 if ((chan->flags & IEEE80211_CHAN_IR_CONCURRENT) &&
667                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_IR_CONCURRENT))
668                         goto nla_put_failure;
669                 if ((chan->flags & IEEE80211_CHAN_NO_20MHZ) &&
670                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_20MHZ))
671                         goto nla_put_failure;
672                 if ((chan->flags & IEEE80211_CHAN_NO_10MHZ) &&
673                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_10MHZ))
674                         goto nla_put_failure;
675         }
676
677         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_MAX_TX_POWER,
678                         DBM_TO_MBM(chan->max_power)))
679                 goto nla_put_failure;
680
681         return 0;
682
683  nla_put_failure:
684         return -ENOBUFS;
685 }
686
687 /* netlink command implementations */
688
689 struct key_parse {
690         struct key_params p;
691         int idx;
692         int type;
693         bool def, defmgmt;
694         bool def_uni, def_multi;
695 };
696
697 static int nl80211_parse_key_new(struct nlattr *key, struct key_parse *k)
698 {
699         struct nlattr *tb[NL80211_KEY_MAX + 1];
700         int err = nla_parse_nested(tb, NL80211_KEY_MAX, key,
701                                    nl80211_key_policy);
702         if (err)
703                 return err;
704
705         k->def = !!tb[NL80211_KEY_DEFAULT];
706         k->defmgmt = !!tb[NL80211_KEY_DEFAULT_MGMT];
707
708         if (k->def) {
709                 k->def_uni = true;
710                 k->def_multi = true;
711         }
712         if (k->defmgmt)
713                 k->def_multi = true;
714
715         if (tb[NL80211_KEY_IDX])
716                 k->idx = nla_get_u8(tb[NL80211_KEY_IDX]);
717
718         if (tb[NL80211_KEY_DATA]) {
719                 k->p.key = nla_data(tb[NL80211_KEY_DATA]);
720                 k->p.key_len = nla_len(tb[NL80211_KEY_DATA]);
721         }
722
723         if (tb[NL80211_KEY_SEQ]) {
724                 k->p.seq = nla_data(tb[NL80211_KEY_SEQ]);
725                 k->p.seq_len = nla_len(tb[NL80211_KEY_SEQ]);
726         }
727
728         if (tb[NL80211_KEY_CIPHER])
729                 k->p.cipher = nla_get_u32(tb[NL80211_KEY_CIPHER]);
730
731         if (tb[NL80211_KEY_TYPE]) {
732                 k->type = nla_get_u32(tb[NL80211_KEY_TYPE]);
733                 if (k->type < 0 || k->type >= NUM_NL80211_KEYTYPES)
734                         return -EINVAL;
735         }
736
737         if (tb[NL80211_KEY_DEFAULT_TYPES]) {
738                 struct nlattr *kdt[NUM_NL80211_KEY_DEFAULT_TYPES];
739
740                 err = nla_parse_nested(kdt, NUM_NL80211_KEY_DEFAULT_TYPES - 1,
741                                        tb[NL80211_KEY_DEFAULT_TYPES],
742                                        nl80211_key_default_policy);
743                 if (err)
744                         return err;
745
746                 k->def_uni = kdt[NL80211_KEY_DEFAULT_TYPE_UNICAST];
747                 k->def_multi = kdt[NL80211_KEY_DEFAULT_TYPE_MULTICAST];
748         }
749
750         return 0;
751 }
752
753 static int nl80211_parse_key_old(struct genl_info *info, struct key_parse *k)
754 {
755         if (info->attrs[NL80211_ATTR_KEY_DATA]) {
756                 k->p.key = nla_data(info->attrs[NL80211_ATTR_KEY_DATA]);
757                 k->p.key_len = nla_len(info->attrs[NL80211_ATTR_KEY_DATA]);
758         }
759
760         if (info->attrs[NL80211_ATTR_KEY_SEQ]) {
761                 k->p.seq = nla_data(info->attrs[NL80211_ATTR_KEY_SEQ]);
762                 k->p.seq_len = nla_len(info->attrs[NL80211_ATTR_KEY_SEQ]);
763         }
764
765         if (info->attrs[NL80211_ATTR_KEY_IDX])
766                 k->idx = nla_get_u8(info->attrs[NL80211_ATTR_KEY_IDX]);
767
768         if (info->attrs[NL80211_ATTR_KEY_CIPHER])
769                 k->p.cipher = nla_get_u32(info->attrs[NL80211_ATTR_KEY_CIPHER]);
770
771         k->def = !!info->attrs[NL80211_ATTR_KEY_DEFAULT];
772         k->defmgmt = !!info->attrs[NL80211_ATTR_KEY_DEFAULT_MGMT];
773
774         if (k->def) {
775                 k->def_uni = true;
776                 k->def_multi = true;
777         }
778         if (k->defmgmt)
779                 k->def_multi = true;
780
781         if (info->attrs[NL80211_ATTR_KEY_TYPE]) {
782                 k->type = nla_get_u32(info->attrs[NL80211_ATTR_KEY_TYPE]);
783                 if (k->type < 0 || k->type >= NUM_NL80211_KEYTYPES)
784                         return -EINVAL;
785         }
786
787         if (info->attrs[NL80211_ATTR_KEY_DEFAULT_TYPES]) {
788                 struct nlattr *kdt[NUM_NL80211_KEY_DEFAULT_TYPES];
789                 int err = nla_parse_nested(
790                                 kdt, NUM_NL80211_KEY_DEFAULT_TYPES - 1,
791                                 info->attrs[NL80211_ATTR_KEY_DEFAULT_TYPES],
792                                 nl80211_key_default_policy);
793                 if (err)
794                         return err;
795
796                 k->def_uni = kdt[NL80211_KEY_DEFAULT_TYPE_UNICAST];
797                 k->def_multi = kdt[NL80211_KEY_DEFAULT_TYPE_MULTICAST];
798         }
799
800         return 0;
801 }
802
803 static int nl80211_parse_key(struct genl_info *info, struct key_parse *k)
804 {
805         int err;
806
807         memset(k, 0, sizeof(*k));
808         k->idx = -1;
809         k->type = -1;
810
811         if (info->attrs[NL80211_ATTR_KEY])
812                 err = nl80211_parse_key_new(info->attrs[NL80211_ATTR_KEY], k);
813         else
814                 err = nl80211_parse_key_old(info, k);
815
816         if (err)
817                 return err;
818
819         if (k->def && k->defmgmt)
820                 return -EINVAL;
821
822         if (k->defmgmt) {
823                 if (k->def_uni || !k->def_multi)
824                         return -EINVAL;
825         }
826
827         if (k->idx != -1) {
828                 if (k->defmgmt) {
829                         if (k->idx < 4 || k->idx > 5)
830                                 return -EINVAL;
831                 } else if (k->def) {
832                         if (k->idx < 0 || k->idx > 3)
833                                 return -EINVAL;
834                 } else {
835                         if (k->idx < 0 || k->idx > 5)
836                                 return -EINVAL;
837                 }
838         }
839
840         return 0;
841 }
842
843 static struct cfg80211_cached_keys *
844 nl80211_parse_connkeys(struct cfg80211_registered_device *rdev,
845                        struct nlattr *keys, bool *no_ht)
846 {
847         struct key_parse parse;
848         struct nlattr *key;
849         struct cfg80211_cached_keys *result;
850         int rem, err, def = 0;
851
852         result = kzalloc(sizeof(*result), GFP_KERNEL);
853         if (!result)
854                 return ERR_PTR(-ENOMEM);
855
856         result->def = -1;
857         result->defmgmt = -1;
858
859         nla_for_each_nested(key, keys, rem) {
860                 memset(&parse, 0, sizeof(parse));
861                 parse.idx = -1;
862
863                 err = nl80211_parse_key_new(key, &parse);
864                 if (err)
865                         goto error;
866                 err = -EINVAL;
867                 if (!parse.p.key)
868                         goto error;
869                 if (parse.idx < 0 || parse.idx > 4)
870                         goto error;
871                 if (parse.def) {
872                         if (def)
873                                 goto error;
874                         def = 1;
875                         result->def = parse.idx;
876                         if (!parse.def_uni || !parse.def_multi)
877                                 goto error;
878                 } else if (parse.defmgmt)
879                         goto error;
880                 err = cfg80211_validate_key_settings(rdev, &parse.p,
881                                                      parse.idx, false, NULL);
882                 if (err)
883                         goto error;
884                 result->params[parse.idx].cipher = parse.p.cipher;
885                 result->params[parse.idx].key_len = parse.p.key_len;
886                 result->params[parse.idx].key = result->data[parse.idx];
887                 memcpy(result->data[parse.idx], parse.p.key, parse.p.key_len);
888
889                 if (parse.p.cipher == WLAN_CIPHER_SUITE_WEP40 ||
890                     parse.p.cipher == WLAN_CIPHER_SUITE_WEP104) {
891                         if (no_ht)
892                                 *no_ht = true;
893                 }
894         }
895
896         return result;
897  error:
898         kfree(result);
899         return ERR_PTR(err);
900 }
901
902 static int nl80211_key_allowed(struct wireless_dev *wdev)
903 {
904         ASSERT_WDEV_LOCK(wdev);
905
906         switch (wdev->iftype) {
907         case NL80211_IFTYPE_AP:
908         case NL80211_IFTYPE_AP_VLAN:
909         case NL80211_IFTYPE_P2P_GO:
910         case NL80211_IFTYPE_MESH_POINT:
911                 break;
912         case NL80211_IFTYPE_ADHOC:
913         case NL80211_IFTYPE_STATION:
914         case NL80211_IFTYPE_P2P_CLIENT:
915                 if (!wdev->current_bss)
916                         return -ENOLINK;
917                 break;
918         case NL80211_IFTYPE_UNSPECIFIED:
919         case NL80211_IFTYPE_OCB:
920         case NL80211_IFTYPE_MONITOR:
921         case NL80211_IFTYPE_P2P_DEVICE:
922         case NL80211_IFTYPE_WDS:
923         case NUM_NL80211_IFTYPES:
924                 return -EINVAL;
925         }
926
927         return 0;
928 }
929
930 static struct ieee80211_channel *nl80211_get_valid_chan(struct wiphy *wiphy,
931                                                         struct nlattr *tb)
932 {
933         struct ieee80211_channel *chan;
934
935         if (tb == NULL)
936                 return NULL;
937         chan = ieee80211_get_channel(wiphy, nla_get_u32(tb));
938         if (!chan || chan->flags & IEEE80211_CHAN_DISABLED)
939                 return NULL;
940         return chan;
941 }
942
943 static int nl80211_put_iftypes(struct sk_buff *msg, u32 attr, u16 ifmodes)
944 {
945         struct nlattr *nl_modes = nla_nest_start(msg, attr);
946         int i;
947
948         if (!nl_modes)
949                 goto nla_put_failure;
950
951         i = 0;
952         while (ifmodes) {
953                 if ((ifmodes & 1) && nla_put_flag(msg, i))
954                         goto nla_put_failure;
955                 ifmodes >>= 1;
956                 i++;
957         }
958
959         nla_nest_end(msg, nl_modes);
960         return 0;
961
962 nla_put_failure:
963         return -ENOBUFS;
964 }
965
966 static int nl80211_put_iface_combinations(struct wiphy *wiphy,
967                                           struct sk_buff *msg,
968                                           bool large)
969 {
970         struct nlattr *nl_combis;
971         int i, j;
972
973         nl_combis = nla_nest_start(msg,
974                                 NL80211_ATTR_INTERFACE_COMBINATIONS);
975         if (!nl_combis)
976                 goto nla_put_failure;
977
978         for (i = 0; i < wiphy->n_iface_combinations; i++) {
979                 const struct ieee80211_iface_combination *c;
980                 struct nlattr *nl_combi, *nl_limits;
981
982                 c = &wiphy->iface_combinations[i];
983
984                 nl_combi = nla_nest_start(msg, i + 1);
985                 if (!nl_combi)
986                         goto nla_put_failure;
987
988                 nl_limits = nla_nest_start(msg, NL80211_IFACE_COMB_LIMITS);
989                 if (!nl_limits)
990                         goto nla_put_failure;
991
992                 for (j = 0; j < c->n_limits; j++) {
993                         struct nlattr *nl_limit;
994
995                         nl_limit = nla_nest_start(msg, j + 1);
996                         if (!nl_limit)
997                                 goto nla_put_failure;
998                         if (nla_put_u32(msg, NL80211_IFACE_LIMIT_MAX,
999                                         c->limits[j].max))
1000                                 goto nla_put_failure;
1001                         if (nl80211_put_iftypes(msg, NL80211_IFACE_LIMIT_TYPES,
1002                                                 c->limits[j].types))
1003                                 goto nla_put_failure;
1004                         nla_nest_end(msg, nl_limit);
1005                 }
1006
1007                 nla_nest_end(msg, nl_limits);
1008
1009                 if (c->beacon_int_infra_match &&
1010                     nla_put_flag(msg, NL80211_IFACE_COMB_STA_AP_BI_MATCH))
1011                         goto nla_put_failure;
1012                 if (nla_put_u32(msg, NL80211_IFACE_COMB_NUM_CHANNELS,
1013                                 c->num_different_channels) ||
1014                     nla_put_u32(msg, NL80211_IFACE_COMB_MAXNUM,
1015                                 c->max_interfaces))
1016                         goto nla_put_failure;
1017                 if (large &&
1018                     (nla_put_u32(msg, NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS,
1019                                 c->radar_detect_widths) ||
1020                      nla_put_u32(msg, NL80211_IFACE_COMB_RADAR_DETECT_REGIONS,
1021                                 c->radar_detect_regions)))
1022                         goto nla_put_failure;
1023
1024                 nla_nest_end(msg, nl_combi);
1025         }
1026
1027         nla_nest_end(msg, nl_combis);
1028
1029         return 0;
1030 nla_put_failure:
1031         return -ENOBUFS;
1032 }
1033
1034 #ifdef CONFIG_PM
1035 static int nl80211_send_wowlan_tcp_caps(struct cfg80211_registered_device *rdev,
1036                                         struct sk_buff *msg)
1037 {
1038         const struct wiphy_wowlan_tcp_support *tcp = rdev->wiphy.wowlan->tcp;
1039         struct nlattr *nl_tcp;
1040
1041         if (!tcp)
1042                 return 0;
1043
1044         nl_tcp = nla_nest_start(msg, NL80211_WOWLAN_TRIG_TCP_CONNECTION);
1045         if (!nl_tcp)
1046                 return -ENOBUFS;
1047
1048         if (nla_put_u32(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD,
1049                         tcp->data_payload_max))
1050                 return -ENOBUFS;
1051
1052         if (nla_put_u32(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD,
1053                         tcp->data_payload_max))
1054                 return -ENOBUFS;
1055
1056         if (tcp->seq && nla_put_flag(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ))
1057                 return -ENOBUFS;
1058
1059         if (tcp->tok && nla_put(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN,
1060                                 sizeof(*tcp->tok), tcp->tok))
1061                 return -ENOBUFS;
1062
1063         if (nla_put_u32(msg, NL80211_WOWLAN_TCP_DATA_INTERVAL,
1064                         tcp->data_interval_max))
1065                 return -ENOBUFS;
1066
1067         if (nla_put_u32(msg, NL80211_WOWLAN_TCP_WAKE_PAYLOAD,
1068                         tcp->wake_payload_max))
1069                 return -ENOBUFS;
1070
1071         nla_nest_end(msg, nl_tcp);
1072         return 0;
1073 }
1074
1075 static int nl80211_send_wowlan(struct sk_buff *msg,
1076                                struct cfg80211_registered_device *rdev,
1077                                bool large)
1078 {
1079         struct nlattr *nl_wowlan;
1080
1081         if (!rdev->wiphy.wowlan)
1082                 return 0;
1083
1084         nl_wowlan = nla_nest_start(msg, NL80211_ATTR_WOWLAN_TRIGGERS_SUPPORTED);
1085         if (!nl_wowlan)
1086                 return -ENOBUFS;
1087
1088         if (((rdev->wiphy.wowlan->flags & WIPHY_WOWLAN_ANY) &&
1089              nla_put_flag(msg, NL80211_WOWLAN_TRIG_ANY)) ||
1090             ((rdev->wiphy.wowlan->flags & WIPHY_WOWLAN_DISCONNECT) &&
1091              nla_put_flag(msg, NL80211_WOWLAN_TRIG_DISCONNECT)) ||
1092             ((rdev->wiphy.wowlan->flags & WIPHY_WOWLAN_MAGIC_PKT) &&
1093              nla_put_flag(msg, NL80211_WOWLAN_TRIG_MAGIC_PKT)) ||
1094             ((rdev->wiphy.wowlan->flags & WIPHY_WOWLAN_SUPPORTS_GTK_REKEY) &&
1095              nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_SUPPORTED)) ||
1096             ((rdev->wiphy.wowlan->flags & WIPHY_WOWLAN_GTK_REKEY_FAILURE) &&
1097              nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE)) ||
1098             ((rdev->wiphy.wowlan->flags & WIPHY_WOWLAN_EAP_IDENTITY_REQ) &&
1099              nla_put_flag(msg, NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST)) ||
1100             ((rdev->wiphy.wowlan->flags & WIPHY_WOWLAN_4WAY_HANDSHAKE) &&
1101              nla_put_flag(msg, NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE)) ||
1102             ((rdev->wiphy.wowlan->flags & WIPHY_WOWLAN_RFKILL_RELEASE) &&
1103              nla_put_flag(msg, NL80211_WOWLAN_TRIG_RFKILL_RELEASE)))
1104                 return -ENOBUFS;
1105
1106         if (rdev->wiphy.wowlan->n_patterns) {
1107                 struct nl80211_pattern_support pat = {
1108                         .max_patterns = rdev->wiphy.wowlan->n_patterns,
1109                         .min_pattern_len = rdev->wiphy.wowlan->pattern_min_len,
1110                         .max_pattern_len = rdev->wiphy.wowlan->pattern_max_len,
1111                         .max_pkt_offset = rdev->wiphy.wowlan->max_pkt_offset,
1112                 };
1113
1114                 if (nla_put(msg, NL80211_WOWLAN_TRIG_PKT_PATTERN,
1115                             sizeof(pat), &pat))
1116                         return -ENOBUFS;
1117         }
1118
1119         if ((rdev->wiphy.wowlan->flags & WIPHY_WOWLAN_NET_DETECT) &&
1120             nla_put_u32(msg, NL80211_WOWLAN_TRIG_NET_DETECT,
1121                         rdev->wiphy.wowlan->max_nd_match_sets))
1122                 return -ENOBUFS;
1123
1124         if (large && nl80211_send_wowlan_tcp_caps(rdev, msg))
1125                 return -ENOBUFS;
1126
1127         nla_nest_end(msg, nl_wowlan);
1128
1129         return 0;
1130 }
1131 #endif
1132
1133 static int nl80211_send_coalesce(struct sk_buff *msg,
1134                                  struct cfg80211_registered_device *rdev)
1135 {
1136         struct nl80211_coalesce_rule_support rule;
1137
1138         if (!rdev->wiphy.coalesce)
1139                 return 0;
1140
1141         rule.max_rules = rdev->wiphy.coalesce->n_rules;
1142         rule.max_delay = rdev->wiphy.coalesce->max_delay;
1143         rule.pat.max_patterns = rdev->wiphy.coalesce->n_patterns;
1144         rule.pat.min_pattern_len = rdev->wiphy.coalesce->pattern_min_len;
1145         rule.pat.max_pattern_len = rdev->wiphy.coalesce->pattern_max_len;
1146         rule.pat.max_pkt_offset = rdev->wiphy.coalesce->max_pkt_offset;
1147
1148         if (nla_put(msg, NL80211_ATTR_COALESCE_RULE, sizeof(rule), &rule))
1149                 return -ENOBUFS;
1150
1151         return 0;
1152 }
1153
1154 static int nl80211_send_band_rateinfo(struct sk_buff *msg,
1155                                       struct ieee80211_supported_band *sband)
1156 {
1157         struct nlattr *nl_rates, *nl_rate;
1158         struct ieee80211_rate *rate;
1159         int i;
1160
1161         /* add HT info */
1162         if (sband->ht_cap.ht_supported &&
1163             (nla_put(msg, NL80211_BAND_ATTR_HT_MCS_SET,
1164                      sizeof(sband->ht_cap.mcs),
1165                      &sband->ht_cap.mcs) ||
1166              nla_put_u16(msg, NL80211_BAND_ATTR_HT_CAPA,
1167                          sband->ht_cap.cap) ||
1168              nla_put_u8(msg, NL80211_BAND_ATTR_HT_AMPDU_FACTOR,
1169                         sband->ht_cap.ampdu_factor) ||
1170              nla_put_u8(msg, NL80211_BAND_ATTR_HT_AMPDU_DENSITY,
1171                         sband->ht_cap.ampdu_density)))
1172                 return -ENOBUFS;
1173
1174         /* add VHT info */
1175         if (sband->vht_cap.vht_supported &&
1176             (nla_put(msg, NL80211_BAND_ATTR_VHT_MCS_SET,
1177                      sizeof(sband->vht_cap.vht_mcs),
1178                      &sband->vht_cap.vht_mcs) ||
1179              nla_put_u32(msg, NL80211_BAND_ATTR_VHT_CAPA,
1180                          sband->vht_cap.cap)))
1181                 return -ENOBUFS;
1182
1183         /* add bitrates */
1184         nl_rates = nla_nest_start(msg, NL80211_BAND_ATTR_RATES);
1185         if (!nl_rates)
1186                 return -ENOBUFS;
1187
1188         for (i = 0; i < sband->n_bitrates; i++) {
1189                 nl_rate = nla_nest_start(msg, i);
1190                 if (!nl_rate)
1191                         return -ENOBUFS;
1192
1193                 rate = &sband->bitrates[i];
1194                 if (nla_put_u32(msg, NL80211_BITRATE_ATTR_RATE,
1195                                 rate->bitrate))
1196                         return -ENOBUFS;
1197                 if ((rate->flags & IEEE80211_RATE_SHORT_PREAMBLE) &&
1198                     nla_put_flag(msg,
1199                                  NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE))
1200                         return -ENOBUFS;
1201
1202                 nla_nest_end(msg, nl_rate);
1203         }
1204
1205         nla_nest_end(msg, nl_rates);
1206
1207         return 0;
1208 }
1209
1210 static int
1211 nl80211_send_mgmt_stypes(struct sk_buff *msg,
1212                          const struct ieee80211_txrx_stypes *mgmt_stypes)
1213 {
1214         u16 stypes;
1215         struct nlattr *nl_ftypes, *nl_ifs;
1216         enum nl80211_iftype ift;
1217         int i;
1218
1219         if (!mgmt_stypes)
1220                 return 0;
1221
1222         nl_ifs = nla_nest_start(msg, NL80211_ATTR_TX_FRAME_TYPES);
1223         if (!nl_ifs)
1224                 return -ENOBUFS;
1225
1226         for (ift = 0; ift < NUM_NL80211_IFTYPES; ift++) {
1227                 nl_ftypes = nla_nest_start(msg, ift);
1228                 if (!nl_ftypes)
1229                         return -ENOBUFS;
1230                 i = 0;
1231                 stypes = mgmt_stypes[ift].tx;
1232                 while (stypes) {
1233                         if ((stypes & 1) &&
1234                             nla_put_u16(msg, NL80211_ATTR_FRAME_TYPE,
1235                                         (i << 4) | IEEE80211_FTYPE_MGMT))
1236                                 return -ENOBUFS;
1237                         stypes >>= 1;
1238                         i++;
1239                 }
1240                 nla_nest_end(msg, nl_ftypes);
1241         }
1242
1243         nla_nest_end(msg, nl_ifs);
1244
1245         nl_ifs = nla_nest_start(msg, NL80211_ATTR_RX_FRAME_TYPES);
1246         if (!nl_ifs)
1247                 return -ENOBUFS;
1248
1249         for (ift = 0; ift < NUM_NL80211_IFTYPES; ift++) {
1250                 nl_ftypes = nla_nest_start(msg, ift);
1251                 if (!nl_ftypes)
1252                         return -ENOBUFS;
1253                 i = 0;
1254                 stypes = mgmt_stypes[ift].rx;
1255                 while (stypes) {
1256                         if ((stypes & 1) &&
1257                             nla_put_u16(msg, NL80211_ATTR_FRAME_TYPE,
1258                                         (i << 4) | IEEE80211_FTYPE_MGMT))
1259                                 return -ENOBUFS;
1260                         stypes >>= 1;
1261                         i++;
1262                 }
1263                 nla_nest_end(msg, nl_ftypes);
1264         }
1265         nla_nest_end(msg, nl_ifs);
1266
1267         return 0;
1268 }
1269
1270 struct nl80211_dump_wiphy_state {
1271         s64 filter_wiphy;
1272         long start;
1273         long split_start, band_start, chan_start, capa_start;
1274         bool split;
1275 };
1276
1277 static int nl80211_send_wiphy(struct cfg80211_registered_device *rdev,
1278                               enum nl80211_commands cmd,
1279                               struct sk_buff *msg, u32 portid, u32 seq,
1280                               int flags, struct nl80211_dump_wiphy_state *state)
1281 {
1282         void *hdr;
1283         struct nlattr *nl_bands, *nl_band;
1284         struct nlattr *nl_freqs, *nl_freq;
1285         struct nlattr *nl_cmds;
1286         enum nl80211_band band;
1287         struct ieee80211_channel *chan;
1288         int i;
1289         const struct ieee80211_txrx_stypes *mgmt_stypes =
1290                                 rdev->wiphy.mgmt_stypes;
1291         u32 features;
1292
1293         hdr = nl80211hdr_put(msg, portid, seq, flags, cmd);
1294         if (!hdr)
1295                 return -ENOBUFS;
1296
1297         if (WARN_ON(!state))
1298                 return -EINVAL;
1299
1300         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
1301             nla_put_string(msg, NL80211_ATTR_WIPHY_NAME,
1302                            wiphy_name(&rdev->wiphy)) ||
1303             nla_put_u32(msg, NL80211_ATTR_GENERATION,
1304                         cfg80211_rdev_list_generation))
1305                 goto nla_put_failure;
1306
1307         if (cmd != NL80211_CMD_NEW_WIPHY)
1308                 goto finish;
1309
1310         switch (state->split_start) {
1311         case 0:
1312                 if (nla_put_u8(msg, NL80211_ATTR_WIPHY_RETRY_SHORT,
1313                                rdev->wiphy.retry_short) ||
1314                     nla_put_u8(msg, NL80211_ATTR_WIPHY_RETRY_LONG,
1315                                rdev->wiphy.retry_long) ||
1316                     nla_put_u32(msg, NL80211_ATTR_WIPHY_FRAG_THRESHOLD,
1317                                 rdev->wiphy.frag_threshold) ||
1318                     nla_put_u32(msg, NL80211_ATTR_WIPHY_RTS_THRESHOLD,
1319                                 rdev->wiphy.rts_threshold) ||
1320                     nla_put_u8(msg, NL80211_ATTR_WIPHY_COVERAGE_CLASS,
1321                                rdev->wiphy.coverage_class) ||
1322                     nla_put_u8(msg, NL80211_ATTR_MAX_NUM_SCAN_SSIDS,
1323                                rdev->wiphy.max_scan_ssids) ||
1324                     nla_put_u8(msg, NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS,
1325                                rdev->wiphy.max_sched_scan_ssids) ||
1326                     nla_put_u16(msg, NL80211_ATTR_MAX_SCAN_IE_LEN,
1327                                 rdev->wiphy.max_scan_ie_len) ||
1328                     nla_put_u16(msg, NL80211_ATTR_MAX_SCHED_SCAN_IE_LEN,
1329                                 rdev->wiphy.max_sched_scan_ie_len) ||
1330                     nla_put_u8(msg, NL80211_ATTR_MAX_MATCH_SETS,
1331                                rdev->wiphy.max_match_sets) ||
1332                     nla_put_u32(msg, NL80211_ATTR_MAX_NUM_SCHED_SCAN_PLANS,
1333                                 rdev->wiphy.max_sched_scan_plans) ||
1334                     nla_put_u32(msg, NL80211_ATTR_MAX_SCAN_PLAN_INTERVAL,
1335                                 rdev->wiphy.max_sched_scan_plan_interval) ||
1336                     nla_put_u32(msg, NL80211_ATTR_MAX_SCAN_PLAN_ITERATIONS,
1337                                 rdev->wiphy.max_sched_scan_plan_iterations))
1338                         goto nla_put_failure;
1339
1340                 if ((rdev->wiphy.flags & WIPHY_FLAG_IBSS_RSN) &&
1341                     nla_put_flag(msg, NL80211_ATTR_SUPPORT_IBSS_RSN))
1342                         goto nla_put_failure;
1343                 if ((rdev->wiphy.flags & WIPHY_FLAG_MESH_AUTH) &&
1344                     nla_put_flag(msg, NL80211_ATTR_SUPPORT_MESH_AUTH))
1345                         goto nla_put_failure;
1346                 if ((rdev->wiphy.flags & WIPHY_FLAG_AP_UAPSD) &&
1347                     nla_put_flag(msg, NL80211_ATTR_SUPPORT_AP_UAPSD))
1348                         goto nla_put_failure;
1349                 if ((rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_FW_ROAM) &&
1350                     nla_put_flag(msg, NL80211_ATTR_ROAM_SUPPORT))
1351                         goto nla_put_failure;
1352                 if ((rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) &&
1353                     nla_put_flag(msg, NL80211_ATTR_TDLS_SUPPORT))
1354                         goto nla_put_failure;
1355                 if ((rdev->wiphy.flags & WIPHY_FLAG_TDLS_EXTERNAL_SETUP) &&
1356                     nla_put_flag(msg, NL80211_ATTR_TDLS_EXTERNAL_SETUP))
1357                         goto nla_put_failure;
1358                 state->split_start++;
1359                 if (state->split)
1360                         break;
1361         case 1:
1362                 if (nla_put(msg, NL80211_ATTR_CIPHER_SUITES,
1363                             sizeof(u32) * rdev->wiphy.n_cipher_suites,
1364                             rdev->wiphy.cipher_suites))
1365                         goto nla_put_failure;
1366
1367                 if (nla_put_u8(msg, NL80211_ATTR_MAX_NUM_PMKIDS,
1368                                rdev->wiphy.max_num_pmkids))
1369                         goto nla_put_failure;
1370
1371                 if ((rdev->wiphy.flags & WIPHY_FLAG_CONTROL_PORT_PROTOCOL) &&
1372                     nla_put_flag(msg, NL80211_ATTR_CONTROL_PORT_ETHERTYPE))
1373                         goto nla_put_failure;
1374
1375                 if (nla_put_u32(msg, NL80211_ATTR_WIPHY_ANTENNA_AVAIL_TX,
1376                                 rdev->wiphy.available_antennas_tx) ||
1377                     nla_put_u32(msg, NL80211_ATTR_WIPHY_ANTENNA_AVAIL_RX,
1378                                 rdev->wiphy.available_antennas_rx))
1379                         goto nla_put_failure;
1380
1381                 if ((rdev->wiphy.flags & WIPHY_FLAG_AP_PROBE_RESP_OFFLOAD) &&
1382                     nla_put_u32(msg, NL80211_ATTR_PROBE_RESP_OFFLOAD,
1383                                 rdev->wiphy.probe_resp_offload))
1384                         goto nla_put_failure;
1385
1386                 if ((rdev->wiphy.available_antennas_tx ||
1387                      rdev->wiphy.available_antennas_rx) &&
1388                     rdev->ops->get_antenna) {
1389                         u32 tx_ant = 0, rx_ant = 0;
1390                         int res;
1391
1392                         res = rdev_get_antenna(rdev, &tx_ant, &rx_ant);
1393                         if (!res) {
1394                                 if (nla_put_u32(msg,
1395                                                 NL80211_ATTR_WIPHY_ANTENNA_TX,
1396                                                 tx_ant) ||
1397                                     nla_put_u32(msg,
1398                                                 NL80211_ATTR_WIPHY_ANTENNA_RX,
1399                                                 rx_ant))
1400                                         goto nla_put_failure;
1401                         }
1402                 }
1403
1404                 state->split_start++;
1405                 if (state->split)
1406                         break;
1407         case 2:
1408                 if (nl80211_put_iftypes(msg, NL80211_ATTR_SUPPORTED_IFTYPES,
1409                                         rdev->wiphy.interface_modes))
1410                                 goto nla_put_failure;
1411                 state->split_start++;
1412                 if (state->split)
1413                         break;
1414         case 3:
1415                 nl_bands = nla_nest_start(msg, NL80211_ATTR_WIPHY_BANDS);
1416                 if (!nl_bands)
1417                         goto nla_put_failure;
1418
1419                 for (band = state->band_start;
1420                      band < NUM_NL80211_BANDS; band++) {
1421                         struct ieee80211_supported_band *sband;
1422
1423                         sband = rdev->wiphy.bands[band];
1424
1425                         if (!sband)
1426                                 continue;
1427
1428                         nl_band = nla_nest_start(msg, band);
1429                         if (!nl_band)
1430                                 goto nla_put_failure;
1431
1432                         switch (state->chan_start) {
1433                         case 0:
1434                                 if (nl80211_send_band_rateinfo(msg, sband))
1435                                         goto nla_put_failure;
1436                                 state->chan_start++;
1437                                 if (state->split)
1438                                         break;
1439                         default:
1440                                 /* add frequencies */
1441                                 nl_freqs = nla_nest_start(
1442                                         msg, NL80211_BAND_ATTR_FREQS);
1443                                 if (!nl_freqs)
1444                                         goto nla_put_failure;
1445
1446                                 for (i = state->chan_start - 1;
1447                                      i < sband->n_channels;
1448                                      i++) {
1449                                         nl_freq = nla_nest_start(msg, i);
1450                                         if (!nl_freq)
1451                                                 goto nla_put_failure;
1452
1453                                         chan = &sband->channels[i];
1454
1455                                         if (nl80211_msg_put_channel(
1456                                                         msg, chan,
1457                                                         state->split))
1458                                                 goto nla_put_failure;
1459
1460                                         nla_nest_end(msg, nl_freq);
1461                                         if (state->split)
1462                                                 break;
1463                                 }
1464                                 if (i < sband->n_channels)
1465                                         state->chan_start = i + 2;
1466                                 else
1467                                         state->chan_start = 0;
1468                                 nla_nest_end(msg, nl_freqs);
1469                         }
1470
1471                         nla_nest_end(msg, nl_band);
1472
1473                         if (state->split) {
1474                                 /* start again here */
1475                                 if (state->chan_start)
1476                                         band--;
1477                                 break;
1478                         }
1479                 }
1480                 nla_nest_end(msg, nl_bands);
1481
1482                 if (band < NUM_NL80211_BANDS)
1483                         state->band_start = band + 1;
1484                 else
1485                         state->band_start = 0;
1486
1487                 /* if bands & channels are done, continue outside */
1488                 if (state->band_start == 0 && state->chan_start == 0)
1489                         state->split_start++;
1490                 if (state->split)
1491                         break;
1492         case 4:
1493                 nl_cmds = nla_nest_start(msg, NL80211_ATTR_SUPPORTED_COMMANDS);
1494                 if (!nl_cmds)
1495                         goto nla_put_failure;
1496
1497                 i = 0;
1498 #define CMD(op, n)                                                      \
1499                  do {                                                   \
1500                         if (rdev->ops->op) {                            \
1501                                 i++;                                    \
1502                                 if (nla_put_u32(msg, i, NL80211_CMD_ ## n)) \
1503                                         goto nla_put_failure;           \
1504                         }                                               \
1505                 } while (0)
1506
1507                 CMD(add_virtual_intf, NEW_INTERFACE);
1508                 CMD(change_virtual_intf, SET_INTERFACE);
1509                 CMD(add_key, NEW_KEY);
1510                 CMD(start_ap, START_AP);
1511                 CMD(add_station, NEW_STATION);
1512                 CMD(add_mpath, NEW_MPATH);
1513                 CMD(update_mesh_config, SET_MESH_CONFIG);
1514                 CMD(change_bss, SET_BSS);
1515                 CMD(auth, AUTHENTICATE);
1516                 CMD(assoc, ASSOCIATE);
1517                 CMD(deauth, DEAUTHENTICATE);
1518                 CMD(disassoc, DISASSOCIATE);
1519                 CMD(join_ibss, JOIN_IBSS);
1520                 CMD(join_mesh, JOIN_MESH);
1521                 CMD(set_pmksa, SET_PMKSA);
1522                 CMD(del_pmksa, DEL_PMKSA);
1523                 CMD(flush_pmksa, FLUSH_PMKSA);
1524                 if (rdev->wiphy.flags & WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL)
1525                         CMD(remain_on_channel, REMAIN_ON_CHANNEL);
1526                 CMD(set_bitrate_mask, SET_TX_BITRATE_MASK);
1527                 CMD(mgmt_tx, FRAME);
1528                 CMD(mgmt_tx_cancel_wait, FRAME_WAIT_CANCEL);
1529                 if (rdev->wiphy.flags & WIPHY_FLAG_NETNS_OK) {
1530                         i++;
1531                         if (nla_put_u32(msg, i, NL80211_CMD_SET_WIPHY_NETNS))
1532                                 goto nla_put_failure;
1533                 }
1534                 if (rdev->ops->set_monitor_channel || rdev->ops->start_ap ||
1535                     rdev->ops->join_mesh) {
1536                         i++;
1537                         if (nla_put_u32(msg, i, NL80211_CMD_SET_CHANNEL))
1538                                 goto nla_put_failure;
1539                 }
1540                 CMD(set_wds_peer, SET_WDS_PEER);
1541                 if (rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) {
1542                         CMD(tdls_mgmt, TDLS_MGMT);
1543                         CMD(tdls_oper, TDLS_OPER);
1544                 }
1545                 if (rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN)
1546                         CMD(sched_scan_start, START_SCHED_SCAN);
1547                 CMD(probe_client, PROBE_CLIENT);
1548                 CMD(set_noack_map, SET_NOACK_MAP);
1549                 if (rdev->wiphy.flags & WIPHY_FLAG_REPORTS_OBSS) {
1550                         i++;
1551                         if (nla_put_u32(msg, i, NL80211_CMD_REGISTER_BEACONS))
1552                                 goto nla_put_failure;
1553                 }
1554                 CMD(start_p2p_device, START_P2P_DEVICE);
1555                 CMD(set_mcast_rate, SET_MCAST_RATE);
1556 #ifdef CONFIG_NL80211_TESTMODE
1557                 CMD(testmode_cmd, TESTMODE);
1558 #endif
1559                 if (state->split) {
1560                         CMD(crit_proto_start, CRIT_PROTOCOL_START);
1561                         CMD(crit_proto_stop, CRIT_PROTOCOL_STOP);
1562                         if (rdev->wiphy.flags & WIPHY_FLAG_HAS_CHANNEL_SWITCH)
1563                                 CMD(channel_switch, CHANNEL_SWITCH);
1564                         CMD(set_qos_map, SET_QOS_MAP);
1565                         if (rdev->wiphy.features &
1566                                         NL80211_FEATURE_SUPPORTS_WMM_ADMISSION)
1567                                 CMD(add_tx_ts, ADD_TX_TS);
1568                 }
1569                 /* add into the if now */
1570 #undef CMD
1571
1572                 if (rdev->ops->connect || rdev->ops->auth) {
1573                         i++;
1574                         if (nla_put_u32(msg, i, NL80211_CMD_CONNECT))
1575                                 goto nla_put_failure;
1576                 }
1577
1578                 if (rdev->ops->disconnect || rdev->ops->deauth) {
1579                         i++;
1580                         if (nla_put_u32(msg, i, NL80211_CMD_DISCONNECT))
1581                                 goto nla_put_failure;
1582                 }
1583
1584                 nla_nest_end(msg, nl_cmds);
1585                 state->split_start++;
1586                 if (state->split)
1587                         break;
1588         case 5:
1589                 if (rdev->ops->remain_on_channel &&
1590                     (rdev->wiphy.flags & WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL) &&
1591                     nla_put_u32(msg,
1592                                 NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION,
1593                                 rdev->wiphy.max_remain_on_channel_duration))
1594                         goto nla_put_failure;
1595
1596                 if ((rdev->wiphy.flags & WIPHY_FLAG_OFFCHAN_TX) &&
1597                     nla_put_flag(msg, NL80211_ATTR_OFFCHANNEL_TX_OK))
1598                         goto nla_put_failure;
1599
1600                 if (nl80211_send_mgmt_stypes(msg, mgmt_stypes))
1601                         goto nla_put_failure;
1602                 state->split_start++;
1603                 if (state->split)
1604                         break;
1605         case 6:
1606 #ifdef CONFIG_PM
1607                 if (nl80211_send_wowlan(msg, rdev, state->split))
1608                         goto nla_put_failure;
1609                 state->split_start++;
1610                 if (state->split)
1611                         break;
1612 #else
1613                 state->split_start++;
1614 #endif
1615         case 7:
1616                 if (nl80211_put_iftypes(msg, NL80211_ATTR_SOFTWARE_IFTYPES,
1617                                         rdev->wiphy.software_iftypes))
1618                         goto nla_put_failure;
1619
1620                 if (nl80211_put_iface_combinations(&rdev->wiphy, msg,
1621                                                    state->split))
1622                         goto nla_put_failure;
1623
1624                 state->split_start++;
1625                 if (state->split)
1626                         break;
1627         case 8:
1628                 if ((rdev->wiphy.flags & WIPHY_FLAG_HAVE_AP_SME) &&
1629                     nla_put_u32(msg, NL80211_ATTR_DEVICE_AP_SME,
1630                                 rdev->wiphy.ap_sme_capa))
1631                         goto nla_put_failure;
1632
1633                 features = rdev->wiphy.features;
1634                 /*
1635                  * We can only add the per-channel limit information if the
1636                  * dump is split, otherwise it makes it too big. Therefore
1637                  * only advertise it in that case.
1638                  */
1639                 if (state->split)
1640                         features |= NL80211_FEATURE_ADVERTISE_CHAN_LIMITS;
1641                 if (nla_put_u32(msg, NL80211_ATTR_FEATURE_FLAGS, features))
1642                         goto nla_put_failure;
1643
1644                 if (rdev->wiphy.ht_capa_mod_mask &&
1645                     nla_put(msg, NL80211_ATTR_HT_CAPABILITY_MASK,
1646                             sizeof(*rdev->wiphy.ht_capa_mod_mask),
1647                             rdev->wiphy.ht_capa_mod_mask))
1648                         goto nla_put_failure;
1649
1650                 if (rdev->wiphy.flags & WIPHY_FLAG_HAVE_AP_SME &&
1651                     rdev->wiphy.max_acl_mac_addrs &&
1652                     nla_put_u32(msg, NL80211_ATTR_MAC_ACL_MAX,
1653                                 rdev->wiphy.max_acl_mac_addrs))
1654                         goto nla_put_failure;
1655
1656                 /*
1657                  * Any information below this point is only available to
1658                  * applications that can deal with it being split. This
1659                  * helps ensure that newly added capabilities don't break
1660                  * older tools by overrunning their buffers.
1661                  *
1662                  * We still increment split_start so that in the split
1663                  * case we'll continue with more data in the next round,
1664                  * but break unconditionally so unsplit data stops here.
1665                  */
1666                 state->split_start++;
1667                 break;
1668         case 9:
1669                 if (rdev->wiphy.extended_capabilities &&
1670                     (nla_put(msg, NL80211_ATTR_EXT_CAPA,
1671                              rdev->wiphy.extended_capabilities_len,
1672                              rdev->wiphy.extended_capabilities) ||
1673                      nla_put(msg, NL80211_ATTR_EXT_CAPA_MASK,
1674                              rdev->wiphy.extended_capabilities_len,
1675                              rdev->wiphy.extended_capabilities_mask)))
1676                         goto nla_put_failure;
1677
1678                 if (rdev->wiphy.vht_capa_mod_mask &&
1679                     nla_put(msg, NL80211_ATTR_VHT_CAPABILITY_MASK,
1680                             sizeof(*rdev->wiphy.vht_capa_mod_mask),
1681                             rdev->wiphy.vht_capa_mod_mask))
1682                         goto nla_put_failure;
1683
1684                 state->split_start++;
1685                 break;
1686         case 10:
1687                 if (nl80211_send_coalesce(msg, rdev))
1688                         goto nla_put_failure;
1689
1690                 if ((rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_5_10_MHZ) &&
1691                     (nla_put_flag(msg, NL80211_ATTR_SUPPORT_5_MHZ) ||
1692                      nla_put_flag(msg, NL80211_ATTR_SUPPORT_10_MHZ)))
1693                         goto nla_put_failure;
1694
1695                 if (rdev->wiphy.max_ap_assoc_sta &&
1696                     nla_put_u32(msg, NL80211_ATTR_MAX_AP_ASSOC_STA,
1697                                 rdev->wiphy.max_ap_assoc_sta))
1698                         goto nla_put_failure;
1699
1700                 state->split_start++;
1701                 break;
1702         case 11:
1703                 if (rdev->wiphy.n_vendor_commands) {
1704                         const struct nl80211_vendor_cmd_info *info;
1705                         struct nlattr *nested;
1706
1707                         nested = nla_nest_start(msg, NL80211_ATTR_VENDOR_DATA);
1708                         if (!nested)
1709                                 goto nla_put_failure;
1710
1711                         for (i = 0; i < rdev->wiphy.n_vendor_commands; i++) {
1712                                 info = &rdev->wiphy.vendor_commands[i].info;
1713                                 if (nla_put(msg, i + 1, sizeof(*info), info))
1714                                         goto nla_put_failure;
1715                         }
1716                         nla_nest_end(msg, nested);
1717                 }
1718
1719                 if (rdev->wiphy.n_vendor_events) {
1720                         const struct nl80211_vendor_cmd_info *info;
1721                         struct nlattr *nested;
1722
1723                         nested = nla_nest_start(msg,
1724                                                 NL80211_ATTR_VENDOR_EVENTS);
1725                         if (!nested)
1726                                 goto nla_put_failure;
1727
1728                         for (i = 0; i < rdev->wiphy.n_vendor_events; i++) {
1729                                 info = &rdev->wiphy.vendor_events[i];
1730                                 if (nla_put(msg, i + 1, sizeof(*info), info))
1731                                         goto nla_put_failure;
1732                         }
1733                         nla_nest_end(msg, nested);
1734                 }
1735                 state->split_start++;
1736                 break;
1737         case 12:
1738                 if (rdev->wiphy.flags & WIPHY_FLAG_HAS_CHANNEL_SWITCH &&
1739                     nla_put_u8(msg, NL80211_ATTR_MAX_CSA_COUNTERS,
1740                                rdev->wiphy.max_num_csa_counters))
1741                         goto nla_put_failure;
1742
1743                 if (rdev->wiphy.regulatory_flags & REGULATORY_WIPHY_SELF_MANAGED &&
1744                     nla_put_flag(msg, NL80211_ATTR_WIPHY_SELF_MANAGED_REG))
1745                         goto nla_put_failure;
1746
1747                 if (nla_put(msg, NL80211_ATTR_EXT_FEATURES,
1748                             sizeof(rdev->wiphy.ext_features),
1749                             rdev->wiphy.ext_features))
1750                         goto nla_put_failure;
1751
1752                 if (rdev->wiphy.bss_select_support) {
1753                         struct nlattr *nested;
1754                         u32 bss_select_support = rdev->wiphy.bss_select_support;
1755
1756                         nested = nla_nest_start(msg, NL80211_ATTR_BSS_SELECT);
1757                         if (!nested)
1758                                 goto nla_put_failure;
1759
1760                         i = 0;
1761                         while (bss_select_support) {
1762                                 if ((bss_select_support & 1) &&
1763                                     nla_put_flag(msg, i))
1764                                         goto nla_put_failure;
1765                                 i++;
1766                                 bss_select_support >>= 1;
1767                         }
1768                         nla_nest_end(msg, nested);
1769                 }
1770
1771                 state->split_start++;
1772                 break;
1773         case 13:
1774                 if (rdev->wiphy.num_iftype_ext_capab &&
1775                     rdev->wiphy.iftype_ext_capab) {
1776                         struct nlattr *nested_ext_capab, *nested;
1777
1778                         nested = nla_nest_start(msg,
1779                                                 NL80211_ATTR_IFTYPE_EXT_CAPA);
1780                         if (!nested)
1781                                 goto nla_put_failure;
1782
1783                         for (i = state->capa_start;
1784                              i < rdev->wiphy.num_iftype_ext_capab; i++) {
1785                                 const struct wiphy_iftype_ext_capab *capab;
1786
1787                                 capab = &rdev->wiphy.iftype_ext_capab[i];
1788
1789                                 nested_ext_capab = nla_nest_start(msg, i);
1790                                 if (!nested_ext_capab ||
1791                                     nla_put_u32(msg, NL80211_ATTR_IFTYPE,
1792                                                 capab->iftype) ||
1793                                     nla_put(msg, NL80211_ATTR_EXT_CAPA,
1794                                             capab->extended_capabilities_len,
1795                                             capab->extended_capabilities) ||
1796                                     nla_put(msg, NL80211_ATTR_EXT_CAPA_MASK,
1797                                             capab->extended_capabilities_len,
1798                                             capab->extended_capabilities_mask))
1799                                         goto nla_put_failure;
1800
1801                                 nla_nest_end(msg, nested_ext_capab);
1802                                 if (state->split)
1803                                         break;
1804                         }
1805                         nla_nest_end(msg, nested);
1806                         if (i < rdev->wiphy.num_iftype_ext_capab) {
1807                                 state->capa_start = i + 1;
1808                                 break;
1809                         }
1810                 }
1811
1812                 /* done */
1813                 state->split_start = 0;
1814                 break;
1815         }
1816  finish:
1817         genlmsg_end(msg, hdr);
1818         return 0;
1819
1820  nla_put_failure:
1821         genlmsg_cancel(msg, hdr);
1822         return -EMSGSIZE;
1823 }
1824
1825 static int nl80211_dump_wiphy_parse(struct sk_buff *skb,
1826                                     struct netlink_callback *cb,
1827                                     struct nl80211_dump_wiphy_state *state)
1828 {
1829         struct nlattr **tb = nl80211_fam.attrbuf;
1830         int ret = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
1831                               tb, nl80211_fam.maxattr, nl80211_policy);
1832         /* ignore parse errors for backward compatibility */
1833         if (ret)
1834                 return 0;
1835
1836         state->split = tb[NL80211_ATTR_SPLIT_WIPHY_DUMP];
1837         if (tb[NL80211_ATTR_WIPHY])
1838                 state->filter_wiphy = nla_get_u32(tb[NL80211_ATTR_WIPHY]);
1839         if (tb[NL80211_ATTR_WDEV])
1840                 state->filter_wiphy = nla_get_u64(tb[NL80211_ATTR_WDEV]) >> 32;
1841         if (tb[NL80211_ATTR_IFINDEX]) {
1842                 struct net_device *netdev;
1843                 struct cfg80211_registered_device *rdev;
1844                 int ifidx = nla_get_u32(tb[NL80211_ATTR_IFINDEX]);
1845
1846                 netdev = __dev_get_by_index(sock_net(skb->sk), ifidx);
1847                 if (!netdev)
1848                         return -ENODEV;
1849                 if (netdev->ieee80211_ptr) {
1850                         rdev = wiphy_to_rdev(
1851                                 netdev->ieee80211_ptr->wiphy);
1852                         state->filter_wiphy = rdev->wiphy_idx;
1853                 }
1854         }
1855
1856         return 0;
1857 }
1858
1859 static int nl80211_dump_wiphy(struct sk_buff *skb, struct netlink_callback *cb)
1860 {
1861         int idx = 0, ret;
1862         struct nl80211_dump_wiphy_state *state = (void *)cb->args[0];
1863         struct cfg80211_registered_device *rdev;
1864
1865         rtnl_lock();
1866         if (!state) {
1867                 state = kzalloc(sizeof(*state), GFP_KERNEL);
1868                 if (!state) {
1869                         rtnl_unlock();
1870                         return -ENOMEM;
1871                 }
1872                 state->filter_wiphy = -1;
1873                 ret = nl80211_dump_wiphy_parse(skb, cb, state);
1874                 if (ret) {
1875                         kfree(state);
1876                         rtnl_unlock();
1877                         return ret;
1878                 }
1879                 cb->args[0] = (long)state;
1880         }
1881
1882         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
1883                 if (!net_eq(wiphy_net(&rdev->wiphy), sock_net(skb->sk)))
1884                         continue;
1885                 if (++idx <= state->start)
1886                         continue;
1887                 if (state->filter_wiphy != -1 &&
1888                     state->filter_wiphy != rdev->wiphy_idx)
1889                         continue;
1890                 /* attempt to fit multiple wiphy data chunks into the skb */
1891                 do {
1892                         ret = nl80211_send_wiphy(rdev, NL80211_CMD_NEW_WIPHY,
1893                                                  skb,
1894                                                  NETLINK_CB(cb->skb).portid,
1895                                                  cb->nlh->nlmsg_seq,
1896                                                  NLM_F_MULTI, state);
1897                         if (ret < 0) {
1898                                 /*
1899                                  * If sending the wiphy data didn't fit (ENOBUFS
1900                                  * or EMSGSIZE returned), this SKB is still
1901                                  * empty (so it's not too big because another
1902                                  * wiphy dataset is already in the skb) and
1903                                  * we've not tried to adjust the dump allocation
1904                                  * yet ... then adjust the alloc size to be
1905                                  * bigger, and return 1 but with the empty skb.
1906                                  * This results in an empty message being RX'ed
1907                                  * in userspace, but that is ignored.
1908                                  *
1909                                  * We can then retry with the larger buffer.
1910                                  */
1911                                 if ((ret == -ENOBUFS || ret == -EMSGSIZE) &&
1912                                     !skb->len && !state->split &&
1913                                     cb->min_dump_alloc < 4096) {
1914                                         cb->min_dump_alloc = 4096;
1915                                         state->split_start = 0;
1916                                         rtnl_unlock();
1917                                         return 1;
1918                                 }
1919                                 idx--;
1920                                 break;
1921                         }
1922                 } while (state->split_start > 0);
1923                 break;
1924         }
1925         rtnl_unlock();
1926
1927         state->start = idx;
1928
1929         return skb->len;
1930 }
1931
1932 static int nl80211_dump_wiphy_done(struct netlink_callback *cb)
1933 {
1934         kfree((void *)cb->args[0]);
1935         return 0;
1936 }
1937
1938 static int nl80211_get_wiphy(struct sk_buff *skb, struct genl_info *info)
1939 {
1940         struct sk_buff *msg;
1941         struct cfg80211_registered_device *rdev = info->user_ptr[0];
1942         struct nl80211_dump_wiphy_state state = {};
1943
1944         msg = nlmsg_new(4096, GFP_KERNEL);
1945         if (!msg)
1946                 return -ENOMEM;
1947
1948         if (nl80211_send_wiphy(rdev, NL80211_CMD_NEW_WIPHY, msg,
1949                                info->snd_portid, info->snd_seq, 0,
1950                                &state) < 0) {
1951                 nlmsg_free(msg);
1952                 return -ENOBUFS;
1953         }
1954
1955         return genlmsg_reply(msg, info);
1956 }
1957
1958 static const struct nla_policy txq_params_policy[NL80211_TXQ_ATTR_MAX + 1] = {
1959         [NL80211_TXQ_ATTR_QUEUE]                = { .type = NLA_U8 },
1960         [NL80211_TXQ_ATTR_TXOP]                 = { .type = NLA_U16 },
1961         [NL80211_TXQ_ATTR_CWMIN]                = { .type = NLA_U16 },
1962         [NL80211_TXQ_ATTR_CWMAX]                = { .type = NLA_U16 },
1963         [NL80211_TXQ_ATTR_AIFS]                 = { .type = NLA_U8 },
1964 };
1965
1966 static int parse_txq_params(struct nlattr *tb[],
1967                             struct ieee80211_txq_params *txq_params)
1968 {
1969         if (!tb[NL80211_TXQ_ATTR_AC] || !tb[NL80211_TXQ_ATTR_TXOP] ||
1970             !tb[NL80211_TXQ_ATTR_CWMIN] || !tb[NL80211_TXQ_ATTR_CWMAX] ||
1971             !tb[NL80211_TXQ_ATTR_AIFS])
1972                 return -EINVAL;
1973
1974         txq_params->ac = nla_get_u8(tb[NL80211_TXQ_ATTR_AC]);
1975         txq_params->txop = nla_get_u16(tb[NL80211_TXQ_ATTR_TXOP]);
1976         txq_params->cwmin = nla_get_u16(tb[NL80211_TXQ_ATTR_CWMIN]);
1977         txq_params->cwmax = nla_get_u16(tb[NL80211_TXQ_ATTR_CWMAX]);
1978         txq_params->aifs = nla_get_u8(tb[NL80211_TXQ_ATTR_AIFS]);
1979
1980         if (txq_params->ac >= NL80211_NUM_ACS)
1981                 return -EINVAL;
1982
1983         return 0;
1984 }
1985
1986 static bool nl80211_can_set_dev_channel(struct wireless_dev *wdev)
1987 {
1988         /*
1989          * You can only set the channel explicitly for WDS interfaces,
1990          * all others have their channel managed via their respective
1991          * "establish a connection" command (connect, join, ...)
1992          *
1993          * For AP/GO and mesh mode, the channel can be set with the
1994          * channel userspace API, but is only stored and passed to the
1995          * low-level driver when the AP starts or the mesh is joined.
1996          * This is for backward compatibility, userspace can also give
1997          * the channel in the start-ap or join-mesh commands instead.
1998          *
1999          * Monitors are special as they are normally slaved to
2000          * whatever else is going on, so they have their own special
2001          * operation to set the monitor channel if possible.
2002          */
2003         return !wdev ||
2004                 wdev->iftype == NL80211_IFTYPE_AP ||
2005                 wdev->iftype == NL80211_IFTYPE_MESH_POINT ||
2006                 wdev->iftype == NL80211_IFTYPE_MONITOR ||
2007                 wdev->iftype == NL80211_IFTYPE_P2P_GO;
2008 }
2009
2010 static int nl80211_parse_chandef(struct cfg80211_registered_device *rdev,
2011                                  struct genl_info *info,
2012                                  struct cfg80211_chan_def *chandef)
2013 {
2014         u32 control_freq;
2015
2016         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ])
2017                 return -EINVAL;
2018
2019         control_freq = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]);
2020
2021         chandef->chan = ieee80211_get_channel(&rdev->wiphy, control_freq);
2022         chandef->width = NL80211_CHAN_WIDTH_20_NOHT;
2023         chandef->center_freq1 = control_freq;
2024         chandef->center_freq2 = 0;
2025
2026         /* Primary channel not allowed */
2027         if (!chandef->chan || chandef->chan->flags & IEEE80211_CHAN_DISABLED)
2028                 return -EINVAL;
2029
2030         if (info->attrs[NL80211_ATTR_WIPHY_CHANNEL_TYPE]) {
2031                 enum nl80211_channel_type chantype;
2032
2033                 chantype = nla_get_u32(
2034                                 info->attrs[NL80211_ATTR_WIPHY_CHANNEL_TYPE]);
2035
2036                 switch (chantype) {
2037                 case NL80211_CHAN_NO_HT:
2038                 case NL80211_CHAN_HT20:
2039                 case NL80211_CHAN_HT40PLUS:
2040                 case NL80211_CHAN_HT40MINUS:
2041                         cfg80211_chandef_create(chandef, chandef->chan,
2042                                                 chantype);
2043                         break;
2044                 default:
2045                         return -EINVAL;
2046                 }
2047         } else if (info->attrs[NL80211_ATTR_CHANNEL_WIDTH]) {
2048                 chandef->width =
2049                         nla_get_u32(info->attrs[NL80211_ATTR_CHANNEL_WIDTH]);
2050                 if (info->attrs[NL80211_ATTR_CENTER_FREQ1])
2051                         chandef->center_freq1 =
2052                                 nla_get_u32(
2053                                         info->attrs[NL80211_ATTR_CENTER_FREQ1]);
2054                 if (info->attrs[NL80211_ATTR_CENTER_FREQ2])
2055                         chandef->center_freq2 =
2056                                 nla_get_u32(
2057                                         info->attrs[NL80211_ATTR_CENTER_FREQ2]);
2058         }
2059
2060         if (!cfg80211_chandef_valid(chandef))
2061                 return -EINVAL;
2062
2063         if (!cfg80211_chandef_usable(&rdev->wiphy, chandef,
2064                                      IEEE80211_CHAN_DISABLED))
2065                 return -EINVAL;
2066
2067         if ((chandef->width == NL80211_CHAN_WIDTH_5 ||
2068              chandef->width == NL80211_CHAN_WIDTH_10) &&
2069             !(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_5_10_MHZ))
2070                 return -EINVAL;
2071
2072         return 0;
2073 }
2074
2075 static int __nl80211_set_channel(struct cfg80211_registered_device *rdev,
2076                                  struct net_device *dev,
2077                                  struct genl_info *info)
2078 {
2079         struct cfg80211_chan_def chandef;
2080         int result;
2081         enum nl80211_iftype iftype = NL80211_IFTYPE_MONITOR;
2082         struct wireless_dev *wdev = NULL;
2083
2084         if (dev)
2085                 wdev = dev->ieee80211_ptr;
2086         if (!nl80211_can_set_dev_channel(wdev))
2087                 return -EOPNOTSUPP;
2088         if (wdev)
2089                 iftype = wdev->iftype;
2090
2091         result = nl80211_parse_chandef(rdev, info, &chandef);
2092         if (result)
2093                 return result;
2094
2095         switch (iftype) {
2096         case NL80211_IFTYPE_AP:
2097         case NL80211_IFTYPE_P2P_GO:
2098                 if (!cfg80211_reg_can_beacon_relax(&rdev->wiphy, &chandef,
2099                                                    iftype)) {
2100                         result = -EINVAL;
2101                         break;
2102                 }
2103                 if (wdev->beacon_interval) {
2104                         if (!dev || !rdev->ops->set_ap_chanwidth ||
2105                             !(rdev->wiphy.features &
2106                               NL80211_FEATURE_AP_MODE_CHAN_WIDTH_CHANGE)) {
2107                                 result = -EBUSY;
2108                                 break;
2109                         }
2110
2111                         /* Only allow dynamic channel width changes */
2112                         if (chandef.chan != wdev->preset_chandef.chan) {
2113                                 result = -EBUSY;
2114                                 break;
2115                         }
2116                         result = rdev_set_ap_chanwidth(rdev, dev, &chandef);
2117                         if (result)
2118                                 break;
2119                 }
2120                 wdev->preset_chandef = chandef;
2121                 result = 0;
2122                 break;
2123         case NL80211_IFTYPE_MESH_POINT:
2124                 result = cfg80211_set_mesh_channel(rdev, wdev, &chandef);
2125                 break;
2126         case NL80211_IFTYPE_MONITOR:
2127                 result = cfg80211_set_monitor_channel(rdev, &chandef);
2128                 break;
2129         default:
2130                 result = -EINVAL;
2131         }
2132
2133         return result;
2134 }
2135
2136 static int nl80211_set_channel(struct sk_buff *skb, struct genl_info *info)
2137 {
2138         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2139         struct net_device *netdev = info->user_ptr[1];
2140
2141         return __nl80211_set_channel(rdev, netdev, info);
2142 }
2143
2144 static int nl80211_set_wds_peer(struct sk_buff *skb, struct genl_info *info)
2145 {
2146         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2147         struct net_device *dev = info->user_ptr[1];
2148         struct wireless_dev *wdev = dev->ieee80211_ptr;
2149         const u8 *bssid;
2150
2151         if (!info->attrs[NL80211_ATTR_MAC])
2152                 return -EINVAL;
2153
2154         if (netif_running(dev))
2155                 return -EBUSY;
2156
2157         if (!rdev->ops->set_wds_peer)
2158                 return -EOPNOTSUPP;
2159
2160         if (wdev->iftype != NL80211_IFTYPE_WDS)
2161                 return -EOPNOTSUPP;
2162
2163         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
2164         return rdev_set_wds_peer(rdev, dev, bssid);
2165 }
2166
2167 static int nl80211_set_wiphy(struct sk_buff *skb, struct genl_info *info)
2168 {
2169         struct cfg80211_registered_device *rdev;
2170         struct net_device *netdev = NULL;
2171         struct wireless_dev *wdev;
2172         int result = 0, rem_txq_params = 0;
2173         struct nlattr *nl_txq_params;
2174         u32 changed;
2175         u8 retry_short = 0, retry_long = 0;
2176         u32 frag_threshold = 0, rts_threshold = 0;
2177         u8 coverage_class = 0;
2178
2179         ASSERT_RTNL();
2180
2181         /*
2182          * Try to find the wiphy and netdev. Normally this
2183          * function shouldn't need the netdev, but this is
2184          * done for backward compatibility -- previously
2185          * setting the channel was done per wiphy, but now
2186          * it is per netdev. Previous userland like hostapd
2187          * also passed a netdev to set_wiphy, so that it is
2188          * possible to let that go to the right netdev!
2189          */
2190
2191         if (info->attrs[NL80211_ATTR_IFINDEX]) {
2192                 int ifindex = nla_get_u32(info->attrs[NL80211_ATTR_IFINDEX]);
2193
2194                 netdev = __dev_get_by_index(genl_info_net(info), ifindex);
2195                 if (netdev && netdev->ieee80211_ptr)
2196                         rdev = wiphy_to_rdev(netdev->ieee80211_ptr->wiphy);
2197                 else
2198                         netdev = NULL;
2199         }
2200
2201         if (!netdev) {
2202                 rdev = __cfg80211_rdev_from_attrs(genl_info_net(info),
2203                                                   info->attrs);
2204                 if (IS_ERR(rdev))
2205                         return PTR_ERR(rdev);
2206                 wdev = NULL;
2207                 netdev = NULL;
2208                 result = 0;
2209         } else
2210                 wdev = netdev->ieee80211_ptr;
2211
2212         /*
2213          * end workaround code, by now the rdev is available
2214          * and locked, and wdev may or may not be NULL.
2215          */
2216
2217         if (info->attrs[NL80211_ATTR_WIPHY_NAME])
2218                 result = cfg80211_dev_rename(
2219                         rdev, nla_data(info->attrs[NL80211_ATTR_WIPHY_NAME]));
2220
2221         if (result)
2222                 return result;
2223
2224         if (info->attrs[NL80211_ATTR_WIPHY_TXQ_PARAMS]) {
2225                 struct ieee80211_txq_params txq_params;
2226                 struct nlattr *tb[NL80211_TXQ_ATTR_MAX + 1];
2227
2228                 if (!rdev->ops->set_txq_params)
2229                         return -EOPNOTSUPP;
2230
2231                 if (!netdev)
2232                         return -EINVAL;
2233
2234                 if (netdev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
2235                     netdev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
2236                         return -EINVAL;
2237
2238                 if (!netif_running(netdev))
2239                         return -ENETDOWN;
2240
2241                 nla_for_each_nested(nl_txq_params,
2242                                     info->attrs[NL80211_ATTR_WIPHY_TXQ_PARAMS],
2243                                     rem_txq_params) {
2244                         result = nla_parse(tb, NL80211_TXQ_ATTR_MAX,
2245                                            nla_data(nl_txq_params),
2246                                            nla_len(nl_txq_params),
2247                                            txq_params_policy);
2248                         if (result)
2249                                 return result;
2250                         result = parse_txq_params(tb, &txq_params);
2251                         if (result)
2252                                 return result;
2253
2254                         result = rdev_set_txq_params(rdev, netdev,
2255                                                      &txq_params);
2256                         if (result)
2257                                 return result;
2258                 }
2259         }
2260
2261         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
2262                 result = __nl80211_set_channel(
2263                         rdev,
2264                         nl80211_can_set_dev_channel(wdev) ? netdev : NULL,
2265                         info);
2266                 if (result)
2267                         return result;
2268         }
2269
2270         if (info->attrs[NL80211_ATTR_WIPHY_TX_POWER_SETTING]) {
2271                 struct wireless_dev *txp_wdev = wdev;
2272                 enum nl80211_tx_power_setting type;
2273                 int idx, mbm = 0;
2274
2275                 if (!(rdev->wiphy.features & NL80211_FEATURE_VIF_TXPOWER))
2276                         txp_wdev = NULL;
2277
2278                 if (!rdev->ops->set_tx_power)
2279                         return -EOPNOTSUPP;
2280
2281                 idx = NL80211_ATTR_WIPHY_TX_POWER_SETTING;
2282                 type = nla_get_u32(info->attrs[idx]);
2283
2284                 if (!info->attrs[NL80211_ATTR_WIPHY_TX_POWER_LEVEL] &&
2285                     (type != NL80211_TX_POWER_AUTOMATIC))
2286                         return -EINVAL;
2287
2288                 if (type != NL80211_TX_POWER_AUTOMATIC) {
2289                         idx = NL80211_ATTR_WIPHY_TX_POWER_LEVEL;
2290                         mbm = nla_get_u32(info->attrs[idx]);
2291                 }
2292
2293                 result = rdev_set_tx_power(rdev, txp_wdev, type, mbm);
2294                 if (result)
2295                         return result;
2296         }
2297
2298         if (info->attrs[NL80211_ATTR_WIPHY_ANTENNA_TX] &&
2299             info->attrs[NL80211_ATTR_WIPHY_ANTENNA_RX]) {
2300                 u32 tx_ant, rx_ant;
2301
2302                 if ((!rdev->wiphy.available_antennas_tx &&
2303                      !rdev->wiphy.available_antennas_rx) ||
2304                     !rdev->ops->set_antenna)
2305                         return -EOPNOTSUPP;
2306
2307                 tx_ant = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_ANTENNA_TX]);
2308                 rx_ant = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_ANTENNA_RX]);
2309
2310                 /* reject antenna configurations which don't match the
2311                  * available antenna masks, except for the "all" mask */
2312                 if ((~tx_ant && (tx_ant & ~rdev->wiphy.available_antennas_tx)) ||
2313                     (~rx_ant && (rx_ant & ~rdev->wiphy.available_antennas_rx)))
2314                         return -EINVAL;
2315
2316                 tx_ant = tx_ant & rdev->wiphy.available_antennas_tx;
2317                 rx_ant = rx_ant & rdev->wiphy.available_antennas_rx;
2318
2319                 result = rdev_set_antenna(rdev, tx_ant, rx_ant);
2320                 if (result)
2321                         return result;
2322         }
2323
2324         changed = 0;
2325
2326         if (info->attrs[NL80211_ATTR_WIPHY_RETRY_SHORT]) {
2327                 retry_short = nla_get_u8(
2328                         info->attrs[NL80211_ATTR_WIPHY_RETRY_SHORT]);
2329                 if (retry_short == 0)
2330                         return -EINVAL;
2331
2332                 changed |= WIPHY_PARAM_RETRY_SHORT;
2333         }
2334
2335         if (info->attrs[NL80211_ATTR_WIPHY_RETRY_LONG]) {
2336                 retry_long = nla_get_u8(
2337                         info->attrs[NL80211_ATTR_WIPHY_RETRY_LONG]);
2338                 if (retry_long == 0)
2339                         return -EINVAL;
2340
2341                 changed |= WIPHY_PARAM_RETRY_LONG;
2342         }
2343
2344         if (info->attrs[NL80211_ATTR_WIPHY_FRAG_THRESHOLD]) {
2345                 frag_threshold = nla_get_u32(
2346                         info->attrs[NL80211_ATTR_WIPHY_FRAG_THRESHOLD]);
2347                 if (frag_threshold < 256)
2348                         return -EINVAL;
2349
2350                 if (frag_threshold != (u32) -1) {
2351                         /*
2352                          * Fragments (apart from the last one) are required to
2353                          * have even length. Make the fragmentation code
2354                          * simpler by stripping LSB should someone try to use
2355                          * odd threshold value.
2356                          */
2357                         frag_threshold &= ~0x1;
2358                 }
2359                 changed |= WIPHY_PARAM_FRAG_THRESHOLD;
2360         }
2361
2362         if (info->attrs[NL80211_ATTR_WIPHY_RTS_THRESHOLD]) {
2363                 rts_threshold = nla_get_u32(
2364                         info->attrs[NL80211_ATTR_WIPHY_RTS_THRESHOLD]);
2365                 changed |= WIPHY_PARAM_RTS_THRESHOLD;
2366         }
2367
2368         if (info->attrs[NL80211_ATTR_WIPHY_COVERAGE_CLASS]) {
2369                 if (info->attrs[NL80211_ATTR_WIPHY_DYN_ACK])
2370                         return -EINVAL;
2371
2372                 coverage_class = nla_get_u8(
2373                         info->attrs[NL80211_ATTR_WIPHY_COVERAGE_CLASS]);
2374                 changed |= WIPHY_PARAM_COVERAGE_CLASS;
2375         }
2376
2377         if (info->attrs[NL80211_ATTR_WIPHY_DYN_ACK]) {
2378                 if (!(rdev->wiphy.features & NL80211_FEATURE_ACKTO_ESTIMATION))
2379                         return -EOPNOTSUPP;
2380
2381                 changed |= WIPHY_PARAM_DYN_ACK;
2382         }
2383
2384         if (changed) {
2385                 u8 old_retry_short, old_retry_long;
2386                 u32 old_frag_threshold, old_rts_threshold;
2387                 u8 old_coverage_class;
2388
2389                 if (!rdev->ops->set_wiphy_params)
2390                         return -EOPNOTSUPP;
2391
2392                 old_retry_short = rdev->wiphy.retry_short;
2393                 old_retry_long = rdev->wiphy.retry_long;
2394                 old_frag_threshold = rdev->wiphy.frag_threshold;
2395                 old_rts_threshold = rdev->wiphy.rts_threshold;
2396                 old_coverage_class = rdev->wiphy.coverage_class;
2397
2398                 if (changed & WIPHY_PARAM_RETRY_SHORT)
2399                         rdev->wiphy.retry_short = retry_short;
2400                 if (changed & WIPHY_PARAM_RETRY_LONG)
2401                         rdev->wiphy.retry_long = retry_long;
2402                 if (changed & WIPHY_PARAM_FRAG_THRESHOLD)
2403                         rdev->wiphy.frag_threshold = frag_threshold;
2404                 if (changed & WIPHY_PARAM_RTS_THRESHOLD)
2405                         rdev->wiphy.rts_threshold = rts_threshold;
2406                 if (changed & WIPHY_PARAM_COVERAGE_CLASS)
2407                         rdev->wiphy.coverage_class = coverage_class;
2408
2409                 result = rdev_set_wiphy_params(rdev, changed);
2410                 if (result) {
2411                         rdev->wiphy.retry_short = old_retry_short;
2412                         rdev->wiphy.retry_long = old_retry_long;
2413                         rdev->wiphy.frag_threshold = old_frag_threshold;
2414                         rdev->wiphy.rts_threshold = old_rts_threshold;
2415                         rdev->wiphy.coverage_class = old_coverage_class;
2416                         return result;
2417                 }
2418         }
2419         return 0;
2420 }
2421
2422 static inline u64 wdev_id(struct wireless_dev *wdev)
2423 {
2424         return (u64)wdev->identifier |
2425                ((u64)wiphy_to_rdev(wdev->wiphy)->wiphy_idx << 32);
2426 }
2427
2428 static int nl80211_send_chandef(struct sk_buff *msg,
2429                                 const struct cfg80211_chan_def *chandef)
2430 {
2431         if (WARN_ON(!cfg80211_chandef_valid(chandef)))
2432                 return -EINVAL;
2433
2434         if (nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ,
2435                         chandef->chan->center_freq))
2436                 return -ENOBUFS;
2437         switch (chandef->width) {
2438         case NL80211_CHAN_WIDTH_20_NOHT:
2439         case NL80211_CHAN_WIDTH_20:
2440         case NL80211_CHAN_WIDTH_40:
2441                 if (nla_put_u32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
2442                                 cfg80211_get_chandef_type(chandef)))
2443                         return -ENOBUFS;
2444                 break;
2445         default:
2446                 break;
2447         }
2448         if (nla_put_u32(msg, NL80211_ATTR_CHANNEL_WIDTH, chandef->width))
2449                 return -ENOBUFS;
2450         if (nla_put_u32(msg, NL80211_ATTR_CENTER_FREQ1, chandef->center_freq1))
2451                 return -ENOBUFS;
2452         if (chandef->center_freq2 &&
2453             nla_put_u32(msg, NL80211_ATTR_CENTER_FREQ2, chandef->center_freq2))
2454                 return -ENOBUFS;
2455         return 0;
2456 }
2457
2458 static int nl80211_send_iface(struct sk_buff *msg, u32 portid, u32 seq, int flags,
2459                               struct cfg80211_registered_device *rdev,
2460                               struct wireless_dev *wdev, bool removal)
2461 {
2462         struct net_device *dev = wdev->netdev;
2463         u8 cmd = NL80211_CMD_NEW_INTERFACE;
2464         void *hdr;
2465
2466         if (removal)
2467                 cmd = NL80211_CMD_DEL_INTERFACE;
2468
2469         hdr = nl80211hdr_put(msg, portid, seq, flags, cmd);
2470         if (!hdr)
2471                 return -1;
2472
2473         if (dev &&
2474             (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
2475              nla_put_string(msg, NL80211_ATTR_IFNAME, dev->name)))
2476                 goto nla_put_failure;
2477
2478         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
2479             nla_put_u32(msg, NL80211_ATTR_IFTYPE, wdev->iftype) ||
2480             nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
2481                               NL80211_ATTR_PAD) ||
2482             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, wdev_address(wdev)) ||
2483             nla_put_u32(msg, NL80211_ATTR_GENERATION,
2484                         rdev->devlist_generation ^
2485                         (cfg80211_rdev_list_generation << 2)))
2486                 goto nla_put_failure;
2487
2488         if (rdev->ops->get_channel) {
2489                 int ret;
2490                 struct cfg80211_chan_def chandef;
2491
2492                 ret = rdev_get_channel(rdev, wdev, &chandef);
2493                 if (ret == 0) {
2494                         if (nl80211_send_chandef(msg, &chandef))
2495                                 goto nla_put_failure;
2496                 }
2497         }
2498
2499         if (rdev->ops->get_tx_power) {
2500                 int dbm, ret;
2501
2502                 ret = rdev_get_tx_power(rdev, wdev, &dbm);
2503                 if (ret == 0 &&
2504                     nla_put_u32(msg, NL80211_ATTR_WIPHY_TX_POWER_LEVEL,
2505                                 DBM_TO_MBM(dbm)))
2506                         goto nla_put_failure;
2507         }
2508
2509         if (wdev->ssid_len) {
2510                 if (nla_put(msg, NL80211_ATTR_SSID, wdev->ssid_len, wdev->ssid))
2511                         goto nla_put_failure;
2512         }
2513
2514         genlmsg_end(msg, hdr);
2515         return 0;
2516
2517  nla_put_failure:
2518         genlmsg_cancel(msg, hdr);
2519         return -EMSGSIZE;
2520 }
2521
2522 static int nl80211_dump_interface(struct sk_buff *skb, struct netlink_callback *cb)
2523 {
2524         int wp_idx = 0;
2525         int if_idx = 0;
2526         int wp_start = cb->args[0];
2527         int if_start = cb->args[1];
2528         struct cfg80211_registered_device *rdev;
2529         struct wireless_dev *wdev;
2530
2531         rtnl_lock();
2532         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
2533                 if (!net_eq(wiphy_net(&rdev->wiphy), sock_net(skb->sk)))
2534                         continue;
2535                 if (wp_idx < wp_start) {
2536                         wp_idx++;
2537                         continue;
2538                 }
2539                 if_idx = 0;
2540
2541                 list_for_each_entry(wdev, &rdev->wiphy.wdev_list, list) {
2542                         if (if_idx < if_start) {
2543                                 if_idx++;
2544                                 continue;
2545                         }
2546                         if (nl80211_send_iface(skb, NETLINK_CB(cb->skb).portid,
2547                                                cb->nlh->nlmsg_seq, NLM_F_MULTI,
2548                                                rdev, wdev, false) < 0) {
2549                                 goto out;
2550                         }
2551                         if_idx++;
2552                 }
2553
2554                 wp_idx++;
2555         }
2556  out:
2557         rtnl_unlock();
2558
2559         cb->args[0] = wp_idx;
2560         cb->args[1] = if_idx;
2561
2562         return skb->len;
2563 }
2564
2565 static int nl80211_get_interface(struct sk_buff *skb, struct genl_info *info)
2566 {
2567         struct sk_buff *msg;
2568         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2569         struct wireless_dev *wdev = info->user_ptr[1];
2570
2571         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2572         if (!msg)
2573                 return -ENOMEM;
2574
2575         if (nl80211_send_iface(msg, info->snd_portid, info->snd_seq, 0,
2576                                rdev, wdev, false) < 0) {
2577                 nlmsg_free(msg);
2578                 return -ENOBUFS;
2579         }
2580
2581         return genlmsg_reply(msg, info);
2582 }
2583
2584 static const struct nla_policy mntr_flags_policy[NL80211_MNTR_FLAG_MAX + 1] = {
2585         [NL80211_MNTR_FLAG_FCSFAIL] = { .type = NLA_FLAG },
2586         [NL80211_MNTR_FLAG_PLCPFAIL] = { .type = NLA_FLAG },
2587         [NL80211_MNTR_FLAG_CONTROL] = { .type = NLA_FLAG },
2588         [NL80211_MNTR_FLAG_OTHER_BSS] = { .type = NLA_FLAG },
2589         [NL80211_MNTR_FLAG_COOK_FRAMES] = { .type = NLA_FLAG },
2590         [NL80211_MNTR_FLAG_ACTIVE] = { .type = NLA_FLAG },
2591 };
2592
2593 static int parse_monitor_flags(struct nlattr *nla, u32 *mntrflags)
2594 {
2595         struct nlattr *flags[NL80211_MNTR_FLAG_MAX + 1];
2596         int flag;
2597
2598         *mntrflags = 0;
2599
2600         if (!nla)
2601                 return -EINVAL;
2602
2603         if (nla_parse_nested(flags, NL80211_MNTR_FLAG_MAX,
2604                              nla, mntr_flags_policy))
2605                 return -EINVAL;
2606
2607         for (flag = 1; flag <= NL80211_MNTR_FLAG_MAX; flag++)
2608                 if (flags[flag])
2609                         *mntrflags |= (1<<flag);
2610
2611         return 0;
2612 }
2613
2614 static int nl80211_valid_4addr(struct cfg80211_registered_device *rdev,
2615                                struct net_device *netdev, u8 use_4addr,
2616                                enum nl80211_iftype iftype)
2617 {
2618         if (!use_4addr) {
2619                 if (netdev && (netdev->priv_flags & IFF_BRIDGE_PORT))
2620                         return -EBUSY;
2621                 return 0;
2622         }
2623
2624         switch (iftype) {
2625         case NL80211_IFTYPE_AP_VLAN:
2626                 if (rdev->wiphy.flags & WIPHY_FLAG_4ADDR_AP)
2627                         return 0;
2628                 break;
2629         case NL80211_IFTYPE_STATION:
2630                 if (rdev->wiphy.flags & WIPHY_FLAG_4ADDR_STATION)
2631                         return 0;
2632                 break;
2633         default:
2634                 break;
2635         }
2636
2637         return -EOPNOTSUPP;
2638 }
2639
2640 static int nl80211_set_interface(struct sk_buff *skb, struct genl_info *info)
2641 {
2642         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2643         struct vif_params params;
2644         int err;
2645         enum nl80211_iftype otype, ntype;
2646         struct net_device *dev = info->user_ptr[1];
2647         u32 _flags, *flags = NULL;
2648         bool change = false;
2649
2650         memset(&params, 0, sizeof(params));
2651
2652         otype = ntype = dev->ieee80211_ptr->iftype;
2653
2654         if (info->attrs[NL80211_ATTR_IFTYPE]) {
2655                 ntype = nla_get_u32(info->attrs[NL80211_ATTR_IFTYPE]);
2656                 if (otype != ntype)
2657                         change = true;
2658                 if (ntype > NL80211_IFTYPE_MAX)
2659                         return -EINVAL;
2660         }
2661
2662         if (info->attrs[NL80211_ATTR_MESH_ID]) {
2663                 struct wireless_dev *wdev = dev->ieee80211_ptr;
2664
2665                 if (ntype != NL80211_IFTYPE_MESH_POINT)
2666                         return -EINVAL;
2667                 if (netif_running(dev))
2668                         return -EBUSY;
2669
2670                 wdev_lock(wdev);
2671                 BUILD_BUG_ON(IEEE80211_MAX_SSID_LEN !=
2672                              IEEE80211_MAX_MESH_ID_LEN);
2673                 wdev->mesh_id_up_len =
2674                         nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
2675                 memcpy(wdev->ssid, nla_data(info->attrs[NL80211_ATTR_MESH_ID]),
2676                        wdev->mesh_id_up_len);
2677                 wdev_unlock(wdev);
2678         }
2679
2680         if (info->attrs[NL80211_ATTR_4ADDR]) {
2681                 params.use_4addr = !!nla_get_u8(info->attrs[NL80211_ATTR_4ADDR]);
2682                 change = true;
2683                 err = nl80211_valid_4addr(rdev, dev, params.use_4addr, ntype);
2684                 if (err)
2685                         return err;
2686         } else {
2687                 params.use_4addr = -1;
2688         }
2689
2690         if (info->attrs[NL80211_ATTR_MNTR_FLAGS]) {
2691                 if (ntype != NL80211_IFTYPE_MONITOR)
2692                         return -EINVAL;
2693                 err = parse_monitor_flags(info->attrs[NL80211_ATTR_MNTR_FLAGS],
2694                                           &_flags);
2695                 if (err)
2696                         return err;
2697
2698                 flags = &_flags;
2699                 change = true;
2700         }
2701
2702         if (info->attrs[NL80211_ATTR_MU_MIMO_GROUP_DATA]) {
2703                 const u8 *mumimo_groups;
2704                 u32 cap_flag = NL80211_EXT_FEATURE_MU_MIMO_AIR_SNIFFER;
2705
2706                 if (!wiphy_ext_feature_isset(&rdev->wiphy, cap_flag))
2707                         return -EOPNOTSUPP;
2708
2709                 mumimo_groups =
2710                         nla_data(info->attrs[NL80211_ATTR_MU_MIMO_GROUP_DATA]);
2711
2712                 /* bits 0 and 63 are reserved and must be zero */
2713                 if ((mumimo_groups[0] & BIT(7)) ||
2714                     (mumimo_groups[VHT_MUMIMO_GROUPS_DATA_LEN - 1] & BIT(0)))
2715                         return -EINVAL;
2716
2717                 memcpy(params.vht_mumimo_groups, mumimo_groups,
2718                        VHT_MUMIMO_GROUPS_DATA_LEN);
2719                 change = true;
2720         }
2721
2722         if (info->attrs[NL80211_ATTR_MU_MIMO_FOLLOW_MAC_ADDR]) {
2723                 u32 cap_flag = NL80211_EXT_FEATURE_MU_MIMO_AIR_SNIFFER;
2724
2725                 if (!wiphy_ext_feature_isset(&rdev->wiphy, cap_flag))
2726                         return -EOPNOTSUPP;
2727
2728                 nla_memcpy(params.macaddr,
2729                            info->attrs[NL80211_ATTR_MU_MIMO_FOLLOW_MAC_ADDR],
2730                            ETH_ALEN);
2731                 change = true;
2732         }
2733
2734         if (flags && (*flags & MONITOR_FLAG_ACTIVE) &&
2735             !(rdev->wiphy.features & NL80211_FEATURE_ACTIVE_MONITOR))
2736                 return -EOPNOTSUPP;
2737
2738         if (change)
2739                 err = cfg80211_change_iface(rdev, dev, ntype, flags, &params);
2740         else
2741                 err = 0;
2742
2743         if (!err && params.use_4addr != -1)
2744                 dev->ieee80211_ptr->use_4addr = params.use_4addr;
2745
2746         return err;
2747 }
2748
2749 static int nl80211_new_interface(struct sk_buff *skb, struct genl_info *info)
2750 {
2751         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2752         struct vif_params params;
2753         struct wireless_dev *wdev;
2754         struct sk_buff *msg;
2755         int err;
2756         enum nl80211_iftype type = NL80211_IFTYPE_UNSPECIFIED;
2757         u32 flags;
2758
2759         /* to avoid failing a new interface creation due to pending removal */
2760         cfg80211_destroy_ifaces(rdev);
2761
2762         memset(&params, 0, sizeof(params));
2763
2764         if (!info->attrs[NL80211_ATTR_IFNAME])
2765                 return -EINVAL;
2766
2767         if (info->attrs[NL80211_ATTR_IFTYPE]) {
2768                 type = nla_get_u32(info->attrs[NL80211_ATTR_IFTYPE]);
2769                 if (type > NL80211_IFTYPE_MAX)
2770                         return -EINVAL;
2771         }
2772
2773         if (!rdev->ops->add_virtual_intf ||
2774             !(rdev->wiphy.interface_modes & (1 << type)))
2775                 return -EOPNOTSUPP;
2776
2777         if ((type == NL80211_IFTYPE_P2P_DEVICE ||
2778              rdev->wiphy.features & NL80211_FEATURE_MAC_ON_CREATE) &&
2779             info->attrs[NL80211_ATTR_MAC]) {
2780                 nla_memcpy(params.macaddr, info->attrs[NL80211_ATTR_MAC],
2781                            ETH_ALEN);
2782                 if (!is_valid_ether_addr(params.macaddr))
2783                         return -EADDRNOTAVAIL;
2784         }
2785
2786         if (info->attrs[NL80211_ATTR_4ADDR]) {
2787                 params.use_4addr = !!nla_get_u8(info->attrs[NL80211_ATTR_4ADDR]);
2788                 err = nl80211_valid_4addr(rdev, NULL, params.use_4addr, type);
2789                 if (err)
2790                         return err;
2791         }
2792
2793         err = parse_monitor_flags(type == NL80211_IFTYPE_MONITOR ?
2794                                   info->attrs[NL80211_ATTR_MNTR_FLAGS] : NULL,
2795                                   &flags);
2796
2797         if (!err && (flags & MONITOR_FLAG_ACTIVE) &&
2798             !(rdev->wiphy.features & NL80211_FEATURE_ACTIVE_MONITOR))
2799                 return -EOPNOTSUPP;
2800
2801         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2802         if (!msg)
2803                 return -ENOMEM;
2804
2805         wdev = rdev_add_virtual_intf(rdev,
2806                                 nla_data(info->attrs[NL80211_ATTR_IFNAME]),
2807                                 NET_NAME_USER, type, err ? NULL : &flags,
2808                                 &params);
2809         if (WARN_ON(!wdev)) {
2810                 nlmsg_free(msg);
2811                 return -EPROTO;
2812         } else if (IS_ERR(wdev)) {
2813                 nlmsg_free(msg);
2814                 return PTR_ERR(wdev);
2815         }
2816
2817         if (info->attrs[NL80211_ATTR_SOCKET_OWNER])
2818                 wdev->owner_nlportid = info->snd_portid;
2819
2820         switch (type) {
2821         case NL80211_IFTYPE_MESH_POINT:
2822                 if (!info->attrs[NL80211_ATTR_MESH_ID])
2823                         break;
2824                 wdev_lock(wdev);
2825                 BUILD_BUG_ON(IEEE80211_MAX_SSID_LEN !=
2826                              IEEE80211_MAX_MESH_ID_LEN);
2827                 wdev->mesh_id_up_len =
2828                         nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
2829                 memcpy(wdev->ssid, nla_data(info->attrs[NL80211_ATTR_MESH_ID]),
2830                        wdev->mesh_id_up_len);
2831                 wdev_unlock(wdev);
2832                 break;
2833         case NL80211_IFTYPE_P2P_DEVICE:
2834                 /*
2835                  * P2P Device doesn't have a netdev, so doesn't go
2836                  * through the netdev notifier and must be added here
2837                  */
2838                 mutex_init(&wdev->mtx);
2839                 INIT_LIST_HEAD(&wdev->event_list);
2840                 spin_lock_init(&wdev->event_lock);
2841                 INIT_LIST_HEAD(&wdev->mgmt_registrations);
2842                 spin_lock_init(&wdev->mgmt_registrations_lock);
2843
2844                 wdev->identifier = ++rdev->wdev_id;
2845                 list_add_rcu(&wdev->list, &rdev->wiphy.wdev_list);
2846                 rdev->devlist_generation++;
2847                 break;
2848         default:
2849                 break;
2850         }
2851
2852         if (nl80211_send_iface(msg, info->snd_portid, info->snd_seq, 0,
2853                                rdev, wdev, false) < 0) {
2854                 nlmsg_free(msg);
2855                 return -ENOBUFS;
2856         }
2857
2858         /*
2859          * For wdevs which have no associated netdev object (e.g. of type
2860          * NL80211_IFTYPE_P2P_DEVICE), emit the NEW_INTERFACE event here.
2861          * For all other types, the event will be generated from the
2862          * netdev notifier
2863          */
2864         if (!wdev->netdev)
2865                 nl80211_notify_iface(rdev, wdev, NL80211_CMD_NEW_INTERFACE);
2866
2867         return genlmsg_reply(msg, info);
2868 }
2869
2870 static int nl80211_del_interface(struct sk_buff *skb, struct genl_info *info)
2871 {
2872         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2873         struct wireless_dev *wdev = info->user_ptr[1];
2874         struct sk_buff *msg;
2875         int status;
2876
2877         if (!rdev->ops->del_virtual_intf)
2878                 return -EOPNOTSUPP;
2879
2880         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2881         if (msg && nl80211_send_iface(msg, 0, 0, 0, rdev, wdev, true) < 0) {
2882                 nlmsg_free(msg);
2883                 msg = NULL;
2884         }
2885
2886         /*
2887          * If we remove a wireless device without a netdev then clear
2888          * user_ptr[1] so that nl80211_post_doit won't dereference it
2889          * to check if it needs to do dev_put(). Otherwise it crashes
2890          * since the wdev has been freed, unlike with a netdev where
2891          * we need the dev_put() for the netdev to really be freed.
2892          */
2893         if (!wdev->netdev)
2894                 info->user_ptr[1] = NULL;
2895
2896         status = rdev_del_virtual_intf(rdev, wdev);
2897         if (status >= 0 && msg)
2898                 genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy),
2899                                         msg, 0, NL80211_MCGRP_CONFIG,
2900                                         GFP_KERNEL);
2901         else
2902                 nlmsg_free(msg);
2903
2904         return status;
2905 }
2906
2907 static int nl80211_set_noack_map(struct sk_buff *skb, struct genl_info *info)
2908 {
2909         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2910         struct net_device *dev = info->user_ptr[1];
2911         u16 noack_map;
2912
2913         if (!info->attrs[NL80211_ATTR_NOACK_MAP])
2914                 return -EINVAL;
2915
2916         if (!rdev->ops->set_noack_map)
2917                 return -EOPNOTSUPP;
2918
2919         noack_map = nla_get_u16(info->attrs[NL80211_ATTR_NOACK_MAP]);
2920
2921         return rdev_set_noack_map(rdev, dev, noack_map);
2922 }
2923
2924 struct get_key_cookie {
2925         struct sk_buff *msg;
2926         int error;
2927         int idx;
2928 };
2929
2930 static void get_key_callback(void *c, struct key_params *params)
2931 {
2932         struct nlattr *key;
2933         struct get_key_cookie *cookie = c;
2934
2935         if ((params->key &&
2936              nla_put(cookie->msg, NL80211_ATTR_KEY_DATA,
2937                      params->key_len, params->key)) ||
2938             (params->seq &&
2939              nla_put(cookie->msg, NL80211_ATTR_KEY_SEQ,
2940                      params->seq_len, params->seq)) ||
2941             (params->cipher &&
2942              nla_put_u32(cookie->msg, NL80211_ATTR_KEY_CIPHER,
2943                          params->cipher)))
2944                 goto nla_put_failure;
2945
2946         key = nla_nest_start(cookie->msg, NL80211_ATTR_KEY);
2947         if (!key)
2948                 goto nla_put_failure;
2949
2950         if ((params->key &&
2951              nla_put(cookie->msg, NL80211_KEY_DATA,
2952                      params->key_len, params->key)) ||
2953             (params->seq &&
2954              nla_put(cookie->msg, NL80211_KEY_SEQ,
2955                      params->seq_len, params->seq)) ||
2956             (params->cipher &&
2957              nla_put_u32(cookie->msg, NL80211_KEY_CIPHER,
2958                          params->cipher)))
2959                 goto nla_put_failure;
2960
2961         if (nla_put_u8(cookie->msg, NL80211_ATTR_KEY_IDX, cookie->idx))
2962                 goto nla_put_failure;
2963
2964         nla_nest_end(cookie->msg, key);
2965
2966         return;
2967  nla_put_failure:
2968         cookie->error = 1;
2969 }
2970
2971 static int nl80211_get_key(struct sk_buff *skb, struct genl_info *info)
2972 {
2973         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2974         int err;
2975         struct net_device *dev = info->user_ptr[1];
2976         u8 key_idx = 0;
2977         const u8 *mac_addr = NULL;
2978         bool pairwise;
2979         struct get_key_cookie cookie = {
2980                 .error = 0,
2981         };
2982         void *hdr;
2983         struct sk_buff *msg;
2984
2985         if (info->attrs[NL80211_ATTR_KEY_IDX])
2986                 key_idx = nla_get_u8(info->attrs[NL80211_ATTR_KEY_IDX]);
2987
2988         if (key_idx > 5)
2989                 return -EINVAL;
2990
2991         if (info->attrs[NL80211_ATTR_MAC])
2992                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
2993
2994         pairwise = !!mac_addr;
2995         if (info->attrs[NL80211_ATTR_KEY_TYPE]) {
2996                 u32 kt = nla_get_u32(info->attrs[NL80211_ATTR_KEY_TYPE]);
2997
2998                 if (kt >= NUM_NL80211_KEYTYPES)
2999                         return -EINVAL;
3000                 if (kt != NL80211_KEYTYPE_GROUP &&
3001                     kt != NL80211_KEYTYPE_PAIRWISE)
3002                         return -EINVAL;
3003                 pairwise = kt == NL80211_KEYTYPE_PAIRWISE;
3004         }
3005
3006         if (!rdev->ops->get_key)
3007                 return -EOPNOTSUPP;
3008
3009         if (!pairwise && mac_addr && !(rdev->wiphy.flags & WIPHY_FLAG_IBSS_RSN))
3010                 return -ENOENT;
3011
3012         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3013         if (!msg)
3014                 return -ENOMEM;
3015
3016         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
3017                              NL80211_CMD_NEW_KEY);
3018         if (!hdr)
3019                 goto nla_put_failure;
3020
3021         cookie.msg = msg;
3022         cookie.idx = key_idx;
3023
3024         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
3025             nla_put_u8(msg, NL80211_ATTR_KEY_IDX, key_idx))
3026                 goto nla_put_failure;
3027         if (mac_addr &&
3028             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr))
3029                 goto nla_put_failure;
3030
3031         err = rdev_get_key(rdev, dev, key_idx, pairwise, mac_addr, &cookie,
3032                            get_key_callback);
3033
3034         if (err)
3035                 goto free_msg;
3036
3037         if (cookie.error)
3038                 goto nla_put_failure;
3039
3040         genlmsg_end(msg, hdr);
3041         return genlmsg_reply(msg, info);
3042
3043  nla_put_failure:
3044         err = -ENOBUFS;
3045  free_msg:
3046         nlmsg_free(msg);
3047         return err;
3048 }
3049
3050 static int nl80211_set_key(struct sk_buff *skb, struct genl_info *info)
3051 {
3052         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3053         struct key_parse key;
3054         int err;
3055         struct net_device *dev = info->user_ptr[1];
3056
3057         err = nl80211_parse_key(info, &key);
3058         if (err)
3059                 return err;
3060
3061         if (key.idx < 0)
3062                 return -EINVAL;
3063
3064         /* only support setting default key */
3065         if (!key.def && !key.defmgmt)
3066                 return -EINVAL;
3067
3068         wdev_lock(dev->ieee80211_ptr);
3069
3070         if (key.def) {
3071                 if (!rdev->ops->set_default_key) {
3072                         err = -EOPNOTSUPP;
3073                         goto out;
3074                 }
3075
3076                 err = nl80211_key_allowed(dev->ieee80211_ptr);
3077                 if (err)
3078                         goto out;
3079
3080                 err = rdev_set_default_key(rdev, dev, key.idx,
3081                                                  key.def_uni, key.def_multi);
3082
3083                 if (err)
3084                         goto out;
3085
3086 #ifdef CONFIG_CFG80211_WEXT
3087                 dev->ieee80211_ptr->wext.default_key = key.idx;
3088 #endif
3089         } else {
3090                 if (key.def_uni || !key.def_multi) {
3091                         err = -EINVAL;
3092                         goto out;
3093                 }
3094
3095                 if (!rdev->ops->set_default_mgmt_key) {
3096                         err = -EOPNOTSUPP;
3097                         goto out;
3098                 }
3099
3100                 err = nl80211_key_allowed(dev->ieee80211_ptr);
3101                 if (err)
3102                         goto out;
3103
3104                 err = rdev_set_default_mgmt_key(rdev, dev, key.idx);
3105                 if (err)
3106                         goto out;
3107
3108 #ifdef CONFIG_CFG80211_WEXT
3109                 dev->ieee80211_ptr->wext.default_mgmt_key = key.idx;
3110 #endif
3111         }
3112
3113  out:
3114         wdev_unlock(dev->ieee80211_ptr);
3115
3116         return err;
3117 }
3118
3119 static int nl80211_new_key(struct sk_buff *skb, struct genl_info *info)
3120 {
3121         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3122         int err;
3123         struct net_device *dev = info->user_ptr[1];
3124         struct key_parse key;
3125         const u8 *mac_addr = NULL;
3126
3127         err = nl80211_parse_key(info, &key);
3128         if (err)
3129                 return err;
3130
3131         if (!key.p.key)
3132                 return -EINVAL;
3133
3134         if (info->attrs[NL80211_ATTR_MAC])
3135                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
3136
3137         if (key.type == -1) {
3138                 if (mac_addr)
3139                         key.type = NL80211_KEYTYPE_PAIRWISE;
3140                 else
3141                         key.type = NL80211_KEYTYPE_GROUP;
3142         }
3143
3144         /* for now */
3145         if (key.type != NL80211_KEYTYPE_PAIRWISE &&
3146             key.type != NL80211_KEYTYPE_GROUP)
3147                 return -EINVAL;
3148
3149         if (!rdev->ops->add_key)
3150                 return -EOPNOTSUPP;
3151
3152         if (cfg80211_validate_key_settings(rdev, &key.p, key.idx,
3153                                            key.type == NL80211_KEYTYPE_PAIRWISE,
3154                                            mac_addr))
3155                 return -EINVAL;
3156
3157         wdev_lock(dev->ieee80211_ptr);
3158         err = nl80211_key_allowed(dev->ieee80211_ptr);
3159         if (!err)
3160                 err = rdev_add_key(rdev, dev, key.idx,
3161                                    key.type == NL80211_KEYTYPE_PAIRWISE,
3162                                     mac_addr, &key.p);
3163         wdev_unlock(dev->ieee80211_ptr);
3164
3165         return err;
3166 }
3167
3168 static int nl80211_del_key(struct sk_buff *skb, struct genl_info *info)
3169 {
3170         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3171         int err;
3172         struct net_device *dev = info->user_ptr[1];
3173         u8 *mac_addr = NULL;
3174         struct key_parse key;
3175
3176         err = nl80211_parse_key(info, &key);
3177         if (err)
3178                 return err;
3179
3180         if (info->attrs[NL80211_ATTR_MAC])
3181                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
3182
3183         if (key.type == -1) {
3184                 if (mac_addr)
3185                         key.type = NL80211_KEYTYPE_PAIRWISE;
3186                 else
3187                         key.type = NL80211_KEYTYPE_GROUP;
3188         }
3189
3190         /* for now */
3191         if (key.type != NL80211_KEYTYPE_PAIRWISE &&
3192             key.type != NL80211_KEYTYPE_GROUP)
3193                 return -EINVAL;
3194
3195         if (!rdev->ops->del_key)
3196                 return -EOPNOTSUPP;
3197
3198         wdev_lock(dev->ieee80211_ptr);
3199         err = nl80211_key_allowed(dev->ieee80211_ptr);
3200
3201         if (key.type == NL80211_KEYTYPE_GROUP && mac_addr &&
3202             !(rdev->wiphy.flags & WIPHY_FLAG_IBSS_RSN))
3203                 err = -ENOENT;
3204
3205         if (!err)
3206                 err = rdev_del_key(rdev, dev, key.idx,
3207                                    key.type == NL80211_KEYTYPE_PAIRWISE,
3208                                    mac_addr);
3209
3210 #ifdef CONFIG_CFG80211_WEXT
3211         if (!err) {
3212                 if (key.idx == dev->ieee80211_ptr->wext.default_key)
3213                         dev->ieee80211_ptr->wext.default_key = -1;
3214                 else if (key.idx == dev->ieee80211_ptr->wext.default_mgmt_key)
3215                         dev->ieee80211_ptr->wext.default_mgmt_key = -1;
3216         }
3217 #endif
3218         wdev_unlock(dev->ieee80211_ptr);
3219
3220         return err;
3221 }
3222
3223 /* This function returns an error or the number of nested attributes */
3224 static int validate_acl_mac_addrs(struct nlattr *nl_attr)
3225 {
3226         struct nlattr *attr;
3227         int n_entries = 0, tmp;
3228
3229         nla_for_each_nested(attr, nl_attr, tmp) {
3230                 if (nla_len(attr) != ETH_ALEN)
3231                         return -EINVAL;
3232
3233                 n_entries++;
3234         }
3235
3236         return n_entries;
3237 }
3238
3239 /*
3240  * This function parses ACL information and allocates memory for ACL data.
3241  * On successful return, the calling function is responsible to free the
3242  * ACL buffer returned by this function.
3243  */
3244 static struct cfg80211_acl_data *parse_acl_data(struct wiphy *wiphy,
3245                                                 struct genl_info *info)
3246 {
3247         enum nl80211_acl_policy acl_policy;
3248         struct nlattr *attr;
3249         struct cfg80211_acl_data *acl;
3250         int i = 0, n_entries, tmp;
3251
3252         if (!wiphy->max_acl_mac_addrs)
3253                 return ERR_PTR(-EOPNOTSUPP);
3254
3255         if (!info->attrs[NL80211_ATTR_ACL_POLICY])
3256                 return ERR_PTR(-EINVAL);
3257
3258         acl_policy = nla_get_u32(info->attrs[NL80211_ATTR_ACL_POLICY]);
3259         if (acl_policy != NL80211_ACL_POLICY_ACCEPT_UNLESS_LISTED &&
3260             acl_policy != NL80211_ACL_POLICY_DENY_UNLESS_LISTED)
3261                 return ERR_PTR(-EINVAL);
3262
3263         if (!info->attrs[NL80211_ATTR_MAC_ADDRS])
3264                 return ERR_PTR(-EINVAL);
3265
3266         n_entries = validate_acl_mac_addrs(info->attrs[NL80211_ATTR_MAC_ADDRS]);
3267         if (n_entries < 0)
3268                 return ERR_PTR(n_entries);
3269
3270         if (n_entries > wiphy->max_acl_mac_addrs)
3271                 return ERR_PTR(-ENOTSUPP);
3272
3273         acl = kzalloc(sizeof(*acl) + (sizeof(struct mac_address) * n_entries),
3274                       GFP_KERNEL);
3275         if (!acl)
3276                 return ERR_PTR(-ENOMEM);
3277
3278         nla_for_each_nested(attr, info->attrs[NL80211_ATTR_MAC_ADDRS], tmp) {
3279                 memcpy(acl->mac_addrs[i].addr, nla_data(attr), ETH_ALEN);
3280                 i++;
3281         }
3282
3283         acl->n_acl_entries = n_entries;
3284         acl->acl_policy = acl_policy;
3285
3286         return acl;
3287 }
3288
3289 static int nl80211_set_mac_acl(struct sk_buff *skb, struct genl_info *info)
3290 {
3291         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3292         struct net_device *dev = info->user_ptr[1];
3293         struct cfg80211_acl_data *acl;
3294         int err;
3295
3296         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
3297             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3298                 return -EOPNOTSUPP;
3299
3300         if (!dev->ieee80211_ptr->beacon_interval)
3301                 return -EINVAL;
3302
3303         acl = parse_acl_data(&rdev->wiphy, info);
3304         if (IS_ERR(acl))
3305                 return PTR_ERR(acl);
3306
3307         err = rdev_set_mac_acl(rdev, dev, acl);
3308
3309         kfree(acl);
3310
3311         return err;
3312 }
3313
3314 static int nl80211_parse_beacon(struct nlattr *attrs[],
3315                                 struct cfg80211_beacon_data *bcn)
3316 {
3317         bool haveinfo = false;
3318
3319         if (!is_valid_ie_attr(attrs[NL80211_ATTR_BEACON_TAIL]) ||
3320             !is_valid_ie_attr(attrs[NL80211_ATTR_IE]) ||
3321             !is_valid_ie_attr(attrs[NL80211_ATTR_IE_PROBE_RESP]) ||
3322             !is_valid_ie_attr(attrs[NL80211_ATTR_IE_ASSOC_RESP]))
3323                 return -EINVAL;
3324
3325         memset(bcn, 0, sizeof(*bcn));
3326
3327         if (attrs[NL80211_ATTR_BEACON_HEAD]) {
3328                 bcn->head = nla_data(attrs[NL80211_ATTR_BEACON_HEAD]);
3329                 bcn->head_len = nla_len(attrs[NL80211_ATTR_BEACON_HEAD]);
3330                 if (!bcn->head_len)
3331                         return -EINVAL;
3332                 haveinfo = true;
3333         }
3334
3335         if (attrs[NL80211_ATTR_BEACON_TAIL]) {
3336                 bcn->tail = nla_data(attrs[NL80211_ATTR_BEACON_TAIL]);
3337                 bcn->tail_len = nla_len(attrs[NL80211_ATTR_BEACON_TAIL]);
3338                 haveinfo = true;
3339         }
3340
3341         if (!haveinfo)
3342                 return -EINVAL;
3343
3344         if (attrs[NL80211_ATTR_IE]) {
3345                 bcn->beacon_ies = nla_data(attrs[NL80211_ATTR_IE]);
3346                 bcn->beacon_ies_len = nla_len(attrs[NL80211_ATTR_IE]);
3347         }
3348
3349         if (attrs[NL80211_ATTR_IE_PROBE_RESP]) {
3350                 bcn->proberesp_ies =
3351                         nla_data(attrs[NL80211_ATTR_IE_PROBE_RESP]);
3352                 bcn->proberesp_ies_len =
3353                         nla_len(attrs[NL80211_ATTR_IE_PROBE_RESP]);
3354         }
3355
3356         if (attrs[NL80211_ATTR_IE_ASSOC_RESP]) {
3357                 bcn->assocresp_ies =
3358                         nla_data(attrs[NL80211_ATTR_IE_ASSOC_RESP]);
3359                 bcn->assocresp_ies_len =
3360                         nla_len(attrs[NL80211_ATTR_IE_ASSOC_RESP]);
3361         }
3362
3363         if (attrs[NL80211_ATTR_PROBE_RESP]) {
3364                 bcn->probe_resp = nla_data(attrs[NL80211_ATTR_PROBE_RESP]);
3365                 bcn->probe_resp_len = nla_len(attrs[NL80211_ATTR_PROBE_RESP]);
3366         }
3367
3368         return 0;
3369 }
3370
3371 static bool nl80211_get_ap_channel(struct cfg80211_registered_device *rdev,
3372                                    struct cfg80211_ap_settings *params)
3373 {
3374         struct wireless_dev *wdev;
3375         bool ret = false;
3376
3377         list_for_each_entry(wdev, &rdev->wiphy.wdev_list, list) {
3378                 if (wdev->iftype != NL80211_IFTYPE_AP &&
3379                     wdev->iftype != NL80211_IFTYPE_P2P_GO)
3380                         continue;
3381
3382                 if (!wdev->preset_chandef.chan)
3383                         continue;
3384
3385                 params->chandef = wdev->preset_chandef;
3386                 ret = true;
3387                 break;
3388         }
3389
3390         return ret;
3391 }
3392
3393 static bool nl80211_valid_auth_type(struct cfg80211_registered_device *rdev,
3394                                     enum nl80211_auth_type auth_type,
3395                                     enum nl80211_commands cmd)
3396 {
3397         if (auth_type > NL80211_AUTHTYPE_MAX)
3398                 return false;
3399
3400         switch (cmd) {
3401         case NL80211_CMD_AUTHENTICATE:
3402                 if (!(rdev->wiphy.features & NL80211_FEATURE_SAE) &&
3403                     auth_type == NL80211_AUTHTYPE_SAE)
3404                         return false;
3405                 return true;
3406         case NL80211_CMD_CONNECT:
3407         case NL80211_CMD_START_AP:
3408                 /* SAE not supported yet */
3409                 if (auth_type == NL80211_AUTHTYPE_SAE)
3410                         return false;
3411                 return true;
3412         default:
3413                 return false;
3414         }
3415 }
3416
3417 static int nl80211_start_ap(struct sk_buff *skb, struct genl_info *info)
3418 {
3419         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3420         struct net_device *dev = info->user_ptr[1];
3421         struct wireless_dev *wdev = dev->ieee80211_ptr;
3422         struct cfg80211_ap_settings params;
3423         int err;
3424
3425         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
3426             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3427                 return -EOPNOTSUPP;
3428
3429         if (!rdev->ops->start_ap)
3430                 return -EOPNOTSUPP;
3431
3432         if (wdev->beacon_interval)
3433                 return -EALREADY;
3434
3435         memset(&params, 0, sizeof(params));
3436
3437         /* these are required for START_AP */
3438         if (!info->attrs[NL80211_ATTR_BEACON_INTERVAL] ||
3439             !info->attrs[NL80211_ATTR_DTIM_PERIOD] ||
3440             !info->attrs[NL80211_ATTR_BEACON_HEAD])
3441                 return -EINVAL;
3442
3443         err = nl80211_parse_beacon(info->attrs, &params.beacon);
3444         if (err)
3445                 return err;
3446
3447         params.beacon_interval =
3448                 nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
3449         params.dtim_period =
3450                 nla_get_u32(info->attrs[NL80211_ATTR_DTIM_PERIOD]);
3451
3452         err = cfg80211_validate_beacon_int(rdev, params.beacon_interval);
3453         if (err)
3454                 return err;
3455
3456         /*
3457          * In theory, some of these attributes should be required here
3458          * but since they were not used when the command was originally
3459          * added, keep them optional for old user space programs to let
3460          * them continue to work with drivers that do not need the
3461          * additional information -- drivers must check!
3462          */
3463         if (info->attrs[NL80211_ATTR_SSID]) {
3464                 params.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
3465                 params.ssid_len =
3466                         nla_len(info->attrs[NL80211_ATTR_SSID]);
3467                 if (params.ssid_len == 0 ||
3468                     params.ssid_len > IEEE80211_MAX_SSID_LEN)
3469                         return -EINVAL;
3470         }
3471
3472         if (info->attrs[NL80211_ATTR_HIDDEN_SSID]) {
3473                 params.hidden_ssid = nla_get_u32(
3474                         info->attrs[NL80211_ATTR_HIDDEN_SSID]);
3475                 if (params.hidden_ssid != NL80211_HIDDEN_SSID_NOT_IN_USE &&
3476                     params.hidden_ssid != NL80211_HIDDEN_SSID_ZERO_LEN &&
3477                     params.hidden_ssid != NL80211_HIDDEN_SSID_ZERO_CONTENTS)
3478                         return -EINVAL;
3479         }
3480
3481         params.privacy = !!info->attrs[NL80211_ATTR_PRIVACY];
3482
3483         if (info->attrs[NL80211_ATTR_AUTH_TYPE]) {
3484                 params.auth_type = nla_get_u32(
3485                         info->attrs[NL80211_ATTR_AUTH_TYPE]);
3486                 if (!nl80211_valid_auth_type(rdev, params.auth_type,
3487                                              NL80211_CMD_START_AP))
3488                         return -EINVAL;
3489         } else
3490                 params.auth_type = NL80211_AUTHTYPE_AUTOMATIC;
3491
3492         err = nl80211_crypto_settings(rdev, info, &params.crypto,
3493                                       NL80211_MAX_NR_CIPHER_SUITES);
3494         if (err)
3495                 return err;
3496
3497         if (info->attrs[NL80211_ATTR_INACTIVITY_TIMEOUT]) {
3498                 if (!(rdev->wiphy.features & NL80211_FEATURE_INACTIVITY_TIMER))
3499                         return -EOPNOTSUPP;
3500                 params.inactivity_timeout = nla_get_u16(
3501                         info->attrs[NL80211_ATTR_INACTIVITY_TIMEOUT]);
3502         }
3503
3504         if (info->attrs[NL80211_ATTR_P2P_CTWINDOW]) {
3505                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3506                         return -EINVAL;
3507                 params.p2p_ctwindow =
3508                         nla_get_u8(info->attrs[NL80211_ATTR_P2P_CTWINDOW]);
3509                 if (params.p2p_ctwindow > 127)
3510                         return -EINVAL;
3511                 if (params.p2p_ctwindow != 0 &&
3512                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_CTWIN))
3513                         return -EINVAL;
3514         }
3515
3516         if (info->attrs[NL80211_ATTR_P2P_OPPPS]) {
3517                 u8 tmp;
3518
3519                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3520                         return -EINVAL;
3521                 tmp = nla_get_u8(info->attrs[NL80211_ATTR_P2P_OPPPS]);
3522                 if (tmp > 1)
3523                         return -EINVAL;
3524                 params.p2p_opp_ps = tmp;
3525                 if (params.p2p_opp_ps != 0 &&
3526                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_OPPPS))
3527                         return -EINVAL;
3528         }
3529
3530         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
3531                 err = nl80211_parse_chandef(rdev, info, &params.chandef);
3532                 if (err)
3533                         return err;
3534         } else if (wdev->preset_chandef.chan) {
3535                 params.chandef = wdev->preset_chandef;
3536         } else if (!nl80211_get_ap_channel(rdev, &params))
3537                 return -EINVAL;
3538
3539         if (!cfg80211_reg_can_beacon_relax(&rdev->wiphy, &params.chandef,
3540                                            wdev->iftype))
3541                 return -EINVAL;
3542
3543         if (info->attrs[NL80211_ATTR_SMPS_MODE]) {
3544                 params.smps_mode =
3545                         nla_get_u8(info->attrs[NL80211_ATTR_SMPS_MODE]);
3546                 switch (params.smps_mode) {
3547                 case NL80211_SMPS_OFF:
3548                         break;
3549                 case NL80211_SMPS_STATIC:
3550                         if (!(rdev->wiphy.features &
3551                               NL80211_FEATURE_STATIC_SMPS))
3552                                 return -EINVAL;
3553                         break;
3554                 case NL80211_SMPS_DYNAMIC:
3555                         if (!(rdev->wiphy.features &
3556                               NL80211_FEATURE_DYNAMIC_SMPS))
3557                                 return -EINVAL;
3558                         break;
3559                 default:
3560                         return -EINVAL;
3561                 }
3562         } else {
3563                 params.smps_mode = NL80211_SMPS_OFF;
3564         }
3565
3566         params.pbss = nla_get_flag(info->attrs[NL80211_ATTR_PBSS]);
3567         if (params.pbss && !rdev->wiphy.bands[NL80211_BAND_60GHZ])
3568                 return -EOPNOTSUPP;
3569
3570         if (info->attrs[NL80211_ATTR_ACL_POLICY]) {
3571                 params.acl = parse_acl_data(&rdev->wiphy, info);
3572                 if (IS_ERR(params.acl))
3573                         return PTR_ERR(params.acl);
3574         }
3575
3576         wdev_lock(wdev);
3577         err = rdev_start_ap(rdev, dev, &params);
3578         if (!err) {
3579                 wdev->preset_chandef = params.chandef;
3580                 wdev->beacon_interval = params.beacon_interval;
3581                 wdev->chandef = params.chandef;
3582                 wdev->ssid_len = params.ssid_len;
3583                 memcpy(wdev->ssid, params.ssid, wdev->ssid_len);
3584         }
3585         wdev_unlock(wdev);
3586
3587         kfree(params.acl);
3588
3589         return err;
3590 }
3591
3592 static int nl80211_set_beacon(struct sk_buff *skb, struct genl_info *info)
3593 {
3594         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3595         struct net_device *dev = info->user_ptr[1];
3596         struct wireless_dev *wdev = dev->ieee80211_ptr;
3597         struct cfg80211_beacon_data params;
3598         int err;
3599
3600         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
3601             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3602                 return -EOPNOTSUPP;
3603
3604         if (!rdev->ops->change_beacon)
3605                 return -EOPNOTSUPP;
3606
3607         if (!wdev->beacon_interval)
3608                 return -EINVAL;
3609
3610         err = nl80211_parse_beacon(info->attrs, &params);
3611         if (err)
3612                 return err;
3613
3614         wdev_lock(wdev);
3615         err = rdev_change_beacon(rdev, dev, &params);
3616         wdev_unlock(wdev);
3617
3618         return err;
3619 }
3620
3621 static int nl80211_stop_ap(struct sk_buff *skb, struct genl_info *info)
3622 {
3623         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3624         struct net_device *dev = info->user_ptr[1];
3625
3626         return cfg80211_stop_ap(rdev, dev, false);
3627 }
3628
3629 static const struct nla_policy sta_flags_policy[NL80211_STA_FLAG_MAX + 1] = {
3630         [NL80211_STA_FLAG_AUTHORIZED] = { .type = NLA_FLAG },
3631         [NL80211_STA_FLAG_SHORT_PREAMBLE] = { .type = NLA_FLAG },
3632         [NL80211_STA_FLAG_WME] = { .type = NLA_FLAG },
3633         [NL80211_STA_FLAG_MFP] = { .type = NLA_FLAG },
3634         [NL80211_STA_FLAG_AUTHENTICATED] = { .type = NLA_FLAG },
3635         [NL80211_STA_FLAG_TDLS_PEER] = { .type = NLA_FLAG },
3636 };
3637
3638 static int parse_station_flags(struct genl_info *info,
3639                                enum nl80211_iftype iftype,
3640                                struct station_parameters *params)
3641 {
3642         struct nlattr *flags[NL80211_STA_FLAG_MAX + 1];
3643         struct nlattr *nla;
3644         int flag;
3645
3646         /*
3647          * Try parsing the new attribute first so userspace
3648          * can specify both for older kernels.
3649          */
3650         nla = info->attrs[NL80211_ATTR_STA_FLAGS2];
3651         if (nla) {
3652                 struct nl80211_sta_flag_update *sta_flags;
3653
3654                 sta_flags = nla_data(nla);
3655                 params->sta_flags_mask = sta_flags->mask;
3656                 params->sta_flags_set = sta_flags->set;
3657                 params->sta_flags_set &= params->sta_flags_mask;
3658                 if ((params->sta_flags_mask |
3659                      params->sta_flags_set) & BIT(__NL80211_STA_FLAG_INVALID))
3660                         return -EINVAL;
3661                 return 0;
3662         }
3663
3664         /* if present, parse the old attribute */
3665
3666         nla = info->attrs[NL80211_ATTR_STA_FLAGS];
3667         if (!nla)
3668                 return 0;
3669
3670         if (nla_parse_nested(flags, NL80211_STA_FLAG_MAX,
3671                              nla, sta_flags_policy))
3672                 return -EINVAL;
3673
3674         /*
3675          * Only allow certain flags for interface types so that
3676          * other attributes are silently ignored. Remember that
3677          * this is backward compatibility code with old userspace
3678          * and shouldn't be hit in other cases anyway.
3679          */
3680         switch (iftype) {
3681         case NL80211_IFTYPE_AP:
3682         case NL80211_IFTYPE_AP_VLAN:
3683         case NL80211_IFTYPE_P2P_GO:
3684                 params->sta_flags_mask = BIT(NL80211_STA_FLAG_AUTHORIZED) |
3685                                          BIT(NL80211_STA_FLAG_SHORT_PREAMBLE) |
3686                                          BIT(NL80211_STA_FLAG_WME) |
3687                                          BIT(NL80211_STA_FLAG_MFP);
3688                 break;
3689         case NL80211_IFTYPE_P2P_CLIENT:
3690         case NL80211_IFTYPE_STATION:
3691                 params->sta_flags_mask = BIT(NL80211_STA_FLAG_AUTHORIZED) |
3692                                          BIT(NL80211_STA_FLAG_TDLS_PEER);
3693                 break;
3694         case NL80211_IFTYPE_MESH_POINT:
3695                 params->sta_flags_mask = BIT(NL80211_STA_FLAG_AUTHENTICATED) |
3696                                          BIT(NL80211_STA_FLAG_MFP) |
3697                                          BIT(NL80211_STA_FLAG_AUTHORIZED);
3698         default:
3699                 return -EINVAL;
3700         }
3701
3702         for (flag = 1; flag <= NL80211_STA_FLAG_MAX; flag++) {
3703                 if (flags[flag]) {
3704                         params->sta_flags_set |= (1<<flag);
3705
3706                         /* no longer support new API additions in old API */
3707                         if (flag > NL80211_STA_FLAG_MAX_OLD_API)
3708                                 return -EINVAL;
3709                 }
3710         }
3711
3712         return 0;
3713 }
3714
3715 static bool nl80211_put_sta_rate(struct sk_buff *msg, struct rate_info *info,
3716                                  int attr)
3717 {
3718         struct nlattr *rate;
3719         u32 bitrate;
3720         u16 bitrate_compat;
3721         enum nl80211_attrs rate_flg;
3722
3723         rate = nla_nest_start(msg, attr);
3724         if (!rate)
3725                 return false;
3726
3727         /* cfg80211_calculate_bitrate will return 0 for mcs >= 32 */
3728         bitrate = cfg80211_calculate_bitrate(info);
3729         /* report 16-bit bitrate only if we can */
3730         bitrate_compat = bitrate < (1UL << 16) ? bitrate : 0;
3731         if (bitrate > 0 &&
3732             nla_put_u32(msg, NL80211_RATE_INFO_BITRATE32, bitrate))
3733                 return false;
3734         if (bitrate_compat > 0 &&
3735             nla_put_u16(msg, NL80211_RATE_INFO_BITRATE, bitrate_compat))
3736                 return false;
3737
3738         switch (info->bw) {
3739         case RATE_INFO_BW_5:
3740                 rate_flg = NL80211_RATE_INFO_5_MHZ_WIDTH;
3741                 break;
3742         case RATE_INFO_BW_10:
3743                 rate_flg = NL80211_RATE_INFO_10_MHZ_WIDTH;
3744                 break;
3745         default:
3746                 WARN_ON(1);
3747                 /* fall through */
3748         case RATE_INFO_BW_20:
3749                 rate_flg = 0;
3750                 break;
3751         case RATE_INFO_BW_40:
3752                 rate_flg = NL80211_RATE_INFO_40_MHZ_WIDTH;
3753                 break;
3754         case RATE_INFO_BW_80:
3755                 rate_flg = NL80211_RATE_INFO_80_MHZ_WIDTH;
3756                 break;
3757         case RATE_INFO_BW_160:
3758                 rate_flg = NL80211_RATE_INFO_160_MHZ_WIDTH;
3759                 break;
3760         }
3761
3762         if (rate_flg && nla_put_flag(msg, rate_flg))
3763                 return false;
3764
3765         if (info->flags & RATE_INFO_FLAGS_MCS) {
3766                 if (nla_put_u8(msg, NL80211_RATE_INFO_MCS, info->mcs))
3767                         return false;
3768                 if (info->flags & RATE_INFO_FLAGS_SHORT_GI &&
3769                     nla_put_flag(msg, NL80211_RATE_INFO_SHORT_GI))
3770                         return false;
3771         } else if (info->flags & RATE_INFO_FLAGS_VHT_MCS) {
3772                 if (nla_put_u8(msg, NL80211_RATE_INFO_VHT_MCS, info->mcs))
3773                         return false;
3774                 if (nla_put_u8(msg, NL80211_RATE_INFO_VHT_NSS, info->nss))
3775                         return false;
3776                 if (info->flags & RATE_INFO_FLAGS_SHORT_GI &&
3777                     nla_put_flag(msg, NL80211_RATE_INFO_SHORT_GI))
3778                         return false;
3779         }
3780
3781         nla_nest_end(msg, rate);
3782         return true;
3783 }
3784
3785 static bool nl80211_put_signal(struct sk_buff *msg, u8 mask, s8 *signal,
3786                                int id)
3787 {
3788         void *attr;
3789         int i = 0;
3790
3791         if (!mask)
3792                 return true;
3793
3794         attr = nla_nest_start(msg, id);
3795         if (!attr)
3796                 return false;
3797
3798         for (i = 0; i < IEEE80211_MAX_CHAINS; i++) {
3799                 if (!(mask & BIT(i)))
3800                         continue;
3801
3802                 if (nla_put_u8(msg, i, signal[i]))
3803                         return false;
3804         }
3805
3806         nla_nest_end(msg, attr);
3807
3808         return true;
3809 }
3810
3811 static int nl80211_send_station(struct sk_buff *msg, u32 cmd, u32 portid,
3812                                 u32 seq, int flags,
3813                                 struct cfg80211_registered_device *rdev,
3814                                 struct net_device *dev,
3815                                 const u8 *mac_addr, struct station_info *sinfo)
3816 {
3817         void *hdr;
3818         struct nlattr *sinfoattr, *bss_param;
3819
3820         hdr = nl80211hdr_put(msg, portid, seq, flags, cmd);
3821         if (!hdr)
3822                 return -1;
3823
3824         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
3825             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr) ||
3826             nla_put_u32(msg, NL80211_ATTR_GENERATION, sinfo->generation))
3827                 goto nla_put_failure;
3828
3829         sinfoattr = nla_nest_start(msg, NL80211_ATTR_STA_INFO);
3830         if (!sinfoattr)
3831                 goto nla_put_failure;
3832
3833 #define PUT_SINFO(attr, memb, type) do {                                \
3834         BUILD_BUG_ON(sizeof(type) == sizeof(u64));                      \
3835         if (sinfo->filled & (1ULL << NL80211_STA_INFO_ ## attr) &&      \
3836             nla_put_ ## type(msg, NL80211_STA_INFO_ ## attr,            \
3837                              sinfo->memb))                              \
3838                 goto nla_put_failure;                                   \
3839         } while (0)
3840 #define PUT_SINFO_U64(attr, memb) do {                                  \
3841         if (sinfo->filled & (1ULL << NL80211_STA_INFO_ ## attr) &&      \
3842             nla_put_u64_64bit(msg, NL80211_STA_INFO_ ## attr,           \
3843                               sinfo->memb, NL80211_STA_INFO_PAD))       \
3844                 goto nla_put_failure;                                   \
3845         } while (0)
3846
3847         PUT_SINFO(CONNECTED_TIME, connected_time, u32);
3848         PUT_SINFO(INACTIVE_TIME, inactive_time, u32);
3849
3850         if (sinfo->filled & (BIT(NL80211_STA_INFO_RX_BYTES) |
3851                              BIT(NL80211_STA_INFO_RX_BYTES64)) &&
3852             nla_put_u32(msg, NL80211_STA_INFO_RX_BYTES,
3853                         (u32)sinfo->rx_bytes))
3854                 goto nla_put_failure;
3855
3856         if (sinfo->filled & (BIT(NL80211_STA_INFO_TX_BYTES) |
3857                              BIT(NL80211_STA_INFO_TX_BYTES64)) &&
3858             nla_put_u32(msg, NL80211_STA_INFO_TX_BYTES,
3859                         (u32)sinfo->tx_bytes))
3860                 goto nla_put_failure;
3861
3862         PUT_SINFO_U64(RX_BYTES64, rx_bytes);
3863         PUT_SINFO_U64(TX_BYTES64, tx_bytes);
3864         PUT_SINFO(LLID, llid, u16);
3865         PUT_SINFO(PLID, plid, u16);
3866         PUT_SINFO(PLINK_STATE, plink_state, u8);
3867         PUT_SINFO_U64(RX_DURATION, rx_duration);
3868
3869         switch (rdev->wiphy.signal_type) {
3870         case CFG80211_SIGNAL_TYPE_MBM:
3871                 PUT_SINFO(SIGNAL, signal, u8);
3872                 PUT_SINFO(SIGNAL_AVG, signal_avg, u8);
3873                 break;
3874         default:
3875                 break;
3876         }
3877         if (sinfo->filled & BIT(NL80211_STA_INFO_CHAIN_SIGNAL)) {
3878                 if (!nl80211_put_signal(msg, sinfo->chains,
3879                                         sinfo->chain_signal,
3880                                         NL80211_STA_INFO_CHAIN_SIGNAL))
3881                         goto nla_put_failure;
3882         }
3883         if (sinfo->filled & BIT(NL80211_STA_INFO_CHAIN_SIGNAL_AVG)) {
3884                 if (!nl80211_put_signal(msg, sinfo->chains,
3885                                         sinfo->chain_signal_avg,
3886                                         NL80211_STA_INFO_CHAIN_SIGNAL_AVG))
3887                         goto nla_put_failure;
3888         }
3889         if (sinfo->filled & BIT(NL80211_STA_INFO_TX_BITRATE)) {
3890                 if (!nl80211_put_sta_rate(msg, &sinfo->txrate,
3891                                           NL80211_STA_INFO_TX_BITRATE))
3892                         goto nla_put_failure;
3893         }
3894         if (sinfo->filled & BIT(NL80211_STA_INFO_RX_BITRATE)) {
3895                 if (!nl80211_put_sta_rate(msg, &sinfo->rxrate,
3896                                           NL80211_STA_INFO_RX_BITRATE))
3897                         goto nla_put_failure;
3898         }
3899
3900         PUT_SINFO(RX_PACKETS, rx_packets, u32);
3901         PUT_SINFO(TX_PACKETS, tx_packets, u32);
3902         PUT_SINFO(TX_RETRIES, tx_retries, u32);
3903         PUT_SINFO(TX_FAILED, tx_failed, u32);
3904         PUT_SINFO(EXPECTED_THROUGHPUT, expected_throughput, u32);
3905         PUT_SINFO(BEACON_LOSS, beacon_loss_count, u32);
3906         PUT_SINFO(LOCAL_PM, local_pm, u32);
3907         PUT_SINFO(PEER_PM, peer_pm, u32);
3908         PUT_SINFO(NONPEER_PM, nonpeer_pm, u32);
3909
3910         if (sinfo->filled & BIT(NL80211_STA_INFO_BSS_PARAM)) {
3911                 bss_param = nla_nest_start(msg, NL80211_STA_INFO_BSS_PARAM);
3912                 if (!bss_param)
3913                         goto nla_put_failure;
3914
3915                 if (((sinfo->bss_param.flags & BSS_PARAM_FLAGS_CTS_PROT) &&
3916                      nla_put_flag(msg, NL80211_STA_BSS_PARAM_CTS_PROT)) ||
3917                     ((sinfo->bss_param.flags & BSS_PARAM_FLAGS_SHORT_PREAMBLE) &&
3918                      nla_put_flag(msg, NL80211_STA_BSS_PARAM_SHORT_PREAMBLE)) ||
3919                     ((sinfo->bss_param.flags & BSS_PARAM_FLAGS_SHORT_SLOT_TIME) &&
3920                      nla_put_flag(msg, NL80211_STA_BSS_PARAM_SHORT_SLOT_TIME)) ||
3921                     nla_put_u8(msg, NL80211_STA_BSS_PARAM_DTIM_PERIOD,
3922                                sinfo->bss_param.dtim_period) ||
3923                     nla_put_u16(msg, NL80211_STA_BSS_PARAM_BEACON_INTERVAL,
3924                                 sinfo->bss_param.beacon_interval))
3925                         goto nla_put_failure;
3926
3927                 nla_nest_end(msg, bss_param);
3928         }
3929         if ((sinfo->filled & BIT(NL80211_STA_INFO_STA_FLAGS)) &&
3930             nla_put(msg, NL80211_STA_INFO_STA_FLAGS,
3931                     sizeof(struct nl80211_sta_flag_update),
3932                     &sinfo->sta_flags))
3933                 goto nla_put_failure;
3934
3935         PUT_SINFO_U64(T_OFFSET, t_offset);
3936         PUT_SINFO_U64(RX_DROP_MISC, rx_dropped_misc);
3937         PUT_SINFO_U64(BEACON_RX, rx_beacon);
3938         PUT_SINFO(BEACON_SIGNAL_AVG, rx_beacon_signal_avg, u8);
3939
3940 #undef PUT_SINFO
3941 #undef PUT_SINFO_U64
3942
3943         if (sinfo->filled & BIT(NL80211_STA_INFO_TID_STATS)) {
3944                 struct nlattr *tidsattr;
3945                 int tid;
3946
3947                 tidsattr = nla_nest_start(msg, NL80211_STA_INFO_TID_STATS);
3948                 if (!tidsattr)
3949                         goto nla_put_failure;
3950
3951                 for (tid = 0; tid < IEEE80211_NUM_TIDS + 1; tid++) {
3952                         struct cfg80211_tid_stats *tidstats;
3953                         struct nlattr *tidattr;
3954
3955                         tidstats = &sinfo->pertid[tid];
3956
3957                         if (!tidstats->filled)
3958                                 continue;
3959
3960                         tidattr = nla_nest_start(msg, tid + 1);
3961                         if (!tidattr)
3962                                 goto nla_put_failure;
3963
3964 #define PUT_TIDVAL_U64(attr, memb) do {                                 \
3965         if (tidstats->filled & BIT(NL80211_TID_STATS_ ## attr) &&       \
3966             nla_put_u64_64bit(msg, NL80211_TID_STATS_ ## attr,          \
3967                               tidstats->memb, NL80211_TID_STATS_PAD))   \
3968                 goto nla_put_failure;                                   \
3969         } while (0)
3970
3971                         PUT_TIDVAL_U64(RX_MSDU, rx_msdu);
3972                         PUT_TIDVAL_U64(TX_MSDU, tx_msdu);
3973                         PUT_TIDVAL_U64(TX_MSDU_RETRIES, tx_msdu_retries);
3974                         PUT_TIDVAL_U64(TX_MSDU_FAILED, tx_msdu_failed);
3975
3976 #undef PUT_TIDVAL_U64
3977                         nla_nest_end(msg, tidattr);
3978                 }
3979
3980                 nla_nest_end(msg, tidsattr);
3981         }
3982
3983         nla_nest_end(msg, sinfoattr);
3984
3985         if (sinfo->assoc_req_ies_len &&
3986             nla_put(msg, NL80211_ATTR_IE, sinfo->assoc_req_ies_len,
3987                     sinfo->assoc_req_ies))
3988                 goto nla_put_failure;
3989
3990         genlmsg_end(msg, hdr);
3991         return 0;
3992
3993  nla_put_failure:
3994         genlmsg_cancel(msg, hdr);
3995         return -EMSGSIZE;
3996 }
3997
3998 static int nl80211_dump_station(struct sk_buff *skb,
3999                                 struct netlink_callback *cb)
4000 {
4001         struct station_info sinfo;
4002         struct cfg80211_registered_device *rdev;
4003         struct wireless_dev *wdev;
4004         u8 mac_addr[ETH_ALEN];
4005         int sta_idx = cb->args[2];
4006         int err;
4007
4008         err = nl80211_prepare_wdev_dump(skb, cb, &rdev, &wdev);
4009         if (err)
4010                 return err;
4011
4012         if (!wdev->netdev) {
4013                 err = -EINVAL;
4014                 goto out_err;
4015         }
4016
4017         if (!rdev->ops->dump_station) {
4018                 err = -EOPNOTSUPP;
4019                 goto out_err;
4020         }
4021
4022         while (1) {
4023                 memset(&sinfo, 0, sizeof(sinfo));
4024                 err = rdev_dump_station(rdev, wdev->netdev, sta_idx,
4025                                         mac_addr, &sinfo);
4026                 if (err == -ENOENT)
4027                         break;
4028                 if (err)
4029                         goto out_err;
4030
4031                 if (nl80211_send_station(skb, NL80211_CMD_NEW_STATION,
4032                                 NETLINK_CB(cb->skb).portid,
4033                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
4034                                 rdev, wdev->netdev, mac_addr,
4035                                 &sinfo) < 0)
4036                         goto out;
4037
4038                 sta_idx++;
4039         }
4040
4041  out:
4042         cb->args[2] = sta_idx;
4043         err = skb->len;
4044  out_err:
4045         nl80211_finish_wdev_dump(rdev);
4046
4047         return err;
4048 }
4049
4050 static int nl80211_get_station(struct sk_buff *skb, struct genl_info *info)
4051 {
4052         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4053         struct net_device *dev = info->user_ptr[1];
4054         struct station_info sinfo;
4055         struct sk_buff *msg;
4056         u8 *mac_addr = NULL;
4057         int err;
4058
4059         memset(&sinfo, 0, sizeof(sinfo));
4060
4061         if (!info->attrs[NL80211_ATTR_MAC])
4062                 return -EINVAL;
4063
4064         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
4065
4066         if (!rdev->ops->get_station)
4067                 return -EOPNOTSUPP;
4068
4069         err = rdev_get_station(rdev, dev, mac_addr, &sinfo);
4070         if (err)
4071                 return err;
4072
4073         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4074         if (!msg)
4075                 return -ENOMEM;
4076
4077         if (nl80211_send_station(msg, NL80211_CMD_NEW_STATION,
4078                                  info->snd_portid, info->snd_seq, 0,
4079                                  rdev, dev, mac_addr, &sinfo) < 0) {
4080                 nlmsg_free(msg);
4081                 return -ENOBUFS;
4082         }
4083
4084         return genlmsg_reply(msg, info);
4085 }
4086
4087 int cfg80211_check_station_change(struct wiphy *wiphy,
4088                                   struct station_parameters *params,
4089                                   enum cfg80211_station_type statype)
4090 {
4091         if (params->listen_interval != -1 &&
4092             statype != CFG80211_STA_AP_CLIENT_UNASSOC)
4093                 return -EINVAL;
4094
4095         if (params->support_p2p_ps != -1 &&
4096             statype != CFG80211_STA_AP_CLIENT_UNASSOC)
4097                 return -EINVAL;
4098
4099         if (params->aid &&
4100             !(params->sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)) &&
4101             statype != CFG80211_STA_AP_CLIENT_UNASSOC)
4102                 return -EINVAL;
4103
4104         /* When you run into this, adjust the code below for the new flag */
4105         BUILD_BUG_ON(NL80211_STA_FLAG_MAX != 7);
4106
4107         switch (statype) {
4108         case CFG80211_STA_MESH_PEER_KERNEL:
4109         case CFG80211_STA_MESH_PEER_USER:
4110                 /*
4111                  * No ignoring the TDLS flag here -- the userspace mesh
4112                  * code doesn't have the bug of including TDLS in the
4113                  * mask everywhere.
4114                  */
4115                 if (params->sta_flags_mask &
4116                                 ~(BIT(NL80211_STA_FLAG_AUTHENTICATED) |
4117                                   BIT(NL80211_STA_FLAG_MFP) |
4118                                   BIT(NL80211_STA_FLAG_AUTHORIZED)))
4119                         return -EINVAL;
4120                 break;
4121         case CFG80211_STA_TDLS_PEER_SETUP:
4122         case CFG80211_STA_TDLS_PEER_ACTIVE:
4123                 if (!(params->sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)))
4124                         return -EINVAL;
4125                 /* ignore since it can't change */
4126                 params->sta_flags_mask &= ~BIT(NL80211_STA_FLAG_TDLS_PEER);
4127                 break;
4128         default:
4129                 /* disallow mesh-specific things */
4130                 if (params->plink_action != NL80211_PLINK_ACTION_NO_ACTION)
4131                         return -EINVAL;
4132                 if (params->local_pm)
4133                         return -EINVAL;
4134                 if (params->sta_modify_mask & STATION_PARAM_APPLY_PLINK_STATE)
4135                         return -EINVAL;
4136         }
4137
4138         if (statype != CFG80211_STA_TDLS_PEER_SETUP &&
4139             statype != CFG80211_STA_TDLS_PEER_ACTIVE) {
4140                 /* TDLS can't be set, ... */
4141                 if (params->sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER))
4142                         return -EINVAL;
4143                 /*
4144                  * ... but don't bother the driver with it. This works around
4145                  * a hostapd/wpa_supplicant issue -- it always includes the
4146                  * TLDS_PEER flag in the mask even for AP mode.
4147                  */
4148                 params->sta_flags_mask &= ~BIT(NL80211_STA_FLAG_TDLS_PEER);
4149         }
4150
4151         if (statype != CFG80211_STA_TDLS_PEER_SETUP &&
4152             statype != CFG80211_STA_AP_CLIENT_UNASSOC) {
4153                 /* reject other things that can't change */
4154                 if (params->sta_modify_mask & STATION_PARAM_APPLY_UAPSD)
4155                         return -EINVAL;
4156                 if (params->sta_modify_mask & STATION_PARAM_APPLY_CAPABILITY)
4157                         return -EINVAL;
4158                 if (params->supported_rates)
4159                         return -EINVAL;
4160                 if (params->ext_capab || params->ht_capa || params->vht_capa)
4161                         return -EINVAL;
4162         }
4163
4164         if (statype != CFG80211_STA_AP_CLIENT &&
4165             statype != CFG80211_STA_AP_CLIENT_UNASSOC) {
4166                 if (params->vlan)
4167                         return -EINVAL;
4168         }
4169
4170         switch (statype) {
4171         case CFG80211_STA_AP_MLME_CLIENT:
4172                 /* Use this only for authorizing/unauthorizing a station */
4173                 if (!(params->sta_flags_mask & BIT(NL80211_STA_FLAG_AUTHORIZED)))
4174                         return -EOPNOTSUPP;
4175                 break;
4176         case CFG80211_STA_AP_CLIENT:
4177         case CFG80211_STA_AP_CLIENT_UNASSOC:
4178                 /* accept only the listed bits */
4179                 if (params->sta_flags_mask &
4180                                 ~(BIT(NL80211_STA_FLAG_AUTHORIZED) |
4181                                   BIT(NL80211_STA_FLAG_AUTHENTICATED) |
4182                                   BIT(NL80211_STA_FLAG_ASSOCIATED) |
4183                                   BIT(NL80211_STA_FLAG_SHORT_PREAMBLE) |
4184                                   BIT(NL80211_STA_FLAG_WME) |
4185                                   BIT(NL80211_STA_FLAG_MFP)))
4186                         return -EINVAL;
4187
4188                 /* but authenticated/associated only if driver handles it */
4189                 if (!(wiphy->features & NL80211_FEATURE_FULL_AP_CLIENT_STATE) &&
4190                     params->sta_flags_mask &
4191                                 (BIT(NL80211_STA_FLAG_AUTHENTICATED) |
4192                                  BIT(NL80211_STA_FLAG_ASSOCIATED)))
4193                         return -EINVAL;
4194                 break;
4195         case CFG80211_STA_IBSS:
4196         case CFG80211_STA_AP_STA:
4197                 /* reject any changes other than AUTHORIZED */
4198                 if (params->sta_flags_mask & ~BIT(NL80211_STA_FLAG_AUTHORIZED))
4199                         return -EINVAL;
4200                 break;
4201         case CFG80211_STA_TDLS_PEER_SETUP:
4202                 /* reject any changes other than AUTHORIZED or WME */
4203                 if (params->sta_flags_mask & ~(BIT(NL80211_STA_FLAG_AUTHORIZED) |
4204                                                BIT(NL80211_STA_FLAG_WME)))
4205                         return -EINVAL;
4206                 /* force (at least) rates when authorizing */
4207                 if (params->sta_flags_set & BIT(NL80211_STA_FLAG_AUTHORIZED) &&
4208                     !params->supported_rates)
4209                         return -EINVAL;
4210                 break;
4211         case CFG80211_STA_TDLS_PEER_ACTIVE:
4212                 /* reject any changes */
4213                 return -EINVAL;
4214         case CFG80211_STA_MESH_PEER_KERNEL:
4215                 if (params->sta_modify_mask & STATION_PARAM_APPLY_PLINK_STATE)
4216                         return -EINVAL;
4217                 break;
4218         case CFG80211_STA_MESH_PEER_USER:
4219                 if (params->plink_action != NL80211_PLINK_ACTION_NO_ACTION &&
4220                     params->plink_action != NL80211_PLINK_ACTION_BLOCK)
4221                         return -EINVAL;
4222                 break;
4223         }
4224
4225         return 0;
4226 }
4227 EXPORT_SYMBOL(cfg80211_check_station_change);
4228
4229 /*
4230  * Get vlan interface making sure it is running and on the right wiphy.
4231  */
4232 static struct net_device *get_vlan(struct genl_info *info,
4233                                    struct cfg80211_registered_device *rdev)
4234 {
4235         struct nlattr *vlanattr = info->attrs[NL80211_ATTR_STA_VLAN];
4236         struct net_device *v;
4237         int ret;
4238
4239         if (!vlanattr)
4240                 return NULL;
4241
4242         v = dev_get_by_index(genl_info_net(info), nla_get_u32(vlanattr));
4243         if (!v)
4244                 return ERR_PTR(-ENODEV);
4245
4246         if (!v->ieee80211_ptr || v->ieee80211_ptr->wiphy != &rdev->wiphy) {
4247                 ret = -EINVAL;
4248                 goto error;
4249         }
4250
4251         if (v->ieee80211_ptr->iftype != NL80211_IFTYPE_AP_VLAN &&
4252             v->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
4253             v->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO) {
4254                 ret = -EINVAL;
4255                 goto error;
4256         }
4257
4258         if (!netif_running(v)) {
4259                 ret = -ENETDOWN;
4260                 goto error;
4261         }
4262
4263         return v;
4264  error:
4265         dev_put(v);
4266         return ERR_PTR(ret);
4267 }
4268
4269 static const struct nla_policy
4270 nl80211_sta_wme_policy[NL80211_STA_WME_MAX + 1] = {
4271         [NL80211_STA_WME_UAPSD_QUEUES] = { .type = NLA_U8 },
4272         [NL80211_STA_WME_MAX_SP] = { .type = NLA_U8 },
4273 };
4274
4275 static int nl80211_parse_sta_wme(struct genl_info *info,
4276                                  struct station_parameters *params)
4277 {
4278         struct nlattr *tb[NL80211_STA_WME_MAX + 1];
4279         struct nlattr *nla;
4280         int err;
4281
4282         /* parse WME attributes if present */
4283         if (!info->attrs[NL80211_ATTR_STA_WME])
4284                 return 0;
4285
4286         nla = info->attrs[NL80211_ATTR_STA_WME];
4287         err = nla_parse_nested(tb, NL80211_STA_WME_MAX, nla,
4288                                nl80211_sta_wme_policy);
4289         if (err)
4290                 return err;
4291
4292         if (tb[NL80211_STA_WME_UAPSD_QUEUES])
4293                 params->uapsd_queues = nla_get_u8(
4294                         tb[NL80211_STA_WME_UAPSD_QUEUES]);
4295         if (params->uapsd_queues & ~IEEE80211_WMM_IE_STA_QOSINFO_AC_MASK)
4296                 return -EINVAL;
4297
4298         if (tb[NL80211_STA_WME_MAX_SP])
4299                 params->max_sp = nla_get_u8(tb[NL80211_STA_WME_MAX_SP]);
4300
4301         if (params->max_sp & ~IEEE80211_WMM_IE_STA_QOSINFO_SP_MASK)
4302                 return -EINVAL;
4303
4304         params->sta_modify_mask |= STATION_PARAM_APPLY_UAPSD;
4305
4306         return 0;
4307 }
4308
4309 static int nl80211_parse_sta_channel_info(struct genl_info *info,
4310                                       struct station_parameters *params)
4311 {
4312         if (info->attrs[NL80211_ATTR_STA_SUPPORTED_CHANNELS]) {
4313                 params->supported_channels =
4314                      nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_CHANNELS]);
4315                 params->supported_channels_len =
4316                      nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_CHANNELS]);
4317                 /*
4318                  * Need to include at least one (first channel, number of
4319                  * channels) tuple for each subband, and must have proper
4320                  * tuples for the rest of the data as well.
4321                  */
4322                 if (params->supported_channels_len < 2)
4323                         return -EINVAL;
4324                 if (params->supported_channels_len % 2)
4325                         return -EINVAL;
4326         }
4327
4328         if (info->attrs[NL80211_ATTR_STA_SUPPORTED_OPER_CLASSES]) {
4329                 params->supported_oper_classes =
4330                  nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_OPER_CLASSES]);
4331                 params->supported_oper_classes_len =
4332                   nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_OPER_CLASSES]);
4333                 /*
4334                  * The value of the Length field of the Supported Operating
4335                  * Classes element is between 2 and 253.
4336                  */
4337                 if (params->supported_oper_classes_len < 2 ||
4338                     params->supported_oper_classes_len > 253)
4339                         return -EINVAL;
4340         }
4341         return 0;
4342 }
4343
4344 static int nl80211_set_station_tdls(struct genl_info *info,
4345                                     struct station_parameters *params)
4346 {
4347         int err;
4348         /* Dummy STA entry gets updated once the peer capabilities are known */
4349         if (info->attrs[NL80211_ATTR_PEER_AID])
4350                 params->aid = nla_get_u16(info->attrs[NL80211_ATTR_PEER_AID]);
4351         if (info->attrs[NL80211_ATTR_HT_CAPABILITY])
4352                 params->ht_capa =
4353                         nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]);
4354         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY])
4355                 params->vht_capa =
4356                         nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY]);
4357
4358         err = nl80211_parse_sta_channel_info(info, params);
4359         if (err)
4360                 return err;
4361
4362         return nl80211_parse_sta_wme(info, params);
4363 }
4364
4365 static int nl80211_set_station(struct sk_buff *skb, struct genl_info *info)
4366 {
4367         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4368         struct net_device *dev = info->user_ptr[1];
4369         struct station_parameters params;
4370         u8 *mac_addr;
4371         int err;
4372
4373         memset(&params, 0, sizeof(params));
4374
4375         if (!rdev->ops->change_station)
4376                 return -EOPNOTSUPP;
4377
4378         /*
4379          * AID and listen_interval properties can be set only for unassociated
4380          * station. Include these parameters here and will check them in
4381          * cfg80211_check_station_change().
4382          */
4383         if (info->attrs[NL80211_ATTR_STA_AID])
4384                 params.aid = nla_get_u16(info->attrs[NL80211_ATTR_STA_AID]);
4385
4386         if (info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL])
4387                 params.listen_interval =
4388                      nla_get_u16(info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL]);
4389         else
4390                 params.listen_interval = -1;
4391
4392         if (info->attrs[NL80211_ATTR_STA_SUPPORT_P2P_PS]) {
4393                 u8 tmp;
4394
4395                 tmp = nla_get_u8(info->attrs[NL80211_ATTR_STA_SUPPORT_P2P_PS]);
4396                 if (tmp >= NUM_NL80211_P2P_PS_STATUS)
4397                         return -EINVAL;
4398
4399                 params.support_p2p_ps = tmp;
4400         } else {
4401                 params.support_p2p_ps = -1;
4402         }
4403
4404         if (!info->attrs[NL80211_ATTR_MAC])
4405                 return -EINVAL;
4406
4407         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
4408
4409         if (info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]) {
4410                 params.supported_rates =
4411                         nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
4412                 params.supported_rates_len =
4413                         nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
4414         }
4415
4416         if (info->attrs[NL80211_ATTR_STA_CAPABILITY]) {
4417                 params.capability =
4418                         nla_get_u16(info->attrs[NL80211_ATTR_STA_CAPABILITY]);
4419                 params.sta_modify_mask |= STATION_PARAM_APPLY_CAPABILITY;
4420         }
4421
4422         if (info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]) {
4423                 params.ext_capab =
4424                         nla_data(info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]);
4425                 params.ext_capab_len =
4426                         nla_len(info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]);
4427         }
4428
4429         if (parse_station_flags(info, dev->ieee80211_ptr->iftype, &params))
4430                 return -EINVAL;
4431
4432         if (info->attrs[NL80211_ATTR_STA_PLINK_ACTION]) {
4433                 params.plink_action =
4434                         nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_ACTION]);
4435                 if (params.plink_action >= NUM_NL80211_PLINK_ACTIONS)
4436                         return -EINVAL;
4437         }
4438
4439         if (info->attrs[NL80211_ATTR_STA_PLINK_STATE]) {
4440                 params.plink_state =
4441                         nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_STATE]);
4442                 if (params.plink_state >= NUM_NL80211_PLINK_STATES)
4443                         return -EINVAL;
4444                 if (info->attrs[NL80211_ATTR_MESH_PEER_AID]) {
4445                         params.peer_aid = nla_get_u16(
4446                                 info->attrs[NL80211_ATTR_MESH_PEER_AID]);
4447                         if (params.peer_aid > IEEE80211_MAX_AID)
4448                                 return -EINVAL;
4449                 }
4450                 params.sta_modify_mask |= STATION_PARAM_APPLY_PLINK_STATE;
4451         }
4452
4453         if (info->attrs[NL80211_ATTR_LOCAL_MESH_POWER_MODE]) {
4454                 enum nl80211_mesh_power_mode pm = nla_get_u32(
4455                         info->attrs[NL80211_ATTR_LOCAL_MESH_POWER_MODE]);
4456
4457                 if (pm <= NL80211_MESH_POWER_UNKNOWN ||
4458                     pm > NL80211_MESH_POWER_MAX)
4459                         return -EINVAL;
4460
4461                 params.local_pm = pm;
4462         }
4463
4464         /* Include parameters for TDLS peer (will check later) */
4465         err = nl80211_set_station_tdls(info, &params);
4466         if (err)
4467                 return err;
4468
4469         params.vlan = get_vlan(info, rdev);
4470         if (IS_ERR(params.vlan))
4471                 return PTR_ERR(params.vlan);
4472
4473         switch (dev->ieee80211_ptr->iftype) {
4474         case NL80211_IFTYPE_AP:
4475         case NL80211_IFTYPE_AP_VLAN:
4476         case NL80211_IFTYPE_P2P_GO:
4477         case NL80211_IFTYPE_P2P_CLIENT:
4478         case NL80211_IFTYPE_STATION:
4479         case NL80211_IFTYPE_ADHOC:
4480         case NL80211_IFTYPE_MESH_POINT:
4481                 break;
4482         default:
4483                 err = -EOPNOTSUPP;
4484                 goto out_put_vlan;
4485         }
4486
4487         /* driver will call cfg80211_check_station_change() */
4488         err = rdev_change_station(rdev, dev, mac_addr, &params);
4489
4490  out_put_vlan:
4491         if (params.vlan)
4492                 dev_put(params.vlan);
4493
4494         return err;
4495 }
4496
4497 static int nl80211_new_station(struct sk_buff *skb, struct genl_info *info)
4498 {
4499         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4500         int err;
4501         struct net_device *dev = info->user_ptr[1];
4502         struct station_parameters params;
4503         u8 *mac_addr = NULL;
4504         u32 auth_assoc = BIT(NL80211_STA_FLAG_AUTHENTICATED) |
4505                          BIT(NL80211_STA_FLAG_ASSOCIATED);
4506
4507         memset(&params, 0, sizeof(params));
4508
4509         if (!rdev->ops->add_station)
4510                 return -EOPNOTSUPP;
4511
4512         if (!info->attrs[NL80211_ATTR_MAC])
4513                 return -EINVAL;
4514
4515         if (!info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL])
4516                 return -EINVAL;
4517
4518         if (!info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES])
4519                 return -EINVAL;
4520
4521         if (!info->attrs[NL80211_ATTR_STA_AID] &&
4522             !info->attrs[NL80211_ATTR_PEER_AID])
4523                 return -EINVAL;
4524
4525         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
4526         params.supported_rates =
4527                 nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
4528         params.supported_rates_len =
4529                 nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
4530         params.listen_interval =
4531                 nla_get_u16(info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL]);
4532
4533         if (info->attrs[NL80211_ATTR_STA_SUPPORT_P2P_PS]) {
4534                 u8 tmp;
4535
4536                 tmp = nla_get_u8(info->attrs[NL80211_ATTR_STA_SUPPORT_P2P_PS]);
4537                 if (tmp >= NUM_NL80211_P2P_PS_STATUS)
4538                         return -EINVAL;
4539
4540                 params.support_p2p_ps = tmp;
4541         } else {
4542                 /*
4543                  * if not specified, assume it's supported for P2P GO interface,
4544                  * and is NOT supported for AP interface
4545                  */
4546                 params.support_p2p_ps =
4547                         dev->ieee80211_ptr->iftype == NL80211_IFTYPE_P2P_GO;
4548         }
4549
4550         if (info->attrs[NL80211_ATTR_PEER_AID])
4551                 params.aid = nla_get_u16(info->attrs[NL80211_ATTR_PEER_AID]);
4552         else
4553                 params.aid = nla_get_u16(info->attrs[NL80211_ATTR_STA_AID]);
4554         if (!params.aid || params.aid > IEEE80211_MAX_AID)
4555                 return -EINVAL;
4556
4557         if (info->attrs[NL80211_ATTR_STA_CAPABILITY]) {
4558                 params.capability =
4559                         nla_get_u16(info->attrs[NL80211_ATTR_STA_CAPABILITY]);
4560                 params.sta_modify_mask |= STATION_PARAM_APPLY_CAPABILITY;
4561         }
4562
4563         if (info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]) {
4564                 params.ext_capab =
4565                         nla_data(info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]);
4566                 params.ext_capab_len =
4567                         nla_len(info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]);
4568         }
4569
4570         if (info->attrs[NL80211_ATTR_HT_CAPABILITY])
4571                 params.ht_capa =
4572                         nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]);
4573
4574         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY])
4575                 params.vht_capa =
4576                         nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY]);
4577
4578         if (info->attrs[NL80211_ATTR_OPMODE_NOTIF]) {
4579                 params.opmode_notif_used = true;
4580                 params.opmode_notif =
4581                         nla_get_u8(info->attrs[NL80211_ATTR_OPMODE_NOTIF]);
4582         }
4583
4584         if (info->attrs[NL80211_ATTR_STA_PLINK_ACTION]) {
4585                 params.plink_action =
4586                         nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_ACTION]);
4587                 if (params.plink_action >= NUM_NL80211_PLINK_ACTIONS)
4588                         return -EINVAL;
4589         }
4590
4591         err = nl80211_parse_sta_channel_info(info, &params);
4592         if (err)
4593                 return err;
4594
4595         err = nl80211_parse_sta_wme(info, &params);
4596         if (err)
4597                 return err;
4598
4599         if (parse_station_flags(info, dev->ieee80211_ptr->iftype, &params))
4600                 return -EINVAL;
4601
4602         /* HT/VHT requires QoS, but if we don't have that just ignore HT/VHT
4603          * as userspace might just pass through the capabilities from the IEs
4604          * directly, rather than enforcing this restriction and returning an
4605          * error in this case.
4606          */
4607         if (!(params.sta_flags_set & BIT(NL80211_STA_FLAG_WME))) {
4608                 params.ht_capa = NULL;
4609                 params.vht_capa = NULL;
4610         }
4611
4612         /* When you run into this, adjust the code below for the new flag */
4613         BUILD_BUG_ON(NL80211_STA_FLAG_MAX != 7);
4614
4615         switch (dev->ieee80211_ptr->iftype) {
4616         case NL80211_IFTYPE_AP:
4617         case NL80211_IFTYPE_AP_VLAN:
4618         case NL80211_IFTYPE_P2P_GO:
4619                 /* ignore WME attributes if iface/sta is not capable */
4620                 if (!(rdev->wiphy.flags & WIPHY_FLAG_AP_UAPSD) ||
4621                     !(params.sta_flags_set & BIT(NL80211_STA_FLAG_WME)))
4622                         params.sta_modify_mask &= ~STATION_PARAM_APPLY_UAPSD;
4623
4624                 /* TDLS peers cannot be added */
4625                 if ((params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)) ||
4626                     info->attrs[NL80211_ATTR_PEER_AID])
4627                         return -EINVAL;
4628                 /* but don't bother the driver with it */
4629                 params.sta_flags_mask &= ~BIT(NL80211_STA_FLAG_TDLS_PEER);
4630
4631                 /* allow authenticated/associated only if driver handles it */
4632                 if (!(rdev->wiphy.features &
4633                                 NL80211_FEATURE_FULL_AP_CLIENT_STATE) &&
4634                     params.sta_flags_mask & auth_assoc)
4635                         return -EINVAL;
4636
4637                 /* Older userspace, or userspace wanting to be compatible with
4638                  * !NL80211_FEATURE_FULL_AP_CLIENT_STATE, will not set the auth
4639                  * and assoc flags in the mask, but assumes the station will be
4640                  * added as associated anyway since this was the required driver
4641                  * behaviour before NL80211_FEATURE_FULL_AP_CLIENT_STATE was
4642                  * introduced.
4643                  * In order to not bother drivers with this quirk in the API
4644                  * set the flags in both the mask and set for new stations in
4645                  * this case.
4646                  */
4647                 if (!(params.sta_flags_mask & auth_assoc)) {
4648                         params.sta_flags_mask |= auth_assoc;
4649                         params.sta_flags_set |= auth_assoc;
4650                 }
4651
4652                 /* must be last in here for error handling */
4653                 params.vlan = get_vlan(info, rdev);
4654                 if (IS_ERR(params.vlan))
4655                         return PTR_ERR(params.vlan);
4656                 break;
4657         case NL80211_IFTYPE_MESH_POINT:
4658                 /* ignore uAPSD data */
4659                 params.sta_modify_mask &= ~STATION_PARAM_APPLY_UAPSD;
4660
4661                 /* associated is disallowed */
4662                 if (params.sta_flags_mask & BIT(NL80211_STA_FLAG_ASSOCIATED))
4663                         return -EINVAL;
4664                 /* TDLS peers cannot be added */
4665                 if ((params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)) ||
4666                     info->attrs[NL80211_ATTR_PEER_AID])
4667                         return -EINVAL;
4668                 break;
4669         case NL80211_IFTYPE_STATION:
4670         case NL80211_IFTYPE_P2P_CLIENT:
4671                 /* ignore uAPSD data */
4672                 params.sta_modify_mask &= ~STATION_PARAM_APPLY_UAPSD;
4673
4674                 /* these are disallowed */
4675                 if (params.sta_flags_mask &
4676                                 (BIT(NL80211_STA_FLAG_ASSOCIATED) |
4677                                  BIT(NL80211_STA_FLAG_AUTHENTICATED)))
4678                         return -EINVAL;
4679                 /* Only TDLS peers can be added */
4680                 if (!(params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)))
4681                         return -EINVAL;
4682                 /* Can only add if TDLS ... */
4683                 if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS))
4684                         return -EOPNOTSUPP;
4685                 /* ... with external setup is supported */
4686                 if (!(rdev->wiphy.flags & WIPHY_FLAG_TDLS_EXTERNAL_SETUP))
4687                         return -EOPNOTSUPP;
4688                 /*
4689                  * Older wpa_supplicant versions always mark the TDLS peer
4690                  * as authorized, but it shouldn't yet be.
4691                  */
4692                 params.sta_flags_mask &= ~BIT(NL80211_STA_FLAG_AUTHORIZED);
4693                 break;
4694         default:
4695                 return -EOPNOTSUPP;
4696         }
4697
4698         /* be aware of params.vlan when changing code here */
4699
4700         err = rdev_add_station(rdev, dev, mac_addr, &params);
4701
4702         if (params.vlan)
4703                 dev_put(params.vlan);
4704         return err;
4705 }
4706
4707 static int nl80211_del_station(struct sk_buff *skb, struct genl_info *info)
4708 {
4709         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4710         struct net_device *dev = info->user_ptr[1];
4711         struct station_del_parameters params;
4712
4713         memset(&params, 0, sizeof(params));
4714
4715         if (info->attrs[NL80211_ATTR_MAC])
4716                 params.mac = nla_data(info->attrs[NL80211_ATTR_MAC]);
4717
4718         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
4719             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP_VLAN &&
4720             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT &&
4721             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
4722                 return -EINVAL;
4723
4724         if (!rdev->ops->del_station)
4725                 return -EOPNOTSUPP;
4726
4727         if (info->attrs[NL80211_ATTR_MGMT_SUBTYPE]) {
4728                 params.subtype =
4729                         nla_get_u8(info->attrs[NL80211_ATTR_MGMT_SUBTYPE]);
4730                 if (params.subtype != IEEE80211_STYPE_DISASSOC >> 4 &&
4731                     params.subtype != IEEE80211_STYPE_DEAUTH >> 4)
4732                         return -EINVAL;
4733         } else {
4734                 /* Default to Deauthentication frame */
4735                 params.subtype = IEEE80211_STYPE_DEAUTH >> 4;
4736         }
4737
4738         if (info->attrs[NL80211_ATTR_REASON_CODE]) {
4739                 params.reason_code =
4740                         nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
4741                 if (params.reason_code == 0)
4742                         return -EINVAL; /* 0 is reserved */
4743         } else {
4744                 /* Default to reason code 2 */
4745                 params.reason_code = WLAN_REASON_PREV_AUTH_NOT_VALID;
4746         }
4747
4748         return rdev_del_station(rdev, dev, &params);
4749 }
4750
4751 static int nl80211_send_mpath(struct sk_buff *msg, u32 portid, u32 seq,
4752                                 int flags, struct net_device *dev,
4753                                 u8 *dst, u8 *next_hop,
4754                                 struct mpath_info *pinfo)
4755 {
4756         void *hdr;
4757         struct nlattr *pinfoattr;
4758
4759         hdr = nl80211hdr_put(msg, portid, seq, flags, NL80211_CMD_NEW_MPATH);
4760         if (!hdr)
4761                 return -1;
4762
4763         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
4764             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, dst) ||
4765             nla_put(msg, NL80211_ATTR_MPATH_NEXT_HOP, ETH_ALEN, next_hop) ||
4766             nla_put_u32(msg, NL80211_ATTR_GENERATION, pinfo->generation))
4767                 goto nla_put_failure;
4768
4769         pinfoattr = nla_nest_start(msg, NL80211_ATTR_MPATH_INFO);
4770         if (!pinfoattr)
4771                 goto nla_put_failure;
4772         if ((pinfo->filled & MPATH_INFO_FRAME_QLEN) &&
4773             nla_put_u32(msg, NL80211_MPATH_INFO_FRAME_QLEN,
4774                         pinfo->frame_qlen))
4775                 goto nla_put_failure;
4776         if (((pinfo->filled & MPATH_INFO_SN) &&
4777              nla_put_u32(msg, NL80211_MPATH_INFO_SN, pinfo->sn)) ||
4778             ((pinfo->filled & MPATH_INFO_METRIC) &&
4779              nla_put_u32(msg, NL80211_MPATH_INFO_METRIC,
4780                          pinfo->metric)) ||
4781             ((pinfo->filled & MPATH_INFO_EXPTIME) &&
4782              nla_put_u32(msg, NL80211_MPATH_INFO_EXPTIME,
4783                          pinfo->exptime)) ||
4784             ((pinfo->filled & MPATH_INFO_FLAGS) &&
4785              nla_put_u8(msg, NL80211_MPATH_INFO_FLAGS,
4786                         pinfo->flags)) ||
4787             ((pinfo->filled & MPATH_INFO_DISCOVERY_TIMEOUT) &&
4788              nla_put_u32(msg, NL80211_MPATH_INFO_DISCOVERY_TIMEOUT,
4789                          pinfo->discovery_timeout)) ||
4790             ((pinfo->filled & MPATH_INFO_DISCOVERY_RETRIES) &&
4791              nla_put_u8(msg, NL80211_MPATH_INFO_DISCOVERY_RETRIES,
4792                         pinfo->discovery_retries)))
4793                 goto nla_put_failure;
4794
4795         nla_nest_end(msg, pinfoattr);
4796
4797         genlmsg_end(msg, hdr);
4798         return 0;
4799
4800  nla_put_failure:
4801         genlmsg_cancel(msg, hdr);
4802         return -EMSGSIZE;
4803 }
4804
4805 static int nl80211_dump_mpath(struct sk_buff *skb,
4806                               struct netlink_callback *cb)
4807 {
4808         struct mpath_info pinfo;
4809         struct cfg80211_registered_device *rdev;
4810         struct wireless_dev *wdev;
4811         u8 dst[ETH_ALEN];
4812         u8 next_hop[ETH_ALEN];
4813         int path_idx = cb->args[2];
4814         int err;
4815
4816         err = nl80211_prepare_wdev_dump(skb, cb, &rdev, &wdev);
4817         if (err)
4818                 return err;
4819
4820         if (!rdev->ops->dump_mpath) {
4821                 err = -EOPNOTSUPP;
4822                 goto out_err;
4823         }
4824
4825         if (wdev->iftype != NL80211_IFTYPE_MESH_POINT) {
4826                 err = -EOPNOTSUPP;
4827                 goto out_err;
4828         }
4829
4830         while (1) {
4831                 err = rdev_dump_mpath(rdev, wdev->netdev, path_idx, dst,
4832                                       next_hop, &pinfo);
4833                 if (err == -ENOENT)
4834                         break;
4835                 if (err)
4836                         goto out_err;
4837
4838                 if (nl80211_send_mpath(skb, NETLINK_CB(cb->skb).portid,
4839                                        cb->nlh->nlmsg_seq, NLM_F_MULTI,
4840                                        wdev->netdev, dst, next_hop,
4841                                        &pinfo) < 0)
4842                         goto out;
4843
4844                 path_idx++;
4845         }
4846
4847  out:
4848         cb->args[2] = path_idx;
4849         err = skb->len;
4850  out_err:
4851         nl80211_finish_wdev_dump(rdev);
4852         return err;
4853 }
4854
4855 static int nl80211_get_mpath(struct sk_buff *skb, struct genl_info *info)
4856 {
4857         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4858         int err;
4859         struct net_device *dev = info->user_ptr[1];
4860         struct mpath_info pinfo;
4861         struct sk_buff *msg;
4862         u8 *dst = NULL;
4863         u8 next_hop[ETH_ALEN];
4864
4865         memset(&pinfo, 0, sizeof(pinfo));
4866
4867         if (!info->attrs[NL80211_ATTR_MAC])
4868                 return -EINVAL;
4869
4870         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
4871
4872         if (!rdev->ops->get_mpath)
4873                 return -EOPNOTSUPP;
4874
4875         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
4876                 return -EOPNOTSUPP;
4877
4878         err = rdev_get_mpath(rdev, dev, dst, next_hop, &pinfo);
4879         if (err)
4880                 return err;
4881
4882         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4883         if (!msg)
4884                 return -ENOMEM;
4885
4886         if (nl80211_send_mpath(msg, info->snd_portid, info->snd_seq, 0,
4887                                  dev, dst, next_hop, &pinfo) < 0) {
4888                 nlmsg_free(msg);
4889                 return -ENOBUFS;
4890         }
4891
4892         return genlmsg_reply(msg, info);
4893 }
4894
4895 static int nl80211_set_mpath(struct sk_buff *skb, struct genl_info *info)
4896 {
4897         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4898         struct net_device *dev = info->user_ptr[1];
4899         u8 *dst = NULL;
4900         u8 *next_hop = NULL;
4901
4902         if (!info->attrs[NL80211_ATTR_MAC])
4903                 return -EINVAL;
4904
4905         if (!info->attrs[NL80211_ATTR_MPATH_NEXT_HOP])
4906                 return -EINVAL;
4907
4908         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
4909         next_hop = nla_data(info->attrs[NL80211_ATTR_MPATH_NEXT_HOP]);
4910
4911         if (!rdev->ops->change_mpath)
4912                 return -EOPNOTSUPP;
4913
4914         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
4915                 return -EOPNOTSUPP;
4916
4917         return rdev_change_mpath(rdev, dev, dst, next_hop);
4918 }
4919
4920 static int nl80211_new_mpath(struct sk_buff *skb, struct genl_info *info)
4921 {
4922         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4923         struct net_device *dev = info->user_ptr[1];
4924         u8 *dst = NULL;
4925         u8 *next_hop = NULL;
4926
4927         if (!info->attrs[NL80211_ATTR_MAC])
4928                 return -EINVAL;
4929
4930         if (!info->attrs[NL80211_ATTR_MPATH_NEXT_HOP])
4931                 return -EINVAL;
4932
4933         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
4934         next_hop = nla_data(info->attrs[NL80211_ATTR_MPATH_NEXT_HOP]);
4935
4936         if (!rdev->ops->add_mpath)
4937                 return -EOPNOTSUPP;
4938
4939         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
4940                 return -EOPNOTSUPP;
4941
4942         return rdev_add_mpath(rdev, dev, dst, next_hop);
4943 }
4944
4945 static int nl80211_del_mpath(struct sk_buff *skb, struct genl_info *info)
4946 {
4947         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4948         struct net_device *dev = info->user_ptr[1];
4949         u8 *dst = NULL;
4950
4951         if (info->attrs[NL80211_ATTR_MAC])
4952                 dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
4953
4954         if (!rdev->ops->del_mpath)
4955                 return -EOPNOTSUPP;
4956
4957         return rdev_del_mpath(rdev, dev, dst);
4958 }
4959
4960 static int nl80211_get_mpp(struct sk_buff *skb, struct genl_info *info)
4961 {
4962         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4963         int err;
4964         struct net_device *dev = info->user_ptr[1];
4965         struct mpath_info pinfo;
4966         struct sk_buff *msg;
4967         u8 *dst = NULL;
4968         u8 mpp[ETH_ALEN];
4969
4970         memset(&pinfo, 0, sizeof(pinfo));
4971
4972         if (!info->attrs[NL80211_ATTR_MAC])
4973                 return -EINVAL;
4974
4975         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
4976
4977         if (!rdev->ops->get_mpp)
4978                 return -EOPNOTSUPP;
4979
4980         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
4981                 return -EOPNOTSUPP;
4982
4983         err = rdev_get_mpp(rdev, dev, dst, mpp, &pinfo);
4984         if (err)
4985                 return err;
4986
4987         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4988         if (!msg)
4989                 return -ENOMEM;
4990
4991         if (nl80211_send_mpath(msg, info->snd_portid, info->snd_seq, 0,
4992                                dev, dst, mpp, &pinfo) < 0) {
4993                 nlmsg_free(msg);
4994                 return -ENOBUFS;
4995         }
4996
4997         return genlmsg_reply(msg, info);
4998 }
4999
5000 static int nl80211_dump_mpp(struct sk_buff *skb,
5001                             struct netlink_callback *cb)
5002 {
5003         struct mpath_info pinfo;
5004         struct cfg80211_registered_device *rdev;
5005         struct wireless_dev *wdev;
5006         u8 dst[ETH_ALEN];
5007         u8 mpp[ETH_ALEN];
5008         int path_idx = cb->args[2];
5009         int err;
5010
5011         err = nl80211_prepare_wdev_dump(skb, cb, &rdev, &wdev);
5012         if (err)
5013                 return err;
5014
5015         if (!rdev->ops->dump_mpp) {
5016                 err = -EOPNOTSUPP;
5017                 goto out_err;
5018         }
5019
5020         if (wdev->iftype != NL80211_IFTYPE_MESH_POINT) {
5021                 err = -EOPNOTSUPP;
5022                 goto out_err;
5023         }
5024
5025         while (1) {
5026                 err = rdev_dump_mpp(rdev, wdev->netdev, path_idx, dst,
5027                                     mpp, &pinfo);
5028                 if (err == -ENOENT)
5029                         break;
5030                 if (err)
5031                         goto out_err;
5032
5033                 if (nl80211_send_mpath(skb, NETLINK_CB(cb->skb).portid,
5034                                        cb->nlh->nlmsg_seq, NLM_F_MULTI,
5035                                        wdev->netdev, dst, mpp,
5036                                        &pinfo) < 0)
5037                         goto out;
5038
5039                 path_idx++;
5040         }
5041
5042  out:
5043         cb->args[2] = path_idx;
5044         err = skb->len;
5045  out_err:
5046         nl80211_finish_wdev_dump(rdev);
5047         return err;
5048 }
5049
5050 static int nl80211_set_bss(struct sk_buff *skb, struct genl_info *info)
5051 {
5052         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5053         struct net_device *dev = info->user_ptr[1];
5054         struct wireless_dev *wdev = dev->ieee80211_ptr;
5055         struct bss_parameters params;
5056         int err;
5057
5058         memset(&params, 0, sizeof(params));
5059         /* default to not changing parameters */
5060         params.use_cts_prot = -1;
5061         params.use_short_preamble = -1;
5062         params.use_short_slot_time = -1;
5063         params.ap_isolate = -1;
5064         params.ht_opmode = -1;
5065         params.p2p_ctwindow = -1;
5066         params.p2p_opp_ps = -1;
5067
5068         if (info->attrs[NL80211_ATTR_BSS_CTS_PROT])
5069                 params.use_cts_prot =
5070                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_CTS_PROT]);
5071         if (info->attrs[NL80211_ATTR_BSS_SHORT_PREAMBLE])
5072                 params.use_short_preamble =
5073                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_SHORT_PREAMBLE]);
5074         if (info->attrs[NL80211_ATTR_BSS_SHORT_SLOT_TIME])
5075                 params.use_short_slot_time =
5076                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_SHORT_SLOT_TIME]);
5077         if (info->attrs[NL80211_ATTR_BSS_BASIC_RATES]) {
5078                 params.basic_rates =
5079                         nla_data(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
5080                 params.basic_rates_len =
5081                         nla_len(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
5082         }
5083         if (info->attrs[NL80211_ATTR_AP_ISOLATE])
5084                 params.ap_isolate = !!nla_get_u8(info->attrs[NL80211_ATTR_AP_ISOLATE]);
5085         if (info->attrs[NL80211_ATTR_BSS_HT_OPMODE])
5086                 params.ht_opmode =
5087                         nla_get_u16(info->attrs[NL80211_ATTR_BSS_HT_OPMODE]);
5088
5089         if (info->attrs[NL80211_ATTR_P2P_CTWINDOW]) {
5090                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
5091                         return -EINVAL;
5092                 params.p2p_ctwindow =
5093                         nla_get_s8(info->attrs[NL80211_ATTR_P2P_CTWINDOW]);
5094                 if (params.p2p_ctwindow < 0)
5095                         return -EINVAL;
5096                 if (params.p2p_ctwindow != 0 &&
5097                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_CTWIN))
5098                         return -EINVAL;
5099         }
5100
5101         if (info->attrs[NL80211_ATTR_P2P_OPPPS]) {
5102                 u8 tmp;
5103
5104                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
5105                         return -EINVAL;
5106                 tmp = nla_get_u8(info->attrs[NL80211_ATTR_P2P_OPPPS]);
5107                 if (tmp > 1)
5108                         return -EINVAL;
5109                 params.p2p_opp_ps = tmp;
5110                 if (params.p2p_opp_ps &&
5111                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_OPPPS))
5112                         return -EINVAL;
5113         }
5114
5115         if (!rdev->ops->change_bss)
5116                 return -EOPNOTSUPP;
5117
5118         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
5119             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
5120                 return -EOPNOTSUPP;
5121
5122         wdev_lock(wdev);
5123         err = rdev_change_bss(rdev, dev, &params);
5124         wdev_unlock(wdev);
5125
5126         return err;
5127 }
5128
5129 static int nl80211_req_set_reg(struct sk_buff *skb, struct genl_info *info)
5130 {
5131         char *data = NULL;
5132         bool is_indoor;
5133         enum nl80211_user_reg_hint_type user_reg_hint_type;
5134         u32 owner_nlportid;
5135
5136         /*
5137          * You should only get this when cfg80211 hasn't yet initialized
5138          * completely when built-in to the kernel right between the time
5139          * window between nl80211_init() and regulatory_init(), if that is
5140          * even possible.
5141          */
5142         if (unlikely(!rcu_access_pointer(cfg80211_regdomain)))
5143                 return -EINPROGRESS;
5144
5145         if (info->attrs[NL80211_ATTR_USER_REG_HINT_TYPE])
5146                 user_reg_hint_type =
5147                   nla_get_u32(info->attrs[NL80211_ATTR_USER_REG_HINT_TYPE]);
5148         else
5149                 user_reg_hint_type = NL80211_USER_REG_HINT_USER;
5150
5151         switch (user_reg_hint_type) {
5152         case NL80211_USER_REG_HINT_USER:
5153         case NL80211_USER_REG_HINT_CELL_BASE:
5154                 if (!info->attrs[NL80211_ATTR_REG_ALPHA2])
5155                         return -EINVAL;
5156
5157                 data = nla_data(info->attrs[NL80211_ATTR_REG_ALPHA2]);
5158                 return regulatory_hint_user(data, user_reg_hint_type);
5159         case NL80211_USER_REG_HINT_INDOOR:
5160                 if (info->attrs[NL80211_ATTR_SOCKET_OWNER]) {
5161                         owner_nlportid = info->snd_portid;
5162                         is_indoor = !!info->attrs[NL80211_ATTR_REG_INDOOR];
5163                 } else {
5164                         owner_nlportid = 0;
5165                         is_indoor = true;
5166                 }
5167
5168                 return regulatory_hint_indoor(is_indoor, owner_nlportid);
5169         default:
5170                 return -EINVAL;
5171         }
5172 }
5173
5174 static int nl80211_get_mesh_config(struct sk_buff *skb,
5175                                    struct genl_info *info)
5176 {
5177         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5178         struct net_device *dev = info->user_ptr[1];
5179         struct wireless_dev *wdev = dev->ieee80211_ptr;
5180         struct mesh_config cur_params;
5181         int err = 0;
5182         void *hdr;
5183         struct nlattr *pinfoattr;
5184         struct sk_buff *msg;
5185
5186         if (wdev->iftype != NL80211_IFTYPE_MESH_POINT)
5187                 return -EOPNOTSUPP;
5188
5189         if (!rdev->ops->get_mesh_config)
5190                 return -EOPNOTSUPP;
5191
5192         wdev_lock(wdev);
5193         /* If not connected, get default parameters */
5194         if (!wdev->mesh_id_len)
5195                 memcpy(&cur_params, &default_mesh_config, sizeof(cur_params));
5196         else
5197                 err = rdev_get_mesh_config(rdev, dev, &cur_params);
5198         wdev_unlock(wdev);
5199
5200         if (err)
5201                 return err;
5202
5203         /* Draw up a netlink message to send back */
5204         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
5205         if (!msg)
5206                 return -ENOMEM;
5207         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
5208                              NL80211_CMD_GET_MESH_CONFIG);
5209         if (!hdr)
5210                 goto out;
5211         pinfoattr = nla_nest_start(msg, NL80211_ATTR_MESH_CONFIG);
5212         if (!pinfoattr)
5213                 goto nla_put_failure;
5214         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
5215             nla_put_u16(msg, NL80211_MESHCONF_RETRY_TIMEOUT,
5216                         cur_params.dot11MeshRetryTimeout) ||
5217             nla_put_u16(msg, NL80211_MESHCONF_CONFIRM_TIMEOUT,
5218                         cur_params.dot11MeshConfirmTimeout) ||
5219             nla_put_u16(msg, NL80211_MESHCONF_HOLDING_TIMEOUT,
5220                         cur_params.dot11MeshHoldingTimeout) ||
5221             nla_put_u16(msg, NL80211_MESHCONF_MAX_PEER_LINKS,
5222                         cur_params.dot11MeshMaxPeerLinks) ||
5223             nla_put_u8(msg, NL80211_MESHCONF_MAX_RETRIES,
5224                        cur_params.dot11MeshMaxRetries) ||
5225             nla_put_u8(msg, NL80211_MESHCONF_TTL,
5226                        cur_params.dot11MeshTTL) ||
5227             nla_put_u8(msg, NL80211_MESHCONF_ELEMENT_TTL,
5228                        cur_params.element_ttl) ||
5229             nla_put_u8(msg, NL80211_MESHCONF_AUTO_OPEN_PLINKS,
5230                        cur_params.auto_open_plinks) ||
5231             nla_put_u32(msg, NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR,
5232                         cur_params.dot11MeshNbrOffsetMaxNeighbor) ||
5233             nla_put_u8(msg, NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES,
5234                        cur_params.dot11MeshHWMPmaxPREQretries) ||
5235             nla_put_u32(msg, NL80211_MESHCONF_PATH_REFRESH_TIME,
5236                         cur_params.path_refresh_time) ||
5237             nla_put_u16(msg, NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT,
5238                         cur_params.min_discovery_timeout) ||
5239             nla_put_u32(msg, NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT,
5240                         cur_params.dot11MeshHWMPactivePathTimeout) ||
5241             nla_put_u16(msg, NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL,
5242                         cur_params.dot11MeshHWMPpreqMinInterval) ||
5243             nla_put_u16(msg, NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL,
5244                         cur_params.dot11MeshHWMPperrMinInterval) ||
5245             nla_put_u16(msg, NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
5246                         cur_params.dot11MeshHWMPnetDiameterTraversalTime) ||
5247             nla_put_u8(msg, NL80211_MESHCONF_HWMP_ROOTMODE,
5248                        cur_params.dot11MeshHWMPRootMode) ||
5249             nla_put_u16(msg, NL80211_MESHCONF_HWMP_RANN_INTERVAL,
5250                         cur_params.dot11MeshHWMPRannInterval) ||
5251             nla_put_u8(msg, NL80211_MESHCONF_GATE_ANNOUNCEMENTS,
5252                        cur_params.dot11MeshGateAnnouncementProtocol) ||
5253             nla_put_u8(msg, NL80211_MESHCONF_FORWARDING,
5254                        cur_params.dot11MeshForwarding) ||
5255             nla_put_u32(msg, NL80211_MESHCONF_RSSI_THRESHOLD,
5256                         cur_params.rssi_threshold) ||
5257             nla_put_u32(msg, NL80211_MESHCONF_HT_OPMODE,
5258                         cur_params.ht_opmode) ||
5259             nla_put_u32(msg, NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT,
5260                         cur_params.dot11MeshHWMPactivePathToRootTimeout) ||
5261             nla_put_u16(msg, NL80211_MESHCONF_HWMP_ROOT_INTERVAL,
5262                         cur_params.dot11MeshHWMProotInterval) ||
5263             nla_put_u16(msg, NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL,
5264                         cur_params.dot11MeshHWMPconfirmationInterval) ||
5265             nla_put_u32(msg, NL80211_MESHCONF_POWER_MODE,
5266                         cur_params.power_mode) ||
5267             nla_put_u16(msg, NL80211_MESHCONF_AWAKE_WINDOW,
5268                         cur_params.dot11MeshAwakeWindowDuration) ||
5269             nla_put_u32(msg, NL80211_MESHCONF_PLINK_TIMEOUT,
5270                         cur_params.plink_timeout))
5271                 goto nla_put_failure;
5272         nla_nest_end(msg, pinfoattr);
5273         genlmsg_end(msg, hdr);
5274         return genlmsg_reply(msg, info);
5275
5276  nla_put_failure:
5277         genlmsg_cancel(msg, hdr);
5278  out:
5279         nlmsg_free(msg);
5280         return -ENOBUFS;
5281 }
5282
5283 static const struct nla_policy nl80211_meshconf_params_policy[NL80211_MESHCONF_ATTR_MAX+1] = {
5284         [NL80211_MESHCONF_RETRY_TIMEOUT] = { .type = NLA_U16 },
5285         [NL80211_MESHCONF_CONFIRM_TIMEOUT] = { .type = NLA_U16 },
5286         [NL80211_MESHCONF_HOLDING_TIMEOUT] = { .type = NLA_U16 },
5287         [NL80211_MESHCONF_MAX_PEER_LINKS] = { .type = NLA_U16 },
5288         [NL80211_MESHCONF_MAX_RETRIES] = { .type = NLA_U8 },
5289         [NL80211_MESHCONF_TTL] = { .type = NLA_U8 },
5290         [NL80211_MESHCONF_ELEMENT_TTL] = { .type = NLA_U8 },
5291         [NL80211_MESHCONF_AUTO_OPEN_PLINKS] = { .type = NLA_U8 },
5292         [NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR] = { .type = NLA_U32 },
5293         [NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES] = { .type = NLA_U8 },
5294         [NL80211_MESHCONF_PATH_REFRESH_TIME] = { .type = NLA_U32 },
5295         [NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT] = { .type = NLA_U16 },
5296         [NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT] = { .type = NLA_U32 },
5297         [NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL] = { .type = NLA_U16 },
5298         [NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL] = { .type = NLA_U16 },
5299         [NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME] = { .type = NLA_U16 },
5300         [NL80211_MESHCONF_HWMP_ROOTMODE] = { .type = NLA_U8 },
5301         [NL80211_MESHCONF_HWMP_RANN_INTERVAL] = { .type = NLA_U16 },
5302         [NL80211_MESHCONF_GATE_ANNOUNCEMENTS] = { .type = NLA_U8 },
5303         [NL80211_MESHCONF_FORWARDING] = { .type = NLA_U8 },
5304         [NL80211_MESHCONF_RSSI_THRESHOLD] = { .type = NLA_U32 },
5305         [NL80211_MESHCONF_HT_OPMODE] = { .type = NLA_U16 },
5306         [NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT] = { .type = NLA_U32 },
5307         [NL80211_MESHCONF_HWMP_ROOT_INTERVAL] = { .type = NLA_U16 },
5308         [NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL] = { .type = NLA_U16 },
5309         [NL80211_MESHCONF_POWER_MODE] = { .type = NLA_U32 },
5310         [NL80211_MESHCONF_AWAKE_WINDOW] = { .type = NLA_U16 },
5311         [NL80211_MESHCONF_PLINK_TIMEOUT] = { .type = NLA_U32 },
5312 };
5313
5314 static const struct nla_policy
5315         nl80211_mesh_setup_params_policy[NL80211_MESH_SETUP_ATTR_MAX+1] = {
5316         [NL80211_MESH_SETUP_ENABLE_VENDOR_SYNC] = { .type = NLA_U8 },
5317         [NL80211_MESH_SETUP_ENABLE_VENDOR_PATH_SEL] = { .type = NLA_U8 },
5318         [NL80211_MESH_SETUP_ENABLE_VENDOR_METRIC] = { .type = NLA_U8 },
5319         [NL80211_MESH_SETUP_USERSPACE_AUTH] = { .type = NLA_FLAG },
5320         [NL80211_MESH_SETUP_AUTH_PROTOCOL] = { .type = NLA_U8 },
5321         [NL80211_MESH_SETUP_USERSPACE_MPM] = { .type = NLA_FLAG },
5322         [NL80211_MESH_SETUP_IE] = { .type = NLA_BINARY,
5323                                     .len = IEEE80211_MAX_DATA_LEN },
5324         [NL80211_MESH_SETUP_USERSPACE_AMPE] = { .type = NLA_FLAG },
5325 };
5326
5327 static int nl80211_check_bool(const struct nlattr *nla, u8 min, u8 max, bool *out)
5328 {
5329         u8 val = nla_get_u8(nla);
5330         if (val < min || val > max)
5331                 return -EINVAL;
5332         *out = val;
5333         return 0;
5334 }
5335
5336 static int nl80211_check_u8(const struct nlattr *nla, u8 min, u8 max, u8 *out)
5337 {
5338         u8 val = nla_get_u8(nla);
5339         if (val < min || val > max)
5340                 return -EINVAL;
5341         *out = val;
5342         return 0;
5343 }
5344
5345 static int nl80211_check_u16(const struct nlattr *nla, u16 min, u16 max, u16 *out)
5346 {
5347         u16 val = nla_get_u16(nla);
5348         if (val < min || val > max)
5349                 return -EINVAL;
5350         *out = val;
5351         return 0;
5352 }
5353
5354 static int nl80211_check_u32(const struct nlattr *nla, u32 min, u32 max, u32 *out)
5355 {
5356         u32 val = nla_get_u32(nla);
5357         if (val < min || val > max)
5358                 return -EINVAL;
5359         *out = val;
5360         return 0;
5361 }
5362
5363 static int nl80211_check_s32(const struct nlattr *nla, s32 min, s32 max, s32 *out)
5364 {
5365         s32 val = nla_get_s32(nla);
5366         if (val < min || val > max)
5367                 return -EINVAL;
5368         *out = val;
5369         return 0;
5370 }
5371
5372 static int nl80211_parse_mesh_config(struct genl_info *info,
5373                                      struct mesh_config *cfg,
5374                                      u32 *mask_out)
5375 {
5376         struct nlattr *tb[NL80211_MESHCONF_ATTR_MAX + 1];
5377         u32 mask = 0;
5378         u16 ht_opmode;
5379
5380 #define FILL_IN_MESH_PARAM_IF_SET(tb, cfg, param, min, max, mask, attr, fn) \
5381 do {                                                                        \
5382         if (tb[attr]) {                                                     \
5383                 if (fn(tb[attr], min, max, &cfg->param))                    \
5384                         return -EINVAL;                                     \
5385                 mask |= (1 << (attr - 1));                                  \
5386         }                                                                   \
5387 } while (0)
5388
5389         if (!info->attrs[NL80211_ATTR_MESH_CONFIG])
5390                 return -EINVAL;
5391         if (nla_parse_nested(tb, NL80211_MESHCONF_ATTR_MAX,
5392                              info->attrs[NL80211_ATTR_MESH_CONFIG],
5393                              nl80211_meshconf_params_policy))
5394                 return -EINVAL;
5395
5396         /* This makes sure that there aren't more than 32 mesh config
5397          * parameters (otherwise our bitfield scheme would not work.) */
5398         BUILD_BUG_ON(NL80211_MESHCONF_ATTR_MAX > 32);
5399
5400         /* Fill in the params struct */
5401         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshRetryTimeout, 1, 255,
5402                                   mask, NL80211_MESHCONF_RETRY_TIMEOUT,
5403                                   nl80211_check_u16);
5404         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshConfirmTimeout, 1, 255,
5405                                   mask, NL80211_MESHCONF_CONFIRM_TIMEOUT,
5406                                   nl80211_check_u16);
5407         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHoldingTimeout, 1, 255,
5408                                   mask, NL80211_MESHCONF_HOLDING_TIMEOUT,
5409                                   nl80211_check_u16);
5410         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshMaxPeerLinks, 0, 255,
5411                                   mask, NL80211_MESHCONF_MAX_PEER_LINKS,
5412                                   nl80211_check_u16);
5413         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshMaxRetries, 0, 16,
5414                                   mask, NL80211_MESHCONF_MAX_RETRIES,
5415                                   nl80211_check_u8);
5416         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshTTL, 1, 255,
5417                                   mask, NL80211_MESHCONF_TTL, nl80211_check_u8);
5418         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, element_ttl, 1, 255,
5419                                   mask, NL80211_MESHCONF_ELEMENT_TTL,
5420                                   nl80211_check_u8);
5421         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, auto_open_plinks, 0, 1,
5422                                   mask, NL80211_MESHCONF_AUTO_OPEN_PLINKS,
5423                                   nl80211_check_bool);
5424         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshNbrOffsetMaxNeighbor,
5425                                   1, 255, mask,
5426                                   NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR,
5427                                   nl80211_check_u32);
5428         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPmaxPREQretries, 0, 255,
5429                                   mask, NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES,
5430                                   nl80211_check_u8);
5431         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, path_refresh_time, 1, 65535,
5432                                   mask, NL80211_MESHCONF_PATH_REFRESH_TIME,
5433                                   nl80211_check_u32);
5434         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, min_discovery_timeout, 1, 65535,
5435                                   mask, NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT,
5436                                   nl80211_check_u16);
5437         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPactivePathTimeout,
5438                                   1, 65535, mask,
5439                                   NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT,
5440                                   nl80211_check_u32);
5441         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPpreqMinInterval,
5442                                   1, 65535, mask,
5443                                   NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL,
5444                                   nl80211_check_u16);
5445         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPperrMinInterval,
5446                                   1, 65535, mask,
5447                                   NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL,
5448                                   nl80211_check_u16);
5449         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
5450                                   dot11MeshHWMPnetDiameterTraversalTime,
5451                                   1, 65535, mask,
5452                                   NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
5453                                   nl80211_check_u16);
5454         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPRootMode, 0, 4,
5455                                   mask, NL80211_MESHCONF_HWMP_ROOTMODE,
5456                                   nl80211_check_u8);
5457         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPRannInterval, 1, 65535,
5458                                   mask, NL80211_MESHCONF_HWMP_RANN_INTERVAL,
5459                                   nl80211_check_u16);
5460         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
5461                                   dot11MeshGateAnnouncementProtocol, 0, 1,
5462                                   mask, NL80211_MESHCONF_GATE_ANNOUNCEMENTS,
5463                                   nl80211_check_bool);
5464         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshForwarding, 0, 1,
5465                                   mask, NL80211_MESHCONF_FORWARDING,
5466                                   nl80211_check_bool);
5467         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, rssi_threshold, -255, 0,
5468                                   mask, NL80211_MESHCONF_RSSI_THRESHOLD,
5469                                   nl80211_check_s32);
5470         /*
5471          * Check HT operation mode based on
5472          * IEEE 802.11 2012 8.4.2.59 HT Operation element.
5473          */
5474         if (tb[NL80211_MESHCONF_HT_OPMODE]) {
5475                 ht_opmode = nla_get_u16(tb[NL80211_MESHCONF_HT_OPMODE]);
5476
5477                 if (ht_opmode & ~(IEEE80211_HT_OP_MODE_PROTECTION |
5478                                   IEEE80211_HT_OP_MODE_NON_GF_STA_PRSNT |
5479                                   IEEE80211_HT_OP_MODE_NON_HT_STA_PRSNT))
5480                         return -EINVAL;
5481
5482                 if ((ht_opmode & IEEE80211_HT_OP_MODE_NON_GF_STA_PRSNT) &&
5483                     (ht_opmode & IEEE80211_HT_OP_MODE_NON_HT_STA_PRSNT))
5484                         return -EINVAL;
5485
5486                 switch (ht_opmode & IEEE80211_HT_OP_MODE_PROTECTION) {
5487                 case IEEE80211_HT_OP_MODE_PROTECTION_NONE:
5488                 case IEEE80211_HT_OP_MODE_PROTECTION_20MHZ:
5489                         if (ht_opmode & IEEE80211_HT_OP_MODE_NON_HT_STA_PRSNT)
5490                                 return -EINVAL;
5491                         break;
5492                 case IEEE80211_HT_OP_MODE_PROTECTION_NONMEMBER:
5493                 case IEEE80211_HT_OP_MODE_PROTECTION_NONHT_MIXED:
5494                         if (!(ht_opmode & IEEE80211_HT_OP_MODE_NON_HT_STA_PRSNT))
5495                                 return -EINVAL;
5496                         break;
5497                 }
5498                 cfg->ht_opmode = ht_opmode;
5499         }
5500         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPactivePathToRootTimeout,
5501                                   1, 65535, mask,
5502                                   NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT,
5503                                   nl80211_check_u32);
5504         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMProotInterval, 1, 65535,
5505                                   mask, NL80211_MESHCONF_HWMP_ROOT_INTERVAL,
5506                                   nl80211_check_u16);
5507         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
5508                                   dot11MeshHWMPconfirmationInterval,
5509                                   1, 65535, mask,
5510                                   NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL,
5511                                   nl80211_check_u16);
5512         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, power_mode,
5513                                   NL80211_MESH_POWER_ACTIVE,
5514                                   NL80211_MESH_POWER_MAX,
5515                                   mask, NL80211_MESHCONF_POWER_MODE,
5516                                   nl80211_check_u32);
5517         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshAwakeWindowDuration,
5518                                   0, 65535, mask,
5519                                   NL80211_MESHCONF_AWAKE_WINDOW, nl80211_check_u16);
5520         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, plink_timeout, 0, 0xffffffff,
5521                                   mask, NL80211_MESHCONF_PLINK_TIMEOUT,
5522                                   nl80211_check_u32);
5523         if (mask_out)
5524                 *mask_out = mask;
5525
5526         return 0;
5527
5528 #undef FILL_IN_MESH_PARAM_IF_SET
5529 }
5530
5531 static int nl80211_parse_mesh_setup(struct genl_info *info,
5532                                      struct mesh_setup *setup)
5533 {
5534         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5535         struct nlattr *tb[NL80211_MESH_SETUP_ATTR_MAX + 1];
5536
5537         if (!info->attrs[NL80211_ATTR_MESH_SETUP])
5538                 return -EINVAL;
5539         if (nla_parse_nested(tb, NL80211_MESH_SETUP_ATTR_MAX,
5540                              info->attrs[NL80211_ATTR_MESH_SETUP],
5541                              nl80211_mesh_setup_params_policy))
5542                 return -EINVAL;
5543
5544         if (tb[NL80211_MESH_SETUP_ENABLE_VENDOR_SYNC])
5545                 setup->sync_method =
5546                 (nla_get_u8(tb[NL80211_MESH_SETUP_ENABLE_VENDOR_SYNC])) ?
5547                  IEEE80211_SYNC_METHOD_VENDOR :
5548                  IEEE80211_SYNC_METHOD_NEIGHBOR_OFFSET;
5549
5550         if (tb[NL80211_MESH_SETUP_ENABLE_VENDOR_PATH_SEL])
5551                 setup->path_sel_proto =
5552                 (nla_get_u8(tb[NL80211_MESH_SETUP_ENABLE_VENDOR_PATH_SEL])) ?
5553                  IEEE80211_PATH_PROTOCOL_VENDOR :
5554                  IEEE80211_PATH_PROTOCOL_HWMP;
5555
5556         if (tb[NL80211_MESH_SETUP_ENABLE_VENDOR_METRIC])
5557                 setup->path_metric =
5558                 (nla_get_u8(tb[NL80211_MESH_SETUP_ENABLE_VENDOR_METRIC])) ?
5559                  IEEE80211_PATH_METRIC_VENDOR :
5560                  IEEE80211_PATH_METRIC_AIRTIME;
5561
5562         if (tb[NL80211_MESH_SETUP_IE]) {
5563                 struct nlattr *ieattr =
5564                         tb[NL80211_MESH_SETUP_IE];
5565                 if (!is_valid_ie_attr(ieattr))
5566                         return -EINVAL;
5567                 setup->ie = nla_data(ieattr);
5568                 setup->ie_len = nla_len(ieattr);
5569         }
5570         if (tb[NL80211_MESH_SETUP_USERSPACE_MPM] &&
5571             !(rdev->wiphy.features & NL80211_FEATURE_USERSPACE_MPM))
5572                 return -EINVAL;
5573         setup->user_mpm = nla_get_flag(tb[NL80211_MESH_SETUP_USERSPACE_MPM]);
5574         setup->is_authenticated = nla_get_flag(tb[NL80211_MESH_SETUP_USERSPACE_AUTH]);
5575         setup->is_secure = nla_get_flag(tb[NL80211_MESH_SETUP_USERSPACE_AMPE]);
5576         if (setup->is_secure)
5577                 setup->user_mpm = true;
5578
5579         if (tb[NL80211_MESH_SETUP_AUTH_PROTOCOL]) {
5580                 if (!setup->user_mpm)
5581                         return -EINVAL;
5582                 setup->auth_id =
5583                         nla_get_u8(tb[NL80211_MESH_SETUP_AUTH_PROTOCOL]);
5584         }
5585
5586         return 0;
5587 }
5588
5589 static int nl80211_update_mesh_config(struct sk_buff *skb,
5590                                       struct genl_info *info)
5591 {
5592         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5593         struct net_device *dev = info->user_ptr[1];
5594         struct wireless_dev *wdev = dev->ieee80211_ptr;
5595         struct mesh_config cfg;
5596         u32 mask;
5597         int err;
5598
5599         if (wdev->iftype != NL80211_IFTYPE_MESH_POINT)
5600                 return -EOPNOTSUPP;
5601
5602         if (!rdev->ops->update_mesh_config)
5603                 return -EOPNOTSUPP;
5604
5605         err = nl80211_parse_mesh_config(info, &cfg, &mask);
5606         if (err)
5607                 return err;
5608
5609         wdev_lock(wdev);
5610         if (!wdev->mesh_id_len)
5611                 err = -ENOLINK;
5612
5613         if (!err)
5614                 err = rdev_update_mesh_config(rdev, dev, mask, &cfg);
5615
5616         wdev_unlock(wdev);
5617
5618         return err;
5619 }
5620
5621 static int nl80211_put_regdom(const struct ieee80211_regdomain *regdom,
5622                               struct sk_buff *msg)
5623 {
5624         struct nlattr *nl_reg_rules;
5625         unsigned int i;
5626
5627         if (nla_put_string(msg, NL80211_ATTR_REG_ALPHA2, regdom->alpha2) ||
5628             (regdom->dfs_region &&
5629              nla_put_u8(msg, NL80211_ATTR_DFS_REGION, regdom->dfs_region)))
5630                 goto nla_put_failure;
5631
5632         nl_reg_rules = nla_nest_start(msg, NL80211_ATTR_REG_RULES);
5633         if (!nl_reg_rules)
5634                 goto nla_put_failure;
5635
5636         for (i = 0; i < regdom->n_reg_rules; i++) {
5637                 struct nlattr *nl_reg_rule;
5638                 const struct ieee80211_reg_rule *reg_rule;
5639                 const struct ieee80211_freq_range *freq_range;
5640                 const struct ieee80211_power_rule *power_rule;
5641                 unsigned int max_bandwidth_khz;
5642
5643                 reg_rule = &regdom->reg_rules[i];
5644                 freq_range = &reg_rule->freq_range;
5645                 power_rule = &reg_rule->power_rule;
5646
5647                 nl_reg_rule = nla_nest_start(msg, i);
5648                 if (!nl_reg_rule)
5649                         goto nla_put_failure;
5650
5651                 max_bandwidth_khz = freq_range->max_bandwidth_khz;
5652                 if (!max_bandwidth_khz)
5653                         max_bandwidth_khz = reg_get_max_bandwidth(regdom,
5654                                                                   reg_rule);
5655
5656                 if (nla_put_u32(msg, NL80211_ATTR_REG_RULE_FLAGS,
5657                                 reg_rule->flags) ||
5658                     nla_put_u32(msg, NL80211_ATTR_FREQ_RANGE_START,
5659                                 freq_range->start_freq_khz) ||
5660                     nla_put_u32(msg, NL80211_ATTR_FREQ_RANGE_END,
5661                                 freq_range->end_freq_khz) ||
5662                     nla_put_u32(msg, NL80211_ATTR_FREQ_RANGE_MAX_BW,
5663                                 max_bandwidth_khz) ||
5664                     nla_put_u32(msg, NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN,
5665                                 power_rule->max_antenna_gain) ||
5666                     nla_put_u32(msg, NL80211_ATTR_POWER_RULE_MAX_EIRP,
5667                                 power_rule->max_eirp) ||
5668                     nla_put_u32(msg, NL80211_ATTR_DFS_CAC_TIME,
5669                                 reg_rule->dfs_cac_ms))
5670                         goto nla_put_failure;
5671
5672                 nla_nest_end(msg, nl_reg_rule);
5673         }
5674
5675         nla_nest_end(msg, nl_reg_rules);
5676         return 0;
5677
5678 nla_put_failure:
5679         return -EMSGSIZE;
5680 }
5681
5682 static int nl80211_get_reg_do(struct sk_buff *skb, struct genl_info *info)
5683 {
5684         const struct ieee80211_regdomain *regdom = NULL;
5685         struct cfg80211_registered_device *rdev;
5686         struct wiphy *wiphy = NULL;
5687         struct sk_buff *msg;
5688         void *hdr;
5689
5690         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
5691         if (!msg)
5692                 return -ENOBUFS;
5693
5694         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
5695                              NL80211_CMD_GET_REG);
5696         if (!hdr)
5697                 goto put_failure;
5698
5699         if (info->attrs[NL80211_ATTR_WIPHY]) {
5700                 bool self_managed;
5701
5702                 rdev = cfg80211_get_dev_from_info(genl_info_net(info), info);
5703                 if (IS_ERR(rdev)) {
5704                         nlmsg_free(msg);
5705                         return PTR_ERR(rdev);
5706                 }
5707
5708                 wiphy = &rdev->wiphy;
5709                 self_managed = wiphy->regulatory_flags &
5710                                REGULATORY_WIPHY_SELF_MANAGED;
5711                 regdom = get_wiphy_regdom(wiphy);
5712
5713                 /* a self-managed-reg device must have a private regdom */
5714                 if (WARN_ON(!regdom && self_managed)) {
5715                         nlmsg_free(msg);
5716                         return -EINVAL;
5717                 }
5718
5719                 if (regdom &&
5720                     nla_put_u32(msg, NL80211_ATTR_WIPHY, get_wiphy_idx(wiphy)))
5721                         goto nla_put_failure;
5722         }
5723
5724         if (!wiphy && reg_last_request_cell_base() &&
5725             nla_put_u32(msg, NL80211_ATTR_USER_REG_HINT_TYPE,
5726                         NL80211_USER_REG_HINT_CELL_BASE))
5727                 goto nla_put_failure;
5728
5729         rcu_read_lock();
5730
5731         if (!regdom)
5732                 regdom = rcu_dereference(cfg80211_regdomain);
5733
5734         if (nl80211_put_regdom(regdom, msg))
5735                 goto nla_put_failure_rcu;
5736
5737         rcu_read_unlock();
5738
5739         genlmsg_end(msg, hdr);
5740         return genlmsg_reply(msg, info);
5741
5742 nla_put_failure_rcu:
5743         rcu_read_unlock();
5744 nla_put_failure:
5745         genlmsg_cancel(msg, hdr);
5746 put_failure:
5747         nlmsg_free(msg);
5748         return -EMSGSIZE;
5749 }
5750
5751 static int nl80211_send_regdom(struct sk_buff *msg, struct netlink_callback *cb,
5752                                u32 seq, int flags, struct wiphy *wiphy,
5753                                const struct ieee80211_regdomain *regdom)
5754 {
5755         void *hdr = nl80211hdr_put(msg, NETLINK_CB(cb->skb).portid, seq, flags,
5756                                    NL80211_CMD_GET_REG);
5757
5758         if (!hdr)
5759                 return -1;
5760
5761         genl_dump_check_consistent(cb, hdr, &nl80211_fam);
5762
5763         if (nl80211_put_regdom(regdom, msg))
5764                 goto nla_put_failure;
5765
5766         if (!wiphy && reg_last_request_cell_base() &&
5767             nla_put_u32(msg, NL80211_ATTR_USER_REG_HINT_TYPE,
5768                         NL80211_USER_REG_HINT_CELL_BASE))
5769                 goto nla_put_failure;
5770
5771         if (wiphy &&
5772             nla_put_u32(msg, NL80211_ATTR_WIPHY, get_wiphy_idx(wiphy)))
5773                 goto nla_put_failure;
5774
5775         if (wiphy && wiphy->regulatory_flags & REGULATORY_WIPHY_SELF_MANAGED &&
5776             nla_put_flag(msg, NL80211_ATTR_WIPHY_SELF_MANAGED_REG))
5777                 goto nla_put_failure;
5778
5779         genlmsg_end(msg, hdr);
5780         return 0;
5781
5782 nla_put_failure:
5783         genlmsg_cancel(msg, hdr);
5784         return -EMSGSIZE;
5785 }
5786
5787 static int nl80211_get_reg_dump(struct sk_buff *skb,
5788                                 struct netlink_callback *cb)
5789 {
5790         const struct ieee80211_regdomain *regdom = NULL;
5791         struct cfg80211_registered_device *rdev;
5792         int err, reg_idx, start = cb->args[2];
5793
5794         rtnl_lock();
5795
5796         if (cfg80211_regdomain && start == 0) {
5797                 err = nl80211_send_regdom(skb, cb, cb->nlh->nlmsg_seq,
5798                                           NLM_F_MULTI, NULL,
5799                                           rtnl_dereference(cfg80211_regdomain));
5800                 if (err < 0)
5801                         goto out_err;
5802         }
5803
5804         /* the global regdom is idx 0 */
5805         reg_idx = 1;
5806         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
5807                 regdom = get_wiphy_regdom(&rdev->wiphy);
5808                 if (!regdom)
5809                         continue;
5810
5811                 if (++reg_idx <= start)
5812                         continue;
5813
5814                 err = nl80211_send_regdom(skb, cb, cb->nlh->nlmsg_seq,
5815                                           NLM_F_MULTI, &rdev->wiphy, regdom);
5816                 if (err < 0) {
5817                         reg_idx--;
5818                         break;
5819                 }
5820         }
5821
5822         cb->args[2] = reg_idx;
5823         err = skb->len;
5824 out_err:
5825         rtnl_unlock();
5826         return err;
5827 }
5828
5829 #ifdef CONFIG_CFG80211_CRDA_SUPPORT
5830 static const struct nla_policy reg_rule_policy[NL80211_REG_RULE_ATTR_MAX + 1] = {
5831         [NL80211_ATTR_REG_RULE_FLAGS]           = { .type = NLA_U32 },
5832         [NL80211_ATTR_FREQ_RANGE_START]         = { .type = NLA_U32 },
5833         [NL80211_ATTR_FREQ_RANGE_END]           = { .type = NLA_U32 },
5834         [NL80211_ATTR_FREQ_RANGE_MAX_BW]        = { .type = NLA_U32 },
5835         [NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN]  = { .type = NLA_U32 },
5836         [NL80211_ATTR_POWER_RULE_MAX_EIRP]      = { .type = NLA_U32 },
5837         [NL80211_ATTR_DFS_CAC_TIME]             = { .type = NLA_U32 },
5838 };
5839
5840 static int parse_reg_rule(struct nlattr *tb[],
5841         struct ieee80211_reg_rule *reg_rule)
5842 {
5843         struct ieee80211_freq_range *freq_range = &reg_rule->freq_range;
5844         struct ieee80211_power_rule *power_rule = &reg_rule->power_rule;
5845
5846         if (!tb[NL80211_ATTR_REG_RULE_FLAGS])
5847                 return -EINVAL;
5848         if (!tb[NL80211_ATTR_FREQ_RANGE_START])
5849                 return -EINVAL;
5850         if (!tb[NL80211_ATTR_FREQ_RANGE_END])
5851                 return -EINVAL;
5852         if (!tb[NL80211_ATTR_FREQ_RANGE_MAX_BW])
5853                 return -EINVAL;
5854         if (!tb[NL80211_ATTR_POWER_RULE_MAX_EIRP])
5855                 return -EINVAL;
5856
5857         reg_rule->flags = nla_get_u32(tb[NL80211_ATTR_REG_RULE_FLAGS]);
5858
5859         freq_range->start_freq_khz =
5860                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]);
5861         freq_range->end_freq_khz =
5862                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]);
5863         freq_range->max_bandwidth_khz =
5864                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]);
5865
5866         power_rule->max_eirp =
5867                 nla_get_u32(tb[NL80211_ATTR_POWER_RULE_MAX_EIRP]);
5868
5869         if (tb[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN])
5870                 power_rule->max_antenna_gain =
5871                         nla_get_u32(tb[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN]);
5872
5873         if (tb[NL80211_ATTR_DFS_CAC_TIME])
5874                 reg_rule->dfs_cac_ms =
5875                         nla_get_u32(tb[NL80211_ATTR_DFS_CAC_TIME]);
5876
5877         return 0;
5878 }
5879
5880 static int nl80211_set_reg(struct sk_buff *skb, struct genl_info *info)
5881 {
5882         struct nlattr *tb[NL80211_REG_RULE_ATTR_MAX + 1];
5883         struct nlattr *nl_reg_rule;
5884         char *alpha2;
5885         int rem_reg_rules, r;
5886         u32 num_rules = 0, rule_idx = 0, size_of_regd;
5887         enum nl80211_dfs_regions dfs_region = NL80211_DFS_UNSET;
5888         struct ieee80211_regdomain *rd;
5889
5890         if (!info->attrs[NL80211_ATTR_REG_ALPHA2])
5891                 return -EINVAL;
5892
5893         if (!info->attrs[NL80211_ATTR_REG_RULES])
5894                 return -EINVAL;
5895
5896         alpha2 = nla_data(info->attrs[NL80211_ATTR_REG_ALPHA2]);
5897
5898         if (info->attrs[NL80211_ATTR_DFS_REGION])
5899                 dfs_region = nla_get_u8(info->attrs[NL80211_ATTR_DFS_REGION]);
5900
5901         nla_for_each_nested(nl_reg_rule, info->attrs[NL80211_ATTR_REG_RULES],
5902                             rem_reg_rules) {
5903                 num_rules++;
5904                 if (num_rules > NL80211_MAX_SUPP_REG_RULES)
5905                         return -EINVAL;
5906         }
5907
5908         if (!reg_is_valid_request(alpha2))
5909                 return -EINVAL;
5910
5911         size_of_regd = sizeof(struct ieee80211_regdomain) +
5912                        num_rules * sizeof(struct ieee80211_reg_rule);
5913
5914         rd = kzalloc(size_of_regd, GFP_KERNEL);
5915         if (!rd)
5916                 return -ENOMEM;
5917
5918         rd->n_reg_rules = num_rules;
5919         rd->alpha2[0] = alpha2[0];
5920         rd->alpha2[1] = alpha2[1];
5921
5922         /*
5923          * Disable DFS master mode if the DFS region was
5924          * not supported or known on this kernel.
5925          */
5926         if (reg_supported_dfs_region(dfs_region))
5927                 rd->dfs_region = dfs_region;
5928
5929         nla_for_each_nested(nl_reg_rule, info->attrs[NL80211_ATTR_REG_RULES],
5930                             rem_reg_rules) {
5931                 r = nla_parse(tb, NL80211_REG_RULE_ATTR_MAX,
5932                               nla_data(nl_reg_rule), nla_len(nl_reg_rule),
5933                               reg_rule_policy);
5934                 if (r)
5935                         goto bad_reg;
5936                 r = parse_reg_rule(tb, &rd->reg_rules[rule_idx]);
5937                 if (r)
5938                         goto bad_reg;
5939
5940                 rule_idx++;
5941
5942                 if (rule_idx > NL80211_MAX_SUPP_REG_RULES) {
5943                         r = -EINVAL;
5944                         goto bad_reg;
5945                 }
5946         }
5947
5948         /* set_regdom takes ownership of rd */
5949         return set_regdom(rd, REGD_SOURCE_CRDA);
5950  bad_reg:
5951         kfree(rd);
5952         return r;
5953 }
5954 #endif /* CONFIG_CFG80211_CRDA_SUPPORT */
5955
5956 static int validate_scan_freqs(struct nlattr *freqs)
5957 {
5958         struct nlattr *attr1, *attr2;
5959         int n_channels = 0, tmp1, tmp2;
5960
5961         nla_for_each_nested(attr1, freqs, tmp1) {
5962                 n_channels++;
5963                 /*
5964                  * Some hardware has a limited channel list for
5965                  * scanning, and it is pretty much nonsensical
5966                  * to scan for a channel twice, so disallow that
5967                  * and don't require drivers to check that the
5968                  * channel list they get isn't longer than what
5969                  * they can scan, as long as they can scan all
5970                  * the channels they registered at once.
5971                  */
5972                 nla_for_each_nested(attr2, freqs, tmp2)
5973                         if (attr1 != attr2 &&
5974                             nla_get_u32(attr1) == nla_get_u32(attr2))
5975                                 return 0;
5976         }
5977
5978         return n_channels;
5979 }
5980
5981 static bool is_band_valid(struct wiphy *wiphy, enum nl80211_band b)
5982 {
5983         return b < NUM_NL80211_BANDS && wiphy->bands[b];
5984 }
5985
5986 static int parse_bss_select(struct nlattr *nla, struct wiphy *wiphy,
5987                             struct cfg80211_bss_selection *bss_select)
5988 {
5989         struct nlattr *attr[NL80211_BSS_SELECT_ATTR_MAX + 1];
5990         struct nlattr *nest;
5991         int err;
5992         bool found = false;
5993         int i;
5994
5995         /* only process one nested attribute */
5996         nest = nla_data(nla);
5997         if (!nla_ok(nest, nla_len(nest)))
5998                 return -EINVAL;
5999
6000         err = nla_parse(attr, NL80211_BSS_SELECT_ATTR_MAX, nla_data(nest),
6001                         nla_len(nest), nl80211_bss_select_policy);
6002         if (err)
6003                 return err;
6004
6005         /* only one attribute may be given */
6006         for (i = 0; i <= NL80211_BSS_SELECT_ATTR_MAX; i++) {
6007                 if (attr[i]) {
6008                         if (found)
6009                                 return -EINVAL;
6010                         found = true;
6011                 }
6012         }
6013
6014         bss_select->behaviour = __NL80211_BSS_SELECT_ATTR_INVALID;
6015
6016         if (attr[NL80211_BSS_SELECT_ATTR_RSSI])
6017                 bss_select->behaviour = NL80211_BSS_SELECT_ATTR_RSSI;
6018
6019         if (attr[NL80211_BSS_SELECT_ATTR_BAND_PREF]) {
6020                 bss_select->behaviour = NL80211_BSS_SELECT_ATTR_BAND_PREF;
6021                 bss_select->param.band_pref =
6022                         nla_get_u32(attr[NL80211_BSS_SELECT_ATTR_BAND_PREF]);
6023                 if (!is_band_valid(wiphy, bss_select->param.band_pref))
6024                         return -EINVAL;
6025         }
6026
6027         if (attr[NL80211_BSS_SELECT_ATTR_RSSI_ADJUST]) {
6028                 struct nl80211_bss_select_rssi_adjust *adj_param;
6029
6030                 adj_param = nla_data(attr[NL80211_BSS_SELECT_ATTR_RSSI_ADJUST]);
6031                 bss_select->behaviour = NL80211_BSS_SELECT_ATTR_RSSI_ADJUST;
6032                 bss_select->param.adjust.band = adj_param->band;
6033                 bss_select->param.adjust.delta = adj_param->delta;
6034                 if (!is_band_valid(wiphy, bss_select->param.adjust.band))
6035                         return -EINVAL;
6036         }
6037
6038         /* user-space did not provide behaviour attribute */
6039         if (bss_select->behaviour == __NL80211_BSS_SELECT_ATTR_INVALID)
6040                 return -EINVAL;
6041
6042         if (!(wiphy->bss_select_support & BIT(bss_select->behaviour)))
6043                 return -EINVAL;
6044
6045         return 0;
6046 }
6047
6048 static int nl80211_parse_random_mac(struct nlattr **attrs,
6049                                     u8 *mac_addr, u8 *mac_addr_mask)
6050 {
6051         int i;
6052
6053         if (!attrs[NL80211_ATTR_MAC] && !attrs[NL80211_ATTR_MAC_MASK]) {
6054                 eth_zero_addr(mac_addr);
6055                 eth_zero_addr(mac_addr_mask);
6056                 mac_addr[0] = 0x2;
6057                 mac_addr_mask[0] = 0x3;
6058
6059                 return 0;
6060         }
6061
6062         /* need both or none */
6063         if (!attrs[NL80211_ATTR_MAC] || !attrs[NL80211_ATTR_MAC_MASK])
6064                 return -EINVAL;
6065
6066         memcpy(mac_addr, nla_data(attrs[NL80211_ATTR_MAC]), ETH_ALEN);
6067         memcpy(mac_addr_mask, nla_data(attrs[NL80211_ATTR_MAC_MASK]), ETH_ALEN);
6068
6069         /* don't allow or configure an mcast address */
6070         if (!is_multicast_ether_addr(mac_addr_mask) ||
6071             is_multicast_ether_addr(mac_addr))
6072                 return -EINVAL;
6073
6074         /*
6075          * allow users to pass a MAC address that has bits set outside
6076          * of the mask, but don't bother drivers with having to deal
6077          * with such bits
6078          */
6079         for (i = 0; i < ETH_ALEN; i++)
6080                 mac_addr[i] &= mac_addr_mask[i];
6081
6082         return 0;
6083 }
6084
6085 static int nl80211_trigger_scan(struct sk_buff *skb, struct genl_info *info)
6086 {
6087         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6088         struct wireless_dev *wdev = info->user_ptr[1];
6089         struct cfg80211_scan_request *request;
6090         struct nlattr *attr;
6091         struct wiphy *wiphy;
6092         int err, tmp, n_ssids = 0, n_channels, i;
6093         size_t ie_len;
6094
6095         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
6096                 return -EINVAL;
6097
6098         wiphy = &rdev->wiphy;
6099
6100         if (!rdev->ops->scan)
6101                 return -EOPNOTSUPP;
6102
6103         if (rdev->scan_req || rdev->scan_msg) {
6104                 err = -EBUSY;
6105                 goto unlock;
6106         }
6107
6108         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
6109                 n_channels = validate_scan_freqs(
6110                                 info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]);
6111                 if (!n_channels) {
6112                         err = -EINVAL;
6113                         goto unlock;
6114                 }
6115         } else {
6116                 n_channels = ieee80211_get_num_supported_channels(wiphy);
6117         }
6118
6119         if (info->attrs[NL80211_ATTR_SCAN_SSIDS])
6120                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS], tmp)
6121                         n_ssids++;
6122
6123         if (n_ssids > wiphy->max_scan_ssids) {
6124                 err = -EINVAL;
6125                 goto unlock;
6126         }
6127
6128         if (info->attrs[NL80211_ATTR_IE])
6129                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
6130         else
6131                 ie_len = 0;
6132
6133         if (ie_len > wiphy->max_scan_ie_len) {
6134                 err = -EINVAL;
6135                 goto unlock;
6136         }
6137
6138         request = kzalloc(sizeof(*request)
6139                         + sizeof(*request->ssids) * n_ssids
6140                         + sizeof(*request->channels) * n_channels
6141                         + ie_len, GFP_KERNEL);
6142         if (!request) {
6143                 err = -ENOMEM;
6144                 goto unlock;
6145         }
6146
6147         if (n_ssids)
6148                 request->ssids = (void *)&request->channels[n_channels];
6149         request->n_ssids = n_ssids;
6150         if (ie_len) {
6151                 if (n_ssids)
6152                         request->ie = (void *)(request->ssids + n_ssids);
6153                 else
6154                         request->ie = (void *)(request->channels + n_channels);
6155         }
6156
6157         i = 0;
6158         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
6159                 /* user specified, bail out if channel not found */
6160                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_FREQUENCIES], tmp) {
6161                         struct ieee80211_channel *chan;
6162
6163                         chan = ieee80211_get_channel(wiphy, nla_get_u32(attr));
6164
6165                         if (!chan) {
6166                                 err = -EINVAL;
6167                                 goto out_free;
6168                         }
6169
6170                         /* ignore disabled channels */
6171                         if (chan->flags & IEEE80211_CHAN_DISABLED)
6172                                 continue;
6173
6174                         request->channels[i] = chan;
6175                         i++;
6176                 }
6177         } else {
6178                 enum nl80211_band band;
6179
6180                 /* all channels */
6181                 for (band = 0; band < NUM_NL80211_BANDS; band++) {
6182                         int j;
6183
6184                         if (!wiphy->bands[band])
6185                                 continue;
6186                         for (j = 0; j < wiphy->bands[band]->n_channels; j++) {
6187                                 struct ieee80211_channel *chan;
6188
6189                                 chan = &wiphy->bands[band]->channels[j];
6190
6191                                 if (chan->flags & IEEE80211_CHAN_DISABLED)
6192                                         continue;
6193
6194                                 request->channels[i] = chan;
6195                                 i++;
6196                         }
6197                 }
6198         }
6199
6200         if (!i) {
6201                 err = -EINVAL;
6202                 goto out_free;
6203         }
6204
6205         request->n_channels = i;
6206
6207         i = 0;
6208         if (n_ssids) {
6209                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS], tmp) {
6210                         if (nla_len(attr) > IEEE80211_MAX_SSID_LEN) {
6211                                 err = -EINVAL;
6212                                 goto out_free;
6213                         }
6214                         request->ssids[i].ssid_len = nla_len(attr);
6215                         memcpy(request->ssids[i].ssid, nla_data(attr), nla_len(attr));
6216                         i++;
6217                 }
6218         }
6219
6220         if (info->attrs[NL80211_ATTR_IE]) {
6221                 request->ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
6222                 memcpy((void *)request->ie,
6223                        nla_data(info->attrs[NL80211_ATTR_IE]),
6224                        request->ie_len);
6225         }
6226
6227         for (i = 0; i < NUM_NL80211_BANDS; i++)
6228                 if (wiphy->bands[i])
6229                         request->rates[i] =
6230                                 (1 << wiphy->bands[i]->n_bitrates) - 1;
6231
6232         if (info->attrs[NL80211_ATTR_SCAN_SUPP_RATES]) {
6233                 nla_for_each_nested(attr,
6234                                     info->attrs[NL80211_ATTR_SCAN_SUPP_RATES],
6235                                     tmp) {
6236                         enum nl80211_band band = nla_type(attr);
6237
6238                         if (band < 0 || band >= NUM_NL80211_BANDS) {
6239                                 err = -EINVAL;
6240                                 goto out_free;
6241                         }
6242
6243                         if (!wiphy->bands[band])
6244                                 continue;
6245
6246                         err = ieee80211_get_ratemask(wiphy->bands[band],
6247                                                      nla_data(attr),
6248                                                      nla_len(attr),
6249                                                      &request->rates[band]);
6250                         if (err)
6251                                 goto out_free;
6252                 }
6253         }
6254
6255         if (info->attrs[NL80211_ATTR_MEASUREMENT_DURATION]) {
6256                 if (!wiphy_ext_feature_isset(wiphy,
6257                                         NL80211_EXT_FEATURE_SET_SCAN_DWELL)) {
6258                         err = -EOPNOTSUPP;
6259                         goto out_free;
6260                 }
6261
6262                 request->duration =
6263                         nla_get_u16(info->attrs[NL80211_ATTR_MEASUREMENT_DURATION]);
6264                 request->duration_mandatory =
6265                         nla_get_flag(info->attrs[NL80211_ATTR_MEASUREMENT_DURATION_MANDATORY]);
6266         }
6267
6268         if (info->attrs[NL80211_ATTR_SCAN_FLAGS]) {
6269                 request->flags = nla_get_u32(
6270                         info->attrs[NL80211_ATTR_SCAN_FLAGS]);
6271                 if ((request->flags & NL80211_SCAN_FLAG_LOW_PRIORITY) &&
6272                     !(wiphy->features & NL80211_FEATURE_LOW_PRIORITY_SCAN)) {
6273                         err = -EOPNOTSUPP;
6274                         goto out_free;
6275                 }
6276
6277                 if (request->flags & NL80211_SCAN_FLAG_RANDOM_ADDR) {
6278                         if (!(wiphy->features &
6279                                         NL80211_FEATURE_SCAN_RANDOM_MAC_ADDR)) {
6280                                 err = -EOPNOTSUPP;
6281                                 goto out_free;
6282                         }
6283
6284                         if (wdev->current_bss) {
6285                                 err = -EOPNOTSUPP;
6286                                 goto out_free;
6287                         }
6288
6289                         err = nl80211_parse_random_mac(info->attrs,
6290                                                        request->mac_addr,
6291                                                        request->mac_addr_mask);
6292                         if (err)
6293                                 goto out_free;
6294                 }
6295         }
6296
6297         request->no_cck =
6298                 nla_get_flag(info->attrs[NL80211_ATTR_TX_NO_CCK_RATE]);
6299
6300         if (info->attrs[NL80211_ATTR_MAC])
6301                 memcpy(request->bssid, nla_data(info->attrs[NL80211_ATTR_MAC]),
6302                        ETH_ALEN);
6303         else
6304                 eth_broadcast_addr(request->bssid);
6305
6306         request->wdev = wdev;
6307         request->wiphy = &rdev->wiphy;
6308         request->scan_start = jiffies;
6309
6310         rdev->scan_req = request;
6311         err = rdev_scan(rdev, request);
6312
6313         if (!err) {
6314                 nl80211_send_scan_start(rdev, wdev);
6315                 if (wdev->netdev)
6316                         dev_hold(wdev->netdev);
6317         } else {
6318  out_free:
6319                 rdev->scan_req = NULL;
6320                 kfree(request);
6321         }
6322
6323  unlock:
6324         return err;
6325 }
6326
6327 static int nl80211_abort_scan(struct sk_buff *skb, struct genl_info *info)
6328 {
6329         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6330         struct wireless_dev *wdev = info->user_ptr[1];
6331
6332         if (!rdev->ops->abort_scan)
6333                 return -EOPNOTSUPP;
6334
6335         if (rdev->scan_msg)
6336                 return 0;
6337
6338         if (!rdev->scan_req)
6339                 return -ENOENT;
6340
6341         rdev_abort_scan(rdev, wdev);
6342         return 0;
6343 }
6344
6345 static int
6346 nl80211_parse_sched_scan_plans(struct wiphy *wiphy, int n_plans,
6347                                struct cfg80211_sched_scan_request *request,
6348                                struct nlattr **attrs)
6349 {
6350         int tmp, err, i = 0;
6351         struct nlattr *attr;
6352
6353         if (!attrs[NL80211_ATTR_SCHED_SCAN_PLANS]) {
6354                 u32 interval;
6355
6356                 /*
6357                  * If scan plans are not specified,
6358                  * %NL80211_ATTR_SCHED_SCAN_INTERVAL must be specified. In this
6359                  * case one scan plan will be set with the specified scan
6360                  * interval and infinite number of iterations.
6361                  */
6362                 if (!attrs[NL80211_ATTR_SCHED_SCAN_INTERVAL])
6363                         return -EINVAL;
6364
6365                 interval = nla_get_u32(attrs[NL80211_ATTR_SCHED_SCAN_INTERVAL]);
6366                 if (!interval)
6367                         return -EINVAL;
6368
6369                 request->scan_plans[0].interval =
6370                         DIV_ROUND_UP(interval, MSEC_PER_SEC);
6371                 if (!request->scan_plans[0].interval)
6372                         return -EINVAL;
6373
6374                 if (request->scan_plans[0].interval >
6375                     wiphy->max_sched_scan_plan_interval)
6376                         request->scan_plans[0].interval =
6377                                 wiphy->max_sched_scan_plan_interval;
6378
6379                 return 0;
6380         }
6381
6382         nla_for_each_nested(attr, attrs[NL80211_ATTR_SCHED_SCAN_PLANS], tmp) {
6383                 struct nlattr *plan[NL80211_SCHED_SCAN_PLAN_MAX + 1];
6384
6385                 if (WARN_ON(i >= n_plans))
6386                         return -EINVAL;
6387
6388                 err = nla_parse(plan, NL80211_SCHED_SCAN_PLAN_MAX,
6389                                 nla_data(attr), nla_len(attr),
6390                                 nl80211_plan_policy);
6391                 if (err)
6392                         return err;
6393
6394                 if (!plan[NL80211_SCHED_SCAN_PLAN_INTERVAL])
6395                         return -EINVAL;
6396
6397                 request->scan_plans[i].interval =
6398                         nla_get_u32(plan[NL80211_SCHED_SCAN_PLAN_INTERVAL]);
6399                 if (!request->scan_plans[i].interval ||
6400                     request->scan_plans[i].interval >
6401                     wiphy->max_sched_scan_plan_interval)
6402                         return -EINVAL;
6403
6404                 if (plan[NL80211_SCHED_SCAN_PLAN_ITERATIONS]) {
6405                         request->scan_plans[i].iterations =
6406                                 nla_get_u32(plan[NL80211_SCHED_SCAN_PLAN_ITERATIONS]);
6407                         if (!request->scan_plans[i].iterations ||
6408                             (request->scan_plans[i].iterations >
6409                              wiphy->max_sched_scan_plan_iterations))
6410                                 return -EINVAL;
6411                 } else if (i < n_plans - 1) {
6412                         /*
6413                          * All scan plans but the last one must specify
6414                          * a finite number of iterations
6415                          */
6416                         return -EINVAL;
6417                 }
6418
6419                 i++;
6420         }
6421
6422         /*
6423          * The last scan plan must not specify the number of
6424          * iterations, it is supposed to run infinitely
6425          */
6426         if (request->scan_plans[n_plans - 1].iterations)
6427                 return  -EINVAL;
6428
6429         return 0;
6430 }
6431
6432 static struct cfg80211_sched_scan_request *
6433 nl80211_parse_sched_scan(struct wiphy *wiphy, struct wireless_dev *wdev,
6434                          struct nlattr **attrs)
6435 {
6436         struct cfg80211_sched_scan_request *request;
6437         struct nlattr *attr;
6438         int err, tmp, n_ssids = 0, n_match_sets = 0, n_channels, i, n_plans = 0;
6439         enum nl80211_band band;
6440         size_t ie_len;
6441         struct nlattr *tb[NL80211_SCHED_SCAN_MATCH_ATTR_MAX + 1];
6442         s32 default_match_rssi = NL80211_SCAN_RSSI_THOLD_OFF;
6443
6444         if (!is_valid_ie_attr(attrs[NL80211_ATTR_IE]))
6445                 return ERR_PTR(-EINVAL);
6446
6447         if (attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
6448                 n_channels = validate_scan_freqs(
6449                                 attrs[NL80211_ATTR_SCAN_FREQUENCIES]);
6450                 if (!n_channels)
6451                         return ERR_PTR(-EINVAL);
6452         } else {
6453                 n_channels = ieee80211_get_num_supported_channels(wiphy);
6454         }
6455
6456         if (attrs[NL80211_ATTR_SCAN_SSIDS])
6457                 nla_for_each_nested(attr, attrs[NL80211_ATTR_SCAN_SSIDS],
6458                                     tmp)
6459                         n_ssids++;
6460
6461         if (n_ssids > wiphy->max_sched_scan_ssids)
6462                 return ERR_PTR(-EINVAL);
6463
6464         /*
6465          * First, count the number of 'real' matchsets. Due to an issue with
6466          * the old implementation, matchsets containing only the RSSI attribute
6467          * (NL80211_SCHED_SCAN_MATCH_ATTR_RSSI) are considered as the 'default'
6468          * RSSI for all matchsets, rather than their own matchset for reporting
6469          * all APs with a strong RSSI. This is needed to be compatible with
6470          * older userspace that treated a matchset with only the RSSI as the
6471          * global RSSI for all other matchsets - if there are other matchsets.
6472          */
6473         if (attrs[NL80211_ATTR_SCHED_SCAN_MATCH]) {
6474                 nla_for_each_nested(attr,
6475                                     attrs[NL80211_ATTR_SCHED_SCAN_MATCH],
6476                                     tmp) {
6477                         struct nlattr *rssi;
6478
6479                         err = nla_parse(tb, NL80211_SCHED_SCAN_MATCH_ATTR_MAX,
6480                                         nla_data(attr), nla_len(attr),
6481                                         nl80211_match_policy);
6482                         if (err)
6483                                 return ERR_PTR(err);
6484                         /* add other standalone attributes here */
6485                         if (tb[NL80211_SCHED_SCAN_MATCH_ATTR_SSID]) {
6486                                 n_match_sets++;
6487                                 continue;
6488                         }
6489                         rssi = tb[NL80211_SCHED_SCAN_MATCH_ATTR_RSSI];
6490                         if (rssi)
6491                                 default_match_rssi = nla_get_s32(rssi);
6492                 }
6493         }
6494
6495         /* However, if there's no other matchset, add the RSSI one */
6496         if (!n_match_sets && default_match_rssi != NL80211_SCAN_RSSI_THOLD_OFF)
6497                 n_match_sets = 1;
6498
6499         if (n_match_sets > wiphy->max_match_sets)
6500                 return ERR_PTR(-EINVAL);
6501
6502         if (attrs[NL80211_ATTR_IE])
6503                 ie_len = nla_len(attrs[NL80211_ATTR_IE]);
6504         else
6505                 ie_len = 0;
6506
6507         if (ie_len > wiphy->max_sched_scan_ie_len)
6508                 return ERR_PTR(-EINVAL);
6509
6510         if (attrs[NL80211_ATTR_SCHED_SCAN_PLANS]) {
6511                 /*
6512                  * NL80211_ATTR_SCHED_SCAN_INTERVAL must not be specified since
6513                  * each scan plan already specifies its own interval
6514                  */
6515                 if (attrs[NL80211_ATTR_SCHED_SCAN_INTERVAL])
6516                         return ERR_PTR(-EINVAL);
6517
6518                 nla_for_each_nested(attr,
6519                                     attrs[NL80211_ATTR_SCHED_SCAN_PLANS], tmp)
6520                         n_plans++;
6521         } else {
6522                 /*
6523                  * The scan interval attribute is kept for backward
6524                  * compatibility. If no scan plans are specified and sched scan
6525                  * interval is specified, one scan plan will be set with this
6526                  * scan interval and infinite number of iterations.
6527                  */
6528                 if (!attrs[NL80211_ATTR_SCHED_SCAN_INTERVAL])
6529                         return ERR_PTR(-EINVAL);
6530
6531                 n_plans = 1;
6532         }
6533
6534         if (!n_plans || n_plans > wiphy->max_sched_scan_plans)
6535                 return ERR_PTR(-EINVAL);
6536
6537         request = kzalloc(sizeof(*request)
6538                         + sizeof(*request->ssids) * n_ssids
6539                         + sizeof(*request->match_sets) * n_match_sets
6540                         + sizeof(*request->scan_plans) * n_plans
6541                         + sizeof(*request->channels) * n_channels
6542                         + ie_len, GFP_KERNEL);
6543         if (!request)
6544                 return ERR_PTR(-ENOMEM);
6545
6546         if (n_ssids)
6547                 request->ssids = (void *)&request->channels[n_channels];
6548         request->n_ssids = n_ssids;
6549         if (ie_len) {
6550                 if (n_ssids)
6551                         request->ie = (void *)(request->ssids + n_ssids);
6552                 else
6553                         request->ie = (void *)(request->channels + n_channels);
6554         }
6555
6556         if (n_match_sets) {
6557                 if (request->ie)
6558                         request->match_sets = (void *)(request->ie + ie_len);
6559                 else if (n_ssids)
6560                         request->match_sets =
6561                                 (void *)(request->ssids + n_ssids);
6562                 else
6563                         request->match_sets =
6564                                 (void *)(request->channels + n_channels);
6565         }
6566         request->n_match_sets = n_match_sets;
6567
6568         if (n_match_sets)
6569                 request->scan_plans = (void *)(request->match_sets +
6570                                                n_match_sets);
6571         else if (request->ie)
6572                 request->scan_plans = (void *)(request->ie + ie_len);
6573         else if (n_ssids)
6574                 request->scan_plans = (void *)(request->ssids + n_ssids);
6575         else
6576                 request->scan_plans = (void *)(request->channels + n_channels);
6577
6578         request->n_scan_plans = n_plans;
6579
6580         i = 0;
6581         if (attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
6582                 /* user specified, bail out if channel not found */
6583                 nla_for_each_nested(attr,
6584                                     attrs[NL80211_ATTR_SCAN_FREQUENCIES],
6585                                     tmp) {
6586                         struct ieee80211_channel *chan;
6587
6588                         chan = ieee80211_get_channel(wiphy, nla_get_u32(attr));
6589
6590                         if (!chan) {
6591                                 err = -EINVAL;
6592                                 goto out_free;
6593                         }
6594
6595                         /* ignore disabled channels */
6596                         if (chan->flags & IEEE80211_CHAN_DISABLED)
6597                                 continue;
6598
6599                         request->channels[i] = chan;
6600                         i++;
6601                 }
6602         } else {
6603                 /* all channels */
6604                 for (band = 0; band < NUM_NL80211_BANDS; band++) {
6605                         int j;
6606
6607                         if (!wiphy->bands[band])
6608                                 continue;
6609                         for (j = 0; j < wiphy->bands[band]->n_channels; j++) {
6610                                 struct ieee80211_channel *chan;
6611
6612                                 chan = &wiphy->bands[band]->channels[j];
6613
6614                                 if (chan->flags & IEEE80211_CHAN_DISABLED)
6615                                         continue;
6616
6617                                 request->channels[i] = chan;
6618                                 i++;
6619                         }
6620                 }
6621         }
6622
6623         if (!i) {
6624                 err = -EINVAL;
6625                 goto out_free;
6626         }
6627
6628         request->n_channels = i;
6629
6630         i = 0;
6631         if (n_ssids) {
6632                 nla_for_each_nested(attr, attrs[NL80211_ATTR_SCAN_SSIDS],
6633                                     tmp) {
6634                         if (nla_len(attr) > IEEE80211_MAX_SSID_LEN) {
6635                                 err = -EINVAL;
6636                                 goto out_free;
6637                         }
6638                         request->ssids[i].ssid_len = nla_len(attr);
6639                         memcpy(request->ssids[i].ssid, nla_data(attr),
6640                                nla_len(attr));
6641                         i++;
6642                 }
6643         }
6644
6645         i = 0;
6646         if (attrs[NL80211_ATTR_SCHED_SCAN_MATCH]) {
6647                 nla_for_each_nested(attr,
6648                                     attrs[NL80211_ATTR_SCHED_SCAN_MATCH],
6649                                     tmp) {
6650                         struct nlattr *ssid, *rssi;
6651
6652                         err = nla_parse(tb, NL80211_SCHED_SCAN_MATCH_ATTR_MAX,
6653                                         nla_data(attr), nla_len(attr),
6654                                         nl80211_match_policy);
6655                         if (err)
6656                                 goto out_free;
6657                         ssid = tb[NL80211_SCHED_SCAN_MATCH_ATTR_SSID];
6658                         if (ssid) {
6659                                 if (WARN_ON(i >= n_match_sets)) {
6660                                         /* this indicates a programming error,
6661                                          * the loop above should have verified
6662                                          * things properly
6663                                          */
6664                                         err = -EINVAL;
6665                                         goto out_free;
6666                                 }
6667
6668                                 if (nla_len(ssid) > IEEE80211_MAX_SSID_LEN) {
6669                                         err = -EINVAL;
6670                                         goto out_free;
6671                                 }
6672                                 memcpy(request->match_sets[i].ssid.ssid,
6673                                        nla_data(ssid), nla_len(ssid));
6674                                 request->match_sets[i].ssid.ssid_len =
6675                                         nla_len(ssid);
6676                                 /* special attribute - old implementation w/a */
6677                                 request->match_sets[i].rssi_thold =
6678                                         default_match_rssi;
6679                                 rssi = tb[NL80211_SCHED_SCAN_MATCH_ATTR_RSSI];
6680                                 if (rssi)
6681                                         request->match_sets[i].rssi_thold =
6682                                                 nla_get_s32(rssi);
6683                         }
6684                         i++;
6685                 }
6686
6687                 /* there was no other matchset, so the RSSI one is alone */
6688                 if (i == 0 && n_match_sets)
6689                         request->match_sets[0].rssi_thold = default_match_rssi;
6690
6691                 request->min_rssi_thold = INT_MAX;
6692                 for (i = 0; i < n_match_sets; i++)
6693                         request->min_rssi_thold =
6694                                 min(request->match_sets[i].rssi_thold,
6695                                     request->min_rssi_thold);
6696         } else {
6697                 request->min_rssi_thold = NL80211_SCAN_RSSI_THOLD_OFF;
6698         }
6699
6700         if (ie_len) {
6701                 request->ie_len = ie_len;
6702                 memcpy((void *)request->ie,
6703                        nla_data(attrs[NL80211_ATTR_IE]),
6704                        request->ie_len);
6705         }
6706
6707         if (attrs[NL80211_ATTR_SCAN_FLAGS]) {
6708                 request->flags = nla_get_u32(
6709                         attrs[NL80211_ATTR_SCAN_FLAGS]);
6710                 if ((request->flags & NL80211_SCAN_FLAG_LOW_PRIORITY) &&
6711                     !(wiphy->features & NL80211_FEATURE_LOW_PRIORITY_SCAN)) {
6712                         err = -EOPNOTSUPP;
6713                         goto out_free;
6714                 }
6715
6716                 if (request->flags & NL80211_SCAN_FLAG_RANDOM_ADDR) {
6717                         u32 flg = NL80211_FEATURE_SCHED_SCAN_RANDOM_MAC_ADDR;
6718
6719                         if (!wdev) /* must be net-detect */
6720                                 flg = NL80211_FEATURE_ND_RANDOM_MAC_ADDR;
6721
6722                         if (!(wiphy->features & flg)) {
6723                                 err = -EOPNOTSUPP;
6724                                 goto out_free;
6725                         }
6726
6727                         if (wdev && wdev->current_bss) {
6728                                 err = -EOPNOTSUPP;
6729                                 goto out_free;
6730                         }
6731
6732                         err = nl80211_parse_random_mac(attrs, request->mac_addr,
6733                                                        request->mac_addr_mask);
6734                         if (err)
6735                                 goto out_free;
6736                 }
6737         }
6738
6739         if (attrs[NL80211_ATTR_SCHED_SCAN_DELAY])
6740                 request->delay =
6741                         nla_get_u32(attrs[NL80211_ATTR_SCHED_SCAN_DELAY]);
6742
6743         err = nl80211_parse_sched_scan_plans(wiphy, n_plans, request, attrs);
6744         if (err)
6745                 goto out_free;
6746
6747         request->scan_start = jiffies;
6748
6749         return request;
6750
6751 out_free:
6752         kfree(request);
6753         return ERR_PTR(err);
6754 }
6755
6756 static int nl80211_start_sched_scan(struct sk_buff *skb,
6757                                     struct genl_info *info)
6758 {
6759         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6760         struct net_device *dev = info->user_ptr[1];
6761         struct wireless_dev *wdev = dev->ieee80211_ptr;
6762         struct cfg80211_sched_scan_request *sched_scan_req;
6763         int err;
6764
6765         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN) ||
6766             !rdev->ops->sched_scan_start)
6767                 return -EOPNOTSUPP;
6768
6769         if (rdev->sched_scan_req)
6770                 return -EINPROGRESS;
6771
6772         sched_scan_req = nl80211_parse_sched_scan(&rdev->wiphy, wdev,
6773                                                   info->attrs);
6774
6775         err = PTR_ERR_OR_ZERO(sched_scan_req);
6776         if (err)
6777                 goto out_err;
6778
6779         err = rdev_sched_scan_start(rdev, dev, sched_scan_req);
6780         if (err)
6781                 goto out_free;
6782
6783         sched_scan_req->dev = dev;
6784         sched_scan_req->wiphy = &rdev->wiphy;
6785
6786         if (info->attrs[NL80211_ATTR_SOCKET_OWNER])
6787                 sched_scan_req->owner_nlportid = info->snd_portid;
6788
6789         rcu_assign_pointer(rdev->sched_scan_req, sched_scan_req);
6790
6791         nl80211_send_sched_scan(rdev, dev,
6792                                 NL80211_CMD_START_SCHED_SCAN);
6793         return 0;
6794
6795 out_free:
6796         kfree(sched_scan_req);
6797 out_err:
6798         return err;
6799 }
6800
6801 static int nl80211_stop_sched_scan(struct sk_buff *skb,
6802                                    struct genl_info *info)
6803 {
6804         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6805
6806         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN) ||
6807             !rdev->ops->sched_scan_stop)
6808                 return -EOPNOTSUPP;
6809
6810         return __cfg80211_stop_sched_scan(rdev, false);
6811 }
6812
6813 static int nl80211_start_radar_detection(struct sk_buff *skb,
6814                                          struct genl_info *info)
6815 {
6816         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6817         struct net_device *dev = info->user_ptr[1];
6818         struct wireless_dev *wdev = dev->ieee80211_ptr;
6819         struct cfg80211_chan_def chandef;
6820         enum nl80211_dfs_regions dfs_region;
6821         unsigned int cac_time_ms;
6822         int err;
6823
6824         dfs_region = reg_get_dfs_region(wdev->wiphy);
6825         if (dfs_region == NL80211_DFS_UNSET)
6826                 return -EINVAL;
6827
6828         err = nl80211_parse_chandef(rdev, info, &chandef);
6829         if (err)
6830                 return err;
6831
6832         if (netif_carrier_ok(dev))
6833                 return -EBUSY;
6834
6835         if (wdev->cac_started)
6836                 return -EBUSY;
6837
6838         err = cfg80211_chandef_dfs_required(wdev->wiphy, &chandef,
6839                                             wdev->iftype);
6840         if (err < 0)
6841                 return err;
6842
6843         if (err == 0)
6844                 return -EINVAL;
6845
6846         if (!cfg80211_chandef_dfs_usable(wdev->wiphy, &chandef))
6847                 return -EINVAL;
6848
6849         if (!rdev->ops->start_radar_detection)
6850                 return -EOPNOTSUPP;
6851
6852         cac_time_ms = cfg80211_chandef_dfs_cac_time(&rdev->wiphy, &chandef);
6853         if (WARN_ON(!cac_time_ms))
6854                 cac_time_ms = IEEE80211_DFS_MIN_CAC_TIME_MS;
6855
6856         err = rdev_start_radar_detection(rdev, dev, &chandef, cac_time_ms);
6857         if (!err) {
6858                 wdev->chandef = chandef;
6859                 wdev->cac_started = true;
6860                 wdev->cac_start_time = jiffies;
6861                 wdev->cac_time_ms = cac_time_ms;
6862         }
6863         return err;
6864 }
6865
6866 static int nl80211_channel_switch(struct sk_buff *skb, struct genl_info *info)
6867 {
6868         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6869         struct net_device *dev = info->user_ptr[1];
6870         struct wireless_dev *wdev = dev->ieee80211_ptr;
6871         struct cfg80211_csa_settings params;
6872         /* csa_attrs is defined static to avoid waste of stack size - this
6873          * function is called under RTNL lock, so this should not be a problem.
6874          */
6875         static struct nlattr *csa_attrs[NL80211_ATTR_MAX+1];
6876         int err;
6877         bool need_new_beacon = false;
6878         int len, i;
6879         u32 cs_count;
6880
6881         if (!rdev->ops->channel_switch ||
6882             !(rdev->wiphy.flags & WIPHY_FLAG_HAS_CHANNEL_SWITCH))
6883                 return -EOPNOTSUPP;
6884
6885         switch (dev->ieee80211_ptr->iftype) {
6886         case NL80211_IFTYPE_AP:
6887         case NL80211_IFTYPE_P2P_GO:
6888                 need_new_beacon = true;
6889
6890                 /* useless if AP is not running */
6891                 if (!wdev->beacon_interval)
6892                         return -ENOTCONN;
6893                 break;
6894         case NL80211_IFTYPE_ADHOC:
6895                 if (!wdev->ssid_len)
6896                         return -ENOTCONN;
6897                 break;
6898         case NL80211_IFTYPE_MESH_POINT:
6899                 if (!wdev->mesh_id_len)
6900                         return -ENOTCONN;
6901                 break;
6902         default:
6903                 return -EOPNOTSUPP;
6904         }
6905
6906         memset(&params, 0, sizeof(params));
6907
6908         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ] ||
6909             !info->attrs[NL80211_ATTR_CH_SWITCH_COUNT])
6910                 return -EINVAL;
6911
6912         /* only important for AP, IBSS and mesh create IEs internally */
6913         if (need_new_beacon && !info->attrs[NL80211_ATTR_CSA_IES])
6914                 return -EINVAL;
6915
6916         /* Even though the attribute is u32, the specification says
6917          * u8, so let's make sure we don't overflow.
6918          */
6919         cs_count = nla_get_u32(info->attrs[NL80211_ATTR_CH_SWITCH_COUNT]);
6920         if (cs_count > 255)
6921                 return -EINVAL;
6922
6923         params.count = cs_count;
6924
6925         if (!need_new_beacon)
6926                 goto skip_beacons;
6927
6928         err = nl80211_parse_beacon(info->attrs, &params.beacon_after);
6929         if (err)
6930                 return err;
6931
6932         err = nla_parse_nested(csa_attrs, NL80211_ATTR_MAX,
6933                                info->attrs[NL80211_ATTR_CSA_IES],
6934                                nl80211_policy);
6935         if (err)
6936                 return err;
6937
6938         err = nl80211_parse_beacon(csa_attrs, &params.beacon_csa);
6939         if (err)
6940                 return err;
6941
6942         if (!csa_attrs[NL80211_ATTR_CSA_C_OFF_BEACON])
6943                 return -EINVAL;
6944
6945         len = nla_len(csa_attrs[NL80211_ATTR_CSA_C_OFF_BEACON]);
6946         if (!len || (len % sizeof(u16)))
6947                 return -EINVAL;
6948
6949         params.n_counter_offsets_beacon = len / sizeof(u16);
6950         if (rdev->wiphy.max_num_csa_counters &&
6951             (params.n_counter_offsets_beacon >
6952              rdev->wiphy.max_num_csa_counters))
6953                 return -EINVAL;
6954
6955         params.counter_offsets_beacon =
6956                 nla_data(csa_attrs[NL80211_ATTR_CSA_C_OFF_BEACON]);
6957
6958         /* sanity checks - counters should fit and be the same */
6959         for (i = 0; i < params.n_counter_offsets_beacon; i++) {
6960                 u16 offset = params.counter_offsets_beacon[i];
6961
6962                 if (offset >= params.beacon_csa.tail_len)
6963                         return -EINVAL;
6964
6965                 if (params.beacon_csa.tail[offset] != params.count)
6966                         return -EINVAL;
6967         }
6968
6969         if (csa_attrs[NL80211_ATTR_CSA_C_OFF_PRESP]) {
6970                 len = nla_len(csa_attrs[NL80211_ATTR_CSA_C_OFF_PRESP]);
6971                 if (!len || (len % sizeof(u16)))
6972                         return -EINVAL;
6973
6974                 params.n_counter_offsets_presp = len / sizeof(u16);
6975                 if (rdev->wiphy.max_num_csa_counters &&
6976                     (params.n_counter_offsets_beacon >
6977                      rdev->wiphy.max_num_csa_counters))
6978                         return -EINVAL;
6979
6980                 params.counter_offsets_presp =
6981                         nla_data(csa_attrs[NL80211_ATTR_CSA_C_OFF_PRESP]);
6982
6983                 /* sanity checks - counters should fit and be the same */
6984                 for (i = 0; i < params.n_counter_offsets_presp; i++) {
6985                         u16 offset = params.counter_offsets_presp[i];
6986
6987                         if (offset >= params.beacon_csa.probe_resp_len)
6988                                 return -EINVAL;
6989
6990                         if (params.beacon_csa.probe_resp[offset] !=
6991                             params.count)
6992                                 return -EINVAL;
6993                 }
6994         }
6995
6996 skip_beacons:
6997         err = nl80211_parse_chandef(rdev, info, &params.chandef);
6998         if (err)
6999                 return err;
7000
7001         if (!cfg80211_reg_can_beacon_relax(&rdev->wiphy, &params.chandef,
7002                                            wdev->iftype))
7003                 return -EINVAL;
7004
7005         err = cfg80211_chandef_dfs_required(wdev->wiphy,
7006                                             &params.chandef,
7007                                             wdev->iftype);
7008         if (err < 0)
7009                 return err;
7010
7011         if (err > 0)
7012                 params.radar_required = true;
7013
7014         if (info->attrs[NL80211_ATTR_CH_SWITCH_BLOCK_TX])
7015                 params.block_tx = true;
7016
7017         wdev_lock(wdev);
7018         err = rdev_channel_switch(rdev, dev, &params);
7019         wdev_unlock(wdev);
7020
7021         return err;
7022 }
7023
7024 static int nl80211_send_bss(struct sk_buff *msg, struct netlink_callback *cb,
7025                             u32 seq, int flags,
7026                             struct cfg80211_registered_device *rdev,
7027                             struct wireless_dev *wdev,
7028                             struct cfg80211_internal_bss *intbss)
7029 {
7030         struct cfg80211_bss *res = &intbss->pub;
7031         const struct cfg80211_bss_ies *ies;
7032         void *hdr;
7033         struct nlattr *bss;
7034
7035         ASSERT_WDEV_LOCK(wdev);
7036
7037         hdr = nl80211hdr_put(msg, NETLINK_CB(cb->skb).portid, seq, flags,
7038                              NL80211_CMD_NEW_SCAN_RESULTS);
7039         if (!hdr)
7040                 return -1;
7041
7042         genl_dump_check_consistent(cb, hdr, &nl80211_fam);
7043
7044         if (nla_put_u32(msg, NL80211_ATTR_GENERATION, rdev->bss_generation))
7045                 goto nla_put_failure;
7046         if (wdev->netdev &&
7047             nla_put_u32(msg, NL80211_ATTR_IFINDEX, wdev->netdev->ifindex))
7048                 goto nla_put_failure;
7049         if (nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
7050                               NL80211_ATTR_PAD))
7051                 goto nla_put_failure;
7052
7053         bss = nla_nest_start(msg, NL80211_ATTR_BSS);
7054         if (!bss)
7055                 goto nla_put_failure;
7056         if ((!is_zero_ether_addr(res->bssid) &&
7057              nla_put(msg, NL80211_BSS_BSSID, ETH_ALEN, res->bssid)))
7058                 goto nla_put_failure;
7059
7060         rcu_read_lock();
7061         /* indicate whether we have probe response data or not */
7062         if (rcu_access_pointer(res->proberesp_ies) &&
7063             nla_put_flag(msg, NL80211_BSS_PRESP_DATA))
7064                 goto fail_unlock_rcu;
7065
7066         /* this pointer prefers to be pointed to probe response data
7067          * but is always valid
7068          */
7069         ies = rcu_dereference(res->ies);
7070         if (ies) {
7071                 if (nla_put_u64_64bit(msg, NL80211_BSS_TSF, ies->tsf,
7072                                       NL80211_BSS_PAD))
7073                         goto fail_unlock_rcu;
7074                 if (ies->len && nla_put(msg, NL80211_BSS_INFORMATION_ELEMENTS,
7075                                         ies->len, ies->data))
7076                         goto fail_unlock_rcu;
7077         }
7078
7079         /* and this pointer is always (unless driver didn't know) beacon data */
7080         ies = rcu_dereference(res->beacon_ies);
7081         if (ies && ies->from_beacon) {
7082                 if (nla_put_u64_64bit(msg, NL80211_BSS_BEACON_TSF, ies->tsf,
7083                                       NL80211_BSS_PAD))
7084                         goto fail_unlock_rcu;
7085                 if (ies->len && nla_put(msg, NL80211_BSS_BEACON_IES,
7086                                         ies->len, ies->data))
7087                         goto fail_unlock_rcu;
7088         }
7089         rcu_read_unlock();
7090
7091         if (res->beacon_interval &&
7092             nla_put_u16(msg, NL80211_BSS_BEACON_INTERVAL, res->beacon_interval))
7093                 goto nla_put_failure;
7094         if (nla_put_u16(msg, NL80211_BSS_CAPABILITY, res->capability) ||
7095             nla_put_u32(msg, NL80211_BSS_FREQUENCY, res->channel->center_freq) ||
7096             nla_put_u32(msg, NL80211_BSS_CHAN_WIDTH, res->scan_width) ||
7097             nla_put_u32(msg, NL80211_BSS_SEEN_MS_AGO,
7098                         jiffies_to_msecs(jiffies - intbss->ts)))
7099                 goto nla_put_failure;
7100
7101         if (intbss->parent_tsf &&
7102             (nla_put_u64_64bit(msg, NL80211_BSS_PARENT_TSF,
7103                                intbss->parent_tsf, NL80211_BSS_PAD) ||
7104              nla_put(msg, NL80211_BSS_PARENT_BSSID, ETH_ALEN,
7105                      intbss->parent_bssid)))
7106                 goto nla_put_failure;
7107
7108         if (intbss->ts_boottime &&
7109             nla_put_u64_64bit(msg, NL80211_BSS_LAST_SEEN_BOOTTIME,
7110                               intbss->ts_boottime, NL80211_BSS_PAD))
7111                 goto nla_put_failure;
7112
7113         switch (rdev->wiphy.signal_type) {
7114         case CFG80211_SIGNAL_TYPE_MBM:
7115                 if (nla_put_u32(msg, NL80211_BSS_SIGNAL_MBM, res->signal))
7116                         goto nla_put_failure;
7117                 break;
7118         case CFG80211_SIGNAL_TYPE_UNSPEC:
7119                 if (nla_put_u8(msg, NL80211_BSS_SIGNAL_UNSPEC, res->signal))
7120                         goto nla_put_failure;
7121                 break;
7122         default:
7123                 break;
7124         }
7125
7126         switch (wdev->iftype) {
7127         case NL80211_IFTYPE_P2P_CLIENT:
7128         case NL80211_IFTYPE_STATION:
7129                 if (intbss == wdev->current_bss &&
7130                     nla_put_u32(msg, NL80211_BSS_STATUS,
7131                                 NL80211_BSS_STATUS_ASSOCIATED))
7132                         goto nla_put_failure;
7133                 break;
7134         case NL80211_IFTYPE_ADHOC:
7135                 if (intbss == wdev->current_bss &&
7136                     nla_put_u32(msg, NL80211_BSS_STATUS,
7137                                 NL80211_BSS_STATUS_IBSS_JOINED))
7138                         goto nla_put_failure;
7139                 break;
7140         default:
7141                 break;
7142         }
7143
7144         nla_nest_end(msg, bss);
7145
7146         genlmsg_end(msg, hdr);
7147         return 0;
7148
7149  fail_unlock_rcu:
7150         rcu_read_unlock();
7151  nla_put_failure:
7152         genlmsg_cancel(msg, hdr);
7153         return -EMSGSIZE;
7154 }
7155
7156 static int nl80211_dump_scan(struct sk_buff *skb, struct netlink_callback *cb)
7157 {
7158         struct cfg80211_registered_device *rdev;
7159         struct cfg80211_internal_bss *scan;
7160         struct wireless_dev *wdev;
7161         int start = cb->args[2], idx = 0;
7162         int err;
7163
7164         err = nl80211_prepare_wdev_dump(skb, cb, &rdev, &wdev);
7165         if (err)
7166                 return err;
7167
7168         wdev_lock(wdev);
7169         spin_lock_bh(&rdev->bss_lock);
7170         cfg80211_bss_expire(rdev);
7171
7172         cb->seq = rdev->bss_generation;
7173
7174         list_for_each_entry(scan, &rdev->bss_list, list) {
7175                 if (++idx <= start)
7176                         continue;
7177                 if (nl80211_send_bss(skb, cb,
7178                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
7179                                 rdev, wdev, scan) < 0) {
7180                         idx--;
7181                         break;
7182                 }
7183         }
7184
7185         spin_unlock_bh(&rdev->bss_lock);
7186         wdev_unlock(wdev);
7187
7188         cb->args[2] = idx;
7189         nl80211_finish_wdev_dump(rdev);
7190
7191         return skb->len;
7192 }
7193
7194 static int nl80211_send_survey(struct sk_buff *msg, u32 portid, u32 seq,
7195                                int flags, struct net_device *dev,
7196                                bool allow_radio_stats,
7197                                struct survey_info *survey)
7198 {
7199         void *hdr;
7200         struct nlattr *infoattr;
7201
7202         /* skip radio stats if userspace didn't request them */
7203         if (!survey->channel && !allow_radio_stats)
7204                 return 0;
7205
7206         hdr = nl80211hdr_put(msg, portid, seq, flags,
7207                              NL80211_CMD_NEW_SURVEY_RESULTS);
7208         if (!hdr)
7209                 return -ENOMEM;
7210
7211         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex))
7212                 goto nla_put_failure;
7213
7214         infoattr = nla_nest_start(msg, NL80211_ATTR_SURVEY_INFO);
7215         if (!infoattr)
7216                 goto nla_put_failure;
7217
7218         if (survey->channel &&
7219             nla_put_u32(msg, NL80211_SURVEY_INFO_FREQUENCY,
7220                         survey->channel->center_freq))
7221                 goto nla_put_failure;
7222
7223         if ((survey->filled & SURVEY_INFO_NOISE_DBM) &&
7224             nla_put_u8(msg, NL80211_SURVEY_INFO_NOISE, survey->noise))
7225                 goto nla_put_failure;
7226         if ((survey->filled & SURVEY_INFO_IN_USE) &&
7227             nla_put_flag(msg, NL80211_SURVEY_INFO_IN_USE))
7228                 goto nla_put_failure;
7229         if ((survey->filled & SURVEY_INFO_TIME) &&
7230             nla_put_u64_64bit(msg, NL80211_SURVEY_INFO_TIME,
7231                         survey->time, NL80211_SURVEY_INFO_PAD))
7232                 goto nla_put_failure;
7233         if ((survey->filled & SURVEY_INFO_TIME_BUSY) &&
7234             nla_put_u64_64bit(msg, NL80211_SURVEY_INFO_TIME_BUSY,
7235                               survey->time_busy, NL80211_SURVEY_INFO_PAD))
7236                 goto nla_put_failure;
7237         if ((survey->filled & SURVEY_INFO_TIME_EXT_BUSY) &&
7238             nla_put_u64_64bit(msg, NL80211_SURVEY_INFO_TIME_EXT_BUSY,
7239                               survey->time_ext_busy, NL80211_SURVEY_INFO_PAD))
7240                 goto nla_put_failure;
7241         if ((survey->filled & SURVEY_INFO_TIME_RX) &&
7242             nla_put_u64_64bit(msg, NL80211_SURVEY_INFO_TIME_RX,
7243                               survey->time_rx, NL80211_SURVEY_INFO_PAD))
7244                 goto nla_put_failure;
7245         if ((survey->filled & SURVEY_INFO_TIME_TX) &&
7246             nla_put_u64_64bit(msg, NL80211_SURVEY_INFO_TIME_TX,
7247                               survey->time_tx, NL80211_SURVEY_INFO_PAD))
7248                 goto nla_put_failure;
7249         if ((survey->filled & SURVEY_INFO_TIME_SCAN) &&
7250             nla_put_u64_64bit(msg, NL80211_SURVEY_INFO_TIME_SCAN,
7251                               survey->time_scan, NL80211_SURVEY_INFO_PAD))
7252                 goto nla_put_failure;
7253
7254         nla_nest_end(msg, infoattr);
7255
7256         genlmsg_end(msg, hdr);
7257         return 0;
7258
7259  nla_put_failure:
7260         genlmsg_cancel(msg, hdr);
7261         return -EMSGSIZE;
7262 }
7263
7264 static int nl80211_dump_survey(struct sk_buff *skb, struct netlink_callback *cb)
7265 {
7266         struct survey_info survey;
7267         struct cfg80211_registered_device *rdev;
7268         struct wireless_dev *wdev;
7269         int survey_idx = cb->args[2];
7270         int res;
7271         bool radio_stats;
7272
7273         res = nl80211_prepare_wdev_dump(skb, cb, &rdev, &wdev);
7274         if (res)
7275                 return res;
7276
7277         /* prepare_wdev_dump parsed the attributes */
7278         radio_stats = nl80211_fam.attrbuf[NL80211_ATTR_SURVEY_RADIO_STATS];
7279
7280         if (!wdev->netdev) {
7281                 res = -EINVAL;
7282                 goto out_err;
7283         }
7284
7285         if (!rdev->ops->dump_survey) {
7286                 res = -EOPNOTSUPP;
7287                 goto out_err;
7288         }
7289
7290         while (1) {
7291                 res = rdev_dump_survey(rdev, wdev->netdev, survey_idx, &survey);
7292                 if (res == -ENOENT)
7293                         break;
7294                 if (res)
7295                         goto out_err;
7296
7297                 /* don't send disabled channels, but do send non-channel data */
7298                 if (survey.channel &&
7299                     survey.channel->flags & IEEE80211_CHAN_DISABLED) {
7300                         survey_idx++;
7301                         continue;
7302                 }
7303
7304                 if (nl80211_send_survey(skb,
7305                                 NETLINK_CB(cb->skb).portid,
7306                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
7307                                 wdev->netdev, radio_stats, &survey) < 0)
7308                         goto out;
7309                 survey_idx++;
7310         }
7311
7312  out:
7313         cb->args[2] = survey_idx;
7314         res = skb->len;
7315  out_err:
7316         nl80211_finish_wdev_dump(rdev);
7317         return res;
7318 }
7319
7320 static bool nl80211_valid_wpa_versions(u32 wpa_versions)
7321 {
7322         return !(wpa_versions & ~(NL80211_WPA_VERSION_1 |
7323                                   NL80211_WPA_VERSION_2));
7324 }
7325
7326 static int nl80211_authenticate(struct sk_buff *skb, struct genl_info *info)
7327 {
7328         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7329         struct net_device *dev = info->user_ptr[1];
7330         struct ieee80211_channel *chan;
7331         const u8 *bssid, *ssid, *ie = NULL, *sae_data = NULL;
7332         int err, ssid_len, ie_len = 0, sae_data_len = 0;
7333         enum nl80211_auth_type auth_type;
7334         struct key_parse key;
7335         bool local_state_change;
7336
7337         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
7338                 return -EINVAL;
7339
7340         if (!info->attrs[NL80211_ATTR_MAC])
7341                 return -EINVAL;
7342
7343         if (!info->attrs[NL80211_ATTR_AUTH_TYPE])
7344                 return -EINVAL;
7345
7346         if (!info->attrs[NL80211_ATTR_SSID])
7347                 return -EINVAL;
7348
7349         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ])
7350                 return -EINVAL;
7351
7352         err = nl80211_parse_key(info, &key);
7353         if (err)
7354                 return err;
7355
7356         if (key.idx >= 0) {
7357                 if (key.type != -1 && key.type != NL80211_KEYTYPE_GROUP)
7358                         return -EINVAL;
7359                 if (!key.p.key || !key.p.key_len)
7360                         return -EINVAL;
7361                 if ((key.p.cipher != WLAN_CIPHER_SUITE_WEP40 ||
7362                      key.p.key_len != WLAN_KEY_LEN_WEP40) &&
7363                     (key.p.cipher != WLAN_CIPHER_SUITE_WEP104 ||
7364                      key.p.key_len != WLAN_KEY_LEN_WEP104))
7365                         return -EINVAL;
7366                 if (key.idx > 4)
7367                         return -EINVAL;
7368         } else {
7369                 key.p.key_len = 0;
7370                 key.p.key = NULL;
7371         }
7372
7373         if (key.idx >= 0) {
7374                 int i;
7375                 bool ok = false;
7376
7377                 for (i = 0; i < rdev->wiphy.n_cipher_suites; i++) {
7378                         if (key.p.cipher == rdev->wiphy.cipher_suites[i]) {
7379                                 ok = true;
7380                                 break;
7381                         }
7382                 }
7383                 if (!ok)
7384                         return -EINVAL;
7385         }
7386
7387         if (!rdev->ops->auth)
7388                 return -EOPNOTSUPP;
7389
7390         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
7391             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
7392                 return -EOPNOTSUPP;
7393
7394         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
7395         chan = nl80211_get_valid_chan(&rdev->wiphy,
7396                                       info->attrs[NL80211_ATTR_WIPHY_FREQ]);
7397         if (!chan)
7398                 return -EINVAL;
7399
7400         ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
7401         ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
7402
7403         if (info->attrs[NL80211_ATTR_IE]) {
7404                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
7405                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
7406         }
7407
7408         auth_type = nla_get_u32(info->attrs[NL80211_ATTR_AUTH_TYPE]);
7409         if (!nl80211_valid_auth_type(rdev, auth_type, NL80211_CMD_AUTHENTICATE))
7410                 return -EINVAL;
7411
7412         if (auth_type == NL80211_AUTHTYPE_SAE &&
7413             !info->attrs[NL80211_ATTR_SAE_DATA])
7414                 return -EINVAL;
7415
7416         if (info->attrs[NL80211_ATTR_SAE_DATA]) {
7417                 if (auth_type != NL80211_AUTHTYPE_SAE)
7418                         return -EINVAL;
7419                 sae_data = nla_data(info->attrs[NL80211_ATTR_SAE_DATA]);
7420                 sae_data_len = nla_len(info->attrs[NL80211_ATTR_SAE_DATA]);
7421                 /* need to include at least Auth Transaction and Status Code */
7422                 if (sae_data_len < 4)
7423                         return -EINVAL;
7424         }
7425
7426         local_state_change = !!info->attrs[NL80211_ATTR_LOCAL_STATE_CHANGE];
7427
7428         /*
7429          * Since we no longer track auth state, ignore
7430          * requests to only change local state.
7431          */
7432         if (local_state_change)
7433                 return 0;
7434
7435         wdev_lock(dev->ieee80211_ptr);
7436         err = cfg80211_mlme_auth(rdev, dev, chan, auth_type, bssid,
7437                                  ssid, ssid_len, ie, ie_len,
7438                                  key.p.key, key.p.key_len, key.idx,
7439                                  sae_data, sae_data_len);
7440         wdev_unlock(dev->ieee80211_ptr);
7441         return err;
7442 }
7443
7444 static int nl80211_crypto_settings(struct cfg80211_registered_device *rdev,
7445                                    struct genl_info *info,
7446                                    struct cfg80211_crypto_settings *settings,
7447                                    int cipher_limit)
7448 {
7449         memset(settings, 0, sizeof(*settings));
7450
7451         settings->control_port = info->attrs[NL80211_ATTR_CONTROL_PORT];
7452
7453         if (info->attrs[NL80211_ATTR_CONTROL_PORT_ETHERTYPE]) {
7454                 u16 proto;
7455
7456                 proto = nla_get_u16(
7457                         info->attrs[NL80211_ATTR_CONTROL_PORT_ETHERTYPE]);
7458                 settings->control_port_ethertype = cpu_to_be16(proto);
7459                 if (!(rdev->wiphy.flags & WIPHY_FLAG_CONTROL_PORT_PROTOCOL) &&
7460                     proto != ETH_P_PAE)
7461                         return -EINVAL;
7462                 if (info->attrs[NL80211_ATTR_CONTROL_PORT_NO_ENCRYPT])
7463                         settings->control_port_no_encrypt = true;
7464         } else
7465                 settings->control_port_ethertype = cpu_to_be16(ETH_P_PAE);
7466
7467         if (info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]) {
7468                 void *data;
7469                 int len, i;
7470
7471                 data = nla_data(info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]);
7472                 len = nla_len(info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]);
7473                 settings->n_ciphers_pairwise = len / sizeof(u32);
7474
7475                 if (len % sizeof(u32))
7476                         return -EINVAL;
7477
7478                 if (settings->n_ciphers_pairwise > cipher_limit)
7479                         return -EINVAL;
7480
7481                 memcpy(settings->ciphers_pairwise, data, len);
7482
7483                 for (i = 0; i < settings->n_ciphers_pairwise; i++)
7484                         if (!cfg80211_supported_cipher_suite(
7485                                         &rdev->wiphy,
7486                                         settings->ciphers_pairwise[i]))
7487                                 return -EINVAL;
7488         }
7489
7490         if (info->attrs[NL80211_ATTR_CIPHER_SUITE_GROUP]) {
7491                 settings->cipher_group =
7492                         nla_get_u32(info->attrs[NL80211_ATTR_CIPHER_SUITE_GROUP]);
7493                 if (!cfg80211_supported_cipher_suite(&rdev->wiphy,
7494                                                      settings->cipher_group))
7495                         return -EINVAL;
7496         }
7497
7498         if (info->attrs[NL80211_ATTR_WPA_VERSIONS]) {
7499                 settings->wpa_versions =
7500                         nla_get_u32(info->attrs[NL80211_ATTR_WPA_VERSIONS]);
7501                 if (!nl80211_valid_wpa_versions(settings->wpa_versions))
7502                         return -EINVAL;
7503         }
7504
7505         if (info->attrs[NL80211_ATTR_AKM_SUITES]) {
7506                 void *data;
7507                 int len;
7508
7509                 data = nla_data(info->attrs[NL80211_ATTR_AKM_SUITES]);
7510                 len = nla_len(info->attrs[NL80211_ATTR_AKM_SUITES]);
7511                 settings->n_akm_suites = len / sizeof(u32);
7512
7513                 if (len % sizeof(u32))
7514                         return -EINVAL;
7515
7516                 if (settings->n_akm_suites > NL80211_MAX_NR_AKM_SUITES)
7517                         return -EINVAL;
7518
7519                 memcpy(settings->akm_suites, data, len);
7520         }
7521
7522         return 0;
7523 }
7524
7525 static int nl80211_associate(struct sk_buff *skb, struct genl_info *info)
7526 {
7527         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7528         struct net_device *dev = info->user_ptr[1];
7529         struct ieee80211_channel *chan;
7530         struct cfg80211_assoc_request req = {};
7531         const u8 *bssid, *ssid;
7532         int err, ssid_len = 0;
7533
7534         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
7535                 return -EINVAL;
7536
7537         if (!info->attrs[NL80211_ATTR_MAC] ||
7538             !info->attrs[NL80211_ATTR_SSID] ||
7539             !info->attrs[NL80211_ATTR_WIPHY_FREQ])
7540                 return -EINVAL;
7541
7542         if (!rdev->ops->assoc)
7543                 return -EOPNOTSUPP;
7544
7545         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
7546             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
7547                 return -EOPNOTSUPP;
7548
7549         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
7550
7551         chan = nl80211_get_valid_chan(&rdev->wiphy,
7552                                       info->attrs[NL80211_ATTR_WIPHY_FREQ]);
7553         if (!chan)
7554                 return -EINVAL;
7555
7556         ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
7557         ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
7558
7559         if (info->attrs[NL80211_ATTR_IE]) {
7560                 req.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
7561                 req.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
7562         }
7563
7564         if (info->attrs[NL80211_ATTR_USE_MFP]) {
7565                 enum nl80211_mfp mfp =
7566                         nla_get_u32(info->attrs[NL80211_ATTR_USE_MFP]);
7567                 if (mfp == NL80211_MFP_REQUIRED)
7568                         req.use_mfp = true;
7569                 else if (mfp != NL80211_MFP_NO)
7570                         return -EINVAL;
7571         }
7572
7573         if (info->attrs[NL80211_ATTR_PREV_BSSID])
7574                 req.prev_bssid = nla_data(info->attrs[NL80211_ATTR_PREV_BSSID]);
7575
7576         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_HT]))
7577                 req.flags |= ASSOC_REQ_DISABLE_HT;
7578
7579         if (info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
7580                 memcpy(&req.ht_capa_mask,
7581                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]),
7582                        sizeof(req.ht_capa_mask));
7583
7584         if (info->attrs[NL80211_ATTR_HT_CAPABILITY]) {
7585                 if (!info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
7586                         return -EINVAL;
7587                 memcpy(&req.ht_capa,
7588                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]),
7589                        sizeof(req.ht_capa));
7590         }
7591
7592         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_VHT]))
7593                 req.flags |= ASSOC_REQ_DISABLE_VHT;
7594
7595         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK])
7596                 memcpy(&req.vht_capa_mask,
7597                        nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK]),
7598                        sizeof(req.vht_capa_mask));
7599
7600         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY]) {
7601                 if (!info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK])
7602                         return -EINVAL;
7603                 memcpy(&req.vht_capa,
7604                        nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY]),
7605                        sizeof(req.vht_capa));
7606         }
7607
7608         if (nla_get_flag(info->attrs[NL80211_ATTR_USE_RRM])) {
7609                 if (!((rdev->wiphy.features &
7610                         NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES) &&
7611                        (rdev->wiphy.features & NL80211_FEATURE_QUIET)) &&
7612                     !wiphy_ext_feature_isset(&rdev->wiphy,
7613                                              NL80211_EXT_FEATURE_RRM))
7614                         return -EINVAL;
7615                 req.flags |= ASSOC_REQ_USE_RRM;
7616         }
7617
7618         err = nl80211_crypto_settings(rdev, info, &req.crypto, 1);
7619         if (!err) {
7620                 wdev_lock(dev->ieee80211_ptr);
7621                 err = cfg80211_mlme_assoc(rdev, dev, chan, bssid,
7622                                           ssid, ssid_len, &req);
7623                 wdev_unlock(dev->ieee80211_ptr);
7624         }
7625
7626         return err;
7627 }
7628
7629 static int nl80211_deauthenticate(struct sk_buff *skb, struct genl_info *info)
7630 {
7631         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7632         struct net_device *dev = info->user_ptr[1];
7633         const u8 *ie = NULL, *bssid;
7634         int ie_len = 0, err;
7635         u16 reason_code;
7636         bool local_state_change;
7637
7638         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
7639                 return -EINVAL;
7640
7641         if (!info->attrs[NL80211_ATTR_MAC])
7642                 return -EINVAL;
7643
7644         if (!info->attrs[NL80211_ATTR_REASON_CODE])
7645                 return -EINVAL;
7646
7647         if (!rdev->ops->deauth)
7648                 return -EOPNOTSUPP;
7649
7650         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
7651             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
7652                 return -EOPNOTSUPP;
7653
7654         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
7655
7656         reason_code = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
7657         if (reason_code == 0) {
7658                 /* Reason Code 0 is reserved */
7659                 return -EINVAL;
7660         }
7661
7662         if (info->attrs[NL80211_ATTR_IE]) {
7663                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
7664                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
7665         }
7666
7667         local_state_change = !!info->attrs[NL80211_ATTR_LOCAL_STATE_CHANGE];
7668
7669         wdev_lock(dev->ieee80211_ptr);
7670         err = cfg80211_mlme_deauth(rdev, dev, bssid, ie, ie_len, reason_code,
7671                                    local_state_change);
7672         wdev_unlock(dev->ieee80211_ptr);
7673         return err;
7674 }
7675
7676 static int nl80211_disassociate(struct sk_buff *skb, struct genl_info *info)
7677 {
7678         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7679         struct net_device *dev = info->user_ptr[1];
7680         const u8 *ie = NULL, *bssid;
7681         int ie_len = 0, err;
7682         u16 reason_code;
7683         bool local_state_change;
7684
7685         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
7686                 return -EINVAL;
7687
7688         if (!info->attrs[NL80211_ATTR_MAC])
7689                 return -EINVAL;
7690
7691         if (!info->attrs[NL80211_ATTR_REASON_CODE])
7692                 return -EINVAL;
7693
7694         if (!rdev->ops->disassoc)
7695                 return -EOPNOTSUPP;
7696
7697         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
7698             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
7699                 return -EOPNOTSUPP;
7700
7701         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
7702
7703         reason_code = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
7704         if (reason_code == 0) {
7705                 /* Reason Code 0 is reserved */
7706                 return -EINVAL;
7707         }
7708
7709         if (info->attrs[NL80211_ATTR_IE]) {
7710                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
7711                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
7712         }
7713
7714         local_state_change = !!info->attrs[NL80211_ATTR_LOCAL_STATE_CHANGE];
7715
7716         wdev_lock(dev->ieee80211_ptr);
7717         err = cfg80211_mlme_disassoc(rdev, dev, bssid, ie, ie_len, reason_code,
7718                                      local_state_change);
7719         wdev_unlock(dev->ieee80211_ptr);
7720         return err;
7721 }
7722
7723 static bool
7724 nl80211_parse_mcast_rate(struct cfg80211_registered_device *rdev,
7725                          int mcast_rate[NUM_NL80211_BANDS],
7726                          int rateval)
7727 {
7728         struct wiphy *wiphy = &rdev->wiphy;
7729         bool found = false;
7730         int band, i;
7731
7732         for (band = 0; band < NUM_NL80211_BANDS; band++) {
7733                 struct ieee80211_supported_band *sband;
7734
7735                 sband = wiphy->bands[band];
7736                 if (!sband)
7737                         continue;
7738
7739                 for (i = 0; i < sband->n_bitrates; i++) {
7740                         if (sband->bitrates[i].bitrate == rateval) {
7741                                 mcast_rate[band] = i + 1;
7742                                 found = true;
7743                                 break;
7744                         }
7745                 }
7746         }
7747
7748         return found;
7749 }
7750
7751 static int nl80211_join_ibss(struct sk_buff *skb, struct genl_info *info)
7752 {
7753         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7754         struct net_device *dev = info->user_ptr[1];
7755         struct cfg80211_ibss_params ibss;
7756         struct wiphy *wiphy;
7757         struct cfg80211_cached_keys *connkeys = NULL;
7758         int err;
7759
7760         memset(&ibss, 0, sizeof(ibss));
7761
7762         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
7763                 return -EINVAL;
7764
7765         if (!info->attrs[NL80211_ATTR_SSID] ||
7766             !nla_len(info->attrs[NL80211_ATTR_SSID]))
7767                 return -EINVAL;
7768
7769         ibss.beacon_interval = 100;
7770
7771         if (info->attrs[NL80211_ATTR_BEACON_INTERVAL]) {
7772                 ibss.beacon_interval =
7773                         nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
7774                 if (ibss.beacon_interval < 1 || ibss.beacon_interval > 10000)
7775                         return -EINVAL;
7776         }
7777
7778         if (!rdev->ops->join_ibss)
7779                 return -EOPNOTSUPP;
7780
7781         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC)
7782                 return -EOPNOTSUPP;
7783
7784         wiphy = &rdev->wiphy;
7785
7786         if (info->attrs[NL80211_ATTR_MAC]) {
7787                 ibss.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
7788
7789                 if (!is_valid_ether_addr(ibss.bssid))
7790                         return -EINVAL;
7791         }
7792         ibss.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
7793         ibss.ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
7794
7795         if (info->attrs[NL80211_ATTR_IE]) {
7796                 ibss.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
7797                 ibss.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
7798         }
7799
7800         err = nl80211_parse_chandef(rdev, info, &ibss.chandef);
7801         if (err)
7802                 return err;
7803
7804         if (!cfg80211_reg_can_beacon(&rdev->wiphy, &ibss.chandef,
7805                                      NL80211_IFTYPE_ADHOC))
7806                 return -EINVAL;
7807
7808         switch (ibss.chandef.width) {
7809         case NL80211_CHAN_WIDTH_5:
7810         case NL80211_CHAN_WIDTH_10:
7811         case NL80211_CHAN_WIDTH_20_NOHT:
7812                 break;
7813         case NL80211_CHAN_WIDTH_20:
7814         case NL80211_CHAN_WIDTH_40:
7815                 if (!(rdev->wiphy.features & NL80211_FEATURE_HT_IBSS))
7816                         return -EINVAL;
7817                 break;
7818         case NL80211_CHAN_WIDTH_80:
7819         case NL80211_CHAN_WIDTH_80P80:
7820         case NL80211_CHAN_WIDTH_160:
7821                 if (!(rdev->wiphy.features & NL80211_FEATURE_HT_IBSS))
7822                         return -EINVAL;
7823                 if (!wiphy_ext_feature_isset(&rdev->wiphy,
7824                                              NL80211_EXT_FEATURE_VHT_IBSS))
7825                         return -EINVAL;
7826                 break;
7827         default:
7828                 return -EINVAL;
7829         }
7830
7831         ibss.channel_fixed = !!info->attrs[NL80211_ATTR_FREQ_FIXED];
7832         ibss.privacy = !!info->attrs[NL80211_ATTR_PRIVACY];
7833
7834         if (info->attrs[NL80211_ATTR_BSS_BASIC_RATES]) {
7835                 u8 *rates =
7836                         nla_data(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
7837                 int n_rates =
7838                         nla_len(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
7839                 struct ieee80211_supported_band *sband =
7840                         wiphy->bands[ibss.chandef.chan->band];
7841
7842                 err = ieee80211_get_ratemask(sband, rates, n_rates,
7843                                              &ibss.basic_rates);
7844                 if (err)
7845                         return err;
7846         }
7847
7848         if (info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
7849                 memcpy(&ibss.ht_capa_mask,
7850                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]),
7851                        sizeof(ibss.ht_capa_mask));
7852
7853         if (info->attrs[NL80211_ATTR_HT_CAPABILITY]) {
7854                 if (!info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
7855                         return -EINVAL;
7856                 memcpy(&ibss.ht_capa,
7857                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]),
7858                        sizeof(ibss.ht_capa));
7859         }
7860
7861         if (info->attrs[NL80211_ATTR_MCAST_RATE] &&
7862             !nl80211_parse_mcast_rate(rdev, ibss.mcast_rate,
7863                         nla_get_u32(info->attrs[NL80211_ATTR_MCAST_RATE])))
7864                 return -EINVAL;
7865
7866         if (ibss.privacy && info->attrs[NL80211_ATTR_KEYS]) {
7867                 bool no_ht = false;
7868
7869                 connkeys = nl80211_parse_connkeys(rdev,
7870                                           info->attrs[NL80211_ATTR_KEYS],
7871                                           &no_ht);
7872                 if (IS_ERR(connkeys))
7873                         return PTR_ERR(connkeys);
7874
7875                 if ((ibss.chandef.width != NL80211_CHAN_WIDTH_20_NOHT) &&
7876                     no_ht) {
7877                         kzfree(connkeys);
7878                         return -EINVAL;
7879                 }
7880         }
7881
7882         ibss.control_port =
7883                 nla_get_flag(info->attrs[NL80211_ATTR_CONTROL_PORT]);
7884
7885         ibss.userspace_handles_dfs =
7886                 nla_get_flag(info->attrs[NL80211_ATTR_HANDLE_DFS]);
7887
7888         err = cfg80211_join_ibss(rdev, dev, &ibss, connkeys);
7889         if (err)
7890                 kzfree(connkeys);
7891         return err;
7892 }
7893
7894 static int nl80211_leave_ibss(struct sk_buff *skb, struct genl_info *info)
7895 {
7896         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7897         struct net_device *dev = info->user_ptr[1];
7898
7899         if (!rdev->ops->leave_ibss)
7900                 return -EOPNOTSUPP;
7901
7902         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC)
7903                 return -EOPNOTSUPP;
7904
7905         return cfg80211_leave_ibss(rdev, dev, false);
7906 }
7907
7908 static int nl80211_set_mcast_rate(struct sk_buff *skb, struct genl_info *info)
7909 {
7910         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7911         struct net_device *dev = info->user_ptr[1];
7912         int mcast_rate[NUM_NL80211_BANDS];
7913         u32 nla_rate;
7914         int err;
7915
7916         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC &&
7917             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT &&
7918             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_OCB)
7919                 return -EOPNOTSUPP;
7920
7921         if (!rdev->ops->set_mcast_rate)
7922                 return -EOPNOTSUPP;
7923
7924         memset(mcast_rate, 0, sizeof(mcast_rate));
7925
7926         if (!info->attrs[NL80211_ATTR_MCAST_RATE])
7927                 return -EINVAL;
7928
7929         nla_rate = nla_get_u32(info->attrs[NL80211_ATTR_MCAST_RATE]);
7930         if (!nl80211_parse_mcast_rate(rdev, mcast_rate, nla_rate))
7931                 return -EINVAL;
7932
7933         err = rdev_set_mcast_rate(rdev, dev, mcast_rate);
7934
7935         return err;
7936 }
7937
7938 static struct sk_buff *
7939 __cfg80211_alloc_vendor_skb(struct cfg80211_registered_device *rdev,
7940                             struct wireless_dev *wdev, int approxlen,
7941                             u32 portid, u32 seq, enum nl80211_commands cmd,
7942                             enum nl80211_attrs attr,
7943                             const struct nl80211_vendor_cmd_info *info,
7944                             gfp_t gfp)
7945 {
7946         struct sk_buff *skb;
7947         void *hdr;
7948         struct nlattr *data;
7949
7950         skb = nlmsg_new(approxlen + 100, gfp);
7951         if (!skb)
7952                 return NULL;
7953
7954         hdr = nl80211hdr_put(skb, portid, seq, 0, cmd);
7955         if (!hdr) {
7956                 kfree_skb(skb);
7957                 return NULL;
7958         }
7959
7960         if (nla_put_u32(skb, NL80211_ATTR_WIPHY, rdev->wiphy_idx))
7961                 goto nla_put_failure;
7962
7963         if (info) {
7964                 if (nla_put_u32(skb, NL80211_ATTR_VENDOR_ID,
7965                                 info->vendor_id))
7966                         goto nla_put_failure;
7967                 if (nla_put_u32(skb, NL80211_ATTR_VENDOR_SUBCMD,
7968                                 info->subcmd))
7969                         goto nla_put_failure;
7970         }
7971
7972         if (wdev) {
7973                 if (nla_put_u64_64bit(skb, NL80211_ATTR_WDEV,
7974                                       wdev_id(wdev), NL80211_ATTR_PAD))
7975                         goto nla_put_failure;
7976                 if (wdev->netdev &&
7977                     nla_put_u32(skb, NL80211_ATTR_IFINDEX,
7978                                 wdev->netdev->ifindex))
7979                         goto nla_put_failure;
7980         }
7981
7982         data = nla_nest_start(skb, attr);
7983
7984         ((void **)skb->cb)[0] = rdev;
7985         ((void **)skb->cb)[1] = hdr;
7986         ((void **)skb->cb)[2] = data;
7987
7988         return skb;
7989
7990  nla_put_failure:
7991         kfree_skb(skb);
7992         return NULL;
7993 }
7994
7995 struct sk_buff *__cfg80211_alloc_event_skb(struct wiphy *wiphy,
7996                                            struct wireless_dev *wdev,
7997                                            enum nl80211_commands cmd,
7998                                            enum nl80211_attrs attr,
7999                                            int vendor_event_idx,
8000                                            int approxlen, gfp_t gfp)
8001 {
8002         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
8003         const struct nl80211_vendor_cmd_info *info;
8004
8005         switch (cmd) {
8006         case NL80211_CMD_TESTMODE:
8007                 if (WARN_ON(vendor_event_idx != -1))
8008                         return NULL;
8009                 info = NULL;
8010                 break;
8011         case NL80211_CMD_VENDOR:
8012                 if (WARN_ON(vendor_event_idx < 0 ||
8013                             vendor_event_idx >= wiphy->n_vendor_events))
8014                         return NULL;
8015                 info = &wiphy->vendor_events[vendor_event_idx];
8016                 break;
8017         default:
8018                 WARN_ON(1);
8019                 return NULL;
8020         }
8021
8022         return __cfg80211_alloc_vendor_skb(rdev, wdev, approxlen, 0, 0,
8023                                            cmd, attr, info, gfp);
8024 }
8025 EXPORT_SYMBOL(__cfg80211_alloc_event_skb);
8026
8027 void __cfg80211_send_event_skb(struct sk_buff *skb, gfp_t gfp)
8028 {
8029         struct cfg80211_registered_device *rdev = ((void **)skb->cb)[0];
8030         void *hdr = ((void **)skb->cb)[1];
8031         struct nlattr *data = ((void **)skb->cb)[2];
8032         enum nl80211_multicast_groups mcgrp = NL80211_MCGRP_TESTMODE;
8033
8034         /* clear CB data for netlink core to own from now on */
8035         memset(skb->cb, 0, sizeof(skb->cb));
8036
8037         nla_nest_end(skb, data);
8038         genlmsg_end(skb, hdr);
8039
8040         if (data->nla_type == NL80211_ATTR_VENDOR_DATA)
8041                 mcgrp = NL80211_MCGRP_VENDOR;
8042
8043         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), skb, 0,
8044                                 mcgrp, gfp);
8045 }
8046 EXPORT_SYMBOL(__cfg80211_send_event_skb);
8047
8048 #ifdef CONFIG_NL80211_TESTMODE
8049 static int nl80211_testmode_do(struct sk_buff *skb, struct genl_info *info)
8050 {
8051         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8052         struct wireless_dev *wdev =
8053                 __cfg80211_wdev_from_attrs(genl_info_net(info), info->attrs);
8054         int err;
8055
8056         if (!rdev->ops->testmode_cmd)
8057                 return -EOPNOTSUPP;
8058
8059         if (IS_ERR(wdev)) {
8060                 err = PTR_ERR(wdev);
8061                 if (err != -EINVAL)
8062                         return err;
8063                 wdev = NULL;
8064         } else if (wdev->wiphy != &rdev->wiphy) {
8065                 return -EINVAL;
8066         }
8067
8068         if (!info->attrs[NL80211_ATTR_TESTDATA])
8069                 return -EINVAL;
8070
8071         rdev->cur_cmd_info = info;
8072         err = rdev_testmode_cmd(rdev, wdev,
8073                                 nla_data(info->attrs[NL80211_ATTR_TESTDATA]),
8074                                 nla_len(info->attrs[NL80211_ATTR_TESTDATA]));
8075         rdev->cur_cmd_info = NULL;
8076
8077         return err;
8078 }
8079
8080 static int nl80211_testmode_dump(struct sk_buff *skb,
8081                                  struct netlink_callback *cb)
8082 {
8083         struct cfg80211_registered_device *rdev;
8084         int err;
8085         long phy_idx;
8086         void *data = NULL;
8087         int data_len = 0;
8088
8089         rtnl_lock();
8090
8091         if (cb->args[0]) {
8092                 /*
8093                  * 0 is a valid index, but not valid for args[0],
8094                  * so we need to offset by 1.
8095                  */
8096                 phy_idx = cb->args[0] - 1;
8097         } else {
8098                 err = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
8099                                   nl80211_fam.attrbuf, nl80211_fam.maxattr,
8100                                   nl80211_policy);
8101                 if (err)
8102                         goto out_err;
8103
8104                 rdev = __cfg80211_rdev_from_attrs(sock_net(skb->sk),
8105                                                   nl80211_fam.attrbuf);
8106                 if (IS_ERR(rdev)) {
8107                         err = PTR_ERR(rdev);
8108                         goto out_err;
8109                 }
8110                 phy_idx = rdev->wiphy_idx;
8111                 rdev = NULL;
8112
8113                 if (nl80211_fam.attrbuf[NL80211_ATTR_TESTDATA])
8114                         cb->args[1] =
8115                                 (long)nl80211_fam.attrbuf[NL80211_ATTR_TESTDATA];
8116         }
8117
8118         if (cb->args[1]) {
8119                 data = nla_data((void *)cb->args[1]);
8120                 data_len = nla_len((void *)cb->args[1]);
8121         }
8122
8123         rdev = cfg80211_rdev_by_wiphy_idx(phy_idx);
8124         if (!rdev) {
8125                 err = -ENOENT;
8126                 goto out_err;
8127         }
8128
8129         if (!rdev->ops->testmode_dump) {
8130                 err = -EOPNOTSUPP;
8131                 goto out_err;
8132         }
8133
8134         while (1) {
8135                 void *hdr = nl80211hdr_put(skb, NETLINK_CB(cb->skb).portid,
8136                                            cb->nlh->nlmsg_seq, NLM_F_MULTI,
8137                                            NL80211_CMD_TESTMODE);
8138                 struct nlattr *tmdata;
8139
8140                 if (!hdr)
8141                         break;
8142
8143                 if (nla_put_u32(skb, NL80211_ATTR_WIPHY, phy_idx)) {
8144                         genlmsg_cancel(skb, hdr);
8145                         break;
8146                 }
8147
8148                 tmdata = nla_nest_start(skb, NL80211_ATTR_TESTDATA);
8149                 if (!tmdata) {
8150                         genlmsg_cancel(skb, hdr);
8151                         break;
8152                 }
8153                 err = rdev_testmode_dump(rdev, skb, cb, data, data_len);
8154                 nla_nest_end(skb, tmdata);
8155
8156                 if (err == -ENOBUFS || err == -ENOENT) {
8157                         genlmsg_cancel(skb, hdr);
8158                         break;
8159                 } else if (err) {
8160                         genlmsg_cancel(skb, hdr);
8161                         goto out_err;
8162                 }
8163
8164                 genlmsg_end(skb, hdr);
8165         }
8166
8167         err = skb->len;
8168         /* see above */
8169         cb->args[0] = phy_idx + 1;
8170  out_err:
8171         rtnl_unlock();
8172         return err;
8173 }
8174 #endif
8175
8176 static int nl80211_connect(struct sk_buff *skb, struct genl_info *info)
8177 {
8178         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8179         struct net_device *dev = info->user_ptr[1];
8180         struct cfg80211_connect_params connect;
8181         struct wiphy *wiphy;
8182         struct cfg80211_cached_keys *connkeys = NULL;
8183         int err;
8184
8185         memset(&connect, 0, sizeof(connect));
8186
8187         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
8188                 return -EINVAL;
8189
8190         if (!info->attrs[NL80211_ATTR_SSID] ||
8191             !nla_len(info->attrs[NL80211_ATTR_SSID]))
8192                 return -EINVAL;
8193
8194         if (info->attrs[NL80211_ATTR_AUTH_TYPE]) {
8195                 connect.auth_type =
8196                         nla_get_u32(info->attrs[NL80211_ATTR_AUTH_TYPE]);
8197                 if (!nl80211_valid_auth_type(rdev, connect.auth_type,
8198                                              NL80211_CMD_CONNECT))
8199                         return -EINVAL;
8200         } else
8201                 connect.auth_type = NL80211_AUTHTYPE_AUTOMATIC;
8202
8203         connect.privacy = info->attrs[NL80211_ATTR_PRIVACY];
8204
8205         err = nl80211_crypto_settings(rdev, info, &connect.crypto,
8206                                       NL80211_MAX_NR_CIPHER_SUITES);
8207         if (err)
8208                 return err;
8209
8210         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
8211             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
8212                 return -EOPNOTSUPP;
8213
8214         wiphy = &rdev->wiphy;
8215
8216         connect.bg_scan_period = -1;
8217         if (info->attrs[NL80211_ATTR_BG_SCAN_PERIOD] &&
8218                 (wiphy->flags & WIPHY_FLAG_SUPPORTS_FW_ROAM)) {
8219                 connect.bg_scan_period =
8220                         nla_get_u16(info->attrs[NL80211_ATTR_BG_SCAN_PERIOD]);
8221         }
8222
8223         if (info->attrs[NL80211_ATTR_MAC])
8224                 connect.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
8225         else if (info->attrs[NL80211_ATTR_MAC_HINT])
8226                 connect.bssid_hint =
8227                         nla_data(info->attrs[NL80211_ATTR_MAC_HINT]);
8228         connect.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
8229         connect.ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
8230
8231         if (info->attrs[NL80211_ATTR_IE]) {
8232                 connect.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
8233                 connect.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
8234         }
8235
8236         if (info->attrs[NL80211_ATTR_USE_MFP]) {
8237                 connect.mfp = nla_get_u32(info->attrs[NL80211_ATTR_USE_MFP]);
8238                 if (connect.mfp != NL80211_MFP_REQUIRED &&
8239                     connect.mfp != NL80211_MFP_NO)
8240                         return -EINVAL;
8241         } else {
8242                 connect.mfp = NL80211_MFP_NO;
8243         }
8244
8245         if (info->attrs[NL80211_ATTR_PREV_BSSID])
8246                 connect.prev_bssid =
8247                         nla_data(info->attrs[NL80211_ATTR_PREV_BSSID]);
8248
8249         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
8250                 connect.channel = nl80211_get_valid_chan(
8251                         wiphy, info->attrs[NL80211_ATTR_WIPHY_FREQ]);
8252                 if (!connect.channel)
8253                         return -EINVAL;
8254         } else if (info->attrs[NL80211_ATTR_WIPHY_FREQ_HINT]) {
8255                 connect.channel_hint = nl80211_get_valid_chan(
8256                         wiphy, info->attrs[NL80211_ATTR_WIPHY_FREQ_HINT]);
8257                 if (!connect.channel_hint)
8258                         return -EINVAL;
8259         }
8260
8261         if (connect.privacy && info->attrs[NL80211_ATTR_KEYS]) {
8262                 connkeys = nl80211_parse_connkeys(rdev,
8263                                           info->attrs[NL80211_ATTR_KEYS], NULL);
8264                 if (IS_ERR(connkeys))
8265                         return PTR_ERR(connkeys);
8266         }
8267
8268         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_HT]))
8269                 connect.flags |= ASSOC_REQ_DISABLE_HT;
8270
8271         if (info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
8272                 memcpy(&connect.ht_capa_mask,
8273                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]),
8274                        sizeof(connect.ht_capa_mask));
8275
8276         if (info->attrs[NL80211_ATTR_HT_CAPABILITY]) {
8277                 if (!info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]) {
8278                         kzfree(connkeys);
8279                         return -EINVAL;
8280                 }
8281                 memcpy(&connect.ht_capa,
8282                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]),
8283                        sizeof(connect.ht_capa));
8284         }
8285
8286         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_VHT]))
8287                 connect.flags |= ASSOC_REQ_DISABLE_VHT;
8288
8289         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK])
8290                 memcpy(&connect.vht_capa_mask,
8291                        nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK]),
8292                        sizeof(connect.vht_capa_mask));
8293
8294         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY]) {
8295                 if (!info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK]) {
8296                         kzfree(connkeys);
8297                         return -EINVAL;
8298                 }
8299                 memcpy(&connect.vht_capa,
8300                        nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY]),
8301                        sizeof(connect.vht_capa));
8302         }
8303
8304         if (nla_get_flag(info->attrs[NL80211_ATTR_USE_RRM])) {
8305                 if (!((rdev->wiphy.features &
8306                         NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES) &&
8307                        (rdev->wiphy.features & NL80211_FEATURE_QUIET)) &&
8308                     !wiphy_ext_feature_isset(&rdev->wiphy,
8309                                              NL80211_EXT_FEATURE_RRM)) {
8310                         kzfree(connkeys);
8311                         return -EINVAL;
8312                 }
8313                 connect.flags |= ASSOC_REQ_USE_RRM;
8314         }
8315
8316         connect.pbss = nla_get_flag(info->attrs[NL80211_ATTR_PBSS]);
8317         if (connect.pbss && !rdev->wiphy.bands[NL80211_BAND_60GHZ]) {
8318                 kzfree(connkeys);
8319                 return -EOPNOTSUPP;
8320         }
8321
8322         if (info->attrs[NL80211_ATTR_BSS_SELECT]) {
8323                 /* bss selection makes no sense if bssid is set */
8324                 if (connect.bssid) {
8325                         kzfree(connkeys);
8326                         return -EINVAL;
8327                 }
8328
8329                 err = parse_bss_select(info->attrs[NL80211_ATTR_BSS_SELECT],
8330                                        wiphy, &connect.bss_select);
8331                 if (err) {
8332                         kzfree(connkeys);
8333                         return err;
8334                 }
8335         }
8336
8337         wdev_lock(dev->ieee80211_ptr);
8338         err = cfg80211_connect(rdev, dev, &connect, connkeys,
8339                                connect.prev_bssid);
8340         wdev_unlock(dev->ieee80211_ptr);
8341         if (err)
8342                 kzfree(connkeys);
8343         return err;
8344 }
8345
8346 static int nl80211_disconnect(struct sk_buff *skb, struct genl_info *info)
8347 {
8348         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8349         struct net_device *dev = info->user_ptr[1];
8350         u16 reason;
8351         int ret;
8352
8353         if (!info->attrs[NL80211_ATTR_REASON_CODE])
8354                 reason = WLAN_REASON_DEAUTH_LEAVING;
8355         else
8356                 reason = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
8357
8358         if (reason == 0)
8359                 return -EINVAL;
8360
8361         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
8362             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
8363                 return -EOPNOTSUPP;
8364
8365         wdev_lock(dev->ieee80211_ptr);
8366         ret = cfg80211_disconnect(rdev, dev, reason, true);
8367         wdev_unlock(dev->ieee80211_ptr);
8368         return ret;
8369 }
8370
8371 static int nl80211_wiphy_netns(struct sk_buff *skb, struct genl_info *info)
8372 {
8373         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8374         struct net *net;
8375         int err;
8376
8377         if (info->attrs[NL80211_ATTR_PID]) {
8378                 u32 pid = nla_get_u32(info->attrs[NL80211_ATTR_PID]);
8379
8380                 net = get_net_ns_by_pid(pid);
8381         } else if (info->attrs[NL80211_ATTR_NETNS_FD]) {
8382                 u32 fd = nla_get_u32(info->attrs[NL80211_ATTR_NETNS_FD]);
8383
8384                 net = get_net_ns_by_fd(fd);
8385         } else {
8386                 return -EINVAL;
8387         }
8388
8389         if (IS_ERR(net))
8390                 return PTR_ERR(net);
8391
8392         err = 0;
8393
8394         /* check if anything to do */
8395         if (!net_eq(wiphy_net(&rdev->wiphy), net))
8396                 err = cfg80211_switch_netns(rdev, net);
8397
8398         put_net(net);
8399         return err;
8400 }
8401
8402 static int nl80211_setdel_pmksa(struct sk_buff *skb, struct genl_info *info)
8403 {
8404         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8405         int (*rdev_ops)(struct wiphy *wiphy, struct net_device *dev,
8406                         struct cfg80211_pmksa *pmksa) = NULL;
8407         struct net_device *dev = info->user_ptr[1];
8408         struct cfg80211_pmksa pmksa;
8409
8410         memset(&pmksa, 0, sizeof(struct cfg80211_pmksa));
8411
8412         if (!info->attrs[NL80211_ATTR_MAC])
8413                 return -EINVAL;
8414
8415         if (!info->attrs[NL80211_ATTR_PMKID])
8416                 return -EINVAL;
8417
8418         pmksa.pmkid = nla_data(info->attrs[NL80211_ATTR_PMKID]);
8419         pmksa.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
8420
8421         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
8422             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
8423                 return -EOPNOTSUPP;
8424
8425         switch (info->genlhdr->cmd) {
8426         case NL80211_CMD_SET_PMKSA:
8427                 rdev_ops = rdev->ops->set_pmksa;
8428                 break;
8429         case NL80211_CMD_DEL_PMKSA:
8430                 rdev_ops = rdev->ops->del_pmksa;
8431                 break;
8432         default:
8433                 WARN_ON(1);
8434                 break;
8435         }
8436
8437         if (!rdev_ops)
8438                 return -EOPNOTSUPP;
8439
8440         return rdev_ops(&rdev->wiphy, dev, &pmksa);
8441 }
8442
8443 static int nl80211_flush_pmksa(struct sk_buff *skb, struct genl_info *info)
8444 {
8445         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8446         struct net_device *dev = info->user_ptr[1];
8447
8448         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
8449             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
8450                 return -EOPNOTSUPP;
8451
8452         if (!rdev->ops->flush_pmksa)
8453                 return -EOPNOTSUPP;
8454
8455         return rdev_flush_pmksa(rdev, dev);
8456 }
8457
8458 static int nl80211_tdls_mgmt(struct sk_buff *skb, struct genl_info *info)
8459 {
8460         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8461         struct net_device *dev = info->user_ptr[1];
8462         u8 action_code, dialog_token;
8463         u32 peer_capability = 0;
8464         u16 status_code;
8465         u8 *peer;
8466         bool initiator;
8467
8468         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) ||
8469             !rdev->ops->tdls_mgmt)
8470                 return -EOPNOTSUPP;
8471
8472         if (!info->attrs[NL80211_ATTR_TDLS_ACTION] ||
8473             !info->attrs[NL80211_ATTR_STATUS_CODE] ||
8474             !info->attrs[NL80211_ATTR_TDLS_DIALOG_TOKEN] ||
8475             !info->attrs[NL80211_ATTR_IE] ||
8476             !info->attrs[NL80211_ATTR_MAC])
8477                 return -EINVAL;
8478
8479         peer = nla_data(info->attrs[NL80211_ATTR_MAC]);
8480         action_code = nla_get_u8(info->attrs[NL80211_ATTR_TDLS_ACTION]);
8481         status_code = nla_get_u16(info->attrs[NL80211_ATTR_STATUS_CODE]);
8482         dialog_token = nla_get_u8(info->attrs[NL80211_ATTR_TDLS_DIALOG_TOKEN]);
8483         initiator = nla_get_flag(info->attrs[NL80211_ATTR_TDLS_INITIATOR]);
8484         if (info->attrs[NL80211_ATTR_TDLS_PEER_CAPABILITY])
8485                 peer_capability =
8486                         nla_get_u32(info->attrs[NL80211_ATTR_TDLS_PEER_CAPABILITY]);
8487
8488         return rdev_tdls_mgmt(rdev, dev, peer, action_code,
8489                               dialog_token, status_code, peer_capability,
8490                               initiator,
8491                               nla_data(info->attrs[NL80211_ATTR_IE]),
8492                               nla_len(info->attrs[NL80211_ATTR_IE]));
8493 }
8494
8495 static int nl80211_tdls_oper(struct sk_buff *skb, struct genl_info *info)
8496 {
8497         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8498         struct net_device *dev = info->user_ptr[1];
8499         enum nl80211_tdls_operation operation;
8500         u8 *peer;
8501
8502         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) ||
8503             !rdev->ops->tdls_oper)
8504                 return -EOPNOTSUPP;
8505
8506         if (!info->attrs[NL80211_ATTR_TDLS_OPERATION] ||
8507             !info->attrs[NL80211_ATTR_MAC])
8508                 return -EINVAL;
8509
8510         operation = nla_get_u8(info->attrs[NL80211_ATTR_TDLS_OPERATION]);
8511         peer = nla_data(info->attrs[NL80211_ATTR_MAC]);
8512
8513         return rdev_tdls_oper(rdev, dev, peer, operation);
8514 }
8515
8516 static int nl80211_remain_on_channel(struct sk_buff *skb,
8517                                      struct genl_info *info)
8518 {
8519         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8520         struct wireless_dev *wdev = info->user_ptr[1];
8521         struct cfg80211_chan_def chandef;
8522         struct sk_buff *msg;
8523         void *hdr;
8524         u64 cookie;
8525         u32 duration;
8526         int err;
8527
8528         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ] ||
8529             !info->attrs[NL80211_ATTR_DURATION])
8530                 return -EINVAL;
8531
8532         duration = nla_get_u32(info->attrs[NL80211_ATTR_DURATION]);
8533
8534         if (!rdev->ops->remain_on_channel ||
8535             !(rdev->wiphy.flags & WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL))
8536                 return -EOPNOTSUPP;
8537
8538         /*
8539          * We should be on that channel for at least a minimum amount of
8540          * time (10ms) but no longer than the driver supports.
8541          */
8542         if (duration < NL80211_MIN_REMAIN_ON_CHANNEL_TIME ||
8543             duration > rdev->wiphy.max_remain_on_channel_duration)
8544                 return -EINVAL;
8545
8546         err = nl80211_parse_chandef(rdev, info, &chandef);
8547         if (err)
8548                 return err;
8549
8550         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8551         if (!msg)
8552                 return -ENOMEM;
8553
8554         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
8555                              NL80211_CMD_REMAIN_ON_CHANNEL);
8556         if (!hdr) {
8557                 err = -ENOBUFS;
8558                 goto free_msg;
8559         }
8560
8561         err = rdev_remain_on_channel(rdev, wdev, chandef.chan,
8562                                      duration, &cookie);
8563
8564         if (err)
8565                 goto free_msg;
8566
8567         if (nla_put_u64_64bit(msg, NL80211_ATTR_COOKIE, cookie,
8568                               NL80211_ATTR_PAD))
8569                 goto nla_put_failure;
8570
8571         genlmsg_end(msg, hdr);
8572
8573         return genlmsg_reply(msg, info);
8574
8575  nla_put_failure:
8576         err = -ENOBUFS;
8577  free_msg:
8578         nlmsg_free(msg);
8579         return err;
8580 }
8581
8582 static int nl80211_cancel_remain_on_channel(struct sk_buff *skb,
8583                                             struct genl_info *info)
8584 {
8585         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8586         struct wireless_dev *wdev = info->user_ptr[1];
8587         u64 cookie;
8588
8589         if (!info->attrs[NL80211_ATTR_COOKIE])
8590                 return -EINVAL;
8591
8592         if (!rdev->ops->cancel_remain_on_channel)
8593                 return -EOPNOTSUPP;
8594
8595         cookie = nla_get_u64(info->attrs[NL80211_ATTR_COOKIE]);
8596
8597         return rdev_cancel_remain_on_channel(rdev, wdev, cookie);
8598 }
8599
8600 static u32 rateset_to_mask(struct ieee80211_supported_band *sband,
8601                            u8 *rates, u8 rates_len)
8602 {
8603         u8 i;
8604         u32 mask = 0;
8605
8606         for (i = 0; i < rates_len; i++) {
8607                 int rate = (rates[i] & 0x7f) * 5;
8608                 int ridx;
8609
8610                 for (ridx = 0; ridx < sband->n_bitrates; ridx++) {
8611                         struct ieee80211_rate *srate =
8612                                 &sband->bitrates[ridx];
8613                         if (rate == srate->bitrate) {
8614                                 mask |= 1 << ridx;
8615                                 break;
8616                         }
8617                 }
8618                 if (ridx == sband->n_bitrates)
8619                         return 0; /* rate not found */
8620         }
8621
8622         return mask;
8623 }
8624
8625 static bool ht_rateset_to_mask(struct ieee80211_supported_band *sband,
8626                                u8 *rates, u8 rates_len,
8627                                u8 mcs[IEEE80211_HT_MCS_MASK_LEN])
8628 {
8629         u8 i;
8630
8631         memset(mcs, 0, IEEE80211_HT_MCS_MASK_LEN);
8632
8633         for (i = 0; i < rates_len; i++) {
8634                 int ridx, rbit;
8635
8636                 ridx = rates[i] / 8;
8637                 rbit = BIT(rates[i] % 8);
8638
8639                 /* check validity */
8640                 if ((ridx < 0) || (ridx >= IEEE80211_HT_MCS_MASK_LEN))
8641                         return false;
8642
8643                 /* check availability */
8644                 if (sband->ht_cap.mcs.rx_mask[ridx] & rbit)
8645                         mcs[ridx] |= rbit;
8646                 else
8647                         return false;
8648         }
8649
8650         return true;
8651 }
8652
8653 static u16 vht_mcs_map_to_mcs_mask(u8 vht_mcs_map)
8654 {
8655         u16 mcs_mask = 0;
8656
8657         switch (vht_mcs_map) {
8658         case IEEE80211_VHT_MCS_NOT_SUPPORTED:
8659                 break;
8660         case IEEE80211_VHT_MCS_SUPPORT_0_7:
8661                 mcs_mask = 0x00FF;
8662                 break;
8663         case IEEE80211_VHT_MCS_SUPPORT_0_8:
8664                 mcs_mask = 0x01FF;
8665                 break;
8666         case IEEE80211_VHT_MCS_SUPPORT_0_9:
8667                 mcs_mask = 0x03FF;
8668                 break;
8669         default:
8670                 break;
8671         }
8672
8673         return mcs_mask;
8674 }
8675
8676 static void vht_build_mcs_mask(u16 vht_mcs_map,
8677                                u16 vht_mcs_mask[NL80211_VHT_NSS_MAX])
8678 {
8679         u8 nss;
8680
8681         for (nss = 0; nss < NL80211_VHT_NSS_MAX; nss++) {
8682                 vht_mcs_mask[nss] = vht_mcs_map_to_mcs_mask(vht_mcs_map & 0x03);
8683                 vht_mcs_map >>= 2;
8684         }
8685 }
8686
8687 static bool vht_set_mcs_mask(struct ieee80211_supported_band *sband,
8688                              struct nl80211_txrate_vht *txrate,
8689                              u16 mcs[NL80211_VHT_NSS_MAX])
8690 {
8691         u16 tx_mcs_map = le16_to_cpu(sband->vht_cap.vht_mcs.tx_mcs_map);
8692         u16 tx_mcs_mask[NL80211_VHT_NSS_MAX] = {};
8693         u8 i;
8694
8695         if (!sband->vht_cap.vht_supported)
8696                 return false;
8697
8698         memset(mcs, 0, sizeof(u16) * NL80211_VHT_NSS_MAX);
8699
8700         /* Build vht_mcs_mask from VHT capabilities */
8701         vht_build_mcs_mask(tx_mcs_map, tx_mcs_mask);
8702
8703         for (i = 0; i < NL80211_VHT_NSS_MAX; i++) {
8704                 if ((tx_mcs_mask[i] & txrate->mcs[i]) == txrate->mcs[i])
8705                         mcs[i] = txrate->mcs[i];
8706                 else
8707                         return false;
8708         }
8709
8710         return true;
8711 }
8712
8713 static const struct nla_policy nl80211_txattr_policy[NL80211_TXRATE_MAX + 1] = {
8714         [NL80211_TXRATE_LEGACY] = { .type = NLA_BINARY,
8715                                     .len = NL80211_MAX_SUPP_RATES },
8716         [NL80211_TXRATE_HT] = { .type = NLA_BINARY,
8717                                 .len = NL80211_MAX_SUPP_HT_RATES },
8718         [NL80211_TXRATE_VHT] = { .len = sizeof(struct nl80211_txrate_vht)},
8719         [NL80211_TXRATE_GI] = { .type = NLA_U8 },
8720 };
8721
8722 static int nl80211_set_tx_bitrate_mask(struct sk_buff *skb,
8723                                        struct genl_info *info)
8724 {
8725         struct nlattr *tb[NL80211_TXRATE_MAX + 1];
8726         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8727         struct cfg80211_bitrate_mask mask;
8728         int rem, i;
8729         struct net_device *dev = info->user_ptr[1];
8730         struct nlattr *tx_rates;
8731         struct ieee80211_supported_band *sband;
8732         u16 vht_tx_mcs_map;
8733
8734         if (!rdev->ops->set_bitrate_mask)
8735                 return -EOPNOTSUPP;
8736
8737         memset(&mask, 0, sizeof(mask));
8738         /* Default to all rates enabled */
8739         for (i = 0; i < NUM_NL80211_BANDS; i++) {
8740                 sband = rdev->wiphy.bands[i];
8741
8742                 if (!sband)
8743                         continue;
8744
8745                 mask.control[i].legacy = (1 << sband->n_bitrates) - 1;
8746                 memcpy(mask.control[i].ht_mcs,
8747                        sband->ht_cap.mcs.rx_mask,
8748                        sizeof(mask.control[i].ht_mcs));
8749
8750                 if (!sband->vht_cap.vht_supported)
8751                         continue;
8752
8753                 vht_tx_mcs_map = le16_to_cpu(sband->vht_cap.vht_mcs.tx_mcs_map);
8754                 vht_build_mcs_mask(vht_tx_mcs_map, mask.control[i].vht_mcs);
8755         }
8756
8757         /* if no rates are given set it back to the defaults */
8758         if (!info->attrs[NL80211_ATTR_TX_RATES])
8759                 goto out;
8760
8761         /*
8762          * The nested attribute uses enum nl80211_band as the index. This maps
8763          * directly to the enum nl80211_band values used in cfg80211.
8764          */
8765         BUILD_BUG_ON(NL80211_MAX_SUPP_HT_RATES > IEEE80211_HT_MCS_MASK_LEN * 8);
8766         nla_for_each_nested(tx_rates, info->attrs[NL80211_ATTR_TX_RATES], rem) {
8767                 enum nl80211_band band = nla_type(tx_rates);
8768                 int err;
8769
8770                 if (band < 0 || band >= NUM_NL80211_BANDS)
8771                         return -EINVAL;
8772                 sband = rdev->wiphy.bands[band];
8773                 if (sband == NULL)
8774                         return -EINVAL;
8775                 err = nla_parse(tb, NL80211_TXRATE_MAX, nla_data(tx_rates),
8776                                 nla_len(tx_rates), nl80211_txattr_policy);
8777                 if (err)
8778                         return err;
8779                 if (tb[NL80211_TXRATE_LEGACY]) {
8780                         mask.control[band].legacy = rateset_to_mask(
8781                                 sband,
8782                                 nla_data(tb[NL80211_TXRATE_LEGACY]),
8783                                 nla_len(tb[NL80211_TXRATE_LEGACY]));
8784                         if ((mask.control[band].legacy == 0) &&
8785                             nla_len(tb[NL80211_TXRATE_LEGACY]))
8786                                 return -EINVAL;
8787                 }
8788                 if (tb[NL80211_TXRATE_HT]) {
8789                         if (!ht_rateset_to_mask(
8790                                         sband,
8791                                         nla_data(tb[NL80211_TXRATE_HT]),
8792                                         nla_len(tb[NL80211_TXRATE_HT]),
8793                                         mask.control[band].ht_mcs))
8794                                 return -EINVAL;
8795                 }
8796                 if (tb[NL80211_TXRATE_VHT]) {
8797                         if (!vht_set_mcs_mask(
8798                                         sband,
8799                                         nla_data(tb[NL80211_TXRATE_VHT]),
8800                                         mask.control[band].vht_mcs))
8801                                 return -EINVAL;
8802                 }
8803                 if (tb[NL80211_TXRATE_GI]) {
8804                         mask.control[band].gi =
8805                                 nla_get_u8(tb[NL80211_TXRATE_GI]);
8806                         if (mask.control[band].gi > NL80211_TXRATE_FORCE_LGI)
8807                                 return -EINVAL;
8808                 }
8809
8810                 if (mask.control[band].legacy == 0) {
8811                         /* don't allow empty legacy rates if HT or VHT
8812                          * are not even supported.
8813                          */
8814                         if (!(rdev->wiphy.bands[band]->ht_cap.ht_supported ||
8815                               rdev->wiphy.bands[band]->vht_cap.vht_supported))
8816                                 return -EINVAL;
8817
8818                         for (i = 0; i < IEEE80211_HT_MCS_MASK_LEN; i++)
8819                                 if (mask.control[band].ht_mcs[i])
8820                                         goto out;
8821
8822                         for (i = 0; i < NL80211_VHT_NSS_MAX; i++)
8823                                 if (mask.control[band].vht_mcs[i])
8824                                         goto out;
8825
8826                         /* legacy and mcs rates may not be both empty */
8827                         return -EINVAL;
8828                 }
8829         }
8830
8831 out:
8832         return rdev_set_bitrate_mask(rdev, dev, NULL, &mask);
8833 }
8834
8835 static int nl80211_register_mgmt(struct sk_buff *skb, struct genl_info *info)
8836 {
8837         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8838         struct wireless_dev *wdev = info->user_ptr[1];
8839         u16 frame_type = IEEE80211_FTYPE_MGMT | IEEE80211_STYPE_ACTION;
8840
8841         if (!info->attrs[NL80211_ATTR_FRAME_MATCH])
8842                 return -EINVAL;
8843
8844         if (info->attrs[NL80211_ATTR_FRAME_TYPE])
8845                 frame_type = nla_get_u16(info->attrs[NL80211_ATTR_FRAME_TYPE]);
8846
8847         switch (wdev->iftype) {
8848         case NL80211_IFTYPE_STATION:
8849         case NL80211_IFTYPE_ADHOC:
8850         case NL80211_IFTYPE_P2P_CLIENT:
8851         case NL80211_IFTYPE_AP:
8852         case NL80211_IFTYPE_AP_VLAN:
8853         case NL80211_IFTYPE_MESH_POINT:
8854         case NL80211_IFTYPE_P2P_GO:
8855         case NL80211_IFTYPE_P2P_DEVICE:
8856                 break;
8857         default:
8858                 return -EOPNOTSUPP;
8859         }
8860
8861         /* not much point in registering if we can't reply */
8862         if (!rdev->ops->mgmt_tx)
8863                 return -EOPNOTSUPP;
8864
8865         return cfg80211_mlme_register_mgmt(wdev, info->snd_portid, frame_type,
8866                         nla_data(info->attrs[NL80211_ATTR_FRAME_MATCH]),
8867                         nla_len(info->attrs[NL80211_ATTR_FRAME_MATCH]));
8868 }
8869
8870 static int nl80211_tx_mgmt(struct sk_buff *skb, struct genl_info *info)
8871 {
8872         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8873         struct wireless_dev *wdev = info->user_ptr[1];
8874         struct cfg80211_chan_def chandef;
8875         int err;
8876         void *hdr = NULL;
8877         u64 cookie;
8878         struct sk_buff *msg = NULL;
8879         struct cfg80211_mgmt_tx_params params = {
8880                 .dont_wait_for_ack =
8881                         info->attrs[NL80211_ATTR_DONT_WAIT_FOR_ACK],
8882         };
8883
8884         if (!info->attrs[NL80211_ATTR_FRAME])
8885                 return -EINVAL;
8886
8887         if (!rdev->ops->mgmt_tx)
8888                 return -EOPNOTSUPP;
8889
8890         switch (wdev->iftype) {
8891         case NL80211_IFTYPE_P2P_DEVICE:
8892                 if (!info->attrs[NL80211_ATTR_WIPHY_FREQ])
8893                         return -EINVAL;
8894         case NL80211_IFTYPE_STATION:
8895         case NL80211_IFTYPE_ADHOC:
8896         case NL80211_IFTYPE_P2P_CLIENT:
8897         case NL80211_IFTYPE_AP:
8898         case NL80211_IFTYPE_AP_VLAN:
8899         case NL80211_IFTYPE_MESH_POINT:
8900         case NL80211_IFTYPE_P2P_GO:
8901                 break;
8902         default:
8903                 return -EOPNOTSUPP;
8904         }
8905
8906         if (info->attrs[NL80211_ATTR_DURATION]) {
8907                 if (!(rdev->wiphy.flags & WIPHY_FLAG_OFFCHAN_TX))
8908                         return -EINVAL;
8909                 params.wait = nla_get_u32(info->attrs[NL80211_ATTR_DURATION]);
8910
8911                 /*
8912                  * We should wait on the channel for at least a minimum amount
8913                  * of time (10ms) but no longer than the driver supports.
8914                  */
8915                 if (params.wait < NL80211_MIN_REMAIN_ON_CHANNEL_TIME ||
8916                     params.wait > rdev->wiphy.max_remain_on_channel_duration)
8917                         return -EINVAL;
8918         }
8919
8920         params.offchan = info->attrs[NL80211_ATTR_OFFCHANNEL_TX_OK];
8921
8922         if (params.offchan && !(rdev->wiphy.flags & WIPHY_FLAG_OFFCHAN_TX))
8923                 return -EINVAL;
8924
8925         params.no_cck = nla_get_flag(info->attrs[NL80211_ATTR_TX_NO_CCK_RATE]);
8926
8927         /* get the channel if any has been specified, otherwise pass NULL to
8928          * the driver. The latter will use the current one
8929          */
8930         chandef.chan = NULL;
8931         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
8932                 err = nl80211_parse_chandef(rdev, info, &chandef);
8933                 if (err)
8934                         return err;
8935         }
8936
8937         if (!chandef.chan && params.offchan)
8938                 return -EINVAL;
8939
8940         params.buf = nla_data(info->attrs[NL80211_ATTR_FRAME]);
8941         params.len = nla_len(info->attrs[NL80211_ATTR_FRAME]);
8942
8943         if (info->attrs[NL80211_ATTR_CSA_C_OFFSETS_TX]) {
8944                 int len = nla_len(info->attrs[NL80211_ATTR_CSA_C_OFFSETS_TX]);
8945                 int i;
8946
8947                 if (len % sizeof(u16))
8948                         return -EINVAL;
8949
8950                 params.n_csa_offsets = len / sizeof(u16);
8951                 params.csa_offsets =
8952                         nla_data(info->attrs[NL80211_ATTR_CSA_C_OFFSETS_TX]);
8953
8954                 /* check that all the offsets fit the frame */
8955                 for (i = 0; i < params.n_csa_offsets; i++) {
8956                         if (params.csa_offsets[i] >= params.len)
8957                                 return -EINVAL;
8958                 }
8959         }
8960
8961         if (!params.dont_wait_for_ack) {
8962                 msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8963                 if (!msg)
8964                         return -ENOMEM;
8965
8966                 hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
8967                                      NL80211_CMD_FRAME);
8968                 if (!hdr) {
8969                         err = -ENOBUFS;
8970                         goto free_msg;
8971                 }
8972         }
8973
8974         params.chan = chandef.chan;
8975         err = cfg80211_mlme_mgmt_tx(rdev, wdev, &params, &cookie);
8976         if (err)
8977                 goto free_msg;
8978
8979         if (msg) {
8980                 if (nla_put_u64_64bit(msg, NL80211_ATTR_COOKIE, cookie,
8981                                       NL80211_ATTR_PAD))
8982                         goto nla_put_failure;
8983
8984                 genlmsg_end(msg, hdr);
8985                 return genlmsg_reply(msg, info);
8986         }
8987
8988         return 0;
8989
8990  nla_put_failure:
8991         err = -ENOBUFS;
8992  free_msg:
8993         nlmsg_free(msg);
8994         return err;
8995 }
8996
8997 static int nl80211_tx_mgmt_cancel_wait(struct sk_buff *skb, struct genl_info *info)
8998 {
8999         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9000         struct wireless_dev *wdev = info->user_ptr[1];
9001         u64 cookie;
9002
9003         if (!info->attrs[NL80211_ATTR_COOKIE])
9004                 return -EINVAL;
9005
9006         if (!rdev->ops->mgmt_tx_cancel_wait)
9007                 return -EOPNOTSUPP;
9008
9009         switch (wdev->iftype) {
9010         case NL80211_IFTYPE_STATION:
9011         case NL80211_IFTYPE_ADHOC:
9012         case NL80211_IFTYPE_P2P_CLIENT:
9013         case NL80211_IFTYPE_AP:
9014         case NL80211_IFTYPE_AP_VLAN:
9015         case NL80211_IFTYPE_P2P_GO:
9016         case NL80211_IFTYPE_P2P_DEVICE:
9017                 break;
9018         default:
9019                 return -EOPNOTSUPP;
9020         }
9021
9022         cookie = nla_get_u64(info->attrs[NL80211_ATTR_COOKIE]);
9023
9024         return rdev_mgmt_tx_cancel_wait(rdev, wdev, cookie);
9025 }
9026
9027 static int nl80211_set_power_save(struct sk_buff *skb, struct genl_info *info)
9028 {
9029         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9030         struct wireless_dev *wdev;
9031         struct net_device *dev = info->user_ptr[1];
9032         u8 ps_state;
9033         bool state;
9034         int err;
9035
9036         if (!info->attrs[NL80211_ATTR_PS_STATE])
9037                 return -EINVAL;
9038
9039         ps_state = nla_get_u32(info->attrs[NL80211_ATTR_PS_STATE]);
9040
9041         if (ps_state != NL80211_PS_DISABLED && ps_state != NL80211_PS_ENABLED)
9042                 return -EINVAL;
9043
9044         wdev = dev->ieee80211_ptr;
9045
9046         if (!rdev->ops->set_power_mgmt)
9047                 return -EOPNOTSUPP;
9048
9049         state = (ps_state == NL80211_PS_ENABLED) ? true : false;
9050
9051         if (state == wdev->ps)
9052                 return 0;
9053
9054         err = rdev_set_power_mgmt(rdev, dev, state, wdev->ps_timeout);
9055         if (!err)
9056                 wdev->ps = state;
9057         return err;
9058 }
9059
9060 static int nl80211_get_power_save(struct sk_buff *skb, struct genl_info *info)
9061 {
9062         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9063         enum nl80211_ps_state ps_state;
9064         struct wireless_dev *wdev;
9065         struct net_device *dev = info->user_ptr[1];
9066         struct sk_buff *msg;
9067         void *hdr;
9068         int err;
9069
9070         wdev = dev->ieee80211_ptr;
9071
9072         if (!rdev->ops->set_power_mgmt)
9073                 return -EOPNOTSUPP;
9074
9075         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9076         if (!msg)
9077                 return -ENOMEM;
9078
9079         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
9080                              NL80211_CMD_GET_POWER_SAVE);
9081         if (!hdr) {
9082                 err = -ENOBUFS;
9083                 goto free_msg;
9084         }
9085
9086         if (wdev->ps)
9087                 ps_state = NL80211_PS_ENABLED;
9088         else
9089                 ps_state = NL80211_PS_DISABLED;
9090
9091         if (nla_put_u32(msg, NL80211_ATTR_PS_STATE, ps_state))
9092                 goto nla_put_failure;
9093
9094         genlmsg_end(msg, hdr);
9095         return genlmsg_reply(msg, info);
9096
9097  nla_put_failure:
9098         err = -ENOBUFS;
9099  free_msg:
9100         nlmsg_free(msg);
9101         return err;
9102 }
9103
9104 static const struct nla_policy
9105 nl80211_attr_cqm_policy[NL80211_ATTR_CQM_MAX + 1] = {
9106         [NL80211_ATTR_CQM_RSSI_THOLD] = { .type = NLA_U32 },
9107         [NL80211_ATTR_CQM_RSSI_HYST] = { .type = NLA_U32 },
9108         [NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT] = { .type = NLA_U32 },
9109         [NL80211_ATTR_CQM_TXE_RATE] = { .type = NLA_U32 },
9110         [NL80211_ATTR_CQM_TXE_PKTS] = { .type = NLA_U32 },
9111         [NL80211_ATTR_CQM_TXE_INTVL] = { .type = NLA_U32 },
9112 };
9113
9114 static int nl80211_set_cqm_txe(struct genl_info *info,
9115                                u32 rate, u32 pkts, u32 intvl)
9116 {
9117         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9118         struct net_device *dev = info->user_ptr[1];
9119         struct wireless_dev *wdev = dev->ieee80211_ptr;
9120
9121         if (rate > 100 || intvl > NL80211_CQM_TXE_MAX_INTVL)
9122                 return -EINVAL;
9123
9124         if (!rdev->ops->set_cqm_txe_config)
9125                 return -EOPNOTSUPP;
9126
9127         if (wdev->iftype != NL80211_IFTYPE_STATION &&
9128             wdev->iftype != NL80211_IFTYPE_P2P_CLIENT)
9129                 return -EOPNOTSUPP;
9130
9131         return rdev_set_cqm_txe_config(rdev, dev, rate, pkts, intvl);
9132 }
9133
9134 static int nl80211_set_cqm_rssi(struct genl_info *info,
9135                                 s32 threshold, u32 hysteresis)
9136 {
9137         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9138         struct net_device *dev = info->user_ptr[1];
9139         struct wireless_dev *wdev = dev->ieee80211_ptr;
9140
9141         if (threshold > 0)
9142                 return -EINVAL;
9143
9144         /* disabling - hysteresis should also be zero then */
9145         if (threshold == 0)
9146                 hysteresis = 0;
9147
9148         if (!rdev->ops->set_cqm_rssi_config)
9149                 return -EOPNOTSUPP;
9150
9151         if (wdev->iftype != NL80211_IFTYPE_STATION &&
9152             wdev->iftype != NL80211_IFTYPE_P2P_CLIENT)
9153                 return -EOPNOTSUPP;
9154
9155         return rdev_set_cqm_rssi_config(rdev, dev, threshold, hysteresis);
9156 }
9157
9158 static int nl80211_set_cqm(struct sk_buff *skb, struct genl_info *info)
9159 {
9160         struct nlattr *attrs[NL80211_ATTR_CQM_MAX + 1];
9161         struct nlattr *cqm;
9162         int err;
9163
9164         cqm = info->attrs[NL80211_ATTR_CQM];
9165         if (!cqm)
9166                 return -EINVAL;
9167
9168         err = nla_parse_nested(attrs, NL80211_ATTR_CQM_MAX, cqm,
9169                                nl80211_attr_cqm_policy);
9170         if (err)
9171                 return err;
9172
9173         if (attrs[NL80211_ATTR_CQM_RSSI_THOLD] &&
9174             attrs[NL80211_ATTR_CQM_RSSI_HYST]) {
9175                 s32 threshold = nla_get_s32(attrs[NL80211_ATTR_CQM_RSSI_THOLD]);
9176                 u32 hysteresis = nla_get_u32(attrs[NL80211_ATTR_CQM_RSSI_HYST]);
9177
9178                 return nl80211_set_cqm_rssi(info, threshold, hysteresis);
9179         }
9180
9181         if (attrs[NL80211_ATTR_CQM_TXE_RATE] &&
9182             attrs[NL80211_ATTR_CQM_TXE_PKTS] &&
9183             attrs[NL80211_ATTR_CQM_TXE_INTVL]) {
9184                 u32 rate = nla_get_u32(attrs[NL80211_ATTR_CQM_TXE_RATE]);
9185                 u32 pkts = nla_get_u32(attrs[NL80211_ATTR_CQM_TXE_PKTS]);
9186                 u32 intvl = nla_get_u32(attrs[NL80211_ATTR_CQM_TXE_INTVL]);
9187
9188                 return nl80211_set_cqm_txe(info, rate, pkts, intvl);
9189         }
9190
9191         return -EINVAL;
9192 }
9193
9194 static int nl80211_join_ocb(struct sk_buff *skb, struct genl_info *info)
9195 {
9196         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9197         struct net_device *dev = info->user_ptr[1];
9198         struct ocb_setup setup = {};
9199         int err;
9200
9201         err = nl80211_parse_chandef(rdev, info, &setup.chandef);
9202         if (err)
9203                 return err;
9204
9205         return cfg80211_join_ocb(rdev, dev, &setup);
9206 }
9207
9208 static int nl80211_leave_ocb(struct sk_buff *skb, struct genl_info *info)
9209 {
9210         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9211         struct net_device *dev = info->user_ptr[1];
9212
9213         return cfg80211_leave_ocb(rdev, dev);
9214 }
9215
9216 static int nl80211_join_mesh(struct sk_buff *skb, struct genl_info *info)
9217 {
9218         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9219         struct net_device *dev = info->user_ptr[1];
9220         struct mesh_config cfg;
9221         struct mesh_setup setup;
9222         int err;
9223
9224         /* start with default */
9225         memcpy(&cfg, &default_mesh_config, sizeof(cfg));
9226         memcpy(&setup, &default_mesh_setup, sizeof(setup));
9227
9228         if (info->attrs[NL80211_ATTR_MESH_CONFIG]) {
9229                 /* and parse parameters if given */
9230                 err = nl80211_parse_mesh_config(info, &cfg, NULL);
9231                 if (err)
9232                         return err;
9233         }
9234
9235         if (!info->attrs[NL80211_ATTR_MESH_ID] ||
9236             !nla_len(info->attrs[NL80211_ATTR_MESH_ID]))
9237                 return -EINVAL;
9238
9239         setup.mesh_id = nla_data(info->attrs[NL80211_ATTR_MESH_ID]);
9240         setup.mesh_id_len = nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
9241
9242         if (info->attrs[NL80211_ATTR_MCAST_RATE] &&
9243             !nl80211_parse_mcast_rate(rdev, setup.mcast_rate,
9244                             nla_get_u32(info->attrs[NL80211_ATTR_MCAST_RATE])))
9245                         return -EINVAL;
9246
9247         if (info->attrs[NL80211_ATTR_BEACON_INTERVAL]) {
9248                 setup.beacon_interval =
9249                         nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
9250                 if (setup.beacon_interval < 10 ||
9251                     setup.beacon_interval > 10000)
9252                         return -EINVAL;
9253         }
9254
9255         if (info->attrs[NL80211_ATTR_DTIM_PERIOD]) {
9256                 setup.dtim_period =
9257                         nla_get_u32(info->attrs[NL80211_ATTR_DTIM_PERIOD]);
9258                 if (setup.dtim_period < 1 || setup.dtim_period > 100)
9259                         return -EINVAL;
9260         }
9261
9262         if (info->attrs[NL80211_ATTR_MESH_SETUP]) {
9263                 /* parse additional setup parameters if given */
9264                 err = nl80211_parse_mesh_setup(info, &setup);
9265                 if (err)
9266                         return err;
9267         }
9268
9269         if (setup.user_mpm)
9270                 cfg.auto_open_plinks = false;
9271
9272         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
9273                 err = nl80211_parse_chandef(rdev, info, &setup.chandef);
9274                 if (err)
9275                         return err;
9276         } else {
9277                 /* cfg80211_join_mesh() will sort it out */
9278                 setup.chandef.chan = NULL;
9279         }
9280
9281         if (info->attrs[NL80211_ATTR_BSS_BASIC_RATES]) {
9282                 u8 *rates = nla_data(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
9283                 int n_rates =
9284                         nla_len(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
9285                 struct ieee80211_supported_band *sband;
9286
9287                 if (!setup.chandef.chan)
9288                         return -EINVAL;
9289
9290                 sband = rdev->wiphy.bands[setup.chandef.chan->band];
9291
9292                 err = ieee80211_get_ratemask(sband, rates, n_rates,
9293                                              &setup.basic_rates);
9294                 if (err)
9295                         return err;
9296         }
9297
9298         return cfg80211_join_mesh(rdev, dev, &setup, &cfg);
9299 }
9300
9301 static int nl80211_leave_mesh(struct sk_buff *skb, struct genl_info *info)
9302 {
9303         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9304         struct net_device *dev = info->user_ptr[1];
9305
9306         return cfg80211_leave_mesh(rdev, dev);
9307 }
9308
9309 #ifdef CONFIG_PM
9310 static int nl80211_send_wowlan_patterns(struct sk_buff *msg,
9311                                         struct cfg80211_registered_device *rdev)
9312 {
9313         struct cfg80211_wowlan *wowlan = rdev->wiphy.wowlan_config;
9314         struct nlattr *nl_pats, *nl_pat;
9315         int i, pat_len;
9316
9317         if (!wowlan->n_patterns)
9318                 return 0;
9319
9320         nl_pats = nla_nest_start(msg, NL80211_WOWLAN_TRIG_PKT_PATTERN);
9321         if (!nl_pats)
9322                 return -ENOBUFS;
9323
9324         for (i = 0; i < wowlan->n_patterns; i++) {
9325                 nl_pat = nla_nest_start(msg, i + 1);
9326                 if (!nl_pat)
9327                         return -ENOBUFS;
9328                 pat_len = wowlan->patterns[i].pattern_len;
9329                 if (nla_put(msg, NL80211_PKTPAT_MASK, DIV_ROUND_UP(pat_len, 8),
9330                             wowlan->patterns[i].mask) ||
9331                     nla_put(msg, NL80211_PKTPAT_PATTERN, pat_len,
9332                             wowlan->patterns[i].pattern) ||
9333                     nla_put_u32(msg, NL80211_PKTPAT_OFFSET,
9334                                 wowlan->patterns[i].pkt_offset))
9335                         return -ENOBUFS;
9336                 nla_nest_end(msg, nl_pat);
9337         }
9338         nla_nest_end(msg, nl_pats);
9339
9340         return 0;
9341 }
9342
9343 static int nl80211_send_wowlan_tcp(struct sk_buff *msg,
9344                                    struct cfg80211_wowlan_tcp *tcp)
9345 {
9346         struct nlattr *nl_tcp;
9347
9348         if (!tcp)
9349                 return 0;
9350
9351         nl_tcp = nla_nest_start(msg, NL80211_WOWLAN_TRIG_TCP_CONNECTION);
9352         if (!nl_tcp)
9353                 return -ENOBUFS;
9354
9355         if (nla_put_in_addr(msg, NL80211_WOWLAN_TCP_SRC_IPV4, tcp->src) ||
9356             nla_put_in_addr(msg, NL80211_WOWLAN_TCP_DST_IPV4, tcp->dst) ||
9357             nla_put(msg, NL80211_WOWLAN_TCP_DST_MAC, ETH_ALEN, tcp->dst_mac) ||
9358             nla_put_u16(msg, NL80211_WOWLAN_TCP_SRC_PORT, tcp->src_port) ||
9359             nla_put_u16(msg, NL80211_WOWLAN_TCP_DST_PORT, tcp->dst_port) ||
9360             nla_put(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD,
9361                     tcp->payload_len, tcp->payload) ||
9362             nla_put_u32(msg, NL80211_WOWLAN_TCP_DATA_INTERVAL,
9363                         tcp->data_interval) ||
9364             nla_put(msg, NL80211_WOWLAN_TCP_WAKE_PAYLOAD,
9365                     tcp->wake_len, tcp->wake_data) ||
9366             nla_put(msg, NL80211_WOWLAN_TCP_WAKE_MASK,
9367                     DIV_ROUND_UP(tcp->wake_len, 8), tcp->wake_mask))
9368                 return -ENOBUFS;
9369
9370         if (tcp->payload_seq.len &&
9371             nla_put(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ,
9372                     sizeof(tcp->payload_seq), &tcp->payload_seq))
9373                 return -ENOBUFS;
9374
9375         if (tcp->payload_tok.len &&
9376             nla_put(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN,
9377                     sizeof(tcp->payload_tok) + tcp->tokens_size,
9378                     &tcp->payload_tok))
9379                 return -ENOBUFS;
9380
9381         nla_nest_end(msg, nl_tcp);
9382
9383         return 0;
9384 }
9385
9386 static int nl80211_send_wowlan_nd(struct sk_buff *msg,
9387                                   struct cfg80211_sched_scan_request *req)
9388 {
9389         struct nlattr *nd, *freqs, *matches, *match, *scan_plans, *scan_plan;
9390         int i;
9391
9392         if (!req)
9393                 return 0;
9394
9395         nd = nla_nest_start(msg, NL80211_WOWLAN_TRIG_NET_DETECT);
9396         if (!nd)
9397                 return -ENOBUFS;
9398
9399         if (req->n_scan_plans == 1 &&
9400             nla_put_u32(msg, NL80211_ATTR_SCHED_SCAN_INTERVAL,
9401                         req->scan_plans[0].interval * 1000))
9402                 return -ENOBUFS;
9403
9404         if (nla_put_u32(msg, NL80211_ATTR_SCHED_SCAN_DELAY, req->delay))
9405                 return -ENOBUFS;
9406
9407         freqs = nla_nest_start(msg, NL80211_ATTR_SCAN_FREQUENCIES);
9408         if (!freqs)
9409                 return -ENOBUFS;
9410
9411         for (i = 0; i < req->n_channels; i++)
9412                 nla_put_u32(msg, i, req->channels[i]->center_freq);
9413
9414         nla_nest_end(msg, freqs);
9415
9416         if (req->n_match_sets) {
9417                 matches = nla_nest_start(msg, NL80211_ATTR_SCHED_SCAN_MATCH);
9418                 for (i = 0; i < req->n_match_sets; i++) {
9419                         match = nla_nest_start(msg, i);
9420                         nla_put(msg, NL80211_SCHED_SCAN_MATCH_ATTR_SSID,
9421                                 req->match_sets[i].ssid.ssid_len,
9422                                 req->match_sets[i].ssid.ssid);
9423                         nla_nest_end(msg, match);
9424                 }
9425                 nla_nest_end(msg, matches);
9426         }
9427
9428         scan_plans = nla_nest_start(msg, NL80211_ATTR_SCHED_SCAN_PLANS);
9429         if (!scan_plans)
9430                 return -ENOBUFS;
9431
9432         for (i = 0; i < req->n_scan_plans; i++) {
9433                 scan_plan = nla_nest_start(msg, i + 1);
9434                 if (!scan_plan ||
9435                     nla_put_u32(msg, NL80211_SCHED_SCAN_PLAN_INTERVAL,
9436                                 req->scan_plans[i].interval) ||
9437                     (req->scan_plans[i].iterations &&
9438                      nla_put_u32(msg, NL80211_SCHED_SCAN_PLAN_ITERATIONS,
9439                                  req->scan_plans[i].iterations)))
9440                         return -ENOBUFS;
9441                 nla_nest_end(msg, scan_plan);
9442         }
9443         nla_nest_end(msg, scan_plans);
9444
9445         nla_nest_end(msg, nd);
9446
9447         return 0;
9448 }
9449
9450 static int nl80211_get_wowlan(struct sk_buff *skb, struct genl_info *info)
9451 {
9452         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9453         struct sk_buff *msg;
9454         void *hdr;
9455         u32 size = NLMSG_DEFAULT_SIZE;
9456
9457         if (!rdev->wiphy.wowlan)
9458                 return -EOPNOTSUPP;
9459
9460         if (rdev->wiphy.wowlan_config && rdev->wiphy.wowlan_config->tcp) {
9461                 /* adjust size to have room for all the data */
9462                 size += rdev->wiphy.wowlan_config->tcp->tokens_size +
9463                         rdev->wiphy.wowlan_config->tcp->payload_len +
9464                         rdev->wiphy.wowlan_config->tcp->wake_len +
9465                         rdev->wiphy.wowlan_config->tcp->wake_len / 8;
9466         }
9467
9468         msg = nlmsg_new(size, GFP_KERNEL);
9469         if (!msg)
9470                 return -ENOMEM;
9471
9472         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
9473                              NL80211_CMD_GET_WOWLAN);
9474         if (!hdr)
9475                 goto nla_put_failure;
9476
9477         if (rdev->wiphy.wowlan_config) {
9478                 struct nlattr *nl_wowlan;
9479
9480                 nl_wowlan = nla_nest_start(msg, NL80211_ATTR_WOWLAN_TRIGGERS);
9481                 if (!nl_wowlan)
9482                         goto nla_put_failure;
9483
9484                 if ((rdev->wiphy.wowlan_config->any &&
9485                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_ANY)) ||
9486                     (rdev->wiphy.wowlan_config->disconnect &&
9487                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_DISCONNECT)) ||
9488                     (rdev->wiphy.wowlan_config->magic_pkt &&
9489                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_MAGIC_PKT)) ||
9490                     (rdev->wiphy.wowlan_config->gtk_rekey_failure &&
9491                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE)) ||
9492                     (rdev->wiphy.wowlan_config->eap_identity_req &&
9493                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST)) ||
9494                     (rdev->wiphy.wowlan_config->four_way_handshake &&
9495                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE)) ||
9496                     (rdev->wiphy.wowlan_config->rfkill_release &&
9497                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_RFKILL_RELEASE)))
9498                         goto nla_put_failure;
9499
9500                 if (nl80211_send_wowlan_patterns(msg, rdev))
9501                         goto nla_put_failure;
9502
9503                 if (nl80211_send_wowlan_tcp(msg,
9504                                             rdev->wiphy.wowlan_config->tcp))
9505                         goto nla_put_failure;
9506
9507                 if (nl80211_send_wowlan_nd(
9508                             msg,
9509                             rdev->wiphy.wowlan_config->nd_config))
9510                         goto nla_put_failure;
9511
9512                 nla_nest_end(msg, nl_wowlan);
9513         }
9514
9515         genlmsg_end(msg, hdr);
9516         return genlmsg_reply(msg, info);
9517
9518 nla_put_failure:
9519         nlmsg_free(msg);
9520         return -ENOBUFS;
9521 }
9522
9523 static int nl80211_parse_wowlan_tcp(struct cfg80211_registered_device *rdev,
9524                                     struct nlattr *attr,
9525                                     struct cfg80211_wowlan *trig)
9526 {
9527         struct nlattr *tb[NUM_NL80211_WOWLAN_TCP];
9528         struct cfg80211_wowlan_tcp *cfg;
9529         struct nl80211_wowlan_tcp_data_token *tok = NULL;
9530         struct nl80211_wowlan_tcp_data_seq *seq = NULL;
9531         u32 size;
9532         u32 data_size, wake_size, tokens_size = 0, wake_mask_size;
9533         int err, port;
9534
9535         if (!rdev->wiphy.wowlan->tcp)
9536                 return -EINVAL;
9537
9538         err = nla_parse(tb, MAX_NL80211_WOWLAN_TCP,
9539                         nla_data(attr), nla_len(attr),
9540                         nl80211_wowlan_tcp_policy);
9541         if (err)
9542                 return err;
9543
9544         if (!tb[NL80211_WOWLAN_TCP_SRC_IPV4] ||
9545             !tb[NL80211_WOWLAN_TCP_DST_IPV4] ||
9546             !tb[NL80211_WOWLAN_TCP_DST_MAC] ||
9547             !tb[NL80211_WOWLAN_TCP_DST_PORT] ||
9548             !tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD] ||
9549             !tb[NL80211_WOWLAN_TCP_DATA_INTERVAL] ||
9550             !tb[NL80211_WOWLAN_TCP_WAKE_PAYLOAD] ||
9551             !tb[NL80211_WOWLAN_TCP_WAKE_MASK])
9552                 return -EINVAL;
9553
9554         data_size = nla_len(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD]);
9555         if (data_size > rdev->wiphy.wowlan->tcp->data_payload_max)
9556                 return -EINVAL;
9557
9558         if (nla_get_u32(tb[NL80211_WOWLAN_TCP_DATA_INTERVAL]) >
9559                         rdev->wiphy.wowlan->tcp->data_interval_max ||
9560             nla_get_u32(tb[NL80211_WOWLAN_TCP_DATA_INTERVAL]) == 0)
9561                 return -EINVAL;
9562
9563         wake_size = nla_len(tb[NL80211_WOWLAN_TCP_WAKE_PAYLOAD]);
9564         if (wake_size > rdev->wiphy.wowlan->tcp->wake_payload_max)
9565                 return -EINVAL;
9566
9567         wake_mask_size = nla_len(tb[NL80211_WOWLAN_TCP_WAKE_MASK]);
9568         if (wake_mask_size != DIV_ROUND_UP(wake_size, 8))
9569                 return -EINVAL;
9570
9571         if (tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN]) {
9572                 u32 tokln = nla_len(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN]);
9573
9574                 tok = nla_data(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN]);
9575                 tokens_size = tokln - sizeof(*tok);
9576
9577                 if (!tok->len || tokens_size % tok->len)
9578                         return -EINVAL;
9579                 if (!rdev->wiphy.wowlan->tcp->tok)
9580                         return -EINVAL;
9581                 if (tok->len > rdev->wiphy.wowlan->tcp->tok->max_len)
9582                         return -EINVAL;
9583                 if (tok->len < rdev->wiphy.wowlan->tcp->tok->min_len)
9584                         return -EINVAL;
9585                 if (tokens_size > rdev->wiphy.wowlan->tcp->tok->bufsize)
9586                         return -EINVAL;
9587                 if (tok->offset + tok->len > data_size)
9588                         return -EINVAL;
9589         }
9590
9591         if (tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ]) {
9592                 seq = nla_data(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ]);
9593                 if (!rdev->wiphy.wowlan->tcp->seq)
9594                         return -EINVAL;
9595                 if (seq->len == 0 || seq->len > 4)
9596                         return -EINVAL;
9597                 if (seq->len + seq->offset > data_size)
9598                         return -EINVAL;
9599         }
9600
9601         size = sizeof(*cfg);
9602         size += data_size;
9603         size += wake_size + wake_mask_size;
9604         size += tokens_size;
9605
9606         cfg = kzalloc(size, GFP_KERNEL);
9607         if (!cfg)
9608                 return -ENOMEM;
9609         cfg->src = nla_get_in_addr(tb[NL80211_WOWLAN_TCP_SRC_IPV4]);
9610         cfg->dst = nla_get_in_addr(tb[NL80211_WOWLAN_TCP_DST_IPV4]);
9611         memcpy(cfg->dst_mac, nla_data(tb[NL80211_WOWLAN_TCP_DST_MAC]),
9612                ETH_ALEN);
9613         if (tb[NL80211_WOWLAN_TCP_SRC_PORT])
9614                 port = nla_get_u16(tb[NL80211_WOWLAN_TCP_SRC_PORT]);
9615         else
9616                 port = 0;
9617 #ifdef CONFIG_INET
9618         /* allocate a socket and port for it and use it */
9619         err = __sock_create(wiphy_net(&rdev->wiphy), PF_INET, SOCK_STREAM,
9620                             IPPROTO_TCP, &cfg->sock, 1);
9621         if (err) {
9622                 kfree(cfg);
9623                 return err;
9624         }
9625         if (inet_csk_get_port(cfg->sock->sk, port)) {
9626                 sock_release(cfg->sock);
9627                 kfree(cfg);
9628                 return -EADDRINUSE;
9629         }
9630         cfg->src_port = inet_sk(cfg->sock->sk)->inet_num;
9631 #else
9632         if (!port) {
9633                 kfree(cfg);
9634                 return -EINVAL;
9635         }
9636         cfg->src_port = port;
9637 #endif
9638
9639         cfg->dst_port = nla_get_u16(tb[NL80211_WOWLAN_TCP_DST_PORT]);
9640         cfg->payload_len = data_size;
9641         cfg->payload = (u8 *)cfg + sizeof(*cfg) + tokens_size;
9642         memcpy((void *)cfg->payload,
9643                nla_data(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD]),
9644                data_size);
9645         if (seq)
9646                 cfg->payload_seq = *seq;
9647         cfg->data_interval = nla_get_u32(tb[NL80211_WOWLAN_TCP_DATA_INTERVAL]);
9648         cfg->wake_len = wake_size;
9649         cfg->wake_data = (u8 *)cfg + sizeof(*cfg) + tokens_size + data_size;
9650         memcpy((void *)cfg->wake_data,
9651                nla_data(tb[NL80211_WOWLAN_TCP_WAKE_PAYLOAD]),
9652                wake_size);
9653         cfg->wake_mask = (u8 *)cfg + sizeof(*cfg) + tokens_size +
9654                          data_size + wake_size;
9655         memcpy((void *)cfg->wake_mask,
9656                nla_data(tb[NL80211_WOWLAN_TCP_WAKE_MASK]),
9657                wake_mask_size);
9658         if (tok) {
9659                 cfg->tokens_size = tokens_size;
9660                 memcpy(&cfg->payload_tok, tok, sizeof(*tok) + tokens_size);
9661         }
9662
9663         trig->tcp = cfg;
9664
9665         return 0;
9666 }
9667
9668 static int nl80211_parse_wowlan_nd(struct cfg80211_registered_device *rdev,
9669                                    const struct wiphy_wowlan_support *wowlan,
9670                                    struct nlattr *attr,
9671                                    struct cfg80211_wowlan *trig)
9672 {
9673         struct nlattr **tb;
9674         int err;
9675
9676         tb = kzalloc(NUM_NL80211_ATTR * sizeof(*tb), GFP_KERNEL);
9677         if (!tb)
9678                 return -ENOMEM;
9679
9680         if (!(wowlan->flags & WIPHY_WOWLAN_NET_DETECT)) {
9681                 err = -EOPNOTSUPP;
9682                 goto out;
9683         }
9684
9685         err = nla_parse(tb, NL80211_ATTR_MAX,
9686                         nla_data(attr), nla_len(attr),
9687                         nl80211_policy);
9688         if (err)
9689                 goto out;
9690
9691         trig->nd_config = nl80211_parse_sched_scan(&rdev->wiphy, NULL, tb);
9692         err = PTR_ERR_OR_ZERO(trig->nd_config);
9693         if (err)
9694                 trig->nd_config = NULL;
9695
9696 out:
9697         kfree(tb);
9698         return err;
9699 }
9700
9701 static int nl80211_set_wowlan(struct sk_buff *skb, struct genl_info *info)
9702 {
9703         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9704         struct nlattr *tb[NUM_NL80211_WOWLAN_TRIG];
9705         struct cfg80211_wowlan new_triggers = {};
9706         struct cfg80211_wowlan *ntrig;
9707         const struct wiphy_wowlan_support *wowlan = rdev->wiphy.wowlan;
9708         int err, i;
9709         bool prev_enabled = rdev->wiphy.wowlan_config;
9710         bool regular = false;
9711
9712         if (!wowlan)
9713                 return -EOPNOTSUPP;
9714
9715         if (!info->attrs[NL80211_ATTR_WOWLAN_TRIGGERS]) {
9716                 cfg80211_rdev_free_wowlan(rdev);
9717                 rdev->wiphy.wowlan_config = NULL;
9718                 goto set_wakeup;
9719         }
9720
9721         err = nla_parse(tb, MAX_NL80211_WOWLAN_TRIG,
9722                         nla_data(info->attrs[NL80211_ATTR_WOWLAN_TRIGGERS]),
9723                         nla_len(info->attrs[NL80211_ATTR_WOWLAN_TRIGGERS]),
9724                         nl80211_wowlan_policy);
9725         if (err)
9726                 return err;
9727
9728         if (tb[NL80211_WOWLAN_TRIG_ANY]) {
9729                 if (!(wowlan->flags & WIPHY_WOWLAN_ANY))
9730                         return -EINVAL;
9731                 new_triggers.any = true;
9732         }
9733
9734         if (tb[NL80211_WOWLAN_TRIG_DISCONNECT]) {
9735                 if (!(wowlan->flags & WIPHY_WOWLAN_DISCONNECT))
9736                         return -EINVAL;
9737                 new_triggers.disconnect = true;
9738                 regular = true;
9739         }
9740
9741         if (tb[NL80211_WOWLAN_TRIG_MAGIC_PKT]) {
9742                 if (!(wowlan->flags & WIPHY_WOWLAN_MAGIC_PKT))
9743                         return -EINVAL;
9744                 new_triggers.magic_pkt = true;
9745                 regular = true;
9746         }
9747
9748         if (tb[NL80211_WOWLAN_TRIG_GTK_REKEY_SUPPORTED])
9749                 return -EINVAL;
9750
9751         if (tb[NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE]) {
9752                 if (!(wowlan->flags & WIPHY_WOWLAN_GTK_REKEY_FAILURE))
9753                         return -EINVAL;
9754                 new_triggers.gtk_rekey_failure = true;
9755                 regular = true;
9756         }
9757
9758         if (tb[NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST]) {
9759                 if (!(wowlan->flags & WIPHY_WOWLAN_EAP_IDENTITY_REQ))
9760                         return -EINVAL;
9761                 new_triggers.eap_identity_req = true;
9762                 regular = true;
9763         }
9764
9765         if (tb[NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE]) {
9766                 if (!(wowlan->flags & WIPHY_WOWLAN_4WAY_HANDSHAKE))
9767                         return -EINVAL;
9768                 new_triggers.four_way_handshake = true;
9769                 regular = true;
9770         }
9771
9772         if (tb[NL80211_WOWLAN_TRIG_RFKILL_RELEASE]) {
9773                 if (!(wowlan->flags & WIPHY_WOWLAN_RFKILL_RELEASE))
9774                         return -EINVAL;
9775                 new_triggers.rfkill_release = true;
9776                 regular = true;
9777         }
9778
9779         if (tb[NL80211_WOWLAN_TRIG_PKT_PATTERN]) {
9780                 struct nlattr *pat;
9781                 int n_patterns = 0;
9782                 int rem, pat_len, mask_len, pkt_offset;
9783                 struct nlattr *pat_tb[NUM_NL80211_PKTPAT];
9784
9785                 regular = true;
9786
9787                 nla_for_each_nested(pat, tb[NL80211_WOWLAN_TRIG_PKT_PATTERN],
9788                                     rem)
9789                         n_patterns++;
9790                 if (n_patterns > wowlan->n_patterns)
9791                         return -EINVAL;
9792
9793                 new_triggers.patterns = kcalloc(n_patterns,
9794                                                 sizeof(new_triggers.patterns[0]),
9795                                                 GFP_KERNEL);
9796                 if (!new_triggers.patterns)
9797                         return -ENOMEM;
9798
9799                 new_triggers.n_patterns = n_patterns;
9800                 i = 0;
9801
9802                 nla_for_each_nested(pat, tb[NL80211_WOWLAN_TRIG_PKT_PATTERN],
9803                                     rem) {
9804                         u8 *mask_pat;
9805
9806                         nla_parse(pat_tb, MAX_NL80211_PKTPAT, nla_data(pat),
9807                                   nla_len(pat), NULL);
9808                         err = -EINVAL;
9809                         if (!pat_tb[NL80211_PKTPAT_MASK] ||
9810                             !pat_tb[NL80211_PKTPAT_PATTERN])
9811                                 goto error;
9812                         pat_len = nla_len(pat_tb[NL80211_PKTPAT_PATTERN]);
9813                         mask_len = DIV_ROUND_UP(pat_len, 8);
9814                         if (nla_len(pat_tb[NL80211_PKTPAT_MASK]) != mask_len)
9815                                 goto error;
9816                         if (pat_len > wowlan->pattern_max_len ||
9817                             pat_len < wowlan->pattern_min_len)
9818                                 goto error;
9819
9820                         if (!pat_tb[NL80211_PKTPAT_OFFSET])
9821                                 pkt_offset = 0;
9822                         else
9823                                 pkt_offset = nla_get_u32(
9824                                         pat_tb[NL80211_PKTPAT_OFFSET]);
9825                         if (pkt_offset > wowlan->max_pkt_offset)
9826                                 goto error;
9827                         new_triggers.patterns[i].pkt_offset = pkt_offset;
9828
9829                         mask_pat = kmalloc(mask_len + pat_len, GFP_KERNEL);
9830                         if (!mask_pat) {
9831                                 err = -ENOMEM;
9832                                 goto error;
9833                         }
9834                         new_triggers.patterns[i].mask = mask_pat;
9835                         memcpy(mask_pat, nla_data(pat_tb[NL80211_PKTPAT_MASK]),
9836                                mask_len);
9837                         mask_pat += mask_len;
9838                         new_triggers.patterns[i].pattern = mask_pat;
9839                         new_triggers.patterns[i].pattern_len = pat_len;
9840                         memcpy(mask_pat,
9841                                nla_data(pat_tb[NL80211_PKTPAT_PATTERN]),
9842                                pat_len);
9843                         i++;
9844                 }
9845         }
9846
9847         if (tb[NL80211_WOWLAN_TRIG_TCP_CONNECTION]) {
9848                 regular = true;
9849                 err = nl80211_parse_wowlan_tcp(
9850                         rdev, tb[NL80211_WOWLAN_TRIG_TCP_CONNECTION],
9851                         &new_triggers);
9852                 if (err)
9853                         goto error;
9854         }
9855
9856         if (tb[NL80211_WOWLAN_TRIG_NET_DETECT]) {
9857                 regular = true;
9858                 err = nl80211_parse_wowlan_nd(
9859                         rdev, wowlan, tb[NL80211_WOWLAN_TRIG_NET_DETECT],
9860                         &new_triggers);
9861                 if (err)
9862                         goto error;
9863         }
9864
9865         /* The 'any' trigger means the device continues operating more or less
9866          * as in its normal operation mode and wakes up the host on most of the
9867          * normal interrupts (like packet RX, ...)
9868          * It therefore makes little sense to combine with the more constrained
9869          * wakeup trigger modes.
9870          */
9871         if (new_triggers.any && regular) {
9872                 err = -EINVAL;
9873                 goto error;
9874         }
9875
9876         ntrig = kmemdup(&new_triggers, sizeof(new_triggers), GFP_KERNEL);
9877         if (!ntrig) {
9878                 err = -ENOMEM;
9879                 goto error;
9880         }
9881         cfg80211_rdev_free_wowlan(rdev);
9882         rdev->wiphy.wowlan_config = ntrig;
9883
9884  set_wakeup:
9885         if (rdev->ops->set_wakeup &&
9886             prev_enabled != !!rdev->wiphy.wowlan_config)
9887                 rdev_set_wakeup(rdev, rdev->wiphy.wowlan_config);
9888
9889         return 0;
9890  error:
9891         for (i = 0; i < new_triggers.n_patterns; i++)
9892                 kfree(new_triggers.patterns[i].mask);
9893         kfree(new_triggers.patterns);
9894         if (new_triggers.tcp && new_triggers.tcp->sock)
9895                 sock_release(new_triggers.tcp->sock);
9896         kfree(new_triggers.tcp);
9897         kfree(new_triggers.nd_config);
9898         return err;
9899 }
9900 #endif
9901
9902 static int nl80211_send_coalesce_rules(struct sk_buff *msg,
9903                                        struct cfg80211_registered_device *rdev)
9904 {
9905         struct nlattr *nl_pats, *nl_pat, *nl_rule, *nl_rules;
9906         int i, j, pat_len;
9907         struct cfg80211_coalesce_rules *rule;
9908
9909         if (!rdev->coalesce->n_rules)
9910                 return 0;
9911
9912         nl_rules = nla_nest_start(msg, NL80211_ATTR_COALESCE_RULE);
9913         if (!nl_rules)
9914                 return -ENOBUFS;
9915
9916         for (i = 0; i < rdev->coalesce->n_rules; i++) {
9917                 nl_rule = nla_nest_start(msg, i + 1);
9918                 if (!nl_rule)
9919                         return -ENOBUFS;
9920
9921                 rule = &rdev->coalesce->rules[i];
9922                 if (nla_put_u32(msg, NL80211_ATTR_COALESCE_RULE_DELAY,
9923                                 rule->delay))
9924                         return -ENOBUFS;
9925
9926                 if (nla_put_u32(msg, NL80211_ATTR_COALESCE_RULE_CONDITION,
9927                                 rule->condition))
9928                         return -ENOBUFS;
9929
9930                 nl_pats = nla_nest_start(msg,
9931                                 NL80211_ATTR_COALESCE_RULE_PKT_PATTERN);
9932                 if (!nl_pats)
9933                         return -ENOBUFS;
9934
9935                 for (j = 0; j < rule->n_patterns; j++) {
9936                         nl_pat = nla_nest_start(msg, j + 1);
9937                         if (!nl_pat)
9938                                 return -ENOBUFS;
9939                         pat_len = rule->patterns[j].pattern_len;
9940                         if (nla_put(msg, NL80211_PKTPAT_MASK,
9941                                     DIV_ROUND_UP(pat_len, 8),
9942                                     rule->patterns[j].mask) ||
9943                             nla_put(msg, NL80211_PKTPAT_PATTERN, pat_len,
9944                                     rule->patterns[j].pattern) ||
9945                             nla_put_u32(msg, NL80211_PKTPAT_OFFSET,
9946                                         rule->patterns[j].pkt_offset))
9947                                 return -ENOBUFS;
9948                         nla_nest_end(msg, nl_pat);
9949                 }
9950                 nla_nest_end(msg, nl_pats);
9951                 nla_nest_end(msg, nl_rule);
9952         }
9953         nla_nest_end(msg, nl_rules);
9954
9955         return 0;
9956 }
9957
9958 static int nl80211_get_coalesce(struct sk_buff *skb, struct genl_info *info)
9959 {
9960         struct cfg80211_registered_device *rdev = info->user_ptr[0];
9961         struct sk_buff *msg;
9962         void *hdr;
9963
9964         if (!rdev->wiphy.coalesce)
9965                 return -EOPNOTSUPP;
9966
9967         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9968         if (!msg)
9969                 return -ENOMEM;
9970
9971         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
9972                              NL80211_CMD_GET_COALESCE);
9973         if (!hdr)
9974                 goto nla_put_failure;
9975
9976         if (rdev->coalesce && nl80211_send_coalesce_rules(msg, rdev))
9977                 goto nla_put_failure;
9978
9979         genlmsg_end(msg, hdr);
9980         return genlmsg_reply(msg, info);
9981
9982 nla_put_failure:
9983         nlmsg_free(msg);
9984         return -ENOBUFS;
9985 }
9986
9987 void cfg80211_rdev_free_coalesce(struct cfg80211_registered_device *rdev)
9988 {
9989         struct cfg80211_coalesce *coalesce = rdev->coalesce;
9990         int i, j;
9991         struct cfg80211_coalesce_rules *rule;
9992
9993         if (!coalesce)
9994                 return;
9995
9996         for (i = 0; i < coalesce->n_rules; i++) {
9997                 rule = &coalesce->rules[i];
9998                 for (j = 0; j < rule->n_patterns; j++)
9999                         kfree(rule->patterns[j].mask);
10000                 kfree(rule->patterns);
10001         }
10002         kfree(coalesce->rules);
10003         kfree(coalesce);
10004         rdev->coalesce = NULL;
10005 }
10006
10007 static int nl80211_parse_coalesce_rule(struct cfg80211_registered_device *rdev,
10008                                        struct nlattr *rule,
10009                                        struct cfg80211_coalesce_rules *new_rule)
10010 {
10011         int err, i;
10012         const struct wiphy_coalesce_support *coalesce = rdev->wiphy.coalesce;
10013         struct nlattr *tb[NUM_NL80211_ATTR_COALESCE_RULE], *pat;
10014         int rem, pat_len, mask_len, pkt_offset, n_patterns = 0;
10015         struct nlattr *pat_tb[NUM_NL80211_PKTPAT];
10016
10017         err = nla_parse(tb, NL80211_ATTR_COALESCE_RULE_MAX, nla_data(rule),
10018                         nla_len(rule), nl80211_coalesce_policy);
10019         if (err)
10020                 return err;
10021
10022         if (tb[NL80211_ATTR_COALESCE_RULE_DELAY])
10023                 new_rule->delay =
10024                         nla_get_u32(tb[NL80211_ATTR_COALESCE_RULE_DELAY]);
10025         if (new_rule->delay > coalesce->max_delay)
10026                 return -EINVAL;
10027
10028         if (tb[NL80211_ATTR_COALESCE_RULE_CONDITION])
10029                 new_rule->condition =
10030                         nla_get_u32(tb[NL80211_ATTR_COALESCE_RULE_CONDITION]);
10031         if (new_rule->condition != NL80211_COALESCE_CONDITION_MATCH &&
10032             new_rule->condition != NL80211_COALESCE_CONDITION_NO_MATCH)
10033                 return -EINVAL;
10034
10035         if (!tb[NL80211_ATTR_COALESCE_RULE_PKT_PATTERN])
10036                 return -EINVAL;
10037
10038         nla_for_each_nested(pat, tb[NL80211_ATTR_COALESCE_RULE_PKT_PATTERN],
10039                             rem)
10040                 n_patterns++;
10041         if (n_patterns > coalesce->n_patterns)
10042                 return -EINVAL;
10043
10044         new_rule->patterns = kcalloc(n_patterns, sizeof(new_rule->patterns[0]),
10045                                      GFP_KERNEL);
10046         if (!new_rule->patterns)
10047                 return -ENOMEM;
10048
10049         new_rule->n_patterns = n_patterns;
10050         i = 0;
10051
10052         nla_for_each_nested(pat, tb[NL80211_ATTR_COALESCE_RULE_PKT_PATTERN],
10053                             rem) {
10054                 u8 *mask_pat;
10055
10056                 nla_parse(pat_tb, MAX_NL80211_PKTPAT, nla_data(pat),
10057                           nla_len(pat), NULL);
10058                 if (!pat_tb[NL80211_PKTPAT_MASK] ||
10059                     !pat_tb[NL80211_PKTPAT_PATTERN])
10060                         return -EINVAL;
10061                 pat_len = nla_len(pat_tb[NL80211_PKTPAT_PATTERN]);
10062                 mask_len = DIV_ROUND_UP(pat_len, 8);
10063                 if (nla_len(pat_tb[NL80211_PKTPAT_MASK]) != mask_len)
10064                         return -EINVAL;
10065                 if (pat_len > coalesce->pattern_max_len ||
10066                     pat_len < coalesce->pattern_min_len)
10067                         return -EINVAL;
10068
10069                 if (!pat_tb[NL80211_PKTPAT_OFFSET])
10070                         pkt_offset = 0;
10071                 else
10072                         pkt_offset = nla_get_u32(pat_tb[NL80211_PKTPAT_OFFSET]);
10073                 if (pkt_offset > coalesce->max_pkt_offset)
10074                         return -EINVAL;
10075                 new_rule->patterns[i].pkt_offset = pkt_offset;
10076
10077                 mask_pat = kmalloc(mask_len + pat_len, GFP_KERNEL);
10078                 if (!mask_pat)
10079                         return -ENOMEM;
10080
10081                 new_rule->patterns[i].mask = mask_pat;
10082                 memcpy(mask_pat, nla_data(pat_tb[NL80211_PKTPAT_MASK]),
10083                        mask_len);
10084
10085                 mask_pat += mask_len;
10086                 new_rule->patterns[i].pattern = mask_pat;
10087                 new_rule->patterns[i].pattern_len = pat_len;
10088                 memcpy(mask_pat, nla_data(pat_tb[NL80211_PKTPAT_PATTERN]),
10089                        pat_len);
10090                 i++;
10091         }
10092
10093         return 0;
10094 }
10095
10096 static int nl80211_set_coalesce(struct sk_buff *skb, struct genl_info *info)
10097 {
10098         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10099         const struct wiphy_coalesce_support *coalesce = rdev->wiphy.coalesce;
10100         struct cfg80211_coalesce new_coalesce = {};
10101         struct cfg80211_coalesce *n_coalesce;
10102         int err, rem_rule, n_rules = 0, i, j;
10103         struct nlattr *rule;
10104         struct cfg80211_coalesce_rules *tmp_rule;
10105
10106         if (!rdev->wiphy.coalesce || !rdev->ops->set_coalesce)
10107                 return -EOPNOTSUPP;
10108
10109         if (!info->attrs[NL80211_ATTR_COALESCE_RULE]) {
10110                 cfg80211_rdev_free_coalesce(rdev);
10111                 rdev_set_coalesce(rdev, NULL);
10112                 return 0;
10113         }
10114
10115         nla_for_each_nested(rule, info->attrs[NL80211_ATTR_COALESCE_RULE],
10116                             rem_rule)
10117                 n_rules++;
10118         if (n_rules > coalesce->n_rules)
10119                 return -EINVAL;
10120
10121         new_coalesce.rules = kcalloc(n_rules, sizeof(new_coalesce.rules[0]),
10122                                      GFP_KERNEL);
10123         if (!new_coalesce.rules)
10124                 return -ENOMEM;
10125
10126         new_coalesce.n_rules = n_rules;
10127         i = 0;
10128
10129         nla_for_each_nested(rule, info->attrs[NL80211_ATTR_COALESCE_RULE],
10130                             rem_rule) {
10131                 err = nl80211_parse_coalesce_rule(rdev, rule,
10132                                                   &new_coalesce.rules[i]);
10133                 if (err)
10134                         goto error;
10135
10136                 i++;
10137         }
10138
10139         err = rdev_set_coalesce(rdev, &new_coalesce);
10140         if (err)
10141                 goto error;
10142
10143         n_coalesce = kmemdup(&new_coalesce, sizeof(new_coalesce), GFP_KERNEL);
10144         if (!n_coalesce) {
10145                 err = -ENOMEM;
10146                 goto error;
10147         }
10148         cfg80211_rdev_free_coalesce(rdev);
10149         rdev->coalesce = n_coalesce;
10150
10151         return 0;
10152 error:
10153         for (i = 0; i < new_coalesce.n_rules; i++) {
10154                 tmp_rule = &new_coalesce.rules[i];
10155                 for (j = 0; j < tmp_rule->n_patterns; j++)
10156                         kfree(tmp_rule->patterns[j].mask);
10157                 kfree(tmp_rule->patterns);
10158         }
10159         kfree(new_coalesce.rules);
10160
10161         return err;
10162 }
10163
10164 static int nl80211_set_rekey_data(struct sk_buff *skb, struct genl_info *info)
10165 {
10166         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10167         struct net_device *dev = info->user_ptr[1];
10168         struct wireless_dev *wdev = dev->ieee80211_ptr;
10169         struct nlattr *tb[NUM_NL80211_REKEY_DATA];
10170         struct cfg80211_gtk_rekey_data rekey_data;
10171         int err;
10172
10173         if (!info->attrs[NL80211_ATTR_REKEY_DATA])
10174                 return -EINVAL;
10175
10176         err = nla_parse(tb, MAX_NL80211_REKEY_DATA,
10177                         nla_data(info->attrs[NL80211_ATTR_REKEY_DATA]),
10178                         nla_len(info->attrs[NL80211_ATTR_REKEY_DATA]),
10179                         nl80211_rekey_policy);
10180         if (err)
10181                 return err;
10182
10183         if (nla_len(tb[NL80211_REKEY_DATA_REPLAY_CTR]) != NL80211_REPLAY_CTR_LEN)
10184                 return -ERANGE;
10185         if (nla_len(tb[NL80211_REKEY_DATA_KEK]) != NL80211_KEK_LEN)
10186                 return -ERANGE;
10187         if (nla_len(tb[NL80211_REKEY_DATA_KCK]) != NL80211_KCK_LEN)
10188                 return -ERANGE;
10189
10190         rekey_data.kek = nla_data(tb[NL80211_REKEY_DATA_KEK]);
10191         rekey_data.kck = nla_data(tb[NL80211_REKEY_DATA_KCK]);
10192         rekey_data.replay_ctr = nla_data(tb[NL80211_REKEY_DATA_REPLAY_CTR]);
10193
10194         wdev_lock(wdev);
10195         if (!wdev->current_bss) {
10196                 err = -ENOTCONN;
10197                 goto out;
10198         }
10199
10200         if (!rdev->ops->set_rekey_data) {
10201                 err = -EOPNOTSUPP;
10202                 goto out;
10203         }
10204
10205         err = rdev_set_rekey_data(rdev, dev, &rekey_data);
10206  out:
10207         wdev_unlock(wdev);
10208         return err;
10209 }
10210
10211 static int nl80211_register_unexpected_frame(struct sk_buff *skb,
10212                                              struct genl_info *info)
10213 {
10214         struct net_device *dev = info->user_ptr[1];
10215         struct wireless_dev *wdev = dev->ieee80211_ptr;
10216
10217         if (wdev->iftype != NL80211_IFTYPE_AP &&
10218             wdev->iftype != NL80211_IFTYPE_P2P_GO)
10219                 return -EINVAL;
10220
10221         if (wdev->ap_unexpected_nlportid)
10222                 return -EBUSY;
10223
10224         wdev->ap_unexpected_nlportid = info->snd_portid;
10225         return 0;
10226 }
10227
10228 static int nl80211_probe_client(struct sk_buff *skb,
10229                                 struct genl_info *info)
10230 {
10231         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10232         struct net_device *dev = info->user_ptr[1];
10233         struct wireless_dev *wdev = dev->ieee80211_ptr;
10234         struct sk_buff *msg;
10235         void *hdr;
10236         const u8 *addr;
10237         u64 cookie;
10238         int err;
10239
10240         if (wdev->iftype != NL80211_IFTYPE_AP &&
10241             wdev->iftype != NL80211_IFTYPE_P2P_GO)
10242                 return -EOPNOTSUPP;
10243
10244         if (!info->attrs[NL80211_ATTR_MAC])
10245                 return -EINVAL;
10246
10247         if (!rdev->ops->probe_client)
10248                 return -EOPNOTSUPP;
10249
10250         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
10251         if (!msg)
10252                 return -ENOMEM;
10253
10254         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
10255                              NL80211_CMD_PROBE_CLIENT);
10256         if (!hdr) {
10257                 err = -ENOBUFS;
10258                 goto free_msg;
10259         }
10260
10261         addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
10262
10263         err = rdev_probe_client(rdev, dev, addr, &cookie);
10264         if (err)
10265                 goto free_msg;
10266
10267         if (nla_put_u64_64bit(msg, NL80211_ATTR_COOKIE, cookie,
10268                               NL80211_ATTR_PAD))
10269                 goto nla_put_failure;
10270
10271         genlmsg_end(msg, hdr);
10272
10273         return genlmsg_reply(msg, info);
10274
10275  nla_put_failure:
10276         err = -ENOBUFS;
10277  free_msg:
10278         nlmsg_free(msg);
10279         return err;
10280 }
10281
10282 static int nl80211_register_beacons(struct sk_buff *skb, struct genl_info *info)
10283 {
10284         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10285         struct cfg80211_beacon_registration *reg, *nreg;
10286         int rv;
10287
10288         if (!(rdev->wiphy.flags & WIPHY_FLAG_REPORTS_OBSS))
10289                 return -EOPNOTSUPP;
10290
10291         nreg = kzalloc(sizeof(*nreg), GFP_KERNEL);
10292         if (!nreg)
10293                 return -ENOMEM;
10294
10295         /* First, check if already registered. */
10296         spin_lock_bh(&rdev->beacon_registrations_lock);
10297         list_for_each_entry(reg, &rdev->beacon_registrations, list) {
10298                 if (reg->nlportid == info->snd_portid) {
10299                         rv = -EALREADY;
10300                         goto out_err;
10301                 }
10302         }
10303         /* Add it to the list */
10304         nreg->nlportid = info->snd_portid;
10305         list_add(&nreg->list, &rdev->beacon_registrations);
10306
10307         spin_unlock_bh(&rdev->beacon_registrations_lock);
10308
10309         return 0;
10310 out_err:
10311         spin_unlock_bh(&rdev->beacon_registrations_lock);
10312         kfree(nreg);
10313         return rv;
10314 }
10315
10316 static int nl80211_start_p2p_device(struct sk_buff *skb, struct genl_info *info)
10317 {
10318         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10319         struct wireless_dev *wdev = info->user_ptr[1];
10320         int err;
10321
10322         if (!rdev->ops->start_p2p_device)
10323                 return -EOPNOTSUPP;
10324
10325         if (wdev->iftype != NL80211_IFTYPE_P2P_DEVICE)
10326                 return -EOPNOTSUPP;
10327
10328         if (wdev->p2p_started)
10329                 return 0;
10330
10331         if (rfkill_blocked(rdev->rfkill))
10332                 return -ERFKILL;
10333
10334         err = rdev_start_p2p_device(rdev, wdev);
10335         if (err)
10336                 return err;
10337
10338         wdev->p2p_started = true;
10339         rdev->opencount++;
10340
10341         return 0;
10342 }
10343
10344 static int nl80211_stop_p2p_device(struct sk_buff *skb, struct genl_info *info)
10345 {
10346         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10347         struct wireless_dev *wdev = info->user_ptr[1];
10348
10349         if (wdev->iftype != NL80211_IFTYPE_P2P_DEVICE)
10350                 return -EOPNOTSUPP;
10351
10352         if (!rdev->ops->stop_p2p_device)
10353                 return -EOPNOTSUPP;
10354
10355         cfg80211_stop_p2p_device(rdev, wdev);
10356
10357         return 0;
10358 }
10359
10360 static int nl80211_get_protocol_features(struct sk_buff *skb,
10361                                          struct genl_info *info)
10362 {
10363         void *hdr;
10364         struct sk_buff *msg;
10365
10366         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
10367         if (!msg)
10368                 return -ENOMEM;
10369
10370         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
10371                              NL80211_CMD_GET_PROTOCOL_FEATURES);
10372         if (!hdr)
10373                 goto nla_put_failure;
10374
10375         if (nla_put_u32(msg, NL80211_ATTR_PROTOCOL_FEATURES,
10376                         NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP))
10377                 goto nla_put_failure;
10378
10379         genlmsg_end(msg, hdr);
10380         return genlmsg_reply(msg, info);
10381
10382  nla_put_failure:
10383         kfree_skb(msg);
10384         return -ENOBUFS;
10385 }
10386
10387 static int nl80211_update_ft_ies(struct sk_buff *skb, struct genl_info *info)
10388 {
10389         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10390         struct cfg80211_update_ft_ies_params ft_params;
10391         struct net_device *dev = info->user_ptr[1];
10392
10393         if (!rdev->ops->update_ft_ies)
10394                 return -EOPNOTSUPP;
10395
10396         if (!info->attrs[NL80211_ATTR_MDID] ||
10397             !is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
10398                 return -EINVAL;
10399
10400         memset(&ft_params, 0, sizeof(ft_params));
10401         ft_params.md = nla_get_u16(info->attrs[NL80211_ATTR_MDID]);
10402         ft_params.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
10403         ft_params.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
10404
10405         return rdev_update_ft_ies(rdev, dev, &ft_params);
10406 }
10407
10408 static int nl80211_crit_protocol_start(struct sk_buff *skb,
10409                                        struct genl_info *info)
10410 {
10411         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10412         struct wireless_dev *wdev = info->user_ptr[1];
10413         enum nl80211_crit_proto_id proto = NL80211_CRIT_PROTO_UNSPEC;
10414         u16 duration;
10415         int ret;
10416
10417         if (!rdev->ops->crit_proto_start)
10418                 return -EOPNOTSUPP;
10419
10420         if (WARN_ON(!rdev->ops->crit_proto_stop))
10421                 return -EINVAL;
10422
10423         if (rdev->crit_proto_nlportid)
10424                 return -EBUSY;
10425
10426         /* determine protocol if provided */
10427         if (info->attrs[NL80211_ATTR_CRIT_PROT_ID])
10428                 proto = nla_get_u16(info->attrs[NL80211_ATTR_CRIT_PROT_ID]);
10429
10430         if (proto >= NUM_NL80211_CRIT_PROTO)
10431                 return -EINVAL;
10432
10433         /* timeout must be provided */
10434         if (!info->attrs[NL80211_ATTR_MAX_CRIT_PROT_DURATION])
10435                 return -EINVAL;
10436
10437         duration =
10438                 nla_get_u16(info->attrs[NL80211_ATTR_MAX_CRIT_PROT_DURATION]);
10439
10440         if (duration > NL80211_CRIT_PROTO_MAX_DURATION)
10441                 return -ERANGE;
10442
10443         ret = rdev_crit_proto_start(rdev, wdev, proto, duration);
10444         if (!ret)
10445                 rdev->crit_proto_nlportid = info->snd_portid;
10446
10447         return ret;
10448 }
10449
10450 static int nl80211_crit_protocol_stop(struct sk_buff *skb,
10451                                       struct genl_info *info)
10452 {
10453         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10454         struct wireless_dev *wdev = info->user_ptr[1];
10455
10456         if (!rdev->ops->crit_proto_stop)
10457                 return -EOPNOTSUPP;
10458
10459         if (rdev->crit_proto_nlportid) {
10460                 rdev->crit_proto_nlportid = 0;
10461                 rdev_crit_proto_stop(rdev, wdev);
10462         }
10463         return 0;
10464 }
10465
10466 static int nl80211_vendor_cmd(struct sk_buff *skb, struct genl_info *info)
10467 {
10468         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10469         struct wireless_dev *wdev =
10470                 __cfg80211_wdev_from_attrs(genl_info_net(info), info->attrs);
10471         int i, err;
10472         u32 vid, subcmd;
10473
10474         if (!rdev->wiphy.vendor_commands)
10475                 return -EOPNOTSUPP;
10476
10477         if (IS_ERR(wdev)) {
10478                 err = PTR_ERR(wdev);
10479                 if (err != -EINVAL)
10480                         return err;
10481                 wdev = NULL;
10482         } else if (wdev->wiphy != &rdev->wiphy) {
10483                 return -EINVAL;
10484         }
10485
10486         if (!info->attrs[NL80211_ATTR_VENDOR_ID] ||
10487             !info->attrs[NL80211_ATTR_VENDOR_SUBCMD])
10488                 return -EINVAL;
10489
10490         vid = nla_get_u32(info->attrs[NL80211_ATTR_VENDOR_ID]);
10491         subcmd = nla_get_u32(info->attrs[NL80211_ATTR_VENDOR_SUBCMD]);
10492         for (i = 0; i < rdev->wiphy.n_vendor_commands; i++) {
10493                 const struct wiphy_vendor_command *vcmd;
10494                 void *data = NULL;
10495                 int len = 0;
10496
10497                 vcmd = &rdev->wiphy.vendor_commands[i];
10498
10499                 if (vcmd->info.vendor_id != vid || vcmd->info.subcmd != subcmd)
10500                         continue;
10501
10502                 if (vcmd->flags & (WIPHY_VENDOR_CMD_NEED_WDEV |
10503                                    WIPHY_VENDOR_CMD_NEED_NETDEV)) {
10504                         if (!wdev)
10505                                 return -EINVAL;
10506                         if (vcmd->flags & WIPHY_VENDOR_CMD_NEED_NETDEV &&
10507                             !wdev->netdev)
10508                                 return -EINVAL;
10509
10510                         if (vcmd->flags & WIPHY_VENDOR_CMD_NEED_RUNNING) {
10511                                 if (wdev->netdev &&
10512                                     !netif_running(wdev->netdev))
10513                                         return -ENETDOWN;
10514                                 if (!wdev->netdev && !wdev->p2p_started)
10515                                         return -ENETDOWN;
10516                         }
10517
10518                         if (!vcmd->doit)
10519                                 return -EOPNOTSUPP;
10520                 } else {
10521                         wdev = NULL;
10522                 }
10523
10524                 if (info->attrs[NL80211_ATTR_VENDOR_DATA]) {
10525                         data = nla_data(info->attrs[NL80211_ATTR_VENDOR_DATA]);
10526                         len = nla_len(info->attrs[NL80211_ATTR_VENDOR_DATA]);
10527                 }
10528
10529                 rdev->cur_cmd_info = info;
10530                 err = rdev->wiphy.vendor_commands[i].doit(&rdev->wiphy, wdev,
10531                                                           data, len);
10532                 rdev->cur_cmd_info = NULL;
10533                 return err;
10534         }
10535
10536         return -EOPNOTSUPP;
10537 }
10538
10539 static int nl80211_prepare_vendor_dump(struct sk_buff *skb,
10540                                        struct netlink_callback *cb,
10541                                        struct cfg80211_registered_device **rdev,
10542                                        struct wireless_dev **wdev)
10543 {
10544         u32 vid, subcmd;
10545         unsigned int i;
10546         int vcmd_idx = -1;
10547         int err;
10548         void *data = NULL;
10549         unsigned int data_len = 0;
10550
10551         rtnl_lock();
10552
10553         if (cb->args[0]) {
10554                 /* subtract the 1 again here */
10555                 struct wiphy *wiphy = wiphy_idx_to_wiphy(cb->args[0] - 1);
10556                 struct wireless_dev *tmp;
10557
10558                 if (!wiphy) {
10559                         err = -ENODEV;
10560                         goto out_unlock;
10561                 }
10562                 *rdev = wiphy_to_rdev(wiphy);
10563                 *wdev = NULL;
10564
10565                 if (cb->args[1]) {
10566                         list_for_each_entry(tmp, &wiphy->wdev_list, list) {
10567                                 if (tmp->identifier == cb->args[1] - 1) {
10568                                         *wdev = tmp;
10569                                         break;
10570                                 }
10571                         }
10572                 }
10573
10574                 /* keep rtnl locked in successful case */
10575                 return 0;
10576         }
10577
10578         err = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
10579                           nl80211_fam.attrbuf, nl80211_fam.maxattr,
10580                           nl80211_policy);
10581         if (err)
10582                 goto out_unlock;
10583
10584         if (!nl80211_fam.attrbuf[NL80211_ATTR_VENDOR_ID] ||
10585             !nl80211_fam.attrbuf[NL80211_ATTR_VENDOR_SUBCMD]) {
10586                 err = -EINVAL;
10587                 goto out_unlock;
10588         }
10589
10590         *wdev = __cfg80211_wdev_from_attrs(sock_net(skb->sk),
10591                                            nl80211_fam.attrbuf);
10592         if (IS_ERR(*wdev))
10593                 *wdev = NULL;
10594
10595         *rdev = __cfg80211_rdev_from_attrs(sock_net(skb->sk),
10596                                            nl80211_fam.attrbuf);
10597         if (IS_ERR(*rdev)) {
10598                 err = PTR_ERR(*rdev);
10599                 goto out_unlock;
10600         }
10601
10602         vid = nla_get_u32(nl80211_fam.attrbuf[NL80211_ATTR_VENDOR_ID]);
10603         subcmd = nla_get_u32(nl80211_fam.attrbuf[NL80211_ATTR_VENDOR_SUBCMD]);
10604
10605         for (i = 0; i < (*rdev)->wiphy.n_vendor_commands; i++) {
10606                 const struct wiphy_vendor_command *vcmd;
10607
10608                 vcmd = &(*rdev)->wiphy.vendor_commands[i];
10609
10610                 if (vcmd->info.vendor_id != vid || vcmd->info.subcmd != subcmd)
10611                         continue;
10612
10613                 if (!vcmd->dumpit) {
10614                         err = -EOPNOTSUPP;
10615                         goto out_unlock;
10616                 }
10617
10618                 vcmd_idx = i;
10619                 break;
10620         }
10621
10622         if (vcmd_idx < 0) {
10623                 err = -EOPNOTSUPP;
10624                 goto out_unlock;
10625         }
10626
10627         if (nl80211_fam.attrbuf[NL80211_ATTR_VENDOR_DATA]) {
10628                 data = nla_data(nl80211_fam.attrbuf[NL80211_ATTR_VENDOR_DATA]);
10629                 data_len = nla_len(nl80211_fam.attrbuf[NL80211_ATTR_VENDOR_DATA]);
10630         }
10631
10632         /* 0 is the first index - add 1 to parse only once */
10633         cb->args[0] = (*rdev)->wiphy_idx + 1;
10634         /* add 1 to know if it was NULL */
10635         cb->args[1] = *wdev ? (*wdev)->identifier + 1 : 0;
10636         cb->args[2] = vcmd_idx;
10637         cb->args[3] = (unsigned long)data;
10638         cb->args[4] = data_len;
10639
10640         /* keep rtnl locked in successful case */
10641         return 0;
10642  out_unlock:
10643         rtnl_unlock();
10644         return err;
10645 }
10646
10647 static int nl80211_vendor_cmd_dump(struct sk_buff *skb,
10648                                    struct netlink_callback *cb)
10649 {
10650         struct cfg80211_registered_device *rdev;
10651         struct wireless_dev *wdev;
10652         unsigned int vcmd_idx;
10653         const struct wiphy_vendor_command *vcmd;
10654         void *data;
10655         int data_len;
10656         int err;
10657         struct nlattr *vendor_data;
10658
10659         err = nl80211_prepare_vendor_dump(skb, cb, &rdev, &wdev);
10660         if (err)
10661                 return err;
10662
10663         vcmd_idx = cb->args[2];
10664         data = (void *)cb->args[3];
10665         data_len = cb->args[4];
10666         vcmd = &rdev->wiphy.vendor_commands[vcmd_idx];
10667
10668         if (vcmd->flags & (WIPHY_VENDOR_CMD_NEED_WDEV |
10669                            WIPHY_VENDOR_CMD_NEED_NETDEV)) {
10670                 if (!wdev)
10671                         return -EINVAL;
10672                 if (vcmd->flags & WIPHY_VENDOR_CMD_NEED_NETDEV &&
10673                     !wdev->netdev)
10674                         return -EINVAL;
10675
10676                 if (vcmd->flags & WIPHY_VENDOR_CMD_NEED_RUNNING) {
10677                         if (wdev->netdev &&
10678                             !netif_running(wdev->netdev))
10679                                 return -ENETDOWN;
10680                         if (!wdev->netdev && !wdev->p2p_started)
10681                                 return -ENETDOWN;
10682                 }
10683         }
10684
10685         while (1) {
10686                 void *hdr = nl80211hdr_put(skb, NETLINK_CB(cb->skb).portid,
10687                                            cb->nlh->nlmsg_seq, NLM_F_MULTI,
10688                                            NL80211_CMD_VENDOR);
10689                 if (!hdr)
10690                         break;
10691
10692                 if (nla_put_u32(skb, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10693                     (wdev && nla_put_u64_64bit(skb, NL80211_ATTR_WDEV,
10694                                                wdev_id(wdev),
10695                                                NL80211_ATTR_PAD))) {
10696                         genlmsg_cancel(skb, hdr);
10697                         break;
10698                 }
10699
10700                 vendor_data = nla_nest_start(skb, NL80211_ATTR_VENDOR_DATA);
10701                 if (!vendor_data) {
10702                         genlmsg_cancel(skb, hdr);
10703                         break;
10704                 }
10705
10706                 err = vcmd->dumpit(&rdev->wiphy, wdev, skb, data, data_len,
10707                                    (unsigned long *)&cb->args[5]);
10708                 nla_nest_end(skb, vendor_data);
10709
10710                 if (err == -ENOBUFS || err == -ENOENT) {
10711                         genlmsg_cancel(skb, hdr);
10712                         break;
10713                 } else if (err) {
10714                         genlmsg_cancel(skb, hdr);
10715                         goto out;
10716                 }
10717
10718                 genlmsg_end(skb, hdr);
10719         }
10720
10721         err = skb->len;
10722  out:
10723         rtnl_unlock();
10724         return err;
10725 }
10726
10727 struct sk_buff *__cfg80211_alloc_reply_skb(struct wiphy *wiphy,
10728                                            enum nl80211_commands cmd,
10729                                            enum nl80211_attrs attr,
10730                                            int approxlen)
10731 {
10732         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
10733
10734         if (WARN_ON(!rdev->cur_cmd_info))
10735                 return NULL;
10736
10737         return __cfg80211_alloc_vendor_skb(rdev, NULL, approxlen,
10738                                            rdev->cur_cmd_info->snd_portid,
10739                                            rdev->cur_cmd_info->snd_seq,
10740                                            cmd, attr, NULL, GFP_KERNEL);
10741 }
10742 EXPORT_SYMBOL(__cfg80211_alloc_reply_skb);
10743
10744 int cfg80211_vendor_cmd_reply(struct sk_buff *skb)
10745 {
10746         struct cfg80211_registered_device *rdev = ((void **)skb->cb)[0];
10747         void *hdr = ((void **)skb->cb)[1];
10748         struct nlattr *data = ((void **)skb->cb)[2];
10749
10750         /* clear CB data for netlink core to own from now on */
10751         memset(skb->cb, 0, sizeof(skb->cb));
10752
10753         if (WARN_ON(!rdev->cur_cmd_info)) {
10754                 kfree_skb(skb);
10755                 return -EINVAL;
10756         }
10757
10758         nla_nest_end(skb, data);
10759         genlmsg_end(skb, hdr);
10760         return genlmsg_reply(skb, rdev->cur_cmd_info);
10761 }
10762 EXPORT_SYMBOL_GPL(cfg80211_vendor_cmd_reply);
10763
10764 static int nl80211_set_qos_map(struct sk_buff *skb,
10765                                struct genl_info *info)
10766 {
10767         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10768         struct cfg80211_qos_map *qos_map = NULL;
10769         struct net_device *dev = info->user_ptr[1];
10770         u8 *pos, len, num_des, des_len, des;
10771         int ret;
10772
10773         if (!rdev->ops->set_qos_map)
10774                 return -EOPNOTSUPP;
10775
10776         if (info->attrs[NL80211_ATTR_QOS_MAP]) {
10777                 pos = nla_data(info->attrs[NL80211_ATTR_QOS_MAP]);
10778                 len = nla_len(info->attrs[NL80211_ATTR_QOS_MAP]);
10779
10780                 if (len % 2 || len < IEEE80211_QOS_MAP_LEN_MIN ||
10781                     len > IEEE80211_QOS_MAP_LEN_MAX)
10782                         return -EINVAL;
10783
10784                 qos_map = kzalloc(sizeof(struct cfg80211_qos_map), GFP_KERNEL);
10785                 if (!qos_map)
10786                         return -ENOMEM;
10787
10788                 num_des = (len - IEEE80211_QOS_MAP_LEN_MIN) >> 1;
10789                 if (num_des) {
10790                         des_len = num_des *
10791                                 sizeof(struct cfg80211_dscp_exception);
10792                         memcpy(qos_map->dscp_exception, pos, des_len);
10793                         qos_map->num_des = num_des;
10794                         for (des = 0; des < num_des; des++) {
10795                                 if (qos_map->dscp_exception[des].up > 7) {
10796                                         kfree(qos_map);
10797                                         return -EINVAL;
10798                                 }
10799                         }
10800                         pos += des_len;
10801                 }
10802                 memcpy(qos_map->up, pos, IEEE80211_QOS_MAP_LEN_MIN);
10803         }
10804
10805         wdev_lock(dev->ieee80211_ptr);
10806         ret = nl80211_key_allowed(dev->ieee80211_ptr);
10807         if (!ret)
10808                 ret = rdev_set_qos_map(rdev, dev, qos_map);
10809         wdev_unlock(dev->ieee80211_ptr);
10810
10811         kfree(qos_map);
10812         return ret;
10813 }
10814
10815 static int nl80211_add_tx_ts(struct sk_buff *skb, struct genl_info *info)
10816 {
10817         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10818         struct net_device *dev = info->user_ptr[1];
10819         struct wireless_dev *wdev = dev->ieee80211_ptr;
10820         const u8 *peer;
10821         u8 tsid, up;
10822         u16 admitted_time = 0;
10823         int err;
10824
10825         if (!(rdev->wiphy.features & NL80211_FEATURE_SUPPORTS_WMM_ADMISSION))
10826                 return -EOPNOTSUPP;
10827
10828         if (!info->attrs[NL80211_ATTR_TSID] || !info->attrs[NL80211_ATTR_MAC] ||
10829             !info->attrs[NL80211_ATTR_USER_PRIO])
10830                 return -EINVAL;
10831
10832         tsid = nla_get_u8(info->attrs[NL80211_ATTR_TSID]);
10833         if (tsid >= IEEE80211_NUM_TIDS)
10834                 return -EINVAL;
10835
10836         up = nla_get_u8(info->attrs[NL80211_ATTR_USER_PRIO]);
10837         if (up >= IEEE80211_NUM_UPS)
10838                 return -EINVAL;
10839
10840         /* WMM uses TIDs 0-7 even for TSPEC */
10841         if (tsid >= IEEE80211_FIRST_TSPEC_TSID) {
10842                 /* TODO: handle 802.11 TSPEC/admission control
10843                  * need more attributes for that (e.g. BA session requirement);
10844                  * change the WMM adminssion test above to allow both then
10845                  */
10846                 return -EINVAL;
10847         }
10848
10849         peer = nla_data(info->attrs[NL80211_ATTR_MAC]);
10850
10851         if (info->attrs[NL80211_ATTR_ADMITTED_TIME]) {
10852                 admitted_time =
10853                         nla_get_u16(info->attrs[NL80211_ATTR_ADMITTED_TIME]);
10854                 if (!admitted_time)
10855                         return -EINVAL;
10856         }
10857
10858         wdev_lock(wdev);
10859         switch (wdev->iftype) {
10860         case NL80211_IFTYPE_STATION:
10861         case NL80211_IFTYPE_P2P_CLIENT:
10862                 if (wdev->current_bss)
10863                         break;
10864                 err = -ENOTCONN;
10865                 goto out;
10866         default:
10867                 err = -EOPNOTSUPP;
10868                 goto out;
10869         }
10870
10871         err = rdev_add_tx_ts(rdev, dev, tsid, peer, up, admitted_time);
10872
10873  out:
10874         wdev_unlock(wdev);
10875         return err;
10876 }
10877
10878 static int nl80211_del_tx_ts(struct sk_buff *skb, struct genl_info *info)
10879 {
10880         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10881         struct net_device *dev = info->user_ptr[1];
10882         struct wireless_dev *wdev = dev->ieee80211_ptr;
10883         const u8 *peer;
10884         u8 tsid;
10885         int err;
10886
10887         if (!info->attrs[NL80211_ATTR_TSID] || !info->attrs[NL80211_ATTR_MAC])
10888                 return -EINVAL;
10889
10890         tsid = nla_get_u8(info->attrs[NL80211_ATTR_TSID]);
10891         peer = nla_data(info->attrs[NL80211_ATTR_MAC]);
10892
10893         wdev_lock(wdev);
10894         err = rdev_del_tx_ts(rdev, dev, tsid, peer);
10895         wdev_unlock(wdev);
10896
10897         return err;
10898 }
10899
10900 static int nl80211_tdls_channel_switch(struct sk_buff *skb,
10901                                        struct genl_info *info)
10902 {
10903         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10904         struct net_device *dev = info->user_ptr[1];
10905         struct wireless_dev *wdev = dev->ieee80211_ptr;
10906         struct cfg80211_chan_def chandef = {};
10907         const u8 *addr;
10908         u8 oper_class;
10909         int err;
10910
10911         if (!rdev->ops->tdls_channel_switch ||
10912             !(rdev->wiphy.features & NL80211_FEATURE_TDLS_CHANNEL_SWITCH))
10913                 return -EOPNOTSUPP;
10914
10915         switch (dev->ieee80211_ptr->iftype) {
10916         case NL80211_IFTYPE_STATION:
10917         case NL80211_IFTYPE_P2P_CLIENT:
10918                 break;
10919         default:
10920                 return -EOPNOTSUPP;
10921         }
10922
10923         if (!info->attrs[NL80211_ATTR_MAC] ||
10924             !info->attrs[NL80211_ATTR_OPER_CLASS])
10925                 return -EINVAL;
10926
10927         err = nl80211_parse_chandef(rdev, info, &chandef);
10928         if (err)
10929                 return err;
10930
10931         /*
10932          * Don't allow wide channels on the 2.4Ghz band, as per IEEE802.11-2012
10933          * section 10.22.6.2.1. Disallow 5/10Mhz channels as well for now, the
10934          * specification is not defined for them.
10935          */
10936         if (chandef.chan->band == NL80211_BAND_2GHZ &&
10937             chandef.width != NL80211_CHAN_WIDTH_20_NOHT &&
10938             chandef.width != NL80211_CHAN_WIDTH_20)
10939                 return -EINVAL;
10940
10941         /* we will be active on the TDLS link */
10942         if (!cfg80211_reg_can_beacon_relax(&rdev->wiphy, &chandef,
10943                                            wdev->iftype))
10944                 return -EINVAL;
10945
10946         /* don't allow switching to DFS channels */
10947         if (cfg80211_chandef_dfs_required(wdev->wiphy, &chandef, wdev->iftype))
10948                 return -EINVAL;
10949
10950         addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
10951         oper_class = nla_get_u8(info->attrs[NL80211_ATTR_OPER_CLASS]);
10952
10953         wdev_lock(wdev);
10954         err = rdev_tdls_channel_switch(rdev, dev, addr, oper_class, &chandef);
10955         wdev_unlock(wdev);
10956
10957         return err;
10958 }
10959
10960 static int nl80211_tdls_cancel_channel_switch(struct sk_buff *skb,
10961                                               struct genl_info *info)
10962 {
10963         struct cfg80211_registered_device *rdev = info->user_ptr[0];
10964         struct net_device *dev = info->user_ptr[1];
10965         struct wireless_dev *wdev = dev->ieee80211_ptr;
10966         const u8 *addr;
10967
10968         if (!rdev->ops->tdls_channel_switch ||
10969             !rdev->ops->tdls_cancel_channel_switch ||
10970             !(rdev->wiphy.features & NL80211_FEATURE_TDLS_CHANNEL_SWITCH))
10971                 return -EOPNOTSUPP;
10972
10973         switch (dev->ieee80211_ptr->iftype) {
10974         case NL80211_IFTYPE_STATION:
10975         case NL80211_IFTYPE_P2P_CLIENT:
10976                 break;
10977         default:
10978                 return -EOPNOTSUPP;
10979         }
10980
10981         if (!info->attrs[NL80211_ATTR_MAC])
10982                 return -EINVAL;
10983
10984         addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
10985
10986         wdev_lock(wdev);
10987         rdev_tdls_cancel_channel_switch(rdev, dev, addr);
10988         wdev_unlock(wdev);
10989
10990         return 0;
10991 }
10992
10993 #define NL80211_FLAG_NEED_WIPHY         0x01
10994 #define NL80211_FLAG_NEED_NETDEV        0x02
10995 #define NL80211_FLAG_NEED_RTNL          0x04
10996 #define NL80211_FLAG_CHECK_NETDEV_UP    0x08
10997 #define NL80211_FLAG_NEED_NETDEV_UP     (NL80211_FLAG_NEED_NETDEV |\
10998                                          NL80211_FLAG_CHECK_NETDEV_UP)
10999 #define NL80211_FLAG_NEED_WDEV          0x10
11000 /* If a netdev is associated, it must be UP, P2P must be started */
11001 #define NL80211_FLAG_NEED_WDEV_UP       (NL80211_FLAG_NEED_WDEV |\
11002                                          NL80211_FLAG_CHECK_NETDEV_UP)
11003 #define NL80211_FLAG_CLEAR_SKB          0x20
11004
11005 static int nl80211_pre_doit(const struct genl_ops *ops, struct sk_buff *skb,
11006                             struct genl_info *info)
11007 {
11008         struct cfg80211_registered_device *rdev;
11009         struct wireless_dev *wdev;
11010         struct net_device *dev;
11011         bool rtnl = ops->internal_flags & NL80211_FLAG_NEED_RTNL;
11012
11013         if (rtnl)
11014                 rtnl_lock();
11015
11016         if (ops->internal_flags & NL80211_FLAG_NEED_WIPHY) {
11017                 rdev = cfg80211_get_dev_from_info(genl_info_net(info), info);
11018                 if (IS_ERR(rdev)) {
11019                         if (rtnl)
11020                                 rtnl_unlock();
11021                         return PTR_ERR(rdev);
11022                 }
11023                 info->user_ptr[0] = rdev;
11024         } else if (ops->internal_flags & NL80211_FLAG_NEED_NETDEV ||
11025                    ops->internal_flags & NL80211_FLAG_NEED_WDEV) {
11026                 ASSERT_RTNL();
11027
11028                 wdev = __cfg80211_wdev_from_attrs(genl_info_net(info),
11029                                                   info->attrs);
11030                 if (IS_ERR(wdev)) {
11031                         if (rtnl)
11032                                 rtnl_unlock();
11033                         return PTR_ERR(wdev);
11034                 }
11035
11036                 dev = wdev->netdev;
11037                 rdev = wiphy_to_rdev(wdev->wiphy);
11038
11039                 if (ops->internal_flags & NL80211_FLAG_NEED_NETDEV) {
11040                         if (!dev) {
11041                                 if (rtnl)
11042                                         rtnl_unlock();
11043                                 return -EINVAL;
11044                         }
11045
11046                         info->user_ptr[1] = dev;
11047                 } else {
11048                         info->user_ptr[1] = wdev;
11049                 }
11050
11051                 if (dev) {
11052                         if (ops->internal_flags & NL80211_FLAG_CHECK_NETDEV_UP &&
11053                             !netif_running(dev)) {
11054                                 if (rtnl)
11055                                         rtnl_unlock();
11056                                 return -ENETDOWN;
11057                         }
11058
11059                         dev_hold(dev);
11060                 } else if (ops->internal_flags & NL80211_FLAG_CHECK_NETDEV_UP) {
11061                         if (!wdev->p2p_started) {
11062                                 if (rtnl)
11063                                         rtnl_unlock();
11064                                 return -ENETDOWN;
11065                         }
11066                 }
11067
11068                 info->user_ptr[0] = rdev;
11069         }
11070
11071         return 0;
11072 }
11073
11074 static void nl80211_post_doit(const struct genl_ops *ops, struct sk_buff *skb,
11075                               struct genl_info *info)
11076 {
11077         if (info->user_ptr[1]) {
11078                 if (ops->internal_flags & NL80211_FLAG_NEED_WDEV) {
11079                         struct wireless_dev *wdev = info->user_ptr[1];
11080
11081                         if (wdev->netdev)
11082                                 dev_put(wdev->netdev);
11083                 } else {
11084                         dev_put(info->user_ptr[1]);
11085                 }
11086         }
11087
11088         if (ops->internal_flags & NL80211_FLAG_NEED_RTNL)
11089                 rtnl_unlock();
11090
11091         /* If needed, clear the netlink message payload from the SKB
11092          * as it might contain key data that shouldn't stick around on
11093          * the heap after the SKB is freed. The netlink message header
11094          * is still needed for further processing, so leave it intact.
11095          */
11096         if (ops->internal_flags & NL80211_FLAG_CLEAR_SKB) {
11097                 struct nlmsghdr *nlh = nlmsg_hdr(skb);
11098
11099                 memset(nlmsg_data(nlh), 0, nlmsg_len(nlh));
11100         }
11101 }
11102
11103 static const struct genl_ops nl80211_ops[] = {
11104         {
11105                 .cmd = NL80211_CMD_GET_WIPHY,
11106                 .doit = nl80211_get_wiphy,
11107                 .dumpit = nl80211_dump_wiphy,
11108                 .done = nl80211_dump_wiphy_done,
11109                 .policy = nl80211_policy,
11110                 /* can be retrieved by unprivileged users */
11111                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11112                                   NL80211_FLAG_NEED_RTNL,
11113         },
11114         {
11115                 .cmd = NL80211_CMD_SET_WIPHY,
11116                 .doit = nl80211_set_wiphy,
11117                 .policy = nl80211_policy,
11118                 .flags = GENL_UNS_ADMIN_PERM,
11119                 .internal_flags = NL80211_FLAG_NEED_RTNL,
11120         },
11121         {
11122                 .cmd = NL80211_CMD_GET_INTERFACE,
11123                 .doit = nl80211_get_interface,
11124                 .dumpit = nl80211_dump_interface,
11125                 .policy = nl80211_policy,
11126                 /* can be retrieved by unprivileged users */
11127                 .internal_flags = NL80211_FLAG_NEED_WDEV |
11128                                   NL80211_FLAG_NEED_RTNL,
11129         },
11130         {
11131                 .cmd = NL80211_CMD_SET_INTERFACE,
11132                 .doit = nl80211_set_interface,
11133                 .policy = nl80211_policy,
11134                 .flags = GENL_UNS_ADMIN_PERM,
11135                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11136                                   NL80211_FLAG_NEED_RTNL,
11137         },
11138         {
11139                 .cmd = NL80211_CMD_NEW_INTERFACE,
11140                 .doit = nl80211_new_interface,
11141                 .policy = nl80211_policy,
11142                 .flags = GENL_UNS_ADMIN_PERM,
11143                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11144                                   NL80211_FLAG_NEED_RTNL,
11145         },
11146         {
11147                 .cmd = NL80211_CMD_DEL_INTERFACE,
11148                 .doit = nl80211_del_interface,
11149                 .policy = nl80211_policy,
11150                 .flags = GENL_UNS_ADMIN_PERM,
11151                 .internal_flags = NL80211_FLAG_NEED_WDEV |
11152                                   NL80211_FLAG_NEED_RTNL,
11153         },
11154         {
11155                 .cmd = NL80211_CMD_GET_KEY,
11156                 .doit = nl80211_get_key,
11157                 .policy = nl80211_policy,
11158                 .flags = GENL_UNS_ADMIN_PERM,
11159                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11160                                   NL80211_FLAG_NEED_RTNL,
11161         },
11162         {
11163                 .cmd = NL80211_CMD_SET_KEY,
11164                 .doit = nl80211_set_key,
11165                 .policy = nl80211_policy,
11166                 .flags = GENL_UNS_ADMIN_PERM,
11167                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11168                                   NL80211_FLAG_NEED_RTNL |
11169                                   NL80211_FLAG_CLEAR_SKB,
11170         },
11171         {
11172                 .cmd = NL80211_CMD_NEW_KEY,
11173                 .doit = nl80211_new_key,
11174                 .policy = nl80211_policy,
11175                 .flags = GENL_UNS_ADMIN_PERM,
11176                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11177                                   NL80211_FLAG_NEED_RTNL |
11178                                   NL80211_FLAG_CLEAR_SKB,
11179         },
11180         {
11181                 .cmd = NL80211_CMD_DEL_KEY,
11182                 .doit = nl80211_del_key,
11183                 .policy = nl80211_policy,
11184                 .flags = GENL_UNS_ADMIN_PERM,
11185                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11186                                   NL80211_FLAG_NEED_RTNL,
11187         },
11188         {
11189                 .cmd = NL80211_CMD_SET_BEACON,
11190                 .policy = nl80211_policy,
11191                 .flags = GENL_UNS_ADMIN_PERM,
11192                 .doit = nl80211_set_beacon,
11193                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11194                                   NL80211_FLAG_NEED_RTNL,
11195         },
11196         {
11197                 .cmd = NL80211_CMD_START_AP,
11198                 .policy = nl80211_policy,
11199                 .flags = GENL_UNS_ADMIN_PERM,
11200                 .doit = nl80211_start_ap,
11201                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11202                                   NL80211_FLAG_NEED_RTNL,
11203         },
11204         {
11205                 .cmd = NL80211_CMD_STOP_AP,
11206                 .policy = nl80211_policy,
11207                 .flags = GENL_UNS_ADMIN_PERM,
11208                 .doit = nl80211_stop_ap,
11209                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11210                                   NL80211_FLAG_NEED_RTNL,
11211         },
11212         {
11213                 .cmd = NL80211_CMD_GET_STATION,
11214                 .doit = nl80211_get_station,
11215                 .dumpit = nl80211_dump_station,
11216                 .policy = nl80211_policy,
11217                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11218                                   NL80211_FLAG_NEED_RTNL,
11219         },
11220         {
11221                 .cmd = NL80211_CMD_SET_STATION,
11222                 .doit = nl80211_set_station,
11223                 .policy = nl80211_policy,
11224                 .flags = GENL_UNS_ADMIN_PERM,
11225                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11226                                   NL80211_FLAG_NEED_RTNL,
11227         },
11228         {
11229                 .cmd = NL80211_CMD_NEW_STATION,
11230                 .doit = nl80211_new_station,
11231                 .policy = nl80211_policy,
11232                 .flags = GENL_UNS_ADMIN_PERM,
11233                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11234                                   NL80211_FLAG_NEED_RTNL,
11235         },
11236         {
11237                 .cmd = NL80211_CMD_DEL_STATION,
11238                 .doit = nl80211_del_station,
11239                 .policy = nl80211_policy,
11240                 .flags = GENL_UNS_ADMIN_PERM,
11241                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11242                                   NL80211_FLAG_NEED_RTNL,
11243         },
11244         {
11245                 .cmd = NL80211_CMD_GET_MPATH,
11246                 .doit = nl80211_get_mpath,
11247                 .dumpit = nl80211_dump_mpath,
11248                 .policy = nl80211_policy,
11249                 .flags = GENL_UNS_ADMIN_PERM,
11250                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11251                                   NL80211_FLAG_NEED_RTNL,
11252         },
11253         {
11254                 .cmd = NL80211_CMD_GET_MPP,
11255                 .doit = nl80211_get_mpp,
11256                 .dumpit = nl80211_dump_mpp,
11257                 .policy = nl80211_policy,
11258                 .flags = GENL_UNS_ADMIN_PERM,
11259                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11260                                   NL80211_FLAG_NEED_RTNL,
11261         },
11262         {
11263                 .cmd = NL80211_CMD_SET_MPATH,
11264                 .doit = nl80211_set_mpath,
11265                 .policy = nl80211_policy,
11266                 .flags = GENL_UNS_ADMIN_PERM,
11267                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11268                                   NL80211_FLAG_NEED_RTNL,
11269         },
11270         {
11271                 .cmd = NL80211_CMD_NEW_MPATH,
11272                 .doit = nl80211_new_mpath,
11273                 .policy = nl80211_policy,
11274                 .flags = GENL_UNS_ADMIN_PERM,
11275                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11276                                   NL80211_FLAG_NEED_RTNL,
11277         },
11278         {
11279                 .cmd = NL80211_CMD_DEL_MPATH,
11280                 .doit = nl80211_del_mpath,
11281                 .policy = nl80211_policy,
11282                 .flags = GENL_UNS_ADMIN_PERM,
11283                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11284                                   NL80211_FLAG_NEED_RTNL,
11285         },
11286         {
11287                 .cmd = NL80211_CMD_SET_BSS,
11288                 .doit = nl80211_set_bss,
11289                 .policy = nl80211_policy,
11290                 .flags = GENL_UNS_ADMIN_PERM,
11291                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11292                                   NL80211_FLAG_NEED_RTNL,
11293         },
11294         {
11295                 .cmd = NL80211_CMD_GET_REG,
11296                 .doit = nl80211_get_reg_do,
11297                 .dumpit = nl80211_get_reg_dump,
11298                 .policy = nl80211_policy,
11299                 .internal_flags = NL80211_FLAG_NEED_RTNL,
11300                 /* can be retrieved by unprivileged users */
11301         },
11302 #ifdef CONFIG_CFG80211_CRDA_SUPPORT
11303         {
11304                 .cmd = NL80211_CMD_SET_REG,
11305                 .doit = nl80211_set_reg,
11306                 .policy = nl80211_policy,
11307                 .flags = GENL_ADMIN_PERM,
11308                 .internal_flags = NL80211_FLAG_NEED_RTNL,
11309         },
11310 #endif
11311         {
11312                 .cmd = NL80211_CMD_REQ_SET_REG,
11313                 .doit = nl80211_req_set_reg,
11314                 .policy = nl80211_policy,
11315                 .flags = GENL_ADMIN_PERM,
11316         },
11317         {
11318                 .cmd = NL80211_CMD_GET_MESH_CONFIG,
11319                 .doit = nl80211_get_mesh_config,
11320                 .policy = nl80211_policy,
11321                 /* can be retrieved by unprivileged users */
11322                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11323                                   NL80211_FLAG_NEED_RTNL,
11324         },
11325         {
11326                 .cmd = NL80211_CMD_SET_MESH_CONFIG,
11327                 .doit = nl80211_update_mesh_config,
11328                 .policy = nl80211_policy,
11329                 .flags = GENL_UNS_ADMIN_PERM,
11330                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11331                                   NL80211_FLAG_NEED_RTNL,
11332         },
11333         {
11334                 .cmd = NL80211_CMD_TRIGGER_SCAN,
11335                 .doit = nl80211_trigger_scan,
11336                 .policy = nl80211_policy,
11337                 .flags = GENL_UNS_ADMIN_PERM,
11338                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
11339                                   NL80211_FLAG_NEED_RTNL,
11340         },
11341         {
11342                 .cmd = NL80211_CMD_ABORT_SCAN,
11343                 .doit = nl80211_abort_scan,
11344                 .policy = nl80211_policy,
11345                 .flags = GENL_UNS_ADMIN_PERM,
11346                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
11347                                   NL80211_FLAG_NEED_RTNL,
11348         },
11349         {
11350                 .cmd = NL80211_CMD_GET_SCAN,
11351                 .policy = nl80211_policy,
11352                 .dumpit = nl80211_dump_scan,
11353         },
11354         {
11355                 .cmd = NL80211_CMD_START_SCHED_SCAN,
11356                 .doit = nl80211_start_sched_scan,
11357                 .policy = nl80211_policy,
11358                 .flags = GENL_UNS_ADMIN_PERM,
11359                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11360                                   NL80211_FLAG_NEED_RTNL,
11361         },
11362         {
11363                 .cmd = NL80211_CMD_STOP_SCHED_SCAN,
11364                 .doit = nl80211_stop_sched_scan,
11365                 .policy = nl80211_policy,
11366                 .flags = GENL_UNS_ADMIN_PERM,
11367                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11368                                   NL80211_FLAG_NEED_RTNL,
11369         },
11370         {
11371                 .cmd = NL80211_CMD_AUTHENTICATE,
11372                 .doit = nl80211_authenticate,
11373                 .policy = nl80211_policy,
11374                 .flags = GENL_UNS_ADMIN_PERM,
11375                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11376                                   NL80211_FLAG_NEED_RTNL |
11377                                   NL80211_FLAG_CLEAR_SKB,
11378         },
11379         {
11380                 .cmd = NL80211_CMD_ASSOCIATE,
11381                 .doit = nl80211_associate,
11382                 .policy = nl80211_policy,
11383                 .flags = GENL_UNS_ADMIN_PERM,
11384                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11385                                   NL80211_FLAG_NEED_RTNL,
11386         },
11387         {
11388                 .cmd = NL80211_CMD_DEAUTHENTICATE,
11389                 .doit = nl80211_deauthenticate,
11390                 .policy = nl80211_policy,
11391                 .flags = GENL_UNS_ADMIN_PERM,
11392                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11393                                   NL80211_FLAG_NEED_RTNL,
11394         },
11395         {
11396                 .cmd = NL80211_CMD_DISASSOCIATE,
11397                 .doit = nl80211_disassociate,
11398                 .policy = nl80211_policy,
11399                 .flags = GENL_UNS_ADMIN_PERM,
11400                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11401                                   NL80211_FLAG_NEED_RTNL,
11402         },
11403         {
11404                 .cmd = NL80211_CMD_JOIN_IBSS,
11405                 .doit = nl80211_join_ibss,
11406                 .policy = nl80211_policy,
11407                 .flags = GENL_UNS_ADMIN_PERM,
11408                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11409                                   NL80211_FLAG_NEED_RTNL,
11410         },
11411         {
11412                 .cmd = NL80211_CMD_LEAVE_IBSS,
11413                 .doit = nl80211_leave_ibss,
11414                 .policy = nl80211_policy,
11415                 .flags = GENL_UNS_ADMIN_PERM,
11416                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11417                                   NL80211_FLAG_NEED_RTNL,
11418         },
11419 #ifdef CONFIG_NL80211_TESTMODE
11420         {
11421                 .cmd = NL80211_CMD_TESTMODE,
11422                 .doit = nl80211_testmode_do,
11423                 .dumpit = nl80211_testmode_dump,
11424                 .policy = nl80211_policy,
11425                 .flags = GENL_UNS_ADMIN_PERM,
11426                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11427                                   NL80211_FLAG_NEED_RTNL,
11428         },
11429 #endif
11430         {
11431                 .cmd = NL80211_CMD_CONNECT,
11432                 .doit = nl80211_connect,
11433                 .policy = nl80211_policy,
11434                 .flags = GENL_UNS_ADMIN_PERM,
11435                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11436                                   NL80211_FLAG_NEED_RTNL,
11437         },
11438         {
11439                 .cmd = NL80211_CMD_DISCONNECT,
11440                 .doit = nl80211_disconnect,
11441                 .policy = nl80211_policy,
11442                 .flags = GENL_UNS_ADMIN_PERM,
11443                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11444                                   NL80211_FLAG_NEED_RTNL,
11445         },
11446         {
11447                 .cmd = NL80211_CMD_SET_WIPHY_NETNS,
11448                 .doit = nl80211_wiphy_netns,
11449                 .policy = nl80211_policy,
11450                 .flags = GENL_UNS_ADMIN_PERM,
11451                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11452                                   NL80211_FLAG_NEED_RTNL,
11453         },
11454         {
11455                 .cmd = NL80211_CMD_GET_SURVEY,
11456                 .policy = nl80211_policy,
11457                 .dumpit = nl80211_dump_survey,
11458         },
11459         {
11460                 .cmd = NL80211_CMD_SET_PMKSA,
11461                 .doit = nl80211_setdel_pmksa,
11462                 .policy = nl80211_policy,
11463                 .flags = GENL_UNS_ADMIN_PERM,
11464                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11465                                   NL80211_FLAG_NEED_RTNL,
11466         },
11467         {
11468                 .cmd = NL80211_CMD_DEL_PMKSA,
11469                 .doit = nl80211_setdel_pmksa,
11470                 .policy = nl80211_policy,
11471                 .flags = GENL_UNS_ADMIN_PERM,
11472                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11473                                   NL80211_FLAG_NEED_RTNL,
11474         },
11475         {
11476                 .cmd = NL80211_CMD_FLUSH_PMKSA,
11477                 .doit = nl80211_flush_pmksa,
11478                 .policy = nl80211_policy,
11479                 .flags = GENL_UNS_ADMIN_PERM,
11480                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11481                                   NL80211_FLAG_NEED_RTNL,
11482         },
11483         {
11484                 .cmd = NL80211_CMD_REMAIN_ON_CHANNEL,
11485                 .doit = nl80211_remain_on_channel,
11486                 .policy = nl80211_policy,
11487                 .flags = GENL_UNS_ADMIN_PERM,
11488                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
11489                                   NL80211_FLAG_NEED_RTNL,
11490         },
11491         {
11492                 .cmd = NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL,
11493                 .doit = nl80211_cancel_remain_on_channel,
11494                 .policy = nl80211_policy,
11495                 .flags = GENL_UNS_ADMIN_PERM,
11496                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
11497                                   NL80211_FLAG_NEED_RTNL,
11498         },
11499         {
11500                 .cmd = NL80211_CMD_SET_TX_BITRATE_MASK,
11501                 .doit = nl80211_set_tx_bitrate_mask,
11502                 .policy = nl80211_policy,
11503                 .flags = GENL_UNS_ADMIN_PERM,
11504                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11505                                   NL80211_FLAG_NEED_RTNL,
11506         },
11507         {
11508                 .cmd = NL80211_CMD_REGISTER_FRAME,
11509                 .doit = nl80211_register_mgmt,
11510                 .policy = nl80211_policy,
11511                 .flags = GENL_UNS_ADMIN_PERM,
11512                 .internal_flags = NL80211_FLAG_NEED_WDEV |
11513                                   NL80211_FLAG_NEED_RTNL,
11514         },
11515         {
11516                 .cmd = NL80211_CMD_FRAME,
11517                 .doit = nl80211_tx_mgmt,
11518                 .policy = nl80211_policy,
11519                 .flags = GENL_UNS_ADMIN_PERM,
11520                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
11521                                   NL80211_FLAG_NEED_RTNL,
11522         },
11523         {
11524                 .cmd = NL80211_CMD_FRAME_WAIT_CANCEL,
11525                 .doit = nl80211_tx_mgmt_cancel_wait,
11526                 .policy = nl80211_policy,
11527                 .flags = GENL_UNS_ADMIN_PERM,
11528                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
11529                                   NL80211_FLAG_NEED_RTNL,
11530         },
11531         {
11532                 .cmd = NL80211_CMD_SET_POWER_SAVE,
11533                 .doit = nl80211_set_power_save,
11534                 .policy = nl80211_policy,
11535                 .flags = GENL_UNS_ADMIN_PERM,
11536                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11537                                   NL80211_FLAG_NEED_RTNL,
11538         },
11539         {
11540                 .cmd = NL80211_CMD_GET_POWER_SAVE,
11541                 .doit = nl80211_get_power_save,
11542                 .policy = nl80211_policy,
11543                 /* can be retrieved by unprivileged users */
11544                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11545                                   NL80211_FLAG_NEED_RTNL,
11546         },
11547         {
11548                 .cmd = NL80211_CMD_SET_CQM,
11549                 .doit = nl80211_set_cqm,
11550                 .policy = nl80211_policy,
11551                 .flags = GENL_UNS_ADMIN_PERM,
11552                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11553                                   NL80211_FLAG_NEED_RTNL,
11554         },
11555         {
11556                 .cmd = NL80211_CMD_SET_CHANNEL,
11557                 .doit = nl80211_set_channel,
11558                 .policy = nl80211_policy,
11559                 .flags = GENL_UNS_ADMIN_PERM,
11560                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11561                                   NL80211_FLAG_NEED_RTNL,
11562         },
11563         {
11564                 .cmd = NL80211_CMD_SET_WDS_PEER,
11565                 .doit = nl80211_set_wds_peer,
11566                 .policy = nl80211_policy,
11567                 .flags = GENL_UNS_ADMIN_PERM,
11568                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11569                                   NL80211_FLAG_NEED_RTNL,
11570         },
11571         {
11572                 .cmd = NL80211_CMD_JOIN_MESH,
11573                 .doit = nl80211_join_mesh,
11574                 .policy = nl80211_policy,
11575                 .flags = GENL_UNS_ADMIN_PERM,
11576                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11577                                   NL80211_FLAG_NEED_RTNL,
11578         },
11579         {
11580                 .cmd = NL80211_CMD_LEAVE_MESH,
11581                 .doit = nl80211_leave_mesh,
11582                 .policy = nl80211_policy,
11583                 .flags = GENL_UNS_ADMIN_PERM,
11584                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11585                                   NL80211_FLAG_NEED_RTNL,
11586         },
11587         {
11588                 .cmd = NL80211_CMD_JOIN_OCB,
11589                 .doit = nl80211_join_ocb,
11590                 .policy = nl80211_policy,
11591                 .flags = GENL_UNS_ADMIN_PERM,
11592                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11593                                   NL80211_FLAG_NEED_RTNL,
11594         },
11595         {
11596                 .cmd = NL80211_CMD_LEAVE_OCB,
11597                 .doit = nl80211_leave_ocb,
11598                 .policy = nl80211_policy,
11599                 .flags = GENL_UNS_ADMIN_PERM,
11600                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11601                                   NL80211_FLAG_NEED_RTNL,
11602         },
11603 #ifdef CONFIG_PM
11604         {
11605                 .cmd = NL80211_CMD_GET_WOWLAN,
11606                 .doit = nl80211_get_wowlan,
11607                 .policy = nl80211_policy,
11608                 /* can be retrieved by unprivileged users */
11609                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11610                                   NL80211_FLAG_NEED_RTNL,
11611         },
11612         {
11613                 .cmd = NL80211_CMD_SET_WOWLAN,
11614                 .doit = nl80211_set_wowlan,
11615                 .policy = nl80211_policy,
11616                 .flags = GENL_UNS_ADMIN_PERM,
11617                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11618                                   NL80211_FLAG_NEED_RTNL,
11619         },
11620 #endif
11621         {
11622                 .cmd = NL80211_CMD_SET_REKEY_OFFLOAD,
11623                 .doit = nl80211_set_rekey_data,
11624                 .policy = nl80211_policy,
11625                 .flags = GENL_UNS_ADMIN_PERM,
11626                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11627                                   NL80211_FLAG_NEED_RTNL |
11628                                   NL80211_FLAG_CLEAR_SKB,
11629         },
11630         {
11631                 .cmd = NL80211_CMD_TDLS_MGMT,
11632                 .doit = nl80211_tdls_mgmt,
11633                 .policy = nl80211_policy,
11634                 .flags = GENL_UNS_ADMIN_PERM,
11635                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11636                                   NL80211_FLAG_NEED_RTNL,
11637         },
11638         {
11639                 .cmd = NL80211_CMD_TDLS_OPER,
11640                 .doit = nl80211_tdls_oper,
11641                 .policy = nl80211_policy,
11642                 .flags = GENL_UNS_ADMIN_PERM,
11643                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11644                                   NL80211_FLAG_NEED_RTNL,
11645         },
11646         {
11647                 .cmd = NL80211_CMD_UNEXPECTED_FRAME,
11648                 .doit = nl80211_register_unexpected_frame,
11649                 .policy = nl80211_policy,
11650                 .flags = GENL_UNS_ADMIN_PERM,
11651                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11652                                   NL80211_FLAG_NEED_RTNL,
11653         },
11654         {
11655                 .cmd = NL80211_CMD_PROBE_CLIENT,
11656                 .doit = nl80211_probe_client,
11657                 .policy = nl80211_policy,
11658                 .flags = GENL_UNS_ADMIN_PERM,
11659                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11660                                   NL80211_FLAG_NEED_RTNL,
11661         },
11662         {
11663                 .cmd = NL80211_CMD_REGISTER_BEACONS,
11664                 .doit = nl80211_register_beacons,
11665                 .policy = nl80211_policy,
11666                 .flags = GENL_UNS_ADMIN_PERM,
11667                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11668                                   NL80211_FLAG_NEED_RTNL,
11669         },
11670         {
11671                 .cmd = NL80211_CMD_SET_NOACK_MAP,
11672                 .doit = nl80211_set_noack_map,
11673                 .policy = nl80211_policy,
11674                 .flags = GENL_UNS_ADMIN_PERM,
11675                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11676                                   NL80211_FLAG_NEED_RTNL,
11677         },
11678         {
11679                 .cmd = NL80211_CMD_START_P2P_DEVICE,
11680                 .doit = nl80211_start_p2p_device,
11681                 .policy = nl80211_policy,
11682                 .flags = GENL_UNS_ADMIN_PERM,
11683                 .internal_flags = NL80211_FLAG_NEED_WDEV |
11684                                   NL80211_FLAG_NEED_RTNL,
11685         },
11686         {
11687                 .cmd = NL80211_CMD_STOP_P2P_DEVICE,
11688                 .doit = nl80211_stop_p2p_device,
11689                 .policy = nl80211_policy,
11690                 .flags = GENL_UNS_ADMIN_PERM,
11691                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
11692                                   NL80211_FLAG_NEED_RTNL,
11693         },
11694         {
11695                 .cmd = NL80211_CMD_SET_MCAST_RATE,
11696                 .doit = nl80211_set_mcast_rate,
11697                 .policy = nl80211_policy,
11698                 .flags = GENL_UNS_ADMIN_PERM,
11699                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11700                                   NL80211_FLAG_NEED_RTNL,
11701         },
11702         {
11703                 .cmd = NL80211_CMD_SET_MAC_ACL,
11704                 .doit = nl80211_set_mac_acl,
11705                 .policy = nl80211_policy,
11706                 .flags = GENL_UNS_ADMIN_PERM,
11707                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
11708                                   NL80211_FLAG_NEED_RTNL,
11709         },
11710         {
11711                 .cmd = NL80211_CMD_RADAR_DETECT,
11712                 .doit = nl80211_start_radar_detection,
11713                 .policy = nl80211_policy,
11714                 .flags = GENL_UNS_ADMIN_PERM,
11715                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11716                                   NL80211_FLAG_NEED_RTNL,
11717         },
11718         {
11719                 .cmd = NL80211_CMD_GET_PROTOCOL_FEATURES,
11720                 .doit = nl80211_get_protocol_features,
11721                 .policy = nl80211_policy,
11722         },
11723         {
11724                 .cmd = NL80211_CMD_UPDATE_FT_IES,
11725                 .doit = nl80211_update_ft_ies,
11726                 .policy = nl80211_policy,
11727                 .flags = GENL_UNS_ADMIN_PERM,
11728                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11729                                   NL80211_FLAG_NEED_RTNL,
11730         },
11731         {
11732                 .cmd = NL80211_CMD_CRIT_PROTOCOL_START,
11733                 .doit = nl80211_crit_protocol_start,
11734                 .policy = nl80211_policy,
11735                 .flags = GENL_UNS_ADMIN_PERM,
11736                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
11737                                   NL80211_FLAG_NEED_RTNL,
11738         },
11739         {
11740                 .cmd = NL80211_CMD_CRIT_PROTOCOL_STOP,
11741                 .doit = nl80211_crit_protocol_stop,
11742                 .policy = nl80211_policy,
11743                 .flags = GENL_UNS_ADMIN_PERM,
11744                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
11745                                   NL80211_FLAG_NEED_RTNL,
11746         },
11747         {
11748                 .cmd = NL80211_CMD_GET_COALESCE,
11749                 .doit = nl80211_get_coalesce,
11750                 .policy = nl80211_policy,
11751                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11752                                   NL80211_FLAG_NEED_RTNL,
11753         },
11754         {
11755                 .cmd = NL80211_CMD_SET_COALESCE,
11756                 .doit = nl80211_set_coalesce,
11757                 .policy = nl80211_policy,
11758                 .flags = GENL_UNS_ADMIN_PERM,
11759                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11760                                   NL80211_FLAG_NEED_RTNL,
11761         },
11762         {
11763                 .cmd = NL80211_CMD_CHANNEL_SWITCH,
11764                 .doit = nl80211_channel_switch,
11765                 .policy = nl80211_policy,
11766                 .flags = GENL_UNS_ADMIN_PERM,
11767                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11768                                   NL80211_FLAG_NEED_RTNL,
11769         },
11770         {
11771                 .cmd = NL80211_CMD_VENDOR,
11772                 .doit = nl80211_vendor_cmd,
11773                 .dumpit = nl80211_vendor_cmd_dump,
11774                 .policy = nl80211_policy,
11775                 .flags = GENL_UNS_ADMIN_PERM,
11776                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
11777                                   NL80211_FLAG_NEED_RTNL,
11778         },
11779         {
11780                 .cmd = NL80211_CMD_SET_QOS_MAP,
11781                 .doit = nl80211_set_qos_map,
11782                 .policy = nl80211_policy,
11783                 .flags = GENL_UNS_ADMIN_PERM,
11784                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11785                                   NL80211_FLAG_NEED_RTNL,
11786         },
11787         {
11788                 .cmd = NL80211_CMD_ADD_TX_TS,
11789                 .doit = nl80211_add_tx_ts,
11790                 .policy = nl80211_policy,
11791                 .flags = GENL_UNS_ADMIN_PERM,
11792                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11793                                   NL80211_FLAG_NEED_RTNL,
11794         },
11795         {
11796                 .cmd = NL80211_CMD_DEL_TX_TS,
11797                 .doit = nl80211_del_tx_ts,
11798                 .policy = nl80211_policy,
11799                 .flags = GENL_UNS_ADMIN_PERM,
11800                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11801                                   NL80211_FLAG_NEED_RTNL,
11802         },
11803         {
11804                 .cmd = NL80211_CMD_TDLS_CHANNEL_SWITCH,
11805                 .doit = nl80211_tdls_channel_switch,
11806                 .policy = nl80211_policy,
11807                 .flags = GENL_UNS_ADMIN_PERM,
11808                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11809                                   NL80211_FLAG_NEED_RTNL,
11810         },
11811         {
11812                 .cmd = NL80211_CMD_TDLS_CANCEL_CHANNEL_SWITCH,
11813                 .doit = nl80211_tdls_cancel_channel_switch,
11814                 .policy = nl80211_policy,
11815                 .flags = GENL_UNS_ADMIN_PERM,
11816                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
11817                                   NL80211_FLAG_NEED_RTNL,
11818         },
11819 };
11820
11821 /* notification functions */
11822
11823 void nl80211_notify_wiphy(struct cfg80211_registered_device *rdev,
11824                           enum nl80211_commands cmd)
11825 {
11826         struct sk_buff *msg;
11827         struct nl80211_dump_wiphy_state state = {};
11828
11829         WARN_ON(cmd != NL80211_CMD_NEW_WIPHY &&
11830                 cmd != NL80211_CMD_DEL_WIPHY);
11831
11832         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
11833         if (!msg)
11834                 return;
11835
11836         if (nl80211_send_wiphy(rdev, cmd, msg, 0, 0, 0, &state) < 0) {
11837                 nlmsg_free(msg);
11838                 return;
11839         }
11840
11841         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
11842                                 NL80211_MCGRP_CONFIG, GFP_KERNEL);
11843 }
11844
11845 void nl80211_notify_iface(struct cfg80211_registered_device *rdev,
11846                                 struct wireless_dev *wdev,
11847                                 enum nl80211_commands cmd)
11848 {
11849         struct sk_buff *msg;
11850
11851         WARN_ON(cmd != NL80211_CMD_NEW_INTERFACE &&
11852                 cmd != NL80211_CMD_DEL_INTERFACE);
11853
11854         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
11855         if (!msg)
11856                 return;
11857
11858         if (nl80211_send_iface(msg, 0, 0, 0, rdev, wdev,
11859                                cmd == NL80211_CMD_DEL_INTERFACE) < 0) {
11860                 nlmsg_free(msg);
11861                 return;
11862         }
11863
11864         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
11865                                 NL80211_MCGRP_CONFIG, GFP_KERNEL);
11866 }
11867
11868 static int nl80211_add_scan_req(struct sk_buff *msg,
11869                                 struct cfg80211_registered_device *rdev)
11870 {
11871         struct cfg80211_scan_request *req = rdev->scan_req;
11872         struct nlattr *nest;
11873         int i;
11874
11875         if (WARN_ON(!req))
11876                 return 0;
11877
11878         nest = nla_nest_start(msg, NL80211_ATTR_SCAN_SSIDS);
11879         if (!nest)
11880                 goto nla_put_failure;
11881         for (i = 0; i < req->n_ssids; i++) {
11882                 if (nla_put(msg, i, req->ssids[i].ssid_len, req->ssids[i].ssid))
11883                         goto nla_put_failure;
11884         }
11885         nla_nest_end(msg, nest);
11886
11887         nest = nla_nest_start(msg, NL80211_ATTR_SCAN_FREQUENCIES);
11888         if (!nest)
11889                 goto nla_put_failure;
11890         for (i = 0; i < req->n_channels; i++) {
11891                 if (nla_put_u32(msg, i, req->channels[i]->center_freq))
11892                         goto nla_put_failure;
11893         }
11894         nla_nest_end(msg, nest);
11895
11896         if (req->ie &&
11897             nla_put(msg, NL80211_ATTR_IE, req->ie_len, req->ie))
11898                 goto nla_put_failure;
11899
11900         if (req->flags &&
11901             nla_put_u32(msg, NL80211_ATTR_SCAN_FLAGS, req->flags))
11902                 goto nla_put_failure;
11903
11904         if (req->info.scan_start_tsf &&
11905             (nla_put_u64_64bit(msg, NL80211_ATTR_SCAN_START_TIME_TSF,
11906                                req->info.scan_start_tsf, NL80211_BSS_PAD) ||
11907              nla_put(msg, NL80211_ATTR_SCAN_START_TIME_TSF_BSSID, ETH_ALEN,
11908                      req->info.tsf_bssid)))
11909                 goto nla_put_failure;
11910
11911         return 0;
11912  nla_put_failure:
11913         return -ENOBUFS;
11914 }
11915
11916 static int nl80211_send_scan_msg(struct sk_buff *msg,
11917                                  struct cfg80211_registered_device *rdev,
11918                                  struct wireless_dev *wdev,
11919                                  u32 portid, u32 seq, int flags,
11920                                  u32 cmd)
11921 {
11922         void *hdr;
11923
11924         hdr = nl80211hdr_put(msg, portid, seq, flags, cmd);
11925         if (!hdr)
11926                 return -1;
11927
11928         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
11929             (wdev->netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
11930                                          wdev->netdev->ifindex)) ||
11931             nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
11932                               NL80211_ATTR_PAD))
11933                 goto nla_put_failure;
11934
11935         /* ignore errors and send incomplete event anyway */
11936         nl80211_add_scan_req(msg, rdev);
11937
11938         genlmsg_end(msg, hdr);
11939         return 0;
11940
11941  nla_put_failure:
11942         genlmsg_cancel(msg, hdr);
11943         return -EMSGSIZE;
11944 }
11945
11946 static int
11947 nl80211_send_sched_scan_msg(struct sk_buff *msg,
11948                             struct cfg80211_registered_device *rdev,
11949                             struct net_device *netdev,
11950                             u32 portid, u32 seq, int flags, u32 cmd)
11951 {
11952         void *hdr;
11953
11954         hdr = nl80211hdr_put(msg, portid, seq, flags, cmd);
11955         if (!hdr)
11956                 return -1;
11957
11958         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
11959             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
11960                 goto nla_put_failure;
11961
11962         genlmsg_end(msg, hdr);
11963         return 0;
11964
11965  nla_put_failure:
11966         genlmsg_cancel(msg, hdr);
11967         return -EMSGSIZE;
11968 }
11969
11970 void nl80211_send_scan_start(struct cfg80211_registered_device *rdev,
11971                              struct wireless_dev *wdev)
11972 {
11973         struct sk_buff *msg;
11974
11975         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
11976         if (!msg)
11977                 return;
11978
11979         if (nl80211_send_scan_msg(msg, rdev, wdev, 0, 0, 0,
11980                                   NL80211_CMD_TRIGGER_SCAN) < 0) {
11981                 nlmsg_free(msg);
11982                 return;
11983         }
11984
11985         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
11986                                 NL80211_MCGRP_SCAN, GFP_KERNEL);
11987 }
11988
11989 struct sk_buff *nl80211_build_scan_msg(struct cfg80211_registered_device *rdev,
11990                                        struct wireless_dev *wdev, bool aborted)
11991 {
11992         struct sk_buff *msg;
11993
11994         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
11995         if (!msg)
11996                 return NULL;
11997
11998         if (nl80211_send_scan_msg(msg, rdev, wdev, 0, 0, 0,
11999                                   aborted ? NL80211_CMD_SCAN_ABORTED :
12000                                             NL80211_CMD_NEW_SCAN_RESULTS) < 0) {
12001                 nlmsg_free(msg);
12002                 return NULL;
12003         }
12004
12005         return msg;
12006 }
12007
12008 void nl80211_send_scan_result(struct cfg80211_registered_device *rdev,
12009                               struct sk_buff *msg)
12010 {
12011         if (!msg)
12012                 return;
12013
12014         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12015                                 NL80211_MCGRP_SCAN, GFP_KERNEL);
12016 }
12017
12018 void nl80211_send_sched_scan_results(struct cfg80211_registered_device *rdev,
12019                                      struct net_device *netdev)
12020 {
12021         struct sk_buff *msg;
12022
12023         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
12024         if (!msg)
12025                 return;
12026
12027         if (nl80211_send_sched_scan_msg(msg, rdev, netdev, 0, 0, 0,
12028                                         NL80211_CMD_SCHED_SCAN_RESULTS) < 0) {
12029                 nlmsg_free(msg);
12030                 return;
12031         }
12032
12033         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12034                                 NL80211_MCGRP_SCAN, GFP_KERNEL);
12035 }
12036
12037 void nl80211_send_sched_scan(struct cfg80211_registered_device *rdev,
12038                              struct net_device *netdev, u32 cmd)
12039 {
12040         struct sk_buff *msg;
12041
12042         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
12043         if (!msg)
12044                 return;
12045
12046         if (nl80211_send_sched_scan_msg(msg, rdev, netdev, 0, 0, 0, cmd) < 0) {
12047                 nlmsg_free(msg);
12048                 return;
12049         }
12050
12051         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12052                                 NL80211_MCGRP_SCAN, GFP_KERNEL);
12053 }
12054
12055 static bool nl80211_reg_change_event_fill(struct sk_buff *msg,
12056                                           struct regulatory_request *request)
12057 {
12058         /* Userspace can always count this one always being set */
12059         if (nla_put_u8(msg, NL80211_ATTR_REG_INITIATOR, request->initiator))
12060                 goto nla_put_failure;
12061
12062         if (request->alpha2[0] == '0' && request->alpha2[1] == '0') {
12063                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
12064                                NL80211_REGDOM_TYPE_WORLD))
12065                         goto nla_put_failure;
12066         } else if (request->alpha2[0] == '9' && request->alpha2[1] == '9') {
12067                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
12068                                NL80211_REGDOM_TYPE_CUSTOM_WORLD))
12069                         goto nla_put_failure;
12070         } else if ((request->alpha2[0] == '9' && request->alpha2[1] == '8') ||
12071                    request->intersect) {
12072                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
12073                                NL80211_REGDOM_TYPE_INTERSECTION))
12074                         goto nla_put_failure;
12075         } else {
12076                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
12077                                NL80211_REGDOM_TYPE_COUNTRY) ||
12078                     nla_put_string(msg, NL80211_ATTR_REG_ALPHA2,
12079                                    request->alpha2))
12080                         goto nla_put_failure;
12081         }
12082
12083         if (request->wiphy_idx != WIPHY_IDX_INVALID) {
12084                 struct wiphy *wiphy = wiphy_idx_to_wiphy(request->wiphy_idx);
12085
12086                 if (wiphy &&
12087                     nla_put_u32(msg, NL80211_ATTR_WIPHY, request->wiphy_idx))
12088                         goto nla_put_failure;
12089
12090                 if (wiphy &&
12091                     wiphy->regulatory_flags & REGULATORY_WIPHY_SELF_MANAGED &&
12092                     nla_put_flag(msg, NL80211_ATTR_WIPHY_SELF_MANAGED_REG))
12093                         goto nla_put_failure;
12094         }
12095
12096         return true;
12097
12098 nla_put_failure:
12099         return false;
12100 }
12101
12102 /*
12103  * This can happen on global regulatory changes or device specific settings
12104  * based on custom regulatory domains.
12105  */
12106 void nl80211_common_reg_change_event(enum nl80211_commands cmd_id,
12107                                      struct regulatory_request *request)
12108 {
12109         struct sk_buff *msg;
12110         void *hdr;
12111
12112         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
12113         if (!msg)
12114                 return;
12115
12116         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd_id);
12117         if (!hdr) {
12118                 nlmsg_free(msg);
12119                 return;
12120         }
12121
12122         if (nl80211_reg_change_event_fill(msg, request) == false)
12123                 goto nla_put_failure;
12124
12125         genlmsg_end(msg, hdr);
12126
12127         rcu_read_lock();
12128         genlmsg_multicast_allns(&nl80211_fam, msg, 0,
12129                                 NL80211_MCGRP_REGULATORY, GFP_ATOMIC);
12130         rcu_read_unlock();
12131
12132         return;
12133
12134 nla_put_failure:
12135         genlmsg_cancel(msg, hdr);
12136         nlmsg_free(msg);
12137 }
12138
12139 static void nl80211_send_mlme_event(struct cfg80211_registered_device *rdev,
12140                                     struct net_device *netdev,
12141                                     const u8 *buf, size_t len,
12142                                     enum nl80211_commands cmd, gfp_t gfp,
12143                                     int uapsd_queues)
12144 {
12145         struct sk_buff *msg;
12146         void *hdr;
12147
12148         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12149         if (!msg)
12150                 return;
12151
12152         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
12153         if (!hdr) {
12154                 nlmsg_free(msg);
12155                 return;
12156         }
12157
12158         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12159             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
12160             nla_put(msg, NL80211_ATTR_FRAME, len, buf))
12161                 goto nla_put_failure;
12162
12163         if (uapsd_queues >= 0) {
12164                 struct nlattr *nla_wmm =
12165                         nla_nest_start(msg, NL80211_ATTR_STA_WME);
12166                 if (!nla_wmm)
12167                         goto nla_put_failure;
12168
12169                 if (nla_put_u8(msg, NL80211_STA_WME_UAPSD_QUEUES,
12170                                uapsd_queues))
12171                         goto nla_put_failure;
12172
12173                 nla_nest_end(msg, nla_wmm);
12174         }
12175
12176         genlmsg_end(msg, hdr);
12177
12178         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12179                                 NL80211_MCGRP_MLME, gfp);
12180         return;
12181
12182  nla_put_failure:
12183         genlmsg_cancel(msg, hdr);
12184         nlmsg_free(msg);
12185 }
12186
12187 void nl80211_send_rx_auth(struct cfg80211_registered_device *rdev,
12188                           struct net_device *netdev, const u8 *buf,
12189                           size_t len, gfp_t gfp)
12190 {
12191         nl80211_send_mlme_event(rdev, netdev, buf, len,
12192                                 NL80211_CMD_AUTHENTICATE, gfp, -1);
12193 }
12194
12195 void nl80211_send_rx_assoc(struct cfg80211_registered_device *rdev,
12196                            struct net_device *netdev, const u8 *buf,
12197                            size_t len, gfp_t gfp, int uapsd_queues)
12198 {
12199         nl80211_send_mlme_event(rdev, netdev, buf, len,
12200                                 NL80211_CMD_ASSOCIATE, gfp, uapsd_queues);
12201 }
12202
12203 void nl80211_send_deauth(struct cfg80211_registered_device *rdev,
12204                          struct net_device *netdev, const u8 *buf,
12205                          size_t len, gfp_t gfp)
12206 {
12207         nl80211_send_mlme_event(rdev, netdev, buf, len,
12208                                 NL80211_CMD_DEAUTHENTICATE, gfp, -1);
12209 }
12210
12211 void nl80211_send_disassoc(struct cfg80211_registered_device *rdev,
12212                            struct net_device *netdev, const u8 *buf,
12213                            size_t len, gfp_t gfp)
12214 {
12215         nl80211_send_mlme_event(rdev, netdev, buf, len,
12216                                 NL80211_CMD_DISASSOCIATE, gfp, -1);
12217 }
12218
12219 void cfg80211_rx_unprot_mlme_mgmt(struct net_device *dev, const u8 *buf,
12220                                   size_t len)
12221 {
12222         struct wireless_dev *wdev = dev->ieee80211_ptr;
12223         struct wiphy *wiphy = wdev->wiphy;
12224         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
12225         const struct ieee80211_mgmt *mgmt = (void *)buf;
12226         u32 cmd;
12227
12228         if (WARN_ON(len < 2))
12229                 return;
12230
12231         if (ieee80211_is_deauth(mgmt->frame_control))
12232                 cmd = NL80211_CMD_UNPROT_DEAUTHENTICATE;
12233         else
12234                 cmd = NL80211_CMD_UNPROT_DISASSOCIATE;
12235
12236         trace_cfg80211_rx_unprot_mlme_mgmt(dev, buf, len);
12237         nl80211_send_mlme_event(rdev, dev, buf, len, cmd, GFP_ATOMIC, -1);
12238 }
12239 EXPORT_SYMBOL(cfg80211_rx_unprot_mlme_mgmt);
12240
12241 static void nl80211_send_mlme_timeout(struct cfg80211_registered_device *rdev,
12242                                       struct net_device *netdev, int cmd,
12243                                       const u8 *addr, gfp_t gfp)
12244 {
12245         struct sk_buff *msg;
12246         void *hdr;
12247
12248         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12249         if (!msg)
12250                 return;
12251
12252         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
12253         if (!hdr) {
12254                 nlmsg_free(msg);
12255                 return;
12256         }
12257
12258         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12259             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
12260             nla_put_flag(msg, NL80211_ATTR_TIMED_OUT) ||
12261             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr))
12262                 goto nla_put_failure;
12263
12264         genlmsg_end(msg, hdr);
12265
12266         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12267                                 NL80211_MCGRP_MLME, gfp);
12268         return;
12269
12270  nla_put_failure:
12271         genlmsg_cancel(msg, hdr);
12272         nlmsg_free(msg);
12273 }
12274
12275 void nl80211_send_auth_timeout(struct cfg80211_registered_device *rdev,
12276                                struct net_device *netdev, const u8 *addr,
12277                                gfp_t gfp)
12278 {
12279         nl80211_send_mlme_timeout(rdev, netdev, NL80211_CMD_AUTHENTICATE,
12280                                   addr, gfp);
12281 }
12282
12283 void nl80211_send_assoc_timeout(struct cfg80211_registered_device *rdev,
12284                                 struct net_device *netdev, const u8 *addr,
12285                                 gfp_t gfp)
12286 {
12287         nl80211_send_mlme_timeout(rdev, netdev, NL80211_CMD_ASSOCIATE,
12288                                   addr, gfp);
12289 }
12290
12291 void nl80211_send_connect_result(struct cfg80211_registered_device *rdev,
12292                                  struct net_device *netdev, const u8 *bssid,
12293                                  const u8 *req_ie, size_t req_ie_len,
12294                                  const u8 *resp_ie, size_t resp_ie_len,
12295                                  int status, gfp_t gfp)
12296 {
12297         struct sk_buff *msg;
12298         void *hdr;
12299
12300         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12301         if (!msg)
12302                 return;
12303
12304         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CONNECT);
12305         if (!hdr) {
12306                 nlmsg_free(msg);
12307                 return;
12308         }
12309
12310         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12311             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
12312             (bssid && nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid)) ||
12313             nla_put_u16(msg, NL80211_ATTR_STATUS_CODE,
12314                         status < 0 ? WLAN_STATUS_UNSPECIFIED_FAILURE :
12315                         status) ||
12316             (status < 0 && nla_put_flag(msg, NL80211_ATTR_TIMED_OUT)) ||
12317             (req_ie &&
12318              nla_put(msg, NL80211_ATTR_REQ_IE, req_ie_len, req_ie)) ||
12319             (resp_ie &&
12320              nla_put(msg, NL80211_ATTR_RESP_IE, resp_ie_len, resp_ie)))
12321                 goto nla_put_failure;
12322
12323         genlmsg_end(msg, hdr);
12324
12325         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12326                                 NL80211_MCGRP_MLME, gfp);
12327         return;
12328
12329  nla_put_failure:
12330         genlmsg_cancel(msg, hdr);
12331         nlmsg_free(msg);
12332 }
12333
12334 void nl80211_send_roamed(struct cfg80211_registered_device *rdev,
12335                          struct net_device *netdev, const u8 *bssid,
12336                          const u8 *req_ie, size_t req_ie_len,
12337                          const u8 *resp_ie, size_t resp_ie_len, gfp_t gfp)
12338 {
12339         struct sk_buff *msg;
12340         void *hdr;
12341
12342         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12343         if (!msg)
12344                 return;
12345
12346         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_ROAM);
12347         if (!hdr) {
12348                 nlmsg_free(msg);
12349                 return;
12350         }
12351
12352         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12353             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
12354             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid) ||
12355             (req_ie &&
12356              nla_put(msg, NL80211_ATTR_REQ_IE, req_ie_len, req_ie)) ||
12357             (resp_ie &&
12358              nla_put(msg, NL80211_ATTR_RESP_IE, resp_ie_len, resp_ie)))
12359                 goto nla_put_failure;
12360
12361         genlmsg_end(msg, hdr);
12362
12363         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12364                                 NL80211_MCGRP_MLME, gfp);
12365         return;
12366
12367  nla_put_failure:
12368         genlmsg_cancel(msg, hdr);
12369         nlmsg_free(msg);
12370 }
12371
12372 void nl80211_send_disconnected(struct cfg80211_registered_device *rdev,
12373                                struct net_device *netdev, u16 reason,
12374                                const u8 *ie, size_t ie_len, bool from_ap)
12375 {
12376         struct sk_buff *msg;
12377         void *hdr;
12378
12379         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
12380         if (!msg)
12381                 return;
12382
12383         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_DISCONNECT);
12384         if (!hdr) {
12385                 nlmsg_free(msg);
12386                 return;
12387         }
12388
12389         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12390             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
12391             (from_ap && reason &&
12392              nla_put_u16(msg, NL80211_ATTR_REASON_CODE, reason)) ||
12393             (from_ap &&
12394              nla_put_flag(msg, NL80211_ATTR_DISCONNECTED_BY_AP)) ||
12395             (ie && nla_put(msg, NL80211_ATTR_IE, ie_len, ie)))
12396                 goto nla_put_failure;
12397
12398         genlmsg_end(msg, hdr);
12399
12400         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12401                                 NL80211_MCGRP_MLME, GFP_KERNEL);
12402         return;
12403
12404  nla_put_failure:
12405         genlmsg_cancel(msg, hdr);
12406         nlmsg_free(msg);
12407 }
12408
12409 void nl80211_send_ibss_bssid(struct cfg80211_registered_device *rdev,
12410                              struct net_device *netdev, const u8 *bssid,
12411                              gfp_t gfp)
12412 {
12413         struct sk_buff *msg;
12414         void *hdr;
12415
12416         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12417         if (!msg)
12418                 return;
12419
12420         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_JOIN_IBSS);
12421         if (!hdr) {
12422                 nlmsg_free(msg);
12423                 return;
12424         }
12425
12426         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12427             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
12428             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid))
12429                 goto nla_put_failure;
12430
12431         genlmsg_end(msg, hdr);
12432
12433         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12434                                 NL80211_MCGRP_MLME, gfp);
12435         return;
12436
12437  nla_put_failure:
12438         genlmsg_cancel(msg, hdr);
12439         nlmsg_free(msg);
12440 }
12441
12442 void cfg80211_notify_new_peer_candidate(struct net_device *dev, const u8 *addr,
12443                                         const u8* ie, u8 ie_len, gfp_t gfp)
12444 {
12445         struct wireless_dev *wdev = dev->ieee80211_ptr;
12446         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
12447         struct sk_buff *msg;
12448         void *hdr;
12449
12450         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_MESH_POINT))
12451                 return;
12452
12453         trace_cfg80211_notify_new_peer_candidate(dev, addr);
12454
12455         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12456         if (!msg)
12457                 return;
12458
12459         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NEW_PEER_CANDIDATE);
12460         if (!hdr) {
12461                 nlmsg_free(msg);
12462                 return;
12463         }
12464
12465         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12466             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
12467             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr) ||
12468             (ie_len && ie &&
12469              nla_put(msg, NL80211_ATTR_IE, ie_len , ie)))
12470                 goto nla_put_failure;
12471
12472         genlmsg_end(msg, hdr);
12473
12474         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12475                                 NL80211_MCGRP_MLME, gfp);
12476         return;
12477
12478  nla_put_failure:
12479         genlmsg_cancel(msg, hdr);
12480         nlmsg_free(msg);
12481 }
12482 EXPORT_SYMBOL(cfg80211_notify_new_peer_candidate);
12483
12484 void nl80211_michael_mic_failure(struct cfg80211_registered_device *rdev,
12485                                  struct net_device *netdev, const u8 *addr,
12486                                  enum nl80211_key_type key_type, int key_id,
12487                                  const u8 *tsc, gfp_t gfp)
12488 {
12489         struct sk_buff *msg;
12490         void *hdr;
12491
12492         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12493         if (!msg)
12494                 return;
12495
12496         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_MICHAEL_MIC_FAILURE);
12497         if (!hdr) {
12498                 nlmsg_free(msg);
12499                 return;
12500         }
12501
12502         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12503             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
12504             (addr && nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr)) ||
12505             nla_put_u32(msg, NL80211_ATTR_KEY_TYPE, key_type) ||
12506             (key_id != -1 &&
12507              nla_put_u8(msg, NL80211_ATTR_KEY_IDX, key_id)) ||
12508             (tsc && nla_put(msg, NL80211_ATTR_KEY_SEQ, 6, tsc)))
12509                 goto nla_put_failure;
12510
12511         genlmsg_end(msg, hdr);
12512
12513         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12514                                 NL80211_MCGRP_MLME, gfp);
12515         return;
12516
12517  nla_put_failure:
12518         genlmsg_cancel(msg, hdr);
12519         nlmsg_free(msg);
12520 }
12521
12522 void nl80211_send_beacon_hint_event(struct wiphy *wiphy,
12523                                     struct ieee80211_channel *channel_before,
12524                                     struct ieee80211_channel *channel_after)
12525 {
12526         struct sk_buff *msg;
12527         void *hdr;
12528         struct nlattr *nl_freq;
12529
12530         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_ATOMIC);
12531         if (!msg)
12532                 return;
12533
12534         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_REG_BEACON_HINT);
12535         if (!hdr) {
12536                 nlmsg_free(msg);
12537                 return;
12538         }
12539
12540         /*
12541          * Since we are applying the beacon hint to a wiphy we know its
12542          * wiphy_idx is valid
12543          */
12544         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, get_wiphy_idx(wiphy)))
12545                 goto nla_put_failure;
12546
12547         /* Before */
12548         nl_freq = nla_nest_start(msg, NL80211_ATTR_FREQ_BEFORE);
12549         if (!nl_freq)
12550                 goto nla_put_failure;
12551         if (nl80211_msg_put_channel(msg, channel_before, false))
12552                 goto nla_put_failure;
12553         nla_nest_end(msg, nl_freq);
12554
12555         /* After */
12556         nl_freq = nla_nest_start(msg, NL80211_ATTR_FREQ_AFTER);
12557         if (!nl_freq)
12558                 goto nla_put_failure;
12559         if (nl80211_msg_put_channel(msg, channel_after, false))
12560                 goto nla_put_failure;
12561         nla_nest_end(msg, nl_freq);
12562
12563         genlmsg_end(msg, hdr);
12564
12565         rcu_read_lock();
12566         genlmsg_multicast_allns(&nl80211_fam, msg, 0,
12567                                 NL80211_MCGRP_REGULATORY, GFP_ATOMIC);
12568         rcu_read_unlock();
12569
12570         return;
12571
12572 nla_put_failure:
12573         genlmsg_cancel(msg, hdr);
12574         nlmsg_free(msg);
12575 }
12576
12577 static void nl80211_send_remain_on_chan_event(
12578         int cmd, struct cfg80211_registered_device *rdev,
12579         struct wireless_dev *wdev, u64 cookie,
12580         struct ieee80211_channel *chan,
12581         unsigned int duration, gfp_t gfp)
12582 {
12583         struct sk_buff *msg;
12584         void *hdr;
12585
12586         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12587         if (!msg)
12588                 return;
12589
12590         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
12591         if (!hdr) {
12592                 nlmsg_free(msg);
12593                 return;
12594         }
12595
12596         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12597             (wdev->netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
12598                                          wdev->netdev->ifindex)) ||
12599             nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
12600                               NL80211_ATTR_PAD) ||
12601             nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ, chan->center_freq) ||
12602             nla_put_u32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
12603                         NL80211_CHAN_NO_HT) ||
12604             nla_put_u64_64bit(msg, NL80211_ATTR_COOKIE, cookie,
12605                               NL80211_ATTR_PAD))
12606                 goto nla_put_failure;
12607
12608         if (cmd == NL80211_CMD_REMAIN_ON_CHANNEL &&
12609             nla_put_u32(msg, NL80211_ATTR_DURATION, duration))
12610                 goto nla_put_failure;
12611
12612         genlmsg_end(msg, hdr);
12613
12614         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12615                                 NL80211_MCGRP_MLME, gfp);
12616         return;
12617
12618  nla_put_failure:
12619         genlmsg_cancel(msg, hdr);
12620         nlmsg_free(msg);
12621 }
12622
12623 void cfg80211_ready_on_channel(struct wireless_dev *wdev, u64 cookie,
12624                                struct ieee80211_channel *chan,
12625                                unsigned int duration, gfp_t gfp)
12626 {
12627         struct wiphy *wiphy = wdev->wiphy;
12628         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
12629
12630         trace_cfg80211_ready_on_channel(wdev, cookie, chan, duration);
12631         nl80211_send_remain_on_chan_event(NL80211_CMD_REMAIN_ON_CHANNEL,
12632                                           rdev, wdev, cookie, chan,
12633                                           duration, gfp);
12634 }
12635 EXPORT_SYMBOL(cfg80211_ready_on_channel);
12636
12637 void cfg80211_remain_on_channel_expired(struct wireless_dev *wdev, u64 cookie,
12638                                         struct ieee80211_channel *chan,
12639                                         gfp_t gfp)
12640 {
12641         struct wiphy *wiphy = wdev->wiphy;
12642         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
12643
12644         trace_cfg80211_ready_on_channel_expired(wdev, cookie, chan);
12645         nl80211_send_remain_on_chan_event(NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL,
12646                                           rdev, wdev, cookie, chan, 0, gfp);
12647 }
12648 EXPORT_SYMBOL(cfg80211_remain_on_channel_expired);
12649
12650 void cfg80211_new_sta(struct net_device *dev, const u8 *mac_addr,
12651                       struct station_info *sinfo, gfp_t gfp)
12652 {
12653         struct wiphy *wiphy = dev->ieee80211_ptr->wiphy;
12654         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
12655         struct sk_buff *msg;
12656
12657         trace_cfg80211_new_sta(dev, mac_addr, sinfo);
12658
12659         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12660         if (!msg)
12661                 return;
12662
12663         if (nl80211_send_station(msg, NL80211_CMD_NEW_STATION, 0, 0, 0,
12664                                  rdev, dev, mac_addr, sinfo) < 0) {
12665                 nlmsg_free(msg);
12666                 return;
12667         }
12668
12669         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12670                                 NL80211_MCGRP_MLME, gfp);
12671 }
12672 EXPORT_SYMBOL(cfg80211_new_sta);
12673
12674 void cfg80211_del_sta_sinfo(struct net_device *dev, const u8 *mac_addr,
12675                             struct station_info *sinfo, gfp_t gfp)
12676 {
12677         struct wiphy *wiphy = dev->ieee80211_ptr->wiphy;
12678         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
12679         struct sk_buff *msg;
12680         struct station_info empty_sinfo = {};
12681
12682         if (!sinfo)
12683                 sinfo = &empty_sinfo;
12684
12685         trace_cfg80211_del_sta(dev, mac_addr);
12686
12687         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12688         if (!msg)
12689                 return;
12690
12691         if (nl80211_send_station(msg, NL80211_CMD_DEL_STATION, 0, 0, 0,
12692                                  rdev, dev, mac_addr, sinfo) < 0) {
12693                 nlmsg_free(msg);
12694                 return;
12695         }
12696
12697         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12698                                 NL80211_MCGRP_MLME, gfp);
12699 }
12700 EXPORT_SYMBOL(cfg80211_del_sta_sinfo);
12701
12702 void cfg80211_conn_failed(struct net_device *dev, const u8 *mac_addr,
12703                           enum nl80211_connect_failed_reason reason,
12704                           gfp_t gfp)
12705 {
12706         struct wiphy *wiphy = dev->ieee80211_ptr->wiphy;
12707         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
12708         struct sk_buff *msg;
12709         void *hdr;
12710
12711         msg = nlmsg_new(NLMSG_GOODSIZE, gfp);
12712         if (!msg)
12713                 return;
12714
12715         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CONN_FAILED);
12716         if (!hdr) {
12717                 nlmsg_free(msg);
12718                 return;
12719         }
12720
12721         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
12722             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr) ||
12723             nla_put_u32(msg, NL80211_ATTR_CONN_FAILED_REASON, reason))
12724                 goto nla_put_failure;
12725
12726         genlmsg_end(msg, hdr);
12727
12728         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12729                                 NL80211_MCGRP_MLME, gfp);
12730         return;
12731
12732  nla_put_failure:
12733         genlmsg_cancel(msg, hdr);
12734         nlmsg_free(msg);
12735 }
12736 EXPORT_SYMBOL(cfg80211_conn_failed);
12737
12738 static bool __nl80211_unexpected_frame(struct net_device *dev, u8 cmd,
12739                                        const u8 *addr, gfp_t gfp)
12740 {
12741         struct wireless_dev *wdev = dev->ieee80211_ptr;
12742         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
12743         struct sk_buff *msg;
12744         void *hdr;
12745         u32 nlportid = ACCESS_ONCE(wdev->ap_unexpected_nlportid);
12746
12747         if (!nlportid)
12748                 return false;
12749
12750         msg = nlmsg_new(100, gfp);
12751         if (!msg)
12752                 return true;
12753
12754         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
12755         if (!hdr) {
12756                 nlmsg_free(msg);
12757                 return true;
12758         }
12759
12760         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12761             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
12762             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr))
12763                 goto nla_put_failure;
12764
12765         genlmsg_end(msg, hdr);
12766         genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, nlportid);
12767         return true;
12768
12769  nla_put_failure:
12770         genlmsg_cancel(msg, hdr);
12771         nlmsg_free(msg);
12772         return true;
12773 }
12774
12775 bool cfg80211_rx_spurious_frame(struct net_device *dev,
12776                                 const u8 *addr, gfp_t gfp)
12777 {
12778         struct wireless_dev *wdev = dev->ieee80211_ptr;
12779         bool ret;
12780
12781         trace_cfg80211_rx_spurious_frame(dev, addr);
12782
12783         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_AP &&
12784                     wdev->iftype != NL80211_IFTYPE_P2P_GO)) {
12785                 trace_cfg80211_return_bool(false);
12786                 return false;
12787         }
12788         ret = __nl80211_unexpected_frame(dev, NL80211_CMD_UNEXPECTED_FRAME,
12789                                          addr, gfp);
12790         trace_cfg80211_return_bool(ret);
12791         return ret;
12792 }
12793 EXPORT_SYMBOL(cfg80211_rx_spurious_frame);
12794
12795 bool cfg80211_rx_unexpected_4addr_frame(struct net_device *dev,
12796                                         const u8 *addr, gfp_t gfp)
12797 {
12798         struct wireless_dev *wdev = dev->ieee80211_ptr;
12799         bool ret;
12800
12801         trace_cfg80211_rx_unexpected_4addr_frame(dev, addr);
12802
12803         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_AP &&
12804                     wdev->iftype != NL80211_IFTYPE_P2P_GO &&
12805                     wdev->iftype != NL80211_IFTYPE_AP_VLAN)) {
12806                 trace_cfg80211_return_bool(false);
12807                 return false;
12808         }
12809         ret = __nl80211_unexpected_frame(dev,
12810                                          NL80211_CMD_UNEXPECTED_4ADDR_FRAME,
12811                                          addr, gfp);
12812         trace_cfg80211_return_bool(ret);
12813         return ret;
12814 }
12815 EXPORT_SYMBOL(cfg80211_rx_unexpected_4addr_frame);
12816
12817 int nl80211_send_mgmt(struct cfg80211_registered_device *rdev,
12818                       struct wireless_dev *wdev, u32 nlportid,
12819                       int freq, int sig_dbm,
12820                       const u8 *buf, size_t len, u32 flags, gfp_t gfp)
12821 {
12822         struct net_device *netdev = wdev->netdev;
12823         struct sk_buff *msg;
12824         void *hdr;
12825
12826         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12827         if (!msg)
12828                 return -ENOMEM;
12829
12830         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FRAME);
12831         if (!hdr) {
12832                 nlmsg_free(msg);
12833                 return -ENOMEM;
12834         }
12835
12836         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12837             (netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
12838                                         netdev->ifindex)) ||
12839             nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
12840                               NL80211_ATTR_PAD) ||
12841             nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ, freq) ||
12842             (sig_dbm &&
12843              nla_put_u32(msg, NL80211_ATTR_RX_SIGNAL_DBM, sig_dbm)) ||
12844             nla_put(msg, NL80211_ATTR_FRAME, len, buf) ||
12845             (flags &&
12846              nla_put_u32(msg, NL80211_ATTR_RXMGMT_FLAGS, flags)))
12847                 goto nla_put_failure;
12848
12849         genlmsg_end(msg, hdr);
12850
12851         return genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, nlportid);
12852
12853  nla_put_failure:
12854         genlmsg_cancel(msg, hdr);
12855         nlmsg_free(msg);
12856         return -ENOBUFS;
12857 }
12858
12859 void cfg80211_mgmt_tx_status(struct wireless_dev *wdev, u64 cookie,
12860                              const u8 *buf, size_t len, bool ack, gfp_t gfp)
12861 {
12862         struct wiphy *wiphy = wdev->wiphy;
12863         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
12864         struct net_device *netdev = wdev->netdev;
12865         struct sk_buff *msg;
12866         void *hdr;
12867
12868         trace_cfg80211_mgmt_tx_status(wdev, cookie, ack);
12869
12870         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12871         if (!msg)
12872                 return;
12873
12874         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FRAME_TX_STATUS);
12875         if (!hdr) {
12876                 nlmsg_free(msg);
12877                 return;
12878         }
12879
12880         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12881             (netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
12882                                    netdev->ifindex)) ||
12883             nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
12884                               NL80211_ATTR_PAD) ||
12885             nla_put(msg, NL80211_ATTR_FRAME, len, buf) ||
12886             nla_put_u64_64bit(msg, NL80211_ATTR_COOKIE, cookie,
12887                               NL80211_ATTR_PAD) ||
12888             (ack && nla_put_flag(msg, NL80211_ATTR_ACK)))
12889                 goto nla_put_failure;
12890
12891         genlmsg_end(msg, hdr);
12892
12893         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12894                                 NL80211_MCGRP_MLME, gfp);
12895         return;
12896
12897  nla_put_failure:
12898         genlmsg_cancel(msg, hdr);
12899         nlmsg_free(msg);
12900 }
12901 EXPORT_SYMBOL(cfg80211_mgmt_tx_status);
12902
12903 static struct sk_buff *cfg80211_prepare_cqm(struct net_device *dev,
12904                                             const char *mac, gfp_t gfp)
12905 {
12906         struct wireless_dev *wdev = dev->ieee80211_ptr;
12907         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
12908         struct sk_buff *msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
12909         void **cb;
12910
12911         if (!msg)
12912                 return NULL;
12913
12914         cb = (void **)msg->cb;
12915
12916         cb[0] = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NOTIFY_CQM);
12917         if (!cb[0]) {
12918                 nlmsg_free(msg);
12919                 return NULL;
12920         }
12921
12922         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
12923             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex))
12924                 goto nla_put_failure;
12925
12926         if (mac && nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac))
12927                 goto nla_put_failure;
12928
12929         cb[1] = nla_nest_start(msg, NL80211_ATTR_CQM);
12930         if (!cb[1])
12931                 goto nla_put_failure;
12932
12933         cb[2] = rdev;
12934
12935         return msg;
12936  nla_put_failure:
12937         nlmsg_free(msg);
12938         return NULL;
12939 }
12940
12941 static void cfg80211_send_cqm(struct sk_buff *msg, gfp_t gfp)
12942 {
12943         void **cb = (void **)msg->cb;
12944         struct cfg80211_registered_device *rdev = cb[2];
12945
12946         nla_nest_end(msg, cb[1]);
12947         genlmsg_end(msg, cb[0]);
12948
12949         memset(msg->cb, 0, sizeof(msg->cb));
12950
12951         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
12952                                 NL80211_MCGRP_MLME, gfp);
12953 }
12954
12955 void cfg80211_cqm_rssi_notify(struct net_device *dev,
12956                               enum nl80211_cqm_rssi_threshold_event rssi_event,
12957                               gfp_t gfp)
12958 {
12959         struct sk_buff *msg;
12960
12961         trace_cfg80211_cqm_rssi_notify(dev, rssi_event);
12962
12963         if (WARN_ON(rssi_event != NL80211_CQM_RSSI_THRESHOLD_EVENT_LOW &&
12964                     rssi_event != NL80211_CQM_RSSI_THRESHOLD_EVENT_HIGH))
12965                 return;
12966
12967         msg = cfg80211_prepare_cqm(dev, NULL, gfp);
12968         if (!msg)
12969                 return;
12970
12971         if (nla_put_u32(msg, NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT,
12972                         rssi_event))
12973                 goto nla_put_failure;
12974
12975         cfg80211_send_cqm(msg, gfp);
12976
12977         return;
12978
12979  nla_put_failure:
12980         nlmsg_free(msg);
12981 }
12982 EXPORT_SYMBOL(cfg80211_cqm_rssi_notify);
12983
12984 void cfg80211_cqm_txe_notify(struct net_device *dev,
12985                              const u8 *peer, u32 num_packets,
12986                              u32 rate, u32 intvl, gfp_t gfp)
12987 {
12988         struct sk_buff *msg;
12989
12990         msg = cfg80211_prepare_cqm(dev, peer, gfp);
12991         if (!msg)
12992                 return;
12993
12994         if (nla_put_u32(msg, NL80211_ATTR_CQM_TXE_PKTS, num_packets))
12995                 goto nla_put_failure;
12996
12997         if (nla_put_u32(msg, NL80211_ATTR_CQM_TXE_RATE, rate))
12998                 goto nla_put_failure;
12999
13000         if (nla_put_u32(msg, NL80211_ATTR_CQM_TXE_INTVL, intvl))
13001                 goto nla_put_failure;
13002
13003         cfg80211_send_cqm(msg, gfp);
13004         return;
13005
13006  nla_put_failure:
13007         nlmsg_free(msg);
13008 }
13009 EXPORT_SYMBOL(cfg80211_cqm_txe_notify);
13010
13011 void cfg80211_cqm_pktloss_notify(struct net_device *dev,
13012                                  const u8 *peer, u32 num_packets, gfp_t gfp)
13013 {
13014         struct sk_buff *msg;
13015
13016         trace_cfg80211_cqm_pktloss_notify(dev, peer, num_packets);
13017
13018         msg = cfg80211_prepare_cqm(dev, peer, gfp);
13019         if (!msg)
13020                 return;
13021
13022         if (nla_put_u32(msg, NL80211_ATTR_CQM_PKT_LOSS_EVENT, num_packets))
13023                 goto nla_put_failure;
13024
13025         cfg80211_send_cqm(msg, gfp);
13026         return;
13027
13028  nla_put_failure:
13029         nlmsg_free(msg);
13030 }
13031 EXPORT_SYMBOL(cfg80211_cqm_pktloss_notify);
13032
13033 void cfg80211_cqm_beacon_loss_notify(struct net_device *dev, gfp_t gfp)
13034 {
13035         struct sk_buff *msg;
13036
13037         msg = cfg80211_prepare_cqm(dev, NULL, gfp);
13038         if (!msg)
13039                 return;
13040
13041         if (nla_put_flag(msg, NL80211_ATTR_CQM_BEACON_LOSS_EVENT))
13042                 goto nla_put_failure;
13043
13044         cfg80211_send_cqm(msg, gfp);
13045         return;
13046
13047  nla_put_failure:
13048         nlmsg_free(msg);
13049 }
13050 EXPORT_SYMBOL(cfg80211_cqm_beacon_loss_notify);
13051
13052 static void nl80211_gtk_rekey_notify(struct cfg80211_registered_device *rdev,
13053                                      struct net_device *netdev, const u8 *bssid,
13054                                      const u8 *replay_ctr, gfp_t gfp)
13055 {
13056         struct sk_buff *msg;
13057         struct nlattr *rekey_attr;
13058         void *hdr;
13059
13060         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
13061         if (!msg)
13062                 return;
13063
13064         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_SET_REKEY_OFFLOAD);
13065         if (!hdr) {
13066                 nlmsg_free(msg);
13067                 return;
13068         }
13069
13070         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
13071             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
13072             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid))
13073                 goto nla_put_failure;
13074
13075         rekey_attr = nla_nest_start(msg, NL80211_ATTR_REKEY_DATA);
13076         if (!rekey_attr)
13077                 goto nla_put_failure;
13078
13079         if (nla_put(msg, NL80211_REKEY_DATA_REPLAY_CTR,
13080                     NL80211_REPLAY_CTR_LEN, replay_ctr))
13081                 goto nla_put_failure;
13082
13083         nla_nest_end(msg, rekey_attr);
13084
13085         genlmsg_end(msg, hdr);
13086
13087         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
13088                                 NL80211_MCGRP_MLME, gfp);
13089         return;
13090
13091  nla_put_failure:
13092         genlmsg_cancel(msg, hdr);
13093         nlmsg_free(msg);
13094 }
13095
13096 void cfg80211_gtk_rekey_notify(struct net_device *dev, const u8 *bssid,
13097                                const u8 *replay_ctr, gfp_t gfp)
13098 {
13099         struct wireless_dev *wdev = dev->ieee80211_ptr;
13100         struct wiphy *wiphy = wdev->wiphy;
13101         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
13102
13103         trace_cfg80211_gtk_rekey_notify(dev, bssid);
13104         nl80211_gtk_rekey_notify(rdev, dev, bssid, replay_ctr, gfp);
13105 }
13106 EXPORT_SYMBOL(cfg80211_gtk_rekey_notify);
13107
13108 static void
13109 nl80211_pmksa_candidate_notify(struct cfg80211_registered_device *rdev,
13110                                struct net_device *netdev, int index,
13111                                const u8 *bssid, bool preauth, gfp_t gfp)
13112 {
13113         struct sk_buff *msg;
13114         struct nlattr *attr;
13115         void *hdr;
13116
13117         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
13118         if (!msg)
13119                 return;
13120
13121         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_PMKSA_CANDIDATE);
13122         if (!hdr) {
13123                 nlmsg_free(msg);
13124                 return;
13125         }
13126
13127         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
13128             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
13129                 goto nla_put_failure;
13130
13131         attr = nla_nest_start(msg, NL80211_ATTR_PMKSA_CANDIDATE);
13132         if (!attr)
13133                 goto nla_put_failure;
13134
13135         if (nla_put_u32(msg, NL80211_PMKSA_CANDIDATE_INDEX, index) ||
13136             nla_put(msg, NL80211_PMKSA_CANDIDATE_BSSID, ETH_ALEN, bssid) ||
13137             (preauth &&
13138              nla_put_flag(msg, NL80211_PMKSA_CANDIDATE_PREAUTH)))
13139                 goto nla_put_failure;
13140
13141         nla_nest_end(msg, attr);
13142
13143         genlmsg_end(msg, hdr);
13144
13145         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
13146                                 NL80211_MCGRP_MLME, gfp);
13147         return;
13148
13149  nla_put_failure:
13150         genlmsg_cancel(msg, hdr);
13151         nlmsg_free(msg);
13152 }
13153
13154 void cfg80211_pmksa_candidate_notify(struct net_device *dev, int index,
13155                                      const u8 *bssid, bool preauth, gfp_t gfp)
13156 {
13157         struct wireless_dev *wdev = dev->ieee80211_ptr;
13158         struct wiphy *wiphy = wdev->wiphy;
13159         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
13160
13161         trace_cfg80211_pmksa_candidate_notify(dev, index, bssid, preauth);
13162         nl80211_pmksa_candidate_notify(rdev, dev, index, bssid, preauth, gfp);
13163 }
13164 EXPORT_SYMBOL(cfg80211_pmksa_candidate_notify);
13165
13166 static void nl80211_ch_switch_notify(struct cfg80211_registered_device *rdev,
13167                                      struct net_device *netdev,
13168                                      struct cfg80211_chan_def *chandef,
13169                                      gfp_t gfp,
13170                                      enum nl80211_commands notif,
13171                                      u8 count)
13172 {
13173         struct sk_buff *msg;
13174         void *hdr;
13175
13176         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
13177         if (!msg)
13178                 return;
13179
13180         hdr = nl80211hdr_put(msg, 0, 0, 0, notif);
13181         if (!hdr) {
13182                 nlmsg_free(msg);
13183                 return;
13184         }
13185
13186         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
13187                 goto nla_put_failure;
13188
13189         if (nl80211_send_chandef(msg, chandef))
13190                 goto nla_put_failure;
13191
13192         if ((notif == NL80211_CMD_CH_SWITCH_STARTED_NOTIFY) &&
13193             (nla_put_u32(msg, NL80211_ATTR_CH_SWITCH_COUNT, count)))
13194                         goto nla_put_failure;
13195
13196         genlmsg_end(msg, hdr);
13197
13198         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
13199                                 NL80211_MCGRP_MLME, gfp);
13200         return;
13201
13202  nla_put_failure:
13203         genlmsg_cancel(msg, hdr);
13204         nlmsg_free(msg);
13205 }
13206
13207 void cfg80211_ch_switch_notify(struct net_device *dev,
13208                                struct cfg80211_chan_def *chandef)
13209 {
13210         struct wireless_dev *wdev = dev->ieee80211_ptr;
13211         struct wiphy *wiphy = wdev->wiphy;
13212         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
13213
13214         ASSERT_WDEV_LOCK(wdev);
13215
13216         trace_cfg80211_ch_switch_notify(dev, chandef);
13217
13218         wdev->chandef = *chandef;
13219         wdev->preset_chandef = *chandef;
13220         nl80211_ch_switch_notify(rdev, dev, chandef, GFP_KERNEL,
13221                                  NL80211_CMD_CH_SWITCH_NOTIFY, 0);
13222 }
13223 EXPORT_SYMBOL(cfg80211_ch_switch_notify);
13224
13225 void cfg80211_ch_switch_started_notify(struct net_device *dev,
13226                                        struct cfg80211_chan_def *chandef,
13227                                        u8 count)
13228 {
13229         struct wireless_dev *wdev = dev->ieee80211_ptr;
13230         struct wiphy *wiphy = wdev->wiphy;
13231         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
13232
13233         trace_cfg80211_ch_switch_started_notify(dev, chandef);
13234
13235         nl80211_ch_switch_notify(rdev, dev, chandef, GFP_KERNEL,
13236                                  NL80211_CMD_CH_SWITCH_STARTED_NOTIFY, count);
13237 }
13238 EXPORT_SYMBOL(cfg80211_ch_switch_started_notify);
13239
13240 void
13241 nl80211_radar_notify(struct cfg80211_registered_device *rdev,
13242                      const struct cfg80211_chan_def *chandef,
13243                      enum nl80211_radar_event event,
13244                      struct net_device *netdev, gfp_t gfp)
13245 {
13246         struct sk_buff *msg;
13247         void *hdr;
13248
13249         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
13250         if (!msg)
13251                 return;
13252
13253         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_RADAR_DETECT);
13254         if (!hdr) {
13255                 nlmsg_free(msg);
13256                 return;
13257         }
13258
13259         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx))
13260                 goto nla_put_failure;
13261
13262         /* NOP and radar events don't need a netdev parameter */
13263         if (netdev) {
13264                 struct wireless_dev *wdev = netdev->ieee80211_ptr;
13265
13266                 if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
13267                     nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
13268                                       NL80211_ATTR_PAD))
13269                         goto nla_put_failure;
13270         }
13271
13272         if (nla_put_u32(msg, NL80211_ATTR_RADAR_EVENT, event))
13273                 goto nla_put_failure;
13274
13275         if (nl80211_send_chandef(msg, chandef))
13276                 goto nla_put_failure;
13277
13278         genlmsg_end(msg, hdr);
13279
13280         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
13281                                 NL80211_MCGRP_MLME, gfp);
13282         return;
13283
13284  nla_put_failure:
13285         genlmsg_cancel(msg, hdr);
13286         nlmsg_free(msg);
13287 }
13288
13289 void cfg80211_probe_status(struct net_device *dev, const u8 *addr,
13290                            u64 cookie, bool acked, gfp_t gfp)
13291 {
13292         struct wireless_dev *wdev = dev->ieee80211_ptr;
13293         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
13294         struct sk_buff *msg;
13295         void *hdr;
13296
13297         trace_cfg80211_probe_status(dev, addr, cookie, acked);
13298
13299         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
13300
13301         if (!msg)
13302                 return;
13303
13304         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_PROBE_CLIENT);
13305         if (!hdr) {
13306                 nlmsg_free(msg);
13307                 return;
13308         }
13309
13310         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
13311             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
13312             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr) ||
13313             nla_put_u64_64bit(msg, NL80211_ATTR_COOKIE, cookie,
13314                               NL80211_ATTR_PAD) ||
13315             (acked && nla_put_flag(msg, NL80211_ATTR_ACK)))
13316                 goto nla_put_failure;
13317
13318         genlmsg_end(msg, hdr);
13319
13320         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
13321                                 NL80211_MCGRP_MLME, gfp);
13322         return;
13323
13324  nla_put_failure:
13325         genlmsg_cancel(msg, hdr);
13326         nlmsg_free(msg);
13327 }
13328 EXPORT_SYMBOL(cfg80211_probe_status);
13329
13330 void cfg80211_report_obss_beacon(struct wiphy *wiphy,
13331                                  const u8 *frame, size_t len,
13332                                  int freq, int sig_dbm)
13333 {
13334         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
13335         struct sk_buff *msg;
13336         void *hdr;
13337         struct cfg80211_beacon_registration *reg;
13338
13339         trace_cfg80211_report_obss_beacon(wiphy, frame, len, freq, sig_dbm);
13340
13341         spin_lock_bh(&rdev->beacon_registrations_lock);
13342         list_for_each_entry(reg, &rdev->beacon_registrations, list) {
13343                 msg = nlmsg_new(len + 100, GFP_ATOMIC);
13344                 if (!msg) {
13345                         spin_unlock_bh(&rdev->beacon_registrations_lock);
13346                         return;
13347                 }
13348
13349                 hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FRAME);
13350                 if (!hdr)
13351                         goto nla_put_failure;
13352
13353                 if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
13354                     (freq &&
13355                      nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ, freq)) ||
13356                     (sig_dbm &&
13357                      nla_put_u32(msg, NL80211_ATTR_RX_SIGNAL_DBM, sig_dbm)) ||
13358                     nla_put(msg, NL80211_ATTR_FRAME, len, frame))
13359                         goto nla_put_failure;
13360
13361                 genlmsg_end(msg, hdr);
13362
13363                 genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, reg->nlportid);
13364         }
13365         spin_unlock_bh(&rdev->beacon_registrations_lock);
13366         return;
13367
13368  nla_put_failure:
13369         spin_unlock_bh(&rdev->beacon_registrations_lock);
13370         if (hdr)
13371                 genlmsg_cancel(msg, hdr);
13372         nlmsg_free(msg);
13373 }
13374 EXPORT_SYMBOL(cfg80211_report_obss_beacon);
13375
13376 #ifdef CONFIG_PM
13377 static int cfg80211_net_detect_results(struct sk_buff *msg,
13378                                        struct cfg80211_wowlan_wakeup *wakeup)
13379 {
13380         struct cfg80211_wowlan_nd_info *nd = wakeup->net_detect;
13381         struct nlattr *nl_results, *nl_match, *nl_freqs;
13382         int i, j;
13383
13384         nl_results = nla_nest_start(
13385                 msg, NL80211_WOWLAN_TRIG_NET_DETECT_RESULTS);
13386         if (!nl_results)
13387                 return -EMSGSIZE;
13388
13389         for (i = 0; i < nd->n_matches; i++) {
13390                 struct cfg80211_wowlan_nd_match *match = nd->matches[i];
13391
13392                 nl_match = nla_nest_start(msg, i);
13393                 if (!nl_match)
13394                         break;
13395
13396                 /* The SSID attribute is optional in nl80211, but for
13397                  * simplicity reasons it's always present in the
13398                  * cfg80211 structure.  If a driver can't pass the
13399                  * SSID, that needs to be changed.  A zero length SSID
13400                  * is still a valid SSID (wildcard), so it cannot be
13401                  * used for this purpose.
13402                  */
13403                 if (nla_put(msg, NL80211_ATTR_SSID, match->ssid.ssid_len,
13404                             match->ssid.ssid)) {
13405                         nla_nest_cancel(msg, nl_match);
13406                         goto out;
13407                 }
13408
13409                 if (match->n_channels) {
13410                         nl_freqs = nla_nest_start(
13411                                 msg, NL80211_ATTR_SCAN_FREQUENCIES);
13412                         if (!nl_freqs) {
13413                                 nla_nest_cancel(msg, nl_match);
13414                                 goto out;
13415                         }
13416
13417                         for (j = 0; j < match->n_channels; j++) {
13418                                 if (nla_put_u32(msg, j, match->channels[j])) {
13419                                         nla_nest_cancel(msg, nl_freqs);
13420                                         nla_nest_cancel(msg, nl_match);
13421                                         goto out;
13422                                 }
13423                         }
13424
13425                         nla_nest_end(msg, nl_freqs);
13426                 }
13427
13428                 nla_nest_end(msg, nl_match);
13429         }
13430
13431 out:
13432         nla_nest_end(msg, nl_results);
13433         return 0;
13434 }
13435
13436 void cfg80211_report_wowlan_wakeup(struct wireless_dev *wdev,
13437                                    struct cfg80211_wowlan_wakeup *wakeup,
13438                                    gfp_t gfp)
13439 {
13440         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
13441         struct sk_buff *msg;
13442         void *hdr;
13443         int size = 200;
13444
13445         trace_cfg80211_report_wowlan_wakeup(wdev->wiphy, wdev, wakeup);
13446
13447         if (wakeup)
13448                 size += wakeup->packet_present_len;
13449
13450         msg = nlmsg_new(size, gfp);
13451         if (!msg)
13452                 return;
13453
13454         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_SET_WOWLAN);
13455         if (!hdr)
13456                 goto free_msg;
13457
13458         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
13459             nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
13460                               NL80211_ATTR_PAD))
13461                 goto free_msg;
13462
13463         if (wdev->netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
13464                                         wdev->netdev->ifindex))
13465                 goto free_msg;
13466
13467         if (wakeup) {
13468                 struct nlattr *reasons;
13469
13470                 reasons = nla_nest_start(msg, NL80211_ATTR_WOWLAN_TRIGGERS);
13471                 if (!reasons)
13472                         goto free_msg;
13473
13474                 if (wakeup->disconnect &&
13475                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_DISCONNECT))
13476                         goto free_msg;
13477                 if (wakeup->magic_pkt &&
13478                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_MAGIC_PKT))
13479                         goto free_msg;
13480                 if (wakeup->gtk_rekey_failure &&
13481                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE))
13482                         goto free_msg;
13483                 if (wakeup->eap_identity_req &&
13484                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST))
13485                         goto free_msg;
13486                 if (wakeup->four_way_handshake &&
13487                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE))
13488                         goto free_msg;
13489                 if (wakeup->rfkill_release &&
13490                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_RFKILL_RELEASE))
13491                         goto free_msg;
13492
13493                 if (wakeup->pattern_idx >= 0 &&
13494                     nla_put_u32(msg, NL80211_WOWLAN_TRIG_PKT_PATTERN,
13495                                 wakeup->pattern_idx))
13496                         goto free_msg;
13497
13498                 if (wakeup->tcp_match &&
13499                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_WAKEUP_TCP_MATCH))
13500                         goto free_msg;
13501
13502                 if (wakeup->tcp_connlost &&
13503                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_WAKEUP_TCP_CONNLOST))
13504                         goto free_msg;
13505
13506                 if (wakeup->tcp_nomoretokens &&
13507                     nla_put_flag(msg,
13508                                  NL80211_WOWLAN_TRIG_WAKEUP_TCP_NOMORETOKENS))
13509                         goto free_msg;
13510
13511                 if (wakeup->packet) {
13512                         u32 pkt_attr = NL80211_WOWLAN_TRIG_WAKEUP_PKT_80211;
13513                         u32 len_attr = NL80211_WOWLAN_TRIG_WAKEUP_PKT_80211_LEN;
13514
13515                         if (!wakeup->packet_80211) {
13516                                 pkt_attr =
13517                                         NL80211_WOWLAN_TRIG_WAKEUP_PKT_8023;
13518                                 len_attr =
13519                                         NL80211_WOWLAN_TRIG_WAKEUP_PKT_8023_LEN;
13520                         }
13521
13522                         if (wakeup->packet_len &&
13523                             nla_put_u32(msg, len_attr, wakeup->packet_len))
13524                                 goto free_msg;
13525
13526                         if (nla_put(msg, pkt_attr, wakeup->packet_present_len,
13527                                     wakeup->packet))
13528                                 goto free_msg;
13529                 }
13530
13531                 if (wakeup->net_detect &&
13532                     cfg80211_net_detect_results(msg, wakeup))
13533                                 goto free_msg;
13534
13535                 nla_nest_end(msg, reasons);
13536         }
13537
13538         genlmsg_end(msg, hdr);
13539
13540         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
13541                                 NL80211_MCGRP_MLME, gfp);
13542         return;
13543
13544  free_msg:
13545         nlmsg_free(msg);
13546 }
13547 EXPORT_SYMBOL(cfg80211_report_wowlan_wakeup);
13548 #endif
13549
13550 void cfg80211_tdls_oper_request(struct net_device *dev, const u8 *peer,
13551                                 enum nl80211_tdls_operation oper,
13552                                 u16 reason_code, gfp_t gfp)
13553 {
13554         struct wireless_dev *wdev = dev->ieee80211_ptr;
13555         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
13556         struct sk_buff *msg;
13557         void *hdr;
13558
13559         trace_cfg80211_tdls_oper_request(wdev->wiphy, dev, peer, oper,
13560                                          reason_code);
13561
13562         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
13563         if (!msg)
13564                 return;
13565
13566         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_TDLS_OPER);
13567         if (!hdr) {
13568                 nlmsg_free(msg);
13569                 return;
13570         }
13571
13572         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
13573             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
13574             nla_put_u8(msg, NL80211_ATTR_TDLS_OPERATION, oper) ||
13575             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, peer) ||
13576             (reason_code > 0 &&
13577              nla_put_u16(msg, NL80211_ATTR_REASON_CODE, reason_code)))
13578                 goto nla_put_failure;
13579
13580         genlmsg_end(msg, hdr);
13581
13582         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
13583                                 NL80211_MCGRP_MLME, gfp);
13584         return;
13585
13586  nla_put_failure:
13587         genlmsg_cancel(msg, hdr);
13588         nlmsg_free(msg);
13589 }
13590 EXPORT_SYMBOL(cfg80211_tdls_oper_request);
13591
13592 static int nl80211_netlink_notify(struct notifier_block * nb,
13593                                   unsigned long state,
13594                                   void *_notify)
13595 {
13596         struct netlink_notify *notify = _notify;
13597         struct cfg80211_registered_device *rdev;
13598         struct wireless_dev *wdev;
13599         struct cfg80211_beacon_registration *reg, *tmp;
13600
13601         if (state != NETLINK_URELEASE || notify->protocol != NETLINK_GENERIC)
13602                 return NOTIFY_DONE;
13603
13604         rcu_read_lock();
13605
13606         list_for_each_entry_rcu(rdev, &cfg80211_rdev_list, list) {
13607                 bool schedule_destroy_work = false;
13608                 bool schedule_scan_stop = false;
13609                 struct cfg80211_sched_scan_request *sched_scan_req =
13610                         rcu_dereference(rdev->sched_scan_req);
13611
13612                 if (sched_scan_req && notify->portid &&
13613                     sched_scan_req->owner_nlportid == notify->portid)
13614                         schedule_scan_stop = true;
13615
13616                 list_for_each_entry_rcu(wdev, &rdev->wiphy.wdev_list, list) {
13617                         cfg80211_mlme_unregister_socket(wdev, notify->portid);
13618
13619                         if (wdev->owner_nlportid == notify->portid)
13620                                 schedule_destroy_work = true;
13621                 }
13622
13623                 spin_lock_bh(&rdev->beacon_registrations_lock);
13624                 list_for_each_entry_safe(reg, tmp, &rdev->beacon_registrations,
13625                                          list) {
13626                         if (reg->nlportid == notify->portid) {
13627                                 list_del(&reg->list);
13628                                 kfree(reg);
13629                                 break;
13630                         }
13631                 }
13632                 spin_unlock_bh(&rdev->beacon_registrations_lock);
13633
13634                 if (schedule_destroy_work) {
13635                         struct cfg80211_iface_destroy *destroy;
13636
13637                         destroy = kzalloc(sizeof(*destroy), GFP_ATOMIC);
13638                         if (destroy) {
13639                                 destroy->nlportid = notify->portid;
13640                                 spin_lock(&rdev->destroy_list_lock);
13641                                 list_add(&destroy->list, &rdev->destroy_list);
13642                                 spin_unlock(&rdev->destroy_list_lock);
13643                                 schedule_work(&rdev->destroy_work);
13644                         }
13645                 } else if (schedule_scan_stop) {
13646                         sched_scan_req->owner_nlportid = 0;
13647
13648                         if (rdev->ops->sched_scan_stop &&
13649                             rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN)
13650                                 schedule_work(&rdev->sched_scan_stop_wk);
13651                 }
13652         }
13653
13654         rcu_read_unlock();
13655
13656         /*
13657          * It is possible that the user space process that is controlling the
13658          * indoor setting disappeared, so notify the regulatory core.
13659          */
13660         regulatory_netlink_notify(notify->portid);
13661         return NOTIFY_OK;
13662 }
13663
13664 static struct notifier_block nl80211_netlink_notifier = {
13665         .notifier_call = nl80211_netlink_notify,
13666 };
13667
13668 void cfg80211_ft_event(struct net_device *netdev,
13669                        struct cfg80211_ft_event_params *ft_event)
13670 {
13671         struct wiphy *wiphy = netdev->ieee80211_ptr->wiphy;
13672         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
13673         struct sk_buff *msg;
13674         void *hdr;
13675
13676         trace_cfg80211_ft_event(wiphy, netdev, ft_event);
13677
13678         if (!ft_event->target_ap)
13679                 return;
13680
13681         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
13682         if (!msg)
13683                 return;
13684
13685         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FT_EVENT);
13686         if (!hdr)
13687                 goto out;
13688
13689         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
13690             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
13691             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, ft_event->target_ap))
13692                 goto out;
13693
13694         if (ft_event->ies &&
13695             nla_put(msg, NL80211_ATTR_IE, ft_event->ies_len, ft_event->ies))
13696                 goto out;
13697         if (ft_event->ric_ies &&
13698             nla_put(msg, NL80211_ATTR_IE_RIC, ft_event->ric_ies_len,
13699                     ft_event->ric_ies))
13700                 goto out;
13701
13702         genlmsg_end(msg, hdr);
13703
13704         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(&rdev->wiphy), msg, 0,
13705                                 NL80211_MCGRP_MLME, GFP_KERNEL);
13706         return;
13707  out:
13708         nlmsg_free(msg);
13709 }
13710 EXPORT_SYMBOL(cfg80211_ft_event);
13711
13712 void cfg80211_crit_proto_stopped(struct wireless_dev *wdev, gfp_t gfp)
13713 {
13714         struct cfg80211_registered_device *rdev;
13715         struct sk_buff *msg;
13716         void *hdr;
13717         u32 nlportid;
13718
13719         rdev = wiphy_to_rdev(wdev->wiphy);
13720         if (!rdev->crit_proto_nlportid)
13721                 return;
13722
13723         nlportid = rdev->crit_proto_nlportid;
13724         rdev->crit_proto_nlportid = 0;
13725
13726         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
13727         if (!msg)
13728                 return;
13729
13730         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CRIT_PROTOCOL_STOP);
13731         if (!hdr)
13732                 goto nla_put_failure;
13733
13734         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
13735             nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
13736                               NL80211_ATTR_PAD))
13737                 goto nla_put_failure;
13738
13739         genlmsg_end(msg, hdr);
13740
13741         genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, nlportid);
13742         return;
13743
13744  nla_put_failure:
13745         if (hdr)
13746                 genlmsg_cancel(msg, hdr);
13747         nlmsg_free(msg);
13748 }
13749 EXPORT_SYMBOL(cfg80211_crit_proto_stopped);
13750
13751 void nl80211_send_ap_stopped(struct wireless_dev *wdev)
13752 {
13753         struct wiphy *wiphy = wdev->wiphy;
13754         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
13755         struct sk_buff *msg;
13756         void *hdr;
13757
13758         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
13759         if (!msg)
13760                 return;
13761
13762         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_STOP_AP);
13763         if (!hdr)
13764                 goto out;
13765
13766         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
13767             nla_put_u32(msg, NL80211_ATTR_IFINDEX, wdev->netdev->ifindex) ||
13768             nla_put_u64_64bit(msg, NL80211_ATTR_WDEV, wdev_id(wdev),
13769                               NL80211_ATTR_PAD))
13770                 goto out;
13771
13772         genlmsg_end(msg, hdr);
13773
13774         genlmsg_multicast_netns(&nl80211_fam, wiphy_net(wiphy), msg, 0,
13775                                 NL80211_MCGRP_MLME, GFP_KERNEL);
13776         return;
13777  out:
13778         nlmsg_free(msg);
13779 }
13780
13781 /* initialisation/exit functions */
13782
13783 int nl80211_init(void)
13784 {
13785         int err;
13786
13787         err = genl_register_family_with_ops_groups(&nl80211_fam, nl80211_ops,
13788                                                    nl80211_mcgrps);
13789         if (err)
13790                 return err;
13791
13792         err = netlink_register_notifier(&nl80211_netlink_notifier);
13793         if (err)
13794                 goto err_out;
13795
13796         return 0;
13797  err_out:
13798         genl_unregister_family(&nl80211_fam);
13799         return err;
13800 }
13801
13802 void nl80211_exit(void)
13803 {
13804         netlink_unregister_notifier(&nl80211_netlink_notifier);
13805         genl_unregister_family(&nl80211_fam);
13806 }