blob: 1fa7145bfee09d30917317e16f49d8317e92b3b0 [file] [log] [blame]
/*
* (C) Copyright 2018, Linaro Limited
*
* SPDX-License-Identifier: GPL-2.0+
*/
#include <avb_verify.h>
#include <command.h>
#include <env.h>
#include <image.h>
#include <malloc.h>
#define AVB_BOOTARGS "avb_bootargs"
static struct AvbOps *avb_ops;
int do_avb_init(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
{
const char *iface;
const char *devnum;
if (argc != 3)
return CMD_RET_USAGE;
iface = argv[1];
devnum = argv[2];
if (avb_ops)
avb_ops_free(avb_ops);
avb_ops = avb_ops_alloc(iface, devnum);
if (avb_ops)
return CMD_RET_SUCCESS;
printf("Failed to initialize avb2\n");
return CMD_RET_FAILURE;
}
int do_avb_read_part(struct cmd_tbl *cmdtp, int flag, int argc,
char *const argv[])
{
const char *part;
s64 offset;
size_t bytes, bytes_read = 0;
void *buffer;
if (!avb_ops) {
printf("AVB 2.0 is not initialized, please run 'avb init'\n");
return CMD_RET_USAGE;
}
if (argc != 5)
return CMD_RET_USAGE;
part = argv[1];
offset = hextoul(argv[2], NULL);
bytes = hextoul(argv[3], NULL);
buffer = (void *)hextoul(argv[4], NULL);
if (avb_ops->read_from_partition(avb_ops, part, offset, bytes,
buffer, &bytes_read) ==
AVB_IO_RESULT_OK) {
printf("Read %zu bytes\n", bytes_read);
return CMD_RET_SUCCESS;
}
printf("Failed to read from partition\n");
return CMD_RET_FAILURE;
}
int do_avb_read_part_hex(struct cmd_tbl *cmdtp, int flag, int argc,
char *const argv[])
{
const char *part;
s64 offset;
size_t bytes, bytes_read = 0;
char *buffer;
if (!avb_ops) {
printf("AVB 2.0 is not initialized, please run 'avb init'\n");
return CMD_RET_USAGE;
}
if (argc != 4)
return CMD_RET_USAGE;
part = argv[1];
offset = hextoul(argv[2], NULL);
bytes = hextoul(argv[3], NULL);
buffer = malloc(bytes);
if (!buffer) {
printf("Failed to tlb_allocate buffer for data\n");
return CMD_RET_FAILURE;
}
memset(buffer, 0, bytes);
if (avb_ops->read_from_partition(avb_ops, part, offset, bytes, buffer,
&bytes_read) == AVB_IO_RESULT_OK) {
printf("Requested %zu, read %zu bytes\n", bytes, bytes_read);
printf("Data: ");
for (int i = 0; i < bytes_read; i++)
printf("%02X", buffer[i]);
printf("\n");
free(buffer);
return CMD_RET_SUCCESS;
}
printf("Failed to read from partition\n");
free(buffer);
return CMD_RET_FAILURE;
}
int do_avb_write_part(struct cmd_tbl *cmdtp, int flag, int argc,
char *const argv[])
{
const char *part;
s64 offset;
size_t bytes;
void *buffer;
if (!avb_ops) {
printf("AVB 2.0 is not initialized, run 'avb init' first\n");
return CMD_RET_FAILURE;
}
if (argc != 5)
return CMD_RET_USAGE;
part = argv[1];
offset = hextoul(argv[2], NULL);
bytes = hextoul(argv[3], NULL);
buffer = (void *)hextoul(argv[4], NULL);
if (avb_ops->write_to_partition(avb_ops, part, offset, bytes, buffer) ==
AVB_IO_RESULT_OK) {
printf("Wrote %zu bytes\n", bytes);
return CMD_RET_SUCCESS;
}
printf("Failed to write in partition\n");
return CMD_RET_FAILURE;
}
int do_avb_read_rb(struct cmd_tbl *cmdtp, int flag, int argc,
char *const argv[])
{
size_t index;
u64 rb_idx;
if (!avb_ops) {
printf("AVB 2.0 is not initialized, run 'avb init' first\n");
return CMD_RET_FAILURE;
}
if (argc != 2)
return CMD_RET_USAGE;
index = (size_t)hextoul(argv[1], NULL);
if (avb_ops->read_rollback_index(avb_ops, index, &rb_idx) ==
AVB_IO_RESULT_OK) {
printf("Rollback index: %llx\n", rb_idx);
return CMD_RET_SUCCESS;
}
printf("Failed to read rollback index\n");
return CMD_RET_FAILURE;
}
int do_avb_write_rb(struct cmd_tbl *cmdtp, int flag, int argc,
char *const argv[])
{
size_t index;
u64 rb_idx;
if (!avb_ops) {
printf("AVB 2.0 is not initialized, run 'avb init' first\n");
return CMD_RET_FAILURE;
}
if (argc != 3)
return CMD_RET_USAGE;
index = (size_t)hextoul(argv[1], NULL);
rb_idx = hextoul(argv[2], NULL);
if (avb_ops->write_rollback_index(avb_ops, index, rb_idx) ==
AVB_IO_RESULT_OK)
return CMD_RET_SUCCESS;
printf("Failed to write rollback index\n");
return CMD_RET_FAILURE;
}
int do_avb_get_uuid(struct cmd_tbl *cmdtp, int flag,
int argc, char *const argv[])
{
const char *part;
char buffer[UUID_STR_LEN + 1];
if (!avb_ops) {
printf("AVB 2.0 is not initialized, run 'avb init' first\n");
return CMD_RET_FAILURE;
}
if (argc != 2)
return CMD_RET_USAGE;
part = argv[1];
if (avb_ops->get_unique_guid_for_partition(avb_ops, part, buffer,
UUID_STR_LEN + 1) ==
AVB_IO_RESULT_OK) {
printf("'%s' UUID: %s\n", part, buffer);
return CMD_RET_SUCCESS;
}
printf("Failed to read UUID\n");
return CMD_RET_FAILURE;
}
int do_avb_verify_part(struct cmd_tbl *cmdtp, int flag,
int argc, char *const argv[])
{
AvbSlotVerifyData *out_data = NULL;
char *out_cmdline = NULL;
char *slot_suffix = "";
int res = CMD_RET_FAILURE;
if (!avb_ops) {
printf("AVB 2.0 is not initialized, run 'avb init' first\n");
return CMD_RET_FAILURE;
}
if (argc < 1 || argc > 2)
return CMD_RET_USAGE;
if (argc == 2)
slot_suffix = argv[1];
if (avb_verify(avb_ops, slot_suffix, &out_data, &out_cmdline) == CMD_RET_SUCCESS) {
env_set(AVB_BOOTARGS, out_cmdline);
res = CMD_RET_SUCCESS;
}
if (out_cmdline)
free(out_cmdline);
if (out_data)
avb_slot_verify_data_free(out_data);
return res;
}
int do_avb_is_unlocked(struct cmd_tbl *cmdtp, int flag,
int argc, char *const argv[])
{
bool unlock;
if (!avb_ops) {
printf("AVB not initialized, run 'avb init' first\n");
return CMD_RET_FAILURE;
}
if (argc != 1) {
printf("--%s(-1)\n", __func__);
return CMD_RET_USAGE;
}
if (avb_ops->read_is_device_unlocked(avb_ops, &unlock) ==
AVB_IO_RESULT_OK) {
printf("Unlocked = %d\n", unlock);
return CMD_RET_SUCCESS;
}
printf("Can't determine device lock state.\n");
return CMD_RET_FAILURE;
}
int do_avb_read_pvalue(struct cmd_tbl *cmdtp, int flag, int argc,
char *const argv[])
{
const char *name;
size_t bytes;
size_t bytes_read;
void *buffer;
char *endp;
if (!avb_ops) {
printf("AVB 2.0 is not initialized, run 'avb init' first\n");
return CMD_RET_FAILURE;
}
if (argc != 3)
return CMD_RET_USAGE;
name = argv[1];
bytes = dectoul(argv[2], &endp);
if (*endp && *endp != '\n')
return CMD_RET_USAGE;
buffer = malloc(bytes);
if (!buffer)
return CMD_RET_FAILURE;
if (avb_ops->read_persistent_value(avb_ops, name, bytes, buffer,
&bytes_read) == AVB_IO_RESULT_OK) {
printf("Read %zu bytes, value = %s\n", bytes_read,
(char *)buffer);
free(buffer);
return CMD_RET_SUCCESS;
}
printf("Failed to read persistent value\n");
free(buffer);
return CMD_RET_FAILURE;
}
int do_avb_write_pvalue(struct cmd_tbl *cmdtp, int flag, int argc,
char *const argv[])
{
const char *name;
const char *value;
if (!avb_ops) {
printf("AVB 2.0 is not initialized, run 'avb init' first\n");
return CMD_RET_FAILURE;
}
if (argc != 3)
return CMD_RET_USAGE;
name = argv[1];
value = argv[2];
if (avb_ops->write_persistent_value(avb_ops, name, strlen(value) + 1,
(const uint8_t *)value) ==
AVB_IO_RESULT_OK) {
printf("Wrote %zu bytes\n", strlen(value) + 1);
return CMD_RET_SUCCESS;
}
printf("Failed to write persistent value\n");
return CMD_RET_FAILURE;
}
static struct cmd_tbl cmd_avb[] = {
U_BOOT_CMD_MKENT(init, 3, 0, do_avb_init, "", ""),
U_BOOT_CMD_MKENT(read_rb, 2, 0, do_avb_read_rb, "", ""),
U_BOOT_CMD_MKENT(write_rb, 3, 0, do_avb_write_rb, "", ""),
U_BOOT_CMD_MKENT(is_unlocked, 1, 0, do_avb_is_unlocked, "", ""),
U_BOOT_CMD_MKENT(get_uuid, 2, 0, do_avb_get_uuid, "", ""),
U_BOOT_CMD_MKENT(read_part, 5, 0, do_avb_read_part, "", ""),
U_BOOT_CMD_MKENT(read_part_hex, 4, 0, do_avb_read_part_hex, "", ""),
U_BOOT_CMD_MKENT(write_part, 5, 0, do_avb_write_part, "", ""),
U_BOOT_CMD_MKENT(verify, 2, 0, do_avb_verify_part, "", ""),
#ifdef CONFIG_OPTEE_TA_AVB
U_BOOT_CMD_MKENT(read_pvalue, 3, 0, do_avb_read_pvalue, "", ""),
U_BOOT_CMD_MKENT(write_pvalue, 3, 0, do_avb_write_pvalue, "", ""),
#endif
};
static int do_avb(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
{
struct cmd_tbl *cp;
cp = find_cmd_tbl(argv[1], cmd_avb, ARRAY_SIZE(cmd_avb));
argc--;
argv++;
if (!cp || argc > cp->maxargs)
return CMD_RET_USAGE;
if (flag == CMD_FLAG_REPEAT)
return CMD_RET_FAILURE;
return cp->cmd(cmdtp, flag, argc, argv);
}
U_BOOT_CMD(
avb, 29, 0, do_avb,
"Provides commands for testing Android Verified Boot 2.0 functionality",
"init <interface> <devnum> - initialize avb2 for the disk <devnum> which\n"
" is on the interface <interface>\n"
"avb read_rb <num> - read rollback index at location <num>\n"
"avb write_rb <num> <rb> - write rollback index <rb> to <num>\n"
"avb is_unlocked - returns unlock status of the device\n"
"avb get_uuid <partname> - read and print uuid of partition <part>\n"
"avb read_part <partname> <offset> <num> <addr> - read <num> bytes from\n"
" partition <partname> to buffer <addr>\n"
"avb read_part_hex <partname> <offset> <num> - read <num> bytes from\n"
" partition <partname> and print to stdout\n"
"avb write_part <partname> <offset> <num> <addr> - write <num> bytes to\n"
" <partname> by <offset> using data from <addr>\n"
#ifdef CONFIG_OPTEE_TA_AVB
"avb read_pvalue <name> <bytes> - read a persistent value <name>\n"
"avb write_pvalue <name> <value> - write a persistent value <name>\n"
#endif
"avb verify [slot_suffix] - run verification process using hash data\n"
" from vbmeta structure\n"
" [slot_suffix] - _a, _b, etc (if vbmeta partition is slotted)\n"
);