aboot: Check size of unsparsed images before flashing
Check size of unsparsed image against partition size before starting
to flash it to target.
Change-Id: Ie82248bbc3a2e04a5af5e1ab3f6e6bc21b8cb3c8
CRs-Fixed: 432780
diff --git a/app/aboot/aboot.c b/app/aboot/aboot.c
index d50be73..a2594d3 100644
--- a/app/aboot/aboot.c
+++ b/app/aboot/aboot.c
@@ -1401,6 +1401,11 @@
/* Read and skip over sparse image header */
sparse_header = (sparse_header_t *) data;
+ if ((sparse_header->total_blks * sparse_header->blk_sz) > size) {
+ fastboot_fail("size too large");
+ return;
+ }
+
data += sparse_header->file_hdr_sz;
if(sparse_header->file_hdr_sz > sizeof(sparse_header_t))
{