1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * MFD core driver for Rockchip RK808/RK818
4  *
5  * Copyright (c) 2014-2018, Fuzhou Rockchip Electronics Co., Ltd
6  *
7  * Author: Chris Zhong <zyw@rock-chips.com>
8  * Author: Zhang Qing <zhangqing@rock-chips.com>
9  *
10  * Copyright (C) 2016 PHYTEC Messtechnik GmbH
11  *
12  * Author: Wadim Egorov <w.egorov@phytec.de>
13  */
14 
15 #include <linux/i2c.h>
16 #include <linux/interrupt.h>
17 #include <linux/mfd/rk808.h>
18 #include <linux/mfd/core.h>
19 #include <linux/module.h>
20 #include <linux/of_device.h>
21 #include <linux/reboot.h>
22 #include <linux/regmap.h>
23 #include <linux/syscore_ops.h>
24 #include <linux/pinctrl/consumer.h>
25 #include <linux/pinctrl/devinfo.h>
26 
27 struct rk808_reg_data {
28     int addr;
29     int mask;
30     int value;
31 };
32 
rk808_is_volatile_reg(struct device *dev, unsigned int reg)33 static bool rk808_is_volatile_reg(struct device *dev, unsigned int reg)
34 {
35     /*
36      * Notes:
37      * - Technically the ROUND_30s bit makes RTC_CTRL_REG volatile, but
38      *   we don't use that feature.  It's better to cache.
39      * - It's unlikely we care that RK808_DEVCTRL_REG is volatile since
40      *   bits are cleared in case when we shutoff anyway, but better safe.
41      */
42 
43     switch (reg) {
44         case RK808_SECONDS_REG ... RK808_WEEKS_REG:
45         case RK808_RTC_STATUS_REG:
46         case RK808_VB_MON_REG:
47         case RK808_THERMAL_REG:
48         case RK808_DCDC_UV_STS_REG:
49         case RK808_LDO_UV_STS_REG:
50         case RK808_DCDC_PG_REG:
51         case RK808_LDO_PG_REG:
52         case RK808_DEVCTRL_REG:
53         case RK808_INT_STS_REG1:
54         case RK808_INT_STS_REG2:
55             return true;
56         default:
57             break;
58     }
59 
60     return false;
61 }
62 
rk817_is_volatile_reg(struct device *dev, unsigned int reg)63 static bool rk817_is_volatile_reg(struct device *dev, unsigned int reg)
64 {
65     /*
66      * Notes:
67      * - Technically the ROUND_30s bit makes RTC_CTRL_REG volatile, but
68      *   we don't use that feature.  It's better to cache.
69      */
70 
71     switch (reg) {
72         case RK817_SECONDS_REG ... RK817_WEEKS_REG:
73         case RK817_RTC_STATUS_REG:
74         case RK817_ADC_CONFIG0 ... RK817_CURE_ADC_K0:
75         case RK817_CHRG_STS:
76         case RK817_CHRG_OUT:
77         case RK817_CHRG_IN:
78         case RK817_SYS_STS:
79         case RK817_INT_STS_REG0:
80         case RK817_INT_STS_REG1:
81         case RK817_INT_STS_REG2:
82             return true;
83         default:
84             break;
85     }
86 
87     return false;
88 }
89 
rk818_is_volatile_reg(struct device *dev, unsigned int reg)90 static bool rk818_is_volatile_reg(struct device *dev, unsigned int reg)
91 {
92     /*
93      * Notes:
94      * - Technically the ROUND_30s bit makes RTC_CTRL_REG volatile, but
95      *   we don't use that feature.  It's better to cache.
96      * - It's unlikely we care that RK808_DEVCTRL_REG is volatile since
97      *   bits are cleared in case when we shutoff anyway, but better safe.
98      */
99 
100     switch (reg) {
101         case RK808_SECONDS_REG ... RK808_WEEKS_REG:
102         case RK808_RTC_STATUS_REG:
103         case RK808_VB_MON_REG:
104         case RK808_THERMAL_REG:
105         case RK808_DCDC_EN_REG:
106         case RK808_LDO_EN_REG:
107         case RK808_DCDC_UV_STS_REG:
108         case RK808_LDO_UV_STS_REG:
109         case RK808_DCDC_PG_REG:
110         case RK808_LDO_PG_REG:
111         case RK808_DEVCTRL_REG:
112         case RK808_INT_STS_REG1:
113         case RK808_INT_STS_REG2:
114         case RK808_INT_STS_MSK_REG1:
115         case RK808_INT_STS_MSK_REG2:
116         case RK816_INT_STS_REG1:
117         case RK816_INT_STS_MSK_REG1:
118         case RK818_SUP_STS_REG ... RK818_SAVE_DATA19:
119             return true;
120         default:
121             break;
122     }
123 
124     return false;
125 }
126 
127 static const struct regmap_config rk818_regmap_config = {
128     .reg_bits = RK_REG_BITS,
129     .val_bits = RK_VAL_BITS,
130     .max_register = RK818_SAVE_DATA19,
131     .cache_type = REGCACHE_RBTREE,
132     .volatile_reg = rk818_is_volatile_reg,
133 };
134 
135 static const struct regmap_config rk805_regmap_config = {
136     .reg_bits = RK_REG_BITS,
137     .val_bits = RK_VAL_BITS,
138     .max_register = RK805_OFF_SOURCE_REG,
139     .cache_type = REGCACHE_RBTREE,
140     .volatile_reg = rk808_is_volatile_reg,
141 };
142 
143 static const struct regmap_config rk808_regmap_config = {
144     .reg_bits = RK_REG_BITS,
145     .val_bits = RK_VAL_BITS,
146     .max_register = RK808_IO_POL_REG,
147     .cache_type = REGCACHE_RBTREE,
148     .volatile_reg = rk808_is_volatile_reg,
149 };
150 
151 static const struct regmap_config rk816_regmap_config = {
152     .reg_bits = RK_REG_BITS,
153     .val_bits = RK_VAL_BITS,
154     .max_register = RK816_DATA18_REG,
155     .cache_type = REGCACHE_RBTREE,
156     .volatile_reg = rk818_is_volatile_reg,
157 };
158 
159 static const struct regmap_config rk817_regmap_config = {
160     .reg_bits = RK_REG_BITS,
161     .val_bits = RK_VAL_BITS,
162     .max_register = RK817_GPIO_INT_CFG,
163     .num_reg_defaults_raw = RK817_GPIO_INT_CFG + 1,
164     .cache_type = REGCACHE_RBTREE,
165     .volatile_reg = rk817_is_volatile_reg,
166 };
167 
168 static struct resource rtc_resources[] = {
169     DEFINE_RES_IRQ(RK808_IRQ_RTC_ALARM),
170 };
171 
172 static struct resource rk816_rtc_resources[] = {
173     DEFINE_RES_IRQ(RK816_IRQ_RTC_ALARM),
174 };
175 
176 static struct resource rk817_rtc_resources[] = {
177     DEFINE_RES_IRQ(RK817_IRQ_RTC_ALARM),
178 };
179 
180 static struct resource rk805_key_resources[] = {
181     DEFINE_RES_IRQ(RK805_IRQ_PWRON_FALL),
182     DEFINE_RES_IRQ(RK805_IRQ_PWRON_RISE),
183 };
184 
185 static struct resource rk816_pwrkey_resources[] = {
186     DEFINE_RES_IRQ(RK816_IRQ_PWRON_FALL),
187     DEFINE_RES_IRQ(RK816_IRQ_PWRON_RISE),
188 };
189 
190 static struct resource rk817_pwrkey_resources[] = {
191     DEFINE_RES_IRQ(RK817_IRQ_PWRON_FALL),
192     DEFINE_RES_IRQ(RK817_IRQ_PWRON_RISE),
193 };
194 
195 static const struct mfd_cell rk805s[] = {
196     {
197         .name = "rk808-clkout",
198     },
199     {
200         .name = "rk808-regulator",
201     },
202     {
203         .name = "rk805-pinctrl",
204     },
205     {
206         .name = "rk808-rtc",
207         .num_resources = ARRAY_SIZE(rtc_resources),
208         .resources = &rtc_resources[0],
209     },
210     {
211         .name = "rk805-pwrkey",
212         .num_resources = ARRAY_SIZE(rk805_key_resources),
213         .resources = &rk805_key_resources[0],
214     },
215 };
216 
217 static const struct mfd_cell rk808s[] = {
218     {
219         .name = "rk808-clkout",
220     },
221     {
222         .name = "rk808-regulator",
223     },
224     {
225         .name = "rk808-rtc",
226         .num_resources = ARRAY_SIZE(rtc_resources),
227         .resources = rtc_resources,
228     },
229 };
230 
231 static const struct mfd_cell rk816s[] = {
232     {
233         .name = "rk808-clkout",
234     },
235     {
236         .name = "rk808-regulator",
237     },
238     {
239         .name = "rk805-pinctrl",
240     },
241     {
242         .name = "rk816-battery",
243         .of_compatible = "rk816-battery",
244     },
245     {
246         .name = "rk805-pwrkey",
247         .num_resources = ARRAY_SIZE(rk816_pwrkey_resources),
248         .resources = &rk816_pwrkey_resources[0],
249     },
250     {
251         .name = "rk808-rtc",
252         .num_resources = ARRAY_SIZE(rk816_rtc_resources),
253         .resources = &rk816_rtc_resources[0],
254     },
255 };
256 
257 static const struct mfd_cell rk817s[] = {
258     {
259         .name = "rk808-clkout",
260     },
261     {
262         .name = "rk808-regulator",
263     },
264     {
265         .name = "rk817-battery",
266         .of_compatible = "rk817,battery",
267     },
268     {
269         .name = "rk817-charger",
270         .of_compatible = "rk817,charger",
271     },
272     {
273         .name = "rk805-pwrkey",
274         .num_resources = ARRAY_SIZE(rk817_pwrkey_resources),
275         .resources = &rk817_pwrkey_resources[0],
276     },
277     {
278         .name = "rk808-rtc",
279         .num_resources = ARRAY_SIZE(rk817_rtc_resources),
280         .resources = &rk817_rtc_resources[0],
281     },
282     {
283         .name = "rk817-codec",
284         .of_compatible = "rockchip,rk817-codec",
285     },
286 };
287 
288 static const struct mfd_cell rk818s[] = {
289     {
290         .name = "rk808-clkout",
291     },
292     {
293         .name = "rk808-regulator",
294     },
295     {
296         .name = "rk818-battery",
297         .of_compatible = "rk818-battery",
298     },
299     {
300         .name = "rk818-charger",
301     },
302     {
303         .name = "rk808-rtc",
304         .num_resources = ARRAY_SIZE(rtc_resources),
305         .resources = rtc_resources,
306     },
307 };
308 
309 static const struct rk808_reg_data rk805_pre_init_reg[] = {
310     {RK805_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK, BUCK_ILMIN_400MA},
311     {RK805_GPIO_IO_POL_REG, SLP_SD_MSK, SLEEP_FUN},
312     {RK805_THERMAL_REG, TEMP_HOTDIE_MSK, TEMP115C},
313     {RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
314 };
315 
316 static struct rk808_reg_data rk805_suspend_reg[] = {
317     {RK805_BUCK3_CONFIG_REG, PWM_MODE_MSK, AUTO_PWM_MODE},
318 };
319 
320 static struct rk808_reg_data rk805_resume_reg[] = {
321     {RK805_BUCK3_CONFIG_REG, PWM_MODE_MSK, FPWM_MODE},
322 };
323 
324 static const struct rk808_reg_data rk808_pre_init_reg[] = {
325     {RK808_BUCK3_CONFIG_REG, BUCK_ILMIN_MASK, BUCK_ILMIN_150MA},
326     {RK808_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK, BUCK_ILMIN_200MA},
327     {RK808_BOOST_CONFIG_REG, BOOST_ILMIN_MASK, BOOST_ILMIN_100MA},
328     {RK808_BUCK1_CONFIG_REG, BUCK1_RATE_MASK, BUCK_ILMIN_200MA},
329     {RK808_BUCK2_CONFIG_REG, BUCK2_RATE_MASK, BUCK_ILMIN_200MA},
330     {RK808_DCDC_UV_ACT_REG, BUCK_UV_ACT_MASK, BUCK_UV_ACT_DISABLE},
331     {RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
332     {RK808_VB_MON_REG, MASK_ALL, VB_LO_ACT | VB_LO_SEL_3500MV},
333 };
334 
335 static const struct rk808_reg_data rk816_pre_init_reg[] = {
336     /* buck4 Max ILMIT */
337     {RK816_BUCK4_CONFIG_REG, REG_WRITE_MSK, BUCK4_MAX_ILIMIT},
338     /* hotdie temperature: 105c */
339     {RK816_THERMAL_REG, REG_WRITE_MSK, TEMP105C},
340     /* set buck 12.5mv/us */
341     {RK816_BUCK1_CONFIG_REG, BUCK_RATE_MSK, BUCK_RATE_12_5MV_US},
342     {RK816_BUCK2_CONFIG_REG, BUCK_RATE_MSK, BUCK_RATE_12_5MV_US},
343     /* enable RTC_PERIOD & RTC_ALARM int */
344     {RK816_INT_STS_MSK_REG2, REG_WRITE_MSK, RTC_PERIOD_ALARM_INT_EN},
345     /* set bat 3.0 low and act shutdown */
346     {RK816_VB_MON_REG, VBAT_LOW_VOL_MASK | VBAT_LOW_ACT_MASK, RK816_VBAT_LOW_3V0 | EN_VABT_LOW_SHUT_DOWN},
347     /* enable PWRON rising/faling int */
348     {RK816_INT_STS_MSK_REG1, REG_WRITE_MSK, RK816_PWRON_FALL_RISE_INT_EN},
349     /* enable PLUG IN/OUT int */
350     {RK816_INT_STS_MSK_REG3, REG_WRITE_MSK, PLUGIN_OUT_INT_EN},
351     /* clear int flags */
352     {RK816_INT_STS_REG1, REG_WRITE_MSK, ALL_INT_FLAGS_ST},
353     {RK816_INT_STS_REG2, REG_WRITE_MSK, ALL_INT_FLAGS_ST},
354     {RK816_INT_STS_REG3, REG_WRITE_MSK, ALL_INT_FLAGS_ST},
355     {RK816_DCDC_EN_REG2, BOOST_EN_MASK, BOOST_DISABLE},
356     /* set write mask bit 1, otherwise 'is_enabled()' get wrong status */
357     {RK816_LDO_EN_REG1, REGS_WMSK, REGS_WMSK},
358     {RK816_LDO_EN_REG2, REGS_WMSK, REGS_WMSK},
359 };
360 
361 static const struct rk808_reg_data rk817_pre_init_reg[] = {
362     {RK817_SYS_CFG(3), RK817_SLPPOL_MSK, RK817_SLPPOL_L},
363     {RK817_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
364     {RK817_GPIO_INT_CFG, RK817_INT_POL_MSK, RK817_INT_POL_L},
365     {RK817_SYS_CFG(1), RK817_HOTDIE_TEMP_MSK | RK817_TSD_TEMP_MSK, RK817_HOTDIE_105 | RK817_TSD_140},
366 };
367 
368 static const struct rk808_reg_data rk818_pre_init_reg[] = {
369     /* improve efficiency */
370     {RK818_BUCK2_CONFIG_REG, BUCK2_RATE_MASK, BUCK_ILMIN_250MA},
371     {RK818_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK, BUCK_ILMIN_250MA},
372     {RK818_BOOST_CONFIG_REG, BOOST_ILMIN_MASK, BOOST_ILMIN_100MA},
373     {RK818_USB_CTRL_REG, RK818_USB_ILIM_SEL_MASK, RK818_USB_ILMIN_2000MA},
374     /* close charger when usb lower then 3.4V */
375     {RK818_USB_CTRL_REG, RK818_USB_CHG_SD_VSEL_MASK, (0x7 << 4)},
376     /* no action when vref */
377     {RK818_H5V_EN_REG, BIT(1), RK818_REF_RDY_CTRL},
378     /* enable HDMI 5V */
379     {RK818_H5V_EN_REG, BIT(0), RK818_H5V_EN},
380     {RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
381     {RK808_VB_MON_REG, MASK_ALL, VB_LO_ACT | VB_LO_SEL_3500MV},
382     {RK808_CLK32OUT_REG, CLK32KOUT2_FUNC_MASK, CLK32KOUT2_FUNC},
383 };
384 
385 static const struct regmap_irq rk805_irqs[] = {
386     [RK805_IRQ_PWRON_RISE] =
387         {
388             .mask = RK805_IRQ_PWRON_RISE_MSK,
389             .reg_offset = 0,
390         },
391     [RK805_IRQ_VB_LOW] =
392         {
393             .mask = RK805_IRQ_VB_LOW_MSK,
394             .reg_offset = 0,
395         },
396     [RK805_IRQ_PWRON] =
397         {
398             .mask = RK805_IRQ_PWRON_MSK,
399             .reg_offset = 0,
400         },
401     [RK805_IRQ_PWRON_LP] =
402         {
403             .mask = RK805_IRQ_PWRON_LP_MSK,
404             .reg_offset = 0,
405         },
406     [RK805_IRQ_HOTDIE] =
407         {
408             .mask = RK805_IRQ_HOTDIE_MSK,
409             .reg_offset = 0,
410         },
411     [RK805_IRQ_RTC_ALARM] =
412         {
413             .mask = RK805_IRQ_RTC_ALARM_MSK,
414             .reg_offset = 0,
415         },
416     [RK805_IRQ_RTC_PERIOD] =
417         {
418             .mask = RK805_IRQ_RTC_PERIOD_MSK,
419             .reg_offset = 0,
420         },
421     [RK805_IRQ_PWRON_FALL] =
422         {
423             .mask = RK805_IRQ_PWRON_FALL_MSK,
424             .reg_offset = 0,
425         },
426 };
427 
428 static const struct regmap_irq rk808_irqs[] = {
429     /* INT_STS */
430     [RK808_IRQ_VOUT_LO] =
431         {
432             .mask = RK808_IRQ_VOUT_LO_MSK,
433             .reg_offset = 0,
434         },
435     [RK808_IRQ_VB_LO] =
436         {
437             .mask = RK808_IRQ_VB_LO_MSK,
438             .reg_offset = 0,
439         },
440     [RK808_IRQ_PWRON] =
441         {
442             .mask = RK808_IRQ_PWRON_MSK,
443             .reg_offset = 0,
444         },
445     [RK808_IRQ_PWRON_LP] =
446         {
447             .mask = RK808_IRQ_PWRON_LP_MSK,
448             .reg_offset = 0,
449         },
450     [RK808_IRQ_HOTDIE] =
451         {
452             .mask = RK808_IRQ_HOTDIE_MSK,
453             .reg_offset = 0,
454         },
455     [RK808_IRQ_RTC_ALARM] =
456         {
457             .mask = RK808_IRQ_RTC_ALARM_MSK,
458             .reg_offset = 0,
459         },
460     [RK808_IRQ_RTC_PERIOD] =
461         {
462             .mask = RK808_IRQ_RTC_PERIOD_MSK,
463             .reg_offset = 0,
464         },
465 
466     /* INT_STS2 */
467     [RK808_IRQ_PLUG_IN_INT] =
468         {
469             .mask = RK808_IRQ_PLUG_IN_INT_MSK,
470             .reg_offset = 1,
471         },
472     [RK808_IRQ_PLUG_OUT_INT] =
473         {
474             .mask = RK808_IRQ_PLUG_OUT_INT_MSK,
475             .reg_offset = 1,
476         },
477 };
478 
479 static struct rk808_reg_data rk816_suspend_reg[] = {
480     /* set bat 3.4v low and act irq */
481     {RK816_VB_MON_REG, VBAT_LOW_VOL_MASK | VBAT_LOW_ACT_MASK, RK816_VBAT_LOW_3V4 | EN_VBAT_LOW_IRQ},
482 };
483 
484 static struct rk808_reg_data rk816_resume_reg[] = {
485     /* set bat 3.0v low and act shutdown */
486     {RK816_VB_MON_REG, VBAT_LOW_VOL_MASK | VBAT_LOW_ACT_MASK, RK816_VBAT_LOW_3V0 | EN_VABT_LOW_SHUT_DOWN},
487 };
488 
489 static const struct regmap_irq rk816_irqs[] = {
490     /* INT_STS */
491     [RK816_IRQ_PWRON_FALL] =
492         {
493             .mask = RK816_IRQ_PWRON_FALL_MSK,
494             .reg_offset = 0,
495         },
496     [RK816_IRQ_PWRON_RISE] =
497         {
498             .mask = RK816_IRQ_PWRON_RISE_MSK,
499             .reg_offset = 0,
500         },
501     [RK816_IRQ_VB_LOW] =
502         {
503             .mask = RK816_IRQ_VB_LOW_MSK,
504             .reg_offset = 1,
505         },
506     [RK816_IRQ_PWRON] =
507         {
508             .mask = RK816_IRQ_PWRON_MSK,
509             .reg_offset = 1,
510         },
511     [RK816_IRQ_PWRON_LP] =
512         {
513             .mask = RK816_IRQ_PWRON_LP_MSK,
514             .reg_offset = 1,
515         },
516     [RK816_IRQ_HOTDIE] =
517         {
518             .mask = RK816_IRQ_HOTDIE_MSK,
519             .reg_offset = 1,
520         },
521     [RK816_IRQ_RTC_ALARM] =
522         {
523             .mask = RK816_IRQ_RTC_ALARM_MSK,
524             .reg_offset = 1,
525         },
526     [RK816_IRQ_RTC_PERIOD] =
527         {
528             .mask = RK816_IRQ_RTC_PERIOD_MSK,
529             .reg_offset = 1,
530         },
531     [RK816_IRQ_USB_OV] =
532         {
533             .mask = RK816_IRQ_USB_OV_MSK,
534             .reg_offset = 1,
535         },
536 };
537 
538 static struct rk808_reg_data rk818_suspend_reg[] = {
539     /* set bat 3.4v low and act irq */
540     {RK808_VB_MON_REG, VBAT_LOW_VOL_MASK | VBAT_LOW_ACT_MASK, RK808_VBAT_LOW_3V4 | EN_VBAT_LOW_IRQ},
541 };
542 
543 static struct rk808_reg_data rk818_resume_reg[] = {
544     /* set bat 3.0v low and act shutdown */
545     {RK808_VB_MON_REG, VBAT_LOW_VOL_MASK | VBAT_LOW_ACT_MASK, RK808_VBAT_LOW_3V0 | EN_VABT_LOW_SHUT_DOWN},
546 };
547 
548 static const struct regmap_irq rk818_irqs[] = {
549     /* INT_STS */
550     [RK818_IRQ_VOUT_LO] =
551         {
552             .mask = RK818_IRQ_VOUT_LO_MSK,
553             .reg_offset = 0,
554         },
555     [RK818_IRQ_VB_LO] =
556         {
557             .mask = RK818_IRQ_VB_LO_MSK,
558             .reg_offset = 0,
559         },
560     [RK818_IRQ_PWRON] =
561         {
562             .mask = RK818_IRQ_PWRON_MSK,
563             .reg_offset = 0,
564         },
565     [RK818_IRQ_PWRON_LP] =
566         {
567             .mask = RK818_IRQ_PWRON_LP_MSK,
568             .reg_offset = 0,
569         },
570     [RK818_IRQ_HOTDIE] =
571         {
572             .mask = RK818_IRQ_HOTDIE_MSK,
573             .reg_offset = 0,
574         },
575     [RK818_IRQ_RTC_ALARM] =
576         {
577             .mask = RK818_IRQ_RTC_ALARM_MSK,
578             .reg_offset = 0,
579         },
580     [RK818_IRQ_RTC_PERIOD] =
581         {
582             .mask = RK818_IRQ_RTC_PERIOD_MSK,
583             .reg_offset = 0,
584         },
585     [RK818_IRQ_USB_OV] =
586         {
587             .mask = RK818_IRQ_USB_OV_MSK,
588             .reg_offset = 0,
589         },
590 
591     /* INT_STS2 */
592     [RK818_IRQ_PLUG_IN] =
593         {
594             .mask = RK818_IRQ_PLUG_IN_MSK,
595             .reg_offset = 1,
596         },
597     [RK818_IRQ_PLUG_OUT] =
598         {
599             .mask = RK818_IRQ_PLUG_OUT_MSK,
600             .reg_offset = 1,
601         },
602     [RK818_IRQ_CHG_OK] =
603         {
604             .mask = RK818_IRQ_CHG_OK_MSK,
605             .reg_offset = 1,
606         },
607     [RK818_IRQ_CHG_TE] =
608         {
609             .mask = RK818_IRQ_CHG_TE_MSK,
610             .reg_offset = 1,
611         },
612     [RK818_IRQ_CHG_TS1] =
613         {
614             .mask = RK818_IRQ_CHG_TS1_MSK,
615             .reg_offset = 1,
616         },
617     [RK818_IRQ_TS2] =
618         {
619             .mask = RK818_IRQ_TS2_MSK,
620             .reg_offset = 1,
621         },
622     [RK818_IRQ_CHG_CVTLIM] =
623         {
624             .mask = RK818_IRQ_CHG_CVTLIM_MSK,
625             .reg_offset = 1,
626         },
627     [RK818_IRQ_DISCHG_ILIM] =
628         {
629             .mask = RK818_IRQ_DISCHG_ILIM_MSK,
630             .reg_offset = 1,
631         },
632 };
633 
634 static const struct regmap_irq rk817_irqs[RK817_IRQ_END] = {
635     REGMAP_IRQ_REG_LINE(0, 8),  REGMAP_IRQ_REG_LINE(1, 8),  REGMAP_IRQ_REG_LINE(2, 8),  REGMAP_IRQ_REG_LINE(3, 8),
636     REGMAP_IRQ_REG_LINE(4, 8),  REGMAP_IRQ_REG_LINE(5, 8),  REGMAP_IRQ_REG_LINE(6, 8),  REGMAP_IRQ_REG_LINE(7, 8),
637     REGMAP_IRQ_REG_LINE(8, 8),  REGMAP_IRQ_REG_LINE(9, 8),  REGMAP_IRQ_REG_LINE(10, 8), REGMAP_IRQ_REG_LINE(11, 8),
638     REGMAP_IRQ_REG_LINE(12, 8), REGMAP_IRQ_REG_LINE(13, 8), REGMAP_IRQ_REG_LINE(14, 8), REGMAP_IRQ_REG_LINE(15, 8),
639     REGMAP_IRQ_REG_LINE(16, 8), REGMAP_IRQ_REG_LINE(17, 8), REGMAP_IRQ_REG_LINE(18, 8), REGMAP_IRQ_REG_LINE(19, 8),
640     REGMAP_IRQ_REG_LINE(20, 8), REGMAP_IRQ_REG_LINE(21, 8), REGMAP_IRQ_REG_LINE(22, 8), REGMAP_IRQ_REG_LINE(23, 8)};
641 
642 static struct regmap_irq_chip rk805_irq_chip = {
643     .name = "rk805",
644     .irqs = rk805_irqs,
645     .num_irqs = ARRAY_SIZE(rk805_irqs),
646     .num_regs = 1,
647     .status_base = RK805_INT_STS_REG,
648     .mask_base = RK805_INT_STS_MSK_REG,
649     .ack_base = RK805_INT_STS_REG,
650     .init_ack_masked = true,
651 };
652 
653 static const struct regmap_irq_chip rk808_irq_chip = {
654     .name = "rk808",
655     .irqs = rk808_irqs,
656     .num_irqs = ARRAY_SIZE(rk808_irqs),
657     .num_regs = 2,
658     .irq_reg_stride = 2,
659     .status_base = RK808_INT_STS_REG1,
660     .mask_base = RK808_INT_STS_MSK_REG1,
661     .ack_base = RK808_INT_STS_REG1,
662     .init_ack_masked = true,
663 };
664 
665 static const struct regmap_irq rk816_battery_irqs[] = {
666     /* INT_STS */
667     [RK816_IRQ_PLUG_IN] =
668         {
669             .mask = RK816_IRQ_PLUG_IN_MSK,
670             .reg_offset = 0,
671         },
672     [RK816_IRQ_PLUG_OUT] =
673         {
674             .mask = RK816_IRQ_PLUG_OUT_MSK,
675             .reg_offset = 0,
676         },
677     [RK816_IRQ_CHG_OK] =
678         {
679             .mask = RK816_IRQ_CHG_OK_MSK,
680             .reg_offset = 0,
681         },
682     [RK816_IRQ_CHG_TE] =
683         {
684             .mask = RK816_IRQ_CHG_TE_MSK,
685             .reg_offset = 0,
686         },
687     [RK816_IRQ_CHG_TS] =
688         {
689             .mask = RK816_IRQ_CHG_TS_MSK,
690             .reg_offset = 0,
691         },
692     [RK816_IRQ_CHG_CVTLIM] =
693         {
694             .mask = RK816_IRQ_CHG_CVTLIM_MSK,
695             .reg_offset = 0,
696         },
697     [RK816_IRQ_DISCHG_ILIM] =
698         {
699             .mask = RK816_IRQ_DISCHG_ILIM_MSK,
700             .reg_offset = 0,
701         },
702 };
703 
704 static struct regmap_irq_chip rk816_irq_chip = {
705     .name = "rk816",
706     .irqs = rk816_irqs,
707     .num_irqs = ARRAY_SIZE(rk816_irqs),
708     .num_regs = 2,
709     .irq_reg_stride = 3,
710     .status_base = RK816_INT_STS_REG1,
711     .mask_base = RK816_INT_STS_MSK_REG1,
712     .ack_base = RK816_INT_STS_REG1,
713     .init_ack_masked = true,
714 };
715 
716 static struct regmap_irq_chip rk816_battery_irq_chip = {
717     .name = "rk816_battery",
718     .irqs = rk816_battery_irqs,
719     .num_irqs = ARRAY_SIZE(rk816_battery_irqs),
720     .num_regs = 1,
721     .status_base = RK816_INT_STS_REG3,
722     .mask_base = RK816_INT_STS_MSK_REG3,
723     .ack_base = RK816_INT_STS_REG3,
724     .init_ack_masked = true,
725 };
726 
727 static struct regmap_irq_chip rk817_irq_chip = {
728     .name = "rk817",
729     .irqs = rk817_irqs,
730     .num_irqs = ARRAY_SIZE(rk817_irqs),
731     .num_regs = 3,
732     .irq_reg_stride = 2,
733     .status_base = RK817_INT_STS_REG0,
734     .mask_base = RK817_INT_STS_MSK_REG0,
735     .ack_base = RK817_INT_STS_REG0,
736     .init_ack_masked = true,
737 };
738 
739 static const struct regmap_irq_chip rk818_irq_chip = {
740     .name = "rk818",
741     .irqs = rk818_irqs,
742     .num_irqs = ARRAY_SIZE(rk818_irqs),
743     .num_regs = 2,
744     .irq_reg_stride = 2,
745     .status_base = RK818_INT_STS_REG1,
746     .mask_base = RK818_INT_STS_MSK_REG1,
747     .ack_base = RK818_INT_STS_REG1,
748     .init_ack_masked = true,
749 };
750 
751 static struct i2c_client *rk808_i2c_client;
752 static struct rk808_reg_data *suspend_reg, *resume_reg;
753 static int suspend_reg_num, resume_reg_num;
754 
rk805_device_shutdown_prepare(void)755 static void rk805_device_shutdown_prepare(void)
756 {
757     int ret;
758     struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
759 
760     if (!rk808) {
761         return;
762     }
763 
764     ret = regmap_update_bits(rk808->regmap, RK805_GPIO_IO_POL_REG, SLP_SD_MSK, SHUTDOWN_FUN);
765     if (ret) {
766         dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
767     }
768 }
769 
rk817_shutdown_prepare(void)770 static void rk817_shutdown_prepare(void)
771 {
772     int ret;
773     struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
774 
775     /* close rtc int when power off */
776     regmap_update_bits(rk808->regmap, RK817_INT_STS_MSK_REG0, (RK_REG_VAL_THREE << RK_REG_SHIFT_MASK_FIVE),
777                        (RK_REG_VAL_THREE << RK_REG_SHIFT_MASK_FIVE));
778     regmap_update_bits(rk808->regmap, RK817_RTC_INT_REG, (RK_REG_VAL_THREE << RK_REG_SHIFT_MASK_TWO),
779                        (RK_REG_VAL_ZERO << RK_REG_SHIFT_MASK_TWO));
780 
781     if (rk808->pins && rk808->pins->p && rk808->pins->power_off) {
782         ret =
783             regmap_update_bits(rk808->regmap, RK817_SYS_CFG(RK_REG_VAL_THREE), RK817_SLPPIN_FUNC_MSK, SLPPIN_NULL_FUN);
784         if (ret) {
785             pr_err("shutdown: config SLPPIN_NULL_FUN error!\n");
786         }
787 
788         ret = regmap_update_bits(rk808->regmap, RK817_SYS_CFG(RK_REG_VAL_THREE), RK817_SLPPOL_MSK, RK817_SLPPOL_H);
789         if (ret) {
790             pr_err("shutdown: config RK817_SLPPOL_H error!\n");
791         }
792 
793         ret = pinctrl_select_state(rk808->pins->p, rk808->pins->power_off);
794         if (ret) {
795             pr_info("%s:failed to activate pwroff state\n", __func__);
796         }
797     }
798 
799     /* pmic sleep shutdown function */
800     ret = regmap_update_bits(rk808->regmap, RK817_SYS_CFG(RK_REG_VAL_THREE), RK817_SLPPIN_FUNC_MSK, SLPPIN_DN_FUN);
801     if (ret) {
802         dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
803     }
804     /* pmic need the SCL clock to synchronize register */
805     mdelay(RK_DELAY_COUNT_TWO_MS);
806 }
807 
rk8xx_device_shutdown(void)808 static void rk8xx_device_shutdown(void)
809 {
810     int ret;
811     unsigned int reg, bit;
812     struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
813 
814     switch (rk808->variant) {
815         case RK805_ID:
816             reg = RK805_DEV_CTRL_REG;
817             bit = DEV_OFF;
818             break;
819         case RK808_ID:
820             reg = RK808_DEVCTRL_REG, bit = DEV_OFF_RST;
821             break;
822         case RK816_ID:
823             reg = RK816_DEV_CTRL_REG;
824             bit = DEV_OFF;
825             break;
826         case RK818_ID:
827             reg = RK818_DEVCTRL_REG;
828             bit = DEV_OFF;
829             break;
830         default:
831             return;
832     }
833 
834     ret = regmap_update_bits(rk808->regmap, reg, bit, bit);
835     if (ret) {
836         dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
837     }
838 }
839 
840 /* Called in syscore shutdown */
841 static void (*pm_shutdown)(void);
842 
rk8xx_syscore_shutdown(void)843 static void rk8xx_syscore_shutdown(void)
844 {
845     int ret;
846     struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
847 
848     if (!rk808) {
849         dev_warn(&rk808_i2c_client->dev, "have no rk808, so do nothing here\n");
850         return;
851     }
852 
853     /* close rtc int when power off */
854     regmap_update_bits(rk808->regmap, RK808_INT_STS_MSK_REG1, (RK_REG_VAL_THREE << RK_REG_SHIFT_MASK_FIVE),
855                        (RK_REG_VAL_THREE << RK_REG_SHIFT_MASK_FIVE));
856     regmap_update_bits(rk808->regmap, RK808_RTC_INT_REG, (RK_REG_VAL_THREE << RK_REG_SHIFT_MASK_TWO),
857                        (RK_REG_VAL_ZERO << RK_REG_SHIFT_MASK_TWO));
858     /*
859      * For PMIC that power off supplies by write register via i2c bus,
860      * it's better to do power off at syscore shutdown here.
861      *
862      * Because when run to kernel's "pm_power_off" call, i2c may has
863      * been stopped or PMIC may not be able to get i2c transfer while
864      * there are too many devices are competiting.
865      */
866     if (system_state == SYSTEM_POWER_OFF) {
867         if (rk808->variant == RK809_ID || rk808->variant == RK817_ID) {
868             ret = regmap_update_bits(rk808->regmap, RK817_SYS_CFG(RK_REG_VAL_THREE), RK817_SLPPIN_FUNC_MSK,
869                                      SLPPIN_DN_FUN);
870             if (ret) {
871                 dev_warn(&rk808_i2c_client->dev, "Cannot switch to power down function\n");
872             }
873         }
874 
875         if (pm_shutdown) {
876             dev_info(&rk808_i2c_client->dev, "System power off\n");
877             pm_shutdown();
878             mdelay(RK_DELAY_COUNT_TEN_MS);
879             dev_info(&rk808_i2c_client->dev, "Power off failed !\n");
880             while (1) {
881                 ;
882             }
883         }
884     }
885 }
886 
887 static struct syscore_ops rk808_syscore_ops = {
888     .shutdown = rk8xx_syscore_shutdown,
889 };
890 
891 /*
892  * RK8xx PMICs would do real power off in syscore shutdown, if "pm_power_off"
893  * is not assigned(e.g. PSCI is not enabled), we have to provide a dummy
894  * callback for it, otherwise there comes a halt in Reboot system call:
895  *
896  * if ((cmd == LINUX_REBOOT_CMD_POWER_OFF) && !pm_power_off)
897  *        cmd = LINUX_REBOOT_CMD_HALT;
898  */
rk808_pm_power_off_dummy(void)899 static void rk808_pm_power_off_dummy(void)
900 {
901     pr_info("Dummy power off for RK8xx PMICs, should never reach here!\n");
902 
903     while (1) {
904         ;
905     }
906 }
907 
rk8xx_dbg_store(struct device *dev, struct device_attribute *attr, const char *buf, size_t count)908 static ssize_t rk8xx_dbg_store(struct device *dev, struct device_attribute *attr, const char *buf, size_t count)
909 {
910     int ret;
911     char cmd;
912     u32 input[2], addr, data;
913     struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
914 
915     ret = sscanf(buf, "%c ", &cmd);
916     if (ret != RK_SSCAN_RET_RESULT_ONE) {
917         pr_err("Unknown command\n");
918         goto out;
919     }
920     switch (cmd) {
921         case 'w':
922             ret = sscanf(buf, "%c %x %x ", &cmd, &input[0], &input[1]);
923             if (ret != RK_SSCAN_RET_RESULT_THREE) {
924                 pr_err("error! cmd format: echo w [addr] [value]\n");
925                 goto out;
926             };
927             addr = input[0] & RK_INPUT_MASK;
928             data = input[1] & RK_INPUT_MASK;
929             pr_info("cmd : %c %x %x\n\n", cmd, input[0], input[1]);
930             regmap_write(rk808->regmap, addr, data);
931             regmap_read(rk808->regmap, addr, &data);
932             pr_info("new: %x %x\n", addr, data);
933             break;
934         case 'r':
935             ret = sscanf(buf, "%c %x ", &cmd, &input[0]);
936             if (ret != RK_SSCAN_RET_RESULT_TWO) {
937                 pr_err("error! cmd format: echo r [addr]\n");
938                 goto out;
939             };
940             pr_info("cmd : %c %x\n\n", cmd, input[0]);
941             addr = input[0] & RK_INPUT_MASK;
942             regmap_read(rk808->regmap, addr, &data);
943             pr_info("%x %x\n", input[0], data);
944             break;
945         default:
946             pr_err("Unknown command\n");
947             break;
948     }
949 
950 out:
951     return count;
952 }
953 
rk817_pinctrl_init(struct device *dev, struct rk808 *rk808)954 static int rk817_pinctrl_init(struct device *dev, struct rk808 *rk808)
955 {
956     int ret;
957     struct platform_device *pinctrl_dev;
958     struct pinctrl_state *default_st;
959 
960     pinctrl_dev = platform_device_alloc("rk805-pinctrl", -1);
961     if (!pinctrl_dev) {
962         dev_err(dev, "Alloc pinctrl dev failed!\n");
963         return -ENOMEM;
964     }
965 
966     pinctrl_dev->dev.parent = dev;
967     ret = platform_device_add(pinctrl_dev);
968     if (ret) {
969         platform_device_put(pinctrl_dev);
970         dev_err(dev, "Add rk805-pinctrl dev failed!\n");
971         return ret;
972     }
973     if (dev->pins && !IS_ERR(dev->pins->p)) {
974         dev_info(dev, "had get a pinctrl!\n");
975         return 0;
976     }
977 
978     rk808->pins = devm_kzalloc(dev, sizeof(struct rk808_pin_info), GFP_KERNEL);
979     if (!rk808->pins) {
980         return -ENOMEM;
981     }
982 
983     rk808->pins->p = devm_pinctrl_get(dev);
984     if (IS_ERR(rk808->pins->p)) {
985         rk808->pins->p = NULL;
986         dev_err(dev, "no pinctrl handle\n");
987         return 0;
988     }
989     default_st = pinctrl_lookup_state(rk808->pins->p, PINCTRL_STATE_DEFAULT);
990     if (IS_ERR(default_st)) {
991         dev_dbg(dev, "no default pinctrl state\n");
992         return -EINVAL;
993     }
994 
995     ret = pinctrl_select_state(rk808->pins->p, default_st);
996     if (ret) {
997         dev_dbg(dev, "failed to activate default pinctrl state\n");
998         return -EINVAL;
999     }
1000 
1001     rk808->pins->power_off = pinctrl_lookup_state(rk808->pins->p, "pmic-power-off");
1002     if (IS_ERR(rk808->pins->power_off)) {
1003         rk808->pins->power_off = NULL;
1004         dev_dbg(dev, "no power-off pinctrl state\n");
1005     }
1006 
1007     rk808->pins->sleep = pinctrl_lookup_state(rk808->pins->p, "pmic-sleep");
1008     if (IS_ERR(rk808->pins->sleep)) {
1009         rk808->pins->sleep = NULL;
1010         dev_dbg(dev, "no sleep-setting state\n");
1011     }
1012 
1013     rk808->pins->reset = pinctrl_lookup_state(rk808->pins->p, "pmic-reset");
1014     if (IS_ERR(rk808->pins->reset)) {
1015         rk808->pins->reset = NULL;
1016         dev_dbg(dev, "no reset-setting pinctrl state\n");
1017         return 0;
1018     }
1019     ret = pinctrl_select_state(rk808->pins->p, rk808->pins->reset);
1020     if (ret) {
1021         dev_dbg(dev, "failed to activate reset-setting pinctrl state\n");
1022     }
1023 
1024     return 0;
1025 }
1026 
1027 struct rk817_reboot_data_t {
1028     struct rk808 *rk808;
1029     struct notifier_block reboot_notifier;
1030 };
1031 
1032 static struct rk817_reboot_data_t rk817_reboot_data;
1033 
rk817_reboot_notifier_handler(struct notifier_block *nb, unsigned long action, void *cmd)1034 static int rk817_reboot_notifier_handler(struct notifier_block *nb, unsigned long action, void *cmd)
1035 {
1036     struct rk817_reboot_data_t *data;
1037     struct device *dev;
1038     int value, power_en_active0, power_en_active1;
1039     int ret, i;
1040     static const char *const pmic_rst_reg_only_cmd[] = {
1041         "loader", "bootloader", "fastboot", "recovery", "ums", "panic", "watchdog", "charge",
1042     };
1043 
1044     data = container_of(nb, struct rk817_reboot_data_t, reboot_notifier);
1045     dev = &data->rk808->i2c->dev;
1046 
1047     regmap_read(data->rk808->regmap, RK817_POWER_EN_SAVE0, &power_en_active0);
1048     if (power_en_active0 != 0) {
1049         regmap_read(data->rk808->regmap, RK817_POWER_EN_SAVE1, &power_en_active1);
1050         value = power_en_active0 & RK_BYTE_LOW_FOUR_BIT_MASK;
1051         regmap_write(data->rk808->regmap, RK817_POWER_EN_REG(RK_POWER_EN_REG_VALUE_ZERO),
1052                      value | RK_BYTE_HIGH_FOUR_BIT_MASK);
1053         value = (power_en_active0 & RK_BYTE_HIGH_FOUR_BIT_MASK) >> RK_BYTE_FOUR_BIT_SHIFT_MASK;
1054         regmap_write(data->rk808->regmap, RK817_POWER_EN_REG(RK_POWER_EN_REG_VALUE_ONE),
1055                      value | RK_BYTE_HIGH_FOUR_BIT_MASK);
1056         value = power_en_active1 & RK_BYTE_LOW_FOUR_BIT_MASK;
1057         regmap_write(data->rk808->regmap, RK817_POWER_EN_REG(RK_POWER_EN_REG_VALUE_TWO),
1058                      value | RK_BYTE_HIGH_FOUR_BIT_MASK);
1059         value = (power_en_active1 & RK_BYTE_HIGH_FOUR_BIT_MASK) >> RK_BYTE_FOUR_BIT_SHIFT_MASK;
1060         regmap_write(data->rk808->regmap, RK817_POWER_EN_REG(RK_POWER_EN_REG_VALUE_THREE),
1061                      value | RK_BYTE_HIGH_FOUR_BIT_MASK);
1062     } else {
1063         dev_info(dev, "reboot: not restore POWER_EN\n");
1064     }
1065 
1066     if (action != SYS_RESTART || !cmd) {
1067         return NOTIFY_OK;
1068     }
1069 
1070     /*
1071      * When system restart, there are two rst actions of PMIC sleep if
1072      * board hardware support:
1073      *
1074      *    0b'00: reset the PMIC itself completely.
1075      *    0b'01: reset the 'RST' related register only.
1076      *
1077      * In the case of 0b'00, PMIC reset itself which triggers SoC NPOR-reset
1078      * at the same time, so the command: reboot load/bootload/recovery, etc
1079      * is not effect any more.
1080      *
1081      * Here we check if this reboot cmd is what we expect for 0b'01.
1082      */
1083     for (i = 0; i < ARRAY_SIZE(pmic_rst_reg_only_cmd); i++) {
1084         if (!strcmp(cmd, pmic_rst_reg_only_cmd[i])) {
1085             ret = regmap_update_bits(data->rk808->regmap, RK817_SYS_CFG(RK_REG_VAL_THREE), RK817_RST_FUNC_MSK,
1086                                      RK817_RST_FUNC_REG);
1087             if (ret) {
1088                 dev_err(dev, "reboot: force RK817_RST_FUNC_REG error!\n");
1089             } else {
1090                 dev_info(dev, "reboot: force RK817_RST_FUNC_REG ok!\n");
1091             }
1092             break;
1093         }
1094     }
1095 
1096     return NOTIFY_OK;
1097 }
1098 
rk817_of_property_prepare(struct rk808 *rk808, struct device *dev)1099 static void rk817_of_property_prepare(struct rk808 *rk808, struct device *dev)
1100 {
1101     u32 inner;
1102     int ret, func, msk, val;
1103     struct device_node *np = dev->of_node;
1104 
1105     ret = of_property_read_u32_index(np, "fb-inner-reg-idxs", 0, &inner);
1106     if (!ret && inner == RK817_ID_DCDC3) {
1107         regmap_update_bits(rk808->regmap, RK817_POWER_CONFIG, RK817_BUCK3_FB_RES_MSK, RK817_BUCK3_FB_RES_INTER);
1108     } else {
1109         regmap_update_bits(rk808->regmap, RK817_POWER_CONFIG, RK817_BUCK3_FB_RES_MSK, RK817_BUCK3_FB_RES_EXT);
1110     }
1111     dev_info(dev, "support dcdc3 fb mode:%d, %d\n", ret, inner);
1112 
1113     ret = of_property_read_u32(np, "pmic-reset-func", &func);
1114 
1115     msk = RK817_SLPPIN_FUNC_MSK | RK817_RST_FUNC_MSK;
1116     val = SLPPIN_NULL_FUN;
1117 
1118     if (!ret && func < RK817_RST_FUNC_CNT) {
1119         val |= RK817_RST_FUNC_MSK & (func << RK817_RST_FUNC_SFT);
1120     } else {
1121         val |= RK817_RST_FUNC_REG;
1122     }
1123 
1124     regmap_update_bits(rk808->regmap, RK817_SYS_CFG(RK_REG_VAL_THREE), msk, val);
1125 
1126     dev_info(dev, "support pmic reset mode:%d,%d\n", ret, func);
1127 
1128     rk817_reboot_data.rk808 = rk808;
1129     rk817_reboot_data.reboot_notifier.notifier_call = rk817_reboot_notifier_handler;
1130     ret = register_reboot_notifier(&rk817_reboot_data.reboot_notifier);
1131     if (ret) {
1132         dev_err(dev, "failed to register reboot nb\n");
1133     }
1134 }
1135 
1136 static struct kobject *rk8xx_kobj;
1137 static struct device_attribute rk8xx_attrs = __ATTR(rk8xx_dbg, 0200, NULL, rk8xx_dbg_store);
1138 
1139 static const struct of_device_id rk808_of_match[] = {
1140     {.compatible = "rockchip,rk805"},
1141     {.compatible = "rockchip,rk808"},
1142     {.compatible = "rockchip,rk809"},
1143     {.compatible = "rockchip,rk816"},
1144     {.compatible = "rockchip,rk817"},
1145     {.compatible = "rockchip,rk818"},
1146     {},
1147 };
1148 MODULE_DEVICE_TABLE(of, rk808_of_match);
1149 
rk808_probe(struct i2c_client *client, const struct i2c_device_id *id)1150 static int rk808_probe(struct i2c_client *client, const struct i2c_device_id *id)
1151 {
1152     struct device_node *np = client->dev.of_node;
1153     struct rk808 *rk808;
1154     const struct rk808_reg_data *pre_init_reg;
1155     const struct regmap_irq_chip *battery_irq_chip = NULL;
1156     const struct mfd_cell *cells;
1157     unsigned char pmic_id_msb, pmic_id_lsb;
1158     u8 on_source = 0, off_source = 0;
1159     unsigned int on, off;
1160     int pm_off = 0, msb, lsb;
1161     int nr_pre_init_regs;
1162     int nr_cells;
1163     int ret;
1164     int i;
1165     void (*of_property_prepare_fn)(struct rk808 * rk808, struct device * dev) = NULL;
1166     int (*pinctrl_init)(struct device * dev, struct rk808 * rk808) = NULL;
1167     void (*device_shutdown_fn)(void) = NULL;
1168 
1169     rk808 = devm_kzalloc(&client->dev, sizeof(*rk808), GFP_KERNEL);
1170     if (!rk808) {
1171         return -ENOMEM;
1172     }
1173 
1174     if (of_device_is_compatible(np, "rockchip,rk817") || of_device_is_compatible(np, "rockchip,rk809")) {
1175         pmic_id_msb = RK817_ID_MSB;
1176         pmic_id_lsb = RK817_ID_LSB;
1177     } else {
1178         pmic_id_msb = RK808_ID_MSB;
1179         pmic_id_lsb = RK808_ID_LSB;
1180     }
1181 
1182     /* Read chip variant */
1183     msb = i2c_smbus_read_byte_data(client, pmic_id_msb);
1184     if (msb < 0) {
1185         dev_err(&client->dev, "failed to read the chip id at 0x%x\n", RK808_ID_MSB);
1186         return msb;
1187     }
1188 
1189     lsb = i2c_smbus_read_byte_data(client, pmic_id_lsb);
1190     if (lsb < 0) {
1191         dev_err(&client->dev, "failed to read the chip id at 0x%x\n", RK808_ID_LSB);
1192         return lsb;
1193     }
1194 
1195     rk808->variant = ((msb << RK_REG_SHIFT_MASK_EIGHT) | lsb) & RK8XX_ID_MSK;
1196     dev_info(&client->dev, "chip id: 0x%x\n", (unsigned int)rk808->variant);
1197 
1198     switch (rk808->variant) {
1199         case RK805_ID:
1200             rk808->regmap_cfg = &rk805_regmap_config;
1201             rk808->regmap_irq_chip = &rk805_irq_chip;
1202             pre_init_reg = rk805_pre_init_reg;
1203             nr_pre_init_regs = ARRAY_SIZE(rk805_pre_init_reg);
1204             cells = rk805s;
1205             nr_cells = ARRAY_SIZE(rk805s);
1206             on_source = RK805_ON_SOURCE_REG;
1207             off_source = RK805_OFF_SOURCE_REG;
1208             suspend_reg = rk805_suspend_reg;
1209             suspend_reg_num = ARRAY_SIZE(rk805_suspend_reg);
1210             resume_reg = rk805_resume_reg;
1211             resume_reg_num = ARRAY_SIZE(rk805_resume_reg);
1212             device_shutdown_fn = rk8xx_device_shutdown;
1213             rk808->pm_pwroff_prep_fn = rk805_device_shutdown_prepare;
1214             break;
1215         case RK808_ID:
1216             rk808->regmap_cfg = &rk808_regmap_config;
1217             rk808->regmap_irq_chip = &rk808_irq_chip;
1218             pre_init_reg = rk808_pre_init_reg;
1219             nr_pre_init_regs = ARRAY_SIZE(rk808_pre_init_reg);
1220             cells = rk808s;
1221             nr_cells = ARRAY_SIZE(rk808s);
1222             device_shutdown_fn = rk8xx_device_shutdown;
1223             break;
1224         case RK816_ID:
1225             rk808->regmap_cfg = &rk816_regmap_config;
1226             rk808->regmap_irq_chip = &rk816_irq_chip;
1227             battery_irq_chip = &rk816_battery_irq_chip;
1228             pre_init_reg = rk816_pre_init_reg;
1229             nr_pre_init_regs = ARRAY_SIZE(rk816_pre_init_reg);
1230             cells = rk816s;
1231             nr_cells = ARRAY_SIZE(rk816s);
1232             on_source = RK816_ON_SOURCE_REG;
1233             off_source = RK816_OFF_SOURCE_REG;
1234             suspend_reg = rk816_suspend_reg;
1235             suspend_reg_num = ARRAY_SIZE(rk816_suspend_reg);
1236             resume_reg = rk816_resume_reg;
1237             resume_reg_num = ARRAY_SIZE(rk816_resume_reg);
1238             device_shutdown_fn = rk8xx_device_shutdown;
1239             break;
1240         case RK818_ID:
1241             rk808->regmap_cfg = &rk818_regmap_config;
1242             rk808->regmap_irq_chip = &rk818_irq_chip;
1243             pre_init_reg = rk818_pre_init_reg;
1244             nr_pre_init_regs = ARRAY_SIZE(rk818_pre_init_reg);
1245             cells = rk818s;
1246             nr_cells = ARRAY_SIZE(rk818s);
1247             on_source = RK818_ON_SOURCE_REG;
1248             off_source = RK818_OFF_SOURCE_REG;
1249             suspend_reg = rk818_suspend_reg;
1250             suspend_reg_num = ARRAY_SIZE(rk818_suspend_reg);
1251             resume_reg = rk818_resume_reg;
1252             resume_reg_num = ARRAY_SIZE(rk818_resume_reg);
1253             device_shutdown_fn = rk8xx_device_shutdown;
1254             break;
1255         case RK809_ID:
1256         case RK817_ID:
1257             rk808->regmap_cfg = &rk817_regmap_config;
1258             rk808->regmap_irq_chip = &rk817_irq_chip;
1259             pre_init_reg = rk817_pre_init_reg;
1260             nr_pre_init_regs = ARRAY_SIZE(rk817_pre_init_reg);
1261             cells = rk817s;
1262             nr_cells = ARRAY_SIZE(rk817s);
1263             on_source = RK817_ON_SOURCE_REG;
1264             off_source = RK817_OFF_SOURCE_REG;
1265             rk808->pm_pwroff_prep_fn = rk817_shutdown_prepare;
1266             of_property_prepare_fn = rk817_of_property_prepare;
1267             pinctrl_init = rk817_pinctrl_init;
1268             break;
1269         default:
1270             dev_err(&client->dev, "Unsupported RK8XX ID %lu\n", rk808->variant);
1271             return -EINVAL;
1272     }
1273 
1274     rk808->i2c = client;
1275     rk808_i2c_client = client;
1276     i2c_set_clientdata(client, rk808);
1277 
1278     rk808->regmap = devm_regmap_init_i2c(client, rk808->regmap_cfg);
1279     if (IS_ERR(rk808->regmap)) {
1280         dev_err(&client->dev, "regmap initialization failed\n");
1281         return PTR_ERR(rk808->regmap);
1282     }
1283 
1284     if (on_source && off_source) {
1285         ret = regmap_read(rk808->regmap, on_source, &on);
1286         if (ret) {
1287             dev_err(&client->dev, "read 0x%x failed\n", on_source);
1288             return ret;
1289         }
1290 
1291         ret = regmap_read(rk808->regmap, off_source, &off);
1292         if (ret) {
1293             dev_err(&client->dev, "read 0x%x failed\n", off_source);
1294             return ret;
1295         }
1296 
1297         dev_info(&client->dev, "source: on=0x%02x, off=0x%02x\n", on, off);
1298     }
1299 
1300     if (!client->irq) {
1301         dev_err(&client->dev, "No interrupt support, no core IRQ\n");
1302         return -EINVAL;
1303     }
1304 
1305     if (of_property_prepare_fn) {
1306         of_property_prepare_fn(rk808, &client->dev);
1307     }
1308 
1309     for (i = 0; i < nr_pre_init_regs; i++) {
1310         ret = regmap_update_bits(rk808->regmap, pre_init_reg[i].addr, pre_init_reg[i].mask, pre_init_reg[i].value);
1311         if (ret) {
1312             dev_err(&client->dev, "0x%x write err\n", pre_init_reg[i].addr);
1313             return ret;
1314         }
1315     }
1316 
1317     if (pinctrl_init) {
1318         ret = pinctrl_init(&client->dev, rk808);
1319         if (ret) {
1320             return ret;
1321         }
1322     }
1323 
1324     ret = regmap_add_irq_chip(rk808->regmap, client->irq, IRQF_ONESHOT, -1, rk808->regmap_irq_chip, &rk808->irq_data);
1325     if (ret) {
1326         dev_err(&client->dev, "Failed to add irq_chip %d\n", ret);
1327         return ret;
1328     }
1329 
1330     if (battery_irq_chip) {
1331         ret = regmap_add_irq_chip(rk808->regmap, client->irq, IRQF_ONESHOT | IRQF_SHARED, -1, battery_irq_chip,
1332                                   &rk808->battery_irq_data);
1333         if (ret) {
1334             dev_err(&client->dev, "Failed to add batterry irq_chip %d\n", ret);
1335             regmap_del_irq_chip(client->irq, rk808->irq_data);
1336             return ret;
1337         }
1338     }
1339 
1340     ret = devm_mfd_add_devices(&client->dev, PLATFORM_DEVID_NONE, cells, nr_cells, NULL, 0,
1341                                regmap_irq_get_domain(rk808->irq_data));
1342     if (ret) {
1343         dev_err(&client->dev, "failed to add MFD devices %d\n", ret);
1344         goto err_irq;
1345     }
1346 
1347     pm_off = of_property_read_bool(np, "rockchip,system-power-controller");
1348     if (pm_off) {
1349         if (!pm_power_off_prepare) {
1350             pm_power_off_prepare = rk808->pm_pwroff_prep_fn;
1351         }
1352 
1353         if (device_shutdown_fn) {
1354             register_syscore_ops(&rk808_syscore_ops);
1355             /* power off system in the syscore shutdown ! */
1356             pm_shutdown = device_shutdown_fn;
1357         }
1358     }
1359 
1360     rk8xx_kobj = kobject_create_and_add("rk8xx", NULL);
1361     if (rk8xx_kobj) {
1362         ret = sysfs_create_file(rk8xx_kobj, &rk8xx_attrs.attr);
1363         if (ret) {
1364             dev_err(&client->dev, "create rk8xx sysfs error\n");
1365         }
1366     }
1367 
1368     if (!pm_power_off) {
1369         pm_power_off = rk808_pm_power_off_dummy;
1370     }
1371 
1372     return 0;
1373 
1374 err_irq:
1375     regmap_del_irq_chip(client->irq, rk808->irq_data);
1376     if (battery_irq_chip) {
1377         regmap_del_irq_chip(client->irq, rk808->battery_irq_data);
1378     }
1379     return ret;
1380 }
1381 
rk808_remove(struct i2c_client *client)1382 static int rk808_remove(struct i2c_client *client)
1383 {
1384     struct rk808 *rk808 = i2c_get_clientdata(client);
1385 
1386     regmap_del_irq_chip(client->irq, rk808->irq_data);
1387     mfd_remove_devices(&client->dev);
1388 
1389     /**
1390      * pm_power_off may points to a function from another module.
1391      * Check if the pointer is set by us and only then overwrite it.
1392      */
1393     if (pm_power_off == rk808_pm_power_off_dummy) {
1394         pm_power_off = NULL;
1395     }
1396 
1397     /**
1398      * As above, check if the pointer is set by us before overwrite.
1399      */
1400     if (rk808->pm_pwroff_prep_fn && pm_power_off_prepare == rk808->pm_pwroff_prep_fn) {
1401         pm_power_off_prepare = NULL;
1402     }
1403 
1404     if (pm_shutdown) {
1405         unregister_syscore_ops(&rk808_syscore_ops);
1406     }
1407 
1408     return 0;
1409 }
1410 
rk8xx_suspend(struct device *dev)1411 static int __maybe_unused rk8xx_suspend(struct device *dev)
1412 {
1413     struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
1414     int i, ret = 0;
1415     int value;
1416 
1417     for (i = 0; i < suspend_reg_num; i++) {
1418         ret = regmap_update_bits(rk808->regmap, suspend_reg[i].addr, suspend_reg[i].mask, suspend_reg[i].value);
1419         if (ret) {
1420             dev_err(dev, "0x%x write err\n", suspend_reg[i].addr);
1421             return ret;
1422         }
1423     }
1424 
1425     switch (rk808->variant) {
1426         case RK805_ID:
1427             ret = regmap_update_bits(rk808->regmap, RK805_GPIO_IO_POL_REG, SLP_SD_MSK, SLEEP_FUN);
1428             break;
1429         case RK809_ID:
1430         case RK817_ID:
1431             if (rk808->pins && rk808->pins->p && rk808->pins->sleep) {
1432                 ret = regmap_update_bits(rk808->regmap, RK817_SYS_CFG(3), RK817_SLPPIN_FUNC_MSK, SLPPIN_NULL_FUN);
1433                 if (ret) {
1434                     dev_err(dev, "suspend: config SLPPIN_NULL_FUN error!\n");
1435                     return ret;
1436                 }
1437 
1438                 ret = regmap_update_bits(rk808->regmap, RK817_SYS_CFG(3), RK817_SLPPOL_MSK, RK817_SLPPOL_H);
1439                 if (ret) {
1440                     dev_err(dev, "suspend: config RK817_SLPPOL_H error!\n");
1441                     return ret;
1442                 }
1443 
1444                 /* pmic need the SCL clock to synchronize register */
1445                 regmap_read(rk808->regmap, RK817_SYS_STS, &value);
1446                 mdelay(2);
1447                 ret = pinctrl_select_state(rk808->pins->p, rk808->pins->sleep);
1448                 if (ret) {
1449                     dev_err(dev, "failed to act slp pinctrl state\n");
1450                     return ret;
1451                 }
1452             }
1453             break;
1454         default:
1455             break;
1456     }
1457 
1458     return ret;
1459 }
1460 
rk8xx_resume(struct device *dev)1461 static int __maybe_unused rk8xx_resume(struct device *dev)
1462 {
1463     struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
1464     int i, ret = 0;
1465     int value;
1466 
1467     for (i = 0; i < resume_reg_num; i++) {
1468         ret = regmap_update_bits(rk808->regmap, resume_reg[i].addr, resume_reg[i].mask, resume_reg[i].value);
1469         if (ret) {
1470             dev_err(dev, "0x%x write err\n", resume_reg[i].addr);
1471             return ret;
1472         }
1473     }
1474 
1475     switch (rk808->variant) {
1476         case RK809_ID:
1477         case RK817_ID:
1478             if (rk808->pins && rk808->pins->p && rk808->pins->reset) {
1479                 ret = regmap_update_bits(rk808->regmap, RK817_SYS_CFG(3), RK817_SLPPIN_FUNC_MSK, SLPPIN_NULL_FUN);
1480                 if (ret) {
1481                     dev_err(dev, "resume: config SLPPIN_NULL_FUN error!\n");
1482                     return ret;
1483                 }
1484 
1485                 ret = regmap_update_bits(rk808->regmap, RK817_SYS_CFG(3), RK817_SLPPOL_MSK, RK817_SLPPOL_L);
1486                 if (ret) {
1487                     dev_err(dev, "resume: config RK817_SLPPOL_L error!\n");
1488                     return ret;
1489                 }
1490 
1491                 /* pmic need the SCL clock to synchronize register */
1492                 regmap_read(rk808->regmap, RK817_SYS_STS, &value);
1493                 mdelay(2);
1494                 ret = pinctrl_select_state(rk808->pins->p, rk808->pins->reset);
1495                 if (ret) {
1496                     dev_dbg(dev, "failed to act reset pinctrl state\n");
1497                 }
1498             }
1499             break;
1500         default:
1501             break;
1502     }
1503 
1504     return ret;
1505 }
1506 SIMPLE_DEV_PM_OPS(rk8xx_pm_ops, rk8xx_suspend, rk8xx_resume);
1507 
1508 static struct i2c_driver rk808_i2c_driver = {
1509     .driver =
1510         {
1511             .name = "rk808",
1512             .of_match_table = rk808_of_match,
1513             .pm = &rk8xx_pm_ops,
1514         },
1515     .probe = rk808_probe,
1516     .remove = rk808_remove,
1517 };
1518 
1519 #ifdef CONFIG_ROCKCHIP_THUNDER_BOOT
rk808_i2c_driver_init(void)1520 static int __init rk808_i2c_driver_init(void)
1521 {
1522     return i2c_add_driver(&rk808_i2c_driver);
1523 }
1524 subsys_initcall(rk808_i2c_driver_init);
1525 
rk808_i2c_driver_exit(void)1526 static void __exit rk808_i2c_driver_exit(void)
1527 {
1528     i2c_del_driver(&rk808_i2c_driver);
1529 }
1530 module_exit(rk808_i2c_driver_exit);
1531 #else
1532 module_i2c_driver(rk808_i2c_driver);
1533 #endif
1534 
1535 MODULE_LICENSE("GPL");
1536 MODULE_AUTHOR("Chris Zhong <zyw@rock-chips.com>");
1537 MODULE_AUTHOR("Zhang Qing <zhangqing@rock-chips.com>");
1538 MODULE_AUTHOR("Wadim Egorov <w.egorov@phytec.de>");
1539 MODULE_DESCRIPTION("RK808/RK818 PMIC driver");
1540