/* * Convert the unicode UEFI command line to ASCII to pass to kernel. * Size of memory allocated return in *cmd_line_len. * Returns NULL on error. */ static char *efi_convert_cmdline_to_ascii(efi_system_table_t *sys_table_arg, efi_loaded_image_t *image, int *cmd_line_len) { u16 *s2; u8 *s1 = NULL; unsigned long cmdline_addr = 0; int load_options_size = image->load_options_size / 2; /* ASCII */ void *options = image->load_options; int options_size = 0; efi_status_t status; int i; u16 zero = 0; if (options) { s2 = options; while (*s2 && *s2 != '\n' && options_size < load_options_size) { s2++; options_size++; } } if (options_size == 0) { /* No command line options, so return empty string*/ options_size = 1; options = &zero; } options_size++; /* NUL termination */ #ifdef CONFIG_ARM /* * For ARM, allocate at a high address to avoid reserved * regions at low addresses that we don't know the specfics of * at the time we are processing the command line. */ status = efi_high_alloc(sys_table_arg, options_size, 0, &cmdline_addr, 0xfffff000); #else status = efi_low_alloc(sys_table_arg, options_size, 0, &cmdline_addr); #endif if (status != EFI_SUCCESS) return NULL; s1 = (u8 *)cmdline_addr; s2 = (u16 *)options; for (i = 0; i < options_size - 1; i++) *s1++ = *s2++; *s1 = '\0'; *cmd_line_len = options_size; return (char *)cmdline_addr; }
efi_status_t allocate_new_fdt_and_exit_boot(efi_system_table_t *sys_table, void *handle, unsigned long *new_fdt_addr, unsigned long max_addr, u64 initrd_addr, u64 initrd_size, char *cmdline_ptr, unsigned long fdt_addr, unsigned long fdt_size) { unsigned long map_size, desc_size; u32 desc_ver; unsigned long mmap_key; efi_memory_desc_t *memory_map; unsigned long new_fdt_size; efi_status_t status; /* * Estimate size of new FDT, and allocate memory for it. We * will allocate a bigger buffer if this ends up being too * small, so a rough guess is OK here. */ new_fdt_size = fdt_size + EFI_PAGE_SIZE; while (1) { status = efi_high_alloc(sys_table, new_fdt_size, EFI_FDT_ALIGN, new_fdt_addr, max_addr); if (status != EFI_SUCCESS) { pr_efi_err(sys_table, "Unable to allocate memory for new device tree.\n"); goto fail; } /* * Now that we have done our final memory allocation (and free) * we can get the memory map key needed for * exit_boot_services(). */ status = efi_get_memory_map(sys_table, &memory_map, &map_size, &desc_size, &desc_ver, &mmap_key); if (status != EFI_SUCCESS) goto fail_free_new_fdt; status = update_fdt(sys_table, (void *)fdt_addr, fdt_size, (void *)*new_fdt_addr, new_fdt_size, cmdline_ptr, initrd_addr, initrd_size, memory_map, map_size, desc_size, desc_ver); /* Succeeding the first time is the expected case. */ if (status == EFI_SUCCESS) break; if (status == EFI_BUFFER_TOO_SMALL) { /* * We need to allocate more space for the new * device tree, so free existing buffer that is * too small. Also free memory map, as we will need * to get new one that reflects the free/alloc we do * on the device tree buffer. */ efi_free(sys_table, new_fdt_size, *new_fdt_addr); sys_table->boottime->free_pool(memory_map); new_fdt_size += EFI_PAGE_SIZE; } else { pr_efi_err(sys_table, "Unable to constuct new device tree.\n"); goto fail_free_mmap; } } /* Now we are ready to exit_boot_services.*/ status = sys_table->boottime->exit_boot_services(handle, mmap_key); if (status == EFI_SUCCESS) return status; pr_efi_err(sys_table, "Exit boot services failed.\n"); fail_free_mmap: sys_table->boottime->free_pool(memory_map); fail_free_new_fdt: efi_free(sys_table, new_fdt_size, *new_fdt_addr); fail: return EFI_LOAD_ERROR; }
/* * Check the cmdline for a LILO-style file= arguments. * * We only support loading a file from the same filesystem as * the kernel image. */ static efi_status_t handle_cmdline_files(efi_system_table_t *sys_table_arg, efi_loaded_image_t *image, char *cmd_line, char *option_string, unsigned long max_addr, unsigned long *load_addr, unsigned long *load_size) { struct file_info *files; unsigned long file_addr; u64 file_size_total; efi_file_handle_t *fh; efi_status_t status; int nr_files; char *str; int i, j, k; file_addr = 0; file_size_total = 0; str = cmd_line; j = 0; /* See close_handles */ if (!load_addr || !load_size) return EFI_INVALID_PARAMETER; *load_addr = 0; *load_size = 0; if (!str || !*str) return EFI_SUCCESS; for (nr_files = 0; *str; nr_files++) { str = strstr(str, option_string); if (!str) break; str += strlen(option_string); /* Skip any leading slashes */ while (*str == '/' || *str == '\\') str++; while (*str && *str != ' ' && *str != '\n') str++; } if (!nr_files) return EFI_SUCCESS; status = efi_call_early(allocate_pool, EFI_LOADER_DATA, nr_files * sizeof(*files), (void **)&files); if (status != EFI_SUCCESS) { efi_printk(sys_table_arg, "Failed to alloc mem for file handle list\n"); goto fail; } str = cmd_line; for (i = 0; i < nr_files; i++) { struct file_info *file; efi_char16_t filename_16[256]; efi_char16_t *p; str = strstr(str, option_string); if (!str) break; str += strlen(option_string); file = &files[i]; p = filename_16; /* Skip any leading slashes */ while (*str == '/' || *str == '\\') str++; while (*str && *str != ' ' && *str != '\n') { if ((u8 *)p >= (u8 *)filename_16 + sizeof(filename_16)) break; if (*str == '/') { *p++ = '\\'; str++; } else { *p++ = *str++; } } *p = '\0'; /* Only open the volume once. */ if (!i) { status = efi_open_volume(sys_table_arg, image, (void **)&fh); if (status != EFI_SUCCESS) goto free_files; } status = efi_file_size(sys_table_arg, fh, filename_16, (void **)&file->handle, &file->size); if (status != EFI_SUCCESS) goto close_handles; file_size_total += file->size; } if (file_size_total) { unsigned long addr; /* * Multiple files need to be at consecutive addresses in memory, * so allocate enough memory for all the files. This is used * for loading multiple files. */ status = efi_high_alloc(sys_table_arg, file_size_total, 0x1000, &file_addr, max_addr); if (status != EFI_SUCCESS) { efi_printk(sys_table_arg, "Failed to alloc highmem for files\n"); goto close_handles; } /* We've run out of free low memory. */ if (file_addr > max_addr) { efi_printk(sys_table_arg, "We've run out of free low memory\n"); status = EFI_INVALID_PARAMETER; goto free_file_total; } addr = file_addr; for (j = 0; j < nr_files; j++) { unsigned long size; size = files[j].size; while (size) { unsigned long chunksize; if (size > EFI_READ_CHUNK_SIZE) chunksize = EFI_READ_CHUNK_SIZE; else chunksize = size; status = efi_file_read(fh, files[j].handle, &chunksize, (void *)addr); if (status != EFI_SUCCESS) { efi_printk(sys_table_arg, "Failed to read file\n"); goto free_file_total; } addr += chunksize; size -= chunksize; } efi_file_close(fh, files[j].handle); } } efi_call_early(free_pool, files); *load_addr = file_addr; *load_size = file_size_total; return status; free_file_total: efi_free(sys_table_arg, file_size_total, file_addr); close_handles: for (k = j; k < i; k++) efi_file_close(fh, files[k].handle); free_files: efi_call_early(free_pool, files); fail: *load_addr = 0; *load_size = 0; return status; }
efi_status_t allocate_new_fdt_and_exit_boot(efi_system_table_t *sys_table, void *handle, unsigned long *new_fdt_addr, unsigned long max_addr, u64 initrd_addr, u64 initrd_size, char *cmdline_ptr, unsigned long fdt_addr, unsigned long fdt_size) { unsigned long map_size, desc_size, buff_size; u32 desc_ver; unsigned long mmap_key; efi_memory_desc_t *memory_map, *runtime_map; unsigned long new_fdt_size; efi_status_t status; int runtime_entry_count = 0; struct efi_boot_memmap map; struct exit_boot_struct priv; map.map = &runtime_map; map.map_size = &map_size; map.desc_size = &desc_size; map.desc_ver = &desc_ver; map.key_ptr = &mmap_key; map.buff_size = &buff_size; /* * Get a copy of the current memory map that we will use to prepare * the input for SetVirtualAddressMap(). We don't have to worry about * subsequent allocations adding entries, since they could not affect * the number of EFI_MEMORY_RUNTIME regions. */ status = efi_get_memory_map(sys_table, &map); if (status != EFI_SUCCESS) { pr_efi_err(sys_table, "Unable to retrieve UEFI memory map.\n"); return status; } pr_efi(sys_table, "Exiting boot services and installing virtual address map...\n"); map.map = &memory_map; /* * Estimate size of new FDT, and allocate memory for it. We * will allocate a bigger buffer if this ends up being too * small, so a rough guess is OK here. */ new_fdt_size = fdt_size + EFI_PAGE_SIZE; while (1) { status = efi_high_alloc(sys_table, new_fdt_size, EFI_FDT_ALIGN, new_fdt_addr, max_addr); if (status != EFI_SUCCESS) { pr_efi_err(sys_table, "Unable to allocate memory for new device tree.\n"); goto fail; } /* * Now that we have done our final memory allocation (and free) * we can get the memory map key needed for * exit_boot_services(). */ status = efi_get_memory_map(sys_table, &map); if (status != EFI_SUCCESS) goto fail_free_new_fdt; status = update_fdt(sys_table, (void *)fdt_addr, fdt_size, (void *)*new_fdt_addr, new_fdt_size, cmdline_ptr, initrd_addr, initrd_size, memory_map, map_size, desc_size, desc_ver); /* Succeeding the first time is the expected case. */ if (status == EFI_SUCCESS) break; if (status == EFI_BUFFER_TOO_SMALL) { /* * We need to allocate more space for the new * device tree, so free existing buffer that is * too small. Also free memory map, as we will need * to get new one that reflects the free/alloc we do * on the device tree buffer. */ efi_free(sys_table, new_fdt_size, *new_fdt_addr); sys_table->boottime->free_pool(memory_map); new_fdt_size += EFI_PAGE_SIZE; } else { pr_efi_err(sys_table, "Unable to construct new device tree.\n"); goto fail_free_mmap; } } sys_table->boottime->free_pool(memory_map); priv.runtime_map = runtime_map; priv.runtime_entry_count = &runtime_entry_count; status = efi_exit_boot_services(sys_table, handle, &map, &priv, exit_boot_func); if (status == EFI_SUCCESS) { efi_set_virtual_address_map_t *svam; /* Install the new virtual address map */ svam = sys_table->runtime->set_virtual_address_map; status = svam(runtime_entry_count * desc_size, desc_size, desc_ver, runtime_map); /* * We are beyond the point of no return here, so if the call to * SetVirtualAddressMap() failed, we need to signal that to the * incoming kernel but proceed normally otherwise. */ if (status != EFI_SUCCESS) { int l; /* * Set the virtual address field of all * EFI_MEMORY_RUNTIME entries to 0. This will signal * the incoming kernel that no virtual translation has * been installed. */ for (l = 0; l < map_size; l += desc_size) { efi_memory_desc_t *p = (void *)memory_map + l; if (p->attribute & EFI_MEMORY_RUNTIME) p->virt_addr = 0; } } return EFI_SUCCESS; } pr_efi_err(sys_table, "Exit boot services failed.\n"); fail_free_mmap: sys_table->boottime->free_pool(memory_map); fail_free_new_fdt: efi_free(sys_table, new_fdt_size, *new_fdt_addr); fail: sys_table->boottime->free_pool(runtime_map); return EFI_LOAD_ERROR; }