abootimg.c 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * (C) Copyright 2020
  4. * Sam Protsenko <joe.skb7@gmail.com>
  5. */
  6. #include <android_image.h>
  7. #include <common.h>
  8. #include <command.h>
  9. #include <image.h>
  10. #include <mapmem.h>
  11. #define abootimg_addr() \
  12. (_abootimg_addr == -1 ? image_load_addr : _abootimg_addr)
  13. /* Please use abootimg_addr() macro to obtain the boot image address */
  14. static ulong _abootimg_addr = -1;
  15. static ulong _avendor_bootimg_addr = -1;
  16. ulong get_abootimg_addr(void)
  17. {
  18. return (_abootimg_addr == -1 ? image_load_addr : _abootimg_addr);
  19. }
  20. ulong get_avendor_bootimg_addr(void)
  21. {
  22. return _avendor_bootimg_addr;
  23. }
  24. static int abootimg_get_ver(int argc, char *const argv[])
  25. {
  26. const struct andr_boot_img_hdr_v0 *hdr;
  27. int res = CMD_RET_SUCCESS;
  28. if (argc > 1)
  29. return CMD_RET_USAGE;
  30. hdr = map_sysmem(abootimg_addr(), sizeof(*hdr));
  31. if (!is_android_boot_image_header(hdr)) {
  32. printf("Error: Boot Image header is incorrect\n");
  33. res = CMD_RET_FAILURE;
  34. goto exit;
  35. }
  36. if (argc == 0)
  37. printf("%u\n", hdr->header_version);
  38. else
  39. env_set_ulong(argv[0], hdr->header_version);
  40. exit:
  41. unmap_sysmem(hdr);
  42. return res;
  43. }
  44. static int abootimg_get_recovery_dtbo(int argc, char *const argv[])
  45. {
  46. ulong addr;
  47. u32 size;
  48. if (argc > 2)
  49. return CMD_RET_USAGE;
  50. if (!android_image_get_dtbo(abootimg_addr(), &addr, &size))
  51. return CMD_RET_FAILURE;
  52. if (argc == 0) {
  53. printf("%lx\n", addr);
  54. } else {
  55. env_set_hex(argv[0], addr);
  56. if (argc == 2)
  57. env_set_hex(argv[1], size);
  58. }
  59. return CMD_RET_SUCCESS;
  60. }
  61. static int abootimg_get_dtb_load_addr(int argc, char *const argv[])
  62. {
  63. if (argc > 1)
  64. return CMD_RET_USAGE;
  65. struct andr_image_data img_data = {0};
  66. const struct andr_boot_img_hdr_v0 *hdr;
  67. const struct andr_vnd_boot_img_hdr *vhdr;
  68. hdr = map_sysmem(abootimg_addr(), sizeof(*hdr));
  69. if (get_avendor_bootimg_addr() != -1)
  70. vhdr = map_sysmem(get_avendor_bootimg_addr(), sizeof(*vhdr));
  71. if (!android_image_get_data(hdr, vhdr, &img_data)) {
  72. if (get_avendor_bootimg_addr() != -1)
  73. unmap_sysmem(vhdr);
  74. unmap_sysmem(hdr);
  75. return CMD_RET_FAILURE;
  76. }
  77. if (get_avendor_bootimg_addr() != -1)
  78. unmap_sysmem(vhdr);
  79. unmap_sysmem(hdr);
  80. if (img_data.header_version < 2) {
  81. printf("Error: header_version must be >= 2 for this\n");
  82. return CMD_RET_FAILURE;
  83. }
  84. if (!img_data.dtb_load_addr) {
  85. printf("Error: failed to read dtb_load_addr\n");
  86. return CMD_RET_FAILURE;
  87. }
  88. if (argc == 0)
  89. printf("%lx\n", (ulong)img_data.dtb_load_addr);
  90. else
  91. env_set_hex(argv[0], (ulong)img_data.dtb_load_addr);
  92. return CMD_RET_SUCCESS;
  93. }
  94. static int abootimg_get_dtb_by_index(int argc, char *const argv[])
  95. {
  96. const char *index_str;
  97. u32 num;
  98. char *endp;
  99. ulong addr;
  100. u32 size;
  101. if (argc < 1 || argc > 3)
  102. return CMD_RET_USAGE;
  103. index_str = argv[0] + strlen("--index=");
  104. if (index_str[0] == '\0') {
  105. printf("Error: Wrong index num\n");
  106. return CMD_RET_FAILURE;
  107. }
  108. num = simple_strtoul(index_str, &endp, 0);
  109. if (*endp != '\0') {
  110. printf("Error: Wrong index num\n");
  111. return CMD_RET_FAILURE;
  112. }
  113. if (!android_image_get_dtb_by_index(abootimg_addr(),
  114. get_avendor_bootimg_addr(), num,
  115. &addr, &size)) {
  116. return CMD_RET_FAILURE;
  117. }
  118. if (argc == 1) {
  119. printf("%lx\n", addr);
  120. } else {
  121. if (env_set_hex(argv[1], addr)) {
  122. printf("Error: Can't set [addr_var]\n");
  123. return CMD_RET_FAILURE;
  124. }
  125. if (argc == 3) {
  126. if (env_set_hex(argv[2], size)) {
  127. printf("Error: Can't set [size_var]\n");
  128. return CMD_RET_FAILURE;
  129. }
  130. }
  131. }
  132. return CMD_RET_SUCCESS;
  133. }
  134. static int abootimg_get_dtb(int argc, char *const argv[])
  135. {
  136. if (argc < 1)
  137. return CMD_RET_USAGE;
  138. if (strstr(argv[0], "--index="))
  139. return abootimg_get_dtb_by_index(argc, argv);
  140. return CMD_RET_USAGE;
  141. }
  142. static int do_abootimg_addr(struct cmd_tbl *cmdtp, int flag, int argc,
  143. char *const argv[])
  144. {
  145. char *endp;
  146. ulong img_addr;
  147. if (argc < 2 || argc > 3)
  148. return CMD_RET_USAGE;
  149. img_addr = hextoul(argv[1], &endp);
  150. if (*endp != '\0') {
  151. printf("Error: Wrong image address\n");
  152. return CMD_RET_FAILURE;
  153. }
  154. _abootimg_addr = img_addr;
  155. if (argc == 3) {
  156. img_addr = simple_strtoul(argv[2], &endp, 16);
  157. if (*endp != '\0') {
  158. printf("Error: Wrong vendor image address\n");
  159. return CMD_RET_FAILURE;
  160. }
  161. _avendor_bootimg_addr = img_addr;
  162. }
  163. return CMD_RET_SUCCESS;
  164. }
  165. static int do_abootimg_get(struct cmd_tbl *cmdtp, int flag, int argc,
  166. char *const argv[])
  167. {
  168. const char *param;
  169. if (argc < 2)
  170. return CMD_RET_USAGE;
  171. param = argv[1];
  172. argc -= 2;
  173. argv += 2;
  174. if (!strcmp(param, "ver"))
  175. return abootimg_get_ver(argc, argv);
  176. else if (!strcmp(param, "recovery_dtbo"))
  177. return abootimg_get_recovery_dtbo(argc, argv);
  178. else if (!strcmp(param, "dtb_load_addr"))
  179. return abootimg_get_dtb_load_addr(argc, argv);
  180. else if (!strcmp(param, "dtb"))
  181. return abootimg_get_dtb(argc, argv);
  182. return CMD_RET_USAGE;
  183. }
  184. static int do_abootimg_dump(struct cmd_tbl *cmdtp, int flag, int argc,
  185. char *const argv[])
  186. {
  187. if (argc != 2)
  188. return CMD_RET_USAGE;
  189. if (!strcmp(argv[1], "dtb")) {
  190. if (android_image_print_dtb_contents(abootimg_addr()))
  191. return CMD_RET_FAILURE;
  192. } else {
  193. return CMD_RET_USAGE;
  194. }
  195. return CMD_RET_SUCCESS;
  196. }
  197. static struct cmd_tbl cmd_abootimg_sub[] = {
  198. U_BOOT_CMD_MKENT(addr, 3, 1, do_abootimg_addr, "", ""),
  199. U_BOOT_CMD_MKENT(dump, 2, 1, do_abootimg_dump, "", ""),
  200. U_BOOT_CMD_MKENT(get, 5, 1, do_abootimg_get, "", ""),
  201. };
  202. static int do_abootimg(struct cmd_tbl *cmdtp, int flag, int argc,
  203. char *const argv[])
  204. {
  205. struct cmd_tbl *cp;
  206. cp = find_cmd_tbl(argv[1], cmd_abootimg_sub,
  207. ARRAY_SIZE(cmd_abootimg_sub));
  208. /* Strip off leading 'abootimg' command argument */
  209. argc--;
  210. argv++;
  211. if (!cp || argc > cp->maxargs)
  212. return CMD_RET_USAGE;
  213. if (flag == CMD_FLAG_REPEAT && !cmd_is_repeatable(cp))
  214. return CMD_RET_SUCCESS;
  215. return cp->cmd(cmdtp, flag, argc, argv);
  216. }
  217. U_BOOT_CMD(
  218. abootimg, CONFIG_SYS_MAXARGS, 0, do_abootimg,
  219. "manipulate Android Boot Image",
  220. "addr <boot_img_addr> [<vendor_boot_img_addr>]>\n"
  221. " - set the address in RAM where boot image is located\n"
  222. " ($loadaddr is used by default)\n"
  223. "abootimg dump dtb\n"
  224. " - print info for all DT blobs in DTB area\n"
  225. "abootimg get ver [varname]\n"
  226. " - get header version\n"
  227. "abootimg get recovery_dtbo [addr_var [size_var]]\n"
  228. " - get address and size (hex) of recovery DTBO area in the image\n"
  229. " [addr_var]: variable name to contain DTBO area address\n"
  230. " [size_var]: variable name to contain DTBO area size\n"
  231. "abootimg get dtb_load_addr [varname]\n"
  232. " - get load address (hex) of DTB, from image header\n"
  233. "abootimg get dtb --index=<num> [addr_var [size_var]]\n"
  234. " - get address and size (hex) of DT blob in the image by index\n"
  235. " <num>: index number of desired DT blob in DTB area\n"
  236. " [addr_var]: variable name to contain DT blob address\n"
  237. " [size_var]: variable name to contain DT blob size"
  238. );