123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260 |
- // SPDX-License-Identifier: GPL-2.0+
- /*
- * (C) Copyright 2020
- * Sam Protsenko <joe.skb7@gmail.com>
- */
- #include <android_image.h>
- #include <common.h>
- #include <command.h>
- #include <image.h>
- #include <mapmem.h>
- #define abootimg_addr() \
- (_abootimg_addr == -1 ? image_load_addr : _abootimg_addr)
- /* Please use abootimg_addr() macro to obtain the boot image address */
- static ulong _abootimg_addr = -1;
- static int abootimg_get_ver(int argc, char *const argv[])
- {
- const struct andr_img_hdr *hdr;
- int res = CMD_RET_SUCCESS;
- if (argc > 1)
- return CMD_RET_USAGE;
- hdr = map_sysmem(abootimg_addr(), sizeof(*hdr));
- if (android_image_check_header(hdr)) {
- printf("Error: Boot Image header is incorrect\n");
- res = CMD_RET_FAILURE;
- goto exit;
- }
- if (argc == 0)
- printf("%u\n", hdr->header_version);
- else
- env_set_ulong(argv[0], hdr->header_version);
- exit:
- unmap_sysmem(hdr);
- return res;
- }
- static int abootimg_get_recovery_dtbo(int argc, char *const argv[])
- {
- ulong addr;
- u32 size;
- if (argc > 2)
- return CMD_RET_USAGE;
- if (!android_image_get_dtbo(abootimg_addr(), &addr, &size))
- return CMD_RET_FAILURE;
- if (argc == 0) {
- printf("%lx\n", addr);
- } else {
- env_set_hex(argv[0], addr);
- if (argc == 2)
- env_set_hex(argv[1], size);
- }
- return CMD_RET_SUCCESS;
- }
- static int abootimg_get_dtb_load_addr(int argc, char *const argv[])
- {
- const struct andr_img_hdr *hdr;
- int res = CMD_RET_SUCCESS;
- if (argc > 1)
- return CMD_RET_USAGE;
- hdr = map_sysmem(abootimg_addr(), sizeof(*hdr));
- if (android_image_check_header(hdr)) {
- printf("Error: Boot Image header is incorrect\n");
- res = CMD_RET_FAILURE;
- goto exit;
- }
- if (hdr->header_version < 2) {
- printf("Error: header_version must be >= 2 for this\n");
- res = CMD_RET_FAILURE;
- goto exit;
- }
- if (argc == 0)
- printf("%lx\n", (ulong)hdr->dtb_addr);
- else
- env_set_hex(argv[0], (ulong)hdr->dtb_addr);
- exit:
- unmap_sysmem(hdr);
- return res;
- }
- static int abootimg_get_dtb_by_index(int argc, char *const argv[])
- {
- const char *index_str;
- u32 num;
- char *endp;
- ulong addr;
- u32 size;
- if (argc < 1 || argc > 3)
- return CMD_RET_USAGE;
- index_str = argv[0] + strlen("--index=");
- if (index_str[0] == '\0') {
- printf("Error: Wrong index num\n");
- return CMD_RET_FAILURE;
- }
- num = simple_strtoul(index_str, &endp, 0);
- if (*endp != '\0') {
- printf("Error: Wrong index num\n");
- return CMD_RET_FAILURE;
- }
- if (!android_image_get_dtb_by_index(abootimg_addr(), num,
- &addr, &size)) {
- return CMD_RET_FAILURE;
- }
- if (argc == 1) {
- printf("%lx\n", addr);
- } else {
- if (env_set_hex(argv[1], addr)) {
- printf("Error: Can't set [addr_var]\n");
- return CMD_RET_FAILURE;
- }
- if (argc == 3) {
- if (env_set_hex(argv[2], size)) {
- printf("Error: Can't set [size_var]\n");
- return CMD_RET_FAILURE;
- }
- }
- }
- return CMD_RET_SUCCESS;
- }
- static int abootimg_get_dtb(int argc, char *const argv[])
- {
- if (argc < 1)
- return CMD_RET_USAGE;
- if (strstr(argv[0], "--index="))
- return abootimg_get_dtb_by_index(argc, argv);
- return CMD_RET_USAGE;
- }
- static int do_abootimg_addr(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- char *endp;
- ulong img_addr;
- if (argc != 2)
- return CMD_RET_USAGE;
- img_addr = hextoul(argv[1], &endp);
- if (*endp != '\0') {
- printf("Error: Wrong image address\n");
- return CMD_RET_FAILURE;
- }
- _abootimg_addr = img_addr;
- return CMD_RET_SUCCESS;
- }
- static int do_abootimg_get(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- const char *param;
- if (argc < 2)
- return CMD_RET_USAGE;
- param = argv[1];
- argc -= 2;
- argv += 2;
- if (!strcmp(param, "ver"))
- return abootimg_get_ver(argc, argv);
- else if (!strcmp(param, "recovery_dtbo"))
- return abootimg_get_recovery_dtbo(argc, argv);
- else if (!strcmp(param, "dtb_load_addr"))
- return abootimg_get_dtb_load_addr(argc, argv);
- else if (!strcmp(param, "dtb"))
- return abootimg_get_dtb(argc, argv);
- return CMD_RET_USAGE;
- }
- static int do_abootimg_dump(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- if (argc != 2)
- return CMD_RET_USAGE;
- if (!strcmp(argv[1], "dtb")) {
- if (android_image_print_dtb_contents(abootimg_addr()))
- return CMD_RET_FAILURE;
- } else {
- return CMD_RET_USAGE;
- }
- return CMD_RET_SUCCESS;
- }
- static struct cmd_tbl cmd_abootimg_sub[] = {
- U_BOOT_CMD_MKENT(addr, 2, 1, do_abootimg_addr, "", ""),
- U_BOOT_CMD_MKENT(dump, 2, 1, do_abootimg_dump, "", ""),
- U_BOOT_CMD_MKENT(get, 5, 1, do_abootimg_get, "", ""),
- };
- static int do_abootimg(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- struct cmd_tbl *cp;
- cp = find_cmd_tbl(argv[1], cmd_abootimg_sub,
- ARRAY_SIZE(cmd_abootimg_sub));
- /* Strip off leading 'abootimg' command argument */
- argc--;
- argv++;
- if (!cp || argc > cp->maxargs)
- return CMD_RET_USAGE;
- if (flag == CMD_FLAG_REPEAT && !cmd_is_repeatable(cp))
- return CMD_RET_SUCCESS;
- return cp->cmd(cmdtp, flag, argc, argv);
- }
- U_BOOT_CMD(
- abootimg, CONFIG_SYS_MAXARGS, 0, do_abootimg,
- "manipulate Android Boot Image",
- "addr <addr>\n"
- " - set the address in RAM where boot image is located\n"
- " ($loadaddr is used by default)\n"
- "abootimg dump dtb\n"
- " - print info for all DT blobs in DTB area\n"
- "abootimg get ver [varname]\n"
- " - get header version\n"
- "abootimg get recovery_dtbo [addr_var [size_var]]\n"
- " - get address and size (hex) of recovery DTBO area in the image\n"
- " [addr_var]: variable name to contain DTBO area address\n"
- " [size_var]: variable name to contain DTBO area size\n"
- "abootimg get dtb_load_addr [varname]\n"
- " - get load address (hex) of DTB, from image header\n"
- "abootimg get dtb --index=<num> [addr_var [size_var]]\n"
- " - get address and size (hex) of DT blob in the image by index\n"
- " <num>: index number of desired DT blob in DTB area\n"
- " [addr_var]: variable name to contain DT blob address\n"
- " [size_var]: variable name to contain DT blob size"
- );
|