diff mbox series

[v7,2/8] efi_loader: install device-tree on configuration table on every invocation

Message ID 20250310115750.1019051-3-sughosh.ganu@linaro.org
State New
Headers show
Series Add pmem node for preserving distro ISO's | expand

Commit Message

Sughosh Ganu March 10, 2025, 11:57 a.m. UTC
The efi_install_fdt() function is called before booting an EFI binary,
either directly, or through a bootmanager. This function installs a
copy of the device-tree(DT) on the EFI configuration table, which is
passed on to the OS.

The current logic in this function does not install a DT if the table
already has a DT installed on it. However, this existing copy of the
DT might not be up-to-date, or it could be a wrong DT for the image
that is being booted. Always install a DT afresh to the configuration
table before booting the EFI binary.

Installing a new DT also involves some additional checks that are
needed to clean up memory associated with the existing DT copy. Check
for an existing copy, and free up that memory.

Signed-off-by: Sughosh Ganu <sughosh.ganu@linaro.org>
---
Changes since V6: New patch

 lib/efi_loader/efi_helper.c | 37 +++++++++++++++++++++++++++----------
 1 file changed, 27 insertions(+), 10 deletions(-)
diff mbox series

Patch

diff --git a/lib/efi_loader/efi_helper.c b/lib/efi_loader/efi_helper.c
index 15ad042bc61..fe25eabeaac 100644
--- a/lib/efi_loader/efi_helper.c
+++ b/lib/efi_loader/efi_helper.c
@@ -454,11 +454,27 @@  efi_status_t efi_env_set_load_options(efi_handle_t handle,
  */
 static efi_status_t copy_fdt(void **fdtp)
 {
-	unsigned long fdt_pages;
 	efi_status_t ret = 0;
 	void *fdt, *new_fdt;
-	u64 new_fdt_addr;
-	uint fdt_size;
+	static u64 new_fdt_addr;
+	static ulong fdt_pages;
+	ulong fdt_size;
+
+	if ((new_fdt_addr && !fdt_pages) || (fdt_pages && !new_fdt_addr)) {
+		log_err("%s: Fdt address/size not in the right state\n",
+			__func__);
+		return EFI_INVALID_PARAMETER;
+	}
+
+	if (new_fdt_addr) {
+		log_debug("%s: Found allocated memory at %#llx, with %#lx pages\n",
+			__func__, new_fdt_addr, fdt_pages);
+		ret = efi_free_pages(new_fdt_addr, fdt_pages);
+		if (ret != EFI_SUCCESS) {
+			log_err("Unable to free up existing FDT memory region\n");
+			return ret;
+		}
+	}
 
 	/*
 	 * Give us at least 12 KiB of breathing room in case the device tree
@@ -472,16 +488,20 @@  static efi_status_t copy_fdt(void **fdtp)
 				 EFI_ACPI_RECLAIM_MEMORY, fdt_pages,
 				 &new_fdt_addr);
 	if (ret != EFI_SUCCESS) {
+		new_fdt_addr = fdt_pages = 0;
 		log_err("Failed to reserve space for FDT\n");
-		goto done;
+		return ret;
 	}
+	log_debug("%s: Allocated memory at %#llx, with %#lx pages\n",
+		  __func__, new_fdt_addr, fdt_pages);
+
 	new_fdt = (void *)(uintptr_t)new_fdt_addr;
 	memcpy(new_fdt, fdt, fdt_totalsize(fdt));
 	fdt_set_totalsize(new_fdt, fdt_size);
 
-	*fdtp = (void *)(uintptr_t)new_fdt_addr;
-done:
-	return ret;
+	*fdtp = new_fdt;
+
+	return EFI_SUCCESS;
 }
 
 /**
@@ -534,9 +554,6 @@  efi_status_t efi_install_fdt(void *fdt)
 		const char *fdt_opt;
 		uintptr_t fdt_addr;
 
-		/* Look for device tree that is already installed */
-		if (efi_get_configuration_table(&efi_guid_fdt))
-			return EFI_SUCCESS;
 		/* Check if there is a hardware device tree */
 		fdt_opt = env_get("fdt_addr");
 		/* Use our own device tree as fallback */