u-boot/cmd/dtimg.c
Boris Brezillon 80a48dd47e common: command: Rework the 'cmd is repeatable' logic
The repeatable property is currently attached to the main command and
sub-commands have no way to change the repeatable value (the
->repeatable field in sub-command entries is ignored).

Replace the ->repeatable field by an extended ->cmd() hook (called
->cmd_rep()) which takes a new int pointer to store the repeatable cap
of the command being executed.

With this trick, we can let sub-commands decide whether they are
repeatable or not.

We also patch mmc and dtimg who are testing the ->repeatable field
directly (they now use cmd_is_repeatable() instead), and fix the help
entry manually since it doesn't use the U_BOOT_CMD() macro.

Signed-off-by: Boris Brezillon <boris.brezillon@bootlin.com>
Reviewed-by: Tom Rini <trini@konsulko.com>
2019-01-15 15:28:54 -05:00

142 lines
3.4 KiB
C

// SPDX-License-Identifier: GPL-2.0+
/*
* (C) Copyright 2018 Linaro Ltd.
* Sam Protsenko <semen.protsenko@linaro.org>
*/
#include <image-android-dt.h>
#include <common.h>
enum cmd_dtimg_info {
CMD_DTIMG_START = 0,
CMD_DTIMG_SIZE,
};
static int do_dtimg_dump(cmd_tbl_t *cmdtp, int flag, int argc,
char * const argv[])
{
char *endp;
ulong hdr_addr;
if (argc != 2)
return CMD_RET_USAGE;
hdr_addr = simple_strtoul(argv[1], &endp, 16);
if (*endp != '\0') {
printf("Error: Wrong image address\n");
return CMD_RET_FAILURE;
}
if (!android_dt_check_header(hdr_addr)) {
printf("Error: DT image header is incorrect\n");
return CMD_RET_FAILURE;
}
android_dt_print_contents(hdr_addr);
return CMD_RET_SUCCESS;
}
static int dtimg_get_fdt(int argc, char * const argv[], enum cmd_dtimg_info cmd)
{
ulong hdr_addr;
u32 index;
char *endp;
ulong fdt_addr;
u32 fdt_size;
char buf[65];
if (argc != 4)
return CMD_RET_USAGE;
hdr_addr = simple_strtoul(argv[1], &endp, 16);
if (*endp != '\0') {
printf("Error: Wrong image address\n");
return CMD_RET_FAILURE;
}
if (!android_dt_check_header(hdr_addr)) {
printf("Error: DT image header is incorrect\n");
return CMD_RET_FAILURE;
}
index = simple_strtoul(argv[2], &endp, 0);
if (*endp != '\0') {
printf("Error: Wrong index\n");
return CMD_RET_FAILURE;
}
if (!android_dt_get_fdt_by_index(hdr_addr, index, &fdt_addr, &fdt_size))
return CMD_RET_FAILURE;
switch (cmd) {
case CMD_DTIMG_START:
snprintf(buf, sizeof(buf), "%lx", fdt_addr);
break;
case CMD_DTIMG_SIZE:
snprintf(buf, sizeof(buf), "%x", fdt_size);
break;
default:
printf("Error: Unknown cmd_dtimg_info value: %d\n", cmd);
return CMD_RET_FAILURE;
}
env_set(argv[3], buf);
return CMD_RET_SUCCESS;
}
static int do_dtimg_start(cmd_tbl_t *cmdtp, int flag, int argc,
char * const argv[])
{
return dtimg_get_fdt(argc, argv, CMD_DTIMG_START);
}
static int do_dtimg_size(cmd_tbl_t *cmdtp, int flag, int argc,
char * const argv[])
{
return dtimg_get_fdt(argc, argv, CMD_DTIMG_SIZE);
}
static cmd_tbl_t cmd_dtimg_sub[] = {
U_BOOT_CMD_MKENT(dump, 2, 0, do_dtimg_dump, "", ""),
U_BOOT_CMD_MKENT(start, 4, 0, do_dtimg_start, "", ""),
U_BOOT_CMD_MKENT(size, 4, 0, do_dtimg_size, "", ""),
};
static int do_dtimg(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
cmd_tbl_t *cp;
cp = find_cmd_tbl(argv[1], cmd_dtimg_sub, ARRAY_SIZE(cmd_dtimg_sub));
/* Strip off leading 'dtimg' command argument */
argc--;
argv++;
if (!cp || argc > cp->maxargs)
return CMD_RET_USAGE;
if (flag == CMD_FLAG_REPEAT && !cmd_is_repeatable(cp))
return CMD_RET_SUCCESS;
return cp->cmd(cmdtp, flag, argc, argv);
}
U_BOOT_CMD(
dtimg, CONFIG_SYS_MAXARGS, 0, do_dtimg,
"manipulate dtb/dtbo Android image",
"dump <addr>\n"
" - parse specified image and print its structure info\n"
" <addr>: image address in RAM, in hex\n"
"dtimg start <addr> <index> <varname>\n"
" - get address (hex) of FDT in the image, by index\n"
" <addr>: image address in RAM, in hex\n"
" <index>: index of desired FDT in the image\n"
" <varname>: name of variable where to store address of FDT\n"
"dtimg size <addr> <index> <varname>\n"
" - get size (hex, bytes) of FDT in the image, by index\n"
" <addr>: image address in RAM, in hex\n"
" <index>: index of desired FDT in the image\n"
" <varname>: name of variable where to store size of FDT"
);