[PATCH 2/3] timer-uclass: relocate ops pointers for CONFIG_NEEDS_MANUAL_RELOC

[Ovidiu Panait]

Relocate timer_ops pointers when CONFIG_NEEDS_MANUAL_RELOC is enabled.

The (gd->flags & GD_FLG_RELOC) check was added to make sure the reloc_done
logic works for drivers that use DM_FLAG_PRE_RELOC.

Signed-off-by: Ovidiu Panait <ovpan...@gmail.com>
---

 drivers/timer/timer-uclass.c | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/drivers/timer/timer-uclass.c b/drivers/timer/timer-uclass.c
index bdc77b3822..bb71979213 100644
--- a/drivers/timer/timer-uclass.c
+++ b/drivers/timer/timer-uclass.c
@@ -18,6 +18,7 @@
 #include <init.h>
 #include <timer.h>
 #include <linux/err.h>
+#include <relocate.h>
 
 DECLARE_GLOBAL_DATA_PTR;
 
@@ -50,6 +51,19 @@ unsigned long notrace timer_get_rate(struct udevice *dev)
 
 static int timer_pre_probe(struct udevice *dev)
 {
+       if (IS_ENABLED(CONFIG_NEEDS_MANUAL_RELOC) &&
+           (gd->flags & GD_FLG_RELOC)) {
+               struct timer_ops *ops = timer_get_ops(dev);
+               static int reloc_done;
+
+               if (!reloc_done) {
+                       if (ops->get_count)
+                               MANUAL_RELOC(ops->get_count);
+
+                       reloc_done++;
+               }
+       }
+
        if (CONFIG_IS_ENABLED(OF_REAL)) {
                struct timer_dev_priv *uc_priv = dev_get_uclass_priv(dev);
                struct clk timer_clk;
-- 
2.25.1