UBI: fix memory leak in checking code

Reported-by: Eric Sesterhenn / Snakebyte <snakebyte@gmx.de>
Signed-off-by: Artem Bityutskiy <Artem.Bityutskiy@nokia.com>

1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/mtd/ubi/scan.c b/drivers/mtd/ubi/scan.c
index 473f3200b868..e4456869e753 100644
--- a/drivers/mtd/ubi/scan.c
+++ b/drivers/mtd/ubi/scan.c
@@ -1310,8 +1310,10 @@ static int paranoid_check_si(const struct ubi_device *ubi,
         memset(buf, 1, ubi->peb_count);
         for (pnum = 0; pnum < ubi->peb_count; pnum++) {
                 err = ubi_io_is_bad(ubi, pnum);
-                if (err < 0)
+                if (err < 0) {
+                        kfree(buf);
                         return err;
+                }
                 else if (err)
                         buf[pnum] = 0;
         }