[PATCH 2/3] regulator: core: Link consumer with regulator driver

[Pascal PAILLET-LME]

From: pascal paillet <p.pail...@st.com>

Add a device link between the consumer and the driver so that
the consumer is not suspended before the driver. The goal is to avoid
implementing suspend_late ops in regulator drivers.

Signed-off-by: pascal paillet <p.pail...@st.com>
---
 drivers/regulator/core.c | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/drivers/regulator/core.c b/drivers/regulator/core.c
index 6ed568b..607ec47 100644
--- a/drivers/regulator/core.c
+++ b/drivers/regulator/core.c
@@ -1740,6 +1740,8 @@ struct regulator *_regulator_get(struct device *dev, 
const char *id,
                        rdev->use_count = 0;
        }
 
+       device_link_add(dev, &rdev->dev, DL_FLAG_STATELESS);
+
        return regulator;
 }
 
@@ -1829,9 +1831,21 @@ static void _regulator_put(struct regulator *regulator)
 
        debugfs_remove_recursive(regulator->debugfs);
 
-       /* remove any sysfs entries */
-       if (regulator->dev)
+       if (regulator->dev) {
+               int count = 0;
+               struct regulator *r;
+
+               list_for_each_entry(r, &rdev->consumer_list, list)
+                       if (r->dev == regulator->dev)
+                               count++;
+
+               if (count == 1)
+                       device_link_remove(regulator->dev, &rdev->dev);
+
+               /* remove any sysfs entries */
                sysfs_remove_link(&rdev->dev.kobj, regulator->supply_name);
+       }
+
        regulator_lock(rdev);
        list_del(&regulator->list);
 
-- 
1.9.1