diff options
Diffstat (limited to 'scripts/imx/imx-image.c')
-rw-r--r-- | scripts/imx/imx-image.c | 99 |
1 files changed, 69 insertions, 30 deletions
diff --git a/scripts/imx/imx-image.c b/scripts/imx/imx-image.c index 294f51a902..6ebae22562 100644 --- a/scripts/imx/imx-image.c +++ b/scripts/imx/imx-image.c @@ -36,6 +36,7 @@ #include <include/filetype.h> #define FLASH_HEADER_OFFSET 0x400 +#define ARM_HEAD_SIZE_INDEX (ARM_HEAD_SIZE_OFFSET / sizeof(uint32_t)) /* * Conservative DCD element limit set to restriction v2 header size to @@ -46,7 +47,6 @@ static uint32_t dcdtable[MAX_DCD]; static int curdcd; -static int add_barebox_header; static int create_usb_image; static char *prgname; @@ -57,8 +57,7 @@ static char *prgname; */ -static uint32_t bb_header[] = { - 0xea0003fe, /* b 0x1000 */ +static uint32_t bb_header_aarch32[] = { 0xeafffffe, /* 1: b 1b */ 0xeafffffe, /* 1: b 1b */ 0xeafffffe, /* 1: b 1b */ @@ -66,6 +65,30 @@ static uint32_t bb_header[] = { 0xeafffffe, /* 1: b 1b */ 0xeafffffe, /* 1: b 1b */ 0xeafffffe, /* 1: b 1b */ + 0xeafffffe, /* 1: b 1b */ + 0x65726162, /* 'bare' */ + 0x00786f62, /* 'box\0' */ + 0x00000000, + 0x00000000, + 0x55555555, + 0x55555555, + 0x55555555, + 0x55555555, + 0x55555555, + 0x55555555, + 0x55555555, + 0x55555555, +}; + +static uint32_t bb_header_aarch64[] = { + 0x14000000, /* 1: b 1b */ + 0x14000000, /* 1: b 1b */ + 0x14000000, /* 1: b 1b */ + 0x14000000, /* 1: b 1b */ + 0x14000000, /* 1: b 1b */ + 0x14000000, /* 1: b 1b */ + 0x14000000, /* 1: b 1b */ + 0x14000000, /* 1: b 1b */ 0x65726162, /* 'bare' */ 0x00786f62, /* 'box\0' */ 0x00000000, @@ -207,18 +230,14 @@ static int add_srk(void *buf, int offset, uint32_t loadaddr, const char *srkfile static int dcd_ptr_offset; static uint32_t dcd_ptr_content; -static int add_header_v1(struct config_data *data, void *buf) +static size_t add_header_v1(struct config_data *data, void *buf) { struct imx_flash_header *hdr; int dcdsize = curdcd * sizeof(uint32_t); - uint32_t *psize = buf + ARM_HEAD_SIZE_OFFSET; int offset = data->image_dcd_offset; uint32_t loadaddr = data->image_load_addr; uint32_t imagesize = data->load_size; - if (add_barebox_header) - memcpy(buf, bb_header, sizeof(bb_header)); - buf += offset; hdr = buf; @@ -250,12 +269,9 @@ static int add_header_v1(struct config_data *data, void *buf) imagesize += CSF_LEN; } - if (add_barebox_header) - *psize = imagesize; - *(uint32_t *)buf = imagesize; - return 0; + return imagesize; } static int write_mem_v1(uint32_t addr, uint32_t val, int width, int set_bits, int clear_bits) @@ -283,18 +299,14 @@ static int write_mem_v1(uint32_t addr, uint32_t val, int width, int set_bits, in * ============================================================================ */ -static int add_header_v2(const struct config_data *data, void *buf) +static size_t add_header_v2(const struct config_data *data, void *buf) { struct imx_flash_header_v2 *hdr; int dcdsize = curdcd * sizeof(uint32_t); - uint32_t *psize = buf + ARM_HEAD_SIZE_OFFSET; int offset = data->image_dcd_offset; uint32_t loadaddr = data->image_load_addr; uint32_t imagesize = data->load_size; - if (add_barebox_header) - memcpy(buf, bb_header, sizeof(bb_header)); - buf += offset; hdr = buf; @@ -320,9 +332,6 @@ static int add_header_v2(const struct config_data *data, void *buf) hdr->boot_data.size += CSF_LEN; } - if (add_barebox_header) - *psize = hdr->boot_data.size; - hdr->dcd_header.tag = TAG_DCD_HEADER; hdr->dcd_header.length = htobe16(sizeof(uint32_t) + dcdsize); hdr->dcd_header.version = DCD_VERSION; @@ -331,7 +340,7 @@ static int add_header_v2(const struct config_data *data, void *buf) memcpy(buf, dcdtable, dcdsize); - return 0; + return imagesize; } static void usage(const char *prgname) @@ -666,6 +675,11 @@ static void *read_file(const char *filename, size_t *size) return buf; } +static bool cpu_is_aarch64(const struct config_data *data) +{ + return data->cpu_type == IMX_CPU_IMX8MQ; +} + int main(int argc, char *argv[]) { int opt, ret; @@ -679,12 +693,17 @@ int main(int argc, char *argv[]) int dcd_only = 0; int now = 0; int sign_image = 0; + int i, header_copies; + int add_barebox_header; + uint32_t barebox_image_size; struct config_data data = { .image_dcd_offset = 0xffffffff, .write_mem = write_mem, .check = check, .nop = nop, }; + uint32_t *bb_header; + size_t sizeof_bb_header; prgname = argv[0]; @@ -791,7 +810,7 @@ int main(int argc, char *argv[]) switch (data.header_version) { case 1: - add_header_v1(&data, buf); + barebox_image_size = add_header_v1(&data, buf); if (data.srkfile) { ret = add_srk(buf, data.image_dcd_offset, data.image_load_addr, data.srkfile); @@ -806,7 +825,7 @@ int main(int argc, char *argv[]) exit(1); } - add_header_v2(&data, buf); + barebox_image_size = add_header_v2(&data, buf); break; default: fprintf(stderr, "Congratulations! You're welcome to implement header version %d\n", @@ -814,6 +833,17 @@ int main(int argc, char *argv[]) exit(1); } + if (cpu_is_aarch64(&data)) { + bb_header = bb_header_aarch64; + sizeof_bb_header = sizeof(bb_header_aarch64); + } else { + bb_header = bb_header_aarch32; + sizeof_bb_header = sizeof(bb_header_aarch32); + } + + bb_header[0] = data.first_opcode; + bb_header[ARM_HEAD_SIZE_INDEX] = barebox_image_size; + infile = read_file(imagename, &insize); if (!infile) exit(1); @@ -824,14 +854,23 @@ int main(int argc, char *argv[]) exit(1); } - ret = xwrite(outfd, buf, HEADER_LEN); - if (ret < 0) { - perror("write"); - exit(1); - } + header_copies = (data.cpu_type == IMX_CPU_IMX35) ? 2 : 1; + + for (i = 0; i < header_copies; i++) { + ret = xwrite(outfd, add_barebox_header ? bb_header : buf, + sizeof_bb_header); + if (ret < 0) { + perror("write"); + exit(1); + } + + if (lseek(outfd, data.header_gap, SEEK_CUR) < 0) { + perror("lseek"); + exit(1); + } - if (data.cpu_type == IMX_CPU_IMX35) { - ret = xwrite(outfd, buf, HEADER_LEN); + ret = xwrite(outfd, buf + sizeof_bb_header, + HEADER_LEN - sizeof_bb_header); if (ret < 0) { perror("write"); exit(1); |