diff options
| -rw-r--r-- | common/cmd_doc.c | 19 | ||||
| -rw-r--r-- | common/cmd_fdc.c | 15 | ||||
| -rw-r--r-- | common/cmd_ide.c | 19 | ||||
| -rw-r--r-- | common/cmd_nand.c | 38 | ||||
| -rw-r--r-- | common/cmd_scsi.c | 15 | ||||
| -rw-r--r-- | common/cmd_usb.c | 15 | 
6 files changed, 64 insertions, 57 deletions
| diff --git a/common/cmd_doc.c b/common/cmd_doc.c index 83aba3744..d7b2f535f 100644 --- a/common/cmd_doc.c +++ b/common/cmd_doc.c @@ -206,7 +206,7 @@ int do_docboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  	image_header_t *hdr;  	int rcode = 0;  #if defined(CONFIG_FIT) -	const void *fit_hdr; +	const void *fit_hdr = NULL;  #endif  	show_boot_progress (34); @@ -275,12 +275,6 @@ int do_docboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	case IMAGE_FORMAT_FIT:  		fit_hdr = (const void *)addr; -		if (!fit_check_format (fit_hdr)) { -			show_boot_progress (-130); -			puts ("** Bad FIT image format\n"); -			return 1; -		} -		show_boot_progress (131);  		puts ("Fit image detected...\n");  		cnt = fit_get_size (fit_hdr); @@ -304,8 +298,15 @@ int do_docboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	/* This cannot be done earlier, we need complete FIT image in RAM first */ -	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) -		fit_print_contents ((const void *)addr); +	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) { +		if (!fit_check_format (fit_hdr)) { +			show_boot_progress (-130); +			puts ("** Bad FIT image format\n"); +			return 1; +		} +		show_boot_progress (131); +		fit_print_contents (fit_hdr); +	}  #endif  	/* Loading ok, update default load address */ diff --git a/common/cmd_fdc.c b/common/cmd_fdc.c index 0293d186a..e4fbf29a2 100644 --- a/common/cmd_fdc.c +++ b/common/cmd_fdc.c @@ -787,7 +787,7 @@ int do_fdcboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  	char *ep;  	int rcode = 0;  #if defined(CONFIG_FIT) -	const void *fit_hdr; +	const void *fit_hdr = NULL;  #endif  	switch (argc) { @@ -847,10 +847,6 @@ int do_fdcboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	case IMAGE_FORMAT_FIT:  		fit_hdr = (const void *)addr; -		if (!fit_check_format (fit_hdr)) { -			puts ("** Bad FIT image format\n"); -			return 1; -		}  		puts ("Fit image detected...\n");  		imsize = fit_get_size (fit_hdr); @@ -879,8 +875,13 @@ int do_fdcboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	/* This cannot be done earlier, we need complete FIT image in RAM first */ -	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) -		fit_print_contents ((const void *)addr); +	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) { +		if (!fit_check_format (fit_hdr)) { +			puts ("** Bad FIT image format\n"); +			return 1; +		} +		fit_print_contents (fit_hdr); +	}  #endif  	/* Loading ok, update default load address */ diff --git a/common/cmd_ide.c b/common/cmd_ide.c index 65607022e..97a873d1c 100644 --- a/common/cmd_ide.c +++ b/common/cmd_ide.c @@ -367,7 +367,7 @@ int do_diskboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  	image_header_t *hdr;  	int rcode = 0;  #if defined(CONFIG_FIT) -	const void *fit_hdr; +	const void *fit_hdr = NULL;  #endif  	show_boot_progress (41); @@ -465,12 +465,6 @@ int do_diskboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	case IMAGE_FORMAT_FIT:  		fit_hdr = (const void *)addr; -		if (!fit_check_format (fit_hdr)) { -			show_boot_progress (-140); -			puts ("** Bad FIT image format\n"); -			return 1; -		} -		show_boot_progress (141);  		puts ("Fit image detected...\n");  		cnt = fit_get_size (fit_hdr); @@ -496,8 +490,15 @@ int do_diskboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	/* This cannot be done earlier, we need complete FIT image in RAM first */ -	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) -		fit_print_contents ((const void *)addr); +	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) { +		if (!fit_check_format (fit_hdr)) { +			show_boot_progress (-140); +			puts ("** Bad FIT image format\n"); +			return 1; +		} +		show_boot_progress (141); +		fit_print_contents (fit_hdr); +	}  #endif  	/* Loading ok, update default load address */ diff --git a/common/cmd_nand.c b/common/cmd_nand.c index 37198d21e..e20c1af35 100644 --- a/common/cmd_nand.c +++ b/common/cmd_nand.c @@ -484,7 +484,7 @@ static int nand_load_image(cmd_tbl_t *cmdtp, nand_info_t *nand,  	image_header_t *hdr;  	int jffs2 = 0;  #if defined(CONFIG_FIT) -	const void *fit_hdr; +	const void *fit_hdr = NULL;  #endif  	s = strchr(cmd, '.'); @@ -526,12 +526,6 @@ static int nand_load_image(cmd_tbl_t *cmdtp, nand_info_t *nand,  #if defined(CONFIG_FIT)  	case IMAGE_FORMAT_FIT:  		fit_hdr = (const void *)addr; -		if (!fit_check_format (fit_hdr)) { -			show_boot_progress (-150); -			puts ("** Bad FIT image format\n"); -			return 1; -		} -		show_boot_progress (151);  		puts ("Fit image detected...\n");  		cnt = fit_get_size (fit_hdr); @@ -564,8 +558,15 @@ static int nand_load_image(cmd_tbl_t *cmdtp, nand_info_t *nand,  #if defined(CONFIG_FIT)  	/* This cannot be done earlier, we need complete FIT image in RAM first */ -	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) -		fit_print_contents ((const void *)addr); +	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) { +		if (!fit_check_format (fit_hdr)) { +			show_boot_progress (-150); +			puts ("** Bad FIT image format\n"); +			return 1; +		} +		show_boot_progress (151); +		fit_print_contents (fit_hdr); +	}  #endif  	/* Loading ok, update default load address */ @@ -952,7 +953,7 @@ int do_nandboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  	image_header_t *hdr;  	int rcode = 0;  #if defined(CONFIG_FIT) -	const void *fit_hdr; +	const void *fit_hdr = NULL;  #endif  	show_boot_progress (52); @@ -1021,12 +1022,6 @@ int do_nandboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	case IMAGE_FORMAT_FIT:  		fit_hdr = (const void *)addr; -		if (!fit_check_format (fit_hdr)) { -			show_boot_progress (-150); -			puts ("** Bad FIT image format\n"); -			return 1; -		} -		show_boot_progress (151);  		puts ("Fit image detected...\n");  		cnt = fit_get_size (fit_hdr); @@ -1050,8 +1045,15 @@ int do_nandboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	/* This cannot be done earlier, we need complete FIT image in RAM first */ -	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) -		fit_print_contents ((const void *)addr); +	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) { +		if (!fit_check_format (fit_hdr)) { +			show_boot_progress (-150); +			puts ("** Bad FIT image format\n"); +			return 1; +		} +		show_boot_progress (151); +		fit_print_contents (fit_hdr); +	}  #endif  	/* Loading ok, update default load address */ diff --git a/common/cmd_scsi.c b/common/cmd_scsi.c index 69028f3b6..e648f7d60 100644 --- a/common/cmd_scsi.c +++ b/common/cmd_scsi.c @@ -212,7 +212,7 @@ int do_scsiboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  	image_header_t *hdr;  	int rcode = 0;  #if defined(CONFIG_FIT) -	const void *fit_hdr; +	const void *fit_hdr = NULL;  #endif  	switch (argc) { @@ -291,10 +291,6 @@ int do_scsiboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	case IMAGE_FORMAT_FIT:  		fit_hdr = (const void *)addr; -		if (!fit_check_format (fit_hdr)) { -			puts ("** Bad FIT image format\n"); -			return 1; -		}  		puts ("Fit image detected...\n");  		cnt = fit_get_size (fit_hdr); @@ -317,8 +313,13 @@ int do_scsiboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	/* This cannot be done earlier, we need complete FIT image in RAM first */ -	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) -		fit_print_contents ((const void *)addr); +	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) { +		if (!fit_check_format (fit_hdr)) { +			puts ("** Bad FIT image format\n"); +			return 1; +		} +		fit_print_contents (fit_hdr); +	}  #endif  	/* Loading ok, update default load address */ diff --git a/common/cmd_usb.c b/common/cmd_usb.c index 9be86b874..f2795d3f4 100644 --- a/common/cmd_usb.c +++ b/common/cmd_usb.c @@ -316,7 +316,7 @@ int do_usbboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  	image_header_t *hdr;  	block_dev_desc_t *stor_dev;  #if defined(CONFIG_FIT) -	const void *fit_hdr; +	const void *fit_hdr = NULL;  #endif  	switch (argc) { @@ -404,10 +404,6 @@ int do_usbboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	case IMAGE_FORMAT_FIT:  		fit_hdr = (const void *)addr; -		if (!fit_check_format (fit_hdr)) { -			puts ("** Bad FIT image format\n"); -			return 1; -		}  		puts ("Fit image detected...\n");  		cnt = fit_get_size (fit_hdr); @@ -430,8 +426,13 @@ int do_usbboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])  #if defined(CONFIG_FIT)  	/* This cannot be done earlier, we need complete FIT image in RAM first */ -	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) -		fit_print_contents ((const void *)addr); +	if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) { +		if (!fit_check_format (fit_hdr)) { +			puts ("** Bad FIT image format\n"); +			return 1; +		} +		fit_print_contents (fit_hdr); +	}  #endif  	/* Loading ok, update default load address */ |