1 files changed, 20 insertions, 7 deletions

diff --git a/drivers/watchdog/orion_wdt.c b/drivers/watchdog/orion_wdt.c
index afa38314fb18..75f623f5e6c3 100644
--- a/drivers/watchdog/orion_wdt.c
+++ b/drivers/watchdog/orion_wdt.c
@@ -262,10 +262,6 @@ static void __iomem *orion_wdt_ioremap_rstout(struct platform_device *pdev,
                 return devm_ioremap(&pdev->dev, res->start,
                                     resource_size(res));
 
-        /* This workaround works only for "orion-wdt", DT-enabled */
-        if (!of_device_is_compatible(pdev->dev.of_node, "marvell,orion-wdt"))
-                return NULL;
-
         rstout = internal_regs + ORION_RSTOUT_MASK_OFFSET;
 
         WARN(1, FW_BUG "falling back to harcoded RSTOUT reg %pa\n", &rstout);
@@ -316,6 +312,7 @@ MODULE_DEVICE_TABLE(of, orion_wdt_of_match_table);
 static int orion_wdt_get_regs(struct platform_device *pdev,
                               struct orion_watchdog *dev)
 {
+        struct device_node *node = pdev->dev.of_node;
         struct resource *res;
 
         res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
@@ -326,10 +323,26 @@ static int orion_wdt_get_regs(struct platform_device *pdev,
         if (!dev->reg)
                 return -ENOMEM;
 
-        dev->rstout = orion_wdt_ioremap_rstout(pdev, res->start &
-                                                     INTERNAL_REGS_MASK);
-        if (!dev->rstout)
+        /* Each supported compatible has some RSTOUT register quirk */
+        if (of_device_is_compatible(node, "marvell,orion-wdt")) {
+
+                dev->rstout = orion_wdt_ioremap_rstout(pdev, res->start &
+                                                       INTERNAL_REGS_MASK);
+                if (!dev->rstout)
+                        return -ENODEV;
+
+        } else if (of_device_is_compatible(node, "marvell,armada-370-wdt") ||
+                   of_device_is_compatible(node, "marvell,armada-xp-wdt")) {
+
+                /* Dedicated RSTOUT register, can be requested. */
+                res = platform_get_resource(pdev, IORESOURCE_MEM, 1);
+                dev->rstout = devm_ioremap_resource(&pdev->dev, res);
+                if (IS_ERR(dev->rstout))
+                        return PTR_ERR(dev->rstout);
+
+        } else {
                 return -ENODEV;
+        }
 
         return 0;
 }