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