UBI: use vmalloc for large buffers

UBI allocates temporary buffers of PEB size, which may be 256KiB.
Use vmalloc instead of kmalloc for such big temporary buffers.

Signed-off-by: Artem Bityutskiy <Artem.Bityutskiy@nokia.com>

1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/mtd/ubi/scan.c b/drivers/mtd/ubi/scan.c
index b24af2104a2a..cbd588d60168 100644
--- a/drivers/mtd/ubi/scan.c
+++ b/drivers/mtd/ubi/scan.c
@@ -356,7 +356,7 @@ static int compare_lebs(const struct ubi_device *ubi,
         /* Read the data of the copy and check the CRC */
 
         len = ubi32_to_cpu(vid_hdr->data_size);
-        buf = kmalloc(len, GFP_KERNEL);
+        buf = vmalloc(len);
         if (!buf) {
                 err = -ENOMEM;
                 goto out_free_vidh;
@@ -379,7 +379,7 @@ static int compare_lebs(const struct ubi_device *ubi,
                 bitflips = !!err;
         }
 
-        kfree(buf);
+        vfree(buf);
         ubi_free_vid_hdr(ubi, vidh);
 
         if (second_is_newer)
@@ -390,7 +390,7 @@ static int compare_lebs(const struct ubi_device *ubi,
         return second_is_newer | (bitflips << 1) | (corrupted << 2);
 
 out_free_buf:
-        kfree(buf);
+        vfree(buf);
 out_free_vidh:
         ubi_free_vid_hdr(ubi, vidh);
         ubi_assert(err < 0);