[linux-block.git] / drivers / power / reset / atc260x-poweroff.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Poweroff & reset driver for Actions Semi ATC260x PMICs
 *
 * Copyright (c) 2020 Cristian Ciocaltea <cristian.ciocaltea@gmail.com>
 */

#include <linux/delay.h>
#include <linux/mfd/atc260x/core.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/power_supply.h>
#include <linux/reboot.h>
#include <linux/regmap.h>

struct atc260x_pwrc {
	struct device *dev;
	struct regmap *regmap;
	struct notifier_block restart_nb;
	int (*do_poweroff)(const struct atc260x_pwrc *pwrc, bool restart);
};

/* Global variable needed only for pm_power_off */
static struct atc260x_pwrc *atc260x_pwrc_data;

static int atc2603c_do_poweroff(const struct atc260x_pwrc *pwrc, bool restart)
{
	int ret, deep_sleep = 0;
	uint reg_mask, reg_val;

	/* S4-Deep Sleep Mode is NOT available for WALL/USB power */
	if (!restart && !power_supply_is_system_supplied()) {
		deep_sleep = 1;
		dev_info(pwrc->dev, "Enabling S4-Deep Sleep Mode");
	}

	/* Update wakeup sources */
	reg_val = ATC2603C_PMU_SYS_CTL0_ONOFF_LONG_WK_EN |
		  (restart ? ATC2603C_PMU_SYS_CTL0_RESET_WK_EN
			   : ATC2603C_PMU_SYS_CTL0_ONOFF_SHORT_WK_EN);

	ret = regmap_update_bits(pwrc->regmap, ATC2603C_PMU_SYS_CTL0,
				 ATC2603C_PMU_SYS_CTL0_WK_ALL, reg_val);
	if (ret)
		dev_warn(pwrc->dev, "failed to write SYS_CTL0: %d\n", ret);

	/* Update power mode */
	reg_mask = ATC2603C_PMU_SYS_CTL3_EN_S2 | ATC2603C_PMU_SYS_CTL3_EN_S3;

	ret = regmap_update_bits(pwrc->regmap, ATC2603C_PMU_SYS_CTL3, reg_mask,
				 deep_sleep ? 0 : ATC2603C_PMU_SYS_CTL3_EN_S3);
	if (ret) {
		dev_err(pwrc->dev, "failed to write SYS_CTL3: %d\n", ret);
		return ret;
	}

	/* Trigger poweroff / restart sequence */
	reg_mask = restart ? ATC2603C_PMU_SYS_CTL0_RESTART_EN
			   : ATC2603C_PMU_SYS_CTL1_EN_S1;
	reg_val = restart ? ATC2603C_PMU_SYS_CTL0_RESTART_EN : 0;

	ret = regmap_update_bits(pwrc->regmap,
				 restart ? ATC2603C_PMU_SYS_CTL0 : ATC2603C_PMU_SYS_CTL1,
				 reg_mask, reg_val);
	if (ret) {
		dev_err(pwrc->dev, "failed to write SYS_CTL%d: %d\n",
			restart ? 0 : 1, ret);
		return ret;
	}

	/* Wait for trigger completion */
	mdelay(200);

	return 0;
}

static int atc2609a_do_poweroff(const struct atc260x_pwrc *pwrc, bool restart)
{
	int ret, deep_sleep = 0;
	uint reg_mask, reg_val;

	/* S4-Deep Sleep Mode is NOT available for WALL/USB power */
	if (!restart && !power_supply_is_system_supplied()) {
		deep_sleep = 1;
		dev_info(pwrc->dev, "Enabling S4-Deep Sleep Mode");
	}

	/* Update wakeup sources */
	reg_val = ATC2609A_PMU_SYS_CTL0_ONOFF_LONG_WK_EN |
		  (restart ? ATC2609A_PMU_SYS_CTL0_RESET_WK_EN
			   : ATC2609A_PMU_SYS_CTL0_ONOFF_SHORT_WK_EN);

	ret = regmap_update_bits(pwrc->regmap, ATC2609A_PMU_SYS_CTL0,
				 ATC2609A_PMU_SYS_CTL0_WK_ALL, reg_val);
	if (ret)
		dev_warn(pwrc->dev, "failed to write SYS_CTL0: %d\n", ret);

	/* Update power mode */
	reg_mask = ATC2609A_PMU_SYS_CTL3_EN_S2 | ATC2609A_PMU_SYS_CTL3_EN_S3;

	ret = regmap_update_bits(pwrc->regmap, ATC2609A_PMU_SYS_CTL3, reg_mask,
				 deep_sleep ? 0 : ATC2609A_PMU_SYS_CTL3_EN_S3);
	if (ret) {
		dev_err(pwrc->dev, "failed to write SYS_CTL3: %d\n", ret);
		return ret;
	}

	/* Trigger poweroff / restart sequence */
	reg_mask = restart ? ATC2609A_PMU_SYS_CTL0_RESTART_EN
			   : ATC2609A_PMU_SYS_CTL1_EN_S1;
	reg_val = restart ? ATC2609A_PMU_SYS_CTL0_RESTART_EN : 0;

	ret = regmap_update_bits(pwrc->regmap,
				 restart ? ATC2609A_PMU_SYS_CTL0 : ATC2609A_PMU_SYS_CTL1,
				 reg_mask, reg_val);
	if (ret) {
		dev_err(pwrc->dev, "failed to write SYS_CTL%d: %d\n",
			restart ? 0 : 1, ret);
		return ret;
	}

	/* Wait for trigger completion */
	mdelay(200);

	return 0;
}

static int atc2603c_init(const struct atc260x_pwrc *pwrc)
{
	int ret;

	/*
	 * Delay transition from S2/S3 to S1 in order to avoid
	 * DDR init failure in Bootloader.
	 */
	ret = regmap_update_bits(pwrc->regmap, ATC2603C_PMU_SYS_CTL3,
				 ATC2603C_PMU_SYS_CTL3_S2S3TOS1_TIMER_EN,
				 ATC2603C_PMU_SYS_CTL3_S2S3TOS1_TIMER_EN);
	if (ret)
		dev_warn(pwrc->dev, "failed to write SYS_CTL3: %d\n", ret);

	/* Set wakeup sources */
	ret = regmap_update_bits(pwrc->regmap, ATC2603C_PMU_SYS_CTL0,
				 ATC2603C_PMU_SYS_CTL0_WK_ALL,
				 ATC2603C_PMU_SYS_CTL0_HDSW_WK_EN |
				 ATC2603C_PMU_SYS_CTL0_ONOFF_LONG_WK_EN);
	if (ret)
		dev_warn(pwrc->dev, "failed to write SYS_CTL0: %d\n", ret);

	return ret;
}

static int atc2609a_init(const struct atc260x_pwrc *pwrc)
{
	int ret;

	/* Set wakeup sources */
	ret = regmap_update_bits(pwrc->regmap, ATC2609A_PMU_SYS_CTL0,
				 ATC2609A_PMU_SYS_CTL0_WK_ALL,
				 ATC2609A_PMU_SYS_CTL0_HDSW_WK_EN |
				 ATC2609A_PMU_SYS_CTL0_ONOFF_LONG_WK_EN);
	if (ret)
		dev_warn(pwrc->dev, "failed to write SYS_CTL0: %d\n", ret);

	return ret;
}

static void atc260x_pwrc_pm_handler(void)
{
	atc260x_pwrc_data->do_poweroff(atc260x_pwrc_data, false);

	WARN_ONCE(1, "Unable to power off system\n");
}

static int atc260x_pwrc_restart_handler(struct notifier_block *nb,
					unsigned long mode, void *cmd)
{
	struct atc260x_pwrc *pwrc = container_of(nb, struct atc260x_pwrc,
						 restart_nb);
	pwrc->do_poweroff(pwrc, true);

	return NOTIFY_DONE;
}

static int atc260x_pwrc_probe(struct platform_device *pdev)
{
	struct atc260x *atc260x = dev_get_drvdata(pdev->dev.parent);
	struct atc260x_pwrc *priv;
	int ret;

	priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
	if (!priv)
		return -ENOMEM;

	priv->dev = &pdev->dev;
	priv->regmap = atc260x->regmap;
	priv->restart_nb.notifier_call = atc260x_pwrc_restart_handler;
	priv->restart_nb.priority = 192;

	switch (atc260x->ic_type) {
	case ATC2603C:
		priv->do_poweroff = atc2603c_do_poweroff;
		ret = atc2603c_init(priv);
		break;
	case ATC2609A:
		priv->do_poweroff = atc2609a_do_poweroff;
		ret = atc2609a_init(priv);
		break;
	default:
		dev_err(priv->dev,
			"Poweroff not supported for ATC260x PMIC type: %u\n",
			atc260x->ic_type);
		return -EINVAL;
	}

	if (ret)
		return ret;

	platform_set_drvdata(pdev, priv);

	if (!pm_power_off) {
		atc260x_pwrc_data = priv;
		pm_power_off = atc260x_pwrc_pm_handler;
	} else {
		dev_warn(priv->dev, "Poweroff callback already assigned\n");
	}

	ret = register_restart_handler(&priv->restart_nb);
	if (ret)
		dev_err(priv->dev, "failed to register restart handler: %d\n",
			ret);

	return ret;
}

static int atc260x_pwrc_remove(struct platform_device *pdev)
{
	struct atc260x_pwrc *priv = platform_get_drvdata(pdev);

	if (atc260x_pwrc_data == priv) {
		pm_power_off = NULL;
		atc260x_pwrc_data = NULL;
	}

	unregister_restart_handler(&priv->restart_nb);

	return 0;
}

static struct platform_driver atc260x_pwrc_driver = {
	.probe = atc260x_pwrc_probe,
	.remove = atc260x_pwrc_remove,
	.driver = {
		.name = "atc260x-pwrc",
	},
};

module_platform_driver(atc260x_pwrc_driver);

MODULE_DESCRIPTION("Poweroff & reset driver for ATC260x PMICs");
MODULE_AUTHOR("Cristian Ciocaltea <cristian.ciocaltea@gmail.com>");
MODULE_LICENSE("GPL");
Commit	Line	Data
3235d130 CC	1	// SPDX-License-Identifier: GPL-2.0+
	2	/*
	3	* Poweroff & reset driver for Actions Semi ATC260x PMICs
	4	*
	5	* Copyright (c) 2020 Cristian Ciocaltea <cristian.ciocaltea@gmail.com>
	6	*/
	7
	8	#include <linux/delay.h>
	9	#include <linux/mfd/atc260x/core.h>
	10	#include <linux/module.h>
	11	#include <linux/platform_device.h>
	12	#include <linux/power_supply.h>
	13	#include <linux/reboot.h>
	14	#include <linux/regmap.h>
	15
	16	struct atc260x_pwrc {
	17	struct device *dev;
	18	struct regmap *regmap;
	19	struct notifier_block restart_nb;
	20	int (do_poweroff)(const struct atc260x_pwrc pwrc, bool restart);
	21	};
	22
	23	/* Global variable needed only for pm_power_off */
	24	static struct atc260x_pwrc *atc260x_pwrc_data;
	25
	26	static int atc2603c_do_poweroff(const struct atc260x_pwrc *pwrc, bool restart)
	27	{
	28	int ret, deep_sleep = 0;
	29	uint reg_mask, reg_val;
	30
	31	/* S4-Deep Sleep Mode is NOT available for WALL/USB power */
	32	if (!restart && !power_supply_is_system_supplied()) {
	33	deep_sleep = 1;
	34	dev_info(pwrc->dev, "Enabling S4-Deep Sleep Mode");
	35	}
	36
	37	/* Update wakeup sources */
	38	reg_val = ATC2603C_PMU_SYS_CTL0_ONOFF_LONG_WK_EN \|
	39	(restart ? ATC2603C_PMU_SYS_CTL0_RESET_WK_EN
	40	: ATC2603C_PMU_SYS_CTL0_ONOFF_SHORT_WK_EN);
	41
	42	ret = regmap_update_bits(pwrc->regmap, ATC2603C_PMU_SYS_CTL0,
	43	ATC2603C_PMU_SYS_CTL0_WK_ALL, reg_val);
	44	if (ret)
	45	dev_warn(pwrc->dev, "failed to write SYS_CTL0: %d\n", ret);
	46
	47	/* Update power mode */
	48	reg_mask = ATC2603C_PMU_SYS_CTL3_EN_S2 \| ATC2603C_PMU_SYS_CTL3_EN_S3;
	49
	50	ret = regmap_update_bits(pwrc->regmap, ATC2603C_PMU_SYS_CTL3, reg_mask,
	51	deep_sleep ? 0 : ATC2603C_PMU_SYS_CTL3_EN_S3);
	52	if (ret) {
	53	dev_err(pwrc->dev, "failed to write SYS_CTL3: %d\n", ret);
	54	return ret;
	55	}
	56
	57	/* Trigger poweroff / restart sequence */
	58	reg_mask = restart ? ATC2603C_PMU_SYS_CTL0_RESTART_EN
	59	: ATC2603C_PMU_SYS_CTL1_EN_S1;
	60	reg_val = restart ? ATC2603C_PMU_SYS_CTL0_RESTART_EN : 0;
	61
	62	ret = regmap_update_bits(pwrc->regmap,
	63	restart ? ATC2603C_PMU_SYS_CTL0 : ATC2603C_PMU_SYS_CTL1,
	64	reg_mask, reg_val);
65	if (ret) {
66	dev_err(pwrc->dev, "failed to write SYS_CTL%d: %d\n",
67	restart ? 0 : 1, ret);
68	return ret;
69	}
70
71	/* Wait for trigger completion */
72	mdelay(200);
73
74	return 0;
75	}
76
77	static int atc2609a_do_poweroff(const struct atc260x_pwrc *pwrc, bool restart)
78	{
79	int ret, deep_sleep = 0;
80	uint reg_mask, reg_val;
81
82	/* S4-Deep Sleep Mode is NOT available for WALL/USB power */
83	if (!restart && !power_supply_is_system_supplied()) {
84	deep_sleep = 1;
85	dev_info(pwrc->dev, "Enabling S4-Deep Sleep Mode");
86	}
87
88	/* Update wakeup sources */
89	reg_val = ATC2609A_PMU_SYS_CTL0_ONOFF_LONG_WK_EN \|
90	(restart ? ATC2609A_PMU_SYS_CTL0_RESET_WK_EN
91	: ATC2609A_PMU_SYS_CTL0_ONOFF_SHORT_WK_EN);
92
93	ret = regmap_update_bits(pwrc->regmap, ATC2609A_PMU_SYS_CTL0,
94	ATC2609A_PMU_SYS_CTL0_WK_ALL, reg_val);
95	if (ret)
96	dev_warn(pwrc->dev, "failed to write SYS_CTL0: %d\n", ret);
97
98	/* Update power mode */
99	reg_mask = ATC2609A_PMU_SYS_CTL3_EN_S2 \| ATC2609A_PMU_SYS_CTL3_EN_S3;
100
101	ret = regmap_update_bits(pwrc->regmap, ATC2609A_PMU_SYS_CTL3, reg_mask,
102	deep_sleep ? 0 : ATC2609A_PMU_SYS_CTL3_EN_S3);
103	if (ret) {
104	dev_err(pwrc->dev, "failed to write SYS_CTL3: %d\n", ret);
105	return ret;
106	}
107
108	/* Trigger poweroff / restart sequence */
109	reg_mask = restart ? ATC2609A_PMU_SYS_CTL0_RESTART_EN
110	: ATC2609A_PMU_SYS_CTL1_EN_S1;
111	reg_val = restart ? ATC2609A_PMU_SYS_CTL0_RESTART_EN : 0;
112
113	ret = regmap_update_bits(pwrc->regmap,
114	restart ? ATC2609A_PMU_SYS_CTL0 : ATC2609A_PMU_SYS_CTL1,
115	reg_mask, reg_val);
116	if (ret) {
117	dev_err(pwrc->dev, "failed to write SYS_CTL%d: %d\n",
118	restart ? 0 : 1, ret);
119	return ret;
120	}
121
122	/* Wait for trigger completion */
123	mdelay(200);
124
125	return 0;
126	}
127
128	static int atc2603c_init(const struct atc260x_pwrc *pwrc)
129	{
130	int ret;
131
132	/*
133	* Delay transition from S2/S3 to S1 in order to avoid
134	* DDR init failure in Bootloader.
135	*/
136	ret = regmap_update_bits(pwrc->regmap, ATC2603C_PMU_SYS_CTL3,
137	ATC2603C_PMU_SYS_CTL3_S2S3TOS1_TIMER_EN,
138	ATC2603C_PMU_SYS_CTL3_S2S3TOS1_TIMER_EN);
139	if (ret)
140	dev_warn(pwrc->dev, "failed to write SYS_CTL3: %d\n", ret);
141
142	/* Set wakeup sources */
143	ret = regmap_update_bits(pwrc->regmap, ATC2603C_PMU_SYS_CTL0,
144	ATC2603C_PMU_SYS_CTL0_WK_ALL,
145	ATC2603C_PMU_SYS_CTL0_HDSW_WK_EN \|
146	ATC2603C_PMU_SYS_CTL0_ONOFF_LONG_WK_EN);
147	if (ret)
148	dev_warn(pwrc->dev, "failed to write SYS_CTL0: %d\n", ret);
149
150	return ret;
151	}
152
153	static int atc2609a_init(const struct atc260x_pwrc *pwrc)
154	{
155	int ret;
156
157	/* Set wakeup sources */
158	ret = regmap_update_bits(pwrc->regmap, ATC2609A_PMU_SYS_CTL0,
159	ATC2609A_PMU_SYS_CTL0_WK_ALL,
160	ATC2609A_PMU_SYS_CTL0_HDSW_WK_EN \|
161	ATC2609A_PMU_SYS_CTL0_ONOFF_LONG_WK_EN);
162	if (ret)
163	dev_warn(pwrc->dev, "failed to write SYS_CTL0: %d\n", ret);
164
165	return ret;
166	}
167
168	static void atc260x_pwrc_pm_handler(void)
169	{
170	atc260x_pwrc_data->do_poweroff(atc260x_pwrc_data, false);
171
172	WARN_ONCE(1, "Unable to power off system\n");
173	}
174
175	static int atc260x_pwrc_restart_handler(struct notifier_block *nb,
176	unsigned long mode, void *cmd)
177	{
178	struct atc260x_pwrc *pwrc = container_of(nb, struct atc260x_pwrc,
179	restart_nb);
180	pwrc->do_poweroff(pwrc, true);
181
182	return NOTIFY_DONE;
183	}
184
185	static int atc260x_pwrc_probe(struct platform_device *pdev)
186	{
187	struct atc260x *atc260x = dev_get_drvdata(pdev->dev.parent);
188	struct atc260x_pwrc *priv;
189	int ret;
190
191	priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
192	if (!priv)
193	return -ENOMEM;
194
195	priv->dev = &pdev->dev;
196	priv->regmap = atc260x->regmap;
197	priv->restart_nb.notifier_call = atc260x_pwrc_restart_handler;
198	priv->restart_nb.priority = 192;
199
200	switch (atc260x->ic_type) {
201	case ATC2603C:
202	priv->do_poweroff = atc2603c_do_poweroff;
203	ret = atc2603c_init(priv);
204	break;
205	case ATC2609A:
206	priv->do_poweroff = atc2609a_do_poweroff;
207	ret = atc2609a_init(priv);
208	break;
209	default:
210	dev_err(priv->dev,
211	"Poweroff not supported for ATC260x PMIC type: %u\n",
212	atc260x->ic_type);
213	return -EINVAL;
214	}
215
216	if (ret)
217	return ret;
218
219	platform_set_drvdata(pdev, priv);
220
221	if (!pm_power_off) {
222	atc260x_pwrc_data = priv;
223	pm_power_off = atc260x_pwrc_pm_handler;
224	} else {
225	dev_warn(priv->dev, "Poweroff callback already assigned\n");
226	}
227
228	ret = register_restart_handler(&priv->restart_nb);
229	if (ret)
230	dev_err(priv->dev, "failed to register restart handler: %d\n",
231	ret);
232
233	return ret;
234	}
235
236	static int atc260x_pwrc_remove(struct platform_device *pdev)
237	{
238	struct atc260x_pwrc *priv = platform_get_drvdata(pdev);
239
240	if (atc260x_pwrc_data == priv) {
241	pm_power_off = NULL;
242	atc260x_pwrc_data = NULL;
243	}
244
245	unregister_restart_handler(&priv->restart_nb);
246
247	return 0;
248	}
249
250	static struct platform_driver atc260x_pwrc_driver = {
251	.probe = atc260x_pwrc_probe,
252	.remove = atc260x_pwrc_remove,
253	.driver = {
254	.name = "atc260x-pwrc",
255	},
256	};
257
258	module_platform_driver(atc260x_pwrc_driver);
259
260	MODULE_DESCRIPTION("Poweroff & reset driver for ATC260x PMICs");
261	MODULE_AUTHOR("Cristian Ciocaltea <cristian.ciocaltea@gmail.com>");
262	MODULE_LICENSE("GPL");