blob: 0035f6ea265d926837ed576418dc81fbed3970c9 [file] [log] [blame]
Ian Campbell50827a52014-05-05 11:52:30 +01001/*
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>
18
19/* boot head definition from sun4i boot code */
20struct boot_file_head {
21 uint32_t b_instruction; /* one intruction jumping to real code */
22 uint8_t magic[8]; /* ="eGON.BT0" or "eGON.BT1", not C-style str */
23 uint32_t check_sum; /* generated by PC */
24 uint32_t length; /* generated by PC */
25 /*
26 * We use a simplified header, only filling in what is needed
27 * by the boot ROM. To be compatible with Allwinner tools we
28 * would need to implement the proper fields here instead of
29 * padding.
30 */
31 uint8_t pad[12]; /* align to 32 bytes */
32};
33
34#define BOOT0_MAGIC "eGON.BT0"
35#define STAMP_VALUE 0x5F0A6C39
36
37/* check sum functon from sun4i boot code */
38int gen_check_sum(struct boot_file_head *head_p)
39{
40 uint32_t length;
41 uint32_t *buf;
42 uint32_t loop;
43 uint32_t i;
44 uint32_t sum;
45
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +020046 length = le32_to_cpu(head_p->length);
Ian Campbell50827a52014-05-05 11:52:30 +010047 if ((length & 0x3) != 0) /* must 4-byte-aligned */
48 return -1;
49 buf = (uint32_t *)head_p;
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +020050 head_p->check_sum = cpu_to_le32(STAMP_VALUE); /* fill stamp */
Ian Campbell50827a52014-05-05 11:52:30 +010051 loop = length >> 2;
52
53 /* calculate the sum */
54 for (i = 0, sum = 0; i < loop; i++)
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +020055 sum += le32_to_cpu(buf[i]);
Ian Campbell50827a52014-05-05 11:52:30 +010056
57 /* write back check sum */
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +020058 head_p->check_sum = cpu_to_le32(sum);
Ian Campbell50827a52014-05-05 11:52:30 +010059
60 return 0;
61}
62
63#define ALIGN(x, a) __ALIGN_MASK((x), (typeof(x))(a)-1)
64#define __ALIGN_MASK(x, mask) (((x)+(mask))&~(mask))
65
66#define SUN4I_SRAM_SIZE 0x7600 /* 0x7748+ is used by BROM */
67#define SRAM_LOAD_MAX_SIZE (SUN4I_SRAM_SIZE - sizeof(struct boot_file_head))
68#define BLOCK_SIZE 512
69
70struct boot_img {
71 struct boot_file_head header;
72 char code[SRAM_LOAD_MAX_SIZE];
73 char pad[BLOCK_SIZE];
74};
75
76int main(int argc, char *argv[])
77{
78 int fd_in, fd_out;
79 struct boot_img img;
Hans de Goede4ba73a52014-06-09 11:36:53 +020080 unsigned file_size;
Ian Campbell50827a52014-05-05 11:52:30 +010081 int count;
82
83 if (argc < 2) {
84 printf("\tThis program makes an input bin file to sun4i " \
85 "bootable image.\n" \
86 "\tUsage: %s input_file out_putfile\n", argv[0]);
87 return EXIT_FAILURE;
88 }
89
90 fd_in = open(argv[1], O_RDONLY);
91 if (fd_in < 0) {
92 perror("Open input file");
93 return EXIT_FAILURE;
94 }
95
96 memset(img.pad, 0, BLOCK_SIZE);
97
98 /* get input file size */
99 file_size = lseek(fd_in, 0, SEEK_END);
100
101 if (file_size > SRAM_LOAD_MAX_SIZE) {
102 fprintf(stderr, "ERROR: File too large!\n");
103 return EXIT_FAILURE;
Ian Campbell50827a52014-05-05 11:52:30 +0100104 }
105
106 fd_out = open(argv[2], O_WRONLY | O_CREAT, 0666);
107 if (fd_out < 0) {
108 perror("Open output file");
109 return EXIT_FAILURE;
110 }
111
112 /* read file to buffer to calculate checksum */
113 lseek(fd_in, 0, SEEK_SET);
Hans de Goede4ba73a52014-06-09 11:36:53 +0200114 count = read(fd_in, img.code, file_size);
115 if (count != file_size) {
Ian Campbell50827a52014-05-05 11:52:30 +0100116 perror("Reading input image");
117 return EXIT_FAILURE;
118 }
119
120 /* fill the header */
121 img.header.b_instruction = /* b instruction */
122 0xEA000000 | /* jump to the first instr after the header */
123 ((sizeof(struct boot_file_head) / sizeof(int) - 2)
124 & 0x00FFFFFF);
125 memcpy(img.header.magic, BOOT0_MAGIC, 8); /* no '0' termination */
126 img.header.length =
Hans de Goede4ba73a52014-06-09 11:36:53 +0200127 ALIGN(file_size + sizeof(struct boot_file_head), BLOCK_SIZE);
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +0200128 img.header.b_instruction = cpu_to_le32(img.header.b_instruction);
129 img.header.length = cpu_to_le32(img.header.length);
Ian Campbell50827a52014-05-05 11:52:30 +0100130 gen_check_sum(&img.header);
131
Siarhei Siamashkac924e2a2015-02-08 07:05:27 +0200132 count = write(fd_out, &img, le32_to_cpu(img.header.length));
133 if (count != le32_to_cpu(img.header.length)) {
Ian Campbell50827a52014-05-05 11:52:30 +0100134 perror("Writing output");
135 return EXIT_FAILURE;
136 }
137
138 close(fd_in);
139 close(fd_out);
140
141 return EXIT_SUCCESS;
142}