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