123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340 |
- // SPDX-License-Identifier: GPL-2.0+
- /*
- * Copyright (C) 2019 Eugeniu Rosca <rosca.eugeniu@gmail.com>
- *
- * Command to read/modify/write Android BCB fields
- */
- #include <android_bootloader_message.h>
- #include <command.h>
- #include <common.h>
- #include <log.h>
- #include <part.h>
- enum bcb_cmd {
- BCB_CMD_LOAD,
- BCB_CMD_FIELD_SET,
- BCB_CMD_FIELD_CLEAR,
- BCB_CMD_FIELD_TEST,
- BCB_CMD_FIELD_DUMP,
- BCB_CMD_STORE,
- };
- static int bcb_dev = -1;
- static int bcb_part = -1;
- static struct bootloader_message bcb = { { 0 } };
- static int bcb_cmd_get(char *cmd)
- {
- if (!strcmp(cmd, "load"))
- return BCB_CMD_LOAD;
- if (!strcmp(cmd, "set"))
- return BCB_CMD_FIELD_SET;
- if (!strcmp(cmd, "clear"))
- return BCB_CMD_FIELD_CLEAR;
- if (!strcmp(cmd, "test"))
- return BCB_CMD_FIELD_TEST;
- if (!strcmp(cmd, "store"))
- return BCB_CMD_STORE;
- if (!strcmp(cmd, "dump"))
- return BCB_CMD_FIELD_DUMP;
- else
- return -1;
- }
- static int bcb_is_misused(int argc, char *const argv[])
- {
- int cmd = bcb_cmd_get(argv[0]);
- switch (cmd) {
- case BCB_CMD_LOAD:
- case BCB_CMD_FIELD_SET:
- if (argc != 3)
- goto err;
- break;
- case BCB_CMD_FIELD_TEST:
- if (argc != 4)
- goto err;
- break;
- case BCB_CMD_FIELD_CLEAR:
- if (argc != 1 && argc != 2)
- goto err;
- break;
- case BCB_CMD_STORE:
- if (argc != 1)
- goto err;
- break;
- case BCB_CMD_FIELD_DUMP:
- if (argc != 2)
- goto err;
- break;
- default:
- printf("Error: 'bcb %s' not supported\n", argv[0]);
- return -1;
- }
- if (cmd != BCB_CMD_LOAD && (bcb_dev < 0 || bcb_part < 0)) {
- printf("Error: Please, load BCB first!\n");
- return -1;
- }
- return 0;
- err:
- printf("Error: Bad usage of 'bcb %s'\n", argv[0]);
- return -1;
- }
- static int bcb_field_get(char *name, char **fieldp, int *sizep)
- {
- if (!strcmp(name, "command")) {
- *fieldp = bcb.command;
- *sizep = sizeof(bcb.command);
- } else if (!strcmp(name, "status")) {
- *fieldp = bcb.status;
- *sizep = sizeof(bcb.status);
- } else if (!strcmp(name, "recovery")) {
- *fieldp = bcb.recovery;
- *sizep = sizeof(bcb.recovery);
- } else if (!strcmp(name, "stage")) {
- *fieldp = bcb.stage;
- *sizep = sizeof(bcb.stage);
- } else if (!strcmp(name, "reserved")) {
- *fieldp = bcb.reserved;
- *sizep = sizeof(bcb.reserved);
- } else {
- printf("Error: Unknown bcb field '%s'\n", name);
- return -1;
- }
- return 0;
- }
- static int do_bcb_load(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- struct blk_desc *desc;
- struct disk_partition info;
- u64 cnt;
- char *endp;
- int part, ret;
- ret = blk_get_device_by_str("mmc", argv[1], &desc);
- if (ret < 0)
- goto err_read_fail;
- part = simple_strtoul(argv[2], &endp, 0);
- if (*endp == '\0') {
- ret = part_get_info(desc, part, &info);
- if (ret)
- goto err_read_fail;
- } else {
- part = part_get_info_by_name(desc, argv[2], &info);
- if (part < 0) {
- ret = part;
- goto err_read_fail;
- }
- }
- cnt = DIV_ROUND_UP(sizeof(struct bootloader_message), info.blksz);
- if (cnt > info.size)
- goto err_too_small;
- if (blk_dread(desc, info.start, cnt, &bcb) != cnt) {
- ret = -EIO;
- goto err_read_fail;
- }
- bcb_dev = desc->devnum;
- bcb_part = part;
- debug("%s: Loaded from mmc %d:%d\n", __func__, bcb_dev, bcb_part);
- return CMD_RET_SUCCESS;
- err_read_fail:
- printf("Error: mmc %s:%s read failed (%d)\n", argv[1], argv[2], ret);
- goto err;
- err_too_small:
- printf("Error: mmc %s:%s too small!", argv[1], argv[2]);
- goto err;
- err:
- bcb_dev = -1;
- bcb_part = -1;
- return CMD_RET_FAILURE;
- }
- static int do_bcb_set(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- int size, len;
- char *field, *str, *found;
- if (bcb_field_get(argv[1], &field, &size))
- return CMD_RET_FAILURE;
- len = strlen(argv[2]);
- if (len >= size) {
- printf("Error: sizeof('%s') = %d >= %d = sizeof(bcb.%s)\n",
- argv[2], len, size, argv[1]);
- return CMD_RET_FAILURE;
- }
- str = argv[2];
- field[0] = '\0';
- while ((found = strsep(&str, ":"))) {
- if (field[0] != '\0')
- strcat(field, "\n");
- strcat(field, found);
- }
- return CMD_RET_SUCCESS;
- }
- static int do_bcb_clear(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- int size;
- char *field;
- if (argc == 1) {
- memset(&bcb, 0, sizeof(bcb));
- return CMD_RET_SUCCESS;
- }
- if (bcb_field_get(argv[1], &field, &size))
- return CMD_RET_FAILURE;
- memset(field, 0, size);
- return CMD_RET_SUCCESS;
- }
- static int do_bcb_test(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- int size;
- char *field;
- char *op = argv[2];
- if (bcb_field_get(argv[1], &field, &size))
- return CMD_RET_FAILURE;
- if (*op == '=' && *(op + 1) == '\0') {
- if (!strncmp(argv[3], field, size))
- return CMD_RET_SUCCESS;
- else
- return CMD_RET_FAILURE;
- } else if (*op == '~' && *(op + 1) == '\0') {
- if (!strstr(field, argv[3]))
- return CMD_RET_FAILURE;
- else
- return CMD_RET_SUCCESS;
- } else {
- printf("Error: Unknown operator '%s'\n", op);
- }
- return CMD_RET_FAILURE;
- }
- static int do_bcb_dump(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- int size;
- char *field;
- if (bcb_field_get(argv[1], &field, &size))
- return CMD_RET_FAILURE;
- print_buffer((ulong)field - (ulong)&bcb, (void *)field, 1, size, 16);
- return CMD_RET_SUCCESS;
- }
- static int do_bcb_store(struct cmd_tbl *cmdtp, int flag, int argc,
- char *const argv[])
- {
- struct blk_desc *desc;
- struct disk_partition info;
- u64 cnt;
- int ret;
- desc = blk_get_devnum_by_type(IF_TYPE_MMC, bcb_dev);
- if (!desc) {
- ret = -ENODEV;
- goto err;
- }
- ret = part_get_info(desc, bcb_part, &info);
- if (ret)
- goto err;
- cnt = DIV_ROUND_UP(sizeof(struct bootloader_message), info.blksz);
- if (blk_dwrite(desc, info.start, cnt, &bcb) != cnt) {
- ret = -EIO;
- goto err;
- }
- return CMD_RET_SUCCESS;
- err:
- printf("Error: mmc %d:%d write failed (%d)\n", bcb_dev, bcb_part, ret);
- return CMD_RET_FAILURE;
- }
- static struct cmd_tbl cmd_bcb_sub[] = {
- U_BOOT_CMD_MKENT(load, CONFIG_SYS_MAXARGS, 1, do_bcb_load, "", ""),
- U_BOOT_CMD_MKENT(set, CONFIG_SYS_MAXARGS, 1, do_bcb_set, "", ""),
- U_BOOT_CMD_MKENT(clear, CONFIG_SYS_MAXARGS, 1, do_bcb_clear, "", ""),
- U_BOOT_CMD_MKENT(test, CONFIG_SYS_MAXARGS, 1, do_bcb_test, "", ""),
- U_BOOT_CMD_MKENT(dump, CONFIG_SYS_MAXARGS, 1, do_bcb_dump, "", ""),
- U_BOOT_CMD_MKENT(store, CONFIG_SYS_MAXARGS, 1, do_bcb_store, "", ""),
- };
- static int do_bcb(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
- {
- struct cmd_tbl *c;
- if (argc < 2)
- return CMD_RET_USAGE;
- argc--;
- argv++;
- c = find_cmd_tbl(argv[0], cmd_bcb_sub, ARRAY_SIZE(cmd_bcb_sub));
- if (!c)
- return CMD_RET_USAGE;
- if (bcb_is_misused(argc, argv)) {
- /*
- * We try to improve the user experience by reporting the
- * root-cause of misusage, so don't return CMD_RET_USAGE,
- * since the latter prints out the full-blown help text
- */
- return CMD_RET_FAILURE;
- }
- return c->cmd(cmdtp, flag, argc, argv);
- }
- U_BOOT_CMD(
- bcb, CONFIG_SYS_MAXARGS, 1, do_bcb,
- "Load/set/clear/test/dump/store Android BCB fields",
- "load <dev> <part> - load BCB from mmc <dev>:<part>\n"
- "bcb set <field> <val> - set BCB <field> to <val>\n"
- "bcb clear [<field>] - clear BCB <field> or all fields\n"
- "bcb test <field> <op> <val> - test BCB <field> against <val>\n"
- "bcb dump <field> - dump BCB <field>\n"
- "bcb store - store BCB back to mmc\n"
- "\n"
- "Legend:\n"
- "<dev> - MMC device index containing the BCB partition\n"
- "<part> - MMC partition index or name containing the BCB\n"
- "<field> - one of {command,status,recovery,stage,reserved}\n"
- "<op> - the binary operator used in 'bcb test':\n"
- " '=' returns true if <val> matches the string stored in <field>\n"
- " '~' returns true if <val> matches a subset of <field>'s string\n"
- "<val> - string/text provided as input to bcb {set,test}\n"
- " NOTE: any ':' character in <val> will be replaced by line feed\n"
- " during 'bcb set' and used as separator by upper layers\n"
- );
|