#include <linux/err.h>
 #include <linux/io.h>
 #include <linux/module.h>
+#include <linux/of_device.h>
 #include <linux/platform_device.h>
 
 #include <crypto/internal/rng.h>
 
 #define EXYNOS_RNG_CONTROL             0x0
 #define EXYNOS_RNG_STATUS              0x10
+
+#define EXYNOS_RNG_SEED_CONF           0x14
+#define EXYNOS_RNG_GEN_PRNG            BIT(1)
+
 #define EXYNOS_RNG_SEED_BASE           0x140
 #define EXYNOS_RNG_SEED(n)             (EXYNOS_RNG_SEED_BASE + (n * 0x4))
 #define EXYNOS_RNG_OUT_BASE            0x160
 #define EXYNOS_RNG_SEED_REGS           5
 #define EXYNOS_RNG_SEED_SIZE           (EXYNOS_RNG_SEED_REGS * 4)
 
+enum exynos_prng_type {
+       EXYNOS_PRNG_UNKNOWN = 0,
+       EXYNOS_PRNG_EXYNOS4,
+       EXYNOS_PRNG_EXYNOS5,
+};
+
 /*
  * Driver re-seeds itself with generated random numbers to increase
  * the randomness.
 /* Device associated memory */
 struct exynos_rng_dev {
        struct device                   *dev;
+       enum exynos_prng_type           type;
        void __iomem                    *mem;
        struct clk                      *clk;
        /* Generated numbers stored for seeding during resume */
 {
        int retry = EXYNOS_RNG_WAIT_RETRIES;
 
-       exynos_rng_writel(rng, EXYNOS_RNG_CONTROL_START,
-                         EXYNOS_RNG_CONTROL);
+       if (rng->type == EXYNOS_PRNG_EXYNOS4) {
+               exynos_rng_writel(rng, EXYNOS_RNG_CONTROL_START,
+                                 EXYNOS_RNG_CONTROL);
+       } else if (rng->type == EXYNOS_PRNG_EXYNOS5) {
+               exynos_rng_writel(rng, EXYNOS_RNG_GEN_PRNG,
+                                 EXYNOS_RNG_SEED_CONF);
+       }
 
        while (!(exynos_rng_readl(rng,
                        EXYNOS_RNG_STATUS) & EXYNOS_RNG_STATUS_RNG_DONE) && --retry)
        if (!rng)
                return -ENOMEM;
 
+       rng->type = (enum exynos_prng_type)of_device_get_match_data(&pdev->dev);
+
        rng->dev = &pdev->dev;
        rng->clk = devm_clk_get(&pdev->dev, "secss");
        if (IS_ERR(rng->clk)) {
 static const struct of_device_id exynos_rng_dt_match[] = {
        {
                .compatible = "samsung,exynos4-rng",
+               .data = (const void *)EXYNOS_PRNG_EXYNOS4,
+       }, {
+               .compatible = "samsung,exynos5250-prng",
+               .data = (const void *)EXYNOS_PRNG_EXYNOS5,
        },
        { },
 };