[U-Boot] [PATCH v3 2/8] cmd: efishell: add devices command

AKASHI Takahiro
2018-12-18T05:05:04Z

"devices" command prints all the uefi variables on the system.
=> efishell devices
Device Name
============================================
/VenHw(e61d73b9-a384-4acc-aeab-82e828f3628b)
/VenHw(e61d73b9-a384-4acc-aeab-82e828f3628b)/SD(0)/SD(0)
/VenHw(e61d73b9-a384-4acc-aeab-82e828f3628b)/SD(0)/SD(0)/\
HD(2,MBR,0x086246ba,0x40800,0x3f800)
/VenHw(e61d73b9-a384-4acc-aeab-82e828f3628b)/SD(0)/SD(0)/\
HD(1,MBR,0x086246ba,0x800,0x40000)

Signed-off-by: AKASHI Takahiro <takahiro.akashi@linaro.org>
---
cmd/efishell.c | 88 +++++++++++++++++++++++++++++++++++++++++++++++++-
1 file changed, 87 insertions(+), 1 deletion(-)

diff --git a/cmd/efishell.c b/cmd/efishell.c
index 5819e52cf575..929b6343b1b2 100644
--- a/cmd/efishell.c
+++ b/cmd/efishell.c
@@ -18,6 +18,8 @@
#include <linux/ctype.h>
#include <asm/global_data.h>

+static const struct efi_boot_services *bs;
+
static void dump_var_data(char *data, unsigned long len)
{
char *start, *end, *p;
@@ -263,6 +265,84 @@ out:
return ret;
}

+static efi_handle_t *efi_get_handles_by_proto(efi_guid_t *guid)
+{
+ efi_handle_t *handles = NULL;
+ efi_uintn_t size = 0;
+ efi_status_t ret;
+
+ if (guid) {
+ ret = bs->locate_handle(BY_PROTOCOL, guid, NULL, &size,
+ handles);
+ if (ret == EFI_BUFFER_TOO_SMALL) {
+ handles = calloc(1, size);
+ if (!handles)
+ return NULL;
+
+ ret = bs->locate_handle(BY_PROTOCOL, guid, NULL, &size,
+ handles);
+ }
+ if (ret != EFI_SUCCESS) {
+ free(handles);
+ return NULL;
+ }
+ } else {
+ ret = bs->locate_handle(ALL_HANDLES, NULL, NULL, &size, NULL);
+ if (ret == EFI_BUFFER_TOO_SMALL) {
+ handles = calloc(1, size);
+ if (!handles)
+ return NULL;
+
+ ret = bs->locate_handle(ALL_HANDLES, NULL, NULL, &size,
+ handles);
+ }
+ if (ret != EFI_SUCCESS) {
+ free(handles);
+ return NULL;
+ }
+ }
+
+ return handles;
+}
+
+static int efi_get_device_handle_info(efi_handle_t handle, u16 **name)
+{
+ struct efi_device_path *dp;
+ efi_status_t ret;
+
+ ret = bs->open_protocol(handle, &efi_guid_device_path,
+ (void **)&dp, NULL /* FIXME */, NULL,
+ EFI_OPEN_PROTOCOL_GET_PROTOCOL);
+ if (ret == EFI_SUCCESS) {
+ *name = efi_dp_str(dp);
+ return 0;
+ } else {
+ return -1;
+ }
+}
+
+static int do_efi_show_devices(int argc, char * const argv[])
+{
+ efi_handle_t *handles = NULL, *handle;
+ u16 *devname;
+
+ handles = efi_get_handles_by_proto(NULL);
+ if (!handles)
+ return CMD_RET_SUCCESS;
+
+ printf("Device Name\n");
+ printf("============================================\n");
+ for (handle = handles; *handle; handle++) {
+ if (!efi_get_device_handle_info(*handle, &devname)) {
+ printf("%ls\n", devname);
+ efi_free_pool(devname);
+ }
+ handle++;
+ }
+
+ return CMD_RET_SUCCESS;
+}
+
static int do_efi_boot_add(int argc, char * const argv[])
{
int id;
@@ -625,6 +705,8 @@ static int do_efishell(cmd_tbl_t *cmdtp, int flag,
if (argc < 2)
return CMD_RET_USAGE;

+ bs = systab.boottime;
+
argc--; argv++;
command = argv[0];

@@ -642,6 +724,8 @@ static int do_efishell(cmd_tbl_t *cmdtp, int flag,
return do_efi_dump_var(argc, argv);
else if (!strcmp(command, "setvar"))
return do_efi_set_var(argc, argv);
+ else if (!strcmp(command, "devices"))
+ return do_efi_show_devices(argc, argv);
else
return CMD_RET_USAGE;
}
@@ -663,7 +747,9 @@ static char efishell_help_text[] =
" - get uefi variable's value\n"
"efishell setvar <name> [<value>]\n"
" - set/delete uefi variable's value\n"
- " <value> may be \"=\"...\"\", \"=0x...\" (set) or \"=\" (delete)\n";
+ " <value> may be \"=\"...\"\", \"=0x...\" (set) or \"=\" (delete)\n"
+ "efishell devices\n"
+ " - show uefi devices\n";
#endif

U_BOOT_CMD(
--
2.19.1