iommu/tegra: smmu: Use helper function to check for valid register offset

Do not repeat the checking loop in the read and write
functions. Use a single helper function for that check and
call it in both accessors.

Signed-off-by: Joerg Roedel <joro@8bytes.org>

diff --git a/drivers/iommu/tegra-smmu.c b/drivers/iommu/tegra-smmu.c
index 2fecbe7..7747283 100644 (file)
--- a/drivers/iommu/tegra-smmu.c
+++ b/drivers/iommu/tegra-smmu.c
@@ -327,36 +327,37 @@ static struct smmu_device *smmu_handle; /* unique for a system */
 /*
  *     SMMU register accessors
  */
-static inline u32 smmu_read(struct smmu_device *smmu, size_t offs)
+static bool inline smmu_valid_reg(struct smmu_device *smmu,
+                                 void __iomem *addr)
 {
        int i;
 
        for (i = 0; i < smmu->nregs; i++) {
-               void __iomem *addr = smmu->regbase + offs;
-
-               BUG_ON(addr < smmu->regs[i]);
+               if (addr < smmu->regs[i])
+                       break;
                if (addr <= smmu->rege[i])
-                       return readl(addr);
+                       return true;
        }
 
-       BUG();
+       return false;
 }
 
-static inline void smmu_write(struct smmu_device *smmu, u32 val, size_t offs)
+static inline u32 smmu_read(struct smmu_device *smmu, size_t offs)
 {
-       int i;
+       void __iomem *addr = smmu->regbase + offs;
 
-       for (i = 0; i < smmu->nregs; i++) {
-               void __iomem *addr = smmu->regbase + offs;
+       BUG_ON(!smmu_valid_reg(smmu, addr));
 
-               BUG_ON(addr < smmu->regs[i]);
-               if (addr <= smmu->rege[i]) {
-                       writel(val, addr);
-                       return;
-               }
-       }
+       return readl(addr);
+}
+
+static inline void smmu_write(struct smmu_device *smmu, u32 val, size_t offs)
+{
+       void __iomem *addr = smmu->regbase + offs;
+
+       BUG_ON(!smmu_valid_reg(smmu, addr));
 
-       BUG();
+       writel(val, addr);
 }
 
 #define VA_PAGE_TO_PA(va, page)        \