Merge "aboot : mdtp: Fixed heap overflow issue"
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.
 		*/