From: Chali Anis <chalian...@gmail.com> Add the ability to install an initrd media protocol from an initrd file or a fitImage, support both the initrd media protocol or installing the initramfs directly to efi like what linux do.
Signed-off-by: Chali Anis <chalian...@gmail.com> --- efi/guid.c | 7 +++ efi/payload/Makefile | 1 + efi/payload/efi-initrd.c | 123 ++++++++++++++++++++++++++++++++++++++ include/efi.h | 4 ++ include/efi/efi-payload.h | 3 + 5 files changed, 138 insertions(+) create mode 100644 efi/payload/efi-initrd.c diff --git a/efi/guid.c b/efi/guid.c index ef230a2b2403..7b572acb768b 100644 --- a/efi/guid.c +++ b/efi/guid.c @@ -44,6 +44,7 @@ const efi_guid_t efi_load_file_protocol_guid = EFI_LOAD_FILE_PROTOCOL_GUID; const efi_guid_t efi_load_file2_protocol_guid = EFI_LOAD_FILE2_PROTOCOL_GUID; const efi_guid_t efi_device_path_utilities_protocol_guid = EFI_DEVICE_PATH_UTILITIES_PROTOCOL_GUID; +const efi_guid_t efi_linux_initrd_media_guid = EFI_LINUX_INITRD_MEDIA_GUID; #define EFI_GUID_STRING(guid, short, long) do { \ if (!efi_guidcmp(guid, *g)) \ @@ -128,5 +129,11 @@ const char *efi_guid_string(efi_guid_t *g) EFI_GUID_STRING(EFI_ISCSIDXE_INF_GUID, "IScsiDxe.inf", "EFI IScsiDxe.inf File GUID"); EFI_GUID_STRING(EFI_VLANCONFIGDXE_INF_GUID, "VlanConfigDxe.inf", "EFI VlanConfigDxe.inf File GUID"); + /* FDT */ + EFI_GUID_STRING(EFI_DEVICE_TREE_GUID, "DeviceTree", "EFI Device Tree GUID"); + + /* Ramdisk */ + EFI_GUID_STRING(EFI_LINUX_INITRD_MEDIA_GUID, "Initrd Media", "EFI Linux Initrd Media GUID"); + return "unknown"; } diff --git a/efi/payload/Makefile b/efi/payload/Makefile index a4a0332a8288..1250535302d3 100644 --- a/efi/payload/Makefile +++ b/efi/payload/Makefile @@ -2,6 +2,7 @@ obj-y += init.o obj-y += image.o +obj-y += efi-initrd.o obj-$(CONFIG_OFTREE) += fdt.o bbenv-y += env-efi obj-$(CONFIG_CMD_IOMEM) += iomem.o diff --git a/efi/payload/efi-initrd.c b/efi/payload/efi-initrd.c new file mode 100644 index 000000000000..633fbc29c137 --- /dev/null +++ b/efi/payload/efi-initrd.c @@ -0,0 +1,123 @@ +// SPDX-License-Identifier: GPL-2.0-only +/* + * efi-initrd.c - barebox EFI payload support + * + * Copyright (c) 2025 Anis Chali <chalian...@gmail.com> + */ +#include <common.h> +#include <driver.h> +#include <init.h> +#include <linux/hw_random.h> +#include <efi.h> +#include <efi/efi-device.h> +#include <efi/device-path.h> +#include <efi/efi-payload.h> + +static efi_status_t EFIAPI efi_initrd_load_file2( + struct efi_load_file_protocol *this, struct efi_device_path *file_path, + bool boot_policy, unsigned long *buffer_size, void *buffer); + +static const struct { + struct efi_device_path_vendor vendor; + struct efi_device_path end; +} __packed initrd_dev_path = { { { + DEVICE_PATH_TYPE_MEDIA_DEVICE, + DEVICE_PATH_SUB_TYPE_VENDOR_PATH, + sizeof(initrd_dev_path.vendor), + }, + EFI_LINUX_INITRD_MEDIA_GUID }, + { DEVICE_PATH_TYPE_END, DEVICE_PATH_SUB_TYPE_END, + DEVICE_PATH_END_LENGTH } }; + +static struct efi_device_path_memory *initrd_dev; +static efi_handle_t lf2_handle; +static struct efi_load_file_protocol efi_lf2_p = { + .load_file = efi_initrd_load_file2 +}; + +static efi_status_t EFIAPI efi_initrd_load_file2( + struct efi_load_file_protocol *this, struct efi_device_path *file_path, + bool boot_policy, unsigned long *buffer_size, void *buffer) +{ + size_t initrd_size; + + if (!this || this != &efi_lf2_p || !buffer_size) + return EFI_INVALID_PARAMETER; + + if (file_path->type != initrd_dev_path.end.type || + file_path->sub_type != initrd_dev_path.end.sub_type) + return EFI_INVALID_PARAMETER; + + if (boot_policy) + return EFI_UNSUPPORTED; + + initrd_size = initrd_dev->ending_address - initrd_dev->starting_address; + if (!buffer || *buffer_size < initrd_size) { + *buffer_size = initrd_size; + return EFI_BUFFER_TOO_SMALL; + } + + else { + memcpy(buffer, (void *)(uintptr_t)initrd_dev->starting_address, + initrd_size); + *buffer_size = initrd_size; + } + + return EFI_SUCCESS; +} + +int efi_initrd_register(void *initrd, size_t initrd_sz) +{ + efi_physical_addr_t mem; + efi_status_t efiret; + size_t sz; + int ret; + + sz = sizeof(struct efi_device_path_memory) + + sizeof(struct efi_device_path); + efiret = BS->allocate_pool(EFI_BOOT_SERVICES_DATA, sz, (void **)&mem); + if (EFI_ERROR(efiret)) { + pr_err("Failed to allocate memory for INITRD %s\n", + efi_strerror(efiret)); + ret = -efi_errno(efiret); + return ret; + } + + initrd_dev = efi_phys_to_virt(mem); + initrd_dev->header.type = DEVICE_PATH_TYPE_HARDWARE_DEVICE; + initrd_dev->header.sub_type = DEVICE_PATH_SUB_TYPE_MEMORY; + initrd_dev->header.length = sizeof(struct efi_device_path_memory); + initrd_dev->memory_type = EFI_LOADER_DATA; + initrd_dev->starting_address = efi_virt_to_phys(initrd); + initrd_dev->ending_address = efi_virt_to_phys(initrd) + initrd_sz; + + memcpy(&initrd_dev[1], &initrd_dev_path.end, DEVICE_PATH_END_LENGTH); + + efiret = BS->install_multiple_protocol_interfaces( + &lf2_handle, &efi_load_file2_protocol_guid, &efi_lf2_p, + &efi_device_path_protocol_guid, &initrd_dev_path, NULL); + if (EFI_ERROR(efiret)) { + pr_err("Failed to install protocols for INITRD %s\n", + efi_strerror(efiret)); + ret = -efi_errno(efiret); + goto out; + } + + return 0; +out: + BS->free_pool(efi_phys_to_virt(mem)); + return ret; +} + +void efi_initrd_unregister(void) +{ + if (!initrd_dev) + return; + + BS->uninstall_multiple_protocol_interfaces( + lf2_handle, &efi_device_path_protocol_guid, &initrd_dev_path, + &efi_load_file2_protocol_guid, &efi_lf2_p, NULL); + + BS->free_pool(initrd_dev); + initrd_dev = NULL; +} diff --git a/include/efi.h b/include/efi.h index 40b69fdb0188..bd339ea097c3 100644 --- a/include/efi.h +++ b/include/efi.h @@ -558,6 +558,9 @@ extern struct efi_runtime_services *RT; #define EFI_DRIVER_BINDING_PROTOCOL_GUID \ EFI_GUID(0x18a031ab, 0xb443, 0x4d1a, 0xa5, 0xc0, 0x0c, 0x09, 0x26, 0x1e, 0x9f, 0x71) +#define EFI_LINUX_INITRD_MEDIA_GUID \ + EFI_GUID(0x5568e427, 0x68fc, 0x4f3d, 0xac, 0x74, 0xca, 0x55, 0x52, 0x31, 0xcc, 0x68) + struct efi_driver_binding_protocol { efi_status_t (EFIAPI * supported)( struct efi_driver_binding_protocol *this, @@ -603,6 +606,7 @@ extern const efi_guid_t efi_guid_event_group_reset_system; extern const efi_guid_t efi_load_file_protocol_guid; extern const efi_guid_t efi_load_file2_protocol_guid; extern const efi_guid_t efi_device_path_utilities_protocol_guid; +extern const efi_guid_t efi_linux_initrd_media_guid; struct efi_config_table { efi_guid_t guid; diff --git a/include/efi/efi-payload.h b/include/efi/efi-payload.h index fe45864dd8a7..d8e66a187a87 100644 --- a/include/efi/efi-payload.h +++ b/include/efi/efi-payload.h @@ -34,4 +34,7 @@ int efi_set_variable_usec(char *name, efi_guid_t *vendor, uint64_t usec); void *efi_earlymem_alloc(const struct efi_system_table *sys_table, size_t *memsize, enum efi_memory_type mem_type); +int efi_initrd_register(void *initrd, size_t initrd_size); +void efi_initrd_unregister(void); + #endif -- 2.34.1