dtimg.c 3.4 KB

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