Merge https://gitlab.denx.de/u-boot/custodians/u-boot-riscv
[platform/kernel/u-boot.git] / cmd / bootefi.c
index 54b4b8f..06563d2 100644 (file)
@@ -5,6 +5,8 @@
  *  Copyright (c) 2016 Alexander Graf
  */
 
+#define LOG_CATEGORY LOGC_EFI
+
 #include <common.h>
 #include <charset.h>
 #include <command.h>
@@ -13,6 +15,8 @@
 #include <efi_selftest.h>
 #include <env.h>
 #include <errno.h>
+#include <image.h>
+#include <log.h>
 #include <malloc.h>
 #include <linux/libfdt.h>
 #include <linux/libfdt_env.h>
@@ -27,55 +31,37 @@ static struct efi_device_path *bootefi_image_path;
 static struct efi_device_path *bootefi_device_path;
 
 /**
- * Set the load options of an image from an environment variable.
+ * efi_env_set_load_options() - set load options from environment variable
  *
  * @handle:            the image handle
  * @env_var:           name of the environment variable
  * @load_options:      pointer to load options (output)
  * Return:             status code
  */
-static efi_status_t set_load_options(efi_handle_t handle, const char *env_var,
-                                    u16 **load_options)
+static efi_status_t efi_env_set_load_options(efi_handle_t handle,
+                                            const char *env_var,
+                                            u16 **load_options)
 {
-       struct efi_loaded_image *loaded_image_info;
-       size_t size;
        const char *env = env_get(env_var);
+       size_t size;
        u16 *pos;
        efi_status_t ret;
 
        *load_options = NULL;
-       ret = EFI_CALL(systab.boottime->open_protocol(
-                                       handle,
-                                       &efi_guid_loaded_image,
-                                       (void **)&loaded_image_info,
-                                       efi_root, NULL,
-                                       EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL));
-       if (ret != EFI_SUCCESS)
-               return EFI_INVALID_PARAMETER;
-
-       loaded_image_info->load_options = NULL;
-       loaded_image_info->load_options_size = 0;
        if (!env)
-               goto out;
-
-       size = utf8_utf16_strlen(env) + 1;
-       loaded_image_info->load_options = calloc(size, sizeof(u16));
-       if (!loaded_image_info->load_options) {
-               printf("ERROR: Out of memory\n");
-               EFI_CALL(systab.boottime->close_protocol(handle,
-                                                        &efi_guid_loaded_image,
-                                                        efi_root, NULL));
+               return EFI_SUCCESS;
+       size = sizeof(u16) * (utf8_utf16_strlen(env) + 1);
+       pos = calloc(size, 1);
+       if (!pos)
                return EFI_OUT_OF_RESOURCES;
-       }
-       pos = loaded_image_info->load_options;
        *load_options = pos;
        utf8_utf16_strcpy(&pos, env);
-       loaded_image_info->load_options_size = size * 2;
-
-out:
-       return EFI_CALL(systab.boottime->close_protocol(handle,
-                                                       &efi_guid_loaded_image,
-                                                       efi_root, NULL));
+       ret = efi_set_load_options(handle, size, *load_options);
+       if (ret != EFI_SUCCESS) {
+               free(*load_options);
+               *load_options = NULL;
+       }
+       return ret;
 }
 
 #if !CONFIG_IS_ENABLED(GENERATE_ACPI_TABLE)
@@ -127,16 +113,16 @@ static efi_status_t copy_fdt(void **fdtp)
        new_fdt_addr = (uintptr_t)map_sysmem(fdt_ram_start + 0x7f00000 +
                                             fdt_size, 0);
        ret = efi_allocate_pages(EFI_ALLOCATE_MAX_ADDRESS,
-                                EFI_BOOT_SERVICES_DATA, fdt_pages,
+                                EFI_ACPI_RECLAIM_MEMORY, fdt_pages,
                                 &new_fdt_addr);
        if (ret != EFI_SUCCESS) {
                /* If we can't put it there, put it somewhere */
                new_fdt_addr = (ulong)memalign(EFI_PAGE_SIZE, fdt_size);
                ret = efi_allocate_pages(EFI_ALLOCATE_MAX_ADDRESS,
-                                        EFI_BOOT_SERVICES_DATA, fdt_pages,
+                                        EFI_ACPI_RECLAIM_MEMORY, fdt_pages,
                                         &new_fdt_addr);
                if (ret != EFI_SUCCESS) {
-                       printf("ERROR: Failed to reserve space for FDT\n");
+                       log_err("ERROR: Failed to reserve space for FDT\n");
                        goto done;
                }
        }
@@ -151,16 +137,12 @@ done:
 
 static void efi_reserve_memory(u64 addr, u64 size)
 {
-       u64 pages;
-
        /* Convert from sandbox address space. */
        addr = (uintptr_t)map_sysmem(addr, 0);
-       pages = efi_size_in_pages(size + (addr & EFI_PAGE_MASK));
-       addr &= ~EFI_PAGE_MASK;
-       if (efi_add_memory_map(addr, pages, EFI_RESERVED_MEMORY_TYPE,
-                              false) != EFI_SUCCESS)
-               printf("Reserved memory mapping failed addr %llx size %llx\n",
-                      addr, size);
+       if (efi_add_memory_map(addr, size,
+                              EFI_RESERVED_MEMORY_TYPE) != EFI_SUCCESS)
+               log_err("Reserved memory mapping failed addr %llx size %llx\n",
+                       addr, size);
 }
 
 /**
@@ -192,16 +174,20 @@ static void efi_carve_out_dt_rsv(void *fdt)
        if (nodeoffset >= 0) {
                subnode = fdt_first_subnode(fdt, nodeoffset);
                while (subnode >= 0) {
+                       fdt_addr_t fdt_addr;
+                       fdt_size_t fdt_size;
+
                        /* check if this subnode has a reg property */
-                       addr = fdtdec_get_addr_size(fdt, subnode, "reg",
-                                                   (fdt_size_t *)&size);
+                       fdt_addr = fdtdec_get_addr_size_auto_parent(
+                                               fdt, nodeoffset, subnode,
+                                               "reg", 0, &fdt_size, false);
                        /*
                         * The /reserved-memory node may have children with
                         * a size instead of a reg property.
                         */
-                       if (addr != FDT_ADDR_T_NONE &&
+                       if (fdt_addr != FDT_ADDR_T_NONE &&
                            fdtdec_get_is_enabled(fdt, subnode))
-                               efi_reserve_memory(addr, size);
+                               efi_reserve_memory(fdt_addr, fdt_size);
                        subnode = fdt_next_subnode(fdt, subnode);
                }
        }
@@ -251,7 +237,7 @@ efi_status_t efi_install_fdt(void *fdt)
         */
 #if CONFIG_IS_ENABLED(GENERATE_ACPI_TABLE)
        if (fdt) {
-               printf("ERROR: can't have ACPI table and device tree.\n");
+               log_err("ERROR: can't have ACPI table and device tree.\n");
                return EFI_LOAD_ERROR;
        }
 #else
@@ -271,13 +257,13 @@ efi_status_t efi_install_fdt(void *fdt)
                if (!fdt_opt) {
                        fdt_opt = env_get("fdtcontroladdr");
                        if (!fdt_opt) {
-                               printf("ERROR: need device tree\n");
+                               log_err("ERROR: need device tree\n");
                                return EFI_NOT_FOUND;
                        }
                }
                fdt_addr = simple_strtoul(fdt_opt, NULL, 16);
                if (!fdt_addr) {
-                       printf("ERROR: invalid $fdt_addr or $fdtcontroladdr\n");
+                       log_err("ERROR: invalid $fdt_addr or $fdtcontroladdr\n");
                        return EFI_LOAD_ERROR;
                }
                fdt = map_sysmem(fdt_addr, 0);
@@ -285,19 +271,19 @@ efi_status_t efi_install_fdt(void *fdt)
 
        /* Install device tree */
        if (fdt_check_header(fdt)) {
-               printf("ERROR: invalid device tree\n");
+               log_err("ERROR: invalid device tree\n");
                return EFI_LOAD_ERROR;
        }
 
        /* Prepare device tree for payload */
        ret = copy_fdt(&fdt);
        if (ret) {
-               printf("ERROR: out of memory\n");
+               log_err("ERROR: out of memory\n");
                return EFI_OUT_OF_RESOURCES;
        }
 
        if (image_setup_libfdt(&img, fdt, 0, NULL)) {
-               printf("ERROR: failed to process device tree\n");
+               log_err("ERROR: failed to process device tree\n");
                return EFI_LOAD_ERROR;
        }
 
@@ -307,7 +293,7 @@ efi_status_t efi_install_fdt(void *fdt)
        /* Install device tree as UEFI table */
        ret = efi_install_configuration_table(&efi_guid_fdt, fdt);
        if (ret != EFI_SUCCESS) {
-               printf("ERROR: failed to install device tree\n");
+               log_err("ERROR: failed to install device tree\n");
                return ret;
        }
 #endif /* GENERATE_ACPI_TABLE */
@@ -318,30 +304,31 @@ efi_status_t efi_install_fdt(void *fdt)
 /**
  * do_bootefi_exec() - execute EFI binary
  *
+ * The image indicated by @handle is started. When it returns the allocated
+ * memory for the @load_options is freed.
+ *
  * @handle:            handle of loaded image
+ * @load_options:      load options
  * Return:             status code
  *
  * Load the EFI binary into a newly assigned memory unwinding the relocation
  * information, install the loaded image protocol, and call the binary.
  */
-static efi_status_t do_bootefi_exec(efi_handle_t handle)
+static efi_status_t do_bootefi_exec(efi_handle_t handle, void *load_options)
 {
        efi_status_t ret;
        efi_uintn_t exit_data_size = 0;
        u16 *exit_data = NULL;
-       u16 *load_options;
-
-       /* Transfer environment variable as load options */
-       ret = set_load_options(handle, "bootargs", &load_options);
-       if (ret != EFI_SUCCESS)
-               return ret;
 
        /* Call our payload! */
        ret = EFI_CALL(efi_start_image(handle, &exit_data_size, &exit_data));
-       printf("## Application terminated, r = %lu\n", ret & ~EFI_ERROR_MASK);
-       if (ret && exit_data) {
-               printf("## %ls\n", exit_data);
-               efi_free_pool(exit_data);
+       if (ret != EFI_SUCCESS) {
+               log_err("## Application failed, r = %lu\n",
+                       ret & ~EFI_ERROR_MASK);
+               if (exit_data) {
+                       log_err("## %ls\n", exit_data);
+                       efi_free_pool(exit_data);
+               }
        }
 
        efi_restore_gd();
@@ -360,14 +347,15 @@ static int do_efibootmgr(void)
 {
        efi_handle_t handle;
        efi_status_t ret;
+       void *load_options;
 
-       ret = efi_bootmgr_load(&handle);
+       ret = efi_bootmgr_load(&handle, &load_options);
        if (ret != EFI_SUCCESS) {
-               printf("EFI boot manager: Cannot load any image\n");
+               log_notice("EFI boot manager: Cannot load any image\n");
                return CMD_RET_FAILURE;
        }
 
-       ret = do_bootefi_exec(handle);
+       ret = do_bootefi_exec(handle, load_options);
 
        if (ret != EFI_SUCCESS)
                return CMD_RET_FAILURE;
@@ -478,7 +466,14 @@ efi_status_t efi_run_image(void *source_buffer, efi_uintn_t source_size)
        if (ret != EFI_SUCCESS)
                goto out;
 
-       ret = do_bootefi_exec(handle);
+       u16 *load_options;
+
+       /* Transfer environment variable as load options */
+       ret = efi_env_set_load_options(handle, "bootargs", &load_options);
+       if (ret != EFI_SUCCESS)
+               goto out;
+
+       ret = do_bootefi_exec(handle, load_options);
 
 out:
        efi_delete_handle(mem_handle);
@@ -502,8 +497,9 @@ static efi_status_t bootefi_run_prepare(const char *load_options_path,
                return ret;
 
        /* Transfer environment variable as load options */
-       return set_load_options((efi_handle_t)*image_objp, load_options_path,
-                               &load_options);
+       return efi_env_set_load_options((efi_handle_t)*image_objp,
+                                       load_options_path,
+                                       &load_options);
 }
 
 /**
@@ -598,7 +594,8 @@ static int do_efi_selftest(void)
  * @argv:      command line arguments
  * Return:     status code
  */
-static int do_bootefi(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
+static int do_bootefi(struct cmd_tbl *cmdtp, int flag, int argc,
+                     char *const argv[])
 {
        efi_status_t ret;
        void *fdt;
@@ -609,8 +606,8 @@ static int do_bootefi(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
        /* Initialize EFI drivers */
        ret = efi_init_obj_list();
        if (ret != EFI_SUCCESS) {
-               printf("Error: Cannot initialize UEFI sub-system, r = %lu\n",
-                      ret & ~EFI_ERROR_MASK);
+               log_err("Error: Cannot initialize UEFI sub-system, r = %lu\n",
+                       ret & ~EFI_ERROR_MASK);
                return CMD_RET_FAILURE;
        }