commit c512668af7e4141148781ec6b66f3c60a1c3fc00
author Sean Anderson <seanga2@gmail.com> Sat Oct 14 16:47:43 2023 -0400
committer Tom Rini <trini@konsulko.com> Tue Oct 17 20:50:52 2023 -0400
tree 99c53c4fa49a41cc9ce5051f85761c05fa8d2cc8
parent 5d5c7bdf89d10fcda24385c34da17cdd9f3c6a5f

arm: imx: Add function to validate i.MX8 containers

Add a function to abstract the common task of validating i.MX8 container
image headers.

Signed-off-by: Sean Anderson <seanga2@gmail.com>

arch/arm/mach-imx/spl_imx_romapi.c

diff --git a/arch/arm/mach-imx/spl_imx_romapi.c b/arch/arm/mach-imx/spl_imx_romapi.c
index c4a4185..80909ae 100644
--- a/arch/arm/mach-imx/spl_imx_romapi.c
+++ b/arch/arm/mach-imx/spl_imx_romapi.c
@@ -202,7 +202,8 @@
 
 	for (i = 0; i < size; i += 4) {
 		hdr = p + i;
-		if (*(hdr + 3) == 0x87 && *hdr == 0 && (*(hdr + 1) != 0 || *(hdr + 2) != 0))
+		if (valid_container_hdr((void *)hdr) &&
+		    (*(hdr + 1) != 0 || *(hdr + 2) != 0))
 			return p + i;
 	}