1 files changed, 6 insertions, 4 deletions

diff --git a/arch/arm64/kernel/efi.c b/arch/arm64/kernel/efi.c
index a98415b5979c..c9cb0fbe7aa4 100644
--- a/arch/arm64/kernel/efi.c
+++ b/arch/arm64/kernel/efi.c
@@ -229,19 +229,21 @@ static int __init arm64_enable_runtime_services(void)
                 return -1;
         }
 
-        mapsize = memmap.map_end - memmap.map;
-
         if (efi_runtime_disabled()) {
                 pr_info("EFI runtime services will be disabled.\n");
                 return -1;
         }
 
         pr_info("Remapping and enabling EFI services.\n");
-        /* replace early memmap mapping with permanent mapping */
+
+        mapsize = memmap.map_end - memmap.map;
         memmap.map = (__force void *)ioremap_cache((phys_addr_t)memmap.phys_map,
                                                    mapsize);
+        if (!memmap.map) {
+                pr_err("Failed to remap EFI memory map\n");
+                return -1;
+        }
         memmap.map_end = memmap.map + mapsize;
-
         efi.memmap = &memmap;
 
         efi.systab = (__force void *)ioremap_cache(efi_system_table,