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 19 /* boot head definition from sun4i boot code */ 20 struct 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 */ 38 int 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 46 length = le32_to_cpu(head_p->length); 47 if ((length & 0x3) != 0) /* must 4-byte-aligned */ 48 return -1; 49 buf = (uint32_t *)head_p; 50 head_p->check_sum = cpu_to_le32(STAMP_VALUE); /* fill stamp */ 51 loop = length >> 2; 52 53 /* calculate the sum */ 54 for (i = 0, sum = 0; i < loop; i++) 55 sum += le32_to_cpu(buf[i]); 56 57 /* write back check sum */ 58 head_p->check_sum = cpu_to_le32(sum); 59 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 69 /* 70 * BROM (at least on A10 and A20) requires NAND-images to be explicitly aligned 71 * to a multiple of 8K, and rejects the image otherwise. MMC-images are fine 72 * with 512B blocks. To cater for both, align to the largest of the two. 73 */ 74 #define BLOCK_SIZE 0x2000 75 76 struct boot_img { 77 struct boot_file_head header; 78 char code[SRAM_LOAD_MAX_SIZE]; 79 char pad[BLOCK_SIZE]; 80 }; 81 82 int main(int argc, char *argv[]) 83 { 84 int fd_in, fd_out; 85 struct boot_img img; 86 unsigned file_size; 87 int count; 88 89 if (argc < 2) { 90 printf("\tThis program makes an input bin file to sun4i " \ 91 "bootable image.\n" \ 92 "\tUsage: %s input_file out_putfile\n", argv[0]); 93 return EXIT_FAILURE; 94 } 95 96 fd_in = open(argv[1], O_RDONLY); 97 if (fd_in < 0) { 98 perror("Open input file"); 99 return EXIT_FAILURE; 100 } 101 102 memset(img.pad, 0, BLOCK_SIZE); 103 104 /* get input file size */ 105 file_size = lseek(fd_in, 0, SEEK_END); 106 107 if (file_size > SRAM_LOAD_MAX_SIZE) { 108 fprintf(stderr, "ERROR: File too large!\n"); 109 return EXIT_FAILURE; 110 } 111 112 fd_out = open(argv[2], O_WRONLY | O_CREAT, 0666); 113 if (fd_out < 0) { 114 perror("Open output file"); 115 return EXIT_FAILURE; 116 } 117 118 /* read file to buffer to calculate checksum */ 119 lseek(fd_in, 0, SEEK_SET); 120 count = read(fd_in, img.code, file_size); 121 if (count != file_size) { 122 perror("Reading input image"); 123 return EXIT_FAILURE; 124 } 125 126 /* fill the header */ 127 img.header.b_instruction = /* b instruction */ 128 0xEA000000 | /* jump to the first instr after the header */ 129 ((sizeof(struct boot_file_head) / sizeof(int) - 2) 130 & 0x00FFFFFF); 131 memcpy(img.header.magic, BOOT0_MAGIC, 8); /* no '0' termination */ 132 img.header.length = 133 ALIGN(file_size + sizeof(struct boot_file_head), BLOCK_SIZE); 134 img.header.b_instruction = cpu_to_le32(img.header.b_instruction); 135 img.header.length = cpu_to_le32(img.header.length); 136 gen_check_sum(&img.header); 137 138 count = write(fd_out, &img, le32_to_cpu(img.header.length)); 139 if (count != le32_to_cpu(img.header.length)) { 140 perror("Writing output"); 141 return EXIT_FAILURE; 142 } 143 144 close(fd_in); 145 close(fd_out); 146 147 return EXIT_SUCCESS; 148 } 149