When return errors, the clock driver does not unmap
the mapped memory, so fix this issue.

Signed-off-by: Dongjiu Geng <[email protected]>
---
 drivers/clk/hisilicon/clk-hi3620.c | 8 ++++++--
 drivers/clk/hisilicon/clk.c        | 5 ++++-
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/drivers/clk/hisilicon/clk-hi3620.c 
b/drivers/clk/hisilicon/clk-hi3620.c
index a3d04c7c3da8..d5f1a8df4b1c 100644
--- a/drivers/clk/hisilicon/clk-hi3620.c
+++ b/drivers/clk/hisilicon/clk-hi3620.c
@@ -463,12 +463,16 @@ static void __init hi3620_mmc_clk_init(struct device_node 
*node)
        }
 
        clk_data = kzalloc(sizeof(*clk_data), GFP_KERNEL);
-       if (WARN_ON(!clk_data))
+       if (WARN_ON(!clk_data)) {
+               iounmap(base);
                return;
+       }
 
        clk_data->clks = kcalloc(num, sizeof(*clk_data->clks), GFP_KERNEL);
-       if (!clk_data->clks)
+       if (!clk_data->clks) {
+               iounmap(base);
                return;
+       }
 
        for (i = 0; i < num; i++) {
                struct hisi_mmc_clock *mmc_clk = &hi3620_mmc_clks[i];
diff --git a/drivers/clk/hisilicon/clk.c b/drivers/clk/hisilicon/clk.c
index 54d9fdc93599..53acaff32934 100644
--- a/drivers/clk/hisilicon/clk.c
+++ b/drivers/clk/hisilicon/clk.c
@@ -69,8 +69,10 @@ struct hisi_clock_data *hisi_clk_init(struct device_node *np,
        }
 
        clk_data = kzalloc(sizeof(*clk_data), GFP_KERNEL);
-       if (!clk_data)
+       if (!clk_data) {
+               iounmap(base);
                goto err;
+       }
 
        clk_data->base = base;
        clk_table = kcalloc(nr_clks, sizeof(*clk_table), GFP_KERNEL);
@@ -82,6 +84,7 @@ struct hisi_clock_data *hisi_clk_init(struct device_node *np,
        of_clk_add_provider(np, of_clk_src_onecell_get, &clk_data->clk_data);
        return clk_data;
 err_data:
+       iounmap(base);
        kfree(clk_data);
 err:
        return NULL;
-- 
2.17.1

Reply via email to