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