image: Adjust the workings of fit_check_format()
At present this function does not accept a size for the FIT. This means that it must be read from the FIT itself, introducing potential security risk. Update the function to include a size parameter, which can be invalid, in which case fit_check_format() calculates it. For now no callers pass the size, but this can be updated later. Also adjust the return value to an error code so that all the different types of problems can be distinguished by the user. Signed-off-by: Simon Glass <sjg@chromium.org> Reported-by: Bruce Monroe <bruce.monroe@intel.com> Reported-by: Arie Haenel <arie.haenel@intel.com> Reported-by: Julien Lenoir <julien.lenoir@intel.com>
This commit is contained in:
@@ -286,7 +286,7 @@ int update_tftp(ulong addr, char *interface, char *devstring)
|
||||
got_update_file:
|
||||
fit = map_sysmem(addr, 0);
|
||||
|
||||
if (!fit_check_format((void *)fit)) {
|
||||
if (fit_check_format((void *)fit, IMAGE_SIZE_INVAL)) {
|
||||
printf("Bad FIT format of the update file, aborting "
|
||||
"auto-update\n");
|
||||
return 1;
|
||||
@@ -363,7 +363,7 @@ int fit_update(const void *fit)
|
||||
if (!fit)
|
||||
return -EINVAL;
|
||||
|
||||
if (!fit_check_format((void *)fit)) {
|
||||
if (fit_check_format((void *)fit, IMAGE_SIZE_INVAL)) {
|
||||
printf("Bad FIT format of the update file, aborting auto-update\n");
|
||||
return -EINVAL;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user