contrib/wpa/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_common.h"
  16 #include "common/wpa_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 auth_supported:1;
  70         unsigned int connect_supported:1;
  71         unsigned int p2p_go_supported:1;
  72         unsigned int p2p_client_supported:1;
  73         unsigned int p2p_go_ctwindow_supported:1;
  74         unsigned int p2p_concurrent:1;
  75         unsigned int channel_switch_supported:1;
  76         unsigned int set_qos_map_supported:1;
  77         unsigned int have_low_prio_scan:1;
  78         unsigned int wmm_ac_supported:1;
  79         unsigned int mac_addr_rand_scan_supported:1;
  80         unsigned int mac_addr_rand_sched_scan_supported:1;
  81 };
  82
  83
  84 static unsigned int probe_resp_offload_support(int supp_protocols)
  85 {
  86         unsigned int prot = 0;
  87
  88         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS)
  89                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS;
  90         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS2)
  91                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS2;
  92         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_P2P)
  93                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_P2P;
  94         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_80211U)
  95                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_INTERWORKING;
  96
  97         return prot;
  98 }
  99
 100
 101 static void wiphy_info_supported_iftypes(struct wiphy_info_data *info,
 102                                          struct nlattr *tb)
 103 {
 104         struct nlattr *nl_mode;
 105         int i;
 106
 107         if (tb == NULL)
 108                 return;
 109
 110         nla_for_each_nested(nl_mode, tb, i) {
 111                 switch (nla_type(nl_mode)) {
 112                 case NL80211_IFTYPE_AP:
 113                         info->capa->flags |= WPA_DRIVER_FLAGS_AP;
 114                         break;
 115                 case NL80211_IFTYPE_MESH_POINT:
 116                         info->capa->flags |= WPA_DRIVER_FLAGS_MESH;
 117                         break;
 118                 case NL80211_IFTYPE_ADHOC:
 119                         info->capa->flags |= WPA_DRIVER_FLAGS_IBSS;
 120                         break;
 121                 case NL80211_IFTYPE_P2P_DEVICE:
 122                         info->capa->flags |=
 123                                 WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE;
 124                         break;
 125                 case NL80211_IFTYPE_P2P_GO:
 126                         info->p2p_go_supported = 1;
 127                         break;
 128                 case NL80211_IFTYPE_P2P_CLIENT:
 129                         info->p2p_client_supported = 1;
 130                         break;
 131                 }
 132         }
 133 }
 134
 135
 136 static int wiphy_info_iface_comb_process(struct wiphy_info_data *info,
 137                                          struct nlattr *nl_combi)
 138 {
 139         struct nlattr *tb_comb[NUM_NL80211_IFACE_COMB];
 140         struct nlattr *tb_limit[NUM_NL80211_IFACE_LIMIT];
 141         struct nlattr *nl_limit, *nl_mode;
 142         int err, rem_limit, rem_mode;
 143         int combination_has_p2p = 0, combination_has_mgd = 0;
 144         static struct nla_policy
 145         iface_combination_policy[NUM_NL80211_IFACE_COMB] = {
 146                 [NL80211_IFACE_COMB_LIMITS] = { .type = NLA_NESTED },
 147                 [NL80211_IFACE_COMB_MAXNUM] = { .type = NLA_U32 },
 148                 [NL80211_IFACE_COMB_STA_AP_BI_MATCH] = { .type = NLA_FLAG },
 149                 [NL80211_IFACE_COMB_NUM_CHANNELS] = { .type = NLA_U32 },
 150                 [NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS] = { .type = NLA_U32 },
 151         },
 152         iface_limit_policy[NUM_NL80211_IFACE_LIMIT] = {
 153                 [NL80211_IFACE_LIMIT_TYPES] = { .type = NLA_NESTED },
 154                 [NL80211_IFACE_LIMIT_MAX] = { .type = NLA_U32 },
 155         };
 156
 157         err = nla_parse_nested(tb_comb, MAX_NL80211_IFACE_COMB,
 158                                nl_combi, iface_combination_policy);
 159         if (err || !tb_comb[NL80211_IFACE_COMB_LIMITS] ||
 160             !tb_comb[NL80211_IFACE_COMB_MAXNUM] ||
 161             !tb_comb[NL80211_IFACE_COMB_NUM_CHANNELS])
 162                 return 0; /* broken combination */
 163
 164         if (tb_comb[NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS])
 165                 info->capa->flags |= WPA_DRIVER_FLAGS_RADAR;
 166
 167         nla_for_each_nested(nl_limit, tb_comb[NL80211_IFACE_COMB_LIMITS],
 168                             rem_limit) {
 169                 err = nla_parse_nested(tb_limit, MAX_NL80211_IFACE_LIMIT,
 170                                        nl_limit, iface_limit_policy);
 171                 if (err || !tb_limit[NL80211_IFACE_LIMIT_TYPES])
 172                         return 0; /* broken combination */
 173
 174                 nla_for_each_nested(nl_mode,
 175                                     tb_limit[NL80211_IFACE_LIMIT_TYPES],
 176                                     rem_mode) {
 177                         int ift = nla_type(nl_mode);
 178                         if (ift == NL80211_IFTYPE_P2P_GO ||
 179                             ift == NL80211_IFTYPE_P2P_CLIENT)
 180                                 combination_has_p2p = 1;
 181                         if (ift == NL80211_IFTYPE_STATION)
 182                                 combination_has_mgd = 1;
 183                 }
 184                 if (combination_has_p2p && combination_has_mgd)
 185                         break;
 186         }
 187
 188         if (combination_has_p2p && combination_has_mgd) {
 189                 unsigned int num_channels =
 190                         nla_get_u32(tb_comb[NL80211_IFACE_COMB_NUM_CHANNELS]);
 191
 192                 info->p2p_concurrent = 1;
 193                 if (info->num_multichan_concurrent < num_channels)
 194                         info->num_multichan_concurrent = num_channels;
 195         }
 196
 197         return 0;
 198 }
 199
 200
 201 static void wiphy_info_iface_comb(struct wiphy_info_data *info,
 202                                   struct nlattr *tb)
 203 {
 204         struct nlattr *nl_combi;
 205         int rem_combi;
 206
 207         if (tb == NULL)
 208                 return;
 209
 210         nla_for_each_nested(nl_combi, tb, rem_combi) {
 211                 if (wiphy_info_iface_comb_process(info, nl_combi) > 0)
 212                         break;
 213         }
 214 }
 215
 216
 217 static void wiphy_info_supp_cmds(struct wiphy_info_data *info,
 218                                  struct nlattr *tb)
 219 {
 220         struct nlattr *nl_cmd;
 221         int i;
 222
 223         if (tb == NULL)
 224                 return;
 225
 226         nla_for_each_nested(nl_cmd, tb, i) {
 227                 switch (nla_get_u32(nl_cmd)) {
 228                 case NL80211_CMD_AUTHENTICATE:
 229                         info->auth_supported = 1;
 230                         break;
 231                 case NL80211_CMD_CONNECT:
 232                         info->connect_supported = 1;
 233                         break;
 234                 case NL80211_CMD_START_SCHED_SCAN:
 235                         info->capa->sched_scan_supported = 1;
 236                         break;
 237                 case NL80211_CMD_PROBE_CLIENT:
 238                         info->poll_command_supported = 1;
 239                         break;
 240                 case NL80211_CMD_CHANNEL_SWITCH:
 241                         info->channel_switch_supported = 1;
 242                         break;
 243                 case NL80211_CMD_SET_QOS_MAP:
 244                         info->set_qos_map_supported = 1;
 245                         break;
 246                 }
 247         }
 248 }
 249
 250
 251 static void wiphy_info_cipher_suites(struct wiphy_info_data *info,
 252                                      struct nlattr *tb)
 253 {
 254         int i, num;
 255         u32 *ciphers;
 256
 257         if (tb == NULL)
 258                 return;
 259
 260         num = nla_len(tb) / sizeof(u32);
 261         ciphers = nla_data(tb);
 262         for (i = 0; i < num; i++) {
 263                 u32 c = ciphers[i];
 264
 265                 wpa_printf(MSG_DEBUG, "nl80211: Supported cipher %02x-%02x-%02x:%d",
 266                            c >> 24, (c >> 16) & 0xff,
 267                            (c >> 8) & 0xff, c & 0xff);
 268                 switch (c) {
 269                 case RSN_CIPHER_SUITE_CCMP_256:
 270                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP_256;
 271                         break;
 272                 case RSN_CIPHER_SUITE_GCMP_256:
 273                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP_256;
 274                         break;
 275                 case RSN_CIPHER_SUITE_CCMP:
 276                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP;
 277                         break;
 278                 case RSN_CIPHER_SUITE_GCMP:
 279                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP;
 280                         break;
 281                 case RSN_CIPHER_SUITE_TKIP:
 282                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_TKIP;
 283                         break;
 284                 case RSN_CIPHER_SUITE_WEP104:
 285                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP104;
 286                         break;
 287                 case RSN_CIPHER_SUITE_WEP40:
 288                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP40;
 289                         break;
 290                 case RSN_CIPHER_SUITE_AES_128_CMAC:
 291                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP;
 292                         break;
 293                 case RSN_CIPHER_SUITE_BIP_GMAC_128:
 294                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_128;
 295                         break;
 296                 case RSN_CIPHER_SUITE_BIP_GMAC_256:
 297                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_256;
 298                         break;
 299                 case RSN_CIPHER_SUITE_BIP_CMAC_256:
 300                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_CMAC_256;
 301                         break;
 302                 case RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED:
 303                         info->capa->enc |= WPA_DRIVER_CAPA_ENC_GTK_NOT_USED;
 304                         break;
 305                 }
 306         }
 307 }
 308
 309
 310 static void wiphy_info_max_roc(struct wpa_driver_capa *capa,
 311                                struct nlattr *tb)
 312 {
 313         if (tb)
 314                 capa->max_remain_on_chan = nla_get_u32(tb);
 315 }
 316
 317
 318 static void wiphy_info_tdls(struct wpa_driver_capa *capa, struct nlattr *tdls,
 319                             struct nlattr *ext_setup)
 320 {
 321         if (tdls == NULL)
 322                 return;
 323
 324         wpa_printf(MSG_DEBUG, "nl80211: TDLS supported");
 325         capa->flags |= WPA_DRIVER_FLAGS_TDLS_SUPPORT;
 326
 327         if (ext_setup) {
 328                 wpa_printf(MSG_DEBUG, "nl80211: TDLS external setup");
 329                 capa->flags |= WPA_DRIVER_FLAGS_TDLS_EXTERNAL_SETUP;
 330         }
 331 }
 332
 333
 334 static int ext_feature_isset(const u8 *ext_features, int ext_features_len,
 335                              enum nl80211_ext_feature_index ftidx)
 336 {
 337         u8 ft_byte;
 338
 339         if ((int) ftidx / 8 >= ext_features_len)
 340                 return 0;
 341
 342         ft_byte = ext_features[ftidx / 8];
 343         return (ft_byte & BIT(ftidx % 8)) != 0;
 344 }
 345
 346
 347 static void wiphy_info_ext_feature_flags(struct wiphy_info_data *info,
 348                                          struct nlattr *tb)
 349 {
 350         struct wpa_driver_capa *capa = info->capa;
 351         u8 *ext_features;
 352         int len;
 353
 354         if (tb == NULL)
 355                 return;
 356
 357         ext_features = nla_data(tb);
 358         len = nla_len(tb);
 359
 360         if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_VHT_IBSS))
 361                 capa->flags |= WPA_DRIVER_FLAGS_VHT_IBSS;
 362
 363         if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_RRM))
 364                 capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_RRM;
 365
 366         if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_FILS_STA))
 367                 capa->flags |= WPA_DRIVER_FLAGS_SUPPORT_FILS;
 368
 369         if (ext_feature_isset(ext_features, len,
 370                               NL80211_EXT_FEATURE_BEACON_RATE_LEGACY))
 371                 capa->flags |= WPA_DRIVER_FLAGS_BEACON_RATE_LEGACY;
 372
 373         if (ext_feature_isset(ext_features, len,
 374                               NL80211_EXT_FEATURE_BEACON_RATE_HT))
 375                 capa->flags |= WPA_DRIVER_FLAGS_BEACON_RATE_HT;
 376
 377         if (ext_feature_isset(ext_features, len,
 378                               NL80211_EXT_FEATURE_BEACON_RATE_VHT))
 379                 capa->flags |= WPA_DRIVER_FLAGS_BEACON_RATE_VHT;
 380
 381         if (ext_feature_isset(ext_features, len,
 382                               NL80211_EXT_FEATURE_SET_SCAN_DWELL))
 383                 capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_SET_SCAN_DWELL;
 384
 385         if (ext_feature_isset(ext_features, len,
 386                               NL80211_EXT_FEATURE_SCAN_START_TIME) &&
 387             ext_feature_isset(ext_features, len,
 388                               NL80211_EXT_FEATURE_BSS_PARENT_TSF) &&
 389             ext_feature_isset(ext_features, len,
 390                               NL80211_EXT_FEATURE_SET_SCAN_DWELL))
 391                 capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_BEACON_REPORT;
 392         if (ext_feature_isset(ext_features, len,
 393                               NL80211_EXT_FEATURE_MGMT_TX_RANDOM_TA))
 394                 capa->flags |= WPA_DRIVER_FLAGS_MGMT_TX_RANDOM_TA;
 395         if (ext_feature_isset(ext_features, len,
 396                               NL80211_EXT_FEATURE_MGMT_TX_RANDOM_TA_CONNECTED))
 397                 capa->flags |= WPA_DRIVER_FLAGS_MGMT_TX_RANDOM_TA_CONNECTED;
 398         if (ext_feature_isset(ext_features, len,
 399                               NL80211_EXT_FEATURE_SCHED_SCAN_RELATIVE_RSSI))
 400                 capa->flags |= WPA_DRIVER_FLAGS_SCHED_SCAN_RELATIVE_RSSI;
 401         if (ext_feature_isset(ext_features, len,
 402                               NL80211_EXT_FEATURE_FILS_SK_OFFLOAD))
 403                 capa->flags |= WPA_DRIVER_FLAGS_FILS_SK_OFFLOAD;
 404
 405         if (ext_feature_isset(ext_features, len,
 406                               NL80211_EXT_FEATURE_4WAY_HANDSHAKE_STA_PSK))
 407                 capa->flags |= WPA_DRIVER_FLAGS_4WAY_HANDSHAKE_PSK;
 408         if (ext_feature_isset(ext_features, len,
 409                               NL80211_EXT_FEATURE_4WAY_HANDSHAKE_STA_1X))
 410                 capa->flags |= WPA_DRIVER_FLAGS_4WAY_HANDSHAKE_8021X;
 411
 412         if (ext_feature_isset(ext_features, len,
 413                               NL80211_EXT_FEATURE_MFP_OPTIONAL))
 414                 capa->flags |= WPA_DRIVER_FLAGS_MFP_OPTIONAL;
 415
 416         if (ext_feature_isset(ext_features, len,
 417                               NL80211_EXT_FEATURE_DFS_OFFLOAD))
 418                 capa->flags |= WPA_DRIVER_FLAGS_DFS_OFFLOAD;
 419
 420 #ifdef CONFIG_MBO
 421         if (ext_feature_isset(ext_features, len,
 422                               NL80211_EXT_FEATURE_FILS_MAX_CHANNEL_TIME) &&
 423             ext_feature_isset(ext_features, len,
 424                               NL80211_EXT_FEATURE_ACCEPT_BCAST_PROBE_RESP) &&
 425             ext_feature_isset(ext_features, len,
 426                               NL80211_EXT_FEATURE_OCE_PROBE_REQ_HIGH_TX_RATE) &&
 427             ext_feature_isset(
 428                     ext_features, len,
 429                     NL80211_EXT_FEATURE_OCE_PROBE_REQ_DEFERRAL_SUPPRESSION))
 430                 capa->flags |= WPA_DRIVER_FLAGS_OCE_STA;
 431 #endif /* CONFIG_MBO */
 432
 433         if (ext_feature_isset(ext_features, len,
 434                               NL80211_EXT_FEATURE_ENABLE_FTM_RESPONDER))
 435                 capa->flags |= WPA_DRIVER_FLAGS_FTM_RESPONDER;
 436 }
 437
 438
 439 static void wiphy_info_feature_flags(struct wiphy_info_data *info,
 440                                      struct nlattr *tb)
 441 {
 442         u32 flags;
 443         struct wpa_driver_capa *capa = info->capa;
 444
 445         if (tb == NULL)
 446                 return;
 447
 448         flags = nla_get_u32(tb);
 449
 450         if (flags & NL80211_FEATURE_SK_TX_STATUS)
 451                 info->data_tx_status = 1;
 452
 453         if (flags & NL80211_FEATURE_INACTIVITY_TIMER)
 454                 capa->flags |= WPA_DRIVER_FLAGS_INACTIVITY_TIMER;
 455
 456         if (flags & NL80211_FEATURE_SAE)
 457                 capa->flags |= WPA_DRIVER_FLAGS_SAE;
 458
 459         if (flags & NL80211_FEATURE_NEED_OBSS_SCAN)
 460                 capa->flags |= WPA_DRIVER_FLAGS_OBSS_SCAN;
 461
 462         if (flags & NL80211_FEATURE_AP_MODE_CHAN_WIDTH_CHANGE)
 463                 capa->flags |= WPA_DRIVER_FLAGS_HT_2040_COEX;
 464
 465         if (flags & NL80211_FEATURE_TDLS_CHANNEL_SWITCH) {
 466                 wpa_printf(MSG_DEBUG, "nl80211: TDLS channel switch");
 467                 capa->flags |= WPA_DRIVER_FLAGS_TDLS_CHANNEL_SWITCH;
 468         }
 469
 470         if (flags & NL80211_FEATURE_P2P_GO_CTWIN)
 471                 info->p2p_go_ctwindow_supported = 1;
 472
 473         if (flags & NL80211_FEATURE_LOW_PRIORITY_SCAN)
 474                 info->have_low_prio_scan = 1;
 475
 476         if (flags & NL80211_FEATURE_SCAN_RANDOM_MAC_ADDR)
 477                 info->mac_addr_rand_scan_supported = 1;
 478
 479         if (flags & NL80211_FEATURE_SCHED_SCAN_RANDOM_MAC_ADDR)
 480                 info->mac_addr_rand_sched_scan_supported = 1;
 481
 482         if (flags & NL80211_FEATURE_STATIC_SMPS)
 483                 capa->smps_modes |= WPA_DRIVER_SMPS_MODE_STATIC;
 484
 485         if (flags & NL80211_FEATURE_DYNAMIC_SMPS)
 486                 capa->smps_modes |= WPA_DRIVER_SMPS_MODE_DYNAMIC;
 487
 488         if (flags & NL80211_FEATURE_SUPPORTS_WMM_ADMISSION)
 489                 info->wmm_ac_supported = 1;
 490
 491         if (flags & NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES)
 492                 capa->rrm_flags |= WPA_DRIVER_FLAGS_DS_PARAM_SET_IE_IN_PROBES;
 493
 494         if (flags & NL80211_FEATURE_WFA_TPC_IE_IN_PROBES)
 495                 capa->rrm_flags |= WPA_DRIVER_FLAGS_WFA_TPC_IE_IN_PROBES;
 496
 497         if (flags & NL80211_FEATURE_QUIET)
 498                 capa->rrm_flags |= WPA_DRIVER_FLAGS_QUIET;
 499
 500         if (flags & NL80211_FEATURE_TX_POWER_INSERTION)
 501                 capa->rrm_flags |= WPA_DRIVER_FLAGS_TX_POWER_INSERTION;
 502
 503         if (flags & NL80211_FEATURE_HT_IBSS)
 504                 capa->flags |= WPA_DRIVER_FLAGS_HT_IBSS;
 505
 506         if (flags & NL80211_FEATURE_FULL_AP_CLIENT_STATE)
 507                 capa->flags |= WPA_DRIVER_FLAGS_FULL_AP_CLIENT_STATE;
 508 }
 509
 510
 511 static void wiphy_info_probe_resp_offload(struct wpa_driver_capa *capa,
 512                                           struct nlattr *tb)
 513 {
 514         u32 protocols;
 515
 516         if (tb == NULL)
 517                 return;
 518
 519         protocols = nla_get_u32(tb);
 520         wpa_printf(MSG_DEBUG, "nl80211: Supports Probe Response offload in AP "
 521                    "mode");
 522         capa->flags |= WPA_DRIVER_FLAGS_PROBE_RESP_OFFLOAD;
 523         capa->probe_resp_offloads = probe_resp_offload_support(protocols);
 524 }
 525
 526
 527 static void wiphy_info_wowlan_triggers(struct wpa_driver_capa *capa,
 528                                        struct nlattr *tb)
 529 {
 530         struct nlattr *triggers[MAX_NL80211_WOWLAN_TRIG + 1];
 531
 532         if (tb == NULL)
 533                 return;
 534
 535         if (nla_parse_nested(triggers, MAX_NL80211_WOWLAN_TRIG,
 536                              tb, NULL))
 537                 return;
 538
 539         if (triggers[NL80211_WOWLAN_TRIG_ANY])
 540                 capa->wowlan_triggers.any = 1;
 541         if (triggers[NL80211_WOWLAN_TRIG_DISCONNECT])
 542                 capa->wowlan_triggers.disconnect = 1;
 543         if (triggers[NL80211_WOWLAN_TRIG_MAGIC_PKT])
 544                 capa->wowlan_triggers.magic_pkt = 1;
 545         if (triggers[NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE])
 546                 capa->wowlan_triggers.gtk_rekey_failure = 1;
 547         if (triggers[NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST])
 548                 capa->wowlan_triggers.eap_identity_req = 1;
 549         if (triggers[NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE])
 550                 capa->wowlan_triggers.four_way_handshake = 1;
 551         if (triggers[NL80211_WOWLAN_TRIG_RFKILL_RELEASE])
 552                 capa->wowlan_triggers.rfkill_release = 1;
 553 }
 554
 555
 556 static void wiphy_info_extended_capab(struct wpa_driver_nl80211_data *drv,
 557                                       struct nlattr *tb)
 558 {
 559         int rem = 0, i;
 560         struct nlattr *tb1[NL80211_ATTR_MAX + 1], *attr;
 561
 562         if (!tb || drv->num_iface_ext_capa == NL80211_IFTYPE_MAX)
 563                 return;
 564
 565         nla_for_each_nested(attr, tb, rem) {
 566                 unsigned int len;
 567                 struct drv_nl80211_ext_capa *capa;
 568
 569                 nla_parse(tb1, NL80211_ATTR_MAX, nla_data(attr),
 570                           nla_len(attr), NULL);
 571
 572                 if (!tb1[NL80211_ATTR_IFTYPE] ||
 573                     !tb1[NL80211_ATTR_EXT_CAPA] ||
 574                     !tb1[NL80211_ATTR_EXT_CAPA_MASK])
 575                         continue;
 576
 577                 capa = &drv->iface_ext_capa[drv->num_iface_ext_capa];
 578                 capa->iftype = nla_get_u32(tb1[NL80211_ATTR_IFTYPE]);
 579                 wpa_printf(MSG_DEBUG,
 580                            "nl80211: Driver-advertised extended capabilities for interface type %s",
 581                            nl80211_iftype_str(capa->iftype));
 582
 583                 len = nla_len(tb1[NL80211_ATTR_EXT_CAPA]);
 584                 capa->ext_capa = os_memdup(nla_data(tb1[NL80211_ATTR_EXT_CAPA]),
 585                                            len);
 586                 if (!capa->ext_capa)
 587                         goto err;
 588
 589                 capa->ext_capa_len = len;
 590                 wpa_hexdump(MSG_DEBUG, "nl80211: Extended capabilities",
 591                             capa->ext_capa, capa->ext_capa_len);
 592
 593                 len = nla_len(tb1[NL80211_ATTR_EXT_CAPA_MASK]);
 594                 capa->ext_capa_mask =
 595                         os_memdup(nla_data(tb1[NL80211_ATTR_EXT_CAPA_MASK]),
 596                                   len);
 597                 if (!capa->ext_capa_mask)
 598                         goto err;
 599
 600                 wpa_hexdump(MSG_DEBUG, "nl80211: Extended capabilities mask",
 601                             capa->ext_capa_mask, capa->ext_capa_len);
 602
 603                 drv->num_iface_ext_capa++;
 604                 if (drv->num_iface_ext_capa == NL80211_IFTYPE_MAX)
 605                         break;
 606         }
 607
 608         return;
 609
 610 err:
 611         /* Cleanup allocated memory on error */
 612         for (i = 0; i < NL80211_IFTYPE_MAX; i++) {
 613                 os_free(drv->iface_ext_capa[i].ext_capa);
 614                 drv->iface_ext_capa[i].ext_capa = NULL;
 615                 os_free(drv->iface_ext_capa[i].ext_capa_mask);
 616                 drv->iface_ext_capa[i].ext_capa_mask = NULL;
 617                 drv->iface_ext_capa[i].ext_capa_len = 0;
 618         }
 619         drv->num_iface_ext_capa = 0;
 620 }
 621
 622
 623 static int wiphy_info_handler(struct nl_msg *msg, void *arg)
 624 {
 625         struct nlattr *tb[NL80211_ATTR_MAX + 1];
 626         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 627         struct wiphy_info_data *info = arg;
 628         struct wpa_driver_capa *capa = info->capa;
 629         struct wpa_driver_nl80211_data *drv = info->drv;
 630
 631         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 632                   genlmsg_attrlen(gnlh, 0), NULL);
 633
 634         if (tb[NL80211_ATTR_WIPHY])
 635                 drv->wiphy_idx = nla_get_u32(tb[NL80211_ATTR_WIPHY]);
 636
 637         if (tb[NL80211_ATTR_WIPHY_NAME])
 638                 os_strlcpy(drv->phyname,
 639                            nla_get_string(tb[NL80211_ATTR_WIPHY_NAME]),
 640                            sizeof(drv->phyname));
 641         if (tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS])
 642                 capa->max_scan_ssids =
 643                         nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS]);
 644
 645         if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS])
 646                 capa->max_sched_scan_ssids =
 647                         nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS]);
 648
 649         if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_PLANS] &&
 650             tb[NL80211_ATTR_MAX_SCAN_PLAN_INTERVAL] &&
 651             tb[NL80211_ATTR_MAX_SCAN_PLAN_ITERATIONS]) {
 652                 capa->max_sched_scan_plans =
 653                         nla_get_u32(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_PLANS]);
 654
 655                 capa->max_sched_scan_plan_interval =
 656                         nla_get_u32(tb[NL80211_ATTR_MAX_SCAN_PLAN_INTERVAL]);
 657
 658                 capa->max_sched_scan_plan_iterations =
 659                         nla_get_u32(tb[NL80211_ATTR_MAX_SCAN_PLAN_ITERATIONS]);
 660         }
 661
 662         if (tb[NL80211_ATTR_MAX_MATCH_SETS])
 663                 capa->max_match_sets =
 664                         nla_get_u8(tb[NL80211_ATTR_MAX_MATCH_SETS]);
 665
 666         if (tb[NL80211_ATTR_MAC_ACL_MAX])
 667                 capa->max_acl_mac_addrs =
 668                         nla_get_u8(tb[NL80211_ATTR_MAC_ACL_MAX]);
 669
 670         wiphy_info_supported_iftypes(info, tb[NL80211_ATTR_SUPPORTED_IFTYPES]);
 671         wiphy_info_iface_comb(info, tb[NL80211_ATTR_INTERFACE_COMBINATIONS]);
 672         wiphy_info_supp_cmds(info, tb[NL80211_ATTR_SUPPORTED_COMMANDS]);
 673         wiphy_info_cipher_suites(info, tb[NL80211_ATTR_CIPHER_SUITES]);
 674
 675         if (tb[NL80211_ATTR_OFFCHANNEL_TX_OK]) {
 676                 wpa_printf(MSG_DEBUG, "nl80211: Using driver-based "
 677                            "off-channel TX");
 678                 capa->flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_TX;
 679         }
 680
 681         if (tb[NL80211_ATTR_ROAM_SUPPORT]) {
 682                 wpa_printf(MSG_DEBUG, "nl80211: Using driver-based roaming");
 683                 capa->flags |= WPA_DRIVER_FLAGS_BSS_SELECTION;
 684         }
 685
 686         wiphy_info_max_roc(capa,
 687                            tb[NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION]);
 688
 689         if (tb[NL80211_ATTR_SUPPORT_AP_UAPSD])
 690                 capa->flags |= WPA_DRIVER_FLAGS_AP_UAPSD;
 691
 692         wiphy_info_tdls(capa, tb[NL80211_ATTR_TDLS_SUPPORT],
 693                         tb[NL80211_ATTR_TDLS_EXTERNAL_SETUP]);
 694
 695         if (tb[NL80211_ATTR_DEVICE_AP_SME])
 696                 info->device_ap_sme = 1;
 697
 698         wiphy_info_feature_flags(info, tb[NL80211_ATTR_FEATURE_FLAGS]);
 699         wiphy_info_ext_feature_flags(info, tb[NL80211_ATTR_EXT_FEATURES]);
 700         wiphy_info_probe_resp_offload(capa,
 701                                       tb[NL80211_ATTR_PROBE_RESP_OFFLOAD]);
 702
 703         if (tb[NL80211_ATTR_EXT_CAPA] && tb[NL80211_ATTR_EXT_CAPA_MASK] &&
 704             drv->extended_capa == NULL) {
 705                 drv->extended_capa =
 706                         os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA]));
 707                 if (drv->extended_capa) {
 708                         os_memcpy(drv->extended_capa,
 709                                   nla_data(tb[NL80211_ATTR_EXT_CAPA]),
 710                                   nla_len(tb[NL80211_ATTR_EXT_CAPA]));
 711                         drv->extended_capa_len =
 712                                 nla_len(tb[NL80211_ATTR_EXT_CAPA]);
 713                         wpa_hexdump(MSG_DEBUG,
 714                                     "nl80211: Driver-advertised extended capabilities (default)",
 715                                     drv->extended_capa, drv->extended_capa_len);
 716                 }
 717                 drv->extended_capa_mask =
 718                         os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
 719                 if (drv->extended_capa_mask) {
 720                         os_memcpy(drv->extended_capa_mask,
 721                                   nla_data(tb[NL80211_ATTR_EXT_CAPA_MASK]),
 722                                   nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
 723                         wpa_hexdump(MSG_DEBUG,
 724                                     "nl80211: Driver-advertised extended capabilities mask (default)",
 725                                     drv->extended_capa_mask,
 726                                     drv->extended_capa_len);
 727                 } else {
 728                         os_free(drv->extended_capa);
 729                         drv->extended_capa = NULL;
 730                         drv->extended_capa_len = 0;
 731                 }
 732         }
 733
 734         wiphy_info_extended_capab(drv, tb[NL80211_ATTR_IFTYPE_EXT_CAPA]);
 735
 736         if (tb[NL80211_ATTR_VENDOR_DATA]) {
 737                 struct nlattr *nl;
 738                 int rem;
 739
 740                 nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_DATA], rem) {
 741                         struct nl80211_vendor_cmd_info *vinfo;
 742                         if (nla_len(nl) != sizeof(*vinfo)) {
 743                                 wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
 744                                 continue;
 745                         }
 746                         vinfo = nla_data(nl);
 747                         if (vinfo->vendor_id == OUI_QCA) {
 748                                 switch (vinfo->subcmd) {
 749                                 case QCA_NL80211_VENDOR_SUBCMD_TEST:
 750                                         drv->vendor_cmd_test_avail = 1;
 751                                         break;
 752 #ifdef CONFIG_DRIVER_NL80211_QCA
 753                                 case QCA_NL80211_VENDOR_SUBCMD_ROAMING:
 754                                         drv->roaming_vendor_cmd_avail = 1;
 755                                         break;
 756                                 case QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY:
 757                                         drv->dfs_vendor_cmd_avail = 1;
 758                                         break;
 759                                 case QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES:
 760                                         drv->get_features_vendor_cmd_avail = 1;
 761                                         break;
 762                                 case QCA_NL80211_VENDOR_SUBCMD_GET_PREFERRED_FREQ_LIST:
 763                                         drv->get_pref_freq_list = 1;
 764                                         break;
 765                                 case QCA_NL80211_VENDOR_SUBCMD_SET_PROBABLE_OPER_CHANNEL:
 766                                         drv->set_prob_oper_freq = 1;
 767                                         break;
 768                                 case QCA_NL80211_VENDOR_SUBCMD_DO_ACS:
 769                                         drv->capa.flags |=
 770                                                 WPA_DRIVER_FLAGS_ACS_OFFLOAD;
 771                                         break;
 772                                 case QCA_NL80211_VENDOR_SUBCMD_SETBAND:
 773                                         drv->setband_vendor_cmd_avail = 1;
 774                                         break;
 775                                 case QCA_NL80211_VENDOR_SUBCMD_TRIGGER_SCAN:
 776                                         drv->scan_vendor_cmd_avail = 1;
 777                                         break;
 778                                 case QCA_NL80211_VENDOR_SUBCMD_SET_WIFI_CONFIGURATION:
 779                                         drv->set_wifi_conf_vendor_cmd_avail = 1;
 780                                         break;
 781                                 case QCA_NL80211_VENDOR_SUBCMD_GET_HE_CAPABILITIES:
 782                                         drv->he_capab_vendor_cmd_avail = 1;
 783                                         break;
 784                                 case QCA_NL80211_VENDOR_SUBCMD_FETCH_BSS_TRANSITION_STATUS:
 785                                         drv->fetch_bss_trans_status = 1;
 786                                         break;
 787                                 case QCA_NL80211_VENDOR_SUBCMD_ROAM:
 788                                         drv->roam_vendor_cmd_avail = 1;
 789                                         break;
 790                                 case QCA_NL80211_VENDOR_SUBCMD_GET_SUPPORTED_AKMS:
 791                                         drv->get_supported_akm_suites_avail = 1;
 792                                         break;
 793 #endif /* CONFIG_DRIVER_NL80211_QCA */
 794                                 }
 795                         }
 796
 797                         wpa_printf(MSG_DEBUG, "nl80211: Supported vendor command: vendor_id=0x%x subcmd=%u",
 798                                    vinfo->vendor_id, vinfo->subcmd);
 799                 }
 800         }
 801
 802         if (tb[NL80211_ATTR_VENDOR_EVENTS]) {
 803                 struct nlattr *nl;
 804                 int rem;
 805
 806                 nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_EVENTS], rem) {
 807                         struct nl80211_vendor_cmd_info *vinfo;
 808                         if (nla_len(nl) != sizeof(*vinfo)) {
 809                                 wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
 810                                 continue;
 811                         }
 812                         vinfo = nla_data(nl);
 813                         wpa_printf(MSG_DEBUG, "nl80211: Supported vendor event: vendor_id=0x%x subcmd=%u",
 814                                    vinfo->vendor_id, vinfo->subcmd);
 815                 }
 816         }
 817
 818         wiphy_info_wowlan_triggers(capa,
 819                                    tb[NL80211_ATTR_WOWLAN_TRIGGERS_SUPPORTED]);
 820
 821         if (tb[NL80211_ATTR_MAX_AP_ASSOC_STA])
 822                 capa->max_stations =
 823                         nla_get_u32(tb[NL80211_ATTR_MAX_AP_ASSOC_STA]);
 824
 825         if (tb[NL80211_ATTR_MAX_CSA_COUNTERS])
 826                 capa->max_csa_counters =
 827                         nla_get_u8(tb[NL80211_ATTR_MAX_CSA_COUNTERS]);
 828
 829         if (tb[NL80211_ATTR_WIPHY_SELF_MANAGED_REG])
 830                 capa->flags |= WPA_DRIVER_FLAGS_SELF_MANAGED_REGULATORY;
 831
 832         return NL_SKIP;
 833 }
 834
 835
 836 static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
 837                                        struct wiphy_info_data *info)
 838 {
 839         u32 feat;
 840         struct nl_msg *msg;
 841         int flags = 0;
 842
 843         os_memset(info, 0, sizeof(*info));
 844         info->capa = &drv->capa;
 845         info->drv = drv;
 846
 847         feat = get_nl80211_protocol_features(drv);
 848         if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
 849                 flags = NLM_F_DUMP;
 850         msg = nl80211_cmd_msg(drv->first_bss, flags, NL80211_CMD_GET_WIPHY);
 851         if (!msg || nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
 852                 nlmsg_free(msg);
 853                 return -1;
 854         }
 855
 856         if (send_and_recv_msgs(drv, msg, wiphy_info_handler, info))
 857                 return -1;
 858
 859         if (info->auth_supported)
 860                 drv->capa.flags |= WPA_DRIVER_FLAGS_SME;
 861         else if (!info->connect_supported) {
 862                 wpa_printf(MSG_INFO, "nl80211: Driver does not support "
 863                            "authentication/association or connect commands");
 864                 info->error = 1;
 865         }
 866
 867         if (info->p2p_go_supported && info->p2p_client_supported)
 868                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CAPABLE;
 869         if (info->p2p_concurrent) {
 870                 wpa_printf(MSG_DEBUG, "nl80211: Use separate P2P group "
 871                            "interface (driver advertised support)");
 872                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT;
 873                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P;
 874         }
 875         if (info->num_multichan_concurrent > 1) {
 876                 wpa_printf(MSG_DEBUG, "nl80211: Enable multi-channel "
 877                            "concurrent (driver advertised support)");
 878                 drv->capa.num_multichan_concurrent =
 879                         info->num_multichan_concurrent;
 880         }
 881         if (drv->capa.flags & WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE)
 882                 wpa_printf(MSG_DEBUG, "nl80211: use P2P_DEVICE support");
 883
 884         /* default to 5000 since early versions of mac80211 don't set it */
 885         if (!drv->capa.max_remain_on_chan)
 886                 drv->capa.max_remain_on_chan = 5000;
 887
 888         drv->capa.wmm_ac_supported = info->wmm_ac_supported;
 889
 890         drv->capa.mac_addr_rand_sched_scan_supported =
 891                 info->mac_addr_rand_sched_scan_supported;
 892         drv->capa.mac_addr_rand_scan_supported =
 893                 info->mac_addr_rand_scan_supported;
 894
 895         if (info->channel_switch_supported) {
 896                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP_CSA;
 897                 if (!drv->capa.max_csa_counters)
 898                         drv->capa.max_csa_counters = 1;
 899         }
 900
 901         if (!drv->capa.max_sched_scan_plans) {
 902                 drv->capa.max_sched_scan_plans = 1;
 903                 drv->capa.max_sched_scan_plan_interval = UINT32_MAX;
 904                 drv->capa.max_sched_scan_plan_iterations = 0;
 905         }
 906
 907         return 0;
 908 }
 909
 910
 911 #ifdef CONFIG_DRIVER_NL80211_QCA
 912
 913 static int dfs_info_handler(struct nl_msg *msg, void *arg)
 914 {
 915         struct nlattr *tb[NL80211_ATTR_MAX + 1];
 916         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 917         int *dfs_capability_ptr = arg;
 918
 919         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 920                   genlmsg_attrlen(gnlh, 0), NULL);
 921
 922         if (tb[NL80211_ATTR_VENDOR_DATA]) {
 923                 struct nlattr *nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
 924                 struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
 925
 926                 nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
 927                           nla_data(nl_vend), nla_len(nl_vend), NULL);
 928
 929                 if (tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]) {
 930                         u32 val;
 931                         val = nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]);
 932                         wpa_printf(MSG_DEBUG, "nl80211: DFS offload capability: %u",
 933                                    val);
 934                         *dfs_capability_ptr = val;
 935                 }
 936         }
 937
 938         return NL_SKIP;
 939 }
 940
 941
 942 static void qca_nl80211_check_dfs_capa(struct wpa_driver_nl80211_data *drv)
 943 {
 944         struct nl_msg *msg;
 945         int dfs_capability = 0;
 946         int ret;
 947
 948         if (!drv->dfs_vendor_cmd_avail)
 949                 return;
 950
 951         if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
 952             nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
 953             nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
 954                         QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY)) {
 955                 nlmsg_free(msg);
 956                 return;
 957         }
 958
 959         ret = send_and_recv_msgs(drv, msg, dfs_info_handler, &dfs_capability);
 960         if (!ret && dfs_capability)
 961                 drv->capa.flags |= WPA_DRIVER_FLAGS_DFS_OFFLOAD;
 962 }
 963
 964
 965 static unsigned int get_akm_suites_info(struct nlattr *tb)
 966 {
 967         int i, num;
 968         unsigned int key_mgmt = 0;
 969         u32 *akms;
 970
 971         if (!tb)
 972                 return 0;
 973
 974         num = nla_len(tb) / sizeof(u32);
 975         akms = nla_data(tb);
 976         for (i = 0; i < num; i++) {
 977                 u32 a = akms[i];
 978
 979                 wpa_printf(MSG_DEBUG,
 980                            "nl80211: Supported AKM %02x-%02x-%02x:%u",
 981                            a >> 24, (a >> 16) & 0xff,
 982                            (a >> 8) & 0xff, a & 0xff);
 983                 switch (a) {
 984                 case RSN_AUTH_KEY_MGMT_UNSPEC_802_1X:
 985                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA |
 986                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2;
 987                         break;
 988                 case RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X:
 989                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
 990                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
 991                         break;
 992                 case RSN_AUTH_KEY_MGMT_FT_802_1X:
 993                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT;
 994                         break;
 995                 case RSN_AUTH_KEY_MGMT_FT_PSK:
 996                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT_PSK;
 997                         break;
 998                 case RSN_AUTH_KEY_MGMT_802_1X_SUITE_B:
 999                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B;
1000                         break;
1001                 case RSN_AUTH_KEY_MGMT_802_1X_SUITE_B_192:
1002                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B_192;
1003                         break;
1004                 case RSN_AUTH_KEY_MGMT_OWE:
1005                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_OWE;
1006                         break;
1007                 case RSN_AUTH_KEY_MGMT_DPP:
1008                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_DPP;
1009                         break;
1010                 case RSN_AUTH_KEY_MGMT_FILS_SHA256:
1011                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA256;
1012                         break;
1013                 case RSN_AUTH_KEY_MGMT_FILS_SHA384:
1014                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA384;
1015                         break;
1016                 case RSN_AUTH_KEY_MGMT_FT_FILS_SHA256:
1017                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA256;
1018                         break;
1019                 case RSN_AUTH_KEY_MGMT_FT_FILS_SHA384:
1020                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA384;
1021                         break;
1022                 case RSN_AUTH_KEY_MGMT_SAE:
1023                         key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_SAE;
1024                         break;
1025                 }
1026         }
1027
1028         return key_mgmt;
1029 }
1030
1031
1032 static int get_akm_suites_handler(struct nl_msg *msg, void *arg)
1033 {
1034         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1035         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1036         unsigned int *key_mgmt = arg;
1037
1038         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1039                   genlmsg_attrlen(gnlh, 0), NULL);
1040
1041         if (tb[NL80211_ATTR_VENDOR_DATA]) {
1042                 struct nlattr *nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
1043                 struct nlattr *tb_data[NL80211_ATTR_MAX + 1];
1044
1045                 nla_parse(tb_data, NL80211_ATTR_MAX,
1046                           nla_data(nl_vend), nla_len(nl_vend), NULL);
1047
1048                 *key_mgmt =
1049                         get_akm_suites_info(tb_data[NL80211_ATTR_AKM_SUITES]);
1050         }
1051
1052         return NL_SKIP;
1053 }
1054
1055
1056 static int qca_nl80211_get_akm_suites(struct wpa_driver_nl80211_data *drv)
1057 {
1058         struct nl_msg *msg;
1059         unsigned int key_mgmt = 0;
1060         int ret;
1061
1062         if (!drv->get_supported_akm_suites_avail)
1063                 return -1;
1064
1065         if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
1066             nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
1067             nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
1068                         QCA_NL80211_VENDOR_SUBCMD_GET_SUPPORTED_AKMS)) {
1069                 nlmsg_free(msg);
1070                 return -1;
1071         }
1072
1073         ret = send_and_recv_msgs(drv, msg, get_akm_suites_handler, &key_mgmt);
1074         if (!ret) {
1075                 wpa_printf(MSG_DEBUG,
1076                            "nl80211: Replace capa.key_mgmt based on driver advertised capabilities: 0x%x",
1077                            key_mgmt);
1078                 drv->capa.key_mgmt = key_mgmt;
1079         }
1080
1081         return ret;
1082 }
1083
1084
1085 static int qca_nl80211_he_capab_handler(struct nl_msg *msg, void *arg)
1086 {
1087         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1088         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1089         struct he_capabilities *he_capab = arg;
1090         struct nlattr *nl_vend;
1091         struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_CAPABILITIES_MAX + 1];
1092         size_t len;
1093
1094         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1095                   genlmsg_attrlen(gnlh, 0), NULL);
1096
1097         if (!tb[NL80211_ATTR_VENDOR_DATA])
1098                 return NL_SKIP;
1099
1100         nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
1101         nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_HE_CAPABILITIES_MAX,
1102                   nla_data(nl_vend), nla_len(nl_vend), NULL);
1103
1104         if (tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_SUPPORTED]) {
1105                 u8 he_supported;
1106
1107                 he_supported = nla_get_u8(
1108                         tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_SUPPORTED]);
1109                 wpa_printf(MSG_DEBUG, "nl80211: HE capabilities supported: %u",
1110                            he_supported);
1111                 he_capab->he_supported = he_supported;
1112                 if (!he_supported)
1113                         return NL_SKIP;
1114         }
1115
1116         if (tb_vendor[QCA_WLAN_VENDOR_ATTR_PHY_CAPAB]) {
1117                 len = nla_len(tb_vendor[QCA_WLAN_VENDOR_ATTR_PHY_CAPAB]);
1118
1119                 if (len > sizeof(he_capab->phy_cap))
1120                         len = sizeof(he_capab->phy_cap);
1121                 os_memcpy(he_capab->phy_cap,
1122                           nla_data(tb_vendor[QCA_WLAN_VENDOR_ATTR_PHY_CAPAB]),
1123                           len);
1124         }
1125
1126         if (tb_vendor[QCA_WLAN_VENDOR_ATTR_MAC_CAPAB])
1127                 he_capab->mac_cap =
1128                         nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_MAC_CAPAB]);
1129
1130         if (tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_MCS])
1131                 he_capab->mcs =
1132                         nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_MCS]);
1133
1134         if (tb_vendor[QCA_WLAN_VENDOR_ATTR_NUM_SS])
1135                 he_capab->ppet.numss_m1 =
1136                         nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_NUM_SS]);
1137
1138         if (tb_vendor[QCA_WLAN_VENDOR_ATTR_RU_IDX_MASK])
1139                 he_capab->ppet.ru_count =
1140                         nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_RU_IDX_MASK]);
1141
1142         if (tb_vendor[QCA_WLAN_VENDOR_ATTR_PPE_THRESHOLD]) {
1143                 len = nla_len(tb_vendor[QCA_WLAN_VENDOR_ATTR_PPE_THRESHOLD]);
1144
1145                 if (len > sizeof(he_capab->ppet.ppet16_ppet8_ru3_ru0))
1146                         len = sizeof(he_capab->ppet.ppet16_ppet8_ru3_ru0);
1147                 os_memcpy(he_capab->ppet.ppet16_ppet8_ru3_ru0,
1148                           nla_data(tb_vendor[QCA_WLAN_VENDOR_ATTR_PPE_THRESHOLD]),
1149                           len);
1150         }
1151
1152         return NL_SKIP;
1153 }
1154
1155
1156 static void qca_nl80211_check_he_capab(struct wpa_driver_nl80211_data *drv)
1157 {
1158         struct nl_msg *msg;
1159         int ret;
1160
1161         if (!drv->he_capab_vendor_cmd_avail)
1162                 return;
1163
1164         if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
1165                 nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
1166                 nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
1167                             QCA_NL80211_VENDOR_SUBCMD_GET_HE_CAPABILITIES)) {
1168                 nlmsg_free(msg);
1169                 return;
1170         }
1171
1172         ret = send_and_recv_msgs(drv, msg, qca_nl80211_he_capab_handler,
1173                                  &drv->he_capab);
1174         if (!ret && drv->he_capab.he_supported)
1175                 drv->capa.flags |= WPA_DRIVER_FLAGS_HE_CAPABILITIES;
1176 }
1177
1178
1179 struct features_info {
1180         u8 *flags;
1181         size_t flags_len;
1182         struct wpa_driver_capa *capa;
1183 };
1184
1185
1186 static int features_info_handler(struct nl_msg *msg, void *arg)
1187 {
1188         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1189         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1190         struct features_info *info = arg;
1191         struct nlattr *nl_vend, *attr;
1192
1193         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1194                   genlmsg_attrlen(gnlh, 0), NULL);
1195
1196         nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
1197         if (nl_vend) {
1198                 struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
1199
1200                 nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
1201                           nla_data(nl_vend), nla_len(nl_vend), NULL);
1202
1203                 attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_FEATURE_FLAGS];
1204                 if (attr) {
1205                         int len = nla_len(attr);
1206                         info->flags = os_malloc(len);
1207                         if (info->flags != NULL) {
1208                                 os_memcpy(info->flags, nla_data(attr), len);
1209                                 info->flags_len = len;
1210                         }
1211                 }
1212                 attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_CONCURRENCY_CAPA];
1213                 if (attr)
1214                         info->capa->conc_capab = nla_get_u32(attr);
1215
1216                 attr = tb_vendor[
1217                         QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_2_4_BAND];
1218                 if (attr)
1219                         info->capa->max_conc_chan_2_4 = nla_get_u32(attr);
1220
1221                 attr = tb_vendor[
1222                         QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_5_0_BAND];
1223                 if (attr)
1224                         info->capa->max_conc_chan_5_0 = nla_get_u32(attr);
1225         }
1226
1227         return NL_SKIP;
1228 }
1229
1230
1231 static int check_feature(enum qca_wlan_vendor_features feature,
1232                          struct features_info *info)
1233 {
1234         size_t idx = feature / 8;
1235
1236         return (idx < info->flags_len) &&
1237                 (info->flags[idx] & BIT(feature % 8));
1238 }
1239
1240
1241 static void qca_nl80211_get_features(struct wpa_driver_nl80211_data *drv)
1242 {
1243         struct nl_msg *msg;
1244         struct features_info info;
1245         int ret;
1246
1247         if (!drv->get_features_vendor_cmd_avail)
1248                 return;
1249
1250         if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
1251             nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
1252             nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
1253                         QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES)) {
1254                 nlmsg_free(msg);
1255                 return;
1256         }
1257
1258         os_memset(&info, 0, sizeof(info));
1259         info.capa = &drv->capa;
1260         ret = send_and_recv_msgs(drv, msg, features_info_handler, &info);
1261         if (ret || !info.flags)
1262                 return;
1263
1264         if (check_feature(QCA_WLAN_VENDOR_FEATURE_KEY_MGMT_OFFLOAD, &info))
1265                 drv->capa.flags |= WPA_DRIVER_FLAGS_KEY_MGMT_OFFLOAD;
1266
1267         if (check_feature(QCA_WLAN_VENDOR_FEATURE_SUPPORT_HW_MODE_ANY, &info))
1268                 drv->capa.flags |= WPA_DRIVER_FLAGS_SUPPORT_HW_MODE_ANY;
1269
1270         if (check_feature(QCA_WLAN_VENDOR_FEATURE_OFFCHANNEL_SIMULTANEOUS,
1271                           &info))
1272                 drv->capa.flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_SIMULTANEOUS;
1273         if (check_feature(QCA_WLAN_VENDOR_FEATURE_P2P_LISTEN_OFFLOAD, &info))
1274                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_LISTEN_OFFLOAD;
1275         if (check_feature(QCA_WLAN_VENDOR_FEATURE_OCE_STA, &info))
1276                 drv->capa.flags |= WPA_DRIVER_FLAGS_OCE_STA;
1277         if (check_feature(QCA_WLAN_VENDOR_FEATURE_OCE_AP, &info))
1278                 drv->capa.flags |= WPA_DRIVER_FLAGS_OCE_AP;
1279         if (check_feature(QCA_WLAN_VENDOR_FEATURE_OCE_STA_CFON, &info))
1280                 drv->capa.flags |= WPA_DRIVER_FLAGS_OCE_STA_CFON;
1281         os_free(info.flags);
1282 }
1283
1284 #endif /* CONFIG_DRIVER_NL80211_QCA */
1285
1286
1287 int wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
1288 {
1289         struct wiphy_info_data info;
1290         if (wpa_driver_nl80211_get_info(drv, &info))
1291                 return -1;
1292
1293         if (info.error)
1294                 return -1;
1295
1296         drv->has_capability = 1;
1297         drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
1298                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
1299                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
1300                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK |
1301                 WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B |
1302                 WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B_192 |
1303                 WPA_DRIVER_CAPA_KEY_MGMT_OWE |
1304                 WPA_DRIVER_CAPA_KEY_MGMT_DPP;
1305
1306         if (drv->capa.flags & WPA_DRIVER_FLAGS_SME)
1307                 drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA256 |
1308                         WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA384 |
1309                         WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA256 |
1310                         WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA384 |
1311                         WPA_DRIVER_CAPA_KEY_MGMT_SAE;
1312         else if (drv->capa.flags & WPA_DRIVER_FLAGS_FILS_SK_OFFLOAD)
1313                 drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA256 |
1314                         WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA384;
1315
1316 #ifdef CONFIG_DRIVER_NL80211_QCA
1317         /* Override drv->capa.key_mgmt based on driver advertised capability
1318          * constraints, if available. */
1319         qca_nl80211_get_akm_suites(drv);
1320 #endif /* CONFIG_DRIVER_NL80211_QCA */
1321
1322         drv->capa.auth = WPA_DRIVER_AUTH_OPEN |
1323                 WPA_DRIVER_AUTH_SHARED |
1324                 WPA_DRIVER_AUTH_LEAP;
1325
1326         drv->capa.flags |= WPA_DRIVER_FLAGS_SANE_ERROR_CODES;
1327         drv->capa.flags |= WPA_DRIVER_FLAGS_SET_KEYS_AFTER_ASSOC_DONE;
1328         drv->capa.flags |= WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
1329
1330         /*
1331          * As all cfg80211 drivers must support cases where the AP interface is
1332          * removed without the knowledge of wpa_supplicant/hostapd, e.g., in
1333          * case that the user space daemon has crashed, they must be able to
1334          * cleanup all stations and key entries in the AP tear down flow. Thus,
1335          * this flag can/should always be set for cfg80211 drivers.
1336          */
1337         drv->capa.flags |= WPA_DRIVER_FLAGS_AP_TEARDOWN_SUPPORT;
1338
1339         if (!info.device_ap_sme) {
1340                 drv->capa.flags |= WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS;
1341
1342                 /*
1343                  * No AP SME is currently assumed to also indicate no AP MLME
1344                  * in the driver/firmware.
1345                  */
1346                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP_MLME;
1347         }
1348
1349         drv->device_ap_sme = info.device_ap_sme;
1350         drv->poll_command_supported = info.poll_command_supported;
1351         drv->data_tx_status = info.data_tx_status;
1352         drv->p2p_go_ctwindow_supported = info.p2p_go_ctwindow_supported;
1353         if (info.set_qos_map_supported)
1354                 drv->capa.flags |= WPA_DRIVER_FLAGS_QOS_MAPPING;
1355         drv->have_low_prio_scan = info.have_low_prio_scan;
1356
1357         /*
1358          * If poll command and tx status are supported, mac80211 is new enough
1359          * to have everything we need to not need monitor interfaces.
1360          */
1361         drv->use_monitor = !info.device_ap_sme &&
1362                 (!info.poll_command_supported || !info.data_tx_status);
1363
1364         /*
1365          * If we aren't going to use monitor interfaces, but the
1366          * driver doesn't support data TX status, we won't get TX
1367          * status for EAPOL frames.
1368          */
1369         if (!drv->use_monitor && !info.data_tx_status)
1370                 drv->capa.flags &= ~WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
1371
1372 #ifdef CONFIG_DRIVER_NL80211_QCA
1373         if (!(info.capa->flags & WPA_DRIVER_FLAGS_DFS_OFFLOAD))
1374                 qca_nl80211_check_dfs_capa(drv);
1375         qca_nl80211_get_features(drv);
1376         qca_nl80211_check_he_capab(drv);
1377
1378         /*
1379          * To enable offchannel simultaneous support in wpa_supplicant, the
1380          * underlying driver needs to support the same along with offchannel TX.
1381          * Offchannel TX support is needed since remain_on_channel and
1382          * action_tx use some common data structures and hence cannot be
1383          * scheduled simultaneously.
1384          */
1385         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_OFFCHANNEL_TX))
1386                 drv->capa.flags &= ~WPA_DRIVER_FLAGS_OFFCHANNEL_SIMULTANEOUS;
1387 #endif /* CONFIG_DRIVER_NL80211_QCA */
1388
1389         return 0;
1390 }
1391
1392
1393 struct phy_info_arg {
1394         u16 *num_modes;
1395         struct hostapd_hw_modes *modes;
1396         int last_mode, last_chan_idx;
1397         int failed;
1398         u8 dfs_domain;
1399 };
1400
1401 static void phy_info_ht_capa(struct hostapd_hw_modes *mode, struct nlattr *capa,
1402                              struct nlattr *ampdu_factor,
1403                              struct nlattr *ampdu_density,
1404                              struct nlattr *mcs_set)
1405 {
1406         if (capa)
1407                 mode->ht_capab = nla_get_u16(capa);
1408
1409         if (ampdu_factor)
1410                 mode->a_mpdu_params |= nla_get_u8(ampdu_factor) & 0x03;
1411
1412         if (ampdu_density)
1413                 mode->a_mpdu_params |= nla_get_u8(ampdu_density) << 2;
1414
1415         if (mcs_set && nla_len(mcs_set) >= 16) {
1416                 u8 *mcs;
1417                 mcs = nla_data(mcs_set);
1418                 os_memcpy(mode->mcs_set, mcs, 16);
1419         }
1420 }
1421
1422
1423 static void phy_info_vht_capa(struct hostapd_hw_modes *mode,
1424                               struct nlattr *capa,
1425                               struct nlattr *mcs_set)
1426 {
1427         if (capa)
1428                 mode->vht_capab = nla_get_u32(capa);
1429
1430         if (mcs_set && nla_len(mcs_set) >= 8) {
1431                 u8 *mcs;
1432                 mcs = nla_data(mcs_set);
1433                 os_memcpy(mode->vht_mcs_set, mcs, 8);
1434         }
1435 }
1436
1437
1438 static void phy_info_freq(struct hostapd_hw_modes *mode,
1439                           struct hostapd_channel_data *chan,
1440                           struct nlattr *tb_freq[])
1441 {
1442         u8 channel;
1443         chan->freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
1444         chan->flag = 0;
1445         chan->allowed_bw = ~0;
1446         chan->dfs_cac_ms = 0;
1447         if (ieee80211_freq_to_chan(chan->freq, &channel) != NUM_HOSTAPD_MODES)
1448                 chan->chan = channel;
1449
1450         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1451                 chan->flag |= HOSTAPD_CHAN_DISABLED;
1452         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IR])
1453                 chan->flag |= HOSTAPD_CHAN_NO_IR;
1454         if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
1455                 chan->flag |= HOSTAPD_CHAN_RADAR;
1456         if (tb_freq[NL80211_FREQUENCY_ATTR_INDOOR_ONLY])
1457                 chan->flag |= HOSTAPD_CHAN_INDOOR_ONLY;
1458         if (tb_freq[NL80211_FREQUENCY_ATTR_GO_CONCURRENT])
1459                 chan->flag |= HOSTAPD_CHAN_GO_CONCURRENT;
1460
1461         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_10MHZ])
1462                 chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_10;
1463         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_20MHZ])
1464                 chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_20;
1465         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_HT40_PLUS])
1466                 chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_40P;
1467         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_HT40_MINUS])
1468                 chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_40M;
1469         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_80MHZ])
1470                 chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_80;
1471         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_160MHZ])
1472                 chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_160;
1473
1474         if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]) {
1475                 enum nl80211_dfs_state state =
1476                         nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]);
1477
1478                 switch (state) {
1479                 case NL80211_DFS_USABLE:
1480                         chan->flag |= HOSTAPD_CHAN_DFS_USABLE;
1481                         break;
1482                 case NL80211_DFS_AVAILABLE:
1483                         chan->flag |= HOSTAPD_CHAN_DFS_AVAILABLE;
1484                         break;
1485                 case NL80211_DFS_UNAVAILABLE:
1486                         chan->flag |= HOSTAPD_CHAN_DFS_UNAVAILABLE;
1487                         break;
1488                 }
1489         }
1490
1491         if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]) {
1492                 chan->dfs_cac_ms = nla_get_u32(
1493                         tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]);
1494         }
1495 }
1496
1497
1498 static int phy_info_freqs(struct phy_info_arg *phy_info,
1499                           struct hostapd_hw_modes *mode, struct nlattr *tb)
1500 {
1501         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1502                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1503                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1504                 [NL80211_FREQUENCY_ATTR_NO_IR] = { .type = NLA_FLAG },
1505                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1506                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
1507                 [NL80211_FREQUENCY_ATTR_DFS_STATE] = { .type = NLA_U32 },
1508                 [NL80211_FREQUENCY_ATTR_NO_10MHZ] = { .type = NLA_FLAG },
1509                 [NL80211_FREQUENCY_ATTR_NO_20MHZ] = { .type = NLA_FLAG },
1510                 [NL80211_FREQUENCY_ATTR_NO_HT40_PLUS] = { .type = NLA_FLAG },
1511                 [NL80211_FREQUENCY_ATTR_NO_HT40_MINUS] = { .type = NLA_FLAG },
1512                 [NL80211_FREQUENCY_ATTR_NO_80MHZ] = { .type = NLA_FLAG },
1513                 [NL80211_FREQUENCY_ATTR_NO_160MHZ] = { .type = NLA_FLAG },
1514         };
1515         int new_channels = 0;
1516         struct hostapd_channel_data *channel;
1517         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1518         struct nlattr *nl_freq;
1519         int rem_freq, idx;
1520
1521         if (tb == NULL)
1522                 return NL_OK;
1523
1524         nla_for_each_nested(nl_freq, tb, rem_freq) {
1525                 nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1526                           nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1527                 if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1528                         continue;
1529                 new_channels++;
1530         }
1531
1532         channel = os_realloc_array(mode->channels,
1533                                    mode->num_channels + new_channels,
1534                                    sizeof(struct hostapd_channel_data));
1535         if (!channel)
1536                 return NL_STOP;
1537
1538         mode->channels = channel;
1539         mode->num_channels += new_channels;
1540
1541         idx = phy_info->last_chan_idx;
1542
1543         nla_for_each_nested(nl_freq, tb, rem_freq) {
1544                 nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1545                           nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1546                 if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1547                         continue;
1548                 phy_info_freq(mode, &mode->channels[idx], tb_freq);
1549                 idx++;
1550         }
1551         phy_info->last_chan_idx = idx;
1552
1553         return NL_OK;
1554 }
1555
1556
1557 static int phy_info_rates(struct hostapd_hw_modes *mode, struct nlattr *tb)
1558 {
1559         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1560                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1561                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] =
1562                 { .type = NLA_FLAG },
1563         };
1564         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1565         struct nlattr *nl_rate;
1566         int rem_rate, idx;
1567
1568         if (tb == NULL)
1569                 return NL_OK;
1570
1571         nla_for_each_nested(nl_rate, tb, rem_rate) {
1572                 nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1573                           nla_data(nl_rate), nla_len(nl_rate),
1574                           rate_policy);
1575                 if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1576                         continue;
1577                 mode->num_rates++;
1578         }
1579
1580         mode->rates = os_calloc(mode->num_rates, sizeof(int));
1581         if (!mode->rates)
1582                 return NL_STOP;
1583
1584         idx = 0;
1585
1586         nla_for_each_nested(nl_rate, tb, rem_rate) {
1587                 nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1588                           nla_data(nl_rate), nla_len(nl_rate),
1589                           rate_policy);
1590                 if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1591                         continue;
1592                 mode->rates[idx] = nla_get_u32(
1593                         tb_rate[NL80211_BITRATE_ATTR_RATE]);
1594                 idx++;
1595         }
1596
1597         return NL_OK;
1598 }
1599
1600
1601 static int phy_info_band(struct phy_info_arg *phy_info, struct nlattr *nl_band)
1602 {
1603         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
1604         struct hostapd_hw_modes *mode;
1605         int ret;
1606
1607         if (phy_info->last_mode != nl_band->nla_type) {
1608                 mode = os_realloc_array(phy_info->modes,
1609                                         *phy_info->num_modes + 1,
1610                                         sizeof(*mode));
1611                 if (!mode) {
1612                         phy_info->failed = 1;
1613                         return NL_STOP;
1614                 }
1615                 phy_info->modes = mode;
1616
1617                 mode = &phy_info->modes[*(phy_info->num_modes)];
1618                 os_memset(mode, 0, sizeof(*mode));
1619                 mode->mode = NUM_HOSTAPD_MODES;
1620                 mode->flags = HOSTAPD_MODE_FLAG_HT_INFO_KNOWN |
1621                         HOSTAPD_MODE_FLAG_VHT_INFO_KNOWN;
1622
1623                 /*
1624                  * Unsupported VHT MCS stream is defined as value 3, so the VHT
1625                  * MCS RX/TX map must be initialized with 0xffff to mark all 8
1626                  * possible streams as unsupported. This will be overridden if
1627                  * driver advertises VHT support.
1628                  */
1629                 mode->vht_mcs_set[0] = 0xff;
1630                 mode->vht_mcs_set[1] = 0xff;
1631                 mode->vht_mcs_set[4] = 0xff;
1632                 mode->vht_mcs_set[5] = 0xff;
1633
1634                 *(phy_info->num_modes) += 1;
1635                 phy_info->last_mode = nl_band->nla_type;
1636                 phy_info->last_chan_idx = 0;
1637         } else
1638                 mode = &phy_info->modes[*(phy_info->num_modes) - 1];
1639
1640         nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
1641                   nla_len(nl_band), NULL);
1642
1643         phy_info_ht_capa(mode, tb_band[NL80211_BAND_ATTR_HT_CAPA],
1644                          tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR],
1645                          tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY],
1646                          tb_band[NL80211_BAND_ATTR_HT_MCS_SET]);
1647         phy_info_vht_capa(mode, tb_band[NL80211_BAND_ATTR_VHT_CAPA],
1648                           tb_band[NL80211_BAND_ATTR_VHT_MCS_SET]);
1649         ret = phy_info_freqs(phy_info, mode, tb_band[NL80211_BAND_ATTR_FREQS]);
1650         if (ret == NL_OK)
1651                 ret = phy_info_rates(mode, tb_band[NL80211_BAND_ATTR_RATES]);
1652         if (ret != NL_OK) {
1653                 phy_info->failed = 1;
1654                 return ret;
1655         }
1656
1657         return NL_OK;
1658 }
1659
1660
1661 static int phy_info_handler(struct nl_msg *msg, void *arg)
1662 {
1663         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1664         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1665         struct phy_info_arg *phy_info = arg;
1666         struct nlattr *nl_band;
1667         int rem_band;
1668
1669         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1670                   genlmsg_attrlen(gnlh, 0), NULL);
1671
1672         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
1673                 return NL_SKIP;
1674
1675         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band)
1676         {
1677                 int res = phy_info_band(phy_info, nl_band);
1678                 if (res != NL_OK)
1679                         return res;
1680         }
1681
1682         return NL_SKIP;
1683 }
1684
1685
1686 static struct hostapd_hw_modes *
1687 wpa_driver_nl80211_postprocess_modes(struct hostapd_hw_modes *modes,
1688                                      u16 *num_modes)
1689 {
1690         u16 m;
1691         struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
1692         int i, mode11g_idx = -1;
1693
1694         /* heuristic to set up modes */
1695         for (m = 0; m < *num_modes; m++) {
1696                 if (!modes[m].num_channels)
1697                         continue;
1698                 if (modes[m].channels[0].freq < 4000) {
1699                         modes[m].mode = HOSTAPD_MODE_IEEE80211B;
1700                         for (i = 0; i < modes[m].num_rates; i++) {
1701                                 if (modes[m].rates[i] > 200) {
1702                                         modes[m].mode = HOSTAPD_MODE_IEEE80211G;
1703                                         break;
1704                                 }
1705                         }
1706                 } else if (modes[m].channels[0].freq > 50000)
1707                         modes[m].mode = HOSTAPD_MODE_IEEE80211AD;
1708                 else
1709                         modes[m].mode = HOSTAPD_MODE_IEEE80211A;
1710         }
1711
1712         /* If only 802.11g mode is included, use it to construct matching
1713          * 802.11b mode data. */
1714
1715         for (m = 0; m < *num_modes; m++) {
1716                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
1717                         return modes; /* 802.11b already included */
1718                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1719                         mode11g_idx = m;
1720         }
1721
1722         if (mode11g_idx < 0)
1723                 return modes; /* 2.4 GHz band not supported at all */
1724
1725         nmodes = os_realloc_array(modes, *num_modes + 1, sizeof(*nmodes));
1726         if (nmodes == NULL)
1727                 return modes; /* Could not add 802.11b mode */
1728
1729         mode = &nmodes[*num_modes];
1730         os_memset(mode, 0, sizeof(*mode));
1731         (*num_modes)++;
1732         modes = nmodes;
1733
1734         mode->mode = HOSTAPD_MODE_IEEE80211B;
1735
1736         mode11g = &modes[mode11g_idx];
1737         mode->num_channels = mode11g->num_channels;
1738         mode->channels = os_memdup(mode11g->channels,
1739                                    mode11g->num_channels *
1740                                    sizeof(struct hostapd_channel_data));
1741         if (mode->channels == NULL) {
1742                 (*num_modes)--;
1743                 return modes; /* Could not add 802.11b mode */
1744         }
1745
1746         mode->num_rates = 0;
1747         mode->rates = os_malloc(4 * sizeof(int));
1748         if (mode->rates == NULL) {
1749                 os_free(mode->channels);
1750                 (*num_modes)--;
1751                 return modes; /* Could not add 802.11b mode */
1752         }
1753
1754         for (i = 0; i < mode11g->num_rates; i++) {
1755                 if (mode11g->rates[i] != 10 && mode11g->rates[i] != 20 &&
1756                     mode11g->rates[i] != 55 && mode11g->rates[i] != 110)
1757                         continue;
1758                 mode->rates[mode->num_rates] = mode11g->rates[i];
1759                 mode->num_rates++;
1760                 if (mode->num_rates == 4)
1761                         break;
1762         }
1763
1764         if (mode->num_rates == 0) {
1765                 os_free(mode->channels);
1766                 os_free(mode->rates);
1767                 (*num_modes)--;
1768                 return modes; /* No 802.11b rates */
1769         }
1770
1771         wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
1772                    "information");
1773
1774         return modes;
1775 }
1776
1777
1778 static void nl80211_set_ht40_mode(struct hostapd_hw_modes *mode, int start,
1779                                   int end)
1780 {
1781         int c;
1782
1783         for (c = 0; c < mode->num_channels; c++) {
1784                 struct hostapd_channel_data *chan = &mode->channels[c];
1785                 if (chan->freq - 10 >= start && chan->freq + 10 <= end)
1786                         chan->flag |= HOSTAPD_CHAN_HT40;
1787         }
1788 }
1789
1790
1791 static void nl80211_set_ht40_mode_sec(struct hostapd_hw_modes *mode, int start,
1792                                       int end)
1793 {
1794         int c;
1795
1796         for (c = 0; c < mode->num_channels; c++) {
1797                 struct hostapd_channel_data *chan = &mode->channels[c];
1798                 if (!(chan->flag & HOSTAPD_CHAN_HT40))
1799                         continue;
1800                 if (chan->freq - 30 >= start && chan->freq - 10 <= end)
1801                         chan->flag |= HOSTAPD_CHAN_HT40MINUS;
1802                 if (chan->freq + 10 >= start && chan->freq + 30 <= end)
1803                         chan->flag |= HOSTAPD_CHAN_HT40PLUS;
1804         }
1805 }
1806
1807
1808 static void nl80211_reg_rule_max_eirp(u32 start, u32 end, u32 max_eirp,
1809                                       struct phy_info_arg *results)
1810 {
1811         u16 m;
1812
1813         for (m = 0; m < *results->num_modes; m++) {
1814                 int c;
1815                 struct hostapd_hw_modes *mode = &results->modes[m];
1816
1817                 for (c = 0; c < mode->num_channels; c++) {
1818                         struct hostapd_channel_data *chan = &mode->channels[c];
1819                         if ((u32) chan->freq - 10 >= start &&
1820                             (u32) chan->freq + 10 <= end)
1821                                 chan->max_tx_power = max_eirp;
1822                 }
1823         }
1824 }
1825
1826
1827 static void nl80211_reg_rule_ht40(u32 start, u32 end,
1828                                   struct phy_info_arg *results)
1829 {
1830         u16 m;
1831
1832         for (m = 0; m < *results->num_modes; m++) {
1833                 if (!(results->modes[m].ht_capab &
1834                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1835                         continue;
1836                 nl80211_set_ht40_mode(&results->modes[m], start, end);
1837         }
1838 }
1839
1840
1841 static void nl80211_reg_rule_sec(struct nlattr *tb[],
1842                                  struct phy_info_arg *results)
1843 {
1844         u32 start, end, max_bw;
1845         u16 m;
1846
1847         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1848             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1849             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1850                 return;
1851
1852         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1853         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1854         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1855
1856         if (max_bw < 20)
1857                 return;
1858
1859         for (m = 0; m < *results->num_modes; m++) {
1860                 if (!(results->modes[m].ht_capab &
1861                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1862                         continue;
1863                 nl80211_set_ht40_mode_sec(&results->modes[m], start, end);
1864         }
1865 }
1866
1867
1868 static void nl80211_set_vht_mode(struct hostapd_hw_modes *mode, int start,
1869                                  int end, int max_bw)
1870 {
1871         int c;
1872
1873         for (c = 0; c < mode->num_channels; c++) {
1874                 struct hostapd_channel_data *chan = &mode->channels[c];
1875                 if (chan->freq - 10 >= start && chan->freq + 70 <= end)
1876                         chan->flag |= HOSTAPD_CHAN_VHT_10_70;
1877
1878                 if (chan->freq - 30 >= start && chan->freq + 50 <= end)
1879                         chan->flag |= HOSTAPD_CHAN_VHT_30_50;
1880
1881                 if (chan->freq - 50 >= start && chan->freq + 30 <= end)
1882                         chan->flag |= HOSTAPD_CHAN_VHT_50_30;
1883
1884                 if (chan->freq - 70 >= start && chan->freq + 10 <= end)
1885                         chan->flag |= HOSTAPD_CHAN_VHT_70_10;
1886
1887                 if (max_bw >= 160) {
1888                         if (chan->freq - 10 >= start && chan->freq + 150 <= end)
1889                                 chan->flag |= HOSTAPD_CHAN_VHT_10_150;
1890
1891                         if (chan->freq - 30 >= start && chan->freq + 130 <= end)
1892                                 chan->flag |= HOSTAPD_CHAN_VHT_30_130;
1893
1894                         if (chan->freq - 50 >= start && chan->freq + 110 <= end)
1895                                 chan->flag |= HOSTAPD_CHAN_VHT_50_110;
1896
1897                         if (chan->freq - 70 >= start && chan->freq + 90 <= end)
1898                                 chan->flag |= HOSTAPD_CHAN_VHT_70_90;
1899
1900                         if (chan->freq - 90 >= start && chan->freq + 70 <= end)
1901                                 chan->flag |= HOSTAPD_CHAN_VHT_90_70;
1902
1903                         if (chan->freq - 110 >= start && chan->freq + 50 <= end)
1904                                 chan->flag |= HOSTAPD_CHAN_VHT_110_50;
1905
1906                         if (chan->freq - 130 >= start && chan->freq + 30 <= end)
1907                                 chan->flag |= HOSTAPD_CHAN_VHT_130_30;
1908
1909                         if (chan->freq - 150 >= start && chan->freq + 10 <= end)
1910                                 chan->flag |= HOSTAPD_CHAN_VHT_150_10;
1911                 }
1912         }
1913 }
1914
1915
1916 static void nl80211_reg_rule_vht(struct nlattr *tb[],
1917                                  struct phy_info_arg *results)
1918 {
1919         u32 start, end, max_bw;
1920         u16 m;
1921
1922         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1923             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1924             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1925                 return;
1926
1927         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1928         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1929         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1930
1931         if (max_bw < 80)
1932                 return;
1933
1934         for (m = 0; m < *results->num_modes; m++) {
1935                 if (!(results->modes[m].ht_capab &
1936                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1937                         continue;
1938                 /* TODO: use a real VHT support indication */
1939                 if (!results->modes[m].vht_capab)
1940                         continue;
1941
1942                 nl80211_set_vht_mode(&results->modes[m], start, end, max_bw);
1943         }
1944 }
1945
1946
1947 static void nl80211_set_dfs_domain(enum nl80211_dfs_regions region,
1948                                    u8 *dfs_domain)
1949 {
1950         if (region == NL80211_DFS_FCC)
1951                 *dfs_domain = HOSTAPD_DFS_REGION_FCC;
1952         else if (region == NL80211_DFS_ETSI)
1953                 *dfs_domain = HOSTAPD_DFS_REGION_ETSI;
1954         else if (region == NL80211_DFS_JP)
1955                 *dfs_domain = HOSTAPD_DFS_REGION_JP;
1956         else
1957                 *dfs_domain = 0;
1958 }
1959
1960
1961 static const char * dfs_domain_name(enum nl80211_dfs_regions region)
1962 {
1963         switch (region) {
1964         case NL80211_DFS_UNSET:
1965                 return "DFS-UNSET";
1966         case NL80211_DFS_FCC:
1967                 return "DFS-FCC";
1968         case NL80211_DFS_ETSI:
1969                 return "DFS-ETSI";
1970         case NL80211_DFS_JP:
1971                 return "DFS-JP";
1972         default:
1973                 return "DFS-invalid";
1974         }
1975 }
1976
1977
1978 static int nl80211_get_reg(struct nl_msg *msg, void *arg)
1979 {
1980         struct phy_info_arg *results = arg;
1981         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1982         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1983         struct nlattr *nl_rule;
1984         struct nlattr *tb_rule[NL80211_FREQUENCY_ATTR_MAX + 1];
1985         int rem_rule;
1986         static struct nla_policy reg_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1987                 [NL80211_ATTR_REG_RULE_FLAGS] = { .type = NLA_U32 },
1988                 [NL80211_ATTR_FREQ_RANGE_START] = { .type = NLA_U32 },
1989                 [NL80211_ATTR_FREQ_RANGE_END] = { .type = NLA_U32 },
1990                 [NL80211_ATTR_FREQ_RANGE_MAX_BW] = { .type = NLA_U32 },
1991                 [NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN] = { .type = NLA_U32 },
1992                 [NL80211_ATTR_POWER_RULE_MAX_EIRP] = { .type = NLA_U32 },
1993         };
1994
1995         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1996                   genlmsg_attrlen(gnlh, 0), NULL);
1997         if (!tb_msg[NL80211_ATTR_REG_ALPHA2] ||
1998             !tb_msg[NL80211_ATTR_REG_RULES]) {
1999                 wpa_printf(MSG_DEBUG, "nl80211: No regulatory information "
2000                            "available");
2001                 return NL_SKIP;
2002         }
2003
2004         if (tb_msg[NL80211_ATTR_DFS_REGION]) {
2005                 enum nl80211_dfs_regions dfs_domain;
2006                 dfs_domain = nla_get_u8(tb_msg[NL80211_ATTR_DFS_REGION]);
2007                 nl80211_set_dfs_domain(dfs_domain, &results->dfs_domain);
2008                 wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s (%s)",
2009                            (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]),
2010                            dfs_domain_name(dfs_domain));
2011         } else {
2012                 wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s",
2013                            (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]));
2014         }
2015
2016         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
2017         {
2018                 u32 start, end, max_eirp = 0, max_bw = 0, flags = 0;
2019                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
2020                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
2021                 if (tb_rule[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
2022                     tb_rule[NL80211_ATTR_FREQ_RANGE_END] == NULL)
2023                         continue;
2024                 start = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
2025                 end = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
2026                 if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
2027                         max_eirp = nla_get_u32(tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP]) / 100;
2028                 if (tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW])
2029                         max_bw = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
2030                 if (tb_rule[NL80211_ATTR_REG_RULE_FLAGS])
2031                         flags = nla_get_u32(tb_rule[NL80211_ATTR_REG_RULE_FLAGS]);
2032
2033                 wpa_printf(MSG_DEBUG, "nl80211: %u-%u @ %u MHz %u mBm%s%s%s%s%s%s%s%s",
2034                            start, end, max_bw, max_eirp,
2035                            flags & NL80211_RRF_NO_OFDM ? " (no OFDM)" : "",
2036                            flags & NL80211_RRF_NO_CCK ? " (no CCK)" : "",
2037                            flags & NL80211_RRF_NO_INDOOR ? " (no indoor)" : "",
2038                            flags & NL80211_RRF_NO_OUTDOOR ? " (no outdoor)" :
2039                            "",
2040                            flags & NL80211_RRF_DFS ? " (DFS)" : "",
2041                            flags & NL80211_RRF_PTP_ONLY ? " (PTP only)" : "",
2042                            flags & NL80211_RRF_PTMP_ONLY ? " (PTMP only)" : "",
2043                            flags & NL80211_RRF_NO_IR ? " (no IR)" : "");
2044                 if (max_bw >= 40)
2045                         nl80211_reg_rule_ht40(start, end, results);
2046                 if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
2047                         nl80211_reg_rule_max_eirp(start, end, max_eirp,
2048                                                   results);
2049         }
2050
2051         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
2052         {
2053                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
2054                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
2055                 nl80211_reg_rule_sec(tb_rule, results);
2056         }
2057
2058         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
2059         {
2060                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
2061                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
2062                 nl80211_reg_rule_vht(tb_rule, results);
2063         }
2064
2065         return NL_SKIP;
2066 }
2067
2068
2069 static int nl80211_set_regulatory_flags(struct wpa_driver_nl80211_data *drv,
2070                                         struct phy_info_arg *results)
2071 {
2072         struct nl_msg *msg;
2073
2074         msg = nlmsg_alloc();
2075         if (!msg)
2076                 return -ENOMEM;
2077
2078         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_REG);
2079         if (drv->capa.flags & WPA_DRIVER_FLAGS_SELF_MANAGED_REGULATORY) {
2080                 if (nla_put_u32(msg, NL80211_ATTR_WIPHY, drv->wiphy_idx)) {
2081                         nlmsg_free(msg);
2082                         return -1;
2083                 }
2084         }
2085
2086         return send_and_recv_msgs(drv, msg, nl80211_get_reg, results);
2087 }
2088
2089
2090 static const char * modestr(enum hostapd_hw_mode mode)
2091 {
2092         switch (mode) {
2093         case HOSTAPD_MODE_IEEE80211B:
2094                 return "802.11b";
2095         case HOSTAPD_MODE_IEEE80211G:
2096                 return "802.11g";
2097         case HOSTAPD_MODE_IEEE80211A:
2098                 return "802.11a";
2099         case HOSTAPD_MODE_IEEE80211AD:
2100                 return "802.11ad";
2101         default:
2102                 return "?";
2103         }
2104 }
2105
2106
2107 static void nl80211_dump_chan_list(struct hostapd_hw_modes *modes,
2108                                    u16 num_modes)
2109 {
2110         int i;
2111
2112         if (!modes)
2113                 return;
2114
2115         for (i = 0; i < num_modes; i++) {
2116                 struct hostapd_hw_modes *mode = &modes[i];
2117                 char str[200];
2118                 char *pos = str;
2119                 char *end = pos + sizeof(str);
2120                 int j, res;
2121
2122                 for (j = 0; j < mode->num_channels; j++) {
2123                         struct hostapd_channel_data *chan = &mode->channels[j];
2124
2125                         res = os_snprintf(pos, end - pos, " %d%s%s%s",
2126                                           chan->freq,
2127                                           (chan->flag & HOSTAPD_CHAN_DISABLED) ?
2128                                           "[DISABLED]" : "",
2129                                           (chan->flag & HOSTAPD_CHAN_NO_IR) ?
2130                                           "[NO_IR]" : "",
2131                                           (chan->flag & HOSTAPD_CHAN_RADAR) ?
2132                                           "[RADAR]" : "");
2133                         if (os_snprintf_error(end - pos, res))
2134                                 break;
2135                         pos += res;
2136                 }
2137
2138                 *pos = '\0';
2139                 wpa_printf(MSG_DEBUG, "nl80211: Mode IEEE %s:%s",
2140                            modestr(mode->mode), str);
2141         }
2142 }
2143
2144
2145 struct hostapd_hw_modes *
2146 nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags,
2147                             u8 *dfs_domain)
2148 {
2149         u32 feat;
2150         struct i802_bss *bss = priv;
2151         struct wpa_driver_nl80211_data *drv = bss->drv;
2152         int nl_flags = 0;
2153         struct nl_msg *msg;
2154         struct phy_info_arg result = {
2155                 .num_modes = num_modes,
2156                 .modes = NULL,
2157                 .last_mode = -1,
2158                 .failed = 0,
2159                 .dfs_domain = 0,
2160         };
2161
2162         *num_modes = 0;
2163         *flags = 0;
2164         *dfs_domain = 0;
2165
2166         feat = get_nl80211_protocol_features(drv);
2167         if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
2168                 nl_flags = NLM_F_DUMP;
2169         if (!(msg = nl80211_cmd_msg(bss, nl_flags, NL80211_CMD_GET_WIPHY)) ||
2170             nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
2171                 nlmsg_free(msg);
2172                 return NULL;
2173         }
2174
2175         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0) {
2176                 struct hostapd_hw_modes *modes;
2177
2178                 nl80211_set_regulatory_flags(drv, &result);
2179                 if (result.failed) {
2180                         int i;
2181
2182                         for (i = 0; result.modes && i < *num_modes; i++) {
2183                                 os_free(result.modes[i].channels);
2184                                 os_free(result.modes[i].rates);
2185                         }
2186                         os_free(result.modes);
2187                         *num_modes = 0;
2188                         return NULL;
2189                 }
2190
2191                 *dfs_domain = result.dfs_domain;
2192
2193                 modes = wpa_driver_nl80211_postprocess_modes(result.modes,
2194                                                              num_modes);
2195                 nl80211_dump_chan_list(modes, *num_modes);
2196                 return modes;
2197         }
2198
2199         return NULL;
2200 }