From 443eb41376d93f291049738b018e9f0c0c5a56e1 Mon Sep 17 00:00:00 2001 From: Enrico Jorns Date: Tue, 20 Feb 2018 12:17:06 +0100 Subject: efi: add bootsource detection We check the supported protocol of the parent EFI device of 'efi_loaded_image' to see if we are booted from USB or not (HD) Signed-off-by: Enrico Jorns Signed-off-by: Sascha Hauer --- drivers/efi/efi-device.c | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) (limited to 'drivers/efi') diff --git a/drivers/efi/efi-device.c b/drivers/efi/efi-device.c index 9975aea6f7..3a27323a00 100644 --- a/drivers/efi/efi-device.c +++ b/drivers/efi/efi-device.c @@ -17,6 +17,7 @@ * */ +#include #include #include #include @@ -385,6 +386,51 @@ static int efi_is_setup_mode(void) return ret != 1; } +static int is_bio_usbdev(struct efi_device *efidev) +{ + int i; + + for (i = 0; i < efidev->num_guids; i++) { + if (!efi_guidcmp(efidev->guids[i], EFI_USB_IO_PROTOCOL_GUID)) + return 1; + } + + return 0; +} + +static void efi_set_bootsource(void) +{ + enum bootsource src = BOOTSOURCE_UNKNOWN; + int instance = BOOTSOURCE_INSTANCE_UNKNOWN; + + efi_handle_t *efi_parent; + struct efi_device *bootdev; + + if (!efi_loaded_image->parent_handle) + goto out; + + efi_parent = efi_find_parent(efi_loaded_image->device_handle); + + if (!efi_parent) + goto out; + + bootdev = efi_find_device(efi_parent); + + if (!bootdev) + goto out; + + if (is_bio_usbdev(bootdev)) { + src = BOOTSOURCE_USB; + } else { + src = BOOTSOURCE_HD; + } + +out: + + bootsource_set(src); + bootsource_set_instance(instance); +} + static int efi_init_devices(void) { char *fw_vendor = NULL; @@ -415,6 +461,8 @@ static int efi_init_devices(void) efi_register_devices(); + efi_set_bootsource(); + return 0; } core_initcall(efi_init_devices); -- cgit v1.2.3