Register with kernel poweroff handler instead of setting pm_power_off
directly. Register with low priority to reflect that the original code
only sets pm_power_off if it was not already set.

Cc: Beniamino Galvani <b.galv...@gmail.com>
Signed-off-by: Guenter Roeck <li...@roeck-us.net>
---
v2: New patch

 drivers/mfd/rn5t618.c       | 32 ++++++++++++++++----------------
 include/linux/mfd/rn5t618.h |  2 ++
 2 files changed, 18 insertions(+), 16 deletions(-)

diff --git a/drivers/mfd/rn5t618.c b/drivers/mfd/rn5t618.c
index 6668571..045cd43 100644
--- a/drivers/mfd/rn5t618.c
+++ b/drivers/mfd/rn5t618.c
@@ -15,6 +15,8 @@
 #include <linux/mfd/core.h>
 #include <linux/mfd/rn5t618.h>
 #include <linux/module.h>
+#include <linux/notifier.h>
+#include <linux/pm.h>
 #include <linux/regmap.h>
 
 static const struct mfd_cell rn5t618_cells[] = {
@@ -47,16 +49,20 @@ static const struct regmap_config rn5t618_regmap_config = {
        .cache_type     = REGCACHE_RBTREE,
 };
 
-static struct rn5t618 *rn5t618_pm_power_off;
-
-static void rn5t618_power_off(void)
+static int rn5t618_power_off(struct notifier_block *this,
+                            unsigned long unused1, void *unused2)
 {
+       struct rn5t618 *rn5t618 = container_of(this, struct rn5t618,
+                                              poweroff_nb);
+
        /* disable automatic repower-on */
-       regmap_update_bits(rn5t618_pm_power_off->regmap, RN5T618_REPCNT,
+       regmap_update_bits(rn5t618->regmap, RN5T618_REPCNT,
                           RN5T618_REPCNT_REPWRON, 0);
        /* start power-off sequence */
-       regmap_update_bits(rn5t618_pm_power_off->regmap, RN5T618_SLPCNT,
+       regmap_update_bits(rn5t618->regmap, RN5T618_SLPCNT,
                           RN5T618_SLPCNT_SWPWROFF, RN5T618_SLPCNT_SWPWROFF);
+
+       return NOTIFY_DONE;
 }
 
 static int rn5t618_i2c_probe(struct i2c_client *i2c,
@@ -85,23 +91,17 @@ static int rn5t618_i2c_probe(struct i2c_client *i2c,
                return ret;
        }
 
-       if (!pm_power_off) {
-               rn5t618_pm_power_off = priv;
-               pm_power_off = rn5t618_power_off;
-       }
+       priv->poweroff_nb.notifier_call = rn5t618_power_off;
+       priv->poweroff_nb.priority = POWEROFF_PRIORITY_LOW;
+       ret = devm_register_power_off_handler(&i2c->dev, &priv->poweroff_nb);
+       if (ret)
+               dev_warn(&i2c->dev, "Failed to register poweroff handler\n");
 
        return 0;
 }
 
 static int rn5t618_i2c_remove(struct i2c_client *i2c)
 {
-       struct rn5t618 *priv = i2c_get_clientdata(i2c);
-
-       if (priv == rn5t618_pm_power_off) {
-               rn5t618_pm_power_off = NULL;
-               pm_power_off = NULL;
-       }
-
        mfd_remove_devices(&i2c->dev);
        return 0;
 }
diff --git a/include/linux/mfd/rn5t618.h b/include/linux/mfd/rn5t618.h
index c72d534..3b0f0e6 100644
--- a/include/linux/mfd/rn5t618.h
+++ b/include/linux/mfd/rn5t618.h
@@ -14,6 +14,7 @@
 #ifndef __LINUX_MFD_RN5T618_H
 #define __LINUX_MFD_RN5T618_H
 
+#include <linux/notifier.h>
 #include <linux/regmap.h>
 
 #define RN5T618_LSIVER                 0x00
@@ -223,6 +224,7 @@ enum {
 
 struct rn5t618 {
        struct regmap *regmap;
+       struct notifier_block poweroff_nb;
 };
 
 #endif /* __LINUX_MFD_RN5T618_H */
-- 
1.9.1

--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to