xref: /linux/drivers/net/wireless/realtek/rtw89/fw.c (revision ee975351cf0c2a11cdf97eae58265c126cb32850)
1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2 /* Copyright(c) 2019-2020  Realtek Corporation
3  */
4 
5 #include "cam.h"
6 #include "chan.h"
7 #include "coex.h"
8 #include "debug.h"
9 #include "fw.h"
10 #include "mac.h"
11 #include "phy.h"
12 #include "ps.h"
13 #include "reg.h"
14 #include "util.h"
15 
16 static const u8 mss_signature[] = {0x4D, 0x53, 0x53, 0x4B, 0x50, 0x4F, 0x4F, 0x4C};
17 
18 union rtw89_fw_element_arg {
19 	size_t offset;
20 	enum rtw89_rf_path rf_path;
21 	enum rtw89_fw_type fw_type;
22 };
23 
24 struct rtw89_fw_element_handler {
25 	int (*fn)(struct rtw89_dev *rtwdev,
26 		  const struct rtw89_fw_element_hdr *elm,
27 		  const union rtw89_fw_element_arg arg);
28 	const union rtw89_fw_element_arg arg;
29 	const char *name;
30 };
31 
32 static void rtw89_fw_c2h_cmd_handle(struct rtw89_dev *rtwdev,
33 				    struct sk_buff *skb);
34 static int rtw89_h2c_tx_and_wait(struct rtw89_dev *rtwdev, struct sk_buff *skb,
35 				 struct rtw89_wait_info *wait, unsigned int cond);
36 
37 static struct sk_buff *rtw89_fw_h2c_alloc_skb(struct rtw89_dev *rtwdev, u32 len,
38 					      bool header)
39 {
40 	struct sk_buff *skb;
41 	u32 header_len = 0;
42 	u32 h2c_desc_size = rtwdev->chip->h2c_desc_size;
43 
44 	if (header)
45 		header_len = H2C_HEADER_LEN;
46 
47 	skb = dev_alloc_skb(len + header_len + h2c_desc_size);
48 	if (!skb)
49 		return NULL;
50 	skb_reserve(skb, header_len + h2c_desc_size);
51 	memset(skb->data, 0, len);
52 
53 	return skb;
54 }
55 
56 struct sk_buff *rtw89_fw_h2c_alloc_skb_with_hdr(struct rtw89_dev *rtwdev, u32 len)
57 {
58 	return rtw89_fw_h2c_alloc_skb(rtwdev, len, true);
59 }
60 
61 struct sk_buff *rtw89_fw_h2c_alloc_skb_no_hdr(struct rtw89_dev *rtwdev, u32 len)
62 {
63 	return rtw89_fw_h2c_alloc_skb(rtwdev, len, false);
64 }
65 
66 int rtw89_fw_check_rdy(struct rtw89_dev *rtwdev, enum rtw89_fwdl_check_type type)
67 {
68 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
69 	u8 val;
70 	int ret;
71 
72 	ret = read_poll_timeout_atomic(mac->fwdl_get_status, val,
73 				       val == RTW89_FWDL_WCPU_FW_INIT_RDY,
74 				       1, FWDL_WAIT_CNT, false, rtwdev, type);
75 	if (ret) {
76 		switch (val) {
77 		case RTW89_FWDL_CHECKSUM_FAIL:
78 			rtw89_err(rtwdev, "fw checksum fail\n");
79 			return -EINVAL;
80 
81 		case RTW89_FWDL_SECURITY_FAIL:
82 			rtw89_err(rtwdev, "fw security fail\n");
83 			return -EINVAL;
84 
85 		case RTW89_FWDL_CV_NOT_MATCH:
86 			rtw89_err(rtwdev, "fw cv not match\n");
87 			return -EINVAL;
88 
89 		default:
90 			rtw89_err(rtwdev, "fw unexpected status %d\n", val);
91 			return -EBUSY;
92 		}
93 	}
94 
95 	set_bit(RTW89_FLAG_FW_RDY, rtwdev->flags);
96 
97 	return 0;
98 }
99 
100 static int rtw89_fw_hdr_parser_v0(struct rtw89_dev *rtwdev, const u8 *fw, u32 len,
101 				  struct rtw89_fw_bin_info *info)
102 {
103 	const struct rtw89_fw_hdr *fw_hdr = (const struct rtw89_fw_hdr *)fw;
104 	struct rtw89_fw_hdr_section_info *section_info;
105 	const struct rtw89_fw_dynhdr_hdr *fwdynhdr;
106 	const struct rtw89_fw_hdr_section *section;
107 	const u8 *fw_end = fw + len;
108 	const u8 *bin;
109 	u32 base_hdr_len;
110 	u32 mssc_len = 0;
111 	u32 i;
112 
113 	if (!info)
114 		return -EINVAL;
115 
116 	info->section_num = le32_get_bits(fw_hdr->w6, FW_HDR_W6_SEC_NUM);
117 	base_hdr_len = struct_size(fw_hdr, sections, info->section_num);
118 	info->dynamic_hdr_en = le32_get_bits(fw_hdr->w7, FW_HDR_W7_DYN_HDR);
119 
120 	if (info->dynamic_hdr_en) {
121 		info->hdr_len = le32_get_bits(fw_hdr->w3, FW_HDR_W3_LEN);
122 		info->dynamic_hdr_len = info->hdr_len - base_hdr_len;
123 		fwdynhdr = (const struct rtw89_fw_dynhdr_hdr *)(fw + base_hdr_len);
124 		if (le32_to_cpu(fwdynhdr->hdr_len) != info->dynamic_hdr_len) {
125 			rtw89_err(rtwdev, "[ERR]invalid fw dynamic header len\n");
126 			return -EINVAL;
127 		}
128 	} else {
129 		info->hdr_len = base_hdr_len;
130 		info->dynamic_hdr_len = 0;
131 	}
132 
133 	bin = fw + info->hdr_len;
134 
135 	/* jump to section header */
136 	section_info = info->section_info;
137 	for (i = 0; i < info->section_num; i++) {
138 		section = &fw_hdr->sections[i];
139 		section_info->type =
140 			le32_get_bits(section->w1, FWSECTION_HDR_W1_SECTIONTYPE);
141 		if (section_info->type == FWDL_SECURITY_SECTION_TYPE) {
142 			section_info->mssc =
143 				le32_get_bits(section->w2, FWSECTION_HDR_W2_MSSC);
144 			mssc_len += section_info->mssc * FWDL_SECURITY_SIGLEN;
145 		} else {
146 			section_info->mssc = 0;
147 		}
148 
149 		section_info->len = le32_get_bits(section->w1, FWSECTION_HDR_W1_SEC_SIZE);
150 		if (le32_get_bits(section->w1, FWSECTION_HDR_W1_CHECKSUM))
151 			section_info->len += FWDL_SECTION_CHKSUM_LEN;
152 		section_info->redl = le32_get_bits(section->w1, FWSECTION_HDR_W1_REDL);
153 		section_info->dladdr =
154 			le32_get_bits(section->w0, FWSECTION_HDR_W0_DL_ADDR) & 0x1fffffff;
155 		section_info->addr = bin;
156 		bin += section_info->len;
157 		section_info++;
158 	}
159 
160 	if (fw_end != bin + mssc_len) {
161 		rtw89_err(rtwdev, "[ERR]fw bin size\n");
162 		return -EINVAL;
163 	}
164 
165 	return 0;
166 }
167 
168 static int __get_mssc_key_idx(struct rtw89_dev *rtwdev,
169 			      const struct rtw89_fw_mss_pool_hdr *mss_hdr,
170 			      u32 rmp_tbl_size, u32 *key_idx)
171 {
172 	struct rtw89_fw_secure *sec = &rtwdev->fw.sec;
173 	u32 sel_byte_idx;
174 	u32 mss_sel_idx;
175 	u8 sel_bit_idx;
176 	int i;
177 
178 	if (sec->mss_dev_type == RTW89_FW_MSS_DEV_TYPE_FWSEC_DEF) {
179 		if (!mss_hdr->defen)
180 			return -ENOENT;
181 
182 		mss_sel_idx = sec->mss_cust_idx * le16_to_cpu(mss_hdr->msskey_num_max) +
183 			      sec->mss_key_num;
184 	} else {
185 		if (mss_hdr->defen)
186 			mss_sel_idx = FWDL_MSS_POOL_DEFKEYSETS_SIZE << 3;
187 		else
188 			mss_sel_idx = 0;
189 		mss_sel_idx += sec->mss_dev_type * le16_to_cpu(mss_hdr->msskey_num_max) *
190 						   le16_to_cpu(mss_hdr->msscust_max) +
191 			       sec->mss_cust_idx * le16_to_cpu(mss_hdr->msskey_num_max) +
192 			       sec->mss_key_num;
193 	}
194 
195 	sel_byte_idx = mss_sel_idx >> 3;
196 	sel_bit_idx = mss_sel_idx & 0x7;
197 
198 	if (sel_byte_idx >= rmp_tbl_size)
199 		return -EFAULT;
200 
201 	if (!(mss_hdr->rmp_tbl[sel_byte_idx] & BIT(sel_bit_idx)))
202 		return -ENOENT;
203 
204 	*key_idx = hweight8(mss_hdr->rmp_tbl[sel_byte_idx] & (BIT(sel_bit_idx) - 1));
205 
206 	for (i = 0; i < sel_byte_idx; i++)
207 		*key_idx += hweight8(mss_hdr->rmp_tbl[i]);
208 
209 	return 0;
210 }
211 
212 static int __parse_formatted_mssc(struct rtw89_dev *rtwdev,
213 				  struct rtw89_fw_bin_info *info,
214 				  struct rtw89_fw_hdr_section_info *section_info,
215 				  const struct rtw89_fw_hdr_section_v1 *section,
216 				  const void *content,
217 				  u32 *mssc_len)
218 {
219 	const struct rtw89_fw_mss_pool_hdr *mss_hdr = content + section_info->len;
220 	const union rtw89_fw_section_mssc_content *section_content = content;
221 	struct rtw89_fw_secure *sec = &rtwdev->fw.sec;
222 	u32 rmp_tbl_size;
223 	u32 key_sign_len;
224 	u32 real_key_idx;
225 	u32 sb_sel_ver;
226 	int ret;
227 
228 	if (memcmp(mss_signature, mss_hdr->signature, sizeof(mss_signature)) != 0) {
229 		rtw89_err(rtwdev, "[ERR] wrong MSS signature\n");
230 		return -ENOENT;
231 	}
232 
233 	if (mss_hdr->rmpfmt == MSS_POOL_RMP_TBL_BITMASK) {
234 		rmp_tbl_size = (le16_to_cpu(mss_hdr->msskey_num_max) *
235 				le16_to_cpu(mss_hdr->msscust_max) *
236 				mss_hdr->mssdev_max) >> 3;
237 		if (mss_hdr->defen)
238 			rmp_tbl_size += FWDL_MSS_POOL_DEFKEYSETS_SIZE;
239 	} else {
240 		rtw89_err(rtwdev, "[ERR] MSS Key Pool Remap Table Format Unsupport:%X\n",
241 			  mss_hdr->rmpfmt);
242 		return -EINVAL;
243 	}
244 
245 	if (rmp_tbl_size + sizeof(*mss_hdr) != le32_to_cpu(mss_hdr->key_raw_offset)) {
246 		rtw89_err(rtwdev, "[ERR] MSS Key Pool Format Error:0x%X + 0x%X != 0x%X\n",
247 			  rmp_tbl_size, (int)sizeof(*mss_hdr),
248 			  le32_to_cpu(mss_hdr->key_raw_offset));
249 		return -EINVAL;
250 	}
251 
252 	key_sign_len = le16_to_cpu(section_content->key_sign_len.v) >> 2;
253 	if (!key_sign_len)
254 		key_sign_len = 512;
255 
256 	if (info->dsp_checksum)
257 		key_sign_len += FWDL_SECURITY_CHKSUM_LEN;
258 
259 	*mssc_len = sizeof(*mss_hdr) + rmp_tbl_size +
260 		    le16_to_cpu(mss_hdr->keypair_num) * key_sign_len;
261 
262 	if (!sec->secure_boot)
263 		goto out;
264 
265 	sb_sel_ver = le32_to_cpu(section_content->sb_sel_ver.v);
266 	if (sb_sel_ver && sb_sel_ver != sec->sb_sel_mgn)
267 		goto ignore;
268 
269 	ret = __get_mssc_key_idx(rtwdev, mss_hdr, rmp_tbl_size, &real_key_idx);
270 	if (ret)
271 		goto ignore;
272 
273 	section_info->key_addr = content + section_info->len +
274 				le32_to_cpu(mss_hdr->key_raw_offset) +
275 				key_sign_len * real_key_idx;
276 	section_info->key_len = key_sign_len;
277 	section_info->key_idx = real_key_idx;
278 
279 out:
280 	if (info->secure_section_exist) {
281 		section_info->ignore = true;
282 		return 0;
283 	}
284 
285 	info->secure_section_exist = true;
286 
287 	return 0;
288 
289 ignore:
290 	section_info->ignore = true;
291 
292 	return 0;
293 }
294 
295 static int __parse_security_section(struct rtw89_dev *rtwdev,
296 				    struct rtw89_fw_bin_info *info,
297 				    struct rtw89_fw_hdr_section_info *section_info,
298 				    const struct rtw89_fw_hdr_section_v1 *section,
299 				    const void *content,
300 				    u32 *mssc_len)
301 {
302 	int ret;
303 
304 	section_info->mssc =
305 		le32_get_bits(section->w2, FWSECTION_HDR_V1_W2_MSSC);
306 
307 	if (section_info->mssc == FORMATTED_MSSC) {
308 		ret = __parse_formatted_mssc(rtwdev, info, section_info,
309 					     section, content, mssc_len);
310 		if (ret)
311 			return -EINVAL;
312 	} else {
313 		*mssc_len = section_info->mssc * FWDL_SECURITY_SIGLEN;
314 		if (info->dsp_checksum)
315 			*mssc_len += section_info->mssc * FWDL_SECURITY_CHKSUM_LEN;
316 
317 		info->secure_section_exist = true;
318 	}
319 
320 	return 0;
321 }
322 
323 static int rtw89_fw_hdr_parser_v1(struct rtw89_dev *rtwdev, const u8 *fw, u32 len,
324 				  struct rtw89_fw_bin_info *info)
325 {
326 	const struct rtw89_fw_hdr_v1 *fw_hdr = (const struct rtw89_fw_hdr_v1 *)fw;
327 	struct rtw89_fw_hdr_section_info *section_info;
328 	const struct rtw89_fw_dynhdr_hdr *fwdynhdr;
329 	const struct rtw89_fw_hdr_section_v1 *section;
330 	const u8 *fw_end = fw + len;
331 	const u8 *bin;
332 	u32 base_hdr_len;
333 	u32 mssc_len;
334 	int ret;
335 	u32 i;
336 
337 	info->section_num = le32_get_bits(fw_hdr->w6, FW_HDR_V1_W6_SEC_NUM);
338 	info->dsp_checksum = le32_get_bits(fw_hdr->w6, FW_HDR_V1_W6_DSP_CHKSUM);
339 	base_hdr_len = struct_size(fw_hdr, sections, info->section_num);
340 	info->dynamic_hdr_en = le32_get_bits(fw_hdr->w7, FW_HDR_V1_W7_DYN_HDR);
341 
342 	if (info->dynamic_hdr_en) {
343 		info->hdr_len = le32_get_bits(fw_hdr->w5, FW_HDR_V1_W5_HDR_SIZE);
344 		info->dynamic_hdr_len = info->hdr_len - base_hdr_len;
345 		fwdynhdr = (const struct rtw89_fw_dynhdr_hdr *)(fw + base_hdr_len);
346 		if (le32_to_cpu(fwdynhdr->hdr_len) != info->dynamic_hdr_len) {
347 			rtw89_err(rtwdev, "[ERR]invalid fw dynamic header len\n");
348 			return -EINVAL;
349 		}
350 	} else {
351 		info->hdr_len = base_hdr_len;
352 		info->dynamic_hdr_len = 0;
353 	}
354 
355 	bin = fw + info->hdr_len;
356 
357 	/* jump to section header */
358 	section_info = info->section_info;
359 	for (i = 0; i < info->section_num; i++) {
360 		section = &fw_hdr->sections[i];
361 
362 		section_info->type =
363 			le32_get_bits(section->w1, FWSECTION_HDR_V1_W1_SECTIONTYPE);
364 		section_info->len =
365 			le32_get_bits(section->w1, FWSECTION_HDR_V1_W1_SEC_SIZE);
366 		if (le32_get_bits(section->w1, FWSECTION_HDR_V1_W1_CHECKSUM))
367 			section_info->len += FWDL_SECTION_CHKSUM_LEN;
368 		section_info->redl = le32_get_bits(section->w1, FWSECTION_HDR_V1_W1_REDL);
369 		section_info->dladdr =
370 			le32_get_bits(section->w0, FWSECTION_HDR_V1_W0_DL_ADDR);
371 		section_info->addr = bin;
372 
373 		if (section_info->type == FWDL_SECURITY_SECTION_TYPE) {
374 			ret = __parse_security_section(rtwdev, info, section_info,
375 						       section, bin, &mssc_len);
376 			if (ret)
377 				return ret;
378 		} else {
379 			section_info->mssc = 0;
380 			mssc_len = 0;
381 		}
382 
383 		rtw89_debug(rtwdev, RTW89_DBG_FW,
384 			    "section[%d] type=%d len=0x%-6x mssc=%d mssc_len=%d addr=%tx\n",
385 			    i, section_info->type, section_info->len,
386 			    section_info->mssc, mssc_len, bin - fw);
387 		rtw89_debug(rtwdev, RTW89_DBG_FW,
388 			    "           ignore=%d key_addr=%p (0x%tx) key_len=%d key_idx=%d\n",
389 			    section_info->ignore, section_info->key_addr,
390 			    section_info->key_addr ?
391 			    section_info->key_addr - section_info->addr : 0,
392 			    section_info->key_len, section_info->key_idx);
393 
394 		bin += section_info->len + mssc_len;
395 		section_info++;
396 	}
397 
398 	if (fw_end != bin) {
399 		rtw89_err(rtwdev, "[ERR]fw bin size\n");
400 		return -EINVAL;
401 	}
402 
403 	if (!info->secure_section_exist)
404 		rtw89_warn(rtwdev, "no firmware secure section\n");
405 
406 	return 0;
407 }
408 
409 static int rtw89_fw_hdr_parser(struct rtw89_dev *rtwdev,
410 			       const struct rtw89_fw_suit *fw_suit,
411 			       struct rtw89_fw_bin_info *info)
412 {
413 	const u8 *fw = fw_suit->data;
414 	u32 len = fw_suit->size;
415 
416 	if (!fw || !len) {
417 		rtw89_err(rtwdev, "fw type %d isn't recognized\n", fw_suit->type);
418 		return -ENOENT;
419 	}
420 
421 	switch (fw_suit->hdr_ver) {
422 	case 0:
423 		return rtw89_fw_hdr_parser_v0(rtwdev, fw, len, info);
424 	case 1:
425 		return rtw89_fw_hdr_parser_v1(rtwdev, fw, len, info);
426 	default:
427 		return -ENOENT;
428 	}
429 }
430 
431 static
432 int rtw89_mfw_recognize(struct rtw89_dev *rtwdev, enum rtw89_fw_type type,
433 			struct rtw89_fw_suit *fw_suit, bool nowarn)
434 {
435 	struct rtw89_fw_info *fw_info = &rtwdev->fw;
436 	const struct firmware *firmware = fw_info->req.firmware;
437 	const u8 *mfw = firmware->data;
438 	u32 mfw_len = firmware->size;
439 	const struct rtw89_mfw_hdr *mfw_hdr = (const struct rtw89_mfw_hdr *)mfw;
440 	const struct rtw89_mfw_info *mfw_info;
441 	int i;
442 
443 	if (mfw_hdr->sig != RTW89_MFW_SIG) {
444 		rtw89_debug(rtwdev, RTW89_DBG_FW, "use legacy firmware\n");
445 		/* legacy firmware support normal type only */
446 		if (type != RTW89_FW_NORMAL)
447 			return -EINVAL;
448 		fw_suit->data = mfw;
449 		fw_suit->size = mfw_len;
450 		return 0;
451 	}
452 
453 	for (i = 0; i < mfw_hdr->fw_nr; i++) {
454 		mfw_info = &mfw_hdr->info[i];
455 		if (mfw_info->type == type) {
456 			if (mfw_info->cv == rtwdev->hal.cv && !mfw_info->mp)
457 				goto found;
458 			if (type == RTW89_FW_LOGFMT)
459 				goto found;
460 		}
461 	}
462 
463 	if (!nowarn)
464 		rtw89_err(rtwdev, "no suitable firmware found\n");
465 	return -ENOENT;
466 
467 found:
468 	fw_suit->data = mfw + le32_to_cpu(mfw_info->shift);
469 	fw_suit->size = le32_to_cpu(mfw_info->size);
470 	return 0;
471 }
472 
473 static u32 rtw89_mfw_get_size(struct rtw89_dev *rtwdev)
474 {
475 	struct rtw89_fw_info *fw_info = &rtwdev->fw;
476 	const struct firmware *firmware = fw_info->req.firmware;
477 	const struct rtw89_mfw_hdr *mfw_hdr =
478 		(const struct rtw89_mfw_hdr *)firmware->data;
479 	const struct rtw89_mfw_info *mfw_info;
480 	u32 size;
481 
482 	if (mfw_hdr->sig != RTW89_MFW_SIG) {
483 		rtw89_warn(rtwdev, "not mfw format\n");
484 		return 0;
485 	}
486 
487 	mfw_info = &mfw_hdr->info[mfw_hdr->fw_nr - 1];
488 	size = le32_to_cpu(mfw_info->shift) + le32_to_cpu(mfw_info->size);
489 
490 	return size;
491 }
492 
493 static void rtw89_fw_update_ver_v0(struct rtw89_dev *rtwdev,
494 				   struct rtw89_fw_suit *fw_suit,
495 				   const struct rtw89_fw_hdr *hdr)
496 {
497 	fw_suit->major_ver = le32_get_bits(hdr->w1, FW_HDR_W1_MAJOR_VERSION);
498 	fw_suit->minor_ver = le32_get_bits(hdr->w1, FW_HDR_W1_MINOR_VERSION);
499 	fw_suit->sub_ver = le32_get_bits(hdr->w1, FW_HDR_W1_SUBVERSION);
500 	fw_suit->sub_idex = le32_get_bits(hdr->w1, FW_HDR_W1_SUBINDEX);
501 	fw_suit->commitid = le32_get_bits(hdr->w2, FW_HDR_W2_COMMITID);
502 	fw_suit->build_year = le32_get_bits(hdr->w5, FW_HDR_W5_YEAR);
503 	fw_suit->build_mon = le32_get_bits(hdr->w4, FW_HDR_W4_MONTH);
504 	fw_suit->build_date = le32_get_bits(hdr->w4, FW_HDR_W4_DATE);
505 	fw_suit->build_hour = le32_get_bits(hdr->w4, FW_HDR_W4_HOUR);
506 	fw_suit->build_min = le32_get_bits(hdr->w4, FW_HDR_W4_MIN);
507 	fw_suit->cmd_ver = le32_get_bits(hdr->w7, FW_HDR_W7_CMD_VERSERION);
508 }
509 
510 static void rtw89_fw_update_ver_v1(struct rtw89_dev *rtwdev,
511 				   struct rtw89_fw_suit *fw_suit,
512 				   const struct rtw89_fw_hdr_v1 *hdr)
513 {
514 	fw_suit->major_ver = le32_get_bits(hdr->w1, FW_HDR_V1_W1_MAJOR_VERSION);
515 	fw_suit->minor_ver = le32_get_bits(hdr->w1, FW_HDR_V1_W1_MINOR_VERSION);
516 	fw_suit->sub_ver = le32_get_bits(hdr->w1, FW_HDR_V1_W1_SUBVERSION);
517 	fw_suit->sub_idex = le32_get_bits(hdr->w1, FW_HDR_V1_W1_SUBINDEX);
518 	fw_suit->commitid = le32_get_bits(hdr->w2, FW_HDR_V1_W2_COMMITID);
519 	fw_suit->build_year = le32_get_bits(hdr->w5, FW_HDR_V1_W5_YEAR);
520 	fw_suit->build_mon = le32_get_bits(hdr->w4, FW_HDR_V1_W4_MONTH);
521 	fw_suit->build_date = le32_get_bits(hdr->w4, FW_HDR_V1_W4_DATE);
522 	fw_suit->build_hour = le32_get_bits(hdr->w4, FW_HDR_V1_W4_HOUR);
523 	fw_suit->build_min = le32_get_bits(hdr->w4, FW_HDR_V1_W4_MIN);
524 	fw_suit->cmd_ver = le32_get_bits(hdr->w7, FW_HDR_V1_W3_CMD_VERSERION);
525 }
526 
527 static int rtw89_fw_update_ver(struct rtw89_dev *rtwdev,
528 			       enum rtw89_fw_type type,
529 			       struct rtw89_fw_suit *fw_suit)
530 {
531 	const struct rtw89_fw_hdr *v0 = (const struct rtw89_fw_hdr *)fw_suit->data;
532 	const struct rtw89_fw_hdr_v1 *v1 = (const struct rtw89_fw_hdr_v1 *)fw_suit->data;
533 
534 	if (type == RTW89_FW_LOGFMT)
535 		return 0;
536 
537 	fw_suit->type = type;
538 	fw_suit->hdr_ver = le32_get_bits(v0->w3, FW_HDR_W3_HDR_VER);
539 
540 	switch (fw_suit->hdr_ver) {
541 	case 0:
542 		rtw89_fw_update_ver_v0(rtwdev, fw_suit, v0);
543 		break;
544 	case 1:
545 		rtw89_fw_update_ver_v1(rtwdev, fw_suit, v1);
546 		break;
547 	default:
548 		rtw89_err(rtwdev, "Unknown firmware header version %u\n",
549 			  fw_suit->hdr_ver);
550 		return -ENOENT;
551 	}
552 
553 	rtw89_info(rtwdev,
554 		   "Firmware version %u.%u.%u.%u (%08x), cmd version %u, type %u\n",
555 		   fw_suit->major_ver, fw_suit->minor_ver, fw_suit->sub_ver,
556 		   fw_suit->sub_idex, fw_suit->commitid, fw_suit->cmd_ver, type);
557 
558 	return 0;
559 }
560 
561 static
562 int __rtw89_fw_recognize(struct rtw89_dev *rtwdev, enum rtw89_fw_type type,
563 			 bool nowarn)
564 {
565 	struct rtw89_fw_suit *fw_suit = rtw89_fw_suit_get(rtwdev, type);
566 	int ret;
567 
568 	ret = rtw89_mfw_recognize(rtwdev, type, fw_suit, nowarn);
569 	if (ret)
570 		return ret;
571 
572 	return rtw89_fw_update_ver(rtwdev, type, fw_suit);
573 }
574 
575 static
576 int __rtw89_fw_recognize_from_elm(struct rtw89_dev *rtwdev,
577 				  const struct rtw89_fw_element_hdr *elm,
578 				  const union rtw89_fw_element_arg arg)
579 {
580 	enum rtw89_fw_type type = arg.fw_type;
581 	struct rtw89_hal *hal = &rtwdev->hal;
582 	struct rtw89_fw_suit *fw_suit;
583 
584 	if (hal->cv != elm->u.bbmcu.cv)
585 		return 1; /* ignore this element */
586 
587 	fw_suit = rtw89_fw_suit_get(rtwdev, type);
588 	fw_suit->data = elm->u.bbmcu.contents;
589 	fw_suit->size = le32_to_cpu(elm->size);
590 
591 	return rtw89_fw_update_ver(rtwdev, type, fw_suit);
592 }
593 
594 #define __DEF_FW_FEAT_COND(__cond, __op) \
595 static bool __fw_feat_cond_ ## __cond(u32 suit_ver_code, u32 comp_ver_code) \
596 { \
597 	return suit_ver_code __op comp_ver_code; \
598 }
599 
600 __DEF_FW_FEAT_COND(ge, >=); /* greater or equal */
601 __DEF_FW_FEAT_COND(le, <=); /* less or equal */
602 __DEF_FW_FEAT_COND(lt, <); /* less than */
603 
604 struct __fw_feat_cfg {
605 	enum rtw89_core_chip_id chip_id;
606 	enum rtw89_fw_feature feature;
607 	u32 ver_code;
608 	bool (*cond)(u32 suit_ver_code, u32 comp_ver_code);
609 };
610 
611 #define __CFG_FW_FEAT(_chip, _cond, _maj, _min, _sub, _idx, _feat) \
612 	{ \
613 		.chip_id = _chip, \
614 		.feature = RTW89_FW_FEATURE_ ## _feat, \
615 		.ver_code = RTW89_FW_VER_CODE(_maj, _min, _sub, _idx), \
616 		.cond = __fw_feat_cond_ ## _cond, \
617 	}
618 
619 static const struct __fw_feat_cfg fw_feat_tbl[] = {
620 	__CFG_FW_FEAT(RTL8851B, ge, 0, 29, 37, 1, TX_WAKE),
621 	__CFG_FW_FEAT(RTL8851B, ge, 0, 29, 37, 1, SCAN_OFFLOAD),
622 	__CFG_FW_FEAT(RTL8851B, ge, 0, 29, 41, 0, CRASH_TRIGGER),
623 	__CFG_FW_FEAT(RTL8852A, le, 0, 13, 29, 0, OLD_HT_RA_FORMAT),
624 	__CFG_FW_FEAT(RTL8852A, ge, 0, 13, 35, 0, SCAN_OFFLOAD),
625 	__CFG_FW_FEAT(RTL8852A, ge, 0, 13, 35, 0, TX_WAKE),
626 	__CFG_FW_FEAT(RTL8852A, ge, 0, 13, 36, 0, CRASH_TRIGGER),
627 	__CFG_FW_FEAT(RTL8852A, lt, 0, 13, 38, 0, NO_PACKET_DROP),
628 	__CFG_FW_FEAT(RTL8852B, ge, 0, 29, 26, 0, NO_LPS_PG),
629 	__CFG_FW_FEAT(RTL8852B, ge, 0, 29, 26, 0, TX_WAKE),
630 	__CFG_FW_FEAT(RTL8852B, ge, 0, 29, 29, 0, CRASH_TRIGGER),
631 	__CFG_FW_FEAT(RTL8852B, ge, 0, 29, 29, 0, SCAN_OFFLOAD),
632 	__CFG_FW_FEAT(RTL8852C, le, 0, 27, 33, 0, NO_DEEP_PS),
633 	__CFG_FW_FEAT(RTL8852C, ge, 0, 27, 34, 0, TX_WAKE),
634 	__CFG_FW_FEAT(RTL8852C, ge, 0, 27, 36, 0, SCAN_OFFLOAD),
635 	__CFG_FW_FEAT(RTL8852C, ge, 0, 27, 40, 0, CRASH_TRIGGER),
636 	__CFG_FW_FEAT(RTL8852C, ge, 0, 27, 56, 10, BEACON_FILTER),
637 	__CFG_FW_FEAT(RTL8922A, ge, 0, 34, 30, 0, CRASH_TRIGGER),
638 	__CFG_FW_FEAT(RTL8922A, ge, 0, 34, 11, 0, MACID_PAUSE_SLEEP),
639 	__CFG_FW_FEAT(RTL8922A, ge, 0, 34, 35, 0, SCAN_OFFLOAD),
640 };
641 
642 static void rtw89_fw_iterate_feature_cfg(struct rtw89_fw_info *fw,
643 					 const struct rtw89_chip_info *chip,
644 					 u32 ver_code)
645 {
646 	int i;
647 
648 	for (i = 0; i < ARRAY_SIZE(fw_feat_tbl); i++) {
649 		const struct __fw_feat_cfg *ent = &fw_feat_tbl[i];
650 
651 		if (chip->chip_id != ent->chip_id)
652 			continue;
653 
654 		if (ent->cond(ver_code, ent->ver_code))
655 			RTW89_SET_FW_FEATURE(ent->feature, fw);
656 	}
657 }
658 
659 static void rtw89_fw_recognize_features(struct rtw89_dev *rtwdev)
660 {
661 	const struct rtw89_chip_info *chip = rtwdev->chip;
662 	const struct rtw89_fw_suit *fw_suit;
663 	u32 suit_ver_code;
664 
665 	fw_suit = rtw89_fw_suit_get(rtwdev, RTW89_FW_NORMAL);
666 	suit_ver_code = RTW89_FW_SUIT_VER_CODE(fw_suit);
667 
668 	rtw89_fw_iterate_feature_cfg(&rtwdev->fw, chip, suit_ver_code);
669 }
670 
671 const struct firmware *
672 rtw89_early_fw_feature_recognize(struct device *device,
673 				 const struct rtw89_chip_info *chip,
674 				 struct rtw89_fw_info *early_fw,
675 				 int *used_fw_format)
676 {
677 	const struct firmware *firmware;
678 	char fw_name[64];
679 	int fw_format;
680 	u32 ver_code;
681 	int ret;
682 
683 	for (fw_format = chip->fw_format_max; fw_format >= 0; fw_format--) {
684 		rtw89_fw_get_filename(fw_name, sizeof(fw_name),
685 				      chip->fw_basename, fw_format);
686 
687 		ret = request_firmware(&firmware, fw_name, device);
688 		if (!ret) {
689 			dev_info(device, "loaded firmware %s\n", fw_name);
690 			*used_fw_format = fw_format;
691 			break;
692 		}
693 	}
694 
695 	if (ret) {
696 		dev_err(device, "failed to early request firmware: %d\n", ret);
697 		return NULL;
698 	}
699 
700 	ver_code = rtw89_compat_fw_hdr_ver_code(firmware->data);
701 
702 	if (!ver_code)
703 		goto out;
704 
705 	rtw89_fw_iterate_feature_cfg(early_fw, chip, ver_code);
706 
707 out:
708 	return firmware;
709 }
710 
711 int rtw89_fw_recognize(struct rtw89_dev *rtwdev)
712 {
713 	const struct rtw89_chip_info *chip = rtwdev->chip;
714 	int ret;
715 
716 	if (chip->try_ce_fw) {
717 		ret = __rtw89_fw_recognize(rtwdev, RTW89_FW_NORMAL_CE, true);
718 		if (!ret)
719 			goto normal_done;
720 	}
721 
722 	ret = __rtw89_fw_recognize(rtwdev, RTW89_FW_NORMAL, false);
723 	if (ret)
724 		return ret;
725 
726 normal_done:
727 	/* It still works if wowlan firmware isn't existing. */
728 	__rtw89_fw_recognize(rtwdev, RTW89_FW_WOWLAN, false);
729 
730 	/* It still works if log format file isn't existing. */
731 	__rtw89_fw_recognize(rtwdev, RTW89_FW_LOGFMT, true);
732 
733 	rtw89_fw_recognize_features(rtwdev);
734 
735 	rtw89_coex_recognize_ver(rtwdev);
736 
737 	return 0;
738 }
739 
740 static
741 int rtw89_build_phy_tbl_from_elm(struct rtw89_dev *rtwdev,
742 				 const struct rtw89_fw_element_hdr *elm,
743 				 const union rtw89_fw_element_arg arg)
744 {
745 	struct rtw89_fw_elm_info *elm_info = &rtwdev->fw.elm_info;
746 	struct rtw89_phy_table *tbl;
747 	struct rtw89_reg2_def *regs;
748 	enum rtw89_rf_path rf_path;
749 	u32 n_regs, i;
750 	u8 idx;
751 
752 	tbl = kzalloc(sizeof(*tbl), GFP_KERNEL);
753 	if (!tbl)
754 		return -ENOMEM;
755 
756 	switch (le32_to_cpu(elm->id)) {
757 	case RTW89_FW_ELEMENT_ID_BB_REG:
758 		elm_info->bb_tbl = tbl;
759 		break;
760 	case RTW89_FW_ELEMENT_ID_BB_GAIN:
761 		elm_info->bb_gain = tbl;
762 		break;
763 	case RTW89_FW_ELEMENT_ID_RADIO_A:
764 	case RTW89_FW_ELEMENT_ID_RADIO_B:
765 	case RTW89_FW_ELEMENT_ID_RADIO_C:
766 	case RTW89_FW_ELEMENT_ID_RADIO_D:
767 		rf_path = arg.rf_path;
768 		idx = elm->u.reg2.idx;
769 
770 		elm_info->rf_radio[idx] = tbl;
771 		tbl->rf_path = rf_path;
772 		tbl->config = rtw89_phy_config_rf_reg_v1;
773 		break;
774 	case RTW89_FW_ELEMENT_ID_RF_NCTL:
775 		elm_info->rf_nctl = tbl;
776 		break;
777 	default:
778 		kfree(tbl);
779 		return -ENOENT;
780 	}
781 
782 	n_regs = le32_to_cpu(elm->size) / sizeof(tbl->regs[0]);
783 	regs = kcalloc(n_regs, sizeof(tbl->regs[0]), GFP_KERNEL);
784 	if (!regs)
785 		goto out;
786 
787 	for (i = 0; i < n_regs; i++) {
788 		regs[i].addr = le32_to_cpu(elm->u.reg2.regs[i].addr);
789 		regs[i].data = le32_to_cpu(elm->u.reg2.regs[i].data);
790 	}
791 
792 	tbl->n_regs = n_regs;
793 	tbl->regs = regs;
794 
795 	return 0;
796 
797 out:
798 	kfree(tbl);
799 	return -ENOMEM;
800 }
801 
802 static
803 int rtw89_fw_recognize_txpwr_from_elm(struct rtw89_dev *rtwdev,
804 				      const struct rtw89_fw_element_hdr *elm,
805 				      const union rtw89_fw_element_arg arg)
806 {
807 	const struct __rtw89_fw_txpwr_element *txpwr_elm = &elm->u.txpwr;
808 	const unsigned long offset = arg.offset;
809 	struct rtw89_efuse *efuse = &rtwdev->efuse;
810 	struct rtw89_txpwr_conf *conf;
811 
812 	if (!rtwdev->rfe_data) {
813 		rtwdev->rfe_data = kzalloc(sizeof(*rtwdev->rfe_data), GFP_KERNEL);
814 		if (!rtwdev->rfe_data)
815 			return -ENOMEM;
816 	}
817 
818 	conf = (void *)rtwdev->rfe_data + offset;
819 
820 	/* if multiple matched, take the last eventually */
821 	if (txpwr_elm->rfe_type == efuse->rfe_type)
822 		goto setup;
823 
824 	/* without one is matched, accept default */
825 	if (txpwr_elm->rfe_type == RTW89_TXPWR_CONF_DFLT_RFE_TYPE &&
826 	    (!rtw89_txpwr_conf_valid(conf) ||
827 	     conf->rfe_type == RTW89_TXPWR_CONF_DFLT_RFE_TYPE))
828 		goto setup;
829 
830 	rtw89_debug(rtwdev, RTW89_DBG_FW, "skip txpwr element ID %u RFE %u\n",
831 		    elm->id, txpwr_elm->rfe_type);
832 	return 0;
833 
834 setup:
835 	rtw89_debug(rtwdev, RTW89_DBG_FW, "take txpwr element ID %u RFE %u\n",
836 		    elm->id, txpwr_elm->rfe_type);
837 
838 	conf->rfe_type = txpwr_elm->rfe_type;
839 	conf->ent_sz = txpwr_elm->ent_sz;
840 	conf->num_ents = le32_to_cpu(txpwr_elm->num_ents);
841 	conf->data = txpwr_elm->content;
842 	return 0;
843 }
844 
845 static
846 int rtw89_build_txpwr_trk_tbl_from_elm(struct rtw89_dev *rtwdev,
847 				       const struct rtw89_fw_element_hdr *elm,
848 				       const union rtw89_fw_element_arg arg)
849 {
850 	struct rtw89_fw_elm_info *elm_info = &rtwdev->fw.elm_info;
851 	const struct rtw89_chip_info *chip = rtwdev->chip;
852 	u32 needed_bitmap = 0;
853 	u32 offset = 0;
854 	int subband;
855 	u32 bitmap;
856 	int type;
857 
858 	if (chip->support_bands & BIT(NL80211_BAND_6GHZ))
859 		needed_bitmap |= RTW89_DEFAULT_NEEDED_FW_TXPWR_TRK_6GHZ;
860 	if (chip->support_bands & BIT(NL80211_BAND_5GHZ))
861 		needed_bitmap |= RTW89_DEFAULT_NEEDED_FW_TXPWR_TRK_5GHZ;
862 	if (chip->support_bands & BIT(NL80211_BAND_2GHZ))
863 		needed_bitmap |= RTW89_DEFAULT_NEEDED_FW_TXPWR_TRK_2GHZ;
864 
865 	bitmap = le32_to_cpu(elm->u.txpwr_trk.bitmap);
866 
867 	if ((bitmap & needed_bitmap) != needed_bitmap) {
868 		rtw89_warn(rtwdev, "needed txpwr trk bitmap %08x but %0x8x\n",
869 			   needed_bitmap, bitmap);
870 		return -ENOENT;
871 	}
872 
873 	elm_info->txpwr_trk = kzalloc(sizeof(*elm_info->txpwr_trk), GFP_KERNEL);
874 	if (!elm_info->txpwr_trk)
875 		return -ENOMEM;
876 
877 	for (type = 0; bitmap; type++, bitmap >>= 1) {
878 		if (!(bitmap & BIT(0)))
879 			continue;
880 
881 		if (type >= __RTW89_FW_TXPWR_TRK_TYPE_6GHZ_START &&
882 		    type <= __RTW89_FW_TXPWR_TRK_TYPE_6GHZ_MAX)
883 			subband = 4;
884 		else if (type >= __RTW89_FW_TXPWR_TRK_TYPE_5GHZ_START &&
885 			 type <= __RTW89_FW_TXPWR_TRK_TYPE_5GHZ_MAX)
886 			subband = 3;
887 		else if (type >= __RTW89_FW_TXPWR_TRK_TYPE_2GHZ_START &&
888 			 type <= __RTW89_FW_TXPWR_TRK_TYPE_2GHZ_MAX)
889 			subband = 1;
890 		else
891 			break;
892 
893 		elm_info->txpwr_trk->delta[type] = &elm->u.txpwr_trk.contents[offset];
894 
895 		offset += subband;
896 		if (offset * DELTA_SWINGIDX_SIZE > le32_to_cpu(elm->size))
897 			goto err;
898 	}
899 
900 	return 0;
901 
902 err:
903 	rtw89_warn(rtwdev, "unexpected txpwr trk offset %d over size %d\n",
904 		   offset, le32_to_cpu(elm->size));
905 	kfree(elm_info->txpwr_trk);
906 	elm_info->txpwr_trk = NULL;
907 
908 	return -EFAULT;
909 }
910 
911 static
912 int rtw89_build_rfk_log_fmt_from_elm(struct rtw89_dev *rtwdev,
913 				     const struct rtw89_fw_element_hdr *elm,
914 				     const union rtw89_fw_element_arg arg)
915 {
916 	struct rtw89_fw_elm_info *elm_info = &rtwdev->fw.elm_info;
917 	u8 rfk_id;
918 
919 	if (elm_info->rfk_log_fmt)
920 		goto allocated;
921 
922 	elm_info->rfk_log_fmt = kzalloc(sizeof(*elm_info->rfk_log_fmt), GFP_KERNEL);
923 	if (!elm_info->rfk_log_fmt)
924 		return 1; /* this is an optional element, so just ignore this */
925 
926 allocated:
927 	rfk_id = elm->u.rfk_log_fmt.rfk_id;
928 	if (rfk_id >= RTW89_PHY_C2H_RFK_LOG_FUNC_NUM)
929 		return 1;
930 
931 	elm_info->rfk_log_fmt->elm[rfk_id] = elm;
932 
933 	return 0;
934 }
935 
936 static const struct rtw89_fw_element_handler __fw_element_handlers[] = {
937 	[RTW89_FW_ELEMENT_ID_BBMCU0] = {__rtw89_fw_recognize_from_elm,
938 					{ .fw_type = RTW89_FW_BBMCU0 }, NULL},
939 	[RTW89_FW_ELEMENT_ID_BBMCU1] = {__rtw89_fw_recognize_from_elm,
940 					{ .fw_type = RTW89_FW_BBMCU1 }, NULL},
941 	[RTW89_FW_ELEMENT_ID_BB_REG] = {rtw89_build_phy_tbl_from_elm, {}, "BB"},
942 	[RTW89_FW_ELEMENT_ID_BB_GAIN] = {rtw89_build_phy_tbl_from_elm, {}, NULL},
943 	[RTW89_FW_ELEMENT_ID_RADIO_A] = {rtw89_build_phy_tbl_from_elm,
944 					 { .rf_path =  RF_PATH_A }, "radio A"},
945 	[RTW89_FW_ELEMENT_ID_RADIO_B] = {rtw89_build_phy_tbl_from_elm,
946 					 { .rf_path =  RF_PATH_B }, NULL},
947 	[RTW89_FW_ELEMENT_ID_RADIO_C] = {rtw89_build_phy_tbl_from_elm,
948 					 { .rf_path =  RF_PATH_C }, NULL},
949 	[RTW89_FW_ELEMENT_ID_RADIO_D] = {rtw89_build_phy_tbl_from_elm,
950 					 { .rf_path =  RF_PATH_D }, NULL},
951 	[RTW89_FW_ELEMENT_ID_RF_NCTL] = {rtw89_build_phy_tbl_from_elm, {}, "NCTL"},
952 	[RTW89_FW_ELEMENT_ID_TXPWR_BYRATE] = {
953 		rtw89_fw_recognize_txpwr_from_elm,
954 		{ .offset = offsetof(struct rtw89_rfe_data, byrate.conf) }, "TXPWR",
955 	},
956 	[RTW89_FW_ELEMENT_ID_TXPWR_LMT_2GHZ] = {
957 		rtw89_fw_recognize_txpwr_from_elm,
958 		{ .offset = offsetof(struct rtw89_rfe_data, lmt_2ghz.conf) }, NULL,
959 	},
960 	[RTW89_FW_ELEMENT_ID_TXPWR_LMT_5GHZ] = {
961 		rtw89_fw_recognize_txpwr_from_elm,
962 		{ .offset = offsetof(struct rtw89_rfe_data, lmt_5ghz.conf) }, NULL,
963 	},
964 	[RTW89_FW_ELEMENT_ID_TXPWR_LMT_6GHZ] = {
965 		rtw89_fw_recognize_txpwr_from_elm,
966 		{ .offset = offsetof(struct rtw89_rfe_data, lmt_6ghz.conf) }, NULL,
967 	},
968 	[RTW89_FW_ELEMENT_ID_TXPWR_LMT_RU_2GHZ] = {
969 		rtw89_fw_recognize_txpwr_from_elm,
970 		{ .offset = offsetof(struct rtw89_rfe_data, lmt_ru_2ghz.conf) }, NULL,
971 	},
972 	[RTW89_FW_ELEMENT_ID_TXPWR_LMT_RU_5GHZ] = {
973 		rtw89_fw_recognize_txpwr_from_elm,
974 		{ .offset = offsetof(struct rtw89_rfe_data, lmt_ru_5ghz.conf) }, NULL,
975 	},
976 	[RTW89_FW_ELEMENT_ID_TXPWR_LMT_RU_6GHZ] = {
977 		rtw89_fw_recognize_txpwr_from_elm,
978 		{ .offset = offsetof(struct rtw89_rfe_data, lmt_ru_6ghz.conf) }, NULL,
979 	},
980 	[RTW89_FW_ELEMENT_ID_TX_SHAPE_LMT] = {
981 		rtw89_fw_recognize_txpwr_from_elm,
982 		{ .offset = offsetof(struct rtw89_rfe_data, tx_shape_lmt.conf) }, NULL,
983 	},
984 	[RTW89_FW_ELEMENT_ID_TX_SHAPE_LMT_RU] = {
985 		rtw89_fw_recognize_txpwr_from_elm,
986 		{ .offset = offsetof(struct rtw89_rfe_data, tx_shape_lmt_ru.conf) }, NULL,
987 	},
988 	[RTW89_FW_ELEMENT_ID_TXPWR_TRK] = {
989 		rtw89_build_txpwr_trk_tbl_from_elm, {}, "PWR_TRK",
990 	},
991 	[RTW89_FW_ELEMENT_ID_RFKLOG_FMT] = {
992 		rtw89_build_rfk_log_fmt_from_elm, {}, NULL,
993 	},
994 };
995 
996 int rtw89_fw_recognize_elements(struct rtw89_dev *rtwdev)
997 {
998 	struct rtw89_fw_info *fw_info = &rtwdev->fw;
999 	const struct firmware *firmware = fw_info->req.firmware;
1000 	const struct rtw89_chip_info *chip = rtwdev->chip;
1001 	u32 unrecognized_elements = chip->needed_fw_elms;
1002 	const struct rtw89_fw_element_handler *handler;
1003 	const struct rtw89_fw_element_hdr *hdr;
1004 	u32 elm_size;
1005 	u32 elem_id;
1006 	u32 offset;
1007 	int ret;
1008 
1009 	BUILD_BUG_ON(sizeof(chip->needed_fw_elms) * 8 < RTW89_FW_ELEMENT_ID_NUM);
1010 
1011 	offset = rtw89_mfw_get_size(rtwdev);
1012 	offset = ALIGN(offset, RTW89_FW_ELEMENT_ALIGN);
1013 	if (offset == 0)
1014 		return -EINVAL;
1015 
1016 	while (offset + sizeof(*hdr) < firmware->size) {
1017 		hdr = (const struct rtw89_fw_element_hdr *)(firmware->data + offset);
1018 
1019 		elm_size = le32_to_cpu(hdr->size);
1020 		if (offset + elm_size >= firmware->size) {
1021 			rtw89_warn(rtwdev, "firmware element size exceeds\n");
1022 			break;
1023 		}
1024 
1025 		elem_id = le32_to_cpu(hdr->id);
1026 		if (elem_id >= ARRAY_SIZE(__fw_element_handlers))
1027 			goto next;
1028 
1029 		handler = &__fw_element_handlers[elem_id];
1030 		if (!handler->fn)
1031 			goto next;
1032 
1033 		ret = handler->fn(rtwdev, hdr, handler->arg);
1034 		if (ret == 1) /* ignore this element */
1035 			goto next;
1036 		if (ret)
1037 			return ret;
1038 
1039 		if (handler->name)
1040 			rtw89_info(rtwdev, "Firmware element %s version: %4ph\n",
1041 				   handler->name, hdr->ver);
1042 
1043 		unrecognized_elements &= ~BIT(elem_id);
1044 next:
1045 		offset += sizeof(*hdr) + elm_size;
1046 		offset = ALIGN(offset, RTW89_FW_ELEMENT_ALIGN);
1047 	}
1048 
1049 	if (unrecognized_elements) {
1050 		rtw89_err(rtwdev, "Firmware elements 0x%08x are unrecognized\n",
1051 			  unrecognized_elements);
1052 		return -ENOENT;
1053 	}
1054 
1055 	return 0;
1056 }
1057 
1058 void rtw89_h2c_pkt_set_hdr(struct rtw89_dev *rtwdev, struct sk_buff *skb,
1059 			   u8 type, u8 cat, u8 class, u8 func,
1060 			   bool rack, bool dack, u32 len)
1061 {
1062 	struct fwcmd_hdr *hdr;
1063 
1064 	hdr = (struct fwcmd_hdr *)skb_push(skb, 8);
1065 
1066 	if (!(rtwdev->fw.h2c_seq % 4))
1067 		rack = true;
1068 	hdr->hdr0 = cpu_to_le32(FIELD_PREP(H2C_HDR_DEL_TYPE, type) |
1069 				FIELD_PREP(H2C_HDR_CAT, cat) |
1070 				FIELD_PREP(H2C_HDR_CLASS, class) |
1071 				FIELD_PREP(H2C_HDR_FUNC, func) |
1072 				FIELD_PREP(H2C_HDR_H2C_SEQ, rtwdev->fw.h2c_seq));
1073 
1074 	hdr->hdr1 = cpu_to_le32(FIELD_PREP(H2C_HDR_TOTAL_LEN,
1075 					   len + H2C_HEADER_LEN) |
1076 				(rack ? H2C_HDR_REC_ACK : 0) |
1077 				(dack ? H2C_HDR_DONE_ACK : 0));
1078 
1079 	rtwdev->fw.h2c_seq++;
1080 }
1081 
1082 static void rtw89_h2c_pkt_set_hdr_fwdl(struct rtw89_dev *rtwdev,
1083 				       struct sk_buff *skb,
1084 				       u8 type, u8 cat, u8 class, u8 func,
1085 				       u32 len)
1086 {
1087 	struct fwcmd_hdr *hdr;
1088 
1089 	hdr = (struct fwcmd_hdr *)skb_push(skb, 8);
1090 
1091 	hdr->hdr0 = cpu_to_le32(FIELD_PREP(H2C_HDR_DEL_TYPE, type) |
1092 				FIELD_PREP(H2C_HDR_CAT, cat) |
1093 				FIELD_PREP(H2C_HDR_CLASS, class) |
1094 				FIELD_PREP(H2C_HDR_FUNC, func) |
1095 				FIELD_PREP(H2C_HDR_H2C_SEQ, rtwdev->fw.h2c_seq));
1096 
1097 	hdr->hdr1 = cpu_to_le32(FIELD_PREP(H2C_HDR_TOTAL_LEN,
1098 					   len + H2C_HEADER_LEN));
1099 }
1100 
1101 static u32 __rtw89_fw_download_tweak_hdr_v0(struct rtw89_dev *rtwdev,
1102 					    struct rtw89_fw_bin_info *info,
1103 					    struct rtw89_fw_hdr *fw_hdr)
1104 {
1105 	le32p_replace_bits(&fw_hdr->w7, FWDL_SECTION_PER_PKT_LEN,
1106 			   FW_HDR_W7_PART_SIZE);
1107 
1108 	return 0;
1109 }
1110 
1111 static u32 __rtw89_fw_download_tweak_hdr_v1(struct rtw89_dev *rtwdev,
1112 					    struct rtw89_fw_bin_info *info,
1113 					    struct rtw89_fw_hdr_v1 *fw_hdr)
1114 {
1115 	struct rtw89_fw_hdr_section_info *section_info;
1116 	struct rtw89_fw_hdr_section_v1 *section;
1117 	u8 dst_sec_idx = 0;
1118 	u8 sec_idx;
1119 
1120 	le32p_replace_bits(&fw_hdr->w7, FWDL_SECTION_PER_PKT_LEN,
1121 			   FW_HDR_V1_W7_PART_SIZE);
1122 
1123 	for (sec_idx = 0; sec_idx < info->section_num; sec_idx++) {
1124 		section_info = &info->section_info[sec_idx];
1125 		section = &fw_hdr->sections[sec_idx];
1126 
1127 		if (section_info->ignore)
1128 			continue;
1129 
1130 		if (dst_sec_idx != sec_idx)
1131 			fw_hdr->sections[dst_sec_idx] = *section;
1132 
1133 		dst_sec_idx++;
1134 	}
1135 
1136 	le32p_replace_bits(&fw_hdr->w6, dst_sec_idx, FW_HDR_V1_W6_SEC_NUM);
1137 
1138 	return (info->section_num - dst_sec_idx) * sizeof(*section);
1139 }
1140 
1141 static int __rtw89_fw_download_hdr(struct rtw89_dev *rtwdev,
1142 				   const struct rtw89_fw_suit *fw_suit,
1143 				   struct rtw89_fw_bin_info *info)
1144 {
1145 	u32 len = info->hdr_len - info->dynamic_hdr_len;
1146 	struct rtw89_fw_hdr_v1 *fw_hdr_v1;
1147 	const u8 *fw = fw_suit->data;
1148 	struct rtw89_fw_hdr *fw_hdr;
1149 	struct sk_buff *skb;
1150 	u32 truncated;
1151 	u32 ret = 0;
1152 
1153 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
1154 	if (!skb) {
1155 		rtw89_err(rtwdev, "failed to alloc skb for fw hdr dl\n");
1156 		return -ENOMEM;
1157 	}
1158 
1159 	skb_put_data(skb, fw, len);
1160 
1161 	switch (fw_suit->hdr_ver) {
1162 	case 0:
1163 		fw_hdr = (struct rtw89_fw_hdr *)skb->data;
1164 		truncated = __rtw89_fw_download_tweak_hdr_v0(rtwdev, info, fw_hdr);
1165 		break;
1166 	case 1:
1167 		fw_hdr_v1 = (struct rtw89_fw_hdr_v1 *)skb->data;
1168 		truncated = __rtw89_fw_download_tweak_hdr_v1(rtwdev, info, fw_hdr_v1);
1169 		break;
1170 	default:
1171 		ret = -EOPNOTSUPP;
1172 		goto fail;
1173 	}
1174 
1175 	if (truncated) {
1176 		len -= truncated;
1177 		skb_trim(skb, len);
1178 	}
1179 
1180 	rtw89_h2c_pkt_set_hdr_fwdl(rtwdev, skb, FWCMD_TYPE_H2C,
1181 				   H2C_CAT_MAC, H2C_CL_MAC_FWDL,
1182 				   H2C_FUNC_MAC_FWHDR_DL, len);
1183 
1184 	ret = rtw89_h2c_tx(rtwdev, skb, false);
1185 	if (ret) {
1186 		rtw89_err(rtwdev, "failed to send h2c\n");
1187 		ret = -1;
1188 		goto fail;
1189 	}
1190 
1191 	return 0;
1192 fail:
1193 	dev_kfree_skb_any(skb);
1194 
1195 	return ret;
1196 }
1197 
1198 static int rtw89_fw_download_hdr(struct rtw89_dev *rtwdev,
1199 				 const struct rtw89_fw_suit *fw_suit,
1200 				 struct rtw89_fw_bin_info *info)
1201 {
1202 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
1203 	int ret;
1204 
1205 	ret = __rtw89_fw_download_hdr(rtwdev, fw_suit, info);
1206 	if (ret) {
1207 		rtw89_err(rtwdev, "[ERR]FW header download\n");
1208 		return ret;
1209 	}
1210 
1211 	ret = mac->fwdl_check_path_ready(rtwdev, false);
1212 	if (ret) {
1213 		rtw89_err(rtwdev, "[ERR]FWDL path ready\n");
1214 		return ret;
1215 	}
1216 
1217 	rtw89_write32(rtwdev, R_AX_HALT_H2C_CTRL, 0);
1218 	rtw89_write32(rtwdev, R_AX_HALT_C2H_CTRL, 0);
1219 
1220 	return 0;
1221 }
1222 
1223 static int __rtw89_fw_download_main(struct rtw89_dev *rtwdev,
1224 				    struct rtw89_fw_hdr_section_info *info)
1225 {
1226 	struct sk_buff *skb;
1227 	const u8 *section = info->addr;
1228 	u32 residue_len = info->len;
1229 	bool copy_key = false;
1230 	u32 pkt_len;
1231 	int ret;
1232 
1233 	if (info->ignore)
1234 		return 0;
1235 
1236 	if (info->key_addr && info->key_len) {
1237 		if (info->len > FWDL_SECTION_PER_PKT_LEN || info->len < info->key_len)
1238 			rtw89_warn(rtwdev, "ignore to copy key data because of len %d, %d, %d\n",
1239 				   info->len, FWDL_SECTION_PER_PKT_LEN, info->key_len);
1240 		else
1241 			copy_key = true;
1242 	}
1243 
1244 	while (residue_len) {
1245 		if (residue_len >= FWDL_SECTION_PER_PKT_LEN)
1246 			pkt_len = FWDL_SECTION_PER_PKT_LEN;
1247 		else
1248 			pkt_len = residue_len;
1249 
1250 		skb = rtw89_fw_h2c_alloc_skb_no_hdr(rtwdev, pkt_len);
1251 		if (!skb) {
1252 			rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
1253 			return -ENOMEM;
1254 		}
1255 		skb_put_data(skb, section, pkt_len);
1256 
1257 		if (copy_key)
1258 			memcpy(skb->data + pkt_len - info->key_len,
1259 			       info->key_addr, info->key_len);
1260 
1261 		ret = rtw89_h2c_tx(rtwdev, skb, true);
1262 		if (ret) {
1263 			rtw89_err(rtwdev, "failed to send h2c\n");
1264 			ret = -1;
1265 			goto fail;
1266 		}
1267 
1268 		section += pkt_len;
1269 		residue_len -= pkt_len;
1270 	}
1271 
1272 	return 0;
1273 fail:
1274 	dev_kfree_skb_any(skb);
1275 
1276 	return ret;
1277 }
1278 
1279 static enum rtw89_fwdl_check_type
1280 rtw89_fw_get_fwdl_chk_type_from_suit(struct rtw89_dev *rtwdev,
1281 				     const struct rtw89_fw_suit *fw_suit)
1282 {
1283 	switch (fw_suit->type) {
1284 	case RTW89_FW_BBMCU0:
1285 		return RTW89_FWDL_CHECK_BB0_FWDL_DONE;
1286 	case RTW89_FW_BBMCU1:
1287 		return RTW89_FWDL_CHECK_BB1_FWDL_DONE;
1288 	default:
1289 		return RTW89_FWDL_CHECK_WCPU_FWDL_DONE;
1290 	}
1291 }
1292 
1293 static int rtw89_fw_download_main(struct rtw89_dev *rtwdev,
1294 				  const struct rtw89_fw_suit *fw_suit,
1295 				  struct rtw89_fw_bin_info *info)
1296 {
1297 	struct rtw89_fw_hdr_section_info *section_info = info->section_info;
1298 	const struct rtw89_chip_info *chip = rtwdev->chip;
1299 	enum rtw89_fwdl_check_type chk_type;
1300 	u8 section_num = info->section_num;
1301 	int ret;
1302 
1303 	while (section_num--) {
1304 		ret = __rtw89_fw_download_main(rtwdev, section_info);
1305 		if (ret)
1306 			return ret;
1307 		section_info++;
1308 	}
1309 
1310 	if (chip->chip_gen == RTW89_CHIP_AX)
1311 		return 0;
1312 
1313 	chk_type = rtw89_fw_get_fwdl_chk_type_from_suit(rtwdev, fw_suit);
1314 	ret = rtw89_fw_check_rdy(rtwdev, chk_type);
1315 	if (ret) {
1316 		rtw89_warn(rtwdev, "failed to download firmware type %u\n",
1317 			   fw_suit->type);
1318 		return ret;
1319 	}
1320 
1321 	return 0;
1322 }
1323 
1324 static void rtw89_fw_prog_cnt_dump(struct rtw89_dev *rtwdev)
1325 {
1326 	enum rtw89_chip_gen chip_gen = rtwdev->chip->chip_gen;
1327 	u32 addr = R_AX_DBG_PORT_SEL;
1328 	u32 val32;
1329 	u16 index;
1330 
1331 	if (chip_gen == RTW89_CHIP_BE) {
1332 		addr = R_BE_WLCPU_PORT_PC;
1333 		goto dump;
1334 	}
1335 
1336 	rtw89_write32(rtwdev, R_AX_DBG_CTRL,
1337 		      FIELD_PREP(B_AX_DBG_SEL0, FW_PROG_CNTR_DBG_SEL) |
1338 		      FIELD_PREP(B_AX_DBG_SEL1, FW_PROG_CNTR_DBG_SEL));
1339 	rtw89_write32_mask(rtwdev, R_AX_SYS_STATUS1, B_AX_SEL_0XC0_MASK, MAC_DBG_SEL);
1340 
1341 dump:
1342 	for (index = 0; index < 15; index++) {
1343 		val32 = rtw89_read32(rtwdev, addr);
1344 		rtw89_err(rtwdev, "[ERR]fw PC = 0x%x\n", val32);
1345 		fsleep(10);
1346 	}
1347 }
1348 
1349 static void rtw89_fw_dl_fail_dump(struct rtw89_dev *rtwdev)
1350 {
1351 	u32 val32;
1352 	u16 val16;
1353 
1354 	val32 = rtw89_read32(rtwdev, R_AX_WCPU_FW_CTRL);
1355 	rtw89_err(rtwdev, "[ERR]fwdl 0x1E0 = 0x%x\n", val32);
1356 
1357 	val16 = rtw89_read16(rtwdev, R_AX_BOOT_DBG + 2);
1358 	rtw89_err(rtwdev, "[ERR]fwdl 0x83F2 = 0x%x\n", val16);
1359 
1360 	rtw89_fw_prog_cnt_dump(rtwdev);
1361 }
1362 
1363 static int rtw89_fw_download_suit(struct rtw89_dev *rtwdev,
1364 				  struct rtw89_fw_suit *fw_suit)
1365 {
1366 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
1367 	struct rtw89_fw_bin_info info = {};
1368 	int ret;
1369 
1370 	ret = rtw89_fw_hdr_parser(rtwdev, fw_suit, &info);
1371 	if (ret) {
1372 		rtw89_err(rtwdev, "parse fw header fail\n");
1373 		return ret;
1374 	}
1375 
1376 	if (rtwdev->chip->chip_id == RTL8922A &&
1377 	    (fw_suit->type == RTW89_FW_NORMAL || fw_suit->type == RTW89_FW_WOWLAN))
1378 		rtw89_write32(rtwdev, R_BE_SECURE_BOOT_MALLOC_INFO, 0x20248000);
1379 
1380 	ret = mac->fwdl_check_path_ready(rtwdev, true);
1381 	if (ret) {
1382 		rtw89_err(rtwdev, "[ERR]H2C path ready\n");
1383 		return ret;
1384 	}
1385 
1386 	ret = rtw89_fw_download_hdr(rtwdev, fw_suit, &info);
1387 	if (ret)
1388 		return ret;
1389 
1390 	ret = rtw89_fw_download_main(rtwdev, fw_suit, &info);
1391 	if (ret)
1392 		return ret;
1393 
1394 	return 0;
1395 }
1396 
1397 int rtw89_fw_download(struct rtw89_dev *rtwdev, enum rtw89_fw_type type,
1398 		      bool include_bb)
1399 {
1400 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
1401 	struct rtw89_fw_info *fw_info = &rtwdev->fw;
1402 	struct rtw89_fw_suit *fw_suit = rtw89_fw_suit_get(rtwdev, type);
1403 	u8 bbmcu_nr = rtwdev->chip->bbmcu_nr;
1404 	int ret;
1405 	int i;
1406 
1407 	mac->disable_cpu(rtwdev);
1408 	ret = mac->fwdl_enable_wcpu(rtwdev, 0, true, include_bb);
1409 	if (ret)
1410 		return ret;
1411 
1412 	ret = rtw89_fw_download_suit(rtwdev, fw_suit);
1413 	if (ret)
1414 		goto fwdl_err;
1415 
1416 	for (i = 0; i < bbmcu_nr && include_bb; i++) {
1417 		fw_suit = rtw89_fw_suit_get(rtwdev, RTW89_FW_BBMCU0 + i);
1418 
1419 		ret = rtw89_fw_download_suit(rtwdev, fw_suit);
1420 		if (ret)
1421 			goto fwdl_err;
1422 	}
1423 
1424 	fw_info->h2c_seq = 0;
1425 	fw_info->rec_seq = 0;
1426 	fw_info->h2c_counter = 0;
1427 	fw_info->c2h_counter = 0;
1428 	rtwdev->mac.rpwm_seq_num = RPWM_SEQ_NUM_MAX;
1429 	rtwdev->mac.cpwm_seq_num = CPWM_SEQ_NUM_MAX;
1430 
1431 	mdelay(5);
1432 
1433 	ret = rtw89_fw_check_rdy(rtwdev, RTW89_FWDL_CHECK_FREERTOS_DONE);
1434 	if (ret) {
1435 		rtw89_warn(rtwdev, "download firmware fail\n");
1436 		return ret;
1437 	}
1438 
1439 	return ret;
1440 
1441 fwdl_err:
1442 	rtw89_fw_dl_fail_dump(rtwdev);
1443 	return ret;
1444 }
1445 
1446 int rtw89_wait_firmware_completion(struct rtw89_dev *rtwdev)
1447 {
1448 	struct rtw89_fw_info *fw = &rtwdev->fw;
1449 
1450 	wait_for_completion(&fw->req.completion);
1451 	if (!fw->req.firmware)
1452 		return -EINVAL;
1453 
1454 	return 0;
1455 }
1456 
1457 static int rtw89_load_firmware_req(struct rtw89_dev *rtwdev,
1458 				   struct rtw89_fw_req_info *req,
1459 				   const char *fw_name, bool nowarn)
1460 {
1461 	int ret;
1462 
1463 	if (req->firmware) {
1464 		rtw89_debug(rtwdev, RTW89_DBG_FW,
1465 			    "full firmware has been early requested\n");
1466 		complete_all(&req->completion);
1467 		return 0;
1468 	}
1469 
1470 	if (nowarn)
1471 		ret = firmware_request_nowarn(&req->firmware, fw_name, rtwdev->dev);
1472 	else
1473 		ret = request_firmware(&req->firmware, fw_name, rtwdev->dev);
1474 
1475 	complete_all(&req->completion);
1476 
1477 	return ret;
1478 }
1479 
1480 void rtw89_load_firmware_work(struct work_struct *work)
1481 {
1482 	struct rtw89_dev *rtwdev =
1483 		container_of(work, struct rtw89_dev, load_firmware_work);
1484 	const struct rtw89_chip_info *chip = rtwdev->chip;
1485 	char fw_name[64];
1486 
1487 	rtw89_fw_get_filename(fw_name, sizeof(fw_name),
1488 			      chip->fw_basename, rtwdev->fw.fw_format);
1489 
1490 	rtw89_load_firmware_req(rtwdev, &rtwdev->fw.req, fw_name, false);
1491 }
1492 
1493 static void rtw89_free_phy_tbl_from_elm(struct rtw89_phy_table *tbl)
1494 {
1495 	if (!tbl)
1496 		return;
1497 
1498 	kfree(tbl->regs);
1499 	kfree(tbl);
1500 }
1501 
1502 static void rtw89_unload_firmware_elements(struct rtw89_dev *rtwdev)
1503 {
1504 	struct rtw89_fw_elm_info *elm_info = &rtwdev->fw.elm_info;
1505 	int i;
1506 
1507 	rtw89_free_phy_tbl_from_elm(elm_info->bb_tbl);
1508 	rtw89_free_phy_tbl_from_elm(elm_info->bb_gain);
1509 	for (i = 0; i < ARRAY_SIZE(elm_info->rf_radio); i++)
1510 		rtw89_free_phy_tbl_from_elm(elm_info->rf_radio[i]);
1511 	rtw89_free_phy_tbl_from_elm(elm_info->rf_nctl);
1512 
1513 	kfree(elm_info->txpwr_trk);
1514 	kfree(elm_info->rfk_log_fmt);
1515 }
1516 
1517 void rtw89_unload_firmware(struct rtw89_dev *rtwdev)
1518 {
1519 	struct rtw89_fw_info *fw = &rtwdev->fw;
1520 
1521 	cancel_work_sync(&rtwdev->load_firmware_work);
1522 
1523 	if (fw->req.firmware) {
1524 		release_firmware(fw->req.firmware);
1525 
1526 		/* assign NULL back in case rtw89_free_ieee80211_hw()
1527 		 * try to release the same one again.
1528 		 */
1529 		fw->req.firmware = NULL;
1530 	}
1531 
1532 	kfree(fw->log.fmts);
1533 	rtw89_unload_firmware_elements(rtwdev);
1534 }
1535 
1536 static u32 rtw89_fw_log_get_fmt_idx(struct rtw89_dev *rtwdev, u32 fmt_id)
1537 {
1538 	struct rtw89_fw_log *fw_log = &rtwdev->fw.log;
1539 	u32 i;
1540 
1541 	if (fmt_id > fw_log->last_fmt_id)
1542 		return 0;
1543 
1544 	for (i = 0; i < fw_log->fmt_count; i++) {
1545 		if (le32_to_cpu(fw_log->fmt_ids[i]) == fmt_id)
1546 			return i;
1547 	}
1548 	return 0;
1549 }
1550 
1551 static int rtw89_fw_log_create_fmts_dict(struct rtw89_dev *rtwdev)
1552 {
1553 	struct rtw89_fw_log *log = &rtwdev->fw.log;
1554 	const struct rtw89_fw_logsuit_hdr *suit_hdr;
1555 	struct rtw89_fw_suit *suit = &log->suit;
1556 	const void *fmts_ptr, *fmts_end_ptr;
1557 	u32 fmt_count;
1558 	int i;
1559 
1560 	suit_hdr = (const struct rtw89_fw_logsuit_hdr *)suit->data;
1561 	fmt_count = le32_to_cpu(suit_hdr->count);
1562 	log->fmt_ids = suit_hdr->ids;
1563 	fmts_ptr = &suit_hdr->ids[fmt_count];
1564 	fmts_end_ptr = suit->data + suit->size;
1565 	log->fmts = kcalloc(fmt_count, sizeof(char *), GFP_KERNEL);
1566 	if (!log->fmts)
1567 		return -ENOMEM;
1568 
1569 	for (i = 0; i < fmt_count; i++) {
1570 		fmts_ptr = memchr_inv(fmts_ptr, 0, fmts_end_ptr - fmts_ptr);
1571 		if (!fmts_ptr)
1572 			break;
1573 
1574 		(*log->fmts)[i] = fmts_ptr;
1575 		log->last_fmt_id = le32_to_cpu(log->fmt_ids[i]);
1576 		log->fmt_count++;
1577 		fmts_ptr += strlen(fmts_ptr);
1578 	}
1579 
1580 	return 0;
1581 }
1582 
1583 int rtw89_fw_log_prepare(struct rtw89_dev *rtwdev)
1584 {
1585 	struct rtw89_fw_log *log = &rtwdev->fw.log;
1586 	struct rtw89_fw_suit *suit = &log->suit;
1587 
1588 	if (!suit || !suit->data) {
1589 		rtw89_debug(rtwdev, RTW89_DBG_FW, "no log format file\n");
1590 		return -EINVAL;
1591 	}
1592 	if (log->fmts)
1593 		return 0;
1594 
1595 	return rtw89_fw_log_create_fmts_dict(rtwdev);
1596 }
1597 
1598 static void rtw89_fw_log_dump_data(struct rtw89_dev *rtwdev,
1599 				   const struct rtw89_fw_c2h_log_fmt *log_fmt,
1600 				   u32 fmt_idx, u8 para_int, bool raw_data)
1601 {
1602 	const char *(*fmts)[] = rtwdev->fw.log.fmts;
1603 	char str_buf[RTW89_C2H_FW_LOG_STR_BUF_SIZE];
1604 	u32 args[RTW89_C2H_FW_LOG_MAX_PARA_NUM] = {0};
1605 	int i;
1606 
1607 	if (log_fmt->argc > RTW89_C2H_FW_LOG_MAX_PARA_NUM) {
1608 		rtw89_warn(rtwdev, "C2H log: Arg count is unexpected %d\n",
1609 			   log_fmt->argc);
1610 		return;
1611 	}
1612 
1613 	if (para_int)
1614 		for (i = 0 ; i < log_fmt->argc; i++)
1615 			args[i] = le32_to_cpu(log_fmt->u.argv[i]);
1616 
1617 	if (raw_data) {
1618 		if (para_int)
1619 			snprintf(str_buf, RTW89_C2H_FW_LOG_STR_BUF_SIZE,
1620 				 "fw_enc(%d, %d, %d) %*ph", le32_to_cpu(log_fmt->fmt_id),
1621 				 para_int, log_fmt->argc, (int)sizeof(args), args);
1622 		else
1623 			snprintf(str_buf, RTW89_C2H_FW_LOG_STR_BUF_SIZE,
1624 				 "fw_enc(%d, %d, %d, %s)", le32_to_cpu(log_fmt->fmt_id),
1625 				 para_int, log_fmt->argc, log_fmt->u.raw);
1626 	} else {
1627 		snprintf(str_buf, RTW89_C2H_FW_LOG_STR_BUF_SIZE, (*fmts)[fmt_idx],
1628 			 args[0x0], args[0x1], args[0x2], args[0x3], args[0x4],
1629 			 args[0x5], args[0x6], args[0x7], args[0x8], args[0x9],
1630 			 args[0xa], args[0xb], args[0xc], args[0xd], args[0xe],
1631 			 args[0xf]);
1632 	}
1633 
1634 	rtw89_info(rtwdev, "C2H log: %s", str_buf);
1635 }
1636 
1637 void rtw89_fw_log_dump(struct rtw89_dev *rtwdev, u8 *buf, u32 len)
1638 {
1639 	const struct rtw89_fw_c2h_log_fmt *log_fmt;
1640 	u8 para_int;
1641 	u32 fmt_idx;
1642 
1643 	if (len < RTW89_C2H_HEADER_LEN) {
1644 		rtw89_err(rtwdev, "c2h log length is wrong!\n");
1645 		return;
1646 	}
1647 
1648 	buf += RTW89_C2H_HEADER_LEN;
1649 	len -= RTW89_C2H_HEADER_LEN;
1650 	log_fmt = (const struct rtw89_fw_c2h_log_fmt *)buf;
1651 
1652 	if (len < RTW89_C2H_FW_FORMATTED_LOG_MIN_LEN)
1653 		goto plain_log;
1654 
1655 	if (log_fmt->signature != cpu_to_le16(RTW89_C2H_FW_LOG_SIGNATURE))
1656 		goto plain_log;
1657 
1658 	if (!rtwdev->fw.log.fmts)
1659 		return;
1660 
1661 	para_int = u8_get_bits(log_fmt->feature, RTW89_C2H_FW_LOG_FEATURE_PARA_INT);
1662 	fmt_idx = rtw89_fw_log_get_fmt_idx(rtwdev, le32_to_cpu(log_fmt->fmt_id));
1663 
1664 	if (!para_int && log_fmt->argc != 0 && fmt_idx != 0)
1665 		rtw89_info(rtwdev, "C2H log: %s%s",
1666 			   (*rtwdev->fw.log.fmts)[fmt_idx], log_fmt->u.raw);
1667 	else if (fmt_idx != 0 && para_int)
1668 		rtw89_fw_log_dump_data(rtwdev, log_fmt, fmt_idx, para_int, false);
1669 	else
1670 		rtw89_fw_log_dump_data(rtwdev, log_fmt, fmt_idx, para_int, true);
1671 	return;
1672 
1673 plain_log:
1674 	rtw89_info(rtwdev, "C2H log: %.*s", len, buf);
1675 
1676 }
1677 
1678 #define H2C_CAM_LEN 60
1679 int rtw89_fw_h2c_cam(struct rtw89_dev *rtwdev, struct rtw89_vif *rtwvif,
1680 		     struct rtw89_sta *rtwsta, const u8 *scan_mac_addr)
1681 {
1682 	struct sk_buff *skb;
1683 	int ret;
1684 
1685 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_CAM_LEN);
1686 	if (!skb) {
1687 		rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
1688 		return -ENOMEM;
1689 	}
1690 	skb_put(skb, H2C_CAM_LEN);
1691 	rtw89_cam_fill_addr_cam_info(rtwdev, rtwvif, rtwsta, scan_mac_addr, skb->data);
1692 	rtw89_cam_fill_bssid_cam_info(rtwdev, rtwvif, rtwsta, skb->data);
1693 
1694 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
1695 			      H2C_CAT_MAC,
1696 			      H2C_CL_MAC_ADDR_CAM_UPDATE,
1697 			      H2C_FUNC_MAC_ADDR_CAM_UPD, 0, 1,
1698 			      H2C_CAM_LEN);
1699 
1700 	ret = rtw89_h2c_tx(rtwdev, skb, false);
1701 	if (ret) {
1702 		rtw89_err(rtwdev, "failed to send h2c\n");
1703 		goto fail;
1704 	}
1705 
1706 	return 0;
1707 fail:
1708 	dev_kfree_skb_any(skb);
1709 
1710 	return ret;
1711 }
1712 
1713 #define H2C_DCTL_SEC_CAM_LEN 68
1714 int rtw89_fw_h2c_dctl_sec_cam_v1(struct rtw89_dev *rtwdev,
1715 				 struct rtw89_vif *rtwvif,
1716 				 struct rtw89_sta *rtwsta)
1717 {
1718 	struct sk_buff *skb;
1719 	int ret;
1720 
1721 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_DCTL_SEC_CAM_LEN);
1722 	if (!skb) {
1723 		rtw89_err(rtwdev, "failed to alloc skb for dctl sec cam\n");
1724 		return -ENOMEM;
1725 	}
1726 	skb_put(skb, H2C_DCTL_SEC_CAM_LEN);
1727 
1728 	rtw89_cam_fill_dctl_sec_cam_info_v1(rtwdev, rtwvif, rtwsta, skb->data);
1729 
1730 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
1731 			      H2C_CAT_MAC,
1732 			      H2C_CL_MAC_FR_EXCHG,
1733 			      H2C_FUNC_MAC_DCTLINFO_UD_V1, 0, 0,
1734 			      H2C_DCTL_SEC_CAM_LEN);
1735 
1736 	ret = rtw89_h2c_tx(rtwdev, skb, false);
1737 	if (ret) {
1738 		rtw89_err(rtwdev, "failed to send h2c\n");
1739 		goto fail;
1740 	}
1741 
1742 	return 0;
1743 fail:
1744 	dev_kfree_skb_any(skb);
1745 
1746 	return ret;
1747 }
1748 EXPORT_SYMBOL(rtw89_fw_h2c_dctl_sec_cam_v1);
1749 
1750 int rtw89_fw_h2c_dctl_sec_cam_v2(struct rtw89_dev *rtwdev,
1751 				 struct rtw89_vif *rtwvif,
1752 				 struct rtw89_sta *rtwsta)
1753 {
1754 	struct rtw89_h2c_dctlinfo_ud_v2 *h2c;
1755 	u32 len = sizeof(*h2c);
1756 	struct sk_buff *skb;
1757 	int ret;
1758 
1759 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
1760 	if (!skb) {
1761 		rtw89_err(rtwdev, "failed to alloc skb for dctl sec cam\n");
1762 		return -ENOMEM;
1763 	}
1764 	skb_put(skb, len);
1765 	h2c = (struct rtw89_h2c_dctlinfo_ud_v2 *)skb->data;
1766 
1767 	rtw89_cam_fill_dctl_sec_cam_info_v2(rtwdev, rtwvif, rtwsta, h2c);
1768 
1769 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
1770 			      H2C_CAT_MAC,
1771 			      H2C_CL_MAC_FR_EXCHG,
1772 			      H2C_FUNC_MAC_DCTLINFO_UD_V2, 0, 0,
1773 			      len);
1774 
1775 	ret = rtw89_h2c_tx(rtwdev, skb, false);
1776 	if (ret) {
1777 		rtw89_err(rtwdev, "failed to send h2c\n");
1778 		goto fail;
1779 	}
1780 
1781 	return 0;
1782 fail:
1783 	dev_kfree_skb_any(skb);
1784 
1785 	return ret;
1786 }
1787 EXPORT_SYMBOL(rtw89_fw_h2c_dctl_sec_cam_v2);
1788 
1789 int rtw89_fw_h2c_default_dmac_tbl_v2(struct rtw89_dev *rtwdev,
1790 				     struct rtw89_vif *rtwvif,
1791 				     struct rtw89_sta *rtwsta)
1792 {
1793 	u8 mac_id = rtwsta ? rtwsta->mac_id : rtwvif->mac_id;
1794 	struct rtw89_h2c_dctlinfo_ud_v2 *h2c;
1795 	u32 len = sizeof(*h2c);
1796 	struct sk_buff *skb;
1797 	int ret;
1798 
1799 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
1800 	if (!skb) {
1801 		rtw89_err(rtwdev, "failed to alloc skb for dctl v2\n");
1802 		return -ENOMEM;
1803 	}
1804 	skb_put(skb, len);
1805 	h2c = (struct rtw89_h2c_dctlinfo_ud_v2 *)skb->data;
1806 
1807 	h2c->c0 = le32_encode_bits(mac_id, DCTLINFO_V2_C0_MACID) |
1808 		  le32_encode_bits(1, DCTLINFO_V2_C0_OP);
1809 
1810 	h2c->m0 = cpu_to_le32(DCTLINFO_V2_W0_ALL);
1811 	h2c->m1 = cpu_to_le32(DCTLINFO_V2_W1_ALL);
1812 	h2c->m2 = cpu_to_le32(DCTLINFO_V2_W2_ALL);
1813 	h2c->m3 = cpu_to_le32(DCTLINFO_V2_W3_ALL);
1814 	h2c->m4 = cpu_to_le32(DCTLINFO_V2_W4_ALL);
1815 	h2c->m5 = cpu_to_le32(DCTLINFO_V2_W5_ALL);
1816 	h2c->m6 = cpu_to_le32(DCTLINFO_V2_W6_ALL);
1817 	h2c->m7 = cpu_to_le32(DCTLINFO_V2_W7_ALL);
1818 	h2c->m8 = cpu_to_le32(DCTLINFO_V2_W8_ALL);
1819 	h2c->m9 = cpu_to_le32(DCTLINFO_V2_W9_ALL);
1820 	h2c->m10 = cpu_to_le32(DCTLINFO_V2_W10_ALL);
1821 	h2c->m11 = cpu_to_le32(DCTLINFO_V2_W11_ALL);
1822 	h2c->m12 = cpu_to_le32(DCTLINFO_V2_W12_ALL);
1823 
1824 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
1825 			      H2C_CAT_MAC,
1826 			      H2C_CL_MAC_FR_EXCHG,
1827 			      H2C_FUNC_MAC_DCTLINFO_UD_V2, 0, 0,
1828 			      len);
1829 
1830 	ret = rtw89_h2c_tx(rtwdev, skb, false);
1831 	if (ret) {
1832 		rtw89_err(rtwdev, "failed to send h2c\n");
1833 		goto fail;
1834 	}
1835 
1836 	return 0;
1837 fail:
1838 	dev_kfree_skb_any(skb);
1839 
1840 	return ret;
1841 }
1842 EXPORT_SYMBOL(rtw89_fw_h2c_default_dmac_tbl_v2);
1843 
1844 int rtw89_fw_h2c_ba_cam(struct rtw89_dev *rtwdev, struct rtw89_sta *rtwsta,
1845 			bool valid, struct ieee80211_ampdu_params *params)
1846 {
1847 	const struct rtw89_chip_info *chip = rtwdev->chip;
1848 	struct rtw89_vif *rtwvif = rtwsta->rtwvif;
1849 	struct rtw89_h2c_ba_cam *h2c;
1850 	u8 macid = rtwsta->mac_id;
1851 	u32 len = sizeof(*h2c);
1852 	struct sk_buff *skb;
1853 	u8 entry_idx;
1854 	int ret;
1855 
1856 	ret = valid ?
1857 	      rtw89_core_acquire_sta_ba_entry(rtwdev, rtwsta, params->tid, &entry_idx) :
1858 	      rtw89_core_release_sta_ba_entry(rtwdev, rtwsta, params->tid, &entry_idx);
1859 	if (ret) {
1860 		/* it still works even if we don't have static BA CAM, because
1861 		 * hardware can create dynamic BA CAM automatically.
1862 		 */
1863 		rtw89_debug(rtwdev, RTW89_DBG_TXRX,
1864 			    "failed to %s entry tid=%d for h2c ba cam\n",
1865 			    valid ? "alloc" : "free", params->tid);
1866 		return 0;
1867 	}
1868 
1869 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
1870 	if (!skb) {
1871 		rtw89_err(rtwdev, "failed to alloc skb for h2c ba cam\n");
1872 		return -ENOMEM;
1873 	}
1874 	skb_put(skb, len);
1875 	h2c = (struct rtw89_h2c_ba_cam *)skb->data;
1876 
1877 	h2c->w0 = le32_encode_bits(macid, RTW89_H2C_BA_CAM_W0_MACID);
1878 	if (chip->bacam_ver == RTW89_BACAM_V0_EXT)
1879 		h2c->w1 |= le32_encode_bits(entry_idx, RTW89_H2C_BA_CAM_W1_ENTRY_IDX_V1);
1880 	else
1881 		h2c->w0 |= le32_encode_bits(entry_idx, RTW89_H2C_BA_CAM_W0_ENTRY_IDX);
1882 	if (!valid)
1883 		goto end;
1884 	h2c->w0 |= le32_encode_bits(valid, RTW89_H2C_BA_CAM_W0_VALID) |
1885 		   le32_encode_bits(params->tid, RTW89_H2C_BA_CAM_W0_TID);
1886 	if (params->buf_size > 64)
1887 		h2c->w0 |= le32_encode_bits(4, RTW89_H2C_BA_CAM_W0_BMAP_SIZE);
1888 	else
1889 		h2c->w0 |= le32_encode_bits(0, RTW89_H2C_BA_CAM_W0_BMAP_SIZE);
1890 	/* If init req is set, hw will set the ssn */
1891 	h2c->w0 |= le32_encode_bits(1, RTW89_H2C_BA_CAM_W0_INIT_REQ) |
1892 		   le32_encode_bits(params->ssn, RTW89_H2C_BA_CAM_W0_SSN);
1893 
1894 	if (chip->bacam_ver == RTW89_BACAM_V0_EXT) {
1895 		h2c->w1 |= le32_encode_bits(1, RTW89_H2C_BA_CAM_W1_STD_EN) |
1896 			   le32_encode_bits(rtwvif->mac_idx, RTW89_H2C_BA_CAM_W1_BAND);
1897 	}
1898 
1899 end:
1900 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
1901 			      H2C_CAT_MAC,
1902 			      H2C_CL_BA_CAM,
1903 			      H2C_FUNC_MAC_BA_CAM, 0, 1,
1904 			      len);
1905 
1906 	ret = rtw89_h2c_tx(rtwdev, skb, false);
1907 	if (ret) {
1908 		rtw89_err(rtwdev, "failed to send h2c\n");
1909 		goto fail;
1910 	}
1911 
1912 	return 0;
1913 fail:
1914 	dev_kfree_skb_any(skb);
1915 
1916 	return ret;
1917 }
1918 EXPORT_SYMBOL(rtw89_fw_h2c_ba_cam);
1919 
1920 static int rtw89_fw_h2c_init_ba_cam_v0_ext(struct rtw89_dev *rtwdev,
1921 					   u8 entry_idx, u8 uid)
1922 {
1923 	struct rtw89_h2c_ba_cam *h2c;
1924 	u32 len = sizeof(*h2c);
1925 	struct sk_buff *skb;
1926 	int ret;
1927 
1928 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
1929 	if (!skb) {
1930 		rtw89_err(rtwdev, "failed to alloc skb for dynamic h2c ba cam\n");
1931 		return -ENOMEM;
1932 	}
1933 	skb_put(skb, len);
1934 	h2c = (struct rtw89_h2c_ba_cam *)skb->data;
1935 
1936 	h2c->w0 = le32_encode_bits(1, RTW89_H2C_BA_CAM_W0_VALID);
1937 	h2c->w1 = le32_encode_bits(entry_idx, RTW89_H2C_BA_CAM_W1_ENTRY_IDX_V1) |
1938 		  le32_encode_bits(uid, RTW89_H2C_BA_CAM_W1_UID) |
1939 		  le32_encode_bits(0, RTW89_H2C_BA_CAM_W1_BAND) |
1940 		  le32_encode_bits(0, RTW89_H2C_BA_CAM_W1_STD_EN);
1941 
1942 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
1943 			      H2C_CAT_MAC,
1944 			      H2C_CL_BA_CAM,
1945 			      H2C_FUNC_MAC_BA_CAM, 0, 1,
1946 			      len);
1947 
1948 	ret = rtw89_h2c_tx(rtwdev, skb, false);
1949 	if (ret) {
1950 		rtw89_err(rtwdev, "failed to send h2c\n");
1951 		goto fail;
1952 	}
1953 
1954 	return 0;
1955 fail:
1956 	dev_kfree_skb_any(skb);
1957 
1958 	return ret;
1959 }
1960 
1961 void rtw89_fw_h2c_init_dynamic_ba_cam_v0_ext(struct rtw89_dev *rtwdev)
1962 {
1963 	const struct rtw89_chip_info *chip = rtwdev->chip;
1964 	u8 entry_idx = chip->bacam_num;
1965 	u8 uid = 0;
1966 	int i;
1967 
1968 	for (i = 0; i < chip->bacam_dynamic_num; i++) {
1969 		rtw89_fw_h2c_init_ba_cam_v0_ext(rtwdev, entry_idx, uid);
1970 		entry_idx++;
1971 		uid++;
1972 	}
1973 }
1974 
1975 int rtw89_fw_h2c_ba_cam_v1(struct rtw89_dev *rtwdev, struct rtw89_sta *rtwsta,
1976 			   bool valid, struct ieee80211_ampdu_params *params)
1977 {
1978 	const struct rtw89_chip_info *chip = rtwdev->chip;
1979 	struct rtw89_vif *rtwvif = rtwsta->rtwvif;
1980 	struct rtw89_h2c_ba_cam_v1 *h2c;
1981 	u8 macid = rtwsta->mac_id;
1982 	u32 len = sizeof(*h2c);
1983 	struct sk_buff *skb;
1984 	u8 entry_idx;
1985 	u8 bmap_size;
1986 	int ret;
1987 
1988 	ret = valid ?
1989 	      rtw89_core_acquire_sta_ba_entry(rtwdev, rtwsta, params->tid, &entry_idx) :
1990 	      rtw89_core_release_sta_ba_entry(rtwdev, rtwsta, params->tid, &entry_idx);
1991 	if (ret) {
1992 		/* it still works even if we don't have static BA CAM, because
1993 		 * hardware can create dynamic BA CAM automatically.
1994 		 */
1995 		rtw89_debug(rtwdev, RTW89_DBG_TXRX,
1996 			    "failed to %s entry tid=%d for h2c ba cam\n",
1997 			    valid ? "alloc" : "free", params->tid);
1998 		return 0;
1999 	}
2000 
2001 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
2002 	if (!skb) {
2003 		rtw89_err(rtwdev, "failed to alloc skb for h2c ba cam\n");
2004 		return -ENOMEM;
2005 	}
2006 	skb_put(skb, len);
2007 	h2c = (struct rtw89_h2c_ba_cam_v1 *)skb->data;
2008 
2009 	if (params->buf_size > 512)
2010 		bmap_size = 10;
2011 	else if (params->buf_size > 256)
2012 		bmap_size = 8;
2013 	else if (params->buf_size > 64)
2014 		bmap_size = 4;
2015 	else
2016 		bmap_size = 0;
2017 
2018 	h2c->w0 = le32_encode_bits(valid, RTW89_H2C_BA_CAM_V1_W0_VALID) |
2019 		  le32_encode_bits(1, RTW89_H2C_BA_CAM_V1_W0_INIT_REQ) |
2020 		  le32_encode_bits(macid, RTW89_H2C_BA_CAM_V1_W0_MACID_MASK) |
2021 		  le32_encode_bits(params->tid, RTW89_H2C_BA_CAM_V1_W0_TID_MASK) |
2022 		  le32_encode_bits(bmap_size, RTW89_H2C_BA_CAM_V1_W0_BMAP_SIZE_MASK) |
2023 		  le32_encode_bits(params->ssn, RTW89_H2C_BA_CAM_V1_W0_SSN_MASK);
2024 
2025 	entry_idx += chip->bacam_dynamic_num; /* std entry right after dynamic ones */
2026 	h2c->w1 = le32_encode_bits(entry_idx, RTW89_H2C_BA_CAM_V1_W1_ENTRY_IDX_MASK) |
2027 		  le32_encode_bits(1, RTW89_H2C_BA_CAM_V1_W1_STD_ENTRY_EN) |
2028 		  le32_encode_bits(!!rtwvif->mac_idx, RTW89_H2C_BA_CAM_V1_W1_BAND_SEL);
2029 
2030 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2031 			      H2C_CAT_MAC,
2032 			      H2C_CL_BA_CAM,
2033 			      H2C_FUNC_MAC_BA_CAM_V1, 0, 1,
2034 			      len);
2035 
2036 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2037 	if (ret) {
2038 		rtw89_err(rtwdev, "failed to send h2c\n");
2039 		goto fail;
2040 	}
2041 
2042 	return 0;
2043 fail:
2044 	dev_kfree_skb_any(skb);
2045 
2046 	return ret;
2047 }
2048 EXPORT_SYMBOL(rtw89_fw_h2c_ba_cam_v1);
2049 
2050 int rtw89_fw_h2c_init_ba_cam_users(struct rtw89_dev *rtwdev, u8 users,
2051 				   u8 offset, u8 mac_idx)
2052 {
2053 	struct rtw89_h2c_ba_cam_init *h2c;
2054 	u32 len = sizeof(*h2c);
2055 	struct sk_buff *skb;
2056 	int ret;
2057 
2058 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
2059 	if (!skb) {
2060 		rtw89_err(rtwdev, "failed to alloc skb for h2c ba cam init\n");
2061 		return -ENOMEM;
2062 	}
2063 	skb_put(skb, len);
2064 	h2c = (struct rtw89_h2c_ba_cam_init *)skb->data;
2065 
2066 	h2c->w0 = le32_encode_bits(users, RTW89_H2C_BA_CAM_INIT_USERS_MASK) |
2067 		  le32_encode_bits(offset, RTW89_H2C_BA_CAM_INIT_OFFSET_MASK) |
2068 		  le32_encode_bits(mac_idx, RTW89_H2C_BA_CAM_INIT_BAND_SEL);
2069 
2070 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2071 			      H2C_CAT_MAC,
2072 			      H2C_CL_BA_CAM,
2073 			      H2C_FUNC_MAC_BA_CAM_INIT, 0, 1,
2074 			      len);
2075 
2076 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2077 	if (ret) {
2078 		rtw89_err(rtwdev, "failed to send h2c\n");
2079 		goto fail;
2080 	}
2081 
2082 	return 0;
2083 fail:
2084 	dev_kfree_skb_any(skb);
2085 
2086 	return ret;
2087 }
2088 
2089 #define H2C_LOG_CFG_LEN 12
2090 int rtw89_fw_h2c_fw_log(struct rtw89_dev *rtwdev, bool enable)
2091 {
2092 	struct sk_buff *skb;
2093 	u32 comp = 0;
2094 	int ret;
2095 
2096 	if (enable)
2097 		comp = BIT(RTW89_FW_LOG_COMP_INIT) | BIT(RTW89_FW_LOG_COMP_TASK) |
2098 		       BIT(RTW89_FW_LOG_COMP_PS) | BIT(RTW89_FW_LOG_COMP_ERROR) |
2099 		       BIT(RTW89_FW_LOG_COMP_SCAN);
2100 
2101 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_LOG_CFG_LEN);
2102 	if (!skb) {
2103 		rtw89_err(rtwdev, "failed to alloc skb for fw log cfg\n");
2104 		return -ENOMEM;
2105 	}
2106 
2107 	skb_put(skb, H2C_LOG_CFG_LEN);
2108 	SET_LOG_CFG_LEVEL(skb->data, RTW89_FW_LOG_LEVEL_LOUD);
2109 	SET_LOG_CFG_PATH(skb->data, BIT(RTW89_FW_LOG_LEVEL_C2H));
2110 	SET_LOG_CFG_COMP(skb->data, comp);
2111 	SET_LOG_CFG_COMP_EXT(skb->data, 0);
2112 
2113 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2114 			      H2C_CAT_MAC,
2115 			      H2C_CL_FW_INFO,
2116 			      H2C_FUNC_LOG_CFG, 0, 0,
2117 			      H2C_LOG_CFG_LEN);
2118 
2119 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2120 	if (ret) {
2121 		rtw89_err(rtwdev, "failed to send h2c\n");
2122 		goto fail;
2123 	}
2124 
2125 	return 0;
2126 fail:
2127 	dev_kfree_skb_any(skb);
2128 
2129 	return ret;
2130 }
2131 
2132 static int rtw89_fw_h2c_add_general_pkt(struct rtw89_dev *rtwdev,
2133 					struct rtw89_vif *rtwvif,
2134 					enum rtw89_fw_pkt_ofld_type type,
2135 					u8 *id)
2136 {
2137 	struct ieee80211_vif *vif = rtwvif_to_vif(rtwvif);
2138 	struct rtw89_pktofld_info *info;
2139 	struct sk_buff *skb;
2140 	int ret;
2141 
2142 	info = kzalloc(sizeof(*info), GFP_KERNEL);
2143 	if (!info)
2144 		return -ENOMEM;
2145 
2146 	switch (type) {
2147 	case RTW89_PKT_OFLD_TYPE_PS_POLL:
2148 		skb = ieee80211_pspoll_get(rtwdev->hw, vif);
2149 		break;
2150 	case RTW89_PKT_OFLD_TYPE_PROBE_RSP:
2151 		skb = ieee80211_proberesp_get(rtwdev->hw, vif);
2152 		break;
2153 	case RTW89_PKT_OFLD_TYPE_NULL_DATA:
2154 		skb = ieee80211_nullfunc_get(rtwdev->hw, vif, -1, false);
2155 		break;
2156 	case RTW89_PKT_OFLD_TYPE_QOS_NULL:
2157 		skb = ieee80211_nullfunc_get(rtwdev->hw, vif, -1, true);
2158 		break;
2159 	default:
2160 		goto err;
2161 	}
2162 
2163 	if (!skb)
2164 		goto err;
2165 
2166 	ret = rtw89_fw_h2c_add_pkt_offload(rtwdev, &info->id, skb);
2167 	kfree_skb(skb);
2168 
2169 	if (ret)
2170 		goto err;
2171 
2172 	list_add_tail(&info->list, &rtwvif->general_pkt_list);
2173 	*id = info->id;
2174 	return 0;
2175 
2176 err:
2177 	kfree(info);
2178 	return -ENOMEM;
2179 }
2180 
2181 void rtw89_fw_release_general_pkt_list_vif(struct rtw89_dev *rtwdev,
2182 					   struct rtw89_vif *rtwvif, bool notify_fw)
2183 {
2184 	struct list_head *pkt_list = &rtwvif->general_pkt_list;
2185 	struct rtw89_pktofld_info *info, *tmp;
2186 
2187 	list_for_each_entry_safe(info, tmp, pkt_list, list) {
2188 		if (notify_fw)
2189 			rtw89_fw_h2c_del_pkt_offload(rtwdev, info->id);
2190 		else
2191 			rtw89_core_release_bit_map(rtwdev->pkt_offload, info->id);
2192 		list_del(&info->list);
2193 		kfree(info);
2194 	}
2195 }
2196 
2197 void rtw89_fw_release_general_pkt_list(struct rtw89_dev *rtwdev, bool notify_fw)
2198 {
2199 	struct rtw89_vif *rtwvif;
2200 
2201 	rtw89_for_each_rtwvif(rtwdev, rtwvif)
2202 		rtw89_fw_release_general_pkt_list_vif(rtwdev, rtwvif, notify_fw);
2203 }
2204 
2205 #define H2C_GENERAL_PKT_LEN 6
2206 #define H2C_GENERAL_PKT_ID_UND 0xff
2207 int rtw89_fw_h2c_general_pkt(struct rtw89_dev *rtwdev,
2208 			     struct rtw89_vif *rtwvif, u8 macid)
2209 {
2210 	u8 pkt_id_ps_poll = H2C_GENERAL_PKT_ID_UND;
2211 	u8 pkt_id_null = H2C_GENERAL_PKT_ID_UND;
2212 	u8 pkt_id_qos_null = H2C_GENERAL_PKT_ID_UND;
2213 	struct sk_buff *skb;
2214 	int ret;
2215 
2216 	rtw89_fw_h2c_add_general_pkt(rtwdev, rtwvif,
2217 				     RTW89_PKT_OFLD_TYPE_PS_POLL, &pkt_id_ps_poll);
2218 	rtw89_fw_h2c_add_general_pkt(rtwdev, rtwvif,
2219 				     RTW89_PKT_OFLD_TYPE_NULL_DATA, &pkt_id_null);
2220 	rtw89_fw_h2c_add_general_pkt(rtwdev, rtwvif,
2221 				     RTW89_PKT_OFLD_TYPE_QOS_NULL, &pkt_id_qos_null);
2222 
2223 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_GENERAL_PKT_LEN);
2224 	if (!skb) {
2225 		rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
2226 		return -ENOMEM;
2227 	}
2228 	skb_put(skb, H2C_GENERAL_PKT_LEN);
2229 	SET_GENERAL_PKT_MACID(skb->data, macid);
2230 	SET_GENERAL_PKT_PROBRSP_ID(skb->data, H2C_GENERAL_PKT_ID_UND);
2231 	SET_GENERAL_PKT_PSPOLL_ID(skb->data, pkt_id_ps_poll);
2232 	SET_GENERAL_PKT_NULL_ID(skb->data, pkt_id_null);
2233 	SET_GENERAL_PKT_QOS_NULL_ID(skb->data, pkt_id_qos_null);
2234 	SET_GENERAL_PKT_CTS2SELF_ID(skb->data, H2C_GENERAL_PKT_ID_UND);
2235 
2236 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2237 			      H2C_CAT_MAC,
2238 			      H2C_CL_FW_INFO,
2239 			      H2C_FUNC_MAC_GENERAL_PKT, 0, 1,
2240 			      H2C_GENERAL_PKT_LEN);
2241 
2242 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2243 	if (ret) {
2244 		rtw89_err(rtwdev, "failed to send h2c\n");
2245 		goto fail;
2246 	}
2247 
2248 	return 0;
2249 fail:
2250 	dev_kfree_skb_any(skb);
2251 
2252 	return ret;
2253 }
2254 
2255 #define H2C_LPS_PARM_LEN 8
2256 int rtw89_fw_h2c_lps_parm(struct rtw89_dev *rtwdev,
2257 			  struct rtw89_lps_parm *lps_param)
2258 {
2259 	struct sk_buff *skb;
2260 	int ret;
2261 
2262 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_LPS_PARM_LEN);
2263 	if (!skb) {
2264 		rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
2265 		return -ENOMEM;
2266 	}
2267 	skb_put(skb, H2C_LPS_PARM_LEN);
2268 
2269 	SET_LPS_PARM_MACID(skb->data, lps_param->macid);
2270 	SET_LPS_PARM_PSMODE(skb->data, lps_param->psmode);
2271 	SET_LPS_PARM_LASTRPWM(skb->data, lps_param->lastrpwm);
2272 	SET_LPS_PARM_RLBM(skb->data, 1);
2273 	SET_LPS_PARM_SMARTPS(skb->data, 1);
2274 	SET_LPS_PARM_AWAKEINTERVAL(skb->data, 1);
2275 	SET_LPS_PARM_VOUAPSD(skb->data, 0);
2276 	SET_LPS_PARM_VIUAPSD(skb->data, 0);
2277 	SET_LPS_PARM_BEUAPSD(skb->data, 0);
2278 	SET_LPS_PARM_BKUAPSD(skb->data, 0);
2279 
2280 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2281 			      H2C_CAT_MAC,
2282 			      H2C_CL_MAC_PS,
2283 			      H2C_FUNC_MAC_LPS_PARM, 0, 1,
2284 			      H2C_LPS_PARM_LEN);
2285 
2286 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2287 	if (ret) {
2288 		rtw89_err(rtwdev, "failed to send h2c\n");
2289 		goto fail;
2290 	}
2291 
2292 	return 0;
2293 fail:
2294 	dev_kfree_skb_any(skb);
2295 
2296 	return ret;
2297 }
2298 
2299 int rtw89_fw_h2c_lps_ch_info(struct rtw89_dev *rtwdev, struct rtw89_vif *rtwvif)
2300 {
2301 	const struct rtw89_chan *chan = rtw89_chan_get(rtwdev,
2302 						       rtwvif->sub_entity_idx);
2303 	const struct rtw89_chip_info *chip = rtwdev->chip;
2304 	struct rtw89_h2c_lps_ch_info *h2c;
2305 	u32 len = sizeof(*h2c);
2306 	struct sk_buff *skb;
2307 	int ret;
2308 
2309 	if (chip->chip_gen != RTW89_CHIP_BE)
2310 		return 0;
2311 
2312 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
2313 	if (!skb) {
2314 		rtw89_err(rtwdev, "failed to alloc skb for h2c lps_ch_info\n");
2315 		return -ENOMEM;
2316 	}
2317 	skb_put(skb, len);
2318 	h2c = (struct rtw89_h2c_lps_ch_info *)skb->data;
2319 
2320 	h2c->info[0].central_ch = chan->channel;
2321 	h2c->info[0].pri_ch = chan->primary_channel;
2322 	h2c->info[0].band = chan->band_type;
2323 	h2c->info[0].bw = chan->band_width;
2324 	h2c->mlo_dbcc_mode_lps = cpu_to_le32(MLO_2_PLUS_0_1RF);
2325 
2326 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2327 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_DM,
2328 			      H2C_FUNC_FW_LPS_CH_INFO, 0, 0, len);
2329 
2330 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2331 	if (ret) {
2332 		rtw89_err(rtwdev, "failed to send h2c\n");
2333 		goto fail;
2334 	}
2335 
2336 	return 0;
2337 fail:
2338 	dev_kfree_skb_any(skb);
2339 
2340 	return ret;
2341 }
2342 
2343 #define H2C_P2P_ACT_LEN 20
2344 int rtw89_fw_h2c_p2p_act(struct rtw89_dev *rtwdev, struct ieee80211_vif *vif,
2345 			 struct ieee80211_p2p_noa_desc *desc,
2346 			 u8 act, u8 noa_id)
2347 {
2348 	struct rtw89_vif *rtwvif = (struct rtw89_vif *)vif->drv_priv;
2349 	bool p2p_type_gc = rtwvif->wifi_role == RTW89_WIFI_ROLE_P2P_CLIENT;
2350 	u8 ctwindow_oppps = vif->bss_conf.p2p_noa_attr.oppps_ctwindow;
2351 	struct sk_buff *skb;
2352 	u8 *cmd;
2353 	int ret;
2354 
2355 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_P2P_ACT_LEN);
2356 	if (!skb) {
2357 		rtw89_err(rtwdev, "failed to alloc skb for h2c p2p act\n");
2358 		return -ENOMEM;
2359 	}
2360 	skb_put(skb, H2C_P2P_ACT_LEN);
2361 	cmd = skb->data;
2362 
2363 	RTW89_SET_FWCMD_P2P_MACID(cmd, rtwvif->mac_id);
2364 	RTW89_SET_FWCMD_P2P_P2PID(cmd, 0);
2365 	RTW89_SET_FWCMD_P2P_NOAID(cmd, noa_id);
2366 	RTW89_SET_FWCMD_P2P_ACT(cmd, act);
2367 	RTW89_SET_FWCMD_P2P_TYPE(cmd, p2p_type_gc);
2368 	RTW89_SET_FWCMD_P2P_ALL_SLEP(cmd, 0);
2369 	if (desc) {
2370 		RTW89_SET_FWCMD_NOA_START_TIME(cmd, desc->start_time);
2371 		RTW89_SET_FWCMD_NOA_INTERVAL(cmd, desc->interval);
2372 		RTW89_SET_FWCMD_NOA_DURATION(cmd, desc->duration);
2373 		RTW89_SET_FWCMD_NOA_COUNT(cmd, desc->count);
2374 		RTW89_SET_FWCMD_NOA_CTWINDOW(cmd, ctwindow_oppps);
2375 	}
2376 
2377 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2378 			      H2C_CAT_MAC, H2C_CL_MAC_PS,
2379 			      H2C_FUNC_P2P_ACT, 0, 0,
2380 			      H2C_P2P_ACT_LEN);
2381 
2382 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2383 	if (ret) {
2384 		rtw89_err(rtwdev, "failed to send h2c\n");
2385 		goto fail;
2386 	}
2387 
2388 	return 0;
2389 fail:
2390 	dev_kfree_skb_any(skb);
2391 
2392 	return ret;
2393 }
2394 
2395 static void __rtw89_fw_h2c_set_tx_path(struct rtw89_dev *rtwdev,
2396 				       struct sk_buff *skb)
2397 {
2398 	const struct rtw89_chip_info *chip = rtwdev->chip;
2399 	struct rtw89_hal *hal = &rtwdev->hal;
2400 	u8 ntx_path;
2401 	u8 map_b;
2402 
2403 	if (chip->rf_path_num == 1) {
2404 		ntx_path = RF_A;
2405 		map_b = 0;
2406 	} else {
2407 		ntx_path = hal->antenna_tx ? hal->antenna_tx : RF_B;
2408 		map_b = hal->antenna_tx == RF_AB ? 1 : 0;
2409 	}
2410 
2411 	SET_CMC_TBL_NTX_PATH_EN(skb->data, ntx_path);
2412 	SET_CMC_TBL_PATH_MAP_A(skb->data, 0);
2413 	SET_CMC_TBL_PATH_MAP_B(skb->data, map_b);
2414 	SET_CMC_TBL_PATH_MAP_C(skb->data, 0);
2415 	SET_CMC_TBL_PATH_MAP_D(skb->data, 0);
2416 }
2417 
2418 #define H2C_CMC_TBL_LEN 68
2419 int rtw89_fw_h2c_default_cmac_tbl(struct rtw89_dev *rtwdev,
2420 				  struct rtw89_vif *rtwvif,
2421 				  struct rtw89_sta *rtwsta)
2422 {
2423 	const struct rtw89_chip_info *chip = rtwdev->chip;
2424 	u8 macid = rtwsta ? rtwsta->mac_id : rtwvif->mac_id;
2425 	struct sk_buff *skb;
2426 	int ret;
2427 
2428 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_CMC_TBL_LEN);
2429 	if (!skb) {
2430 		rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
2431 		return -ENOMEM;
2432 	}
2433 	skb_put(skb, H2C_CMC_TBL_LEN);
2434 	SET_CTRL_INFO_MACID(skb->data, macid);
2435 	SET_CTRL_INFO_OPERATION(skb->data, 1);
2436 	if (chip->h2c_cctl_func_id == H2C_FUNC_MAC_CCTLINFO_UD) {
2437 		SET_CMC_TBL_TXPWR_MODE(skb->data, 0);
2438 		__rtw89_fw_h2c_set_tx_path(rtwdev, skb);
2439 		SET_CMC_TBL_ANTSEL_A(skb->data, 0);
2440 		SET_CMC_TBL_ANTSEL_B(skb->data, 0);
2441 		SET_CMC_TBL_ANTSEL_C(skb->data, 0);
2442 		SET_CMC_TBL_ANTSEL_D(skb->data, 0);
2443 	}
2444 	SET_CMC_TBL_DOPPLER_CTRL(skb->data, 0);
2445 	SET_CMC_TBL_TXPWR_TOLERENCE(skb->data, 0);
2446 	if (rtwvif->net_type == RTW89_NET_TYPE_AP_MODE)
2447 		SET_CMC_TBL_DATA_DCM(skb->data, 0);
2448 
2449 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2450 			      H2C_CAT_MAC, H2C_CL_MAC_FR_EXCHG,
2451 			      chip->h2c_cctl_func_id, 0, 1,
2452 			      H2C_CMC_TBL_LEN);
2453 
2454 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2455 	if (ret) {
2456 		rtw89_err(rtwdev, "failed to send h2c\n");
2457 		goto fail;
2458 	}
2459 
2460 	return 0;
2461 fail:
2462 	dev_kfree_skb_any(skb);
2463 
2464 	return ret;
2465 }
2466 EXPORT_SYMBOL(rtw89_fw_h2c_default_cmac_tbl);
2467 
2468 int rtw89_fw_h2c_default_cmac_tbl_g7(struct rtw89_dev *rtwdev,
2469 				     struct rtw89_vif *rtwvif,
2470 				     struct rtw89_sta *rtwsta)
2471 {
2472 	u8 mac_id = rtwsta ? rtwsta->mac_id : rtwvif->mac_id;
2473 	struct rtw89_h2c_cctlinfo_ud_g7 *h2c;
2474 	u32 len = sizeof(*h2c);
2475 	struct sk_buff *skb;
2476 	int ret;
2477 
2478 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
2479 	if (!skb) {
2480 		rtw89_err(rtwdev, "failed to alloc skb for cmac g7\n");
2481 		return -ENOMEM;
2482 	}
2483 	skb_put(skb, len);
2484 	h2c = (struct rtw89_h2c_cctlinfo_ud_g7 *)skb->data;
2485 
2486 	h2c->c0 = le32_encode_bits(mac_id, CCTLINFO_G7_C0_MACID) |
2487 		  le32_encode_bits(1, CCTLINFO_G7_C0_OP);
2488 
2489 	h2c->w0 = le32_encode_bits(4, CCTLINFO_G7_W0_DATARATE);
2490 	h2c->m0 = cpu_to_le32(CCTLINFO_G7_W0_ALL);
2491 
2492 	h2c->w1 = le32_encode_bits(4, CCTLINFO_G7_W1_DATA_RTY_LOWEST_RATE) |
2493 		  le32_encode_bits(0xa, CCTLINFO_G7_W1_RTSRATE) |
2494 		  le32_encode_bits(4, CCTLINFO_G7_W1_RTS_RTY_LOWEST_RATE);
2495 	h2c->m1 = cpu_to_le32(CCTLINFO_G7_W1_ALL);
2496 
2497 	h2c->m2 = cpu_to_le32(CCTLINFO_G7_W2_ALL);
2498 
2499 	h2c->m3 = cpu_to_le32(CCTLINFO_G7_W3_ALL);
2500 
2501 	h2c->w4 = le32_encode_bits(0xFFFF, CCTLINFO_G7_W4_ACT_SUBCH_CBW);
2502 	h2c->m4 = cpu_to_le32(CCTLINFO_G7_W4_ALL);
2503 
2504 	h2c->w5 = le32_encode_bits(2, CCTLINFO_G7_W5_NOMINAL_PKT_PADDING0) |
2505 		  le32_encode_bits(2, CCTLINFO_G7_W5_NOMINAL_PKT_PADDING1) |
2506 		  le32_encode_bits(2, CCTLINFO_G7_W5_NOMINAL_PKT_PADDING2) |
2507 		  le32_encode_bits(2, CCTLINFO_G7_W5_NOMINAL_PKT_PADDING3) |
2508 		  le32_encode_bits(2, CCTLINFO_G7_W5_NOMINAL_PKT_PADDING4);
2509 	h2c->m5 = cpu_to_le32(CCTLINFO_G7_W5_ALL);
2510 
2511 	h2c->w6 = le32_encode_bits(0xb, CCTLINFO_G7_W6_RESP_REF_RATE);
2512 	h2c->m6 = cpu_to_le32(CCTLINFO_G7_W6_ALL);
2513 
2514 	h2c->w7 = le32_encode_bits(1, CCTLINFO_G7_W7_NC) |
2515 		  le32_encode_bits(1, CCTLINFO_G7_W7_NR) |
2516 		  le32_encode_bits(1, CCTLINFO_G7_W7_CB) |
2517 		  le32_encode_bits(0x1, CCTLINFO_G7_W7_CSI_PARA_EN) |
2518 		  le32_encode_bits(0xb, CCTLINFO_G7_W7_CSI_FIX_RATE);
2519 	h2c->m7 = cpu_to_le32(CCTLINFO_G7_W7_ALL);
2520 
2521 	h2c->m8 = cpu_to_le32(CCTLINFO_G7_W8_ALL);
2522 
2523 	h2c->w14 = le32_encode_bits(0, CCTLINFO_G7_W14_VO_CURR_RATE) |
2524 		   le32_encode_bits(0, CCTLINFO_G7_W14_VI_CURR_RATE) |
2525 		   le32_encode_bits(0, CCTLINFO_G7_W14_BE_CURR_RATE_L);
2526 	h2c->m14 = cpu_to_le32(CCTLINFO_G7_W14_ALL);
2527 
2528 	h2c->w15 = le32_encode_bits(0, CCTLINFO_G7_W15_BE_CURR_RATE_H) |
2529 		   le32_encode_bits(0, CCTLINFO_G7_W15_BK_CURR_RATE) |
2530 		   le32_encode_bits(0, CCTLINFO_G7_W15_MGNT_CURR_RATE);
2531 	h2c->m15 = cpu_to_le32(CCTLINFO_G7_W15_ALL);
2532 
2533 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2534 			      H2C_CAT_MAC, H2C_CL_MAC_FR_EXCHG,
2535 			      H2C_FUNC_MAC_CCTLINFO_UD_G7, 0, 1,
2536 			      len);
2537 
2538 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2539 	if (ret) {
2540 		rtw89_err(rtwdev, "failed to send h2c\n");
2541 		goto fail;
2542 	}
2543 
2544 	return 0;
2545 fail:
2546 	dev_kfree_skb_any(skb);
2547 
2548 	return ret;
2549 }
2550 EXPORT_SYMBOL(rtw89_fw_h2c_default_cmac_tbl_g7);
2551 
2552 static void __get_sta_he_pkt_padding(struct rtw89_dev *rtwdev,
2553 				     struct ieee80211_sta *sta, u8 *pads)
2554 {
2555 	bool ppe_th;
2556 	u8 ppe16, ppe8;
2557 	u8 nss = min(sta->deflink.rx_nss, rtwdev->hal.tx_nss) - 1;
2558 	u8 ppe_thres_hdr = sta->deflink.he_cap.ppe_thres[0];
2559 	u8 ru_bitmap;
2560 	u8 n, idx, sh;
2561 	u16 ppe;
2562 	int i;
2563 
2564 	ppe_th = FIELD_GET(IEEE80211_HE_PHY_CAP6_PPE_THRESHOLD_PRESENT,
2565 			   sta->deflink.he_cap.he_cap_elem.phy_cap_info[6]);
2566 	if (!ppe_th) {
2567 		u8 pad;
2568 
2569 		pad = FIELD_GET(IEEE80211_HE_PHY_CAP9_NOMINAL_PKT_PADDING_MASK,
2570 				sta->deflink.he_cap.he_cap_elem.phy_cap_info[9]);
2571 
2572 		for (i = 0; i < RTW89_PPE_BW_NUM; i++)
2573 			pads[i] = pad;
2574 
2575 		return;
2576 	}
2577 
2578 	ru_bitmap = FIELD_GET(IEEE80211_PPE_THRES_RU_INDEX_BITMASK_MASK, ppe_thres_hdr);
2579 	n = hweight8(ru_bitmap);
2580 	n = 7 + (n * IEEE80211_PPE_THRES_INFO_PPET_SIZE * 2) * nss;
2581 
2582 	for (i = 0; i < RTW89_PPE_BW_NUM; i++) {
2583 		if (!(ru_bitmap & BIT(i))) {
2584 			pads[i] = 1;
2585 			continue;
2586 		}
2587 
2588 		idx = n >> 3;
2589 		sh = n & 7;
2590 		n += IEEE80211_PPE_THRES_INFO_PPET_SIZE * 2;
2591 
2592 		ppe = le16_to_cpu(*((__le16 *)&sta->deflink.he_cap.ppe_thres[idx]));
2593 		ppe16 = (ppe >> sh) & IEEE80211_PPE_THRES_NSS_MASK;
2594 		sh += IEEE80211_PPE_THRES_INFO_PPET_SIZE;
2595 		ppe8 = (ppe >> sh) & IEEE80211_PPE_THRES_NSS_MASK;
2596 
2597 		if (ppe16 != 7 && ppe8 == 7)
2598 			pads[i] = 2;
2599 		else if (ppe8 != 7)
2600 			pads[i] = 1;
2601 		else
2602 			pads[i] = 0;
2603 	}
2604 }
2605 
2606 int rtw89_fw_h2c_assoc_cmac_tbl(struct rtw89_dev *rtwdev,
2607 				struct ieee80211_vif *vif,
2608 				struct ieee80211_sta *sta)
2609 {
2610 	const struct rtw89_chip_info *chip = rtwdev->chip;
2611 	struct rtw89_sta *rtwsta = sta_to_rtwsta_safe(sta);
2612 	struct rtw89_vif *rtwvif = (struct rtw89_vif *)vif->drv_priv;
2613 	const struct rtw89_chan *chan = rtw89_chan_get(rtwdev,
2614 						       rtwvif->sub_entity_idx);
2615 	struct sk_buff *skb;
2616 	u8 pads[RTW89_PPE_BW_NUM];
2617 	u8 mac_id = rtwsta ? rtwsta->mac_id : rtwvif->mac_id;
2618 	u16 lowest_rate;
2619 	int ret;
2620 
2621 	memset(pads, 0, sizeof(pads));
2622 	if (sta && sta->deflink.he_cap.has_he)
2623 		__get_sta_he_pkt_padding(rtwdev, sta, pads);
2624 
2625 	if (vif->p2p)
2626 		lowest_rate = RTW89_HW_RATE_OFDM6;
2627 	else if (chan->band_type == RTW89_BAND_2G)
2628 		lowest_rate = RTW89_HW_RATE_CCK1;
2629 	else
2630 		lowest_rate = RTW89_HW_RATE_OFDM6;
2631 
2632 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_CMC_TBL_LEN);
2633 	if (!skb) {
2634 		rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
2635 		return -ENOMEM;
2636 	}
2637 	skb_put(skb, H2C_CMC_TBL_LEN);
2638 	SET_CTRL_INFO_MACID(skb->data, mac_id);
2639 	SET_CTRL_INFO_OPERATION(skb->data, 1);
2640 	SET_CMC_TBL_DISRTSFB(skb->data, 1);
2641 	SET_CMC_TBL_DISDATAFB(skb->data, 1);
2642 	SET_CMC_TBL_RTS_RTY_LOWEST_RATE(skb->data, lowest_rate);
2643 	SET_CMC_TBL_RTS_TXCNT_LMT_SEL(skb->data, 0);
2644 	SET_CMC_TBL_DATA_TXCNT_LMT_SEL(skb->data, 0);
2645 	if (vif->type == NL80211_IFTYPE_STATION)
2646 		SET_CMC_TBL_ULDL(skb->data, 1);
2647 	else
2648 		SET_CMC_TBL_ULDL(skb->data, 0);
2649 	SET_CMC_TBL_MULTI_PORT_ID(skb->data, rtwvif->port);
2650 	if (chip->h2c_cctl_func_id == H2C_FUNC_MAC_CCTLINFO_UD_V1) {
2651 		SET_CMC_TBL_NOMINAL_PKT_PADDING_V1(skb->data, pads[RTW89_CHANNEL_WIDTH_20]);
2652 		SET_CMC_TBL_NOMINAL_PKT_PADDING40_V1(skb->data, pads[RTW89_CHANNEL_WIDTH_40]);
2653 		SET_CMC_TBL_NOMINAL_PKT_PADDING80_V1(skb->data, pads[RTW89_CHANNEL_WIDTH_80]);
2654 		SET_CMC_TBL_NOMINAL_PKT_PADDING160_V1(skb->data, pads[RTW89_CHANNEL_WIDTH_160]);
2655 	} else if (chip->h2c_cctl_func_id == H2C_FUNC_MAC_CCTLINFO_UD) {
2656 		SET_CMC_TBL_NOMINAL_PKT_PADDING(skb->data, pads[RTW89_CHANNEL_WIDTH_20]);
2657 		SET_CMC_TBL_NOMINAL_PKT_PADDING40(skb->data, pads[RTW89_CHANNEL_WIDTH_40]);
2658 		SET_CMC_TBL_NOMINAL_PKT_PADDING80(skb->data, pads[RTW89_CHANNEL_WIDTH_80]);
2659 		SET_CMC_TBL_NOMINAL_PKT_PADDING160(skb->data, pads[RTW89_CHANNEL_WIDTH_160]);
2660 	}
2661 	if (sta)
2662 		SET_CMC_TBL_BSR_QUEUE_SIZE_FORMAT(skb->data,
2663 						  sta->deflink.he_cap.has_he);
2664 	if (rtwvif->net_type == RTW89_NET_TYPE_AP_MODE)
2665 		SET_CMC_TBL_DATA_DCM(skb->data, 0);
2666 
2667 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2668 			      H2C_CAT_MAC, H2C_CL_MAC_FR_EXCHG,
2669 			      chip->h2c_cctl_func_id, 0, 1,
2670 			      H2C_CMC_TBL_LEN);
2671 
2672 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2673 	if (ret) {
2674 		rtw89_err(rtwdev, "failed to send h2c\n");
2675 		goto fail;
2676 	}
2677 
2678 	return 0;
2679 fail:
2680 	dev_kfree_skb_any(skb);
2681 
2682 	return ret;
2683 }
2684 EXPORT_SYMBOL(rtw89_fw_h2c_assoc_cmac_tbl);
2685 
2686 static void __get_sta_eht_pkt_padding(struct rtw89_dev *rtwdev,
2687 				      struct ieee80211_sta *sta, u8 *pads)
2688 {
2689 	u8 nss = min(sta->deflink.rx_nss, rtwdev->hal.tx_nss) - 1;
2690 	u16 ppe_thres_hdr;
2691 	u8 ppe16, ppe8;
2692 	u8 n, idx, sh;
2693 	u8 ru_bitmap;
2694 	bool ppe_th;
2695 	u16 ppe;
2696 	int i;
2697 
2698 	ppe_th = !!u8_get_bits(sta->deflink.eht_cap.eht_cap_elem.phy_cap_info[5],
2699 			       IEEE80211_EHT_PHY_CAP5_PPE_THRESHOLD_PRESENT);
2700 	if (!ppe_th) {
2701 		u8 pad;
2702 
2703 		pad = u8_get_bits(sta->deflink.eht_cap.eht_cap_elem.phy_cap_info[5],
2704 				  IEEE80211_EHT_PHY_CAP5_COMMON_NOMINAL_PKT_PAD_MASK);
2705 
2706 		for (i = 0; i < RTW89_PPE_BW_NUM; i++)
2707 			pads[i] = pad;
2708 
2709 		return;
2710 	}
2711 
2712 	ppe_thres_hdr = get_unaligned_le16(sta->deflink.eht_cap.eht_ppe_thres);
2713 	ru_bitmap = u16_get_bits(ppe_thres_hdr,
2714 				 IEEE80211_EHT_PPE_THRES_RU_INDEX_BITMASK_MASK);
2715 	n = hweight8(ru_bitmap);
2716 	n = IEEE80211_EHT_PPE_THRES_INFO_HEADER_SIZE +
2717 	    (n * IEEE80211_EHT_PPE_THRES_INFO_PPET_SIZE * 2) * nss;
2718 
2719 	for (i = 0; i < RTW89_PPE_BW_NUM; i++) {
2720 		if (!(ru_bitmap & BIT(i))) {
2721 			pads[i] = 1;
2722 			continue;
2723 		}
2724 
2725 		idx = n >> 3;
2726 		sh = n & 7;
2727 		n += IEEE80211_EHT_PPE_THRES_INFO_PPET_SIZE * 2;
2728 
2729 		ppe = get_unaligned_le16(sta->deflink.eht_cap.eht_ppe_thres + idx);
2730 		ppe16 = (ppe >> sh) & IEEE80211_PPE_THRES_NSS_MASK;
2731 		sh += IEEE80211_EHT_PPE_THRES_INFO_PPET_SIZE;
2732 		ppe8 = (ppe >> sh) & IEEE80211_PPE_THRES_NSS_MASK;
2733 
2734 		if (ppe16 != 7 && ppe8 == 7)
2735 			pads[i] = 2;
2736 		else if (ppe8 != 7)
2737 			pads[i] = 1;
2738 		else
2739 			pads[i] = 0;
2740 	}
2741 }
2742 
2743 int rtw89_fw_h2c_assoc_cmac_tbl_g7(struct rtw89_dev *rtwdev,
2744 				   struct ieee80211_vif *vif,
2745 				   struct ieee80211_sta *sta)
2746 {
2747 	const struct rtw89_chan *chan = rtw89_chan_get(rtwdev, RTW89_SUB_ENTITY_0);
2748 	struct rtw89_vif *rtwvif = (struct rtw89_vif *)vif->drv_priv;
2749 	struct rtw89_sta *rtwsta = sta_to_rtwsta_safe(sta);
2750 	u8 mac_id = rtwsta ? rtwsta->mac_id : rtwvif->mac_id;
2751 	struct rtw89_h2c_cctlinfo_ud_g7 *h2c;
2752 	u8 pads[RTW89_PPE_BW_NUM];
2753 	u32 len = sizeof(*h2c);
2754 	struct sk_buff *skb;
2755 	u16 lowest_rate;
2756 	int ret;
2757 
2758 	memset(pads, 0, sizeof(pads));
2759 	if (sta) {
2760 		if (sta->deflink.eht_cap.has_eht)
2761 			__get_sta_eht_pkt_padding(rtwdev, sta, pads);
2762 		else if (sta->deflink.he_cap.has_he)
2763 			__get_sta_he_pkt_padding(rtwdev, sta, pads);
2764 	}
2765 
2766 	if (vif->p2p)
2767 		lowest_rate = RTW89_HW_RATE_OFDM6;
2768 	else if (chan->band_type == RTW89_BAND_2G)
2769 		lowest_rate = RTW89_HW_RATE_CCK1;
2770 	else
2771 		lowest_rate = RTW89_HW_RATE_OFDM6;
2772 
2773 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
2774 	if (!skb) {
2775 		rtw89_err(rtwdev, "failed to alloc skb for cmac g7\n");
2776 		return -ENOMEM;
2777 	}
2778 	skb_put(skb, len);
2779 	h2c = (struct rtw89_h2c_cctlinfo_ud_g7 *)skb->data;
2780 
2781 	h2c->c0 = le32_encode_bits(mac_id, CCTLINFO_G7_C0_MACID) |
2782 		  le32_encode_bits(1, CCTLINFO_G7_C0_OP);
2783 
2784 	h2c->w0 = le32_encode_bits(1, CCTLINFO_G7_W0_DISRTSFB) |
2785 		  le32_encode_bits(1, CCTLINFO_G7_W0_DISDATAFB);
2786 	h2c->m0 = cpu_to_le32(CCTLINFO_G7_W0_DISRTSFB |
2787 			      CCTLINFO_G7_W0_DISDATAFB);
2788 
2789 	h2c->w1 = le32_encode_bits(lowest_rate, CCTLINFO_G7_W1_RTS_RTY_LOWEST_RATE);
2790 	h2c->m1 = cpu_to_le32(CCTLINFO_G7_W1_RTS_RTY_LOWEST_RATE);
2791 
2792 	h2c->w2 = le32_encode_bits(0, CCTLINFO_G7_W2_DATA_TXCNT_LMT_SEL);
2793 	h2c->m2 = cpu_to_le32(CCTLINFO_G7_W2_DATA_TXCNT_LMT_SEL);
2794 
2795 	h2c->w3 = le32_encode_bits(0, CCTLINFO_G7_W3_RTS_TXCNT_LMT_SEL);
2796 	h2c->m3 = cpu_to_le32(CCTLINFO_G7_W3_RTS_TXCNT_LMT_SEL);
2797 
2798 	h2c->w4 = le32_encode_bits(rtwvif->port, CCTLINFO_G7_W4_MULTI_PORT_ID);
2799 	h2c->m4 = cpu_to_le32(CCTLINFO_G7_W4_MULTI_PORT_ID);
2800 
2801 	if (rtwvif->net_type == RTW89_NET_TYPE_AP_MODE) {
2802 		h2c->w4 |= le32_encode_bits(0, CCTLINFO_G7_W4_DATA_DCM);
2803 		h2c->m4 |= cpu_to_le32(CCTLINFO_G7_W4_DATA_DCM);
2804 	}
2805 
2806 	if (vif->bss_conf.eht_support) {
2807 		u16 punct = vif->bss_conf.chanreq.oper.punctured;
2808 
2809 		h2c->w4 |= le32_encode_bits(~punct,
2810 					    CCTLINFO_G7_W4_ACT_SUBCH_CBW);
2811 		h2c->m4 |= cpu_to_le32(CCTLINFO_G7_W4_ACT_SUBCH_CBW);
2812 	}
2813 
2814 	h2c->w5 = le32_encode_bits(pads[RTW89_CHANNEL_WIDTH_20],
2815 				   CCTLINFO_G7_W5_NOMINAL_PKT_PADDING0) |
2816 		  le32_encode_bits(pads[RTW89_CHANNEL_WIDTH_40],
2817 				   CCTLINFO_G7_W5_NOMINAL_PKT_PADDING1) |
2818 		  le32_encode_bits(pads[RTW89_CHANNEL_WIDTH_80],
2819 				   CCTLINFO_G7_W5_NOMINAL_PKT_PADDING2) |
2820 		  le32_encode_bits(pads[RTW89_CHANNEL_WIDTH_160],
2821 				   CCTLINFO_G7_W5_NOMINAL_PKT_PADDING3) |
2822 		  le32_encode_bits(pads[RTW89_CHANNEL_WIDTH_320],
2823 				   CCTLINFO_G7_W5_NOMINAL_PKT_PADDING4);
2824 	h2c->m5 = cpu_to_le32(CCTLINFO_G7_W5_NOMINAL_PKT_PADDING0 |
2825 			      CCTLINFO_G7_W5_NOMINAL_PKT_PADDING1 |
2826 			      CCTLINFO_G7_W5_NOMINAL_PKT_PADDING2 |
2827 			      CCTLINFO_G7_W5_NOMINAL_PKT_PADDING3 |
2828 			      CCTLINFO_G7_W5_NOMINAL_PKT_PADDING4);
2829 
2830 	h2c->w6 = le32_encode_bits(vif->type == NL80211_IFTYPE_STATION ? 1 : 0,
2831 				   CCTLINFO_G7_W6_ULDL);
2832 	h2c->m6 = cpu_to_le32(CCTLINFO_G7_W6_ULDL);
2833 
2834 	if (sta) {
2835 		h2c->w8 = le32_encode_bits(sta->deflink.he_cap.has_he,
2836 					   CCTLINFO_G7_W8_BSR_QUEUE_SIZE_FORMAT);
2837 		h2c->m8 = cpu_to_le32(CCTLINFO_G7_W8_BSR_QUEUE_SIZE_FORMAT);
2838 	}
2839 
2840 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2841 			      H2C_CAT_MAC, H2C_CL_MAC_FR_EXCHG,
2842 			      H2C_FUNC_MAC_CCTLINFO_UD_G7, 0, 1,
2843 			      len);
2844 
2845 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2846 	if (ret) {
2847 		rtw89_err(rtwdev, "failed to send h2c\n");
2848 		goto fail;
2849 	}
2850 
2851 	return 0;
2852 fail:
2853 	dev_kfree_skb_any(skb);
2854 
2855 	return ret;
2856 }
2857 EXPORT_SYMBOL(rtw89_fw_h2c_assoc_cmac_tbl_g7);
2858 
2859 int rtw89_fw_h2c_ampdu_cmac_tbl_g7(struct rtw89_dev *rtwdev,
2860 				   struct ieee80211_vif *vif,
2861 				   struct ieee80211_sta *sta)
2862 {
2863 	struct rtw89_sta *rtwsta = (struct rtw89_sta *)sta->drv_priv;
2864 	struct rtw89_h2c_cctlinfo_ud_g7 *h2c;
2865 	u32 len = sizeof(*h2c);
2866 	struct sk_buff *skb;
2867 	u16 agg_num = 0;
2868 	u8 ba_bmap = 0;
2869 	int ret;
2870 	u8 tid;
2871 
2872 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
2873 	if (!skb) {
2874 		rtw89_err(rtwdev, "failed to alloc skb for ampdu cmac g7\n");
2875 		return -ENOMEM;
2876 	}
2877 	skb_put(skb, len);
2878 	h2c = (struct rtw89_h2c_cctlinfo_ud_g7 *)skb->data;
2879 
2880 	for_each_set_bit(tid, rtwsta->ampdu_map, IEEE80211_NUM_TIDS) {
2881 		if (agg_num == 0)
2882 			agg_num = rtwsta->ampdu_params[tid].agg_num;
2883 		else
2884 			agg_num = min(agg_num, rtwsta->ampdu_params[tid].agg_num);
2885 	}
2886 
2887 	if (agg_num <= 0x20)
2888 		ba_bmap = 3;
2889 	else if (agg_num > 0x20 && agg_num <= 0x40)
2890 		ba_bmap = 0;
2891 	else if (agg_num > 0x40 && agg_num <= 0x80)
2892 		ba_bmap = 1;
2893 	else if (agg_num > 0x80 && agg_num <= 0x100)
2894 		ba_bmap = 2;
2895 	else if (agg_num > 0x100 && agg_num <= 0x200)
2896 		ba_bmap = 4;
2897 	else if (agg_num > 0x200 && agg_num <= 0x400)
2898 		ba_bmap = 5;
2899 
2900 	h2c->c0 = le32_encode_bits(rtwsta->mac_id, CCTLINFO_G7_C0_MACID) |
2901 		  le32_encode_bits(1, CCTLINFO_G7_C0_OP);
2902 
2903 	h2c->w3 = le32_encode_bits(ba_bmap, CCTLINFO_G7_W3_BA_BMAP);
2904 	h2c->m3 = cpu_to_le32(CCTLINFO_G7_W3_BA_BMAP);
2905 
2906 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2907 			      H2C_CAT_MAC, H2C_CL_MAC_FR_EXCHG,
2908 			      H2C_FUNC_MAC_CCTLINFO_UD_G7, 0, 0,
2909 			      len);
2910 
2911 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2912 	if (ret) {
2913 		rtw89_err(rtwdev, "failed to send h2c\n");
2914 		goto fail;
2915 	}
2916 
2917 	return 0;
2918 fail:
2919 	dev_kfree_skb_any(skb);
2920 
2921 	return ret;
2922 }
2923 EXPORT_SYMBOL(rtw89_fw_h2c_ampdu_cmac_tbl_g7);
2924 
2925 int rtw89_fw_h2c_txtime_cmac_tbl(struct rtw89_dev *rtwdev,
2926 				 struct rtw89_sta *rtwsta)
2927 {
2928 	const struct rtw89_chip_info *chip = rtwdev->chip;
2929 	struct sk_buff *skb;
2930 	int ret;
2931 
2932 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_CMC_TBL_LEN);
2933 	if (!skb) {
2934 		rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
2935 		return -ENOMEM;
2936 	}
2937 	skb_put(skb, H2C_CMC_TBL_LEN);
2938 	SET_CTRL_INFO_MACID(skb->data, rtwsta->mac_id);
2939 	SET_CTRL_INFO_OPERATION(skb->data, 1);
2940 	if (rtwsta->cctl_tx_time) {
2941 		SET_CMC_TBL_AMPDU_TIME_SEL(skb->data, 1);
2942 		SET_CMC_TBL_AMPDU_MAX_TIME(skb->data, rtwsta->ampdu_max_time);
2943 	}
2944 	if (rtwsta->cctl_tx_retry_limit) {
2945 		SET_CMC_TBL_DATA_TXCNT_LMT_SEL(skb->data, 1);
2946 		SET_CMC_TBL_DATA_TX_CNT_LMT(skb->data, rtwsta->data_tx_cnt_lmt);
2947 	}
2948 
2949 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2950 			      H2C_CAT_MAC, H2C_CL_MAC_FR_EXCHG,
2951 			      chip->h2c_cctl_func_id, 0, 1,
2952 			      H2C_CMC_TBL_LEN);
2953 
2954 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2955 	if (ret) {
2956 		rtw89_err(rtwdev, "failed to send h2c\n");
2957 		goto fail;
2958 	}
2959 
2960 	return 0;
2961 fail:
2962 	dev_kfree_skb_any(skb);
2963 
2964 	return ret;
2965 }
2966 
2967 int rtw89_fw_h2c_txpath_cmac_tbl(struct rtw89_dev *rtwdev,
2968 				 struct rtw89_sta *rtwsta)
2969 {
2970 	const struct rtw89_chip_info *chip = rtwdev->chip;
2971 	struct sk_buff *skb;
2972 	int ret;
2973 
2974 	if (chip->h2c_cctl_func_id != H2C_FUNC_MAC_CCTLINFO_UD)
2975 		return 0;
2976 
2977 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_CMC_TBL_LEN);
2978 	if (!skb) {
2979 		rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
2980 		return -ENOMEM;
2981 	}
2982 	skb_put(skb, H2C_CMC_TBL_LEN);
2983 	SET_CTRL_INFO_MACID(skb->data, rtwsta->mac_id);
2984 	SET_CTRL_INFO_OPERATION(skb->data, 1);
2985 
2986 	__rtw89_fw_h2c_set_tx_path(rtwdev, skb);
2987 
2988 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
2989 			      H2C_CAT_MAC, H2C_CL_MAC_FR_EXCHG,
2990 			      H2C_FUNC_MAC_CCTLINFO_UD, 0, 1,
2991 			      H2C_CMC_TBL_LEN);
2992 
2993 	ret = rtw89_h2c_tx(rtwdev, skb, false);
2994 	if (ret) {
2995 		rtw89_err(rtwdev, "failed to send h2c\n");
2996 		goto fail;
2997 	}
2998 
2999 	return 0;
3000 fail:
3001 	dev_kfree_skb_any(skb);
3002 
3003 	return ret;
3004 }
3005 
3006 int rtw89_fw_h2c_update_beacon(struct rtw89_dev *rtwdev,
3007 			       struct rtw89_vif *rtwvif)
3008 {
3009 	const struct rtw89_chan *chan = rtw89_chan_get(rtwdev,
3010 						       rtwvif->sub_entity_idx);
3011 	struct ieee80211_vif *vif = rtwvif_to_vif(rtwvif);
3012 	struct rtw89_h2c_bcn_upd *h2c;
3013 	struct sk_buff *skb_beacon;
3014 	struct ieee80211_hdr *hdr;
3015 	u32 len = sizeof(*h2c);
3016 	struct sk_buff *skb;
3017 	int bcn_total_len;
3018 	u16 beacon_rate;
3019 	u16 tim_offset;
3020 	void *noa_data;
3021 	u8 noa_len;
3022 	int ret;
3023 
3024 	if (vif->p2p)
3025 		beacon_rate = RTW89_HW_RATE_OFDM6;
3026 	else if (chan->band_type == RTW89_BAND_2G)
3027 		beacon_rate = RTW89_HW_RATE_CCK1;
3028 	else
3029 		beacon_rate = RTW89_HW_RATE_OFDM6;
3030 
3031 	skb_beacon = ieee80211_beacon_get_tim(rtwdev->hw, vif, &tim_offset,
3032 					      NULL, 0);
3033 	if (!skb_beacon) {
3034 		rtw89_err(rtwdev, "failed to get beacon skb\n");
3035 		return -ENOMEM;
3036 	}
3037 
3038 	noa_len = rtw89_p2p_noa_fetch(rtwvif, &noa_data);
3039 	if (noa_len &&
3040 	    (noa_len <= skb_tailroom(skb_beacon) ||
3041 	     pskb_expand_head(skb_beacon, 0, noa_len, GFP_KERNEL) == 0)) {
3042 		skb_put_data(skb_beacon, noa_data, noa_len);
3043 	}
3044 
3045 	hdr = (struct ieee80211_hdr *)skb_beacon;
3046 	tim_offset -= ieee80211_hdrlen(hdr->frame_control);
3047 
3048 	bcn_total_len = len + skb_beacon->len;
3049 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, bcn_total_len);
3050 	if (!skb) {
3051 		rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
3052 		dev_kfree_skb_any(skb_beacon);
3053 		return -ENOMEM;
3054 	}
3055 	skb_put(skb, len);
3056 	h2c = (struct rtw89_h2c_bcn_upd *)skb->data;
3057 
3058 	h2c->w0 = le32_encode_bits(rtwvif->port, RTW89_H2C_BCN_UPD_W0_PORT) |
3059 		  le32_encode_bits(0, RTW89_H2C_BCN_UPD_W0_MBSSID) |
3060 		  le32_encode_bits(rtwvif->mac_idx, RTW89_H2C_BCN_UPD_W0_BAND) |
3061 		  le32_encode_bits(tim_offset | BIT(7), RTW89_H2C_BCN_UPD_W0_GRP_IE_OFST);
3062 	h2c->w1 = le32_encode_bits(rtwvif->mac_id, RTW89_H2C_BCN_UPD_W1_MACID) |
3063 		  le32_encode_bits(RTW89_MGMT_HW_SSN_SEL, RTW89_H2C_BCN_UPD_W1_SSN_SEL) |
3064 		  le32_encode_bits(RTW89_MGMT_HW_SEQ_MODE, RTW89_H2C_BCN_UPD_W1_SSN_MODE) |
3065 		  le32_encode_bits(beacon_rate, RTW89_H2C_BCN_UPD_W1_RATE);
3066 
3067 	skb_put_data(skb, skb_beacon->data, skb_beacon->len);
3068 	dev_kfree_skb_any(skb_beacon);
3069 
3070 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3071 			      H2C_CAT_MAC, H2C_CL_MAC_FR_EXCHG,
3072 			      H2C_FUNC_MAC_BCN_UPD, 0, 1,
3073 			      bcn_total_len);
3074 
3075 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3076 	if (ret) {
3077 		rtw89_err(rtwdev, "failed to send h2c\n");
3078 		dev_kfree_skb_any(skb);
3079 		return ret;
3080 	}
3081 
3082 	return 0;
3083 }
3084 EXPORT_SYMBOL(rtw89_fw_h2c_update_beacon);
3085 
3086 int rtw89_fw_h2c_update_beacon_be(struct rtw89_dev *rtwdev,
3087 				  struct rtw89_vif *rtwvif)
3088 {
3089 	const struct rtw89_chan *chan = rtw89_chan_get(rtwdev, RTW89_SUB_ENTITY_0);
3090 	struct ieee80211_vif *vif = rtwvif_to_vif(rtwvif);
3091 	struct rtw89_h2c_bcn_upd_be *h2c;
3092 	struct sk_buff *skb_beacon;
3093 	struct ieee80211_hdr *hdr;
3094 	u32 len = sizeof(*h2c);
3095 	struct sk_buff *skb;
3096 	int bcn_total_len;
3097 	u16 beacon_rate;
3098 	u16 tim_offset;
3099 	void *noa_data;
3100 	u8 noa_len;
3101 	int ret;
3102 
3103 	if (vif->p2p)
3104 		beacon_rate = RTW89_HW_RATE_OFDM6;
3105 	else if (chan->band_type == RTW89_BAND_2G)
3106 		beacon_rate = RTW89_HW_RATE_CCK1;
3107 	else
3108 		beacon_rate = RTW89_HW_RATE_OFDM6;
3109 
3110 	skb_beacon = ieee80211_beacon_get_tim(rtwdev->hw, vif, &tim_offset,
3111 					      NULL, 0);
3112 	if (!skb_beacon) {
3113 		rtw89_err(rtwdev, "failed to get beacon skb\n");
3114 		return -ENOMEM;
3115 	}
3116 
3117 	noa_len = rtw89_p2p_noa_fetch(rtwvif, &noa_data);
3118 	if (noa_len &&
3119 	    (noa_len <= skb_tailroom(skb_beacon) ||
3120 	     pskb_expand_head(skb_beacon, 0, noa_len, GFP_KERNEL) == 0)) {
3121 		skb_put_data(skb_beacon, noa_data, noa_len);
3122 	}
3123 
3124 	hdr = (struct ieee80211_hdr *)skb_beacon;
3125 	tim_offset -= ieee80211_hdrlen(hdr->frame_control);
3126 
3127 	bcn_total_len = len + skb_beacon->len;
3128 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, bcn_total_len);
3129 	if (!skb) {
3130 		rtw89_err(rtwdev, "failed to alloc skb for fw dl\n");
3131 		dev_kfree_skb_any(skb_beacon);
3132 		return -ENOMEM;
3133 	}
3134 	skb_put(skb, len);
3135 	h2c = (struct rtw89_h2c_bcn_upd_be *)skb->data;
3136 
3137 	h2c->w0 = le32_encode_bits(rtwvif->port, RTW89_H2C_BCN_UPD_BE_W0_PORT) |
3138 		  le32_encode_bits(0, RTW89_H2C_BCN_UPD_BE_W0_MBSSID) |
3139 		  le32_encode_bits(rtwvif->mac_idx, RTW89_H2C_BCN_UPD_BE_W0_BAND) |
3140 		  le32_encode_bits(tim_offset | BIT(7), RTW89_H2C_BCN_UPD_BE_W0_GRP_IE_OFST);
3141 	h2c->w1 = le32_encode_bits(rtwvif->mac_id, RTW89_H2C_BCN_UPD_BE_W1_MACID) |
3142 		  le32_encode_bits(RTW89_MGMT_HW_SSN_SEL, RTW89_H2C_BCN_UPD_BE_W1_SSN_SEL) |
3143 		  le32_encode_bits(RTW89_MGMT_HW_SEQ_MODE, RTW89_H2C_BCN_UPD_BE_W1_SSN_MODE) |
3144 		  le32_encode_bits(beacon_rate, RTW89_H2C_BCN_UPD_BE_W1_RATE);
3145 
3146 	skb_put_data(skb, skb_beacon->data, skb_beacon->len);
3147 	dev_kfree_skb_any(skb_beacon);
3148 
3149 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3150 			      H2C_CAT_MAC, H2C_CL_MAC_FR_EXCHG,
3151 			      H2C_FUNC_MAC_BCN_UPD_BE, 0, 1,
3152 			      bcn_total_len);
3153 
3154 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3155 	if (ret) {
3156 		rtw89_err(rtwdev, "failed to send h2c\n");
3157 		goto fail;
3158 	}
3159 
3160 	return 0;
3161 
3162 fail:
3163 	dev_kfree_skb_any(skb);
3164 
3165 	return ret;
3166 }
3167 EXPORT_SYMBOL(rtw89_fw_h2c_update_beacon_be);
3168 
3169 #define H2C_ROLE_MAINTAIN_LEN 4
3170 int rtw89_fw_h2c_role_maintain(struct rtw89_dev *rtwdev,
3171 			       struct rtw89_vif *rtwvif,
3172 			       struct rtw89_sta *rtwsta,
3173 			       enum rtw89_upd_mode upd_mode)
3174 {
3175 	struct sk_buff *skb;
3176 	u8 mac_id = rtwsta ? rtwsta->mac_id : rtwvif->mac_id;
3177 	u8 self_role;
3178 	int ret;
3179 
3180 	if (rtwvif->net_type == RTW89_NET_TYPE_AP_MODE) {
3181 		if (rtwsta)
3182 			self_role = RTW89_SELF_ROLE_AP_CLIENT;
3183 		else
3184 			self_role = rtwvif->self_role;
3185 	} else {
3186 		self_role = rtwvif->self_role;
3187 	}
3188 
3189 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_ROLE_MAINTAIN_LEN);
3190 	if (!skb) {
3191 		rtw89_err(rtwdev, "failed to alloc skb for h2c join\n");
3192 		return -ENOMEM;
3193 	}
3194 	skb_put(skb, H2C_ROLE_MAINTAIN_LEN);
3195 	SET_FWROLE_MAINTAIN_MACID(skb->data, mac_id);
3196 	SET_FWROLE_MAINTAIN_SELF_ROLE(skb->data, self_role);
3197 	SET_FWROLE_MAINTAIN_UPD_MODE(skb->data, upd_mode);
3198 	SET_FWROLE_MAINTAIN_WIFI_ROLE(skb->data, rtwvif->wifi_role);
3199 
3200 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3201 			      H2C_CAT_MAC, H2C_CL_MAC_MEDIA_RPT,
3202 			      H2C_FUNC_MAC_FWROLE_MAINTAIN, 0, 1,
3203 			      H2C_ROLE_MAINTAIN_LEN);
3204 
3205 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3206 	if (ret) {
3207 		rtw89_err(rtwdev, "failed to send h2c\n");
3208 		goto fail;
3209 	}
3210 
3211 	return 0;
3212 fail:
3213 	dev_kfree_skb_any(skb);
3214 
3215 	return ret;
3216 }
3217 
3218 static enum rtw89_fw_sta_type
3219 rtw89_fw_get_sta_type(struct rtw89_dev *rtwdev, struct rtw89_vif *rtwvif,
3220 		      struct rtw89_sta *rtwsta)
3221 {
3222 	struct ieee80211_sta *sta = rtwsta_to_sta_safe(rtwsta);
3223 	struct ieee80211_vif *vif = rtwvif_to_vif(rtwvif);
3224 
3225 	if (!sta)
3226 		goto by_vif;
3227 
3228 	if (sta->deflink.eht_cap.has_eht)
3229 		return RTW89_FW_BE_STA;
3230 	else if (sta->deflink.he_cap.has_he)
3231 		return RTW89_FW_AX_STA;
3232 	else
3233 		return RTW89_FW_N_AC_STA;
3234 
3235 by_vif:
3236 	if (vif->bss_conf.eht_support)
3237 		return RTW89_FW_BE_STA;
3238 	else if (vif->bss_conf.he_support)
3239 		return RTW89_FW_AX_STA;
3240 	else
3241 		return RTW89_FW_N_AC_STA;
3242 }
3243 
3244 int rtw89_fw_h2c_join_info(struct rtw89_dev *rtwdev, struct rtw89_vif *rtwvif,
3245 			   struct rtw89_sta *rtwsta, bool dis_conn)
3246 {
3247 	struct sk_buff *skb;
3248 	u8 mac_id = rtwsta ? rtwsta->mac_id : rtwvif->mac_id;
3249 	u8 self_role = rtwvif->self_role;
3250 	enum rtw89_fw_sta_type sta_type;
3251 	u8 net_type = rtwvif->net_type;
3252 	struct rtw89_h2c_join_v1 *h2c_v1;
3253 	struct rtw89_h2c_join *h2c;
3254 	u32 len = sizeof(*h2c);
3255 	bool format_v1 = false;
3256 	int ret;
3257 
3258 	if (rtwdev->chip->chip_gen == RTW89_CHIP_BE) {
3259 		len = sizeof(*h2c_v1);
3260 		format_v1 = true;
3261 	}
3262 
3263 	if (net_type == RTW89_NET_TYPE_AP_MODE && rtwsta) {
3264 		self_role = RTW89_SELF_ROLE_AP_CLIENT;
3265 		net_type = dis_conn ? RTW89_NET_TYPE_NO_LINK : net_type;
3266 	}
3267 
3268 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3269 	if (!skb) {
3270 		rtw89_err(rtwdev, "failed to alloc skb for h2c join\n");
3271 		return -ENOMEM;
3272 	}
3273 	skb_put(skb, len);
3274 	h2c = (struct rtw89_h2c_join *)skb->data;
3275 
3276 	h2c->w0 = le32_encode_bits(mac_id, RTW89_H2C_JOININFO_W0_MACID) |
3277 		  le32_encode_bits(dis_conn, RTW89_H2C_JOININFO_W0_OP) |
3278 		  le32_encode_bits(rtwvif->mac_idx, RTW89_H2C_JOININFO_W0_BAND) |
3279 		  le32_encode_bits(rtwvif->wmm, RTW89_H2C_JOININFO_W0_WMM) |
3280 		  le32_encode_bits(rtwvif->trigger, RTW89_H2C_JOININFO_W0_TGR) |
3281 		  le32_encode_bits(0, RTW89_H2C_JOININFO_W0_ISHESTA) |
3282 		  le32_encode_bits(0, RTW89_H2C_JOININFO_W0_DLBW) |
3283 		  le32_encode_bits(0, RTW89_H2C_JOININFO_W0_TF_MAC_PAD) |
3284 		  le32_encode_bits(0, RTW89_H2C_JOININFO_W0_DL_T_PE) |
3285 		  le32_encode_bits(rtwvif->port, RTW89_H2C_JOININFO_W0_PORT_ID) |
3286 		  le32_encode_bits(net_type, RTW89_H2C_JOININFO_W0_NET_TYPE) |
3287 		  le32_encode_bits(rtwvif->wifi_role, RTW89_H2C_JOININFO_W0_WIFI_ROLE) |
3288 		  le32_encode_bits(self_role, RTW89_H2C_JOININFO_W0_SELF_ROLE);
3289 
3290 	if (!format_v1)
3291 		goto done;
3292 
3293 	h2c_v1 = (struct rtw89_h2c_join_v1 *)skb->data;
3294 
3295 	sta_type = rtw89_fw_get_sta_type(rtwdev, rtwvif, rtwsta);
3296 
3297 	h2c_v1->w1 = le32_encode_bits(sta_type, RTW89_H2C_JOININFO_W1_STA_TYPE);
3298 	h2c_v1->w2 = 0;
3299 
3300 done:
3301 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3302 			      H2C_CAT_MAC, H2C_CL_MAC_MEDIA_RPT,
3303 			      H2C_FUNC_MAC_JOININFO, 0, 1,
3304 			      len);
3305 
3306 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3307 	if (ret) {
3308 		rtw89_err(rtwdev, "failed to send h2c\n");
3309 		goto fail;
3310 	}
3311 
3312 	return 0;
3313 fail:
3314 	dev_kfree_skb_any(skb);
3315 
3316 	return ret;
3317 }
3318 
3319 int rtw89_fw_h2c_notify_dbcc(struct rtw89_dev *rtwdev, bool en)
3320 {
3321 	struct rtw89_h2c_notify_dbcc *h2c;
3322 	u32 len = sizeof(*h2c);
3323 	struct sk_buff *skb;
3324 	int ret;
3325 
3326 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3327 	if (!skb) {
3328 		rtw89_err(rtwdev, "failed to alloc skb for h2c notify dbcc\n");
3329 		return -ENOMEM;
3330 	}
3331 	skb_put(skb, len);
3332 	h2c = (struct rtw89_h2c_notify_dbcc *)skb->data;
3333 
3334 	h2c->w0 = le32_encode_bits(en, RTW89_H2C_NOTIFY_DBCC_EN);
3335 
3336 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3337 			      H2C_CAT_MAC, H2C_CL_MAC_MEDIA_RPT,
3338 			      H2C_FUNC_NOTIFY_DBCC, 0, 1,
3339 			      len);
3340 
3341 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3342 	if (ret) {
3343 		rtw89_err(rtwdev, "failed to send h2c\n");
3344 		goto fail;
3345 	}
3346 
3347 	return 0;
3348 fail:
3349 	dev_kfree_skb_any(skb);
3350 
3351 	return ret;
3352 }
3353 
3354 int rtw89_fw_h2c_macid_pause(struct rtw89_dev *rtwdev, u8 sh, u8 grp,
3355 			     bool pause)
3356 {
3357 	struct rtw89_fw_macid_pause_sleep_grp *h2c_new;
3358 	struct rtw89_fw_macid_pause_grp *h2c;
3359 	__le32 set = cpu_to_le32(BIT(sh));
3360 	u8 h2c_macid_pause_id;
3361 	struct sk_buff *skb;
3362 	u32 len;
3363 	int ret;
3364 
3365 	if (RTW89_CHK_FW_FEATURE(MACID_PAUSE_SLEEP, &rtwdev->fw)) {
3366 		h2c_macid_pause_id = H2C_FUNC_MAC_MACID_PAUSE_SLEEP;
3367 		len = sizeof(*h2c_new);
3368 	} else {
3369 		h2c_macid_pause_id = H2C_FUNC_MAC_MACID_PAUSE;
3370 		len = sizeof(*h2c);
3371 	}
3372 
3373 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3374 	if (!skb) {
3375 		rtw89_err(rtwdev, "failed to alloc skb for h2c macid pause\n");
3376 		return -ENOMEM;
3377 	}
3378 	skb_put(skb, len);
3379 
3380 	if (h2c_macid_pause_id == H2C_FUNC_MAC_MACID_PAUSE_SLEEP) {
3381 		h2c_new = (struct rtw89_fw_macid_pause_sleep_grp *)skb->data;
3382 
3383 		h2c_new->n[0].pause_mask_grp[grp] = set;
3384 		h2c_new->n[0].sleep_mask_grp[grp] = set;
3385 		if (pause) {
3386 			h2c_new->n[0].pause_grp[grp] = set;
3387 			h2c_new->n[0].sleep_grp[grp] = set;
3388 		}
3389 	} else {
3390 		h2c = (struct rtw89_fw_macid_pause_grp *)skb->data;
3391 
3392 		h2c->mask_grp[grp] = set;
3393 		if (pause)
3394 			h2c->pause_grp[grp] = set;
3395 	}
3396 
3397 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3398 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
3399 			      h2c_macid_pause_id, 1, 0,
3400 			      len);
3401 
3402 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3403 	if (ret) {
3404 		rtw89_err(rtwdev, "failed to send h2c\n");
3405 		goto fail;
3406 	}
3407 
3408 	return 0;
3409 fail:
3410 	dev_kfree_skb_any(skb);
3411 
3412 	return ret;
3413 }
3414 
3415 #define H2C_EDCA_LEN 12
3416 int rtw89_fw_h2c_set_edca(struct rtw89_dev *rtwdev, struct rtw89_vif *rtwvif,
3417 			  u8 ac, u32 val)
3418 {
3419 	struct sk_buff *skb;
3420 	int ret;
3421 
3422 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_EDCA_LEN);
3423 	if (!skb) {
3424 		rtw89_err(rtwdev, "failed to alloc skb for h2c edca\n");
3425 		return -ENOMEM;
3426 	}
3427 	skb_put(skb, H2C_EDCA_LEN);
3428 	RTW89_SET_EDCA_SEL(skb->data, 0);
3429 	RTW89_SET_EDCA_BAND(skb->data, rtwvif->mac_idx);
3430 	RTW89_SET_EDCA_WMM(skb->data, 0);
3431 	RTW89_SET_EDCA_AC(skb->data, ac);
3432 	RTW89_SET_EDCA_PARAM(skb->data, val);
3433 
3434 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3435 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
3436 			      H2C_FUNC_USR_EDCA, 0, 1,
3437 			      H2C_EDCA_LEN);
3438 
3439 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3440 	if (ret) {
3441 		rtw89_err(rtwdev, "failed to send h2c\n");
3442 		goto fail;
3443 	}
3444 
3445 	return 0;
3446 fail:
3447 	dev_kfree_skb_any(skb);
3448 
3449 	return ret;
3450 }
3451 
3452 #define H2C_TSF32_TOGL_LEN 4
3453 int rtw89_fw_h2c_tsf32_toggle(struct rtw89_dev *rtwdev, struct rtw89_vif *rtwvif,
3454 			      bool en)
3455 {
3456 	struct sk_buff *skb;
3457 	u16 early_us = en ? 2000 : 0;
3458 	u8 *cmd;
3459 	int ret;
3460 
3461 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_TSF32_TOGL_LEN);
3462 	if (!skb) {
3463 		rtw89_err(rtwdev, "failed to alloc skb for h2c p2p act\n");
3464 		return -ENOMEM;
3465 	}
3466 	skb_put(skb, H2C_TSF32_TOGL_LEN);
3467 	cmd = skb->data;
3468 
3469 	RTW89_SET_FWCMD_TSF32_TOGL_BAND(cmd, rtwvif->mac_idx);
3470 	RTW89_SET_FWCMD_TSF32_TOGL_EN(cmd, en);
3471 	RTW89_SET_FWCMD_TSF32_TOGL_PORT(cmd, rtwvif->port);
3472 	RTW89_SET_FWCMD_TSF32_TOGL_EARLY(cmd, early_us);
3473 
3474 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3475 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
3476 			      H2C_FUNC_TSF32_TOGL, 0, 0,
3477 			      H2C_TSF32_TOGL_LEN);
3478 
3479 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3480 	if (ret) {
3481 		rtw89_err(rtwdev, "failed to send h2c\n");
3482 		goto fail;
3483 	}
3484 
3485 	return 0;
3486 fail:
3487 	dev_kfree_skb_any(skb);
3488 
3489 	return ret;
3490 }
3491 
3492 #define H2C_OFLD_CFG_LEN 8
3493 int rtw89_fw_h2c_set_ofld_cfg(struct rtw89_dev *rtwdev)
3494 {
3495 	static const u8 cfg[] = {0x09, 0x00, 0x00, 0x00, 0x5e, 0x00, 0x00, 0x00};
3496 	struct sk_buff *skb;
3497 	int ret;
3498 
3499 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_OFLD_CFG_LEN);
3500 	if (!skb) {
3501 		rtw89_err(rtwdev, "failed to alloc skb for h2c ofld\n");
3502 		return -ENOMEM;
3503 	}
3504 	skb_put_data(skb, cfg, H2C_OFLD_CFG_LEN);
3505 
3506 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3507 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
3508 			      H2C_FUNC_OFLD_CFG, 0, 1,
3509 			      H2C_OFLD_CFG_LEN);
3510 
3511 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3512 	if (ret) {
3513 		rtw89_err(rtwdev, "failed to send h2c\n");
3514 		goto fail;
3515 	}
3516 
3517 	return 0;
3518 fail:
3519 	dev_kfree_skb_any(skb);
3520 
3521 	return ret;
3522 }
3523 
3524 int rtw89_fw_h2c_set_bcn_fltr_cfg(struct rtw89_dev *rtwdev,
3525 				  struct ieee80211_vif *vif,
3526 				  bool connect)
3527 {
3528 	struct rtw89_vif *rtwvif = vif_to_rtwvif_safe(vif);
3529 	struct ieee80211_bss_conf *bss_conf = vif ? &vif->bss_conf : NULL;
3530 	s32 thold = RTW89_DEFAULT_CQM_THOLD;
3531 	u32 hyst = RTW89_DEFAULT_CQM_HYST;
3532 	struct rtw89_h2c_bcnfltr *h2c;
3533 	u32 len = sizeof(*h2c);
3534 	struct sk_buff *skb;
3535 	int ret;
3536 
3537 	if (!RTW89_CHK_FW_FEATURE(BEACON_FILTER, &rtwdev->fw))
3538 		return -EINVAL;
3539 
3540 	if (!rtwvif || !bss_conf || rtwvif->net_type != RTW89_NET_TYPE_INFRA)
3541 		return -EINVAL;
3542 
3543 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3544 	if (!skb) {
3545 		rtw89_err(rtwdev, "failed to alloc skb for h2c bcn filter\n");
3546 		return -ENOMEM;
3547 	}
3548 
3549 	skb_put(skb, len);
3550 	h2c = (struct rtw89_h2c_bcnfltr *)skb->data;
3551 
3552 	if (bss_conf->cqm_rssi_hyst)
3553 		hyst = bss_conf->cqm_rssi_hyst;
3554 	if (bss_conf->cqm_rssi_thold)
3555 		thold = bss_conf->cqm_rssi_thold;
3556 
3557 	h2c->w0 = le32_encode_bits(connect, RTW89_H2C_BCNFLTR_W0_MON_RSSI) |
3558 		  le32_encode_bits(connect, RTW89_H2C_BCNFLTR_W0_MON_BCN) |
3559 		  le32_encode_bits(connect, RTW89_H2C_BCNFLTR_W0_MON_EN) |
3560 		  le32_encode_bits(RTW89_BCN_FLTR_OFFLOAD_MODE_DEFAULT,
3561 				   RTW89_H2C_BCNFLTR_W0_MODE) |
3562 		  le32_encode_bits(RTW89_BCN_LOSS_CNT, RTW89_H2C_BCNFLTR_W0_BCN_LOSS_CNT) |
3563 		  le32_encode_bits(hyst, RTW89_H2C_BCNFLTR_W0_RSSI_HYST) |
3564 		  le32_encode_bits(thold + MAX_RSSI,
3565 				   RTW89_H2C_BCNFLTR_W0_RSSI_THRESHOLD) |
3566 		  le32_encode_bits(rtwvif->mac_id, RTW89_H2C_BCNFLTR_W0_MAC_ID);
3567 
3568 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3569 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
3570 			      H2C_FUNC_CFG_BCNFLTR, 0, 1, len);
3571 
3572 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3573 	if (ret) {
3574 		rtw89_err(rtwdev, "failed to send h2c\n");
3575 		goto fail;
3576 	}
3577 
3578 	return 0;
3579 fail:
3580 	dev_kfree_skb_any(skb);
3581 
3582 	return ret;
3583 }
3584 
3585 int rtw89_fw_h2c_rssi_offload(struct rtw89_dev *rtwdev,
3586 			      struct rtw89_rx_phy_ppdu *phy_ppdu)
3587 {
3588 	struct rtw89_h2c_ofld_rssi *h2c;
3589 	u32 len = sizeof(*h2c);
3590 	struct sk_buff *skb;
3591 	s8 rssi;
3592 	int ret;
3593 
3594 	if (!RTW89_CHK_FW_FEATURE(BEACON_FILTER, &rtwdev->fw))
3595 		return -EINVAL;
3596 
3597 	if (!phy_ppdu)
3598 		return -EINVAL;
3599 
3600 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3601 	if (!skb) {
3602 		rtw89_err(rtwdev, "failed to alloc skb for h2c rssi\n");
3603 		return -ENOMEM;
3604 	}
3605 
3606 	rssi = phy_ppdu->rssi_avg >> RSSI_FACTOR;
3607 	skb_put(skb, len);
3608 	h2c = (struct rtw89_h2c_ofld_rssi *)skb->data;
3609 
3610 	h2c->w0 = le32_encode_bits(phy_ppdu->mac_id, RTW89_H2C_OFLD_RSSI_W0_MACID) |
3611 		  le32_encode_bits(1, RTW89_H2C_OFLD_RSSI_W0_NUM);
3612 	h2c->w1 = le32_encode_bits(rssi, RTW89_H2C_OFLD_RSSI_W1_VAL);
3613 
3614 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3615 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
3616 			      H2C_FUNC_OFLD_RSSI, 0, 1, len);
3617 
3618 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3619 	if (ret) {
3620 		rtw89_err(rtwdev, "failed to send h2c\n");
3621 		goto fail;
3622 	}
3623 
3624 	return 0;
3625 fail:
3626 	dev_kfree_skb_any(skb);
3627 
3628 	return ret;
3629 }
3630 
3631 int rtw89_fw_h2c_tp_offload(struct rtw89_dev *rtwdev, struct rtw89_vif *rtwvif)
3632 {
3633 	struct rtw89_traffic_stats *stats = &rtwvif->stats;
3634 	struct rtw89_h2c_ofld *h2c;
3635 	u32 len = sizeof(*h2c);
3636 	struct sk_buff *skb;
3637 	int ret;
3638 
3639 	if (rtwvif->net_type != RTW89_NET_TYPE_INFRA)
3640 		return -EINVAL;
3641 
3642 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3643 	if (!skb) {
3644 		rtw89_err(rtwdev, "failed to alloc skb for h2c tp\n");
3645 		return -ENOMEM;
3646 	}
3647 
3648 	skb_put(skb, len);
3649 	h2c = (struct rtw89_h2c_ofld *)skb->data;
3650 
3651 	h2c->w0 = le32_encode_bits(rtwvif->mac_id, RTW89_H2C_OFLD_W0_MAC_ID) |
3652 		  le32_encode_bits(stats->tx_throughput, RTW89_H2C_OFLD_W0_TX_TP) |
3653 		  le32_encode_bits(stats->rx_throughput, RTW89_H2C_OFLD_W0_RX_TP);
3654 
3655 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3656 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
3657 			      H2C_FUNC_OFLD_TP, 0, 1, len);
3658 
3659 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3660 	if (ret) {
3661 		rtw89_err(rtwdev, "failed to send h2c\n");
3662 		goto fail;
3663 	}
3664 
3665 	return 0;
3666 fail:
3667 	dev_kfree_skb_any(skb);
3668 
3669 	return ret;
3670 }
3671 
3672 int rtw89_fw_h2c_ra(struct rtw89_dev *rtwdev, struct rtw89_ra_info *ra, bool csi)
3673 {
3674 	const struct rtw89_chip_info *chip = rtwdev->chip;
3675 	struct rtw89_h2c_ra_v1 *h2c_v1;
3676 	struct rtw89_h2c_ra *h2c;
3677 	u32 len = sizeof(*h2c);
3678 	bool format_v1 = false;
3679 	struct sk_buff *skb;
3680 	int ret;
3681 
3682 	if (chip->chip_gen == RTW89_CHIP_BE) {
3683 		len = sizeof(*h2c_v1);
3684 		format_v1 = true;
3685 	}
3686 
3687 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3688 	if (!skb) {
3689 		rtw89_err(rtwdev, "failed to alloc skb for h2c join\n");
3690 		return -ENOMEM;
3691 	}
3692 	skb_put(skb, len);
3693 	h2c = (struct rtw89_h2c_ra *)skb->data;
3694 	rtw89_debug(rtwdev, RTW89_DBG_RA,
3695 		    "ra cmd msk: %llx ", ra->ra_mask);
3696 
3697 	h2c->w0 = le32_encode_bits(ra->mode_ctrl, RTW89_H2C_RA_W0_MODE) |
3698 		  le32_encode_bits(ra->bw_cap, RTW89_H2C_RA_W0_BW_CAP) |
3699 		  le32_encode_bits(ra->macid, RTW89_H2C_RA_W0_MACID) |
3700 		  le32_encode_bits(ra->dcm_cap, RTW89_H2C_RA_W0_DCM) |
3701 		  le32_encode_bits(ra->er_cap, RTW89_H2C_RA_W0_ER) |
3702 		  le32_encode_bits(ra->init_rate_lv, RTW89_H2C_RA_W0_INIT_RATE_LV) |
3703 		  le32_encode_bits(ra->upd_all, RTW89_H2C_RA_W0_UPD_ALL) |
3704 		  le32_encode_bits(ra->en_sgi, RTW89_H2C_RA_W0_SGI) |
3705 		  le32_encode_bits(ra->ldpc_cap, RTW89_H2C_RA_W0_LDPC) |
3706 		  le32_encode_bits(ra->stbc_cap, RTW89_H2C_RA_W0_STBC) |
3707 		  le32_encode_bits(ra->ss_num, RTW89_H2C_RA_W0_SS_NUM) |
3708 		  le32_encode_bits(ra->giltf, RTW89_H2C_RA_W0_GILTF) |
3709 		  le32_encode_bits(ra->upd_bw_nss_mask, RTW89_H2C_RA_W0_UPD_BW_NSS_MASK) |
3710 		  le32_encode_bits(ra->upd_mask, RTW89_H2C_RA_W0_UPD_MASK);
3711 	h2c->w1 = le32_encode_bits(ra->ra_mask, RTW89_H2C_RA_W1_RAMASK_LO32);
3712 	h2c->w2 = le32_encode_bits(ra->ra_mask >> 32, RTW89_H2C_RA_W2_RAMASK_HI32);
3713 	h2c->w3 = le32_encode_bits(ra->fix_giltf_en, RTW89_H2C_RA_W3_FIX_GILTF_EN) |
3714 		  le32_encode_bits(ra->fix_giltf, RTW89_H2C_RA_W3_FIX_GILTF);
3715 
3716 	if (!format_v1)
3717 		goto csi;
3718 
3719 	h2c_v1 = (struct rtw89_h2c_ra_v1 *)h2c;
3720 	h2c_v1->w4 = le32_encode_bits(ra->mode_ctrl, RTW89_H2C_RA_V1_W4_MODE_EHT) |
3721 		     le32_encode_bits(ra->bw_cap, RTW89_H2C_RA_V1_W4_BW_EHT);
3722 
3723 csi:
3724 	if (!csi)
3725 		goto done;
3726 
3727 	h2c->w2 |= le32_encode_bits(1, RTW89_H2C_RA_W2_BFEE_CSI_CTL);
3728 	h2c->w3 |= le32_encode_bits(ra->band_num, RTW89_H2C_RA_W3_BAND_NUM) |
3729 		   le32_encode_bits(ra->cr_tbl_sel, RTW89_H2C_RA_W3_CR_TBL_SEL) |
3730 		   le32_encode_bits(ra->fixed_csi_rate_en, RTW89_H2C_RA_W3_FIXED_CSI_RATE_EN) |
3731 		   le32_encode_bits(ra->ra_csi_rate_en, RTW89_H2C_RA_W3_RA_CSI_RATE_EN) |
3732 		   le32_encode_bits(ra->csi_mcs_ss_idx, RTW89_H2C_RA_W3_FIXED_CSI_MCS_SS_IDX) |
3733 		   le32_encode_bits(ra->csi_mode, RTW89_H2C_RA_W3_FIXED_CSI_MODE) |
3734 		   le32_encode_bits(ra->csi_gi_ltf, RTW89_H2C_RA_W3_FIXED_CSI_GI_LTF) |
3735 		   le32_encode_bits(ra->csi_bw, RTW89_H2C_RA_W3_FIXED_CSI_BW);
3736 
3737 done:
3738 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3739 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_RA,
3740 			      H2C_FUNC_OUTSRC_RA_MACIDCFG, 0, 0,
3741 			      len);
3742 
3743 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3744 	if (ret) {
3745 		rtw89_err(rtwdev, "failed to send h2c\n");
3746 		goto fail;
3747 	}
3748 
3749 	return 0;
3750 fail:
3751 	dev_kfree_skb_any(skb);
3752 
3753 	return ret;
3754 }
3755 
3756 int rtw89_fw_h2c_cxdrv_init(struct rtw89_dev *rtwdev)
3757 {
3758 	struct rtw89_btc *btc = &rtwdev->btc;
3759 	struct rtw89_btc_dm *dm = &btc->dm;
3760 	struct rtw89_btc_init_info *init_info = &dm->init_info;
3761 	struct rtw89_btc_module *module = &init_info->module;
3762 	struct rtw89_btc_ant_info *ant = &module->ant;
3763 	struct rtw89_h2c_cxinit *h2c;
3764 	u32 len = sizeof(*h2c);
3765 	struct sk_buff *skb;
3766 	int ret;
3767 
3768 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3769 	if (!skb) {
3770 		rtw89_err(rtwdev, "failed to alloc skb for h2c cxdrv_init\n");
3771 		return -ENOMEM;
3772 	}
3773 	skb_put(skb, len);
3774 	h2c = (struct rtw89_h2c_cxinit *)skb->data;
3775 
3776 	h2c->hdr.type = CXDRVINFO_INIT;
3777 	h2c->hdr.len = len - H2C_LEN_CXDRVHDR;
3778 
3779 	h2c->ant_type = ant->type;
3780 	h2c->ant_num = ant->num;
3781 	h2c->ant_iso = ant->isolation;
3782 	h2c->ant_info =
3783 		u8_encode_bits(ant->single_pos, RTW89_H2C_CXINIT_ANT_INFO_POS) |
3784 		u8_encode_bits(ant->diversity, RTW89_H2C_CXINIT_ANT_INFO_DIVERSITY) |
3785 		u8_encode_bits(ant->btg_pos, RTW89_H2C_CXINIT_ANT_INFO_BTG_POS) |
3786 		u8_encode_bits(ant->stream_cnt, RTW89_H2C_CXINIT_ANT_INFO_STREAM_CNT);
3787 
3788 	h2c->mod_rfe = module->rfe_type;
3789 	h2c->mod_cv = module->cv;
3790 	h2c->mod_info =
3791 		u8_encode_bits(module->bt_solo, RTW89_H2C_CXINIT_MOD_INFO_BT_SOLO) |
3792 		u8_encode_bits(module->bt_pos, RTW89_H2C_CXINIT_MOD_INFO_BT_POS) |
3793 		u8_encode_bits(module->switch_type, RTW89_H2C_CXINIT_MOD_INFO_SW_TYPE) |
3794 		u8_encode_bits(module->wa_type, RTW89_H2C_CXINIT_MOD_INFO_WA_TYPE);
3795 	h2c->mod_adie_kt = module->kt_ver_adie;
3796 	h2c->wl_gch = init_info->wl_guard_ch;
3797 
3798 	h2c->info =
3799 		u8_encode_bits(init_info->wl_only, RTW89_H2C_CXINIT_INFO_WL_ONLY) |
3800 		u8_encode_bits(init_info->wl_init_ok, RTW89_H2C_CXINIT_INFO_WL_INITOK) |
3801 		u8_encode_bits(init_info->dbcc_en, RTW89_H2C_CXINIT_INFO_DBCC_EN) |
3802 		u8_encode_bits(init_info->cx_other, RTW89_H2C_CXINIT_INFO_CX_OTHER) |
3803 		u8_encode_bits(init_info->bt_only, RTW89_H2C_CXINIT_INFO_BT_ONLY);
3804 
3805 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3806 			      H2C_CAT_OUTSRC, BTFC_SET,
3807 			      SET_DRV_INFO, 0, 0,
3808 			      len);
3809 
3810 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3811 	if (ret) {
3812 		rtw89_err(rtwdev, "failed to send h2c\n");
3813 		goto fail;
3814 	}
3815 
3816 	return 0;
3817 fail:
3818 	dev_kfree_skb_any(skb);
3819 
3820 	return ret;
3821 }
3822 
3823 #define PORT_DATA_OFFSET 4
3824 #define H2C_LEN_CXDRVINFO_ROLE_DBCC_LEN 12
3825 #define H2C_LEN_CXDRVINFO_ROLE_SIZE(max_role_num) \
3826 	(4 + 12 * (max_role_num) + H2C_LEN_CXDRVHDR)
3827 
3828 int rtw89_fw_h2c_cxdrv_role(struct rtw89_dev *rtwdev)
3829 {
3830 	struct rtw89_btc *btc = &rtwdev->btc;
3831 	const struct rtw89_btc_ver *ver = btc->ver;
3832 	struct rtw89_btc_wl_info *wl = &btc->cx.wl;
3833 	struct rtw89_btc_wl_role_info *role_info = &wl->role_info;
3834 	struct rtw89_btc_wl_role_info_bpos *bpos = &role_info->role_map.role;
3835 	struct rtw89_btc_wl_active_role *active = role_info->active_role;
3836 	struct sk_buff *skb;
3837 	u32 len;
3838 	u8 offset = 0;
3839 	u8 *cmd;
3840 	int ret;
3841 	int i;
3842 
3843 	len = H2C_LEN_CXDRVINFO_ROLE_SIZE(ver->max_role_num);
3844 
3845 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3846 	if (!skb) {
3847 		rtw89_err(rtwdev, "failed to alloc skb for h2c cxdrv_role\n");
3848 		return -ENOMEM;
3849 	}
3850 	skb_put(skb, len);
3851 	cmd = skb->data;
3852 
3853 	RTW89_SET_FWCMD_CXHDR_TYPE(cmd, CXDRVINFO_ROLE);
3854 	RTW89_SET_FWCMD_CXHDR_LEN(cmd, len - H2C_LEN_CXDRVHDR);
3855 
3856 	RTW89_SET_FWCMD_CXROLE_CONNECT_CNT(cmd, role_info->connect_cnt);
3857 	RTW89_SET_FWCMD_CXROLE_LINK_MODE(cmd, role_info->link_mode);
3858 
3859 	RTW89_SET_FWCMD_CXROLE_ROLE_NONE(cmd, bpos->none);
3860 	RTW89_SET_FWCMD_CXROLE_ROLE_STA(cmd, bpos->station);
3861 	RTW89_SET_FWCMD_CXROLE_ROLE_AP(cmd, bpos->ap);
3862 	RTW89_SET_FWCMD_CXROLE_ROLE_VAP(cmd, bpos->vap);
3863 	RTW89_SET_FWCMD_CXROLE_ROLE_ADHOC(cmd, bpos->adhoc);
3864 	RTW89_SET_FWCMD_CXROLE_ROLE_ADHOC_MASTER(cmd, bpos->adhoc_master);
3865 	RTW89_SET_FWCMD_CXROLE_ROLE_MESH(cmd, bpos->mesh);
3866 	RTW89_SET_FWCMD_CXROLE_ROLE_MONITOR(cmd, bpos->moniter);
3867 	RTW89_SET_FWCMD_CXROLE_ROLE_P2P_DEV(cmd, bpos->p2p_device);
3868 	RTW89_SET_FWCMD_CXROLE_ROLE_P2P_GC(cmd, bpos->p2p_gc);
3869 	RTW89_SET_FWCMD_CXROLE_ROLE_P2P_GO(cmd, bpos->p2p_go);
3870 	RTW89_SET_FWCMD_CXROLE_ROLE_NAN(cmd, bpos->nan);
3871 
3872 	for (i = 0; i < RTW89_PORT_NUM; i++, active++) {
3873 		RTW89_SET_FWCMD_CXROLE_ACT_CONNECTED(cmd, active->connected, i, offset);
3874 		RTW89_SET_FWCMD_CXROLE_ACT_PID(cmd, active->pid, i, offset);
3875 		RTW89_SET_FWCMD_CXROLE_ACT_PHY(cmd, active->phy, i, offset);
3876 		RTW89_SET_FWCMD_CXROLE_ACT_NOA(cmd, active->noa, i, offset);
3877 		RTW89_SET_FWCMD_CXROLE_ACT_BAND(cmd, active->band, i, offset);
3878 		RTW89_SET_FWCMD_CXROLE_ACT_CLIENT_PS(cmd, active->client_ps, i, offset);
3879 		RTW89_SET_FWCMD_CXROLE_ACT_BW(cmd, active->bw, i, offset);
3880 		RTW89_SET_FWCMD_CXROLE_ACT_ROLE(cmd, active->role, i, offset);
3881 		RTW89_SET_FWCMD_CXROLE_ACT_CH(cmd, active->ch, i, offset);
3882 		RTW89_SET_FWCMD_CXROLE_ACT_TX_LVL(cmd, active->tx_lvl, i, offset);
3883 		RTW89_SET_FWCMD_CXROLE_ACT_RX_LVL(cmd, active->rx_lvl, i, offset);
3884 		RTW89_SET_FWCMD_CXROLE_ACT_TX_RATE(cmd, active->tx_rate, i, offset);
3885 		RTW89_SET_FWCMD_CXROLE_ACT_RX_RATE(cmd, active->rx_rate, i, offset);
3886 	}
3887 
3888 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3889 			      H2C_CAT_OUTSRC, BTFC_SET,
3890 			      SET_DRV_INFO, 0, 0,
3891 			      len);
3892 
3893 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3894 	if (ret) {
3895 		rtw89_err(rtwdev, "failed to send h2c\n");
3896 		goto fail;
3897 	}
3898 
3899 	return 0;
3900 fail:
3901 	dev_kfree_skb_any(skb);
3902 
3903 	return ret;
3904 }
3905 
3906 #define H2C_LEN_CXDRVINFO_ROLE_SIZE_V1(max_role_num) \
3907 	(4 + 16 * (max_role_num) + H2C_LEN_CXDRVINFO_ROLE_DBCC_LEN + H2C_LEN_CXDRVHDR)
3908 
3909 int rtw89_fw_h2c_cxdrv_role_v1(struct rtw89_dev *rtwdev)
3910 {
3911 	struct rtw89_btc *btc = &rtwdev->btc;
3912 	const struct rtw89_btc_ver *ver = btc->ver;
3913 	struct rtw89_btc_wl_info *wl = &btc->cx.wl;
3914 	struct rtw89_btc_wl_role_info_v1 *role_info = &wl->role_info_v1;
3915 	struct rtw89_btc_wl_role_info_bpos *bpos = &role_info->role_map.role;
3916 	struct rtw89_btc_wl_active_role_v1 *active = role_info->active_role_v1;
3917 	struct sk_buff *skb;
3918 	u32 len;
3919 	u8 *cmd, offset;
3920 	int ret;
3921 	int i;
3922 
3923 	len = H2C_LEN_CXDRVINFO_ROLE_SIZE_V1(ver->max_role_num);
3924 
3925 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
3926 	if (!skb) {
3927 		rtw89_err(rtwdev, "failed to alloc skb for h2c cxdrv_role\n");
3928 		return -ENOMEM;
3929 	}
3930 	skb_put(skb, len);
3931 	cmd = skb->data;
3932 
3933 	RTW89_SET_FWCMD_CXHDR_TYPE(cmd, CXDRVINFO_ROLE);
3934 	RTW89_SET_FWCMD_CXHDR_LEN(cmd, len - H2C_LEN_CXDRVHDR);
3935 
3936 	RTW89_SET_FWCMD_CXROLE_CONNECT_CNT(cmd, role_info->connect_cnt);
3937 	RTW89_SET_FWCMD_CXROLE_LINK_MODE(cmd, role_info->link_mode);
3938 
3939 	RTW89_SET_FWCMD_CXROLE_ROLE_NONE(cmd, bpos->none);
3940 	RTW89_SET_FWCMD_CXROLE_ROLE_STA(cmd, bpos->station);
3941 	RTW89_SET_FWCMD_CXROLE_ROLE_AP(cmd, bpos->ap);
3942 	RTW89_SET_FWCMD_CXROLE_ROLE_VAP(cmd, bpos->vap);
3943 	RTW89_SET_FWCMD_CXROLE_ROLE_ADHOC(cmd, bpos->adhoc);
3944 	RTW89_SET_FWCMD_CXROLE_ROLE_ADHOC_MASTER(cmd, bpos->adhoc_master);
3945 	RTW89_SET_FWCMD_CXROLE_ROLE_MESH(cmd, bpos->mesh);
3946 	RTW89_SET_FWCMD_CXROLE_ROLE_MONITOR(cmd, bpos->moniter);
3947 	RTW89_SET_FWCMD_CXROLE_ROLE_P2P_DEV(cmd, bpos->p2p_device);
3948 	RTW89_SET_FWCMD_CXROLE_ROLE_P2P_GC(cmd, bpos->p2p_gc);
3949 	RTW89_SET_FWCMD_CXROLE_ROLE_P2P_GO(cmd, bpos->p2p_go);
3950 	RTW89_SET_FWCMD_CXROLE_ROLE_NAN(cmd, bpos->nan);
3951 
3952 	offset = PORT_DATA_OFFSET;
3953 	for (i = 0; i < RTW89_PORT_NUM; i++, active++) {
3954 		RTW89_SET_FWCMD_CXROLE_ACT_CONNECTED(cmd, active->connected, i, offset);
3955 		RTW89_SET_FWCMD_CXROLE_ACT_PID(cmd, active->pid, i, offset);
3956 		RTW89_SET_FWCMD_CXROLE_ACT_PHY(cmd, active->phy, i, offset);
3957 		RTW89_SET_FWCMD_CXROLE_ACT_NOA(cmd, active->noa, i, offset);
3958 		RTW89_SET_FWCMD_CXROLE_ACT_BAND(cmd, active->band, i, offset);
3959 		RTW89_SET_FWCMD_CXROLE_ACT_CLIENT_PS(cmd, active->client_ps, i, offset);
3960 		RTW89_SET_FWCMD_CXROLE_ACT_BW(cmd, active->bw, i, offset);
3961 		RTW89_SET_FWCMD_CXROLE_ACT_ROLE(cmd, active->role, i, offset);
3962 		RTW89_SET_FWCMD_CXROLE_ACT_CH(cmd, active->ch, i, offset);
3963 		RTW89_SET_FWCMD_CXROLE_ACT_TX_LVL(cmd, active->tx_lvl, i, offset);
3964 		RTW89_SET_FWCMD_CXROLE_ACT_RX_LVL(cmd, active->rx_lvl, i, offset);
3965 		RTW89_SET_FWCMD_CXROLE_ACT_TX_RATE(cmd, active->tx_rate, i, offset);
3966 		RTW89_SET_FWCMD_CXROLE_ACT_RX_RATE(cmd, active->rx_rate, i, offset);
3967 		RTW89_SET_FWCMD_CXROLE_ACT_NOA_DUR(cmd, active->noa_duration, i, offset);
3968 	}
3969 
3970 	offset = len - H2C_LEN_CXDRVINFO_ROLE_DBCC_LEN;
3971 	RTW89_SET_FWCMD_CXROLE_MROLE_TYPE(cmd, role_info->mrole_type, offset);
3972 	RTW89_SET_FWCMD_CXROLE_MROLE_NOA(cmd, role_info->mrole_noa_duration, offset);
3973 	RTW89_SET_FWCMD_CXROLE_DBCC_EN(cmd, role_info->dbcc_en, offset);
3974 	RTW89_SET_FWCMD_CXROLE_DBCC_CHG(cmd, role_info->dbcc_chg, offset);
3975 	RTW89_SET_FWCMD_CXROLE_DBCC_2G_PHY(cmd, role_info->dbcc_2g_phy, offset);
3976 	RTW89_SET_FWCMD_CXROLE_LINK_MODE_CHG(cmd, role_info->link_mode_chg, offset);
3977 
3978 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
3979 			      H2C_CAT_OUTSRC, BTFC_SET,
3980 			      SET_DRV_INFO, 0, 0,
3981 			      len);
3982 
3983 	ret = rtw89_h2c_tx(rtwdev, skb, false);
3984 	if (ret) {
3985 		rtw89_err(rtwdev, "failed to send h2c\n");
3986 		goto fail;
3987 	}
3988 
3989 	return 0;
3990 fail:
3991 	dev_kfree_skb_any(skb);
3992 
3993 	return ret;
3994 }
3995 
3996 #define H2C_LEN_CXDRVINFO_ROLE_SIZE_V2(max_role_num) \
3997 	(4 + 8 * (max_role_num) + H2C_LEN_CXDRVINFO_ROLE_DBCC_LEN + H2C_LEN_CXDRVHDR)
3998 
3999 int rtw89_fw_h2c_cxdrv_role_v2(struct rtw89_dev *rtwdev)
4000 {
4001 	struct rtw89_btc *btc = &rtwdev->btc;
4002 	const struct rtw89_btc_ver *ver = btc->ver;
4003 	struct rtw89_btc_wl_info *wl = &btc->cx.wl;
4004 	struct rtw89_btc_wl_role_info_v2 *role_info = &wl->role_info_v2;
4005 	struct rtw89_btc_wl_role_info_bpos *bpos = &role_info->role_map.role;
4006 	struct rtw89_btc_wl_active_role_v2 *active = role_info->active_role_v2;
4007 	struct sk_buff *skb;
4008 	u32 len;
4009 	u8 *cmd, offset;
4010 	int ret;
4011 	int i;
4012 
4013 	len = H2C_LEN_CXDRVINFO_ROLE_SIZE_V2(ver->max_role_num);
4014 
4015 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
4016 	if (!skb) {
4017 		rtw89_err(rtwdev, "failed to alloc skb for h2c cxdrv_role\n");
4018 		return -ENOMEM;
4019 	}
4020 	skb_put(skb, len);
4021 	cmd = skb->data;
4022 
4023 	RTW89_SET_FWCMD_CXHDR_TYPE(cmd, CXDRVINFO_ROLE);
4024 	RTW89_SET_FWCMD_CXHDR_LEN(cmd, len - H2C_LEN_CXDRVHDR);
4025 
4026 	RTW89_SET_FWCMD_CXROLE_CONNECT_CNT(cmd, role_info->connect_cnt);
4027 	RTW89_SET_FWCMD_CXROLE_LINK_MODE(cmd, role_info->link_mode);
4028 
4029 	RTW89_SET_FWCMD_CXROLE_ROLE_NONE(cmd, bpos->none);
4030 	RTW89_SET_FWCMD_CXROLE_ROLE_STA(cmd, bpos->station);
4031 	RTW89_SET_FWCMD_CXROLE_ROLE_AP(cmd, bpos->ap);
4032 	RTW89_SET_FWCMD_CXROLE_ROLE_VAP(cmd, bpos->vap);
4033 	RTW89_SET_FWCMD_CXROLE_ROLE_ADHOC(cmd, bpos->adhoc);
4034 	RTW89_SET_FWCMD_CXROLE_ROLE_ADHOC_MASTER(cmd, bpos->adhoc_master);
4035 	RTW89_SET_FWCMD_CXROLE_ROLE_MESH(cmd, bpos->mesh);
4036 	RTW89_SET_FWCMD_CXROLE_ROLE_MONITOR(cmd, bpos->moniter);
4037 	RTW89_SET_FWCMD_CXROLE_ROLE_P2P_DEV(cmd, bpos->p2p_device);
4038 	RTW89_SET_FWCMD_CXROLE_ROLE_P2P_GC(cmd, bpos->p2p_gc);
4039 	RTW89_SET_FWCMD_CXROLE_ROLE_P2P_GO(cmd, bpos->p2p_go);
4040 	RTW89_SET_FWCMD_CXROLE_ROLE_NAN(cmd, bpos->nan);
4041 
4042 	offset = PORT_DATA_OFFSET;
4043 	for (i = 0; i < RTW89_PORT_NUM; i++, active++) {
4044 		RTW89_SET_FWCMD_CXROLE_ACT_CONNECTED_V2(cmd, active->connected, i, offset);
4045 		RTW89_SET_FWCMD_CXROLE_ACT_PID_V2(cmd, active->pid, i, offset);
4046 		RTW89_SET_FWCMD_CXROLE_ACT_PHY_V2(cmd, active->phy, i, offset);
4047 		RTW89_SET_FWCMD_CXROLE_ACT_NOA_V2(cmd, active->noa, i, offset);
4048 		RTW89_SET_FWCMD_CXROLE_ACT_BAND_V2(cmd, active->band, i, offset);
4049 		RTW89_SET_FWCMD_CXROLE_ACT_CLIENT_PS_V2(cmd, active->client_ps, i, offset);
4050 		RTW89_SET_FWCMD_CXROLE_ACT_BW_V2(cmd, active->bw, i, offset);
4051 		RTW89_SET_FWCMD_CXROLE_ACT_ROLE_V2(cmd, active->role, i, offset);
4052 		RTW89_SET_FWCMD_CXROLE_ACT_CH_V2(cmd, active->ch, i, offset);
4053 		RTW89_SET_FWCMD_CXROLE_ACT_NOA_DUR_V2(cmd, active->noa_duration, i, offset);
4054 	}
4055 
4056 	offset = len - H2C_LEN_CXDRVINFO_ROLE_DBCC_LEN;
4057 	RTW89_SET_FWCMD_CXROLE_MROLE_TYPE(cmd, role_info->mrole_type, offset);
4058 	RTW89_SET_FWCMD_CXROLE_MROLE_NOA(cmd, role_info->mrole_noa_duration, offset);
4059 	RTW89_SET_FWCMD_CXROLE_DBCC_EN(cmd, role_info->dbcc_en, offset);
4060 	RTW89_SET_FWCMD_CXROLE_DBCC_CHG(cmd, role_info->dbcc_chg, offset);
4061 	RTW89_SET_FWCMD_CXROLE_DBCC_2G_PHY(cmd, role_info->dbcc_2g_phy, offset);
4062 	RTW89_SET_FWCMD_CXROLE_LINK_MODE_CHG(cmd, role_info->link_mode_chg, offset);
4063 
4064 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4065 			      H2C_CAT_OUTSRC, BTFC_SET,
4066 			      SET_DRV_INFO, 0, 0,
4067 			      len);
4068 
4069 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4070 	if (ret) {
4071 		rtw89_err(rtwdev, "failed to send h2c\n");
4072 		goto fail;
4073 	}
4074 
4075 	return 0;
4076 fail:
4077 	dev_kfree_skb_any(skb);
4078 
4079 	return ret;
4080 }
4081 
4082 #define H2C_LEN_CXDRVINFO_CTRL (4 + H2C_LEN_CXDRVHDR)
4083 int rtw89_fw_h2c_cxdrv_ctrl(struct rtw89_dev *rtwdev)
4084 {
4085 	struct rtw89_btc *btc = &rtwdev->btc;
4086 	const struct rtw89_btc_ver *ver = btc->ver;
4087 	struct rtw89_btc_ctrl *ctrl = &btc->ctrl;
4088 	struct sk_buff *skb;
4089 	u8 *cmd;
4090 	int ret;
4091 
4092 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_LEN_CXDRVINFO_CTRL);
4093 	if (!skb) {
4094 		rtw89_err(rtwdev, "failed to alloc skb for h2c cxdrv_ctrl\n");
4095 		return -ENOMEM;
4096 	}
4097 	skb_put(skb, H2C_LEN_CXDRVINFO_CTRL);
4098 	cmd = skb->data;
4099 
4100 	RTW89_SET_FWCMD_CXHDR_TYPE(cmd, CXDRVINFO_CTRL);
4101 	RTW89_SET_FWCMD_CXHDR_LEN(cmd, H2C_LEN_CXDRVINFO_CTRL - H2C_LEN_CXDRVHDR);
4102 
4103 	RTW89_SET_FWCMD_CXCTRL_MANUAL(cmd, ctrl->manual);
4104 	RTW89_SET_FWCMD_CXCTRL_IGNORE_BT(cmd, ctrl->igno_bt);
4105 	RTW89_SET_FWCMD_CXCTRL_ALWAYS_FREERUN(cmd, ctrl->always_freerun);
4106 	if (ver->fcxctrl == 0)
4107 		RTW89_SET_FWCMD_CXCTRL_TRACE_STEP(cmd, ctrl->trace_step);
4108 
4109 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4110 			      H2C_CAT_OUTSRC, BTFC_SET,
4111 			      SET_DRV_INFO, 0, 0,
4112 			      H2C_LEN_CXDRVINFO_CTRL);
4113 
4114 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4115 	if (ret) {
4116 		rtw89_err(rtwdev, "failed to send h2c\n");
4117 		goto fail;
4118 	}
4119 
4120 	return 0;
4121 fail:
4122 	dev_kfree_skb_any(skb);
4123 
4124 	return ret;
4125 }
4126 
4127 #define H2C_LEN_CXDRVINFO_TRX (28 + H2C_LEN_CXDRVHDR)
4128 int rtw89_fw_h2c_cxdrv_trx(struct rtw89_dev *rtwdev)
4129 {
4130 	struct rtw89_btc *btc = &rtwdev->btc;
4131 	struct rtw89_btc_trx_info *trx = &btc->dm.trx_info;
4132 	struct sk_buff *skb;
4133 	u8 *cmd;
4134 	int ret;
4135 
4136 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_LEN_CXDRVINFO_TRX);
4137 	if (!skb) {
4138 		rtw89_err(rtwdev, "failed to alloc skb for h2c cxdrv_trx\n");
4139 		return -ENOMEM;
4140 	}
4141 	skb_put(skb, H2C_LEN_CXDRVINFO_TRX);
4142 	cmd = skb->data;
4143 
4144 	RTW89_SET_FWCMD_CXHDR_TYPE(cmd, CXDRVINFO_TRX);
4145 	RTW89_SET_FWCMD_CXHDR_LEN(cmd, H2C_LEN_CXDRVINFO_TRX - H2C_LEN_CXDRVHDR);
4146 
4147 	RTW89_SET_FWCMD_CXTRX_TXLV(cmd, trx->tx_lvl);
4148 	RTW89_SET_FWCMD_CXTRX_RXLV(cmd, trx->rx_lvl);
4149 	RTW89_SET_FWCMD_CXTRX_WLRSSI(cmd, trx->wl_rssi);
4150 	RTW89_SET_FWCMD_CXTRX_BTRSSI(cmd, trx->bt_rssi);
4151 	RTW89_SET_FWCMD_CXTRX_TXPWR(cmd, trx->tx_power);
4152 	RTW89_SET_FWCMD_CXTRX_RXGAIN(cmd, trx->rx_gain);
4153 	RTW89_SET_FWCMD_CXTRX_BTTXPWR(cmd, trx->bt_tx_power);
4154 	RTW89_SET_FWCMD_CXTRX_BTRXGAIN(cmd, trx->bt_rx_gain);
4155 	RTW89_SET_FWCMD_CXTRX_CN(cmd, trx->cn);
4156 	RTW89_SET_FWCMD_CXTRX_NHM(cmd, trx->nhm);
4157 	RTW89_SET_FWCMD_CXTRX_BTPROFILE(cmd, trx->bt_profile);
4158 	RTW89_SET_FWCMD_CXTRX_RSVD2(cmd, trx->rsvd2);
4159 	RTW89_SET_FWCMD_CXTRX_TXRATE(cmd, trx->tx_rate);
4160 	RTW89_SET_FWCMD_CXTRX_RXRATE(cmd, trx->rx_rate);
4161 	RTW89_SET_FWCMD_CXTRX_TXTP(cmd, trx->tx_tp);
4162 	RTW89_SET_FWCMD_CXTRX_RXTP(cmd, trx->rx_tp);
4163 	RTW89_SET_FWCMD_CXTRX_RXERRRA(cmd, trx->rx_err_ratio);
4164 
4165 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4166 			      H2C_CAT_OUTSRC, BTFC_SET,
4167 			      SET_DRV_INFO, 0, 0,
4168 			      H2C_LEN_CXDRVINFO_TRX);
4169 
4170 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4171 	if (ret) {
4172 		rtw89_err(rtwdev, "failed to send h2c\n");
4173 		goto fail;
4174 	}
4175 
4176 	return 0;
4177 fail:
4178 	dev_kfree_skb_any(skb);
4179 
4180 	return ret;
4181 }
4182 
4183 #define H2C_LEN_CXDRVINFO_RFK (4 + H2C_LEN_CXDRVHDR)
4184 int rtw89_fw_h2c_cxdrv_rfk(struct rtw89_dev *rtwdev)
4185 {
4186 	struct rtw89_btc *btc = &rtwdev->btc;
4187 	struct rtw89_btc_wl_info *wl = &btc->cx.wl;
4188 	struct rtw89_btc_wl_rfk_info *rfk_info = &wl->rfk_info;
4189 	struct sk_buff *skb;
4190 	u8 *cmd;
4191 	int ret;
4192 
4193 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_LEN_CXDRVINFO_RFK);
4194 	if (!skb) {
4195 		rtw89_err(rtwdev, "failed to alloc skb for h2c cxdrv_ctrl\n");
4196 		return -ENOMEM;
4197 	}
4198 	skb_put(skb, H2C_LEN_CXDRVINFO_RFK);
4199 	cmd = skb->data;
4200 
4201 	RTW89_SET_FWCMD_CXHDR_TYPE(cmd, CXDRVINFO_RFK);
4202 	RTW89_SET_FWCMD_CXHDR_LEN(cmd, H2C_LEN_CXDRVINFO_RFK - H2C_LEN_CXDRVHDR);
4203 
4204 	RTW89_SET_FWCMD_CXRFK_STATE(cmd, rfk_info->state);
4205 	RTW89_SET_FWCMD_CXRFK_PATH_MAP(cmd, rfk_info->path_map);
4206 	RTW89_SET_FWCMD_CXRFK_PHY_MAP(cmd, rfk_info->phy_map);
4207 	RTW89_SET_FWCMD_CXRFK_BAND(cmd, rfk_info->band);
4208 	RTW89_SET_FWCMD_CXRFK_TYPE(cmd, rfk_info->type);
4209 
4210 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4211 			      H2C_CAT_OUTSRC, BTFC_SET,
4212 			      SET_DRV_INFO, 0, 0,
4213 			      H2C_LEN_CXDRVINFO_RFK);
4214 
4215 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4216 	if (ret) {
4217 		rtw89_err(rtwdev, "failed to send h2c\n");
4218 		goto fail;
4219 	}
4220 
4221 	return 0;
4222 fail:
4223 	dev_kfree_skb_any(skb);
4224 
4225 	return ret;
4226 }
4227 
4228 #define H2C_LEN_PKT_OFLD 4
4229 int rtw89_fw_h2c_del_pkt_offload(struct rtw89_dev *rtwdev, u8 id)
4230 {
4231 	struct rtw89_wait_info *wait = &rtwdev->mac.fw_ofld_wait;
4232 	struct sk_buff *skb;
4233 	unsigned int cond;
4234 	u8 *cmd;
4235 	int ret;
4236 
4237 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_LEN_PKT_OFLD);
4238 	if (!skb) {
4239 		rtw89_err(rtwdev, "failed to alloc skb for h2c pkt offload\n");
4240 		return -ENOMEM;
4241 	}
4242 	skb_put(skb, H2C_LEN_PKT_OFLD);
4243 	cmd = skb->data;
4244 
4245 	RTW89_SET_FWCMD_PACKET_OFLD_PKT_IDX(cmd, id);
4246 	RTW89_SET_FWCMD_PACKET_OFLD_PKT_OP(cmd, RTW89_PKT_OFLD_OP_DEL);
4247 
4248 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4249 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
4250 			      H2C_FUNC_PACKET_OFLD, 1, 1,
4251 			      H2C_LEN_PKT_OFLD);
4252 
4253 	cond = RTW89_FW_OFLD_WAIT_COND_PKT_OFLD(id, RTW89_PKT_OFLD_OP_DEL);
4254 
4255 	ret = rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
4256 	if (ret < 0) {
4257 		rtw89_debug(rtwdev, RTW89_DBG_FW,
4258 			    "failed to del pkt ofld: id %d, ret %d\n",
4259 			    id, ret);
4260 		return ret;
4261 	}
4262 
4263 	rtw89_core_release_bit_map(rtwdev->pkt_offload, id);
4264 	return 0;
4265 }
4266 
4267 int rtw89_fw_h2c_add_pkt_offload(struct rtw89_dev *rtwdev, u8 *id,
4268 				 struct sk_buff *skb_ofld)
4269 {
4270 	struct rtw89_wait_info *wait = &rtwdev->mac.fw_ofld_wait;
4271 	struct sk_buff *skb;
4272 	unsigned int cond;
4273 	u8 *cmd;
4274 	u8 alloc_id;
4275 	int ret;
4276 
4277 	alloc_id = rtw89_core_acquire_bit_map(rtwdev->pkt_offload,
4278 					      RTW89_MAX_PKT_OFLD_NUM);
4279 	if (alloc_id == RTW89_MAX_PKT_OFLD_NUM)
4280 		return -ENOSPC;
4281 
4282 	*id = alloc_id;
4283 
4284 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_LEN_PKT_OFLD + skb_ofld->len);
4285 	if (!skb) {
4286 		rtw89_err(rtwdev, "failed to alloc skb for h2c pkt offload\n");
4287 		rtw89_core_release_bit_map(rtwdev->pkt_offload, alloc_id);
4288 		return -ENOMEM;
4289 	}
4290 	skb_put(skb, H2C_LEN_PKT_OFLD);
4291 	cmd = skb->data;
4292 
4293 	RTW89_SET_FWCMD_PACKET_OFLD_PKT_IDX(cmd, alloc_id);
4294 	RTW89_SET_FWCMD_PACKET_OFLD_PKT_OP(cmd, RTW89_PKT_OFLD_OP_ADD);
4295 	RTW89_SET_FWCMD_PACKET_OFLD_PKT_LENGTH(cmd, skb_ofld->len);
4296 	skb_put_data(skb, skb_ofld->data, skb_ofld->len);
4297 
4298 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4299 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
4300 			      H2C_FUNC_PACKET_OFLD, 1, 1,
4301 			      H2C_LEN_PKT_OFLD + skb_ofld->len);
4302 
4303 	cond = RTW89_FW_OFLD_WAIT_COND_PKT_OFLD(alloc_id, RTW89_PKT_OFLD_OP_ADD);
4304 
4305 	ret = rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
4306 	if (ret < 0) {
4307 		rtw89_debug(rtwdev, RTW89_DBG_FW,
4308 			    "failed to add pkt ofld: id %d, ret %d\n",
4309 			    alloc_id, ret);
4310 		rtw89_core_release_bit_map(rtwdev->pkt_offload, alloc_id);
4311 		return ret;
4312 	}
4313 
4314 	return 0;
4315 }
4316 
4317 int rtw89_fw_h2c_scan_list_offload(struct rtw89_dev *rtwdev, int ch_num,
4318 				   struct list_head *chan_list)
4319 {
4320 	struct rtw89_wait_info *wait = &rtwdev->mac.fw_ofld_wait;
4321 	struct rtw89_h2c_chinfo_elem *elem;
4322 	struct rtw89_mac_chinfo *ch_info;
4323 	struct rtw89_h2c_chinfo *h2c;
4324 	struct sk_buff *skb;
4325 	unsigned int cond;
4326 	int skb_len;
4327 	int ret;
4328 
4329 	static_assert(sizeof(*elem) == RTW89_MAC_CHINFO_SIZE);
4330 
4331 	skb_len = struct_size(h2c, elem, ch_num);
4332 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, skb_len);
4333 	if (!skb) {
4334 		rtw89_err(rtwdev, "failed to alloc skb for h2c scan list\n");
4335 		return -ENOMEM;
4336 	}
4337 	skb_put(skb, sizeof(*h2c));
4338 	h2c = (struct rtw89_h2c_chinfo *)skb->data;
4339 
4340 	h2c->ch_num = ch_num;
4341 	h2c->elem_size = sizeof(*elem) / 4; /* in unit of 4 bytes */
4342 
4343 	list_for_each_entry(ch_info, chan_list, list) {
4344 		elem = (struct rtw89_h2c_chinfo_elem *)skb_put(skb, sizeof(*elem));
4345 
4346 		elem->w0 = le32_encode_bits(ch_info->period, RTW89_H2C_CHINFO_W0_PERIOD) |
4347 			   le32_encode_bits(ch_info->dwell_time, RTW89_H2C_CHINFO_W0_DWELL) |
4348 			   le32_encode_bits(ch_info->central_ch, RTW89_H2C_CHINFO_W0_CENTER_CH) |
4349 			   le32_encode_bits(ch_info->pri_ch, RTW89_H2C_CHINFO_W0_PRI_CH);
4350 
4351 		elem->w1 = le32_encode_bits(ch_info->bw, RTW89_H2C_CHINFO_W1_BW) |
4352 			   le32_encode_bits(ch_info->notify_action, RTW89_H2C_CHINFO_W1_ACTION) |
4353 			   le32_encode_bits(ch_info->num_pkt, RTW89_H2C_CHINFO_W1_NUM_PKT) |
4354 			   le32_encode_bits(ch_info->tx_pkt, RTW89_H2C_CHINFO_W1_TX) |
4355 			   le32_encode_bits(ch_info->pause_data, RTW89_H2C_CHINFO_W1_PAUSE_DATA) |
4356 			   le32_encode_bits(ch_info->ch_band, RTW89_H2C_CHINFO_W1_BAND) |
4357 			   le32_encode_bits(ch_info->probe_id, RTW89_H2C_CHINFO_W1_PKT_ID) |
4358 			   le32_encode_bits(ch_info->dfs_ch, RTW89_H2C_CHINFO_W1_DFS) |
4359 			   le32_encode_bits(ch_info->tx_null, RTW89_H2C_CHINFO_W1_TX_NULL) |
4360 			   le32_encode_bits(ch_info->rand_seq_num, RTW89_H2C_CHINFO_W1_RANDOM);
4361 
4362 		elem->w2 = le32_encode_bits(ch_info->pkt_id[0], RTW89_H2C_CHINFO_W2_PKT0) |
4363 			   le32_encode_bits(ch_info->pkt_id[1], RTW89_H2C_CHINFO_W2_PKT1) |
4364 			   le32_encode_bits(ch_info->pkt_id[2], RTW89_H2C_CHINFO_W2_PKT2) |
4365 			   le32_encode_bits(ch_info->pkt_id[3], RTW89_H2C_CHINFO_W2_PKT3);
4366 
4367 		elem->w3 = le32_encode_bits(ch_info->pkt_id[4], RTW89_H2C_CHINFO_W3_PKT4) |
4368 			   le32_encode_bits(ch_info->pkt_id[5], RTW89_H2C_CHINFO_W3_PKT5) |
4369 			   le32_encode_bits(ch_info->pkt_id[6], RTW89_H2C_CHINFO_W3_PKT6) |
4370 			   le32_encode_bits(ch_info->pkt_id[7], RTW89_H2C_CHINFO_W3_PKT7);
4371 	}
4372 
4373 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4374 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
4375 			      H2C_FUNC_ADD_SCANOFLD_CH, 1, 1, skb_len);
4376 
4377 	cond = RTW89_SCANOFLD_WAIT_COND_ADD_CH;
4378 
4379 	ret = rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
4380 	if (ret) {
4381 		rtw89_debug(rtwdev, RTW89_DBG_FW, "failed to add scan ofld ch\n");
4382 		return ret;
4383 	}
4384 
4385 	return 0;
4386 }
4387 
4388 int rtw89_fw_h2c_scan_list_offload_be(struct rtw89_dev *rtwdev, int ch_num,
4389 				      struct list_head *chan_list)
4390 {
4391 	struct rtw89_wait_info *wait = &rtwdev->mac.fw_ofld_wait;
4392 	struct rtw89_h2c_chinfo_elem_be *elem;
4393 	struct rtw89_mac_chinfo_be *ch_info;
4394 	struct rtw89_h2c_chinfo *h2c;
4395 	struct sk_buff *skb;
4396 	unsigned int cond;
4397 	int skb_len;
4398 	int ret;
4399 
4400 	static_assert(sizeof(*elem) == RTW89_MAC_CHINFO_SIZE);
4401 
4402 	skb_len = struct_size(h2c, elem, ch_num);
4403 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, skb_len);
4404 	if (!skb) {
4405 		rtw89_err(rtwdev, "failed to alloc skb for h2c scan list\n");
4406 		return -ENOMEM;
4407 	}
4408 
4409 	skb_put(skb, sizeof(*h2c));
4410 	h2c = (struct rtw89_h2c_chinfo *)skb->data;
4411 
4412 	h2c->ch_num = ch_num;
4413 	h2c->elem_size = sizeof(*elem) / 4; /* in unit of 4 bytes */
4414 	h2c->arg = u8_encode_bits(RTW89_PHY_0, RTW89_H2C_CHINFO_ARG_MAC_IDX_MASK);
4415 
4416 	list_for_each_entry(ch_info, chan_list, list) {
4417 		elem = (struct rtw89_h2c_chinfo_elem_be *)skb_put(skb, sizeof(*elem));
4418 
4419 		elem->w0 = le32_encode_bits(ch_info->period, RTW89_H2C_CHINFO_BE_W0_PERIOD) |
4420 			   le32_encode_bits(ch_info->dwell_time, RTW89_H2C_CHINFO_BE_W0_DWELL) |
4421 			   le32_encode_bits(ch_info->central_ch,
4422 					    RTW89_H2C_CHINFO_BE_W0_CENTER_CH) |
4423 			   le32_encode_bits(ch_info->pri_ch, RTW89_H2C_CHINFO_BE_W0_PRI_CH);
4424 
4425 		elem->w1 = le32_encode_bits(ch_info->bw, RTW89_H2C_CHINFO_BE_W1_BW) |
4426 			   le32_encode_bits(ch_info->ch_band, RTW89_H2C_CHINFO_BE_W1_CH_BAND) |
4427 			   le32_encode_bits(ch_info->dfs_ch, RTW89_H2C_CHINFO_BE_W1_DFS) |
4428 			   le32_encode_bits(ch_info->pause_data,
4429 					    RTW89_H2C_CHINFO_BE_W1_PAUSE_DATA) |
4430 			   le32_encode_bits(ch_info->tx_null, RTW89_H2C_CHINFO_BE_W1_TX_NULL) |
4431 			   le32_encode_bits(ch_info->rand_seq_num,
4432 					    RTW89_H2C_CHINFO_BE_W1_RANDOM) |
4433 			   le32_encode_bits(ch_info->notify_action,
4434 					    RTW89_H2C_CHINFO_BE_W1_NOTIFY) |
4435 			   le32_encode_bits(ch_info->probe_id != 0xff ? 1 : 0,
4436 					    RTW89_H2C_CHINFO_BE_W1_PROBE) |
4437 			   le32_encode_bits(ch_info->leave_crit,
4438 					    RTW89_H2C_CHINFO_BE_W1_EARLY_LEAVE_CRIT) |
4439 			   le32_encode_bits(ch_info->chkpt_timer,
4440 					    RTW89_H2C_CHINFO_BE_W1_CHKPT_TIMER);
4441 
4442 		elem->w2 = le32_encode_bits(ch_info->leave_time,
4443 					    RTW89_H2C_CHINFO_BE_W2_EARLY_LEAVE_TIME) |
4444 			   le32_encode_bits(ch_info->leave_th,
4445 					    RTW89_H2C_CHINFO_BE_W2_EARLY_LEAVE_TH) |
4446 			   le32_encode_bits(ch_info->tx_pkt_ctrl,
4447 					    RTW89_H2C_CHINFO_BE_W2_TX_PKT_CTRL);
4448 
4449 		elem->w3 = le32_encode_bits(ch_info->pkt_id[0], RTW89_H2C_CHINFO_BE_W3_PKT0) |
4450 			   le32_encode_bits(ch_info->pkt_id[1], RTW89_H2C_CHINFO_BE_W3_PKT1) |
4451 			   le32_encode_bits(ch_info->pkt_id[2], RTW89_H2C_CHINFO_BE_W3_PKT2) |
4452 			   le32_encode_bits(ch_info->pkt_id[3], RTW89_H2C_CHINFO_BE_W3_PKT3);
4453 
4454 		elem->w4 = le32_encode_bits(ch_info->pkt_id[4], RTW89_H2C_CHINFO_BE_W4_PKT4) |
4455 			   le32_encode_bits(ch_info->pkt_id[5], RTW89_H2C_CHINFO_BE_W4_PKT5) |
4456 			   le32_encode_bits(ch_info->pkt_id[6], RTW89_H2C_CHINFO_BE_W4_PKT6) |
4457 			   le32_encode_bits(ch_info->pkt_id[7], RTW89_H2C_CHINFO_BE_W4_PKT7);
4458 
4459 		elem->w5 = le32_encode_bits(ch_info->sw_def, RTW89_H2C_CHINFO_BE_W5_SW_DEF) |
4460 			   le32_encode_bits(ch_info->fw_probe0_ssids,
4461 					    RTW89_H2C_CHINFO_BE_W5_FW_PROBE0_SSIDS);
4462 
4463 		elem->w6 = le32_encode_bits(ch_info->fw_probe0_shortssids,
4464 					    RTW89_H2C_CHINFO_BE_W6_FW_PROBE0_SHORTSSIDS) |
4465 			   le32_encode_bits(ch_info->fw_probe0_bssids,
4466 					    RTW89_H2C_CHINFO_BE_W6_FW_PROBE0_BSSIDS);
4467 	}
4468 
4469 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4470 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
4471 			      H2C_FUNC_ADD_SCANOFLD_CH, 1, 1, skb_len);
4472 
4473 	cond = RTW89_SCANOFLD_WAIT_COND_ADD_CH;
4474 
4475 	ret = rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
4476 	if (ret) {
4477 		rtw89_debug(rtwdev, RTW89_DBG_FW, "failed to add scan ofld ch\n");
4478 		return ret;
4479 	}
4480 
4481 	return 0;
4482 }
4483 
4484 int rtw89_fw_h2c_scan_offload(struct rtw89_dev *rtwdev,
4485 			      struct rtw89_scan_option *option,
4486 			      struct rtw89_vif *rtwvif)
4487 {
4488 	struct rtw89_wait_info *wait = &rtwdev->mac.fw_ofld_wait;
4489 	struct rtw89_chan *op = &rtwdev->scan_info.op_chan;
4490 	struct rtw89_h2c_scanofld *h2c;
4491 	u32 len = sizeof(*h2c);
4492 	struct sk_buff *skb;
4493 	unsigned int cond;
4494 	int ret;
4495 
4496 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
4497 	if (!skb) {
4498 		rtw89_err(rtwdev, "failed to alloc skb for h2c scan offload\n");
4499 		return -ENOMEM;
4500 	}
4501 	skb_put(skb, len);
4502 	h2c = (struct rtw89_h2c_scanofld *)skb->data;
4503 
4504 	h2c->w0 = le32_encode_bits(rtwvif->mac_id, RTW89_H2C_SCANOFLD_W0_MACID) |
4505 		  le32_encode_bits(rtwvif->port, RTW89_H2C_SCANOFLD_W0_PORT_ID) |
4506 		  le32_encode_bits(RTW89_PHY_0, RTW89_H2C_SCANOFLD_W0_BAND) |
4507 		  le32_encode_bits(option->enable, RTW89_H2C_SCANOFLD_W0_OPERATION);
4508 
4509 	h2c->w1 = le32_encode_bits(true, RTW89_H2C_SCANOFLD_W1_NOTIFY_END) |
4510 		  le32_encode_bits(option->target_ch_mode,
4511 				   RTW89_H2C_SCANOFLD_W1_TARGET_CH_MODE) |
4512 		  le32_encode_bits(RTW89_SCAN_IMMEDIATE,
4513 				   RTW89_H2C_SCANOFLD_W1_START_MODE) |
4514 		  le32_encode_bits(RTW89_SCAN_ONCE, RTW89_H2C_SCANOFLD_W1_SCAN_TYPE);
4515 
4516 	if (option->target_ch_mode) {
4517 		h2c->w1 |= le32_encode_bits(op->band_width,
4518 					    RTW89_H2C_SCANOFLD_W1_TARGET_CH_BW) |
4519 			   le32_encode_bits(op->primary_channel,
4520 					    RTW89_H2C_SCANOFLD_W1_TARGET_PRI_CH) |
4521 			   le32_encode_bits(op->channel,
4522 					    RTW89_H2C_SCANOFLD_W1_TARGET_CENTRAL_CH);
4523 		h2c->w0 |= le32_encode_bits(op->band_type,
4524 					    RTW89_H2C_SCANOFLD_W0_TARGET_CH_BAND);
4525 	}
4526 
4527 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4528 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
4529 			      H2C_FUNC_SCANOFLD, 1, 1,
4530 			      len);
4531 
4532 	if (option->enable)
4533 		cond = RTW89_SCANOFLD_WAIT_COND_START;
4534 	else
4535 		cond = RTW89_SCANOFLD_WAIT_COND_STOP;
4536 
4537 	ret = rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
4538 	if (ret) {
4539 		rtw89_debug(rtwdev, RTW89_DBG_FW, "failed to scan ofld\n");
4540 		return ret;
4541 	}
4542 
4543 	return 0;
4544 }
4545 
4546 static void rtw89_scan_get_6g_disabled_chan(struct rtw89_dev *rtwdev,
4547 					    struct rtw89_scan_option *option)
4548 {
4549 	struct ieee80211_supported_band *sband;
4550 	struct ieee80211_channel *chan;
4551 	u8 i, idx;
4552 
4553 	sband = rtwdev->hw->wiphy->bands[NL80211_BAND_6GHZ];
4554 
4555 	for (i = 0; i < sband->n_channels; i++) {
4556 		chan = &sband->channels[i];
4557 		if (chan->flags & IEEE80211_CHAN_DISABLED) {
4558 			idx = (chan->hw_value - 1) / 4;
4559 			option->prohib_chan |= BIT(idx);
4560 		}
4561 	}
4562 }
4563 
4564 int rtw89_fw_h2c_scan_offload_be(struct rtw89_dev *rtwdev,
4565 				 struct rtw89_scan_option *option,
4566 				 struct rtw89_vif *rtwvif)
4567 {
4568 	struct rtw89_hw_scan_info *scan_info = &rtwdev->scan_info;
4569 	struct rtw89_wait_info *wait = &rtwdev->mac.fw_ofld_wait;
4570 	struct rtw89_h2c_scanofld_be_macc_role *macc_role;
4571 	struct rtw89_chan *op = &scan_info->op_chan;
4572 	struct rtw89_h2c_scanofld_be_opch *opch;
4573 	struct rtw89_h2c_scanofld_be *h2c;
4574 	struct sk_buff *skb;
4575 	u8 macc_role_size = sizeof(*macc_role) * option->num_macc_role;
4576 	u8 opch_size = sizeof(*opch) * option->num_opch;
4577 	u8 probe_id[NUM_NL80211_BANDS];
4578 	unsigned int cond;
4579 	void *ptr;
4580 	int ret;
4581 	u32 len;
4582 	u8 i;
4583 
4584 	rtw89_scan_get_6g_disabled_chan(rtwdev, option);
4585 
4586 	len = sizeof(*h2c) + macc_role_size + opch_size;
4587 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
4588 	if (!skb) {
4589 		rtw89_err(rtwdev, "failed to alloc skb for h2c scan offload\n");
4590 		return -ENOMEM;
4591 	}
4592 
4593 	skb_put(skb, len);
4594 	h2c = (struct rtw89_h2c_scanofld_be *)skb->data;
4595 	ptr = skb->data;
4596 
4597 	h2c->w0 = le32_encode_bits(option->operation, RTW89_H2C_SCANOFLD_BE_W0_OP) |
4598 		  le32_encode_bits(option->scan_mode,
4599 				   RTW89_H2C_SCANOFLD_BE_W0_SCAN_MODE) |
4600 		  le32_encode_bits(option->repeat, RTW89_H2C_SCANOFLD_BE_W0_REPEAT) |
4601 		  le32_encode_bits(true, RTW89_H2C_SCANOFLD_BE_W0_NOTIFY_END) |
4602 		  le32_encode_bits(true, RTW89_H2C_SCANOFLD_BE_W0_LEARN_CH) |
4603 		  le32_encode_bits(rtwvif->mac_id, RTW89_H2C_SCANOFLD_BE_W0_MACID) |
4604 		  le32_encode_bits(rtwvif->port, RTW89_H2C_SCANOFLD_BE_W0_PORT) |
4605 		  le32_encode_bits(option->band, RTW89_H2C_SCANOFLD_BE_W0_BAND);
4606 
4607 	h2c->w1 = le32_encode_bits(option->num_macc_role, RTW89_H2C_SCANOFLD_BE_W1_NUM_MACC_ROLE) |
4608 		  le32_encode_bits(option->num_opch, RTW89_H2C_SCANOFLD_BE_W1_NUM_OP) |
4609 		  le32_encode_bits(option->norm_pd, RTW89_H2C_SCANOFLD_BE_W1_NORM_PD);
4610 
4611 	h2c->w2 = le32_encode_bits(option->slow_pd, RTW89_H2C_SCANOFLD_BE_W2_SLOW_PD) |
4612 		  le32_encode_bits(option->norm_cy, RTW89_H2C_SCANOFLD_BE_W2_NORM_CY) |
4613 		  le32_encode_bits(option->opch_end, RTW89_H2C_SCANOFLD_BE_W2_OPCH_END);
4614 
4615 	h2c->w3 = le32_encode_bits(0, RTW89_H2C_SCANOFLD_BE_W3_NUM_SSID) |
4616 		  le32_encode_bits(0, RTW89_H2C_SCANOFLD_BE_W3_NUM_SHORT_SSID) |
4617 		  le32_encode_bits(0, RTW89_H2C_SCANOFLD_BE_W3_NUM_BSSID) |
4618 		  le32_encode_bits(probe_id[NL80211_BAND_2GHZ], RTW89_H2C_SCANOFLD_BE_W3_PROBEID);
4619 
4620 	h2c->w4 = le32_encode_bits(probe_id[NL80211_BAND_5GHZ],
4621 				   RTW89_H2C_SCANOFLD_BE_W4_PROBE_5G) |
4622 		  le32_encode_bits(probe_id[NL80211_BAND_6GHZ],
4623 				   RTW89_H2C_SCANOFLD_BE_W4_PROBE_6G) |
4624 		  le32_encode_bits(0, RTW89_H2C_SCANOFLD_BE_W4_DELAY_START);
4625 
4626 	h2c->w5 = le32_encode_bits(option->mlo_mode, RTW89_H2C_SCANOFLD_BE_W5_MLO_MODE);
4627 
4628 	h2c->w6 = le32_encode_bits(option->prohib_chan,
4629 				   RTW89_H2C_SCANOFLD_BE_W6_CHAN_PROHIB_LOW);
4630 	h2c->w7 = le32_encode_bits(option->prohib_chan >> 32,
4631 				   RTW89_H2C_SCANOFLD_BE_W7_CHAN_PROHIB_HIGH);
4632 	ptr += sizeof(*h2c);
4633 
4634 	for (i = 0; i < option->num_macc_role; i++) {
4635 		macc_role = (struct rtw89_h2c_scanofld_be_macc_role *)&h2c->role[i];
4636 		macc_role->w0 =
4637 			le32_encode_bits(0, RTW89_H2C_SCANOFLD_BE_MACC_ROLE_W0_BAND) |
4638 			le32_encode_bits(0, RTW89_H2C_SCANOFLD_BE_MACC_ROLE_W0_PORT) |
4639 			le32_encode_bits(0, RTW89_H2C_SCANOFLD_BE_MACC_ROLE_W0_MACID) |
4640 			le32_encode_bits(0, RTW89_H2C_SCANOFLD_BE_MACC_ROLE_W0_OPCH_END);
4641 		ptr += sizeof(*macc_role);
4642 	}
4643 
4644 	for (i = 0; i < option->num_opch; i++) {
4645 		opch = ptr;
4646 		opch->w0 = le32_encode_bits(rtwvif->mac_id,
4647 					    RTW89_H2C_SCANOFLD_BE_OPCH_W0_MACID) |
4648 			   le32_encode_bits(option->band,
4649 					    RTW89_H2C_SCANOFLD_BE_OPCH_W0_BAND) |
4650 			   le32_encode_bits(rtwvif->port,
4651 					    RTW89_H2C_SCANOFLD_BE_OPCH_W0_PORT) |
4652 			   le32_encode_bits(RTW89_SCAN_OPMODE_INTV,
4653 					    RTW89_H2C_SCANOFLD_BE_OPCH_W0_POLICY) |
4654 			   le32_encode_bits(true,
4655 					    RTW89_H2C_SCANOFLD_BE_OPCH_W0_TXNULL) |
4656 			   le32_encode_bits(RTW89_OFF_CHAN_TIME / 10,
4657 					    RTW89_H2C_SCANOFLD_BE_OPCH_W0_POLICY_VAL);
4658 
4659 		opch->w1 = le32_encode_bits(RTW89_CHANNEL_TIME,
4660 					    RTW89_H2C_SCANOFLD_BE_OPCH_W1_DURATION) |
4661 			   le32_encode_bits(op->band_type,
4662 					    RTW89_H2C_SCANOFLD_BE_OPCH_W1_CH_BAND) |
4663 			   le32_encode_bits(op->band_width,
4664 					    RTW89_H2C_SCANOFLD_BE_OPCH_W1_BW) |
4665 			   le32_encode_bits(0x3,
4666 					    RTW89_H2C_SCANOFLD_BE_OPCH_W1_NOTIFY) |
4667 			   le32_encode_bits(op->primary_channel,
4668 					    RTW89_H2C_SCANOFLD_BE_OPCH_W1_PRI_CH) |
4669 			   le32_encode_bits(op->channel,
4670 					    RTW89_H2C_SCANOFLD_BE_OPCH_W1_CENTRAL_CH);
4671 
4672 		opch->w2 = le32_encode_bits(0,
4673 					    RTW89_H2C_SCANOFLD_BE_OPCH_W2_PKTS_CTRL) |
4674 			   le32_encode_bits(0,
4675 					    RTW89_H2C_SCANOFLD_BE_OPCH_W2_SW_DEF) |
4676 			   le32_encode_bits(2,
4677 					    RTW89_H2C_SCANOFLD_BE_OPCH_W2_SS);
4678 
4679 		opch->w3 = le32_encode_bits(RTW89_SCANOFLD_PKT_NONE,
4680 					    RTW89_H2C_SCANOFLD_BE_OPCH_W3_PKT0) |
4681 			   le32_encode_bits(RTW89_SCANOFLD_PKT_NONE,
4682 					    RTW89_H2C_SCANOFLD_BE_OPCH_W3_PKT1) |
4683 			   le32_encode_bits(RTW89_SCANOFLD_PKT_NONE,
4684 					    RTW89_H2C_SCANOFLD_BE_OPCH_W3_PKT2) |
4685 			   le32_encode_bits(RTW89_SCANOFLD_PKT_NONE,
4686 					    RTW89_H2C_SCANOFLD_BE_OPCH_W3_PKT3);
4687 		ptr += sizeof(*opch);
4688 	}
4689 
4690 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4691 			      H2C_CAT_MAC, H2C_CL_MAC_FW_OFLD,
4692 			      H2C_FUNC_SCANOFLD_BE, 1, 1,
4693 			      len);
4694 
4695 	if (option->enable)
4696 		cond = RTW89_SCANOFLD_BE_WAIT_COND_START;
4697 	else
4698 		cond = RTW89_SCANOFLD_BE_WAIT_COND_STOP;
4699 
4700 	ret = rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
4701 	if (ret) {
4702 		rtw89_debug(rtwdev, RTW89_DBG_FW, "failed to scan be ofld\n");
4703 		return ret;
4704 	}
4705 
4706 	return 0;
4707 }
4708 
4709 int rtw89_fw_h2c_rf_reg(struct rtw89_dev *rtwdev,
4710 			struct rtw89_fw_h2c_rf_reg_info *info,
4711 			u16 len, u8 page)
4712 {
4713 	struct sk_buff *skb;
4714 	u8 class = info->rf_path == RF_PATH_A ?
4715 		   H2C_CL_OUTSRC_RF_REG_A : H2C_CL_OUTSRC_RF_REG_B;
4716 	int ret;
4717 
4718 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
4719 	if (!skb) {
4720 		rtw89_err(rtwdev, "failed to alloc skb for h2c rf reg\n");
4721 		return -ENOMEM;
4722 	}
4723 	skb_put_data(skb, info->rtw89_phy_config_rf_h2c[page], len);
4724 
4725 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4726 			      H2C_CAT_OUTSRC, class, page, 0, 0,
4727 			      len);
4728 
4729 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4730 	if (ret) {
4731 		rtw89_err(rtwdev, "failed to send h2c\n");
4732 		goto fail;
4733 	}
4734 
4735 	return 0;
4736 fail:
4737 	dev_kfree_skb_any(skb);
4738 
4739 	return ret;
4740 }
4741 
4742 int rtw89_fw_h2c_rf_ntfy_mcc(struct rtw89_dev *rtwdev)
4743 {
4744 	struct rtw89_rfk_mcc_info *rfk_mcc = &rtwdev->rfk_mcc;
4745 	struct rtw89_fw_h2c_rf_get_mccch *mccch;
4746 	struct sk_buff *skb;
4747 	int ret;
4748 	u8 idx;
4749 
4750 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, sizeof(*mccch));
4751 	if (!skb) {
4752 		rtw89_err(rtwdev, "failed to alloc skb for h2c cxdrv_ctrl\n");
4753 		return -ENOMEM;
4754 	}
4755 	skb_put(skb, sizeof(*mccch));
4756 	mccch = (struct rtw89_fw_h2c_rf_get_mccch *)skb->data;
4757 
4758 	idx = rfk_mcc->table_idx;
4759 	mccch->ch_0 = cpu_to_le32(rfk_mcc->ch[0]);
4760 	mccch->ch_1 = cpu_to_le32(rfk_mcc->ch[1]);
4761 	mccch->band_0 = cpu_to_le32(rfk_mcc->band[0]);
4762 	mccch->band_1 = cpu_to_le32(rfk_mcc->band[1]);
4763 	mccch->current_channel = cpu_to_le32(rfk_mcc->ch[idx]);
4764 	mccch->current_band_type = cpu_to_le32(rfk_mcc->band[idx]);
4765 
4766 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4767 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_RF_FW_NOTIFY,
4768 			      H2C_FUNC_OUTSRC_RF_GET_MCCCH, 0, 0,
4769 			      sizeof(*mccch));
4770 
4771 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4772 	if (ret) {
4773 		rtw89_err(rtwdev, "failed to send h2c\n");
4774 		goto fail;
4775 	}
4776 
4777 	return 0;
4778 fail:
4779 	dev_kfree_skb_any(skb);
4780 
4781 	return ret;
4782 }
4783 EXPORT_SYMBOL(rtw89_fw_h2c_rf_ntfy_mcc);
4784 
4785 int rtw89_fw_h2c_rf_pre_ntfy(struct rtw89_dev *rtwdev,
4786 			     enum rtw89_phy_idx phy_idx)
4787 {
4788 	struct rtw89_rfk_mcc_info *rfk_mcc = &rtwdev->rfk_mcc;
4789 	struct rtw89_fw_h2c_rfk_pre_info *h2c;
4790 	u8 tbl_sel = rfk_mcc->table_idx;
4791 	u32 len = sizeof(*h2c);
4792 	struct sk_buff *skb;
4793 	u8 tbl, path;
4794 	u32 val32;
4795 	int ret;
4796 
4797 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
4798 	if (!skb) {
4799 		rtw89_err(rtwdev, "failed to alloc skb for h2c rfk_pre_ntfy\n");
4800 		return -ENOMEM;
4801 	}
4802 	skb_put(skb, len);
4803 	h2c = (struct rtw89_fw_h2c_rfk_pre_info *)skb->data;
4804 
4805 	h2c->mlo_mode = cpu_to_le32(rtwdev->mlo_dbcc_mode);
4806 
4807 	BUILD_BUG_ON(NUM_OF_RTW89_FW_RFK_TBL > RTW89_RFK_CHS_NR);
4808 
4809 	for (tbl = 0; tbl < NUM_OF_RTW89_FW_RFK_TBL; tbl++) {
4810 		for (path = 0; path < NUM_OF_RTW89_FW_RFK_PATH; path++) {
4811 			h2c->dbcc.ch[path][tbl] = cpu_to_le32(rfk_mcc->ch[tbl]);
4812 			h2c->dbcc.band[path][tbl] = cpu_to_le32(rfk_mcc->band[tbl]);
4813 		}
4814 	}
4815 
4816 	for (path = 0; path < NUM_OF_RTW89_FW_RFK_PATH; path++) {
4817 		h2c->tbl.cur_ch[path] = cpu_to_le32(rfk_mcc->ch[tbl_sel]);
4818 		h2c->tbl.cur_band[path] = cpu_to_le32(rfk_mcc->band[tbl_sel]);
4819 	}
4820 
4821 	h2c->phy_idx = cpu_to_le32(phy_idx);
4822 	h2c->cur_band = cpu_to_le32(rfk_mcc->band[tbl_sel]);
4823 	h2c->cur_bw = cpu_to_le32(rfk_mcc->bw[tbl_sel]);
4824 	h2c->cur_center_ch = cpu_to_le32(rfk_mcc->ch[tbl_sel]);
4825 
4826 	val32 = rtw89_phy_read32_mask(rtwdev, R_COEF_SEL, B_COEF_SEL_IQC_V1);
4827 	h2c->ktbl_sel0 = cpu_to_le32(val32);
4828 	val32 = rtw89_phy_read32_mask(rtwdev, R_COEF_SEL_C1, B_COEF_SEL_IQC_V1);
4829 	h2c->ktbl_sel1 = cpu_to_le32(val32);
4830 	val32 = rtw89_read_rf(rtwdev, RF_PATH_A, RR_CFGCH, RFREG_MASK);
4831 	h2c->rfmod0 = cpu_to_le32(val32);
4832 	val32 = rtw89_read_rf(rtwdev, RF_PATH_B, RR_CFGCH, RFREG_MASK);
4833 	h2c->rfmod1 = cpu_to_le32(val32);
4834 
4835 	if (rtw89_is_mlo_1_1(rtwdev))
4836 		h2c->mlo_1_1 = cpu_to_le32(1);
4837 
4838 	h2c->rfe_type = cpu_to_le32(rtwdev->efuse.rfe_type);
4839 
4840 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4841 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_RF_FW_RFK,
4842 			      H2C_FUNC_RFK_PRE_NOTIFY, 0, 0,
4843 			      len);
4844 
4845 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4846 	if (ret) {
4847 		rtw89_err(rtwdev, "failed to send h2c\n");
4848 		goto fail;
4849 	}
4850 
4851 	return 0;
4852 fail:
4853 	dev_kfree_skb_any(skb);
4854 
4855 	return ret;
4856 }
4857 
4858 int rtw89_fw_h2c_rf_tssi(struct rtw89_dev *rtwdev, enum rtw89_phy_idx phy_idx,
4859 			 enum rtw89_tssi_mode tssi_mode)
4860 {
4861 	const struct rtw89_chan *chan = rtw89_chan_get(rtwdev,
4862 						       RTW89_SUB_ENTITY_0);
4863 	struct rtw89_hal *hal = &rtwdev->hal;
4864 	struct rtw89_h2c_rf_tssi *h2c;
4865 	u32 len = sizeof(*h2c);
4866 	struct sk_buff *skb;
4867 	int ret;
4868 
4869 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
4870 	if (!skb) {
4871 		rtw89_err(rtwdev, "failed to alloc skb for h2c RF TSSI\n");
4872 		return -ENOMEM;
4873 	}
4874 	skb_put(skb, len);
4875 	h2c = (struct rtw89_h2c_rf_tssi *)skb->data;
4876 
4877 	h2c->len = cpu_to_le16(len);
4878 	h2c->phy = phy_idx;
4879 	h2c->ch = chan->channel;
4880 	h2c->bw = chan->band_width;
4881 	h2c->band = chan->band_type;
4882 	h2c->hwtx_en = true;
4883 	h2c->cv = hal->cv;
4884 	h2c->tssi_mode = tssi_mode;
4885 
4886 	rtw89_phy_rfk_tssi_fill_fwcmd_efuse_to_de(rtwdev, phy_idx, chan, h2c);
4887 	rtw89_phy_rfk_tssi_fill_fwcmd_tmeter_tbl(rtwdev, phy_idx, chan, h2c);
4888 
4889 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4890 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_RF_FW_RFK,
4891 			      H2C_FUNC_RFK_TSSI_OFFLOAD, 0, 0, len);
4892 
4893 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4894 	if (ret) {
4895 		rtw89_err(rtwdev, "failed to send h2c\n");
4896 		goto fail;
4897 	}
4898 
4899 	return 0;
4900 fail:
4901 	dev_kfree_skb_any(skb);
4902 
4903 	return ret;
4904 }
4905 
4906 int rtw89_fw_h2c_rf_iqk(struct rtw89_dev *rtwdev, enum rtw89_phy_idx phy_idx)
4907 {
4908 	struct rtw89_h2c_rf_iqk *h2c;
4909 	u32 len = sizeof(*h2c);
4910 	struct sk_buff *skb;
4911 	int ret;
4912 
4913 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
4914 	if (!skb) {
4915 		rtw89_err(rtwdev, "failed to alloc skb for h2c RF IQK\n");
4916 		return -ENOMEM;
4917 	}
4918 	skb_put(skb, len);
4919 	h2c = (struct rtw89_h2c_rf_iqk *)skb->data;
4920 
4921 	h2c->phy_idx = cpu_to_le32(phy_idx);
4922 	h2c->dbcc = cpu_to_le32(rtwdev->dbcc_en);
4923 
4924 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4925 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_RF_FW_RFK,
4926 			      H2C_FUNC_RFK_IQK_OFFLOAD, 0, 0, len);
4927 
4928 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4929 	if (ret) {
4930 		rtw89_err(rtwdev, "failed to send h2c\n");
4931 		goto fail;
4932 	}
4933 
4934 	return 0;
4935 fail:
4936 	dev_kfree_skb_any(skb);
4937 
4938 	return ret;
4939 }
4940 
4941 int rtw89_fw_h2c_rf_dpk(struct rtw89_dev *rtwdev, enum rtw89_phy_idx phy_idx)
4942 {
4943 	const struct rtw89_chan *chan = rtw89_chan_get(rtwdev,
4944 						       RTW89_SUB_ENTITY_0);
4945 	struct rtw89_h2c_rf_dpk *h2c;
4946 	u32 len = sizeof(*h2c);
4947 	struct sk_buff *skb;
4948 	int ret;
4949 
4950 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
4951 	if (!skb) {
4952 		rtw89_err(rtwdev, "failed to alloc skb for h2c RF DPK\n");
4953 		return -ENOMEM;
4954 	}
4955 	skb_put(skb, len);
4956 	h2c = (struct rtw89_h2c_rf_dpk *)skb->data;
4957 
4958 	h2c->len = len;
4959 	h2c->phy = phy_idx;
4960 	h2c->dpk_enable = true;
4961 	h2c->kpath = RF_AB;
4962 	h2c->cur_band = chan->band_type;
4963 	h2c->cur_bw = chan->band_width;
4964 	h2c->cur_ch = chan->channel;
4965 	h2c->dpk_dbg_en = rtw89_debug_is_enabled(rtwdev, RTW89_DBG_RFK);
4966 
4967 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
4968 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_RF_FW_RFK,
4969 			      H2C_FUNC_RFK_DPK_OFFLOAD, 0, 0, len);
4970 
4971 	ret = rtw89_h2c_tx(rtwdev, skb, false);
4972 	if (ret) {
4973 		rtw89_err(rtwdev, "failed to send h2c\n");
4974 		goto fail;
4975 	}
4976 
4977 	return 0;
4978 fail:
4979 	dev_kfree_skb_any(skb);
4980 
4981 	return ret;
4982 }
4983 
4984 int rtw89_fw_h2c_rf_txgapk(struct rtw89_dev *rtwdev, enum rtw89_phy_idx phy_idx)
4985 {
4986 	const struct rtw89_chan *chan = rtw89_chan_get(rtwdev,
4987 						       RTW89_SUB_ENTITY_0);
4988 	struct rtw89_hal *hal = &rtwdev->hal;
4989 	struct rtw89_h2c_rf_txgapk *h2c;
4990 	u32 len = sizeof(*h2c);
4991 	struct sk_buff *skb;
4992 	int ret;
4993 
4994 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
4995 	if (!skb) {
4996 		rtw89_err(rtwdev, "failed to alloc skb for h2c RF TXGAPK\n");
4997 		return -ENOMEM;
4998 	}
4999 	skb_put(skb, len);
5000 	h2c = (struct rtw89_h2c_rf_txgapk *)skb->data;
5001 
5002 	h2c->len = len;
5003 	h2c->ktype = 2;
5004 	h2c->phy = phy_idx;
5005 	h2c->kpath = RF_AB;
5006 	h2c->band = chan->band_type;
5007 	h2c->bw = chan->band_width;
5008 	h2c->ch = chan->channel;
5009 	h2c->cv = hal->cv;
5010 
5011 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
5012 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_RF_FW_RFK,
5013 			      H2C_FUNC_RFK_TXGAPK_OFFLOAD, 0, 0, len);
5014 
5015 	ret = rtw89_h2c_tx(rtwdev, skb, false);
5016 	if (ret) {
5017 		rtw89_err(rtwdev, "failed to send h2c\n");
5018 		goto fail;
5019 	}
5020 
5021 	return 0;
5022 fail:
5023 	dev_kfree_skb_any(skb);
5024 
5025 	return ret;
5026 }
5027 
5028 int rtw89_fw_h2c_rf_dack(struct rtw89_dev *rtwdev, enum rtw89_phy_idx phy_idx)
5029 {
5030 	struct rtw89_h2c_rf_dack *h2c;
5031 	u32 len = sizeof(*h2c);
5032 	struct sk_buff *skb;
5033 	int ret;
5034 
5035 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
5036 	if (!skb) {
5037 		rtw89_err(rtwdev, "failed to alloc skb for h2c RF DACK\n");
5038 		return -ENOMEM;
5039 	}
5040 	skb_put(skb, len);
5041 	h2c = (struct rtw89_h2c_rf_dack *)skb->data;
5042 
5043 	h2c->len = cpu_to_le32(len);
5044 	h2c->phy = cpu_to_le32(phy_idx);
5045 	h2c->type = cpu_to_le32(0);
5046 
5047 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
5048 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_RF_FW_RFK,
5049 			      H2C_FUNC_RFK_DACK_OFFLOAD, 0, 0, len);
5050 
5051 	ret = rtw89_h2c_tx(rtwdev, skb, false);
5052 	if (ret) {
5053 		rtw89_err(rtwdev, "failed to send h2c\n");
5054 		goto fail;
5055 	}
5056 
5057 	return 0;
5058 fail:
5059 	dev_kfree_skb_any(skb);
5060 
5061 	return ret;
5062 }
5063 
5064 int rtw89_fw_h2c_rf_rxdck(struct rtw89_dev *rtwdev, enum rtw89_phy_idx phy_idx)
5065 {
5066 	const struct rtw89_chan *chan = rtw89_chan_get(rtwdev,
5067 						       RTW89_SUB_ENTITY_0);
5068 	struct rtw89_h2c_rf_rxdck *h2c;
5069 	u32 len = sizeof(*h2c);
5070 	struct sk_buff *skb;
5071 	int ret;
5072 
5073 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
5074 	if (!skb) {
5075 		rtw89_err(rtwdev, "failed to alloc skb for h2c RF RXDCK\n");
5076 		return -ENOMEM;
5077 	}
5078 	skb_put(skb, len);
5079 	h2c = (struct rtw89_h2c_rf_rxdck *)skb->data;
5080 
5081 	h2c->len = len;
5082 	h2c->phy = phy_idx;
5083 	h2c->is_afe = false;
5084 	h2c->kpath = RF_AB;
5085 	h2c->cur_band = chan->band_type;
5086 	h2c->cur_bw = chan->band_width;
5087 	h2c->cur_ch = chan->channel;
5088 	h2c->rxdck_dbg_en = rtw89_debug_is_enabled(rtwdev, RTW89_DBG_RFK);
5089 
5090 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
5091 			      H2C_CAT_OUTSRC, H2C_CL_OUTSRC_RF_FW_RFK,
5092 			      H2C_FUNC_RFK_RXDCK_OFFLOAD, 0, 0, len);
5093 
5094 	ret = rtw89_h2c_tx(rtwdev, skb, false);
5095 	if (ret) {
5096 		rtw89_err(rtwdev, "failed to send h2c\n");
5097 		goto fail;
5098 	}
5099 
5100 	return 0;
5101 fail:
5102 	dev_kfree_skb_any(skb);
5103 
5104 	return ret;
5105 }
5106 
5107 int rtw89_fw_h2c_raw_with_hdr(struct rtw89_dev *rtwdev,
5108 			      u8 h2c_class, u8 h2c_func, u8 *buf, u16 len,
5109 			      bool rack, bool dack)
5110 {
5111 	struct sk_buff *skb;
5112 	int ret;
5113 
5114 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
5115 	if (!skb) {
5116 		rtw89_err(rtwdev, "failed to alloc skb for raw with hdr\n");
5117 		return -ENOMEM;
5118 	}
5119 	skb_put_data(skb, buf, len);
5120 
5121 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
5122 			      H2C_CAT_OUTSRC, h2c_class, h2c_func, rack, dack,
5123 			      len);
5124 
5125 	ret = rtw89_h2c_tx(rtwdev, skb, false);
5126 	if (ret) {
5127 		rtw89_err(rtwdev, "failed to send h2c\n");
5128 		goto fail;
5129 	}
5130 
5131 	return 0;
5132 fail:
5133 	dev_kfree_skb_any(skb);
5134 
5135 	return ret;
5136 }
5137 
5138 int rtw89_fw_h2c_raw(struct rtw89_dev *rtwdev, const u8 *buf, u16 len)
5139 {
5140 	struct sk_buff *skb;
5141 	int ret;
5142 
5143 	skb = rtw89_fw_h2c_alloc_skb_no_hdr(rtwdev, len);
5144 	if (!skb) {
5145 		rtw89_err(rtwdev, "failed to alloc skb for h2c raw\n");
5146 		return -ENOMEM;
5147 	}
5148 	skb_put_data(skb, buf, len);
5149 
5150 	ret = rtw89_h2c_tx(rtwdev, skb, false);
5151 	if (ret) {
5152 		rtw89_err(rtwdev, "failed to send h2c\n");
5153 		goto fail;
5154 	}
5155 
5156 	return 0;
5157 fail:
5158 	dev_kfree_skb_any(skb);
5159 
5160 	return ret;
5161 }
5162 
5163 void rtw89_fw_send_all_early_h2c(struct rtw89_dev *rtwdev)
5164 {
5165 	struct rtw89_early_h2c *early_h2c;
5166 
5167 	lockdep_assert_held(&rtwdev->mutex);
5168 
5169 	list_for_each_entry(early_h2c, &rtwdev->early_h2c_list, list) {
5170 		rtw89_fw_h2c_raw(rtwdev, early_h2c->h2c, early_h2c->h2c_len);
5171 	}
5172 }
5173 
5174 void rtw89_fw_free_all_early_h2c(struct rtw89_dev *rtwdev)
5175 {
5176 	struct rtw89_early_h2c *early_h2c, *tmp;
5177 
5178 	mutex_lock(&rtwdev->mutex);
5179 	list_for_each_entry_safe(early_h2c, tmp, &rtwdev->early_h2c_list, list) {
5180 		list_del(&early_h2c->list);
5181 		kfree(early_h2c->h2c);
5182 		kfree(early_h2c);
5183 	}
5184 	mutex_unlock(&rtwdev->mutex);
5185 }
5186 
5187 static void rtw89_fw_c2h_parse_attr(struct sk_buff *c2h)
5188 {
5189 	const struct rtw89_c2h_hdr *hdr = (const struct rtw89_c2h_hdr *)c2h->data;
5190 	struct rtw89_fw_c2h_attr *attr = RTW89_SKB_C2H_CB(c2h);
5191 
5192 	attr->category = le32_get_bits(hdr->w0, RTW89_C2H_HDR_W0_CATEGORY);
5193 	attr->class = le32_get_bits(hdr->w0, RTW89_C2H_HDR_W0_CLASS);
5194 	attr->func = le32_get_bits(hdr->w0, RTW89_C2H_HDR_W0_FUNC);
5195 	attr->len = le32_get_bits(hdr->w1, RTW89_C2H_HDR_W1_LEN);
5196 }
5197 
5198 static bool rtw89_fw_c2h_chk_atomic(struct rtw89_dev *rtwdev,
5199 				    struct sk_buff *c2h)
5200 {
5201 	struct rtw89_fw_c2h_attr *attr = RTW89_SKB_C2H_CB(c2h);
5202 	u8 category = attr->category;
5203 	u8 class = attr->class;
5204 	u8 func = attr->func;
5205 
5206 	switch (category) {
5207 	default:
5208 		return false;
5209 	case RTW89_C2H_CAT_MAC:
5210 		return rtw89_mac_c2h_chk_atomic(rtwdev, c2h, class, func);
5211 	case RTW89_C2H_CAT_OUTSRC:
5212 		return rtw89_phy_c2h_chk_atomic(rtwdev, class, func);
5213 	}
5214 }
5215 
5216 void rtw89_fw_c2h_irqsafe(struct rtw89_dev *rtwdev, struct sk_buff *c2h)
5217 {
5218 	rtw89_fw_c2h_parse_attr(c2h);
5219 	if (!rtw89_fw_c2h_chk_atomic(rtwdev, c2h))
5220 		goto enqueue;
5221 
5222 	rtw89_fw_c2h_cmd_handle(rtwdev, c2h);
5223 	dev_kfree_skb_any(c2h);
5224 	return;
5225 
5226 enqueue:
5227 	skb_queue_tail(&rtwdev->c2h_queue, c2h);
5228 	ieee80211_queue_work(rtwdev->hw, &rtwdev->c2h_work);
5229 }
5230 
5231 static void rtw89_fw_c2h_cmd_handle(struct rtw89_dev *rtwdev,
5232 				    struct sk_buff *skb)
5233 {
5234 	struct rtw89_fw_c2h_attr *attr = RTW89_SKB_C2H_CB(skb);
5235 	u8 category = attr->category;
5236 	u8 class = attr->class;
5237 	u8 func = attr->func;
5238 	u16 len = attr->len;
5239 	bool dump = true;
5240 
5241 	if (!test_bit(RTW89_FLAG_RUNNING, rtwdev->flags))
5242 		return;
5243 
5244 	switch (category) {
5245 	case RTW89_C2H_CAT_TEST:
5246 		break;
5247 	case RTW89_C2H_CAT_MAC:
5248 		rtw89_mac_c2h_handle(rtwdev, skb, len, class, func);
5249 		if (class == RTW89_MAC_C2H_CLASS_INFO &&
5250 		    func == RTW89_MAC_C2H_FUNC_C2H_LOG)
5251 			dump = false;
5252 		break;
5253 	case RTW89_C2H_CAT_OUTSRC:
5254 		if (class >= RTW89_PHY_C2H_CLASS_BTC_MIN &&
5255 		    class <= RTW89_PHY_C2H_CLASS_BTC_MAX)
5256 			rtw89_btc_c2h_handle(rtwdev, skb, len, class, func);
5257 		else
5258 			rtw89_phy_c2h_handle(rtwdev, skb, len, class, func);
5259 		break;
5260 	}
5261 
5262 	if (dump)
5263 		rtw89_hex_dump(rtwdev, RTW89_DBG_FW, "C2H: ", skb->data, skb->len);
5264 }
5265 
5266 void rtw89_fw_c2h_work(struct work_struct *work)
5267 {
5268 	struct rtw89_dev *rtwdev = container_of(work, struct rtw89_dev,
5269 						c2h_work);
5270 	struct sk_buff *skb, *tmp;
5271 
5272 	skb_queue_walk_safe(&rtwdev->c2h_queue, skb, tmp) {
5273 		skb_unlink(skb, &rtwdev->c2h_queue);
5274 		mutex_lock(&rtwdev->mutex);
5275 		rtw89_fw_c2h_cmd_handle(rtwdev, skb);
5276 		mutex_unlock(&rtwdev->mutex);
5277 		dev_kfree_skb_any(skb);
5278 	}
5279 }
5280 
5281 static int rtw89_fw_write_h2c_reg(struct rtw89_dev *rtwdev,
5282 				  struct rtw89_mac_h2c_info *info)
5283 {
5284 	const struct rtw89_chip_info *chip = rtwdev->chip;
5285 	struct rtw89_fw_info *fw_info = &rtwdev->fw;
5286 	const u32 *h2c_reg = chip->h2c_regs;
5287 	u8 i, val, len;
5288 	int ret;
5289 
5290 	ret = read_poll_timeout(rtw89_read8, val, val == 0, 1000, 5000, false,
5291 				rtwdev, chip->h2c_ctrl_reg);
5292 	if (ret) {
5293 		rtw89_warn(rtwdev, "FW does not process h2c registers\n");
5294 		return ret;
5295 	}
5296 
5297 	len = DIV_ROUND_UP(info->content_len + RTW89_H2CREG_HDR_LEN,
5298 			   sizeof(info->u.h2creg[0]));
5299 
5300 	u32p_replace_bits(&info->u.hdr.w0, info->id, RTW89_H2CREG_HDR_FUNC_MASK);
5301 	u32p_replace_bits(&info->u.hdr.w0, len, RTW89_H2CREG_HDR_LEN_MASK);
5302 
5303 	for (i = 0; i < RTW89_H2CREG_MAX; i++)
5304 		rtw89_write32(rtwdev, h2c_reg[i], info->u.h2creg[i]);
5305 
5306 	fw_info->h2c_counter++;
5307 	rtw89_write8_mask(rtwdev, chip->h2c_counter_reg.addr,
5308 			  chip->h2c_counter_reg.mask, fw_info->h2c_counter);
5309 	rtw89_write8(rtwdev, chip->h2c_ctrl_reg, B_AX_H2CREG_TRIGGER);
5310 
5311 	return 0;
5312 }
5313 
5314 static int rtw89_fw_read_c2h_reg(struct rtw89_dev *rtwdev,
5315 				 struct rtw89_mac_c2h_info *info)
5316 {
5317 	const struct rtw89_chip_info *chip = rtwdev->chip;
5318 	struct rtw89_fw_info *fw_info = &rtwdev->fw;
5319 	const u32 *c2h_reg = chip->c2h_regs;
5320 	u32 ret;
5321 	u8 i, val;
5322 
5323 	info->id = RTW89_FWCMD_C2HREG_FUNC_NULL;
5324 
5325 	ret = read_poll_timeout_atomic(rtw89_read8, val, val, 1,
5326 				       RTW89_C2H_TIMEOUT, false, rtwdev,
5327 				       chip->c2h_ctrl_reg);
5328 	if (ret) {
5329 		rtw89_warn(rtwdev, "c2h reg timeout\n");
5330 		return ret;
5331 	}
5332 
5333 	for (i = 0; i < RTW89_C2HREG_MAX; i++)
5334 		info->u.c2hreg[i] = rtw89_read32(rtwdev, c2h_reg[i]);
5335 
5336 	rtw89_write8(rtwdev, chip->c2h_ctrl_reg, 0);
5337 
5338 	info->id = u32_get_bits(info->u.hdr.w0, RTW89_C2HREG_HDR_FUNC_MASK);
5339 	info->content_len =
5340 		(u32_get_bits(info->u.hdr.w0, RTW89_C2HREG_HDR_LEN_MASK) << 2) -
5341 		RTW89_C2HREG_HDR_LEN;
5342 
5343 	fw_info->c2h_counter++;
5344 	rtw89_write8_mask(rtwdev, chip->c2h_counter_reg.addr,
5345 			  chip->c2h_counter_reg.mask, fw_info->c2h_counter);
5346 
5347 	return 0;
5348 }
5349 
5350 int rtw89_fw_msg_reg(struct rtw89_dev *rtwdev,
5351 		     struct rtw89_mac_h2c_info *h2c_info,
5352 		     struct rtw89_mac_c2h_info *c2h_info)
5353 {
5354 	u32 ret;
5355 
5356 	if (h2c_info && h2c_info->id != RTW89_FWCMD_H2CREG_FUNC_GET_FEATURE)
5357 		lockdep_assert_held(&rtwdev->mutex);
5358 
5359 	if (!h2c_info && !c2h_info)
5360 		return -EINVAL;
5361 
5362 	if (!h2c_info)
5363 		goto recv_c2h;
5364 
5365 	ret = rtw89_fw_write_h2c_reg(rtwdev, h2c_info);
5366 	if (ret)
5367 		return ret;
5368 
5369 recv_c2h:
5370 	if (!c2h_info)
5371 		return 0;
5372 
5373 	ret = rtw89_fw_read_c2h_reg(rtwdev, c2h_info);
5374 	if (ret)
5375 		return ret;
5376 
5377 	return 0;
5378 }
5379 
5380 void rtw89_fw_st_dbg_dump(struct rtw89_dev *rtwdev)
5381 {
5382 	if (!test_bit(RTW89_FLAG_POWERON, rtwdev->flags)) {
5383 		rtw89_err(rtwdev, "[ERR]pwr is off\n");
5384 		return;
5385 	}
5386 
5387 	rtw89_info(rtwdev, "FW status = 0x%x\n", rtw89_read32(rtwdev, R_AX_UDM0));
5388 	rtw89_info(rtwdev, "FW BADADDR = 0x%x\n", rtw89_read32(rtwdev, R_AX_UDM1));
5389 	rtw89_info(rtwdev, "FW EPC/RA = 0x%x\n", rtw89_read32(rtwdev, R_AX_UDM2));
5390 	rtw89_info(rtwdev, "FW MISC = 0x%x\n", rtw89_read32(rtwdev, R_AX_UDM3));
5391 	rtw89_info(rtwdev, "R_AX_HALT_C2H = 0x%x\n",
5392 		   rtw89_read32(rtwdev, R_AX_HALT_C2H));
5393 	rtw89_info(rtwdev, "R_AX_SER_DBG_INFO = 0x%x\n",
5394 		   rtw89_read32(rtwdev, R_AX_SER_DBG_INFO));
5395 
5396 	rtw89_fw_prog_cnt_dump(rtwdev);
5397 }
5398 
5399 static void rtw89_release_pkt_list(struct rtw89_dev *rtwdev)
5400 {
5401 	struct list_head *pkt_list = rtwdev->scan_info.pkt_list;
5402 	struct rtw89_pktofld_info *info, *tmp;
5403 	u8 idx;
5404 
5405 	for (idx = NL80211_BAND_2GHZ; idx < NUM_NL80211_BANDS; idx++) {
5406 		if (!(rtwdev->chip->support_bands & BIT(idx)))
5407 			continue;
5408 
5409 		list_for_each_entry_safe(info, tmp, &pkt_list[idx], list) {
5410 			if (test_bit(info->id, rtwdev->pkt_offload))
5411 				rtw89_fw_h2c_del_pkt_offload(rtwdev, info->id);
5412 			list_del(&info->list);
5413 			kfree(info);
5414 		}
5415 	}
5416 }
5417 
5418 static bool rtw89_is_6ghz_wildcard_probe_req(struct rtw89_dev *rtwdev,
5419 					     struct rtw89_vif *rtwvif,
5420 					     struct rtw89_pktofld_info *info,
5421 					     enum nl80211_band band, u8 ssid_idx)
5422 {
5423 	struct cfg80211_scan_request *req = rtwvif->scan_req;
5424 
5425 	if (band != NL80211_BAND_6GHZ)
5426 		return false;
5427 
5428 	if (req->ssids[ssid_idx].ssid_len) {
5429 		memcpy(info->ssid, req->ssids[ssid_idx].ssid,
5430 		       req->ssids[ssid_idx].ssid_len);
5431 		info->ssid_len = req->ssids[ssid_idx].ssid_len;
5432 		return false;
5433 	} else {
5434 		return true;
5435 	}
5436 }
5437 
5438 static int rtw89_append_probe_req_ie(struct rtw89_dev *rtwdev,
5439 				     struct rtw89_vif *rtwvif,
5440 				     struct sk_buff *skb, u8 ssid_idx)
5441 {
5442 	struct rtw89_hw_scan_info *scan_info = &rtwdev->scan_info;
5443 	struct ieee80211_scan_ies *ies = rtwvif->scan_ies;
5444 	struct rtw89_pktofld_info *info;
5445 	struct sk_buff *new;
5446 	int ret = 0;
5447 	u8 band;
5448 
5449 	for (band = NL80211_BAND_2GHZ; band < NUM_NL80211_BANDS; band++) {
5450 		if (!(rtwdev->chip->support_bands & BIT(band)))
5451 			continue;
5452 
5453 		new = skb_copy(skb, GFP_KERNEL);
5454 		if (!new) {
5455 			ret = -ENOMEM;
5456 			goto out;
5457 		}
5458 		skb_put_data(new, ies->ies[band], ies->len[band]);
5459 		skb_put_data(new, ies->common_ies, ies->common_ie_len);
5460 
5461 		info = kzalloc(sizeof(*info), GFP_KERNEL);
5462 		if (!info) {
5463 			ret = -ENOMEM;
5464 			kfree_skb(new);
5465 			goto out;
5466 		}
5467 
5468 		if (rtw89_is_6ghz_wildcard_probe_req(rtwdev, rtwvif, info, band,
5469 						     ssid_idx)) {
5470 			kfree_skb(new);
5471 			kfree(info);
5472 			goto out;
5473 		}
5474 
5475 		ret = rtw89_fw_h2c_add_pkt_offload(rtwdev, &info->id, new);
5476 		if (ret) {
5477 			kfree_skb(new);
5478 			kfree(info);
5479 			goto out;
5480 		}
5481 
5482 		list_add_tail(&info->list, &scan_info->pkt_list[band]);
5483 		kfree_skb(new);
5484 	}
5485 out:
5486 	return ret;
5487 }
5488 
5489 static int rtw89_hw_scan_update_probe_req(struct rtw89_dev *rtwdev,
5490 					  struct rtw89_vif *rtwvif)
5491 {
5492 	struct cfg80211_scan_request *req = rtwvif->scan_req;
5493 	struct sk_buff *skb;
5494 	u8 num = req->n_ssids, i;
5495 	int ret;
5496 
5497 	for (i = 0; i < num; i++) {
5498 		skb = ieee80211_probereq_get(rtwdev->hw, rtwvif->mac_addr,
5499 					     req->ssids[i].ssid,
5500 					     req->ssids[i].ssid_len,
5501 					     req->ie_len);
5502 		if (!skb)
5503 			return -ENOMEM;
5504 
5505 		ret = rtw89_append_probe_req_ie(rtwdev, rtwvif, skb, i);
5506 		kfree_skb(skb);
5507 
5508 		if (ret)
5509 			return ret;
5510 	}
5511 
5512 	return 0;
5513 }
5514 
5515 static int rtw89_update_6ghz_rnr_chan(struct rtw89_dev *rtwdev,
5516 				      struct cfg80211_scan_request *req,
5517 				      struct rtw89_mac_chinfo *ch_info)
5518 {
5519 	struct ieee80211_vif *vif = rtwdev->scan_info.scanning_vif;
5520 	struct list_head *pkt_list = rtwdev->scan_info.pkt_list;
5521 	struct rtw89_vif *rtwvif = vif_to_rtwvif_safe(vif);
5522 	struct ieee80211_scan_ies *ies = rtwvif->scan_ies;
5523 	struct cfg80211_scan_6ghz_params *params;
5524 	struct rtw89_pktofld_info *info, *tmp;
5525 	struct ieee80211_hdr *hdr;
5526 	struct sk_buff *skb;
5527 	bool found;
5528 	int ret = 0;
5529 	u8 i;
5530 
5531 	if (!req->n_6ghz_params)
5532 		return 0;
5533 
5534 	for (i = 0; i < req->n_6ghz_params; i++) {
5535 		params = &req->scan_6ghz_params[i];
5536 
5537 		if (req->channels[params->channel_idx]->hw_value !=
5538 		    ch_info->pri_ch)
5539 			continue;
5540 
5541 		found = false;
5542 		list_for_each_entry(tmp, &pkt_list[NL80211_BAND_6GHZ], list) {
5543 			if (ether_addr_equal(tmp->bssid, params->bssid)) {
5544 				found = true;
5545 				break;
5546 			}
5547 		}
5548 		if (found)
5549 			continue;
5550 
5551 		skb = ieee80211_probereq_get(rtwdev->hw, rtwvif->mac_addr,
5552 					     NULL, 0, req->ie_len);
5553 		skb_put_data(skb, ies->ies[NL80211_BAND_6GHZ], ies->len[NL80211_BAND_6GHZ]);
5554 		skb_put_data(skb, ies->common_ies, ies->common_ie_len);
5555 		hdr = (struct ieee80211_hdr *)skb->data;
5556 		ether_addr_copy(hdr->addr3, params->bssid);
5557 
5558 		info = kzalloc(sizeof(*info), GFP_KERNEL);
5559 		if (!info) {
5560 			ret = -ENOMEM;
5561 			kfree_skb(skb);
5562 			goto out;
5563 		}
5564 
5565 		ret = rtw89_fw_h2c_add_pkt_offload(rtwdev, &info->id, skb);
5566 		if (ret) {
5567 			kfree_skb(skb);
5568 			kfree(info);
5569 			goto out;
5570 		}
5571 
5572 		ether_addr_copy(info->bssid, params->bssid);
5573 		info->channel_6ghz = req->channels[params->channel_idx]->hw_value;
5574 		list_add_tail(&info->list, &rtwdev->scan_info.pkt_list[NL80211_BAND_6GHZ]);
5575 
5576 		ch_info->tx_pkt = true;
5577 		ch_info->period = RTW89_CHANNEL_TIME_6G + RTW89_DWELL_TIME_6G;
5578 
5579 		kfree_skb(skb);
5580 	}
5581 
5582 out:
5583 	return ret;
5584 }
5585 
5586 static void rtw89_hw_scan_add_chan(struct rtw89_dev *rtwdev, int chan_type,
5587 				   int ssid_num,
5588 				   struct rtw89_mac_chinfo *ch_info)
5589 {
5590 	struct rtw89_hw_scan_info *scan_info = &rtwdev->scan_info;
5591 	struct ieee80211_vif *vif = rtwdev->scan_info.scanning_vif;
5592 	struct rtw89_vif *rtwvif = (struct rtw89_vif *)vif->drv_priv;
5593 	struct cfg80211_scan_request *req = rtwvif->scan_req;
5594 	struct rtw89_chan *op = &rtwdev->scan_info.op_chan;
5595 	struct rtw89_pktofld_info *info;
5596 	u8 band, probe_count = 0;
5597 	int ret;
5598 
5599 	ch_info->notify_action = RTW89_SCANOFLD_DEBUG_MASK;
5600 	ch_info->dfs_ch = chan_type == RTW89_CHAN_DFS;
5601 	ch_info->bw = RTW89_SCAN_WIDTH;
5602 	ch_info->tx_pkt = true;
5603 	ch_info->cfg_tx_pwr = false;
5604 	ch_info->tx_pwr_idx = 0;
5605 	ch_info->tx_null = false;
5606 	ch_info->pause_data = false;
5607 	ch_info->probe_id = RTW89_SCANOFLD_PKT_NONE;
5608 
5609 	if (ch_info->ch_band == RTW89_BAND_6G) {
5610 		if ((ssid_num == 1 && req->ssids[0].ssid_len == 0) ||
5611 		    !ch_info->is_psc) {
5612 			ch_info->tx_pkt = false;
5613 			if (!req->duration_mandatory)
5614 				ch_info->period -= RTW89_DWELL_TIME_6G;
5615 		}
5616 	}
5617 
5618 	ret = rtw89_update_6ghz_rnr_chan(rtwdev, req, ch_info);
5619 	if (ret)
5620 		rtw89_warn(rtwdev, "RNR fails: %d\n", ret);
5621 
5622 	if (ssid_num) {
5623 		band = rtw89_hw_to_nl80211_band(ch_info->ch_band);
5624 
5625 		list_for_each_entry(info, &scan_info->pkt_list[band], list) {
5626 			if (info->channel_6ghz &&
5627 			    ch_info->pri_ch != info->channel_6ghz)
5628 				continue;
5629 			else if (info->channel_6ghz && probe_count != 0)
5630 				ch_info->period += RTW89_CHANNEL_TIME_6G;
5631 			ch_info->pkt_id[probe_count++] = info->id;
5632 			if (probe_count >= RTW89_SCANOFLD_MAX_SSID)
5633 				break;
5634 		}
5635 		ch_info->num_pkt = probe_count;
5636 	}
5637 
5638 	switch (chan_type) {
5639 	case RTW89_CHAN_OPERATE:
5640 		ch_info->central_ch = op->channel;
5641 		ch_info->pri_ch = op->primary_channel;
5642 		ch_info->ch_band = op->band_type;
5643 		ch_info->bw = op->band_width;
5644 		ch_info->tx_null = true;
5645 		ch_info->num_pkt = 0;
5646 		break;
5647 	case RTW89_CHAN_DFS:
5648 		if (ch_info->ch_band != RTW89_BAND_6G)
5649 			ch_info->period = max_t(u8, ch_info->period,
5650 						RTW89_DFS_CHAN_TIME);
5651 		ch_info->dwell_time = RTW89_DWELL_TIME;
5652 		break;
5653 	case RTW89_CHAN_ACTIVE:
5654 		break;
5655 	default:
5656 		rtw89_err(rtwdev, "Channel type out of bound\n");
5657 	}
5658 }
5659 
5660 static void rtw89_hw_scan_add_chan_be(struct rtw89_dev *rtwdev, int chan_type,
5661 				      int ssid_num,
5662 				      struct rtw89_mac_chinfo_be *ch_info)
5663 {
5664 	struct rtw89_hw_scan_info *scan_info = &rtwdev->scan_info;
5665 	struct ieee80211_vif *vif = rtwdev->scan_info.scanning_vif;
5666 	struct rtw89_vif *rtwvif = (struct rtw89_vif *)vif->drv_priv;
5667 	struct cfg80211_scan_request *req = rtwvif->scan_req;
5668 	struct rtw89_pktofld_info *info;
5669 	u8 band, probe_count = 0, i;
5670 
5671 	ch_info->notify_action = RTW89_SCANOFLD_DEBUG_MASK;
5672 	ch_info->dfs_ch = chan_type == RTW89_CHAN_DFS;
5673 	ch_info->bw = RTW89_SCAN_WIDTH;
5674 	ch_info->tx_null = false;
5675 	ch_info->pause_data = false;
5676 	ch_info->probe_id = RTW89_SCANOFLD_PKT_NONE;
5677 
5678 	if (ssid_num) {
5679 		band = rtw89_hw_to_nl80211_band(ch_info->ch_band);
5680 
5681 		list_for_each_entry(info, &scan_info->pkt_list[band], list) {
5682 			if (info->channel_6ghz &&
5683 			    ch_info->pri_ch != info->channel_6ghz)
5684 				continue;
5685 			ch_info->pkt_id[probe_count++] = info->id;
5686 			if (probe_count >= RTW89_SCANOFLD_MAX_SSID)
5687 				break;
5688 		}
5689 	}
5690 
5691 	if (ch_info->ch_band == RTW89_BAND_6G) {
5692 		if ((ssid_num == 1 && req->ssids[0].ssid_len == 0) ||
5693 		    !ch_info->is_psc) {
5694 			ch_info->probe_id = RTW89_SCANOFLD_PKT_NONE;
5695 			if (!req->duration_mandatory)
5696 				ch_info->period -= RTW89_DWELL_TIME_6G;
5697 		}
5698 	}
5699 
5700 	for (i = probe_count; i < RTW89_SCANOFLD_MAX_SSID; i++)
5701 		ch_info->pkt_id[i] = RTW89_SCANOFLD_PKT_NONE;
5702 
5703 	switch (chan_type) {
5704 	case RTW89_CHAN_DFS:
5705 		if (ch_info->ch_band != RTW89_BAND_6G)
5706 			ch_info->period =
5707 				max_t(u8, ch_info->period, RTW89_DFS_CHAN_TIME);
5708 		ch_info->dwell_time = RTW89_DWELL_TIME;
5709 		break;
5710 	case RTW89_CHAN_ACTIVE:
5711 		break;
5712 	default:
5713 		rtw89_warn(rtwdev, "Channel type out of bound\n");
5714 		break;
5715 	}
5716 }
5717 
5718 int rtw89_hw_scan_add_chan_list(struct rtw89_dev *rtwdev,
5719 				struct rtw89_vif *rtwvif, bool connected)
5720 {
5721 	struct cfg80211_scan_request *req = rtwvif->scan_req;
5722 	struct rtw89_mac_chinfo	*ch_info, *tmp;
5723 	struct ieee80211_channel *channel;
5724 	struct list_head chan_list;
5725 	bool random_seq = req->flags & NL80211_SCAN_FLAG_RANDOM_SN;
5726 	int list_len, off_chan_time = 0;
5727 	enum rtw89_chan_type type;
5728 	int ret = 0;
5729 	u32 idx;
5730 
5731 	INIT_LIST_HEAD(&chan_list);
5732 	for (idx = rtwdev->scan_info.last_chan_idx, list_len = 0;
5733 	     idx < req->n_channels && list_len < RTW89_SCAN_LIST_LIMIT;
5734 	     idx++, list_len++) {
5735 		channel = req->channels[idx];
5736 		ch_info = kzalloc(sizeof(*ch_info), GFP_KERNEL);
5737 		if (!ch_info) {
5738 			ret = -ENOMEM;
5739 			goto out;
5740 		}
5741 
5742 		if (req->duration_mandatory)
5743 			ch_info->period = req->duration;
5744 		else if (channel->band == NL80211_BAND_6GHZ)
5745 			ch_info->period = RTW89_CHANNEL_TIME_6G +
5746 					  RTW89_DWELL_TIME_6G;
5747 		else
5748 			ch_info->period = RTW89_CHANNEL_TIME;
5749 
5750 		ch_info->ch_band = rtw89_nl80211_to_hw_band(channel->band);
5751 		ch_info->central_ch = channel->hw_value;
5752 		ch_info->pri_ch = channel->hw_value;
5753 		ch_info->rand_seq_num = random_seq;
5754 		ch_info->is_psc = cfg80211_channel_is_psc(channel);
5755 
5756 		if (channel->flags &
5757 		    (IEEE80211_CHAN_RADAR | IEEE80211_CHAN_NO_IR))
5758 			type = RTW89_CHAN_DFS;
5759 		else
5760 			type = RTW89_CHAN_ACTIVE;
5761 		rtw89_hw_scan_add_chan(rtwdev, type, req->n_ssids, ch_info);
5762 
5763 		if (connected &&
5764 		    off_chan_time + ch_info->period > RTW89_OFF_CHAN_TIME) {
5765 			tmp = kzalloc(sizeof(*tmp), GFP_KERNEL);
5766 			if (!tmp) {
5767 				ret = -ENOMEM;
5768 				kfree(ch_info);
5769 				goto out;
5770 			}
5771 
5772 			type = RTW89_CHAN_OPERATE;
5773 			tmp->period = req->duration_mandatory ?
5774 				      req->duration : RTW89_CHANNEL_TIME;
5775 			rtw89_hw_scan_add_chan(rtwdev, type, 0, tmp);
5776 			list_add_tail(&tmp->list, &chan_list);
5777 			off_chan_time = 0;
5778 			list_len++;
5779 		}
5780 		list_add_tail(&ch_info->list, &chan_list);
5781 		off_chan_time += ch_info->period;
5782 	}
5783 	rtwdev->scan_info.last_chan_idx = idx;
5784 	ret = rtw89_fw_h2c_scan_list_offload(rtwdev, list_len, &chan_list);
5785 
5786 out:
5787 	list_for_each_entry_safe(ch_info, tmp, &chan_list, list) {
5788 		list_del(&ch_info->list);
5789 		kfree(ch_info);
5790 	}
5791 
5792 	return ret;
5793 }
5794 
5795 int rtw89_hw_scan_add_chan_list_be(struct rtw89_dev *rtwdev,
5796 				   struct rtw89_vif *rtwvif, bool connected)
5797 {
5798 	struct cfg80211_scan_request *req = rtwvif->scan_req;
5799 	struct rtw89_mac_chinfo_be *ch_info, *tmp;
5800 	struct ieee80211_channel *channel;
5801 	struct list_head chan_list;
5802 	enum rtw89_chan_type type;
5803 	int list_len, ret;
5804 	bool random_seq;
5805 	u32 idx;
5806 
5807 	random_seq = !!(req->flags & NL80211_SCAN_FLAG_RANDOM_SN);
5808 	INIT_LIST_HEAD(&chan_list);
5809 
5810 	for (idx = rtwdev->scan_info.last_chan_idx, list_len = 0;
5811 	     idx < req->n_channels && list_len < RTW89_SCAN_LIST_LIMIT;
5812 	     idx++, list_len++) {
5813 		channel = req->channels[idx];
5814 		ch_info = kzalloc(sizeof(*ch_info), GFP_KERNEL);
5815 		if (!ch_info) {
5816 			ret = -ENOMEM;
5817 			goto out;
5818 		}
5819 
5820 		if (req->duration_mandatory)
5821 			ch_info->period = req->duration;
5822 		else if (channel->band == NL80211_BAND_6GHZ)
5823 			ch_info->period = RTW89_CHANNEL_TIME_6G + RTW89_DWELL_TIME_6G;
5824 		else
5825 			ch_info->period = RTW89_CHANNEL_TIME;
5826 
5827 		ch_info->ch_band = rtw89_nl80211_to_hw_band(channel->band);
5828 		ch_info->central_ch = channel->hw_value;
5829 		ch_info->pri_ch = channel->hw_value;
5830 		ch_info->rand_seq_num = random_seq;
5831 		ch_info->is_psc = cfg80211_channel_is_psc(channel);
5832 
5833 		if (channel->flags & (IEEE80211_CHAN_RADAR | IEEE80211_CHAN_NO_IR))
5834 			type = RTW89_CHAN_DFS;
5835 		else
5836 			type = RTW89_CHAN_ACTIVE;
5837 		rtw89_hw_scan_add_chan_be(rtwdev, type, req->n_ssids, ch_info);
5838 
5839 		list_add_tail(&ch_info->list, &chan_list);
5840 	}
5841 
5842 	rtwdev->scan_info.last_chan_idx = idx;
5843 	ret = rtw89_fw_h2c_scan_list_offload_be(rtwdev, list_len, &chan_list);
5844 
5845 out:
5846 	list_for_each_entry_safe(ch_info, tmp, &chan_list, list) {
5847 		list_del(&ch_info->list);
5848 		kfree(ch_info);
5849 	}
5850 
5851 	return ret;
5852 }
5853 
5854 static int rtw89_hw_scan_prehandle(struct rtw89_dev *rtwdev,
5855 				   struct rtw89_vif *rtwvif, bool connected)
5856 {
5857 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
5858 	int ret;
5859 
5860 	ret = rtw89_hw_scan_update_probe_req(rtwdev, rtwvif);
5861 	if (ret) {
5862 		rtw89_err(rtwdev, "Update probe request failed\n");
5863 		goto out;
5864 	}
5865 	ret = mac->add_chan_list(rtwdev, rtwvif, connected);
5866 out:
5867 	return ret;
5868 }
5869 
5870 void rtw89_hw_scan_start(struct rtw89_dev *rtwdev, struct ieee80211_vif *vif,
5871 			 struct ieee80211_scan_request *scan_req)
5872 {
5873 	struct rtw89_vif *rtwvif = (struct rtw89_vif *)vif->drv_priv;
5874 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
5875 	struct cfg80211_scan_request *req = &scan_req->req;
5876 	u32 rx_fltr = rtwdev->hal.rx_fltr;
5877 	u8 mac_addr[ETH_ALEN];
5878 
5879 	rtw89_get_channel(rtwdev, rtwvif, &rtwdev->scan_info.op_chan);
5880 	rtwdev->scan_info.scanning_vif = vif;
5881 	rtwdev->scan_info.last_chan_idx = 0;
5882 	rtwdev->scan_info.abort = false;
5883 	rtwvif->scan_ies = &scan_req->ies;
5884 	rtwvif->scan_req = req;
5885 	ieee80211_stop_queues(rtwdev->hw);
5886 	rtw89_mac_port_cfg_rx_sync(rtwdev, rtwvif, false);
5887 
5888 	if (req->flags & NL80211_SCAN_FLAG_RANDOM_ADDR)
5889 		get_random_mask_addr(mac_addr, req->mac_addr,
5890 				     req->mac_addr_mask);
5891 	else
5892 		ether_addr_copy(mac_addr, vif->addr);
5893 	rtw89_core_scan_start(rtwdev, rtwvif, mac_addr, true);
5894 
5895 	rx_fltr &= ~B_AX_A_BCN_CHK_EN;
5896 	rx_fltr &= ~B_AX_A_BC;
5897 	rx_fltr &= ~B_AX_A_A1_MATCH;
5898 	rtw89_write32_mask(rtwdev,
5899 			   rtw89_mac_reg_by_idx(rtwdev, mac->rx_fltr, RTW89_MAC_0),
5900 			   B_AX_RX_FLTR_CFG_MASK,
5901 			   rx_fltr);
5902 
5903 	rtw89_chanctx_pause(rtwdev, RTW89_CHANCTX_PAUSE_REASON_HW_SCAN);
5904 }
5905 
5906 void rtw89_hw_scan_complete(struct rtw89_dev *rtwdev, struct ieee80211_vif *vif,
5907 			    bool aborted)
5908 {
5909 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
5910 	struct rtw89_hw_scan_info *scan_info = &rtwdev->scan_info;
5911 	struct rtw89_vif *rtwvif = vif_to_rtwvif_safe(vif);
5912 	struct cfg80211_scan_info info = {
5913 		.aborted = aborted,
5914 	};
5915 
5916 	if (!vif)
5917 		return;
5918 
5919 	rtw89_write32_mask(rtwdev,
5920 			   rtw89_mac_reg_by_idx(rtwdev, mac->rx_fltr, RTW89_MAC_0),
5921 			   B_AX_RX_FLTR_CFG_MASK,
5922 			   rtwdev->hal.rx_fltr);
5923 
5924 	rtw89_core_scan_complete(rtwdev, vif, true);
5925 	ieee80211_scan_completed(rtwdev->hw, &info);
5926 	ieee80211_wake_queues(rtwdev->hw);
5927 	rtw89_mac_port_cfg_rx_sync(rtwdev, rtwvif, true);
5928 	rtw89_mac_enable_beacon_for_ap_vifs(rtwdev, true);
5929 
5930 	rtw89_release_pkt_list(rtwdev);
5931 	rtwvif->scan_req = NULL;
5932 	rtwvif->scan_ies = NULL;
5933 	scan_info->last_chan_idx = 0;
5934 	scan_info->scanning_vif = NULL;
5935 	scan_info->abort = false;
5936 
5937 	rtw89_chanctx_proceed(rtwdev);
5938 }
5939 
5940 void rtw89_hw_scan_abort(struct rtw89_dev *rtwdev, struct ieee80211_vif *vif)
5941 {
5942 	struct rtw89_hw_scan_info *scan_info = &rtwdev->scan_info;
5943 	int ret;
5944 
5945 	scan_info->abort = true;
5946 
5947 	ret = rtw89_hw_scan_offload(rtwdev, vif, false);
5948 	if (ret)
5949 		rtw89_hw_scan_complete(rtwdev, vif, true);
5950 }
5951 
5952 static bool rtw89_is_any_vif_connected_or_connecting(struct rtw89_dev *rtwdev)
5953 {
5954 	struct rtw89_vif *rtwvif;
5955 
5956 	rtw89_for_each_rtwvif(rtwdev, rtwvif) {
5957 		/* This variable implies connected or during attempt to connect */
5958 		if (!is_zero_ether_addr(rtwvif->bssid))
5959 			return true;
5960 	}
5961 
5962 	return false;
5963 }
5964 
5965 int rtw89_hw_scan_offload(struct rtw89_dev *rtwdev, struct ieee80211_vif *vif,
5966 			  bool enable)
5967 {
5968 	const struct rtw89_mac_gen_def *mac = rtwdev->chip->mac_def;
5969 	struct rtw89_scan_option opt = {0};
5970 	struct rtw89_vif *rtwvif;
5971 	bool connected;
5972 	int ret = 0;
5973 
5974 	rtwvif = vif ? (struct rtw89_vif *)vif->drv_priv : NULL;
5975 	if (!rtwvif)
5976 		return -EINVAL;
5977 
5978 	connected = rtw89_is_any_vif_connected_or_connecting(rtwdev);
5979 	opt.enable = enable;
5980 	opt.target_ch_mode = connected;
5981 	if (enable) {
5982 		ret = rtw89_hw_scan_prehandle(rtwdev, rtwvif, connected);
5983 		if (ret)
5984 			goto out;
5985 	}
5986 
5987 	if (rtwdev->chip->chip_gen == RTW89_CHIP_BE) {
5988 		opt.operation = enable ? RTW89_SCAN_OP_START : RTW89_SCAN_OP_STOP;
5989 		opt.scan_mode = RTW89_SCAN_MODE_SA;
5990 		opt.band = RTW89_PHY_0;
5991 		opt.num_macc_role = 0;
5992 		opt.mlo_mode = rtwdev->mlo_dbcc_mode;
5993 		opt.num_opch = connected ? 1 : 0;
5994 		opt.opch_end = connected ? 0 : RTW89_CHAN_INVALID;
5995 	}
5996 
5997 	ret = mac->scan_offload(rtwdev, &opt, rtwvif);
5998 out:
5999 	return ret;
6000 }
6001 
6002 #define H2C_FW_CPU_EXCEPTION_LEN 4
6003 #define H2C_FW_CPU_EXCEPTION_TYPE_DEF 0x5566
6004 int rtw89_fw_h2c_trigger_cpu_exception(struct rtw89_dev *rtwdev)
6005 {
6006 	struct sk_buff *skb;
6007 	int ret;
6008 
6009 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_FW_CPU_EXCEPTION_LEN);
6010 	if (!skb) {
6011 		rtw89_err(rtwdev,
6012 			  "failed to alloc skb for fw cpu exception\n");
6013 		return -ENOMEM;
6014 	}
6015 
6016 	skb_put(skb, H2C_FW_CPU_EXCEPTION_LEN);
6017 	RTW89_SET_FWCMD_CPU_EXCEPTION_TYPE(skb->data,
6018 					   H2C_FW_CPU_EXCEPTION_TYPE_DEF);
6019 
6020 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6021 			      H2C_CAT_TEST,
6022 			      H2C_CL_FW_STATUS_TEST,
6023 			      H2C_FUNC_CPU_EXCEPTION, 0, 0,
6024 			      H2C_FW_CPU_EXCEPTION_LEN);
6025 
6026 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6027 	if (ret) {
6028 		rtw89_err(rtwdev, "failed to send h2c\n");
6029 		goto fail;
6030 	}
6031 
6032 	return 0;
6033 
6034 fail:
6035 	dev_kfree_skb_any(skb);
6036 	return ret;
6037 }
6038 
6039 #define H2C_PKT_DROP_LEN 24
6040 int rtw89_fw_h2c_pkt_drop(struct rtw89_dev *rtwdev,
6041 			  const struct rtw89_pkt_drop_params *params)
6042 {
6043 	struct sk_buff *skb;
6044 	int ret;
6045 
6046 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_PKT_DROP_LEN);
6047 	if (!skb) {
6048 		rtw89_err(rtwdev,
6049 			  "failed to alloc skb for packet drop\n");
6050 		return -ENOMEM;
6051 	}
6052 
6053 	switch (params->sel) {
6054 	case RTW89_PKT_DROP_SEL_MACID_BE_ONCE:
6055 	case RTW89_PKT_DROP_SEL_MACID_BK_ONCE:
6056 	case RTW89_PKT_DROP_SEL_MACID_VI_ONCE:
6057 	case RTW89_PKT_DROP_SEL_MACID_VO_ONCE:
6058 	case RTW89_PKT_DROP_SEL_BAND_ONCE:
6059 		break;
6060 	default:
6061 		rtw89_debug(rtwdev, RTW89_DBG_FW,
6062 			    "H2C of pkt drop might not fully support sel: %d yet\n",
6063 			    params->sel);
6064 		break;
6065 	}
6066 
6067 	skb_put(skb, H2C_PKT_DROP_LEN);
6068 	RTW89_SET_FWCMD_PKT_DROP_SEL(skb->data, params->sel);
6069 	RTW89_SET_FWCMD_PKT_DROP_MACID(skb->data, params->macid);
6070 	RTW89_SET_FWCMD_PKT_DROP_BAND(skb->data, params->mac_band);
6071 	RTW89_SET_FWCMD_PKT_DROP_PORT(skb->data, params->port);
6072 	RTW89_SET_FWCMD_PKT_DROP_MBSSID(skb->data, params->mbssid);
6073 	RTW89_SET_FWCMD_PKT_DROP_ROLE_A_INFO_TF_TRS(skb->data, params->tf_trs);
6074 	RTW89_SET_FWCMD_PKT_DROP_MACID_BAND_SEL_0(skb->data,
6075 						  params->macid_band_sel[0]);
6076 	RTW89_SET_FWCMD_PKT_DROP_MACID_BAND_SEL_1(skb->data,
6077 						  params->macid_band_sel[1]);
6078 	RTW89_SET_FWCMD_PKT_DROP_MACID_BAND_SEL_2(skb->data,
6079 						  params->macid_band_sel[2]);
6080 	RTW89_SET_FWCMD_PKT_DROP_MACID_BAND_SEL_3(skb->data,
6081 						  params->macid_band_sel[3]);
6082 
6083 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6084 			      H2C_CAT_MAC,
6085 			      H2C_CL_MAC_FW_OFLD,
6086 			      H2C_FUNC_PKT_DROP, 0, 0,
6087 			      H2C_PKT_DROP_LEN);
6088 
6089 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6090 	if (ret) {
6091 		rtw89_err(rtwdev, "failed to send h2c\n");
6092 		goto fail;
6093 	}
6094 
6095 	return 0;
6096 
6097 fail:
6098 	dev_kfree_skb_any(skb);
6099 	return ret;
6100 }
6101 
6102 #define H2C_KEEP_ALIVE_LEN 4
6103 int rtw89_fw_h2c_keep_alive(struct rtw89_dev *rtwdev, struct rtw89_vif *rtwvif,
6104 			    bool enable)
6105 {
6106 	struct sk_buff *skb;
6107 	u8 pkt_id = 0;
6108 	int ret;
6109 
6110 	if (enable) {
6111 		ret = rtw89_fw_h2c_add_general_pkt(rtwdev, rtwvif,
6112 						   RTW89_PKT_OFLD_TYPE_NULL_DATA,
6113 						   &pkt_id);
6114 		if (ret)
6115 			return -EPERM;
6116 	}
6117 
6118 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_KEEP_ALIVE_LEN);
6119 	if (!skb) {
6120 		rtw89_err(rtwdev, "failed to alloc skb for keep alive\n");
6121 		return -ENOMEM;
6122 	}
6123 
6124 	skb_put(skb, H2C_KEEP_ALIVE_LEN);
6125 
6126 	RTW89_SET_KEEP_ALIVE_ENABLE(skb->data, enable);
6127 	RTW89_SET_KEEP_ALIVE_PKT_NULL_ID(skb->data, pkt_id);
6128 	RTW89_SET_KEEP_ALIVE_PERIOD(skb->data, 5);
6129 	RTW89_SET_KEEP_ALIVE_MACID(skb->data, rtwvif->mac_id);
6130 
6131 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6132 			      H2C_CAT_MAC,
6133 			      H2C_CL_MAC_WOW,
6134 			      H2C_FUNC_KEEP_ALIVE, 0, 1,
6135 			      H2C_KEEP_ALIVE_LEN);
6136 
6137 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6138 	if (ret) {
6139 		rtw89_err(rtwdev, "failed to send h2c\n");
6140 		goto fail;
6141 	}
6142 
6143 	return 0;
6144 
6145 fail:
6146 	dev_kfree_skb_any(skb);
6147 
6148 	return ret;
6149 }
6150 
6151 #define H2C_DISCONNECT_DETECT_LEN 8
6152 int rtw89_fw_h2c_disconnect_detect(struct rtw89_dev *rtwdev,
6153 				   struct rtw89_vif *rtwvif, bool enable)
6154 {
6155 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
6156 	struct sk_buff *skb;
6157 	u8 macid = rtwvif->mac_id;
6158 	int ret;
6159 
6160 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_DISCONNECT_DETECT_LEN);
6161 	if (!skb) {
6162 		rtw89_err(rtwdev, "failed to alloc skb for keep alive\n");
6163 		return -ENOMEM;
6164 	}
6165 
6166 	skb_put(skb, H2C_DISCONNECT_DETECT_LEN);
6167 
6168 	if (test_bit(RTW89_WOW_FLAG_EN_DISCONNECT, rtw_wow->flags)) {
6169 		RTW89_SET_DISCONNECT_DETECT_ENABLE(skb->data, enable);
6170 		RTW89_SET_DISCONNECT_DETECT_DISCONNECT(skb->data, !enable);
6171 		RTW89_SET_DISCONNECT_DETECT_MAC_ID(skb->data, macid);
6172 		RTW89_SET_DISCONNECT_DETECT_CHECK_PERIOD(skb->data, 100);
6173 		RTW89_SET_DISCONNECT_DETECT_TRY_PKT_COUNT(skb->data, 5);
6174 	}
6175 
6176 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6177 			      H2C_CAT_MAC,
6178 			      H2C_CL_MAC_WOW,
6179 			      H2C_FUNC_DISCONNECT_DETECT, 0, 1,
6180 			      H2C_DISCONNECT_DETECT_LEN);
6181 
6182 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6183 	if (ret) {
6184 		rtw89_err(rtwdev, "failed to send h2c\n");
6185 		goto fail;
6186 	}
6187 
6188 	return 0;
6189 
6190 fail:
6191 	dev_kfree_skb_any(skb);
6192 
6193 	return ret;
6194 }
6195 
6196 #define H2C_WOW_GLOBAL_LEN 8
6197 int rtw89_fw_h2c_wow_global(struct rtw89_dev *rtwdev, struct rtw89_vif *rtwvif,
6198 			    bool enable)
6199 {
6200 	struct sk_buff *skb;
6201 	u8 macid = rtwvif->mac_id;
6202 	int ret;
6203 
6204 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_WOW_GLOBAL_LEN);
6205 	if (!skb) {
6206 		rtw89_err(rtwdev, "failed to alloc skb for keep alive\n");
6207 		return -ENOMEM;
6208 	}
6209 
6210 	skb_put(skb, H2C_WOW_GLOBAL_LEN);
6211 
6212 	RTW89_SET_WOW_GLOBAL_ENABLE(skb->data, enable);
6213 	RTW89_SET_WOW_GLOBAL_MAC_ID(skb->data, macid);
6214 
6215 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6216 			      H2C_CAT_MAC,
6217 			      H2C_CL_MAC_WOW,
6218 			      H2C_FUNC_WOW_GLOBAL, 0, 1,
6219 			      H2C_WOW_GLOBAL_LEN);
6220 
6221 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6222 	if (ret) {
6223 		rtw89_err(rtwdev, "failed to send h2c\n");
6224 		goto fail;
6225 	}
6226 
6227 	return 0;
6228 
6229 fail:
6230 	dev_kfree_skb_any(skb);
6231 
6232 	return ret;
6233 }
6234 
6235 #define H2C_WAKEUP_CTRL_LEN 4
6236 int rtw89_fw_h2c_wow_wakeup_ctrl(struct rtw89_dev *rtwdev,
6237 				 struct rtw89_vif *rtwvif,
6238 				 bool enable)
6239 {
6240 	struct rtw89_wow_param *rtw_wow = &rtwdev->wow;
6241 	struct sk_buff *skb;
6242 	u8 macid = rtwvif->mac_id;
6243 	int ret;
6244 
6245 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_WAKEUP_CTRL_LEN);
6246 	if (!skb) {
6247 		rtw89_err(rtwdev, "failed to alloc skb for keep alive\n");
6248 		return -ENOMEM;
6249 	}
6250 
6251 	skb_put(skb, H2C_WAKEUP_CTRL_LEN);
6252 
6253 	if (rtw_wow->pattern_cnt)
6254 		RTW89_SET_WOW_WAKEUP_CTRL_PATTERN_MATCH_ENABLE(skb->data, enable);
6255 	if (test_bit(RTW89_WOW_FLAG_EN_MAGIC_PKT, rtw_wow->flags))
6256 		RTW89_SET_WOW_WAKEUP_CTRL_MAGIC_ENABLE(skb->data, enable);
6257 	if (test_bit(RTW89_WOW_FLAG_EN_DISCONNECT, rtw_wow->flags))
6258 		RTW89_SET_WOW_WAKEUP_CTRL_DEAUTH_ENABLE(skb->data, enable);
6259 
6260 	RTW89_SET_WOW_WAKEUP_CTRL_MAC_ID(skb->data, macid);
6261 
6262 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6263 			      H2C_CAT_MAC,
6264 			      H2C_CL_MAC_WOW,
6265 			      H2C_FUNC_WAKEUP_CTRL, 0, 1,
6266 			      H2C_WAKEUP_CTRL_LEN);
6267 
6268 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6269 	if (ret) {
6270 		rtw89_err(rtwdev, "failed to send h2c\n");
6271 		goto fail;
6272 	}
6273 
6274 	return 0;
6275 
6276 fail:
6277 	dev_kfree_skb_any(skb);
6278 
6279 	return ret;
6280 }
6281 
6282 #define H2C_WOW_CAM_UPD_LEN 24
6283 int rtw89_fw_wow_cam_update(struct rtw89_dev *rtwdev,
6284 			    struct rtw89_wow_cam_info *cam_info)
6285 {
6286 	struct sk_buff *skb;
6287 	int ret;
6288 
6289 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_WOW_CAM_UPD_LEN);
6290 	if (!skb) {
6291 		rtw89_err(rtwdev, "failed to alloc skb for keep alive\n");
6292 		return -ENOMEM;
6293 	}
6294 
6295 	skb_put(skb, H2C_WOW_CAM_UPD_LEN);
6296 
6297 	RTW89_SET_WOW_CAM_UPD_R_W(skb->data, cam_info->r_w);
6298 	RTW89_SET_WOW_CAM_UPD_IDX(skb->data, cam_info->idx);
6299 	if (cam_info->valid) {
6300 		RTW89_SET_WOW_CAM_UPD_WKFM1(skb->data, cam_info->mask[0]);
6301 		RTW89_SET_WOW_CAM_UPD_WKFM2(skb->data, cam_info->mask[1]);
6302 		RTW89_SET_WOW_CAM_UPD_WKFM3(skb->data, cam_info->mask[2]);
6303 		RTW89_SET_WOW_CAM_UPD_WKFM4(skb->data, cam_info->mask[3]);
6304 		RTW89_SET_WOW_CAM_UPD_CRC(skb->data, cam_info->crc);
6305 		RTW89_SET_WOW_CAM_UPD_NEGATIVE_PATTERN_MATCH(skb->data,
6306 							     cam_info->negative_pattern_match);
6307 		RTW89_SET_WOW_CAM_UPD_SKIP_MAC_HDR(skb->data,
6308 						   cam_info->skip_mac_hdr);
6309 		RTW89_SET_WOW_CAM_UPD_UC(skb->data, cam_info->uc);
6310 		RTW89_SET_WOW_CAM_UPD_MC(skb->data, cam_info->mc);
6311 		RTW89_SET_WOW_CAM_UPD_BC(skb->data, cam_info->bc);
6312 	}
6313 	RTW89_SET_WOW_CAM_UPD_VALID(skb->data, cam_info->valid);
6314 
6315 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6316 			      H2C_CAT_MAC,
6317 			      H2C_CL_MAC_WOW,
6318 			      H2C_FUNC_WOW_CAM_UPD, 0, 1,
6319 			      H2C_WOW_CAM_UPD_LEN);
6320 
6321 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6322 	if (ret) {
6323 		rtw89_err(rtwdev, "failed to send h2c\n");
6324 		goto fail;
6325 	}
6326 
6327 	return 0;
6328 fail:
6329 	dev_kfree_skb_any(skb);
6330 
6331 	return ret;
6332 }
6333 
6334 /* Return < 0, if failures happen during waiting for the condition.
6335  * Return 0, when waiting for the condition succeeds.
6336  * Return > 0, if the wait is considered unreachable due to driver/FW design,
6337  * where 1 means during SER.
6338  */
6339 static int rtw89_h2c_tx_and_wait(struct rtw89_dev *rtwdev, struct sk_buff *skb,
6340 				 struct rtw89_wait_info *wait, unsigned int cond)
6341 {
6342 	int ret;
6343 
6344 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6345 	if (ret) {
6346 		rtw89_err(rtwdev, "failed to send h2c\n");
6347 		dev_kfree_skb_any(skb);
6348 		return -EBUSY;
6349 	}
6350 
6351 	if (test_bit(RTW89_FLAG_SER_HANDLING, rtwdev->flags))
6352 		return 1;
6353 
6354 	return rtw89_wait_for_cond(wait, cond);
6355 }
6356 
6357 #define H2C_ADD_MCC_LEN 16
6358 int rtw89_fw_h2c_add_mcc(struct rtw89_dev *rtwdev,
6359 			 const struct rtw89_fw_mcc_add_req *p)
6360 {
6361 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6362 	struct sk_buff *skb;
6363 	unsigned int cond;
6364 
6365 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_ADD_MCC_LEN);
6366 	if (!skb) {
6367 		rtw89_err(rtwdev,
6368 			  "failed to alloc skb for add mcc\n");
6369 		return -ENOMEM;
6370 	}
6371 
6372 	skb_put(skb, H2C_ADD_MCC_LEN);
6373 	RTW89_SET_FWCMD_ADD_MCC_MACID(skb->data, p->macid);
6374 	RTW89_SET_FWCMD_ADD_MCC_CENTRAL_CH_SEG0(skb->data, p->central_ch_seg0);
6375 	RTW89_SET_FWCMD_ADD_MCC_CENTRAL_CH_SEG1(skb->data, p->central_ch_seg1);
6376 	RTW89_SET_FWCMD_ADD_MCC_PRIMARY_CH(skb->data, p->primary_ch);
6377 	RTW89_SET_FWCMD_ADD_MCC_BANDWIDTH(skb->data, p->bandwidth);
6378 	RTW89_SET_FWCMD_ADD_MCC_GROUP(skb->data, p->group);
6379 	RTW89_SET_FWCMD_ADD_MCC_C2H_RPT(skb->data, p->c2h_rpt);
6380 	RTW89_SET_FWCMD_ADD_MCC_DIS_TX_NULL(skb->data, p->dis_tx_null);
6381 	RTW89_SET_FWCMD_ADD_MCC_DIS_SW_RETRY(skb->data, p->dis_sw_retry);
6382 	RTW89_SET_FWCMD_ADD_MCC_IN_CURR_CH(skb->data, p->in_curr_ch);
6383 	RTW89_SET_FWCMD_ADD_MCC_SW_RETRY_COUNT(skb->data, p->sw_retry_count);
6384 	RTW89_SET_FWCMD_ADD_MCC_TX_NULL_EARLY(skb->data, p->tx_null_early);
6385 	RTW89_SET_FWCMD_ADD_MCC_BTC_IN_2G(skb->data, p->btc_in_2g);
6386 	RTW89_SET_FWCMD_ADD_MCC_PTA_EN(skb->data, p->pta_en);
6387 	RTW89_SET_FWCMD_ADD_MCC_RFK_BY_PASS(skb->data, p->rfk_by_pass);
6388 	RTW89_SET_FWCMD_ADD_MCC_CH_BAND_TYPE(skb->data, p->ch_band_type);
6389 	RTW89_SET_FWCMD_ADD_MCC_DURATION(skb->data, p->duration);
6390 	RTW89_SET_FWCMD_ADD_MCC_COURTESY_EN(skb->data, p->courtesy_en);
6391 	RTW89_SET_FWCMD_ADD_MCC_COURTESY_NUM(skb->data, p->courtesy_num);
6392 	RTW89_SET_FWCMD_ADD_MCC_COURTESY_TARGET(skb->data, p->courtesy_target);
6393 
6394 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6395 			      H2C_CAT_MAC,
6396 			      H2C_CL_MCC,
6397 			      H2C_FUNC_ADD_MCC, 0, 0,
6398 			      H2C_ADD_MCC_LEN);
6399 
6400 	cond = RTW89_MCC_WAIT_COND(p->group, H2C_FUNC_ADD_MCC);
6401 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6402 }
6403 
6404 #define H2C_START_MCC_LEN 12
6405 int rtw89_fw_h2c_start_mcc(struct rtw89_dev *rtwdev,
6406 			   const struct rtw89_fw_mcc_start_req *p)
6407 {
6408 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6409 	struct sk_buff *skb;
6410 	unsigned int cond;
6411 
6412 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_START_MCC_LEN);
6413 	if (!skb) {
6414 		rtw89_err(rtwdev,
6415 			  "failed to alloc skb for start mcc\n");
6416 		return -ENOMEM;
6417 	}
6418 
6419 	skb_put(skb, H2C_START_MCC_LEN);
6420 	RTW89_SET_FWCMD_START_MCC_GROUP(skb->data, p->group);
6421 	RTW89_SET_FWCMD_START_MCC_BTC_IN_GROUP(skb->data, p->btc_in_group);
6422 	RTW89_SET_FWCMD_START_MCC_OLD_GROUP_ACTION(skb->data, p->old_group_action);
6423 	RTW89_SET_FWCMD_START_MCC_OLD_GROUP(skb->data, p->old_group);
6424 	RTW89_SET_FWCMD_START_MCC_NOTIFY_CNT(skb->data, p->notify_cnt);
6425 	RTW89_SET_FWCMD_START_MCC_NOTIFY_RXDBG_EN(skb->data, p->notify_rxdbg_en);
6426 	RTW89_SET_FWCMD_START_MCC_MACID(skb->data, p->macid);
6427 	RTW89_SET_FWCMD_START_MCC_TSF_LOW(skb->data, p->tsf_low);
6428 	RTW89_SET_FWCMD_START_MCC_TSF_HIGH(skb->data, p->tsf_high);
6429 
6430 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6431 			      H2C_CAT_MAC,
6432 			      H2C_CL_MCC,
6433 			      H2C_FUNC_START_MCC, 0, 0,
6434 			      H2C_START_MCC_LEN);
6435 
6436 	cond = RTW89_MCC_WAIT_COND(p->group, H2C_FUNC_START_MCC);
6437 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6438 }
6439 
6440 #define H2C_STOP_MCC_LEN 4
6441 int rtw89_fw_h2c_stop_mcc(struct rtw89_dev *rtwdev, u8 group, u8 macid,
6442 			  bool prev_groups)
6443 {
6444 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6445 	struct sk_buff *skb;
6446 	unsigned int cond;
6447 
6448 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_STOP_MCC_LEN);
6449 	if (!skb) {
6450 		rtw89_err(rtwdev,
6451 			  "failed to alloc skb for stop mcc\n");
6452 		return -ENOMEM;
6453 	}
6454 
6455 	skb_put(skb, H2C_STOP_MCC_LEN);
6456 	RTW89_SET_FWCMD_STOP_MCC_MACID(skb->data, macid);
6457 	RTW89_SET_FWCMD_STOP_MCC_GROUP(skb->data, group);
6458 	RTW89_SET_FWCMD_STOP_MCC_PREV_GROUPS(skb->data, prev_groups);
6459 
6460 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6461 			      H2C_CAT_MAC,
6462 			      H2C_CL_MCC,
6463 			      H2C_FUNC_STOP_MCC, 0, 0,
6464 			      H2C_STOP_MCC_LEN);
6465 
6466 	cond = RTW89_MCC_WAIT_COND(group, H2C_FUNC_STOP_MCC);
6467 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6468 }
6469 
6470 #define H2C_DEL_MCC_GROUP_LEN 4
6471 int rtw89_fw_h2c_del_mcc_group(struct rtw89_dev *rtwdev, u8 group,
6472 			       bool prev_groups)
6473 {
6474 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6475 	struct sk_buff *skb;
6476 	unsigned int cond;
6477 
6478 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_DEL_MCC_GROUP_LEN);
6479 	if (!skb) {
6480 		rtw89_err(rtwdev,
6481 			  "failed to alloc skb for del mcc group\n");
6482 		return -ENOMEM;
6483 	}
6484 
6485 	skb_put(skb, H2C_DEL_MCC_GROUP_LEN);
6486 	RTW89_SET_FWCMD_DEL_MCC_GROUP_GROUP(skb->data, group);
6487 	RTW89_SET_FWCMD_DEL_MCC_GROUP_PREV_GROUPS(skb->data, prev_groups);
6488 
6489 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6490 			      H2C_CAT_MAC,
6491 			      H2C_CL_MCC,
6492 			      H2C_FUNC_DEL_MCC_GROUP, 0, 0,
6493 			      H2C_DEL_MCC_GROUP_LEN);
6494 
6495 	cond = RTW89_MCC_WAIT_COND(group, H2C_FUNC_DEL_MCC_GROUP);
6496 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6497 }
6498 
6499 #define H2C_RESET_MCC_GROUP_LEN 4
6500 int rtw89_fw_h2c_reset_mcc_group(struct rtw89_dev *rtwdev, u8 group)
6501 {
6502 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6503 	struct sk_buff *skb;
6504 	unsigned int cond;
6505 
6506 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_RESET_MCC_GROUP_LEN);
6507 	if (!skb) {
6508 		rtw89_err(rtwdev,
6509 			  "failed to alloc skb for reset mcc group\n");
6510 		return -ENOMEM;
6511 	}
6512 
6513 	skb_put(skb, H2C_RESET_MCC_GROUP_LEN);
6514 	RTW89_SET_FWCMD_RESET_MCC_GROUP_GROUP(skb->data, group);
6515 
6516 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6517 			      H2C_CAT_MAC,
6518 			      H2C_CL_MCC,
6519 			      H2C_FUNC_RESET_MCC_GROUP, 0, 0,
6520 			      H2C_RESET_MCC_GROUP_LEN);
6521 
6522 	cond = RTW89_MCC_WAIT_COND(group, H2C_FUNC_RESET_MCC_GROUP);
6523 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6524 }
6525 
6526 #define H2C_MCC_REQ_TSF_LEN 4
6527 int rtw89_fw_h2c_mcc_req_tsf(struct rtw89_dev *rtwdev,
6528 			     const struct rtw89_fw_mcc_tsf_req *req,
6529 			     struct rtw89_mac_mcc_tsf_rpt *rpt)
6530 {
6531 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6532 	struct rtw89_mac_mcc_tsf_rpt *tmp;
6533 	struct sk_buff *skb;
6534 	unsigned int cond;
6535 	int ret;
6536 
6537 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_MCC_REQ_TSF_LEN);
6538 	if (!skb) {
6539 		rtw89_err(rtwdev,
6540 			  "failed to alloc skb for mcc req tsf\n");
6541 		return -ENOMEM;
6542 	}
6543 
6544 	skb_put(skb, H2C_MCC_REQ_TSF_LEN);
6545 	RTW89_SET_FWCMD_MCC_REQ_TSF_GROUP(skb->data, req->group);
6546 	RTW89_SET_FWCMD_MCC_REQ_TSF_MACID_X(skb->data, req->macid_x);
6547 	RTW89_SET_FWCMD_MCC_REQ_TSF_MACID_Y(skb->data, req->macid_y);
6548 
6549 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6550 			      H2C_CAT_MAC,
6551 			      H2C_CL_MCC,
6552 			      H2C_FUNC_MCC_REQ_TSF, 0, 0,
6553 			      H2C_MCC_REQ_TSF_LEN);
6554 
6555 	cond = RTW89_MCC_WAIT_COND(req->group, H2C_FUNC_MCC_REQ_TSF);
6556 	ret = rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6557 	if (ret)
6558 		return ret;
6559 
6560 	tmp = (struct rtw89_mac_mcc_tsf_rpt *)wait->data.buf;
6561 	*rpt = *tmp;
6562 
6563 	return 0;
6564 }
6565 
6566 #define H2C_MCC_MACID_BITMAP_DSC_LEN 4
6567 int rtw89_fw_h2c_mcc_macid_bitmap(struct rtw89_dev *rtwdev, u8 group, u8 macid,
6568 				  u8 *bitmap)
6569 {
6570 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6571 	struct sk_buff *skb;
6572 	unsigned int cond;
6573 	u8 map_len;
6574 	u8 h2c_len;
6575 
6576 	BUILD_BUG_ON(RTW89_MAX_MAC_ID_NUM % 8);
6577 	map_len = RTW89_MAX_MAC_ID_NUM / 8;
6578 	h2c_len = H2C_MCC_MACID_BITMAP_DSC_LEN + map_len;
6579 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, h2c_len);
6580 	if (!skb) {
6581 		rtw89_err(rtwdev,
6582 			  "failed to alloc skb for mcc macid bitmap\n");
6583 		return -ENOMEM;
6584 	}
6585 
6586 	skb_put(skb, h2c_len);
6587 	RTW89_SET_FWCMD_MCC_MACID_BITMAP_GROUP(skb->data, group);
6588 	RTW89_SET_FWCMD_MCC_MACID_BITMAP_MACID(skb->data, macid);
6589 	RTW89_SET_FWCMD_MCC_MACID_BITMAP_BITMAP_LENGTH(skb->data, map_len);
6590 	RTW89_SET_FWCMD_MCC_MACID_BITMAP_BITMAP(skb->data, bitmap, map_len);
6591 
6592 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6593 			      H2C_CAT_MAC,
6594 			      H2C_CL_MCC,
6595 			      H2C_FUNC_MCC_MACID_BITMAP, 0, 0,
6596 			      h2c_len);
6597 
6598 	cond = RTW89_MCC_WAIT_COND(group, H2C_FUNC_MCC_MACID_BITMAP);
6599 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6600 }
6601 
6602 #define H2C_MCC_SYNC_LEN 4
6603 int rtw89_fw_h2c_mcc_sync(struct rtw89_dev *rtwdev, u8 group, u8 source,
6604 			  u8 target, u8 offset)
6605 {
6606 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6607 	struct sk_buff *skb;
6608 	unsigned int cond;
6609 
6610 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_MCC_SYNC_LEN);
6611 	if (!skb) {
6612 		rtw89_err(rtwdev,
6613 			  "failed to alloc skb for mcc sync\n");
6614 		return -ENOMEM;
6615 	}
6616 
6617 	skb_put(skb, H2C_MCC_SYNC_LEN);
6618 	RTW89_SET_FWCMD_MCC_SYNC_GROUP(skb->data, group);
6619 	RTW89_SET_FWCMD_MCC_SYNC_MACID_SOURCE(skb->data, source);
6620 	RTW89_SET_FWCMD_MCC_SYNC_MACID_TARGET(skb->data, target);
6621 	RTW89_SET_FWCMD_MCC_SYNC_SYNC_OFFSET(skb->data, offset);
6622 
6623 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6624 			      H2C_CAT_MAC,
6625 			      H2C_CL_MCC,
6626 			      H2C_FUNC_MCC_SYNC, 0, 0,
6627 			      H2C_MCC_SYNC_LEN);
6628 
6629 	cond = RTW89_MCC_WAIT_COND(group, H2C_FUNC_MCC_SYNC);
6630 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6631 }
6632 
6633 #define H2C_MCC_SET_DURATION_LEN 20
6634 int rtw89_fw_h2c_mcc_set_duration(struct rtw89_dev *rtwdev,
6635 				  const struct rtw89_fw_mcc_duration *p)
6636 {
6637 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6638 	struct sk_buff *skb;
6639 	unsigned int cond;
6640 
6641 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, H2C_MCC_SET_DURATION_LEN);
6642 	if (!skb) {
6643 		rtw89_err(rtwdev,
6644 			  "failed to alloc skb for mcc set duration\n");
6645 		return -ENOMEM;
6646 	}
6647 
6648 	skb_put(skb, H2C_MCC_SET_DURATION_LEN);
6649 	RTW89_SET_FWCMD_MCC_SET_DURATION_GROUP(skb->data, p->group);
6650 	RTW89_SET_FWCMD_MCC_SET_DURATION_BTC_IN_GROUP(skb->data, p->btc_in_group);
6651 	RTW89_SET_FWCMD_MCC_SET_DURATION_START_MACID(skb->data, p->start_macid);
6652 	RTW89_SET_FWCMD_MCC_SET_DURATION_MACID_X(skb->data, p->macid_x);
6653 	RTW89_SET_FWCMD_MCC_SET_DURATION_MACID_Y(skb->data, p->macid_y);
6654 	RTW89_SET_FWCMD_MCC_SET_DURATION_START_TSF_LOW(skb->data,
6655 						       p->start_tsf_low);
6656 	RTW89_SET_FWCMD_MCC_SET_DURATION_START_TSF_HIGH(skb->data,
6657 							p->start_tsf_high);
6658 	RTW89_SET_FWCMD_MCC_SET_DURATION_DURATION_X(skb->data, p->duration_x);
6659 	RTW89_SET_FWCMD_MCC_SET_DURATION_DURATION_Y(skb->data, p->duration_y);
6660 
6661 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6662 			      H2C_CAT_MAC,
6663 			      H2C_CL_MCC,
6664 			      H2C_FUNC_MCC_SET_DURATION, 0, 0,
6665 			      H2C_MCC_SET_DURATION_LEN);
6666 
6667 	cond = RTW89_MCC_WAIT_COND(p->group, H2C_FUNC_MCC_SET_DURATION);
6668 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6669 }
6670 
6671 static
6672 u32 rtw89_fw_h2c_mrc_add_slot(struct rtw89_dev *rtwdev,
6673 			      const struct rtw89_fw_mrc_add_slot_arg *slot_arg,
6674 			      struct rtw89_h2c_mrc_add_slot *slot_h2c)
6675 {
6676 	bool fill_h2c = !!slot_h2c;
6677 	unsigned int i;
6678 
6679 	if (!fill_h2c)
6680 		goto calc_len;
6681 
6682 	slot_h2c->w0 = le32_encode_bits(slot_arg->duration,
6683 					RTW89_H2C_MRC_ADD_SLOT_W0_DURATION) |
6684 		       le32_encode_bits(slot_arg->courtesy_en,
6685 					RTW89_H2C_MRC_ADD_SLOT_W0_COURTESY_EN) |
6686 		       le32_encode_bits(slot_arg->role_num,
6687 					RTW89_H2C_MRC_ADD_SLOT_W0_ROLE_NUM);
6688 	slot_h2c->w1 = le32_encode_bits(slot_arg->courtesy_period,
6689 					RTW89_H2C_MRC_ADD_SLOT_W1_COURTESY_PERIOD) |
6690 		       le32_encode_bits(slot_arg->courtesy_target,
6691 					RTW89_H2C_MRC_ADD_SLOT_W1_COURTESY_TARGET);
6692 
6693 	for (i = 0; i < slot_arg->role_num; i++) {
6694 		slot_h2c->roles[i].w0 =
6695 			le32_encode_bits(slot_arg->roles[i].macid,
6696 					 RTW89_H2C_MRC_ADD_ROLE_W0_MACID) |
6697 			le32_encode_bits(slot_arg->roles[i].role_type,
6698 					 RTW89_H2C_MRC_ADD_ROLE_W0_ROLE_TYPE) |
6699 			le32_encode_bits(slot_arg->roles[i].is_master,
6700 					 RTW89_H2C_MRC_ADD_ROLE_W0_IS_MASTER) |
6701 			le32_encode_bits(slot_arg->roles[i].en_tx_null,
6702 					 RTW89_H2C_MRC_ADD_ROLE_W0_TX_NULL_EN) |
6703 			le32_encode_bits(false,
6704 					 RTW89_H2C_MRC_ADD_ROLE_W0_IS_ALT_ROLE) |
6705 			le32_encode_bits(false,
6706 					 RTW89_H2C_MRC_ADD_ROLE_W0_ROLE_ALT_EN);
6707 		slot_h2c->roles[i].w1 =
6708 			le32_encode_bits(slot_arg->roles[i].central_ch,
6709 					 RTW89_H2C_MRC_ADD_ROLE_W1_CENTRAL_CH_SEG) |
6710 			le32_encode_bits(slot_arg->roles[i].primary_ch,
6711 					 RTW89_H2C_MRC_ADD_ROLE_W1_PRI_CH) |
6712 			le32_encode_bits(slot_arg->roles[i].bw,
6713 					 RTW89_H2C_MRC_ADD_ROLE_W1_BW) |
6714 			le32_encode_bits(slot_arg->roles[i].band,
6715 					 RTW89_H2C_MRC_ADD_ROLE_W1_CH_BAND_TYPE) |
6716 			le32_encode_bits(slot_arg->roles[i].null_early,
6717 					 RTW89_H2C_MRC_ADD_ROLE_W1_NULL_EARLY) |
6718 			le32_encode_bits(false,
6719 					 RTW89_H2C_MRC_ADD_ROLE_W1_RFK_BY_PASS) |
6720 			le32_encode_bits(true,
6721 					 RTW89_H2C_MRC_ADD_ROLE_W1_CAN_BTC);
6722 		slot_h2c->roles[i].macid_main_bitmap =
6723 			cpu_to_le32(slot_arg->roles[i].macid_main_bitmap);
6724 		slot_h2c->roles[i].macid_paired_bitmap =
6725 			cpu_to_le32(slot_arg->roles[i].macid_paired_bitmap);
6726 	}
6727 
6728 calc_len:
6729 	return struct_size(slot_h2c, roles, slot_arg->role_num);
6730 }
6731 
6732 int rtw89_fw_h2c_mrc_add(struct rtw89_dev *rtwdev,
6733 			 const struct rtw89_fw_mrc_add_arg *arg)
6734 {
6735 	struct rtw89_h2c_mrc_add *h2c_head;
6736 	struct sk_buff *skb;
6737 	unsigned int i;
6738 	void *tmp;
6739 	u32 len;
6740 	int ret;
6741 
6742 	len = sizeof(*h2c_head);
6743 	for (i = 0; i < arg->slot_num; i++)
6744 		len += rtw89_fw_h2c_mrc_add_slot(rtwdev, &arg->slots[i], NULL);
6745 
6746 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
6747 	if (!skb) {
6748 		rtw89_err(rtwdev, "failed to alloc skb for mrc add\n");
6749 		return -ENOMEM;
6750 	}
6751 
6752 	skb_put(skb, len);
6753 	tmp = skb->data;
6754 
6755 	h2c_head = tmp;
6756 	h2c_head->w0 = le32_encode_bits(arg->sch_idx,
6757 					RTW89_H2C_MRC_ADD_W0_SCH_IDX) |
6758 		       le32_encode_bits(arg->sch_type,
6759 					RTW89_H2C_MRC_ADD_W0_SCH_TYPE) |
6760 		       le32_encode_bits(arg->slot_num,
6761 					RTW89_H2C_MRC_ADD_W0_SLOT_NUM) |
6762 		       le32_encode_bits(arg->btc_in_sch,
6763 					RTW89_H2C_MRC_ADD_W0_BTC_IN_SCH);
6764 
6765 	tmp += sizeof(*h2c_head);
6766 	for (i = 0; i < arg->slot_num; i++)
6767 		tmp += rtw89_fw_h2c_mrc_add_slot(rtwdev, &arg->slots[i], tmp);
6768 
6769 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6770 			      H2C_CAT_MAC,
6771 			      H2C_CL_MRC,
6772 			      H2C_FUNC_ADD_MRC, 0, 0,
6773 			      len);
6774 
6775 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6776 	if (ret) {
6777 		rtw89_err(rtwdev, "failed to send h2c\n");
6778 		dev_kfree_skb_any(skb);
6779 		return -EBUSY;
6780 	}
6781 
6782 	return 0;
6783 }
6784 
6785 int rtw89_fw_h2c_mrc_start(struct rtw89_dev *rtwdev,
6786 			   const struct rtw89_fw_mrc_start_arg *arg)
6787 {
6788 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6789 	struct rtw89_h2c_mrc_start *h2c;
6790 	u32 len = sizeof(*h2c);
6791 	struct sk_buff *skb;
6792 	unsigned int cond;
6793 
6794 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
6795 	if (!skb) {
6796 		rtw89_err(rtwdev, "failed to alloc skb for mrc start\n");
6797 		return -ENOMEM;
6798 	}
6799 
6800 	skb_put(skb, len);
6801 	h2c = (struct rtw89_h2c_mrc_start *)skb->data;
6802 
6803 	h2c->w0 = le32_encode_bits(arg->sch_idx,
6804 				   RTW89_H2C_MRC_START_W0_SCH_IDX) |
6805 		  le32_encode_bits(arg->old_sch_idx,
6806 				   RTW89_H2C_MRC_START_W0_OLD_SCH_IDX) |
6807 		  le32_encode_bits(arg->action,
6808 				   RTW89_H2C_MRC_START_W0_ACTION);
6809 
6810 	h2c->start_tsf_high = cpu_to_le32(arg->start_tsf >> 32);
6811 	h2c->start_tsf_low = cpu_to_le32(arg->start_tsf);
6812 
6813 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6814 			      H2C_CAT_MAC,
6815 			      H2C_CL_MRC,
6816 			      H2C_FUNC_START_MRC, 0, 0,
6817 			      len);
6818 
6819 	cond = RTW89_MRC_WAIT_COND(arg->sch_idx, H2C_FUNC_START_MRC);
6820 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6821 }
6822 
6823 int rtw89_fw_h2c_mrc_del(struct rtw89_dev *rtwdev, u8 sch_idx)
6824 {
6825 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6826 	struct rtw89_h2c_mrc_del *h2c;
6827 	u32 len = sizeof(*h2c);
6828 	struct sk_buff *skb;
6829 	unsigned int cond;
6830 
6831 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
6832 	if (!skb) {
6833 		rtw89_err(rtwdev, "failed to alloc skb for mrc del\n");
6834 		return -ENOMEM;
6835 	}
6836 
6837 	skb_put(skb, len);
6838 	h2c = (struct rtw89_h2c_mrc_del *)skb->data;
6839 
6840 	h2c->w0 = le32_encode_bits(sch_idx, RTW89_H2C_MRC_DEL_W0_SCH_IDX);
6841 
6842 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6843 			      H2C_CAT_MAC,
6844 			      H2C_CL_MRC,
6845 			      H2C_FUNC_DEL_MRC, 0, 0,
6846 			      len);
6847 
6848 	cond = RTW89_MRC_WAIT_COND(sch_idx, H2C_FUNC_DEL_MRC);
6849 	return rtw89_h2c_tx_and_wait(rtwdev, skb, wait, cond);
6850 }
6851 
6852 int rtw89_fw_h2c_mrc_req_tsf(struct rtw89_dev *rtwdev,
6853 			     const struct rtw89_fw_mrc_req_tsf_arg *arg,
6854 			     struct rtw89_mac_mrc_tsf_rpt *rpt)
6855 {
6856 	struct rtw89_wait_info *wait = &rtwdev->mcc.wait;
6857 	struct rtw89_h2c_mrc_req_tsf *h2c;
6858 	struct rtw89_mac_mrc_tsf_rpt *tmp;
6859 	struct sk_buff *skb;
6860 	unsigned int i;
6861 	u32 len;
6862 	int ret;
6863 
6864 	len = struct_size(h2c, infos, arg->num);
6865 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
6866 	if (!skb) {
6867 		rtw89_err(rtwdev, "failed to alloc skb for mrc req tsf\n");
6868 		return -ENOMEM;
6869 	}
6870 
6871 	skb_put(skb, len);
6872 	h2c = (struct rtw89_h2c_mrc_req_tsf *)skb->data;
6873 
6874 	h2c->req_tsf_num = arg->num;
6875 	for (i = 0; i < arg->num; i++)
6876 		h2c->infos[i] =
6877 			u8_encode_bits(arg->infos[i].band,
6878 				       RTW89_H2C_MRC_REQ_TSF_INFO_BAND) |
6879 			u8_encode_bits(arg->infos[i].port,
6880 				       RTW89_H2C_MRC_REQ_TSF_INFO_PORT);
6881 
6882 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6883 			      H2C_CAT_MAC,
6884 			      H2C_CL_MRC,
6885 			      H2C_FUNC_MRC_REQ_TSF, 0, 0,
6886 			      len);
6887 
6888 	ret = rtw89_h2c_tx_and_wait(rtwdev, skb, wait, RTW89_MRC_WAIT_COND_REQ_TSF);
6889 	if (ret)
6890 		return ret;
6891 
6892 	tmp = (struct rtw89_mac_mrc_tsf_rpt *)wait->data.buf;
6893 	*rpt = *tmp;
6894 
6895 	return 0;
6896 }
6897 
6898 int rtw89_fw_h2c_mrc_upd_bitmap(struct rtw89_dev *rtwdev,
6899 				const struct rtw89_fw_mrc_upd_bitmap_arg *arg)
6900 {
6901 	struct rtw89_h2c_mrc_upd_bitmap *h2c;
6902 	u32 len = sizeof(*h2c);
6903 	struct sk_buff *skb;
6904 	int ret;
6905 
6906 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
6907 	if (!skb) {
6908 		rtw89_err(rtwdev, "failed to alloc skb for mrc upd bitmap\n");
6909 		return -ENOMEM;
6910 	}
6911 
6912 	skb_put(skb, len);
6913 	h2c = (struct rtw89_h2c_mrc_upd_bitmap *)skb->data;
6914 
6915 	h2c->w0 = le32_encode_bits(arg->sch_idx,
6916 				   RTW89_H2C_MRC_UPD_BITMAP_W0_SCH_IDX) |
6917 		  le32_encode_bits(arg->action,
6918 				   RTW89_H2C_MRC_UPD_BITMAP_W0_ACTION) |
6919 		  le32_encode_bits(arg->macid,
6920 				   RTW89_H2C_MRC_UPD_BITMAP_W0_MACID);
6921 	h2c->w1 = le32_encode_bits(arg->client_macid,
6922 				   RTW89_H2C_MRC_UPD_BITMAP_W1_CLIENT_MACID);
6923 
6924 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6925 			      H2C_CAT_MAC,
6926 			      H2C_CL_MRC,
6927 			      H2C_FUNC_MRC_UPD_BITMAP, 0, 0,
6928 			      len);
6929 
6930 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6931 	if (ret) {
6932 		rtw89_err(rtwdev, "failed to send h2c\n");
6933 		dev_kfree_skb_any(skb);
6934 		return -EBUSY;
6935 	}
6936 
6937 	return 0;
6938 }
6939 
6940 int rtw89_fw_h2c_mrc_sync(struct rtw89_dev *rtwdev,
6941 			  const struct rtw89_fw_mrc_sync_arg *arg)
6942 {
6943 	struct rtw89_h2c_mrc_sync *h2c;
6944 	u32 len = sizeof(*h2c);
6945 	struct sk_buff *skb;
6946 	int ret;
6947 
6948 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
6949 	if (!skb) {
6950 		rtw89_err(rtwdev, "failed to alloc skb for mrc sync\n");
6951 		return -ENOMEM;
6952 	}
6953 
6954 	skb_put(skb, len);
6955 	h2c = (struct rtw89_h2c_mrc_sync *)skb->data;
6956 
6957 	h2c->w0 = le32_encode_bits(true, RTW89_H2C_MRC_SYNC_W0_SYNC_EN) |
6958 		  le32_encode_bits(arg->src.port,
6959 				   RTW89_H2C_MRC_SYNC_W0_SRC_PORT) |
6960 		  le32_encode_bits(arg->src.band,
6961 				   RTW89_H2C_MRC_SYNC_W0_SRC_BAND) |
6962 		  le32_encode_bits(arg->dest.port,
6963 				   RTW89_H2C_MRC_SYNC_W0_DEST_PORT) |
6964 		  le32_encode_bits(arg->dest.band,
6965 				   RTW89_H2C_MRC_SYNC_W0_DEST_BAND);
6966 	h2c->w1 = le32_encode_bits(arg->offset, RTW89_H2C_MRC_SYNC_W1_OFFSET);
6967 
6968 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
6969 			      H2C_CAT_MAC,
6970 			      H2C_CL_MRC,
6971 			      H2C_FUNC_MRC_SYNC, 0, 0,
6972 			      len);
6973 
6974 	ret = rtw89_h2c_tx(rtwdev, skb, false);
6975 	if (ret) {
6976 		rtw89_err(rtwdev, "failed to send h2c\n");
6977 		dev_kfree_skb_any(skb);
6978 		return -EBUSY;
6979 	}
6980 
6981 	return 0;
6982 }
6983 
6984 int rtw89_fw_h2c_mrc_upd_duration(struct rtw89_dev *rtwdev,
6985 				  const struct rtw89_fw_mrc_upd_duration_arg *arg)
6986 {
6987 	struct rtw89_h2c_mrc_upd_duration *h2c;
6988 	struct sk_buff *skb;
6989 	unsigned int i;
6990 	u32 len;
6991 	int ret;
6992 
6993 	len = struct_size(h2c, slots, arg->slot_num);
6994 	skb = rtw89_fw_h2c_alloc_skb_with_hdr(rtwdev, len);
6995 	if (!skb) {
6996 		rtw89_err(rtwdev, "failed to alloc skb for mrc upd duration\n");
6997 		return -ENOMEM;
6998 	}
6999 
7000 	skb_put(skb, len);
7001 	h2c = (struct rtw89_h2c_mrc_upd_duration *)skb->data;
7002 
7003 	h2c->w0 = le32_encode_bits(arg->sch_idx,
7004 				   RTW89_H2C_MRC_UPD_DURATION_W0_SCH_IDX) |
7005 		  le32_encode_bits(arg->slot_num,
7006 				   RTW89_H2C_MRC_UPD_DURATION_W0_SLOT_NUM) |
7007 		  le32_encode_bits(false,
7008 				   RTW89_H2C_MRC_UPD_DURATION_W0_BTC_IN_SCH);
7009 
7010 	h2c->start_tsf_high = cpu_to_le32(arg->start_tsf >> 32);
7011 	h2c->start_tsf_low = cpu_to_le32(arg->start_tsf);
7012 
7013 	for (i = 0; i < arg->slot_num; i++) {
7014 		h2c->slots[i] =
7015 			le32_encode_bits(arg->slots[i].slot_idx,
7016 					 RTW89_H2C_MRC_UPD_DURATION_SLOT_SLOT_IDX) |
7017 			le32_encode_bits(arg->slots[i].duration,
7018 					 RTW89_H2C_MRC_UPD_DURATION_SLOT_DURATION);
7019 	}
7020 
7021 	rtw89_h2c_pkt_set_hdr(rtwdev, skb, FWCMD_TYPE_H2C,
7022 			      H2C_CAT_MAC,
7023 			      H2C_CL_MRC,
7024 			      H2C_FUNC_MRC_UPD_DURATION, 0, 0,
7025 			      len);
7026 
7027 	ret = rtw89_h2c_tx(rtwdev, skb, false);
7028 	if (ret) {
7029 		rtw89_err(rtwdev, "failed to send h2c\n");
7030 		dev_kfree_skb_any(skb);
7031 		return -EBUSY;
7032 	}
7033 
7034 	return 0;
7035 }
7036 
7037 static bool __fw_txpwr_entry_zero_ext(const void *ext_ptr, u8 ext_len)
7038 {
7039 	static const u8 zeros[U8_MAX] = {};
7040 
7041 	return memcmp(ext_ptr, zeros, ext_len) == 0;
7042 }
7043 
7044 #define __fw_txpwr_entry_acceptable(e, cursor, ent_sz)	\
7045 ({							\
7046 	u8 __var_sz = sizeof(*(e));			\
7047 	bool __accept;					\
7048 	if (__var_sz >= (ent_sz))			\
7049 		__accept = true;			\
7050 	else						\
7051 		__accept = __fw_txpwr_entry_zero_ext((cursor) + __var_sz,\
7052 						     (ent_sz) - __var_sz);\
7053 	__accept;					\
7054 })
7055 
7056 static bool
7057 fw_txpwr_byrate_entry_valid(const struct rtw89_fw_txpwr_byrate_entry *e,
7058 			    const void *cursor,
7059 			    const struct rtw89_txpwr_conf *conf)
7060 {
7061 	if (!__fw_txpwr_entry_acceptable(e, cursor, conf->ent_sz))
7062 		return false;
7063 
7064 	if (e->band >= RTW89_BAND_NUM || e->bw >= RTW89_BYR_BW_NUM)
7065 		return false;
7066 
7067 	switch (e->rs) {
7068 	case RTW89_RS_CCK:
7069 		if (e->shf + e->len > RTW89_RATE_CCK_NUM)
7070 			return false;
7071 		break;
7072 	case RTW89_RS_OFDM:
7073 		if (e->shf + e->len > RTW89_RATE_OFDM_NUM)
7074 			return false;
7075 		break;
7076 	case RTW89_RS_MCS:
7077 		if (e->shf + e->len > __RTW89_RATE_MCS_NUM ||
7078 		    e->nss >= RTW89_NSS_NUM ||
7079 		    e->ofdma >= RTW89_OFDMA_NUM)
7080 			return false;
7081 		break;
7082 	case RTW89_RS_HEDCM:
7083 		if (e->shf + e->len > RTW89_RATE_HEDCM_NUM ||
7084 		    e->nss >= RTW89_NSS_HEDCM_NUM ||
7085 		    e->ofdma >= RTW89_OFDMA_NUM)
7086 			return false;
7087 		break;
7088 	case RTW89_RS_OFFSET:
7089 		if (e->shf + e->len > __RTW89_RATE_OFFSET_NUM)
7090 			return false;
7091 		break;
7092 	default:
7093 		return false;
7094 	}
7095 
7096 	return true;
7097 }
7098 
7099 static
7100 void rtw89_fw_load_txpwr_byrate(struct rtw89_dev *rtwdev,
7101 				const struct rtw89_txpwr_table *tbl)
7102 {
7103 	const struct rtw89_txpwr_conf *conf = tbl->data;
7104 	struct rtw89_fw_txpwr_byrate_entry entry = {};
7105 	struct rtw89_txpwr_byrate *byr_head;
7106 	struct rtw89_rate_desc desc = {};
7107 	const void *cursor;
7108 	u32 data;
7109 	s8 *byr;
7110 	int i;
7111 
7112 	rtw89_for_each_in_txpwr_conf(entry, cursor, conf) {
7113 		if (!fw_txpwr_byrate_entry_valid(&entry, cursor, conf))
7114 			continue;
7115 
7116 		byr_head = &rtwdev->byr[entry.band][entry.bw];
7117 		data = le32_to_cpu(entry.data);
7118 		desc.ofdma = entry.ofdma;
7119 		desc.nss = entry.nss;
7120 		desc.rs = entry.rs;
7121 
7122 		for (i = 0; i < entry.len; i++, data >>= 8) {
7123 			desc.idx = entry.shf + i;
7124 			byr = rtw89_phy_raw_byr_seek(rtwdev, byr_head, &desc);
7125 			*byr = data & 0xff;
7126 		}
7127 	}
7128 }
7129 
7130 static bool
7131 fw_txpwr_lmt_2ghz_entry_valid(const struct rtw89_fw_txpwr_lmt_2ghz_entry *e,
7132 			      const void *cursor,
7133 			      const struct rtw89_txpwr_conf *conf)
7134 {
7135 	if (!__fw_txpwr_entry_acceptable(e, cursor, conf->ent_sz))
7136 		return false;
7137 
7138 	if (e->bw >= RTW89_2G_BW_NUM)
7139 		return false;
7140 	if (e->nt >= RTW89_NTX_NUM)
7141 		return false;
7142 	if (e->rs >= RTW89_RS_LMT_NUM)
7143 		return false;
7144 	if (e->bf >= RTW89_BF_NUM)
7145 		return false;
7146 	if (e->regd >= RTW89_REGD_NUM)
7147 		return false;
7148 	if (e->ch_idx >= RTW89_2G_CH_NUM)
7149 		return false;
7150 
7151 	return true;
7152 }
7153 
7154 static
7155 void rtw89_fw_load_txpwr_lmt_2ghz(struct rtw89_txpwr_lmt_2ghz_data *data)
7156 {
7157 	const struct rtw89_txpwr_conf *conf = &data->conf;
7158 	struct rtw89_fw_txpwr_lmt_2ghz_entry entry = {};
7159 	const void *cursor;
7160 
7161 	rtw89_for_each_in_txpwr_conf(entry, cursor, conf) {
7162 		if (!fw_txpwr_lmt_2ghz_entry_valid(&entry, cursor, conf))
7163 			continue;
7164 
7165 		data->v[entry.bw][entry.nt][entry.rs][entry.bf][entry.regd]
7166 		       [entry.ch_idx] = entry.v;
7167 	}
7168 }
7169 
7170 static bool
7171 fw_txpwr_lmt_5ghz_entry_valid(const struct rtw89_fw_txpwr_lmt_5ghz_entry *e,
7172 			      const void *cursor,
7173 			      const struct rtw89_txpwr_conf *conf)
7174 {
7175 	if (!__fw_txpwr_entry_acceptable(e, cursor, conf->ent_sz))
7176 		return false;
7177 
7178 	if (e->bw >= RTW89_5G_BW_NUM)
7179 		return false;
7180 	if (e->nt >= RTW89_NTX_NUM)
7181 		return false;
7182 	if (e->rs >= RTW89_RS_LMT_NUM)
7183 		return false;
7184 	if (e->bf >= RTW89_BF_NUM)
7185 		return false;
7186 	if (e->regd >= RTW89_REGD_NUM)
7187 		return false;
7188 	if (e->ch_idx >= RTW89_5G_CH_NUM)
7189 		return false;
7190 
7191 	return true;
7192 }
7193 
7194 static
7195 void rtw89_fw_load_txpwr_lmt_5ghz(struct rtw89_txpwr_lmt_5ghz_data *data)
7196 {
7197 	const struct rtw89_txpwr_conf *conf = &data->conf;
7198 	struct rtw89_fw_txpwr_lmt_5ghz_entry entry = {};
7199 	const void *cursor;
7200 
7201 	rtw89_for_each_in_txpwr_conf(entry, cursor, conf) {
7202 		if (!fw_txpwr_lmt_5ghz_entry_valid(&entry, cursor, conf))
7203 			continue;
7204 
7205 		data->v[entry.bw][entry.nt][entry.rs][entry.bf][entry.regd]
7206 		       [entry.ch_idx] = entry.v;
7207 	}
7208 }
7209 
7210 static bool
7211 fw_txpwr_lmt_6ghz_entry_valid(const struct rtw89_fw_txpwr_lmt_6ghz_entry *e,
7212 			      const void *cursor,
7213 			      const struct rtw89_txpwr_conf *conf)
7214 {
7215 	if (!__fw_txpwr_entry_acceptable(e, cursor, conf->ent_sz))
7216 		return false;
7217 
7218 	if (e->bw >= RTW89_6G_BW_NUM)
7219 		return false;
7220 	if (e->nt >= RTW89_NTX_NUM)
7221 		return false;
7222 	if (e->rs >= RTW89_RS_LMT_NUM)
7223 		return false;
7224 	if (e->bf >= RTW89_BF_NUM)
7225 		return false;
7226 	if (e->regd >= RTW89_REGD_NUM)
7227 		return false;
7228 	if (e->reg_6ghz_power >= NUM_OF_RTW89_REG_6GHZ_POWER)
7229 		return false;
7230 	if (e->ch_idx >= RTW89_6G_CH_NUM)
7231 		return false;
7232 
7233 	return true;
7234 }
7235 
7236 static
7237 void rtw89_fw_load_txpwr_lmt_6ghz(struct rtw89_txpwr_lmt_6ghz_data *data)
7238 {
7239 	const struct rtw89_txpwr_conf *conf = &data->conf;
7240 	struct rtw89_fw_txpwr_lmt_6ghz_entry entry = {};
7241 	const void *cursor;
7242 
7243 	rtw89_for_each_in_txpwr_conf(entry, cursor, conf) {
7244 		if (!fw_txpwr_lmt_6ghz_entry_valid(&entry, cursor, conf))
7245 			continue;
7246 
7247 		data->v[entry.bw][entry.nt][entry.rs][entry.bf][entry.regd]
7248 		       [entry.reg_6ghz_power][entry.ch_idx] = entry.v;
7249 	}
7250 }
7251 
7252 static bool
7253 fw_txpwr_lmt_ru_2ghz_entry_valid(const struct rtw89_fw_txpwr_lmt_ru_2ghz_entry *e,
7254 				 const void *cursor,
7255 				 const struct rtw89_txpwr_conf *conf)
7256 {
7257 	if (!__fw_txpwr_entry_acceptable(e, cursor, conf->ent_sz))
7258 		return false;
7259 
7260 	if (e->ru >= RTW89_RU_NUM)
7261 		return false;
7262 	if (e->nt >= RTW89_NTX_NUM)
7263 		return false;
7264 	if (e->regd >= RTW89_REGD_NUM)
7265 		return false;
7266 	if (e->ch_idx >= RTW89_2G_CH_NUM)
7267 		return false;
7268 
7269 	return true;
7270 }
7271 
7272 static
7273 void rtw89_fw_load_txpwr_lmt_ru_2ghz(struct rtw89_txpwr_lmt_ru_2ghz_data *data)
7274 {
7275 	const struct rtw89_txpwr_conf *conf = &data->conf;
7276 	struct rtw89_fw_txpwr_lmt_ru_2ghz_entry entry = {};
7277 	const void *cursor;
7278 
7279 	rtw89_for_each_in_txpwr_conf(entry, cursor, conf) {
7280 		if (!fw_txpwr_lmt_ru_2ghz_entry_valid(&entry, cursor, conf))
7281 			continue;
7282 
7283 		data->v[entry.ru][entry.nt][entry.regd][entry.ch_idx] = entry.v;
7284 	}
7285 }
7286 
7287 static bool
7288 fw_txpwr_lmt_ru_5ghz_entry_valid(const struct rtw89_fw_txpwr_lmt_ru_5ghz_entry *e,
7289 				 const void *cursor,
7290 				 const struct rtw89_txpwr_conf *conf)
7291 {
7292 	if (!__fw_txpwr_entry_acceptable(e, cursor, conf->ent_sz))
7293 		return false;
7294 
7295 	if (e->ru >= RTW89_RU_NUM)
7296 		return false;
7297 	if (e->nt >= RTW89_NTX_NUM)
7298 		return false;
7299 	if (e->regd >= RTW89_REGD_NUM)
7300 		return false;
7301 	if (e->ch_idx >= RTW89_5G_CH_NUM)
7302 		return false;
7303 
7304 	return true;
7305 }
7306 
7307 static
7308 void rtw89_fw_load_txpwr_lmt_ru_5ghz(struct rtw89_txpwr_lmt_ru_5ghz_data *data)
7309 {
7310 	const struct rtw89_txpwr_conf *conf = &data->conf;
7311 	struct rtw89_fw_txpwr_lmt_ru_5ghz_entry entry = {};
7312 	const void *cursor;
7313 
7314 	rtw89_for_each_in_txpwr_conf(entry, cursor, conf) {
7315 		if (!fw_txpwr_lmt_ru_5ghz_entry_valid(&entry, cursor, conf))
7316 			continue;
7317 
7318 		data->v[entry.ru][entry.nt][entry.regd][entry.ch_idx] = entry.v;
7319 	}
7320 }
7321 
7322 static bool
7323 fw_txpwr_lmt_ru_6ghz_entry_valid(const struct rtw89_fw_txpwr_lmt_ru_6ghz_entry *e,
7324 				 const void *cursor,
7325 				 const struct rtw89_txpwr_conf *conf)
7326 {
7327 	if (!__fw_txpwr_entry_acceptable(e, cursor, conf->ent_sz))
7328 		return false;
7329 
7330 	if (e->ru >= RTW89_RU_NUM)
7331 		return false;
7332 	if (e->nt >= RTW89_NTX_NUM)
7333 		return false;
7334 	if (e->regd >= RTW89_REGD_NUM)
7335 		return false;
7336 	if (e->reg_6ghz_power >= NUM_OF_RTW89_REG_6GHZ_POWER)
7337 		return false;
7338 	if (e->ch_idx >= RTW89_6G_CH_NUM)
7339 		return false;
7340 
7341 	return true;
7342 }
7343 
7344 static
7345 void rtw89_fw_load_txpwr_lmt_ru_6ghz(struct rtw89_txpwr_lmt_ru_6ghz_data *data)
7346 {
7347 	const struct rtw89_txpwr_conf *conf = &data->conf;
7348 	struct rtw89_fw_txpwr_lmt_ru_6ghz_entry entry = {};
7349 	const void *cursor;
7350 
7351 	rtw89_for_each_in_txpwr_conf(entry, cursor, conf) {
7352 		if (!fw_txpwr_lmt_ru_6ghz_entry_valid(&entry, cursor, conf))
7353 			continue;
7354 
7355 		data->v[entry.ru][entry.nt][entry.regd][entry.reg_6ghz_power]
7356 		       [entry.ch_idx] = entry.v;
7357 	}
7358 }
7359 
7360 static bool
7361 fw_tx_shape_lmt_entry_valid(const struct rtw89_fw_tx_shape_lmt_entry *e,
7362 			    const void *cursor,
7363 			    const struct rtw89_txpwr_conf *conf)
7364 {
7365 	if (!__fw_txpwr_entry_acceptable(e, cursor, conf->ent_sz))
7366 		return false;
7367 
7368 	if (e->band >= RTW89_BAND_NUM)
7369 		return false;
7370 	if (e->tx_shape_rs >= RTW89_RS_TX_SHAPE_NUM)
7371 		return false;
7372 	if (e->regd >= RTW89_REGD_NUM)
7373 		return false;
7374 
7375 	return true;
7376 }
7377 
7378 static
7379 void rtw89_fw_load_tx_shape_lmt(struct rtw89_tx_shape_lmt_data *data)
7380 {
7381 	const struct rtw89_txpwr_conf *conf = &data->conf;
7382 	struct rtw89_fw_tx_shape_lmt_entry entry = {};
7383 	const void *cursor;
7384 
7385 	rtw89_for_each_in_txpwr_conf(entry, cursor, conf) {
7386 		if (!fw_tx_shape_lmt_entry_valid(&entry, cursor, conf))
7387 			continue;
7388 
7389 		data->v[entry.band][entry.tx_shape_rs][entry.regd] = entry.v;
7390 	}
7391 }
7392 
7393 static bool
7394 fw_tx_shape_lmt_ru_entry_valid(const struct rtw89_fw_tx_shape_lmt_ru_entry *e,
7395 			       const void *cursor,
7396 			       const struct rtw89_txpwr_conf *conf)
7397 {
7398 	if (!__fw_txpwr_entry_acceptable(e, cursor, conf->ent_sz))
7399 		return false;
7400 
7401 	if (e->band >= RTW89_BAND_NUM)
7402 		return false;
7403 	if (e->regd >= RTW89_REGD_NUM)
7404 		return false;
7405 
7406 	return true;
7407 }
7408 
7409 static
7410 void rtw89_fw_load_tx_shape_lmt_ru(struct rtw89_tx_shape_lmt_ru_data *data)
7411 {
7412 	const struct rtw89_txpwr_conf *conf = &data->conf;
7413 	struct rtw89_fw_tx_shape_lmt_ru_entry entry = {};
7414 	const void *cursor;
7415 
7416 	rtw89_for_each_in_txpwr_conf(entry, cursor, conf) {
7417 		if (!fw_tx_shape_lmt_ru_entry_valid(&entry, cursor, conf))
7418 			continue;
7419 
7420 		data->v[entry.band][entry.regd] = entry.v;
7421 	}
7422 }
7423 
7424 const struct rtw89_rfe_parms *
7425 rtw89_load_rfe_data_from_fw(struct rtw89_dev *rtwdev,
7426 			    const struct rtw89_rfe_parms *init)
7427 {
7428 	struct rtw89_rfe_data *rfe_data = rtwdev->rfe_data;
7429 	struct rtw89_rfe_parms *parms;
7430 
7431 	if (!rfe_data)
7432 		return init;
7433 
7434 	parms = &rfe_data->rfe_parms;
7435 	if (init)
7436 		*parms = *init;
7437 
7438 	if (rtw89_txpwr_conf_valid(&rfe_data->byrate.conf)) {
7439 		rfe_data->byrate.tbl.data = &rfe_data->byrate.conf;
7440 		rfe_data->byrate.tbl.size = 0; /* don't care here */
7441 		rfe_data->byrate.tbl.load = rtw89_fw_load_txpwr_byrate;
7442 		parms->byr_tbl = &rfe_data->byrate.tbl;
7443 	}
7444 
7445 	if (rtw89_txpwr_conf_valid(&rfe_data->lmt_2ghz.conf)) {
7446 		rtw89_fw_load_txpwr_lmt_2ghz(&rfe_data->lmt_2ghz);
7447 		parms->rule_2ghz.lmt = &rfe_data->lmt_2ghz.v;
7448 	}
7449 
7450 	if (rtw89_txpwr_conf_valid(&rfe_data->lmt_5ghz.conf)) {
7451 		rtw89_fw_load_txpwr_lmt_5ghz(&rfe_data->lmt_5ghz);
7452 		parms->rule_5ghz.lmt = &rfe_data->lmt_5ghz.v;
7453 	}
7454 
7455 	if (rtw89_txpwr_conf_valid(&rfe_data->lmt_6ghz.conf)) {
7456 		rtw89_fw_load_txpwr_lmt_6ghz(&rfe_data->lmt_6ghz);
7457 		parms->rule_6ghz.lmt = &rfe_data->lmt_6ghz.v;
7458 	}
7459 
7460 	if (rtw89_txpwr_conf_valid(&rfe_data->lmt_ru_2ghz.conf)) {
7461 		rtw89_fw_load_txpwr_lmt_ru_2ghz(&rfe_data->lmt_ru_2ghz);
7462 		parms->rule_2ghz.lmt_ru = &rfe_data->lmt_ru_2ghz.v;
7463 	}
7464 
7465 	if (rtw89_txpwr_conf_valid(&rfe_data->lmt_ru_5ghz.conf)) {
7466 		rtw89_fw_load_txpwr_lmt_ru_5ghz(&rfe_data->lmt_ru_5ghz);
7467 		parms->rule_5ghz.lmt_ru = &rfe_data->lmt_ru_5ghz.v;
7468 	}
7469 
7470 	if (rtw89_txpwr_conf_valid(&rfe_data->lmt_ru_6ghz.conf)) {
7471 		rtw89_fw_load_txpwr_lmt_ru_6ghz(&rfe_data->lmt_ru_6ghz);
7472 		parms->rule_6ghz.lmt_ru = &rfe_data->lmt_ru_6ghz.v;
7473 	}
7474 
7475 	if (rtw89_txpwr_conf_valid(&rfe_data->tx_shape_lmt.conf)) {
7476 		rtw89_fw_load_tx_shape_lmt(&rfe_data->tx_shape_lmt);
7477 		parms->tx_shape.lmt = &rfe_data->tx_shape_lmt.v;
7478 	}
7479 
7480 	if (rtw89_txpwr_conf_valid(&rfe_data->tx_shape_lmt_ru.conf)) {
7481 		rtw89_fw_load_tx_shape_lmt_ru(&rfe_data->tx_shape_lmt_ru);
7482 		parms->tx_shape.lmt_ru = &rfe_data->tx_shape_lmt_ru.v;
7483 	}
7484 
7485 	return parms;
7486 }
7487