net/wireless/nl80211.c

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