xref: /openbmc/u-boot/cmd/dtimg.c (revision 308b1307)
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 );
142