dtimg.c 3.4 KB

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