src/drivers/driver_nl80211_capa.c

   1 /*
   2  * Driver interaction with Linux nl80211/cfg80211 - Capabilities
   3  * Copyright (c) 2002-2015, Jouni Malinen <j@w1.fi>
   4  * Copyright (c) 2007, Johannes Berg <johannes@sipsolutions.net>
   5  * Copyright (c) 2009-2010, Atheros Communications
   6  *
   7  * This software may be distributed under the terms of the BSD license.
   8  * See README for more details.
   9  */
  10
  11 #include "includes.h"
  12 #include <netlink/genl/genl.h>
  13
  14 #include "utils/common.h"
  15 #include "common/ieee802_11_defs.h"
  16 #include "common/ieee802_11_common.h"
  17 #include "common/qca-vendor.h"
  18 #include "common/qca-vendor-attr.h"
  19 #include "driver_nl80211.h"
  20
  21
  22 static int protocol_feature_handler(struct nl_msg *msg, void *arg)
  23 {
  24         u32 *feat = arg;
  25         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
  26         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
  27
  28         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
  29                   genlmsg_attrlen(gnlh, 0), NULL);
  30
  31         if (tb_msg[NL80211_ATTR_PROTOCOL_FEATURES])
  32                 *feat = nla_get_u32(tb_msg[NL80211_ATTR_PROTOCOL_FEATURES]);
  33
  34         return NL_SKIP;
  35 }
  36
  37
  38 static u32 get_nl80211_protocol_features(struct wpa_driver_nl80211_data *drv)
  39 {
  40         u32 feat = 0;
  41         struct nl_msg *msg;
  42
  43         msg = nlmsg_alloc();
  44         if (!msg)
  45                 return 0;
  46
  47         if (!nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_PROTOCOL_FEATURES)) {
  48                 nlmsg_free(msg);
  49                 return 0;
  50         }
  51
  52         if (send_and_recv_msgs(drv, msg, protocol_feature_handler, &feat) == 0)
  53                 return feat;
  54
  55         return 0;
  56 }
  57
  58
  59 struct wiphy_info_data {
  60         struct wpa_driver_nl80211_data *drv;
  61         struct wpa_driver_capa *capa;
  62
  63         unsigned int num_multichan_concurrent;
  64
  65         unsigned int error:1;
  66         unsigned int device_ap_sme:1;
  67         unsigned int poll_command_supported:1;
  68         unsigned int data_tx_status:1;
  69         unsigned int monitor_supported:1;
  70         unsigned int auth_supported:1;
  71         unsigned int connect_supported:1;
  72         unsigned int p2p_go_supported:1;
  73         unsigned int p2p_client_supported:1;
  74         unsigned int p2p_go_ctwindow_supported:1;
  75         unsigned int p2p_concurrent:1;
  76         unsigned int channel_switch_supported:1;
  77         unsigned int set_qos_map_supported:1;
  78         unsigned int have_low_prio_scan:1;
  79         unsigned int wmm_ac_supported:1;
  80         unsigned int mac_addr_rand_scan_supported:1;
  81         unsigned int mac_addr_rand_sched_scan_supported:1;
  82 };
  83
  84
  85 static unsigned int probe_resp_offload_support(int supp_protocols)
  86 {
  87         unsigned int prot = 0;
  88
  89         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS)
  90                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS;
  91         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS2)
  92                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS2;
  93         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_P2P)
  94                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_P2P;
  95         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_80211U)
  96                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_INTERWORKING;
  97
  98         return prot;
  99 }
 100
 101
 102 static void wiphy_info_supported_iftypes(struct wiphy_info_data *info,
 103                                          struct nlattr *tb)
 104 {
 105         struct nlattr *nl_mode;
 106         int i;
 107
 108         if (tb == NULL)
 109                 return;
 110
 111         nla_for_each_nested(nl_mode, tb, i) {
 112                 switch (nla_type(nl_mode)) {
 113                 case NL80211_IFTYPE_AP:
 114                         info->capa->flags |= WPA_DRIVER_FLAGS_AP;
 115                         break;
 116                 case NL80211_IFTYPE_MESH_POINT:
 117                         info->capa->flags |= WPA_DRIVER_FLAGS_MESH;
 118                         break;
 119                 case NL80211_IFTYPE_ADHOC:
 120                         info->capa->flags |= WPA_DRIVER_FLAGS_IBSS;
 121                         break;
 122                 case NL80211_IFTYPE_P2P_DEVICE:
 123                         info->capa->flags |=
 124                                 WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE;
 125                         break;
 126                 case NL80211_IFTYPE_P2P_GO:
 127                         info->p2p_go_supported = 1;
 128                         break;
 129                 case NL80211_IFTYPE_P2P_CLIENT:
 130                         info->p2p_client_supported = 1;
 131                         break;
 132                 case NL80211_IFTYPE_MONITOR:
 133                         info->monitor_supported = 1;
 134                         break;
 135                 }
 136         }
 137 }
 138
 139
 140 static int wiphy_info_iface_comb_process(struct wiphy_info_data *info,
 141                                          struct nlattr *nl_combi)
 142 {
 143         struct nlattr *tb_comb[NUM_NL80211_IFACE_COMB];
 144         struct nlattr *tb_limit[NUM_NL80211_IFACE_LIMIT];
 145         struct nlattr *nl_limit, *nl_mode;
 146         int err, rem_limit, rem_mode;
 147         int combination_has_p2p = 0, combination_has_mgd = 0;
 148         static struct nla_policy
 149         iface_combination_policy[NUM_NL80211_IFACE_COMB] = {
 150                 [NL80211_IFACE_COMB_LIMITS] = { .type = NLA_NESTED },
 151                 [NL80211_IFACE_COMB_MAXNUM] = { .type = NLA_U32 },
 152                 [NL80211_IFACE_COMB_STA_AP_BI_MATCH] = { .type = NLA_FLAG },
 153                 [NL80211_IFACE_COMB_NUM_CHANNELS] = { .type = NLA_U32 },
 154                 [NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS] = { .type = NLA_U32 },
 155         },
 156         iface_limit_policy[NUM_NL80211_IFACE_LIMIT] = {
 157                 [NL80211_IFACE_LIMIT_TYPES] = { .type = NLA_NESTED },
 158                 [NL80211_IFACE_LIMIT_MAX] = { .type = NLA_U32 },
 159         };
 160
 161         err = nla_parse_nested(tb_comb, MAX_NL80211_IFACE_COMB,
 162                                nl_combi, iface_combination_policy);
 163         if (err || !tb_comb[NL80211_IFACE_COMB_LIMITS] ||
 164             !tb_comb[NL80211_IFACE_COMB_MAXNUM] ||
 165             !tb_comb[NL80211_IFACE_COMB_NUM_CHANNELS])
 166                 return 0; /* broken combination */
 167
 168         if (tb_comb[NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS])
 169                 info->capa->flags |= WPA_DRIVER_FLAGS_RADAR;
 170
 171         nla_for_each_nested(nl_limit, tb_comb[NL80211_IFACE_COMB_LIMITS],
 172                             rem_limit) {
 173                 err = nla_parse_nested(tb_limit, MAX_NL80211_IFACE_LIMIT,
 174                                        nl_limit, iface_limit_policy);
 175                 if (err || !tb_limit[NL80211_IFACE_LIMIT_TYPES])
 176                         return 0; /* broken combination */
 177
 178                 nla_for_each_nested(nl_mode,
 179                                     tb_limit[NL80211_IFACE_LIMIT_TYPES],
 180                                     rem_mode) {
 181                         int ift = nla_type(nl_mode);
 182                         if (ift == NL80211_IFTYPE_P2P_GO ||
 183                             ift == NL80211_IFTYPE_P2P_CLIENT)
 184                                 combination_has_p2p = 1;
 185                         if (ift == NL80211_IFTYPE_STATION)
 186                                 combination_has_mgd = 1;
 187                 }
 188                 if (combination_has_p2p && combination_has_mgd)
 189                         break;
 190         }
 191
 192         if (combination_has_p2p && combination_has_mgd) {
 193                 unsigned int num_channels =
 194                         nla_get_u32(tb_comb[NL80211_IFACE_COMB_NUM_CHANNELS]);
 195
 196                 info->p2p_concurrent = 1;
 197                 if (info->num_multichan_concurrent < num_channels)
 198                         info->num_multichan_concurrent = num_channels;
 199         }
 200
 201         return 0;
 202 }
 203
 204
 205 static void wiphy_info_iface_comb(struct wiphy_info_data *info,
 206                                   struct nlattr *tb)
 207 {
 208         struct nlattr *nl_combi;
 209         int rem_combi;
 210
 211         if (tb == NULL)
 212                 return;
 213
 214         nla_for_each_nested(nl_combi, tb, rem_combi) {
 215                 if (wiphy_info_iface_comb_process(info, nl_combi) > 0)
 216                         break;
 217         }
 218 }
 219
 220
 221 static void wiphy_info_supp_cmds(struct wiphy_info_data *info,
 222                                  struct nlattr *tb)
 223 {
 224         struct nlattr *nl_cmd;
 225         int i;
 226
 227         if (tb == NULL)
 228                 return;
 229
 230         nla_for_each_nested(nl_cmd, tb, i) {
 231                 switch (nla_get_u32(nl_cmd)) {
 232                 case NL80211_CMD_AUTHENTICATE:
 233                         info->auth_supported = 1;
 234                         break;
 235                 case NL80211_CMD_CONNECT:
 236                         info->connect_supported = 1;
 237                         break;
 238                 case NL80211_CMD_START_SCHED_SCAN:
 239                         info->capa->sched_scan_supported = 1;
 240                         break;
 241                 case NL80211_CMD_PROBE_CLIENT:
 242                         info->poll_command_supported = 1;
 243                         break;
 244                 case NL80211_CMD_CHANNEL_SWITCH:
 245                         info->channel_switch_supported = 1;
 246                         break;
 247                 case NL80211_CMD_SET_QOS_MAP:
 248                         info->set_qos_map_supported = 1;
 249                         break;
 250                 }
 251         }
 252 }
 253
 254
 255 static void wiphy_info_cipher_suites(struct wiphy_info_data *info,
 256                                      struct nlattr *tb)
 257 {
 258         int i, num;
 259         u32 *ciphers;
 260
 261         if (tb == NULL)
 262                 return;
 263
 264         num = nla_len(tb) / sizeof(u32);
 265         ciphers = nla_data(tb);
 266         for (i = 0; i < num; i++) {
 267                 u32 c = ciphers[i];
 268
 269                 wpa_printf(MSG_DEBUG, "nl80211: Supported cipher %02x-%02x-%02x:%d",
 270                            c >> 24, (c >> 16) & 0xff,
 271                            (c >> 8) & 0xff, c & 0xff);
 272                 switch (c) {
 273                 case WLAN_CIPHER_SUITE_CCMP_256:
 274                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP_256;
 275                         break;
 276                 case WLAN_CIPHER_SUITE_GCMP_256:
 277                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP_256;
 278                         break;
 279                 case WLAN_CIPHER_SUITE_CCMP:
 280                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP;
 281                         break;
 282                 case WLAN_CIPHER_SUITE_GCMP:
 283                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP;
 284                         break;
 285                 case WLAN_CIPHER_SUITE_TKIP:
 286                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_TKIP;
 287                         break;
 288                 case WLAN_CIPHER_SUITE_WEP104:
 289                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP104;
 290                         break;
 291                 case WLAN_CIPHER_SUITE_WEP40:
 292                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP40;
 293                         break;
 294                 case WLAN_CIPHER_SUITE_AES_CMAC:
 295                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP;
 296                         break;
 297                 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
 298                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_128;
 299                         break;
 300                 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
 301                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_256;
 302                         break;
 303                 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
 304                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_CMAC_256;
 305                         break;
 306                 case WLAN_CIPHER_SUITE_NO_GROUP_ADDR:
 307                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_GTK_NOT_USED;
 308                         break;
 309                 }
 310         }
 311 }
 312
 313
 314 static void wiphy_info_max_roc(struct wpa_driver_capa *capa,
 315                                struct nlattr *tb)
 316 {
 317         if (tb)
 318                 capa->max_remain_on_chan = nla_get_u32(tb);
 319 }
 320
 321
 322 static void wiphy_info_tdls(struct wpa_driver_capa *capa, struct nlattr *tdls,
 323                             struct nlattr *ext_setup)
 324 {
 325         if (tdls == NULL)
 326                 return;
 327
 328         wpa_printf(MSG_DEBUG, "nl80211: TDLS supported");
 329         capa->flags |= WPA_DRIVER_FLAGS_TDLS_SUPPORT;
 330
 331         if (ext_setup) {
 332                 wpa_printf(MSG_DEBUG, "nl80211: TDLS external setup");
 333                 capa->flags |= WPA_DRIVER_FLAGS_TDLS_EXTERNAL_SETUP;
 334         }
 335 }
 336
 337
 338 static int ext_feature_isset(const u8 *ext_features, int ext_features_len,
 339                              enum nl80211_ext_feature_index ftidx)
 340 {
 341         u8 ft_byte;
 342
 343         if ((int) ftidx / 8 >= ext_features_len)
 344                 return 0;
 345
 346         ft_byte = ext_features[ftidx / 8];
 347         return (ft_byte & BIT(ftidx % 8)) != 0;
 348 }
 349
 350
 351 static void wiphy_info_ext_feature_flags(struct wiphy_info_data *info,
 352                                          struct nlattr *tb)
 353 {
 354         struct wpa_driver_capa *capa = info->capa;
 355         u8 *ext_features;
 356         int len;
 357
 358         if (tb == NULL)
 359                 return;
 360
 361         ext_features = nla_data(tb);
 362         len = nla_len(tb);
 363
 364         if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_VHT_IBSS))
 365                 capa->flags |= WPA_DRIVER_FLAGS_VHT_IBSS;
 366
 367         if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_RRM))
 368                 capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_RRM;
 369 }
 370
 371
 372 static void wiphy_info_feature_flags(struct wiphy_info_data *info,
 373                                      struct nlattr *tb)
 374 {
 375         u32 flags;
 376         struct wpa_driver_capa *capa = info->capa;
 377
 378         if (tb == NULL)
 379                 return;
 380
 381         flags = nla_get_u32(tb);
 382
 383         if (flags & NL80211_FEATURE_SK_TX_STATUS)
 384                 info->data_tx_status = 1;
 385
 386         if (flags & NL80211_FEATURE_INACTIVITY_TIMER)
 387                 capa->flags |= WPA_DRIVER_FLAGS_INACTIVITY_TIMER;
 388
 389         if (flags & NL80211_FEATURE_SAE)
 390                 capa->flags |= WPA_DRIVER_FLAGS_SAE;
 391
 392         if (flags & NL80211_FEATURE_NEED_OBSS_SCAN)
 393                 capa->flags |= WPA_DRIVER_FLAGS_OBSS_SCAN;
 394
 395         if (flags & NL80211_FEATURE_AP_MODE_CHAN_WIDTH_CHANGE)
 396                 capa->flags |= WPA_DRIVER_FLAGS_HT_2040_COEX;
 397
 398         if (flags & NL80211_FEATURE_TDLS_CHANNEL_SWITCH) {
 399                 wpa_printf(MSG_DEBUG, "nl80211: TDLS channel switch");
 400                 capa->flags |= WPA_DRIVER_FLAGS_TDLS_CHANNEL_SWITCH;
 401         }
 402
 403         if (flags & NL80211_FEATURE_P2P_GO_CTWIN)
 404                 info->p2p_go_ctwindow_supported = 1;
 405
 406         if (flags & NL80211_FEATURE_LOW_PRIORITY_SCAN)
 407                 info->have_low_prio_scan = 1;
 408
 409         if (flags & NL80211_FEATURE_SCAN_RANDOM_MAC_ADDR)
 410                 info->mac_addr_rand_scan_supported = 1;
 411
 412         if (flags & NL80211_FEATURE_SCHED_SCAN_RANDOM_MAC_ADDR)
 413                 info->mac_addr_rand_sched_scan_supported = 1;
 414
 415         if (flags & NL80211_FEATURE_STATIC_SMPS)
 416                 capa->smps_modes |= WPA_DRIVER_SMPS_MODE_STATIC;
 417
 418         if (flags & NL80211_FEATURE_DYNAMIC_SMPS)
 419                 capa->smps_modes |= WPA_DRIVER_SMPS_MODE_DYNAMIC;
 420
 421         if (flags & NL80211_FEATURE_SUPPORTS_WMM_ADMISSION)
 422                 info->wmm_ac_supported = 1;
 423
 424         if (flags & NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES)
 425                 capa->rrm_flags |= WPA_DRIVER_FLAGS_DS_PARAM_SET_IE_IN_PROBES;
 426
 427         if (flags & NL80211_FEATURE_WFA_TPC_IE_IN_PROBES)
 428                 capa->rrm_flags |= WPA_DRIVER_FLAGS_WFA_TPC_IE_IN_PROBES;
 429
 430         if (flags & NL80211_FEATURE_QUIET)
 431                 capa->rrm_flags |= WPA_DRIVER_FLAGS_QUIET;
 432
 433         if (flags & NL80211_FEATURE_TX_POWER_INSERTION)
 434                 capa->rrm_flags |= WPA_DRIVER_FLAGS_TX_POWER_INSERTION;
 435
 436         if (flags & NL80211_FEATURE_HT_IBSS)
 437                 capa->flags |= WPA_DRIVER_FLAGS_HT_IBSS;
 438
 439         if (flags & NL80211_FEATURE_FULL_AP_CLIENT_STATE)
 440                 capa->flags |= WPA_DRIVER_FLAGS_FULL_AP_CLIENT_STATE;
 441 }
 442
 443
 444 static void wiphy_info_probe_resp_offload(struct wpa_driver_capa *capa,
 445                                           struct nlattr *tb)
 446 {
 447         u32 protocols;
 448
 449         if (tb == NULL)
 450                 return;
 451
 452         protocols = nla_get_u32(tb);
 453         wpa_printf(MSG_DEBUG, "nl80211: Supports Probe Response offload in AP "
 454                    "mode");
 455         capa->flags |= WPA_DRIVER_FLAGS_PROBE_RESP_OFFLOAD;
 456         capa->probe_resp_offloads = probe_resp_offload_support(protocols);
 457 }
 458
 459
 460 static void wiphy_info_wowlan_triggers(struct wpa_driver_capa *capa,
 461                                        struct nlattr *tb)
 462 {
 463         struct nlattr *triggers[MAX_NL80211_WOWLAN_TRIG + 1];
 464
 465         if (tb == NULL)
 466                 return;
 467
 468         if (nla_parse_nested(triggers, MAX_NL80211_WOWLAN_TRIG,
 469                              tb, NULL))
 470                 return;
 471
 472         if (triggers[NL80211_WOWLAN_TRIG_ANY])
 473                 capa->wowlan_triggers.any = 1;
 474         if (triggers[NL80211_WOWLAN_TRIG_DISCONNECT])
 475                 capa->wowlan_triggers.disconnect = 1;
 476         if (triggers[NL80211_WOWLAN_TRIG_MAGIC_PKT])
 477                 capa->wowlan_triggers.magic_pkt = 1;
 478         if (triggers[NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE])
 479                 capa->wowlan_triggers.gtk_rekey_failure = 1;
 480         if (triggers[NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST])
 481                 capa->wowlan_triggers.eap_identity_req = 1;
 482         if (triggers[NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE])
 483                 capa->wowlan_triggers.four_way_handshake = 1;
 484         if (triggers[NL80211_WOWLAN_TRIG_RFKILL_RELEASE])
 485                 capa->wowlan_triggers.rfkill_release = 1;
 486 }
 487
 488
 489 static void wiphy_info_extended_capab(struct wpa_driver_nl80211_data *drv,
 490                                       struct nlattr *tb)
 491 {
 492         int rem = 0, i;
 493         struct nlattr *tb1[NL80211_ATTR_MAX + 1], *attr;
 494
 495         if (!tb || drv->num_iface_ext_capa == NL80211_IFTYPE_MAX)
 496                 return;
 497
 498         nla_for_each_nested(attr, tb, rem) {
 499                 unsigned int len;
 500                 struct drv_nl80211_ext_capa *capa;
 501
 502                 nla_parse(tb1, NL80211_ATTR_MAX, nla_data(attr),
 503                           nla_len(attr), NULL);
 504
 505                 if (!tb1[NL80211_ATTR_IFTYPE] ||
 506                     !tb1[NL80211_ATTR_EXT_CAPA] ||
 507                     !tb1[NL80211_ATTR_EXT_CAPA_MASK])
 508                         continue;
 509
 510                 capa = &drv->iface_ext_capa[drv->num_iface_ext_capa];
 511                 capa->iftype = nla_get_u32(tb1[NL80211_ATTR_IFTYPE]);
 512                 wpa_printf(MSG_DEBUG,
 513                            "nl80211: Driver-advertised extended capabilities for interface type %s",
 514                            nl80211_iftype_str(capa->iftype));
 515
 516                 len = nla_len(tb1[NL80211_ATTR_EXT_CAPA]);
 517                 capa->ext_capa = os_malloc(len);
 518                 if (!capa->ext_capa)
 519                         goto err;
 520
 521                 os_memcpy(capa->ext_capa, nla_data(tb1[NL80211_ATTR_EXT_CAPA]),
 522                           len);
 523                 capa->ext_capa_len = len;
 524                 wpa_hexdump(MSG_DEBUG, "nl80211: Extended capabilities",
 525                             capa->ext_capa, capa->ext_capa_len);
 526
 527                 len = nla_len(tb1[NL80211_ATTR_EXT_CAPA_MASK]);
 528                 capa->ext_capa_mask = os_malloc(len);
 529                 if (!capa->ext_capa_mask)
 530                         goto err;
 531
 532                 os_memcpy(capa->ext_capa_mask,
 533                           nla_data(tb1[NL80211_ATTR_EXT_CAPA_MASK]), len);
 534                 wpa_hexdump(MSG_DEBUG, "nl80211: Extended capabilities mask",
 535                             capa->ext_capa_mask, capa->ext_capa_len);
 536
 537                 drv->num_iface_ext_capa++;
 538                 if (drv->num_iface_ext_capa == NL80211_IFTYPE_MAX)
 539                         break;
 540         }
 541
 542         return;
 543
 544 err:
 545         /* Cleanup allocated memory on error */
 546         for (i = 0; i < NL80211_IFTYPE_MAX; i++) {
 547                 os_free(drv->iface_ext_capa[i].ext_capa);
 548                 drv->iface_ext_capa[i].ext_capa = NULL;
 549                 os_free(drv->iface_ext_capa[i].ext_capa_mask);
 550                 drv->iface_ext_capa[i].ext_capa_mask = NULL;
 551                 drv->iface_ext_capa[i].ext_capa_len = 0;
 552         }
 553         drv->num_iface_ext_capa = 0;
 554 }
 555
 556
 557 static int wiphy_info_handler(struct nl_msg *msg, void *arg)
 558 {
 559         struct nlattr *tb[NL80211_ATTR_MAX + 1];
 560         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 561         struct wiphy_info_data *info = arg;
 562         struct wpa_driver_capa *capa = info->capa;
 563         struct wpa_driver_nl80211_data *drv = info->drv;
 564
 565         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 566                   genlmsg_attrlen(gnlh, 0), NULL);
 567
 568         if (tb[NL80211_ATTR_WIPHY])
 569                 drv->wiphy_idx = nla_get_u32(tb[NL80211_ATTR_WIPHY]);
 570
 571         if (tb[NL80211_ATTR_WIPHY_NAME])
 572                 os_strlcpy(drv->phyname,
 573                            nla_get_string(tb[NL80211_ATTR_WIPHY_NAME]),
 574                            sizeof(drv->phyname));
 575         if (tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS])
 576                 capa->max_scan_ssids =
 577                         nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS]);
 578
 579         if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS])
 580                 capa->max_sched_scan_ssids =
 581                         nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS]);
 582
 583         if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_PLANS] &&
 584             tb[NL80211_ATTR_MAX_SCAN_PLAN_INTERVAL] &&
 585             tb[NL80211_ATTR_MAX_SCAN_PLAN_ITERATIONS]) {
 586                 capa->max_sched_scan_plans =
 587                         nla_get_u32(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_PLANS]);
 588
 589                 capa->max_sched_scan_plan_interval =
 590                         nla_get_u32(tb[NL80211_ATTR_MAX_SCAN_PLAN_INTERVAL]);
 591
 592                 capa->max_sched_scan_plan_iterations =
 593                         nla_get_u32(tb[NL80211_ATTR_MAX_SCAN_PLAN_ITERATIONS]);
 594         }
 595
 596         if (tb[NL80211_ATTR_MAX_MATCH_SETS])
 597                 capa->max_match_sets =
 598                         nla_get_u8(tb[NL80211_ATTR_MAX_MATCH_SETS]);
 599
 600         if (tb[NL80211_ATTR_MAC_ACL_MAX])
 601                 capa->max_acl_mac_addrs =
 602                         nla_get_u8(tb[NL80211_ATTR_MAC_ACL_MAX]);
 603
 604         wiphy_info_supported_iftypes(info, tb[NL80211_ATTR_SUPPORTED_IFTYPES]);
 605         wiphy_info_iface_comb(info, tb[NL80211_ATTR_INTERFACE_COMBINATIONS]);
 606         wiphy_info_supp_cmds(info, tb[NL80211_ATTR_SUPPORTED_COMMANDS]);
 607         wiphy_info_cipher_suites(info, tb[NL80211_ATTR_CIPHER_SUITES]);
 608
 609         if (tb[NL80211_ATTR_OFFCHANNEL_TX_OK]) {
 610                 wpa_printf(MSG_DEBUG, "nl80211: Using driver-based "
 611                            "off-channel TX");
 612                 capa->flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_TX;
 613         }
 614
 615         if (tb[NL80211_ATTR_ROAM_SUPPORT]) {
 616                 wpa_printf(MSG_DEBUG, "nl80211: Using driver-based roaming");
 617                 capa->flags |= WPA_DRIVER_FLAGS_BSS_SELECTION;
 618         }
 619
 620         wiphy_info_max_roc(capa,
 621                            tb[NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION]);
 622
 623         if (tb[NL80211_ATTR_SUPPORT_AP_UAPSD])
 624                 capa->flags |= WPA_DRIVER_FLAGS_AP_UAPSD;
 625
 626         wiphy_info_tdls(capa, tb[NL80211_ATTR_TDLS_SUPPORT],
 627                         tb[NL80211_ATTR_TDLS_EXTERNAL_SETUP]);
 628
 629         if (tb[NL80211_ATTR_DEVICE_AP_SME])
 630                 info->device_ap_sme = 1;
 631
 632         wiphy_info_feature_flags(info, tb[NL80211_ATTR_FEATURE_FLAGS]);
 633         wiphy_info_ext_feature_flags(info, tb[NL80211_ATTR_EXT_FEATURES]);
 634         wiphy_info_probe_resp_offload(capa,
 635                                       tb[NL80211_ATTR_PROBE_RESP_OFFLOAD]);
 636
 637         if (tb[NL80211_ATTR_EXT_CAPA] && tb[NL80211_ATTR_EXT_CAPA_MASK] &&
 638             drv->extended_capa == NULL) {
 639                 drv->extended_capa =
 640                         os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA]));
 641                 if (drv->extended_capa) {
 642                         os_memcpy(drv->extended_capa,
 643                                   nla_data(tb[NL80211_ATTR_EXT_CAPA]),
 644                                   nla_len(tb[NL80211_ATTR_EXT_CAPA]));
 645                         drv->extended_capa_len =
 646                                 nla_len(tb[NL80211_ATTR_EXT_CAPA]);
 647                         wpa_hexdump(MSG_DEBUG,
 648                                     "nl80211: Driver-advertised extended capabilities (default)",
 649                                     drv->extended_capa, drv->extended_capa_len);
 650                 }
 651                 drv->extended_capa_mask =
 652                         os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
 653                 if (drv->extended_capa_mask) {
 654                         os_memcpy(drv->extended_capa_mask,
 655                                   nla_data(tb[NL80211_ATTR_EXT_CAPA_MASK]),
 656                                   nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
 657                         wpa_hexdump(MSG_DEBUG,
 658                                     "nl80211: Driver-advertised extended capabilities mask (default)",
 659                                     drv->extended_capa_mask,
 660                                     drv->extended_capa_len);
 661                 } else {
 662                         os_free(drv->extended_capa);
 663                         drv->extended_capa = NULL;
 664                         drv->extended_capa_len = 0;
 665                 }
 666         }
 667
 668         wiphy_info_extended_capab(drv, tb[NL80211_ATTR_IFTYPE_EXT_CAPA]);
 669
 670         if (tb[NL80211_ATTR_VENDOR_DATA]) {
 671                 struct nlattr *nl;
 672                 int rem;
 673
 674                 nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_DATA], rem) {
 675                         struct nl80211_vendor_cmd_info *vinfo;
 676                         if (nla_len(nl) != sizeof(*vinfo)) {
 677                                 wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
 678                                 continue;
 679                         }
 680                         vinfo = nla_data(nl);
 681                         if (vinfo->vendor_id == OUI_QCA) {
 682                                 switch (vinfo->subcmd) {
 683                                 case QCA_NL80211_VENDOR_SUBCMD_TEST:
 684                                         drv->vendor_cmd_test_avail = 1;
 685                                         break;
 686 #ifdef CONFIG_DRIVER_NL80211_QCA
 687                                 case QCA_NL80211_VENDOR_SUBCMD_ROAMING:
 688                                         drv->roaming_vendor_cmd_avail = 1;
 689                                         break;
 690                                 case QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY:
 691                                         drv->dfs_vendor_cmd_avail = 1;
 692                                         break;
 693                                 case QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES:
 694                                         drv->get_features_vendor_cmd_avail = 1;
 695                                         break;
 696                                 case QCA_NL80211_VENDOR_SUBCMD_GET_PREFERRED_FREQ_LIST:
 697                                         drv->get_pref_freq_list = 1;
 698                                         break;
 699                                 case QCA_NL80211_VENDOR_SUBCMD_SET_PROBABLE_OPER_CHANNEL:
 700                                         drv->set_prob_oper_freq = 1;
 701                                         break;
 702                                 case QCA_NL80211_VENDOR_SUBCMD_DO_ACS:
 703                                         drv->capa.flags |=
 704                                                 WPA_DRIVER_FLAGS_ACS_OFFLOAD;
 705                                         break;
 706                                 case QCA_NL80211_VENDOR_SUBCMD_SETBAND:
 707                                         drv->setband_vendor_cmd_avail = 1;
 708                                         break;
 709                                 case QCA_NL80211_VENDOR_SUBCMD_TRIGGER_SCAN:
 710                                         drv->scan_vendor_cmd_avail = 1;
 711                                         break;
 712 #endif /* CONFIG_DRIVER_NL80211_QCA */
 713                                 }
 714                         }
 715
 716                         wpa_printf(MSG_DEBUG, "nl80211: Supported vendor command: vendor_id=0x%x subcmd=%u",
 717                                    vinfo->vendor_id, vinfo->subcmd);
 718                 }
 719         }
 720
 721         if (tb[NL80211_ATTR_VENDOR_EVENTS]) {
 722                 struct nlattr *nl;
 723                 int rem;
 724
 725                 nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_EVENTS], rem) {
 726                         struct nl80211_vendor_cmd_info *vinfo;
 727                         if (nla_len(nl) != sizeof(*vinfo)) {
 728                                 wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
 729                                 continue;
 730                         }
 731                         vinfo = nla_data(nl);
 732                         wpa_printf(MSG_DEBUG, "nl80211: Supported vendor event: vendor_id=0x%x subcmd=%u",
 733                                    vinfo->vendor_id, vinfo->subcmd);
 734                 }
 735         }
 736
 737         wiphy_info_wowlan_triggers(capa,
 738                                    tb[NL80211_ATTR_WOWLAN_TRIGGERS_SUPPORTED]);
 739
 740         if (tb[NL80211_ATTR_MAX_AP_ASSOC_STA])
 741                 capa->max_stations =
 742                         nla_get_u32(tb[NL80211_ATTR_MAX_AP_ASSOC_STA]);
 743
 744         if (tb[NL80211_ATTR_MAX_CSA_COUNTERS])
 745                 capa->max_csa_counters =
 746                         nla_get_u8(tb[NL80211_ATTR_MAX_CSA_COUNTERS]);
 747
 748         return NL_SKIP;
 749 }
 750
 751
 752 static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
 753                                        struct wiphy_info_data *info)
 754 {
 755         u32 feat;
 756         struct nl_msg *msg;
 757         int flags = 0;
 758
 759         os_memset(info, 0, sizeof(*info));
 760         info->capa = &drv->capa;
 761         info->drv = drv;
 762
 763         feat = get_nl80211_protocol_features(drv);
 764         if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
 765                 flags = NLM_F_DUMP;
 766         msg = nl80211_cmd_msg(drv->first_bss, flags, NL80211_CMD_GET_WIPHY);
 767         if (!msg || nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
 768                 nlmsg_free(msg);
 769                 return -1;
 770         }
 771
 772         if (send_and_recv_msgs(drv, msg, wiphy_info_handler, info))
 773                 return -1;
 774
 775         if (info->auth_supported)
 776                 drv->capa.flags |= WPA_DRIVER_FLAGS_SME;
 777         else if (!info->connect_supported) {
 778                 wpa_printf(MSG_INFO, "nl80211: Driver does not support "
 779                            "authentication/association or connect commands");
 780                 info->error = 1;
 781         }
 782
 783         if (info->p2p_go_supported && info->p2p_client_supported)
 784                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CAPABLE;
 785         if (info->p2p_concurrent) {
 786                 wpa_printf(MSG_DEBUG, "nl80211: Use separate P2P group "
 787                            "interface (driver advertised support)");
 788                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT;
 789                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P;
 790         }
 791         if (info->num_multichan_concurrent > 1) {
 792                 wpa_printf(MSG_DEBUG, "nl80211: Enable multi-channel "
 793                            "concurrent (driver advertised support)");
 794                 drv->capa.num_multichan_concurrent =
 795                         info->num_multichan_concurrent;
 796         }
 797         if (drv->capa.flags & WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE)
 798                 wpa_printf(MSG_DEBUG, "nl80211: use P2P_DEVICE support");
 799
 800         /* default to 5000 since early versions of mac80211 don't set it */
 801         if (!drv->capa.max_remain_on_chan)
 802                 drv->capa.max_remain_on_chan = 5000;
 803
 804         drv->capa.wmm_ac_supported = info->wmm_ac_supported;
 805
 806         drv->capa.mac_addr_rand_sched_scan_supported =
 807                 info->mac_addr_rand_sched_scan_supported;
 808         drv->capa.mac_addr_rand_scan_supported =
 809                 info->mac_addr_rand_scan_supported;
 810
 811         if (info->channel_switch_supported) {
 812                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP_CSA;
 813                 if (!drv->capa.max_csa_counters)
 814                         drv->capa.max_csa_counters = 1;
 815         }
 816
 817         if (!drv->capa.max_sched_scan_plans) {
 818                 drv->capa.max_sched_scan_plans = 1;
 819                 drv->capa.max_sched_scan_plan_interval = UINT32_MAX;
 820                 drv->capa.max_sched_scan_plan_iterations = 0;
 821         }
 822
 823         return 0;
 824 }
 825
 826
 827 #ifdef CONFIG_DRIVER_NL80211_QCA
 828
 829 static int dfs_info_handler(struct nl_msg *msg, void *arg)
 830 {
 831         struct nlattr *tb[NL80211_ATTR_MAX + 1];
 832         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 833         int *dfs_capability_ptr = arg;
 834
 835         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 836                   genlmsg_attrlen(gnlh, 0), NULL);
 837
 838         if (tb[NL80211_ATTR_VENDOR_DATA]) {
 839                 struct nlattr *nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
 840                 struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
 841
 842                 nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
 843                           nla_data(nl_vend), nla_len(nl_vend), NULL);
 844
 845                 if (tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]) {
 846                         u32 val;
 847                         val = nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]);
 848                         wpa_printf(MSG_DEBUG, "nl80211: DFS offload capability: %u",
 849                                    val);
 850                         *dfs_capability_ptr = val;
 851                 }
 852         }
 853
 854         return NL_SKIP;
 855 }
 856
 857
 858 static void qca_nl80211_check_dfs_capa(struct wpa_driver_nl80211_data *drv)
 859 {
 860         struct nl_msg *msg;
 861         int dfs_capability = 0;
 862         int ret;
 863
 864         if (!drv->dfs_vendor_cmd_avail)
 865                 return;
 866
 867         if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
 868             nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
 869             nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
 870                         QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY)) {
 871                 nlmsg_free(msg);
 872                 return;
 873         }
 874
 875         ret = send_and_recv_msgs(drv, msg, dfs_info_handler, &dfs_capability);
 876         if (!ret && dfs_capability)
 877                 drv->capa.flags |= WPA_DRIVER_FLAGS_DFS_OFFLOAD;
 878 }
 879
 880
 881 struct features_info {
 882         u8 *flags;
 883         size_t flags_len;
 884         struct wpa_driver_capa *capa;
 885 };
 886
 887
 888 static int features_info_handler(struct nl_msg *msg, void *arg)
 889 {
 890         struct nlattr *tb[NL80211_ATTR_MAX + 1];
 891         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 892         struct features_info *info = arg;
 893         struct nlattr *nl_vend, *attr;
 894
 895         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 896                   genlmsg_attrlen(gnlh, 0), NULL);
 897
 898         nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
 899         if (nl_vend) {
 900                 struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
 901
 902                 nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
 903                           nla_data(nl_vend), nla_len(nl_vend), NULL);
 904
 905                 attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_FEATURE_FLAGS];
 906                 if (attr) {
 907                         int len = nla_len(attr);
 908                         info->flags = os_malloc(len);
 909                         if (info->flags != NULL) {
 910                                 os_memcpy(info->flags, nla_data(attr), len);
 911                                 info->flags_len = len;
 912                         }
 913                 }
 914                 attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_CONCURRENCY_CAPA];
 915                 if (attr)
 916                         info->capa->conc_capab = nla_get_u32(attr);
 917
 918                 attr = tb_vendor[
 919                         QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_2_4_BAND];
 920                 if (attr)
 921                         info->capa->max_conc_chan_2_4 = nla_get_u32(attr);
 922
 923                 attr = tb_vendor[
 924                         QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_5_0_BAND];
 925                 if (attr)
 926                         info->capa->max_conc_chan_5_0 = nla_get_u32(attr);
 927         }
 928
 929         return NL_SKIP;
 930 }
 931
 932
 933 static int check_feature(enum qca_wlan_vendor_features feature,
 934                          struct features_info *info)
 935 {
 936         size_t idx = feature / 8;
 937
 938         return (idx < info->flags_len) &&
 939                 (info->flags[idx] & BIT(feature % 8));
 940 }
 941
 942
 943 static void qca_nl80211_get_features(struct wpa_driver_nl80211_data *drv)
 944 {
 945         struct nl_msg *msg;
 946         struct features_info info;
 947         int ret;
 948
 949         if (!drv->get_features_vendor_cmd_avail)
 950                 return;
 951
 952         if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
 953             nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
 954             nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
 955                         QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES)) {
 956                 nlmsg_free(msg);
 957                 return;
 958         }
 959
 960         os_memset(&info, 0, sizeof(info));
 961         info.capa = &drv->capa;
 962         ret = send_and_recv_msgs(drv, msg, features_info_handler, &info);
 963         if (ret || !info.flags)
 964                 return;
 965
 966         if (check_feature(QCA_WLAN_VENDOR_FEATURE_KEY_MGMT_OFFLOAD, &info))
 967                 drv->capa.flags |= WPA_DRIVER_FLAGS_KEY_MGMT_OFFLOAD;
 968
 969         if (check_feature(QCA_WLAN_VENDOR_FEATURE_SUPPORT_HW_MODE_ANY, &info))
 970                 drv->capa.flags |= WPA_DRIVER_FLAGS_SUPPORT_HW_MODE_ANY;
 971
 972         if (check_feature(QCA_WLAN_VENDOR_FEATURE_OFFCHANNEL_SIMULTANEOUS,
 973                           &info))
 974                 drv->capa.flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_SIMULTANEOUS;
 975         if (check_feature(QCA_WLAN_VENDOR_FEATURE_P2P_LISTEN_OFFLOAD, &info))
 976                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_LISTEN_OFFLOAD;
 977         os_free(info.flags);
 978 }
 979
 980 #endif /* CONFIG_DRIVER_NL80211_QCA */
 981
 982
 983 int wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
 984 {
 985         struct wiphy_info_data info;
 986         if (wpa_driver_nl80211_get_info(drv, &info))
 987                 return -1;
 988
 989         if (info.error)
 990                 return -1;
 991
 992         drv->has_capability = 1;
 993         drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
 994                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
 995                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
 996                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK |
 997                 WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B |
 998                 WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B_192;
 999         drv->capa.auth = WPA_DRIVER_AUTH_OPEN |
1000                 WPA_DRIVER_AUTH_SHARED |
1001                 WPA_DRIVER_AUTH_LEAP;
1002
1003         drv->capa.flags |= WPA_DRIVER_FLAGS_SANE_ERROR_CODES;
1004         drv->capa.flags |= WPA_DRIVER_FLAGS_SET_KEYS_AFTER_ASSOC_DONE;
1005         drv->capa.flags |= WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
1006
1007         /*
1008          * As all cfg80211 drivers must support cases where the AP interface is
1009          * removed without the knowledge of wpa_supplicant/hostapd, e.g., in
1010          * case that the user space daemon has crashed, they must be able to
1011          * cleanup all stations and key entries in the AP tear down flow. Thus,
1012          * this flag can/should always be set for cfg80211 drivers.
1013          */
1014         drv->capa.flags |= WPA_DRIVER_FLAGS_AP_TEARDOWN_SUPPORT;
1015
1016         if (!info.device_ap_sme) {
1017                 drv->capa.flags |= WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS;
1018
1019                 /*
1020                  * No AP SME is currently assumed to also indicate no AP MLME
1021                  * in the driver/firmware.
1022                  */
1023                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP_MLME;
1024         }
1025
1026         drv->device_ap_sme = info.device_ap_sme;
1027         drv->poll_command_supported = info.poll_command_supported;
1028         drv->data_tx_status = info.data_tx_status;
1029         drv->p2p_go_ctwindow_supported = info.p2p_go_ctwindow_supported;
1030         if (info.set_qos_map_supported)
1031                 drv->capa.flags |= WPA_DRIVER_FLAGS_QOS_MAPPING;
1032         drv->have_low_prio_scan = info.have_low_prio_scan;
1033
1034         /*
1035          * If poll command and tx status are supported, mac80211 is new enough
1036          * to have everything we need to not need monitor interfaces.
1037          */
1038         drv->use_monitor = !info.poll_command_supported || !info.data_tx_status;
1039
1040         if (drv->device_ap_sme && drv->use_monitor) {
1041                 /*
1042                  * Non-mac80211 drivers may not support monitor interface.
1043                  * Make sure we do not get stuck with incorrect capability here
1044                  * by explicitly testing this.
1045                  */
1046                 if (!info.monitor_supported) {
1047                         wpa_printf(MSG_DEBUG, "nl80211: Disable use_monitor "
1048                                    "with device_ap_sme since no monitor mode "
1049                                    "support detected");
1050                         drv->use_monitor = 0;
1051                 }
1052         }
1053
1054         /*
1055          * If we aren't going to use monitor interfaces, but the
1056          * driver doesn't support data TX status, we won't get TX
1057          * status for EAPOL frames.
1058          */
1059         if (!drv->use_monitor && !info.data_tx_status)
1060                 drv->capa.flags &= ~WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
1061
1062 #ifdef CONFIG_DRIVER_NL80211_QCA
1063         qca_nl80211_check_dfs_capa(drv);
1064         qca_nl80211_get_features(drv);
1065
1066         /*
1067          * To enable offchannel simultaneous support in wpa_supplicant, the
1068          * underlying driver needs to support the same along with offchannel TX.
1069          * Offchannel TX support is needed since remain_on_channel and
1070          * action_tx use some common data structures and hence cannot be
1071          * scheduled simultaneously.
1072          */
1073         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_OFFCHANNEL_TX))
1074                 drv->capa.flags &= ~WPA_DRIVER_FLAGS_OFFCHANNEL_SIMULTANEOUS;
1075 #endif /* CONFIG_DRIVER_NL80211_QCA */
1076
1077         return 0;
1078 }
1079
1080
1081 struct phy_info_arg {
1082         u16 *num_modes;
1083         struct hostapd_hw_modes *modes;
1084         int last_mode, last_chan_idx;
1085         int failed;
1086 };
1087
1088 static void phy_info_ht_capa(struct hostapd_hw_modes *mode, struct nlattr *capa,
1089                              struct nlattr *ampdu_factor,
1090                              struct nlattr *ampdu_density,
1091                              struct nlattr *mcs_set)
1092 {
1093         if (capa)
1094                 mode->ht_capab = nla_get_u16(capa);
1095
1096         if (ampdu_factor)
1097                 mode->a_mpdu_params |= nla_get_u8(ampdu_factor) & 0x03;
1098
1099         if (ampdu_density)
1100                 mode->a_mpdu_params |= nla_get_u8(ampdu_density) << 2;
1101
1102         if (mcs_set && nla_len(mcs_set) >= 16) {
1103                 u8 *mcs;
1104                 mcs = nla_data(mcs_set);
1105                 os_memcpy(mode->mcs_set, mcs, 16);
1106         }
1107 }
1108
1109
1110 static void phy_info_vht_capa(struct hostapd_hw_modes *mode,
1111                               struct nlattr *capa,
1112                               struct nlattr *mcs_set)
1113 {
1114         if (capa)
1115                 mode->vht_capab = nla_get_u32(capa);
1116
1117         if (mcs_set && nla_len(mcs_set) >= 8) {
1118                 u8 *mcs;
1119                 mcs = nla_data(mcs_set);
1120                 os_memcpy(mode->vht_mcs_set, mcs, 8);
1121         }
1122 }
1123
1124
1125 static void phy_info_freq(struct hostapd_hw_modes *mode,
1126                           struct hostapd_channel_data *chan,
1127                           struct nlattr *tb_freq[])
1128 {
1129         u8 channel;
1130         chan->freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
1131         chan->flag = 0;
1132         chan->dfs_cac_ms = 0;
1133         if (ieee80211_freq_to_chan(chan->freq, &channel) != NUM_HOSTAPD_MODES)
1134                 chan->chan = channel;
1135
1136         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1137                 chan->flag |= HOSTAPD_CHAN_DISABLED;
1138         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IR])
1139                 chan->flag |= HOSTAPD_CHAN_NO_IR;
1140         if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
1141                 chan->flag |= HOSTAPD_CHAN_RADAR;
1142         if (tb_freq[NL80211_FREQUENCY_ATTR_INDOOR_ONLY])
1143                 chan->flag |= HOSTAPD_CHAN_INDOOR_ONLY;
1144         if (tb_freq[NL80211_FREQUENCY_ATTR_GO_CONCURRENT])
1145                 chan->flag |= HOSTAPD_CHAN_GO_CONCURRENT;
1146
1147         if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]) {
1148                 enum nl80211_dfs_state state =
1149                         nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]);
1150
1151                 switch (state) {
1152                 case NL80211_DFS_USABLE:
1153                         chan->flag |= HOSTAPD_CHAN_DFS_USABLE;
1154                         break;
1155                 case NL80211_DFS_AVAILABLE:
1156                         chan->flag |= HOSTAPD_CHAN_DFS_AVAILABLE;
1157                         break;
1158                 case NL80211_DFS_UNAVAILABLE:
1159                         chan->flag |= HOSTAPD_CHAN_DFS_UNAVAILABLE;
1160                         break;
1161                 }
1162         }
1163
1164         if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]) {
1165                 chan->dfs_cac_ms = nla_get_u32(
1166                         tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]);
1167         }
1168 }
1169
1170
1171 static int phy_info_freqs(struct phy_info_arg *phy_info,
1172                           struct hostapd_hw_modes *mode, struct nlattr *tb)
1173 {
1174         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1175                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1176                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1177                 [NL80211_FREQUENCY_ATTR_NO_IR] = { .type = NLA_FLAG },
1178                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1179                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
1180                 [NL80211_FREQUENCY_ATTR_DFS_STATE] = { .type = NLA_U32 },
1181         };
1182         int new_channels = 0;
1183         struct hostapd_channel_data *channel;
1184         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1185         struct nlattr *nl_freq;
1186         int rem_freq, idx;
1187
1188         if (tb == NULL)
1189                 return NL_OK;
1190
1191         nla_for_each_nested(nl_freq, tb, rem_freq) {
1192                 nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1193                           nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1194                 if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1195                         continue;
1196                 new_channels++;
1197         }
1198
1199         channel = os_realloc_array(mode->channels,
1200                                    mode->num_channels + new_channels,
1201                                    sizeof(struct hostapd_channel_data));
1202         if (!channel)
1203                 return NL_STOP;
1204
1205         mode->channels = channel;
1206         mode->num_channels += new_channels;
1207
1208         idx = phy_info->last_chan_idx;
1209
1210         nla_for_each_nested(nl_freq, tb, rem_freq) {
1211                 nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1212                           nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1213                 if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1214                         continue;
1215                 phy_info_freq(mode, &mode->channels[idx], tb_freq);
1216                 idx++;
1217         }
1218         phy_info->last_chan_idx = idx;
1219
1220         return NL_OK;
1221 }
1222
1223
1224 static int phy_info_rates(struct hostapd_hw_modes *mode, struct nlattr *tb)
1225 {
1226         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1227                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1228                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] =
1229                 { .type = NLA_FLAG },
1230         };
1231         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1232         struct nlattr *nl_rate;
1233         int rem_rate, idx;
1234
1235         if (tb == NULL)
1236                 return NL_OK;
1237
1238         nla_for_each_nested(nl_rate, tb, rem_rate) {
1239                 nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1240                           nla_data(nl_rate), nla_len(nl_rate),
1241                           rate_policy);
1242                 if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1243                         continue;
1244                 mode->num_rates++;
1245         }
1246
1247         mode->rates = os_calloc(mode->num_rates, sizeof(int));
1248         if (!mode->rates)
1249                 return NL_STOP;
1250
1251         idx = 0;
1252
1253         nla_for_each_nested(nl_rate, tb, rem_rate) {
1254                 nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1255                           nla_data(nl_rate), nla_len(nl_rate),
1256                           rate_policy);
1257                 if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1258                         continue;
1259                 mode->rates[idx] = nla_get_u32(
1260                         tb_rate[NL80211_BITRATE_ATTR_RATE]);
1261                 idx++;
1262         }
1263
1264         return NL_OK;
1265 }
1266
1267
1268 static int phy_info_band(struct phy_info_arg *phy_info, struct nlattr *nl_band)
1269 {
1270         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
1271         struct hostapd_hw_modes *mode;
1272         int ret;
1273
1274         if (phy_info->last_mode != nl_band->nla_type) {
1275                 mode = os_realloc_array(phy_info->modes,
1276                                         *phy_info->num_modes + 1,
1277                                         sizeof(*mode));
1278                 if (!mode) {
1279                         phy_info->failed = 1;
1280                         return NL_STOP;
1281                 }
1282                 phy_info->modes = mode;
1283
1284                 mode = &phy_info->modes[*(phy_info->num_modes)];
1285                 os_memset(mode, 0, sizeof(*mode));
1286                 mode->mode = NUM_HOSTAPD_MODES;
1287                 mode->flags = HOSTAPD_MODE_FLAG_HT_INFO_KNOWN |
1288                         HOSTAPD_MODE_FLAG_VHT_INFO_KNOWN;
1289
1290                 /*
1291                  * Unsupported VHT MCS stream is defined as value 3, so the VHT
1292                  * MCS RX/TX map must be initialized with 0xffff to mark all 8
1293                  * possible streams as unsupported. This will be overridden if
1294                  * driver advertises VHT support.
1295                  */
1296                 mode->vht_mcs_set[0] = 0xff;
1297                 mode->vht_mcs_set[1] = 0xff;
1298                 mode->vht_mcs_set[4] = 0xff;
1299                 mode->vht_mcs_set[5] = 0xff;
1300
1301                 *(phy_info->num_modes) += 1;
1302                 phy_info->last_mode = nl_band->nla_type;
1303                 phy_info->last_chan_idx = 0;
1304         } else
1305                 mode = &phy_info->modes[*(phy_info->num_modes) - 1];
1306
1307         nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
1308                   nla_len(nl_band), NULL);
1309
1310         phy_info_ht_capa(mode, tb_band[NL80211_BAND_ATTR_HT_CAPA],
1311                          tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR],
1312                          tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY],
1313                          tb_band[NL80211_BAND_ATTR_HT_MCS_SET]);
1314         phy_info_vht_capa(mode, tb_band[NL80211_BAND_ATTR_VHT_CAPA],
1315                           tb_band[NL80211_BAND_ATTR_VHT_MCS_SET]);
1316         ret = phy_info_freqs(phy_info, mode, tb_band[NL80211_BAND_ATTR_FREQS]);
1317         if (ret == NL_OK)
1318                 ret = phy_info_rates(mode, tb_band[NL80211_BAND_ATTR_RATES]);
1319         if (ret != NL_OK) {
1320                 phy_info->failed = 1;
1321                 return ret;
1322         }
1323
1324         return NL_OK;
1325 }
1326
1327
1328 static int phy_info_handler(struct nl_msg *msg, void *arg)
1329 {
1330         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1331         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1332         struct phy_info_arg *phy_info = arg;
1333         struct nlattr *nl_band;
1334         int rem_band;
1335
1336         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1337                   genlmsg_attrlen(gnlh, 0), NULL);
1338
1339         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
1340                 return NL_SKIP;
1341
1342         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band)
1343         {
1344                 int res = phy_info_band(phy_info, nl_band);
1345                 if (res != NL_OK)
1346                         return res;
1347         }
1348
1349         return NL_SKIP;
1350 }
1351
1352
1353 static struct hostapd_hw_modes *
1354 wpa_driver_nl80211_postprocess_modes(struct hostapd_hw_modes *modes,
1355                                      u16 *num_modes)
1356 {
1357         u16 m;
1358         struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
1359         int i, mode11g_idx = -1;
1360
1361         /* heuristic to set up modes */
1362         for (m = 0; m < *num_modes; m++) {
1363                 if (!modes[m].num_channels)
1364                         continue;
1365                 if (modes[m].channels[0].freq < 4000) {
1366                         modes[m].mode = HOSTAPD_MODE_IEEE80211B;
1367                         for (i = 0; i < modes[m].num_rates; i++) {
1368                                 if (modes[m].rates[i] > 200) {
1369                                         modes[m].mode = HOSTAPD_MODE_IEEE80211G;
1370                                         break;
1371                                 }
1372                         }
1373                 } else if (modes[m].channels[0].freq > 50000)
1374                         modes[m].mode = HOSTAPD_MODE_IEEE80211AD;
1375                 else
1376                         modes[m].mode = HOSTAPD_MODE_IEEE80211A;
1377         }
1378
1379         /* If only 802.11g mode is included, use it to construct matching
1380          * 802.11b mode data. */
1381
1382         for (m = 0; m < *num_modes; m++) {
1383                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
1384                         return modes; /* 802.11b already included */
1385                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1386                         mode11g_idx = m;
1387         }
1388
1389         if (mode11g_idx < 0)
1390                 return modes; /* 2.4 GHz band not supported at all */
1391
1392         nmodes = os_realloc_array(modes, *num_modes + 1, sizeof(*nmodes));
1393         if (nmodes == NULL)
1394                 return modes; /* Could not add 802.11b mode */
1395
1396         mode = &nmodes[*num_modes];
1397         os_memset(mode, 0, sizeof(*mode));
1398         (*num_modes)++;
1399         modes = nmodes;
1400
1401         mode->mode = HOSTAPD_MODE_IEEE80211B;
1402
1403         mode11g = &modes[mode11g_idx];
1404         mode->num_channels = mode11g->num_channels;
1405         mode->channels = os_malloc(mode11g->num_channels *
1406                                    sizeof(struct hostapd_channel_data));
1407         if (mode->channels == NULL) {
1408                 (*num_modes)--;
1409                 return modes; /* Could not add 802.11b mode */
1410         }
1411         os_memcpy(mode->channels, mode11g->channels,
1412                   mode11g->num_channels * sizeof(struct hostapd_channel_data));
1413
1414         mode->num_rates = 0;
1415         mode->rates = os_malloc(4 * sizeof(int));
1416         if (mode->rates == NULL) {
1417                 os_free(mode->channels);
1418                 (*num_modes)--;
1419                 return modes; /* Could not add 802.11b mode */
1420         }
1421
1422         for (i = 0; i < mode11g->num_rates; i++) {
1423                 if (mode11g->rates[i] != 10 && mode11g->rates[i] != 20 &&
1424                     mode11g->rates[i] != 55 && mode11g->rates[i] != 110)
1425                         continue;
1426                 mode->rates[mode->num_rates] = mode11g->rates[i];
1427                 mode->num_rates++;
1428                 if (mode->num_rates == 4)
1429                         break;
1430         }
1431
1432         if (mode->num_rates == 0) {
1433                 os_free(mode->channels);
1434                 os_free(mode->rates);
1435                 (*num_modes)--;
1436                 return modes; /* No 802.11b rates */
1437         }
1438
1439         wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
1440                    "information");
1441
1442         return modes;
1443 }
1444
1445
1446 static void nl80211_set_ht40_mode(struct hostapd_hw_modes *mode, int start,
1447                                   int end)
1448 {
1449         int c;
1450
1451         for (c = 0; c < mode->num_channels; c++) {
1452                 struct hostapd_channel_data *chan = &mode->channels[c];
1453                 if (chan->freq - 10 >= start && chan->freq + 10 <= end)
1454                         chan->flag |= HOSTAPD_CHAN_HT40;
1455         }
1456 }
1457
1458
1459 static void nl80211_set_ht40_mode_sec(struct hostapd_hw_modes *mode, int start,
1460                                       int end)
1461 {
1462         int c;
1463
1464         for (c = 0; c < mode->num_channels; c++) {
1465                 struct hostapd_channel_data *chan = &mode->channels[c];
1466                 if (!(chan->flag & HOSTAPD_CHAN_HT40))
1467                         continue;
1468                 if (chan->freq - 30 >= start && chan->freq - 10 <= end)
1469                         chan->flag |= HOSTAPD_CHAN_HT40MINUS;
1470                 if (chan->freq + 10 >= start && chan->freq + 30 <= end)
1471                         chan->flag |= HOSTAPD_CHAN_HT40PLUS;
1472         }
1473 }
1474
1475
1476 static void nl80211_reg_rule_max_eirp(u32 start, u32 end, u32 max_eirp,
1477                                       struct phy_info_arg *results)
1478 {
1479         u16 m;
1480
1481         for (m = 0; m < *results->num_modes; m++) {
1482                 int c;
1483                 struct hostapd_hw_modes *mode = &results->modes[m];
1484
1485                 for (c = 0; c < mode->num_channels; c++) {
1486                         struct hostapd_channel_data *chan = &mode->channels[c];
1487                         if ((u32) chan->freq - 10 >= start &&
1488                             (u32) chan->freq + 10 <= end)
1489                                 chan->max_tx_power = max_eirp;
1490                 }
1491         }
1492 }
1493
1494
1495 static void nl80211_reg_rule_ht40(u32 start, u32 end,
1496                                   struct phy_info_arg *results)
1497 {
1498         u16 m;
1499
1500         for (m = 0; m < *results->num_modes; m++) {
1501                 if (!(results->modes[m].ht_capab &
1502                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1503                         continue;
1504                 nl80211_set_ht40_mode(&results->modes[m], start, end);
1505         }
1506 }
1507
1508
1509 static void nl80211_reg_rule_sec(struct nlattr *tb[],
1510                                  struct phy_info_arg *results)
1511 {
1512         u32 start, end, max_bw;
1513         u16 m;
1514
1515         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1516             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1517             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1518                 return;
1519
1520         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1521         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1522         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1523
1524         if (max_bw < 20)
1525                 return;
1526
1527         for (m = 0; m < *results->num_modes; m++) {
1528                 if (!(results->modes[m].ht_capab &
1529                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1530                         continue;
1531                 nl80211_set_ht40_mode_sec(&results->modes[m], start, end);
1532         }
1533 }
1534
1535
1536 static void nl80211_set_vht_mode(struct hostapd_hw_modes *mode, int start,
1537                                  int end, int max_bw)
1538 {
1539         int c;
1540
1541         for (c = 0; c < mode->num_channels; c++) {
1542                 struct hostapd_channel_data *chan = &mode->channels[c];
1543                 if (chan->freq - 10 >= start && chan->freq + 70 <= end)
1544                         chan->flag |= HOSTAPD_CHAN_VHT_10_70;
1545
1546                 if (chan->freq - 30 >= start && chan->freq + 50 <= end)
1547                         chan->flag |= HOSTAPD_CHAN_VHT_30_50;
1548
1549                 if (chan->freq - 50 >= start && chan->freq + 30 <= end)
1550                         chan->flag |= HOSTAPD_CHAN_VHT_50_30;
1551
1552                 if (chan->freq - 70 >= start && chan->freq + 10 <= end)
1553                         chan->flag |= HOSTAPD_CHAN_VHT_70_10;
1554
1555                 if (max_bw >= 160) {
1556                         if (chan->freq - 10 >= start && chan->freq + 150 <= end)
1557                                 chan->flag |= HOSTAPD_CHAN_VHT_10_150;
1558
1559                         if (chan->freq - 30 >= start && chan->freq + 130 <= end)
1560                                 chan->flag |= HOSTAPD_CHAN_VHT_30_130;
1561
1562                         if (chan->freq - 50 >= start && chan->freq + 110 <= end)
1563                                 chan->flag |= HOSTAPD_CHAN_VHT_50_110;
1564
1565                         if (chan->freq - 70 >= start && chan->freq + 90 <= end)
1566                                 chan->flag |= HOSTAPD_CHAN_VHT_70_90;
1567
1568                         if (chan->freq - 90 >= start && chan->freq + 70 <= end)
1569                                 chan->flag |= HOSTAPD_CHAN_VHT_90_70;
1570
1571                         if (chan->freq - 110 >= start && chan->freq + 50 <= end)
1572                                 chan->flag |= HOSTAPD_CHAN_VHT_110_50;
1573
1574                         if (chan->freq - 130 >= start && chan->freq + 30 <= end)
1575                                 chan->flag |= HOSTAPD_CHAN_VHT_130_30;
1576
1577                         if (chan->freq - 150 >= start && chan->freq + 10 <= end)
1578                                 chan->flag |= HOSTAPD_CHAN_VHT_150_10;
1579                 }
1580         }
1581 }
1582
1583
1584 static void nl80211_reg_rule_vht(struct nlattr *tb[],
1585                                  struct phy_info_arg *results)
1586 {
1587         u32 start, end, max_bw;
1588         u16 m;
1589
1590         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1591             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1592             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1593                 return;
1594
1595         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1596         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1597         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1598
1599         if (max_bw < 80)
1600                 return;
1601
1602         for (m = 0; m < *results->num_modes; m++) {
1603                 if (!(results->modes[m].ht_capab &
1604                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1605                         continue;
1606                 /* TODO: use a real VHT support indication */
1607                 if (!results->modes[m].vht_capab)
1608                         continue;
1609
1610                 nl80211_set_vht_mode(&results->modes[m], start, end, max_bw);
1611         }
1612 }
1613
1614
1615 static const char * dfs_domain_name(enum nl80211_dfs_regions region)
1616 {
1617         switch (region) {
1618         case NL80211_DFS_UNSET:
1619                 return "DFS-UNSET";
1620         case NL80211_DFS_FCC:
1621                 return "DFS-FCC";
1622         case NL80211_DFS_ETSI:
1623                 return "DFS-ETSI";
1624         case NL80211_DFS_JP:
1625                 return "DFS-JP";
1626         default:
1627                 return "DFS-invalid";
1628         }
1629 }
1630
1631
1632 static int nl80211_get_reg(struct nl_msg *msg, void *arg)
1633 {
1634         struct phy_info_arg *results = arg;
1635         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1636         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1637         struct nlattr *nl_rule;
1638         struct nlattr *tb_rule[NL80211_FREQUENCY_ATTR_MAX + 1];
1639         int rem_rule;
1640         static struct nla_policy reg_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1641                 [NL80211_ATTR_REG_RULE_FLAGS] = { .type = NLA_U32 },
1642                 [NL80211_ATTR_FREQ_RANGE_START] = { .type = NLA_U32 },
1643                 [NL80211_ATTR_FREQ_RANGE_END] = { .type = NLA_U32 },
1644                 [NL80211_ATTR_FREQ_RANGE_MAX_BW] = { .type = NLA_U32 },
1645                 [NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN] = { .type = NLA_U32 },
1646                 [NL80211_ATTR_POWER_RULE_MAX_EIRP] = { .type = NLA_U32 },
1647         };
1648
1649         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1650                   genlmsg_attrlen(gnlh, 0), NULL);
1651         if (!tb_msg[NL80211_ATTR_REG_ALPHA2] ||
1652             !tb_msg[NL80211_ATTR_REG_RULES]) {
1653                 wpa_printf(MSG_DEBUG, "nl80211: No regulatory information "
1654                            "available");
1655                 return NL_SKIP;
1656         }
1657
1658         if (tb_msg[NL80211_ATTR_DFS_REGION]) {
1659                 enum nl80211_dfs_regions dfs_domain;
1660                 dfs_domain = nla_get_u8(tb_msg[NL80211_ATTR_DFS_REGION]);
1661                 wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s (%s)",
1662                            (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]),
1663                            dfs_domain_name(dfs_domain));
1664         } else {
1665                 wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s",
1666                            (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]));
1667         }
1668
1669         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1670         {
1671                 u32 start, end, max_eirp = 0, max_bw = 0, flags = 0;
1672                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1673                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1674                 if (tb_rule[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1675                     tb_rule[NL80211_ATTR_FREQ_RANGE_END] == NULL)
1676                         continue;
1677                 start = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1678                 end = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1679                 if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
1680                         max_eirp = nla_get_u32(tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP]) / 100;
1681                 if (tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW])
1682                         max_bw = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1683                 if (tb_rule[NL80211_ATTR_REG_RULE_FLAGS])
1684                         flags = nla_get_u32(tb_rule[NL80211_ATTR_REG_RULE_FLAGS]);
1685
1686                 wpa_printf(MSG_DEBUG, "nl80211: %u-%u @ %u MHz %u mBm%s%s%s%s%s%s%s%s",
1687                            start, end, max_bw, max_eirp,
1688                            flags & NL80211_RRF_NO_OFDM ? " (no OFDM)" : "",
1689                            flags & NL80211_RRF_NO_CCK ? " (no CCK)" : "",
1690                            flags & NL80211_RRF_NO_INDOOR ? " (no indoor)" : "",
1691                            flags & NL80211_RRF_NO_OUTDOOR ? " (no outdoor)" :
1692                            "",
1693                            flags & NL80211_RRF_DFS ? " (DFS)" : "",
1694                            flags & NL80211_RRF_PTP_ONLY ? " (PTP only)" : "",
1695                            flags & NL80211_RRF_PTMP_ONLY ? " (PTMP only)" : "",
1696                            flags & NL80211_RRF_NO_IR ? " (no IR)" : "");
1697                 if (max_bw >= 40)
1698                         nl80211_reg_rule_ht40(start, end, results);
1699                 if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
1700                         nl80211_reg_rule_max_eirp(start, end, max_eirp,
1701                                                   results);
1702         }
1703
1704         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1705         {
1706                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1707                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1708                 nl80211_reg_rule_sec(tb_rule, results);
1709         }
1710
1711         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1712         {
1713                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1714                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1715                 nl80211_reg_rule_vht(tb_rule, results);
1716         }
1717
1718         return NL_SKIP;
1719 }
1720
1721
1722 static int nl80211_set_regulatory_flags(struct wpa_driver_nl80211_data *drv,
1723                                         struct phy_info_arg *results)
1724 {
1725         struct nl_msg *msg;
1726
1727         msg = nlmsg_alloc();
1728         if (!msg)
1729                 return -ENOMEM;
1730
1731         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_REG);
1732         return send_and_recv_msgs(drv, msg, nl80211_get_reg, results);
1733 }
1734
1735
1736 struct hostapd_hw_modes *
1737 nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
1738 {
1739         u32 feat;
1740         struct i802_bss *bss = priv;
1741         struct wpa_driver_nl80211_data *drv = bss->drv;
1742         int nl_flags = 0;
1743         struct nl_msg *msg;
1744         struct phy_info_arg result = {
1745                 .num_modes = num_modes,
1746                 .modes = NULL,
1747                 .last_mode = -1,
1748                 .failed = 0,
1749         };
1750
1751         *num_modes = 0;
1752         *flags = 0;
1753
1754         feat = get_nl80211_protocol_features(drv);
1755         if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
1756                 nl_flags = NLM_F_DUMP;
1757         if (!(msg = nl80211_cmd_msg(bss, nl_flags, NL80211_CMD_GET_WIPHY)) ||
1758             nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
1759                 nlmsg_free(msg);
1760                 return NULL;
1761         }
1762
1763         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0) {
1764                 nl80211_set_regulatory_flags(drv, &result);
1765                 if (result.failed) {
1766                         int i;
1767
1768                         for (i = 0; result.modes && i < *num_modes; i++) {
1769                                 os_free(result.modes[i].channels);
1770                                 os_free(result.modes[i].rates);
1771                         }
1772                         os_free(result.modes);
1773                         return NULL;
1774                 }
1775                 return wpa_driver_nl80211_postprocess_modes(result.modes,
1776                                                             num_modes);
1777         }
1778
1779         return NULL;
1780 }