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_device.h>
12 #include <linux/platform_device.h>
13 #include <linux/regmap.h>
14 #include <linux/reset.h>
15 
16 #define PWRAP_MT8135_BRIDGE_IORD_ARB_EN		0x4
17 #define PWRAP_MT8135_BRIDGE_WACS3_EN		0x10
18 #define PWRAP_MT8135_BRIDGE_INIT_DONE3		0x14
19 #define PWRAP_MT8135_BRIDGE_WACS4_EN		0x24
20 #define PWRAP_MT8135_BRIDGE_INIT_DONE4		0x28
21 #define PWRAP_MT8135_BRIDGE_INT_EN		0x38
22 #define PWRAP_MT8135_BRIDGE_TIMER_EN		0x48
23 #define PWRAP_MT8135_BRIDGE_WDT_UNIT		0x50
24 #define PWRAP_MT8135_BRIDGE_WDT_SRC_EN		0x54
25 
26 /* macro for wrapper status */
27 #define PWRAP_GET_WACS_RDATA(x)		(((x) >> 0) & 0x0000ffff)
28 #define PWRAP_GET_WACS_FSM(x)		(((x) >> 16) & 0x00000007)
29 #define PWRAP_GET_WACS_REQ(x)		(((x) >> 19) & 0x00000001)
30 #define PWRAP_STATE_SYNC_IDLE0		(1 << 20)
31 #define PWRAP_STATE_INIT_DONE0		(1 << 21)
32 
33 /* macro for WACS FSM */
34 #define PWRAP_WACS_FSM_IDLE		0x00
35 #define PWRAP_WACS_FSM_REQ		0x02
36 #define PWRAP_WACS_FSM_WFDLE		0x04
37 #define PWRAP_WACS_FSM_WFVLDCLR		0x06
38 #define PWRAP_WACS_INIT_DONE		0x01
39 #define PWRAP_WACS_WACS_SYNC_IDLE	0x01
40 #define PWRAP_WACS_SYNC_BUSY		0x00
41 
42 /* macro for device wrapper default value */
43 #define PWRAP_DEW_READ_TEST_VAL		0x5aa5
44 #define PWRAP_DEW_WRITE_TEST_VAL	0xa55a
45 
46 /* macro for manual command */
47 #define PWRAP_MAN_CMD_SPI_WRITE_NEW	(1 << 14)
48 #define PWRAP_MAN_CMD_SPI_WRITE		(1 << 13)
49 #define PWRAP_MAN_CMD_OP_CSH		(0x0 << 8)
50 #define PWRAP_MAN_CMD_OP_CSL		(0x1 << 8)
51 #define PWRAP_MAN_CMD_OP_CK		(0x2 << 8)
52 #define PWRAP_MAN_CMD_OP_OUTS		(0x8 << 8)
53 #define PWRAP_MAN_CMD_OP_OUTD		(0x9 << 8)
54 #define PWRAP_MAN_CMD_OP_OUTQ		(0xa << 8)
55 
56 /* macro for Watch Dog Timer Source */
57 #define PWRAP_WDT_SRC_EN_STAUPD_TRIG		(1 << 25)
58 #define PWRAP_WDT_SRC_EN_HARB_STAUPD_DLE	(1 << 20)
59 #define PWRAP_WDT_SRC_EN_HARB_STAUPD_ALE	(1 << 6)
60 #define PWRAP_WDT_SRC_MASK_ALL			0xffffffff
61 #define PWRAP_WDT_SRC_MASK_NO_STAUPD	~(PWRAP_WDT_SRC_EN_STAUPD_TRIG | \
62 					  PWRAP_WDT_SRC_EN_HARB_STAUPD_DLE | \
63 					  PWRAP_WDT_SRC_EN_HARB_STAUPD_ALE)
64 
65 /* Group of bits used for shown slave capability */
66 #define PWRAP_SLV_CAP_SPI	BIT(0)
67 #define PWRAP_SLV_CAP_DUALIO	BIT(1)
68 #define PWRAP_SLV_CAP_SECURITY	BIT(2)
69 #define HAS_CAP(_c, _x)	(((_c) & (_x)) == (_x))
70 
71 /* Group of bits used for shown pwrap capability */
72 #define PWRAP_CAP_BRIDGE	BIT(0)
73 #define PWRAP_CAP_RESET		BIT(1)
74 #define PWRAP_CAP_DCM		BIT(2)
75 #define PWRAP_CAP_INT1_EN	BIT(3)
76 #define PWRAP_CAP_WDT_SRC1	BIT(4)
77 
78 /* defines for slave device wrapper registers */
79 enum dew_regs {
80 	PWRAP_DEW_BASE,
81 	PWRAP_DEW_DIO_EN,
82 	PWRAP_DEW_READ_TEST,
83 	PWRAP_DEW_WRITE_TEST,
84 	PWRAP_DEW_CRC_EN,
85 	PWRAP_DEW_CRC_VAL,
86 	PWRAP_DEW_MON_GRP_SEL,
87 	PWRAP_DEW_CIPHER_KEY_SEL,
88 	PWRAP_DEW_CIPHER_IV_SEL,
89 	PWRAP_DEW_CIPHER_RDY,
90 	PWRAP_DEW_CIPHER_MODE,
91 	PWRAP_DEW_CIPHER_SWRST,
92 
93 	/* MT6323 only regs */
94 	PWRAP_DEW_CIPHER_EN,
95 	PWRAP_DEW_RDDMY_NO,
96 
97 	/* MT6358 only regs */
98 	PWRAP_SMT_CON1,
99 	PWRAP_DRV_CON1,
100 	PWRAP_FILTER_CON0,
101 	PWRAP_GPIO_PULLEN0_CLR,
102 	PWRAP_RG_SPI_CON0,
103 	PWRAP_RG_SPI_RECORD0,
104 	PWRAP_RG_SPI_CON2,
105 	PWRAP_RG_SPI_CON3,
106 	PWRAP_RG_SPI_CON4,
107 	PWRAP_RG_SPI_CON5,
108 	PWRAP_RG_SPI_CON6,
109 	PWRAP_RG_SPI_CON7,
110 	PWRAP_RG_SPI_CON8,
111 	PWRAP_RG_SPI_CON13,
112 	PWRAP_SPISLV_KEY,
113 
114 	/* MT6397 only regs */
115 	PWRAP_DEW_EVENT_OUT_EN,
116 	PWRAP_DEW_EVENT_SRC_EN,
117 	PWRAP_DEW_EVENT_SRC,
118 	PWRAP_DEW_EVENT_FLAG,
119 	PWRAP_DEW_MON_FLAG_SEL,
120 	PWRAP_DEW_EVENT_TEST,
121 	PWRAP_DEW_CIPHER_LOAD,
122 	PWRAP_DEW_CIPHER_START,
123 };
124 
125 static const u32 mt6323_regs[] = {
126 	[PWRAP_DEW_BASE] =		0x0000,
127 	[PWRAP_DEW_DIO_EN] =		0x018a,
128 	[PWRAP_DEW_READ_TEST] =		0x018c,
129 	[PWRAP_DEW_WRITE_TEST] =	0x018e,
130 	[PWRAP_DEW_CRC_EN] =		0x0192,
131 	[PWRAP_DEW_CRC_VAL] =		0x0194,
132 	[PWRAP_DEW_MON_GRP_SEL] =	0x0196,
133 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0x0198,
134 	[PWRAP_DEW_CIPHER_IV_SEL] =	0x019a,
135 	[PWRAP_DEW_CIPHER_EN] =		0x019c,
136 	[PWRAP_DEW_CIPHER_RDY] =	0x019e,
137 	[PWRAP_DEW_CIPHER_MODE] =	0x01a0,
138 	[PWRAP_DEW_CIPHER_SWRST] =	0x01a2,
139 	[PWRAP_DEW_RDDMY_NO] =		0x01a4,
140 };
141 
142 static const u32 mt6351_regs[] = {
143 	[PWRAP_DEW_DIO_EN] =		0x02F2,
144 	[PWRAP_DEW_READ_TEST] =		0x02F4,
145 	[PWRAP_DEW_WRITE_TEST] =	0x02F6,
146 	[PWRAP_DEW_CRC_EN] =		0x02FA,
147 	[PWRAP_DEW_CRC_VAL] =		0x02FC,
148 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0x0300,
149 	[PWRAP_DEW_CIPHER_IV_SEL] =	0x0302,
150 	[PWRAP_DEW_CIPHER_EN] =		0x0304,
151 	[PWRAP_DEW_CIPHER_RDY] =	0x0306,
152 	[PWRAP_DEW_CIPHER_MODE] =	0x0308,
153 	[PWRAP_DEW_CIPHER_SWRST] =	0x030A,
154 	[PWRAP_DEW_RDDMY_NO] =		0x030C,
155 };
156 
157 static const u32 mt6357_regs[] = {
158 	[PWRAP_DEW_DIO_EN] =            0x040A,
159 	[PWRAP_DEW_READ_TEST] =         0x040C,
160 	[PWRAP_DEW_WRITE_TEST] =        0x040E,
161 	[PWRAP_DEW_CRC_EN] =            0x0412,
162 	[PWRAP_DEW_CRC_VAL] =           0x0414,
163 	[PWRAP_DEW_CIPHER_KEY_SEL] =    0x0418,
164 	[PWRAP_DEW_CIPHER_IV_SEL] =     0x041A,
165 	[PWRAP_DEW_CIPHER_EN] =         0x041C,
166 	[PWRAP_DEW_CIPHER_RDY] =        0x041E,
167 	[PWRAP_DEW_CIPHER_MODE] =       0x0420,
168 	[PWRAP_DEW_CIPHER_SWRST] =      0x0422,
169 	[PWRAP_DEW_RDDMY_NO] =          0x0424,
170 };
171 
172 static const u32 mt6358_regs[] = {
173 	[PWRAP_SMT_CON1] =		0x0030,
174 	[PWRAP_DRV_CON1] =		0x0038,
175 	[PWRAP_FILTER_CON0] =		0x0040,
176 	[PWRAP_GPIO_PULLEN0_CLR] =	0x0098,
177 	[PWRAP_RG_SPI_CON0] =		0x0408,
178 	[PWRAP_RG_SPI_RECORD0] =	0x040a,
179 	[PWRAP_DEW_DIO_EN] =		0x040c,
180 	[PWRAP_DEW_READ_TEST]	=	0x040e,
181 	[PWRAP_DEW_WRITE_TEST]	=	0x0410,
182 	[PWRAP_DEW_CRC_EN] =		0x0414,
183 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0x041a,
184 	[PWRAP_DEW_CIPHER_IV_SEL] =	0x041c,
185 	[PWRAP_DEW_CIPHER_EN]	=	0x041e,
186 	[PWRAP_DEW_CIPHER_RDY] =	0x0420,
187 	[PWRAP_DEW_CIPHER_MODE] =	0x0422,
188 	[PWRAP_DEW_CIPHER_SWRST] =	0x0424,
189 	[PWRAP_RG_SPI_CON2] =		0x0432,
190 	[PWRAP_RG_SPI_CON3] =		0x0434,
191 	[PWRAP_RG_SPI_CON4] =		0x0436,
192 	[PWRAP_RG_SPI_CON5] =		0x0438,
193 	[PWRAP_RG_SPI_CON6] =		0x043a,
194 	[PWRAP_RG_SPI_CON7] =		0x043c,
195 	[PWRAP_RG_SPI_CON8] =		0x043e,
196 	[PWRAP_RG_SPI_CON13] =		0x0448,
197 	[PWRAP_SPISLV_KEY] =		0x044a,
198 };
199 
200 static const u32 mt6397_regs[] = {
201 	[PWRAP_DEW_BASE] =		0xbc00,
202 	[PWRAP_DEW_EVENT_OUT_EN] =	0xbc00,
203 	[PWRAP_DEW_DIO_EN] =		0xbc02,
204 	[PWRAP_DEW_EVENT_SRC_EN] =	0xbc04,
205 	[PWRAP_DEW_EVENT_SRC] =		0xbc06,
206 	[PWRAP_DEW_EVENT_FLAG] =	0xbc08,
207 	[PWRAP_DEW_READ_TEST] =		0xbc0a,
208 	[PWRAP_DEW_WRITE_TEST] =	0xbc0c,
209 	[PWRAP_DEW_CRC_EN] =		0xbc0e,
210 	[PWRAP_DEW_CRC_VAL] =		0xbc10,
211 	[PWRAP_DEW_MON_GRP_SEL] =	0xbc12,
212 	[PWRAP_DEW_MON_FLAG_SEL] =	0xbc14,
213 	[PWRAP_DEW_EVENT_TEST] =	0xbc16,
214 	[PWRAP_DEW_CIPHER_KEY_SEL] =	0xbc18,
215 	[PWRAP_DEW_CIPHER_IV_SEL] =	0xbc1a,
216 	[PWRAP_DEW_CIPHER_LOAD] =	0xbc1c,
217 	[PWRAP_DEW_CIPHER_START] =	0xbc1e,
218 	[PWRAP_DEW_CIPHER_RDY] =	0xbc20,
219 	[PWRAP_DEW_CIPHER_MODE] =	0xbc22,
220 	[PWRAP_DEW_CIPHER_SWRST] =	0xbc24,
221 };
222 
223 enum pwrap_regs {
224 	PWRAP_MUX_SEL,
225 	PWRAP_WRAP_EN,
226 	PWRAP_DIO_EN,
227 	PWRAP_SIDLY,
228 	PWRAP_CSHEXT_WRITE,
229 	PWRAP_CSHEXT_READ,
230 	PWRAP_CSLEXT_START,
231 	PWRAP_CSLEXT_END,
232 	PWRAP_STAUPD_PRD,
233 	PWRAP_STAUPD_GRPEN,
234 	PWRAP_STAUPD_MAN_TRIG,
235 	PWRAP_STAUPD_STA,
236 	PWRAP_WRAP_STA,
237 	PWRAP_HARB_INIT,
238 	PWRAP_HARB_HPRIO,
239 	PWRAP_HIPRIO_ARB_EN,
240 	PWRAP_HARB_STA0,
241 	PWRAP_HARB_STA1,
242 	PWRAP_MAN_EN,
243 	PWRAP_MAN_CMD,
244 	PWRAP_MAN_RDATA,
245 	PWRAP_MAN_VLDCLR,
246 	PWRAP_WACS0_EN,
247 	PWRAP_INIT_DONE0,
248 	PWRAP_WACS0_CMD,
249 	PWRAP_WACS0_RDATA,
250 	PWRAP_WACS0_VLDCLR,
251 	PWRAP_WACS1_EN,
252 	PWRAP_INIT_DONE1,
253 	PWRAP_WACS1_CMD,
254 	PWRAP_WACS1_RDATA,
255 	PWRAP_WACS1_VLDCLR,
256 	PWRAP_WACS2_EN,
257 	PWRAP_INIT_DONE2,
258 	PWRAP_WACS2_CMD,
259 	PWRAP_WACS2_RDATA,
260 	PWRAP_WACS2_VLDCLR,
261 	PWRAP_INT_EN,
262 	PWRAP_INT_FLG_RAW,
263 	PWRAP_INT_FLG,
264 	PWRAP_INT_CLR,
265 	PWRAP_SIG_ADR,
266 	PWRAP_SIG_MODE,
267 	PWRAP_SIG_VALUE,
268 	PWRAP_SIG_ERRVAL,
269 	PWRAP_CRC_EN,
270 	PWRAP_TIMER_EN,
271 	PWRAP_TIMER_STA,
272 	PWRAP_WDT_UNIT,
273 	PWRAP_WDT_SRC_EN,
274 	PWRAP_WDT_FLG,
275 	PWRAP_DEBUG_INT_SEL,
276 	PWRAP_CIPHER_KEY_SEL,
277 	PWRAP_CIPHER_IV_SEL,
278 	PWRAP_CIPHER_RDY,
279 	PWRAP_CIPHER_MODE,
280 	PWRAP_CIPHER_SWRST,
281 	PWRAP_DCM_EN,
282 	PWRAP_DCM_DBC_PRD,
283 	PWRAP_EINT_STA0_ADR,
284 	PWRAP_EINT_STA1_ADR,
285 
286 	/* MT2701 only regs */
287 	PWRAP_ADC_CMD_ADDR,
288 	PWRAP_PWRAP_ADC_CMD,
289 	PWRAP_ADC_RDY_ADDR,
290 	PWRAP_ADC_RDATA_ADDR1,
291 	PWRAP_ADC_RDATA_ADDR2,
292 
293 	/* MT7622 only regs */
294 	PWRAP_STA,
295 	PWRAP_CLR,
296 	PWRAP_DVFS_ADR8,
297 	PWRAP_DVFS_WDATA8,
298 	PWRAP_DVFS_ADR9,
299 	PWRAP_DVFS_WDATA9,
300 	PWRAP_DVFS_ADR10,
301 	PWRAP_DVFS_WDATA10,
302 	PWRAP_DVFS_ADR11,
303 	PWRAP_DVFS_WDATA11,
304 	PWRAP_DVFS_ADR12,
305 	PWRAP_DVFS_WDATA12,
306 	PWRAP_DVFS_ADR13,
307 	PWRAP_DVFS_WDATA13,
308 	PWRAP_DVFS_ADR14,
309 	PWRAP_DVFS_WDATA14,
310 	PWRAP_DVFS_ADR15,
311 	PWRAP_DVFS_WDATA15,
312 	PWRAP_EXT_CK,
313 	PWRAP_ADC_RDATA_ADDR,
314 	PWRAP_GPS_STA,
315 	PWRAP_SW_RST,
316 	PWRAP_DVFS_STEP_CTRL0,
317 	PWRAP_DVFS_STEP_CTRL1,
318 	PWRAP_DVFS_STEP_CTRL2,
319 	PWRAP_SPI2_CTRL,
320 
321 	/* MT8135 only regs */
322 	PWRAP_CSHEXT,
323 	PWRAP_EVENT_IN_EN,
324 	PWRAP_EVENT_DST_EN,
325 	PWRAP_RRARB_INIT,
326 	PWRAP_RRARB_EN,
327 	PWRAP_RRARB_STA0,
328 	PWRAP_RRARB_STA1,
329 	PWRAP_EVENT_STA,
330 	PWRAP_EVENT_STACLR,
331 	PWRAP_CIPHER_LOAD,
332 	PWRAP_CIPHER_START,
333 
334 	/* MT8173 only regs */
335 	PWRAP_RDDMY,
336 	PWRAP_SI_CK_CON,
337 	PWRAP_DVFS_ADR0,
338 	PWRAP_DVFS_WDATA0,
339 	PWRAP_DVFS_ADR1,
340 	PWRAP_DVFS_WDATA1,
341 	PWRAP_DVFS_ADR2,
342 	PWRAP_DVFS_WDATA2,
343 	PWRAP_DVFS_ADR3,
344 	PWRAP_DVFS_WDATA3,
345 	PWRAP_DVFS_ADR4,
346 	PWRAP_DVFS_WDATA4,
347 	PWRAP_DVFS_ADR5,
348 	PWRAP_DVFS_WDATA5,
349 	PWRAP_DVFS_ADR6,
350 	PWRAP_DVFS_WDATA6,
351 	PWRAP_DVFS_ADR7,
352 	PWRAP_DVFS_WDATA7,
353 	PWRAP_SPMINF_STA,
354 	PWRAP_CIPHER_EN,
355 
356 	/* MT8183 only regs */
357 	PWRAP_SI_SAMPLE_CTRL,
358 	PWRAP_CSLEXT_WRITE,
359 	PWRAP_CSLEXT_READ,
360 	PWRAP_EXT_CK_WRITE,
361 	PWRAP_STAUPD_CTRL,
362 	PWRAP_WACS_P2P_EN,
363 	PWRAP_INIT_DONE_P2P,
364 	PWRAP_WACS_MD32_EN,
365 	PWRAP_INIT_DONE_MD32,
366 	PWRAP_INT1_EN,
367 	PWRAP_INT1_FLG,
368 	PWRAP_INT1_CLR,
369 	PWRAP_WDT_SRC_EN_1,
370 	PWRAP_INT_GPS_AUXADC_CMD_ADDR,
371 	PWRAP_INT_GPS_AUXADC_CMD,
372 	PWRAP_INT_GPS_AUXADC_RDATA_ADDR,
373 	PWRAP_EXT_GPS_AUXADC_RDATA_ADDR,
374 	PWRAP_GPSINF_0_STA,
375 	PWRAP_GPSINF_1_STA,
376 
377 	/* MT8516 only regs */
378 	PWRAP_OP_TYPE,
379 	PWRAP_MSB_FIRST,
380 };
381 
382 static int mt2701_regs[] = {
383 	[PWRAP_MUX_SEL] =		0x0,
384 	[PWRAP_WRAP_EN] =		0x4,
385 	[PWRAP_DIO_EN] =		0x8,
386 	[PWRAP_SIDLY] =			0xc,
387 	[PWRAP_RDDMY] =			0x18,
388 	[PWRAP_SI_CK_CON] =		0x1c,
389 	[PWRAP_CSHEXT_WRITE] =		0x20,
390 	[PWRAP_CSHEXT_READ] =		0x24,
391 	[PWRAP_CSLEXT_START] =		0x28,
392 	[PWRAP_CSLEXT_END] =		0x2c,
393 	[PWRAP_STAUPD_PRD] =		0x30,
394 	[PWRAP_STAUPD_GRPEN] =		0x34,
395 	[PWRAP_STAUPD_MAN_TRIG] =	0x38,
396 	[PWRAP_STAUPD_STA] =		0x3c,
397 	[PWRAP_WRAP_STA] =		0x44,
398 	[PWRAP_HARB_INIT] =		0x48,
399 	[PWRAP_HARB_HPRIO] =		0x4c,
400 	[PWRAP_HIPRIO_ARB_EN] =		0x50,
401 	[PWRAP_HARB_STA0] =		0x54,
402 	[PWRAP_HARB_STA1] =		0x58,
403 	[PWRAP_MAN_EN] =		0x5c,
404 	[PWRAP_MAN_CMD] =		0x60,
405 	[PWRAP_MAN_RDATA] =		0x64,
406 	[PWRAP_MAN_VLDCLR] =		0x68,
407 	[PWRAP_WACS0_EN] =		0x6c,
408 	[PWRAP_INIT_DONE0] =		0x70,
409 	[PWRAP_WACS0_CMD] =		0x74,
410 	[PWRAP_WACS0_RDATA] =		0x78,
411 	[PWRAP_WACS0_VLDCLR] =		0x7c,
412 	[PWRAP_WACS1_EN] =		0x80,
413 	[PWRAP_INIT_DONE1] =		0x84,
414 	[PWRAP_WACS1_CMD] =		0x88,
415 	[PWRAP_WACS1_RDATA] =		0x8c,
416 	[PWRAP_WACS1_VLDCLR] =		0x90,
417 	[PWRAP_WACS2_EN] =		0x94,
418 	[PWRAP_INIT_DONE2] =		0x98,
419 	[PWRAP_WACS2_CMD] =		0x9c,
420 	[PWRAP_WACS2_RDATA] =		0xa0,
421 	[PWRAP_WACS2_VLDCLR] =		0xa4,
422 	[PWRAP_INT_EN] =		0xa8,
423 	[PWRAP_INT_FLG_RAW] =		0xac,
424 	[PWRAP_INT_FLG] =		0xb0,
425 	[PWRAP_INT_CLR] =		0xb4,
426 	[PWRAP_SIG_ADR] =		0xb8,
427 	[PWRAP_SIG_MODE] =		0xbc,
428 	[PWRAP_SIG_VALUE] =		0xc0,
429 	[PWRAP_SIG_ERRVAL] =		0xc4,
430 	[PWRAP_CRC_EN] =		0xc8,
431 	[PWRAP_TIMER_EN] =		0xcc,
432 	[PWRAP_TIMER_STA] =		0xd0,
433 	[PWRAP_WDT_UNIT] =		0xd4,
434 	[PWRAP_WDT_SRC_EN] =		0xd8,
435 	[PWRAP_WDT_FLG] =		0xdc,
436 	[PWRAP_DEBUG_INT_SEL] =		0xe0,
437 	[PWRAP_DVFS_ADR0] =		0xe4,
438 	[PWRAP_DVFS_WDATA0] =		0xe8,
439 	[PWRAP_DVFS_ADR1] =		0xec,
440 	[PWRAP_DVFS_WDATA1] =		0xf0,
441 	[PWRAP_DVFS_ADR2] =		0xf4,
442 	[PWRAP_DVFS_WDATA2] =		0xf8,
443 	[PWRAP_DVFS_ADR3] =		0xfc,
444 	[PWRAP_DVFS_WDATA3] =		0x100,
445 	[PWRAP_DVFS_ADR4] =		0x104,
446 	[PWRAP_DVFS_WDATA4] =		0x108,
447 	[PWRAP_DVFS_ADR5] =		0x10c,
448 	[PWRAP_DVFS_WDATA5] =		0x110,
449 	[PWRAP_DVFS_ADR6] =		0x114,
450 	[PWRAP_DVFS_WDATA6] =		0x118,
451 	[PWRAP_DVFS_ADR7] =		0x11c,
452 	[PWRAP_DVFS_WDATA7] =		0x120,
453 	[PWRAP_CIPHER_KEY_SEL] =	0x124,
454 	[PWRAP_CIPHER_IV_SEL] =		0x128,
455 	[PWRAP_CIPHER_EN] =		0x12c,
456 	[PWRAP_CIPHER_RDY] =		0x130,
457 	[PWRAP_CIPHER_MODE] =		0x134,
458 	[PWRAP_CIPHER_SWRST] =		0x138,
459 	[PWRAP_DCM_EN] =		0x13c,
460 	[PWRAP_DCM_DBC_PRD] =		0x140,
461 	[PWRAP_ADC_CMD_ADDR] =		0x144,
462 	[PWRAP_PWRAP_ADC_CMD] =		0x148,
463 	[PWRAP_ADC_RDY_ADDR] =		0x14c,
464 	[PWRAP_ADC_RDATA_ADDR1] =	0x150,
465 	[PWRAP_ADC_RDATA_ADDR2] =	0x154,
466 };
467 
468 static int mt6765_regs[] = {
469 	[PWRAP_MUX_SEL] =		0x0,
470 	[PWRAP_WRAP_EN] =		0x4,
471 	[PWRAP_DIO_EN] =		0x8,
472 	[PWRAP_RDDMY] =			0x20,
473 	[PWRAP_CSHEXT_WRITE] =		0x24,
474 	[PWRAP_CSHEXT_READ] =		0x28,
475 	[PWRAP_CSLEXT_START] =		0x2C,
476 	[PWRAP_CSLEXT_END] =		0x30,
477 	[PWRAP_STAUPD_PRD] =		0x3C,
478 	[PWRAP_HARB_HPRIO] =		0x68,
479 	[PWRAP_HIPRIO_ARB_EN] =		0x6C,
480 	[PWRAP_MAN_EN] =		0x7C,
481 	[PWRAP_MAN_CMD] =		0x80,
482 	[PWRAP_WACS0_EN] =		0x8C,
483 	[PWRAP_WACS1_EN] =		0x94,
484 	[PWRAP_WACS2_EN] =		0x9C,
485 	[PWRAP_INIT_DONE2] =		0xA0,
486 	[PWRAP_WACS2_CMD] =		0xC20,
487 	[PWRAP_WACS2_RDATA] =		0xC24,
488 	[PWRAP_WACS2_VLDCLR] =		0xC28,
489 	[PWRAP_INT_EN] =		0xB4,
490 	[PWRAP_INT_FLG_RAW] =		0xB8,
491 	[PWRAP_INT_FLG] =		0xBC,
492 	[PWRAP_INT_CLR] =		0xC0,
493 	[PWRAP_TIMER_EN] =		0xE8,
494 	[PWRAP_WDT_UNIT] =		0xF0,
495 	[PWRAP_WDT_SRC_EN] =		0xF4,
496 	[PWRAP_DCM_EN] =		0x1DC,
497 	[PWRAP_DCM_DBC_PRD] =		0x1E0,
498 };
499 
500 static int mt6797_regs[] = {
501 	[PWRAP_MUX_SEL] =		0x0,
502 	[PWRAP_WRAP_EN] =		0x4,
503 	[PWRAP_DIO_EN] =		0x8,
504 	[PWRAP_SIDLY] =			0xC,
505 	[PWRAP_RDDMY] =			0x10,
506 	[PWRAP_CSHEXT_WRITE] =		0x18,
507 	[PWRAP_CSHEXT_READ] =		0x1C,
508 	[PWRAP_CSLEXT_START] =		0x20,
509 	[PWRAP_CSLEXT_END] =		0x24,
510 	[PWRAP_STAUPD_PRD] =		0x28,
511 	[PWRAP_HARB_HPRIO] =		0x50,
512 	[PWRAP_HIPRIO_ARB_EN] =		0x54,
513 	[PWRAP_MAN_EN] =		0x60,
514 	[PWRAP_MAN_CMD] =		0x64,
515 	[PWRAP_WACS0_EN] =		0x70,
516 	[PWRAP_WACS1_EN] =		0x84,
517 	[PWRAP_WACS2_EN] =		0x98,
518 	[PWRAP_INIT_DONE2] =		0x9C,
519 	[PWRAP_WACS2_CMD] =		0xA0,
520 	[PWRAP_WACS2_RDATA] =		0xA4,
521 	[PWRAP_WACS2_VLDCLR] =		0xA8,
522 	[PWRAP_INT_EN] =		0xC0,
523 	[PWRAP_INT_FLG_RAW] =		0xC4,
524 	[PWRAP_INT_FLG] =		0xC8,
525 	[PWRAP_INT_CLR] =		0xCC,
526 	[PWRAP_TIMER_EN] =		0xF4,
527 	[PWRAP_WDT_UNIT] =		0xFC,
528 	[PWRAP_WDT_SRC_EN] =		0x100,
529 	[PWRAP_DCM_EN] =		0x1CC,
530 	[PWRAP_DCM_DBC_PRD] =		0x1D4,
531 };
532 
533 static int mt7622_regs[] = {
534 	[PWRAP_MUX_SEL] =		0x0,
535 	[PWRAP_WRAP_EN] =		0x4,
536 	[PWRAP_DIO_EN] =		0x8,
537 	[PWRAP_SIDLY] =			0xC,
538 	[PWRAP_RDDMY] =			0x10,
539 	[PWRAP_SI_CK_CON] =		0x14,
540 	[PWRAP_CSHEXT_WRITE] =		0x18,
541 	[PWRAP_CSHEXT_READ] =		0x1C,
542 	[PWRAP_CSLEXT_START] =		0x20,
543 	[PWRAP_CSLEXT_END] =		0x24,
544 	[PWRAP_STAUPD_PRD] =		0x28,
545 	[PWRAP_STAUPD_GRPEN] =		0x2C,
546 	[PWRAP_EINT_STA0_ADR] =		0x30,
547 	[PWRAP_EINT_STA1_ADR] =		0x34,
548 	[PWRAP_STA] =			0x38,
549 	[PWRAP_CLR] =			0x3C,
550 	[PWRAP_STAUPD_MAN_TRIG] =	0x40,
551 	[PWRAP_STAUPD_STA] =		0x44,
552 	[PWRAP_WRAP_STA] =		0x48,
553 	[PWRAP_HARB_INIT] =		0x4C,
554 	[PWRAP_HARB_HPRIO] =		0x50,
555 	[PWRAP_HIPRIO_ARB_EN] =		0x54,
556 	[PWRAP_HARB_STA0] =		0x58,
557 	[PWRAP_HARB_STA1] =		0x5C,
558 	[PWRAP_MAN_EN] =		0x60,
559 	[PWRAP_MAN_CMD] =		0x64,
560 	[PWRAP_MAN_RDATA] =		0x68,
561 	[PWRAP_MAN_VLDCLR] =		0x6C,
562 	[PWRAP_WACS0_EN] =		0x70,
563 	[PWRAP_INIT_DONE0] =		0x74,
564 	[PWRAP_WACS0_CMD] =		0x78,
565 	[PWRAP_WACS0_RDATA] =		0x7C,
566 	[PWRAP_WACS0_VLDCLR] =		0x80,
567 	[PWRAP_WACS1_EN] =		0x84,
568 	[PWRAP_INIT_DONE1] =		0x88,
569 	[PWRAP_WACS1_CMD] =		0x8C,
570 	[PWRAP_WACS1_RDATA] =		0x90,
571 	[PWRAP_WACS1_VLDCLR] =		0x94,
572 	[PWRAP_WACS2_EN] =		0x98,
573 	[PWRAP_INIT_DONE2] =		0x9C,
574 	[PWRAP_WACS2_CMD] =		0xA0,
575 	[PWRAP_WACS2_RDATA] =		0xA4,
576 	[PWRAP_WACS2_VLDCLR] =		0xA8,
577 	[PWRAP_INT_EN] =		0xAC,
578 	[PWRAP_INT_FLG_RAW] =		0xB0,
579 	[PWRAP_INT_FLG] =		0xB4,
580 	[PWRAP_INT_CLR] =		0xB8,
581 	[PWRAP_SIG_ADR] =		0xBC,
582 	[PWRAP_SIG_MODE] =		0xC0,
583 	[PWRAP_SIG_VALUE] =		0xC4,
584 	[PWRAP_SIG_ERRVAL] =		0xC8,
585 	[PWRAP_CRC_EN] =		0xCC,
586 	[PWRAP_TIMER_EN] =		0xD0,
587 	[PWRAP_TIMER_STA] =		0xD4,
588 	[PWRAP_WDT_UNIT] =		0xD8,
589 	[PWRAP_WDT_SRC_EN] =		0xDC,
590 	[PWRAP_WDT_FLG] =		0xE0,
591 	[PWRAP_DEBUG_INT_SEL] =		0xE4,
592 	[PWRAP_DVFS_ADR0] =		0xE8,
593 	[PWRAP_DVFS_WDATA0] =		0xEC,
594 	[PWRAP_DVFS_ADR1] =		0xF0,
595 	[PWRAP_DVFS_WDATA1] =		0xF4,
596 	[PWRAP_DVFS_ADR2] =		0xF8,
597 	[PWRAP_DVFS_WDATA2] =		0xFC,
598 	[PWRAP_DVFS_ADR3] =		0x100,
599 	[PWRAP_DVFS_WDATA3] =		0x104,
600 	[PWRAP_DVFS_ADR4] =		0x108,
601 	[PWRAP_DVFS_WDATA4] =		0x10C,
602 	[PWRAP_DVFS_ADR5] =		0x110,
603 	[PWRAP_DVFS_WDATA5] =		0x114,
604 	[PWRAP_DVFS_ADR6] =		0x118,
605 	[PWRAP_DVFS_WDATA6] =		0x11C,
606 	[PWRAP_DVFS_ADR7] =		0x120,
607 	[PWRAP_DVFS_WDATA7] =		0x124,
608 	[PWRAP_DVFS_ADR8] =		0x128,
609 	[PWRAP_DVFS_WDATA8] =		0x12C,
610 	[PWRAP_DVFS_ADR9] =		0x130,
611 	[PWRAP_DVFS_WDATA9] =		0x134,
612 	[PWRAP_DVFS_ADR10] =		0x138,
613 	[PWRAP_DVFS_WDATA10] =		0x13C,
614 	[PWRAP_DVFS_ADR11] =		0x140,
615 	[PWRAP_DVFS_WDATA11] =		0x144,
616 	[PWRAP_DVFS_ADR12] =		0x148,
617 	[PWRAP_DVFS_WDATA12] =		0x14C,
618 	[PWRAP_DVFS_ADR13] =		0x150,
619 	[PWRAP_DVFS_WDATA13] =		0x154,
620 	[PWRAP_DVFS_ADR14] =		0x158,
621 	[PWRAP_DVFS_WDATA14] =		0x15C,
622 	[PWRAP_DVFS_ADR15] =		0x160,
623 	[PWRAP_DVFS_WDATA15] =		0x164,
624 	[PWRAP_SPMINF_STA] =		0x168,
625 	[PWRAP_CIPHER_KEY_SEL] =	0x16C,
626 	[PWRAP_CIPHER_IV_SEL] =		0x170,
627 	[PWRAP_CIPHER_EN] =		0x174,
628 	[PWRAP_CIPHER_RDY] =		0x178,
629 	[PWRAP_CIPHER_MODE] =		0x17C,
630 	[PWRAP_CIPHER_SWRST] =		0x180,
631 	[PWRAP_DCM_EN] =		0x184,
632 	[PWRAP_DCM_DBC_PRD] =		0x188,
633 	[PWRAP_EXT_CK] =		0x18C,
634 	[PWRAP_ADC_CMD_ADDR] =		0x190,
635 	[PWRAP_PWRAP_ADC_CMD] =		0x194,
636 	[PWRAP_ADC_RDATA_ADDR] =	0x198,
637 	[PWRAP_GPS_STA] =		0x19C,
638 	[PWRAP_SW_RST] =		0x1A0,
639 	[PWRAP_DVFS_STEP_CTRL0] =	0x238,
640 	[PWRAP_DVFS_STEP_CTRL1] =	0x23C,
641 	[PWRAP_DVFS_STEP_CTRL2] =	0x240,
642 	[PWRAP_SPI2_CTRL] =		0x244,
643 };
644 
645 static int mt8135_regs[] = {
646 	[PWRAP_MUX_SEL] =		0x0,
647 	[PWRAP_WRAP_EN] =		0x4,
648 	[PWRAP_DIO_EN] =		0x8,
649 	[PWRAP_SIDLY] =			0xc,
650 	[PWRAP_CSHEXT] =		0x10,
651 	[PWRAP_CSHEXT_WRITE] =		0x14,
652 	[PWRAP_CSHEXT_READ] =		0x18,
653 	[PWRAP_CSLEXT_START] =		0x1c,
654 	[PWRAP_CSLEXT_END] =		0x20,
655 	[PWRAP_STAUPD_PRD] =		0x24,
656 	[PWRAP_STAUPD_GRPEN] =		0x28,
657 	[PWRAP_STAUPD_MAN_TRIG] =	0x2c,
658 	[PWRAP_STAUPD_STA] =		0x30,
659 	[PWRAP_EVENT_IN_EN] =		0x34,
660 	[PWRAP_EVENT_DST_EN] =		0x38,
661 	[PWRAP_WRAP_STA] =		0x3c,
662 	[PWRAP_RRARB_INIT] =		0x40,
663 	[PWRAP_RRARB_EN] =		0x44,
664 	[PWRAP_RRARB_STA0] =		0x48,
665 	[PWRAP_RRARB_STA1] =		0x4c,
666 	[PWRAP_HARB_INIT] =		0x50,
667 	[PWRAP_HARB_HPRIO] =		0x54,
668 	[PWRAP_HIPRIO_ARB_EN] =		0x58,
669 	[PWRAP_HARB_STA0] =		0x5c,
670 	[PWRAP_HARB_STA1] =		0x60,
671 	[PWRAP_MAN_EN] =		0x64,
672 	[PWRAP_MAN_CMD] =		0x68,
673 	[PWRAP_MAN_RDATA] =		0x6c,
674 	[PWRAP_MAN_VLDCLR] =		0x70,
675 	[PWRAP_WACS0_EN] =		0x74,
676 	[PWRAP_INIT_DONE0] =		0x78,
677 	[PWRAP_WACS0_CMD] =		0x7c,
678 	[PWRAP_WACS0_RDATA] =		0x80,
679 	[PWRAP_WACS0_VLDCLR] =		0x84,
680 	[PWRAP_WACS1_EN] =		0x88,
681 	[PWRAP_INIT_DONE1] =		0x8c,
682 	[PWRAP_WACS1_CMD] =		0x90,
683 	[PWRAP_WACS1_RDATA] =		0x94,
684 	[PWRAP_WACS1_VLDCLR] =		0x98,
685 	[PWRAP_WACS2_EN] =		0x9c,
686 	[PWRAP_INIT_DONE2] =		0xa0,
687 	[PWRAP_WACS2_CMD] =		0xa4,
688 	[PWRAP_WACS2_RDATA] =		0xa8,
689 	[PWRAP_WACS2_VLDCLR] =		0xac,
690 	[PWRAP_INT_EN] =		0xb0,
691 	[PWRAP_INT_FLG_RAW] =		0xb4,
692 	[PWRAP_INT_FLG] =		0xb8,
693 	[PWRAP_INT_CLR] =		0xbc,
694 	[PWRAP_SIG_ADR] =		0xc0,
695 	[PWRAP_SIG_MODE] =		0xc4,
696 	[PWRAP_SIG_VALUE] =		0xc8,
697 	[PWRAP_SIG_ERRVAL] =		0xcc,
698 	[PWRAP_CRC_EN] =		0xd0,
699 	[PWRAP_EVENT_STA] =		0xd4,
700 	[PWRAP_EVENT_STACLR] =		0xd8,
701 	[PWRAP_TIMER_EN] =		0xdc,
702 	[PWRAP_TIMER_STA] =		0xe0,
703 	[PWRAP_WDT_UNIT] =		0xe4,
704 	[PWRAP_WDT_SRC_EN] =		0xe8,
705 	[PWRAP_WDT_FLG] =		0xec,
706 	[PWRAP_DEBUG_INT_SEL] =		0xf0,
707 	[PWRAP_CIPHER_KEY_SEL] =	0x134,
708 	[PWRAP_CIPHER_IV_SEL] =		0x138,
709 	[PWRAP_CIPHER_LOAD] =		0x13c,
710 	[PWRAP_CIPHER_START] =		0x140,
711 	[PWRAP_CIPHER_RDY] =		0x144,
712 	[PWRAP_CIPHER_MODE] =		0x148,
713 	[PWRAP_CIPHER_SWRST] =		0x14c,
714 	[PWRAP_DCM_EN] =		0x15c,
715 	[PWRAP_DCM_DBC_PRD] =		0x160,
716 };
717 
718 static int mt8173_regs[] = {
719 	[PWRAP_MUX_SEL] =		0x0,
720 	[PWRAP_WRAP_EN] =		0x4,
721 	[PWRAP_DIO_EN] =		0x8,
722 	[PWRAP_SIDLY] =			0xc,
723 	[PWRAP_RDDMY] =			0x10,
724 	[PWRAP_SI_CK_CON] =		0x14,
725 	[PWRAP_CSHEXT_WRITE] =		0x18,
726 	[PWRAP_CSHEXT_READ] =		0x1c,
727 	[PWRAP_CSLEXT_START] =		0x20,
728 	[PWRAP_CSLEXT_END] =		0x24,
729 	[PWRAP_STAUPD_PRD] =		0x28,
730 	[PWRAP_STAUPD_GRPEN] =		0x2c,
731 	[PWRAP_STAUPD_MAN_TRIG] =	0x40,
732 	[PWRAP_STAUPD_STA] =		0x44,
733 	[PWRAP_WRAP_STA] =		0x48,
734 	[PWRAP_HARB_INIT] =		0x4c,
735 	[PWRAP_HARB_HPRIO] =		0x50,
736 	[PWRAP_HIPRIO_ARB_EN] =		0x54,
737 	[PWRAP_HARB_STA0] =		0x58,
738 	[PWRAP_HARB_STA1] =		0x5c,
739 	[PWRAP_MAN_EN] =		0x60,
740 	[PWRAP_MAN_CMD] =		0x64,
741 	[PWRAP_MAN_RDATA] =		0x68,
742 	[PWRAP_MAN_VLDCLR] =		0x6c,
743 	[PWRAP_WACS0_EN] =		0x70,
744 	[PWRAP_INIT_DONE0] =		0x74,
745 	[PWRAP_WACS0_CMD] =		0x78,
746 	[PWRAP_WACS0_RDATA] =		0x7c,
747 	[PWRAP_WACS0_VLDCLR] =		0x80,
748 	[PWRAP_WACS1_EN] =		0x84,
749 	[PWRAP_INIT_DONE1] =		0x88,
750 	[PWRAP_WACS1_CMD] =		0x8c,
751 	[PWRAP_WACS1_RDATA] =		0x90,
752 	[PWRAP_WACS1_VLDCLR] =		0x94,
753 	[PWRAP_WACS2_EN] =		0x98,
754 	[PWRAP_INIT_DONE2] =		0x9c,
755 	[PWRAP_WACS2_CMD] =		0xa0,
756 	[PWRAP_WACS2_RDATA] =		0xa4,
757 	[PWRAP_WACS2_VLDCLR] =		0xa8,
758 	[PWRAP_INT_EN] =		0xac,
759 	[PWRAP_INT_FLG_RAW] =		0xb0,
760 	[PWRAP_INT_FLG] =		0xb4,
761 	[PWRAP_INT_CLR] =		0xb8,
762 	[PWRAP_SIG_ADR] =		0xbc,
763 	[PWRAP_SIG_MODE] =		0xc0,
764 	[PWRAP_SIG_VALUE] =		0xc4,
765 	[PWRAP_SIG_ERRVAL] =		0xc8,
766 	[PWRAP_CRC_EN] =		0xcc,
767 	[PWRAP_TIMER_EN] =		0xd0,
768 	[PWRAP_TIMER_STA] =		0xd4,
769 	[PWRAP_WDT_UNIT] =		0xd8,
770 	[PWRAP_WDT_SRC_EN] =		0xdc,
771 	[PWRAP_WDT_FLG] =		0xe0,
772 	[PWRAP_DEBUG_INT_SEL] =		0xe4,
773 	[PWRAP_DVFS_ADR0] =		0xe8,
774 	[PWRAP_DVFS_WDATA0] =		0xec,
775 	[PWRAP_DVFS_ADR1] =		0xf0,
776 	[PWRAP_DVFS_WDATA1] =		0xf4,
777 	[PWRAP_DVFS_ADR2] =		0xf8,
778 	[PWRAP_DVFS_WDATA2] =		0xfc,
779 	[PWRAP_DVFS_ADR3] =		0x100,
780 	[PWRAP_DVFS_WDATA3] =		0x104,
781 	[PWRAP_DVFS_ADR4] =		0x108,
782 	[PWRAP_DVFS_WDATA4] =		0x10c,
783 	[PWRAP_DVFS_ADR5] =		0x110,
784 	[PWRAP_DVFS_WDATA5] =		0x114,
785 	[PWRAP_DVFS_ADR6] =		0x118,
786 	[PWRAP_DVFS_WDATA6] =		0x11c,
787 	[PWRAP_DVFS_ADR7] =		0x120,
788 	[PWRAP_DVFS_WDATA7] =		0x124,
789 	[PWRAP_SPMINF_STA] =		0x128,
790 	[PWRAP_CIPHER_KEY_SEL] =	0x12c,
791 	[PWRAP_CIPHER_IV_SEL] =		0x130,
792 	[PWRAP_CIPHER_EN] =		0x134,
793 	[PWRAP_CIPHER_RDY] =		0x138,
794 	[PWRAP_CIPHER_MODE] =		0x13c,
795 	[PWRAP_CIPHER_SWRST] =		0x140,
796 	[PWRAP_DCM_EN] =		0x144,
797 	[PWRAP_DCM_DBC_PRD] =		0x148,
798 };
799 
800 static int mt8183_regs[] = {
801 	[PWRAP_MUX_SEL] =			0x0,
802 	[PWRAP_WRAP_EN] =			0x4,
803 	[PWRAP_DIO_EN] =			0x8,
804 	[PWRAP_SI_SAMPLE_CTRL] =		0xC,
805 	[PWRAP_RDDMY] =				0x14,
806 	[PWRAP_CSHEXT_WRITE] =			0x18,
807 	[PWRAP_CSHEXT_READ] =			0x1C,
808 	[PWRAP_CSLEXT_WRITE] =			0x20,
809 	[PWRAP_CSLEXT_READ] =			0x24,
810 	[PWRAP_EXT_CK_WRITE] =			0x28,
811 	[PWRAP_STAUPD_CTRL] =			0x30,
812 	[PWRAP_STAUPD_GRPEN] =			0x34,
813 	[PWRAP_EINT_STA0_ADR] =			0x38,
814 	[PWRAP_HARB_HPRIO] =			0x5C,
815 	[PWRAP_HIPRIO_ARB_EN] =			0x60,
816 	[PWRAP_MAN_EN] =			0x70,
817 	[PWRAP_MAN_CMD] =			0x74,
818 	[PWRAP_WACS0_EN] =			0x80,
819 	[PWRAP_INIT_DONE0] =			0x84,
820 	[PWRAP_WACS1_EN] =			0x88,
821 	[PWRAP_INIT_DONE1] =			0x8C,
822 	[PWRAP_WACS2_EN] =			0x90,
823 	[PWRAP_INIT_DONE2] =			0x94,
824 	[PWRAP_WACS_P2P_EN] =			0xA0,
825 	[PWRAP_INIT_DONE_P2P] =			0xA4,
826 	[PWRAP_WACS_MD32_EN] =			0xA8,
827 	[PWRAP_INIT_DONE_MD32] =		0xAC,
828 	[PWRAP_INT_EN] =			0xB0,
829 	[PWRAP_INT_FLG] =			0xB8,
830 	[PWRAP_INT_CLR] =			0xBC,
831 	[PWRAP_INT1_EN] =			0xC0,
832 	[PWRAP_INT1_FLG] =			0xC8,
833 	[PWRAP_INT1_CLR] =			0xCC,
834 	[PWRAP_SIG_ADR] =			0xD0,
835 	[PWRAP_CRC_EN] =			0xE0,
836 	[PWRAP_TIMER_EN] =			0xE4,
837 	[PWRAP_WDT_UNIT] =			0xEC,
838 	[PWRAP_WDT_SRC_EN] =			0xF0,
839 	[PWRAP_WDT_SRC_EN_1] =			0xF4,
840 	[PWRAP_INT_GPS_AUXADC_CMD_ADDR] =	0x1DC,
841 	[PWRAP_INT_GPS_AUXADC_CMD] =		0x1E0,
842 	[PWRAP_INT_GPS_AUXADC_RDATA_ADDR] =	0x1E4,
843 	[PWRAP_EXT_GPS_AUXADC_RDATA_ADDR] =	0x1E8,
844 	[PWRAP_GPSINF_0_STA] =			0x1EC,
845 	[PWRAP_GPSINF_1_STA] =			0x1F0,
846 	[PWRAP_WACS2_CMD] =			0xC20,
847 	[PWRAP_WACS2_RDATA] =			0xC24,
848 	[PWRAP_WACS2_VLDCLR] =			0xC28,
849 };
850 
851 static int mt8516_regs[] = {
852 	[PWRAP_MUX_SEL] =		0x0,
853 	[PWRAP_WRAP_EN] =		0x4,
854 	[PWRAP_DIO_EN] =		0x8,
855 	[PWRAP_SIDLY] =			0xc,
856 	[PWRAP_RDDMY] =			0x10,
857 	[PWRAP_SI_CK_CON] =		0x14,
858 	[PWRAP_CSHEXT_WRITE] =		0x18,
859 	[PWRAP_CSHEXT_READ] =		0x1c,
860 	[PWRAP_CSLEXT_START] =		0x20,
861 	[PWRAP_CSLEXT_END] =		0x24,
862 	[PWRAP_STAUPD_PRD] =		0x28,
863 	[PWRAP_STAUPD_GRPEN] =		0x2c,
864 	[PWRAP_STAUPD_MAN_TRIG] =	0x40,
865 	[PWRAP_STAUPD_STA] =		0x44,
866 	[PWRAP_WRAP_STA] =		0x48,
867 	[PWRAP_HARB_INIT] =		0x4c,
868 	[PWRAP_HARB_HPRIO] =		0x50,
869 	[PWRAP_HIPRIO_ARB_EN] =		0x54,
870 	[PWRAP_HARB_STA0] =		0x58,
871 	[PWRAP_HARB_STA1] =		0x5c,
872 	[PWRAP_MAN_EN] =		0x60,
873 	[PWRAP_MAN_CMD] =		0x64,
874 	[PWRAP_MAN_RDATA] =		0x68,
875 	[PWRAP_MAN_VLDCLR] =		0x6c,
876 	[PWRAP_WACS0_EN] =		0x70,
877 	[PWRAP_INIT_DONE0] =		0x74,
878 	[PWRAP_WACS0_CMD] =		0x78,
879 	[PWRAP_WACS0_RDATA] =		0x7c,
880 	[PWRAP_WACS0_VLDCLR] =		0x80,
881 	[PWRAP_WACS1_EN] =		0x84,
882 	[PWRAP_INIT_DONE1] =		0x88,
883 	[PWRAP_WACS1_CMD] =		0x8c,
884 	[PWRAP_WACS1_RDATA] =		0x90,
885 	[PWRAP_WACS1_VLDCLR] =		0x94,
886 	[PWRAP_WACS2_EN] =		0x98,
887 	[PWRAP_INIT_DONE2] =		0x9c,
888 	[PWRAP_WACS2_CMD] =		0xa0,
889 	[PWRAP_WACS2_RDATA] =		0xa4,
890 	[PWRAP_WACS2_VLDCLR] =		0xa8,
891 	[PWRAP_INT_EN] =		0xac,
892 	[PWRAP_INT_FLG_RAW] =		0xb0,
893 	[PWRAP_INT_FLG] =		0xb4,
894 	[PWRAP_INT_CLR] =		0xb8,
895 	[PWRAP_SIG_ADR] =		0xbc,
896 	[PWRAP_SIG_MODE] =		0xc0,
897 	[PWRAP_SIG_VALUE] =		0xc4,
898 	[PWRAP_SIG_ERRVAL] =		0xc8,
899 	[PWRAP_CRC_EN] =		0xcc,
900 	[PWRAP_TIMER_EN] =		0xd0,
901 	[PWRAP_TIMER_STA] =		0xd4,
902 	[PWRAP_WDT_UNIT] =		0xd8,
903 	[PWRAP_WDT_SRC_EN] =		0xdc,
904 	[PWRAP_WDT_FLG] =		0xe0,
905 	[PWRAP_DEBUG_INT_SEL] =		0xe4,
906 	[PWRAP_DVFS_ADR0] =		0xe8,
907 	[PWRAP_DVFS_WDATA0] =		0xec,
908 	[PWRAP_DVFS_ADR1] =		0xf0,
909 	[PWRAP_DVFS_WDATA1] =		0xf4,
910 	[PWRAP_DVFS_ADR2] =		0xf8,
911 	[PWRAP_DVFS_WDATA2] =		0xfc,
912 	[PWRAP_DVFS_ADR3] =		0x100,
913 	[PWRAP_DVFS_WDATA3] =		0x104,
914 	[PWRAP_DVFS_ADR4] =		0x108,
915 	[PWRAP_DVFS_WDATA4] =		0x10c,
916 	[PWRAP_DVFS_ADR5] =		0x110,
917 	[PWRAP_DVFS_WDATA5] =		0x114,
918 	[PWRAP_DVFS_ADR6] =		0x118,
919 	[PWRAP_DVFS_WDATA6] =		0x11c,
920 	[PWRAP_DVFS_ADR7] =		0x120,
921 	[PWRAP_DVFS_WDATA7] =		0x124,
922 	[PWRAP_SPMINF_STA] =		0x128,
923 	[PWRAP_CIPHER_KEY_SEL] =	0x12c,
924 	[PWRAP_CIPHER_IV_SEL] =		0x130,
925 	[PWRAP_CIPHER_EN] =		0x134,
926 	[PWRAP_CIPHER_RDY] =		0x138,
927 	[PWRAP_CIPHER_MODE] =		0x13c,
928 	[PWRAP_CIPHER_SWRST] =		0x140,
929 	[PWRAP_DCM_EN] =		0x144,
930 	[PWRAP_DCM_DBC_PRD] =		0x148,
931 	[PWRAP_SW_RST] =		0x168,
932 	[PWRAP_OP_TYPE] =		0x16c,
933 	[PWRAP_MSB_FIRST] =		0x170,
934 };
935 
936 enum pmic_type {
937 	PMIC_MT6323,
938 	PMIC_MT6351,
939 	PMIC_MT6357,
940 	PMIC_MT6358,
941 	PMIC_MT6380,
942 	PMIC_MT6397,
943 };
944 
945 enum pwrap_type {
946 	PWRAP_MT2701,
947 	PWRAP_MT6765,
948 	PWRAP_MT6797,
949 	PWRAP_MT7622,
950 	PWRAP_MT8135,
951 	PWRAP_MT8173,
952 	PWRAP_MT8183,
953 	PWRAP_MT8516,
954 };
955 
956 struct pmic_wrapper;
957 struct pwrap_slv_type {
958 	const u32 *dew_regs;
959 	enum pmic_type type;
960 	const struct regmap_config *regmap;
961 	/* Flags indicating the capability for the target slave */
962 	u32 caps;
963 	/*
964 	 * pwrap operations are highly associated with the PMIC types,
965 	 * so the pointers added increases flexibility allowing determination
966 	 * which type is used by the detection through device tree.
967 	 */
968 	int (*pwrap_read)(struct pmic_wrapper *wrp, u32 adr, u32 *rdata);
969 	int (*pwrap_write)(struct pmic_wrapper *wrp, u32 adr, u32 wdata);
970 };
971 
972 struct pmic_wrapper {
973 	struct device *dev;
974 	void __iomem *base;
975 	struct regmap *regmap;
976 	const struct pmic_wrapper_type *master;
977 	const struct pwrap_slv_type *slave;
978 	struct clk *clk_spi;
979 	struct clk *clk_wrap;
980 	struct reset_control *rstc;
981 
982 	struct reset_control *rstc_bridge;
983 	void __iomem *bridge_base;
984 };
985 
986 struct pmic_wrapper_type {
987 	int *regs;
988 	enum pwrap_type type;
989 	u32 arb_en_all;
990 	u32 int_en_all;
991 	u32 int1_en_all;
992 	u32 spi_w;
993 	u32 wdt_src;
994 	/* Flags indicating the capability for the target pwrap */
995 	u32 caps;
996 	int (*init_reg_clock)(struct pmic_wrapper *wrp);
997 	int (*init_soc_specific)(struct pmic_wrapper *wrp);
998 };
999 
1000 static u32 pwrap_readl(struct pmic_wrapper *wrp, enum pwrap_regs reg)
1001 {
1002 	return readl(wrp->base + wrp->master->regs[reg]);
1003 }
1004 
1005 static void pwrap_writel(struct pmic_wrapper *wrp, u32 val, enum pwrap_regs reg)
1006 {
1007 	writel(val, wrp->base + wrp->master->regs[reg]);
1008 }
1009 
1010 static bool pwrap_is_fsm_idle(struct pmic_wrapper *wrp)
1011 {
1012 	u32 val = pwrap_readl(wrp, PWRAP_WACS2_RDATA);
1013 
1014 	return PWRAP_GET_WACS_FSM(val) == PWRAP_WACS_FSM_IDLE;
1015 }
1016 
1017 static bool pwrap_is_fsm_vldclr(struct pmic_wrapper *wrp)
1018 {
1019 	u32 val = pwrap_readl(wrp, PWRAP_WACS2_RDATA);
1020 
1021 	return PWRAP_GET_WACS_FSM(val) == PWRAP_WACS_FSM_WFVLDCLR;
1022 }
1023 
1024 /*
1025  * Timeout issue sometimes caused by the last read command
1026  * failed because pmic wrap could not got the FSM_VLDCLR
1027  * in time after finishing WACS2_CMD. It made state machine
1028  * still on FSM_VLDCLR and timeout next time.
1029  * Check the status of FSM and clear the vldclr to recovery the
1030  * error.
1031  */
1032 static inline void pwrap_leave_fsm_vldclr(struct pmic_wrapper *wrp)
1033 {
1034 	if (pwrap_is_fsm_vldclr(wrp))
1035 		pwrap_writel(wrp, 1, PWRAP_WACS2_VLDCLR);
1036 }
1037 
1038 static bool pwrap_is_sync_idle(struct pmic_wrapper *wrp)
1039 {
1040 	return pwrap_readl(wrp, PWRAP_WACS2_RDATA) & PWRAP_STATE_SYNC_IDLE0;
1041 }
1042 
1043 static bool pwrap_is_fsm_idle_and_sync_idle(struct pmic_wrapper *wrp)
1044 {
1045 	u32 val = pwrap_readl(wrp, PWRAP_WACS2_RDATA);
1046 
1047 	return (PWRAP_GET_WACS_FSM(val) == PWRAP_WACS_FSM_IDLE) &&
1048 		(val & PWRAP_STATE_SYNC_IDLE0);
1049 }
1050 
1051 static int pwrap_wait_for_state(struct pmic_wrapper *wrp,
1052 		bool (*fp)(struct pmic_wrapper *))
1053 {
1054 	unsigned long timeout;
1055 
1056 	timeout = jiffies + usecs_to_jiffies(10000);
1057 
1058 	do {
1059 		if (time_after(jiffies, timeout))
1060 			return fp(wrp) ? 0 : -ETIMEDOUT;
1061 		if (fp(wrp))
1062 			return 0;
1063 	} while (1);
1064 }
1065 
1066 static int pwrap_read16(struct pmic_wrapper *wrp, u32 adr, u32 *rdata)
1067 {
1068 	int ret;
1069 
1070 	ret = pwrap_wait_for_state(wrp, pwrap_is_fsm_idle);
1071 	if (ret) {
1072 		pwrap_leave_fsm_vldclr(wrp);
1073 		return ret;
1074 	}
1075 
1076 	pwrap_writel(wrp, (adr >> 1) << 16, PWRAP_WACS2_CMD);
1077 
1078 	ret = pwrap_wait_for_state(wrp, pwrap_is_fsm_vldclr);
1079 	if (ret)
1080 		return ret;
1081 
1082 	*rdata = PWRAP_GET_WACS_RDATA(pwrap_readl(wrp, PWRAP_WACS2_RDATA));
1083 
1084 	pwrap_writel(wrp, 1, PWRAP_WACS2_VLDCLR);
1085 
1086 	return 0;
1087 }
1088 
1089 static int pwrap_read32(struct pmic_wrapper *wrp, u32 adr, u32 *rdata)
1090 {
1091 	int ret, msb;
1092 
1093 	*rdata = 0;
1094 	for (msb = 0; msb < 2; msb++) {
1095 		ret = pwrap_wait_for_state(wrp, pwrap_is_fsm_idle);
1096 		if (ret) {
1097 			pwrap_leave_fsm_vldclr(wrp);
1098 			return ret;
1099 		}
1100 
1101 		pwrap_writel(wrp, ((msb << 30) | (adr << 16)),
1102 			     PWRAP_WACS2_CMD);
1103 
1104 		ret = pwrap_wait_for_state(wrp, pwrap_is_fsm_vldclr);
1105 		if (ret)
1106 			return ret;
1107 
1108 		*rdata += (PWRAP_GET_WACS_RDATA(pwrap_readl(wrp,
1109 			   PWRAP_WACS2_RDATA)) << (16 * msb));
1110 
1111 		pwrap_writel(wrp, 1, PWRAP_WACS2_VLDCLR);
1112 	}
1113 
1114 	return 0;
1115 }
1116 
1117 static int pwrap_read(struct pmic_wrapper *wrp, u32 adr, u32 *rdata)
1118 {
1119 	return wrp->slave->pwrap_read(wrp, adr, rdata);
1120 }
1121 
1122 static int pwrap_write16(struct pmic_wrapper *wrp, u32 adr, u32 wdata)
1123 {
1124 	int ret;
1125 
1126 	ret = pwrap_wait_for_state(wrp, pwrap_is_fsm_idle);
1127 	if (ret) {
1128 		pwrap_leave_fsm_vldclr(wrp);
1129 		return ret;
1130 	}
1131 
1132 	pwrap_writel(wrp, (1 << 31) | ((adr >> 1) << 16) | wdata,
1133 		     PWRAP_WACS2_CMD);
1134 
1135 	return 0;
1136 }
1137 
1138 static int pwrap_write32(struct pmic_wrapper *wrp, u32 adr, u32 wdata)
1139 {
1140 	int ret, msb, rdata;
1141 
1142 	for (msb = 0; msb < 2; msb++) {
1143 		ret = pwrap_wait_for_state(wrp, pwrap_is_fsm_idle);
1144 		if (ret) {
1145 			pwrap_leave_fsm_vldclr(wrp);
1146 			return ret;
1147 		}
1148 
1149 		pwrap_writel(wrp, (1 << 31) | (msb << 30) | (adr << 16) |
1150 			     ((wdata >> (msb * 16)) & 0xffff),
1151 			     PWRAP_WACS2_CMD);
1152 
1153 		/*
1154 		 * The pwrap_read operation is the requirement of hardware used
1155 		 * for the synchronization between two successive 16-bit
1156 		 * pwrap_writel operations composing one 32-bit bus writing.
1157 		 * Otherwise, we'll find the result fails on the lower 16-bit
1158 		 * pwrap writing.
1159 		 */
1160 		if (!msb)
1161 			pwrap_read(wrp, adr, &rdata);
1162 	}
1163 
1164 	return 0;
1165 }
1166 
1167 static int pwrap_write(struct pmic_wrapper *wrp, u32 adr, u32 wdata)
1168 {
1169 	return wrp->slave->pwrap_write(wrp, adr, wdata);
1170 }
1171 
1172 static int pwrap_regmap_read(void *context, u32 adr, u32 *rdata)
1173 {
1174 	return pwrap_read(context, adr, rdata);
1175 }
1176 
1177 static int pwrap_regmap_write(void *context, u32 adr, u32 wdata)
1178 {
1179 	return pwrap_write(context, adr, wdata);
1180 }
1181 
1182 static int pwrap_reset_spislave(struct pmic_wrapper *wrp)
1183 {
1184 	int ret, i;
1185 
1186 	pwrap_writel(wrp, 0, PWRAP_HIPRIO_ARB_EN);
1187 	pwrap_writel(wrp, 0, PWRAP_WRAP_EN);
1188 	pwrap_writel(wrp, 1, PWRAP_MUX_SEL);
1189 	pwrap_writel(wrp, 1, PWRAP_MAN_EN);
1190 	pwrap_writel(wrp, 0, PWRAP_DIO_EN);
1191 
1192 	pwrap_writel(wrp, wrp->master->spi_w | PWRAP_MAN_CMD_OP_CSL,
1193 			PWRAP_MAN_CMD);
1194 	pwrap_writel(wrp, wrp->master->spi_w | PWRAP_MAN_CMD_OP_OUTS,
1195 			PWRAP_MAN_CMD);
1196 	pwrap_writel(wrp, wrp->master->spi_w | PWRAP_MAN_CMD_OP_CSH,
1197 			PWRAP_MAN_CMD);
1198 
1199 	for (i = 0; i < 4; i++)
1200 		pwrap_writel(wrp, wrp->master->spi_w | PWRAP_MAN_CMD_OP_OUTS,
1201 				PWRAP_MAN_CMD);
1202 
1203 	ret = pwrap_wait_for_state(wrp, pwrap_is_sync_idle);
1204 	if (ret) {
1205 		dev_err(wrp->dev, "%s fail, ret=%d\n", __func__, ret);
1206 		return ret;
1207 	}
1208 
1209 	pwrap_writel(wrp, 0, PWRAP_MAN_EN);
1210 	pwrap_writel(wrp, 0, PWRAP_MUX_SEL);
1211 
1212 	return 0;
1213 }
1214 
1215 /*
1216  * pwrap_init_sidly - configure serial input delay
1217  *
1218  * This configures the serial input delay. We can configure 0, 2, 4 or 6ns
1219  * delay. Do a read test with all possible values and chose the best delay.
1220  */
1221 static int pwrap_init_sidly(struct pmic_wrapper *wrp)
1222 {
1223 	u32 rdata;
1224 	u32 i;
1225 	u32 pass = 0;
1226 	signed char dly[16] = {
1227 		-1, 0, 1, 0, 2, -1, 1, 1, 3, -1, -1, -1, 3, -1, 2, 1
1228 	};
1229 
1230 	for (i = 0; i < 4; i++) {
1231 		pwrap_writel(wrp, i, PWRAP_SIDLY);
1232 		pwrap_read(wrp, wrp->slave->dew_regs[PWRAP_DEW_READ_TEST],
1233 			   &rdata);
1234 		if (rdata == PWRAP_DEW_READ_TEST_VAL) {
1235 			dev_dbg(wrp->dev, "[Read Test] pass, SIDLY=%x\n", i);
1236 			pass |= 1 << i;
1237 		}
1238 	}
1239 
1240 	if (dly[pass] < 0) {
1241 		dev_err(wrp->dev, "sidly pass range 0x%x not continuous\n",
1242 				pass);
1243 		return -EIO;
1244 	}
1245 
1246 	pwrap_writel(wrp, dly[pass], PWRAP_SIDLY);
1247 
1248 	return 0;
1249 }
1250 
1251 static int pwrap_init_dual_io(struct pmic_wrapper *wrp)
1252 {
1253 	int ret;
1254 	u32 rdata;
1255 
1256 	/* Enable dual IO mode */
1257 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_DIO_EN], 1);
1258 
1259 	/* Check IDLE & INIT_DONE in advance */
1260 	ret = pwrap_wait_for_state(wrp,
1261 				   pwrap_is_fsm_idle_and_sync_idle);
1262 	if (ret) {
1263 		dev_err(wrp->dev, "%s fail, ret=%d\n", __func__, ret);
1264 		return ret;
1265 	}
1266 
1267 	pwrap_writel(wrp, 1, PWRAP_DIO_EN);
1268 
1269 	/* Read Test */
1270 	pwrap_read(wrp,
1271 		   wrp->slave->dew_regs[PWRAP_DEW_READ_TEST], &rdata);
1272 	if (rdata != PWRAP_DEW_READ_TEST_VAL) {
1273 		dev_err(wrp->dev,
1274 			"Read failed on DIO mode: 0x%04x!=0x%04x\n",
1275 			PWRAP_DEW_READ_TEST_VAL, rdata);
1276 		return -EFAULT;
1277 	}
1278 
1279 	return 0;
1280 }
1281 
1282 /*
1283  * pwrap_init_chip_select_ext is used to configure CS extension time for each
1284  * phase during data transactions on the pwrap bus.
1285  */
1286 static void pwrap_init_chip_select_ext(struct pmic_wrapper *wrp, u8 hext_write,
1287 				       u8 hext_read, u8 lext_start,
1288 				       u8 lext_end)
1289 {
1290 	/*
1291 	 * After finishing a write and read transaction, extends CS high time
1292 	 * to be at least xT of BUS CLK as hext_write and hext_read specifies
1293 	 * respectively.
1294 	 */
1295 	pwrap_writel(wrp, hext_write, PWRAP_CSHEXT_WRITE);
1296 	pwrap_writel(wrp, hext_read, PWRAP_CSHEXT_READ);
1297 
1298 	/*
1299 	 * Extends CS low time after CSL and before CSH command to be at
1300 	 * least xT of BUS CLK as lext_start and lext_end specifies
1301 	 * respectively.
1302 	 */
1303 	pwrap_writel(wrp, lext_start, PWRAP_CSLEXT_START);
1304 	pwrap_writel(wrp, lext_end, PWRAP_CSLEXT_END);
1305 }
1306 
1307 static int pwrap_common_init_reg_clock(struct pmic_wrapper *wrp)
1308 {
1309 	switch (wrp->master->type) {
1310 	case PWRAP_MT8173:
1311 		pwrap_init_chip_select_ext(wrp, 0, 4, 2, 2);
1312 		break;
1313 	case PWRAP_MT8135:
1314 		pwrap_writel(wrp, 0x4, PWRAP_CSHEXT);
1315 		pwrap_init_chip_select_ext(wrp, 0, 4, 0, 0);
1316 		break;
1317 	default:
1318 		break;
1319 	}
1320 
1321 	return 0;
1322 }
1323 
1324 static int pwrap_mt2701_init_reg_clock(struct pmic_wrapper *wrp)
1325 {
1326 	switch (wrp->slave->type) {
1327 	case PMIC_MT6397:
1328 		pwrap_writel(wrp, 0xc, PWRAP_RDDMY);
1329 		pwrap_init_chip_select_ext(wrp, 4, 0, 2, 2);
1330 		break;
1331 
1332 	case PMIC_MT6323:
1333 		pwrap_writel(wrp, 0x8, PWRAP_RDDMY);
1334 		pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_RDDMY_NO],
1335 			    0x8);
1336 		pwrap_init_chip_select_ext(wrp, 5, 0, 2, 2);
1337 		break;
1338 	default:
1339 		break;
1340 	}
1341 
1342 	return 0;
1343 }
1344 
1345 static bool pwrap_is_cipher_ready(struct pmic_wrapper *wrp)
1346 {
1347 	return pwrap_readl(wrp, PWRAP_CIPHER_RDY) & 1;
1348 }
1349 
1350 static bool pwrap_is_pmic_cipher_ready(struct pmic_wrapper *wrp)
1351 {
1352 	u32 rdata;
1353 	int ret;
1354 
1355 	ret = pwrap_read(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_RDY],
1356 			 &rdata);
1357 	if (ret)
1358 		return false;
1359 
1360 	return rdata == 1;
1361 }
1362 
1363 static int pwrap_init_cipher(struct pmic_wrapper *wrp)
1364 {
1365 	int ret;
1366 	u32 rdata = 0;
1367 
1368 	pwrap_writel(wrp, 0x1, PWRAP_CIPHER_SWRST);
1369 	pwrap_writel(wrp, 0x0, PWRAP_CIPHER_SWRST);
1370 	pwrap_writel(wrp, 0x1, PWRAP_CIPHER_KEY_SEL);
1371 	pwrap_writel(wrp, 0x2, PWRAP_CIPHER_IV_SEL);
1372 
1373 	switch (wrp->master->type) {
1374 	case PWRAP_MT8135:
1375 		pwrap_writel(wrp, 1, PWRAP_CIPHER_LOAD);
1376 		pwrap_writel(wrp, 1, PWRAP_CIPHER_START);
1377 		break;
1378 	case PWRAP_MT2701:
1379 	case PWRAP_MT6765:
1380 	case PWRAP_MT6797:
1381 	case PWRAP_MT8173:
1382 	case PWRAP_MT8516:
1383 		pwrap_writel(wrp, 1, PWRAP_CIPHER_EN);
1384 		break;
1385 	case PWRAP_MT7622:
1386 		pwrap_writel(wrp, 0, PWRAP_CIPHER_EN);
1387 		break;
1388 	case PWRAP_MT8183:
1389 		break;
1390 	}
1391 
1392 	/* Config cipher mode @PMIC */
1393 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_SWRST], 0x1);
1394 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_SWRST], 0x0);
1395 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_KEY_SEL], 0x1);
1396 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_IV_SEL], 0x2);
1397 
1398 	switch (wrp->slave->type) {
1399 	case PMIC_MT6397:
1400 		pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_LOAD],
1401 			    0x1);
1402 		pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_START],
1403 			    0x1);
1404 		break;
1405 	case PMIC_MT6323:
1406 	case PMIC_MT6351:
1407 	case PMIC_MT6357:
1408 		pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_EN],
1409 			    0x1);
1410 		break;
1411 	default:
1412 		break;
1413 	}
1414 
1415 	/* wait for cipher data ready@AP */
1416 	ret = pwrap_wait_for_state(wrp, pwrap_is_cipher_ready);
1417 	if (ret) {
1418 		dev_err(wrp->dev, "cipher data ready@AP fail, ret=%d\n", ret);
1419 		return ret;
1420 	}
1421 
1422 	/* wait for cipher data ready@PMIC */
1423 	ret = pwrap_wait_for_state(wrp, pwrap_is_pmic_cipher_ready);
1424 	if (ret) {
1425 		dev_err(wrp->dev,
1426 			"timeout waiting for cipher data ready@PMIC\n");
1427 		return ret;
1428 	}
1429 
1430 	/* wait for cipher mode idle */
1431 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CIPHER_MODE], 0x1);
1432 	ret = pwrap_wait_for_state(wrp, pwrap_is_fsm_idle_and_sync_idle);
1433 	if (ret) {
1434 		dev_err(wrp->dev, "cipher mode idle fail, ret=%d\n", ret);
1435 		return ret;
1436 	}
1437 
1438 	pwrap_writel(wrp, 1, PWRAP_CIPHER_MODE);
1439 
1440 	/* Write Test */
1441 	if (pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_WRITE_TEST],
1442 			PWRAP_DEW_WRITE_TEST_VAL) ||
1443 	    pwrap_read(wrp, wrp->slave->dew_regs[PWRAP_DEW_WRITE_TEST],
1444 		       &rdata) ||
1445 	    (rdata != PWRAP_DEW_WRITE_TEST_VAL)) {
1446 		dev_err(wrp->dev, "rdata=0x%04X\n", rdata);
1447 		return -EFAULT;
1448 	}
1449 
1450 	return 0;
1451 }
1452 
1453 static int pwrap_init_security(struct pmic_wrapper *wrp)
1454 {
1455 	int ret;
1456 
1457 	/* Enable encryption */
1458 	ret = pwrap_init_cipher(wrp);
1459 	if (ret)
1460 		return ret;
1461 
1462 	/* Signature checking - using CRC */
1463 	if (pwrap_write(wrp,
1464 			wrp->slave->dew_regs[PWRAP_DEW_CRC_EN], 0x1))
1465 		return -EFAULT;
1466 
1467 	pwrap_writel(wrp, 0x1, PWRAP_CRC_EN);
1468 	pwrap_writel(wrp, 0x0, PWRAP_SIG_MODE);
1469 	pwrap_writel(wrp, wrp->slave->dew_regs[PWRAP_DEW_CRC_VAL],
1470 		     PWRAP_SIG_ADR);
1471 	pwrap_writel(wrp,
1472 		     wrp->master->arb_en_all, PWRAP_HIPRIO_ARB_EN);
1473 
1474 	return 0;
1475 }
1476 
1477 static int pwrap_mt8135_init_soc_specific(struct pmic_wrapper *wrp)
1478 {
1479 	/* enable pwrap events and pwrap bridge in AP side */
1480 	pwrap_writel(wrp, 0x1, PWRAP_EVENT_IN_EN);
1481 	pwrap_writel(wrp, 0xffff, PWRAP_EVENT_DST_EN);
1482 	writel(0x7f, wrp->bridge_base + PWRAP_MT8135_BRIDGE_IORD_ARB_EN);
1483 	writel(0x1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_WACS3_EN);
1484 	writel(0x1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_WACS4_EN);
1485 	writel(0x1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_WDT_UNIT);
1486 	writel(0xffff, wrp->bridge_base + PWRAP_MT8135_BRIDGE_WDT_SRC_EN);
1487 	writel(0x1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_TIMER_EN);
1488 	writel(0x7ff, wrp->bridge_base + PWRAP_MT8135_BRIDGE_INT_EN);
1489 
1490 	/* enable PMIC event out and sources */
1491 	if (pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_EVENT_OUT_EN],
1492 			0x1) ||
1493 	    pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_EVENT_SRC_EN],
1494 			0xffff)) {
1495 		dev_err(wrp->dev, "enable dewrap fail\n");
1496 		return -EFAULT;
1497 	}
1498 
1499 	return 0;
1500 }
1501 
1502 static int pwrap_mt8173_init_soc_specific(struct pmic_wrapper *wrp)
1503 {
1504 	/* PMIC_DEWRAP enables */
1505 	if (pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_EVENT_OUT_EN],
1506 			0x1) ||
1507 	    pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_EVENT_SRC_EN],
1508 			0xffff)) {
1509 		dev_err(wrp->dev, "enable dewrap fail\n");
1510 		return -EFAULT;
1511 	}
1512 
1513 	return 0;
1514 }
1515 
1516 static int pwrap_mt2701_init_soc_specific(struct pmic_wrapper *wrp)
1517 {
1518 	/* GPS_INTF initialization */
1519 	switch (wrp->slave->type) {
1520 	case PMIC_MT6323:
1521 		pwrap_writel(wrp, 0x076c, PWRAP_ADC_CMD_ADDR);
1522 		pwrap_writel(wrp, 0x8000, PWRAP_PWRAP_ADC_CMD);
1523 		pwrap_writel(wrp, 0x072c, PWRAP_ADC_RDY_ADDR);
1524 		pwrap_writel(wrp, 0x072e, PWRAP_ADC_RDATA_ADDR1);
1525 		pwrap_writel(wrp, 0x0730, PWRAP_ADC_RDATA_ADDR2);
1526 		break;
1527 	default:
1528 		break;
1529 	}
1530 
1531 	return 0;
1532 }
1533 
1534 static int pwrap_mt7622_init_soc_specific(struct pmic_wrapper *wrp)
1535 {
1536 	pwrap_writel(wrp, 0, PWRAP_STAUPD_PRD);
1537 	/* enable 2wire SPI master */
1538 	pwrap_writel(wrp, 0x8000000, PWRAP_SPI2_CTRL);
1539 
1540 	return 0;
1541 }
1542 
1543 static int pwrap_mt8183_init_soc_specific(struct pmic_wrapper *wrp)
1544 {
1545 	pwrap_writel(wrp, 0xf5, PWRAP_STAUPD_GRPEN);
1546 
1547 	pwrap_write(wrp, wrp->slave->dew_regs[PWRAP_DEW_CRC_EN], 0x1);
1548 	pwrap_writel(wrp, 1, PWRAP_CRC_EN);
1549 	pwrap_writel(wrp, 0x416, PWRAP_SIG_ADR);
1550 	pwrap_writel(wrp, 0x42e, PWRAP_EINT_STA0_ADR);
1551 
1552 	pwrap_writel(wrp, 1, PWRAP_WACS_P2P_EN);
1553 	pwrap_writel(wrp, 1, PWRAP_WACS_MD32_EN);
1554 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE_P2P);
1555 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE_MD32);
1556 
1557 	return 0;
1558 }
1559 
1560 static int pwrap_init(struct pmic_wrapper *wrp)
1561 {
1562 	int ret;
1563 
1564 	if (wrp->rstc)
1565 		reset_control_reset(wrp->rstc);
1566 	if (wrp->rstc_bridge)
1567 		reset_control_reset(wrp->rstc_bridge);
1568 
1569 	if (wrp->master->type == PWRAP_MT8173) {
1570 		/* Enable DCM */
1571 		pwrap_writel(wrp, 3, PWRAP_DCM_EN);
1572 		pwrap_writel(wrp, 0, PWRAP_DCM_DBC_PRD);
1573 	}
1574 
1575 	if (HAS_CAP(wrp->slave->caps, PWRAP_SLV_CAP_SPI)) {
1576 		/* Reset SPI slave */
1577 		ret = pwrap_reset_spislave(wrp);
1578 		if (ret)
1579 			return ret;
1580 	}
1581 
1582 	pwrap_writel(wrp, 1, PWRAP_WRAP_EN);
1583 
1584 	pwrap_writel(wrp, wrp->master->arb_en_all, PWRAP_HIPRIO_ARB_EN);
1585 
1586 	pwrap_writel(wrp, 1, PWRAP_WACS2_EN);
1587 
1588 	ret = wrp->master->init_reg_clock(wrp);
1589 	if (ret)
1590 		return ret;
1591 
1592 	if (HAS_CAP(wrp->slave->caps, PWRAP_SLV_CAP_SPI)) {
1593 		/* Setup serial input delay */
1594 		ret = pwrap_init_sidly(wrp);
1595 		if (ret)
1596 			return ret;
1597 	}
1598 
1599 	if (HAS_CAP(wrp->slave->caps, PWRAP_SLV_CAP_DUALIO)) {
1600 		/* Enable dual I/O mode */
1601 		ret = pwrap_init_dual_io(wrp);
1602 		if (ret)
1603 			return ret;
1604 	}
1605 
1606 	if (HAS_CAP(wrp->slave->caps, PWRAP_SLV_CAP_SECURITY)) {
1607 		/* Enable security on bus */
1608 		ret = pwrap_init_security(wrp);
1609 		if (ret)
1610 			return ret;
1611 	}
1612 
1613 	if (wrp->master->type == PWRAP_MT8135)
1614 		pwrap_writel(wrp, 0x7, PWRAP_RRARB_EN);
1615 
1616 	pwrap_writel(wrp, 0x1, PWRAP_WACS0_EN);
1617 	pwrap_writel(wrp, 0x1, PWRAP_WACS1_EN);
1618 	pwrap_writel(wrp, 0x1, PWRAP_WACS2_EN);
1619 	pwrap_writel(wrp, 0x5, PWRAP_STAUPD_PRD);
1620 	pwrap_writel(wrp, 0xff, PWRAP_STAUPD_GRPEN);
1621 
1622 	if (wrp->master->init_soc_specific) {
1623 		ret = wrp->master->init_soc_specific(wrp);
1624 		if (ret)
1625 			return ret;
1626 	}
1627 
1628 	/* Setup the init done registers */
1629 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE2);
1630 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE0);
1631 	pwrap_writel(wrp, 1, PWRAP_INIT_DONE1);
1632 
1633 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_BRIDGE)) {
1634 		writel(1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_INIT_DONE3);
1635 		writel(1, wrp->bridge_base + PWRAP_MT8135_BRIDGE_INIT_DONE4);
1636 	}
1637 
1638 	return 0;
1639 }
1640 
1641 static irqreturn_t pwrap_interrupt(int irqno, void *dev_id)
1642 {
1643 	u32 rdata;
1644 	struct pmic_wrapper *wrp = dev_id;
1645 
1646 	rdata = pwrap_readl(wrp, PWRAP_INT_FLG);
1647 	dev_err(wrp->dev, "unexpected interrupt int=0x%x\n", rdata);
1648 	pwrap_writel(wrp, 0xffffffff, PWRAP_INT_CLR);
1649 
1650 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_INT1_EN)) {
1651 		rdata = pwrap_readl(wrp, PWRAP_INT1_FLG);
1652 		dev_err(wrp->dev, "unexpected interrupt int1=0x%x\n", rdata);
1653 		pwrap_writel(wrp, 0xffffffff, PWRAP_INT1_CLR);
1654 	}
1655 
1656 	return IRQ_HANDLED;
1657 }
1658 
1659 static const struct regmap_config pwrap_regmap_config16 = {
1660 	.reg_bits = 16,
1661 	.val_bits = 16,
1662 	.reg_stride = 2,
1663 	.reg_read = pwrap_regmap_read,
1664 	.reg_write = pwrap_regmap_write,
1665 	.max_register = 0xffff,
1666 };
1667 
1668 static const struct regmap_config pwrap_regmap_config32 = {
1669 	.reg_bits = 32,
1670 	.val_bits = 32,
1671 	.reg_stride = 4,
1672 	.reg_read = pwrap_regmap_read,
1673 	.reg_write = pwrap_regmap_write,
1674 	.max_register = 0xffff,
1675 };
1676 
1677 static const struct pwrap_slv_type pmic_mt6323 = {
1678 	.dew_regs = mt6323_regs,
1679 	.type = PMIC_MT6323,
1680 	.regmap = &pwrap_regmap_config16,
1681 	.caps = PWRAP_SLV_CAP_SPI | PWRAP_SLV_CAP_DUALIO |
1682 		PWRAP_SLV_CAP_SECURITY,
1683 	.pwrap_read = pwrap_read16,
1684 	.pwrap_write = pwrap_write16,
1685 };
1686 
1687 static const struct pwrap_slv_type pmic_mt6351 = {
1688 	.dew_regs = mt6351_regs,
1689 	.type = PMIC_MT6351,
1690 	.regmap = &pwrap_regmap_config16,
1691 	.caps = 0,
1692 	.pwrap_read = pwrap_read16,
1693 	.pwrap_write = pwrap_write16,
1694 };
1695 
1696 static const struct pwrap_slv_type pmic_mt6357 = {
1697 	.dew_regs = mt6357_regs,
1698 	.type = PMIC_MT6357,
1699 	.regmap = &pwrap_regmap_config16,
1700 	.caps = 0,
1701 	.pwrap_read = pwrap_read16,
1702 	.pwrap_write = pwrap_write16,
1703 };
1704 
1705 static const struct pwrap_slv_type pmic_mt6358 = {
1706 	.dew_regs = mt6358_regs,
1707 	.type = PMIC_MT6358,
1708 	.regmap = &pwrap_regmap_config16,
1709 	.caps = PWRAP_SLV_CAP_SPI | PWRAP_SLV_CAP_DUALIO,
1710 	.pwrap_read = pwrap_read16,
1711 	.pwrap_write = pwrap_write16,
1712 };
1713 
1714 static const struct pwrap_slv_type pmic_mt6380 = {
1715 	.dew_regs = NULL,
1716 	.type = PMIC_MT6380,
1717 	.regmap = &pwrap_regmap_config32,
1718 	.caps = 0,
1719 	.pwrap_read = pwrap_read32,
1720 	.pwrap_write = pwrap_write32,
1721 };
1722 
1723 static const struct pwrap_slv_type pmic_mt6397 = {
1724 	.dew_regs = mt6397_regs,
1725 	.type = PMIC_MT6397,
1726 	.regmap = &pwrap_regmap_config16,
1727 	.caps = PWRAP_SLV_CAP_SPI | PWRAP_SLV_CAP_DUALIO |
1728 		PWRAP_SLV_CAP_SECURITY,
1729 	.pwrap_read = pwrap_read16,
1730 	.pwrap_write = pwrap_write16,
1731 };
1732 
1733 static const struct of_device_id of_slave_match_tbl[] = {
1734 	{
1735 		.compatible = "mediatek,mt6323",
1736 		.data = &pmic_mt6323,
1737 	}, {
1738 		.compatible = "mediatek,mt6351",
1739 		.data = &pmic_mt6351,
1740 	}, {
1741 		.compatible = "mediatek,mt6357",
1742 		.data = &pmic_mt6357,
1743 	}, {
1744 		.compatible = "mediatek,mt6358",
1745 		.data = &pmic_mt6358,
1746 	}, {
1747 		/* The MT6380 PMIC only implements a regulator, so we bind it
1748 		 * directly instead of using a MFD.
1749 		 */
1750 		.compatible = "mediatek,mt6380-regulator",
1751 		.data = &pmic_mt6380,
1752 	}, {
1753 		.compatible = "mediatek,mt6397",
1754 		.data = &pmic_mt6397,
1755 	}, {
1756 		/* sentinel */
1757 	}
1758 };
1759 MODULE_DEVICE_TABLE(of, of_slave_match_tbl);
1760 
1761 static const struct pmic_wrapper_type pwrap_mt2701 = {
1762 	.regs = mt2701_regs,
1763 	.type = PWRAP_MT2701,
1764 	.arb_en_all = 0x3f,
1765 	.int_en_all = ~(u32)(BIT(31) | BIT(2)),
1766 	.int1_en_all = 0,
1767 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE_NEW,
1768 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
1769 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
1770 	.init_reg_clock = pwrap_mt2701_init_reg_clock,
1771 	.init_soc_specific = pwrap_mt2701_init_soc_specific,
1772 };
1773 
1774 static const struct pmic_wrapper_type pwrap_mt6765 = {
1775 	.regs = mt6765_regs,
1776 	.type = PWRAP_MT6765,
1777 	.arb_en_all = 0x3fd35,
1778 	.int_en_all = 0xffffffff,
1779 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
1780 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
1781 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
1782 	.init_reg_clock = pwrap_common_init_reg_clock,
1783 	.init_soc_specific = NULL,
1784 };
1785 
1786 static const struct pmic_wrapper_type pwrap_mt6797 = {
1787 	.regs = mt6797_regs,
1788 	.type = PWRAP_MT6797,
1789 	.arb_en_all = 0x01fff,
1790 	.int_en_all = 0xffffffc6,
1791 	.int1_en_all = 0,
1792 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
1793 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
1794 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
1795 	.init_reg_clock = pwrap_common_init_reg_clock,
1796 	.init_soc_specific = NULL,
1797 };
1798 
1799 static const struct pmic_wrapper_type pwrap_mt7622 = {
1800 	.regs = mt7622_regs,
1801 	.type = PWRAP_MT7622,
1802 	.arb_en_all = 0xff,
1803 	.int_en_all = ~(u32)BIT(31),
1804 	.int1_en_all = 0,
1805 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
1806 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
1807 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
1808 	.init_reg_clock = pwrap_common_init_reg_clock,
1809 	.init_soc_specific = pwrap_mt7622_init_soc_specific,
1810 };
1811 
1812 static const struct pmic_wrapper_type pwrap_mt8135 = {
1813 	.regs = mt8135_regs,
1814 	.type = PWRAP_MT8135,
1815 	.arb_en_all = 0x1ff,
1816 	.int_en_all = ~(u32)(BIT(31) | BIT(1)),
1817 	.int1_en_all = 0,
1818 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
1819 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
1820 	.caps = PWRAP_CAP_BRIDGE | PWRAP_CAP_RESET | PWRAP_CAP_DCM,
1821 	.init_reg_clock = pwrap_common_init_reg_clock,
1822 	.init_soc_specific = pwrap_mt8135_init_soc_specific,
1823 };
1824 
1825 static const struct pmic_wrapper_type pwrap_mt8173 = {
1826 	.regs = mt8173_regs,
1827 	.type = PWRAP_MT8173,
1828 	.arb_en_all = 0x3f,
1829 	.int_en_all = ~(u32)(BIT(31) | BIT(1)),
1830 	.int1_en_all = 0,
1831 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
1832 	.wdt_src = PWRAP_WDT_SRC_MASK_NO_STAUPD,
1833 	.caps = PWRAP_CAP_RESET | PWRAP_CAP_DCM,
1834 	.init_reg_clock = pwrap_common_init_reg_clock,
1835 	.init_soc_specific = pwrap_mt8173_init_soc_specific,
1836 };
1837 
1838 static const struct pmic_wrapper_type pwrap_mt8183 = {
1839 	.regs = mt8183_regs,
1840 	.type = PWRAP_MT8183,
1841 	.arb_en_all = 0x3fa75,
1842 	.int_en_all = 0xffffffff,
1843 	.int1_en_all = 0xeef7ffff,
1844 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
1845 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
1846 	.caps = PWRAP_CAP_INT1_EN | PWRAP_CAP_WDT_SRC1,
1847 	.init_reg_clock = pwrap_common_init_reg_clock,
1848 	.init_soc_specific = pwrap_mt8183_init_soc_specific,
1849 };
1850 
1851 static struct pmic_wrapper_type pwrap_mt8516 = {
1852 	.regs = mt8516_regs,
1853 	.type = PWRAP_MT8516,
1854 	.arb_en_all = 0xff,
1855 	.int_en_all = ~(u32)(BIT(31) | BIT(2)),
1856 	.spi_w = PWRAP_MAN_CMD_SPI_WRITE,
1857 	.wdt_src = PWRAP_WDT_SRC_MASK_ALL,
1858 	.caps = PWRAP_CAP_DCM,
1859 	.init_reg_clock = pwrap_mt2701_init_reg_clock,
1860 	.init_soc_specific = NULL,
1861 };
1862 
1863 static const struct of_device_id of_pwrap_match_tbl[] = {
1864 	{
1865 		.compatible = "mediatek,mt2701-pwrap",
1866 		.data = &pwrap_mt2701,
1867 	}, {
1868 		.compatible = "mediatek,mt6765-pwrap",
1869 		.data = &pwrap_mt6765,
1870 	}, {
1871 		.compatible = "mediatek,mt6797-pwrap",
1872 		.data = &pwrap_mt6797,
1873 	}, {
1874 		.compatible = "mediatek,mt7622-pwrap",
1875 		.data = &pwrap_mt7622,
1876 	}, {
1877 		.compatible = "mediatek,mt8135-pwrap",
1878 		.data = &pwrap_mt8135,
1879 	}, {
1880 		.compatible = "mediatek,mt8173-pwrap",
1881 		.data = &pwrap_mt8173,
1882 	}, {
1883 		.compatible = "mediatek,mt8183-pwrap",
1884 		.data = &pwrap_mt8183,
1885 	}, {
1886 		.compatible = "mediatek,mt8516-pwrap",
1887 		.data = &pwrap_mt8516,
1888 	}, {
1889 		/* sentinel */
1890 	}
1891 };
1892 MODULE_DEVICE_TABLE(of, of_pwrap_match_tbl);
1893 
1894 static int pwrap_probe(struct platform_device *pdev)
1895 {
1896 	int ret, irq;
1897 	struct pmic_wrapper *wrp;
1898 	struct device_node *np = pdev->dev.of_node;
1899 	const struct of_device_id *of_slave_id = NULL;
1900 	struct resource *res;
1901 
1902 	if (np->child)
1903 		of_slave_id = of_match_node(of_slave_match_tbl, np->child);
1904 
1905 	if (!of_slave_id) {
1906 		dev_dbg(&pdev->dev, "slave pmic should be defined in dts\n");
1907 		return -EINVAL;
1908 	}
1909 
1910 	wrp = devm_kzalloc(&pdev->dev, sizeof(*wrp), GFP_KERNEL);
1911 	if (!wrp)
1912 		return -ENOMEM;
1913 
1914 	platform_set_drvdata(pdev, wrp);
1915 
1916 	wrp->master = of_device_get_match_data(&pdev->dev);
1917 	wrp->slave = of_slave_id->data;
1918 	wrp->dev = &pdev->dev;
1919 
1920 	res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "pwrap");
1921 	wrp->base = devm_ioremap_resource(wrp->dev, res);
1922 	if (IS_ERR(wrp->base))
1923 		return PTR_ERR(wrp->base);
1924 
1925 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_RESET)) {
1926 		wrp->rstc = devm_reset_control_get(wrp->dev, "pwrap");
1927 		if (IS_ERR(wrp->rstc)) {
1928 			ret = PTR_ERR(wrp->rstc);
1929 			dev_dbg(wrp->dev, "cannot get pwrap reset: %d\n", ret);
1930 			return ret;
1931 		}
1932 	}
1933 
1934 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_BRIDGE)) {
1935 		res = platform_get_resource_byname(pdev, IORESOURCE_MEM,
1936 				"pwrap-bridge");
1937 		wrp->bridge_base = devm_ioremap_resource(wrp->dev, res);
1938 		if (IS_ERR(wrp->bridge_base))
1939 			return PTR_ERR(wrp->bridge_base);
1940 
1941 		wrp->rstc_bridge = devm_reset_control_get(wrp->dev,
1942 							  "pwrap-bridge");
1943 		if (IS_ERR(wrp->rstc_bridge)) {
1944 			ret = PTR_ERR(wrp->rstc_bridge);
1945 			dev_dbg(wrp->dev,
1946 				"cannot get pwrap-bridge reset: %d\n", ret);
1947 			return ret;
1948 		}
1949 	}
1950 
1951 	wrp->clk_spi = devm_clk_get(wrp->dev, "spi");
1952 	if (IS_ERR(wrp->clk_spi)) {
1953 		dev_dbg(wrp->dev, "failed to get clock: %ld\n",
1954 			PTR_ERR(wrp->clk_spi));
1955 		return PTR_ERR(wrp->clk_spi);
1956 	}
1957 
1958 	wrp->clk_wrap = devm_clk_get(wrp->dev, "wrap");
1959 	if (IS_ERR(wrp->clk_wrap)) {
1960 		dev_dbg(wrp->dev, "failed to get clock: %ld\n",
1961 			PTR_ERR(wrp->clk_wrap));
1962 		return PTR_ERR(wrp->clk_wrap);
1963 	}
1964 
1965 	ret = clk_prepare_enable(wrp->clk_spi);
1966 	if (ret)
1967 		return ret;
1968 
1969 	ret = clk_prepare_enable(wrp->clk_wrap);
1970 	if (ret)
1971 		goto err_out1;
1972 
1973 	/* Enable internal dynamic clock */
1974 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_DCM)) {
1975 		pwrap_writel(wrp, 1, PWRAP_DCM_EN);
1976 		pwrap_writel(wrp, 0, PWRAP_DCM_DBC_PRD);
1977 	}
1978 
1979 	/*
1980 	 * The PMIC could already be initialized by the bootloader.
1981 	 * Skip initialization here in this case.
1982 	 */
1983 	if (!pwrap_readl(wrp, PWRAP_INIT_DONE2)) {
1984 		ret = pwrap_init(wrp);
1985 		if (ret) {
1986 			dev_dbg(wrp->dev, "init failed with %d\n", ret);
1987 			goto err_out2;
1988 		}
1989 	}
1990 
1991 	if (!(pwrap_readl(wrp, PWRAP_WACS2_RDATA) & PWRAP_STATE_INIT_DONE0)) {
1992 		dev_dbg(wrp->dev, "initialization isn't finished\n");
1993 		ret = -ENODEV;
1994 		goto err_out2;
1995 	}
1996 
1997 	/* Initialize watchdog, may not be done by the bootloader */
1998 	pwrap_writel(wrp, 0xf, PWRAP_WDT_UNIT);
1999 	/*
2000 	 * Since STAUPD was not used on mt8173 platform,
2001 	 * so STAUPD of WDT_SRC which should be turned off
2002 	 */
2003 	pwrap_writel(wrp, wrp->master->wdt_src, PWRAP_WDT_SRC_EN);
2004 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_WDT_SRC1))
2005 		pwrap_writel(wrp, wrp->master->wdt_src, PWRAP_WDT_SRC_EN_1);
2006 
2007 	pwrap_writel(wrp, 0x1, PWRAP_TIMER_EN);
2008 	pwrap_writel(wrp, wrp->master->int_en_all, PWRAP_INT_EN);
2009 	/*
2010 	 * We add INT1 interrupt to handle starvation and request exception
2011 	 * If we support it, we should enable it here.
2012 	 */
2013 	if (HAS_CAP(wrp->master->caps, PWRAP_CAP_INT1_EN))
2014 		pwrap_writel(wrp, wrp->master->int1_en_all, PWRAP_INT1_EN);
2015 
2016 	irq = platform_get_irq(pdev, 0);
2017 	ret = devm_request_irq(wrp->dev, irq, pwrap_interrupt,
2018 			       IRQF_TRIGGER_HIGH,
2019 			       "mt-pmic-pwrap", wrp);
2020 	if (ret)
2021 		goto err_out2;
2022 
2023 	wrp->regmap = devm_regmap_init(wrp->dev, NULL, wrp, wrp->slave->regmap);
2024 	if (IS_ERR(wrp->regmap)) {
2025 		ret = PTR_ERR(wrp->regmap);
2026 		goto err_out2;
2027 	}
2028 
2029 	ret = of_platform_populate(np, NULL, NULL, wrp->dev);
2030 	if (ret) {
2031 		dev_dbg(wrp->dev, "failed to create child devices at %pOF\n",
2032 				np);
2033 		goto err_out2;
2034 	}
2035 
2036 	return 0;
2037 
2038 err_out2:
2039 	clk_disable_unprepare(wrp->clk_wrap);
2040 err_out1:
2041 	clk_disable_unprepare(wrp->clk_spi);
2042 
2043 	return ret;
2044 }
2045 
2046 static struct platform_driver pwrap_drv = {
2047 	.driver = {
2048 		.name = "mt-pmic-pwrap",
2049 		.of_match_table = of_match_ptr(of_pwrap_match_tbl),
2050 	},
2051 	.probe = pwrap_probe,
2052 };
2053 
2054 module_platform_driver(pwrap_drv);
2055 
2056 MODULE_AUTHOR("Flora Fu, MediaTek");
2057 MODULE_DESCRIPTION("MediaTek MT8135 PMIC Wrapper Driver");
2058 MODULE_LICENSE("GPL v2");
2059