[U-Boot] [PATCH v2 05/14] cmd: efishell: add devices command

[AKASHI Takahiro]

"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.aka...@linaro.org>
---
 cmd/efishell.c | 88 +++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 87 insertions(+), 1 deletion(-)

diff --git a/cmd/efishell.c b/cmd/efishell.c
index abc8216c7bd6..f4fa3fdf28a7 100644
--- a/cmd/efishell.c
+++ b/cmd/efishell.c
@@ -21,6 +21,8 @@
 
 DECLARE_GLOBAL_DATA_PTR;
 
+static const struct efi_boot_services *bs;
+
 static void dump_var_data(char *data, unsigned long len)
 {
        char *start, *end, *p;
@@ -266,6 +268,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.0

_______________________________________________
U-Boot mailing list
U-Boot@lists.denx.de
https://lists.denx.de/listinfo/u-boot