cmd/dtimg.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * (C) Copyright 2018 Linaro Ltd.
   4  * Sam Protsenko <semen.protsenko@linaro.org>
   5  */
   6
   7 #include <image-android-dt.h>
   8 #include <common.h>
   9
  10 enum cmd_dtimg_info {
  11         CMD_DTIMG_START = 0,
  12         CMD_DTIMG_SIZE,
  13 };
  14
  15 static int do_dtimg_dump(cmd_tbl_t *cmdtp, int flag, int argc,
  16                          char * const argv[])
  17 {
  18         char *endp;
  19         ulong hdr_addr;
  20
  21         if (argc != 2)
  22                 return CMD_RET_USAGE;
  23
  24         hdr_addr = simple_strtoul(argv[1], &endp, 16);
  25         if (*endp != '\0') {
  26                 printf("Error: Wrong image address\n");
  27                 return CMD_RET_FAILURE;
  28         }
  29
  30         if (!android_dt_check_header(hdr_addr)) {
  31                 printf("Error: DT image header is incorrect\n");
  32                 return CMD_RET_FAILURE;
  33         }
  34
  35         android_dt_print_contents(hdr_addr);
  36
  37         return CMD_RET_SUCCESS;
  38 }
  39
  40 static int dtimg_get_fdt(int argc, char * const argv[], enum cmd_dtimg_info cmd)
  41 {
  42         ulong hdr_addr;
  43         u32 index;
  44         char *endp;
  45         ulong fdt_addr;
  46         u32 fdt_size;
  47         char buf[65];
  48
  49         if (argc != 4)
  50                 return CMD_RET_USAGE;
  51
  52         hdr_addr = simple_strtoul(argv[1], &endp, 16);
  53         if (*endp != '\0') {
  54                 printf("Error: Wrong image address\n");
  55                 return CMD_RET_FAILURE;
  56         }
  57
  58         if (!android_dt_check_header(hdr_addr)) {
  59                 printf("Error: DT image header is incorrect\n");
  60                 return CMD_RET_FAILURE;
  61         }
  62
  63         index = simple_strtoul(argv[2], &endp, 0);
  64         if (*endp != '\0') {
  65                 printf("Error: Wrong index\n");
  66                 return CMD_RET_FAILURE;
  67         }
  68
  69         if (!android_dt_get_fdt_by_index(hdr_addr, index, &fdt_addr, &fdt_size))
  70                 return CMD_RET_FAILURE;
  71
  72         switch (cmd) {
  73         case CMD_DTIMG_START:
  74                 snprintf(buf, sizeof(buf), "%lx", fdt_addr);
  75                 break;
  76         case CMD_DTIMG_SIZE:
  77                 snprintf(buf, sizeof(buf), "%x", fdt_size);
  78                 break;
  79         default:
  80                 printf("Error: Unknown cmd_dtimg_info value: %d\n", cmd);
  81                 return CMD_RET_FAILURE;
  82         }
  83
  84         env_set(argv[3], buf);
  85
  86         return CMD_RET_SUCCESS;
  87 }
  88
  89 static int do_dtimg_start(cmd_tbl_t *cmdtp, int flag, int argc,
  90                           char * const argv[])
  91 {
  92         return dtimg_get_fdt(argc, argv, CMD_DTIMG_START);
  93 }
  94
  95 static int do_dtimg_size(cmd_tbl_t *cmdtp, int flag, int argc,
  96                          char * const argv[])
  97 {
  98         return dtimg_get_fdt(argc, argv, CMD_DTIMG_SIZE);
  99 }
 100
 101 static cmd_tbl_t cmd_dtimg_sub[] = {
 102         U_BOOT_CMD_MKENT(dump, 2, 0, do_dtimg_dump, "", ""),
 103         U_BOOT_CMD_MKENT(start, 4, 0, do_dtimg_start, "", ""),
 104         U_BOOT_CMD_MKENT(size, 4, 0, do_dtimg_size, "", ""),
 105 };
 106
 107 static int do_dtimg(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 108 {
 109         cmd_tbl_t *cp;
 110
 111         cp = find_cmd_tbl(argv[1], cmd_dtimg_sub, ARRAY_SIZE(cmd_dtimg_sub));
 112
 113         /* Strip off leading 'dtimg' command argument */
 114         argc--;
 115         argv++;
 116
 117         if (!cp || argc > cp->maxargs)
 118                 return CMD_RET_USAGE;
 119         if (flag == CMD_FLAG_REPEAT && !cmd_is_repeatable(cp))
 120                 return CMD_RET_SUCCESS;
 121
 122         return cp->cmd(cmdtp, flag, argc, argv);
 123 }
 124
 125 U_BOOT_CMD(
 126         dtimg, CONFIG_SYS_MAXARGS, 0, do_dtimg,
 127         "manipulate dtb/dtbo Android image",
 128         "dump <addr>\n"
 129         "    - parse specified image and print its structure info\n"
 130         "      <addr>: image address in RAM, in hex\n"
 131         "dtimg start <addr> <index> <varname>\n"
 132         "    - get address (hex) of FDT in the image, by index\n"
 133         "      <addr>: image address in RAM, in hex\n"
 134         "      <index>: index of desired FDT in the image\n"
 135         "      <varname>: name of variable where to store address of FDT\n"
 136         "dtimg size <addr> <index> <varname>\n"
 137         "    - get size (hex, bytes) of FDT in the image, by index\n"
 138         "      <addr>: image address in RAM, in hex\n"
 139         "      <index>: index of desired FDT in the image\n"
 140         "      <varname>: name of variable where to store size of FDT"
 141 );