1 // SPDX-License-Identifier: ISC
2 /* Copyright (C) 2020 MediaTek Inc. */
3 
4 #include <linux/kernel.h>
5 #include <linux/module.h>
6 #include <linux/platform_device.h>
7 #include <linux/rtnetlink.h>
8 #include <linux/pci.h>
9 
10 #include "mt7915.h"
11 #include "mac.h"
12 #include "mcu.h"
13 #include "../trace.h"
14 #include "../dma.h"
15 
16 static bool wed_enable;
17 module_param(wed_enable, bool, 0644);
18 MODULE_PARM_DESC(wed_enable, "Enable Wireless Ethernet Dispatch support");
19 
20 static const u32 mt7915_reg[] = {
21 	[INT_SOURCE_CSR]		= 0xd7010,
22 	[INT_MASK_CSR]			= 0xd7014,
23 	[INT1_SOURCE_CSR]		= 0xd7088,
24 	[INT1_MASK_CSR]			= 0xd708c,
25 	[INT_MCU_CMD_SOURCE]		= 0xd51f0,
26 	[INT_MCU_CMD_EVENT]		= 0x3108,
27 	[WFDMA0_ADDR]			= 0xd4000,
28 	[WFDMA0_PCIE1_ADDR]		= 0xd8000,
29 	[WFDMA_EXT_CSR_ADDR]		= 0xd7000,
30 	[CBTOP1_PHY_END]		= 0x77ffffff,
31 	[INFRA_MCU_ADDR_END]		= 0x7c3fffff,
32 	[FW_ASSERT_STAT_ADDR]		= 0x219848,
33 	[FW_EXCEPT_TYPE_ADDR]		= 0x21987c,
34 	[FW_EXCEPT_COUNT_ADDR]		= 0x219848,
35 	[FW_CIRQ_COUNT_ADDR]		= 0x216f94,
36 	[FW_CIRQ_IDX_ADDR]		= 0x216ef8,
37 	[FW_CIRQ_LISR_ADDR]		= 0x2170ac,
38 	[FW_TASK_ID_ADDR]		= 0x216f90,
39 	[FW_TASK_IDX_ADDR]		= 0x216f9c,
40 	[FW_TASK_QID1_ADDR]		= 0x219680,
41 	[FW_TASK_QID2_ADDR]		= 0x219760,
42 	[FW_TASK_START_ADDR]		= 0x219558,
43 	[FW_TASK_END_ADDR]		= 0x219554,
44 	[FW_TASK_SIZE_ADDR]		= 0x219560,
45 	[FW_LAST_MSG_ID_ADDR]		= 0x216f70,
46 	[FW_EINT_INFO_ADDR]		= 0x219818,
47 	[FW_SCHED_INFO_ADDR]		= 0x219828,
48 	[SWDEF_BASE_ADDR]		= 0x41f200,
49 	[TXQ_WED_RING_BASE]		= 0xd7300,
50 	[RXQ_WED_RING_BASE]		= 0xd7410,
51 	[RXQ_WED_DATA_RING_BASE]	= 0xd4500,
52 };
53 
54 static const u32 mt7916_reg[] = {
55 	[INT_SOURCE_CSR]		= 0xd4200,
56 	[INT_MASK_CSR]			= 0xd4204,
57 	[INT1_SOURCE_CSR]		= 0xd8200,
58 	[INT1_MASK_CSR]			= 0xd8204,
59 	[INT_MCU_CMD_SOURCE]		= 0xd41f0,
60 	[INT_MCU_CMD_EVENT]		= 0x2108,
61 	[WFDMA0_ADDR]			= 0xd4000,
62 	[WFDMA0_PCIE1_ADDR]		= 0xd8000,
63 	[WFDMA_EXT_CSR_ADDR]		= 0xd7000,
64 	[CBTOP1_PHY_END]		= 0x7fffffff,
65 	[INFRA_MCU_ADDR_END]		= 0x7c085fff,
66 	[FW_ASSERT_STAT_ADDR]		= 0x02204c14,
67 	[FW_EXCEPT_TYPE_ADDR]		= 0x022051a4,
68 	[FW_EXCEPT_COUNT_ADDR]		= 0x022050bc,
69 	[FW_CIRQ_COUNT_ADDR]		= 0x022001ac,
70 	[FW_CIRQ_IDX_ADDR]		= 0x02204f84,
71 	[FW_CIRQ_LISR_ADDR]		= 0x022050d0,
72 	[FW_TASK_ID_ADDR]		= 0x0220406c,
73 	[FW_TASK_IDX_ADDR]		= 0x0220500c,
74 	[FW_TASK_QID1_ADDR]		= 0x022028c8,
75 	[FW_TASK_QID2_ADDR]		= 0x02202a38,
76 	[FW_TASK_START_ADDR]		= 0x0220286c,
77 	[FW_TASK_END_ADDR]		= 0x02202870,
78 	[FW_TASK_SIZE_ADDR]		= 0x02202878,
79 	[FW_LAST_MSG_ID_ADDR]		= 0x02204fe8,
80 	[FW_EINT_INFO_ADDR]		= 0x0220525c,
81 	[FW_SCHED_INFO_ADDR]		= 0x0220516c,
82 	[SWDEF_BASE_ADDR]		= 0x411400,
83 	[TXQ_WED_RING_BASE]		= 0xd7300,
84 	[RXQ_WED_RING_BASE]		= 0xd7410,
85 	[RXQ_WED_DATA_RING_BASE]	= 0xd4540,
86 };
87 
88 static const u32 mt7986_reg[] = {
89 	[INT_SOURCE_CSR]		= 0x24200,
90 	[INT_MASK_CSR]			= 0x24204,
91 	[INT1_SOURCE_CSR]		= 0x28200,
92 	[INT1_MASK_CSR]			= 0x28204,
93 	[INT_MCU_CMD_SOURCE]		= 0x241f0,
94 	[INT_MCU_CMD_EVENT]		= 0x54000108,
95 	[WFDMA0_ADDR]			= 0x24000,
96 	[WFDMA0_PCIE1_ADDR]		= 0x28000,
97 	[WFDMA_EXT_CSR_ADDR]		= 0x27000,
98 	[CBTOP1_PHY_END]		= 0x7fffffff,
99 	[INFRA_MCU_ADDR_END]		= 0x7c085fff,
100 	[FW_ASSERT_STAT_ADDR]		= 0x02204b54,
101 	[FW_EXCEPT_TYPE_ADDR]		= 0x022050dc,
102 	[FW_EXCEPT_COUNT_ADDR]		= 0x02204ffc,
103 	[FW_CIRQ_COUNT_ADDR]		= 0x022001ac,
104 	[FW_CIRQ_IDX_ADDR]		= 0x02204ec4,
105 	[FW_CIRQ_LISR_ADDR]		= 0x02205010,
106 	[FW_TASK_ID_ADDR]		= 0x02204fac,
107 	[FW_TASK_IDX_ADDR]		= 0x02204f4c,
108 	[FW_TASK_QID1_ADDR]		= 0x02202814,
109 	[FW_TASK_QID2_ADDR]		= 0x02202984,
110 	[FW_TASK_START_ADDR]		= 0x022027b8,
111 	[FW_TASK_END_ADDR]		= 0x022027bc,
112 	[FW_TASK_SIZE_ADDR]		= 0x022027c4,
113 	[FW_LAST_MSG_ID_ADDR]		= 0x02204f28,
114 	[FW_EINT_INFO_ADDR]		= 0x02205194,
115 	[FW_SCHED_INFO_ADDR]		= 0x022051a4,
116 	[SWDEF_BASE_ADDR]		= 0x411400,
117 	[TXQ_WED_RING_BASE]		= 0x24420,
118 	[RXQ_WED_RING_BASE]		= 0x24520,
119 	[RXQ_WED_DATA_RING_BASE]	= 0x24540,
120 };
121 
122 static const u32 mt7915_offs[] = {
123 	[TMAC_CDTR]		= 0x090,
124 	[TMAC_ODTR]		= 0x094,
125 	[TMAC_ATCR]		= 0x098,
126 	[TMAC_TRCR0]		= 0x09c,
127 	[TMAC_ICR0]		= 0x0a4,
128 	[TMAC_ICR1]		= 0x0b4,
129 	[TMAC_CTCR0]		= 0x0f4,
130 	[TMAC_TFCR0]		= 0x1e0,
131 	[MDP_BNRCFR0]		= 0x070,
132 	[MDP_BNRCFR1]		= 0x074,
133 	[ARB_DRNGR0]		= 0x194,
134 	[ARB_SCR]		= 0x080,
135 	[RMAC_MIB_AIRTIME14]	= 0x3b8,
136 	[AGG_AWSCR0]		= 0x05c,
137 	[AGG_PCR0]		= 0x06c,
138 	[AGG_ACR0]		= 0x084,
139 	[AGG_ACR4]		= 0x08c,
140 	[AGG_MRCR]		= 0x098,
141 	[AGG_ATCR1]		= 0x0f0,
142 	[AGG_ATCR3]		= 0x0f4,
143 	[LPON_UTTR0]		= 0x080,
144 	[LPON_UTTR1]		= 0x084,
145 	[LPON_FRCR]		= 0x314,
146 	[MIB_SDR3]		= 0x014,
147 	[MIB_SDR4]		= 0x018,
148 	[MIB_SDR5]		= 0x01c,
149 	[MIB_SDR7]		= 0x024,
150 	[MIB_SDR8]		= 0x028,
151 	[MIB_SDR9]		= 0x02c,
152 	[MIB_SDR10]		= 0x030,
153 	[MIB_SDR11]		= 0x034,
154 	[MIB_SDR12]		= 0x038,
155 	[MIB_SDR13]		= 0x03c,
156 	[MIB_SDR14]		= 0x040,
157 	[MIB_SDR15]		= 0x044,
158 	[MIB_SDR16]		= 0x048,
159 	[MIB_SDR17]		= 0x04c,
160 	[MIB_SDR18]		= 0x050,
161 	[MIB_SDR19]		= 0x054,
162 	[MIB_SDR20]		= 0x058,
163 	[MIB_SDR21]		= 0x05c,
164 	[MIB_SDR22]		= 0x060,
165 	[MIB_SDR23]		= 0x064,
166 	[MIB_SDR24]		= 0x068,
167 	[MIB_SDR25]		= 0x06c,
168 	[MIB_SDR27]		= 0x074,
169 	[MIB_SDR28]		= 0x078,
170 	[MIB_SDR29]		= 0x07c,
171 	[MIB_SDRVEC]		= 0x080,
172 	[MIB_SDR31]		= 0x084,
173 	[MIB_SDR32]		= 0x088,
174 	[MIB_SDRMUBF]		= 0x090,
175 	[MIB_DR8]		= 0x0c0,
176 	[MIB_DR9]		= 0x0c4,
177 	[MIB_DR11]		= 0x0cc,
178 	[MIB_MB_SDR0]		= 0x100,
179 	[MIB_MB_SDR1]		= 0x104,
180 	[TX_AGG_CNT]		= 0x0a8,
181 	[TX_AGG_CNT2]		= 0x164,
182 	[MIB_ARNG]		= 0x4b8,
183 	[WTBLON_TOP_WDUCR]	= 0x0,
184 	[WTBL_UPDATE]		= 0x030,
185 	[PLE_FL_Q_EMPTY]	= 0x0b0,
186 	[PLE_FL_Q_CTRL]		= 0x1b0,
187 	[PLE_AC_QEMPTY]		= 0x500,
188 	[PLE_FREEPG_CNT]	= 0x100,
189 	[PLE_FREEPG_HEAD_TAIL]	= 0x104,
190 	[PLE_PG_HIF_GROUP]	= 0x110,
191 	[PLE_HIF_PG_INFO]	= 0x114,
192 	[AC_OFFSET]		= 0x040,
193 	[ETBF_PAR_RPT0]		= 0x068,
194 };
195 
196 static const u32 mt7916_offs[] = {
197 	[TMAC_CDTR]		= 0x0c8,
198 	[TMAC_ODTR]		= 0x0cc,
199 	[TMAC_ATCR]		= 0x00c,
200 	[TMAC_TRCR0]		= 0x010,
201 	[TMAC_ICR0]		= 0x014,
202 	[TMAC_ICR1]		= 0x018,
203 	[TMAC_CTCR0]		= 0x114,
204 	[TMAC_TFCR0]		= 0x0e4,
205 	[MDP_BNRCFR0]		= 0x090,
206 	[MDP_BNRCFR1]		= 0x094,
207 	[ARB_DRNGR0]		= 0x1e0,
208 	[ARB_SCR]		= 0x000,
209 	[RMAC_MIB_AIRTIME14]	= 0x0398,
210 	[AGG_AWSCR0]		= 0x030,
211 	[AGG_PCR0]		= 0x040,
212 	[AGG_ACR0]		= 0x054,
213 	[AGG_ACR4]		= 0x05c,
214 	[AGG_MRCR]		= 0x068,
215 	[AGG_ATCR1]		= 0x1a8,
216 	[AGG_ATCR3]		= 0x080,
217 	[LPON_UTTR0]		= 0x360,
218 	[LPON_UTTR1]		= 0x364,
219 	[LPON_FRCR]		= 0x37c,
220 	[MIB_SDR3]		= 0x698,
221 	[MIB_SDR4]		= 0x788,
222 	[MIB_SDR5]		= 0x780,
223 	[MIB_SDR7]		= 0x5a8,
224 	[MIB_SDR8]		= 0x78c,
225 	[MIB_SDR9]		= 0x024,
226 	[MIB_SDR10]		= 0x76c,
227 	[MIB_SDR11]		= 0x790,
228 	[MIB_SDR12]		= 0x558,
229 	[MIB_SDR13]		= 0x560,
230 	[MIB_SDR14]		= 0x564,
231 	[MIB_SDR15]		= 0x568,
232 	[MIB_SDR16]		= 0x7fc,
233 	[MIB_SDR17]		= 0x800,
234 	[MIB_SDR18]		= 0x030,
235 	[MIB_SDR19]		= 0x5ac,
236 	[MIB_SDR20]		= 0x5b0,
237 	[MIB_SDR21]		= 0x5b4,
238 	[MIB_SDR22]		= 0x770,
239 	[MIB_SDR23]		= 0x774,
240 	[MIB_SDR24]		= 0x778,
241 	[MIB_SDR25]		= 0x77c,
242 	[MIB_SDR27]		= 0x080,
243 	[MIB_SDR28]		= 0x084,
244 	[MIB_SDR29]		= 0x650,
245 	[MIB_SDRVEC]		= 0x5a8,
246 	[MIB_SDR31]		= 0x55c,
247 	[MIB_SDR32]		= 0x7a8,
248 	[MIB_SDRMUBF]		= 0x7ac,
249 	[MIB_DR8]		= 0x56c,
250 	[MIB_DR9]		= 0x570,
251 	[MIB_DR11]		= 0x574,
252 	[MIB_MB_SDR0]		= 0x688,
253 	[MIB_MB_SDR1]		= 0x690,
254 	[TX_AGG_CNT]		= 0x7dc,
255 	[TX_AGG_CNT2]		= 0x7ec,
256 	[MIB_ARNG]		= 0x0b0,
257 	[WTBLON_TOP_WDUCR]	= 0x200,
258 	[WTBL_UPDATE]		= 0x230,
259 	[PLE_FL_Q_EMPTY]	= 0x360,
260 	[PLE_FL_Q_CTRL]		= 0x3e0,
261 	[PLE_AC_QEMPTY]		= 0x600,
262 	[PLE_FREEPG_CNT]	= 0x380,
263 	[PLE_FREEPG_HEAD_TAIL]	= 0x384,
264 	[PLE_PG_HIF_GROUP]	= 0x00c,
265 	[PLE_HIF_PG_INFO]	= 0x388,
266 	[AC_OFFSET]		= 0x080,
267 	[ETBF_PAR_RPT0]		= 0x100,
268 };
269 
270 static const struct mt76_connac_reg_map mt7915_reg_map[] = {
271 	{ 0x00400000, 0x80000, 0x10000 }, /* WF_MCU_SYSRAM */
272 	{ 0x00410000, 0x90000, 0x10000 }, /* WF_MCU_SYSRAM (configure regs) */
273 	{ 0x40000000, 0x70000, 0x10000 }, /* WF_UMAC_SYSRAM */
274 	{ 0x54000000, 0x02000, 0x01000 }, /* WFDMA PCIE0 MCU DMA0 */
275 	{ 0x55000000, 0x03000, 0x01000 }, /* WFDMA PCIE0 MCU DMA1 */
276 	{ 0x58000000, 0x06000, 0x01000 }, /* WFDMA PCIE1 MCU DMA0 (MEM_DMA) */
277 	{ 0x59000000, 0x07000, 0x01000 }, /* WFDMA PCIE1 MCU DMA1 */
278 	{ 0x7c000000, 0xf0000, 0x10000 }, /* CONN_INFRA */
279 	{ 0x7c020000, 0xd0000, 0x10000 }, /* CONN_INFRA, WFDMA */
280 	{ 0x80020000, 0xb0000, 0x10000 }, /* WF_TOP_MISC_OFF */
281 	{ 0x81020000, 0xc0000, 0x10000 }, /* WF_TOP_MISC_ON */
282 	{ 0x820c0000, 0x08000, 0x04000 }, /* WF_UMAC_TOP (PLE) */
283 	{ 0x820c8000, 0x0c000, 0x02000 }, /* WF_UMAC_TOP (PSE) */
284 	{ 0x820cc000, 0x0e000, 0x02000 }, /* WF_UMAC_TOP (PP) */
285 	{ 0x820ce000, 0x21c00, 0x00200 }, /* WF_LMAC_TOP (WF_SEC) */
286 	{ 0x820cf000, 0x22000, 0x01000 }, /* WF_LMAC_TOP (WF_PF) */
287 	{ 0x820d0000, 0x30000, 0x10000 }, /* WF_LMAC_TOP (WF_WTBLON) */
288 	{ 0x820e0000, 0x20000, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_CFG) */
289 	{ 0x820e1000, 0x20400, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_TRB) */
290 	{ 0x820e2000, 0x20800, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_AGG) */
291 	{ 0x820e3000, 0x20c00, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_ARB) */
292 	{ 0x820e4000, 0x21000, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_TMAC) */
293 	{ 0x820e5000, 0x21400, 0x00800 }, /* WF_LMAC_TOP BN0 (WF_RMAC) */
294 	{ 0x820e7000, 0x21e00, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_DMA) */
295 	{ 0x820e9000, 0x23400, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_WTBLOFF) */
296 	{ 0x820ea000, 0x24000, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_ETBF) */
297 	{ 0x820eb000, 0x24200, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_LPON) */
298 	{ 0x820ec000, 0x24600, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_INT) */
299 	{ 0x820ed000, 0x24800, 0x00800 }, /* WF_LMAC_TOP BN0 (WF_MIB) */
300 	{ 0x820f0000, 0xa0000, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_CFG) */
301 	{ 0x820f1000, 0xa0600, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_TRB) */
302 	{ 0x820f2000, 0xa0800, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_AGG) */
303 	{ 0x820f3000, 0xa0c00, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_ARB) */
304 	{ 0x820f4000, 0xa1000, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_TMAC) */
305 	{ 0x820f5000, 0xa1400, 0x00800 }, /* WF_LMAC_TOP BN1 (WF_RMAC) */
306 	{ 0x820f7000, 0xa1e00, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_DMA) */
307 	{ 0x820f9000, 0xa3400, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_WTBLOFF) */
308 	{ 0x820fa000, 0xa4000, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_ETBF) */
309 	{ 0x820fb000, 0xa4200, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_LPON) */
310 	{ 0x820fc000, 0xa4600, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_INT) */
311 	{ 0x820fd000, 0xa4800, 0x00800 }, /* WF_LMAC_TOP BN1 (WF_MIB) */
312 	{ 0x0, 0x0, 0x0 }, /* imply end of search */
313 };
314 
315 static const struct mt76_connac_reg_map mt7916_reg_map[] = {
316 	{ 0x54000000, 0x02000, 0x01000 }, /* WFDMA_0 (PCIE0 MCU DMA0) */
317 	{ 0x55000000, 0x03000, 0x01000 }, /* WFDMA_1 (PCIE0 MCU DMA1) */
318 	{ 0x56000000, 0x04000, 0x01000 }, /* WFDMA_2 (Reserved) */
319 	{ 0x57000000, 0x05000, 0x01000 }, /* WFDMA_3 (MCU wrap CR) */
320 	{ 0x58000000, 0x06000, 0x01000 }, /* WFDMA_4 (PCIE1 MCU DMA0) */
321 	{ 0x59000000, 0x07000, 0x01000 }, /* WFDMA_5 (PCIE1 MCU DMA1) */
322 	{ 0x820c0000, 0x08000, 0x04000 }, /* WF_UMAC_TOP (PLE) */
323 	{ 0x820c8000, 0x0c000, 0x02000 }, /* WF_UMAC_TOP (PSE) */
324 	{ 0x820cc000, 0x0e000, 0x02000 }, /* WF_UMAC_TOP (PP) */
325 	{ 0x820e0000, 0x20000, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_CFG) */
326 	{ 0x820e1000, 0x20400, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_TRB) */
327 	{ 0x820e2000, 0x20800, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_AGG) */
328 	{ 0x820e3000, 0x20c00, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_ARB) */
329 	{ 0x820e4000, 0x21000, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_TMAC) */
330 	{ 0x820e5000, 0x21400, 0x00800 }, /* WF_LMAC_TOP BN0 (WF_RMAC) */
331 	{ 0x820ce000, 0x21c00, 0x00200 }, /* WF_LMAC_TOP (WF_SEC) */
332 	{ 0x820e7000, 0x21e00, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_DMA) */
333 	{ 0x820cf000, 0x22000, 0x01000 }, /* WF_LMAC_TOP (WF_PF) */
334 	{ 0x820e9000, 0x23400, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_WTBLOFF) */
335 	{ 0x820ea000, 0x24000, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_ETBF) */
336 	{ 0x820eb000, 0x24200, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_LPON) */
337 	{ 0x820ec000, 0x24600, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_INT) */
338 	{ 0x820ed000, 0x24800, 0x00800 }, /* WF_LMAC_TOP BN0 (WF_MIB) */
339 	{ 0x820ca000, 0x26000, 0x02000 }, /* WF_LMAC_TOP BN0 (WF_MUCOP) */
340 	{ 0x820d0000, 0x30000, 0x10000 }, /* WF_LMAC_TOP (WF_WTBLON) */
341 	{ 0x00400000, 0x80000, 0x10000 }, /* WF_MCU_SYSRAM */
342 	{ 0x00410000, 0x90000, 0x10000 }, /* WF_MCU_SYSRAM (configure cr) */
343 	{ 0x820f0000, 0xa0000, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_CFG) */
344 	{ 0x820f1000, 0xa0600, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_TRB) */
345 	{ 0x820f2000, 0xa0800, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_AGG) */
346 	{ 0x820f3000, 0xa0c00, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_ARB) */
347 	{ 0x820f4000, 0xa1000, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_TMAC) */
348 	{ 0x820f5000, 0xa1400, 0x00800 }, /* WF_LMAC_TOP BN1 (WF_RMAC) */
349 	{ 0x820f7000, 0xa1e00, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_DMA) */
350 	{ 0x820f9000, 0xa3400, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_WTBLOFF) */
351 	{ 0x820fa000, 0xa4000, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_ETBF) */
352 	{ 0x820fb000, 0xa4200, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_LPON) */
353 	{ 0x820fc000, 0xa4600, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_INT) */
354 	{ 0x820fd000, 0xa4800, 0x00800 }, /* WF_LMAC_TOP BN1 (WF_MIB) */
355 	{ 0x820c4000, 0xa8000, 0x01000 }, /* WF_LMAC_TOP (WF_UWTBL ) */
356 	{ 0x820b0000, 0xae000, 0x01000 }, /* [APB2] WFSYS_ON */
357 	{ 0x80020000, 0xb0000, 0x10000 }, /* WF_TOP_MISC_OFF */
358 	{ 0x81020000, 0xc0000, 0x10000 }, /* WF_TOP_MISC_ON */
359 	{ 0x0, 0x0, 0x0 }, /* imply end of search */
360 };
361 
362 static const struct mt76_connac_reg_map mt7986_reg_map[] = {
363 	{ 0x54000000, 0x402000, 0x01000 }, /* WFDMA_0 (PCIE0 MCU DMA0) */
364 	{ 0x55000000, 0x403000, 0x01000 }, /* WFDMA_1 (PCIE0 MCU DMA1) */
365 	{ 0x56000000, 0x404000, 0x01000 }, /* WFDMA_2 (Reserved) */
366 	{ 0x57000000, 0x405000, 0x01000 }, /* WFDMA_3 (MCU wrap CR) */
367 	{ 0x58000000, 0x406000, 0x01000 }, /* WFDMA_4 (PCIE1 MCU DMA0) */
368 	{ 0x59000000, 0x407000, 0x01000 }, /* WFDMA_5 (PCIE1 MCU DMA1) */
369 	{ 0x820c0000, 0x408000, 0x04000 }, /* WF_UMAC_TOP (PLE) */
370 	{ 0x820c8000, 0x40c000, 0x02000 }, /* WF_UMAC_TOP (PSE) */
371 	{ 0x820cc000, 0x40e000, 0x02000 }, /* WF_UMAC_TOP (PP) */
372 	{ 0x820e0000, 0x420000, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_CFG) */
373 	{ 0x820e1000, 0x420400, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_TRB) */
374 	{ 0x820e2000, 0x420800, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_AGG) */
375 	{ 0x820e3000, 0x420c00, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_ARB) */
376 	{ 0x820e4000, 0x421000, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_TMAC) */
377 	{ 0x820e5000, 0x421400, 0x00800 }, /* WF_LMAC_TOP BN0 (WF_RMAC) */
378 	{ 0x820ce000, 0x421c00, 0x00200 }, /* WF_LMAC_TOP (WF_SEC) */
379 	{ 0x820e7000, 0x421e00, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_DMA) */
380 	{ 0x820cf000, 0x422000, 0x01000 }, /* WF_LMAC_TOP (WF_PF) */
381 	{ 0x820e9000, 0x423400, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_WTBLOFF) */
382 	{ 0x820ea000, 0x424000, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_ETBF) */
383 	{ 0x820eb000, 0x424200, 0x00400 }, /* WF_LMAC_TOP BN0 (WF_LPON) */
384 	{ 0x820ec000, 0x424600, 0x00200 }, /* WF_LMAC_TOP BN0 (WF_INT) */
385 	{ 0x820ed000, 0x424800, 0x00800 }, /* WF_LMAC_TOP BN0 (WF_MIB) */
386 	{ 0x820ca000, 0x426000, 0x02000 }, /* WF_LMAC_TOP BN0 (WF_MUCOP) */
387 	{ 0x820d0000, 0x430000, 0x10000 }, /* WF_LMAC_TOP (WF_WTBLON) */
388 	{ 0x00400000, 0x480000, 0x10000 }, /* WF_MCU_SYSRAM */
389 	{ 0x00410000, 0x490000, 0x10000 }, /* WF_MCU_SYSRAM */
390 	{ 0x820f0000, 0x4a0000, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_CFG) */
391 	{ 0x820f1000, 0x4a0600, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_TRB) */
392 	{ 0x820f2000, 0x4a0800, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_AGG) */
393 	{ 0x820f3000, 0x4a0c00, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_ARB) */
394 	{ 0x820f4000, 0x4a1000, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_TMAC) */
395 	{ 0x820f5000, 0x4a1400, 0x00800 }, /* WF_LMAC_TOP BN1 (WF_RMAC) */
396 	{ 0x820f7000, 0x4a1e00, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_DMA) */
397 	{ 0x820f9000, 0x4a3400, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_WTBLOFF) */
398 	{ 0x820fa000, 0x4a4000, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_ETBF) */
399 	{ 0x820fb000, 0x4a4200, 0x00400 }, /* WF_LMAC_TOP BN1 (WF_LPON) */
400 	{ 0x820fc000, 0x4a4600, 0x00200 }, /* WF_LMAC_TOP BN1 (WF_INT) */
401 	{ 0x820fd000, 0x4a4800, 0x00800 }, /* WF_LMAC_TOP BN1 (WF_MIB) */
402 	{ 0x820c4000, 0x4a8000, 0x01000 }, /* WF_LMAC_TOP (WF_UWTBL ) */
403 	{ 0x820b0000, 0x4ae000, 0x01000 }, /* [APB2] WFSYS_ON */
404 	{ 0x80020000, 0x4b0000, 0x10000 }, /* WF_TOP_MISC_OFF */
405 	{ 0x81020000, 0x4c0000, 0x10000 }, /* WF_TOP_MISC_ON */
406 	{ 0x89000000, 0x4d0000, 0x01000 }, /* WF_MCU_CFG_ON */
407 	{ 0x89010000, 0x4d1000, 0x01000 }, /* WF_MCU_CIRQ */
408 	{ 0x89020000, 0x4d2000, 0x01000 }, /* WF_MCU_GPT */
409 	{ 0x89030000, 0x4d3000, 0x01000 }, /* WF_MCU_WDT */
410 	{ 0x80010000, 0x4d4000, 0x01000 }, /* WF_AXIDMA */
411 	{ 0x0, 0x0, 0x0 }, /* imply end of search */
412 };
413 
414 static u32 mt7915_reg_map_l1(struct mt7915_dev *dev, u32 addr)
415 {
416 	u32 offset = FIELD_GET(MT_HIF_REMAP_L1_OFFSET, addr);
417 	u32 base = FIELD_GET(MT_HIF_REMAP_L1_BASE, addr);
418 	u32 l1_remap;
419 
420 	if (is_mt798x(&dev->mt76))
421 		return MT_CONN_INFRA_OFFSET(addr);
422 
423 	l1_remap = is_mt7915(&dev->mt76) ?
424 		   MT_HIF_REMAP_L1 : MT_HIF_REMAP_L1_MT7916;
425 
426 	dev->bus_ops->rmw(&dev->mt76, l1_remap,
427 			  MT_HIF_REMAP_L1_MASK,
428 			  FIELD_PREP(MT_HIF_REMAP_L1_MASK, base));
429 	/* use read to push write */
430 	dev->bus_ops->rr(&dev->mt76, l1_remap);
431 
432 	return MT_HIF_REMAP_BASE_L1 + offset;
433 }
434 
435 static u32 mt7915_reg_map_l2(struct mt7915_dev *dev, u32 addr)
436 {
437 	u32 offset, base;
438 
439 	if (is_mt7915(&dev->mt76)) {
440 		offset = FIELD_GET(MT_HIF_REMAP_L2_OFFSET, addr);
441 		base = FIELD_GET(MT_HIF_REMAP_L2_BASE, addr);
442 
443 		dev->bus_ops->rmw(&dev->mt76, MT_HIF_REMAP_L2,
444 				  MT_HIF_REMAP_L2_MASK,
445 				  FIELD_PREP(MT_HIF_REMAP_L2_MASK, base));
446 
447 		/* use read to push write */
448 		dev->bus_ops->rr(&dev->mt76, MT_HIF_REMAP_L2);
449 	} else {
450 		u32 ofs = is_mt798x(&dev->mt76) ? 0x400000 : 0;
451 
452 		offset = FIELD_GET(MT_HIF_REMAP_L2_OFFSET_MT7916, addr);
453 		base = FIELD_GET(MT_HIF_REMAP_L2_BASE_MT7916, addr);
454 
455 		dev->bus_ops->rmw(&dev->mt76, MT_HIF_REMAP_L2_MT7916 + ofs,
456 				  MT_HIF_REMAP_L2_MASK_MT7916,
457 				  FIELD_PREP(MT_HIF_REMAP_L2_MASK_MT7916, base));
458 
459 		/* use read to push write */
460 		dev->bus_ops->rr(&dev->mt76, MT_HIF_REMAP_L2_MT7916 + ofs);
461 
462 		offset += (MT_HIF_REMAP_BASE_L2_MT7916 + ofs);
463 	}
464 
465 	return offset;
466 }
467 
468 static u32 __mt7915_reg_addr(struct mt7915_dev *dev, u32 addr)
469 {
470 	int i;
471 
472 	if (addr < 0x100000)
473 		return addr;
474 
475 	if (!dev->reg.map) {
476 		dev_err(dev->mt76.dev, "err: reg_map is null\n");
477 		return addr;
478 	}
479 
480 	for (i = 0; i < dev->reg.map_size; i++) {
481 		u32 ofs;
482 
483 		if (addr < dev->reg.map[i].phys)
484 			continue;
485 
486 		ofs = addr - dev->reg.map[i].phys;
487 		if (ofs > dev->reg.map[i].size)
488 			continue;
489 
490 		return dev->reg.map[i].maps + ofs;
491 	}
492 
493 	return 0;
494 }
495 
496 static u32 __mt7915_reg_remap_addr(struct mt7915_dev *dev, u32 addr)
497 {
498 	if ((addr >= MT_INFRA_BASE && addr < MT_WFSYS0_PHY_START) ||
499 	    (addr >= MT_WFSYS0_PHY_START && addr < MT_WFSYS1_PHY_START) ||
500 	    (addr >= MT_WFSYS1_PHY_START && addr <= MT_WFSYS1_PHY_END))
501 		return mt7915_reg_map_l1(dev, addr);
502 
503 	if (dev_is_pci(dev->mt76.dev) &&
504 	    ((addr >= MT_CBTOP1_PHY_START && addr <= MT_CBTOP1_PHY_END) ||
505 	    addr >= MT_CBTOP2_PHY_START))
506 		return mt7915_reg_map_l1(dev, addr);
507 
508 	/* CONN_INFRA: covert to phyiscal addr and use layer 1 remap */
509 	if (addr >= MT_INFRA_MCU_START && addr <= MT_INFRA_MCU_END) {
510 		addr = addr - MT_INFRA_MCU_START + MT_INFRA_BASE;
511 		return mt7915_reg_map_l1(dev, addr);
512 	}
513 
514 	return mt7915_reg_map_l2(dev, addr);
515 }
516 
517 void mt7915_memcpy_fromio(struct mt7915_dev *dev, void *buf, u32 offset,
518 			  size_t len)
519 {
520 	u32 addr = __mt7915_reg_addr(dev, offset);
521 
522 	if (addr) {
523 		memcpy_fromio(buf, dev->mt76.mmio.regs + addr, len);
524 		return;
525 	}
526 
527 	spin_lock_bh(&dev->reg_lock);
528 	memcpy_fromio(buf, dev->mt76.mmio.regs +
529 			   __mt7915_reg_remap_addr(dev, offset), len);
530 	spin_unlock_bh(&dev->reg_lock);
531 }
532 
533 static u32 mt7915_rr(struct mt76_dev *mdev, u32 offset)
534 {
535 	struct mt7915_dev *dev = container_of(mdev, struct mt7915_dev, mt76);
536 	u32 addr = __mt7915_reg_addr(dev, offset), val;
537 
538 	if (addr)
539 		return dev->bus_ops->rr(mdev, addr);
540 
541 	spin_lock_bh(&dev->reg_lock);
542 	val = dev->bus_ops->rr(mdev, __mt7915_reg_remap_addr(dev, offset));
543 	spin_unlock_bh(&dev->reg_lock);
544 
545 	return val;
546 }
547 
548 static void mt7915_wr(struct mt76_dev *mdev, u32 offset, u32 val)
549 {
550 	struct mt7915_dev *dev = container_of(mdev, struct mt7915_dev, mt76);
551 	u32 addr = __mt7915_reg_addr(dev, offset);
552 
553 	if (addr) {
554 		dev->bus_ops->wr(mdev, addr, val);
555 		return;
556 	}
557 
558 	spin_lock_bh(&dev->reg_lock);
559 	dev->bus_ops->wr(mdev, __mt7915_reg_remap_addr(dev, offset), val);
560 	spin_unlock_bh(&dev->reg_lock);
561 }
562 
563 static u32 mt7915_rmw(struct mt76_dev *mdev, u32 offset, u32 mask, u32 val)
564 {
565 	struct mt7915_dev *dev = container_of(mdev, struct mt7915_dev, mt76);
566 	u32 addr = __mt7915_reg_addr(dev, offset);
567 
568 	if (addr)
569 		return dev->bus_ops->rmw(mdev, addr, mask, val);
570 
571 	spin_lock_bh(&dev->reg_lock);
572 	val = dev->bus_ops->rmw(mdev, __mt7915_reg_remap_addr(dev, offset), mask, val);
573 	spin_unlock_bh(&dev->reg_lock);
574 
575 	return val;
576 }
577 
578 #ifdef CONFIG_NET_MEDIATEK_SOC_WED
579 static int mt7915_mmio_wed_offload_enable(struct mtk_wed_device *wed)
580 {
581 	struct mt7915_dev *dev;
582 
583 	dev = container_of(wed, struct mt7915_dev, mt76.mmio.wed);
584 
585 	spin_lock_bh(&dev->mt76.token_lock);
586 	dev->mt76.token_size = wed->wlan.token_start;
587 	spin_unlock_bh(&dev->mt76.token_lock);
588 
589 	return !wait_event_timeout(dev->mt76.tx_wait,
590 				   !dev->mt76.wed_token_count, HZ);
591 }
592 
593 static void mt7915_mmio_wed_offload_disable(struct mtk_wed_device *wed)
594 {
595 	struct mt7915_dev *dev;
596 
597 	dev = container_of(wed, struct mt7915_dev, mt76.mmio.wed);
598 
599 	spin_lock_bh(&dev->mt76.token_lock);
600 	dev->mt76.token_size = MT7915_TOKEN_SIZE;
601 	spin_unlock_bh(&dev->mt76.token_lock);
602 }
603 
604 static void mt7915_mmio_wed_release_rx_buf(struct mtk_wed_device *wed)
605 {
606 	struct mt7915_dev *dev;
607 	int i;
608 
609 	dev = container_of(wed, struct mt7915_dev, mt76.mmio.wed);
610 	for (i = 0; i < dev->mt76.rx_token_size; i++) {
611 		struct mt76_txwi_cache *t;
612 
613 		t = mt76_rx_token_release(&dev->mt76, i);
614 		if (!t || !t->ptr)
615 			continue;
616 
617 		mt76_put_page_pool_buf(t->ptr, false);
618 		t->ptr = NULL;
619 
620 		mt76_put_rxwi(&dev->mt76, t);
621 	}
622 
623 	mt76_free_pending_rxwi(&dev->mt76);
624 }
625 
626 static u32 mt7915_mmio_wed_init_rx_buf(struct mtk_wed_device *wed, int size)
627 {
628 	struct mtk_rxbm_desc *desc = wed->rx_buf_ring.desc;
629 	struct mt76_txwi_cache *t = NULL;
630 	struct mt7915_dev *dev;
631 	struct mt76_queue *q;
632 	int i, len;
633 
634 	dev = container_of(wed, struct mt7915_dev, mt76.mmio.wed);
635 	q = &dev->mt76.q_rx[MT_RXQ_MAIN];
636 	len = SKB_WITH_OVERHEAD(q->buf_size);
637 
638 	for (i = 0; i < size; i++) {
639 		enum dma_data_direction dir;
640 		dma_addr_t addr;
641 		u32 offset;
642 		int token;
643 		void *buf;
644 
645 		t = mt76_get_rxwi(&dev->mt76);
646 		if (!t)
647 			goto unmap;
648 
649 		buf = mt76_get_page_pool_buf(q, &offset, q->buf_size);
650 		if (!buf)
651 			goto unmap;
652 
653 		addr = page_pool_get_dma_addr(virt_to_head_page(buf)) + offset;
654 		dir = page_pool_get_dma_dir(q->page_pool);
655 		dma_sync_single_for_device(dev->mt76.dma_dev, addr, len, dir);
656 
657 		desc->buf0 = cpu_to_le32(addr);
658 		token = mt76_rx_token_consume(&dev->mt76, buf, t, addr);
659 		if (token < 0) {
660 			mt76_put_page_pool_buf(buf, false);
661 			goto unmap;
662 		}
663 
664 		desc->token |= cpu_to_le32(FIELD_PREP(MT_DMA_CTL_TOKEN,
665 						      token));
666 		desc++;
667 	}
668 
669 	return 0;
670 
671 unmap:
672 	if (t)
673 		mt76_put_rxwi(&dev->mt76, t);
674 	mt7915_mmio_wed_release_rx_buf(wed);
675 	return -ENOMEM;
676 }
677 
678 static void mt7915_mmio_wed_update_rx_stats(struct mtk_wed_device *wed,
679 					    struct mtk_wed_wo_rx_stats *stats)
680 {
681 	int idx = le16_to_cpu(stats->wlan_idx);
682 	struct mt7915_dev *dev;
683 	struct mt76_wcid *wcid;
684 
685 	dev = container_of(wed, struct mt7915_dev, mt76.mmio.wed);
686 
687 	if (idx >= mt7915_wtbl_size(dev))
688 		return;
689 
690 	rcu_read_lock();
691 
692 	wcid = rcu_dereference(dev->mt76.wcid[idx]);
693 	if (wcid) {
694 		wcid->stats.rx_bytes += le32_to_cpu(stats->rx_byte_cnt);
695 		wcid->stats.rx_packets += le32_to_cpu(stats->rx_pkt_cnt);
696 		wcid->stats.rx_errors += le32_to_cpu(stats->rx_err_cnt);
697 		wcid->stats.rx_drops += le32_to_cpu(stats->rx_drop_cnt);
698 	}
699 
700 	rcu_read_unlock();
701 }
702 
703 static int mt7915_mmio_wed_reset(struct mtk_wed_device *wed)
704 {
705 	struct mt76_dev *mdev = container_of(wed, struct mt76_dev, mmio.wed);
706 	struct mt7915_dev *dev = container_of(mdev, struct mt7915_dev, mt76);
707 	struct mt76_phy *mphy = &dev->mphy;
708 	int ret;
709 
710 	ASSERT_RTNL();
711 
712 	if (test_and_set_bit(MT76_STATE_WED_RESET, &mphy->state))
713 		return -EBUSY;
714 
715 	ret = mt7915_mcu_set_ser(dev, SER_RECOVER, SER_SET_RECOVER_L1,
716 				 mphy->band_idx);
717 	if (ret)
718 		goto out;
719 
720 	rtnl_unlock();
721 	if (!wait_for_completion_timeout(&mdev->mmio.wed_reset, 20 * HZ)) {
722 		dev_err(mdev->dev, "wed reset timeout\n");
723 		ret = -ETIMEDOUT;
724 	}
725 	rtnl_lock();
726 out:
727 	clear_bit(MT76_STATE_WED_RESET, &mphy->state);
728 
729 	return ret;
730 }
731 
732 static void mt7915_mmio_wed_reset_complete(struct mtk_wed_device *wed)
733 {
734 	struct mt76_dev *dev = container_of(wed, struct mt76_dev, mmio.wed);
735 
736 	complete(&dev->mmio.wed_reset_complete);
737 }
738 #endif
739 
740 int mt7915_mmio_wed_init(struct mt7915_dev *dev, void *pdev_ptr,
741 			 bool pci, int *irq)
742 {
743 #ifdef CONFIG_NET_MEDIATEK_SOC_WED
744 	struct mtk_wed_device *wed = &dev->mt76.mmio.wed;
745 	int ret;
746 
747 	if (!wed_enable)
748 		return 0;
749 
750 	if (pci) {
751 		struct pci_dev *pci_dev = pdev_ptr;
752 
753 		wed->wlan.pci_dev = pci_dev;
754 		wed->wlan.bus_type = MTK_WED_BUS_PCIE;
755 		wed->wlan.base = devm_ioremap(dev->mt76.dev,
756 					      pci_resource_start(pci_dev, 0),
757 					      pci_resource_len(pci_dev, 0));
758 		wed->wlan.phy_base = pci_resource_start(pci_dev, 0);
759 		wed->wlan.wpdma_int = pci_resource_start(pci_dev, 0) +
760 				      MT_INT_WED_SOURCE_CSR;
761 		wed->wlan.wpdma_mask = pci_resource_start(pci_dev, 0) +
762 				       MT_INT_WED_MASK_CSR;
763 		wed->wlan.wpdma_phys = pci_resource_start(pci_dev, 0) +
764 				       MT_WFDMA_EXT_CSR_BASE;
765 		wed->wlan.wpdma_tx = pci_resource_start(pci_dev, 0) +
766 				     MT_TXQ_WED_RING_BASE;
767 		wed->wlan.wpdma_txfree = pci_resource_start(pci_dev, 0) +
768 					 MT_RXQ_WED_RING_BASE;
769 		wed->wlan.wpdma_rx_glo = pci_resource_start(pci_dev, 0) +
770 					 MT_WPDMA_GLO_CFG;
771 		wed->wlan.wpdma_rx = pci_resource_start(pci_dev, 0) +
772 				     MT_RXQ_WED_DATA_RING_BASE;
773 	} else {
774 		struct platform_device *plat_dev = pdev_ptr;
775 		struct resource *res;
776 
777 		res = platform_get_resource(plat_dev, IORESOURCE_MEM, 0);
778 		if (!res)
779 			return 0;
780 
781 		wed->wlan.platform_dev = plat_dev;
782 		wed->wlan.bus_type = MTK_WED_BUS_AXI;
783 		wed->wlan.base = devm_ioremap(dev->mt76.dev, res->start,
784 					      resource_size(res));
785 		wed->wlan.phy_base = res->start;
786 		wed->wlan.wpdma_int = res->start + MT_INT_SOURCE_CSR;
787 		wed->wlan.wpdma_mask = res->start + MT_INT_MASK_CSR;
788 		wed->wlan.wpdma_tx = res->start + MT_TXQ_WED_RING_BASE;
789 		wed->wlan.wpdma_txfree = res->start + MT_RXQ_WED_RING_BASE;
790 		wed->wlan.wpdma_rx_glo = res->start + MT_WPDMA_GLO_CFG;
791 		wed->wlan.wpdma_rx = res->start + MT_RXQ_WED_DATA_RING_BASE;
792 	}
793 	wed->wlan.nbuf = MT7915_HW_TOKEN_SIZE;
794 	wed->wlan.tx_tbit[0] = is_mt7915(&dev->mt76) ? 4 : 30;
795 	wed->wlan.tx_tbit[1] = is_mt7915(&dev->mt76) ? 5 : 31;
796 	wed->wlan.txfree_tbit = is_mt798x(&dev->mt76) ? 2 : 1;
797 	wed->wlan.token_start = MT7915_TOKEN_SIZE - wed->wlan.nbuf;
798 	wed->wlan.wcid_512 = !is_mt7915(&dev->mt76);
799 
800 	wed->wlan.rx_nbuf = 65536;
801 	wed->wlan.rx_npkt = MT7915_WED_RX_TOKEN_SIZE;
802 	wed->wlan.rx_size = SKB_WITH_OVERHEAD(MT_RX_BUF_SIZE);
803 	if (is_mt7915(&dev->mt76)) {
804 		wed->wlan.rx_tbit[0] = 16;
805 		wed->wlan.rx_tbit[1] = 17;
806 	} else if (is_mt798x(&dev->mt76)) {
807 		wed->wlan.rx_tbit[0] = 22;
808 		wed->wlan.rx_tbit[1] = 23;
809 	} else {
810 		wed->wlan.rx_tbit[0] = 18;
811 		wed->wlan.rx_tbit[1] = 19;
812 	}
813 
814 	wed->wlan.init_buf = mt7915_wed_init_buf;
815 	wed->wlan.offload_enable = mt7915_mmio_wed_offload_enable;
816 	wed->wlan.offload_disable = mt7915_mmio_wed_offload_disable;
817 	wed->wlan.init_rx_buf = mt7915_mmio_wed_init_rx_buf;
818 	wed->wlan.release_rx_buf = mt7915_mmio_wed_release_rx_buf;
819 	wed->wlan.update_wo_rx_stats = mt7915_mmio_wed_update_rx_stats;
820 	wed->wlan.reset = mt7915_mmio_wed_reset;
821 	wed->wlan.reset_complete = mt7915_mmio_wed_reset_complete;
822 
823 	dev->mt76.rx_token_size = wed->wlan.rx_npkt;
824 
825 	if (mtk_wed_device_attach(wed))
826 		return 0;
827 
828 	*irq = wed->irq;
829 	dev->mt76.dma_dev = wed->dev;
830 
831 	ret = dma_set_mask(wed->dev, DMA_BIT_MASK(32));
832 	if (ret)
833 		return ret;
834 
835 	return 1;
836 #else
837 	return 0;
838 #endif
839 }
840 
841 static int mt7915_mmio_init(struct mt76_dev *mdev,
842 			    void __iomem *mem_base,
843 			    u32 device_id)
844 {
845 	struct mt76_bus_ops *bus_ops;
846 	struct mt7915_dev *dev;
847 
848 	dev = container_of(mdev, struct mt7915_dev, mt76);
849 	mt76_mmio_init(&dev->mt76, mem_base);
850 	spin_lock_init(&dev->reg_lock);
851 
852 	switch (device_id) {
853 	case 0x7915:
854 		dev->reg.reg_rev = mt7915_reg;
855 		dev->reg.offs_rev = mt7915_offs;
856 		dev->reg.map = mt7915_reg_map;
857 		dev->reg.map_size = ARRAY_SIZE(mt7915_reg_map);
858 		break;
859 	case 0x7906:
860 		dev->reg.reg_rev = mt7916_reg;
861 		dev->reg.offs_rev = mt7916_offs;
862 		dev->reg.map = mt7916_reg_map;
863 		dev->reg.map_size = ARRAY_SIZE(mt7916_reg_map);
864 		break;
865 	case 0x7981:
866 	case 0x7986:
867 		dev->reg.reg_rev = mt7986_reg;
868 		dev->reg.offs_rev = mt7916_offs;
869 		dev->reg.map = mt7986_reg_map;
870 		dev->reg.map_size = ARRAY_SIZE(mt7986_reg_map);
871 		break;
872 	default:
873 		return -EINVAL;
874 	}
875 
876 	dev->bus_ops = dev->mt76.bus;
877 	bus_ops = devm_kmemdup(dev->mt76.dev, dev->bus_ops, sizeof(*bus_ops),
878 			       GFP_KERNEL);
879 	if (!bus_ops)
880 		return -ENOMEM;
881 
882 	bus_ops->rr = mt7915_rr;
883 	bus_ops->wr = mt7915_wr;
884 	bus_ops->rmw = mt7915_rmw;
885 	dev->mt76.bus = bus_ops;
886 
887 	mdev->rev = (device_id << 16) |
888 		    (mt76_rr(dev, MT_HW_REV) & 0xff);
889 	dev_dbg(mdev->dev, "ASIC revision: %04x\n", mdev->rev);
890 
891 	return 0;
892 }
893 
894 void mt7915_dual_hif_set_irq_mask(struct mt7915_dev *dev,
895 				  bool write_reg,
896 				  u32 clear, u32 set)
897 {
898 	struct mt76_dev *mdev = &dev->mt76;
899 	unsigned long flags;
900 
901 	spin_lock_irqsave(&mdev->mmio.irq_lock, flags);
902 
903 	mdev->mmio.irqmask &= ~clear;
904 	mdev->mmio.irqmask |= set;
905 
906 	if (write_reg) {
907 		if (mtk_wed_device_active(&mdev->mmio.wed))
908 			mtk_wed_device_irq_set_mask(&mdev->mmio.wed,
909 						    mdev->mmio.irqmask);
910 		else
911 			mt76_wr(dev, MT_INT_MASK_CSR, mdev->mmio.irqmask);
912 		mt76_wr(dev, MT_INT1_MASK_CSR, mdev->mmio.irqmask);
913 	}
914 
915 	spin_unlock_irqrestore(&mdev->mmio.irq_lock, flags);
916 }
917 
918 static void mt7915_rx_poll_complete(struct mt76_dev *mdev,
919 				    enum mt76_rxq_id q)
920 {
921 	struct mt7915_dev *dev = container_of(mdev, struct mt7915_dev, mt76);
922 
923 	mt7915_irq_enable(dev, MT_INT_RX(q));
924 }
925 
926 /* TODO: support 2/4/6/8 MSI-X vectors */
927 static void mt7915_irq_tasklet(struct tasklet_struct *t)
928 {
929 	struct mt7915_dev *dev = from_tasklet(dev, t, mt76.irq_tasklet);
930 	struct mtk_wed_device *wed = &dev->mt76.mmio.wed;
931 	u32 intr, intr1, mask;
932 
933 	if (mtk_wed_device_active(wed)) {
934 		mtk_wed_device_irq_set_mask(wed, 0);
935 		if (dev->hif2)
936 			mt76_wr(dev, MT_INT1_MASK_CSR, 0);
937 		intr = mtk_wed_device_irq_get(wed, dev->mt76.mmio.irqmask);
938 	} else {
939 		mt76_wr(dev, MT_INT_MASK_CSR, 0);
940 		if (dev->hif2)
941 			mt76_wr(dev, MT_INT1_MASK_CSR, 0);
942 
943 		intr = mt76_rr(dev, MT_INT_SOURCE_CSR);
944 		intr &= dev->mt76.mmio.irqmask;
945 		mt76_wr(dev, MT_INT_SOURCE_CSR, intr);
946 	}
947 
948 	if (dev->hif2) {
949 		intr1 = mt76_rr(dev, MT_INT1_SOURCE_CSR);
950 		intr1 &= dev->mt76.mmio.irqmask;
951 		mt76_wr(dev, MT_INT1_SOURCE_CSR, intr1);
952 
953 		intr |= intr1;
954 	}
955 
956 	trace_dev_irq(&dev->mt76, intr, dev->mt76.mmio.irqmask);
957 
958 	mask = intr & MT_INT_RX_DONE_ALL;
959 	if (intr & MT_INT_TX_DONE_MCU)
960 		mask |= MT_INT_TX_DONE_MCU;
961 
962 	mt7915_irq_disable(dev, mask);
963 
964 	if (intr & MT_INT_TX_DONE_MCU)
965 		napi_schedule(&dev->mt76.tx_napi);
966 
967 	if (intr & MT_INT_RX(MT_RXQ_MAIN))
968 		napi_schedule(&dev->mt76.napi[MT_RXQ_MAIN]);
969 
970 	if (intr & MT_INT_RX(MT_RXQ_BAND1))
971 		napi_schedule(&dev->mt76.napi[MT_RXQ_BAND1]);
972 
973 	if (intr & MT_INT_RX(MT_RXQ_MCU))
974 		napi_schedule(&dev->mt76.napi[MT_RXQ_MCU]);
975 
976 	if (intr & MT_INT_RX(MT_RXQ_MCU_WA))
977 		napi_schedule(&dev->mt76.napi[MT_RXQ_MCU_WA]);
978 
979 	if (!is_mt7915(&dev->mt76) &&
980 	    (intr & MT_INT_RX(MT_RXQ_MAIN_WA)))
981 		napi_schedule(&dev->mt76.napi[MT_RXQ_MAIN_WA]);
982 
983 	if (intr & MT_INT_RX(MT_RXQ_BAND1_WA))
984 		napi_schedule(&dev->mt76.napi[MT_RXQ_BAND1_WA]);
985 
986 	if (intr & MT_INT_MCU_CMD) {
987 		u32 val = mt76_rr(dev, MT_MCU_CMD);
988 
989 		mt76_wr(dev, MT_MCU_CMD, val);
990 		if (val & (MT_MCU_CMD_ERROR_MASK | MT_MCU_CMD_WDT_MASK)) {
991 			dev->recovery.state = val;
992 			mt7915_reset(dev);
993 		}
994 	}
995 }
996 
997 irqreturn_t mt7915_irq_handler(int irq, void *dev_instance)
998 {
999 	struct mt7915_dev *dev = dev_instance;
1000 	struct mtk_wed_device *wed = &dev->mt76.mmio.wed;
1001 
1002 	if (mtk_wed_device_active(wed))
1003 		mtk_wed_device_irq_set_mask(wed, 0);
1004 	else
1005 		mt76_wr(dev, MT_INT_MASK_CSR, 0);
1006 
1007 	if (dev->hif2)
1008 		mt76_wr(dev, MT_INT1_MASK_CSR, 0);
1009 
1010 	if (!test_bit(MT76_STATE_INITIALIZED, &dev->mphy.state))
1011 		return IRQ_NONE;
1012 
1013 	tasklet_schedule(&dev->mt76.irq_tasklet);
1014 
1015 	return IRQ_HANDLED;
1016 }
1017 
1018 struct mt7915_dev *mt7915_mmio_probe(struct device *pdev,
1019 				     void __iomem *mem_base, u32 device_id)
1020 {
1021 	static const struct mt76_driver_ops drv_ops = {
1022 		/* txwi_size = txd size + txp size */
1023 		.txwi_size = MT_TXD_SIZE + sizeof(struct mt76_connac_fw_txp),
1024 		.drv_flags = MT_DRV_TXWI_NO_FREE | MT_DRV_HW_MGMT_TXQ |
1025 			     MT_DRV_AMSDU_OFFLOAD,
1026 		.survey_flags = SURVEY_INFO_TIME_TX |
1027 				SURVEY_INFO_TIME_RX |
1028 				SURVEY_INFO_TIME_BSS_RX,
1029 		.token_size = MT7915_TOKEN_SIZE,
1030 		.tx_prepare_skb = mt7915_tx_prepare_skb,
1031 		.tx_complete_skb = mt76_connac_tx_complete_skb,
1032 		.rx_skb = mt7915_queue_rx_skb,
1033 		.rx_check = mt7915_rx_check,
1034 		.rx_poll_complete = mt7915_rx_poll_complete,
1035 		.sta_add = mt7915_mac_sta_add,
1036 		.sta_remove = mt7915_mac_sta_remove,
1037 		.update_survey = mt7915_update_channel,
1038 	};
1039 	struct mt7915_dev *dev;
1040 	struct mt76_dev *mdev;
1041 	int ret;
1042 
1043 	mdev = mt76_alloc_device(pdev, sizeof(*dev), &mt7915_ops, &drv_ops);
1044 	if (!mdev)
1045 		return ERR_PTR(-ENOMEM);
1046 
1047 	dev = container_of(mdev, struct mt7915_dev, mt76);
1048 
1049 	ret = mt7915_mmio_init(mdev, mem_base, device_id);
1050 	if (ret)
1051 		goto error;
1052 
1053 	tasklet_setup(&mdev->irq_tasklet, mt7915_irq_tasklet);
1054 
1055 	return dev;
1056 
1057 error:
1058 	mt76_free_device(&dev->mt76);
1059 
1060 	return ERR_PTR(ret);
1061 }
1062 
1063 static int __init mt7915_init(void)
1064 {
1065 	int ret;
1066 
1067 	ret = pci_register_driver(&mt7915_hif_driver);
1068 	if (ret)
1069 		return ret;
1070 
1071 	ret = pci_register_driver(&mt7915_pci_driver);
1072 	if (ret)
1073 		goto error_pci;
1074 
1075 	if (IS_ENABLED(CONFIG_MT798X_WMAC)) {
1076 		ret = platform_driver_register(&mt798x_wmac_driver);
1077 		if (ret)
1078 			goto error_wmac;
1079 	}
1080 
1081 	return 0;
1082 
1083 error_wmac:
1084 	pci_unregister_driver(&mt7915_pci_driver);
1085 error_pci:
1086 	pci_unregister_driver(&mt7915_hif_driver);
1087 
1088 	return ret;
1089 }
1090 
1091 static void __exit mt7915_exit(void)
1092 {
1093 	if (IS_ENABLED(CONFIG_MT798X_WMAC))
1094 		platform_driver_unregister(&mt798x_wmac_driver);
1095 
1096 	pci_unregister_driver(&mt7915_pci_driver);
1097 	pci_unregister_driver(&mt7915_hif_driver);
1098 }
1099 
1100 module_init(mt7915_init);
1101 module_exit(mt7915_exit);
1102 MODULE_LICENSE("Dual BSD/GPL");
1103