[PATCH v2 4/5] pmdomain: allow callback for when devices are attached

[Ahmad Fatoum <a.fatoum@barebox.org>]

This will be needed for the incoming Rockchip pmdomain drivers.

Signed-off-by: Ahmad Fatoum <a.fatoum@barebox.org>
---
v1 -> v2:
  - no change
---
 drivers/base/power.c | 10 +++++++++-
 include/pm_domain.h  |  3 +++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/drivers/base/power.c b/drivers/base/power.c
index c291f8657cbb..05fb096cf64c 100644
--- a/drivers/base/power.c
+++ b/drivers/base/power.c
@@ -331,14 +331,22 @@ int pm_runtime_resume_and_get_genpd(struct device *dev)
 	return genpd_power_on(genpd);
 }
 
-static void genpd_add_device(struct generic_pm_domain *genpd, struct device *dev)
+static int genpd_add_device(struct generic_pm_domain *genpd, struct device *dev)
 {
 	dev->pm_domain = genpd;
+
+	if (genpd->attach_dev)
+		return genpd->attach_dev(genpd, dev);
+
+	return 0;
 }
 
 static void genpd_remove_device(struct generic_pm_domain *genpd,
 			       struct device *dev)
 {
+	if (genpd->detach_dev)
+		genpd->detach_dev(genpd, dev);
+
 	dev->pm_domain = NULL;
 }
 
diff --git a/include/pm_domain.h b/include/pm_domain.h
index 3678deb9a76c..13dc926448c8 100644
--- a/include/pm_domain.h
+++ b/include/pm_domain.h
@@ -20,6 +20,9 @@ struct generic_pm_domain {
 
 	int (*power_off)(struct generic_pm_domain *domain);
 	int (*power_on)(struct generic_pm_domain *domain);
+
+	int (*attach_dev)(struct generic_pm_domain *genpd, struct device *dev);
+	void (*detach_dev)(struct generic_pm_domain *genpd, struct device *dev);
 };
 
 typedef struct generic_pm_domain *(*genpd_xlate_t)(struct of_phandle_args *args,
-- 
2.47.3