nl80211: Driver interaction for QCA vendor scan
[mech_eap.git] / 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_GET_PREFERRED_FREQ_LIST:
593                                         drv->get_pref_freq_list = 1;
594                                         break;
595                                 case QCA_NL80211_VENDOR_SUBCMD_SET_PROBABLE_OPER_CHANNEL:
596                                         drv->set_prob_oper_freq = 1;
597                                         break;
598                                 case QCA_NL80211_VENDOR_SUBCMD_DO_ACS:
599                                         drv->capa.flags |=
600                                                 WPA_DRIVER_FLAGS_ACS_OFFLOAD;
601                                         break;
602                                 case QCA_NL80211_VENDOR_SUBCMD_SETBAND:
603                                         drv->setband_vendor_cmd_avail = 1;
604                                         break;
605                                 case QCA_NL80211_VENDOR_SUBCMD_TRIGGER_SCAN:
606                                         drv->scan_vendor_cmd_avail = 1;
607                                         break;
608                                 }
609                         }
610
611                         wpa_printf(MSG_DEBUG, "nl80211: Supported vendor command: vendor_id=0x%x subcmd=%u",
612                                    vinfo->vendor_id, vinfo->subcmd);
613                 }
614         }
615
616         if (tb[NL80211_ATTR_VENDOR_EVENTS]) {
617                 struct nlattr *nl;
618                 int rem;
619
620                 nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_EVENTS], rem) {
621                         struct nl80211_vendor_cmd_info *vinfo;
622                         if (nla_len(nl) != sizeof(*vinfo)) {
623                                 wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
624                                 continue;
625                         }
626                         vinfo = nla_data(nl);
627                         wpa_printf(MSG_DEBUG, "nl80211: Supported vendor event: vendor_id=0x%x subcmd=%u",
628                                    vinfo->vendor_id, vinfo->subcmd);
629                 }
630         }
631
632         wiphy_info_wowlan_triggers(capa,
633                                    tb[NL80211_ATTR_WOWLAN_TRIGGERS_SUPPORTED]);
634
635         if (tb[NL80211_ATTR_MAX_AP_ASSOC_STA])
636                 capa->max_stations =
637                         nla_get_u32(tb[NL80211_ATTR_MAX_AP_ASSOC_STA]);
638
639         return NL_SKIP;
640 }
641
642
643 static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
644                                        struct wiphy_info_data *info)
645 {
646         u32 feat;
647         struct nl_msg *msg;
648         int flags = 0;
649
650         os_memset(info, 0, sizeof(*info));
651         info->capa = &drv->capa;
652         info->drv = drv;
653
654         feat = get_nl80211_protocol_features(drv);
655         if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
656                 flags = NLM_F_DUMP;
657         msg = nl80211_cmd_msg(drv->first_bss, flags, NL80211_CMD_GET_WIPHY);
658         if (!msg || nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
659                 nlmsg_free(msg);
660                 return -1;
661         }
662
663         if (send_and_recv_msgs(drv, msg, wiphy_info_handler, info))
664                 return -1;
665
666         if (info->auth_supported)
667                 drv->capa.flags |= WPA_DRIVER_FLAGS_SME;
668         else if (!info->connect_supported) {
669                 wpa_printf(MSG_INFO, "nl80211: Driver does not support "
670                            "authentication/association or connect commands");
671                 info->error = 1;
672         }
673
674         if (info->p2p_go_supported && info->p2p_client_supported)
675                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CAPABLE;
676         if (info->p2p_concurrent) {
677                 wpa_printf(MSG_DEBUG, "nl80211: Use separate P2P group "
678                            "interface (driver advertised support)");
679                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT;
680                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P;
681         }
682         if (info->num_multichan_concurrent > 1) {
683                 wpa_printf(MSG_DEBUG, "nl80211: Enable multi-channel "
684                            "concurrent (driver advertised support)");
685                 drv->capa.num_multichan_concurrent =
686                         info->num_multichan_concurrent;
687         }
688         if (drv->capa.flags & WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE)
689                 wpa_printf(MSG_DEBUG, "nl80211: use P2P_DEVICE support");
690
691         /* default to 5000 since early versions of mac80211 don't set it */
692         if (!drv->capa.max_remain_on_chan)
693                 drv->capa.max_remain_on_chan = 5000;
694
695         if (info->channel_switch_supported)
696                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP_CSA;
697         drv->capa.wmm_ac_supported = info->wmm_ac_supported;
698
699         drv->capa.mac_addr_rand_sched_scan_supported =
700                 info->mac_addr_rand_sched_scan_supported;
701         drv->capa.mac_addr_rand_scan_supported =
702                 info->mac_addr_rand_scan_supported;
703
704         return 0;
705 }
706
707
708 static int dfs_info_handler(struct nl_msg *msg, void *arg)
709 {
710         struct nlattr *tb[NL80211_ATTR_MAX + 1];
711         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
712         int *dfs_capability_ptr = arg;
713
714         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
715                   genlmsg_attrlen(gnlh, 0), NULL);
716
717         if (tb[NL80211_ATTR_VENDOR_DATA]) {
718                 struct nlattr *nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
719                 struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
720
721                 nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
722                           nla_data(nl_vend), nla_len(nl_vend), NULL);
723
724                 if (tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]) {
725                         u32 val;
726                         val = nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]);
727                         wpa_printf(MSG_DEBUG, "nl80211: DFS offload capability: %u",
728                                    val);
729                         *dfs_capability_ptr = val;
730                 }
731         }
732
733         return NL_SKIP;
734 }
735
736
737 static void qca_nl80211_check_dfs_capa(struct wpa_driver_nl80211_data *drv)
738 {
739         struct nl_msg *msg;
740         int dfs_capability = 0;
741         int ret;
742
743         if (!drv->dfs_vendor_cmd_avail)
744                 return;
745
746         if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
747             nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
748             nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
749                         QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY)) {
750                 nlmsg_free(msg);
751                 return;
752         }
753
754         ret = send_and_recv_msgs(drv, msg, dfs_info_handler, &dfs_capability);
755         if (!ret && dfs_capability)
756                 drv->capa.flags |= WPA_DRIVER_FLAGS_DFS_OFFLOAD;
757 }
758
759
760 struct features_info {
761         u8 *flags;
762         size_t flags_len;
763         struct wpa_driver_capa *capa;
764 };
765
766
767 static int features_info_handler(struct nl_msg *msg, void *arg)
768 {
769         struct nlattr *tb[NL80211_ATTR_MAX + 1];
770         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
771         struct features_info *info = arg;
772         struct nlattr *nl_vend, *attr;
773
774         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
775                   genlmsg_attrlen(gnlh, 0), NULL);
776
777         nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
778         if (nl_vend) {
779                 struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
780
781                 nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
782                           nla_data(nl_vend), nla_len(nl_vend), NULL);
783
784                 attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_FEATURE_FLAGS];
785                 if (attr) {
786                         info->flags = nla_data(attr);
787                         info->flags_len = nla_len(attr);
788                 }
789                 attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_CONCURRENCY_CAPA];
790                 if (attr)
791                         info->capa->conc_capab = nla_get_u32(attr);
792
793                 attr = tb_vendor[
794                         QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_2_4_BAND];
795                 if (attr)
796                         info->capa->max_conc_chan_2_4 = nla_get_u32(attr);
797
798                 attr = tb_vendor[
799                         QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_5_0_BAND];
800                 if (attr)
801                         info->capa->max_conc_chan_5_0 = nla_get_u32(attr);
802         }
803
804         return NL_SKIP;
805 }
806
807
808 static int check_feature(enum qca_wlan_vendor_features feature,
809                          struct features_info *info)
810 {
811         size_t idx = feature / 8;
812
813         return (idx < info->flags_len) &&
814                 (info->flags[idx] & BIT(feature % 8));
815 }
816
817
818 static void qca_nl80211_get_features(struct wpa_driver_nl80211_data *drv)
819 {
820         struct nl_msg *msg;
821         struct features_info info;
822         int ret;
823
824         if (!drv->get_features_vendor_cmd_avail)
825                 return;
826
827         if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
828             nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
829             nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
830                         QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES)) {
831                 nlmsg_free(msg);
832                 return;
833         }
834
835         os_memset(&info, 0, sizeof(info));
836         info.capa = &drv->capa;
837         ret = send_and_recv_msgs(drv, msg, features_info_handler, &info);
838         if (ret || !info.flags)
839                 return;
840
841         if (check_feature(QCA_WLAN_VENDOR_FEATURE_KEY_MGMT_OFFLOAD, &info))
842                 drv->capa.flags |= WPA_DRIVER_FLAGS_KEY_MGMT_OFFLOAD;
843
844         if (check_feature(QCA_WLAN_VENDOR_FEATURE_SUPPORT_HW_MODE_ANY, &info))
845                 drv->capa.flags |= WPA_DRIVER_FLAGS_SUPPORT_HW_MODE_ANY;
846 }
847
848
849 int wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
850 {
851         struct wiphy_info_data info;
852         if (wpa_driver_nl80211_get_info(drv, &info))
853                 return -1;
854
855         if (info.error)
856                 return -1;
857
858         drv->has_capability = 1;
859         drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
860                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
861                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
862                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK |
863                 WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B |
864                 WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B_192;
865         drv->capa.auth = WPA_DRIVER_AUTH_OPEN |
866                 WPA_DRIVER_AUTH_SHARED |
867                 WPA_DRIVER_AUTH_LEAP;
868
869         drv->capa.flags |= WPA_DRIVER_FLAGS_SANE_ERROR_CODES;
870         drv->capa.flags |= WPA_DRIVER_FLAGS_SET_KEYS_AFTER_ASSOC_DONE;
871         drv->capa.flags |= WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
872
873         /*
874          * As all cfg80211 drivers must support cases where the AP interface is
875          * removed without the knowledge of wpa_supplicant/hostapd, e.g., in
876          * case that the user space daemon has crashed, they must be able to
877          * cleanup all stations and key entries in the AP tear down flow. Thus,
878          * this flag can/should always be set for cfg80211 drivers.
879          */
880         drv->capa.flags |= WPA_DRIVER_FLAGS_AP_TEARDOWN_SUPPORT;
881
882         if (!info.device_ap_sme) {
883                 drv->capa.flags |= WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS;
884
885                 /*
886                  * No AP SME is currently assumed to also indicate no AP MLME
887                  * in the driver/firmware.
888                  */
889                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP_MLME;
890         }
891
892         drv->device_ap_sme = info.device_ap_sme;
893         drv->poll_command_supported = info.poll_command_supported;
894         drv->data_tx_status = info.data_tx_status;
895         drv->p2p_go_ctwindow_supported = info.p2p_go_ctwindow_supported;
896         if (info.set_qos_map_supported)
897                 drv->capa.flags |= WPA_DRIVER_FLAGS_QOS_MAPPING;
898         drv->have_low_prio_scan = info.have_low_prio_scan;
899
900         /*
901          * If poll command and tx status are supported, mac80211 is new enough
902          * to have everything we need to not need monitor interfaces.
903          */
904         drv->use_monitor = !info.poll_command_supported || !info.data_tx_status;
905
906         if (drv->device_ap_sme && drv->use_monitor) {
907                 /*
908                  * Non-mac80211 drivers may not support monitor interface.
909                  * Make sure we do not get stuck with incorrect capability here
910                  * by explicitly testing this.
911                  */
912                 if (!info.monitor_supported) {
913                         wpa_printf(MSG_DEBUG, "nl80211: Disable use_monitor "
914                                    "with device_ap_sme since no monitor mode "
915                                    "support detected");
916                         drv->use_monitor = 0;
917                 }
918         }
919
920         /*
921          * If we aren't going to use monitor interfaces, but the
922          * driver doesn't support data TX status, we won't get TX
923          * status for EAPOL frames.
924          */
925         if (!drv->use_monitor && !info.data_tx_status)
926                 drv->capa.flags &= ~WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
927
928         qca_nl80211_check_dfs_capa(drv);
929         qca_nl80211_get_features(drv);
930
931         return 0;
932 }
933
934
935 struct phy_info_arg {
936         u16 *num_modes;
937         struct hostapd_hw_modes *modes;
938         int last_mode, last_chan_idx;
939 };
940
941 static void phy_info_ht_capa(struct hostapd_hw_modes *mode, struct nlattr *capa,
942                              struct nlattr *ampdu_factor,
943                              struct nlattr *ampdu_density,
944                              struct nlattr *mcs_set)
945 {
946         if (capa)
947                 mode->ht_capab = nla_get_u16(capa);
948
949         if (ampdu_factor)
950                 mode->a_mpdu_params |= nla_get_u8(ampdu_factor) & 0x03;
951
952         if (ampdu_density)
953                 mode->a_mpdu_params |= nla_get_u8(ampdu_density) << 2;
954
955         if (mcs_set && nla_len(mcs_set) >= 16) {
956                 u8 *mcs;
957                 mcs = nla_data(mcs_set);
958                 os_memcpy(mode->mcs_set, mcs, 16);
959         }
960 }
961
962
963 static void phy_info_vht_capa(struct hostapd_hw_modes *mode,
964                               struct nlattr *capa,
965                               struct nlattr *mcs_set)
966 {
967         if (capa)
968                 mode->vht_capab = nla_get_u32(capa);
969
970         if (mcs_set && nla_len(mcs_set) >= 8) {
971                 u8 *mcs;
972                 mcs = nla_data(mcs_set);
973                 os_memcpy(mode->vht_mcs_set, mcs, 8);
974         }
975 }
976
977
978 static void phy_info_freq(struct hostapd_hw_modes *mode,
979                           struct hostapd_channel_data *chan,
980                           struct nlattr *tb_freq[])
981 {
982         u8 channel;
983         chan->freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
984         chan->flag = 0;
985         chan->dfs_cac_ms = 0;
986         if (ieee80211_freq_to_chan(chan->freq, &channel) != NUM_HOSTAPD_MODES)
987                 chan->chan = channel;
988
989         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
990                 chan->flag |= HOSTAPD_CHAN_DISABLED;
991         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IR])
992                 chan->flag |= HOSTAPD_CHAN_NO_IR;
993         if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
994                 chan->flag |= HOSTAPD_CHAN_RADAR;
995         if (tb_freq[NL80211_FREQUENCY_ATTR_INDOOR_ONLY])
996                 chan->flag |= HOSTAPD_CHAN_INDOOR_ONLY;
997         if (tb_freq[NL80211_FREQUENCY_ATTR_GO_CONCURRENT])
998                 chan->flag |= HOSTAPD_CHAN_GO_CONCURRENT;
999
1000         if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]) {
1001                 enum nl80211_dfs_state state =
1002                         nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]);
1003
1004                 switch (state) {
1005                 case NL80211_DFS_USABLE:
1006                         chan->flag |= HOSTAPD_CHAN_DFS_USABLE;
1007                         break;
1008                 case NL80211_DFS_AVAILABLE:
1009                         chan->flag |= HOSTAPD_CHAN_DFS_AVAILABLE;
1010                         break;
1011                 case NL80211_DFS_UNAVAILABLE:
1012                         chan->flag |= HOSTAPD_CHAN_DFS_UNAVAILABLE;
1013                         break;
1014                 }
1015         }
1016
1017         if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]) {
1018                 chan->dfs_cac_ms = nla_get_u32(
1019                         tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]);
1020         }
1021 }
1022
1023
1024 static int phy_info_freqs(struct phy_info_arg *phy_info,
1025                           struct hostapd_hw_modes *mode, struct nlattr *tb)
1026 {
1027         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1028                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1029                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1030                 [NL80211_FREQUENCY_ATTR_NO_IR] = { .type = NLA_FLAG },
1031                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1032                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
1033                 [NL80211_FREQUENCY_ATTR_DFS_STATE] = { .type = NLA_U32 },
1034         };
1035         int new_channels = 0;
1036         struct hostapd_channel_data *channel;
1037         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1038         struct nlattr *nl_freq;
1039         int rem_freq, idx;
1040
1041         if (tb == NULL)
1042                 return NL_OK;
1043
1044         nla_for_each_nested(nl_freq, tb, rem_freq) {
1045                 nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1046                           nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1047                 if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1048                         continue;
1049                 new_channels++;
1050         }
1051
1052         channel = os_realloc_array(mode->channels,
1053                                    mode->num_channels + new_channels,
1054                                    sizeof(struct hostapd_channel_data));
1055         if (!channel)
1056                 return NL_SKIP;
1057
1058         mode->channels = channel;
1059         mode->num_channels += new_channels;
1060
1061         idx = phy_info->last_chan_idx;
1062
1063         nla_for_each_nested(nl_freq, tb, rem_freq) {
1064                 nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1065                           nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1066                 if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1067                         continue;
1068                 phy_info_freq(mode, &mode->channels[idx], tb_freq);
1069                 idx++;
1070         }
1071         phy_info->last_chan_idx = idx;
1072
1073         return NL_OK;
1074 }
1075
1076
1077 static int phy_info_rates(struct hostapd_hw_modes *mode, struct nlattr *tb)
1078 {
1079         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1080                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1081                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] =
1082                 { .type = NLA_FLAG },
1083         };
1084         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1085         struct nlattr *nl_rate;
1086         int rem_rate, idx;
1087
1088         if (tb == NULL)
1089                 return NL_OK;
1090
1091         nla_for_each_nested(nl_rate, tb, rem_rate) {
1092                 nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1093                           nla_data(nl_rate), nla_len(nl_rate),
1094                           rate_policy);
1095                 if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1096                         continue;
1097                 mode->num_rates++;
1098         }
1099
1100         mode->rates = os_calloc(mode->num_rates, sizeof(int));
1101         if (!mode->rates)
1102                 return NL_SKIP;
1103
1104         idx = 0;
1105
1106         nla_for_each_nested(nl_rate, tb, rem_rate) {
1107                 nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1108                           nla_data(nl_rate), nla_len(nl_rate),
1109                           rate_policy);
1110                 if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1111                         continue;
1112                 mode->rates[idx] = nla_get_u32(
1113                         tb_rate[NL80211_BITRATE_ATTR_RATE]);
1114                 idx++;
1115         }
1116
1117         return NL_OK;
1118 }
1119
1120
1121 static int phy_info_band(struct phy_info_arg *phy_info, struct nlattr *nl_band)
1122 {
1123         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
1124         struct hostapd_hw_modes *mode;
1125         int ret;
1126
1127         if (phy_info->last_mode != nl_band->nla_type) {
1128                 mode = os_realloc_array(phy_info->modes,
1129                                         *phy_info->num_modes + 1,
1130                                         sizeof(*mode));
1131                 if (!mode)
1132                         return NL_SKIP;
1133                 phy_info->modes = mode;
1134
1135                 mode = &phy_info->modes[*(phy_info->num_modes)];
1136                 os_memset(mode, 0, sizeof(*mode));
1137                 mode->mode = NUM_HOSTAPD_MODES;
1138                 mode->flags = HOSTAPD_MODE_FLAG_HT_INFO_KNOWN |
1139                         HOSTAPD_MODE_FLAG_VHT_INFO_KNOWN;
1140
1141                 /*
1142                  * Unsupported VHT MCS stream is defined as value 3, so the VHT
1143                  * MCS RX/TX map must be initialized with 0xffff to mark all 8
1144                  * possible streams as unsupported. This will be overridden if
1145                  * driver advertises VHT support.
1146                  */
1147                 mode->vht_mcs_set[0] = 0xff;
1148                 mode->vht_mcs_set[1] = 0xff;
1149                 mode->vht_mcs_set[4] = 0xff;
1150                 mode->vht_mcs_set[5] = 0xff;
1151
1152                 *(phy_info->num_modes) += 1;
1153                 phy_info->last_mode = nl_band->nla_type;
1154                 phy_info->last_chan_idx = 0;
1155         } else
1156                 mode = &phy_info->modes[*(phy_info->num_modes) - 1];
1157
1158         nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
1159                   nla_len(nl_band), NULL);
1160
1161         phy_info_ht_capa(mode, tb_band[NL80211_BAND_ATTR_HT_CAPA],
1162                          tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR],
1163                          tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY],
1164                          tb_band[NL80211_BAND_ATTR_HT_MCS_SET]);
1165         phy_info_vht_capa(mode, tb_band[NL80211_BAND_ATTR_VHT_CAPA],
1166                           tb_band[NL80211_BAND_ATTR_VHT_MCS_SET]);
1167         ret = phy_info_freqs(phy_info, mode, tb_band[NL80211_BAND_ATTR_FREQS]);
1168         if (ret != NL_OK)
1169                 return ret;
1170         ret = phy_info_rates(mode, tb_band[NL80211_BAND_ATTR_RATES]);
1171         if (ret != NL_OK)
1172                 return ret;
1173
1174         return NL_OK;
1175 }
1176
1177
1178 static int phy_info_handler(struct nl_msg *msg, void *arg)
1179 {
1180         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1181         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1182         struct phy_info_arg *phy_info = arg;
1183         struct nlattr *nl_band;
1184         int rem_band;
1185
1186         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1187                   genlmsg_attrlen(gnlh, 0), NULL);
1188
1189         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
1190                 return NL_SKIP;
1191
1192         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band)
1193         {
1194                 int res = phy_info_band(phy_info, nl_band);
1195                 if (res != NL_OK)
1196                         return res;
1197         }
1198
1199         return NL_SKIP;
1200 }
1201
1202
1203 static struct hostapd_hw_modes *
1204 wpa_driver_nl80211_postprocess_modes(struct hostapd_hw_modes *modes,
1205                                      u16 *num_modes)
1206 {
1207         u16 m;
1208         struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
1209         int i, mode11g_idx = -1;
1210
1211         /* heuristic to set up modes */
1212         for (m = 0; m < *num_modes; m++) {
1213                 if (!modes[m].num_channels)
1214                         continue;
1215                 if (modes[m].channels[0].freq < 4000) {
1216                         modes[m].mode = HOSTAPD_MODE_IEEE80211B;
1217                         for (i = 0; i < modes[m].num_rates; i++) {
1218                                 if (modes[m].rates[i] > 200) {
1219                                         modes[m].mode = HOSTAPD_MODE_IEEE80211G;
1220                                         break;
1221                                 }
1222                         }
1223                 } else if (modes[m].channels[0].freq > 50000)
1224                         modes[m].mode = HOSTAPD_MODE_IEEE80211AD;
1225                 else
1226                         modes[m].mode = HOSTAPD_MODE_IEEE80211A;
1227         }
1228
1229         /* If only 802.11g mode is included, use it to construct matching
1230          * 802.11b mode data. */
1231
1232         for (m = 0; m < *num_modes; m++) {
1233                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
1234                         return modes; /* 802.11b already included */
1235                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1236                         mode11g_idx = m;
1237         }
1238
1239         if (mode11g_idx < 0)
1240                 return modes; /* 2.4 GHz band not supported at all */
1241
1242         nmodes = os_realloc_array(modes, *num_modes + 1, sizeof(*nmodes));
1243         if (nmodes == NULL)
1244                 return modes; /* Could not add 802.11b mode */
1245
1246         mode = &nmodes[*num_modes];
1247         os_memset(mode, 0, sizeof(*mode));
1248         (*num_modes)++;
1249         modes = nmodes;
1250
1251         mode->mode = HOSTAPD_MODE_IEEE80211B;
1252
1253         mode11g = &modes[mode11g_idx];
1254         mode->num_channels = mode11g->num_channels;
1255         mode->channels = os_malloc(mode11g->num_channels *
1256                                    sizeof(struct hostapd_channel_data));
1257         if (mode->channels == NULL) {
1258                 (*num_modes)--;
1259                 return modes; /* Could not add 802.11b mode */
1260         }
1261         os_memcpy(mode->channels, mode11g->channels,
1262                   mode11g->num_channels * sizeof(struct hostapd_channel_data));
1263
1264         mode->num_rates = 0;
1265         mode->rates = os_malloc(4 * sizeof(int));
1266         if (mode->rates == NULL) {
1267                 os_free(mode->channels);
1268                 (*num_modes)--;
1269                 return modes; /* Could not add 802.11b mode */
1270         }
1271
1272         for (i = 0; i < mode11g->num_rates; i++) {
1273                 if (mode11g->rates[i] != 10 && mode11g->rates[i] != 20 &&
1274                     mode11g->rates[i] != 55 && mode11g->rates[i] != 110)
1275                         continue;
1276                 mode->rates[mode->num_rates] = mode11g->rates[i];
1277                 mode->num_rates++;
1278                 if (mode->num_rates == 4)
1279                         break;
1280         }
1281
1282         if (mode->num_rates == 0) {
1283                 os_free(mode->channels);
1284                 os_free(mode->rates);
1285                 (*num_modes)--;
1286                 return modes; /* No 802.11b rates */
1287         }
1288
1289         wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
1290                    "information");
1291
1292         return modes;
1293 }
1294
1295
1296 static void nl80211_set_ht40_mode(struct hostapd_hw_modes *mode, int start,
1297                                   int end)
1298 {
1299         int c;
1300
1301         for (c = 0; c < mode->num_channels; c++) {
1302                 struct hostapd_channel_data *chan = &mode->channels[c];
1303                 if (chan->freq - 10 >= start && chan->freq + 10 <= end)
1304                         chan->flag |= HOSTAPD_CHAN_HT40;
1305         }
1306 }
1307
1308
1309 static void nl80211_set_ht40_mode_sec(struct hostapd_hw_modes *mode, int start,
1310                                       int end)
1311 {
1312         int c;
1313
1314         for (c = 0; c < mode->num_channels; c++) {
1315                 struct hostapd_channel_data *chan = &mode->channels[c];
1316                 if (!(chan->flag & HOSTAPD_CHAN_HT40))
1317                         continue;
1318                 if (chan->freq - 30 >= start && chan->freq - 10 <= end)
1319                         chan->flag |= HOSTAPD_CHAN_HT40MINUS;
1320                 if (chan->freq + 10 >= start && chan->freq + 30 <= end)
1321                         chan->flag |= HOSTAPD_CHAN_HT40PLUS;
1322         }
1323 }
1324
1325
1326 static void nl80211_reg_rule_max_eirp(u32 start, u32 end, u32 max_eirp,
1327                                       struct phy_info_arg *results)
1328 {
1329         u16 m;
1330
1331         for (m = 0; m < *results->num_modes; m++) {
1332                 int c;
1333                 struct hostapd_hw_modes *mode = &results->modes[m];
1334
1335                 for (c = 0; c < mode->num_channels; c++) {
1336                         struct hostapd_channel_data *chan = &mode->channels[c];
1337                         if ((u32) chan->freq - 10 >= start &&
1338                             (u32) chan->freq + 10 <= end)
1339                                 chan->max_tx_power = max_eirp;
1340                 }
1341         }
1342 }
1343
1344
1345 static void nl80211_reg_rule_ht40(u32 start, u32 end,
1346                                   struct phy_info_arg *results)
1347 {
1348         u16 m;
1349
1350         for (m = 0; m < *results->num_modes; m++) {
1351                 if (!(results->modes[m].ht_capab &
1352                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1353                         continue;
1354                 nl80211_set_ht40_mode(&results->modes[m], start, end);
1355         }
1356 }
1357
1358
1359 static void nl80211_reg_rule_sec(struct nlattr *tb[],
1360                                  struct phy_info_arg *results)
1361 {
1362         u32 start, end, max_bw;
1363         u16 m;
1364
1365         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1366             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1367             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1368                 return;
1369
1370         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1371         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1372         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1373
1374         if (max_bw < 20)
1375                 return;
1376
1377         for (m = 0; m < *results->num_modes; m++) {
1378                 if (!(results->modes[m].ht_capab &
1379                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1380                         continue;
1381                 nl80211_set_ht40_mode_sec(&results->modes[m], start, end);
1382         }
1383 }
1384
1385
1386 static void nl80211_set_vht_mode(struct hostapd_hw_modes *mode, int start,
1387                                  int end)
1388 {
1389         int c;
1390
1391         for (c = 0; c < mode->num_channels; c++) {
1392                 struct hostapd_channel_data *chan = &mode->channels[c];
1393                 if (chan->freq - 10 >= start && chan->freq + 70 <= end)
1394                         chan->flag |= HOSTAPD_CHAN_VHT_10_70;
1395
1396                 if (chan->freq - 30 >= start && chan->freq + 50 <= end)
1397                         chan->flag |= HOSTAPD_CHAN_VHT_30_50;
1398
1399                 if (chan->freq - 50 >= start && chan->freq + 30 <= end)
1400                         chan->flag |= HOSTAPD_CHAN_VHT_50_30;
1401
1402                 if (chan->freq - 70 >= start && chan->freq + 10 <= end)
1403                         chan->flag |= HOSTAPD_CHAN_VHT_70_10;
1404         }
1405 }
1406
1407
1408 static void nl80211_reg_rule_vht(struct nlattr *tb[],
1409                                  struct phy_info_arg *results)
1410 {
1411         u32 start, end, max_bw;
1412         u16 m;
1413
1414         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1415             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1416             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1417                 return;
1418
1419         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1420         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1421         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1422
1423         if (max_bw < 80)
1424                 return;
1425
1426         for (m = 0; m < *results->num_modes; m++) {
1427                 if (!(results->modes[m].ht_capab &
1428                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1429                         continue;
1430                 /* TODO: use a real VHT support indication */
1431                 if (!results->modes[m].vht_capab)
1432                         continue;
1433
1434                 nl80211_set_vht_mode(&results->modes[m], start, end);
1435         }
1436 }
1437
1438
1439 static const char * dfs_domain_name(enum nl80211_dfs_regions region)
1440 {
1441         switch (region) {
1442         case NL80211_DFS_UNSET:
1443                 return "DFS-UNSET";
1444         case NL80211_DFS_FCC:
1445                 return "DFS-FCC";
1446         case NL80211_DFS_ETSI:
1447                 return "DFS-ETSI";
1448         case NL80211_DFS_JP:
1449                 return "DFS-JP";
1450         default:
1451                 return "DFS-invalid";
1452         }
1453 }
1454
1455
1456 static int nl80211_get_reg(struct nl_msg *msg, void *arg)
1457 {
1458         struct phy_info_arg *results = arg;
1459         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1460         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1461         struct nlattr *nl_rule;
1462         struct nlattr *tb_rule[NL80211_FREQUENCY_ATTR_MAX + 1];
1463         int rem_rule;
1464         static struct nla_policy reg_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1465                 [NL80211_ATTR_REG_RULE_FLAGS] = { .type = NLA_U32 },
1466                 [NL80211_ATTR_FREQ_RANGE_START] = { .type = NLA_U32 },
1467                 [NL80211_ATTR_FREQ_RANGE_END] = { .type = NLA_U32 },
1468                 [NL80211_ATTR_FREQ_RANGE_MAX_BW] = { .type = NLA_U32 },
1469                 [NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN] = { .type = NLA_U32 },
1470                 [NL80211_ATTR_POWER_RULE_MAX_EIRP] = { .type = NLA_U32 },
1471         };
1472
1473         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1474                   genlmsg_attrlen(gnlh, 0), NULL);
1475         if (!tb_msg[NL80211_ATTR_REG_ALPHA2] ||
1476             !tb_msg[NL80211_ATTR_REG_RULES]) {
1477                 wpa_printf(MSG_DEBUG, "nl80211: No regulatory information "
1478                            "available");
1479                 return NL_SKIP;
1480         }
1481
1482         if (tb_msg[NL80211_ATTR_DFS_REGION]) {
1483                 enum nl80211_dfs_regions dfs_domain;
1484                 dfs_domain = nla_get_u8(tb_msg[NL80211_ATTR_DFS_REGION]);
1485                 wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s (%s)",
1486                            (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]),
1487                            dfs_domain_name(dfs_domain));
1488         } else {
1489                 wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s",
1490                            (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]));
1491         }
1492
1493         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1494         {
1495                 u32 start, end, max_eirp = 0, max_bw = 0, flags = 0;
1496                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1497                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1498                 if (tb_rule[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1499                     tb_rule[NL80211_ATTR_FREQ_RANGE_END] == NULL)
1500                         continue;
1501                 start = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1502                 end = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1503                 if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
1504                         max_eirp = nla_get_u32(tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP]) / 100;
1505                 if (tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW])
1506                         max_bw = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1507                 if (tb_rule[NL80211_ATTR_REG_RULE_FLAGS])
1508                         flags = nla_get_u32(tb_rule[NL80211_ATTR_REG_RULE_FLAGS]);
1509
1510                 wpa_printf(MSG_DEBUG, "nl80211: %u-%u @ %u MHz %u mBm%s%s%s%s%s%s%s%s",
1511                            start, end, max_bw, max_eirp,
1512                            flags & NL80211_RRF_NO_OFDM ? " (no OFDM)" : "",
1513                            flags & NL80211_RRF_NO_CCK ? " (no CCK)" : "",
1514                            flags & NL80211_RRF_NO_INDOOR ? " (no indoor)" : "",
1515                            flags & NL80211_RRF_NO_OUTDOOR ? " (no outdoor)" :
1516                            "",
1517                            flags & NL80211_RRF_DFS ? " (DFS)" : "",
1518                            flags & NL80211_RRF_PTP_ONLY ? " (PTP only)" : "",
1519                            flags & NL80211_RRF_PTMP_ONLY ? " (PTMP only)" : "",
1520                            flags & NL80211_RRF_NO_IR ? " (no IR)" : "");
1521                 if (max_bw >= 40)
1522                         nl80211_reg_rule_ht40(start, end, results);
1523                 if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
1524                         nl80211_reg_rule_max_eirp(start, end, max_eirp,
1525                                                   results);
1526         }
1527
1528         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1529         {
1530                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1531                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1532                 nl80211_reg_rule_sec(tb_rule, results);
1533         }
1534
1535         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1536         {
1537                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1538                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1539                 nl80211_reg_rule_vht(tb_rule, results);
1540         }
1541
1542         return NL_SKIP;
1543 }
1544
1545
1546 static int nl80211_set_regulatory_flags(struct wpa_driver_nl80211_data *drv,
1547                                         struct phy_info_arg *results)
1548 {
1549         struct nl_msg *msg;
1550
1551         msg = nlmsg_alloc();
1552         if (!msg)
1553                 return -ENOMEM;
1554
1555         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_REG);
1556         return send_and_recv_msgs(drv, msg, nl80211_get_reg, results);
1557 }
1558
1559
1560 struct hostapd_hw_modes *
1561 nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
1562 {
1563         u32 feat;
1564         struct i802_bss *bss = priv;
1565         struct wpa_driver_nl80211_data *drv = bss->drv;
1566         int nl_flags = 0;
1567         struct nl_msg *msg;
1568         struct phy_info_arg result = {
1569                 .num_modes = num_modes,
1570                 .modes = NULL,
1571                 .last_mode = -1,
1572         };
1573
1574         *num_modes = 0;
1575         *flags = 0;
1576
1577         feat = get_nl80211_protocol_features(drv);
1578         if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
1579                 nl_flags = NLM_F_DUMP;
1580         if (!(msg = nl80211_cmd_msg(bss, nl_flags, NL80211_CMD_GET_WIPHY)) ||
1581             nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
1582                 nlmsg_free(msg);
1583                 return NULL;
1584         }
1585
1586         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0) {
1587                 nl80211_set_regulatory_flags(drv, &result);
1588                 return wpa_driver_nl80211_postprocess_modes(result.modes,
1589                                                             num_modes);
1590         }
1591
1592         return NULL;
1593 }