tools/mksunxiboot.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   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.
   8  */
   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 <sunxi_image.h>
  16 #include <sys/types.h>
  17 #include <sys/stat.h>
  18 #include "imagetool.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 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))
  50
  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
  57
  58 struct boot_img {
  59         struct boot_file_head header;
  60         char code[SRAM_LOAD_MAX_SIZE];
  61         char pad[BLOCK_SIZE];
  62 };
  63
  64 int main(int argc, char *argv[])
  65 {
  66         int fd_in, fd_out;
  67         struct boot_img img;
  68         unsigned file_size;
  69         int count;
  70         char *tool_name = argv[0];
  71         char *default_dt = NULL;
  72
  73         /* 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");
 104                 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
 113         memset(&img, 0, sizeof(img));
 114
 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;
 121         }
 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);
 131         count = read(fd_in, img.code, file_size);
 132         if (count != file_size) {
 133                 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 =
 144                 ALIGN(file_size + sizeof(struct boot_file_head), BLOCK_SIZE);
 145         img.header.b_instruction = cpu_to_le32(img.header.b_instruction);
 146         img.header.length = cpu_to_le32(img.header.length);
 147
 148         memcpy(img.header.spl_signature, SPL_SIGNATURE, 3); /* "sunxi" marker */
 149         img.header.spl_signature[3] = SPL_HEADER_VERSION;
 150
 151         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
 163         gen_check_sum(&img.header);
 164
 165         count = write(fd_out, &img, le32_to_cpu(img.header.length));
 166         if (count != le32_to_cpu(img.header.length)) {
 167                 perror("Writing output");
 168                 return EXIT_FAILURE;
 169         }
 170
 171         close(fd_in);
 172         close(fd_out);
 173
 174         return EXIT_SUCCESS;
 175 }