[U-Boot] [PATCH] efi_loader: Fix some entry/exit points

[Alexander Graf]

When switching between EFI context and U-Boot context we need to swap
the register that "gd" resides in.

Some functions slipped through here, with efi_allocate_pool / efi_free_pool
not doing the switch correctly and efi_return_handle switching too often.

Fix them all up to make sure we always have consistent register state.

Signed-off-by: Alexander Graf <ag...@suse.de>
---
 lib/efi_loader/efi_boottime.c     | 13 ++++++++++---
 lib/efi_loader/efi_image_loader.c |  5 +----
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/lib/efi_loader/efi_boottime.c b/lib/efi_loader/efi_boottime.c
index 87400de..9daca50 100644
--- a/lib/efi_loader/efi_boottime.c
+++ b/lib/efi_loader/efi_boottime.c
@@ -137,12 +137,20 @@ efi_status_t EFIAPI efi_get_memory_map_ext(unsigned long 
*memory_map_size,
 static efi_status_t EFIAPI efi_allocate_pool(int pool_type, unsigned long size,
                                             void **buffer)
 {
-       return efi_allocate_pages(0, pool_type, (size + 0xfff) >> 12, 
(void*)buffer);
+       efi_status_t r;
+
+       EFI_ENTRY("%d, %ld, %p", pool_type, size, buffer);
+       r = efi_allocate_pages(0, pool_type, (size + 0xfff) >> 12, 
(void*)buffer);
+       return EFI_EXIT(r);
 }
 
 static efi_status_t EFIAPI efi_free_pool(void *buffer)
 {
-       return efi_free_pages((ulong)buffer, 0);
+       efi_status_t r;
+
+       EFI_ENTRY("%p", buffer);
+       r = efi_free_pages((ulong)buffer, 0);
+       return EFI_EXIT(r);
 }
 
 /*
@@ -706,7 +714,6 @@ static efi_status_t EFIAPI efi_handle_protocol(void *handle,
                                               efi_guid_t *protocol,
                                               void **protocol_interface)
 {
-       EFI_ENTRY("%p, %p, %p", handle, protocol, protocol_interface);
        return efi_open_protocol(handle, protocol, protocol_interface,
                                 NULL, NULL, 0);
 }
diff --git a/lib/efi_loader/efi_image_loader.c 
b/lib/efi_loader/efi_image_loader.c
index d558f5a..574b204 100644
--- a/lib/efi_loader/efi_image_loader.c
+++ b/lib/efi_loader/efi_image_loader.c
@@ -22,11 +22,8 @@ efi_status_t EFIAPI efi_return_handle(void *handle, 
efi_guid_t *protocol,
                        void **protocol_interface, void *agent_handle,
                        void *controller_handle, uint32_t attributes)
 {
-       EFI_ENTRY("%p, %p, %p, %p, %p, 0x%x", handle, protocol,
-                 protocol_interface, agent_handle, controller_handle,
-                 attributes);
        *protocol_interface = handle;
-       return EFI_EXIT(EFI_SUCCESS);
+       return EFI_SUCCESS;
 }
 
 static void efi_loader_relocate(const IMAGE_BASE_RELOCATION *rel,
-- 
1.8.5.6

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