mfd: stpmic1: Add PMIC poweroff via sys-off handler

Use devm_register_sys_off_handler() that allows to register multiple
power-off handlers.
This will allow boards using the stpmic1 to power-off.

Signed-off-by: Sean Nyekjaer <sean@geanix.com>
Signed-off-by: Lee Jones <lee@kernel.org>
Link: https://lore.kernel.org/r/20230602062426.3947116-2-sean@geanix.com

diff --git a/drivers/mfd/stpmic1.c b/drivers/mfd/stpmic1.c
index a79dfb9923d73c179f7b30a8c251ebf0e467eebe..3cc7492f828ff9646b97d2d885978c0d371bcb7c 100644 (file)
--- a/drivers/mfd/stpmic1.c
+++ b/drivers/mfd/stpmic1.c
@@ -7,6 +7,7 @@
 #include <linux/mfd/core.h>
 #include <linux/mfd/stpmic1.h>
 #include <linux/module.h>
+#include <linux/reboot.h>
 #include <linux/of.h>
 #include <linux/of_irq.h>
 #include <linux/of_platform.h>
@@ -117,6 +118,16 @@ static const struct regmap_irq_chip stpmic1_regmap_irq_chip = {
        .num_irqs = ARRAY_SIZE(stpmic1_irqs),
 };
 
+static int stpmic1_power_off(struct sys_off_data *data)
+{
+       struct stpmic1 *ddata = data->cb_data;
+
+       regmap_update_bits(ddata->regmap, MAIN_CR,
+                          SOFTWARE_SWITCH_OFF, SOFTWARE_SWITCH_OFF);
+
+       return NOTIFY_DONE;
+}
+
 static int stpmic1_probe(struct i2c_client *i2c)
 {
        struct stpmic1 *ddata;
@@ -159,6 +170,16 @@ static int stpmic1_probe(struct i2c_client *i2c)
                return ret;
        }
 
+       ret = devm_register_sys_off_handler(ddata->dev,
+                                           SYS_OFF_MODE_POWER_OFF,
+                                           SYS_OFF_PRIO_DEFAULT,
+                                           stpmic1_power_off,
+                                           ddata);
+       if (ret) {
+               dev_err(ddata->dev, "failed to register sys-off handler: %d\n", ret);
+               return ret;
+       }
+
        return devm_of_platform_populate(dev);
 }