platform: msm_shared: add check to for minimum size for ubi img.
Add check for minimum size should be more than, ubi magic size in
case of ubi image.
Change-Id: Ib9548cdf0972bf38c2796301e91916399a7ec93d
diff --git a/platform/msm_shared/flash-ubi.c b/platform/msm_shared/flash-ubi.c
index ca46876..e0836c6 100644
--- a/platform/msm_shared/flash-ubi.c
+++ b/platform/msm_shared/flash-ubi.c
@@ -855,6 +855,12 @@
/* Total size of valid data in peb */
peb_valid_sz = num_pages * page_size;
+ if (size < UBI_MAGIC_SIZE)
+ {
+ dprintf(CRITICAL, "flash_ubi_img: invalid size provided.\n");
+ return -1;
+ }
+
/*
* Check for oob access if any in img_peb.
*/