efi_loader: rework bootmgr/bootefi using load_image API

[oweals/u-boot.git] / tools / fit_common.c

diff --git a/tools/fit_common.c b/tools/fit_common.c
index 81ba698abd2594727522559c0ca8560268bdae20..9506390214ce028bbe0045d88161538ec652da2d 100644 (file)
--- a/tools/fit_common.c
+++ b/tools/fit_common.c
@@ -1,3 +1,4 @@
+// SPDX-License-Identifier: GPL-2.0+
 /*
  * (C) Copyright 2014
  * DENX Software Engineering
@@ -14,8 +15,6 @@
  *             some functions added to address abstraction
  *
  * All rights reserved.
- *
- * SPDX-License-Identifier:    GPL-2.0+
  */
 
 #include "imagetool.h"
@@ -27,7 +26,10 @@
 int fit_verify_header(unsigned char *ptr, int image_size,
                        struct image_tool_params *params)
 {
-       return fdt_check_header(ptr);
+       if (fdt_check_header(ptr) != EXIT_SUCCESS || !fit_check_format(ptr))
+               return EXIT_FAILURE;
+
+       return EXIT_SUCCESS;
 }
 
 int fit_check_image_types(uint8_t type)