1 // SPDX-License-Identifier: ISC
2 /* Copyright (C) 2022 MediaTek Inc. */
3 
4 #include <linux/kernel.h>
5 #include <linux/module.h>
6 #include <linux/platform_device.h>
7 #include <linux/pinctrl/consumer.h>
8 #include <linux/of.h>
9 #include <linux/of_device.h>
10 #include <linux/of_reserved_mem.h>
11 #include <linux/of_gpio.h>
12 #include <linux/iopoll.h>
13 #include <linux/reset.h>
14 #include <linux/of_net.h>
15 #include <linux/clk.h>
16 
17 #include "mt7915.h"
18 
19 #define MT7981_CON_INFRA_VERSION 0x02090000
20 #define MT7986_CON_INFRA_VERSION 0x02070000
21 
22 /* INFRACFG */
23 #define MT_INFRACFG_CONN2AP_SLPPROT	0x0d0
24 #define MT_INFRACFG_AP2CONN_SLPPROT	0x0d4
25 
26 #define MT_INFRACFG_RX_EN_MASK		BIT(16)
27 #define MT_INFRACFG_TX_RDY_MASK		BIT(4)
28 #define MT_INFRACFG_TX_EN_MASK		BIT(0)
29 
30 /* TOP POS */
31 #define MT_TOP_POS_FAST_CTRL		0x114
32 #define MT_TOP_POS_FAST_EN_MASK		BIT(3)
33 
34 #define MT_TOP_POS_SKU			0x21c
35 #define MT_TOP_POS_SKU_MASK		GENMASK(31, 28)
36 #define MT_TOP_POS_SKU_ADIE_DBDC_MASK	BIT(2)
37 
38 enum {
39 	ADIE_SB,
40 	ADIE_DBDC
41 };
42 
43 static int
44 mt76_wmac_spi_read(struct mt7915_dev *dev, u8 adie, u32 addr, u32 *val)
45 {
46 	int ret;
47 	u32 cur;
48 
49 	ret = read_poll_timeout(mt76_rr, cur, !(cur & MT_TOP_SPI_POLLING_BIT),
50 				USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
51 				dev, MT_TOP_SPI_BUSY_CR(adie));
52 	if (ret)
53 		return ret;
54 
55 	mt76_wr(dev, MT_TOP_SPI_ADDR_CR(adie),
56 		MT_TOP_SPI_READ_ADDR_FORMAT | addr);
57 	mt76_wr(dev, MT_TOP_SPI_WRITE_DATA_CR(adie), 0);
58 
59 	ret = read_poll_timeout(mt76_rr, cur, !(cur & MT_TOP_SPI_POLLING_BIT),
60 				USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
61 				dev, MT_TOP_SPI_BUSY_CR(adie));
62 	if (ret)
63 		return ret;
64 
65 	*val = mt76_rr(dev, MT_TOP_SPI_READ_DATA_CR(adie));
66 
67 	return 0;
68 }
69 
70 static int
71 mt76_wmac_spi_write(struct mt7915_dev *dev, u8 adie, u32 addr, u32 val)
72 {
73 	int ret;
74 	u32 cur;
75 
76 	ret = read_poll_timeout(mt76_rr, cur, !(cur & MT_TOP_SPI_POLLING_BIT),
77 				USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
78 				dev, MT_TOP_SPI_BUSY_CR(adie));
79 	if (ret)
80 		return ret;
81 
82 	mt76_wr(dev, MT_TOP_SPI_ADDR_CR(adie),
83 		MT_TOP_SPI_WRITE_ADDR_FORMAT | addr);
84 	mt76_wr(dev, MT_TOP_SPI_WRITE_DATA_CR(adie), val);
85 
86 	return read_poll_timeout(mt76_rr, cur, !(cur & MT_TOP_SPI_POLLING_BIT),
87 				 USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
88 				 dev, MT_TOP_SPI_BUSY_CR(adie));
89 }
90 
91 static int
92 mt76_wmac_spi_rmw(struct mt7915_dev *dev, u8 adie,
93 		  u32 addr, u32 mask, u32 val)
94 {
95 	u32 cur, ret;
96 
97 	ret = mt76_wmac_spi_read(dev, adie, addr, &cur);
98 	if (ret)
99 		return ret;
100 
101 	cur &= ~mask;
102 	cur |= val;
103 
104 	return mt76_wmac_spi_write(dev, adie, addr, cur);
105 }
106 
107 static int
108 mt7986_wmac_adie_efuse_read(struct mt7915_dev *dev, u8 adie,
109 			    u32 addr, u32 *data)
110 {
111 	int ret, temp;
112 	u32 val, mask;
113 
114 	ret = mt76_wmac_spi_write(dev, adie, MT_ADIE_EFUSE_CFG,
115 				  MT_ADIE_EFUSE_CTRL_MASK);
116 	if (ret)
117 		return ret;
118 
119 	ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_EFUSE2_CTRL, BIT(30), 0x0);
120 	if (ret)
121 		return ret;
122 
123 	mask = (MT_ADIE_EFUSE_MODE_MASK | MT_ADIE_EFUSE_ADDR_MASK |
124 		MT_ADIE_EFUSE_KICK_MASK);
125 	val = FIELD_PREP(MT_ADIE_EFUSE_MODE_MASK, 0) |
126 	      FIELD_PREP(MT_ADIE_EFUSE_ADDR_MASK, addr) |
127 	      FIELD_PREP(MT_ADIE_EFUSE_KICK_MASK, 1);
128 	ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_EFUSE2_CTRL, mask, val);
129 	if (ret)
130 		return ret;
131 
132 	ret = read_poll_timeout(mt76_wmac_spi_read, temp,
133 				!temp && !FIELD_GET(MT_ADIE_EFUSE_KICK_MASK, val),
134 				USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
135 				dev, adie, MT_ADIE_EFUSE2_CTRL, &val);
136 	if (ret)
137 		return ret;
138 
139 	ret = mt76_wmac_spi_read(dev, adie, MT_ADIE_EFUSE2_CTRL, &val);
140 	if (ret)
141 		return ret;
142 
143 	if (FIELD_GET(MT_ADIE_EFUSE_VALID_MASK, val) == 1)
144 		ret = mt76_wmac_spi_read(dev, adie, MT_ADIE_EFUSE_RDATA0,
145 					 data);
146 
147 	return ret;
148 }
149 
150 static inline void mt76_wmac_spi_lock(struct mt7915_dev *dev)
151 {
152 	u32 cur;
153 
154 	read_poll_timeout(mt76_rr, cur,
155 			  FIELD_GET(MT_SEMA_RFSPI_STATUS_MASK, cur),
156 			  1000, 1000 * MSEC_PER_SEC, false, dev,
157 			  MT_SEMA_RFSPI_STATUS);
158 }
159 
160 static inline void mt76_wmac_spi_unlock(struct mt7915_dev *dev)
161 {
162 	mt76_wr(dev, MT_SEMA_RFSPI_RELEASE, 1);
163 }
164 
165 static u32 mt76_wmac_rmw(void __iomem *base, u32 offset, u32 mask, u32 val)
166 {
167 	val |= readl(base + offset) & ~mask;
168 	writel(val, base + offset);
169 
170 	return val;
171 }
172 
173 static u8 mt798x_wmac_check_adie_type(struct mt7915_dev *dev)
174 {
175 	u32 val;
176 
177 	/* Only DBDC A-die is used with MT7981 */
178 	if (is_mt7981(&dev->mt76))
179 		return ADIE_DBDC;
180 
181 	val = readl(dev->sku + MT_TOP_POS_SKU);
182 
183 	return FIELD_GET(MT_TOP_POS_SKU_ADIE_DBDC_MASK, val);
184 }
185 
186 static int mt7986_wmac_consys_reset(struct mt7915_dev *dev, bool enable)
187 {
188 	if (!enable)
189 		return reset_control_assert(dev->rstc);
190 
191 	mt76_wmac_rmw(dev->sku, MT_TOP_POS_FAST_CTRL,
192 		      MT_TOP_POS_FAST_EN_MASK,
193 		      FIELD_PREP(MT_TOP_POS_FAST_EN_MASK, 0x1));
194 
195 	return reset_control_deassert(dev->rstc);
196 }
197 
198 static int mt7986_wmac_gpio_setup(struct mt7915_dev *dev)
199 {
200 	struct pinctrl_state *state;
201 	struct pinctrl *pinctrl;
202 	int ret;
203 	u8 type;
204 
205 	type = mt798x_wmac_check_adie_type(dev);
206 	pinctrl = devm_pinctrl_get(dev->mt76.dev);
207 	if (IS_ERR(pinctrl))
208 		return PTR_ERR(pinctrl);
209 
210 	switch (type) {
211 	case ADIE_SB:
212 		state = pinctrl_lookup_state(pinctrl, "default");
213 		if (IS_ERR_OR_NULL(state))
214 			return -EINVAL;
215 		break;
216 	case ADIE_DBDC:
217 		state = pinctrl_lookup_state(pinctrl, "dbdc");
218 		if (IS_ERR_OR_NULL(state))
219 			return -EINVAL;
220 		break;
221 	default:
222 		return -EINVAL;
223 	}
224 
225 	ret = pinctrl_select_state(pinctrl, state);
226 	if (ret)
227 		return ret;
228 
229 	usleep_range(500, 1000);
230 
231 	return 0;
232 }
233 
234 static int mt7986_wmac_consys_lockup(struct mt7915_dev *dev, bool enable)
235 {
236 	int ret;
237 	u32 cur;
238 
239 	mt76_wmac_rmw(dev->dcm, MT_INFRACFG_AP2CONN_SLPPROT,
240 		      MT_INFRACFG_RX_EN_MASK,
241 		      FIELD_PREP(MT_INFRACFG_RX_EN_MASK, enable));
242 	ret = read_poll_timeout(readl, cur, !(cur & MT_INFRACFG_RX_EN_MASK),
243 				USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
244 				dev->dcm + MT_INFRACFG_AP2CONN_SLPPROT);
245 	if (ret)
246 		return ret;
247 
248 	mt76_wmac_rmw(dev->dcm, MT_INFRACFG_AP2CONN_SLPPROT,
249 		      MT_INFRACFG_TX_EN_MASK,
250 		      FIELD_PREP(MT_INFRACFG_TX_EN_MASK, enable));
251 	ret = read_poll_timeout(readl, cur, !(cur & MT_INFRACFG_TX_RDY_MASK),
252 				USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
253 				dev->dcm + MT_INFRACFG_AP2CONN_SLPPROT);
254 	if (ret)
255 		return ret;
256 
257 	mt76_wmac_rmw(dev->dcm, MT_INFRACFG_CONN2AP_SLPPROT,
258 		      MT_INFRACFG_RX_EN_MASK,
259 		      FIELD_PREP(MT_INFRACFG_RX_EN_MASK, enable));
260 	mt76_wmac_rmw(dev->dcm, MT_INFRACFG_CONN2AP_SLPPROT,
261 		      MT_INFRACFG_TX_EN_MASK,
262 		      FIELD_PREP(MT_INFRACFG_TX_EN_MASK, enable));
263 
264 	return 0;
265 }
266 
267 static int mt798x_wmac_coninfra_check(struct mt7915_dev *dev)
268 {
269 	u32 cur;
270 	u32 con_infra_version;
271 
272 	if (is_mt7981(&dev->mt76)) {
273 		con_infra_version = MT7981_CON_INFRA_VERSION;
274 	} else if (is_mt7986(&dev->mt76)) {
275 		con_infra_version = MT7986_CON_INFRA_VERSION;
276 	} else {
277 		WARN_ON(1);
278 		return -EINVAL;
279 	}
280 
281 	return read_poll_timeout(mt76_rr, cur, (cur == con_infra_version),
282 				 USEC_PER_MSEC, 50 * USEC_PER_MSEC,
283 				 false, dev, MT_CONN_INFRA_BASE);
284 }
285 
286 static int mt798x_wmac_coninfra_setup(struct mt7915_dev *dev)
287 {
288 	struct device *pdev = dev->mt76.dev;
289 	struct reserved_mem *rmem;
290 	struct device_node *np;
291 	u32 val;
292 
293 	np = of_parse_phandle(pdev->of_node, "memory-region", 0);
294 	if (!np)
295 		return -EINVAL;
296 
297 	rmem = of_reserved_mem_lookup(np);
298 	of_node_put(np);
299 	if (!rmem)
300 		return -EINVAL;
301 
302 	val = (rmem->base >> 16) & MT_TOP_MCU_EMI_BASE_MASK;
303 
304 	if (is_mt7986(&dev->mt76)) {
305 		/* Set conninfra subsys PLL check */
306 		mt76_rmw_field(dev, MT_INFRA_CKGEN_BUS,
307 			       MT_INFRA_CKGEN_BUS_RDY_SEL_MASK, 0x1);
308 		mt76_rmw_field(dev, MT_INFRA_CKGEN_BUS,
309 			       MT_INFRA_CKGEN_BUS_RDY_SEL_MASK, 0x1);
310 	}
311 
312 	mt76_rmw_field(dev, MT_TOP_MCU_EMI_BASE,
313 		       MT_TOP_MCU_EMI_BASE_MASK, val);
314 
315 	if (is_mt7981(&dev->mt76)) {
316 		mt76_rmw_field(dev, MT_TOP_WF_AP_PERI_BASE,
317 			       MT_TOP_WF_AP_PERI_BASE_MASK, 0x300d0000 >> 16);
318 
319 		mt76_rmw_field(dev, MT_TOP_EFUSE_BASE,
320 			       MT_TOP_EFUSE_BASE_MASK, 0x11f20000 >> 16);
321 	}
322 
323 	mt76_wr(dev, MT_INFRA_BUS_EMI_START, rmem->base);
324 	mt76_wr(dev, MT_INFRA_BUS_EMI_END, rmem->size);
325 
326 	mt76_rr(dev, MT_CONN_INFRA_EFUSE);
327 
328 	/* Set conninfra sysram */
329 	mt76_wr(dev, MT_TOP_RGU_SYSRAM_PDN, 0);
330 	mt76_wr(dev, MT_TOP_RGU_SYSRAM_SLP, 1);
331 
332 	return 0;
333 }
334 
335 static int mt798x_wmac_sku_setup(struct mt7915_dev *dev, u32 *adie_type)
336 {
337 	int ret;
338 	u32 adie_main = 0, adie_ext = 0;
339 
340 	mt76_rmw_field(dev, MT_CONN_INFRA_ADIE_RESET,
341 		       MT_CONN_INFRA_ADIE1_RESET_MASK, 0x1);
342 
343 	if (is_mt7986(&dev->mt76)) {
344 		mt76_rmw_field(dev, MT_CONN_INFRA_ADIE_RESET,
345 			       MT_CONN_INFRA_ADIE2_RESET_MASK, 0x1);
346 	}
347 
348 	mt76_wmac_spi_lock(dev);
349 
350 	ret = mt76_wmac_spi_read(dev, 0, MT_ADIE_CHIP_ID, &adie_main);
351 	if (ret)
352 		goto out;
353 
354 	if (is_mt7986(&dev->mt76)) {
355 		ret = mt76_wmac_spi_read(dev, 1, MT_ADIE_CHIP_ID, &adie_ext);
356 		if (ret)
357 			goto out;
358 	}
359 
360 	*adie_type = FIELD_GET(MT_ADIE_CHIP_ID_MASK, adie_main) |
361 		     (MT_ADIE_CHIP_ID_MASK & adie_ext);
362 
363 out:
364 	mt76_wmac_spi_unlock(dev);
365 
366 	return 0;
367 }
368 
369 static inline u16 mt7986_adie_idx(u8 adie, u32 adie_type)
370 {
371 	if (adie == 0)
372 		return u32_get_bits(adie_type, MT_ADIE_IDX0);
373 	else
374 		return u32_get_bits(adie_type, MT_ADIE_IDX1);
375 }
376 
377 static inline bool is_7975(struct mt7915_dev *dev, u8 adie, u32 adie_type)
378 {
379 	return mt7986_adie_idx(adie, adie_type) == 0x7975;
380 }
381 
382 static inline bool is_7976(struct mt7915_dev *dev, u8 adie, u32 adie_type)
383 {
384 	return mt7986_adie_idx(adie, adie_type) == 0x7976;
385 }
386 
387 static int mt7986_wmac_adie_thermal_cal(struct mt7915_dev *dev, u8 adie)
388 {
389 	int ret;
390 	u32 data, val;
391 
392 	ret = mt7986_wmac_adie_efuse_read(dev, adie, MT_ADIE_THADC_ANALOG,
393 					  &data);
394 	if (ret || FIELD_GET(MT_ADIE_ANA_EN_MASK, data)) {
395 		val = FIELD_GET(MT_ADIE_VRPI_SEL_EFUSE_MASK, data);
396 		ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_RG_TOP_THADC_BG,
397 					MT_ADIE_VRPI_SEL_CR_MASK,
398 					FIELD_PREP(MT_ADIE_VRPI_SEL_CR_MASK, val));
399 		if (ret)
400 			return ret;
401 
402 		val = FIELD_GET(MT_ADIE_PGA_GAIN_EFUSE_MASK, data);
403 		ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_RG_TOP_THADC,
404 					MT_ADIE_PGA_GAIN_MASK,
405 					FIELD_PREP(MT_ADIE_PGA_GAIN_MASK, val));
406 		if (ret)
407 			return ret;
408 	}
409 
410 	ret = mt7986_wmac_adie_efuse_read(dev, adie, MT_ADIE_THADC_SLOP,
411 					  &data);
412 	if (ret || FIELD_GET(MT_ADIE_ANA_EN_MASK, data)) {
413 		val = FIELD_GET(MT_ADIE_LDO_CTRL_EFUSE_MASK, data);
414 
415 		return mt76_wmac_spi_rmw(dev, adie, MT_ADIE_RG_TOP_THADC,
416 					 MT_ADIE_LDO_CTRL_MASK,
417 					 FIELD_PREP(MT_ADIE_LDO_CTRL_MASK, val));
418 	}
419 
420 	return 0;
421 }
422 
423 static int
424 mt7986_read_efuse_xo_trim_7976(struct mt7915_dev *dev, u8 adie,
425 			       bool is_40m, int *result)
426 {
427 	int ret;
428 	u32 data, addr;
429 
430 	addr = is_40m ? MT_ADIE_XTAL_AXM_40M_OSC : MT_ADIE_XTAL_AXM_80M_OSC;
431 	ret = mt7986_wmac_adie_efuse_read(dev, adie, addr, &data);
432 	if (ret)
433 		return ret;
434 
435 	if (!FIELD_GET(MT_ADIE_XO_TRIM_EN_MASK, data)) {
436 		*result = 64;
437 	} else {
438 		*result = FIELD_GET(MT_ADIE_TRIM_MASK, data);
439 		addr = is_40m ? MT_ADIE_XTAL_TRIM1_40M_OSC :
440 				MT_ADIE_XTAL_TRIM1_80M_OSC;
441 		ret = mt7986_wmac_adie_efuse_read(dev, adie, addr, &data);
442 		if (ret)
443 			return ret;
444 
445 		if (FIELD_GET(MT_ADIE_XO_TRIM_EN_MASK, data) &&
446 		    FIELD_GET(MT_ADIE_XTAL_DECREASE_MASK, data))
447 			*result -= FIELD_GET(MT_ADIE_EFUSE_TRIM_MASK, data);
448 		else if (FIELD_GET(MT_ADIE_XO_TRIM_EN_MASK, data))
449 			*result += FIELD_GET(MT_ADIE_EFUSE_TRIM_MASK, data);
450 
451 		*result = max(0, min(127, *result));
452 	}
453 
454 	return 0;
455 }
456 
457 static int mt7986_wmac_adie_xtal_trim_7976(struct mt7915_dev *dev, u8 adie)
458 {
459 	int ret, trim_80m, trim_40m;
460 	u32 data, val, mode;
461 
462 	ret = mt7986_wmac_adie_efuse_read(dev, adie, MT_ADIE_XO_TRIM_FLOW,
463 					  &data);
464 	if (ret || !FIELD_GET(BIT(1), data))
465 		return 0;
466 
467 	ret = mt7986_read_efuse_xo_trim_7976(dev, adie, false, &trim_80m);
468 	if (ret)
469 		return ret;
470 
471 	ret = mt7986_read_efuse_xo_trim_7976(dev, adie, true, &trim_40m);
472 	if (ret)
473 		return ret;
474 
475 	ret = mt76_wmac_spi_read(dev, adie, MT_ADIE_RG_STRAP_PIN_IN, &val);
476 	if (ret)
477 		return ret;
478 
479 	mode = FIELD_PREP(GENMASK(6, 4), val);
480 	if (!mode || mode == 0x2) {
481 		ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_XTAL_C1,
482 					GENMASK(31, 24),
483 					FIELD_PREP(GENMASK(31, 24), trim_80m));
484 		if (ret)
485 			return ret;
486 
487 		ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_XTAL_C2,
488 					GENMASK(31, 24),
489 					FIELD_PREP(GENMASK(31, 24), trim_80m));
490 	} else if (mode == 0x3 || mode == 0x4 || mode == 0x6) {
491 		ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_XTAL_C1,
492 					GENMASK(23, 16),
493 					FIELD_PREP(GENMASK(23, 16), trim_40m));
494 		if (ret)
495 			return ret;
496 
497 		ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_XTAL_C2,
498 					GENMASK(23, 16),
499 					FIELD_PREP(GENMASK(23, 16), trim_40m));
500 	}
501 
502 	return ret;
503 }
504 
505 static int mt798x_wmac_adie_patch_7976(struct mt7915_dev *dev, u8 adie)
506 {
507 	u32 id, version, rg_xo_01, rg_xo_03;
508 	int ret;
509 
510 	ret = mt76_wmac_spi_read(dev, adie, MT_ADIE_CHIP_ID, &id);
511 	if (ret)
512 		return ret;
513 
514 	version = FIELD_GET(MT_ADIE_VERSION_MASK, id);
515 
516 	ret = mt76_wmac_spi_write(dev, adie, MT_ADIE_RG_TOP_THADC, 0x4a563b00);
517 	if (ret)
518 		return ret;
519 
520 	if (version == 0x8a00 || version == 0x8a10 || version == 0x8b00) {
521 		rg_xo_01 = 0x1d59080f;
522 		rg_xo_03 = 0x34c00fe0;
523 	} else {
524 		if (is_mt7981(&dev->mt76)) {
525 			rg_xo_01 = 0x1959c80f;
526 		} else if (is_mt7986(&dev->mt76)) {
527 			rg_xo_01 = 0x1959f80f;
528 		} else {
529 			WARN_ON(1);
530 			return -EINVAL;
531 		}
532 		rg_xo_03 = 0x34d00fe0;
533 	}
534 
535 	ret = mt76_wmac_spi_write(dev, adie, MT_ADIE_RG_XO_01, rg_xo_01);
536 	if (ret)
537 		return ret;
538 
539 	return mt76_wmac_spi_write(dev, adie, MT_ADIE_RG_XO_03, rg_xo_03);
540 }
541 
542 static int
543 mt7986_read_efuse_xo_trim_7975(struct mt7915_dev *dev, u8 adie,
544 			       u32 addr, u32 *result)
545 {
546 	int ret;
547 	u32 data;
548 
549 	ret = mt7986_wmac_adie_efuse_read(dev, adie, addr, &data);
550 	if (ret)
551 		return ret;
552 
553 	if ((data & MT_ADIE_XO_TRIM_EN_MASK)) {
554 		if ((data & MT_ADIE_XTAL_DECREASE_MASK))
555 			*result -= (data & MT_ADIE_EFUSE_TRIM_MASK);
556 		else
557 			*result += (data & MT_ADIE_EFUSE_TRIM_MASK);
558 
559 		*result = (*result & MT_ADIE_TRIM_MASK);
560 	}
561 
562 	return 0;
563 }
564 
565 static int mt7986_wmac_adie_xtal_trim_7975(struct mt7915_dev *dev, u8 adie)
566 {
567 	int ret;
568 	u32 data, result = 0, value;
569 
570 	ret = mt7986_wmac_adie_efuse_read(dev, adie, MT_ADIE_7975_XTAL_EN,
571 					  &data);
572 	if (ret || !(data & BIT(1)))
573 		return 0;
574 
575 	ret = mt7986_wmac_adie_efuse_read(dev, adie, MT_ADIE_7975_XTAL_CAL,
576 					  &data);
577 	if (ret)
578 		return ret;
579 
580 	if (data & MT_ADIE_XO_TRIM_EN_MASK)
581 		result = (data & MT_ADIE_TRIM_MASK);
582 
583 	ret = mt7986_read_efuse_xo_trim_7975(dev, adie, MT_ADIE_7975_XO_TRIM2,
584 					     &result);
585 	if (ret)
586 		return ret;
587 
588 	ret = mt7986_read_efuse_xo_trim_7975(dev, adie, MT_ADIE_7975_XO_TRIM3,
589 					     &result);
590 	if (ret)
591 		return ret;
592 
593 	ret = mt7986_read_efuse_xo_trim_7975(dev, adie, MT_ADIE_7975_XO_TRIM4,
594 					     &result);
595 	if (ret)
596 		return ret;
597 
598 	/* Update trim value to C1 and C2*/
599 	value = FIELD_GET(MT_ADIE_7975_XO_CTRL2_C1_MASK, result) |
600 		FIELD_GET(MT_ADIE_7975_XO_CTRL2_C2_MASK, result);
601 	ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_7975_XO_CTRL2,
602 				MT_ADIE_7975_XO_CTRL2_MASK, value);
603 	if (ret)
604 		return ret;
605 
606 	ret = mt76_wmac_spi_read(dev, adie, MT_ADIE_7975_XTAL, &value);
607 	if (ret)
608 		return ret;
609 
610 	if (value & MT_ADIE_7975_XTAL_EN_MASK) {
611 		ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_7975_XO_2,
612 					MT_ADIE_7975_XO_2_FIX_EN, 0x0);
613 		if (ret)
614 			return ret;
615 	}
616 
617 	return mt76_wmac_spi_rmw(dev, adie, MT_ADIE_7975_XO_CTRL6,
618 				 MT_ADIE_7975_XO_CTRL6_MASK, 0x1);
619 }
620 
621 static int mt7986_wmac_adie_patch_7975(struct mt7915_dev *dev, u8 adie)
622 {
623 	int ret;
624 
625 	/* disable CAL LDO and fine tune RFDIG LDO */
626 	ret = mt76_wmac_spi_write(dev, adie, 0x348, 0x00000002);
627 	if (ret)
628 		return ret;
629 
630 	ret = mt76_wmac_spi_write(dev, adie, 0x378, 0x00000002);
631 	if (ret)
632 		return ret;
633 
634 	ret = mt76_wmac_spi_write(dev, adie, 0x3a8, 0x00000002);
635 	if (ret)
636 		return ret;
637 
638 	ret = mt76_wmac_spi_write(dev, adie, 0x3d8, 0x00000002);
639 	if (ret)
640 		return ret;
641 
642 	/* set CKA driving and filter */
643 	ret = mt76_wmac_spi_write(dev, adie, 0xa1c, 0x30000aaa);
644 	if (ret)
645 		return ret;
646 
647 	/* set CKB LDO to 1.4V */
648 	ret = mt76_wmac_spi_write(dev, adie, 0xa84, 0x8470008a);
649 	if (ret)
650 		return ret;
651 
652 	/* turn on SX0 LTBUF */
653 	if (is_mt7981(&dev->mt76)) {
654 		ret = mt76_wmac_spi_write(dev, adie, 0x074, 0x00000007);
655 	} else if (is_mt7986(&dev->mt76)) {
656 		ret = mt76_wmac_spi_write(dev, adie, 0x074, 0x00000002);
657 	} else {
658 		WARN_ON(1);
659 		return -EINVAL;
660 	}
661 
662 	if (ret)
663 		return ret;
664 
665 	/* CK_BUF_SW_EN = 1 (all buf in manual mode.) */
666 	ret = mt76_wmac_spi_write(dev, adie, 0xaa4, 0x01001fc0);
667 	if (ret)
668 		return ret;
669 
670 	/* BT mode/WF normal mode 00000005 */
671 	ret = mt76_wmac_spi_write(dev, adie, 0x070, 0x00000005);
672 	if (ret)
673 		return ret;
674 
675 	/* BG thermal sensor offset update */
676 	ret = mt76_wmac_spi_write(dev, adie, 0x344, 0x00000088);
677 	if (ret)
678 		return ret;
679 
680 	ret = mt76_wmac_spi_write(dev, adie, 0x374, 0x00000088);
681 	if (ret)
682 		return ret;
683 
684 	ret = mt76_wmac_spi_write(dev, adie, 0x3a4, 0x00000088);
685 	if (ret)
686 		return ret;
687 
688 	ret = mt76_wmac_spi_write(dev, adie, 0x3d4, 0x00000088);
689 	if (ret)
690 		return ret;
691 
692 	/* set WCON VDD IPTAT to "0000" */
693 	ret = mt76_wmac_spi_write(dev, adie, 0xa80, 0x44d07000);
694 	if (ret)
695 		return ret;
696 
697 	/* change back LTBUF SX3 drving to default value */
698 	ret = mt76_wmac_spi_write(dev, adie, 0xa88, 0x3900aaaa);
699 	if (ret)
700 		return ret;
701 
702 	/* SM input cap off */
703 	ret = mt76_wmac_spi_write(dev, adie, 0x2c4, 0x00000000);
704 	if (ret)
705 		return ret;
706 
707 	/* set CKB driving and filter */
708 	if (is_mt7986(&dev->mt76))
709 		return mt76_wmac_spi_write(dev, adie, 0x2c8, 0x00000072);
710 
711 	return ret;
712 }
713 
714 static int mt7986_wmac_adie_cfg(struct mt7915_dev *dev, u8 adie, u32 adie_type)
715 {
716 	int ret;
717 
718 	mt76_wmac_spi_lock(dev);
719 	ret = mt76_wmac_spi_write(dev, adie, MT_ADIE_CLK_EN, ~0);
720 	if (ret)
721 		goto out;
722 
723 	if (is_7975(dev, adie, adie_type)) {
724 		ret = mt76_wmac_spi_rmw(dev, adie, MT_ADIE_7975_COCLK,
725 					BIT(1), 0x1);
726 		if (ret)
727 			goto out;
728 
729 		ret = mt7986_wmac_adie_thermal_cal(dev, adie);
730 		if (ret)
731 			goto out;
732 
733 		ret = mt7986_wmac_adie_xtal_trim_7975(dev, adie);
734 		if (ret)
735 			goto out;
736 
737 		ret = mt7986_wmac_adie_patch_7975(dev, adie);
738 	} else if (is_7976(dev, adie, adie_type)) {
739 		if (mt798x_wmac_check_adie_type(dev) == ADIE_DBDC) {
740 			ret = mt76_wmac_spi_write(dev, adie,
741 						  MT_ADIE_WRI_CK_SEL, 0x1c);
742 			if (ret)
743 				goto out;
744 		}
745 
746 		ret = mt7986_wmac_adie_thermal_cal(dev, adie);
747 		if (ret)
748 			goto out;
749 
750 		ret = mt7986_wmac_adie_xtal_trim_7976(dev, adie);
751 		if (ret)
752 			goto out;
753 
754 		ret = mt798x_wmac_adie_patch_7976(dev, adie);
755 	}
756 out:
757 	mt76_wmac_spi_unlock(dev);
758 
759 	return ret;
760 }
761 
762 static int
763 mt7986_wmac_afe_cal(struct mt7915_dev *dev, u8 adie, bool dbdc, u32 adie_type)
764 {
765 	int ret;
766 	u8 idx;
767 	u32 txcal;
768 
769 	mt76_wmac_spi_lock(dev);
770 	if (is_7975(dev, adie, adie_type))
771 		ret = mt76_wmac_spi_write(dev, adie,
772 					  MT_AFE_RG_ENCAL_WBTAC_IF_SW,
773 					  0x80000000);
774 	else
775 		ret = mt76_wmac_spi_write(dev, adie,
776 					  MT_AFE_RG_ENCAL_WBTAC_IF_SW,
777 					  0x88888005);
778 	if (ret)
779 		goto out;
780 
781 	idx = dbdc ? ADIE_DBDC : adie;
782 
783 	mt76_rmw_field(dev, MT_AFE_DIG_EN_01(idx),
784 		       MT_AFE_RG_WBG_EN_RCK_MASK, 0x1);
785 	usleep_range(60, 100);
786 
787 	mt76_rmw(dev, MT_AFE_DIG_EN_01(idx),
788 		 MT_AFE_RG_WBG_EN_RCK_MASK, 0x0);
789 
790 	mt76_rmw_field(dev, MT_AFE_DIG_EN_03(idx),
791 		       MT_AFE_RG_WBG_EN_BPLL_UP_MASK, 0x1);
792 	usleep_range(30, 100);
793 
794 	mt76_rmw_field(dev, MT_AFE_DIG_EN_03(idx),
795 		       MT_AFE_RG_WBG_EN_WPLL_UP_MASK, 0x1);
796 	usleep_range(60, 100);
797 
798 	txcal = (MT_AFE_RG_WBG_EN_TXCAL_BT |
799 		      MT_AFE_RG_WBG_EN_TXCAL_WF0 |
800 		      MT_AFE_RG_WBG_EN_TXCAL_WF1 |
801 		      MT_AFE_RG_WBG_EN_TXCAL_WF2 |
802 		      MT_AFE_RG_WBG_EN_TXCAL_WF3);
803 	if (is_mt7981(&dev->mt76))
804 		txcal |= MT_AFE_RG_WBG_EN_TXCAL_WF4;
805 
806 	mt76_set(dev, MT_AFE_DIG_EN_01(idx), txcal);
807 	usleep_range(800, 1000);
808 
809 	mt76_clear(dev, MT_AFE_DIG_EN_01(idx), txcal);
810 	mt76_rmw(dev, MT_AFE_DIG_EN_03(idx),
811 		 MT_AFE_RG_WBG_EN_PLL_UP_MASK, 0x0);
812 
813 	ret = mt76_wmac_spi_write(dev, adie, MT_AFE_RG_ENCAL_WBTAC_IF_SW,
814 				  0x5);
815 
816 out:
817 	mt76_wmac_spi_unlock(dev);
818 
819 	return ret;
820 }
821 
822 static void mt7986_wmac_subsys_pll_initial(struct mt7915_dev *dev, u8 band)
823 {
824 	mt76_rmw(dev, MT_AFE_PLL_STB_TIME(band),
825 		 MT_AFE_PLL_STB_TIME_MASK, MT_AFE_PLL_STB_TIME_VAL);
826 
827 	mt76_rmw(dev, MT_AFE_DIG_EN_02(band),
828 		 MT_AFE_PLL_CFG_MASK, MT_AFE_PLL_CFG_VAL);
829 
830 	mt76_rmw(dev, MT_AFE_DIG_TOP_01(band),
831 		 MT_AFE_DIG_TOP_01_MASK, MT_AFE_DIG_TOP_01_VAL);
832 }
833 
834 static void mt7986_wmac_subsys_setting(struct mt7915_dev *dev)
835 {
836 	/* Subsys pll init */
837 	mt7986_wmac_subsys_pll_initial(dev, 0);
838 	mt7986_wmac_subsys_pll_initial(dev, 1);
839 
840 	/* Set legacy OSC control stable time*/
841 	mt76_rmw(dev, MT_CONN_INFRA_OSC_RC_EN,
842 		 MT_CONN_INFRA_OSC_RC_EN_MASK, 0x0);
843 	mt76_rmw(dev, MT_CONN_INFRA_OSC_CTRL,
844 		 MT_CONN_INFRA_OSC_STB_TIME_MASK, 0x80706);
845 
846 	/* prevent subsys from power on/of in a short time interval */
847 	mt76_rmw(dev, MT_TOP_WFSYS_PWR,
848 		 MT_TOP_PWR_ACK_MASK | MT_TOP_PWR_KEY_MASK,
849 		 MT_TOP_PWR_KEY);
850 }
851 
852 static int mt7986_wmac_bus_timeout(struct mt7915_dev *dev)
853 {
854 	mt76_rmw_field(dev, MT_INFRA_BUS_OFF_TIMEOUT,
855 		       MT_INFRA_BUS_TIMEOUT_LIMIT_MASK, 0x2);
856 
857 	mt76_rmw_field(dev, MT_INFRA_BUS_OFF_TIMEOUT,
858 		       MT_INFRA_BUS_TIMEOUT_EN_MASK, 0xf);
859 
860 	mt76_rmw_field(dev, MT_INFRA_BUS_ON_TIMEOUT,
861 		       MT_INFRA_BUS_TIMEOUT_LIMIT_MASK, 0xc);
862 
863 	mt76_rmw_field(dev, MT_INFRA_BUS_ON_TIMEOUT,
864 		       MT_INFRA_BUS_TIMEOUT_EN_MASK, 0xf);
865 
866 	return mt798x_wmac_coninfra_check(dev);
867 }
868 
869 static void mt7986_wmac_clock_enable(struct mt7915_dev *dev, u32 adie_type)
870 {
871 	u32 cur;
872 
873 	mt76_rmw_field(dev, MT_INFRA_CKGEN_BUS_WPLL_DIV_1,
874 		       MT_INFRA_CKGEN_DIV_SEL_MASK, 0x1);
875 
876 	mt76_rmw_field(dev, MT_INFRA_CKGEN_BUS_WPLL_DIV_2,
877 		       MT_INFRA_CKGEN_DIV_SEL_MASK, 0x1);
878 
879 	mt76_rmw_field(dev, MT_INFRA_CKGEN_BUS_WPLL_DIV_1,
880 		       MT_INFRA_CKGEN_DIV_EN_MASK, 0x1);
881 
882 	mt76_rmw_field(dev, MT_INFRA_CKGEN_BUS_WPLL_DIV_2,
883 		       MT_INFRA_CKGEN_DIV_EN_MASK, 0x1);
884 
885 	mt76_rmw_field(dev, MT_INFRA_CKGEN_RFSPI_WPLL_DIV,
886 		       MT_INFRA_CKGEN_DIV_SEL_MASK, 0x8);
887 
888 	mt76_rmw_field(dev, MT_INFRA_CKGEN_RFSPI_WPLL_DIV,
889 		       MT_INFRA_CKGEN_DIV_EN_MASK, 0x1);
890 
891 	mt76_rmw_field(dev, MT_INFRA_CKGEN_BUS,
892 		       MT_INFRA_CKGEN_BUS_CLK_SEL_MASK, 0x0);
893 
894 	mt76_rmw_field(dev, MT_CONN_INFRA_HW_CTRL,
895 		       MT_CONN_INFRA_HW_CTRL_MASK, 0x1);
896 
897 	mt76_rmw(dev, MT_TOP_CONN_INFRA_WAKEUP,
898 		 MT_TOP_CONN_INFRA_WAKEUP_MASK, 0x1);
899 
900 	usleep_range(900, 1000);
901 
902 	mt76_wmac_spi_lock(dev);
903 	if (is_7975(dev, 0, adie_type) || is_7976(dev, 0, adie_type)) {
904 		mt76_rmw_field(dev, MT_ADIE_SLP_CTRL_CK0(0),
905 			       MT_SLP_CTRL_EN_MASK, 0x1);
906 
907 		read_poll_timeout(mt76_rr, cur, !(cur & MT_SLP_CTRL_BSY_MASK),
908 				  USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
909 				  dev, MT_ADIE_SLP_CTRL_CK0(0));
910 	}
911 	if (is_7975(dev, 1, adie_type) || is_7976(dev, 1, adie_type)) {
912 		mt76_rmw_field(dev, MT_ADIE_SLP_CTRL_CK0(1),
913 			       MT_SLP_CTRL_EN_MASK, 0x1);
914 
915 		read_poll_timeout(mt76_rr, cur, !(cur & MT_SLP_CTRL_BSY_MASK),
916 				  USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
917 				  dev, MT_ADIE_SLP_CTRL_CK0(0));
918 	}
919 	mt76_wmac_spi_unlock(dev);
920 
921 	mt76_rmw(dev, MT_TOP_CONN_INFRA_WAKEUP,
922 		 MT_TOP_CONN_INFRA_WAKEUP_MASK, 0x0);
923 	usleep_range(900, 1000);
924 }
925 
926 static int mt7986_wmac_top_wfsys_wakeup(struct mt7915_dev *dev, bool enable)
927 {
928 	mt76_rmw_field(dev, MT_TOP_WFSYS_WAKEUP,
929 		       MT_TOP_WFSYS_WAKEUP_MASK, enable);
930 
931 	usleep_range(900, 1000);
932 
933 	if (!enable)
934 		return 0;
935 
936 	return mt798x_wmac_coninfra_check(dev);
937 }
938 
939 static int mt7986_wmac_wm_enable(struct mt7915_dev *dev, bool enable)
940 {
941 	u32 cur;
942 
943 	if (is_mt7986(&dev->mt76))
944 		mt76_wr(dev, MT_CONNINFRA_SKU_DEC_ADDR, 0);
945 
946 	mt76_rmw_field(dev, MT7986_TOP_WM_RESET,
947 		       MT7986_TOP_WM_RESET_MASK, enable);
948 	if (!enable)
949 		return 0;
950 
951 	return read_poll_timeout(mt76_rr, cur, (cur == 0x1d1e),
952 				 USEC_PER_MSEC, 5000 * USEC_PER_MSEC, false,
953 				 dev, MT_TOP_CFG_ON_ROM_IDX);
954 }
955 
956 static int mt7986_wmac_wfsys_poweron(struct mt7915_dev *dev, bool enable)
957 {
958 	u32 mask = MT_TOP_PWR_EN_MASK | MT_TOP_PWR_KEY_MASK;
959 	u32 cur;
960 
961 	mt76_rmw(dev, MT_TOP_WFSYS_PWR, mask,
962 		 MT_TOP_PWR_KEY | FIELD_PREP(MT_TOP_PWR_EN_MASK, enable));
963 
964 	return read_poll_timeout(mt76_rr, cur,
965 		(FIELD_GET(MT_TOP_WFSYS_RESET_STATUS_MASK, cur) == enable),
966 		USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
967 		dev, MT_TOP_WFSYS_RESET_STATUS);
968 }
969 
970 static int mt7986_wmac_wfsys_setting(struct mt7915_dev *dev)
971 {
972 	int ret;
973 	u32 cur;
974 
975 	/* Turn off wfsys2conn bus sleep protect */
976 	mt76_rmw(dev, MT_CONN_INFRA_WF_SLP_PROT,
977 		 MT_CONN_INFRA_WF_SLP_PROT_MASK, 0x0);
978 
979 	ret = mt7986_wmac_wfsys_poweron(dev, true);
980 	if (ret)
981 		return ret;
982 
983 	/* Check bus sleep protect */
984 
985 	ret = read_poll_timeout(mt76_rr, cur,
986 				!(cur & MT_CONN_INFRA_CONN_WF_MASK),
987 				USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
988 				dev, MT_CONN_INFRA_WF_SLP_PROT_RDY);
989 	if (ret)
990 		return ret;
991 
992 	ret = read_poll_timeout(mt76_rr, cur, !(cur & MT_SLP_WFDMA2CONN_MASK),
993 				USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
994 				dev, MT_SLP_STATUS);
995 	if (ret)
996 		return ret;
997 
998 	return read_poll_timeout(mt76_rr, cur, (cur == 0x02060000),
999 				 USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
1000 				 dev, MT_TOP_CFG_IP_VERSION_ADDR);
1001 }
1002 
1003 static void mt7986_wmac_wfsys_set_timeout(struct mt7915_dev *dev)
1004 {
1005 	u32 mask = MT_MCU_BUS_TIMEOUT_SET_MASK |
1006 		   MT_MCU_BUS_TIMEOUT_CG_EN_MASK |
1007 		   MT_MCU_BUS_TIMEOUT_EN_MASK;
1008 	u32 val = FIELD_PREP(MT_MCU_BUS_TIMEOUT_SET_MASK, 1) |
1009 		  FIELD_PREP(MT_MCU_BUS_TIMEOUT_CG_EN_MASK, 1) |
1010 		  FIELD_PREP(MT_MCU_BUS_TIMEOUT_EN_MASK, 1);
1011 
1012 	mt76_rmw(dev, MT_MCU_BUS_TIMEOUT, mask, val);
1013 
1014 	mt76_wr(dev, MT_MCU_BUS_REMAP, 0x810f0000);
1015 
1016 	mask = MT_MCU_BUS_DBG_TIMEOUT_SET_MASK |
1017 	       MT_MCU_BUS_DBG_TIMEOUT_CK_EN_MASK |
1018 	       MT_MCU_BUS_DBG_TIMEOUT_EN_MASK;
1019 	val = FIELD_PREP(MT_MCU_BUS_DBG_TIMEOUT_SET_MASK, 0x3aa) |
1020 	      FIELD_PREP(MT_MCU_BUS_DBG_TIMEOUT_CK_EN_MASK, 1) |
1021 	      FIELD_PREP(MT_MCU_BUS_DBG_TIMEOUT_EN_MASK, 1);
1022 
1023 	mt76_rmw(dev, MT_MCU_BUS_DBG_TIMEOUT, mask, val);
1024 }
1025 
1026 static int mt7986_wmac_sku_update(struct mt7915_dev *dev, u32 adie_type)
1027 {
1028 	u32 val;
1029 
1030 	if (is_7976(dev, 0, adie_type) && is_7976(dev, 1, adie_type))
1031 		val = 0xf;
1032 	else if (is_7975(dev, 0, adie_type) && is_7975(dev, 1, adie_type))
1033 		val = 0xd;
1034 	else if (is_7976(dev, 0, adie_type))
1035 		val = 0x7;
1036 	else if (is_7975(dev, 1, adie_type))
1037 		val = 0x8;
1038 	else if (is_7976(dev, 1, adie_type))
1039 		val = 0xa;
1040 	else
1041 		return -EINVAL;
1042 
1043 	mt76_wmac_rmw(dev->sku, MT_TOP_POS_SKU, MT_TOP_POS_SKU_MASK,
1044 		      FIELD_PREP(MT_TOP_POS_SKU_MASK, val));
1045 
1046 	mt76_wr(dev, MT_CONNINFRA_SKU_DEC_ADDR, val);
1047 
1048 	return 0;
1049 }
1050 
1051 static int
1052 mt7986_wmac_adie_setup(struct mt7915_dev *dev, u8 adie, u32 adie_type)
1053 {
1054 	int ret;
1055 
1056 	if (!(is_7975(dev, adie, adie_type) || is_7976(dev, adie, adie_type)))
1057 		return 0;
1058 
1059 	ret = mt7986_wmac_adie_cfg(dev, adie, adie_type);
1060 	if (ret)
1061 		return ret;
1062 
1063 	ret = mt7986_wmac_afe_cal(dev, adie, false, adie_type);
1064 	if (ret)
1065 		return ret;
1066 
1067 	if (!adie && (mt798x_wmac_check_adie_type(dev) == ADIE_DBDC))
1068 		ret = mt7986_wmac_afe_cal(dev, adie, true, adie_type);
1069 
1070 	return ret;
1071 }
1072 
1073 static int mt7986_wmac_subsys_powerup(struct mt7915_dev *dev, u32 adie_type)
1074 {
1075 	int ret;
1076 
1077 	mt7986_wmac_subsys_setting(dev);
1078 
1079 	ret = mt7986_wmac_bus_timeout(dev);
1080 	if (ret)
1081 		return ret;
1082 
1083 	mt7986_wmac_clock_enable(dev, adie_type);
1084 
1085 	return 0;
1086 }
1087 
1088 static int mt7986_wmac_wfsys_powerup(struct mt7915_dev *dev)
1089 {
1090 	int ret;
1091 
1092 	ret = mt7986_wmac_wm_enable(dev, false);
1093 	if (ret)
1094 		return ret;
1095 
1096 	ret = mt7986_wmac_wfsys_setting(dev);
1097 	if (ret)
1098 		return ret;
1099 
1100 	mt7986_wmac_wfsys_set_timeout(dev);
1101 
1102 	return mt7986_wmac_wm_enable(dev, true);
1103 }
1104 
1105 int mt7986_wmac_enable(struct mt7915_dev *dev)
1106 {
1107 	int ret;
1108 	u32 adie_type;
1109 
1110 	ret = mt7986_wmac_consys_reset(dev, true);
1111 	if (ret)
1112 		return ret;
1113 
1114 	ret = mt7986_wmac_gpio_setup(dev);
1115 	if (ret)
1116 		return ret;
1117 
1118 	ret = mt7986_wmac_consys_lockup(dev, false);
1119 	if (ret)
1120 		return ret;
1121 
1122 	ret = mt798x_wmac_coninfra_check(dev);
1123 	if (ret)
1124 		return ret;
1125 
1126 	ret = mt798x_wmac_coninfra_setup(dev);
1127 	if (ret)
1128 		return ret;
1129 
1130 	ret = mt798x_wmac_sku_setup(dev, &adie_type);
1131 	if (ret)
1132 		return ret;
1133 
1134 	ret = mt7986_wmac_adie_setup(dev, 0, adie_type);
1135 	if (ret)
1136 		return ret;
1137 
1138 	/* mt7981 doesn't support a second a-die */
1139 	if (is_mt7986(&dev->mt76)) {
1140 		ret = mt7986_wmac_adie_setup(dev, 1, adie_type);
1141 		if (ret)
1142 			return ret;
1143 	}
1144 
1145 	ret = mt7986_wmac_subsys_powerup(dev, adie_type);
1146 	if (ret)
1147 		return ret;
1148 
1149 	ret = mt7986_wmac_top_wfsys_wakeup(dev, true);
1150 	if (ret)
1151 		return ret;
1152 
1153 	ret = mt7986_wmac_wfsys_powerup(dev);
1154 	if (ret)
1155 		return ret;
1156 
1157 	return mt7986_wmac_sku_update(dev, adie_type);
1158 }
1159 
1160 void mt7986_wmac_disable(struct mt7915_dev *dev)
1161 {
1162 	u32 cur;
1163 
1164 	mt7986_wmac_top_wfsys_wakeup(dev, true);
1165 
1166 	/* Turn on wfsys2conn bus sleep protect */
1167 	mt76_rmw_field(dev, MT_CONN_INFRA_WF_SLP_PROT,
1168 		       MT_CONN_INFRA_WF_SLP_PROT_MASK, 0x1);
1169 
1170 	/* Check wfsys2conn bus sleep protect */
1171 	read_poll_timeout(mt76_rr, cur, !(cur ^ MT_CONN_INFRA_CONN),
1172 			  USEC_PER_MSEC, 50 * USEC_PER_MSEC, false,
1173 			  dev, MT_CONN_INFRA_WF_SLP_PROT_RDY);
1174 
1175 	mt7986_wmac_wfsys_poweron(dev, false);
1176 
1177 	/* Turn back wpll setting */
1178 	mt76_rmw_field(dev, MT_AFE_DIG_EN_02(0), MT_AFE_MCU_BPLL_CFG_MASK, 0x2);
1179 	mt76_rmw_field(dev, MT_AFE_DIG_EN_02(0), MT_AFE_WPLL_CFG_MASK, 0x2);
1180 
1181 	/* Reset EMI */
1182 	mt76_rmw_field(dev, MT_CONN_INFRA_EMI_REQ,
1183 		       MT_CONN_INFRA_EMI_REQ_MASK, 0x1);
1184 	mt76_rmw_field(dev, MT_CONN_INFRA_EMI_REQ,
1185 		       MT_CONN_INFRA_EMI_REQ_MASK, 0x0);
1186 	mt76_rmw_field(dev, MT_CONN_INFRA_EMI_REQ,
1187 		       MT_CONN_INFRA_INFRA_REQ_MASK, 0x1);
1188 	mt76_rmw_field(dev, MT_CONN_INFRA_EMI_REQ,
1189 		       MT_CONN_INFRA_INFRA_REQ_MASK, 0x0);
1190 
1191 	mt7986_wmac_top_wfsys_wakeup(dev, false);
1192 	mt7986_wmac_consys_lockup(dev, true);
1193 	mt7986_wmac_consys_reset(dev, false);
1194 }
1195 
1196 static int mt798x_wmac_init(struct mt7915_dev *dev)
1197 {
1198 	struct device *pdev = dev->mt76.dev;
1199 	struct platform_device *pfdev = to_platform_device(pdev);
1200 	struct clk *mcu_clk, *ap_conn_clk;
1201 
1202 	mcu_clk = devm_clk_get(pdev, "mcu");
1203 	if (IS_ERR(mcu_clk))
1204 		dev_err(pdev, "mcu clock not found\n");
1205 	else if (clk_prepare_enable(mcu_clk))
1206 		dev_err(pdev, "mcu clock configuration failed\n");
1207 
1208 	ap_conn_clk = devm_clk_get(pdev, "ap2conn");
1209 	if (IS_ERR(ap_conn_clk))
1210 		dev_err(pdev, "ap2conn clock not found\n");
1211 	else if (clk_prepare_enable(ap_conn_clk))
1212 		dev_err(pdev, "ap2conn clock configuration failed\n");
1213 
1214 	dev->dcm = devm_platform_ioremap_resource(pfdev, 1);
1215 	if (IS_ERR(dev->dcm))
1216 		return PTR_ERR(dev->dcm);
1217 
1218 	dev->sku = devm_platform_ioremap_resource(pfdev, 2);
1219 	if (IS_ERR(dev->sku))
1220 		return PTR_ERR(dev->sku);
1221 
1222 	dev->rstc = devm_reset_control_get(pdev, "consys");
1223 	if (IS_ERR(dev->rstc))
1224 		return PTR_ERR(dev->rstc);
1225 
1226 	return 0;
1227 }
1228 
1229 static int mt798x_wmac_probe(struct platform_device *pdev)
1230 {
1231 	void __iomem *mem_base;
1232 	struct mt7915_dev *dev;
1233 	struct mt76_dev *mdev;
1234 	int irq, ret;
1235 	u32 chip_id;
1236 
1237 	chip_id = (uintptr_t)of_device_get_match_data(&pdev->dev);
1238 
1239 	mem_base = devm_platform_ioremap_resource(pdev, 0);
1240 	if (IS_ERR(mem_base)) {
1241 		dev_err(&pdev->dev, "Failed to get memory resource\n");
1242 		return PTR_ERR(mem_base);
1243 	}
1244 
1245 	dev = mt7915_mmio_probe(&pdev->dev, mem_base, chip_id);
1246 	if (IS_ERR(dev))
1247 		return PTR_ERR(dev);
1248 
1249 	mdev = &dev->mt76;
1250 	ret = mt7915_mmio_wed_init(dev, pdev, false, &irq);
1251 	if (ret < 0)
1252 		goto free_device;
1253 
1254 	if (!ret) {
1255 		irq = platform_get_irq(pdev, 0);
1256 		if (irq < 0) {
1257 			ret = irq;
1258 			goto free_device;
1259 		}
1260 	}
1261 
1262 	ret = devm_request_irq(mdev->dev, irq, mt7915_irq_handler,
1263 			       IRQF_SHARED, KBUILD_MODNAME, dev);
1264 	if (ret)
1265 		goto free_device;
1266 
1267 	ret = mt798x_wmac_init(dev);
1268 	if (ret)
1269 		goto free_irq;
1270 
1271 	mt7915_wfsys_reset(dev);
1272 
1273 	ret = mt7915_register_device(dev);
1274 	if (ret)
1275 		goto free_irq;
1276 
1277 	return 0;
1278 
1279 free_irq:
1280 	devm_free_irq(mdev->dev, irq, dev);
1281 free_device:
1282 	if (mtk_wed_device_active(&mdev->mmio.wed))
1283 		mtk_wed_device_detach(&mdev->mmio.wed);
1284 	mt76_free_device(mdev);
1285 
1286 	return ret;
1287 }
1288 
1289 static int mt798x_wmac_remove(struct platform_device *pdev)
1290 {
1291 	struct mt7915_dev *dev = platform_get_drvdata(pdev);
1292 
1293 	mt7915_unregister_device(dev);
1294 
1295 	return 0;
1296 }
1297 
1298 static const struct of_device_id mt798x_wmac_of_match[] = {
1299 	{ .compatible = "mediatek,mt7981-wmac", .data = (u32 *)0x7981 },
1300 	{ .compatible = "mediatek,mt7986-wmac", .data = (u32 *)0x7986 },
1301 	{},
1302 };
1303 
1304 MODULE_DEVICE_TABLE(of, mt798x_wmac_of_match);
1305 
1306 struct platform_driver mt798x_wmac_driver = {
1307 	.driver = {
1308 		.name = "mt798x-wmac",
1309 		.of_match_table = mt798x_wmac_of_match,
1310 	},
1311 	.probe = mt798x_wmac_probe,
1312 	.remove = mt798x_wmac_remove,
1313 };
1314 
1315 MODULE_FIRMWARE(MT7986_FIRMWARE_WA);
1316 MODULE_FIRMWARE(MT7986_FIRMWARE_WM);
1317 MODULE_FIRMWARE(MT7986_FIRMWARE_WM_MT7975);
1318 MODULE_FIRMWARE(MT7986_ROM_PATCH);
1319 MODULE_FIRMWARE(MT7986_ROM_PATCH_MT7975);
1320 
1321 MODULE_FIRMWARE(MT7981_FIRMWARE_WA);
1322 MODULE_FIRMWARE(MT7981_FIRMWARE_WM);
1323 MODULE_FIRMWARE(MT7981_ROM_PATCH);
1324