/* Copyright (C) 2020 Felix Fietkau <nbd@nbd.name> */
 
 #include <linux/kernel.h>
-#include <linux/jiffies.h>
-#include <linux/delay.h>
 #include <linux/io.h>
+#include <linux/iopoll.h>
 #include <linux/etherdevice.h>
 #include <linux/platform_device.h>
 #include "mtk_ppe.h"
 
 static int mtk_ppe_wait_busy(struct mtk_ppe *ppe)
 {
-       unsigned long timeout = jiffies + HZ;
-
-       while (time_is_before_jiffies(timeout)) {
-               if (!(ppe_r32(ppe, MTK_PPE_GLO_CFG) & MTK_PPE_GLO_CFG_BUSY))
-                       return 0;
+       int ret;
+       u32 val;
 
-               usleep_range(10, 20);
-       }
+       ret = readl_poll_timeout(ppe->base + MTK_PPE_GLO_CFG, val,
+                                !(val & MTK_PPE_GLO_CFG_BUSY),
+                                20, MTK_PPE_WAIT_TIMEOUT_US);
 
-       dev_err(ppe->dev, "PPE table busy");
+       if (ret)
+               dev_err(ppe->dev, "PPE table busy");
 
-       return -ETIMEDOUT;
+       return ret;
 }
 
 static void mtk_ppe_cache_clear(struct mtk_ppe *ppe)
 
 #define MTK_PPE_ENTRIES_SHIFT          3
 #define MTK_PPE_ENTRIES                        (1024 << MTK_PPE_ENTRIES_SHIFT)
 #define MTK_PPE_HASH_MASK              (MTK_PPE_ENTRIES - 1)
+#define MTK_PPE_WAIT_TIMEOUT_US                1000000
 
 #define MTK_FOE_IB1_UNBIND_TIMESTAMP   GENMASK(7, 0)
 #define MTK_FOE_IB1_UNBIND_PACKETS     GENMASK(23, 8)