power: reset: zx-reboot: Unmap region obtained by of_iomap

Free memory mapping, if probe is not successful.

Signed-off-by: Arvind Yadav <arvind.yadav.cs@gmail.com>
Signed-off-by: Sebastian Reichel <sre@kernel.org>

diff --git a/drivers/power/reset/zx-reboot.c b/drivers/power/reset/zx-reboot.c
index a5b0096..b0b1eb3 100644 (file)
--- a/drivers/power/reset/zx-reboot.c
+++ b/drivers/power/reset/zx-reboot.c
@@ -58,9 +58,12 @@ static int zx_reboot_probe(struct platform_device *pdev)
        }
 
        err = register_restart_handler(&zx_restart_nb);
-       if (err)
+       if (err) {
+               iounmap(base);
+               iounmap(pcu_base);
                dev_err(&pdev->dev, "Register restart handler failed(err=%d)\n",
                        err);
+       }
 
        return err;
 }