Merge branch 'master' of master.kernel.org:/pub/scm/linux/kernel/git/davem/net-2.6
[cascardo/linux.git] / net / wireless / nl80211.c
1 /*
2  * This is the new netlink-based wireless configuration interface.
3  *
4  * Copyright 2006-2009  Johannes Berg <johannes@sipsolutions.net>
5  */
6
7 #include <linux/if.h>
8 #include <linux/module.h>
9 #include <linux/err.h>
10 #include <linux/list.h>
11 #include <linux/if_ether.h>
12 #include <linux/ieee80211.h>
13 #include <linux/nl80211.h>
14 #include <linux/rtnetlink.h>
15 #include <linux/netlink.h>
16 #include <linux/etherdevice.h>
17 #include <net/net_namespace.h>
18 #include <net/genetlink.h>
19 #include <net/cfg80211.h>
20 #include <net/sock.h>
21 #include "core.h"
22 #include "nl80211.h"
23 #include "reg.h"
24
25 /* the netlink family */
26 static struct genl_family nl80211_fam = {
27         .id = GENL_ID_GENERATE, /* don't bother with a hardcoded ID */
28         .name = "nl80211",      /* have users key off the name instead */
29         .hdrsize = 0,           /* no private header */
30         .version = 1,           /* no particular meaning now */
31         .maxattr = NL80211_ATTR_MAX,
32         .netnsok = true,
33 };
34
35 /* internal helper: get rdev and dev */
36 static int get_rdev_dev_by_info_ifindex(struct genl_info *info,
37                                        struct cfg80211_registered_device **rdev,
38                                        struct net_device **dev)
39 {
40         struct nlattr **attrs = info->attrs;
41         int ifindex;
42
43         if (!attrs[NL80211_ATTR_IFINDEX])
44                 return -EINVAL;
45
46         ifindex = nla_get_u32(attrs[NL80211_ATTR_IFINDEX]);
47         *dev = dev_get_by_index(genl_info_net(info), ifindex);
48         if (!*dev)
49                 return -ENODEV;
50
51         *rdev = cfg80211_get_dev_from_ifindex(genl_info_net(info), ifindex);
52         if (IS_ERR(*rdev)) {
53                 dev_put(*dev);
54                 return PTR_ERR(*rdev);
55         }
56
57         return 0;
58 }
59
60 /* policy for the attributes */
61 static struct nla_policy nl80211_policy[NL80211_ATTR_MAX+1] __read_mostly = {
62         [NL80211_ATTR_WIPHY] = { .type = NLA_U32 },
63         [NL80211_ATTR_WIPHY_NAME] = { .type = NLA_NUL_STRING,
64                                       .len = 20-1 },
65         [NL80211_ATTR_WIPHY_TXQ_PARAMS] = { .type = NLA_NESTED },
66         [NL80211_ATTR_WIPHY_FREQ] = { .type = NLA_U32 },
67         [NL80211_ATTR_WIPHY_CHANNEL_TYPE] = { .type = NLA_U32 },
68         [NL80211_ATTR_WIPHY_RETRY_SHORT] = { .type = NLA_U8 },
69         [NL80211_ATTR_WIPHY_RETRY_LONG] = { .type = NLA_U8 },
70         [NL80211_ATTR_WIPHY_FRAG_THRESHOLD] = { .type = NLA_U32 },
71         [NL80211_ATTR_WIPHY_RTS_THRESHOLD] = { .type = NLA_U32 },
72         [NL80211_ATTR_WIPHY_COVERAGE_CLASS] = { .type = NLA_U8 },
73
74         [NL80211_ATTR_IFTYPE] = { .type = NLA_U32 },
75         [NL80211_ATTR_IFINDEX] = { .type = NLA_U32 },
76         [NL80211_ATTR_IFNAME] = { .type = NLA_NUL_STRING, .len = IFNAMSIZ-1 },
77
78         [NL80211_ATTR_MAC] = { .type = NLA_BINARY, .len = ETH_ALEN },
79         [NL80211_ATTR_PREV_BSSID] = { .type = NLA_BINARY, .len = ETH_ALEN },
80
81         [NL80211_ATTR_KEY] = { .type = NLA_NESTED, },
82         [NL80211_ATTR_KEY_DATA] = { .type = NLA_BINARY,
83                                     .len = WLAN_MAX_KEY_LEN },
84         [NL80211_ATTR_KEY_IDX] = { .type = NLA_U8 },
85         [NL80211_ATTR_KEY_CIPHER] = { .type = NLA_U32 },
86         [NL80211_ATTR_KEY_DEFAULT] = { .type = NLA_FLAG },
87         [NL80211_ATTR_KEY_SEQ] = { .type = NLA_BINARY, .len = 8 },
88
89         [NL80211_ATTR_BEACON_INTERVAL] = { .type = NLA_U32 },
90         [NL80211_ATTR_DTIM_PERIOD] = { .type = NLA_U32 },
91         [NL80211_ATTR_BEACON_HEAD] = { .type = NLA_BINARY,
92                                        .len = IEEE80211_MAX_DATA_LEN },
93         [NL80211_ATTR_BEACON_TAIL] = { .type = NLA_BINARY,
94                                        .len = IEEE80211_MAX_DATA_LEN },
95         [NL80211_ATTR_STA_AID] = { .type = NLA_U16 },
96         [NL80211_ATTR_STA_FLAGS] = { .type = NLA_NESTED },
97         [NL80211_ATTR_STA_LISTEN_INTERVAL] = { .type = NLA_U16 },
98         [NL80211_ATTR_STA_SUPPORTED_RATES] = { .type = NLA_BINARY,
99                                                .len = NL80211_MAX_SUPP_RATES },
100         [NL80211_ATTR_STA_PLINK_ACTION] = { .type = NLA_U8 },
101         [NL80211_ATTR_STA_VLAN] = { .type = NLA_U32 },
102         [NL80211_ATTR_MNTR_FLAGS] = { /* NLA_NESTED can't be empty */ },
103         [NL80211_ATTR_MESH_ID] = { .type = NLA_BINARY,
104                                 .len = IEEE80211_MAX_MESH_ID_LEN },
105         [NL80211_ATTR_MPATH_NEXT_HOP] = { .type = NLA_U32 },
106
107         [NL80211_ATTR_REG_ALPHA2] = { .type = NLA_STRING, .len = 2 },
108         [NL80211_ATTR_REG_RULES] = { .type = NLA_NESTED },
109
110         [NL80211_ATTR_BSS_CTS_PROT] = { .type = NLA_U8 },
111         [NL80211_ATTR_BSS_SHORT_PREAMBLE] = { .type = NLA_U8 },
112         [NL80211_ATTR_BSS_SHORT_SLOT_TIME] = { .type = NLA_U8 },
113         [NL80211_ATTR_BSS_BASIC_RATES] = { .type = NLA_BINARY,
114                                            .len = NL80211_MAX_SUPP_RATES },
115
116         [NL80211_ATTR_MESH_PARAMS] = { .type = NLA_NESTED },
117
118         [NL80211_ATTR_HT_CAPABILITY] = { .type = NLA_BINARY,
119                                          .len = NL80211_HT_CAPABILITY_LEN },
120
121         [NL80211_ATTR_MGMT_SUBTYPE] = { .type = NLA_U8 },
122         [NL80211_ATTR_IE] = { .type = NLA_BINARY,
123                               .len = IEEE80211_MAX_DATA_LEN },
124         [NL80211_ATTR_SCAN_FREQUENCIES] = { .type = NLA_NESTED },
125         [NL80211_ATTR_SCAN_SSIDS] = { .type = NLA_NESTED },
126
127         [NL80211_ATTR_SSID] = { .type = NLA_BINARY,
128                                 .len = IEEE80211_MAX_SSID_LEN },
129         [NL80211_ATTR_AUTH_TYPE] = { .type = NLA_U32 },
130         [NL80211_ATTR_REASON_CODE] = { .type = NLA_U16 },
131         [NL80211_ATTR_FREQ_FIXED] = { .type = NLA_FLAG },
132         [NL80211_ATTR_TIMED_OUT] = { .type = NLA_FLAG },
133         [NL80211_ATTR_USE_MFP] = { .type = NLA_U32 },
134         [NL80211_ATTR_STA_FLAGS2] = {
135                 .len = sizeof(struct nl80211_sta_flag_update),
136         },
137         [NL80211_ATTR_CONTROL_PORT] = { .type = NLA_FLAG },
138         [NL80211_ATTR_PRIVACY] = { .type = NLA_FLAG },
139         [NL80211_ATTR_CIPHER_SUITE_GROUP] = { .type = NLA_U32 },
140         [NL80211_ATTR_WPA_VERSIONS] = { .type = NLA_U32 },
141         [NL80211_ATTR_PID] = { .type = NLA_U32 },
142         [NL80211_ATTR_4ADDR] = { .type = NLA_U8 },
143         [NL80211_ATTR_PMKID] = { .type = NLA_BINARY,
144                                  .len = WLAN_PMKID_LEN },
145         [NL80211_ATTR_DURATION] = { .type = NLA_U32 },
146         [NL80211_ATTR_COOKIE] = { .type = NLA_U64 },
147         [NL80211_ATTR_TX_RATES] = { .type = NLA_NESTED },
148 };
149
150 /* policy for the attributes */
151 static struct nla_policy
152 nl80211_key_policy[NL80211_KEY_MAX + 1] __read_mostly = {
153         [NL80211_KEY_DATA] = { .type = NLA_BINARY, .len = WLAN_MAX_KEY_LEN },
154         [NL80211_KEY_IDX] = { .type = NLA_U8 },
155         [NL80211_KEY_CIPHER] = { .type = NLA_U32 },
156         [NL80211_KEY_SEQ] = { .type = NLA_BINARY, .len = 8 },
157         [NL80211_KEY_DEFAULT] = { .type = NLA_FLAG },
158         [NL80211_KEY_DEFAULT_MGMT] = { .type = NLA_FLAG },
159 };
160
161 /* ifidx get helper */
162 static int nl80211_get_ifidx(struct netlink_callback *cb)
163 {
164         int res;
165
166         res = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
167                           nl80211_fam.attrbuf, nl80211_fam.maxattr,
168                           nl80211_policy);
169         if (res)
170                 return res;
171
172         if (!nl80211_fam.attrbuf[NL80211_ATTR_IFINDEX])
173                 return -EINVAL;
174
175         res = nla_get_u32(nl80211_fam.attrbuf[NL80211_ATTR_IFINDEX]);
176         if (!res)
177                 return -EINVAL;
178         return res;
179 }
180
181 /* IE validation */
182 static bool is_valid_ie_attr(const struct nlattr *attr)
183 {
184         const u8 *pos;
185         int len;
186
187         if (!attr)
188                 return true;
189
190         pos = nla_data(attr);
191         len = nla_len(attr);
192
193         while (len) {
194                 u8 elemlen;
195
196                 if (len < 2)
197                         return false;
198                 len -= 2;
199
200                 elemlen = pos[1];
201                 if (elemlen > len)
202                         return false;
203
204                 len -= elemlen;
205                 pos += 2 + elemlen;
206         }
207
208         return true;
209 }
210
211 /* message building helper */
212 static inline void *nl80211hdr_put(struct sk_buff *skb, u32 pid, u32 seq,
213                                    int flags, u8 cmd)
214 {
215         /* since there is no private header just add the generic one */
216         return genlmsg_put(skb, pid, seq, &nl80211_fam, flags, cmd);
217 }
218
219 static int nl80211_msg_put_channel(struct sk_buff *msg,
220                                    struct ieee80211_channel *chan)
221 {
222         NLA_PUT_U32(msg, NL80211_FREQUENCY_ATTR_FREQ,
223                     chan->center_freq);
224
225         if (chan->flags & IEEE80211_CHAN_DISABLED)
226                 NLA_PUT_FLAG(msg, NL80211_FREQUENCY_ATTR_DISABLED);
227         if (chan->flags & IEEE80211_CHAN_PASSIVE_SCAN)
228                 NLA_PUT_FLAG(msg, NL80211_FREQUENCY_ATTR_PASSIVE_SCAN);
229         if (chan->flags & IEEE80211_CHAN_NO_IBSS)
230                 NLA_PUT_FLAG(msg, NL80211_FREQUENCY_ATTR_NO_IBSS);
231         if (chan->flags & IEEE80211_CHAN_RADAR)
232                 NLA_PUT_FLAG(msg, NL80211_FREQUENCY_ATTR_RADAR);
233
234         NLA_PUT_U32(msg, NL80211_FREQUENCY_ATTR_MAX_TX_POWER,
235                     DBM_TO_MBM(chan->max_power));
236
237         return 0;
238
239  nla_put_failure:
240         return -ENOBUFS;
241 }
242
243 /* netlink command implementations */
244
245 struct key_parse {
246         struct key_params p;
247         int idx;
248         bool def, defmgmt;
249 };
250
251 static int nl80211_parse_key_new(struct nlattr *key, struct key_parse *k)
252 {
253         struct nlattr *tb[NL80211_KEY_MAX + 1];
254         int err = nla_parse_nested(tb, NL80211_KEY_MAX, key,
255                                    nl80211_key_policy);
256         if (err)
257                 return err;
258
259         k->def = !!tb[NL80211_KEY_DEFAULT];
260         k->defmgmt = !!tb[NL80211_KEY_DEFAULT_MGMT];
261
262         if (tb[NL80211_KEY_IDX])
263                 k->idx = nla_get_u8(tb[NL80211_KEY_IDX]);
264
265         if (tb[NL80211_KEY_DATA]) {
266                 k->p.key = nla_data(tb[NL80211_KEY_DATA]);
267                 k->p.key_len = nla_len(tb[NL80211_KEY_DATA]);
268         }
269
270         if (tb[NL80211_KEY_SEQ]) {
271                 k->p.seq = nla_data(tb[NL80211_KEY_SEQ]);
272                 k->p.seq_len = nla_len(tb[NL80211_KEY_SEQ]);
273         }
274
275         if (tb[NL80211_KEY_CIPHER])
276                 k->p.cipher = nla_get_u32(tb[NL80211_KEY_CIPHER]);
277
278         return 0;
279 }
280
281 static int nl80211_parse_key_old(struct genl_info *info, struct key_parse *k)
282 {
283         if (info->attrs[NL80211_ATTR_KEY_DATA]) {
284                 k->p.key = nla_data(info->attrs[NL80211_ATTR_KEY_DATA]);
285                 k->p.key_len = nla_len(info->attrs[NL80211_ATTR_KEY_DATA]);
286         }
287
288         if (info->attrs[NL80211_ATTR_KEY_SEQ]) {
289                 k->p.seq = nla_data(info->attrs[NL80211_ATTR_KEY_SEQ]);
290                 k->p.seq_len = nla_len(info->attrs[NL80211_ATTR_KEY_SEQ]);
291         }
292
293         if (info->attrs[NL80211_ATTR_KEY_IDX])
294                 k->idx = nla_get_u8(info->attrs[NL80211_ATTR_KEY_IDX]);
295
296         if (info->attrs[NL80211_ATTR_KEY_CIPHER])
297                 k->p.cipher = nla_get_u32(info->attrs[NL80211_ATTR_KEY_CIPHER]);
298
299         k->def = !!info->attrs[NL80211_ATTR_KEY_DEFAULT];
300         k->defmgmt = !!info->attrs[NL80211_ATTR_KEY_DEFAULT_MGMT];
301
302         return 0;
303 }
304
305 static int nl80211_parse_key(struct genl_info *info, struct key_parse *k)
306 {
307         int err;
308
309         memset(k, 0, sizeof(*k));
310         k->idx = -1;
311
312         if (info->attrs[NL80211_ATTR_KEY])
313                 err = nl80211_parse_key_new(info->attrs[NL80211_ATTR_KEY], k);
314         else
315                 err = nl80211_parse_key_old(info, k);
316
317         if (err)
318                 return err;
319
320         if (k->def && k->defmgmt)
321                 return -EINVAL;
322
323         if (k->idx != -1) {
324                 if (k->defmgmt) {
325                         if (k->idx < 4 || k->idx > 5)
326                                 return -EINVAL;
327                 } else if (k->def) {
328                         if (k->idx < 0 || k->idx > 3)
329                                 return -EINVAL;
330                 } else {
331                         if (k->idx < 0 || k->idx > 5)
332                                 return -EINVAL;
333                 }
334         }
335
336         return 0;
337 }
338
339 static struct cfg80211_cached_keys *
340 nl80211_parse_connkeys(struct cfg80211_registered_device *rdev,
341                        struct nlattr *keys)
342 {
343         struct key_parse parse;
344         struct nlattr *key;
345         struct cfg80211_cached_keys *result;
346         int rem, err, def = 0;
347
348         result = kzalloc(sizeof(*result), GFP_KERNEL);
349         if (!result)
350                 return ERR_PTR(-ENOMEM);
351
352         result->def = -1;
353         result->defmgmt = -1;
354
355         nla_for_each_nested(key, keys, rem) {
356                 memset(&parse, 0, sizeof(parse));
357                 parse.idx = -1;
358
359                 err = nl80211_parse_key_new(key, &parse);
360                 if (err)
361                         goto error;
362                 err = -EINVAL;
363                 if (!parse.p.key)
364                         goto error;
365                 if (parse.idx < 0 || parse.idx > 4)
366                         goto error;
367                 if (parse.def) {
368                         if (def)
369                                 goto error;
370                         def = 1;
371                         result->def = parse.idx;
372                 } else if (parse.defmgmt)
373                         goto error;
374                 err = cfg80211_validate_key_settings(rdev, &parse.p,
375                                                      parse.idx, NULL);
376                 if (err)
377                         goto error;
378                 result->params[parse.idx].cipher = parse.p.cipher;
379                 result->params[parse.idx].key_len = parse.p.key_len;
380                 result->params[parse.idx].key = result->data[parse.idx];
381                 memcpy(result->data[parse.idx], parse.p.key, parse.p.key_len);
382         }
383
384         return result;
385  error:
386         kfree(result);
387         return ERR_PTR(err);
388 }
389
390 static int nl80211_key_allowed(struct wireless_dev *wdev)
391 {
392         ASSERT_WDEV_LOCK(wdev);
393
394         if (!netif_running(wdev->netdev))
395                 return -ENETDOWN;
396
397         switch (wdev->iftype) {
398         case NL80211_IFTYPE_AP:
399         case NL80211_IFTYPE_AP_VLAN:
400                 break;
401         case NL80211_IFTYPE_ADHOC:
402                 if (!wdev->current_bss)
403                         return -ENOLINK;
404                 break;
405         case NL80211_IFTYPE_STATION:
406                 if (wdev->sme_state != CFG80211_SME_CONNECTED)
407                         return -ENOLINK;
408                 break;
409         default:
410                 return -EINVAL;
411         }
412
413         return 0;
414 }
415
416 static int nl80211_send_wiphy(struct sk_buff *msg, u32 pid, u32 seq, int flags,
417                               struct cfg80211_registered_device *dev)
418 {
419         void *hdr;
420         struct nlattr *nl_bands, *nl_band;
421         struct nlattr *nl_freqs, *nl_freq;
422         struct nlattr *nl_rates, *nl_rate;
423         struct nlattr *nl_modes;
424         struct nlattr *nl_cmds;
425         enum ieee80211_band band;
426         struct ieee80211_channel *chan;
427         struct ieee80211_rate *rate;
428         int i;
429         u16 ifmodes = dev->wiphy.interface_modes;
430
431         hdr = nl80211hdr_put(msg, pid, seq, flags, NL80211_CMD_NEW_WIPHY);
432         if (!hdr)
433                 return -1;
434
435         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, dev->wiphy_idx);
436         NLA_PUT_STRING(msg, NL80211_ATTR_WIPHY_NAME, wiphy_name(&dev->wiphy));
437
438         NLA_PUT_U32(msg, NL80211_ATTR_GENERATION,
439                     cfg80211_rdev_list_generation);
440
441         NLA_PUT_U8(msg, NL80211_ATTR_WIPHY_RETRY_SHORT,
442                    dev->wiphy.retry_short);
443         NLA_PUT_U8(msg, NL80211_ATTR_WIPHY_RETRY_LONG,
444                    dev->wiphy.retry_long);
445         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FRAG_THRESHOLD,
446                     dev->wiphy.frag_threshold);
447         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_RTS_THRESHOLD,
448                     dev->wiphy.rts_threshold);
449         NLA_PUT_U8(msg, NL80211_ATTR_WIPHY_COVERAGE_CLASS,
450                     dev->wiphy.coverage_class);
451
452         NLA_PUT_U8(msg, NL80211_ATTR_MAX_NUM_SCAN_SSIDS,
453                    dev->wiphy.max_scan_ssids);
454         NLA_PUT_U16(msg, NL80211_ATTR_MAX_SCAN_IE_LEN,
455                     dev->wiphy.max_scan_ie_len);
456
457         NLA_PUT(msg, NL80211_ATTR_CIPHER_SUITES,
458                 sizeof(u32) * dev->wiphy.n_cipher_suites,
459                 dev->wiphy.cipher_suites);
460
461         NLA_PUT_U8(msg, NL80211_ATTR_MAX_NUM_PMKIDS,
462                    dev->wiphy.max_num_pmkids);
463
464         nl_modes = nla_nest_start(msg, NL80211_ATTR_SUPPORTED_IFTYPES);
465         if (!nl_modes)
466                 goto nla_put_failure;
467
468         i = 0;
469         while (ifmodes) {
470                 if (ifmodes & 1)
471                         NLA_PUT_FLAG(msg, i);
472                 ifmodes >>= 1;
473                 i++;
474         }
475
476         nla_nest_end(msg, nl_modes);
477
478         nl_bands = nla_nest_start(msg, NL80211_ATTR_WIPHY_BANDS);
479         if (!nl_bands)
480                 goto nla_put_failure;
481
482         for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
483                 if (!dev->wiphy.bands[band])
484                         continue;
485
486                 nl_band = nla_nest_start(msg, band);
487                 if (!nl_band)
488                         goto nla_put_failure;
489
490                 /* add HT info */
491                 if (dev->wiphy.bands[band]->ht_cap.ht_supported) {
492                         NLA_PUT(msg, NL80211_BAND_ATTR_HT_MCS_SET,
493                                 sizeof(dev->wiphy.bands[band]->ht_cap.mcs),
494                                 &dev->wiphy.bands[band]->ht_cap.mcs);
495                         NLA_PUT_U16(msg, NL80211_BAND_ATTR_HT_CAPA,
496                                 dev->wiphy.bands[band]->ht_cap.cap);
497                         NLA_PUT_U8(msg, NL80211_BAND_ATTR_HT_AMPDU_FACTOR,
498                                 dev->wiphy.bands[band]->ht_cap.ampdu_factor);
499                         NLA_PUT_U8(msg, NL80211_BAND_ATTR_HT_AMPDU_DENSITY,
500                                 dev->wiphy.bands[band]->ht_cap.ampdu_density);
501                 }
502
503                 /* add frequencies */
504                 nl_freqs = nla_nest_start(msg, NL80211_BAND_ATTR_FREQS);
505                 if (!nl_freqs)
506                         goto nla_put_failure;
507
508                 for (i = 0; i < dev->wiphy.bands[band]->n_channels; i++) {
509                         nl_freq = nla_nest_start(msg, i);
510                         if (!nl_freq)
511                                 goto nla_put_failure;
512
513                         chan = &dev->wiphy.bands[band]->channels[i];
514
515                         if (nl80211_msg_put_channel(msg, chan))
516                                 goto nla_put_failure;
517
518                         nla_nest_end(msg, nl_freq);
519                 }
520
521                 nla_nest_end(msg, nl_freqs);
522
523                 /* add bitrates */
524                 nl_rates = nla_nest_start(msg, NL80211_BAND_ATTR_RATES);
525                 if (!nl_rates)
526                         goto nla_put_failure;
527
528                 for (i = 0; i < dev->wiphy.bands[band]->n_bitrates; i++) {
529                         nl_rate = nla_nest_start(msg, i);
530                         if (!nl_rate)
531                                 goto nla_put_failure;
532
533                         rate = &dev->wiphy.bands[band]->bitrates[i];
534                         NLA_PUT_U32(msg, NL80211_BITRATE_ATTR_RATE,
535                                     rate->bitrate);
536                         if (rate->flags & IEEE80211_RATE_SHORT_PREAMBLE)
537                                 NLA_PUT_FLAG(msg,
538                                         NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE);
539
540                         nla_nest_end(msg, nl_rate);
541                 }
542
543                 nla_nest_end(msg, nl_rates);
544
545                 nla_nest_end(msg, nl_band);
546         }
547         nla_nest_end(msg, nl_bands);
548
549         nl_cmds = nla_nest_start(msg, NL80211_ATTR_SUPPORTED_COMMANDS);
550         if (!nl_cmds)
551                 goto nla_put_failure;
552
553         i = 0;
554 #define CMD(op, n)                                              \
555          do {                                                   \
556                 if (dev->ops->op) {                             \
557                         i++;                                    \
558                         NLA_PUT_U32(msg, i, NL80211_CMD_ ## n); \
559                 }                                               \
560         } while (0)
561
562         CMD(add_virtual_intf, NEW_INTERFACE);
563         CMD(change_virtual_intf, SET_INTERFACE);
564         CMD(add_key, NEW_KEY);
565         CMD(add_beacon, NEW_BEACON);
566         CMD(add_station, NEW_STATION);
567         CMD(add_mpath, NEW_MPATH);
568         CMD(set_mesh_params, SET_MESH_PARAMS);
569         CMD(change_bss, SET_BSS);
570         CMD(auth, AUTHENTICATE);
571         CMD(assoc, ASSOCIATE);
572         CMD(deauth, DEAUTHENTICATE);
573         CMD(disassoc, DISASSOCIATE);
574         CMD(join_ibss, JOIN_IBSS);
575         CMD(set_pmksa, SET_PMKSA);
576         CMD(del_pmksa, DEL_PMKSA);
577         CMD(flush_pmksa, FLUSH_PMKSA);
578         CMD(remain_on_channel, REMAIN_ON_CHANNEL);
579         CMD(set_bitrate_mask, SET_TX_BITRATE_MASK);
580         if (dev->wiphy.flags & WIPHY_FLAG_NETNS_OK) {
581                 i++;
582                 NLA_PUT_U32(msg, i, NL80211_CMD_SET_WIPHY_NETNS);
583         }
584
585 #undef CMD
586
587         if (dev->ops->connect || dev->ops->auth) {
588                 i++;
589                 NLA_PUT_U32(msg, i, NL80211_CMD_CONNECT);
590         }
591
592         if (dev->ops->disconnect || dev->ops->deauth) {
593                 i++;
594                 NLA_PUT_U32(msg, i, NL80211_CMD_DISCONNECT);
595         }
596
597         nla_nest_end(msg, nl_cmds);
598
599         return genlmsg_end(msg, hdr);
600
601  nla_put_failure:
602         genlmsg_cancel(msg, hdr);
603         return -EMSGSIZE;
604 }
605
606 static int nl80211_dump_wiphy(struct sk_buff *skb, struct netlink_callback *cb)
607 {
608         int idx = 0;
609         int start = cb->args[0];
610         struct cfg80211_registered_device *dev;
611
612         mutex_lock(&cfg80211_mutex);
613         list_for_each_entry(dev, &cfg80211_rdev_list, list) {
614                 if (!net_eq(wiphy_net(&dev->wiphy), sock_net(skb->sk)))
615                         continue;
616                 if (++idx <= start)
617                         continue;
618                 if (nl80211_send_wiphy(skb, NETLINK_CB(cb->skb).pid,
619                                        cb->nlh->nlmsg_seq, NLM_F_MULTI,
620                                        dev) < 0) {
621                         idx--;
622                         break;
623                 }
624         }
625         mutex_unlock(&cfg80211_mutex);
626
627         cb->args[0] = idx;
628
629         return skb->len;
630 }
631
632 static int nl80211_get_wiphy(struct sk_buff *skb, struct genl_info *info)
633 {
634         struct sk_buff *msg;
635         struct cfg80211_registered_device *dev;
636
637         dev = cfg80211_get_dev_from_info(info);
638         if (IS_ERR(dev))
639                 return PTR_ERR(dev);
640
641         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
642         if (!msg)
643                 goto out_err;
644
645         if (nl80211_send_wiphy(msg, info->snd_pid, info->snd_seq, 0, dev) < 0)
646                 goto out_free;
647
648         cfg80211_unlock_rdev(dev);
649
650         return genlmsg_reply(msg, info);
651
652  out_free:
653         nlmsg_free(msg);
654  out_err:
655         cfg80211_unlock_rdev(dev);
656         return -ENOBUFS;
657 }
658
659 static const struct nla_policy txq_params_policy[NL80211_TXQ_ATTR_MAX + 1] = {
660         [NL80211_TXQ_ATTR_QUEUE]                = { .type = NLA_U8 },
661         [NL80211_TXQ_ATTR_TXOP]                 = { .type = NLA_U16 },
662         [NL80211_TXQ_ATTR_CWMIN]                = { .type = NLA_U16 },
663         [NL80211_TXQ_ATTR_CWMAX]                = { .type = NLA_U16 },
664         [NL80211_TXQ_ATTR_AIFS]                 = { .type = NLA_U8 },
665 };
666
667 static int parse_txq_params(struct nlattr *tb[],
668                             struct ieee80211_txq_params *txq_params)
669 {
670         if (!tb[NL80211_TXQ_ATTR_QUEUE] || !tb[NL80211_TXQ_ATTR_TXOP] ||
671             !tb[NL80211_TXQ_ATTR_CWMIN] || !tb[NL80211_TXQ_ATTR_CWMAX] ||
672             !tb[NL80211_TXQ_ATTR_AIFS])
673                 return -EINVAL;
674
675         txq_params->queue = nla_get_u8(tb[NL80211_TXQ_ATTR_QUEUE]);
676         txq_params->txop = nla_get_u16(tb[NL80211_TXQ_ATTR_TXOP]);
677         txq_params->cwmin = nla_get_u16(tb[NL80211_TXQ_ATTR_CWMIN]);
678         txq_params->cwmax = nla_get_u16(tb[NL80211_TXQ_ATTR_CWMAX]);
679         txq_params->aifs = nla_get_u8(tb[NL80211_TXQ_ATTR_AIFS]);
680
681         return 0;
682 }
683
684 static int nl80211_set_wiphy(struct sk_buff *skb, struct genl_info *info)
685 {
686         struct cfg80211_registered_device *rdev;
687         int result = 0, rem_txq_params = 0;
688         struct nlattr *nl_txq_params;
689         u32 changed;
690         u8 retry_short = 0, retry_long = 0;
691         u32 frag_threshold = 0, rts_threshold = 0;
692         u8 coverage_class = 0;
693
694         rtnl_lock();
695
696         mutex_lock(&cfg80211_mutex);
697
698         rdev = __cfg80211_rdev_from_info(info);
699         if (IS_ERR(rdev)) {
700                 mutex_unlock(&cfg80211_mutex);
701                 result = PTR_ERR(rdev);
702                 goto unlock;
703         }
704
705         mutex_lock(&rdev->mtx);
706
707         if (info->attrs[NL80211_ATTR_WIPHY_NAME])
708                 result = cfg80211_dev_rename(
709                         rdev, nla_data(info->attrs[NL80211_ATTR_WIPHY_NAME]));
710
711         mutex_unlock(&cfg80211_mutex);
712
713         if (result)
714                 goto bad_res;
715
716         if (info->attrs[NL80211_ATTR_WIPHY_TXQ_PARAMS]) {
717                 struct ieee80211_txq_params txq_params;
718                 struct nlattr *tb[NL80211_TXQ_ATTR_MAX + 1];
719
720                 if (!rdev->ops->set_txq_params) {
721                         result = -EOPNOTSUPP;
722                         goto bad_res;
723                 }
724
725                 nla_for_each_nested(nl_txq_params,
726                                     info->attrs[NL80211_ATTR_WIPHY_TXQ_PARAMS],
727                                     rem_txq_params) {
728                         nla_parse(tb, NL80211_TXQ_ATTR_MAX,
729                                   nla_data(nl_txq_params),
730                                   nla_len(nl_txq_params),
731                                   txq_params_policy);
732                         result = parse_txq_params(tb, &txq_params);
733                         if (result)
734                                 goto bad_res;
735
736                         result = rdev->ops->set_txq_params(&rdev->wiphy,
737                                                            &txq_params);
738                         if (result)
739                                 goto bad_res;
740                 }
741         }
742
743         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
744                 enum nl80211_channel_type channel_type = NL80211_CHAN_NO_HT;
745                 u32 freq;
746
747                 result = -EINVAL;
748
749                 if (info->attrs[NL80211_ATTR_WIPHY_CHANNEL_TYPE]) {
750                         channel_type = nla_get_u32(info->attrs[
751                                            NL80211_ATTR_WIPHY_CHANNEL_TYPE]);
752                         if (channel_type != NL80211_CHAN_NO_HT &&
753                             channel_type != NL80211_CHAN_HT20 &&
754                             channel_type != NL80211_CHAN_HT40PLUS &&
755                             channel_type != NL80211_CHAN_HT40MINUS)
756                                 goto bad_res;
757                 }
758
759                 freq = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]);
760
761                 mutex_lock(&rdev->devlist_mtx);
762                 result = rdev_set_freq(rdev, NULL, freq, channel_type);
763                 mutex_unlock(&rdev->devlist_mtx);
764                 if (result)
765                         goto bad_res;
766         }
767
768         changed = 0;
769
770         if (info->attrs[NL80211_ATTR_WIPHY_RETRY_SHORT]) {
771                 retry_short = nla_get_u8(
772                         info->attrs[NL80211_ATTR_WIPHY_RETRY_SHORT]);
773                 if (retry_short == 0) {
774                         result = -EINVAL;
775                         goto bad_res;
776                 }
777                 changed |= WIPHY_PARAM_RETRY_SHORT;
778         }
779
780         if (info->attrs[NL80211_ATTR_WIPHY_RETRY_LONG]) {
781                 retry_long = nla_get_u8(
782                         info->attrs[NL80211_ATTR_WIPHY_RETRY_LONG]);
783                 if (retry_long == 0) {
784                         result = -EINVAL;
785                         goto bad_res;
786                 }
787                 changed |= WIPHY_PARAM_RETRY_LONG;
788         }
789
790         if (info->attrs[NL80211_ATTR_WIPHY_FRAG_THRESHOLD]) {
791                 frag_threshold = nla_get_u32(
792                         info->attrs[NL80211_ATTR_WIPHY_FRAG_THRESHOLD]);
793                 if (frag_threshold < 256) {
794                         result = -EINVAL;
795                         goto bad_res;
796                 }
797                 if (frag_threshold != (u32) -1) {
798                         /*
799                          * Fragments (apart from the last one) are required to
800                          * have even length. Make the fragmentation code
801                          * simpler by stripping LSB should someone try to use
802                          * odd threshold value.
803                          */
804                         frag_threshold &= ~0x1;
805                 }
806                 changed |= WIPHY_PARAM_FRAG_THRESHOLD;
807         }
808
809         if (info->attrs[NL80211_ATTR_WIPHY_RTS_THRESHOLD]) {
810                 rts_threshold = nla_get_u32(
811                         info->attrs[NL80211_ATTR_WIPHY_RTS_THRESHOLD]);
812                 changed |= WIPHY_PARAM_RTS_THRESHOLD;
813         }
814
815         if (info->attrs[NL80211_ATTR_WIPHY_COVERAGE_CLASS]) {
816                 coverage_class = nla_get_u8(
817                         info->attrs[NL80211_ATTR_WIPHY_COVERAGE_CLASS]);
818                 changed |= WIPHY_PARAM_COVERAGE_CLASS;
819         }
820
821         if (changed) {
822                 u8 old_retry_short, old_retry_long;
823                 u32 old_frag_threshold, old_rts_threshold;
824                 u8 old_coverage_class;
825
826                 if (!rdev->ops->set_wiphy_params) {
827                         result = -EOPNOTSUPP;
828                         goto bad_res;
829                 }
830
831                 old_retry_short = rdev->wiphy.retry_short;
832                 old_retry_long = rdev->wiphy.retry_long;
833                 old_frag_threshold = rdev->wiphy.frag_threshold;
834                 old_rts_threshold = rdev->wiphy.rts_threshold;
835                 old_coverage_class = rdev->wiphy.coverage_class;
836
837                 if (changed & WIPHY_PARAM_RETRY_SHORT)
838                         rdev->wiphy.retry_short = retry_short;
839                 if (changed & WIPHY_PARAM_RETRY_LONG)
840                         rdev->wiphy.retry_long = retry_long;
841                 if (changed & WIPHY_PARAM_FRAG_THRESHOLD)
842                         rdev->wiphy.frag_threshold = frag_threshold;
843                 if (changed & WIPHY_PARAM_RTS_THRESHOLD)
844                         rdev->wiphy.rts_threshold = rts_threshold;
845                 if (changed & WIPHY_PARAM_COVERAGE_CLASS)
846                         rdev->wiphy.coverage_class = coverage_class;
847
848                 result = rdev->ops->set_wiphy_params(&rdev->wiphy, changed);
849                 if (result) {
850                         rdev->wiphy.retry_short = old_retry_short;
851                         rdev->wiphy.retry_long = old_retry_long;
852                         rdev->wiphy.frag_threshold = old_frag_threshold;
853                         rdev->wiphy.rts_threshold = old_rts_threshold;
854                         rdev->wiphy.coverage_class = old_coverage_class;
855                 }
856         }
857
858  bad_res:
859         mutex_unlock(&rdev->mtx);
860  unlock:
861         rtnl_unlock();
862         return result;
863 }
864
865
866 static int nl80211_send_iface(struct sk_buff *msg, u32 pid, u32 seq, int flags,
867                               struct cfg80211_registered_device *rdev,
868                               struct net_device *dev)
869 {
870         void *hdr;
871
872         hdr = nl80211hdr_put(msg, pid, seq, flags, NL80211_CMD_NEW_INTERFACE);
873         if (!hdr)
874                 return -1;
875
876         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, dev->ifindex);
877         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
878         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, dev->name);
879         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, dev->ieee80211_ptr->iftype);
880
881         NLA_PUT_U32(msg, NL80211_ATTR_GENERATION,
882                     rdev->devlist_generation ^
883                         (cfg80211_rdev_list_generation << 2));
884
885         return genlmsg_end(msg, hdr);
886
887  nla_put_failure:
888         genlmsg_cancel(msg, hdr);
889         return -EMSGSIZE;
890 }
891
892 static int nl80211_dump_interface(struct sk_buff *skb, struct netlink_callback *cb)
893 {
894         int wp_idx = 0;
895         int if_idx = 0;
896         int wp_start = cb->args[0];
897         int if_start = cb->args[1];
898         struct cfg80211_registered_device *rdev;
899         struct wireless_dev *wdev;
900
901         mutex_lock(&cfg80211_mutex);
902         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
903                 if (!net_eq(wiphy_net(&rdev->wiphy), sock_net(skb->sk)))
904                         continue;
905                 if (wp_idx < wp_start) {
906                         wp_idx++;
907                         continue;
908                 }
909                 if_idx = 0;
910
911                 mutex_lock(&rdev->devlist_mtx);
912                 list_for_each_entry(wdev, &rdev->netdev_list, list) {
913                         if (if_idx < if_start) {
914                                 if_idx++;
915                                 continue;
916                         }
917                         if (nl80211_send_iface(skb, NETLINK_CB(cb->skb).pid,
918                                                cb->nlh->nlmsg_seq, NLM_F_MULTI,
919                                                rdev, wdev->netdev) < 0) {
920                                 mutex_unlock(&rdev->devlist_mtx);
921                                 goto out;
922                         }
923                         if_idx++;
924                 }
925                 mutex_unlock(&rdev->devlist_mtx);
926
927                 wp_idx++;
928         }
929  out:
930         mutex_unlock(&cfg80211_mutex);
931
932         cb->args[0] = wp_idx;
933         cb->args[1] = if_idx;
934
935         return skb->len;
936 }
937
938 static int nl80211_get_interface(struct sk_buff *skb, struct genl_info *info)
939 {
940         struct sk_buff *msg;
941         struct cfg80211_registered_device *dev;
942         struct net_device *netdev;
943         int err;
944
945         err = get_rdev_dev_by_info_ifindex(info, &dev, &netdev);
946         if (err)
947                 return err;
948
949         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
950         if (!msg)
951                 goto out_err;
952
953         if (nl80211_send_iface(msg, info->snd_pid, info->snd_seq, 0,
954                                dev, netdev) < 0)
955                 goto out_free;
956
957         dev_put(netdev);
958         cfg80211_unlock_rdev(dev);
959
960         return genlmsg_reply(msg, info);
961
962  out_free:
963         nlmsg_free(msg);
964  out_err:
965         dev_put(netdev);
966         cfg80211_unlock_rdev(dev);
967         return -ENOBUFS;
968 }
969
970 static const struct nla_policy mntr_flags_policy[NL80211_MNTR_FLAG_MAX + 1] = {
971         [NL80211_MNTR_FLAG_FCSFAIL] = { .type = NLA_FLAG },
972         [NL80211_MNTR_FLAG_PLCPFAIL] = { .type = NLA_FLAG },
973         [NL80211_MNTR_FLAG_CONTROL] = { .type = NLA_FLAG },
974         [NL80211_MNTR_FLAG_OTHER_BSS] = { .type = NLA_FLAG },
975         [NL80211_MNTR_FLAG_COOK_FRAMES] = { .type = NLA_FLAG },
976 };
977
978 static int parse_monitor_flags(struct nlattr *nla, u32 *mntrflags)
979 {
980         struct nlattr *flags[NL80211_MNTR_FLAG_MAX + 1];
981         int flag;
982
983         *mntrflags = 0;
984
985         if (!nla)
986                 return -EINVAL;
987
988         if (nla_parse_nested(flags, NL80211_MNTR_FLAG_MAX,
989                              nla, mntr_flags_policy))
990                 return -EINVAL;
991
992         for (flag = 1; flag <= NL80211_MNTR_FLAG_MAX; flag++)
993                 if (flags[flag])
994                         *mntrflags |= (1<<flag);
995
996         return 0;
997 }
998
999 static int nl80211_valid_4addr(struct cfg80211_registered_device *rdev,
1000                                struct net_device *netdev, u8 use_4addr,
1001                                enum nl80211_iftype iftype)
1002 {
1003         if (!use_4addr) {
1004                 if (netdev && netdev->br_port)
1005                         return -EBUSY;
1006                 return 0;
1007         }
1008
1009         switch (iftype) {
1010         case NL80211_IFTYPE_AP_VLAN:
1011                 if (rdev->wiphy.flags & WIPHY_FLAG_4ADDR_AP)
1012                         return 0;
1013                 break;
1014         case NL80211_IFTYPE_STATION:
1015                 if (rdev->wiphy.flags & WIPHY_FLAG_4ADDR_STATION)
1016                         return 0;
1017                 break;
1018         default:
1019                 break;
1020         }
1021
1022         return -EOPNOTSUPP;
1023 }
1024
1025 static int nl80211_set_interface(struct sk_buff *skb, struct genl_info *info)
1026 {
1027         struct cfg80211_registered_device *rdev;
1028         struct vif_params params;
1029         int err;
1030         enum nl80211_iftype otype, ntype;
1031         struct net_device *dev;
1032         u32 _flags, *flags = NULL;
1033         bool change = false;
1034
1035         memset(&params, 0, sizeof(params));
1036
1037         rtnl_lock();
1038
1039         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1040         if (err)
1041                 goto unlock_rtnl;
1042
1043         otype = ntype = dev->ieee80211_ptr->iftype;
1044
1045         if (info->attrs[NL80211_ATTR_IFTYPE]) {
1046                 ntype = nla_get_u32(info->attrs[NL80211_ATTR_IFTYPE]);
1047                 if (otype != ntype)
1048                         change = true;
1049                 if (ntype > NL80211_IFTYPE_MAX) {
1050                         err = -EINVAL;
1051                         goto unlock;
1052                 }
1053         }
1054
1055         if (info->attrs[NL80211_ATTR_MESH_ID]) {
1056                 if (ntype != NL80211_IFTYPE_MESH_POINT) {
1057                         err = -EINVAL;
1058                         goto unlock;
1059                 }
1060                 params.mesh_id = nla_data(info->attrs[NL80211_ATTR_MESH_ID]);
1061                 params.mesh_id_len = nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
1062                 change = true;
1063         }
1064
1065         if (info->attrs[NL80211_ATTR_4ADDR]) {
1066                 params.use_4addr = !!nla_get_u8(info->attrs[NL80211_ATTR_4ADDR]);
1067                 change = true;
1068                 err = nl80211_valid_4addr(rdev, dev, params.use_4addr, ntype);
1069                 if (err)
1070                         goto unlock;
1071         } else {
1072                 params.use_4addr = -1;
1073         }
1074
1075         if (info->attrs[NL80211_ATTR_MNTR_FLAGS]) {
1076                 if (ntype != NL80211_IFTYPE_MONITOR) {
1077                         err = -EINVAL;
1078                         goto unlock;
1079                 }
1080                 err = parse_monitor_flags(info->attrs[NL80211_ATTR_MNTR_FLAGS],
1081                                           &_flags);
1082                 if (err)
1083                         goto unlock;
1084
1085                 flags = &_flags;
1086                 change = true;
1087         }
1088
1089         if (change)
1090                 err = cfg80211_change_iface(rdev, dev, ntype, flags, &params);
1091         else
1092                 err = 0;
1093
1094         if (!err && params.use_4addr != -1)
1095                 dev->ieee80211_ptr->use_4addr = params.use_4addr;
1096
1097  unlock:
1098         dev_put(dev);
1099         cfg80211_unlock_rdev(rdev);
1100  unlock_rtnl:
1101         rtnl_unlock();
1102         return err;
1103 }
1104
1105 static int nl80211_new_interface(struct sk_buff *skb, struct genl_info *info)
1106 {
1107         struct cfg80211_registered_device *rdev;
1108         struct vif_params params;
1109         int err;
1110         enum nl80211_iftype type = NL80211_IFTYPE_UNSPECIFIED;
1111         u32 flags;
1112
1113         memset(&params, 0, sizeof(params));
1114
1115         if (!info->attrs[NL80211_ATTR_IFNAME])
1116                 return -EINVAL;
1117
1118         if (info->attrs[NL80211_ATTR_IFTYPE]) {
1119                 type = nla_get_u32(info->attrs[NL80211_ATTR_IFTYPE]);
1120                 if (type > NL80211_IFTYPE_MAX)
1121                         return -EINVAL;
1122         }
1123
1124         rtnl_lock();
1125
1126         rdev = cfg80211_get_dev_from_info(info);
1127         if (IS_ERR(rdev)) {
1128                 err = PTR_ERR(rdev);
1129                 goto unlock_rtnl;
1130         }
1131
1132         if (!rdev->ops->add_virtual_intf ||
1133             !(rdev->wiphy.interface_modes & (1 << type))) {
1134                 err = -EOPNOTSUPP;
1135                 goto unlock;
1136         }
1137
1138         if (type == NL80211_IFTYPE_MESH_POINT &&
1139             info->attrs[NL80211_ATTR_MESH_ID]) {
1140                 params.mesh_id = nla_data(info->attrs[NL80211_ATTR_MESH_ID]);
1141                 params.mesh_id_len = nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
1142         }
1143
1144         if (info->attrs[NL80211_ATTR_4ADDR]) {
1145                 params.use_4addr = !!nla_get_u8(info->attrs[NL80211_ATTR_4ADDR]);
1146                 err = nl80211_valid_4addr(rdev, NULL, params.use_4addr, type);
1147                 if (err)
1148                         goto unlock;
1149         }
1150
1151         err = parse_monitor_flags(type == NL80211_IFTYPE_MONITOR ?
1152                                   info->attrs[NL80211_ATTR_MNTR_FLAGS] : NULL,
1153                                   &flags);
1154         err = rdev->ops->add_virtual_intf(&rdev->wiphy,
1155                 nla_data(info->attrs[NL80211_ATTR_IFNAME]),
1156                 type, err ? NULL : &flags, &params);
1157
1158  unlock:
1159         cfg80211_unlock_rdev(rdev);
1160  unlock_rtnl:
1161         rtnl_unlock();
1162         return err;
1163 }
1164
1165 static int nl80211_del_interface(struct sk_buff *skb, struct genl_info *info)
1166 {
1167         struct cfg80211_registered_device *rdev;
1168         int err;
1169         struct net_device *dev;
1170
1171         rtnl_lock();
1172
1173         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1174         if (err)
1175                 goto unlock_rtnl;
1176
1177         if (!rdev->ops->del_virtual_intf) {
1178                 err = -EOPNOTSUPP;
1179                 goto out;
1180         }
1181
1182         err = rdev->ops->del_virtual_intf(&rdev->wiphy, dev);
1183
1184  out:
1185         cfg80211_unlock_rdev(rdev);
1186         dev_put(dev);
1187  unlock_rtnl:
1188         rtnl_unlock();
1189         return err;
1190 }
1191
1192 struct get_key_cookie {
1193         struct sk_buff *msg;
1194         int error;
1195         int idx;
1196 };
1197
1198 static void get_key_callback(void *c, struct key_params *params)
1199 {
1200         struct nlattr *key;
1201         struct get_key_cookie *cookie = c;
1202
1203         if (params->key)
1204                 NLA_PUT(cookie->msg, NL80211_ATTR_KEY_DATA,
1205                         params->key_len, params->key);
1206
1207         if (params->seq)
1208                 NLA_PUT(cookie->msg, NL80211_ATTR_KEY_SEQ,
1209                         params->seq_len, params->seq);
1210
1211         if (params->cipher)
1212                 NLA_PUT_U32(cookie->msg, NL80211_ATTR_KEY_CIPHER,
1213                             params->cipher);
1214
1215         key = nla_nest_start(cookie->msg, NL80211_ATTR_KEY);
1216         if (!key)
1217                 goto nla_put_failure;
1218
1219         if (params->key)
1220                 NLA_PUT(cookie->msg, NL80211_KEY_DATA,
1221                         params->key_len, params->key);
1222
1223         if (params->seq)
1224                 NLA_PUT(cookie->msg, NL80211_KEY_SEQ,
1225                         params->seq_len, params->seq);
1226
1227         if (params->cipher)
1228                 NLA_PUT_U32(cookie->msg, NL80211_KEY_CIPHER,
1229                             params->cipher);
1230
1231         NLA_PUT_U8(cookie->msg, NL80211_ATTR_KEY_IDX, cookie->idx);
1232
1233         nla_nest_end(cookie->msg, key);
1234
1235         return;
1236  nla_put_failure:
1237         cookie->error = 1;
1238 }
1239
1240 static int nl80211_get_key(struct sk_buff *skb, struct genl_info *info)
1241 {
1242         struct cfg80211_registered_device *rdev;
1243         int err;
1244         struct net_device *dev;
1245         u8 key_idx = 0;
1246         u8 *mac_addr = NULL;
1247         struct get_key_cookie cookie = {
1248                 .error = 0,
1249         };
1250         void *hdr;
1251         struct sk_buff *msg;
1252
1253         if (info->attrs[NL80211_ATTR_KEY_IDX])
1254                 key_idx = nla_get_u8(info->attrs[NL80211_ATTR_KEY_IDX]);
1255
1256         if (key_idx > 5)
1257                 return -EINVAL;
1258
1259         if (info->attrs[NL80211_ATTR_MAC])
1260                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
1261
1262         rtnl_lock();
1263
1264         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1265         if (err)
1266                 goto unlock_rtnl;
1267
1268         if (!rdev->ops->get_key) {
1269                 err = -EOPNOTSUPP;
1270                 goto out;
1271         }
1272
1273         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1274         if (!msg) {
1275                 err = -ENOMEM;
1276                 goto out;
1277         }
1278
1279         hdr = nl80211hdr_put(msg, info->snd_pid, info->snd_seq, 0,
1280                              NL80211_CMD_NEW_KEY);
1281
1282         if (IS_ERR(hdr)) {
1283                 err = PTR_ERR(hdr);
1284                 goto free_msg;
1285         }
1286
1287         cookie.msg = msg;
1288         cookie.idx = key_idx;
1289
1290         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, dev->ifindex);
1291         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
1292         if (mac_addr)
1293                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr);
1294
1295         err = rdev->ops->get_key(&rdev->wiphy, dev, key_idx, mac_addr,
1296                                 &cookie, get_key_callback);
1297
1298         if (err)
1299                 goto free_msg;
1300
1301         if (cookie.error)
1302                 goto nla_put_failure;
1303
1304         genlmsg_end(msg, hdr);
1305         err = genlmsg_reply(msg, info);
1306         goto out;
1307
1308  nla_put_failure:
1309         err = -ENOBUFS;
1310  free_msg:
1311         nlmsg_free(msg);
1312  out:
1313         cfg80211_unlock_rdev(rdev);
1314         dev_put(dev);
1315  unlock_rtnl:
1316         rtnl_unlock();
1317
1318         return err;
1319 }
1320
1321 static int nl80211_set_key(struct sk_buff *skb, struct genl_info *info)
1322 {
1323         struct cfg80211_registered_device *rdev;
1324         struct key_parse key;
1325         int err;
1326         struct net_device *dev;
1327         int (*func)(struct wiphy *wiphy, struct net_device *netdev,
1328                     u8 key_index);
1329
1330         err = nl80211_parse_key(info, &key);
1331         if (err)
1332                 return err;
1333
1334         if (key.idx < 0)
1335                 return -EINVAL;
1336
1337         /* only support setting default key */
1338         if (!key.def && !key.defmgmt)
1339                 return -EINVAL;
1340
1341         rtnl_lock();
1342
1343         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1344         if (err)
1345                 goto unlock_rtnl;
1346
1347         if (key.def)
1348                 func = rdev->ops->set_default_key;
1349         else
1350                 func = rdev->ops->set_default_mgmt_key;
1351
1352         if (!func) {
1353                 err = -EOPNOTSUPP;
1354                 goto out;
1355         }
1356
1357         wdev_lock(dev->ieee80211_ptr);
1358         err = nl80211_key_allowed(dev->ieee80211_ptr);
1359         if (!err)
1360                 err = func(&rdev->wiphy, dev, key.idx);
1361
1362 #ifdef CONFIG_CFG80211_WEXT
1363         if (!err) {
1364                 if (func == rdev->ops->set_default_key)
1365                         dev->ieee80211_ptr->wext.default_key = key.idx;
1366                 else
1367                         dev->ieee80211_ptr->wext.default_mgmt_key = key.idx;
1368         }
1369 #endif
1370         wdev_unlock(dev->ieee80211_ptr);
1371
1372  out:
1373         cfg80211_unlock_rdev(rdev);
1374         dev_put(dev);
1375
1376  unlock_rtnl:
1377         rtnl_unlock();
1378
1379         return err;
1380 }
1381
1382 static int nl80211_new_key(struct sk_buff *skb, struct genl_info *info)
1383 {
1384         struct cfg80211_registered_device *rdev;
1385         int err;
1386         struct net_device *dev;
1387         struct key_parse key;
1388         u8 *mac_addr = NULL;
1389
1390         err = nl80211_parse_key(info, &key);
1391         if (err)
1392                 return err;
1393
1394         if (!key.p.key)
1395                 return -EINVAL;
1396
1397         if (info->attrs[NL80211_ATTR_MAC])
1398                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
1399
1400         rtnl_lock();
1401
1402         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1403         if (err)
1404                 goto unlock_rtnl;
1405
1406         if (!rdev->ops->add_key) {
1407                 err = -EOPNOTSUPP;
1408                 goto out;
1409         }
1410
1411         if (cfg80211_validate_key_settings(rdev, &key.p, key.idx, mac_addr)) {
1412                 err = -EINVAL;
1413                 goto out;
1414         }
1415
1416         wdev_lock(dev->ieee80211_ptr);
1417         err = nl80211_key_allowed(dev->ieee80211_ptr);
1418         if (!err)
1419                 err = rdev->ops->add_key(&rdev->wiphy, dev, key.idx,
1420                                          mac_addr, &key.p);
1421         wdev_unlock(dev->ieee80211_ptr);
1422
1423  out:
1424         cfg80211_unlock_rdev(rdev);
1425         dev_put(dev);
1426  unlock_rtnl:
1427         rtnl_unlock();
1428
1429         return err;
1430 }
1431
1432 static int nl80211_del_key(struct sk_buff *skb, struct genl_info *info)
1433 {
1434         struct cfg80211_registered_device *rdev;
1435         int err;
1436         struct net_device *dev;
1437         u8 *mac_addr = NULL;
1438         struct key_parse key;
1439
1440         err = nl80211_parse_key(info, &key);
1441         if (err)
1442                 return err;
1443
1444         if (info->attrs[NL80211_ATTR_MAC])
1445                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
1446
1447         rtnl_lock();
1448
1449         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1450         if (err)
1451                 goto unlock_rtnl;
1452
1453         if (!rdev->ops->del_key) {
1454                 err = -EOPNOTSUPP;
1455                 goto out;
1456         }
1457
1458         wdev_lock(dev->ieee80211_ptr);
1459         err = nl80211_key_allowed(dev->ieee80211_ptr);
1460         if (!err)
1461                 err = rdev->ops->del_key(&rdev->wiphy, dev, key.idx, mac_addr);
1462
1463 #ifdef CONFIG_CFG80211_WEXT
1464         if (!err) {
1465                 if (key.idx == dev->ieee80211_ptr->wext.default_key)
1466                         dev->ieee80211_ptr->wext.default_key = -1;
1467                 else if (key.idx == dev->ieee80211_ptr->wext.default_mgmt_key)
1468                         dev->ieee80211_ptr->wext.default_mgmt_key = -1;
1469         }
1470 #endif
1471         wdev_unlock(dev->ieee80211_ptr);
1472
1473  out:
1474         cfg80211_unlock_rdev(rdev);
1475         dev_put(dev);
1476
1477  unlock_rtnl:
1478         rtnl_unlock();
1479
1480         return err;
1481 }
1482
1483 static int nl80211_addset_beacon(struct sk_buff *skb, struct genl_info *info)
1484 {
1485         int (*call)(struct wiphy *wiphy, struct net_device *dev,
1486                     struct beacon_parameters *info);
1487         struct cfg80211_registered_device *rdev;
1488         int err;
1489         struct net_device *dev;
1490         struct beacon_parameters params;
1491         int haveinfo = 0;
1492
1493         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_BEACON_TAIL]))
1494                 return -EINVAL;
1495
1496         rtnl_lock();
1497
1498         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1499         if (err)
1500                 goto unlock_rtnl;
1501
1502         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP) {
1503                 err = -EOPNOTSUPP;
1504                 goto out;
1505         }
1506
1507         switch (info->genlhdr->cmd) {
1508         case NL80211_CMD_NEW_BEACON:
1509                 /* these are required for NEW_BEACON */
1510                 if (!info->attrs[NL80211_ATTR_BEACON_INTERVAL] ||
1511                     !info->attrs[NL80211_ATTR_DTIM_PERIOD] ||
1512                     !info->attrs[NL80211_ATTR_BEACON_HEAD]) {
1513                         err = -EINVAL;
1514                         goto out;
1515                 }
1516
1517                 call = rdev->ops->add_beacon;
1518                 break;
1519         case NL80211_CMD_SET_BEACON:
1520                 call = rdev->ops->set_beacon;
1521                 break;
1522         default:
1523                 WARN_ON(1);
1524                 err = -EOPNOTSUPP;
1525                 goto out;
1526         }
1527
1528         if (!call) {
1529                 err = -EOPNOTSUPP;
1530                 goto out;
1531         }
1532
1533         memset(&params, 0, sizeof(params));
1534
1535         if (info->attrs[NL80211_ATTR_BEACON_INTERVAL]) {
1536                 params.interval =
1537                     nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
1538                 haveinfo = 1;
1539         }
1540
1541         if (info->attrs[NL80211_ATTR_DTIM_PERIOD]) {
1542                 params.dtim_period =
1543                     nla_get_u32(info->attrs[NL80211_ATTR_DTIM_PERIOD]);
1544                 haveinfo = 1;
1545         }
1546
1547         if (info->attrs[NL80211_ATTR_BEACON_HEAD]) {
1548                 params.head = nla_data(info->attrs[NL80211_ATTR_BEACON_HEAD]);
1549                 params.head_len =
1550                     nla_len(info->attrs[NL80211_ATTR_BEACON_HEAD]);
1551                 haveinfo = 1;
1552         }
1553
1554         if (info->attrs[NL80211_ATTR_BEACON_TAIL]) {
1555                 params.tail = nla_data(info->attrs[NL80211_ATTR_BEACON_TAIL]);
1556                 params.tail_len =
1557                     nla_len(info->attrs[NL80211_ATTR_BEACON_TAIL]);
1558                 haveinfo = 1;
1559         }
1560
1561         if (!haveinfo) {
1562                 err = -EINVAL;
1563                 goto out;
1564         }
1565
1566         err = call(&rdev->wiphy, dev, &params);
1567
1568  out:
1569         cfg80211_unlock_rdev(rdev);
1570         dev_put(dev);
1571  unlock_rtnl:
1572         rtnl_unlock();
1573
1574         return err;
1575 }
1576
1577 static int nl80211_del_beacon(struct sk_buff *skb, struct genl_info *info)
1578 {
1579         struct cfg80211_registered_device *rdev;
1580         int err;
1581         struct net_device *dev;
1582
1583         rtnl_lock();
1584
1585         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1586         if (err)
1587                 goto unlock_rtnl;
1588
1589         if (!rdev->ops->del_beacon) {
1590                 err = -EOPNOTSUPP;
1591                 goto out;
1592         }
1593
1594         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP) {
1595                 err = -EOPNOTSUPP;
1596                 goto out;
1597         }
1598         err = rdev->ops->del_beacon(&rdev->wiphy, dev);
1599
1600  out:
1601         cfg80211_unlock_rdev(rdev);
1602         dev_put(dev);
1603  unlock_rtnl:
1604         rtnl_unlock();
1605
1606         return err;
1607 }
1608
1609 static const struct nla_policy sta_flags_policy[NL80211_STA_FLAG_MAX + 1] = {
1610         [NL80211_STA_FLAG_AUTHORIZED] = { .type = NLA_FLAG },
1611         [NL80211_STA_FLAG_SHORT_PREAMBLE] = { .type = NLA_FLAG },
1612         [NL80211_STA_FLAG_WME] = { .type = NLA_FLAG },
1613         [NL80211_STA_FLAG_MFP] = { .type = NLA_FLAG },
1614 };
1615
1616 static int parse_station_flags(struct genl_info *info,
1617                                struct station_parameters *params)
1618 {
1619         struct nlattr *flags[NL80211_STA_FLAG_MAX + 1];
1620         struct nlattr *nla;
1621         int flag;
1622
1623         /*
1624          * Try parsing the new attribute first so userspace
1625          * can specify both for older kernels.
1626          */
1627         nla = info->attrs[NL80211_ATTR_STA_FLAGS2];
1628         if (nla) {
1629                 struct nl80211_sta_flag_update *sta_flags;
1630
1631                 sta_flags = nla_data(nla);
1632                 params->sta_flags_mask = sta_flags->mask;
1633                 params->sta_flags_set = sta_flags->set;
1634                 if ((params->sta_flags_mask |
1635                      params->sta_flags_set) & BIT(__NL80211_STA_FLAG_INVALID))
1636                         return -EINVAL;
1637                 return 0;
1638         }
1639
1640         /* if present, parse the old attribute */
1641
1642         nla = info->attrs[NL80211_ATTR_STA_FLAGS];
1643         if (!nla)
1644                 return 0;
1645
1646         if (nla_parse_nested(flags, NL80211_STA_FLAG_MAX,
1647                              nla, sta_flags_policy))
1648                 return -EINVAL;
1649
1650         params->sta_flags_mask = (1 << __NL80211_STA_FLAG_AFTER_LAST) - 1;
1651         params->sta_flags_mask &= ~1;
1652
1653         for (flag = 1; flag <= NL80211_STA_FLAG_MAX; flag++)
1654                 if (flags[flag])
1655                         params->sta_flags_set |= (1<<flag);
1656
1657         return 0;
1658 }
1659
1660 static int nl80211_send_station(struct sk_buff *msg, u32 pid, u32 seq,
1661                                 int flags, struct net_device *dev,
1662                                 const u8 *mac_addr, struct station_info *sinfo)
1663 {
1664         void *hdr;
1665         struct nlattr *sinfoattr, *txrate;
1666         u16 bitrate;
1667
1668         hdr = nl80211hdr_put(msg, pid, seq, flags, NL80211_CMD_NEW_STATION);
1669         if (!hdr)
1670                 return -1;
1671
1672         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, dev->ifindex);
1673         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr);
1674
1675         NLA_PUT_U32(msg, NL80211_ATTR_GENERATION, sinfo->generation);
1676
1677         sinfoattr = nla_nest_start(msg, NL80211_ATTR_STA_INFO);
1678         if (!sinfoattr)
1679                 goto nla_put_failure;
1680         if (sinfo->filled & STATION_INFO_INACTIVE_TIME)
1681                 NLA_PUT_U32(msg, NL80211_STA_INFO_INACTIVE_TIME,
1682                             sinfo->inactive_time);
1683         if (sinfo->filled & STATION_INFO_RX_BYTES)
1684                 NLA_PUT_U32(msg, NL80211_STA_INFO_RX_BYTES,
1685                             sinfo->rx_bytes);
1686         if (sinfo->filled & STATION_INFO_TX_BYTES)
1687                 NLA_PUT_U32(msg, NL80211_STA_INFO_TX_BYTES,
1688                             sinfo->tx_bytes);
1689         if (sinfo->filled & STATION_INFO_LLID)
1690                 NLA_PUT_U16(msg, NL80211_STA_INFO_LLID,
1691                             sinfo->llid);
1692         if (sinfo->filled & STATION_INFO_PLID)
1693                 NLA_PUT_U16(msg, NL80211_STA_INFO_PLID,
1694                             sinfo->plid);
1695         if (sinfo->filled & STATION_INFO_PLINK_STATE)
1696                 NLA_PUT_U8(msg, NL80211_STA_INFO_PLINK_STATE,
1697                             sinfo->plink_state);
1698         if (sinfo->filled & STATION_INFO_SIGNAL)
1699                 NLA_PUT_U8(msg, NL80211_STA_INFO_SIGNAL,
1700                            sinfo->signal);
1701         if (sinfo->filled & STATION_INFO_TX_BITRATE) {
1702                 txrate = nla_nest_start(msg, NL80211_STA_INFO_TX_BITRATE);
1703                 if (!txrate)
1704                         goto nla_put_failure;
1705
1706                 /* cfg80211_calculate_bitrate will return 0 for mcs >= 32 */
1707                 bitrate = cfg80211_calculate_bitrate(&sinfo->txrate);
1708                 if (bitrate > 0)
1709                         NLA_PUT_U16(msg, NL80211_RATE_INFO_BITRATE, bitrate);
1710
1711                 if (sinfo->txrate.flags & RATE_INFO_FLAGS_MCS)
1712                         NLA_PUT_U8(msg, NL80211_RATE_INFO_MCS,
1713                                     sinfo->txrate.mcs);
1714                 if (sinfo->txrate.flags & RATE_INFO_FLAGS_40_MHZ_WIDTH)
1715                         NLA_PUT_FLAG(msg, NL80211_RATE_INFO_40_MHZ_WIDTH);
1716                 if (sinfo->txrate.flags & RATE_INFO_FLAGS_SHORT_GI)
1717                         NLA_PUT_FLAG(msg, NL80211_RATE_INFO_SHORT_GI);
1718
1719                 nla_nest_end(msg, txrate);
1720         }
1721         if (sinfo->filled & STATION_INFO_RX_PACKETS)
1722                 NLA_PUT_U32(msg, NL80211_STA_INFO_RX_PACKETS,
1723                             sinfo->rx_packets);
1724         if (sinfo->filled & STATION_INFO_TX_PACKETS)
1725                 NLA_PUT_U32(msg, NL80211_STA_INFO_TX_PACKETS,
1726                             sinfo->tx_packets);
1727         nla_nest_end(msg, sinfoattr);
1728
1729         return genlmsg_end(msg, hdr);
1730
1731  nla_put_failure:
1732         genlmsg_cancel(msg, hdr);
1733         return -EMSGSIZE;
1734 }
1735
1736 static int nl80211_dump_station(struct sk_buff *skb,
1737                                 struct netlink_callback *cb)
1738 {
1739         struct station_info sinfo;
1740         struct cfg80211_registered_device *dev;
1741         struct net_device *netdev;
1742         u8 mac_addr[ETH_ALEN];
1743         int ifidx = cb->args[0];
1744         int sta_idx = cb->args[1];
1745         int err;
1746
1747         if (!ifidx)
1748                 ifidx = nl80211_get_ifidx(cb);
1749         if (ifidx < 0)
1750                 return ifidx;
1751
1752         rtnl_lock();
1753
1754         netdev = __dev_get_by_index(sock_net(skb->sk), ifidx);
1755         if (!netdev) {
1756                 err = -ENODEV;
1757                 goto out_rtnl;
1758         }
1759
1760         dev = cfg80211_get_dev_from_ifindex(sock_net(skb->sk), ifidx);
1761         if (IS_ERR(dev)) {
1762                 err = PTR_ERR(dev);
1763                 goto out_rtnl;
1764         }
1765
1766         if (!dev->ops->dump_station) {
1767                 err = -EOPNOTSUPP;
1768                 goto out_err;
1769         }
1770
1771         while (1) {
1772                 err = dev->ops->dump_station(&dev->wiphy, netdev, sta_idx,
1773                                              mac_addr, &sinfo);
1774                 if (err == -ENOENT)
1775                         break;
1776                 if (err)
1777                         goto out_err;
1778
1779                 if (nl80211_send_station(skb,
1780                                 NETLINK_CB(cb->skb).pid,
1781                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
1782                                 netdev, mac_addr,
1783                                 &sinfo) < 0)
1784                         goto out;
1785
1786                 sta_idx++;
1787         }
1788
1789
1790  out:
1791         cb->args[1] = sta_idx;
1792         err = skb->len;
1793  out_err:
1794         cfg80211_unlock_rdev(dev);
1795  out_rtnl:
1796         rtnl_unlock();
1797
1798         return err;
1799 }
1800
1801 static int nl80211_get_station(struct sk_buff *skb, struct genl_info *info)
1802 {
1803         struct cfg80211_registered_device *rdev;
1804         int err;
1805         struct net_device *dev;
1806         struct station_info sinfo;
1807         struct sk_buff *msg;
1808         u8 *mac_addr = NULL;
1809
1810         memset(&sinfo, 0, sizeof(sinfo));
1811
1812         if (!info->attrs[NL80211_ATTR_MAC])
1813                 return -EINVAL;
1814
1815         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
1816
1817         rtnl_lock();
1818
1819         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1820         if (err)
1821                 goto out_rtnl;
1822
1823         if (!rdev->ops->get_station) {
1824                 err = -EOPNOTSUPP;
1825                 goto out;
1826         }
1827
1828         err = rdev->ops->get_station(&rdev->wiphy, dev, mac_addr, &sinfo);
1829         if (err)
1830                 goto out;
1831
1832         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1833         if (!msg)
1834                 goto out;
1835
1836         if (nl80211_send_station(msg, info->snd_pid, info->snd_seq, 0,
1837                                  dev, mac_addr, &sinfo) < 0)
1838                 goto out_free;
1839
1840         err = genlmsg_reply(msg, info);
1841         goto out;
1842
1843  out_free:
1844         nlmsg_free(msg);
1845  out:
1846         cfg80211_unlock_rdev(rdev);
1847         dev_put(dev);
1848  out_rtnl:
1849         rtnl_unlock();
1850
1851         return err;
1852 }
1853
1854 /*
1855  * Get vlan interface making sure it is running and on the right wiphy.
1856  */
1857 static int get_vlan(struct genl_info *info,
1858                     struct cfg80211_registered_device *rdev,
1859                     struct net_device **vlan)
1860 {
1861         struct nlattr *vlanattr = info->attrs[NL80211_ATTR_STA_VLAN];
1862         *vlan = NULL;
1863
1864         if (vlanattr) {
1865                 *vlan = dev_get_by_index(genl_info_net(info),
1866                                          nla_get_u32(vlanattr));
1867                 if (!*vlan)
1868                         return -ENODEV;
1869                 if (!(*vlan)->ieee80211_ptr)
1870                         return -EINVAL;
1871                 if ((*vlan)->ieee80211_ptr->wiphy != &rdev->wiphy)
1872                         return -EINVAL;
1873                 if (!netif_running(*vlan))
1874                         return -ENETDOWN;
1875         }
1876         return 0;
1877 }
1878
1879 static int nl80211_set_station(struct sk_buff *skb, struct genl_info *info)
1880 {
1881         struct cfg80211_registered_device *rdev;
1882         int err;
1883         struct net_device *dev;
1884         struct station_parameters params;
1885         u8 *mac_addr = NULL;
1886
1887         memset(&params, 0, sizeof(params));
1888
1889         params.listen_interval = -1;
1890
1891         if (info->attrs[NL80211_ATTR_STA_AID])
1892                 return -EINVAL;
1893
1894         if (!info->attrs[NL80211_ATTR_MAC])
1895                 return -EINVAL;
1896
1897         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
1898
1899         if (info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]) {
1900                 params.supported_rates =
1901                         nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
1902                 params.supported_rates_len =
1903                         nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
1904         }
1905
1906         if (info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL])
1907                 params.listen_interval =
1908                     nla_get_u16(info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL]);
1909
1910         if (info->attrs[NL80211_ATTR_HT_CAPABILITY])
1911                 params.ht_capa =
1912                         nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]);
1913
1914         if (parse_station_flags(info, &params))
1915                 return -EINVAL;
1916
1917         if (info->attrs[NL80211_ATTR_STA_PLINK_ACTION])
1918                 params.plink_action =
1919                     nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_ACTION]);
1920
1921         rtnl_lock();
1922
1923         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
1924         if (err)
1925                 goto out_rtnl;
1926
1927         err = get_vlan(info, rdev, &params.vlan);
1928         if (err)
1929                 goto out;
1930
1931         /* validate settings */
1932         err = 0;
1933
1934         switch (dev->ieee80211_ptr->iftype) {
1935         case NL80211_IFTYPE_AP:
1936         case NL80211_IFTYPE_AP_VLAN:
1937                 /* disallow mesh-specific things */
1938                 if (params.plink_action)
1939                         err = -EINVAL;
1940                 break;
1941         case NL80211_IFTYPE_STATION:
1942                 /* disallow everything but AUTHORIZED flag */
1943                 if (params.plink_action)
1944                         err = -EINVAL;
1945                 if (params.vlan)
1946                         err = -EINVAL;
1947                 if (params.supported_rates)
1948                         err = -EINVAL;
1949                 if (params.ht_capa)
1950                         err = -EINVAL;
1951                 if (params.listen_interval >= 0)
1952                         err = -EINVAL;
1953                 if (params.sta_flags_mask & ~BIT(NL80211_STA_FLAG_AUTHORIZED))
1954                         err = -EINVAL;
1955                 break;
1956         case NL80211_IFTYPE_MESH_POINT:
1957                 /* disallow things mesh doesn't support */
1958                 if (params.vlan)
1959                         err = -EINVAL;
1960                 if (params.ht_capa)
1961                         err = -EINVAL;
1962                 if (params.listen_interval >= 0)
1963                         err = -EINVAL;
1964                 if (params.supported_rates)
1965                         err = -EINVAL;
1966                 if (params.sta_flags_mask)
1967                         err = -EINVAL;
1968                 break;
1969         default:
1970                 err = -EINVAL;
1971         }
1972
1973         if (err)
1974                 goto out;
1975
1976         if (!rdev->ops->change_station) {
1977                 err = -EOPNOTSUPP;
1978                 goto out;
1979         }
1980
1981         err = rdev->ops->change_station(&rdev->wiphy, dev, mac_addr, &params);
1982
1983  out:
1984         if (params.vlan)
1985                 dev_put(params.vlan);
1986         cfg80211_unlock_rdev(rdev);
1987         dev_put(dev);
1988  out_rtnl:
1989         rtnl_unlock();
1990
1991         return err;
1992 }
1993
1994 static int nl80211_new_station(struct sk_buff *skb, struct genl_info *info)
1995 {
1996         struct cfg80211_registered_device *rdev;
1997         int err;
1998         struct net_device *dev;
1999         struct station_parameters params;
2000         u8 *mac_addr = NULL;
2001
2002         memset(&params, 0, sizeof(params));
2003
2004         if (!info->attrs[NL80211_ATTR_MAC])
2005                 return -EINVAL;
2006
2007         if (!info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL])
2008                 return -EINVAL;
2009
2010         if (!info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES])
2011                 return -EINVAL;
2012
2013         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
2014         params.supported_rates =
2015                 nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
2016         params.supported_rates_len =
2017                 nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
2018         params.listen_interval =
2019                 nla_get_u16(info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL]);
2020
2021         if (info->attrs[NL80211_ATTR_STA_AID]) {
2022                 params.aid = nla_get_u16(info->attrs[NL80211_ATTR_STA_AID]);
2023                 if (!params.aid || params.aid > IEEE80211_MAX_AID)
2024                         return -EINVAL;
2025         }
2026
2027         if (info->attrs[NL80211_ATTR_HT_CAPABILITY])
2028                 params.ht_capa =
2029                         nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]);
2030
2031         if (parse_station_flags(info, &params))
2032                 return -EINVAL;
2033
2034         rtnl_lock();
2035
2036         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2037         if (err)
2038                 goto out_rtnl;
2039
2040         err = get_vlan(info, rdev, &params.vlan);
2041         if (err)
2042                 goto out;
2043
2044         /* validate settings */
2045         err = 0;
2046
2047         switch (dev->ieee80211_ptr->iftype) {
2048         case NL80211_IFTYPE_AP:
2049         case NL80211_IFTYPE_AP_VLAN:
2050                 /* all ok but must have AID */
2051                 if (!params.aid)
2052                         err = -EINVAL;
2053                 break;
2054         case NL80211_IFTYPE_MESH_POINT:
2055                 /* disallow things mesh doesn't support */
2056                 if (params.vlan)
2057                         err = -EINVAL;
2058                 if (params.aid)
2059                         err = -EINVAL;
2060                 if (params.ht_capa)
2061                         err = -EINVAL;
2062                 if (params.listen_interval >= 0)
2063                         err = -EINVAL;
2064                 if (params.supported_rates)
2065                         err = -EINVAL;
2066                 if (params.sta_flags_mask)
2067                         err = -EINVAL;
2068                 break;
2069         default:
2070                 err = -EINVAL;
2071         }
2072
2073         if (err)
2074                 goto out;
2075
2076         if (!rdev->ops->add_station) {
2077                 err = -EOPNOTSUPP;
2078                 goto out;
2079         }
2080
2081         if (!netif_running(dev)) {
2082                 err = -ENETDOWN;
2083                 goto out;
2084         }
2085
2086         err = rdev->ops->add_station(&rdev->wiphy, dev, mac_addr, &params);
2087
2088  out:
2089         if (params.vlan)
2090                 dev_put(params.vlan);
2091         cfg80211_unlock_rdev(rdev);
2092         dev_put(dev);
2093  out_rtnl:
2094         rtnl_unlock();
2095
2096         return err;
2097 }
2098
2099 static int nl80211_del_station(struct sk_buff *skb, struct genl_info *info)
2100 {
2101         struct cfg80211_registered_device *rdev;
2102         int err;
2103         struct net_device *dev;
2104         u8 *mac_addr = NULL;
2105
2106         if (info->attrs[NL80211_ATTR_MAC])
2107                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
2108
2109         rtnl_lock();
2110
2111         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2112         if (err)
2113                 goto out_rtnl;
2114
2115         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
2116             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP_VLAN &&
2117             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT) {
2118                 err = -EINVAL;
2119                 goto out;
2120         }
2121
2122         if (!rdev->ops->del_station) {
2123                 err = -EOPNOTSUPP;
2124                 goto out;
2125         }
2126
2127         err = rdev->ops->del_station(&rdev->wiphy, dev, mac_addr);
2128
2129  out:
2130         cfg80211_unlock_rdev(rdev);
2131         dev_put(dev);
2132  out_rtnl:
2133         rtnl_unlock();
2134
2135         return err;
2136 }
2137
2138 static int nl80211_send_mpath(struct sk_buff *msg, u32 pid, u32 seq,
2139                                 int flags, struct net_device *dev,
2140                                 u8 *dst, u8 *next_hop,
2141                                 struct mpath_info *pinfo)
2142 {
2143         void *hdr;
2144         struct nlattr *pinfoattr;
2145
2146         hdr = nl80211hdr_put(msg, pid, seq, flags, NL80211_CMD_NEW_STATION);
2147         if (!hdr)
2148                 return -1;
2149
2150         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, dev->ifindex);
2151         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, dst);
2152         NLA_PUT(msg, NL80211_ATTR_MPATH_NEXT_HOP, ETH_ALEN, next_hop);
2153
2154         NLA_PUT_U32(msg, NL80211_ATTR_GENERATION, pinfo->generation);
2155
2156         pinfoattr = nla_nest_start(msg, NL80211_ATTR_MPATH_INFO);
2157         if (!pinfoattr)
2158                 goto nla_put_failure;
2159         if (pinfo->filled & MPATH_INFO_FRAME_QLEN)
2160                 NLA_PUT_U32(msg, NL80211_MPATH_INFO_FRAME_QLEN,
2161                             pinfo->frame_qlen);
2162         if (pinfo->filled & MPATH_INFO_SN)
2163                 NLA_PUT_U32(msg, NL80211_MPATH_INFO_SN,
2164                             pinfo->sn);
2165         if (pinfo->filled & MPATH_INFO_METRIC)
2166                 NLA_PUT_U32(msg, NL80211_MPATH_INFO_METRIC,
2167                             pinfo->metric);
2168         if (pinfo->filled & MPATH_INFO_EXPTIME)
2169                 NLA_PUT_U32(msg, NL80211_MPATH_INFO_EXPTIME,
2170                             pinfo->exptime);
2171         if (pinfo->filled & MPATH_INFO_FLAGS)
2172                 NLA_PUT_U8(msg, NL80211_MPATH_INFO_FLAGS,
2173                             pinfo->flags);
2174         if (pinfo->filled & MPATH_INFO_DISCOVERY_TIMEOUT)
2175                 NLA_PUT_U32(msg, NL80211_MPATH_INFO_DISCOVERY_TIMEOUT,
2176                             pinfo->discovery_timeout);
2177         if (pinfo->filled & MPATH_INFO_DISCOVERY_RETRIES)
2178                 NLA_PUT_U8(msg, NL80211_MPATH_INFO_DISCOVERY_RETRIES,
2179                             pinfo->discovery_retries);
2180
2181         nla_nest_end(msg, pinfoattr);
2182
2183         return genlmsg_end(msg, hdr);
2184
2185  nla_put_failure:
2186         genlmsg_cancel(msg, hdr);
2187         return -EMSGSIZE;
2188 }
2189
2190 static int nl80211_dump_mpath(struct sk_buff *skb,
2191                               struct netlink_callback *cb)
2192 {
2193         struct mpath_info pinfo;
2194         struct cfg80211_registered_device *dev;
2195         struct net_device *netdev;
2196         u8 dst[ETH_ALEN];
2197         u8 next_hop[ETH_ALEN];
2198         int ifidx = cb->args[0];
2199         int path_idx = cb->args[1];
2200         int err;
2201
2202         if (!ifidx)
2203                 ifidx = nl80211_get_ifidx(cb);
2204         if (ifidx < 0)
2205                 return ifidx;
2206
2207         rtnl_lock();
2208
2209         netdev = __dev_get_by_index(sock_net(skb->sk), ifidx);
2210         if (!netdev) {
2211                 err = -ENODEV;
2212                 goto out_rtnl;
2213         }
2214
2215         dev = cfg80211_get_dev_from_ifindex(sock_net(skb->sk), ifidx);
2216         if (IS_ERR(dev)) {
2217                 err = PTR_ERR(dev);
2218                 goto out_rtnl;
2219         }
2220
2221         if (!dev->ops->dump_mpath) {
2222                 err = -EOPNOTSUPP;
2223                 goto out_err;
2224         }
2225
2226         if (netdev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT) {
2227                 err = -EOPNOTSUPP;
2228                 goto out_err;
2229         }
2230
2231         while (1) {
2232                 err = dev->ops->dump_mpath(&dev->wiphy, netdev, path_idx,
2233                                            dst, next_hop, &pinfo);
2234                 if (err == -ENOENT)
2235                         break;
2236                 if (err)
2237                         goto out_err;
2238
2239                 if (nl80211_send_mpath(skb, NETLINK_CB(cb->skb).pid,
2240                                        cb->nlh->nlmsg_seq, NLM_F_MULTI,
2241                                        netdev, dst, next_hop,
2242                                        &pinfo) < 0)
2243                         goto out;
2244
2245                 path_idx++;
2246         }
2247
2248
2249  out:
2250         cb->args[1] = path_idx;
2251         err = skb->len;
2252  out_err:
2253         cfg80211_unlock_rdev(dev);
2254  out_rtnl:
2255         rtnl_unlock();
2256
2257         return err;
2258 }
2259
2260 static int nl80211_get_mpath(struct sk_buff *skb, struct genl_info *info)
2261 {
2262         struct cfg80211_registered_device *rdev;
2263         int err;
2264         struct net_device *dev;
2265         struct mpath_info pinfo;
2266         struct sk_buff *msg;
2267         u8 *dst = NULL;
2268         u8 next_hop[ETH_ALEN];
2269
2270         memset(&pinfo, 0, sizeof(pinfo));
2271
2272         if (!info->attrs[NL80211_ATTR_MAC])
2273                 return -EINVAL;
2274
2275         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
2276
2277         rtnl_lock();
2278
2279         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2280         if (err)
2281                 goto out_rtnl;
2282
2283         if (!rdev->ops->get_mpath) {
2284                 err = -EOPNOTSUPP;
2285                 goto out;
2286         }
2287
2288         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT) {
2289                 err = -EOPNOTSUPP;
2290                 goto out;
2291         }
2292
2293         err = rdev->ops->get_mpath(&rdev->wiphy, dev, dst, next_hop, &pinfo);
2294         if (err)
2295                 goto out;
2296
2297         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2298         if (!msg)
2299                 goto out;
2300
2301         if (nl80211_send_mpath(msg, info->snd_pid, info->snd_seq, 0,
2302                                  dev, dst, next_hop, &pinfo) < 0)
2303                 goto out_free;
2304
2305         err = genlmsg_reply(msg, info);
2306         goto out;
2307
2308  out_free:
2309         nlmsg_free(msg);
2310  out:
2311         cfg80211_unlock_rdev(rdev);
2312         dev_put(dev);
2313  out_rtnl:
2314         rtnl_unlock();
2315
2316         return err;
2317 }
2318
2319 static int nl80211_set_mpath(struct sk_buff *skb, struct genl_info *info)
2320 {
2321         struct cfg80211_registered_device *rdev;
2322         int err;
2323         struct net_device *dev;
2324         u8 *dst = NULL;
2325         u8 *next_hop = NULL;
2326
2327         if (!info->attrs[NL80211_ATTR_MAC])
2328                 return -EINVAL;
2329
2330         if (!info->attrs[NL80211_ATTR_MPATH_NEXT_HOP])
2331                 return -EINVAL;
2332
2333         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
2334         next_hop = nla_data(info->attrs[NL80211_ATTR_MPATH_NEXT_HOP]);
2335
2336         rtnl_lock();
2337
2338         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2339         if (err)
2340                 goto out_rtnl;
2341
2342         if (!rdev->ops->change_mpath) {
2343                 err = -EOPNOTSUPP;
2344                 goto out;
2345         }
2346
2347         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT) {
2348                 err = -EOPNOTSUPP;
2349                 goto out;
2350         }
2351
2352         if (!netif_running(dev)) {
2353                 err = -ENETDOWN;
2354                 goto out;
2355         }
2356
2357         err = rdev->ops->change_mpath(&rdev->wiphy, dev, dst, next_hop);
2358
2359  out:
2360         cfg80211_unlock_rdev(rdev);
2361         dev_put(dev);
2362  out_rtnl:
2363         rtnl_unlock();
2364
2365         return err;
2366 }
2367 static int nl80211_new_mpath(struct sk_buff *skb, struct genl_info *info)
2368 {
2369         struct cfg80211_registered_device *rdev;
2370         int err;
2371         struct net_device *dev;
2372         u8 *dst = NULL;
2373         u8 *next_hop = NULL;
2374
2375         if (!info->attrs[NL80211_ATTR_MAC])
2376                 return -EINVAL;
2377
2378         if (!info->attrs[NL80211_ATTR_MPATH_NEXT_HOP])
2379                 return -EINVAL;
2380
2381         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
2382         next_hop = nla_data(info->attrs[NL80211_ATTR_MPATH_NEXT_HOP]);
2383
2384         rtnl_lock();
2385
2386         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2387         if (err)
2388                 goto out_rtnl;
2389
2390         if (!rdev->ops->add_mpath) {
2391                 err = -EOPNOTSUPP;
2392                 goto out;
2393         }
2394
2395         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT) {
2396                 err = -EOPNOTSUPP;
2397                 goto out;
2398         }
2399
2400         if (!netif_running(dev)) {
2401                 err = -ENETDOWN;
2402                 goto out;
2403         }
2404
2405         err = rdev->ops->add_mpath(&rdev->wiphy, dev, dst, next_hop);
2406
2407  out:
2408         cfg80211_unlock_rdev(rdev);
2409         dev_put(dev);
2410  out_rtnl:
2411         rtnl_unlock();
2412
2413         return err;
2414 }
2415
2416 static int nl80211_del_mpath(struct sk_buff *skb, struct genl_info *info)
2417 {
2418         struct cfg80211_registered_device *rdev;
2419         int err;
2420         struct net_device *dev;
2421         u8 *dst = NULL;
2422
2423         if (info->attrs[NL80211_ATTR_MAC])
2424                 dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
2425
2426         rtnl_lock();
2427
2428         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2429         if (err)
2430                 goto out_rtnl;
2431
2432         if (!rdev->ops->del_mpath) {
2433                 err = -EOPNOTSUPP;
2434                 goto out;
2435         }
2436
2437         err = rdev->ops->del_mpath(&rdev->wiphy, dev, dst);
2438
2439  out:
2440         cfg80211_unlock_rdev(rdev);
2441         dev_put(dev);
2442  out_rtnl:
2443         rtnl_unlock();
2444
2445         return err;
2446 }
2447
2448 static int nl80211_set_bss(struct sk_buff *skb, struct genl_info *info)
2449 {
2450         struct cfg80211_registered_device *rdev;
2451         int err;
2452         struct net_device *dev;
2453         struct bss_parameters params;
2454
2455         memset(&params, 0, sizeof(params));
2456         /* default to not changing parameters */
2457         params.use_cts_prot = -1;
2458         params.use_short_preamble = -1;
2459         params.use_short_slot_time = -1;
2460
2461         if (info->attrs[NL80211_ATTR_BSS_CTS_PROT])
2462                 params.use_cts_prot =
2463                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_CTS_PROT]);
2464         if (info->attrs[NL80211_ATTR_BSS_SHORT_PREAMBLE])
2465                 params.use_short_preamble =
2466                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_SHORT_PREAMBLE]);
2467         if (info->attrs[NL80211_ATTR_BSS_SHORT_SLOT_TIME])
2468                 params.use_short_slot_time =
2469                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_SHORT_SLOT_TIME]);
2470         if (info->attrs[NL80211_ATTR_BSS_BASIC_RATES]) {
2471                 params.basic_rates =
2472                         nla_data(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
2473                 params.basic_rates_len =
2474                         nla_len(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
2475         }
2476
2477         rtnl_lock();
2478
2479         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2480         if (err)
2481                 goto out_rtnl;
2482
2483         if (!rdev->ops->change_bss) {
2484                 err = -EOPNOTSUPP;
2485                 goto out;
2486         }
2487
2488         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP) {
2489                 err = -EOPNOTSUPP;
2490                 goto out;
2491         }
2492
2493         err = rdev->ops->change_bss(&rdev->wiphy, dev, &params);
2494
2495  out:
2496         cfg80211_unlock_rdev(rdev);
2497         dev_put(dev);
2498  out_rtnl:
2499         rtnl_unlock();
2500
2501         return err;
2502 }
2503
2504 static const struct nla_policy
2505         reg_rule_policy[NL80211_REG_RULE_ATTR_MAX + 1] = {
2506         [NL80211_ATTR_REG_RULE_FLAGS]           = { .type = NLA_U32 },
2507         [NL80211_ATTR_FREQ_RANGE_START]         = { .type = NLA_U32 },
2508         [NL80211_ATTR_FREQ_RANGE_END]           = { .type = NLA_U32 },
2509         [NL80211_ATTR_FREQ_RANGE_MAX_BW]        = { .type = NLA_U32 },
2510         [NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN]  = { .type = NLA_U32 },
2511         [NL80211_ATTR_POWER_RULE_MAX_EIRP]      = { .type = NLA_U32 },
2512 };
2513
2514 static int parse_reg_rule(struct nlattr *tb[],
2515         struct ieee80211_reg_rule *reg_rule)
2516 {
2517         struct ieee80211_freq_range *freq_range = &reg_rule->freq_range;
2518         struct ieee80211_power_rule *power_rule = &reg_rule->power_rule;
2519
2520         if (!tb[NL80211_ATTR_REG_RULE_FLAGS])
2521                 return -EINVAL;
2522         if (!tb[NL80211_ATTR_FREQ_RANGE_START])
2523                 return -EINVAL;
2524         if (!tb[NL80211_ATTR_FREQ_RANGE_END])
2525                 return -EINVAL;
2526         if (!tb[NL80211_ATTR_FREQ_RANGE_MAX_BW])
2527                 return -EINVAL;
2528         if (!tb[NL80211_ATTR_POWER_RULE_MAX_EIRP])
2529                 return -EINVAL;
2530
2531         reg_rule->flags = nla_get_u32(tb[NL80211_ATTR_REG_RULE_FLAGS]);
2532
2533         freq_range->start_freq_khz =
2534                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]);
2535         freq_range->end_freq_khz =
2536                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]);
2537         freq_range->max_bandwidth_khz =
2538                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]);
2539
2540         power_rule->max_eirp =
2541                 nla_get_u32(tb[NL80211_ATTR_POWER_RULE_MAX_EIRP]);
2542
2543         if (tb[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN])
2544                 power_rule->max_antenna_gain =
2545                         nla_get_u32(tb[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN]);
2546
2547         return 0;
2548 }
2549
2550 static int nl80211_req_set_reg(struct sk_buff *skb, struct genl_info *info)
2551 {
2552         int r;
2553         char *data = NULL;
2554
2555         /*
2556          * You should only get this when cfg80211 hasn't yet initialized
2557          * completely when built-in to the kernel right between the time
2558          * window between nl80211_init() and regulatory_init(), if that is
2559          * even possible.
2560          */
2561         mutex_lock(&cfg80211_mutex);
2562         if (unlikely(!cfg80211_regdomain)) {
2563                 mutex_unlock(&cfg80211_mutex);
2564                 return -EINPROGRESS;
2565         }
2566         mutex_unlock(&cfg80211_mutex);
2567
2568         if (!info->attrs[NL80211_ATTR_REG_ALPHA2])
2569                 return -EINVAL;
2570
2571         data = nla_data(info->attrs[NL80211_ATTR_REG_ALPHA2]);
2572
2573         r = regulatory_hint_user(data);
2574
2575         return r;
2576 }
2577
2578 static int nl80211_get_mesh_params(struct sk_buff *skb,
2579         struct genl_info *info)
2580 {
2581         struct cfg80211_registered_device *rdev;
2582         struct mesh_config cur_params;
2583         int err;
2584         struct net_device *dev;
2585         void *hdr;
2586         struct nlattr *pinfoattr;
2587         struct sk_buff *msg;
2588
2589         rtnl_lock();
2590
2591         /* Look up our device */
2592         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2593         if (err)
2594                 goto out_rtnl;
2595
2596         if (!rdev->ops->get_mesh_params) {
2597                 err = -EOPNOTSUPP;
2598                 goto out;
2599         }
2600
2601         /* Get the mesh params */
2602         err = rdev->ops->get_mesh_params(&rdev->wiphy, dev, &cur_params);
2603         if (err)
2604                 goto out;
2605
2606         /* Draw up a netlink message to send back */
2607         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2608         if (!msg) {
2609                 err = -ENOBUFS;
2610                 goto out;
2611         }
2612         hdr = nl80211hdr_put(msg, info->snd_pid, info->snd_seq, 0,
2613                              NL80211_CMD_GET_MESH_PARAMS);
2614         if (!hdr)
2615                 goto nla_put_failure;
2616         pinfoattr = nla_nest_start(msg, NL80211_ATTR_MESH_PARAMS);
2617         if (!pinfoattr)
2618                 goto nla_put_failure;
2619         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, dev->ifindex);
2620         NLA_PUT_U16(msg, NL80211_MESHCONF_RETRY_TIMEOUT,
2621                         cur_params.dot11MeshRetryTimeout);
2622         NLA_PUT_U16(msg, NL80211_MESHCONF_CONFIRM_TIMEOUT,
2623                         cur_params.dot11MeshConfirmTimeout);
2624         NLA_PUT_U16(msg, NL80211_MESHCONF_HOLDING_TIMEOUT,
2625                         cur_params.dot11MeshHoldingTimeout);
2626         NLA_PUT_U16(msg, NL80211_MESHCONF_MAX_PEER_LINKS,
2627                         cur_params.dot11MeshMaxPeerLinks);
2628         NLA_PUT_U8(msg, NL80211_MESHCONF_MAX_RETRIES,
2629                         cur_params.dot11MeshMaxRetries);
2630         NLA_PUT_U8(msg, NL80211_MESHCONF_TTL,
2631                         cur_params.dot11MeshTTL);
2632         NLA_PUT_U8(msg, NL80211_MESHCONF_AUTO_OPEN_PLINKS,
2633                         cur_params.auto_open_plinks);
2634         NLA_PUT_U8(msg, NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES,
2635                         cur_params.dot11MeshHWMPmaxPREQretries);
2636         NLA_PUT_U32(msg, NL80211_MESHCONF_PATH_REFRESH_TIME,
2637                         cur_params.path_refresh_time);
2638         NLA_PUT_U16(msg, NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT,
2639                         cur_params.min_discovery_timeout);
2640         NLA_PUT_U32(msg, NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT,
2641                         cur_params.dot11MeshHWMPactivePathTimeout);
2642         NLA_PUT_U16(msg, NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL,
2643                         cur_params.dot11MeshHWMPpreqMinInterval);
2644         NLA_PUT_U16(msg, NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
2645                         cur_params.dot11MeshHWMPnetDiameterTraversalTime);
2646         NLA_PUT_U8(msg, NL80211_MESHCONF_HWMP_ROOTMODE,
2647                         cur_params.dot11MeshHWMPRootMode);
2648         nla_nest_end(msg, pinfoattr);
2649         genlmsg_end(msg, hdr);
2650         err = genlmsg_reply(msg, info);
2651         goto out;
2652
2653  nla_put_failure:
2654         genlmsg_cancel(msg, hdr);
2655         err = -EMSGSIZE;
2656  out:
2657         /* Cleanup */
2658         cfg80211_unlock_rdev(rdev);
2659         dev_put(dev);
2660  out_rtnl:
2661         rtnl_unlock();
2662
2663         return err;
2664 }
2665
2666 #define FILL_IN_MESH_PARAM_IF_SET(table, cfg, param, mask, attr_num, nla_fn) \
2667 do {\
2668         if (table[attr_num]) {\
2669                 cfg.param = nla_fn(table[attr_num]); \
2670                 mask |= (1 << (attr_num - 1)); \
2671         } \
2672 } while (0);\
2673
2674 static struct nla_policy
2675 nl80211_meshconf_params_policy[NL80211_MESHCONF_ATTR_MAX+1] __read_mostly = {
2676         [NL80211_MESHCONF_RETRY_TIMEOUT] = { .type = NLA_U16 },
2677         [NL80211_MESHCONF_CONFIRM_TIMEOUT] = { .type = NLA_U16 },
2678         [NL80211_MESHCONF_HOLDING_TIMEOUT] = { .type = NLA_U16 },
2679         [NL80211_MESHCONF_MAX_PEER_LINKS] = { .type = NLA_U16 },
2680         [NL80211_MESHCONF_MAX_RETRIES] = { .type = NLA_U8 },
2681         [NL80211_MESHCONF_TTL] = { .type = NLA_U8 },
2682         [NL80211_MESHCONF_AUTO_OPEN_PLINKS] = { .type = NLA_U8 },
2683
2684         [NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES] = { .type = NLA_U8 },
2685         [NL80211_MESHCONF_PATH_REFRESH_TIME] = { .type = NLA_U32 },
2686         [NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT] = { .type = NLA_U16 },
2687         [NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT] = { .type = NLA_U32 },
2688         [NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL] = { .type = NLA_U16 },
2689         [NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME] = { .type = NLA_U16 },
2690 };
2691
2692 static int nl80211_set_mesh_params(struct sk_buff *skb, struct genl_info *info)
2693 {
2694         int err;
2695         u32 mask;
2696         struct cfg80211_registered_device *rdev;
2697         struct net_device *dev;
2698         struct mesh_config cfg;
2699         struct nlattr *tb[NL80211_MESHCONF_ATTR_MAX + 1];
2700         struct nlattr *parent_attr;
2701
2702         parent_attr = info->attrs[NL80211_ATTR_MESH_PARAMS];
2703         if (!parent_attr)
2704                 return -EINVAL;
2705         if (nla_parse_nested(tb, NL80211_MESHCONF_ATTR_MAX,
2706                         parent_attr, nl80211_meshconf_params_policy))
2707                 return -EINVAL;
2708
2709         rtnl_lock();
2710
2711         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2712         if (err)
2713                 goto out_rtnl;
2714
2715         if (!rdev->ops->set_mesh_params) {
2716                 err = -EOPNOTSUPP;
2717                 goto out;
2718         }
2719
2720         /* This makes sure that there aren't more than 32 mesh config
2721          * parameters (otherwise our bitfield scheme would not work.) */
2722         BUILD_BUG_ON(NL80211_MESHCONF_ATTR_MAX > 32);
2723
2724         /* Fill in the params struct */
2725         mask = 0;
2726         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshRetryTimeout,
2727                         mask, NL80211_MESHCONF_RETRY_TIMEOUT, nla_get_u16);
2728         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshConfirmTimeout,
2729                         mask, NL80211_MESHCONF_CONFIRM_TIMEOUT, nla_get_u16);
2730         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHoldingTimeout,
2731                         mask, NL80211_MESHCONF_HOLDING_TIMEOUT, nla_get_u16);
2732         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshMaxPeerLinks,
2733                         mask, NL80211_MESHCONF_MAX_PEER_LINKS, nla_get_u16);
2734         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshMaxRetries,
2735                         mask, NL80211_MESHCONF_MAX_RETRIES, nla_get_u8);
2736         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshTTL,
2737                         mask, NL80211_MESHCONF_TTL, nla_get_u8);
2738         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, auto_open_plinks,
2739                         mask, NL80211_MESHCONF_AUTO_OPEN_PLINKS, nla_get_u8);
2740         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPmaxPREQretries,
2741                         mask, NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES,
2742                         nla_get_u8);
2743         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, path_refresh_time,
2744                         mask, NL80211_MESHCONF_PATH_REFRESH_TIME, nla_get_u32);
2745         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, min_discovery_timeout,
2746                         mask, NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT,
2747                         nla_get_u16);
2748         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPactivePathTimeout,
2749                         mask, NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT,
2750                         nla_get_u32);
2751         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPpreqMinInterval,
2752                         mask, NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL,
2753                         nla_get_u16);
2754         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
2755                         dot11MeshHWMPnetDiameterTraversalTime,
2756                         mask, NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
2757                         nla_get_u16);
2758         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
2759                         dot11MeshHWMPRootMode, mask,
2760                         NL80211_MESHCONF_HWMP_ROOTMODE,
2761                         nla_get_u8);
2762
2763         /* Apply changes */
2764         err = rdev->ops->set_mesh_params(&rdev->wiphy, dev, &cfg, mask);
2765
2766  out:
2767         /* cleanup */
2768         cfg80211_unlock_rdev(rdev);
2769         dev_put(dev);
2770  out_rtnl:
2771         rtnl_unlock();
2772
2773         return err;
2774 }
2775
2776 #undef FILL_IN_MESH_PARAM_IF_SET
2777
2778 static int nl80211_get_reg(struct sk_buff *skb, struct genl_info *info)
2779 {
2780         struct sk_buff *msg;
2781         void *hdr = NULL;
2782         struct nlattr *nl_reg_rules;
2783         unsigned int i;
2784         int err = -EINVAL;
2785
2786         mutex_lock(&cfg80211_mutex);
2787
2788         if (!cfg80211_regdomain)
2789                 goto out;
2790
2791         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2792         if (!msg) {
2793                 err = -ENOBUFS;
2794                 goto out;
2795         }
2796
2797         hdr = nl80211hdr_put(msg, info->snd_pid, info->snd_seq, 0,
2798                              NL80211_CMD_GET_REG);
2799         if (!hdr)
2800                 goto nla_put_failure;
2801
2802         NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2,
2803                 cfg80211_regdomain->alpha2);
2804
2805         nl_reg_rules = nla_nest_start(msg, NL80211_ATTR_REG_RULES);
2806         if (!nl_reg_rules)
2807                 goto nla_put_failure;
2808
2809         for (i = 0; i < cfg80211_regdomain->n_reg_rules; i++) {
2810                 struct nlattr *nl_reg_rule;
2811                 const struct ieee80211_reg_rule *reg_rule;
2812                 const struct ieee80211_freq_range *freq_range;
2813                 const struct ieee80211_power_rule *power_rule;
2814
2815                 reg_rule = &cfg80211_regdomain->reg_rules[i];
2816                 freq_range = &reg_rule->freq_range;
2817                 power_rule = &reg_rule->power_rule;
2818
2819                 nl_reg_rule = nla_nest_start(msg, i);
2820                 if (!nl_reg_rule)
2821                         goto nla_put_failure;
2822
2823                 NLA_PUT_U32(msg, NL80211_ATTR_REG_RULE_FLAGS,
2824                         reg_rule->flags);
2825                 NLA_PUT_U32(msg, NL80211_ATTR_FREQ_RANGE_START,
2826                         freq_range->start_freq_khz);
2827                 NLA_PUT_U32(msg, NL80211_ATTR_FREQ_RANGE_END,
2828                         freq_range->end_freq_khz);
2829                 NLA_PUT_U32(msg, NL80211_ATTR_FREQ_RANGE_MAX_BW,
2830                         freq_range->max_bandwidth_khz);
2831                 NLA_PUT_U32(msg, NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN,
2832                         power_rule->max_antenna_gain);
2833                 NLA_PUT_U32(msg, NL80211_ATTR_POWER_RULE_MAX_EIRP,
2834                         power_rule->max_eirp);
2835
2836                 nla_nest_end(msg, nl_reg_rule);
2837         }
2838
2839         nla_nest_end(msg, nl_reg_rules);
2840
2841         genlmsg_end(msg, hdr);
2842         err = genlmsg_reply(msg, info);
2843         goto out;
2844
2845 nla_put_failure:
2846         genlmsg_cancel(msg, hdr);
2847         err = -EMSGSIZE;
2848 out:
2849         mutex_unlock(&cfg80211_mutex);
2850         return err;
2851 }
2852
2853 static int nl80211_set_reg(struct sk_buff *skb, struct genl_info *info)
2854 {
2855         struct nlattr *tb[NL80211_REG_RULE_ATTR_MAX + 1];
2856         struct nlattr *nl_reg_rule;
2857         char *alpha2 = NULL;
2858         int rem_reg_rules = 0, r = 0;
2859         u32 num_rules = 0, rule_idx = 0, size_of_regd;
2860         struct ieee80211_regdomain *rd = NULL;
2861
2862         if (!info->attrs[NL80211_ATTR_REG_ALPHA2])
2863                 return -EINVAL;
2864
2865         if (!info->attrs[NL80211_ATTR_REG_RULES])
2866                 return -EINVAL;
2867
2868         alpha2 = nla_data(info->attrs[NL80211_ATTR_REG_ALPHA2]);
2869
2870         nla_for_each_nested(nl_reg_rule, info->attrs[NL80211_ATTR_REG_RULES],
2871                         rem_reg_rules) {
2872                 num_rules++;
2873                 if (num_rules > NL80211_MAX_SUPP_REG_RULES)
2874                         return -EINVAL;
2875         }
2876
2877         mutex_lock(&cfg80211_mutex);
2878
2879         if (!reg_is_valid_request(alpha2)) {
2880                 r = -EINVAL;
2881                 goto bad_reg;
2882         }
2883
2884         size_of_regd = sizeof(struct ieee80211_regdomain) +
2885                 (num_rules * sizeof(struct ieee80211_reg_rule));
2886
2887         rd = kzalloc(size_of_regd, GFP_KERNEL);
2888         if (!rd) {
2889                 r = -ENOMEM;
2890                 goto bad_reg;
2891         }
2892
2893         rd->n_reg_rules = num_rules;
2894         rd->alpha2[0] = alpha2[0];
2895         rd->alpha2[1] = alpha2[1];
2896
2897         nla_for_each_nested(nl_reg_rule, info->attrs[NL80211_ATTR_REG_RULES],
2898                         rem_reg_rules) {
2899                 nla_parse(tb, NL80211_REG_RULE_ATTR_MAX,
2900                         nla_data(nl_reg_rule), nla_len(nl_reg_rule),
2901                         reg_rule_policy);
2902                 r = parse_reg_rule(tb, &rd->reg_rules[rule_idx]);
2903                 if (r)
2904                         goto bad_reg;
2905
2906                 rule_idx++;
2907
2908                 if (rule_idx > NL80211_MAX_SUPP_REG_RULES) {
2909                         r = -EINVAL;
2910                         goto bad_reg;
2911                 }
2912         }
2913
2914         BUG_ON(rule_idx != num_rules);
2915
2916         r = set_regdom(rd);
2917
2918         mutex_unlock(&cfg80211_mutex);
2919
2920         return r;
2921
2922  bad_reg:
2923         mutex_unlock(&cfg80211_mutex);
2924         kfree(rd);
2925         return r;
2926 }
2927
2928 static int validate_scan_freqs(struct nlattr *freqs)
2929 {
2930         struct nlattr *attr1, *attr2;
2931         int n_channels = 0, tmp1, tmp2;
2932
2933         nla_for_each_nested(attr1, freqs, tmp1) {
2934                 n_channels++;
2935                 /*
2936                  * Some hardware has a limited channel list for
2937                  * scanning, and it is pretty much nonsensical
2938                  * to scan for a channel twice, so disallow that
2939                  * and don't require drivers to check that the
2940                  * channel list they get isn't longer than what
2941                  * they can scan, as long as they can scan all
2942                  * the channels they registered at once.
2943                  */
2944                 nla_for_each_nested(attr2, freqs, tmp2)
2945                         if (attr1 != attr2 &&
2946                             nla_get_u32(attr1) == nla_get_u32(attr2))
2947                                 return 0;
2948         }
2949
2950         return n_channels;
2951 }
2952
2953 static int nl80211_trigger_scan(struct sk_buff *skb, struct genl_info *info)
2954 {
2955         struct cfg80211_registered_device *rdev;
2956         struct net_device *dev;
2957         struct cfg80211_scan_request *request;
2958         struct cfg80211_ssid *ssid;
2959         struct ieee80211_channel *channel;
2960         struct nlattr *attr;
2961         struct wiphy *wiphy;
2962         int err, tmp, n_ssids = 0, n_channels, i;
2963         enum ieee80211_band band;
2964         size_t ie_len;
2965
2966         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
2967                 return -EINVAL;
2968
2969         rtnl_lock();
2970
2971         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
2972         if (err)
2973                 goto out_rtnl;
2974
2975         wiphy = &rdev->wiphy;
2976
2977         if (!rdev->ops->scan) {
2978                 err = -EOPNOTSUPP;
2979                 goto out;
2980         }
2981
2982         if (!netif_running(dev)) {
2983                 err = -ENETDOWN;
2984                 goto out;
2985         }
2986
2987         if (rdev->scan_req) {
2988                 err = -EBUSY;
2989                 goto out;
2990         }
2991
2992         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
2993                 n_channels = validate_scan_freqs(
2994                                 info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]);
2995                 if (!n_channels) {
2996                         err = -EINVAL;
2997                         goto out;
2998                 }
2999         } else {
3000                 n_channels = 0;
3001
3002                 for (band = 0; band < IEEE80211_NUM_BANDS; band++)
3003                         if (wiphy->bands[band])
3004                                 n_channels += wiphy->bands[band]->n_channels;
3005         }
3006
3007         if (info->attrs[NL80211_ATTR_SCAN_SSIDS])
3008                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS], tmp)
3009                         n_ssids++;
3010
3011         if (n_ssids > wiphy->max_scan_ssids) {
3012                 err = -EINVAL;
3013                 goto out;
3014         }
3015
3016         if (info->attrs[NL80211_ATTR_IE])
3017                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
3018         else
3019                 ie_len = 0;
3020
3021         if (ie_len > wiphy->max_scan_ie_len) {
3022                 err = -EINVAL;
3023                 goto out;
3024         }
3025
3026         request = kzalloc(sizeof(*request)
3027                         + sizeof(*ssid) * n_ssids
3028                         + sizeof(channel) * n_channels
3029                         + ie_len, GFP_KERNEL);
3030         if (!request) {
3031                 err = -ENOMEM;
3032                 goto out;
3033         }
3034
3035         if (n_ssids)
3036                 request->ssids = (void *)&request->channels[n_channels];
3037         request->n_ssids = n_ssids;
3038         if (ie_len) {
3039                 if (request->ssids)
3040                         request->ie = (void *)(request->ssids + n_ssids);
3041                 else
3042                         request->ie = (void *)(request->channels + n_channels);
3043         }
3044
3045         i = 0;
3046         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
3047                 /* user specified, bail out if channel not found */
3048                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_FREQUENCIES], tmp) {
3049                         struct ieee80211_channel *chan;
3050
3051                         chan = ieee80211_get_channel(wiphy, nla_get_u32(attr));
3052
3053                         if (!chan) {
3054                                 err = -EINVAL;
3055                                 goto out_free;
3056                         }
3057
3058                         /* ignore disabled channels */
3059                         if (chan->flags & IEEE80211_CHAN_DISABLED)
3060                                 continue;
3061
3062                         request->channels[i] = chan;
3063                         i++;
3064                 }
3065         } else {
3066                 /* all channels */
3067                 for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
3068                         int j;
3069                         if (!wiphy->bands[band])
3070                                 continue;
3071                         for (j = 0; j < wiphy->bands[band]->n_channels; j++) {
3072                                 struct ieee80211_channel *chan;
3073
3074                                 chan = &wiphy->bands[band]->channels[j];
3075
3076                                 if (chan->flags & IEEE80211_CHAN_DISABLED)
3077                                         continue;
3078
3079                                 request->channels[i] = chan;
3080                                 i++;
3081                         }
3082                 }
3083         }
3084
3085         if (!i) {
3086                 err = -EINVAL;
3087                 goto out_free;
3088         }
3089
3090         request->n_channels = i;
3091
3092         i = 0;
3093         if (info->attrs[NL80211_ATTR_SCAN_SSIDS]) {
3094                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS], tmp) {
3095                         if (request->ssids[i].ssid_len > IEEE80211_MAX_SSID_LEN) {
3096                                 err = -EINVAL;
3097                                 goto out_free;
3098                         }
3099                         memcpy(request->ssids[i].ssid, nla_data(attr), nla_len(attr));
3100                         request->ssids[i].ssid_len = nla_len(attr);
3101                         i++;
3102                 }
3103         }
3104
3105         if (info->attrs[NL80211_ATTR_IE]) {
3106                 request->ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
3107                 memcpy((void *)request->ie,
3108                        nla_data(info->attrs[NL80211_ATTR_IE]),
3109                        request->ie_len);
3110         }
3111
3112         request->dev = dev;
3113         request->wiphy = &rdev->wiphy;
3114
3115         rdev->scan_req = request;
3116         err = rdev->ops->scan(&rdev->wiphy, dev, request);
3117
3118         if (!err) {
3119                 nl80211_send_scan_start(rdev, dev);
3120                 dev_hold(dev);
3121         }
3122
3123  out_free:
3124         if (err) {
3125                 rdev->scan_req = NULL;
3126                 kfree(request);
3127         }
3128  out:
3129         cfg80211_unlock_rdev(rdev);
3130         dev_put(dev);
3131  out_rtnl:
3132         rtnl_unlock();
3133
3134         return err;
3135 }
3136
3137 static int nl80211_send_bss(struct sk_buff *msg, u32 pid, u32 seq, int flags,
3138                             struct cfg80211_registered_device *rdev,
3139                             struct wireless_dev *wdev,
3140                             struct cfg80211_internal_bss *intbss)
3141 {
3142         struct cfg80211_bss *res = &intbss->pub;
3143         void *hdr;
3144         struct nlattr *bss;
3145         int i;
3146
3147         ASSERT_WDEV_LOCK(wdev);
3148
3149         hdr = nl80211hdr_put(msg, pid, seq, flags,
3150                              NL80211_CMD_NEW_SCAN_RESULTS);
3151         if (!hdr)
3152                 return -1;
3153
3154         NLA_PUT_U32(msg, NL80211_ATTR_GENERATION, rdev->bss_generation);
3155         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, wdev->netdev->ifindex);
3156
3157         bss = nla_nest_start(msg, NL80211_ATTR_BSS);
3158         if (!bss)
3159                 goto nla_put_failure;
3160         if (!is_zero_ether_addr(res->bssid))
3161                 NLA_PUT(msg, NL80211_BSS_BSSID, ETH_ALEN, res->bssid);
3162         if (res->information_elements && res->len_information_elements)
3163                 NLA_PUT(msg, NL80211_BSS_INFORMATION_ELEMENTS,
3164                         res->len_information_elements,
3165                         res->information_elements);
3166         if (res->beacon_ies && res->len_beacon_ies &&
3167             res->beacon_ies != res->information_elements)
3168                 NLA_PUT(msg, NL80211_BSS_BEACON_IES,
3169                         res->len_beacon_ies, res->beacon_ies);
3170         if (res->tsf)
3171                 NLA_PUT_U64(msg, NL80211_BSS_TSF, res->tsf);
3172         if (res->beacon_interval)
3173                 NLA_PUT_U16(msg, NL80211_BSS_BEACON_INTERVAL, res->beacon_interval);
3174         NLA_PUT_U16(msg, NL80211_BSS_CAPABILITY, res->capability);
3175         NLA_PUT_U32(msg, NL80211_BSS_FREQUENCY, res->channel->center_freq);
3176         NLA_PUT_U32(msg, NL80211_BSS_SEEN_MS_AGO,
3177                 jiffies_to_msecs(jiffies - intbss->ts));
3178
3179         switch (rdev->wiphy.signal_type) {
3180         case CFG80211_SIGNAL_TYPE_MBM:
3181                 NLA_PUT_U32(msg, NL80211_BSS_SIGNAL_MBM, res->signal);
3182                 break;
3183         case CFG80211_SIGNAL_TYPE_UNSPEC:
3184                 NLA_PUT_U8(msg, NL80211_BSS_SIGNAL_UNSPEC, res->signal);
3185                 break;
3186         default:
3187                 break;
3188         }
3189
3190         switch (wdev->iftype) {
3191         case NL80211_IFTYPE_STATION:
3192                 if (intbss == wdev->current_bss)
3193                         NLA_PUT_U32(msg, NL80211_BSS_STATUS,
3194                                     NL80211_BSS_STATUS_ASSOCIATED);
3195                 else for (i = 0; i < MAX_AUTH_BSSES; i++) {
3196                         if (intbss != wdev->auth_bsses[i])
3197                                 continue;
3198                         NLA_PUT_U32(msg, NL80211_BSS_STATUS,
3199                                     NL80211_BSS_STATUS_AUTHENTICATED);
3200                         break;
3201                 }
3202                 break;
3203         case NL80211_IFTYPE_ADHOC:
3204                 if (intbss == wdev->current_bss)
3205                         NLA_PUT_U32(msg, NL80211_BSS_STATUS,
3206                                     NL80211_BSS_STATUS_IBSS_JOINED);
3207                 break;
3208         default:
3209                 break;
3210         }
3211
3212         nla_nest_end(msg, bss);
3213
3214         return genlmsg_end(msg, hdr);
3215
3216  nla_put_failure:
3217         genlmsg_cancel(msg, hdr);
3218         return -EMSGSIZE;
3219 }
3220
3221 static int nl80211_dump_scan(struct sk_buff *skb,
3222                              struct netlink_callback *cb)
3223 {
3224         struct cfg80211_registered_device *rdev;
3225         struct net_device *dev;
3226         struct cfg80211_internal_bss *scan;
3227         struct wireless_dev *wdev;
3228         int ifidx = cb->args[0];
3229         int start = cb->args[1], idx = 0;
3230         int err;
3231
3232         if (!ifidx)
3233                 ifidx = nl80211_get_ifidx(cb);
3234         if (ifidx < 0)
3235                 return ifidx;
3236         cb->args[0] = ifidx;
3237
3238         dev = dev_get_by_index(sock_net(skb->sk), ifidx);
3239         if (!dev)
3240                 return -ENODEV;
3241
3242         rdev = cfg80211_get_dev_from_ifindex(sock_net(skb->sk), ifidx);
3243         if (IS_ERR(rdev)) {
3244                 err = PTR_ERR(rdev);
3245                 goto out_put_netdev;
3246         }
3247
3248         wdev = dev->ieee80211_ptr;
3249
3250         wdev_lock(wdev);
3251         spin_lock_bh(&rdev->bss_lock);
3252         cfg80211_bss_expire(rdev);
3253
3254         list_for_each_entry(scan, &rdev->bss_list, list) {
3255                 if (++idx <= start)
3256                         continue;
3257                 if (nl80211_send_bss(skb,
3258                                 NETLINK_CB(cb->skb).pid,
3259                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
3260                                 rdev, wdev, scan) < 0) {
3261                         idx--;
3262                         goto out;
3263                 }
3264         }
3265
3266  out:
3267         spin_unlock_bh(&rdev->bss_lock);
3268         wdev_unlock(wdev);
3269
3270         cb->args[1] = idx;
3271         err = skb->len;
3272         cfg80211_unlock_rdev(rdev);
3273  out_put_netdev:
3274         dev_put(dev);
3275
3276         return err;
3277 }
3278
3279 static int nl80211_send_survey(struct sk_buff *msg, u32 pid, u32 seq,
3280                                 int flags, struct net_device *dev,
3281                                 struct survey_info *survey)
3282 {
3283         void *hdr;
3284         struct nlattr *infoattr;
3285
3286         /* Survey without a channel doesn't make sense */
3287         if (!survey->channel)
3288                 return -EINVAL;
3289
3290         hdr = nl80211hdr_put(msg, pid, seq, flags,
3291                              NL80211_CMD_NEW_SURVEY_RESULTS);
3292         if (!hdr)
3293                 return -ENOMEM;
3294
3295         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, dev->ifindex);
3296
3297         infoattr = nla_nest_start(msg, NL80211_ATTR_SURVEY_INFO);
3298         if (!infoattr)
3299                 goto nla_put_failure;
3300
3301         NLA_PUT_U32(msg, NL80211_SURVEY_INFO_FREQUENCY,
3302                     survey->channel->center_freq);
3303         if (survey->filled & SURVEY_INFO_NOISE_DBM)
3304                 NLA_PUT_U8(msg, NL80211_SURVEY_INFO_NOISE,
3305                             survey->noise);
3306
3307         nla_nest_end(msg, infoattr);
3308
3309         return genlmsg_end(msg, hdr);
3310
3311  nla_put_failure:
3312         genlmsg_cancel(msg, hdr);
3313         return -EMSGSIZE;
3314 }
3315
3316 static int nl80211_dump_survey(struct sk_buff *skb,
3317                         struct netlink_callback *cb)
3318 {
3319         struct survey_info survey;
3320         struct cfg80211_registered_device *dev;
3321         struct net_device *netdev;
3322         int ifidx = cb->args[0];
3323         int survey_idx = cb->args[1];
3324         int res;
3325
3326         if (!ifidx)
3327                 ifidx = nl80211_get_ifidx(cb);
3328         if (ifidx < 0)
3329                 return ifidx;
3330         cb->args[0] = ifidx;
3331
3332         rtnl_lock();
3333
3334         netdev = __dev_get_by_index(sock_net(skb->sk), ifidx);
3335         if (!netdev) {
3336                 res = -ENODEV;
3337                 goto out_rtnl;
3338         }
3339
3340         dev = cfg80211_get_dev_from_ifindex(sock_net(skb->sk), ifidx);
3341         if (IS_ERR(dev)) {
3342                 res = PTR_ERR(dev);
3343                 goto out_rtnl;
3344         }
3345
3346         if (!dev->ops->dump_survey) {
3347                 res = -EOPNOTSUPP;
3348                 goto out_err;
3349         }
3350
3351         while (1) {
3352                 res = dev->ops->dump_survey(&dev->wiphy, netdev, survey_idx,
3353                                             &survey);
3354                 if (res == -ENOENT)
3355                         break;
3356                 if (res)
3357                         goto out_err;
3358
3359                 if (nl80211_send_survey(skb,
3360                                 NETLINK_CB(cb->skb).pid,
3361                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
3362                                 netdev,
3363                                 &survey) < 0)
3364                         goto out;
3365                 survey_idx++;
3366         }
3367
3368  out:
3369         cb->args[1] = survey_idx;
3370         res = skb->len;
3371  out_err:
3372         cfg80211_unlock_rdev(dev);
3373  out_rtnl:
3374         rtnl_unlock();
3375
3376         return res;
3377 }
3378
3379 static bool nl80211_valid_auth_type(enum nl80211_auth_type auth_type)
3380 {
3381         return auth_type <= NL80211_AUTHTYPE_MAX;
3382 }
3383
3384 static bool nl80211_valid_wpa_versions(u32 wpa_versions)
3385 {
3386         return !(wpa_versions & ~(NL80211_WPA_VERSION_1 |
3387                                   NL80211_WPA_VERSION_2));
3388 }
3389
3390 static bool nl80211_valid_akm_suite(u32 akm)
3391 {
3392         return akm == WLAN_AKM_SUITE_8021X ||
3393                 akm == WLAN_AKM_SUITE_PSK;
3394 }
3395
3396 static bool nl80211_valid_cipher_suite(u32 cipher)
3397 {
3398         return cipher == WLAN_CIPHER_SUITE_WEP40 ||
3399                 cipher == WLAN_CIPHER_SUITE_WEP104 ||
3400                 cipher == WLAN_CIPHER_SUITE_TKIP ||
3401                 cipher == WLAN_CIPHER_SUITE_CCMP ||
3402                 cipher == WLAN_CIPHER_SUITE_AES_CMAC;
3403 }
3404
3405
3406 static int nl80211_authenticate(struct sk_buff *skb, struct genl_info *info)
3407 {
3408         struct cfg80211_registered_device *rdev;
3409         struct net_device *dev;
3410         struct ieee80211_channel *chan;
3411         const u8 *bssid, *ssid, *ie = NULL;
3412         int err, ssid_len, ie_len = 0;
3413         enum nl80211_auth_type auth_type;
3414         struct key_parse key;
3415
3416         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
3417                 return -EINVAL;
3418
3419         if (!info->attrs[NL80211_ATTR_MAC])
3420                 return -EINVAL;
3421
3422         if (!info->attrs[NL80211_ATTR_AUTH_TYPE])
3423                 return -EINVAL;
3424
3425         if (!info->attrs[NL80211_ATTR_SSID])
3426                 return -EINVAL;
3427
3428         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ])
3429                 return -EINVAL;
3430
3431         err = nl80211_parse_key(info, &key);
3432         if (err)
3433                 return err;
3434
3435         if (key.idx >= 0) {
3436                 if (!key.p.key || !key.p.key_len)
3437                         return -EINVAL;
3438                 if ((key.p.cipher != WLAN_CIPHER_SUITE_WEP40 ||
3439                      key.p.key_len != WLAN_KEY_LEN_WEP40) &&
3440                     (key.p.cipher != WLAN_CIPHER_SUITE_WEP104 ||
3441                      key.p.key_len != WLAN_KEY_LEN_WEP104))
3442                         return -EINVAL;
3443                 if (key.idx > 4)
3444                         return -EINVAL;
3445         } else {
3446                 key.p.key_len = 0;
3447                 key.p.key = NULL;
3448         }
3449
3450         rtnl_lock();
3451
3452         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
3453         if (err)
3454                 goto unlock_rtnl;
3455
3456         if (!rdev->ops->auth) {
3457                 err = -EOPNOTSUPP;
3458                 goto out;
3459         }
3460
3461         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION) {
3462                 err = -EOPNOTSUPP;
3463                 goto out;
3464         }
3465
3466         if (!netif_running(dev)) {
3467                 err = -ENETDOWN;
3468                 goto out;
3469         }
3470
3471         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
3472         chan = ieee80211_get_channel(&rdev->wiphy,
3473                 nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
3474         if (!chan || (chan->flags & IEEE80211_CHAN_DISABLED)) {
3475                 err = -EINVAL;
3476                 goto out;
3477         }
3478
3479         ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
3480         ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
3481
3482         if (info->attrs[NL80211_ATTR_IE]) {
3483                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
3484                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
3485         }
3486
3487         auth_type = nla_get_u32(info->attrs[NL80211_ATTR_AUTH_TYPE]);
3488         if (!nl80211_valid_auth_type(auth_type)) {
3489                 err = -EINVAL;
3490                 goto out;
3491         }
3492
3493         err = cfg80211_mlme_auth(rdev, dev, chan, auth_type, bssid,
3494                                  ssid, ssid_len, ie, ie_len,
3495                                  key.p.key, key.p.key_len, key.idx);
3496
3497 out:
3498         cfg80211_unlock_rdev(rdev);
3499         dev_put(dev);
3500 unlock_rtnl:
3501         rtnl_unlock();
3502         return err;
3503 }
3504
3505 static int nl80211_crypto_settings(struct genl_info *info,
3506                                    struct cfg80211_crypto_settings *settings,
3507                                    int cipher_limit)
3508 {
3509         memset(settings, 0, sizeof(*settings));
3510
3511         settings->control_port = info->attrs[NL80211_ATTR_CONTROL_PORT];
3512
3513         if (info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]) {
3514                 void *data;
3515                 int len, i;
3516
3517                 data = nla_data(info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]);
3518                 len = nla_len(info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]);
3519                 settings->n_ciphers_pairwise = len / sizeof(u32);
3520
3521                 if (len % sizeof(u32))
3522                         return -EINVAL;
3523
3524                 if (settings->n_ciphers_pairwise > cipher_limit)
3525                         return -EINVAL;
3526
3527                 memcpy(settings->ciphers_pairwise, data, len);
3528
3529                 for (i = 0; i < settings->n_ciphers_pairwise; i++)
3530                         if (!nl80211_valid_cipher_suite(
3531                                         settings->ciphers_pairwise[i]))
3532                                 return -EINVAL;
3533         }
3534
3535         if (info->attrs[NL80211_ATTR_CIPHER_SUITE_GROUP]) {
3536                 settings->cipher_group =
3537                         nla_get_u32(info->attrs[NL80211_ATTR_CIPHER_SUITE_GROUP]);
3538                 if (!nl80211_valid_cipher_suite(settings->cipher_group))
3539                         return -EINVAL;
3540         }
3541
3542         if (info->attrs[NL80211_ATTR_WPA_VERSIONS]) {
3543                 settings->wpa_versions =
3544                         nla_get_u32(info->attrs[NL80211_ATTR_WPA_VERSIONS]);
3545                 if (!nl80211_valid_wpa_versions(settings->wpa_versions))
3546                         return -EINVAL;
3547         }
3548
3549         if (info->attrs[NL80211_ATTR_AKM_SUITES]) {
3550                 void *data;
3551                 int len, i;
3552
3553                 data = nla_data(info->attrs[NL80211_ATTR_AKM_SUITES]);
3554                 len = nla_len(info->attrs[NL80211_ATTR_AKM_SUITES]);
3555                 settings->n_akm_suites = len / sizeof(u32);
3556
3557                 if (len % sizeof(u32))
3558                         return -EINVAL;
3559
3560                 memcpy(settings->akm_suites, data, len);
3561
3562                 for (i = 0; i < settings->n_ciphers_pairwise; i++)
3563                         if (!nl80211_valid_akm_suite(settings->akm_suites[i]))
3564                                 return -EINVAL;
3565         }
3566
3567         return 0;
3568 }
3569
3570 static int nl80211_associate(struct sk_buff *skb, struct genl_info *info)
3571 {
3572         struct cfg80211_registered_device *rdev;
3573         struct net_device *dev;
3574         struct cfg80211_crypto_settings crypto;
3575         struct ieee80211_channel *chan, *fixedchan;
3576         const u8 *bssid, *ssid, *ie = NULL, *prev_bssid = NULL;
3577         int err, ssid_len, ie_len = 0;
3578         bool use_mfp = false;
3579
3580         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
3581                 return -EINVAL;
3582
3583         if (!info->attrs[NL80211_ATTR_MAC] ||
3584             !info->attrs[NL80211_ATTR_SSID] ||
3585             !info->attrs[NL80211_ATTR_WIPHY_FREQ])
3586                 return -EINVAL;
3587
3588         rtnl_lock();
3589
3590         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
3591         if (err)
3592                 goto unlock_rtnl;
3593
3594         if (!rdev->ops->assoc) {
3595                 err = -EOPNOTSUPP;
3596                 goto out;
3597         }
3598
3599         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION) {
3600                 err = -EOPNOTSUPP;
3601                 goto out;
3602         }
3603
3604         if (!netif_running(dev)) {
3605                 err = -ENETDOWN;
3606                 goto out;
3607         }
3608
3609         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
3610
3611         chan = ieee80211_get_channel(&rdev->wiphy,
3612                 nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
3613         if (!chan || (chan->flags & IEEE80211_CHAN_DISABLED)) {
3614                 err = -EINVAL;
3615                 goto out;
3616         }
3617
3618         mutex_lock(&rdev->devlist_mtx);
3619         fixedchan = rdev_fixed_channel(rdev, NULL);
3620         if (fixedchan && chan != fixedchan) {
3621                 err = -EBUSY;
3622                 mutex_unlock(&rdev->devlist_mtx);
3623                 goto out;
3624         }
3625         mutex_unlock(&rdev->devlist_mtx);
3626
3627         ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
3628         ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
3629
3630         if (info->attrs[NL80211_ATTR_IE]) {
3631                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
3632                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
3633         }
3634
3635         if (info->attrs[NL80211_ATTR_USE_MFP]) {
3636                 enum nl80211_mfp mfp =
3637                         nla_get_u32(info->attrs[NL80211_ATTR_USE_MFP]);
3638                 if (mfp == NL80211_MFP_REQUIRED)
3639                         use_mfp = true;
3640                 else if (mfp != NL80211_MFP_NO) {
3641                         err = -EINVAL;
3642                         goto out;
3643                 }
3644         }
3645
3646         if (info->attrs[NL80211_ATTR_PREV_BSSID])
3647                 prev_bssid = nla_data(info->attrs[NL80211_ATTR_PREV_BSSID]);
3648
3649         err = nl80211_crypto_settings(info, &crypto, 1);
3650         if (!err)
3651                 err = cfg80211_mlme_assoc(rdev, dev, chan, bssid, prev_bssid,
3652                                           ssid, ssid_len, ie, ie_len, use_mfp,
3653                                           &crypto);
3654
3655 out:
3656         cfg80211_unlock_rdev(rdev);
3657         dev_put(dev);
3658 unlock_rtnl:
3659         rtnl_unlock();
3660         return err;
3661 }
3662
3663 static int nl80211_deauthenticate(struct sk_buff *skb, struct genl_info *info)
3664 {
3665         struct cfg80211_registered_device *rdev;
3666         struct net_device *dev;
3667         const u8 *ie = NULL, *bssid;
3668         int err, ie_len = 0;
3669         u16 reason_code;
3670
3671         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
3672                 return -EINVAL;
3673
3674         if (!info->attrs[NL80211_ATTR_MAC])
3675                 return -EINVAL;
3676
3677         if (!info->attrs[NL80211_ATTR_REASON_CODE])
3678                 return -EINVAL;
3679
3680         rtnl_lock();
3681
3682         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
3683         if (err)
3684                 goto unlock_rtnl;
3685
3686         if (!rdev->ops->deauth) {
3687                 err = -EOPNOTSUPP;
3688                 goto out;
3689         }
3690
3691         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION) {
3692                 err = -EOPNOTSUPP;
3693                 goto out;
3694         }
3695
3696         if (!netif_running(dev)) {
3697                 err = -ENETDOWN;
3698                 goto out;
3699         }
3700
3701         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
3702
3703         reason_code = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
3704         if (reason_code == 0) {
3705                 /* Reason Code 0 is reserved */
3706                 err = -EINVAL;
3707                 goto out;
3708         }
3709
3710         if (info->attrs[NL80211_ATTR_IE]) {
3711                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
3712                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
3713         }
3714
3715         err = cfg80211_mlme_deauth(rdev, dev, bssid, ie, ie_len, reason_code);
3716
3717 out:
3718         cfg80211_unlock_rdev(rdev);
3719         dev_put(dev);
3720 unlock_rtnl:
3721         rtnl_unlock();
3722         return err;
3723 }
3724
3725 static int nl80211_disassociate(struct sk_buff *skb, struct genl_info *info)
3726 {
3727         struct cfg80211_registered_device *rdev;
3728         struct net_device *dev;
3729         const u8 *ie = NULL, *bssid;
3730         int err, ie_len = 0;
3731         u16 reason_code;
3732
3733         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
3734                 return -EINVAL;
3735
3736         if (!info->attrs[NL80211_ATTR_MAC])
3737                 return -EINVAL;
3738
3739         if (!info->attrs[NL80211_ATTR_REASON_CODE])
3740                 return -EINVAL;
3741
3742         rtnl_lock();
3743
3744         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
3745         if (err)
3746                 goto unlock_rtnl;
3747
3748         if (!rdev->ops->disassoc) {
3749                 err = -EOPNOTSUPP;
3750                 goto out;
3751         }
3752
3753         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION) {
3754                 err = -EOPNOTSUPP;
3755                 goto out;
3756         }
3757
3758         if (!netif_running(dev)) {
3759                 err = -ENETDOWN;
3760                 goto out;
3761         }
3762
3763         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
3764
3765         reason_code = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
3766         if (reason_code == 0) {
3767                 /* Reason Code 0 is reserved */
3768                 err = -EINVAL;
3769                 goto out;
3770         }
3771
3772         if (info->attrs[NL80211_ATTR_IE]) {
3773                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
3774                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
3775         }
3776
3777         err = cfg80211_mlme_disassoc(rdev, dev, bssid, ie, ie_len, reason_code);
3778
3779 out:
3780         cfg80211_unlock_rdev(rdev);
3781         dev_put(dev);
3782 unlock_rtnl:
3783         rtnl_unlock();
3784         return err;
3785 }
3786
3787 static int nl80211_join_ibss(struct sk_buff *skb, struct genl_info *info)
3788 {
3789         struct cfg80211_registered_device *rdev;
3790         struct net_device *dev;
3791         struct cfg80211_ibss_params ibss;
3792         struct wiphy *wiphy;
3793         struct cfg80211_cached_keys *connkeys = NULL;
3794         int err;
3795
3796         memset(&ibss, 0, sizeof(ibss));
3797
3798         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
3799                 return -EINVAL;
3800
3801         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ] ||
3802             !info->attrs[NL80211_ATTR_SSID] ||
3803             !nla_len(info->attrs[NL80211_ATTR_SSID]))
3804                 return -EINVAL;
3805
3806         ibss.beacon_interval = 100;
3807
3808         if (info->attrs[NL80211_ATTR_BEACON_INTERVAL]) {
3809                 ibss.beacon_interval =
3810                         nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
3811                 if (ibss.beacon_interval < 1 || ibss.beacon_interval > 10000)
3812                         return -EINVAL;
3813         }
3814
3815         rtnl_lock();
3816
3817         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
3818         if (err)
3819                 goto unlock_rtnl;
3820
3821         if (!rdev->ops->join_ibss) {
3822                 err = -EOPNOTSUPP;
3823                 goto out;
3824         }
3825
3826         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC) {
3827                 err = -EOPNOTSUPP;
3828                 goto out;
3829         }
3830
3831         if (!netif_running(dev)) {
3832                 err = -ENETDOWN;
3833                 goto out;
3834         }
3835
3836         wiphy = &rdev->wiphy;
3837
3838         if (info->attrs[NL80211_ATTR_MAC])
3839                 ibss.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
3840         ibss.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
3841         ibss.ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
3842
3843         if (info->attrs[NL80211_ATTR_IE]) {
3844                 ibss.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
3845                 ibss.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
3846         }
3847
3848         ibss.channel = ieee80211_get_channel(wiphy,
3849                 nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
3850         if (!ibss.channel ||
3851             ibss.channel->flags & IEEE80211_CHAN_NO_IBSS ||
3852             ibss.channel->flags & IEEE80211_CHAN_DISABLED) {
3853                 err = -EINVAL;
3854                 goto out;
3855         }
3856
3857         ibss.channel_fixed = !!info->attrs[NL80211_ATTR_FREQ_FIXED];
3858         ibss.privacy = !!info->attrs[NL80211_ATTR_PRIVACY];
3859
3860         if (ibss.privacy && info->attrs[NL80211_ATTR_KEYS]) {
3861                 connkeys = nl80211_parse_connkeys(rdev,
3862                                         info->attrs[NL80211_ATTR_KEYS]);
3863                 if (IS_ERR(connkeys)) {
3864                         err = PTR_ERR(connkeys);
3865                         connkeys = NULL;
3866                         goto out;
3867                 }
3868         }
3869
3870         err = cfg80211_join_ibss(rdev, dev, &ibss, connkeys);
3871
3872 out:
3873         cfg80211_unlock_rdev(rdev);
3874         dev_put(dev);
3875 unlock_rtnl:
3876         if (err)
3877                 kfree(connkeys);
3878         rtnl_unlock();
3879         return err;
3880 }
3881
3882 static int nl80211_leave_ibss(struct sk_buff *skb, struct genl_info *info)
3883 {
3884         struct cfg80211_registered_device *rdev;
3885         struct net_device *dev;
3886         int err;
3887
3888         rtnl_lock();
3889
3890         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
3891         if (err)
3892                 goto unlock_rtnl;
3893
3894         if (!rdev->ops->leave_ibss) {
3895                 err = -EOPNOTSUPP;
3896                 goto out;
3897         }
3898
3899         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC) {
3900                 err = -EOPNOTSUPP;
3901                 goto out;
3902         }
3903
3904         if (!netif_running(dev)) {
3905                 err = -ENETDOWN;
3906                 goto out;
3907         }
3908
3909         err = cfg80211_leave_ibss(rdev, dev, false);
3910
3911 out:
3912         cfg80211_unlock_rdev(rdev);
3913         dev_put(dev);
3914 unlock_rtnl:
3915         rtnl_unlock();
3916         return err;
3917 }
3918
3919 #ifdef CONFIG_NL80211_TESTMODE
3920 static struct genl_multicast_group nl80211_testmode_mcgrp = {
3921         .name = "testmode",
3922 };
3923
3924 static int nl80211_testmode_do(struct sk_buff *skb, struct genl_info *info)
3925 {
3926         struct cfg80211_registered_device *rdev;
3927         int err;
3928
3929         if (!info->attrs[NL80211_ATTR_TESTDATA])
3930                 return -EINVAL;
3931
3932         rtnl_lock();
3933
3934         rdev = cfg80211_get_dev_from_info(info);
3935         if (IS_ERR(rdev)) {
3936                 err = PTR_ERR(rdev);
3937                 goto unlock_rtnl;
3938         }
3939
3940         err = -EOPNOTSUPP;
3941         if (rdev->ops->testmode_cmd) {
3942                 rdev->testmode_info = info;
3943                 err = rdev->ops->testmode_cmd(&rdev->wiphy,
3944                                 nla_data(info->attrs[NL80211_ATTR_TESTDATA]),
3945                                 nla_len(info->attrs[NL80211_ATTR_TESTDATA]));
3946                 rdev->testmode_info = NULL;
3947         }
3948
3949         cfg80211_unlock_rdev(rdev);
3950
3951  unlock_rtnl:
3952         rtnl_unlock();
3953         return err;
3954 }
3955
3956 static struct sk_buff *
3957 __cfg80211_testmode_alloc_skb(struct cfg80211_registered_device *rdev,
3958                               int approxlen, u32 pid, u32 seq, gfp_t gfp)
3959 {
3960         struct sk_buff *skb;
3961         void *hdr;
3962         struct nlattr *data;
3963
3964         skb = nlmsg_new(approxlen + 100, gfp);
3965         if (!skb)
3966                 return NULL;
3967
3968         hdr = nl80211hdr_put(skb, pid, seq, 0, NL80211_CMD_TESTMODE);
3969         if (!hdr) {
3970                 kfree_skb(skb);
3971                 return NULL;
3972         }
3973
3974         NLA_PUT_U32(skb, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
3975         data = nla_nest_start(skb, NL80211_ATTR_TESTDATA);
3976
3977         ((void **)skb->cb)[0] = rdev;
3978         ((void **)skb->cb)[1] = hdr;
3979         ((void **)skb->cb)[2] = data;
3980
3981         return skb;
3982
3983  nla_put_failure:
3984         kfree_skb(skb);
3985         return NULL;
3986 }
3987
3988 struct sk_buff *cfg80211_testmode_alloc_reply_skb(struct wiphy *wiphy,
3989                                                   int approxlen)
3990 {
3991         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
3992
3993         if (WARN_ON(!rdev->testmode_info))
3994                 return NULL;
3995
3996         return __cfg80211_testmode_alloc_skb(rdev, approxlen,
3997                                 rdev->testmode_info->snd_pid,
3998                                 rdev->testmode_info->snd_seq,
3999                                 GFP_KERNEL);
4000 }
4001 EXPORT_SYMBOL(cfg80211_testmode_alloc_reply_skb);
4002
4003 int cfg80211_testmode_reply(struct sk_buff *skb)
4004 {
4005         struct cfg80211_registered_device *rdev = ((void **)skb->cb)[0];
4006         void *hdr = ((void **)skb->cb)[1];
4007         struct nlattr *data = ((void **)skb->cb)[2];
4008
4009         if (WARN_ON(!rdev->testmode_info)) {
4010                 kfree_skb(skb);
4011                 return -EINVAL;
4012         }
4013
4014         nla_nest_end(skb, data);
4015         genlmsg_end(skb, hdr);
4016         return genlmsg_reply(skb, rdev->testmode_info);
4017 }
4018 EXPORT_SYMBOL(cfg80211_testmode_reply);
4019
4020 struct sk_buff *cfg80211_testmode_alloc_event_skb(struct wiphy *wiphy,
4021                                                   int approxlen, gfp_t gfp)
4022 {
4023         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
4024
4025         return __cfg80211_testmode_alloc_skb(rdev, approxlen, 0, 0, gfp);
4026 }
4027 EXPORT_SYMBOL(cfg80211_testmode_alloc_event_skb);
4028
4029 void cfg80211_testmode_event(struct sk_buff *skb, gfp_t gfp)
4030 {
4031         void *hdr = ((void **)skb->cb)[1];
4032         struct nlattr *data = ((void **)skb->cb)[2];
4033
4034         nla_nest_end(skb, data);
4035         genlmsg_end(skb, hdr);
4036         genlmsg_multicast(skb, 0, nl80211_testmode_mcgrp.id, gfp);
4037 }
4038 EXPORT_SYMBOL(cfg80211_testmode_event);
4039 #endif
4040
4041 static int nl80211_connect(struct sk_buff *skb, struct genl_info *info)
4042 {
4043         struct cfg80211_registered_device *rdev;
4044         struct net_device *dev;
4045         struct cfg80211_connect_params connect;
4046         struct wiphy *wiphy;
4047         struct cfg80211_cached_keys *connkeys = NULL;
4048         int err;
4049
4050         memset(&connect, 0, sizeof(connect));
4051
4052         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
4053                 return -EINVAL;
4054
4055         if (!info->attrs[NL80211_ATTR_SSID] ||
4056             !nla_len(info->attrs[NL80211_ATTR_SSID]))
4057                 return -EINVAL;
4058
4059         if (info->attrs[NL80211_ATTR_AUTH_TYPE]) {
4060                 connect.auth_type =
4061                         nla_get_u32(info->attrs[NL80211_ATTR_AUTH_TYPE]);
4062                 if (!nl80211_valid_auth_type(connect.auth_type))
4063                         return -EINVAL;
4064         } else
4065                 connect.auth_type = NL80211_AUTHTYPE_AUTOMATIC;
4066
4067         connect.privacy = info->attrs[NL80211_ATTR_PRIVACY];
4068
4069         err = nl80211_crypto_settings(info, &connect.crypto,
4070                                       NL80211_MAX_NR_CIPHER_SUITES);
4071         if (err)
4072                 return err;
4073         rtnl_lock();
4074
4075         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
4076         if (err)
4077                 goto unlock_rtnl;
4078
4079         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION) {
4080                 err = -EOPNOTSUPP;
4081                 goto out;
4082         }
4083
4084         if (!netif_running(dev)) {
4085                 err = -ENETDOWN;
4086                 goto out;
4087         }
4088
4089         wiphy = &rdev->wiphy;
4090
4091         if (info->attrs[NL80211_ATTR_MAC])
4092                 connect.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
4093         connect.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
4094         connect.ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
4095
4096         if (info->attrs[NL80211_ATTR_IE]) {
4097                 connect.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
4098                 connect.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
4099         }
4100
4101         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
4102                 connect.channel =
4103                         ieee80211_get_channel(wiphy,
4104                             nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
4105                 if (!connect.channel ||
4106                     connect.channel->flags & IEEE80211_CHAN_DISABLED) {
4107                         err = -EINVAL;
4108                         goto out;
4109                 }
4110         }
4111
4112         if (connect.privacy && info->attrs[NL80211_ATTR_KEYS]) {
4113                 connkeys = nl80211_parse_connkeys(rdev,
4114                                         info->attrs[NL80211_ATTR_KEYS]);
4115                 if (IS_ERR(connkeys)) {
4116                         err = PTR_ERR(connkeys);
4117                         connkeys = NULL;
4118                         goto out;
4119                 }
4120         }
4121
4122         err = cfg80211_connect(rdev, dev, &connect, connkeys);
4123
4124 out:
4125         cfg80211_unlock_rdev(rdev);
4126         dev_put(dev);
4127 unlock_rtnl:
4128         if (err)
4129                 kfree(connkeys);
4130         rtnl_unlock();
4131         return err;
4132 }
4133
4134 static int nl80211_disconnect(struct sk_buff *skb, struct genl_info *info)
4135 {
4136         struct cfg80211_registered_device *rdev;
4137         struct net_device *dev;
4138         int err;
4139         u16 reason;
4140
4141         if (!info->attrs[NL80211_ATTR_REASON_CODE])
4142                 reason = WLAN_REASON_DEAUTH_LEAVING;
4143         else
4144                 reason = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
4145
4146         if (reason == 0)
4147                 return -EINVAL;
4148
4149         rtnl_lock();
4150
4151         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
4152         if (err)
4153                 goto unlock_rtnl;
4154
4155         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION) {
4156                 err = -EOPNOTSUPP;
4157                 goto out;
4158         }
4159
4160         if (!netif_running(dev)) {
4161                 err = -ENETDOWN;
4162                 goto out;
4163         }
4164
4165         err = cfg80211_disconnect(rdev, dev, reason, true);
4166
4167 out:
4168         cfg80211_unlock_rdev(rdev);
4169         dev_put(dev);
4170 unlock_rtnl:
4171         rtnl_unlock();
4172         return err;
4173 }
4174
4175 static int nl80211_wiphy_netns(struct sk_buff *skb, struct genl_info *info)
4176 {
4177         struct cfg80211_registered_device *rdev;
4178         struct net *net;
4179         int err;
4180         u32 pid;
4181
4182         if (!info->attrs[NL80211_ATTR_PID])
4183                 return -EINVAL;
4184
4185         pid = nla_get_u32(info->attrs[NL80211_ATTR_PID]);
4186
4187         rtnl_lock();
4188
4189         rdev = cfg80211_get_dev_from_info(info);
4190         if (IS_ERR(rdev)) {
4191                 err = PTR_ERR(rdev);
4192                 goto out_rtnl;
4193         }
4194
4195         net = get_net_ns_by_pid(pid);
4196         if (IS_ERR(net)) {
4197                 err = PTR_ERR(net);
4198                 goto out;
4199         }
4200
4201         err = 0;
4202
4203         /* check if anything to do */
4204         if (net_eq(wiphy_net(&rdev->wiphy), net))
4205                 goto out_put_net;
4206
4207         err = cfg80211_switch_netns(rdev, net);
4208  out_put_net:
4209         put_net(net);
4210  out:
4211         cfg80211_unlock_rdev(rdev);
4212  out_rtnl:
4213         rtnl_unlock();
4214         return err;
4215 }
4216
4217 static int nl80211_setdel_pmksa(struct sk_buff *skb, struct genl_info *info)
4218 {
4219         struct cfg80211_registered_device *rdev;
4220         int (*rdev_ops)(struct wiphy *wiphy, struct net_device *dev,
4221                         struct cfg80211_pmksa *pmksa) = NULL;
4222         int err;
4223         struct net_device *dev;
4224         struct cfg80211_pmksa pmksa;
4225
4226         memset(&pmksa, 0, sizeof(struct cfg80211_pmksa));
4227
4228         if (!info->attrs[NL80211_ATTR_MAC])
4229                 return -EINVAL;
4230
4231         if (!info->attrs[NL80211_ATTR_PMKID])
4232                 return -EINVAL;
4233
4234         rtnl_lock();
4235
4236         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
4237         if (err)
4238                 goto out_rtnl;
4239
4240         pmksa.pmkid = nla_data(info->attrs[NL80211_ATTR_PMKID]);
4241         pmksa.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
4242
4243         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION) {
4244                 err = -EOPNOTSUPP;
4245                 goto out;
4246         }
4247
4248         switch (info->genlhdr->cmd) {
4249         case NL80211_CMD_SET_PMKSA:
4250                 rdev_ops = rdev->ops->set_pmksa;
4251                 break;
4252         case NL80211_CMD_DEL_PMKSA:
4253                 rdev_ops = rdev->ops->del_pmksa;
4254                 break;
4255         default:
4256                 WARN_ON(1);
4257                 break;
4258         }
4259
4260         if (!rdev_ops) {
4261                 err = -EOPNOTSUPP;
4262                 goto out;
4263         }
4264
4265         err = rdev_ops(&rdev->wiphy, dev, &pmksa);
4266
4267  out:
4268         cfg80211_unlock_rdev(rdev);
4269         dev_put(dev);
4270  out_rtnl:
4271         rtnl_unlock();
4272
4273         return err;
4274 }
4275
4276 static int nl80211_flush_pmksa(struct sk_buff *skb, struct genl_info *info)
4277 {
4278         struct cfg80211_registered_device *rdev;
4279         int err;
4280         struct net_device *dev;
4281
4282         rtnl_lock();
4283
4284         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
4285         if (err)
4286                 goto out_rtnl;
4287
4288         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION) {
4289                 err = -EOPNOTSUPP;
4290                 goto out;
4291         }
4292
4293         if (!rdev->ops->flush_pmksa) {
4294                 err = -EOPNOTSUPP;
4295                 goto out;
4296         }
4297
4298         err = rdev->ops->flush_pmksa(&rdev->wiphy, dev);
4299
4300  out:
4301         cfg80211_unlock_rdev(rdev);
4302         dev_put(dev);
4303  out_rtnl:
4304         rtnl_unlock();
4305
4306         return err;
4307
4308 }
4309
4310 static int nl80211_remain_on_channel(struct sk_buff *skb,
4311                                      struct genl_info *info)
4312 {
4313         struct cfg80211_registered_device *rdev;
4314         struct net_device *dev;
4315         struct ieee80211_channel *chan;
4316         struct sk_buff *msg;
4317         void *hdr;
4318         u64 cookie;
4319         enum nl80211_channel_type channel_type = NL80211_CHAN_NO_HT;
4320         u32 freq, duration;
4321         int err;
4322
4323         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ] ||
4324             !info->attrs[NL80211_ATTR_DURATION])
4325                 return -EINVAL;
4326
4327         duration = nla_get_u32(info->attrs[NL80211_ATTR_DURATION]);
4328
4329         /*
4330          * We should be on that channel for at least one jiffie,
4331          * and more than 5 seconds seems excessive.
4332          */
4333         if (!duration || !msecs_to_jiffies(duration) || duration > 5000)
4334                 return -EINVAL;
4335
4336         rtnl_lock();
4337
4338         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
4339         if (err)
4340                 goto unlock_rtnl;
4341
4342         if (!rdev->ops->remain_on_channel) {
4343                 err = -EOPNOTSUPP;
4344                 goto out;
4345         }
4346
4347         if (!netif_running(dev)) {
4348                 err = -ENETDOWN;
4349                 goto out;
4350         }
4351
4352         if (info->attrs[NL80211_ATTR_WIPHY_CHANNEL_TYPE]) {
4353                 channel_type = nla_get_u32(
4354                         info->attrs[NL80211_ATTR_WIPHY_CHANNEL_TYPE]);
4355                 if (channel_type != NL80211_CHAN_NO_HT &&
4356                     channel_type != NL80211_CHAN_HT20 &&
4357                     channel_type != NL80211_CHAN_HT40PLUS &&
4358                     channel_type != NL80211_CHAN_HT40MINUS)
4359                         err = -EINVAL;
4360                         goto out;
4361         }
4362
4363         freq = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]);
4364         chan = rdev_freq_to_chan(rdev, freq, channel_type);
4365         if (chan == NULL) {
4366                 err = -EINVAL;
4367                 goto out;
4368         }
4369
4370         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4371         if (!msg) {
4372                 err = -ENOMEM;
4373                 goto out;
4374         }
4375
4376         hdr = nl80211hdr_put(msg, info->snd_pid, info->snd_seq, 0,
4377                              NL80211_CMD_REMAIN_ON_CHANNEL);
4378
4379         if (IS_ERR(hdr)) {
4380                 err = PTR_ERR(hdr);
4381                 goto free_msg;
4382         }
4383
4384         err = rdev->ops->remain_on_channel(&rdev->wiphy, dev, chan,
4385                                            channel_type, duration, &cookie);
4386
4387         if (err)
4388                 goto free_msg;
4389
4390         NLA_PUT_U64(msg, NL80211_ATTR_COOKIE, cookie);
4391
4392         genlmsg_end(msg, hdr);
4393         err = genlmsg_reply(msg, info);
4394         goto out;
4395
4396  nla_put_failure:
4397         err = -ENOBUFS;
4398  free_msg:
4399         nlmsg_free(msg);
4400  out:
4401         cfg80211_unlock_rdev(rdev);
4402         dev_put(dev);
4403  unlock_rtnl:
4404         rtnl_unlock();
4405         return err;
4406 }
4407
4408 static int nl80211_cancel_remain_on_channel(struct sk_buff *skb,
4409                                             struct genl_info *info)
4410 {
4411         struct cfg80211_registered_device *rdev;
4412         struct net_device *dev;
4413         u64 cookie;
4414         int err;
4415
4416         if (!info->attrs[NL80211_ATTR_COOKIE])
4417                 return -EINVAL;
4418
4419         rtnl_lock();
4420
4421         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
4422         if (err)
4423                 goto unlock_rtnl;
4424
4425         if (!rdev->ops->cancel_remain_on_channel) {
4426                 err = -EOPNOTSUPP;
4427                 goto out;
4428         }
4429
4430         if (!netif_running(dev)) {
4431                 err = -ENETDOWN;
4432                 goto out;
4433         }
4434
4435         cookie = nla_get_u64(info->attrs[NL80211_ATTR_COOKIE]);
4436
4437         err = rdev->ops->cancel_remain_on_channel(&rdev->wiphy, dev, cookie);
4438
4439  out:
4440         cfg80211_unlock_rdev(rdev);
4441         dev_put(dev);
4442  unlock_rtnl:
4443         rtnl_unlock();
4444         return err;
4445 }
4446
4447 static u32 rateset_to_mask(struct ieee80211_supported_band *sband,
4448                            u8 *rates, u8 rates_len)
4449 {
4450         u8 i;
4451         u32 mask = 0;
4452
4453         for (i = 0; i < rates_len; i++) {
4454                 int rate = (rates[i] & 0x7f) * 5;
4455                 int ridx;
4456                 for (ridx = 0; ridx < sband->n_bitrates; ridx++) {
4457                         struct ieee80211_rate *srate =
4458                                 &sband->bitrates[ridx];
4459                         if (rate == srate->bitrate) {
4460                                 mask |= 1 << ridx;
4461                                 break;
4462                         }
4463                 }
4464                 if (ridx == sband->n_bitrates)
4465                         return 0; /* rate not found */
4466         }
4467
4468         return mask;
4469 }
4470
4471 static struct nla_policy
4472 nl80211_txattr_policy[NL80211_TXRATE_MAX + 1] __read_mostly = {
4473         [NL80211_TXRATE_LEGACY] = { .type = NLA_BINARY,
4474                                     .len = NL80211_MAX_SUPP_RATES },
4475 };
4476
4477 static int nl80211_set_tx_bitrate_mask(struct sk_buff *skb,
4478                                        struct genl_info *info)
4479 {
4480         struct nlattr *tb[NL80211_TXRATE_MAX + 1];
4481         struct cfg80211_registered_device *rdev;
4482         struct cfg80211_bitrate_mask mask;
4483         int err, rem, i;
4484         struct net_device *dev;
4485         struct nlattr *tx_rates;
4486         struct ieee80211_supported_band *sband;
4487
4488         if (info->attrs[NL80211_ATTR_TX_RATES] == NULL)
4489                 return -EINVAL;
4490
4491         rtnl_lock();
4492
4493         err = get_rdev_dev_by_info_ifindex(info, &rdev, &dev);
4494         if (err)
4495                 goto unlock_rtnl;
4496
4497         if (!rdev->ops->set_bitrate_mask) {
4498                 err = -EOPNOTSUPP;
4499                 goto unlock;
4500         }
4501
4502         memset(&mask, 0, sizeof(mask));
4503         /* Default to all rates enabled */
4504         for (i = 0; i < IEEE80211_NUM_BANDS; i++) {
4505                 sband = rdev->wiphy.bands[i];
4506                 mask.control[i].legacy =
4507                         sband ? (1 << sband->n_bitrates) - 1 : 0;
4508         }
4509
4510         /*
4511          * The nested attribute uses enum nl80211_band as the index. This maps
4512          * directly to the enum ieee80211_band values used in cfg80211.
4513          */
4514         nla_for_each_nested(tx_rates, info->attrs[NL80211_ATTR_TX_RATES], rem)
4515         {
4516                 enum ieee80211_band band = nla_type(tx_rates);
4517                 if (band < 0 || band >= IEEE80211_NUM_BANDS) {
4518                         err = -EINVAL;
4519                         goto unlock;
4520                 }
4521                 sband = rdev->wiphy.bands[band];
4522                 if (sband == NULL) {
4523                         err = -EINVAL;
4524                         goto unlock;
4525                 }
4526                 nla_parse(tb, NL80211_TXRATE_MAX, nla_data(tx_rates),
4527                           nla_len(tx_rates), nl80211_txattr_policy);
4528                 if (tb[NL80211_TXRATE_LEGACY]) {
4529                         mask.control[band].legacy = rateset_to_mask(
4530                                 sband,
4531                                 nla_data(tb[NL80211_TXRATE_LEGACY]),
4532                                 nla_len(tb[NL80211_TXRATE_LEGACY]));
4533                         if (mask.control[band].legacy == 0) {
4534                                 err = -EINVAL;
4535                                 goto unlock;
4536                         }
4537                 }
4538         }
4539
4540         err = rdev->ops->set_bitrate_mask(&rdev->wiphy, dev, NULL, &mask);
4541
4542  unlock:
4543         dev_put(dev);
4544         cfg80211_unlock_rdev(rdev);
4545  unlock_rtnl:
4546         rtnl_unlock();
4547         return err;
4548 }
4549
4550 static struct genl_ops nl80211_ops[] = {
4551         {
4552                 .cmd = NL80211_CMD_GET_WIPHY,
4553                 .doit = nl80211_get_wiphy,
4554                 .dumpit = nl80211_dump_wiphy,
4555                 .policy = nl80211_policy,
4556                 /* can be retrieved by unprivileged users */
4557         },
4558         {
4559                 .cmd = NL80211_CMD_SET_WIPHY,
4560                 .doit = nl80211_set_wiphy,
4561                 .policy = nl80211_policy,
4562                 .flags = GENL_ADMIN_PERM,
4563         },
4564         {
4565                 .cmd = NL80211_CMD_GET_INTERFACE,
4566                 .doit = nl80211_get_interface,
4567                 .dumpit = nl80211_dump_interface,
4568                 .policy = nl80211_policy,
4569                 /* can be retrieved by unprivileged users */
4570         },
4571         {
4572                 .cmd = NL80211_CMD_SET_INTERFACE,
4573                 .doit = nl80211_set_interface,
4574                 .policy = nl80211_policy,
4575                 .flags = GENL_ADMIN_PERM,
4576         },
4577         {
4578                 .cmd = NL80211_CMD_NEW_INTERFACE,
4579                 .doit = nl80211_new_interface,
4580                 .policy = nl80211_policy,
4581                 .flags = GENL_ADMIN_PERM,
4582         },
4583         {
4584                 .cmd = NL80211_CMD_DEL_INTERFACE,
4585                 .doit = nl80211_del_interface,
4586                 .policy = nl80211_policy,
4587                 .flags = GENL_ADMIN_PERM,
4588         },
4589         {
4590                 .cmd = NL80211_CMD_GET_KEY,
4591                 .doit = nl80211_get_key,
4592                 .policy = nl80211_policy,
4593                 .flags = GENL_ADMIN_PERM,
4594         },
4595         {
4596                 .cmd = NL80211_CMD_SET_KEY,
4597                 .doit = nl80211_set_key,
4598                 .policy = nl80211_policy,
4599                 .flags = GENL_ADMIN_PERM,
4600         },
4601         {
4602                 .cmd = NL80211_CMD_NEW_KEY,
4603                 .doit = nl80211_new_key,
4604                 .policy = nl80211_policy,
4605                 .flags = GENL_ADMIN_PERM,
4606         },
4607         {
4608                 .cmd = NL80211_CMD_DEL_KEY,
4609                 .doit = nl80211_del_key,
4610                 .policy = nl80211_policy,
4611                 .flags = GENL_ADMIN_PERM,
4612         },
4613         {
4614                 .cmd = NL80211_CMD_SET_BEACON,
4615                 .policy = nl80211_policy,
4616                 .flags = GENL_ADMIN_PERM,
4617                 .doit = nl80211_addset_beacon,
4618         },
4619         {
4620                 .cmd = NL80211_CMD_NEW_BEACON,
4621                 .policy = nl80211_policy,
4622                 .flags = GENL_ADMIN_PERM,
4623                 .doit = nl80211_addset_beacon,
4624         },
4625         {
4626                 .cmd = NL80211_CMD_DEL_BEACON,
4627                 .policy = nl80211_policy,
4628                 .flags = GENL_ADMIN_PERM,
4629                 .doit = nl80211_del_beacon,
4630         },
4631         {
4632                 .cmd = NL80211_CMD_GET_STATION,
4633                 .doit = nl80211_get_station,
4634                 .dumpit = nl80211_dump_station,
4635                 .policy = nl80211_policy,
4636         },
4637         {
4638                 .cmd = NL80211_CMD_SET_STATION,
4639                 .doit = nl80211_set_station,
4640                 .policy = nl80211_policy,
4641                 .flags = GENL_ADMIN_PERM,
4642         },
4643         {
4644                 .cmd = NL80211_CMD_NEW_STATION,
4645                 .doit = nl80211_new_station,
4646                 .policy = nl80211_policy,
4647                 .flags = GENL_ADMIN_PERM,
4648         },
4649         {
4650                 .cmd = NL80211_CMD_DEL_STATION,
4651                 .doit = nl80211_del_station,
4652                 .policy = nl80211_policy,
4653                 .flags = GENL_ADMIN_PERM,
4654         },
4655         {
4656                 .cmd = NL80211_CMD_GET_MPATH,
4657                 .doit = nl80211_get_mpath,
4658                 .dumpit = nl80211_dump_mpath,
4659                 .policy = nl80211_policy,
4660                 .flags = GENL_ADMIN_PERM,
4661         },
4662         {
4663                 .cmd = NL80211_CMD_SET_MPATH,
4664                 .doit = nl80211_set_mpath,
4665                 .policy = nl80211_policy,
4666                 .flags = GENL_ADMIN_PERM,
4667         },
4668         {
4669                 .cmd = NL80211_CMD_NEW_MPATH,
4670                 .doit = nl80211_new_mpath,
4671                 .policy = nl80211_policy,
4672                 .flags = GENL_ADMIN_PERM,
4673         },
4674         {
4675                 .cmd = NL80211_CMD_DEL_MPATH,
4676                 .doit = nl80211_del_mpath,
4677                 .policy = nl80211_policy,
4678                 .flags = GENL_ADMIN_PERM,
4679         },
4680         {
4681                 .cmd = NL80211_CMD_SET_BSS,
4682                 .doit = nl80211_set_bss,
4683                 .policy = nl80211_policy,
4684                 .flags = GENL_ADMIN_PERM,
4685         },
4686         {
4687                 .cmd = NL80211_CMD_GET_REG,
4688                 .doit = nl80211_get_reg,
4689                 .policy = nl80211_policy,
4690                 /* can be retrieved by unprivileged users */
4691         },
4692         {
4693                 .cmd = NL80211_CMD_SET_REG,
4694                 .doit = nl80211_set_reg,
4695                 .policy = nl80211_policy,
4696                 .flags = GENL_ADMIN_PERM,
4697         },
4698         {
4699                 .cmd = NL80211_CMD_REQ_SET_REG,
4700                 .doit = nl80211_req_set_reg,
4701                 .policy = nl80211_policy,
4702                 .flags = GENL_ADMIN_PERM,
4703         },
4704         {
4705                 .cmd = NL80211_CMD_GET_MESH_PARAMS,
4706                 .doit = nl80211_get_mesh_params,
4707                 .policy = nl80211_policy,
4708                 /* can be retrieved by unprivileged users */
4709         },
4710         {
4711                 .cmd = NL80211_CMD_SET_MESH_PARAMS,
4712                 .doit = nl80211_set_mesh_params,
4713                 .policy = nl80211_policy,
4714                 .flags = GENL_ADMIN_PERM,
4715         },
4716         {
4717                 .cmd = NL80211_CMD_TRIGGER_SCAN,
4718                 .doit = nl80211_trigger_scan,
4719                 .policy = nl80211_policy,
4720                 .flags = GENL_ADMIN_PERM,
4721         },
4722         {
4723                 .cmd = NL80211_CMD_GET_SCAN,
4724                 .policy = nl80211_policy,
4725                 .dumpit = nl80211_dump_scan,
4726         },
4727         {
4728                 .cmd = NL80211_CMD_AUTHENTICATE,
4729                 .doit = nl80211_authenticate,
4730                 .policy = nl80211_policy,
4731                 .flags = GENL_ADMIN_PERM,
4732         },
4733         {
4734                 .cmd = NL80211_CMD_ASSOCIATE,
4735                 .doit = nl80211_associate,
4736                 .policy = nl80211_policy,
4737                 .flags = GENL_ADMIN_PERM,
4738         },
4739         {
4740                 .cmd = NL80211_CMD_DEAUTHENTICATE,
4741                 .doit = nl80211_deauthenticate,
4742                 .policy = nl80211_policy,
4743                 .flags = GENL_ADMIN_PERM,
4744         },
4745         {
4746                 .cmd = NL80211_CMD_DISASSOCIATE,
4747                 .doit = nl80211_disassociate,
4748                 .policy = nl80211_policy,
4749                 .flags = GENL_ADMIN_PERM,
4750         },
4751         {
4752                 .cmd = NL80211_CMD_JOIN_IBSS,
4753                 .doit = nl80211_join_ibss,
4754                 .policy = nl80211_policy,
4755                 .flags = GENL_ADMIN_PERM,
4756         },
4757         {
4758                 .cmd = NL80211_CMD_LEAVE_IBSS,
4759                 .doit = nl80211_leave_ibss,
4760                 .policy = nl80211_policy,
4761                 .flags = GENL_ADMIN_PERM,
4762         },
4763 #ifdef CONFIG_NL80211_TESTMODE
4764         {
4765                 .cmd = NL80211_CMD_TESTMODE,
4766                 .doit = nl80211_testmode_do,
4767                 .policy = nl80211_policy,
4768                 .flags = GENL_ADMIN_PERM,
4769         },
4770 #endif
4771         {
4772                 .cmd = NL80211_CMD_CONNECT,
4773                 .doit = nl80211_connect,
4774                 .policy = nl80211_policy,
4775                 .flags = GENL_ADMIN_PERM,
4776         },
4777         {
4778                 .cmd = NL80211_CMD_DISCONNECT,
4779                 .doit = nl80211_disconnect,
4780                 .policy = nl80211_policy,
4781                 .flags = GENL_ADMIN_PERM,
4782         },
4783         {
4784                 .cmd = NL80211_CMD_SET_WIPHY_NETNS,
4785                 .doit = nl80211_wiphy_netns,
4786                 .policy = nl80211_policy,
4787                 .flags = GENL_ADMIN_PERM,
4788         },
4789         {
4790                 .cmd = NL80211_CMD_GET_SURVEY,
4791                 .policy = nl80211_policy,
4792                 .dumpit = nl80211_dump_survey,
4793         },
4794         {
4795                 .cmd = NL80211_CMD_SET_PMKSA,
4796                 .doit = nl80211_setdel_pmksa,
4797                 .policy = nl80211_policy,
4798                 .flags = GENL_ADMIN_PERM,
4799         },
4800         {
4801                 .cmd = NL80211_CMD_DEL_PMKSA,
4802                 .doit = nl80211_setdel_pmksa,
4803                 .policy = nl80211_policy,
4804                 .flags = GENL_ADMIN_PERM,
4805         },
4806         {
4807                 .cmd = NL80211_CMD_FLUSH_PMKSA,
4808                 .doit = nl80211_flush_pmksa,
4809                 .policy = nl80211_policy,
4810                 .flags = GENL_ADMIN_PERM,
4811         },
4812         {
4813                 .cmd = NL80211_CMD_REMAIN_ON_CHANNEL,
4814                 .doit = nl80211_remain_on_channel,
4815                 .policy = nl80211_policy,
4816                 .flags = GENL_ADMIN_PERM,
4817         },
4818         {
4819                 .cmd = NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL,
4820                 .doit = nl80211_cancel_remain_on_channel,
4821                 .policy = nl80211_policy,
4822                 .flags = GENL_ADMIN_PERM,
4823         },
4824         {
4825                 .cmd = NL80211_CMD_SET_TX_BITRATE_MASK,
4826                 .doit = nl80211_set_tx_bitrate_mask,
4827                 .policy = nl80211_policy,
4828                 .flags = GENL_ADMIN_PERM,
4829         },
4830 };
4831
4832 static struct genl_multicast_group nl80211_mlme_mcgrp = {
4833         .name = "mlme",
4834 };
4835
4836 /* multicast groups */
4837 static struct genl_multicast_group nl80211_config_mcgrp = {
4838         .name = "config",
4839 };
4840 static struct genl_multicast_group nl80211_scan_mcgrp = {
4841         .name = "scan",
4842 };
4843 static struct genl_multicast_group nl80211_regulatory_mcgrp = {
4844         .name = "regulatory",
4845 };
4846
4847 /* notification functions */
4848
4849 void nl80211_notify_dev_rename(struct cfg80211_registered_device *rdev)
4850 {
4851         struct sk_buff *msg;
4852
4853         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4854         if (!msg)
4855                 return;
4856
4857         if (nl80211_send_wiphy(msg, 0, 0, 0, rdev) < 0) {
4858                 nlmsg_free(msg);
4859                 return;
4860         }
4861
4862         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
4863                                 nl80211_config_mcgrp.id, GFP_KERNEL);
4864 }
4865
4866 static int nl80211_add_scan_req(struct sk_buff *msg,
4867                                 struct cfg80211_registered_device *rdev)
4868 {
4869         struct cfg80211_scan_request *req = rdev->scan_req;
4870         struct nlattr *nest;
4871         int i;
4872
4873         ASSERT_RDEV_LOCK(rdev);
4874
4875         if (WARN_ON(!req))
4876                 return 0;
4877
4878         nest = nla_nest_start(msg, NL80211_ATTR_SCAN_SSIDS);
4879         if (!nest)
4880                 goto nla_put_failure;
4881         for (i = 0; i < req->n_ssids; i++)
4882                 NLA_PUT(msg, i, req->ssids[i].ssid_len, req->ssids[i].ssid);
4883         nla_nest_end(msg, nest);
4884
4885         nest = nla_nest_start(msg, NL80211_ATTR_SCAN_FREQUENCIES);
4886         if (!nest)
4887                 goto nla_put_failure;
4888         for (i = 0; i < req->n_channels; i++)
4889                 NLA_PUT_U32(msg, i, req->channels[i]->center_freq);
4890         nla_nest_end(msg, nest);
4891
4892         if (req->ie)
4893                 NLA_PUT(msg, NL80211_ATTR_IE, req->ie_len, req->ie);
4894
4895         return 0;
4896  nla_put_failure:
4897         return -ENOBUFS;
4898 }
4899
4900 static int nl80211_send_scan_msg(struct sk_buff *msg,
4901                                  struct cfg80211_registered_device *rdev,
4902                                  struct net_device *netdev,
4903                                  u32 pid, u32 seq, int flags,
4904                                  u32 cmd)
4905 {
4906         void *hdr;
4907
4908         hdr = nl80211hdr_put(msg, pid, seq, flags, cmd);
4909         if (!hdr)
4910                 return -1;
4911
4912         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
4913         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
4914
4915         /* ignore errors and send incomplete event anyway */
4916         nl80211_add_scan_req(msg, rdev);
4917
4918         return genlmsg_end(msg, hdr);
4919
4920  nla_put_failure:
4921         genlmsg_cancel(msg, hdr);
4922         return -EMSGSIZE;
4923 }
4924
4925 void nl80211_send_scan_start(struct cfg80211_registered_device *rdev,
4926                              struct net_device *netdev)
4927 {
4928         struct sk_buff *msg;
4929
4930         msg = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
4931         if (!msg)
4932                 return;
4933
4934         if (nl80211_send_scan_msg(msg, rdev, netdev, 0, 0, 0,
4935                                   NL80211_CMD_TRIGGER_SCAN) < 0) {
4936                 nlmsg_free(msg);
4937                 return;
4938         }
4939
4940         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
4941                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
4942 }
4943
4944 void nl80211_send_scan_done(struct cfg80211_registered_device *rdev,
4945                             struct net_device *netdev)
4946 {
4947         struct sk_buff *msg;
4948
4949         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4950         if (!msg)
4951                 return;
4952
4953         if (nl80211_send_scan_msg(msg, rdev, netdev, 0, 0, 0,
4954                                   NL80211_CMD_NEW_SCAN_RESULTS) < 0) {
4955                 nlmsg_free(msg);
4956                 return;
4957         }
4958
4959         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
4960                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
4961 }
4962
4963 void nl80211_send_scan_aborted(struct cfg80211_registered_device *rdev,
4964                                struct net_device *netdev)
4965 {
4966         struct sk_buff *msg;
4967
4968         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4969         if (!msg)
4970                 return;
4971
4972         if (nl80211_send_scan_msg(msg, rdev, netdev, 0, 0, 0,
4973                                   NL80211_CMD_SCAN_ABORTED) < 0) {
4974                 nlmsg_free(msg);
4975                 return;
4976         }
4977
4978         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
4979                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
4980 }
4981
4982 /*
4983  * This can happen on global regulatory changes or device specific settings
4984  * based on custom world regulatory domains.
4985  */
4986 void nl80211_send_reg_change_event(struct regulatory_request *request)
4987 {
4988         struct sk_buff *msg;
4989         void *hdr;
4990
4991         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4992         if (!msg)
4993                 return;
4994
4995         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_REG_CHANGE);
4996         if (!hdr) {
4997                 nlmsg_free(msg);
4998                 return;
4999         }
5000
5001         /* Userspace can always count this one always being set */
5002         NLA_PUT_U8(msg, NL80211_ATTR_REG_INITIATOR, request->initiator);
5003
5004         if (request->alpha2[0] == '0' && request->alpha2[1] == '0')
5005                 NLA_PUT_U8(msg, NL80211_ATTR_REG_TYPE,
5006                            NL80211_REGDOM_TYPE_WORLD);
5007         else if (request->alpha2[0] == '9' && request->alpha2[1] == '9')
5008                 NLA_PUT_U8(msg, NL80211_ATTR_REG_TYPE,
5009                            NL80211_REGDOM_TYPE_CUSTOM_WORLD);
5010         else if ((request->alpha2[0] == '9' && request->alpha2[1] == '8') ||
5011                  request->intersect)
5012                 NLA_PUT_U8(msg, NL80211_ATTR_REG_TYPE,
5013                            NL80211_REGDOM_TYPE_INTERSECTION);
5014         else {
5015                 NLA_PUT_U8(msg, NL80211_ATTR_REG_TYPE,
5016                            NL80211_REGDOM_TYPE_COUNTRY);
5017                 NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2, request->alpha2);
5018         }
5019
5020         if (wiphy_idx_valid(request->wiphy_idx))
5021                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, request->wiphy_idx);
5022
5023         if (genlmsg_end(msg, hdr) < 0) {
5024                 nlmsg_free(msg);
5025                 return;
5026         }
5027
5028         rcu_read_lock();
5029         genlmsg_multicast_allns(msg, 0, nl80211_regulatory_mcgrp.id,
5030                                 GFP_ATOMIC);
5031         rcu_read_unlock();
5032
5033         return;
5034
5035 nla_put_failure:
5036         genlmsg_cancel(msg, hdr);
5037         nlmsg_free(msg);
5038 }
5039
5040 static void nl80211_send_mlme_event(struct cfg80211_registered_device *rdev,
5041                                     struct net_device *netdev,
5042                                     const u8 *buf, size_t len,
5043                                     enum nl80211_commands cmd, gfp_t gfp)
5044 {
5045         struct sk_buff *msg;
5046         void *hdr;
5047
5048         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
5049         if (!msg)
5050                 return;
5051
5052         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
5053         if (!hdr) {
5054                 nlmsg_free(msg);
5055                 return;
5056         }
5057
5058         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
5059         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
5060         NLA_PUT(msg, NL80211_ATTR_FRAME, len, buf);
5061
5062         if (genlmsg_end(msg, hdr) < 0) {
5063                 nlmsg_free(msg);
5064                 return;
5065         }
5066
5067         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
5068                                 nl80211_mlme_mcgrp.id, gfp);
5069         return;
5070
5071  nla_put_failure:
5072         genlmsg_cancel(msg, hdr);
5073         nlmsg_free(msg);
5074 }
5075
5076 void nl80211_send_rx_auth(struct cfg80211_registered_device *rdev,
5077                           struct net_device *netdev, const u8 *buf,
5078                           size_t len, gfp_t gfp)
5079 {
5080         nl80211_send_mlme_event(rdev, netdev, buf, len,
5081                                 NL80211_CMD_AUTHENTICATE, gfp);
5082 }
5083
5084 void nl80211_send_rx_assoc(struct cfg80211_registered_device *rdev,
5085                            struct net_device *netdev, const u8 *buf,
5086                            size_t len, gfp_t gfp)
5087 {
5088         nl80211_send_mlme_event(rdev, netdev, buf, len,
5089                                 NL80211_CMD_ASSOCIATE, gfp);
5090 }
5091
5092 void nl80211_send_deauth(struct cfg80211_registered_device *rdev,
5093                          struct net_device *netdev, const u8 *buf,
5094                          size_t len, gfp_t gfp)
5095 {
5096         nl80211_send_mlme_event(rdev, netdev, buf, len,
5097                                 NL80211_CMD_DEAUTHENTICATE, gfp);
5098 }
5099
5100 void nl80211_send_disassoc(struct cfg80211_registered_device *rdev,
5101                            struct net_device *netdev, const u8 *buf,
5102                            size_t len, gfp_t gfp)
5103 {
5104         nl80211_send_mlme_event(rdev, netdev, buf, len,
5105                                 NL80211_CMD_DISASSOCIATE, gfp);
5106 }
5107
5108 static void nl80211_send_mlme_timeout(struct cfg80211_registered_device *rdev,
5109                                       struct net_device *netdev, int cmd,
5110                                       const u8 *addr, gfp_t gfp)
5111 {
5112         struct sk_buff *msg;
5113         void *hdr;
5114
5115         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
5116         if (!msg)
5117                 return;
5118
5119         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
5120         if (!hdr) {
5121                 nlmsg_free(msg);
5122                 return;
5123         }
5124
5125         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
5126         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
5127         NLA_PUT_FLAG(msg, NL80211_ATTR_TIMED_OUT);
5128         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
5129
5130         if (genlmsg_end(msg, hdr) < 0) {
5131                 nlmsg_free(msg);
5132                 return;
5133         }
5134
5135         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
5136                                 nl80211_mlme_mcgrp.id, gfp);
5137         return;
5138
5139  nla_put_failure:
5140         genlmsg_cancel(msg, hdr);
5141         nlmsg_free(msg);
5142 }
5143
5144 void nl80211_send_auth_timeout(struct cfg80211_registered_device *rdev,
5145                                struct net_device *netdev, const u8 *addr,
5146                                gfp_t gfp)
5147 {
5148         nl80211_send_mlme_timeout(rdev, netdev, NL80211_CMD_AUTHENTICATE,
5149                                   addr, gfp);
5150 }
5151
5152 void nl80211_send_assoc_timeout(struct cfg80211_registered_device *rdev,
5153                                 struct net_device *netdev, const u8 *addr,
5154                                 gfp_t gfp)
5155 {
5156         nl80211_send_mlme_timeout(rdev, netdev, NL80211_CMD_ASSOCIATE,
5157                                   addr, gfp);
5158 }
5159
5160 void nl80211_send_connect_result(struct cfg80211_registered_device *rdev,
5161                                  struct net_device *netdev, const u8 *bssid,
5162                                  const u8 *req_ie, size_t req_ie_len,
5163                                  const u8 *resp_ie, size_t resp_ie_len,
5164                                  u16 status, gfp_t gfp)
5165 {
5166         struct sk_buff *msg;
5167         void *hdr;
5168
5169         msg = nlmsg_new(NLMSG_GOODSIZE, gfp);
5170         if (!msg)
5171                 return;
5172
5173         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CONNECT);
5174         if (!hdr) {
5175                 nlmsg_free(msg);
5176                 return;
5177         }
5178
5179         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
5180         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
5181         if (bssid)
5182                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid);
5183         NLA_PUT_U16(msg, NL80211_ATTR_STATUS_CODE, status);
5184         if (req_ie)
5185                 NLA_PUT(msg, NL80211_ATTR_REQ_IE, req_ie_len, req_ie);
5186         if (resp_ie)
5187                 NLA_PUT(msg, NL80211_ATTR_RESP_IE, resp_ie_len, resp_ie);
5188
5189         if (genlmsg_end(msg, hdr) < 0) {
5190                 nlmsg_free(msg);
5191                 return;
5192         }
5193
5194         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
5195                                 nl80211_mlme_mcgrp.id, gfp);
5196         return;
5197
5198  nla_put_failure:
5199         genlmsg_cancel(msg, hdr);
5200         nlmsg_free(msg);
5201
5202 }
5203
5204 void nl80211_send_roamed(struct cfg80211_registered_device *rdev,
5205                          struct net_device *netdev, const u8 *bssid,
5206                          const u8 *req_ie, size_t req_ie_len,
5207                          const u8 *resp_ie, size_t resp_ie_len, gfp_t gfp)
5208 {
5209         struct sk_buff *msg;
5210         void *hdr;
5211
5212         msg = nlmsg_new(NLMSG_GOODSIZE, gfp);
5213         if (!msg)
5214                 return;
5215
5216         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_ROAM);
5217         if (!hdr) {
5218                 nlmsg_free(msg);
5219                 return;
5220         }
5221
5222         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
5223         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
5224         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid);
5225         if (req_ie)
5226                 NLA_PUT(msg, NL80211_ATTR_REQ_IE, req_ie_len, req_ie);
5227         if (resp_ie)
5228                 NLA_PUT(msg, NL80211_ATTR_RESP_IE, resp_ie_len, resp_ie);
5229
5230         if (genlmsg_end(msg, hdr) < 0) {
5231                 nlmsg_free(msg);
5232                 return;
5233         }
5234
5235         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
5236                                 nl80211_mlme_mcgrp.id, gfp);
5237         return;
5238
5239  nla_put_failure:
5240         genlmsg_cancel(msg, hdr);
5241         nlmsg_free(msg);
5242
5243 }
5244
5245 void nl80211_send_disconnected(struct cfg80211_registered_device *rdev,
5246                                struct net_device *netdev, u16 reason,
5247                                const u8 *ie, size_t ie_len, bool from_ap)
5248 {
5249         struct sk_buff *msg;
5250         void *hdr;
5251
5252         msg = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
5253         if (!msg)
5254                 return;
5255
5256         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_DISCONNECT);
5257         if (!hdr) {
5258                 nlmsg_free(msg);
5259                 return;
5260         }
5261
5262         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
5263         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
5264         if (from_ap && reason)
5265                 NLA_PUT_U16(msg, NL80211_ATTR_REASON_CODE, reason);
5266         if (from_ap)
5267                 NLA_PUT_FLAG(msg, NL80211_ATTR_DISCONNECTED_BY_AP);
5268         if (ie)
5269                 NLA_PUT(msg, NL80211_ATTR_IE, ie_len, ie);
5270
5271         if (genlmsg_end(msg, hdr) < 0) {
5272                 nlmsg_free(msg);
5273                 return;
5274         }
5275
5276         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
5277                                 nl80211_mlme_mcgrp.id, GFP_KERNEL);
5278         return;
5279
5280  nla_put_failure:
5281         genlmsg_cancel(msg, hdr);
5282         nlmsg_free(msg);
5283
5284 }
5285
5286 void nl80211_send_ibss_bssid(struct cfg80211_registered_device *rdev,
5287                              struct net_device *netdev, const u8 *bssid,
5288                              gfp_t gfp)
5289 {
5290         struct sk_buff *msg;
5291         void *hdr;
5292
5293         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
5294         if (!msg)
5295                 return;
5296
5297         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_JOIN_IBSS);
5298         if (!hdr) {
5299                 nlmsg_free(msg);
5300                 return;
5301         }
5302
5303         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
5304         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
5305         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid);
5306
5307         if (genlmsg_end(msg, hdr) < 0) {
5308                 nlmsg_free(msg);
5309                 return;
5310         }
5311
5312         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
5313                                 nl80211_mlme_mcgrp.id, gfp);
5314         return;
5315
5316  nla_put_failure:
5317         genlmsg_cancel(msg, hdr);
5318         nlmsg_free(msg);
5319 }
5320
5321 void nl80211_michael_mic_failure(struct cfg80211_registered_device *rdev,
5322                                  struct net_device *netdev, const u8 *addr,
5323                                  enum nl80211_key_type key_type, int key_id,
5324                                  const u8 *tsc, gfp_t gfp)
5325 {
5326         struct sk_buff *msg;
5327         void *hdr;
5328
5329         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
5330         if (!msg)
5331                 return;
5332
5333         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_MICHAEL_MIC_FAILURE);
5334         if (!hdr) {
5335                 nlmsg_free(msg);
5336                 return;
5337         }
5338
5339         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
5340         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
5341         if (addr)
5342                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
5343         NLA_PUT_U32(msg, NL80211_ATTR_KEY_TYPE, key_type);
5344         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_id);
5345         if (tsc)
5346                 NLA_PUT(msg, NL80211_ATTR_KEY_SEQ, 6, tsc);
5347
5348         if (genlmsg_end(msg, hdr) < 0) {
5349                 nlmsg_free(msg);
5350                 return;
5351         }
5352
5353         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
5354                                 nl80211_mlme_mcgrp.id, gfp);
5355         return;
5356
5357  nla_put_failure:
5358         genlmsg_cancel(msg, hdr);
5359         nlmsg_free(msg);
5360 }
5361
5362 void nl80211_send_beacon_hint_event(struct wiphy *wiphy,
5363                                     struct ieee80211_channel *channel_before,
5364                                     struct ieee80211_channel *channel_after)
5365 {
5366         struct sk_buff *msg;
5367         void *hdr;
5368         struct nlattr *nl_freq;
5369
5370         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_ATOMIC);
5371         if (!msg)
5372                 return;
5373
5374         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_REG_BEACON_HINT);
5375         if (!hdr) {
5376                 nlmsg_free(msg);
5377                 return;
5378         }
5379
5380         /*
5381          * Since we are applying the beacon hint to a wiphy we know its
5382          * wiphy_idx is valid
5383          */
5384         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, get_wiphy_idx(wiphy));
5385
5386         /* Before */
5387         nl_freq = nla_nest_start(msg, NL80211_ATTR_FREQ_BEFORE);
5388         if (!nl_freq)
5389                 goto nla_put_failure;
5390         if (nl80211_msg_put_channel(msg, channel_before))
5391                 goto nla_put_failure;
5392         nla_nest_end(msg, nl_freq);
5393
5394         /* After */
5395         nl_freq = nla_nest_start(msg, NL80211_ATTR_FREQ_AFTER);
5396         if (!nl_freq)
5397                 goto nla_put_failure;
5398         if (nl80211_msg_put_channel(msg, channel_after))
5399                 goto nla_put_failure;
5400         nla_nest_end(msg, nl_freq);
5401
5402         if (genlmsg_end(msg, hdr) < 0) {
5403                 nlmsg_free(msg);
5404                 return;
5405         }
5406
5407         rcu_read_lock();
5408         genlmsg_multicast_allns(msg, 0, nl80211_regulatory_mcgrp.id,
5409                                 GFP_ATOMIC);
5410         rcu_read_unlock();
5411
5412         return;
5413
5414 nla_put_failure:
5415         genlmsg_cancel(msg, hdr);
5416         nlmsg_free(msg);
5417 }
5418
5419 static void nl80211_send_remain_on_chan_event(
5420         int cmd, struct cfg80211_registered_device *rdev,
5421         struct net_device *netdev, u64 cookie,
5422         struct ieee80211_channel *chan,
5423         enum nl80211_channel_type channel_type,
5424         unsigned int duration, gfp_t gfp)
5425 {
5426         struct sk_buff *msg;
5427         void *hdr;
5428
5429         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
5430         if (!msg)
5431                 return;
5432
5433         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
5434         if (!hdr) {
5435                 nlmsg_free(msg);
5436                 return;
5437         }
5438
5439         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
5440         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
5441         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, chan->center_freq);
5442         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE, channel_type);
5443         NLA_PUT_U64(msg, NL80211_ATTR_COOKIE, cookie);
5444
5445         if (cmd == NL80211_CMD_REMAIN_ON_CHANNEL)
5446                 NLA_PUT_U32(msg, NL80211_ATTR_DURATION, duration);
5447
5448         if (genlmsg_end(msg, hdr) < 0) {
5449                 nlmsg_free(msg);
5450                 return;
5451         }
5452
5453         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
5454                                 nl80211_mlme_mcgrp.id, gfp);
5455         return;
5456
5457  nla_put_failure:
5458         genlmsg_cancel(msg, hdr);
5459         nlmsg_free(msg);
5460 }
5461
5462 void nl80211_send_remain_on_channel(struct cfg80211_registered_device *rdev,
5463                                     struct net_device *netdev, u64 cookie,
5464                                     struct ieee80211_channel *chan,
5465                                     enum nl80211_channel_type channel_type,
5466                                     unsigned int duration, gfp_t gfp)
5467 {
5468         nl80211_send_remain_on_chan_event(NL80211_CMD_REMAIN_ON_CHANNEL,
5469                                           rdev, netdev, cookie, chan,
5470                                           channel_type, duration, gfp);
5471 }
5472
5473 void nl80211_send_remain_on_channel_cancel(
5474         struct cfg80211_registered_device *rdev, struct net_device *netdev,
5475         u64 cookie, struct ieee80211_channel *chan,
5476         enum nl80211_channel_type channel_type, gfp_t gfp)
5477 {
5478         nl80211_send_remain_on_chan_event(NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL,
5479                                           rdev, netdev, cookie, chan,
5480                                           channel_type, 0, gfp);
5481 }
5482
5483 void nl80211_send_sta_event(struct cfg80211_registered_device *rdev,
5484                             struct net_device *dev, const u8 *mac_addr,
5485                             struct station_info *sinfo, gfp_t gfp)
5486 {
5487         struct sk_buff *msg;
5488
5489         msg = nlmsg_new(NLMSG_GOODSIZE, gfp);
5490         if (!msg)
5491                 return;
5492
5493         if (nl80211_send_station(msg, 0, 0, 0, dev, mac_addr, sinfo) < 0) {
5494                 nlmsg_free(msg);
5495                 return;
5496         }
5497
5498         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
5499                                 nl80211_mlme_mcgrp.id, gfp);
5500 }
5501
5502 /* initialisation/exit functions */
5503
5504 int nl80211_init(void)
5505 {
5506         int err;
5507
5508         err = genl_register_family_with_ops(&nl80211_fam,
5509                 nl80211_ops, ARRAY_SIZE(nl80211_ops));
5510         if (err)
5511                 return err;
5512
5513         err = genl_register_mc_group(&nl80211_fam, &nl80211_config_mcgrp);
5514         if (err)
5515                 goto err_out;
5516
5517         err = genl_register_mc_group(&nl80211_fam, &nl80211_scan_mcgrp);
5518         if (err)
5519                 goto err_out;
5520
5521         err = genl_register_mc_group(&nl80211_fam, &nl80211_regulatory_mcgrp);
5522         if (err)
5523                 goto err_out;
5524
5525         err = genl_register_mc_group(&nl80211_fam, &nl80211_mlme_mcgrp);
5526         if (err)
5527                 goto err_out;
5528
5529 #ifdef CONFIG_NL80211_TESTMODE
5530         err = genl_register_mc_group(&nl80211_fam, &nl80211_testmode_mcgrp);
5531         if (err)
5532                 goto err_out;
5533 #endif
5534
5535         return 0;
5536  err_out:
5537         genl_unregister_family(&nl80211_fam);
5538         return err;
5539 }
5540
5541 void nl80211_exit(void)
5542 {
5543         genl_unregister_family(&nl80211_fam);
5544 }