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> 18 #include "../arch/arm/include/asm/arch-sunxi/spl.h" 19 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 31 length = le32_to_cpu(head_p->length); 32 if ((length & 0x3) != 0) /* must 4-byte-aligned */ 33 return -1; 34 buf = (uint32_t *)head_p; 35 head_p->check_sum = cpu_to_le32(STAMP_VALUE); /* fill stamp */ 36 loop = length >> 2; 37 38 /* calculate the sum */ 39 for (i = 0, sum = 0; i < loop; i++) 40 sum += le32_to_cpu(buf[i]); 41 42 /* write back check sum */ 43 head_p->check_sum = cpu_to_le32(sum); 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 SUNXI_SRAM_SIZE 0x8000 /* SoC with smaller size are limited before */ 52 #define SRAM_LOAD_MAX_SIZE (SUNXI_SRAM_SIZE - sizeof(struct boot_file_head)) 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 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; 71 unsigned file_size; 72 int count; 73 char *tool_name = argv[0]; 74 char *default_dt = NULL; 75 76 /* a sanity check */ 77 if ((sizeof(img.header) % 32) != 0) { 78 fprintf(stderr, "ERROR: the SPL header must be a multiple "); 79 fprintf(stderr, "of 32 bytes.\n"); 80 return EXIT_FAILURE; 81 } 82 83 /* process optional command line switches */ 84 while (argc >= 2 && argv[1][0] == '-') { 85 if (strcmp(argv[1], "--default-dt") == 0) { 86 if (argc >= 3) { 87 default_dt = argv[2]; 88 argv += 2; 89 argc -= 2; 90 continue; 91 } 92 fprintf(stderr, "ERROR: no --default-dt arg\n"); 93 return EXIT_FAILURE; 94 } else { 95 fprintf(stderr, "ERROR: bad option '%s'\n", argv[1]); 96 return EXIT_FAILURE; 97 } 98 } 99 100 if (argc < 3) { 101 printf("This program converts an input binary file to a sunxi bootable image.\n"); 102 printf("\nUsage: %s [options] input_file output_file\n", 103 tool_name); 104 printf("Where [options] may be:\n"); 105 printf(" --default-dt arg - 'arg' is the default device tree name\n"); 106 printf(" (CONFIG_DEFAULT_DEVICE_TREE).\n"); 107 return EXIT_FAILURE; 108 } 109 110 fd_in = open(argv[1], O_RDONLY); 111 if (fd_in < 0) { 112 perror("Open input file"); 113 return EXIT_FAILURE; 114 } 115 116 memset(&img, 0, sizeof(img)); 117 118 /* get input file size */ 119 file_size = lseek(fd_in, 0, SEEK_END); 120 121 if (file_size > SRAM_LOAD_MAX_SIZE) { 122 fprintf(stderr, "ERROR: File too large!\n"); 123 return EXIT_FAILURE; 124 } 125 126 fd_out = open(argv[2], O_WRONLY | O_CREAT, 0666); 127 if (fd_out < 0) { 128 perror("Open output file"); 129 return EXIT_FAILURE; 130 } 131 132 /* read file to buffer to calculate checksum */ 133 lseek(fd_in, 0, SEEK_SET); 134 count = read(fd_in, img.code, file_size); 135 if (count != file_size) { 136 perror("Reading input image"); 137 return EXIT_FAILURE; 138 } 139 140 /* fill the header */ 141 img.header.b_instruction = /* b instruction */ 142 0xEA000000 | /* jump to the first instr after the header */ 143 ((sizeof(struct boot_file_head) / sizeof(int) - 2) 144 & 0x00FFFFFF); 145 memcpy(img.header.magic, BOOT0_MAGIC, 8); /* no '0' termination */ 146 img.header.length = 147 ALIGN(file_size + sizeof(struct boot_file_head), BLOCK_SIZE); 148 img.header.b_instruction = cpu_to_le32(img.header.b_instruction); 149 img.header.length = cpu_to_le32(img.header.length); 150 151 memcpy(img.header.spl_signature, SPL_SIGNATURE, 3); /* "sunxi" marker */ 152 img.header.spl_signature[3] = SPL_HEADER_VERSION; 153 154 if (default_dt) { 155 if (strlen(default_dt) + 1 <= sizeof(img.header.string_pool)) { 156 strcpy((char *)img.header.string_pool, default_dt); 157 img.header.dt_name_offset = 158 cpu_to_le32(offsetof(struct boot_file_head, 159 string_pool)); 160 } else { 161 printf("WARNING: The SPL header is too small\n"); 162 printf(" and has no space to store the dt name.\n"); 163 } 164 } 165 166 gen_check_sum(&img.header); 167 168 count = write(fd_out, &img, le32_to_cpu(img.header.length)); 169 if (count != le32_to_cpu(img.header.length)) { 170 perror("Writing output"); 171 return EXIT_FAILURE; 172 } 173 174 close(fd_in); 175 close(fd_out); 176 177 return EXIT_SUCCESS; 178 } 179