[J-u-boot.git] / tools / pblimage.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2012-2014 Freescale Semiconductor, Inc.
 */
#include "imagetool.h"
#include <image.h>
#include "pblimage.h"
#include "pbl_crc32.h"
#include <u-boot/crc.h>

#define roundup(x, y)		((((x) + ((y) - 1)) / (y)) * (y))
#define PBL_ACS_CONT_CMD	0x81000000
#define PBL_ADDR_24BIT_MASK	0x00ffffff

/*
 * Initialize to an invalid value.
 */
static uint32_t next_pbl_cmd = 0x82000000;
/*
 * need to store all bytes in memory for calculating crc32, then write the
 * bytes to image file for PBL boot.
 */
static unsigned char mem_buf[1000000];
static unsigned char *pmem_buf = mem_buf;
static int pbl_size;
static char *fname = "Unknown";
static int lineno = -1;
static struct pbl_header pblimage_header;
static int uboot_size;
static int arch_flag;

static uint32_t pbl_cmd_initaddr;
static uint32_t pbi_crc_cmd1;
static uint32_t pbi_crc_cmd2;
static uint32_t pbl_end_cmd[4];

static union
{
	char c[4];
	unsigned char l;
} endian_test = { {'l', '?', '?', 'b'} };

#define ENDIANNESS ((char)endian_test.l)

/*
 * The PBL can load up to 64 bytes at a time, so we split the U-Boot
 * image into 64 byte chunks. PBL needs a command for each piece, of
 * the form "81xxxxxx", where "xxxxxx" is the offset. Calculate the
 * start offset by subtracting the size of the u-boot image from the
 * top of the allowable 24-bit range.
 */
static void generate_pbl_cmd(void)
{
	uint32_t val = next_pbl_cmd;
	next_pbl_cmd += 0x40;
	int i;

	for (i = 3; i >= 0; i--) {
		*pmem_buf++ = (val >> (i * 8)) & 0xff;
		pbl_size++;
	}
}

static void pbl_fget(size_t size, FILE *stream)
{
	unsigned char c = 0xff;
	int c_temp;

	while (size) {
		c_temp = fgetc(stream);
		if (c_temp != EOF)
			c = (unsigned char)c_temp;
		else if ((c_temp == EOF) && (arch_flag == IH_ARCH_ARM))
			c = 0xff;
		*pmem_buf++ = c;
		pbl_size++;
		size--;
	}
}

/* load split u-boot with PBI command 81xxxxxx. */
static void load_uboot(FILE *fp_uboot)
{
	next_pbl_cmd = pbl_cmd_initaddr - uboot_size;
	while (next_pbl_cmd < pbl_cmd_initaddr) {
		generate_pbl_cmd();
		pbl_fget(64, fp_uboot);
	}
}

static void check_get_hexval(char *token)
{
	uint32_t hexval;
	int i;

	if (!sscanf(token, "%x", &hexval)) {
		printf("Error:%s[%d] - Invalid hex data(%s)\n", fname,
			lineno, token);
		exit(EXIT_FAILURE);
	}
	for (i = 3; i >= 0; i--) {
		*pmem_buf++ = (hexval >> (i * 8)) & 0xff;
		pbl_size++;
	}
}

static void pbl_parser(char *name)
{
	FILE *fd = NULL;
	char *line = NULL;
	char *token, *saveptr1, *saveptr2;
	size_t len = 0;

	fname = name;
	fd = fopen(name, "r");
	if (fd == NULL) {
		printf("Error:%s - Can't open\n", fname);
		exit(EXIT_FAILURE);
	}

	while ((getline(&line, &len, fd)) > 0) {
		lineno++;
		token = strtok_r(line, "\r\n", &saveptr1);
		/* drop all lines with zero tokens (= empty lines) */
		if (token == NULL)
			continue;
		for (line = token;; line = NULL) {
			token = strtok_r(line, " \t", &saveptr2);
			if (token == NULL)
				break;
			/* Drop all text starting with '#' as comments */
			if (token[0] == '#')
				break;
			check_get_hexval(token);
		}
	}
	if (line)
		free(line);
	fclose(fd);
}

static uint32_t reverse_byte(uint32_t val)
{
	uint32_t temp;
	unsigned char *p1;
	int j;

	temp = val;
	p1 = (unsigned char *)&temp;
	for (j = 3; j >= 0; j--)
		*p1++ = (val >> (j * 8)) & 0xff;
	return temp;
}

/* write end command and crc command to memory. */
static void add_end_cmd(void)
{
	uint32_t crc32_pbl;
	int i;
	unsigned char *p = (unsigned char *)&pbl_end_cmd;

	if (ENDIANNESS == 'l') {
		for (i = 0; i < 4; i++)
			pbl_end_cmd[i] = reverse_byte(pbl_end_cmd[i]);
	}

	for (i = 0; i < 16; i++) {
		*pmem_buf++ = *p++;
		pbl_size++;
	}

	/* Add PBI CRC command. */
	*pmem_buf++ = 0x08;
	*pmem_buf++ = pbi_crc_cmd1;
	*pmem_buf++ = pbi_crc_cmd2;
	*pmem_buf++ = 0x40;
	pbl_size += 4;

	/* calculated CRC32 and write it to memory. */
	crc32_pbl = pbl_crc32(0, (const char *)mem_buf, pbl_size);
	*pmem_buf++ = (crc32_pbl >> 24) & 0xff;
	*pmem_buf++ = (crc32_pbl >> 16) & 0xff;
	*pmem_buf++ = (crc32_pbl >> 8) & 0xff;
	*pmem_buf++ = (crc32_pbl) & 0xff;
	pbl_size += 4;
}

void pbl_load_uboot(int ifd, struct image_tool_params *params)
{
	FILE *fp_uboot;
	int size;

	/* parse the rcw.cfg file. */
	pbl_parser(params->imagename);

	/* parse the pbi.cfg file. */
	if (params->imagename2[0] != '\0')
		pbl_parser(params->imagename2);

	if (params->datafile) {
		fp_uboot = fopen(params->datafile, "r");
		if (fp_uboot == NULL) {
			printf("Error: %s open failed\n", params->datafile);
			exit(EXIT_FAILURE);
		}

		load_uboot(fp_uboot);
		fclose(fp_uboot);
	}
	add_end_cmd();
	lseek(ifd, 0, SEEK_SET);

	size = pbl_size;
	if (write(ifd, (const void *)&mem_buf, size) != size) {
		fprintf(stderr, "Write error on %s: %s\n",
			params->imagefile, strerror(errno));
		exit(EXIT_FAILURE);
	}
}

static int pblimage_check_image_types(uint8_t type)
{
	if (type == IH_TYPE_PBLIMAGE)
		return EXIT_SUCCESS;
	else
		return EXIT_FAILURE;
}

static int pblimage_verify_header(unsigned char *ptr, int image_size,
			struct image_tool_params *params)
{
	struct pbl_header *pbl_hdr = (struct pbl_header *) ptr;

	/* Only a few checks can be done: search for magic numbers */
	if (ENDIANNESS == 'l') {
		if (pbl_hdr->preamble != reverse_byte(RCW_PREAMBLE))
			return -FDT_ERR_BADSTRUCTURE;

		if (pbl_hdr->rcwheader != reverse_byte(RCW_HEADER))
			return -FDT_ERR_BADSTRUCTURE;
	} else {
		if (pbl_hdr->preamble != RCW_PREAMBLE)
			return -FDT_ERR_BADSTRUCTURE;

		if (pbl_hdr->rcwheader != RCW_HEADER)
			return -FDT_ERR_BADSTRUCTURE;
	}
	return 0;
}

static void pblimage_print_header(const void *ptr)
{
	printf("Image Type:   Freescale PBL Boot Image\n");
}

static void pblimage_set_header(void *ptr, struct stat *sbuf, int ifd,
				struct image_tool_params *params)
{
	/*nothing need to do, pbl_load_uboot takes care of whole file. */
}

int pblimage_check_params(struct image_tool_params *params)
{
	FILE *fp_uboot;
	int fd;
	struct stat st;

	if (!params)
		return EXIT_FAILURE;

	if (params->datafile) {
		fp_uboot = fopen(params->datafile, "r");
		if (fp_uboot == NULL) {
			printf("Error: %s open failed\n", params->datafile);
			exit(EXIT_FAILURE);
		}
		fd = fileno(fp_uboot);

		if (fstat(fd, &st) == -1) {
			printf("Error: Could not determine u-boot image size. %s\n",
			       strerror(errno));
			exit(EXIT_FAILURE);
		}

		/* For the variable size, pad it to 64 byte boundary */
		uboot_size = roundup(st.st_size, 64);
		fclose(fp_uboot);
	}

	if (params->arch == IH_ARCH_ARM) {
		arch_flag = IH_ARCH_ARM;
		pbi_crc_cmd1 = 0x61;
		pbi_crc_cmd2 = 0;
		pbl_cmd_initaddr = params->addr & PBL_ADDR_24BIT_MASK;
		pbl_cmd_initaddr |= PBL_ACS_CONT_CMD;
		pbl_cmd_initaddr += uboot_size;
		pbl_end_cmd[0] = 0x09610000;
		pbl_end_cmd[1] = 0x00000000;
		pbl_end_cmd[2] = 0x096100c0;
		pbl_end_cmd[3] = 0x00000000;
	} else if (params->arch == IH_ARCH_PPC) {
		arch_flag = IH_ARCH_PPC;
		pbi_crc_cmd1 = 0x13;
		pbi_crc_cmd2 = 0x80;
		pbl_cmd_initaddr = 0x82000000;
		pbl_end_cmd[0] = 0x091380c0;
		pbl_end_cmd[1] = 0x00000000;
		pbl_end_cmd[2] = 0x091380c0;
		pbl_end_cmd[3] = 0x00000000;
	}

	next_pbl_cmd = pbl_cmd_initaddr;
	return 0;
};

/* pblimage parameters */
U_BOOT_IMAGE_TYPE(
	pblimage,
	"Freescale PBL Boot Image support",
	sizeof(struct pbl_header),
	(void *)&pblimage_header,
	pblimage_check_params,
	pblimage_verify_header,
	pblimage_print_header,
	pblimage_set_header,
	NULL,
	pblimage_check_image_types,
	NULL,
	NULL
);
Commit	Line	Data
83d290c5	1	// SPDX-License-Identifier: GPL-2.0+
5d898a00	2	/*
94cb17d0	3	* Copyright 2012-2014 Freescale Semiconductor, Inc.
5d898a00	4	*/
f86ed6a8	5	#include "imagetool.h"
5d898a00 SX	6	#include <image.h>
5d898a00 SX	7	#include "pblimage.h"
abbc67ee	8	#include "pbl_crc32.h"
3db71108	9	#include <u-boot/crc.h>
5d898a00	10
94cb17d0 AW	11	#define roundup(x, y) ((((x) + ((y) - 1)) / (y)) * (y))
	12	#define PBL_ACS_CONT_CMD 0x81000000
	13	#define PBL_ADDR_24BIT_MASK 0x00ffffff
	14
5d898a00	15	/*
0d3efd80	16	* Initialize to an invalid value.
5d898a00	17	*/
0d3efd80	18	static uint32_t next_pbl_cmd = 0x82000000;
5d898a00 SX	19	/*
	20	* need to store all bytes in memory for calculating crc32, then write the
	21	* bytes to image file for PBL boot.
	22	*/
0d3efd80	23	static unsigned char mem_buf[1000000];
5d898a00 SX	24	static unsigned char *pmem_buf = mem_buf;
	25	static int pbl_size;
	26	static char *fname = "Unknown";
	27	static int lineno = -1;
	28	static struct pbl_header pblimage_header;
94cb17d0 AW	29	static int uboot_size;
	30	static int arch_flag;
	31
	32	static uint32_t pbl_cmd_initaddr;
	33	static uint32_t pbi_crc_cmd1;
	34	static uint32_t pbi_crc_cmd2;
	35	static uint32_t pbl_end_cmd[4];
5d898a00 SX	36
	37	static union
	38	{
	39	char c[4];
	40	unsigned char l;
	41	} endian_test = { {'l', '?', '?', 'b'} };
	42
	43	#define ENDIANNESS ((char)endian_test.l)
	44
0d3efd80 CP	45	/*
	46	* The PBL can load up to 64 bytes at a time, so we split the U-Boot
	47	* image into 64 byte chunks. PBL needs a command for each piece, of
	48	* the form "81xxxxxx", where "xxxxxx" is the offset. Calculate the
	49	* start offset by subtracting the size of the u-boot image from the
	50	* top of the allowable 24-bit range.
	51	*/
5d898a00 SX	52	static void generate_pbl_cmd(void)
	53	{
	54	uint32_t val = next_pbl_cmd;
	55	next_pbl_cmd += 0x40;
	56	int i;
	57
	58	for (i = 3; i >= 0; i--) {
	59	pmem_buf++ = (val >> (i 8)) & 0xff;
	60	pbl_size++;
	61	}
	62	}
	63
	64	static void pbl_fget(size_t size, FILE *stream)
	65	{
94cb17d0	66	unsigned char c = 0xff;
5d898a00 SX	67	int c_temp;
5d898a00 SX	68
94cb17d0 AW	69	while (size) {
	70	c_temp = fgetc(stream);
	71	if (c_temp != EOF)
	72	c = (unsigned char)c_temp;
	73	else if ((c_temp == EOF) && (arch_flag == IH_ARCH_ARM))
	74	c = 0xff;
5d898a00 SX	75	*pmem_buf++ = c;
	76	pbl_size++;
	77	size--;
	78	}
	79	}
	80
	81	/* load split u-boot with PBI command 81xxxxxx. */
	82	static void load_uboot(FILE *fp_uboot)
	83	{
94cb17d0 AW	84	next_pbl_cmd = pbl_cmd_initaddr - uboot_size;
94cb17d0 AW	85	while (next_pbl_cmd < pbl_cmd_initaddr) {
5d898a00 SX	86	generate_pbl_cmd();
	87	pbl_fget(64, fp_uboot);
	88	}
	89	}
	90
	91	static void check_get_hexval(char *token)
	92	{
	93	uint32_t hexval;
	94	int i;
	95
	96	if (!sscanf(token, "%x", &hexval)) {
	97	printf("Error:%s[%d] - Invalid hex data(%s)\n", fname,
	98	lineno, token);
	99	exit(EXIT_FAILURE);
	100	}
	101	for (i = 3; i >= 0; i--) {
	102	pmem_buf++ = (hexval >> (i 8)) & 0xff;
	103	pbl_size++;
	104	}
	105	}
	106
	107	static void pbl_parser(char *name)
	108	{
	109	FILE *fd = NULL;
	110	char *line = NULL;
	111	char token, saveptr1, *saveptr2;
	112	size_t len = 0;
	113
	114	fname = name;
	115	fd = fopen(name, "r");
	116	if (fd == NULL) {
	117	printf("Error:%s - Can't open\n", fname);
	118	exit(EXIT_FAILURE);
	119	}
	120
	121	while ((getline(&line, &len, fd)) > 0) {
	122	lineno++;
	123	token = strtok_r(line, "\r\n", &saveptr1);
	124	/* drop all lines with zero tokens (= empty lines) */
	125	if (token == NULL)
	126	continue;
	127	for (line = token;; line = NULL) {
	128	token = strtok_r(line, " \t", &saveptr2);
	129	if (token == NULL)
	130	break;
	131	/* Drop all text starting with '#' as comments */
	132	if (token[0] == '#')
	133	break;
	134	check_get_hexval(token);
	135	}
	136	}
	137	if (line)
	138	free(line);
	139	fclose(fd);
	140	}
	141
5d898a00 SX	142	static uint32_t reverse_byte(uint32_t val)
	143	{
	144	uint32_t temp;
	145	unsigned char *p1;
	146	int j;
	147
	148	temp = val;
	149	p1 = (unsigned char *)&temp;
	150	for (j = 3; j >= 0; j--)
	151	p1++ = (val >> (j 8)) & 0xff;
	152	return temp;
	153	}
	154
	155	/* write end command and crc command to memory. */
	156	static void add_end_cmd(void)
	157	{
5d898a00 SX	158	uint32_t crc32_pbl;
	159	int i;
	160	unsigned char p = (unsigned char )&pbl_end_cmd;
	161
	162	if (ENDIANNESS == 'l') {
	163	for (i = 0; i < 4; i++)
	164	pbl_end_cmd[i] = reverse_byte(pbl_end_cmd[i]);
	165	}
	166
	167	for (i = 0; i < 16; i++) {
	168	pmem_buf++ = p++;
	169	pbl_size++;
	170	}
	171
	172	/* Add PBI CRC command. */
	173	*pmem_buf++ = 0x08;
94cb17d0 AW	174	*pmem_buf++ = pbi_crc_cmd1;
94cb17d0 AW	175	*pmem_buf++ = pbi_crc_cmd2;
5d898a00 SX	176	*pmem_buf++ = 0x40;
	177	pbl_size += 4;
	178
	179	/* calculated CRC32 and write it to memory. */
	180	crc32_pbl = pbl_crc32(0, (const char *)mem_buf, pbl_size);
	181	*pmem_buf++ = (crc32_pbl >> 24) & 0xff;
	182	*pmem_buf++ = (crc32_pbl >> 16) & 0xff;
	183	*pmem_buf++ = (crc32_pbl >> 8) & 0xff;
	184	*pmem_buf++ = (crc32_pbl) & 0xff;
	185	pbl_size += 4;
5d898a00 SX	186	}
5d898a00 SX	187
f86ed6a8	188	void pbl_load_uboot(int ifd, struct image_tool_params *params)
5d898a00 SX	189	{
	190	FILE *fp_uboot;
	191	int size;
	192
	193	/* parse the rcw.cfg file. */
	194	pbl_parser(params->imagename);
	195
	196	/* parse the pbi.cfg file. */
33ed5749	197	if (params->imagename2[0] != '\0')
	198	pbl_parser(params->imagename2);
	199
	200	if (params->datafile) {
	201	fp_uboot = fopen(params->datafile, "r");
	202	if (fp_uboot == NULL) {
	203	printf("Error: %s open failed\n", params->datafile);
	204	exit(EXIT_FAILURE);
	205	}
5d898a00	206
33ed5749	207	load_uboot(fp_uboot);
33ed5749	208	fclose(fp_uboot);
5d898a00	209	}
5d898a00	210	add_end_cmd();
5d898a00 SX	211	lseek(ifd, 0, SEEK_SET);
	212
	213	size = pbl_size;
	214	if (write(ifd, (const void *)&mem_buf, size) != size) {
	215	fprintf(stderr, "Write error on %s: %s\n",
	216	params->imagefile, strerror(errno));
	217	exit(EXIT_FAILURE);
	218	}
	219	}
	220
	221	static int pblimage_check_image_types(uint8_t type)
	222	{
	223	if (type == IH_TYPE_PBLIMAGE)
	224	return EXIT_SUCCESS;
	225	else
	226	return EXIT_FAILURE;
	227	}
	228
	229	static int pblimage_verify_header(unsigned char *ptr, int image_size,
f86ed6a8	230	struct image_tool_params *params)
5d898a00 SX	231	{
	232	struct pbl_header pbl_hdr = (struct pbl_header ) ptr;
	233
	234	/* Only a few checks can be done: search for magic numbers */
	235	if (ENDIANNESS == 'l') {
	236	if (pbl_hdr->preamble != reverse_byte(RCW_PREAMBLE))
	237	return -FDT_ERR_BADSTRUCTURE;
	238
	239	if (pbl_hdr->rcwheader != reverse_byte(RCW_HEADER))
	240	return -FDT_ERR_BADSTRUCTURE;
	241	} else {
	242	if (pbl_hdr->preamble != RCW_PREAMBLE)
	243	return -FDT_ERR_BADSTRUCTURE;
	244
	245	if (pbl_hdr->rcwheader != RCW_HEADER)
	246	return -FDT_ERR_BADSTRUCTURE;
	247	}
	248	return 0;
	249	}
	250
	251	static void pblimage_print_header(const void *ptr)
	252	{
	253	printf("Image Type: Freescale PBL Boot Image\n");
	254	}
	255
	256	static void pblimage_set_header(void ptr, struct stat sbuf, int ifd,
f86ed6a8	257	struct image_tool_params *params)
5d898a00 SX	258	{
	259	/nothing need to do, pbl_load_uboot takes care of whole file. /
	260	}
	261
94cb17d0 AW	262	int pblimage_check_params(struct image_tool_params *params)
	263	{
	264	FILE *fp_uboot;
	265	int fd;
	266	struct stat st;
	267
	268	if (!params)
	269	return EXIT_FAILURE;
	270
33ed5749	271	if (params->datafile) {
	272	fp_uboot = fopen(params->datafile, "r");
	273	if (fp_uboot == NULL) {
	274	printf("Error: %s open failed\n", params->datafile);
	275	exit(EXIT_FAILURE);
	276	}
	277	fd = fileno(fp_uboot);
94cb17d0	278
33ed5749	279	if (fstat(fd, &st) == -1) {
	280	printf("Error: Could not determine u-boot image size. %s\n",
	281	strerror(errno));
	282	exit(EXIT_FAILURE);
	283	}
94cb17d0	284
33ed5749	285	/* For the variable size, pad it to 64 byte boundary */
	286	uboot_size = roundup(st.st_size, 64);
	287	fclose(fp_uboot);
	288	}
94cb17d0 AW	289
	290	if (params->arch == IH_ARCH_ARM) {
	291	arch_flag = IH_ARCH_ARM;
	292	pbi_crc_cmd1 = 0x61;
	293	pbi_crc_cmd2 = 0;
	294	pbl_cmd_initaddr = params->addr & PBL_ADDR_24BIT_MASK;
	295	pbl_cmd_initaddr \|= PBL_ACS_CONT_CMD;
05e35d26	296	pbl_cmd_initaddr += uboot_size;
94cb17d0 AW	297	pbl_end_cmd[0] = 0x09610000;
	298	pbl_end_cmd[1] = 0x00000000;
	299	pbl_end_cmd[2] = 0x096100c0;
	300	pbl_end_cmd[3] = 0x00000000;
	301	} else if (params->arch == IH_ARCH_PPC) {
	302	arch_flag = IH_ARCH_PPC;
	303	pbi_crc_cmd1 = 0x13;
	304	pbi_crc_cmd2 = 0x80;
	305	pbl_cmd_initaddr = 0x82000000;
c5938c10	306	pbl_end_cmd[0] = 0x091380c0;
94cb17d0 AW	307	pbl_end_cmd[1] = 0x00000000;
	308	pbl_end_cmd[2] = 0x091380c0;
	309	pbl_end_cmd[3] = 0x00000000;
	310	}
	311
	312	next_pbl_cmd = pbl_cmd_initaddr;
	313	return 0;
	314	};
	315
5d898a00	316	/* pblimage parameters */
a93648d1 GMF	317	U_BOOT_IMAGE_TYPE(
	318	pblimage,
	319	"Freescale PBL Boot Image support",
	320	sizeof(struct pbl_header),
	321	(void *)&pblimage_header,
	322	pblimage_check_params,
	323	pblimage_verify_header,
	324	pblimage_print_header,
	325	pblimage_set_header,
	326	NULL,
	327	pblimage_check_image_types,
	328	NULL,
	329	NULL
	330	);