drivers/firmware/efi/libstub/zboot.c

   1 // SPDX-License-Identifier: GPL-2.0
   2
   3 #include <linux/efi.h>
   4 #include <linux/pe.h>
   5 #include <asm/efi.h>
   6 #include <linux/unaligned.h>
   7
   8 #include "efistub.h"
   9
  10 static unsigned char zboot_heap[SZ_256K] __aligned(64);
  11 static unsigned long free_mem_ptr, free_mem_end_ptr;
  12
  13 #define STATIC static
  14 #if defined(CONFIG_KERNEL_GZIP)
  15 #include "../../../../lib/decompress_inflate.c"
  16 #elif defined(CONFIG_KERNEL_LZ4)
  17 #include "../../../../lib/decompress_unlz4.c"
  18 #elif defined(CONFIG_KERNEL_LZMA)
  19 #include "../../../../lib/decompress_unlzma.c"
  20 #elif defined(CONFIG_KERNEL_LZO)
  21 #include "../../../../lib/decompress_unlzo.c"
  22 #elif defined(CONFIG_KERNEL_XZ)
  23 #undef memcpy
  24 #define memcpy memcpy
  25 #undef memmove
  26 #define memmove memmove
  27 #include "../../../../lib/decompress_unxz.c"
  28 #elif defined(CONFIG_KERNEL_ZSTD)
  29 #include "../../../../lib/decompress_unzstd.c"
  30 #endif
  31
  32 extern char efi_zboot_header[];
  33 extern char _gzdata_start[], _gzdata_end[];
  34
  35 static void error(char *x)
  36 {
  37         efi_err("EFI decompressor: %s\n", x);
  38 }
  39
  40 static unsigned long alloc_preferred_address(unsigned long alloc_size)
  41 {
  42 #ifdef EFI_KIMG_PREFERRED_ADDRESS
  43         efi_physical_addr_t efi_addr = EFI_KIMG_PREFERRED_ADDRESS;
  44
  45         if (efi_bs_call(allocate_pages, EFI_ALLOCATE_ADDRESS, EFI_LOADER_DATA,
  46                         alloc_size / EFI_PAGE_SIZE, &efi_addr) == EFI_SUCCESS)
  47                 return efi_addr;
  48 #endif
  49         return ULONG_MAX;
  50 }
  51
  52 void __weak efi_cache_sync_image(unsigned long image_base,
  53                                  unsigned long alloc_size)
  54 {
  55         // Provided by the arch to perform the cache maintenance necessary for
  56         // executable code loaded into memory to be safe for execution.
  57 }
  58
  59 struct screen_info *alloc_screen_info(void)
  60 {
  61         return __alloc_screen_info();
  62 }
  63
  64 asmlinkage efi_status_t __efiapi
  65 efi_zboot_entry(efi_handle_t handle, efi_system_table_t *systab)
  66 {
  67         unsigned long compressed_size = _gzdata_end - _gzdata_start;
  68         unsigned long image_base, alloc_size;
  69         efi_loaded_image_t *image;
  70         efi_status_t status;
  71         char *cmdline_ptr;
  72         int ret;
  73
  74         WRITE_ONCE(efi_system_table, systab);
  75
  76         free_mem_ptr = (unsigned long)&zboot_heap;
  77         free_mem_end_ptr = free_mem_ptr + sizeof(zboot_heap);
  78
  79         status = efi_bs_call(handle_protocol, handle,
  80                              &LOADED_IMAGE_PROTOCOL_GUID, (void **)&image);
  81         if (status != EFI_SUCCESS) {
  82                 error("Failed to locate parent's loaded image protocol");
  83                 return status;
  84         }
  85
  86         status = efi_handle_cmdline(image, &cmdline_ptr);
  87         if (status != EFI_SUCCESS)
  88                 return status;
  89
  90         efi_info("Decompressing Linux Kernel...\n");
  91
  92         // SizeOfImage from the compressee's PE/COFF header
  93         alloc_size = round_up(get_unaligned_le32(_gzdata_end - 4),
  94                               EFI_ALLOC_ALIGN);
  95
  96          // If the architecture has a preferred address for the image,
  97          // try that first.
  98         image_base = alloc_preferred_address(alloc_size);
  99         if (image_base == ULONG_MAX) {
 100                 unsigned long min_kimg_align = efi_get_kimg_min_align();
 101                 u32 seed = U32_MAX;
 102
 103                 if (!IS_ENABLED(CONFIG_RANDOMIZE_BASE)) {
 104                         // Setting the random seed to 0x0 is the same as
 105                         // allocating as low as possible
 106                         seed = 0;
 107                 } else if (efi_nokaslr) {
 108                         efi_info("KASLR disabled on kernel command line\n");
 109                 } else {
 110                         status = efi_get_random_bytes(sizeof(seed), (u8 *)&seed);
 111                         if (status == EFI_NOT_FOUND) {
 112                                 efi_info("EFI_RNG_PROTOCOL unavailable\n");
 113                                 efi_nokaslr = true;
 114                         } else if (status != EFI_SUCCESS) {
 115                                 efi_err("efi_get_random_bytes() failed (0x%lx)\n",
 116                                         status);
 117                                 efi_nokaslr = true;
 118                         }
 119                 }
 120
 121                 status = efi_random_alloc(alloc_size, min_kimg_align, &image_base,
 122                                           seed, EFI_LOADER_CODE, 0, EFI_ALLOC_LIMIT);
 123                 if (status != EFI_SUCCESS) {
 124                         efi_err("Failed to allocate memory\n");
 125                         goto free_cmdline;
 126                 }
 127         }
 128
 129         // Decompress the payload into the newly allocated buffer.
 130         ret = __decompress(_gzdata_start, compressed_size, NULL, NULL,
 131                            (void *)image_base, alloc_size, NULL, error);
 132         if (ret < 0) {
 133                 error("Decompression failed");
 134                 status = EFI_DEVICE_ERROR;
 135                 goto free_image;
 136         }
 137
 138         efi_cache_sync_image(image_base, alloc_size);
 139
 140         status = efi_stub_common(handle, image, image_base, cmdline_ptr);
 141
 142 free_image:
 143         efi_free(alloc_size, image_base);
 144 free_cmdline:
 145         efi_bs_call(free_pool, cmdline_ptr);
 146         return status;
 147 }