diff options
Diffstat (limited to 'common/cmd_nand.c')
| -rw-r--r-- | common/cmd_nand.c | 38 | 
1 files changed, 20 insertions, 18 deletions
| 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 */ |