tools: kwbimage: Add support for more BINARY headers

The kwbimage v1 format supports multiple BINARY executable headers.
Add support for it into mkimage/kwbimage tool.

Signed-off-by: Pali Rohár <pali@kernel.org>
Reviewed-by: Marek Behún <marek.behun@nic.cz>
Reviewed-by: Chris Packham <judge.packham@gmail.com>
Reviewed-by: Stefan Roese <sr@denx.de>
Tested-by: Chris Packham <judge.packham@gmail.com>
This commit is contained in:
Pali Rohár 2021-07-23 11:14:09 +02:00 committed by Stefan Roese
parent 6458fd4fb1
commit d9fb82c508

View file

@ -942,6 +942,7 @@ static size_t image_headersz_v1(int *hasext)
{ {
struct image_cfg_element *binarye; struct image_cfg_element *binarye;
size_t headersz; size_t headersz;
int cfgi;
/* /*
* Calculate the size of the header and the size of the * Calculate the size of the header and the size of the
@ -949,21 +950,19 @@ static size_t image_headersz_v1(int *hasext)
*/ */
headersz = sizeof(struct main_hdr_v1); headersz = sizeof(struct main_hdr_v1);
if (image_count_options(IMAGE_CFG_BINARY) > 1) {
fprintf(stderr, "More than one binary blob, not supported\n");
return 0;
}
if (image_count_options(IMAGE_CFG_PAYLOAD) > 1) { if (image_count_options(IMAGE_CFG_PAYLOAD) > 1) {
fprintf(stderr, "More than one payload, not possible\n"); fprintf(stderr, "More than one payload, not possible\n");
return 0; return 0;
} }
binarye = image_find_option(IMAGE_CFG_BINARY); for (cfgi = 0; cfgi < cfgn; cfgi++) {
if (binarye) {
int ret; int ret;
struct stat s; struct stat s;
binarye = &image_cfg[cfgi];
if (binarye->type != IMAGE_CFG_BINARY)
continue;
ret = stat(binarye->binary.file, &s); ret = stat(binarye->binary.file, &s);
if (ret < 0) { if (ret < 0) {
char cwd[PATH_MAX]; char cwd[PATH_MAX];
@ -1018,10 +1017,10 @@ static size_t image_headersz_v1(int *hasext)
return ALIGN(headersz, 4096); return ALIGN(headersz, 4096);
} }
int add_binary_header_v1(uint8_t *cur) int add_binary_header_v1(uint8_t **cur, uint8_t **next_ext,
struct image_cfg_element *binarye)
{ {
struct image_cfg_element *binarye; struct opt_hdr_v1 *hdr = (struct opt_hdr_v1 *)*cur;
struct opt_hdr_v1 *hdr = (struct opt_hdr_v1 *)cur;
uint32_t *args; uint32_t *args;
size_t binhdrsz; size_t binhdrsz;
struct stat s; struct stat s;
@ -1029,11 +1028,6 @@ int add_binary_header_v1(uint8_t *cur)
FILE *bin; FILE *bin;
int ret; int ret;
binarye = image_find_option(IMAGE_CFG_BINARY);
if (!binarye)
return 0;
hdr->headertype = OPT_HDR_V1_BINARY_TYPE; hdr->headertype = OPT_HDR_V1_BINARY_TYPE;
bin = fopen(binarye->binary.file, "r"); bin = fopen(binarye->binary.file, "r");
@ -1055,17 +1049,17 @@ int add_binary_header_v1(uint8_t *cur)
hdr->headersz_lsb = cpu_to_le16(binhdrsz & 0xFFFF); hdr->headersz_lsb = cpu_to_le16(binhdrsz & 0xFFFF);
hdr->headersz_msb = (binhdrsz & 0xFFFF0000) >> 16; hdr->headersz_msb = (binhdrsz & 0xFFFF0000) >> 16;
cur += sizeof(struct opt_hdr_v1); *cur += sizeof(struct opt_hdr_v1);
args = (uint32_t *)cur; args = (uint32_t *)*cur;
*args = cpu_to_le32(binarye->binary.nargs); *args = cpu_to_le32(binarye->binary.nargs);
args++; args++;
for (argi = 0; argi < binarye->binary.nargs; argi++) for (argi = 0; argi < binarye->binary.nargs; argi++)
args[argi] = cpu_to_le32(binarye->binary.args[argi]); args[argi] = cpu_to_le32(binarye->binary.args[argi]);
cur += (binarye->binary.nargs + 1) * sizeof(uint32_t); *cur += (binarye->binary.nargs + 1) * sizeof(uint32_t);
ret = fread(cur, s.st_size, 1, bin); ret = fread(*cur, s.st_size, 1, bin);
if (ret != 1) { if (ret != 1) {
fprintf(stderr, fprintf(stderr,
"Could not read binary image %s\n", "Could not read binary image %s\n",
@ -1075,17 +1069,13 @@ int add_binary_header_v1(uint8_t *cur)
fclose(bin); fclose(bin);
cur += ALIGN(s.st_size, 4); *cur += ALIGN(s.st_size, 4);
/* *((uint32_t *)*cur) = 0x00000000;
* For now, we don't support more than one binary **next_ext = 1;
* header, and no other header types are *next_ext = *cur;
* supported. So, the binary header is necessarily the
* last one
*/
*((uint32_t *)cur) = 0x00000000;
cur += sizeof(uint32_t); *cur += sizeof(uint32_t);
return 0; return 0;
@ -1218,6 +1208,7 @@ static void *image_create_v1(size_t *imagesz, struct image_tool_params *params,
uint8_t *image, *cur; uint8_t *image, *cur;
int hasext = 0; int hasext = 0;
uint8_t *next_ext = NULL; uint8_t *next_ext = NULL;
int cfgi;
/* /*
* Calculate the size of the header and the size of the * Calculate the size of the header and the size of the
@ -1296,13 +1287,19 @@ static void *image_create_v1(size_t *imagesz, struct image_tool_params *params,
*/ */
secure_hdr = (struct secure_hdr_v1 *)cur; secure_hdr = (struct secure_hdr_v1 *)cur;
cur += sizeof(struct secure_hdr_v1); cur += sizeof(struct secure_hdr_v1);
*next_ext = 1;
next_ext = &secure_hdr->next; next_ext = &secure_hdr->next;
} }
#endif #endif
*next_ext = 1;
if (add_binary_header_v1(cur)) for (cfgi = 0; cfgi < cfgn; cfgi++) {
return NULL; e = &image_cfg[cfgi];
if (e->type != IMAGE_CFG_BINARY)
continue;
if (add_binary_header_v1(&cur, &next_ext, e))
return NULL;
}
#if defined(CONFIG_KWB_SECURE) #if defined(CONFIG_KWB_SECURE)
if (secure_hdr && add_secure_header_v1(params, ptr, payloadsz, if (secure_hdr && add_secure_header_v1(params, ptr, payloadsz,