return container_of(rng, struct bcm2835_rng_priv, rng);
 }
 
+static inline u32 rng_readl(struct bcm2835_rng_priv *priv, u32 offset)
+{
+       return readl(priv->base + offset);
+}
+
+static inline void rng_writel(struct bcm2835_rng_priv *priv, u32 val,
+                             u32 offset)
+{
+       writel(val, priv->base + offset);
+}
+
 static int bcm2835_rng_read(struct hwrng *rng, void *buf, size_t max,
                               bool wait)
 {
        u32 max_words = max / sizeof(u32);
        u32 num_words, count;
 
-       while ((__raw_readl(priv->base + RNG_STATUS) >> 24) == 0) {
+       while ((rng_readl(priv, RNG_STATUS) >> 24) == 0) {
                if (!wait)
                        return 0;
                cpu_relax();
        }
 
-       num_words = readl(priv->base + RNG_STATUS) >> 24;
+       num_words = rng_readl(priv, RNG_STATUS) >> 24;
        if (num_words > max_words)
                num_words = max_words;
 
        for (count = 0; count < num_words; count++)
-               ((u32 *)buf)[count] = readl(priv->base + RNG_DATA);
+               ((u32 *)buf)[count] = rng_readl(priv, RNG_DATA);
 
        return num_words * sizeof(u32);
 }
 
        if (priv->mask_interrupts) {
                /* mask the interrupt */
-               val = readl(priv->base + RNG_INT_MASK);
+               val = rng_readl(priv, RNG_INT_MASK);
                val |= RNG_INT_OFF;
-               writel(val, priv->base + RNG_INT_MASK);
+               rng_writel(priv, val, RNG_INT_MASK);
        }
 
        /* set warm-up count & enable */
-       __raw_writel(RNG_WARMUP_COUNT, priv->base + RNG_STATUS);
-       __raw_writel(RNG_RBGEN, priv->base + RNG_CTRL);
+       rng_writel(priv, RNG_WARMUP_COUNT, RNG_STATUS);
+       rng_writel(priv, RNG_RBGEN, RNG_CTRL);
 
        return ret;
 }
        struct bcm2835_rng_priv *priv = to_rng_priv(rng);
 
        /* disable rng hardware */
-       __raw_writel(0, priv->base + RNG_CTRL);
+       rng_writel(priv, 0, RNG_CTRL);
 
        if (!IS_ERR(priv->clk))
                clk_disable_unprepare(priv->clk);