Djam/u-boot
1
0
Fork 0
mirror of https://github.com/u-boot/u-boot.git synced 2025-05-08 19:11:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

image: Add a function to find a script in an image

Browse source 
Split this functionality out of the 'source' command so it can be used
from another place. For now leave it where it is, but a future patch will
move it out of cmd/

Signed-off-by: Simon Glass <sjg@chromium.org>
This commit is contained in:
Simon Glass 2023-01-06 08:52:27 -06:00 • committed by Tom Rini
parent 32bab0eae5
commit 858fefd5fc
2 changed files with 136 additions and 107 deletions
Download patch file Download diff file Expand all files Collapse all files

61
cmd/source.c
View file

@ -24,7 +24,6 @@
#include <asm/byteorder.h>
#include <asm/io.h>
#if defined(CONFIG_FIT)
/**
* get_default_image() - Return default property from /images
*
@ -40,30 +39,24 @@ static const char *get_default_image(const void *fit)
return fdt_getprop(fit, images_noffset, FIT_DEFAULT_PROP, NULL);
}
#endif
int image_source_script(ulong addr, const char *fit_uname, const char *confname)
int image_locate_script(void *buf, int size, const char *fit_uname,
const char *confname, char **datap, uint *lenp)
{
ulong len;
#if defined(CONFIG_LEGACY_IMAGE_FORMAT)
const struct legacy_img_hdr *hdr;
#endif
u32 *data;
int verify;
void *buf;
#if defined(CONFIG_FIT)
const void* fit_hdr;
int noffset;
const void *fit_data;
size_t fit_len;
#endif
verify = env_get_yesno("verify");
buf = map_sysmem(addr, 0);
switch (genimg_get_format(buf)) {
#if defined(CONFIG_LEGACY_IMAGE_FORMAT)
case IMAGE_FORMAT_LEGACY:
if (IS_ENABLED(CONFIG_LEGACY_IMAGE_FORMAT)) {
hdr = buf;
if (!image_check_magic(hdr)) {
@ -91,21 +84,22 @@ int image_source_script(ulong addr, const char *fit_uname, const char *confname)
/* get length of script */
data = (u32 *)image_get_data(hdr);
if ((len = uimage_to_cpu (*data)) == 0) {
len = uimage_to_cpu(*data);
if (!len) {
puts("Empty Script\n");
return 1;
}
/*
* scripts are just multi-image files with one component, seek
* past the zero-terminated sequence of image lengths to get
* to the actual image data
* scripts are just multi-image files with one
* component, so seek past the zero-terminated sequence
* of image lengths to get to the actual image data
*/
while (*data++);
}
break;
#endif
#if defined(CONFIG_FIT)
case IMAGE_FORMAT_FIT:
if (IS_ENABLED(CONFIG_FIT)) {
fit_hdr = buf;
if (fit_check_format(fit_hdr, IMAGE_SIZE_INVAL)) {
puts("Bad FIT image format\n");
@ -128,7 +122,8 @@ int image_source_script(ulong addr, const char *fit_uname, const char *confname)
if (verify && fit_config_verify(fit_hdr, noffset))
return 1;
noffset = fit_conf_get_prop_node(fit_hdr, noffset,
noffset = fit_conf_get_prop_node(fit_hdr,
noffset,
FIT_SCRIPT_PROP,
IH_PHASE_NONE);
if (noffset < 0) {
@ -155,8 +150,9 @@ fallback:
}
}
if (!fit_image_check_type (fit_hdr, noffset, IH_TYPE_SCRIPT)) {
puts("Not a script image\n");
if (!fit_image_check_type(fit_hdr, noffset,
IH_TYPE_SCRIPT)) {
puts("Not a image image\n");
return 1;
}
@ -174,15 +170,34 @@ fallback:
data = (u32 *)fit_data;
len = (ulong)fit_len;
}
break;
#endif
default:
puts("Wrong image format for \"source\" command\n");
return 1;
return -EPERM;
}
debug("** Script length: %ld\n", len);
return run_command_list((char *)data, len, 0);
*datap = (char *)data;
*lenp = len;
return 0;
}
int image_source_script(ulong addr, const char *fit_uname, const char *confname)
{
char *data;
void *buf;
uint len;
int ret;
buf = map_sysmem(addr, 0);
ret = image_locate_script(buf, 0, fit_uname, confname, &data, &len);
unmap_sysmem(buf);
if (ret)
return CMD_RET_FAILURE;
debug("** Script length: %d\n", len);
return run_command_list(data, len, 0);
}
/**************************************************/

14
include/image.h
View file

@ -729,6 +729,20 @@ int fit_image_load(struct bootm_headers *images, ulong addr,
int image_source_script(ulong addr, const char *fit_uname,
const char *confname);
/**
* image_locate_script() - Locate the raw script in an image
*
* @buf: Address of image
* @size: Size of image in bytes
* @fit_uname: Node name of FIT image to read
* @confname: Node name of FIT config to read
* @datap: Returns pointer to raw script on success
* @lenp: Returns size of raw script on success
* @return 0 if OK, non-zero on error
*/
int image_locate_script(void *buf, int size, const char *fit_uname,
const char *confname, char **datap, uint *lenp);
/**
* fit_get_node_from_config() - Look up an image a FIT by type
*