1 files changed, 28 insertions, 1 deletions

diff --git a/drivers/clk/samsung/clk-exynos5440.c b/drivers/clk/samsung/clk-exynos5440.c
index 00d1d00a41de..979e81389cdd 100644
--- a/drivers/clk/samsung/clk-exynos5440.c
+++ b/drivers/clk/samsung/clk-exynos5440.c
@@ -15,6 +15,8 @@
 #include <linux/clk-provider.h>
 #include <linux/of.h>
 #include <linux/of_address.h>
+#include <linux/notifier.h>
+#include <linux/reboot.h>
 
 #include "clk.h"
 #include "clk-pll.h"
@@ -23,6 +25,8 @@
 #define CPU_CLK_STATUS          0xfc
 #define MISC_DOUT1              0x558
 
+static void __iomem *reg_base;
+
 /* parent clock name list */
 PNAME(mout_armclk_p)    = { "cplla", "cpllb" };
 PNAME(mout_spi_p)       = { "div125", "div200" };
@@ -89,10 +93,30 @@ static const struct of_device_id ext_clk_match[] __initconst = {
         {},
 };
 
+static int exynos5440_clk_restart_notify(struct notifier_block *this,
+                unsigned long code, void *unused)
+{
+        u32 val, status;
+
+        status = readl_relaxed(reg_base + 0xbc);
+        val = readl_relaxed(reg_base + 0xcc);
+        val = (val & 0xffff0000) | (status & 0xffff);
+        writel_relaxed(val, reg_base + 0xcc);
+
+        return NOTIFY_DONE;
+}
+
+/*
+ * Exynos5440 Clock restart notifier, handles restart functionality
+ */
+static struct notifier_block exynos5440_clk_restart_handler = {
+        .notifier_call = exynos5440_clk_restart_notify,
+        .priority = 128,
+};
+
 /* register exynos5440 clocks */
 static void __init exynos5440_clk_init(struct device_node *np)
 {
-        void __iomem *reg_base;
         struct samsung_clk_provider *ctx;
 
         reg_base = of_iomap(np, 0);
@@ -125,6 +149,9 @@ static void __init exynos5440_clk_init(struct device_node *np)
 
         samsung_clk_of_add_provider(np, ctx);
 
+        if (register_restart_handler(&exynos5440_clk_restart_handler))
+                pr_warn("exynos5440 clock can't register restart handler\n");
+
         pr_info("Exynos5440: arm_clk = %ldHz\n", _get_rate("arm_clk"));
         pr_info("exynos5440 clock initialization complete\n");
 }