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