[people/ms/u-boot.git] / tools / mksunxiboot.c

/*
 * (C) Copyright 2007-2011
 * Allwinner Technology Co., Ltd. <www.allwinnertech.com>
 * Tom Cubie <tangliang@allwinnertech.com>
 *
 * a simple tool to generate bootable image for sunxi platform.
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */
#include <fcntl.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include "asm/arch/spl.h"

#define STAMP_VALUE                     0x5F0A6C39

/* check sum functon from sun4i boot code */
int gen_check_sum(struct boot_file_head *head_p)
{
	uint32_t length;
	uint32_t *buf;
	uint32_t loop;
	uint32_t i;
	uint32_t sum;

	length = le32_to_cpu(head_p->length);
	if ((length & 0x3) != 0)	/* must 4-byte-aligned */
		return -1;
	buf = (uint32_t *)head_p;
	head_p->check_sum = cpu_to_le32(STAMP_VALUE);	/* fill stamp */
	loop = length >> 2;

	/* calculate the sum */
	for (i = 0, sum = 0; i < loop; i++)
		sum += le32_to_cpu(buf[i]);

	/* write back check sum */
	head_p->check_sum = cpu_to_le32(sum);

	return 0;
}

#define ALIGN(x, a) __ALIGN_MASK((x), (typeof(x))(a)-1)
#define __ALIGN_MASK(x, mask) (((x)+(mask))&~(mask))

#define SUN4I_SRAM_SIZE 0x7600	/* 0x7748+ is used by BROM */
#define SRAM_LOAD_MAX_SIZE (SUN4I_SRAM_SIZE - sizeof(struct boot_file_head))

/*
 * BROM (at least on A10 and A20) requires NAND-images to be explicitly aligned
 * to a multiple of 8K, and rejects the image otherwise. MMC-images are fine
 * with 512B blocks. To cater for both, align to the largest of the two.
 */
#define BLOCK_SIZE 0x2000

struct boot_img {
	struct boot_file_head header;
	char code[SRAM_LOAD_MAX_SIZE];
	char pad[BLOCK_SIZE];
};

int main(int argc, char *argv[])
{
	int fd_in, fd_out;
	struct boot_img img;
	unsigned file_size;
	int count;

	if (argc < 2) {
		printf("\tThis program makes an input bin file to sun4i " \
		       "bootable image.\n" \
		       "\tUsage: %s input_file out_putfile\n", argv[0]);
		return EXIT_FAILURE;
	}

	fd_in = open(argv[1], O_RDONLY);
	if (fd_in < 0) {
		perror("Open input file");
		return EXIT_FAILURE;
	}

	memset(&img, 0, sizeof(img));

	/* get input file size */
	file_size = lseek(fd_in, 0, SEEK_END);

	if (file_size > SRAM_LOAD_MAX_SIZE) {
		fprintf(stderr, "ERROR: File too large!\n");
		return EXIT_FAILURE;
	}

	fd_out = open(argv[2], O_WRONLY | O_CREAT, 0666);
	if (fd_out < 0) {
		perror("Open output file");
		return EXIT_FAILURE;
	}

	/* read file to buffer to calculate checksum */
	lseek(fd_in, 0, SEEK_SET);
	count = read(fd_in, img.code, file_size);
	if (count != file_size) {
		perror("Reading input image");
		return EXIT_FAILURE;
	}

	/* fill the header */
	img.header.b_instruction =	/* b instruction */
		0xEA000000 |	/* jump to the first instr after the header */
		((sizeof(struct boot_file_head) / sizeof(int) - 2)
		 & 0x00FFFFFF);
	memcpy(img.header.magic, BOOT0_MAGIC, 8);	/* no '0' termination */
	img.header.length =
		ALIGN(file_size + sizeof(struct boot_file_head), BLOCK_SIZE);
	img.header.b_instruction = cpu_to_le32(img.header.b_instruction);
	img.header.length = cpu_to_le32(img.header.length);

	memcpy(img.header.spl_signature, SPL_SIGNATURE, 3); /* "sunxi" marker */
	img.header.spl_signature[3] = SPL_HEADER_VERSION;

	gen_check_sum(&img.header);

	count = write(fd_out, &img, le32_to_cpu(img.header.length));
	if (count != le32_to_cpu(img.header.length)) {
		perror("Writing output");
		return EXIT_FAILURE;
	}

	close(fd_in);
	close(fd_out);

	return EXIT_SUCCESS;
}
Commit	Line	Data
50827a59 IC	1	/*
	2	* (C) Copyright 2007-2011
	3	* Allwinner Technology Co., Ltd. <www.allwinnertech.com>
	4	* Tom Cubie <tangliang@allwinnertech.com>
	5	*
	6	* a simple tool to generate bootable image for sunxi platform.
	7	*
	8	* SPDX-License-Identifier: GPL-2.0+
	9	*/
	10	#include <fcntl.h>
	11	#include <stdio.h>
	12	#include <unistd.h>
	13	#include <stdlib.h>
	14	#include <string.h>
	15	#include <errno.h>
	16	#include <sys/types.h>
	17	#include <sys/stat.h>
e954eb80	18	#include "asm/arch/spl.h"
50827a59	19
50827a59 IC	20	#define STAMP_VALUE 0x5F0A6C39
	21
	22	/* check sum functon from sun4i boot code */
	23	int gen_check_sum(struct boot_file_head *head_p)
	24	{
	25	uint32_t length;
	26	uint32_t *buf;
	27	uint32_t loop;
	28	uint32_t i;
	29	uint32_t sum;
	30
c924e2a8	31	length = le32_to_cpu(head_p->length);
50827a59 IC	32	if ((length & 0x3) != 0) /* must 4-byte-aligned */
	33	return -1;
	34	buf = (uint32_t *)head_p;
c924e2a8	35	head_p->check_sum = cpu_to_le32(STAMP_VALUE); /* fill stamp */
50827a59 IC	36	loop = length >> 2;
	37
	38	/* calculate the sum */
	39	for (i = 0, sum = 0; i < loop; i++)
c924e2a8	40	sum += le32_to_cpu(buf[i]);
50827a59 IC	41
50827a59 IC	42	/* write back check sum */
c924e2a8	43	head_p->check_sum = cpu_to_le32(sum);
50827a59 IC	44
	45	return 0;
	46	}
	47
	48	#define ALIGN(x, a) __ALIGN_MASK((x), (typeof(x))(a)-1)
	49	#define __ALIGN_MASK(x, mask) (((x)+(mask))&~(mask))
	50
	51	#define SUN4I_SRAM_SIZE 0x7600 /* 0x7748+ is used by BROM */
	52	#define SRAM_LOAD_MAX_SIZE (SUN4I_SRAM_SIZE - sizeof(struct boot_file_head))
1f6f61fe DK	53
	54	/*
	55	* BROM (at least on A10 and A20) requires NAND-images to be explicitly aligned
	56	* to a multiple of 8K, and rejects the image otherwise. MMC-images are fine
	57	* with 512B blocks. To cater for both, align to the largest of the two.
	58	*/
	59	#define BLOCK_SIZE 0x2000
50827a59 IC	60
	61	struct boot_img {
	62	struct boot_file_head header;
	63	char code[SRAM_LOAD_MAX_SIZE];
	64	char pad[BLOCK_SIZE];
	65	};
	66
	67	int main(int argc, char *argv[])
	68	{
	69	int fd_in, fd_out;
	70	struct boot_img img;
4ba73a5a	71	unsigned file_size;
50827a59 IC	72	int count;
	73
	74	if (argc < 2) {
	75	printf("\tThis program makes an input bin file to sun4i " \
	76	"bootable image.\n" \
	77	"\tUsage: %s input_file out_putfile\n", argv[0]);
	78	return EXIT_FAILURE;
	79	}
	80
	81	fd_in = open(argv[1], O_RDONLY);
	82	if (fd_in < 0) {
	83	perror("Open input file");
	84	return EXIT_FAILURE;
	85	}
	86
bfb05d01	87	memset(&img, 0, sizeof(img));
50827a59 IC	88
	89	/* get input file size */
	90	file_size = lseek(fd_in, 0, SEEK_END);
	91
	92	if (file_size > SRAM_LOAD_MAX_SIZE) {
	93	fprintf(stderr, "ERROR: File too large!\n");
	94	return EXIT_FAILURE;
50827a59 IC	95	}
	96
	97	fd_out = open(argv[2], O_WRONLY \| O_CREAT, 0666);
	98	if (fd_out < 0) {
	99	perror("Open output file");
	100	return EXIT_FAILURE;
	101	}
	102
	103	/* read file to buffer to calculate checksum */
	104	lseek(fd_in, 0, SEEK_SET);
4ba73a5a HG	105	count = read(fd_in, img.code, file_size);
4ba73a5a HG	106	if (count != file_size) {
50827a59 IC	107	perror("Reading input image");
	108	return EXIT_FAILURE;
	109	}
	110
	111	/* fill the header */
	112	img.header.b_instruction = /* b instruction */
	113	0xEA000000 \| /* jump to the first instr after the header */
	114	((sizeof(struct boot_file_head) / sizeof(int) - 2)
	115	& 0x00FFFFFF);
	116	memcpy(img.header.magic, BOOT0_MAGIC, 8); /* no '0' termination */
	117	img.header.length =
4ba73a5a	118	ALIGN(file_size + sizeof(struct boot_file_head), BLOCK_SIZE);
c924e2a8 SS	119	img.header.b_instruction = cpu_to_le32(img.header.b_instruction);
c924e2a8 SS	120	img.header.length = cpu_to_le32(img.header.length);
a1884381 BN	121
	122	memcpy(img.header.spl_signature, SPL_SIGNATURE, 3); /* "sunxi" marker */
	123	img.header.spl_signature[3] = SPL_HEADER_VERSION;
	124
50827a59 IC	125	gen_check_sum(&img.header);
50827a59 IC	126
c924e2a8 SS	127	count = write(fd_out, &img, le32_to_cpu(img.header.length));
c924e2a8 SS	128	if (count != le32_to_cpu(img.header.length)) {
50827a59 IC	129	perror("Writing output");
	130	return EXIT_FAILURE;
	131	}
	132
	133	close(fd_in);
	134	close(fd_out);
	135
	136	return EXIT_SUCCESS;
	137	}