Pileus Git - ~andy/linux/blob - 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 <net/inet_connection_sock.h>
  23 #include "core.h"
  24 #include "nl80211.h"
  25 #include "reg.h"
  26 #include "rdev-ops.h"
  27
  28 static int nl80211_crypto_settings(struct cfg80211_registered_device *rdev,
  29                                    struct genl_info *info,
  30                                    struct cfg80211_crypto_settings *settings,
  31                                    int cipher_limit);
  32
  33 static int nl80211_pre_doit(struct genl_ops *ops, struct sk_buff *skb,
  34                             struct genl_info *info);
  35 static void nl80211_post_doit(struct genl_ops *ops, struct sk_buff *skb,
  36                               struct genl_info *info);
  37
  38 /* the netlink family */
  39 static struct genl_family nl80211_fam = {
  40         .id = GENL_ID_GENERATE,         /* don't bother with a hardcoded ID */
  41         .name = NL80211_GENL_NAME,      /* have users key off the name instead */
  42         .hdrsize = 0,                   /* no private header */
  43         .version = 1,                   /* no particular meaning now */
  44         .maxattr = NL80211_ATTR_MAX,
  45         .netnsok = true,
  46         .pre_doit = nl80211_pre_doit,
  47         .post_doit = nl80211_post_doit,
  48 };
  49
  50 /* returns ERR_PTR values */
  51 static struct wireless_dev *
  52 __cfg80211_wdev_from_attrs(struct net *netns, struct nlattr **attrs)
  53 {
  54         struct cfg80211_registered_device *rdev;
  55         struct wireless_dev *result = NULL;
  56         bool have_ifidx = attrs[NL80211_ATTR_IFINDEX];
  57         bool have_wdev_id = attrs[NL80211_ATTR_WDEV];
  58         u64 wdev_id;
  59         int wiphy_idx = -1;
  60         int ifidx = -1;
  61
  62         ASSERT_RTNL();
  63
  64         if (!have_ifidx && !have_wdev_id)
  65                 return ERR_PTR(-EINVAL);
  66
  67         if (have_ifidx)
  68                 ifidx = nla_get_u32(attrs[NL80211_ATTR_IFINDEX]);
  69         if (have_wdev_id) {
  70                 wdev_id = nla_get_u64(attrs[NL80211_ATTR_WDEV]);
  71                 wiphy_idx = wdev_id >> 32;
  72         }
  73
  74         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
  75                 struct wireless_dev *wdev;
  76
  77                 if (wiphy_net(&rdev->wiphy) != netns)
  78                         continue;
  79
  80                 if (have_wdev_id && rdev->wiphy_idx != wiphy_idx)
  81                         continue;
  82
  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
  95                 if (result)
  96                         break;
  97         }
  98
  99         if (result)
 100                 return result;
 101         return ERR_PTR(-ENODEV);
 102 }
 103
 104 static struct cfg80211_registered_device *
 105 __cfg80211_rdev_from_attrs(struct net *netns, struct nlattr **attrs)
 106 {
 107         struct cfg80211_registered_device *rdev = NULL, *tmp;
 108         struct net_device *netdev;
 109
 110         ASSERT_RTNL();
 111
 112         if (!attrs[NL80211_ATTR_WIPHY] &&
 113             !attrs[NL80211_ATTR_IFINDEX] &&
 114             !attrs[NL80211_ATTR_WDEV])
 115                 return ERR_PTR(-EINVAL);
 116
 117         if (attrs[NL80211_ATTR_WIPHY])
 118                 rdev = cfg80211_rdev_by_wiphy_idx(
 119                                 nla_get_u32(attrs[NL80211_ATTR_WIPHY]));
 120
 121         if (attrs[NL80211_ATTR_WDEV]) {
 122                 u64 wdev_id = nla_get_u64(attrs[NL80211_ATTR_WDEV]);
 123                 struct wireless_dev *wdev;
 124                 bool found = false;
 125
 126                 tmp = cfg80211_rdev_by_wiphy_idx(wdev_id >> 32);
 127                 if (tmp) {
 128                         /* make sure wdev exists */
 129                         list_for_each_entry(wdev, &tmp->wdev_list, list) {
 130                                 if (wdev->identifier != (u32)wdev_id)
 131                                         continue;
 132                                 found = true;
 133                                 break;
 134                         }
 135
 136                         if (!found)
 137                                 tmp = NULL;
 138
 139                         if (rdev && tmp != rdev)
 140                                 return ERR_PTR(-EINVAL);
 141                         rdev = tmp;
 142                 }
 143         }
 144
 145         if (attrs[NL80211_ATTR_IFINDEX]) {
 146                 int ifindex = nla_get_u32(attrs[NL80211_ATTR_IFINDEX]);
 147                 netdev = dev_get_by_index(netns, ifindex);
 148                 if (netdev) {
 149                         if (netdev->ieee80211_ptr)
 150                                 tmp = wiphy_to_dev(
 151                                                 netdev->ieee80211_ptr->wiphy);
 152                         else
 153                                 tmp = NULL;
 154
 155                         dev_put(netdev);
 156
 157                         /* not wireless device -- return error */
 158                         if (!tmp)
 159                                 return ERR_PTR(-EINVAL);
 160
 161                         /* mismatch -- return error */
 162                         if (rdev && tmp != rdev)
 163                                 return ERR_PTR(-EINVAL);
 164
 165                         rdev = tmp;
 166                 }
 167         }
 168
 169         if (!rdev)
 170                 return ERR_PTR(-ENODEV);
 171
 172         if (netns != wiphy_net(&rdev->wiphy))
 173                 return ERR_PTR(-ENODEV);
 174
 175         return rdev;
 176 }
 177
 178 /*
 179  * This function returns a pointer to the driver
 180  * that the genl_info item that is passed refers to.
 181  *
 182  * The result of this can be a PTR_ERR and hence must
 183  * be checked with IS_ERR() for errors.
 184  */
 185 static struct cfg80211_registered_device *
 186 cfg80211_get_dev_from_info(struct net *netns, struct genl_info *info)
 187 {
 188         return __cfg80211_rdev_from_attrs(netns, info->attrs);
 189 }
 190
 191 /* policy for the attributes */
 192 static const struct nla_policy nl80211_policy[NL80211_ATTR_MAX+1] = {
 193         [NL80211_ATTR_WIPHY] = { .type = NLA_U32 },
 194         [NL80211_ATTR_WIPHY_NAME] = { .type = NLA_NUL_STRING,
 195                                       .len = 20-1 },
 196         [NL80211_ATTR_WIPHY_TXQ_PARAMS] = { .type = NLA_NESTED },
 197
 198         [NL80211_ATTR_WIPHY_FREQ] = { .type = NLA_U32 },
 199         [NL80211_ATTR_WIPHY_CHANNEL_TYPE] = { .type = NLA_U32 },
 200         [NL80211_ATTR_CHANNEL_WIDTH] = { .type = NLA_U32 },
 201         [NL80211_ATTR_CENTER_FREQ1] = { .type = NLA_U32 },
 202         [NL80211_ATTR_CENTER_FREQ2] = { .type = NLA_U32 },
 203
 204         [NL80211_ATTR_WIPHY_RETRY_SHORT] = { .type = NLA_U8 },
 205         [NL80211_ATTR_WIPHY_RETRY_LONG] = { .type = NLA_U8 },
 206         [NL80211_ATTR_WIPHY_FRAG_THRESHOLD] = { .type = NLA_U32 },
 207         [NL80211_ATTR_WIPHY_RTS_THRESHOLD] = { .type = NLA_U32 },
 208         [NL80211_ATTR_WIPHY_COVERAGE_CLASS] = { .type = NLA_U8 },
 209
 210         [NL80211_ATTR_IFTYPE] = { .type = NLA_U32 },
 211         [NL80211_ATTR_IFINDEX] = { .type = NLA_U32 },
 212         [NL80211_ATTR_IFNAME] = { .type = NLA_NUL_STRING, .len = IFNAMSIZ-1 },
 213
 214         [NL80211_ATTR_MAC] = { .len = ETH_ALEN },
 215         [NL80211_ATTR_PREV_BSSID] = { .len = ETH_ALEN },
 216
 217         [NL80211_ATTR_KEY] = { .type = NLA_NESTED, },
 218         [NL80211_ATTR_KEY_DATA] = { .type = NLA_BINARY,
 219                                     .len = WLAN_MAX_KEY_LEN },
 220         [NL80211_ATTR_KEY_IDX] = { .type = NLA_U8 },
 221         [NL80211_ATTR_KEY_CIPHER] = { .type = NLA_U32 },
 222         [NL80211_ATTR_KEY_DEFAULT] = { .type = NLA_FLAG },
 223         [NL80211_ATTR_KEY_SEQ] = { .type = NLA_BINARY, .len = 16 },
 224         [NL80211_ATTR_KEY_TYPE] = { .type = NLA_U32 },
 225
 226         [NL80211_ATTR_BEACON_INTERVAL] = { .type = NLA_U32 },
 227         [NL80211_ATTR_DTIM_PERIOD] = { .type = NLA_U32 },
 228         [NL80211_ATTR_BEACON_HEAD] = { .type = NLA_BINARY,
 229                                        .len = IEEE80211_MAX_DATA_LEN },
 230         [NL80211_ATTR_BEACON_TAIL] = { .type = NLA_BINARY,
 231                                        .len = IEEE80211_MAX_DATA_LEN },
 232         [NL80211_ATTR_STA_AID] = { .type = NLA_U16 },
 233         [NL80211_ATTR_STA_FLAGS] = { .type = NLA_NESTED },
 234         [NL80211_ATTR_STA_LISTEN_INTERVAL] = { .type = NLA_U16 },
 235         [NL80211_ATTR_STA_SUPPORTED_RATES] = { .type = NLA_BINARY,
 236                                                .len = NL80211_MAX_SUPP_RATES },
 237         [NL80211_ATTR_STA_PLINK_ACTION] = { .type = NLA_U8 },
 238         [NL80211_ATTR_STA_VLAN] = { .type = NLA_U32 },
 239         [NL80211_ATTR_MNTR_FLAGS] = { /* NLA_NESTED can't be empty */ },
 240         [NL80211_ATTR_MESH_ID] = { .type = NLA_BINARY,
 241                                    .len = IEEE80211_MAX_MESH_ID_LEN },
 242         [NL80211_ATTR_MPATH_NEXT_HOP] = { .type = NLA_U32 },
 243
 244         [NL80211_ATTR_REG_ALPHA2] = { .type = NLA_STRING, .len = 2 },
 245         [NL80211_ATTR_REG_RULES] = { .type = NLA_NESTED },
 246
 247         [NL80211_ATTR_BSS_CTS_PROT] = { .type = NLA_U8 },
 248         [NL80211_ATTR_BSS_SHORT_PREAMBLE] = { .type = NLA_U8 },
 249         [NL80211_ATTR_BSS_SHORT_SLOT_TIME] = { .type = NLA_U8 },
 250         [NL80211_ATTR_BSS_BASIC_RATES] = { .type = NLA_BINARY,
 251                                            .len = NL80211_MAX_SUPP_RATES },
 252         [NL80211_ATTR_BSS_HT_OPMODE] = { .type = NLA_U16 },
 253
 254         [NL80211_ATTR_MESH_CONFIG] = { .type = NLA_NESTED },
 255         [NL80211_ATTR_SUPPORT_MESH_AUTH] = { .type = NLA_FLAG },
 256
 257         [NL80211_ATTR_HT_CAPABILITY] = { .len = NL80211_HT_CAPABILITY_LEN },
 258
 259         [NL80211_ATTR_MGMT_SUBTYPE] = { .type = NLA_U8 },
 260         [NL80211_ATTR_IE] = { .type = NLA_BINARY,
 261                               .len = IEEE80211_MAX_DATA_LEN },
 262         [NL80211_ATTR_SCAN_FREQUENCIES] = { .type = NLA_NESTED },
 263         [NL80211_ATTR_SCAN_SSIDS] = { .type = NLA_NESTED },
 264
 265         [NL80211_ATTR_SSID] = { .type = NLA_BINARY,
 266                                 .len = IEEE80211_MAX_SSID_LEN },
 267         [NL80211_ATTR_AUTH_TYPE] = { .type = NLA_U32 },
 268         [NL80211_ATTR_REASON_CODE] = { .type = NLA_U16 },
 269         [NL80211_ATTR_FREQ_FIXED] = { .type = NLA_FLAG },
 270         [NL80211_ATTR_TIMED_OUT] = { .type = NLA_FLAG },
 271         [NL80211_ATTR_USE_MFP] = { .type = NLA_U32 },
 272         [NL80211_ATTR_STA_FLAGS2] = {
 273                 .len = sizeof(struct nl80211_sta_flag_update),
 274         },
 275         [NL80211_ATTR_CONTROL_PORT] = { .type = NLA_FLAG },
 276         [NL80211_ATTR_CONTROL_PORT_ETHERTYPE] = { .type = NLA_U16 },
 277         [NL80211_ATTR_CONTROL_PORT_NO_ENCRYPT] = { .type = NLA_FLAG },
 278         [NL80211_ATTR_PRIVACY] = { .type = NLA_FLAG },
 279         [NL80211_ATTR_CIPHER_SUITE_GROUP] = { .type = NLA_U32 },
 280         [NL80211_ATTR_WPA_VERSIONS] = { .type = NLA_U32 },
 281         [NL80211_ATTR_PID] = { .type = NLA_U32 },
 282         [NL80211_ATTR_4ADDR] = { .type = NLA_U8 },
 283         [NL80211_ATTR_PMKID] = { .type = NLA_BINARY,
 284                                  .len = WLAN_PMKID_LEN },
 285         [NL80211_ATTR_DURATION] = { .type = NLA_U32 },
 286         [NL80211_ATTR_COOKIE] = { .type = NLA_U64 },
 287         [NL80211_ATTR_TX_RATES] = { .type = NLA_NESTED },
 288         [NL80211_ATTR_FRAME] = { .type = NLA_BINARY,
 289                                  .len = IEEE80211_MAX_DATA_LEN },
 290         [NL80211_ATTR_FRAME_MATCH] = { .type = NLA_BINARY, },
 291         [NL80211_ATTR_PS_STATE] = { .type = NLA_U32 },
 292         [NL80211_ATTR_CQM] = { .type = NLA_NESTED, },
 293         [NL80211_ATTR_LOCAL_STATE_CHANGE] = { .type = NLA_FLAG },
 294         [NL80211_ATTR_AP_ISOLATE] = { .type = NLA_U8 },
 295         [NL80211_ATTR_WIPHY_TX_POWER_SETTING] = { .type = NLA_U32 },
 296         [NL80211_ATTR_WIPHY_TX_POWER_LEVEL] = { .type = NLA_U32 },
 297         [NL80211_ATTR_FRAME_TYPE] = { .type = NLA_U16 },
 298         [NL80211_ATTR_WIPHY_ANTENNA_TX] = { .type = NLA_U32 },
 299         [NL80211_ATTR_WIPHY_ANTENNA_RX] = { .type = NLA_U32 },
 300         [NL80211_ATTR_MCAST_RATE] = { .type = NLA_U32 },
 301         [NL80211_ATTR_OFFCHANNEL_TX_OK] = { .type = NLA_FLAG },
 302         [NL80211_ATTR_KEY_DEFAULT_TYPES] = { .type = NLA_NESTED },
 303         [NL80211_ATTR_WOWLAN_TRIGGERS] = { .type = NLA_NESTED },
 304         [NL80211_ATTR_STA_PLINK_STATE] = { .type = NLA_U8 },
 305         [NL80211_ATTR_SCHED_SCAN_INTERVAL] = { .type = NLA_U32 },
 306         [NL80211_ATTR_REKEY_DATA] = { .type = NLA_NESTED },
 307         [NL80211_ATTR_SCAN_SUPP_RATES] = { .type = NLA_NESTED },
 308         [NL80211_ATTR_HIDDEN_SSID] = { .type = NLA_U32 },
 309         [NL80211_ATTR_IE_PROBE_RESP] = { .type = NLA_BINARY,
 310                                          .len = IEEE80211_MAX_DATA_LEN },
 311         [NL80211_ATTR_IE_ASSOC_RESP] = { .type = NLA_BINARY,
 312                                          .len = IEEE80211_MAX_DATA_LEN },
 313         [NL80211_ATTR_ROAM_SUPPORT] = { .type = NLA_FLAG },
 314         [NL80211_ATTR_SCHED_SCAN_MATCH] = { .type = NLA_NESTED },
 315         [NL80211_ATTR_TX_NO_CCK_RATE] = { .type = NLA_FLAG },
 316         [NL80211_ATTR_TDLS_ACTION] = { .type = NLA_U8 },
 317         [NL80211_ATTR_TDLS_DIALOG_TOKEN] = { .type = NLA_U8 },
 318         [NL80211_ATTR_TDLS_OPERATION] = { .type = NLA_U8 },
 319         [NL80211_ATTR_TDLS_SUPPORT] = { .type = NLA_FLAG },
 320         [NL80211_ATTR_TDLS_EXTERNAL_SETUP] = { .type = NLA_FLAG },
 321         [NL80211_ATTR_DONT_WAIT_FOR_ACK] = { .type = NLA_FLAG },
 322         [NL80211_ATTR_PROBE_RESP] = { .type = NLA_BINARY,
 323                                       .len = IEEE80211_MAX_DATA_LEN },
 324         [NL80211_ATTR_DFS_REGION] = { .type = NLA_U8 },
 325         [NL80211_ATTR_DISABLE_HT] = { .type = NLA_FLAG },
 326         [NL80211_ATTR_HT_CAPABILITY_MASK] = {
 327                 .len = NL80211_HT_CAPABILITY_LEN
 328         },
 329         [NL80211_ATTR_NOACK_MAP] = { .type = NLA_U16 },
 330         [NL80211_ATTR_INACTIVITY_TIMEOUT] = { .type = NLA_U16 },
 331         [NL80211_ATTR_BG_SCAN_PERIOD] = { .type = NLA_U16 },
 332         [NL80211_ATTR_WDEV] = { .type = NLA_U64 },
 333         [NL80211_ATTR_USER_REG_HINT_TYPE] = { .type = NLA_U32 },
 334         [NL80211_ATTR_SAE_DATA] = { .type = NLA_BINARY, },
 335         [NL80211_ATTR_VHT_CAPABILITY] = { .len = NL80211_VHT_CAPABILITY_LEN },
 336         [NL80211_ATTR_SCAN_FLAGS] = { .type = NLA_U32 },
 337         [NL80211_ATTR_P2P_CTWINDOW] = { .type = NLA_U8 },
 338         [NL80211_ATTR_P2P_OPPPS] = { .type = NLA_U8 },
 339         [NL80211_ATTR_ACL_POLICY] = {. type = NLA_U32 },
 340         [NL80211_ATTR_MAC_ADDRS] = { .type = NLA_NESTED },
 341         [NL80211_ATTR_STA_CAPABILITY] = { .type = NLA_U16 },
 342         [NL80211_ATTR_STA_EXT_CAPABILITY] = { .type = NLA_BINARY, },
 343         [NL80211_ATTR_SPLIT_WIPHY_DUMP] = { .type = NLA_FLAG, },
 344         [NL80211_ATTR_DISABLE_VHT] = { .type = NLA_FLAG },
 345         [NL80211_ATTR_VHT_CAPABILITY_MASK] = {
 346                 .len = NL80211_VHT_CAPABILITY_LEN,
 347         },
 348         [NL80211_ATTR_MDID] = { .type = NLA_U16 },
 349         [NL80211_ATTR_IE_RIC] = { .type = NLA_BINARY,
 350                                   .len = IEEE80211_MAX_DATA_LEN },
 351         [NL80211_ATTR_PEER_AID] = { .type = NLA_U16 },
 352         [NL80211_ATTR_CH_SWITCH_COUNT] = { .type = NLA_U32 },
 353         [NL80211_ATTR_CH_SWITCH_BLOCK_TX] = { .type = NLA_FLAG },
 354         [NL80211_ATTR_CSA_IES] = { .type = NLA_NESTED },
 355         [NL80211_ATTR_CSA_C_OFF_BEACON] = { .type = NLA_U16 },
 356         [NL80211_ATTR_CSA_C_OFF_PRESP] = { .type = NLA_U16 },
 357         [NL80211_ATTR_STA_SUPPORTED_CHANNELS] = { .type = NLA_BINARY },
 358         [NL80211_ATTR_STA_SUPPORTED_OPER_CLASSES] = { .type = NLA_BINARY },
 359         [NL80211_ATTR_HANDLE_DFS] = { .type = NLA_FLAG },
 360 };
 361
 362 /* policy for the key attributes */
 363 static const struct nla_policy nl80211_key_policy[NL80211_KEY_MAX + 1] = {
 364         [NL80211_KEY_DATA] = { .type = NLA_BINARY, .len = WLAN_MAX_KEY_LEN },
 365         [NL80211_KEY_IDX] = { .type = NLA_U8 },
 366         [NL80211_KEY_CIPHER] = { .type = NLA_U32 },
 367         [NL80211_KEY_SEQ] = { .type = NLA_BINARY, .len = 16 },
 368         [NL80211_KEY_DEFAULT] = { .type = NLA_FLAG },
 369         [NL80211_KEY_DEFAULT_MGMT] = { .type = NLA_FLAG },
 370         [NL80211_KEY_TYPE] = { .type = NLA_U32 },
 371         [NL80211_KEY_DEFAULT_TYPES] = { .type = NLA_NESTED },
 372 };
 373
 374 /* policy for the key default flags */
 375 static const struct nla_policy
 376 nl80211_key_default_policy[NUM_NL80211_KEY_DEFAULT_TYPES] = {
 377         [NL80211_KEY_DEFAULT_TYPE_UNICAST] = { .type = NLA_FLAG },
 378         [NL80211_KEY_DEFAULT_TYPE_MULTICAST] = { .type = NLA_FLAG },
 379 };
 380
 381 /* policy for WoWLAN attributes */
 382 static const struct nla_policy
 383 nl80211_wowlan_policy[NUM_NL80211_WOWLAN_TRIG] = {
 384         [NL80211_WOWLAN_TRIG_ANY] = { .type = NLA_FLAG },
 385         [NL80211_WOWLAN_TRIG_DISCONNECT] = { .type = NLA_FLAG },
 386         [NL80211_WOWLAN_TRIG_MAGIC_PKT] = { .type = NLA_FLAG },
 387         [NL80211_WOWLAN_TRIG_PKT_PATTERN] = { .type = NLA_NESTED },
 388         [NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE] = { .type = NLA_FLAG },
 389         [NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST] = { .type = NLA_FLAG },
 390         [NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE] = { .type = NLA_FLAG },
 391         [NL80211_WOWLAN_TRIG_RFKILL_RELEASE] = { .type = NLA_FLAG },
 392         [NL80211_WOWLAN_TRIG_TCP_CONNECTION] = { .type = NLA_NESTED },
 393 };
 394
 395 static const struct nla_policy
 396 nl80211_wowlan_tcp_policy[NUM_NL80211_WOWLAN_TCP] = {
 397         [NL80211_WOWLAN_TCP_SRC_IPV4] = { .type = NLA_U32 },
 398         [NL80211_WOWLAN_TCP_DST_IPV4] = { .type = NLA_U32 },
 399         [NL80211_WOWLAN_TCP_DST_MAC] = { .len = ETH_ALEN },
 400         [NL80211_WOWLAN_TCP_SRC_PORT] = { .type = NLA_U16 },
 401         [NL80211_WOWLAN_TCP_DST_PORT] = { .type = NLA_U16 },
 402         [NL80211_WOWLAN_TCP_DATA_PAYLOAD] = { .len = 1 },
 403         [NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ] = {
 404                 .len = sizeof(struct nl80211_wowlan_tcp_data_seq)
 405         },
 406         [NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN] = {
 407                 .len = sizeof(struct nl80211_wowlan_tcp_data_token)
 408         },
 409         [NL80211_WOWLAN_TCP_DATA_INTERVAL] = { .type = NLA_U32 },
 410         [NL80211_WOWLAN_TCP_WAKE_PAYLOAD] = { .len = 1 },
 411         [NL80211_WOWLAN_TCP_WAKE_MASK] = { .len = 1 },
 412 };
 413
 414 /* policy for coalesce rule attributes */
 415 static const struct nla_policy
 416 nl80211_coalesce_policy[NUM_NL80211_ATTR_COALESCE_RULE] = {
 417         [NL80211_ATTR_COALESCE_RULE_DELAY] = { .type = NLA_U32 },
 418         [NL80211_ATTR_COALESCE_RULE_CONDITION] = { .type = NLA_U32 },
 419         [NL80211_ATTR_COALESCE_RULE_PKT_PATTERN] = { .type = NLA_NESTED },
 420 };
 421
 422 /* policy for GTK rekey offload attributes */
 423 static const struct nla_policy
 424 nl80211_rekey_policy[NUM_NL80211_REKEY_DATA] = {
 425         [NL80211_REKEY_DATA_KEK] = { .len = NL80211_KEK_LEN },
 426         [NL80211_REKEY_DATA_KCK] = { .len = NL80211_KCK_LEN },
 427         [NL80211_REKEY_DATA_REPLAY_CTR] = { .len = NL80211_REPLAY_CTR_LEN },
 428 };
 429
 430 static const struct nla_policy
 431 nl80211_match_policy[NL80211_SCHED_SCAN_MATCH_ATTR_MAX + 1] = {
 432         [NL80211_SCHED_SCAN_MATCH_ATTR_SSID] = { .type = NLA_BINARY,
 433                                                  .len = IEEE80211_MAX_SSID_LEN },
 434         [NL80211_SCHED_SCAN_MATCH_ATTR_RSSI] = { .type = NLA_U32 },
 435 };
 436
 437 static int nl80211_prepare_wdev_dump(struct sk_buff *skb,
 438                                      struct netlink_callback *cb,
 439                                      struct cfg80211_registered_device **rdev,
 440                                      struct wireless_dev **wdev)
 441 {
 442         int err;
 443
 444         rtnl_lock();
 445
 446         if (!cb->args[0]) {
 447                 err = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
 448                                   nl80211_fam.attrbuf, nl80211_fam.maxattr,
 449                                   nl80211_policy);
 450                 if (err)
 451                         goto out_unlock;
 452
 453                 *wdev = __cfg80211_wdev_from_attrs(sock_net(skb->sk),
 454                                                    nl80211_fam.attrbuf);
 455                 if (IS_ERR(*wdev)) {
 456                         err = PTR_ERR(*wdev);
 457                         goto out_unlock;
 458                 }
 459                 *rdev = wiphy_to_dev((*wdev)->wiphy);
 460                 /* 0 is the first index - add 1 to parse only once */
 461                 cb->args[0] = (*rdev)->wiphy_idx + 1;
 462                 cb->args[1] = (*wdev)->identifier;
 463         } else {
 464                 /* subtract the 1 again here */
 465                 struct wiphy *wiphy = wiphy_idx_to_wiphy(cb->args[0] - 1);
 466                 struct wireless_dev *tmp;
 467
 468                 if (!wiphy) {
 469                         err = -ENODEV;
 470                         goto out_unlock;
 471                 }
 472                 *rdev = wiphy_to_dev(wiphy);
 473                 *wdev = NULL;
 474
 475                 list_for_each_entry(tmp, &(*rdev)->wdev_list, list) {
 476                         if (tmp->identifier == cb->args[1]) {
 477                                 *wdev = tmp;
 478                                 break;
 479                         }
 480                 }
 481
 482                 if (!*wdev) {
 483                         err = -ENODEV;
 484                         goto out_unlock;
 485                 }
 486         }
 487
 488         return 0;
 489  out_unlock:
 490         rtnl_unlock();
 491         return err;
 492 }
 493
 494 static void nl80211_finish_wdev_dump(struct cfg80211_registered_device *rdev)
 495 {
 496         rtnl_unlock();
 497 }
 498
 499 /* IE validation */
 500 static bool is_valid_ie_attr(const struct nlattr *attr)
 501 {
 502         const u8 *pos;
 503         int len;
 504
 505         if (!attr)
 506                 return true;
 507
 508         pos = nla_data(attr);
 509         len = nla_len(attr);
 510
 511         while (len) {
 512                 u8 elemlen;
 513
 514                 if (len < 2)
 515                         return false;
 516                 len -= 2;
 517
 518                 elemlen = pos[1];
 519                 if (elemlen > len)
 520                         return false;
 521
 522                 len -= elemlen;
 523                 pos += 2 + elemlen;
 524         }
 525
 526         return true;
 527 }
 528
 529 /* message building helper */
 530 static inline void *nl80211hdr_put(struct sk_buff *skb, u32 portid, u32 seq,
 531                                    int flags, u8 cmd)
 532 {
 533         /* since there is no private header just add the generic one */
 534         return genlmsg_put(skb, portid, seq, &nl80211_fam, flags, cmd);
 535 }
 536
 537 static int nl80211_msg_put_channel(struct sk_buff *msg,
 538                                    struct ieee80211_channel *chan,
 539                                    bool large)
 540 {
 541         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_FREQ,
 542                         chan->center_freq))
 543                 goto nla_put_failure;
 544
 545         if ((chan->flags & IEEE80211_CHAN_DISABLED) &&
 546             nla_put_flag(msg, NL80211_FREQUENCY_ATTR_DISABLED))
 547                 goto nla_put_failure;
 548         if (chan->flags & IEEE80211_CHAN_NO_IR) {
 549                 if (nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_IR))
 550                         goto nla_put_failure;
 551                 if (nla_put_flag(msg, __NL80211_FREQUENCY_ATTR_NO_IBSS))
 552                         goto nla_put_failure;
 553         }
 554         if (chan->flags & IEEE80211_CHAN_RADAR) {
 555                 if (nla_put_flag(msg, NL80211_FREQUENCY_ATTR_RADAR))
 556                         goto nla_put_failure;
 557                 if (large) {
 558                         u32 time;
 559
 560                         time = elapsed_jiffies_msecs(chan->dfs_state_entered);
 561
 562                         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_DFS_STATE,
 563                                         chan->dfs_state))
 564                                 goto nla_put_failure;
 565                         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_DFS_TIME,
 566                                         time))
 567                                 goto nla_put_failure;
 568                 }
 569         }
 570
 571         if (large) {
 572                 if ((chan->flags & IEEE80211_CHAN_NO_HT40MINUS) &&
 573                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_HT40_MINUS))
 574                         goto nla_put_failure;
 575                 if ((chan->flags & IEEE80211_CHAN_NO_HT40PLUS) &&
 576                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_HT40_PLUS))
 577                         goto nla_put_failure;
 578                 if ((chan->flags & IEEE80211_CHAN_NO_80MHZ) &&
 579                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_80MHZ))
 580                         goto nla_put_failure;
 581                 if ((chan->flags & IEEE80211_CHAN_NO_160MHZ) &&
 582                     nla_put_flag(msg, NL80211_FREQUENCY_ATTR_NO_160MHZ))
 583                         goto nla_put_failure;
 584         }
 585
 586         if (nla_put_u32(msg, NL80211_FREQUENCY_ATTR_MAX_TX_POWER,
 587                         DBM_TO_MBM(chan->max_power)))
 588                 goto nla_put_failure;
 589
 590         return 0;
 591
 592  nla_put_failure:
 593         return -ENOBUFS;
 594 }
 595
 596 /* netlink command implementations */
 597
 598 struct key_parse {
 599         struct key_params p;
 600         int idx;
 601         int type;
 602         bool def, defmgmt;
 603         bool def_uni, def_multi;
 604 };
 605
 606 static int nl80211_parse_key_new(struct nlattr *key, struct key_parse *k)
 607 {
 608         struct nlattr *tb[NL80211_KEY_MAX + 1];
 609         int err = nla_parse_nested(tb, NL80211_KEY_MAX, key,
 610                                    nl80211_key_policy);
 611         if (err)
 612                 return err;
 613
 614         k->def = !!tb[NL80211_KEY_DEFAULT];
 615         k->defmgmt = !!tb[NL80211_KEY_DEFAULT_MGMT];
 616
 617         if (k->def) {
 618                 k->def_uni = true;
 619                 k->def_multi = true;
 620         }
 621         if (k->defmgmt)
 622                 k->def_multi = true;
 623
 624         if (tb[NL80211_KEY_IDX])
 625                 k->idx = nla_get_u8(tb[NL80211_KEY_IDX]);
 626
 627         if (tb[NL80211_KEY_DATA]) {
 628                 k->p.key = nla_data(tb[NL80211_KEY_DATA]);
 629                 k->p.key_len = nla_len(tb[NL80211_KEY_DATA]);
 630         }
 631
 632         if (tb[NL80211_KEY_SEQ]) {
 633                 k->p.seq = nla_data(tb[NL80211_KEY_SEQ]);
 634                 k->p.seq_len = nla_len(tb[NL80211_KEY_SEQ]);
 635         }
 636
 637         if (tb[NL80211_KEY_CIPHER])
 638                 k->p.cipher = nla_get_u32(tb[NL80211_KEY_CIPHER]);
 639
 640         if (tb[NL80211_KEY_TYPE]) {
 641                 k->type = nla_get_u32(tb[NL80211_KEY_TYPE]);
 642                 if (k->type < 0 || k->type >= NUM_NL80211_KEYTYPES)
 643                         return -EINVAL;
 644         }
 645
 646         if (tb[NL80211_KEY_DEFAULT_TYPES]) {
 647                 struct nlattr *kdt[NUM_NL80211_KEY_DEFAULT_TYPES];
 648                 err = nla_parse_nested(kdt, NUM_NL80211_KEY_DEFAULT_TYPES - 1,
 649                                        tb[NL80211_KEY_DEFAULT_TYPES],
 650                                        nl80211_key_default_policy);
 651                 if (err)
 652                         return err;
 653
 654                 k->def_uni = kdt[NL80211_KEY_DEFAULT_TYPE_UNICAST];
 655                 k->def_multi = kdt[NL80211_KEY_DEFAULT_TYPE_MULTICAST];
 656         }
 657
 658         return 0;
 659 }
 660
 661 static int nl80211_parse_key_old(struct genl_info *info, struct key_parse *k)
 662 {
 663         if (info->attrs[NL80211_ATTR_KEY_DATA]) {
 664                 k->p.key = nla_data(info->attrs[NL80211_ATTR_KEY_DATA]);
 665                 k->p.key_len = nla_len(info->attrs[NL80211_ATTR_KEY_DATA]);
 666         }
 667
 668         if (info->attrs[NL80211_ATTR_KEY_SEQ]) {
 669                 k->p.seq = nla_data(info->attrs[NL80211_ATTR_KEY_SEQ]);
 670                 k->p.seq_len = nla_len(info->attrs[NL80211_ATTR_KEY_SEQ]);
 671         }
 672
 673         if (info->attrs[NL80211_ATTR_KEY_IDX])
 674                 k->idx = nla_get_u8(info->attrs[NL80211_ATTR_KEY_IDX]);
 675
 676         if (info->attrs[NL80211_ATTR_KEY_CIPHER])
 677                 k->p.cipher = nla_get_u32(info->attrs[NL80211_ATTR_KEY_CIPHER]);
 678
 679         k->def = !!info->attrs[NL80211_ATTR_KEY_DEFAULT];
 680         k->defmgmt = !!info->attrs[NL80211_ATTR_KEY_DEFAULT_MGMT];
 681
 682         if (k->def) {
 683                 k->def_uni = true;
 684                 k->def_multi = true;
 685         }
 686         if (k->defmgmt)
 687                 k->def_multi = true;
 688
 689         if (info->attrs[NL80211_ATTR_KEY_TYPE]) {
 690                 k->type = nla_get_u32(info->attrs[NL80211_ATTR_KEY_TYPE]);
 691                 if (k->type < 0 || k->type >= NUM_NL80211_KEYTYPES)
 692                         return -EINVAL;
 693         }
 694
 695         if (info->attrs[NL80211_ATTR_KEY_DEFAULT_TYPES]) {
 696                 struct nlattr *kdt[NUM_NL80211_KEY_DEFAULT_TYPES];
 697                 int err = nla_parse_nested(
 698                                 kdt, NUM_NL80211_KEY_DEFAULT_TYPES - 1,
 699                                 info->attrs[NL80211_ATTR_KEY_DEFAULT_TYPES],
 700                                 nl80211_key_default_policy);
 701                 if (err)
 702                         return err;
 703
 704                 k->def_uni = kdt[NL80211_KEY_DEFAULT_TYPE_UNICAST];
 705                 k->def_multi = kdt[NL80211_KEY_DEFAULT_TYPE_MULTICAST];
 706         }
 707
 708         return 0;
 709 }
 710
 711 static int nl80211_parse_key(struct genl_info *info, struct key_parse *k)
 712 {
 713         int err;
 714
 715         memset(k, 0, sizeof(*k));
 716         k->idx = -1;
 717         k->type = -1;
 718
 719         if (info->attrs[NL80211_ATTR_KEY])
 720                 err = nl80211_parse_key_new(info->attrs[NL80211_ATTR_KEY], k);
 721         else
 722                 err = nl80211_parse_key_old(info, k);
 723
 724         if (err)
 725                 return err;
 726
 727         if (k->def && k->defmgmt)
 728                 return -EINVAL;
 729
 730         if (k->defmgmt) {
 731                 if (k->def_uni || !k->def_multi)
 732                         return -EINVAL;
 733         }
 734
 735         if (k->idx != -1) {
 736                 if (k->defmgmt) {
 737                         if (k->idx < 4 || k->idx > 5)
 738                                 return -EINVAL;
 739                 } else if (k->def) {
 740                         if (k->idx < 0 || k->idx > 3)
 741                                 return -EINVAL;
 742                 } else {
 743                         if (k->idx < 0 || k->idx > 5)
 744                                 return -EINVAL;
 745                 }
 746         }
 747
 748         return 0;
 749 }
 750
 751 static struct cfg80211_cached_keys *
 752 nl80211_parse_connkeys(struct cfg80211_registered_device *rdev,
 753                        struct nlattr *keys, bool *no_ht)
 754 {
 755         struct key_parse parse;
 756         struct nlattr *key;
 757         struct cfg80211_cached_keys *result;
 758         int rem, err, def = 0;
 759
 760         result = kzalloc(sizeof(*result), GFP_KERNEL);
 761         if (!result)
 762                 return ERR_PTR(-ENOMEM);
 763
 764         result->def = -1;
 765         result->defmgmt = -1;
 766
 767         nla_for_each_nested(key, keys, rem) {
 768                 memset(&parse, 0, sizeof(parse));
 769                 parse.idx = -1;
 770
 771                 err = nl80211_parse_key_new(key, &parse);
 772                 if (err)
 773                         goto error;
 774                 err = -EINVAL;
 775                 if (!parse.p.key)
 776                         goto error;
 777                 if (parse.idx < 0 || parse.idx > 4)
 778                         goto error;
 779                 if (parse.def) {
 780                         if (def)
 781                                 goto error;
 782                         def = 1;
 783                         result->def = parse.idx;
 784                         if (!parse.def_uni || !parse.def_multi)
 785                                 goto error;
 786                 } else if (parse.defmgmt)
 787                         goto error;
 788                 err = cfg80211_validate_key_settings(rdev, &parse.p,
 789                                                      parse.idx, false, NULL);
 790                 if (err)
 791                         goto error;
 792                 result->params[parse.idx].cipher = parse.p.cipher;
 793                 result->params[parse.idx].key_len = parse.p.key_len;
 794                 result->params[parse.idx].key = result->data[parse.idx];
 795                 memcpy(result->data[parse.idx], parse.p.key, parse.p.key_len);
 796
 797                 if (parse.p.cipher == WLAN_CIPHER_SUITE_WEP40 ||
 798                     parse.p.cipher == WLAN_CIPHER_SUITE_WEP104) {
 799                         if (no_ht)
 800                                 *no_ht = true;
 801                 }
 802         }
 803
 804         return result;
 805  error:
 806         kfree(result);
 807         return ERR_PTR(err);
 808 }
 809
 810 static int nl80211_key_allowed(struct wireless_dev *wdev)
 811 {
 812         ASSERT_WDEV_LOCK(wdev);
 813
 814         switch (wdev->iftype) {
 815         case NL80211_IFTYPE_AP:
 816         case NL80211_IFTYPE_AP_VLAN:
 817         case NL80211_IFTYPE_P2P_GO:
 818         case NL80211_IFTYPE_MESH_POINT:
 819                 break;
 820         case NL80211_IFTYPE_ADHOC:
 821         case NL80211_IFTYPE_STATION:
 822         case NL80211_IFTYPE_P2P_CLIENT:
 823                 if (!wdev->current_bss)
 824                         return -ENOLINK;
 825                 break;
 826         default:
 827                 return -EINVAL;
 828         }
 829
 830         return 0;
 831 }
 832
 833 static int nl80211_put_iftypes(struct sk_buff *msg, u32 attr, u16 ifmodes)
 834 {
 835         struct nlattr *nl_modes = nla_nest_start(msg, attr);
 836         int i;
 837
 838         if (!nl_modes)
 839                 goto nla_put_failure;
 840
 841         i = 0;
 842         while (ifmodes) {
 843                 if ((ifmodes & 1) && nla_put_flag(msg, i))
 844                         goto nla_put_failure;
 845                 ifmodes >>= 1;
 846                 i++;
 847         }
 848
 849         nla_nest_end(msg, nl_modes);
 850         return 0;
 851
 852 nla_put_failure:
 853         return -ENOBUFS;
 854 }
 855
 856 static int nl80211_put_iface_combinations(struct wiphy *wiphy,
 857                                           struct sk_buff *msg,
 858                                           bool large)
 859 {
 860         struct nlattr *nl_combis;
 861         int i, j;
 862
 863         nl_combis = nla_nest_start(msg,
 864                                 NL80211_ATTR_INTERFACE_COMBINATIONS);
 865         if (!nl_combis)
 866                 goto nla_put_failure;
 867
 868         for (i = 0; i < wiphy->n_iface_combinations; i++) {
 869                 const struct ieee80211_iface_combination *c;
 870                 struct nlattr *nl_combi, *nl_limits;
 871
 872                 c = &wiphy->iface_combinations[i];
 873
 874                 nl_combi = nla_nest_start(msg, i + 1);
 875                 if (!nl_combi)
 876                         goto nla_put_failure;
 877
 878                 nl_limits = nla_nest_start(msg, NL80211_IFACE_COMB_LIMITS);
 879                 if (!nl_limits)
 880                         goto nla_put_failure;
 881
 882                 for (j = 0; j < c->n_limits; j++) {
 883                         struct nlattr *nl_limit;
 884
 885                         nl_limit = nla_nest_start(msg, j + 1);
 886                         if (!nl_limit)
 887                                 goto nla_put_failure;
 888                         if (nla_put_u32(msg, NL80211_IFACE_LIMIT_MAX,
 889                                         c->limits[j].max))
 890                                 goto nla_put_failure;
 891                         if (nl80211_put_iftypes(msg, NL80211_IFACE_LIMIT_TYPES,
 892                                                 c->limits[j].types))
 893                                 goto nla_put_failure;
 894                         nla_nest_end(msg, nl_limit);
 895                 }
 896
 897                 nla_nest_end(msg, nl_limits);
 898
 899                 if (c->beacon_int_infra_match &&
 900                     nla_put_flag(msg, NL80211_IFACE_COMB_STA_AP_BI_MATCH))
 901                         goto nla_put_failure;
 902                 if (nla_put_u32(msg, NL80211_IFACE_COMB_NUM_CHANNELS,
 903                                 c->num_different_channels) ||
 904                     nla_put_u32(msg, NL80211_IFACE_COMB_MAXNUM,
 905                                 c->max_interfaces))
 906                         goto nla_put_failure;
 907                 if (large &&
 908                     nla_put_u32(msg, NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS,
 909                                 c->radar_detect_widths))
 910                         goto nla_put_failure;
 911
 912                 nla_nest_end(msg, nl_combi);
 913         }
 914
 915         nla_nest_end(msg, nl_combis);
 916
 917         return 0;
 918 nla_put_failure:
 919         return -ENOBUFS;
 920 }
 921
 922 #ifdef CONFIG_PM
 923 static int nl80211_send_wowlan_tcp_caps(struct cfg80211_registered_device *rdev,
 924                                         struct sk_buff *msg)
 925 {
 926         const struct wiphy_wowlan_tcp_support *tcp = rdev->wiphy.wowlan->tcp;
 927         struct nlattr *nl_tcp;
 928
 929         if (!tcp)
 930                 return 0;
 931
 932         nl_tcp = nla_nest_start(msg, NL80211_WOWLAN_TRIG_TCP_CONNECTION);
 933         if (!nl_tcp)
 934                 return -ENOBUFS;
 935
 936         if (nla_put_u32(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD,
 937                         tcp->data_payload_max))
 938                 return -ENOBUFS;
 939
 940         if (nla_put_u32(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD,
 941                         tcp->data_payload_max))
 942                 return -ENOBUFS;
 943
 944         if (tcp->seq && nla_put_flag(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ))
 945                 return -ENOBUFS;
 946
 947         if (tcp->tok && nla_put(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN,
 948                                 sizeof(*tcp->tok), tcp->tok))
 949                 return -ENOBUFS;
 950
 951         if (nla_put_u32(msg, NL80211_WOWLAN_TCP_DATA_INTERVAL,
 952                         tcp->data_interval_max))
 953                 return -ENOBUFS;
 954
 955         if (nla_put_u32(msg, NL80211_WOWLAN_TCP_WAKE_PAYLOAD,
 956                         tcp->wake_payload_max))
 957                 return -ENOBUFS;
 958
 959         nla_nest_end(msg, nl_tcp);
 960         return 0;
 961 }
 962
 963 static int nl80211_send_wowlan(struct sk_buff *msg,
 964                                struct cfg80211_registered_device *dev,
 965                                bool large)
 966 {
 967         struct nlattr *nl_wowlan;
 968
 969         if (!dev->wiphy.wowlan)
 970                 return 0;
 971
 972         nl_wowlan = nla_nest_start(msg, NL80211_ATTR_WOWLAN_TRIGGERS_SUPPORTED);
 973         if (!nl_wowlan)
 974                 return -ENOBUFS;
 975
 976         if (((dev->wiphy.wowlan->flags & WIPHY_WOWLAN_ANY) &&
 977              nla_put_flag(msg, NL80211_WOWLAN_TRIG_ANY)) ||
 978             ((dev->wiphy.wowlan->flags & WIPHY_WOWLAN_DISCONNECT) &&
 979              nla_put_flag(msg, NL80211_WOWLAN_TRIG_DISCONNECT)) ||
 980             ((dev->wiphy.wowlan->flags & WIPHY_WOWLAN_MAGIC_PKT) &&
 981              nla_put_flag(msg, NL80211_WOWLAN_TRIG_MAGIC_PKT)) ||
 982             ((dev->wiphy.wowlan->flags & WIPHY_WOWLAN_SUPPORTS_GTK_REKEY) &&
 983              nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_SUPPORTED)) ||
 984             ((dev->wiphy.wowlan->flags & WIPHY_WOWLAN_GTK_REKEY_FAILURE) &&
 985              nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE)) ||
 986             ((dev->wiphy.wowlan->flags & WIPHY_WOWLAN_EAP_IDENTITY_REQ) &&
 987              nla_put_flag(msg, NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST)) ||
 988             ((dev->wiphy.wowlan->flags & WIPHY_WOWLAN_4WAY_HANDSHAKE) &&
 989              nla_put_flag(msg, NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE)) ||
 990             ((dev->wiphy.wowlan->flags & WIPHY_WOWLAN_RFKILL_RELEASE) &&
 991              nla_put_flag(msg, NL80211_WOWLAN_TRIG_RFKILL_RELEASE)))
 992                 return -ENOBUFS;
 993
 994         if (dev->wiphy.wowlan->n_patterns) {
 995                 struct nl80211_pattern_support pat = {
 996                         .max_patterns = dev->wiphy.wowlan->n_patterns,
 997                         .min_pattern_len = dev->wiphy.wowlan->pattern_min_len,
 998                         .max_pattern_len = dev->wiphy.wowlan->pattern_max_len,
 999                         .max_pkt_offset = dev->wiphy.wowlan->max_pkt_offset,
1000                 };
1001
1002                 if (nla_put(msg, NL80211_WOWLAN_TRIG_PKT_PATTERN,
1003                             sizeof(pat), &pat))
1004                         return -ENOBUFS;
1005         }
1006
1007         if (large && nl80211_send_wowlan_tcp_caps(dev, msg))
1008                 return -ENOBUFS;
1009
1010         nla_nest_end(msg, nl_wowlan);
1011
1012         return 0;
1013 }
1014 #endif
1015
1016 static int nl80211_send_coalesce(struct sk_buff *msg,
1017                                  struct cfg80211_registered_device *dev)
1018 {
1019         struct nl80211_coalesce_rule_support rule;
1020
1021         if (!dev->wiphy.coalesce)
1022                 return 0;
1023
1024         rule.max_rules = dev->wiphy.coalesce->n_rules;
1025         rule.max_delay = dev->wiphy.coalesce->max_delay;
1026         rule.pat.max_patterns = dev->wiphy.coalesce->n_patterns;
1027         rule.pat.min_pattern_len = dev->wiphy.coalesce->pattern_min_len;
1028         rule.pat.max_pattern_len = dev->wiphy.coalesce->pattern_max_len;
1029         rule.pat.max_pkt_offset = dev->wiphy.coalesce->max_pkt_offset;
1030
1031         if (nla_put(msg, NL80211_ATTR_COALESCE_RULE, sizeof(rule), &rule))
1032                 return -ENOBUFS;
1033
1034         return 0;
1035 }
1036
1037 static int nl80211_send_band_rateinfo(struct sk_buff *msg,
1038                                       struct ieee80211_supported_band *sband)
1039 {
1040         struct nlattr *nl_rates, *nl_rate;
1041         struct ieee80211_rate *rate;
1042         int i;
1043
1044         /* add HT info */
1045         if (sband->ht_cap.ht_supported &&
1046             (nla_put(msg, NL80211_BAND_ATTR_HT_MCS_SET,
1047                      sizeof(sband->ht_cap.mcs),
1048                      &sband->ht_cap.mcs) ||
1049              nla_put_u16(msg, NL80211_BAND_ATTR_HT_CAPA,
1050                          sband->ht_cap.cap) ||
1051              nla_put_u8(msg, NL80211_BAND_ATTR_HT_AMPDU_FACTOR,
1052                         sband->ht_cap.ampdu_factor) ||
1053              nla_put_u8(msg, NL80211_BAND_ATTR_HT_AMPDU_DENSITY,
1054                         sband->ht_cap.ampdu_density)))
1055                 return -ENOBUFS;
1056
1057         /* add VHT info */
1058         if (sband->vht_cap.vht_supported &&
1059             (nla_put(msg, NL80211_BAND_ATTR_VHT_MCS_SET,
1060                      sizeof(sband->vht_cap.vht_mcs),
1061                      &sband->vht_cap.vht_mcs) ||
1062              nla_put_u32(msg, NL80211_BAND_ATTR_VHT_CAPA,
1063                          sband->vht_cap.cap)))
1064                 return -ENOBUFS;
1065
1066         /* add bitrates */
1067         nl_rates = nla_nest_start(msg, NL80211_BAND_ATTR_RATES);
1068         if (!nl_rates)
1069                 return -ENOBUFS;
1070
1071         for (i = 0; i < sband->n_bitrates; i++) {
1072                 nl_rate = nla_nest_start(msg, i);
1073                 if (!nl_rate)
1074                         return -ENOBUFS;
1075
1076                 rate = &sband->bitrates[i];
1077                 if (nla_put_u32(msg, NL80211_BITRATE_ATTR_RATE,
1078                                 rate->bitrate))
1079                         return -ENOBUFS;
1080                 if ((rate->flags & IEEE80211_RATE_SHORT_PREAMBLE) &&
1081                     nla_put_flag(msg,
1082                                  NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE))
1083                         return -ENOBUFS;
1084
1085                 nla_nest_end(msg, nl_rate);
1086         }
1087
1088         nla_nest_end(msg, nl_rates);
1089
1090         return 0;
1091 }
1092
1093 static int
1094 nl80211_send_mgmt_stypes(struct sk_buff *msg,
1095                          const struct ieee80211_txrx_stypes *mgmt_stypes)
1096 {
1097         u16 stypes;
1098         struct nlattr *nl_ftypes, *nl_ifs;
1099         enum nl80211_iftype ift;
1100         int i;
1101
1102         if (!mgmt_stypes)
1103                 return 0;
1104
1105         nl_ifs = nla_nest_start(msg, NL80211_ATTR_TX_FRAME_TYPES);
1106         if (!nl_ifs)
1107                 return -ENOBUFS;
1108
1109         for (ift = 0; ift < NUM_NL80211_IFTYPES; ift++) {
1110                 nl_ftypes = nla_nest_start(msg, ift);
1111                 if (!nl_ftypes)
1112                         return -ENOBUFS;
1113                 i = 0;
1114                 stypes = mgmt_stypes[ift].tx;
1115                 while (stypes) {
1116                         if ((stypes & 1) &&
1117                             nla_put_u16(msg, NL80211_ATTR_FRAME_TYPE,
1118                                         (i << 4) | IEEE80211_FTYPE_MGMT))
1119                                 return -ENOBUFS;
1120                         stypes >>= 1;
1121                         i++;
1122                 }
1123                 nla_nest_end(msg, nl_ftypes);
1124         }
1125
1126         nla_nest_end(msg, nl_ifs);
1127
1128         nl_ifs = nla_nest_start(msg, NL80211_ATTR_RX_FRAME_TYPES);
1129         if (!nl_ifs)
1130                 return -ENOBUFS;
1131
1132         for (ift = 0; ift < NUM_NL80211_IFTYPES; ift++) {
1133                 nl_ftypes = nla_nest_start(msg, ift);
1134                 if (!nl_ftypes)
1135                         return -ENOBUFS;
1136                 i = 0;
1137                 stypes = mgmt_stypes[ift].rx;
1138                 while (stypes) {
1139                         if ((stypes & 1) &&
1140                             nla_put_u16(msg, NL80211_ATTR_FRAME_TYPE,
1141                                         (i << 4) | IEEE80211_FTYPE_MGMT))
1142                                 return -ENOBUFS;
1143                         stypes >>= 1;
1144                         i++;
1145                 }
1146                 nla_nest_end(msg, nl_ftypes);
1147         }
1148         nla_nest_end(msg, nl_ifs);
1149
1150         return 0;
1151 }
1152
1153 struct nl80211_dump_wiphy_state {
1154         s64 filter_wiphy;
1155         long start;
1156         long split_start, band_start, chan_start;
1157         bool split;
1158 };
1159
1160 static int nl80211_send_wiphy(struct cfg80211_registered_device *dev,
1161                               struct sk_buff *msg, u32 portid, u32 seq,
1162                               int flags, struct nl80211_dump_wiphy_state *state)
1163 {
1164         void *hdr;
1165         struct nlattr *nl_bands, *nl_band;
1166         struct nlattr *nl_freqs, *nl_freq;
1167         struct nlattr *nl_cmds;
1168         enum ieee80211_band band;
1169         struct ieee80211_channel *chan;
1170         int i;
1171         const struct ieee80211_txrx_stypes *mgmt_stypes =
1172                                 dev->wiphy.mgmt_stypes;
1173         u32 features;
1174
1175         hdr = nl80211hdr_put(msg, portid, seq, flags, NL80211_CMD_NEW_WIPHY);
1176         if (!hdr)
1177                 return -ENOBUFS;
1178
1179         if (WARN_ON(!state))
1180                 return -EINVAL;
1181
1182         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, dev->wiphy_idx) ||
1183             nla_put_string(msg, NL80211_ATTR_WIPHY_NAME,
1184                            wiphy_name(&dev->wiphy)) ||
1185             nla_put_u32(msg, NL80211_ATTR_GENERATION,
1186                         cfg80211_rdev_list_generation))
1187                 goto nla_put_failure;
1188
1189         switch (state->split_start) {
1190         case 0:
1191                 if (nla_put_u8(msg, NL80211_ATTR_WIPHY_RETRY_SHORT,
1192                                dev->wiphy.retry_short) ||
1193                     nla_put_u8(msg, NL80211_ATTR_WIPHY_RETRY_LONG,
1194                                dev->wiphy.retry_long) ||
1195                     nla_put_u32(msg, NL80211_ATTR_WIPHY_FRAG_THRESHOLD,
1196                                 dev->wiphy.frag_threshold) ||
1197                     nla_put_u32(msg, NL80211_ATTR_WIPHY_RTS_THRESHOLD,
1198                                 dev->wiphy.rts_threshold) ||
1199                     nla_put_u8(msg, NL80211_ATTR_WIPHY_COVERAGE_CLASS,
1200                                dev->wiphy.coverage_class) ||
1201                     nla_put_u8(msg, NL80211_ATTR_MAX_NUM_SCAN_SSIDS,
1202                                dev->wiphy.max_scan_ssids) ||
1203                     nla_put_u8(msg, NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS,
1204                                dev->wiphy.max_sched_scan_ssids) ||
1205                     nla_put_u16(msg, NL80211_ATTR_MAX_SCAN_IE_LEN,
1206                                 dev->wiphy.max_scan_ie_len) ||
1207                     nla_put_u16(msg, NL80211_ATTR_MAX_SCHED_SCAN_IE_LEN,
1208                                 dev->wiphy.max_sched_scan_ie_len) ||
1209                     nla_put_u8(msg, NL80211_ATTR_MAX_MATCH_SETS,
1210                                dev->wiphy.max_match_sets))
1211                         goto nla_put_failure;
1212
1213                 if ((dev->wiphy.flags & WIPHY_FLAG_IBSS_RSN) &&
1214                     nla_put_flag(msg, NL80211_ATTR_SUPPORT_IBSS_RSN))
1215                         goto nla_put_failure;
1216                 if ((dev->wiphy.flags & WIPHY_FLAG_MESH_AUTH) &&
1217                     nla_put_flag(msg, NL80211_ATTR_SUPPORT_MESH_AUTH))
1218                         goto nla_put_failure;
1219                 if ((dev->wiphy.flags & WIPHY_FLAG_AP_UAPSD) &&
1220                     nla_put_flag(msg, NL80211_ATTR_SUPPORT_AP_UAPSD))
1221                         goto nla_put_failure;
1222                 if ((dev->wiphy.flags & WIPHY_FLAG_SUPPORTS_FW_ROAM) &&
1223                     nla_put_flag(msg, NL80211_ATTR_ROAM_SUPPORT))
1224                         goto nla_put_failure;
1225                 if ((dev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) &&
1226                     nla_put_flag(msg, NL80211_ATTR_TDLS_SUPPORT))
1227                         goto nla_put_failure;
1228                 if ((dev->wiphy.flags & WIPHY_FLAG_TDLS_EXTERNAL_SETUP) &&
1229                     nla_put_flag(msg, NL80211_ATTR_TDLS_EXTERNAL_SETUP))
1230                         goto nla_put_failure;
1231                 if ((dev->wiphy.flags & WIPHY_FLAG_SUPPORTS_5_10_MHZ) &&
1232                     nla_put_flag(msg, WIPHY_FLAG_SUPPORTS_5_10_MHZ))
1233                         goto nla_put_failure;
1234
1235                 state->split_start++;
1236                 if (state->split)
1237                         break;
1238         case 1:
1239                 if (nla_put(msg, NL80211_ATTR_CIPHER_SUITES,
1240                             sizeof(u32) * dev->wiphy.n_cipher_suites,
1241                             dev->wiphy.cipher_suites))
1242                         goto nla_put_failure;
1243
1244                 if (nla_put_u8(msg, NL80211_ATTR_MAX_NUM_PMKIDS,
1245                                dev->wiphy.max_num_pmkids))
1246                         goto nla_put_failure;
1247
1248                 if ((dev->wiphy.flags & WIPHY_FLAG_CONTROL_PORT_PROTOCOL) &&
1249                     nla_put_flag(msg, NL80211_ATTR_CONTROL_PORT_ETHERTYPE))
1250                         goto nla_put_failure;
1251
1252                 if (nla_put_u32(msg, NL80211_ATTR_WIPHY_ANTENNA_AVAIL_TX,
1253                                 dev->wiphy.available_antennas_tx) ||
1254                     nla_put_u32(msg, NL80211_ATTR_WIPHY_ANTENNA_AVAIL_RX,
1255                                 dev->wiphy.available_antennas_rx))
1256                         goto nla_put_failure;
1257
1258                 if ((dev->wiphy.flags & WIPHY_FLAG_AP_PROBE_RESP_OFFLOAD) &&
1259                     nla_put_u32(msg, NL80211_ATTR_PROBE_RESP_OFFLOAD,
1260                                 dev->wiphy.probe_resp_offload))
1261                         goto nla_put_failure;
1262
1263                 if ((dev->wiphy.available_antennas_tx ||
1264                      dev->wiphy.available_antennas_rx) &&
1265                     dev->ops->get_antenna) {
1266                         u32 tx_ant = 0, rx_ant = 0;
1267                         int res;
1268                         res = rdev_get_antenna(dev, &tx_ant, &rx_ant);
1269                         if (!res) {
1270                                 if (nla_put_u32(msg,
1271                                                 NL80211_ATTR_WIPHY_ANTENNA_TX,
1272                                                 tx_ant) ||
1273                                     nla_put_u32(msg,
1274                                                 NL80211_ATTR_WIPHY_ANTENNA_RX,
1275                                                 rx_ant))
1276                                         goto nla_put_failure;
1277                         }
1278                 }
1279
1280                 state->split_start++;
1281                 if (state->split)
1282                         break;
1283         case 2:
1284                 if (nl80211_put_iftypes(msg, NL80211_ATTR_SUPPORTED_IFTYPES,
1285                                         dev->wiphy.interface_modes))
1286                                 goto nla_put_failure;
1287                 state->split_start++;
1288                 if (state->split)
1289                         break;
1290         case 3:
1291                 nl_bands = nla_nest_start(msg, NL80211_ATTR_WIPHY_BANDS);
1292                 if (!nl_bands)
1293                         goto nla_put_failure;
1294
1295                 for (band = state->band_start;
1296                      band < IEEE80211_NUM_BANDS; band++) {
1297                         struct ieee80211_supported_band *sband;
1298
1299                         sband = dev->wiphy.bands[band];
1300
1301                         if (!sband)
1302                                 continue;
1303
1304                         nl_band = nla_nest_start(msg, band);
1305                         if (!nl_band)
1306                                 goto nla_put_failure;
1307
1308                         switch (state->chan_start) {
1309                         case 0:
1310                                 if (nl80211_send_band_rateinfo(msg, sband))
1311                                         goto nla_put_failure;
1312                                 state->chan_start++;
1313                                 if (state->split)
1314                                         break;
1315                         default:
1316                                 /* add frequencies */
1317                                 nl_freqs = nla_nest_start(
1318                                         msg, NL80211_BAND_ATTR_FREQS);
1319                                 if (!nl_freqs)
1320                                         goto nla_put_failure;
1321
1322                                 for (i = state->chan_start - 1;
1323                                      i < sband->n_channels;
1324                                      i++) {
1325                                         nl_freq = nla_nest_start(msg, i);
1326                                         if (!nl_freq)
1327                                                 goto nla_put_failure;
1328
1329                                         chan = &sband->channels[i];
1330
1331                                         if (nl80211_msg_put_channel(
1332                                                         msg, chan,
1333                                                         state->split))
1334                                                 goto nla_put_failure;
1335
1336                                         nla_nest_end(msg, nl_freq);
1337                                         if (state->split)
1338                                                 break;
1339                                 }
1340                                 if (i < sband->n_channels)
1341                                         state->chan_start = i + 2;
1342                                 else
1343                                         state->chan_start = 0;
1344                                 nla_nest_end(msg, nl_freqs);
1345                         }
1346
1347                         nla_nest_end(msg, nl_band);
1348
1349                         if (state->split) {
1350                                 /* start again here */
1351                                 if (state->chan_start)
1352                                         band--;
1353                                 break;
1354                         }
1355                 }
1356                 nla_nest_end(msg, nl_bands);
1357
1358                 if (band < IEEE80211_NUM_BANDS)
1359                         state->band_start = band + 1;
1360                 else
1361                         state->band_start = 0;
1362
1363                 /* if bands & channels are done, continue outside */
1364                 if (state->band_start == 0 && state->chan_start == 0)
1365                         state->split_start++;
1366                 if (state->split)
1367                         break;
1368         case 4:
1369                 nl_cmds = nla_nest_start(msg, NL80211_ATTR_SUPPORTED_COMMANDS);
1370                 if (!nl_cmds)
1371                         goto nla_put_failure;
1372
1373                 i = 0;
1374 #define CMD(op, n)                                                      \
1375                  do {                                                   \
1376                         if (dev->ops->op) {                             \
1377                                 i++;                                    \
1378                                 if (nla_put_u32(msg, i, NL80211_CMD_ ## n)) \
1379                                         goto nla_put_failure;           \
1380                         }                                               \
1381                 } while (0)
1382
1383                 CMD(add_virtual_intf, NEW_INTERFACE);
1384                 CMD(change_virtual_intf, SET_INTERFACE);
1385                 CMD(add_key, NEW_KEY);
1386                 CMD(start_ap, START_AP);
1387                 CMD(add_station, NEW_STATION);
1388                 CMD(add_mpath, NEW_MPATH);
1389                 CMD(update_mesh_config, SET_MESH_CONFIG);
1390                 CMD(change_bss, SET_BSS);
1391                 CMD(auth, AUTHENTICATE);
1392                 CMD(assoc, ASSOCIATE);
1393                 CMD(deauth, DEAUTHENTICATE);
1394                 CMD(disassoc, DISASSOCIATE);
1395                 CMD(join_ibss, JOIN_IBSS);
1396                 CMD(join_mesh, JOIN_MESH);
1397                 CMD(set_pmksa, SET_PMKSA);
1398                 CMD(del_pmksa, DEL_PMKSA);
1399                 CMD(flush_pmksa, FLUSH_PMKSA);
1400                 if (dev->wiphy.flags & WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL)
1401                         CMD(remain_on_channel, REMAIN_ON_CHANNEL);
1402                 CMD(set_bitrate_mask, SET_TX_BITRATE_MASK);
1403                 CMD(mgmt_tx, FRAME);
1404                 CMD(mgmt_tx_cancel_wait, FRAME_WAIT_CANCEL);
1405                 if (dev->wiphy.flags & WIPHY_FLAG_NETNS_OK) {
1406                         i++;
1407                         if (nla_put_u32(msg, i, NL80211_CMD_SET_WIPHY_NETNS))
1408                                 goto nla_put_failure;
1409                 }
1410                 if (dev->ops->set_monitor_channel || dev->ops->start_ap ||
1411                     dev->ops->join_mesh) {
1412                         i++;
1413                         if (nla_put_u32(msg, i, NL80211_CMD_SET_CHANNEL))
1414                                 goto nla_put_failure;
1415                 }
1416                 CMD(set_wds_peer, SET_WDS_PEER);
1417                 if (dev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) {
1418                         CMD(tdls_mgmt, TDLS_MGMT);
1419                         CMD(tdls_oper, TDLS_OPER);
1420                 }
1421                 if (dev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN)
1422                         CMD(sched_scan_start, START_SCHED_SCAN);
1423                 CMD(probe_client, PROBE_CLIENT);
1424                 CMD(set_noack_map, SET_NOACK_MAP);
1425                 if (dev->wiphy.flags & WIPHY_FLAG_REPORTS_OBSS) {
1426                         i++;
1427                         if (nla_put_u32(msg, i, NL80211_CMD_REGISTER_BEACONS))
1428                                 goto nla_put_failure;
1429                 }
1430                 CMD(start_p2p_device, START_P2P_DEVICE);
1431                 CMD(set_mcast_rate, SET_MCAST_RATE);
1432                 if (state->split) {
1433                         CMD(crit_proto_start, CRIT_PROTOCOL_START);
1434                         CMD(crit_proto_stop, CRIT_PROTOCOL_STOP);
1435                         if (dev->wiphy.flags & WIPHY_FLAG_HAS_CHANNEL_SWITCH)
1436                                 CMD(channel_switch, CHANNEL_SWITCH);
1437                 }
1438
1439 #ifdef CONFIG_NL80211_TESTMODE
1440                 CMD(testmode_cmd, TESTMODE);
1441 #endif
1442
1443 #undef CMD
1444
1445                 if (dev->ops->connect || dev->ops->auth) {
1446                         i++;
1447                         if (nla_put_u32(msg, i, NL80211_CMD_CONNECT))
1448                                 goto nla_put_failure;
1449                 }
1450
1451                 if (dev->ops->disconnect || dev->ops->deauth) {
1452                         i++;
1453                         if (nla_put_u32(msg, i, NL80211_CMD_DISCONNECT))
1454                                 goto nla_put_failure;
1455                 }
1456
1457                 nla_nest_end(msg, nl_cmds);
1458                 state->split_start++;
1459                 if (state->split)
1460                         break;
1461         case 5:
1462                 if (dev->ops->remain_on_channel &&
1463                     (dev->wiphy.flags & WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL) &&
1464                     nla_put_u32(msg,
1465                                 NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION,
1466                                 dev->wiphy.max_remain_on_channel_duration))
1467                         goto nla_put_failure;
1468
1469                 if ((dev->wiphy.flags & WIPHY_FLAG_OFFCHAN_TX) &&
1470                     nla_put_flag(msg, NL80211_ATTR_OFFCHANNEL_TX_OK))
1471                         goto nla_put_failure;
1472
1473                 if (nl80211_send_mgmt_stypes(msg, mgmt_stypes))
1474                         goto nla_put_failure;
1475                 state->split_start++;
1476                 if (state->split)
1477                         break;
1478         case 6:
1479 #ifdef CONFIG_PM
1480                 if (nl80211_send_wowlan(msg, dev, state->split))
1481                         goto nla_put_failure;
1482                 state->split_start++;
1483                 if (state->split)
1484                         break;
1485 #else
1486                 state->split_start++;
1487 #endif
1488         case 7:
1489                 if (nl80211_put_iftypes(msg, NL80211_ATTR_SOFTWARE_IFTYPES,
1490                                         dev->wiphy.software_iftypes))
1491                         goto nla_put_failure;
1492
1493                 if (nl80211_put_iface_combinations(&dev->wiphy, msg,
1494                                                    state->split))
1495                         goto nla_put_failure;
1496
1497                 state->split_start++;
1498                 if (state->split)
1499                         break;
1500         case 8:
1501                 if ((dev->wiphy.flags & WIPHY_FLAG_HAVE_AP_SME) &&
1502                     nla_put_u32(msg, NL80211_ATTR_DEVICE_AP_SME,
1503                                 dev->wiphy.ap_sme_capa))
1504                         goto nla_put_failure;
1505
1506                 features = dev->wiphy.features;
1507                 /*
1508                  * We can only add the per-channel limit information if the
1509                  * dump is split, otherwise it makes it too big. Therefore
1510                  * only advertise it in that case.
1511                  */
1512                 if (state->split)
1513                         features |= NL80211_FEATURE_ADVERTISE_CHAN_LIMITS;
1514                 if (nla_put_u32(msg, NL80211_ATTR_FEATURE_FLAGS, features))
1515                         goto nla_put_failure;
1516
1517                 if (dev->wiphy.ht_capa_mod_mask &&
1518                     nla_put(msg, NL80211_ATTR_HT_CAPABILITY_MASK,
1519                             sizeof(*dev->wiphy.ht_capa_mod_mask),
1520                             dev->wiphy.ht_capa_mod_mask))
1521                         goto nla_put_failure;
1522
1523                 if (dev->wiphy.flags & WIPHY_FLAG_HAVE_AP_SME &&
1524                     dev->wiphy.max_acl_mac_addrs &&
1525                     nla_put_u32(msg, NL80211_ATTR_MAC_ACL_MAX,
1526                                 dev->wiphy.max_acl_mac_addrs))
1527                         goto nla_put_failure;
1528
1529                 /*
1530                  * Any information below this point is only available to
1531                  * applications that can deal with it being split. This
1532                  * helps ensure that newly added capabilities don't break
1533                  * older tools by overrunning their buffers.
1534                  *
1535                  * We still increment split_start so that in the split
1536                  * case we'll continue with more data in the next round,
1537                  * but break unconditionally so unsplit data stops here.
1538                  */
1539                 state->split_start++;
1540                 break;
1541         case 9:
1542                 if (dev->wiphy.extended_capabilities &&
1543                     (nla_put(msg, NL80211_ATTR_EXT_CAPA,
1544                              dev->wiphy.extended_capabilities_len,
1545                              dev->wiphy.extended_capabilities) ||
1546                      nla_put(msg, NL80211_ATTR_EXT_CAPA_MASK,
1547                              dev->wiphy.extended_capabilities_len,
1548                              dev->wiphy.extended_capabilities_mask)))
1549                         goto nla_put_failure;
1550
1551                 if (dev->wiphy.vht_capa_mod_mask &&
1552                     nla_put(msg, NL80211_ATTR_VHT_CAPABILITY_MASK,
1553                             sizeof(*dev->wiphy.vht_capa_mod_mask),
1554                             dev->wiphy.vht_capa_mod_mask))
1555                         goto nla_put_failure;
1556
1557                 state->split_start++;
1558                 break;
1559         case 10:
1560                 if (nl80211_send_coalesce(msg, dev))
1561                         goto nla_put_failure;
1562
1563                 /* done */
1564                 state->split_start = 0;
1565                 break;
1566         }
1567         return genlmsg_end(msg, hdr);
1568
1569  nla_put_failure:
1570         genlmsg_cancel(msg, hdr);
1571         return -EMSGSIZE;
1572 }
1573
1574 static int nl80211_dump_wiphy_parse(struct sk_buff *skb,
1575                                     struct netlink_callback *cb,
1576                                     struct nl80211_dump_wiphy_state *state)
1577 {
1578         struct nlattr **tb = nl80211_fam.attrbuf;
1579         int ret = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
1580                               tb, nl80211_fam.maxattr, nl80211_policy);
1581         /* ignore parse errors for backward compatibility */
1582         if (ret)
1583                 return 0;
1584
1585         state->split = tb[NL80211_ATTR_SPLIT_WIPHY_DUMP];
1586         if (tb[NL80211_ATTR_WIPHY])
1587                 state->filter_wiphy = nla_get_u32(tb[NL80211_ATTR_WIPHY]);
1588         if (tb[NL80211_ATTR_WDEV])
1589                 state->filter_wiphy = nla_get_u64(tb[NL80211_ATTR_WDEV]) >> 32;
1590         if (tb[NL80211_ATTR_IFINDEX]) {
1591                 struct net_device *netdev;
1592                 struct cfg80211_registered_device *rdev;
1593                 int ifidx = nla_get_u32(tb[NL80211_ATTR_IFINDEX]);
1594
1595                 netdev = dev_get_by_index(sock_net(skb->sk), ifidx);
1596                 if (!netdev)
1597                         return -ENODEV;
1598                 if (netdev->ieee80211_ptr) {
1599                         rdev = wiphy_to_dev(
1600                                 netdev->ieee80211_ptr->wiphy);
1601                         state->filter_wiphy = rdev->wiphy_idx;
1602                 }
1603                 dev_put(netdev);
1604         }
1605
1606         return 0;
1607 }
1608
1609 static int nl80211_dump_wiphy(struct sk_buff *skb, struct netlink_callback *cb)
1610 {
1611         int idx = 0, ret;
1612         struct nl80211_dump_wiphy_state *state = (void *)cb->args[0];
1613         struct cfg80211_registered_device *dev;
1614
1615         rtnl_lock();
1616         if (!state) {
1617                 state = kzalloc(sizeof(*state), GFP_KERNEL);
1618                 if (!state) {
1619                         rtnl_unlock();
1620                         return -ENOMEM;
1621                 }
1622                 state->filter_wiphy = -1;
1623                 ret = nl80211_dump_wiphy_parse(skb, cb, state);
1624                 if (ret) {
1625                         kfree(state);
1626                         rtnl_unlock();
1627                         return ret;
1628                 }
1629                 cb->args[0] = (long)state;
1630         }
1631
1632         list_for_each_entry(dev, &cfg80211_rdev_list, list) {
1633                 if (!net_eq(wiphy_net(&dev->wiphy), sock_net(skb->sk)))
1634                         continue;
1635                 if (++idx <= state->start)
1636                         continue;
1637                 if (state->filter_wiphy != -1 &&
1638                     state->filter_wiphy != dev->wiphy_idx)
1639                         continue;
1640                 /* attempt to fit multiple wiphy data chunks into the skb */
1641                 do {
1642                         ret = nl80211_send_wiphy(dev, skb,
1643                                                  NETLINK_CB(cb->skb).portid,
1644                                                  cb->nlh->nlmsg_seq,
1645                                                  NLM_F_MULTI, state);
1646                         if (ret < 0) {
1647                                 /*
1648                                  * If sending the wiphy data didn't fit (ENOBUFS
1649                                  * or EMSGSIZE returned), this SKB is still
1650                                  * empty (so it's not too big because another
1651                                  * wiphy dataset is already in the skb) and
1652                                  * we've not tried to adjust the dump allocation
1653                                  * yet ... then adjust the alloc size to be
1654                                  * bigger, and return 1 but with the empty skb.
1655                                  * This results in an empty message being RX'ed
1656                                  * in userspace, but that is ignored.
1657                                  *
1658                                  * We can then retry with the larger buffer.
1659                                  */
1660                                 if ((ret == -ENOBUFS || ret == -EMSGSIZE) &&
1661                                     !skb->len &&
1662                                     cb->min_dump_alloc < 4096) {
1663                                         cb->min_dump_alloc = 4096;
1664                                         rtnl_unlock();
1665                                         return 1;
1666                                 }
1667                                 idx--;
1668                                 break;
1669                         }
1670                 } while (state->split_start > 0);
1671                 break;
1672         }
1673         rtnl_unlock();
1674
1675         state->start = idx;
1676
1677         return skb->len;
1678 }
1679
1680 static int nl80211_dump_wiphy_done(struct netlink_callback *cb)
1681 {
1682         kfree((void *)cb->args[0]);
1683         return 0;
1684 }
1685
1686 static int nl80211_get_wiphy(struct sk_buff *skb, struct genl_info *info)
1687 {
1688         struct sk_buff *msg;
1689         struct cfg80211_registered_device *dev = info->user_ptr[0];
1690         struct nl80211_dump_wiphy_state state = {};
1691
1692         msg = nlmsg_new(4096, GFP_KERNEL);
1693         if (!msg)
1694                 return -ENOMEM;
1695
1696         if (nl80211_send_wiphy(dev, msg, info->snd_portid, info->snd_seq, 0,
1697                                &state) < 0) {
1698                 nlmsg_free(msg);
1699                 return -ENOBUFS;
1700         }
1701
1702         return genlmsg_reply(msg, info);
1703 }
1704
1705 static const struct nla_policy txq_params_policy[NL80211_TXQ_ATTR_MAX + 1] = {
1706         [NL80211_TXQ_ATTR_QUEUE]                = { .type = NLA_U8 },
1707         [NL80211_TXQ_ATTR_TXOP]                 = { .type = NLA_U16 },
1708         [NL80211_TXQ_ATTR_CWMIN]                = { .type = NLA_U16 },
1709         [NL80211_TXQ_ATTR_CWMAX]                = { .type = NLA_U16 },
1710         [NL80211_TXQ_ATTR_AIFS]                 = { .type = NLA_U8 },
1711 };
1712
1713 static int parse_txq_params(struct nlattr *tb[],
1714                             struct ieee80211_txq_params *txq_params)
1715 {
1716         if (!tb[NL80211_TXQ_ATTR_AC] || !tb[NL80211_TXQ_ATTR_TXOP] ||
1717             !tb[NL80211_TXQ_ATTR_CWMIN] || !tb[NL80211_TXQ_ATTR_CWMAX] ||
1718             !tb[NL80211_TXQ_ATTR_AIFS])
1719                 return -EINVAL;
1720
1721         txq_params->ac = nla_get_u8(tb[NL80211_TXQ_ATTR_AC]);
1722         txq_params->txop = nla_get_u16(tb[NL80211_TXQ_ATTR_TXOP]);
1723         txq_params->cwmin = nla_get_u16(tb[NL80211_TXQ_ATTR_CWMIN]);
1724         txq_params->cwmax = nla_get_u16(tb[NL80211_TXQ_ATTR_CWMAX]);
1725         txq_params->aifs = nla_get_u8(tb[NL80211_TXQ_ATTR_AIFS]);
1726
1727         if (txq_params->ac >= NL80211_NUM_ACS)
1728                 return -EINVAL;
1729
1730         return 0;
1731 }
1732
1733 static bool nl80211_can_set_dev_channel(struct wireless_dev *wdev)
1734 {
1735         /*
1736          * You can only set the channel explicitly for WDS interfaces,
1737          * all others have their channel managed via their respective
1738          * "establish a connection" command (connect, join, ...)
1739          *
1740          * For AP/GO and mesh mode, the channel can be set with the
1741          * channel userspace API, but is only stored and passed to the
1742          * low-level driver when the AP starts or the mesh is joined.
1743          * This is for backward compatibility, userspace can also give
1744          * the channel in the start-ap or join-mesh commands instead.
1745          *
1746          * Monitors are special as they are normally slaved to
1747          * whatever else is going on, so they have their own special
1748          * operation to set the monitor channel if possible.
1749          */
1750         return !wdev ||
1751                 wdev->iftype == NL80211_IFTYPE_AP ||
1752                 wdev->iftype == NL80211_IFTYPE_MESH_POINT ||
1753                 wdev->iftype == NL80211_IFTYPE_MONITOR ||
1754                 wdev->iftype == NL80211_IFTYPE_P2P_GO;
1755 }
1756
1757 static int nl80211_parse_chandef(struct cfg80211_registered_device *rdev,
1758                                  struct genl_info *info,
1759                                  struct cfg80211_chan_def *chandef)
1760 {
1761         u32 control_freq;
1762
1763         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ])
1764                 return -EINVAL;
1765
1766         control_freq = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]);
1767
1768         chandef->chan = ieee80211_get_channel(&rdev->wiphy, control_freq);
1769         chandef->width = NL80211_CHAN_WIDTH_20_NOHT;
1770         chandef->center_freq1 = control_freq;
1771         chandef->center_freq2 = 0;
1772
1773         /* Primary channel not allowed */
1774         if (!chandef->chan || chandef->chan->flags & IEEE80211_CHAN_DISABLED)
1775                 return -EINVAL;
1776
1777         if (info->attrs[NL80211_ATTR_WIPHY_CHANNEL_TYPE]) {
1778                 enum nl80211_channel_type chantype;
1779
1780                 chantype = nla_get_u32(
1781                                 info->attrs[NL80211_ATTR_WIPHY_CHANNEL_TYPE]);
1782
1783                 switch (chantype) {
1784                 case NL80211_CHAN_NO_HT:
1785                 case NL80211_CHAN_HT20:
1786                 case NL80211_CHAN_HT40PLUS:
1787                 case NL80211_CHAN_HT40MINUS:
1788                         cfg80211_chandef_create(chandef, chandef->chan,
1789                                                 chantype);
1790                         break;
1791                 default:
1792                         return -EINVAL;
1793                 }
1794         } else if (info->attrs[NL80211_ATTR_CHANNEL_WIDTH]) {
1795                 chandef->width =
1796                         nla_get_u32(info->attrs[NL80211_ATTR_CHANNEL_WIDTH]);
1797                 if (info->attrs[NL80211_ATTR_CENTER_FREQ1])
1798                         chandef->center_freq1 =
1799                                 nla_get_u32(
1800                                         info->attrs[NL80211_ATTR_CENTER_FREQ1]);
1801                 if (info->attrs[NL80211_ATTR_CENTER_FREQ2])
1802                         chandef->center_freq2 =
1803                                 nla_get_u32(
1804                                         info->attrs[NL80211_ATTR_CENTER_FREQ2]);
1805         }
1806
1807         if (!cfg80211_chandef_valid(chandef))
1808                 return -EINVAL;
1809
1810         if (!cfg80211_chandef_usable(&rdev->wiphy, chandef,
1811                                      IEEE80211_CHAN_DISABLED))
1812                 return -EINVAL;
1813
1814         if ((chandef->width == NL80211_CHAN_WIDTH_5 ||
1815              chandef->width == NL80211_CHAN_WIDTH_10) &&
1816             !(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_5_10_MHZ))
1817                 return -EINVAL;
1818
1819         return 0;
1820 }
1821
1822 static int __nl80211_set_channel(struct cfg80211_registered_device *rdev,
1823                                  struct wireless_dev *wdev,
1824                                  struct genl_info *info)
1825 {
1826         struct cfg80211_chan_def chandef;
1827         int result;
1828         enum nl80211_iftype iftype = NL80211_IFTYPE_MONITOR;
1829
1830         if (wdev)
1831                 iftype = wdev->iftype;
1832
1833         if (!nl80211_can_set_dev_channel(wdev))
1834                 return -EOPNOTSUPP;
1835
1836         result = nl80211_parse_chandef(rdev, info, &chandef);
1837         if (result)
1838                 return result;
1839
1840         switch (iftype) {
1841         case NL80211_IFTYPE_AP:
1842         case NL80211_IFTYPE_P2P_GO:
1843                 if (wdev->beacon_interval) {
1844                         result = -EBUSY;
1845                         break;
1846                 }
1847                 if (!cfg80211_reg_can_beacon(&rdev->wiphy, &chandef)) {
1848                         result = -EINVAL;
1849                         break;
1850                 }
1851                 wdev->preset_chandef = chandef;
1852                 result = 0;
1853                 break;
1854         case NL80211_IFTYPE_MESH_POINT:
1855                 result = cfg80211_set_mesh_channel(rdev, wdev, &chandef);
1856                 break;
1857         case NL80211_IFTYPE_MONITOR:
1858                 result = cfg80211_set_monitor_channel(rdev, &chandef);
1859                 break;
1860         default:
1861                 result = -EINVAL;
1862         }
1863
1864         return result;
1865 }
1866
1867 static int nl80211_set_channel(struct sk_buff *skb, struct genl_info *info)
1868 {
1869         struct cfg80211_registered_device *rdev = info->user_ptr[0];
1870         struct net_device *netdev = info->user_ptr[1];
1871
1872         return __nl80211_set_channel(rdev, netdev->ieee80211_ptr, info);
1873 }
1874
1875 static int nl80211_set_wds_peer(struct sk_buff *skb, struct genl_info *info)
1876 {
1877         struct cfg80211_registered_device *rdev = info->user_ptr[0];
1878         struct net_device *dev = info->user_ptr[1];
1879         struct wireless_dev *wdev = dev->ieee80211_ptr;
1880         const u8 *bssid;
1881
1882         if (!info->attrs[NL80211_ATTR_MAC])
1883                 return -EINVAL;
1884
1885         if (netif_running(dev))
1886                 return -EBUSY;
1887
1888         if (!rdev->ops->set_wds_peer)
1889                 return -EOPNOTSUPP;
1890
1891         if (wdev->iftype != NL80211_IFTYPE_WDS)
1892                 return -EOPNOTSUPP;
1893
1894         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
1895         return rdev_set_wds_peer(rdev, dev, bssid);
1896 }
1897
1898
1899 static int nl80211_set_wiphy(struct sk_buff *skb, struct genl_info *info)
1900 {
1901         struct cfg80211_registered_device *rdev;
1902         struct net_device *netdev = NULL;
1903         struct wireless_dev *wdev;
1904         int result = 0, rem_txq_params = 0;
1905         struct nlattr *nl_txq_params;
1906         u32 changed;
1907         u8 retry_short = 0, retry_long = 0;
1908         u32 frag_threshold = 0, rts_threshold = 0;
1909         u8 coverage_class = 0;
1910
1911         ASSERT_RTNL();
1912
1913         /*
1914          * Try to find the wiphy and netdev. Normally this
1915          * function shouldn't need the netdev, but this is
1916          * done for backward compatibility -- previously
1917          * setting the channel was done per wiphy, but now
1918          * it is per netdev. Previous userland like hostapd
1919          * also passed a netdev to set_wiphy, so that it is
1920          * possible to let that go to the right netdev!
1921          */
1922
1923         if (info->attrs[NL80211_ATTR_IFINDEX]) {
1924                 int ifindex = nla_get_u32(info->attrs[NL80211_ATTR_IFINDEX]);
1925
1926                 netdev = dev_get_by_index(genl_info_net(info), ifindex);
1927                 if (netdev && netdev->ieee80211_ptr)
1928                         rdev = wiphy_to_dev(netdev->ieee80211_ptr->wiphy);
1929                 else
1930                         netdev = NULL;
1931         }
1932
1933         if (!netdev) {
1934                 rdev = __cfg80211_rdev_from_attrs(genl_info_net(info),
1935                                                   info->attrs);
1936                 if (IS_ERR(rdev))
1937                         return PTR_ERR(rdev);
1938                 wdev = NULL;
1939                 netdev = NULL;
1940                 result = 0;
1941         } else
1942                 wdev = netdev->ieee80211_ptr;
1943
1944         /*
1945          * end workaround code, by now the rdev is available
1946          * and locked, and wdev may or may not be NULL.
1947          */
1948
1949         if (info->attrs[NL80211_ATTR_WIPHY_NAME])
1950                 result = cfg80211_dev_rename(
1951                         rdev, nla_data(info->attrs[NL80211_ATTR_WIPHY_NAME]));
1952
1953         if (result)
1954                 goto bad_res;
1955
1956         if (info->attrs[NL80211_ATTR_WIPHY_TXQ_PARAMS]) {
1957                 struct ieee80211_txq_params txq_params;
1958                 struct nlattr *tb[NL80211_TXQ_ATTR_MAX + 1];
1959
1960                 if (!rdev->ops->set_txq_params) {
1961                         result = -EOPNOTSUPP;
1962                         goto bad_res;
1963                 }
1964
1965                 if (!netdev) {
1966                         result = -EINVAL;
1967                         goto bad_res;
1968                 }
1969
1970                 if (netdev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
1971                     netdev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO) {
1972                         result = -EINVAL;
1973                         goto bad_res;
1974                 }
1975
1976                 if (!netif_running(netdev)) {
1977                         result = -ENETDOWN;
1978                         goto bad_res;
1979                 }
1980
1981                 nla_for_each_nested(nl_txq_params,
1982                                     info->attrs[NL80211_ATTR_WIPHY_TXQ_PARAMS],
1983                                     rem_txq_params) {
1984                         nla_parse(tb, NL80211_TXQ_ATTR_MAX,
1985                                   nla_data(nl_txq_params),
1986                                   nla_len(nl_txq_params),
1987                                   txq_params_policy);
1988                         result = parse_txq_params(tb, &txq_params);
1989                         if (result)
1990                                 goto bad_res;
1991
1992                         result = rdev_set_txq_params(rdev, netdev,
1993                                                      &txq_params);
1994                         if (result)
1995                                 goto bad_res;
1996                 }
1997         }
1998
1999         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
2000                 result = __nl80211_set_channel(rdev,
2001                                 nl80211_can_set_dev_channel(wdev) ? wdev : NULL,
2002                                 info);
2003                 if (result)
2004                         goto bad_res;
2005         }
2006
2007         if (info->attrs[NL80211_ATTR_WIPHY_TX_POWER_SETTING]) {
2008                 struct wireless_dev *txp_wdev = wdev;
2009                 enum nl80211_tx_power_setting type;
2010                 int idx, mbm = 0;
2011
2012                 if (!(rdev->wiphy.features & NL80211_FEATURE_VIF_TXPOWER))
2013                         txp_wdev = NULL;
2014
2015                 if (!rdev->ops->set_tx_power) {
2016                         result = -EOPNOTSUPP;
2017                         goto bad_res;
2018                 }
2019
2020                 idx = NL80211_ATTR_WIPHY_TX_POWER_SETTING;
2021                 type = nla_get_u32(info->attrs[idx]);
2022
2023                 if (!info->attrs[NL80211_ATTR_WIPHY_TX_POWER_LEVEL] &&
2024                     (type != NL80211_TX_POWER_AUTOMATIC)) {
2025                         result = -EINVAL;
2026                         goto bad_res;
2027                 }
2028
2029                 if (type != NL80211_TX_POWER_AUTOMATIC) {
2030                         idx = NL80211_ATTR_WIPHY_TX_POWER_LEVEL;
2031                         mbm = nla_get_u32(info->attrs[idx]);
2032                 }
2033
2034                 result = rdev_set_tx_power(rdev, txp_wdev, type, mbm);
2035                 if (result)
2036                         goto bad_res;
2037         }
2038
2039         if (info->attrs[NL80211_ATTR_WIPHY_ANTENNA_TX] &&
2040             info->attrs[NL80211_ATTR_WIPHY_ANTENNA_RX]) {
2041                 u32 tx_ant, rx_ant;
2042                 if ((!rdev->wiphy.available_antennas_tx &&
2043                      !rdev->wiphy.available_antennas_rx) ||
2044                     !rdev->ops->set_antenna) {
2045                         result = -EOPNOTSUPP;
2046                         goto bad_res;
2047                 }
2048
2049                 tx_ant = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_ANTENNA_TX]);
2050                 rx_ant = nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_ANTENNA_RX]);
2051
2052                 /* reject antenna configurations which don't match the
2053                  * available antenna masks, except for the "all" mask */
2054                 if ((~tx_ant && (tx_ant & ~rdev->wiphy.available_antennas_tx)) ||
2055                     (~rx_ant && (rx_ant & ~rdev->wiphy.available_antennas_rx))) {
2056                         result = -EINVAL;
2057                         goto bad_res;
2058                 }
2059
2060                 tx_ant = tx_ant & rdev->wiphy.available_antennas_tx;
2061                 rx_ant = rx_ant & rdev->wiphy.available_antennas_rx;
2062
2063                 result = rdev_set_antenna(rdev, tx_ant, rx_ant);
2064                 if (result)
2065                         goto bad_res;
2066         }
2067
2068         changed = 0;
2069
2070         if (info->attrs[NL80211_ATTR_WIPHY_RETRY_SHORT]) {
2071                 retry_short = nla_get_u8(
2072                         info->attrs[NL80211_ATTR_WIPHY_RETRY_SHORT]);
2073                 if (retry_short == 0) {
2074                         result = -EINVAL;
2075                         goto bad_res;
2076                 }
2077                 changed |= WIPHY_PARAM_RETRY_SHORT;
2078         }
2079
2080         if (info->attrs[NL80211_ATTR_WIPHY_RETRY_LONG]) {
2081                 retry_long = nla_get_u8(
2082                         info->attrs[NL80211_ATTR_WIPHY_RETRY_LONG]);
2083                 if (retry_long == 0) {
2084                         result = -EINVAL;
2085                         goto bad_res;
2086                 }
2087                 changed |= WIPHY_PARAM_RETRY_LONG;
2088         }
2089
2090         if (info->attrs[NL80211_ATTR_WIPHY_FRAG_THRESHOLD]) {
2091                 frag_threshold = nla_get_u32(
2092                         info->attrs[NL80211_ATTR_WIPHY_FRAG_THRESHOLD]);
2093                 if (frag_threshold < 256) {
2094                         result = -EINVAL;
2095                         goto bad_res;
2096                 }
2097                 if (frag_threshold != (u32) -1) {
2098                         /*
2099                          * Fragments (apart from the last one) are required to
2100                          * have even length. Make the fragmentation code
2101                          * simpler by stripping LSB should someone try to use
2102                          * odd threshold value.
2103                          */
2104                         frag_threshold &= ~0x1;
2105                 }
2106                 changed |= WIPHY_PARAM_FRAG_THRESHOLD;
2107         }
2108
2109         if (info->attrs[NL80211_ATTR_WIPHY_RTS_THRESHOLD]) {
2110                 rts_threshold = nla_get_u32(
2111                         info->attrs[NL80211_ATTR_WIPHY_RTS_THRESHOLD]);
2112                 changed |= WIPHY_PARAM_RTS_THRESHOLD;
2113         }
2114
2115         if (info->attrs[NL80211_ATTR_WIPHY_COVERAGE_CLASS]) {
2116                 coverage_class = nla_get_u8(
2117                         info->attrs[NL80211_ATTR_WIPHY_COVERAGE_CLASS]);
2118                 changed |= WIPHY_PARAM_COVERAGE_CLASS;
2119         }
2120
2121         if (changed) {
2122                 u8 old_retry_short, old_retry_long;
2123                 u32 old_frag_threshold, old_rts_threshold;
2124                 u8 old_coverage_class;
2125
2126                 if (!rdev->ops->set_wiphy_params) {
2127                         result = -EOPNOTSUPP;
2128                         goto bad_res;
2129                 }
2130
2131                 old_retry_short = rdev->wiphy.retry_short;
2132                 old_retry_long = rdev->wiphy.retry_long;
2133                 old_frag_threshold = rdev->wiphy.frag_threshold;
2134                 old_rts_threshold = rdev->wiphy.rts_threshold;
2135                 old_coverage_class = rdev->wiphy.coverage_class;
2136
2137                 if (changed & WIPHY_PARAM_RETRY_SHORT)
2138                         rdev->wiphy.retry_short = retry_short;
2139                 if (changed & WIPHY_PARAM_RETRY_LONG)
2140                         rdev->wiphy.retry_long = retry_long;
2141                 if (changed & WIPHY_PARAM_FRAG_THRESHOLD)
2142                         rdev->wiphy.frag_threshold = frag_threshold;
2143                 if (changed & WIPHY_PARAM_RTS_THRESHOLD)
2144                         rdev->wiphy.rts_threshold = rts_threshold;
2145                 if (changed & WIPHY_PARAM_COVERAGE_CLASS)
2146                         rdev->wiphy.coverage_class = coverage_class;
2147
2148                 result = rdev_set_wiphy_params(rdev, changed);
2149                 if (result) {
2150                         rdev->wiphy.retry_short = old_retry_short;
2151                         rdev->wiphy.retry_long = old_retry_long;
2152                         rdev->wiphy.frag_threshold = old_frag_threshold;
2153                         rdev->wiphy.rts_threshold = old_rts_threshold;
2154                         rdev->wiphy.coverage_class = old_coverage_class;
2155                 }
2156         }
2157
2158  bad_res:
2159         if (netdev)
2160                 dev_put(netdev);
2161         return result;
2162 }
2163
2164 static inline u64 wdev_id(struct wireless_dev *wdev)
2165 {
2166         return (u64)wdev->identifier |
2167                ((u64)wiphy_to_dev(wdev->wiphy)->wiphy_idx << 32);
2168 }
2169
2170 static int nl80211_send_chandef(struct sk_buff *msg,
2171                                 const struct cfg80211_chan_def *chandef)
2172 {
2173         WARN_ON(!cfg80211_chandef_valid(chandef));
2174
2175         if (nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ,
2176                         chandef->chan->center_freq))
2177                 return -ENOBUFS;
2178         switch (chandef->width) {
2179         case NL80211_CHAN_WIDTH_20_NOHT:
2180         case NL80211_CHAN_WIDTH_20:
2181         case NL80211_CHAN_WIDTH_40:
2182                 if (nla_put_u32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
2183                                 cfg80211_get_chandef_type(chandef)))
2184                         return -ENOBUFS;
2185                 break;
2186         default:
2187                 break;
2188         }
2189         if (nla_put_u32(msg, NL80211_ATTR_CHANNEL_WIDTH, chandef->width))
2190                 return -ENOBUFS;
2191         if (nla_put_u32(msg, NL80211_ATTR_CENTER_FREQ1, chandef->center_freq1))
2192                 return -ENOBUFS;
2193         if (chandef->center_freq2 &&
2194             nla_put_u32(msg, NL80211_ATTR_CENTER_FREQ2, chandef->center_freq2))
2195                 return -ENOBUFS;
2196         return 0;
2197 }
2198
2199 static int nl80211_send_iface(struct sk_buff *msg, u32 portid, u32 seq, int flags,
2200                               struct cfg80211_registered_device *rdev,
2201                               struct wireless_dev *wdev)
2202 {
2203         struct net_device *dev = wdev->netdev;
2204         void *hdr;
2205
2206         hdr = nl80211hdr_put(msg, portid, seq, flags, NL80211_CMD_NEW_INTERFACE);
2207         if (!hdr)
2208                 return -1;
2209
2210         if (dev &&
2211             (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
2212              nla_put_string(msg, NL80211_ATTR_IFNAME, dev->name)))
2213                 goto nla_put_failure;
2214
2215         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
2216             nla_put_u32(msg, NL80211_ATTR_IFTYPE, wdev->iftype) ||
2217             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)) ||
2218             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, wdev_address(wdev)) ||
2219             nla_put_u32(msg, NL80211_ATTR_GENERATION,
2220                         rdev->devlist_generation ^
2221                         (cfg80211_rdev_list_generation << 2)))
2222                 goto nla_put_failure;
2223
2224         if (rdev->ops->get_channel) {
2225                 int ret;
2226                 struct cfg80211_chan_def chandef;
2227
2228                 ret = rdev_get_channel(rdev, wdev, &chandef);
2229                 if (ret == 0) {
2230                         if (nl80211_send_chandef(msg, &chandef))
2231                                 goto nla_put_failure;
2232                 }
2233         }
2234
2235         if (wdev->ssid_len) {
2236                 if (nla_put(msg, NL80211_ATTR_SSID, wdev->ssid_len, wdev->ssid))
2237                         goto nla_put_failure;
2238         }
2239
2240         return genlmsg_end(msg, hdr);
2241
2242  nla_put_failure:
2243         genlmsg_cancel(msg, hdr);
2244         return -EMSGSIZE;
2245 }
2246
2247 static int nl80211_dump_interface(struct sk_buff *skb, struct netlink_callback *cb)
2248 {
2249         int wp_idx = 0;
2250         int if_idx = 0;
2251         int wp_start = cb->args[0];
2252         int if_start = cb->args[1];
2253         struct cfg80211_registered_device *rdev;
2254         struct wireless_dev *wdev;
2255
2256         rtnl_lock();
2257         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
2258                 if (!net_eq(wiphy_net(&rdev->wiphy), sock_net(skb->sk)))
2259                         continue;
2260                 if (wp_idx < wp_start) {
2261                         wp_idx++;
2262                         continue;
2263                 }
2264                 if_idx = 0;
2265
2266                 list_for_each_entry(wdev, &rdev->wdev_list, list) {
2267                         if (if_idx < if_start) {
2268                                 if_idx++;
2269                                 continue;
2270                         }
2271                         if (nl80211_send_iface(skb, NETLINK_CB(cb->skb).portid,
2272                                                cb->nlh->nlmsg_seq, NLM_F_MULTI,
2273                                                rdev, wdev) < 0) {
2274                                 goto out;
2275                         }
2276                         if_idx++;
2277                 }
2278
2279                 wp_idx++;
2280         }
2281  out:
2282         rtnl_unlock();
2283
2284         cb->args[0] = wp_idx;
2285         cb->args[1] = if_idx;
2286
2287         return skb->len;
2288 }
2289
2290 static int nl80211_get_interface(struct sk_buff *skb, struct genl_info *info)
2291 {
2292         struct sk_buff *msg;
2293         struct cfg80211_registered_device *dev = info->user_ptr[0];
2294         struct wireless_dev *wdev = info->user_ptr[1];
2295
2296         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2297         if (!msg)
2298                 return -ENOMEM;
2299
2300         if (nl80211_send_iface(msg, info->snd_portid, info->snd_seq, 0,
2301                                dev, wdev) < 0) {
2302                 nlmsg_free(msg);
2303                 return -ENOBUFS;
2304         }
2305
2306         return genlmsg_reply(msg, info);
2307 }
2308
2309 static const struct nla_policy mntr_flags_policy[NL80211_MNTR_FLAG_MAX + 1] = {
2310         [NL80211_MNTR_FLAG_FCSFAIL] = { .type = NLA_FLAG },
2311         [NL80211_MNTR_FLAG_PLCPFAIL] = { .type = NLA_FLAG },
2312         [NL80211_MNTR_FLAG_CONTROL] = { .type = NLA_FLAG },
2313         [NL80211_MNTR_FLAG_OTHER_BSS] = { .type = NLA_FLAG },
2314         [NL80211_MNTR_FLAG_COOK_FRAMES] = { .type = NLA_FLAG },
2315         [NL80211_MNTR_FLAG_ACTIVE] = { .type = NLA_FLAG },
2316 };
2317
2318 static int parse_monitor_flags(struct nlattr *nla, u32 *mntrflags)
2319 {
2320         struct nlattr *flags[NL80211_MNTR_FLAG_MAX + 1];
2321         int flag;
2322
2323         *mntrflags = 0;
2324
2325         if (!nla)
2326                 return -EINVAL;
2327
2328         if (nla_parse_nested(flags, NL80211_MNTR_FLAG_MAX,
2329                              nla, mntr_flags_policy))
2330                 return -EINVAL;
2331
2332         for (flag = 1; flag <= NL80211_MNTR_FLAG_MAX; flag++)
2333                 if (flags[flag])
2334                         *mntrflags |= (1<<flag);
2335
2336         return 0;
2337 }
2338
2339 static int nl80211_valid_4addr(struct cfg80211_registered_device *rdev,
2340                                struct net_device *netdev, u8 use_4addr,
2341                                enum nl80211_iftype iftype)
2342 {
2343         if (!use_4addr) {
2344                 if (netdev && (netdev->priv_flags & IFF_BRIDGE_PORT))
2345                         return -EBUSY;
2346                 return 0;
2347         }
2348
2349         switch (iftype) {
2350         case NL80211_IFTYPE_AP_VLAN:
2351                 if (rdev->wiphy.flags & WIPHY_FLAG_4ADDR_AP)
2352                         return 0;
2353                 break;
2354         case NL80211_IFTYPE_STATION:
2355                 if (rdev->wiphy.flags & WIPHY_FLAG_4ADDR_STATION)
2356                         return 0;
2357                 break;
2358         default:
2359                 break;
2360         }
2361
2362         return -EOPNOTSUPP;
2363 }
2364
2365 static int nl80211_set_interface(struct sk_buff *skb, struct genl_info *info)
2366 {
2367         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2368         struct vif_params params;
2369         int err;
2370         enum nl80211_iftype otype, ntype;
2371         struct net_device *dev = info->user_ptr[1];
2372         u32 _flags, *flags = NULL;
2373         bool change = false;
2374
2375         memset(&params, 0, sizeof(params));
2376
2377         otype = ntype = dev->ieee80211_ptr->iftype;
2378
2379         if (info->attrs[NL80211_ATTR_IFTYPE]) {
2380                 ntype = nla_get_u32(info->attrs[NL80211_ATTR_IFTYPE]);
2381                 if (otype != ntype)
2382                         change = true;
2383                 if (ntype > NL80211_IFTYPE_MAX)
2384                         return -EINVAL;
2385         }
2386
2387         if (info->attrs[NL80211_ATTR_MESH_ID]) {
2388                 struct wireless_dev *wdev = dev->ieee80211_ptr;
2389
2390                 if (ntype != NL80211_IFTYPE_MESH_POINT)
2391                         return -EINVAL;
2392                 if (netif_running(dev))
2393                         return -EBUSY;
2394
2395                 wdev_lock(wdev);
2396                 BUILD_BUG_ON(IEEE80211_MAX_SSID_LEN !=
2397                              IEEE80211_MAX_MESH_ID_LEN);
2398                 wdev->mesh_id_up_len =
2399                         nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
2400                 memcpy(wdev->ssid, nla_data(info->attrs[NL80211_ATTR_MESH_ID]),
2401                        wdev->mesh_id_up_len);
2402                 wdev_unlock(wdev);
2403         }
2404
2405         if (info->attrs[NL80211_ATTR_4ADDR]) {
2406                 params.use_4addr = !!nla_get_u8(info->attrs[NL80211_ATTR_4ADDR]);
2407                 change = true;
2408                 err = nl80211_valid_4addr(rdev, dev, params.use_4addr, ntype);
2409                 if (err)
2410                         return err;
2411         } else {
2412                 params.use_4addr = -1;
2413         }
2414
2415         if (info->attrs[NL80211_ATTR_MNTR_FLAGS]) {
2416                 if (ntype != NL80211_IFTYPE_MONITOR)
2417                         return -EINVAL;
2418                 err = parse_monitor_flags(info->attrs[NL80211_ATTR_MNTR_FLAGS],
2419                                           &_flags);
2420                 if (err)
2421                         return err;
2422
2423                 flags = &_flags;
2424                 change = true;
2425         }
2426
2427         if (flags && (*flags & MONITOR_FLAG_ACTIVE) &&
2428             !(rdev->wiphy.features & NL80211_FEATURE_ACTIVE_MONITOR))
2429                 return -EOPNOTSUPP;
2430
2431         if (change)
2432                 err = cfg80211_change_iface(rdev, dev, ntype, flags, &params);
2433         else
2434                 err = 0;
2435
2436         if (!err && params.use_4addr != -1)
2437                 dev->ieee80211_ptr->use_4addr = params.use_4addr;
2438
2439         return err;
2440 }
2441
2442 static int nl80211_new_interface(struct sk_buff *skb, struct genl_info *info)
2443 {
2444         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2445         struct vif_params params;
2446         struct wireless_dev *wdev;
2447         struct sk_buff *msg;
2448         int err;
2449         enum nl80211_iftype type = NL80211_IFTYPE_UNSPECIFIED;
2450         u32 flags;
2451
2452         memset(&params, 0, sizeof(params));
2453
2454         if (!info->attrs[NL80211_ATTR_IFNAME])
2455                 return -EINVAL;
2456
2457         if (info->attrs[NL80211_ATTR_IFTYPE]) {
2458                 type = nla_get_u32(info->attrs[NL80211_ATTR_IFTYPE]);
2459                 if (type > NL80211_IFTYPE_MAX)
2460                         return -EINVAL;
2461         }
2462
2463         if (!rdev->ops->add_virtual_intf ||
2464             !(rdev->wiphy.interface_modes & (1 << type)))
2465                 return -EOPNOTSUPP;
2466
2467         if (type == NL80211_IFTYPE_P2P_DEVICE && info->attrs[NL80211_ATTR_MAC]) {
2468                 nla_memcpy(params.macaddr, info->attrs[NL80211_ATTR_MAC],
2469                            ETH_ALEN);
2470                 if (!is_valid_ether_addr(params.macaddr))
2471                         return -EADDRNOTAVAIL;
2472         }
2473
2474         if (info->attrs[NL80211_ATTR_4ADDR]) {
2475                 params.use_4addr = !!nla_get_u8(info->attrs[NL80211_ATTR_4ADDR]);
2476                 err = nl80211_valid_4addr(rdev, NULL, params.use_4addr, type);
2477                 if (err)
2478                         return err;
2479         }
2480
2481         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2482         if (!msg)
2483                 return -ENOMEM;
2484
2485         err = parse_monitor_flags(type == NL80211_IFTYPE_MONITOR ?
2486                                   info->attrs[NL80211_ATTR_MNTR_FLAGS] : NULL,
2487                                   &flags);
2488
2489         if (!err && (flags & MONITOR_FLAG_ACTIVE) &&
2490             !(rdev->wiphy.features & NL80211_FEATURE_ACTIVE_MONITOR))
2491                 return -EOPNOTSUPP;
2492
2493         wdev = rdev_add_virtual_intf(rdev,
2494                                 nla_data(info->attrs[NL80211_ATTR_IFNAME]),
2495                                 type, err ? NULL : &flags, &params);
2496         if (IS_ERR(wdev)) {
2497                 nlmsg_free(msg);
2498                 return PTR_ERR(wdev);
2499         }
2500
2501         switch (type) {
2502         case NL80211_IFTYPE_MESH_POINT:
2503                 if (!info->attrs[NL80211_ATTR_MESH_ID])
2504                         break;
2505                 wdev_lock(wdev);
2506                 BUILD_BUG_ON(IEEE80211_MAX_SSID_LEN !=
2507                              IEEE80211_MAX_MESH_ID_LEN);
2508                 wdev->mesh_id_up_len =
2509                         nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
2510                 memcpy(wdev->ssid, nla_data(info->attrs[NL80211_ATTR_MESH_ID]),
2511                        wdev->mesh_id_up_len);
2512                 wdev_unlock(wdev);
2513                 break;
2514         case NL80211_IFTYPE_P2P_DEVICE:
2515                 /*
2516                  * P2P Device doesn't have a netdev, so doesn't go
2517                  * through the netdev notifier and must be added here
2518                  */
2519                 mutex_init(&wdev->mtx);
2520                 INIT_LIST_HEAD(&wdev->event_list);
2521                 spin_lock_init(&wdev->event_lock);
2522                 INIT_LIST_HEAD(&wdev->mgmt_registrations);
2523                 spin_lock_init(&wdev->mgmt_registrations_lock);
2524
2525                 wdev->identifier = ++rdev->wdev_id;
2526                 list_add_rcu(&wdev->list, &rdev->wdev_list);
2527                 rdev->devlist_generation++;
2528                 break;
2529         default:
2530                 break;
2531         }
2532
2533         if (nl80211_send_iface(msg, info->snd_portid, info->snd_seq, 0,
2534                                rdev, wdev) < 0) {
2535                 nlmsg_free(msg);
2536                 return -ENOBUFS;
2537         }
2538
2539         return genlmsg_reply(msg, info);
2540 }
2541
2542 static int nl80211_del_interface(struct sk_buff *skb, struct genl_info *info)
2543 {
2544         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2545         struct wireless_dev *wdev = info->user_ptr[1];
2546
2547         if (!rdev->ops->del_virtual_intf)
2548                 return -EOPNOTSUPP;
2549
2550         /*
2551          * If we remove a wireless device without a netdev then clear
2552          * user_ptr[1] so that nl80211_post_doit won't dereference it
2553          * to check if it needs to do dev_put(). Otherwise it crashes
2554          * since the wdev has been freed, unlike with a netdev where
2555          * we need the dev_put() for the netdev to really be freed.
2556          */
2557         if (!wdev->netdev)
2558                 info->user_ptr[1] = NULL;
2559
2560         return rdev_del_virtual_intf(rdev, wdev);
2561 }
2562
2563 static int nl80211_set_noack_map(struct sk_buff *skb, struct genl_info *info)
2564 {
2565         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2566         struct net_device *dev = info->user_ptr[1];
2567         u16 noack_map;
2568
2569         if (!info->attrs[NL80211_ATTR_NOACK_MAP])
2570                 return -EINVAL;
2571
2572         if (!rdev->ops->set_noack_map)
2573                 return -EOPNOTSUPP;
2574
2575         noack_map = nla_get_u16(info->attrs[NL80211_ATTR_NOACK_MAP]);
2576
2577         return rdev_set_noack_map(rdev, dev, noack_map);
2578 }
2579
2580 struct get_key_cookie {
2581         struct sk_buff *msg;
2582         int error;
2583         int idx;
2584 };
2585
2586 static void get_key_callback(void *c, struct key_params *params)
2587 {
2588         struct nlattr *key;
2589         struct get_key_cookie *cookie = c;
2590
2591         if ((params->key &&
2592              nla_put(cookie->msg, NL80211_ATTR_KEY_DATA,
2593                      params->key_len, params->key)) ||
2594             (params->seq &&
2595              nla_put(cookie->msg, NL80211_ATTR_KEY_SEQ,
2596                      params->seq_len, params->seq)) ||
2597             (params->cipher &&
2598              nla_put_u32(cookie->msg, NL80211_ATTR_KEY_CIPHER,
2599                          params->cipher)))
2600                 goto nla_put_failure;
2601
2602         key = nla_nest_start(cookie->msg, NL80211_ATTR_KEY);
2603         if (!key)
2604                 goto nla_put_failure;
2605
2606         if ((params->key &&
2607              nla_put(cookie->msg, NL80211_KEY_DATA,
2608                      params->key_len, params->key)) ||
2609             (params->seq &&
2610              nla_put(cookie->msg, NL80211_KEY_SEQ,
2611                      params->seq_len, params->seq)) ||
2612             (params->cipher &&
2613              nla_put_u32(cookie->msg, NL80211_KEY_CIPHER,
2614                          params->cipher)))
2615                 goto nla_put_failure;
2616
2617         if (nla_put_u8(cookie->msg, NL80211_ATTR_KEY_IDX, cookie->idx))
2618                 goto nla_put_failure;
2619
2620         nla_nest_end(cookie->msg, key);
2621
2622         return;
2623  nla_put_failure:
2624         cookie->error = 1;
2625 }
2626
2627 static int nl80211_get_key(struct sk_buff *skb, struct genl_info *info)
2628 {
2629         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2630         int err;
2631         struct net_device *dev = info->user_ptr[1];
2632         u8 key_idx = 0;
2633         const u8 *mac_addr = NULL;
2634         bool pairwise;
2635         struct get_key_cookie cookie = {
2636                 .error = 0,
2637         };
2638         void *hdr;
2639         struct sk_buff *msg;
2640
2641         if (info->attrs[NL80211_ATTR_KEY_IDX])
2642                 key_idx = nla_get_u8(info->attrs[NL80211_ATTR_KEY_IDX]);
2643
2644         if (key_idx > 5)
2645                 return -EINVAL;
2646
2647         if (info->attrs[NL80211_ATTR_MAC])
2648                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
2649
2650         pairwise = !!mac_addr;
2651         if (info->attrs[NL80211_ATTR_KEY_TYPE]) {
2652                 u32 kt = nla_get_u32(info->attrs[NL80211_ATTR_KEY_TYPE]);
2653                 if (kt >= NUM_NL80211_KEYTYPES)
2654                         return -EINVAL;
2655                 if (kt != NL80211_KEYTYPE_GROUP &&
2656                     kt != NL80211_KEYTYPE_PAIRWISE)
2657                         return -EINVAL;
2658                 pairwise = kt == NL80211_KEYTYPE_PAIRWISE;
2659         }
2660
2661         if (!rdev->ops->get_key)
2662                 return -EOPNOTSUPP;
2663
2664         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2665         if (!msg)
2666                 return -ENOMEM;
2667
2668         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
2669                              NL80211_CMD_NEW_KEY);
2670         if (!hdr)
2671                 return -ENOBUFS;
2672
2673         cookie.msg = msg;
2674         cookie.idx = key_idx;
2675
2676         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
2677             nla_put_u8(msg, NL80211_ATTR_KEY_IDX, key_idx))
2678                 goto nla_put_failure;
2679         if (mac_addr &&
2680             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr))
2681                 goto nla_put_failure;
2682
2683         if (pairwise && mac_addr &&
2684             !(rdev->wiphy.flags & WIPHY_FLAG_IBSS_RSN))
2685                 return -ENOENT;
2686
2687         err = rdev_get_key(rdev, dev, key_idx, pairwise, mac_addr, &cookie,
2688                            get_key_callback);
2689
2690         if (err)
2691                 goto free_msg;
2692
2693         if (cookie.error)
2694                 goto nla_put_failure;
2695
2696         genlmsg_end(msg, hdr);
2697         return genlmsg_reply(msg, info);
2698
2699  nla_put_failure:
2700         err = -ENOBUFS;
2701  free_msg:
2702         nlmsg_free(msg);
2703         return err;
2704 }
2705
2706 static int nl80211_set_key(struct sk_buff *skb, struct genl_info *info)
2707 {
2708         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2709         struct key_parse key;
2710         int err;
2711         struct net_device *dev = info->user_ptr[1];
2712
2713         err = nl80211_parse_key(info, &key);
2714         if (err)
2715                 return err;
2716
2717         if (key.idx < 0)
2718                 return -EINVAL;
2719
2720         /* only support setting default key */
2721         if (!key.def && !key.defmgmt)
2722                 return -EINVAL;
2723
2724         wdev_lock(dev->ieee80211_ptr);
2725
2726         if (key.def) {
2727                 if (!rdev->ops->set_default_key) {
2728                         err = -EOPNOTSUPP;
2729                         goto out;
2730                 }
2731
2732                 err = nl80211_key_allowed(dev->ieee80211_ptr);
2733                 if (err)
2734                         goto out;
2735
2736                 err = rdev_set_default_key(rdev, dev, key.idx,
2737                                                  key.def_uni, key.def_multi);
2738
2739                 if (err)
2740                         goto out;
2741
2742 #ifdef CONFIG_CFG80211_WEXT
2743                 dev->ieee80211_ptr->wext.default_key = key.idx;
2744 #endif
2745         } else {
2746                 if (key.def_uni || !key.def_multi) {
2747                         err = -EINVAL;
2748                         goto out;
2749                 }
2750
2751                 if (!rdev->ops->set_default_mgmt_key) {
2752                         err = -EOPNOTSUPP;
2753                         goto out;
2754                 }
2755
2756                 err = nl80211_key_allowed(dev->ieee80211_ptr);
2757                 if (err)
2758                         goto out;
2759
2760                 err = rdev_set_default_mgmt_key(rdev, dev, key.idx);
2761                 if (err)
2762                         goto out;
2763
2764 #ifdef CONFIG_CFG80211_WEXT
2765                 dev->ieee80211_ptr->wext.default_mgmt_key = key.idx;
2766 #endif
2767         }
2768
2769  out:
2770         wdev_unlock(dev->ieee80211_ptr);
2771
2772         return err;
2773 }
2774
2775 static int nl80211_new_key(struct sk_buff *skb, struct genl_info *info)
2776 {
2777         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2778         int err;
2779         struct net_device *dev = info->user_ptr[1];
2780         struct key_parse key;
2781         const u8 *mac_addr = NULL;
2782
2783         err = nl80211_parse_key(info, &key);
2784         if (err)
2785                 return err;
2786
2787         if (!key.p.key)
2788                 return -EINVAL;
2789
2790         if (info->attrs[NL80211_ATTR_MAC])
2791                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
2792
2793         if (key.type == -1) {
2794                 if (mac_addr)
2795                         key.type = NL80211_KEYTYPE_PAIRWISE;
2796                 else
2797                         key.type = NL80211_KEYTYPE_GROUP;
2798         }
2799
2800         /* for now */
2801         if (key.type != NL80211_KEYTYPE_PAIRWISE &&
2802             key.type != NL80211_KEYTYPE_GROUP)
2803                 return -EINVAL;
2804
2805         if (!rdev->ops->add_key)
2806                 return -EOPNOTSUPP;
2807
2808         if (cfg80211_validate_key_settings(rdev, &key.p, key.idx,
2809                                            key.type == NL80211_KEYTYPE_PAIRWISE,
2810                                            mac_addr))
2811                 return -EINVAL;
2812
2813         wdev_lock(dev->ieee80211_ptr);
2814         err = nl80211_key_allowed(dev->ieee80211_ptr);
2815         if (!err)
2816                 err = rdev_add_key(rdev, dev, key.idx,
2817                                    key.type == NL80211_KEYTYPE_PAIRWISE,
2818                                     mac_addr, &key.p);
2819         wdev_unlock(dev->ieee80211_ptr);
2820
2821         return err;
2822 }
2823
2824 static int nl80211_del_key(struct sk_buff *skb, struct genl_info *info)
2825 {
2826         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2827         int err;
2828         struct net_device *dev = info->user_ptr[1];
2829         u8 *mac_addr = NULL;
2830         struct key_parse key;
2831
2832         err = nl80211_parse_key(info, &key);
2833         if (err)
2834                 return err;
2835
2836         if (info->attrs[NL80211_ATTR_MAC])
2837                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
2838
2839         if (key.type == -1) {
2840                 if (mac_addr)
2841                         key.type = NL80211_KEYTYPE_PAIRWISE;
2842                 else
2843                         key.type = NL80211_KEYTYPE_GROUP;
2844         }
2845
2846         /* for now */
2847         if (key.type != NL80211_KEYTYPE_PAIRWISE &&
2848             key.type != NL80211_KEYTYPE_GROUP)
2849                 return -EINVAL;
2850
2851         if (!rdev->ops->del_key)
2852                 return -EOPNOTSUPP;
2853
2854         wdev_lock(dev->ieee80211_ptr);
2855         err = nl80211_key_allowed(dev->ieee80211_ptr);
2856
2857         if (key.type == NL80211_KEYTYPE_PAIRWISE && mac_addr &&
2858             !(rdev->wiphy.flags & WIPHY_FLAG_IBSS_RSN))
2859                 err = -ENOENT;
2860
2861         if (!err)
2862                 err = rdev_del_key(rdev, dev, key.idx,
2863                                    key.type == NL80211_KEYTYPE_PAIRWISE,
2864                                    mac_addr);
2865
2866 #ifdef CONFIG_CFG80211_WEXT
2867         if (!err) {
2868                 if (key.idx == dev->ieee80211_ptr->wext.default_key)
2869                         dev->ieee80211_ptr->wext.default_key = -1;
2870                 else if (key.idx == dev->ieee80211_ptr->wext.default_mgmt_key)
2871                         dev->ieee80211_ptr->wext.default_mgmt_key = -1;
2872         }
2873 #endif
2874         wdev_unlock(dev->ieee80211_ptr);
2875
2876         return err;
2877 }
2878
2879 /* This function returns an error or the number of nested attributes */
2880 static int validate_acl_mac_addrs(struct nlattr *nl_attr)
2881 {
2882         struct nlattr *attr;
2883         int n_entries = 0, tmp;
2884
2885         nla_for_each_nested(attr, nl_attr, tmp) {
2886                 if (nla_len(attr) != ETH_ALEN)
2887                         return -EINVAL;
2888
2889                 n_entries++;
2890         }
2891
2892         return n_entries;
2893 }
2894
2895 /*
2896  * This function parses ACL information and allocates memory for ACL data.
2897  * On successful return, the calling function is responsible to free the
2898  * ACL buffer returned by this function.
2899  */
2900 static struct cfg80211_acl_data *parse_acl_data(struct wiphy *wiphy,
2901                                                 struct genl_info *info)
2902 {
2903         enum nl80211_acl_policy acl_policy;
2904         struct nlattr *attr;
2905         struct cfg80211_acl_data *acl;
2906         int i = 0, n_entries, tmp;
2907
2908         if (!wiphy->max_acl_mac_addrs)
2909                 return ERR_PTR(-EOPNOTSUPP);
2910
2911         if (!info->attrs[NL80211_ATTR_ACL_POLICY])
2912                 return ERR_PTR(-EINVAL);
2913
2914         acl_policy = nla_get_u32(info->attrs[NL80211_ATTR_ACL_POLICY]);
2915         if (acl_policy != NL80211_ACL_POLICY_ACCEPT_UNLESS_LISTED &&
2916             acl_policy != NL80211_ACL_POLICY_DENY_UNLESS_LISTED)
2917                 return ERR_PTR(-EINVAL);
2918
2919         if (!info->attrs[NL80211_ATTR_MAC_ADDRS])
2920                 return ERR_PTR(-EINVAL);
2921
2922         n_entries = validate_acl_mac_addrs(info->attrs[NL80211_ATTR_MAC_ADDRS]);
2923         if (n_entries < 0)
2924                 return ERR_PTR(n_entries);
2925
2926         if (n_entries > wiphy->max_acl_mac_addrs)
2927                 return ERR_PTR(-ENOTSUPP);
2928
2929         acl = kzalloc(sizeof(*acl) + (sizeof(struct mac_address) * n_entries),
2930                       GFP_KERNEL);
2931         if (!acl)
2932                 return ERR_PTR(-ENOMEM);
2933
2934         nla_for_each_nested(attr, info->attrs[NL80211_ATTR_MAC_ADDRS], tmp) {
2935                 memcpy(acl->mac_addrs[i].addr, nla_data(attr), ETH_ALEN);
2936                 i++;
2937         }
2938
2939         acl->n_acl_entries = n_entries;
2940         acl->acl_policy = acl_policy;
2941
2942         return acl;
2943 }
2944
2945 static int nl80211_set_mac_acl(struct sk_buff *skb, struct genl_info *info)
2946 {
2947         struct cfg80211_registered_device *rdev = info->user_ptr[0];
2948         struct net_device *dev = info->user_ptr[1];
2949         struct cfg80211_acl_data *acl;
2950         int err;
2951
2952         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
2953             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
2954                 return -EOPNOTSUPP;
2955
2956         if (!dev->ieee80211_ptr->beacon_interval)
2957                 return -EINVAL;
2958
2959         acl = parse_acl_data(&rdev->wiphy, info);
2960         if (IS_ERR(acl))
2961                 return PTR_ERR(acl);
2962
2963         err = rdev_set_mac_acl(rdev, dev, acl);
2964
2965         kfree(acl);
2966
2967         return err;
2968 }
2969
2970 static int nl80211_parse_beacon(struct nlattr *attrs[],
2971                                 struct cfg80211_beacon_data *bcn)
2972 {
2973         bool haveinfo = false;
2974
2975         if (!is_valid_ie_attr(attrs[NL80211_ATTR_BEACON_TAIL]) ||
2976             !is_valid_ie_attr(attrs[NL80211_ATTR_IE]) ||
2977             !is_valid_ie_attr(attrs[NL80211_ATTR_IE_PROBE_RESP]) ||
2978             !is_valid_ie_attr(attrs[NL80211_ATTR_IE_ASSOC_RESP]))
2979                 return -EINVAL;
2980
2981         memset(bcn, 0, sizeof(*bcn));
2982
2983         if (attrs[NL80211_ATTR_BEACON_HEAD]) {
2984                 bcn->head = nla_data(attrs[NL80211_ATTR_BEACON_HEAD]);
2985                 bcn->head_len = nla_len(attrs[NL80211_ATTR_BEACON_HEAD]);
2986                 if (!bcn->head_len)
2987                         return -EINVAL;
2988                 haveinfo = true;
2989         }
2990
2991         if (attrs[NL80211_ATTR_BEACON_TAIL]) {
2992                 bcn->tail = nla_data(attrs[NL80211_ATTR_BEACON_TAIL]);
2993                 bcn->tail_len = nla_len(attrs[NL80211_ATTR_BEACON_TAIL]);
2994                 haveinfo = true;
2995         }
2996
2997         if (!haveinfo)
2998                 return -EINVAL;
2999
3000         if (attrs[NL80211_ATTR_IE]) {
3001                 bcn->beacon_ies = nla_data(attrs[NL80211_ATTR_IE]);
3002                 bcn->beacon_ies_len = nla_len(attrs[NL80211_ATTR_IE]);
3003         }
3004
3005         if (attrs[NL80211_ATTR_IE_PROBE_RESP]) {
3006                 bcn->proberesp_ies =
3007                         nla_data(attrs[NL80211_ATTR_IE_PROBE_RESP]);
3008                 bcn->proberesp_ies_len =
3009                         nla_len(attrs[NL80211_ATTR_IE_PROBE_RESP]);
3010         }
3011
3012         if (attrs[NL80211_ATTR_IE_ASSOC_RESP]) {
3013                 bcn->assocresp_ies =
3014                         nla_data(attrs[NL80211_ATTR_IE_ASSOC_RESP]);
3015                 bcn->assocresp_ies_len =
3016                         nla_len(attrs[NL80211_ATTR_IE_ASSOC_RESP]);
3017         }
3018
3019         if (attrs[NL80211_ATTR_PROBE_RESP]) {
3020                 bcn->probe_resp = nla_data(attrs[NL80211_ATTR_PROBE_RESP]);
3021                 bcn->probe_resp_len = nla_len(attrs[NL80211_ATTR_PROBE_RESP]);
3022         }
3023
3024         return 0;
3025 }
3026
3027 static bool nl80211_get_ap_channel(struct cfg80211_registered_device *rdev,
3028                                    struct cfg80211_ap_settings *params)
3029 {
3030         struct wireless_dev *wdev;
3031         bool ret = false;
3032
3033         list_for_each_entry(wdev, &rdev->wdev_list, list) {
3034                 if (wdev->iftype != NL80211_IFTYPE_AP &&
3035                     wdev->iftype != NL80211_IFTYPE_P2P_GO)
3036                         continue;
3037
3038                 if (!wdev->preset_chandef.chan)
3039                         continue;
3040
3041                 params->chandef = wdev->preset_chandef;
3042                 ret = true;
3043                 break;
3044         }
3045
3046         return ret;
3047 }
3048
3049 static bool nl80211_valid_auth_type(struct cfg80211_registered_device *rdev,
3050                                     enum nl80211_auth_type auth_type,
3051                                     enum nl80211_commands cmd)
3052 {
3053         if (auth_type > NL80211_AUTHTYPE_MAX)
3054                 return false;
3055
3056         switch (cmd) {
3057         case NL80211_CMD_AUTHENTICATE:
3058                 if (!(rdev->wiphy.features & NL80211_FEATURE_SAE) &&
3059                     auth_type == NL80211_AUTHTYPE_SAE)
3060                         return false;
3061                 return true;
3062         case NL80211_CMD_CONNECT:
3063         case NL80211_CMD_START_AP:
3064                 /* SAE not supported yet */
3065                 if (auth_type == NL80211_AUTHTYPE_SAE)
3066                         return false;
3067                 return true;
3068         default:
3069                 return false;
3070         }
3071 }
3072
3073 static int nl80211_start_ap(struct sk_buff *skb, struct genl_info *info)
3074 {
3075         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3076         struct net_device *dev = info->user_ptr[1];
3077         struct wireless_dev *wdev = dev->ieee80211_ptr;
3078         struct cfg80211_ap_settings params;
3079         int err;
3080         u8 radar_detect_width = 0;
3081
3082         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
3083             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3084                 return -EOPNOTSUPP;
3085
3086         if (!rdev->ops->start_ap)
3087                 return -EOPNOTSUPP;
3088
3089         if (wdev->beacon_interval)
3090                 return -EALREADY;
3091
3092         memset(&params, 0, sizeof(params));
3093
3094         /* these are required for START_AP */
3095         if (!info->attrs[NL80211_ATTR_BEACON_INTERVAL] ||
3096             !info->attrs[NL80211_ATTR_DTIM_PERIOD] ||
3097             !info->attrs[NL80211_ATTR_BEACON_HEAD])
3098                 return -EINVAL;
3099
3100         err = nl80211_parse_beacon(info->attrs, &params.beacon);
3101         if (err)
3102                 return err;
3103
3104         params.beacon_interval =
3105                 nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
3106         params.dtim_period =
3107                 nla_get_u32(info->attrs[NL80211_ATTR_DTIM_PERIOD]);
3108
3109         err = cfg80211_validate_beacon_int(rdev, params.beacon_interval);
3110         if (err)
3111                 return err;
3112
3113         /*
3114          * In theory, some of these attributes should be required here
3115          * but since they were not used when the command was originally
3116          * added, keep them optional for old user space programs to let
3117          * them continue to work with drivers that do not need the
3118          * additional information -- drivers must check!
3119          */
3120         if (info->attrs[NL80211_ATTR_SSID]) {
3121                 params.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
3122                 params.ssid_len =
3123                         nla_len(info->attrs[NL80211_ATTR_SSID]);
3124                 if (params.ssid_len == 0 ||
3125                     params.ssid_len > IEEE80211_MAX_SSID_LEN)
3126                         return -EINVAL;
3127         }
3128
3129         if (info->attrs[NL80211_ATTR_HIDDEN_SSID]) {
3130                 params.hidden_ssid = nla_get_u32(
3131                         info->attrs[NL80211_ATTR_HIDDEN_SSID]);
3132                 if (params.hidden_ssid != NL80211_HIDDEN_SSID_NOT_IN_USE &&
3133                     params.hidden_ssid != NL80211_HIDDEN_SSID_ZERO_LEN &&
3134                     params.hidden_ssid != NL80211_HIDDEN_SSID_ZERO_CONTENTS)
3135                         return -EINVAL;
3136         }
3137
3138         params.privacy = !!info->attrs[NL80211_ATTR_PRIVACY];
3139
3140         if (info->attrs[NL80211_ATTR_AUTH_TYPE]) {
3141                 params.auth_type = nla_get_u32(
3142                         info->attrs[NL80211_ATTR_AUTH_TYPE]);
3143                 if (!nl80211_valid_auth_type(rdev, params.auth_type,
3144                                              NL80211_CMD_START_AP))
3145                         return -EINVAL;
3146         } else
3147                 params.auth_type = NL80211_AUTHTYPE_AUTOMATIC;
3148
3149         err = nl80211_crypto_settings(rdev, info, &params.crypto,
3150                                       NL80211_MAX_NR_CIPHER_SUITES);
3151         if (err)
3152                 return err;
3153
3154         if (info->attrs[NL80211_ATTR_INACTIVITY_TIMEOUT]) {
3155                 if (!(rdev->wiphy.features & NL80211_FEATURE_INACTIVITY_TIMER))
3156                         return -EOPNOTSUPP;
3157                 params.inactivity_timeout = nla_get_u16(
3158                         info->attrs[NL80211_ATTR_INACTIVITY_TIMEOUT]);
3159         }
3160
3161         if (info->attrs[NL80211_ATTR_P2P_CTWINDOW]) {
3162                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3163                         return -EINVAL;
3164                 params.p2p_ctwindow =
3165                         nla_get_u8(info->attrs[NL80211_ATTR_P2P_CTWINDOW]);
3166                 if (params.p2p_ctwindow > 127)
3167                         return -EINVAL;
3168                 if (params.p2p_ctwindow != 0 &&
3169                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_CTWIN))
3170                         return -EINVAL;
3171         }
3172
3173         if (info->attrs[NL80211_ATTR_P2P_OPPPS]) {
3174                 u8 tmp;
3175
3176                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3177                         return -EINVAL;
3178                 tmp = nla_get_u8(info->attrs[NL80211_ATTR_P2P_OPPPS]);
3179                 if (tmp > 1)
3180                         return -EINVAL;
3181                 params.p2p_opp_ps = tmp;
3182                 if (params.p2p_opp_ps != 0 &&
3183                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_OPPPS))
3184                         return -EINVAL;
3185         }
3186
3187         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
3188                 err = nl80211_parse_chandef(rdev, info, &params.chandef);
3189                 if (err)
3190                         return err;
3191         } else if (wdev->preset_chandef.chan) {
3192                 params.chandef = wdev->preset_chandef;
3193         } else if (!nl80211_get_ap_channel(rdev, &params))
3194                 return -EINVAL;
3195
3196         if (!cfg80211_reg_can_beacon(&rdev->wiphy, &params.chandef))
3197                 return -EINVAL;
3198
3199         err = cfg80211_chandef_dfs_required(wdev->wiphy, &params.chandef);
3200         if (err < 0)
3201                 return err;
3202         if (err) {
3203                 radar_detect_width = BIT(params.chandef.width);
3204                 params.radar_required = true;
3205         }
3206
3207         err = cfg80211_can_use_iftype_chan(rdev, wdev, wdev->iftype,
3208                                            params.chandef.chan,
3209                                            CHAN_MODE_SHARED,
3210                                            radar_detect_width);
3211         if (err)
3212                 return err;
3213
3214         if (info->attrs[NL80211_ATTR_ACL_POLICY]) {
3215                 params.acl = parse_acl_data(&rdev->wiphy, info);
3216                 if (IS_ERR(params.acl))
3217                         return PTR_ERR(params.acl);
3218         }
3219
3220         err = rdev_start_ap(rdev, dev, &params);
3221         if (!err) {
3222                 wdev->preset_chandef = params.chandef;
3223                 wdev->beacon_interval = params.beacon_interval;
3224                 wdev->channel = params.chandef.chan;
3225                 wdev->ssid_len = params.ssid_len;
3226                 memcpy(wdev->ssid, params.ssid, wdev->ssid_len);
3227         }
3228
3229         kfree(params.acl);
3230
3231         return err;
3232 }
3233
3234 static int nl80211_set_beacon(struct sk_buff *skb, struct genl_info *info)
3235 {
3236         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3237         struct net_device *dev = info->user_ptr[1];
3238         struct wireless_dev *wdev = dev->ieee80211_ptr;
3239         struct cfg80211_beacon_data params;
3240         int err;
3241
3242         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
3243             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
3244                 return -EOPNOTSUPP;
3245
3246         if (!rdev->ops->change_beacon)
3247                 return -EOPNOTSUPP;
3248
3249         if (!wdev->beacon_interval)
3250                 return -EINVAL;
3251
3252         err = nl80211_parse_beacon(info->attrs, &params);
3253         if (err)
3254                 return err;
3255
3256         return rdev_change_beacon(rdev, dev, &params);
3257 }
3258
3259 static int nl80211_stop_ap(struct sk_buff *skb, struct genl_info *info)
3260 {
3261         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3262         struct net_device *dev = info->user_ptr[1];
3263
3264         return cfg80211_stop_ap(rdev, dev);
3265 }
3266
3267 static const struct nla_policy sta_flags_policy[NL80211_STA_FLAG_MAX + 1] = {
3268         [NL80211_STA_FLAG_AUTHORIZED] = { .type = NLA_FLAG },
3269         [NL80211_STA_FLAG_SHORT_PREAMBLE] = { .type = NLA_FLAG },
3270         [NL80211_STA_FLAG_WME] = { .type = NLA_FLAG },
3271         [NL80211_STA_FLAG_MFP] = { .type = NLA_FLAG },
3272         [NL80211_STA_FLAG_AUTHENTICATED] = { .type = NLA_FLAG },
3273         [NL80211_STA_FLAG_TDLS_PEER] = { .type = NLA_FLAG },
3274 };
3275
3276 static int parse_station_flags(struct genl_info *info,
3277                                enum nl80211_iftype iftype,
3278                                struct station_parameters *params)
3279 {
3280         struct nlattr *flags[NL80211_STA_FLAG_MAX + 1];
3281         struct nlattr *nla;
3282         int flag;
3283
3284         /*
3285          * Try parsing the new attribute first so userspace
3286          * can specify both for older kernels.
3287          */
3288         nla = info->attrs[NL80211_ATTR_STA_FLAGS2];
3289         if (nla) {
3290                 struct nl80211_sta_flag_update *sta_flags;
3291
3292                 sta_flags = nla_data(nla);
3293                 params->sta_flags_mask = sta_flags->mask;
3294                 params->sta_flags_set = sta_flags->set;
3295                 params->sta_flags_set &= params->sta_flags_mask;
3296                 if ((params->sta_flags_mask |
3297                      params->sta_flags_set) & BIT(__NL80211_STA_FLAG_INVALID))
3298                         return -EINVAL;
3299                 return 0;
3300         }
3301
3302         /* if present, parse the old attribute */
3303
3304         nla = info->attrs[NL80211_ATTR_STA_FLAGS];
3305         if (!nla)
3306                 return 0;
3307
3308         if (nla_parse_nested(flags, NL80211_STA_FLAG_MAX,
3309                              nla, sta_flags_policy))
3310                 return -EINVAL;
3311
3312         /*
3313          * Only allow certain flags for interface types so that
3314          * other attributes are silently ignored. Remember that
3315          * this is backward compatibility code with old userspace
3316          * and shouldn't be hit in other cases anyway.
3317          */
3318         switch (iftype) {
3319         case NL80211_IFTYPE_AP:
3320         case NL80211_IFTYPE_AP_VLAN:
3321         case NL80211_IFTYPE_P2P_GO:
3322                 params->sta_flags_mask = BIT(NL80211_STA_FLAG_AUTHORIZED) |
3323                                          BIT(NL80211_STA_FLAG_SHORT_PREAMBLE) |
3324                                          BIT(NL80211_STA_FLAG_WME) |
3325                                          BIT(NL80211_STA_FLAG_MFP);
3326                 break;
3327         case NL80211_IFTYPE_P2P_CLIENT:
3328         case NL80211_IFTYPE_STATION:
3329                 params->sta_flags_mask = BIT(NL80211_STA_FLAG_AUTHORIZED) |
3330                                          BIT(NL80211_STA_FLAG_TDLS_PEER);
3331                 break;
3332         case NL80211_IFTYPE_MESH_POINT:
3333                 params->sta_flags_mask = BIT(NL80211_STA_FLAG_AUTHENTICATED) |
3334                                          BIT(NL80211_STA_FLAG_MFP) |
3335                                          BIT(NL80211_STA_FLAG_AUTHORIZED);
3336         default:
3337                 return -EINVAL;
3338         }
3339
3340         for (flag = 1; flag <= NL80211_STA_FLAG_MAX; flag++) {
3341                 if (flags[flag]) {
3342                         params->sta_flags_set |= (1<<flag);
3343
3344                         /* no longer support new API additions in old API */
3345                         if (flag > NL80211_STA_FLAG_MAX_OLD_API)
3346                                 return -EINVAL;
3347                 }
3348         }
3349
3350         return 0;
3351 }
3352
3353 static bool nl80211_put_sta_rate(struct sk_buff *msg, struct rate_info *info,
3354                                  int attr)
3355 {
3356         struct nlattr *rate;
3357         u32 bitrate;
3358         u16 bitrate_compat;
3359
3360         rate = nla_nest_start(msg, attr);
3361         if (!rate)
3362                 return false;
3363
3364         /* cfg80211_calculate_bitrate will return 0 for mcs >= 32 */
3365         bitrate = cfg80211_calculate_bitrate(info);
3366         /* report 16-bit bitrate only if we can */
3367         bitrate_compat = bitrate < (1UL << 16) ? bitrate : 0;
3368         if (bitrate > 0 &&
3369             nla_put_u32(msg, NL80211_RATE_INFO_BITRATE32, bitrate))
3370                 return false;
3371         if (bitrate_compat > 0 &&
3372             nla_put_u16(msg, NL80211_RATE_INFO_BITRATE, bitrate_compat))
3373                 return false;
3374
3375         if (info->flags & RATE_INFO_FLAGS_MCS) {
3376                 if (nla_put_u8(msg, NL80211_RATE_INFO_MCS, info->mcs))
3377                         return false;
3378                 if (info->flags & RATE_INFO_FLAGS_40_MHZ_WIDTH &&
3379                     nla_put_flag(msg, NL80211_RATE_INFO_40_MHZ_WIDTH))
3380                         return false;
3381                 if (info->flags & RATE_INFO_FLAGS_SHORT_GI &&
3382                     nla_put_flag(msg, NL80211_RATE_INFO_SHORT_GI))
3383                         return false;
3384         } else if (info->flags & RATE_INFO_FLAGS_VHT_MCS) {
3385                 if (nla_put_u8(msg, NL80211_RATE_INFO_VHT_MCS, info->mcs))
3386                         return false;
3387                 if (nla_put_u8(msg, NL80211_RATE_INFO_VHT_NSS, info->nss))
3388                         return false;
3389                 if (info->flags & RATE_INFO_FLAGS_40_MHZ_WIDTH &&
3390                     nla_put_flag(msg, NL80211_RATE_INFO_40_MHZ_WIDTH))
3391                         return false;
3392                 if (info->flags & RATE_INFO_FLAGS_80_MHZ_WIDTH &&
3393                     nla_put_flag(msg, NL80211_RATE_INFO_80_MHZ_WIDTH))
3394                         return false;
3395                 if (info->flags & RATE_INFO_FLAGS_80P80_MHZ_WIDTH &&
3396                     nla_put_flag(msg, NL80211_RATE_INFO_80P80_MHZ_WIDTH))
3397                         return false;
3398                 if (info->flags & RATE_INFO_FLAGS_160_MHZ_WIDTH &&
3399                     nla_put_flag(msg, NL80211_RATE_INFO_160_MHZ_WIDTH))
3400                         return false;
3401                 if (info->flags & RATE_INFO_FLAGS_SHORT_GI &&
3402                     nla_put_flag(msg, NL80211_RATE_INFO_SHORT_GI))
3403                         return false;
3404         }
3405
3406         nla_nest_end(msg, rate);
3407         return true;
3408 }
3409
3410 static bool nl80211_put_signal(struct sk_buff *msg, u8 mask, s8 *signal,
3411                                int id)
3412 {
3413         void *attr;
3414         int i = 0;
3415
3416         if (!mask)
3417                 return true;
3418
3419         attr = nla_nest_start(msg, id);
3420         if (!attr)
3421                 return false;
3422
3423         for (i = 0; i < IEEE80211_MAX_CHAINS; i++) {
3424                 if (!(mask & BIT(i)))
3425                         continue;
3426
3427                 if (nla_put_u8(msg, i, signal[i]))
3428                         return false;
3429         }
3430
3431         nla_nest_end(msg, attr);
3432
3433         return true;
3434 }
3435
3436 static int nl80211_send_station(struct sk_buff *msg, u32 portid, u32 seq,
3437                                 int flags,
3438                                 struct cfg80211_registered_device *rdev,
3439                                 struct net_device *dev,
3440                                 const u8 *mac_addr, struct station_info *sinfo)
3441 {
3442         void *hdr;
3443         struct nlattr *sinfoattr, *bss_param;
3444
3445         hdr = nl80211hdr_put(msg, portid, seq, flags, NL80211_CMD_NEW_STATION);
3446         if (!hdr)
3447                 return -1;
3448
3449         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
3450             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr) ||
3451             nla_put_u32(msg, NL80211_ATTR_GENERATION, sinfo->generation))
3452                 goto nla_put_failure;
3453
3454         sinfoattr = nla_nest_start(msg, NL80211_ATTR_STA_INFO);
3455         if (!sinfoattr)
3456                 goto nla_put_failure;
3457         if ((sinfo->filled & STATION_INFO_CONNECTED_TIME) &&
3458             nla_put_u32(msg, NL80211_STA_INFO_CONNECTED_TIME,
3459                         sinfo->connected_time))
3460                 goto nla_put_failure;
3461         if ((sinfo->filled & STATION_INFO_INACTIVE_TIME) &&
3462             nla_put_u32(msg, NL80211_STA_INFO_INACTIVE_TIME,
3463                         sinfo->inactive_time))
3464                 goto nla_put_failure;
3465         if ((sinfo->filled & (STATION_INFO_RX_BYTES |
3466                               STATION_INFO_RX_BYTES64)) &&
3467             nla_put_u32(msg, NL80211_STA_INFO_RX_BYTES,
3468                         (u32)sinfo->rx_bytes))
3469                 goto nla_put_failure;
3470         if ((sinfo->filled & (STATION_INFO_TX_BYTES |
3471                               STATION_INFO_TX_BYTES64)) &&
3472             nla_put_u32(msg, NL80211_STA_INFO_TX_BYTES,
3473                         (u32)sinfo->tx_bytes))
3474                 goto nla_put_failure;
3475         if ((sinfo->filled & STATION_INFO_RX_BYTES64) &&
3476             nla_put_u64(msg, NL80211_STA_INFO_RX_BYTES64,
3477                         sinfo->rx_bytes))
3478                 goto nla_put_failure;
3479         if ((sinfo->filled & STATION_INFO_TX_BYTES64) &&
3480             nla_put_u64(msg, NL80211_STA_INFO_TX_BYTES64,
3481                         sinfo->tx_bytes))
3482                 goto nla_put_failure;
3483         if ((sinfo->filled & STATION_INFO_LLID) &&
3484             nla_put_u16(msg, NL80211_STA_INFO_LLID, sinfo->llid))
3485                 goto nla_put_failure;
3486         if ((sinfo->filled & STATION_INFO_PLID) &&
3487             nla_put_u16(msg, NL80211_STA_INFO_PLID, sinfo->plid))
3488                 goto nla_put_failure;
3489         if ((sinfo->filled & STATION_INFO_PLINK_STATE) &&
3490             nla_put_u8(msg, NL80211_STA_INFO_PLINK_STATE,
3491                        sinfo->plink_state))
3492                 goto nla_put_failure;
3493         switch (rdev->wiphy.signal_type) {
3494         case CFG80211_SIGNAL_TYPE_MBM:
3495                 if ((sinfo->filled & STATION_INFO_SIGNAL) &&
3496                     nla_put_u8(msg, NL80211_STA_INFO_SIGNAL,
3497                                sinfo->signal))
3498                         goto nla_put_failure;
3499                 if ((sinfo->filled & STATION_INFO_SIGNAL_AVG) &&
3500                     nla_put_u8(msg, NL80211_STA_INFO_SIGNAL_AVG,
3501                                sinfo->signal_avg))
3502                         goto nla_put_failure;
3503                 break;
3504         default:
3505                 break;
3506         }
3507         if (sinfo->filled & STATION_INFO_CHAIN_SIGNAL) {
3508                 if (!nl80211_put_signal(msg, sinfo->chains,
3509                                         sinfo->chain_signal,
3510                                         NL80211_STA_INFO_CHAIN_SIGNAL))
3511                         goto nla_put_failure;
3512         }
3513         if (sinfo->filled & STATION_INFO_CHAIN_SIGNAL_AVG) {
3514                 if (!nl80211_put_signal(msg, sinfo->chains,
3515                                         sinfo->chain_signal_avg,
3516                                         NL80211_STA_INFO_CHAIN_SIGNAL_AVG))
3517                         goto nla_put_failure;
3518         }
3519         if (sinfo->filled & STATION_INFO_TX_BITRATE) {
3520                 if (!nl80211_put_sta_rate(msg, &sinfo->txrate,
3521                                           NL80211_STA_INFO_TX_BITRATE))
3522                         goto nla_put_failure;
3523         }
3524         if (sinfo->filled & STATION_INFO_RX_BITRATE) {
3525                 if (!nl80211_put_sta_rate(msg, &sinfo->rxrate,
3526                                           NL80211_STA_INFO_RX_BITRATE))
3527                         goto nla_put_failure;
3528         }
3529         if ((sinfo->filled & STATION_INFO_RX_PACKETS) &&
3530             nla_put_u32(msg, NL80211_STA_INFO_RX_PACKETS,
3531                         sinfo->rx_packets))
3532                 goto nla_put_failure;
3533         if ((sinfo->filled & STATION_INFO_TX_PACKETS) &&
3534             nla_put_u32(msg, NL80211_STA_INFO_TX_PACKETS,
3535                         sinfo->tx_packets))
3536                 goto nla_put_failure;
3537         if ((sinfo->filled & STATION_INFO_TX_RETRIES) &&
3538             nla_put_u32(msg, NL80211_STA_INFO_TX_RETRIES,
3539                         sinfo->tx_retries))
3540                 goto nla_put_failure;
3541         if ((sinfo->filled & STATION_INFO_TX_FAILED) &&
3542             nla_put_u32(msg, NL80211_STA_INFO_TX_FAILED,
3543                         sinfo->tx_failed))
3544                 goto nla_put_failure;
3545         if ((sinfo->filled & STATION_INFO_BEACON_LOSS_COUNT) &&
3546             nla_put_u32(msg, NL80211_STA_INFO_BEACON_LOSS,
3547                         sinfo->beacon_loss_count))
3548                 goto nla_put_failure;
3549         if ((sinfo->filled & STATION_INFO_LOCAL_PM) &&
3550             nla_put_u32(msg, NL80211_STA_INFO_LOCAL_PM,
3551                         sinfo->local_pm))
3552                 goto nla_put_failure;
3553         if ((sinfo->filled & STATION_INFO_PEER_PM) &&
3554             nla_put_u32(msg, NL80211_STA_INFO_PEER_PM,
3555                         sinfo->peer_pm))
3556                 goto nla_put_failure;
3557         if ((sinfo->filled & STATION_INFO_NONPEER_PM) &&
3558             nla_put_u32(msg, NL80211_STA_INFO_NONPEER_PM,
3559                         sinfo->nonpeer_pm))
3560                 goto nla_put_failure;
3561         if (sinfo->filled & STATION_INFO_BSS_PARAM) {
3562                 bss_param = nla_nest_start(msg, NL80211_STA_INFO_BSS_PARAM);
3563                 if (!bss_param)
3564                         goto nla_put_failure;
3565
3566                 if (((sinfo->bss_param.flags & BSS_PARAM_FLAGS_CTS_PROT) &&
3567                      nla_put_flag(msg, NL80211_STA_BSS_PARAM_CTS_PROT)) ||
3568                     ((sinfo->bss_param.flags & BSS_PARAM_FLAGS_SHORT_PREAMBLE) &&
3569                      nla_put_flag(msg, NL80211_STA_BSS_PARAM_SHORT_PREAMBLE)) ||
3570                     ((sinfo->bss_param.flags & BSS_PARAM_FLAGS_SHORT_SLOT_TIME) &&
3571                      nla_put_flag(msg, NL80211_STA_BSS_PARAM_SHORT_SLOT_TIME)) ||
3572                     nla_put_u8(msg, NL80211_STA_BSS_PARAM_DTIM_PERIOD,
3573                                sinfo->bss_param.dtim_period) ||
3574                     nla_put_u16(msg, NL80211_STA_BSS_PARAM_BEACON_INTERVAL,
3575                                 sinfo->bss_param.beacon_interval))
3576                         goto nla_put_failure;
3577
3578                 nla_nest_end(msg, bss_param);
3579         }
3580         if ((sinfo->filled & STATION_INFO_STA_FLAGS) &&
3581             nla_put(msg, NL80211_STA_INFO_STA_FLAGS,
3582                     sizeof(struct nl80211_sta_flag_update),
3583                     &sinfo->sta_flags))
3584                 goto nla_put_failure;
3585         if ((sinfo->filled & STATION_INFO_T_OFFSET) &&
3586                 nla_put_u64(msg, NL80211_STA_INFO_T_OFFSET,
3587                             sinfo->t_offset))
3588                 goto nla_put_failure;
3589         nla_nest_end(msg, sinfoattr);
3590
3591         if ((sinfo->filled & STATION_INFO_ASSOC_REQ_IES) &&
3592             nla_put(msg, NL80211_ATTR_IE, sinfo->assoc_req_ies_len,
3593                     sinfo->assoc_req_ies))
3594                 goto nla_put_failure;
3595
3596         return genlmsg_end(msg, hdr);
3597
3598  nla_put_failure:
3599         genlmsg_cancel(msg, hdr);
3600         return -EMSGSIZE;
3601 }
3602
3603 static int nl80211_dump_station(struct sk_buff *skb,
3604                                 struct netlink_callback *cb)
3605 {
3606         struct station_info sinfo;
3607         struct cfg80211_registered_device *dev;
3608         struct wireless_dev *wdev;
3609         u8 mac_addr[ETH_ALEN];
3610         int sta_idx = cb->args[2];
3611         int err;
3612
3613         err = nl80211_prepare_wdev_dump(skb, cb, &dev, &wdev);
3614         if (err)
3615                 return err;
3616
3617         if (!wdev->netdev) {
3618                 err = -EINVAL;
3619                 goto out_err;
3620         }
3621
3622         if (!dev->ops->dump_station) {
3623                 err = -EOPNOTSUPP;
3624                 goto out_err;
3625         }
3626
3627         while (1) {
3628                 memset(&sinfo, 0, sizeof(sinfo));
3629                 err = rdev_dump_station(dev, wdev->netdev, sta_idx,
3630                                         mac_addr, &sinfo);
3631                 if (err == -ENOENT)
3632                         break;
3633                 if (err)
3634                         goto out_err;
3635
3636                 if (nl80211_send_station(skb,
3637                                 NETLINK_CB(cb->skb).portid,
3638                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
3639                                 dev, wdev->netdev, mac_addr,
3640                                 &sinfo) < 0)
3641                         goto out;
3642
3643                 sta_idx++;
3644         }
3645
3646
3647  out:
3648         cb->args[2] = sta_idx;
3649         err = skb->len;
3650  out_err:
3651         nl80211_finish_wdev_dump(dev);
3652
3653         return err;
3654 }
3655
3656 static int nl80211_get_station(struct sk_buff *skb, struct genl_info *info)
3657 {
3658         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3659         struct net_device *dev = info->user_ptr[1];
3660         struct station_info sinfo;
3661         struct sk_buff *msg;
3662         u8 *mac_addr = NULL;
3663         int err;
3664
3665         memset(&sinfo, 0, sizeof(sinfo));
3666
3667         if (!info->attrs[NL80211_ATTR_MAC])
3668                 return -EINVAL;
3669
3670         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
3671
3672         if (!rdev->ops->get_station)
3673                 return -EOPNOTSUPP;
3674
3675         err = rdev_get_station(rdev, dev, mac_addr, &sinfo);
3676         if (err)
3677                 return err;
3678
3679         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3680         if (!msg)
3681                 return -ENOMEM;
3682
3683         if (nl80211_send_station(msg, info->snd_portid, info->snd_seq, 0,
3684                                  rdev, dev, mac_addr, &sinfo) < 0) {
3685                 nlmsg_free(msg);
3686                 return -ENOBUFS;
3687         }
3688
3689         return genlmsg_reply(msg, info);
3690 }
3691
3692 int cfg80211_check_station_change(struct wiphy *wiphy,
3693                                   struct station_parameters *params,
3694                                   enum cfg80211_station_type statype)
3695 {
3696         if (params->listen_interval != -1)
3697                 return -EINVAL;
3698         if (params->aid)
3699                 return -EINVAL;
3700
3701         /* When you run into this, adjust the code below for the new flag */
3702         BUILD_BUG_ON(NL80211_STA_FLAG_MAX != 7);
3703
3704         switch (statype) {
3705         case CFG80211_STA_MESH_PEER_KERNEL:
3706         case CFG80211_STA_MESH_PEER_USER:
3707                 /*
3708                  * No ignoring the TDLS flag here -- the userspace mesh
3709                  * code doesn't have the bug of including TDLS in the
3710                  * mask everywhere.
3711                  */
3712                 if (params->sta_flags_mask &
3713                                 ~(BIT(NL80211_STA_FLAG_AUTHENTICATED) |
3714                                   BIT(NL80211_STA_FLAG_MFP) |
3715                                   BIT(NL80211_STA_FLAG_AUTHORIZED)))
3716                         return -EINVAL;
3717                 break;
3718         case CFG80211_STA_TDLS_PEER_SETUP:
3719         case CFG80211_STA_TDLS_PEER_ACTIVE:
3720                 if (!(params->sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)))
3721                         return -EINVAL;
3722                 /* ignore since it can't change */
3723                 params->sta_flags_mask &= ~BIT(NL80211_STA_FLAG_TDLS_PEER);
3724                 break;
3725         default:
3726                 /* disallow mesh-specific things */
3727                 if (params->plink_action != NL80211_PLINK_ACTION_NO_ACTION)
3728                         return -EINVAL;
3729                 if (params->local_pm)
3730                         return -EINVAL;
3731                 if (params->sta_modify_mask & STATION_PARAM_APPLY_PLINK_STATE)
3732                         return -EINVAL;
3733         }
3734
3735         if (statype != CFG80211_STA_TDLS_PEER_SETUP &&
3736             statype != CFG80211_STA_TDLS_PEER_ACTIVE) {
3737                 /* TDLS can't be set, ... */
3738                 if (params->sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER))
3739                         return -EINVAL;
3740                 /*
3741                  * ... but don't bother the driver with it. This works around
3742                  * a hostapd/wpa_supplicant issue -- it always includes the
3743                  * TLDS_PEER flag in the mask even for AP mode.
3744                  */
3745                 params->sta_flags_mask &= ~BIT(NL80211_STA_FLAG_TDLS_PEER);
3746         }
3747
3748         if (statype != CFG80211_STA_TDLS_PEER_SETUP) {
3749                 /* reject other things that can't change */
3750                 if (params->sta_modify_mask & STATION_PARAM_APPLY_UAPSD)
3751                         return -EINVAL;
3752                 if (params->sta_modify_mask & STATION_PARAM_APPLY_CAPABILITY)
3753                         return -EINVAL;
3754                 if (params->supported_rates)
3755                         return -EINVAL;
3756                 if (params->ext_capab || params->ht_capa || params->vht_capa)
3757                         return -EINVAL;
3758         }
3759
3760         if (statype != CFG80211_STA_AP_CLIENT) {
3761                 if (params->vlan)
3762                         return -EINVAL;
3763         }
3764
3765         switch (statype) {
3766         case CFG80211_STA_AP_MLME_CLIENT:
3767                 /* Use this only for authorizing/unauthorizing a station */
3768                 if (!(params->sta_flags_mask & BIT(NL80211_STA_FLAG_AUTHORIZED)))
3769                         return -EOPNOTSUPP;
3770                 break;
3771         case CFG80211_STA_AP_CLIENT:
3772                 /* accept only the listed bits */
3773                 if (params->sta_flags_mask &
3774                                 ~(BIT(NL80211_STA_FLAG_AUTHORIZED) |
3775                                   BIT(NL80211_STA_FLAG_AUTHENTICATED) |
3776                                   BIT(NL80211_STA_FLAG_ASSOCIATED) |
3777                                   BIT(NL80211_STA_FLAG_SHORT_PREAMBLE) |
3778                                   BIT(NL80211_STA_FLAG_WME) |
3779                                   BIT(NL80211_STA_FLAG_MFP)))
3780                         return -EINVAL;
3781
3782                 /* but authenticated/associated only if driver handles it */
3783                 if (!(wiphy->features & NL80211_FEATURE_FULL_AP_CLIENT_STATE) &&
3784                     params->sta_flags_mask &
3785                                 (BIT(NL80211_STA_FLAG_AUTHENTICATED) |
3786                                  BIT(NL80211_STA_FLAG_ASSOCIATED)))
3787                         return -EINVAL;
3788                 break;
3789         case CFG80211_STA_IBSS:
3790         case CFG80211_STA_AP_STA:
3791                 /* reject any changes other than AUTHORIZED */
3792                 if (params->sta_flags_mask & ~BIT(NL80211_STA_FLAG_AUTHORIZED))
3793                         return -EINVAL;
3794                 break;
3795         case CFG80211_STA_TDLS_PEER_SETUP:
3796                 /* reject any changes other than AUTHORIZED or WME */
3797                 if (params->sta_flags_mask & ~(BIT(NL80211_STA_FLAG_AUTHORIZED) |
3798                                                BIT(NL80211_STA_FLAG_WME)))
3799                         return -EINVAL;
3800                 /* force (at least) rates when authorizing */
3801                 if (params->sta_flags_set & BIT(NL80211_STA_FLAG_AUTHORIZED) &&
3802                     !params->supported_rates)
3803                         return -EINVAL;
3804                 break;
3805         case CFG80211_STA_TDLS_PEER_ACTIVE:
3806                 /* reject any changes */
3807                 return -EINVAL;
3808         case CFG80211_STA_MESH_PEER_KERNEL:
3809                 if (params->sta_modify_mask & STATION_PARAM_APPLY_PLINK_STATE)
3810                         return -EINVAL;
3811                 break;
3812         case CFG80211_STA_MESH_PEER_USER:
3813                 if (params->plink_action != NL80211_PLINK_ACTION_NO_ACTION)
3814                         return -EINVAL;
3815                 break;
3816         }
3817
3818         return 0;
3819 }
3820 EXPORT_SYMBOL(cfg80211_check_station_change);
3821
3822 /*
3823  * Get vlan interface making sure it is running and on the right wiphy.
3824  */
3825 static struct net_device *get_vlan(struct genl_info *info,
3826                                    struct cfg80211_registered_device *rdev)
3827 {
3828         struct nlattr *vlanattr = info->attrs[NL80211_ATTR_STA_VLAN];
3829         struct net_device *v;
3830         int ret;
3831
3832         if (!vlanattr)
3833                 return NULL;
3834
3835         v = dev_get_by_index(genl_info_net(info), nla_get_u32(vlanattr));
3836         if (!v)
3837                 return ERR_PTR(-ENODEV);
3838
3839         if (!v->ieee80211_ptr || v->ieee80211_ptr->wiphy != &rdev->wiphy) {
3840                 ret = -EINVAL;
3841                 goto error;
3842         }
3843
3844         if (v->ieee80211_ptr->iftype != NL80211_IFTYPE_AP_VLAN &&
3845             v->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
3846             v->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO) {
3847                 ret = -EINVAL;
3848                 goto error;
3849         }
3850
3851         if (!netif_running(v)) {
3852                 ret = -ENETDOWN;
3853                 goto error;
3854         }
3855
3856         return v;
3857  error:
3858         dev_put(v);
3859         return ERR_PTR(ret);
3860 }
3861
3862 static struct nla_policy
3863 nl80211_sta_wme_policy[NL80211_STA_WME_MAX + 1] __read_mostly = {
3864         [NL80211_STA_WME_UAPSD_QUEUES] = { .type = NLA_U8 },
3865         [NL80211_STA_WME_MAX_SP] = { .type = NLA_U8 },
3866 };
3867
3868 static int nl80211_parse_sta_wme(struct genl_info *info,
3869                                  struct station_parameters *params)
3870 {
3871         struct nlattr *tb[NL80211_STA_WME_MAX + 1];
3872         struct nlattr *nla;
3873         int err;
3874
3875         /* parse WME attributes if present */
3876         if (!info->attrs[NL80211_ATTR_STA_WME])
3877                 return 0;
3878
3879         nla = info->attrs[NL80211_ATTR_STA_WME];
3880         err = nla_parse_nested(tb, NL80211_STA_WME_MAX, nla,
3881                                nl80211_sta_wme_policy);
3882         if (err)
3883                 return err;
3884
3885         if (tb[NL80211_STA_WME_UAPSD_QUEUES])
3886                 params->uapsd_queues = nla_get_u8(
3887                         tb[NL80211_STA_WME_UAPSD_QUEUES]);
3888         if (params->uapsd_queues & ~IEEE80211_WMM_IE_STA_QOSINFO_AC_MASK)
3889                 return -EINVAL;
3890
3891         if (tb[NL80211_STA_WME_MAX_SP])
3892                 params->max_sp = nla_get_u8(tb[NL80211_STA_WME_MAX_SP]);
3893
3894         if (params->max_sp & ~IEEE80211_WMM_IE_STA_QOSINFO_SP_MASK)
3895                 return -EINVAL;
3896
3897         params->sta_modify_mask |= STATION_PARAM_APPLY_UAPSD;
3898
3899         return 0;
3900 }
3901
3902 static int nl80211_parse_sta_channel_info(struct genl_info *info,
3903                                       struct station_parameters *params)
3904 {
3905         if (info->attrs[NL80211_ATTR_STA_SUPPORTED_CHANNELS]) {
3906                 params->supported_channels =
3907                      nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_CHANNELS]);
3908                 params->supported_channels_len =
3909                      nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_CHANNELS]);
3910                 /*
3911                  * Need to include at least one (first channel, number of
3912                  * channels) tuple for each subband, and must have proper
3913                  * tuples for the rest of the data as well.
3914                  */
3915                 if (params->supported_channels_len < 2)
3916                         return -EINVAL;
3917                 if (params->supported_channels_len % 2)
3918                         return -EINVAL;
3919         }
3920
3921         if (info->attrs[NL80211_ATTR_STA_SUPPORTED_OPER_CLASSES]) {
3922                 params->supported_oper_classes =
3923                  nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_OPER_CLASSES]);
3924                 params->supported_oper_classes_len =
3925                   nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_OPER_CLASSES]);
3926                 /*
3927                  * The value of the Length field of the Supported Operating
3928                  * Classes element is between 2 and 253.
3929                  */
3930                 if (params->supported_oper_classes_len < 2 ||
3931                     params->supported_oper_classes_len > 253)
3932                         return -EINVAL;
3933         }
3934         return 0;
3935 }
3936
3937 static int nl80211_set_station_tdls(struct genl_info *info,
3938                                     struct station_parameters *params)
3939 {
3940         int err;
3941         /* Dummy STA entry gets updated once the peer capabilities are known */
3942         if (info->attrs[NL80211_ATTR_PEER_AID])
3943                 params->aid = nla_get_u16(info->attrs[NL80211_ATTR_PEER_AID]);
3944         if (info->attrs[NL80211_ATTR_HT_CAPABILITY])
3945                 params->ht_capa =
3946                         nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]);
3947         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY])
3948                 params->vht_capa =
3949                         nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY]);
3950
3951         err = nl80211_parse_sta_channel_info(info, params);
3952         if (err)
3953                 return err;
3954
3955         return nl80211_parse_sta_wme(info, params);
3956 }
3957
3958 static int nl80211_set_station(struct sk_buff *skb, struct genl_info *info)
3959 {
3960         struct cfg80211_registered_device *rdev = info->user_ptr[0];
3961         struct net_device *dev = info->user_ptr[1];
3962         struct station_parameters params;
3963         u8 *mac_addr;
3964         int err;
3965
3966         memset(&params, 0, sizeof(params));
3967
3968         params.listen_interval = -1;
3969
3970         if (!rdev->ops->change_station)
3971                 return -EOPNOTSUPP;
3972
3973         if (info->attrs[NL80211_ATTR_STA_AID])
3974                 return -EINVAL;
3975
3976         if (!info->attrs[NL80211_ATTR_MAC])
3977                 return -EINVAL;
3978
3979         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
3980
3981         if (info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]) {
3982                 params.supported_rates =
3983                         nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
3984                 params.supported_rates_len =
3985                         nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
3986         }
3987
3988         if (info->attrs[NL80211_ATTR_STA_CAPABILITY]) {
3989                 params.capability =
3990                         nla_get_u16(info->attrs[NL80211_ATTR_STA_CAPABILITY]);
3991                 params.sta_modify_mask |= STATION_PARAM_APPLY_CAPABILITY;
3992         }
3993
3994         if (info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]) {
3995                 params.ext_capab =
3996                         nla_data(info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]);
3997                 params.ext_capab_len =
3998                         nla_len(info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]);
3999         }
4000
4001         if (info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL])
4002                 return -EINVAL;
4003
4004         if (parse_station_flags(info, dev->ieee80211_ptr->iftype, &params))
4005                 return -EINVAL;
4006
4007         if (info->attrs[NL80211_ATTR_STA_PLINK_ACTION]) {
4008                 params.plink_action =
4009                         nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_ACTION]);
4010                 if (params.plink_action >= NUM_NL80211_PLINK_ACTIONS)
4011                         return -EINVAL;
4012         }
4013
4014         if (info->attrs[NL80211_ATTR_STA_PLINK_STATE]) {
4015                 params.plink_state =
4016                         nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_STATE]);
4017                 if (params.plink_state >= NUM_NL80211_PLINK_STATES)
4018                         return -EINVAL;
4019                 params.sta_modify_mask |= STATION_PARAM_APPLY_PLINK_STATE;
4020         }
4021
4022         if (info->attrs[NL80211_ATTR_LOCAL_MESH_POWER_MODE]) {
4023                 enum nl80211_mesh_power_mode pm = nla_get_u32(
4024                         info->attrs[NL80211_ATTR_LOCAL_MESH_POWER_MODE]);
4025
4026                 if (pm <= NL80211_MESH_POWER_UNKNOWN ||
4027                     pm > NL80211_MESH_POWER_MAX)
4028                         return -EINVAL;
4029
4030                 params.local_pm = pm;
4031         }
4032
4033         /* Include parameters for TDLS peer (will check later) */
4034         err = nl80211_set_station_tdls(info, &params);
4035         if (err)
4036                 return err;
4037
4038         params.vlan = get_vlan(info, rdev);
4039         if (IS_ERR(params.vlan))
4040                 return PTR_ERR(params.vlan);
4041
4042         switch (dev->ieee80211_ptr->iftype) {
4043         case NL80211_IFTYPE_AP:
4044         case NL80211_IFTYPE_AP_VLAN:
4045         case NL80211_IFTYPE_P2P_GO:
4046         case NL80211_IFTYPE_P2P_CLIENT:
4047         case NL80211_IFTYPE_STATION:
4048         case NL80211_IFTYPE_ADHOC:
4049         case NL80211_IFTYPE_MESH_POINT:
4050                 break;
4051         default:
4052                 err = -EOPNOTSUPP;
4053                 goto out_put_vlan;
4054         }
4055
4056         /* driver will call cfg80211_check_station_change() */
4057         err = rdev_change_station(rdev, dev, mac_addr, &params);
4058
4059  out_put_vlan:
4060         if (params.vlan)
4061                 dev_put(params.vlan);
4062
4063         return err;
4064 }
4065
4066 static int nl80211_new_station(struct sk_buff *skb, struct genl_info *info)
4067 {
4068         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4069         int err;
4070         struct net_device *dev = info->user_ptr[1];
4071         struct station_parameters params;
4072         u8 *mac_addr = NULL;
4073
4074         memset(&params, 0, sizeof(params));
4075
4076         if (!rdev->ops->add_station)
4077                 return -EOPNOTSUPP;
4078
4079         if (!info->attrs[NL80211_ATTR_MAC])
4080                 return -EINVAL;
4081
4082         if (!info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL])
4083                 return -EINVAL;
4084
4085         if (!info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES])
4086                 return -EINVAL;
4087
4088         if (!info->attrs[NL80211_ATTR_STA_AID] &&
4089             !info->attrs[NL80211_ATTR_PEER_AID])
4090                 return -EINVAL;
4091
4092         mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
4093         params.supported_rates =
4094                 nla_data(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
4095         params.supported_rates_len =
4096                 nla_len(info->attrs[NL80211_ATTR_STA_SUPPORTED_RATES]);
4097         params.listen_interval =
4098                 nla_get_u16(info->attrs[NL80211_ATTR_STA_LISTEN_INTERVAL]);
4099
4100         if (info->attrs[NL80211_ATTR_PEER_AID])
4101                 params.aid = nla_get_u16(info->attrs[NL80211_ATTR_PEER_AID]);
4102         else
4103                 params.aid = nla_get_u16(info->attrs[NL80211_ATTR_STA_AID]);
4104         if (!params.aid || params.aid > IEEE80211_MAX_AID)
4105                 return -EINVAL;
4106
4107         if (info->attrs[NL80211_ATTR_STA_CAPABILITY]) {
4108                 params.capability =
4109                         nla_get_u16(info->attrs[NL80211_ATTR_STA_CAPABILITY]);
4110                 params.sta_modify_mask |= STATION_PARAM_APPLY_CAPABILITY;
4111         }
4112
4113         if (info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]) {
4114                 params.ext_capab =
4115                         nla_data(info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]);
4116                 params.ext_capab_len =
4117                         nla_len(info->attrs[NL80211_ATTR_STA_EXT_CAPABILITY]);
4118         }
4119
4120         if (info->attrs[NL80211_ATTR_HT_CAPABILITY])
4121                 params.ht_capa =
4122                         nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]);
4123
4124         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY])
4125                 params.vht_capa =
4126                         nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY]);
4127
4128         if (info->attrs[NL80211_ATTR_STA_PLINK_ACTION]) {
4129                 params.plink_action =
4130                         nla_get_u8(info->attrs[NL80211_ATTR_STA_PLINK_ACTION]);
4131                 if (params.plink_action >= NUM_NL80211_PLINK_ACTIONS)
4132                         return -EINVAL;
4133         }
4134
4135         err = nl80211_parse_sta_channel_info(info, &params);
4136         if (err)
4137                 return err;
4138
4139         err = nl80211_parse_sta_wme(info, &params);
4140         if (err)
4141                 return err;
4142
4143         if (parse_station_flags(info, dev->ieee80211_ptr->iftype, &params))
4144                 return -EINVAL;
4145
4146         /* When you run into this, adjust the code below for the new flag */
4147         BUILD_BUG_ON(NL80211_STA_FLAG_MAX != 7);
4148
4149         switch (dev->ieee80211_ptr->iftype) {
4150         case NL80211_IFTYPE_AP:
4151         case NL80211_IFTYPE_AP_VLAN:
4152         case NL80211_IFTYPE_P2P_GO:
4153                 /* ignore WME attributes if iface/sta is not capable */
4154                 if (!(rdev->wiphy.flags & WIPHY_FLAG_AP_UAPSD) ||
4155                     !(params.sta_flags_set & BIT(NL80211_STA_FLAG_WME)))
4156                         params.sta_modify_mask &= ~STATION_PARAM_APPLY_UAPSD;
4157
4158                 /* TDLS peers cannot be added */
4159                 if ((params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)) ||
4160                     info->attrs[NL80211_ATTR_PEER_AID])
4161                         return -EINVAL;
4162                 /* but don't bother the driver with it */
4163                 params.sta_flags_mask &= ~BIT(NL80211_STA_FLAG_TDLS_PEER);
4164
4165                 /* allow authenticated/associated only if driver handles it */
4166                 if (!(rdev->wiphy.features &
4167                                 NL80211_FEATURE_FULL_AP_CLIENT_STATE) &&
4168                     params.sta_flags_mask &
4169                                 (BIT(NL80211_STA_FLAG_AUTHENTICATED) |
4170                                  BIT(NL80211_STA_FLAG_ASSOCIATED)))
4171                         return -EINVAL;
4172
4173                 /* must be last in here for error handling */
4174                 params.vlan = get_vlan(info, rdev);
4175                 if (IS_ERR(params.vlan))
4176                         return PTR_ERR(params.vlan);
4177                 break;
4178         case NL80211_IFTYPE_MESH_POINT:
4179                 /* ignore uAPSD data */
4180                 params.sta_modify_mask &= ~STATION_PARAM_APPLY_UAPSD;
4181
4182                 /* associated is disallowed */
4183                 if (params.sta_flags_mask & BIT(NL80211_STA_FLAG_ASSOCIATED))
4184                         return -EINVAL;
4185                 /* TDLS peers cannot be added */
4186                 if ((params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)) ||
4187                     info->attrs[NL80211_ATTR_PEER_AID])
4188                         return -EINVAL;
4189                 break;
4190         case NL80211_IFTYPE_STATION:
4191         case NL80211_IFTYPE_P2P_CLIENT:
4192                 /* ignore uAPSD data */
4193                 params.sta_modify_mask &= ~STATION_PARAM_APPLY_UAPSD;
4194
4195                 /* these are disallowed */
4196                 if (params.sta_flags_mask &
4197                                 (BIT(NL80211_STA_FLAG_ASSOCIATED) |
4198                                  BIT(NL80211_STA_FLAG_AUTHENTICATED)))
4199                         return -EINVAL;
4200                 /* Only TDLS peers can be added */
4201                 if (!(params.sta_flags_set & BIT(NL80211_STA_FLAG_TDLS_PEER)))
4202                         return -EINVAL;
4203                 /* Can only add if TDLS ... */
4204                 if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS))
4205                         return -EOPNOTSUPP;
4206                 /* ... with external setup is supported */
4207                 if (!(rdev->wiphy.flags & WIPHY_FLAG_TDLS_EXTERNAL_SETUP))
4208                         return -EOPNOTSUPP;
4209                 /*
4210                  * Older wpa_supplicant versions always mark the TDLS peer
4211                  * as authorized, but it shouldn't yet be.
4212                  */
4213                 params.sta_flags_mask &= ~BIT(NL80211_STA_FLAG_AUTHORIZED);
4214                 break;
4215         default:
4216                 return -EOPNOTSUPP;
4217         }
4218
4219         /* be aware of params.vlan when changing code here */
4220
4221         err = rdev_add_station(rdev, dev, mac_addr, &params);
4222
4223         if (params.vlan)
4224                 dev_put(params.vlan);
4225         return err;
4226 }
4227
4228 static int nl80211_del_station(struct sk_buff *skb, struct genl_info *info)
4229 {
4230         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4231         struct net_device *dev = info->user_ptr[1];
4232         u8 *mac_addr = NULL;
4233
4234         if (info->attrs[NL80211_ATTR_MAC])
4235                 mac_addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
4236
4237         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
4238             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP_VLAN &&
4239             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT &&
4240             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
4241                 return -EINVAL;
4242
4243         if (!rdev->ops->del_station)
4244                 return -EOPNOTSUPP;
4245
4246         return rdev_del_station(rdev, dev, mac_addr);
4247 }
4248
4249 static int nl80211_send_mpath(struct sk_buff *msg, u32 portid, u32 seq,
4250                                 int flags, struct net_device *dev,
4251                                 u8 *dst, u8 *next_hop,
4252                                 struct mpath_info *pinfo)
4253 {
4254         void *hdr;
4255         struct nlattr *pinfoattr;
4256
4257         hdr = nl80211hdr_put(msg, portid, seq, flags, NL80211_CMD_NEW_STATION);
4258         if (!hdr)
4259                 return -1;
4260
4261         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
4262             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, dst) ||
4263             nla_put(msg, NL80211_ATTR_MPATH_NEXT_HOP, ETH_ALEN, next_hop) ||
4264             nla_put_u32(msg, NL80211_ATTR_GENERATION, pinfo->generation))
4265                 goto nla_put_failure;
4266
4267         pinfoattr = nla_nest_start(msg, NL80211_ATTR_MPATH_INFO);
4268         if (!pinfoattr)
4269                 goto nla_put_failure;
4270         if ((pinfo->filled & MPATH_INFO_FRAME_QLEN) &&
4271             nla_put_u32(msg, NL80211_MPATH_INFO_FRAME_QLEN,
4272                         pinfo->frame_qlen))
4273                 goto nla_put_failure;
4274         if (((pinfo->filled & MPATH_INFO_SN) &&
4275              nla_put_u32(msg, NL80211_MPATH_INFO_SN, pinfo->sn)) ||
4276             ((pinfo->filled & MPATH_INFO_METRIC) &&
4277              nla_put_u32(msg, NL80211_MPATH_INFO_METRIC,
4278                          pinfo->metric)) ||
4279             ((pinfo->filled & MPATH_INFO_EXPTIME) &&
4280              nla_put_u32(msg, NL80211_MPATH_INFO_EXPTIME,
4281                          pinfo->exptime)) ||
4282             ((pinfo->filled & MPATH_INFO_FLAGS) &&
4283              nla_put_u8(msg, NL80211_MPATH_INFO_FLAGS,
4284                         pinfo->flags)) ||
4285             ((pinfo->filled & MPATH_INFO_DISCOVERY_TIMEOUT) &&
4286              nla_put_u32(msg, NL80211_MPATH_INFO_DISCOVERY_TIMEOUT,
4287                          pinfo->discovery_timeout)) ||
4288             ((pinfo->filled & MPATH_INFO_DISCOVERY_RETRIES) &&
4289              nla_put_u8(msg, NL80211_MPATH_INFO_DISCOVERY_RETRIES,
4290                         pinfo->discovery_retries)))
4291                 goto nla_put_failure;
4292
4293         nla_nest_end(msg, pinfoattr);
4294
4295         return genlmsg_end(msg, hdr);
4296
4297  nla_put_failure:
4298         genlmsg_cancel(msg, hdr);
4299         return -EMSGSIZE;
4300 }
4301
4302 static int nl80211_dump_mpath(struct sk_buff *skb,
4303                               struct netlink_callback *cb)
4304 {
4305         struct mpath_info pinfo;
4306         struct cfg80211_registered_device *dev;
4307         struct wireless_dev *wdev;
4308         u8 dst[ETH_ALEN];
4309         u8 next_hop[ETH_ALEN];
4310         int path_idx = cb->args[2];
4311         int err;
4312
4313         err = nl80211_prepare_wdev_dump(skb, cb, &dev, &wdev);
4314         if (err)
4315                 return err;
4316
4317         if (!dev->ops->dump_mpath) {
4318                 err = -EOPNOTSUPP;
4319                 goto out_err;
4320         }
4321
4322         if (wdev->iftype != NL80211_IFTYPE_MESH_POINT) {
4323                 err = -EOPNOTSUPP;
4324                 goto out_err;
4325         }
4326
4327         while (1) {
4328                 err = rdev_dump_mpath(dev, wdev->netdev, path_idx, dst,
4329                                       next_hop, &pinfo);
4330                 if (err == -ENOENT)
4331                         break;
4332                 if (err)
4333                         goto out_err;
4334
4335                 if (nl80211_send_mpath(skb, NETLINK_CB(cb->skb).portid,
4336                                        cb->nlh->nlmsg_seq, NLM_F_MULTI,
4337                                        wdev->netdev, dst, next_hop,
4338                                        &pinfo) < 0)
4339                         goto out;
4340
4341                 path_idx++;
4342         }
4343
4344
4345  out:
4346         cb->args[2] = path_idx;
4347         err = skb->len;
4348  out_err:
4349         nl80211_finish_wdev_dump(dev);
4350         return err;
4351 }
4352
4353 static int nl80211_get_mpath(struct sk_buff *skb, struct genl_info *info)
4354 {
4355         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4356         int err;
4357         struct net_device *dev = info->user_ptr[1];
4358         struct mpath_info pinfo;
4359         struct sk_buff *msg;
4360         u8 *dst = NULL;
4361         u8 next_hop[ETH_ALEN];
4362
4363         memset(&pinfo, 0, sizeof(pinfo));
4364
4365         if (!info->attrs[NL80211_ATTR_MAC])
4366                 return -EINVAL;
4367
4368         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
4369
4370         if (!rdev->ops->get_mpath)
4371                 return -EOPNOTSUPP;
4372
4373         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
4374                 return -EOPNOTSUPP;
4375
4376         err = rdev_get_mpath(rdev, dev, dst, next_hop, &pinfo);
4377         if (err)
4378                 return err;
4379
4380         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4381         if (!msg)
4382                 return -ENOMEM;
4383
4384         if (nl80211_send_mpath(msg, info->snd_portid, info->snd_seq, 0,
4385                                  dev, dst, next_hop, &pinfo) < 0) {
4386                 nlmsg_free(msg);
4387                 return -ENOBUFS;
4388         }
4389
4390         return genlmsg_reply(msg, info);
4391 }
4392
4393 static int nl80211_set_mpath(struct sk_buff *skb, struct genl_info *info)
4394 {
4395         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4396         struct net_device *dev = info->user_ptr[1];
4397         u8 *dst = NULL;
4398         u8 *next_hop = NULL;
4399
4400         if (!info->attrs[NL80211_ATTR_MAC])
4401                 return -EINVAL;
4402
4403         if (!info->attrs[NL80211_ATTR_MPATH_NEXT_HOP])
4404                 return -EINVAL;
4405
4406         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
4407         next_hop = nla_data(info->attrs[NL80211_ATTR_MPATH_NEXT_HOP]);
4408
4409         if (!rdev->ops->change_mpath)
4410                 return -EOPNOTSUPP;
4411
4412         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
4413                 return -EOPNOTSUPP;
4414
4415         return rdev_change_mpath(rdev, dev, dst, next_hop);
4416 }
4417
4418 static int nl80211_new_mpath(struct sk_buff *skb, struct genl_info *info)
4419 {
4420         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4421         struct net_device *dev = info->user_ptr[1];
4422         u8 *dst = NULL;
4423         u8 *next_hop = NULL;
4424
4425         if (!info->attrs[NL80211_ATTR_MAC])
4426                 return -EINVAL;
4427
4428         if (!info->attrs[NL80211_ATTR_MPATH_NEXT_HOP])
4429                 return -EINVAL;
4430
4431         dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
4432         next_hop = nla_data(info->attrs[NL80211_ATTR_MPATH_NEXT_HOP]);
4433
4434         if (!rdev->ops->add_mpath)
4435                 return -EOPNOTSUPP;
4436
4437         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
4438                 return -EOPNOTSUPP;
4439
4440         return rdev_add_mpath(rdev, dev, dst, next_hop);
4441 }
4442
4443 static int nl80211_del_mpath(struct sk_buff *skb, struct genl_info *info)
4444 {
4445         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4446         struct net_device *dev = info->user_ptr[1];
4447         u8 *dst = NULL;
4448
4449         if (info->attrs[NL80211_ATTR_MAC])
4450                 dst = nla_data(info->attrs[NL80211_ATTR_MAC]);
4451
4452         if (!rdev->ops->del_mpath)
4453                 return -EOPNOTSUPP;
4454
4455         return rdev_del_mpath(rdev, dev, dst);
4456 }
4457
4458 static int nl80211_set_bss(struct sk_buff *skb, struct genl_info *info)
4459 {
4460         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4461         struct net_device *dev = info->user_ptr[1];
4462         struct bss_parameters params;
4463
4464         memset(&params, 0, sizeof(params));
4465         /* default to not changing parameters */
4466         params.use_cts_prot = -1;
4467         params.use_short_preamble = -1;
4468         params.use_short_slot_time = -1;
4469         params.ap_isolate = -1;
4470         params.ht_opmode = -1;
4471         params.p2p_ctwindow = -1;
4472         params.p2p_opp_ps = -1;
4473
4474         if (info->attrs[NL80211_ATTR_BSS_CTS_PROT])
4475                 params.use_cts_prot =
4476                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_CTS_PROT]);
4477         if (info->attrs[NL80211_ATTR_BSS_SHORT_PREAMBLE])
4478                 params.use_short_preamble =
4479                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_SHORT_PREAMBLE]);
4480         if (info->attrs[NL80211_ATTR_BSS_SHORT_SLOT_TIME])
4481                 params.use_short_slot_time =
4482                     nla_get_u8(info->attrs[NL80211_ATTR_BSS_SHORT_SLOT_TIME]);
4483         if (info->attrs[NL80211_ATTR_BSS_BASIC_RATES]) {
4484                 params.basic_rates =
4485                         nla_data(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
4486                 params.basic_rates_len =
4487                         nla_len(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
4488         }
4489         if (info->attrs[NL80211_ATTR_AP_ISOLATE])
4490                 params.ap_isolate = !!nla_get_u8(info->attrs[NL80211_ATTR_AP_ISOLATE]);
4491         if (info->attrs[NL80211_ATTR_BSS_HT_OPMODE])
4492                 params.ht_opmode =
4493                         nla_get_u16(info->attrs[NL80211_ATTR_BSS_HT_OPMODE]);
4494
4495         if (info->attrs[NL80211_ATTR_P2P_CTWINDOW]) {
4496                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
4497                         return -EINVAL;
4498                 params.p2p_ctwindow =
4499                         nla_get_s8(info->attrs[NL80211_ATTR_P2P_CTWINDOW]);
4500                 if (params.p2p_ctwindow < 0)
4501                         return -EINVAL;
4502                 if (params.p2p_ctwindow != 0 &&
4503                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_CTWIN))
4504                         return -EINVAL;
4505         }
4506
4507         if (info->attrs[NL80211_ATTR_P2P_OPPPS]) {
4508                 u8 tmp;
4509
4510                 if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
4511                         return -EINVAL;
4512                 tmp = nla_get_u8(info->attrs[NL80211_ATTR_P2P_OPPPS]);
4513                 if (tmp > 1)
4514                         return -EINVAL;
4515                 params.p2p_opp_ps = tmp;
4516                 if (params.p2p_opp_ps &&
4517                     !(rdev->wiphy.features & NL80211_FEATURE_P2P_GO_OPPPS))
4518                         return -EINVAL;
4519         }
4520
4521         if (!rdev->ops->change_bss)
4522                 return -EOPNOTSUPP;
4523
4524         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_AP &&
4525             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_GO)
4526                 return -EOPNOTSUPP;
4527
4528         return rdev_change_bss(rdev, dev, &params);
4529 }
4530
4531 static const struct nla_policy reg_rule_policy[NL80211_REG_RULE_ATTR_MAX + 1] = {
4532         [NL80211_ATTR_REG_RULE_FLAGS]           = { .type = NLA_U32 },
4533         [NL80211_ATTR_FREQ_RANGE_START]         = { .type = NLA_U32 },
4534         [NL80211_ATTR_FREQ_RANGE_END]           = { .type = NLA_U32 },
4535         [NL80211_ATTR_FREQ_RANGE_MAX_BW]        = { .type = NLA_U32 },
4536         [NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN]  = { .type = NLA_U32 },
4537         [NL80211_ATTR_POWER_RULE_MAX_EIRP]      = { .type = NLA_U32 },
4538 };
4539
4540 static int parse_reg_rule(struct nlattr *tb[],
4541         struct ieee80211_reg_rule *reg_rule)
4542 {
4543         struct ieee80211_freq_range *freq_range = &reg_rule->freq_range;
4544         struct ieee80211_power_rule *power_rule = &reg_rule->power_rule;
4545
4546         if (!tb[NL80211_ATTR_REG_RULE_FLAGS])
4547                 return -EINVAL;
4548         if (!tb[NL80211_ATTR_FREQ_RANGE_START])
4549                 return -EINVAL;
4550         if (!tb[NL80211_ATTR_FREQ_RANGE_END])
4551                 return -EINVAL;
4552         if (!tb[NL80211_ATTR_FREQ_RANGE_MAX_BW])
4553                 return -EINVAL;
4554         if (!tb[NL80211_ATTR_POWER_RULE_MAX_EIRP])
4555                 return -EINVAL;
4556
4557         reg_rule->flags = nla_get_u32(tb[NL80211_ATTR_REG_RULE_FLAGS]);
4558
4559         freq_range->start_freq_khz =
4560                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]);
4561         freq_range->end_freq_khz =
4562                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]);
4563         freq_range->max_bandwidth_khz =
4564                 nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]);
4565
4566         power_rule->max_eirp =
4567                 nla_get_u32(tb[NL80211_ATTR_POWER_RULE_MAX_EIRP]);
4568
4569         if (tb[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN])
4570                 power_rule->max_antenna_gain =
4571                         nla_get_u32(tb[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN]);
4572
4573         return 0;
4574 }
4575
4576 static int nl80211_req_set_reg(struct sk_buff *skb, struct genl_info *info)
4577 {
4578         int r;
4579         char *data = NULL;
4580         enum nl80211_user_reg_hint_type user_reg_hint_type;
4581
4582         /*
4583          * You should only get this when cfg80211 hasn't yet initialized
4584          * completely when built-in to the kernel right between the time
4585          * window between nl80211_init() and regulatory_init(), if that is
4586          * even possible.
4587          */
4588         if (unlikely(!rcu_access_pointer(cfg80211_regdomain)))
4589                 return -EINPROGRESS;
4590
4591         if (!info->attrs[NL80211_ATTR_REG_ALPHA2])
4592                 return -EINVAL;
4593
4594         data = nla_data(info->attrs[NL80211_ATTR_REG_ALPHA2]);
4595
4596         if (info->attrs[NL80211_ATTR_USER_REG_HINT_TYPE])
4597                 user_reg_hint_type =
4598                   nla_get_u32(info->attrs[NL80211_ATTR_USER_REG_HINT_TYPE]);
4599         else
4600                 user_reg_hint_type = NL80211_USER_REG_HINT_USER;
4601
4602         switch (user_reg_hint_type) {
4603         case NL80211_USER_REG_HINT_USER:
4604         case NL80211_USER_REG_HINT_CELL_BASE:
4605                 break;
4606         default:
4607                 return -EINVAL;
4608         }
4609
4610         r = regulatory_hint_user(data, user_reg_hint_type);
4611
4612         return r;
4613 }
4614
4615 static int nl80211_get_mesh_config(struct sk_buff *skb,
4616                                    struct genl_info *info)
4617 {
4618         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4619         struct net_device *dev = info->user_ptr[1];
4620         struct wireless_dev *wdev = dev->ieee80211_ptr;
4621         struct mesh_config cur_params;
4622         int err = 0;
4623         void *hdr;
4624         struct nlattr *pinfoattr;
4625         struct sk_buff *msg;
4626
4627         if (wdev->iftype != NL80211_IFTYPE_MESH_POINT)
4628                 return -EOPNOTSUPP;
4629
4630         if (!rdev->ops->get_mesh_config)
4631                 return -EOPNOTSUPP;
4632
4633         wdev_lock(wdev);
4634         /* If not connected, get default parameters */
4635         if (!wdev->mesh_id_len)
4636                 memcpy(&cur_params, &default_mesh_config, sizeof(cur_params));
4637         else
4638                 err = rdev_get_mesh_config(rdev, dev, &cur_params);
4639         wdev_unlock(wdev);
4640
4641         if (err)
4642                 return err;
4643
4644         /* Draw up a netlink message to send back */
4645         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4646         if (!msg)
4647                 return -ENOMEM;
4648         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
4649                              NL80211_CMD_GET_MESH_CONFIG);
4650         if (!hdr)
4651                 goto out;
4652         pinfoattr = nla_nest_start(msg, NL80211_ATTR_MESH_CONFIG);
4653         if (!pinfoattr)
4654                 goto nla_put_failure;
4655         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
4656             nla_put_u16(msg, NL80211_MESHCONF_RETRY_TIMEOUT,
4657                         cur_params.dot11MeshRetryTimeout) ||
4658             nla_put_u16(msg, NL80211_MESHCONF_CONFIRM_TIMEOUT,
4659                         cur_params.dot11MeshConfirmTimeout) ||
4660             nla_put_u16(msg, NL80211_MESHCONF_HOLDING_TIMEOUT,
4661                         cur_params.dot11MeshHoldingTimeout) ||
4662             nla_put_u16(msg, NL80211_MESHCONF_MAX_PEER_LINKS,
4663                         cur_params.dot11MeshMaxPeerLinks) ||
4664             nla_put_u8(msg, NL80211_MESHCONF_MAX_RETRIES,
4665                        cur_params.dot11MeshMaxRetries) ||
4666             nla_put_u8(msg, NL80211_MESHCONF_TTL,
4667                        cur_params.dot11MeshTTL) ||
4668             nla_put_u8(msg, NL80211_MESHCONF_ELEMENT_TTL,
4669                        cur_params.element_ttl) ||
4670             nla_put_u8(msg, NL80211_MESHCONF_AUTO_OPEN_PLINKS,
4671                        cur_params.auto_open_plinks) ||
4672             nla_put_u32(msg, NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR,
4673                         cur_params.dot11MeshNbrOffsetMaxNeighbor) ||
4674             nla_put_u8(msg, NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES,
4675                        cur_params.dot11MeshHWMPmaxPREQretries) ||
4676             nla_put_u32(msg, NL80211_MESHCONF_PATH_REFRESH_TIME,
4677                         cur_params.path_refresh_time) ||
4678             nla_put_u16(msg, NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT,
4679                         cur_params.min_discovery_timeout) ||
4680             nla_put_u32(msg, NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT,
4681                         cur_params.dot11MeshHWMPactivePathTimeout) ||
4682             nla_put_u16(msg, NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL,
4683                         cur_params.dot11MeshHWMPpreqMinInterval) ||
4684             nla_put_u16(msg, NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL,
4685                         cur_params.dot11MeshHWMPperrMinInterval) ||
4686             nla_put_u16(msg, NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
4687                         cur_params.dot11MeshHWMPnetDiameterTraversalTime) ||
4688             nla_put_u8(msg, NL80211_MESHCONF_HWMP_ROOTMODE,
4689                        cur_params.dot11MeshHWMPRootMode) ||
4690             nla_put_u16(msg, NL80211_MESHCONF_HWMP_RANN_INTERVAL,
4691                         cur_params.dot11MeshHWMPRannInterval) ||
4692             nla_put_u8(msg, NL80211_MESHCONF_GATE_ANNOUNCEMENTS,
4693                        cur_params.dot11MeshGateAnnouncementProtocol) ||
4694             nla_put_u8(msg, NL80211_MESHCONF_FORWARDING,
4695                        cur_params.dot11MeshForwarding) ||
4696             nla_put_u32(msg, NL80211_MESHCONF_RSSI_THRESHOLD,
4697                         cur_params.rssi_threshold) ||
4698             nla_put_u32(msg, NL80211_MESHCONF_HT_OPMODE,
4699                         cur_params.ht_opmode) ||
4700             nla_put_u32(msg, NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT,
4701                         cur_params.dot11MeshHWMPactivePathToRootTimeout) ||
4702             nla_put_u16(msg, NL80211_MESHCONF_HWMP_ROOT_INTERVAL,
4703                         cur_params.dot11MeshHWMProotInterval) ||
4704             nla_put_u16(msg, NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL,
4705                         cur_params.dot11MeshHWMPconfirmationInterval) ||
4706             nla_put_u32(msg, NL80211_MESHCONF_POWER_MODE,
4707                         cur_params.power_mode) ||
4708             nla_put_u16(msg, NL80211_MESHCONF_AWAKE_WINDOW,
4709                         cur_params.dot11MeshAwakeWindowDuration) ||
4710             nla_put_u32(msg, NL80211_MESHCONF_PLINK_TIMEOUT,
4711                         cur_params.plink_timeout))
4712                 goto nla_put_failure;
4713         nla_nest_end(msg, pinfoattr);
4714         genlmsg_end(msg, hdr);
4715         return genlmsg_reply(msg, info);
4716
4717  nla_put_failure:
4718         genlmsg_cancel(msg, hdr);
4719  out:
4720         nlmsg_free(msg);
4721         return -ENOBUFS;
4722 }
4723
4724 static const struct nla_policy nl80211_meshconf_params_policy[NL80211_MESHCONF_ATTR_MAX+1] = {
4725         [NL80211_MESHCONF_RETRY_TIMEOUT] = { .type = NLA_U16 },
4726         [NL80211_MESHCONF_CONFIRM_TIMEOUT] = { .type = NLA_U16 },
4727         [NL80211_MESHCONF_HOLDING_TIMEOUT] = { .type = NLA_U16 },
4728         [NL80211_MESHCONF_MAX_PEER_LINKS] = { .type = NLA_U16 },
4729         [NL80211_MESHCONF_MAX_RETRIES] = { .type = NLA_U8 },
4730         [NL80211_MESHCONF_TTL] = { .type = NLA_U8 },
4731         [NL80211_MESHCONF_ELEMENT_TTL] = { .type = NLA_U8 },
4732         [NL80211_MESHCONF_AUTO_OPEN_PLINKS] = { .type = NLA_U8 },
4733         [NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR] = { .type = NLA_U32 },
4734         [NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES] = { .type = NLA_U8 },
4735         [NL80211_MESHCONF_PATH_REFRESH_TIME] = { .type = NLA_U32 },
4736         [NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT] = { .type = NLA_U16 },
4737         [NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT] = { .type = NLA_U32 },
4738         [NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL] = { .type = NLA_U16 },
4739         [NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL] = { .type = NLA_U16 },
4740         [NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME] = { .type = NLA_U16 },
4741         [NL80211_MESHCONF_HWMP_ROOTMODE] = { .type = NLA_U8 },
4742         [NL80211_MESHCONF_HWMP_RANN_INTERVAL] = { .type = NLA_U16 },
4743         [NL80211_MESHCONF_GATE_ANNOUNCEMENTS] = { .type = NLA_U8 },
4744         [NL80211_MESHCONF_FORWARDING] = { .type = NLA_U8 },
4745         [NL80211_MESHCONF_RSSI_THRESHOLD] = { .type = NLA_U32 },
4746         [NL80211_MESHCONF_HT_OPMODE] = { .type = NLA_U16 },
4747         [NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT] = { .type = NLA_U32 },
4748         [NL80211_MESHCONF_HWMP_ROOT_INTERVAL] = { .type = NLA_U16 },
4749         [NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL] = { .type = NLA_U16 },
4750         [NL80211_MESHCONF_POWER_MODE] = { .type = NLA_U32 },
4751         [NL80211_MESHCONF_AWAKE_WINDOW] = { .type = NLA_U16 },
4752         [NL80211_MESHCONF_PLINK_TIMEOUT] = { .type = NLA_U32 },
4753 };
4754
4755 static const struct nla_policy
4756         nl80211_mesh_setup_params_policy[NL80211_MESH_SETUP_ATTR_MAX+1] = {
4757         [NL80211_MESH_SETUP_ENABLE_VENDOR_SYNC] = { .type = NLA_U8 },
4758         [NL80211_MESH_SETUP_ENABLE_VENDOR_PATH_SEL] = { .type = NLA_U8 },
4759         [NL80211_MESH_SETUP_ENABLE_VENDOR_METRIC] = { .type = NLA_U8 },
4760         [NL80211_MESH_SETUP_USERSPACE_AUTH] = { .type = NLA_FLAG },
4761         [NL80211_MESH_SETUP_AUTH_PROTOCOL] = { .type = NLA_U8 },
4762         [NL80211_MESH_SETUP_USERSPACE_MPM] = { .type = NLA_FLAG },
4763         [NL80211_MESH_SETUP_IE] = { .type = NLA_BINARY,
4764                                     .len = IEEE80211_MAX_DATA_LEN },
4765         [NL80211_MESH_SETUP_USERSPACE_AMPE] = { .type = NLA_FLAG },
4766 };
4767
4768 static int nl80211_parse_mesh_config(struct genl_info *info,
4769                                      struct mesh_config *cfg,
4770                                      u32 *mask_out)
4771 {
4772         struct nlattr *tb[NL80211_MESHCONF_ATTR_MAX + 1];
4773         u32 mask = 0;
4774
4775 #define FILL_IN_MESH_PARAM_IF_SET(tb, cfg, param, min, max, mask, attr, fn) \
4776 do {                                                                        \
4777         if (tb[attr]) {                                                     \
4778                 if (fn(tb[attr]) < min || fn(tb[attr]) > max)               \
4779                         return -EINVAL;                                     \
4780                 cfg->param = fn(tb[attr]);                                  \
4781                 mask |= (1 << (attr - 1));                                  \
4782         }                                                                   \
4783 } while (0)
4784
4785
4786         if (!info->attrs[NL80211_ATTR_MESH_CONFIG])
4787                 return -EINVAL;
4788         if (nla_parse_nested(tb, NL80211_MESHCONF_ATTR_MAX,
4789                              info->attrs[NL80211_ATTR_MESH_CONFIG],
4790                              nl80211_meshconf_params_policy))
4791                 return -EINVAL;
4792
4793         /* This makes sure that there aren't more than 32 mesh config
4794          * parameters (otherwise our bitfield scheme would not work.) */
4795         BUILD_BUG_ON(NL80211_MESHCONF_ATTR_MAX > 32);
4796
4797         /* Fill in the params struct */
4798         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshRetryTimeout, 1, 255,
4799                                   mask, NL80211_MESHCONF_RETRY_TIMEOUT,
4800                                   nla_get_u16);
4801         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshConfirmTimeout, 1, 255,
4802                                   mask, NL80211_MESHCONF_CONFIRM_TIMEOUT,
4803                                   nla_get_u16);
4804         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHoldingTimeout, 1, 255,
4805                                   mask, NL80211_MESHCONF_HOLDING_TIMEOUT,
4806                                   nla_get_u16);
4807         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshMaxPeerLinks, 0, 255,
4808                                   mask, NL80211_MESHCONF_MAX_PEER_LINKS,
4809                                   nla_get_u16);
4810         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshMaxRetries, 0, 16,
4811                                   mask, NL80211_MESHCONF_MAX_RETRIES,
4812                                   nla_get_u8);
4813         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshTTL, 1, 255,
4814                                   mask, NL80211_MESHCONF_TTL, nla_get_u8);
4815         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, element_ttl, 1, 255,
4816                                   mask, NL80211_MESHCONF_ELEMENT_TTL,
4817                                   nla_get_u8);
4818         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, auto_open_plinks, 0, 1,
4819                                   mask, NL80211_MESHCONF_AUTO_OPEN_PLINKS,
4820                                   nla_get_u8);
4821         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshNbrOffsetMaxNeighbor,
4822                                   1, 255, mask,
4823                                   NL80211_MESHCONF_SYNC_OFFSET_MAX_NEIGHBOR,
4824                                   nla_get_u32);
4825         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPmaxPREQretries, 0, 255,
4826                                   mask, NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES,
4827                                   nla_get_u8);
4828         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, path_refresh_time, 1, 65535,
4829                                   mask, NL80211_MESHCONF_PATH_REFRESH_TIME,
4830                                   nla_get_u32);
4831         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, min_discovery_timeout, 1, 65535,
4832                                   mask, NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT,
4833                                   nla_get_u16);
4834         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPactivePathTimeout,
4835                                   1, 65535, mask,
4836                                   NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT,
4837                                   nla_get_u32);
4838         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPpreqMinInterval,
4839                                   1, 65535, mask,
4840                                   NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL,
4841                                   nla_get_u16);
4842         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPperrMinInterval,
4843                                   1, 65535, mask,
4844                                   NL80211_MESHCONF_HWMP_PERR_MIN_INTERVAL,
4845                                   nla_get_u16);
4846         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
4847                                   dot11MeshHWMPnetDiameterTraversalTime,
4848                                   1, 65535, mask,
4849                                   NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME,
4850                                   nla_get_u16);
4851         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPRootMode, 0, 4,
4852                                   mask, NL80211_MESHCONF_HWMP_ROOTMODE,
4853                                   nla_get_u8);
4854         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPRannInterval, 1, 65535,
4855                                   mask, NL80211_MESHCONF_HWMP_RANN_INTERVAL,
4856                                   nla_get_u16);
4857         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
4858                                   dot11MeshGateAnnouncementProtocol, 0, 1,
4859                                   mask, NL80211_MESHCONF_GATE_ANNOUNCEMENTS,
4860                                   nla_get_u8);
4861         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshForwarding, 0, 1,
4862                                   mask, NL80211_MESHCONF_FORWARDING,
4863                                   nla_get_u8);
4864         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, rssi_threshold, -255, 0,
4865                                   mask, NL80211_MESHCONF_RSSI_THRESHOLD,
4866                                   nla_get_s32);
4867         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, ht_opmode, 0, 16,
4868                                   mask, NL80211_MESHCONF_HT_OPMODE,
4869                                   nla_get_u16);
4870         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMPactivePathToRootTimeout,
4871                                   1, 65535, mask,
4872                                   NL80211_MESHCONF_HWMP_PATH_TO_ROOT_TIMEOUT,
4873                                   nla_get_u32);
4874         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshHWMProotInterval, 1, 65535,
4875                                   mask, NL80211_MESHCONF_HWMP_ROOT_INTERVAL,
4876                                   nla_get_u16);
4877         FILL_IN_MESH_PARAM_IF_SET(tb, cfg,
4878                                   dot11MeshHWMPconfirmationInterval,
4879                                   1, 65535, mask,
4880                                   NL80211_MESHCONF_HWMP_CONFIRMATION_INTERVAL,
4881                                   nla_get_u16);
4882         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, power_mode,
4883                                   NL80211_MESH_POWER_ACTIVE,
4884                                   NL80211_MESH_POWER_MAX,
4885                                   mask, NL80211_MESHCONF_POWER_MODE,
4886                                   nla_get_u32);
4887         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, dot11MeshAwakeWindowDuration,
4888                                   0, 65535, mask,
4889                                   NL80211_MESHCONF_AWAKE_WINDOW, nla_get_u16);
4890         FILL_IN_MESH_PARAM_IF_SET(tb, cfg, plink_timeout, 1, 0xffffffff,
4891                                   mask, NL80211_MESHCONF_PLINK_TIMEOUT,
4892                                   nla_get_u32);
4893         if (mask_out)
4894                 *mask_out = mask;
4895
4896         return 0;
4897
4898 #undef FILL_IN_MESH_PARAM_IF_SET
4899 }
4900
4901 static int nl80211_parse_mesh_setup(struct genl_info *info,
4902                                      struct mesh_setup *setup)
4903 {
4904         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4905         struct nlattr *tb[NL80211_MESH_SETUP_ATTR_MAX + 1];
4906
4907         if (!info->attrs[NL80211_ATTR_MESH_SETUP])
4908                 return -EINVAL;
4909         if (nla_parse_nested(tb, NL80211_MESH_SETUP_ATTR_MAX,
4910                              info->attrs[NL80211_ATTR_MESH_SETUP],
4911                              nl80211_mesh_setup_params_policy))
4912                 return -EINVAL;
4913
4914         if (tb[NL80211_MESH_SETUP_ENABLE_VENDOR_SYNC])
4915                 setup->sync_method =
4916                 (nla_get_u8(tb[NL80211_MESH_SETUP_ENABLE_VENDOR_SYNC])) ?
4917                  IEEE80211_SYNC_METHOD_VENDOR :
4918                  IEEE80211_SYNC_METHOD_NEIGHBOR_OFFSET;
4919
4920         if (tb[NL80211_MESH_SETUP_ENABLE_VENDOR_PATH_SEL])
4921                 setup->path_sel_proto =
4922                 (nla_get_u8(tb[NL80211_MESH_SETUP_ENABLE_VENDOR_PATH_SEL])) ?
4923                  IEEE80211_PATH_PROTOCOL_VENDOR :
4924                  IEEE80211_PATH_PROTOCOL_HWMP;
4925
4926         if (tb[NL80211_MESH_SETUP_ENABLE_VENDOR_METRIC])
4927                 setup->path_metric =
4928                 (nla_get_u8(tb[NL80211_MESH_SETUP_ENABLE_VENDOR_METRIC])) ?
4929                  IEEE80211_PATH_METRIC_VENDOR :
4930                  IEEE80211_PATH_METRIC_AIRTIME;
4931
4932
4933         if (tb[NL80211_MESH_SETUP_IE]) {
4934                 struct nlattr *ieattr =
4935                         tb[NL80211_MESH_SETUP_IE];
4936                 if (!is_valid_ie_attr(ieattr))
4937                         return -EINVAL;
4938                 setup->ie = nla_data(ieattr);
4939                 setup->ie_len = nla_len(ieattr);
4940         }
4941         if (tb[NL80211_MESH_SETUP_USERSPACE_MPM] &&
4942             !(rdev->wiphy.features & NL80211_FEATURE_USERSPACE_MPM))
4943                 return -EINVAL;
4944         setup->user_mpm = nla_get_flag(tb[NL80211_MESH_SETUP_USERSPACE_MPM]);
4945         setup->is_authenticated = nla_get_flag(tb[NL80211_MESH_SETUP_USERSPACE_AUTH]);
4946         setup->is_secure = nla_get_flag(tb[NL80211_MESH_SETUP_USERSPACE_AMPE]);
4947         if (setup->is_secure)
4948                 setup->user_mpm = true;
4949
4950         if (tb[NL80211_MESH_SETUP_AUTH_PROTOCOL]) {
4951                 if (!setup->user_mpm)
4952                         return -EINVAL;
4953                 setup->auth_id =
4954                         nla_get_u8(tb[NL80211_MESH_SETUP_AUTH_PROTOCOL]);
4955         }
4956
4957         return 0;
4958 }
4959
4960 static int nl80211_update_mesh_config(struct sk_buff *skb,
4961                                       struct genl_info *info)
4962 {
4963         struct cfg80211_registered_device *rdev = info->user_ptr[0];
4964         struct net_device *dev = info->user_ptr[1];
4965         struct wireless_dev *wdev = dev->ieee80211_ptr;
4966         struct mesh_config cfg;
4967         u32 mask;
4968         int err;
4969
4970         if (wdev->iftype != NL80211_IFTYPE_MESH_POINT)
4971                 return -EOPNOTSUPP;
4972
4973         if (!rdev->ops->update_mesh_config)
4974                 return -EOPNOTSUPP;
4975
4976         err = nl80211_parse_mesh_config(info, &cfg, &mask);
4977         if (err)
4978                 return err;
4979
4980         wdev_lock(wdev);
4981         if (!wdev->mesh_id_len)
4982                 err = -ENOLINK;
4983
4984         if (!err)
4985                 err = rdev_update_mesh_config(rdev, dev, mask, &cfg);
4986
4987         wdev_unlock(wdev);
4988
4989         return err;
4990 }
4991
4992 static int nl80211_get_reg(struct sk_buff *skb, struct genl_info *info)
4993 {
4994         const struct ieee80211_regdomain *regdom;
4995         struct sk_buff *msg;
4996         void *hdr = NULL;
4997         struct nlattr *nl_reg_rules;
4998         unsigned int i;
4999
5000         if (!cfg80211_regdomain)
5001                 return -EINVAL;
5002
5003         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
5004         if (!msg)
5005                 return -ENOBUFS;
5006
5007         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
5008                              NL80211_CMD_GET_REG);
5009         if (!hdr)
5010                 goto put_failure;
5011
5012         if (reg_last_request_cell_base() &&
5013             nla_put_u32(msg, NL80211_ATTR_USER_REG_HINT_TYPE,
5014                         NL80211_USER_REG_HINT_CELL_BASE))
5015                 goto nla_put_failure;
5016
5017         rcu_read_lock();
5018         regdom = rcu_dereference(cfg80211_regdomain);
5019
5020         if (nla_put_string(msg, NL80211_ATTR_REG_ALPHA2, regdom->alpha2) ||
5021             (regdom->dfs_region &&
5022              nla_put_u8(msg, NL80211_ATTR_DFS_REGION, regdom->dfs_region)))
5023                 goto nla_put_failure_rcu;
5024
5025         nl_reg_rules = nla_nest_start(msg, NL80211_ATTR_REG_RULES);
5026         if (!nl_reg_rules)
5027                 goto nla_put_failure_rcu;
5028
5029         for (i = 0; i < regdom->n_reg_rules; i++) {
5030                 struct nlattr *nl_reg_rule;
5031                 const struct ieee80211_reg_rule *reg_rule;
5032                 const struct ieee80211_freq_range *freq_range;
5033                 const struct ieee80211_power_rule *power_rule;
5034
5035                 reg_rule = &regdom->reg_rules[i];
5036                 freq_range = &reg_rule->freq_range;
5037                 power_rule = &reg_rule->power_rule;
5038
5039                 nl_reg_rule = nla_nest_start(msg, i);
5040                 if (!nl_reg_rule)
5041                         goto nla_put_failure_rcu;
5042
5043                 if (nla_put_u32(msg, NL80211_ATTR_REG_RULE_FLAGS,
5044                                 reg_rule->flags) ||
5045                     nla_put_u32(msg, NL80211_ATTR_FREQ_RANGE_START,
5046                                 freq_range->start_freq_khz) ||
5047                     nla_put_u32(msg, NL80211_ATTR_FREQ_RANGE_END,
5048                                 freq_range->end_freq_khz) ||
5049                     nla_put_u32(msg, NL80211_ATTR_FREQ_RANGE_MAX_BW,
5050                                 freq_range->max_bandwidth_khz) ||
5051                     nla_put_u32(msg, NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN,
5052                                 power_rule->max_antenna_gain) ||
5053                     nla_put_u32(msg, NL80211_ATTR_POWER_RULE_MAX_EIRP,
5054                                 power_rule->max_eirp))
5055                         goto nla_put_failure_rcu;
5056
5057                 nla_nest_end(msg, nl_reg_rule);
5058         }
5059         rcu_read_unlock();
5060
5061         nla_nest_end(msg, nl_reg_rules);
5062
5063         genlmsg_end(msg, hdr);
5064         return genlmsg_reply(msg, info);
5065
5066 nla_put_failure_rcu:
5067         rcu_read_unlock();
5068 nla_put_failure:
5069         genlmsg_cancel(msg, hdr);
5070 put_failure:
5071         nlmsg_free(msg);
5072         return -EMSGSIZE;
5073 }
5074
5075 static int nl80211_set_reg(struct sk_buff *skb, struct genl_info *info)
5076 {
5077         struct nlattr *tb[NL80211_REG_RULE_ATTR_MAX + 1];
5078         struct nlattr *nl_reg_rule;
5079         char *alpha2 = NULL;
5080         int rem_reg_rules = 0, r = 0;
5081         u32 num_rules = 0, rule_idx = 0, size_of_regd;
5082         u8 dfs_region = 0;
5083         struct ieee80211_regdomain *rd = NULL;
5084
5085         if (!info->attrs[NL80211_ATTR_REG_ALPHA2])
5086                 return -EINVAL;
5087
5088         if (!info->attrs[NL80211_ATTR_REG_RULES])
5089                 return -EINVAL;
5090
5091         alpha2 = nla_data(info->attrs[NL80211_ATTR_REG_ALPHA2]);
5092
5093         if (info->attrs[NL80211_ATTR_DFS_REGION])
5094                 dfs_region = nla_get_u8(info->attrs[NL80211_ATTR_DFS_REGION]);
5095
5096         nla_for_each_nested(nl_reg_rule, info->attrs[NL80211_ATTR_REG_RULES],
5097                             rem_reg_rules) {
5098                 num_rules++;
5099                 if (num_rules > NL80211_MAX_SUPP_REG_RULES)
5100                         return -EINVAL;
5101         }
5102
5103         size_of_regd = sizeof(struct ieee80211_regdomain) +
5104                        num_rules * sizeof(struct ieee80211_reg_rule);
5105
5106         rd = kzalloc(size_of_regd, GFP_KERNEL);
5107         if (!rd)
5108                 return -ENOMEM;
5109
5110         rd->n_reg_rules = num_rules;
5111         rd->alpha2[0] = alpha2[0];
5112         rd->alpha2[1] = alpha2[1];
5113
5114         /*
5115          * Disable DFS master mode if the DFS region was
5116          * not supported or known on this kernel.
5117          */
5118         if (reg_supported_dfs_region(dfs_region))
5119                 rd->dfs_region = dfs_region;
5120
5121         nla_for_each_nested(nl_reg_rule, info->attrs[NL80211_ATTR_REG_RULES],
5122                             rem_reg_rules) {
5123                 nla_parse(tb, NL80211_REG_RULE_ATTR_MAX,
5124                           nla_data(nl_reg_rule), nla_len(nl_reg_rule),
5125                           reg_rule_policy);
5126                 r = parse_reg_rule(tb, &rd->reg_rules[rule_idx]);
5127                 if (r)
5128                         goto bad_reg;
5129
5130                 rule_idx++;
5131
5132                 if (rule_idx > NL80211_MAX_SUPP_REG_RULES) {
5133                         r = -EINVAL;
5134                         goto bad_reg;
5135                 }
5136         }
5137
5138         r = set_regdom(rd);
5139         /* set_regdom took ownership */
5140         rd = NULL;
5141
5142  bad_reg:
5143         kfree(rd);
5144         return r;
5145 }
5146
5147 static int validate_scan_freqs(struct nlattr *freqs)
5148 {
5149         struct nlattr *attr1, *attr2;
5150         int n_channels = 0, tmp1, tmp2;
5151
5152         nla_for_each_nested(attr1, freqs, tmp1) {
5153                 n_channels++;
5154                 /*
5155                  * Some hardware has a limited channel list for
5156                  * scanning, and it is pretty much nonsensical
5157                  * to scan for a channel twice, so disallow that
5158                  * and don't require drivers to check that the
5159                  * channel list they get isn't longer than what
5160                  * they can scan, as long as they can scan all
5161                  * the channels they registered at once.
5162                  */
5163                 nla_for_each_nested(attr2, freqs, tmp2)
5164                         if (attr1 != attr2 &&
5165                             nla_get_u32(attr1) == nla_get_u32(attr2))
5166                                 return 0;
5167         }
5168
5169         return n_channels;
5170 }
5171
5172 static int nl80211_trigger_scan(struct sk_buff *skb, struct genl_info *info)
5173 {
5174         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5175         struct wireless_dev *wdev = info->user_ptr[1];
5176         struct cfg80211_scan_request *request;
5177         struct nlattr *attr;
5178         struct wiphy *wiphy;
5179         int err, tmp, n_ssids = 0, n_channels, i;
5180         size_t ie_len;
5181
5182         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
5183                 return -EINVAL;
5184
5185         wiphy = &rdev->wiphy;
5186
5187         if (!rdev->ops->scan)
5188                 return -EOPNOTSUPP;
5189
5190         if (rdev->scan_req) {
5191                 err = -EBUSY;
5192                 goto unlock;
5193         }
5194
5195         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
5196                 n_channels = validate_scan_freqs(
5197                                 info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]);
5198                 if (!n_channels) {
5199                         err = -EINVAL;
5200                         goto unlock;
5201                 }
5202         } else {
5203                 enum ieee80211_band band;
5204                 n_channels = 0;
5205
5206                 for (band = 0; band < IEEE80211_NUM_BANDS; band++)
5207                         if (wiphy->bands[band])
5208                                 n_channels += wiphy->bands[band]->n_channels;
5209         }
5210
5211         if (info->attrs[NL80211_ATTR_SCAN_SSIDS])
5212                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS], tmp)
5213                         n_ssids++;
5214
5215         if (n_ssids > wiphy->max_scan_ssids) {
5216                 err = -EINVAL;
5217                 goto unlock;
5218         }
5219
5220         if (info->attrs[NL80211_ATTR_IE])
5221                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5222         else
5223                 ie_len = 0;
5224
5225         if (ie_len > wiphy->max_scan_ie_len) {
5226                 err = -EINVAL;
5227                 goto unlock;
5228         }
5229
5230         request = kzalloc(sizeof(*request)
5231                         + sizeof(*request->ssids) * n_ssids
5232                         + sizeof(*request->channels) * n_channels
5233                         + ie_len, GFP_KERNEL);
5234         if (!request) {
5235                 err = -ENOMEM;
5236                 goto unlock;
5237         }
5238
5239         if (n_ssids)
5240                 request->ssids = (void *)&request->channels[n_channels];
5241         request->n_ssids = n_ssids;
5242         if (ie_len) {
5243                 if (request->ssids)
5244                         request->ie = (void *)(request->ssids + n_ssids);
5245                 else
5246                         request->ie = (void *)(request->channels + n_channels);
5247         }
5248
5249         i = 0;
5250         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
5251                 /* user specified, bail out if channel not found */
5252                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_FREQUENCIES], tmp) {
5253                         struct ieee80211_channel *chan;
5254
5255                         chan = ieee80211_get_channel(wiphy, nla_get_u32(attr));
5256
5257                         if (!chan) {
5258                                 err = -EINVAL;
5259                                 goto out_free;
5260                         }
5261
5262                         /* ignore disabled channels */
5263                         if (chan->flags & IEEE80211_CHAN_DISABLED)
5264                                 continue;
5265
5266                         request->channels[i] = chan;
5267                         i++;
5268                 }
5269         } else {
5270                 enum ieee80211_band band;
5271
5272                 /* all channels */
5273                 for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
5274                         int j;
5275                         if (!wiphy->bands[band])
5276                                 continue;
5277                         for (j = 0; j < wiphy->bands[band]->n_channels; j++) {
5278                                 struct ieee80211_channel *chan;
5279
5280                                 chan = &wiphy->bands[band]->channels[j];
5281
5282                                 if (chan->flags & IEEE80211_CHAN_DISABLED)
5283                                         continue;
5284
5285                                 request->channels[i] = chan;
5286                                 i++;
5287                         }
5288                 }
5289         }
5290
5291         if (!i) {
5292                 err = -EINVAL;
5293                 goto out_free;
5294         }
5295
5296         request->n_channels = i;
5297
5298         i = 0;
5299         if (info->attrs[NL80211_ATTR_SCAN_SSIDS]) {
5300                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS], tmp) {
5301                         if (nla_len(attr) > IEEE80211_MAX_SSID_LEN) {
5302                                 err = -EINVAL;
5303                                 goto out_free;
5304                         }
5305                         request->ssids[i].ssid_len = nla_len(attr);
5306                         memcpy(request->ssids[i].ssid, nla_data(attr), nla_len(attr));
5307                         i++;
5308                 }
5309         }
5310
5311         if (info->attrs[NL80211_ATTR_IE]) {
5312                 request->ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5313                 memcpy((void *)request->ie,
5314                        nla_data(info->attrs[NL80211_ATTR_IE]),
5315                        request->ie_len);
5316         }
5317
5318         for (i = 0; i < IEEE80211_NUM_BANDS; i++)
5319                 if (wiphy->bands[i])
5320                         request->rates[i] =
5321                                 (1 << wiphy->bands[i]->n_bitrates) - 1;
5322
5323         if (info->attrs[NL80211_ATTR_SCAN_SUPP_RATES]) {
5324                 nla_for_each_nested(attr,
5325                                     info->attrs[NL80211_ATTR_SCAN_SUPP_RATES],
5326                                     tmp) {
5327                         enum ieee80211_band band = nla_type(attr);
5328
5329                         if (band < 0 || band >= IEEE80211_NUM_BANDS) {
5330                                 err = -EINVAL;
5331                                 goto out_free;
5332                         }
5333                         err = ieee80211_get_ratemask(wiphy->bands[band],
5334                                                      nla_data(attr),
5335                                                      nla_len(attr),
5336                                                      &request->rates[band]);
5337                         if (err)
5338                                 goto out_free;
5339                 }
5340         }
5341
5342         if (info->attrs[NL80211_ATTR_SCAN_FLAGS]) {
5343                 request->flags = nla_get_u32(
5344                         info->attrs[NL80211_ATTR_SCAN_FLAGS]);
5345                 if ((request->flags & NL80211_SCAN_FLAG_LOW_PRIORITY) &&
5346                     !(wiphy->features & NL80211_FEATURE_LOW_PRIORITY_SCAN)) {
5347                         err = -EOPNOTSUPP;
5348                         goto out_free;
5349                 }
5350         }
5351
5352         request->no_cck =
5353                 nla_get_flag(info->attrs[NL80211_ATTR_TX_NO_CCK_RATE]);
5354
5355         request->wdev = wdev;
5356         request->wiphy = &rdev->wiphy;
5357         request->scan_start = jiffies;
5358
5359         rdev->scan_req = request;
5360         err = rdev_scan(rdev, request);
5361
5362         if (!err) {
5363                 nl80211_send_scan_start(rdev, wdev);
5364                 if (wdev->netdev)
5365                         dev_hold(wdev->netdev);
5366         } else {
5367  out_free:
5368                 rdev->scan_req = NULL;
5369                 kfree(request);
5370         }
5371
5372  unlock:
5373         return err;
5374 }
5375
5376 static int nl80211_start_sched_scan(struct sk_buff *skb,
5377                                     struct genl_info *info)
5378 {
5379         struct cfg80211_sched_scan_request *request;
5380         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5381         struct net_device *dev = info->user_ptr[1];
5382         struct nlattr *attr;
5383         struct wiphy *wiphy;
5384         int err, tmp, n_ssids = 0, n_match_sets = 0, n_channels, i;
5385         u32 interval;
5386         enum ieee80211_band band;
5387         size_t ie_len;
5388         struct nlattr *tb[NL80211_SCHED_SCAN_MATCH_ATTR_MAX + 1];
5389
5390         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN) ||
5391             !rdev->ops->sched_scan_start)
5392                 return -EOPNOTSUPP;
5393
5394         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
5395                 return -EINVAL;
5396
5397         if (!info->attrs[NL80211_ATTR_SCHED_SCAN_INTERVAL])
5398                 return -EINVAL;
5399
5400         interval = nla_get_u32(info->attrs[NL80211_ATTR_SCHED_SCAN_INTERVAL]);
5401         if (interval == 0)
5402                 return -EINVAL;
5403
5404         wiphy = &rdev->wiphy;
5405
5406         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
5407                 n_channels = validate_scan_freqs(
5408                                 info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]);
5409                 if (!n_channels)
5410                         return -EINVAL;
5411         } else {
5412                 n_channels = 0;
5413
5414                 for (band = 0; band < IEEE80211_NUM_BANDS; band++)
5415                         if (wiphy->bands[band])
5416                                 n_channels += wiphy->bands[band]->n_channels;
5417         }
5418
5419         if (info->attrs[NL80211_ATTR_SCAN_SSIDS])
5420                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS],
5421                                     tmp)
5422                         n_ssids++;
5423
5424         if (n_ssids > wiphy->max_sched_scan_ssids)
5425                 return -EINVAL;
5426
5427         if (info->attrs[NL80211_ATTR_SCHED_SCAN_MATCH])
5428                 nla_for_each_nested(attr,
5429                                     info->attrs[NL80211_ATTR_SCHED_SCAN_MATCH],
5430                                     tmp)
5431                         n_match_sets++;
5432
5433         if (n_match_sets > wiphy->max_match_sets)
5434                 return -EINVAL;
5435
5436         if (info->attrs[NL80211_ATTR_IE])
5437                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5438         else
5439                 ie_len = 0;
5440
5441         if (ie_len > wiphy->max_sched_scan_ie_len)
5442                 return -EINVAL;
5443
5444         if (rdev->sched_scan_req) {
5445                 err = -EINPROGRESS;
5446                 goto out;
5447         }
5448
5449         request = kzalloc(sizeof(*request)
5450                         + sizeof(*request->ssids) * n_ssids
5451                         + sizeof(*request->match_sets) * n_match_sets
5452                         + sizeof(*request->channels) * n_channels
5453                         + ie_len, GFP_KERNEL);
5454         if (!request) {
5455                 err = -ENOMEM;
5456                 goto out;
5457         }
5458
5459         if (n_ssids)
5460                 request->ssids = (void *)&request->channels[n_channels];
5461         request->n_ssids = n_ssids;
5462         if (ie_len) {
5463                 if (request->ssids)
5464                         request->ie = (void *)(request->ssids + n_ssids);
5465                 else
5466                         request->ie = (void *)(request->channels + n_channels);
5467         }
5468
5469         if (n_match_sets) {
5470                 if (request->ie)
5471                         request->match_sets = (void *)(request->ie + ie_len);
5472                 else if (request->ssids)
5473                         request->match_sets =
5474                                 (void *)(request->ssids + n_ssids);
5475                 else
5476                         request->match_sets =
5477                                 (void *)(request->channels + n_channels);
5478         }
5479         request->n_match_sets = n_match_sets;
5480
5481         i = 0;
5482         if (info->attrs[NL80211_ATTR_SCAN_FREQUENCIES]) {
5483                 /* user specified, bail out if channel not found */
5484                 nla_for_each_nested(attr,
5485                                     info->attrs[NL80211_ATTR_SCAN_FREQUENCIES],
5486                                     tmp) {
5487                         struct ieee80211_channel *chan;
5488
5489                         chan = ieee80211_get_channel(wiphy, nla_get_u32(attr));
5490
5491                         if (!chan) {
5492                                 err = -EINVAL;
5493                                 goto out_free;
5494                         }
5495
5496                         /* ignore disabled channels */
5497                         if (chan->flags & IEEE80211_CHAN_DISABLED)
5498                                 continue;
5499
5500                         request->channels[i] = chan;
5501                         i++;
5502                 }
5503         } else {
5504                 /* all channels */
5505                 for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
5506                         int j;
5507                         if (!wiphy->bands[band])
5508                                 continue;
5509                         for (j = 0; j < wiphy->bands[band]->n_channels; j++) {
5510                                 struct ieee80211_channel *chan;
5511
5512                                 chan = &wiphy->bands[band]->channels[j];
5513
5514                                 if (chan->flags & IEEE80211_CHAN_DISABLED)
5515                                         continue;
5516
5517                                 request->channels[i] = chan;
5518                                 i++;
5519                         }
5520                 }
5521         }
5522
5523         if (!i) {
5524                 err = -EINVAL;
5525                 goto out_free;
5526         }
5527
5528         request->n_channels = i;
5529
5530         i = 0;
5531         if (info->attrs[NL80211_ATTR_SCAN_SSIDS]) {
5532                 nla_for_each_nested(attr, info->attrs[NL80211_ATTR_SCAN_SSIDS],
5533                                     tmp) {
5534                         if (nla_len(attr) > IEEE80211_MAX_SSID_LEN) {
5535                                 err = -EINVAL;
5536                                 goto out_free;
5537                         }
5538                         request->ssids[i].ssid_len = nla_len(attr);
5539                         memcpy(request->ssids[i].ssid, nla_data(attr),
5540                                nla_len(attr));
5541                         i++;
5542                 }
5543         }
5544
5545         i = 0;
5546         if (info->attrs[NL80211_ATTR_SCHED_SCAN_MATCH]) {
5547                 nla_for_each_nested(attr,
5548                                     info->attrs[NL80211_ATTR_SCHED_SCAN_MATCH],
5549                                     tmp) {
5550                         struct nlattr *ssid, *rssi;
5551
5552                         nla_parse(tb, NL80211_SCHED_SCAN_MATCH_ATTR_MAX,
5553                                   nla_data(attr), nla_len(attr),
5554                                   nl80211_match_policy);
5555                         ssid = tb[NL80211_SCHED_SCAN_MATCH_ATTR_SSID];
5556                         if (ssid) {
5557                                 if (nla_len(ssid) > IEEE80211_MAX_SSID_LEN) {
5558                                         err = -EINVAL;
5559                                         goto out_free;
5560                                 }
5561                                 memcpy(request->match_sets[i].ssid.ssid,
5562                                        nla_data(ssid), nla_len(ssid));
5563                                 request->match_sets[i].ssid.ssid_len =
5564                                         nla_len(ssid);
5565                         }
5566                         rssi = tb[NL80211_SCHED_SCAN_MATCH_ATTR_RSSI];
5567                         if (rssi)
5568                                 request->rssi_thold = nla_get_u32(rssi);
5569                         else
5570                                 request->rssi_thold =
5571                                                    NL80211_SCAN_RSSI_THOLD_OFF;
5572                         i++;
5573                 }
5574         }
5575
5576         if (info->attrs[NL80211_ATTR_IE]) {
5577                 request->ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
5578                 memcpy((void *)request->ie,
5579                        nla_data(info->attrs[NL80211_ATTR_IE]),
5580                        request->ie_len);
5581         }
5582
5583         if (info->attrs[NL80211_ATTR_SCAN_FLAGS]) {
5584                 request->flags = nla_get_u32(
5585                         info->attrs[NL80211_ATTR_SCAN_FLAGS]);
5586                 if ((request->flags & NL80211_SCAN_FLAG_LOW_PRIORITY) &&
5587                     !(wiphy->features & NL80211_FEATURE_LOW_PRIORITY_SCAN)) {
5588                         err = -EOPNOTSUPP;
5589                         goto out_free;
5590                 }
5591         }
5592
5593         request->dev = dev;
5594         request->wiphy = &rdev->wiphy;
5595         request->interval = interval;
5596         request->scan_start = jiffies;
5597
5598         err = rdev_sched_scan_start(rdev, dev, request);
5599         if (!err) {
5600                 rdev->sched_scan_req = request;
5601                 nl80211_send_sched_scan(rdev, dev,
5602                                         NL80211_CMD_START_SCHED_SCAN);
5603                 goto out;
5604         }
5605
5606 out_free:
5607         kfree(request);
5608 out:
5609         return err;
5610 }
5611
5612 static int nl80211_stop_sched_scan(struct sk_buff *skb,
5613                                    struct genl_info *info)
5614 {
5615         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5616
5617         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_SCHED_SCAN) ||
5618             !rdev->ops->sched_scan_stop)
5619                 return -EOPNOTSUPP;
5620
5621         return __cfg80211_stop_sched_scan(rdev, false);
5622 }
5623
5624 static int nl80211_start_radar_detection(struct sk_buff *skb,
5625                                          struct genl_info *info)
5626 {
5627         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5628         struct net_device *dev = info->user_ptr[1];
5629         struct wireless_dev *wdev = dev->ieee80211_ptr;
5630         struct cfg80211_chan_def chandef;
5631         int err;
5632
5633         err = nl80211_parse_chandef(rdev, info, &chandef);
5634         if (err)
5635                 return err;
5636
5637         if (netif_carrier_ok(dev))
5638                 return -EBUSY;
5639
5640         if (wdev->cac_started)
5641                 return -EBUSY;
5642
5643         err = cfg80211_chandef_dfs_required(wdev->wiphy, &chandef);
5644         if (err < 0)
5645                 return err;
5646
5647         if (err == 0)
5648                 return -EINVAL;
5649
5650         if (!cfg80211_chandef_dfs_usable(wdev->wiphy, &chandef))
5651                 return -EINVAL;
5652
5653         if (!rdev->ops->start_radar_detection)
5654                 return -EOPNOTSUPP;
5655
5656         err = cfg80211_can_use_iftype_chan(rdev, wdev, wdev->iftype,
5657                                            chandef.chan, CHAN_MODE_SHARED,
5658                                            BIT(chandef.width));
5659         if (err)
5660                 return err;
5661
5662         err = rdev->ops->start_radar_detection(&rdev->wiphy, dev, &chandef);
5663         if (!err) {
5664                 wdev->channel = chandef.chan;
5665                 wdev->cac_started = true;
5666                 wdev->cac_start_time = jiffies;
5667         }
5668         return err;
5669 }
5670
5671 static int nl80211_channel_switch(struct sk_buff *skb, struct genl_info *info)
5672 {
5673         struct cfg80211_registered_device *rdev = info->user_ptr[0];
5674         struct net_device *dev = info->user_ptr[1];
5675         struct wireless_dev *wdev = dev->ieee80211_ptr;
5676         struct cfg80211_csa_settings params;
5677         /* csa_attrs is defined static to avoid waste of stack size - this
5678          * function is called under RTNL lock, so this should not be a problem.
5679          */
5680         static struct nlattr *csa_attrs[NL80211_ATTR_MAX+1];
5681         u8 radar_detect_width = 0;
5682         int err;
5683         bool need_new_beacon = false;
5684
5685         if (!rdev->ops->channel_switch ||
5686             !(rdev->wiphy.flags & WIPHY_FLAG_HAS_CHANNEL_SWITCH))
5687                 return -EOPNOTSUPP;
5688
5689         switch (dev->ieee80211_ptr->iftype) {
5690         case NL80211_IFTYPE_AP:
5691         case NL80211_IFTYPE_P2P_GO:
5692                 need_new_beacon = true;
5693
5694                 /* useless if AP is not running */
5695                 if (!wdev->beacon_interval)
5696                         return -EINVAL;
5697                 break;
5698         case NL80211_IFTYPE_ADHOC:
5699         case NL80211_IFTYPE_MESH_POINT:
5700                 break;
5701         default:
5702                 return -EOPNOTSUPP;
5703         }
5704
5705         memset(&params, 0, sizeof(params));
5706
5707         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ] ||
5708             !info->attrs[NL80211_ATTR_CH_SWITCH_COUNT])
5709                 return -EINVAL;
5710
5711         /* only important for AP, IBSS and mesh create IEs internally */
5712         if (need_new_beacon && !info->attrs[NL80211_ATTR_CSA_IES])
5713                 return -EINVAL;
5714
5715         params.count = nla_get_u32(info->attrs[NL80211_ATTR_CH_SWITCH_COUNT]);
5716
5717         if (!need_new_beacon)
5718                 goto skip_beacons;
5719
5720         err = nl80211_parse_beacon(info->attrs, &params.beacon_after);
5721         if (err)
5722                 return err;
5723
5724         err = nla_parse_nested(csa_attrs, NL80211_ATTR_MAX,
5725                                info->attrs[NL80211_ATTR_CSA_IES],
5726                                nl80211_policy);
5727         if (err)
5728                 return err;
5729
5730         err = nl80211_parse_beacon(csa_attrs, &params.beacon_csa);
5731         if (err)
5732                 return err;
5733
5734         if (!csa_attrs[NL80211_ATTR_CSA_C_OFF_BEACON])
5735                 return -EINVAL;
5736
5737         params.counter_offset_beacon =
5738                 nla_get_u16(csa_attrs[NL80211_ATTR_CSA_C_OFF_BEACON]);
5739         if (params.counter_offset_beacon >= params.beacon_csa.tail_len)
5740                 return -EINVAL;
5741
5742         /* sanity check - counters should be the same */
5743         if (params.beacon_csa.tail[params.counter_offset_beacon] !=
5744             params.count)
5745                 return -EINVAL;
5746
5747         if (csa_attrs[NL80211_ATTR_CSA_C_OFF_PRESP]) {
5748                 params.counter_offset_presp =
5749                         nla_get_u16(csa_attrs[NL80211_ATTR_CSA_C_OFF_PRESP]);
5750                 if (params.counter_offset_presp >=
5751                     params.beacon_csa.probe_resp_len)
5752                         return -EINVAL;
5753
5754                 if (params.beacon_csa.probe_resp[params.counter_offset_presp] !=
5755                     params.count)
5756                         return -EINVAL;
5757         }
5758
5759 skip_beacons:
5760         err = nl80211_parse_chandef(rdev, info, &params.chandef);
5761         if (err)
5762                 return err;
5763
5764         if (!cfg80211_reg_can_beacon(&rdev->wiphy, &params.chandef))
5765                 return -EINVAL;
5766
5767         if (dev->ieee80211_ptr->iftype == NL80211_IFTYPE_AP ||
5768             dev->ieee80211_ptr->iftype == NL80211_IFTYPE_P2P_GO ||
5769             dev->ieee80211_ptr->iftype == NL80211_IFTYPE_ADHOC) {
5770                 err = cfg80211_chandef_dfs_required(wdev->wiphy,
5771                                                     &params.chandef);
5772                 if (err < 0) {
5773                         return err;
5774                 } else if (err) {
5775                         radar_detect_width = BIT(params.chandef.width);
5776                         params.radar_required = true;
5777                 }
5778         }
5779
5780         err = cfg80211_can_use_iftype_chan(rdev, wdev, wdev->iftype,
5781                                            params.chandef.chan,
5782                                            CHAN_MODE_SHARED,
5783                                            radar_detect_width);
5784         if (err)
5785                 return err;
5786
5787         if (info->attrs[NL80211_ATTR_CH_SWITCH_BLOCK_TX])
5788                 params.block_tx = true;
5789
5790         return rdev_channel_switch(rdev, dev, &params);
5791 }
5792
5793 static int nl80211_send_bss(struct sk_buff *msg, struct netlink_callback *cb,
5794                             u32 seq, int flags,
5795                             struct cfg80211_registered_device *rdev,
5796                             struct wireless_dev *wdev,
5797                             struct cfg80211_internal_bss *intbss)
5798 {
5799         struct cfg80211_bss *res = &intbss->pub;
5800         const struct cfg80211_bss_ies *ies;
5801         void *hdr;
5802         struct nlattr *bss;
5803         bool tsf = false;
5804
5805         ASSERT_WDEV_LOCK(wdev);
5806
5807         hdr = nl80211hdr_put(msg, NETLINK_CB(cb->skb).portid, seq, flags,
5808                              NL80211_CMD_NEW_SCAN_RESULTS);
5809         if (!hdr)
5810                 return -1;
5811
5812         genl_dump_check_consistent(cb, hdr, &nl80211_fam);
5813
5814         if (nla_put_u32(msg, NL80211_ATTR_GENERATION, rdev->bss_generation))
5815                 goto nla_put_failure;
5816         if (wdev->netdev &&
5817             nla_put_u32(msg, NL80211_ATTR_IFINDEX, wdev->netdev->ifindex))
5818                 goto nla_put_failure;
5819         if (nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)))
5820                 goto nla_put_failure;
5821
5822         bss = nla_nest_start(msg, NL80211_ATTR_BSS);
5823         if (!bss)
5824                 goto nla_put_failure;
5825         if ((!is_zero_ether_addr(res->bssid) &&
5826              nla_put(msg, NL80211_BSS_BSSID, ETH_ALEN, res->bssid)))
5827                 goto nla_put_failure;
5828
5829         rcu_read_lock();
5830         ies = rcu_dereference(res->ies);
5831         if (ies) {
5832                 if (nla_put_u64(msg, NL80211_BSS_TSF, ies->tsf))
5833                         goto fail_unlock_rcu;
5834                 tsf = true;
5835                 if (ies->len && nla_put(msg, NL80211_BSS_INFORMATION_ELEMENTS,
5836                                         ies->len, ies->data))
5837                         goto fail_unlock_rcu;
5838         }
5839         ies = rcu_dereference(res->beacon_ies);
5840         if (ies) {
5841                 if (!tsf && nla_put_u64(msg, NL80211_BSS_TSF, ies->tsf))
5842                         goto fail_unlock_rcu;
5843                 if (ies->len && nla_put(msg, NL80211_BSS_BEACON_IES,
5844                                         ies->len, ies->data))
5845                         goto fail_unlock_rcu;
5846         }
5847         rcu_read_unlock();
5848
5849         if (res->beacon_interval &&
5850             nla_put_u16(msg, NL80211_BSS_BEACON_INTERVAL, res->beacon_interval))
5851                 goto nla_put_failure;
5852         if (nla_put_u16(msg, NL80211_BSS_CAPABILITY, res->capability) ||
5853             nla_put_u32(msg, NL80211_BSS_FREQUENCY, res->channel->center_freq) ||
5854             nla_put_u32(msg, NL80211_BSS_CHAN_WIDTH, res->scan_width) ||
5855             nla_put_u32(msg, NL80211_BSS_SEEN_MS_AGO,
5856                         jiffies_to_msecs(jiffies - intbss->ts)))
5857                 goto nla_put_failure;
5858
5859         switch (rdev->wiphy.signal_type) {
5860         case CFG80211_SIGNAL_TYPE_MBM:
5861                 if (nla_put_u32(msg, NL80211_BSS_SIGNAL_MBM, res->signal))
5862                         goto nla_put_failure;
5863                 break;
5864         case CFG80211_SIGNAL_TYPE_UNSPEC:
5865                 if (nla_put_u8(msg, NL80211_BSS_SIGNAL_UNSPEC, res->signal))
5866                         goto nla_put_failure;
5867                 break;
5868         default:
5869                 break;
5870         }
5871
5872         switch (wdev->iftype) {
5873         case NL80211_IFTYPE_P2P_CLIENT:
5874         case NL80211_IFTYPE_STATION:
5875                 if (intbss == wdev->current_bss &&
5876                     nla_put_u32(msg, NL80211_BSS_STATUS,
5877                                 NL80211_BSS_STATUS_ASSOCIATED))
5878                         goto nla_put_failure;
5879                 break;
5880         case NL80211_IFTYPE_ADHOC:
5881                 if (intbss == wdev->current_bss &&
5882                     nla_put_u32(msg, NL80211_BSS_STATUS,
5883                                 NL80211_BSS_STATUS_IBSS_JOINED))
5884                         goto nla_put_failure;
5885                 break;
5886         default:
5887                 break;
5888         }
5889
5890         nla_nest_end(msg, bss);
5891
5892         return genlmsg_end(msg, hdr);
5893
5894  fail_unlock_rcu:
5895         rcu_read_unlock();
5896  nla_put_failure:
5897         genlmsg_cancel(msg, hdr);
5898         return -EMSGSIZE;
5899 }
5900
5901 static int nl80211_dump_scan(struct sk_buff *skb, struct netlink_callback *cb)
5902 {
5903         struct cfg80211_registered_device *rdev;
5904         struct cfg80211_internal_bss *scan;
5905         struct wireless_dev *wdev;
5906         int start = cb->args[2], idx = 0;
5907         int err;
5908
5909         err = nl80211_prepare_wdev_dump(skb, cb, &rdev, &wdev);
5910         if (err)
5911                 return err;
5912
5913         wdev_lock(wdev);
5914         spin_lock_bh(&rdev->bss_lock);
5915         cfg80211_bss_expire(rdev);
5916
5917         cb->seq = rdev->bss_generation;
5918
5919         list_for_each_entry(scan, &rdev->bss_list, list) {
5920                 if (++idx <= start)
5921                         continue;
5922                 if (nl80211_send_bss(skb, cb,
5923                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
5924                                 rdev, wdev, scan) < 0) {
5925                         idx--;
5926                         break;
5927                 }
5928         }
5929
5930         spin_unlock_bh(&rdev->bss_lock);
5931         wdev_unlock(wdev);
5932
5933         cb->args[2] = idx;
5934         nl80211_finish_wdev_dump(rdev);
5935
5936         return skb->len;
5937 }
5938
5939 static int nl80211_send_survey(struct sk_buff *msg, u32 portid, u32 seq,
5940                                 int flags, struct net_device *dev,
5941                                 struct survey_info *survey)
5942 {
5943         void *hdr;
5944         struct nlattr *infoattr;
5945
5946         hdr = nl80211hdr_put(msg, portid, seq, flags,
5947                              NL80211_CMD_NEW_SURVEY_RESULTS);
5948         if (!hdr)
5949                 return -ENOMEM;
5950
5951         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex))
5952                 goto nla_put_failure;
5953
5954         infoattr = nla_nest_start(msg, NL80211_ATTR_SURVEY_INFO);
5955         if (!infoattr)
5956                 goto nla_put_failure;
5957
5958         if (nla_put_u32(msg, NL80211_SURVEY_INFO_FREQUENCY,
5959                         survey->channel->center_freq))
5960                 goto nla_put_failure;
5961
5962         if ((survey->filled & SURVEY_INFO_NOISE_DBM) &&
5963             nla_put_u8(msg, NL80211_SURVEY_INFO_NOISE, survey->noise))
5964                 goto nla_put_failure;
5965         if ((survey->filled & SURVEY_INFO_IN_USE) &&
5966             nla_put_flag(msg, NL80211_SURVEY_INFO_IN_USE))
5967                 goto nla_put_failure;
5968         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME) &&
5969             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME,
5970                         survey->channel_time))
5971                 goto nla_put_failure;
5972         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME_BUSY) &&
5973             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME_BUSY,
5974                         survey->channel_time_busy))
5975                 goto nla_put_failure;
5976         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME_EXT_BUSY) &&
5977             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME_EXT_BUSY,
5978                         survey->channel_time_ext_busy))
5979                 goto nla_put_failure;
5980         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME_RX) &&
5981             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME_RX,
5982                         survey->channel_time_rx))
5983                 goto nla_put_failure;
5984         if ((survey->filled & SURVEY_INFO_CHANNEL_TIME_TX) &&
5985             nla_put_u64(msg, NL80211_SURVEY_INFO_CHANNEL_TIME_TX,
5986                         survey->channel_time_tx))
5987                 goto nla_put_failure;
5988
5989         nla_nest_end(msg, infoattr);
5990
5991         return genlmsg_end(msg, hdr);
5992
5993  nla_put_failure:
5994         genlmsg_cancel(msg, hdr);
5995         return -EMSGSIZE;
5996 }
5997
5998 static int nl80211_dump_survey(struct sk_buff *skb,
5999                         struct netlink_callback *cb)
6000 {
6001         struct survey_info survey;
6002         struct cfg80211_registered_device *dev;
6003         struct wireless_dev *wdev;
6004         int survey_idx = cb->args[2];
6005         int res;
6006
6007         res = nl80211_prepare_wdev_dump(skb, cb, &dev, &wdev);
6008         if (res)
6009                 return res;
6010
6011         if (!wdev->netdev) {
6012                 res = -EINVAL;
6013                 goto out_err;
6014         }
6015
6016         if (!dev->ops->dump_survey) {
6017                 res = -EOPNOTSUPP;
6018                 goto out_err;
6019         }
6020
6021         while (1) {
6022                 struct ieee80211_channel *chan;
6023
6024                 res = rdev_dump_survey(dev, wdev->netdev, survey_idx, &survey);
6025                 if (res == -ENOENT)
6026                         break;
6027                 if (res)
6028                         goto out_err;
6029
6030                 /* Survey without a channel doesn't make sense */
6031                 if (!survey.channel) {
6032                         res = -EINVAL;
6033                         goto out;
6034                 }
6035
6036                 chan = ieee80211_get_channel(&dev->wiphy,
6037                                              survey.channel->center_freq);
6038                 if (!chan || chan->flags & IEEE80211_CHAN_DISABLED) {
6039                         survey_idx++;
6040                         continue;
6041                 }
6042
6043                 if (nl80211_send_survey(skb,
6044                                 NETLINK_CB(cb->skb).portid,
6045                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
6046                                 wdev->netdev, &survey) < 0)
6047                         goto out;
6048                 survey_idx++;
6049         }
6050
6051  out:
6052         cb->args[2] = survey_idx;
6053         res = skb->len;
6054  out_err:
6055         nl80211_finish_wdev_dump(dev);
6056         return res;
6057 }
6058
6059 static bool nl80211_valid_wpa_versions(u32 wpa_versions)
6060 {
6061         return !(wpa_versions & ~(NL80211_WPA_VERSION_1 |
6062                                   NL80211_WPA_VERSION_2));
6063 }
6064
6065 static int nl80211_authenticate(struct sk_buff *skb, struct genl_info *info)
6066 {
6067         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6068         struct net_device *dev = info->user_ptr[1];
6069         struct ieee80211_channel *chan;
6070         const u8 *bssid, *ssid, *ie = NULL, *sae_data = NULL;
6071         int err, ssid_len, ie_len = 0, sae_data_len = 0;
6072         enum nl80211_auth_type auth_type;
6073         struct key_parse key;
6074         bool local_state_change;
6075
6076         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
6077                 return -EINVAL;
6078
6079         if (!info->attrs[NL80211_ATTR_MAC])
6080                 return -EINVAL;
6081
6082         if (!info->attrs[NL80211_ATTR_AUTH_TYPE])
6083                 return -EINVAL;
6084
6085         if (!info->attrs[NL80211_ATTR_SSID])
6086                 return -EINVAL;
6087
6088         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ])
6089                 return -EINVAL;
6090
6091         err = nl80211_parse_key(info, &key);
6092         if (err)
6093                 return err;
6094
6095         if (key.idx >= 0) {
6096                 if (key.type != -1 && key.type != NL80211_KEYTYPE_GROUP)
6097                         return -EINVAL;
6098                 if (!key.p.key || !key.p.key_len)
6099                         return -EINVAL;
6100                 if ((key.p.cipher != WLAN_CIPHER_SUITE_WEP40 ||
6101                      key.p.key_len != WLAN_KEY_LEN_WEP40) &&
6102                     (key.p.cipher != WLAN_CIPHER_SUITE_WEP104 ||
6103                      key.p.key_len != WLAN_KEY_LEN_WEP104))
6104                         return -EINVAL;
6105                 if (key.idx > 4)
6106                         return -EINVAL;
6107         } else {
6108                 key.p.key_len = 0;
6109                 key.p.key = NULL;
6110         }
6111
6112         if (key.idx >= 0) {
6113                 int i;
6114                 bool ok = false;
6115                 for (i = 0; i < rdev->wiphy.n_cipher_suites; i++) {
6116                         if (key.p.cipher == rdev->wiphy.cipher_suites[i]) {
6117                                 ok = true;
6118                                 break;
6119                         }
6120                 }
6121                 if (!ok)
6122                         return -EINVAL;
6123         }
6124
6125         if (!rdev->ops->auth)
6126                 return -EOPNOTSUPP;
6127
6128         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
6129             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
6130                 return -EOPNOTSUPP;
6131
6132         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
6133         chan = ieee80211_get_channel(&rdev->wiphy,
6134                 nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
6135         if (!chan || (chan->flags & IEEE80211_CHAN_DISABLED))
6136                 return -EINVAL;
6137
6138         ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
6139         ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
6140
6141         if (info->attrs[NL80211_ATTR_IE]) {
6142                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
6143                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
6144         }
6145
6146         auth_type = nla_get_u32(info->attrs[NL80211_ATTR_AUTH_TYPE]);
6147         if (!nl80211_valid_auth_type(rdev, auth_type, NL80211_CMD_AUTHENTICATE))
6148                 return -EINVAL;
6149
6150         if (auth_type == NL80211_AUTHTYPE_SAE &&
6151             !info->attrs[NL80211_ATTR_SAE_DATA])
6152                 return -EINVAL;
6153
6154         if (info->attrs[NL80211_ATTR_SAE_DATA]) {
6155                 if (auth_type != NL80211_AUTHTYPE_SAE)
6156                         return -EINVAL;
6157                 sae_data = nla_data(info->attrs[NL80211_ATTR_SAE_DATA]);
6158                 sae_data_len = nla_len(info->attrs[NL80211_ATTR_SAE_DATA]);
6159                 /* need to include at least Auth Transaction and Status Code */
6160                 if (sae_data_len < 4)
6161                         return -EINVAL;
6162         }
6163
6164         local_state_change = !!info->attrs[NL80211_ATTR_LOCAL_STATE_CHANGE];
6165
6166         /*
6167          * Since we no longer track auth state, ignore
6168          * requests to only change local state.
6169          */
6170         if (local_state_change)
6171                 return 0;
6172
6173         wdev_lock(dev->ieee80211_ptr);
6174         err = cfg80211_mlme_auth(rdev, dev, chan, auth_type, bssid,
6175                                  ssid, ssid_len, ie, ie_len,
6176                                  key.p.key, key.p.key_len, key.idx,
6177                                  sae_data, sae_data_len);
6178         wdev_unlock(dev->ieee80211_ptr);
6179         return err;
6180 }
6181
6182 static int nl80211_crypto_settings(struct cfg80211_registered_device *rdev,
6183                                    struct genl_info *info,
6184                                    struct cfg80211_crypto_settings *settings,
6185                                    int cipher_limit)
6186 {
6187         memset(settings, 0, sizeof(*settings));
6188
6189         settings->control_port = info->attrs[NL80211_ATTR_CONTROL_PORT];
6190
6191         if (info->attrs[NL80211_ATTR_CONTROL_PORT_ETHERTYPE]) {
6192                 u16 proto;
6193                 proto = nla_get_u16(
6194                         info->attrs[NL80211_ATTR_CONTROL_PORT_ETHERTYPE]);
6195                 settings->control_port_ethertype = cpu_to_be16(proto);
6196                 if (!(rdev->wiphy.flags & WIPHY_FLAG_CONTROL_PORT_PROTOCOL) &&
6197                     proto != ETH_P_PAE)
6198                         return -EINVAL;
6199                 if (info->attrs[NL80211_ATTR_CONTROL_PORT_NO_ENCRYPT])
6200                         settings->control_port_no_encrypt = true;
6201         } else
6202                 settings->control_port_ethertype = cpu_to_be16(ETH_P_PAE);
6203
6204         if (info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]) {
6205                 void *data;
6206                 int len, i;
6207
6208                 data = nla_data(info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]);
6209                 len = nla_len(info->attrs[NL80211_ATTR_CIPHER_SUITES_PAIRWISE]);
6210                 settings->n_ciphers_pairwise = len / sizeof(u32);
6211
6212                 if (len % sizeof(u32))
6213                         return -EINVAL;
6214
6215                 if (settings->n_ciphers_pairwise > cipher_limit)
6216                         return -EINVAL;
6217
6218                 memcpy(settings->ciphers_pairwise, data, len);
6219
6220                 for (i = 0; i < settings->n_ciphers_pairwise; i++)
6221                         if (!cfg80211_supported_cipher_suite(
6222                                         &rdev->wiphy,
6223                                         settings->ciphers_pairwise[i]))
6224                                 return -EINVAL;
6225         }
6226
6227         if (info->attrs[NL80211_ATTR_CIPHER_SUITE_GROUP]) {
6228                 settings->cipher_group =
6229                         nla_get_u32(info->attrs[NL80211_ATTR_CIPHER_SUITE_GROUP]);
6230                 if (!cfg80211_supported_cipher_suite(&rdev->wiphy,
6231                                                      settings->cipher_group))
6232                         return -EINVAL;
6233         }
6234
6235         if (info->attrs[NL80211_ATTR_WPA_VERSIONS]) {
6236                 settings->wpa_versions =
6237                         nla_get_u32(info->attrs[NL80211_ATTR_WPA_VERSIONS]);
6238                 if (!nl80211_valid_wpa_versions(settings->wpa_versions))
6239                         return -EINVAL;
6240         }
6241
6242         if (info->attrs[NL80211_ATTR_AKM_SUITES]) {
6243                 void *data;
6244                 int len;
6245
6246                 data = nla_data(info->attrs[NL80211_ATTR_AKM_SUITES]);
6247                 len = nla_len(info->attrs[NL80211_ATTR_AKM_SUITES]);
6248                 settings->n_akm_suites = len / sizeof(u32);
6249
6250                 if (len % sizeof(u32))
6251                         return -EINVAL;
6252
6253                 if (settings->n_akm_suites > NL80211_MAX_NR_AKM_SUITES)
6254                         return -EINVAL;
6255
6256                 memcpy(settings->akm_suites, data, len);
6257         }
6258
6259         return 0;
6260 }
6261
6262 static int nl80211_associate(struct sk_buff *skb, struct genl_info *info)
6263 {
6264         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6265         struct net_device *dev = info->user_ptr[1];
6266         struct ieee80211_channel *chan;
6267         struct cfg80211_assoc_request req = {};
6268         const u8 *bssid, *ssid;
6269         int err, ssid_len = 0;
6270
6271         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
6272                 return -EINVAL;
6273
6274         if (!info->attrs[NL80211_ATTR_MAC] ||
6275             !info->attrs[NL80211_ATTR_SSID] ||
6276             !info->attrs[NL80211_ATTR_WIPHY_FREQ])
6277                 return -EINVAL;
6278
6279         if (!rdev->ops->assoc)
6280                 return -EOPNOTSUPP;
6281
6282         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
6283             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
6284                 return -EOPNOTSUPP;
6285
6286         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
6287
6288         chan = ieee80211_get_channel(&rdev->wiphy,
6289                 nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
6290         if (!chan || (chan->flags & IEEE80211_CHAN_DISABLED))
6291                 return -EINVAL;
6292
6293         ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
6294         ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
6295
6296         if (info->attrs[NL80211_ATTR_IE]) {
6297                 req.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
6298                 req.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
6299         }
6300
6301         if (info->attrs[NL80211_ATTR_USE_MFP]) {
6302                 enum nl80211_mfp mfp =
6303                         nla_get_u32(info->attrs[NL80211_ATTR_USE_MFP]);
6304                 if (mfp == NL80211_MFP_REQUIRED)
6305                         req.use_mfp = true;
6306                 else if (mfp != NL80211_MFP_NO)
6307                         return -EINVAL;
6308         }
6309
6310         if (info->attrs[NL80211_ATTR_PREV_BSSID])
6311                 req.prev_bssid = nla_data(info->attrs[NL80211_ATTR_PREV_BSSID]);
6312
6313         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_HT]))
6314                 req.flags |= ASSOC_REQ_DISABLE_HT;
6315
6316         if (info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
6317                 memcpy(&req.ht_capa_mask,
6318                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]),
6319                        sizeof(req.ht_capa_mask));
6320
6321         if (info->attrs[NL80211_ATTR_HT_CAPABILITY]) {
6322                 if (!info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
6323                         return -EINVAL;
6324                 memcpy(&req.ht_capa,
6325                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]),
6326                        sizeof(req.ht_capa));
6327         }
6328
6329         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_VHT]))
6330                 req.flags |= ASSOC_REQ_DISABLE_VHT;
6331
6332         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK])
6333                 memcpy(&req.vht_capa_mask,
6334                        nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK]),
6335                        sizeof(req.vht_capa_mask));
6336
6337         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY]) {
6338                 if (!info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK])
6339                         return -EINVAL;
6340                 memcpy(&req.vht_capa,
6341                        nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY]),
6342                        sizeof(req.vht_capa));
6343         }
6344
6345         err = nl80211_crypto_settings(rdev, info, &req.crypto, 1);
6346         if (!err) {
6347                 wdev_lock(dev->ieee80211_ptr);
6348                 err = cfg80211_mlme_assoc(rdev, dev, chan, bssid,
6349                                           ssid, ssid_len, &req);
6350                 wdev_unlock(dev->ieee80211_ptr);
6351         }
6352
6353         return err;
6354 }
6355
6356 static int nl80211_deauthenticate(struct sk_buff *skb, struct genl_info *info)
6357 {
6358         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6359         struct net_device *dev = info->user_ptr[1];
6360         const u8 *ie = NULL, *bssid;
6361         int ie_len = 0, err;
6362         u16 reason_code;
6363         bool local_state_change;
6364
6365         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
6366                 return -EINVAL;
6367
6368         if (!info->attrs[NL80211_ATTR_MAC])
6369                 return -EINVAL;
6370
6371         if (!info->attrs[NL80211_ATTR_REASON_CODE])
6372                 return -EINVAL;
6373
6374         if (!rdev->ops->deauth)
6375                 return -EOPNOTSUPP;
6376
6377         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
6378             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
6379                 return -EOPNOTSUPP;
6380
6381         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
6382
6383         reason_code = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
6384         if (reason_code == 0) {
6385                 /* Reason Code 0 is reserved */
6386                 return -EINVAL;
6387         }
6388
6389         if (info->attrs[NL80211_ATTR_IE]) {
6390                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
6391                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
6392         }
6393
6394         local_state_change = !!info->attrs[NL80211_ATTR_LOCAL_STATE_CHANGE];
6395
6396         wdev_lock(dev->ieee80211_ptr);
6397         err = cfg80211_mlme_deauth(rdev, dev, bssid, ie, ie_len, reason_code,
6398                                    local_state_change);
6399         wdev_unlock(dev->ieee80211_ptr);
6400         return err;
6401 }
6402
6403 static int nl80211_disassociate(struct sk_buff *skb, struct genl_info *info)
6404 {
6405         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6406         struct net_device *dev = info->user_ptr[1];
6407         const u8 *ie = NULL, *bssid;
6408         int ie_len = 0, err;
6409         u16 reason_code;
6410         bool local_state_change;
6411
6412         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
6413                 return -EINVAL;
6414
6415         if (!info->attrs[NL80211_ATTR_MAC])
6416                 return -EINVAL;
6417
6418         if (!info->attrs[NL80211_ATTR_REASON_CODE])
6419                 return -EINVAL;
6420
6421         if (!rdev->ops->disassoc)
6422                 return -EOPNOTSUPP;
6423
6424         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
6425             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
6426                 return -EOPNOTSUPP;
6427
6428         bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
6429
6430         reason_code = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
6431         if (reason_code == 0) {
6432                 /* Reason Code 0 is reserved */
6433                 return -EINVAL;
6434         }
6435
6436         if (info->attrs[NL80211_ATTR_IE]) {
6437                 ie = nla_data(info->attrs[NL80211_ATTR_IE]);
6438                 ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
6439         }
6440
6441         local_state_change = !!info->attrs[NL80211_ATTR_LOCAL_STATE_CHANGE];
6442
6443         wdev_lock(dev->ieee80211_ptr);
6444         err = cfg80211_mlme_disassoc(rdev, dev, bssid, ie, ie_len, reason_code,
6445                                      local_state_change);
6446         wdev_unlock(dev->ieee80211_ptr);
6447         return err;
6448 }
6449
6450 static bool
6451 nl80211_parse_mcast_rate(struct cfg80211_registered_device *rdev,
6452                          int mcast_rate[IEEE80211_NUM_BANDS],
6453                          int rateval)
6454 {
6455         struct wiphy *wiphy = &rdev->wiphy;
6456         bool found = false;
6457         int band, i;
6458
6459         for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
6460                 struct ieee80211_supported_band *sband;
6461
6462                 sband = wiphy->bands[band];
6463                 if (!sband)
6464                         continue;
6465
6466                 for (i = 0; i < sband->n_bitrates; i++) {
6467                         if (sband->bitrates[i].bitrate == rateval) {
6468                                 mcast_rate[band] = i + 1;
6469                                 found = true;
6470                                 break;
6471                         }
6472                 }
6473         }
6474
6475         return found;
6476 }
6477
6478 static int nl80211_join_ibss(struct sk_buff *skb, struct genl_info *info)
6479 {
6480         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6481         struct net_device *dev = info->user_ptr[1];
6482         struct cfg80211_ibss_params ibss;
6483         struct wiphy *wiphy;
6484         struct cfg80211_cached_keys *connkeys = NULL;
6485         int err;
6486
6487         memset(&ibss, 0, sizeof(ibss));
6488
6489         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
6490                 return -EINVAL;
6491
6492         if (!info->attrs[NL80211_ATTR_SSID] ||
6493             !nla_len(info->attrs[NL80211_ATTR_SSID]))
6494                 return -EINVAL;
6495
6496         ibss.beacon_interval = 100;
6497
6498         if (info->attrs[NL80211_ATTR_BEACON_INTERVAL]) {
6499                 ibss.beacon_interval =
6500                         nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
6501                 if (ibss.beacon_interval < 1 || ibss.beacon_interval > 10000)
6502                         return -EINVAL;
6503         }
6504
6505         if (!rdev->ops->join_ibss)
6506                 return -EOPNOTSUPP;
6507
6508         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC)
6509                 return -EOPNOTSUPP;
6510
6511         wiphy = &rdev->wiphy;
6512
6513         if (info->attrs[NL80211_ATTR_MAC]) {
6514                 ibss.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
6515
6516                 if (!is_valid_ether_addr(ibss.bssid))
6517                         return -EINVAL;
6518         }
6519         ibss.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
6520         ibss.ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
6521
6522         if (info->attrs[NL80211_ATTR_IE]) {
6523                 ibss.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
6524                 ibss.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
6525         }
6526
6527         err = nl80211_parse_chandef(rdev, info, &ibss.chandef);
6528         if (err)
6529                 return err;
6530
6531         if (!cfg80211_reg_can_beacon(&rdev->wiphy, &ibss.chandef))
6532                 return -EINVAL;
6533
6534         switch (ibss.chandef.width) {
6535         case NL80211_CHAN_WIDTH_5:
6536         case NL80211_CHAN_WIDTH_10:
6537         case NL80211_CHAN_WIDTH_20_NOHT:
6538                 break;
6539         case NL80211_CHAN_WIDTH_20:
6540         case NL80211_CHAN_WIDTH_40:
6541                 if (rdev->wiphy.features & NL80211_FEATURE_HT_IBSS)
6542                         break;
6543         default:
6544                 return -EINVAL;
6545         }
6546
6547         ibss.channel_fixed = !!info->attrs[NL80211_ATTR_FREQ_FIXED];
6548         ibss.privacy = !!info->attrs[NL80211_ATTR_PRIVACY];
6549
6550         if (info->attrs[NL80211_ATTR_BSS_BASIC_RATES]) {
6551                 u8 *rates =
6552                         nla_data(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
6553                 int n_rates =
6554                         nla_len(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
6555                 struct ieee80211_supported_band *sband =
6556                         wiphy->bands[ibss.chandef.chan->band];
6557
6558                 err = ieee80211_get_ratemask(sband, rates, n_rates,
6559                                              &ibss.basic_rates);
6560                 if (err)
6561                         return err;
6562         }
6563
6564         if (info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
6565                 memcpy(&ibss.ht_capa_mask,
6566                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]),
6567                        sizeof(ibss.ht_capa_mask));
6568
6569         if (info->attrs[NL80211_ATTR_HT_CAPABILITY]) {
6570                 if (!info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
6571                         return -EINVAL;
6572                 memcpy(&ibss.ht_capa,
6573                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]),
6574                        sizeof(ibss.ht_capa));
6575         }
6576
6577         if (info->attrs[NL80211_ATTR_MCAST_RATE] &&
6578             !nl80211_parse_mcast_rate(rdev, ibss.mcast_rate,
6579                         nla_get_u32(info->attrs[NL80211_ATTR_MCAST_RATE])))
6580                 return -EINVAL;
6581
6582         if (ibss.privacy && info->attrs[NL80211_ATTR_KEYS]) {
6583                 bool no_ht = false;
6584
6585                 connkeys = nl80211_parse_connkeys(rdev,
6586                                           info->attrs[NL80211_ATTR_KEYS],
6587                                           &no_ht);
6588                 if (IS_ERR(connkeys))
6589                         return PTR_ERR(connkeys);
6590
6591                 if ((ibss.chandef.width != NL80211_CHAN_WIDTH_20_NOHT) &&
6592                     no_ht) {
6593                         kfree(connkeys);
6594                         return -EINVAL;
6595                 }
6596         }
6597
6598         ibss.control_port =
6599                 nla_get_flag(info->attrs[NL80211_ATTR_CONTROL_PORT]);
6600
6601         ibss.userspace_handles_dfs =
6602                 nla_get_flag(info->attrs[NL80211_ATTR_HANDLE_DFS]);
6603
6604         err = cfg80211_join_ibss(rdev, dev, &ibss, connkeys);
6605         if (err)
6606                 kfree(connkeys);
6607         return err;
6608 }
6609
6610 static int nl80211_leave_ibss(struct sk_buff *skb, struct genl_info *info)
6611 {
6612         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6613         struct net_device *dev = info->user_ptr[1];
6614
6615         if (!rdev->ops->leave_ibss)
6616                 return -EOPNOTSUPP;
6617
6618         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC)
6619                 return -EOPNOTSUPP;
6620
6621         return cfg80211_leave_ibss(rdev, dev, false);
6622 }
6623
6624 static int nl80211_set_mcast_rate(struct sk_buff *skb, struct genl_info *info)
6625 {
6626         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6627         struct net_device *dev = info->user_ptr[1];
6628         int mcast_rate[IEEE80211_NUM_BANDS];
6629         u32 nla_rate;
6630         int err;
6631
6632         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_ADHOC &&
6633             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_MESH_POINT)
6634                 return -EOPNOTSUPP;
6635
6636         if (!rdev->ops->set_mcast_rate)
6637                 return -EOPNOTSUPP;
6638
6639         memset(mcast_rate, 0, sizeof(mcast_rate));
6640
6641         if (!info->attrs[NL80211_ATTR_MCAST_RATE])
6642                 return -EINVAL;
6643
6644         nla_rate = nla_get_u32(info->attrs[NL80211_ATTR_MCAST_RATE]);
6645         if (!nl80211_parse_mcast_rate(rdev, mcast_rate, nla_rate))
6646                 return -EINVAL;
6647
6648         err = rdev->ops->set_mcast_rate(&rdev->wiphy, dev, mcast_rate);
6649
6650         return err;
6651 }
6652
6653
6654 #ifdef CONFIG_NL80211_TESTMODE
6655 static struct genl_multicast_group nl80211_testmode_mcgrp = {
6656         .name = "testmode",
6657 };
6658
6659 static int nl80211_testmode_do(struct sk_buff *skb, struct genl_info *info)
6660 {
6661         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6662         struct wireless_dev *wdev =
6663                 __cfg80211_wdev_from_attrs(genl_info_net(info), info->attrs);
6664         int err;
6665
6666         if (!rdev->ops->testmode_cmd)
6667                 return -EOPNOTSUPP;
6668
6669         if (IS_ERR(wdev)) {
6670                 err = PTR_ERR(wdev);
6671                 if (err != -EINVAL)
6672                         return err;
6673                 wdev = NULL;
6674         } else if (wdev->wiphy != &rdev->wiphy) {
6675                 return -EINVAL;
6676         }
6677
6678         if (!info->attrs[NL80211_ATTR_TESTDATA])
6679                 return -EINVAL;
6680
6681         rdev->testmode_info = info;
6682         err = rdev_testmode_cmd(rdev, wdev,
6683                                 nla_data(info->attrs[NL80211_ATTR_TESTDATA]),
6684                                 nla_len(info->attrs[NL80211_ATTR_TESTDATA]));
6685         rdev->testmode_info = NULL;
6686
6687         return err;
6688 }
6689
6690 static int nl80211_testmode_dump(struct sk_buff *skb,
6691                                  struct netlink_callback *cb)
6692 {
6693         struct cfg80211_registered_device *rdev;
6694         int err;
6695         long phy_idx;
6696         void *data = NULL;
6697         int data_len = 0;
6698
6699         rtnl_lock();
6700
6701         if (cb->args[0]) {
6702                 /*
6703                  * 0 is a valid index, but not valid for args[0],
6704                  * so we need to offset by 1.
6705                  */
6706                 phy_idx = cb->args[0] - 1;
6707         } else {
6708                 err = nlmsg_parse(cb->nlh, GENL_HDRLEN + nl80211_fam.hdrsize,
6709                                   nl80211_fam.attrbuf, nl80211_fam.maxattr,
6710                                   nl80211_policy);
6711                 if (err)
6712                         goto out_err;
6713
6714                 rdev = __cfg80211_rdev_from_attrs(sock_net(skb->sk),
6715                                                   nl80211_fam.attrbuf);
6716                 if (IS_ERR(rdev)) {
6717                         err = PTR_ERR(rdev);
6718                         goto out_err;
6719                 }
6720                 phy_idx = rdev->wiphy_idx;
6721                 rdev = NULL;
6722
6723                 if (nl80211_fam.attrbuf[NL80211_ATTR_TESTDATA])
6724                         cb->args[1] =
6725                                 (long)nl80211_fam.attrbuf[NL80211_ATTR_TESTDATA];
6726         }
6727
6728         if (cb->args[1]) {
6729                 data = nla_data((void *)cb->args[1]);
6730                 data_len = nla_len((void *)cb->args[1]);
6731         }
6732
6733         rdev = cfg80211_rdev_by_wiphy_idx(phy_idx);
6734         if (!rdev) {
6735                 err = -ENOENT;
6736                 goto out_err;
6737         }
6738
6739         if (!rdev->ops->testmode_dump) {
6740                 err = -EOPNOTSUPP;
6741                 goto out_err;
6742         }
6743
6744         while (1) {
6745                 void *hdr = nl80211hdr_put(skb, NETLINK_CB(cb->skb).portid,
6746                                            cb->nlh->nlmsg_seq, NLM_F_MULTI,
6747                                            NL80211_CMD_TESTMODE);
6748                 struct nlattr *tmdata;
6749
6750                 if (!hdr)
6751                         break;
6752
6753                 if (nla_put_u32(skb, NL80211_ATTR_WIPHY, phy_idx)) {
6754                         genlmsg_cancel(skb, hdr);
6755                         break;
6756                 }
6757
6758                 tmdata = nla_nest_start(skb, NL80211_ATTR_TESTDATA);
6759                 if (!tmdata) {
6760                         genlmsg_cancel(skb, hdr);
6761                         break;
6762                 }
6763                 err = rdev_testmode_dump(rdev, skb, cb, data, data_len);
6764                 nla_nest_end(skb, tmdata);
6765
6766                 if (err == -ENOBUFS || err == -ENOENT) {
6767                         genlmsg_cancel(skb, hdr);
6768                         break;
6769                 } else if (err) {
6770                         genlmsg_cancel(skb, hdr);
6771                         goto out_err;
6772                 }
6773
6774                 genlmsg_end(skb, hdr);
6775         }
6776
6777         err = skb->len;
6778         /* see above */
6779         cb->args[0] = phy_idx + 1;
6780  out_err:
6781         rtnl_unlock();
6782         return err;
6783 }
6784
6785 static struct sk_buff *
6786 __cfg80211_testmode_alloc_skb(struct cfg80211_registered_device *rdev,
6787                               int approxlen, u32 portid, u32 seq, gfp_t gfp)
6788 {
6789         struct sk_buff *skb;
6790         void *hdr;
6791         struct nlattr *data;
6792
6793         skb = nlmsg_new(approxlen + 100, gfp);
6794         if (!skb)
6795                 return NULL;
6796
6797         hdr = nl80211hdr_put(skb, portid, seq, 0, NL80211_CMD_TESTMODE);
6798         if (!hdr) {
6799                 kfree_skb(skb);
6800                 return NULL;
6801         }
6802
6803         if (nla_put_u32(skb, NL80211_ATTR_WIPHY, rdev->wiphy_idx))
6804                 goto nla_put_failure;
6805         data = nla_nest_start(skb, NL80211_ATTR_TESTDATA);
6806
6807         ((void **)skb->cb)[0] = rdev;
6808         ((void **)skb->cb)[1] = hdr;
6809         ((void **)skb->cb)[2] = data;
6810
6811         return skb;
6812
6813  nla_put_failure:
6814         kfree_skb(skb);
6815         return NULL;
6816 }
6817
6818 struct sk_buff *cfg80211_testmode_alloc_reply_skb(struct wiphy *wiphy,
6819                                                   int approxlen)
6820 {
6821         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
6822
6823         if (WARN_ON(!rdev->testmode_info))
6824                 return NULL;
6825
6826         return __cfg80211_testmode_alloc_skb(rdev, approxlen,
6827                                 rdev->testmode_info->snd_portid,
6828                                 rdev->testmode_info->snd_seq,
6829                                 GFP_KERNEL);
6830 }
6831 EXPORT_SYMBOL(cfg80211_testmode_alloc_reply_skb);
6832
6833 int cfg80211_testmode_reply(struct sk_buff *skb)
6834 {
6835         struct cfg80211_registered_device *rdev = ((void **)skb->cb)[0];
6836         void *hdr = ((void **)skb->cb)[1];
6837         struct nlattr *data = ((void **)skb->cb)[2];
6838
6839         if (WARN_ON(!rdev->testmode_info)) {
6840                 kfree_skb(skb);
6841                 return -EINVAL;
6842         }
6843
6844         nla_nest_end(skb, data);
6845         genlmsg_end(skb, hdr);
6846         return genlmsg_reply(skb, rdev->testmode_info);
6847 }
6848 EXPORT_SYMBOL(cfg80211_testmode_reply);
6849
6850 struct sk_buff *cfg80211_testmode_alloc_event_skb(struct wiphy *wiphy,
6851                                                   int approxlen, gfp_t gfp)
6852 {
6853         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
6854
6855         return __cfg80211_testmode_alloc_skb(rdev, approxlen, 0, 0, gfp);
6856 }
6857 EXPORT_SYMBOL(cfg80211_testmode_alloc_event_skb);
6858
6859 void cfg80211_testmode_event(struct sk_buff *skb, gfp_t gfp)
6860 {
6861         struct cfg80211_registered_device *rdev = ((void **)skb->cb)[0];
6862         void *hdr = ((void **)skb->cb)[1];
6863         struct nlattr *data = ((void **)skb->cb)[2];
6864
6865         nla_nest_end(skb, data);
6866         genlmsg_end(skb, hdr);
6867         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), skb, 0,
6868                                 nl80211_testmode_mcgrp.id, gfp);
6869 }
6870 EXPORT_SYMBOL(cfg80211_testmode_event);
6871 #endif
6872
6873 static int nl80211_connect(struct sk_buff *skb, struct genl_info *info)
6874 {
6875         struct cfg80211_registered_device *rdev = info->user_ptr[0];
6876         struct net_device *dev = info->user_ptr[1];
6877         struct cfg80211_connect_params connect;
6878         struct wiphy *wiphy;
6879         struct cfg80211_cached_keys *connkeys = NULL;
6880         int err;
6881
6882         memset(&connect, 0, sizeof(connect));
6883
6884         if (!is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
6885                 return -EINVAL;
6886
6887         if (!info->attrs[NL80211_ATTR_SSID] ||
6888             !nla_len(info->attrs[NL80211_ATTR_SSID]))
6889                 return -EINVAL;
6890
6891         if (info->attrs[NL80211_ATTR_AUTH_TYPE]) {
6892                 connect.auth_type =
6893                         nla_get_u32(info->attrs[NL80211_ATTR_AUTH_TYPE]);
6894                 if (!nl80211_valid_auth_type(rdev, connect.auth_type,
6895                                              NL80211_CMD_CONNECT))
6896                         return -EINVAL;
6897         } else
6898                 connect.auth_type = NL80211_AUTHTYPE_AUTOMATIC;
6899
6900         connect.privacy = info->attrs[NL80211_ATTR_PRIVACY];
6901
6902         err = nl80211_crypto_settings(rdev, info, &connect.crypto,
6903                                       NL80211_MAX_NR_CIPHER_SUITES);
6904         if (err)
6905                 return err;
6906
6907         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
6908             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
6909                 return -EOPNOTSUPP;
6910
6911         wiphy = &rdev->wiphy;
6912
6913         connect.bg_scan_period = -1;
6914         if (info->attrs[NL80211_ATTR_BG_SCAN_PERIOD] &&
6915                 (wiphy->flags & WIPHY_FLAG_SUPPORTS_FW_ROAM)) {
6916                 connect.bg_scan_period =
6917                         nla_get_u16(info->attrs[NL80211_ATTR_BG_SCAN_PERIOD]);
6918         }
6919
6920         if (info->attrs[NL80211_ATTR_MAC])
6921                 connect.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
6922         connect.ssid = nla_data(info->attrs[NL80211_ATTR_SSID]);
6923         connect.ssid_len = nla_len(info->attrs[NL80211_ATTR_SSID]);
6924
6925         if (info->attrs[NL80211_ATTR_IE]) {
6926                 connect.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
6927                 connect.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
6928         }
6929
6930         if (info->attrs[NL80211_ATTR_USE_MFP]) {
6931                 connect.mfp = nla_get_u32(info->attrs[NL80211_ATTR_USE_MFP]);
6932                 if (connect.mfp != NL80211_MFP_REQUIRED &&
6933                     connect.mfp != NL80211_MFP_NO)
6934                         return -EINVAL;
6935         } else {
6936                 connect.mfp = NL80211_MFP_NO;
6937         }
6938
6939         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
6940                 connect.channel =
6941                         ieee80211_get_channel(wiphy,
6942                             nla_get_u32(info->attrs[NL80211_ATTR_WIPHY_FREQ]));
6943                 if (!connect.channel ||
6944                     connect.channel->flags & IEEE80211_CHAN_DISABLED)
6945                         return -EINVAL;
6946         }
6947
6948         if (connect.privacy && info->attrs[NL80211_ATTR_KEYS]) {
6949                 connkeys = nl80211_parse_connkeys(rdev,
6950                                           info->attrs[NL80211_ATTR_KEYS], NULL);
6951                 if (IS_ERR(connkeys))
6952                         return PTR_ERR(connkeys);
6953         }
6954
6955         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_HT]))
6956                 connect.flags |= ASSOC_REQ_DISABLE_HT;
6957
6958         if (info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK])
6959                 memcpy(&connect.ht_capa_mask,
6960                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]),
6961                        sizeof(connect.ht_capa_mask));
6962
6963         if (info->attrs[NL80211_ATTR_HT_CAPABILITY]) {
6964                 if (!info->attrs[NL80211_ATTR_HT_CAPABILITY_MASK]) {
6965                         kfree(connkeys);
6966                         return -EINVAL;
6967                 }
6968                 memcpy(&connect.ht_capa,
6969                        nla_data(info->attrs[NL80211_ATTR_HT_CAPABILITY]),
6970                        sizeof(connect.ht_capa));
6971         }
6972
6973         if (nla_get_flag(info->attrs[NL80211_ATTR_DISABLE_VHT]))
6974                 connect.flags |= ASSOC_REQ_DISABLE_VHT;
6975
6976         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK])
6977                 memcpy(&connect.vht_capa_mask,
6978                        nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK]),
6979                        sizeof(connect.vht_capa_mask));
6980
6981         if (info->attrs[NL80211_ATTR_VHT_CAPABILITY]) {
6982                 if (!info->attrs[NL80211_ATTR_VHT_CAPABILITY_MASK]) {
6983                         kfree(connkeys);
6984                         return -EINVAL;
6985                 }
6986                 memcpy(&connect.vht_capa,
6987                        nla_data(info->attrs[NL80211_ATTR_VHT_CAPABILITY]),
6988                        sizeof(connect.vht_capa));
6989         }
6990
6991         wdev_lock(dev->ieee80211_ptr);
6992         err = cfg80211_connect(rdev, dev, &connect, connkeys, NULL);
6993         wdev_unlock(dev->ieee80211_ptr);
6994         if (err)
6995                 kfree(connkeys);
6996         return err;
6997 }
6998
6999 static int nl80211_disconnect(struct sk_buff *skb, struct genl_info *info)
7000 {
7001         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7002         struct net_device *dev = info->user_ptr[1];
7003         u16 reason;
7004         int ret;
7005
7006         if (!info->attrs[NL80211_ATTR_REASON_CODE])
7007                 reason = WLAN_REASON_DEAUTH_LEAVING;
7008         else
7009                 reason = nla_get_u16(info->attrs[NL80211_ATTR_REASON_CODE]);
7010
7011         if (reason == 0)
7012                 return -EINVAL;
7013
7014         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
7015             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
7016                 return -EOPNOTSUPP;
7017
7018         wdev_lock(dev->ieee80211_ptr);
7019         ret = cfg80211_disconnect(rdev, dev, reason, true);
7020         wdev_unlock(dev->ieee80211_ptr);
7021         return ret;
7022 }
7023
7024 static int nl80211_wiphy_netns(struct sk_buff *skb, struct genl_info *info)
7025 {
7026         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7027         struct net *net;
7028         int err;
7029         u32 pid;
7030
7031         if (!info->attrs[NL80211_ATTR_PID])
7032                 return -EINVAL;
7033
7034         pid = nla_get_u32(info->attrs[NL80211_ATTR_PID]);
7035
7036         net = get_net_ns_by_pid(pid);
7037         if (IS_ERR(net))
7038                 return PTR_ERR(net);
7039
7040         err = 0;
7041
7042         /* check if anything to do */
7043         if (!net_eq(wiphy_net(&rdev->wiphy), net))
7044                 err = cfg80211_switch_netns(rdev, net);
7045
7046         put_net(net);
7047         return err;
7048 }
7049
7050 static int nl80211_setdel_pmksa(struct sk_buff *skb, struct genl_info *info)
7051 {
7052         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7053         int (*rdev_ops)(struct wiphy *wiphy, struct net_device *dev,
7054                         struct cfg80211_pmksa *pmksa) = NULL;
7055         struct net_device *dev = info->user_ptr[1];
7056         struct cfg80211_pmksa pmksa;
7057
7058         memset(&pmksa, 0, sizeof(struct cfg80211_pmksa));
7059
7060         if (!info->attrs[NL80211_ATTR_MAC])
7061                 return -EINVAL;
7062
7063         if (!info->attrs[NL80211_ATTR_PMKID])
7064                 return -EINVAL;
7065
7066         pmksa.pmkid = nla_data(info->attrs[NL80211_ATTR_PMKID]);
7067         pmksa.bssid = nla_data(info->attrs[NL80211_ATTR_MAC]);
7068
7069         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
7070             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
7071                 return -EOPNOTSUPP;
7072
7073         switch (info->genlhdr->cmd) {
7074         case NL80211_CMD_SET_PMKSA:
7075                 rdev_ops = rdev->ops->set_pmksa;
7076                 break;
7077         case NL80211_CMD_DEL_PMKSA:
7078                 rdev_ops = rdev->ops->del_pmksa;
7079                 break;
7080         default:
7081                 WARN_ON(1);
7082                 break;
7083         }
7084
7085         if (!rdev_ops)
7086                 return -EOPNOTSUPP;
7087
7088         return rdev_ops(&rdev->wiphy, dev, &pmksa);
7089 }
7090
7091 static int nl80211_flush_pmksa(struct sk_buff *skb, struct genl_info *info)
7092 {
7093         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7094         struct net_device *dev = info->user_ptr[1];
7095
7096         if (dev->ieee80211_ptr->iftype != NL80211_IFTYPE_STATION &&
7097             dev->ieee80211_ptr->iftype != NL80211_IFTYPE_P2P_CLIENT)
7098                 return -EOPNOTSUPP;
7099
7100         if (!rdev->ops->flush_pmksa)
7101                 return -EOPNOTSUPP;
7102
7103         return rdev_flush_pmksa(rdev, dev);
7104 }
7105
7106 static int nl80211_tdls_mgmt(struct sk_buff *skb, struct genl_info *info)
7107 {
7108         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7109         struct net_device *dev = info->user_ptr[1];
7110         u8 action_code, dialog_token;
7111         u16 status_code;
7112         u8 *peer;
7113
7114         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) ||
7115             !rdev->ops->tdls_mgmt)
7116                 return -EOPNOTSUPP;
7117
7118         if (!info->attrs[NL80211_ATTR_TDLS_ACTION] ||
7119             !info->attrs[NL80211_ATTR_STATUS_CODE] ||
7120             !info->attrs[NL80211_ATTR_TDLS_DIALOG_TOKEN] ||
7121             !info->attrs[NL80211_ATTR_IE] ||
7122             !info->attrs[NL80211_ATTR_MAC])
7123                 return -EINVAL;
7124
7125         peer = nla_data(info->attrs[NL80211_ATTR_MAC]);
7126         action_code = nla_get_u8(info->attrs[NL80211_ATTR_TDLS_ACTION]);
7127         status_code = nla_get_u16(info->attrs[NL80211_ATTR_STATUS_CODE]);
7128         dialog_token = nla_get_u8(info->attrs[NL80211_ATTR_TDLS_DIALOG_TOKEN]);
7129
7130         return rdev_tdls_mgmt(rdev, dev, peer, action_code,
7131                               dialog_token, status_code,
7132                               nla_data(info->attrs[NL80211_ATTR_IE]),
7133                               nla_len(info->attrs[NL80211_ATTR_IE]));
7134 }
7135
7136 static int nl80211_tdls_oper(struct sk_buff *skb, struct genl_info *info)
7137 {
7138         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7139         struct net_device *dev = info->user_ptr[1];
7140         enum nl80211_tdls_operation operation;
7141         u8 *peer;
7142
7143         if (!(rdev->wiphy.flags & WIPHY_FLAG_SUPPORTS_TDLS) ||
7144             !rdev->ops->tdls_oper)
7145                 return -EOPNOTSUPP;
7146
7147         if (!info->attrs[NL80211_ATTR_TDLS_OPERATION] ||
7148             !info->attrs[NL80211_ATTR_MAC])
7149                 return -EINVAL;
7150
7151         operation = nla_get_u8(info->attrs[NL80211_ATTR_TDLS_OPERATION]);
7152         peer = nla_data(info->attrs[NL80211_ATTR_MAC]);
7153
7154         return rdev_tdls_oper(rdev, dev, peer, operation);
7155 }
7156
7157 static int nl80211_remain_on_channel(struct sk_buff *skb,
7158                                      struct genl_info *info)
7159 {
7160         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7161         struct wireless_dev *wdev = info->user_ptr[1];
7162         struct cfg80211_chan_def chandef;
7163         struct sk_buff *msg;
7164         void *hdr;
7165         u64 cookie;
7166         u32 duration;
7167         int err;
7168
7169         if (!info->attrs[NL80211_ATTR_WIPHY_FREQ] ||
7170             !info->attrs[NL80211_ATTR_DURATION])
7171                 return -EINVAL;
7172
7173         duration = nla_get_u32(info->attrs[NL80211_ATTR_DURATION]);
7174
7175         if (!rdev->ops->remain_on_channel ||
7176             !(rdev->wiphy.flags & WIPHY_FLAG_HAS_REMAIN_ON_CHANNEL))
7177                 return -EOPNOTSUPP;
7178
7179         /*
7180          * We should be on that channel for at least a minimum amount of
7181          * time (10ms) but no longer than the driver supports.
7182          */
7183         if (duration < NL80211_MIN_REMAIN_ON_CHANNEL_TIME ||
7184             duration > rdev->wiphy.max_remain_on_channel_duration)
7185                 return -EINVAL;
7186
7187         err = nl80211_parse_chandef(rdev, info, &chandef);
7188         if (err)
7189                 return err;
7190
7191         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
7192         if (!msg)
7193                 return -ENOMEM;
7194
7195         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
7196                              NL80211_CMD_REMAIN_ON_CHANNEL);
7197         if (!hdr) {
7198                 err = -ENOBUFS;
7199                 goto free_msg;
7200         }
7201
7202         err = rdev_remain_on_channel(rdev, wdev, chandef.chan,
7203                                      duration, &cookie);
7204
7205         if (err)
7206                 goto free_msg;
7207
7208         if (nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie))
7209                 goto nla_put_failure;
7210
7211         genlmsg_end(msg, hdr);
7212
7213         return genlmsg_reply(msg, info);
7214
7215  nla_put_failure:
7216         err = -ENOBUFS;
7217  free_msg:
7218         nlmsg_free(msg);
7219         return err;
7220 }
7221
7222 static int nl80211_cancel_remain_on_channel(struct sk_buff *skb,
7223                                             struct genl_info *info)
7224 {
7225         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7226         struct wireless_dev *wdev = info->user_ptr[1];
7227         u64 cookie;
7228
7229         if (!info->attrs[NL80211_ATTR_COOKIE])
7230                 return -EINVAL;
7231
7232         if (!rdev->ops->cancel_remain_on_channel)
7233                 return -EOPNOTSUPP;
7234
7235         cookie = nla_get_u64(info->attrs[NL80211_ATTR_COOKIE]);
7236
7237         return rdev_cancel_remain_on_channel(rdev, wdev, cookie);
7238 }
7239
7240 static u32 rateset_to_mask(struct ieee80211_supported_band *sband,
7241                            u8 *rates, u8 rates_len)
7242 {
7243         u8 i;
7244         u32 mask = 0;
7245
7246         for (i = 0; i < rates_len; i++) {
7247                 int rate = (rates[i] & 0x7f) * 5;
7248                 int ridx;
7249                 for (ridx = 0; ridx < sband->n_bitrates; ridx++) {
7250                         struct ieee80211_rate *srate =
7251                                 &sband->bitrates[ridx];
7252                         if (rate == srate->bitrate) {
7253                                 mask |= 1 << ridx;
7254                                 break;
7255                         }
7256                 }
7257                 if (ridx == sband->n_bitrates)
7258                         return 0; /* rate not found */
7259         }
7260
7261         return mask;
7262 }
7263
7264 static bool ht_rateset_to_mask(struct ieee80211_supported_band *sband,
7265                                u8 *rates, u8 rates_len,
7266                                u8 mcs[IEEE80211_HT_MCS_MASK_LEN])
7267 {
7268         u8 i;
7269
7270         memset(mcs, 0, IEEE80211_HT_MCS_MASK_LEN);
7271
7272         for (i = 0; i < rates_len; i++) {
7273                 int ridx, rbit;
7274
7275                 ridx = rates[i] / 8;
7276                 rbit = BIT(rates[i] % 8);
7277
7278                 /* check validity */
7279                 if ((ridx < 0) || (ridx >= IEEE80211_HT_MCS_MASK_LEN))
7280                         return false;
7281
7282                 /* check availability */
7283                 if (sband->ht_cap.mcs.rx_mask[ridx] & rbit)
7284                         mcs[ridx] |= rbit;
7285                 else
7286                         return false;
7287         }
7288
7289         return true;
7290 }
7291
7292 static const struct nla_policy nl80211_txattr_policy[NL80211_TXRATE_MAX + 1] = {
7293         [NL80211_TXRATE_LEGACY] = { .type = NLA_BINARY,
7294                                     .len = NL80211_MAX_SUPP_RATES },
7295         [NL80211_TXRATE_MCS] = { .type = NLA_BINARY,
7296                                  .len = NL80211_MAX_SUPP_HT_RATES },
7297 };
7298
7299 static int nl80211_set_tx_bitrate_mask(struct sk_buff *skb,
7300                                        struct genl_info *info)
7301 {
7302         struct nlattr *tb[NL80211_TXRATE_MAX + 1];
7303         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7304         struct cfg80211_bitrate_mask mask;
7305         int rem, i;
7306         struct net_device *dev = info->user_ptr[1];
7307         struct nlattr *tx_rates;
7308         struct ieee80211_supported_band *sband;
7309
7310         if (info->attrs[NL80211_ATTR_TX_RATES] == NULL)
7311                 return -EINVAL;
7312
7313         if (!rdev->ops->set_bitrate_mask)
7314                 return -EOPNOTSUPP;
7315
7316         memset(&mask, 0, sizeof(mask));
7317         /* Default to all rates enabled */
7318         for (i = 0; i < IEEE80211_NUM_BANDS; i++) {
7319                 sband = rdev->wiphy.bands[i];
7320                 mask.control[i].legacy =
7321                         sband ? (1 << sband->n_bitrates) - 1 : 0;
7322                 if (sband)
7323                         memcpy(mask.control[i].mcs,
7324                                sband->ht_cap.mcs.rx_mask,
7325                                sizeof(mask.control[i].mcs));
7326                 else
7327                         memset(mask.control[i].mcs, 0,
7328                                sizeof(mask.control[i].mcs));
7329         }
7330
7331         /*
7332          * The nested attribute uses enum nl80211_band as the index. This maps
7333          * directly to the enum ieee80211_band values used in cfg80211.
7334          */
7335         BUILD_BUG_ON(NL80211_MAX_SUPP_HT_RATES > IEEE80211_HT_MCS_MASK_LEN * 8);
7336         nla_for_each_nested(tx_rates, info->attrs[NL80211_ATTR_TX_RATES], rem)
7337         {
7338                 enum ieee80211_band band = nla_type(tx_rates);
7339                 if (band < 0 || band >= IEEE80211_NUM_BANDS)
7340                         return -EINVAL;
7341                 sband = rdev->wiphy.bands[band];
7342                 if (sband == NULL)
7343                         return -EINVAL;
7344                 nla_parse(tb, NL80211_TXRATE_MAX, nla_data(tx_rates),
7345                           nla_len(tx_rates), nl80211_txattr_policy);
7346                 if (tb[NL80211_TXRATE_LEGACY]) {
7347                         mask.control[band].legacy = rateset_to_mask(
7348                                 sband,
7349                                 nla_data(tb[NL80211_TXRATE_LEGACY]),
7350                                 nla_len(tb[NL80211_TXRATE_LEGACY]));
7351                         if ((mask.control[band].legacy == 0) &&
7352                             nla_len(tb[NL80211_TXRATE_LEGACY]))
7353                                 return -EINVAL;
7354                 }
7355                 if (tb[NL80211_TXRATE_MCS]) {
7356                         if (!ht_rateset_to_mask(
7357                                         sband,
7358                                         nla_data(tb[NL80211_TXRATE_MCS]),
7359                                         nla_len(tb[NL80211_TXRATE_MCS]),
7360                                         mask.control[band].mcs))
7361                                 return -EINVAL;
7362                 }
7363
7364                 if (mask.control[band].legacy == 0) {
7365                         /* don't allow empty legacy rates if HT
7366                          * is not even supported. */
7367                         if (!rdev->wiphy.bands[band]->ht_cap.ht_supported)
7368                                 return -EINVAL;
7369
7370                         for (i = 0; i < IEEE80211_HT_MCS_MASK_LEN; i++)
7371                                 if (mask.control[band].mcs[i])
7372                                         break;
7373
7374                         /* legacy and mcs rates may not be both empty */
7375                         if (i == IEEE80211_HT_MCS_MASK_LEN)
7376                                 return -EINVAL;
7377                 }
7378         }
7379
7380         return rdev_set_bitrate_mask(rdev, dev, NULL, &mask);
7381 }
7382
7383 static int nl80211_register_mgmt(struct sk_buff *skb, struct genl_info *info)
7384 {
7385         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7386         struct wireless_dev *wdev = info->user_ptr[1];
7387         u16 frame_type = IEEE80211_FTYPE_MGMT | IEEE80211_STYPE_ACTION;
7388
7389         if (!info->attrs[NL80211_ATTR_FRAME_MATCH])
7390                 return -EINVAL;
7391
7392         if (info->attrs[NL80211_ATTR_FRAME_TYPE])
7393                 frame_type = nla_get_u16(info->attrs[NL80211_ATTR_FRAME_TYPE]);
7394
7395         switch (wdev->iftype) {
7396         case NL80211_IFTYPE_STATION:
7397         case NL80211_IFTYPE_ADHOC:
7398         case NL80211_IFTYPE_P2P_CLIENT:
7399         case NL80211_IFTYPE_AP:
7400         case NL80211_IFTYPE_AP_VLAN:
7401         case NL80211_IFTYPE_MESH_POINT:
7402         case NL80211_IFTYPE_P2P_GO:
7403         case NL80211_IFTYPE_P2P_DEVICE:
7404                 break;
7405         default:
7406                 return -EOPNOTSUPP;
7407         }
7408
7409         /* not much point in registering if we can't reply */
7410         if (!rdev->ops->mgmt_tx)
7411                 return -EOPNOTSUPP;
7412
7413         return cfg80211_mlme_register_mgmt(wdev, info->snd_portid, frame_type,
7414                         nla_data(info->attrs[NL80211_ATTR_FRAME_MATCH]),
7415                         nla_len(info->attrs[NL80211_ATTR_FRAME_MATCH]));
7416 }
7417
7418 static int nl80211_tx_mgmt(struct sk_buff *skb, struct genl_info *info)
7419 {
7420         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7421         struct wireless_dev *wdev = info->user_ptr[1];
7422         struct cfg80211_chan_def chandef;
7423         int err;
7424         void *hdr = NULL;
7425         u64 cookie;
7426         struct sk_buff *msg = NULL;
7427         unsigned int wait = 0;
7428         bool offchan, no_cck, dont_wait_for_ack;
7429
7430         dont_wait_for_ack = info->attrs[NL80211_ATTR_DONT_WAIT_FOR_ACK];
7431
7432         if (!info->attrs[NL80211_ATTR_FRAME])
7433                 return -EINVAL;
7434
7435         if (!rdev->ops->mgmt_tx)
7436                 return -EOPNOTSUPP;
7437
7438         switch (wdev->iftype) {
7439         case NL80211_IFTYPE_P2P_DEVICE:
7440                 if (!info->attrs[NL80211_ATTR_WIPHY_FREQ])
7441                         return -EINVAL;
7442         case NL80211_IFTYPE_STATION:
7443         case NL80211_IFTYPE_ADHOC:
7444         case NL80211_IFTYPE_P2P_CLIENT:
7445         case NL80211_IFTYPE_AP:
7446         case NL80211_IFTYPE_AP_VLAN:
7447         case NL80211_IFTYPE_MESH_POINT:
7448         case NL80211_IFTYPE_P2P_GO:
7449                 break;
7450         default:
7451                 return -EOPNOTSUPP;
7452         }
7453
7454         if (info->attrs[NL80211_ATTR_DURATION]) {
7455                 if (!(rdev->wiphy.flags & WIPHY_FLAG_OFFCHAN_TX))
7456                         return -EINVAL;
7457                 wait = nla_get_u32(info->attrs[NL80211_ATTR_DURATION]);
7458
7459                 /*
7460                  * We should wait on the channel for at least a minimum amount
7461                  * of time (10ms) but no longer than the driver supports.
7462                  */
7463                 if (wait < NL80211_MIN_REMAIN_ON_CHANNEL_TIME ||
7464                     wait > rdev->wiphy.max_remain_on_channel_duration)
7465                         return -EINVAL;
7466
7467         }
7468
7469         offchan = info->attrs[NL80211_ATTR_OFFCHANNEL_TX_OK];
7470
7471         if (offchan && !(rdev->wiphy.flags & WIPHY_FLAG_OFFCHAN_TX))
7472                 return -EINVAL;
7473
7474         no_cck = nla_get_flag(info->attrs[NL80211_ATTR_TX_NO_CCK_RATE]);
7475
7476         /* get the channel if any has been specified, otherwise pass NULL to
7477          * the driver. The latter will use the current one
7478          */
7479         chandef.chan = NULL;
7480         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
7481                 err = nl80211_parse_chandef(rdev, info, &chandef);
7482                 if (err)
7483                         return err;
7484         }
7485
7486         if (!chandef.chan && offchan)
7487                 return -EINVAL;
7488
7489         if (!dont_wait_for_ack) {
7490                 msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
7491                 if (!msg)
7492                         return -ENOMEM;
7493
7494                 hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
7495                                      NL80211_CMD_FRAME);
7496                 if (!hdr) {
7497                         err = -ENOBUFS;
7498                         goto free_msg;
7499                 }
7500         }
7501
7502         err = cfg80211_mlme_mgmt_tx(rdev, wdev, chandef.chan, offchan, wait,
7503                                     nla_data(info->attrs[NL80211_ATTR_FRAME]),
7504                                     nla_len(info->attrs[NL80211_ATTR_FRAME]),
7505                                     no_cck, dont_wait_for_ack, &cookie);
7506         if (err)
7507                 goto free_msg;
7508
7509         if (msg) {
7510                 if (nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie))
7511                         goto nla_put_failure;
7512
7513                 genlmsg_end(msg, hdr);
7514                 return genlmsg_reply(msg, info);
7515         }
7516
7517         return 0;
7518
7519  nla_put_failure:
7520         err = -ENOBUFS;
7521  free_msg:
7522         nlmsg_free(msg);
7523         return err;
7524 }
7525
7526 static int nl80211_tx_mgmt_cancel_wait(struct sk_buff *skb, struct genl_info *info)
7527 {
7528         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7529         struct wireless_dev *wdev = info->user_ptr[1];
7530         u64 cookie;
7531
7532         if (!info->attrs[NL80211_ATTR_COOKIE])
7533                 return -EINVAL;
7534
7535         if (!rdev->ops->mgmt_tx_cancel_wait)
7536                 return -EOPNOTSUPP;
7537
7538         switch (wdev->iftype) {
7539         case NL80211_IFTYPE_STATION:
7540         case NL80211_IFTYPE_ADHOC:
7541         case NL80211_IFTYPE_P2P_CLIENT:
7542         case NL80211_IFTYPE_AP:
7543         case NL80211_IFTYPE_AP_VLAN:
7544         case NL80211_IFTYPE_P2P_GO:
7545         case NL80211_IFTYPE_P2P_DEVICE:
7546                 break;
7547         default:
7548                 return -EOPNOTSUPP;
7549         }
7550
7551         cookie = nla_get_u64(info->attrs[NL80211_ATTR_COOKIE]);
7552
7553         return rdev_mgmt_tx_cancel_wait(rdev, wdev, cookie);
7554 }
7555
7556 static int nl80211_set_power_save(struct sk_buff *skb, struct genl_info *info)
7557 {
7558         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7559         struct wireless_dev *wdev;
7560         struct net_device *dev = info->user_ptr[1];
7561         u8 ps_state;
7562         bool state;
7563         int err;
7564
7565         if (!info->attrs[NL80211_ATTR_PS_STATE])
7566                 return -EINVAL;
7567
7568         ps_state = nla_get_u32(info->attrs[NL80211_ATTR_PS_STATE]);
7569
7570         if (ps_state != NL80211_PS_DISABLED && ps_state != NL80211_PS_ENABLED)
7571                 return -EINVAL;
7572
7573         wdev = dev->ieee80211_ptr;
7574
7575         if (!rdev->ops->set_power_mgmt)
7576                 return -EOPNOTSUPP;
7577
7578         state = (ps_state == NL80211_PS_ENABLED) ? true : false;
7579
7580         if (state == wdev->ps)
7581                 return 0;
7582
7583         err = rdev_set_power_mgmt(rdev, dev, state, wdev->ps_timeout);
7584         if (!err)
7585                 wdev->ps = state;
7586         return err;
7587 }
7588
7589 static int nl80211_get_power_save(struct sk_buff *skb, struct genl_info *info)
7590 {
7591         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7592         enum nl80211_ps_state ps_state;
7593         struct wireless_dev *wdev;
7594         struct net_device *dev = info->user_ptr[1];
7595         struct sk_buff *msg;
7596         void *hdr;
7597         int err;
7598
7599         wdev = dev->ieee80211_ptr;
7600
7601         if (!rdev->ops->set_power_mgmt)
7602                 return -EOPNOTSUPP;
7603
7604         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
7605         if (!msg)
7606                 return -ENOMEM;
7607
7608         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
7609                              NL80211_CMD_GET_POWER_SAVE);
7610         if (!hdr) {
7611                 err = -ENOBUFS;
7612                 goto free_msg;
7613         }
7614
7615         if (wdev->ps)
7616                 ps_state = NL80211_PS_ENABLED;
7617         else
7618                 ps_state = NL80211_PS_DISABLED;
7619
7620         if (nla_put_u32(msg, NL80211_ATTR_PS_STATE, ps_state))
7621                 goto nla_put_failure;
7622
7623         genlmsg_end(msg, hdr);
7624         return genlmsg_reply(msg, info);
7625
7626  nla_put_failure:
7627         err = -ENOBUFS;
7628  free_msg:
7629         nlmsg_free(msg);
7630         return err;
7631 }
7632
7633 static struct nla_policy
7634 nl80211_attr_cqm_policy[NL80211_ATTR_CQM_MAX + 1] __read_mostly = {
7635         [NL80211_ATTR_CQM_RSSI_THOLD] = { .type = NLA_U32 },
7636         [NL80211_ATTR_CQM_RSSI_HYST] = { .type = NLA_U32 },
7637         [NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT] = { .type = NLA_U32 },
7638         [NL80211_ATTR_CQM_TXE_RATE] = { .type = NLA_U32 },
7639         [NL80211_ATTR_CQM_TXE_PKTS] = { .type = NLA_U32 },
7640         [NL80211_ATTR_CQM_TXE_INTVL] = { .type = NLA_U32 },
7641 };
7642
7643 static int nl80211_set_cqm_txe(struct genl_info *info,
7644                                u32 rate, u32 pkts, u32 intvl)
7645 {
7646         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7647         struct net_device *dev = info->user_ptr[1];
7648         struct wireless_dev *wdev = dev->ieee80211_ptr;
7649
7650         if (rate > 100 || intvl > NL80211_CQM_TXE_MAX_INTVL)
7651                 return -EINVAL;
7652
7653         if (!rdev->ops->set_cqm_txe_config)
7654                 return -EOPNOTSUPP;
7655
7656         if (wdev->iftype != NL80211_IFTYPE_STATION &&
7657             wdev->iftype != NL80211_IFTYPE_P2P_CLIENT)
7658                 return -EOPNOTSUPP;
7659
7660         return rdev_set_cqm_txe_config(rdev, dev, rate, pkts, intvl);
7661 }
7662
7663 static int nl80211_set_cqm_rssi(struct genl_info *info,
7664                                 s32 threshold, u32 hysteresis)
7665 {
7666         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7667         struct net_device *dev = info->user_ptr[1];
7668         struct wireless_dev *wdev = dev->ieee80211_ptr;
7669
7670         if (threshold > 0)
7671                 return -EINVAL;
7672
7673         /* disabling - hysteresis should also be zero then */
7674         if (threshold == 0)
7675                 hysteresis = 0;
7676
7677         if (!rdev->ops->set_cqm_rssi_config)
7678                 return -EOPNOTSUPP;
7679
7680         if (wdev->iftype != NL80211_IFTYPE_STATION &&
7681             wdev->iftype != NL80211_IFTYPE_P2P_CLIENT)
7682                 return -EOPNOTSUPP;
7683
7684         return rdev_set_cqm_rssi_config(rdev, dev, threshold, hysteresis);
7685 }
7686
7687 static int nl80211_set_cqm(struct sk_buff *skb, struct genl_info *info)
7688 {
7689         struct nlattr *attrs[NL80211_ATTR_CQM_MAX + 1];
7690         struct nlattr *cqm;
7691         int err;
7692
7693         cqm = info->attrs[NL80211_ATTR_CQM];
7694         if (!cqm)
7695                 return -EINVAL;
7696
7697         err = nla_parse_nested(attrs, NL80211_ATTR_CQM_MAX, cqm,
7698                                nl80211_attr_cqm_policy);
7699         if (err)
7700                 return err;
7701
7702         if (attrs[NL80211_ATTR_CQM_RSSI_THOLD] &&
7703             attrs[NL80211_ATTR_CQM_RSSI_HYST]) {
7704                 s32 threshold = nla_get_s32(attrs[NL80211_ATTR_CQM_RSSI_THOLD]);
7705                 u32 hysteresis = nla_get_u32(attrs[NL80211_ATTR_CQM_RSSI_HYST]);
7706
7707                 return nl80211_set_cqm_rssi(info, threshold, hysteresis);
7708         }
7709
7710         if (attrs[NL80211_ATTR_CQM_TXE_RATE] &&
7711             attrs[NL80211_ATTR_CQM_TXE_PKTS] &&
7712             attrs[NL80211_ATTR_CQM_TXE_INTVL]) {
7713                 u32 rate = nla_get_u32(attrs[NL80211_ATTR_CQM_TXE_RATE]);
7714                 u32 pkts = nla_get_u32(attrs[NL80211_ATTR_CQM_TXE_PKTS]);
7715                 u32 intvl = nla_get_u32(attrs[NL80211_ATTR_CQM_TXE_INTVL]);
7716
7717                 return nl80211_set_cqm_txe(info, rate, pkts, intvl);
7718         }
7719
7720         return -EINVAL;
7721 }
7722
7723 static int nl80211_join_mesh(struct sk_buff *skb, struct genl_info *info)
7724 {
7725         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7726         struct net_device *dev = info->user_ptr[1];
7727         struct mesh_config cfg;
7728         struct mesh_setup setup;
7729         int err;
7730
7731         /* start with default */
7732         memcpy(&cfg, &default_mesh_config, sizeof(cfg));
7733         memcpy(&setup, &default_mesh_setup, sizeof(setup));
7734
7735         if (info->attrs[NL80211_ATTR_MESH_CONFIG]) {
7736                 /* and parse parameters if given */
7737                 err = nl80211_parse_mesh_config(info, &cfg, NULL);
7738                 if (err)
7739                         return err;
7740         }
7741
7742         if (!info->attrs[NL80211_ATTR_MESH_ID] ||
7743             !nla_len(info->attrs[NL80211_ATTR_MESH_ID]))
7744                 return -EINVAL;
7745
7746         setup.mesh_id = nla_data(info->attrs[NL80211_ATTR_MESH_ID]);
7747         setup.mesh_id_len = nla_len(info->attrs[NL80211_ATTR_MESH_ID]);
7748
7749         if (info->attrs[NL80211_ATTR_MCAST_RATE] &&
7750             !nl80211_parse_mcast_rate(rdev, setup.mcast_rate,
7751                             nla_get_u32(info->attrs[NL80211_ATTR_MCAST_RATE])))
7752                         return -EINVAL;
7753
7754         if (info->attrs[NL80211_ATTR_BEACON_INTERVAL]) {
7755                 setup.beacon_interval =
7756                         nla_get_u32(info->attrs[NL80211_ATTR_BEACON_INTERVAL]);
7757                 if (setup.beacon_interval < 10 ||
7758                     setup.beacon_interval > 10000)
7759                         return -EINVAL;
7760         }
7761
7762         if (info->attrs[NL80211_ATTR_DTIM_PERIOD]) {
7763                 setup.dtim_period =
7764                         nla_get_u32(info->attrs[NL80211_ATTR_DTIM_PERIOD]);
7765                 if (setup.dtim_period < 1 || setup.dtim_period > 100)
7766                         return -EINVAL;
7767         }
7768
7769         if (info->attrs[NL80211_ATTR_MESH_SETUP]) {
7770                 /* parse additional setup parameters if given */
7771                 err = nl80211_parse_mesh_setup(info, &setup);
7772                 if (err)
7773                         return err;
7774         }
7775
7776         if (setup.user_mpm)
7777                 cfg.auto_open_plinks = false;
7778
7779         if (info->attrs[NL80211_ATTR_WIPHY_FREQ]) {
7780                 err = nl80211_parse_chandef(rdev, info, &setup.chandef);
7781                 if (err)
7782                         return err;
7783         } else {
7784                 /* cfg80211_join_mesh() will sort it out */
7785                 setup.chandef.chan = NULL;
7786         }
7787
7788         if (info->attrs[NL80211_ATTR_BSS_BASIC_RATES]) {
7789                 u8 *rates = nla_data(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
7790                 int n_rates =
7791                         nla_len(info->attrs[NL80211_ATTR_BSS_BASIC_RATES]);
7792                 struct ieee80211_supported_band *sband;
7793
7794                 if (!setup.chandef.chan)
7795                         return -EINVAL;
7796
7797                 sband = rdev->wiphy.bands[setup.chandef.chan->band];
7798
7799                 err = ieee80211_get_ratemask(sband, rates, n_rates,
7800                                              &setup.basic_rates);
7801                 if (err)
7802                         return err;
7803         }
7804
7805         return cfg80211_join_mesh(rdev, dev, &setup, &cfg);
7806 }
7807
7808 static int nl80211_leave_mesh(struct sk_buff *skb, struct genl_info *info)
7809 {
7810         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7811         struct net_device *dev = info->user_ptr[1];
7812
7813         return cfg80211_leave_mesh(rdev, dev);
7814 }
7815
7816 #ifdef CONFIG_PM
7817 static int nl80211_send_wowlan_patterns(struct sk_buff *msg,
7818                                         struct cfg80211_registered_device *rdev)
7819 {
7820         struct cfg80211_wowlan *wowlan = rdev->wiphy.wowlan_config;
7821         struct nlattr *nl_pats, *nl_pat;
7822         int i, pat_len;
7823
7824         if (!wowlan->n_patterns)
7825                 return 0;
7826
7827         nl_pats = nla_nest_start(msg, NL80211_WOWLAN_TRIG_PKT_PATTERN);
7828         if (!nl_pats)
7829                 return -ENOBUFS;
7830
7831         for (i = 0; i < wowlan->n_patterns; i++) {
7832                 nl_pat = nla_nest_start(msg, i + 1);
7833                 if (!nl_pat)
7834                         return -ENOBUFS;
7835                 pat_len = wowlan->patterns[i].pattern_len;
7836                 if (nla_put(msg, NL80211_PKTPAT_MASK, DIV_ROUND_UP(pat_len, 8),
7837                             wowlan->patterns[i].mask) ||
7838                     nla_put(msg, NL80211_PKTPAT_PATTERN, pat_len,
7839                             wowlan->patterns[i].pattern) ||
7840                     nla_put_u32(msg, NL80211_PKTPAT_OFFSET,
7841                                 wowlan->patterns[i].pkt_offset))
7842                         return -ENOBUFS;
7843                 nla_nest_end(msg, nl_pat);
7844         }
7845         nla_nest_end(msg, nl_pats);
7846
7847         return 0;
7848 }
7849
7850 static int nl80211_send_wowlan_tcp(struct sk_buff *msg,
7851                                    struct cfg80211_wowlan_tcp *tcp)
7852 {
7853         struct nlattr *nl_tcp;
7854
7855         if (!tcp)
7856                 return 0;
7857
7858         nl_tcp = nla_nest_start(msg, NL80211_WOWLAN_TRIG_TCP_CONNECTION);
7859         if (!nl_tcp)
7860                 return -ENOBUFS;
7861
7862         if (nla_put_be32(msg, NL80211_WOWLAN_TCP_SRC_IPV4, tcp->src) ||
7863             nla_put_be32(msg, NL80211_WOWLAN_TCP_DST_IPV4, tcp->dst) ||
7864             nla_put(msg, NL80211_WOWLAN_TCP_DST_MAC, ETH_ALEN, tcp->dst_mac) ||
7865             nla_put_u16(msg, NL80211_WOWLAN_TCP_SRC_PORT, tcp->src_port) ||
7866             nla_put_u16(msg, NL80211_WOWLAN_TCP_DST_PORT, tcp->dst_port) ||
7867             nla_put(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD,
7868                     tcp->payload_len, tcp->payload) ||
7869             nla_put_u32(msg, NL80211_WOWLAN_TCP_DATA_INTERVAL,
7870                         tcp->data_interval) ||
7871             nla_put(msg, NL80211_WOWLAN_TCP_WAKE_PAYLOAD,
7872                     tcp->wake_len, tcp->wake_data) ||
7873             nla_put(msg, NL80211_WOWLAN_TCP_WAKE_MASK,
7874                     DIV_ROUND_UP(tcp->wake_len, 8), tcp->wake_mask))
7875                 return -ENOBUFS;
7876
7877         if (tcp->payload_seq.len &&
7878             nla_put(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ,
7879                     sizeof(tcp->payload_seq), &tcp->payload_seq))
7880                 return -ENOBUFS;
7881
7882         if (tcp->payload_tok.len &&
7883             nla_put(msg, NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN,
7884                     sizeof(tcp->payload_tok) + tcp->tokens_size,
7885                     &tcp->payload_tok))
7886                 return -ENOBUFS;
7887
7888         nla_nest_end(msg, nl_tcp);
7889
7890         return 0;
7891 }
7892
7893 static int nl80211_get_wowlan(struct sk_buff *skb, struct genl_info *info)
7894 {
7895         struct cfg80211_registered_device *rdev = info->user_ptr[0];
7896         struct sk_buff *msg;
7897         void *hdr;
7898         u32 size = NLMSG_DEFAULT_SIZE;
7899
7900         if (!rdev->wiphy.wowlan)
7901                 return -EOPNOTSUPP;
7902
7903         if (rdev->wiphy.wowlan_config && rdev->wiphy.wowlan_config->tcp) {
7904                 /* adjust size to have room for all the data */
7905                 size += rdev->wiphy.wowlan_config->tcp->tokens_size +
7906                         rdev->wiphy.wowlan_config->tcp->payload_len +
7907                         rdev->wiphy.wowlan_config->tcp->wake_len +
7908                         rdev->wiphy.wowlan_config->tcp->wake_len / 8;
7909         }
7910
7911         msg = nlmsg_new(size, GFP_KERNEL);
7912         if (!msg)
7913                 return -ENOMEM;
7914
7915         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
7916                              NL80211_CMD_GET_WOWLAN);
7917         if (!hdr)
7918                 goto nla_put_failure;
7919
7920         if (rdev->wiphy.wowlan_config) {
7921                 struct nlattr *nl_wowlan;
7922
7923                 nl_wowlan = nla_nest_start(msg, NL80211_ATTR_WOWLAN_TRIGGERS);
7924                 if (!nl_wowlan)
7925                         goto nla_put_failure;
7926
7927                 if ((rdev->wiphy.wowlan_config->any &&
7928                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_ANY)) ||
7929                     (rdev->wiphy.wowlan_config->disconnect &&
7930                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_DISCONNECT)) ||
7931                     (rdev->wiphy.wowlan_config->magic_pkt &&
7932                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_MAGIC_PKT)) ||
7933                     (rdev->wiphy.wowlan_config->gtk_rekey_failure &&
7934                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE)) ||
7935                     (rdev->wiphy.wowlan_config->eap_identity_req &&
7936                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST)) ||
7937                     (rdev->wiphy.wowlan_config->four_way_handshake &&
7938                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE)) ||
7939                     (rdev->wiphy.wowlan_config->rfkill_release &&
7940                      nla_put_flag(msg, NL80211_WOWLAN_TRIG_RFKILL_RELEASE)))
7941                         goto nla_put_failure;
7942
7943                 if (nl80211_send_wowlan_patterns(msg, rdev))
7944                         goto nla_put_failure;
7945
7946                 if (nl80211_send_wowlan_tcp(msg,
7947                                             rdev->wiphy.wowlan_config->tcp))
7948                         goto nla_put_failure;
7949
7950                 nla_nest_end(msg, nl_wowlan);
7951         }
7952
7953         genlmsg_end(msg, hdr);
7954         return genlmsg_reply(msg, info);
7955
7956 nla_put_failure:
7957         nlmsg_free(msg);
7958         return -ENOBUFS;
7959 }
7960
7961 static int nl80211_parse_wowlan_tcp(struct cfg80211_registered_device *rdev,
7962                                     struct nlattr *attr,
7963                                     struct cfg80211_wowlan *trig)
7964 {
7965         struct nlattr *tb[NUM_NL80211_WOWLAN_TCP];
7966         struct cfg80211_wowlan_tcp *cfg;
7967         struct nl80211_wowlan_tcp_data_token *tok = NULL;
7968         struct nl80211_wowlan_tcp_data_seq *seq = NULL;
7969         u32 size;
7970         u32 data_size, wake_size, tokens_size = 0, wake_mask_size;
7971         int err, port;
7972
7973         if (!rdev->wiphy.wowlan->tcp)
7974                 return -EINVAL;
7975
7976         err = nla_parse(tb, MAX_NL80211_WOWLAN_TCP,
7977                         nla_data(attr), nla_len(attr),
7978                         nl80211_wowlan_tcp_policy);
7979         if (err)
7980                 return err;
7981
7982         if (!tb[NL80211_WOWLAN_TCP_SRC_IPV4] ||
7983             !tb[NL80211_WOWLAN_TCP_DST_IPV4] ||
7984             !tb[NL80211_WOWLAN_TCP_DST_MAC] ||
7985             !tb[NL80211_WOWLAN_TCP_DST_PORT] ||
7986             !tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD] ||
7987             !tb[NL80211_WOWLAN_TCP_DATA_INTERVAL] ||
7988             !tb[NL80211_WOWLAN_TCP_WAKE_PAYLOAD] ||
7989             !tb[NL80211_WOWLAN_TCP_WAKE_MASK])
7990                 return -EINVAL;
7991
7992         data_size = nla_len(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD]);
7993         if (data_size > rdev->wiphy.wowlan->tcp->data_payload_max)
7994                 return -EINVAL;
7995
7996         if (nla_get_u32(tb[NL80211_WOWLAN_TCP_DATA_INTERVAL]) >
7997                         rdev->wiphy.wowlan->tcp->data_interval_max ||
7998             nla_get_u32(tb[NL80211_WOWLAN_TCP_DATA_INTERVAL]) == 0)
7999                 return -EINVAL;
8000
8001         wake_size = nla_len(tb[NL80211_WOWLAN_TCP_WAKE_PAYLOAD]);
8002         if (wake_size > rdev->wiphy.wowlan->tcp->wake_payload_max)
8003                 return -EINVAL;
8004
8005         wake_mask_size = nla_len(tb[NL80211_WOWLAN_TCP_WAKE_MASK]);
8006         if (wake_mask_size != DIV_ROUND_UP(wake_size, 8))
8007                 return -EINVAL;
8008
8009         if (tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN]) {
8010                 u32 tokln = nla_len(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN]);
8011
8012                 tok = nla_data(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_TOKEN]);
8013                 tokens_size = tokln - sizeof(*tok);
8014
8015                 if (!tok->len || tokens_size % tok->len)
8016                         return -EINVAL;
8017                 if (!rdev->wiphy.wowlan->tcp->tok)
8018                         return -EINVAL;
8019                 if (tok->len > rdev->wiphy.wowlan->tcp->tok->max_len)
8020                         return -EINVAL;
8021                 if (tok->len < rdev->wiphy.wowlan->tcp->tok->min_len)
8022                         return -EINVAL;
8023                 if (tokens_size > rdev->wiphy.wowlan->tcp->tok->bufsize)
8024                         return -EINVAL;
8025                 if (tok->offset + tok->len > data_size)
8026                         return -EINVAL;
8027         }
8028
8029         if (tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ]) {
8030                 seq = nla_data(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD_SEQ]);
8031                 if (!rdev->wiphy.wowlan->tcp->seq)
8032                         return -EINVAL;
8033                 if (seq->len == 0 || seq->len > 4)
8034                         return -EINVAL;
8035                 if (seq->len + seq->offset > data_size)
8036                         return -EINVAL;
8037         }
8038
8039         size = sizeof(*cfg);
8040         size += data_size;
8041         size += wake_size + wake_mask_size;
8042         size += tokens_size;
8043
8044         cfg = kzalloc(size, GFP_KERNEL);
8045         if (!cfg)
8046                 return -ENOMEM;
8047         cfg->src = nla_get_be32(tb[NL80211_WOWLAN_TCP_SRC_IPV4]);
8048         cfg->dst = nla_get_be32(tb[NL80211_WOWLAN_TCP_DST_IPV4]);
8049         memcpy(cfg->dst_mac, nla_data(tb[NL80211_WOWLAN_TCP_DST_MAC]),
8050                ETH_ALEN);
8051         if (tb[NL80211_WOWLAN_TCP_SRC_PORT])
8052                 port = nla_get_u16(tb[NL80211_WOWLAN_TCP_SRC_PORT]);
8053         else
8054                 port = 0;
8055 #ifdef CONFIG_INET
8056         /* allocate a socket and port for it and use it */
8057         err = __sock_create(wiphy_net(&rdev->wiphy), PF_INET, SOCK_STREAM,
8058                             IPPROTO_TCP, &cfg->sock, 1);
8059         if (err) {
8060                 kfree(cfg);
8061                 return err;
8062         }
8063         if (inet_csk_get_port(cfg->sock->sk, port)) {
8064                 sock_release(cfg->sock);
8065                 kfree(cfg);
8066                 return -EADDRINUSE;
8067         }
8068         cfg->src_port = inet_sk(cfg->sock->sk)->inet_num;
8069 #else
8070         if (!port) {
8071                 kfree(cfg);
8072                 return -EINVAL;
8073         }
8074         cfg->src_port = port;
8075 #endif
8076
8077         cfg->dst_port = nla_get_u16(tb[NL80211_WOWLAN_TCP_DST_PORT]);
8078         cfg->payload_len = data_size;
8079         cfg->payload = (u8 *)cfg + sizeof(*cfg) + tokens_size;
8080         memcpy((void *)cfg->payload,
8081                nla_data(tb[NL80211_WOWLAN_TCP_DATA_PAYLOAD]),
8082                data_size);
8083         if (seq)
8084                 cfg->payload_seq = *seq;
8085         cfg->data_interval = nla_get_u32(tb[NL80211_WOWLAN_TCP_DATA_INTERVAL]);
8086         cfg->wake_len = wake_size;
8087         cfg->wake_data = (u8 *)cfg + sizeof(*cfg) + tokens_size + data_size;
8088         memcpy((void *)cfg->wake_data,
8089                nla_data(tb[NL80211_WOWLAN_TCP_WAKE_PAYLOAD]),
8090                wake_size);
8091         cfg->wake_mask = (u8 *)cfg + sizeof(*cfg) + tokens_size +
8092                          data_size + wake_size;
8093         memcpy((void *)cfg->wake_mask,
8094                nla_data(tb[NL80211_WOWLAN_TCP_WAKE_MASK]),
8095                wake_mask_size);
8096         if (tok) {
8097                 cfg->tokens_size = tokens_size;
8098                 memcpy(&cfg->payload_tok, tok, sizeof(*tok) + tokens_size);
8099         }
8100
8101         trig->tcp = cfg;
8102
8103         return 0;
8104 }
8105
8106 static int nl80211_set_wowlan(struct sk_buff *skb, struct genl_info *info)
8107 {
8108         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8109         struct nlattr *tb[NUM_NL80211_WOWLAN_TRIG];
8110         struct cfg80211_wowlan new_triggers = {};
8111         struct cfg80211_wowlan *ntrig;
8112         const struct wiphy_wowlan_support *wowlan = rdev->wiphy.wowlan;
8113         int err, i;
8114         bool prev_enabled = rdev->wiphy.wowlan_config;
8115
8116         if (!wowlan)
8117                 return -EOPNOTSUPP;
8118
8119         if (!info->attrs[NL80211_ATTR_WOWLAN_TRIGGERS]) {
8120                 cfg80211_rdev_free_wowlan(rdev);
8121                 rdev->wiphy.wowlan_config = NULL;
8122                 goto set_wakeup;
8123         }
8124
8125         err = nla_parse(tb, MAX_NL80211_WOWLAN_TRIG,
8126                         nla_data(info->attrs[NL80211_ATTR_WOWLAN_TRIGGERS]),
8127                         nla_len(info->attrs[NL80211_ATTR_WOWLAN_TRIGGERS]),
8128                         nl80211_wowlan_policy);
8129         if (err)
8130                 return err;
8131
8132         if (tb[NL80211_WOWLAN_TRIG_ANY]) {
8133                 if (!(wowlan->flags & WIPHY_WOWLAN_ANY))
8134                         return -EINVAL;
8135                 new_triggers.any = true;
8136         }
8137
8138         if (tb[NL80211_WOWLAN_TRIG_DISCONNECT]) {
8139                 if (!(wowlan->flags & WIPHY_WOWLAN_DISCONNECT))
8140                         return -EINVAL;
8141                 new_triggers.disconnect = true;
8142         }
8143
8144         if (tb[NL80211_WOWLAN_TRIG_MAGIC_PKT]) {
8145                 if (!(wowlan->flags & WIPHY_WOWLAN_MAGIC_PKT))
8146                         return -EINVAL;
8147                 new_triggers.magic_pkt = true;
8148         }
8149
8150         if (tb[NL80211_WOWLAN_TRIG_GTK_REKEY_SUPPORTED])
8151                 return -EINVAL;
8152
8153         if (tb[NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE]) {
8154                 if (!(wowlan->flags & WIPHY_WOWLAN_GTK_REKEY_FAILURE))
8155                         return -EINVAL;
8156                 new_triggers.gtk_rekey_failure = true;
8157         }
8158
8159         if (tb[NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST]) {
8160                 if (!(wowlan->flags & WIPHY_WOWLAN_EAP_IDENTITY_REQ))
8161                         return -EINVAL;
8162                 new_triggers.eap_identity_req = true;
8163         }
8164
8165         if (tb[NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE]) {
8166                 if (!(wowlan->flags & WIPHY_WOWLAN_4WAY_HANDSHAKE))
8167                         return -EINVAL;
8168                 new_triggers.four_way_handshake = true;
8169         }
8170
8171         if (tb[NL80211_WOWLAN_TRIG_RFKILL_RELEASE]) {
8172                 if (!(wowlan->flags & WIPHY_WOWLAN_RFKILL_RELEASE))
8173                         return -EINVAL;
8174                 new_triggers.rfkill_release = true;
8175         }
8176
8177         if (tb[NL80211_WOWLAN_TRIG_PKT_PATTERN]) {
8178                 struct nlattr *pat;
8179                 int n_patterns = 0;
8180                 int rem, pat_len, mask_len, pkt_offset;
8181                 struct nlattr *pat_tb[NUM_NL80211_PKTPAT];
8182
8183                 nla_for_each_nested(pat, tb[NL80211_WOWLAN_TRIG_PKT_PATTERN],
8184                                     rem)
8185                         n_patterns++;
8186                 if (n_patterns > wowlan->n_patterns)
8187                         return -EINVAL;
8188
8189                 new_triggers.patterns = kcalloc(n_patterns,
8190                                                 sizeof(new_triggers.patterns[0]),
8191                                                 GFP_KERNEL);
8192                 if (!new_triggers.patterns)
8193                         return -ENOMEM;
8194
8195                 new_triggers.n_patterns = n_patterns;
8196                 i = 0;
8197
8198                 nla_for_each_nested(pat, tb[NL80211_WOWLAN_TRIG_PKT_PATTERN],
8199                                     rem) {
8200                         nla_parse(pat_tb, MAX_NL80211_PKTPAT, nla_data(pat),
8201                                   nla_len(pat), NULL);
8202                         err = -EINVAL;
8203                         if (!pat_tb[NL80211_PKTPAT_MASK] ||
8204                             !pat_tb[NL80211_PKTPAT_PATTERN])
8205                                 goto error;
8206                         pat_len = nla_len(pat_tb[NL80211_PKTPAT_PATTERN]);
8207                         mask_len = DIV_ROUND_UP(pat_len, 8);
8208                         if (nla_len(pat_tb[NL80211_PKTPAT_MASK]) != mask_len)
8209                                 goto error;
8210                         if (pat_len > wowlan->pattern_max_len ||
8211                             pat_len < wowlan->pattern_min_len)
8212                                 goto error;
8213
8214                         if (!pat_tb[NL80211_PKTPAT_OFFSET])
8215                                 pkt_offset = 0;
8216                         else
8217                                 pkt_offset = nla_get_u32(
8218                                         pat_tb[NL80211_PKTPAT_OFFSET]);
8219                         if (pkt_offset > wowlan->max_pkt_offset)
8220                                 goto error;
8221                         new_triggers.patterns[i].pkt_offset = pkt_offset;
8222
8223                         new_triggers.patterns[i].mask =
8224                                 kmalloc(mask_len + pat_len, GFP_KERNEL);
8225                         if (!new_triggers.patterns[i].mask) {
8226                                 err = -ENOMEM;
8227                                 goto error;
8228                         }
8229                         new_triggers.patterns[i].pattern =
8230                                 new_triggers.patterns[i].mask + mask_len;
8231                         memcpy(new_triggers.patterns[i].mask,
8232                                nla_data(pat_tb[NL80211_PKTPAT_MASK]),
8233                                mask_len);
8234                         new_triggers.patterns[i].pattern_len = pat_len;
8235                         memcpy(new_triggers.patterns[i].pattern,
8236                                nla_data(pat_tb[NL80211_PKTPAT_PATTERN]),
8237                                pat_len);
8238                         i++;
8239                 }
8240         }
8241
8242         if (tb[NL80211_WOWLAN_TRIG_TCP_CONNECTION]) {
8243                 err = nl80211_parse_wowlan_tcp(
8244                         rdev, tb[NL80211_WOWLAN_TRIG_TCP_CONNECTION],
8245                         &new_triggers);
8246                 if (err)
8247                         goto error;
8248         }
8249
8250         ntrig = kmemdup(&new_triggers, sizeof(new_triggers), GFP_KERNEL);
8251         if (!ntrig) {
8252                 err = -ENOMEM;
8253                 goto error;
8254         }
8255         cfg80211_rdev_free_wowlan(rdev);
8256         rdev->wiphy.wowlan_config = ntrig;
8257
8258  set_wakeup:
8259         if (rdev->ops->set_wakeup &&
8260             prev_enabled != !!rdev->wiphy.wowlan_config)
8261                 rdev_set_wakeup(rdev, rdev->wiphy.wowlan_config);
8262
8263         return 0;
8264  error:
8265         for (i = 0; i < new_triggers.n_patterns; i++)
8266                 kfree(new_triggers.patterns[i].mask);
8267         kfree(new_triggers.patterns);
8268         if (new_triggers.tcp && new_triggers.tcp->sock)
8269                 sock_release(new_triggers.tcp->sock);
8270         kfree(new_triggers.tcp);
8271         return err;
8272 }
8273 #endif
8274
8275 static int nl80211_send_coalesce_rules(struct sk_buff *msg,
8276                                        struct cfg80211_registered_device *rdev)
8277 {
8278         struct nlattr *nl_pats, *nl_pat, *nl_rule, *nl_rules;
8279         int i, j, pat_len;
8280         struct cfg80211_coalesce_rules *rule;
8281
8282         if (!rdev->coalesce->n_rules)
8283                 return 0;
8284
8285         nl_rules = nla_nest_start(msg, NL80211_ATTR_COALESCE_RULE);
8286         if (!nl_rules)
8287                 return -ENOBUFS;
8288
8289         for (i = 0; i < rdev->coalesce->n_rules; i++) {
8290                 nl_rule = nla_nest_start(msg, i + 1);
8291                 if (!nl_rule)
8292                         return -ENOBUFS;
8293
8294                 rule = &rdev->coalesce->rules[i];
8295                 if (nla_put_u32(msg, NL80211_ATTR_COALESCE_RULE_DELAY,
8296                                 rule->delay))
8297                         return -ENOBUFS;
8298
8299                 if (nla_put_u32(msg, NL80211_ATTR_COALESCE_RULE_CONDITION,
8300                                 rule->condition))
8301                         return -ENOBUFS;
8302
8303                 nl_pats = nla_nest_start(msg,
8304                                 NL80211_ATTR_COALESCE_RULE_PKT_PATTERN);
8305                 if (!nl_pats)
8306                         return -ENOBUFS;
8307
8308                 for (j = 0; j < rule->n_patterns; j++) {
8309                         nl_pat = nla_nest_start(msg, j + 1);
8310                         if (!nl_pat)
8311                                 return -ENOBUFS;
8312                         pat_len = rule->patterns[j].pattern_len;
8313                         if (nla_put(msg, NL80211_PKTPAT_MASK,
8314                                     DIV_ROUND_UP(pat_len, 8),
8315                                     rule->patterns[j].mask) ||
8316                             nla_put(msg, NL80211_PKTPAT_PATTERN, pat_len,
8317                                     rule->patterns[j].pattern) ||
8318                             nla_put_u32(msg, NL80211_PKTPAT_OFFSET,
8319                                         rule->patterns[j].pkt_offset))
8320                                 return -ENOBUFS;
8321                         nla_nest_end(msg, nl_pat);
8322                 }
8323                 nla_nest_end(msg, nl_pats);
8324                 nla_nest_end(msg, nl_rule);
8325         }
8326         nla_nest_end(msg, nl_rules);
8327
8328         return 0;
8329 }
8330
8331 static int nl80211_get_coalesce(struct sk_buff *skb, struct genl_info *info)
8332 {
8333         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8334         struct sk_buff *msg;
8335         void *hdr;
8336
8337         if (!rdev->wiphy.coalesce)
8338                 return -EOPNOTSUPP;
8339
8340         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8341         if (!msg)
8342                 return -ENOMEM;
8343
8344         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
8345                              NL80211_CMD_GET_COALESCE);
8346         if (!hdr)
8347                 goto nla_put_failure;
8348
8349         if (rdev->coalesce && nl80211_send_coalesce_rules(msg, rdev))
8350                 goto nla_put_failure;
8351
8352         genlmsg_end(msg, hdr);
8353         return genlmsg_reply(msg, info);
8354
8355 nla_put_failure:
8356         nlmsg_free(msg);
8357         return -ENOBUFS;
8358 }
8359
8360 void cfg80211_rdev_free_coalesce(struct cfg80211_registered_device *rdev)
8361 {
8362         struct cfg80211_coalesce *coalesce = rdev->coalesce;
8363         int i, j;
8364         struct cfg80211_coalesce_rules *rule;
8365
8366         if (!coalesce)
8367                 return;
8368
8369         for (i = 0; i < coalesce->n_rules; i++) {
8370                 rule = &coalesce->rules[i];
8371                 for (j = 0; j < rule->n_patterns; j++)
8372                         kfree(rule->patterns[j].mask);
8373                 kfree(rule->patterns);
8374         }
8375         kfree(coalesce->rules);
8376         kfree(coalesce);
8377         rdev->coalesce = NULL;
8378 }
8379
8380 static int nl80211_parse_coalesce_rule(struct cfg80211_registered_device *rdev,
8381                                        struct nlattr *rule,
8382                                        struct cfg80211_coalesce_rules *new_rule)
8383 {
8384         int err, i;
8385         const struct wiphy_coalesce_support *coalesce = rdev->wiphy.coalesce;
8386         struct nlattr *tb[NUM_NL80211_ATTR_COALESCE_RULE], *pat;
8387         int rem, pat_len, mask_len, pkt_offset, n_patterns = 0;
8388         struct nlattr *pat_tb[NUM_NL80211_PKTPAT];
8389
8390         err = nla_parse(tb, NL80211_ATTR_COALESCE_RULE_MAX, nla_data(rule),
8391                         nla_len(rule), nl80211_coalesce_policy);
8392         if (err)
8393                 return err;
8394
8395         if (tb[NL80211_ATTR_COALESCE_RULE_DELAY])
8396                 new_rule->delay =
8397                         nla_get_u32(tb[NL80211_ATTR_COALESCE_RULE_DELAY]);
8398         if (new_rule->delay > coalesce->max_delay)
8399                 return -EINVAL;
8400
8401         if (tb[NL80211_ATTR_COALESCE_RULE_CONDITION])
8402                 new_rule->condition =
8403                         nla_get_u32(tb[NL80211_ATTR_COALESCE_RULE_CONDITION]);
8404         if (new_rule->condition != NL80211_COALESCE_CONDITION_MATCH &&
8405             new_rule->condition != NL80211_COALESCE_CONDITION_NO_MATCH)
8406                 return -EINVAL;
8407
8408         if (!tb[NL80211_ATTR_COALESCE_RULE_PKT_PATTERN])
8409                 return -EINVAL;
8410
8411         nla_for_each_nested(pat, tb[NL80211_ATTR_COALESCE_RULE_PKT_PATTERN],
8412                             rem)
8413                 n_patterns++;
8414         if (n_patterns > coalesce->n_patterns)
8415                 return -EINVAL;
8416
8417         new_rule->patterns = kcalloc(n_patterns, sizeof(new_rule->patterns[0]),
8418                                      GFP_KERNEL);
8419         if (!new_rule->patterns)
8420                 return -ENOMEM;
8421
8422         new_rule->n_patterns = n_patterns;
8423         i = 0;
8424
8425         nla_for_each_nested(pat, tb[NL80211_ATTR_COALESCE_RULE_PKT_PATTERN],
8426                             rem) {
8427                 nla_parse(pat_tb, MAX_NL80211_PKTPAT, nla_data(pat),
8428                           nla_len(pat), NULL);
8429                 if (!pat_tb[NL80211_PKTPAT_MASK] ||
8430                     !pat_tb[NL80211_PKTPAT_PATTERN])
8431                         return -EINVAL;
8432                 pat_len = nla_len(pat_tb[NL80211_PKTPAT_PATTERN]);
8433                 mask_len = DIV_ROUND_UP(pat_len, 8);
8434                 if (nla_len(pat_tb[NL80211_PKTPAT_MASK]) != mask_len)
8435                         return -EINVAL;
8436                 if (pat_len > coalesce->pattern_max_len ||
8437                     pat_len < coalesce->pattern_min_len)
8438                         return -EINVAL;
8439
8440                 if (!pat_tb[NL80211_PKTPAT_OFFSET])
8441                         pkt_offset = 0;
8442                 else
8443                         pkt_offset = nla_get_u32(pat_tb[NL80211_PKTPAT_OFFSET]);
8444                 if (pkt_offset > coalesce->max_pkt_offset)
8445                         return -EINVAL;
8446                 new_rule->patterns[i].pkt_offset = pkt_offset;
8447
8448                 new_rule->patterns[i].mask =
8449                         kmalloc(mask_len + pat_len, GFP_KERNEL);
8450                 if (!new_rule->patterns[i].mask)
8451                         return -ENOMEM;
8452                 new_rule->patterns[i].pattern =
8453                         new_rule->patterns[i].mask + mask_len;
8454                 memcpy(new_rule->patterns[i].mask,
8455                        nla_data(pat_tb[NL80211_PKTPAT_MASK]), mask_len);
8456                 new_rule->patterns[i].pattern_len = pat_len;
8457                 memcpy(new_rule->patterns[i].pattern,
8458                        nla_data(pat_tb[NL80211_PKTPAT_PATTERN]), pat_len);
8459                 i++;
8460         }
8461
8462         return 0;
8463 }
8464
8465 static int nl80211_set_coalesce(struct sk_buff *skb, struct genl_info *info)
8466 {
8467         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8468         const struct wiphy_coalesce_support *coalesce = rdev->wiphy.coalesce;
8469         struct cfg80211_coalesce new_coalesce = {};
8470         struct cfg80211_coalesce *n_coalesce;
8471         int err, rem_rule, n_rules = 0, i, j;
8472         struct nlattr *rule;
8473         struct cfg80211_coalesce_rules *tmp_rule;
8474
8475         if (!rdev->wiphy.coalesce || !rdev->ops->set_coalesce)
8476                 return -EOPNOTSUPP;
8477
8478         if (!info->attrs[NL80211_ATTR_COALESCE_RULE]) {
8479                 cfg80211_rdev_free_coalesce(rdev);
8480                 rdev->ops->set_coalesce(&rdev->wiphy, NULL);
8481                 return 0;
8482         }
8483
8484         nla_for_each_nested(rule, info->attrs[NL80211_ATTR_COALESCE_RULE],
8485                             rem_rule)
8486                 n_rules++;
8487         if (n_rules > coalesce->n_rules)
8488                 return -EINVAL;
8489
8490         new_coalesce.rules = kcalloc(n_rules, sizeof(new_coalesce.rules[0]),
8491                                      GFP_KERNEL);
8492         if (!new_coalesce.rules)
8493                 return -ENOMEM;
8494
8495         new_coalesce.n_rules = n_rules;
8496         i = 0;
8497
8498         nla_for_each_nested(rule, info->attrs[NL80211_ATTR_COALESCE_RULE],
8499                             rem_rule) {
8500                 err = nl80211_parse_coalesce_rule(rdev, rule,
8501                                                   &new_coalesce.rules[i]);
8502                 if (err)
8503                         goto error;
8504
8505                 i++;
8506         }
8507
8508         err = rdev->ops->set_coalesce(&rdev->wiphy, &new_coalesce);
8509         if (err)
8510                 goto error;
8511
8512         n_coalesce = kmemdup(&new_coalesce, sizeof(new_coalesce), GFP_KERNEL);
8513         if (!n_coalesce) {
8514                 err = -ENOMEM;
8515                 goto error;
8516         }
8517         cfg80211_rdev_free_coalesce(rdev);
8518         rdev->coalesce = n_coalesce;
8519
8520         return 0;
8521 error:
8522         for (i = 0; i < new_coalesce.n_rules; i++) {
8523                 tmp_rule = &new_coalesce.rules[i];
8524                 for (j = 0; j < tmp_rule->n_patterns; j++)
8525                         kfree(tmp_rule->patterns[j].mask);
8526                 kfree(tmp_rule->patterns);
8527         }
8528         kfree(new_coalesce.rules);
8529
8530         return err;
8531 }
8532
8533 static int nl80211_set_rekey_data(struct sk_buff *skb, struct genl_info *info)
8534 {
8535         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8536         struct net_device *dev = info->user_ptr[1];
8537         struct wireless_dev *wdev = dev->ieee80211_ptr;
8538         struct nlattr *tb[NUM_NL80211_REKEY_DATA];
8539         struct cfg80211_gtk_rekey_data rekey_data;
8540         int err;
8541
8542         if (!info->attrs[NL80211_ATTR_REKEY_DATA])
8543                 return -EINVAL;
8544
8545         err = nla_parse(tb, MAX_NL80211_REKEY_DATA,
8546                         nla_data(info->attrs[NL80211_ATTR_REKEY_DATA]),
8547                         nla_len(info->attrs[NL80211_ATTR_REKEY_DATA]),
8548                         nl80211_rekey_policy);
8549         if (err)
8550                 return err;
8551
8552         if (nla_len(tb[NL80211_REKEY_DATA_REPLAY_CTR]) != NL80211_REPLAY_CTR_LEN)
8553                 return -ERANGE;
8554         if (nla_len(tb[NL80211_REKEY_DATA_KEK]) != NL80211_KEK_LEN)
8555                 return -ERANGE;
8556         if (nla_len(tb[NL80211_REKEY_DATA_KCK]) != NL80211_KCK_LEN)
8557                 return -ERANGE;
8558
8559         memcpy(rekey_data.kek, nla_data(tb[NL80211_REKEY_DATA_KEK]),
8560                NL80211_KEK_LEN);
8561         memcpy(rekey_data.kck, nla_data(tb[NL80211_REKEY_DATA_KCK]),
8562                NL80211_KCK_LEN);
8563         memcpy(rekey_data.replay_ctr,
8564                nla_data(tb[NL80211_REKEY_DATA_REPLAY_CTR]),
8565                NL80211_REPLAY_CTR_LEN);
8566
8567         wdev_lock(wdev);
8568         if (!wdev->current_bss) {
8569                 err = -ENOTCONN;
8570                 goto out;
8571         }
8572
8573         if (!rdev->ops->set_rekey_data) {
8574                 err = -EOPNOTSUPP;
8575                 goto out;
8576         }
8577
8578         err = rdev_set_rekey_data(rdev, dev, &rekey_data);
8579  out:
8580         wdev_unlock(wdev);
8581         return err;
8582 }
8583
8584 static int nl80211_register_unexpected_frame(struct sk_buff *skb,
8585                                              struct genl_info *info)
8586 {
8587         struct net_device *dev = info->user_ptr[1];
8588         struct wireless_dev *wdev = dev->ieee80211_ptr;
8589
8590         if (wdev->iftype != NL80211_IFTYPE_AP &&
8591             wdev->iftype != NL80211_IFTYPE_P2P_GO)
8592                 return -EINVAL;
8593
8594         if (wdev->ap_unexpected_nlportid)
8595                 return -EBUSY;
8596
8597         wdev->ap_unexpected_nlportid = info->snd_portid;
8598         return 0;
8599 }
8600
8601 static int nl80211_probe_client(struct sk_buff *skb,
8602                                 struct genl_info *info)
8603 {
8604         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8605         struct net_device *dev = info->user_ptr[1];
8606         struct wireless_dev *wdev = dev->ieee80211_ptr;
8607         struct sk_buff *msg;
8608         void *hdr;
8609         const u8 *addr;
8610         u64 cookie;
8611         int err;
8612
8613         if (wdev->iftype != NL80211_IFTYPE_AP &&
8614             wdev->iftype != NL80211_IFTYPE_P2P_GO)
8615                 return -EOPNOTSUPP;
8616
8617         if (!info->attrs[NL80211_ATTR_MAC])
8618                 return -EINVAL;
8619
8620         if (!rdev->ops->probe_client)
8621                 return -EOPNOTSUPP;
8622
8623         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8624         if (!msg)
8625                 return -ENOMEM;
8626
8627         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
8628                              NL80211_CMD_PROBE_CLIENT);
8629         if (!hdr) {
8630                 err = -ENOBUFS;
8631                 goto free_msg;
8632         }
8633
8634         addr = nla_data(info->attrs[NL80211_ATTR_MAC]);
8635
8636         err = rdev_probe_client(rdev, dev, addr, &cookie);
8637         if (err)
8638                 goto free_msg;
8639
8640         if (nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie))
8641                 goto nla_put_failure;
8642
8643         genlmsg_end(msg, hdr);
8644
8645         return genlmsg_reply(msg, info);
8646
8647  nla_put_failure:
8648         err = -ENOBUFS;
8649  free_msg:
8650         nlmsg_free(msg);
8651         return err;
8652 }
8653
8654 static int nl80211_register_beacons(struct sk_buff *skb, struct genl_info *info)
8655 {
8656         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8657         struct cfg80211_beacon_registration *reg, *nreg;
8658         int rv;
8659
8660         if (!(rdev->wiphy.flags & WIPHY_FLAG_REPORTS_OBSS))
8661                 return -EOPNOTSUPP;
8662
8663         nreg = kzalloc(sizeof(*nreg), GFP_KERNEL);
8664         if (!nreg)
8665                 return -ENOMEM;
8666
8667         /* First, check if already registered. */
8668         spin_lock_bh(&rdev->beacon_registrations_lock);
8669         list_for_each_entry(reg, &rdev->beacon_registrations, list) {
8670                 if (reg->nlportid == info->snd_portid) {
8671                         rv = -EALREADY;
8672                         goto out_err;
8673                 }
8674         }
8675         /* Add it to the list */
8676         nreg->nlportid = info->snd_portid;
8677         list_add(&nreg->list, &rdev->beacon_registrations);
8678
8679         spin_unlock_bh(&rdev->beacon_registrations_lock);
8680
8681         return 0;
8682 out_err:
8683         spin_unlock_bh(&rdev->beacon_registrations_lock);
8684         kfree(nreg);
8685         return rv;
8686 }
8687
8688 static int nl80211_start_p2p_device(struct sk_buff *skb, struct genl_info *info)
8689 {
8690         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8691         struct wireless_dev *wdev = info->user_ptr[1];
8692         int err;
8693
8694         if (!rdev->ops->start_p2p_device)
8695                 return -EOPNOTSUPP;
8696
8697         if (wdev->iftype != NL80211_IFTYPE_P2P_DEVICE)
8698                 return -EOPNOTSUPP;
8699
8700         if (wdev->p2p_started)
8701                 return 0;
8702
8703         err = cfg80211_can_add_interface(rdev, wdev->iftype);
8704         if (err)
8705                 return err;
8706
8707         err = rdev_start_p2p_device(rdev, wdev);
8708         if (err)
8709                 return err;
8710
8711         wdev->p2p_started = true;
8712         rdev->opencount++;
8713
8714         return 0;
8715 }
8716
8717 static int nl80211_stop_p2p_device(struct sk_buff *skb, struct genl_info *info)
8718 {
8719         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8720         struct wireless_dev *wdev = info->user_ptr[1];
8721
8722         if (wdev->iftype != NL80211_IFTYPE_P2P_DEVICE)
8723                 return -EOPNOTSUPP;
8724
8725         if (!rdev->ops->stop_p2p_device)
8726                 return -EOPNOTSUPP;
8727
8728         cfg80211_stop_p2p_device(rdev, wdev);
8729
8730         return 0;
8731 }
8732
8733 static int nl80211_get_protocol_features(struct sk_buff *skb,
8734                                          struct genl_info *info)
8735 {
8736         void *hdr;
8737         struct sk_buff *msg;
8738
8739         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
8740         if (!msg)
8741                 return -ENOMEM;
8742
8743         hdr = nl80211hdr_put(msg, info->snd_portid, info->snd_seq, 0,
8744                              NL80211_CMD_GET_PROTOCOL_FEATURES);
8745         if (!hdr)
8746                 goto nla_put_failure;
8747
8748         if (nla_put_u32(msg, NL80211_ATTR_PROTOCOL_FEATURES,
8749                         NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP))
8750                 goto nla_put_failure;
8751
8752         genlmsg_end(msg, hdr);
8753         return genlmsg_reply(msg, info);
8754
8755  nla_put_failure:
8756         kfree_skb(msg);
8757         return -ENOBUFS;
8758 }
8759
8760 static int nl80211_update_ft_ies(struct sk_buff *skb, struct genl_info *info)
8761 {
8762         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8763         struct cfg80211_update_ft_ies_params ft_params;
8764         struct net_device *dev = info->user_ptr[1];
8765
8766         if (!rdev->ops->update_ft_ies)
8767                 return -EOPNOTSUPP;
8768
8769         if (!info->attrs[NL80211_ATTR_MDID] ||
8770             !is_valid_ie_attr(info->attrs[NL80211_ATTR_IE]))
8771                 return -EINVAL;
8772
8773         memset(&ft_params, 0, sizeof(ft_params));
8774         ft_params.md = nla_get_u16(info->attrs[NL80211_ATTR_MDID]);
8775         ft_params.ie = nla_data(info->attrs[NL80211_ATTR_IE]);
8776         ft_params.ie_len = nla_len(info->attrs[NL80211_ATTR_IE]);
8777
8778         return rdev_update_ft_ies(rdev, dev, &ft_params);
8779 }
8780
8781 static int nl80211_crit_protocol_start(struct sk_buff *skb,
8782                                        struct genl_info *info)
8783 {
8784         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8785         struct wireless_dev *wdev = info->user_ptr[1];
8786         enum nl80211_crit_proto_id proto = NL80211_CRIT_PROTO_UNSPEC;
8787         u16 duration;
8788         int ret;
8789
8790         if (!rdev->ops->crit_proto_start)
8791                 return -EOPNOTSUPP;
8792
8793         if (WARN_ON(!rdev->ops->crit_proto_stop))
8794                 return -EINVAL;
8795
8796         if (rdev->crit_proto_nlportid)
8797                 return -EBUSY;
8798
8799         /* determine protocol if provided */
8800         if (info->attrs[NL80211_ATTR_CRIT_PROT_ID])
8801                 proto = nla_get_u16(info->attrs[NL80211_ATTR_CRIT_PROT_ID]);
8802
8803         if (proto >= NUM_NL80211_CRIT_PROTO)
8804                 return -EINVAL;
8805
8806         /* timeout must be provided */
8807         if (!info->attrs[NL80211_ATTR_MAX_CRIT_PROT_DURATION])
8808                 return -EINVAL;
8809
8810         duration =
8811                 nla_get_u16(info->attrs[NL80211_ATTR_MAX_CRIT_PROT_DURATION]);
8812
8813         if (duration > NL80211_CRIT_PROTO_MAX_DURATION)
8814                 return -ERANGE;
8815
8816         ret = rdev_crit_proto_start(rdev, wdev, proto, duration);
8817         if (!ret)
8818                 rdev->crit_proto_nlportid = info->snd_portid;
8819
8820         return ret;
8821 }
8822
8823 static int nl80211_crit_protocol_stop(struct sk_buff *skb,
8824                                       struct genl_info *info)
8825 {
8826         struct cfg80211_registered_device *rdev = info->user_ptr[0];
8827         struct wireless_dev *wdev = info->user_ptr[1];
8828
8829         if (!rdev->ops->crit_proto_stop)
8830                 return -EOPNOTSUPP;
8831
8832         if (rdev->crit_proto_nlportid) {
8833                 rdev->crit_proto_nlportid = 0;
8834                 rdev_crit_proto_stop(rdev, wdev);
8835         }
8836         return 0;
8837 }
8838
8839 #define NL80211_FLAG_NEED_WIPHY         0x01
8840 #define NL80211_FLAG_NEED_NETDEV        0x02
8841 #define NL80211_FLAG_NEED_RTNL          0x04
8842 #define NL80211_FLAG_CHECK_NETDEV_UP    0x08
8843 #define NL80211_FLAG_NEED_NETDEV_UP     (NL80211_FLAG_NEED_NETDEV |\
8844                                          NL80211_FLAG_CHECK_NETDEV_UP)
8845 #define NL80211_FLAG_NEED_WDEV          0x10
8846 /* If a netdev is associated, it must be UP, P2P must be started */
8847 #define NL80211_FLAG_NEED_WDEV_UP       (NL80211_FLAG_NEED_WDEV |\
8848                                          NL80211_FLAG_CHECK_NETDEV_UP)
8849
8850 static int nl80211_pre_doit(struct genl_ops *ops, struct sk_buff *skb,
8851                             struct genl_info *info)
8852 {
8853         struct cfg80211_registered_device *rdev;
8854         struct wireless_dev *wdev;
8855         struct net_device *dev;
8856         bool rtnl = ops->internal_flags & NL80211_FLAG_NEED_RTNL;
8857
8858         if (rtnl)
8859                 rtnl_lock();
8860
8861         if (ops->internal_flags & NL80211_FLAG_NEED_WIPHY) {
8862                 rdev = cfg80211_get_dev_from_info(genl_info_net(info), info);
8863                 if (IS_ERR(rdev)) {
8864                         if (rtnl)
8865                                 rtnl_unlock();
8866                         return PTR_ERR(rdev);
8867                 }
8868                 info->user_ptr[0] = rdev;
8869         } else if (ops->internal_flags & NL80211_FLAG_NEED_NETDEV ||
8870                    ops->internal_flags & NL80211_FLAG_NEED_WDEV) {
8871                 ASSERT_RTNL();
8872
8873                 wdev = __cfg80211_wdev_from_attrs(genl_info_net(info),
8874                                                   info->attrs);
8875                 if (IS_ERR(wdev)) {
8876                         if (rtnl)
8877                                 rtnl_unlock();
8878                         return PTR_ERR(wdev);
8879                 }
8880
8881                 dev = wdev->netdev;
8882                 rdev = wiphy_to_dev(wdev->wiphy);
8883
8884                 if (ops->internal_flags & NL80211_FLAG_NEED_NETDEV) {
8885                         if (!dev) {
8886                                 if (rtnl)
8887                                         rtnl_unlock();
8888                                 return -EINVAL;
8889                         }
8890
8891                         info->user_ptr[1] = dev;
8892                 } else {
8893                         info->user_ptr[1] = wdev;
8894                 }
8895
8896                 if (dev) {
8897                         if (ops->internal_flags & NL80211_FLAG_CHECK_NETDEV_UP &&
8898                             !netif_running(dev)) {
8899                                 if (rtnl)
8900                                         rtnl_unlock();
8901                                 return -ENETDOWN;
8902                         }
8903
8904                         dev_hold(dev);
8905                 } else if (ops->internal_flags & NL80211_FLAG_CHECK_NETDEV_UP) {
8906                         if (!wdev->p2p_started) {
8907                                 if (rtnl)
8908                                         rtnl_unlock();
8909                                 return -ENETDOWN;
8910                         }
8911                 }
8912
8913                 info->user_ptr[0] = rdev;
8914         }
8915
8916         return 0;
8917 }
8918
8919 static void nl80211_post_doit(struct genl_ops *ops, struct sk_buff *skb,
8920                               struct genl_info *info)
8921 {
8922         if (info->user_ptr[1]) {
8923                 if (ops->internal_flags & NL80211_FLAG_NEED_WDEV) {
8924                         struct wireless_dev *wdev = info->user_ptr[1];
8925
8926                         if (wdev->netdev)
8927                                 dev_put(wdev->netdev);
8928                 } else {
8929                         dev_put(info->user_ptr[1]);
8930                 }
8931         }
8932         if (ops->internal_flags & NL80211_FLAG_NEED_RTNL)
8933                 rtnl_unlock();
8934 }
8935
8936 static struct genl_ops nl80211_ops[] = {
8937         {
8938                 .cmd = NL80211_CMD_GET_WIPHY,
8939                 .doit = nl80211_get_wiphy,
8940                 .dumpit = nl80211_dump_wiphy,
8941                 .done = nl80211_dump_wiphy_done,
8942                 .policy = nl80211_policy,
8943                 /* can be retrieved by unprivileged users */
8944                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
8945                                   NL80211_FLAG_NEED_RTNL,
8946         },
8947         {
8948                 .cmd = NL80211_CMD_SET_WIPHY,
8949                 .doit = nl80211_set_wiphy,
8950                 .policy = nl80211_policy,
8951                 .flags = GENL_ADMIN_PERM,
8952                 .internal_flags = NL80211_FLAG_NEED_RTNL,
8953         },
8954         {
8955                 .cmd = NL80211_CMD_GET_INTERFACE,
8956                 .doit = nl80211_get_interface,
8957                 .dumpit = nl80211_dump_interface,
8958                 .policy = nl80211_policy,
8959                 /* can be retrieved by unprivileged users */
8960                 .internal_flags = NL80211_FLAG_NEED_WDEV |
8961                                   NL80211_FLAG_NEED_RTNL,
8962         },
8963         {
8964                 .cmd = NL80211_CMD_SET_INTERFACE,
8965                 .doit = nl80211_set_interface,
8966                 .policy = nl80211_policy,
8967                 .flags = GENL_ADMIN_PERM,
8968                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
8969                                   NL80211_FLAG_NEED_RTNL,
8970         },
8971         {
8972                 .cmd = NL80211_CMD_NEW_INTERFACE,
8973                 .doit = nl80211_new_interface,
8974                 .policy = nl80211_policy,
8975                 .flags = GENL_ADMIN_PERM,
8976                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
8977                                   NL80211_FLAG_NEED_RTNL,
8978         },
8979         {
8980                 .cmd = NL80211_CMD_DEL_INTERFACE,
8981                 .doit = nl80211_del_interface,
8982                 .policy = nl80211_policy,
8983                 .flags = GENL_ADMIN_PERM,
8984                 .internal_flags = NL80211_FLAG_NEED_WDEV |
8985                                   NL80211_FLAG_NEED_RTNL,
8986         },
8987         {
8988                 .cmd = NL80211_CMD_GET_KEY,
8989                 .doit = nl80211_get_key,
8990                 .policy = nl80211_policy,
8991                 .flags = GENL_ADMIN_PERM,
8992                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
8993                                   NL80211_FLAG_NEED_RTNL,
8994         },
8995         {
8996                 .cmd = NL80211_CMD_SET_KEY,
8997                 .doit = nl80211_set_key,
8998                 .policy = nl80211_policy,
8999                 .flags = GENL_ADMIN_PERM,
9000                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9001                                   NL80211_FLAG_NEED_RTNL,
9002         },
9003         {
9004                 .cmd = NL80211_CMD_NEW_KEY,
9005                 .doit = nl80211_new_key,
9006                 .policy = nl80211_policy,
9007                 .flags = GENL_ADMIN_PERM,
9008                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9009                                   NL80211_FLAG_NEED_RTNL,
9010         },
9011         {
9012                 .cmd = NL80211_CMD_DEL_KEY,
9013                 .doit = nl80211_del_key,
9014                 .policy = nl80211_policy,
9015                 .flags = GENL_ADMIN_PERM,
9016                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9017                                   NL80211_FLAG_NEED_RTNL,
9018         },
9019         {
9020                 .cmd = NL80211_CMD_SET_BEACON,
9021                 .policy = nl80211_policy,
9022                 .flags = GENL_ADMIN_PERM,
9023                 .doit = nl80211_set_beacon,
9024                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9025                                   NL80211_FLAG_NEED_RTNL,
9026         },
9027         {
9028                 .cmd = NL80211_CMD_START_AP,
9029                 .policy = nl80211_policy,
9030                 .flags = GENL_ADMIN_PERM,
9031                 .doit = nl80211_start_ap,
9032                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9033                                   NL80211_FLAG_NEED_RTNL,
9034         },
9035         {
9036                 .cmd = NL80211_CMD_STOP_AP,
9037                 .policy = nl80211_policy,
9038                 .flags = GENL_ADMIN_PERM,
9039                 .doit = nl80211_stop_ap,
9040                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9041                                   NL80211_FLAG_NEED_RTNL,
9042         },
9043         {
9044                 .cmd = NL80211_CMD_GET_STATION,
9045                 .doit = nl80211_get_station,
9046                 .dumpit = nl80211_dump_station,
9047                 .policy = nl80211_policy,
9048                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9049                                   NL80211_FLAG_NEED_RTNL,
9050         },
9051         {
9052                 .cmd = NL80211_CMD_SET_STATION,
9053                 .doit = nl80211_set_station,
9054                 .policy = nl80211_policy,
9055                 .flags = GENL_ADMIN_PERM,
9056                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9057                                   NL80211_FLAG_NEED_RTNL,
9058         },
9059         {
9060                 .cmd = NL80211_CMD_NEW_STATION,
9061                 .doit = nl80211_new_station,
9062                 .policy = nl80211_policy,
9063                 .flags = GENL_ADMIN_PERM,
9064                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9065                                   NL80211_FLAG_NEED_RTNL,
9066         },
9067         {
9068                 .cmd = NL80211_CMD_DEL_STATION,
9069                 .doit = nl80211_del_station,
9070                 .policy = nl80211_policy,
9071                 .flags = GENL_ADMIN_PERM,
9072                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9073                                   NL80211_FLAG_NEED_RTNL,
9074         },
9075         {
9076                 .cmd = NL80211_CMD_GET_MPATH,
9077                 .doit = nl80211_get_mpath,
9078                 .dumpit = nl80211_dump_mpath,
9079                 .policy = nl80211_policy,
9080                 .flags = GENL_ADMIN_PERM,
9081                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9082                                   NL80211_FLAG_NEED_RTNL,
9083         },
9084         {
9085                 .cmd = NL80211_CMD_SET_MPATH,
9086                 .doit = nl80211_set_mpath,
9087                 .policy = nl80211_policy,
9088                 .flags = GENL_ADMIN_PERM,
9089                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9090                                   NL80211_FLAG_NEED_RTNL,
9091         },
9092         {
9093                 .cmd = NL80211_CMD_NEW_MPATH,
9094                 .doit = nl80211_new_mpath,
9095                 .policy = nl80211_policy,
9096                 .flags = GENL_ADMIN_PERM,
9097                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9098                                   NL80211_FLAG_NEED_RTNL,
9099         },
9100         {
9101                 .cmd = NL80211_CMD_DEL_MPATH,
9102                 .doit = nl80211_del_mpath,
9103                 .policy = nl80211_policy,
9104                 .flags = GENL_ADMIN_PERM,
9105                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9106                                   NL80211_FLAG_NEED_RTNL,
9107         },
9108         {
9109                 .cmd = NL80211_CMD_SET_BSS,
9110                 .doit = nl80211_set_bss,
9111                 .policy = nl80211_policy,
9112                 .flags = GENL_ADMIN_PERM,
9113                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9114                                   NL80211_FLAG_NEED_RTNL,
9115         },
9116         {
9117                 .cmd = NL80211_CMD_GET_REG,
9118                 .doit = nl80211_get_reg,
9119                 .policy = nl80211_policy,
9120                 .internal_flags = NL80211_FLAG_NEED_RTNL,
9121                 /* can be retrieved by unprivileged users */
9122         },
9123         {
9124                 .cmd = NL80211_CMD_SET_REG,
9125                 .doit = nl80211_set_reg,
9126                 .policy = nl80211_policy,
9127                 .flags = GENL_ADMIN_PERM,
9128                 .internal_flags = NL80211_FLAG_NEED_RTNL,
9129         },
9130         {
9131                 .cmd = NL80211_CMD_REQ_SET_REG,
9132                 .doit = nl80211_req_set_reg,
9133                 .policy = nl80211_policy,
9134                 .flags = GENL_ADMIN_PERM,
9135         },
9136         {
9137                 .cmd = NL80211_CMD_GET_MESH_CONFIG,
9138                 .doit = nl80211_get_mesh_config,
9139                 .policy = nl80211_policy,
9140                 /* can be retrieved by unprivileged users */
9141                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9142                                   NL80211_FLAG_NEED_RTNL,
9143         },
9144         {
9145                 .cmd = NL80211_CMD_SET_MESH_CONFIG,
9146                 .doit = nl80211_update_mesh_config,
9147                 .policy = nl80211_policy,
9148                 .flags = GENL_ADMIN_PERM,
9149                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9150                                   NL80211_FLAG_NEED_RTNL,
9151         },
9152         {
9153                 .cmd = NL80211_CMD_TRIGGER_SCAN,
9154                 .doit = nl80211_trigger_scan,
9155                 .policy = nl80211_policy,
9156                 .flags = GENL_ADMIN_PERM,
9157                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
9158                                   NL80211_FLAG_NEED_RTNL,
9159         },
9160         {
9161                 .cmd = NL80211_CMD_GET_SCAN,
9162                 .policy = nl80211_policy,
9163                 .dumpit = nl80211_dump_scan,
9164         },
9165         {
9166                 .cmd = NL80211_CMD_START_SCHED_SCAN,
9167                 .doit = nl80211_start_sched_scan,
9168                 .policy = nl80211_policy,
9169                 .flags = GENL_ADMIN_PERM,
9170                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9171                                   NL80211_FLAG_NEED_RTNL,
9172         },
9173         {
9174                 .cmd = NL80211_CMD_STOP_SCHED_SCAN,
9175                 .doit = nl80211_stop_sched_scan,
9176                 .policy = nl80211_policy,
9177                 .flags = GENL_ADMIN_PERM,
9178                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9179                                   NL80211_FLAG_NEED_RTNL,
9180         },
9181         {
9182                 .cmd = NL80211_CMD_AUTHENTICATE,
9183                 .doit = nl80211_authenticate,
9184                 .policy = nl80211_policy,
9185                 .flags = GENL_ADMIN_PERM,
9186                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9187                                   NL80211_FLAG_NEED_RTNL,
9188         },
9189         {
9190                 .cmd = NL80211_CMD_ASSOCIATE,
9191                 .doit = nl80211_associate,
9192                 .policy = nl80211_policy,
9193                 .flags = GENL_ADMIN_PERM,
9194                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9195                                   NL80211_FLAG_NEED_RTNL,
9196         },
9197         {
9198                 .cmd = NL80211_CMD_DEAUTHENTICATE,
9199                 .doit = nl80211_deauthenticate,
9200                 .policy = nl80211_policy,
9201                 .flags = GENL_ADMIN_PERM,
9202                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9203                                   NL80211_FLAG_NEED_RTNL,
9204         },
9205         {
9206                 .cmd = NL80211_CMD_DISASSOCIATE,
9207                 .doit = nl80211_disassociate,
9208                 .policy = nl80211_policy,
9209                 .flags = GENL_ADMIN_PERM,
9210                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9211                                   NL80211_FLAG_NEED_RTNL,
9212         },
9213         {
9214                 .cmd = NL80211_CMD_JOIN_IBSS,
9215                 .doit = nl80211_join_ibss,
9216                 .policy = nl80211_policy,
9217                 .flags = GENL_ADMIN_PERM,
9218                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9219                                   NL80211_FLAG_NEED_RTNL,
9220         },
9221         {
9222                 .cmd = NL80211_CMD_LEAVE_IBSS,
9223                 .doit = nl80211_leave_ibss,
9224                 .policy = nl80211_policy,
9225                 .flags = GENL_ADMIN_PERM,
9226                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9227                                   NL80211_FLAG_NEED_RTNL,
9228         },
9229 #ifdef CONFIG_NL80211_TESTMODE
9230         {
9231                 .cmd = NL80211_CMD_TESTMODE,
9232                 .doit = nl80211_testmode_do,
9233                 .dumpit = nl80211_testmode_dump,
9234                 .policy = nl80211_policy,
9235                 .flags = GENL_ADMIN_PERM,
9236                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
9237                                   NL80211_FLAG_NEED_RTNL,
9238         },
9239 #endif
9240         {
9241                 .cmd = NL80211_CMD_CONNECT,
9242                 .doit = nl80211_connect,
9243                 .policy = nl80211_policy,
9244                 .flags = GENL_ADMIN_PERM,
9245                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9246                                   NL80211_FLAG_NEED_RTNL,
9247         },
9248         {
9249                 .cmd = NL80211_CMD_DISCONNECT,
9250                 .doit = nl80211_disconnect,
9251                 .policy = nl80211_policy,
9252                 .flags = GENL_ADMIN_PERM,
9253                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9254                                   NL80211_FLAG_NEED_RTNL,
9255         },
9256         {
9257                 .cmd = NL80211_CMD_SET_WIPHY_NETNS,
9258                 .doit = nl80211_wiphy_netns,
9259                 .policy = nl80211_policy,
9260                 .flags = GENL_ADMIN_PERM,
9261                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
9262                                   NL80211_FLAG_NEED_RTNL,
9263         },
9264         {
9265                 .cmd = NL80211_CMD_GET_SURVEY,
9266                 .policy = nl80211_policy,
9267                 .dumpit = nl80211_dump_survey,
9268         },
9269         {
9270                 .cmd = NL80211_CMD_SET_PMKSA,
9271                 .doit = nl80211_setdel_pmksa,
9272                 .policy = nl80211_policy,
9273                 .flags = GENL_ADMIN_PERM,
9274                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9275                                   NL80211_FLAG_NEED_RTNL,
9276         },
9277         {
9278                 .cmd = NL80211_CMD_DEL_PMKSA,
9279                 .doit = nl80211_setdel_pmksa,
9280                 .policy = nl80211_policy,
9281                 .flags = GENL_ADMIN_PERM,
9282                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9283                                   NL80211_FLAG_NEED_RTNL,
9284         },
9285         {
9286                 .cmd = NL80211_CMD_FLUSH_PMKSA,
9287                 .doit = nl80211_flush_pmksa,
9288                 .policy = nl80211_policy,
9289                 .flags = GENL_ADMIN_PERM,
9290                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9291                                   NL80211_FLAG_NEED_RTNL,
9292         },
9293         {
9294                 .cmd = NL80211_CMD_REMAIN_ON_CHANNEL,
9295                 .doit = nl80211_remain_on_channel,
9296                 .policy = nl80211_policy,
9297                 .flags = GENL_ADMIN_PERM,
9298                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
9299                                   NL80211_FLAG_NEED_RTNL,
9300         },
9301         {
9302                 .cmd = NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL,
9303                 .doit = nl80211_cancel_remain_on_channel,
9304                 .policy = nl80211_policy,
9305                 .flags = GENL_ADMIN_PERM,
9306                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
9307                                   NL80211_FLAG_NEED_RTNL,
9308         },
9309         {
9310                 .cmd = NL80211_CMD_SET_TX_BITRATE_MASK,
9311                 .doit = nl80211_set_tx_bitrate_mask,
9312                 .policy = nl80211_policy,
9313                 .flags = GENL_ADMIN_PERM,
9314                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9315                                   NL80211_FLAG_NEED_RTNL,
9316         },
9317         {
9318                 .cmd = NL80211_CMD_REGISTER_FRAME,
9319                 .doit = nl80211_register_mgmt,
9320                 .policy = nl80211_policy,
9321                 .flags = GENL_ADMIN_PERM,
9322                 .internal_flags = NL80211_FLAG_NEED_WDEV |
9323                                   NL80211_FLAG_NEED_RTNL,
9324         },
9325         {
9326                 .cmd = NL80211_CMD_FRAME,
9327                 .doit = nl80211_tx_mgmt,
9328                 .policy = nl80211_policy,
9329                 .flags = GENL_ADMIN_PERM,
9330                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
9331                                   NL80211_FLAG_NEED_RTNL,
9332         },
9333         {
9334                 .cmd = NL80211_CMD_FRAME_WAIT_CANCEL,
9335                 .doit = nl80211_tx_mgmt_cancel_wait,
9336                 .policy = nl80211_policy,
9337                 .flags = GENL_ADMIN_PERM,
9338                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
9339                                   NL80211_FLAG_NEED_RTNL,
9340         },
9341         {
9342                 .cmd = NL80211_CMD_SET_POWER_SAVE,
9343                 .doit = nl80211_set_power_save,
9344                 .policy = nl80211_policy,
9345                 .flags = GENL_ADMIN_PERM,
9346                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9347                                   NL80211_FLAG_NEED_RTNL,
9348         },
9349         {
9350                 .cmd = NL80211_CMD_GET_POWER_SAVE,
9351                 .doit = nl80211_get_power_save,
9352                 .policy = nl80211_policy,
9353                 /* can be retrieved by unprivileged users */
9354                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9355                                   NL80211_FLAG_NEED_RTNL,
9356         },
9357         {
9358                 .cmd = NL80211_CMD_SET_CQM,
9359                 .doit = nl80211_set_cqm,
9360                 .policy = nl80211_policy,
9361                 .flags = GENL_ADMIN_PERM,
9362                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9363                                   NL80211_FLAG_NEED_RTNL,
9364         },
9365         {
9366                 .cmd = NL80211_CMD_SET_CHANNEL,
9367                 .doit = nl80211_set_channel,
9368                 .policy = nl80211_policy,
9369                 .flags = GENL_ADMIN_PERM,
9370                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9371                                   NL80211_FLAG_NEED_RTNL,
9372         },
9373         {
9374                 .cmd = NL80211_CMD_SET_WDS_PEER,
9375                 .doit = nl80211_set_wds_peer,
9376                 .policy = nl80211_policy,
9377                 .flags = GENL_ADMIN_PERM,
9378                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9379                                   NL80211_FLAG_NEED_RTNL,
9380         },
9381         {
9382                 .cmd = NL80211_CMD_JOIN_MESH,
9383                 .doit = nl80211_join_mesh,
9384                 .policy = nl80211_policy,
9385                 .flags = GENL_ADMIN_PERM,
9386                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9387                                   NL80211_FLAG_NEED_RTNL,
9388         },
9389         {
9390                 .cmd = NL80211_CMD_LEAVE_MESH,
9391                 .doit = nl80211_leave_mesh,
9392                 .policy = nl80211_policy,
9393                 .flags = GENL_ADMIN_PERM,
9394                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9395                                   NL80211_FLAG_NEED_RTNL,
9396         },
9397 #ifdef CONFIG_PM
9398         {
9399                 .cmd = NL80211_CMD_GET_WOWLAN,
9400                 .doit = nl80211_get_wowlan,
9401                 .policy = nl80211_policy,
9402                 /* can be retrieved by unprivileged users */
9403                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
9404                                   NL80211_FLAG_NEED_RTNL,
9405         },
9406         {
9407                 .cmd = NL80211_CMD_SET_WOWLAN,
9408                 .doit = nl80211_set_wowlan,
9409                 .policy = nl80211_policy,
9410                 .flags = GENL_ADMIN_PERM,
9411                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
9412                                   NL80211_FLAG_NEED_RTNL,
9413         },
9414 #endif
9415         {
9416                 .cmd = NL80211_CMD_SET_REKEY_OFFLOAD,
9417                 .doit = nl80211_set_rekey_data,
9418                 .policy = nl80211_policy,
9419                 .flags = GENL_ADMIN_PERM,
9420                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9421                                   NL80211_FLAG_NEED_RTNL,
9422         },
9423         {
9424                 .cmd = NL80211_CMD_TDLS_MGMT,
9425                 .doit = nl80211_tdls_mgmt,
9426                 .policy = nl80211_policy,
9427                 .flags = GENL_ADMIN_PERM,
9428                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9429                                   NL80211_FLAG_NEED_RTNL,
9430         },
9431         {
9432                 .cmd = NL80211_CMD_TDLS_OPER,
9433                 .doit = nl80211_tdls_oper,
9434                 .policy = nl80211_policy,
9435                 .flags = GENL_ADMIN_PERM,
9436                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9437                                   NL80211_FLAG_NEED_RTNL,
9438         },
9439         {
9440                 .cmd = NL80211_CMD_UNEXPECTED_FRAME,
9441                 .doit = nl80211_register_unexpected_frame,
9442                 .policy = nl80211_policy,
9443                 .flags = GENL_ADMIN_PERM,
9444                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9445                                   NL80211_FLAG_NEED_RTNL,
9446         },
9447         {
9448                 .cmd = NL80211_CMD_PROBE_CLIENT,
9449                 .doit = nl80211_probe_client,
9450                 .policy = nl80211_policy,
9451                 .flags = GENL_ADMIN_PERM,
9452                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9453                                   NL80211_FLAG_NEED_RTNL,
9454         },
9455         {
9456                 .cmd = NL80211_CMD_REGISTER_BEACONS,
9457                 .doit = nl80211_register_beacons,
9458                 .policy = nl80211_policy,
9459                 .flags = GENL_ADMIN_PERM,
9460                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
9461                                   NL80211_FLAG_NEED_RTNL,
9462         },
9463         {
9464                 .cmd = NL80211_CMD_SET_NOACK_MAP,
9465                 .doit = nl80211_set_noack_map,
9466                 .policy = nl80211_policy,
9467                 .flags = GENL_ADMIN_PERM,
9468                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9469                                   NL80211_FLAG_NEED_RTNL,
9470         },
9471         {
9472                 .cmd = NL80211_CMD_START_P2P_DEVICE,
9473                 .doit = nl80211_start_p2p_device,
9474                 .policy = nl80211_policy,
9475                 .flags = GENL_ADMIN_PERM,
9476                 .internal_flags = NL80211_FLAG_NEED_WDEV |
9477                                   NL80211_FLAG_NEED_RTNL,
9478         },
9479         {
9480                 .cmd = NL80211_CMD_STOP_P2P_DEVICE,
9481                 .doit = nl80211_stop_p2p_device,
9482                 .policy = nl80211_policy,
9483                 .flags = GENL_ADMIN_PERM,
9484                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
9485                                   NL80211_FLAG_NEED_RTNL,
9486         },
9487         {
9488                 .cmd = NL80211_CMD_SET_MCAST_RATE,
9489                 .doit = nl80211_set_mcast_rate,
9490                 .policy = nl80211_policy,
9491                 .flags = GENL_ADMIN_PERM,
9492                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9493                                   NL80211_FLAG_NEED_RTNL,
9494         },
9495         {
9496                 .cmd = NL80211_CMD_SET_MAC_ACL,
9497                 .doit = nl80211_set_mac_acl,
9498                 .policy = nl80211_policy,
9499                 .flags = GENL_ADMIN_PERM,
9500                 .internal_flags = NL80211_FLAG_NEED_NETDEV |
9501                                   NL80211_FLAG_NEED_RTNL,
9502         },
9503         {
9504                 .cmd = NL80211_CMD_RADAR_DETECT,
9505                 .doit = nl80211_start_radar_detection,
9506                 .policy = nl80211_policy,
9507                 .flags = GENL_ADMIN_PERM,
9508                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9509                                   NL80211_FLAG_NEED_RTNL,
9510         },
9511         {
9512                 .cmd = NL80211_CMD_GET_PROTOCOL_FEATURES,
9513                 .doit = nl80211_get_protocol_features,
9514                 .policy = nl80211_policy,
9515         },
9516         {
9517                 .cmd = NL80211_CMD_UPDATE_FT_IES,
9518                 .doit = nl80211_update_ft_ies,
9519                 .policy = nl80211_policy,
9520                 .flags = GENL_ADMIN_PERM,
9521                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9522                                   NL80211_FLAG_NEED_RTNL,
9523         },
9524         {
9525                 .cmd = NL80211_CMD_CRIT_PROTOCOL_START,
9526                 .doit = nl80211_crit_protocol_start,
9527                 .policy = nl80211_policy,
9528                 .flags = GENL_ADMIN_PERM,
9529                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
9530                                   NL80211_FLAG_NEED_RTNL,
9531         },
9532         {
9533                 .cmd = NL80211_CMD_CRIT_PROTOCOL_STOP,
9534                 .doit = nl80211_crit_protocol_stop,
9535                 .policy = nl80211_policy,
9536                 .flags = GENL_ADMIN_PERM,
9537                 .internal_flags = NL80211_FLAG_NEED_WDEV_UP |
9538                                   NL80211_FLAG_NEED_RTNL,
9539         },
9540         {
9541                 .cmd = NL80211_CMD_GET_COALESCE,
9542                 .doit = nl80211_get_coalesce,
9543                 .policy = nl80211_policy,
9544                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
9545                                   NL80211_FLAG_NEED_RTNL,
9546         },
9547         {
9548                 .cmd = NL80211_CMD_SET_COALESCE,
9549                 .doit = nl80211_set_coalesce,
9550                 .policy = nl80211_policy,
9551                 .flags = GENL_ADMIN_PERM,
9552                 .internal_flags = NL80211_FLAG_NEED_WIPHY |
9553                                   NL80211_FLAG_NEED_RTNL,
9554         },
9555         {
9556                 .cmd = NL80211_CMD_CHANNEL_SWITCH,
9557                 .doit = nl80211_channel_switch,
9558                 .policy = nl80211_policy,
9559                 .flags = GENL_ADMIN_PERM,
9560                 .internal_flags = NL80211_FLAG_NEED_NETDEV_UP |
9561                                   NL80211_FLAG_NEED_RTNL,
9562         },
9563 };
9564
9565 static struct genl_multicast_group nl80211_mlme_mcgrp = {
9566         .name = "mlme",
9567 };
9568
9569 /* multicast groups */
9570 static struct genl_multicast_group nl80211_config_mcgrp = {
9571         .name = "config",
9572 };
9573 static struct genl_multicast_group nl80211_scan_mcgrp = {
9574         .name = "scan",
9575 };
9576 static struct genl_multicast_group nl80211_regulatory_mcgrp = {
9577         .name = "regulatory",
9578 };
9579
9580 /* notification functions */
9581
9582 void nl80211_notify_dev_rename(struct cfg80211_registered_device *rdev)
9583 {
9584         struct sk_buff *msg;
9585         struct nl80211_dump_wiphy_state state = {};
9586
9587         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9588         if (!msg)
9589                 return;
9590
9591         if (nl80211_send_wiphy(rdev, msg, 0, 0, 0, &state) < 0) {
9592                 nlmsg_free(msg);
9593                 return;
9594         }
9595
9596         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9597                                 nl80211_config_mcgrp.id, GFP_KERNEL);
9598 }
9599
9600 static int nl80211_add_scan_req(struct sk_buff *msg,
9601                                 struct cfg80211_registered_device *rdev)
9602 {
9603         struct cfg80211_scan_request *req = rdev->scan_req;
9604         struct nlattr *nest;
9605         int i;
9606
9607         if (WARN_ON(!req))
9608                 return 0;
9609
9610         nest = nla_nest_start(msg, NL80211_ATTR_SCAN_SSIDS);
9611         if (!nest)
9612                 goto nla_put_failure;
9613         for (i = 0; i < req->n_ssids; i++) {
9614                 if (nla_put(msg, i, req->ssids[i].ssid_len, req->ssids[i].ssid))
9615                         goto nla_put_failure;
9616         }
9617         nla_nest_end(msg, nest);
9618
9619         nest = nla_nest_start(msg, NL80211_ATTR_SCAN_FREQUENCIES);
9620         if (!nest)
9621                 goto nla_put_failure;
9622         for (i = 0; i < req->n_channels; i++) {
9623                 if (nla_put_u32(msg, i, req->channels[i]->center_freq))
9624                         goto nla_put_failure;
9625         }
9626         nla_nest_end(msg, nest);
9627
9628         if (req->ie &&
9629             nla_put(msg, NL80211_ATTR_IE, req->ie_len, req->ie))
9630                 goto nla_put_failure;
9631
9632         if (req->flags)
9633                 nla_put_u32(msg, NL80211_ATTR_SCAN_FLAGS, req->flags);
9634
9635         return 0;
9636  nla_put_failure:
9637         return -ENOBUFS;
9638 }
9639
9640 static int nl80211_send_scan_msg(struct sk_buff *msg,
9641                                  struct cfg80211_registered_device *rdev,
9642                                  struct wireless_dev *wdev,
9643                                  u32 portid, u32 seq, int flags,
9644                                  u32 cmd)
9645 {
9646         void *hdr;
9647
9648         hdr = nl80211hdr_put(msg, portid, seq, flags, cmd);
9649         if (!hdr)
9650                 return -1;
9651
9652         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
9653             (wdev->netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
9654                                          wdev->netdev->ifindex)) ||
9655             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)))
9656                 goto nla_put_failure;
9657
9658         /* ignore errors and send incomplete event anyway */
9659         nl80211_add_scan_req(msg, rdev);
9660
9661         return genlmsg_end(msg, hdr);
9662
9663  nla_put_failure:
9664         genlmsg_cancel(msg, hdr);
9665         return -EMSGSIZE;
9666 }
9667
9668 static int
9669 nl80211_send_sched_scan_msg(struct sk_buff *msg,
9670                             struct cfg80211_registered_device *rdev,
9671                             struct net_device *netdev,
9672                             u32 portid, u32 seq, int flags, u32 cmd)
9673 {
9674         void *hdr;
9675
9676         hdr = nl80211hdr_put(msg, portid, seq, flags, cmd);
9677         if (!hdr)
9678                 return -1;
9679
9680         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
9681             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
9682                 goto nla_put_failure;
9683
9684         return genlmsg_end(msg, hdr);
9685
9686  nla_put_failure:
9687         genlmsg_cancel(msg, hdr);
9688         return -EMSGSIZE;
9689 }
9690
9691 void nl80211_send_scan_start(struct cfg80211_registered_device *rdev,
9692                              struct wireless_dev *wdev)
9693 {
9694         struct sk_buff *msg;
9695
9696         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9697         if (!msg)
9698                 return;
9699
9700         if (nl80211_send_scan_msg(msg, rdev, wdev, 0, 0, 0,
9701                                   NL80211_CMD_TRIGGER_SCAN) < 0) {
9702                 nlmsg_free(msg);
9703                 return;
9704         }
9705
9706         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9707                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
9708 }
9709
9710 void nl80211_send_scan_done(struct cfg80211_registered_device *rdev,
9711                             struct wireless_dev *wdev)
9712 {
9713         struct sk_buff *msg;
9714
9715         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9716         if (!msg)
9717                 return;
9718
9719         if (nl80211_send_scan_msg(msg, rdev, wdev, 0, 0, 0,
9720                                   NL80211_CMD_NEW_SCAN_RESULTS) < 0) {
9721                 nlmsg_free(msg);
9722                 return;
9723         }
9724
9725         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9726                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
9727 }
9728
9729 void nl80211_send_scan_aborted(struct cfg80211_registered_device *rdev,
9730                                struct wireless_dev *wdev)
9731 {
9732         struct sk_buff *msg;
9733
9734         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9735         if (!msg)
9736                 return;
9737
9738         if (nl80211_send_scan_msg(msg, rdev, wdev, 0, 0, 0,
9739                                   NL80211_CMD_SCAN_ABORTED) < 0) {
9740                 nlmsg_free(msg);
9741                 return;
9742         }
9743
9744         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9745                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
9746 }
9747
9748 void nl80211_send_sched_scan_results(struct cfg80211_registered_device *rdev,
9749                                      struct net_device *netdev)
9750 {
9751         struct sk_buff *msg;
9752
9753         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9754         if (!msg)
9755                 return;
9756
9757         if (nl80211_send_sched_scan_msg(msg, rdev, netdev, 0, 0, 0,
9758                                         NL80211_CMD_SCHED_SCAN_RESULTS) < 0) {
9759                 nlmsg_free(msg);
9760                 return;
9761         }
9762
9763         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9764                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
9765 }
9766
9767 void nl80211_send_sched_scan(struct cfg80211_registered_device *rdev,
9768                              struct net_device *netdev, u32 cmd)
9769 {
9770         struct sk_buff *msg;
9771
9772         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9773         if (!msg)
9774                 return;
9775
9776         if (nl80211_send_sched_scan_msg(msg, rdev, netdev, 0, 0, 0, cmd) < 0) {
9777                 nlmsg_free(msg);
9778                 return;
9779         }
9780
9781         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9782                                 nl80211_scan_mcgrp.id, GFP_KERNEL);
9783 }
9784
9785 /*
9786  * This can happen on global regulatory changes or device specific settings
9787  * based on custom world regulatory domains.
9788  */
9789 void nl80211_send_reg_change_event(struct regulatory_request *request)
9790 {
9791         struct sk_buff *msg;
9792         void *hdr;
9793
9794         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9795         if (!msg)
9796                 return;
9797
9798         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_REG_CHANGE);
9799         if (!hdr) {
9800                 nlmsg_free(msg);
9801                 return;
9802         }
9803
9804         /* Userspace can always count this one always being set */
9805         if (nla_put_u8(msg, NL80211_ATTR_REG_INITIATOR, request->initiator))
9806                 goto nla_put_failure;
9807
9808         if (request->alpha2[0] == '0' && request->alpha2[1] == '0') {
9809                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
9810                                NL80211_REGDOM_TYPE_WORLD))
9811                         goto nla_put_failure;
9812         } else if (request->alpha2[0] == '9' && request->alpha2[1] == '9') {
9813                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
9814                                NL80211_REGDOM_TYPE_CUSTOM_WORLD))
9815                         goto nla_put_failure;
9816         } else if ((request->alpha2[0] == '9' && request->alpha2[1] == '8') ||
9817                    request->intersect) {
9818                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
9819                                NL80211_REGDOM_TYPE_INTERSECTION))
9820                         goto nla_put_failure;
9821         } else {
9822                 if (nla_put_u8(msg, NL80211_ATTR_REG_TYPE,
9823                                NL80211_REGDOM_TYPE_COUNTRY) ||
9824                     nla_put_string(msg, NL80211_ATTR_REG_ALPHA2,
9825                                    request->alpha2))
9826                         goto nla_put_failure;
9827         }
9828
9829         if (request->wiphy_idx != WIPHY_IDX_INVALID &&
9830             nla_put_u32(msg, NL80211_ATTR_WIPHY, request->wiphy_idx))
9831                 goto nla_put_failure;
9832
9833         genlmsg_end(msg, hdr);
9834
9835         rcu_read_lock();
9836         genlmsg_multicast_allns(msg, 0, nl80211_regulatory_mcgrp.id,
9837                                 GFP_ATOMIC);
9838         rcu_read_unlock();
9839
9840         return;
9841
9842 nla_put_failure:
9843         genlmsg_cancel(msg, hdr);
9844         nlmsg_free(msg);
9845 }
9846
9847 static void nl80211_send_mlme_event(struct cfg80211_registered_device *rdev,
9848                                     struct net_device *netdev,
9849                                     const u8 *buf, size_t len,
9850                                     enum nl80211_commands cmd, gfp_t gfp)
9851 {
9852         struct sk_buff *msg;
9853         void *hdr;
9854
9855         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
9856         if (!msg)
9857                 return;
9858
9859         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
9860         if (!hdr) {
9861                 nlmsg_free(msg);
9862                 return;
9863         }
9864
9865         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
9866             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
9867             nla_put(msg, NL80211_ATTR_FRAME, len, buf))
9868                 goto nla_put_failure;
9869
9870         genlmsg_end(msg, hdr);
9871
9872         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9873                                 nl80211_mlme_mcgrp.id, gfp);
9874         return;
9875
9876  nla_put_failure:
9877         genlmsg_cancel(msg, hdr);
9878         nlmsg_free(msg);
9879 }
9880
9881 void nl80211_send_rx_auth(struct cfg80211_registered_device *rdev,
9882                           struct net_device *netdev, const u8 *buf,
9883                           size_t len, gfp_t gfp)
9884 {
9885         nl80211_send_mlme_event(rdev, netdev, buf, len,
9886                                 NL80211_CMD_AUTHENTICATE, gfp);
9887 }
9888
9889 void nl80211_send_rx_assoc(struct cfg80211_registered_device *rdev,
9890                            struct net_device *netdev, const u8 *buf,
9891                            size_t len, gfp_t gfp)
9892 {
9893         nl80211_send_mlme_event(rdev, netdev, buf, len,
9894                                 NL80211_CMD_ASSOCIATE, gfp);
9895 }
9896
9897 void nl80211_send_deauth(struct cfg80211_registered_device *rdev,
9898                          struct net_device *netdev, const u8 *buf,
9899                          size_t len, gfp_t gfp)
9900 {
9901         nl80211_send_mlme_event(rdev, netdev, buf, len,
9902                                 NL80211_CMD_DEAUTHENTICATE, gfp);
9903 }
9904
9905 void nl80211_send_disassoc(struct cfg80211_registered_device *rdev,
9906                            struct net_device *netdev, const u8 *buf,
9907                            size_t len, gfp_t gfp)
9908 {
9909         nl80211_send_mlme_event(rdev, netdev, buf, len,
9910                                 NL80211_CMD_DISASSOCIATE, gfp);
9911 }
9912
9913 void cfg80211_rx_unprot_mlme_mgmt(struct net_device *dev, const u8 *buf,
9914                                   size_t len)
9915 {
9916         struct wireless_dev *wdev = dev->ieee80211_ptr;
9917         struct wiphy *wiphy = wdev->wiphy;
9918         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
9919         const struct ieee80211_mgmt *mgmt = (void *)buf;
9920         u32 cmd;
9921
9922         if (WARN_ON(len < 2))
9923                 return;
9924
9925         if (ieee80211_is_deauth(mgmt->frame_control))
9926                 cmd = NL80211_CMD_UNPROT_DEAUTHENTICATE;
9927         else
9928                 cmd = NL80211_CMD_UNPROT_DISASSOCIATE;
9929
9930         trace_cfg80211_rx_unprot_mlme_mgmt(dev, buf, len);
9931         nl80211_send_mlme_event(rdev, dev, buf, len, cmd, GFP_ATOMIC);
9932 }
9933 EXPORT_SYMBOL(cfg80211_rx_unprot_mlme_mgmt);
9934
9935 static void nl80211_send_mlme_timeout(struct cfg80211_registered_device *rdev,
9936                                       struct net_device *netdev, int cmd,
9937                                       const u8 *addr, gfp_t gfp)
9938 {
9939         struct sk_buff *msg;
9940         void *hdr;
9941
9942         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
9943         if (!msg)
9944                 return;
9945
9946         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
9947         if (!hdr) {
9948                 nlmsg_free(msg);
9949                 return;
9950         }
9951
9952         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
9953             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
9954             nla_put_flag(msg, NL80211_ATTR_TIMED_OUT) ||
9955             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr))
9956                 goto nla_put_failure;
9957
9958         genlmsg_end(msg, hdr);
9959
9960         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
9961                                 nl80211_mlme_mcgrp.id, gfp);
9962         return;
9963
9964  nla_put_failure:
9965         genlmsg_cancel(msg, hdr);
9966         nlmsg_free(msg);
9967 }
9968
9969 void nl80211_send_auth_timeout(struct cfg80211_registered_device *rdev,
9970                                struct net_device *netdev, const u8 *addr,
9971                                gfp_t gfp)
9972 {
9973         nl80211_send_mlme_timeout(rdev, netdev, NL80211_CMD_AUTHENTICATE,
9974                                   addr, gfp);
9975 }
9976
9977 void nl80211_send_assoc_timeout(struct cfg80211_registered_device *rdev,
9978                                 struct net_device *netdev, const u8 *addr,
9979                                 gfp_t gfp)
9980 {
9981         nl80211_send_mlme_timeout(rdev, netdev, NL80211_CMD_ASSOCIATE,
9982                                   addr, gfp);
9983 }
9984
9985 void nl80211_send_connect_result(struct cfg80211_registered_device *rdev,
9986                                  struct net_device *netdev, const u8 *bssid,
9987                                  const u8 *req_ie, size_t req_ie_len,
9988                                  const u8 *resp_ie, size_t resp_ie_len,
9989                                  u16 status, gfp_t gfp)
9990 {
9991         struct sk_buff *msg;
9992         void *hdr;
9993
9994         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
9995         if (!msg)
9996                 return;
9997
9998         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CONNECT);
9999         if (!hdr) {
10000                 nlmsg_free(msg);
10001                 return;
10002         }
10003
10004         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10005             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
10006             (bssid && nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid)) ||
10007             nla_put_u16(msg, NL80211_ATTR_STATUS_CODE, status) ||
10008             (req_ie &&
10009              nla_put(msg, NL80211_ATTR_REQ_IE, req_ie_len, req_ie)) ||
10010             (resp_ie &&
10011              nla_put(msg, NL80211_ATTR_RESP_IE, resp_ie_len, resp_ie)))
10012                 goto nla_put_failure;
10013
10014         genlmsg_end(msg, hdr);
10015
10016         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10017                                 nl80211_mlme_mcgrp.id, gfp);
10018         return;
10019
10020  nla_put_failure:
10021         genlmsg_cancel(msg, hdr);
10022         nlmsg_free(msg);
10023
10024 }
10025
10026 void nl80211_send_roamed(struct cfg80211_registered_device *rdev,
10027                          struct net_device *netdev, const u8 *bssid,
10028                          const u8 *req_ie, size_t req_ie_len,
10029                          const u8 *resp_ie, size_t resp_ie_len, gfp_t gfp)
10030 {
10031         struct sk_buff *msg;
10032         void *hdr;
10033
10034         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10035         if (!msg)
10036                 return;
10037
10038         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_ROAM);
10039         if (!hdr) {
10040                 nlmsg_free(msg);
10041                 return;
10042         }
10043
10044         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10045             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
10046             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid) ||
10047             (req_ie &&
10048              nla_put(msg, NL80211_ATTR_REQ_IE, req_ie_len, req_ie)) ||
10049             (resp_ie &&
10050              nla_put(msg, NL80211_ATTR_RESP_IE, resp_ie_len, resp_ie)))
10051                 goto nla_put_failure;
10052
10053         genlmsg_end(msg, hdr);
10054
10055         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10056                                 nl80211_mlme_mcgrp.id, gfp);
10057         return;
10058
10059  nla_put_failure:
10060         genlmsg_cancel(msg, hdr);
10061         nlmsg_free(msg);
10062
10063 }
10064
10065 void nl80211_send_disconnected(struct cfg80211_registered_device *rdev,
10066                                struct net_device *netdev, u16 reason,
10067                                const u8 *ie, size_t ie_len, bool from_ap)
10068 {
10069         struct sk_buff *msg;
10070         void *hdr;
10071
10072         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
10073         if (!msg)
10074                 return;
10075
10076         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_DISCONNECT);
10077         if (!hdr) {
10078                 nlmsg_free(msg);
10079                 return;
10080         }
10081
10082         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10083             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
10084             (from_ap && reason &&
10085              nla_put_u16(msg, NL80211_ATTR_REASON_CODE, reason)) ||
10086             (from_ap &&
10087              nla_put_flag(msg, NL80211_ATTR_DISCONNECTED_BY_AP)) ||
10088             (ie && nla_put(msg, NL80211_ATTR_IE, ie_len, ie)))
10089                 goto nla_put_failure;
10090
10091         genlmsg_end(msg, hdr);
10092
10093         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10094                                 nl80211_mlme_mcgrp.id, GFP_KERNEL);
10095         return;
10096
10097  nla_put_failure:
10098         genlmsg_cancel(msg, hdr);
10099         nlmsg_free(msg);
10100
10101 }
10102
10103 void nl80211_send_ibss_bssid(struct cfg80211_registered_device *rdev,
10104                              struct net_device *netdev, const u8 *bssid,
10105                              gfp_t gfp)
10106 {
10107         struct sk_buff *msg;
10108         void *hdr;
10109
10110         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10111         if (!msg)
10112                 return;
10113
10114         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_JOIN_IBSS);
10115         if (!hdr) {
10116                 nlmsg_free(msg);
10117                 return;
10118         }
10119
10120         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10121             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
10122             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid))
10123                 goto nla_put_failure;
10124
10125         genlmsg_end(msg, hdr);
10126
10127         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10128                                 nl80211_mlme_mcgrp.id, gfp);
10129         return;
10130
10131  nla_put_failure:
10132         genlmsg_cancel(msg, hdr);
10133         nlmsg_free(msg);
10134 }
10135
10136 void cfg80211_notify_new_peer_candidate(struct net_device *dev, const u8 *addr,
10137                                         const u8* ie, u8 ie_len, gfp_t gfp)
10138 {
10139         struct wireless_dev *wdev = dev->ieee80211_ptr;
10140         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
10141         struct sk_buff *msg;
10142         void *hdr;
10143
10144         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_MESH_POINT))
10145                 return;
10146
10147         trace_cfg80211_notify_new_peer_candidate(dev, addr);
10148
10149         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10150         if (!msg)
10151                 return;
10152
10153         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NEW_PEER_CANDIDATE);
10154         if (!hdr) {
10155                 nlmsg_free(msg);
10156                 return;
10157         }
10158
10159         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10160             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
10161             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr) ||
10162             (ie_len && ie &&
10163              nla_put(msg, NL80211_ATTR_IE, ie_len , ie)))
10164                 goto nla_put_failure;
10165
10166         genlmsg_end(msg, hdr);
10167
10168         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10169                                 nl80211_mlme_mcgrp.id, gfp);
10170         return;
10171
10172  nla_put_failure:
10173         genlmsg_cancel(msg, hdr);
10174         nlmsg_free(msg);
10175 }
10176 EXPORT_SYMBOL(cfg80211_notify_new_peer_candidate);
10177
10178 void nl80211_michael_mic_failure(struct cfg80211_registered_device *rdev,
10179                                  struct net_device *netdev, const u8 *addr,
10180                                  enum nl80211_key_type key_type, int key_id,
10181                                  const u8 *tsc, gfp_t gfp)
10182 {
10183         struct sk_buff *msg;
10184         void *hdr;
10185
10186         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10187         if (!msg)
10188                 return;
10189
10190         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_MICHAEL_MIC_FAILURE);
10191         if (!hdr) {
10192                 nlmsg_free(msg);
10193                 return;
10194         }
10195
10196         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10197             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
10198             (addr && nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr)) ||
10199             nla_put_u32(msg, NL80211_ATTR_KEY_TYPE, key_type) ||
10200             (key_id != -1 &&
10201              nla_put_u8(msg, NL80211_ATTR_KEY_IDX, key_id)) ||
10202             (tsc && nla_put(msg, NL80211_ATTR_KEY_SEQ, 6, tsc)))
10203                 goto nla_put_failure;
10204
10205         genlmsg_end(msg, hdr);
10206
10207         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10208                                 nl80211_mlme_mcgrp.id, gfp);
10209         return;
10210
10211  nla_put_failure:
10212         genlmsg_cancel(msg, hdr);
10213         nlmsg_free(msg);
10214 }
10215
10216 void nl80211_send_beacon_hint_event(struct wiphy *wiphy,
10217                                     struct ieee80211_channel *channel_before,
10218                                     struct ieee80211_channel *channel_after)
10219 {
10220         struct sk_buff *msg;
10221         void *hdr;
10222         struct nlattr *nl_freq;
10223
10224         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_ATOMIC);
10225         if (!msg)
10226                 return;
10227
10228         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_REG_BEACON_HINT);
10229         if (!hdr) {
10230                 nlmsg_free(msg);
10231                 return;
10232         }
10233
10234         /*
10235          * Since we are applying the beacon hint to a wiphy we know its
10236          * wiphy_idx is valid
10237          */
10238         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, get_wiphy_idx(wiphy)))
10239                 goto nla_put_failure;
10240
10241         /* Before */
10242         nl_freq = nla_nest_start(msg, NL80211_ATTR_FREQ_BEFORE);
10243         if (!nl_freq)
10244                 goto nla_put_failure;
10245         if (nl80211_msg_put_channel(msg, channel_before, false))
10246                 goto nla_put_failure;
10247         nla_nest_end(msg, nl_freq);
10248
10249         /* After */
10250         nl_freq = nla_nest_start(msg, NL80211_ATTR_FREQ_AFTER);
10251         if (!nl_freq)
10252                 goto nla_put_failure;
10253         if (nl80211_msg_put_channel(msg, channel_after, false))
10254                 goto nla_put_failure;
10255         nla_nest_end(msg, nl_freq);
10256
10257         genlmsg_end(msg, hdr);
10258
10259         rcu_read_lock();
10260         genlmsg_multicast_allns(msg, 0, nl80211_regulatory_mcgrp.id,
10261                                 GFP_ATOMIC);
10262         rcu_read_unlock();
10263
10264         return;
10265
10266 nla_put_failure:
10267         genlmsg_cancel(msg, hdr);
10268         nlmsg_free(msg);
10269 }
10270
10271 static void nl80211_send_remain_on_chan_event(
10272         int cmd, struct cfg80211_registered_device *rdev,
10273         struct wireless_dev *wdev, u64 cookie,
10274         struct ieee80211_channel *chan,
10275         unsigned int duration, gfp_t gfp)
10276 {
10277         struct sk_buff *msg;
10278         void *hdr;
10279
10280         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10281         if (!msg)
10282                 return;
10283
10284         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
10285         if (!hdr) {
10286                 nlmsg_free(msg);
10287                 return;
10288         }
10289
10290         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10291             (wdev->netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
10292                                          wdev->netdev->ifindex)) ||
10293             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)) ||
10294             nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ, chan->center_freq) ||
10295             nla_put_u32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
10296                         NL80211_CHAN_NO_HT) ||
10297             nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie))
10298                 goto nla_put_failure;
10299
10300         if (cmd == NL80211_CMD_REMAIN_ON_CHANNEL &&
10301             nla_put_u32(msg, NL80211_ATTR_DURATION, duration))
10302                 goto nla_put_failure;
10303
10304         genlmsg_end(msg, hdr);
10305
10306         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10307                                 nl80211_mlme_mcgrp.id, gfp);
10308         return;
10309
10310  nla_put_failure:
10311         genlmsg_cancel(msg, hdr);
10312         nlmsg_free(msg);
10313 }
10314
10315 void cfg80211_ready_on_channel(struct wireless_dev *wdev, u64 cookie,
10316                                struct ieee80211_channel *chan,
10317                                unsigned int duration, gfp_t gfp)
10318 {
10319         struct wiphy *wiphy = wdev->wiphy;
10320         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10321
10322         trace_cfg80211_ready_on_channel(wdev, cookie, chan, duration);
10323         nl80211_send_remain_on_chan_event(NL80211_CMD_REMAIN_ON_CHANNEL,
10324                                           rdev, wdev, cookie, chan,
10325                                           duration, gfp);
10326 }
10327 EXPORT_SYMBOL(cfg80211_ready_on_channel);
10328
10329 void cfg80211_remain_on_channel_expired(struct wireless_dev *wdev, u64 cookie,
10330                                         struct ieee80211_channel *chan,
10331                                         gfp_t gfp)
10332 {
10333         struct wiphy *wiphy = wdev->wiphy;
10334         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10335
10336         trace_cfg80211_ready_on_channel_expired(wdev, cookie, chan);
10337         nl80211_send_remain_on_chan_event(NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL,
10338                                           rdev, wdev, cookie, chan, 0, gfp);
10339 }
10340 EXPORT_SYMBOL(cfg80211_remain_on_channel_expired);
10341
10342 void cfg80211_new_sta(struct net_device *dev, const u8 *mac_addr,
10343                       struct station_info *sinfo, gfp_t gfp)
10344 {
10345         struct wiphy *wiphy = dev->ieee80211_ptr->wiphy;
10346         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10347         struct sk_buff *msg;
10348
10349         trace_cfg80211_new_sta(dev, mac_addr, sinfo);
10350
10351         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10352         if (!msg)
10353                 return;
10354
10355         if (nl80211_send_station(msg, 0, 0, 0,
10356                                  rdev, dev, mac_addr, sinfo) < 0) {
10357                 nlmsg_free(msg);
10358                 return;
10359         }
10360
10361         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10362                                 nl80211_mlme_mcgrp.id, gfp);
10363 }
10364 EXPORT_SYMBOL(cfg80211_new_sta);
10365
10366 void cfg80211_del_sta(struct net_device *dev, const u8 *mac_addr, gfp_t gfp)
10367 {
10368         struct wiphy *wiphy = dev->ieee80211_ptr->wiphy;
10369         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10370         struct sk_buff *msg;
10371         void *hdr;
10372
10373         trace_cfg80211_del_sta(dev, mac_addr);
10374
10375         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10376         if (!msg)
10377                 return;
10378
10379         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_DEL_STATION);
10380         if (!hdr) {
10381                 nlmsg_free(msg);
10382                 return;
10383         }
10384
10385         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
10386             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr))
10387                 goto nla_put_failure;
10388
10389         genlmsg_end(msg, hdr);
10390
10391         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10392                                 nl80211_mlme_mcgrp.id, gfp);
10393         return;
10394
10395  nla_put_failure:
10396         genlmsg_cancel(msg, hdr);
10397         nlmsg_free(msg);
10398 }
10399 EXPORT_SYMBOL(cfg80211_del_sta);
10400
10401 void cfg80211_conn_failed(struct net_device *dev, const u8 *mac_addr,
10402                           enum nl80211_connect_failed_reason reason,
10403                           gfp_t gfp)
10404 {
10405         struct wiphy *wiphy = dev->ieee80211_ptr->wiphy;
10406         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10407         struct sk_buff *msg;
10408         void *hdr;
10409
10410         msg = nlmsg_new(NLMSG_GOODSIZE, gfp);
10411         if (!msg)
10412                 return;
10413
10414         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CONN_FAILED);
10415         if (!hdr) {
10416                 nlmsg_free(msg);
10417                 return;
10418         }
10419
10420         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
10421             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr) ||
10422             nla_put_u32(msg, NL80211_ATTR_CONN_FAILED_REASON, reason))
10423                 goto nla_put_failure;
10424
10425         genlmsg_end(msg, hdr);
10426
10427         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10428                                 nl80211_mlme_mcgrp.id, gfp);
10429         return;
10430
10431  nla_put_failure:
10432         genlmsg_cancel(msg, hdr);
10433         nlmsg_free(msg);
10434 }
10435 EXPORT_SYMBOL(cfg80211_conn_failed);
10436
10437 static bool __nl80211_unexpected_frame(struct net_device *dev, u8 cmd,
10438                                        const u8 *addr, gfp_t gfp)
10439 {
10440         struct wireless_dev *wdev = dev->ieee80211_ptr;
10441         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
10442         struct sk_buff *msg;
10443         void *hdr;
10444         u32 nlportid = ACCESS_ONCE(wdev->ap_unexpected_nlportid);
10445
10446         if (!nlportid)
10447                 return false;
10448
10449         msg = nlmsg_new(100, gfp);
10450         if (!msg)
10451                 return true;
10452
10453         hdr = nl80211hdr_put(msg, 0, 0, 0, cmd);
10454         if (!hdr) {
10455                 nlmsg_free(msg);
10456                 return true;
10457         }
10458
10459         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10460             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
10461             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr))
10462                 goto nla_put_failure;
10463
10464         genlmsg_end(msg, hdr);
10465         genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, nlportid);
10466         return true;
10467
10468  nla_put_failure:
10469         genlmsg_cancel(msg, hdr);
10470         nlmsg_free(msg);
10471         return true;
10472 }
10473
10474 bool cfg80211_rx_spurious_frame(struct net_device *dev,
10475                                 const u8 *addr, gfp_t gfp)
10476 {
10477         struct wireless_dev *wdev = dev->ieee80211_ptr;
10478         bool ret;
10479
10480         trace_cfg80211_rx_spurious_frame(dev, addr);
10481
10482         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_AP &&
10483                     wdev->iftype != NL80211_IFTYPE_P2P_GO)) {
10484                 trace_cfg80211_return_bool(false);
10485                 return false;
10486         }
10487         ret = __nl80211_unexpected_frame(dev, NL80211_CMD_UNEXPECTED_FRAME,
10488                                          addr, gfp);
10489         trace_cfg80211_return_bool(ret);
10490         return ret;
10491 }
10492 EXPORT_SYMBOL(cfg80211_rx_spurious_frame);
10493
10494 bool cfg80211_rx_unexpected_4addr_frame(struct net_device *dev,
10495                                         const u8 *addr, gfp_t gfp)
10496 {
10497         struct wireless_dev *wdev = dev->ieee80211_ptr;
10498         bool ret;
10499
10500         trace_cfg80211_rx_unexpected_4addr_frame(dev, addr);
10501
10502         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_AP &&
10503                     wdev->iftype != NL80211_IFTYPE_P2P_GO &&
10504                     wdev->iftype != NL80211_IFTYPE_AP_VLAN)) {
10505                 trace_cfg80211_return_bool(false);
10506                 return false;
10507         }
10508         ret = __nl80211_unexpected_frame(dev,
10509                                          NL80211_CMD_UNEXPECTED_4ADDR_FRAME,
10510                                          addr, gfp);
10511         trace_cfg80211_return_bool(ret);
10512         return ret;
10513 }
10514 EXPORT_SYMBOL(cfg80211_rx_unexpected_4addr_frame);
10515
10516 int nl80211_send_mgmt(struct cfg80211_registered_device *rdev,
10517                       struct wireless_dev *wdev, u32 nlportid,
10518                       int freq, int sig_dbm,
10519                       const u8 *buf, size_t len, u32 flags, gfp_t gfp)
10520 {
10521         struct net_device *netdev = wdev->netdev;
10522         struct sk_buff *msg;
10523         void *hdr;
10524
10525         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10526         if (!msg)
10527                 return -ENOMEM;
10528
10529         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FRAME);
10530         if (!hdr) {
10531                 nlmsg_free(msg);
10532                 return -ENOMEM;
10533         }
10534
10535         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10536             (netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
10537                                         netdev->ifindex)) ||
10538             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)) ||
10539             nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ, freq) ||
10540             (sig_dbm &&
10541              nla_put_u32(msg, NL80211_ATTR_RX_SIGNAL_DBM, sig_dbm)) ||
10542             nla_put(msg, NL80211_ATTR_FRAME, len, buf) ||
10543             (flags &&
10544              nla_put_u32(msg, NL80211_ATTR_RXMGMT_FLAGS, flags)))
10545                 goto nla_put_failure;
10546
10547         genlmsg_end(msg, hdr);
10548
10549         return genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, nlportid);
10550
10551  nla_put_failure:
10552         genlmsg_cancel(msg, hdr);
10553         nlmsg_free(msg);
10554         return -ENOBUFS;
10555 }
10556
10557 void cfg80211_mgmt_tx_status(struct wireless_dev *wdev, u64 cookie,
10558                              const u8 *buf, size_t len, bool ack, gfp_t gfp)
10559 {
10560         struct wiphy *wiphy = wdev->wiphy;
10561         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10562         struct net_device *netdev = wdev->netdev;
10563         struct sk_buff *msg;
10564         void *hdr;
10565
10566         trace_cfg80211_mgmt_tx_status(wdev, cookie, ack);
10567
10568         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10569         if (!msg)
10570                 return;
10571
10572         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FRAME_TX_STATUS);
10573         if (!hdr) {
10574                 nlmsg_free(msg);
10575                 return;
10576         }
10577
10578         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10579             (netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
10580                                    netdev->ifindex)) ||
10581             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)) ||
10582             nla_put(msg, NL80211_ATTR_FRAME, len, buf) ||
10583             nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie) ||
10584             (ack && nla_put_flag(msg, NL80211_ATTR_ACK)))
10585                 goto nla_put_failure;
10586
10587         genlmsg_end(msg, hdr);
10588
10589         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10590                                 nl80211_mlme_mcgrp.id, gfp);
10591         return;
10592
10593  nla_put_failure:
10594         genlmsg_cancel(msg, hdr);
10595         nlmsg_free(msg);
10596 }
10597 EXPORT_SYMBOL(cfg80211_mgmt_tx_status);
10598
10599 void cfg80211_cqm_rssi_notify(struct net_device *dev,
10600                               enum nl80211_cqm_rssi_threshold_event rssi_event,
10601                               gfp_t gfp)
10602 {
10603         struct wireless_dev *wdev = dev->ieee80211_ptr;
10604         struct wiphy *wiphy = wdev->wiphy;
10605         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10606         struct sk_buff *msg;
10607         struct nlattr *pinfoattr;
10608         void *hdr;
10609
10610         trace_cfg80211_cqm_rssi_notify(dev, rssi_event);
10611
10612         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10613         if (!msg)
10614                 return;
10615
10616         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NOTIFY_CQM);
10617         if (!hdr) {
10618                 nlmsg_free(msg);
10619                 return;
10620         }
10621
10622         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10623             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex))
10624                 goto nla_put_failure;
10625
10626         pinfoattr = nla_nest_start(msg, NL80211_ATTR_CQM);
10627         if (!pinfoattr)
10628                 goto nla_put_failure;
10629
10630         if (nla_put_u32(msg, NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT,
10631                         rssi_event))
10632                 goto nla_put_failure;
10633
10634         nla_nest_end(msg, pinfoattr);
10635
10636         genlmsg_end(msg, hdr);
10637
10638         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10639                                 nl80211_mlme_mcgrp.id, gfp);
10640         return;
10641
10642  nla_put_failure:
10643         genlmsg_cancel(msg, hdr);
10644         nlmsg_free(msg);
10645 }
10646 EXPORT_SYMBOL(cfg80211_cqm_rssi_notify);
10647
10648 static void nl80211_gtk_rekey_notify(struct cfg80211_registered_device *rdev,
10649                                      struct net_device *netdev, const u8 *bssid,
10650                                      const u8 *replay_ctr, gfp_t gfp)
10651 {
10652         struct sk_buff *msg;
10653         struct nlattr *rekey_attr;
10654         void *hdr;
10655
10656         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10657         if (!msg)
10658                 return;
10659
10660         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_SET_REKEY_OFFLOAD);
10661         if (!hdr) {
10662                 nlmsg_free(msg);
10663                 return;
10664         }
10665
10666         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10667             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
10668             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid))
10669                 goto nla_put_failure;
10670
10671         rekey_attr = nla_nest_start(msg, NL80211_ATTR_REKEY_DATA);
10672         if (!rekey_attr)
10673                 goto nla_put_failure;
10674
10675         if (nla_put(msg, NL80211_REKEY_DATA_REPLAY_CTR,
10676                     NL80211_REPLAY_CTR_LEN, replay_ctr))
10677                 goto nla_put_failure;
10678
10679         nla_nest_end(msg, rekey_attr);
10680
10681         genlmsg_end(msg, hdr);
10682
10683         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10684                                 nl80211_mlme_mcgrp.id, gfp);
10685         return;
10686
10687  nla_put_failure:
10688         genlmsg_cancel(msg, hdr);
10689         nlmsg_free(msg);
10690 }
10691
10692 void cfg80211_gtk_rekey_notify(struct net_device *dev, const u8 *bssid,
10693                                const u8 *replay_ctr, gfp_t gfp)
10694 {
10695         struct wireless_dev *wdev = dev->ieee80211_ptr;
10696         struct wiphy *wiphy = wdev->wiphy;
10697         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10698
10699         trace_cfg80211_gtk_rekey_notify(dev, bssid);
10700         nl80211_gtk_rekey_notify(rdev, dev, bssid, replay_ctr, gfp);
10701 }
10702 EXPORT_SYMBOL(cfg80211_gtk_rekey_notify);
10703
10704 static void
10705 nl80211_pmksa_candidate_notify(struct cfg80211_registered_device *rdev,
10706                                struct net_device *netdev, int index,
10707                                const u8 *bssid, bool preauth, gfp_t gfp)
10708 {
10709         struct sk_buff *msg;
10710         struct nlattr *attr;
10711         void *hdr;
10712
10713         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10714         if (!msg)
10715                 return;
10716
10717         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_PMKSA_CANDIDATE);
10718         if (!hdr) {
10719                 nlmsg_free(msg);
10720                 return;
10721         }
10722
10723         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10724             nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
10725                 goto nla_put_failure;
10726
10727         attr = nla_nest_start(msg, NL80211_ATTR_PMKSA_CANDIDATE);
10728         if (!attr)
10729                 goto nla_put_failure;
10730
10731         if (nla_put_u32(msg, NL80211_PMKSA_CANDIDATE_INDEX, index) ||
10732             nla_put(msg, NL80211_PMKSA_CANDIDATE_BSSID, ETH_ALEN, bssid) ||
10733             (preauth &&
10734              nla_put_flag(msg, NL80211_PMKSA_CANDIDATE_PREAUTH)))
10735                 goto nla_put_failure;
10736
10737         nla_nest_end(msg, attr);
10738
10739         genlmsg_end(msg, hdr);
10740
10741         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10742                                 nl80211_mlme_mcgrp.id, gfp);
10743         return;
10744
10745  nla_put_failure:
10746         genlmsg_cancel(msg, hdr);
10747         nlmsg_free(msg);
10748 }
10749
10750 void cfg80211_pmksa_candidate_notify(struct net_device *dev, int index,
10751                                      const u8 *bssid, bool preauth, gfp_t gfp)
10752 {
10753         struct wireless_dev *wdev = dev->ieee80211_ptr;
10754         struct wiphy *wiphy = wdev->wiphy;
10755         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10756
10757         trace_cfg80211_pmksa_candidate_notify(dev, index, bssid, preauth);
10758         nl80211_pmksa_candidate_notify(rdev, dev, index, bssid, preauth, gfp);
10759 }
10760 EXPORT_SYMBOL(cfg80211_pmksa_candidate_notify);
10761
10762 static void nl80211_ch_switch_notify(struct cfg80211_registered_device *rdev,
10763                                      struct net_device *netdev,
10764                                      struct cfg80211_chan_def *chandef,
10765                                      gfp_t gfp)
10766 {
10767         struct sk_buff *msg;
10768         void *hdr;
10769
10770         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10771         if (!msg)
10772                 return;
10773
10774         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CH_SWITCH_NOTIFY);
10775         if (!hdr) {
10776                 nlmsg_free(msg);
10777                 return;
10778         }
10779
10780         if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex))
10781                 goto nla_put_failure;
10782
10783         if (nl80211_send_chandef(msg, chandef))
10784                 goto nla_put_failure;
10785
10786         genlmsg_end(msg, hdr);
10787
10788         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10789                                 nl80211_mlme_mcgrp.id, gfp);
10790         return;
10791
10792  nla_put_failure:
10793         genlmsg_cancel(msg, hdr);
10794         nlmsg_free(msg);
10795 }
10796
10797 void cfg80211_ch_switch_notify(struct net_device *dev,
10798                                struct cfg80211_chan_def *chandef)
10799 {
10800         struct wireless_dev *wdev = dev->ieee80211_ptr;
10801         struct wiphy *wiphy = wdev->wiphy;
10802         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10803
10804         trace_cfg80211_ch_switch_notify(dev, chandef);
10805
10806         wdev_lock(wdev);
10807
10808         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_AP &&
10809                     wdev->iftype != NL80211_IFTYPE_P2P_GO &&
10810                     wdev->iftype != NL80211_IFTYPE_ADHOC &&
10811                     wdev->iftype != NL80211_IFTYPE_MESH_POINT))
10812                 goto out;
10813
10814         wdev->channel = chandef->chan;
10815         nl80211_ch_switch_notify(rdev, dev, chandef, GFP_KERNEL);
10816 out:
10817         wdev_unlock(wdev);
10818         return;
10819 }
10820 EXPORT_SYMBOL(cfg80211_ch_switch_notify);
10821
10822 void cfg80211_cqm_txe_notify(struct net_device *dev,
10823                              const u8 *peer, u32 num_packets,
10824                              u32 rate, u32 intvl, gfp_t gfp)
10825 {
10826         struct wireless_dev *wdev = dev->ieee80211_ptr;
10827         struct wiphy *wiphy = wdev->wiphy;
10828         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10829         struct sk_buff *msg;
10830         struct nlattr *pinfoattr;
10831         void *hdr;
10832
10833         msg = nlmsg_new(NLMSG_GOODSIZE, gfp);
10834         if (!msg)
10835                 return;
10836
10837         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NOTIFY_CQM);
10838         if (!hdr) {
10839                 nlmsg_free(msg);
10840                 return;
10841         }
10842
10843         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10844             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
10845             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, peer))
10846                 goto nla_put_failure;
10847
10848         pinfoattr = nla_nest_start(msg, NL80211_ATTR_CQM);
10849         if (!pinfoattr)
10850                 goto nla_put_failure;
10851
10852         if (nla_put_u32(msg, NL80211_ATTR_CQM_TXE_PKTS, num_packets))
10853                 goto nla_put_failure;
10854
10855         if (nla_put_u32(msg, NL80211_ATTR_CQM_TXE_RATE, rate))
10856                 goto nla_put_failure;
10857
10858         if (nla_put_u32(msg, NL80211_ATTR_CQM_TXE_INTVL, intvl))
10859                 goto nla_put_failure;
10860
10861         nla_nest_end(msg, pinfoattr);
10862
10863         genlmsg_end(msg, hdr);
10864
10865         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10866                                 nl80211_mlme_mcgrp.id, gfp);
10867         return;
10868
10869  nla_put_failure:
10870         genlmsg_cancel(msg, hdr);
10871         nlmsg_free(msg);
10872 }
10873 EXPORT_SYMBOL(cfg80211_cqm_txe_notify);
10874
10875 void
10876 nl80211_radar_notify(struct cfg80211_registered_device *rdev,
10877                      const struct cfg80211_chan_def *chandef,
10878                      enum nl80211_radar_event event,
10879                      struct net_device *netdev, gfp_t gfp)
10880 {
10881         struct sk_buff *msg;
10882         void *hdr;
10883
10884         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10885         if (!msg)
10886                 return;
10887
10888         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_RADAR_DETECT);
10889         if (!hdr) {
10890                 nlmsg_free(msg);
10891                 return;
10892         }
10893
10894         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx))
10895                 goto nla_put_failure;
10896
10897         /* NOP and radar events don't need a netdev parameter */
10898         if (netdev) {
10899                 struct wireless_dev *wdev = netdev->ieee80211_ptr;
10900
10901                 if (nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex) ||
10902                     nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)))
10903                         goto nla_put_failure;
10904         }
10905
10906         if (nla_put_u32(msg, NL80211_ATTR_RADAR_EVENT, event))
10907                 goto nla_put_failure;
10908
10909         if (nl80211_send_chandef(msg, chandef))
10910                 goto nla_put_failure;
10911
10912         genlmsg_end(msg, hdr);
10913
10914         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10915                                 nl80211_mlme_mcgrp.id, gfp);
10916         return;
10917
10918  nla_put_failure:
10919         genlmsg_cancel(msg, hdr);
10920         nlmsg_free(msg);
10921 }
10922
10923 void cfg80211_cqm_pktloss_notify(struct net_device *dev,
10924                                  const u8 *peer, u32 num_packets, gfp_t gfp)
10925 {
10926         struct wireless_dev *wdev = dev->ieee80211_ptr;
10927         struct wiphy *wiphy = wdev->wiphy;
10928         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
10929         struct sk_buff *msg;
10930         struct nlattr *pinfoattr;
10931         void *hdr;
10932
10933         trace_cfg80211_cqm_pktloss_notify(dev, peer, num_packets);
10934
10935         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10936         if (!msg)
10937                 return;
10938
10939         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_NOTIFY_CQM);
10940         if (!hdr) {
10941                 nlmsg_free(msg);
10942                 return;
10943         }
10944
10945         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10946             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
10947             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, peer))
10948                 goto nla_put_failure;
10949
10950         pinfoattr = nla_nest_start(msg, NL80211_ATTR_CQM);
10951         if (!pinfoattr)
10952                 goto nla_put_failure;
10953
10954         if (nla_put_u32(msg, NL80211_ATTR_CQM_PKT_LOSS_EVENT, num_packets))
10955                 goto nla_put_failure;
10956
10957         nla_nest_end(msg, pinfoattr);
10958
10959         genlmsg_end(msg, hdr);
10960
10961         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
10962                                 nl80211_mlme_mcgrp.id, gfp);
10963         return;
10964
10965  nla_put_failure:
10966         genlmsg_cancel(msg, hdr);
10967         nlmsg_free(msg);
10968 }
10969 EXPORT_SYMBOL(cfg80211_cqm_pktloss_notify);
10970
10971 void cfg80211_probe_status(struct net_device *dev, const u8 *addr,
10972                            u64 cookie, bool acked, gfp_t gfp)
10973 {
10974         struct wireless_dev *wdev = dev->ieee80211_ptr;
10975         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
10976         struct sk_buff *msg;
10977         void *hdr;
10978
10979         trace_cfg80211_probe_status(dev, addr, cookie, acked);
10980
10981         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
10982
10983         if (!msg)
10984                 return;
10985
10986         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_PROBE_CLIENT);
10987         if (!hdr) {
10988                 nlmsg_free(msg);
10989                 return;
10990         }
10991
10992         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
10993             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
10994             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, addr) ||
10995             nla_put_u64(msg, NL80211_ATTR_COOKIE, cookie) ||
10996             (acked && nla_put_flag(msg, NL80211_ATTR_ACK)))
10997                 goto nla_put_failure;
10998
10999         genlmsg_end(msg, hdr);
11000
11001         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
11002                                 nl80211_mlme_mcgrp.id, gfp);
11003         return;
11004
11005  nla_put_failure:
11006         genlmsg_cancel(msg, hdr);
11007         nlmsg_free(msg);
11008 }
11009 EXPORT_SYMBOL(cfg80211_probe_status);
11010
11011 void cfg80211_report_obss_beacon(struct wiphy *wiphy,
11012                                  const u8 *frame, size_t len,
11013                                  int freq, int sig_dbm)
11014 {
11015         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
11016         struct sk_buff *msg;
11017         void *hdr;
11018         struct cfg80211_beacon_registration *reg;
11019
11020         trace_cfg80211_report_obss_beacon(wiphy, frame, len, freq, sig_dbm);
11021
11022         spin_lock_bh(&rdev->beacon_registrations_lock);
11023         list_for_each_entry(reg, &rdev->beacon_registrations, list) {
11024                 msg = nlmsg_new(len + 100, GFP_ATOMIC);
11025                 if (!msg) {
11026                         spin_unlock_bh(&rdev->beacon_registrations_lock);
11027                         return;
11028                 }
11029
11030                 hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FRAME);
11031                 if (!hdr)
11032                         goto nla_put_failure;
11033
11034                 if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
11035                     (freq &&
11036                      nla_put_u32(msg, NL80211_ATTR_WIPHY_FREQ, freq)) ||
11037                     (sig_dbm &&
11038                      nla_put_u32(msg, NL80211_ATTR_RX_SIGNAL_DBM, sig_dbm)) ||
11039                     nla_put(msg, NL80211_ATTR_FRAME, len, frame))
11040                         goto nla_put_failure;
11041
11042                 genlmsg_end(msg, hdr);
11043
11044                 genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, reg->nlportid);
11045         }
11046         spin_unlock_bh(&rdev->beacon_registrations_lock);
11047         return;
11048
11049  nla_put_failure:
11050         spin_unlock_bh(&rdev->beacon_registrations_lock);
11051         if (hdr)
11052                 genlmsg_cancel(msg, hdr);
11053         nlmsg_free(msg);
11054 }
11055 EXPORT_SYMBOL(cfg80211_report_obss_beacon);
11056
11057 #ifdef CONFIG_PM
11058 void cfg80211_report_wowlan_wakeup(struct wireless_dev *wdev,
11059                                    struct cfg80211_wowlan_wakeup *wakeup,
11060                                    gfp_t gfp)
11061 {
11062         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
11063         struct sk_buff *msg;
11064         void *hdr;
11065         int size = 200;
11066
11067         trace_cfg80211_report_wowlan_wakeup(wdev->wiphy, wdev, wakeup);
11068
11069         if (wakeup)
11070                 size += wakeup->packet_present_len;
11071
11072         msg = nlmsg_new(size, gfp);
11073         if (!msg)
11074                 return;
11075
11076         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_SET_WOWLAN);
11077         if (!hdr)
11078                 goto free_msg;
11079
11080         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
11081             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)))
11082                 goto free_msg;
11083
11084         if (wdev->netdev && nla_put_u32(msg, NL80211_ATTR_IFINDEX,
11085                                         wdev->netdev->ifindex))
11086                 goto free_msg;
11087
11088         if (wakeup) {
11089                 struct nlattr *reasons;
11090
11091                 reasons = nla_nest_start(msg, NL80211_ATTR_WOWLAN_TRIGGERS);
11092
11093                 if (wakeup->disconnect &&
11094                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_DISCONNECT))
11095                         goto free_msg;
11096                 if (wakeup->magic_pkt &&
11097                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_MAGIC_PKT))
11098                         goto free_msg;
11099                 if (wakeup->gtk_rekey_failure &&
11100                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE))
11101                         goto free_msg;
11102                 if (wakeup->eap_identity_req &&
11103                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST))
11104                         goto free_msg;
11105                 if (wakeup->four_way_handshake &&
11106                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE))
11107                         goto free_msg;
11108                 if (wakeup->rfkill_release &&
11109                     nla_put_flag(msg, NL80211_WOWLAN_TRIG_RFKILL_RELEASE))
11110                         goto free_msg;
11111
11112                 if (wakeup->pattern_idx >= 0 &&
11113                     nla_put_u32(msg, NL80211_WOWLAN_TRIG_PKT_PATTERN,
11114                                 wakeup->pattern_idx))
11115                         goto free_msg;
11116
11117                 if (wakeup->tcp_match)
11118                         nla_put_flag(msg, NL80211_WOWLAN_TRIG_WAKEUP_TCP_MATCH);
11119
11120                 if (wakeup->tcp_connlost)
11121                         nla_put_flag(msg,
11122                                      NL80211_WOWLAN_TRIG_WAKEUP_TCP_CONNLOST);
11123
11124                 if (wakeup->tcp_nomoretokens)
11125                         nla_put_flag(msg,
11126                                 NL80211_WOWLAN_TRIG_WAKEUP_TCP_NOMORETOKENS);
11127
11128                 if (wakeup->packet) {
11129                         u32 pkt_attr = NL80211_WOWLAN_TRIG_WAKEUP_PKT_80211;
11130                         u32 len_attr = NL80211_WOWLAN_TRIG_WAKEUP_PKT_80211_LEN;
11131
11132                         if (!wakeup->packet_80211) {
11133                                 pkt_attr =
11134                                         NL80211_WOWLAN_TRIG_WAKEUP_PKT_8023;
11135                                 len_attr =
11136                                         NL80211_WOWLAN_TRIG_WAKEUP_PKT_8023_LEN;
11137                         }
11138
11139                         if (wakeup->packet_len &&
11140                             nla_put_u32(msg, len_attr, wakeup->packet_len))
11141                                 goto free_msg;
11142
11143                         if (nla_put(msg, pkt_attr, wakeup->packet_present_len,
11144                                     wakeup->packet))
11145                                 goto free_msg;
11146                 }
11147
11148                 nla_nest_end(msg, reasons);
11149         }
11150
11151         genlmsg_end(msg, hdr);
11152
11153         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
11154                                 nl80211_mlme_mcgrp.id, gfp);
11155         return;
11156
11157  free_msg:
11158         nlmsg_free(msg);
11159 }
11160 EXPORT_SYMBOL(cfg80211_report_wowlan_wakeup);
11161 #endif
11162
11163 void cfg80211_tdls_oper_request(struct net_device *dev, const u8 *peer,
11164                                 enum nl80211_tdls_operation oper,
11165                                 u16 reason_code, gfp_t gfp)
11166 {
11167         struct wireless_dev *wdev = dev->ieee80211_ptr;
11168         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
11169         struct sk_buff *msg;
11170         void *hdr;
11171
11172         trace_cfg80211_tdls_oper_request(wdev->wiphy, dev, peer, oper,
11173                                          reason_code);
11174
11175         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
11176         if (!msg)
11177                 return;
11178
11179         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_TDLS_OPER);
11180         if (!hdr) {
11181                 nlmsg_free(msg);
11182                 return;
11183         }
11184
11185         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
11186             nla_put_u32(msg, NL80211_ATTR_IFINDEX, dev->ifindex) ||
11187             nla_put_u8(msg, NL80211_ATTR_TDLS_OPERATION, oper) ||
11188             nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, peer) ||
11189             (reason_code > 0 &&
11190              nla_put_u16(msg, NL80211_ATTR_REASON_CODE, reason_code)))
11191                 goto nla_put_failure;
11192
11193         genlmsg_end(msg, hdr);
11194
11195         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
11196                                 nl80211_mlme_mcgrp.id, gfp);
11197         return;
11198
11199  nla_put_failure:
11200         genlmsg_cancel(msg, hdr);
11201         nlmsg_free(msg);
11202 }
11203 EXPORT_SYMBOL(cfg80211_tdls_oper_request);
11204
11205 static int nl80211_netlink_notify(struct notifier_block * nb,
11206                                   unsigned long state,
11207                                   void *_notify)
11208 {
11209         struct netlink_notify *notify = _notify;
11210         struct cfg80211_registered_device *rdev;
11211         struct wireless_dev *wdev;
11212         struct cfg80211_beacon_registration *reg, *tmp;
11213
11214         if (state != NETLINK_URELEASE)
11215                 return NOTIFY_DONE;
11216
11217         rcu_read_lock();
11218
11219         list_for_each_entry_rcu(rdev, &cfg80211_rdev_list, list) {
11220                 list_for_each_entry_rcu(wdev, &rdev->wdev_list, list)
11221                         cfg80211_mlme_unregister_socket(wdev, notify->portid);
11222
11223                 spin_lock_bh(&rdev->beacon_registrations_lock);
11224                 list_for_each_entry_safe(reg, tmp, &rdev->beacon_registrations,
11225                                          list) {
11226                         if (reg->nlportid == notify->portid) {
11227                                 list_del(&reg->list);
11228                                 kfree(reg);
11229                                 break;
11230                         }
11231                 }
11232                 spin_unlock_bh(&rdev->beacon_registrations_lock);
11233         }
11234
11235         rcu_read_unlock();
11236
11237         return NOTIFY_DONE;
11238 }
11239
11240 static struct notifier_block nl80211_netlink_notifier = {
11241         .notifier_call = nl80211_netlink_notify,
11242 };
11243
11244 void cfg80211_ft_event(struct net_device *netdev,
11245                        struct cfg80211_ft_event_params *ft_event)
11246 {
11247         struct wiphy *wiphy = netdev->ieee80211_ptr->wiphy;
11248         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
11249         struct sk_buff *msg;
11250         void *hdr;
11251
11252         trace_cfg80211_ft_event(wiphy, netdev, ft_event);
11253
11254         if (!ft_event->target_ap)
11255                 return;
11256
11257         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
11258         if (!msg)
11259                 return;
11260
11261         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_FT_EVENT);
11262         if (!hdr) {
11263                 nlmsg_free(msg);
11264                 return;
11265         }
11266
11267         nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx);
11268         nla_put_u32(msg, NL80211_ATTR_IFINDEX, netdev->ifindex);
11269         nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, ft_event->target_ap);
11270         if (ft_event->ies)
11271                 nla_put(msg, NL80211_ATTR_IE, ft_event->ies_len, ft_event->ies);
11272         if (ft_event->ric_ies)
11273                 nla_put(msg, NL80211_ATTR_IE_RIC, ft_event->ric_ies_len,
11274                         ft_event->ric_ies);
11275
11276         genlmsg_end(msg, hdr);
11277
11278         genlmsg_multicast_netns(wiphy_net(&rdev->wiphy), msg, 0,
11279                                 nl80211_mlme_mcgrp.id, GFP_KERNEL);
11280 }
11281 EXPORT_SYMBOL(cfg80211_ft_event);
11282
11283 void cfg80211_crit_proto_stopped(struct wireless_dev *wdev, gfp_t gfp)
11284 {
11285         struct cfg80211_registered_device *rdev;
11286         struct sk_buff *msg;
11287         void *hdr;
11288         u32 nlportid;
11289
11290         rdev = wiphy_to_dev(wdev->wiphy);
11291         if (!rdev->crit_proto_nlportid)
11292                 return;
11293
11294         nlportid = rdev->crit_proto_nlportid;
11295         rdev->crit_proto_nlportid = 0;
11296
11297         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, gfp);
11298         if (!msg)
11299                 return;
11300
11301         hdr = nl80211hdr_put(msg, 0, 0, 0, NL80211_CMD_CRIT_PROTOCOL_STOP);
11302         if (!hdr)
11303                 goto nla_put_failure;
11304
11305         if (nla_put_u32(msg, NL80211_ATTR_WIPHY, rdev->wiphy_idx) ||
11306             nla_put_u64(msg, NL80211_ATTR_WDEV, wdev_id(wdev)))
11307                 goto nla_put_failure;
11308
11309         genlmsg_end(msg, hdr);
11310
11311         genlmsg_unicast(wiphy_net(&rdev->wiphy), msg, nlportid);
11312         return;
11313
11314  nla_put_failure:
11315         if (hdr)
11316                 genlmsg_cancel(msg, hdr);
11317         nlmsg_free(msg);
11318
11319 }
11320 EXPORT_SYMBOL(cfg80211_crit_proto_stopped);
11321
11322 /* initialisation/exit functions */
11323
11324 int nl80211_init(void)
11325 {
11326         int err;
11327
11328         err = genl_register_family_with_ops(&nl80211_fam,
11329                 nl80211_ops, ARRAY_SIZE(nl80211_ops));
11330         if (err)
11331                 return err;
11332
11333         err = genl_register_mc_group(&nl80211_fam, &nl80211_config_mcgrp);
11334         if (err)
11335                 goto err_out;
11336
11337         err = genl_register_mc_group(&nl80211_fam, &nl80211_scan_mcgrp);
11338         if (err)
11339                 goto err_out;
11340
11341         err = genl_register_mc_group(&nl80211_fam, &nl80211_regulatory_mcgrp);
11342         if (err)
11343                 goto err_out;
11344
11345         err = genl_register_mc_group(&nl80211_fam, &nl80211_mlme_mcgrp);
11346         if (err)
11347                 goto err_out;
11348
11349 #ifdef CONFIG_NL80211_TESTMODE
11350         err = genl_register_mc_group(&nl80211_fam, &nl80211_testmode_mcgrp);
11351         if (err)
11352                 goto err_out;
11353 #endif
11354
11355         err = netlink_register_notifier(&nl80211_netlink_notifier);
11356         if (err)
11357                 goto err_out;
11358
11359         return 0;
11360  err_out:
11361         genl_unregister_family(&nl80211_fam);
11362         return err;
11363 }
11364
11365 void nl80211_exit(void)
11366 {
11367         netlink_unregister_notifier(&nl80211_netlink_notifier);
11368         genl_unregister_family(&nl80211_fam);
11369 }