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