UBI: fix memory leak in checking code

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

UBI: fix memory leak in checking code
Reported-by: Eric Sesterhenn / Snakebyte <snakebyte@gmx.de> Signed-off-by: Artem Bityutskiy <Artem.Bityutskiy@nokia.com>
Artem Bityutskiy
1 parent cb32da0416
Showing 1 changed file with 3 additions and 1 deletions Side-by-side Diff
drivers/mtd/ubi/scan.c
@@ -1310,8 +1310,10 @@
 	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;
 	}
...	...	@@ -1310,8 +1310,10 @@
1310	1310	memset(buf, 1, ubi->peb_count);
1311	1311	for (pnum = 0; pnum < ubi->peb_count; pnum++) {
1312	1312	err = ubi_io_is_bad(ubi, pnum);
1313		- if (err < 0)
	1313	+ if (err < 0) {
	1314	+ kfree(buf);
1314	1315	return err;
	1316	+ }
1315	1317	else if (err)
1316	1318	buf[pnum] = 0;
1317	1319	}