blob: a18c9d98bc7d1b814b1e0c775bcfcbc1809ab512 [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Ian Campbell50827a52014-05-05 11:52:30 +01002/*
3 * (C) Copyright 2007-2011
4 * Allwinner Technology Co., Ltd. <www.allwinnertech.com>
5 * Tom Cubie <tangliang@allwinnertech.com>
6 *
7 * a simple tool to generate bootable image for sunxi platform.
Ian Campbell50827a52014-05-05 11:52:30 +01008 */
9#include <fcntl.h>
10#include <stdio.h>
11#include <unistd.h>
12#include <stdlib.h>
13#include <string.h>
14#include <errno.h>
15#include <sys/types.h>
16#include <sys/stat.h>
Kever Yangc738adb2020-03-30 11:56:17 +080017#include "imagetool.h"
Stefan Brünsfed329a2016-10-29 12:23:18 +020018#include "../arch/arm/include/asm/arch-sunxi/spl.h"
Ian Campbell50827a52014-05-05 11:52:30 +010019
Ian Campbell50827a52014-05-05 11:52:30 +010020#define STAMP_VALUE 0x5F0A6C39
21
22/* check sum functon from sun4i boot code */
23int 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
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +020031 length = le32_to_cpu(head_p->length);
Ian Campbell50827a52014-05-05 11:52:30 +010032 if ((length & 0x3) != 0) /* must 4-byte-aligned */
33 return -1;
34 buf = (uint32_t *)head_p;
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +020035 head_p->check_sum = cpu_to_le32(STAMP_VALUE); /* fill stamp */
Ian Campbell50827a52014-05-05 11:52:30 +010036 loop = length >> 2;
37
38 /* calculate the sum */
39 for (i = 0, sum = 0; i < loop; i++)
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +020040 sum += le32_to_cpu(buf[i]);
Ian Campbell50827a52014-05-05 11:52:30 +010041
42 /* write back check sum */
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +020043 head_p->check_sum = cpu_to_le32(sum);
Ian Campbell50827a52014-05-05 11:52:30 +010044
45 return 0;
46}
47
Andre Przywara45e2d062017-04-26 01:32:39 +010048#define SUNXI_SRAM_SIZE 0x8000 /* SoC with smaller size are limited before */
49#define SRAM_LOAD_MAX_SIZE (SUNXI_SRAM_SIZE - sizeof(struct boot_file_head))
Daniel Kochmański1f6f61f2015-05-26 17:00:39 +020050
51/*
52 * BROM (at least on A10 and A20) requires NAND-images to be explicitly aligned
53 * to a multiple of 8K, and rejects the image otherwise. MMC-images are fine
54 * with 512B blocks. To cater for both, align to the largest of the two.
55 */
56#define BLOCK_SIZE 0x2000
Ian Campbell50827a52014-05-05 11:52:30 +010057
58struct boot_img {
59 struct boot_file_head header;
60 char code[SRAM_LOAD_MAX_SIZE];
61 char pad[BLOCK_SIZE];
62};
63
64int main(int argc, char *argv[])
65{
66 int fd_in, fd_out;
67 struct boot_img img;
Hans de Goede4ba73a52014-06-09 11:36:53 +020068 unsigned file_size;
Ian Campbell50827a52014-05-05 11:52:30 +010069 int count;
Siarhei Siamashka7f0ef5a2017-04-26 01:32:49 +010070 char *tool_name = argv[0];
71 char *default_dt = NULL;
Ian Campbell50827a52014-05-05 11:52:30 +010072
Siarhei Siamashka7f0ef5a2017-04-26 01:32:49 +010073 /* a sanity check */
74 if ((sizeof(img.header) % 32) != 0) {
75 fprintf(stderr, "ERROR: the SPL header must be a multiple ");
76 fprintf(stderr, "of 32 bytes.\n");
77 return EXIT_FAILURE;
78 }
79
80 /* process optional command line switches */
81 while (argc >= 2 && argv[1][0] == '-') {
82 if (strcmp(argv[1], "--default-dt") == 0) {
83 if (argc >= 3) {
84 default_dt = argv[2];
85 argv += 2;
86 argc -= 2;
87 continue;
88 }
89 fprintf(stderr, "ERROR: no --default-dt arg\n");
90 return EXIT_FAILURE;
91 } else {
92 fprintf(stderr, "ERROR: bad option '%s'\n", argv[1]);
93 return EXIT_FAILURE;
94 }
95 }
96
97 if (argc < 3) {
98 printf("This program converts an input binary file to a sunxi bootable image.\n");
99 printf("\nUsage: %s [options] input_file output_file\n",
100 tool_name);
101 printf("Where [options] may be:\n");
102 printf(" --default-dt arg - 'arg' is the default device tree name\n");
103 printf(" (CONFIG_DEFAULT_DEVICE_TREE).\n");
Ian Campbell50827a52014-05-05 11:52:30 +0100104 return EXIT_FAILURE;
105 }
106
107 fd_in = open(argv[1], O_RDONLY);
108 if (fd_in < 0) {
109 perror("Open input file");
110 return EXIT_FAILURE;
111 }
112
Siarhei Siamashkabfb05d02015-09-03 02:36:39 +0300113 memset(&img, 0, sizeof(img));
Ian Campbell50827a52014-05-05 11:52:30 +0100114
115 /* get input file size */
116 file_size = lseek(fd_in, 0, SEEK_END);
117
118 if (file_size > SRAM_LOAD_MAX_SIZE) {
119 fprintf(stderr, "ERROR: File too large!\n");
120 return EXIT_FAILURE;
Ian Campbell50827a52014-05-05 11:52:30 +0100121 }
122
123 fd_out = open(argv[2], O_WRONLY | O_CREAT, 0666);
124 if (fd_out < 0) {
125 perror("Open output file");
126 return EXIT_FAILURE;
127 }
128
129 /* read file to buffer to calculate checksum */
130 lseek(fd_in, 0, SEEK_SET);
Hans de Goede4ba73a52014-06-09 11:36:53 +0200131 count = read(fd_in, img.code, file_size);
132 if (count != file_size) {
Ian Campbell50827a52014-05-05 11:52:30 +0100133 perror("Reading input image");
134 return EXIT_FAILURE;
135 }
136
137 /* fill the header */
138 img.header.b_instruction = /* b instruction */
139 0xEA000000 | /* jump to the first instr after the header */
140 ((sizeof(struct boot_file_head) / sizeof(int) - 2)
141 & 0x00FFFFFF);
142 memcpy(img.header.magic, BOOT0_MAGIC, 8); /* no '0' termination */
143 img.header.length =
Hans de Goede4ba73a52014-06-09 11:36:53 +0200144 ALIGN(file_size + sizeof(struct boot_file_head), BLOCK_SIZE);
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +0200145 img.header.b_instruction = cpu_to_le32(img.header.b_instruction);
146 img.header.length = cpu_to_le32(img.header.length);
Bernhard Nortmanna1884382015-09-17 18:52:51 +0200147
148 memcpy(img.header.spl_signature, SPL_SIGNATURE, 3); /* "sunxi" marker */
149 img.header.spl_signature[3] = SPL_HEADER_VERSION;
150
Siarhei Siamashka7f0ef5a2017-04-26 01:32:49 +0100151 if (default_dt) {
152 if (strlen(default_dt) + 1 <= sizeof(img.header.string_pool)) {
153 strcpy((char *)img.header.string_pool, default_dt);
154 img.header.dt_name_offset =
155 cpu_to_le32(offsetof(struct boot_file_head,
156 string_pool));
157 } else {
158 printf("WARNING: The SPL header is too small\n");
159 printf(" and has no space to store the dt name.\n");
160 }
161 }
162
Ian Campbell50827a52014-05-05 11:52:30 +0100163 gen_check_sum(&img.header);
164
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +0200165 count = write(fd_out, &img, le32_to_cpu(img.header.length));
166 if (count != le32_to_cpu(img.header.length)) {
Ian Campbell50827a52014-05-05 11:52:30 +0100167 perror("Writing output");
168 return EXIT_FAILURE;
169 }
170
171 close(fd_in);
172 close(fd_out);
173
174 return EXIT_SUCCESS;
175}