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

[thirdparty/u-boot.git] / 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 c4a4185eed1fb5c5cdad17494c7504e68b2a7171..80909aec3ea7d753ecc31ca4882ced668cd95b64 100644 (file)
--- a/arch/arm/mach-imx/spl_imx_romapi.c
+++ b/arch/arm/mach-imx/spl_imx_romapi.c
@@ -202,7 +202,8 @@ static u8 *search_container_header(u8 *p, int size)
 
        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;
        }