tools/mksunxiboot.c

   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 }