blob: 53f33764fbe8e0088c436fd8fd3e0652bc034c5c [file] [log] [blame]
Sam Protsenkod03e76a2018-08-16 23:34:13 +03001// SPDX-License-Identifier: GPL-2.0+
2/*
3 * (C) Copyright 2018 Linaro Ltd.
4 * Sam Protsenko <semen.protsenko@linaro.org>
Eugeniu Rosca4f731c72019-12-24 17:51:08 +01005 * Eugeniu Rosca <rosca.eugeniu@gmail.com>
Sam Protsenkod03e76a2018-08-16 23:34:13 +03006 */
7
Simon Glass09140112020-05-10 11:40:03 -06008#include <command.h>
Simon Glass9fb625c2019-08-01 09:46:51 -06009#include <env.h>
Tom Rini03de3052024-05-20 13:35:03 -060010#include <vsprintf.h>
Sam Protsenkod03e76a2018-08-16 23:34:13 +030011#include <image-android-dt.h>
Sam Protsenkod03e76a2018-08-16 23:34:13 +030012
Eugeniu Rosca4f731c72019-12-24 17:51:08 +010013#define OPT_INDEX "--index"
Sam Protsenkod03e76a2018-08-16 23:34:13 +030014
Eugeniu Rosca4f731c72019-12-24 17:51:08 +010015/*
16 * Current/working DTB/DTBO Android image address.
17 * Similar to 'working_fdt' variable in 'fdt' command.
18 */
19static ulong working_img;
20
Simon Glass09140112020-05-10 11:40:03 -060021static int do_adtimg_addr(struct cmd_tbl *cmdtp, int flag, int argc,
22 char *const argv[])
Sam Protsenkod03e76a2018-08-16 23:34:13 +030023{
24 char *endp;
25 ulong hdr_addr;
26
27 if (argc != 2)
28 return CMD_RET_USAGE;
29
Simon Glass7e5f4602021-07-24 09:03:29 -060030 hdr_addr = hextoul(argv[1], &endp);
Sam Protsenkod03e76a2018-08-16 23:34:13 +030031 if (*endp != '\0') {
Eugeniu Rosca4f731c72019-12-24 17:51:08 +010032 printf("Error: Wrong image address '%s'\n", argv[1]);
Sam Protsenkod03e76a2018-08-16 23:34:13 +030033 return CMD_RET_FAILURE;
34 }
35
Eugeniu Rosca4f731c72019-12-24 17:51:08 +010036 /*
37 * Allow users to set an address prior to copying the DTB/DTBO
38 * image to that same address, i.e. skip header verification.
39 */
40
41 working_img = hdr_addr;
42 return CMD_RET_SUCCESS;
43}
44
45static int adtimg_check_working_img(void)
46{
47 if (!working_img) {
48 printf("Error: Please, call 'adtimg addr <addr>'. Aborting!\n");
Sam Protsenkod03e76a2018-08-16 23:34:13 +030049 return CMD_RET_FAILURE;
50 }
51
Eugeniu Rosca4f731c72019-12-24 17:51:08 +010052 if (!android_dt_check_header(working_img)) {
53 printf("Error: Invalid image header at 0x%lx\n", working_img);
54 return CMD_RET_FAILURE;
55 }
Sam Protsenkod03e76a2018-08-16 23:34:13 +030056
57 return CMD_RET_SUCCESS;
58}
59
Simon Glass09140112020-05-10 11:40:03 -060060static int do_adtimg_dump(struct cmd_tbl *cmdtp, int flag, int argc,
61 char *const argv[])
Sam Protsenkod03e76a2018-08-16 23:34:13 +030062{
Eugeniu Rosca4f731c72019-12-24 17:51:08 +010063 if (argc != 1)
64 return CMD_RET_USAGE;
65
66 if (adtimg_check_working_img() != CMD_RET_SUCCESS)
67 return CMD_RET_FAILURE;
68
69 android_dt_print_contents(working_img);
70
71 return CMD_RET_SUCCESS;
72}
73
74static int adtimg_getopt_u32(char * const opt, char * const name, u32 *optval)
75{
76 char *endp, *str;
77 u32 val;
78
79 if (!opt || !name || !optval)
80 return CMD_RET_FAILURE;
81
82 str = strchr(opt, '=');
83 if (!str) {
84 printf("Error: Option '%s' not followed by '='\n", name);
85 return CMD_RET_FAILURE;
86 }
87
88 if (*++str == '\0') {
89 printf("Error: Option '%s=' not followed by value\n", name);
90 return CMD_RET_FAILURE;
91 }
92
93 val = simple_strtoul(str, &endp, 0);
94 if (*endp != '\0') {
95 printf("Error: Wrong integer value '%s=%s'\n", name, str);
96 return CMD_RET_FAILURE;
97 }
98
99 *optval = val;
100 return CMD_RET_SUCCESS;
101}
102
Simon Glass09140112020-05-10 11:40:03 -0600103static int adtimg_getopt_index(int argc, char *const argv[], u32 *index,
Eugeniu Rosca4f731c72019-12-24 17:51:08 +0100104 char **avar, char **svar)
105{
106 int ret;
107
108 if (!argv || !avar || !svar)
109 return CMD_RET_FAILURE;
110
111 if (argc > 3) {
112 printf("Error: Unexpected argument '%s'\n", argv[3]);
113 return CMD_RET_FAILURE;
114 }
115
116 ret = adtimg_getopt_u32(argv[0], OPT_INDEX, index);
117 if (ret != CMD_RET_SUCCESS)
118 return ret;
119
120 if (argc > 1)
121 *avar = argv[1];
122 if (argc > 2)
123 *svar = argv[2];
124
125 return CMD_RET_SUCCESS;
126}
127
Simon Glass09140112020-05-10 11:40:03 -0600128static int adtimg_get_dt_by_index(int argc, char *const argv[])
Eugeniu Rosca4f731c72019-12-24 17:51:08 +0100129{
130 ulong addr;
131 u32 index, size;
132 int ret;
133 char *avar = NULL, *svar = NULL;
134
135 ret = adtimg_getopt_index(argc, argv, &index, &avar, &svar);
136 if (ret != CMD_RET_SUCCESS)
137 return ret;
138
139 if (!android_dt_get_fdt_by_index(working_img, index, &addr, &size))
140 return CMD_RET_FAILURE;
141
142 if (avar && svar) {
143 ret = env_set_hex(avar, addr);
144 if (ret) {
145 printf("Error: Can't set '%s' to 0x%lx\n", avar, addr);
146 return CMD_RET_FAILURE;
147 }
148 ret = env_set_hex(svar, size);
149 if (ret) {
150 printf("Error: Can't set '%s' to 0x%x\n", svar, size);
151 return CMD_RET_FAILURE;
152 }
153 } else if (avar) {
154 ret = env_set_hex(avar, addr);
155 if (ret) {
156 printf("Error: Can't set '%s' to 0x%lx\n", avar, addr);
157 return CMD_RET_FAILURE;
158 }
159 printf("0x%x (%d)\n", size, size);
160 } else {
161 printf("0x%lx, 0x%x (%d)\n", addr, size, size);
162 }
163
164 return CMD_RET_SUCCESS;
165}
166
Simon Glass09140112020-05-10 11:40:03 -0600167static int adtimg_get_dt(int argc, char *const argv[])
Eugeniu Rosca4f731c72019-12-24 17:51:08 +0100168{
169 if (argc < 2) {
170 printf("Error: No options passed to '%s'\n", argv[0]);
171 return CMD_RET_FAILURE;
172 }
173
174 /* Strip off leading 'dt' command argument */
175 argc--;
176 argv++;
177
178 if (!strncmp(argv[0], OPT_INDEX, sizeof(OPT_INDEX) - 1))
179 return adtimg_get_dt_by_index(argc, argv);
180
181 printf("Error: Option '%s' not supported\n", argv[0]);
182 return CMD_RET_FAILURE;
183}
184
Simon Glass09140112020-05-10 11:40:03 -0600185static int do_adtimg_get(struct cmd_tbl *cmdtp, int flag, int argc,
186 char *const argv[])
Eugeniu Rosca4f731c72019-12-24 17:51:08 +0100187{
188 if (argc < 2) {
189 printf("Error: No arguments passed to '%s'\n", argv[0]);
190 return CMD_RET_FAILURE;
191 }
192
193 if (adtimg_check_working_img() != CMD_RET_SUCCESS)
194 return CMD_RET_FAILURE;
195
196 /* Strip off leading 'get' command argument */
197 argc--;
198 argv++;
199
200 if (!strcmp(argv[0], "dt"))
201 return adtimg_get_dt(argc, argv);
202
203 printf("Error: Wrong argument '%s'\n", argv[0]);
204 return CMD_RET_FAILURE;
Sam Protsenkod03e76a2018-08-16 23:34:13 +0300205}
206
Simon Glass09140112020-05-10 11:40:03 -0600207static struct cmd_tbl cmd_adtimg_sub[] = {
Eugeniu Rosca4f731c72019-12-24 17:51:08 +0100208 U_BOOT_CMD_MKENT(addr, CONFIG_SYS_MAXARGS, 1, do_adtimg_addr, "", ""),
209 U_BOOT_CMD_MKENT(dump, CONFIG_SYS_MAXARGS, 1, do_adtimg_dump, "", ""),
210 U_BOOT_CMD_MKENT(get, CONFIG_SYS_MAXARGS, 1, do_adtimg_get, "", ""),
Sam Protsenkod03e76a2018-08-16 23:34:13 +0300211};
212
Simon Glass09140112020-05-10 11:40:03 -0600213static int do_adtimg(struct cmd_tbl *cmdtp, int flag, int argc,
214 char *const argv[])
Sam Protsenkod03e76a2018-08-16 23:34:13 +0300215{
Simon Glass09140112020-05-10 11:40:03 -0600216 struct cmd_tbl *cp;
Sam Protsenkod03e76a2018-08-16 23:34:13 +0300217
Eugeniu Rosca4c6edc22019-12-24 17:51:07 +0100218 cp = find_cmd_tbl(argv[1], cmd_adtimg_sub, ARRAY_SIZE(cmd_adtimg_sub));
Sam Protsenkod03e76a2018-08-16 23:34:13 +0300219
Eugeniu Rosca4c6edc22019-12-24 17:51:07 +0100220 /* Strip off leading 'adtimg' command argument */
Sam Protsenkod03e76a2018-08-16 23:34:13 +0300221 argc--;
222 argv++;
223
224 if (!cp || argc > cp->maxargs)
225 return CMD_RET_USAGE;
Boris Brezillon80a48dd2018-12-03 22:54:20 +0100226 if (flag == CMD_FLAG_REPEAT && !cmd_is_repeatable(cp))
Sam Protsenkod03e76a2018-08-16 23:34:13 +0300227 return CMD_RET_SUCCESS;
228
229 return cp->cmd(cmdtp, flag, argc, argv);
230}
231
232U_BOOT_CMD(
Eugeniu Rosca4c6edc22019-12-24 17:51:07 +0100233 adtimg, CONFIG_SYS_MAXARGS, 0, do_adtimg,
Sam Protsenkod03e76a2018-08-16 23:34:13 +0300234 "manipulate dtb/dtbo Android image",
Eugeniu Rosca4f731c72019-12-24 17:51:08 +0100235 "addr <addr> - Set image location to <addr>\n"
236 "adtimg dump - Print out image contents\n"
237 "adtimg get dt --index=<index> [avar [svar]] - Get DT address/size by index\n"
238 "\n"
239 "Legend:\n"
240 " - <addr>: DTB/DTBO image address (hex) in RAM\n"
241 " - <index>: index (hex/dec) of desired DT in the image\n"
242 " - <avar>: variable name to contain DT address (hex)\n"
243 " - <svar>: variable name to contain DT size (hex)"
Sam Protsenkod03e76a2018-08-16 23:34:13 +0300244);