The current way how command parameters, particularly "fdt addr," are
handled makes it a bit complicated to add a subcommand-specific parameter.
So just refactor the code and extract efi_handle_fdt().

This commit is a preparatory change for enhancing bootmgr sub-command.

Signed-off-by: AKASHI Takahiro <takahiro.aka...@linaro.org>
---
 cmd/bootefi.c | 51 ++++++++++++++++++++++++++++++++++-----------------
 1 file changed, 34 insertions(+), 17 deletions(-)

diff --git a/cmd/bootefi.c b/cmd/bootefi.c
index b3e2845120fe..5772f7422e5f 100644
--- a/cmd/bootefi.c
+++ b/cmd/bootefi.c
@@ -349,6 +349,31 @@ static efi_status_t efi_install_fdt(ulong fdt_addr)
        return ret;
 }
 
+static int efi_handle_fdt(char *fdt_opt)
+{
+       unsigned long fdt_addr;
+       efi_status_t r;
+
+       if (fdt_opt) {
+               fdt_addr = simple_strtoul(fdt_opt, NULL, 16);
+               if (!fdt_addr && *fdt_opt != '0')
+                       return CMD_RET_USAGE;
+
+               /* Install device tree */
+               r = efi_install_fdt(fdt_addr);
+               if (r != EFI_SUCCESS) {
+                       printf("ERROR: failed to install device tree\n");
+                       return CMD_RET_FAILURE;
+               }
+       } else {
+               /* Remove device tree. EFI_NOT_FOUND can be ignored here */
+               efi_install_configuration_table(&efi_guid_fdt, NULL);
+               printf("WARNING: booting without device tree\n");
+       }
+
+       return CMD_RET_SUCCESS;
+}
+
 /**
  * do_bootefi_exec() - execute EFI binary
  *
@@ -516,7 +541,6 @@ static int do_bootefi(cmd_tbl_t *cmdtp, int flag, int argc, 
char * const argv[])
        unsigned long addr;
        char *saddr;
        efi_status_t r;
-       unsigned long fdt_addr;
 
        /* Allow unaligned memory access */
        allow_unaligned();
@@ -532,21 +556,6 @@ static int do_bootefi(cmd_tbl_t *cmdtp, int flag, int 
argc, char * const argv[])
        if (argc < 2)
                return CMD_RET_USAGE;
 
-       if (argc > 2) {
-               fdt_addr = simple_strtoul(argv[2], NULL, 16);
-               if (!fdt_addr && *argv[2] != '0')
-                       return CMD_RET_USAGE;
-               /* Install device tree */
-               r = efi_install_fdt(fdt_addr);
-               if (r != EFI_SUCCESS) {
-                       printf("ERROR: failed to install device tree\n");
-                       return CMD_RET_FAILURE;
-               }
-       } else {
-               /* Remove device tree. EFI_NOT_FOUND can be ignored here */
-               efi_install_configuration_table(&efi_guid_fdt, NULL);
-               printf("WARNING: booting without device tree\n");
-       }
 #ifdef CONFIG_CMD_BOOTEFI_HELLO
        if (!strcmp(argv[1], "hello")) {
                ulong size = __efi_helloworld_end - __efi_helloworld_begin;
@@ -564,6 +573,9 @@ static int do_bootefi(cmd_tbl_t *cmdtp, int flag, int argc, 
char * const argv[])
                struct efi_loaded_image_obj *image_handle;
                struct efi_loaded_image *loaded_image_info;
 
+               if (efi_handle_fdt(argc > 2 ? argv[2] : NULL))
+                       return CMD_RET_FAILURE;
+
                /* Construct a dummy device path. */
                bootefi_device_path = efi_dp_from_mem(EFI_RESERVED_MEMORY_TYPE,
                                                      (uintptr_t)&efi_selftest,
@@ -593,7 +605,10 @@ static int do_bootefi(cmd_tbl_t *cmdtp, int flag, int 
argc, char * const argv[])
        } else
 #endif
        if (!strcmp(argv[1], "bootmgr")) {
-               return do_bootefi_bootmgr_exec();
+               if (efi_handle_fdt(argc > 2 ? argv[2] : NULL))
+                       return CMD_RET_FAILURE;
+
+               return do_bootefi_bootmgr_exec(boot_id);
        } else {
                saddr = argv[1];
 
@@ -602,6 +617,8 @@ static int do_bootefi(cmd_tbl_t *cmdtp, int flag, int argc, 
char * const argv[])
                if (!addr && *saddr != '0')
                        return CMD_RET_USAGE;
 
+               if (efi_handle_fdt(argc > 2 ? argv[2] : NULL))
+                       return CMD_RET_FAILURE;
        }
 
        printf("## Starting EFI application at %08lx ...\n", addr);
-- 
2.19.0

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

Reply via email to