[thirdparty/u-boot.git] / tools / sunxi_egon.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * (C) Copyright 2018 Arm Ltd.
 */

#include "imagetool.h"
#include <image.h>

#include <sunxi_image.h>

/*
 * NAND requires 8K padding. SD/eMMC gets away with 512 bytes,
 * but let's use the larger padding by default to cover both.
 */
#define PAD_SIZE			8192
#define PAD_SIZE_MIN			512

static int egon_get_arch(struct image_tool_params *params)
{
	if (params->Aflag)
		return params->arch;

	/* For compatibility, assume ARM when no architecture specified */
	return IH_ARCH_ARM;
}

static int egon_check_params(struct image_tool_params *params)
{
	/*
	 * Check whether the architecture is supported.
	 */
	switch (egon_get_arch(params)) {
	case IH_ARCH_ARM:
	case IH_ARCH_RISCV:
		break;
	default:
		return EXIT_FAILURE;
	}

	/* We need a binary image file. */
	return !params->dflag;
}

static int egon_verify_header(unsigned char *ptr, int image_size,
			      struct image_tool_params *params)
{
	const struct boot_file_head *header = (void *)ptr;
	uint32_t length;

	/*
	 * First 4 bytes must be a branch instruction of the corresponding
	 * architecture.
	 */
	switch (egon_get_arch(params)) {
	case IH_ARCH_ARM:
		if ((le32_to_cpu(header->b_instruction) & 0xff000000) != 0xea000000)
			return EXIT_FAILURE;
		break;
	case IH_ARCH_RISCV:
		if ((le32_to_cpu(header->b_instruction) & 0x00000fff) != 0x0000006f)
			return EXIT_FAILURE;
		break;
	default:
		return EXIT_FAILURE; /* Unknown architecture */
	}

	if (memcmp(header->magic, BOOT0_MAGIC, sizeof(header->magic)))
		return EXIT_FAILURE;

	length = le32_to_cpu(header->length);
	/* Must be at least 512 byte aligned. */
	if (length & 511)
		return EXIT_FAILURE;

	/*
	 * Image could also contain U-Boot proper, so could be bigger.
	 * But it must not be shorter.
	 */
	if (image_size < length)
		return EXIT_FAILURE;

	return EXIT_SUCCESS;
}

static void egon_print_header(const void *buf, struct image_tool_params *params)
{
	const struct boot_file_head *header = buf;

	printf("Allwinner eGON image, size: %d bytes\n",
	       le32_to_cpu(header->length));

	if (memcmp(header->spl_signature, SPL_SIGNATURE, 3))
		return;

	printf("\tSPL header version %d.%d\n",
	       header->spl_signature[3] >> SPL_MINOR_BITS,
	       header->spl_signature[3] & ((1U << SPL_MINOR_BITS) - 1));
	if (header->spl_signature[3] >= SPL_DT_HEADER_VERSION) {
		uint32_t dt_name_offs = le32_to_cpu(header->dt_name_offset);

		if (dt_name_offs > 0)
			printf("\tDT name: %s\n", (char *)buf + dt_name_offs);
	}
}

static void egon_set_header(void *buf, struct stat *sbuf, int infd,
			    struct image_tool_params *params)
{
	struct boot_file_head *header = buf;
	uint32_t *buf32 = buf;
	uint32_t checksum = 0, value;
	int i;

	/*
	 * Different architectures need different first instruction to
	 * branch to the body.
	 */
	switch (egon_get_arch(params)) {
	case IH_ARCH_ARM:
		/* Generate an ARM branch instruction to jump over the header. */
		value = 0xea000000 | (sizeof(struct boot_file_head) / 4 - 2);
		header->b_instruction = cpu_to_le32(value);
		break;
	case IH_ARCH_RISCV:
		/*
		 * Generate a RISC-V JAL instruction with rd=x0
		 * (pseudo instruction J, jump without side effects).
		 *
		 * The following weird bit operation maps imm[20]
		 * to inst[31], imm[10:1] to inst[30:21],
		 * imm[11] to inst[20], imm[19:12] to inst[19:12],
		 * and imm[0] is dropped (because 1-byte RISC-V instruction
		 * is not allowed).
		 */
		value = 0x0000006f |
			((sizeof(struct boot_file_head) & 0x00100000) << 11) |
			((sizeof(struct boot_file_head) & 0x000007fe) << 20) |
			((sizeof(struct boot_file_head) & 0x00000800) << 9) |
			((sizeof(struct boot_file_head) & 0x000ff000) << 0);
		header->b_instruction = cpu_to_le32(value);
		break;
	}

	memcpy(header->magic, BOOT0_MAGIC, sizeof(header->magic));
	header->check_sum = cpu_to_le32(BROM_STAMP_VALUE);
	header->length = cpu_to_le32(params->file_size);

	memcpy(header->spl_signature, SPL_SIGNATURE, 3);
	header->spl_signature[3] = SPL_ENV_HEADER_VERSION;

	/* If an image name has been provided, use it as the DT name. */
	if (params->imagename && params->imagename[0]) {
		if (strlen(params->imagename) > sizeof(header->string_pool) - 1)
			printf("WARNING: DT name too long for SPL header!\n");
		else {
			strcpy((char *)header->string_pool, params->imagename);
			value = offsetof(struct boot_file_head, string_pool);
			header->dt_name_offset = cpu_to_le32(value);
			header->spl_signature[3] = SPL_DT_HEADER_VERSION;
		}
	}

	/* Calculate the checksum. Yes, it's that simple. */
	for (i = 0; i < sbuf->st_size / 4; i++)
		checksum += le32_to_cpu(buf32[i]);
	header->check_sum = cpu_to_le32(checksum);
}

static int egon_check_image_type(uint8_t type)
{
	return type == IH_TYPE_SUNXI_EGON ? 0 : 1;
}

static int egon_vrec_header(struct image_tool_params *params,
			    struct image_type_params *tparams)
{
	int pad_size = ALIGN(params->bl_len ?: PAD_SIZE, PAD_SIZE_MIN);

	tparams->hdr = calloc(sizeof(struct boot_file_head), 1);

	/* Return padding to complete blocks. */
	return ALIGN(params->file_size, pad_size) - params->file_size;
}

U_BOOT_IMAGE_TYPE(
	sunxi_egon,
	"Allwinner eGON Boot Image support",
	sizeof(struct boot_file_head),
	NULL,
	egon_check_params,
	egon_verify_header,
	egon_print_header,
	egon_set_header,
	NULL,
	egon_check_image_type,
	NULL,
	egon_vrec_header
);
Commit	Line	Data
6d295099 AP	1	// SPDX-License-Identifier: GPL-2.0+
	2	/*
	3	* (C) Copyright 2018 Arm Ltd.
	4	*/
	5
	6	#include "imagetool.h"
	7	#include <image.h>
	8
	9	#include <sunxi_image.h>
	10
	11	/*
	12	* NAND requires 8K padding. SD/eMMC gets away with 512 bytes,
50d5c642	13	* but let's use the larger padding by default to cover both.
6d295099 AP	14	*/
6d295099 AP	15	#define PAD_SIZE 8192
50d5c642	16	#define PAD_SIZE_MIN 512
6d295099	17
82ae151a IZ	18	static int egon_get_arch(struct image_tool_params *params)
	19	{
	20	if (params->Aflag)
	21	return params->arch;
	22
	23	/* For compatibility, assume ARM when no architecture specified */
	24	return IH_ARCH_ARM;
	25	}
	26
6d295099 AP	27	static int egon_check_params(struct image_tool_params *params)
6d295099 AP	28	{
82ae151a IZ	29	/*
	30	* Check whether the architecture is supported.
	31	*/
	32	switch (egon_get_arch(params)) {
	33	case IH_ARCH_ARM:
78ac2c0f	34	case IH_ARCH_RISCV:
82ae151a IZ	35	break;
	36	default:
	37	return EXIT_FAILURE;
	38	}
	39
	40	/* We need a binary image file. */
6d295099 AP	41	return !params->dflag;
	42	}
	43
	44	static int egon_verify_header(unsigned char *ptr, int image_size,
	45	struct image_tool_params *params)
	46	{
	47	const struct boot_file_head header = (void )ptr;
	48	uint32_t length;
	49
82ae151a IZ	50	/*
	51	* First 4 bytes must be a branch instruction of the corresponding
	52	* architecture.
	53	*/
	54	switch (egon_get_arch(params)) {
	55	case IH_ARCH_ARM:
	56	if ((le32_to_cpu(header->b_instruction) & 0xff000000) != 0xea000000)
	57	return EXIT_FAILURE;
	58	break;
78ac2c0f IZ	59	case IH_ARCH_RISCV:
	60	if ((le32_to_cpu(header->b_instruction) & 0x00000fff) != 0x0000006f)
	61	return EXIT_FAILURE;
	62	break;
82ae151a IZ	63	default:
	64	return EXIT_FAILURE; /* Unknown architecture */
	65	}
6d295099 AP	66
	67	if (memcmp(header->magic, BOOT0_MAGIC, sizeof(header->magic)))
	68	return EXIT_FAILURE;
	69
	70	length = le32_to_cpu(header->length);
	71	/* Must be at least 512 byte aligned. */
	72	if (length & 511)
	73	return EXIT_FAILURE;
	74
	75	/*
	76	* Image could also contain U-Boot proper, so could be bigger.
	77	* But it must not be shorter.
	78	*/
	79	if (image_size < length)
	80	return EXIT_FAILURE;
	81
	82	return EXIT_SUCCESS;
	83	}
	84
2972d7d6	85	static void egon_print_header(const void buf, struct image_tool_params params)
6d295099 AP	86	{
	87	const struct boot_file_head *header = buf;
	88
	89	printf("Allwinner eGON image, size: %d bytes\n",
	90	le32_to_cpu(header->length));
	91
	92	if (memcmp(header->spl_signature, SPL_SIGNATURE, 3))
	93	return;
	94
	95	printf("\tSPL header version %d.%d\n",
	96	header->spl_signature[3] >> SPL_MINOR_BITS,
	97	header->spl_signature[3] & ((1U << SPL_MINOR_BITS) - 1));
	98	if (header->spl_signature[3] >= SPL_DT_HEADER_VERSION) {
	99	uint32_t dt_name_offs = le32_to_cpu(header->dt_name_offset);
	100
	101	if (dt_name_offs > 0)
	102	printf("\tDT name: %s\n", (char *)buf + dt_name_offs);
	103	}
	104	}
	105
	106	static void egon_set_header(void buf, struct stat sbuf, int infd,
	107	struct image_tool_params *params)
	108	{
	109	struct boot_file_head *header = buf;
	110	uint32_t *buf32 = buf;
	111	uint32_t checksum = 0, value;
	112	int i;
	113
82ae151a IZ	114	/*
	115	* Different architectures need different first instruction to
	116	* branch to the body.
	117	*/
	118	switch (egon_get_arch(params)) {
	119	case IH_ARCH_ARM:
	120	/* Generate an ARM branch instruction to jump over the header. */
	121	value = 0xea000000 \| (sizeof(struct boot_file_head) / 4 - 2);
	122	header->b_instruction = cpu_to_le32(value);
	123	break;
78ac2c0f IZ	124	case IH_ARCH_RISCV:
	125	/*
	126	* Generate a RISC-V JAL instruction with rd=x0
	127	* (pseudo instruction J, jump without side effects).
	128	*
	129	* The following weird bit operation maps imm[20]
	130	* to inst[31], imm[10:1] to inst[30:21],
	131	* imm[11] to inst[20], imm[19:12] to inst[19:12],
	132	* and imm[0] is dropped (because 1-byte RISC-V instruction
	133	* is not allowed).
	134	*/
	135	value = 0x0000006f \|
	136	((sizeof(struct boot_file_head) & 0x00100000) << 11) \|
	137	((sizeof(struct boot_file_head) & 0x000007fe) << 20) \|
	138	((sizeof(struct boot_file_head) & 0x00000800) << 9) \|
	139	((sizeof(struct boot_file_head) & 0x000ff000) << 0);
	140	header->b_instruction = cpu_to_le32(value);
	141	break;
82ae151a	142	}
6d295099 AP	143
	144	memcpy(header->magic, BOOT0_MAGIC, sizeof(header->magic));
	145	header->check_sum = cpu_to_le32(BROM_STAMP_VALUE);
	146	header->length = cpu_to_le32(params->file_size);
	147
	148	memcpy(header->spl_signature, SPL_SIGNATURE, 3);
	149	header->spl_signature[3] = SPL_ENV_HEADER_VERSION;
	150
	151	/* If an image name has been provided, use it as the DT name. */
	152	if (params->imagename && params->imagename[0]) {
	153	if (strlen(params->imagename) > sizeof(header->string_pool) - 1)
	154	printf("WARNING: DT name too long for SPL header!\n");
	155	else {
	156	strcpy((char *)header->string_pool, params->imagename);
	157	value = offsetof(struct boot_file_head, string_pool);
	158	header->dt_name_offset = cpu_to_le32(value);
	159	header->spl_signature[3] = SPL_DT_HEADER_VERSION;
	160	}
	161	}
	162
	163	/* Calculate the checksum. Yes, it's that simple. */
	164	for (i = 0; i < sbuf->st_size / 4; i++)
	165	checksum += le32_to_cpu(buf32[i]);
	166	header->check_sum = cpu_to_le32(checksum);
	167	}
	168
	169	static int egon_check_image_type(uint8_t type)
	170	{
	171	return type == IH_TYPE_SUNXI_EGON ? 0 : 1;
	172	}
	173
	174	static int egon_vrec_header(struct image_tool_params *params,
	175	struct image_type_params *tparams)
	176	{
50d5c642 SH	177	int pad_size = ALIGN(params->bl_len ?: PAD_SIZE, PAD_SIZE_MIN);
50d5c642 SH	178
6d295099 AP	179	tparams->hdr = calloc(sizeof(struct boot_file_head), 1);
6d295099 AP	180
50d5c642 SH	181	/* Return padding to complete blocks. */
50d5c642 SH	182	return ALIGN(params->file_size, pad_size) - params->file_size;
6d295099 AP	183	}
	184
	185	U_BOOT_IMAGE_TYPE(
	186	sunxi_egon,
	187	"Allwinner eGON Boot Image support",
	188	sizeof(struct boot_file_head),
	189	NULL,
	190	egon_check_params,
	191	egon_verify_header,
	192	egon_print_header,
	193	egon_set_header,
	194	NULL,
	195	egon_check_image_type,
	196	NULL,
	197	egon_vrec_header
	198	);