powercap: DTPM: Fix missing cpufreq_cpu_put() calls

[platform/kernel/linux-starfive.git] / drivers / powercap / dtpm_cpu.c

diff --git a/drivers/powercap/dtpm_cpu.c b/drivers/powercap/dtpm_cpu.c
index 8a2f18f..9193c3b 100644 (file)
--- a/drivers/powercap/dtpm_cpu.c
+++ b/drivers/powercap/dtpm_cpu.c
@@ -140,6 +140,8 @@ static void pd_release(struct dtpm *dtpm)
        if (policy) {
                for_each_cpu(dtpm_cpu->cpu, policy->related_cpus)
                        per_cpu(dtpm_per_cpu, dtpm_cpu->cpu) = NULL;
+
+               cpufreq_cpu_put(policy);
        }
        
        kfree(dtpm_cpu);
@@ -191,12 +193,16 @@ static int __dtpm_cpu_setup(int cpu, struct dtpm *parent)
                return 0;
 
        pd = em_cpu_get(cpu);
-       if (!pd || em_is_artificial(pd))
-               return -EINVAL;
+       if (!pd || em_is_artificial(pd)) {
+               ret = -EINVAL;
+               goto release_policy;
+       }
 
        dtpm_cpu = kzalloc(sizeof(*dtpm_cpu), GFP_KERNEL);
-       if (!dtpm_cpu)
-               return -ENOMEM;
+       if (!dtpm_cpu) {
+               ret = -ENOMEM;
+               goto release_policy;
+       }
 
        dtpm_init(&dtpm_cpu->dtpm, &dtpm_ops);
        dtpm_cpu->cpu = cpu;
@@ -216,6 +222,7 @@ static int __dtpm_cpu_setup(int cpu, struct dtpm *parent)
        if (ret)
                goto out_dtpm_unregister;
 
+       cpufreq_cpu_put(policy);
        return 0;
 
 out_dtpm_unregister:
@@ -227,6 +234,8 @@ out_kfree_dtpm_cpu:
                per_cpu(dtpm_per_cpu, cpu) = NULL;
        kfree(dtpm_cpu);
 
+release_policy:
+       cpufreq_cpu_put(policy);
        return ret;
 }