xref: /openbmc/u-boot/tools/imximage.c (revision 2ae23a28)
183d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
28edcde5eSStefano Babic /*
38edcde5eSStefano Babic  * (C) Copyright 2009
48edcde5eSStefano Babic  * Stefano Babic, DENX Software Engineering, sbabic@denx.de.
58edcde5eSStefano Babic  *
68edcde5eSStefano Babic  * (C) Copyright 2008
78edcde5eSStefano Babic  * Marvell Semiconductor <www.marvell.com>
88edcde5eSStefano Babic  * Written-by: Prafulla Wadaskar <prafulla@marvell.com>
98edcde5eSStefano Babic  */
108edcde5eSStefano Babic 
11f86ed6a8SGuilherme Maciel Ferreira #include "imagetool.h"
128edcde5eSStefano Babic #include <image.h>
138edcde5eSStefano Babic #include "imximage.h"
148edcde5eSStefano Babic 
150187c985SStefano Babic #define UNDEFINED 0xFFFFFFFF
160187c985SStefano Babic 
178edcde5eSStefano Babic /*
188edcde5eSStefano Babic  * Supported commands for configuration file
198edcde5eSStefano Babic  */
208edcde5eSStefano Babic static table_entry_t imximage_cmds[] = {
21f581e3a2SStefano Babic 	{CMD_BOOT_FROM,         "BOOT_FROM",            "boot command",	  },
226cb83829SMarek Vasut 	{CMD_BOOT_OFFSET,       "BOOT_OFFSET",          "Boot offset",	  },
230b7f7c33SAdrian Alonso 	{CMD_WRITE_DATA,        "DATA",                 "Reg Write Data", },
240b7f7c33SAdrian Alonso 	{CMD_WRITE_CLR_BIT,     "CLR_BIT",              "Reg clear bit",  },
253e0a71c1SPeng Fan 	{CMD_WRITE_SET_BIT,     "SET_BIT",              "Reg set bit",  },
260b7f7c33SAdrian Alonso 	{CMD_CHECK_BITS_SET,    "CHECK_BITS_SET",   "Reg Check bits set", },
270b7f7c33SAdrian Alonso 	{CMD_CHECK_BITS_CLR,    "CHECK_BITS_CLR",   "Reg Check bits clr", },
280187c985SStefano Babic 	{CMD_CSF,               "CSF",           "Command Sequence File", },
298a1edd7dSLiu Hui-R64343 	{CMD_IMAGE_VERSION,     "IMAGE_VERSION",        "image version",  },
30b55e4f48SPeng Fan 	{CMD_PLUGIN,            "PLUGIN",               "file plugin_addr",  },
318edcde5eSStefano Babic 	{-1,                    "",                     "",	          },
328edcde5eSStefano Babic };
338edcde5eSStefano Babic 
348edcde5eSStefano Babic /*
358edcde5eSStefano Babic  * Supported Boot options for configuration file
368edcde5eSStefano Babic  * this is needed to set the correct flash offset
378edcde5eSStefano Babic  */
38377e367aSStefano Babic static table_entry_t imximage_boot_offset[] = {
398edcde5eSStefano Babic 	{FLASH_OFFSET_ONENAND,	"onenand",	"OneNAND Flash",},
40bd25864cSDirk Behme 	{FLASH_OFFSET_NAND,	"nand",		"NAND Flash",	},
4119b409c0SDirk Behme 	{FLASH_OFFSET_NOR,	"nor",		"NOR Flash",	},
4219b409c0SDirk Behme 	{FLASH_OFFSET_SATA,	"sata",		"SATA Disk",	},
43bd25864cSDirk Behme 	{FLASH_OFFSET_SD,	"sd",		"SD Card",	},
44bd25864cSDirk Behme 	{FLASH_OFFSET_SPI,	"spi",		"SPI Flash",	},
459598f8c3SYe.Li 	{FLASH_OFFSET_QSPI,	"qspi",		"QSPI NOR Flash",},
468edcde5eSStefano Babic 	{-1,			"",		"Invalid",	},
478edcde5eSStefano Babic };
488edcde5eSStefano Babic 
498a1edd7dSLiu Hui-R64343 /*
50377e367aSStefano Babic  * Supported Boot options for configuration file
51377e367aSStefano Babic  * this is needed to determine the initial load size
52377e367aSStefano Babic  */
53377e367aSStefano Babic static table_entry_t imximage_boot_loadsize[] = {
54377e367aSStefano Babic 	{FLASH_LOADSIZE_ONENAND,	"onenand",	"OneNAND Flash",},
55377e367aSStefano Babic 	{FLASH_LOADSIZE_NAND,		"nand",		"NAND Flash",	},
56377e367aSStefano Babic 	{FLASH_LOADSIZE_NOR,		"nor",		"NOR Flash",	},
57377e367aSStefano Babic 	{FLASH_LOADSIZE_SATA,		"sata",		"SATA Disk",	},
58377e367aSStefano Babic 	{FLASH_LOADSIZE_SD,		"sd",		"SD Card",	},
59377e367aSStefano Babic 	{FLASH_LOADSIZE_SPI,		"spi",		"SPI Flash",	},
609598f8c3SYe.Li 	{FLASH_LOADSIZE_QSPI,		"qspi",		"QSPI NOR Flash",},
61377e367aSStefano Babic 	{-1,				"",		"Invalid",	},
62377e367aSStefano Babic };
63377e367aSStefano Babic 
64377e367aSStefano Babic /*
658a1edd7dSLiu Hui-R64343  * IMXIMAGE version definition for i.MX chips
668a1edd7dSLiu Hui-R64343  */
678a1edd7dSLiu Hui-R64343 static table_entry_t imximage_versions[] = {
688a1edd7dSLiu Hui-R64343 	{IMXIMAGE_V1,	"",	" (i.MX25/35/51 compatible)", },
690b7f7c33SAdrian Alonso 	{IMXIMAGE_V2,	"",	" (i.MX53/6/7 compatible)",   },
708a1edd7dSLiu Hui-R64343 	{-1,            "",     " (Invalid)",                 },
718a1edd7dSLiu Hui-R64343 };
728edcde5eSStefano Babic 
738edcde5eSStefano Babic static struct imx_header imximage_header;
748a1edd7dSLiu Hui-R64343 static uint32_t imximage_version;
750187c985SStefano Babic /*
760187c985SStefano Babic  * Image Vector Table Offset
770187c985SStefano Babic  * Initialized to a wrong not 4-bytes aligned address to
780187c985SStefano Babic  * check if it is was set by the cfg file.
790187c985SStefano Babic  */
800187c985SStefano Babic static uint32_t imximage_ivt_offset = UNDEFINED;
810187c985SStefano Babic static uint32_t imximage_csf_size = UNDEFINED;
82377e367aSStefano Babic /* Initial Load Region Size */
83377e367aSStefano Babic static uint32_t imximage_init_loadsize;
84b55e4f48SPeng Fan static uint32_t imximage_iram_free_start;
85b55e4f48SPeng Fan static uint32_t imximage_plugin_size;
86b55e4f48SPeng Fan static uint32_t plugin_image;
878a1edd7dSLiu Hui-R64343 
888a1edd7dSLiu Hui-R64343 static set_dcd_val_t set_dcd_val;
890b7f7c33SAdrian Alonso static set_dcd_param_t set_dcd_param;
908a1edd7dSLiu Hui-R64343 static set_dcd_rst_t set_dcd_rst;
918a1edd7dSLiu Hui-R64343 static set_imx_hdr_t set_imx_hdr;
924d5fa985STroy Kisky static uint32_t max_dcd_entries;
9324331982STroy Kisky static uint32_t *header_size_ptr;
940187c985SStefano Babic static uint32_t *csf_ptr;
958edcde5eSStefano Babic 
get_cfg_value(char * token,char * name,int linenr)968edcde5eSStefano Babic static uint32_t get_cfg_value(char *token, char *name,  int linenr)
978edcde5eSStefano Babic {
988edcde5eSStefano Babic 	char *endptr;
998edcde5eSStefano Babic 	uint32_t value;
1008edcde5eSStefano Babic 
1018edcde5eSStefano Babic 	errno = 0;
1028edcde5eSStefano Babic 	value = strtoul(token, &endptr, 16);
1038edcde5eSStefano Babic 	if (errno || (token == endptr)) {
1048edcde5eSStefano Babic 		fprintf(stderr, "Error: %s[%d] - Invalid hex data(%s)\n",
1058edcde5eSStefano Babic 			name,  linenr, token);
1068edcde5eSStefano Babic 		exit(EXIT_FAILURE);
1078edcde5eSStefano Babic 	}
1088edcde5eSStefano Babic 	return value;
1098edcde5eSStefano Babic }
1108edcde5eSStefano Babic 
detect_imximage_version(struct imx_header * imx_hdr)1118a1edd7dSLiu Hui-R64343 static uint32_t detect_imximage_version(struct imx_header *imx_hdr)
1128edcde5eSStefano Babic {
1138a1edd7dSLiu Hui-R64343 	imx_header_v1_t *hdr_v1 = &imx_hdr->header.hdr_v1;
1148a1edd7dSLiu Hui-R64343 	imx_header_v2_t *hdr_v2 = &imx_hdr->header.hdr_v2;
1158a1edd7dSLiu Hui-R64343 	flash_header_v1_t *fhdr_v1 = &hdr_v1->fhdr;
1168a1edd7dSLiu Hui-R64343 	flash_header_v2_t *fhdr_v2 = &hdr_v2->fhdr;
1178a1edd7dSLiu Hui-R64343 
1188a1edd7dSLiu Hui-R64343 	/* Try to detect V1 */
1198a1edd7dSLiu Hui-R64343 	if ((fhdr_v1->app_code_barker == APP_CODE_BARKER) &&
1208a1edd7dSLiu Hui-R64343 		(hdr_v1->dcd_table.preamble.barker == DCD_BARKER))
1218a1edd7dSLiu Hui-R64343 		return IMXIMAGE_V1;
1228a1edd7dSLiu Hui-R64343 
1238a1edd7dSLiu Hui-R64343 	/* Try to detect V2 */
1248a1edd7dSLiu Hui-R64343 	if ((fhdr_v2->header.tag == IVT_HEADER_TAG) &&
125b55e4f48SPeng Fan 		(hdr_v2->data.dcd_table.header.tag == DCD_HEADER_TAG))
126b55e4f48SPeng Fan 		return IMXIMAGE_V2;
127b55e4f48SPeng Fan 
128b55e4f48SPeng Fan 	if ((fhdr_v2->header.tag == IVT_HEADER_TAG) &&
129b55e4f48SPeng Fan 	    hdr_v2->boot_data.plugin)
1308a1edd7dSLiu Hui-R64343 		return IMXIMAGE_V2;
1318a1edd7dSLiu Hui-R64343 
1328a1edd7dSLiu Hui-R64343 	return IMXIMAGE_VER_INVALID;
1338edcde5eSStefano Babic }
1348edcde5eSStefano Babic 
err_imximage_version(int version)1358a1edd7dSLiu Hui-R64343 static void err_imximage_version(int version)
1368edcde5eSStefano Babic {
1378edcde5eSStefano Babic 	fprintf(stderr,
1388a1edd7dSLiu Hui-R64343 		"Error: Unsupported imximage version:%d\n", version);
1398a1edd7dSLiu Hui-R64343 
1408edcde5eSStefano Babic 	exit(EXIT_FAILURE);
1418edcde5eSStefano Babic }
1428edcde5eSStefano Babic 
set_dcd_val_v1(struct imx_header * imxhdr,char * name,int lineno,int fld,uint32_t value,uint32_t off)1438a1edd7dSLiu Hui-R64343 static void set_dcd_val_v1(struct imx_header *imxhdr, char *name, int lineno,
1448a1edd7dSLiu Hui-R64343 					int fld, uint32_t value, uint32_t off)
1458a1edd7dSLiu Hui-R64343 {
1468a1edd7dSLiu Hui-R64343 	dcd_v1_t *dcd_v1 = &imxhdr->header.hdr_v1.dcd_table;
1478edcde5eSStefano Babic 
1488a1edd7dSLiu Hui-R64343 	switch (fld) {
1498a1edd7dSLiu Hui-R64343 	case CFG_REG_SIZE:
1508a1edd7dSLiu Hui-R64343 		/* Byte, halfword, word */
1518a1edd7dSLiu Hui-R64343 		if ((value != 1) && (value != 2) && (value != 4)) {
1528a1edd7dSLiu Hui-R64343 			fprintf(stderr, "Error: %s[%d] - "
1538a1edd7dSLiu Hui-R64343 				"Invalid register size " "(%d)\n",
1548a1edd7dSLiu Hui-R64343 				name, lineno, value);
1558a1edd7dSLiu Hui-R64343 			exit(EXIT_FAILURE);
1568a1edd7dSLiu Hui-R64343 		}
1578a1edd7dSLiu Hui-R64343 		dcd_v1->addr_data[off].type = value;
1588a1edd7dSLiu Hui-R64343 		break;
1598a1edd7dSLiu Hui-R64343 	case CFG_REG_ADDRESS:
1608a1edd7dSLiu Hui-R64343 		dcd_v1->addr_data[off].addr = value;
1618a1edd7dSLiu Hui-R64343 		break;
1628a1edd7dSLiu Hui-R64343 	case CFG_REG_VALUE:
1638a1edd7dSLiu Hui-R64343 		dcd_v1->addr_data[off].value = value;
1648a1edd7dSLiu Hui-R64343 		break;
1658a1edd7dSLiu Hui-R64343 	default:
1668a1edd7dSLiu Hui-R64343 		break;
1678a1edd7dSLiu Hui-R64343 
1688a1edd7dSLiu Hui-R64343 	}
1698edcde5eSStefano Babic }
1708edcde5eSStefano Babic 
17161903b75STroy Kisky static struct dcd_v2_cmd *gd_last_cmd;
17261903b75STroy Kisky 
set_dcd_param_v2(struct imx_header * imxhdr,uint32_t dcd_len,int32_t cmd)1730b7f7c33SAdrian Alonso static void set_dcd_param_v2(struct imx_header *imxhdr, uint32_t dcd_len,
1740b7f7c33SAdrian Alonso 		int32_t cmd)
1750b7f7c33SAdrian Alonso {
176b55e4f48SPeng Fan 	dcd_v2_t *dcd_v2 = &imxhdr->header.hdr_v2.data.dcd_table;
17761903b75STroy Kisky 	struct dcd_v2_cmd *d = gd_last_cmd;
17861903b75STroy Kisky 	struct dcd_v2_cmd *d2;
17961903b75STroy Kisky 	int len;
18061903b75STroy Kisky 
18161903b75STroy Kisky 	if (!d)
18261903b75STroy Kisky 		d = &dcd_v2->dcd_cmd;
18361903b75STroy Kisky 	d2 = d;
18461903b75STroy Kisky 	len = be16_to_cpu(d->write_dcd_command.length);
18561903b75STroy Kisky 	if (len > 4)
18661903b75STroy Kisky 		d2 = (struct dcd_v2_cmd *)(((char *)d) + len);
1870b7f7c33SAdrian Alonso 
1880b7f7c33SAdrian Alonso 	switch (cmd) {
1890b7f7c33SAdrian Alonso 	case CMD_WRITE_DATA:
19061903b75STroy Kisky 		if ((d->write_dcd_command.tag == DCD_WRITE_DATA_COMMAND_TAG) &&
19161903b75STroy Kisky 		    (d->write_dcd_command.param == DCD_WRITE_DATA_PARAM))
19261903b75STroy Kisky 			break;
19361903b75STroy Kisky 		d = d2;
19461903b75STroy Kisky 		d->write_dcd_command.tag = DCD_WRITE_DATA_COMMAND_TAG;
19561903b75STroy Kisky 		d->write_dcd_command.length = cpu_to_be16(4);
19661903b75STroy Kisky 		d->write_dcd_command.param = DCD_WRITE_DATA_PARAM;
1970b7f7c33SAdrian Alonso 		break;
1980b7f7c33SAdrian Alonso 	case CMD_WRITE_CLR_BIT:
19961903b75STroy Kisky 		if ((d->write_dcd_command.tag == DCD_WRITE_DATA_COMMAND_TAG) &&
20061903b75STroy Kisky 		    (d->write_dcd_command.param == DCD_WRITE_CLR_BIT_PARAM))
20161903b75STroy Kisky 			break;
20261903b75STroy Kisky 		d = d2;
20361903b75STroy Kisky 		d->write_dcd_command.tag = DCD_WRITE_DATA_COMMAND_TAG;
20461903b75STroy Kisky 		d->write_dcd_command.length = cpu_to_be16(4);
20561903b75STroy Kisky 		d->write_dcd_command.param = DCD_WRITE_CLR_BIT_PARAM;
2060b7f7c33SAdrian Alonso 		break;
2073e0a71c1SPeng Fan 	case CMD_WRITE_SET_BIT:
2083e0a71c1SPeng Fan 		if ((d->write_dcd_command.tag == DCD_WRITE_DATA_COMMAND_TAG) &&
2093e0a71c1SPeng Fan 		    (d->write_dcd_command.param == DCD_WRITE_SET_BIT_PARAM))
2103e0a71c1SPeng Fan 			break;
2113e0a71c1SPeng Fan 		d = d2;
2123e0a71c1SPeng Fan 		d->write_dcd_command.tag = DCD_WRITE_DATA_COMMAND_TAG;
2133e0a71c1SPeng Fan 		d->write_dcd_command.length = cpu_to_be16(4);
2143e0a71c1SPeng Fan 		d->write_dcd_command.param = DCD_WRITE_SET_BIT_PARAM;
2153e0a71c1SPeng Fan 		break;
2160b7f7c33SAdrian Alonso 	/*
2170b7f7c33SAdrian Alonso 	 * Check data command only supports one entry,
2180b7f7c33SAdrian Alonso 	 */
2190b7f7c33SAdrian Alonso 	case CMD_CHECK_BITS_SET:
22061903b75STroy Kisky 		d = d2;
22161903b75STroy Kisky 		d->write_dcd_command.tag = DCD_CHECK_DATA_COMMAND_TAG;
22261903b75STroy Kisky 		d->write_dcd_command.length = cpu_to_be16(4);
22361903b75STroy Kisky 		d->write_dcd_command.param = DCD_CHECK_BITS_SET_PARAM;
2240b7f7c33SAdrian Alonso 		break;
2250b7f7c33SAdrian Alonso 	case CMD_CHECK_BITS_CLR:
22661903b75STroy Kisky 		d = d2;
22761903b75STroy Kisky 		d->write_dcd_command.tag = DCD_CHECK_DATA_COMMAND_TAG;
22861903b75STroy Kisky 		d->write_dcd_command.length = cpu_to_be16(4);
2290782a880SAdrian Alonso 		d->write_dcd_command.param = DCD_CHECK_BITS_CLR_PARAM;
2300b7f7c33SAdrian Alonso 		break;
2310b7f7c33SAdrian Alonso 	default:
2320b7f7c33SAdrian Alonso 		break;
2330b7f7c33SAdrian Alonso 	}
23461903b75STroy Kisky 	gd_last_cmd = d;
2350b7f7c33SAdrian Alonso }
2360b7f7c33SAdrian Alonso 
set_dcd_val_v2(struct imx_header * imxhdr,char * name,int lineno,int fld,uint32_t value,uint32_t off)2378a1edd7dSLiu Hui-R64343 static void set_dcd_val_v2(struct imx_header *imxhdr, char *name, int lineno,
2388a1edd7dSLiu Hui-R64343 					int fld, uint32_t value, uint32_t off)
2398a1edd7dSLiu Hui-R64343 {
24061903b75STroy Kisky 	struct dcd_v2_cmd *d = gd_last_cmd;
24161903b75STroy Kisky 	int len;
24261903b75STroy Kisky 
24361903b75STroy Kisky 	len = be16_to_cpu(d->write_dcd_command.length);
24461903b75STroy Kisky 	off = (len - 4) >> 3;
2458a1edd7dSLiu Hui-R64343 
2468a1edd7dSLiu Hui-R64343 	switch (fld) {
2478a1edd7dSLiu Hui-R64343 	case CFG_REG_ADDRESS:
24861903b75STroy Kisky 		d->addr_data[off].addr = cpu_to_be32(value);
2498a1edd7dSLiu Hui-R64343 		break;
2508a1edd7dSLiu Hui-R64343 	case CFG_REG_VALUE:
25161903b75STroy Kisky 		d->addr_data[off].value = cpu_to_be32(value);
25261903b75STroy Kisky 		off++;
25361903b75STroy Kisky 		d->write_dcd_command.length = cpu_to_be16((off << 3) + 4);
2548a1edd7dSLiu Hui-R64343 		break;
2558a1edd7dSLiu Hui-R64343 	default:
2568a1edd7dSLiu Hui-R64343 		break;
2578a1edd7dSLiu Hui-R64343 
2588a1edd7dSLiu Hui-R64343 	}
2598a1edd7dSLiu Hui-R64343 }
2608a1edd7dSLiu Hui-R64343 
2618a1edd7dSLiu Hui-R64343 /*
2628a1edd7dSLiu Hui-R64343  * Complete setting up the rest field of DCD of V1
2638a1edd7dSLiu Hui-R64343  * such as barker code and DCD data length.
2648a1edd7dSLiu Hui-R64343  */
set_dcd_rst_v1(struct imx_header * imxhdr,uint32_t dcd_len,char * name,int lineno)2658a1edd7dSLiu Hui-R64343 static void set_dcd_rst_v1(struct imx_header *imxhdr, uint32_t dcd_len,
2668a1edd7dSLiu Hui-R64343 						char *name, int lineno)
2678a1edd7dSLiu Hui-R64343 {
2688a1edd7dSLiu Hui-R64343 	dcd_v1_t *dcd_v1 = &imxhdr->header.hdr_v1.dcd_table;
2698a1edd7dSLiu Hui-R64343 
2708a1edd7dSLiu Hui-R64343 	dcd_v1->preamble.barker = DCD_BARKER;
2718a1edd7dSLiu Hui-R64343 	dcd_v1->preamble.length = dcd_len * sizeof(dcd_type_addr_data_t);
2728a1edd7dSLiu Hui-R64343 }
2738a1edd7dSLiu Hui-R64343 
2748a1edd7dSLiu Hui-R64343 /*
2758a1edd7dSLiu Hui-R64343  * Complete setting up the reset field of DCD of V2
2768a1edd7dSLiu Hui-R64343  * such as DCD tag, version, length, etc.
2778a1edd7dSLiu Hui-R64343  */
set_dcd_rst_v2(struct imx_header * imxhdr,uint32_t dcd_len,char * name,int lineno)2788a1edd7dSLiu Hui-R64343 static void set_dcd_rst_v2(struct imx_header *imxhdr, uint32_t dcd_len,
2798a1edd7dSLiu Hui-R64343 						char *name, int lineno)
2808a1edd7dSLiu Hui-R64343 {
281b55e4f48SPeng Fan 	if (!imxhdr->header.hdr_v2.boot_data.plugin) {
282b55e4f48SPeng Fan 		dcd_v2_t *dcd_v2 = &imxhdr->header.hdr_v2.data.dcd_table;
28361903b75STroy Kisky 		struct dcd_v2_cmd *d = gd_last_cmd;
28461903b75STroy Kisky 		int len;
28561903b75STroy Kisky 
28661903b75STroy Kisky 		if (!d)
28761903b75STroy Kisky 			d = &dcd_v2->dcd_cmd;
28861903b75STroy Kisky 		len = be16_to_cpu(d->write_dcd_command.length);
28961903b75STroy Kisky 		if (len > 4)
29061903b75STroy Kisky 			d = (struct dcd_v2_cmd *)(((char *)d) + len);
29161903b75STroy Kisky 
29261903b75STroy Kisky 		len = (char *)d - (char *)&dcd_v2->header;
2938a1edd7dSLiu Hui-R64343 		dcd_v2->header.tag = DCD_HEADER_TAG;
29461903b75STroy Kisky 		dcd_v2->header.length = cpu_to_be16(len);
2958a1edd7dSLiu Hui-R64343 		dcd_v2->header.version = DCD_VERSION;
2968a1edd7dSLiu Hui-R64343 	}
297b55e4f48SPeng Fan }
2988a1edd7dSLiu Hui-R64343 
set_imx_hdr_v1(struct imx_header * imxhdr,uint32_t dcd_len,uint32_t entry_point,uint32_t flash_offset)2998a1edd7dSLiu Hui-R64343 static void set_imx_hdr_v1(struct imx_header *imxhdr, uint32_t dcd_len,
300ad0826dcSTroy Kisky 		uint32_t entry_point, uint32_t flash_offset)
3018a1edd7dSLiu Hui-R64343 {
3028a1edd7dSLiu Hui-R64343 	imx_header_v1_t *hdr_v1 = &imxhdr->header.hdr_v1;
3038a1edd7dSLiu Hui-R64343 	flash_header_v1_t *fhdr_v1 = &hdr_v1->fhdr;
3048a1edd7dSLiu Hui-R64343 	dcd_v1_t *dcd_v1 = &hdr_v1->dcd_table;
305ab857f26STroy Kisky 	uint32_t hdr_base;
30624331982STroy Kisky 	uint32_t header_length = (((char *)&dcd_v1->addr_data[dcd_len].addr)
30724331982STroy Kisky 			- ((char *)imxhdr));
3088a1edd7dSLiu Hui-R64343 
3098a1edd7dSLiu Hui-R64343 	/* Set magic number */
3108a1edd7dSLiu Hui-R64343 	fhdr_v1->app_code_barker = APP_CODE_BARKER;
3118a1edd7dSLiu Hui-R64343 
31297f17fa6SMartin Kaiser 	hdr_base = entry_point - imximage_init_loadsize + flash_offset;
313ab857f26STroy Kisky 	fhdr_v1->app_dest_ptr = hdr_base - flash_offset;
314ad0826dcSTroy Kisky 	fhdr_v1->app_code_jump_vector = entry_point;
3158a1edd7dSLiu Hui-R64343 
316ab857f26STroy Kisky 	fhdr_v1->dcd_ptr_ptr = hdr_base + offsetof(flash_header_v1_t, dcd_ptr);
317ab857f26STroy Kisky 	fhdr_v1->dcd_ptr = hdr_base + offsetof(imx_header_v1_t, dcd_table);
3188a1edd7dSLiu Hui-R64343 
3198a1edd7dSLiu Hui-R64343 	/* Security feature are not supported */
3208a1edd7dSLiu Hui-R64343 	fhdr_v1->app_code_csf = 0;
3218a1edd7dSLiu Hui-R64343 	fhdr_v1->super_root_key = 0;
32224331982STroy Kisky 	header_size_ptr = (uint32_t *)(((char *)imxhdr) + header_length - 4);
3238a1edd7dSLiu Hui-R64343 }
3248a1edd7dSLiu Hui-R64343 
set_imx_hdr_v2(struct imx_header * imxhdr,uint32_t dcd_len,uint32_t entry_point,uint32_t flash_offset)3258a1edd7dSLiu Hui-R64343 static void set_imx_hdr_v2(struct imx_header *imxhdr, uint32_t dcd_len,
326ad0826dcSTroy Kisky 		uint32_t entry_point, uint32_t flash_offset)
3278a1edd7dSLiu Hui-R64343 {
3288a1edd7dSLiu Hui-R64343 	imx_header_v2_t *hdr_v2 = &imxhdr->header.hdr_v2;
3298a1edd7dSLiu Hui-R64343 	flash_header_v2_t *fhdr_v2 = &hdr_v2->fhdr;
330ab857f26STroy Kisky 	uint32_t hdr_base;
3318a1edd7dSLiu Hui-R64343 
3328a1edd7dSLiu Hui-R64343 	/* Set magic number */
3338a1edd7dSLiu Hui-R64343 	fhdr_v2->header.tag = IVT_HEADER_TAG; /* 0xD1 */
3348a1edd7dSLiu Hui-R64343 	fhdr_v2->header.length = cpu_to_be16(sizeof(flash_header_v2_t));
3358a1edd7dSLiu Hui-R64343 	fhdr_v2->header.version = IVT_VERSION; /* 0x40 */
3368a1edd7dSLiu Hui-R64343 
337b55e4f48SPeng Fan 	if (!hdr_v2->boot_data.plugin) {
338ad0826dcSTroy Kisky 		fhdr_v2->entry = entry_point;
339b55e4f48SPeng Fan 		fhdr_v2->reserved1 = 0;
340b55e4f48SPeng Fan 		fhdr_v2->reserved1 = 0;
341377e367aSStefano Babic 		hdr_base = entry_point - imximage_init_loadsize +
342377e367aSStefano Babic 			flash_offset;
343377e367aSStefano Babic 		fhdr_v2->self = hdr_base;
344b893c989SBaruch Siach 		if (dcd_len > 0)
345b55e4f48SPeng Fan 			fhdr_v2->dcd_ptr = hdr_base +
346b55e4f48SPeng Fan 				offsetof(imx_header_v2_t, data);
347b893c989SBaruch Siach 		else
348b893c989SBaruch Siach 			fhdr_v2->dcd_ptr = 0;
349ab857f26STroy Kisky 		fhdr_v2->boot_data_ptr = hdr_base
350ab857f26STroy Kisky 				+ offsetof(imx_header_v2_t, boot_data);
351377e367aSStefano Babic 		hdr_v2->boot_data.start = entry_point - imximage_init_loadsize;
3528a1edd7dSLiu Hui-R64343 
3538a1edd7dSLiu Hui-R64343 		fhdr_v2->csf = 0;
3540187c985SStefano Babic 
35524331982STroy Kisky 		header_size_ptr = &hdr_v2->boot_data.size;
3560187c985SStefano Babic 		csf_ptr = &fhdr_v2->csf;
357b55e4f48SPeng Fan 	} else {
358b55e4f48SPeng Fan 		imx_header_v2_t *next_hdr_v2;
359b55e4f48SPeng Fan 		flash_header_v2_t *next_fhdr_v2;
360b55e4f48SPeng Fan 
361b55e4f48SPeng Fan 		if (imximage_csf_size != 0) {
362b55e4f48SPeng Fan 			fprintf(stderr, "Error: Header v2: SECURE_BOOT is only supported in DCD mode!");
363b55e4f48SPeng Fan 			exit(EXIT_FAILURE);
364b55e4f48SPeng Fan 		}
365b55e4f48SPeng Fan 
366b55e4f48SPeng Fan 		fhdr_v2->entry = imximage_iram_free_start +
367b55e4f48SPeng Fan 			flash_offset + sizeof(flash_header_v2_t) +
368b55e4f48SPeng Fan 			sizeof(boot_data_t);
369b55e4f48SPeng Fan 
370b55e4f48SPeng Fan 		fhdr_v2->reserved1 = 0;
371b55e4f48SPeng Fan 		fhdr_v2->reserved2 = 0;
372b55e4f48SPeng Fan 		fhdr_v2->self = imximage_iram_free_start + flash_offset;
373b55e4f48SPeng Fan 
374b55e4f48SPeng Fan 		fhdr_v2->dcd_ptr = 0;
375b55e4f48SPeng Fan 
376b55e4f48SPeng Fan 		fhdr_v2->boot_data_ptr = fhdr_v2->self +
377b55e4f48SPeng Fan 				offsetof(imx_header_v2_t, boot_data);
378b55e4f48SPeng Fan 
379b55e4f48SPeng Fan 		hdr_v2->boot_data.start = imximage_iram_free_start;
380b55e4f48SPeng Fan 		/*
381b55e4f48SPeng Fan 		 * The actural size of plugin image is "imximage_plugin_size +
382b55e4f48SPeng Fan 		 * sizeof(flash_header_v2_t) + sizeof(boot_data_t)", plus the
383b55e4f48SPeng Fan 		 * flash_offset space.The ROM code only need to copy this size
384b55e4f48SPeng Fan 		 * to run the plugin code. However, later when copy the whole
385b55e4f48SPeng Fan 		 * U-Boot image to DDR, the ROM code use memcpy to copy the
386b55e4f48SPeng Fan 		 * first part of the image, and use the storage read function
387b55e4f48SPeng Fan 		 * to get the remaining part. This requires the dividing point
388b55e4f48SPeng Fan 		 * must be multiple of storage sector size. Here we set the
389b55e4f48SPeng Fan 		 * first section to be MAX_PLUGIN_CODE_SIZE(64KB) for this
390b55e4f48SPeng Fan 		 * purpose.
391b55e4f48SPeng Fan 		 */
392b55e4f48SPeng Fan 		hdr_v2->boot_data.size = MAX_PLUGIN_CODE_SIZE;
393b55e4f48SPeng Fan 
394b55e4f48SPeng Fan 		/* Security feature are not supported */
395b55e4f48SPeng Fan 		fhdr_v2->csf = 0;
396b55e4f48SPeng Fan 
397b55e4f48SPeng Fan 		next_hdr_v2 = (imx_header_v2_t *)((char *)hdr_v2 +
398b55e4f48SPeng Fan 			       imximage_plugin_size);
399b55e4f48SPeng Fan 
400b55e4f48SPeng Fan 		next_fhdr_v2 = &next_hdr_v2->fhdr;
401b55e4f48SPeng Fan 
402b55e4f48SPeng Fan 		next_fhdr_v2->header.tag = IVT_HEADER_TAG; /* 0xD1 */
403b55e4f48SPeng Fan 		next_fhdr_v2->header.length =
404b55e4f48SPeng Fan 			cpu_to_be16(sizeof(flash_header_v2_t));
405b55e4f48SPeng Fan 		next_fhdr_v2->header.version = IVT_VERSION; /* 0x40 */
406b55e4f48SPeng Fan 
407b55e4f48SPeng Fan 		next_fhdr_v2->entry = entry_point;
408b55e4f48SPeng Fan 		hdr_base = entry_point - sizeof(struct imx_header);
409b55e4f48SPeng Fan 		next_fhdr_v2->reserved1 = 0;
410b55e4f48SPeng Fan 		next_fhdr_v2->reserved2 = 0;
411b55e4f48SPeng Fan 		next_fhdr_v2->self = hdr_base + imximage_plugin_size;
412b55e4f48SPeng Fan 
413b55e4f48SPeng Fan 		next_fhdr_v2->dcd_ptr = 0;
414b55e4f48SPeng Fan 		next_fhdr_v2->boot_data_ptr = next_fhdr_v2->self +
415b55e4f48SPeng Fan 				offsetof(imx_header_v2_t, boot_data);
416b55e4f48SPeng Fan 
417b55e4f48SPeng Fan 		next_hdr_v2->boot_data.start = hdr_base - flash_offset;
418b55e4f48SPeng Fan 
419b55e4f48SPeng Fan 		header_size_ptr = &next_hdr_v2->boot_data.size;
420b55e4f48SPeng Fan 
421b55e4f48SPeng Fan 		next_hdr_v2->boot_data.plugin = 0;
422b55e4f48SPeng Fan 
423b55e4f48SPeng Fan 		next_fhdr_v2->csf = 0;
424b55e4f48SPeng Fan 	}
4258a1edd7dSLiu Hui-R64343 }
4268a1edd7dSLiu Hui-R64343 
set_hdr_func(void)42772048bc3SYork Sun static void set_hdr_func(void)
4288a1edd7dSLiu Hui-R64343 {
4298a1edd7dSLiu Hui-R64343 	switch (imximage_version) {
4308a1edd7dSLiu Hui-R64343 	case IMXIMAGE_V1:
4318a1edd7dSLiu Hui-R64343 		set_dcd_val = set_dcd_val_v1;
4320b7f7c33SAdrian Alonso 		set_dcd_param = NULL;
4338a1edd7dSLiu Hui-R64343 		set_dcd_rst = set_dcd_rst_v1;
4348a1edd7dSLiu Hui-R64343 		set_imx_hdr = set_imx_hdr_v1;
4354d5fa985STroy Kisky 		max_dcd_entries = MAX_HW_CFG_SIZE_V1;
4368a1edd7dSLiu Hui-R64343 		break;
4378a1edd7dSLiu Hui-R64343 	case IMXIMAGE_V2:
43861903b75STroy Kisky 		gd_last_cmd = NULL;
4398a1edd7dSLiu Hui-R64343 		set_dcd_val = set_dcd_val_v2;
4400b7f7c33SAdrian Alonso 		set_dcd_param = set_dcd_param_v2;
4418a1edd7dSLiu Hui-R64343 		set_dcd_rst = set_dcd_rst_v2;
4428a1edd7dSLiu Hui-R64343 		set_imx_hdr = set_imx_hdr_v2;
4434d5fa985STroy Kisky 		max_dcd_entries = MAX_HW_CFG_SIZE_V2;
4448a1edd7dSLiu Hui-R64343 		break;
4458a1edd7dSLiu Hui-R64343 	default:
4468a1edd7dSLiu Hui-R64343 		err_imximage_version(imximage_version);
4478a1edd7dSLiu Hui-R64343 		break;
4488a1edd7dSLiu Hui-R64343 	}
4498a1edd7dSLiu Hui-R64343 }
4508a1edd7dSLiu Hui-R64343 
print_hdr_v1(struct imx_header * imx_hdr)4518a1edd7dSLiu Hui-R64343 static void print_hdr_v1(struct imx_header *imx_hdr)
4528a1edd7dSLiu Hui-R64343 {
4538a1edd7dSLiu Hui-R64343 	imx_header_v1_t *hdr_v1 = &imx_hdr->header.hdr_v1;
4548a1edd7dSLiu Hui-R64343 	flash_header_v1_t *fhdr_v1 = &hdr_v1->fhdr;
4558a1edd7dSLiu Hui-R64343 	dcd_v1_t *dcd_v1 = &hdr_v1->dcd_table;
4568a1edd7dSLiu Hui-R64343 	uint32_t size, length, ver;
4578a1edd7dSLiu Hui-R64343 
4588a1edd7dSLiu Hui-R64343 	size = dcd_v1->preamble.length;
4598a1edd7dSLiu Hui-R64343 	if (size > (MAX_HW_CFG_SIZE_V1 * sizeof(dcd_type_addr_data_t))) {
4608a1edd7dSLiu Hui-R64343 		fprintf(stderr,
4618a1edd7dSLiu Hui-R64343 			"Error: Image corrupt DCD size %d exceed maximum %d\n",
4628a1edd7dSLiu Hui-R64343 			(uint32_t)(size / sizeof(dcd_type_addr_data_t)),
4638a1edd7dSLiu Hui-R64343 			MAX_HW_CFG_SIZE_V1);
4648a1edd7dSLiu Hui-R64343 		exit(EXIT_FAILURE);
4658a1edd7dSLiu Hui-R64343 	}
4668a1edd7dSLiu Hui-R64343 
4678a1edd7dSLiu Hui-R64343 	length = dcd_v1->preamble.length / sizeof(dcd_type_addr_data_t);
4688a1edd7dSLiu Hui-R64343 	ver = detect_imximage_version(imx_hdr);
4698a1edd7dSLiu Hui-R64343 
4708a1edd7dSLiu Hui-R64343 	printf("Image Type:   Freescale IMX Boot Image\n");
4718a1edd7dSLiu Hui-R64343 	printf("Image Ver:    %x", ver);
4728a1edd7dSLiu Hui-R64343 	printf("%s\n", get_table_entry_name(imximage_versions, NULL, ver));
4738a1edd7dSLiu Hui-R64343 	printf("Data Size:    ");
4748a1edd7dSLiu Hui-R64343 	genimg_print_size(dcd_v1->addr_data[length].type);
4758a1edd7dSLiu Hui-R64343 	printf("Load Address: %08x\n", (uint32_t)fhdr_v1->app_dest_ptr);
4768a1edd7dSLiu Hui-R64343 	printf("Entry Point:  %08x\n", (uint32_t)fhdr_v1->app_code_jump_vector);
4778a1edd7dSLiu Hui-R64343 }
4788a1edd7dSLiu Hui-R64343 
print_hdr_v2(struct imx_header * imx_hdr)4798a1edd7dSLiu Hui-R64343 static void print_hdr_v2(struct imx_header *imx_hdr)
4808a1edd7dSLiu Hui-R64343 {
4818a1edd7dSLiu Hui-R64343 	imx_header_v2_t *hdr_v2 = &imx_hdr->header.hdr_v2;
4828a1edd7dSLiu Hui-R64343 	flash_header_v2_t *fhdr_v2 = &hdr_v2->fhdr;
483b55e4f48SPeng Fan 	dcd_v2_t *dcd_v2 = &hdr_v2->data.dcd_table;
484b55e4f48SPeng Fan 	uint32_t size, version, plugin;
4858a1edd7dSLiu Hui-R64343 
486b55e4f48SPeng Fan 	plugin = hdr_v2->boot_data.plugin;
487b55e4f48SPeng Fan 	if (!plugin) {
488835c30e3STroy Kisky 		size = be16_to_cpu(dcd_v2->header.length);
489b55e4f48SPeng Fan 		if (size > (MAX_HW_CFG_SIZE_V2 * sizeof(dcd_addr_data_t))) {
4908a1edd7dSLiu Hui-R64343 			fprintf(stderr,
4918a1edd7dSLiu Hui-R64343 				"Error: Image corrupt DCD size %d exceed maximum %d\n",
4928a1edd7dSLiu Hui-R64343 				(uint32_t)(size / sizeof(dcd_addr_data_t)),
4938a1edd7dSLiu Hui-R64343 				MAX_HW_CFG_SIZE_V2);
4948a1edd7dSLiu Hui-R64343 			exit(EXIT_FAILURE);
4958a1edd7dSLiu Hui-R64343 		}
496b55e4f48SPeng Fan 	}
4978a1edd7dSLiu Hui-R64343 
4988a1edd7dSLiu Hui-R64343 	version = detect_imximage_version(imx_hdr);
4998a1edd7dSLiu Hui-R64343 
5008a1edd7dSLiu Hui-R64343 	printf("Image Type:   Freescale IMX Boot Image\n");
5018a1edd7dSLiu Hui-R64343 	printf("Image Ver:    %x", version);
5028a1edd7dSLiu Hui-R64343 	printf("%s\n", get_table_entry_name(imximage_versions, NULL, version));
503b55e4f48SPeng Fan 	printf("Mode:         %s\n", plugin ? "PLUGIN" : "DCD");
504b55e4f48SPeng Fan 	if (!plugin) {
5058a1edd7dSLiu Hui-R64343 		printf("Data Size:    ");
5068a1edd7dSLiu Hui-R64343 		genimg_print_size(hdr_v2->boot_data.size);
5078a1edd7dSLiu Hui-R64343 		printf("Load Address: %08x\n", (uint32_t)fhdr_v2->boot_data_ptr);
5088a1edd7dSLiu Hui-R64343 		printf("Entry Point:  %08x\n", (uint32_t)fhdr_v2->entry);
509*7e1a0483SHolger Dengler 		if (fhdr_v2->csf) {
510e5491f3eSEric Nelson 			uint16_t dcdlen;
511e5491f3eSEric Nelson 			int offs;
512e5491f3eSEric Nelson 
513e5491f3eSEric Nelson 			dcdlen = hdr_v2->data.dcd_table.header.length;
514e5491f3eSEric Nelson 			offs = (char *)&hdr_v2->data.dcd_table
515e5491f3eSEric Nelson 				- (char *)hdr_v2;
516e5491f3eSEric Nelson 
517*7e1a0483SHolger Dengler 			/*
518*7e1a0483SHolger Dengler 			 * The HAB block is the first part of the image, from
519*7e1a0483SHolger Dengler 			 * start of IVT header (fhdr_v2->self) to the start of
520*7e1a0483SHolger Dengler 			 * the CSF block (fhdr_v2->csf). So HAB size is
521*7e1a0483SHolger Dengler 			 * calculated as:
522*7e1a0483SHolger Dengler 			 * HAB_size = fhdr_v2->csf - fhdr_v2->self
523*7e1a0483SHolger Dengler 			 */
5248519c9c9SRasmus Villemoes 			printf("HAB Blocks:   0x%08x 0x%08x 0x%08x\n",
5250187c985SStefano Babic 			       (uint32_t)fhdr_v2->self, 0,
526*7e1a0483SHolger Dengler 			       (uint32_t)(fhdr_v2->csf - fhdr_v2->self));
527af1b492dSBryan O'Donoghue 			printf("DCD Blocks:   0x00910000 0x%08x 0x%08x\n",
528e5491f3eSEric Nelson 			       offs, be16_to_cpu(dcdlen));
5290187c985SStefano Babic 		}
530b55e4f48SPeng Fan 	} else {
531b55e4f48SPeng Fan 		imx_header_v2_t *next_hdr_v2;
532b55e4f48SPeng Fan 		flash_header_v2_t *next_fhdr_v2;
533b55e4f48SPeng Fan 
534b55e4f48SPeng Fan 		/*First Header*/
535b55e4f48SPeng Fan 		printf("Plugin Data Size:     ");
536b55e4f48SPeng Fan 		genimg_print_size(hdr_v2->boot_data.size);
537b55e4f48SPeng Fan 		printf("Plugin Code Size:     ");
538b55e4f48SPeng Fan 		genimg_print_size(imximage_plugin_size);
539b55e4f48SPeng Fan 		printf("Plugin Load Address:  %08x\n", hdr_v2->boot_data.start);
540b55e4f48SPeng Fan 		printf("Plugin Entry Point:   %08x\n", (uint32_t)fhdr_v2->entry);
541b55e4f48SPeng Fan 
542b55e4f48SPeng Fan 		/*Second Header*/
543b55e4f48SPeng Fan 		next_hdr_v2 = (imx_header_v2_t *)((char *)hdr_v2 +
544b55e4f48SPeng Fan 				imximage_plugin_size);
545b55e4f48SPeng Fan 		next_fhdr_v2 = &next_hdr_v2->fhdr;
546b55e4f48SPeng Fan 		printf("U-Boot Data Size:     ");
547b55e4f48SPeng Fan 		genimg_print_size(next_hdr_v2->boot_data.size);
548b55e4f48SPeng Fan 		printf("U-Boot Load Address:  %08x\n",
549b55e4f48SPeng Fan 		       next_hdr_v2->boot_data.start);
550b55e4f48SPeng Fan 		printf("U-Boot Entry Point:   %08x\n",
551b55e4f48SPeng Fan 		       (uint32_t)next_fhdr_v2->entry);
552b55e4f48SPeng Fan 	}
553b55e4f48SPeng Fan }
554b55e4f48SPeng Fan 
copy_plugin_code(struct imx_header * imxhdr,char * plugin_file)555b55e4f48SPeng Fan static void copy_plugin_code(struct imx_header *imxhdr, char *plugin_file)
556b55e4f48SPeng Fan {
5572a380cccSPeng Fan 	int ifd;
558b55e4f48SPeng Fan 	struct stat sbuf;
559b55e4f48SPeng Fan 	char *plugin_buf = imxhdr->header.hdr_v2.data.plugin_code;
560b55e4f48SPeng Fan 	char *ptr;
561b55e4f48SPeng Fan 
562b55e4f48SPeng Fan 	ifd = open(plugin_file, O_RDONLY|O_BINARY);
5632a380cccSPeng Fan 	if (ifd < 0) {
5642a380cccSPeng Fan 		fprintf(stderr, "Can't open %s: %s\n",
5652a380cccSPeng Fan 			plugin_file,
5662a380cccSPeng Fan 			strerror(errno));
5672a380cccSPeng Fan 		exit(EXIT_FAILURE);
5682a380cccSPeng Fan 	}
5692a380cccSPeng Fan 
570b55e4f48SPeng Fan 	if (fstat(ifd, &sbuf) < 0) {
571b55e4f48SPeng Fan 		fprintf(stderr, "Can't stat %s: %s\n",
572b55e4f48SPeng Fan 			plugin_file,
573b55e4f48SPeng Fan 			strerror(errno));
574b55e4f48SPeng Fan 		exit(EXIT_FAILURE);
575b55e4f48SPeng Fan 	}
576b55e4f48SPeng Fan 
577b55e4f48SPeng Fan 	ptr = mmap(0, sbuf.st_size, PROT_READ, MAP_SHARED, ifd, 0);
578b55e4f48SPeng Fan 	if (ptr == MAP_FAILED) {
579b55e4f48SPeng Fan 		fprintf(stderr, "Can't read %s: %s\n",
580b55e4f48SPeng Fan 			plugin_file,
581b55e4f48SPeng Fan 			strerror(errno));
582b55e4f48SPeng Fan 		exit(EXIT_FAILURE);
583b55e4f48SPeng Fan 	}
584b55e4f48SPeng Fan 
585b55e4f48SPeng Fan 	if (sbuf.st_size > MAX_PLUGIN_CODE_SIZE) {
586b55e4f48SPeng Fan 		printf("plugin binary size too large\n");
587b55e4f48SPeng Fan 		exit(EXIT_FAILURE);
588b55e4f48SPeng Fan 	}
589b55e4f48SPeng Fan 
590b55e4f48SPeng Fan 	memcpy(plugin_buf, ptr, sbuf.st_size);
591b55e4f48SPeng Fan 	imximage_plugin_size = sbuf.st_size;
592b55e4f48SPeng Fan 
593b55e4f48SPeng Fan 	(void) munmap((void *)ptr, sbuf.st_size);
594b55e4f48SPeng Fan 	(void) close(ifd);
595b55e4f48SPeng Fan 
596b55e4f48SPeng Fan 	imxhdr->header.hdr_v2.boot_data.plugin = 1;
5978a1edd7dSLiu Hui-R64343 }
5988a1edd7dSLiu Hui-R64343 
parse_cfg_cmd(struct imx_header * imxhdr,int32_t cmd,char * token,char * name,int lineno,int fld,int dcd_len)5998a1edd7dSLiu Hui-R64343 static void parse_cfg_cmd(struct imx_header *imxhdr, int32_t cmd, char *token,
6008a1edd7dSLiu Hui-R64343 				char *name, int lineno, int fld, int dcd_len)
6018a1edd7dSLiu Hui-R64343 {
6028a1edd7dSLiu Hui-R64343 	int value;
6038a1edd7dSLiu Hui-R64343 	static int cmd_ver_first = ~0;
6048a1edd7dSLiu Hui-R64343 
6058a1edd7dSLiu Hui-R64343 	switch (cmd) {
6068a1edd7dSLiu Hui-R64343 	case CMD_IMAGE_VERSION:
6078a1edd7dSLiu Hui-R64343 		imximage_version = get_cfg_value(token, name, lineno);
6088a1edd7dSLiu Hui-R64343 		if (cmd_ver_first == 0) {
6098a1edd7dSLiu Hui-R64343 			fprintf(stderr, "Error: %s[%d] - IMAGE_VERSION "
6108a1edd7dSLiu Hui-R64343 				"command need be the first before other "
6118a1edd7dSLiu Hui-R64343 				"valid command in the file\n", name, lineno);
6128a1edd7dSLiu Hui-R64343 			exit(EXIT_FAILURE);
6138a1edd7dSLiu Hui-R64343 		}
6148a1edd7dSLiu Hui-R64343 		cmd_ver_first = 1;
61572048bc3SYork Sun 		set_hdr_func();
6168a1edd7dSLiu Hui-R64343 		break;
6178a1edd7dSLiu Hui-R64343 	case CMD_BOOT_FROM:
618377e367aSStefano Babic 		imximage_ivt_offset = get_table_entry_id(imximage_boot_offset,
6198a1edd7dSLiu Hui-R64343 					"imximage boot option", token);
6203150f92cSStefano Babic 		if (imximage_ivt_offset == -1) {
6218a1edd7dSLiu Hui-R64343 			fprintf(stderr, "Error: %s[%d] -Invalid boot device"
6228a1edd7dSLiu Hui-R64343 				"(%s)\n", name, lineno, token);
6238a1edd7dSLiu Hui-R64343 			exit(EXIT_FAILURE);
6248a1edd7dSLiu Hui-R64343 		}
625377e367aSStefano Babic 
626377e367aSStefano Babic 		imximage_init_loadsize =
627377e367aSStefano Babic 			get_table_entry_id(imximage_boot_loadsize,
628377e367aSStefano Babic 					   "imximage boot option", token);
629377e367aSStefano Babic 
630377e367aSStefano Babic 		if (imximage_init_loadsize == -1) {
631377e367aSStefano Babic 			fprintf(stderr,
632377e367aSStefano Babic 				"Error: %s[%d] -Invalid boot device(%s)\n",
633377e367aSStefano Babic 				name, lineno, token);
634377e367aSStefano Babic 			exit(EXIT_FAILURE);
635377e367aSStefano Babic 		}
63601390affSStefano Babic 
63701390affSStefano Babic 		/*
63801390affSStefano Babic 		 * The SOC loads from the storage starting at address 0
63901390affSStefano Babic 		 * then ensures that the load size contains the offset
64001390affSStefano Babic 		 */
64101390affSStefano Babic 		if (imximage_init_loadsize < imximage_ivt_offset)
64201390affSStefano Babic 			imximage_init_loadsize = imximage_ivt_offset;
6438a1edd7dSLiu Hui-R64343 		if (unlikely(cmd_ver_first != 1))
6448a1edd7dSLiu Hui-R64343 			cmd_ver_first = 0;
6458a1edd7dSLiu Hui-R64343 		break;
6466cb83829SMarek Vasut 	case CMD_BOOT_OFFSET:
6473150f92cSStefano Babic 		imximage_ivt_offset = get_cfg_value(token, name, lineno);
6486cb83829SMarek Vasut 		if (unlikely(cmd_ver_first != 1))
6496cb83829SMarek Vasut 			cmd_ver_first = 0;
6506cb83829SMarek Vasut 		break;
6510b7f7c33SAdrian Alonso 	case CMD_WRITE_DATA:
6520b7f7c33SAdrian Alonso 	case CMD_WRITE_CLR_BIT:
6533e0a71c1SPeng Fan 	case CMD_WRITE_SET_BIT:
6540b7f7c33SAdrian Alonso 	case CMD_CHECK_BITS_SET:
6550b7f7c33SAdrian Alonso 	case CMD_CHECK_BITS_CLR:
6568a1edd7dSLiu Hui-R64343 		value = get_cfg_value(token, name, lineno);
6570b7f7c33SAdrian Alonso 		if (set_dcd_param)
6580b7f7c33SAdrian Alonso 			(*set_dcd_param)(imxhdr, dcd_len, cmd);
6598a1edd7dSLiu Hui-R64343 		(*set_dcd_val)(imxhdr, name, lineno, fld, value, dcd_len);
6608a1edd7dSLiu Hui-R64343 		if (unlikely(cmd_ver_first != 1))
6618a1edd7dSLiu Hui-R64343 			cmd_ver_first = 0;
6628a1edd7dSLiu Hui-R64343 		break;
6630187c985SStefano Babic 	case CMD_CSF:
6640187c985SStefano Babic 		if (imximage_version != 2) {
6650187c985SStefano Babic 			fprintf(stderr,
6660187c985SStefano Babic 				"Error: %s[%d] - CSF only supported for VERSION 2(%s)\n",
6670187c985SStefano Babic 				name, lineno, token);
6680187c985SStefano Babic 			exit(EXIT_FAILURE);
6690187c985SStefano Babic 		}
6700187c985SStefano Babic 		imximage_csf_size = get_cfg_value(token, name, lineno);
6710187c985SStefano Babic 		if (unlikely(cmd_ver_first != 1))
6720187c985SStefano Babic 			cmd_ver_first = 0;
6730187c985SStefano Babic 		break;
674b55e4f48SPeng Fan 	case CMD_PLUGIN:
675b55e4f48SPeng Fan 		plugin_image = 1;
676b55e4f48SPeng Fan 		copy_plugin_code(imxhdr, token);
677b55e4f48SPeng Fan 		break;
6788a1edd7dSLiu Hui-R64343 	}
6798a1edd7dSLiu Hui-R64343 }
6808a1edd7dSLiu Hui-R64343 
parse_cfg_fld(struct imx_header * imxhdr,int32_t * cmd,char * token,char * name,int lineno,int fld,int * dcd_len)6818a1edd7dSLiu Hui-R64343 static void parse_cfg_fld(struct imx_header *imxhdr, int32_t *cmd,
6828a1edd7dSLiu Hui-R64343 		char *token, char *name, int lineno, int fld, int *dcd_len)
6838a1edd7dSLiu Hui-R64343 {
6848a1edd7dSLiu Hui-R64343 	int value;
6858a1edd7dSLiu Hui-R64343 
6868a1edd7dSLiu Hui-R64343 	switch (fld) {
6878a1edd7dSLiu Hui-R64343 	case CFG_COMMAND:
6888a1edd7dSLiu Hui-R64343 		*cmd = get_table_entry_id(imximage_cmds,
6898a1edd7dSLiu Hui-R64343 			"imximage commands", token);
6908a1edd7dSLiu Hui-R64343 		if (*cmd < 0) {
6918a1edd7dSLiu Hui-R64343 			fprintf(stderr, "Error: %s[%d] - Invalid command"
6928a1edd7dSLiu Hui-R64343 			"(%s)\n", name, lineno, token);
6938a1edd7dSLiu Hui-R64343 			exit(EXIT_FAILURE);
6948a1edd7dSLiu Hui-R64343 		}
6958a1edd7dSLiu Hui-R64343 		break;
6968a1edd7dSLiu Hui-R64343 	case CFG_REG_SIZE:
6978a1edd7dSLiu Hui-R64343 		parse_cfg_cmd(imxhdr, *cmd, token, name, lineno, fld, *dcd_len);
6988a1edd7dSLiu Hui-R64343 		break;
6998a1edd7dSLiu Hui-R64343 	case CFG_REG_ADDRESS:
7008a1edd7dSLiu Hui-R64343 	case CFG_REG_VALUE:
7010b7f7c33SAdrian Alonso 		switch(*cmd) {
7020b7f7c33SAdrian Alonso 		case CMD_WRITE_DATA:
7030b7f7c33SAdrian Alonso 		case CMD_WRITE_CLR_BIT:
7043e0a71c1SPeng Fan 		case CMD_WRITE_SET_BIT:
7050b7f7c33SAdrian Alonso 		case CMD_CHECK_BITS_SET:
7060b7f7c33SAdrian Alonso 		case CMD_CHECK_BITS_CLR:
7078a1edd7dSLiu Hui-R64343 
7088a1edd7dSLiu Hui-R64343 			value = get_cfg_value(token, name, lineno);
7090b7f7c33SAdrian Alonso 			if (set_dcd_param)
7100b7f7c33SAdrian Alonso 				(*set_dcd_param)(imxhdr, *dcd_len, *cmd);
7110b7f7c33SAdrian Alonso 			(*set_dcd_val)(imxhdr, name, lineno, fld, value,
7120b7f7c33SAdrian Alonso 					*dcd_len);
7138a1edd7dSLiu Hui-R64343 
7144d5fa985STroy Kisky 			if (fld == CFG_REG_VALUE) {
7158a1edd7dSLiu Hui-R64343 				(*dcd_len)++;
7164d5fa985STroy Kisky 				if (*dcd_len > max_dcd_entries) {
7174d5fa985STroy Kisky 					fprintf(stderr, "Error: %s[%d] -"
7184d5fa985STroy Kisky 						"DCD table exceeds maximum size(%d)\n",
7194d5fa985STroy Kisky 						name, lineno, max_dcd_entries);
7204d5fa985STroy Kisky 					exit(EXIT_FAILURE);
7214d5fa985STroy Kisky 				}
7224d5fa985STroy Kisky 			}
7238a1edd7dSLiu Hui-R64343 			break;
724b55e4f48SPeng Fan 		case CMD_PLUGIN:
725b55e4f48SPeng Fan 			value = get_cfg_value(token, name, lineno);
726b55e4f48SPeng Fan 			imximage_iram_free_start = value;
727b55e4f48SPeng Fan 			break;
7288a1edd7dSLiu Hui-R64343 		default:
7298a1edd7dSLiu Hui-R64343 			break;
7308a1edd7dSLiu Hui-R64343 		}
7310b7f7c33SAdrian Alonso 		break;
7320b7f7c33SAdrian Alonso 	default:
7330b7f7c33SAdrian Alonso 		break;
7340b7f7c33SAdrian Alonso 	}
7358a1edd7dSLiu Hui-R64343 }
parse_cfg_file(struct imx_header * imxhdr,char * name)7368a1edd7dSLiu Hui-R64343 static uint32_t parse_cfg_file(struct imx_header *imxhdr, char *name)
7378edcde5eSStefano Babic {
7388edcde5eSStefano Babic 	FILE *fd = NULL;
7398edcde5eSStefano Babic 	char *line = NULL;
7408edcde5eSStefano Babic 	char *token, *saveptr1, *saveptr2;
7418edcde5eSStefano Babic 	int lineno = 0;
7428a1edd7dSLiu Hui-R64343 	int fld;
7430ad22703SKim Phillips 	size_t len;
7448edcde5eSStefano Babic 	int dcd_len = 0;
7458edcde5eSStefano Babic 	int32_t cmd;
7468edcde5eSStefano Babic 
7478edcde5eSStefano Babic 	fd = fopen(name, "r");
7488edcde5eSStefano Babic 	if (fd == 0) {
7498edcde5eSStefano Babic 		fprintf(stderr, "Error: %s - Can't open DCD file\n", name);
7508edcde5eSStefano Babic 		exit(EXIT_FAILURE);
7518edcde5eSStefano Babic 	}
7528edcde5eSStefano Babic 
75301390affSStefano Babic 	/*
75401390affSStefano Babic 	 * Very simple parsing, line starting with # are comments
7558edcde5eSStefano Babic 	 * and are dropped
7568edcde5eSStefano Babic 	 */
7578edcde5eSStefano Babic 	while ((getline(&line, &len, fd)) > 0) {
7588edcde5eSStefano Babic 		lineno++;
7598edcde5eSStefano Babic 
7608edcde5eSStefano Babic 		token = strtok_r(line, "\r\n", &saveptr1);
7618edcde5eSStefano Babic 		if (token == NULL)
7628edcde5eSStefano Babic 			continue;
7638edcde5eSStefano Babic 
7648edcde5eSStefano Babic 		/* Check inside the single line */
7658edcde5eSStefano Babic 		for (fld = CFG_COMMAND, cmd = CMD_INVALID,
7668edcde5eSStefano Babic 				line = token; ; line = NULL, fld++) {
7678edcde5eSStefano Babic 			token = strtok_r(line, " \t", &saveptr2);
7688edcde5eSStefano Babic 			if (token == NULL)
7698edcde5eSStefano Babic 				break;
7708edcde5eSStefano Babic 
7718edcde5eSStefano Babic 			/* Drop all text starting with '#' as comments */
7728edcde5eSStefano Babic 			if (token[0] == '#')
7738edcde5eSStefano Babic 				break;
7748edcde5eSStefano Babic 
7758a1edd7dSLiu Hui-R64343 			parse_cfg_fld(imxhdr, &cmd, token, name,
7768a1edd7dSLiu Hui-R64343 					lineno, fld, &dcd_len);
7778edcde5eSStefano Babic 		}
7788edcde5eSStefano Babic 
7798edcde5eSStefano Babic 	}
7808edcde5eSStefano Babic 
7818a1edd7dSLiu Hui-R64343 	(*set_dcd_rst)(imxhdr, dcd_len, name, lineno);
7828edcde5eSStefano Babic 	fclose(fd);
7838edcde5eSStefano Babic 
784f3c32628SFabio Estevam 	/* Exit if there is no BOOT_FROM field specifying the flash_offset */
785f3c32628SFabio Estevam 	if (imximage_ivt_offset == FLASH_OFFSET_UNDEFINED) {
786f3c32628SFabio Estevam 		fprintf(stderr, "Error: No BOOT_FROM tag in %s\n", name);
787f3c32628SFabio Estevam 		exit(EXIT_FAILURE);
788f3c32628SFabio Estevam 	}
7895b28e913SStefano Babic 	return dcd_len;
7908edcde5eSStefano Babic }
7918edcde5eSStefano Babic 
7928a1edd7dSLiu Hui-R64343 
imximage_check_image_types(uint8_t type)7938a1edd7dSLiu Hui-R64343 static int imximage_check_image_types(uint8_t type)
7948a1edd7dSLiu Hui-R64343 {
7958a1edd7dSLiu Hui-R64343 	if (type == IH_TYPE_IMXIMAGE)
7968a1edd7dSLiu Hui-R64343 		return EXIT_SUCCESS;
7978a1edd7dSLiu Hui-R64343 	else
7988a1edd7dSLiu Hui-R64343 		return EXIT_FAILURE;
7998a1edd7dSLiu Hui-R64343 }
8008a1edd7dSLiu Hui-R64343 
imximage_verify_header(unsigned char * ptr,int image_size,struct image_tool_params * params)8018a1edd7dSLiu Hui-R64343 static int imximage_verify_header(unsigned char *ptr, int image_size,
802f86ed6a8SGuilherme Maciel Ferreira 			struct image_tool_params *params)
8038a1edd7dSLiu Hui-R64343 {
8048a1edd7dSLiu Hui-R64343 	struct imx_header *imx_hdr = (struct imx_header *) ptr;
8058a1edd7dSLiu Hui-R64343 
8068a1edd7dSLiu Hui-R64343 	if (detect_imximage_version(imx_hdr) == IMXIMAGE_VER_INVALID)
8078a1edd7dSLiu Hui-R64343 		return -FDT_ERR_BADSTRUCTURE;
8088a1edd7dSLiu Hui-R64343 
8098a1edd7dSLiu Hui-R64343 	return 0;
8108a1edd7dSLiu Hui-R64343 }
8118a1edd7dSLiu Hui-R64343 
imximage_print_header(const void * ptr)8128a1edd7dSLiu Hui-R64343 static void imximage_print_header(const void *ptr)
8138a1edd7dSLiu Hui-R64343 {
8148a1edd7dSLiu Hui-R64343 	struct imx_header *imx_hdr = (struct imx_header *) ptr;
8158a1edd7dSLiu Hui-R64343 	uint32_t version = detect_imximage_version(imx_hdr);
8168a1edd7dSLiu Hui-R64343 
8178a1edd7dSLiu Hui-R64343 	switch (version) {
8188a1edd7dSLiu Hui-R64343 	case IMXIMAGE_V1:
8198a1edd7dSLiu Hui-R64343 		print_hdr_v1(imx_hdr);
8208a1edd7dSLiu Hui-R64343 		break;
8218a1edd7dSLiu Hui-R64343 	case IMXIMAGE_V2:
8228a1edd7dSLiu Hui-R64343 		print_hdr_v2(imx_hdr);
8238a1edd7dSLiu Hui-R64343 		break;
8248a1edd7dSLiu Hui-R64343 	default:
8258a1edd7dSLiu Hui-R64343 		err_imximage_version(version);
8268a1edd7dSLiu Hui-R64343 		break;
8278a1edd7dSLiu Hui-R64343 	}
8288a1edd7dSLiu Hui-R64343 }
8298a1edd7dSLiu Hui-R64343 
imximage_set_header(void * ptr,struct stat * sbuf,int ifd,struct image_tool_params * params)8308edcde5eSStefano Babic static void imximage_set_header(void *ptr, struct stat *sbuf, int ifd,
831f86ed6a8SGuilherme Maciel Ferreira 				struct image_tool_params *params)
8328edcde5eSStefano Babic {
8338a1edd7dSLiu Hui-R64343 	struct imx_header *imxhdr = (struct imx_header *)ptr;
8348a1edd7dSLiu Hui-R64343 	uint32_t dcd_len;
835b55e4f48SPeng Fan 	uint32_t header_size;
8368edcde5eSStefano Babic 
8378a1edd7dSLiu Hui-R64343 	/*
8388a1edd7dSLiu Hui-R64343 	 * In order to not change the old imx cfg file
8398a1edd7dSLiu Hui-R64343 	 * by adding VERSION command into it, here need
8408a1edd7dSLiu Hui-R64343 	 * set up function ptr group to V1 by default.
8418a1edd7dSLiu Hui-R64343 	 */
8428a1edd7dSLiu Hui-R64343 	imximage_version = IMXIMAGE_V1;
84349d3e272SDirk Behme 	/* Be able to detect if the cfg file has no BOOT_FROM tag */
8443150f92cSStefano Babic 	imximage_ivt_offset = FLASH_OFFSET_UNDEFINED;
8450187c985SStefano Babic 	imximage_csf_size = 0;
84672048bc3SYork Sun 	set_hdr_func();
8478edcde5eSStefano Babic 
8488edcde5eSStefano Babic 	/* Parse dcd configuration file */
8498a1edd7dSLiu Hui-R64343 	dcd_len = parse_cfg_file(imxhdr, params->imagename);
8508edcde5eSStefano Babic 
85197f17fa6SMartin Kaiser 	if (imximage_version == IMXIMAGE_V1)
85297f17fa6SMartin Kaiser 		header_size = sizeof(flash_header_v1_t);
85397f17fa6SMartin Kaiser 	else {
854b55e4f48SPeng Fan 		header_size = sizeof(flash_header_v2_t) + sizeof(boot_data_t);
855b55e4f48SPeng Fan 		if (!plugin_image)
856b55e4f48SPeng Fan 			header_size += sizeof(dcd_v2_t);
857b55e4f48SPeng Fan 		else
858b55e4f48SPeng Fan 			header_size += MAX_PLUGIN_CODE_SIZE;
85997f17fa6SMartin Kaiser 	}
860b55e4f48SPeng Fan 
861b55e4f48SPeng Fan 	if (imximage_init_loadsize < imximage_ivt_offset + header_size)
86297f17fa6SMartin Kaiser 			imximage_init_loadsize = imximage_ivt_offset + header_size;
86303ea24b2SYe.Li 
8648a1edd7dSLiu Hui-R64343 	/* Set the imx header */
8653150f92cSStefano Babic 	(*set_imx_hdr)(imxhdr, dcd_len, params->ep, imximage_ivt_offset);
8661411fb37SFabio Estevam 
8671411fb37SFabio Estevam 	/*
8681411fb37SFabio Estevam 	 * ROM bug alert
869895d9966SMarek Vasut 	 *
870895d9966SMarek Vasut 	 * MX53 only loads 512 byte multiples in case of SD boot.
871895d9966SMarek Vasut 	 * MX53 only loads NAND page multiples in case of NAND boot and
872895d9966SMarek Vasut 	 * supports up to 4096 byte large pages, thus align to 4096.
873895d9966SMarek Vasut 	 *
874895d9966SMarek Vasut 	 * The remaining fraction of a block bytes would not be loaded!
8751411fb37SFabio Estevam 	 */
876de979804SYe.Li 	*header_size_ptr = ROUND((sbuf->st_size + imximage_ivt_offset), 4096);
8770187c985SStefano Babic 
8780187c985SStefano Babic 	if (csf_ptr && imximage_csf_size) {
8790187c985SStefano Babic 		*csf_ptr = params->ep - imximage_init_loadsize +
8800187c985SStefano Babic 			*header_size_ptr;
8810187c985SStefano Babic 		*header_size_ptr += imximage_csf_size;
8820187c985SStefano Babic 	}
8838edcde5eSStefano Babic }
8848edcde5eSStefano Babic 
imximage_check_params(struct image_tool_params * params)885f86ed6a8SGuilherme Maciel Ferreira int imximage_check_params(struct image_tool_params *params)
8868edcde5eSStefano Babic {
8878edcde5eSStefano Babic 	if (!params)
8888edcde5eSStefano Babic 		return CFG_INVALID;
8898edcde5eSStefano Babic 	if (!strlen(params->imagename)) {
8908edcde5eSStefano Babic 		fprintf(stderr, "Error: %s - Configuration file not specified, "
8918edcde5eSStefano Babic 			"it is needed for imximage generation\n",
8928edcde5eSStefano Babic 			params->cmdname);
8938edcde5eSStefano Babic 		return CFG_INVALID;
8948edcde5eSStefano Babic 	}
8958edcde5eSStefano Babic 	/*
8968edcde5eSStefano Babic 	 * Check parameters:
8978edcde5eSStefano Babic 	 * XIP is not allowed and verify that incompatible
8988edcde5eSStefano Babic 	 * parameters are not sent at the same time
8998edcde5eSStefano Babic 	 * For example, if list is required a data image must not be provided
9008edcde5eSStefano Babic 	 */
9018edcde5eSStefano Babic 	return	(params->dflag && (params->fflag || params->lflag)) ||
9028edcde5eSStefano Babic 		(params->fflag && (params->dflag || params->lflag)) ||
9038edcde5eSStefano Babic 		(params->lflag && (params->dflag || params->fflag)) ||
9048edcde5eSStefano Babic 		(params->xflag) || !(strlen(params->imagename));
9058edcde5eSStefano Babic }
9068edcde5eSStefano Babic 
imximage_generate(struct image_tool_params * params,struct image_type_params * tparams)907f86ed6a8SGuilherme Maciel Ferreira static int imximage_generate(struct image_tool_params *params,
90801390affSStefano Babic 	struct image_type_params *tparams)
90901390affSStefano Babic {
91001390affSStefano Babic 	struct imx_header *imxhdr;
91101390affSStefano Babic 	size_t alloc_len;
91201390affSStefano Babic 	struct stat sbuf;
91301390affSStefano Babic 	char *datafile = params->datafile;
914b55e4f48SPeng Fan 	uint32_t pad_len, header_size;
91501390affSStefano Babic 
91601390affSStefano Babic 	memset(&imximage_header, 0, sizeof(imximage_header));
91701390affSStefano Babic 
91801390affSStefano Babic 	/*
91901390affSStefano Babic 	 * In order to not change the old imx cfg file
92001390affSStefano Babic 	 * by adding VERSION command into it, here need
92101390affSStefano Babic 	 * set up function ptr group to V1 by default.
92201390affSStefano Babic 	 */
92301390affSStefano Babic 	imximage_version = IMXIMAGE_V1;
92401390affSStefano Babic 	/* Be able to detect if the cfg file has no BOOT_FROM tag */
92501390affSStefano Babic 	imximage_ivt_offset = FLASH_OFFSET_UNDEFINED;
92601390affSStefano Babic 	imximage_csf_size = 0;
92772048bc3SYork Sun 	set_hdr_func();
92801390affSStefano Babic 
92901390affSStefano Babic 	/* Parse dcd configuration file */
93001390affSStefano Babic 	parse_cfg_file(&imximage_header, params->imagename);
93101390affSStefano Babic 
93297f17fa6SMartin Kaiser 	if (imximage_version == IMXIMAGE_V1)
93397f17fa6SMartin Kaiser 		header_size = sizeof(imx_header_v1_t);
93497f17fa6SMartin Kaiser 	else {
935b55e4f48SPeng Fan 		header_size = sizeof(flash_header_v2_t) + sizeof(boot_data_t);
936b55e4f48SPeng Fan 		if (!plugin_image)
937b55e4f48SPeng Fan 			header_size += sizeof(dcd_v2_t);
938b55e4f48SPeng Fan 		else
939b55e4f48SPeng Fan 			header_size += MAX_PLUGIN_CODE_SIZE;
94097f17fa6SMartin Kaiser 	}
941b55e4f48SPeng Fan 
942b55e4f48SPeng Fan 	if (imximage_init_loadsize < imximage_ivt_offset + header_size)
94397f17fa6SMartin Kaiser 			imximage_init_loadsize = imximage_ivt_offset + header_size;
94497f17fa6SMartin Kaiser 
94501390affSStefano Babic 	alloc_len = imximage_init_loadsize - imximage_ivt_offset;
94601390affSStefano Babic 
947b55e4f48SPeng Fan 	if (alloc_len < header_size) {
94801390affSStefano Babic 		fprintf(stderr, "%s: header error\n",
94901390affSStefano Babic 			params->cmdname);
95001390affSStefano Babic 		exit(EXIT_FAILURE);
95101390affSStefano Babic 	}
95201390affSStefano Babic 
95301390affSStefano Babic 	imxhdr = malloc(alloc_len);
95401390affSStefano Babic 
95501390affSStefano Babic 	if (!imxhdr) {
95601390affSStefano Babic 		fprintf(stderr, "%s: malloc return failure: %s\n",
95701390affSStefano Babic 			params->cmdname, strerror(errno));
95801390affSStefano Babic 		exit(EXIT_FAILURE);
95901390affSStefano Babic 	}
96001390affSStefano Babic 
96101390affSStefano Babic 	memset(imxhdr, 0, alloc_len);
96201390affSStefano Babic 
96301390affSStefano Babic 	tparams->header_size = alloc_len;
96401390affSStefano Babic 	tparams->hdr         = imxhdr;
96501390affSStefano Babic 
96601390affSStefano Babic 	/* determine data image file length */
96701390affSStefano Babic 
96801390affSStefano Babic 	if (stat(datafile, &sbuf) < 0) {
96901390affSStefano Babic 		fprintf(stderr, "%s: Can't stat %s: %s\n",
97001390affSStefano Babic 			params->cmdname, datafile, strerror(errno));
97101390affSStefano Babic 		exit(EXIT_FAILURE);
97201390affSStefano Babic 	}
97301390affSStefano Babic 
97401390affSStefano Babic 	pad_len = ROUND(sbuf.st_size, 4096) - sbuf.st_size;
97501390affSStefano Babic 
97601390affSStefano Babic 	return pad_len;
97701390affSStefano Babic }
97801390affSStefano Babic 
97901390affSStefano Babic 
9808edcde5eSStefano Babic /*
9818edcde5eSStefano Babic  * imximage parameters
9828edcde5eSStefano Babic  */
983a93648d1SGuilherme Maciel Ferreira U_BOOT_IMAGE_TYPE(
984a93648d1SGuilherme Maciel Ferreira 	imximage,
985a93648d1SGuilherme Maciel Ferreira 	"Freescale i.MX Boot Image support",
986a93648d1SGuilherme Maciel Ferreira 	0,
987a93648d1SGuilherme Maciel Ferreira 	NULL,
988a93648d1SGuilherme Maciel Ferreira 	imximage_check_params,
989a93648d1SGuilherme Maciel Ferreira 	imximage_verify_header,
990a93648d1SGuilherme Maciel Ferreira 	imximage_print_header,
991a93648d1SGuilherme Maciel Ferreira 	imximage_set_header,
992a93648d1SGuilherme Maciel Ferreira 	NULL,
993a93648d1SGuilherme Maciel Ferreira 	imximage_check_image_types,
994a93648d1SGuilherme Maciel Ferreira 	NULL,
995a93648d1SGuilherme Maciel Ferreira 	imximage_generate
996a93648d1SGuilherme Maciel Ferreira );
997