865 lines
		
	
	
		
			20 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			865 lines
		
	
	
		
			20 KiB
		
	
	
	
		
			C
		
	
	
	
| /*
 | |
|  * Image manipulator for Marvell SoCs
 | |
|  *  supports Kirkwood, Dove, Armada 370, and Armada XP
 | |
|  *
 | |
|  * (C) Copyright 2013 Thomas Petazzoni
 | |
|  * <thomas.petazzoni@free-electrons.com>
 | |
|  *
 | |
|  * SPDX-License-Identifier:	GPL-2.0+
 | |
|  *
 | |
|  * Not implemented: support for the register headers and secure
 | |
|  * headers in v1 images
 | |
|  */
 | |
| 
 | |
| #include "imagetool.h"
 | |
| #include <limits.h>
 | |
| #include <image.h>
 | |
| #include <stdint.h>
 | |
| #include "kwbimage.h"
 | |
| 
 | |
| static struct image_cfg_element *image_cfg;
 | |
| static int cfgn;
 | |
| 
 | |
| struct boot_mode {
 | |
| 	unsigned int id;
 | |
| 	const char *name;
 | |
| };
 | |
| 
 | |
| struct boot_mode boot_modes[] = {
 | |
| 	{ 0x4D, "i2c"  },
 | |
| 	{ 0x5A, "spi"  },
 | |
| 	{ 0x8B, "nand" },
 | |
| 	{ 0x78, "sata" },
 | |
| 	{ 0x9C, "pex"  },
 | |
| 	{ 0x69, "uart" },
 | |
| 	{ 0xAE, "sdio" },
 | |
| 	{},
 | |
| };
 | |
| 
 | |
| struct nand_ecc_mode {
 | |
| 	unsigned int id;
 | |
| 	const char *name;
 | |
| };
 | |
| 
 | |
| struct nand_ecc_mode nand_ecc_modes[] = {
 | |
| 	{ 0x00, "default" },
 | |
| 	{ 0x01, "hamming" },
 | |
| 	{ 0x02, "rs" },
 | |
| 	{ 0x03, "disabled" },
 | |
| 	{},
 | |
| };
 | |
| 
 | |
| /* Used to identify an undefined execution or destination address */
 | |
| #define ADDR_INVALID ((uint32_t)-1)
 | |
| 
 | |
| #define BINARY_MAX_ARGS 8
 | |
| 
 | |
| /* In-memory representation of a line of the configuration file */
 | |
| struct image_cfg_element {
 | |
| 	enum {
 | |
| 		IMAGE_CFG_VERSION = 0x1,
 | |
| 		IMAGE_CFG_BOOT_FROM,
 | |
| 		IMAGE_CFG_DEST_ADDR,
 | |
| 		IMAGE_CFG_EXEC_ADDR,
 | |
| 		IMAGE_CFG_NAND_BLKSZ,
 | |
| 		IMAGE_CFG_NAND_BADBLK_LOCATION,
 | |
| 		IMAGE_CFG_NAND_ECC_MODE,
 | |
| 		IMAGE_CFG_NAND_PAGESZ,
 | |
| 		IMAGE_CFG_BINARY,
 | |
| 		IMAGE_CFG_PAYLOAD,
 | |
| 		IMAGE_CFG_DATA,
 | |
| 	} type;
 | |
| 	union {
 | |
| 		unsigned int version;
 | |
| 		unsigned int bootfrom;
 | |
| 		struct {
 | |
| 			const char *file;
 | |
| 			unsigned int args[BINARY_MAX_ARGS];
 | |
| 			unsigned int nargs;
 | |
| 		} binary;
 | |
| 		const char *payload;
 | |
| 		unsigned int dstaddr;
 | |
| 		unsigned int execaddr;
 | |
| 		unsigned int nandblksz;
 | |
| 		unsigned int nandbadblklocation;
 | |
| 		unsigned int nandeccmode;
 | |
| 		unsigned int nandpagesz;
 | |
| 		struct ext_hdr_v0_reg regdata;
 | |
| 	};
 | |
| };
 | |
| 
 | |
| #define IMAGE_CFG_ELEMENT_MAX 256
 | |
| 
 | |
| /*
 | |
|  * Utility functions to manipulate boot mode and ecc modes (convert
 | |
|  * them back and forth between description strings and the
 | |
|  * corresponding numerical identifiers).
 | |
|  */
 | |
| 
 | |
| static const char *image_boot_mode_name(unsigned int id)
 | |
| {
 | |
| 	int i;
 | |
| 	for (i = 0; boot_modes[i].name; i++)
 | |
| 		if (boot_modes[i].id == id)
 | |
| 			return boot_modes[i].name;
 | |
| 	return NULL;
 | |
| }
 | |
| 
 | |
| int image_boot_mode_id(const char *boot_mode_name)
 | |
| {
 | |
| 	int i;
 | |
| 	for (i = 0; boot_modes[i].name; i++)
 | |
| 		if (!strcmp(boot_modes[i].name, boot_mode_name))
 | |
| 			return boot_modes[i].id;
 | |
| 
 | |
| 	return -1;
 | |
| }
 | |
| 
 | |
| int image_nand_ecc_mode_id(const char *nand_ecc_mode_name)
 | |
| {
 | |
| 	int i;
 | |
| 	for (i = 0; nand_ecc_modes[i].name; i++)
 | |
| 		if (!strcmp(nand_ecc_modes[i].name, nand_ecc_mode_name))
 | |
| 			return nand_ecc_modes[i].id;
 | |
| 	return -1;
 | |
| }
 | |
| 
 | |
| static struct image_cfg_element *
 | |
| image_find_option(unsigned int optiontype)
 | |
| {
 | |
| 	int i;
 | |
| 
 | |
| 	for (i = 0; i < cfgn; i++) {
 | |
| 		if (image_cfg[i].type == optiontype)
 | |
| 			return &image_cfg[i];
 | |
| 	}
 | |
| 
 | |
| 	return NULL;
 | |
| }
 | |
| 
 | |
| static unsigned int
 | |
| image_count_options(unsigned int optiontype)
 | |
| {
 | |
| 	int i;
 | |
| 	unsigned int count = 0;
 | |
| 
 | |
| 	for (i = 0; i < cfgn; i++)
 | |
| 		if (image_cfg[i].type == optiontype)
 | |
| 			count++;
 | |
| 
 | |
| 	return count;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * Compute a 8-bit checksum of a memory area. This algorithm follows
 | |
|  * the requirements of the Marvell SoC BootROM specifications.
 | |
|  */
 | |
| static uint8_t image_checksum8(void *start, uint32_t len)
 | |
| {
 | |
| 	uint8_t csum = 0;
 | |
| 	uint8_t *p = start;
 | |
| 
 | |
| 	/* check len and return zero checksum if invalid */
 | |
| 	if (!len)
 | |
| 		return 0;
 | |
| 
 | |
| 	do {
 | |
| 		csum += *p;
 | |
| 		p++;
 | |
| 	} while (--len);
 | |
| 
 | |
| 	return csum;
 | |
| }
 | |
| 
 | |
| static uint32_t image_checksum32(void *start, uint32_t len)
 | |
| {
 | |
| 	uint32_t csum = 0;
 | |
| 	uint32_t *p = start;
 | |
| 
 | |
| 	/* check len and return zero checksum if invalid */
 | |
| 	if (!len)
 | |
| 		return 0;
 | |
| 
 | |
| 	if (len % sizeof(uint32_t)) {
 | |
| 		fprintf(stderr, "Length %d is not in multiple of %zu\n",
 | |
| 			len, sizeof(uint32_t));
 | |
| 		return 0;
 | |
| 	}
 | |
| 
 | |
| 	do {
 | |
| 		csum += *p;
 | |
| 		p++;
 | |
| 		len -= sizeof(uint32_t);
 | |
| 	} while (len > 0);
 | |
| 
 | |
| 	return csum;
 | |
| }
 | |
| 
 | |
| static void *image_create_v0(size_t *imagesz, struct image_tool_params *params,
 | |
| 			     int payloadsz)
 | |
| {
 | |
| 	struct image_cfg_element *e;
 | |
| 	size_t headersz;
 | |
| 	struct main_hdr_v0 *main_hdr;
 | |
| 	struct ext_hdr_v0 *ext_hdr;
 | |
| 	void *image;
 | |
| 	int has_ext = 0;
 | |
| 
 | |
| 	/*
 | |
| 	 * Calculate the size of the header and the size of the
 | |
| 	 * payload
 | |
| 	 */
 | |
| 	headersz  = sizeof(struct main_hdr_v0);
 | |
| 
 | |
| 	if (image_count_options(IMAGE_CFG_DATA) > 0) {
 | |
| 		has_ext = 1;
 | |
| 		headersz += sizeof(struct ext_hdr_v0);
 | |
| 	}
 | |
| 
 | |
| 	if (image_count_options(IMAGE_CFG_PAYLOAD) > 1) {
 | |
| 		fprintf(stderr, "More than one payload, not possible\n");
 | |
| 		return NULL;
 | |
| 	}
 | |
| 
 | |
| 	image = malloc(headersz);
 | |
| 	if (!image) {
 | |
| 		fprintf(stderr, "Cannot allocate memory for image\n");
 | |
| 		return NULL;
 | |
| 	}
 | |
| 
 | |
| 	memset(image, 0, headersz);
 | |
| 
 | |
| 	main_hdr = image;
 | |
| 
 | |
| 	/* Fill in the main header */
 | |
| 	main_hdr->blocksize =
 | |
| 		cpu_to_le32(payloadsz + sizeof(uint32_t) - headersz);
 | |
| 	main_hdr->srcaddr   = cpu_to_le32(headersz);
 | |
| 	main_hdr->ext       = has_ext;
 | |
| 	main_hdr->destaddr  = cpu_to_le32(params->addr);
 | |
| 	main_hdr->execaddr  = cpu_to_le32(params->ep);
 | |
| 
 | |
| 	e = image_find_option(IMAGE_CFG_BOOT_FROM);
 | |
| 	if (e)
 | |
| 		main_hdr->blockid = e->bootfrom;
 | |
| 	e = image_find_option(IMAGE_CFG_NAND_ECC_MODE);
 | |
| 	if (e)
 | |
| 		main_hdr->nandeccmode = e->nandeccmode;
 | |
| 	e = image_find_option(IMAGE_CFG_NAND_PAGESZ);
 | |
| 	if (e)
 | |
| 		main_hdr->nandpagesize = cpu_to_le16(e->nandpagesz);
 | |
| 	main_hdr->checksum = image_checksum8(image,
 | |
| 					     sizeof(struct main_hdr_v0));
 | |
| 
 | |
| 	/* Generate the ext header */
 | |
| 	if (has_ext) {
 | |
| 		int cfgi, datai;
 | |
| 
 | |
| 		ext_hdr = image + sizeof(struct main_hdr_v0);
 | |
| 		ext_hdr->offset = cpu_to_le32(0x40);
 | |
| 
 | |
| 		for (cfgi = 0, datai = 0; cfgi < cfgn; cfgi++) {
 | |
| 			e = &image_cfg[cfgi];
 | |
| 			if (e->type != IMAGE_CFG_DATA)
 | |
| 				continue;
 | |
| 
 | |
| 			ext_hdr->rcfg[datai].raddr =
 | |
| 				cpu_to_le32(e->regdata.raddr);
 | |
| 			ext_hdr->rcfg[datai].rdata =
 | |
| 				cpu_to_le32(e->regdata.rdata);
 | |
| 			datai++;
 | |
| 		}
 | |
| 
 | |
| 		ext_hdr->checksum = image_checksum8(ext_hdr,
 | |
| 						    sizeof(struct ext_hdr_v0));
 | |
| 	}
 | |
| 
 | |
| 	*imagesz = headersz;
 | |
| 	return image;
 | |
| }
 | |
| 
 | |
| static size_t image_headersz_v1(struct image_tool_params *params,
 | |
| 				int *hasext)
 | |
| {
 | |
| 	struct image_cfg_element *binarye;
 | |
| 	size_t headersz;
 | |
| 	int ret;
 | |
| 
 | |
| 	/*
 | |
| 	 * Calculate the size of the header and the size of the
 | |
| 	 * payload
 | |
| 	 */
 | |
| 	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) {
 | |
| 		fprintf(stderr, "More than one payload, not possible\n");
 | |
| 		return 0;
 | |
| 	}
 | |
| 
 | |
| 	binarye = image_find_option(IMAGE_CFG_BINARY);
 | |
| 	if (binarye) {
 | |
| 		struct stat s;
 | |
| 
 | |
| 		ret = stat(binarye->binary.file, &s);
 | |
| 		if (ret < 0) {
 | |
| 			char cwd[PATH_MAX];
 | |
| 			char *dir = cwd;
 | |
| 
 | |
| 			memset(cwd, 0, sizeof(cwd));
 | |
| 			if (!getcwd(cwd, sizeof(cwd))) {
 | |
| 				dir = "current working directory";
 | |
| 				perror("getcwd() failed");
 | |
| 			}
 | |
| 
 | |
| 			fprintf(stderr,
 | |
| 				"Didn't find the file '%s' in '%s' which is mandatory to generate the image\n"
 | |
| 				"This file generally contains the DDR3 training code, and should be extracted from an existing bootable\n"
 | |
| 				"image for your board. See 'kwbimage -x' to extract it from an existing image.\n",
 | |
| 				binarye->binary.file, dir);
 | |
| 			return 0;
 | |
| 		}
 | |
| 
 | |
| 		headersz += sizeof(struct opt_hdr_v1) +
 | |
| 			s.st_size +
 | |
| 			(binarye->binary.nargs + 2) * sizeof(uint32_t);
 | |
| 		if (hasext)
 | |
| 			*hasext = 1;
 | |
| 	}
 | |
| 
 | |
| #if defined(CONFIG_SYS_U_BOOT_OFFS)
 | |
| 	if (headersz > CONFIG_SYS_U_BOOT_OFFS) {
 | |
| 		fprintf(stderr, "Error: Image header (incl. SPL image) too big!\n");
 | |
| 		fprintf(stderr, "header=0x%x CONFIG_SYS_U_BOOT_OFFS=0x%x!\n",
 | |
| 			(int)headersz, CONFIG_SYS_U_BOOT_OFFS);
 | |
| 		fprintf(stderr, "Increase CONFIG_SYS_U_BOOT_OFFS!\n");
 | |
| 		return 0;
 | |
| 	} else {
 | |
| 		headersz = CONFIG_SYS_U_BOOT_OFFS;
 | |
| 	}
 | |
| #endif
 | |
| 
 | |
| 	/*
 | |
| 	 * The payload should be aligned on some reasonable
 | |
| 	 * boundary
 | |
| 	 */
 | |
| 	return ALIGN_SUP(headersz, 4096);
 | |
| }
 | |
| 
 | |
| static void *image_create_v1(size_t *imagesz, struct image_tool_params *params,
 | |
| 			     int payloadsz)
 | |
| {
 | |
| 	struct image_cfg_element *e, *binarye;
 | |
| 	struct main_hdr_v1 *main_hdr;
 | |
| 	size_t headersz;
 | |
| 	void *image, *cur;
 | |
| 	int hasext = 0;
 | |
| 	int ret;
 | |
| 
 | |
| 	/*
 | |
| 	 * Calculate the size of the header and the size of the
 | |
| 	 * payload
 | |
| 	 */
 | |
| 	headersz = image_headersz_v1(params, &hasext);
 | |
| 	if (headersz == 0)
 | |
| 		return NULL;
 | |
| 
 | |
| 	image = malloc(headersz);
 | |
| 	if (!image) {
 | |
| 		fprintf(stderr, "Cannot allocate memory for image\n");
 | |
| 		return NULL;
 | |
| 	}
 | |
| 
 | |
| 	memset(image, 0, headersz);
 | |
| 
 | |
| 	cur = main_hdr = image;
 | |
| 	cur += sizeof(struct main_hdr_v1);
 | |
| 
 | |
| 	/* Fill the main header */
 | |
| 	main_hdr->blocksize    =
 | |
| 		cpu_to_le32(payloadsz - headersz + sizeof(uint32_t));
 | |
| 	main_hdr->headersz_lsb = cpu_to_le16(headersz & 0xFFFF);
 | |
| 	main_hdr->headersz_msb = (headersz & 0xFFFF0000) >> 16;
 | |
| 	main_hdr->destaddr     = cpu_to_le32(params->addr);
 | |
| 	main_hdr->execaddr     = cpu_to_le32(params->ep);
 | |
| 	main_hdr->srcaddr      = cpu_to_le32(headersz);
 | |
| 	main_hdr->ext          = hasext;
 | |
| 	main_hdr->version      = 1;
 | |
| 	e = image_find_option(IMAGE_CFG_BOOT_FROM);
 | |
| 	if (e)
 | |
| 		main_hdr->blockid = e->bootfrom;
 | |
| 	e = image_find_option(IMAGE_CFG_NAND_BLKSZ);
 | |
| 	if (e)
 | |
| 		main_hdr->nandblocksize = e->nandblksz / (64 * 1024);
 | |
| 	e = image_find_option(IMAGE_CFG_NAND_BADBLK_LOCATION);
 | |
| 	if (e)
 | |
| 		main_hdr->nandbadblklocation = e->nandbadblklocation;
 | |
| 
 | |
| 	binarye = image_find_option(IMAGE_CFG_BINARY);
 | |
| 	if (binarye) {
 | |
| 		struct opt_hdr_v1 *hdr = cur;
 | |
| 		uint32_t *args;
 | |
| 		size_t binhdrsz;
 | |
| 		struct stat s;
 | |
| 		int argi;
 | |
| 		FILE *bin;
 | |
| 
 | |
| 		hdr->headertype = OPT_HDR_V1_BINARY_TYPE;
 | |
| 
 | |
| 		bin = fopen(binarye->binary.file, "r");
 | |
| 		if (!bin) {
 | |
| 			fprintf(stderr, "Cannot open binary file %s\n",
 | |
| 				binarye->binary.file);
 | |
| 			return NULL;
 | |
| 		}
 | |
| 
 | |
| 		fstat(fileno(bin), &s);
 | |
| 
 | |
| 		binhdrsz = sizeof(struct opt_hdr_v1) +
 | |
| 			(binarye->binary.nargs + 2) * sizeof(uint32_t) +
 | |
| 			s.st_size;
 | |
| 
 | |
| 		/*
 | |
| 		 * The size includes the binary image size, rounded
 | |
| 		 * up to a 4-byte boundary. Plus 4 bytes for the
 | |
| 		 * next-header byte and 3-byte alignment at the end.
 | |
| 		 */
 | |
| 		binhdrsz = ALIGN_SUP(binhdrsz, 4) + 4;
 | |
| 		hdr->headersz_lsb = cpu_to_le16(binhdrsz & 0xFFFF);
 | |
| 		hdr->headersz_msb = (binhdrsz & 0xFFFF0000) >> 16;
 | |
| 
 | |
| 		cur += sizeof(struct opt_hdr_v1);
 | |
| 
 | |
| 		args = cur;
 | |
| 		*args = cpu_to_le32(binarye->binary.nargs);
 | |
| 		args++;
 | |
| 		for (argi = 0; argi < binarye->binary.nargs; argi++)
 | |
| 			args[argi] = cpu_to_le32(binarye->binary.args[argi]);
 | |
| 
 | |
| 		cur += (binarye->binary.nargs + 1) * sizeof(uint32_t);
 | |
| 
 | |
| 		ret = fread(cur, s.st_size, 1, bin);
 | |
| 		if (ret != 1) {
 | |
| 			fprintf(stderr,
 | |
| 				"Could not read binary image %s\n",
 | |
| 				binarye->binary.file);
 | |
| 			return NULL;
 | |
| 		}
 | |
| 
 | |
| 		fclose(bin);
 | |
| 
 | |
| 		cur += ALIGN_SUP(s.st_size, 4);
 | |
| 
 | |
| 		/*
 | |
| 		 * For now, we don't support more than one binary
 | |
| 		 * header, and no other header types are
 | |
| 		 * supported. So, the binary header is necessarily the
 | |
| 		 * last one
 | |
| 		 */
 | |
| 		*((uint32_t *)cur) = 0x00000000;
 | |
| 
 | |
| 		cur += sizeof(uint32_t);
 | |
| 	}
 | |
| 
 | |
| 	/* Calculate and set the header checksum */
 | |
| 	main_hdr->checksum = image_checksum8(main_hdr, headersz);
 | |
| 
 | |
| 	*imagesz = headersz;
 | |
| 	return image;
 | |
| }
 | |
| 
 | |
| static int image_create_config_parse_oneline(char *line,
 | |
| 					     struct image_cfg_element *el)
 | |
| {
 | |
| 	char *keyword, *saveptr;
 | |
| 	char deliminiters[] = " \t";
 | |
| 
 | |
| 	keyword = strtok_r(line, deliminiters, &saveptr);
 | |
| 	if (!strcmp(keyword, "VERSION")) {
 | |
| 		char *value = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 		el->type = IMAGE_CFG_VERSION;
 | |
| 		el->version = atoi(value);
 | |
| 	} else if (!strcmp(keyword, "BOOT_FROM")) {
 | |
| 		char *value = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 		int ret = image_boot_mode_id(value);
 | |
| 		if (ret < 0) {
 | |
| 			fprintf(stderr,
 | |
| 				"Invalid boot media '%s'\n", value);
 | |
| 			return -1;
 | |
| 		}
 | |
| 		el->type = IMAGE_CFG_BOOT_FROM;
 | |
| 		el->bootfrom = ret;
 | |
| 	} else if (!strcmp(keyword, "NAND_BLKSZ")) {
 | |
| 		char *value = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 		el->type = IMAGE_CFG_NAND_BLKSZ;
 | |
| 		el->nandblksz = strtoul(value, NULL, 16);
 | |
| 	} else if (!strcmp(keyword, "NAND_BADBLK_LOCATION")) {
 | |
| 		char *value = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 		el->type = IMAGE_CFG_NAND_BADBLK_LOCATION;
 | |
| 		el->nandbadblklocation =
 | |
| 			strtoul(value, NULL, 16);
 | |
| 	} else if (!strcmp(keyword, "NAND_ECC_MODE")) {
 | |
| 		char *value = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 		int ret = image_nand_ecc_mode_id(value);
 | |
| 		if (ret < 0) {
 | |
| 			fprintf(stderr,
 | |
| 				"Invalid NAND ECC mode '%s'\n", value);
 | |
| 			return -1;
 | |
| 		}
 | |
| 		el->type = IMAGE_CFG_NAND_ECC_MODE;
 | |
| 		el->nandeccmode = ret;
 | |
| 	} else if (!strcmp(keyword, "NAND_PAGE_SIZE")) {
 | |
| 		char *value = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 		el->type = IMAGE_CFG_NAND_PAGESZ;
 | |
| 		el->nandpagesz = strtoul(value, NULL, 16);
 | |
| 	} else if (!strcmp(keyword, "BINARY")) {
 | |
| 		char *value = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 		int argi = 0;
 | |
| 
 | |
| 		el->type = IMAGE_CFG_BINARY;
 | |
| 		el->binary.file = strdup(value);
 | |
| 		while (1) {
 | |
| 			value = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 			if (!value)
 | |
| 				break;
 | |
| 			el->binary.args[argi] = strtoul(value, NULL, 16);
 | |
| 			argi++;
 | |
| 			if (argi >= BINARY_MAX_ARGS) {
 | |
| 				fprintf(stderr,
 | |
| 					"Too many argument for binary\n");
 | |
| 				return -1;
 | |
| 			}
 | |
| 		}
 | |
| 		el->binary.nargs = argi;
 | |
| 	} else if (!strcmp(keyword, "DATA")) {
 | |
| 		char *value1 = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 		char *value2 = strtok_r(NULL, deliminiters, &saveptr);
 | |
| 
 | |
| 		if (!value1 || !value2) {
 | |
| 			fprintf(stderr,
 | |
| 				"Invalid number of arguments for DATA\n");
 | |
| 			return -1;
 | |
| 		}
 | |
| 
 | |
| 		el->type = IMAGE_CFG_DATA;
 | |
| 		el->regdata.raddr = strtoul(value1, NULL, 16);
 | |
| 		el->regdata.rdata = strtoul(value2, NULL, 16);
 | |
| 	} else {
 | |
| 		fprintf(stderr, "Ignoring unknown line '%s'\n", line);
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * Parse the configuration file 'fcfg' into the array of configuration
 | |
|  * elements 'image_cfg', and return the number of configuration
 | |
|  * elements in 'cfgn'.
 | |
|  */
 | |
| static int image_create_config_parse(FILE *fcfg)
 | |
| {
 | |
| 	int ret;
 | |
| 	int cfgi = 0;
 | |
| 
 | |
| 	/* Parse the configuration file */
 | |
| 	while (!feof(fcfg)) {
 | |
| 		char *line;
 | |
| 		char buf[256];
 | |
| 
 | |
| 		/* Read the current line */
 | |
| 		memset(buf, 0, sizeof(buf));
 | |
| 		line = fgets(buf, sizeof(buf), fcfg);
 | |
| 		if (!line)
 | |
| 			break;
 | |
| 
 | |
| 		/* Ignore useless lines */
 | |
| 		if (line[0] == '\n' || line[0] == '#')
 | |
| 			continue;
 | |
| 
 | |
| 		/* Strip final newline */
 | |
| 		if (line[strlen(line) - 1] == '\n')
 | |
| 			line[strlen(line) - 1] = 0;
 | |
| 
 | |
| 		/* Parse the current line */
 | |
| 		ret = image_create_config_parse_oneline(line,
 | |
| 							&image_cfg[cfgi]);
 | |
| 		if (ret)
 | |
| 			return ret;
 | |
| 
 | |
| 		cfgi++;
 | |
| 
 | |
| 		if (cfgi >= IMAGE_CFG_ELEMENT_MAX) {
 | |
| 			fprintf(stderr,
 | |
| 				"Too many configuration elements in .cfg file\n");
 | |
| 			return -1;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	cfgn = cfgi;
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static int image_get_version(void)
 | |
| {
 | |
| 	struct image_cfg_element *e;
 | |
| 
 | |
| 	e = image_find_option(IMAGE_CFG_VERSION);
 | |
| 	if (!e)
 | |
| 		return -1;
 | |
| 
 | |
| 	return e->version;
 | |
| }
 | |
| 
 | |
| static int image_version_file(const char *input)
 | |
| {
 | |
| 	FILE *fcfg;
 | |
| 	int version;
 | |
| 	int ret;
 | |
| 
 | |
| 	fcfg = fopen(input, "r");
 | |
| 	if (!fcfg) {
 | |
| 		fprintf(stderr, "Could not open input file %s\n", input);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	image_cfg = malloc(IMAGE_CFG_ELEMENT_MAX *
 | |
| 			   sizeof(struct image_cfg_element));
 | |
| 	if (!image_cfg) {
 | |
| 		fprintf(stderr, "Cannot allocate memory\n");
 | |
| 		fclose(fcfg);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	memset(image_cfg, 0,
 | |
| 	       IMAGE_CFG_ELEMENT_MAX * sizeof(struct image_cfg_element));
 | |
| 	rewind(fcfg);
 | |
| 
 | |
| 	ret = image_create_config_parse(fcfg);
 | |
| 	fclose(fcfg);
 | |
| 	if (ret) {
 | |
| 		free(image_cfg);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	version = image_get_version();
 | |
| 	/* Fallback to version 0 is no version is provided in the cfg file */
 | |
| 	if (version == -1)
 | |
| 		version = 0;
 | |
| 
 | |
| 	free(image_cfg);
 | |
| 
 | |
| 	return version;
 | |
| }
 | |
| 
 | |
| static void kwbimage_set_header(void *ptr, struct stat *sbuf, int ifd,
 | |
| 				struct image_tool_params *params)
 | |
| {
 | |
| 	FILE *fcfg;
 | |
| 	void *image = NULL;
 | |
| 	int version;
 | |
| 	size_t headersz = 0;
 | |
| 	uint32_t checksum;
 | |
| 	int ret;
 | |
| 	int size;
 | |
| 
 | |
| 	fcfg = fopen(params->imagename, "r");
 | |
| 	if (!fcfg) {
 | |
| 		fprintf(stderr, "Could not open input file %s\n",
 | |
| 			params->imagename);
 | |
| 		exit(EXIT_FAILURE);
 | |
| 	}
 | |
| 
 | |
| 	image_cfg = malloc(IMAGE_CFG_ELEMENT_MAX *
 | |
| 			   sizeof(struct image_cfg_element));
 | |
| 	if (!image_cfg) {
 | |
| 		fprintf(stderr, "Cannot allocate memory\n");
 | |
| 		fclose(fcfg);
 | |
| 		exit(EXIT_FAILURE);
 | |
| 	}
 | |
| 
 | |
| 	memset(image_cfg, 0,
 | |
| 	       IMAGE_CFG_ELEMENT_MAX * sizeof(struct image_cfg_element));
 | |
| 	rewind(fcfg);
 | |
| 
 | |
| 	ret = image_create_config_parse(fcfg);
 | |
| 	fclose(fcfg);
 | |
| 	if (ret) {
 | |
| 		free(image_cfg);
 | |
| 		exit(EXIT_FAILURE);
 | |
| 	}
 | |
| 
 | |
| 	/* The MVEBU BootROM does not allow non word aligned payloads */
 | |
| 	sbuf->st_size = ALIGN_SUP(sbuf->st_size, 4);
 | |
| 
 | |
| 	version = image_get_version();
 | |
| 	switch (version) {
 | |
| 		/*
 | |
| 		 * Fallback to version 0 if no version is provided in the
 | |
| 		 * cfg file
 | |
| 		 */
 | |
| 	case -1:
 | |
| 	case 0:
 | |
| 		image = image_create_v0(&headersz, params, sbuf->st_size);
 | |
| 		break;
 | |
| 
 | |
| 	case 1:
 | |
| 		image = image_create_v1(&headersz, params, sbuf->st_size);
 | |
| 		break;
 | |
| 
 | |
| 	default:
 | |
| 		fprintf(stderr, "Unsupported version %d\n", version);
 | |
| 		free(image_cfg);
 | |
| 		exit(EXIT_FAILURE);
 | |
| 	}
 | |
| 
 | |
| 	if (!image) {
 | |
| 		fprintf(stderr, "Could not create image\n");
 | |
| 		free(image_cfg);
 | |
| 		exit(EXIT_FAILURE);
 | |
| 	}
 | |
| 
 | |
| 	free(image_cfg);
 | |
| 
 | |
| 	/* Build and add image checksum header */
 | |
| 	checksum =
 | |
| 		cpu_to_le32(image_checksum32((uint32_t *)ptr, sbuf->st_size));
 | |
| 	size = write(ifd, &checksum, sizeof(uint32_t));
 | |
| 	if (size != sizeof(uint32_t)) {
 | |
| 		fprintf(stderr, "Error:%s - Checksum write %d bytes %s\n",
 | |
| 			params->cmdname, size, params->imagefile);
 | |
| 		exit(EXIT_FAILURE);
 | |
| 	}
 | |
| 
 | |
| 	sbuf->st_size += sizeof(uint32_t);
 | |
| 
 | |
| 	/* Finally copy the header into the image area */
 | |
| 	memcpy(ptr, image, headersz);
 | |
| 
 | |
| 	free(image);
 | |
| }
 | |
| 
 | |
| static void kwbimage_print_header(const void *ptr)
 | |
| {
 | |
| 	struct main_hdr_v0 *mhdr = (struct main_hdr_v0 *)ptr;
 | |
| 
 | |
| 	printf("Image Type:   MVEBU Boot from %s Image\n",
 | |
| 	       image_boot_mode_name(mhdr->blockid));
 | |
| 	printf("Image version:%d\n", image_version((void *)ptr));
 | |
| 	printf("Data Size:    ");
 | |
| 	genimg_print_size(mhdr->blocksize - sizeof(uint32_t));
 | |
| 	printf("Load Address: %08x\n", mhdr->destaddr);
 | |
| 	printf("Entry Point:  %08x\n", mhdr->execaddr);
 | |
| }
 | |
| 
 | |
| static int kwbimage_check_image_types(uint8_t type)
 | |
| {
 | |
| 	if (type == IH_TYPE_KWBIMAGE)
 | |
| 		return EXIT_SUCCESS;
 | |
| 	else
 | |
| 		return EXIT_FAILURE;
 | |
| }
 | |
| 
 | |
| static int kwbimage_verify_header(unsigned char *ptr, int image_size,
 | |
| 				  struct image_tool_params *params)
 | |
| {
 | |
| 	struct main_hdr_v0 *main_hdr;
 | |
| 	struct ext_hdr_v0 *ext_hdr;
 | |
| 	uint8_t checksum;
 | |
| 
 | |
| 	main_hdr = (void *)ptr;
 | |
| 	checksum = image_checksum8(ptr,
 | |
| 				   sizeof(struct main_hdr_v0)
 | |
| 				   - sizeof(uint8_t));
 | |
| 	if (checksum != main_hdr->checksum)
 | |
| 		return -FDT_ERR_BADSTRUCTURE;
 | |
| 
 | |
| 	/* Only version 0 extended header has checksum */
 | |
| 	if (image_version((void *)ptr) == 0) {
 | |
| 		ext_hdr = (void *)ptr + sizeof(struct main_hdr_v0);
 | |
| 		checksum = image_checksum8(ext_hdr,
 | |
| 					   sizeof(struct ext_hdr_v0)
 | |
| 					   - sizeof(uint8_t));
 | |
| 		if (checksum != ext_hdr->checksum)
 | |
| 			return -FDT_ERR_BADSTRUCTURE;
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static int kwbimage_generate(struct image_tool_params *params,
 | |
| 			     struct image_type_params *tparams)
 | |
| {
 | |
| 	int alloc_len;
 | |
| 	void *hdr;
 | |
| 	int version = 0;
 | |
| 
 | |
| 	version = image_version_file(params->imagename);
 | |
| 	if (version == 0) {
 | |
| 		alloc_len = sizeof(struct main_hdr_v0) +
 | |
| 			sizeof(struct ext_hdr_v0);
 | |
| 	} else {
 | |
| 		alloc_len = image_headersz_v1(params, NULL);
 | |
| 	}
 | |
| 
 | |
| 	hdr = malloc(alloc_len);
 | |
| 	if (!hdr) {
 | |
| 		fprintf(stderr, "%s: malloc return failure: %s\n",
 | |
| 			params->cmdname, strerror(errno));
 | |
| 		exit(EXIT_FAILURE);
 | |
| 	}
 | |
| 
 | |
| 	memset(hdr, 0, alloc_len);
 | |
| 	tparams->header_size = alloc_len;
 | |
| 	tparams->hdr = hdr;
 | |
| 
 | |
| 	/*
 | |
| 	 * The resulting image needs to be 4-byte aligned. At least
 | |
| 	 * the Marvell hdrparser tool complains if its unaligned.
 | |
| 	 * By returning 1 here in this function, called via
 | |
| 	 * tparams->vrec_header() in mkimage.c, mkimage will
 | |
| 	 * automatically pad the the resulting image to a 4-byte
 | |
| 	 * size if necessary.
 | |
| 	 */
 | |
| 	return 1;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * Report Error if xflag is set in addition to default
 | |
|  */
 | |
| static int kwbimage_check_params(struct image_tool_params *params)
 | |
| {
 | |
| 	if (!strlen(params->imagename)) {
 | |
| 		fprintf(stderr, "Error:%s - Configuration file not specified, "
 | |
| 			"it is needed for kwbimage generation\n",
 | |
| 			params->cmdname);
 | |
| 		return CFG_INVALID;
 | |
| 	}
 | |
| 
 | |
| 	return (params->dflag && (params->fflag || params->lflag)) ||
 | |
| 		(params->fflag && (params->dflag || params->lflag)) ||
 | |
| 		(params->lflag && (params->dflag || params->fflag)) ||
 | |
| 		(params->xflag) || !(strlen(params->imagename));
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * kwbimage type parameters definition
 | |
|  */
 | |
| U_BOOT_IMAGE_TYPE(
 | |
| 	kwbimage,
 | |
| 	"Marvell MVEBU Boot Image support",
 | |
| 	0,
 | |
| 	NULL,
 | |
| 	kwbimage_check_params,
 | |
| 	kwbimage_verify_header,
 | |
| 	kwbimage_print_header,
 | |
| 	kwbimage_set_header,
 | |
| 	NULL,
 | |
| 	kwbimage_check_image_types,
 | |
| 	NULL,
 | |
| 	kwbimage_generate
 | |
| );
 |