drivers/net/wireless/ath/ath11k/wmi.c

   1 // SPDX-License-Identifier: BSD-3-Clause-Clear
   2 /*
   3  * Copyright (c) 2018-2019 The Linux Foundation. All rights reserved.
   4  * Copyright (c) 2021-2023 Qualcomm Innovation Center, Inc. All rights reserved.
   5  */
   6 #include <linux/skbuff.h>
   7 #include <linux/ctype.h>
   8 #include <net/mac80211.h>
   9 #include <net/cfg80211.h>
  10 #include <linux/completion.h>
  11 #include <linux/if_ether.h>
  12 #include <linux/types.h>
  13 #include <linux/pci.h>
  14 #include <linux/uuid.h>
  15 #include <linux/time.h>
  16 #include <linux/of.h>
  17 #include "core.h"
  18 #include "debug.h"
  19 #include "mac.h"
  20 #include "hw.h"
  21 #include "peer.h"
  22 #include "testmode.h"
  23
  24 struct wmi_tlv_policy {
  25         size_t min_len;
  26 };
  27
  28 struct wmi_tlv_svc_ready_parse {
  29         bool wmi_svc_bitmap_done;
  30 };
  31
  32 struct wmi_tlv_dma_ring_caps_parse {
  33         struct wmi_dma_ring_capabilities *dma_ring_caps;
  34         u32 n_dma_ring_caps;
  35 };
  36
  37 struct wmi_tlv_svc_rdy_ext_parse {
  38         struct ath11k_service_ext_param param;
  39         struct wmi_soc_mac_phy_hw_mode_caps *hw_caps;
  40         struct wmi_hw_mode_capabilities *hw_mode_caps;
  41         u32 n_hw_mode_caps;
  42         u32 tot_phy_id;
  43         struct wmi_hw_mode_capabilities pref_hw_mode_caps;
  44         struct wmi_mac_phy_capabilities *mac_phy_caps;
  45         u32 n_mac_phy_caps;
  46         struct wmi_soc_hal_reg_capabilities *soc_hal_reg_caps;
  47         struct wmi_hal_reg_capabilities_ext *ext_hal_reg_caps;
  48         u32 n_ext_hal_reg_caps;
  49         struct wmi_tlv_dma_ring_caps_parse dma_caps_parse;
  50         bool hw_mode_done;
  51         bool mac_phy_done;
  52         bool ext_hal_reg_done;
  53         bool mac_phy_chainmask_combo_done;
  54         bool mac_phy_chainmask_cap_done;
  55         bool oem_dma_ring_cap_done;
  56         bool dma_ring_cap_done;
  57 };
  58
  59 struct wmi_tlv_svc_rdy_ext2_parse {
  60         struct wmi_tlv_dma_ring_caps_parse dma_caps_parse;
  61         bool dma_ring_cap_done;
  62 };
  63
  64 struct wmi_tlv_rdy_parse {
  65         u32 num_extra_mac_addr;
  66 };
  67
  68 struct wmi_tlv_dma_buf_release_parse {
  69         struct ath11k_wmi_dma_buf_release_fixed_param fixed;
  70         struct wmi_dma_buf_release_entry *buf_entry;
  71         struct wmi_dma_buf_release_meta_data *meta_data;
  72         u32 num_buf_entry;
  73         u32 num_meta;
  74         bool buf_entry_done;
  75         bool meta_data_done;
  76 };
  77
  78 struct wmi_tlv_fw_stats_parse {
  79         const struct wmi_stats_event *ev;
  80         const struct wmi_per_chain_rssi_stats *rssi;
  81         struct ath11k_fw_stats *stats;
  82         int rssi_num;
  83         bool chain_rssi_done;
  84 };
  85
  86 struct wmi_tlv_mgmt_rx_parse {
  87         const struct wmi_mgmt_rx_hdr *fixed;
  88         const u8 *frame_buf;
  89         bool frame_buf_done;
  90 };
  91
  92 static const struct wmi_tlv_policy wmi_tlv_policies[] = {
  93         [WMI_TAG_ARRAY_BYTE]
  94                 = { .min_len = 0 },
  95         [WMI_TAG_ARRAY_UINT32]
  96                 = { .min_len = 0 },
  97         [WMI_TAG_SERVICE_READY_EVENT]
  98                 = { .min_len = sizeof(struct wmi_service_ready_event) },
  99         [WMI_TAG_SERVICE_READY_EXT_EVENT]
 100                 = { .min_len =  sizeof(struct wmi_service_ready_ext_event) },
 101         [WMI_TAG_SOC_MAC_PHY_HW_MODE_CAPS]
 102                 = { .min_len = sizeof(struct wmi_soc_mac_phy_hw_mode_caps) },
 103         [WMI_TAG_SOC_HAL_REG_CAPABILITIES]
 104                 = { .min_len = sizeof(struct wmi_soc_hal_reg_capabilities) },
 105         [WMI_TAG_VDEV_START_RESPONSE_EVENT]
 106                 = { .min_len = sizeof(struct wmi_vdev_start_resp_event) },
 107         [WMI_TAG_PEER_DELETE_RESP_EVENT]
 108                 = { .min_len = sizeof(struct wmi_peer_delete_resp_event) },
 109         [WMI_TAG_OFFLOAD_BCN_TX_STATUS_EVENT]
 110                 = { .min_len = sizeof(struct wmi_bcn_tx_status_event) },
 111         [WMI_TAG_VDEV_STOPPED_EVENT]
 112                 = { .min_len = sizeof(struct wmi_vdev_stopped_event) },
 113         [WMI_TAG_REG_CHAN_LIST_CC_EVENT]
 114                 = { .min_len = sizeof(struct wmi_reg_chan_list_cc_event) },
 115         [WMI_TAG_REG_CHAN_LIST_CC_EXT_EVENT]
 116                 = { .min_len = sizeof(struct wmi_reg_chan_list_cc_ext_event) },
 117         [WMI_TAG_MGMT_RX_HDR]
 118                 = { .min_len = sizeof(struct wmi_mgmt_rx_hdr) },
 119         [WMI_TAG_MGMT_TX_COMPL_EVENT]
 120                 = { .min_len = sizeof(struct wmi_mgmt_tx_compl_event) },
 121         [WMI_TAG_SCAN_EVENT]
 122                 = { .min_len = sizeof(struct wmi_scan_event) },
 123         [WMI_TAG_PEER_STA_KICKOUT_EVENT]
 124                 = { .min_len = sizeof(struct wmi_peer_sta_kickout_event) },
 125         [WMI_TAG_ROAM_EVENT]
 126                 = { .min_len = sizeof(struct wmi_roam_event) },
 127         [WMI_TAG_CHAN_INFO_EVENT]
 128                 = { .min_len = sizeof(struct wmi_chan_info_event) },
 129         [WMI_TAG_PDEV_BSS_CHAN_INFO_EVENT]
 130                 = { .min_len = sizeof(struct wmi_pdev_bss_chan_info_event) },
 131         [WMI_TAG_VDEV_INSTALL_KEY_COMPLETE_EVENT]
 132                 = { .min_len = sizeof(struct wmi_vdev_install_key_compl_event) },
 133         [WMI_TAG_READY_EVENT] = {
 134                 .min_len = sizeof(struct wmi_ready_event_min) },
 135         [WMI_TAG_SERVICE_AVAILABLE_EVENT]
 136                 = {.min_len = sizeof(struct wmi_service_available_event) },
 137         [WMI_TAG_PEER_ASSOC_CONF_EVENT]
 138                 = { .min_len = sizeof(struct wmi_peer_assoc_conf_event) },
 139         [WMI_TAG_STATS_EVENT]
 140                 = { .min_len = sizeof(struct wmi_stats_event) },
 141         [WMI_TAG_PDEV_CTL_FAILSAFE_CHECK_EVENT]
 142                 = { .min_len = sizeof(struct wmi_pdev_ctl_failsafe_chk_event) },
 143         [WMI_TAG_HOST_SWFDA_EVENT] = {
 144                 .min_len = sizeof(struct wmi_fils_discovery_event) },
 145         [WMI_TAG_OFFLOAD_PRB_RSP_TX_STATUS_EVENT] = {
 146                 .min_len = sizeof(struct wmi_probe_resp_tx_status_event) },
 147         [WMI_TAG_VDEV_DELETE_RESP_EVENT] = {
 148                 .min_len = sizeof(struct wmi_vdev_delete_resp_event) },
 149         [WMI_TAG_OBSS_COLOR_COLLISION_EVT] = {
 150                 .min_len = sizeof(struct wmi_obss_color_collision_event) },
 151         [WMI_TAG_11D_NEW_COUNTRY_EVENT] = {
 152                 .min_len = sizeof(struct wmi_11d_new_cc_ev) },
 153         [WMI_TAG_PER_CHAIN_RSSI_STATS] = {
 154                 .min_len = sizeof(struct wmi_per_chain_rssi_stats) },
 155         [WMI_TAG_TWT_ADD_DIALOG_COMPLETE_EVENT] = {
 156                 .min_len = sizeof(struct wmi_twt_add_dialog_event) },
 157 };
 158
 159 #define PRIMAP(_hw_mode_) \
 160         [_hw_mode_] = _hw_mode_##_PRI
 161
 162 static const int ath11k_hw_mode_pri_map[] = {
 163         PRIMAP(WMI_HOST_HW_MODE_SINGLE),
 164         PRIMAP(WMI_HOST_HW_MODE_DBS),
 165         PRIMAP(WMI_HOST_HW_MODE_SBS_PASSIVE),
 166         PRIMAP(WMI_HOST_HW_MODE_SBS),
 167         PRIMAP(WMI_HOST_HW_MODE_DBS_SBS),
 168         PRIMAP(WMI_HOST_HW_MODE_DBS_OR_SBS),
 169         /* keep last */
 170         PRIMAP(WMI_HOST_HW_MODE_MAX),
 171 };
 172
 173 static int
 174 ath11k_wmi_tlv_iter(struct ath11k_base *ab, const void *ptr, size_t len,
 175                     int (*iter)(struct ath11k_base *ab, u16 tag, u16 len,
 176                                 const void *ptr, void *data),
 177                     void *data)
 178 {
 179         const void *begin = ptr;
 180         const struct wmi_tlv *tlv;
 181         u16 tlv_tag, tlv_len;
 182         int ret;
 183
 184         while (len > 0) {
 185                 if (len < sizeof(*tlv)) {
 186                         ath11k_err(ab, "wmi tlv parse failure at byte %zd (%zu bytes left, %zu expected)\n",
 187                                    ptr - begin, len, sizeof(*tlv));
 188                         return -EINVAL;
 189                 }
 190
 191                 tlv = ptr;
 192                 tlv_tag = FIELD_GET(WMI_TLV_TAG, tlv->header);
 193                 tlv_len = FIELD_GET(WMI_TLV_LEN, tlv->header);
 194                 ptr += sizeof(*tlv);
 195                 len -= sizeof(*tlv);
 196
 197                 if (tlv_len > len) {
 198                         ath11k_err(ab, "wmi tlv parse failure of tag %u at byte %zd (%zu bytes left, %u expected)\n",
 199                                    tlv_tag, ptr - begin, len, tlv_len);
 200                         return -EINVAL;
 201                 }
 202
 203                 if (tlv_tag < ARRAY_SIZE(wmi_tlv_policies) &&
 204                     wmi_tlv_policies[tlv_tag].min_len &&
 205                     wmi_tlv_policies[tlv_tag].min_len > tlv_len) {
 206                         ath11k_err(ab, "wmi tlv parse failure of tag %u at byte %zd (%u bytes is less than min length %zu)\n",
 207                                    tlv_tag, ptr - begin, tlv_len,
 208                                    wmi_tlv_policies[tlv_tag].min_len);
 209                         return -EINVAL;
 210                 }
 211
 212                 ret = iter(ab, tlv_tag, tlv_len, ptr, data);
 213                 if (ret)
 214                         return ret;
 215
 216                 ptr += tlv_len;
 217                 len -= tlv_len;
 218         }
 219
 220         return 0;
 221 }
 222
 223 static int ath11k_wmi_tlv_iter_parse(struct ath11k_base *ab, u16 tag, u16 len,
 224                                      const void *ptr, void *data)
 225 {
 226         const void **tb = data;
 227
 228         if (tag < WMI_TAG_MAX)
 229                 tb[tag] = ptr;
 230
 231         return 0;
 232 }
 233
 234 static int ath11k_wmi_tlv_parse(struct ath11k_base *ar, const void **tb,
 235                                 const void *ptr, size_t len)
 236 {
 237         return ath11k_wmi_tlv_iter(ar, ptr, len, ath11k_wmi_tlv_iter_parse,
 238                                    (void *)tb);
 239 }
 240
 241 const void **ath11k_wmi_tlv_parse_alloc(struct ath11k_base *ab, const void *ptr,
 242                                         size_t len, gfp_t gfp)
 243 {
 244         const void **tb;
 245         int ret;
 246
 247         tb = kcalloc(WMI_TAG_MAX, sizeof(*tb), gfp);
 248         if (!tb)
 249                 return ERR_PTR(-ENOMEM);
 250
 251         ret = ath11k_wmi_tlv_parse(ab, tb, ptr, len);
 252         if (ret) {
 253                 kfree(tb);
 254                 return ERR_PTR(ret);
 255         }
 256
 257         return tb;
 258 }
 259
 260 static int ath11k_wmi_cmd_send_nowait(struct ath11k_pdev_wmi *wmi, struct sk_buff *skb,
 261                                       u32 cmd_id)
 262 {
 263         struct ath11k_skb_cb *skb_cb = ATH11K_SKB_CB(skb);
 264         struct ath11k_base *ab = wmi->wmi_ab->ab;
 265         struct wmi_cmd_hdr *cmd_hdr;
 266         int ret;
 267         u32 cmd = 0;
 268
 269         if (skb_push(skb, sizeof(struct wmi_cmd_hdr)) == NULL)
 270                 return -ENOMEM;
 271
 272         cmd |= FIELD_PREP(WMI_CMD_HDR_CMD_ID, cmd_id);
 273
 274         cmd_hdr = (struct wmi_cmd_hdr *)skb->data;
 275         cmd_hdr->cmd_id = cmd;
 276
 277         trace_ath11k_wmi_cmd(ab, cmd_id, skb->data, skb->len);
 278
 279         memset(skb_cb, 0, sizeof(*skb_cb));
 280         ret = ath11k_htc_send(&ab->htc, wmi->eid, skb);
 281
 282         if (ret)
 283                 goto err_pull;
 284
 285         return 0;
 286
 287 err_pull:
 288         skb_pull(skb, sizeof(struct wmi_cmd_hdr));
 289         return ret;
 290 }
 291
 292 int ath11k_wmi_cmd_send(struct ath11k_pdev_wmi *wmi, struct sk_buff *skb,
 293                         u32 cmd_id)
 294 {
 295         struct ath11k_wmi_base *wmi_ab = wmi->wmi_ab;
 296         int ret = -EOPNOTSUPP;
 297         struct ath11k_base *ab = wmi_ab->ab;
 298
 299         might_sleep();
 300
 301         if (ab->hw_params.credit_flow) {
 302                 wait_event_timeout(wmi_ab->tx_credits_wq, ({
 303                         ret = ath11k_wmi_cmd_send_nowait(wmi, skb, cmd_id);
 304
 305                         if (ret && test_bit(ATH11K_FLAG_CRASH_FLUSH,
 306                                             &wmi_ab->ab->dev_flags))
 307                                 ret = -ESHUTDOWN;
 308
 309                         (ret != -EAGAIN);
 310                         }), WMI_SEND_TIMEOUT_HZ);
 311         } else {
 312                 wait_event_timeout(wmi->tx_ce_desc_wq, ({
 313                         ret = ath11k_wmi_cmd_send_nowait(wmi, skb, cmd_id);
 314
 315                         if (ret && test_bit(ATH11K_FLAG_CRASH_FLUSH,
 316                                             &wmi_ab->ab->dev_flags))
 317                                 ret = -ESHUTDOWN;
 318
 319                         (ret != -ENOBUFS);
 320                         }), WMI_SEND_TIMEOUT_HZ);
 321         }
 322
 323         if (ret == -EAGAIN)
 324                 ath11k_warn(wmi_ab->ab, "wmi command %d timeout\n", cmd_id);
 325
 326         if (ret == -ENOBUFS)
 327                 ath11k_warn(wmi_ab->ab, "ce desc not available for wmi command %d\n",
 328                             cmd_id);
 329
 330         return ret;
 331 }
 332
 333 static int ath11k_pull_svc_ready_ext(struct ath11k_pdev_wmi *wmi_handle,
 334                                      const void *ptr,
 335                                      struct ath11k_service_ext_param *param)
 336 {
 337         const struct wmi_service_ready_ext_event *ev = ptr;
 338
 339         if (!ev)
 340                 return -EINVAL;
 341
 342         /* Move this to host based bitmap */
 343         param->default_conc_scan_config_bits = ev->default_conc_scan_config_bits;
 344         param->default_fw_config_bits = ev->default_fw_config_bits;
 345         param->he_cap_info = ev->he_cap_info;
 346         param->mpdu_density = ev->mpdu_density;
 347         param->max_bssid_rx_filters = ev->max_bssid_rx_filters;
 348         memcpy(&param->ppet, &ev->ppet, sizeof(param->ppet));
 349
 350         return 0;
 351 }
 352
 353 static int
 354 ath11k_pull_mac_phy_cap_svc_ready_ext(struct ath11k_pdev_wmi *wmi_handle,
 355                                       struct wmi_soc_mac_phy_hw_mode_caps *hw_caps,
 356                                       struct wmi_hw_mode_capabilities *wmi_hw_mode_caps,
 357                                       struct wmi_soc_hal_reg_capabilities *hal_reg_caps,
 358                                       struct wmi_mac_phy_capabilities *wmi_mac_phy_caps,
 359                                       u8 hw_mode_id, u8 phy_id,
 360                                       struct ath11k_pdev *pdev)
 361 {
 362         struct wmi_mac_phy_capabilities *mac_phy_caps;
 363         struct ath11k_base *ab = wmi_handle->wmi_ab->ab;
 364         struct ath11k_band_cap *cap_band;
 365         struct ath11k_pdev_cap *pdev_cap = &pdev->cap;
 366         u32 phy_map;
 367         u32 hw_idx, phy_idx = 0;
 368
 369         if (!hw_caps || !wmi_hw_mode_caps || !hal_reg_caps)
 370                 return -EINVAL;
 371
 372         for (hw_idx = 0; hw_idx < hw_caps->num_hw_modes; hw_idx++) {
 373                 if (hw_mode_id == wmi_hw_mode_caps[hw_idx].hw_mode_id)
 374                         break;
 375
 376                 phy_map = wmi_hw_mode_caps[hw_idx].phy_id_map;
 377                 while (phy_map) {
 378                         phy_map >>= 1;
 379                         phy_idx++;
 380                 }
 381         }
 382
 383         if (hw_idx == hw_caps->num_hw_modes)
 384                 return -EINVAL;
 385
 386         phy_idx += phy_id;
 387         if (phy_id >= hal_reg_caps->num_phy)
 388                 return -EINVAL;
 389
 390         mac_phy_caps = wmi_mac_phy_caps + phy_idx;
 391
 392         pdev->pdev_id = mac_phy_caps->pdev_id;
 393         pdev_cap->supported_bands |= mac_phy_caps->supported_bands;
 394         pdev_cap->ampdu_density = mac_phy_caps->ampdu_density;
 395         ab->target_pdev_ids[ab->target_pdev_count].supported_bands =
 396                 mac_phy_caps->supported_bands;
 397         ab->target_pdev_ids[ab->target_pdev_count].pdev_id = mac_phy_caps->pdev_id;
 398         ab->target_pdev_count++;
 399
 400         if (!(mac_phy_caps->supported_bands & WMI_HOST_WLAN_2G_CAP) &&
 401             !(mac_phy_caps->supported_bands & WMI_HOST_WLAN_5G_CAP))
 402                 return -EINVAL;
 403
 404         /* Take non-zero tx/rx chainmask. If tx/rx chainmask differs from
 405          * band to band for a single radio, need to see how this should be
 406          * handled.
 407          */
 408         if (mac_phy_caps->supported_bands & WMI_HOST_WLAN_2G_CAP) {
 409                 pdev_cap->tx_chain_mask = mac_phy_caps->tx_chain_mask_2g;
 410                 pdev_cap->rx_chain_mask = mac_phy_caps->rx_chain_mask_2g;
 411         }
 412
 413         if (mac_phy_caps->supported_bands & WMI_HOST_WLAN_5G_CAP) {
 414                 pdev_cap->vht_cap = mac_phy_caps->vht_cap_info_5g;
 415                 pdev_cap->vht_mcs = mac_phy_caps->vht_supp_mcs_5g;
 416                 pdev_cap->he_mcs = mac_phy_caps->he_supp_mcs_5g;
 417                 pdev_cap->tx_chain_mask = mac_phy_caps->tx_chain_mask_5g;
 418                 pdev_cap->rx_chain_mask = mac_phy_caps->rx_chain_mask_5g;
 419                 pdev_cap->nss_ratio_enabled =
 420                         WMI_NSS_RATIO_ENABLE_DISABLE_GET(mac_phy_caps->nss_ratio);
 421                 pdev_cap->nss_ratio_info =
 422                         WMI_NSS_RATIO_INFO_GET(mac_phy_caps->nss_ratio);
 423         }
 424
 425         /* tx/rx chainmask reported from fw depends on the actual hw chains used,
 426          * For example, for 4x4 capable macphys, first 4 chains can be used for first
 427          * mac and the remaining 4 chains can be used for the second mac or vice-versa.
 428          * In this case, tx/rx chainmask 0xf will be advertised for first mac and 0xf0
 429          * will be advertised for second mac or vice-versa. Compute the shift value
 430          * for tx/rx chainmask which will be used to advertise supported ht/vht rates to
 431          * mac80211.
 432          */
 433         pdev_cap->tx_chain_mask_shift =
 434                         find_first_bit((unsigned long *)&pdev_cap->tx_chain_mask, 32);
 435         pdev_cap->rx_chain_mask_shift =
 436                         find_first_bit((unsigned long *)&pdev_cap->rx_chain_mask, 32);
 437
 438         if (mac_phy_caps->supported_bands & WMI_HOST_WLAN_2G_CAP) {
 439                 cap_band = &pdev_cap->band[NL80211_BAND_2GHZ];
 440                 cap_band->phy_id = mac_phy_caps->phy_id;
 441                 cap_band->max_bw_supported = mac_phy_caps->max_bw_supported_2g;
 442                 cap_band->ht_cap_info = mac_phy_caps->ht_cap_info_2g;
 443                 cap_band->he_cap_info[0] = mac_phy_caps->he_cap_info_2g;
 444                 cap_band->he_cap_info[1] = mac_phy_caps->he_cap_info_2g_ext;
 445                 cap_band->he_mcs = mac_phy_caps->he_supp_mcs_2g;
 446                 memcpy(cap_band->he_cap_phy_info, &mac_phy_caps->he_cap_phy_info_2g,
 447                        sizeof(u32) * PSOC_HOST_MAX_PHY_SIZE);
 448                 memcpy(&cap_band->he_ppet, &mac_phy_caps->he_ppet2g,
 449                        sizeof(struct ath11k_ppe_threshold));
 450         }
 451
 452         if (mac_phy_caps->supported_bands & WMI_HOST_WLAN_5G_CAP) {
 453                 cap_band = &pdev_cap->band[NL80211_BAND_5GHZ];
 454                 cap_band->phy_id = mac_phy_caps->phy_id;
 455                 cap_band->max_bw_supported = mac_phy_caps->max_bw_supported_5g;
 456                 cap_band->ht_cap_info = mac_phy_caps->ht_cap_info_5g;
 457                 cap_band->he_cap_info[0] = mac_phy_caps->he_cap_info_5g;
 458                 cap_band->he_cap_info[1] = mac_phy_caps->he_cap_info_5g_ext;
 459                 cap_band->he_mcs = mac_phy_caps->he_supp_mcs_5g;
 460                 memcpy(cap_band->he_cap_phy_info, &mac_phy_caps->he_cap_phy_info_5g,
 461                        sizeof(u32) * PSOC_HOST_MAX_PHY_SIZE);
 462                 memcpy(&cap_band->he_ppet, &mac_phy_caps->he_ppet5g,
 463                        sizeof(struct ath11k_ppe_threshold));
 464
 465                 cap_band = &pdev_cap->band[NL80211_BAND_6GHZ];
 466                 cap_band->max_bw_supported = mac_phy_caps->max_bw_supported_5g;
 467                 cap_band->ht_cap_info = mac_phy_caps->ht_cap_info_5g;
 468                 cap_band->he_cap_info[0] = mac_phy_caps->he_cap_info_5g;
 469                 cap_band->he_cap_info[1] = mac_phy_caps->he_cap_info_5g_ext;
 470                 cap_band->he_mcs = mac_phy_caps->he_supp_mcs_5g;
 471                 memcpy(cap_band->he_cap_phy_info, &mac_phy_caps->he_cap_phy_info_5g,
 472                        sizeof(u32) * PSOC_HOST_MAX_PHY_SIZE);
 473                 memcpy(&cap_band->he_ppet, &mac_phy_caps->he_ppet5g,
 474                        sizeof(struct ath11k_ppe_threshold));
 475         }
 476
 477         return 0;
 478 }
 479
 480 static int
 481 ath11k_pull_reg_cap_svc_rdy_ext(struct ath11k_pdev_wmi *wmi_handle,
 482                                 struct wmi_soc_hal_reg_capabilities *reg_caps,
 483                                 struct wmi_hal_reg_capabilities_ext *wmi_ext_reg_cap,
 484                                 u8 phy_idx,
 485                                 struct ath11k_hal_reg_capabilities_ext *param)
 486 {
 487         struct wmi_hal_reg_capabilities_ext *ext_reg_cap;
 488
 489         if (!reg_caps || !wmi_ext_reg_cap)
 490                 return -EINVAL;
 491
 492         if (phy_idx >= reg_caps->num_phy)
 493                 return -EINVAL;
 494
 495         ext_reg_cap = &wmi_ext_reg_cap[phy_idx];
 496
 497         param->phy_id = ext_reg_cap->phy_id;
 498         param->eeprom_reg_domain = ext_reg_cap->eeprom_reg_domain;
 499         param->eeprom_reg_domain_ext =
 500                               ext_reg_cap->eeprom_reg_domain_ext;
 501         param->regcap1 = ext_reg_cap->regcap1;
 502         param->regcap2 = ext_reg_cap->regcap2;
 503         /* check if param->wireless_mode is needed */
 504         param->low_2ghz_chan = ext_reg_cap->low_2ghz_chan;
 505         param->high_2ghz_chan = ext_reg_cap->high_2ghz_chan;
 506         param->low_5ghz_chan = ext_reg_cap->low_5ghz_chan;
 507         param->high_5ghz_chan = ext_reg_cap->high_5ghz_chan;
 508
 509         return 0;
 510 }
 511
 512 static int ath11k_pull_service_ready_tlv(struct ath11k_base *ab,
 513                                          const void *evt_buf,
 514                                          struct ath11k_targ_cap *cap)
 515 {
 516         const struct wmi_service_ready_event *ev = evt_buf;
 517
 518         if (!ev) {
 519                 ath11k_err(ab, "%s: failed by NULL param\n",
 520                            __func__);
 521                 return -EINVAL;
 522         }
 523
 524         cap->phy_capability = ev->phy_capability;
 525         cap->max_frag_entry = ev->max_frag_entry;
 526         cap->num_rf_chains = ev->num_rf_chains;
 527         cap->ht_cap_info = ev->ht_cap_info;
 528         cap->vht_cap_info = ev->vht_cap_info;
 529         cap->vht_supp_mcs = ev->vht_supp_mcs;
 530         cap->hw_min_tx_power = ev->hw_min_tx_power;
 531         cap->hw_max_tx_power = ev->hw_max_tx_power;
 532         cap->sys_cap_info = ev->sys_cap_info;
 533         cap->min_pkt_size_enable = ev->min_pkt_size_enable;
 534         cap->max_bcn_ie_size = ev->max_bcn_ie_size;
 535         cap->max_num_scan_channels = ev->max_num_scan_channels;
 536         cap->max_supported_macs = ev->max_supported_macs;
 537         cap->wmi_fw_sub_feat_caps = ev->wmi_fw_sub_feat_caps;
 538         cap->txrx_chainmask = ev->txrx_chainmask;
 539         cap->default_dbs_hw_mode_index = ev->default_dbs_hw_mode_index;
 540         cap->num_msdu_desc = ev->num_msdu_desc;
 541
 542         return 0;
 543 }
 544
 545 /* Save the wmi_service_bitmap into a linear bitmap. The wmi_services in
 546  * wmi_service ready event are advertised in b0-b3 (LSB 4-bits) of each
 547  * 4-byte word.
 548  */
 549 static void ath11k_wmi_service_bitmap_copy(struct ath11k_pdev_wmi *wmi,
 550                                            const u32 *wmi_svc_bm)
 551 {
 552         int i, j;
 553
 554         for (i = 0, j = 0; i < WMI_SERVICE_BM_SIZE && j < WMI_MAX_SERVICE; i++) {
 555                 do {
 556                         if (wmi_svc_bm[i] & BIT(j % WMI_SERVICE_BITS_IN_SIZE32))
 557                                 set_bit(j, wmi->wmi_ab->svc_map);
 558                 } while (++j % WMI_SERVICE_BITS_IN_SIZE32);
 559         }
 560 }
 561
 562 static int ath11k_wmi_tlv_svc_rdy_parse(struct ath11k_base *ab, u16 tag, u16 len,
 563                                         const void *ptr, void *data)
 564 {
 565         struct wmi_tlv_svc_ready_parse *svc_ready = data;
 566         struct ath11k_pdev_wmi *wmi_handle = &ab->wmi_ab.wmi[0];
 567         u16 expect_len;
 568
 569         switch (tag) {
 570         case WMI_TAG_SERVICE_READY_EVENT:
 571                 if (ath11k_pull_service_ready_tlv(ab, ptr, &ab->target_caps))
 572                         return -EINVAL;
 573                 break;
 574
 575         case WMI_TAG_ARRAY_UINT32:
 576                 if (!svc_ready->wmi_svc_bitmap_done) {
 577                         expect_len = WMI_SERVICE_BM_SIZE * sizeof(u32);
 578                         if (len < expect_len) {
 579                                 ath11k_warn(ab, "invalid len %d for the tag 0x%x\n",
 580                                             len, tag);
 581                                 return -EINVAL;
 582                         }
 583
 584                         ath11k_wmi_service_bitmap_copy(wmi_handle, ptr);
 585
 586                         svc_ready->wmi_svc_bitmap_done = true;
 587                 }
 588                 break;
 589         default:
 590                 break;
 591         }
 592
 593         return 0;
 594 }
 595
 596 static int ath11k_service_ready_event(struct ath11k_base *ab, struct sk_buff *skb)
 597 {
 598         struct wmi_tlv_svc_ready_parse svc_ready = { };
 599         int ret;
 600
 601         ret = ath11k_wmi_tlv_iter(ab, skb->data, skb->len,
 602                                   ath11k_wmi_tlv_svc_rdy_parse,
 603                                   &svc_ready);
 604         if (ret) {
 605                 ath11k_warn(ab, "failed to parse tlv %d\n", ret);
 606                 return ret;
 607         }
 608
 609         ath11k_dbg(ab, ATH11K_DBG_WMI, "event service ready");
 610
 611         return 0;
 612 }
 613
 614 struct sk_buff *ath11k_wmi_alloc_skb(struct ath11k_wmi_base *wmi_ab, u32 len)
 615 {
 616         struct sk_buff *skb;
 617         struct ath11k_base *ab = wmi_ab->ab;
 618         u32 round_len = roundup(len, 4);
 619
 620         skb = ath11k_htc_alloc_skb(ab, WMI_SKB_HEADROOM + round_len);
 621         if (!skb)
 622                 return NULL;
 623
 624         skb_reserve(skb, WMI_SKB_HEADROOM);
 625         if (!IS_ALIGNED((unsigned long)skb->data, 4))
 626                 ath11k_warn(ab, "unaligned WMI skb data\n");
 627
 628         skb_put(skb, round_len);
 629         memset(skb->data, 0, round_len);
 630
 631         return skb;
 632 }
 633
 634 static u32 ath11k_wmi_mgmt_get_freq(struct ath11k *ar,
 635                                     struct ieee80211_tx_info *info)
 636 {
 637         struct ath11k_base *ab = ar->ab;
 638         u32 freq = 0;
 639
 640         if (ab->hw_params.support_off_channel_tx &&
 641             ar->scan.is_roc &&
 642             (info->flags & IEEE80211_TX_CTL_TX_OFFCHAN))
 643                 freq = ar->scan.roc_freq;
 644
 645         return freq;
 646 }
 647
 648 int ath11k_wmi_mgmt_send(struct ath11k *ar, u32 vdev_id, u32 buf_id,
 649                          struct sk_buff *frame)
 650 {
 651         struct ath11k_pdev_wmi *wmi = ar->wmi;
 652         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(frame);
 653         struct wmi_mgmt_send_cmd *cmd;
 654         struct wmi_tlv *frame_tlv;
 655         struct sk_buff *skb;
 656         u32 buf_len;
 657         int ret, len;
 658
 659         buf_len = frame->len < WMI_MGMT_SEND_DOWNLD_LEN ?
 660                   frame->len : WMI_MGMT_SEND_DOWNLD_LEN;
 661
 662         len = sizeof(*cmd) + sizeof(*frame_tlv) + roundup(buf_len, 4);
 663
 664         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
 665         if (!skb)
 666                 return -ENOMEM;
 667
 668         cmd = (struct wmi_mgmt_send_cmd *)skb->data;
 669         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_MGMT_TX_SEND_CMD) |
 670                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
 671         cmd->vdev_id = vdev_id;
 672         cmd->desc_id = buf_id;
 673         cmd->chanfreq = ath11k_wmi_mgmt_get_freq(ar, info);
 674         cmd->paddr_lo = lower_32_bits(ATH11K_SKB_CB(frame)->paddr);
 675         cmd->paddr_hi = upper_32_bits(ATH11K_SKB_CB(frame)->paddr);
 676         cmd->frame_len = frame->len;
 677         cmd->buf_len = buf_len;
 678         cmd->tx_params_valid = 0;
 679
 680         frame_tlv = (struct wmi_tlv *)(skb->data + sizeof(*cmd));
 681         frame_tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
 682                             FIELD_PREP(WMI_TLV_LEN, buf_len);
 683
 684         memcpy(frame_tlv->value, frame->data, buf_len);
 685
 686         ath11k_ce_byte_swap(frame_tlv->value, buf_len);
 687
 688         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_MGMT_TX_SEND_CMDID);
 689         if (ret) {
 690                 ath11k_warn(ar->ab,
 691                             "failed to submit WMI_MGMT_TX_SEND_CMDID cmd\n");
 692                 dev_kfree_skb(skb);
 693         }
 694
 695         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd mgmt tx send");
 696
 697         return ret;
 698 }
 699
 700 int ath11k_wmi_vdev_create(struct ath11k *ar, u8 *macaddr,
 701                            struct vdev_create_params *param)
 702 {
 703         struct ath11k_pdev_wmi *wmi = ar->wmi;
 704         struct wmi_vdev_create_cmd *cmd;
 705         struct sk_buff *skb;
 706         struct wmi_vdev_txrx_streams *txrx_streams;
 707         struct wmi_tlv *tlv;
 708         int ret, len;
 709         void *ptr;
 710
 711         /* It can be optimized my sending tx/rx chain configuration
 712          * only for supported bands instead of always sending it for
 713          * both the bands.
 714          */
 715         len = sizeof(*cmd) + TLV_HDR_SIZE +
 716                 (WMI_NUM_SUPPORTED_BAND_MAX * sizeof(*txrx_streams));
 717
 718         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
 719         if (!skb)
 720                 return -ENOMEM;
 721
 722         cmd = (struct wmi_vdev_create_cmd *)skb->data;
 723         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VDEV_CREATE_CMD) |
 724                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
 725
 726         cmd->vdev_id = param->if_id;
 727         cmd->vdev_type = param->type;
 728         cmd->vdev_subtype = param->subtype;
 729         cmd->num_cfg_txrx_streams = WMI_NUM_SUPPORTED_BAND_MAX;
 730         cmd->pdev_id = param->pdev_id;
 731         cmd->mbssid_flags = param->mbssid_flags;
 732         cmd->mbssid_tx_vdev_id = param->mbssid_tx_vdev_id;
 733
 734         ether_addr_copy(cmd->vdev_macaddr.addr, macaddr);
 735
 736         ptr = skb->data + sizeof(*cmd);
 737         len = WMI_NUM_SUPPORTED_BAND_MAX * sizeof(*txrx_streams);
 738
 739         tlv = ptr;
 740         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
 741                       FIELD_PREP(WMI_TLV_LEN, len);
 742
 743         ptr += TLV_HDR_SIZE;
 744         txrx_streams = ptr;
 745         len = sizeof(*txrx_streams);
 746         txrx_streams->tlv_header =
 747                 FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VDEV_TXRX_STREAMS) |
 748                 FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
 749         txrx_streams->band = WMI_TPC_CHAINMASK_CONFIG_BAND_2G;
 750         txrx_streams->supported_tx_streams =
 751                                  param->chains[NL80211_BAND_2GHZ].tx;
 752         txrx_streams->supported_rx_streams =
 753                                  param->chains[NL80211_BAND_2GHZ].rx;
 754
 755         txrx_streams++;
 756         txrx_streams->tlv_header =
 757                 FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VDEV_TXRX_STREAMS) |
 758                 FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
 759         txrx_streams->band = WMI_TPC_CHAINMASK_CONFIG_BAND_5G;
 760         txrx_streams->supported_tx_streams =
 761                                  param->chains[NL80211_BAND_5GHZ].tx;
 762         txrx_streams->supported_rx_streams =
 763                                  param->chains[NL80211_BAND_5GHZ].rx;
 764
 765         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_VDEV_CREATE_CMDID);
 766         if (ret) {
 767                 ath11k_warn(ar->ab,
 768                             "failed to submit WMI_VDEV_CREATE_CMDID\n");
 769                 dev_kfree_skb(skb);
 770         }
 771
 772         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
 773                    "cmd vdev create id %d type %d subtype %d macaddr %pM pdevid %d\n",
 774                    param->if_id, param->type, param->subtype,
 775                    macaddr, param->pdev_id);
 776
 777         return ret;
 778 }
 779
 780 int ath11k_wmi_vdev_delete(struct ath11k *ar, u8 vdev_id)
 781 {
 782         struct ath11k_pdev_wmi *wmi = ar->wmi;
 783         struct wmi_vdev_delete_cmd *cmd;
 784         struct sk_buff *skb;
 785         int ret;
 786
 787         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
 788         if (!skb)
 789                 return -ENOMEM;
 790
 791         cmd = (struct wmi_vdev_delete_cmd *)skb->data;
 792         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VDEV_DELETE_CMD) |
 793                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
 794         cmd->vdev_id = vdev_id;
 795
 796         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_VDEV_DELETE_CMDID);
 797         if (ret) {
 798                 ath11k_warn(ar->ab, "failed to submit WMI_VDEV_DELETE_CMDID\n");
 799                 dev_kfree_skb(skb);
 800         }
 801
 802         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd vdev delete id %d\n", vdev_id);
 803
 804         return ret;
 805 }
 806
 807 int ath11k_wmi_vdev_stop(struct ath11k *ar, u8 vdev_id)
 808 {
 809         struct ath11k_pdev_wmi *wmi = ar->wmi;
 810         struct wmi_vdev_stop_cmd *cmd;
 811         struct sk_buff *skb;
 812         int ret;
 813
 814         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
 815         if (!skb)
 816                 return -ENOMEM;
 817
 818         cmd = (struct wmi_vdev_stop_cmd *)skb->data;
 819
 820         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VDEV_STOP_CMD) |
 821                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
 822         cmd->vdev_id = vdev_id;
 823
 824         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_VDEV_STOP_CMDID);
 825         if (ret) {
 826                 ath11k_warn(ar->ab, "failed to submit WMI_VDEV_STOP cmd\n");
 827                 dev_kfree_skb(skb);
 828         }
 829
 830         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd vdev stop id 0x%x\n", vdev_id);
 831
 832         return ret;
 833 }
 834
 835 int ath11k_wmi_vdev_down(struct ath11k *ar, u8 vdev_id)
 836 {
 837         struct ath11k_pdev_wmi *wmi = ar->wmi;
 838         struct wmi_vdev_down_cmd *cmd;
 839         struct sk_buff *skb;
 840         int ret;
 841
 842         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
 843         if (!skb)
 844                 return -ENOMEM;
 845
 846         cmd = (struct wmi_vdev_down_cmd *)skb->data;
 847
 848         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VDEV_DOWN_CMD) |
 849                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
 850         cmd->vdev_id = vdev_id;
 851
 852         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_VDEV_DOWN_CMDID);
 853         if (ret) {
 854                 ath11k_warn(ar->ab, "failed to submit WMI_VDEV_DOWN cmd\n");
 855                 dev_kfree_skb(skb);
 856         }
 857
 858         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd vdev down id 0x%x\n", vdev_id);
 859
 860         return ret;
 861 }
 862
 863 static void ath11k_wmi_put_wmi_channel(struct wmi_channel *chan,
 864                                        struct wmi_vdev_start_req_arg *arg)
 865 {
 866         u32 center_freq1 = arg->channel.band_center_freq1;
 867
 868         memset(chan, 0, sizeof(*chan));
 869
 870         chan->mhz = arg->channel.freq;
 871         chan->band_center_freq1 = arg->channel.band_center_freq1;
 872
 873         if (arg->channel.mode == MODE_11AX_HE160) {
 874                 if (arg->channel.freq > arg->channel.band_center_freq1)
 875                         chan->band_center_freq1 = center_freq1 + 40;
 876                 else
 877                         chan->band_center_freq1 = center_freq1 - 40;
 878
 879                 chan->band_center_freq2 = arg->channel.band_center_freq1;
 880
 881         } else if ((arg->channel.mode == MODE_11AC_VHT80_80) ||
 882                    (arg->channel.mode == MODE_11AX_HE80_80)) {
 883                 chan->band_center_freq2 = arg->channel.band_center_freq2;
 884         } else {
 885                 chan->band_center_freq2 = 0;
 886         }
 887
 888         chan->info |= FIELD_PREP(WMI_CHAN_INFO_MODE, arg->channel.mode);
 889         if (arg->channel.passive)
 890                 chan->info |= WMI_CHAN_INFO_PASSIVE;
 891         if (arg->channel.allow_ibss)
 892                 chan->info |= WMI_CHAN_INFO_ADHOC_ALLOWED;
 893         if (arg->channel.allow_ht)
 894                 chan->info |= WMI_CHAN_INFO_ALLOW_HT;
 895         if (arg->channel.allow_vht)
 896                 chan->info |= WMI_CHAN_INFO_ALLOW_VHT;
 897         if (arg->channel.allow_he)
 898                 chan->info |= WMI_CHAN_INFO_ALLOW_HE;
 899         if (arg->channel.ht40plus)
 900                 chan->info |= WMI_CHAN_INFO_HT40_PLUS;
 901         if (arg->channel.chan_radar)
 902                 chan->info |= WMI_CHAN_INFO_DFS;
 903         if (arg->channel.freq2_radar)
 904                 chan->info |= WMI_CHAN_INFO_DFS_FREQ2;
 905
 906         chan->reg_info_1 = FIELD_PREP(WMI_CHAN_REG_INFO1_MAX_PWR,
 907                                       arg->channel.max_power) |
 908                 FIELD_PREP(WMI_CHAN_REG_INFO1_MAX_REG_PWR,
 909                            arg->channel.max_reg_power);
 910
 911         chan->reg_info_2 = FIELD_PREP(WMI_CHAN_REG_INFO2_ANT_MAX,
 912                                       arg->channel.max_antenna_gain) |
 913                 FIELD_PREP(WMI_CHAN_REG_INFO2_MAX_TX_PWR,
 914                            arg->channel.max_power);
 915 }
 916
 917 int ath11k_wmi_vdev_start(struct ath11k *ar, struct wmi_vdev_start_req_arg *arg,
 918                           bool restart)
 919 {
 920         struct ath11k_pdev_wmi *wmi = ar->wmi;
 921         struct wmi_vdev_start_request_cmd *cmd;
 922         struct sk_buff *skb;
 923         struct wmi_channel *chan;
 924         struct wmi_tlv *tlv;
 925         void *ptr;
 926         int ret, len;
 927
 928         if (WARN_ON(arg->ssid_len > sizeof(cmd->ssid.ssid)))
 929                 return -EINVAL;
 930
 931         len = sizeof(*cmd) + sizeof(*chan) + TLV_HDR_SIZE;
 932
 933         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
 934         if (!skb)
 935                 return -ENOMEM;
 936
 937         cmd = (struct wmi_vdev_start_request_cmd *)skb->data;
 938         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
 939                                      WMI_TAG_VDEV_START_REQUEST_CMD) |
 940                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
 941         cmd->vdev_id = arg->vdev_id;
 942         cmd->beacon_interval = arg->bcn_intval;
 943         cmd->bcn_tx_rate = arg->bcn_tx_rate;
 944         cmd->dtim_period = arg->dtim_period;
 945         cmd->num_noa_descriptors = arg->num_noa_descriptors;
 946         cmd->preferred_rx_streams = arg->pref_rx_streams;
 947         cmd->preferred_tx_streams = arg->pref_tx_streams;
 948         cmd->cac_duration_ms = arg->cac_duration_ms;
 949         cmd->regdomain = arg->regdomain;
 950         cmd->he_ops = arg->he_ops;
 951         cmd->mbssid_flags = arg->mbssid_flags;
 952         cmd->mbssid_tx_vdev_id = arg->mbssid_tx_vdev_id;
 953
 954         if (!restart) {
 955                 if (arg->ssid) {
 956                         cmd->ssid.ssid_len = arg->ssid_len;
 957                         memcpy(cmd->ssid.ssid, arg->ssid, arg->ssid_len);
 958                 }
 959                 if (arg->hidden_ssid)
 960                         cmd->flags |= WMI_VDEV_START_HIDDEN_SSID;
 961                 if (arg->pmf_enabled)
 962                         cmd->flags |= WMI_VDEV_START_PMF_ENABLED;
 963         }
 964
 965         cmd->flags |= WMI_VDEV_START_LDPC_RX_ENABLED;
 966         if (test_bit(ATH11K_FLAG_HW_CRYPTO_DISABLED, &ar->ab->dev_flags))
 967                 cmd->flags |= WMI_VDEV_START_HW_ENCRYPTION_DISABLED;
 968
 969         ptr = skb->data + sizeof(*cmd);
 970         chan = ptr;
 971
 972         ath11k_wmi_put_wmi_channel(chan, arg);
 973
 974         chan->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_CHANNEL) |
 975                            FIELD_PREP(WMI_TLV_LEN,
 976                                       sizeof(*chan) - TLV_HDR_SIZE);
 977         ptr += sizeof(*chan);
 978
 979         tlv = ptr;
 980         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
 981                       FIELD_PREP(WMI_TLV_LEN, 0);
 982
 983         /* Note: This is a nested TLV containing:
 984          * [wmi_tlv][wmi_p2p_noa_descriptor][wmi_tlv]..
 985          */
 986
 987         ptr += sizeof(*tlv);
 988
 989         if (restart)
 990                 ret = ath11k_wmi_cmd_send(wmi, skb,
 991                                           WMI_VDEV_RESTART_REQUEST_CMDID);
 992         else
 993                 ret = ath11k_wmi_cmd_send(wmi, skb,
 994                                           WMI_VDEV_START_REQUEST_CMDID);
 995         if (ret) {
 996                 ath11k_warn(ar->ab, "failed to submit vdev_%s cmd\n",
 997                             restart ? "restart" : "start");
 998                 dev_kfree_skb(skb);
 999         }
1000
1001         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd vdev %s id 0x%x freq 0x%x mode 0x%x\n",
1002                    restart ? "restart" : "start", arg->vdev_id,
1003                    arg->channel.freq, arg->channel.mode);
1004
1005         return ret;
1006 }
1007
1008 int ath11k_wmi_vdev_up(struct ath11k *ar, u32 vdev_id, u32 aid, const u8 *bssid,
1009                        u8 *tx_bssid, u32 nontx_profile_idx, u32 nontx_profile_cnt)
1010 {
1011         struct ath11k_pdev_wmi *wmi = ar->wmi;
1012         struct wmi_vdev_up_cmd *cmd;
1013         struct ieee80211_bss_conf *bss_conf;
1014         struct ath11k_vif *arvif;
1015         struct sk_buff *skb;
1016         int ret;
1017
1018         arvif = ath11k_mac_get_arvif(ar, vdev_id);
1019
1020         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1021         if (!skb)
1022                 return -ENOMEM;
1023
1024         cmd = (struct wmi_vdev_up_cmd *)skb->data;
1025
1026         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VDEV_UP_CMD) |
1027                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1028         cmd->vdev_id = vdev_id;
1029         cmd->vdev_assoc_id = aid;
1030
1031         ether_addr_copy(cmd->vdev_bssid.addr, bssid);
1032
1033         cmd->nontx_profile_idx = nontx_profile_idx;
1034         cmd->nontx_profile_cnt = nontx_profile_cnt;
1035         if (tx_bssid)
1036                 ether_addr_copy(cmd->tx_vdev_bssid.addr, tx_bssid);
1037
1038         if (arvif && arvif->vif->type == NL80211_IFTYPE_STATION) {
1039                 bss_conf = &arvif->vif->bss_conf;
1040
1041                 if (bss_conf->nontransmitted) {
1042                         ether_addr_copy(cmd->tx_vdev_bssid.addr,
1043                                         bss_conf->transmitter_bssid);
1044                         cmd->nontx_profile_idx = bss_conf->bssid_index;
1045                         cmd->nontx_profile_cnt = bss_conf->bssid_indicator;
1046                 }
1047         }
1048
1049         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_VDEV_UP_CMDID);
1050         if (ret) {
1051                 ath11k_warn(ar->ab, "failed to submit WMI_VDEV_UP cmd\n");
1052                 dev_kfree_skb(skb);
1053         }
1054
1055         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1056                    "cmd vdev up id 0x%x assoc id %d bssid %pM\n",
1057                    vdev_id, aid, bssid);
1058
1059         return ret;
1060 }
1061
1062 int ath11k_wmi_send_peer_create_cmd(struct ath11k *ar,
1063                                     struct peer_create_params *param)
1064 {
1065         struct ath11k_pdev_wmi *wmi = ar->wmi;
1066         struct wmi_peer_create_cmd *cmd;
1067         struct sk_buff *skb;
1068         int ret;
1069
1070         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1071         if (!skb)
1072                 return -ENOMEM;
1073
1074         cmd = (struct wmi_peer_create_cmd *)skb->data;
1075         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PEER_CREATE_CMD) |
1076                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1077
1078         ether_addr_copy(cmd->peer_macaddr.addr, param->peer_addr);
1079         cmd->peer_type = param->peer_type;
1080         cmd->vdev_id = param->vdev_id;
1081
1082         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PEER_CREATE_CMDID);
1083         if (ret) {
1084                 ath11k_warn(ar->ab, "failed to submit WMI_PEER_CREATE cmd\n");
1085                 dev_kfree_skb(skb);
1086         }
1087
1088         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1089                    "cmd peer create vdev_id %d peer_addr %pM\n",
1090                    param->vdev_id, param->peer_addr);
1091
1092         return ret;
1093 }
1094
1095 int ath11k_wmi_send_peer_delete_cmd(struct ath11k *ar,
1096                                     const u8 *peer_addr, u8 vdev_id)
1097 {
1098         struct ath11k_pdev_wmi *wmi = ar->wmi;
1099         struct wmi_peer_delete_cmd *cmd;
1100         struct sk_buff *skb;
1101         int ret;
1102
1103         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1104         if (!skb)
1105                 return -ENOMEM;
1106
1107         cmd = (struct wmi_peer_delete_cmd *)skb->data;
1108         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PEER_DELETE_CMD) |
1109                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1110
1111         ether_addr_copy(cmd->peer_macaddr.addr, peer_addr);
1112         cmd->vdev_id = vdev_id;
1113
1114         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PEER_DELETE_CMDID);
1115         if (ret) {
1116                 ath11k_warn(ar->ab, "failed to send WMI_PEER_DELETE cmd\n");
1117                 dev_kfree_skb(skb);
1118         }
1119
1120         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1121                    "cmd peer delete vdev_id %d peer_addr %pM\n",
1122                    vdev_id,  peer_addr);
1123
1124         return ret;
1125 }
1126
1127 int ath11k_wmi_send_pdev_set_regdomain(struct ath11k *ar,
1128                                        struct pdev_set_regdomain_params *param)
1129 {
1130         struct ath11k_pdev_wmi *wmi = ar->wmi;
1131         struct wmi_pdev_set_regdomain_cmd *cmd;
1132         struct sk_buff *skb;
1133         int ret;
1134
1135         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1136         if (!skb)
1137                 return -ENOMEM;
1138
1139         cmd = (struct wmi_pdev_set_regdomain_cmd *)skb->data;
1140         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
1141                                      WMI_TAG_PDEV_SET_REGDOMAIN_CMD) |
1142                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1143
1144         cmd->reg_domain = param->current_rd_in_use;
1145         cmd->reg_domain_2g = param->current_rd_2g;
1146         cmd->reg_domain_5g = param->current_rd_5g;
1147         cmd->conformance_test_limit_2g = param->ctl_2g;
1148         cmd->conformance_test_limit_5g = param->ctl_5g;
1149         cmd->dfs_domain = param->dfs_domain;
1150         cmd->pdev_id = param->pdev_id;
1151
1152         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PDEV_SET_REGDOMAIN_CMDID);
1153         if (ret) {
1154                 ath11k_warn(ar->ab,
1155                             "failed to send WMI_PDEV_SET_REGDOMAIN cmd\n");
1156                 dev_kfree_skb(skb);
1157         }
1158
1159         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1160                    "cmd pdev regd rd %d rd2g %d rd5g %d domain %d pdev id %d\n",
1161                    param->current_rd_in_use, param->current_rd_2g,
1162                    param->current_rd_5g, param->dfs_domain, param->pdev_id);
1163
1164         return ret;
1165 }
1166
1167 int ath11k_wmi_set_peer_param(struct ath11k *ar, const u8 *peer_addr,
1168                               u32 vdev_id, u32 param_id, u32 param_val)
1169 {
1170         struct ath11k_pdev_wmi *wmi = ar->wmi;
1171         struct wmi_peer_set_param_cmd *cmd;
1172         struct sk_buff *skb;
1173         int ret;
1174
1175         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1176         if (!skb)
1177                 return -ENOMEM;
1178
1179         cmd = (struct wmi_peer_set_param_cmd *)skb->data;
1180         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PEER_SET_PARAM_CMD) |
1181                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1182         ether_addr_copy(cmd->peer_macaddr.addr, peer_addr);
1183         cmd->vdev_id = vdev_id;
1184         cmd->param_id = param_id;
1185         cmd->param_value = param_val;
1186
1187         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PEER_SET_PARAM_CMDID);
1188         if (ret) {
1189                 ath11k_warn(ar->ab, "failed to send WMI_PEER_SET_PARAM cmd\n");
1190                 dev_kfree_skb(skb);
1191         }
1192
1193         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1194                    "cmd peer set param vdev %d peer 0x%pM set param %d value %d\n",
1195                    vdev_id, peer_addr, param_id, param_val);
1196
1197         return ret;
1198 }
1199
1200 int ath11k_wmi_send_peer_flush_tids_cmd(struct ath11k *ar,
1201                                         u8 peer_addr[ETH_ALEN],
1202                                         struct peer_flush_params *param)
1203 {
1204         struct ath11k_pdev_wmi *wmi = ar->wmi;
1205         struct wmi_peer_flush_tids_cmd *cmd;
1206         struct sk_buff *skb;
1207         int ret;
1208
1209         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1210         if (!skb)
1211                 return -ENOMEM;
1212
1213         cmd = (struct wmi_peer_flush_tids_cmd *)skb->data;
1214         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PEER_FLUSH_TIDS_CMD) |
1215                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1216
1217         ether_addr_copy(cmd->peer_macaddr.addr, peer_addr);
1218         cmd->peer_tid_bitmap = param->peer_tid_bitmap;
1219         cmd->vdev_id = param->vdev_id;
1220
1221         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PEER_FLUSH_TIDS_CMDID);
1222         if (ret) {
1223                 ath11k_warn(ar->ab,
1224                             "failed to send WMI_PEER_FLUSH_TIDS cmd\n");
1225                 dev_kfree_skb(skb);
1226         }
1227
1228         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1229                    "cmd peer flush tids vdev_id %d peer_addr %pM tids %08x\n",
1230                    param->vdev_id, peer_addr, param->peer_tid_bitmap);
1231
1232         return ret;
1233 }
1234
1235 int ath11k_wmi_peer_rx_reorder_queue_setup(struct ath11k *ar,
1236                                            int vdev_id, const u8 *addr,
1237                                            dma_addr_t paddr, u8 tid,
1238                                            u8 ba_window_size_valid,
1239                                            u32 ba_window_size)
1240 {
1241         struct wmi_peer_reorder_queue_setup_cmd *cmd;
1242         struct sk_buff *skb;
1243         int ret;
1244
1245         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, sizeof(*cmd));
1246         if (!skb)
1247                 return -ENOMEM;
1248
1249         cmd = (struct wmi_peer_reorder_queue_setup_cmd *)skb->data;
1250         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
1251                                      WMI_TAG_REORDER_QUEUE_SETUP_CMD) |
1252                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1253
1254         ether_addr_copy(cmd->peer_macaddr.addr, addr);
1255         cmd->vdev_id = vdev_id;
1256         cmd->tid = tid;
1257         cmd->queue_ptr_lo = lower_32_bits(paddr);
1258         cmd->queue_ptr_hi = upper_32_bits(paddr);
1259         cmd->queue_no = tid;
1260         cmd->ba_window_size_valid = ba_window_size_valid;
1261         cmd->ba_window_size = ba_window_size;
1262
1263         ret = ath11k_wmi_cmd_send(ar->wmi, skb,
1264                                   WMI_PEER_REORDER_QUEUE_SETUP_CMDID);
1265         if (ret) {
1266                 ath11k_warn(ar->ab,
1267                             "failed to send WMI_PEER_REORDER_QUEUE_SETUP\n");
1268                 dev_kfree_skb(skb);
1269         }
1270
1271         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1272                    "cmd peer reorder queue setup addr %pM vdev_id %d tid %d\n",
1273                    addr, vdev_id, tid);
1274
1275         return ret;
1276 }
1277
1278 int
1279 ath11k_wmi_rx_reord_queue_remove(struct ath11k *ar,
1280                                  struct rx_reorder_queue_remove_params *param)
1281 {
1282         struct ath11k_pdev_wmi *wmi = ar->wmi;
1283         struct wmi_peer_reorder_queue_remove_cmd *cmd;
1284         struct sk_buff *skb;
1285         int ret;
1286
1287         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1288         if (!skb)
1289                 return -ENOMEM;
1290
1291         cmd = (struct wmi_peer_reorder_queue_remove_cmd *)skb->data;
1292         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
1293                                      WMI_TAG_REORDER_QUEUE_REMOVE_CMD) |
1294                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1295
1296         ether_addr_copy(cmd->peer_macaddr.addr, param->peer_macaddr);
1297         cmd->vdev_id = param->vdev_id;
1298         cmd->tid_mask = param->peer_tid_bitmap;
1299
1300         ret = ath11k_wmi_cmd_send(wmi, skb,
1301                                   WMI_PEER_REORDER_QUEUE_REMOVE_CMDID);
1302         if (ret) {
1303                 ath11k_warn(ar->ab,
1304                             "failed to send WMI_PEER_REORDER_QUEUE_REMOVE_CMDID");
1305                 dev_kfree_skb(skb);
1306         }
1307
1308         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1309                    "cmd peer reorder queue remove peer_macaddr %pM vdev_id %d tid_map %d",
1310                    param->peer_macaddr, param->vdev_id, param->peer_tid_bitmap);
1311
1312         return ret;
1313 }
1314
1315 int ath11k_wmi_pdev_set_param(struct ath11k *ar, u32 param_id,
1316                               u32 param_value, u8 pdev_id)
1317 {
1318         struct ath11k_pdev_wmi *wmi = ar->wmi;
1319         struct wmi_pdev_set_param_cmd *cmd;
1320         struct sk_buff *skb;
1321         int ret;
1322
1323         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1324         if (!skb)
1325                 return -ENOMEM;
1326
1327         cmd = (struct wmi_pdev_set_param_cmd *)skb->data;
1328         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_SET_PARAM_CMD) |
1329                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1330         cmd->pdev_id = pdev_id;
1331         cmd->param_id = param_id;
1332         cmd->param_value = param_value;
1333
1334         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PDEV_SET_PARAM_CMDID);
1335         if (ret) {
1336                 ath11k_warn(ar->ab, "failed to send WMI_PDEV_SET_PARAM cmd\n");
1337                 dev_kfree_skb(skb);
1338         }
1339
1340         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1341                    "cmd pdev set param %d pdev id %d value %d\n",
1342                    param_id, pdev_id, param_value);
1343
1344         return ret;
1345 }
1346
1347 int ath11k_wmi_pdev_set_ps_mode(struct ath11k *ar, int vdev_id,
1348                                 enum wmi_sta_ps_mode psmode)
1349 {
1350         struct ath11k_pdev_wmi *wmi = ar->wmi;
1351         struct wmi_pdev_set_ps_mode_cmd *cmd;
1352         struct sk_buff *skb;
1353         int ret;
1354
1355         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1356         if (!skb)
1357                 return -ENOMEM;
1358
1359         cmd = (struct wmi_pdev_set_ps_mode_cmd *)skb->data;
1360         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_STA_POWERSAVE_MODE_CMD) |
1361                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1362         cmd->vdev_id = vdev_id;
1363         cmd->sta_ps_mode = psmode;
1364
1365         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_STA_POWERSAVE_MODE_CMDID);
1366         if (ret) {
1367                 ath11k_warn(ar->ab, "failed to send WMI_PDEV_SET_PARAM cmd\n");
1368                 dev_kfree_skb(skb);
1369         }
1370
1371         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1372                    "cmd sta powersave mode psmode %d vdev id %d\n",
1373                    psmode, vdev_id);
1374
1375         return ret;
1376 }
1377
1378 int ath11k_wmi_pdev_suspend(struct ath11k *ar, u32 suspend_opt,
1379                             u32 pdev_id)
1380 {
1381         struct ath11k_pdev_wmi *wmi = ar->wmi;
1382         struct wmi_pdev_suspend_cmd *cmd;
1383         struct sk_buff *skb;
1384         int ret;
1385
1386         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1387         if (!skb)
1388                 return -ENOMEM;
1389
1390         cmd = (struct wmi_pdev_suspend_cmd *)skb->data;
1391
1392         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_SUSPEND_CMD) |
1393                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1394
1395         cmd->suspend_opt = suspend_opt;
1396         cmd->pdev_id = pdev_id;
1397
1398         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PDEV_SUSPEND_CMDID);
1399         if (ret) {
1400                 ath11k_warn(ar->ab, "failed to send WMI_PDEV_SUSPEND cmd\n");
1401                 dev_kfree_skb(skb);
1402         }
1403
1404         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1405                    "cmd pdev suspend pdev_id %d\n", pdev_id);
1406
1407         return ret;
1408 }
1409
1410 int ath11k_wmi_pdev_resume(struct ath11k *ar, u32 pdev_id)
1411 {
1412         struct ath11k_pdev_wmi *wmi = ar->wmi;
1413         struct wmi_pdev_resume_cmd *cmd;
1414         struct sk_buff *skb;
1415         int ret;
1416
1417         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1418         if (!skb)
1419                 return -ENOMEM;
1420
1421         cmd = (struct wmi_pdev_resume_cmd *)skb->data;
1422
1423         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_RESUME_CMD) |
1424                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1425         cmd->pdev_id = pdev_id;
1426
1427         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PDEV_RESUME_CMDID);
1428         if (ret) {
1429                 ath11k_warn(ar->ab, "failed to send WMI_PDEV_RESUME cmd\n");
1430                 dev_kfree_skb(skb);
1431         }
1432
1433         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1434                    "cmd pdev resume pdev id %d\n", pdev_id);
1435
1436         return ret;
1437 }
1438
1439 /* TODO FW Support for the cmd is not available yet.
1440  * Can be tested once the command and corresponding
1441  * event is implemented in FW
1442  */
1443 int ath11k_wmi_pdev_bss_chan_info_request(struct ath11k *ar,
1444                                           enum wmi_bss_chan_info_req_type type)
1445 {
1446         struct ath11k_pdev_wmi *wmi = ar->wmi;
1447         struct wmi_pdev_bss_chan_info_req_cmd *cmd;
1448         struct sk_buff *skb;
1449         int ret;
1450
1451         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1452         if (!skb)
1453                 return -ENOMEM;
1454
1455         cmd = (struct wmi_pdev_bss_chan_info_req_cmd *)skb->data;
1456
1457         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
1458                                      WMI_TAG_PDEV_BSS_CHAN_INFO_REQUEST) |
1459                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1460         cmd->req_type = type;
1461         cmd->pdev_id = ar->pdev->pdev_id;
1462
1463         ret = ath11k_wmi_cmd_send(wmi, skb,
1464                                   WMI_PDEV_BSS_CHAN_INFO_REQUEST_CMDID);
1465         if (ret) {
1466                 ath11k_warn(ar->ab,
1467                             "failed to send WMI_PDEV_BSS_CHAN_INFO_REQUEST cmd\n");
1468                 dev_kfree_skb(skb);
1469         }
1470
1471         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1472                    "cmd pdev bss chan info request type %d\n", type);
1473
1474         return ret;
1475 }
1476
1477 int ath11k_wmi_send_set_ap_ps_param_cmd(struct ath11k *ar, u8 *peer_addr,
1478                                         struct ap_ps_params *param)
1479 {
1480         struct ath11k_pdev_wmi *wmi = ar->wmi;
1481         struct wmi_ap_ps_peer_cmd *cmd;
1482         struct sk_buff *skb;
1483         int ret;
1484
1485         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1486         if (!skb)
1487                 return -ENOMEM;
1488
1489         cmd = (struct wmi_ap_ps_peer_cmd *)skb->data;
1490         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_AP_PS_PEER_CMD) |
1491                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1492
1493         cmd->vdev_id = param->vdev_id;
1494         ether_addr_copy(cmd->peer_macaddr.addr, peer_addr);
1495         cmd->param = param->param;
1496         cmd->value = param->value;
1497
1498         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_AP_PS_PEER_PARAM_CMDID);
1499         if (ret) {
1500                 ath11k_warn(ar->ab,
1501                             "failed to send WMI_AP_PS_PEER_PARAM_CMDID\n");
1502                 dev_kfree_skb(skb);
1503         }
1504
1505         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1506                    "cmd ap ps peer param vdev id %d peer %pM param %d value %d\n",
1507                    param->vdev_id, peer_addr, param->param, param->value);
1508
1509         return ret;
1510 }
1511
1512 int ath11k_wmi_set_sta_ps_param(struct ath11k *ar, u32 vdev_id,
1513                                 u32 param, u32 param_value)
1514 {
1515         struct ath11k_pdev_wmi *wmi = ar->wmi;
1516         struct wmi_sta_powersave_param_cmd *cmd;
1517         struct sk_buff *skb;
1518         int ret;
1519
1520         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1521         if (!skb)
1522                 return -ENOMEM;
1523
1524         cmd = (struct wmi_sta_powersave_param_cmd *)skb->data;
1525         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
1526                                      WMI_TAG_STA_POWERSAVE_PARAM_CMD) |
1527                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1528
1529         cmd->vdev_id = vdev_id;
1530         cmd->param = param;
1531         cmd->value = param_value;
1532
1533         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_STA_POWERSAVE_PARAM_CMDID);
1534         if (ret) {
1535                 ath11k_warn(ar->ab, "failed to send WMI_STA_POWERSAVE_PARAM_CMDID");
1536                 dev_kfree_skb(skb);
1537         }
1538
1539         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1540                    "cmd set powersave param vdev_id %d param %d value %d\n",
1541                    vdev_id, param, param_value);
1542
1543         return ret;
1544 }
1545
1546 int ath11k_wmi_force_fw_hang_cmd(struct ath11k *ar, u32 type, u32 delay_time_ms)
1547 {
1548         struct ath11k_pdev_wmi *wmi = ar->wmi;
1549         struct wmi_force_fw_hang_cmd *cmd;
1550         struct sk_buff *skb;
1551         int ret, len;
1552
1553         len = sizeof(*cmd);
1554
1555         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
1556         if (!skb)
1557                 return -ENOMEM;
1558
1559         cmd = (struct wmi_force_fw_hang_cmd *)skb->data;
1560         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_FORCE_FW_HANG_CMD) |
1561                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
1562
1563         cmd->type = type;
1564         cmd->delay_time_ms = delay_time_ms;
1565
1566         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_FORCE_FW_HANG_CMDID);
1567
1568         if (ret) {
1569                 ath11k_warn(ar->ab, "Failed to send WMI_FORCE_FW_HANG_CMDID");
1570                 dev_kfree_skb(skb);
1571         }
1572
1573         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd force fw hang");
1574
1575         return ret;
1576 }
1577
1578 int ath11k_wmi_vdev_set_param_cmd(struct ath11k *ar, u32 vdev_id,
1579                                   u32 param_id, u32 param_value)
1580 {
1581         struct ath11k_pdev_wmi *wmi = ar->wmi;
1582         struct wmi_vdev_set_param_cmd *cmd;
1583         struct sk_buff *skb;
1584         int ret;
1585
1586         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1587         if (!skb)
1588                 return -ENOMEM;
1589
1590         cmd = (struct wmi_vdev_set_param_cmd *)skb->data;
1591         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VDEV_SET_PARAM_CMD) |
1592                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1593
1594         cmd->vdev_id = vdev_id;
1595         cmd->param_id = param_id;
1596         cmd->param_value = param_value;
1597
1598         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_VDEV_SET_PARAM_CMDID);
1599         if (ret) {
1600                 ath11k_warn(ar->ab,
1601                             "failed to send WMI_VDEV_SET_PARAM_CMDID\n");
1602                 dev_kfree_skb(skb);
1603         }
1604
1605         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1606                    "cmd vdev set param vdev 0x%x param %d value %d\n",
1607                    vdev_id, param_id, param_value);
1608
1609         return ret;
1610 }
1611
1612 int ath11k_wmi_send_stats_request_cmd(struct ath11k *ar,
1613                                       struct stats_request_params *param)
1614 {
1615         struct ath11k_pdev_wmi *wmi = ar->wmi;
1616         struct wmi_request_stats_cmd *cmd;
1617         struct sk_buff *skb;
1618         int ret;
1619
1620         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1621         if (!skb)
1622                 return -ENOMEM;
1623
1624         cmd = (struct wmi_request_stats_cmd *)skb->data;
1625         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_REQUEST_STATS_CMD) |
1626                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1627
1628         cmd->stats_id = param->stats_id;
1629         cmd->vdev_id = param->vdev_id;
1630         cmd->pdev_id = param->pdev_id;
1631
1632         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_REQUEST_STATS_CMDID);
1633         if (ret) {
1634                 ath11k_warn(ar->ab, "failed to send WMI_REQUEST_STATS cmd\n");
1635                 dev_kfree_skb(skb);
1636         }
1637
1638         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1639                    "cmd request stats 0x%x vdev id %d pdev id %d\n",
1640                    param->stats_id, param->vdev_id, param->pdev_id);
1641
1642         return ret;
1643 }
1644
1645 int ath11k_wmi_send_pdev_temperature_cmd(struct ath11k *ar)
1646 {
1647         struct ath11k_pdev_wmi *wmi = ar->wmi;
1648         struct wmi_get_pdev_temperature_cmd *cmd;
1649         struct sk_buff *skb;
1650         int ret;
1651
1652         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1653         if (!skb)
1654                 return -ENOMEM;
1655
1656         cmd = (struct wmi_get_pdev_temperature_cmd *)skb->data;
1657         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_GET_TEMPERATURE_CMD) |
1658                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1659         cmd->pdev_id = ar->pdev->pdev_id;
1660
1661         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PDEV_GET_TEMPERATURE_CMDID);
1662         if (ret) {
1663                 ath11k_warn(ar->ab, "failed to send WMI_PDEV_GET_TEMPERATURE cmd\n");
1664                 dev_kfree_skb(skb);
1665         }
1666
1667         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1668                    "cmd pdev get temperature for pdev_id %d\n", ar->pdev->pdev_id);
1669
1670         return ret;
1671 }
1672
1673 int ath11k_wmi_send_bcn_offload_control_cmd(struct ath11k *ar,
1674                                             u32 vdev_id, u32 bcn_ctrl_op)
1675 {
1676         struct ath11k_pdev_wmi *wmi = ar->wmi;
1677         struct wmi_bcn_offload_ctrl_cmd *cmd;
1678         struct sk_buff *skb;
1679         int ret;
1680
1681         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
1682         if (!skb)
1683                 return -ENOMEM;
1684
1685         cmd = (struct wmi_bcn_offload_ctrl_cmd *)skb->data;
1686         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
1687                                      WMI_TAG_BCN_OFFLOAD_CTRL_CMD) |
1688                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1689
1690         cmd->vdev_id = vdev_id;
1691         cmd->bcn_ctrl_op = bcn_ctrl_op;
1692
1693         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_BCN_OFFLOAD_CTRL_CMDID);
1694         if (ret) {
1695                 ath11k_warn(ar->ab,
1696                             "failed to send WMI_BCN_OFFLOAD_CTRL_CMDID\n");
1697                 dev_kfree_skb(skb);
1698         }
1699
1700         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1701                    "cmd bcn offload ctrl vdev id %d ctrl_op %d\n",
1702                    vdev_id, bcn_ctrl_op);
1703
1704         return ret;
1705 }
1706
1707 int ath11k_wmi_bcn_tmpl(struct ath11k *ar, u32 vdev_id,
1708                         struct ieee80211_mutable_offsets *offs,
1709                         struct sk_buff *bcn, u32 ema_params)
1710 {
1711         struct ath11k_pdev_wmi *wmi = ar->wmi;
1712         struct wmi_bcn_tmpl_cmd *cmd;
1713         struct wmi_bcn_prb_info *bcn_prb_info;
1714         struct wmi_tlv *tlv;
1715         struct sk_buff *skb;
1716         void *ptr;
1717         int ret, len;
1718         size_t aligned_len = roundup(bcn->len, 4);
1719         struct ieee80211_vif *vif;
1720         struct ath11k_vif *arvif = ath11k_mac_get_arvif(ar, vdev_id);
1721
1722         if (!arvif) {
1723                 ath11k_warn(ar->ab, "failed to find arvif with vdev id %d\n", vdev_id);
1724                 return -EINVAL;
1725         }
1726
1727         vif = arvif->vif;
1728
1729         len = sizeof(*cmd) + sizeof(*bcn_prb_info) + TLV_HDR_SIZE + aligned_len;
1730
1731         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
1732         if (!skb)
1733                 return -ENOMEM;
1734
1735         cmd = (struct wmi_bcn_tmpl_cmd *)skb->data;
1736         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_BCN_TMPL_CMD) |
1737                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1738         cmd->vdev_id = vdev_id;
1739         cmd->tim_ie_offset = offs->tim_offset;
1740
1741         if (vif->bss_conf.csa_active) {
1742                 cmd->csa_switch_count_offset = offs->cntdwn_counter_offs[0];
1743                 cmd->ext_csa_switch_count_offset = offs->cntdwn_counter_offs[1];
1744         }
1745
1746         cmd->buf_len = bcn->len;
1747         cmd->mbssid_ie_offset = offs->mbssid_off;
1748         cmd->ema_params = ema_params;
1749
1750         ptr = skb->data + sizeof(*cmd);
1751
1752         bcn_prb_info = ptr;
1753         len = sizeof(*bcn_prb_info);
1754         bcn_prb_info->tlv_header = FIELD_PREP(WMI_TLV_TAG,
1755                                               WMI_TAG_BCN_PRB_INFO) |
1756                                    FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
1757         bcn_prb_info->caps = 0;
1758         bcn_prb_info->erp = 0;
1759
1760         ptr += sizeof(*bcn_prb_info);
1761
1762         tlv = ptr;
1763         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
1764                       FIELD_PREP(WMI_TLV_LEN, aligned_len);
1765         memcpy(tlv->value, bcn->data, bcn->len);
1766
1767         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_BCN_TMPL_CMDID);
1768         if (ret) {
1769                 ath11k_warn(ar->ab, "failed to send WMI_BCN_TMPL_CMDID\n");
1770                 dev_kfree_skb(skb);
1771         }
1772
1773         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd bcn tmpl");
1774
1775         return ret;
1776 }
1777
1778 int ath11k_wmi_vdev_install_key(struct ath11k *ar,
1779                                 struct wmi_vdev_install_key_arg *arg)
1780 {
1781         struct ath11k_pdev_wmi *wmi = ar->wmi;
1782         struct wmi_vdev_install_key_cmd *cmd;
1783         struct wmi_tlv *tlv;
1784         struct sk_buff *skb;
1785         int ret, len;
1786         int key_len_aligned = roundup(arg->key_len, sizeof(uint32_t));
1787
1788         len = sizeof(*cmd) + TLV_HDR_SIZE + key_len_aligned;
1789
1790         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
1791         if (!skb)
1792                 return -ENOMEM;
1793
1794         cmd = (struct wmi_vdev_install_key_cmd *)skb->data;
1795         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VDEV_INSTALL_KEY_CMD) |
1796                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1797         cmd->vdev_id = arg->vdev_id;
1798         ether_addr_copy(cmd->peer_macaddr.addr, arg->macaddr);
1799         cmd->key_idx = arg->key_idx;
1800         cmd->key_flags = arg->key_flags;
1801         cmd->key_cipher = arg->key_cipher;
1802         cmd->key_len = arg->key_len;
1803         cmd->key_txmic_len = arg->key_txmic_len;
1804         cmd->key_rxmic_len = arg->key_rxmic_len;
1805
1806         if (arg->key_rsc_counter)
1807                 memcpy(&cmd->key_rsc_counter, &arg->key_rsc_counter,
1808                        sizeof(struct wmi_key_seq_counter));
1809
1810         tlv = (struct wmi_tlv *)(skb->data + sizeof(*cmd));
1811         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
1812                       FIELD_PREP(WMI_TLV_LEN, key_len_aligned);
1813         if (arg->key_data)
1814                 memcpy(tlv->value, (u8 *)arg->key_data, key_len_aligned);
1815
1816         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_VDEV_INSTALL_KEY_CMDID);
1817         if (ret) {
1818                 ath11k_warn(ar->ab,
1819                             "failed to send WMI_VDEV_INSTALL_KEY cmd\n");
1820                 dev_kfree_skb(skb);
1821         }
1822
1823         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
1824                    "cmd vdev install key idx %d cipher %d len %d\n",
1825                    arg->key_idx, arg->key_cipher, arg->key_len);
1826
1827         return ret;
1828 }
1829
1830 static inline void
1831 ath11k_wmi_copy_peer_flags(struct wmi_peer_assoc_complete_cmd *cmd,
1832                            struct peer_assoc_params *param,
1833                            bool hw_crypto_disabled)
1834 {
1835         cmd->peer_flags = 0;
1836
1837         if (param->is_wme_set) {
1838                 if (param->qos_flag)
1839                         cmd->peer_flags |= WMI_PEER_QOS;
1840                 if (param->apsd_flag)
1841                         cmd->peer_flags |= WMI_PEER_APSD;
1842                 if (param->ht_flag)
1843                         cmd->peer_flags |= WMI_PEER_HT;
1844                 if (param->bw_40)
1845                         cmd->peer_flags |= WMI_PEER_40MHZ;
1846                 if (param->bw_80)
1847                         cmd->peer_flags |= WMI_PEER_80MHZ;
1848                 if (param->bw_160)
1849                         cmd->peer_flags |= WMI_PEER_160MHZ;
1850
1851                 /* Typically if STBC is enabled for VHT it should be enabled
1852                  * for HT as well
1853                  **/
1854                 if (param->stbc_flag)
1855                         cmd->peer_flags |= WMI_PEER_STBC;
1856
1857                 /* Typically if LDPC is enabled for VHT it should be enabled
1858                  * for HT as well
1859                  **/
1860                 if (param->ldpc_flag)
1861                         cmd->peer_flags |= WMI_PEER_LDPC;
1862
1863                 if (param->static_mimops_flag)
1864                         cmd->peer_flags |= WMI_PEER_STATIC_MIMOPS;
1865                 if (param->dynamic_mimops_flag)
1866                         cmd->peer_flags |= WMI_PEER_DYN_MIMOPS;
1867                 if (param->spatial_mux_flag)
1868                         cmd->peer_flags |= WMI_PEER_SPATIAL_MUX;
1869                 if (param->vht_flag)
1870                         cmd->peer_flags |= WMI_PEER_VHT;
1871                 if (param->he_flag)
1872                         cmd->peer_flags |= WMI_PEER_HE;
1873                 if (param->twt_requester)
1874                         cmd->peer_flags |= WMI_PEER_TWT_REQ;
1875                 if (param->twt_responder)
1876                         cmd->peer_flags |= WMI_PEER_TWT_RESP;
1877         }
1878
1879         /* Suppress authorization for all AUTH modes that need 4-way handshake
1880          * (during re-association).
1881          * Authorization will be done for these modes on key installation.
1882          */
1883         if (param->auth_flag)
1884                 cmd->peer_flags |= WMI_PEER_AUTH;
1885         if (param->need_ptk_4_way) {
1886                 cmd->peer_flags |= WMI_PEER_NEED_PTK_4_WAY;
1887                 if (!hw_crypto_disabled && param->is_assoc)
1888                         cmd->peer_flags &= ~WMI_PEER_AUTH;
1889         }
1890         if (param->need_gtk_2_way)
1891                 cmd->peer_flags |= WMI_PEER_NEED_GTK_2_WAY;
1892         /* safe mode bypass the 4-way handshake */
1893         if (param->safe_mode_enabled)
1894                 cmd->peer_flags &= ~(WMI_PEER_NEED_PTK_4_WAY |
1895                                      WMI_PEER_NEED_GTK_2_WAY);
1896
1897         if (param->is_pmf_enabled)
1898                 cmd->peer_flags |= WMI_PEER_PMF;
1899
1900         /* Disable AMSDU for station transmit, if user configures it */
1901         /* Disable AMSDU for AP transmit to 11n Stations, if user configures
1902          * it
1903          * if (param->amsdu_disable) Add after FW support
1904          **/
1905
1906         /* Target asserts if node is marked HT and all MCS is set to 0.
1907          * Mark the node as non-HT if all the mcs rates are disabled through
1908          * iwpriv
1909          **/
1910         if (param->peer_ht_rates.num_rates == 0)
1911                 cmd->peer_flags &= ~WMI_PEER_HT;
1912 }
1913
1914 int ath11k_wmi_send_peer_assoc_cmd(struct ath11k *ar,
1915                                    struct peer_assoc_params *param)
1916 {
1917         struct ath11k_pdev_wmi *wmi = ar->wmi;
1918         struct wmi_peer_assoc_complete_cmd *cmd;
1919         struct wmi_vht_rate_set *mcs;
1920         struct wmi_he_rate_set *he_mcs;
1921         struct sk_buff *skb;
1922         struct wmi_tlv *tlv;
1923         void *ptr;
1924         u32 peer_legacy_rates_align;
1925         u32 peer_ht_rates_align;
1926         int i, ret, len;
1927
1928         peer_legacy_rates_align = roundup(param->peer_legacy_rates.num_rates,
1929                                           sizeof(u32));
1930         peer_ht_rates_align = roundup(param->peer_ht_rates.num_rates,
1931                                       sizeof(u32));
1932
1933         len = sizeof(*cmd) +
1934               TLV_HDR_SIZE + (peer_legacy_rates_align * sizeof(u8)) +
1935               TLV_HDR_SIZE + (peer_ht_rates_align * sizeof(u8)) +
1936               sizeof(*mcs) + TLV_HDR_SIZE +
1937               (sizeof(*he_mcs) * param->peer_he_mcs_count);
1938
1939         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
1940         if (!skb)
1941                 return -ENOMEM;
1942
1943         ptr = skb->data;
1944
1945         cmd = ptr;
1946         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
1947                                      WMI_TAG_PEER_ASSOC_COMPLETE_CMD) |
1948                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
1949
1950         cmd->vdev_id = param->vdev_id;
1951
1952         cmd->peer_new_assoc = param->peer_new_assoc;
1953         cmd->peer_associd = param->peer_associd;
1954
1955         ath11k_wmi_copy_peer_flags(cmd, param,
1956                                    test_bit(ATH11K_FLAG_HW_CRYPTO_DISABLED,
1957                                             &ar->ab->dev_flags));
1958
1959         ether_addr_copy(cmd->peer_macaddr.addr, param->peer_mac);
1960
1961         cmd->peer_rate_caps = param->peer_rate_caps;
1962         cmd->peer_caps = param->peer_caps;
1963         cmd->peer_listen_intval = param->peer_listen_intval;
1964         cmd->peer_ht_caps = param->peer_ht_caps;
1965         cmd->peer_max_mpdu = param->peer_max_mpdu;
1966         cmd->peer_mpdu_density = param->peer_mpdu_density;
1967         cmd->peer_vht_caps = param->peer_vht_caps;
1968         cmd->peer_phymode = param->peer_phymode;
1969
1970         /* Update 11ax capabilities */
1971         cmd->peer_he_cap_info = param->peer_he_cap_macinfo[0];
1972         cmd->peer_he_cap_info_ext = param->peer_he_cap_macinfo[1];
1973         cmd->peer_he_cap_info_internal = param->peer_he_cap_macinfo_internal;
1974         cmd->peer_he_caps_6ghz = param->peer_he_caps_6ghz;
1975         cmd->peer_he_ops = param->peer_he_ops;
1976         memcpy(&cmd->peer_he_cap_phy, &param->peer_he_cap_phyinfo,
1977                sizeof(param->peer_he_cap_phyinfo));
1978         memcpy(&cmd->peer_ppet, &param->peer_ppet,
1979                sizeof(param->peer_ppet));
1980
1981         /* Update peer legacy rate information */
1982         ptr += sizeof(*cmd);
1983
1984         tlv = ptr;
1985         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
1986                       FIELD_PREP(WMI_TLV_LEN, peer_legacy_rates_align);
1987
1988         ptr += TLV_HDR_SIZE;
1989
1990         cmd->num_peer_legacy_rates = param->peer_legacy_rates.num_rates;
1991         memcpy(ptr, param->peer_legacy_rates.rates,
1992                param->peer_legacy_rates.num_rates);
1993
1994         /* Update peer HT rate information */
1995         ptr += peer_legacy_rates_align;
1996
1997         tlv = ptr;
1998         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
1999                       FIELD_PREP(WMI_TLV_LEN, peer_ht_rates_align);
2000         ptr += TLV_HDR_SIZE;
2001         cmd->num_peer_ht_rates = param->peer_ht_rates.num_rates;
2002         memcpy(ptr, param->peer_ht_rates.rates,
2003                param->peer_ht_rates.num_rates);
2004
2005         /* VHT Rates */
2006         ptr += peer_ht_rates_align;
2007
2008         mcs = ptr;
2009
2010         mcs->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_VHT_RATE_SET) |
2011                           FIELD_PREP(WMI_TLV_LEN, sizeof(*mcs) - TLV_HDR_SIZE);
2012
2013         cmd->peer_nss = param->peer_nss;
2014
2015         /* Update bandwidth-NSS mapping */
2016         cmd->peer_bw_rxnss_override = 0;
2017         cmd->peer_bw_rxnss_override |= param->peer_bw_rxnss_override;
2018
2019         if (param->vht_capable) {
2020                 mcs->rx_max_rate = param->rx_max_rate;
2021                 mcs->rx_mcs_set = param->rx_mcs_set;
2022                 mcs->tx_max_rate = param->tx_max_rate;
2023                 mcs->tx_mcs_set = param->tx_mcs_set;
2024         }
2025
2026         /* HE Rates */
2027         cmd->peer_he_mcs = param->peer_he_mcs_count;
2028         cmd->min_data_rate = param->min_data_rate;
2029
2030         ptr += sizeof(*mcs);
2031
2032         len = param->peer_he_mcs_count * sizeof(*he_mcs);
2033
2034         tlv = ptr;
2035         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
2036                       FIELD_PREP(WMI_TLV_LEN, len);
2037         ptr += TLV_HDR_SIZE;
2038
2039         /* Loop through the HE rate set */
2040         for (i = 0; i < param->peer_he_mcs_count; i++) {
2041                 he_mcs = ptr;
2042                 he_mcs->tlv_header = FIELD_PREP(WMI_TLV_TAG,
2043                                                 WMI_TAG_HE_RATE_SET) |
2044                                      FIELD_PREP(WMI_TLV_LEN,
2045                                                 sizeof(*he_mcs) - TLV_HDR_SIZE);
2046
2047                 he_mcs->rx_mcs_set = param->peer_he_tx_mcs_set[i];
2048                 he_mcs->tx_mcs_set = param->peer_he_rx_mcs_set[i];
2049                 ptr += sizeof(*he_mcs);
2050         }
2051
2052         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_PEER_ASSOC_CMDID);
2053         if (ret) {
2054                 ath11k_warn(ar->ab,
2055                             "failed to send WMI_PEER_ASSOC_CMDID\n");
2056                 dev_kfree_skb(skb);
2057         }
2058
2059         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2060                    "cmd peer assoc vdev id %d assoc id %d peer mac %pM peer_flags %x rate_caps %x peer_caps %x listen_intval %d ht_caps %x max_mpdu %d nss %d phymode %d peer_mpdu_density %d vht_caps %x he cap_info %x he ops %x he cap_info_ext %x he phy %x %x %x peer_bw_rxnss_override %x\n",
2061                    cmd->vdev_id, cmd->peer_associd, param->peer_mac,
2062                    cmd->peer_flags, cmd->peer_rate_caps, cmd->peer_caps,
2063                    cmd->peer_listen_intval, cmd->peer_ht_caps,
2064                    cmd->peer_max_mpdu, cmd->peer_nss, cmd->peer_phymode,
2065                    cmd->peer_mpdu_density,
2066                    cmd->peer_vht_caps, cmd->peer_he_cap_info,
2067                    cmd->peer_he_ops, cmd->peer_he_cap_info_ext,
2068                    cmd->peer_he_cap_phy[0], cmd->peer_he_cap_phy[1],
2069                    cmd->peer_he_cap_phy[2],
2070                    cmd->peer_bw_rxnss_override);
2071
2072         return ret;
2073 }
2074
2075 void ath11k_wmi_start_scan_init(struct ath11k *ar,
2076                                 struct scan_req_params *arg)
2077 {
2078         /* setup commonly used values */
2079         arg->scan_req_id = 1;
2080         if (ar->state_11d == ATH11K_11D_PREPARING)
2081                 arg->scan_priority = WMI_SCAN_PRIORITY_MEDIUM;
2082         else
2083                 arg->scan_priority = WMI_SCAN_PRIORITY_LOW;
2084         arg->dwell_time_active = 50;
2085         arg->dwell_time_active_2g = 0;
2086         arg->dwell_time_passive = 150;
2087         arg->dwell_time_active_6g = 40;
2088         arg->dwell_time_passive_6g = 30;
2089         arg->min_rest_time = 50;
2090         arg->max_rest_time = 500;
2091         arg->repeat_probe_time = 0;
2092         arg->probe_spacing_time = 0;
2093         arg->idle_time = 0;
2094         arg->max_scan_time = 20000;
2095         arg->probe_delay = 5;
2096         arg->notify_scan_events = WMI_SCAN_EVENT_STARTED |
2097                                   WMI_SCAN_EVENT_COMPLETED |
2098                                   WMI_SCAN_EVENT_BSS_CHANNEL |
2099                                   WMI_SCAN_EVENT_FOREIGN_CHAN |
2100                                   WMI_SCAN_EVENT_DEQUEUED;
2101         arg->scan_flags |= WMI_SCAN_CHAN_STAT_EVENT;
2102
2103         if (test_bit(WMI_TLV_SERVICE_PASSIVE_SCAN_START_TIME_ENHANCE,
2104                      ar->ab->wmi_ab.svc_map))
2105                 arg->scan_ctrl_flags_ext |=
2106                         WMI_SCAN_FLAG_EXT_PASSIVE_SCAN_START_TIME_ENHANCE;
2107
2108         arg->num_bssid = 1;
2109
2110         /* fill bssid_list[0] with 0xff, otherwise bssid and RA will be
2111          * ZEROs in probe request
2112          */
2113         eth_broadcast_addr(arg->bssid_list[0].addr);
2114 }
2115
2116 static inline void
2117 ath11k_wmi_copy_scan_event_cntrl_flags(struct wmi_start_scan_cmd *cmd,
2118                                        struct scan_req_params *param)
2119 {
2120         /* Scan events subscription */
2121         if (param->scan_ev_started)
2122                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_STARTED;
2123         if (param->scan_ev_completed)
2124                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_COMPLETED;
2125         if (param->scan_ev_bss_chan)
2126                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_BSS_CHANNEL;
2127         if (param->scan_ev_foreign_chan)
2128                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_FOREIGN_CHAN;
2129         if (param->scan_ev_dequeued)
2130                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_DEQUEUED;
2131         if (param->scan_ev_preempted)
2132                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_PREEMPTED;
2133         if (param->scan_ev_start_failed)
2134                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_START_FAILED;
2135         if (param->scan_ev_restarted)
2136                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_RESTARTED;
2137         if (param->scan_ev_foreign_chn_exit)
2138                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_FOREIGN_CHAN_EXIT;
2139         if (param->scan_ev_suspended)
2140                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_SUSPENDED;
2141         if (param->scan_ev_resumed)
2142                 cmd->notify_scan_events |=  WMI_SCAN_EVENT_RESUMED;
2143
2144         /** Set scan control flags */
2145         cmd->scan_ctrl_flags = 0;
2146         if (param->scan_f_passive)
2147                 cmd->scan_ctrl_flags |=  WMI_SCAN_FLAG_PASSIVE;
2148         if (param->scan_f_strict_passive_pch)
2149                 cmd->scan_ctrl_flags |=  WMI_SCAN_FLAG_STRICT_PASSIVE_ON_PCHN;
2150         if (param->scan_f_promisc_mode)
2151                 cmd->scan_ctrl_flags |=  WMI_SCAN_FILTER_PROMISCUOS;
2152         if (param->scan_f_capture_phy_err)
2153                 cmd->scan_ctrl_flags |=  WMI_SCAN_CAPTURE_PHY_ERROR;
2154         if (param->scan_f_half_rate)
2155                 cmd->scan_ctrl_flags |=  WMI_SCAN_FLAG_HALF_RATE_SUPPORT;
2156         if (param->scan_f_quarter_rate)
2157                 cmd->scan_ctrl_flags |=  WMI_SCAN_FLAG_QUARTER_RATE_SUPPORT;
2158         if (param->scan_f_cck_rates)
2159                 cmd->scan_ctrl_flags |=  WMI_SCAN_ADD_CCK_RATES;
2160         if (param->scan_f_ofdm_rates)
2161                 cmd->scan_ctrl_flags |=  WMI_SCAN_ADD_OFDM_RATES;
2162         if (param->scan_f_chan_stat_evnt)
2163                 cmd->scan_ctrl_flags |=  WMI_SCAN_CHAN_STAT_EVENT;
2164         if (param->scan_f_filter_prb_req)
2165                 cmd->scan_ctrl_flags |=  WMI_SCAN_FILTER_PROBE_REQ;
2166         if (param->scan_f_bcast_probe)
2167                 cmd->scan_ctrl_flags |=  WMI_SCAN_ADD_BCAST_PROBE_REQ;
2168         if (param->scan_f_offchan_mgmt_tx)
2169                 cmd->scan_ctrl_flags |=  WMI_SCAN_OFFCHAN_MGMT_TX;
2170         if (param->scan_f_offchan_data_tx)
2171                 cmd->scan_ctrl_flags |=  WMI_SCAN_OFFCHAN_DATA_TX;
2172         if (param->scan_f_force_active_dfs_chn)
2173                 cmd->scan_ctrl_flags |=  WMI_SCAN_FLAG_FORCE_ACTIVE_ON_DFS;
2174         if (param->scan_f_add_tpc_ie_in_probe)
2175                 cmd->scan_ctrl_flags |=  WMI_SCAN_ADD_TPC_IE_IN_PROBE_REQ;
2176         if (param->scan_f_add_ds_ie_in_probe)
2177                 cmd->scan_ctrl_flags |=  WMI_SCAN_ADD_DS_IE_IN_PROBE_REQ;
2178         if (param->scan_f_add_spoofed_mac_in_probe)
2179                 cmd->scan_ctrl_flags |=  WMI_SCAN_ADD_SPOOF_MAC_IN_PROBE_REQ;
2180         if (param->scan_f_add_rand_seq_in_probe)
2181                 cmd->scan_ctrl_flags |=  WMI_SCAN_RANDOM_SEQ_NO_IN_PROBE_REQ;
2182         if (param->scan_f_en_ie_whitelist_in_probe)
2183                 cmd->scan_ctrl_flags |=
2184                          WMI_SCAN_ENABLE_IE_WHTELIST_IN_PROBE_REQ;
2185
2186         /* for adaptive scan mode using 3 bits (21 - 23 bits) */
2187         WMI_SCAN_SET_DWELL_MODE(cmd->scan_ctrl_flags,
2188                                 param->adaptive_dwell_time_mode);
2189
2190         cmd->scan_ctrl_flags_ext = param->scan_ctrl_flags_ext;
2191 }
2192
2193 int ath11k_wmi_send_scan_start_cmd(struct ath11k *ar,
2194                                    struct scan_req_params *params)
2195 {
2196         struct ath11k_pdev_wmi *wmi = ar->wmi;
2197         struct wmi_start_scan_cmd *cmd;
2198         struct wmi_ssid *ssid = NULL;
2199         struct wmi_mac_addr *bssid;
2200         struct sk_buff *skb;
2201         struct wmi_tlv *tlv;
2202         void *ptr;
2203         int i, ret, len;
2204         u32 *tmp_ptr;
2205         u16 extraie_len_with_pad = 0;
2206         struct hint_short_ssid *s_ssid = NULL;
2207         struct hint_bssid *hint_bssid = NULL;
2208
2209         len = sizeof(*cmd);
2210
2211         len += TLV_HDR_SIZE;
2212         if (params->num_chan)
2213                 len += params->num_chan * sizeof(u32);
2214
2215         len += TLV_HDR_SIZE;
2216         if (params->num_ssids)
2217                 len += params->num_ssids * sizeof(*ssid);
2218
2219         len += TLV_HDR_SIZE;
2220         if (params->num_bssid)
2221                 len += sizeof(*bssid) * params->num_bssid;
2222
2223         len += TLV_HDR_SIZE;
2224         if (params->extraie.len && params->extraie.len <= 0xFFFF)
2225                 extraie_len_with_pad =
2226                         roundup(params->extraie.len, sizeof(u32));
2227         len += extraie_len_with_pad;
2228
2229         if (params->num_hint_bssid)
2230                 len += TLV_HDR_SIZE +
2231                        params->num_hint_bssid * sizeof(struct hint_bssid);
2232
2233         if (params->num_hint_s_ssid)
2234                 len += TLV_HDR_SIZE +
2235                        params->num_hint_s_ssid * sizeof(struct hint_short_ssid);
2236
2237         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
2238         if (!skb)
2239                 return -ENOMEM;
2240
2241         ptr = skb->data;
2242
2243         cmd = ptr;
2244         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_START_SCAN_CMD) |
2245                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2246
2247         cmd->scan_id = params->scan_id;
2248         cmd->scan_req_id = params->scan_req_id;
2249         cmd->vdev_id = params->vdev_id;
2250         cmd->scan_priority = params->scan_priority;
2251         cmd->notify_scan_events = params->notify_scan_events;
2252
2253         ath11k_wmi_copy_scan_event_cntrl_flags(cmd, params);
2254
2255         cmd->dwell_time_active = params->dwell_time_active;
2256         cmd->dwell_time_active_2g = params->dwell_time_active_2g;
2257         cmd->dwell_time_passive = params->dwell_time_passive;
2258         cmd->dwell_time_active_6g = params->dwell_time_active_6g;
2259         cmd->dwell_time_passive_6g = params->dwell_time_passive_6g;
2260         cmd->min_rest_time = params->min_rest_time;
2261         cmd->max_rest_time = params->max_rest_time;
2262         cmd->repeat_probe_time = params->repeat_probe_time;
2263         cmd->probe_spacing_time = params->probe_spacing_time;
2264         cmd->idle_time = params->idle_time;
2265         cmd->max_scan_time = params->max_scan_time;
2266         cmd->probe_delay = params->probe_delay;
2267         cmd->burst_duration = params->burst_duration;
2268         cmd->num_chan = params->num_chan;
2269         cmd->num_bssid = params->num_bssid;
2270         cmd->num_ssids = params->num_ssids;
2271         cmd->ie_len = params->extraie.len;
2272         cmd->n_probes = params->n_probes;
2273         ether_addr_copy(cmd->mac_addr.addr, params->mac_addr.addr);
2274         ether_addr_copy(cmd->mac_mask.addr, params->mac_mask.addr);
2275
2276         ptr += sizeof(*cmd);
2277
2278         len = params->num_chan * sizeof(u32);
2279
2280         tlv = ptr;
2281         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_UINT32) |
2282                       FIELD_PREP(WMI_TLV_LEN, len);
2283         ptr += TLV_HDR_SIZE;
2284         tmp_ptr = ptr;
2285
2286         for (i = 0; i < params->num_chan; ++i)
2287                 tmp_ptr[i] = params->chan_list[i];
2288
2289         ptr += len;
2290
2291         len = params->num_ssids * sizeof(*ssid);
2292         tlv = ptr;
2293         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_FIXED_STRUCT) |
2294                       FIELD_PREP(WMI_TLV_LEN, len);
2295
2296         ptr += TLV_HDR_SIZE;
2297
2298         if (params->num_ssids) {
2299                 ssid = ptr;
2300                 for (i = 0; i < params->num_ssids; ++i) {
2301                         ssid->ssid_len = params->ssid[i].length;
2302                         memcpy(ssid->ssid, params->ssid[i].ssid,
2303                                params->ssid[i].length);
2304                         ssid++;
2305                 }
2306         }
2307
2308         ptr += (params->num_ssids * sizeof(*ssid));
2309         len = params->num_bssid * sizeof(*bssid);
2310         tlv = ptr;
2311         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_FIXED_STRUCT) |
2312                       FIELD_PREP(WMI_TLV_LEN, len);
2313
2314         ptr += TLV_HDR_SIZE;
2315         bssid = ptr;
2316
2317         if (params->num_bssid) {
2318                 for (i = 0; i < params->num_bssid; ++i) {
2319                         ether_addr_copy(bssid->addr,
2320                                         params->bssid_list[i].addr);
2321                         bssid++;
2322                 }
2323         }
2324
2325         ptr += params->num_bssid * sizeof(*bssid);
2326
2327         len = extraie_len_with_pad;
2328         tlv = ptr;
2329         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
2330                       FIELD_PREP(WMI_TLV_LEN, len);
2331         ptr += TLV_HDR_SIZE;
2332
2333         if (extraie_len_with_pad)
2334                 memcpy(ptr, params->extraie.ptr,
2335                        params->extraie.len);
2336
2337         ptr += extraie_len_with_pad;
2338
2339         if (params->num_hint_s_ssid) {
2340                 len = params->num_hint_s_ssid * sizeof(struct hint_short_ssid);
2341                 tlv = ptr;
2342                 tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_FIXED_STRUCT) |
2343                               FIELD_PREP(WMI_TLV_LEN, len);
2344                 ptr += TLV_HDR_SIZE;
2345                 s_ssid = ptr;
2346                 for (i = 0; i < params->num_hint_s_ssid; ++i) {
2347                         s_ssid->freq_flags = params->hint_s_ssid[i].freq_flags;
2348                         s_ssid->short_ssid = params->hint_s_ssid[i].short_ssid;
2349                         s_ssid++;
2350                 }
2351                 ptr += len;
2352         }
2353
2354         if (params->num_hint_bssid) {
2355                 len = params->num_hint_bssid * sizeof(struct hint_bssid);
2356                 tlv = ptr;
2357                 tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_FIXED_STRUCT) |
2358                               FIELD_PREP(WMI_TLV_LEN, len);
2359                 ptr += TLV_HDR_SIZE;
2360                 hint_bssid = ptr;
2361                 for (i = 0; i < params->num_hint_bssid; ++i) {
2362                         hint_bssid->freq_flags =
2363                                 params->hint_bssid[i].freq_flags;
2364                         ether_addr_copy(&params->hint_bssid[i].bssid.addr[0],
2365                                         &hint_bssid->bssid.addr[0]);
2366                         hint_bssid++;
2367                 }
2368         }
2369
2370         ret = ath11k_wmi_cmd_send(wmi, skb,
2371                                   WMI_START_SCAN_CMDID);
2372         if (ret) {
2373                 ath11k_warn(ar->ab, "failed to send WMI_START_SCAN_CMDID\n");
2374                 dev_kfree_skb(skb);
2375         }
2376
2377         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd start scan");
2378
2379         return ret;
2380 }
2381
2382 int ath11k_wmi_send_scan_stop_cmd(struct ath11k *ar,
2383                                   struct scan_cancel_param *param)
2384 {
2385         struct ath11k_pdev_wmi *wmi = ar->wmi;
2386         struct wmi_stop_scan_cmd *cmd;
2387         struct sk_buff *skb;
2388         int ret;
2389
2390         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
2391         if (!skb)
2392                 return -ENOMEM;
2393
2394         cmd = (struct wmi_stop_scan_cmd *)skb->data;
2395
2396         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_STOP_SCAN_CMD) |
2397                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2398
2399         cmd->vdev_id = param->vdev_id;
2400         cmd->requestor = param->requester;
2401         cmd->scan_id = param->scan_id;
2402         cmd->pdev_id = param->pdev_id;
2403         /* stop the scan with the corresponding scan_id */
2404         if (param->req_type == WLAN_SCAN_CANCEL_PDEV_ALL) {
2405                 /* Cancelling all scans */
2406                 cmd->req_type =  WMI_SCAN_STOP_ALL;
2407         } else if (param->req_type == WLAN_SCAN_CANCEL_VDEV_ALL) {
2408                 /* Cancelling VAP scans */
2409                 cmd->req_type =  WMI_SCN_STOP_VAP_ALL;
2410         } else if (param->req_type == WLAN_SCAN_CANCEL_SINGLE) {
2411                 /* Cancelling specific scan */
2412                 cmd->req_type =  WMI_SCAN_STOP_ONE;
2413         } else {
2414                 ath11k_warn(ar->ab, "invalid scan cancel param %d",
2415                             param->req_type);
2416                 dev_kfree_skb(skb);
2417                 return -EINVAL;
2418         }
2419
2420         ret = ath11k_wmi_cmd_send(wmi, skb,
2421                                   WMI_STOP_SCAN_CMDID);
2422         if (ret) {
2423                 ath11k_warn(ar->ab, "failed to send WMI_STOP_SCAN_CMDID\n");
2424                 dev_kfree_skb(skb);
2425         }
2426
2427         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd stop scan");
2428
2429         return ret;
2430 }
2431
2432 int ath11k_wmi_send_scan_chan_list_cmd(struct ath11k *ar,
2433                                        struct scan_chan_list_params *chan_list)
2434 {
2435         struct ath11k_pdev_wmi *wmi = ar->wmi;
2436         struct wmi_scan_chan_list_cmd *cmd;
2437         struct sk_buff *skb;
2438         struct wmi_channel *chan_info;
2439         struct channel_param *tchan_info;
2440         struct wmi_tlv *tlv;
2441         void *ptr;
2442         int i, ret, len;
2443         u16 num_send_chans, num_sends = 0, max_chan_limit = 0;
2444         u32 *reg1, *reg2;
2445
2446         tchan_info = chan_list->ch_param;
2447         while (chan_list->nallchans) {
2448                 len = sizeof(*cmd) + TLV_HDR_SIZE;
2449                 max_chan_limit = (wmi->wmi_ab->max_msg_len[ar->pdev_idx] - len) /
2450                         sizeof(*chan_info);
2451
2452                 if (chan_list->nallchans > max_chan_limit)
2453                         num_send_chans = max_chan_limit;
2454                 else
2455                         num_send_chans = chan_list->nallchans;
2456
2457                 chan_list->nallchans -= num_send_chans;
2458                 len += sizeof(*chan_info) * num_send_chans;
2459
2460                 skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
2461                 if (!skb)
2462                         return -ENOMEM;
2463
2464                 cmd = (struct wmi_scan_chan_list_cmd *)skb->data;
2465                 cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_SCAN_CHAN_LIST_CMD) |
2466                         FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2467                 cmd->pdev_id = chan_list->pdev_id;
2468                 cmd->num_scan_chans = num_send_chans;
2469                 if (num_sends)
2470                         cmd->flags |= WMI_APPEND_TO_EXISTING_CHAN_LIST_FLAG;
2471
2472                 ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2473                            "no.of chan = %d len = %d pdev_id = %d num_sends = %d\n",
2474                            num_send_chans, len, cmd->pdev_id, num_sends);
2475
2476                 ptr = skb->data + sizeof(*cmd);
2477
2478                 len = sizeof(*chan_info) * num_send_chans;
2479                 tlv = ptr;
2480                 tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
2481                               FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
2482                 ptr += TLV_HDR_SIZE;
2483
2484                 for (i = 0; i < num_send_chans; ++i) {
2485                         chan_info = ptr;
2486                         memset(chan_info, 0, sizeof(*chan_info));
2487                         len = sizeof(*chan_info);
2488                         chan_info->tlv_header = FIELD_PREP(WMI_TLV_TAG,
2489                                                            WMI_TAG_CHANNEL) |
2490                                                 FIELD_PREP(WMI_TLV_LEN,
2491                                                            len - TLV_HDR_SIZE);
2492
2493                         reg1 = &chan_info->reg_info_1;
2494                         reg2 = &chan_info->reg_info_2;
2495                         chan_info->mhz = tchan_info->mhz;
2496                         chan_info->band_center_freq1 = tchan_info->cfreq1;
2497                         chan_info->band_center_freq2 = tchan_info->cfreq2;
2498
2499                         if (tchan_info->is_chan_passive)
2500                                 chan_info->info |= WMI_CHAN_INFO_PASSIVE;
2501                         if (tchan_info->allow_he)
2502                                 chan_info->info |= WMI_CHAN_INFO_ALLOW_HE;
2503                         else if (tchan_info->allow_vht)
2504                                 chan_info->info |= WMI_CHAN_INFO_ALLOW_VHT;
2505                         else if (tchan_info->allow_ht)
2506                                 chan_info->info |= WMI_CHAN_INFO_ALLOW_HT;
2507                         if (tchan_info->half_rate)
2508                                 chan_info->info |= WMI_CHAN_INFO_HALF_RATE;
2509                         if (tchan_info->quarter_rate)
2510                                 chan_info->info |= WMI_CHAN_INFO_QUARTER_RATE;
2511                         if (tchan_info->psc_channel)
2512                                 chan_info->info |= WMI_CHAN_INFO_PSC;
2513                         if (tchan_info->dfs_set)
2514                                 chan_info->info |= WMI_CHAN_INFO_DFS;
2515
2516                         chan_info->info |= FIELD_PREP(WMI_CHAN_INFO_MODE,
2517                                                       tchan_info->phy_mode);
2518                         *reg1 |= FIELD_PREP(WMI_CHAN_REG_INFO1_MIN_PWR,
2519                                             tchan_info->minpower);
2520                         *reg1 |= FIELD_PREP(WMI_CHAN_REG_INFO1_MAX_PWR,
2521                                             tchan_info->maxpower);
2522                         *reg1 |= FIELD_PREP(WMI_CHAN_REG_INFO1_MAX_REG_PWR,
2523                                             tchan_info->maxregpower);
2524                         *reg1 |= FIELD_PREP(WMI_CHAN_REG_INFO1_REG_CLS,
2525                                             tchan_info->reg_class_id);
2526                         *reg2 |= FIELD_PREP(WMI_CHAN_REG_INFO2_ANT_MAX,
2527                                             tchan_info->antennamax);
2528                         *reg2 |= FIELD_PREP(WMI_CHAN_REG_INFO2_MAX_TX_PWR,
2529                                             tchan_info->maxregpower);
2530
2531                         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2532                                    "chan scan list chan[%d] = %u, chan_info->info %8x\n",
2533                                    i, chan_info->mhz, chan_info->info);
2534
2535                         ptr += sizeof(*chan_info);
2536
2537                         tchan_info++;
2538                 }
2539
2540                 ret = ath11k_wmi_cmd_send(wmi, skb, WMI_SCAN_CHAN_LIST_CMDID);
2541                 if (ret) {
2542                         ath11k_warn(ar->ab, "failed to send WMI_SCAN_CHAN_LIST cmd\n");
2543                         dev_kfree_skb(skb);
2544                         return ret;
2545                 }
2546
2547                 ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd scan chan list channels %d",
2548                            num_send_chans);
2549
2550                 num_sends++;
2551         }
2552
2553         return 0;
2554 }
2555
2556 int ath11k_wmi_send_wmm_update_cmd_tlv(struct ath11k *ar, u32 vdev_id,
2557                                        struct wmi_wmm_params_all_arg *param)
2558 {
2559         struct ath11k_pdev_wmi *wmi = ar->wmi;
2560         struct wmi_vdev_set_wmm_params_cmd *cmd;
2561         struct wmi_wmm_params *wmm_param;
2562         struct wmi_wmm_params_arg *wmi_wmm_arg;
2563         struct sk_buff *skb;
2564         int ret, ac;
2565
2566         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
2567         if (!skb)
2568                 return -ENOMEM;
2569
2570         cmd = (struct wmi_vdev_set_wmm_params_cmd *)skb->data;
2571         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
2572                                      WMI_TAG_VDEV_SET_WMM_PARAMS_CMD) |
2573                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2574
2575         cmd->vdev_id = vdev_id;
2576         cmd->wmm_param_type = 0;
2577
2578         for (ac = 0; ac < WME_NUM_AC; ac++) {
2579                 switch (ac) {
2580                 case WME_AC_BE:
2581                         wmi_wmm_arg = &param->ac_be;
2582                         break;
2583                 case WME_AC_BK:
2584                         wmi_wmm_arg = &param->ac_bk;
2585                         break;
2586                 case WME_AC_VI:
2587                         wmi_wmm_arg = &param->ac_vi;
2588                         break;
2589                 case WME_AC_VO:
2590                         wmi_wmm_arg = &param->ac_vo;
2591                         break;
2592                 }
2593
2594                 wmm_param = (struct wmi_wmm_params *)&cmd->wmm_params[ac];
2595                 wmm_param->tlv_header =
2596                                 FIELD_PREP(WMI_TLV_TAG,
2597                                            WMI_TAG_VDEV_SET_WMM_PARAMS_CMD) |
2598                                 FIELD_PREP(WMI_TLV_LEN,
2599                                            sizeof(*wmm_param) - TLV_HDR_SIZE);
2600
2601                 wmm_param->aifs = wmi_wmm_arg->aifs;
2602                 wmm_param->cwmin = wmi_wmm_arg->cwmin;
2603                 wmm_param->cwmax = wmi_wmm_arg->cwmax;
2604                 wmm_param->txoplimit = wmi_wmm_arg->txop;
2605                 wmm_param->acm = wmi_wmm_arg->acm;
2606                 wmm_param->no_ack = wmi_wmm_arg->no_ack;
2607
2608                 ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2609                            "wmm set ac %d aifs %d cwmin %d cwmax %d txop %d acm %d no_ack %d\n",
2610                            ac, wmm_param->aifs, wmm_param->cwmin,
2611                            wmm_param->cwmax, wmm_param->txoplimit,
2612                            wmm_param->acm, wmm_param->no_ack);
2613         }
2614         ret = ath11k_wmi_cmd_send(wmi, skb,
2615                                   WMI_VDEV_SET_WMM_PARAMS_CMDID);
2616         if (ret) {
2617                 ath11k_warn(ar->ab,
2618                             "failed to send WMI_VDEV_SET_WMM_PARAMS_CMDID");
2619                 dev_kfree_skb(skb);
2620         }
2621
2622         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd vdev set wmm params");
2623
2624         return ret;
2625 }
2626
2627 int ath11k_wmi_send_dfs_phyerr_offload_enable_cmd(struct ath11k *ar,
2628                                                   u32 pdev_id)
2629 {
2630         struct ath11k_pdev_wmi *wmi = ar->wmi;
2631         struct wmi_dfs_phyerr_offload_cmd *cmd;
2632         struct sk_buff *skb;
2633         int ret;
2634
2635         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
2636         if (!skb)
2637                 return -ENOMEM;
2638
2639         cmd = (struct wmi_dfs_phyerr_offload_cmd *)skb->data;
2640         cmd->tlv_header =
2641                 FIELD_PREP(WMI_TLV_TAG,
2642                            WMI_TAG_PDEV_DFS_PHYERR_OFFLOAD_ENABLE_CMD) |
2643                 FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2644
2645         cmd->pdev_id = pdev_id;
2646
2647         ret = ath11k_wmi_cmd_send(wmi, skb,
2648                                   WMI_PDEV_DFS_PHYERR_OFFLOAD_ENABLE_CMDID);
2649         if (ret) {
2650                 ath11k_warn(ar->ab,
2651                             "failed to send WMI_PDEV_DFS_PHYERR_OFFLOAD_ENABLE cmd\n");
2652                 dev_kfree_skb(skb);
2653         }
2654
2655         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2656                    "cmd pdev dfs phyerr offload enable pdev id %d\n", pdev_id);
2657
2658         return ret;
2659 }
2660
2661 int ath11k_wmi_delba_send(struct ath11k *ar, u32 vdev_id, const u8 *mac,
2662                           u32 tid, u32 initiator, u32 reason)
2663 {
2664         struct ath11k_pdev_wmi *wmi = ar->wmi;
2665         struct wmi_delba_send_cmd *cmd;
2666         struct sk_buff *skb;
2667         int ret;
2668
2669         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
2670         if (!skb)
2671                 return -ENOMEM;
2672
2673         cmd = (struct wmi_delba_send_cmd *)skb->data;
2674         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_DELBA_SEND_CMD) |
2675                         FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2676         cmd->vdev_id = vdev_id;
2677         ether_addr_copy(cmd->peer_macaddr.addr, mac);
2678         cmd->tid = tid;
2679         cmd->initiator = initiator;
2680         cmd->reasoncode = reason;
2681
2682         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_DELBA_SEND_CMDID);
2683
2684         if (ret) {
2685                 ath11k_warn(ar->ab,
2686                             "failed to send WMI_DELBA_SEND_CMDID cmd\n");
2687                 dev_kfree_skb(skb);
2688         }
2689
2690         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2691                    "cmd delba send vdev_id 0x%X mac_addr %pM tid %u initiator %u reason %u\n",
2692                    vdev_id, mac, tid, initiator, reason);
2693
2694         return ret;
2695 }
2696
2697 int ath11k_wmi_addba_set_resp(struct ath11k *ar, u32 vdev_id, const u8 *mac,
2698                               u32 tid, u32 status)
2699 {
2700         struct ath11k_pdev_wmi *wmi = ar->wmi;
2701         struct wmi_addba_setresponse_cmd *cmd;
2702         struct sk_buff *skb;
2703         int ret;
2704
2705         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
2706         if (!skb)
2707                 return -ENOMEM;
2708
2709         cmd = (struct wmi_addba_setresponse_cmd *)skb->data;
2710         cmd->tlv_header =
2711                 FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ADDBA_SETRESPONSE_CMD) |
2712                 FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2713         cmd->vdev_id = vdev_id;
2714         ether_addr_copy(cmd->peer_macaddr.addr, mac);
2715         cmd->tid = tid;
2716         cmd->statuscode = status;
2717
2718         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_ADDBA_SET_RESP_CMDID);
2719
2720         if (ret) {
2721                 ath11k_warn(ar->ab,
2722                             "failed to send WMI_ADDBA_SET_RESP_CMDID cmd\n");
2723                 dev_kfree_skb(skb);
2724         }
2725
2726         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2727                    "cmd addba set resp vdev_id 0x%X mac_addr %pM tid %u status %u\n",
2728                    vdev_id, mac, tid, status);
2729
2730         return ret;
2731 }
2732
2733 int ath11k_wmi_addba_send(struct ath11k *ar, u32 vdev_id, const u8 *mac,
2734                           u32 tid, u32 buf_size)
2735 {
2736         struct ath11k_pdev_wmi *wmi = ar->wmi;
2737         struct wmi_addba_send_cmd *cmd;
2738         struct sk_buff *skb;
2739         int ret;
2740
2741         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
2742         if (!skb)
2743                 return -ENOMEM;
2744
2745         cmd = (struct wmi_addba_send_cmd *)skb->data;
2746         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ADDBA_SEND_CMD) |
2747                 FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2748         cmd->vdev_id = vdev_id;
2749         ether_addr_copy(cmd->peer_macaddr.addr, mac);
2750         cmd->tid = tid;
2751         cmd->buffersize = buf_size;
2752
2753         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_ADDBA_SEND_CMDID);
2754
2755         if (ret) {
2756                 ath11k_warn(ar->ab,
2757                             "failed to send WMI_ADDBA_SEND_CMDID cmd\n");
2758                 dev_kfree_skb(skb);
2759         }
2760
2761         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2762                    "cmd addba send vdev_id 0x%X mac_addr %pM tid %u bufsize %u\n",
2763                    vdev_id, mac, tid, buf_size);
2764
2765         return ret;
2766 }
2767
2768 int ath11k_wmi_addba_clear_resp(struct ath11k *ar, u32 vdev_id, const u8 *mac)
2769 {
2770         struct ath11k_pdev_wmi *wmi = ar->wmi;
2771         struct wmi_addba_clear_resp_cmd *cmd;
2772         struct sk_buff *skb;
2773         int ret;
2774
2775         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
2776         if (!skb)
2777                 return -ENOMEM;
2778
2779         cmd = (struct wmi_addba_clear_resp_cmd *)skb->data;
2780         cmd->tlv_header =
2781                 FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ADDBA_CLEAR_RESP_CMD) |
2782                 FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2783         cmd->vdev_id = vdev_id;
2784         ether_addr_copy(cmd->peer_macaddr.addr, mac);
2785
2786         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_ADDBA_CLEAR_RESP_CMDID);
2787
2788         if (ret) {
2789                 ath11k_warn(ar->ab,
2790                             "failed to send WMI_ADDBA_CLEAR_RESP_CMDID cmd\n");
2791                 dev_kfree_skb(skb);
2792         }
2793
2794         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2795                    "cmd addba clear resp vdev_id 0x%X mac_addr %pM\n",
2796                    vdev_id, mac);
2797
2798         return ret;
2799 }
2800
2801 int ath11k_wmi_pdev_peer_pktlog_filter(struct ath11k *ar, u8 *addr, u8 enable)
2802 {
2803         struct ath11k_pdev_wmi *wmi = ar->wmi;
2804         struct wmi_pdev_pktlog_filter_cmd *cmd;
2805         struct wmi_pdev_pktlog_filter_info *info;
2806         struct sk_buff *skb;
2807         struct wmi_tlv *tlv;
2808         void *ptr;
2809         int ret, len;
2810
2811         len = sizeof(*cmd) + sizeof(*info) + TLV_HDR_SIZE;
2812         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
2813         if (!skb)
2814                 return -ENOMEM;
2815
2816         cmd = (struct wmi_pdev_pktlog_filter_cmd *)skb->data;
2817
2818         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_PEER_PKTLOG_FILTER_CMD) |
2819                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2820
2821         cmd->pdev_id = DP_HW2SW_MACID(ar->pdev->pdev_id);
2822         cmd->num_mac = 1;
2823         cmd->enable = enable;
2824
2825         ptr = skb->data + sizeof(*cmd);
2826
2827         tlv = ptr;
2828         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
2829                       FIELD_PREP(WMI_TLV_LEN, sizeof(*info));
2830
2831         ptr += TLV_HDR_SIZE;
2832         info = ptr;
2833
2834         ether_addr_copy(info->peer_macaddr.addr, addr);
2835         info->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_PEER_PKTLOG_FILTER_INFO) |
2836                            FIELD_PREP(WMI_TLV_LEN,
2837                                       sizeof(*info) - TLV_HDR_SIZE);
2838
2839         ret = ath11k_wmi_cmd_send(wmi, skb,
2840                                   WMI_PDEV_PKTLOG_FILTER_CMDID);
2841         if (ret) {
2842                 ath11k_warn(ar->ab, "failed to send WMI_PDEV_PKTLOG_ENABLE_CMDID\n");
2843                 dev_kfree_skb(skb);
2844         }
2845
2846         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd pdev pktlog filter");
2847
2848         return ret;
2849 }
2850
2851 int
2852 ath11k_wmi_send_init_country_cmd(struct ath11k *ar,
2853                                  struct wmi_init_country_params init_cc_params)
2854 {
2855         struct ath11k_pdev_wmi *wmi = ar->wmi;
2856         struct wmi_init_country_cmd *cmd;
2857         struct sk_buff *skb;
2858         int ret;
2859
2860         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
2861         if (!skb)
2862                 return -ENOMEM;
2863
2864         cmd = (struct wmi_init_country_cmd *)skb->data;
2865         cmd->tlv_header =
2866                 FIELD_PREP(WMI_TLV_TAG,
2867                            WMI_TAG_SET_INIT_COUNTRY_CMD) |
2868                 FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2869
2870         cmd->pdev_id = ar->pdev->pdev_id;
2871
2872         switch (init_cc_params.flags) {
2873         case ALPHA_IS_SET:
2874                 cmd->init_cc_type = WMI_COUNTRY_INFO_TYPE_ALPHA;
2875                 memcpy((u8 *)&cmd->cc_info.alpha2,
2876                        init_cc_params.cc_info.alpha2, 3);
2877                 break;
2878         case CC_IS_SET:
2879                 cmd->init_cc_type = WMI_COUNTRY_INFO_TYPE_COUNTRY_CODE;
2880                 cmd->cc_info.country_code = init_cc_params.cc_info.country_code;
2881                 break;
2882         case REGDMN_IS_SET:
2883                 cmd->init_cc_type = WMI_COUNTRY_INFO_TYPE_REGDOMAIN;
2884                 cmd->cc_info.regdom_id = init_cc_params.cc_info.regdom_id;
2885                 break;
2886         default:
2887                 ath11k_warn(ar->ab, "unknown cc params flags: 0x%x",
2888                             init_cc_params.flags);
2889                 ret = -EINVAL;
2890                 goto err;
2891         }
2892
2893         ret = ath11k_wmi_cmd_send(wmi, skb,
2894                                   WMI_SET_INIT_COUNTRY_CMDID);
2895         if (ret) {
2896                 ath11k_warn(ar->ab,
2897                             "failed to send WMI_SET_INIT_COUNTRY CMD :%d\n",
2898                             ret);
2899                 goto err;
2900         }
2901
2902         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd set init country");
2903
2904         return 0;
2905
2906 err:
2907         dev_kfree_skb(skb);
2908         return ret;
2909 }
2910
2911 int ath11k_wmi_send_set_current_country_cmd(struct ath11k *ar,
2912                                             struct wmi_set_current_country_params *param)
2913 {
2914         struct ath11k_pdev_wmi *wmi = ar->wmi;
2915         struct wmi_set_current_country_cmd *cmd;
2916         struct sk_buff *skb;
2917         int ret;
2918
2919         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
2920         if (!skb)
2921                 return -ENOMEM;
2922
2923         cmd = (struct wmi_set_current_country_cmd *)skb->data;
2924         cmd->tlv_header =
2925                 FIELD_PREP(WMI_TLV_TAG, WMI_TAG_SET_CURRENT_COUNTRY_CMD) |
2926                 FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2927
2928         cmd->pdev_id = ar->pdev->pdev_id;
2929         memcpy(&cmd->new_alpha2, &param->alpha2, 3);
2930
2931         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_SET_CURRENT_COUNTRY_CMDID);
2932         if (ret) {
2933                 ath11k_warn(ar->ab,
2934                             "failed to send WMI_SET_CURRENT_COUNTRY_CMDID: %d\n", ret);
2935                 dev_kfree_skb(skb);
2936         }
2937
2938         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
2939                    "cmd set current country pdev id %d alpha2 %c%c\n",
2940                    ar->pdev->pdev_id,
2941                    param->alpha2[0],
2942                    param->alpha2[1]);
2943
2944         return ret;
2945 }
2946
2947 int
2948 ath11k_wmi_send_thermal_mitigation_param_cmd(struct ath11k *ar,
2949                                              struct thermal_mitigation_params *param)
2950 {
2951         struct ath11k_pdev_wmi *wmi = ar->wmi;
2952         struct wmi_therm_throt_config_request_cmd *cmd;
2953         struct wmi_therm_throt_level_config_info *lvl_conf;
2954         struct wmi_tlv *tlv;
2955         struct sk_buff *skb;
2956         int i, ret, len;
2957
2958         len = sizeof(*cmd) + TLV_HDR_SIZE +
2959               THERMAL_LEVELS * sizeof(struct wmi_therm_throt_level_config_info);
2960
2961         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
2962         if (!skb)
2963                 return -ENOMEM;
2964
2965         cmd = (struct wmi_therm_throt_config_request_cmd *)skb->data;
2966
2967         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_THERM_THROT_CONFIG_REQUEST) |
2968                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
2969
2970         cmd->pdev_id = ar->pdev->pdev_id;
2971         cmd->enable = param->enable;
2972         cmd->dc = param->dc;
2973         cmd->dc_per_event = param->dc_per_event;
2974         cmd->therm_throt_levels = THERMAL_LEVELS;
2975
2976         tlv = (struct wmi_tlv *)(skb->data + sizeof(*cmd));
2977         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
2978                       FIELD_PREP(WMI_TLV_LEN,
2979                                  (THERMAL_LEVELS *
2980                                   sizeof(struct wmi_therm_throt_level_config_info)));
2981
2982         lvl_conf = (struct wmi_therm_throt_level_config_info *)(skb->data +
2983                                                                 sizeof(*cmd) +
2984                                                                 TLV_HDR_SIZE);
2985         for (i = 0; i < THERMAL_LEVELS; i++) {
2986                 lvl_conf->tlv_header =
2987                         FIELD_PREP(WMI_TLV_TAG, WMI_TAG_THERM_THROT_LEVEL_CONFIG_INFO) |
2988                         FIELD_PREP(WMI_TLV_LEN, sizeof(*lvl_conf) - TLV_HDR_SIZE);
2989
2990                 lvl_conf->temp_lwm = param->levelconf[i].tmplwm;
2991                 lvl_conf->temp_hwm = param->levelconf[i].tmphwm;
2992                 lvl_conf->dc_off_percent = param->levelconf[i].dcoffpercent;
2993                 lvl_conf->prio = param->levelconf[i].priority;
2994                 lvl_conf++;
2995         }
2996
2997         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_THERM_THROT_SET_CONF_CMDID);
2998         if (ret) {
2999                 ath11k_warn(ar->ab, "failed to send THERM_THROT_SET_CONF cmd\n");
3000                 dev_kfree_skb(skb);
3001         }
3002
3003         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3004                    "cmd therm throt set conf pdev_id %d enable %d dc %d dc_per_event %x levels %d\n",
3005                    ar->pdev->pdev_id, param->enable, param->dc,
3006                    param->dc_per_event, THERMAL_LEVELS);
3007
3008         return ret;
3009 }
3010
3011 int ath11k_wmi_send_11d_scan_start_cmd(struct ath11k *ar,
3012                                        struct wmi_11d_scan_start_params *param)
3013 {
3014         struct ath11k_pdev_wmi *wmi = ar->wmi;
3015         struct wmi_11d_scan_start_cmd *cmd;
3016         struct sk_buff *skb;
3017         int ret;
3018
3019         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
3020         if (!skb)
3021                 return -ENOMEM;
3022
3023         cmd = (struct wmi_11d_scan_start_cmd *)skb->data;
3024         cmd->tlv_header =
3025                 FIELD_PREP(WMI_TLV_TAG, WMI_TAG_11D_SCAN_START_CMD) |
3026                 FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
3027
3028         cmd->vdev_id = param->vdev_id;
3029         cmd->scan_period_msec = param->scan_period_msec;
3030         cmd->start_interval_msec = param->start_interval_msec;
3031
3032         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_11D_SCAN_START_CMDID);
3033         if (ret) {
3034                 ath11k_warn(ar->ab,
3035                             "failed to send WMI_11D_SCAN_START_CMDID: %d\n", ret);
3036                 dev_kfree_skb(skb);
3037         }
3038
3039         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3040                    "cmd 11d scan start vdev id %d period %d ms internal %d ms\n",
3041                    cmd->vdev_id,
3042                    cmd->scan_period_msec,
3043                    cmd->start_interval_msec);
3044
3045         return ret;
3046 }
3047
3048 int ath11k_wmi_send_11d_scan_stop_cmd(struct ath11k *ar, u32 vdev_id)
3049 {
3050         struct ath11k_pdev_wmi *wmi = ar->wmi;
3051         struct wmi_11d_scan_stop_cmd *cmd;
3052         struct sk_buff *skb;
3053         int ret;
3054
3055         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
3056         if (!skb)
3057                 return -ENOMEM;
3058
3059         cmd = (struct wmi_11d_scan_stop_cmd *)skb->data;
3060         cmd->tlv_header =
3061                 FIELD_PREP(WMI_TLV_TAG, WMI_TAG_11D_SCAN_STOP_CMD) |
3062                 FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
3063
3064         cmd->vdev_id = vdev_id;
3065
3066         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_11D_SCAN_STOP_CMDID);
3067         if (ret) {
3068                 ath11k_warn(ar->ab,
3069                             "failed to send WMI_11D_SCAN_STOP_CMDID: %d\n", ret);
3070                 dev_kfree_skb(skb);
3071         }
3072
3073         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3074                    "cmd 11d scan stop vdev id %d\n",
3075                    cmd->vdev_id);
3076
3077         return ret;
3078 }
3079
3080 int ath11k_wmi_pdev_pktlog_enable(struct ath11k *ar, u32 pktlog_filter)
3081 {
3082         struct ath11k_pdev_wmi *wmi = ar->wmi;
3083         struct wmi_pktlog_enable_cmd *cmd;
3084         struct sk_buff *skb;
3085         int ret;
3086
3087         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
3088         if (!skb)
3089                 return -ENOMEM;
3090
3091         cmd = (struct wmi_pktlog_enable_cmd *)skb->data;
3092
3093         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_PKTLOG_ENABLE_CMD) |
3094                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
3095
3096         cmd->pdev_id = DP_HW2SW_MACID(ar->pdev->pdev_id);
3097         cmd->evlist = pktlog_filter;
3098         cmd->enable = ATH11K_WMI_PKTLOG_ENABLE_FORCE;
3099
3100         ret = ath11k_wmi_cmd_send(wmi, skb,
3101                                   WMI_PDEV_PKTLOG_ENABLE_CMDID);
3102         if (ret) {
3103                 ath11k_warn(ar->ab, "failed to send WMI_PDEV_PKTLOG_ENABLE_CMDID\n");
3104                 dev_kfree_skb(skb);
3105         }
3106
3107         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd pdev pktlog enable");
3108
3109         return ret;
3110 }
3111
3112 int ath11k_wmi_pdev_pktlog_disable(struct ath11k *ar)
3113 {
3114         struct ath11k_pdev_wmi *wmi = ar->wmi;
3115         struct wmi_pktlog_disable_cmd *cmd;
3116         struct sk_buff *skb;
3117         int ret;
3118
3119         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, sizeof(*cmd));
3120         if (!skb)
3121                 return -ENOMEM;
3122
3123         cmd = (struct wmi_pktlog_disable_cmd *)skb->data;
3124
3125         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_PKTLOG_DISABLE_CMD) |
3126                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
3127
3128         cmd->pdev_id = DP_HW2SW_MACID(ar->pdev->pdev_id);
3129
3130         ret = ath11k_wmi_cmd_send(wmi, skb,
3131                                   WMI_PDEV_PKTLOG_DISABLE_CMDID);
3132         if (ret) {
3133                 ath11k_warn(ar->ab, "failed to send WMI_PDEV_PKTLOG_ENABLE_CMDID\n");
3134                 dev_kfree_skb(skb);
3135         }
3136
3137         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd pdev pktlog disable");
3138
3139         return ret;
3140 }
3141
3142 void ath11k_wmi_fill_default_twt_params(struct wmi_twt_enable_params *twt_params)
3143 {
3144         twt_params->sta_cong_timer_ms = ATH11K_TWT_DEF_STA_CONG_TIMER_MS;
3145         twt_params->default_slot_size = ATH11K_TWT_DEF_DEFAULT_SLOT_SIZE;
3146         twt_params->congestion_thresh_setup = ATH11K_TWT_DEF_CONGESTION_THRESH_SETUP;
3147         twt_params->congestion_thresh_teardown =
3148                 ATH11K_TWT_DEF_CONGESTION_THRESH_TEARDOWN;
3149         twt_params->congestion_thresh_critical =
3150                 ATH11K_TWT_DEF_CONGESTION_THRESH_CRITICAL;
3151         twt_params->interference_thresh_teardown =
3152                 ATH11K_TWT_DEF_INTERFERENCE_THRESH_TEARDOWN;
3153         twt_params->interference_thresh_setup =
3154                 ATH11K_TWT_DEF_INTERFERENCE_THRESH_SETUP;
3155         twt_params->min_no_sta_setup = ATH11K_TWT_DEF_MIN_NO_STA_SETUP;
3156         twt_params->min_no_sta_teardown = ATH11K_TWT_DEF_MIN_NO_STA_TEARDOWN;
3157         twt_params->no_of_bcast_mcast_slots = ATH11K_TWT_DEF_NO_OF_BCAST_MCAST_SLOTS;
3158         twt_params->min_no_twt_slots = ATH11K_TWT_DEF_MIN_NO_TWT_SLOTS;
3159         twt_params->max_no_sta_twt = ATH11K_TWT_DEF_MAX_NO_STA_TWT;
3160         twt_params->mode_check_interval = ATH11K_TWT_DEF_MODE_CHECK_INTERVAL;
3161         twt_params->add_sta_slot_interval = ATH11K_TWT_DEF_ADD_STA_SLOT_INTERVAL;
3162         twt_params->remove_sta_slot_interval =
3163                 ATH11K_TWT_DEF_REMOVE_STA_SLOT_INTERVAL;
3164         /* TODO add MBSSID support */
3165         twt_params->mbss_support = 0;
3166 }
3167
3168 int ath11k_wmi_send_twt_enable_cmd(struct ath11k *ar, u32 pdev_id,
3169                                    struct wmi_twt_enable_params *params)
3170 {
3171         struct ath11k_pdev_wmi *wmi = ar->wmi;
3172         struct ath11k_base *ab = wmi->wmi_ab->ab;
3173         struct wmi_twt_enable_params_cmd *cmd;
3174         struct sk_buff *skb;
3175         int ret, len;
3176
3177         len = sizeof(*cmd);
3178
3179         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3180         if (!skb)
3181                 return -ENOMEM;
3182
3183         cmd = (struct wmi_twt_enable_params_cmd *)skb->data;
3184         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_TWT_ENABLE_CMD) |
3185                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3186         cmd->pdev_id = pdev_id;
3187         cmd->sta_cong_timer_ms = params->sta_cong_timer_ms;
3188         cmd->default_slot_size = params->default_slot_size;
3189         cmd->congestion_thresh_setup = params->congestion_thresh_setup;
3190         cmd->congestion_thresh_teardown = params->congestion_thresh_teardown;
3191         cmd->congestion_thresh_critical = params->congestion_thresh_critical;
3192         cmd->interference_thresh_teardown = params->interference_thresh_teardown;
3193         cmd->interference_thresh_setup = params->interference_thresh_setup;
3194         cmd->min_no_sta_setup = params->min_no_sta_setup;
3195         cmd->min_no_sta_teardown = params->min_no_sta_teardown;
3196         cmd->no_of_bcast_mcast_slots = params->no_of_bcast_mcast_slots;
3197         cmd->min_no_twt_slots = params->min_no_twt_slots;
3198         cmd->max_no_sta_twt = params->max_no_sta_twt;
3199         cmd->mode_check_interval = params->mode_check_interval;
3200         cmd->add_sta_slot_interval = params->add_sta_slot_interval;
3201         cmd->remove_sta_slot_interval = params->remove_sta_slot_interval;
3202         cmd->mbss_support = params->mbss_support;
3203
3204         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_TWT_ENABLE_CMDID);
3205         if (ret) {
3206                 ath11k_warn(ab, "Failed to send WMI_TWT_ENABLE_CMDID");
3207                 dev_kfree_skb(skb);
3208                 return ret;
3209         }
3210
3211         ar->twt_enabled = 1;
3212
3213         ath11k_dbg(ab, ATH11K_DBG_WMI, "cmd twt enable");
3214
3215         return 0;
3216 }
3217
3218 int
3219 ath11k_wmi_send_twt_disable_cmd(struct ath11k *ar, u32 pdev_id)
3220 {
3221         struct ath11k_pdev_wmi *wmi = ar->wmi;
3222         struct ath11k_base *ab = wmi->wmi_ab->ab;
3223         struct wmi_twt_disable_params_cmd *cmd;
3224         struct sk_buff *skb;
3225         int ret, len;
3226
3227         len = sizeof(*cmd);
3228
3229         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3230         if (!skb)
3231                 return -ENOMEM;
3232
3233         cmd = (struct wmi_twt_disable_params_cmd *)skb->data;
3234         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_TWT_DISABLE_CMD) |
3235                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3236         cmd->pdev_id = pdev_id;
3237
3238         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_TWT_DISABLE_CMDID);
3239         if (ret) {
3240                 ath11k_warn(ab, "Failed to send WMI_TWT_DISABLE_CMDID");
3241                 dev_kfree_skb(skb);
3242                 return ret;
3243         }
3244
3245         ath11k_dbg(ab, ATH11K_DBG_WMI, "cmd twt disable");
3246
3247         ar->twt_enabled = 0;
3248
3249         return 0;
3250 }
3251
3252 int ath11k_wmi_send_twt_add_dialog_cmd(struct ath11k *ar,
3253                                        struct wmi_twt_add_dialog_params *params)
3254 {
3255         struct ath11k_pdev_wmi *wmi = ar->wmi;
3256         struct ath11k_base *ab = wmi->wmi_ab->ab;
3257         struct wmi_twt_add_dialog_params_cmd *cmd;
3258         struct sk_buff *skb;
3259         int ret, len;
3260
3261         len = sizeof(*cmd);
3262
3263         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3264         if (!skb)
3265                 return -ENOMEM;
3266
3267         cmd = (struct wmi_twt_add_dialog_params_cmd *)skb->data;
3268         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_TWT_ADD_DIALOG_CMD) |
3269                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3270
3271         cmd->vdev_id = params->vdev_id;
3272         ether_addr_copy(cmd->peer_macaddr.addr, params->peer_macaddr);
3273         cmd->dialog_id = params->dialog_id;
3274         cmd->wake_intvl_us = params->wake_intvl_us;
3275         cmd->wake_intvl_mantis = params->wake_intvl_mantis;
3276         cmd->wake_dura_us = params->wake_dura_us;
3277         cmd->sp_offset_us = params->sp_offset_us;
3278         cmd->flags = params->twt_cmd;
3279         if (params->flag_bcast)
3280                 cmd->flags |= WMI_TWT_ADD_DIALOG_FLAG_BCAST;
3281         if (params->flag_trigger)
3282                 cmd->flags |= WMI_TWT_ADD_DIALOG_FLAG_TRIGGER;
3283         if (params->flag_flow_type)
3284                 cmd->flags |= WMI_TWT_ADD_DIALOG_FLAG_FLOW_TYPE;
3285         if (params->flag_protection)
3286                 cmd->flags |= WMI_TWT_ADD_DIALOG_FLAG_PROTECTION;
3287
3288         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_TWT_ADD_DIALOG_CMDID);
3289         if (ret) {
3290                 ath11k_warn(ab,
3291                             "failed to send wmi command to add twt dialog: %d",
3292                             ret);
3293                 dev_kfree_skb(skb);
3294                 return ret;
3295         }
3296
3297         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3298                    "cmd twt add dialog vdev %u dialog id %u wake interval %u mantissa %u wake duration %u service period offset %u flags 0x%x\n",
3299                    cmd->vdev_id, cmd->dialog_id, cmd->wake_intvl_us,
3300                    cmd->wake_intvl_mantis, cmd->wake_dura_us, cmd->sp_offset_us,
3301                    cmd->flags);
3302
3303         return 0;
3304 }
3305
3306 int ath11k_wmi_send_twt_del_dialog_cmd(struct ath11k *ar,
3307                                        struct wmi_twt_del_dialog_params *params)
3308 {
3309         struct ath11k_pdev_wmi *wmi = ar->wmi;
3310         struct ath11k_base *ab = wmi->wmi_ab->ab;
3311         struct wmi_twt_del_dialog_params_cmd *cmd;
3312         struct sk_buff *skb;
3313         int ret, len;
3314
3315         len = sizeof(*cmd);
3316
3317         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3318         if (!skb)
3319                 return -ENOMEM;
3320
3321         cmd = (struct wmi_twt_del_dialog_params_cmd *)skb->data;
3322         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_TWT_DEL_DIALOG_CMD) |
3323                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3324
3325         cmd->vdev_id = params->vdev_id;
3326         ether_addr_copy(cmd->peer_macaddr.addr, params->peer_macaddr);
3327         cmd->dialog_id = params->dialog_id;
3328
3329         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_TWT_DEL_DIALOG_CMDID);
3330         if (ret) {
3331                 ath11k_warn(ab,
3332                             "failed to send wmi command to delete twt dialog: %d",
3333                             ret);
3334                 dev_kfree_skb(skb);
3335                 return ret;
3336         }
3337
3338         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3339                    "cmd twt del dialog vdev %u dialog id %u\n",
3340                    cmd->vdev_id, cmd->dialog_id);
3341
3342         return 0;
3343 }
3344
3345 int ath11k_wmi_send_twt_pause_dialog_cmd(struct ath11k *ar,
3346                                          struct wmi_twt_pause_dialog_params *params)
3347 {
3348         struct ath11k_pdev_wmi *wmi = ar->wmi;
3349         struct ath11k_base *ab = wmi->wmi_ab->ab;
3350         struct wmi_twt_pause_dialog_params_cmd *cmd;
3351         struct sk_buff *skb;
3352         int ret, len;
3353
3354         len = sizeof(*cmd);
3355
3356         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3357         if (!skb)
3358                 return -ENOMEM;
3359
3360         cmd = (struct wmi_twt_pause_dialog_params_cmd *)skb->data;
3361         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
3362                                      WMI_TAG_TWT_PAUSE_DIALOG_CMD) |
3363                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3364
3365         cmd->vdev_id = params->vdev_id;
3366         ether_addr_copy(cmd->peer_macaddr.addr, params->peer_macaddr);
3367         cmd->dialog_id = params->dialog_id;
3368
3369         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_TWT_PAUSE_DIALOG_CMDID);
3370         if (ret) {
3371                 ath11k_warn(ab,
3372                             "failed to send wmi command to pause twt dialog: %d",
3373                             ret);
3374                 dev_kfree_skb(skb);
3375                 return ret;
3376         }
3377
3378         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3379                    "cmd twt pause dialog vdev %u dialog id %u\n",
3380                    cmd->vdev_id, cmd->dialog_id);
3381
3382         return 0;
3383 }
3384
3385 int ath11k_wmi_send_twt_resume_dialog_cmd(struct ath11k *ar,
3386                                           struct wmi_twt_resume_dialog_params *params)
3387 {
3388         struct ath11k_pdev_wmi *wmi = ar->wmi;
3389         struct ath11k_base *ab = wmi->wmi_ab->ab;
3390         struct wmi_twt_resume_dialog_params_cmd *cmd;
3391         struct sk_buff *skb;
3392         int ret, len;
3393
3394         len = sizeof(*cmd);
3395
3396         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3397         if (!skb)
3398                 return -ENOMEM;
3399
3400         cmd = (struct wmi_twt_resume_dialog_params_cmd *)skb->data;
3401         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
3402                                      WMI_TAG_TWT_RESUME_DIALOG_CMD) |
3403                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3404
3405         cmd->vdev_id = params->vdev_id;
3406         ether_addr_copy(cmd->peer_macaddr.addr, params->peer_macaddr);
3407         cmd->dialog_id = params->dialog_id;
3408         cmd->sp_offset_us = params->sp_offset_us;
3409         cmd->next_twt_size = params->next_twt_size;
3410
3411         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_TWT_RESUME_DIALOG_CMDID);
3412         if (ret) {
3413                 ath11k_warn(ab,
3414                             "failed to send wmi command to resume twt dialog: %d",
3415                             ret);
3416                 dev_kfree_skb(skb);
3417                 return ret;
3418         }
3419
3420         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3421                    "cmd twt resume dialog vdev %u dialog id %u service period offset %u next twt subfield size %u\n",
3422                    cmd->vdev_id, cmd->dialog_id, cmd->sp_offset_us,
3423                    cmd->next_twt_size);
3424
3425         return 0;
3426 }
3427
3428 int
3429 ath11k_wmi_send_obss_spr_cmd(struct ath11k *ar, u32 vdev_id,
3430                              struct ieee80211_he_obss_pd *he_obss_pd)
3431 {
3432         struct ath11k_pdev_wmi *wmi = ar->wmi;
3433         struct ath11k_base *ab = wmi->wmi_ab->ab;
3434         struct wmi_obss_spatial_reuse_params_cmd *cmd;
3435         struct sk_buff *skb;
3436         int ret, len;
3437
3438         len = sizeof(*cmd);
3439
3440         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3441         if (!skb)
3442                 return -ENOMEM;
3443
3444         cmd = (struct wmi_obss_spatial_reuse_params_cmd *)skb->data;
3445         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
3446                                      WMI_TAG_OBSS_SPATIAL_REUSE_SET_CMD) |
3447                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3448         cmd->vdev_id = vdev_id;
3449         cmd->enable = he_obss_pd->enable;
3450         cmd->obss_min = he_obss_pd->min_offset;
3451         cmd->obss_max = he_obss_pd->max_offset;
3452
3453         ret = ath11k_wmi_cmd_send(wmi, skb,
3454                                   WMI_PDEV_OBSS_PD_SPATIAL_REUSE_CMDID);
3455         if (ret) {
3456                 ath11k_warn(ab,
3457                             "Failed to send WMI_PDEV_OBSS_PD_SPATIAL_REUSE_CMDID");
3458                 dev_kfree_skb(skb);
3459                 return ret;
3460         }
3461
3462         ath11k_dbg(ab, ATH11K_DBG_WMI, "cmd pdev obss pd spatial reuse");
3463
3464         return 0;
3465 }
3466
3467 int
3468 ath11k_wmi_pdev_set_srg_bss_color_bitmap(struct ath11k *ar, u32 *bitmap)
3469 {
3470         struct ath11k_pdev_wmi *wmi = ar->wmi;
3471         struct ath11k_base *ab = wmi->wmi_ab->ab;
3472         struct wmi_pdev_obss_pd_bitmap_cmd *cmd;
3473         struct sk_buff *skb;
3474         int ret, len;
3475
3476         len = sizeof(*cmd);
3477
3478         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3479         if (!skb)
3480                 return -ENOMEM;
3481
3482         cmd = (struct wmi_pdev_obss_pd_bitmap_cmd *)skb->data;
3483         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
3484                                      WMI_TAG_PDEV_SRG_BSS_COLOR_BITMAP_CMD) |
3485                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3486         cmd->pdev_id = ar->pdev->pdev_id;
3487         memcpy(cmd->bitmap, bitmap, sizeof(cmd->bitmap));
3488
3489         ret = ath11k_wmi_cmd_send(wmi, skb,
3490                                   WMI_PDEV_SET_SRG_BSS_COLOR_BITMAP_CMDID);
3491         if (ret) {
3492                 ath11k_warn(ab,
3493                             "failed to send WMI_PDEV_SET_SRG_BSS_COLOR_BITMAP_CMDID");
3494                 dev_kfree_skb(skb);
3495                 return ret;
3496         }
3497
3498         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3499                    "cmd pdev set srg bss color bitmap pdev_id %d bss color bitmap %08x %08x\n",
3500                    cmd->pdev_id, cmd->bitmap[0], cmd->bitmap[1]);
3501
3502         return 0;
3503 }
3504
3505 int
3506 ath11k_wmi_pdev_set_srg_patial_bssid_bitmap(struct ath11k *ar, u32 *bitmap)
3507 {
3508         struct ath11k_pdev_wmi *wmi = ar->wmi;
3509         struct ath11k_base *ab = wmi->wmi_ab->ab;
3510         struct wmi_pdev_obss_pd_bitmap_cmd *cmd;
3511         struct sk_buff *skb;
3512         int ret, len;
3513
3514         len = sizeof(*cmd);
3515
3516         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3517         if (!skb)
3518                 return -ENOMEM;
3519
3520         cmd = (struct wmi_pdev_obss_pd_bitmap_cmd *)skb->data;
3521         cmd->tlv_header =
3522                 FIELD_PREP(WMI_TLV_TAG,
3523                            WMI_TAG_PDEV_SRG_PARTIAL_BSSID_BITMAP_CMD) |
3524                 FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3525         cmd->pdev_id = ar->pdev->pdev_id;
3526         memcpy(cmd->bitmap, bitmap, sizeof(cmd->bitmap));
3527
3528         ret = ath11k_wmi_cmd_send(wmi, skb,
3529                                   WMI_PDEV_SET_SRG_PARTIAL_BSSID_BITMAP_CMDID);
3530         if (ret) {
3531                 ath11k_warn(ab,
3532                             "failed to send WMI_PDEV_SET_SRG_PARTIAL_BSSID_BITMAP_CMDID");
3533                 dev_kfree_skb(skb);
3534                 return ret;
3535         }
3536
3537         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3538                    "cmd pdev set srg partial bssid bitmap pdev_id %d partial bssid bitmap %08x %08x\n",
3539                    cmd->pdev_id, cmd->bitmap[0], cmd->bitmap[1]);
3540
3541         return 0;
3542 }
3543
3544 int
3545 ath11k_wmi_pdev_srg_obss_color_enable_bitmap(struct ath11k *ar, u32 *bitmap)
3546 {
3547         struct ath11k_pdev_wmi *wmi = ar->wmi;
3548         struct ath11k_base *ab = wmi->wmi_ab->ab;
3549         struct wmi_pdev_obss_pd_bitmap_cmd *cmd;
3550         struct sk_buff *skb;
3551         int ret, len;
3552
3553         len = sizeof(*cmd);
3554
3555         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3556         if (!skb)
3557                 return -ENOMEM;
3558
3559         cmd = (struct wmi_pdev_obss_pd_bitmap_cmd *)skb->data;
3560         cmd->tlv_header =
3561                 FIELD_PREP(WMI_TLV_TAG,
3562                            WMI_TAG_PDEV_SRG_OBSS_COLOR_ENABLE_BITMAP_CMD) |
3563                 FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3564         cmd->pdev_id = ar->pdev->pdev_id;
3565         memcpy(cmd->bitmap, bitmap, sizeof(cmd->bitmap));
3566
3567         ret = ath11k_wmi_cmd_send(wmi, skb,
3568                                   WMI_PDEV_SET_SRG_OBSS_COLOR_ENABLE_BITMAP_CMDID);
3569         if (ret) {
3570                 ath11k_warn(ab,
3571                             "failed to send WMI_PDEV_SET_SRG_OBSS_COLOR_ENABLE_BITMAP_CMDID");
3572                 dev_kfree_skb(skb);
3573                 return ret;
3574         }
3575
3576         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3577                    "cmd pdev set srg obsscolor enable pdev_id %d bss color enable bitmap %08x %08x\n",
3578                    cmd->pdev_id, cmd->bitmap[0], cmd->bitmap[1]);
3579
3580         return 0;
3581 }
3582
3583 int
3584 ath11k_wmi_pdev_srg_obss_bssid_enable_bitmap(struct ath11k *ar, u32 *bitmap)
3585 {
3586         struct ath11k_pdev_wmi *wmi = ar->wmi;
3587         struct ath11k_base *ab = wmi->wmi_ab->ab;
3588         struct wmi_pdev_obss_pd_bitmap_cmd *cmd;
3589         struct sk_buff *skb;
3590         int ret, len;
3591
3592         len = sizeof(*cmd);
3593
3594         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3595         if (!skb)
3596                 return -ENOMEM;
3597
3598         cmd = (struct wmi_pdev_obss_pd_bitmap_cmd *)skb->data;
3599         cmd->tlv_header =
3600                 FIELD_PREP(WMI_TLV_TAG,
3601                            WMI_TAG_PDEV_SRG_OBSS_BSSID_ENABLE_BITMAP_CMD) |
3602                 FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3603         cmd->pdev_id = ar->pdev->pdev_id;
3604         memcpy(cmd->bitmap, bitmap, sizeof(cmd->bitmap));
3605
3606         ret = ath11k_wmi_cmd_send(wmi, skb,
3607                                   WMI_PDEV_SET_SRG_OBSS_BSSID_ENABLE_BITMAP_CMDID);
3608         if (ret) {
3609                 ath11k_warn(ab,
3610                             "failed to send WMI_PDEV_SET_SRG_OBSS_BSSID_ENABLE_BITMAP_CMDID");
3611                 dev_kfree_skb(skb);
3612                 return ret;
3613         }
3614
3615         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3616                    "cmd pdev set srg obss bssid enable bitmap pdev_id %d bssid enable bitmap %08x %08x\n",
3617                    cmd->pdev_id, cmd->bitmap[0], cmd->bitmap[1]);
3618
3619         return 0;
3620 }
3621
3622 int
3623 ath11k_wmi_pdev_non_srg_obss_color_enable_bitmap(struct ath11k *ar, u32 *bitmap)
3624 {
3625         struct ath11k_pdev_wmi *wmi = ar->wmi;
3626         struct ath11k_base *ab = wmi->wmi_ab->ab;
3627         struct wmi_pdev_obss_pd_bitmap_cmd *cmd;
3628         struct sk_buff *skb;
3629         int ret, len;
3630
3631         len = sizeof(*cmd);
3632
3633         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3634         if (!skb)
3635                 return -ENOMEM;
3636
3637         cmd = (struct wmi_pdev_obss_pd_bitmap_cmd *)skb->data;
3638         cmd->tlv_header =
3639                 FIELD_PREP(WMI_TLV_TAG,
3640                            WMI_TAG_PDEV_NON_SRG_OBSS_COLOR_ENABLE_BITMAP_CMD) |
3641                 FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3642         cmd->pdev_id = ar->pdev->pdev_id;
3643         memcpy(cmd->bitmap, bitmap, sizeof(cmd->bitmap));
3644
3645         ret = ath11k_wmi_cmd_send(wmi, skb,
3646                                   WMI_PDEV_SET_NON_SRG_OBSS_COLOR_ENABLE_BITMAP_CMDID);
3647         if (ret) {
3648                 ath11k_warn(ab,
3649                             "failed to send WMI_PDEV_SET_NON_SRG_OBSS_COLOR_ENABLE_BITMAP_CMDID");
3650                 dev_kfree_skb(skb);
3651                 return ret;
3652         }
3653
3654         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3655                    "cmd pdev set non srg obss color enable bitmap pdev_id %d bss color enable bitmap %08x %08x\n",
3656                    cmd->pdev_id, cmd->bitmap[0], cmd->bitmap[1]);
3657
3658         return 0;
3659 }
3660
3661 int
3662 ath11k_wmi_pdev_non_srg_obss_bssid_enable_bitmap(struct ath11k *ar, u32 *bitmap)
3663 {
3664         struct ath11k_pdev_wmi *wmi = ar->wmi;
3665         struct ath11k_base *ab = wmi->wmi_ab->ab;
3666         struct wmi_pdev_obss_pd_bitmap_cmd *cmd;
3667         struct sk_buff *skb;
3668         int ret, len;
3669
3670         len = sizeof(*cmd);
3671
3672         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3673         if (!skb)
3674                 return -ENOMEM;
3675
3676         cmd = (struct wmi_pdev_obss_pd_bitmap_cmd *)skb->data;
3677         cmd->tlv_header =
3678                 FIELD_PREP(WMI_TLV_TAG,
3679                            WMI_TAG_PDEV_NON_SRG_OBSS_BSSID_ENABLE_BITMAP_CMD) |
3680                 FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3681         cmd->pdev_id = ar->pdev->pdev_id;
3682         memcpy(cmd->bitmap, bitmap, sizeof(cmd->bitmap));
3683
3684         ret = ath11k_wmi_cmd_send(wmi, skb,
3685                                   WMI_PDEV_SET_NON_SRG_OBSS_BSSID_ENABLE_BITMAP_CMDID);
3686         if (ret) {
3687                 ath11k_warn(ab,
3688                             "failed to send WMI_PDEV_SET_NON_SRG_OBSS_BSSID_ENABLE_BITMAP_CMDID");
3689                 dev_kfree_skb(skb);
3690                 return ret;
3691         }
3692
3693         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3694                    "cmd pdev set non srg obss bssid enable bitmap pdev_id %d bssid enable bitmap %08x %08x\n",
3695                    cmd->pdev_id, cmd->bitmap[0], cmd->bitmap[1]);
3696
3697         return 0;
3698 }
3699
3700 int
3701 ath11k_wmi_send_obss_color_collision_cfg_cmd(struct ath11k *ar, u32 vdev_id,
3702                                              u8 bss_color, u32 period,
3703                                              bool enable)
3704 {
3705         struct ath11k_pdev_wmi *wmi = ar->wmi;
3706         struct ath11k_base *ab = wmi->wmi_ab->ab;
3707         struct wmi_obss_color_collision_cfg_params_cmd *cmd;
3708         struct sk_buff *skb;
3709         int ret, len;
3710
3711         len = sizeof(*cmd);
3712
3713         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3714         if (!skb)
3715                 return -ENOMEM;
3716
3717         cmd = (struct wmi_obss_color_collision_cfg_params_cmd *)skb->data;
3718         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
3719                                      WMI_TAG_OBSS_COLOR_COLLISION_DET_CONFIG) |
3720                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3721         cmd->vdev_id = vdev_id;
3722         cmd->evt_type = enable ? ATH11K_OBSS_COLOR_COLLISION_DETECTION :
3723                                  ATH11K_OBSS_COLOR_COLLISION_DETECTION_DISABLE;
3724         cmd->current_bss_color = bss_color;
3725         cmd->detection_period_ms = period;
3726         cmd->scan_period_ms = ATH11K_BSS_COLOR_COLLISION_SCAN_PERIOD_MS;
3727         cmd->free_slot_expiry_time_ms = 0;
3728         cmd->flags = 0;
3729
3730         ret = ath11k_wmi_cmd_send(wmi, skb,
3731                                   WMI_OBSS_COLOR_COLLISION_DET_CONFIG_CMDID);
3732         if (ret) {
3733                 ath11k_warn(ab, "Failed to send WMI_OBSS_COLOR_COLLISION_DET_CONFIG_CMDID");
3734                 dev_kfree_skb(skb);
3735                 return ret;
3736         }
3737
3738         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3739                    "cmd obss color collision det config id %d type %d bss_color %d detect_period %d scan_period %d\n",
3740                    cmd->vdev_id, cmd->evt_type, cmd->current_bss_color,
3741                    cmd->detection_period_ms, cmd->scan_period_ms);
3742
3743         return 0;
3744 }
3745
3746 int ath11k_wmi_send_bss_color_change_enable_cmd(struct ath11k *ar, u32 vdev_id,
3747                                                 bool enable)
3748 {
3749         struct ath11k_pdev_wmi *wmi = ar->wmi;
3750         struct ath11k_base *ab = wmi->wmi_ab->ab;
3751         struct wmi_bss_color_change_enable_params_cmd *cmd;
3752         struct sk_buff *skb;
3753         int ret, len;
3754
3755         len = sizeof(*cmd);
3756
3757         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
3758         if (!skb)
3759                 return -ENOMEM;
3760
3761         cmd = (struct wmi_bss_color_change_enable_params_cmd *)skb->data;
3762         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_BSS_COLOR_CHANGE_ENABLE) |
3763                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3764         cmd->vdev_id = vdev_id;
3765         cmd->enable = enable ? 1 : 0;
3766
3767         ret = ath11k_wmi_cmd_send(wmi, skb,
3768                                   WMI_BSS_COLOR_CHANGE_ENABLE_CMDID);
3769         if (ret) {
3770                 ath11k_warn(ab, "Failed to send WMI_BSS_COLOR_CHANGE_ENABLE_CMDID");
3771                 dev_kfree_skb(skb);
3772                 return ret;
3773         }
3774
3775         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3776                    "cmd bss color change enable id %d enable %d\n",
3777                    cmd->vdev_id, cmd->enable);
3778
3779         return 0;
3780 }
3781
3782 int ath11k_wmi_fils_discovery_tmpl(struct ath11k *ar, u32 vdev_id,
3783                                    struct sk_buff *tmpl)
3784 {
3785         struct wmi_tlv *tlv;
3786         struct sk_buff *skb;
3787         void *ptr;
3788         int ret, len;
3789         size_t aligned_len;
3790         struct wmi_fils_discovery_tmpl_cmd *cmd;
3791
3792         aligned_len = roundup(tmpl->len, 4);
3793         len = sizeof(*cmd) + TLV_HDR_SIZE + aligned_len;
3794
3795         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3796                    "vdev %i set FILS discovery template\n", vdev_id);
3797
3798         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
3799         if (!skb)
3800                 return -ENOMEM;
3801
3802         cmd = (struct wmi_fils_discovery_tmpl_cmd *)skb->data;
3803         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
3804                                      WMI_TAG_FILS_DISCOVERY_TMPL_CMD) |
3805                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
3806         cmd->vdev_id = vdev_id;
3807         cmd->buf_len = tmpl->len;
3808         ptr = skb->data + sizeof(*cmd);
3809
3810         tlv = ptr;
3811         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
3812                       FIELD_PREP(WMI_TLV_LEN, aligned_len);
3813         memcpy(tlv->value, tmpl->data, tmpl->len);
3814
3815         ret = ath11k_wmi_cmd_send(ar->wmi, skb, WMI_FILS_DISCOVERY_TMPL_CMDID);
3816         if (ret) {
3817                 ath11k_warn(ar->ab,
3818                             "WMI vdev %i failed to send FILS discovery template command\n",
3819                             vdev_id);
3820                 dev_kfree_skb(skb);
3821                 return ret;
3822         }
3823
3824         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd fils discovery tmpl");
3825
3826         return 0;
3827 }
3828
3829 int ath11k_wmi_probe_resp_tmpl(struct ath11k *ar, u32 vdev_id,
3830                                struct sk_buff *tmpl)
3831 {
3832         struct wmi_probe_tmpl_cmd *cmd;
3833         struct wmi_bcn_prb_info *probe_info;
3834         struct wmi_tlv *tlv;
3835         struct sk_buff *skb;
3836         void *ptr;
3837         int ret, len;
3838         size_t aligned_len = roundup(tmpl->len, 4);
3839
3840         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3841                    "vdev %i set probe response template\n", vdev_id);
3842
3843         len = sizeof(*cmd) + sizeof(*probe_info) + TLV_HDR_SIZE + aligned_len;
3844
3845         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
3846         if (!skb)
3847                 return -ENOMEM;
3848
3849         cmd = (struct wmi_probe_tmpl_cmd *)skb->data;
3850         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PRB_TMPL_CMD) |
3851                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
3852         cmd->vdev_id = vdev_id;
3853         cmd->buf_len = tmpl->len;
3854
3855         ptr = skb->data + sizeof(*cmd);
3856
3857         probe_info = ptr;
3858         len = sizeof(*probe_info);
3859         probe_info->tlv_header = FIELD_PREP(WMI_TLV_TAG,
3860                                             WMI_TAG_BCN_PRB_INFO) |
3861                                  FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3862         probe_info->caps = 0;
3863         probe_info->erp = 0;
3864
3865         ptr += sizeof(*probe_info);
3866
3867         tlv = ptr;
3868         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
3869                       FIELD_PREP(WMI_TLV_LEN, aligned_len);
3870         memcpy(tlv->value, tmpl->data, tmpl->len);
3871
3872         ret = ath11k_wmi_cmd_send(ar->wmi, skb, WMI_PRB_TMPL_CMDID);
3873         if (ret) {
3874                 ath11k_warn(ar->ab,
3875                             "WMI vdev %i failed to send probe response template command\n",
3876                             vdev_id);
3877                 dev_kfree_skb(skb);
3878                 return ret;
3879         }
3880
3881         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd ");
3882
3883         return 0;
3884 }
3885
3886 int ath11k_wmi_fils_discovery(struct ath11k *ar, u32 vdev_id, u32 interval,
3887                               bool unsol_bcast_probe_resp_enabled)
3888 {
3889         struct sk_buff *skb;
3890         int ret, len;
3891         struct wmi_fils_discovery_cmd *cmd;
3892
3893         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
3894                    "vdev %i set %s interval to %u TU\n",
3895                    vdev_id, unsol_bcast_probe_resp_enabled ?
3896                    "unsolicited broadcast probe response" : "FILS discovery",
3897                    interval);
3898
3899         len = sizeof(*cmd);
3900         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
3901         if (!skb)
3902                 return -ENOMEM;
3903
3904         cmd = (struct wmi_fils_discovery_cmd *)skb->data;
3905         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ENABLE_FILS_CMD) |
3906                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
3907         cmd->vdev_id = vdev_id;
3908         cmd->interval = interval;
3909         cmd->config = unsol_bcast_probe_resp_enabled;
3910
3911         ret = ath11k_wmi_cmd_send(ar->wmi, skb, WMI_ENABLE_FILS_CMDID);
3912         if (ret) {
3913                 ath11k_warn(ar->ab,
3914                             "WMI vdev %i failed to send FILS discovery enable/disable command\n",
3915                             vdev_id);
3916                 dev_kfree_skb(skb);
3917                 return ret;
3918         }
3919
3920         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd enable fils");
3921
3922         return 0;
3923 }
3924
3925 static void
3926 ath11k_wmi_obss_color_collision_event(struct ath11k_base *ab, struct sk_buff *skb)
3927 {
3928         const void **tb;
3929         const struct wmi_obss_color_collision_event *ev;
3930         struct ath11k_vif *arvif;
3931         int ret;
3932
3933         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
3934         if (IS_ERR(tb)) {
3935                 ret = PTR_ERR(tb);
3936                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
3937                 return;
3938         }
3939
3940         ath11k_dbg(ab, ATH11K_DBG_WMI, "event obss color collision");
3941
3942         rcu_read_lock();
3943
3944         ev = tb[WMI_TAG_OBSS_COLOR_COLLISION_EVT];
3945         if (!ev) {
3946                 ath11k_warn(ab, "failed to fetch obss color collision ev");
3947                 goto exit;
3948         }
3949
3950         arvif = ath11k_mac_get_arvif_by_vdev_id(ab, ev->vdev_id);
3951         if (!arvif) {
3952                 ath11k_warn(ab, "failed to find arvif with vedv id %d in obss_color_collision_event\n",
3953                             ev->vdev_id);
3954                 goto exit;
3955         }
3956
3957         switch (ev->evt_type) {
3958         case WMI_BSS_COLOR_COLLISION_DETECTION:
3959                 ieee80211_obss_color_collision_notify(arvif->vif, ev->obss_color_bitmap,
3960                                                       GFP_KERNEL);
3961                 ath11k_dbg(ab, ATH11K_DBG_WMI,
3962                            "OBSS color collision detected vdev:%d, event:%d, bitmap:%08llx\n",
3963                            ev->vdev_id, ev->evt_type, ev->obss_color_bitmap);
3964                 break;
3965         case WMI_BSS_COLOR_COLLISION_DISABLE:
3966         case WMI_BSS_COLOR_FREE_SLOT_TIMER_EXPIRY:
3967         case WMI_BSS_COLOR_FREE_SLOT_AVAILABLE:
3968                 break;
3969         default:
3970                 ath11k_warn(ab, "received unknown obss color collision detection event\n");
3971         }
3972
3973 exit:
3974         kfree(tb);
3975         rcu_read_unlock();
3976 }
3977
3978 static void
3979 ath11k_fill_band_to_mac_param(struct ath11k_base  *soc,
3980                               struct wmi_host_pdev_band_to_mac *band_to_mac)
3981 {
3982         u8 i;
3983         struct ath11k_hal_reg_capabilities_ext *hal_reg_cap;
3984         struct ath11k_pdev *pdev;
3985
3986         for (i = 0; i < soc->num_radios; i++) {
3987                 pdev = &soc->pdevs[i];
3988                 hal_reg_cap = &soc->hal_reg_cap[i];
3989                 band_to_mac[i].pdev_id = pdev->pdev_id;
3990
3991                 switch (pdev->cap.supported_bands) {
3992                 case WMI_HOST_WLAN_2G_5G_CAP:
3993                         band_to_mac[i].start_freq = hal_reg_cap->low_2ghz_chan;
3994                         band_to_mac[i].end_freq = hal_reg_cap->high_5ghz_chan;
3995                         break;
3996                 case WMI_HOST_WLAN_2G_CAP:
3997                         band_to_mac[i].start_freq = hal_reg_cap->low_2ghz_chan;
3998                         band_to_mac[i].end_freq = hal_reg_cap->high_2ghz_chan;
3999                         break;
4000                 case WMI_HOST_WLAN_5G_CAP:
4001                         band_to_mac[i].start_freq = hal_reg_cap->low_5ghz_chan;
4002                         band_to_mac[i].end_freq = hal_reg_cap->high_5ghz_chan;
4003                         break;
4004                 default:
4005                         break;
4006                 }
4007         }
4008 }
4009
4010 static void
4011 ath11k_wmi_copy_resource_config(struct wmi_resource_config *wmi_cfg,
4012                                 struct target_resource_config *tg_cfg)
4013 {
4014         wmi_cfg->num_vdevs = tg_cfg->num_vdevs;
4015         wmi_cfg->num_peers = tg_cfg->num_peers;
4016         wmi_cfg->num_offload_peers = tg_cfg->num_offload_peers;
4017         wmi_cfg->num_offload_reorder_buffs = tg_cfg->num_offload_reorder_buffs;
4018         wmi_cfg->num_peer_keys = tg_cfg->num_peer_keys;
4019         wmi_cfg->num_tids = tg_cfg->num_tids;
4020         wmi_cfg->ast_skid_limit = tg_cfg->ast_skid_limit;
4021         wmi_cfg->tx_chain_mask = tg_cfg->tx_chain_mask;
4022         wmi_cfg->rx_chain_mask = tg_cfg->rx_chain_mask;
4023         wmi_cfg->rx_timeout_pri[0] = tg_cfg->rx_timeout_pri[0];
4024         wmi_cfg->rx_timeout_pri[1] = tg_cfg->rx_timeout_pri[1];
4025         wmi_cfg->rx_timeout_pri[2] = tg_cfg->rx_timeout_pri[2];
4026         wmi_cfg->rx_timeout_pri[3] = tg_cfg->rx_timeout_pri[3];
4027         wmi_cfg->rx_decap_mode = tg_cfg->rx_decap_mode;
4028         wmi_cfg->scan_max_pending_req = tg_cfg->scan_max_pending_req;
4029         wmi_cfg->bmiss_offload_max_vdev = tg_cfg->bmiss_offload_max_vdev;
4030         wmi_cfg->roam_offload_max_vdev = tg_cfg->roam_offload_max_vdev;
4031         wmi_cfg->roam_offload_max_ap_profiles =
4032                 tg_cfg->roam_offload_max_ap_profiles;
4033         wmi_cfg->num_mcast_groups = tg_cfg->num_mcast_groups;
4034         wmi_cfg->num_mcast_table_elems = tg_cfg->num_mcast_table_elems;
4035         wmi_cfg->mcast2ucast_mode = tg_cfg->mcast2ucast_mode;
4036         wmi_cfg->tx_dbg_log_size = tg_cfg->tx_dbg_log_size;
4037         wmi_cfg->num_wds_entries = tg_cfg->num_wds_entries;
4038         wmi_cfg->dma_burst_size = tg_cfg->dma_burst_size;
4039         wmi_cfg->mac_aggr_delim = tg_cfg->mac_aggr_delim;
4040         wmi_cfg->rx_skip_defrag_timeout_dup_detection_check =
4041                 tg_cfg->rx_skip_defrag_timeout_dup_detection_check;
4042         wmi_cfg->vow_config = tg_cfg->vow_config;
4043         wmi_cfg->gtk_offload_max_vdev = tg_cfg->gtk_offload_max_vdev;
4044         wmi_cfg->num_msdu_desc = tg_cfg->num_msdu_desc;
4045         wmi_cfg->max_frag_entries = tg_cfg->max_frag_entries;
4046         wmi_cfg->num_tdls_vdevs = tg_cfg->num_tdls_vdevs;
4047         wmi_cfg->num_tdls_conn_table_entries =
4048                 tg_cfg->num_tdls_conn_table_entries;
4049         wmi_cfg->beacon_tx_offload_max_vdev =
4050                 tg_cfg->beacon_tx_offload_max_vdev;
4051         wmi_cfg->num_multicast_filter_entries =
4052                 tg_cfg->num_multicast_filter_entries;
4053         wmi_cfg->num_wow_filters = tg_cfg->num_wow_filters;
4054         wmi_cfg->num_keep_alive_pattern = tg_cfg->num_keep_alive_pattern;
4055         wmi_cfg->keep_alive_pattern_size = tg_cfg->keep_alive_pattern_size;
4056         wmi_cfg->max_tdls_concurrent_sleep_sta =
4057                 tg_cfg->max_tdls_concurrent_sleep_sta;
4058         wmi_cfg->max_tdls_concurrent_buffer_sta =
4059                 tg_cfg->max_tdls_concurrent_buffer_sta;
4060         wmi_cfg->wmi_send_separate = tg_cfg->wmi_send_separate;
4061         wmi_cfg->num_ocb_vdevs = tg_cfg->num_ocb_vdevs;
4062         wmi_cfg->num_ocb_channels = tg_cfg->num_ocb_channels;
4063         wmi_cfg->num_ocb_schedules = tg_cfg->num_ocb_schedules;
4064         wmi_cfg->bpf_instruction_size = tg_cfg->bpf_instruction_size;
4065         wmi_cfg->max_bssid_rx_filters = tg_cfg->max_bssid_rx_filters;
4066         wmi_cfg->use_pdev_id = tg_cfg->use_pdev_id;
4067         wmi_cfg->flag1 = tg_cfg->flag1;
4068         wmi_cfg->peer_map_unmap_v2_support = tg_cfg->peer_map_unmap_v2_support;
4069         wmi_cfg->sched_params = tg_cfg->sched_params;
4070         wmi_cfg->twt_ap_pdev_count = tg_cfg->twt_ap_pdev_count;
4071         wmi_cfg->twt_ap_sta_count = tg_cfg->twt_ap_sta_count;
4072         wmi_cfg->host_service_flags &=
4073                 ~(1 << WMI_CFG_HOST_SERVICE_FLAG_REG_CC_EXT);
4074         wmi_cfg->host_service_flags |= (tg_cfg->is_reg_cc_ext_event_supported <<
4075                                         WMI_CFG_HOST_SERVICE_FLAG_REG_CC_EXT);
4076         wmi_cfg->flags2 = WMI_RSRC_CFG_FLAG2_CALC_NEXT_DTIM_COUNT_SET;
4077         wmi_cfg->ema_max_vap_cnt = tg_cfg->ema_max_vap_cnt;
4078         wmi_cfg->ema_max_profile_period = tg_cfg->ema_max_profile_period;
4079 }
4080
4081 static int ath11k_init_cmd_send(struct ath11k_pdev_wmi *wmi,
4082                                 struct wmi_init_cmd_param *param)
4083 {
4084         struct ath11k_base *ab = wmi->wmi_ab->ab;
4085         struct sk_buff *skb;
4086         struct wmi_init_cmd *cmd;
4087         struct wmi_resource_config *cfg;
4088         struct wmi_pdev_set_hw_mode_cmd_param *hw_mode;
4089         struct wmi_pdev_band_to_mac *band_to_mac;
4090         struct wlan_host_mem_chunk *host_mem_chunks;
4091         struct wmi_tlv *tlv;
4092         size_t ret, len;
4093         void *ptr;
4094         u32 hw_mode_len = 0;
4095         u16 idx;
4096
4097         if (param->hw_mode_id != WMI_HOST_HW_MODE_MAX)
4098                 hw_mode_len = sizeof(*hw_mode) + TLV_HDR_SIZE +
4099                               (param->num_band_to_mac * sizeof(*band_to_mac));
4100
4101         len = sizeof(*cmd) + TLV_HDR_SIZE + sizeof(*cfg) + hw_mode_len +
4102               (param->num_mem_chunks ? (sizeof(*host_mem_chunks) * WMI_MAX_MEM_REQS) : 0);
4103
4104         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
4105         if (!skb)
4106                 return -ENOMEM;
4107
4108         cmd = (struct wmi_init_cmd *)skb->data;
4109
4110         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_INIT_CMD) |
4111                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
4112
4113         ptr = skb->data + sizeof(*cmd);
4114         cfg = ptr;
4115
4116         ath11k_wmi_copy_resource_config(cfg, param->res_cfg);
4117
4118         cfg->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_RESOURCE_CONFIG) |
4119                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cfg) - TLV_HDR_SIZE);
4120
4121         ptr += sizeof(*cfg);
4122         host_mem_chunks = ptr + TLV_HDR_SIZE;
4123         len = sizeof(struct wlan_host_mem_chunk);
4124
4125         for (idx = 0; idx < param->num_mem_chunks; ++idx) {
4126                 host_mem_chunks[idx].tlv_header =
4127                                 FIELD_PREP(WMI_TLV_TAG,
4128                                            WMI_TAG_WLAN_HOST_MEMORY_CHUNK) |
4129                                 FIELD_PREP(WMI_TLV_LEN, len);
4130
4131                 host_mem_chunks[idx].ptr = param->mem_chunks[idx].paddr;
4132                 host_mem_chunks[idx].size = param->mem_chunks[idx].len;
4133                 host_mem_chunks[idx].req_id = param->mem_chunks[idx].req_id;
4134
4135                 ath11k_dbg(ab, ATH11K_DBG_WMI,
4136                            "host mem chunk req_id %d paddr 0x%llx len %d\n",
4137                            param->mem_chunks[idx].req_id,
4138                            (u64)param->mem_chunks[idx].paddr,
4139                            param->mem_chunks[idx].len);
4140         }
4141         cmd->num_host_mem_chunks = param->num_mem_chunks;
4142         len = sizeof(struct wlan_host_mem_chunk) * param->num_mem_chunks;
4143
4144         /* num_mem_chunks is zero */
4145         tlv = ptr;
4146         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
4147                       FIELD_PREP(WMI_TLV_LEN, len);
4148         ptr += TLV_HDR_SIZE + len;
4149
4150         if (param->hw_mode_id != WMI_HOST_HW_MODE_MAX) {
4151                 hw_mode = ptr;
4152                 hw_mode->tlv_header = FIELD_PREP(WMI_TLV_TAG,
4153                                                  WMI_TAG_PDEV_SET_HW_MODE_CMD) |
4154                                       FIELD_PREP(WMI_TLV_LEN,
4155                                                  sizeof(*hw_mode) - TLV_HDR_SIZE);
4156
4157                 hw_mode->hw_mode_index = param->hw_mode_id;
4158                 hw_mode->num_band_to_mac = param->num_band_to_mac;
4159
4160                 ptr += sizeof(*hw_mode);
4161
4162                 len = param->num_band_to_mac * sizeof(*band_to_mac);
4163                 tlv = ptr;
4164                 tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
4165                               FIELD_PREP(WMI_TLV_LEN, len);
4166
4167                 ptr += TLV_HDR_SIZE;
4168                 len = sizeof(*band_to_mac);
4169
4170                 for (idx = 0; idx < param->num_band_to_mac; idx++) {
4171                         band_to_mac = ptr;
4172
4173                         band_to_mac->tlv_header = FIELD_PREP(WMI_TLV_TAG,
4174                                                              WMI_TAG_PDEV_BAND_TO_MAC) |
4175                                                   FIELD_PREP(WMI_TLV_LEN,
4176                                                              len - TLV_HDR_SIZE);
4177                         band_to_mac->pdev_id = param->band_to_mac[idx].pdev_id;
4178                         band_to_mac->start_freq =
4179                                 param->band_to_mac[idx].start_freq;
4180                         band_to_mac->end_freq =
4181                                 param->band_to_mac[idx].end_freq;
4182                         ptr += sizeof(*band_to_mac);
4183                 }
4184         }
4185
4186         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_INIT_CMDID);
4187         if (ret) {
4188                 ath11k_warn(ab, "failed to send WMI_INIT_CMDID\n");
4189                 dev_kfree_skb(skb);
4190                 return ret;
4191         }
4192
4193         ath11k_dbg(ab, ATH11K_DBG_WMI, "cmd wmi init");
4194
4195         return 0;
4196 }
4197
4198 int ath11k_wmi_pdev_lro_cfg(struct ath11k *ar,
4199                             int pdev_id)
4200 {
4201         struct ath11k_wmi_pdev_lro_config_cmd *cmd;
4202         struct sk_buff *skb;
4203         int ret;
4204
4205         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, sizeof(*cmd));
4206         if (!skb)
4207                 return -ENOMEM;
4208
4209         cmd = (struct ath11k_wmi_pdev_lro_config_cmd *)skb->data;
4210         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_LRO_INFO_CMD) |
4211                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
4212
4213         get_random_bytes(cmd->th_4, sizeof(uint32_t) * ATH11K_IPV4_TH_SEED_SIZE);
4214         get_random_bytes(cmd->th_6, sizeof(uint32_t) * ATH11K_IPV6_TH_SEED_SIZE);
4215
4216         cmd->pdev_id = pdev_id;
4217
4218         ret = ath11k_wmi_cmd_send(ar->wmi, skb, WMI_LRO_CONFIG_CMDID);
4219         if (ret) {
4220                 ath11k_warn(ar->ab,
4221                             "failed to send lro cfg req wmi cmd\n");
4222                 goto err;
4223         }
4224
4225         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
4226                    "cmd lro config pdev_id 0x%x\n", pdev_id);
4227         return 0;
4228 err:
4229         dev_kfree_skb(skb);
4230         return ret;
4231 }
4232
4233 int ath11k_wmi_wait_for_service_ready(struct ath11k_base *ab)
4234 {
4235         unsigned long time_left;
4236
4237         time_left = wait_for_completion_timeout(&ab->wmi_ab.service_ready,
4238                                                 WMI_SERVICE_READY_TIMEOUT_HZ);
4239         if (!time_left)
4240                 return -ETIMEDOUT;
4241
4242         return 0;
4243 }
4244
4245 int ath11k_wmi_wait_for_unified_ready(struct ath11k_base *ab)
4246 {
4247         unsigned long time_left;
4248
4249         time_left = wait_for_completion_timeout(&ab->wmi_ab.unified_ready,
4250                                                 WMI_SERVICE_READY_TIMEOUT_HZ);
4251         if (!time_left)
4252                 return -ETIMEDOUT;
4253
4254         return 0;
4255 }
4256
4257 int ath11k_wmi_set_hw_mode(struct ath11k_base *ab,
4258                            enum wmi_host_hw_mode_config_type mode)
4259 {
4260         struct wmi_pdev_set_hw_mode_cmd_param *cmd;
4261         struct sk_buff *skb;
4262         struct ath11k_wmi_base *wmi_ab = &ab->wmi_ab;
4263         int len;
4264         int ret;
4265
4266         len = sizeof(*cmd);
4267
4268         skb = ath11k_wmi_alloc_skb(wmi_ab, len);
4269         if (!skb)
4270                 return -ENOMEM;
4271
4272         cmd = (struct wmi_pdev_set_hw_mode_cmd_param *)skb->data;
4273
4274         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_SET_HW_MODE_CMD) |
4275                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
4276
4277         cmd->pdev_id = WMI_PDEV_ID_SOC;
4278         cmd->hw_mode_index = mode;
4279
4280         ret = ath11k_wmi_cmd_send(&wmi_ab->wmi[0], skb, WMI_PDEV_SET_HW_MODE_CMDID);
4281         if (ret) {
4282                 ath11k_warn(ab, "failed to send WMI_PDEV_SET_HW_MODE_CMDID\n");
4283                 dev_kfree_skb(skb);
4284                 return ret;
4285         }
4286
4287         ath11k_dbg(ab, ATH11K_DBG_WMI, "cmd pdev set hw mode %d", cmd->hw_mode_index);
4288
4289         return 0;
4290 }
4291
4292 int ath11k_wmi_cmd_init(struct ath11k_base *ab)
4293 {
4294         struct ath11k_wmi_base *wmi_ab = &ab->wmi_ab;
4295         struct wmi_init_cmd_param init_param;
4296         struct target_resource_config  config;
4297
4298         memset(&init_param, 0, sizeof(init_param));
4299         memset(&config, 0, sizeof(config));
4300
4301         ab->hw_params.hw_ops->wmi_init_config(ab, &config);
4302
4303         if (test_bit(WMI_TLV_SERVICE_REG_CC_EXT_EVENT_SUPPORT,
4304                      ab->wmi_ab.svc_map))
4305                 config.is_reg_cc_ext_event_supported = 1;
4306
4307         memcpy(&wmi_ab->wlan_resource_config, &config, sizeof(config));
4308
4309         init_param.res_cfg = &wmi_ab->wlan_resource_config;
4310         init_param.num_mem_chunks = wmi_ab->num_mem_chunks;
4311         init_param.hw_mode_id = wmi_ab->preferred_hw_mode;
4312         init_param.mem_chunks = wmi_ab->mem_chunks;
4313
4314         if (ab->hw_params.single_pdev_only)
4315                 init_param.hw_mode_id = WMI_HOST_HW_MODE_MAX;
4316
4317         init_param.num_band_to_mac = ab->num_radios;
4318         ath11k_fill_band_to_mac_param(ab, init_param.band_to_mac);
4319
4320         return ath11k_init_cmd_send(&wmi_ab->wmi[0], &init_param);
4321 }
4322
4323 int ath11k_wmi_vdev_spectral_conf(struct ath11k *ar,
4324                                   struct ath11k_wmi_vdev_spectral_conf_param *param)
4325 {
4326         struct ath11k_wmi_vdev_spectral_conf_cmd *cmd;
4327         struct sk_buff *skb;
4328         int ret;
4329
4330         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, sizeof(*cmd));
4331         if (!skb)
4332                 return -ENOMEM;
4333
4334         cmd = (struct ath11k_wmi_vdev_spectral_conf_cmd *)skb->data;
4335         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
4336                                      WMI_TAG_VDEV_SPECTRAL_CONFIGURE_CMD) |
4337                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
4338
4339         memcpy(&cmd->param, param, sizeof(*param));
4340
4341         ret = ath11k_wmi_cmd_send(ar->wmi, skb,
4342                                   WMI_VDEV_SPECTRAL_SCAN_CONFIGURE_CMDID);
4343         if (ret) {
4344                 ath11k_warn(ar->ab,
4345                             "failed to send spectral scan config wmi cmd\n");
4346                 goto err;
4347         }
4348
4349         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
4350                    "cmd vdev spectral scan configure vdev_id 0x%x\n",
4351                    param->vdev_id);
4352
4353         return 0;
4354 err:
4355         dev_kfree_skb(skb);
4356         return ret;
4357 }
4358
4359 int ath11k_wmi_vdev_spectral_enable(struct ath11k *ar, u32 vdev_id,
4360                                     u32 trigger, u32 enable)
4361 {
4362         struct ath11k_wmi_vdev_spectral_enable_cmd *cmd;
4363         struct sk_buff *skb;
4364         int ret;
4365
4366         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, sizeof(*cmd));
4367         if (!skb)
4368                 return -ENOMEM;
4369
4370         cmd = (struct ath11k_wmi_vdev_spectral_enable_cmd *)skb->data;
4371         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
4372                                      WMI_TAG_VDEV_SPECTRAL_ENABLE_CMD) |
4373                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
4374
4375         cmd->vdev_id = vdev_id;
4376         cmd->trigger_cmd = trigger;
4377         cmd->enable_cmd = enable;
4378
4379         ret = ath11k_wmi_cmd_send(ar->wmi, skb,
4380                                   WMI_VDEV_SPECTRAL_SCAN_ENABLE_CMDID);
4381         if (ret) {
4382                 ath11k_warn(ar->ab,
4383                             "failed to send spectral enable wmi cmd\n");
4384                 goto err;
4385         }
4386
4387         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
4388                    "cmd vdev spectral scan enable vdev id 0x%x\n",
4389                    vdev_id);
4390
4391         return 0;
4392 err:
4393         dev_kfree_skb(skb);
4394         return ret;
4395 }
4396
4397 int ath11k_wmi_pdev_dma_ring_cfg(struct ath11k *ar,
4398                                  struct ath11k_wmi_pdev_dma_ring_cfg_req_cmd *param)
4399 {
4400         struct ath11k_wmi_pdev_dma_ring_cfg_req_cmd *cmd;
4401         struct sk_buff *skb;
4402         int ret;
4403
4404         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, sizeof(*cmd));
4405         if (!skb)
4406                 return -ENOMEM;
4407
4408         cmd = (struct ath11k_wmi_pdev_dma_ring_cfg_req_cmd *)skb->data;
4409         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_DMA_RING_CFG_REQ) |
4410                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
4411
4412         cmd->pdev_id            = param->pdev_id;
4413         cmd->module_id          = param->module_id;
4414         cmd->base_paddr_lo      = param->base_paddr_lo;
4415         cmd->base_paddr_hi      = param->base_paddr_hi;
4416         cmd->head_idx_paddr_lo  = param->head_idx_paddr_lo;
4417         cmd->head_idx_paddr_hi  = param->head_idx_paddr_hi;
4418         cmd->tail_idx_paddr_lo  = param->tail_idx_paddr_lo;
4419         cmd->tail_idx_paddr_hi  = param->tail_idx_paddr_hi;
4420         cmd->num_elems          = param->num_elems;
4421         cmd->buf_size           = param->buf_size;
4422         cmd->num_resp_per_event = param->num_resp_per_event;
4423         cmd->event_timeout_ms   = param->event_timeout_ms;
4424
4425         ret = ath11k_wmi_cmd_send(ar->wmi, skb,
4426                                   WMI_PDEV_DMA_RING_CFG_REQ_CMDID);
4427         if (ret) {
4428                 ath11k_warn(ar->ab,
4429                             "failed to send dma ring cfg req wmi cmd\n");
4430                 goto err;
4431         }
4432
4433         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
4434                    "cmd pdev dma ring cfg req pdev_id 0x%x\n",
4435                    param->pdev_id);
4436
4437         return 0;
4438 err:
4439         dev_kfree_skb(skb);
4440         return ret;
4441 }
4442
4443 static int ath11k_wmi_tlv_dma_buf_entry_parse(struct ath11k_base *soc,
4444                                               u16 tag, u16 len,
4445                                               const void *ptr, void *data)
4446 {
4447         struct wmi_tlv_dma_buf_release_parse *parse = data;
4448
4449         if (tag != WMI_TAG_DMA_BUF_RELEASE_ENTRY)
4450                 return -EPROTO;
4451
4452         if (parse->num_buf_entry >= parse->fixed.num_buf_release_entry)
4453                 return -ENOBUFS;
4454
4455         parse->num_buf_entry++;
4456         return 0;
4457 }
4458
4459 static int ath11k_wmi_tlv_dma_buf_meta_parse(struct ath11k_base *soc,
4460                                              u16 tag, u16 len,
4461                                              const void *ptr, void *data)
4462 {
4463         struct wmi_tlv_dma_buf_release_parse *parse = data;
4464
4465         if (tag != WMI_TAG_DMA_BUF_RELEASE_SPECTRAL_META_DATA)
4466                 return -EPROTO;
4467
4468         if (parse->num_meta >= parse->fixed.num_meta_data_entry)
4469                 return -ENOBUFS;
4470
4471         parse->num_meta++;
4472         return 0;
4473 }
4474
4475 static int ath11k_wmi_tlv_dma_buf_parse(struct ath11k_base *ab,
4476                                         u16 tag, u16 len,
4477                                         const void *ptr, void *data)
4478 {
4479         struct wmi_tlv_dma_buf_release_parse *parse = data;
4480         int ret;
4481
4482         switch (tag) {
4483         case WMI_TAG_DMA_BUF_RELEASE:
4484                 memcpy(&parse->fixed, ptr,
4485                        sizeof(struct ath11k_wmi_dma_buf_release_fixed_param));
4486                 parse->fixed.pdev_id = DP_HW2SW_MACID(parse->fixed.pdev_id);
4487                 break;
4488         case WMI_TAG_ARRAY_STRUCT:
4489                 if (!parse->buf_entry_done) {
4490                         parse->num_buf_entry = 0;
4491                         parse->buf_entry = (struct wmi_dma_buf_release_entry *)ptr;
4492
4493                         ret = ath11k_wmi_tlv_iter(ab, ptr, len,
4494                                                   ath11k_wmi_tlv_dma_buf_entry_parse,
4495                                                   parse);
4496                         if (ret) {
4497                                 ath11k_warn(ab, "failed to parse dma buf entry tlv %d\n",
4498                                             ret);
4499                                 return ret;
4500                         }
4501
4502                         parse->buf_entry_done = true;
4503                 } else if (!parse->meta_data_done) {
4504                         parse->num_meta = 0;
4505                         parse->meta_data = (struct wmi_dma_buf_release_meta_data *)ptr;
4506
4507                         ret = ath11k_wmi_tlv_iter(ab, ptr, len,
4508                                                   ath11k_wmi_tlv_dma_buf_meta_parse,
4509                                                   parse);
4510                         if (ret) {
4511                                 ath11k_warn(ab, "failed to parse dma buf meta tlv %d\n",
4512                                             ret);
4513                                 return ret;
4514                         }
4515
4516                         parse->meta_data_done = true;
4517                 }
4518                 break;
4519         default:
4520                 break;
4521         }
4522         return 0;
4523 }
4524
4525 static void ath11k_wmi_pdev_dma_ring_buf_release_event(struct ath11k_base *ab,
4526                                                        struct sk_buff *skb)
4527 {
4528         struct wmi_tlv_dma_buf_release_parse parse = { };
4529         struct ath11k_dbring_buf_release_event param;
4530         int ret;
4531
4532         ret = ath11k_wmi_tlv_iter(ab, skb->data, skb->len,
4533                                   ath11k_wmi_tlv_dma_buf_parse,
4534                                   &parse);
4535         if (ret) {
4536                 ath11k_warn(ab, "failed to parse dma buf release tlv %d\n", ret);
4537                 return;
4538         }
4539
4540         ath11k_dbg(ab, ATH11K_DBG_WMI, "event pdev dma ring buf release");
4541
4542         param.fixed             = parse.fixed;
4543         param.buf_entry         = parse.buf_entry;
4544         param.num_buf_entry     = parse.num_buf_entry;
4545         param.meta_data         = parse.meta_data;
4546         param.num_meta          = parse.num_meta;
4547
4548         ret = ath11k_dbring_buffer_release_event(ab, &param);
4549         if (ret) {
4550                 ath11k_warn(ab, "failed to handle dma buf release event %d\n", ret);
4551                 return;
4552         }
4553 }
4554
4555 static int ath11k_wmi_tlv_hw_mode_caps_parse(struct ath11k_base *soc,
4556                                              u16 tag, u16 len,
4557                                              const void *ptr, void *data)
4558 {
4559         struct wmi_tlv_svc_rdy_ext_parse *svc_rdy_ext = data;
4560         struct wmi_hw_mode_capabilities *hw_mode_cap;
4561         u32 phy_map = 0;
4562
4563         if (tag != WMI_TAG_HW_MODE_CAPABILITIES)
4564                 return -EPROTO;
4565
4566         if (svc_rdy_ext->n_hw_mode_caps >= svc_rdy_ext->param.num_hw_modes)
4567                 return -ENOBUFS;
4568
4569         hw_mode_cap = container_of(ptr, struct wmi_hw_mode_capabilities,
4570                                    hw_mode_id);
4571         svc_rdy_ext->n_hw_mode_caps++;
4572
4573         phy_map = hw_mode_cap->phy_id_map;
4574         while (phy_map) {
4575                 svc_rdy_ext->tot_phy_id++;
4576                 phy_map = phy_map >> 1;
4577         }
4578
4579         return 0;
4580 }
4581
4582 static int ath11k_wmi_tlv_hw_mode_caps(struct ath11k_base *soc,
4583                                        u16 len, const void *ptr, void *data)
4584 {
4585         struct wmi_tlv_svc_rdy_ext_parse *svc_rdy_ext = data;
4586         struct wmi_hw_mode_capabilities *hw_mode_caps;
4587         enum wmi_host_hw_mode_config_type mode, pref;
4588         u32 i;
4589         int ret;
4590
4591         svc_rdy_ext->n_hw_mode_caps = 0;
4592         svc_rdy_ext->hw_mode_caps = (struct wmi_hw_mode_capabilities *)ptr;
4593
4594         ret = ath11k_wmi_tlv_iter(soc, ptr, len,
4595                                   ath11k_wmi_tlv_hw_mode_caps_parse,
4596                                   svc_rdy_ext);
4597         if (ret) {
4598                 ath11k_warn(soc, "failed to parse tlv %d\n", ret);
4599                 return ret;
4600         }
4601
4602         i = 0;
4603         while (i < svc_rdy_ext->n_hw_mode_caps) {
4604                 hw_mode_caps = &svc_rdy_ext->hw_mode_caps[i];
4605                 mode = hw_mode_caps->hw_mode_id;
4606                 pref = soc->wmi_ab.preferred_hw_mode;
4607
4608                 if (ath11k_hw_mode_pri_map[mode] < ath11k_hw_mode_pri_map[pref]) {
4609                         svc_rdy_ext->pref_hw_mode_caps = *hw_mode_caps;
4610                         soc->wmi_ab.preferred_hw_mode = mode;
4611                 }
4612                 i++;
4613         }
4614
4615         ath11k_dbg(soc, ATH11K_DBG_WMI, "preferred_hw_mode:%d\n",
4616                    soc->wmi_ab.preferred_hw_mode);
4617         if (soc->wmi_ab.preferred_hw_mode == WMI_HOST_HW_MODE_MAX)
4618                 return -EINVAL;
4619
4620         return 0;
4621 }
4622
4623 static int ath11k_wmi_tlv_mac_phy_caps_parse(struct ath11k_base *soc,
4624                                              u16 tag, u16 len,
4625                                              const void *ptr, void *data)
4626 {
4627         struct wmi_tlv_svc_rdy_ext_parse *svc_rdy_ext = data;
4628
4629         if (tag != WMI_TAG_MAC_PHY_CAPABILITIES)
4630                 return -EPROTO;
4631
4632         if (svc_rdy_ext->n_mac_phy_caps >= svc_rdy_ext->tot_phy_id)
4633                 return -ENOBUFS;
4634
4635         len = min_t(u16, len, sizeof(struct wmi_mac_phy_capabilities));
4636         if (!svc_rdy_ext->n_mac_phy_caps) {
4637                 svc_rdy_ext->mac_phy_caps = kcalloc(svc_rdy_ext->tot_phy_id,
4638                                                     len, GFP_ATOMIC);
4639                 if (!svc_rdy_ext->mac_phy_caps)
4640                         return -ENOMEM;
4641         }
4642
4643         memcpy(svc_rdy_ext->mac_phy_caps + svc_rdy_ext->n_mac_phy_caps, ptr, len);
4644         svc_rdy_ext->n_mac_phy_caps++;
4645         return 0;
4646 }
4647
4648 static int ath11k_wmi_tlv_ext_hal_reg_caps_parse(struct ath11k_base *soc,
4649                                                  u16 tag, u16 len,
4650                                                  const void *ptr, void *data)
4651 {
4652         struct wmi_tlv_svc_rdy_ext_parse *svc_rdy_ext = data;
4653
4654         if (tag != WMI_TAG_HAL_REG_CAPABILITIES_EXT)
4655                 return -EPROTO;
4656
4657         if (svc_rdy_ext->n_ext_hal_reg_caps >= svc_rdy_ext->param.num_phy)
4658                 return -ENOBUFS;
4659
4660         svc_rdy_ext->n_ext_hal_reg_caps++;
4661         return 0;
4662 }
4663
4664 static int ath11k_wmi_tlv_ext_hal_reg_caps(struct ath11k_base *soc,
4665                                            u16 len, const void *ptr, void *data)
4666 {
4667         struct ath11k_pdev_wmi *wmi_handle = &soc->wmi_ab.wmi[0];
4668         struct wmi_tlv_svc_rdy_ext_parse *svc_rdy_ext = data;
4669         struct ath11k_hal_reg_capabilities_ext reg_cap;
4670         int ret;
4671         u32 i;
4672
4673         svc_rdy_ext->n_ext_hal_reg_caps = 0;
4674         svc_rdy_ext->ext_hal_reg_caps = (struct wmi_hal_reg_capabilities_ext *)ptr;
4675         ret = ath11k_wmi_tlv_iter(soc, ptr, len,
4676                                   ath11k_wmi_tlv_ext_hal_reg_caps_parse,
4677                                   svc_rdy_ext);
4678         if (ret) {
4679                 ath11k_warn(soc, "failed to parse tlv %d\n", ret);
4680                 return ret;
4681         }
4682
4683         for (i = 0; i < svc_rdy_ext->param.num_phy; i++) {
4684                 ret = ath11k_pull_reg_cap_svc_rdy_ext(wmi_handle,
4685                                                       svc_rdy_ext->soc_hal_reg_caps,
4686                                                       svc_rdy_ext->ext_hal_reg_caps, i,
4687                                                       &reg_cap);
4688                 if (ret) {
4689                         ath11k_warn(soc, "failed to extract reg cap %d\n", i);
4690                         return ret;
4691                 }
4692
4693                 memcpy(&soc->hal_reg_cap[reg_cap.phy_id],
4694                        &reg_cap, sizeof(reg_cap));
4695         }
4696         return 0;
4697 }
4698
4699 static int ath11k_wmi_tlv_ext_soc_hal_reg_caps_parse(struct ath11k_base *soc,
4700                                                      u16 len, const void *ptr,
4701                                                      void *data)
4702 {
4703         struct ath11k_pdev_wmi *wmi_handle = &soc->wmi_ab.wmi[0];
4704         struct wmi_tlv_svc_rdy_ext_parse *svc_rdy_ext = data;
4705         u8 hw_mode_id = svc_rdy_ext->pref_hw_mode_caps.hw_mode_id;
4706         u32 phy_id_map;
4707         int pdev_index = 0;
4708         int ret;
4709
4710         svc_rdy_ext->soc_hal_reg_caps = (struct wmi_soc_hal_reg_capabilities *)ptr;
4711         svc_rdy_ext->param.num_phy = svc_rdy_ext->soc_hal_reg_caps->num_phy;
4712
4713         soc->num_radios = 0;
4714         soc->target_pdev_count = 0;
4715         phy_id_map = svc_rdy_ext->pref_hw_mode_caps.phy_id_map;
4716
4717         while (phy_id_map && soc->num_radios < MAX_RADIOS) {
4718                 ret = ath11k_pull_mac_phy_cap_svc_ready_ext(wmi_handle,
4719                                                             svc_rdy_ext->hw_caps,
4720                                                             svc_rdy_ext->hw_mode_caps,
4721                                                             svc_rdy_ext->soc_hal_reg_caps,
4722                                                             svc_rdy_ext->mac_phy_caps,
4723                                                             hw_mode_id, soc->num_radios,
4724                                                             &soc->pdevs[pdev_index]);
4725                 if (ret) {
4726                         ath11k_warn(soc, "failed to extract mac caps, idx :%d\n",
4727                                     soc->num_radios);
4728                         return ret;
4729                 }
4730
4731                 soc->num_radios++;
4732
4733                 /* For QCA6390, save mac_phy capability in the same pdev */
4734                 if (soc->hw_params.single_pdev_only)
4735                         pdev_index = 0;
4736                 else
4737                         pdev_index = soc->num_radios;
4738
4739                 /* TODO: mac_phy_cap prints */
4740                 phy_id_map >>= 1;
4741         }
4742
4743         /* For QCA6390, set num_radios to 1 because host manages
4744          * both 2G and 5G radio in one pdev.
4745          * Set pdev_id = 0 and 0 means soc level.
4746          */
4747         if (soc->hw_params.single_pdev_only) {
4748                 soc->num_radios = 1;
4749                 soc->pdevs[0].pdev_id = 0;
4750         }
4751
4752         if (!soc->reg_info_store) {
4753                 soc->reg_info_store = kcalloc(soc->num_radios,
4754                                               sizeof(*soc->reg_info_store),
4755                                               GFP_ATOMIC);
4756                 if (!soc->reg_info_store)
4757                         return -ENOMEM;
4758         }
4759
4760         return 0;
4761 }
4762
4763 static int ath11k_wmi_tlv_dma_ring_caps_parse(struct ath11k_base *soc,
4764                                               u16 tag, u16 len,
4765                                               const void *ptr, void *data)
4766 {
4767         struct wmi_tlv_dma_ring_caps_parse *parse = data;
4768
4769         if (tag != WMI_TAG_DMA_RING_CAPABILITIES)
4770                 return -EPROTO;
4771
4772         parse->n_dma_ring_caps++;
4773         return 0;
4774 }
4775
4776 static int ath11k_wmi_alloc_dbring_caps(struct ath11k_base *ab,
4777                                         u32 num_cap)
4778 {
4779         size_t sz;
4780         void *ptr;
4781
4782         sz = num_cap * sizeof(struct ath11k_dbring_cap);
4783         ptr = kzalloc(sz, GFP_ATOMIC);
4784         if (!ptr)
4785                 return -ENOMEM;
4786
4787         ab->db_caps = ptr;
4788         ab->num_db_cap = num_cap;
4789
4790         return 0;
4791 }
4792
4793 static void ath11k_wmi_free_dbring_caps(struct ath11k_base *ab)
4794 {
4795         kfree(ab->db_caps);
4796         ab->db_caps = NULL;
4797 }
4798
4799 static int ath11k_wmi_tlv_dma_ring_caps(struct ath11k_base *ab,
4800                                         u16 len, const void *ptr, void *data)
4801 {
4802         struct wmi_tlv_dma_ring_caps_parse *dma_caps_parse = data;
4803         struct wmi_dma_ring_capabilities *dma_caps;
4804         struct ath11k_dbring_cap *dir_buff_caps;
4805         int ret;
4806         u32 i;
4807
4808         dma_caps_parse->n_dma_ring_caps = 0;
4809         dma_caps = (struct wmi_dma_ring_capabilities *)ptr;
4810         ret = ath11k_wmi_tlv_iter(ab, ptr, len,
4811                                   ath11k_wmi_tlv_dma_ring_caps_parse,
4812                                   dma_caps_parse);
4813         if (ret) {
4814                 ath11k_warn(ab, "failed to parse dma ring caps tlv %d\n", ret);
4815                 return ret;
4816         }
4817
4818         if (!dma_caps_parse->n_dma_ring_caps)
4819                 return 0;
4820
4821         if (ab->num_db_cap) {
4822                 ath11k_warn(ab, "Already processed, so ignoring dma ring caps\n");
4823                 return 0;
4824         }
4825
4826         ret = ath11k_wmi_alloc_dbring_caps(ab, dma_caps_parse->n_dma_ring_caps);
4827         if (ret)
4828                 return ret;
4829
4830         dir_buff_caps = ab->db_caps;
4831         for (i = 0; i < dma_caps_parse->n_dma_ring_caps; i++) {
4832                 if (dma_caps[i].module_id >= WMI_DIRECT_BUF_MAX) {
4833                         ath11k_warn(ab, "Invalid module id %d\n", dma_caps[i].module_id);
4834                         ret = -EINVAL;
4835                         goto free_dir_buff;
4836                 }
4837
4838                 dir_buff_caps[i].id = dma_caps[i].module_id;
4839                 dir_buff_caps[i].pdev_id = DP_HW2SW_MACID(dma_caps[i].pdev_id);
4840                 dir_buff_caps[i].min_elem = dma_caps[i].min_elem;
4841                 dir_buff_caps[i].min_buf_sz = dma_caps[i].min_buf_sz;
4842                 dir_buff_caps[i].min_buf_align = dma_caps[i].min_buf_align;
4843         }
4844
4845         return 0;
4846
4847 free_dir_buff:
4848         ath11k_wmi_free_dbring_caps(ab);
4849         return ret;
4850 }
4851
4852 static int ath11k_wmi_tlv_svc_rdy_ext_parse(struct ath11k_base *ab,
4853                                             u16 tag, u16 len,
4854                                             const void *ptr, void *data)
4855 {
4856         struct ath11k_pdev_wmi *wmi_handle = &ab->wmi_ab.wmi[0];
4857         struct wmi_tlv_svc_rdy_ext_parse *svc_rdy_ext = data;
4858         int ret;
4859
4860         switch (tag) {
4861         case WMI_TAG_SERVICE_READY_EXT_EVENT:
4862                 ret = ath11k_pull_svc_ready_ext(wmi_handle, ptr,
4863                                                 &svc_rdy_ext->param);
4864                 if (ret) {
4865                         ath11k_warn(ab, "unable to extract ext params\n");
4866                         return ret;
4867                 }
4868                 break;
4869
4870         case WMI_TAG_SOC_MAC_PHY_HW_MODE_CAPS:
4871                 svc_rdy_ext->hw_caps = (struct wmi_soc_mac_phy_hw_mode_caps *)ptr;
4872                 svc_rdy_ext->param.num_hw_modes = svc_rdy_ext->hw_caps->num_hw_modes;
4873                 break;
4874
4875         case WMI_TAG_SOC_HAL_REG_CAPABILITIES:
4876                 ret = ath11k_wmi_tlv_ext_soc_hal_reg_caps_parse(ab, len, ptr,
4877                                                                 svc_rdy_ext);
4878                 if (ret)
4879                         return ret;
4880                 break;
4881
4882         case WMI_TAG_ARRAY_STRUCT:
4883                 if (!svc_rdy_ext->hw_mode_done) {
4884                         ret = ath11k_wmi_tlv_hw_mode_caps(ab, len, ptr,
4885                                                           svc_rdy_ext);
4886                         if (ret)
4887                                 return ret;
4888
4889                         svc_rdy_ext->hw_mode_done = true;
4890                 } else if (!svc_rdy_ext->mac_phy_done) {
4891                         svc_rdy_ext->n_mac_phy_caps = 0;
4892                         ret = ath11k_wmi_tlv_iter(ab, ptr, len,
4893                                                   ath11k_wmi_tlv_mac_phy_caps_parse,
4894                                                   svc_rdy_ext);
4895                         if (ret) {
4896                                 ath11k_warn(ab, "failed to parse tlv %d\n", ret);
4897                                 return ret;
4898                         }
4899
4900                         svc_rdy_ext->mac_phy_done = true;
4901                 } else if (!svc_rdy_ext->ext_hal_reg_done) {
4902                         ret = ath11k_wmi_tlv_ext_hal_reg_caps(ab, len, ptr,
4903                                                               svc_rdy_ext);
4904                         if (ret)
4905                                 return ret;
4906
4907                         svc_rdy_ext->ext_hal_reg_done = true;
4908                 } else if (!svc_rdy_ext->mac_phy_chainmask_combo_done) {
4909                         svc_rdy_ext->mac_phy_chainmask_combo_done = true;
4910                 } else if (!svc_rdy_ext->mac_phy_chainmask_cap_done) {
4911                         svc_rdy_ext->mac_phy_chainmask_cap_done = true;
4912                 } else if (!svc_rdy_ext->oem_dma_ring_cap_done) {
4913                         svc_rdy_ext->oem_dma_ring_cap_done = true;
4914                 } else if (!svc_rdy_ext->dma_ring_cap_done) {
4915                         ret = ath11k_wmi_tlv_dma_ring_caps(ab, len, ptr,
4916                                                            &svc_rdy_ext->dma_caps_parse);
4917                         if (ret)
4918                                 return ret;
4919
4920                         svc_rdy_ext->dma_ring_cap_done = true;
4921                 }
4922                 break;
4923
4924         default:
4925                 break;
4926         }
4927         return 0;
4928 }
4929
4930 static int ath11k_service_ready_ext_event(struct ath11k_base *ab,
4931                                           struct sk_buff *skb)
4932 {
4933         struct wmi_tlv_svc_rdy_ext_parse svc_rdy_ext = { };
4934         int ret;
4935
4936         ret = ath11k_wmi_tlv_iter(ab, skb->data, skb->len,
4937                                   ath11k_wmi_tlv_svc_rdy_ext_parse,
4938                                   &svc_rdy_ext);
4939         if (ret) {
4940                 ath11k_warn(ab, "failed to parse tlv %d\n", ret);
4941                 goto err;
4942         }
4943
4944         ath11k_dbg(ab, ATH11K_DBG_WMI, "event service ready ext");
4945
4946         if (!test_bit(WMI_TLV_SERVICE_EXT2_MSG, ab->wmi_ab.svc_map))
4947                 complete(&ab->wmi_ab.service_ready);
4948
4949         kfree(svc_rdy_ext.mac_phy_caps);
4950         return 0;
4951
4952 err:
4953         ath11k_wmi_free_dbring_caps(ab);
4954         return ret;
4955 }
4956
4957 static int ath11k_wmi_tlv_svc_rdy_ext2_parse(struct ath11k_base *ab,
4958                                              u16 tag, u16 len,
4959                                              const void *ptr, void *data)
4960 {
4961         struct wmi_tlv_svc_rdy_ext2_parse *parse = data;
4962         int ret;
4963
4964         switch (tag) {
4965         case WMI_TAG_ARRAY_STRUCT:
4966                 if (!parse->dma_ring_cap_done) {
4967                         ret = ath11k_wmi_tlv_dma_ring_caps(ab, len, ptr,
4968                                                            &parse->dma_caps_parse);
4969                         if (ret)
4970                                 return ret;
4971
4972                         parse->dma_ring_cap_done = true;
4973                 }
4974                 break;
4975         default:
4976                 break;
4977         }
4978
4979         return 0;
4980 }
4981
4982 static int ath11k_service_ready_ext2_event(struct ath11k_base *ab,
4983                                            struct sk_buff *skb)
4984 {
4985         struct wmi_tlv_svc_rdy_ext2_parse svc_rdy_ext2 = { };
4986         int ret;
4987
4988         ret = ath11k_wmi_tlv_iter(ab, skb->data, skb->len,
4989                                   ath11k_wmi_tlv_svc_rdy_ext2_parse,
4990                                   &svc_rdy_ext2);
4991         if (ret) {
4992                 ath11k_warn(ab, "failed to parse ext2 event tlv %d\n", ret);
4993                 goto err;
4994         }
4995
4996         ath11k_dbg(ab, ATH11K_DBG_WMI, "event service ready ext2");
4997
4998         complete(&ab->wmi_ab.service_ready);
4999
5000         return 0;
5001
5002 err:
5003         ath11k_wmi_free_dbring_caps(ab);
5004         return ret;
5005 }
5006
5007 static int ath11k_pull_vdev_start_resp_tlv(struct ath11k_base *ab, struct sk_buff *skb,
5008                                            struct wmi_vdev_start_resp_event *vdev_rsp)
5009 {
5010         const void **tb;
5011         const struct wmi_vdev_start_resp_event *ev;
5012         int ret;
5013
5014         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
5015         if (IS_ERR(tb)) {
5016                 ret = PTR_ERR(tb);
5017                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
5018                 return ret;
5019         }
5020
5021         ev = tb[WMI_TAG_VDEV_START_RESPONSE_EVENT];
5022         if (!ev) {
5023                 ath11k_warn(ab, "failed to fetch vdev start resp ev");
5024                 kfree(tb);
5025                 return -EPROTO;
5026         }
5027
5028         memset(vdev_rsp, 0, sizeof(*vdev_rsp));
5029
5030         vdev_rsp->vdev_id = ev->vdev_id;
5031         vdev_rsp->requestor_id = ev->requestor_id;
5032         vdev_rsp->resp_type = ev->resp_type;
5033         vdev_rsp->status = ev->status;
5034         vdev_rsp->chain_mask = ev->chain_mask;
5035         vdev_rsp->smps_mode = ev->smps_mode;
5036         vdev_rsp->mac_id = ev->mac_id;
5037         vdev_rsp->cfgd_tx_streams = ev->cfgd_tx_streams;
5038         vdev_rsp->cfgd_rx_streams = ev->cfgd_rx_streams;
5039
5040         kfree(tb);
5041         return 0;
5042 }
5043
5044 static void ath11k_print_reg_rule(struct ath11k_base *ab, const char *band,
5045                                   u32 num_reg_rules,
5046                                   struct cur_reg_rule *reg_rule_ptr)
5047 {
5048         struct cur_reg_rule *reg_rule = reg_rule_ptr;
5049         u32 count;
5050
5051         ath11k_dbg(ab, ATH11K_DBG_WMI, "number of reg rules in %s band: %d\n",
5052                    band, num_reg_rules);
5053
5054         for (count = 0; count < num_reg_rules; count++) {
5055                 ath11k_dbg(ab, ATH11K_DBG_WMI,
5056                            "reg rule %d: (%d - %d @ %d) (%d, %d) (FLAGS %d)\n",
5057                            count + 1, reg_rule->start_freq, reg_rule->end_freq,
5058                            reg_rule->max_bw, reg_rule->ant_gain,
5059                            reg_rule->reg_power, reg_rule->flags);
5060                 reg_rule++;
5061         }
5062 }
5063
5064 static struct cur_reg_rule
5065 *create_reg_rules_from_wmi(u32 num_reg_rules,
5066                            struct wmi_regulatory_rule_struct *wmi_reg_rule)
5067 {
5068         struct cur_reg_rule *reg_rule_ptr;
5069         u32 count;
5070
5071         reg_rule_ptr = kcalloc(num_reg_rules, sizeof(*reg_rule_ptr),
5072                                GFP_ATOMIC);
5073
5074         if (!reg_rule_ptr)
5075                 return NULL;
5076
5077         for (count = 0; count < num_reg_rules; count++) {
5078                 reg_rule_ptr[count].start_freq =
5079                         FIELD_GET(REG_RULE_START_FREQ,
5080                                   wmi_reg_rule[count].freq_info);
5081                 reg_rule_ptr[count].end_freq =
5082                         FIELD_GET(REG_RULE_END_FREQ,
5083                                   wmi_reg_rule[count].freq_info);
5084                 reg_rule_ptr[count].max_bw =
5085                         FIELD_GET(REG_RULE_MAX_BW,
5086                                   wmi_reg_rule[count].bw_pwr_info);
5087                 reg_rule_ptr[count].reg_power =
5088                         FIELD_GET(REG_RULE_REG_PWR,
5089                                   wmi_reg_rule[count].bw_pwr_info);
5090                 reg_rule_ptr[count].ant_gain =
5091                         FIELD_GET(REG_RULE_ANT_GAIN,
5092                                   wmi_reg_rule[count].bw_pwr_info);
5093                 reg_rule_ptr[count].flags =
5094                         FIELD_GET(REG_RULE_FLAGS,
5095                                   wmi_reg_rule[count].flag_info);
5096         }
5097
5098         return reg_rule_ptr;
5099 }
5100
5101 static int ath11k_pull_reg_chan_list_update_ev(struct ath11k_base *ab,
5102                                                struct sk_buff *skb,
5103                                                struct cur_regulatory_info *reg_info)
5104 {
5105         const void **tb;
5106         const struct wmi_reg_chan_list_cc_event *chan_list_event_hdr;
5107         struct wmi_regulatory_rule_struct *wmi_reg_rule;
5108         u32 num_2ghz_reg_rules, num_5ghz_reg_rules;
5109         int ret;
5110
5111         ath11k_dbg(ab, ATH11K_DBG_WMI, "processing regulatory channel list\n");
5112
5113         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
5114         if (IS_ERR(tb)) {
5115                 ret = PTR_ERR(tb);
5116                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
5117                 return ret;
5118         }
5119
5120         chan_list_event_hdr = tb[WMI_TAG_REG_CHAN_LIST_CC_EVENT];
5121         if (!chan_list_event_hdr) {
5122                 ath11k_warn(ab, "failed to fetch reg chan list update ev\n");
5123                 kfree(tb);
5124                 return -EPROTO;
5125         }
5126
5127         reg_info->num_2ghz_reg_rules = chan_list_event_hdr->num_2ghz_reg_rules;
5128         reg_info->num_5ghz_reg_rules = chan_list_event_hdr->num_5ghz_reg_rules;
5129
5130         if (!(reg_info->num_2ghz_reg_rules + reg_info->num_5ghz_reg_rules)) {
5131                 ath11k_warn(ab, "No regulatory rules available in the event info\n");
5132                 kfree(tb);
5133                 return -EINVAL;
5134         }
5135
5136         memcpy(reg_info->alpha2, &chan_list_event_hdr->alpha2,
5137                REG_ALPHA2_LEN);
5138         reg_info->dfs_region = chan_list_event_hdr->dfs_region;
5139         reg_info->phybitmap = chan_list_event_hdr->phybitmap;
5140         reg_info->num_phy = chan_list_event_hdr->num_phy;
5141         reg_info->phy_id = chan_list_event_hdr->phy_id;
5142         reg_info->ctry_code = chan_list_event_hdr->country_id;
5143         reg_info->reg_dmn_pair = chan_list_event_hdr->domain_code;
5144
5145         ath11k_dbg(ab, ATH11K_DBG_WMI,
5146                    "status_code %s",
5147                    ath11k_cc_status_to_str(reg_info->status_code));
5148
5149         reg_info->status_code =
5150                 ath11k_wmi_cc_setting_code_to_reg(chan_list_event_hdr->status_code);
5151
5152         reg_info->is_ext_reg_event = false;
5153
5154         reg_info->min_bw_2ghz = chan_list_event_hdr->min_bw_2ghz;
5155         reg_info->max_bw_2ghz = chan_list_event_hdr->max_bw_2ghz;
5156         reg_info->min_bw_5ghz = chan_list_event_hdr->min_bw_5ghz;
5157         reg_info->max_bw_5ghz = chan_list_event_hdr->max_bw_5ghz;
5158
5159         num_2ghz_reg_rules = reg_info->num_2ghz_reg_rules;
5160         num_5ghz_reg_rules = reg_info->num_5ghz_reg_rules;
5161
5162         ath11k_dbg(ab, ATH11K_DBG_WMI,
5163                    "cc %s dsf %d BW: min_2ghz %d max_2ghz %d min_5ghz %d max_5ghz %d",
5164                    reg_info->alpha2, reg_info->dfs_region,
5165                    reg_info->min_bw_2ghz, reg_info->max_bw_2ghz,
5166                    reg_info->min_bw_5ghz, reg_info->max_bw_5ghz);
5167
5168         ath11k_dbg(ab, ATH11K_DBG_WMI,
5169                    "num_2ghz_reg_rules %d num_5ghz_reg_rules %d",
5170                    num_2ghz_reg_rules, num_5ghz_reg_rules);
5171
5172         wmi_reg_rule =
5173                 (struct wmi_regulatory_rule_struct *)((u8 *)chan_list_event_hdr
5174                                                 + sizeof(*chan_list_event_hdr)
5175                                                 + sizeof(struct wmi_tlv));
5176
5177         if (num_2ghz_reg_rules) {
5178                 reg_info->reg_rules_2ghz_ptr =
5179                                 create_reg_rules_from_wmi(num_2ghz_reg_rules,
5180                                                           wmi_reg_rule);
5181                 if (!reg_info->reg_rules_2ghz_ptr) {
5182                         kfree(tb);
5183                         ath11k_warn(ab, "Unable to Allocate memory for 2 GHz rules\n");
5184                         return -ENOMEM;
5185                 }
5186
5187                 ath11k_print_reg_rule(ab, "2 GHz",
5188                                       num_2ghz_reg_rules,
5189                                       reg_info->reg_rules_2ghz_ptr);
5190         }
5191
5192         if (num_5ghz_reg_rules) {
5193                 wmi_reg_rule += num_2ghz_reg_rules;
5194                 reg_info->reg_rules_5ghz_ptr =
5195                                 create_reg_rules_from_wmi(num_5ghz_reg_rules,
5196                                                           wmi_reg_rule);
5197                 if (!reg_info->reg_rules_5ghz_ptr) {
5198                         kfree(tb);
5199                         ath11k_warn(ab, "Unable to Allocate memory for 5 GHz rules\n");
5200                         return -ENOMEM;
5201                 }
5202
5203                 ath11k_print_reg_rule(ab, "5 GHz",
5204                                       num_5ghz_reg_rules,
5205                                       reg_info->reg_rules_5ghz_ptr);
5206         }
5207
5208         ath11k_dbg(ab, ATH11K_DBG_WMI, "processed regulatory channel list\n");
5209
5210         kfree(tb);
5211         return 0;
5212 }
5213
5214 static struct cur_reg_rule
5215 *create_ext_reg_rules_from_wmi(u32 num_reg_rules,
5216                                struct wmi_regulatory_ext_rule *wmi_reg_rule)
5217 {
5218         struct cur_reg_rule *reg_rule_ptr;
5219         u32 count;
5220
5221         reg_rule_ptr =  kcalloc(num_reg_rules, sizeof(*reg_rule_ptr), GFP_ATOMIC);
5222
5223         if (!reg_rule_ptr)
5224                 return NULL;
5225
5226         for (count = 0; count < num_reg_rules; count++) {
5227                 reg_rule_ptr[count].start_freq =
5228                         u32_get_bits(wmi_reg_rule[count].freq_info,
5229                                      REG_RULE_START_FREQ);
5230                 reg_rule_ptr[count].end_freq =
5231                         u32_get_bits(wmi_reg_rule[count].freq_info,
5232                                      REG_RULE_END_FREQ);
5233                 reg_rule_ptr[count].max_bw =
5234                         u32_get_bits(wmi_reg_rule[count].bw_pwr_info,
5235                                      REG_RULE_MAX_BW);
5236                 reg_rule_ptr[count].reg_power =
5237                         u32_get_bits(wmi_reg_rule[count].bw_pwr_info,
5238                                      REG_RULE_REG_PWR);
5239                 reg_rule_ptr[count].ant_gain =
5240                         u32_get_bits(wmi_reg_rule[count].bw_pwr_info,
5241                                      REG_RULE_ANT_GAIN);
5242                 reg_rule_ptr[count].flags =
5243                         u32_get_bits(wmi_reg_rule[count].flag_info,
5244                                      REG_RULE_FLAGS);
5245                 reg_rule_ptr[count].psd_flag =
5246                         u32_get_bits(wmi_reg_rule[count].psd_power_info,
5247                                      REG_RULE_PSD_INFO);
5248                 reg_rule_ptr[count].psd_eirp =
5249                         u32_get_bits(wmi_reg_rule[count].psd_power_info,
5250                                      REG_RULE_PSD_EIRP);
5251         }
5252
5253         return reg_rule_ptr;
5254 }
5255
5256 static u8
5257 ath11k_invalid_5ghz_reg_ext_rules_from_wmi(u32 num_reg_rules,
5258                                            const struct wmi_regulatory_ext_rule *rule)
5259 {
5260         u8 num_invalid_5ghz_rules = 0;
5261         u32 count, start_freq;
5262
5263         for (count = 0; count < num_reg_rules; count++) {
5264                 start_freq = u32_get_bits(rule[count].freq_info,
5265                                           REG_RULE_START_FREQ);
5266
5267                 if (start_freq >= ATH11K_MIN_6G_FREQ)
5268                         num_invalid_5ghz_rules++;
5269         }
5270
5271         return num_invalid_5ghz_rules;
5272 }
5273
5274 static int ath11k_pull_reg_chan_list_ext_update_ev(struct ath11k_base *ab,
5275                                                    struct sk_buff *skb,
5276                                                    struct cur_regulatory_info *reg_info)
5277 {
5278         const void **tb;
5279         const struct wmi_reg_chan_list_cc_ext_event *ev;
5280         struct wmi_regulatory_ext_rule *ext_wmi_reg_rule;
5281         u32 num_2ghz_reg_rules, num_5ghz_reg_rules;
5282         u32 num_6ghz_reg_rules_ap[WMI_REG_CURRENT_MAX_AP_TYPE];
5283         u32 num_6ghz_client[WMI_REG_CURRENT_MAX_AP_TYPE][WMI_REG_MAX_CLIENT_TYPE];
5284         u32 total_reg_rules = 0;
5285         int ret, i, j, num_invalid_5ghz_ext_rules = 0;
5286
5287         ath11k_dbg(ab, ATH11K_DBG_WMI, "processing regulatory ext channel list\n");
5288
5289         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
5290         if (IS_ERR(tb)) {
5291                 ret = PTR_ERR(tb);
5292                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
5293                 return ret;
5294         }
5295
5296         ev = tb[WMI_TAG_REG_CHAN_LIST_CC_EXT_EVENT];
5297         if (!ev) {
5298                 ath11k_warn(ab, "failed to fetch reg chan list ext update ev\n");
5299                 kfree(tb);
5300                 return -EPROTO;
5301         }
5302
5303         reg_info->num_2ghz_reg_rules = ev->num_2ghz_reg_rules;
5304         reg_info->num_5ghz_reg_rules = ev->num_5ghz_reg_rules;
5305         reg_info->num_6ghz_rules_ap[WMI_REG_INDOOR_AP] =
5306                         ev->num_6ghz_reg_rules_ap_lpi;
5307         reg_info->num_6ghz_rules_ap[WMI_REG_STANDARD_POWER_AP] =
5308                         ev->num_6ghz_reg_rules_ap_sp;
5309         reg_info->num_6ghz_rules_ap[WMI_REG_VERY_LOW_POWER_AP] =
5310                         ev->num_6ghz_reg_rules_ap_vlp;
5311
5312         for (i = 0; i < WMI_REG_MAX_CLIENT_TYPE; i++) {
5313                 reg_info->num_6ghz_rules_client[WMI_REG_INDOOR_AP][i] =
5314                         ev->num_6ghz_reg_rules_client_lpi[i];
5315                 reg_info->num_6ghz_rules_client[WMI_REG_STANDARD_POWER_AP][i] =
5316                         ev->num_6ghz_reg_rules_client_sp[i];
5317                 reg_info->num_6ghz_rules_client[WMI_REG_VERY_LOW_POWER_AP][i] =
5318                         ev->num_6ghz_reg_rules_client_vlp[i];
5319         }
5320
5321         num_2ghz_reg_rules = reg_info->num_2ghz_reg_rules;
5322         num_5ghz_reg_rules = reg_info->num_5ghz_reg_rules;
5323
5324         total_reg_rules += num_2ghz_reg_rules;
5325         total_reg_rules += num_5ghz_reg_rules;
5326
5327         if ((num_2ghz_reg_rules > MAX_REG_RULES) ||
5328             (num_5ghz_reg_rules > MAX_REG_RULES)) {
5329                 ath11k_warn(ab, "Num reg rules for 2.4 GHz/5 GHz exceeds max limit (num_2ghz_reg_rules: %d num_5ghz_reg_rules: %d max_rules: %d)\n",
5330                             num_2ghz_reg_rules, num_5ghz_reg_rules, MAX_REG_RULES);
5331                 kfree(tb);
5332                 return -EINVAL;
5333         }
5334
5335         for (i = 0; i < WMI_REG_CURRENT_MAX_AP_TYPE; i++) {
5336                 num_6ghz_reg_rules_ap[i] = reg_info->num_6ghz_rules_ap[i];
5337
5338                 if (num_6ghz_reg_rules_ap[i] > MAX_6GHZ_REG_RULES) {
5339                         ath11k_warn(ab, "Num 6 GHz reg rules for AP mode(%d) exceeds max limit (num_6ghz_reg_rules_ap: %d, max_rules: %d)\n",
5340                                     i, num_6ghz_reg_rules_ap[i], MAX_6GHZ_REG_RULES);
5341                         kfree(tb);
5342                         return -EINVAL;
5343                 }
5344
5345                 total_reg_rules += num_6ghz_reg_rules_ap[i];
5346         }
5347
5348         for (i = 0; i < WMI_REG_MAX_CLIENT_TYPE; i++) {
5349                 num_6ghz_client[WMI_REG_INDOOR_AP][i] =
5350                         reg_info->num_6ghz_rules_client[WMI_REG_INDOOR_AP][i];
5351                 total_reg_rules += num_6ghz_client[WMI_REG_INDOOR_AP][i];
5352
5353                 num_6ghz_client[WMI_REG_STANDARD_POWER_AP][i] =
5354                         reg_info->num_6ghz_rules_client[WMI_REG_STANDARD_POWER_AP][i];
5355                 total_reg_rules += num_6ghz_client[WMI_REG_STANDARD_POWER_AP][i];
5356
5357                 num_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][i] =
5358                         reg_info->num_6ghz_rules_client[WMI_REG_VERY_LOW_POWER_AP][i];
5359                 total_reg_rules += num_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][i];
5360
5361                 if ((num_6ghz_client[WMI_REG_INDOOR_AP][i] > MAX_6GHZ_REG_RULES) ||
5362                     (num_6ghz_client[WMI_REG_STANDARD_POWER_AP][i] >
5363                                                              MAX_6GHZ_REG_RULES) ||
5364                     (num_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][i] >
5365                                                              MAX_6GHZ_REG_RULES)) {
5366                         ath11k_warn(ab,
5367                                     "Num 6 GHz client reg rules exceeds max limit, for client(type: %d)\n",
5368                                     i);
5369                         kfree(tb);
5370                         return -EINVAL;
5371                 }
5372         }
5373
5374         if (!total_reg_rules) {
5375                 ath11k_warn(ab, "No reg rules available\n");
5376                 kfree(tb);
5377                 return -EINVAL;
5378         }
5379
5380         memcpy(reg_info->alpha2, &ev->alpha2, REG_ALPHA2_LEN);
5381
5382         reg_info->dfs_region = ev->dfs_region;
5383         reg_info->phybitmap = ev->phybitmap;
5384         reg_info->num_phy = ev->num_phy;
5385         reg_info->phy_id = ev->phy_id;
5386         reg_info->ctry_code = ev->country_id;
5387         reg_info->reg_dmn_pair = ev->domain_code;
5388
5389         ath11k_dbg(ab, ATH11K_DBG_WMI,
5390                    "status_code %s",
5391                    ath11k_cc_status_to_str(reg_info->status_code));
5392
5393         reg_info->status_code =
5394                 ath11k_wmi_cc_setting_code_to_reg(ev->status_code);
5395
5396         reg_info->is_ext_reg_event = true;
5397
5398         reg_info->min_bw_2ghz = ev->min_bw_2ghz;
5399         reg_info->max_bw_2ghz = ev->max_bw_2ghz;
5400         reg_info->min_bw_5ghz = ev->min_bw_5ghz;
5401         reg_info->max_bw_5ghz = ev->max_bw_5ghz;
5402
5403         reg_info->min_bw_6ghz_ap[WMI_REG_INDOOR_AP] =
5404                         ev->min_bw_6ghz_ap_lpi;
5405         reg_info->max_bw_6ghz_ap[WMI_REG_INDOOR_AP] =
5406                         ev->max_bw_6ghz_ap_lpi;
5407         reg_info->min_bw_6ghz_ap[WMI_REG_STANDARD_POWER_AP] =
5408                         ev->min_bw_6ghz_ap_sp;
5409         reg_info->max_bw_6ghz_ap[WMI_REG_STANDARD_POWER_AP] =
5410                         ev->max_bw_6ghz_ap_sp;
5411         reg_info->min_bw_6ghz_ap[WMI_REG_VERY_LOW_POWER_AP] =
5412                         ev->min_bw_6ghz_ap_vlp;
5413         reg_info->max_bw_6ghz_ap[WMI_REG_VERY_LOW_POWER_AP] =
5414                         ev->max_bw_6ghz_ap_vlp;
5415
5416         ath11k_dbg(ab, ATH11K_DBG_WMI,
5417                    "6 GHz AP BW: LPI (%d - %d), SP (%d - %d), VLP (%d - %d)\n",
5418                    reg_info->min_bw_6ghz_ap[WMI_REG_INDOOR_AP],
5419                    reg_info->max_bw_6ghz_ap[WMI_REG_INDOOR_AP],
5420                    reg_info->min_bw_6ghz_ap[WMI_REG_STANDARD_POWER_AP],
5421                    reg_info->max_bw_6ghz_ap[WMI_REG_STANDARD_POWER_AP],
5422                    reg_info->min_bw_6ghz_ap[WMI_REG_VERY_LOW_POWER_AP],
5423                    reg_info->max_bw_6ghz_ap[WMI_REG_VERY_LOW_POWER_AP]);
5424
5425         for (i = 0; i < WMI_REG_MAX_CLIENT_TYPE; i++) {
5426                 reg_info->min_bw_6ghz_client[WMI_REG_INDOOR_AP][i] =
5427                                 ev->min_bw_6ghz_client_lpi[i];
5428                 reg_info->max_bw_6ghz_client[WMI_REG_INDOOR_AP][i] =
5429                                 ev->max_bw_6ghz_client_lpi[i];
5430                 reg_info->min_bw_6ghz_client[WMI_REG_STANDARD_POWER_AP][i] =
5431                                 ev->min_bw_6ghz_client_sp[i];
5432                 reg_info->max_bw_6ghz_client[WMI_REG_STANDARD_POWER_AP][i] =
5433                                 ev->max_bw_6ghz_client_sp[i];
5434                 reg_info->min_bw_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][i] =
5435                                 ev->min_bw_6ghz_client_vlp[i];
5436                 reg_info->max_bw_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][i] =
5437                                 ev->max_bw_6ghz_client_vlp[i];
5438
5439                 ath11k_dbg(ab, ATH11K_DBG_WMI,
5440                            "6 GHz %s BW: LPI (%d - %d), SP (%d - %d), VLP (%d - %d)\n",
5441                            ath11k_6ghz_client_type_to_str(i),
5442                            reg_info->min_bw_6ghz_client[WMI_REG_INDOOR_AP][i],
5443                            reg_info->max_bw_6ghz_client[WMI_REG_INDOOR_AP][i],
5444                            reg_info->min_bw_6ghz_client[WMI_REG_STANDARD_POWER_AP][i],
5445                            reg_info->max_bw_6ghz_client[WMI_REG_STANDARD_POWER_AP][i],
5446                            reg_info->min_bw_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][i],
5447                            reg_info->max_bw_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][i]);
5448         }
5449
5450         ath11k_dbg(ab, ATH11K_DBG_WMI,
5451                    "cc_ext %s dfs %d BW: min_2ghz %d max_2ghz %d min_5ghz %d max_5ghz %d phy_bitmap 0x%x",
5452                    reg_info->alpha2, reg_info->dfs_region,
5453                    reg_info->min_bw_2ghz, reg_info->max_bw_2ghz,
5454                    reg_info->min_bw_5ghz, reg_info->max_bw_5ghz,
5455                    reg_info->phybitmap);
5456
5457         ath11k_dbg(ab, ATH11K_DBG_WMI,
5458                    "num_2ghz_reg_rules %d num_5ghz_reg_rules %d",
5459                    num_2ghz_reg_rules, num_5ghz_reg_rules);
5460
5461         ath11k_dbg(ab, ATH11K_DBG_WMI,
5462                    "num_6ghz_reg_rules_ap_lpi: %d num_6ghz_reg_rules_ap_sp: %d num_6ghz_reg_rules_ap_vlp: %d",
5463                    num_6ghz_reg_rules_ap[WMI_REG_INDOOR_AP],
5464                    num_6ghz_reg_rules_ap[WMI_REG_STANDARD_POWER_AP],
5465                    num_6ghz_reg_rules_ap[WMI_REG_VERY_LOW_POWER_AP]);
5466
5467         j = WMI_REG_DEFAULT_CLIENT;
5468         ath11k_dbg(ab, ATH11K_DBG_WMI,
5469                    "6 GHz Regular client: num_6ghz_reg_rules_lpi: %d num_6ghz_reg_rules_sp: %d num_6ghz_reg_rules_vlp: %d",
5470                    num_6ghz_client[WMI_REG_INDOOR_AP][j],
5471                    num_6ghz_client[WMI_REG_STANDARD_POWER_AP][j],
5472                    num_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][j]);
5473
5474         j = WMI_REG_SUBORDINATE_CLIENT;
5475         ath11k_dbg(ab, ATH11K_DBG_WMI,
5476                    "6 GHz Subordinate client: num_6ghz_reg_rules_lpi: %d num_6ghz_reg_rules_sp: %d num_6ghz_reg_rules_vlp: %d",
5477                    num_6ghz_client[WMI_REG_INDOOR_AP][j],
5478                    num_6ghz_client[WMI_REG_STANDARD_POWER_AP][j],
5479                    num_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][j]);
5480
5481         ext_wmi_reg_rule =
5482                 (struct wmi_regulatory_ext_rule *)((u8 *)ev + sizeof(*ev) +
5483                                                    sizeof(struct wmi_tlv));
5484         if (num_2ghz_reg_rules) {
5485                 reg_info->reg_rules_2ghz_ptr =
5486                         create_ext_reg_rules_from_wmi(num_2ghz_reg_rules,
5487                                                       ext_wmi_reg_rule);
5488
5489                 if (!reg_info->reg_rules_2ghz_ptr) {
5490                         kfree(tb);
5491                         ath11k_warn(ab, "Unable to Allocate memory for 2 GHz rules\n");
5492                         return -ENOMEM;
5493                 }
5494
5495                 ath11k_print_reg_rule(ab, "2 GHz",
5496                                       num_2ghz_reg_rules,
5497                                       reg_info->reg_rules_2ghz_ptr);
5498         }
5499
5500         ext_wmi_reg_rule += num_2ghz_reg_rules;
5501
5502         /* Firmware might include 6 GHz reg rule in 5 GHz rule list
5503          * for few countries along with separate 6 GHz rule.
5504          * Having same 6 GHz reg rule in 5 GHz and 6 GHz rules list
5505          * causes intersect check to be true, and same rules will be
5506          * shown multiple times in iw cmd.
5507          * Hence, avoid parsing 6 GHz rule from 5 GHz reg rule list
5508          */
5509         num_invalid_5ghz_ext_rules =
5510                 ath11k_invalid_5ghz_reg_ext_rules_from_wmi(num_5ghz_reg_rules,
5511                                                            ext_wmi_reg_rule);
5512
5513         if (num_invalid_5ghz_ext_rules) {
5514                 ath11k_dbg(ab, ATH11K_DBG_WMI,
5515                            "CC: %s 5 GHz reg rules number %d from fw, %d number of invalid 5 GHz rules",
5516                            reg_info->alpha2, reg_info->num_5ghz_reg_rules,
5517                            num_invalid_5ghz_ext_rules);
5518
5519                 num_5ghz_reg_rules = num_5ghz_reg_rules - num_invalid_5ghz_ext_rules;
5520                 reg_info->num_5ghz_reg_rules = num_5ghz_reg_rules;
5521         }
5522
5523         if (num_5ghz_reg_rules) {
5524                 reg_info->reg_rules_5ghz_ptr =
5525                         create_ext_reg_rules_from_wmi(num_5ghz_reg_rules,
5526                                                       ext_wmi_reg_rule);
5527
5528                 if (!reg_info->reg_rules_5ghz_ptr) {
5529                         kfree(tb);
5530                         ath11k_warn(ab, "Unable to Allocate memory for 5 GHz rules\n");
5531                         return -ENOMEM;
5532                 }
5533
5534                 ath11k_print_reg_rule(ab, "5 GHz",
5535                                       num_5ghz_reg_rules,
5536                                       reg_info->reg_rules_5ghz_ptr);
5537         }
5538
5539         /* We have adjusted the number of 5 GHz reg rules above. But still those
5540          * many rules needs to be adjusted in ext_wmi_reg_rule.
5541          *
5542          * NOTE: num_invalid_5ghz_ext_rules will be 0 for rest other cases.
5543          */
5544         ext_wmi_reg_rule += (num_5ghz_reg_rules + num_invalid_5ghz_ext_rules);
5545
5546         for (i = 0; i < WMI_REG_CURRENT_MAX_AP_TYPE; i++) {
5547                 reg_info->reg_rules_6ghz_ap_ptr[i] =
5548                         create_ext_reg_rules_from_wmi(num_6ghz_reg_rules_ap[i],
5549                                                       ext_wmi_reg_rule);
5550
5551                 if (!reg_info->reg_rules_6ghz_ap_ptr[i]) {
5552                         kfree(tb);
5553                         ath11k_warn(ab, "Unable to Allocate memory for 6 GHz AP rules\n");
5554                         return -ENOMEM;
5555                 }
5556
5557                 ath11k_print_reg_rule(ab, ath11k_6ghz_ap_type_to_str(i),
5558                                       num_6ghz_reg_rules_ap[i],
5559                                       reg_info->reg_rules_6ghz_ap_ptr[i]);
5560
5561                 ext_wmi_reg_rule += num_6ghz_reg_rules_ap[i];
5562         }
5563
5564         for (j = 0; j < WMI_REG_CURRENT_MAX_AP_TYPE; j++) {
5565                 ath11k_dbg(ab, ATH11K_DBG_WMI,
5566                            "6 GHz AP type %s", ath11k_6ghz_ap_type_to_str(j));
5567
5568                 for (i = 0; i < WMI_REG_MAX_CLIENT_TYPE; i++) {
5569                         reg_info->reg_rules_6ghz_client_ptr[j][i] =
5570                                 create_ext_reg_rules_from_wmi(num_6ghz_client[j][i],
5571                                                               ext_wmi_reg_rule);
5572
5573                         if (!reg_info->reg_rules_6ghz_client_ptr[j][i]) {
5574                                 kfree(tb);
5575                                 ath11k_warn(ab, "Unable to Allocate memory for 6 GHz client rules\n");
5576                                 return -ENOMEM;
5577                         }
5578
5579                         ath11k_print_reg_rule(ab,
5580                                               ath11k_6ghz_client_type_to_str(i),
5581                                               num_6ghz_client[j][i],
5582                                               reg_info->reg_rules_6ghz_client_ptr[j][i]);
5583
5584                         ext_wmi_reg_rule += num_6ghz_client[j][i];
5585                 }
5586         }
5587
5588         reg_info->client_type = ev->client_type;
5589         reg_info->rnr_tpe_usable = ev->rnr_tpe_usable;
5590         reg_info->unspecified_ap_usable =
5591                         ev->unspecified_ap_usable;
5592         reg_info->domain_code_6ghz_ap[WMI_REG_INDOOR_AP] =
5593                         ev->domain_code_6ghz_ap_lpi;
5594         reg_info->domain_code_6ghz_ap[WMI_REG_STANDARD_POWER_AP] =
5595                         ev->domain_code_6ghz_ap_sp;
5596         reg_info->domain_code_6ghz_ap[WMI_REG_VERY_LOW_POWER_AP] =
5597                         ev->domain_code_6ghz_ap_vlp;
5598
5599         ath11k_dbg(ab, ATH11K_DBG_WMI,
5600                    "6 GHz reg info client type %s rnr_tpe_usable %d unspecified_ap_usable %d AP sub domain: lpi %s, sp %s, vlp %s\n",
5601                    ath11k_6ghz_client_type_to_str(reg_info->client_type),
5602                    reg_info->rnr_tpe_usable,
5603                    reg_info->unspecified_ap_usable,
5604                    ath11k_sub_reg_6ghz_to_str(ev->domain_code_6ghz_ap_lpi),
5605                    ath11k_sub_reg_6ghz_to_str(ev->domain_code_6ghz_ap_sp),
5606                    ath11k_sub_reg_6ghz_to_str(ev->domain_code_6ghz_ap_vlp));
5607
5608         for (i = 0; i < WMI_REG_MAX_CLIENT_TYPE; i++) {
5609                 reg_info->domain_code_6ghz_client[WMI_REG_INDOOR_AP][i] =
5610                                 ev->domain_code_6ghz_client_lpi[i];
5611                 reg_info->domain_code_6ghz_client[WMI_REG_STANDARD_POWER_AP][i] =
5612                                 ev->domain_code_6ghz_client_sp[i];
5613                 reg_info->domain_code_6ghz_client[WMI_REG_VERY_LOW_POWER_AP][i] =
5614                                 ev->domain_code_6ghz_client_vlp[i];
5615
5616                 ath11k_dbg(ab, ATH11K_DBG_WMI,
5617                            "6 GHz client type %s client sub domain: lpi %s, sp %s, vlp %s\n",
5618                            ath11k_6ghz_client_type_to_str(i),
5619                            ath11k_sub_reg_6ghz_to_str(ev->domain_code_6ghz_client_lpi[i]),
5620                            ath11k_sub_reg_6ghz_to_str(ev->domain_code_6ghz_client_sp[i]),
5621                            ath11k_sub_reg_6ghz_to_str(ev->domain_code_6ghz_client_vlp[i])
5622                           );
5623         }
5624
5625         reg_info->domain_code_6ghz_super_id = ev->domain_code_6ghz_super_id;
5626
5627         ath11k_dbg(ab, ATH11K_DBG_WMI,
5628                    "6 GHz client_type %s 6 GHz super domain %s",
5629                    ath11k_6ghz_client_type_to_str(reg_info->client_type),
5630                    ath11k_super_reg_6ghz_to_str(reg_info->domain_code_6ghz_super_id));
5631
5632         ath11k_dbg(ab, ATH11K_DBG_WMI, "processed regulatory ext channel list\n");
5633
5634         kfree(tb);
5635         return 0;
5636 }
5637
5638 static int ath11k_pull_peer_del_resp_ev(struct ath11k_base *ab, struct sk_buff *skb,
5639                                         struct wmi_peer_delete_resp_event *peer_del_resp)
5640 {
5641         const void **tb;
5642         const struct wmi_peer_delete_resp_event *ev;
5643         int ret;
5644
5645         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
5646         if (IS_ERR(tb)) {
5647                 ret = PTR_ERR(tb);
5648                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
5649                 return ret;
5650         }
5651
5652         ev = tb[WMI_TAG_PEER_DELETE_RESP_EVENT];
5653         if (!ev) {
5654                 ath11k_warn(ab, "failed to fetch peer delete resp ev");
5655                 kfree(tb);
5656                 return -EPROTO;
5657         }
5658
5659         memset(peer_del_resp, 0, sizeof(*peer_del_resp));
5660
5661         peer_del_resp->vdev_id = ev->vdev_id;
5662         ether_addr_copy(peer_del_resp->peer_macaddr.addr,
5663                         ev->peer_macaddr.addr);
5664
5665         kfree(tb);
5666         return 0;
5667 }
5668
5669 static int ath11k_pull_vdev_del_resp_ev(struct ath11k_base *ab,
5670                                         struct sk_buff *skb,
5671                                         u32 *vdev_id)
5672 {
5673         const void **tb;
5674         const struct wmi_vdev_delete_resp_event *ev;
5675         int ret;
5676
5677         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
5678         if (IS_ERR(tb)) {
5679                 ret = PTR_ERR(tb);
5680                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
5681                 return ret;
5682         }
5683
5684         ev = tb[WMI_TAG_VDEV_DELETE_RESP_EVENT];
5685         if (!ev) {
5686                 ath11k_warn(ab, "failed to fetch vdev delete resp ev");
5687                 kfree(tb);
5688                 return -EPROTO;
5689         }
5690
5691         *vdev_id = ev->vdev_id;
5692
5693         kfree(tb);
5694         return 0;
5695 }
5696
5697 static int ath11k_pull_bcn_tx_status_ev(struct ath11k_base *ab, void *evt_buf,
5698                                         u32 len, u32 *vdev_id,
5699                                         u32 *tx_status)
5700 {
5701         const void **tb;
5702         const struct wmi_bcn_tx_status_event *ev;
5703         int ret;
5704
5705         tb = ath11k_wmi_tlv_parse_alloc(ab, evt_buf, len, GFP_ATOMIC);
5706         if (IS_ERR(tb)) {
5707                 ret = PTR_ERR(tb);
5708                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
5709                 return ret;
5710         }
5711
5712         ev = tb[WMI_TAG_OFFLOAD_BCN_TX_STATUS_EVENT];
5713         if (!ev) {
5714                 ath11k_warn(ab, "failed to fetch bcn tx status ev");
5715                 kfree(tb);
5716                 return -EPROTO;
5717         }
5718
5719         *vdev_id   = ev->vdev_id;
5720         *tx_status = ev->tx_status;
5721
5722         kfree(tb);
5723         return 0;
5724 }
5725
5726 static int ath11k_pull_vdev_stopped_param_tlv(struct ath11k_base *ab, struct sk_buff *skb,
5727                                               u32 *vdev_id)
5728 {
5729         const void **tb;
5730         const struct wmi_vdev_stopped_event *ev;
5731         int ret;
5732
5733         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
5734         if (IS_ERR(tb)) {
5735                 ret = PTR_ERR(tb);
5736                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
5737                 return ret;
5738         }
5739
5740         ev = tb[WMI_TAG_VDEV_STOPPED_EVENT];
5741         if (!ev) {
5742                 ath11k_warn(ab, "failed to fetch vdev stop ev");
5743                 kfree(tb);
5744                 return -EPROTO;
5745         }
5746
5747         *vdev_id =  ev->vdev_id;
5748
5749         kfree(tb);
5750         return 0;
5751 }
5752
5753 static int ath11k_wmi_tlv_mgmt_rx_parse(struct ath11k_base *ab,
5754                                         u16 tag, u16 len,
5755                                         const void *ptr, void *data)
5756 {
5757         struct wmi_tlv_mgmt_rx_parse *parse = data;
5758
5759         switch (tag) {
5760         case WMI_TAG_MGMT_RX_HDR:
5761                 parse->fixed = ptr;
5762                 break;
5763         case WMI_TAG_ARRAY_BYTE:
5764                 if (!parse->frame_buf_done) {
5765                         parse->frame_buf = ptr;
5766                         parse->frame_buf_done = true;
5767                 }
5768                 break;
5769         }
5770         return 0;
5771 }
5772
5773 static int ath11k_pull_mgmt_rx_params_tlv(struct ath11k_base *ab,
5774                                           struct sk_buff *skb,
5775                                           struct mgmt_rx_event_params *hdr)
5776 {
5777         struct wmi_tlv_mgmt_rx_parse parse = { };
5778         const struct wmi_mgmt_rx_hdr *ev;
5779         const u8 *frame;
5780         int ret;
5781
5782         ret = ath11k_wmi_tlv_iter(ab, skb->data, skb->len,
5783                                   ath11k_wmi_tlv_mgmt_rx_parse,
5784                                   &parse);
5785         if (ret) {
5786                 ath11k_warn(ab, "failed to parse mgmt rx tlv %d\n",
5787                             ret);
5788                 return ret;
5789         }
5790
5791         ev = parse.fixed;
5792         frame = parse.frame_buf;
5793
5794         if (!ev || !frame) {
5795                 ath11k_warn(ab, "failed to fetch mgmt rx hdr");
5796                 return -EPROTO;
5797         }
5798
5799         hdr->pdev_id =  ev->pdev_id;
5800         hdr->chan_freq = ev->chan_freq;
5801         hdr->channel =  ev->channel;
5802         hdr->snr =  ev->snr;
5803         hdr->rate =  ev->rate;
5804         hdr->phy_mode =  ev->phy_mode;
5805         hdr->buf_len =  ev->buf_len;
5806         hdr->status =  ev->status;
5807         hdr->flags =  ev->flags;
5808         hdr->rssi =  ev->rssi;
5809         hdr->tsf_delta =  ev->tsf_delta;
5810         memcpy(hdr->rssi_ctl, ev->rssi_ctl, sizeof(hdr->rssi_ctl));
5811
5812         if (skb->len < (frame - skb->data) + hdr->buf_len) {
5813                 ath11k_warn(ab, "invalid length in mgmt rx hdr ev");
5814                 return -EPROTO;
5815         }
5816
5817         /* shift the sk_buff to point to `frame` */
5818         skb_trim(skb, 0);
5819         skb_put(skb, frame - skb->data);
5820         skb_pull(skb, frame - skb->data);
5821         skb_put(skb, hdr->buf_len);
5822
5823         ath11k_ce_byte_swap(skb->data, hdr->buf_len);
5824
5825         return 0;
5826 }
5827
5828 static int wmi_process_mgmt_tx_comp(struct ath11k *ar,
5829                                     struct wmi_mgmt_tx_compl_event *tx_compl_param)
5830 {
5831         struct sk_buff *msdu;
5832         struct ieee80211_tx_info *info;
5833         struct ath11k_skb_cb *skb_cb;
5834         int num_mgmt;
5835
5836         spin_lock_bh(&ar->txmgmt_idr_lock);
5837         msdu = idr_find(&ar->txmgmt_idr, tx_compl_param->desc_id);
5838
5839         if (!msdu) {
5840                 ath11k_warn(ar->ab, "received mgmt tx compl for invalid msdu_id: %d\n",
5841                             tx_compl_param->desc_id);
5842                 spin_unlock_bh(&ar->txmgmt_idr_lock);
5843                 return -ENOENT;
5844         }
5845
5846         idr_remove(&ar->txmgmt_idr, tx_compl_param->desc_id);
5847         spin_unlock_bh(&ar->txmgmt_idr_lock);
5848
5849         skb_cb = ATH11K_SKB_CB(msdu);
5850         dma_unmap_single(ar->ab->dev, skb_cb->paddr, msdu->len, DMA_TO_DEVICE);
5851
5852         info = IEEE80211_SKB_CB(msdu);
5853         if ((!(info->flags & IEEE80211_TX_CTL_NO_ACK)) &&
5854             !tx_compl_param->status) {
5855                 info->flags |= IEEE80211_TX_STAT_ACK;
5856                 if (test_bit(WMI_TLV_SERVICE_TX_DATA_MGMT_ACK_RSSI,
5857                              ar->ab->wmi_ab.svc_map))
5858                         info->status.ack_signal = tx_compl_param->ack_rssi;
5859         }
5860
5861         ieee80211_tx_status_irqsafe(ar->hw, msdu);
5862
5863         num_mgmt = atomic_dec_if_positive(&ar->num_pending_mgmt_tx);
5864
5865         /* WARN when we received this event without doing any mgmt tx */
5866         if (num_mgmt < 0)
5867                 WARN_ON_ONCE(1);
5868
5869         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
5870                    "mgmt tx comp pending %d desc id %d\n",
5871                    num_mgmt, tx_compl_param->desc_id);
5872
5873         if (!num_mgmt)
5874                 wake_up(&ar->txmgmt_empty_waitq);
5875
5876         return 0;
5877 }
5878
5879 static int ath11k_pull_mgmt_tx_compl_param_tlv(struct ath11k_base *ab,
5880                                                struct sk_buff *skb,
5881                                                struct wmi_mgmt_tx_compl_event *param)
5882 {
5883         const void **tb;
5884         const struct wmi_mgmt_tx_compl_event *ev;
5885         int ret;
5886
5887         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
5888         if (IS_ERR(tb)) {
5889                 ret = PTR_ERR(tb);
5890                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
5891                 return ret;
5892         }
5893
5894         ev = tb[WMI_TAG_MGMT_TX_COMPL_EVENT];
5895         if (!ev) {
5896                 ath11k_warn(ab, "failed to fetch mgmt tx compl ev");
5897                 kfree(tb);
5898                 return -EPROTO;
5899         }
5900
5901         param->pdev_id = ev->pdev_id;
5902         param->desc_id = ev->desc_id;
5903         param->status = ev->status;
5904         param->ack_rssi = ev->ack_rssi;
5905
5906         kfree(tb);
5907         return 0;
5908 }
5909
5910 static void ath11k_wmi_event_scan_started(struct ath11k *ar)
5911 {
5912         lockdep_assert_held(&ar->data_lock);
5913
5914         switch (ar->scan.state) {
5915         case ATH11K_SCAN_IDLE:
5916         case ATH11K_SCAN_RUNNING:
5917         case ATH11K_SCAN_ABORTING:
5918                 ath11k_warn(ar->ab, "received scan started event in an invalid scan state: %s (%d)\n",
5919                             ath11k_scan_state_str(ar->scan.state),
5920                             ar->scan.state);
5921                 break;
5922         case ATH11K_SCAN_STARTING:
5923                 ar->scan.state = ATH11K_SCAN_RUNNING;
5924                 if (ar->scan.is_roc)
5925                         ieee80211_ready_on_channel(ar->hw);
5926                 complete(&ar->scan.started);
5927                 break;
5928         }
5929 }
5930
5931 static void ath11k_wmi_event_scan_start_failed(struct ath11k *ar)
5932 {
5933         lockdep_assert_held(&ar->data_lock);
5934
5935         switch (ar->scan.state) {
5936         case ATH11K_SCAN_IDLE:
5937         case ATH11K_SCAN_RUNNING:
5938         case ATH11K_SCAN_ABORTING:
5939                 ath11k_warn(ar->ab, "received scan start failed event in an invalid scan state: %s (%d)\n",
5940                             ath11k_scan_state_str(ar->scan.state),
5941                             ar->scan.state);
5942                 break;
5943         case ATH11K_SCAN_STARTING:
5944                 complete(&ar->scan.started);
5945                 __ath11k_mac_scan_finish(ar);
5946                 break;
5947         }
5948 }
5949
5950 static void ath11k_wmi_event_scan_completed(struct ath11k *ar)
5951 {
5952         lockdep_assert_held(&ar->data_lock);
5953
5954         switch (ar->scan.state) {
5955         case ATH11K_SCAN_IDLE:
5956         case ATH11K_SCAN_STARTING:
5957                 /* One suspected reason scan can be completed while starting is
5958                  * if firmware fails to deliver all scan events to the host,
5959                  * e.g. when transport pipe is full. This has been observed
5960                  * with spectral scan phyerr events starving wmi transport
5961                  * pipe. In such case the "scan completed" event should be (and
5962                  * is) ignored by the host as it may be just firmware's scan
5963                  * state machine recovering.
5964                  */
5965                 ath11k_warn(ar->ab, "received scan completed event in an invalid scan state: %s (%d)\n",
5966                             ath11k_scan_state_str(ar->scan.state),
5967                             ar->scan.state);
5968                 break;
5969         case ATH11K_SCAN_RUNNING:
5970         case ATH11K_SCAN_ABORTING:
5971                 __ath11k_mac_scan_finish(ar);
5972                 break;
5973         }
5974 }
5975
5976 static void ath11k_wmi_event_scan_bss_chan(struct ath11k *ar)
5977 {
5978         lockdep_assert_held(&ar->data_lock);
5979
5980         switch (ar->scan.state) {
5981         case ATH11K_SCAN_IDLE:
5982         case ATH11K_SCAN_STARTING:
5983                 ath11k_warn(ar->ab, "received scan bss chan event in an invalid scan state: %s (%d)\n",
5984                             ath11k_scan_state_str(ar->scan.state),
5985                             ar->scan.state);
5986                 break;
5987         case ATH11K_SCAN_RUNNING:
5988         case ATH11K_SCAN_ABORTING:
5989                 ar->scan_channel = NULL;
5990                 break;
5991         }
5992 }
5993
5994 static void ath11k_wmi_event_scan_foreign_chan(struct ath11k *ar, u32 freq)
5995 {
5996         lockdep_assert_held(&ar->data_lock);
5997
5998         switch (ar->scan.state) {
5999         case ATH11K_SCAN_IDLE:
6000         case ATH11K_SCAN_STARTING:
6001                 ath11k_warn(ar->ab, "received scan foreign chan event in an invalid scan state: %s (%d)\n",
6002                             ath11k_scan_state_str(ar->scan.state),
6003                             ar->scan.state);
6004                 break;
6005         case ATH11K_SCAN_RUNNING:
6006         case ATH11K_SCAN_ABORTING:
6007                 ar->scan_channel = ieee80211_get_channel(ar->hw->wiphy, freq);
6008                 if (ar->scan.is_roc && ar->scan.roc_freq == freq)
6009                         complete(&ar->scan.on_channel);
6010                 break;
6011         }
6012 }
6013
6014 static const char *
6015 ath11k_wmi_event_scan_type_str(enum wmi_scan_event_type type,
6016                                enum wmi_scan_completion_reason reason)
6017 {
6018         switch (type) {
6019         case WMI_SCAN_EVENT_STARTED:
6020                 return "started";
6021         case WMI_SCAN_EVENT_COMPLETED:
6022                 switch (reason) {
6023                 case WMI_SCAN_REASON_COMPLETED:
6024                         return "completed";
6025                 case WMI_SCAN_REASON_CANCELLED:
6026                         return "completed [cancelled]";
6027                 case WMI_SCAN_REASON_PREEMPTED:
6028                         return "completed [preempted]";
6029                 case WMI_SCAN_REASON_TIMEDOUT:
6030                         return "completed [timedout]";
6031                 case WMI_SCAN_REASON_INTERNAL_FAILURE:
6032                         return "completed [internal err]";
6033                 case WMI_SCAN_REASON_MAX:
6034                         break;
6035                 }
6036                 return "completed [unknown]";
6037         case WMI_SCAN_EVENT_BSS_CHANNEL:
6038                 return "bss channel";
6039         case WMI_SCAN_EVENT_FOREIGN_CHAN:
6040                 return "foreign channel";
6041         case WMI_SCAN_EVENT_DEQUEUED:
6042                 return "dequeued";
6043         case WMI_SCAN_EVENT_PREEMPTED:
6044                 return "preempted";
6045         case WMI_SCAN_EVENT_START_FAILED:
6046                 return "start failed";
6047         case WMI_SCAN_EVENT_RESTARTED:
6048                 return "restarted";
6049         case WMI_SCAN_EVENT_FOREIGN_CHAN_EXIT:
6050                 return "foreign channel exit";
6051         default:
6052                 return "unknown";
6053         }
6054 }
6055
6056 static int ath11k_pull_scan_ev(struct ath11k_base *ab, struct sk_buff *skb,
6057                                struct wmi_scan_event *scan_evt_param)
6058 {
6059         const void **tb;
6060         const struct wmi_scan_event *ev;
6061         int ret;
6062
6063         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
6064         if (IS_ERR(tb)) {
6065                 ret = PTR_ERR(tb);
6066                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
6067                 return ret;
6068         }
6069
6070         ev = tb[WMI_TAG_SCAN_EVENT];
6071         if (!ev) {
6072                 ath11k_warn(ab, "failed to fetch scan ev");
6073                 kfree(tb);
6074                 return -EPROTO;
6075         }
6076
6077         scan_evt_param->event_type = ev->event_type;
6078         scan_evt_param->reason = ev->reason;
6079         scan_evt_param->channel_freq = ev->channel_freq;
6080         scan_evt_param->scan_req_id = ev->scan_req_id;
6081         scan_evt_param->scan_id = ev->scan_id;
6082         scan_evt_param->vdev_id = ev->vdev_id;
6083         scan_evt_param->tsf_timestamp = ev->tsf_timestamp;
6084
6085         kfree(tb);
6086         return 0;
6087 }
6088
6089 static int ath11k_pull_peer_sta_kickout_ev(struct ath11k_base *ab, struct sk_buff *skb,
6090                                            struct wmi_peer_sta_kickout_arg *arg)
6091 {
6092         const void **tb;
6093         const struct wmi_peer_sta_kickout_event *ev;
6094         int ret;
6095
6096         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
6097         if (IS_ERR(tb)) {
6098                 ret = PTR_ERR(tb);
6099                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
6100                 return ret;
6101         }
6102
6103         ev = tb[WMI_TAG_PEER_STA_KICKOUT_EVENT];
6104         if (!ev) {
6105                 ath11k_warn(ab, "failed to fetch peer sta kickout ev");
6106                 kfree(tb);
6107                 return -EPROTO;
6108         }
6109
6110         arg->mac_addr = ev->peer_macaddr.addr;
6111
6112         kfree(tb);
6113         return 0;
6114 }
6115
6116 static int ath11k_pull_roam_ev(struct ath11k_base *ab, struct sk_buff *skb,
6117                                struct wmi_roam_event *roam_ev)
6118 {
6119         const void **tb;
6120         const struct wmi_roam_event *ev;
6121         int ret;
6122
6123         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
6124         if (IS_ERR(tb)) {
6125                 ret = PTR_ERR(tb);
6126                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
6127                 return ret;
6128         }
6129
6130         ev = tb[WMI_TAG_ROAM_EVENT];
6131         if (!ev) {
6132                 ath11k_warn(ab, "failed to fetch roam ev");
6133                 kfree(tb);
6134                 return -EPROTO;
6135         }
6136
6137         roam_ev->vdev_id = ev->vdev_id;
6138         roam_ev->reason = ev->reason;
6139         roam_ev->rssi = ev->rssi;
6140
6141         kfree(tb);
6142         return 0;
6143 }
6144
6145 static int freq_to_idx(struct ath11k *ar, int freq)
6146 {
6147         struct ieee80211_supported_band *sband;
6148         int band, ch, idx = 0;
6149
6150         for (band = NL80211_BAND_2GHZ; band < NUM_NL80211_BANDS; band++) {
6151                 sband = ar->hw->wiphy->bands[band];
6152                 if (!sband)
6153                         continue;
6154
6155                 for (ch = 0; ch < sband->n_channels; ch++, idx++)
6156                         if (sband->channels[ch].center_freq == freq)
6157                                 goto exit;
6158         }
6159
6160 exit:
6161         return idx;
6162 }
6163
6164 static int ath11k_pull_chan_info_ev(struct ath11k_base *ab, u8 *evt_buf,
6165                                     u32 len, struct wmi_chan_info_event *ch_info_ev)
6166 {
6167         const void **tb;
6168         const struct wmi_chan_info_event *ev;
6169         int ret;
6170
6171         tb = ath11k_wmi_tlv_parse_alloc(ab, evt_buf, len, GFP_ATOMIC);
6172         if (IS_ERR(tb)) {
6173                 ret = PTR_ERR(tb);
6174                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
6175                 return ret;
6176         }
6177
6178         ev = tb[WMI_TAG_CHAN_INFO_EVENT];
6179         if (!ev) {
6180                 ath11k_warn(ab, "failed to fetch chan info ev");
6181                 kfree(tb);
6182                 return -EPROTO;
6183         }
6184
6185         ch_info_ev->err_code = ev->err_code;
6186         ch_info_ev->freq = ev->freq;
6187         ch_info_ev->cmd_flags = ev->cmd_flags;
6188         ch_info_ev->noise_floor = ev->noise_floor;
6189         ch_info_ev->rx_clear_count = ev->rx_clear_count;
6190         ch_info_ev->cycle_count = ev->cycle_count;
6191         ch_info_ev->chan_tx_pwr_range = ev->chan_tx_pwr_range;
6192         ch_info_ev->chan_tx_pwr_tp = ev->chan_tx_pwr_tp;
6193         ch_info_ev->rx_frame_count = ev->rx_frame_count;
6194         ch_info_ev->tx_frame_cnt = ev->tx_frame_cnt;
6195         ch_info_ev->mac_clk_mhz = ev->mac_clk_mhz;
6196         ch_info_ev->vdev_id = ev->vdev_id;
6197
6198         kfree(tb);
6199         return 0;
6200 }
6201
6202 static int
6203 ath11k_pull_pdev_bss_chan_info_ev(struct ath11k_base *ab, struct sk_buff *skb,
6204                                   struct wmi_pdev_bss_chan_info_event *bss_ch_info_ev)
6205 {
6206         const void **tb;
6207         const struct wmi_pdev_bss_chan_info_event *ev;
6208         int ret;
6209
6210         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
6211         if (IS_ERR(tb)) {
6212                 ret = PTR_ERR(tb);
6213                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
6214                 return ret;
6215         }
6216
6217         ev = tb[WMI_TAG_PDEV_BSS_CHAN_INFO_EVENT];
6218         if (!ev) {
6219                 ath11k_warn(ab, "failed to fetch pdev bss chan info ev");
6220                 kfree(tb);
6221                 return -EPROTO;
6222         }
6223
6224         bss_ch_info_ev->pdev_id = ev->pdev_id;
6225         bss_ch_info_ev->freq = ev->freq;
6226         bss_ch_info_ev->noise_floor = ev->noise_floor;
6227         bss_ch_info_ev->rx_clear_count_low = ev->rx_clear_count_low;
6228         bss_ch_info_ev->rx_clear_count_high = ev->rx_clear_count_high;
6229         bss_ch_info_ev->cycle_count_low = ev->cycle_count_low;
6230         bss_ch_info_ev->cycle_count_high = ev->cycle_count_high;
6231         bss_ch_info_ev->tx_cycle_count_low = ev->tx_cycle_count_low;
6232         bss_ch_info_ev->tx_cycle_count_high = ev->tx_cycle_count_high;
6233         bss_ch_info_ev->rx_cycle_count_low = ev->rx_cycle_count_low;
6234         bss_ch_info_ev->rx_cycle_count_high = ev->rx_cycle_count_high;
6235         bss_ch_info_ev->rx_bss_cycle_count_low = ev->rx_bss_cycle_count_low;
6236         bss_ch_info_ev->rx_bss_cycle_count_high = ev->rx_bss_cycle_count_high;
6237
6238         kfree(tb);
6239         return 0;
6240 }
6241
6242 static int
6243 ath11k_pull_vdev_install_key_compl_ev(struct ath11k_base *ab, struct sk_buff *skb,
6244                                       struct wmi_vdev_install_key_complete_arg *arg)
6245 {
6246         const void **tb;
6247         const struct wmi_vdev_install_key_compl_event *ev;
6248         int ret;
6249
6250         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
6251         if (IS_ERR(tb)) {
6252                 ret = PTR_ERR(tb);
6253                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
6254                 return ret;
6255         }
6256
6257         ev = tb[WMI_TAG_VDEV_INSTALL_KEY_COMPLETE_EVENT];
6258         if (!ev) {
6259                 ath11k_warn(ab, "failed to fetch vdev install key compl ev");
6260                 kfree(tb);
6261                 return -EPROTO;
6262         }
6263
6264         arg->vdev_id = ev->vdev_id;
6265         arg->macaddr = ev->peer_macaddr.addr;
6266         arg->key_idx = ev->key_idx;
6267         arg->key_flags = ev->key_flags;
6268         arg->status = ev->status;
6269
6270         kfree(tb);
6271         return 0;
6272 }
6273
6274 static int ath11k_pull_peer_assoc_conf_ev(struct ath11k_base *ab, struct sk_buff *skb,
6275                                           struct wmi_peer_assoc_conf_arg *peer_assoc_conf)
6276 {
6277         const void **tb;
6278         const struct wmi_peer_assoc_conf_event *ev;
6279         int ret;
6280
6281         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
6282         if (IS_ERR(tb)) {
6283                 ret = PTR_ERR(tb);
6284                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
6285                 return ret;
6286         }
6287
6288         ev = tb[WMI_TAG_PEER_ASSOC_CONF_EVENT];
6289         if (!ev) {
6290                 ath11k_warn(ab, "failed to fetch peer assoc conf ev");
6291                 kfree(tb);
6292                 return -EPROTO;
6293         }
6294
6295         peer_assoc_conf->vdev_id = ev->vdev_id;
6296         peer_assoc_conf->macaddr = ev->peer_macaddr.addr;
6297
6298         kfree(tb);
6299         return 0;
6300 }
6301
6302 static void ath11k_wmi_pull_pdev_stats_base(const struct wmi_pdev_stats_base *src,
6303                                             struct ath11k_fw_stats_pdev *dst)
6304 {
6305         dst->ch_noise_floor = src->chan_nf;
6306         dst->tx_frame_count = src->tx_frame_count;
6307         dst->rx_frame_count = src->rx_frame_count;
6308         dst->rx_clear_count = src->rx_clear_count;
6309         dst->cycle_count = src->cycle_count;
6310         dst->phy_err_count = src->phy_err_count;
6311         dst->chan_tx_power = src->chan_tx_pwr;
6312 }
6313
6314 static void
6315 ath11k_wmi_pull_pdev_stats_tx(const struct wmi_pdev_stats_tx *src,
6316                               struct ath11k_fw_stats_pdev *dst)
6317 {
6318         dst->comp_queued = src->comp_queued;
6319         dst->comp_delivered = src->comp_delivered;
6320         dst->msdu_enqued = src->msdu_enqued;
6321         dst->mpdu_enqued = src->mpdu_enqued;
6322         dst->wmm_drop = src->wmm_drop;
6323         dst->local_enqued = src->local_enqued;
6324         dst->local_freed = src->local_freed;
6325         dst->hw_queued = src->hw_queued;
6326         dst->hw_reaped = src->hw_reaped;
6327         dst->underrun = src->underrun;
6328         dst->hw_paused = src->hw_paused;
6329         dst->tx_abort = src->tx_abort;
6330         dst->mpdus_requeued = src->mpdus_requeued;
6331         dst->tx_ko = src->tx_ko;
6332         dst->tx_xretry = src->tx_xretry;
6333         dst->data_rc = src->data_rc;
6334         dst->self_triggers = src->self_triggers;
6335         dst->sw_retry_failure = src->sw_retry_failure;
6336         dst->illgl_rate_phy_err = src->illgl_rate_phy_err;
6337         dst->pdev_cont_xretry = src->pdev_cont_xretry;
6338         dst->pdev_tx_timeout = src->pdev_tx_timeout;
6339         dst->pdev_resets = src->pdev_resets;
6340         dst->stateless_tid_alloc_failure = src->stateless_tid_alloc_failure;
6341         dst->phy_underrun = src->phy_underrun;
6342         dst->txop_ovf = src->txop_ovf;
6343         dst->seq_posted = src->seq_posted;
6344         dst->seq_failed_queueing = src->seq_failed_queueing;
6345         dst->seq_completed = src->seq_completed;
6346         dst->seq_restarted = src->seq_restarted;
6347         dst->mu_seq_posted = src->mu_seq_posted;
6348         dst->mpdus_sw_flush = src->mpdus_sw_flush;
6349         dst->mpdus_hw_filter = src->mpdus_hw_filter;
6350         dst->mpdus_truncated = src->mpdus_truncated;
6351         dst->mpdus_ack_failed = src->mpdus_ack_failed;
6352         dst->mpdus_expired = src->mpdus_expired;
6353 }
6354
6355 static void ath11k_wmi_pull_pdev_stats_rx(const struct wmi_pdev_stats_rx *src,
6356                                           struct ath11k_fw_stats_pdev *dst)
6357 {
6358         dst->mid_ppdu_route_change = src->mid_ppdu_route_change;
6359         dst->status_rcvd = src->status_rcvd;
6360         dst->r0_frags = src->r0_frags;
6361         dst->r1_frags = src->r1_frags;
6362         dst->r2_frags = src->r2_frags;
6363         dst->r3_frags = src->r3_frags;
6364         dst->htt_msdus = src->htt_msdus;
6365         dst->htt_mpdus = src->htt_mpdus;
6366         dst->loc_msdus = src->loc_msdus;
6367         dst->loc_mpdus = src->loc_mpdus;
6368         dst->oversize_amsdu = src->oversize_amsdu;
6369         dst->phy_errs = src->phy_errs;
6370         dst->phy_err_drop = src->phy_err_drop;
6371         dst->mpdu_errs = src->mpdu_errs;
6372         dst->rx_ovfl_errs = src->rx_ovfl_errs;
6373 }
6374
6375 static void
6376 ath11k_wmi_pull_vdev_stats(const struct wmi_vdev_stats *src,
6377                            struct ath11k_fw_stats_vdev *dst)
6378 {
6379         int i;
6380
6381         dst->vdev_id = src->vdev_id;
6382         dst->beacon_snr = src->beacon_snr;
6383         dst->data_snr = src->data_snr;
6384         dst->num_rx_frames = src->num_rx_frames;
6385         dst->num_rts_fail = src->num_rts_fail;
6386         dst->num_rts_success = src->num_rts_success;
6387         dst->num_rx_err = src->num_rx_err;
6388         dst->num_rx_discard = src->num_rx_discard;
6389         dst->num_tx_not_acked = src->num_tx_not_acked;
6390
6391         for (i = 0; i < ARRAY_SIZE(src->num_tx_frames); i++)
6392                 dst->num_tx_frames[i] = src->num_tx_frames[i];
6393
6394         for (i = 0; i < ARRAY_SIZE(src->num_tx_frames_retries); i++)
6395                 dst->num_tx_frames_retries[i] = src->num_tx_frames_retries[i];
6396
6397         for (i = 0; i < ARRAY_SIZE(src->num_tx_frames_failures); i++)
6398                 dst->num_tx_frames_failures[i] = src->num_tx_frames_failures[i];
6399
6400         for (i = 0; i < ARRAY_SIZE(src->tx_rate_history); i++)
6401                 dst->tx_rate_history[i] = src->tx_rate_history[i];
6402
6403         for (i = 0; i < ARRAY_SIZE(src->beacon_rssi_history); i++)
6404                 dst->beacon_rssi_history[i] = src->beacon_rssi_history[i];
6405 }
6406
6407 static void
6408 ath11k_wmi_pull_bcn_stats(const struct wmi_bcn_stats *src,
6409                           struct ath11k_fw_stats_bcn *dst)
6410 {
6411         dst->vdev_id = src->vdev_id;
6412         dst->tx_bcn_succ_cnt = src->tx_bcn_succ_cnt;
6413         dst->tx_bcn_outage_cnt = src->tx_bcn_outage_cnt;
6414 }
6415
6416 static int ath11k_wmi_tlv_rssi_chain_parse(struct ath11k_base *ab,
6417                                            u16 tag, u16 len,
6418                                            const void *ptr, void *data)
6419 {
6420         struct wmi_tlv_fw_stats_parse *parse = data;
6421         const struct wmi_stats_event *ev = parse->ev;
6422         struct ath11k_fw_stats *stats = parse->stats;
6423         struct ath11k *ar;
6424         struct ath11k_vif *arvif;
6425         struct ieee80211_sta *sta;
6426         struct ath11k_sta *arsta;
6427         const struct wmi_rssi_stats *stats_rssi = (const struct wmi_rssi_stats *)ptr;
6428         int j, ret = 0;
6429
6430         if (tag != WMI_TAG_RSSI_STATS)
6431                 return -EPROTO;
6432
6433         rcu_read_lock();
6434
6435         ar = ath11k_mac_get_ar_by_pdev_id(ab, ev->pdev_id);
6436         stats->stats_id = WMI_REQUEST_RSSI_PER_CHAIN_STAT;
6437
6438         ath11k_dbg(ab, ATH11K_DBG_WMI,
6439                    "stats vdev id %d mac %pM\n",
6440                    stats_rssi->vdev_id, stats_rssi->peer_macaddr.addr);
6441
6442         arvif = ath11k_mac_get_arvif(ar, stats_rssi->vdev_id);
6443         if (!arvif) {
6444                 ath11k_warn(ab, "not found vif for vdev id %d\n",
6445                             stats_rssi->vdev_id);
6446                 ret = -EPROTO;
6447                 goto exit;
6448         }
6449
6450         ath11k_dbg(ab, ATH11K_DBG_WMI,
6451                    "stats bssid %pM vif %p\n",
6452                    arvif->bssid, arvif->vif);
6453
6454         sta = ieee80211_find_sta_by_ifaddr(ar->hw,
6455                                            arvif->bssid,
6456                                            NULL);
6457         if (!sta) {
6458                 ath11k_dbg(ab, ATH11K_DBG_WMI,
6459                            "not found station of bssid %pM for rssi chain\n",
6460                            arvif->bssid);
6461                 goto exit;
6462         }
6463
6464         arsta = ath11k_sta_to_arsta(sta);
6465
6466         BUILD_BUG_ON(ARRAY_SIZE(arsta->chain_signal) >
6467                      ARRAY_SIZE(stats_rssi->rssi_avg_beacon));
6468
6469         for (j = 0; j < ARRAY_SIZE(arsta->chain_signal); j++) {
6470                 arsta->chain_signal[j] = stats_rssi->rssi_avg_beacon[j];
6471                 ath11k_dbg(ab, ATH11K_DBG_WMI,
6472                            "stats beacon rssi[%d] %d data rssi[%d] %d\n",
6473                            j,
6474                            stats_rssi->rssi_avg_beacon[j],
6475                            j,
6476                            stats_rssi->rssi_avg_data[j]);
6477         }
6478
6479 exit:
6480         rcu_read_unlock();
6481         return ret;
6482 }
6483
6484 static int ath11k_wmi_tlv_fw_stats_data_parse(struct ath11k_base *ab,
6485                                               struct wmi_tlv_fw_stats_parse *parse,
6486                                               const void *ptr,
6487                                               u16 len)
6488 {
6489         struct ath11k_fw_stats *stats = parse->stats;
6490         const struct wmi_stats_event *ev = parse->ev;
6491         struct ath11k *ar;
6492         struct ath11k_vif *arvif;
6493         struct ieee80211_sta *sta;
6494         struct ath11k_sta *arsta;
6495         int i, ret = 0;
6496         const void *data = ptr;
6497
6498         if (!ev) {
6499                 ath11k_warn(ab, "failed to fetch update stats ev");
6500                 return -EPROTO;
6501         }
6502
6503         stats->stats_id = 0;
6504
6505         rcu_read_lock();
6506
6507         ar = ath11k_mac_get_ar_by_pdev_id(ab, ev->pdev_id);
6508
6509         for (i = 0; i < ev->num_pdev_stats; i++) {
6510                 const struct wmi_pdev_stats *src;
6511                 struct ath11k_fw_stats_pdev *dst;
6512
6513                 src = data;
6514                 if (len < sizeof(*src)) {
6515                         ret = -EPROTO;
6516                         goto exit;
6517                 }
6518
6519                 stats->stats_id = WMI_REQUEST_PDEV_STAT;
6520
6521                 data += sizeof(*src);
6522                 len -= sizeof(*src);
6523
6524                 dst = kzalloc(sizeof(*dst), GFP_ATOMIC);
6525                 if (!dst)
6526                         continue;
6527
6528                 ath11k_wmi_pull_pdev_stats_base(&src->base, dst);
6529                 ath11k_wmi_pull_pdev_stats_tx(&src->tx, dst);
6530                 ath11k_wmi_pull_pdev_stats_rx(&src->rx, dst);
6531                 list_add_tail(&dst->list, &stats->pdevs);
6532         }
6533
6534         for (i = 0; i < ev->num_vdev_stats; i++) {
6535                 const struct wmi_vdev_stats *src;
6536                 struct ath11k_fw_stats_vdev *dst;
6537
6538                 src = data;
6539                 if (len < sizeof(*src)) {
6540                         ret = -EPROTO;
6541                         goto exit;
6542                 }
6543
6544                 stats->stats_id = WMI_REQUEST_VDEV_STAT;
6545
6546                 arvif = ath11k_mac_get_arvif(ar, src->vdev_id);
6547                 if (arvif) {
6548                         sta = ieee80211_find_sta_by_ifaddr(ar->hw,
6549                                                            arvif->bssid,
6550                                                            NULL);
6551                         if (sta) {
6552                                 arsta = ath11k_sta_to_arsta(sta);
6553                                 arsta->rssi_beacon = src->beacon_snr;
6554                                 ath11k_dbg(ab, ATH11K_DBG_WMI,
6555                                            "stats vdev id %d snr %d\n",
6556                                            src->vdev_id, src->beacon_snr);
6557                         } else {
6558                                 ath11k_dbg(ab, ATH11K_DBG_WMI,
6559                                            "not found station of bssid %pM for vdev stat\n",
6560                                            arvif->bssid);
6561                         }
6562                 }
6563
6564                 data += sizeof(*src);
6565                 len -= sizeof(*src);
6566
6567                 dst = kzalloc(sizeof(*dst), GFP_ATOMIC);
6568                 if (!dst)
6569                         continue;
6570
6571                 ath11k_wmi_pull_vdev_stats(src, dst);
6572                 list_add_tail(&dst->list, &stats->vdevs);
6573         }
6574
6575         for (i = 0; i < ev->num_bcn_stats; i++) {
6576                 const struct wmi_bcn_stats *src;
6577                 struct ath11k_fw_stats_bcn *dst;
6578
6579                 src = data;
6580                 if (len < sizeof(*src)) {
6581                         ret = -EPROTO;
6582                         goto exit;
6583                 }
6584
6585                 stats->stats_id = WMI_REQUEST_BCN_STAT;
6586
6587                 data += sizeof(*src);
6588                 len -= sizeof(*src);
6589
6590                 dst = kzalloc(sizeof(*dst), GFP_ATOMIC);
6591                 if (!dst)
6592                         continue;
6593
6594                 ath11k_wmi_pull_bcn_stats(src, dst);
6595                 list_add_tail(&dst->list, &stats->bcn);
6596         }
6597
6598 exit:
6599         rcu_read_unlock();
6600         return ret;
6601 }
6602
6603 static int ath11k_wmi_tlv_fw_stats_parse(struct ath11k_base *ab,
6604                                          u16 tag, u16 len,
6605                                          const void *ptr, void *data)
6606 {
6607         struct wmi_tlv_fw_stats_parse *parse = data;
6608         int ret = 0;
6609
6610         switch (tag) {
6611         case WMI_TAG_STATS_EVENT:
6612                 parse->ev = (struct wmi_stats_event *)ptr;
6613                 parse->stats->pdev_id = parse->ev->pdev_id;
6614                 break;
6615         case WMI_TAG_ARRAY_BYTE:
6616                 ret = ath11k_wmi_tlv_fw_stats_data_parse(ab, parse, ptr, len);
6617                 break;
6618         case WMI_TAG_PER_CHAIN_RSSI_STATS:
6619                 parse->rssi = (struct wmi_per_chain_rssi_stats *)ptr;
6620
6621                 if (parse->ev->stats_id & WMI_REQUEST_RSSI_PER_CHAIN_STAT)
6622                         parse->rssi_num = parse->rssi->num_per_chain_rssi_stats;
6623
6624                 ath11k_dbg(ab, ATH11K_DBG_WMI,
6625                            "stats id 0x%x num chain %d\n",
6626                            parse->ev->stats_id,
6627                            parse->rssi_num);
6628                 break;
6629         case WMI_TAG_ARRAY_STRUCT:
6630                 if (parse->rssi_num && !parse->chain_rssi_done) {
6631                         ret = ath11k_wmi_tlv_iter(ab, ptr, len,
6632                                                   ath11k_wmi_tlv_rssi_chain_parse,
6633                                                   parse);
6634                         if (ret) {
6635                                 ath11k_warn(ab, "failed to parse rssi chain %d\n",
6636                                             ret);
6637                                 return ret;
6638                         }
6639                         parse->chain_rssi_done = true;
6640                 }
6641                 break;
6642         default:
6643                 break;
6644         }
6645         return ret;
6646 }
6647
6648 int ath11k_wmi_pull_fw_stats(struct ath11k_base *ab, struct sk_buff *skb,
6649                              struct ath11k_fw_stats *stats)
6650 {
6651         struct wmi_tlv_fw_stats_parse parse = { };
6652
6653         stats->stats_id = 0;
6654         parse.stats = stats;
6655
6656         return ath11k_wmi_tlv_iter(ab, skb->data, skb->len,
6657                                    ath11k_wmi_tlv_fw_stats_parse,
6658                                    &parse);
6659 }
6660
6661 static void
6662 ath11k_wmi_fw_pdev_base_stats_fill(const struct ath11k_fw_stats_pdev *pdev,
6663                                    char *buf, u32 *length)
6664 {
6665         u32 len = *length;
6666         u32 buf_len = ATH11K_FW_STATS_BUF_SIZE;
6667
6668         len += scnprintf(buf + len, buf_len - len, "\n");
6669         len += scnprintf(buf + len, buf_len - len, "%30s\n",
6670                         "ath11k PDEV stats");
6671         len += scnprintf(buf + len, buf_len - len, "%30s\n\n",
6672                         "=================");
6673
6674         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6675                         "Channel noise floor", pdev->ch_noise_floor);
6676         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6677                         "Channel TX power", pdev->chan_tx_power);
6678         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6679                         "TX frame count", pdev->tx_frame_count);
6680         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6681                         "RX frame count", pdev->rx_frame_count);
6682         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6683                         "RX clear count", pdev->rx_clear_count);
6684         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6685                         "Cycle count", pdev->cycle_count);
6686         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6687                         "PHY error count", pdev->phy_err_count);
6688
6689         *length = len;
6690 }
6691
6692 static void
6693 ath11k_wmi_fw_pdev_tx_stats_fill(const struct ath11k_fw_stats_pdev *pdev,
6694                                  char *buf, u32 *length)
6695 {
6696         u32 len = *length;
6697         u32 buf_len = ATH11K_FW_STATS_BUF_SIZE;
6698
6699         len += scnprintf(buf + len, buf_len - len, "\n%30s\n",
6700                          "ath11k PDEV TX stats");
6701         len += scnprintf(buf + len, buf_len - len, "%30s\n\n",
6702                          "====================");
6703
6704         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6705                          "HTT cookies queued", pdev->comp_queued);
6706         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6707                          "HTT cookies disp.", pdev->comp_delivered);
6708         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6709                          "MSDU queued", pdev->msdu_enqued);
6710         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6711                          "MPDU queued", pdev->mpdu_enqued);
6712         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6713                          "MSDUs dropped", pdev->wmm_drop);
6714         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6715                          "Local enqued", pdev->local_enqued);
6716         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6717                          "Local freed", pdev->local_freed);
6718         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6719                          "HW queued", pdev->hw_queued);
6720         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6721                          "PPDUs reaped", pdev->hw_reaped);
6722         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6723                          "Num underruns", pdev->underrun);
6724         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6725                          "Num HW Paused", pdev->hw_paused);
6726         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6727                          "PPDUs cleaned", pdev->tx_abort);
6728         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6729                          "MPDUs requeued", pdev->mpdus_requeued);
6730         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6731                          "PPDU OK", pdev->tx_ko);
6732         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6733                          "Excessive retries", pdev->tx_xretry);
6734         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6735                          "HW rate", pdev->data_rc);
6736         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6737                          "Sched self triggers", pdev->self_triggers);
6738         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6739                          "Dropped due to SW retries",
6740                          pdev->sw_retry_failure);
6741         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6742                          "Illegal rate phy errors",
6743                          pdev->illgl_rate_phy_err);
6744         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6745                          "PDEV continuous xretry", pdev->pdev_cont_xretry);
6746         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6747                          "TX timeout", pdev->pdev_tx_timeout);
6748         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6749                          "PDEV resets", pdev->pdev_resets);
6750         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6751                          "Stateless TIDs alloc failures",
6752                          pdev->stateless_tid_alloc_failure);
6753         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6754                          "PHY underrun", pdev->phy_underrun);
6755         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6756                          "MPDU is more than txop limit", pdev->txop_ovf);
6757         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6758                          "Num sequences posted", pdev->seq_posted);
6759         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6760                          "Num seq failed queueing ", pdev->seq_failed_queueing);
6761         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6762                          "Num sequences completed ", pdev->seq_completed);
6763         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6764                          "Num sequences restarted ", pdev->seq_restarted);
6765         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6766                          "Num of MU sequences posted ", pdev->mu_seq_posted);
6767         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6768                          "Num of MPDUS SW flushed ", pdev->mpdus_sw_flush);
6769         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6770                          "Num of MPDUS HW filtered ", pdev->mpdus_hw_filter);
6771         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6772                          "Num of MPDUS truncated ", pdev->mpdus_truncated);
6773         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6774                          "Num of MPDUS ACK failed ", pdev->mpdus_ack_failed);
6775         len += scnprintf(buf + len, buf_len - len, "%30s %10u\n",
6776                          "Num of MPDUS expired ", pdev->mpdus_expired);
6777         *length = len;
6778 }
6779
6780 static void
6781 ath11k_wmi_fw_pdev_rx_stats_fill(const struct ath11k_fw_stats_pdev *pdev,
6782                                  char *buf, u32 *length)
6783 {
6784         u32 len = *length;
6785         u32 buf_len = ATH11K_FW_STATS_BUF_SIZE;
6786
6787         len += scnprintf(buf + len, buf_len - len, "\n%30s\n",
6788                          "ath11k PDEV RX stats");
6789         len += scnprintf(buf + len, buf_len - len, "%30s\n\n",
6790                          "====================");
6791
6792         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6793                          "Mid PPDU route change",
6794                          pdev->mid_ppdu_route_change);
6795         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6796                          "Tot. number of statuses", pdev->status_rcvd);
6797         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6798                          "Extra frags on rings 0", pdev->r0_frags);
6799         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6800                          "Extra frags on rings 1", pdev->r1_frags);
6801         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6802                          "Extra frags on rings 2", pdev->r2_frags);
6803         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6804                          "Extra frags on rings 3", pdev->r3_frags);
6805         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6806                          "MSDUs delivered to HTT", pdev->htt_msdus);
6807         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6808                          "MPDUs delivered to HTT", pdev->htt_mpdus);
6809         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6810                          "MSDUs delivered to stack", pdev->loc_msdus);
6811         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6812                          "MPDUs delivered to stack", pdev->loc_mpdus);
6813         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6814                          "Oversized AMSUs", pdev->oversize_amsdu);
6815         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6816                          "PHY errors", pdev->phy_errs);
6817         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6818                          "PHY errors drops", pdev->phy_err_drop);
6819         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6820                          "MPDU errors (FCS, MIC, ENC)", pdev->mpdu_errs);
6821         len += scnprintf(buf + len, buf_len - len, "%30s %10d\n",
6822                          "Overflow errors", pdev->rx_ovfl_errs);
6823         *length = len;
6824 }
6825
6826 static void
6827 ath11k_wmi_fw_vdev_stats_fill(struct ath11k *ar,
6828                               const struct ath11k_fw_stats_vdev *vdev,
6829                               char *buf, u32 *length)
6830 {
6831         u32 len = *length;
6832         u32 buf_len = ATH11K_FW_STATS_BUF_SIZE;
6833         struct ath11k_vif *arvif = ath11k_mac_get_arvif(ar, vdev->vdev_id);
6834         u8 *vif_macaddr;
6835         int i;
6836
6837         /* VDEV stats has all the active VDEVs of other PDEVs as well,
6838          * ignoring those not part of requested PDEV
6839          */
6840         if (!arvif)
6841                 return;
6842
6843         vif_macaddr = arvif->vif->addr;
6844
6845         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6846                          "VDEV ID", vdev->vdev_id);
6847         len += scnprintf(buf + len, buf_len - len, "%30s %pM\n",
6848                          "VDEV MAC address", vif_macaddr);
6849         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6850                          "beacon snr", vdev->beacon_snr);
6851         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6852                          "data snr", vdev->data_snr);
6853         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6854                          "num rx frames", vdev->num_rx_frames);
6855         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6856                          "num rts fail", vdev->num_rts_fail);
6857         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6858                          "num rts success", vdev->num_rts_success);
6859         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6860                          "num rx err", vdev->num_rx_err);
6861         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6862                          "num rx discard", vdev->num_rx_discard);
6863         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6864                          "num tx not acked", vdev->num_tx_not_acked);
6865
6866         for (i = 0 ; i < ARRAY_SIZE(vdev->num_tx_frames); i++)
6867                 len += scnprintf(buf + len, buf_len - len,
6868                                 "%25s [%02d] %u\n",
6869                                 "num tx frames", i,
6870                                 vdev->num_tx_frames[i]);
6871
6872         for (i = 0 ; i < ARRAY_SIZE(vdev->num_tx_frames_retries); i++)
6873                 len += scnprintf(buf + len, buf_len - len,
6874                                 "%25s [%02d] %u\n",
6875                                 "num tx frames retries", i,
6876                                 vdev->num_tx_frames_retries[i]);
6877
6878         for (i = 0 ; i < ARRAY_SIZE(vdev->num_tx_frames_failures); i++)
6879                 len += scnprintf(buf + len, buf_len - len,
6880                                 "%25s [%02d] %u\n",
6881                                 "num tx frames failures", i,
6882                                 vdev->num_tx_frames_failures[i]);
6883
6884         for (i = 0 ; i < ARRAY_SIZE(vdev->tx_rate_history); i++)
6885                 len += scnprintf(buf + len, buf_len - len,
6886                                 "%25s [%02d] 0x%08x\n",
6887                                 "tx rate history", i,
6888                                 vdev->tx_rate_history[i]);
6889
6890         for (i = 0 ; i < ARRAY_SIZE(vdev->beacon_rssi_history); i++)
6891                 len += scnprintf(buf + len, buf_len - len,
6892                                 "%25s [%02d] %u\n",
6893                                 "beacon rssi history", i,
6894                                 vdev->beacon_rssi_history[i]);
6895
6896         len += scnprintf(buf + len, buf_len - len, "\n");
6897         *length = len;
6898 }
6899
6900 static void
6901 ath11k_wmi_fw_bcn_stats_fill(struct ath11k *ar,
6902                              const struct ath11k_fw_stats_bcn *bcn,
6903                              char *buf, u32 *length)
6904 {
6905         u32 len = *length;
6906         u32 buf_len = ATH11K_FW_STATS_BUF_SIZE;
6907         struct ath11k_vif *arvif = ath11k_mac_get_arvif(ar, bcn->vdev_id);
6908         u8 *vdev_macaddr;
6909
6910         if (!arvif) {
6911                 ath11k_warn(ar->ab, "invalid vdev id %d in bcn stats",
6912                             bcn->vdev_id);
6913                 return;
6914         }
6915
6916         vdev_macaddr = arvif->vif->addr;
6917
6918         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6919                          "VDEV ID", bcn->vdev_id);
6920         len += scnprintf(buf + len, buf_len - len, "%30s %pM\n",
6921                          "VDEV MAC address", vdev_macaddr);
6922         len += scnprintf(buf + len, buf_len - len, "%30s\n\n",
6923                          "================");
6924         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6925                          "Num of beacon tx success", bcn->tx_bcn_succ_cnt);
6926         len += scnprintf(buf + len, buf_len - len, "%30s %u\n",
6927                          "Num of beacon tx failures", bcn->tx_bcn_outage_cnt);
6928
6929         len += scnprintf(buf + len, buf_len - len, "\n");
6930         *length = len;
6931 }
6932
6933 void ath11k_wmi_fw_stats_fill(struct ath11k *ar,
6934                               struct ath11k_fw_stats *fw_stats,
6935                               u32 stats_id, char *buf)
6936 {
6937         u32 len = 0;
6938         u32 buf_len = ATH11K_FW_STATS_BUF_SIZE;
6939         const struct ath11k_fw_stats_pdev *pdev;
6940         const struct ath11k_fw_stats_vdev *vdev;
6941         const struct ath11k_fw_stats_bcn *bcn;
6942         size_t num_bcn;
6943
6944         spin_lock_bh(&ar->data_lock);
6945
6946         if (stats_id == WMI_REQUEST_PDEV_STAT) {
6947                 pdev = list_first_entry_or_null(&fw_stats->pdevs,
6948                                                 struct ath11k_fw_stats_pdev, list);
6949                 if (!pdev) {
6950                         ath11k_warn(ar->ab, "failed to get pdev stats\n");
6951                         goto unlock;
6952                 }
6953
6954                 ath11k_wmi_fw_pdev_base_stats_fill(pdev, buf, &len);
6955                 ath11k_wmi_fw_pdev_tx_stats_fill(pdev, buf, &len);
6956                 ath11k_wmi_fw_pdev_rx_stats_fill(pdev, buf, &len);
6957         }
6958
6959         if (stats_id == WMI_REQUEST_VDEV_STAT) {
6960                 len += scnprintf(buf + len, buf_len - len, "\n");
6961                 len += scnprintf(buf + len, buf_len - len, "%30s\n",
6962                                  "ath11k VDEV stats");
6963                 len += scnprintf(buf + len, buf_len - len, "%30s\n\n",
6964                                  "=================");
6965
6966                 list_for_each_entry(vdev, &fw_stats->vdevs, list)
6967                         ath11k_wmi_fw_vdev_stats_fill(ar, vdev, buf, &len);
6968         }
6969
6970         if (stats_id == WMI_REQUEST_BCN_STAT) {
6971                 num_bcn = list_count_nodes(&fw_stats->bcn);
6972
6973                 len += scnprintf(buf + len, buf_len - len, "\n");
6974                 len += scnprintf(buf + len, buf_len - len, "%30s (%zu)\n",
6975                                  "ath11k Beacon stats", num_bcn);
6976                 len += scnprintf(buf + len, buf_len - len, "%30s\n\n",
6977                                  "===================");
6978
6979                 list_for_each_entry(bcn, &fw_stats->bcn, list)
6980                         ath11k_wmi_fw_bcn_stats_fill(ar, bcn, buf, &len);
6981         }
6982
6983 unlock:
6984         spin_unlock_bh(&ar->data_lock);
6985
6986         if (len >= buf_len)
6987                 buf[len - 1] = 0;
6988         else
6989                 buf[len] = 0;
6990 }
6991
6992 static void ath11k_wmi_op_ep_tx_credits(struct ath11k_base *ab)
6993 {
6994         /* try to send pending beacons first. they take priority */
6995         wake_up(&ab->wmi_ab.tx_credits_wq);
6996 }
6997
6998 static int ath11k_reg_11d_new_cc_event(struct ath11k_base *ab, struct sk_buff *skb)
6999 {
7000         const struct wmi_11d_new_cc_ev *ev;
7001         struct ath11k *ar;
7002         struct ath11k_pdev *pdev;
7003         const void **tb;
7004         int ret, i;
7005
7006         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
7007         if (IS_ERR(tb)) {
7008                 ret = PTR_ERR(tb);
7009                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
7010                 return ret;
7011         }
7012
7013         ev = tb[WMI_TAG_11D_NEW_COUNTRY_EVENT];
7014         if (!ev) {
7015                 kfree(tb);
7016                 ath11k_warn(ab, "failed to fetch 11d new cc ev");
7017                 return -EPROTO;
7018         }
7019
7020         spin_lock_bh(&ab->base_lock);
7021         memcpy(&ab->new_alpha2, &ev->new_alpha2, 2);
7022         spin_unlock_bh(&ab->base_lock);
7023
7024         ath11k_dbg(ab, ATH11K_DBG_WMI, "event 11d new cc %c%c\n",
7025                    ab->new_alpha2[0],
7026                    ab->new_alpha2[1]);
7027
7028         kfree(tb);
7029
7030         for (i = 0; i < ab->num_radios; i++) {
7031                 pdev = &ab->pdevs[i];
7032                 ar = pdev->ar;
7033                 ar->state_11d = ATH11K_11D_IDLE;
7034                 complete(&ar->completed_11d_scan);
7035         }
7036
7037         queue_work(ab->workqueue, &ab->update_11d_work);
7038
7039         return 0;
7040 }
7041
7042 static void ath11k_wmi_htc_tx_complete(struct ath11k_base *ab,
7043                                        struct sk_buff *skb)
7044 {
7045         struct ath11k_pdev_wmi *wmi = NULL;
7046         u32 i;
7047         u8 wmi_ep_count;
7048         u8 eid;
7049
7050         eid = ATH11K_SKB_CB(skb)->eid;
7051         dev_kfree_skb(skb);
7052
7053         if (eid >= ATH11K_HTC_EP_COUNT)
7054                 return;
7055
7056         wmi_ep_count = ab->htc.wmi_ep_count;
7057         if (wmi_ep_count > ab->hw_params.max_radios)
7058                 return;
7059
7060         for (i = 0; i < ab->htc.wmi_ep_count; i++) {
7061                 if (ab->wmi_ab.wmi[i].eid == eid) {
7062                         wmi = &ab->wmi_ab.wmi[i];
7063                         break;
7064                 }
7065         }
7066
7067         if (wmi)
7068                 wake_up(&wmi->tx_ce_desc_wq);
7069 }
7070
7071 static int ath11k_reg_chan_list_event(struct ath11k_base *ab, struct sk_buff *skb,
7072                                       enum wmi_reg_chan_list_cmd_type id)
7073 {
7074         struct cur_regulatory_info *reg_info;
7075         int ret;
7076
7077         reg_info = kzalloc(sizeof(*reg_info), GFP_ATOMIC);
7078         if (!reg_info)
7079                 return -ENOMEM;
7080
7081         if (id == WMI_REG_CHAN_LIST_CC_ID)
7082                 ret = ath11k_pull_reg_chan_list_update_ev(ab, skb, reg_info);
7083         else
7084                 ret = ath11k_pull_reg_chan_list_ext_update_ev(ab, skb, reg_info);
7085
7086         if (ret) {
7087                 ath11k_warn(ab, "failed to extract regulatory info\n");
7088                 goto mem_free;
7089         }
7090
7091         ret = ath11k_reg_handle_chan_list(ab, reg_info, IEEE80211_REG_UNSET_AP);
7092         if (ret) {
7093                 ath11k_warn(ab, "failed to process regulatory info %d\n", ret);
7094                 goto mem_free;
7095         }
7096
7097         kfree(reg_info);
7098         return 0;
7099
7100 mem_free:
7101         ath11k_reg_reset_info(reg_info);
7102         kfree(reg_info);
7103         return ret;
7104 }
7105
7106 static int ath11k_wmi_tlv_rdy_parse(struct ath11k_base *ab, u16 tag, u16 len,
7107                                     const void *ptr, void *data)
7108 {
7109         struct wmi_tlv_rdy_parse *rdy_parse = data;
7110         struct wmi_ready_event fixed_param;
7111         struct wmi_mac_addr *addr_list;
7112         struct ath11k_pdev *pdev;
7113         u32 num_mac_addr;
7114         int i;
7115
7116         switch (tag) {
7117         case WMI_TAG_READY_EVENT:
7118                 memset(&fixed_param, 0, sizeof(fixed_param));
7119                 memcpy(&fixed_param, (struct wmi_ready_event *)ptr,
7120                        min_t(u16, sizeof(fixed_param), len));
7121                 rdy_parse->num_extra_mac_addr =
7122                         fixed_param.ready_event_min.num_extra_mac_addr;
7123
7124                 ether_addr_copy(ab->mac_addr,
7125                                 fixed_param.ready_event_min.mac_addr.addr);
7126                 ab->pktlog_defs_checksum = fixed_param.pktlog_defs_checksum;
7127                 break;
7128         case WMI_TAG_ARRAY_FIXED_STRUCT:
7129                 addr_list = (struct wmi_mac_addr *)ptr;
7130                 num_mac_addr = rdy_parse->num_extra_mac_addr;
7131
7132                 if (!(ab->num_radios > 1 && num_mac_addr >= ab->num_radios))
7133                         break;
7134
7135                 for (i = 0; i < ab->num_radios; i++) {
7136                         pdev = &ab->pdevs[i];
7137                         ether_addr_copy(pdev->mac_addr, addr_list[i].addr);
7138                 }
7139                 ab->pdevs_macaddr_valid = true;
7140                 break;
7141         default:
7142                 break;
7143         }
7144
7145         return 0;
7146 }
7147
7148 static int ath11k_ready_event(struct ath11k_base *ab, struct sk_buff *skb)
7149 {
7150         struct wmi_tlv_rdy_parse rdy_parse = { };
7151         int ret;
7152
7153         ret = ath11k_wmi_tlv_iter(ab, skb->data, skb->len,
7154                                   ath11k_wmi_tlv_rdy_parse, &rdy_parse);
7155         if (ret) {
7156                 ath11k_warn(ab, "failed to parse tlv %d\n", ret);
7157                 return ret;
7158         }
7159
7160         ath11k_dbg(ab, ATH11K_DBG_WMI, "event ready");
7161
7162         complete(&ab->wmi_ab.unified_ready);
7163         return 0;
7164 }
7165
7166 static void ath11k_peer_delete_resp_event(struct ath11k_base *ab, struct sk_buff *skb)
7167 {
7168         struct wmi_peer_delete_resp_event peer_del_resp;
7169         struct ath11k *ar;
7170
7171         if (ath11k_pull_peer_del_resp_ev(ab, skb, &peer_del_resp) != 0) {
7172                 ath11k_warn(ab, "failed to extract peer delete resp");
7173                 return;
7174         }
7175
7176         ath11k_dbg(ab, ATH11K_DBG_WMI, "event peer delete resp");
7177
7178         rcu_read_lock();
7179         ar = ath11k_mac_get_ar_by_vdev_id(ab, peer_del_resp.vdev_id);
7180         if (!ar) {
7181                 ath11k_warn(ab, "invalid vdev id in peer delete resp ev %d",
7182                             peer_del_resp.vdev_id);
7183                 rcu_read_unlock();
7184                 return;
7185         }
7186
7187         complete(&ar->peer_delete_done);
7188         rcu_read_unlock();
7189         ath11k_dbg(ab, ATH11K_DBG_WMI, "peer delete resp for vdev id %d addr %pM\n",
7190                    peer_del_resp.vdev_id, peer_del_resp.peer_macaddr.addr);
7191 }
7192
7193 static void ath11k_vdev_delete_resp_event(struct ath11k_base *ab,
7194                                           struct sk_buff *skb)
7195 {
7196         struct ath11k *ar;
7197         u32 vdev_id = 0;
7198
7199         if (ath11k_pull_vdev_del_resp_ev(ab, skb, &vdev_id) != 0) {
7200                 ath11k_warn(ab, "failed to extract vdev delete resp");
7201                 return;
7202         }
7203
7204         rcu_read_lock();
7205         ar = ath11k_mac_get_ar_by_vdev_id(ab, vdev_id);
7206         if (!ar) {
7207                 ath11k_warn(ab, "invalid vdev id in vdev delete resp ev %d",
7208                             vdev_id);
7209                 rcu_read_unlock();
7210                 return;
7211         }
7212
7213         complete(&ar->vdev_delete_done);
7214
7215         rcu_read_unlock();
7216
7217         ath11k_dbg(ab, ATH11K_DBG_WMI, "event vdev delete resp for vdev id %d\n",
7218                    vdev_id);
7219 }
7220
7221 static inline const char *ath11k_wmi_vdev_resp_print(u32 vdev_resp_status)
7222 {
7223         switch (vdev_resp_status) {
7224         case WMI_VDEV_START_RESPONSE_INVALID_VDEVID:
7225                 return "invalid vdev id";
7226         case WMI_VDEV_START_RESPONSE_NOT_SUPPORTED:
7227                 return "not supported";
7228         case WMI_VDEV_START_RESPONSE_DFS_VIOLATION:
7229                 return "dfs violation";
7230         case WMI_VDEV_START_RESPONSE_INVALID_REGDOMAIN:
7231                 return "invalid regdomain";
7232         default:
7233                 return "unknown";
7234         }
7235 }
7236
7237 static void ath11k_vdev_start_resp_event(struct ath11k_base *ab, struct sk_buff *skb)
7238 {
7239         struct wmi_vdev_start_resp_event vdev_start_resp;
7240         struct ath11k *ar;
7241         u32 status;
7242
7243         if (ath11k_pull_vdev_start_resp_tlv(ab, skb, &vdev_start_resp) != 0) {
7244                 ath11k_warn(ab, "failed to extract vdev start resp");
7245                 return;
7246         }
7247
7248         ath11k_dbg(ab, ATH11K_DBG_WMI, "event start resp event");
7249
7250         rcu_read_lock();
7251         ar = ath11k_mac_get_ar_by_vdev_id(ab, vdev_start_resp.vdev_id);
7252         if (!ar) {
7253                 ath11k_warn(ab, "invalid vdev id in vdev start resp ev %d",
7254                             vdev_start_resp.vdev_id);
7255                 rcu_read_unlock();
7256                 return;
7257         }
7258
7259         ar->last_wmi_vdev_start_status = 0;
7260
7261         status = vdev_start_resp.status;
7262
7263         if (WARN_ON_ONCE(status)) {
7264                 ath11k_warn(ab, "vdev start resp error status %d (%s)\n",
7265                             status, ath11k_wmi_vdev_resp_print(status));
7266                 ar->last_wmi_vdev_start_status = status;
7267         }
7268
7269         complete(&ar->vdev_setup_done);
7270
7271         rcu_read_unlock();
7272
7273         ath11k_dbg(ab, ATH11K_DBG_WMI, "vdev start resp for vdev id %d",
7274                    vdev_start_resp.vdev_id);
7275 }
7276
7277 static void ath11k_bcn_tx_status_event(struct ath11k_base *ab, struct sk_buff *skb)
7278 {
7279         struct ath11k_vif *arvif;
7280         u32 vdev_id, tx_status;
7281
7282         if (ath11k_pull_bcn_tx_status_ev(ab, skb->data, skb->len,
7283                                          &vdev_id, &tx_status) != 0) {
7284                 ath11k_warn(ab, "failed to extract bcn tx status");
7285                 return;
7286         }
7287
7288         ath11k_dbg(ab, ATH11K_DBG_WMI, "event offload bcn tx status");
7289
7290         rcu_read_lock();
7291         arvif = ath11k_mac_get_arvif_by_vdev_id(ab, vdev_id);
7292         if (!arvif) {
7293                 ath11k_warn(ab, "invalid vdev id %d in bcn_tx_status",
7294                             vdev_id);
7295                 rcu_read_unlock();
7296                 return;
7297         }
7298         ath11k_mac_bcn_tx_event(arvif);
7299         rcu_read_unlock();
7300 }
7301
7302 static void ath11k_wmi_event_peer_sta_ps_state_chg(struct ath11k_base *ab,
7303                                                    struct sk_buff *skb)
7304 {
7305         const struct wmi_peer_sta_ps_state_chg_event *ev;
7306         struct ieee80211_sta *sta;
7307         struct ath11k_peer *peer;
7308         struct ath11k *ar;
7309         struct ath11k_sta *arsta;
7310         const void **tb;
7311         enum ath11k_wmi_peer_ps_state peer_previous_ps_state;
7312         int ret;
7313
7314         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
7315         if (IS_ERR(tb)) {
7316                 ret = PTR_ERR(tb);
7317                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
7318                 return;
7319         }
7320
7321         ev = tb[WMI_TAG_PEER_STA_PS_STATECHANGE_EVENT];
7322         if (!ev) {
7323                 ath11k_warn(ab, "failed to fetch sta ps change ev");
7324                 kfree(tb);
7325                 return;
7326         }
7327
7328         ath11k_dbg(ab, ATH11K_DBG_WMI,
7329                    "event peer sta ps change ev addr %pM state %u sup_bitmap %x ps_valid %u ts %u\n",
7330                    ev->peer_macaddr.addr, ev->peer_ps_state,
7331                    ev->ps_supported_bitmap, ev->peer_ps_valid,
7332                    ev->peer_ps_timestamp);
7333
7334         rcu_read_lock();
7335
7336         spin_lock_bh(&ab->base_lock);
7337
7338         peer = ath11k_peer_find_by_addr(ab, ev->peer_macaddr.addr);
7339
7340         if (!peer) {
7341                 spin_unlock_bh(&ab->base_lock);
7342                 ath11k_warn(ab, "peer not found %pM\n", ev->peer_macaddr.addr);
7343                 goto exit;
7344         }
7345
7346         ar = ath11k_mac_get_ar_by_vdev_id(ab, peer->vdev_id);
7347
7348         if (!ar) {
7349                 spin_unlock_bh(&ab->base_lock);
7350                 ath11k_warn(ab, "invalid vdev id in peer sta ps state change ev %d",
7351                             peer->vdev_id);
7352
7353                 goto exit;
7354         }
7355
7356         sta = peer->sta;
7357
7358         spin_unlock_bh(&ab->base_lock);
7359
7360         if (!sta) {
7361                 ath11k_warn(ab, "failed to find station entry %pM\n",
7362                             ev->peer_macaddr.addr);
7363                 goto exit;
7364         }
7365
7366         arsta = ath11k_sta_to_arsta(sta);
7367
7368         spin_lock_bh(&ar->data_lock);
7369
7370         peer_previous_ps_state = arsta->peer_ps_state;
7371         arsta->peer_ps_state = ev->peer_ps_state;
7372         arsta->peer_current_ps_valid = !!ev->peer_ps_valid;
7373
7374         if (test_bit(WMI_TLV_SERVICE_PEER_POWER_SAVE_DURATION_SUPPORT,
7375                      ar->ab->wmi_ab.svc_map)) {
7376                 if (!(ev->ps_supported_bitmap & WMI_PEER_PS_VALID) ||
7377                     !(ev->ps_supported_bitmap & WMI_PEER_PS_STATE_TIMESTAMP) ||
7378                     !ev->peer_ps_valid)
7379                         goto out;
7380
7381                 if (arsta->peer_ps_state == WMI_PEER_PS_STATE_ON) {
7382                         arsta->ps_start_time = ev->peer_ps_timestamp;
7383                         arsta->ps_start_jiffies = jiffies;
7384                 } else if (arsta->peer_ps_state == WMI_PEER_PS_STATE_OFF &&
7385                            peer_previous_ps_state == WMI_PEER_PS_STATE_ON) {
7386                         arsta->ps_total_duration = arsta->ps_total_duration +
7387                                         (ev->peer_ps_timestamp - arsta->ps_start_time);
7388                 }
7389
7390                 if (ar->ps_timekeeper_enable)
7391                         trace_ath11k_ps_timekeeper(ar, ev->peer_macaddr.addr,
7392                                                    ev->peer_ps_timestamp,
7393                                                    arsta->peer_ps_state);
7394         }
7395
7396 out:
7397         spin_unlock_bh(&ar->data_lock);
7398 exit:
7399         rcu_read_unlock();
7400         kfree(tb);
7401 }
7402
7403 static void ath11k_vdev_stopped_event(struct ath11k_base *ab, struct sk_buff *skb)
7404 {
7405         struct ath11k *ar;
7406         u32 vdev_id = 0;
7407
7408         if (ath11k_pull_vdev_stopped_param_tlv(ab, skb, &vdev_id) != 0) {
7409                 ath11k_warn(ab, "failed to extract vdev stopped event");
7410                 return;
7411         }
7412
7413         ath11k_dbg(ab, ATH11K_DBG_WMI, "event vdev stopped");
7414
7415         rcu_read_lock();
7416         ar = ath11k_mac_get_ar_by_vdev_id(ab, vdev_id);
7417         if (!ar) {
7418                 ath11k_warn(ab, "invalid vdev id in vdev stopped ev %d",
7419                             vdev_id);
7420                 rcu_read_unlock();
7421                 return;
7422         }
7423
7424         complete(&ar->vdev_setup_done);
7425
7426         rcu_read_unlock();
7427
7428         ath11k_dbg(ab, ATH11K_DBG_WMI, "vdev stopped for vdev id %d", vdev_id);
7429 }
7430
7431 static void ath11k_mgmt_rx_event(struct ath11k_base *ab, struct sk_buff *skb)
7432 {
7433         struct mgmt_rx_event_params rx_ev = {0};
7434         struct ath11k *ar;
7435         struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
7436         struct ieee80211_hdr *hdr;
7437         u16 fc;
7438         struct ieee80211_supported_band *sband;
7439
7440         if (ath11k_pull_mgmt_rx_params_tlv(ab, skb, &rx_ev) != 0) {
7441                 ath11k_warn(ab, "failed to extract mgmt rx event");
7442                 dev_kfree_skb(skb);
7443                 return;
7444         }
7445
7446         memset(status, 0, sizeof(*status));
7447
7448         ath11k_dbg(ab, ATH11K_DBG_MGMT, "event mgmt rx status %08x\n",
7449                    rx_ev.status);
7450
7451         rcu_read_lock();
7452         ar = ath11k_mac_get_ar_by_pdev_id(ab, rx_ev.pdev_id);
7453
7454         if (!ar) {
7455                 ath11k_warn(ab, "invalid pdev_id %d in mgmt_rx_event\n",
7456                             rx_ev.pdev_id);
7457                 dev_kfree_skb(skb);
7458                 goto exit;
7459         }
7460
7461         if ((test_bit(ATH11K_CAC_RUNNING, &ar->dev_flags)) ||
7462             (rx_ev.status & (WMI_RX_STATUS_ERR_DECRYPT |
7463             WMI_RX_STATUS_ERR_KEY_CACHE_MISS | WMI_RX_STATUS_ERR_CRC))) {
7464                 dev_kfree_skb(skb);
7465                 goto exit;
7466         }
7467
7468         if (rx_ev.status & WMI_RX_STATUS_ERR_MIC)
7469                 status->flag |= RX_FLAG_MMIC_ERROR;
7470
7471         if (rx_ev.chan_freq >= ATH11K_MIN_6G_FREQ &&
7472             rx_ev.chan_freq <= ATH11K_MAX_6G_FREQ) {
7473                 status->band = NL80211_BAND_6GHZ;
7474                 status->freq = rx_ev.chan_freq;
7475         } else if (rx_ev.channel >= 1 && rx_ev.channel <= 14) {
7476                 status->band = NL80211_BAND_2GHZ;
7477         } else if (rx_ev.channel >= 36 && rx_ev.channel <= ATH11K_MAX_5G_CHAN) {
7478                 status->band = NL80211_BAND_5GHZ;
7479         } else {
7480                 /* Shouldn't happen unless list of advertised channels to
7481                  * mac80211 has been changed.
7482                  */
7483                 WARN_ON_ONCE(1);
7484                 dev_kfree_skb(skb);
7485                 goto exit;
7486         }
7487
7488         if (rx_ev.phy_mode == MODE_11B &&
7489             (status->band == NL80211_BAND_5GHZ || status->band == NL80211_BAND_6GHZ))
7490                 ath11k_dbg(ab, ATH11K_DBG_WMI,
7491                            "mgmt rx 11b (CCK) on 5/6GHz, band = %d\n", status->band);
7492
7493         sband = &ar->mac.sbands[status->band];
7494
7495         if (status->band != NL80211_BAND_6GHZ)
7496                 status->freq = ieee80211_channel_to_frequency(rx_ev.channel,
7497                                                               status->band);
7498
7499         status->signal = rx_ev.snr + ATH11K_DEFAULT_NOISE_FLOOR;
7500         status->rate_idx = ath11k_mac_bitrate_to_idx(sband, rx_ev.rate / 100);
7501
7502         hdr = (struct ieee80211_hdr *)skb->data;
7503         fc = le16_to_cpu(hdr->frame_control);
7504
7505         /* Firmware is guaranteed to report all essential management frames via
7506          * WMI while it can deliver some extra via HTT. Since there can be
7507          * duplicates split the reporting wrt monitor/sniffing.
7508          */
7509         status->flag |= RX_FLAG_SKIP_MONITOR;
7510
7511         /* In case of PMF, FW delivers decrypted frames with Protected Bit set.
7512          * Don't clear that. Also, FW delivers broadcast management frames
7513          * (ex: group privacy action frames in mesh) as encrypted payload.
7514          */
7515         if (ieee80211_has_protected(hdr->frame_control) &&
7516             !is_multicast_ether_addr(ieee80211_get_DA(hdr))) {
7517                 status->flag |= RX_FLAG_DECRYPTED;
7518
7519                 if (!ieee80211_is_robust_mgmt_frame(skb)) {
7520                         status->flag |= RX_FLAG_IV_STRIPPED |
7521                                         RX_FLAG_MMIC_STRIPPED;
7522                         hdr->frame_control = __cpu_to_le16(fc &
7523                                              ~IEEE80211_FCTL_PROTECTED);
7524                 }
7525         }
7526
7527         if (ieee80211_is_beacon(hdr->frame_control))
7528                 ath11k_mac_handle_beacon(ar, skb);
7529
7530         ath11k_dbg(ab, ATH11K_DBG_MGMT,
7531                    "event mgmt rx skb %p len %d ftype %02x stype %02x\n",
7532                    skb, skb->len,
7533                    fc & IEEE80211_FCTL_FTYPE, fc & IEEE80211_FCTL_STYPE);
7534
7535         ath11k_dbg(ab, ATH11K_DBG_MGMT,
7536                    "event mgmt rx freq %d band %d snr %d, rate_idx %d\n",
7537                    status->freq, status->band, status->signal,
7538                    status->rate_idx);
7539
7540         ieee80211_rx_ni(ar->hw, skb);
7541
7542 exit:
7543         rcu_read_unlock();
7544 }
7545
7546 static void ath11k_mgmt_tx_compl_event(struct ath11k_base *ab, struct sk_buff *skb)
7547 {
7548         struct wmi_mgmt_tx_compl_event tx_compl_param = {0};
7549         struct ath11k *ar;
7550
7551         if (ath11k_pull_mgmt_tx_compl_param_tlv(ab, skb, &tx_compl_param) != 0) {
7552                 ath11k_warn(ab, "failed to extract mgmt tx compl event");
7553                 return;
7554         }
7555
7556         rcu_read_lock();
7557         ar = ath11k_mac_get_ar_by_pdev_id(ab, tx_compl_param.pdev_id);
7558         if (!ar) {
7559                 ath11k_warn(ab, "invalid pdev id %d in mgmt_tx_compl_event\n",
7560                             tx_compl_param.pdev_id);
7561                 goto exit;
7562         }
7563
7564         wmi_process_mgmt_tx_comp(ar, &tx_compl_param);
7565
7566         ath11k_dbg(ab, ATH11K_DBG_MGMT,
7567                    "event mgmt tx compl ev pdev_id %d, desc_id %d, status %d ack_rssi %d",
7568                    tx_compl_param.pdev_id, tx_compl_param.desc_id,
7569                    tx_compl_param.status, tx_compl_param.ack_rssi);
7570
7571 exit:
7572         rcu_read_unlock();
7573 }
7574
7575 static struct ath11k *ath11k_get_ar_on_scan_state(struct ath11k_base *ab,
7576                                                   u32 vdev_id,
7577                                                   enum ath11k_scan_state state)
7578 {
7579         int i;
7580         struct ath11k_pdev *pdev;
7581         struct ath11k *ar;
7582
7583         for (i = 0; i < ab->num_radios; i++) {
7584                 pdev = rcu_dereference(ab->pdevs_active[i]);
7585                 if (pdev && pdev->ar) {
7586                         ar = pdev->ar;
7587
7588                         spin_lock_bh(&ar->data_lock);
7589                         if (ar->scan.state == state &&
7590                             ar->scan.vdev_id == vdev_id) {
7591                                 spin_unlock_bh(&ar->data_lock);
7592                                 return ar;
7593                         }
7594                         spin_unlock_bh(&ar->data_lock);
7595                 }
7596         }
7597         return NULL;
7598 }
7599
7600 static void ath11k_scan_event(struct ath11k_base *ab, struct sk_buff *skb)
7601 {
7602         struct ath11k *ar;
7603         struct wmi_scan_event scan_ev = {0};
7604
7605         if (ath11k_pull_scan_ev(ab, skb, &scan_ev) != 0) {
7606                 ath11k_warn(ab, "failed to extract scan event");
7607                 return;
7608         }
7609
7610         rcu_read_lock();
7611
7612         /* In case the scan was cancelled, ex. during interface teardown,
7613          * the interface will not be found in active interfaces.
7614          * Rather, in such scenarios, iterate over the active pdev's to
7615          * search 'ar' if the corresponding 'ar' scan is ABORTING and the
7616          * aborting scan's vdev id matches this event info.
7617          */
7618         if (scan_ev.event_type == WMI_SCAN_EVENT_COMPLETED &&
7619             scan_ev.reason == WMI_SCAN_REASON_CANCELLED) {
7620                 ar = ath11k_get_ar_on_scan_state(ab, scan_ev.vdev_id,
7621                                                  ATH11K_SCAN_ABORTING);
7622                 if (!ar)
7623                         ar = ath11k_get_ar_on_scan_state(ab, scan_ev.vdev_id,
7624                                                          ATH11K_SCAN_RUNNING);
7625         } else {
7626                 ar = ath11k_mac_get_ar_by_vdev_id(ab, scan_ev.vdev_id);
7627         }
7628
7629         if (!ar) {
7630                 ath11k_warn(ab, "Received scan event for unknown vdev");
7631                 rcu_read_unlock();
7632                 return;
7633         }
7634
7635         spin_lock_bh(&ar->data_lock);
7636
7637         ath11k_dbg(ab, ATH11K_DBG_WMI,
7638                    "event scan %s type %d reason %d freq %d req_id %d scan_id %d vdev_id %d state %s (%d)\n",
7639                    ath11k_wmi_event_scan_type_str(scan_ev.event_type, scan_ev.reason),
7640                    scan_ev.event_type, scan_ev.reason, scan_ev.channel_freq,
7641                    scan_ev.scan_req_id, scan_ev.scan_id, scan_ev.vdev_id,
7642                    ath11k_scan_state_str(ar->scan.state), ar->scan.state);
7643
7644         switch (scan_ev.event_type) {
7645         case WMI_SCAN_EVENT_STARTED:
7646                 ath11k_wmi_event_scan_started(ar);
7647                 break;
7648         case WMI_SCAN_EVENT_COMPLETED:
7649                 ath11k_wmi_event_scan_completed(ar);
7650                 break;
7651         case WMI_SCAN_EVENT_BSS_CHANNEL:
7652                 ath11k_wmi_event_scan_bss_chan(ar);
7653                 break;
7654         case WMI_SCAN_EVENT_FOREIGN_CHAN:
7655                 ath11k_wmi_event_scan_foreign_chan(ar, scan_ev.channel_freq);
7656                 break;
7657         case WMI_SCAN_EVENT_START_FAILED:
7658                 ath11k_warn(ab, "received scan start failure event\n");
7659                 ath11k_wmi_event_scan_start_failed(ar);
7660                 break;
7661         case WMI_SCAN_EVENT_DEQUEUED:
7662                 __ath11k_mac_scan_finish(ar);
7663                 break;
7664         case WMI_SCAN_EVENT_PREEMPTED:
7665         case WMI_SCAN_EVENT_RESTARTED:
7666         case WMI_SCAN_EVENT_FOREIGN_CHAN_EXIT:
7667         default:
7668                 break;
7669         }
7670
7671         spin_unlock_bh(&ar->data_lock);
7672
7673         rcu_read_unlock();
7674 }
7675
7676 static void ath11k_peer_sta_kickout_event(struct ath11k_base *ab, struct sk_buff *skb)
7677 {
7678         struct wmi_peer_sta_kickout_arg arg = {};
7679         struct ieee80211_sta *sta;
7680         struct ath11k_peer *peer;
7681         struct ath11k *ar;
7682         u32 vdev_id;
7683
7684         if (ath11k_pull_peer_sta_kickout_ev(ab, skb, &arg) != 0) {
7685                 ath11k_warn(ab, "failed to extract peer sta kickout event");
7686                 return;
7687         }
7688
7689         rcu_read_lock();
7690
7691         spin_lock_bh(&ab->base_lock);
7692
7693         peer = ath11k_peer_find_by_addr(ab, arg.mac_addr);
7694
7695         if (!peer) {
7696                 ath11k_warn(ab, "peer not found %pM\n",
7697                             arg.mac_addr);
7698                 spin_unlock_bh(&ab->base_lock);
7699                 goto exit;
7700         }
7701
7702         vdev_id = peer->vdev_id;
7703
7704         spin_unlock_bh(&ab->base_lock);
7705
7706         ar = ath11k_mac_get_ar_by_vdev_id(ab, vdev_id);
7707         if (!ar) {
7708                 ath11k_warn(ab, "invalid vdev id in peer sta kickout ev %d",
7709                             peer->vdev_id);
7710                 goto exit;
7711         }
7712
7713         sta = ieee80211_find_sta_by_ifaddr(ar->hw,
7714                                            arg.mac_addr, NULL);
7715         if (!sta) {
7716                 ath11k_warn(ab, "Spurious quick kickout for STA %pM\n",
7717                             arg.mac_addr);
7718                 goto exit;
7719         }
7720
7721         ath11k_dbg(ab, ATH11K_DBG_WMI, "event peer sta kickout %pM",
7722                    arg.mac_addr);
7723
7724         ieee80211_report_low_ack(sta, 10);
7725
7726 exit:
7727         rcu_read_unlock();
7728 }
7729
7730 static void ath11k_roam_event(struct ath11k_base *ab, struct sk_buff *skb)
7731 {
7732         struct wmi_roam_event roam_ev = {};
7733         struct ath11k *ar;
7734
7735         if (ath11k_pull_roam_ev(ab, skb, &roam_ev) != 0) {
7736                 ath11k_warn(ab, "failed to extract roam event");
7737                 return;
7738         }
7739
7740         ath11k_dbg(ab, ATH11K_DBG_WMI,
7741                    "event roam vdev %u reason 0x%08x rssi %d\n",
7742                    roam_ev.vdev_id, roam_ev.reason, roam_ev.rssi);
7743
7744         rcu_read_lock();
7745         ar = ath11k_mac_get_ar_by_vdev_id(ab, roam_ev.vdev_id);
7746         if (!ar) {
7747                 ath11k_warn(ab, "invalid vdev id in roam ev %d",
7748                             roam_ev.vdev_id);
7749                 rcu_read_unlock();
7750                 return;
7751         }
7752
7753         if (roam_ev.reason >= WMI_ROAM_REASON_MAX)
7754                 ath11k_warn(ab, "ignoring unknown roam event reason %d on vdev %i\n",
7755                             roam_ev.reason, roam_ev.vdev_id);
7756
7757         switch (roam_ev.reason) {
7758         case WMI_ROAM_REASON_BEACON_MISS:
7759                 ath11k_mac_handle_beacon_miss(ar, roam_ev.vdev_id);
7760                 break;
7761         case WMI_ROAM_REASON_BETTER_AP:
7762         case WMI_ROAM_REASON_LOW_RSSI:
7763         case WMI_ROAM_REASON_SUITABLE_AP_FOUND:
7764         case WMI_ROAM_REASON_HO_FAILED:
7765                 ath11k_warn(ab, "ignoring not implemented roam event reason %d on vdev %i\n",
7766                             roam_ev.reason, roam_ev.vdev_id);
7767                 break;
7768         }
7769
7770         rcu_read_unlock();
7771 }
7772
7773 static void ath11k_chan_info_event(struct ath11k_base *ab, struct sk_buff *skb)
7774 {
7775         struct wmi_chan_info_event ch_info_ev = {0};
7776         struct ath11k *ar;
7777         struct survey_info *survey;
7778         int idx;
7779         /* HW channel counters frequency value in hertz */
7780         u32 cc_freq_hz = ab->cc_freq_hz;
7781
7782         if (ath11k_pull_chan_info_ev(ab, skb->data, skb->len, &ch_info_ev) != 0) {
7783                 ath11k_warn(ab, "failed to extract chan info event");
7784                 return;
7785         }
7786
7787         ath11k_dbg(ab, ATH11K_DBG_WMI,
7788                    "event chan info vdev_id %d err_code %d freq %d cmd_flags %d noise_floor %d rx_clear_count %d cycle_count %d mac_clk_mhz %d\n",
7789                    ch_info_ev.vdev_id, ch_info_ev.err_code, ch_info_ev.freq,
7790                    ch_info_ev.cmd_flags, ch_info_ev.noise_floor,
7791                    ch_info_ev.rx_clear_count, ch_info_ev.cycle_count,
7792                    ch_info_ev.mac_clk_mhz);
7793
7794         if (ch_info_ev.cmd_flags == WMI_CHAN_INFO_END_RESP) {
7795                 ath11k_dbg(ab, ATH11K_DBG_WMI, "chan info report completed\n");
7796                 return;
7797         }
7798
7799         rcu_read_lock();
7800         ar = ath11k_mac_get_ar_by_vdev_id(ab, ch_info_ev.vdev_id);
7801         if (!ar) {
7802                 ath11k_warn(ab, "invalid vdev id in chan info ev %d",
7803                             ch_info_ev.vdev_id);
7804                 rcu_read_unlock();
7805                 return;
7806         }
7807         spin_lock_bh(&ar->data_lock);
7808
7809         switch (ar->scan.state) {
7810         case ATH11K_SCAN_IDLE:
7811         case ATH11K_SCAN_STARTING:
7812                 ath11k_warn(ab, "received chan info event without a scan request, ignoring\n");
7813                 goto exit;
7814         case ATH11K_SCAN_RUNNING:
7815         case ATH11K_SCAN_ABORTING:
7816                 break;
7817         }
7818
7819         idx = freq_to_idx(ar, ch_info_ev.freq);
7820         if (idx >= ARRAY_SIZE(ar->survey)) {
7821                 ath11k_warn(ab, "chan info: invalid frequency %d (idx %d out of bounds)\n",
7822                             ch_info_ev.freq, idx);
7823                 goto exit;
7824         }
7825
7826         /* If FW provides MAC clock frequency in Mhz, overriding the initialized
7827          * HW channel counters frequency value
7828          */
7829         if (ch_info_ev.mac_clk_mhz)
7830                 cc_freq_hz = (ch_info_ev.mac_clk_mhz * 1000);
7831
7832         if (ch_info_ev.cmd_flags == WMI_CHAN_INFO_START_RESP) {
7833                 survey = &ar->survey[idx];
7834                 memset(survey, 0, sizeof(*survey));
7835                 survey->noise = ch_info_ev.noise_floor;
7836                 survey->filled = SURVEY_INFO_NOISE_DBM | SURVEY_INFO_TIME |
7837                                  SURVEY_INFO_TIME_BUSY;
7838                 survey->time = div_u64(ch_info_ev.cycle_count, cc_freq_hz);
7839                 survey->time_busy = div_u64(ch_info_ev.rx_clear_count, cc_freq_hz);
7840         }
7841 exit:
7842         spin_unlock_bh(&ar->data_lock);
7843         rcu_read_unlock();
7844 }
7845
7846 static void
7847 ath11k_pdev_bss_chan_info_event(struct ath11k_base *ab, struct sk_buff *skb)
7848 {
7849         struct wmi_pdev_bss_chan_info_event bss_ch_info_ev = {};
7850         struct survey_info *survey;
7851         struct ath11k *ar;
7852         u32 cc_freq_hz = ab->cc_freq_hz;
7853         u64 busy, total, tx, rx, rx_bss;
7854         int idx;
7855
7856         if (ath11k_pull_pdev_bss_chan_info_ev(ab, skb, &bss_ch_info_ev) != 0) {
7857                 ath11k_warn(ab, "failed to extract pdev bss chan info event");
7858                 return;
7859         }
7860
7861         busy = (u64)(bss_ch_info_ev.rx_clear_count_high) << 32 |
7862                         bss_ch_info_ev.rx_clear_count_low;
7863
7864         total = (u64)(bss_ch_info_ev.cycle_count_high) << 32 |
7865                         bss_ch_info_ev.cycle_count_low;
7866
7867         tx = (u64)(bss_ch_info_ev.tx_cycle_count_high) << 32 |
7868                         bss_ch_info_ev.tx_cycle_count_low;
7869
7870         rx = (u64)(bss_ch_info_ev.rx_cycle_count_high) << 32 |
7871                         bss_ch_info_ev.rx_cycle_count_low;
7872
7873         rx_bss = (u64)(bss_ch_info_ev.rx_bss_cycle_count_high) << 32 |
7874                         bss_ch_info_ev.rx_bss_cycle_count_low;
7875
7876         ath11k_dbg(ab, ATH11K_DBG_WMI,
7877                    "event pdev bss chan info:\n pdev_id: %d freq: %d noise: %d cycle: busy %llu total %llu tx %llu rx %llu rx_bss %llu\n",
7878                    bss_ch_info_ev.pdev_id, bss_ch_info_ev.freq,
7879                    bss_ch_info_ev.noise_floor, busy, total,
7880                    tx, rx, rx_bss);
7881
7882         rcu_read_lock();
7883         ar = ath11k_mac_get_ar_by_pdev_id(ab, bss_ch_info_ev.pdev_id);
7884
7885         if (!ar) {
7886                 ath11k_warn(ab, "invalid pdev id %d in bss_chan_info event\n",
7887                             bss_ch_info_ev.pdev_id);
7888                 rcu_read_unlock();
7889                 return;
7890         }
7891
7892         spin_lock_bh(&ar->data_lock);
7893         idx = freq_to_idx(ar, bss_ch_info_ev.freq);
7894         if (idx >= ARRAY_SIZE(ar->survey)) {
7895                 ath11k_warn(ab, "bss chan info: invalid frequency %d (idx %d out of bounds)\n",
7896                             bss_ch_info_ev.freq, idx);
7897                 goto exit;
7898         }
7899
7900         survey = &ar->survey[idx];
7901
7902         survey->noise     = bss_ch_info_ev.noise_floor;
7903         survey->time      = div_u64(total, cc_freq_hz);
7904         survey->time_busy = div_u64(busy, cc_freq_hz);
7905         survey->time_rx   = div_u64(rx_bss, cc_freq_hz);
7906         survey->time_tx   = div_u64(tx, cc_freq_hz);
7907         survey->filled   |= (SURVEY_INFO_NOISE_DBM |
7908                              SURVEY_INFO_TIME |
7909                              SURVEY_INFO_TIME_BUSY |
7910                              SURVEY_INFO_TIME_RX |
7911                              SURVEY_INFO_TIME_TX);
7912 exit:
7913         spin_unlock_bh(&ar->data_lock);
7914         complete(&ar->bss_survey_done);
7915
7916         rcu_read_unlock();
7917 }
7918
7919 static void ath11k_vdev_install_key_compl_event(struct ath11k_base *ab,
7920                                                 struct sk_buff *skb)
7921 {
7922         struct wmi_vdev_install_key_complete_arg install_key_compl = {0};
7923         struct ath11k *ar;
7924
7925         if (ath11k_pull_vdev_install_key_compl_ev(ab, skb, &install_key_compl) != 0) {
7926                 ath11k_warn(ab, "failed to extract install key compl event");
7927                 return;
7928         }
7929
7930         ath11k_dbg(ab, ATH11K_DBG_WMI,
7931                    "event vdev install key ev idx %d flags %08x macaddr %pM status %d\n",
7932                    install_key_compl.key_idx, install_key_compl.key_flags,
7933                    install_key_compl.macaddr, install_key_compl.status);
7934
7935         rcu_read_lock();
7936         ar = ath11k_mac_get_ar_by_vdev_id(ab, install_key_compl.vdev_id);
7937         if (!ar) {
7938                 ath11k_warn(ab, "invalid vdev id in install key compl ev %d",
7939                             install_key_compl.vdev_id);
7940                 rcu_read_unlock();
7941                 return;
7942         }
7943
7944         ar->install_key_status = 0;
7945
7946         if (install_key_compl.status != WMI_VDEV_INSTALL_KEY_COMPL_STATUS_SUCCESS) {
7947                 ath11k_warn(ab, "install key failed for %pM status %d\n",
7948                             install_key_compl.macaddr, install_key_compl.status);
7949                 ar->install_key_status = install_key_compl.status;
7950         }
7951
7952         complete(&ar->install_key_done);
7953         rcu_read_unlock();
7954 }
7955
7956 static int  ath11k_wmi_tlv_services_parser(struct ath11k_base *ab,
7957                                            u16 tag, u16 len,
7958                                            const void *ptr, void *data)
7959 {
7960         const struct wmi_service_available_event *ev;
7961         u32 *wmi_ext2_service_bitmap;
7962         int i, j;
7963
7964         switch (tag) {
7965         case WMI_TAG_SERVICE_AVAILABLE_EVENT:
7966                 ev = (struct wmi_service_available_event *)ptr;
7967                 for (i = 0, j = WMI_MAX_SERVICE;
7968                         i < WMI_SERVICE_SEGMENT_BM_SIZE32 && j < WMI_MAX_EXT_SERVICE;
7969                         i++) {
7970                         do {
7971                                 if (ev->wmi_service_segment_bitmap[i] &
7972                                     BIT(j % WMI_AVAIL_SERVICE_BITS_IN_SIZE32))
7973                                         set_bit(j, ab->wmi_ab.svc_map);
7974                         } while (++j % WMI_AVAIL_SERVICE_BITS_IN_SIZE32);
7975                 }
7976
7977                 ath11k_dbg(ab, ATH11K_DBG_WMI,
7978                            "wmi_ext_service_bitmap 0:0x%04x, 1:0x%04x, 2:0x%04x, 3:0x%04x",
7979                            ev->wmi_service_segment_bitmap[0],
7980                            ev->wmi_service_segment_bitmap[1],
7981                            ev->wmi_service_segment_bitmap[2],
7982                            ev->wmi_service_segment_bitmap[3]);
7983                 break;
7984         case WMI_TAG_ARRAY_UINT32:
7985                 wmi_ext2_service_bitmap = (u32 *)ptr;
7986                 for (i = 0, j = WMI_MAX_EXT_SERVICE;
7987                         i < WMI_SERVICE_SEGMENT_BM_SIZE32 && j < WMI_MAX_EXT2_SERVICE;
7988                         i++) {
7989                         do {
7990                                 if (wmi_ext2_service_bitmap[i] &
7991                                     BIT(j % WMI_AVAIL_SERVICE_BITS_IN_SIZE32))
7992                                         set_bit(j, ab->wmi_ab.svc_map);
7993                         } while (++j % WMI_AVAIL_SERVICE_BITS_IN_SIZE32);
7994                 }
7995
7996                 ath11k_dbg(ab, ATH11K_DBG_WMI,
7997                            "wmi_ext2_service__bitmap  0:0x%04x, 1:0x%04x, 2:0x%04x, 3:0x%04x",
7998                            wmi_ext2_service_bitmap[0], wmi_ext2_service_bitmap[1],
7999                            wmi_ext2_service_bitmap[2], wmi_ext2_service_bitmap[3]);
8000                 break;
8001         }
8002         return 0;
8003 }
8004
8005 static void ath11k_service_available_event(struct ath11k_base *ab, struct sk_buff *skb)
8006 {
8007         int ret;
8008
8009         ret = ath11k_wmi_tlv_iter(ab, skb->data, skb->len,
8010                                   ath11k_wmi_tlv_services_parser,
8011                                   NULL);
8012         if (ret)
8013                 ath11k_warn(ab, "failed to parse services available tlv %d\n", ret);
8014
8015         ath11k_dbg(ab, ATH11K_DBG_WMI, "event service available");
8016 }
8017
8018 static void ath11k_peer_assoc_conf_event(struct ath11k_base *ab, struct sk_buff *skb)
8019 {
8020         struct wmi_peer_assoc_conf_arg peer_assoc_conf = {0};
8021         struct ath11k *ar;
8022
8023         if (ath11k_pull_peer_assoc_conf_ev(ab, skb, &peer_assoc_conf) != 0) {
8024                 ath11k_warn(ab, "failed to extract peer assoc conf event");
8025                 return;
8026         }
8027
8028         ath11k_dbg(ab, ATH11K_DBG_WMI,
8029                    "event peer assoc conf ev vdev id %d macaddr %pM\n",
8030                    peer_assoc_conf.vdev_id, peer_assoc_conf.macaddr);
8031
8032         rcu_read_lock();
8033         ar = ath11k_mac_get_ar_by_vdev_id(ab, peer_assoc_conf.vdev_id);
8034
8035         if (!ar) {
8036                 ath11k_warn(ab, "invalid vdev id in peer assoc conf ev %d",
8037                             peer_assoc_conf.vdev_id);
8038                 rcu_read_unlock();
8039                 return;
8040         }
8041
8042         complete(&ar->peer_assoc_done);
8043         rcu_read_unlock();
8044 }
8045
8046 static void ath11k_update_stats_event(struct ath11k_base *ab, struct sk_buff *skb)
8047 {
8048         struct ath11k_fw_stats stats = {};
8049         struct ath11k *ar;
8050         int ret;
8051
8052         INIT_LIST_HEAD(&stats.pdevs);
8053         INIT_LIST_HEAD(&stats.vdevs);
8054         INIT_LIST_HEAD(&stats.bcn);
8055
8056         ret = ath11k_wmi_pull_fw_stats(ab, skb, &stats);
8057         if (ret) {
8058                 ath11k_warn(ab, "failed to pull fw stats: %d\n", ret);
8059                 goto free;
8060         }
8061
8062         ath11k_dbg(ab, ATH11K_DBG_WMI, "event update stats");
8063
8064         rcu_read_lock();
8065         ar = ath11k_mac_get_ar_by_pdev_id(ab, stats.pdev_id);
8066         if (!ar) {
8067                 rcu_read_unlock();
8068                 ath11k_warn(ab, "failed to get ar for pdev_id %d: %d\n",
8069                             stats.pdev_id, ret);
8070                 goto free;
8071         }
8072
8073         spin_lock_bh(&ar->data_lock);
8074
8075         /* WMI_REQUEST_PDEV_STAT can be requested via .get_txpower mac ops or via
8076          * debugfs fw stats. Therefore, processing it separately.
8077          */
8078         if (stats.stats_id == WMI_REQUEST_PDEV_STAT) {
8079                 list_splice_tail_init(&stats.pdevs, &ar->fw_stats.pdevs);
8080                 ar->fw_stats_done = true;
8081                 goto complete;
8082         }
8083
8084         /* WMI_REQUEST_VDEV_STAT, WMI_REQUEST_BCN_STAT and WMI_REQUEST_RSSI_PER_CHAIN_STAT
8085          * are currently requested only via debugfs fw stats. Hence, processing these
8086          * in debugfs context
8087          */
8088         ath11k_debugfs_fw_stats_process(ar, &stats);
8089
8090 complete:
8091         complete(&ar->fw_stats_complete);
8092         rcu_read_unlock();
8093         spin_unlock_bh(&ar->data_lock);
8094
8095         /* Since the stats's pdev, vdev and beacon list are spliced and reinitialised
8096          * at this point, no need to free the individual list.
8097          */
8098         return;
8099
8100 free:
8101         ath11k_fw_stats_free(&stats);
8102 }
8103
8104 /* PDEV_CTL_FAILSAFE_CHECK_EVENT is received from FW when the frequency scanned
8105  * is not part of BDF CTL(Conformance test limits) table entries.
8106  */
8107 static void ath11k_pdev_ctl_failsafe_check_event(struct ath11k_base *ab,
8108                                                  struct sk_buff *skb)
8109 {
8110         const void **tb;
8111         const struct wmi_pdev_ctl_failsafe_chk_event *ev;
8112         int ret;
8113
8114         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
8115         if (IS_ERR(tb)) {
8116                 ret = PTR_ERR(tb);
8117                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
8118                 return;
8119         }
8120
8121         ev = tb[WMI_TAG_PDEV_CTL_FAILSAFE_CHECK_EVENT];
8122         if (!ev) {
8123                 ath11k_warn(ab, "failed to fetch pdev ctl failsafe check ev");
8124                 kfree(tb);
8125                 return;
8126         }
8127
8128         ath11k_dbg(ab, ATH11K_DBG_WMI,
8129                    "event pdev ctl failsafe check status %d\n",
8130                    ev->ctl_failsafe_status);
8131
8132         /* If ctl_failsafe_status is set to 1 FW will max out the Transmit power
8133          * to 10 dBm else the CTL power entry in the BDF would be picked up.
8134          */
8135         if (ev->ctl_failsafe_status != 0)
8136                 ath11k_warn(ab, "pdev ctl failsafe failure status %d",
8137                             ev->ctl_failsafe_status);
8138
8139         kfree(tb);
8140 }
8141
8142 static void
8143 ath11k_wmi_process_csa_switch_count_event(struct ath11k_base *ab,
8144                                           const struct wmi_pdev_csa_switch_ev *ev,
8145                                           const u32 *vdev_ids)
8146 {
8147         int i;
8148         struct ath11k_vif *arvif;
8149
8150         /* Finish CSA once the switch count becomes NULL */
8151         if (ev->current_switch_count)
8152                 return;
8153
8154         rcu_read_lock();
8155         for (i = 0; i < ev->num_vdevs; i++) {
8156                 arvif = ath11k_mac_get_arvif_by_vdev_id(ab, vdev_ids[i]);
8157
8158                 if (!arvif) {
8159                         ath11k_warn(ab, "Recvd csa status for unknown vdev %d",
8160                                     vdev_ids[i]);
8161                         continue;
8162                 }
8163
8164                 if (arvif->is_up && arvif->vif->bss_conf.csa_active)
8165                         ieee80211_csa_finish(arvif->vif);
8166         }
8167         rcu_read_unlock();
8168 }
8169
8170 static void
8171 ath11k_wmi_pdev_csa_switch_count_status_event(struct ath11k_base *ab,
8172                                               struct sk_buff *skb)
8173 {
8174         const void **tb;
8175         const struct wmi_pdev_csa_switch_ev *ev;
8176         const u32 *vdev_ids;
8177         int ret;
8178
8179         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
8180         if (IS_ERR(tb)) {
8181                 ret = PTR_ERR(tb);
8182                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
8183                 return;
8184         }
8185
8186         ev = tb[WMI_TAG_PDEV_CSA_SWITCH_COUNT_STATUS_EVENT];
8187         vdev_ids = tb[WMI_TAG_ARRAY_UINT32];
8188
8189         if (!ev || !vdev_ids) {
8190                 ath11k_warn(ab, "failed to fetch pdev csa switch count ev");
8191                 kfree(tb);
8192                 return;
8193         }
8194
8195         ath11k_dbg(ab, ATH11K_DBG_WMI,
8196                    "event pdev csa switch count %d for pdev %d, num_vdevs %d",
8197                    ev->current_switch_count, ev->pdev_id,
8198                    ev->num_vdevs);
8199
8200         ath11k_wmi_process_csa_switch_count_event(ab, ev, vdev_ids);
8201
8202         kfree(tb);
8203 }
8204
8205 static void
8206 ath11k_wmi_pdev_dfs_radar_detected_event(struct ath11k_base *ab, struct sk_buff *skb)
8207 {
8208         const void **tb;
8209         const struct wmi_pdev_radar_ev *ev;
8210         struct ath11k *ar;
8211         int ret;
8212
8213         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
8214         if (IS_ERR(tb)) {
8215                 ret = PTR_ERR(tb);
8216                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
8217                 return;
8218         }
8219
8220         ev = tb[WMI_TAG_PDEV_DFS_RADAR_DETECTION_EVENT];
8221
8222         if (!ev) {
8223                 ath11k_warn(ab, "failed to fetch pdev dfs radar detected ev");
8224                 kfree(tb);
8225                 return;
8226         }
8227
8228         ath11k_dbg(ab, ATH11K_DBG_WMI,
8229                    "event pdev dfs radar detected on pdev %d, detection mode %d, chan freq %d, chan_width %d, detector id %d, seg id %d, timestamp %d, chirp %d, freq offset %d, sidx %d",
8230                    ev->pdev_id, ev->detection_mode, ev->chan_freq, ev->chan_width,
8231                    ev->detector_id, ev->segment_id, ev->timestamp, ev->is_chirp,
8232                    ev->freq_offset, ev->sidx);
8233
8234         rcu_read_lock();
8235
8236         ar = ath11k_mac_get_ar_by_pdev_id(ab, ev->pdev_id);
8237
8238         if (!ar) {
8239                 ath11k_warn(ab, "radar detected in invalid pdev %d\n",
8240                             ev->pdev_id);
8241                 goto exit;
8242         }
8243
8244         ath11k_dbg(ar->ab, ATH11K_DBG_REG, "DFS Radar Detected in pdev %d\n",
8245                    ev->pdev_id);
8246
8247         if (ar->dfs_block_radar_events)
8248                 ath11k_info(ab, "DFS Radar detected, but ignored as requested\n");
8249         else
8250                 ieee80211_radar_detected(ar->hw);
8251
8252 exit:
8253         rcu_read_unlock();
8254
8255         kfree(tb);
8256 }
8257
8258 static void
8259 ath11k_wmi_pdev_temperature_event(struct ath11k_base *ab,
8260                                   struct sk_buff *skb)
8261 {
8262         struct ath11k *ar;
8263         const void **tb;
8264         const struct wmi_pdev_temperature_event *ev;
8265         int ret;
8266
8267         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
8268         if (IS_ERR(tb)) {
8269                 ret = PTR_ERR(tb);
8270                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
8271                 return;
8272         }
8273
8274         ev = tb[WMI_TAG_PDEV_TEMPERATURE_EVENT];
8275         if (!ev) {
8276                 ath11k_warn(ab, "failed to fetch pdev temp ev");
8277                 kfree(tb);
8278                 return;
8279         }
8280
8281         ath11k_dbg(ab, ATH11K_DBG_WMI, "event pdev temperature ev temp %d pdev_id %d\n",
8282                    ev->temp, ev->pdev_id);
8283
8284         rcu_read_lock();
8285
8286         ar = ath11k_mac_get_ar_by_pdev_id(ab, ev->pdev_id);
8287         if (!ar) {
8288                 ath11k_warn(ab, "invalid pdev id in pdev temperature ev %d", ev->pdev_id);
8289                 goto exit;
8290         }
8291
8292         ath11k_thermal_event_temperature(ar, ev->temp);
8293
8294 exit:
8295         rcu_read_unlock();
8296
8297         kfree(tb);
8298 }
8299
8300 static void ath11k_fils_discovery_event(struct ath11k_base *ab,
8301                                         struct sk_buff *skb)
8302 {
8303         const void **tb;
8304         const struct wmi_fils_discovery_event *ev;
8305         int ret;
8306
8307         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
8308         if (IS_ERR(tb)) {
8309                 ret = PTR_ERR(tb);
8310                 ath11k_warn(ab,
8311                             "failed to parse FILS discovery event tlv %d\n",
8312                             ret);
8313                 return;
8314         }
8315
8316         ath11k_dbg(ab, ATH11K_DBG_WMI, "event fils discovery");
8317
8318         ev = tb[WMI_TAG_HOST_SWFDA_EVENT];
8319         if (!ev) {
8320                 ath11k_warn(ab, "failed to fetch FILS discovery event\n");
8321                 kfree(tb);
8322                 return;
8323         }
8324
8325         ath11k_warn(ab,
8326                     "FILS discovery frame expected from host for vdev_id: %u, transmission scheduled at %u, next TBTT: %u\n",
8327                     ev->vdev_id, ev->fils_tt, ev->tbtt);
8328
8329         kfree(tb);
8330 }
8331
8332 static void ath11k_probe_resp_tx_status_event(struct ath11k_base *ab,
8333                                               struct sk_buff *skb)
8334 {
8335         const void **tb;
8336         const struct wmi_probe_resp_tx_status_event *ev;
8337         int ret;
8338
8339         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
8340         if (IS_ERR(tb)) {
8341                 ret = PTR_ERR(tb);
8342                 ath11k_warn(ab,
8343                             "failed to parse probe response transmission status event tlv: %d\n",
8344                             ret);
8345                 return;
8346         }
8347
8348         ath11k_dbg(ab, ATH11K_DBG_WMI, "event probe resp tx status");
8349
8350         ev = tb[WMI_TAG_OFFLOAD_PRB_RSP_TX_STATUS_EVENT];
8351         if (!ev) {
8352                 ath11k_warn(ab,
8353                             "failed to fetch probe response transmission status event");
8354                 kfree(tb);
8355                 return;
8356         }
8357
8358         if (ev->tx_status)
8359                 ath11k_warn(ab,
8360                             "Probe response transmission failed for vdev_id %u, status %u\n",
8361                             ev->vdev_id, ev->tx_status);
8362
8363         kfree(tb);
8364 }
8365
8366 static int ath11k_wmi_tlv_wow_wakeup_host_parse(struct ath11k_base *ab,
8367                                                 u16 tag, u16 len,
8368                                                 const void *ptr, void *data)
8369 {
8370         struct wmi_wow_ev_arg *ev = data;
8371         const char *wow_pg_fault;
8372         int wow_pg_len;
8373
8374         switch (tag) {
8375         case WMI_TAG_WOW_EVENT_INFO:
8376                 memcpy(ev, ptr, sizeof(*ev));
8377                 ath11k_dbg(ab, ATH11K_DBG_WMI, "wow wakeup host reason %d %s\n",
8378                            ev->wake_reason, wow_reason(ev->wake_reason));
8379                 break;
8380
8381         case WMI_TAG_ARRAY_BYTE:
8382                 if (ev && ev->wake_reason == WOW_REASON_PAGE_FAULT) {
8383                         wow_pg_fault = ptr;
8384                         /* the first 4 bytes are length */
8385                         wow_pg_len = *(int *)wow_pg_fault;
8386                         wow_pg_fault += sizeof(int);
8387                         ath11k_dbg(ab, ATH11K_DBG_WMI, "wow data_len = %d\n",
8388                                    wow_pg_len);
8389                         ath11k_dbg_dump(ab, ATH11K_DBG_WMI,
8390                                         "wow_event_info_type packet present",
8391                                         "wow_pg_fault ",
8392                                         wow_pg_fault,
8393                                         wow_pg_len);
8394                 }
8395                 break;
8396         default:
8397                 break;
8398         }
8399
8400         return 0;
8401 }
8402
8403 static void ath11k_wmi_event_wow_wakeup_host(struct ath11k_base *ab, struct sk_buff *skb)
8404 {
8405         struct wmi_wow_ev_arg ev = { };
8406         int ret;
8407
8408         ret = ath11k_wmi_tlv_iter(ab, skb->data, skb->len,
8409                                   ath11k_wmi_tlv_wow_wakeup_host_parse,
8410                                   &ev);
8411         if (ret) {
8412                 ath11k_warn(ab, "failed to parse wmi wow tlv: %d\n", ret);
8413                 return;
8414         }
8415
8416         ath11k_dbg(ab, ATH11K_DBG_WMI, "event wow wakeup host");
8417
8418         complete(&ab->wow.wakeup_completed);
8419 }
8420
8421 static void
8422 ath11k_wmi_diag_event(struct ath11k_base *ab,
8423                       struct sk_buff *skb)
8424 {
8425         ath11k_dbg(ab, ATH11K_DBG_WMI, "event diag");
8426
8427         trace_ath11k_wmi_diag(ab, skb->data, skb->len);
8428 }
8429
8430 static const char *ath11k_wmi_twt_add_dialog_event_status(u32 status)
8431 {
8432         switch (status) {
8433         case WMI_ADD_TWT_STATUS_OK:
8434                 return "ok";
8435         case WMI_ADD_TWT_STATUS_TWT_NOT_ENABLED:
8436                 return "twt disabled";
8437         case WMI_ADD_TWT_STATUS_USED_DIALOG_ID:
8438                 return "dialog id in use";
8439         case WMI_ADD_TWT_STATUS_INVALID_PARAM:
8440                 return "invalid parameters";
8441         case WMI_ADD_TWT_STATUS_NOT_READY:
8442                 return "not ready";
8443         case WMI_ADD_TWT_STATUS_NO_RESOURCE:
8444                 return "resource unavailable";
8445         case WMI_ADD_TWT_STATUS_NO_ACK:
8446                 return "no ack";
8447         case WMI_ADD_TWT_STATUS_NO_RESPONSE:
8448                 return "no response";
8449         case WMI_ADD_TWT_STATUS_DENIED:
8450                 return "denied";
8451         case WMI_ADD_TWT_STATUS_UNKNOWN_ERROR:
8452                 fallthrough;
8453         default:
8454                 return "unknown error";
8455         }
8456 }
8457
8458 static void ath11k_wmi_twt_add_dialog_event(struct ath11k_base *ab,
8459                                             struct sk_buff *skb)
8460 {
8461         const void **tb;
8462         const struct wmi_twt_add_dialog_event *ev;
8463         int ret;
8464
8465         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
8466         if (IS_ERR(tb)) {
8467                 ret = PTR_ERR(tb);
8468                 ath11k_warn(ab,
8469                             "failed to parse wmi twt add dialog status event tlv: %d\n",
8470                             ret);
8471                 return;
8472         }
8473
8474         ath11k_dbg(ab, ATH11K_DBG_WMI, "event twt add dialog");
8475
8476         ev = tb[WMI_TAG_TWT_ADD_DIALOG_COMPLETE_EVENT];
8477         if (!ev) {
8478                 ath11k_warn(ab, "failed to fetch twt add dialog wmi event\n");
8479                 goto exit;
8480         }
8481
8482         if (ev->status)
8483                 ath11k_warn(ab,
8484                             "wmi add twt dialog event vdev %d dialog id %d status %s\n",
8485                             ev->vdev_id, ev->dialog_id,
8486                             ath11k_wmi_twt_add_dialog_event_status(ev->status));
8487
8488 exit:
8489         kfree(tb);
8490 }
8491
8492 static void ath11k_wmi_gtk_offload_status_event(struct ath11k_base *ab,
8493                                                 struct sk_buff *skb)
8494 {
8495         const void **tb;
8496         const struct wmi_gtk_offload_status_event *ev;
8497         struct ath11k_vif *arvif;
8498         __be64 replay_ctr_be;
8499         u64    replay_ctr;
8500         int ret;
8501
8502         tb = ath11k_wmi_tlv_parse_alloc(ab, skb->data, skb->len, GFP_ATOMIC);
8503         if (IS_ERR(tb)) {
8504                 ret = PTR_ERR(tb);
8505                 ath11k_warn(ab, "failed to parse tlv: %d\n", ret);
8506                 return;
8507         }
8508
8509         ev = tb[WMI_TAG_GTK_OFFLOAD_STATUS_EVENT];
8510         if (!ev) {
8511                 ath11k_warn(ab, "failed to fetch gtk offload status ev");
8512                 kfree(tb);
8513                 return;
8514         }
8515
8516         rcu_read_lock();
8517
8518         arvif = ath11k_mac_get_arvif_by_vdev_id(ab, ev->vdev_id);
8519         if (!arvif) {
8520                 ath11k_warn(ab, "failed to get arvif for vdev_id:%d\n",
8521                             ev->vdev_id);
8522                 goto exit;
8523         }
8524
8525         ath11k_dbg(ab, ATH11K_DBG_WMI, "event gtk offload refresh_cnt %d\n",
8526                    ev->refresh_cnt);
8527         ath11k_dbg_dump(ab, ATH11K_DBG_WMI, "replay_cnt",
8528                         NULL, ev->replay_ctr.counter, GTK_REPLAY_COUNTER_BYTES);
8529
8530         replay_ctr =  ev->replay_ctr.word1;
8531         replay_ctr = (replay_ctr << 32) | ev->replay_ctr.word0;
8532         arvif->rekey_data.replay_ctr = replay_ctr;
8533
8534         /* supplicant expects big-endian replay counter */
8535         replay_ctr_be = cpu_to_be64(replay_ctr);
8536
8537         ieee80211_gtk_rekey_notify(arvif->vif, arvif->bssid,
8538                                    (void *)&replay_ctr_be, GFP_ATOMIC);
8539 exit:
8540         rcu_read_unlock();
8541
8542         kfree(tb);
8543 }
8544
8545 static void ath11k_wmi_tlv_op_rx(struct ath11k_base *ab, struct sk_buff *skb)
8546 {
8547         struct wmi_cmd_hdr *cmd_hdr;
8548         enum wmi_tlv_event_id id;
8549
8550         cmd_hdr = (struct wmi_cmd_hdr *)skb->data;
8551         id = FIELD_GET(WMI_CMD_HDR_CMD_ID, (cmd_hdr->cmd_id));
8552
8553         trace_ath11k_wmi_event(ab, id, skb->data, skb->len);
8554
8555         if (skb_pull(skb, sizeof(struct wmi_cmd_hdr)) == NULL)
8556                 goto out;
8557
8558         switch (id) {
8559                 /* Process all the WMI events here */
8560         case WMI_SERVICE_READY_EVENTID:
8561                 ath11k_service_ready_event(ab, skb);
8562                 break;
8563         case WMI_SERVICE_READY_EXT_EVENTID:
8564                 ath11k_service_ready_ext_event(ab, skb);
8565                 break;
8566         case WMI_SERVICE_READY_EXT2_EVENTID:
8567                 ath11k_service_ready_ext2_event(ab, skb);
8568                 break;
8569         case WMI_REG_CHAN_LIST_CC_EVENTID:
8570                 ath11k_reg_chan_list_event(ab, skb, WMI_REG_CHAN_LIST_CC_ID);
8571                 break;
8572         case WMI_REG_CHAN_LIST_CC_EXT_EVENTID:
8573                 ath11k_reg_chan_list_event(ab, skb, WMI_REG_CHAN_LIST_CC_EXT_ID);
8574                 break;
8575         case WMI_READY_EVENTID:
8576                 ath11k_ready_event(ab, skb);
8577                 break;
8578         case WMI_PEER_DELETE_RESP_EVENTID:
8579                 ath11k_peer_delete_resp_event(ab, skb);
8580                 break;
8581         case WMI_VDEV_START_RESP_EVENTID:
8582                 ath11k_vdev_start_resp_event(ab, skb);
8583                 break;
8584         case WMI_OFFLOAD_BCN_TX_STATUS_EVENTID:
8585                 ath11k_bcn_tx_status_event(ab, skb);
8586                 break;
8587         case WMI_VDEV_STOPPED_EVENTID:
8588                 ath11k_vdev_stopped_event(ab, skb);
8589                 break;
8590         case WMI_MGMT_RX_EVENTID:
8591                 ath11k_mgmt_rx_event(ab, skb);
8592                 /* mgmt_rx_event() owns the skb now! */
8593                 return;
8594         case WMI_MGMT_TX_COMPLETION_EVENTID:
8595                 ath11k_mgmt_tx_compl_event(ab, skb);
8596                 break;
8597         case WMI_SCAN_EVENTID:
8598                 ath11k_scan_event(ab, skb);
8599                 break;
8600         case WMI_PEER_STA_KICKOUT_EVENTID:
8601                 ath11k_peer_sta_kickout_event(ab, skb);
8602                 break;
8603         case WMI_ROAM_EVENTID:
8604                 ath11k_roam_event(ab, skb);
8605                 break;
8606         case WMI_CHAN_INFO_EVENTID:
8607                 ath11k_chan_info_event(ab, skb);
8608                 break;
8609         case WMI_PDEV_BSS_CHAN_INFO_EVENTID:
8610                 ath11k_pdev_bss_chan_info_event(ab, skb);
8611                 break;
8612         case WMI_VDEV_INSTALL_KEY_COMPLETE_EVENTID:
8613                 ath11k_vdev_install_key_compl_event(ab, skb);
8614                 break;
8615         case WMI_SERVICE_AVAILABLE_EVENTID:
8616                 ath11k_service_available_event(ab, skb);
8617                 break;
8618         case WMI_PEER_ASSOC_CONF_EVENTID:
8619                 ath11k_peer_assoc_conf_event(ab, skb);
8620                 break;
8621         case WMI_UPDATE_STATS_EVENTID:
8622                 ath11k_update_stats_event(ab, skb);
8623                 break;
8624         case WMI_PDEV_CTL_FAILSAFE_CHECK_EVENTID:
8625                 ath11k_pdev_ctl_failsafe_check_event(ab, skb);
8626                 break;
8627         case WMI_PDEV_CSA_SWITCH_COUNT_STATUS_EVENTID:
8628                 ath11k_wmi_pdev_csa_switch_count_status_event(ab, skb);
8629                 break;
8630         case WMI_PDEV_UTF_EVENTID:
8631                 ath11k_tm_wmi_event(ab, id, skb);
8632                 break;
8633         case WMI_PDEV_TEMPERATURE_EVENTID:
8634                 ath11k_wmi_pdev_temperature_event(ab, skb);
8635                 break;
8636         case WMI_PDEV_DMA_RING_BUF_RELEASE_EVENTID:
8637                 ath11k_wmi_pdev_dma_ring_buf_release_event(ab, skb);
8638                 break;
8639         case WMI_HOST_FILS_DISCOVERY_EVENTID:
8640                 ath11k_fils_discovery_event(ab, skb);
8641                 break;
8642         case WMI_OFFLOAD_PROB_RESP_TX_STATUS_EVENTID:
8643                 ath11k_probe_resp_tx_status_event(ab, skb);
8644                 break;
8645         case WMI_OBSS_COLOR_COLLISION_DETECTION_EVENTID:
8646                 ath11k_wmi_obss_color_collision_event(ab, skb);
8647                 break;
8648         case WMI_TWT_ADD_DIALOG_EVENTID:
8649                 ath11k_wmi_twt_add_dialog_event(ab, skb);
8650                 break;
8651         case WMI_PDEV_DFS_RADAR_DETECTION_EVENTID:
8652                 ath11k_wmi_pdev_dfs_radar_detected_event(ab, skb);
8653                 break;
8654         case WMI_VDEV_DELETE_RESP_EVENTID:
8655                 ath11k_vdev_delete_resp_event(ab, skb);
8656                 break;
8657         case WMI_WOW_WAKEUP_HOST_EVENTID:
8658                 ath11k_wmi_event_wow_wakeup_host(ab, skb);
8659                 break;
8660         case WMI_11D_NEW_COUNTRY_EVENTID:
8661                 ath11k_reg_11d_new_cc_event(ab, skb);
8662                 break;
8663         case WMI_DIAG_EVENTID:
8664                 ath11k_wmi_diag_event(ab, skb);
8665                 break;
8666         case WMI_PEER_STA_PS_STATECHG_EVENTID:
8667                 ath11k_wmi_event_peer_sta_ps_state_chg(ab, skb);
8668                 break;
8669         case WMI_GTK_OFFLOAD_STATUS_EVENTID:
8670                 ath11k_wmi_gtk_offload_status_event(ab, skb);
8671                 break;
8672         default:
8673                 ath11k_dbg(ab, ATH11K_DBG_WMI, "unsupported event id 0x%x\n", id);
8674                 break;
8675         }
8676
8677 out:
8678         dev_kfree_skb(skb);
8679 }
8680
8681 static int ath11k_connect_pdev_htc_service(struct ath11k_base *ab,
8682                                            u32 pdev_idx)
8683 {
8684         int status;
8685         u32 svc_id[] = { ATH11K_HTC_SVC_ID_WMI_CONTROL,
8686                          ATH11K_HTC_SVC_ID_WMI_CONTROL_MAC1,
8687                          ATH11K_HTC_SVC_ID_WMI_CONTROL_MAC2 };
8688
8689         struct ath11k_htc_svc_conn_req conn_req;
8690         struct ath11k_htc_svc_conn_resp conn_resp;
8691
8692         memset(&conn_req, 0, sizeof(conn_req));
8693         memset(&conn_resp, 0, sizeof(conn_resp));
8694
8695         /* these fields are the same for all service endpoints */
8696         conn_req.ep_ops.ep_tx_complete = ath11k_wmi_htc_tx_complete;
8697         conn_req.ep_ops.ep_rx_complete = ath11k_wmi_tlv_op_rx;
8698         conn_req.ep_ops.ep_tx_credits = ath11k_wmi_op_ep_tx_credits;
8699
8700         /* connect to control service */
8701         conn_req.service_id = svc_id[pdev_idx];
8702
8703         status = ath11k_htc_connect_service(&ab->htc, &conn_req, &conn_resp);
8704         if (status) {
8705                 ath11k_warn(ab, "failed to connect to WMI CONTROL service status: %d\n",
8706                             status);
8707                 return status;
8708         }
8709
8710         ab->wmi_ab.wmi_endpoint_id[pdev_idx] = conn_resp.eid;
8711         ab->wmi_ab.wmi[pdev_idx].eid = conn_resp.eid;
8712         ab->wmi_ab.max_msg_len[pdev_idx] = conn_resp.max_msg_len;
8713         init_waitqueue_head(&ab->wmi_ab.wmi[pdev_idx].tx_ce_desc_wq);
8714
8715         return 0;
8716 }
8717
8718 static int
8719 ath11k_wmi_send_unit_test_cmd(struct ath11k *ar,
8720                               struct wmi_unit_test_cmd ut_cmd,
8721                               u32 *test_args)
8722 {
8723         struct ath11k_pdev_wmi *wmi = ar->wmi;
8724         struct wmi_unit_test_cmd *cmd;
8725         struct sk_buff *skb;
8726         struct wmi_tlv *tlv;
8727         void *ptr;
8728         u32 *ut_cmd_args;
8729         int buf_len, arg_len;
8730         int ret;
8731         int i;
8732
8733         arg_len = sizeof(u32) * ut_cmd.num_args;
8734         buf_len = sizeof(ut_cmd) + arg_len + TLV_HDR_SIZE;
8735
8736         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, buf_len);
8737         if (!skb)
8738                 return -ENOMEM;
8739
8740         cmd = (struct wmi_unit_test_cmd *)skb->data;
8741         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_UNIT_TEST_CMD) |
8742                           FIELD_PREP(WMI_TLV_LEN, sizeof(ut_cmd) - TLV_HDR_SIZE);
8743
8744         cmd->vdev_id = ut_cmd.vdev_id;
8745         cmd->module_id = ut_cmd.module_id;
8746         cmd->num_args = ut_cmd.num_args;
8747         cmd->diag_token = ut_cmd.diag_token;
8748
8749         ptr = skb->data + sizeof(ut_cmd);
8750
8751         tlv = ptr;
8752         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_UINT32) |
8753                       FIELD_PREP(WMI_TLV_LEN, arg_len);
8754
8755         ptr += TLV_HDR_SIZE;
8756
8757         ut_cmd_args = ptr;
8758         for (i = 0; i < ut_cmd.num_args; i++)
8759                 ut_cmd_args[i] = test_args[i];
8760
8761         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_UNIT_TEST_CMDID);
8762
8763         if (ret) {
8764                 ath11k_warn(ar->ab, "failed to send WMI_UNIT_TEST CMD :%d\n",
8765                             ret);
8766                 dev_kfree_skb(skb);
8767         }
8768
8769         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
8770                    "cmd unit test module %d vdev %d n_args %d token %d\n",
8771                    cmd->module_id, cmd->vdev_id, cmd->num_args,
8772                    cmd->diag_token);
8773
8774         return ret;
8775 }
8776
8777 int ath11k_wmi_simulate_radar(struct ath11k *ar)
8778 {
8779         struct ath11k_vif *arvif;
8780         u32 dfs_args[DFS_MAX_TEST_ARGS];
8781         struct wmi_unit_test_cmd wmi_ut;
8782         bool arvif_found = false;
8783
8784         list_for_each_entry(arvif, &ar->arvifs, list) {
8785                 if (arvif->is_started && arvif->vdev_type == WMI_VDEV_TYPE_AP) {
8786                         arvif_found = true;
8787                         break;
8788                 }
8789         }
8790
8791         if (!arvif_found)
8792                 return -EINVAL;
8793
8794         dfs_args[DFS_TEST_CMDID] = 0;
8795         dfs_args[DFS_TEST_PDEV_ID] = ar->pdev->pdev_id;
8796         /* Currently we could pass segment_id(b0 - b1), chirp(b2)
8797          * freq offset (b3 - b10) to unit test. For simulation
8798          * purpose this can be set to 0 which is valid.
8799          */
8800         dfs_args[DFS_TEST_RADAR_PARAM] = 0;
8801
8802         wmi_ut.vdev_id = arvif->vdev_id;
8803         wmi_ut.module_id = DFS_UNIT_TEST_MODULE;
8804         wmi_ut.num_args = DFS_MAX_TEST_ARGS;
8805         wmi_ut.diag_token = DFS_UNIT_TEST_TOKEN;
8806
8807         ath11k_dbg(ar->ab, ATH11K_DBG_REG, "Triggering Radar Simulation\n");
8808
8809         return ath11k_wmi_send_unit_test_cmd(ar, wmi_ut, dfs_args);
8810 }
8811
8812 int ath11k_wmi_fw_dbglog_cfg(struct ath11k *ar, u32 *module_id_bitmap,
8813                              struct ath11k_fw_dbglog *dbglog)
8814 {
8815         struct ath11k_pdev_wmi *wmi = ar->wmi;
8816         struct wmi_debug_log_config_cmd_fixed_param *cmd;
8817         struct sk_buff *skb;
8818         struct wmi_tlv *tlv;
8819         int ret, len;
8820
8821         len = sizeof(*cmd) + TLV_HDR_SIZE + (MAX_MODULE_ID_BITMAP_WORDS * sizeof(u32));
8822         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
8823         if (!skb)
8824                 return -ENOMEM;
8825
8826         cmd = (struct wmi_debug_log_config_cmd_fixed_param *)skb->data;
8827         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_DEBUG_LOG_CONFIG_CMD) |
8828                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
8829         cmd->dbg_log_param = dbglog->param;
8830
8831         tlv = (struct wmi_tlv *)((u8 *)cmd + sizeof(*cmd));
8832         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_UINT32) |
8833                       FIELD_PREP(WMI_TLV_LEN, MAX_MODULE_ID_BITMAP_WORDS * sizeof(u32));
8834
8835         switch (dbglog->param) {
8836         case WMI_DEBUG_LOG_PARAM_LOG_LEVEL:
8837         case WMI_DEBUG_LOG_PARAM_VDEV_ENABLE:
8838         case WMI_DEBUG_LOG_PARAM_VDEV_DISABLE:
8839         case WMI_DEBUG_LOG_PARAM_VDEV_ENABLE_BITMAP:
8840                 cmd->value = dbglog->value;
8841                 break;
8842         case WMI_DEBUG_LOG_PARAM_MOD_ENABLE_BITMAP:
8843         case WMI_DEBUG_LOG_PARAM_WOW_MOD_ENABLE_BITMAP:
8844                 cmd->value = dbglog->value;
8845                 memcpy(tlv->value, module_id_bitmap,
8846                        MAX_MODULE_ID_BITMAP_WORDS * sizeof(u32));
8847                 /* clear current config to be used for next user config */
8848                 memset(module_id_bitmap, 0,
8849                        MAX_MODULE_ID_BITMAP_WORDS * sizeof(u32));
8850                 break;
8851         default:
8852                 dev_kfree_skb(skb);
8853                 return -EINVAL;
8854         }
8855
8856         ret = ath11k_wmi_cmd_send(wmi, skb, WMI_DBGLOG_CFG_CMDID);
8857         if (ret) {
8858                 ath11k_warn(ar->ab,
8859                             "failed to send WMI_DBGLOG_CFG_CMDID\n");
8860                 dev_kfree_skb(skb);
8861         }
8862
8863         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "cmd dbglog cfg");
8864
8865         return ret;
8866 }
8867
8868 int ath11k_wmi_connect(struct ath11k_base *ab)
8869 {
8870         u32 i;
8871         u8 wmi_ep_count;
8872
8873         wmi_ep_count = ab->htc.wmi_ep_count;
8874         if (wmi_ep_count > ab->hw_params.max_radios)
8875                 return -1;
8876
8877         for (i = 0; i < wmi_ep_count; i++)
8878                 ath11k_connect_pdev_htc_service(ab, i);
8879
8880         return 0;
8881 }
8882
8883 static void ath11k_wmi_pdev_detach(struct ath11k_base *ab, u8 pdev_id)
8884 {
8885         if (WARN_ON(pdev_id >= MAX_RADIOS))
8886                 return;
8887
8888         /* TODO: Deinit any pdev specific wmi resource */
8889 }
8890
8891 int ath11k_wmi_pdev_attach(struct ath11k_base *ab,
8892                            u8 pdev_id)
8893 {
8894         struct ath11k_pdev_wmi *wmi_handle;
8895
8896         if (pdev_id >= ab->hw_params.max_radios)
8897                 return -EINVAL;
8898
8899         wmi_handle = &ab->wmi_ab.wmi[pdev_id];
8900
8901         wmi_handle->wmi_ab = &ab->wmi_ab;
8902
8903         ab->wmi_ab.ab = ab;
8904         /* TODO: Init remaining resource specific to pdev */
8905
8906         return 0;
8907 }
8908
8909 int ath11k_wmi_attach(struct ath11k_base *ab)
8910 {
8911         int ret;
8912
8913         ret = ath11k_wmi_pdev_attach(ab, 0);
8914         if (ret)
8915                 return ret;
8916
8917         ab->wmi_ab.ab = ab;
8918         ab->wmi_ab.preferred_hw_mode = WMI_HOST_HW_MODE_MAX;
8919
8920         /* It's overwritten when service_ext_ready is handled */
8921         if (ab->hw_params.single_pdev_only && ab->hw_params.num_rxmda_per_pdev > 1)
8922                 ab->wmi_ab.preferred_hw_mode = WMI_HOST_HW_MODE_SINGLE;
8923
8924         /* TODO: Init remaining wmi soc resources required */
8925         init_completion(&ab->wmi_ab.service_ready);
8926         init_completion(&ab->wmi_ab.unified_ready);
8927
8928         return 0;
8929 }
8930
8931 void ath11k_wmi_detach(struct ath11k_base *ab)
8932 {
8933         int i;
8934
8935         /* TODO: Deinit wmi resource specific to SOC as required */
8936
8937         for (i = 0; i < ab->htc.wmi_ep_count; i++)
8938                 ath11k_wmi_pdev_detach(ab, i);
8939
8940         ath11k_wmi_free_dbring_caps(ab);
8941 }
8942
8943 int ath11k_wmi_hw_data_filter_cmd(struct ath11k *ar, u32 vdev_id,
8944                                   u32 filter_bitmap, bool enable)
8945 {
8946         struct wmi_hw_data_filter_cmd *cmd;
8947         struct sk_buff *skb;
8948         int len;
8949
8950         len = sizeof(*cmd);
8951         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
8952
8953         if (!skb)
8954                 return -ENOMEM;
8955
8956         cmd = (struct wmi_hw_data_filter_cmd *)skb->data;
8957         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_HW_DATA_FILTER_CMD) |
8958                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
8959
8960         cmd->vdev_id = vdev_id;
8961         cmd->enable = enable;
8962
8963         /* Set all modes in case of disable */
8964         if (cmd->enable)
8965                 cmd->hw_filter_bitmap = filter_bitmap;
8966         else
8967                 cmd->hw_filter_bitmap = ((u32)~0U);
8968
8969         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
8970                    "hw data filter enable %d filter_bitmap 0x%x\n",
8971                    enable, filter_bitmap);
8972
8973         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_HW_DATA_FILTER_CMDID);
8974 }
8975
8976 int ath11k_wmi_wow_host_wakeup_ind(struct ath11k *ar)
8977 {
8978         struct wmi_wow_host_wakeup_ind *cmd;
8979         struct sk_buff *skb;
8980         size_t len;
8981
8982         len = sizeof(*cmd);
8983         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
8984         if (!skb)
8985                 return -ENOMEM;
8986
8987         cmd = (struct wmi_wow_host_wakeup_ind *)skb->data;
8988         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
8989                                      WMI_TAG_WOW_HOSTWAKEUP_FROM_SLEEP_CMD) |
8990                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
8991
8992         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "tlv wow host wakeup ind\n");
8993
8994         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_WOW_HOSTWAKEUP_FROM_SLEEP_CMDID);
8995 }
8996
8997 int ath11k_wmi_wow_enable(struct ath11k *ar)
8998 {
8999         struct wmi_wow_enable_cmd *cmd;
9000         struct sk_buff *skb;
9001         int len;
9002
9003         len = sizeof(*cmd);
9004         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9005         if (!skb)
9006                 return -ENOMEM;
9007
9008         cmd = (struct wmi_wow_enable_cmd *)skb->data;
9009         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_WOW_ENABLE_CMD) |
9010                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9011
9012         cmd->enable = 1;
9013         cmd->pause_iface_config = WOW_IFACE_PAUSE_ENABLED;
9014         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "tlv wow enable\n");
9015
9016         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_WOW_ENABLE_CMDID);
9017 }
9018
9019 int ath11k_wmi_scan_prob_req_oui(struct ath11k *ar,
9020                                  const u8 mac_addr[ETH_ALEN])
9021 {
9022         struct sk_buff *skb;
9023         struct wmi_scan_prob_req_oui_cmd *cmd;
9024         u32 prob_req_oui;
9025         int len;
9026
9027         prob_req_oui = (((u32)mac_addr[0]) << 16) |
9028                        (((u32)mac_addr[1]) << 8) | mac_addr[2];
9029
9030         len = sizeof(*cmd);
9031         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9032         if (!skb)
9033                 return -ENOMEM;
9034
9035         cmd = (struct wmi_scan_prob_req_oui_cmd *)skb->data;
9036         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
9037                                      WMI_TAG_SCAN_PROB_REQ_OUI_CMD) |
9038                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9039         cmd->prob_req_oui = prob_req_oui;
9040
9041         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "scan prob req oui %d\n",
9042                    prob_req_oui);
9043
9044         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_SCAN_PROB_REQ_OUI_CMDID);
9045 }
9046
9047 int ath11k_wmi_wow_add_wakeup_event(struct ath11k *ar, u32 vdev_id,
9048                                     enum wmi_wow_wakeup_event event,
9049                                 u32 enable)
9050 {
9051         struct wmi_wow_add_del_event_cmd *cmd;
9052         struct sk_buff *skb;
9053         size_t len;
9054
9055         len = sizeof(*cmd);
9056         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9057         if (!skb)
9058                 return -ENOMEM;
9059
9060         cmd = (struct wmi_wow_add_del_event_cmd *)skb->data;
9061         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_WOW_ADD_DEL_EVT_CMD) |
9062                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9063
9064         cmd->vdev_id = vdev_id;
9065         cmd->is_add = enable;
9066         cmd->event_bitmap = (1 << event);
9067
9068         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "tlv wow add wakeup event %s enable %d vdev_id %d\n",
9069                    wow_wakeup_event(event), enable, vdev_id);
9070
9071         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_WOW_ENABLE_DISABLE_WAKE_EVENT_CMDID);
9072 }
9073
9074 int ath11k_wmi_wow_add_pattern(struct ath11k *ar, u32 vdev_id, u32 pattern_id,
9075                                const u8 *pattern, const u8 *mask,
9076                            int pattern_len, int pattern_offset)
9077 {
9078         struct wmi_wow_add_pattern_cmd *cmd;
9079         struct wmi_wow_bitmap_pattern *bitmap;
9080         struct wmi_tlv *tlv;
9081         struct sk_buff *skb;
9082         u8 *ptr;
9083         size_t len;
9084
9085         len = sizeof(*cmd) +
9086               sizeof(*tlv) +                    /* array struct */
9087               sizeof(*bitmap) +                 /* bitmap */
9088               sizeof(*tlv) +                    /* empty ipv4 sync */
9089               sizeof(*tlv) +                    /* empty ipv6 sync */
9090               sizeof(*tlv) +                    /* empty magic */
9091               sizeof(*tlv) +                    /* empty info timeout */
9092               sizeof(*tlv) + sizeof(u32);       /* ratelimit interval */
9093
9094         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9095         if (!skb)
9096                 return -ENOMEM;
9097
9098         /* cmd */
9099         ptr = (u8 *)skb->data;
9100         cmd = (struct wmi_wow_add_pattern_cmd *)ptr;
9101         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
9102                                      WMI_TAG_WOW_ADD_PATTERN_CMD) |
9103                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9104
9105         cmd->vdev_id = vdev_id;
9106         cmd->pattern_id = pattern_id;
9107         cmd->pattern_type = WOW_BITMAP_PATTERN;
9108
9109         ptr += sizeof(*cmd);
9110
9111         /* bitmap */
9112         tlv = (struct wmi_tlv *)ptr;
9113         tlv->header = FIELD_PREP(WMI_TLV_TAG,
9114                                  WMI_TAG_ARRAY_STRUCT) |
9115                       FIELD_PREP(WMI_TLV_LEN, sizeof(*bitmap));
9116
9117         ptr += sizeof(*tlv);
9118
9119         bitmap = (struct wmi_wow_bitmap_pattern *)ptr;
9120         bitmap->tlv_header = FIELD_PREP(WMI_TLV_TAG,
9121                                         WMI_TAG_WOW_BITMAP_PATTERN_T) |
9122                              FIELD_PREP(WMI_TLV_LEN, sizeof(*bitmap) - TLV_HDR_SIZE);
9123
9124         memcpy(bitmap->patternbuf, pattern, pattern_len);
9125         ath11k_ce_byte_swap(bitmap->patternbuf, roundup(pattern_len, 4));
9126         memcpy(bitmap->bitmaskbuf, mask, pattern_len);
9127         ath11k_ce_byte_swap(bitmap->bitmaskbuf, roundup(pattern_len, 4));
9128         bitmap->pattern_offset = pattern_offset;
9129         bitmap->pattern_len = pattern_len;
9130         bitmap->bitmask_len = pattern_len;
9131         bitmap->pattern_id = pattern_id;
9132
9133         ptr += sizeof(*bitmap);
9134
9135         /* ipv4 sync */
9136         tlv = (struct wmi_tlv *)ptr;
9137         tlv->header = FIELD_PREP(WMI_TLV_TAG,
9138                                  WMI_TAG_ARRAY_STRUCT) |
9139                       FIELD_PREP(WMI_TLV_LEN, 0);
9140
9141         ptr += sizeof(*tlv);
9142
9143         /* ipv6 sync */
9144         tlv = (struct wmi_tlv *)ptr;
9145         tlv->header = FIELD_PREP(WMI_TLV_TAG,
9146                                  WMI_TAG_ARRAY_STRUCT) |
9147                       FIELD_PREP(WMI_TLV_LEN, 0);
9148
9149         ptr += sizeof(*tlv);
9150
9151         /* magic */
9152         tlv = (struct wmi_tlv *)ptr;
9153         tlv->header = FIELD_PREP(WMI_TLV_TAG,
9154                                  WMI_TAG_ARRAY_STRUCT) |
9155                       FIELD_PREP(WMI_TLV_LEN, 0);
9156
9157         ptr += sizeof(*tlv);
9158
9159         /* pattern info timeout */
9160         tlv = (struct wmi_tlv *)ptr;
9161         tlv->header = FIELD_PREP(WMI_TLV_TAG,
9162                                  WMI_TAG_ARRAY_UINT32) |
9163                       FIELD_PREP(WMI_TLV_LEN, 0);
9164
9165         ptr += sizeof(*tlv);
9166
9167         /* ratelimit interval */
9168         tlv = (struct wmi_tlv *)ptr;
9169         tlv->header = FIELD_PREP(WMI_TLV_TAG,
9170                                  WMI_TAG_ARRAY_UINT32) |
9171                       FIELD_PREP(WMI_TLV_LEN, sizeof(u32));
9172
9173         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "tlv wow add pattern vdev_id %d pattern_id %d pattern_offset %d\n",
9174                    vdev_id, pattern_id, pattern_offset);
9175
9176         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_WOW_ADD_WAKE_PATTERN_CMDID);
9177 }
9178
9179 int ath11k_wmi_wow_del_pattern(struct ath11k *ar, u32 vdev_id, u32 pattern_id)
9180 {
9181         struct wmi_wow_del_pattern_cmd *cmd;
9182         struct sk_buff *skb;
9183         size_t len;
9184
9185         len = sizeof(*cmd);
9186         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9187         if (!skb)
9188                 return -ENOMEM;
9189
9190         cmd = (struct wmi_wow_del_pattern_cmd *)skb->data;
9191         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
9192                                      WMI_TAG_WOW_DEL_PATTERN_CMD) |
9193                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9194
9195         cmd->vdev_id = vdev_id;
9196         cmd->pattern_id = pattern_id;
9197         cmd->pattern_type = WOW_BITMAP_PATTERN;
9198
9199         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "tlv wow del pattern vdev_id %d pattern_id %d\n",
9200                    vdev_id, pattern_id);
9201
9202         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_WOW_DEL_WAKE_PATTERN_CMDID);
9203 }
9204
9205 static struct sk_buff *
9206 ath11k_wmi_op_gen_config_pno_start(struct ath11k *ar,
9207                                    u32 vdev_id,
9208                                        struct wmi_pno_scan_req *pno)
9209 {
9210         struct nlo_configured_parameters *nlo_list;
9211         struct wmi_wow_nlo_config_cmd *cmd;
9212         struct wmi_tlv *tlv;
9213         struct sk_buff *skb;
9214         u32 *channel_list;
9215         size_t len, nlo_list_len, channel_list_len;
9216         u8 *ptr;
9217         u32 i;
9218
9219         len = sizeof(*cmd) +
9220               sizeof(*tlv) +
9221               /* TLV place holder for array of structures
9222                * nlo_configured_parameters(nlo_list)
9223                */
9224               sizeof(*tlv);
9225               /* TLV place holder for array of uint32 channel_list */
9226
9227         channel_list_len = sizeof(u32) * pno->a_networks[0].channel_count;
9228         len += channel_list_len;
9229
9230         nlo_list_len = sizeof(*nlo_list) * pno->uc_networks_count;
9231         len += nlo_list_len;
9232
9233         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9234         if (!skb)
9235                 return ERR_PTR(-ENOMEM);
9236
9237         ptr = (u8 *)skb->data;
9238         cmd = (struct wmi_wow_nlo_config_cmd *)ptr;
9239         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_NLO_CONFIG_CMD) |
9240                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9241
9242         cmd->vdev_id = pno->vdev_id;
9243         cmd->flags = WMI_NLO_CONFIG_START | WMI_NLO_CONFIG_SSID_HIDE_EN;
9244
9245         /* current FW does not support min-max range for dwell time */
9246         cmd->active_dwell_time = pno->active_max_time;
9247         cmd->passive_dwell_time = pno->passive_max_time;
9248
9249         if (pno->do_passive_scan)
9250                 cmd->flags |= WMI_NLO_CONFIG_SCAN_PASSIVE;
9251
9252         cmd->fast_scan_period = pno->fast_scan_period;
9253         cmd->slow_scan_period = pno->slow_scan_period;
9254         cmd->fast_scan_max_cycles = pno->fast_scan_max_cycles;
9255         cmd->delay_start_time = pno->delay_start_time;
9256
9257         if (pno->enable_pno_scan_randomization) {
9258                 cmd->flags |= WMI_NLO_CONFIG_SPOOFED_MAC_IN_PROBE_REQ |
9259                                 WMI_NLO_CONFIG_RANDOM_SEQ_NO_IN_PROBE_REQ;
9260                 ether_addr_copy(cmd->mac_addr.addr, pno->mac_addr);
9261                 ether_addr_copy(cmd->mac_mask.addr, pno->mac_addr_mask);
9262                 ath11k_ce_byte_swap(cmd->mac_addr.addr, 8);
9263                 ath11k_ce_byte_swap(cmd->mac_mask.addr, 8);
9264         }
9265
9266         ptr += sizeof(*cmd);
9267
9268         /* nlo_configured_parameters(nlo_list) */
9269         cmd->no_of_ssids = pno->uc_networks_count;
9270         tlv = (struct wmi_tlv *)ptr;
9271         tlv->header = FIELD_PREP(WMI_TLV_TAG,
9272                                  WMI_TAG_ARRAY_STRUCT) |
9273                       FIELD_PREP(WMI_TLV_LEN, nlo_list_len);
9274
9275         ptr += sizeof(*tlv);
9276         nlo_list = (struct nlo_configured_parameters *)ptr;
9277         for (i = 0; i < cmd->no_of_ssids; i++) {
9278                 tlv = (struct wmi_tlv *)(&nlo_list[i].tlv_header);
9279                 tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
9280                               FIELD_PREP(WMI_TLV_LEN, sizeof(*nlo_list) - sizeof(*tlv));
9281
9282                 nlo_list[i].ssid.valid = true;
9283                 nlo_list[i].ssid.ssid.ssid_len = pno->a_networks[i].ssid.ssid_len;
9284                 memcpy(nlo_list[i].ssid.ssid.ssid,
9285                        pno->a_networks[i].ssid.ssid,
9286                        nlo_list[i].ssid.ssid.ssid_len);
9287                 ath11k_ce_byte_swap(nlo_list[i].ssid.ssid.ssid,
9288                                     roundup(nlo_list[i].ssid.ssid.ssid_len, 4));
9289
9290                 if (pno->a_networks[i].rssi_threshold &&
9291                     pno->a_networks[i].rssi_threshold > -300) {
9292                         nlo_list[i].rssi_cond.valid = true;
9293                         nlo_list[i].rssi_cond.rssi =
9294                                 pno->a_networks[i].rssi_threshold;
9295                 }
9296
9297                 nlo_list[i].bcast_nw_type.valid = true;
9298                 nlo_list[i].bcast_nw_type.bcast_nw_type =
9299                         pno->a_networks[i].bcast_nw_type;
9300         }
9301
9302         ptr += nlo_list_len;
9303         cmd->num_of_channels = pno->a_networks[0].channel_count;
9304         tlv = (struct wmi_tlv *)ptr;
9305         tlv->header =  FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_UINT32) |
9306                        FIELD_PREP(WMI_TLV_LEN, channel_list_len);
9307         ptr += sizeof(*tlv);
9308         channel_list = (u32 *)ptr;
9309         for (i = 0; i < cmd->num_of_channels; i++)
9310                 channel_list[i] = pno->a_networks[0].channels[i];
9311
9312         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "tlv start pno config vdev_id %d\n",
9313                    vdev_id);
9314
9315         return skb;
9316 }
9317
9318 static struct sk_buff *ath11k_wmi_op_gen_config_pno_stop(struct ath11k *ar,
9319                                                          u32 vdev_id)
9320 {
9321         struct wmi_wow_nlo_config_cmd *cmd;
9322         struct sk_buff *skb;
9323         size_t len;
9324
9325         len = sizeof(*cmd);
9326         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9327         if (!skb)
9328                 return ERR_PTR(-ENOMEM);
9329
9330         cmd = (struct wmi_wow_nlo_config_cmd *)skb->data;
9331         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_NLO_CONFIG_CMD) |
9332                           FIELD_PREP(WMI_TLV_LEN, len - TLV_HDR_SIZE);
9333
9334         cmd->vdev_id = vdev_id;
9335         cmd->flags = WMI_NLO_CONFIG_STOP;
9336
9337         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
9338                    "tlv stop pno config vdev_id %d\n", vdev_id);
9339         return skb;
9340 }
9341
9342 int ath11k_wmi_wow_config_pno(struct ath11k *ar, u32 vdev_id,
9343                               struct wmi_pno_scan_req  *pno_scan)
9344 {
9345         struct sk_buff *skb;
9346
9347         if (pno_scan->enable)
9348                 skb = ath11k_wmi_op_gen_config_pno_start(ar, vdev_id, pno_scan);
9349         else
9350                 skb = ath11k_wmi_op_gen_config_pno_stop(ar, vdev_id);
9351
9352         if (IS_ERR_OR_NULL(skb))
9353                 return -ENOMEM;
9354
9355         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_NETWORK_LIST_OFFLOAD_CONFIG_CMDID);
9356 }
9357
9358 static void ath11k_wmi_fill_ns_offload(struct ath11k *ar,
9359                                        struct ath11k_arp_ns_offload *offload,
9360                                        u8 **ptr,
9361                                        bool enable,
9362                                        bool ext)
9363 {
9364         struct wmi_ns_offload_tuple *ns;
9365         struct wmi_tlv *tlv;
9366         u8 *buf_ptr = *ptr;
9367         u32 ns_cnt, ns_ext_tuples;
9368         int i, max_offloads;
9369
9370         ns_cnt = offload->ipv6_count;
9371
9372         tlv  = (struct wmi_tlv *)buf_ptr;
9373
9374         if (ext) {
9375                 ns_ext_tuples = offload->ipv6_count - WMI_MAX_NS_OFFLOADS;
9376                 tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
9377                               FIELD_PREP(WMI_TLV_LEN, ns_ext_tuples * sizeof(*ns));
9378                 i = WMI_MAX_NS_OFFLOADS;
9379                 max_offloads = offload->ipv6_count;
9380         } else {
9381                 tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
9382                               FIELD_PREP(WMI_TLV_LEN, WMI_MAX_NS_OFFLOADS * sizeof(*ns));
9383                 i = 0;
9384                 max_offloads = WMI_MAX_NS_OFFLOADS;
9385         }
9386
9387         buf_ptr += sizeof(*tlv);
9388
9389         for (; i < max_offloads; i++) {
9390                 ns = (struct wmi_ns_offload_tuple *)buf_ptr;
9391                 ns->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_NS_OFFLOAD_TUPLE) |
9392                                  FIELD_PREP(WMI_TLV_LEN, sizeof(*ns) - TLV_HDR_SIZE);
9393
9394                 if (enable) {
9395                         if (i < ns_cnt)
9396                                 ns->flags |= WMI_NSOL_FLAGS_VALID;
9397
9398                         memcpy(ns->target_ipaddr[0], offload->ipv6_addr[i], 16);
9399                         memcpy(ns->solicitation_ipaddr, offload->self_ipv6_addr[i], 16);
9400                         ath11k_ce_byte_swap(ns->target_ipaddr[0], 16);
9401                         ath11k_ce_byte_swap(ns->solicitation_ipaddr, 16);
9402
9403                         if (offload->ipv6_type[i])
9404                                 ns->flags |= WMI_NSOL_FLAGS_IS_IPV6_ANYCAST;
9405
9406                         memcpy(ns->target_mac.addr, offload->mac_addr, ETH_ALEN);
9407                         ath11k_ce_byte_swap(ns->target_mac.addr, 8);
9408
9409                         if (ns->target_mac.word0 != 0 ||
9410                             ns->target_mac.word1 != 0) {
9411                                 ns->flags |= WMI_NSOL_FLAGS_MAC_VALID;
9412                         }
9413
9414                         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
9415                                    "index %d ns_solicited %pI6 target %pI6",
9416                                    i, ns->solicitation_ipaddr,
9417                                    ns->target_ipaddr[0]);
9418                 }
9419
9420                 buf_ptr += sizeof(*ns);
9421         }
9422
9423         *ptr = buf_ptr;
9424 }
9425
9426 static void ath11k_wmi_fill_arp_offload(struct ath11k *ar,
9427                                         struct ath11k_arp_ns_offload *offload,
9428                                         u8 **ptr,
9429                                         bool enable)
9430 {
9431         struct wmi_arp_offload_tuple *arp;
9432         struct wmi_tlv *tlv;
9433         u8 *buf_ptr = *ptr;
9434         int i;
9435
9436         /* fill arp tuple */
9437         tlv = (struct wmi_tlv *)buf_ptr;
9438         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_STRUCT) |
9439                       FIELD_PREP(WMI_TLV_LEN, WMI_MAX_ARP_OFFLOADS * sizeof(*arp));
9440         buf_ptr += sizeof(*tlv);
9441
9442         for (i = 0; i < WMI_MAX_ARP_OFFLOADS; i++) {
9443                 arp = (struct wmi_arp_offload_tuple *)buf_ptr;
9444                 arp->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARP_OFFLOAD_TUPLE) |
9445                                   FIELD_PREP(WMI_TLV_LEN, sizeof(*arp) - TLV_HDR_SIZE);
9446
9447                 if (enable && i < offload->ipv4_count) {
9448                         /* Copy the target ip addr and flags */
9449                         arp->flags = WMI_ARPOL_FLAGS_VALID;
9450                         memcpy(arp->target_ipaddr, offload->ipv4_addr[i], 4);
9451                         ath11k_ce_byte_swap(arp->target_ipaddr, 4);
9452
9453                         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "arp offload address %pI4",
9454                                    arp->target_ipaddr);
9455                 }
9456
9457                 buf_ptr += sizeof(*arp);
9458         }
9459
9460         *ptr = buf_ptr;
9461 }
9462
9463 int ath11k_wmi_arp_ns_offload(struct ath11k *ar,
9464                               struct ath11k_vif *arvif, bool enable)
9465 {
9466         struct ath11k_arp_ns_offload *offload;
9467         struct wmi_set_arp_ns_offload_cmd *cmd;
9468         struct wmi_tlv *tlv;
9469         struct sk_buff *skb;
9470         u8 *buf_ptr;
9471         size_t len;
9472         u8 ns_cnt, ns_ext_tuples = 0;
9473
9474         offload = &arvif->arp_ns_offload;
9475         ns_cnt = offload->ipv6_count;
9476
9477         len = sizeof(*cmd) +
9478               sizeof(*tlv) +
9479               WMI_MAX_NS_OFFLOADS * sizeof(struct wmi_ns_offload_tuple) +
9480               sizeof(*tlv) +
9481               WMI_MAX_ARP_OFFLOADS * sizeof(struct wmi_arp_offload_tuple);
9482
9483         if (ns_cnt > WMI_MAX_NS_OFFLOADS) {
9484                 ns_ext_tuples = ns_cnt - WMI_MAX_NS_OFFLOADS;
9485                 len += sizeof(*tlv) +
9486                        ns_ext_tuples * sizeof(struct wmi_ns_offload_tuple);
9487         }
9488
9489         skb = ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9490         if (!skb)
9491                 return -ENOMEM;
9492
9493         buf_ptr = skb->data;
9494         cmd = (struct wmi_set_arp_ns_offload_cmd *)buf_ptr;
9495         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
9496                                      WMI_TAG_SET_ARP_NS_OFFLOAD_CMD) |
9497                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9498
9499         cmd->flags = 0;
9500         cmd->vdev_id = arvif->vdev_id;
9501         cmd->num_ns_ext_tuples = ns_ext_tuples;
9502
9503         buf_ptr += sizeof(*cmd);
9504
9505         ath11k_wmi_fill_ns_offload(ar, offload, &buf_ptr, enable, 0);
9506         ath11k_wmi_fill_arp_offload(ar, offload, &buf_ptr, enable);
9507
9508         if (ns_ext_tuples)
9509                 ath11k_wmi_fill_ns_offload(ar, offload, &buf_ptr, enable, 1);
9510
9511         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_SET_ARP_NS_OFFLOAD_CMDID);
9512 }
9513
9514 int ath11k_wmi_gtk_rekey_offload(struct ath11k *ar,
9515                                  struct ath11k_vif *arvif, bool enable)
9516 {
9517         struct wmi_gtk_rekey_offload_cmd *cmd;
9518         struct ath11k_rekey_data *rekey_data = &arvif->rekey_data;
9519         int len;
9520         struct sk_buff *skb;
9521         __le64 replay_ctr;
9522
9523         len = sizeof(*cmd);
9524         skb =  ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9525         if (!skb)
9526                 return -ENOMEM;
9527
9528         cmd = (struct wmi_gtk_rekey_offload_cmd *)skb->data;
9529         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_GTK_OFFLOAD_CMD) |
9530                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9531
9532         cmd->vdev_id = arvif->vdev_id;
9533
9534         if (enable) {
9535                 cmd->flags = GTK_OFFLOAD_ENABLE_OPCODE;
9536
9537                 /* the length in rekey_data and cmd is equal */
9538                 memcpy(cmd->kck, rekey_data->kck, sizeof(cmd->kck));
9539                 ath11k_ce_byte_swap(cmd->kck, GTK_OFFLOAD_KEK_BYTES);
9540                 memcpy(cmd->kek, rekey_data->kek, sizeof(cmd->kek));
9541                 ath11k_ce_byte_swap(cmd->kek, GTK_OFFLOAD_KEK_BYTES);
9542
9543                 replay_ctr = cpu_to_le64(rekey_data->replay_ctr);
9544                 memcpy(cmd->replay_ctr, &replay_ctr,
9545                        sizeof(replay_ctr));
9546                 ath11k_ce_byte_swap(cmd->replay_ctr, GTK_REPLAY_COUNTER_BYTES);
9547         } else {
9548                 cmd->flags = GTK_OFFLOAD_DISABLE_OPCODE;
9549         }
9550
9551         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "offload gtk rekey vdev: %d %d\n",
9552                    arvif->vdev_id, enable);
9553         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_GTK_OFFLOAD_CMDID);
9554 }
9555
9556 int ath11k_wmi_gtk_rekey_getinfo(struct ath11k *ar,
9557                                  struct ath11k_vif *arvif)
9558 {
9559         struct wmi_gtk_rekey_offload_cmd *cmd;
9560         int len;
9561         struct sk_buff *skb;
9562
9563         len = sizeof(*cmd);
9564         skb =  ath11k_wmi_alloc_skb(ar->wmi->wmi_ab, len);
9565         if (!skb)
9566                 return -ENOMEM;
9567
9568         cmd = (struct wmi_gtk_rekey_offload_cmd *)skb->data;
9569         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_GTK_OFFLOAD_CMD) |
9570                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9571
9572         cmd->vdev_id = arvif->vdev_id;
9573         cmd->flags = GTK_OFFLOAD_REQUEST_STATUS_OPCODE;
9574
9575         ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "get gtk rekey vdev_id: %d\n",
9576                    arvif->vdev_id);
9577         return ath11k_wmi_cmd_send(ar->wmi, skb, WMI_GTK_OFFLOAD_CMDID);
9578 }
9579
9580 int ath11k_wmi_pdev_set_bios_sar_table_param(struct ath11k *ar, const u8 *sar_val)
9581 {       struct ath11k_pdev_wmi *wmi = ar->wmi;
9582         struct wmi_pdev_set_sar_table_cmd *cmd;
9583         struct wmi_tlv *tlv;
9584         struct sk_buff *skb;
9585         u8 *buf_ptr;
9586         u32 len, sar_len_aligned, rsvd_len_aligned;
9587
9588         sar_len_aligned = roundup(BIOS_SAR_TABLE_LEN, sizeof(u32));
9589         rsvd_len_aligned = roundup(BIOS_SAR_RSVD1_LEN, sizeof(u32));
9590         len = sizeof(*cmd) +
9591               TLV_HDR_SIZE + sar_len_aligned +
9592               TLV_HDR_SIZE + rsvd_len_aligned;
9593
9594         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
9595         if (!skb)
9596                 return -ENOMEM;
9597
9598         cmd = (struct wmi_pdev_set_sar_table_cmd *)skb->data;
9599         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_SET_BIOS_SAR_TABLE_CMD) |
9600                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9601         cmd->pdev_id = ar->pdev->pdev_id;
9602         cmd->sar_len = BIOS_SAR_TABLE_LEN;
9603         cmd->rsvd_len = BIOS_SAR_RSVD1_LEN;
9604
9605         buf_ptr = skb->data + sizeof(*cmd);
9606         tlv = (struct wmi_tlv *)buf_ptr;
9607         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
9608                       FIELD_PREP(WMI_TLV_LEN, sar_len_aligned);
9609         buf_ptr += TLV_HDR_SIZE;
9610         memcpy(buf_ptr, sar_val, BIOS_SAR_TABLE_LEN);
9611
9612         buf_ptr += sar_len_aligned;
9613         tlv = (struct wmi_tlv *)buf_ptr;
9614         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
9615                       FIELD_PREP(WMI_TLV_LEN, rsvd_len_aligned);
9616
9617         return ath11k_wmi_cmd_send(wmi, skb, WMI_PDEV_SET_BIOS_SAR_TABLE_CMDID);
9618 }
9619
9620 int ath11k_wmi_pdev_set_bios_geo_table_param(struct ath11k *ar)
9621 {
9622         struct ath11k_pdev_wmi *wmi = ar->wmi;
9623         struct wmi_pdev_set_geo_table_cmd *cmd;
9624         struct wmi_tlv *tlv;
9625         struct sk_buff *skb;
9626         u8 *buf_ptr;
9627         u32 len, rsvd_len_aligned;
9628
9629         rsvd_len_aligned = roundup(BIOS_SAR_RSVD2_LEN, sizeof(u32));
9630         len = sizeof(*cmd) + TLV_HDR_SIZE + rsvd_len_aligned;
9631
9632         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
9633         if (!skb)
9634                 return -ENOMEM;
9635
9636         cmd = (struct wmi_pdev_set_geo_table_cmd *)skb->data;
9637         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_PDEV_SET_BIOS_GEO_TABLE_CMD) |
9638                           FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9639         cmd->pdev_id = ar->pdev->pdev_id;
9640         cmd->rsvd_len = BIOS_SAR_RSVD2_LEN;
9641
9642         buf_ptr = skb->data + sizeof(*cmd);
9643         tlv = (struct wmi_tlv *)buf_ptr;
9644         tlv->header = FIELD_PREP(WMI_TLV_TAG, WMI_TAG_ARRAY_BYTE) |
9645                       FIELD_PREP(WMI_TLV_LEN, rsvd_len_aligned);
9646
9647         return ath11k_wmi_cmd_send(wmi, skb, WMI_PDEV_SET_BIOS_GEO_TABLE_CMDID);
9648 }
9649
9650 int ath11k_wmi_sta_keepalive(struct ath11k *ar,
9651                              const struct wmi_sta_keepalive_arg *arg)
9652 {
9653         struct ath11k_pdev_wmi *wmi = ar->wmi;
9654         struct wmi_sta_keepalive_cmd *cmd;
9655         struct wmi_sta_keepalive_arp_resp *arp;
9656         struct sk_buff *skb;
9657         size_t len;
9658
9659         len = sizeof(*cmd) + sizeof(*arp);
9660         skb = ath11k_wmi_alloc_skb(wmi->wmi_ab, len);
9661         if (!skb)
9662                 return -ENOMEM;
9663
9664         cmd = (struct wmi_sta_keepalive_cmd *)skb->data;
9665         cmd->tlv_header = FIELD_PREP(WMI_TLV_TAG,
9666                                      WMI_TAG_STA_KEEPALIVE_CMD) |
9667                                      FIELD_PREP(WMI_TLV_LEN, sizeof(*cmd) - TLV_HDR_SIZE);
9668         cmd->vdev_id = arg->vdev_id;
9669         cmd->enabled = arg->enabled;
9670         cmd->interval = arg->interval;
9671         cmd->method = arg->method;
9672
9673         arp = (struct wmi_sta_keepalive_arp_resp *)(cmd + 1);
9674         arp->tlv_header = FIELD_PREP(WMI_TLV_TAG,
9675                                      WMI_TAG_STA_KEEPALIVE_ARP_RESPONSE) |
9676                          FIELD_PREP(WMI_TLV_LEN, sizeof(*arp) - TLV_HDR_SIZE);
9677
9678         if (arg->method == WMI_STA_KEEPALIVE_METHOD_UNSOLICITED_ARP_RESPONSE ||
9679             arg->method == WMI_STA_KEEPALIVE_METHOD_GRATUITOUS_ARP_REQUEST) {
9680                 arp->src_ip4_addr = arg->src_ip4_addr;
9681                 arp->dest_ip4_addr = arg->dest_ip4_addr;
9682                 ether_addr_copy(arp->dest_mac_addr.addr, arg->dest_mac_addr);
9683         }
9684
9685         ath11k_dbg(ar->ab, ATH11K_DBG_WMI,
9686                    "sta keepalive vdev %d enabled %d method %d interval %d\n",
9687                    arg->vdev_id, arg->enabled, arg->method, arg->interval);
9688
9689         return ath11k_wmi_cmd_send(wmi, skb, WMI_STA_KEEPALIVE_CMDID);
9690 }