1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (c) 2014 MediaTek Inc.
4  * Author: Flora Fu, MediaTek
5  */
6 #include <linux/clk.h>
7 #include <linux/interrupt.h>
8 #include <linux/io.h>
9 #include <linux/kernel.h>
10 #include <linux/module.h>
11 #include <linux/of.h>
12 #include <linux/of_platform.h>
13 #include <linux/platform_device.h>
14 #include <linux/regmap.h>
15 #include <linux/reset.h>
16 
17 #define PWRAP_POLL_DELAY_US	10
18 #define PWRAP_POLL_TIMEOUT_US	10000
19 
20 #define PWRAP_MT8135_BRIDGE_IORD_ARB_EN		0x4
21 #define PWRAP_MT8135_BRIDGE_WACS3_EN		0x10
22 #define PWRAP_MT8135_BRIDGE_INIT_DONE3		0x14
23 #define PWRAP_MT8135_BRIDGE_WACS4_EN		0x24
24 #define PWRAP_MT8135_BRIDGE_INIT_DONE4		0x28
25 #define PWRAP_MT8135_BRIDGE_INT_EN		0x38
26 #define PWRAP_MT8135_BRIDGE_TIMER_EN		0x48
27 #define PWRAP_MT8135_BRIDGE_WDT_UNIT		0x50
28 #define PWRAP_MT8135_BRIDGE_WDT_SRC_EN		0x54
29 
30 /* macro for wrapper status */
31 #define PWRAP_GET_WACS_RDATA(x)		(((x) >> 0) & 0x0000ffff)
32 #define PWRAP_GET_WACS_ARB_FSM(x)	(((x) >> 1) & 0x00000007)
33 #define PWRAP_GET_WACS_FSM(x)		(((x) >> 16) & 0x00000007)
34 #define PWRAP_GET_WACS_REQ(x)		(((x) >> 19) & 0x00000001)
35 #define PWRAP_STATE_SYNC_IDLE0		BIT(20)
36 #define PWRAP_STATE_INIT_DONE0		BIT(21)
37 #define PWRAP_STATE_INIT_DONE0_MT8186	BIT(22)
38 #define PWRAP_STATE_INIT_DONE1		BIT(15)
39 
40 /* macro for WACS FSM */
41 #define PWRAP_WACS_FSM_IDLE		0x00
42 #define PWRAP_WACS_FSM_REQ		0x02
43 #define PWRAP_WACS_FSM_WFDLE		0x04
44 #define PWRAP_WACS_FSM_WFVLDCLR		0x06
45 #define PWRAP_WACS_INIT_DONE		0x01
46 #define PWRAP_WACS_WACS_SYNC_IDLE	0x01
47 #define PWRAP_WACS_SYNC_BUSY		0x00
48 
49 /* macro for device wrapper default value */
50 #define PWRAP_DEW_READ_TEST_VAL		0x5aa5
51 #define PWRAP_DEW_COMP_READ_TEST_VAL	0xa55a
52 #define PWRAP_DEW_WRITE_TEST_VAL	0xa55a
53 
54 /* macro for manual command */
55 #define PWRAP_MAN_CMD_SPI_WRITE_NEW	(1 << 14)
56 #define PWRAP_MAN_CMD_SPI_WRITE		(1 << 13)
57 #define PWRAP_MAN_CMD_OP_CSH		(0x0 << 8)
58 #define PWRAP_MAN_CMD_OP_CSL		(0x1 << 8)
59 #define PWRAP_MAN_CMD_OP_CK		(0x2 << 8)
60 #define PWRAP_MAN_CMD_OP_OUTS		(0x8 << 8)
61 #define PWRAP_MAN_CMD_OP_OUTD		(0x9 << 8)
62 #define PWRAP_MAN_CMD_OP_OUTQ		(0xa << 8)
63 
64 /* macro for Watch Dog Timer Source */
65 #define PWRAP_WDT_SRC_EN_STAUPD_TRIG		(1 << 25)
66 #define PWRAP_WDT_SRC_EN_HARB_STAUPD_DLE	(1 << 20)
67 #define PWRAP_WDT_SRC_EN_HARB_STAUPD_ALE	(1 << 6)
68 #define PWRAP_WDT_SRC_MASK_ALL			0xffffffff
69 #define PWRAP_WDT_SRC_MASK_NO_STAUPD	~(PWRAP_WDT_SRC_EN_STAUPD_TRIG | \
70 					  PWRAP_WDT_SRC_EN_HARB_STAUPD_DLE | \
71 					  PWRAP_WDT_SRC_EN_HARB_STAUPD_ALE)
72 
73 /* Group of bits used for shown slave capability */
74 #define PWRAP_SLV_CAP_SPI	BIT(0)
75 #define PWRAP_SLV_CAP_DUALIO	BIT(1)
76 #define PWRAP_SLV_CAP_SECURITY	BIT(2)
77 #define HAS_CAP(_c, _x)	(((_c) & (_x)) == (_x))
78 
79 /* Group of bits used for shown pwrap capability */
80 #define PWRAP_CAP_BRIDGE	BIT(0)
81 #define PWRAP_CAP_RESET		BIT(1)
82 #define PWRAP_CAP_DCM		BIT(2)
83 #define PWRAP_CAP_INT1_EN	BIT(3)
84 #define PWRAP_CAP_WDT_SRC1	BIT(4)
85 #define PWRAP_CAP_ARB		BIT(5)
86 #define PWRAP_CAP_ARB_MT8186	BIT(8)
87 
88 /* defines for slave device wrapper registers */
89 enum dew_regs {
90 	PWRAP_DEW_BASE,
91 	PWRAP_DEW_DIO_EN,
92 	PWRAP_DEW_READ_TEST,
93 	PWRAP_DEW_WRITE_TEST,
94 	PWRAP_DEW_CRC_EN,
95 	PWRAP_DEW_CRC_VAL,
96 	PWRAP_DEW_MON_GRP_SEL,
97 	PWRAP_DEW_CIPHER_KEY_SEL,
98 	PWRAP_DEW_CIPHER_IV_SEL,
99 	PWRAP_DEW_CIPHER_RDY,
100 	PWRAP_DEW_CIPHER_MODE,
101 	PWRAP_DEW_CIPHER_SWRST,
102 
103 	/* MT6323 only regs */
104 	PWRAP_DEW_CIPHER_EN,
105 	PWRAP_DEW_RDDMY_NO,
106 
107 	/* MT6358 only regs */
108 	PWRAP_SMT_CON1,
109 	PWRAP_DRV_CON1,
110 	PWRAP_FILTER_CON0,
111 	PWRAP_GPIO_PULLEN0_CLR,
112 	PWRAP_RG_SPI_CON0,
113 	PWRAP_RG_SPI_RECORD0,
114 	PWRAP_RG_SPI_CON2,
115 	PWRAP_RG_SPI_CON3,
116 	PWRAP_RG_SPI_CON4,
117 	PWRAP_RG_SPI_CON5,
118 	PWRAP_RG_SPI_CON6,
119 	PWRAP_RG_SPI_CON7,
120 	PWRAP_RG_SPI_CON8,
121 	PWRAP_RG_SPI_CON13,
122 	PWRAP_SPISLV_KEY,
123 
124 	/* MT6359 only regs */
125 	PWRAP_DEW_CRC_SWRST,
126 	PWRAP_DEW_RG_EN_RECORD,
127 	PWRAP_DEW_RECORD_CMD0,
128 	PWRAP_DEW_RECORD_CMD1,
129 	PWRAP_DEW_RECORD_CMD2,
130 	PWRAP_DEW_RECORD_CMD3,
131 	PWRAP_DEW_RECORD_CMD4,
132 	PWRAP_DEW_RECORD_CMD5,
133 	PWRAP_DEW_RECORD_WDATA0,
134 	PWRAP_DEW_RECORD_WDATA1,
135 	PWRAP_DEW_RECORD_WDATA2,
136 	PWRAP_DEW_RECORD_WDATA3,
137 	PWRAP_DEW_RECORD_WDATA4,
138 	PWRAP_DEW_RECORD_WDATA5,
139 	PWRAP_DEW_RG_ADDR_TARGET,
140 	PWRAP_DEW_RG_ADDR_MASK,
141 	PWRAP_DEW_RG_WDATA_TARGET,
142 	PWRAP_DEW_RG_WDATA_MASK,
143 	PWRAP_DEW_RG_SPI_RECORD_CLR,
144 	PWRAP_DEW_RG_CMD_ALERT_CLR,
145 
146 	/* MT6397 only regs */
147 	PWRAP_DEW_EVENT_OUT_EN,
148 	PWRAP_DEW_EVENT_SRC_EN,
149 	PWRAP_DEW_EVENT_SRC,
150 	PWRAP_DEW_EVENT_FLAG,
151 	PWRAP_DEW_MON_FLAG_SEL,
152 	PWRAP_DEW_EVENT_TEST,
153 	PWRAP_DEW_CIPHER_LOAD,
154 	PWRAP_DEW_CIPHER_START,
155 };
156 
157 static const u32 mt6323_regs[] = {
158 	[PWRAP_DEW_BASE] =		0x0000,
159 	[PWRAP_DEW_DIO_EN] =		0x018a,
160 	[PWRAP_DEW_READ_TEST] =		0x018c,
161 	[PWRAP_DEW_WRITE_TEST] =	0x018e,
162 	[PWRAP_DEW_CRC_EN] =		0x0192,
163 	[PWRAP_DEW_CRC_VAL] =		0x0194,
164 	[PWRAP_DEW_MON_GRP_SEL] =	0x0196,
165 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0x0198,
166 	[PWRAP_DEW_CIPHER_IV_SEL] =	0x019a,
167 	[PWRAP_DEW_CIPHER_EN] =		0x019c,
168 	[PWRAP_DEW_CIPHER_RDY] =	0x019e,
169 	[PWRAP_DEW_CIPHER_MODE] =	0x01a0,
170 	[PWRAP_DEW_CIPHER_SWRST] =	0x01a2,
171 	[PWRAP_DEW_RDDMY_NO] =		0x01a4,
172 };
173 
174 static const u32 mt6331_regs[] = {
175 	[PWRAP_DEW_DIO_EN] =		0x018c,
176 	[PWRAP_DEW_READ_TEST] =		0x018e,
177 	[PWRAP_DEW_WRITE_TEST] =	0x0190,
178 	[PWRAP_DEW_CRC_SWRST] =		0x0192,
179 	[PWRAP_DEW_CRC_EN] =		0x0194,
180 	[PWRAP_DEW_CRC_VAL] =		0x0196,
181 	[PWRAP_DEW_MON_GRP_SEL] =	0x0198,
182 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0x019a,
183 	[PWRAP_DEW_CIPHER_IV_SEL] =	0x019c,
184 	[PWRAP_DEW_CIPHER_EN] =		0x019e,
185 	[PWRAP_DEW_CIPHER_RDY] =	0x01a0,
186 	[PWRAP_DEW_CIPHER_MODE] =	0x01a2,
187 	[PWRAP_DEW_CIPHER_SWRST] =	0x01a4,
188 	[PWRAP_DEW_RDDMY_NO] =		0x01a6,
189 };
190 
191 static const u32 mt6332_regs[] = {
192 	[PWRAP_DEW_DIO_EN] =		0x80f6,
193 	[PWRAP_DEW_READ_TEST] =		0x80f8,
194 	[PWRAP_DEW_WRITE_TEST] =	0x80fa,
195 	[PWRAP_DEW_CRC_SWRST] =		0x80fc,
196 	[PWRAP_DEW_CRC_EN] =		0x80fe,
197 	[PWRAP_DEW_CRC_VAL] =		0x8100,
198 	[PWRAP_DEW_MON_GRP_SEL] =	0x8102,
199 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0x8104,
200 	[PWRAP_DEW_CIPHER_IV_SEL] =	0x8106,
201 	[PWRAP_DEW_CIPHER_EN] =		0x8108,
202 	[PWRAP_DEW_CIPHER_RDY] =	0x810a,
203 	[PWRAP_DEW_CIPHER_MODE] =	0x810c,
204 	[PWRAP_DEW_CIPHER_SWRST] =	0x810e,
205 	[PWRAP_DEW_RDDMY_NO] =		0x8110,
206 };
207 
208 static const u32 mt6351_regs[] = {
209 	[PWRAP_DEW_DIO_EN] =		0x02F2,
210 	[PWRAP_DEW_READ_TEST] =		0x02F4,
211 	[PWRAP_DEW_WRITE_TEST] =	0x02F6,
212 	[PWRAP_DEW_CRC_EN] =		0x02FA,
213 	[PWRAP_DEW_CRC_VAL] =		0x02FC,
214 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0x0300,
215 	[PWRAP_DEW_CIPHER_IV_SEL] =	0x0302,
216 	[PWRAP_DEW_CIPHER_EN] =		0x0304,
217 	[PWRAP_DEW_CIPHER_RDY] =	0x0306,
218 	[PWRAP_DEW_CIPHER_MODE] =	0x0308,
219 	[PWRAP_DEW_CIPHER_SWRST] =	0x030A,
220 	[PWRAP_DEW_RDDMY_NO] =		0x030C,
221 };
222 
223 static const u32 mt6357_regs[] = {
224 	[PWRAP_DEW_DIO_EN] =            0x040A,
225 	[PWRAP_DEW_READ_TEST] =         0x040C,
226 	[PWRAP_DEW_WRITE_TEST] =        0x040E,
227 	[PWRAP_DEW_CRC_EN] =            0x0412,
228 	[PWRAP_DEW_CRC_VAL] =           0x0414,
229 	[PWRAP_DEW_CIPHER_KEY_SEL] =    0x0418,
230 	[PWRAP_DEW_CIPHER_IV_SEL] =     0x041A,
231 	[PWRAP_DEW_CIPHER_EN] =         0x041C,
232 	[PWRAP_DEW_CIPHER_RDY] =        0x041E,
233 	[PWRAP_DEW_CIPHER_MODE] =       0x0420,
234 	[PWRAP_DEW_CIPHER_SWRST] =      0x0422,
235 	[PWRAP_DEW_RDDMY_NO] =          0x0424,
236 };
237 
238 static const u32 mt6358_regs[] = {
239 	[PWRAP_SMT_CON1] =		0x0030,
240 	[PWRAP_DRV_CON1] =		0x0038,
241 	[PWRAP_FILTER_CON0] =		0x0040,
242 	[PWRAP_GPIO_PULLEN0_CLR] =	0x0098,
243 	[PWRAP_RG_SPI_CON0] =		0x0408,
244 	[PWRAP_RG_SPI_RECORD0] =	0x040a,
245 	[PWRAP_DEW_DIO_EN] =		0x040c,
246 	[PWRAP_DEW_READ_TEST]	=	0x040e,
247 	[PWRAP_DEW_WRITE_TEST]	=	0x0410,
248 	[PWRAP_DEW_CRC_EN] =		0x0414,
249 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0x041a,
250 	[PWRAP_DEW_CIPHER_IV_SEL] =	0x041c,
251 	[PWRAP_DEW_CIPHER_EN]	=	0x041e,
252 	[PWRAP_DEW_CIPHER_RDY] =	0x0420,
253 	[PWRAP_DEW_CIPHER_MODE] =	0x0422,
254 	[PWRAP_DEW_CIPHER_SWRST] =	0x0424,
255 	[PWRAP_RG_SPI_CON2] =		0x0432,
256 	[PWRAP_RG_SPI_CON3] =		0x0434,
257 	[PWRAP_RG_SPI_CON4] =		0x0436,
258 	[PWRAP_RG_SPI_CON5] =		0x0438,
259 	[PWRAP_RG_SPI_CON6] =		0x043a,
260 	[PWRAP_RG_SPI_CON7] =		0x043c,
261 	[PWRAP_RG_SPI_CON8] =		0x043e,
262 	[PWRAP_RG_SPI_CON13] =		0x0448,
263 	[PWRAP_SPISLV_KEY] =		0x044a,
264 };
265 
266 static const u32 mt6359_regs[] = {
267 	[PWRAP_DEW_RG_EN_RECORD] =	0x040a,
268 	[PWRAP_DEW_DIO_EN] =		0x040c,
269 	[PWRAP_DEW_READ_TEST] =		0x040e,
270 	[PWRAP_DEW_WRITE_TEST] =	0x0410,
271 	[PWRAP_DEW_CRC_SWRST] =		0x0412,
272 	[PWRAP_DEW_CRC_EN] =		0x0414,
273 	[PWRAP_DEW_CRC_VAL] =		0x0416,
274 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0x0418,
275 	[PWRAP_DEW_CIPHER_IV_SEL] =	0x041a,
276 	[PWRAP_DEW_CIPHER_EN] =		0x041c,
277 	[PWRAP_DEW_CIPHER_RDY] =	0x041e,
278 	[PWRAP_DEW_CIPHER_MODE] =	0x0420,
279 	[PWRAP_DEW_CIPHER_SWRST] =	0x0422,
280 	[PWRAP_DEW_RDDMY_NO] =		0x0424,
281 	[PWRAP_DEW_RECORD_CMD0] =	0x0428,
282 	[PWRAP_DEW_RECORD_CMD1] =	0x042a,
283 	[PWRAP_DEW_RECORD_CMD2] =	0x042c,
284 	[PWRAP_DEW_RECORD_CMD3] =	0x042e,
285 	[PWRAP_DEW_RECORD_CMD4] =	0x0430,
286 	[PWRAP_DEW_RECORD_CMD5] =	0x0432,
287 	[PWRAP_DEW_RECORD_WDATA0] =	0x0434,
288 	[PWRAP_DEW_RECORD_WDATA1] =	0x0436,
289 	[PWRAP_DEW_RECORD_WDATA2] =	0x0438,
290 	[PWRAP_DEW_RECORD_WDATA3] =	0x043a,
291 	[PWRAP_DEW_RECORD_WDATA4] =	0x043c,
292 	[PWRAP_DEW_RECORD_WDATA5] =	0x043e,
293 	[PWRAP_DEW_RG_ADDR_TARGET] =	0x0440,
294 	[PWRAP_DEW_RG_ADDR_MASK] =	0x0442,
295 	[PWRAP_DEW_RG_WDATA_TARGET] =	0x0444,
296 	[PWRAP_DEW_RG_WDATA_MASK] =	0x0446,
297 	[PWRAP_DEW_RG_SPI_RECORD_CLR] =	0x0448,
298 	[PWRAP_DEW_RG_CMD_ALERT_CLR] =	0x0448,
299 	[PWRAP_SPISLV_KEY] =		0x044a,
300 };
301 
302 static const u32 mt6397_regs[] = {
303 	[PWRAP_DEW_BASE] =		0xbc00,
304 	[PWRAP_DEW_EVENT_OUT_EN] =	0xbc00,
305 	[PWRAP_DEW_DIO_EN] =		0xbc02,
306 	[PWRAP_DEW_EVENT_SRC_EN] =	0xbc04,
307 	[PWRAP_DEW_EVENT_SRC] =		0xbc06,
308 	[PWRAP_DEW_EVENT_FLAG] =	0xbc08,
309 	[PWRAP_DEW_READ_TEST] =		0xbc0a,
310 	[PWRAP_DEW_WRITE_TEST] =	0xbc0c,
311 	[PWRAP_DEW_CRC_EN] =		0xbc0e,
312 	[PWRAP_DEW_CRC_VAL] =		0xbc10,
313 	[PWRAP_DEW_MON_GRP_SEL] =	0xbc12,
314 	[PWRAP_DEW_MON_FLAG_SEL] =	0xbc14,
315 	[PWRAP_DEW_EVENT_TEST] =	0xbc16,
316 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0xbc18,
317 	[PWRAP_DEW_CIPHER_IV_SEL] =	0xbc1a,
318 	[PWRAP_DEW_CIPHER_LOAD] =	0xbc1c,
319 	[PWRAP_DEW_CIPHER_START] =	0xbc1e,
320 	[PWRAP_DEW_CIPHER_RDY] =	0xbc20,
321 	[PWRAP_DEW_CIPHER_MODE] =	0xbc22,
322 	[PWRAP_DEW_CIPHER_SWRST] =	0xbc24,
323 };
324 
325 enum pwrap_regs {
326 	PWRAP_MUX_SEL,
327 	PWRAP_WRAP_EN,
328 	PWRAP_DIO_EN,
329 	PWRAP_SIDLY,
330 	PWRAP_CSHEXT_WRITE,
331 	PWRAP_CSHEXT_READ,
332 	PWRAP_CSLEXT_START,
333 	PWRAP_CSLEXT_END,
334 	PWRAP_STAUPD_PRD,
335 	PWRAP_STAUPD_GRPEN,
336 	PWRAP_STAUPD_MAN_TRIG,
337 	PWRAP_STAUPD_STA,
338 	PWRAP_WRAP_STA,
339 	PWRAP_HARB_INIT,
340 	PWRAP_HARB_HPRIO,
341 	PWRAP_HIPRIO_ARB_EN,
342 	PWRAP_HARB_STA0,
343 	PWRAP_HARB_STA1,
344 	PWRAP_MAN_EN,
345 	PWRAP_MAN_CMD,
346 	PWRAP_MAN_RDATA,
347 	PWRAP_MAN_VLDCLR,
348 	PWRAP_WACS0_EN,
349 	PWRAP_INIT_DONE0,
350 	PWRAP_WACS0_CMD,
351 	PWRAP_WACS0_RDATA,
352 	PWRAP_WACS0_VLDCLR,
353 	PWRAP_WACS1_EN,
354 	PWRAP_INIT_DONE1,
355 	PWRAP_WACS1_CMD,
356 	PWRAP_WACS1_RDATA,
357 	PWRAP_WACS1_VLDCLR,
358 	PWRAP_WACS2_EN,
359 	PWRAP_INIT_DONE2,
360 	PWRAP_WACS2_CMD,
361 	PWRAP_WACS2_RDATA,
362 	PWRAP_WACS2_VLDCLR,
363 	PWRAP_INT_EN,
364 	PWRAP_INT_FLG_RAW,
365 	PWRAP_INT_FLG,
366 	PWRAP_INT_CLR,
367 	PWRAP_SIG_ADR,
368 	PWRAP_SIG_MODE,
369 	PWRAP_SIG_VALUE,
370 	PWRAP_SIG_ERRVAL,
371 	PWRAP_CRC_EN,
372 	PWRAP_TIMER_EN,
373 	PWRAP_TIMER_STA,
374 	PWRAP_WDT_UNIT,
375 	PWRAP_WDT_SRC_EN,
376 	PWRAP_WDT_FLG,
377 	PWRAP_DEBUG_INT_SEL,
378 	PWRAP_CIPHER_KEY_SEL,
379 	PWRAP_CIPHER_IV_SEL,
380 	PWRAP_CIPHER_RDY,
381 	PWRAP_CIPHER_MODE,
382 	PWRAP_CIPHER_SWRST,
383 	PWRAP_DCM_EN,
384 	PWRAP_DCM_DBC_PRD,
385 	PWRAP_EINT_STA0_ADR,
386 	PWRAP_EINT_STA1_ADR,
387 	PWRAP_SWINF_2_WDATA_31_0,
388 	PWRAP_SWINF_2_RDATA_31_0,
389 
390 	/* MT2701 only regs */
391 	PWRAP_ADC_CMD_ADDR,
392 	PWRAP_PWRAP_ADC_CMD,
393 	PWRAP_ADC_RDY_ADDR,
394 	PWRAP_ADC_RDATA_ADDR1,
395 	PWRAP_ADC_RDATA_ADDR2,
396 
397 	/* MT7622 only regs */
398 	PWRAP_STA,
399 	PWRAP_CLR,
400 	PWRAP_DVFS_ADR8,
401 	PWRAP_DVFS_WDATA8,
402 	PWRAP_DVFS_ADR9,
403 	PWRAP_DVFS_WDATA9,
404 	PWRAP_DVFS_ADR10,
405 	PWRAP_DVFS_WDATA10,
406 	PWRAP_DVFS_ADR11,
407 	PWRAP_DVFS_WDATA11,
408 	PWRAP_DVFS_ADR12,
409 	PWRAP_DVFS_WDATA12,
410 	PWRAP_DVFS_ADR13,
411 	PWRAP_DVFS_WDATA13,
412 	PWRAP_DVFS_ADR14,
413 	PWRAP_DVFS_WDATA14,
414 	PWRAP_DVFS_ADR15,
415 	PWRAP_DVFS_WDATA15,
416 	PWRAP_EXT_CK,
417 	PWRAP_ADC_RDATA_ADDR,
418 	PWRAP_GPS_STA,
419 	PWRAP_SW_RST,
420 	PWRAP_DVFS_STEP_CTRL0,
421 	PWRAP_DVFS_STEP_CTRL1,
422 	PWRAP_DVFS_STEP_CTRL2,
423 	PWRAP_SPI2_CTRL,
424 
425 	/* MT8135 only regs */
426 	PWRAP_CSHEXT,
427 	PWRAP_EVENT_IN_EN,
428 	PWRAP_EVENT_DST_EN,
429 	PWRAP_RRARB_INIT,
430 	PWRAP_RRARB_EN,
431 	PWRAP_RRARB_STA0,
432 	PWRAP_RRARB_STA1,
433 	PWRAP_EVENT_STA,
434 	PWRAP_EVENT_STACLR,
435 	PWRAP_CIPHER_LOAD,
436 	PWRAP_CIPHER_START,
437 
438 	/* MT8173 only regs */
439 	PWRAP_RDDMY,
440 	PWRAP_SI_CK_CON,
441 	PWRAP_DVFS_ADR0,
442 	PWRAP_DVFS_WDATA0,
443 	PWRAP_DVFS_ADR1,
444 	PWRAP_DVFS_WDATA1,
445 	PWRAP_DVFS_ADR2,
446 	PWRAP_DVFS_WDATA2,
447 	PWRAP_DVFS_ADR3,
448 	PWRAP_DVFS_WDATA3,
449 	PWRAP_DVFS_ADR4,
450 	PWRAP_DVFS_WDATA4,
451 	PWRAP_DVFS_ADR5,
452 	PWRAP_DVFS_WDATA5,
453 	PWRAP_DVFS_ADR6,
454 	PWRAP_DVFS_WDATA6,
455 	PWRAP_DVFS_ADR7,
456 	PWRAP_DVFS_WDATA7,
457 	PWRAP_SPMINF_STA,
458 	PWRAP_CIPHER_EN,
459 
460 	/* MT8183 only regs */
461 	PWRAP_SI_SAMPLE_CTRL,
462 	PWRAP_CSLEXT_WRITE,
463 	PWRAP_CSLEXT_READ,
464 	PWRAP_EXT_CK_WRITE,
465 	PWRAP_STAUPD_CTRL,
466 	PWRAP_WACS_P2P_EN,
467 	PWRAP_INIT_DONE_P2P,
468 	PWRAP_WACS_MD32_EN,
469 	PWRAP_INIT_DONE_MD32,
470 	PWRAP_INT1_EN,
471 	PWRAP_INT1_FLG,
472 	PWRAP_INT1_CLR,
473 	PWRAP_WDT_SRC_EN_1,
474 	PWRAP_INT_GPS_AUXADC_CMD_ADDR,
475 	PWRAP_INT_GPS_AUXADC_CMD,
476 	PWRAP_INT_GPS_AUXADC_RDATA_ADDR,
477 	PWRAP_EXT_GPS_AUXADC_RDATA_ADDR,
478 	PWRAP_GPSINF_0_STA,
479 	PWRAP_GPSINF_1_STA,
480 
481 	/* MT8516 only regs */
482 	PWRAP_OP_TYPE,
483 	PWRAP_MSB_FIRST,
484 };
485 
486 static int mt2701_regs[] = {
487 	[PWRAP_MUX_SEL] =		0x0,
488 	[PWRAP_WRAP_EN] =		0x4,
489 	[PWRAP_DIO_EN] =		0x8,
490 	[PWRAP_SIDLY] =			0xc,
491 	[PWRAP_RDDMY] =			0x18,
492 	[PWRAP_SI_CK_CON] =		0x1c,
493 	[PWRAP_CSHEXT_WRITE] =		0x20,
494 	[PWRAP_CSHEXT_READ] =		0x24,
495 	[PWRAP_CSLEXT_START] =		0x28,
496 	[PWRAP_CSLEXT_END] =		0x2c,
497 	[PWRAP_STAUPD_PRD] =		0x30,
498 	[PWRAP_STAUPD_GRPEN] =		0x34,
499 	[PWRAP_STAUPD_MAN_TRIG] =	0x38,
500 	[PWRAP_STAUPD_STA] =		0x3c,
501 	[PWRAP_WRAP_STA] =		0x44,
502 	[PWRAP_HARB_INIT] =		0x48,
503 	[PWRAP_HARB_HPRIO] =		0x4c,
504 	[PWRAP_HIPRIO_ARB_EN] =		0x50,
505 	[PWRAP_HARB_STA0] =		0x54,
506 	[PWRAP_HARB_STA1] =		0x58,
507 	[PWRAP_MAN_EN] =		0x5c,
508 	[PWRAP_MAN_CMD] =		0x60,
509 	[PWRAP_MAN_RDATA] =		0x64,
510 	[PWRAP_MAN_VLDCLR] =		0x68,
511 	[PWRAP_WACS0_EN] =		0x6c,
512 	[PWRAP_INIT_DONE0] =		0x70,
513 	[PWRAP_WACS0_CMD] =		0x74,
514 	[PWRAP_WACS0_RDATA] =		0x78,
515 	[PWRAP_WACS0_VLDCLR] =		0x7c,
516 	[PWRAP_WACS1_EN] =		0x80,
517 	[PWRAP_INIT_DONE1] =		0x84,
518 	[PWRAP_WACS1_CMD] =		0x88,
519 	[PWRAP_WACS1_RDATA] =		0x8c,
520 	[PWRAP_WACS1_VLDCLR] =		0x90,
521 	[PWRAP_WACS2_EN] =		0x94,
522 	[PWRAP_INIT_DONE2] =		0x98,
523 	[PWRAP_WACS2_CMD] =		0x9c,
524 	[PWRAP_WACS2_RDATA] =		0xa0,
525 	[PWRAP_WACS2_VLDCLR] =		0xa4,
526 	[PWRAP_INT_EN] =		0xa8,
527 	[PWRAP_INT_FLG_RAW] =		0xac,
528 	[PWRAP_INT_FLG] =		0xb0,
529 	[PWRAP_INT_CLR] =		0xb4,
530 	[PWRAP_SIG_ADR] =		0xb8,
531 	[PWRAP_SIG_MODE] =		0xbc,
532 	[PWRAP_SIG_VALUE] =		0xc0,
533 	[PWRAP_SIG_ERRVAL] =		0xc4,
534 	[PWRAP_CRC_EN] =		0xc8,
535 	[PWRAP_TIMER_EN] =		0xcc,
536 	[PWRAP_TIMER_STA] =		0xd0,
537 	[PWRAP_WDT_UNIT] =		0xd4,
538 	[PWRAP_WDT_SRC_EN] =		0xd8,
539 	[PWRAP_WDT_FLG] =		0xdc,
540 	[PWRAP_DEBUG_INT_SEL] =		0xe0,
541 	[PWRAP_DVFS_ADR0] =		0xe4,
542 	[PWRAP_DVFS_WDATA0] =		0xe8,
543 	[PWRAP_DVFS_ADR1] =		0xec,
544 	[PWRAP_DVFS_WDATA1] =		0xf0,
545 	[PWRAP_DVFS_ADR2] =		0xf4,
546 	[PWRAP_DVFS_WDATA2] =		0xf8,
547 	[PWRAP_DVFS_ADR3] =		0xfc,
548 	[PWRAP_DVFS_WDATA3] =		0x100,
549 	[PWRAP_DVFS_ADR4] =		0x104,
550 	[PWRAP_DVFS_WDATA4] =		0x108,
551 	[PWRAP_DVFS_ADR5] =		0x10c,
552 	[PWRAP_DVFS_WDATA5] =		0x110,
553 	[PWRAP_DVFS_ADR6] =		0x114,
554 	[PWRAP_DVFS_WDATA6] =		0x118,
555 	[PWRAP_DVFS_ADR7] =		0x11c,
556 	[PWRAP_DVFS_WDATA7] =		0x120,
557 	[PWRAP_CIPHER_KEY_SEL] =	0x124,
558 	[PWRAP_CIPHER_IV_SEL] =		0x128,
559 	[PWRAP_CIPHER_EN] =		0x12c,
560 	[PWRAP_CIPHER_RDY] =		0x130,
561 	[PWRAP_CIPHER_MODE] =		0x134,
562 	[PWRAP_CIPHER_SWRST] =		0x138,
563 	[PWRAP_DCM_EN] =		0x13c,
564 	[PWRAP_DCM_DBC_PRD] =		0x140,
565 	[PWRAP_ADC_CMD_ADDR] =		0x144,
566 	[PWRAP_PWRAP_ADC_CMD] =		0x148,
567 	[PWRAP_ADC_RDY_ADDR] =		0x14c,
568 	[PWRAP_ADC_RDATA_ADDR1] =	0x150,
569 	[PWRAP_ADC_RDATA_ADDR2] =	0x154,
570 };
571 
572 static int mt6765_regs[] = {
573 	[PWRAP_MUX_SEL] =		0x0,
574 	[PWRAP_WRAP_EN] =		0x4,
575 	[PWRAP_DIO_EN] =		0x8,
576 	[PWRAP_RDDMY] =			0x20,
577 	[PWRAP_CSHEXT_WRITE] =		0x24,
578 	[PWRAP_CSHEXT_READ] =		0x28,
579 	[PWRAP_CSLEXT_START] =		0x2C,
580 	[PWRAP_CSLEXT_END] =		0x30,
581 	[PWRAP_STAUPD_PRD] =		0x3C,
582 	[PWRAP_HARB_HPRIO] =		0x68,
583 	[PWRAP_HIPRIO_ARB_EN] =		0x6C,
584 	[PWRAP_MAN_EN] =		0x7C,
585 	[PWRAP_MAN_CMD] =		0x80,
586 	[PWRAP_WACS0_EN] =		0x8C,
587 	[PWRAP_WACS1_EN] =		0x94,
588 	[PWRAP_WACS2_EN] =		0x9C,
589 	[PWRAP_INIT_DONE2] =		0xA0,
590 	[PWRAP_WACS2_CMD] =		0xC20,
591 	[PWRAP_WACS2_RDATA] =		0xC24,
592 	[PWRAP_WACS2_VLDCLR] =		0xC28,
593 	[PWRAP_INT_EN] =		0xB4,
594 	[PWRAP_INT_FLG_RAW] =		0xB8,
595 	[PWRAP_INT_FLG] =		0xBC,
596 	[PWRAP_INT_CLR] =		0xC0,
597 	[PWRAP_TIMER_EN] =		0xE8,
598 	[PWRAP_WDT_UNIT] =		0xF0,
599 	[PWRAP_WDT_SRC_EN] =		0xF4,
600 	[PWRAP_DCM_EN] =		0x1DC,
601 	[PWRAP_DCM_DBC_PRD] =		0x1E0,
602 };
603 
604 static int mt6779_regs[] = {
605 	[PWRAP_MUX_SEL] =		0x0,
606 	[PWRAP_WRAP_EN] =		0x4,
607 	[PWRAP_DIO_EN] =		0x8,
608 	[PWRAP_RDDMY] =			0x20,
609 	[PWRAP_CSHEXT_WRITE] =		0x24,
610 	[PWRAP_CSHEXT_READ] =		0x28,
611 	[PWRAP_CSLEXT_WRITE] =		0x2C,
612 	[PWRAP_CSLEXT_READ] =		0x30,
613 	[PWRAP_EXT_CK_WRITE] =		0x34,
614 	[PWRAP_STAUPD_CTRL] =		0x3C,
615 	[PWRAP_STAUPD_GRPEN] =		0x40,
616 	[PWRAP_EINT_STA0_ADR] =		0x44,
617 	[PWRAP_HARB_HPRIO] =		0x68,
618 	[PWRAP_HIPRIO_ARB_EN] =		0x6C,
619 	[PWRAP_MAN_EN] =		0x7C,
620 	[PWRAP_MAN_CMD] =		0x80,
621 	[PWRAP_WACS0_EN] =		0x8C,
622 	[PWRAP_INIT_DONE0] =		0x90,
623 	[PWRAP_WACS1_EN] =		0x94,
624 	[PWRAP_WACS2_EN] =		0x9C,
625 	[PWRAP_INIT_DONE1] =		0x98,
626 	[PWRAP_INIT_DONE2] =		0xA0,
627 	[PWRAP_INT_EN] =		0xBC,
628 	[PWRAP_INT_FLG_RAW] =		0xC0,
629 	[PWRAP_INT_FLG] =		0xC4,
630 	[PWRAP_INT_CLR] =		0xC8,
631 	[PWRAP_INT1_EN] =		0xCC,
632 	[PWRAP_INT1_FLG] =		0xD4,
633 	[PWRAP_INT1_CLR] =		0xD8,
634 	[PWRAP_TIMER_EN] =		0xF0,
635 	[PWRAP_WDT_UNIT] =		0xF8,
636 	[PWRAP_WDT_SRC_EN] =		0xFC,
637 	[PWRAP_WDT_SRC_EN_1] =		0x100,
638 	[PWRAP_WACS2_CMD] =		0xC20,
639 	[PWRAP_WACS2_RDATA] =		0xC24,
640 	[PWRAP_WACS2_VLDCLR] =		0xC28,
641 };
642 
643 static int mt6795_regs[] = {
644 	[PWRAP_MUX_SEL] =		0x0,
645 	[PWRAP_WRAP_EN] =		0x4,
646 	[PWRAP_DIO_EN] =		0x8,
647 	[PWRAP_SIDLY] =			0xc,
648 	[PWRAP_RDDMY] =			0x10,
649 	[PWRAP_SI_CK_CON] =		0x14,
650 	[PWRAP_CSHEXT_WRITE] =		0x18,
651 	[PWRAP_CSHEXT_READ] =		0x1c,
652 	[PWRAP_CSLEXT_START] =		0x20,
653 	[PWRAP_CSLEXT_END] =		0x24,
654 	[PWRAP_STAUPD_PRD] =		0x28,
655 	[PWRAP_STAUPD_GRPEN] =		0x2c,
656 	[PWRAP_EINT_STA0_ADR] =		0x30,
657 	[PWRAP_EINT_STA1_ADR] =		0x34,
658 	[PWRAP_STAUPD_MAN_TRIG] =	0x40,
659 	[PWRAP_STAUPD_STA] =		0x44,
660 	[PWRAP_WRAP_STA] =		0x48,
661 	[PWRAP_HARB_INIT] =		0x4c,
662 	[PWRAP_HARB_HPRIO] =		0x50,
663 	[PWRAP_HIPRIO_ARB_EN] =		0x54,
664 	[PWRAP_HARB_STA0] =		0x58,
665 	[PWRAP_HARB_STA1] =		0x5c,
666 	[PWRAP_MAN_EN] =		0x60,
667 	[PWRAP_MAN_CMD] =		0x64,
668 	[PWRAP_MAN_RDATA] =		0x68,
669 	[PWRAP_MAN_VLDCLR] =		0x6c,
670 	[PWRAP_WACS0_EN] =		0x70,
671 	[PWRAP_INIT_DONE0] =		0x74,
672 	[PWRAP_WACS0_CMD] =		0x78,
673 	[PWRAP_WACS0_RDATA] =		0x7c,
674 	[PWRAP_WACS0_VLDCLR] =		0x80,
675 	[PWRAP_WACS1_EN] =		0x84,
676 	[PWRAP_INIT_DONE1] =		0x88,
677 	[PWRAP_WACS1_CMD] =		0x8c,
678 	[PWRAP_WACS1_RDATA] =		0x90,
679 	[PWRAP_WACS1_VLDCLR] =		0x94,
680 	[PWRAP_WACS2_EN] =		0x98,
681 	[PWRAP_INIT_DONE2] =		0x9c,
682 	[PWRAP_WACS2_CMD] =		0xa0,
683 	[PWRAP_WACS2_RDATA] =		0xa4,
684 	[PWRAP_WACS2_VLDCLR] =		0xa8,
685 	[PWRAP_INT_EN] =		0xac,
686 	[PWRAP_INT_FLG_RAW] =		0xb0,
687 	[PWRAP_INT_FLG] =		0xb4,
688 	[PWRAP_INT_CLR] =		0xb8,
689 	[PWRAP_SIG_ADR] =		0xbc,
690 	[PWRAP_SIG_MODE] =		0xc0,
691 	[PWRAP_SIG_VALUE] =		0xc4,
692 	[PWRAP_SIG_ERRVAL] =		0xc8,
693 	[PWRAP_CRC_EN] =		0xcc,
694 	[PWRAP_TIMER_EN] =		0xd0,
695 	[PWRAP_TIMER_STA] =		0xd4,
696 	[PWRAP_WDT_UNIT] =		0xd8,
697 	[PWRAP_WDT_SRC_EN] =		0xdc,
698 	[PWRAP_WDT_FLG] =		0xe0,
699 	[PWRAP_DEBUG_INT_SEL] =		0xe4,
700 	[PWRAP_DVFS_ADR0] =		0xe8,
701 	[PWRAP_DVFS_WDATA0] =		0xec,
702 	[PWRAP_DVFS_ADR1] =		0xf0,
703 	[PWRAP_DVFS_WDATA1] =		0xf4,
704 	[PWRAP_DVFS_ADR2] =		0xf8,
705 	[PWRAP_DVFS_WDATA2] =		0xfc,
706 	[PWRAP_DVFS_ADR3] =		0x100,
707 	[PWRAP_DVFS_WDATA3] =		0x104,
708 	[PWRAP_DVFS_ADR4] =		0x108,
709 	[PWRAP_DVFS_WDATA4] =		0x10c,
710 	[PWRAP_DVFS_ADR5] =		0x110,
711 	[PWRAP_DVFS_WDATA5] =		0x114,
712 	[PWRAP_DVFS_ADR6] =		0x118,
713 	[PWRAP_DVFS_WDATA6] =		0x11c,
714 	[PWRAP_DVFS_ADR7] =		0x120,
715 	[PWRAP_DVFS_WDATA7] =		0x124,
716 	[PWRAP_SPMINF_STA] =		0x128,
717 	[PWRAP_CIPHER_KEY_SEL] =	0x12c,
718 	[PWRAP_CIPHER_IV_SEL] =		0x130,
719 	[PWRAP_CIPHER_EN] =		0x134,
720 	[PWRAP_CIPHER_RDY] =		0x138,
721 	[PWRAP_CIPHER_MODE] =		0x13c,
722 	[PWRAP_CIPHER_SWRST] =		0x140,
723 	[PWRAP_DCM_EN] =		0x144,
724 	[PWRAP_DCM_DBC_PRD] =		0x148,
725 	[PWRAP_EXT_CK] =		0x14c,
726 };
727 
728 static int mt6797_regs[] = {
729 	[PWRAP_MUX_SEL] =		0x0,
730 	[PWRAP_WRAP_EN] =		0x4,
731 	[PWRAP_DIO_EN] =		0x8,
732 	[PWRAP_SIDLY] =			0xC,
733 	[PWRAP_RDDMY] =			0x10,
734 	[PWRAP_CSHEXT_WRITE] =		0x18,
735 	[PWRAP_CSHEXT_READ] =		0x1C,
736 	[PWRAP_CSLEXT_START] =		0x20,
737 	[PWRAP_CSLEXT_END] =		0x24,
738 	[PWRAP_STAUPD_PRD] =		0x28,
739 	[PWRAP_HARB_HPRIO] =		0x50,
740 	[PWRAP_HIPRIO_ARB_EN] =		0x54,
741 	[PWRAP_MAN_EN] =		0x60,
742 	[PWRAP_MAN_CMD] =		0x64,
743 	[PWRAP_WACS0_EN] =		0x70,
744 	[PWRAP_WACS1_EN] =		0x84,
745 	[PWRAP_WACS2_EN] =		0x98,
746 	[PWRAP_INIT_DONE2] =		0x9C,
747 	[PWRAP_WACS2_CMD] =		0xA0,
748 	[PWRAP_WACS2_RDATA] =		0xA4,
749 	[PWRAP_WACS2_VLDCLR] =		0xA8,
750 	[PWRAP_INT_EN] =		0xC0,
751 	[PWRAP_INT_FLG_RAW] =		0xC4,
752 	[PWRAP_INT_FLG] =		0xC8,
753 	[PWRAP_INT_CLR] =		0xCC,
754 	[PWRAP_TIMER_EN] =		0xF4,
755 	[PWRAP_WDT_UNIT] =		0xFC,
756 	[PWRAP_WDT_SRC_EN] =		0x100,
757 	[PWRAP_DCM_EN] =		0x1CC,
758 	[PWRAP_DCM_DBC_PRD] =		0x1D4,
759 };
760 
761 static int mt6873_regs[] = {
762 	[PWRAP_INIT_DONE2] =		0x0,
763 	[PWRAP_TIMER_EN] =		0x3E0,
764 	[PWRAP_INT_EN] =		0x448,
765 	[PWRAP_WACS2_CMD] =		0xC80,
766 	[PWRAP_SWINF_2_WDATA_31_0] =	0xC84,
767 	[PWRAP_SWINF_2_RDATA_31_0] =	0xC94,
768 	[PWRAP_WACS2_VLDCLR] =		0xCA4,
769 	[PWRAP_WACS2_RDATA] =		0xCA8,
770 };
771 
772 static int mt7622_regs[] = {
773 	[PWRAP_MUX_SEL] =		0x0,
774 	[PWRAP_WRAP_EN] =		0x4,
775 	[PWRAP_DIO_EN] =		0x8,
776 	[PWRAP_SIDLY] =			0xC,
777 	[PWRAP_RDDMY] =			0x10,
778 	[PWRAP_SI_CK_CON] =		0x14,
779 	[PWRAP_CSHEXT_WRITE] =		0x18,
780 	[PWRAP_CSHEXT_READ] =		0x1C,
781 	[PWRAP_CSLEXT_START] =		0x20,
782 	[PWRAP_CSLEXT_END] =		0x24,
783 	[PWRAP_STAUPD_PRD] =		0x28,
784 	[PWRAP_STAUPD_GRPEN] =		0x2C,
785 	[PWRAP_EINT_STA0_ADR] =		0x30,
786 	[PWRAP_EINT_STA1_ADR] =		0x34,
787 	[PWRAP_STA] =			0x38,
788 	[PWRAP_CLR] =			0x3C,
789 	[PWRAP_STAUPD_MAN_TRIG] =	0x40,
790 	[PWRAP_STAUPD_STA] =		0x44,
791 	[PWRAP_WRAP_STA] =		0x48,
792 	[PWRAP_HARB_INIT] =		0x4C,
793 	[PWRAP_HARB_HPRIO] =		0x50,
794 	[PWRAP_HIPRIO_ARB_EN] =		0x54,
795 	[PWRAP_HARB_STA0] =		0x58,
796 	[PWRAP_HARB_STA1] =		0x5C,
797 	[PWRAP_MAN_EN] =		0x60,
798 	[PWRAP_MAN_CMD] =		0x64,
799 	[PWRAP_MAN_RDATA] =		0x68,
800 	[PWRAP_MAN_VLDCLR] =		0x6C,
801 	[PWRAP_WACS0_EN] =		0x70,
802 	[PWRAP_INIT_DONE0] =		0x74,
803 	[PWRAP_WACS0_CMD] =		0x78,
804 	[PWRAP_WACS0_RDATA] =		0x7C,
805 	[PWRAP_WACS0_VLDCLR] =		0x80,
806 	[PWRAP_WACS1_EN] =		0x84,
807 	[PWRAP_INIT_DONE1] =		0x88,
808 	[PWRAP_WACS1_CMD] =		0x8C,
809 	[PWRAP_WACS1_RDATA] =		0x90,
810 	[PWRAP_WACS1_VLDCLR] =		0x94,
811 	[PWRAP_WACS2_EN] =		0x98,
812 	[PWRAP_INIT_DONE2] =		0x9C,
813 	[PWRAP_WACS2_CMD] =		0xA0,
814 	[PWRAP_WACS2_RDATA] =		0xA4,
815 	[PWRAP_WACS2_VLDCLR] =		0xA8,
816 	[PWRAP_INT_EN] =		0xAC,
817 	[PWRAP_INT_FLG_RAW] =		0xB0,
818 	[PWRAP_INT_FLG] =		0xB4,
819 	[PWRAP_INT_CLR] =		0xB8,
820 	[PWRAP_SIG_ADR] =		0xBC,
821 	[PWRAP_SIG_MODE] =		0xC0,
822 	[PWRAP_SIG_VALUE] =		0xC4,
823 	[PWRAP_SIG_ERRVAL] =		0xC8,
824 	[PWRAP_CRC_EN] =		0xCC,
825 	[PWRAP_TIMER_EN] =		0xD0,
826 	[PWRAP_TIMER_STA] =		0xD4,
827 	[PWRAP_WDT_UNIT] =		0xD8,
828 	[PWRAP_WDT_SRC_EN] =		0xDC,
829 	[PWRAP_WDT_FLG] =		0xE0,
830 	[PWRAP_DEBUG_INT_SEL] =		0xE4,
831 	[PWRAP_DVFS_ADR0] =		0xE8,
832 	[PWRAP_DVFS_WDATA0] =		0xEC,
833 	[PWRAP_DVFS_ADR1] =		0xF0,
834 	[PWRAP_DVFS_WDATA1] =		0xF4,
835 	[PWRAP_DVFS_ADR2] =		0xF8,
836 	[PWRAP_DVFS_WDATA2] =		0xFC,
837 	[PWRAP_DVFS_ADR3] =		0x100,
838 	[PWRAP_DVFS_WDATA3] =		0x104,
839 	[PWRAP_DVFS_ADR4] =		0x108,
840 	[PWRAP_DVFS_WDATA4] =		0x10C,
841 	[PWRAP_DVFS_ADR5] =		0x110,
842 	[PWRAP_DVFS_WDATA5] =		0x114,
843 	[PWRAP_DVFS_ADR6] =		0x118,
844 	[PWRAP_DVFS_WDATA6] =		0x11C,
845 	[PWRAP_DVFS_ADR7] =		0x120,
846 	[PWRAP_DVFS_WDATA7] =		0x124,
847 	[PWRAP_DVFS_ADR8] =		0x128,
848 	[PWRAP_DVFS_WDATA8] =		0x12C,
849 	[PWRAP_DVFS_ADR9] =		0x130,
850 	[PWRAP_DVFS_WDATA9] =		0x134,
851 	[PWRAP_DVFS_ADR10] =		0x138,
852 	[PWRAP_DVFS_WDATA10] =		0x13C,
853 	[PWRAP_DVFS_ADR11] =		0x140,
854 	[PWRAP_DVFS_WDATA11] =		0x144,
855 	[PWRAP_DVFS_ADR12] =		0x148,
856 	[PWRAP_DVFS_WDATA12] =		0x14C,
857 	[PWRAP_DVFS_ADR13] =		0x150,
858 	[PWRAP_DVFS_WDATA13] =		0x154,
859 	[PWRAP_DVFS_ADR14] =		0x158,
860 	[PWRAP_DVFS_WDATA14] =		0x15C,
861 	[PWRAP_DVFS_ADR15] =		0x160,
862 	[PWRAP_DVFS_WDATA15] =		0x164,
863 	[PWRAP_SPMINF_STA] =		0x168,
864 	[PWRAP_CIPHER_KEY_SEL] =	0x16C,
865 	[PWRAP_CIPHER_IV_SEL] =		0x170,
866 	[PWRAP_CIPHER_EN] =		0x174,
867 	[PWRAP_CIPHER_RDY] =		0x178,
868 	[PWRAP_CIPHER_MODE] =		0x17C,
869 	[PWRAP_CIPHER_SWRST] =		0x180,
870 	[PWRAP_DCM_EN] =		0x184,
871 	[PWRAP_DCM_DBC_PRD] =		0x188,
872 	[PWRAP_EXT_CK] =		0x18C,
873 	[PWRAP_ADC_CMD_ADDR] =		0x190,
874 	[PWRAP_PWRAP_ADC_CMD] =		0x194,
875 	[PWRAP_ADC_RDATA_ADDR] =	0x198,
876 	[PWRAP_GPS_STA] =		0x19C,
877 	[PWRAP_SW_RST] =		0x1A0,
878 	[PWRAP_DVFS_STEP_CTRL0] =	0x238,
879 	[PWRAP_DVFS_STEP_CTRL1] =	0x23C,
880 	[PWRAP_DVFS_STEP_CTRL2] =	0x240,
881 	[PWRAP_SPI2_CTRL] =		0x244,
882 };
883 
884 static int mt8135_regs[] = {
885 	[PWRAP_MUX_SEL] =		0x0,
886 	[PWRAP_WRAP_EN] =		0x4,
887 	[PWRAP_DIO_EN] =		0x8,
888 	[PWRAP_SIDLY] =			0xc,
889 	[PWRAP_CSHEXT] =		0x10,
890 	[PWRAP_CSHEXT_WRITE] =		0x14,
891 	[PWRAP_CSHEXT_READ] =		0x18,
892 	[PWRAP_CSLEXT_START] =		0x1c,
893 	[PWRAP_CSLEXT_END] =		0x20,
894 	[PWRAP_STAUPD_PRD] =		0x24,
895 	[PWRAP_STAUPD_GRPEN] =		0x28,
896 	[PWRAP_STAUPD_MAN_TRIG] =	0x2c,
897 	[PWRAP_STAUPD_STA] =		0x30,
898 	[PWRAP_EVENT_IN_EN] =		0x34,
899 	[PWRAP_EVENT_DST_EN] =		0x38,
900 	[PWRAP_WRAP_STA] =		0x3c,
901 	[PWRAP_RRARB_INIT] =		0x40,
902 	[PWRAP_RRARB_EN] =		0x44,
903 	[PWRAP_RRARB_STA0] =		0x48,
904 	[PWRAP_RRARB_STA1] =		0x4c,
905 	[PWRAP_HARB_INIT] =		0x50,
906 	[PWRAP_HARB_HPRIO] =		0x54,
907 	[PWRAP_HIPRIO_ARB_EN] =		0x58,
908 	[PWRAP_HARB_STA0] =		0x5c,
909 	[PWRAP_HARB_STA1] =		0x60,
910 	[PWRAP_MAN_EN] =		0x64,
911 	[PWRAP_MAN_CMD] =		0x68,
912 	[PWRAP_MAN_RDATA] =		0x6c,
913 	[PWRAP_MAN_VLDCLR] =		0x70,
914 	[PWRAP_WACS0_EN] =		0x74,
915 	[PWRAP_INIT_DONE0] =		0x78,
916 	[PWRAP_WACS0_CMD] =		0x7c,
917 	[PWRAP_WACS0_RDATA] =		0x80,
918 	[PWRAP_WACS0_VLDCLR] =		0x84,
919 	[PWRAP_WACS1_EN] =		0x88,
920 	[PWRAP_INIT_DONE1] =		0x8c,
921 	[PWRAP_WACS1_CMD] =		0x90,
922 	[PWRAP_WACS1_RDATA] =		0x94,
923 	[PWRAP_WACS1_VLDCLR] =		0x98,
924 	[PWRAP_WACS2_EN] =		0x9c,
925 	[PWRAP_INIT_DONE2] =		0xa0,
926 	[PWRAP_WACS2_CMD] =		0xa4,
927 	[PWRAP_WACS2_RDATA] =		0xa8,
928 	[PWRAP_WACS2_VLDCLR] =		0xac,
929 	[PWRAP_INT_EN] =		0xb0,
930 	[PWRAP_INT_FLG_RAW] =		0xb4,
931 	[PWRAP_INT_FLG] =		0xb8,
932 	[PWRAP_INT_CLR] =		0xbc,
933 	[PWRAP_SIG_ADR] =		0xc0,
934 	[PWRAP_SIG_MODE] =		0xc4,
935 	[PWRAP_SIG_VALUE] =		0xc8,
936 	[PWRAP_SIG_ERRVAL] =		0xcc,
937 	[PWRAP_CRC_EN] =		0xd0,
938 	[PWRAP_EVENT_STA] =		0xd4,
939 	[PWRAP_EVENT_STACLR] =		0xd8,
940 	[PWRAP_TIMER_EN] =		0xdc,
941 	[PWRAP_TIMER_STA] =		0xe0,
942 	[PWRAP_WDT_UNIT] =		0xe4,
943 	[PWRAP_WDT_SRC_EN] =		0xe8,
944 	[PWRAP_WDT_FLG] =		0xec,
945 	[PWRAP_DEBUG_INT_SEL] =		0xf0,
946 	[PWRAP_CIPHER_KEY_SEL] =	0x134,
947 	[PWRAP_CIPHER_IV_SEL] =		0x138,
948 	[PWRAP_CIPHER_LOAD] =		0x13c,
949 	[PWRAP_CIPHER_START] =		0x140,
950 	[PWRAP_CIPHER_RDY] =		0x144,
951 	[PWRAP_CIPHER_MODE] =		0x148,
952 	[PWRAP_CIPHER_SWRST] =		0x14c,
953 	[PWRAP_DCM_EN] =		0x15c,
954 	[PWRAP_DCM_DBC_PRD] =		0x160,
955 };
956 
957 static int mt8173_regs[] = {
958 	[PWRAP_MUX_SEL] =		0x0,
959 	[PWRAP_WRAP_EN] =		0x4,
960 	[PWRAP_DIO_EN] =		0x8,
961 	[PWRAP_SIDLY] =			0xc,
962 	[PWRAP_RDDMY] =			0x10,
963 	[PWRAP_SI_CK_CON] =		0x14,
964 	[PWRAP_CSHEXT_WRITE] =		0x18,
965 	[PWRAP_CSHEXT_READ] =		0x1c,
966 	[PWRAP_CSLEXT_START] =		0x20,
967 	[PWRAP_CSLEXT_END] =		0x24,
968 	[PWRAP_STAUPD_PRD] =		0x28,
969 	[PWRAP_STAUPD_GRPEN] =		0x2c,
970 	[PWRAP_STAUPD_MAN_TRIG] =	0x40,
971 	[PWRAP_STAUPD_STA] =		0x44,
972 	[PWRAP_WRAP_STA] =		0x48,
973 	[PWRAP_HARB_INIT] =		0x4c,
974 	[PWRAP_HARB_HPRIO] =		0x50,
975 	[PWRAP_HIPRIO_ARB_EN] =		0x54,
976 	[PWRAP_HARB_STA0] =		0x58,
977 	[PWRAP_HARB_STA1] =		0x5c,
978 	[PWRAP_MAN_EN] =		0x60,
979 	[PWRAP_MAN_CMD] =		0x64,
980 	[PWRAP_MAN_RDATA] =		0x68,
981 	[PWRAP_MAN_VLDCLR] =		0x6c,
982 	[PWRAP_WACS0_EN] =		0x70,
983 	[PWRAP_INIT_DONE0] =		0x74,
984 	[PWRAP_WACS0_CMD] =		0x78,
985 	[PWRAP_WACS0_RDATA] =		0x7c,
986 	[PWRAP_WACS0_VLDCLR] =		0x80,
987 	[PWRAP_WACS1_EN] =		0x84,
988 	[PWRAP_INIT_DONE1] =		0x88,
989 	[PWRAP_WACS1_CMD] =		0x8c,
990 	[PWRAP_WACS1_RDATA] =		0x90,
991 	[PWRAP_WACS1_VLDCLR] =		0x94,
992 	[PWRAP_WACS2_EN] =		0x98,
993 	[PWRAP_INIT_DONE2] =		0x9c,
994 	[PWRAP_WACS2_CMD] =		0xa0,
995 	[PWRAP_WACS2_RDATA] =		0xa4,
996 	[PWRAP_WACS2_VLDCLR] =		0xa8,
997 	[PWRAP_INT_EN] =		0xac,
998 	[PWRAP_INT_FLG_RAW] =		0xb0,
999 	[PWRAP_INT_FLG] =		0xb4,
1000 	[PWRAP_INT_CLR] =		0xb8,
1001 	[PWRAP_SIG_ADR] =		0xbc,
1002 	[PWRAP_SIG_MODE] =		0xc0,
1003 	[PWRAP_SIG_VALUE] =		0xc4,
1004 	[PWRAP_SIG_ERRVAL] =		0xc8,
1005 	[PWRAP_CRC_EN] =		0xcc,
1006 	[PWRAP_TIMER_EN] =		0xd0,
1007 	[PWRAP_TIMER_STA] =		0xd4,
1008 	[PWRAP_WDT_UNIT] =		0xd8,
1009 	[PWRAP_WDT_SRC_EN] =		0xdc,
1010 	[PWRAP_WDT_FLG] =		0xe0,
1011 	[PWRAP_DEBUG_INT_SEL] =		0xe4,
1012 	[PWRAP_DVFS_ADR0] =		0xe8,
1013 	[PWRAP_DVFS_WDATA0] =		0xec,
1014 	[PWRAP_DVFS_ADR1] =		0xf0,
1015 	[PWRAP_DVFS_WDATA1] =		0xf4,
1016 	[PWRAP_DVFS_ADR2] =		0xf8,
1017 	[PWRAP_DVFS_WDATA2] =		0xfc,
1018 	[PWRAP_DVFS_ADR3] =		0x100,
1019 	[PWRAP_DVFS_WDATA3] =		0x104,
1020 	[PWRAP_DVFS_ADR4] =		0x108,
1021 	[PWRAP_DVFS_WDATA4] =		0x10c,
1022 	[PWRAP_DVFS_ADR5] =		0x110,
1023 	[PWRAP_DVFS_WDATA5] =		0x114,
1024 	[PWRAP_DVFS_ADR6] =		0x118,
1025 	[PWRAP_DVFS_WDATA6] =		0x11c,
1026 	[PWRAP_DVFS_ADR7] =		0x120,
1027 	[PWRAP_DVFS_WDATA7] =		0x124,
1028 	[PWRAP_SPMINF_STA] =		0x128,
1029 	[PWRAP_CIPHER_KEY_SEL] =	0x12c,
1030 	[PWRAP_CIPHER_IV_SEL] =		0x130,
1031 	[PWRAP_CIPHER_EN] =		0x134,
1032 	[PWRAP_CIPHER_RDY] =		0x138,
1033 	[PWRAP_CIPHER_MODE] =		0x13c,
1034 	[PWRAP_CIPHER_SWRST] =		0x140,
1035 	[PWRAP_DCM_EN] =		0x144,
1036 	[PWRAP_DCM_DBC_PRD] =		0x148,
1037 };
1038 
1039 static int mt8183_regs[] = {
1040 	[PWRAP_MUX_SEL] =			0x0,
1041 	[PWRAP_WRAP_EN] =			0x4,
1042 	[PWRAP_DIO_EN] =			0x8,
1043 	[PWRAP_SI_SAMPLE_CTRL] =		0xC,
1044 	[PWRAP_RDDMY] =				0x14,
1045 	[PWRAP_CSHEXT_WRITE] =			0x18,
1046 	[PWRAP_CSHEXT_READ] =			0x1C,
1047 	[PWRAP_CSLEXT_WRITE] =			0x20,
1048 	[PWRAP_CSLEXT_READ] =			0x24,
1049 	[PWRAP_EXT_CK_WRITE] =			0x28,
1050 	[PWRAP_STAUPD_CTRL] =			0x30,
1051 	[PWRAP_STAUPD_GRPEN] =			0x34,
1052 	[PWRAP_EINT_STA0_ADR] =			0x38,
1053 	[PWRAP_HARB_HPRIO] =			0x5C,
1054 	[PWRAP_HIPRIO_ARB_EN] =			0x60,
1055 	[PWRAP_MAN_EN] =			0x70,
1056 	[PWRAP_MAN_CMD] =			0x74,
1057 	[PWRAP_WACS0_EN] =			0x80,
1058 	[PWRAP_INIT_DONE0] =			0x84,
1059 	[PWRAP_WACS1_EN] =			0x88,
1060 	[PWRAP_INIT_DONE1] =			0x8C,
1061 	[PWRAP_WACS2_EN] =			0x90,
1062 	[PWRAP_INIT_DONE2] =			0x94,
1063 	[PWRAP_WACS_P2P_EN] =			0xA0,
1064 	[PWRAP_INIT_DONE_P2P] =			0xA4,
1065 	[PWRAP_WACS_MD32_EN] =			0xA8,
1066 	[PWRAP_INIT_DONE_MD32] =		0xAC,
1067 	[PWRAP_INT_EN] =			0xB0,
1068 	[PWRAP_INT_FLG] =			0xB8,
1069 	[PWRAP_INT_CLR] =			0xBC,
1070 	[PWRAP_INT1_EN] =			0xC0,
1071 	[PWRAP_INT1_FLG] =			0xC8,
1072 	[PWRAP_INT1_CLR] =			0xCC,
1073 	[PWRAP_SIG_ADR] =			0xD0,
1074 	[PWRAP_CRC_EN] =			0xE0,
1075 	[PWRAP_TIMER_EN] =			0xE4,
1076 	[PWRAP_WDT_UNIT] =			0xEC,
1077 	[PWRAP_WDT_SRC_EN] =			0xF0,
1078 	[PWRAP_WDT_SRC_EN_1] =			0xF4,
1079 	[PWRAP_INT_GPS_AUXADC_CMD_ADDR] =	0x1DC,
1080 	[PWRAP_INT_GPS_AUXADC_CMD] =		0x1E0,
1081 	[PWRAP_INT_GPS_AUXADC_RDATA_ADDR] =	0x1E4,
1082 	[PWRAP_EXT_GPS_AUXADC_RDATA_ADDR] =	0x1E8,
1083 	[PWRAP_GPSINF_0_STA] =			0x1EC,
1084 	[PWRAP_GPSINF_1_STA] =			0x1F0,
1085 	[PWRAP_WACS2_CMD] =			0xC20,
1086 	[PWRAP_WACS2_RDATA] =			0xC24,
1087 	[PWRAP_WACS2_VLDCLR] =			0xC28,
1088 };
1089 
1090 static int mt8195_regs[] = {
1091 	[PWRAP_INIT_DONE2] =		0x0,
1092 	[PWRAP_STAUPD_CTRL] =		0x4C,
1093 	[PWRAP_TIMER_EN] =		0x3E4,
1094 	[PWRAP_INT_EN] =		0x420,
1095 	[PWRAP_INT_FLG] =		0x428,
1096 	[PWRAP_INT_CLR] =		0x42C,
1097 	[PWRAP_INT1_EN] =		0x450,
1098 	[PWRAP_INT1_FLG] =		0x458,
1099 	[PWRAP_INT1_CLR] =		0x45C,
1100 	[PWRAP_WACS2_CMD] =		0x880,
1101 	[PWRAP_SWINF_2_WDATA_31_0] =	0x884,
1102 	[PWRAP_SWINF_2_RDATA_31_0] =	0x894,
1103 	[PWRAP_WACS2_VLDCLR] =		0x8A4,
1104 	[PWRAP_WACS2_RDATA] =		0x8A8,
1105 };
1106 
1107 static int mt8365_regs[] = {
1108 	[PWRAP_MUX_SEL] =		0x0,
1109 	[PWRAP_WRAP_EN] =		0x4,
1110 	[PWRAP_DIO_EN] =		0x8,
1111 	[PWRAP_CSHEXT_WRITE] =		0x24,
1112 	[PWRAP_CSHEXT_READ] =		0x28,
1113 	[PWRAP_STAUPD_PRD] =		0x3c,
1114 	[PWRAP_STAUPD_GRPEN] =		0x40,
1115 	[PWRAP_STAUPD_MAN_TRIG] =	0x58,
1116 	[PWRAP_STAUPD_STA] =		0x5c,
1117 	[PWRAP_WRAP_STA] =		0x60,
1118 	[PWRAP_HARB_INIT] =		0x64,
1119 	[PWRAP_HARB_HPRIO] =		0x68,
1120 	[PWRAP_HIPRIO_ARB_EN] =		0x6c,
1121 	[PWRAP_HARB_STA0] =		0x70,
1122 	[PWRAP_HARB_STA1] =		0x74,
1123 	[PWRAP_MAN_EN] =		0x7c,
1124 	[PWRAP_MAN_CMD] =		0x80,
1125 	[PWRAP_MAN_RDATA] =		0x84,
1126 	[PWRAP_MAN_VLDCLR] =		0x88,
1127 	[PWRAP_WACS0_EN] =		0x8c,
1128 	[PWRAP_INIT_DONE0] =		0x90,
1129 	[PWRAP_WACS0_CMD] =		0xc00,
1130 	[PWRAP_WACS0_RDATA] =		0xc04,
1131 	[PWRAP_WACS0_VLDCLR] =		0xc08,
1132 	[PWRAP_WACS1_EN] =		0x94,
1133 	[PWRAP_INIT_DONE1] =		0x98,
1134 	[PWRAP_WACS2_EN] =		0x9c,
1135 	[PWRAP_INIT_DONE2] =		0xa0,
1136 	[PWRAP_WACS2_CMD] =		0xc20,
1137 	[PWRAP_WACS2_RDATA] =		0xc24,
1138 	[PWRAP_WACS2_VLDCLR] =		0xc28,
1139 	[PWRAP_INT_EN] =		0xb4,
1140 	[PWRAP_INT_FLG_RAW] =		0xb8,
1141 	[PWRAP_INT_FLG] =		0xbc,
1142 	[PWRAP_INT_CLR] =		0xc0,
1143 	[PWRAP_SIG_ADR] =		0xd4,
1144 	[PWRAP_SIG_MODE] =		0xd8,
1145 	[PWRAP_SIG_VALUE] =		0xdc,
1146 	[PWRAP_SIG_ERRVAL] =		0xe0,
1147 	[PWRAP_CRC_EN] =		0xe4,
1148 	[PWRAP_TIMER_EN] =		0xe8,
1149 	[PWRAP_TIMER_STA] =		0xec,
1150 	[PWRAP_WDT_UNIT] =		0xf0,
1151 	[PWRAP_WDT_SRC_EN] =		0xf4,
1152 	[PWRAP_WDT_FLG] =		0xfc,
1153 	[PWRAP_DEBUG_INT_SEL] =		0x104,
1154 	[PWRAP_CIPHER_KEY_SEL] =	0x1c4,
1155 	[PWRAP_CIPHER_IV_SEL] =		0x1c8,
1156 	[PWRAP_CIPHER_RDY] =		0x1d0,
1157 	[PWRAP_CIPHER_MODE] =		0x1d4,
1158 	[PWRAP_CIPHER_SWRST] =		0x1d8,
1159 	[PWRAP_DCM_EN] =		0x1dc,
1160 	[PWRAP_DCM_DBC_PRD] =		0x1e0,
1161 	[PWRAP_EINT_STA0_ADR] =		0x44,
1162 	[PWRAP_EINT_STA1_ADR] =		0x48,
1163 	[PWRAP_INT1_EN] =		0xc4,
1164 	[PWRAP_INT1_FLG] =		0xcc,
1165 	[PWRAP_INT1_CLR] =		0xd0,
1166 	[PWRAP_WDT_SRC_EN_1] =		0xf8,
1167 };
1168 
1169 static int mt8516_regs[] = {
1170 	[PWRAP_MUX_SEL] =		0x0,
1171 	[PWRAP_WRAP_EN] =		0x4,
1172 	[PWRAP_DIO_EN] =		0x8,
1173 	[PWRAP_SIDLY] =			0xc,
1174 	[PWRAP_RDDMY] =			0x10,
1175 	[PWRAP_SI_CK_CON] =		0x14,
1176 	[PWRAP_CSHEXT_WRITE] =		0x18,
1177 	[PWRAP_CSHEXT_READ] =		0x1c,
1178 	[PWRAP_CSLEXT_START] =		0x20,
1179 	[PWRAP_CSLEXT_END] =		0x24,
1180 	[PWRAP_STAUPD_PRD] =		0x28,
1181 	[PWRAP_STAUPD_GRPEN] =		0x2c,
1182 	[PWRAP_STAUPD_MAN_TRIG] =	0x40,
1183 	[PWRAP_STAUPD_STA] =		0x44,
1184 	[PWRAP_WRAP_STA] =		0x48,
1185 	[PWRAP_HARB_INIT] =		0x4c,
1186 	[PWRAP_HARB_HPRIO] =		0x50,
1187 	[PWRAP_HIPRIO_ARB_EN] =		0x54,
1188 	[PWRAP_HARB_STA0] =		0x58,
1189 	[PWRAP_HARB_STA1] =		0x5c,
1190 	[PWRAP_MAN_EN] =		0x60,
1191 	[PWRAP_MAN_CMD] =		0x64,
1192 	[PWRAP_MAN_RDATA] =		0x68,
1193 	[PWRAP_MAN_VLDCLR] =		0x6c,
1194 	[PWRAP_WACS0_EN] =		0x70,
1195 	[PWRAP_INIT_DONE0] =		0x74,
1196 	[PWRAP_WACS0_CMD] =		0x78,
1197 	[PWRAP_WACS0_RDATA] =		0x7c,
1198 	[PWRAP_WACS0_VLDCLR] =		0x80,
1199 	[PWRAP_WACS1_EN] =		0x84,
1200 	[PWRAP_INIT_DONE1] =		0x88,
1201 	[PWRAP_WACS1_CMD] =		0x8c,
1202 	[PWRAP_WACS1_RDATA] =		0x90,
1203 	[PWRAP_WACS1_VLDCLR] =		0x94,
1204 	[PWRAP_WACS2_EN] =		0x98,
1205 	[PWRAP_INIT_DONE2] =		0x9c,
1206 	[PWRAP_WACS2_CMD] =		0xa0,
1207 	[PWRAP_WACS2_RDATA] =		0xa4,
1208 	[PWRAP_WACS2_VLDCLR] =		0xa8,
1209 	[PWRAP_INT_EN] =		0xac,
1210 	[PWRAP_INT_FLG_RAW] =		0xb0,
1211 	[PWRAP_INT_FLG] =		0xb4,
1212 	[PWRAP_INT_CLR] =		0xb8,
1213 	[PWRAP_SIG_ADR] =		0xbc,
1214 	[PWRAP_SIG_MODE] =		0xc0,
1215 	[PWRAP_SIG_VALUE] =		0xc4,
1216 	[PWRAP_SIG_ERRVAL] =		0xc8,
1217 	[PWRAP_CRC_EN] =		0xcc,
1218 	[PWRAP_TIMER_EN] =		0xd0,
1219 	[PWRAP_TIMER_STA] =		0xd4,
1220 	[PWRAP_WDT_UNIT] =		0xd8,
1221 	[PWRAP_WDT_SRC_EN] =		0xdc,
1222 	[PWRAP_WDT_FLG] =		0xe0,
1223 	[PWRAP_DEBUG_INT_SEL] =		0xe4,
1224 	[PWRAP_DVFS_ADR0] =		0xe8,
1225 	[PWRAP_DVFS_WDATA0] =		0xec,
1226 	[PWRAP_DVFS_ADR1] =		0xf0,
1227 	[PWRAP_DVFS_WDATA1] =		0xf4,
1228 	[PWRAP_DVFS_ADR2] =		0xf8,
1229 	[PWRAP_DVFS_WDATA2] =		0xfc,
1230 	[PWRAP_DVFS_ADR3] =		0x100,
1231 	[PWRAP_DVFS_WDATA3] =		0x104,
1232 	[PWRAP_DVFS_ADR4] =		0x108,
1233 	[PWRAP_DVFS_WDATA4] =		0x10c,
1234 	[PWRAP_DVFS_ADR5] =		0x110,
1235 	[PWRAP_DVFS_WDATA5] =		0x114,
1236 	[PWRAP_DVFS_ADR6] =		0x118,
1237 	[PWRAP_DVFS_WDATA6] =		0x11c,
1238 	[PWRAP_DVFS_ADR7] =		0x120,
1239 	[PWRAP_DVFS_WDATA7] =		0x124,
1240 	[PWRAP_SPMINF_STA] =		0x128,
1241 	[PWRAP_CIPHER_KEY_SEL] =	0x12c,
1242 	[PWRAP_CIPHER_IV_SEL] =		0x130,
1243 	[PWRAP_CIPHER_EN] =		0x134,
1244 	[PWRAP_CIPHER_RDY] =		0x138,
1245 	[PWRAP_CIPHER_MODE] =		0x13c,
1246 	[PWRAP_CIPHER_SWRST] =		0x140,
1247 	[PWRAP_DCM_EN] =		0x144,
1248 	[PWRAP_DCM_DBC_PRD] =		0x148,
1249 	[PWRAP_SW_RST] =		0x168,
1250 	[PWRAP_OP_TYPE] =		0x16c,
1251 	[PWRAP_MSB_FIRST] =		0x170,
1252 };
1253 
1254 static int mt8186_regs[] = {
1255 	[PWRAP_MUX_SEL] =		0x0,
1256 	[PWRAP_WRAP_EN] =		0x4,
1257 	[PWRAP_DIO_EN] =		0x8,
1258 	[PWRAP_RDDMY] =			0x20,
1259 	[PWRAP_CSHEXT_WRITE] =		0x24,
1260 	[PWRAP_CSHEXT_READ] =		0x28,
1261 	[PWRAP_CSLEXT_WRITE] =		0x2C,
1262 	[PWRAP_CSLEXT_READ] =		0x30,
1263 	[PWRAP_EXT_CK_WRITE] =		0x34,
1264 	[PWRAP_STAUPD_CTRL] =		0x3C,
1265 	[PWRAP_STAUPD_GRPEN] =		0x40,
1266 	[PWRAP_EINT_STA0_ADR] =		0x44,
1267 	[PWRAP_EINT_STA1_ADR] =		0x48,
1268 	[PWRAP_INT_CLR] =		0xC8,
1269 	[PWRAP_INT_FLG] =		0xC4,
1270 	[PWRAP_MAN_EN] =		0x7C,
1271 	[PWRAP_MAN_CMD] =		0x80,
1272 	[PWRAP_WACS0_EN] =		0x8C,
1273 	[PWRAP_WACS1_EN] =		0x94,
1274 	[PWRAP_WACS2_EN] =		0x9C,
1275 	[PWRAP_INIT_DONE0] =		0x90,
1276 	[PWRAP_INIT_DONE1] =		0x98,
1277 	[PWRAP_INIT_DONE2] =		0xA0,
1278 	[PWRAP_INT_EN] =		0xBC,
1279 	[PWRAP_INT1_EN] =		0xCC,
1280 	[PWRAP_INT1_FLG] =		0xD4,
1281 	[PWRAP_INT1_CLR] =		0xD8,
1282 	[PWRAP_TIMER_EN] =		0xF0,
1283 	[PWRAP_WDT_UNIT] =		0xF8,
1284 	[PWRAP_WDT_SRC_EN] =		0xFC,
1285 	[PWRAP_WDT_SRC_EN_1] =		0x100,
1286 	[PWRAP_WDT_FLG] =		0x104,
1287 	[PWRAP_SPMINF_STA] =		0x1B4,
1288 	[PWRAP_DCM_EN] =		0x1EC,
1289 	[PWRAP_DCM_DBC_PRD] =		0x1F0,
1290 	[PWRAP_GPSINF_0_STA] =		0x204,
1291 	[PWRAP_GPSINF_1_STA] =		0x208,
1292 	[PWRAP_WACS0_CMD] =		0xC00,
1293 	[PWRAP_WACS0_RDATA] =		0xC04,
1294 	[PWRAP_WACS0_VLDCLR] =		0xC08,
1295 	[PWRAP_WACS1_CMD] =		0xC10,
1296 	[PWRAP_WACS1_RDATA] =		0xC14,
1297 	[PWRAP_WACS1_VLDCLR] =		0xC18,
1298 	[PWRAP_WACS2_CMD] =		0xC20,
1299 	[PWRAP_WACS2_RDATA] =		0xC24,
1300 	[PWRAP_WACS2_VLDCLR] =		0xC28,
1301 };
1302 
1303 enum pmic_type {
1304 	PMIC_MT6323,
1305 	PMIC_MT6331,
1306 	PMIC_MT6332,
1307 	PMIC_MT6351,
1308 	PMIC_MT6357,
1309 	PMIC_MT6358,
1310 	PMIC_MT6359,
1311 	PMIC_MT6380,
1312 	PMIC_MT6397,
1313 };
1314 
1315 enum pwrap_type {
1316 	PWRAP_MT2701,
1317 	PWRAP_MT6765,
1318 	PWRAP_MT6779,
1319 	PWRAP_MT6795,
1320 	PWRAP_MT6797,
1321 	PWRAP_MT6873,
1322 	PWRAP_MT7622,
1323 	PWRAP_MT8135,
1324 	PWRAP_MT8173,
1325 	PWRAP_MT8183,
1326 	PWRAP_MT8186,
1327 	PWRAP_MT8195,
1328 	PWRAP_MT8365,
1329 	PWRAP_MT8516,
1330 };
1331 
1332 struct pmic_wrapper;
1333 
1334 struct pwrap_slv_regops {
1335 	const struct regmap_config *regmap;
1336 	/*
1337 	 * pwrap operations are highly associated with the PMIC types,
1338 	 * so the pointers added increases flexibility allowing determination
1339 	 * which type is used by the detection through device tree.
1340 	 */
1341 	int (*pwrap_read)(struct pmic_wrapper *wrp, u32 adr, u32 *rdata);
1342 	int (*pwrap_write)(struct pmic_wrapper *wrp, u32 adr, u32 wdata);
1343 };
1344 
1345 /**
1346  * struct pwrap_slv_type - PMIC device wrapper definitions
1347  * @dew_regs:      Device Wrapper (DeW) register offsets
1348  * @type:          PMIC Type (model)
1349  * @comp_dew_regs: Device Wrapper (DeW) register offsets for companion device
1350  * @comp_type:     Companion PMIC Type (model)
1351  * @regops:        Register R/W ops
1352  * @caps:          Capability flags for the target device
1353  */
1354 struct pwrap_slv_type {
1355 	const u32 *dew_regs;
1356 	enum pmic_type type;
1357 	const u32 *comp_dew_regs;
1358 	enum pmic_type comp_type;
1359 	const struct pwrap_slv_regops *regops;
1360 	u32 caps;
1361 };
1362 
1363 struct pmic_wrapper {
1364 	struct device *dev;
1365 	void __iomem *base;
1366 	struct regmap *regmap;
1367 	const struct pmic_wrapper_type *master;
1368 	const struct pwrap_slv_type *slave;
1369 	struct clk *clk_spi;
1370 	struct clk *clk_wrap;
1371 	struct clk *clk_sys;
1372 	struct clk *clk_tmr;
1373 	struct reset_control *rstc;
1374 
1375 	struct reset_control *rstc_bridge;
1376 	void __iomem *bridge_base;
1377 };
1378 
1379 struct pmic_wrapper_type {
1380 	int *regs;
1381 	enum pwrap_type type;
1382 	u32 arb_en_all;
1383 	u32 int_en_all;
1384 	u32 int1_en_all;
1385 	u32 spi_w;
1386 	u32 wdt_src;
1387 	/* Flags indicating the capability for the target pwrap */
1388 	u32 caps;
1389 	int (*init_reg_clock)(struct pmic_wrapper *wrp);
1390 	int (*init_soc_specific)(struct pmic_wrapper *wrp);
1391 };
1392 
1393 static u32 pwrap_readl(struct pmic_wrapper *wrp, enum pwrap_regs reg)
1394 {
1395 	return readl(wrp->base + wrp->master->regs[reg]);
1396 }
1397 
1398 static void pwrap_writel(struct pmic_wrapper *wrp, u32 val, enum pwrap_regs reg)
1399 {
1400 	writel(val, wrp->base + wrp->master->regs[reg]);
1401 }
1402 
1403 static u32 pwrap_get_fsm_state(struct pmic_wrapper *wrp)
1404 {
1405 	u32 val;
1406 
1407 	val = pwrap_readl(wrp, PWRAP_WACS2_RDATA);
1408 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_ARB))
1409 		return PWRAP_GET_WACS_ARB_FSM(val);
1410 	else
1411 		return PWRAP_GET_WACS_FSM(val);
1412 }
1413 
1414 static bool pwrap_is_fsm_idle(struct pmic_wrapper *wrp)
1415 {
1416 	return pwrap_get_fsm_state(wrp) == PWRAP_WACS_FSM_IDLE;
1417 }
1418 
1419 static bool pwrap_is_fsm_vldclr(struct pmic_wrapper *wrp)
1420 {
1421 	return pwrap_get_fsm_state(wrp) == PWRAP_WACS_FSM_WFVLDCLR;
1422 }
1423 
1424 /*
1425  * Timeout issue sometimes caused by the last read command
1426  * failed because pmic wrap could not got the FSM_VLDCLR
1427  * in time after finishing WACS2_CMD. It made state machine
1428  * still on FSM_VLDCLR and timeout next time.
1429  * Check the status of FSM and clear the vldclr to recovery the
1430  * error.
1431  */
1432 static inline void pwrap_leave_fsm_vldclr(struct pmic_wrapper *wrp)
1433 {
1434 	if (pwrap_is_fsm_vldclr(wrp))
1435 		pwrap_writel(wrp, 1, PWRAP_WACS2_VLDCLR);
1436 }
1437 
1438 static bool pwrap_is_sync_idle(struct pmic_wrapper *wrp)
1439 {
1440 	return pwrap_readl(wrp, PWRAP_WACS2_RDATA) & PWRAP_STATE_SYNC_IDLE0;
1441 }
1442 
1443 static bool pwrap_is_fsm_idle_and_sync_idle(struct pmic_wrapper *wrp)
1444 {
1445 	u32 val = pwrap_readl(wrp, PWRAP_WACS2_RDATA);
1446 
1447 	return (PWRAP_GET_WACS_FSM(val) == PWRAP_WACS_FSM_IDLE) &&
1448 		(val & PWRAP_STATE_SYNC_IDLE0);
1449 }
1450 
1451 static int pwrap_read16(struct pmic_wrapper *wrp, u32 adr, u32 *rdata)
1452 {
1453 	bool tmp;
1454 	int ret;
1455 	u32 val;
1456 
1457 	ret = readx_poll_timeout(pwrap_is_fsm_idle, wrp, tmp, tmp,
1458 				 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1459 	if (ret) {
1460 		pwrap_leave_fsm_vldclr(wrp);
1461 		return ret;
1462 	}
1463 
1464 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_ARB))
1465 		val = adr;
1466 	else
1467 		val = (adr >> 1) << 16;
1468 	pwrap_writel(wrp, val, PWRAP_WACS2_CMD);
1469 
1470 	ret = readx_poll_timeout(pwrap_is_fsm_vldclr, wrp, tmp, tmp,
1471 				 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1472 	if (ret)
1473 		return ret;
1474 
1475 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_ARB))
1476 		val = pwrap_readl(wrp, PWRAP_SWINF_2_RDATA_31_0);
1477 	else
1478 		val = pwrap_readl(wrp, PWRAP_WACS2_RDATA);
1479 	*rdata = PWRAP_GET_WACS_RDATA(val);
1480 
1481 	pwrap_writel(wrp, 1, PWRAP_WACS2_VLDCLR);
1482 
1483 	return 0;
1484 }
1485 
1486 static int pwrap_read32(struct pmic_wrapper *wrp, u32 adr, u32 *rdata)
1487 {
1488 	bool tmp;
1489 	int ret, msb;
1490 
1491 	*rdata = 0;
1492 	for (msb = 0; msb < 2; msb++) {
1493 		ret = readx_poll_timeout(pwrap_is_fsm_idle, wrp, tmp, tmp,
1494 					 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1495 
1496 		if (ret) {
1497 			pwrap_leave_fsm_vldclr(wrp);
1498 			return ret;
1499 		}
1500 
1501 		pwrap_writel(wrp, ((msb << 30) | (adr << 16)),
1502 			     PWRAP_WACS2_CMD);
1503 
1504 		ret = readx_poll_timeout(pwrap_is_fsm_vldclr, wrp, tmp, tmp,
1505 					 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1506 		if (ret)
1507 			return ret;
1508 
1509 		*rdata += (PWRAP_GET_WACS_RDATA(pwrap_readl(wrp,
1510 			   PWRAP_WACS2_RDATA)) << (16 * msb));
1511 
1512 		pwrap_writel(wrp, 1, PWRAP_WACS2_VLDCLR);
1513 	}
1514 
1515 	return 0;
1516 }
1517 
1518 static int pwrap_read(struct pmic_wrapper *wrp, u32 adr, u32 *rdata)
1519 {
1520 	return wrp->slave->regops->pwrap_read(wrp, adr, rdata);
1521 }
1522 
1523 static int pwrap_write16(struct pmic_wrapper *wrp, u32 adr, u32 wdata)
1524 {
1525 	bool tmp;
1526 	int ret;
1527 
1528 	ret = readx_poll_timeout(pwrap_is_fsm_idle, wrp, tmp, tmp,
1529 				 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1530 	if (ret) {
1531 		pwrap_leave_fsm_vldclr(wrp);
1532 		return ret;
1533 	}
1534 
1535 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_ARB)) {
1536 		pwrap_writel(wrp, wdata, PWRAP_SWINF_2_WDATA_31_0);
1537 		pwrap_writel(wrp, BIT(29) | adr, PWRAP_WACS2_CMD);
1538 	} else {
1539 		pwrap_writel(wrp, BIT(31) | ((adr >> 1) << 16) | wdata,
1540 			     PWRAP_WACS2_CMD);
1541 	}
1542 
1543 	return 0;
1544 }
1545 
1546 static int pwrap_write32(struct pmic_wrapper *wrp, u32 adr, u32 wdata)
1547 {
1548 	bool tmp;
1549 	int ret, msb, rdata;
1550 
1551 	for (msb = 0; msb < 2; msb++) {
1552 		ret = readx_poll_timeout(pwrap_is_fsm_idle, wrp, tmp, tmp,
1553 					 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1554 		if (ret) {
1555 			pwrap_leave_fsm_vldclr(wrp);
1556 			return ret;
1557 		}
1558 
1559 		pwrap_writel(wrp, (1 << 31) | (msb << 30) | (adr << 16) |
1560 			     ((wdata >> (msb * 16)) & 0xffff),
1561 			     PWRAP_WACS2_CMD);
1562 
1563 		/*
1564 		 * The pwrap_read operation is the requirement of hardware used
1565 		 * for the synchronization between two successive 16-bit
1566 		 * pwrap_writel operations composing one 32-bit bus writing.
1567 		 * Otherwise, we'll find the result fails on the lower 16-bit
1568 		 * pwrap writing.
1569 		 */
1570 		if (!msb)
1571 			pwrap_read(wrp, adr, &rdata);
1572 	}
1573 
1574 	return 0;
1575 }
1576 
1577 static int pwrap_write(struct pmic_wrapper *wrp, u32 adr, u32 wdata)
1578 {
1579 	return wrp->slave->regops->pwrap_write(wrp, adr, wdata);
1580 }
1581 
1582 static int pwrap_regmap_read(void *context, u32 adr, u32 *rdata)
1583 {
1584 	return pwrap_read(context, adr, rdata);
1585 }
1586 
1587 static int pwrap_regmap_write(void *context, u32 adr, u32 wdata)
1588 {
1589 	return pwrap_write(context, adr, wdata);
1590 }
1591 
1592 static bool pwrap_pmic_read_test(struct pmic_wrapper *wrp, const u32 *dew_regs,
1593 				 u16 read_test_val)
1594 {
1595 	bool is_success;
1596 	u32 rdata;
1597 
1598 	pwrap_read(wrp, dew_regs[PWRAP_DEW_READ_TEST], &rdata);
1599 	is_success = ((rdata & U16_MAX) == read_test_val);
1600 
1601 	return is_success;
1602 }
1603 
1604 static int pwrap_reset_spislave(struct pmic_wrapper *wrp)
1605 {
1606 	bool tmp;
1607 	int ret, i;
1608 
1609 	pwrap_writel(wrp, 0, PWRAP_HIPRIO_ARB_EN);
1610 	pwrap_writel(wrp, 0, PWRAP_WRAP_EN);
1611 	pwrap_writel(wrp, 1, PWRAP_MUX_SEL);
1612 	pwrap_writel(wrp, 1, PWRAP_MAN_EN);
1613 	pwrap_writel(wrp, 0, PWRAP_DIO_EN);
1614 
1615 	pwrap_writel(wrp, wrp->master->spi_w | PWRAP_MAN_CMD_OP_CSL,
1616 			PWRAP_MAN_CMD);
1617 	pwrap_writel(wrp, wrp->master->spi_w | PWRAP_MAN_CMD_OP_OUTS,
1618 			PWRAP_MAN_CMD);
1619 	pwrap_writel(wrp, wrp->master->spi_w | PWRAP_MAN_CMD_OP_CSH,
1620 			PWRAP_MAN_CMD);
1621 
1622 	for (i = 0; i < 4; i++)
1623 		pwrap_writel(wrp, wrp->master->spi_w | PWRAP_MAN_CMD_OP_OUTS,
1624 				PWRAP_MAN_CMD);
1625 
1626 	ret = readx_poll_timeout(pwrap_is_sync_idle, wrp, tmp, tmp,
1627 				 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1628 	if (ret) {
1629 		dev_err(wrp->dev, "%s fail, ret=%d\n", __func__, ret);
1630 		return ret;
1631 	}
1632 
1633 	pwrap_writel(wrp, 0, PWRAP_MAN_EN);
1634 	pwrap_writel(wrp, 0, PWRAP_MUX_SEL);
1635 
1636 	return 0;
1637 }
1638 
1639 /*
1640  * pwrap_init_sidly - configure serial input delay
1641  *
1642  * This configures the serial input delay. We can configure 0, 2, 4 or 6ns
1643  * delay. Do a read test with all possible values and chose the best delay.
1644  */
1645 static int pwrap_init_sidly(struct pmic_wrapper *wrp)
1646 {
1647 	u32 i;
1648 	u32 pass = 0;
1649 	bool read_ok;
1650 	signed char dly[16] = {
1651 		-1, 0, 1, 0, 2, -1, 1, 1, 3, -1, -1, -1, 3, -1, 2, 1
1652 	};
1653 
1654 	for (i = 0; i < 4; i++) {
1655 		pwrap_writel(wrp, i, PWRAP_SIDLY);
1656 		read_ok = pwrap_pmic_read_test(wrp, wrp->slave->dew_regs,
1657 					       PWRAP_DEW_READ_TEST_VAL);
1658 		if (read_ok) {
1659 			dev_dbg(wrp->dev, "[Read Test] pass, SIDLY=%x\n", i);
1660 			pass |= 1 << i;
1661 		}
1662 	}
1663 
1664 	if (dly[pass] < 0) {
1665 		dev_err(wrp->dev, "sidly pass range 0x%x not continuous\n",
1666 				pass);
1667 		return -EIO;
1668 	}
1669 
1670 	pwrap_writel(wrp, dly[pass], PWRAP_SIDLY);
1671 
1672 	return 0;
1673 }
1674 
1675 static int pwrap_init_dual_io(struct pmic_wrapper *wrp)
1676 {
1677 	int ret;
1678 	bool read_ok, tmp;
1679 	bool comp_read_ok = true;
1680 
1681 	/* Enable dual IO mode */
1682 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_DIO_EN], 1);
1683 	if (wrp->slave->comp_dew_regs)
1684 		pwrap_write(wrp, wrp->slave->comp_dew_regs[PWRAP_DEW_DIO_EN], 1);
1685 
1686 	/* Check IDLE & INIT_DONE in advance */
1687 	ret = readx_poll_timeout(pwrap_is_fsm_idle_and_sync_idle, wrp, tmp, tmp,
1688 				 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1689 	if (ret) {
1690 		dev_err(wrp->dev, "%s fail, ret=%d\n", __func__, ret);
1691 		return ret;
1692 	}
1693 
1694 	pwrap_writel(wrp, 1, PWRAP_DIO_EN);
1695 
1696 	/* Read Test */
1697 	read_ok = pwrap_pmic_read_test(wrp, wrp->slave->dew_regs, PWRAP_DEW_READ_TEST_VAL);
1698 	if (wrp->slave->comp_dew_regs)
1699 		comp_read_ok = pwrap_pmic_read_test(wrp, wrp->slave->comp_dew_regs,
1700 						    PWRAP_DEW_COMP_READ_TEST_VAL);
1701 	if (!read_ok || !comp_read_ok) {
1702 		dev_err(wrp->dev, "Read failed on DIO mode. Main PMIC %s%s\n",
1703 			!read_ok ? "fail" : "success",
1704 			wrp->slave->comp_dew_regs && !comp_read_ok ?
1705 			", Companion PMIC fail" : "");
1706 		return -EFAULT;
1707 	}
1708 
1709 	return 0;
1710 }
1711 
1712 /*
1713  * pwrap_init_chip_select_ext is used to configure CS extension time for each
1714  * phase during data transactions on the pwrap bus.
1715  */
1716 static void pwrap_init_chip_select_ext(struct pmic_wrapper *wrp, u8 hext_write,
1717 				       u8 hext_read, u8 lext_start,
1718 				       u8 lext_end)
1719 {
1720 	/*
1721 	 * After finishing a write and read transaction, extends CS high time
1722 	 * to be at least xT of BUS CLK as hext_write and hext_read specifies
1723 	 * respectively.
1724 	 */
1725 	pwrap_writel(wrp, hext_write, PWRAP_CSHEXT_WRITE);
1726 	pwrap_writel(wrp, hext_read, PWRAP_CSHEXT_READ);
1727 
1728 	/*
1729 	 * Extends CS low time after CSL and before CSH command to be at
1730 	 * least xT of BUS CLK as lext_start and lext_end specifies
1731 	 * respectively.
1732 	 */
1733 	pwrap_writel(wrp, lext_start, PWRAP_CSLEXT_START);
1734 	pwrap_writel(wrp, lext_end, PWRAP_CSLEXT_END);
1735 }
1736 
1737 static int pwrap_common_init_reg_clock(struct pmic_wrapper *wrp)
1738 {
1739 	switch (wrp->master->type) {
1740 	case PWRAP_MT6795:
1741 		if (wrp->slave->type == PMIC_MT6331) {
1742 			const u32 *dew_regs = wrp->slave->dew_regs;
1743 
1744 			pwrap_write(wrp, dew_regs[PWRAP_DEW_RDDMY_NO], 0x8);
1745 
1746 			if (wrp->slave->comp_type == PMIC_MT6332) {
1747 				dew_regs = wrp->slave->comp_dew_regs;
1748 				pwrap_write(wrp, dew_regs[PWRAP_DEW_RDDMY_NO], 0x8);
1749 			}
1750 		}
1751 		pwrap_writel(wrp, 0x88, PWRAP_RDDMY);
1752 		pwrap_init_chip_select_ext(wrp, 15, 15, 15, 15);
1753 		break;
1754 	case PWRAP_MT8173:
1755 		pwrap_init_chip_select_ext(wrp, 0, 4, 2, 2);
1756 		break;
1757 	case PWRAP_MT8135:
1758 		pwrap_writel(wrp, 0x4, PWRAP_CSHEXT);
1759 		pwrap_init_chip_select_ext(wrp, 0, 4, 0, 0);
1760 		break;
1761 	default:
1762 		break;
1763 	}
1764 
1765 	return 0;
1766 }
1767 
1768 static int pwrap_mt2701_init_reg_clock(struct pmic_wrapper *wrp)
1769 {
1770 	switch (wrp->slave->type) {
1771 	case PMIC_MT6397:
1772 		pwrap_writel(wrp, 0xc, PWRAP_RDDMY);
1773 		pwrap_init_chip_select_ext(wrp, 4, 0, 2, 2);
1774 		break;
1775 
1776 	case PMIC_MT6323:
1777 		pwrap_writel(wrp, 0x8, PWRAP_RDDMY);
1778 		pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_RDDMY_NO],
1779 			    0x8);
1780 		pwrap_init_chip_select_ext(wrp, 5, 0, 2, 2);
1781 		break;
1782 	default:
1783 		break;
1784 	}
1785 
1786 	return 0;
1787 }
1788 
1789 static bool pwrap_is_cipher_ready(struct pmic_wrapper *wrp)
1790 {
1791 	return pwrap_readl(wrp, PWRAP_CIPHER_RDY) & 1;
1792 }
1793 
1794 static bool __pwrap_is_pmic_cipher_ready(struct pmic_wrapper *wrp, const u32 *dew_regs)
1795 {
1796 	u32 rdata;
1797 	int ret;
1798 
1799 	ret = pwrap_read(wrp, dew_regs[PWRAP_DEW_CIPHER_RDY], &rdata);
1800 	if (ret)
1801 		return false;
1802 
1803 	return rdata == 1;
1804 }
1805 
1806 
1807 static bool pwrap_is_pmic_cipher_ready(struct pmic_wrapper *wrp)
1808 {
1809 	bool ret = __pwrap_is_pmic_cipher_ready(wrp, wrp->slave->dew_regs);
1810 
1811 	if (!ret)
1812 		return ret;
1813 
1814 	/* If there's any companion, wait for it to be ready too */
1815 	if (wrp->slave->comp_dew_regs)
1816 		ret = __pwrap_is_pmic_cipher_ready(wrp, wrp->slave->comp_dew_regs);
1817 
1818 	return ret;
1819 }
1820 
1821 static void pwrap_config_cipher(struct pmic_wrapper *wrp, const u32 *dew_regs)
1822 {
1823 	pwrap_write(wrp, dew_regs[PWRAP_DEW_CIPHER_SWRST], 0x1);
1824 	pwrap_write(wrp, dew_regs[PWRAP_DEW_CIPHER_SWRST], 0x0);
1825 	pwrap_write(wrp, dew_regs[PWRAP_DEW_CIPHER_KEY_SEL], 0x1);
1826 	pwrap_write(wrp, dew_regs[PWRAP_DEW_CIPHER_IV_SEL], 0x2);
1827 }
1828 
1829 static int pwrap_init_cipher(struct pmic_wrapper *wrp)
1830 {
1831 	int ret;
1832 	bool tmp;
1833 	u32 rdata = 0;
1834 
1835 	pwrap_writel(wrp, 0x1, PWRAP_CIPHER_SWRST);
1836 	pwrap_writel(wrp, 0x0, PWRAP_CIPHER_SWRST);
1837 	pwrap_writel(wrp, 0x1, PWRAP_CIPHER_KEY_SEL);
1838 	pwrap_writel(wrp, 0x2, PWRAP_CIPHER_IV_SEL);
1839 
1840 	switch (wrp->master->type) {
1841 	case PWRAP_MT8135:
1842 		pwrap_writel(wrp, 1, PWRAP_CIPHER_LOAD);
1843 		pwrap_writel(wrp, 1, PWRAP_CIPHER_START);
1844 		break;
1845 	case PWRAP_MT2701:
1846 	case PWRAP_MT6765:
1847 	case PWRAP_MT6779:
1848 	case PWRAP_MT6795:
1849 	case PWRAP_MT6797:
1850 	case PWRAP_MT8173:
1851 	case PWRAP_MT8186:
1852 	case PWRAP_MT8365:
1853 	case PWRAP_MT8516:
1854 		pwrap_writel(wrp, 1, PWRAP_CIPHER_EN);
1855 		break;
1856 	case PWRAP_MT7622:
1857 		pwrap_writel(wrp, 0, PWRAP_CIPHER_EN);
1858 		break;
1859 	case PWRAP_MT6873:
1860 	case PWRAP_MT8183:
1861 	case PWRAP_MT8195:
1862 		break;
1863 	}
1864 
1865 	/* Config cipher mode @PMIC */
1866 	pwrap_config_cipher(wrp, wrp->slave->dew_regs);
1867 
1868 	/* If there is any companion PMIC, configure cipher mode there too */
1869 	if (wrp->slave->comp_type > 0)
1870 		pwrap_config_cipher(wrp, wrp->slave->comp_dew_regs);
1871 
1872 	switch (wrp->slave->type) {
1873 	case PMIC_MT6397:
1874 		pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_LOAD],
1875 			    0x1);
1876 		pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_START],
1877 			    0x1);
1878 		break;
1879 	case PMIC_MT6323:
1880 	case PMIC_MT6351:
1881 	case PMIC_MT6357:
1882 		pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_EN],
1883 			    0x1);
1884 		break;
1885 	default:
1886 		break;
1887 	}
1888 
1889 	/* wait for cipher data ready@AP */
1890 	ret = readx_poll_timeout(pwrap_is_cipher_ready, wrp, tmp, tmp,
1891 				 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1892 	if (ret) {
1893 		dev_err(wrp->dev, "cipher data ready@AP fail, ret=%d\n", ret);
1894 		return ret;
1895 	}
1896 
1897 	/* wait for cipher data ready@PMIC */
1898 	ret = readx_poll_timeout(pwrap_is_pmic_cipher_ready, wrp, tmp, tmp,
1899 				 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1900 	if (ret) {
1901 		dev_err(wrp->dev,
1902 			"timeout waiting for cipher data ready@PMIC\n");
1903 		return ret;
1904 	}
1905 
1906 	/* wait for cipher mode idle */
1907 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_MODE], 0x1);
1908 	ret = readx_poll_timeout(pwrap_is_fsm_idle_and_sync_idle, wrp, tmp, tmp,
1909 				 PWRAP_POLL_DELAY_US, PWRAP_POLL_TIMEOUT_US);
1910 	if (ret) {
1911 		dev_err(wrp->dev, "cipher mode idle fail, ret=%d\n", ret);
1912 		return ret;
1913 	}
1914 
1915 	pwrap_writel(wrp, 1, PWRAP_CIPHER_MODE);
1916 
1917 	/* Write Test */
1918 	if (pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_WRITE_TEST],
1919 			PWRAP_DEW_WRITE_TEST_VAL) ||
1920 	    pwrap_read(wrp, wrp->slave->dew_regs[PWRAP_DEW_WRITE_TEST],
1921 		       &rdata) ||
1922 	    (rdata != PWRAP_DEW_WRITE_TEST_VAL)) {
1923 		dev_err(wrp->dev, "rdata=0x%04X\n", rdata);
1924 		return -EFAULT;
1925 	}
1926 
1927 	return 0;
1928 }
1929 
1930 static int pwrap_init_security(struct pmic_wrapper *wrp)
1931 {
1932 	u32 crc_val;
1933 	int ret;
1934 
1935 	/* Enable encryption */
1936 	ret = pwrap_init_cipher(wrp);
1937 	if (ret)
1938 		return ret;
1939 
1940 	/* Signature checking - using CRC */
1941 	ret = pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CRC_EN], 0x1);
1942 	if (ret == 0 && wrp->slave->comp_dew_regs)
1943 		ret = pwrap_write(wrp, wrp->slave->comp_dew_regs[PWRAP_DEW_CRC_EN], 0x1);
1944 
1945 	pwrap_writel(wrp, 0x1, PWRAP_CRC_EN);
1946 	pwrap_writel(wrp, 0x0, PWRAP_SIG_MODE);
1947 
1948 	/* CRC value */
1949 	crc_val = wrp->slave->dew_regs[PWRAP_DEW_CRC_VAL];
1950 	if (wrp->slave->comp_dew_regs)
1951 		crc_val |= wrp->slave->comp_dew_regs[PWRAP_DEW_CRC_VAL] << 16;
1952 
1953 	pwrap_writel(wrp, crc_val, PWRAP_SIG_ADR);
1954 
1955 	/* PMIC Wrapper Arbiter priority */
1956 	pwrap_writel(wrp,
1957 		     wrp->master->arb_en_all, PWRAP_HIPRIO_ARB_EN);
1958 
1959 	return 0;
1960 }
1961 
1962 static int pwrap_mt8135_init_soc_specific(struct pmic_wrapper *wrp)
1963 {
1964 	/* enable pwrap events and pwrap bridge in AP side */
1965 	pwrap_writel(wrp, 0x1, PWRAP_EVENT_IN_EN);
1966 	pwrap_writel(wrp, 0xffff, PWRAP_EVENT_DST_EN);
1967 	writel(0x7f, wrp->bridge_base + PWRAP_MT8135_BRIDGE_IORD_ARB_EN);
1968 	writel(0x1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_WACS3_EN);
1969 	writel(0x1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_WACS4_EN);
1970 	writel(0x1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_WDT_UNIT);
1971 	writel(0xffff, wrp->bridge_base + PWRAP_MT8135_BRIDGE_WDT_SRC_EN);
1972 	writel(0x1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_TIMER_EN);
1973 	writel(0x7ff, wrp->bridge_base + PWRAP_MT8135_BRIDGE_INT_EN);
1974 
1975 	/* enable PMIC event out and sources */
1976 	if (pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_EVENT_OUT_EN],
1977 			0x1) ||
1978 	    pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_EVENT_SRC_EN],
1979 			0xffff)) {
1980 		dev_err(wrp->dev, "enable dewrap fail\n");
1981 		return -EFAULT;
1982 	}
1983 
1984 	return 0;
1985 }
1986 
1987 static int pwrap_mt8173_init_soc_specific(struct pmic_wrapper *wrp)
1988 {
1989 	/* PMIC_DEWRAP enables */
1990 	if (pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_EVENT_OUT_EN],
1991 			0x1) ||
1992 	    pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_EVENT_SRC_EN],
1993 			0xffff)) {
1994 		dev_err(wrp->dev, "enable dewrap fail\n");
1995 		return -EFAULT;
1996 	}
1997 
1998 	return 0;
1999 }
2000 
2001 static int pwrap_mt2701_init_soc_specific(struct pmic_wrapper *wrp)
2002 {
2003 	/* GPS_INTF initialization */
2004 	switch (wrp->slave->type) {
2005 	case PMIC_MT6323:
2006 		pwrap_writel(wrp, 0x076c, PWRAP_ADC_CMD_ADDR);
2007 		pwrap_writel(wrp, 0x8000, PWRAP_PWRAP_ADC_CMD);
2008 		pwrap_writel(wrp, 0x072c, PWRAP_ADC_RDY_ADDR);
2009 		pwrap_writel(wrp, 0x072e, PWRAP_ADC_RDATA_ADDR1);
2010 		pwrap_writel(wrp, 0x0730, PWRAP_ADC_RDATA_ADDR2);
2011 		break;
2012 	default:
2013 		break;
2014 	}
2015 
2016 	return 0;
2017 }
2018 
2019 static int pwrap_mt6795_init_soc_specific(struct pmic_wrapper *wrp)
2020 {
2021 	pwrap_writel(wrp, 0xf, PWRAP_STAUPD_GRPEN);
2022 
2023 	if (wrp->slave->type == PMIC_MT6331)
2024 		pwrap_writel(wrp, 0x1b4, PWRAP_EINT_STA0_ADR);
2025 
2026 	if (wrp->slave->comp_type == PMIC_MT6332)
2027 		pwrap_writel(wrp, 0x8112, PWRAP_EINT_STA1_ADR);
2028 
2029 	return 0;
2030 }
2031 
2032 static int pwrap_mt7622_init_soc_specific(struct pmic_wrapper *wrp)
2033 {
2034 	pwrap_writel(wrp, 0, PWRAP_STAUPD_PRD);
2035 	/* enable 2wire SPI master */
2036 	pwrap_writel(wrp, 0x8000000, PWRAP_SPI2_CTRL);
2037 
2038 	return 0;
2039 }
2040 
2041 static int pwrap_mt8183_init_soc_specific(struct pmic_wrapper *wrp)
2042 {
2043 	pwrap_writel(wrp, 0xf5, PWRAP_STAUPD_GRPEN);
2044 
2045 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CRC_EN], 0x1);
2046 	pwrap_writel(wrp, 1, PWRAP_CRC_EN);
2047 	pwrap_writel(wrp, 0x416, PWRAP_SIG_ADR);
2048 	pwrap_writel(wrp, 0x42e, PWRAP_EINT_STA0_ADR);
2049 
2050 	pwrap_writel(wrp, 1, PWRAP_WACS_P2P_EN);
2051 	pwrap_writel(wrp, 1, PWRAP_WACS_MD32_EN);
2052 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE_P2P);
2053 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE_MD32);
2054 
2055 	return 0;
2056 }
2057 
2058 static int pwrap_init(struct pmic_wrapper *wrp)
2059 {
2060 	int ret;
2061 
2062 	if (wrp->rstc)
2063 		reset_control_reset(wrp->rstc);
2064 	if (wrp->rstc_bridge)
2065 		reset_control_reset(wrp->rstc_bridge);
2066 
2067 	switch (wrp->master->type) {
2068 	case PWRAP_MT6795:
2069 		fallthrough;
2070 	case PWRAP_MT8173:
2071 		/* Enable DCM */
2072 		pwrap_writel(wrp, 3, PWRAP_DCM_EN);
2073 		pwrap_writel(wrp, 0, PWRAP_DCM_DBC_PRD);
2074 		break;
2075 	default:
2076 		break;
2077 	}
2078 
2079 	if (HAS_CAP(wrp->slave->caps, PWRAP_SLV_CAP_SPI)) {
2080 		/* Reset SPI slave */
2081 		ret = pwrap_reset_spislave(wrp);
2082 		if (ret)
2083 			return ret;
2084 	}
2085 
2086 	pwrap_writel(wrp, 1, PWRAP_WRAP_EN);
2087 
2088 	pwrap_writel(wrp, wrp->master->arb_en_all, PWRAP_HIPRIO_ARB_EN);
2089 
2090 	pwrap_writel(wrp, 1, PWRAP_WACS2_EN);
2091 
2092 	ret = wrp->master->init_reg_clock(wrp);
2093 	if (ret)
2094 		return ret;
2095 
2096 	if (HAS_CAP(wrp->slave->caps, PWRAP_SLV_CAP_SPI)) {
2097 		/* Setup serial input delay */
2098 		ret = pwrap_init_sidly(wrp);
2099 		if (ret)
2100 			return ret;
2101 	}
2102 
2103 	if (HAS_CAP(wrp->slave->caps, PWRAP_SLV_CAP_DUALIO)) {
2104 		/* Enable dual I/O mode */
2105 		ret = pwrap_init_dual_io(wrp);
2106 		if (ret)
2107 			return ret;
2108 	}
2109 
2110 	if (HAS_CAP(wrp->slave->caps, PWRAP_SLV_CAP_SECURITY)) {
2111 		/* Enable security on bus */
2112 		ret = pwrap_init_security(wrp);
2113 		if (ret)
2114 			return ret;
2115 	}
2116 
2117 	if (wrp->master->type == PWRAP_MT8135)
2118 		pwrap_writel(wrp, 0x7, PWRAP_RRARB_EN);
2119 
2120 	pwrap_writel(wrp, 0x1, PWRAP_WACS0_EN);
2121 	pwrap_writel(wrp, 0x1, PWRAP_WACS1_EN);
2122 	pwrap_writel(wrp, 0x1, PWRAP_WACS2_EN);
2123 	pwrap_writel(wrp, 0x5, PWRAP_STAUPD_PRD);
2124 	pwrap_writel(wrp, 0xff, PWRAP_STAUPD_GRPEN);
2125 
2126 	if (wrp->master->init_soc_specific) {
2127 		ret = wrp->master->init_soc_specific(wrp);
2128 		if (ret)
2129 			return ret;
2130 	}
2131 
2132 	/* Setup the init done registers */
2133 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE2);
2134 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE0);
2135 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE1);
2136 
2137 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_BRIDGE)) {
2138 		writel(1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_INIT_DONE3);
2139 		writel(1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_INIT_DONE4);
2140 	}
2141 
2142 	return 0;
2143 }
2144 
2145 static irqreturn_t pwrap_interrupt(int irqno, void *dev_id)
2146 {
2147 	u32 rdata;
2148 	struct pmic_wrapper *wrp = dev_id;
2149 
2150 	rdata = pwrap_readl(wrp, PWRAP_INT_FLG);
2151 	dev_err(wrp->dev, "unexpected interrupt int=0x%x\n", rdata);
2152 	pwrap_writel(wrp, 0xffffffff, PWRAP_INT_CLR);
2153 
2154 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_INT1_EN)) {
2155 		rdata = pwrap_readl(wrp, PWRAP_INT1_FLG);
2156 		dev_err(wrp->dev, "unexpected interrupt int1=0x%x\n", rdata);
2157 		pwrap_writel(wrp, 0xffffffff, PWRAP_INT1_CLR);
2158 	}
2159 
2160 	return IRQ_HANDLED;
2161 }
2162 
2163 static const struct regmap_config pwrap_regmap_config16 = {
2164 	.reg_bits = 16,
2165 	.val_bits = 16,
2166 	.reg_stride = 2,
2167 	.reg_read = pwrap_regmap_read,
2168 	.reg_write = pwrap_regmap_write,
2169 	.max_register = 0xffff,
2170 };
2171 
2172 static const struct regmap_config pwrap_regmap_config32 = {
2173 	.reg_bits = 32,
2174 	.val_bits = 32,
2175 	.reg_stride = 4,
2176 	.reg_read = pwrap_regmap_read,
2177 	.reg_write = pwrap_regmap_write,
2178 	.max_register = 0xffff,
2179 };
2180 
2181 static const struct pwrap_slv_regops pwrap_regops16 = {
2182 	.pwrap_read = pwrap_read16,
2183 	.pwrap_write = pwrap_write16,
2184 	.regmap = &pwrap_regmap_config16,
2185 };
2186 
2187 static const struct pwrap_slv_regops pwrap_regops32 = {
2188 	.pwrap_read = pwrap_read32,
2189 	.pwrap_write = pwrap_write32,
2190 	.regmap = &pwrap_regmap_config32,
2191 };
2192 
2193 static const struct pwrap_slv_type pmic_mt6323 = {
2194 	.dew_regs = mt6323_regs,
2195 	.type = PMIC_MT6323,
2196 	.regops = &pwrap_regops16,
2197 	.caps = PWRAP_SLV_CAP_SPI | PWRAP_SLV_CAP_DUALIO |
2198 		PWRAP_SLV_CAP_SECURITY,
2199 };
2200 
2201 static const struct pwrap_slv_type pmic_mt6331 = {
2202 	.dew_regs = mt6331_regs,
2203 	.type = PMIC_MT6331,
2204 	.comp_dew_regs = mt6332_regs,
2205 	.comp_type = PMIC_MT6332,
2206 	.regops = &pwrap_regops16,
2207 	.caps = PWRAP_SLV_CAP_SPI | PWRAP_SLV_CAP_DUALIO |
2208 		PWRAP_SLV_CAP_SECURITY,
2209 };
2210 
2211 static const struct pwrap_slv_type pmic_mt6351 = {
2212 	.dew_regs = mt6351_regs,
2213 	.type = PMIC_MT6351,
2214 	.regops = &pwrap_regops16,
2215 	.caps = 0,
2216 };
2217 
2218 static const struct pwrap_slv_type pmic_mt6357 = {
2219 	.dew_regs = mt6357_regs,
2220 	.type = PMIC_MT6357,
2221 	.regops = &pwrap_regops16,
2222 	.caps = 0,
2223 };
2224 
2225 static const struct pwrap_slv_type pmic_mt6358 = {
2226 	.dew_regs = mt6358_regs,
2227 	.type = PMIC_MT6358,
2228 	.regops = &pwrap_regops16,
2229 	.caps = PWRAP_SLV_CAP_SPI | PWRAP_SLV_CAP_DUALIO,
2230 };
2231 
2232 static const struct pwrap_slv_type pmic_mt6359 = {
2233 	.dew_regs = mt6359_regs,
2234 	.type = PMIC_MT6359,
2235 	.regops = &pwrap_regops16,
2236 	.caps = PWRAP_SLV_CAP_DUALIO,
2237 };
2238 
2239 static const struct pwrap_slv_type pmic_mt6380 = {
2240 	.dew_regs = NULL,
2241 	.type = PMIC_MT6380,
2242 	.regops = &pwrap_regops32,
2243 	.caps = 0,
2244 };
2245 
2246 static const struct pwrap_slv_type pmic_mt6397 = {
2247 	.dew_regs = mt6397_regs,
2248 	.type = PMIC_MT6397,
2249 	.regops = &pwrap_regops16,
2250 	.caps = PWRAP_SLV_CAP_SPI | PWRAP_SLV_CAP_DUALIO |
2251 		PWRAP_SLV_CAP_SECURITY,
2252 };
2253 
2254 static const struct of_device_id of_slave_match_tbl[] = {
2255 	{ .compatible = "mediatek,mt6323", .data = &pmic_mt6323 },
2256 	{ .compatible = "mediatek,mt6331", .data = &pmic_mt6331 },
2257 	{ .compatible = "mediatek,mt6351", .data = &pmic_mt6351 },
2258 	{ .compatible = "mediatek,mt6357", .data = &pmic_mt6357 },
2259 	{ .compatible = "mediatek,mt6358", .data = &pmic_mt6358 },
2260 	{ .compatible = "mediatek,mt6359", .data = &pmic_mt6359 },
2261 
2262 	/* The MT6380 PMIC only implements a regulator, so we bind it
2263 	 * directly instead of using a MFD.
2264 	 */
2265 	{ .compatible = "mediatek,mt6380-regulator", .data = &pmic_mt6380 },
2266 	{ .compatible = "mediatek,mt6397", .data = &pmic_mt6397 },
2267 	{ /* sentinel */ }
2268 };
2269 MODULE_DEVICE_TABLE(of, of_slave_match_tbl);
2270 
2271 static const struct pmic_wrapper_type pwrap_mt2701 = {
2272 	.regs = mt2701_regs,
2273 	.type = PWRAP_MT2701,
2274 	.arb_en_all = 0x3f,
2275 	.int_en_all = ~(u32)(BIT(31) | BIT(2)),
2276 	.int1_en_all = 0,
2277 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE_NEW,
2278 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2279 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
2280 	.init_reg_clock = pwrap_mt2701_init_reg_clock,
2281 	.init_soc_specific = pwrap_mt2701_init_soc_specific,
2282 };
2283 
2284 static const struct pmic_wrapper_type pwrap_mt6765 = {
2285 	.regs = mt6765_regs,
2286 	.type = PWRAP_MT6765,
2287 	.arb_en_all = 0x3fd35,
2288 	.int_en_all = 0xffffffff,
2289 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2290 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2291 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
2292 	.init_reg_clock = pwrap_common_init_reg_clock,
2293 	.init_soc_specific = NULL,
2294 };
2295 
2296 static const struct pmic_wrapper_type pwrap_mt6779 = {
2297 	.regs = mt6779_regs,
2298 	.type = PWRAP_MT6779,
2299 	.arb_en_all = 0xfbb7f,
2300 	.int_en_all = 0xfffffffe,
2301 	.int1_en_all = 0,
2302 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2303 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2304 	.caps = 0,
2305 	.init_reg_clock = pwrap_common_init_reg_clock,
2306 	.init_soc_specific = NULL,
2307 };
2308 
2309 static const struct pmic_wrapper_type pwrap_mt6795 = {
2310 	.regs = mt6795_regs,
2311 	.type = PWRAP_MT6795,
2312 	.arb_en_all = 0x3f,
2313 	.int_en_all = ~(u32)(BIT(31) | BIT(2) | BIT(1)),
2314 	.int1_en_all = 0,
2315 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2316 	.wdt_src = PWRAP_WDT_SRC_MASK_NO_STAUPD,
2317 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
2318 	.init_reg_clock = pwrap_common_init_reg_clock,
2319 	.init_soc_specific = pwrap_mt6795_init_soc_specific,
2320 };
2321 
2322 static const struct pmic_wrapper_type pwrap_mt6797 = {
2323 	.regs = mt6797_regs,
2324 	.type = PWRAP_MT6797,
2325 	.arb_en_all = 0x01fff,
2326 	.int_en_all = 0xffffffc6,
2327 	.int1_en_all = 0,
2328 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2329 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2330 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
2331 	.init_reg_clock = pwrap_common_init_reg_clock,
2332 	.init_soc_specific = NULL,
2333 };
2334 
2335 static const struct pmic_wrapper_type pwrap_mt6873 = {
2336 	.regs = mt6873_regs,
2337 	.type = PWRAP_MT6873,
2338 	.arb_en_all = 0x777f,
2339 	.int_en_all = BIT(4) | BIT(5),
2340 	.int1_en_all = 0,
2341 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2342 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2343 	.caps = PWRAP_CAP_ARB,
2344 	.init_reg_clock = pwrap_common_init_reg_clock,
2345 	.init_soc_specific = NULL,
2346 };
2347 
2348 static const struct pmic_wrapper_type pwrap_mt7622 = {
2349 	.regs = mt7622_regs,
2350 	.type = PWRAP_MT7622,
2351 	.arb_en_all = 0xff,
2352 	.int_en_all = ~(u32)BIT(31),
2353 	.int1_en_all = 0,
2354 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2355 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2356 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
2357 	.init_reg_clock = pwrap_common_init_reg_clock,
2358 	.init_soc_specific = pwrap_mt7622_init_soc_specific,
2359 };
2360 
2361 static const struct pmic_wrapper_type pwrap_mt8135 = {
2362 	.regs = mt8135_regs,
2363 	.type = PWRAP_MT8135,
2364 	.arb_en_all = 0x1ff,
2365 	.int_en_all = ~(u32)(BIT(31) | BIT(1)),
2366 	.int1_en_all = 0,
2367 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2368 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2369 	.caps = PWRAP_CAP_BRIDGE | PWRAP_CAP_RESET | PWRAP_CAP_DCM,
2370 	.init_reg_clock = pwrap_common_init_reg_clock,
2371 	.init_soc_specific = pwrap_mt8135_init_soc_specific,
2372 };
2373 
2374 static const struct pmic_wrapper_type pwrap_mt8173 = {
2375 	.regs = mt8173_regs,
2376 	.type = PWRAP_MT8173,
2377 	.arb_en_all = 0x3f,
2378 	.int_en_all = ~(u32)(BIT(31) | BIT(1)),
2379 	.int1_en_all = 0,
2380 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2381 	.wdt_src = PWRAP_WDT_SRC_MASK_NO_STAUPD,
2382 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
2383 	.init_reg_clock = pwrap_common_init_reg_clock,
2384 	.init_soc_specific = pwrap_mt8173_init_soc_specific,
2385 };
2386 
2387 static const struct pmic_wrapper_type pwrap_mt8183 = {
2388 	.regs = mt8183_regs,
2389 	.type = PWRAP_MT8183,
2390 	.arb_en_all = 0x3fa75,
2391 	.int_en_all = 0xffffffff,
2392 	.int1_en_all = 0xeef7ffff,
2393 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2394 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2395 	.caps = PWRAP_CAP_INT1_EN | PWRAP_CAP_WDT_SRC1,
2396 	.init_reg_clock = pwrap_common_init_reg_clock,
2397 	.init_soc_specific = pwrap_mt8183_init_soc_specific,
2398 };
2399 
2400 static struct pmic_wrapper_type pwrap_mt8195 = {
2401 	.regs = mt8195_regs,
2402 	.type = PWRAP_MT8195,
2403 	.arb_en_all = 0x777f, /* NEED CONFIRM */
2404 	.int_en_all = 0x180000, /* NEED CONFIRM */
2405 	.int1_en_all = 0,
2406 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2407 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2408 	.caps = PWRAP_CAP_INT1_EN | PWRAP_CAP_ARB,
2409 	.init_reg_clock = pwrap_common_init_reg_clock,
2410 	.init_soc_specific = NULL,
2411 };
2412 
2413 static const struct pmic_wrapper_type pwrap_mt8365 = {
2414 	.regs = mt8365_regs,
2415 	.type = PWRAP_MT8365,
2416 	.arb_en_all = 0x3ffff,
2417 	.int_en_all = 0x7f1fffff,
2418 	.int1_en_all = 0x0,
2419 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2420 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2421 	.caps = PWRAP_CAP_INT1_EN | PWRAP_CAP_WDT_SRC1,
2422 	.init_reg_clock = pwrap_common_init_reg_clock,
2423 	.init_soc_specific = NULL,
2424 };
2425 
2426 static struct pmic_wrapper_type pwrap_mt8516 = {
2427 	.regs = mt8516_regs,
2428 	.type = PWRAP_MT8516,
2429 	.arb_en_all = 0xff,
2430 	.int_en_all = ~(u32)(BIT(31) | BIT(2)),
2431 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2432 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2433 	.caps = PWRAP_CAP_DCM,
2434 	.init_reg_clock = pwrap_mt2701_init_reg_clock,
2435 	.init_soc_specific = NULL,
2436 };
2437 
2438 static struct pmic_wrapper_type pwrap_mt8186 = {
2439 	.regs = mt8186_regs,
2440 	.type = PWRAP_MT8186,
2441 	.arb_en_all = 0xfb27f,
2442 	.int_en_all = 0xfffffffe, /* disable WatchDog Timeout for bit 1 */
2443 	.int1_en_all =  0x000017ff, /* disable Matching interrupt for bit 13 */
2444 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
2445 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
2446 	.caps = PWRAP_CAP_INT1_EN | PWRAP_CAP_ARB_MT8186,
2447 	.init_reg_clock = pwrap_common_init_reg_clock,
2448 	.init_soc_specific = NULL,
2449 };
2450 
2451 static const struct of_device_id of_pwrap_match_tbl[] = {
2452 	{ .compatible = "mediatek,mt2701-pwrap", .data = &pwrap_mt2701 },
2453 	{ .compatible = "mediatek,mt6765-pwrap", .data = &pwrap_mt6765 },
2454 	{ .compatible = "mediatek,mt6779-pwrap", .data = &pwrap_mt6779 },
2455 	{ .compatible = "mediatek,mt6795-pwrap", .data = &pwrap_mt6795 },
2456 	{ .compatible = "mediatek,mt6797-pwrap", .data = &pwrap_mt6797 },
2457 	{ .compatible = "mediatek,mt6873-pwrap", .data = &pwrap_mt6873 },
2458 	{ .compatible = "mediatek,mt7622-pwrap", .data = &pwrap_mt7622 },
2459 	{ .compatible = "mediatek,mt8135-pwrap", .data = &pwrap_mt8135 },
2460 	{ .compatible = "mediatek,mt8173-pwrap", .data = &pwrap_mt8173 },
2461 	{ .compatible = "mediatek,mt8183-pwrap", .data = &pwrap_mt8183 },
2462 	{ .compatible = "mediatek,mt8186-pwrap", .data = &pwrap_mt8186 },
2463 	{ .compatible = "mediatek,mt8195-pwrap", .data = &pwrap_mt8195 },
2464 	{ .compatible = "mediatek,mt8365-pwrap", .data = &pwrap_mt8365 },
2465 	{ .compatible = "mediatek,mt8516-pwrap", .data = &pwrap_mt8516 },
2466 	{ /* sentinel */ }
2467 };
2468 MODULE_DEVICE_TABLE(of, of_pwrap_match_tbl);
2469 
2470 static int pwrap_probe(struct platform_device *pdev)
2471 {
2472 	int ret, irq;
2473 	u32 mask_done;
2474 	struct pmic_wrapper *wrp;
2475 	struct device_node *np = pdev->dev.of_node;
2476 	const struct of_device_id *of_slave_id = NULL;
2477 
2478 	if (np->child)
2479 		of_slave_id = of_match_node(of_slave_match_tbl, np->child);
2480 
2481 	if (!of_slave_id) {
2482 		dev_dbg(&pdev->dev, "slave pmic should be defined in dts\n");
2483 		return -EINVAL;
2484 	}
2485 
2486 	wrp = devm_kzalloc(&pdev->dev, sizeof(*wrp), GFP_KERNEL);
2487 	if (!wrp)
2488 		return -ENOMEM;
2489 
2490 	platform_set_drvdata(pdev, wrp);
2491 
2492 	wrp->master = of_device_get_match_data(&pdev->dev);
2493 	wrp->slave = of_slave_id->data;
2494 	wrp->dev = &pdev->dev;
2495 
2496 	wrp->base = devm_platform_ioremap_resource_byname(pdev, "pwrap");
2497 	if (IS_ERR(wrp->base))
2498 		return PTR_ERR(wrp->base);
2499 
2500 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_RESET)) {
2501 		wrp->rstc = devm_reset_control_get(wrp->dev, "pwrap");
2502 		if (IS_ERR(wrp->rstc)) {
2503 			ret = PTR_ERR(wrp->rstc);
2504 			dev_dbg(wrp->dev, "cannot get pwrap reset: %d\n", ret);
2505 			return ret;
2506 		}
2507 	}
2508 
2509 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_BRIDGE)) {
2510 		wrp->bridge_base = devm_platform_ioremap_resource_byname(pdev, "pwrap-bridge");
2511 		if (IS_ERR(wrp->bridge_base))
2512 			return PTR_ERR(wrp->bridge_base);
2513 
2514 		wrp->rstc_bridge = devm_reset_control_get(wrp->dev,
2515 							  "pwrap-bridge");
2516 		if (IS_ERR(wrp->rstc_bridge)) {
2517 			ret = PTR_ERR(wrp->rstc_bridge);
2518 			dev_dbg(wrp->dev,
2519 				"cannot get pwrap-bridge reset: %d\n", ret);
2520 			return ret;
2521 		}
2522 	}
2523 
2524 	wrp->clk_spi = devm_clk_get(wrp->dev, "spi");
2525 	if (IS_ERR(wrp->clk_spi)) {
2526 		dev_dbg(wrp->dev, "failed to get clock: %ld\n",
2527 			PTR_ERR(wrp->clk_spi));
2528 		return PTR_ERR(wrp->clk_spi);
2529 	}
2530 
2531 	wrp->clk_wrap = devm_clk_get(wrp->dev, "wrap");
2532 	if (IS_ERR(wrp->clk_wrap)) {
2533 		dev_dbg(wrp->dev, "failed to get clock: %ld\n",
2534 			PTR_ERR(wrp->clk_wrap));
2535 		return PTR_ERR(wrp->clk_wrap);
2536 	}
2537 
2538 	wrp->clk_sys = devm_clk_get_optional(wrp->dev, "sys");
2539 	if (IS_ERR(wrp->clk_sys)) {
2540 		return dev_err_probe(wrp->dev, PTR_ERR(wrp->clk_sys),
2541 				     "failed to get clock: %pe\n",
2542 				     wrp->clk_sys);
2543 	}
2544 
2545 	wrp->clk_tmr = devm_clk_get_optional(wrp->dev, "tmr");
2546 	if (IS_ERR(wrp->clk_tmr)) {
2547 		return dev_err_probe(wrp->dev, PTR_ERR(wrp->clk_tmr),
2548 				     "failed to get clock: %pe\n",
2549 				     wrp->clk_tmr);
2550 	}
2551 
2552 	ret = clk_prepare_enable(wrp->clk_spi);
2553 	if (ret)
2554 		return ret;
2555 
2556 	ret = clk_prepare_enable(wrp->clk_wrap);
2557 	if (ret)
2558 		goto err_out1;
2559 
2560 	ret = clk_prepare_enable(wrp->clk_sys);
2561 	if (ret)
2562 		goto err_out2;
2563 
2564 	ret = clk_prepare_enable(wrp->clk_tmr);
2565 	if (ret)
2566 		goto err_out3;
2567 
2568 	/* Enable internal dynamic clock */
2569 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_DCM)) {
2570 		pwrap_writel(wrp, 1, PWRAP_DCM_EN);
2571 		pwrap_writel(wrp, 0, PWRAP_DCM_DBC_PRD);
2572 	}
2573 
2574 	/*
2575 	 * The PMIC could already be initialized by the bootloader.
2576 	 * Skip initialization here in this case.
2577 	 */
2578 	if (!pwrap_readl(wrp, PWRAP_INIT_DONE2)) {
2579 		ret = pwrap_init(wrp);
2580 		if (ret) {
2581 			dev_dbg(wrp->dev, "init failed with %d\n", ret);
2582 			goto err_out4;
2583 		}
2584 	}
2585 
2586 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_ARB))
2587 		mask_done = PWRAP_STATE_INIT_DONE1;
2588 	else if (HAS_CAP(wrp->master->caps, PWRAP_CAP_ARB_MT8186))
2589 		mask_done = PWRAP_STATE_INIT_DONE0_MT8186;
2590 	else
2591 		mask_done = PWRAP_STATE_INIT_DONE0;
2592 
2593 	if (!(pwrap_readl(wrp, PWRAP_WACS2_RDATA) & mask_done)) {
2594 		dev_dbg(wrp->dev, "initialization isn't finished\n");
2595 		ret = -ENODEV;
2596 		goto err_out4;
2597 	}
2598 
2599 	/* Initialize watchdog, may not be done by the bootloader */
2600 	if (!HAS_CAP(wrp->master->caps, PWRAP_CAP_ARB))
2601 		pwrap_writel(wrp, 0xf, PWRAP_WDT_UNIT);
2602 
2603 	/*
2604 	 * Since STAUPD was not used on mt8173 platform,
2605 	 * so STAUPD of WDT_SRC which should be turned off
2606 	 */
2607 	pwrap_writel(wrp, wrp->master->wdt_src, PWRAP_WDT_SRC_EN);
2608 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_WDT_SRC1))
2609 		pwrap_writel(wrp, wrp->master->wdt_src, PWRAP_WDT_SRC_EN_1);
2610 
2611 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_ARB))
2612 		pwrap_writel(wrp, 0x3, PWRAP_TIMER_EN);
2613 	else
2614 		pwrap_writel(wrp, 0x1, PWRAP_TIMER_EN);
2615 
2616 	pwrap_writel(wrp, wrp->master->int_en_all, PWRAP_INT_EN);
2617 	/*
2618 	 * We add INT1 interrupt to handle starvation and request exception
2619 	 * If we support it, we should enable it here.
2620 	 */
2621 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_INT1_EN))
2622 		pwrap_writel(wrp, wrp->master->int1_en_all, PWRAP_INT1_EN);
2623 
2624 	irq = platform_get_irq(pdev, 0);
2625 	if (irq < 0) {
2626 		ret = irq;
2627 		goto err_out2;
2628 	}
2629 
2630 	ret = devm_request_irq(wrp->dev, irq, pwrap_interrupt,
2631 			       IRQF_TRIGGER_HIGH,
2632 			       "mt-pmic-pwrap", wrp);
2633 	if (ret)
2634 		goto err_out4;
2635 
2636 	wrp->regmap = devm_regmap_init(wrp->dev, NULL, wrp, wrp->slave->regops->regmap);
2637 	if (IS_ERR(wrp->regmap)) {
2638 		ret = PTR_ERR(wrp->regmap);
2639 		goto err_out2;
2640 	}
2641 
2642 	ret = of_platform_populate(np, NULL, NULL, wrp->dev);
2643 	if (ret) {
2644 		dev_dbg(wrp->dev, "failed to create child devices at %pOF\n",
2645 				np);
2646 		goto err_out4;
2647 	}
2648 
2649 	return 0;
2650 
2651 err_out4:
2652 	clk_disable_unprepare(wrp->clk_tmr);
2653 err_out3:
2654 	clk_disable_unprepare(wrp->clk_sys);
2655 err_out2:
2656 	clk_disable_unprepare(wrp->clk_wrap);
2657 err_out1:
2658 	clk_disable_unprepare(wrp->clk_spi);
2659 
2660 	return ret;
2661 }
2662 
2663 static struct platform_driver pwrap_drv = {
2664 	.driver = {
2665 		.name = "mt-pmic-pwrap",
2666 		.of_match_table = of_pwrap_match_tbl,
2667 	},
2668 	.probe = pwrap_probe,
2669 };
2670 
2671 module_platform_driver(pwrap_drv);
2672 
2673 MODULE_AUTHOR("Flora Fu, MediaTek");
2674 MODULE_DESCRIPTION("MediaTek MT8135 PMIC Wrapper Driver");
2675 MODULE_LICENSE("GPL v2");
2676