123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813 |
- // SPDX-License-Identifier: GPL-2.0+
- /*
- * Generate MediaTek BootROM header for SPL/U-Boot images
- *
- * Copyright (C) 2018 MediaTek Inc.
- * Author: Weijie Gao <weijie.gao@mediatek.com>
- */
- #include <image.h>
- #include <u-boot/sha256.h>
- #include "imagetool.h"
- #include "mtk_image.h"
- /* NAND header for SPI-NAND with 2KB page + 64B spare */
- static const union nand_boot_header snand_hdr_2k_64_data = {
- .data = {
- 0x42, 0x4F, 0x4F, 0x54, 0x4C, 0x4F, 0x41, 0x44,
- 0x45, 0x52, 0x21, 0x00, 0x56, 0x30, 0x30, 0x36,
- 0x4E, 0x46, 0x49, 0x49, 0x4E, 0x46, 0x4F, 0x00,
- 0x00, 0x00, 0x00, 0x08, 0x03, 0x00, 0x40, 0x00,
- 0x40, 0x00, 0x00, 0x08, 0x10, 0x00, 0x16, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x7B, 0xC4, 0x17, 0x9D,
- 0xCA, 0x42, 0x90, 0xD0, 0x98, 0xD0, 0xE0, 0xF7,
- 0xDB, 0xCD, 0x16, 0xF6, 0x03, 0x73, 0xD2, 0xB8,
- 0x93, 0xB2, 0x56, 0x5A, 0x84, 0x6E, 0x00, 0x00
- }
- };
- /* NAND header for SPI-NAND with 2KB page + 120B/128B spare */
- static const union nand_boot_header snand_hdr_2k_128_data = {
- .data = {
- 0x42, 0x4F, 0x4F, 0x54, 0x4C, 0x4F, 0x41, 0x44,
- 0x45, 0x52, 0x21, 0x00, 0x56, 0x30, 0x30, 0x36,
- 0x4E, 0x46, 0x49, 0x49, 0x4E, 0x46, 0x4F, 0x00,
- 0x00, 0x00, 0x00, 0x08, 0x05, 0x00, 0x70, 0x00,
- 0x40, 0x00, 0x00, 0x08, 0x10, 0x00, 0x16, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x90, 0x28, 0xED, 0x13,
- 0x7F, 0x12, 0x22, 0xCD, 0x3D, 0x06, 0xF1, 0xB3,
- 0x6F, 0x2E, 0xD9, 0xA0, 0x9D, 0x7A, 0xBD, 0xD7,
- 0xB3, 0x28, 0x3C, 0x13, 0xDB, 0x4E, 0x00, 0x00
- }
- };
- /* NAND header for SPI-NAND with 4KB page + 256B spare */
- static const union nand_boot_header snand_hdr_4k_256_data = {
- .data = {
- 0x42, 0x4F, 0x4F, 0x54, 0x4C, 0x4F, 0x41, 0x44,
- 0x45, 0x52, 0x21, 0x00, 0x56, 0x30, 0x30, 0x36,
- 0x4E, 0x46, 0x49, 0x49, 0x4E, 0x46, 0x4F, 0x00,
- 0x00, 0x00, 0x00, 0x10, 0x05, 0x00, 0xE0, 0x00,
- 0x40, 0x00, 0x00, 0x08, 0x10, 0x00, 0x16, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x47, 0xED, 0x0E, 0xC3,
- 0x83, 0xBF, 0x41, 0xD2, 0x85, 0x21, 0x97, 0x57,
- 0xC4, 0x2E, 0x6B, 0x7A, 0x40, 0xE0, 0xCF, 0x8F,
- 0x37, 0xBD, 0x17, 0xB6, 0xC7, 0xFE, 0x00, 0x00
- }
- };
- /* NAND header for Parallel NAND 1Gb with 2KB page + 64B spare */
- static const union nand_boot_header nand_hdr_1gb_2k_64_data = {
- .data = {
- 0x42, 0x4F, 0x4F, 0x54, 0x4C, 0x4F, 0x41, 0x44,
- 0x45, 0x52, 0x21, 0x00, 0x56, 0x30, 0x30, 0x36,
- 0x4E, 0x46, 0x49, 0x49, 0x4E, 0x46, 0x4F, 0x00,
- 0x00, 0x00, 0x00, 0x08, 0x05, 0x00, 0x40, 0x00,
- 0x40, 0x00, 0x00, 0x04, 0x0B, 0x00, 0x11, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x12, 0x28, 0x1C, 0x12,
- 0x8F, 0xFD, 0xF8, 0x32, 0x6F, 0x6D, 0xCF, 0x6C,
- 0xDA, 0x21, 0x70, 0x8C, 0xDA, 0x0A, 0x22, 0x82,
- 0xAA, 0x59, 0xFA, 0x7C, 0x42, 0x2D, 0x00, 0x00
- }
- };
- /* NAND header for Parallel NAND 2Gb with 2KB page + 64B spare */
- static const union nand_boot_header nand_hdr_2gb_2k_64_data = {
- .data = {
- 0x42, 0x4F, 0x4F, 0x54, 0x4C, 0x4F, 0x41, 0x44,
- 0x45, 0x52, 0x21, 0x00, 0x56, 0x30, 0x30, 0x36,
- 0x4E, 0x46, 0x49, 0x49, 0x4E, 0x46, 0x4F, 0x00,
- 0x00, 0x00, 0x00, 0x08, 0x05, 0x00, 0x40, 0x00,
- 0x40, 0x00, 0x00, 0x08, 0x0B, 0x00, 0x11, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x20, 0x9C, 0x3D, 0x2D,
- 0x7B, 0x68, 0x63, 0x52, 0x2E, 0x04, 0x63, 0xF1,
- 0x35, 0x4E, 0x44, 0x3E, 0xF8, 0xAC, 0x9B, 0x95,
- 0xAB, 0xFE, 0xE4, 0xE1, 0xD5, 0xF9, 0x00, 0x00
- }
- };
- /* NAND header for Parallel NAND 4Gb with 2KB page + 64B spare */
- static const union nand_boot_header nand_hdr_4gb_2k_64_data = {
- .data = {
- 0x42, 0x4F, 0x4F, 0x54, 0x4C, 0x4F, 0x41, 0x44,
- 0x45, 0x52, 0x21, 0x00, 0x56, 0x30, 0x30, 0x36,
- 0x4E, 0x46, 0x49, 0x49, 0x4E, 0x46, 0x4F, 0x00,
- 0x00, 0x00, 0x00, 0x08, 0x05, 0x00, 0x40, 0x00,
- 0x40, 0x00, 0x00, 0x10, 0x0B, 0x00, 0x11, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0xE3, 0x0F, 0x86, 0x32,
- 0x68, 0x05, 0xD9, 0xC8, 0x13, 0xDF, 0xC5, 0x0B,
- 0x35, 0x3A, 0x68, 0xA5, 0x3C, 0x0C, 0x73, 0x87,
- 0x63, 0xB0, 0xBE, 0xCC, 0x84, 0x47, 0x00, 0x00
- }
- };
- /* NAND header for Parallel NAND 2Gb with 2KB page + 128B spare */
- static const union nand_boot_header nand_hdr_2gb_2k_128_data = {
- .data = {
- 0x42, 0x4F, 0x4F, 0x54, 0x4C, 0x4F, 0x41, 0x44,
- 0x45, 0x52, 0x21, 0x00, 0x56, 0x30, 0x30, 0x36,
- 0x4E, 0x46, 0x49, 0x49, 0x4E, 0x46, 0x4F, 0x00,
- 0x00, 0x00, 0x00, 0x08, 0x05, 0x00, 0x70, 0x00,
- 0x40, 0x00, 0x00, 0x08, 0x0B, 0x00, 0x11, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x01, 0xA5, 0xE9, 0x5A,
- 0xDF, 0x58, 0x62, 0x41, 0xD6, 0x26, 0x77, 0xBC,
- 0x76, 0x1F, 0x27, 0x4E, 0x4F, 0x6C, 0xC3, 0xF0,
- 0x36, 0xDE, 0xD9, 0xB3, 0xFF, 0x93, 0x00, 0x00
- }
- };
- /* NAND header for Parallel NAND 4Gb with 2KB page + 128B spare */
- static const union nand_boot_header nand_hdr_4gb_2k_128_data = {
- .data = {
- 0x42, 0x4F, 0x4F, 0x54, 0x4C, 0x4F, 0x41, 0x44,
- 0x45, 0x52, 0x21, 0x00, 0x56, 0x30, 0x30, 0x36,
- 0x4E, 0x46, 0x49, 0x49, 0x4E, 0x46, 0x4F, 0x00,
- 0x00, 0x00, 0x00, 0x08, 0x05, 0x00, 0x70, 0x00,
- 0x40, 0x00, 0x00, 0x10, 0x0B, 0x00, 0x11, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0xC2, 0x36, 0x52, 0x45,
- 0xCC, 0x35, 0xD8, 0xDB, 0xEB, 0xFD, 0xD1, 0x46,
- 0x76, 0x6B, 0x0B, 0xD5, 0x8B, 0xCC, 0x2B, 0xE2,
- 0xFE, 0x90, 0x83, 0x9E, 0xAE, 0x2D, 0x00, 0x00
- }
- };
- static const struct nand_header_type {
- const char *name;
- const union nand_boot_header *data;
- } nand_headers[] = {
- {
- .name = "2k+64",
- .data = &snand_hdr_2k_64_data
- }, {
- .name = "2k+120",
- .data = &snand_hdr_2k_128_data
- }, {
- .name = "2k+128",
- .data = &snand_hdr_2k_128_data
- }, {
- .name = "4k+256",
- .data = &snand_hdr_4k_256_data
- }, {
- .name = "1g:2k+64",
- .data = &nand_hdr_1gb_2k_64_data
- }, {
- .name = "2g:2k+64",
- .data = &nand_hdr_2gb_2k_64_data
- }, {
- .name = "4g:2k+64",
- .data = &nand_hdr_4gb_2k_64_data
- }, {
- .name = "2g:2k+128",
- .data = &nand_hdr_2gb_2k_128_data
- }, {
- .name = "4g:2k+128",
- .data = &nand_hdr_4gb_2k_128_data
- }
- };
- static const struct brom_img_type {
- const char *name;
- enum brlyt_img_type type;
- } brom_images[] = {
- {
- .name = "nand",
- .type = BRLYT_TYPE_NAND
- }, {
- .name = "emmc",
- .type = BRLYT_TYPE_EMMC
- }, {
- .name = "nor",
- .type = BRLYT_TYPE_NOR
- }, {
- .name = "sdmmc",
- .type = BRLYT_TYPE_SDMMC
- }, {
- .name = "snand",
- .type = BRLYT_TYPE_SNAND
- }
- };
- /* Indicates whether we're generating or verifying */
- static bool img_gen;
- static uint32_t img_size;
- /* Image type selected by user */
- static enum brlyt_img_type hdr_media;
- static uint32_t hdr_offset;
- static int use_lk_hdr;
- static bool is_arm64_image;
- /* LK image name */
- static char lk_name[32] = "U-Boot";
- /* NAND header selected by user */
- static const union nand_boot_header *hdr_nand;
- /* GFH header + 2 * 4KB pages of NAND */
- static char hdr_tmp[sizeof(struct gfh_header) + 0x2000];
- static int mtk_image_check_image_types(uint8_t type)
- {
- if (type == IH_TYPE_MTKIMAGE)
- return EXIT_SUCCESS;
- else
- return EXIT_FAILURE;
- }
- static int mtk_brom_parse_imagename(const char *imagename)
- {
- #define is_blank_char(c) \
- ((c) == '\t' || (c) == '\n' || (c) == '\r' || (c) == ' ')
- char *buf = strdup(imagename), *key, *val, *end, *next;
- int i;
- /* User passed arguments from image name */
- static const char *media = "";
- static const char *hdr_offs = "";
- static const char *nandinfo = "";
- static const char *lk = "";
- static const char *arm64_param = "";
- key = buf;
- while (key) {
- next = strchr(key, ';');
- if (next)
- *next = 0;
- val = strchr(key, '=');
- if (val) {
- *val++ = 0;
- /* Trim key */
- while (is_blank_char(*key))
- key++;
- end = key + strlen(key) - 1;
- while ((end >= key) && is_blank_char(*end))
- end--;
- end++;
- if (is_blank_char(*end))
- *end = 0;
- /* Trim value */
- while (is_blank_char(*val))
- val++;
- end = val + strlen(val) - 1;
- while ((end >= val) && is_blank_char(*end))
- end--;
- end++;
- if (is_blank_char(*end))
- *end = 0;
- /* record user passed arguments */
- if (!strcmp(key, "media"))
- media = val;
- if (!strcmp(key, "hdroffset"))
- hdr_offs = val;
- if (!strcmp(key, "nandinfo"))
- nandinfo = val;
- if (!strcmp(key, "lk"))
- lk = val;
- if (!strcmp(key, "lkname"))
- snprintf(lk_name, sizeof(lk_name), "%s", val);
- if (!strcmp(key, "arm64"))
- arm64_param = val;
- }
- if (next)
- key = next + 1;
- else
- break;
- }
- /* if user specified LK image header, skip following checks */
- if (lk && lk[0] == '1') {
- use_lk_hdr = 1;
- free(buf);
- return 0;
- }
- /* parse media type */
- for (i = 0; i < ARRAY_SIZE(brom_images); i++) {
- if (!strcmp(brom_images[i].name, media)) {
- hdr_media = brom_images[i].type;
- break;
- }
- }
- /* parse nand header type */
- for (i = 0; i < ARRAY_SIZE(nand_headers); i++) {
- if (!strcmp(nand_headers[i].name, nandinfo)) {
- hdr_nand = nand_headers[i].data;
- break;
- }
- }
- /* parse device header offset */
- if (hdr_offs && hdr_offs[0])
- hdr_offset = strtoul(hdr_offs, NULL, 0);
- if (arm64_param && arm64_param[0] == '1')
- is_arm64_image = true;
- free(buf);
- if (hdr_media == BRLYT_TYPE_INVALID) {
- fprintf(stderr, "Error: media type is invalid or missing.\n");
- fprintf(stderr, " Please specify -n \"media=<type>\"\n");
- return -EINVAL;
- }
- if ((hdr_media == BRLYT_TYPE_NAND || hdr_media == BRLYT_TYPE_SNAND) &&
- !hdr_nand) {
- fprintf(stderr, "Error: nand info is invalid or missing.\n");
- fprintf(stderr, " Please specify -n \"media=%s;"
- "nandinfo=<info>\"\n", media);
- return -EINVAL;
- }
- return 0;
- }
- static int mtk_image_check_params(struct image_tool_params *params)
- {
- if (!params->addr) {
- fprintf(stderr, "Error: Load Address must be set.\n");
- return -EINVAL;
- }
- if (!params->imagename) {
- fprintf(stderr, "Error: Image Name must be set.\n");
- return -EINVAL;
- }
- return mtk_brom_parse_imagename(params->imagename);
- }
- static int mtk_image_vrec_header(struct image_tool_params *params,
- struct image_type_params *tparams)
- {
- if (use_lk_hdr) {
- tparams->header_size = sizeof(union lk_hdr);
- tparams->hdr = &hdr_tmp;
- memset(&hdr_tmp, 0xff, tparams->header_size);
- return 0;
- }
- if (hdr_media == BRLYT_TYPE_NAND || hdr_media == BRLYT_TYPE_SNAND)
- tparams->header_size = 2 * le16_to_cpu(hdr_nand->pagesize);
- else
- tparams->header_size = sizeof(struct gen_device_header);
- tparams->header_size += sizeof(struct gfh_header);
- tparams->hdr = &hdr_tmp;
- memset(&hdr_tmp, 0xff, tparams->header_size);
- return SHA256_SUM_LEN;
- }
- static int mtk_image_verify_gen_header(const uint8_t *ptr, int print)
- {
- union gen_boot_header *gbh = (union gen_boot_header *)ptr;
- uint32_t gfh_offset, total_size, devh_size;
- struct brom_layout_header *bh;
- struct gfh_header *gfh;
- const char *bootmedia;
- if (!strcmp(gbh->name, SF_BOOT_NAME))
- bootmedia = "Serial NOR";
- else if (!strcmp(gbh->name, EMMC_BOOT_NAME))
- bootmedia = "eMMC";
- else if (!strcmp(gbh->name, SDMMC_BOOT_NAME))
- bootmedia = "SD/MMC";
- else
- return -1;
- if (print)
- printf("Boot Media: %s\n", bootmedia);
- if (le32_to_cpu(gbh->version) != 1 ||
- le32_to_cpu(gbh->size) != sizeof(union gen_boot_header))
- return -1;
- bh = (struct brom_layout_header *)(ptr + le32_to_cpu(gbh->size));
- if (strcmp(bh->name, BRLYT_NAME))
- return -1;
- if (le32_to_cpu(bh->magic) != BRLYT_MAGIC ||
- (le32_to_cpu(bh->type) != BRLYT_TYPE_NOR &&
- le32_to_cpu(bh->type) != BRLYT_TYPE_EMMC &&
- le32_to_cpu(bh->type) != BRLYT_TYPE_SDMMC))
- return -1;
- devh_size = sizeof(struct gen_device_header);
- if (img_gen) {
- gfh_offset = devh_size;
- } else {
- gfh_offset = le32_to_cpu(bh->header_size);
- if (gfh_offset + sizeof(struct gfh_header) > img_size) {
- /*
- * This may happen if the hdr_offset used to generate
- * this image is not zero.
- * Since device header size is not fixed, we can't
- * cover all possible cases.
- * Assuming the image is valid only if the real
- * device header size equals to devh_size.
- */
- total_size = le32_to_cpu(bh->total_size);
- if (total_size - gfh_offset > img_size - devh_size)
- return -1;
- gfh_offset = devh_size;
- }
- }
- gfh = (struct gfh_header *)(ptr + gfh_offset);
- if (strcmp(gfh->file_info.name, GFH_FILE_INFO_NAME))
- return -1;
- if (le32_to_cpu(gfh->file_info.flash_type) != GFH_FLASH_TYPE_GEN)
- return -1;
- if (print)
- printf("Load Address: %08x\n",
- le32_to_cpu(gfh->file_info.load_addr) +
- le32_to_cpu(gfh->file_info.jump_offset));
- if (print)
- printf("Architecture: %s\n", is_arm64_image ? "ARM64" : "ARM");
- return 0;
- }
- static int mtk_image_verify_nand_header(const uint8_t *ptr, int print)
- {
- union nand_boot_header *nh = (union nand_boot_header *)ptr;
- struct brom_layout_header *bh;
- struct gfh_header *gfh;
- const char *bootmedia;
- if (strncmp(nh->version, NAND_BOOT_VERSION, sizeof(nh->version)) ||
- strcmp(nh->id, NAND_BOOT_ID))
- return -1;
- bh = (struct brom_layout_header *)(ptr + le16_to_cpu(nh->pagesize));
- if (strcmp(bh->name, BRLYT_NAME))
- return -1;
- if (le32_to_cpu(bh->magic) != BRLYT_MAGIC) {
- return -1;
- } else {
- if (le32_to_cpu(bh->type) == BRLYT_TYPE_NAND)
- bootmedia = "Parallel NAND";
- else if (le32_to_cpu(bh->type) == BRLYT_TYPE_SNAND)
- bootmedia = "Serial NAND";
- else
- return -1;
- }
- if (print) {
- printf("Boot Media: %s\n", bootmedia);
- if (le32_to_cpu(bh->type) == BRLYT_TYPE_NAND) {
- uint64_t capacity =
- (uint64_t)le16_to_cpu(nh->numblocks) *
- (uint64_t)le16_to_cpu(nh->pages_of_block) *
- (uint64_t)le16_to_cpu(nh->pagesize) * 8;
- printf("Capacity: %dGb\n",
- (uint32_t)(capacity >> 30));
- }
- if (le16_to_cpu(nh->pagesize) >= 1024)
- printf("Page Size: %dKB\n",
- le16_to_cpu(nh->pagesize) >> 10);
- else
- printf("Page Size: %dB\n",
- le16_to_cpu(nh->pagesize));
- printf("Spare Size: %dB\n", le16_to_cpu(nh->oobsize));
- }
- gfh = (struct gfh_header *)(ptr + 2 * le16_to_cpu(nh->pagesize));
- if (strcmp(gfh->file_info.name, GFH_FILE_INFO_NAME))
- return -1;
- if (le32_to_cpu(gfh->file_info.flash_type) != GFH_FLASH_TYPE_NAND)
- return -1;
- if (print)
- printf("Load Address: %08x\n",
- le32_to_cpu(gfh->file_info.load_addr) +
- le32_to_cpu(gfh->file_info.jump_offset));
- if (print)
- printf("Architecture: %s\n", is_arm64_image ? "ARM64" : "ARM");
- return 0;
- }
- static int mtk_image_verify_header(unsigned char *ptr, int image_size,
- struct image_tool_params *params)
- {
- union lk_hdr *lk = (union lk_hdr *)ptr;
- /* nothing to verify for LK image header */
- if (le32_to_cpu(lk->magic) == LK_PART_MAGIC)
- return 0;
- img_size = image_size;
- if (!strcmp((char *)ptr, NAND_BOOT_NAME))
- return mtk_image_verify_nand_header(ptr, 0);
- else
- return mtk_image_verify_gen_header(ptr, 0);
- return -1;
- }
- static void mtk_image_print_header(const void *ptr)
- {
- union lk_hdr *lk = (union lk_hdr *)ptr;
- if (le32_to_cpu(lk->magic) == LK_PART_MAGIC) {
- printf("Image Type: MediaTek LK Image\n");
- printf("Load Address: %08x\n", le32_to_cpu(lk->loadaddr));
- return;
- }
- printf("Image Type: MediaTek BootROM Loadable Image\n");
- if (!strcmp((char *)ptr, NAND_BOOT_NAME))
- mtk_image_verify_nand_header(ptr, 1);
- else
- mtk_image_verify_gen_header(ptr, 1);
- }
- static void put_brom_layout_header(struct brom_layout_header *hdr, int type)
- {
- strncpy(hdr->name, BRLYT_NAME, sizeof(hdr->name));
- hdr->version = cpu_to_le32(1);
- hdr->magic = cpu_to_le32(BRLYT_MAGIC);
- hdr->type = cpu_to_le32(type);
- }
- static void put_ghf_common_header(struct gfh_common_header *gfh, int size,
- int type, int ver)
- {
- memcpy(gfh->magic, GFH_HEADER_MAGIC, sizeof(gfh->magic));
- gfh->version = ver;
- gfh->size = cpu_to_le16(size);
- gfh->type = cpu_to_le16(type);
- }
- static void put_ghf_header(struct gfh_header *gfh, int file_size,
- int dev_hdr_size, int load_addr, int flash_type)
- {
- uint32_t cfg_bits;
- memset(gfh, 0, sizeof(struct gfh_header));
- /* GFH_FILE_INFO header */
- put_ghf_common_header(&gfh->file_info.gfh, sizeof(gfh->file_info),
- GFH_TYPE_FILE_INFO, 1);
- strncpy(gfh->file_info.name, GFH_FILE_INFO_NAME,
- sizeof(gfh->file_info.name));
- gfh->file_info.unused = cpu_to_le32(1);
- gfh->file_info.file_type = cpu_to_le16(1);
- gfh->file_info.flash_type = flash_type;
- gfh->file_info.sig_type = GFH_SIG_TYPE_SHA256;
- gfh->file_info.load_addr = cpu_to_le32(load_addr - sizeof(*gfh));
- gfh->file_info.total_size = cpu_to_le32(file_size - dev_hdr_size);
- gfh->file_info.max_size = cpu_to_le32(file_size);
- gfh->file_info.hdr_size = sizeof(*gfh);
- gfh->file_info.sig_size = SHA256_SUM_LEN;
- gfh->file_info.jump_offset = sizeof(*gfh);
- gfh->file_info.processed = cpu_to_le32(1);
- /* GFH_BL_INFO header */
- put_ghf_common_header(&gfh->bl_info.gfh, sizeof(gfh->bl_info),
- GFH_TYPE_BL_INFO, 1);
- gfh->bl_info.attr = cpu_to_le32(1);
- /* GFH_BROM_CFG header */
- put_ghf_common_header(&gfh->brom_cfg.gfh, sizeof(gfh->brom_cfg),
- GFH_TYPE_BROM_CFG, 3);
- cfg_bits = GFH_BROM_CFG_USBDL_AUTO_DETECT_DIS |
- GFH_BROM_CFG_USBDL_BY_KCOL0_TIMEOUT_EN |
- GFH_BROM_CFG_USBDL_BY_FLAG_TIMEOUT_EN;
- gfh->brom_cfg.usbdl_by_kcol0_timeout_ms = cpu_to_le32(5000);
- if (is_arm64_image) {
- gfh->brom_cfg.jump_bl_arm64 = GFH_BROM_CFG_JUMP_BL_ARM64;
- cfg_bits |= GFH_BROM_CFG_JUMP_BL_ARM64_EN;
- }
- gfh->brom_cfg.cfg_bits = cpu_to_le32(cfg_bits);
- /* GFH_BL_SEC_KEY header */
- put_ghf_common_header(&gfh->bl_sec_key.gfh, sizeof(gfh->bl_sec_key),
- GFH_TYPE_BL_SEC_KEY, 1);
- /* GFH_ANTI_CLONE header */
- put_ghf_common_header(&gfh->anti_clone.gfh, sizeof(gfh->anti_clone),
- GFH_TYPE_ANTI_CLONE, 1);
- gfh->anti_clone.ac_offset = cpu_to_le32(0x10);
- gfh->anti_clone.ac_len = cpu_to_le32(0x80);
- /* GFH_BROM_SEC_CFG header */
- put_ghf_common_header(&gfh->brom_sec_cfg.gfh,
- sizeof(gfh->brom_sec_cfg),
- GFH_TYPE_BROM_SEC_CFG, 1);
- gfh->brom_sec_cfg.cfg_bits =
- cpu_to_le32(BROM_SEC_CFG_JTAG_EN | BROM_SEC_CFG_UART_EN);
- }
- static void put_hash(uint8_t *buff, int size)
- {
- sha256_context ctx;
- sha256_starts(&ctx);
- sha256_update(&ctx, buff, size);
- sha256_finish(&ctx, buff + size);
- }
- static void mtk_image_set_gen_header(void *ptr, off_t filesize,
- uint32_t loadaddr)
- {
- struct gen_device_header *hdr = (struct gen_device_header *)ptr;
- struct gfh_header *gfh;
- const char *bootname = NULL;
- if (hdr_media == BRLYT_TYPE_NOR)
- bootname = SF_BOOT_NAME;
- else if (hdr_media == BRLYT_TYPE_EMMC)
- bootname = EMMC_BOOT_NAME;
- else if (hdr_media == BRLYT_TYPE_SDMMC)
- bootname = SDMMC_BOOT_NAME;
- /* Generic device header */
- snprintf(hdr->boot.name, sizeof(hdr->boot.name), "%s", bootname);
- hdr->boot.version = cpu_to_le32(1);
- hdr->boot.size = cpu_to_le32(sizeof(hdr->boot));
- /* BRLYT header */
- put_brom_layout_header(&hdr->brlyt, hdr_media);
- hdr->brlyt.header_size = cpu_to_le32(hdr_offset + sizeof(*hdr));
- hdr->brlyt.total_size = cpu_to_le32(hdr_offset + filesize);
- hdr->brlyt.header_size_2 = hdr->brlyt.header_size;
- hdr->brlyt.total_size_2 = hdr->brlyt.total_size;
- /* GFH header */
- gfh = (struct gfh_header *)(ptr + sizeof(struct gen_device_header));
- put_ghf_header(gfh, filesize, sizeof(struct gen_device_header),
- loadaddr, GFH_FLASH_TYPE_GEN);
- /* Generate SHA256 hash */
- put_hash((uint8_t *)gfh,
- filesize - sizeof(struct gen_device_header) - SHA256_SUM_LEN);
- }
- static void mtk_image_set_nand_header(void *ptr, off_t filesize,
- uint32_t loadaddr)
- {
- union nand_boot_header *nh = (union nand_boot_header *)ptr;
- struct brom_layout_header *brlyt;
- struct gfh_header *gfh;
- uint32_t payload_pages;
- int i;
- /* NAND device header, repeat 4 times */
- for (i = 0; i < 4; i++)
- memcpy(nh + i, hdr_nand, sizeof(union nand_boot_header));
- /* BRLYT header */
- payload_pages = (filesize + le16_to_cpu(hdr_nand->pagesize) - 1) /
- le16_to_cpu(hdr_nand->pagesize);
- brlyt = (struct brom_layout_header *)
- (ptr + le16_to_cpu(hdr_nand->pagesize));
- put_brom_layout_header(brlyt, hdr_media);
- brlyt->header_size = cpu_to_le32(2);
- brlyt->total_size = cpu_to_le32(payload_pages);
- brlyt->header_size_2 = brlyt->header_size;
- brlyt->total_size_2 = brlyt->total_size;
- brlyt->unused = cpu_to_le32(1);
- /* GFH header */
- gfh = (struct gfh_header *)(ptr + 2 * le16_to_cpu(hdr_nand->pagesize));
- put_ghf_header(gfh, filesize, 2 * le16_to_cpu(hdr_nand->pagesize),
- loadaddr, GFH_FLASH_TYPE_NAND);
- /* Generate SHA256 hash */
- put_hash((uint8_t *)gfh,
- filesize - 2 * le16_to_cpu(hdr_nand->pagesize) - SHA256_SUM_LEN);
- }
- static void mtk_image_set_header(void *ptr, struct stat *sbuf, int ifd,
- struct image_tool_params *params)
- {
- union lk_hdr *lk = (union lk_hdr *)ptr;
- if (use_lk_hdr) {
- lk->magic = cpu_to_le32(LK_PART_MAGIC);
- lk->size = cpu_to_le32(sbuf->st_size - sizeof(union lk_hdr));
- lk->loadaddr = cpu_to_le32(params->addr);
- lk->mode = 0xffffffff; /* must be non-zero */
- memset(lk->name, 0, sizeof(lk->name));
- strncpy(lk->name, lk_name, sizeof(lk->name));
- return;
- }
- img_gen = true;
- img_size = sbuf->st_size;
- if (hdr_media == BRLYT_TYPE_NAND || hdr_media == BRLYT_TYPE_SNAND)
- mtk_image_set_nand_header(ptr, sbuf->st_size, params->addr);
- else
- mtk_image_set_gen_header(ptr, sbuf->st_size, params->addr);
- }
- U_BOOT_IMAGE_TYPE(
- mtk_image,
- "MediaTek BootROM Loadable Image support",
- 0,
- NULL,
- mtk_image_check_params,
- mtk_image_verify_header,
- mtk_image_print_header,
- mtk_image_set_header,
- NULL,
- mtk_image_check_image_types,
- NULL,
- mtk_image_vrec_header
- );
|