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 102c7d1e30SArvind Sankar #include <stdarg.h> 112c7d1e30SArvind Sankar 1280b1bfe1SArvind Sankar #include <linux/ctype.h> 13f4f75ad5SArd Biesheuvel #include <linux/efi.h> 14fd0528a2SArvind Sankar #include <linux/kernel.h> 1523d5b73fSArvind Sankar #include <linux/printk.h> /* For CONSOLE_LOGLEVEL_* */ 16f4f75ad5SArd Biesheuvel #include <asm/efi.h> 1780b1bfe1SArvind Sankar #include <asm/setup.h> 18f4f75ad5SArd Biesheuvel 19f4f75ad5SArd Biesheuvel #include "efistub.h" 20f4f75ad5SArd Biesheuvel 21980771f6SArd Biesheuvel bool efi_nochunk; 227c116db2SWill Deacon bool efi_nokaslr = !IS_ENABLED(CONFIG_RANDOMIZE_BASE); 23980771f6SArd Biesheuvel bool efi_noinitrd; 2423d5b73fSArvind Sankar int efi_loglevel = CONSOLE_LOGLEVEL_DEFAULT; 25980771f6SArd Biesheuvel bool efi_novamap; 26980771f6SArd Biesheuvel 2754439370SArvind Sankar static bool efi_nosoftreserve; 2854439370SArvind Sankar static bool efi_disable_pci_dma = IS_ENABLED(CONFIG_EFI_DISABLE_PCI_DMA); 2960f38de7SArd Biesheuvel 30b617c526SDan Williams bool __pure __efi_soft_reserve_enabled(void) 31b617c526SDan Williams { 32b617c526SDan Williams return !efi_nosoftreserve; 33b617c526SDan Williams } 3460f38de7SArd Biesheuvel 358c0a839cSHeinrich Schuchardt /** 368c0a839cSHeinrich Schuchardt * efi_char16_puts() - Write a UCS-2 encoded string to the console 378c0a839cSHeinrich Schuchardt * @str: UCS-2 encoded string 388c0a839cSHeinrich Schuchardt */ 39cb8c90a0SArvind Sankar void efi_char16_puts(efi_char16_t *str) 40cb8c90a0SArvind Sankar { 41cb8c90a0SArvind Sankar efi_call_proto(efi_table_attr(efi_system_table, con_out), 42cb8c90a0SArvind Sankar output_string, str); 43cb8c90a0SArvind Sankar } 44cb8c90a0SArvind Sankar 454b75bd36SArvind Sankar static 464b75bd36SArvind Sankar u32 utf8_to_utf32(const u8 **s8) 474b75bd36SArvind Sankar { 484b75bd36SArvind Sankar u32 c32; 494b75bd36SArvind Sankar u8 c0, cx; 504b75bd36SArvind Sankar size_t clen, i; 514b75bd36SArvind Sankar 524b75bd36SArvind Sankar c0 = cx = *(*s8)++; 534b75bd36SArvind Sankar /* 544b75bd36SArvind Sankar * The position of the most-significant 0 bit gives us the length of 554b75bd36SArvind Sankar * a multi-octet encoding. 564b75bd36SArvind Sankar */ 574b75bd36SArvind Sankar for (clen = 0; cx & 0x80; ++clen) 584b75bd36SArvind Sankar cx <<= 1; 594b75bd36SArvind Sankar /* 604b75bd36SArvind Sankar * If the 0 bit is in position 8, this is a valid single-octet 614b75bd36SArvind Sankar * encoding. If the 0 bit is in position 7 or positions 1-3, the 624b75bd36SArvind Sankar * encoding is invalid. 634b75bd36SArvind Sankar * In either case, we just return the first octet. 644b75bd36SArvind Sankar */ 654b75bd36SArvind Sankar if (clen < 2 || clen > 4) 664b75bd36SArvind Sankar return c0; 674b75bd36SArvind Sankar /* Get the bits from the first octet. */ 684b75bd36SArvind Sankar c32 = cx >> clen--; 694b75bd36SArvind Sankar for (i = 0; i < clen; ++i) { 704b75bd36SArvind Sankar /* Trailing octets must have 10 in most significant bits. */ 714b75bd36SArvind Sankar cx = (*s8)[i] ^ 0x80; 724b75bd36SArvind Sankar if (cx & 0xc0) 734b75bd36SArvind Sankar return c0; 744b75bd36SArvind Sankar c32 = (c32 << 6) | cx; 754b75bd36SArvind Sankar } 764b75bd36SArvind Sankar /* 774b75bd36SArvind Sankar * Check for validity: 784b75bd36SArvind Sankar * - The character must be in the Unicode range. 794b75bd36SArvind Sankar * - It must not be a surrogate. 804b75bd36SArvind Sankar * - It must be encoded using the correct number of octets. 814b75bd36SArvind Sankar */ 824b75bd36SArvind Sankar if (c32 > 0x10ffff || 834b75bd36SArvind Sankar (c32 & 0xf800) == 0xd800 || 844b75bd36SArvind Sankar clen != (c32 >= 0x80) + (c32 >= 0x800) + (c32 >= 0x10000)) 854b75bd36SArvind Sankar return c0; 864b75bd36SArvind Sankar *s8 += clen; 874b75bd36SArvind Sankar return c32; 884b75bd36SArvind Sankar } 894b75bd36SArvind Sankar 908c0a839cSHeinrich Schuchardt /** 918c0a839cSHeinrich Schuchardt * efi_puts() - Write a UTF-8 encoded string to the console 928c0a839cSHeinrich Schuchardt * @str: UTF-8 encoded string 938c0a839cSHeinrich Schuchardt */ 94cb8c90a0SArvind Sankar void efi_puts(const char *str) 95f4f75ad5SArd Biesheuvel { 96fd0528a2SArvind Sankar efi_char16_t buf[128]; 97fd0528a2SArvind Sankar size_t pos = 0, lim = ARRAY_SIZE(buf); 984b75bd36SArvind Sankar const u8 *s8 = (const u8 *)str; 994b75bd36SArvind Sankar u32 c32; 100f4f75ad5SArd Biesheuvel 1014b75bd36SArvind Sankar while (*s8) { 1024b75bd36SArvind Sankar if (*s8 == '\n') 103fd0528a2SArvind Sankar buf[pos++] = L'\r'; 1044b75bd36SArvind Sankar c32 = utf8_to_utf32(&s8); 1054b75bd36SArvind Sankar if (c32 < 0x10000) { 1064b75bd36SArvind Sankar /* Characters in plane 0 use a single word. */ 1074b75bd36SArvind Sankar buf[pos++] = c32; 1084b75bd36SArvind Sankar } else { 1094b75bd36SArvind Sankar /* 1104b75bd36SArvind Sankar * Characters in other planes encode into a surrogate 1114b75bd36SArvind Sankar * pair. 1124b75bd36SArvind Sankar */ 1134b75bd36SArvind Sankar buf[pos++] = (0xd800 - (0x10000 >> 10)) + (c32 >> 10); 1144b75bd36SArvind Sankar buf[pos++] = 0xdc00 + (c32 & 0x3ff); 1154b75bd36SArvind Sankar } 1164b75bd36SArvind Sankar if (*s8 == '\0' || pos >= lim - 2) { 117fd0528a2SArvind Sankar buf[pos] = L'\0'; 118fd0528a2SArvind Sankar efi_char16_puts(buf); 119fd0528a2SArvind Sankar pos = 0; 120fd0528a2SArvind Sankar } 121f4f75ad5SArd Biesheuvel } 122f4f75ad5SArd Biesheuvel } 123f4f75ad5SArd Biesheuvel 1248c0a839cSHeinrich Schuchardt /** 1258c0a839cSHeinrich Schuchardt * efi_printk() - Print a kernel message 1268c0a839cSHeinrich Schuchardt * @fmt: format string 1278c0a839cSHeinrich Schuchardt * 1288c0a839cSHeinrich Schuchardt * The first letter of the format string is used to determine the logging level 1298c0a839cSHeinrich Schuchardt * of the message. If the level is less then the current EFI logging level, the 1308c0a839cSHeinrich Schuchardt * message is suppressed. The message will be truncated to 255 bytes. 1318c0a839cSHeinrich Schuchardt * 1328c0a839cSHeinrich Schuchardt * Return: number of printed characters 1338c0a839cSHeinrich Schuchardt */ 1342c7d1e30SArvind Sankar int efi_printk(const char *fmt, ...) 1352c7d1e30SArvind Sankar { 1362c7d1e30SArvind Sankar char printf_buf[256]; 1372c7d1e30SArvind Sankar va_list args; 1382c7d1e30SArvind Sankar int printed; 13923d5b73fSArvind Sankar int loglevel = printk_get_level(fmt); 14023d5b73fSArvind Sankar 14123d5b73fSArvind Sankar switch (loglevel) { 14223d5b73fSArvind Sankar case '0' ... '9': 14323d5b73fSArvind Sankar loglevel -= '0'; 14423d5b73fSArvind Sankar break; 14523d5b73fSArvind Sankar default: 14623d5b73fSArvind Sankar /* 14723d5b73fSArvind Sankar * Use loglevel -1 for cases where we just want to print to 14823d5b73fSArvind Sankar * the screen. 14923d5b73fSArvind Sankar */ 15023d5b73fSArvind Sankar loglevel = -1; 15123d5b73fSArvind Sankar break; 15223d5b73fSArvind Sankar } 15323d5b73fSArvind Sankar 15423d5b73fSArvind Sankar if (loglevel >= efi_loglevel) 15523d5b73fSArvind Sankar return 0; 15623d5b73fSArvind Sankar 15723d5b73fSArvind Sankar if (loglevel >= 0) 15823d5b73fSArvind Sankar efi_puts("EFI stub: "); 15923d5b73fSArvind Sankar 16023d5b73fSArvind Sankar fmt = printk_skip_level(fmt); 1612c7d1e30SArvind Sankar 1622c7d1e30SArvind Sankar va_start(args, fmt); 1638fb331e1SArvind Sankar printed = vsnprintf(printf_buf, sizeof(printf_buf), fmt, args); 1642c7d1e30SArvind Sankar va_end(args); 1652c7d1e30SArvind Sankar 1662c7d1e30SArvind Sankar efi_puts(printf_buf); 1678fb331e1SArvind Sankar if (printed >= sizeof(printf_buf)) { 1688fb331e1SArvind Sankar efi_puts("[Message truncated]\n"); 1698fb331e1SArvind Sankar return -1; 1708fb331e1SArvind Sankar } 1712c7d1e30SArvind Sankar 1722c7d1e30SArvind Sankar return printed; 1732c7d1e30SArvind Sankar } 1742c7d1e30SArvind Sankar 1758c0a839cSHeinrich Schuchardt /** 1768c0a839cSHeinrich Schuchardt * efi_parse_options() - Parse EFI command line options 1778c0a839cSHeinrich Schuchardt * @cmdline: kernel command line 1788c0a839cSHeinrich Schuchardt * 1798c0a839cSHeinrich Schuchardt * Parse the ASCII string @cmdline for EFI options, denoted by the efi= 1805a17dae4SMatt Fleming * option, e.g. efi=nochunk. 1815a17dae4SMatt Fleming * 1825a17dae4SMatt Fleming * It should be noted that efi= is parsed in two very different 1835a17dae4SMatt Fleming * environments, first in the early boot environment of the EFI boot 1845a17dae4SMatt Fleming * stub, and subsequently during the kernel boot. 1858c0a839cSHeinrich Schuchardt * 1868c0a839cSHeinrich Schuchardt * Return: status code 1875a17dae4SMatt Fleming */ 18860f38de7SArd Biesheuvel efi_status_t efi_parse_options(char const *cmdline) 1895a17dae4SMatt Fleming { 190a37ca6a2SArvind Sankar size_t len; 19191d150c0SArd Biesheuvel efi_status_t status; 19291d150c0SArd Biesheuvel char *str, *buf; 1935a17dae4SMatt Fleming 194a37ca6a2SArvind Sankar if (!cmdline) 195a37ca6a2SArvind Sankar return EFI_SUCCESS; 196a37ca6a2SArvind Sankar 197a37ca6a2SArvind Sankar len = strlen(cmdline) + 1; 19891d150c0SArd Biesheuvel status = efi_bs_call(allocate_pool, EFI_LOADER_DATA, len, (void **)&buf); 19991d150c0SArd Biesheuvel if (status != EFI_SUCCESS) 20091d150c0SArd Biesheuvel return status; 20191d150c0SArd Biesheuvel 20291d150c0SArd Biesheuvel str = skip_spaces(memcpy(buf, cmdline, len)); 20391d150c0SArd Biesheuvel 20491d150c0SArd Biesheuvel while (*str) { 20591d150c0SArd Biesheuvel char *param, *val; 20691d150c0SArd Biesheuvel 20791d150c0SArd Biesheuvel str = next_arg(str, ¶m, &val); 2081fd9717dSArvind Sankar if (!val && !strcmp(param, "--")) 2091fd9717dSArvind Sankar break; 21091d150c0SArd Biesheuvel 21191d150c0SArd Biesheuvel if (!strcmp(param, "nokaslr")) { 2127d4e323dSArd Biesheuvel efi_nokaslr = true; 21391d150c0SArd Biesheuvel } else if (!strcmp(param, "quiet")) { 21423d5b73fSArvind Sankar efi_loglevel = CONSOLE_LOGLEVEL_QUIET; 21579d3219dSArd Biesheuvel } else if (!strcmp(param, "noinitrd")) { 21679d3219dSArd Biesheuvel efi_noinitrd = true; 21791d150c0SArd Biesheuvel } else if (!strcmp(param, "efi") && val) { 21891d150c0SArd Biesheuvel efi_nochunk = parse_option_str(val, "nochunk"); 21991d150c0SArd Biesheuvel efi_novamap = parse_option_str(val, "novamap"); 220eeff7d63SArd Biesheuvel 22191d150c0SArd Biesheuvel efi_nosoftreserve = IS_ENABLED(CONFIG_EFI_SOFT_RESERVE) && 22291d150c0SArd Biesheuvel parse_option_str(val, "nosoftreserve"); 2235a17dae4SMatt Fleming 22491d150c0SArd Biesheuvel if (parse_option_str(val, "disable_early_pci_dma")) 2254444f854SMatthew Garrett efi_disable_pci_dma = true; 22691d150c0SArd Biesheuvel if (parse_option_str(val, "no_disable_early_pci_dma")) 2274444f854SMatthew Garrett efi_disable_pci_dma = false; 22823d5b73fSArvind Sankar if (parse_option_str(val, "debug")) 22923d5b73fSArvind Sankar efi_loglevel = CONSOLE_LOGLEVEL_DEBUG; 230fffb6804SArvind Sankar } else if (!strcmp(param, "video") && 231fffb6804SArvind Sankar val && strstarts(val, "efifb:")) { 232fffb6804SArvind Sankar efi_parse_option_graphics(val + strlen("efifb:")); 2334444f854SMatthew Garrett } 2345a17dae4SMatt Fleming } 23591d150c0SArd Biesheuvel efi_bs_call(free_pool, buf); 2365a17dae4SMatt Fleming return EFI_SUCCESS; 2375a17dae4SMatt Fleming } 238f4f75ad5SArd Biesheuvel 239f4f75ad5SArd Biesheuvel /* 240f4f75ad5SArd Biesheuvel * Convert the unicode UEFI command line to ASCII to pass to kernel. 241f4f75ad5SArd Biesheuvel * Size of memory allocated return in *cmd_line_len. 242f4f75ad5SArd Biesheuvel * Returns NULL on error. 243f4f75ad5SArd Biesheuvel */ 24427cd5511SArd Biesheuvel char *efi_convert_cmdline(efi_loaded_image_t *image, int *cmd_line_len) 245f4f75ad5SArd Biesheuvel { 246f4f75ad5SArd Biesheuvel const u16 *s2; 247f4f75ad5SArd Biesheuvel unsigned long cmdline_addr = 0; 24804b24409SArvind Sankar int options_chars = efi_table_attr(image, load_options_size) / 2; 249f7b85b33SArd Biesheuvel const u16 *options = efi_table_attr(image, load_options); 25080b1bfe1SArvind Sankar int options_bytes = 0, safe_options_bytes = 0; /* UTF-8 bytes */ 25180b1bfe1SArvind Sankar bool in_quote = false; 252f4f75ad5SArd Biesheuvel efi_status_t status; 253f4f75ad5SArd Biesheuvel 254f4f75ad5SArd Biesheuvel if (options) { 255f4f75ad5SArd Biesheuvel s2 = options; 25680b1bfe1SArvind Sankar while (options_bytes < COMMAND_LINE_SIZE && options_chars--) { 25715c316bcSArvind Sankar u16 c = *s2++; 25815c316bcSArvind Sankar 25980b1bfe1SArvind Sankar if (c < 0x80) { 26015c316bcSArvind Sankar if (c == L'\0' || c == L'\n') 26115c316bcSArvind Sankar break; 26280b1bfe1SArvind Sankar if (c == L'"') 26380b1bfe1SArvind Sankar in_quote = !in_quote; 26480b1bfe1SArvind Sankar else if (!in_quote && isspace((char)c)) 26580b1bfe1SArvind Sankar safe_options_bytes = options_bytes; 26680b1bfe1SArvind Sankar 26780b1bfe1SArvind Sankar options_bytes++; 26880b1bfe1SArvind Sankar continue; 26980b1bfe1SArvind Sankar } 27080b1bfe1SArvind Sankar 27115c316bcSArvind Sankar /* 27215c316bcSArvind Sankar * Get the number of UTF-8 bytes corresponding to a 27315c316bcSArvind Sankar * UTF-16 character. 27415c316bcSArvind Sankar * The first part handles everything in the BMP. 27515c316bcSArvind Sankar */ 27680b1bfe1SArvind Sankar options_bytes += 2 + (c >= 0x800); 27715c316bcSArvind Sankar /* 27815c316bcSArvind Sankar * Add one more byte for valid surrogate pairs. Invalid 27915c316bcSArvind Sankar * surrogates will be replaced with 0xfffd and take up 28015c316bcSArvind Sankar * only 3 bytes. 28115c316bcSArvind Sankar */ 28215c316bcSArvind Sankar if ((c & 0xfc00) == 0xd800) { 28315c316bcSArvind Sankar /* 28415c316bcSArvind Sankar * If the very last word is a high surrogate, 28515c316bcSArvind Sankar * we must ignore it since we can't access the 28615c316bcSArvind Sankar * low surrogate. 28715c316bcSArvind Sankar */ 28804b24409SArvind Sankar if (!options_chars) { 28915c316bcSArvind Sankar options_bytes -= 3; 29015c316bcSArvind Sankar } else if ((*s2 & 0xfc00) == 0xdc00) { 29115c316bcSArvind Sankar options_bytes++; 29204b24409SArvind Sankar options_chars--; 29315c316bcSArvind Sankar s2++; 29415c316bcSArvind Sankar } 29515c316bcSArvind Sankar } 296f4f75ad5SArd Biesheuvel } 29780b1bfe1SArvind Sankar if (options_bytes >= COMMAND_LINE_SIZE) { 29880b1bfe1SArvind Sankar options_bytes = safe_options_bytes; 29980b1bfe1SArvind Sankar efi_err("Command line is too long: truncated to %d bytes\n", 30080b1bfe1SArvind Sankar options_bytes); 30180b1bfe1SArvind Sankar } 302f4f75ad5SArd Biesheuvel } 303f4f75ad5SArd Biesheuvel 304f4f75ad5SArd Biesheuvel options_bytes++; /* NUL termination */ 305f4f75ad5SArd Biesheuvel 30627cd5511SArd Biesheuvel status = efi_bs_call(allocate_pool, EFI_LOADER_DATA, options_bytes, 30727cd5511SArd Biesheuvel (void **)&cmdline_addr); 308f4f75ad5SArd Biesheuvel if (status != EFI_SUCCESS) 309f4f75ad5SArd Biesheuvel return NULL; 310f4f75ad5SArd Biesheuvel 31104b24409SArvind Sankar snprintf((char *)cmdline_addr, options_bytes, "%.*ls", 31204b24409SArvind Sankar options_bytes - 1, options); 313f4f75ad5SArd Biesheuvel 314f4f75ad5SArd Biesheuvel *cmd_line_len = options_bytes; 315f4f75ad5SArd Biesheuvel return (char *)cmdline_addr; 316f4f75ad5SArd Biesheuvel } 317fc07716bSJeffrey Hugo 3188c0a839cSHeinrich Schuchardt /** 3198c0a839cSHeinrich Schuchardt * efi_exit_boot_services() - Exit boot services 3208c0a839cSHeinrich Schuchardt * @handle: handle of the exiting image 3218c0a839cSHeinrich Schuchardt * @map: pointer to receive the memory map 3228c0a839cSHeinrich Schuchardt * @priv: argument to be passed to @priv_func 3238c0a839cSHeinrich Schuchardt * @priv_func: function to process the memory map before exiting boot services 3248c0a839cSHeinrich Schuchardt * 325fc07716bSJeffrey Hugo * Handle calling ExitBootServices according to the requirements set out by the 326fc07716bSJeffrey Hugo * spec. Obtains the current memory map, and returns that info after calling 327fc07716bSJeffrey Hugo * ExitBootServices. The client must specify a function to perform any 328fc07716bSJeffrey Hugo * processing of the memory map data prior to ExitBootServices. A client 329fc07716bSJeffrey Hugo * specific structure may be passed to the function via priv. The client 330fc07716bSJeffrey Hugo * function may be called multiple times. 3318c0a839cSHeinrich Schuchardt * 3328c0a839cSHeinrich Schuchardt * Return: status code 333fc07716bSJeffrey Hugo */ 334cd33a5c1SArd Biesheuvel efi_status_t efi_exit_boot_services(void *handle, 335fc07716bSJeffrey Hugo struct efi_boot_memmap *map, 336fc07716bSJeffrey Hugo void *priv, 337fc07716bSJeffrey Hugo efi_exit_boot_map_processing priv_func) 338fc07716bSJeffrey Hugo { 339fc07716bSJeffrey Hugo efi_status_t status; 340fc07716bSJeffrey Hugo 341cd33a5c1SArd Biesheuvel status = efi_get_memory_map(map); 342fc07716bSJeffrey Hugo 343fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 344fc07716bSJeffrey Hugo goto fail; 345fc07716bSJeffrey Hugo 346cd33a5c1SArd Biesheuvel status = priv_func(map, priv); 347fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 348fc07716bSJeffrey Hugo goto free_map; 349fc07716bSJeffrey Hugo 3504444f854SMatthew Garrett if (efi_disable_pci_dma) 3514444f854SMatthew Garrett efi_pci_disable_bridge_busmaster(); 3524444f854SMatthew Garrett 353966291f6SArd Biesheuvel status = efi_bs_call(exit_boot_services, handle, *map->key_ptr); 354fc07716bSJeffrey Hugo 355fc07716bSJeffrey Hugo if (status == EFI_INVALID_PARAMETER) { 356fc07716bSJeffrey Hugo /* 357fc07716bSJeffrey Hugo * The memory map changed between efi_get_memory_map() and 358fc07716bSJeffrey Hugo * exit_boot_services(). Per the UEFI Spec v2.6, Section 6.4: 359fc07716bSJeffrey Hugo * EFI_BOOT_SERVICES.ExitBootServices we need to get the 360fc07716bSJeffrey Hugo * updated map, and try again. The spec implies one retry 361fc07716bSJeffrey Hugo * should be sufficent, which is confirmed against the EDK2 362fc07716bSJeffrey Hugo * implementation. Per the spec, we can only invoke 363fc07716bSJeffrey Hugo * get_memory_map() and exit_boot_services() - we cannot alloc 364fc07716bSJeffrey Hugo * so efi_get_memory_map() cannot be used, and we must reuse 365fc07716bSJeffrey Hugo * the buffer. For all practical purposes, the headroom in the 366fc07716bSJeffrey Hugo * buffer should account for any changes in the map so the call 367fc07716bSJeffrey Hugo * to get_memory_map() is expected to succeed here. 368fc07716bSJeffrey Hugo */ 369fc07716bSJeffrey Hugo *map->map_size = *map->buff_size; 370966291f6SArd Biesheuvel status = efi_bs_call(get_memory_map, 371fc07716bSJeffrey Hugo map->map_size, 372fc07716bSJeffrey Hugo *map->map, 373fc07716bSJeffrey Hugo map->key_ptr, 374fc07716bSJeffrey Hugo map->desc_size, 375fc07716bSJeffrey Hugo map->desc_ver); 376fc07716bSJeffrey Hugo 377fc07716bSJeffrey Hugo /* exit_boot_services() was called, thus cannot free */ 378fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 379fc07716bSJeffrey Hugo goto fail; 380fc07716bSJeffrey Hugo 381cd33a5c1SArd Biesheuvel status = priv_func(map, priv); 382fc07716bSJeffrey Hugo /* exit_boot_services() was called, thus cannot free */ 383fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 384fc07716bSJeffrey Hugo goto fail; 385fc07716bSJeffrey Hugo 386966291f6SArd Biesheuvel status = efi_bs_call(exit_boot_services, handle, *map->key_ptr); 387fc07716bSJeffrey Hugo } 388fc07716bSJeffrey Hugo 389fc07716bSJeffrey Hugo /* exit_boot_services() was called, thus cannot free */ 390fc07716bSJeffrey Hugo if (status != EFI_SUCCESS) 391fc07716bSJeffrey Hugo goto fail; 392fc07716bSJeffrey Hugo 393fc07716bSJeffrey Hugo return EFI_SUCCESS; 394fc07716bSJeffrey Hugo 395fc07716bSJeffrey Hugo free_map: 396966291f6SArd Biesheuvel efi_bs_call(free_pool, *map->map); 397fc07716bSJeffrey Hugo fail: 398fc07716bSJeffrey Hugo return status; 399fc07716bSJeffrey Hugo } 40082d736acSMatthew Garrett 4018c0a839cSHeinrich Schuchardt /** 4028c0a839cSHeinrich Schuchardt * get_efi_config_table() - retrieve UEFI configuration table 4038c0a839cSHeinrich Schuchardt * @guid: GUID of the configuration table to be retrieved 4048c0a839cSHeinrich Schuchardt * Return: pointer to the configuration table or NULL 4058c0a839cSHeinrich Schuchardt */ 406cd33a5c1SArd Biesheuvel void *get_efi_config_table(efi_guid_t guid) 40782d736acSMatthew Garrett { 408ccc27ae7SArd Biesheuvel unsigned long tables = efi_table_attr(efi_system_table, tables); 409ccc27ae7SArd Biesheuvel int nr_tables = efi_table_attr(efi_system_table, nr_tables); 410f958efe9SArd Biesheuvel int i; 411f958efe9SArd Biesheuvel 412f958efe9SArd Biesheuvel for (i = 0; i < nr_tables; i++) { 413f958efe9SArd Biesheuvel efi_config_table_t *t = (void *)tables; 414f958efe9SArd Biesheuvel 415f958efe9SArd Biesheuvel if (efi_guidcmp(t->guid, guid) == 0) 41699ea8b1dSArd Biesheuvel return efi_table_attr(t, table); 417f958efe9SArd Biesheuvel 418f958efe9SArd Biesheuvel tables += efi_is_native() ? sizeof(efi_config_table_t) 419f958efe9SArd Biesheuvel : sizeof(efi_config_table_32_t); 420f958efe9SArd Biesheuvel } 421f958efe9SArd Biesheuvel return NULL; 42282d736acSMatthew Garrett } 423dc29da14SArd Biesheuvel 424ec93fc37SArd Biesheuvel /* 425ec93fc37SArd Biesheuvel * The LINUX_EFI_INITRD_MEDIA_GUID vendor media device path below provides a way 426ec93fc37SArd Biesheuvel * for the firmware or bootloader to expose the initrd data directly to the stub 427ec93fc37SArd Biesheuvel * via the trivial LoadFile2 protocol, which is defined in the UEFI spec, and is 428ec93fc37SArd Biesheuvel * very easy to implement. It is a simple Linux initrd specific conduit between 429ec93fc37SArd Biesheuvel * kernel and firmware, allowing us to put the EFI stub (being part of the 430ec93fc37SArd Biesheuvel * kernel) in charge of where and when to load the initrd, while leaving it up 431ec93fc37SArd Biesheuvel * to the firmware to decide whether it needs to expose its filesystem hierarchy 432ec93fc37SArd Biesheuvel * via EFI protocols. 433ec93fc37SArd Biesheuvel */ 434ec93fc37SArd Biesheuvel static const struct { 435ec93fc37SArd Biesheuvel struct efi_vendor_dev_path vendor; 436ec93fc37SArd Biesheuvel struct efi_generic_dev_path end; 437ec93fc37SArd Biesheuvel } __packed initrd_dev_path = { 438ec93fc37SArd Biesheuvel { 439ec93fc37SArd Biesheuvel { 440ec93fc37SArd Biesheuvel EFI_DEV_MEDIA, 441ec93fc37SArd Biesheuvel EFI_DEV_MEDIA_VENDOR, 442ec93fc37SArd Biesheuvel sizeof(struct efi_vendor_dev_path), 443ec93fc37SArd Biesheuvel }, 444ec93fc37SArd Biesheuvel LINUX_EFI_INITRD_MEDIA_GUID 445ec93fc37SArd Biesheuvel }, { 446ec93fc37SArd Biesheuvel EFI_DEV_END_PATH, 447ec93fc37SArd Biesheuvel EFI_DEV_END_ENTIRE, 448ec93fc37SArd Biesheuvel sizeof(struct efi_generic_dev_path) 449ec93fc37SArd Biesheuvel } 450ec93fc37SArd Biesheuvel }; 451ec93fc37SArd Biesheuvel 452ec93fc37SArd Biesheuvel /** 4538c0a839cSHeinrich Schuchardt * efi_load_initrd_dev_path() - load the initrd from the Linux initrd device path 454ec93fc37SArd Biesheuvel * @load_addr: pointer to store the address where the initrd was loaded 455ec93fc37SArd Biesheuvel * @load_size: pointer to store the size of the loaded initrd 456ec93fc37SArd Biesheuvel * @max: upper limit for the initrd memory allocation 4578c0a839cSHeinrich Schuchardt * 4588c0a839cSHeinrich Schuchardt * Return: 4598c0a839cSHeinrich Schuchardt * * %EFI_SUCCESS if the initrd was loaded successfully, in which 460ec93fc37SArd Biesheuvel * case @load_addr and @load_size are assigned accordingly 4618c0a839cSHeinrich Schuchardt * * %EFI_NOT_FOUND if no LoadFile2 protocol exists on the initrd device path 4628c0a839cSHeinrich Schuchardt * * %EFI_INVALID_PARAMETER if load_addr == NULL or load_size == NULL 4638c0a839cSHeinrich Schuchardt * * %EFI_OUT_OF_RESOURCES if memory allocation failed 4648c0a839cSHeinrich Schuchardt * * %EFI_LOAD_ERROR in all other cases 465ec93fc37SArd Biesheuvel */ 466f61900fdSArvind Sankar static 467ec93fc37SArd Biesheuvel efi_status_t efi_load_initrd_dev_path(unsigned long *load_addr, 468ec93fc37SArd Biesheuvel unsigned long *load_size, 469ec93fc37SArd Biesheuvel unsigned long max) 470ec93fc37SArd Biesheuvel { 471ec93fc37SArd Biesheuvel efi_guid_t lf2_proto_guid = EFI_LOAD_FILE2_PROTOCOL_GUID; 472ec93fc37SArd Biesheuvel efi_device_path_protocol_t *dp; 473ec93fc37SArd Biesheuvel efi_load_file2_protocol_t *lf2; 474ec93fc37SArd Biesheuvel unsigned long initrd_addr; 475ec93fc37SArd Biesheuvel unsigned long initrd_size; 476ec93fc37SArd Biesheuvel efi_handle_t handle; 477ec93fc37SArd Biesheuvel efi_status_t status; 478ec93fc37SArd Biesheuvel 479ec93fc37SArd Biesheuvel dp = (efi_device_path_protocol_t *)&initrd_dev_path; 480ec93fc37SArd Biesheuvel status = efi_bs_call(locate_device_path, &lf2_proto_guid, &dp, &handle); 481ec93fc37SArd Biesheuvel if (status != EFI_SUCCESS) 482ec93fc37SArd Biesheuvel return status; 483ec93fc37SArd Biesheuvel 484ec93fc37SArd Biesheuvel status = efi_bs_call(handle_protocol, handle, &lf2_proto_guid, 485ec93fc37SArd Biesheuvel (void **)&lf2); 486ec93fc37SArd Biesheuvel if (status != EFI_SUCCESS) 487ec93fc37SArd Biesheuvel return status; 488ec93fc37SArd Biesheuvel 489ec93fc37SArd Biesheuvel status = efi_call_proto(lf2, load_file, dp, false, &initrd_size, NULL); 490ec93fc37SArd Biesheuvel if (status != EFI_BUFFER_TOO_SMALL) 491ec93fc37SArd Biesheuvel return EFI_LOAD_ERROR; 492ec93fc37SArd Biesheuvel 493ec93fc37SArd Biesheuvel status = efi_allocate_pages(initrd_size, &initrd_addr, max); 494ec93fc37SArd Biesheuvel if (status != EFI_SUCCESS) 495ec93fc37SArd Biesheuvel return status; 496ec93fc37SArd Biesheuvel 497ec93fc37SArd Biesheuvel status = efi_call_proto(lf2, load_file, dp, false, &initrd_size, 498ec93fc37SArd Biesheuvel (void *)initrd_addr); 499ec93fc37SArd Biesheuvel if (status != EFI_SUCCESS) { 500ec93fc37SArd Biesheuvel efi_free(initrd_size, initrd_addr); 501ec93fc37SArd Biesheuvel return EFI_LOAD_ERROR; 502ec93fc37SArd Biesheuvel } 503ec93fc37SArd Biesheuvel 504ec93fc37SArd Biesheuvel *load_addr = initrd_addr; 505ec93fc37SArd Biesheuvel *load_size = initrd_size; 506ec93fc37SArd Biesheuvel return EFI_SUCCESS; 507ec93fc37SArd Biesheuvel } 508f61900fdSArvind Sankar 509f61900fdSArvind Sankar static 510f61900fdSArvind Sankar efi_status_t efi_load_initrd_cmdline(efi_loaded_image_t *image, 511f61900fdSArvind Sankar unsigned long *load_addr, 512f61900fdSArvind Sankar unsigned long *load_size, 513f61900fdSArvind Sankar unsigned long soft_limit, 514f61900fdSArvind Sankar unsigned long hard_limit) 515f61900fdSArvind Sankar { 516f61900fdSArvind Sankar if (!IS_ENABLED(CONFIG_EFI_GENERIC_STUB_INITRD_CMDLINE_LOADER) || 517f61900fdSArvind Sankar (IS_ENABLED(CONFIG_X86) && (!efi_is_native() || image == NULL))) { 518f61900fdSArvind Sankar *load_addr = *load_size = 0; 519f61900fdSArvind Sankar return EFI_SUCCESS; 520f61900fdSArvind Sankar } 521f61900fdSArvind Sankar 522f61900fdSArvind Sankar return handle_cmdline_files(image, L"initrd=", sizeof(L"initrd=") - 2, 523f61900fdSArvind Sankar soft_limit, hard_limit, 524f61900fdSArvind Sankar load_addr, load_size); 525f61900fdSArvind Sankar } 526f61900fdSArvind Sankar 5278c0a839cSHeinrich Schuchardt /** 5288c0a839cSHeinrich Schuchardt * efi_load_initrd() - Load initial RAM disk 5298c0a839cSHeinrich Schuchardt * @image: EFI loaded image protocol 5308c0a839cSHeinrich Schuchardt * @load_addr: pointer to loaded initrd 5318c0a839cSHeinrich Schuchardt * @load_size: size of loaded initrd 5328c0a839cSHeinrich Schuchardt * @soft_limit: preferred size of allocated memory for loading the initrd 5338c0a839cSHeinrich Schuchardt * @hard_limit: minimum size of allocated memory 5348c0a839cSHeinrich Schuchardt * 5358c0a839cSHeinrich Schuchardt * Return: status code 5368c0a839cSHeinrich Schuchardt */ 537f61900fdSArvind Sankar efi_status_t efi_load_initrd(efi_loaded_image_t *image, 538f61900fdSArvind Sankar unsigned long *load_addr, 539f61900fdSArvind Sankar unsigned long *load_size, 540f61900fdSArvind Sankar unsigned long soft_limit, 541f61900fdSArvind Sankar unsigned long hard_limit) 542f61900fdSArvind Sankar { 543f61900fdSArvind Sankar efi_status_t status; 544f61900fdSArvind Sankar 545f61900fdSArvind Sankar if (!load_addr || !load_size) 546f61900fdSArvind Sankar return EFI_INVALID_PARAMETER; 547f61900fdSArvind Sankar 548f61900fdSArvind Sankar status = efi_load_initrd_dev_path(load_addr, load_size, hard_limit); 549f61900fdSArvind Sankar if (status == EFI_SUCCESS) { 550f61900fdSArvind Sankar efi_info("Loaded initrd from LINUX_EFI_INITRD_MEDIA_GUID device path\n"); 551f61900fdSArvind Sankar } else if (status == EFI_NOT_FOUND) { 552f61900fdSArvind Sankar status = efi_load_initrd_cmdline(image, load_addr, load_size, 553f61900fdSArvind Sankar soft_limit, hard_limit); 554f61900fdSArvind Sankar if (status == EFI_SUCCESS && *load_size > 0) 555f61900fdSArvind Sankar efi_info("Loaded initrd from command line option\n"); 556f61900fdSArvind Sankar } 557f61900fdSArvind Sankar 558f61900fdSArvind Sankar return status; 559f61900fdSArvind Sankar } 56014c574f3SArvind Sankar 5618c0a839cSHeinrich Schuchardt /** 5628c0a839cSHeinrich Schuchardt * efi_wait_for_key() - Wait for key stroke 5638c0a839cSHeinrich Schuchardt * @usec: number of microseconds to wait for key stroke 5648c0a839cSHeinrich Schuchardt * @key: key entered 5658c0a839cSHeinrich Schuchardt * 5668c0a839cSHeinrich Schuchardt * Wait for up to @usec microseconds for a key stroke. 5678c0a839cSHeinrich Schuchardt * 5688c0a839cSHeinrich Schuchardt * Return: status code, EFI_SUCCESS if key received 5698c0a839cSHeinrich Schuchardt */ 57014c574f3SArvind Sankar efi_status_t efi_wait_for_key(unsigned long usec, efi_input_key_t *key) 57114c574f3SArvind Sankar { 57214c574f3SArvind Sankar efi_event_t events[2], timer; 57314c574f3SArvind Sankar unsigned long index; 57414c574f3SArvind Sankar efi_simple_text_input_protocol_t *con_in; 57514c574f3SArvind Sankar efi_status_t status; 57614c574f3SArvind Sankar 57714c574f3SArvind Sankar con_in = efi_table_attr(efi_system_table, con_in); 57814c574f3SArvind Sankar if (!con_in) 57914c574f3SArvind Sankar return EFI_UNSUPPORTED; 58014c574f3SArvind Sankar efi_set_event_at(events, 0, efi_table_attr(con_in, wait_for_key)); 58114c574f3SArvind Sankar 58214c574f3SArvind Sankar status = efi_bs_call(create_event, EFI_EVT_TIMER, 0, NULL, NULL, &timer); 58314c574f3SArvind Sankar if (status != EFI_SUCCESS) 58414c574f3SArvind Sankar return status; 58514c574f3SArvind Sankar 58614c574f3SArvind Sankar status = efi_bs_call(set_timer, timer, EfiTimerRelative, 58714c574f3SArvind Sankar EFI_100NSEC_PER_USEC * usec); 58814c574f3SArvind Sankar if (status != EFI_SUCCESS) 58914c574f3SArvind Sankar return status; 59014c574f3SArvind Sankar efi_set_event_at(events, 1, timer); 59114c574f3SArvind Sankar 59214c574f3SArvind Sankar status = efi_bs_call(wait_for_event, 2, events, &index); 59314c574f3SArvind Sankar if (status == EFI_SUCCESS) { 59414c574f3SArvind Sankar if (index == 0) 59514c574f3SArvind Sankar status = efi_call_proto(con_in, read_keystroke, key); 59614c574f3SArvind Sankar else 59714c574f3SArvind Sankar status = EFI_TIMEOUT; 59814c574f3SArvind Sankar } 59914c574f3SArvind Sankar 60014c574f3SArvind Sankar efi_bs_call(close_event, timer); 60114c574f3SArvind Sankar 60214c574f3SArvind Sankar return status; 60314c574f3SArvind Sankar } 604