From df3ff3726e3ed065a22efbe0c405de03d66fcc01 Mon Sep 17 00:00:00 2001
From: Joseph Chen <chenjh@rock-chips.com>
Date: Fri, 30 Aug 2019 09:44:06 +0800
Subject: [PATCH] cmd: android: round up image header size

Change-Id: I79e1b31ae2d018c0a8da030482f2ac0be44cb177
Signed-off-by: Joseph Chen <chenjh@rock-chips.com>
---
 cmd/android.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cmd/android.c b/cmd/android.c
index 50916b2759..721a67a0f8 100644
--- a/cmd/android.c
+++ b/cmd/android.c
@@ -37,13 +37,13 @@ static int do_android_print_hdr(cmd_tbl_t *cmdtp, int flag,
 		return -ENODEV;
 	}
 
-	hdr = memalign(ARCH_DMA_MINALIGN, sizeof(*hdr));
+	blkcnt = DIV_ROUND_UP(sizeof(*hdr), dev_desc->blksz);
+	hdr = memalign(ARCH_DMA_MINALIGN, dev_desc->blksz * blkcnt);
 	if (!hdr) {
 		printf("%s: out of memory!\n", __func__);
 		return -ENOMEM;
 	}
 
-	blkcnt = sizeof(*hdr) / dev_desc->blksz;
 	ret = blk_dread(dev_desc, part_info.start, blkcnt, hdr);
 	if (ret != blkcnt) {
 		printf("Failed to read %s sector, ret=%d\n", part_info.name, ret);