Register with kernel restart handler instead of setting arm_pm_restart
directly. Select high priority since the restart handler is instantiated
through devicetree, indicating that it should be used if configured.

Signed-off-by: Guenter Roeck <li...@roeck-us.net>
---
 drivers/power/reset/arm-versatile-reboot.c | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/drivers/power/reset/arm-versatile-reboot.c 
b/drivers/power/reset/arm-versatile-reboot.c
index 5b08bff..e533812 100644
--- a/drivers/power/reset/arm-versatile-reboot.c
+++ b/drivers/power/reset/arm-versatile-reboot.c
@@ -13,7 +13,6 @@
 #include <linux/reboot.h>
 #include <linux/regmap.h>
 #include <linux/of.h>
-#include <asm/system_misc.h>
 
 #define REALVIEW_SYS_LOCK_OFFSET       0x20
 #define REALVIEW_SYS_LOCK_VAL          0xA05F
@@ -57,7 +56,8 @@ static const struct of_device_id versatile_reboot_of_match[] 
= {
        },
 };
 
-static void versatile_reboot(enum reboot_mode mode, const char *cmd)
+static int versatile_reboot(struct notifier_block *this, unsigned long mode,
+                           void *cmd)
 {
        /* Unlock the reset register */
        regmap_write(syscon_regmap, REALVIEW_SYS_LOCK_OFFSET,
@@ -87,12 +87,20 @@ static void versatile_reboot(enum reboot_mode mode, const 
char *cmd)
                break;
        }
        dsb();
+
+       return NOTIFY_DONE;
 }
 
+static struct notifier_block versatile_reboot_nb = {
+       .notifier_call = versatile_reboot,
+       .priority = 192,
+};
+
 static int __init versatile_reboot_probe(void)
 {
        const struct of_device_id *reboot_id;
        struct device_node *np;
+       int err;
 
        np = of_find_matching_node_and_match(NULL, versatile_reboot_of_match,
                                                 &reboot_id);
@@ -104,7 +112,10 @@ static int __init versatile_reboot_probe(void)
        if (IS_ERR(syscon_regmap))
                return PTR_ERR(syscon_regmap);
 
-       arm_pm_restart = versatile_reboot;
+       err = register_restart_handler(&versatile_reboot_nb);
+       if (err)
+               return err;
+
        pr_info("versatile reboot driver registered\n");
        return 0;
 }
-- 
2.1.0

--
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