2 files changed, 15 insertions, 2 deletions

diff --git a/drivers/crypto/caam/ctrl.c b/drivers/crypto/caam/ctrl.c
index 77557ebcd337..9a2db9c30630 100644
--- a/drivers/crypto/caam/ctrl.c
+++ b/drivers/crypto/caam/ctrl.c
@@ -82,13 +82,18 @@ static int caam_probe(struct platform_device *pdev)
 
         /*
          * Enable DECO watchdogs and, if this is a PHYS_ADDR_T_64BIT kernel,
-         * 36-bit pointers in master configuration register
+         * long pointers in master configuration register
          */
         setbits32(&topregs->ctrl.mcr, MCFGR_WDENABLE |
                   (sizeof(dma_addr_t) == sizeof(u64) ? MCFGR_LONG_PTR : 0));
 
         if (sizeof(dma_addr_t) == sizeof(u64))
-                dma_set_mask(dev, DMA_BIT_MASK(36));
+                if (of_device_is_compatible(nprop, "fsl,sec-v5.0"))
+                        dma_set_mask(dev, DMA_BIT_MASK(40));
+                else
+                        dma_set_mask(dev, DMA_BIT_MASK(36));
+        else
+                dma_set_mask(dev, DMA_BIT_MASK(32));
 
         /*
          * Detect and enable JobRs
diff --git a/drivers/crypto/caam/jr.c b/drivers/crypto/caam/jr.c
index 6ce4c41c8634..9f16b2c03613 100644
--- a/drivers/crypto/caam/jr.c
+++ b/drivers/crypto/caam/jr.c
@@ -503,6 +503,14 @@ int caam_jr_probe(struct platform_device *pdev, struct device_node *np,
         dev_set_drvdata(jrdev, jrpriv);
         ctrlpriv->jrdev[ring] = jrdev;
 
+        if (sizeof(dma_addr_t) == sizeof(u64))
+                if (of_device_is_compatible(np, "fsl,sec-v5.0-job-ring"))
+                        dma_set_mask(jrdev, DMA_BIT_MASK(40));
+                else
+                        dma_set_mask(jrdev, DMA_BIT_MASK(36));
+        else
+                dma_set_mask(jrdev, DMA_BIT_MASK(32));
+
         /* Identify the interrupt */
         jrpriv->irq = of_irq_to_resource(np, 0, NULL);