14febfb8dSArd Biesheuvel // SPDX-License-Identifier: GPL-2.0 2f4f75ad5SArd Biesheuvel /* 3f4f75ad5SArd Biesheuvel * Helper functions used by the EFI stub on multiple 4f4f75ad5SArd Biesheuvel * architectures. This should be #included by the EFI stub 5f4f75ad5SArd Biesheuvel * implementation files. 6f4f75ad5SArd Biesheuvel * 7f4f75ad5SArd Biesheuvel * Copyright 2011 Intel Corporation; author Matt Fleming 8f4f75ad5SArd Biesheuvel */ 9f4f75ad5SArd Biesheuvel 10f4f75ad5SArd Biesheuvel #include <linux/efi.h> 11f4f75ad5SArd Biesheuvel #include <asm/efi.h> 12f4f75ad5SArd Biesheuvel 13f4f75ad5SArd Biesheuvel #include "efistub.h" 14f4f75ad5SArd Biesheuvel 155a17dae4SMatt Fleming /* 165a17dae4SMatt Fleming * Some firmware implementations have problems reading files in one go. 175a17dae4SMatt Fleming * A read chunk size of 1MB seems to work for most platforms. 185a17dae4SMatt Fleming * 195a17dae4SMatt Fleming * Unfortunately, reading files in chunks triggers *other* bugs on some 205a17dae4SMatt Fleming * platforms, so we provide a way to disable this workaround, which can 215a17dae4SMatt Fleming * be done by passing "efi=nochunk" on the EFI boot stub command line. 225a17dae4SMatt Fleming * 235a17dae4SMatt Fleming * If you experience issues with initrd images being corrupt it's worth 245a17dae4SMatt Fleming * trying efi=nochunk, but chunking is enabled by default because there 255a17dae4SMatt Fleming * are far more machines that require the workaround than those that 265a17dae4SMatt Fleming * break with it enabled. 275a17dae4SMatt Fleming */ 28f4f75ad5SArd Biesheuvel #define EFI_READ_CHUNK_SIZE (1024 * 1024) 29f4f75ad5SArd Biesheuvel 305a17dae4SMatt Fleming static unsigned long __chunk_size = EFI_READ_CHUNK_SIZE; 315a17dae4SMatt Fleming 3260f38de7SArd Biesheuvel static int __section(.data) __nokaslr; 33eeff7d63SArd Biesheuvel static int __section(.data) __quiet; 3460f38de7SArd Biesheuvel 3560f38de7SArd Biesheuvel int __pure nokaslr(void) 3660f38de7SArd Biesheuvel { 3760f38de7SArd Biesheuvel return __nokaslr; 3860f38de7SArd Biesheuvel } 39eeff7d63SArd Biesheuvel int __pure is_quiet(void) 40eeff7d63SArd Biesheuvel { 41eeff7d63SArd Biesheuvel return __quiet; 42eeff7d63SArd Biesheuvel } 4360f38de7SArd Biesheuvel 44dadb57abSJeffrey Hugo #define EFI_MMAP_NR_SLACK_SLOTS 8 45dadb57abSJeffrey Hugo 46f4f75ad5SArd Biesheuvel struct file_info { 47f4f75ad5SArd Biesheuvel efi_file_handle_t *handle; 48f4f75ad5SArd Biesheuvel u64 size; 49f4f75ad5SArd Biesheuvel }; 50f4f75ad5SArd Biesheuvel 51f4f75ad5SArd Biesheuvel void efi_printk(efi_system_table_t *sys_table_arg, char *str) 52f4f75ad5SArd Biesheuvel { 53f4f75ad5SArd Biesheuvel char *s8; 54f4f75ad5SArd Biesheuvel 55f4f75ad5SArd Biesheuvel for (s8 = str; *s8; s8++) { 56f4f75ad5SArd Biesheuvel efi_char16_t ch[2] = { 0 }; 57f4f75ad5SArd Biesheuvel 58f4f75ad5SArd Biesheuvel ch[0] = *s8; 59f4f75ad5SArd Biesheuvel if (*s8 == '\n') { 60f4f75ad5SArd Biesheuvel efi_char16_t nl[2] = { '\r', 0 }; 61f4f75ad5SArd Biesheuvel efi_char16_printk(sys_table_arg, nl); 62f4f75ad5SArd Biesheuvel } 63f4f75ad5SArd Biesheuvel 64f4f75ad5SArd Biesheuvel efi_char16_printk(sys_table_arg, ch); 65f4f75ad5SArd Biesheuvel } 66f4f75ad5SArd Biesheuvel } 67f4f75ad5SArd Biesheuvel 68dadb57abSJeffrey Hugo static inline bool mmap_has_headroom(unsigned long buff_size, 69dadb57abSJeffrey Hugo unsigned long map_size, 70dadb57abSJeffrey Hugo unsigned long desc_size) 71dadb57abSJeffrey Hugo { 72dadb57abSJeffrey Hugo unsigned long slack = buff_size - map_size; 73dadb57abSJeffrey Hugo 74dadb57abSJeffrey Hugo return slack / desc_size >= EFI_MMAP_NR_SLACK_SLOTS; 75dadb57abSJeffrey Hugo } 76dadb57abSJeffrey Hugo 77f4f75ad5SArd Biesheuvel efi_status_t efi_get_memory_map(efi_system_table_t *sys_table_arg, 78dadb57abSJeffrey Hugo struct efi_boot_memmap *map) 79f4f75ad5SArd Biesheuvel { 80f4f75ad5SArd Biesheuvel efi_memory_desc_t *m = NULL; 81f4f75ad5SArd Biesheuvel efi_status_t status; 82f4f75ad5SArd Biesheuvel unsigned long key; 83f4f75ad5SArd Biesheuvel u32 desc_version; 84f4f75ad5SArd Biesheuvel 85dadb57abSJeffrey Hugo *map->desc_size = sizeof(*m); 86dadb57abSJeffrey Hugo *map->map_size = *map->desc_size * 32; 87dadb57abSJeffrey Hugo *map->buff_size = *map->map_size; 8843a9f696SMatt Fleming again: 89f4f75ad5SArd Biesheuvel status = efi_call_early(allocate_pool, EFI_LOADER_DATA, 90dadb57abSJeffrey Hugo *map->map_size, (void **)&m); 91f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) 92f4f75ad5SArd Biesheuvel goto fail; 93f4f75ad5SArd Biesheuvel 94dadb57abSJeffrey Hugo *map->desc_size = 0; 9543a9f696SMatt Fleming key = 0; 96dadb57abSJeffrey Hugo status = efi_call_early(get_memory_map, map->map_size, m, 97dadb57abSJeffrey Hugo &key, map->desc_size, &desc_version); 98dadb57abSJeffrey Hugo if (status == EFI_BUFFER_TOO_SMALL || 99dadb57abSJeffrey Hugo !mmap_has_headroom(*map->buff_size, *map->map_size, 100dadb57abSJeffrey Hugo *map->desc_size)) { 101f4f75ad5SArd Biesheuvel efi_call_early(free_pool, m); 102dadb57abSJeffrey Hugo /* 103dadb57abSJeffrey Hugo * Make sure there is some entries of headroom so that the 104dadb57abSJeffrey Hugo * buffer can be reused for a new map after allocations are 105dadb57abSJeffrey Hugo * no longer permitted. Its unlikely that the map will grow to 106dadb57abSJeffrey Hugo * exceed this headroom once we are ready to trigger 107dadb57abSJeffrey Hugo * ExitBootServices() 108dadb57abSJeffrey Hugo */ 109dadb57abSJeffrey Hugo *map->map_size += *map->desc_size * EFI_MMAP_NR_SLACK_SLOTS; 110dadb57abSJeffrey Hugo *map->buff_size = *map->map_size; 11143a9f696SMatt Fleming goto again; 112f4f75ad5SArd Biesheuvel } 113f4f75ad5SArd Biesheuvel 114f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) 115f4f75ad5SArd Biesheuvel efi_call_early(free_pool, m); 116f4f75ad5SArd Biesheuvel 117dadb57abSJeffrey Hugo if (map->key_ptr && status == EFI_SUCCESS) 118dadb57abSJeffrey Hugo *map->key_ptr = key; 119dadb57abSJeffrey Hugo if (map->desc_ver && status == EFI_SUCCESS) 120dadb57abSJeffrey Hugo *map->desc_ver = desc_version; 121f4f75ad5SArd Biesheuvel 122f4f75ad5SArd Biesheuvel fail: 123dadb57abSJeffrey Hugo *map->map = m; 124f4f75ad5SArd Biesheuvel return status; 125f4f75ad5SArd Biesheuvel } 126f4f75ad5SArd Biesheuvel 127f4f75ad5SArd Biesheuvel 128ddeeefe2SArd Biesheuvel unsigned long get_dram_base(efi_system_table_t *sys_table_arg) 129f4f75ad5SArd Biesheuvel { 130f4f75ad5SArd Biesheuvel efi_status_t status; 131dadb57abSJeffrey Hugo unsigned long map_size, buff_size; 132f4f75ad5SArd Biesheuvel unsigned long membase = EFI_ERROR; 133f4f75ad5SArd Biesheuvel struct efi_memory_map map; 134f4f75ad5SArd Biesheuvel efi_memory_desc_t *md; 135dadb57abSJeffrey Hugo struct efi_boot_memmap boot_map; 136f4f75ad5SArd Biesheuvel 137dadb57abSJeffrey Hugo boot_map.map = (efi_memory_desc_t **)&map.map; 138dadb57abSJeffrey Hugo boot_map.map_size = &map_size; 139dadb57abSJeffrey Hugo boot_map.desc_size = &map.desc_size; 140dadb57abSJeffrey Hugo boot_map.desc_ver = NULL; 141dadb57abSJeffrey Hugo boot_map.key_ptr = NULL; 142dadb57abSJeffrey Hugo boot_map.buff_size = &buff_size; 143dadb57abSJeffrey Hugo 144dadb57abSJeffrey Hugo status = efi_get_memory_map(sys_table_arg, &boot_map); 145f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) 146f4f75ad5SArd Biesheuvel return membase; 147f4f75ad5SArd Biesheuvel 148f4f75ad5SArd Biesheuvel map.map_end = map.map + map_size; 149f4f75ad5SArd Biesheuvel 15078ce248fSMatt Fleming for_each_efi_memory_desc_in_map(&map, md) { 15178ce248fSMatt Fleming if (md->attribute & EFI_MEMORY_WB) { 152f4f75ad5SArd Biesheuvel if (membase > md->phys_addr) 153f4f75ad5SArd Biesheuvel membase = md->phys_addr; 15478ce248fSMatt Fleming } 15578ce248fSMatt Fleming } 156f4f75ad5SArd Biesheuvel 157f4f75ad5SArd Biesheuvel efi_call_early(free_pool, map.map); 158f4f75ad5SArd Biesheuvel 159f4f75ad5SArd Biesheuvel return membase; 160f4f75ad5SArd Biesheuvel } 161f4f75ad5SArd Biesheuvel 162f4f75ad5SArd Biesheuvel /* 163f4f75ad5SArd Biesheuvel * Allocate at the highest possible address that is not above 'max'. 164f4f75ad5SArd Biesheuvel */ 165f4f75ad5SArd Biesheuvel efi_status_t efi_high_alloc(efi_system_table_t *sys_table_arg, 166f4f75ad5SArd Biesheuvel unsigned long size, unsigned long align, 167f4f75ad5SArd Biesheuvel unsigned long *addr, unsigned long max) 168f4f75ad5SArd Biesheuvel { 169dadb57abSJeffrey Hugo unsigned long map_size, desc_size, buff_size; 170f4f75ad5SArd Biesheuvel efi_memory_desc_t *map; 171f4f75ad5SArd Biesheuvel efi_status_t status; 172f4f75ad5SArd Biesheuvel unsigned long nr_pages; 173f4f75ad5SArd Biesheuvel u64 max_addr = 0; 174f4f75ad5SArd Biesheuvel int i; 175dadb57abSJeffrey Hugo struct efi_boot_memmap boot_map; 176f4f75ad5SArd Biesheuvel 177dadb57abSJeffrey Hugo boot_map.map = ↦ 178dadb57abSJeffrey Hugo boot_map.map_size = &map_size; 179dadb57abSJeffrey Hugo boot_map.desc_size = &desc_size; 180dadb57abSJeffrey Hugo boot_map.desc_ver = NULL; 181dadb57abSJeffrey Hugo boot_map.key_ptr = NULL; 182dadb57abSJeffrey Hugo boot_map.buff_size = &buff_size; 183dadb57abSJeffrey Hugo 184dadb57abSJeffrey Hugo status = efi_get_memory_map(sys_table_arg, &boot_map); 185f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) 186f4f75ad5SArd Biesheuvel goto fail; 187f4f75ad5SArd Biesheuvel 188f4f75ad5SArd Biesheuvel /* 1895b88a31cSRoy Franz * Enforce minimum alignment that EFI or Linux requires when 1905b88a31cSRoy Franz * requesting a specific address. We are doing page-based (or 1915b88a31cSRoy Franz * larger) allocations, and both the address and size must meet 1925b88a31cSRoy Franz * alignment constraints. 193f4f75ad5SArd Biesheuvel */ 194cf2b0f10SArd Biesheuvel if (align < EFI_ALLOC_ALIGN) 195cf2b0f10SArd Biesheuvel align = EFI_ALLOC_ALIGN; 196f4f75ad5SArd Biesheuvel 1975b88a31cSRoy Franz size = round_up(size, EFI_ALLOC_ALIGN); 1985b88a31cSRoy Franz nr_pages = size / EFI_PAGE_SIZE; 199f4f75ad5SArd Biesheuvel again: 200f4f75ad5SArd Biesheuvel for (i = 0; i < map_size / desc_size; i++) { 201f4f75ad5SArd Biesheuvel efi_memory_desc_t *desc; 202f4f75ad5SArd Biesheuvel unsigned long m = (unsigned long)map; 203f4f75ad5SArd Biesheuvel u64 start, end; 204f4f75ad5SArd Biesheuvel 20502e43c2dSBaoquan He desc = efi_early_memdesc_ptr(m, desc_size, i); 206f4f75ad5SArd Biesheuvel if (desc->type != EFI_CONVENTIONAL_MEMORY) 207f4f75ad5SArd Biesheuvel continue; 208f4f75ad5SArd Biesheuvel 209f4f75ad5SArd Biesheuvel if (desc->num_pages < nr_pages) 210f4f75ad5SArd Biesheuvel continue; 211f4f75ad5SArd Biesheuvel 212f4f75ad5SArd Biesheuvel start = desc->phys_addr; 2135b88a31cSRoy Franz end = start + desc->num_pages * EFI_PAGE_SIZE; 214f4f75ad5SArd Biesheuvel 2157ed620bbSYinghai Lu if (end > max) 216f4f75ad5SArd Biesheuvel end = max; 217f4f75ad5SArd Biesheuvel 2187ed620bbSYinghai Lu if ((start + size) > end) 2197ed620bbSYinghai Lu continue; 2207ed620bbSYinghai Lu 221f4f75ad5SArd Biesheuvel if (round_down(end - size, align) < start) 222f4f75ad5SArd Biesheuvel continue; 223f4f75ad5SArd Biesheuvel 224f4f75ad5SArd Biesheuvel start = round_down(end - size, align); 225f4f75ad5SArd Biesheuvel 226f4f75ad5SArd Biesheuvel /* 227f4f75ad5SArd Biesheuvel * Don't allocate at 0x0. It will confuse code that 228f4f75ad5SArd Biesheuvel * checks pointers against NULL. 229f4f75ad5SArd Biesheuvel */ 230f4f75ad5SArd Biesheuvel if (start == 0x0) 231f4f75ad5SArd Biesheuvel continue; 232f4f75ad5SArd Biesheuvel 233f4f75ad5SArd Biesheuvel if (start > max_addr) 234f4f75ad5SArd Biesheuvel max_addr = start; 235f4f75ad5SArd Biesheuvel } 236f4f75ad5SArd Biesheuvel 237f4f75ad5SArd Biesheuvel if (!max_addr) 238f4f75ad5SArd Biesheuvel status = EFI_NOT_FOUND; 239f4f75ad5SArd Biesheuvel else { 240f4f75ad5SArd Biesheuvel status = efi_call_early(allocate_pages, 241f4f75ad5SArd Biesheuvel EFI_ALLOCATE_ADDRESS, EFI_LOADER_DATA, 242f4f75ad5SArd Biesheuvel nr_pages, &max_addr); 243f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) { 244f4f75ad5SArd Biesheuvel max = max_addr; 245f4f75ad5SArd Biesheuvel max_addr = 0; 246f4f75ad5SArd Biesheuvel goto again; 247f4f75ad5SArd Biesheuvel } 248f4f75ad5SArd Biesheuvel 249f4f75ad5SArd Biesheuvel *addr = max_addr; 250f4f75ad5SArd Biesheuvel } 251f4f75ad5SArd Biesheuvel 252f4f75ad5SArd Biesheuvel efi_call_early(free_pool, map); 253f4f75ad5SArd Biesheuvel fail: 254f4f75ad5SArd Biesheuvel return status; 255f4f75ad5SArd Biesheuvel } 256f4f75ad5SArd Biesheuvel 257f4f75ad5SArd Biesheuvel /* 258f4f75ad5SArd Biesheuvel * Allocate at the lowest possible address. 259f4f75ad5SArd Biesheuvel */ 260f4f75ad5SArd Biesheuvel efi_status_t efi_low_alloc(efi_system_table_t *sys_table_arg, 261f4f75ad5SArd Biesheuvel unsigned long size, unsigned long align, 262f4f75ad5SArd Biesheuvel unsigned long *addr) 263f4f75ad5SArd Biesheuvel { 264dadb57abSJeffrey Hugo unsigned long map_size, desc_size, buff_size; 265f4f75ad5SArd Biesheuvel efi_memory_desc_t *map; 266f4f75ad5SArd Biesheuvel efi_status_t status; 267f4f75ad5SArd Biesheuvel unsigned long nr_pages; 268f4f75ad5SArd Biesheuvel int i; 269dadb57abSJeffrey Hugo struct efi_boot_memmap boot_map; 270f4f75ad5SArd Biesheuvel 271dadb57abSJeffrey Hugo boot_map.map = ↦ 272dadb57abSJeffrey Hugo boot_map.map_size = &map_size; 273dadb57abSJeffrey Hugo boot_map.desc_size = &desc_size; 274dadb57abSJeffrey Hugo boot_map.desc_ver = NULL; 275dadb57abSJeffrey Hugo boot_map.key_ptr = NULL; 276dadb57abSJeffrey Hugo boot_map.buff_size = &buff_size; 277dadb57abSJeffrey Hugo 278dadb57abSJeffrey Hugo status = efi_get_memory_map(sys_table_arg, &boot_map); 279f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) 280f4f75ad5SArd Biesheuvel goto fail; 281f4f75ad5SArd Biesheuvel 282f4f75ad5SArd Biesheuvel /* 2835b88a31cSRoy Franz * Enforce minimum alignment that EFI or Linux requires when 2845b88a31cSRoy Franz * requesting a specific address. We are doing page-based (or 2855b88a31cSRoy Franz * larger) allocations, and both the address and size must meet 2865b88a31cSRoy Franz * alignment constraints. 287f4f75ad5SArd Biesheuvel */ 288cf2b0f10SArd Biesheuvel if (align < EFI_ALLOC_ALIGN) 289cf2b0f10SArd Biesheuvel align = EFI_ALLOC_ALIGN; 290f4f75ad5SArd Biesheuvel 2915b88a31cSRoy Franz size = round_up(size, EFI_ALLOC_ALIGN); 2925b88a31cSRoy Franz nr_pages = size / EFI_PAGE_SIZE; 293f4f75ad5SArd Biesheuvel for (i = 0; i < map_size / desc_size; i++) { 294f4f75ad5SArd Biesheuvel efi_memory_desc_t *desc; 295f4f75ad5SArd Biesheuvel unsigned long m = (unsigned long)map; 296f4f75ad5SArd Biesheuvel u64 start, end; 297f4f75ad5SArd Biesheuvel 29802e43c2dSBaoquan He desc = efi_early_memdesc_ptr(m, desc_size, i); 299f4f75ad5SArd Biesheuvel 300f4f75ad5SArd Biesheuvel if (desc->type != EFI_CONVENTIONAL_MEMORY) 301f4f75ad5SArd Biesheuvel continue; 302f4f75ad5SArd Biesheuvel 303f4f75ad5SArd Biesheuvel if (desc->num_pages < nr_pages) 304f4f75ad5SArd Biesheuvel continue; 305f4f75ad5SArd Biesheuvel 306f4f75ad5SArd Biesheuvel start = desc->phys_addr; 3075b88a31cSRoy Franz end = start + desc->num_pages * EFI_PAGE_SIZE; 308f4f75ad5SArd Biesheuvel 309f4f75ad5SArd Biesheuvel /* 310f4f75ad5SArd Biesheuvel * Don't allocate at 0x0. It will confuse code that 311f4f75ad5SArd Biesheuvel * checks pointers against NULL. Skip the first 8 312f4f75ad5SArd Biesheuvel * bytes so we start at a nice even number. 313f4f75ad5SArd Biesheuvel */ 314f4f75ad5SArd Biesheuvel if (start == 0x0) 315f4f75ad5SArd Biesheuvel start += 8; 316f4f75ad5SArd Biesheuvel 317f4f75ad5SArd Biesheuvel start = round_up(start, align); 318f4f75ad5SArd Biesheuvel if ((start + size) > end) 319f4f75ad5SArd Biesheuvel continue; 320f4f75ad5SArd Biesheuvel 321f4f75ad5SArd Biesheuvel status = efi_call_early(allocate_pages, 322f4f75ad5SArd Biesheuvel EFI_ALLOCATE_ADDRESS, EFI_LOADER_DATA, 323f4f75ad5SArd Biesheuvel nr_pages, &start); 324f4f75ad5SArd Biesheuvel if (status == EFI_SUCCESS) { 325f4f75ad5SArd Biesheuvel *addr = start; 326f4f75ad5SArd Biesheuvel break; 327f4f75ad5SArd Biesheuvel } 328f4f75ad5SArd Biesheuvel } 329f4f75ad5SArd Biesheuvel 330f4f75ad5SArd Biesheuvel if (i == map_size / desc_size) 331f4f75ad5SArd Biesheuvel status = EFI_NOT_FOUND; 332f4f75ad5SArd Biesheuvel 333f4f75ad5SArd Biesheuvel efi_call_early(free_pool, map); 334f4f75ad5SArd Biesheuvel fail: 335f4f75ad5SArd Biesheuvel return status; 336f4f75ad5SArd Biesheuvel } 337f4f75ad5SArd Biesheuvel 338f4f75ad5SArd Biesheuvel void efi_free(efi_system_table_t *sys_table_arg, unsigned long size, 339f4f75ad5SArd Biesheuvel unsigned long addr) 340f4f75ad5SArd Biesheuvel { 341f4f75ad5SArd Biesheuvel unsigned long nr_pages; 342f4f75ad5SArd Biesheuvel 343f4f75ad5SArd Biesheuvel if (!size) 344f4f75ad5SArd Biesheuvel return; 345f4f75ad5SArd Biesheuvel 346cf2b0f10SArd Biesheuvel nr_pages = round_up(size, EFI_ALLOC_ALIGN) / EFI_PAGE_SIZE; 347f4f75ad5SArd Biesheuvel efi_call_early(free_pages, addr, nr_pages); 348f4f75ad5SArd Biesheuvel } 349f4f75ad5SArd Biesheuvel 3502bd79f30SLukas Wunner static efi_status_t efi_file_size(efi_system_table_t *sys_table_arg, void *__fh, 3512bd79f30SLukas Wunner efi_char16_t *filename_16, void **handle, 3522bd79f30SLukas Wunner u64 *file_sz) 3532bd79f30SLukas Wunner { 3542bd79f30SLukas Wunner efi_file_handle_t *h, *fh = __fh; 3552bd79f30SLukas Wunner efi_file_info_t *info; 3562bd79f30SLukas Wunner efi_status_t status; 3572bd79f30SLukas Wunner efi_guid_t info_guid = EFI_FILE_INFO_ID; 3582bd79f30SLukas Wunner unsigned long info_sz; 3592bd79f30SLukas Wunner 3602bd79f30SLukas Wunner status = efi_call_proto(efi_file_handle, open, fh, &h, filename_16, 3612bd79f30SLukas Wunner EFI_FILE_MODE_READ, (u64)0); 3622bd79f30SLukas Wunner if (status != EFI_SUCCESS) { 3632bd79f30SLukas Wunner efi_printk(sys_table_arg, "Failed to open file: "); 3642bd79f30SLukas Wunner efi_char16_printk(sys_table_arg, filename_16); 3652bd79f30SLukas Wunner efi_printk(sys_table_arg, "\n"); 3662bd79f30SLukas Wunner return status; 3672bd79f30SLukas Wunner } 3682bd79f30SLukas Wunner 3692bd79f30SLukas Wunner *handle = h; 3702bd79f30SLukas Wunner 3712bd79f30SLukas Wunner info_sz = 0; 3722bd79f30SLukas Wunner status = efi_call_proto(efi_file_handle, get_info, h, &info_guid, 3732bd79f30SLukas Wunner &info_sz, NULL); 3742bd79f30SLukas Wunner if (status != EFI_BUFFER_TOO_SMALL) { 3752bd79f30SLukas Wunner efi_printk(sys_table_arg, "Failed to get file info size\n"); 3762bd79f30SLukas Wunner return status; 3772bd79f30SLukas Wunner } 3782bd79f30SLukas Wunner 3792bd79f30SLukas Wunner grow: 3802bd79f30SLukas Wunner status = efi_call_early(allocate_pool, EFI_LOADER_DATA, 3812bd79f30SLukas Wunner info_sz, (void **)&info); 3822bd79f30SLukas Wunner if (status != EFI_SUCCESS) { 3832bd79f30SLukas Wunner efi_printk(sys_table_arg, "Failed to alloc mem for file info\n"); 3842bd79f30SLukas Wunner return status; 3852bd79f30SLukas Wunner } 3862bd79f30SLukas Wunner 3872bd79f30SLukas Wunner status = efi_call_proto(efi_file_handle, get_info, h, &info_guid, 3882bd79f30SLukas Wunner &info_sz, info); 3892bd79f30SLukas Wunner if (status == EFI_BUFFER_TOO_SMALL) { 3902bd79f30SLukas Wunner efi_call_early(free_pool, info); 3912bd79f30SLukas Wunner goto grow; 3922bd79f30SLukas Wunner } 3932bd79f30SLukas Wunner 3942bd79f30SLukas Wunner *file_sz = info->file_size; 3952bd79f30SLukas Wunner efi_call_early(free_pool, info); 3962bd79f30SLukas Wunner 3972bd79f30SLukas Wunner if (status != EFI_SUCCESS) 3982bd79f30SLukas Wunner efi_printk(sys_table_arg, "Failed to get initrd info\n"); 3992bd79f30SLukas Wunner 4002bd79f30SLukas Wunner return status; 4012bd79f30SLukas Wunner } 4022bd79f30SLukas Wunner 4032bd79f30SLukas Wunner static efi_status_t efi_file_read(void *handle, unsigned long *size, void *addr) 4042bd79f30SLukas Wunner { 4052bd79f30SLukas Wunner return efi_call_proto(efi_file_handle, read, handle, size, addr); 4062bd79f30SLukas Wunner } 4072bd79f30SLukas Wunner 4082bd79f30SLukas Wunner static efi_status_t efi_file_close(void *handle) 4092bd79f30SLukas Wunner { 4102bd79f30SLukas Wunner return efi_call_proto(efi_file_handle, close, handle); 4112bd79f30SLukas Wunner } 4122bd79f30SLukas Wunner 413c4db9c1eSLukas Wunner static efi_status_t efi_open_volume(efi_system_table_t *sys_table_arg, 414c4db9c1eSLukas Wunner efi_loaded_image_t *image, 415c4db9c1eSLukas Wunner efi_file_handle_t **__fh) 416c4db9c1eSLukas Wunner { 417c4db9c1eSLukas Wunner efi_file_io_interface_t *io; 418c4db9c1eSLukas Wunner efi_file_handle_t *fh; 419c4db9c1eSLukas Wunner efi_guid_t fs_proto = EFI_FILE_SYSTEM_GUID; 420c4db9c1eSLukas Wunner efi_status_t status; 421c4db9c1eSLukas Wunner void *handle = (void *)(unsigned long)efi_table_attr(efi_loaded_image, 422c4db9c1eSLukas Wunner device_handle, 423c4db9c1eSLukas Wunner image); 424c4db9c1eSLukas Wunner 425c4db9c1eSLukas Wunner status = efi_call_early(handle_protocol, handle, 426c4db9c1eSLukas Wunner &fs_proto, (void **)&io); 427c4db9c1eSLukas Wunner if (status != EFI_SUCCESS) { 428c4db9c1eSLukas Wunner efi_printk(sys_table_arg, "Failed to handle fs_proto\n"); 429c4db9c1eSLukas Wunner return status; 430c4db9c1eSLukas Wunner } 431c4db9c1eSLukas Wunner 432c4db9c1eSLukas Wunner status = efi_call_proto(efi_file_io_interface, open_volume, io, &fh); 433c4db9c1eSLukas Wunner if (status != EFI_SUCCESS) 434c4db9c1eSLukas Wunner efi_printk(sys_table_arg, "Failed to open volume\n"); 435c4db9c1eSLukas Wunner else 436c4db9c1eSLukas Wunner *__fh = fh; 437c4db9c1eSLukas Wunner 438c4db9c1eSLukas Wunner return status; 439c4db9c1eSLukas Wunner } 440c4db9c1eSLukas Wunner 4415a17dae4SMatt Fleming /* 4425a17dae4SMatt Fleming * Parse the ASCII string 'cmdline' for EFI options, denoted by the efi= 4435a17dae4SMatt Fleming * option, e.g. efi=nochunk. 4445a17dae4SMatt Fleming * 4455a17dae4SMatt Fleming * It should be noted that efi= is parsed in two very different 4465a17dae4SMatt Fleming * environments, first in the early boot environment of the EFI boot 4475a17dae4SMatt Fleming * stub, and subsequently during the kernel boot. 4485a17dae4SMatt Fleming */ 44960f38de7SArd Biesheuvel efi_status_t efi_parse_options(char const *cmdline) 4505a17dae4SMatt Fleming { 4515a17dae4SMatt Fleming char *str; 4525a17dae4SMatt Fleming 45360f38de7SArd Biesheuvel str = strstr(cmdline, "nokaslr"); 45460f38de7SArd Biesheuvel if (str == cmdline || (str && str > cmdline && *(str - 1) == ' ')) 45560f38de7SArd Biesheuvel __nokaslr = 1; 456b3879a4dSArd Biesheuvel 457eeff7d63SArd Biesheuvel str = strstr(cmdline, "quiet"); 458eeff7d63SArd Biesheuvel if (str == cmdline || (str && str > cmdline && *(str - 1) == ' ')) 459eeff7d63SArd Biesheuvel __quiet = 1; 460eeff7d63SArd Biesheuvel 461b3879a4dSArd Biesheuvel /* 4625a17dae4SMatt Fleming * If no EFI parameters were specified on the cmdline we've got 4635a17dae4SMatt Fleming * nothing to do. 4645a17dae4SMatt Fleming */ 4655a17dae4SMatt Fleming str = strstr(cmdline, "efi="); 4665a17dae4SMatt Fleming if (!str) 4675a17dae4SMatt Fleming return EFI_SUCCESS; 4685a17dae4SMatt Fleming 4695a17dae4SMatt Fleming /* Skip ahead to first argument */ 4705a17dae4SMatt Fleming str += strlen("efi="); 4715a17dae4SMatt Fleming 4725a17dae4SMatt Fleming /* 4735a17dae4SMatt Fleming * Remember, because efi= is also used by the kernel we need to 4745a17dae4SMatt Fleming * skip over arguments we don't understand. 4755a17dae4SMatt Fleming */ 4764c3f14bbSArd Biesheuvel while (*str && *str != ' ') { 4775a17dae4SMatt Fleming if (!strncmp(str, "nochunk", 7)) { 4785a17dae4SMatt Fleming str += strlen("nochunk"); 4795a17dae4SMatt Fleming __chunk_size = -1UL; 4805a17dae4SMatt Fleming } 4815a17dae4SMatt Fleming 4825a17dae4SMatt Fleming /* Group words together, delimited by "," */ 4834c3f14bbSArd Biesheuvel while (*str && *str != ' ' && *str != ',') 4845a17dae4SMatt Fleming str++; 4855a17dae4SMatt Fleming 4865a17dae4SMatt Fleming if (*str == ',') 4875a17dae4SMatt Fleming str++; 4885a17dae4SMatt Fleming } 4895a17dae4SMatt Fleming 4905a17dae4SMatt Fleming return EFI_SUCCESS; 4915a17dae4SMatt Fleming } 492f4f75ad5SArd Biesheuvel 493f4f75ad5SArd Biesheuvel /* 494f4f75ad5SArd Biesheuvel * Check the cmdline for a LILO-style file= arguments. 495f4f75ad5SArd Biesheuvel * 496f4f75ad5SArd Biesheuvel * We only support loading a file from the same filesystem as 497f4f75ad5SArd Biesheuvel * the kernel image. 498f4f75ad5SArd Biesheuvel */ 499f4f75ad5SArd Biesheuvel efi_status_t handle_cmdline_files(efi_system_table_t *sys_table_arg, 500f4f75ad5SArd Biesheuvel efi_loaded_image_t *image, 501f4f75ad5SArd Biesheuvel char *cmd_line, char *option_string, 502f4f75ad5SArd Biesheuvel unsigned long max_addr, 503f4f75ad5SArd Biesheuvel unsigned long *load_addr, 504f4f75ad5SArd Biesheuvel unsigned long *load_size) 505f4f75ad5SArd Biesheuvel { 506f4f75ad5SArd Biesheuvel struct file_info *files; 507f4f75ad5SArd Biesheuvel unsigned long file_addr; 508f4f75ad5SArd Biesheuvel u64 file_size_total; 509f4f75ad5SArd Biesheuvel efi_file_handle_t *fh = NULL; 510f4f75ad5SArd Biesheuvel efi_status_t status; 511f4f75ad5SArd Biesheuvel int nr_files; 512f4f75ad5SArd Biesheuvel char *str; 513f4f75ad5SArd Biesheuvel int i, j, k; 514f4f75ad5SArd Biesheuvel 515f4f75ad5SArd Biesheuvel file_addr = 0; 516f4f75ad5SArd Biesheuvel file_size_total = 0; 517f4f75ad5SArd Biesheuvel 518f4f75ad5SArd Biesheuvel str = cmd_line; 519f4f75ad5SArd Biesheuvel 520f4f75ad5SArd Biesheuvel j = 0; /* See close_handles */ 521f4f75ad5SArd Biesheuvel 522f4f75ad5SArd Biesheuvel if (!load_addr || !load_size) 523f4f75ad5SArd Biesheuvel return EFI_INVALID_PARAMETER; 524f4f75ad5SArd Biesheuvel 525f4f75ad5SArd Biesheuvel *load_addr = 0; 526f4f75ad5SArd Biesheuvel *load_size = 0; 527f4f75ad5SArd Biesheuvel 528f4f75ad5SArd Biesheuvel if (!str || !*str) 529f4f75ad5SArd Biesheuvel return EFI_SUCCESS; 530f4f75ad5SArd Biesheuvel 531f4f75ad5SArd Biesheuvel for (nr_files = 0; *str; nr_files++) { 532f4f75ad5SArd Biesheuvel str = strstr(str, option_string); 533f4f75ad5SArd Biesheuvel if (!str) 534f4f75ad5SArd Biesheuvel break; 535f4f75ad5SArd Biesheuvel 536f4f75ad5SArd Biesheuvel str += strlen(option_string); 537f4f75ad5SArd Biesheuvel 538f4f75ad5SArd Biesheuvel /* Skip any leading slashes */ 539f4f75ad5SArd Biesheuvel while (*str == '/' || *str == '\\') 540f4f75ad5SArd Biesheuvel str++; 541f4f75ad5SArd Biesheuvel 542f4f75ad5SArd Biesheuvel while (*str && *str != ' ' && *str != '\n') 543f4f75ad5SArd Biesheuvel str++; 544f4f75ad5SArd Biesheuvel } 545f4f75ad5SArd Biesheuvel 546f4f75ad5SArd Biesheuvel if (!nr_files) 547f4f75ad5SArd Biesheuvel return EFI_SUCCESS; 548f4f75ad5SArd Biesheuvel 549f4f75ad5SArd Biesheuvel status = efi_call_early(allocate_pool, EFI_LOADER_DATA, 550f4f75ad5SArd Biesheuvel nr_files * sizeof(*files), (void **)&files); 551f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) { 552f4f75ad5SArd Biesheuvel pr_efi_err(sys_table_arg, "Failed to alloc mem for file handle list\n"); 553f4f75ad5SArd Biesheuvel goto fail; 554f4f75ad5SArd Biesheuvel } 555f4f75ad5SArd Biesheuvel 556f4f75ad5SArd Biesheuvel str = cmd_line; 557f4f75ad5SArd Biesheuvel for (i = 0; i < nr_files; i++) { 558f4f75ad5SArd Biesheuvel struct file_info *file; 559f4f75ad5SArd Biesheuvel efi_char16_t filename_16[256]; 560f4f75ad5SArd Biesheuvel efi_char16_t *p; 561f4f75ad5SArd Biesheuvel 562f4f75ad5SArd Biesheuvel str = strstr(str, option_string); 563f4f75ad5SArd Biesheuvel if (!str) 564f4f75ad5SArd Biesheuvel break; 565f4f75ad5SArd Biesheuvel 566f4f75ad5SArd Biesheuvel str += strlen(option_string); 567f4f75ad5SArd Biesheuvel 568f4f75ad5SArd Biesheuvel file = &files[i]; 569f4f75ad5SArd Biesheuvel p = filename_16; 570f4f75ad5SArd Biesheuvel 571f4f75ad5SArd Biesheuvel /* Skip any leading slashes */ 572f4f75ad5SArd Biesheuvel while (*str == '/' || *str == '\\') 573f4f75ad5SArd Biesheuvel str++; 574f4f75ad5SArd Biesheuvel 575f4f75ad5SArd Biesheuvel while (*str && *str != ' ' && *str != '\n') { 576f4f75ad5SArd Biesheuvel if ((u8 *)p >= (u8 *)filename_16 + sizeof(filename_16)) 577f4f75ad5SArd Biesheuvel break; 578f4f75ad5SArd Biesheuvel 579f4f75ad5SArd Biesheuvel if (*str == '/') { 580f4f75ad5SArd Biesheuvel *p++ = '\\'; 581f4f75ad5SArd Biesheuvel str++; 582f4f75ad5SArd Biesheuvel } else { 583f4f75ad5SArd Biesheuvel *p++ = *str++; 584f4f75ad5SArd Biesheuvel } 585f4f75ad5SArd Biesheuvel } 586f4f75ad5SArd Biesheuvel 587f4f75ad5SArd Biesheuvel *p = '\0'; 588f4f75ad5SArd Biesheuvel 589f4f75ad5SArd Biesheuvel /* Only open the volume once. */ 590f4f75ad5SArd Biesheuvel if (!i) { 591c4db9c1eSLukas Wunner status = efi_open_volume(sys_table_arg, image, &fh); 592f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) 593f4f75ad5SArd Biesheuvel goto free_files; 594f4f75ad5SArd Biesheuvel } 595f4f75ad5SArd Biesheuvel 596f4f75ad5SArd Biesheuvel status = efi_file_size(sys_table_arg, fh, filename_16, 597f4f75ad5SArd Biesheuvel (void **)&file->handle, &file->size); 598f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) 599f4f75ad5SArd Biesheuvel goto close_handles; 600f4f75ad5SArd Biesheuvel 601f4f75ad5SArd Biesheuvel file_size_total += file->size; 602f4f75ad5SArd Biesheuvel } 603f4f75ad5SArd Biesheuvel 604f4f75ad5SArd Biesheuvel if (file_size_total) { 605f4f75ad5SArd Biesheuvel unsigned long addr; 606f4f75ad5SArd Biesheuvel 607f4f75ad5SArd Biesheuvel /* 608f4f75ad5SArd Biesheuvel * Multiple files need to be at consecutive addresses in memory, 609f4f75ad5SArd Biesheuvel * so allocate enough memory for all the files. This is used 610f4f75ad5SArd Biesheuvel * for loading multiple files. 611f4f75ad5SArd Biesheuvel */ 612f4f75ad5SArd Biesheuvel status = efi_high_alloc(sys_table_arg, file_size_total, 0x1000, 613f4f75ad5SArd Biesheuvel &file_addr, max_addr); 614f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) { 615f4f75ad5SArd Biesheuvel pr_efi_err(sys_table_arg, "Failed to alloc highmem for files\n"); 616f4f75ad5SArd Biesheuvel goto close_handles; 617f4f75ad5SArd Biesheuvel } 618f4f75ad5SArd Biesheuvel 619f4f75ad5SArd Biesheuvel /* We've run out of free low memory. */ 620f4f75ad5SArd Biesheuvel if (file_addr > max_addr) { 621f4f75ad5SArd Biesheuvel pr_efi_err(sys_table_arg, "We've run out of free low memory\n"); 622f4f75ad5SArd Biesheuvel status = EFI_INVALID_PARAMETER; 623f4f75ad5SArd Biesheuvel goto free_file_total; 624f4f75ad5SArd Biesheuvel } 625f4f75ad5SArd Biesheuvel 626f4f75ad5SArd Biesheuvel addr = file_addr; 627f4f75ad5SArd Biesheuvel for (j = 0; j < nr_files; j++) { 628f4f75ad5SArd Biesheuvel unsigned long size; 629f4f75ad5SArd Biesheuvel 630f4f75ad5SArd Biesheuvel size = files[j].size; 631f4f75ad5SArd Biesheuvel while (size) { 632f4f75ad5SArd Biesheuvel unsigned long chunksize; 633b3879a4dSArd Biesheuvel 634b3879a4dSArd Biesheuvel if (IS_ENABLED(CONFIG_X86) && size > __chunk_size) 6355a17dae4SMatt Fleming chunksize = __chunk_size; 636f4f75ad5SArd Biesheuvel else 637f4f75ad5SArd Biesheuvel chunksize = size; 638f4f75ad5SArd Biesheuvel 639f4f75ad5SArd Biesheuvel status = efi_file_read(files[j].handle, 640f4f75ad5SArd Biesheuvel &chunksize, 641f4f75ad5SArd Biesheuvel (void *)addr); 642f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) { 643f4f75ad5SArd Biesheuvel pr_efi_err(sys_table_arg, "Failed to read file\n"); 644f4f75ad5SArd Biesheuvel goto free_file_total; 645f4f75ad5SArd Biesheuvel } 646f4f75ad5SArd Biesheuvel addr += chunksize; 647f4f75ad5SArd Biesheuvel size -= chunksize; 648f4f75ad5SArd Biesheuvel } 649f4f75ad5SArd Biesheuvel 650f4f75ad5SArd Biesheuvel efi_file_close(files[j].handle); 651f4f75ad5SArd Biesheuvel } 652f4f75ad5SArd Biesheuvel 653f4f75ad5SArd Biesheuvel } 654f4f75ad5SArd Biesheuvel 655f4f75ad5SArd Biesheuvel efi_call_early(free_pool, files); 656f4f75ad5SArd Biesheuvel 657f4f75ad5SArd Biesheuvel *load_addr = file_addr; 658f4f75ad5SArd Biesheuvel *load_size = file_size_total; 659f4f75ad5SArd Biesheuvel 660f4f75ad5SArd Biesheuvel return status; 661f4f75ad5SArd Biesheuvel 662f4f75ad5SArd Biesheuvel free_file_total: 663f4f75ad5SArd Biesheuvel efi_free(sys_table_arg, file_size_total, file_addr); 664f4f75ad5SArd Biesheuvel 665f4f75ad5SArd Biesheuvel close_handles: 666f4f75ad5SArd Biesheuvel for (k = j; k < i; k++) 667f4f75ad5SArd Biesheuvel efi_file_close(files[k].handle); 668f4f75ad5SArd Biesheuvel free_files: 669f4f75ad5SArd Biesheuvel efi_call_early(free_pool, files); 670f4f75ad5SArd Biesheuvel fail: 671f4f75ad5SArd Biesheuvel *load_addr = 0; 672f4f75ad5SArd Biesheuvel *load_size = 0; 673f4f75ad5SArd Biesheuvel 674f4f75ad5SArd Biesheuvel return status; 675f4f75ad5SArd Biesheuvel } 676f4f75ad5SArd Biesheuvel /* 677f4f75ad5SArd Biesheuvel * Relocate a kernel image, either compressed or uncompressed. 678f4f75ad5SArd Biesheuvel * In the ARM64 case, all kernel images are currently 679f4f75ad5SArd Biesheuvel * uncompressed, and as such when we relocate it we need to 680f4f75ad5SArd Biesheuvel * allocate additional space for the BSS segment. Any low 681f4f75ad5SArd Biesheuvel * memory that this function should avoid needs to be 682f4f75ad5SArd Biesheuvel * unavailable in the EFI memory map, as if the preferred 683f4f75ad5SArd Biesheuvel * address is not available the lowest available address will 684f4f75ad5SArd Biesheuvel * be used. 685f4f75ad5SArd Biesheuvel */ 686f4f75ad5SArd Biesheuvel efi_status_t efi_relocate_kernel(efi_system_table_t *sys_table_arg, 687f4f75ad5SArd Biesheuvel unsigned long *image_addr, 688f4f75ad5SArd Biesheuvel unsigned long image_size, 689f4f75ad5SArd Biesheuvel unsigned long alloc_size, 690f4f75ad5SArd Biesheuvel unsigned long preferred_addr, 691f4f75ad5SArd Biesheuvel unsigned long alignment) 692f4f75ad5SArd Biesheuvel { 693f4f75ad5SArd Biesheuvel unsigned long cur_image_addr; 694f4f75ad5SArd Biesheuvel unsigned long new_addr = 0; 695f4f75ad5SArd Biesheuvel efi_status_t status; 696f4f75ad5SArd Biesheuvel unsigned long nr_pages; 697f4f75ad5SArd Biesheuvel efi_physical_addr_t efi_addr = preferred_addr; 698f4f75ad5SArd Biesheuvel 699f4f75ad5SArd Biesheuvel if (!image_addr || !image_size || !alloc_size) 700f4f75ad5SArd Biesheuvel return EFI_INVALID_PARAMETER; 701f4f75ad5SArd Biesheuvel if (alloc_size < image_size) 702f4f75ad5SArd Biesheuvel return EFI_INVALID_PARAMETER; 703f4f75ad5SArd Biesheuvel 704f4f75ad5SArd Biesheuvel cur_image_addr = *image_addr; 705f4f75ad5SArd Biesheuvel 706f4f75ad5SArd Biesheuvel /* 707f4f75ad5SArd Biesheuvel * The EFI firmware loader could have placed the kernel image 708f4f75ad5SArd Biesheuvel * anywhere in memory, but the kernel has restrictions on the 709f4f75ad5SArd Biesheuvel * max physical address it can run at. Some architectures 710f4f75ad5SArd Biesheuvel * also have a prefered address, so first try to relocate 711f4f75ad5SArd Biesheuvel * to the preferred address. If that fails, allocate as low 712f4f75ad5SArd Biesheuvel * as possible while respecting the required alignment. 713f4f75ad5SArd Biesheuvel */ 714cf2b0f10SArd Biesheuvel nr_pages = round_up(alloc_size, EFI_ALLOC_ALIGN) / EFI_PAGE_SIZE; 715f4f75ad5SArd Biesheuvel status = efi_call_early(allocate_pages, 716f4f75ad5SArd Biesheuvel EFI_ALLOCATE_ADDRESS, EFI_LOADER_DATA, 717f4f75ad5SArd Biesheuvel nr_pages, &efi_addr); 718f4f75ad5SArd Biesheuvel new_addr = efi_addr; 719f4f75ad5SArd Biesheuvel /* 720f4f75ad5SArd Biesheuvel * If preferred address allocation failed allocate as low as 721f4f75ad5SArd Biesheuvel * possible. 722f4f75ad5SArd Biesheuvel */ 723f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) { 724f4f75ad5SArd Biesheuvel status = efi_low_alloc(sys_table_arg, alloc_size, alignment, 725f4f75ad5SArd Biesheuvel &new_addr); 726f4f75ad5SArd Biesheuvel } 727f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) { 728f4f75ad5SArd Biesheuvel pr_efi_err(sys_table_arg, "Failed to allocate usable memory for kernel.\n"); 729f4f75ad5SArd Biesheuvel return status; 730f4f75ad5SArd Biesheuvel } 731f4f75ad5SArd Biesheuvel 732f4f75ad5SArd Biesheuvel /* 733f4f75ad5SArd Biesheuvel * We know source/dest won't overlap since both memory ranges 734f4f75ad5SArd Biesheuvel * have been allocated by UEFI, so we can safely use memcpy. 735f4f75ad5SArd Biesheuvel */ 736f4f75ad5SArd Biesheuvel memcpy((void *)new_addr, (void *)cur_image_addr, image_size); 737f4f75ad5SArd Biesheuvel 738f4f75ad5SArd Biesheuvel /* Return the new address of the relocated image. */ 739f4f75ad5SArd Biesheuvel *image_addr = new_addr; 740f4f75ad5SArd Biesheuvel 741f4f75ad5SArd Biesheuvel return status; 742f4f75ad5SArd Biesheuvel } 743f4f75ad5SArd Biesheuvel 744f4f75ad5SArd Biesheuvel /* 745f4f75ad5SArd Biesheuvel * Get the number of UTF-8 bytes corresponding to an UTF-16 character. 746f4f75ad5SArd Biesheuvel * This overestimates for surrogates, but that is okay. 747f4f75ad5SArd Biesheuvel */ 748f4f75ad5SArd Biesheuvel static int efi_utf8_bytes(u16 c) 749f4f75ad5SArd Biesheuvel { 750f4f75ad5SArd Biesheuvel return 1 + (c >= 0x80) + (c >= 0x800); 751f4f75ad5SArd Biesheuvel } 752f4f75ad5SArd Biesheuvel 753f4f75ad5SArd Biesheuvel /* 754f4f75ad5SArd Biesheuvel * Convert an UTF-16 string, not necessarily null terminated, to UTF-8. 755f4f75ad5SArd Biesheuvel */ 756f4f75ad5SArd Biesheuvel static u8 *efi_utf16_to_utf8(u8 *dst, const u16 *src, int n) 757f4f75ad5SArd Biesheuvel { 758f4f75ad5SArd Biesheuvel unsigned int c; 759f4f75ad5SArd Biesheuvel 760f4f75ad5SArd Biesheuvel while (n--) { 761f4f75ad5SArd Biesheuvel c = *src++; 762f4f75ad5SArd Biesheuvel if (n && c >= 0xd800 && c <= 0xdbff && 763f4f75ad5SArd Biesheuvel *src >= 0xdc00 && *src <= 0xdfff) { 764f4f75ad5SArd Biesheuvel c = 0x10000 + ((c & 0x3ff) << 10) + (*src & 0x3ff); 765f4f75ad5SArd Biesheuvel src++; 766f4f75ad5SArd Biesheuvel n--; 767f4f75ad5SArd Biesheuvel } 768f4f75ad5SArd Biesheuvel if (c >= 0xd800 && c <= 0xdfff) 769f4f75ad5SArd Biesheuvel c = 0xfffd; /* Unmatched surrogate */ 770f4f75ad5SArd Biesheuvel if (c < 0x80) { 771f4f75ad5SArd Biesheuvel *dst++ = c; 772f4f75ad5SArd Biesheuvel continue; 773f4f75ad5SArd Biesheuvel } 774f4f75ad5SArd Biesheuvel if (c < 0x800) { 775f4f75ad5SArd Biesheuvel *dst++ = 0xc0 + (c >> 6); 776f4f75ad5SArd Biesheuvel goto t1; 777f4f75ad5SArd Biesheuvel } 778f4f75ad5SArd Biesheuvel if (c < 0x10000) { 779f4f75ad5SArd Biesheuvel *dst++ = 0xe0 + (c >> 12); 780f4f75ad5SArd Biesheuvel goto t2; 781f4f75ad5SArd Biesheuvel } 782f4f75ad5SArd Biesheuvel *dst++ = 0xf0 + (c >> 18); 783f4f75ad5SArd Biesheuvel *dst++ = 0x80 + ((c >> 12) & 0x3f); 784f4f75ad5SArd Biesheuvel t2: 785f4f75ad5SArd Biesheuvel *dst++ = 0x80 + ((c >> 6) & 0x3f); 786f4f75ad5SArd Biesheuvel t1: 787f4f75ad5SArd Biesheuvel *dst++ = 0x80 + (c & 0x3f); 788f4f75ad5SArd Biesheuvel } 789f4f75ad5SArd Biesheuvel 790f4f75ad5SArd Biesheuvel return dst; 791f4f75ad5SArd Biesheuvel } 792f4f75ad5SArd Biesheuvel 79348fcb2d0SArd Biesheuvel #ifndef MAX_CMDLINE_ADDRESS 79448fcb2d0SArd Biesheuvel #define MAX_CMDLINE_ADDRESS ULONG_MAX 79548fcb2d0SArd Biesheuvel #endif 79648fcb2d0SArd Biesheuvel 797f4f75ad5SArd Biesheuvel /* 798f4f75ad5SArd Biesheuvel * Convert the unicode UEFI command line to ASCII to pass to kernel. 799f4f75ad5SArd Biesheuvel * Size of memory allocated return in *cmd_line_len. 800f4f75ad5SArd Biesheuvel * Returns NULL on error. 801f4f75ad5SArd Biesheuvel */ 802f4f75ad5SArd Biesheuvel char *efi_convert_cmdline(efi_system_table_t *sys_table_arg, 803f4f75ad5SArd Biesheuvel efi_loaded_image_t *image, 804f4f75ad5SArd Biesheuvel int *cmd_line_len) 805f4f75ad5SArd Biesheuvel { 806f4f75ad5SArd Biesheuvel const u16 *s2; 807f4f75ad5SArd Biesheuvel u8 *s1 = NULL; 808f4f75ad5SArd Biesheuvel unsigned long cmdline_addr = 0; 809f4f75ad5SArd Biesheuvel int load_options_chars = image->load_options_size / 2; /* UTF-16 */ 810f4f75ad5SArd Biesheuvel const u16 *options = image->load_options; 811f4f75ad5SArd Biesheuvel int options_bytes = 0; /* UTF-8 bytes */ 812f4f75ad5SArd Biesheuvel int options_chars = 0; /* UTF-16 chars */ 813f4f75ad5SArd Biesheuvel efi_status_t status; 814f4f75ad5SArd Biesheuvel u16 zero = 0; 815f4f75ad5SArd Biesheuvel 816f4f75ad5SArd Biesheuvel if (options) { 817f4f75ad5SArd Biesheuvel s2 = options; 818f4f75ad5SArd Biesheuvel while (*s2 && *s2 != '\n' 819f4f75ad5SArd Biesheuvel && options_chars < load_options_chars) { 820f4f75ad5SArd Biesheuvel options_bytes += efi_utf8_bytes(*s2++); 821f4f75ad5SArd Biesheuvel options_chars++; 822f4f75ad5SArd Biesheuvel } 823f4f75ad5SArd Biesheuvel } 824f4f75ad5SArd Biesheuvel 825f4f75ad5SArd Biesheuvel if (!options_chars) { 826f4f75ad5SArd Biesheuvel /* No command line options, so return empty string*/ 827f4f75ad5SArd Biesheuvel options = &zero; 828f4f75ad5SArd Biesheuvel } 829f4f75ad5SArd Biesheuvel 830f4f75ad5SArd Biesheuvel options_bytes++; /* NUL termination */ 831f4f75ad5SArd Biesheuvel 83248fcb2d0SArd Biesheuvel status = efi_high_alloc(sys_table_arg, options_bytes, 0, 83348fcb2d0SArd Biesheuvel &cmdline_addr, MAX_CMDLINE_ADDRESS); 834f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) 835f4f75ad5SArd Biesheuvel return NULL; 836f4f75ad5SArd Biesheuvel 837f4f75ad5SArd Biesheuvel s1 = (u8 *)cmdline_addr; 838f4f75ad5SArd Biesheuvel s2 = (const u16 *)options; 839f4f75ad5SArd Biesheuvel 840f4f75ad5SArd Biesheuvel s1 = efi_utf16_to_utf8(s1, s2, options_chars); 841f4f75ad5SArd Biesheuvel *s1 = '\0'; 842f4f75ad5SArd Biesheuvel 843f4f75ad5SArd Biesheuvel *cmd_line_len = options_bytes; 844f4f75ad5SArd Biesheuvel return (char *)cmdline_addr; 845f4f75ad5SArd Biesheuvel } 846fc07716bSJeffrey Hugo 847fc07716bSJeffrey Hugo /* 848fc07716bSJeffrey Hugo * Handle calling ExitBootServices according to the requirements set out by the 849fc07716bSJeffrey Hugo * spec. Obtains the current memory map, and returns that info after calling 850fc07716bSJeffrey Hugo * ExitBootServices. The client must specify a function to perform any 851fc07716bSJeffrey Hugo * processing of the memory map data prior to ExitBootServices. A client 852fc07716bSJeffrey Hugo * specific structure may be passed to the function via priv. The client 853fc07716bSJeffrey Hugo * function may be called multiple times. 854fc07716bSJeffrey Hugo */ 855fc07716bSJeffrey Hugo efi_status_t efi_exit_boot_services(efi_system_table_t *sys_table_arg, 856fc07716bSJeffrey Hugo void *handle, 857fc07716bSJeffrey Hugo struct efi_boot_memmap *map, 858fc07716bSJeffrey Hugo void *priv, 859fc07716bSJeffrey Hugo efi_exit_boot_map_processing priv_func) 860fc07716bSJeffrey Hugo { 861fc07716bSJeffrey Hugo efi_status_t status; 862fc07716bSJeffrey Hugo 863fc07716bSJeffrey Hugo status = efi_get_memory_map(sys_table_arg, map); 864fc07716bSJeffrey Hugo 865fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 866fc07716bSJeffrey Hugo goto fail; 867fc07716bSJeffrey Hugo 868fc07716bSJeffrey Hugo status = priv_func(sys_table_arg, map, priv); 869fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 870fc07716bSJeffrey Hugo goto free_map; 871fc07716bSJeffrey Hugo 872fc07716bSJeffrey Hugo status = efi_call_early(exit_boot_services, handle, *map->key_ptr); 873fc07716bSJeffrey Hugo 874fc07716bSJeffrey Hugo if (status == EFI_INVALID_PARAMETER) { 875fc07716bSJeffrey Hugo /* 876fc07716bSJeffrey Hugo * The memory map changed between efi_get_memory_map() and 877fc07716bSJeffrey Hugo * exit_boot_services(). Per the UEFI Spec v2.6, Section 6.4: 878fc07716bSJeffrey Hugo * EFI_BOOT_SERVICES.ExitBootServices we need to get the 879fc07716bSJeffrey Hugo * updated map, and try again. The spec implies one retry 880fc07716bSJeffrey Hugo * should be sufficent, which is confirmed against the EDK2 881fc07716bSJeffrey Hugo * implementation. Per the spec, we can only invoke 882fc07716bSJeffrey Hugo * get_memory_map() and exit_boot_services() - we cannot alloc 883fc07716bSJeffrey Hugo * so efi_get_memory_map() cannot be used, and we must reuse 884fc07716bSJeffrey Hugo * the buffer. For all practical purposes, the headroom in the 885fc07716bSJeffrey Hugo * buffer should account for any changes in the map so the call 886fc07716bSJeffrey Hugo * to get_memory_map() is expected to succeed here. 887fc07716bSJeffrey Hugo */ 888fc07716bSJeffrey Hugo *map->map_size = *map->buff_size; 889fc07716bSJeffrey Hugo status = efi_call_early(get_memory_map, 890fc07716bSJeffrey Hugo map->map_size, 891fc07716bSJeffrey Hugo *map->map, 892fc07716bSJeffrey Hugo map->key_ptr, 893fc07716bSJeffrey Hugo map->desc_size, 894fc07716bSJeffrey Hugo map->desc_ver); 895fc07716bSJeffrey Hugo 896fc07716bSJeffrey Hugo /* exit_boot_services() was called, thus cannot free */ 897fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 898fc07716bSJeffrey Hugo goto fail; 899fc07716bSJeffrey Hugo 900fc07716bSJeffrey Hugo status = priv_func(sys_table_arg, map, priv); 901fc07716bSJeffrey Hugo /* exit_boot_services() was called, thus cannot free */ 902fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 903fc07716bSJeffrey Hugo goto fail; 904fc07716bSJeffrey Hugo 905fc07716bSJeffrey Hugo status = efi_call_early(exit_boot_services, handle, *map->key_ptr); 906fc07716bSJeffrey Hugo } 907fc07716bSJeffrey Hugo 908fc07716bSJeffrey Hugo /* exit_boot_services() was called, thus cannot free */ 909fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 910fc07716bSJeffrey Hugo goto fail; 911fc07716bSJeffrey Hugo 912fc07716bSJeffrey Hugo return EFI_SUCCESS; 913fc07716bSJeffrey Hugo 914fc07716bSJeffrey Hugo free_map: 915fc07716bSJeffrey Hugo efi_call_early(free_pool, *map->map); 916fc07716bSJeffrey Hugo fail: 917fc07716bSJeffrey Hugo return status; 918fc07716bSJeffrey Hugo } 919