@@ -19,10 +19,12 @@
#include <linux/of.h>
#include <linux/platform_device.h>
#include <linux/module.h>
+#include <linux/notifier.h>
#include <linux/reboot.h>
#include <linux/pm.h>
static void __iomem *msm_ps_hold;
+
static int do_msm_restart(struct notifier_block *nb, unsigned long action,
void *data)
{
@@ -37,11 +39,11 @@ static struct notifier_block restart_nb = {
.priority = 128,
};
-static void do_msm_poweroff(void)
-{
+static struct notifier_block msm_power_off_nb = {
/* TODO: Add poweroff capability */
- do_msm_restart(&restart_nb, 0, NULL);
-}
+ .notifier_call = do_msm_restart,
+ .priority = POWER_OFF_PRIORITY_FALLBACK,
+};
static int msm_restart_probe(struct platform_device *pdev)
{
@@ -55,7 +57,8 @@ static int msm_restart_probe(struct platform_device *pdev)
register_restart_handler(&restart_nb);
- pm_power_off = do_msm_poweroff;
+ if (register_power_off_handler(&msm_power_off_nb))
+ dev_warn(&pdev->dev, "Failed to register power-off handler\n");
return 0;
}