1// SPDX-License-Identifier: GPL-2.0-only
2/* Copyright (c) 2010-2011, Code Aurora Forum. All rights reserved.
3 */
4
5#include <linux/module.h>
6#include <linux/kernel.h>
7#include <linux/errno.h>
8#include <linux/slab.h>
9#include <linux/input.h>
10#include <linux/interrupt.h>
11#include <linux/platform_device.h>
12#include <linux/regmap.h>
13#include <linux/log2.h>
14#include <linux/of.h>
15
16#define PON_CNTL_1 0x1C
17#define PON_CNTL_PULL_UP BIT(7)
18#define PON_CNTL_TRIG_DELAY_MASK (0x7)
19#define PON_CNTL_1_PULL_UP_EN			0xe0
20#define PON_CNTL_1_USB_PWR_EN			0x10
21#define PON_CNTL_1_WD_EN_RESET			0x08
22
23#define PM8058_SLEEP_CTRL			0x02b
24#define PM8921_SLEEP_CTRL			0x10a
25
26#define SLEEP_CTRL_SMPL_EN_RESET		0x04
27
28/* Regulator master enable addresses */
29#define REG_PM8058_VREG_EN_MSM			0x018
30#define REG_PM8058_VREG_EN_GRP_5_4		0x1c8
31
32/* Regulator control registers for shutdown/reset */
33#define PM8058_S0_CTRL				0x004
34#define PM8058_S1_CTRL				0x005
35#define PM8058_S3_CTRL				0x111
36#define PM8058_L21_CTRL				0x120
37#define PM8058_L22_CTRL				0x121
38
39#define PM8058_REGULATOR_ENABLE_MASK		0x80
40#define PM8058_REGULATOR_ENABLE			0x80
41#define PM8058_REGULATOR_DISABLE		0x00
42#define PM8058_REGULATOR_PULL_DOWN_MASK		0x40
43#define PM8058_REGULATOR_PULL_DOWN_EN		0x40
44
45/* Buck CTRL register */
46#define PM8058_SMPS_LEGACY_VREF_SEL		0x20
47#define PM8058_SMPS_LEGACY_VPROG_MASK		0x1f
48#define PM8058_SMPS_ADVANCED_BAND_MASK		0xC0
49#define PM8058_SMPS_ADVANCED_BAND_SHIFT		6
50#define PM8058_SMPS_ADVANCED_VPROG_MASK		0x3f
51
52/* Buck TEST2 registers for shutdown/reset */
53#define PM8058_S0_TEST2				0x084
54#define PM8058_S1_TEST2				0x085
55#define PM8058_S3_TEST2				0x11a
56
57#define PM8058_REGULATOR_BANK_WRITE		0x80
58#define PM8058_REGULATOR_BANK_MASK		0x70
59#define PM8058_REGULATOR_BANK_SHIFT		4
60#define PM8058_REGULATOR_BANK_SEL(n)	((n) << PM8058_REGULATOR_BANK_SHIFT)
61
62/* Buck TEST2 register bank 1 */
63#define PM8058_SMPS_LEGACY_VLOW_SEL		0x01
64
65/* Buck TEST2 register bank 7 */
66#define PM8058_SMPS_ADVANCED_MODE_MASK		0x02
67#define PM8058_SMPS_ADVANCED_MODE		0x02
68#define PM8058_SMPS_LEGACY_MODE			0x00
69
70/**
71 * struct pmic8xxx_pwrkey - pmic8xxx pwrkey information
72 * @key_press_irq: key press irq number
73 * @regmap: device regmap
74 * @shutdown_fn: shutdown configuration function
75 */
76struct pmic8xxx_pwrkey {
77	int key_press_irq;
78	struct regmap *regmap;
79	int (*shutdown_fn)(struct pmic8xxx_pwrkey *, bool);
80};
81
82static irqreturn_t pwrkey_press_irq(int irq, void *_pwr)
83{
84	struct input_dev *pwr = _pwr;
85
86	input_report_key(pwr, KEY_POWER, 1);
87	input_sync(pwr);
88
89	return IRQ_HANDLED;
90}
91
92static irqreturn_t pwrkey_release_irq(int irq, void *_pwr)
93{
94	struct input_dev *pwr = _pwr;
95
96	input_report_key(pwr, KEY_POWER, 0);
97	input_sync(pwr);
98
99	return IRQ_HANDLED;
100}
101
102static int pmic8xxx_pwrkey_suspend(struct device *dev)
103{
104	struct pmic8xxx_pwrkey *pwrkey = dev_get_drvdata(dev);
105
106	if (device_may_wakeup(dev))
107		enable_irq_wake(pwrkey->key_press_irq);
108
109	return 0;
110}
111
112static int pmic8xxx_pwrkey_resume(struct device *dev)
113{
114	struct pmic8xxx_pwrkey *pwrkey = dev_get_drvdata(dev);
115
116	if (device_may_wakeup(dev))
117		disable_irq_wake(pwrkey->key_press_irq);
118
119	return 0;
120}
121
122static DEFINE_SIMPLE_DEV_PM_OPS(pm8xxx_pwr_key_pm_ops,
123		pmic8xxx_pwrkey_suspend, pmic8xxx_pwrkey_resume);
124
125static void pmic8xxx_pwrkey_shutdown(struct platform_device *pdev)
126{
127	struct pmic8xxx_pwrkey *pwrkey = platform_get_drvdata(pdev);
128	int error;
129	u8 mask, val;
130	bool reset = system_state == SYSTEM_RESTART;
131
132	if (pwrkey->shutdown_fn) {
133		error = pwrkey->shutdown_fn(pwrkey, reset);
134		if (error)
135			return;
136	}
137
138	/*
139	 * Select action to perform (reset or shutdown) when PS_HOLD goes low.
140	 * Also ensure that KPD, CBL0, and CBL1 pull ups are enabled and that
141	 * USB charging is enabled.
142	 */
143	mask = PON_CNTL_1_PULL_UP_EN | PON_CNTL_1_USB_PWR_EN;
144	mask |= PON_CNTL_1_WD_EN_RESET;
145	val = mask;
146	if (!reset)
147		val &= ~PON_CNTL_1_WD_EN_RESET;
148
149	regmap_update_bits(pwrkey->regmap, PON_CNTL_1, mask, val);
150}
151
152/*
153 * Set an SMPS regulator to be disabled in its CTRL register, but enabled
154 * in the master enable register.  Also set it's pull down enable bit.
155 * Take care to make sure that the output voltage doesn't change if switching
156 * from advanced mode to legacy mode.
157 */
158static int pm8058_disable_smps_locally_set_pull_down(struct regmap *regmap,
159	u16 ctrl_addr, u16 test2_addr, u16 master_enable_addr,
160	u8 master_enable_bit)
161{
162	int error;
163	u8 vref_sel, vlow_sel, band, vprog, bank;
164	unsigned int reg;
165
166	bank = PM8058_REGULATOR_BANK_SEL(7);
167	error = regmap_write(regmap, test2_addr, bank);
168	if (error)
169		return error;
170
171	error = regmap_read(regmap, test2_addr, &reg);
172	if (error)
173		return error;
174
175	reg &= PM8058_SMPS_ADVANCED_MODE_MASK;
176	/* Check if in advanced mode. */
177	if (reg == PM8058_SMPS_ADVANCED_MODE) {
178		/* Determine current output voltage. */
179		error = regmap_read(regmap, ctrl_addr, &reg);
180		if (error)
181			return error;
182
183		band = reg & PM8058_SMPS_ADVANCED_BAND_MASK;
184		band >>= PM8058_SMPS_ADVANCED_BAND_SHIFT;
185		switch (band) {
186		case 3:
187			vref_sel = 0;
188			vlow_sel = 0;
189			break;
190		case 2:
191			vref_sel = PM8058_SMPS_LEGACY_VREF_SEL;
192			vlow_sel = 0;
193			break;
194		case 1:
195			vref_sel = PM8058_SMPS_LEGACY_VREF_SEL;
196			vlow_sel = PM8058_SMPS_LEGACY_VLOW_SEL;
197			break;
198		default:
199			pr_err("%s: regulator already disabled\n", __func__);
200			return -EPERM;
201		}
202		vprog = reg & PM8058_SMPS_ADVANCED_VPROG_MASK;
203		/* Round up if fine step is in use. */
204		vprog = (vprog + 1) >> 1;
205		if (vprog > PM8058_SMPS_LEGACY_VPROG_MASK)
206			vprog = PM8058_SMPS_LEGACY_VPROG_MASK;
207
208		/* Set VLOW_SEL bit. */
209		bank = PM8058_REGULATOR_BANK_SEL(1);
210		error = regmap_write(regmap, test2_addr, bank);
211		if (error)
212			return error;
213
214		error = regmap_update_bits(regmap, test2_addr,
215			PM8058_REGULATOR_BANK_WRITE | PM8058_REGULATOR_BANK_MASK
216				| PM8058_SMPS_LEGACY_VLOW_SEL,
217			PM8058_REGULATOR_BANK_WRITE |
218			PM8058_REGULATOR_BANK_SEL(1) | vlow_sel);
219		if (error)
220			return error;
221
222		/* Switch to legacy mode */
223		bank = PM8058_REGULATOR_BANK_SEL(7);
224		error = regmap_write(regmap, test2_addr, bank);
225		if (error)
226			return error;
227
228		error = regmap_update_bits(regmap, test2_addr,
229				PM8058_REGULATOR_BANK_WRITE |
230				PM8058_REGULATOR_BANK_MASK |
231				PM8058_SMPS_ADVANCED_MODE_MASK,
232				PM8058_REGULATOR_BANK_WRITE |
233				PM8058_REGULATOR_BANK_SEL(7) |
234				PM8058_SMPS_LEGACY_MODE);
235		if (error)
236			return error;
237
238		/* Enable locally, enable pull down, keep voltage the same. */
239		error = regmap_update_bits(regmap, ctrl_addr,
240			PM8058_REGULATOR_ENABLE_MASK |
241			PM8058_REGULATOR_PULL_DOWN_MASK |
242			PM8058_SMPS_LEGACY_VREF_SEL |
243			PM8058_SMPS_LEGACY_VPROG_MASK,
244			PM8058_REGULATOR_ENABLE | PM8058_REGULATOR_PULL_DOWN_EN
245				| vref_sel | vprog);
246		if (error)
247			return error;
248	}
249
250	/* Enable in master control register. */
251	error = regmap_update_bits(regmap, master_enable_addr,
252			master_enable_bit, master_enable_bit);
253	if (error)
254		return error;
255
256	/* Disable locally and enable pull down. */
257	return regmap_update_bits(regmap, ctrl_addr,
258		PM8058_REGULATOR_ENABLE_MASK | PM8058_REGULATOR_PULL_DOWN_MASK,
259		PM8058_REGULATOR_DISABLE | PM8058_REGULATOR_PULL_DOWN_EN);
260}
261
262static int pm8058_disable_ldo_locally_set_pull_down(struct regmap *regmap,
263		u16 ctrl_addr, u16 master_enable_addr, u8 master_enable_bit)
264{
265	int error;
266
267	/* Enable LDO in master control register. */
268	error = regmap_update_bits(regmap, master_enable_addr,
269			master_enable_bit, master_enable_bit);
270	if (error)
271		return error;
272
273	/* Disable LDO in CTRL register and set pull down */
274	return regmap_update_bits(regmap, ctrl_addr,
275		PM8058_REGULATOR_ENABLE_MASK | PM8058_REGULATOR_PULL_DOWN_MASK,
276		PM8058_REGULATOR_DISABLE | PM8058_REGULATOR_PULL_DOWN_EN);
277}
278
279static int pm8058_pwrkey_shutdown(struct pmic8xxx_pwrkey *pwrkey, bool reset)
280{
281	int error;
282	struct regmap *regmap = pwrkey->regmap;
283	u8 mask, val;
284
285	/* When shutting down, enable active pulldowns on important rails. */
286	if (!reset) {
287		/* Disable SMPS's 0,1,3 locally and set pulldown enable bits. */
288		pm8058_disable_smps_locally_set_pull_down(regmap,
289			PM8058_S0_CTRL, PM8058_S0_TEST2,
290			REG_PM8058_VREG_EN_MSM, BIT(7));
291		pm8058_disable_smps_locally_set_pull_down(regmap,
292			PM8058_S1_CTRL, PM8058_S1_TEST2,
293			REG_PM8058_VREG_EN_MSM, BIT(6));
294		pm8058_disable_smps_locally_set_pull_down(regmap,
295			PM8058_S3_CTRL, PM8058_S3_TEST2,
296			REG_PM8058_VREG_EN_GRP_5_4, BIT(7) | BIT(4));
297		/* Disable LDO 21 locally and set pulldown enable bit. */
298		pm8058_disable_ldo_locally_set_pull_down(regmap,
299			PM8058_L21_CTRL, REG_PM8058_VREG_EN_GRP_5_4,
300			BIT(1));
301	}
302
303	/*
304	 * Fix-up: Set regulator LDO22 to 1.225 V in high power mode. Leave its
305	 * pull-down state intact. This ensures a safe shutdown.
306	 */
307	error = regmap_update_bits(regmap, PM8058_L22_CTRL, 0xbf, 0x93);
308	if (error)
309		return error;
310
311	/* Enable SMPL if resetting is desired */
312	mask = SLEEP_CTRL_SMPL_EN_RESET;
313	val = 0;
314	if (reset)
315		val = mask;
316	return regmap_update_bits(regmap, PM8058_SLEEP_CTRL, mask, val);
317}
318
319static int pm8921_pwrkey_shutdown(struct pmic8xxx_pwrkey *pwrkey, bool reset)
320{
321	struct regmap *regmap = pwrkey->regmap;
322	u8 mask = SLEEP_CTRL_SMPL_EN_RESET;
323	u8 val = 0;
324
325	/* Enable SMPL if resetting is desired */
326	if (reset)
327		val = mask;
328	return regmap_update_bits(regmap, PM8921_SLEEP_CTRL, mask, val);
329}
330
331static int pmic8xxx_pwrkey_probe(struct platform_device *pdev)
332{
333	struct input_dev *pwr;
334	int key_release_irq = platform_get_irq(pdev, 0);
335	int key_press_irq = platform_get_irq(pdev, 1);
336	int err;
337	unsigned int delay;
338	unsigned int pon_cntl;
339	struct regmap *regmap;
340	struct pmic8xxx_pwrkey *pwrkey;
341	u32 kpd_delay;
342	bool pull_up;
343
344	if (of_property_read_u32(pdev->dev.of_node, "debounce", &kpd_delay))
345		kpd_delay = 15625;
346
347	/* Valid range of pwr key trigger delay is 1/64 sec to 2 seconds. */
348	if (kpd_delay > USEC_PER_SEC * 2 || kpd_delay < USEC_PER_SEC / 64) {
349		dev_err(&pdev->dev, "invalid power key trigger delay\n");
350		return -EINVAL;
351	}
352
353	pull_up = of_property_read_bool(pdev->dev.of_node, "pull-up");
354
355	regmap = dev_get_regmap(pdev->dev.parent, NULL);
356	if (!regmap) {
357		dev_err(&pdev->dev, "failed to locate regmap for the device\n");
358		return -ENODEV;
359	}
360
361	pwrkey = devm_kzalloc(&pdev->dev, sizeof(*pwrkey), GFP_KERNEL);
362	if (!pwrkey)
363		return -ENOMEM;
364
365	pwrkey->shutdown_fn = of_device_get_match_data(&pdev->dev);
366	pwrkey->regmap = regmap;
367	pwrkey->key_press_irq = key_press_irq;
368
369	pwr = devm_input_allocate_device(&pdev->dev);
370	if (!pwr) {
371		dev_dbg(&pdev->dev, "Can't allocate power button\n");
372		return -ENOMEM;
373	}
374
375	input_set_capability(pwr, EV_KEY, KEY_POWER);
376
377	pwr->name = "pmic8xxx_pwrkey";
378	pwr->phys = "pmic8xxx_pwrkey/input0";
379
380	delay = (kpd_delay << 6) / USEC_PER_SEC;
381	delay = ilog2(delay);
382
383	err = regmap_read(regmap, PON_CNTL_1, &pon_cntl);
384	if (err < 0) {
385		dev_err(&pdev->dev, "failed reading PON_CNTL_1 err=%d\n", err);
386		return err;
387	}
388
389	pon_cntl &= ~PON_CNTL_TRIG_DELAY_MASK;
390	pon_cntl |= (delay & PON_CNTL_TRIG_DELAY_MASK);
391	if (pull_up)
392		pon_cntl |= PON_CNTL_PULL_UP;
393	else
394		pon_cntl &= ~PON_CNTL_PULL_UP;
395
396	err = regmap_write(regmap, PON_CNTL_1, pon_cntl);
397	if (err < 0) {
398		dev_err(&pdev->dev, "failed writing PON_CNTL_1 err=%d\n", err);
399		return err;
400	}
401
402	err = devm_request_irq(&pdev->dev, key_press_irq, pwrkey_press_irq,
403			       IRQF_TRIGGER_RISING,
404			       "pmic8xxx_pwrkey_press", pwr);
405	if (err) {
406		dev_err(&pdev->dev, "Can't get %d IRQ for pwrkey: %d\n",
407			key_press_irq, err);
408		return err;
409	}
410
411	err = devm_request_irq(&pdev->dev, key_release_irq, pwrkey_release_irq,
412			       IRQF_TRIGGER_RISING,
413			       "pmic8xxx_pwrkey_release", pwr);
414	if (err) {
415		dev_err(&pdev->dev, "Can't get %d IRQ for pwrkey: %d\n",
416			key_release_irq, err);
417		return err;
418	}
419
420	err = input_register_device(pwr);
421	if (err) {
422		dev_err(&pdev->dev, "Can't register power key: %d\n", err);
423		return err;
424	}
425
426	platform_set_drvdata(pdev, pwrkey);
427	device_init_wakeup(&pdev->dev, 1);
428
429	return 0;
430}
431
432static const struct of_device_id pm8xxx_pwr_key_id_table[] = {
433	{ .compatible = "qcom,pm8058-pwrkey", .data = &pm8058_pwrkey_shutdown },
434	{ .compatible = "qcom,pm8921-pwrkey", .data = &pm8921_pwrkey_shutdown },
435	{ }
436};
437MODULE_DEVICE_TABLE(of, pm8xxx_pwr_key_id_table);
438
439static struct platform_driver pmic8xxx_pwrkey_driver = {
440	.probe		= pmic8xxx_pwrkey_probe,
441	.shutdown	= pmic8xxx_pwrkey_shutdown,
442	.driver		= {
443		.name	= "pm8xxx-pwrkey",
444		.pm	= pm_sleep_ptr(&pm8xxx_pwr_key_pm_ops),
445		.of_match_table = pm8xxx_pwr_key_id_table,
446	},
447};
448module_platform_driver(pmic8xxx_pwrkey_driver);
449
450MODULE_ALIAS("platform:pmic8xxx_pwrkey");
451MODULE_DESCRIPTION("PMIC8XXX Power Key driver");
452MODULE_LICENSE("GPL v2");
453MODULE_AUTHOR("Trilok Soni <tsoni@codeaurora.org>");
454