src/drivers/driver_nl80211_capa.c

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