misc: sram: fix resource leaks in probe error path

Make sure to disable clocks and deregister any exported partitions
before returning on late probe errors.

Note that since commit ee895ccdf776 ("misc: sram: fix enabled clock leak
on error path"), partitions are deliberately exported before enabling
the clock so we stick to that logic here. A follow up patch will address
this.

Fixes: 2ae2e28852f2 ("misc: sram: add Atmel securam support")
Cc: stable <stable@vger.kernel.org>     # 4.9
Cc: Alexandre Belloni <alexandre.belloni@free-electrons.com>
Signed-off-by: Johan Hovold <johan@kernel.org>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

1 files changed, 8 insertions, 1 deletions

diff --git a/drivers/misc/sram.c b/drivers/misc/sram.c
index c5dc6095686a..679647713e36 100644
--- a/drivers/misc/sram.c
+++ b/drivers/misc/sram.c
@@ -407,13 +407,20 @@ static int sram_probe(struct platform_device *pdev)
         if (init_func) {
                 ret = init_func();
                 if (ret)
-                        return ret;
+                        goto err_disable_clk;
         }
 
         dev_dbg(sram->dev, "SRAM pool: %zu KiB @ 0x%p\n",
                 gen_pool_size(sram->pool) / 1024, sram->virt_base);
 
         return 0;
+
+err_disable_clk:
+        if (sram->clk)
+                clk_disable_unprepare(sram->clk);
+        sram_free_partitions(sram);
+
+        return ret;
 }
 
 static int sram_remove(struct platform_device *pdev)