[PATCH v2] powerpc: powernv: Fix refcount leak bug

[Liang He]

In these driver init functions, there are two kinds of errors:

(1) missing of_put_node() for of_find_compatible_node()'s returned
    pointer (refcount incremented)  in fail path or when it is not
    used anymore.
(2) missing of_put_node() for 'for_each_xxx' loop's break

These bugs are similar with the ones reported by commit-09700c504d.

Signed-off-by: Liang He <win...@126.com>
---
 changelog:
 
 v2: merge all powernv related bugs into one commit 
 v1: only fix bugs in opal-powercap.c


 arch/powerpc/platforms/powernv/opal-powercap.c      | 6 +++++-
 arch/powerpc/platforms/powernv/opal-psr.c           | 6 +++++-
 arch/powerpc/platforms/powernv/opal-sensor-groups.c | 6 +++++-
 3 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/arch/powerpc/platforms/powernv/opal-powercap.c 
b/arch/powerpc/platforms/powernv/opal-powercap.c
index 64506b46e77b..78c359c90093 100644
--- a/arch/powerpc/platforms/powernv/opal-powercap.c
+++ b/arch/powerpc/platforms/powernv/opal-powercap.c
@@ -153,7 +153,7 @@ void __init opal_powercap_init(void)
        pcaps = kcalloc(of_get_child_count(powercap), sizeof(*pcaps),
                        GFP_KERNEL);
        if (!pcaps)
-               return;
+               goto out_powercap;
 
        powercap_kobj = kobject_create_and_add("powercap", opal_kobj);
        if (!powercap_kobj) {
@@ -226,6 +226,7 @@ void __init opal_powercap_init(void)
                }
                i++;
        }
+       of_node_put(powercap);
 
        return;
 
@@ -236,6 +237,9 @@ void __init opal_powercap_init(void)
                kfree(pcaps[i].pg.name);
        }
        kobject_put(powercap_kobj);
+       of_node_put(node);
 out_pcaps:
        kfree(pcaps);
+out_powercap:
+       of_node_put(powercap);
 }
diff --git a/arch/powerpc/platforms/powernv/opal-psr.c 
b/arch/powerpc/platforms/powernv/opal-psr.c
index 69d7e75950d1..ec32e0a93f08 100644
--- a/arch/powerpc/platforms/powernv/opal-psr.c
+++ b/arch/powerpc/platforms/powernv/opal-psr.c
@@ -135,7 +135,7 @@ void __init opal_psr_init(void)
        psr_attrs = kcalloc(of_get_child_count(psr), sizeof(*psr_attrs),
                            GFP_KERNEL);
        if (!psr_attrs)
-               return;
+               goto out_psr;
 
        psr_kobj = kobject_create_and_add("psr", opal_kobj);
        if (!psr_kobj) {
@@ -162,10 +162,14 @@ void __init opal_psr_init(void)
                }
                i++;
        }
+       of_node_put(psr);
 
        return;
 out_kobj:
+       of_node_put(node);
        kobject_put(psr_kobj);
 out:
        kfree(psr_attrs);
+out_psr:
+       of_node_put(psr);
 }
diff --git a/arch/powerpc/platforms/powernv/opal-sensor-groups.c 
b/arch/powerpc/platforms/powernv/opal-sensor-groups.c
index 8fba7d25ae56..9944376b115c 100644
--- a/arch/powerpc/platforms/powernv/opal-sensor-groups.c
+++ b/arch/powerpc/platforms/powernv/opal-sensor-groups.c
@@ -170,7 +170,7 @@ void __init opal_sensor_groups_init(void)
 
        sgs = kcalloc(of_get_child_count(sg), sizeof(*sgs), GFP_KERNEL);
        if (!sgs)
-               return;
+               goto out_sg_put;
 
        sg_kobj = kobject_create_and_add("sensor_groups", opal_kobj);
        if (!sg_kobj) {
@@ -222,6 +222,7 @@ void __init opal_sensor_groups_init(void)
                }
                i++;
        }
+       of_node_put(sg);
 
        return;
 
@@ -231,6 +232,9 @@ void __init opal_sensor_groups_init(void)
                kfree(sgs[i].sg.attrs);
        }
        kobject_put(sg_kobj);
+       of_node_put(node);
 out_sgs:
        kfree(sgs);
+out_sg_put:
+       of_node_put(sg);
 }
-- 
2.25.1