imx: imx8qm/imx8qxp: Power down the resources before SPL jump to u-boot

[oweals/u-boot.git] / drivers / power / domain / imx8-power-domain-legacy.c

diff --git a/drivers/power/domain/imx8-power-domain-legacy.c b/drivers/power/domain/imx8-power-domain-legacy.c
index f679df9e5d1bbe40c0460e916c9308950a31edb9..7ba4056e2d5b48595656f998fa00f8f0ddc7ab9a 100644 (file)
--- a/drivers/power/domain/imx8-power-domain-legacy.c
+++ b/drivers/power/domain/imx8-power-domain-legacy.c
@@ -11,6 +11,7 @@
 #include <asm/arch/power-domain.h>
 #include <dm/device-internal.h>
 #include <dm/device.h>
+#include <dm/uclass-internal.h>
 #include <asm/arch/sci/sci.h>
 
 DECLARE_GLOBAL_DATA_PTR;
@@ -19,6 +20,40 @@ struct imx8_power_domain_priv {
        bool state_on;
 };
 
+static bool check_device_power_off(struct udevice *dev,
+                                  const char *permanent_on_devices[],
+                                  int size)
+{
+       int i;
+
+       for (i = 0; i < size; i++) {
+               if (!strcmp(dev->name, permanent_on_devices[i]))
+                       return false;
+       }
+
+       return true;
+}
+
+void imx8_power_off_pd_devices(const char *permanent_on_devices[], int size)
+{
+       struct udevice *dev;
+       struct power_domain pd;
+
+       for (uclass_find_first_device(UCLASS_POWER_DOMAIN, &dev); dev;
+            uclass_find_next_device(&dev)) {
+               if (!device_active(dev))
+                       continue;
+               /*
+                * Power off active pd devices except the permanent
+                * power on devices
+                */
+               if (check_device_power_off(dev, permanent_on_devices, size)) {
+                       pd.dev = dev;
+                       power_domain_off(&pd);
+               }
+       }
+}
+
 int imx8_power_domain_lookup_name(const char *name,
                                  struct power_domain *power_domain)
 {