Heinrich, On Sat, Feb 16, 2019 at 08:50:43PM +0100, Heinrich Schuchardt wrote: > All code and data sections of PE images are already in the correct relative > location when loaded into memory. There is not need to copy them once > again.
While I'm not very familiar with how PE image is created (in EDK2), what I understand in Alex's code is * All the code and data are located starting 0x0 (in virtual space) * Sections in PE image may not be sorted in ascending order * There may be some gaps (more than one page) between sections, probably, due to alignment requirements or BSS Do you say that those assumptions are no longer correct? Thanks, -Takahiro Akashi > Signed-off-by: Heinrich Schuchardt <[email protected]> > --- > lib/efi_loader/efi_image_loader.c | 64 ++++++------------------------- > 1 file changed, 11 insertions(+), 53 deletions(-) > > diff --git a/lib/efi_loader/efi_image_loader.c > b/lib/efi_loader/efi_image_loader.c > index fe66e7b9ffe..6146acc6b6f 100644 > --- a/lib/efi_loader/efi_image_loader.c > +++ b/lib/efi_loader/efi_image_loader.c > @@ -2,7 +2,7 @@ > /* > * EFI image loader > * > - * based partly on wine code > + * based partly on Wine code > * > * Copyright (c) 2016 Alexander Graf > */ > @@ -219,16 +219,12 @@ efi_status_t efi_load_pe(struct efi_loaded_image_obj > *handle, void *efi, > { > IMAGE_NT_HEADERS32 *nt; > IMAGE_DOS_HEADER *dos; > - IMAGE_SECTION_HEADER *sections; > - int num_sections; > - void *efi_reloc; > int i; > const IMAGE_BASE_RELOCATION *rel; > unsigned long rel_size; > int rel_idx = IMAGE_DIRECTORY_ENTRY_BASERELOC; > uint64_t image_base; > uint64_t image_size; > - unsigned long virt_size = 0; > int supported = 0; > > dos = efi; > @@ -255,85 +251,47 @@ efi_status_t efi_load_pe(struct efi_loaded_image_obj > *handle, void *efi, > return EFI_LOAD_ERROR; > } > > - /* Calculate upper virtual address boundary */ > - num_sections = nt->FileHeader.NumberOfSections; > - sections = (void *)&nt->OptionalHeader + > - nt->FileHeader.SizeOfOptionalHeader; > - > - for (i = num_sections - 1; i >= 0; i--) { > - IMAGE_SECTION_HEADER *sec = §ions[i]; > - virt_size = max_t(unsigned long, virt_size, > - sec->VirtualAddress + sec->Misc.VirtualSize); > - } > - > /* Read 32/64bit specific header bits */ > if (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC) { > IMAGE_NT_HEADERS64 *nt64 = (void *)nt; > IMAGE_OPTIONAL_HEADER64 *opt = &nt64->OptionalHeader; > + > image_base = opt->ImageBase; > image_size = opt->SizeOfImage; > efi_set_code_and_data_type(loaded_image_info, opt->Subsystem); > - efi_reloc = efi_alloc(virt_size, > - loaded_image_info->image_code_type); > - if (!efi_reloc) { > - printf("%s: Could not allocate %lu bytes\n", > - __func__, virt_size); > - return EFI_OUT_OF_RESOURCES; > - } > - handle->entry = efi_reloc + opt->AddressOfEntryPoint; > + handle->entry = efi + opt->AddressOfEntryPoint; > rel_size = opt->DataDirectory[rel_idx].Size; > - rel = efi_reloc + opt->DataDirectory[rel_idx].VirtualAddress; > - virt_size = ALIGN(virt_size, opt->SectionAlignment); > + rel = efi + opt->DataDirectory[rel_idx].VirtualAddress; > } else if (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC) { > IMAGE_OPTIONAL_HEADER32 *opt = &nt->OptionalHeader; > + > image_base = opt->ImageBase; > image_size = opt->SizeOfImage; > efi_set_code_and_data_type(loaded_image_info, opt->Subsystem); > - efi_reloc = efi_alloc(virt_size, > - loaded_image_info->image_code_type); > - if (!efi_reloc) { > - printf("%s: Could not allocate %lu bytes\n", > - __func__, virt_size); > - return EFI_OUT_OF_RESOURCES; > - } > - handle->entry = efi_reloc + opt->AddressOfEntryPoint; > + handle->entry = efi + opt->AddressOfEntryPoint; > rel_size = opt->DataDirectory[rel_idx].Size; > - rel = efi_reloc + opt->DataDirectory[rel_idx].VirtualAddress; > - virt_size = ALIGN(virt_size, opt->SectionAlignment); > + rel = efi + opt->DataDirectory[rel_idx].VirtualAddress; > } else { > printf("%s: Invalid optional header magic %x\n", __func__, > nt->OptionalHeader.Magic); > return EFI_LOAD_ERROR; > } > > - /* Load sections into RAM */ > - for (i = num_sections - 1; i >= 0; i--) { > - IMAGE_SECTION_HEADER *sec = §ions[i]; > - memset(efi_reloc + sec->VirtualAddress, 0, > - sec->Misc.VirtualSize); > - memcpy(efi_reloc + sec->VirtualAddress, > - efi + sec->PointerToRawData, > - sec->SizeOfRawData); > - } > - > /* Run through relocations */ > - if (efi_loader_relocate(rel, rel_size, efi_reloc, > + if (efi_loader_relocate(rel, rel_size, efi, > (unsigned long)image_base) != EFI_SUCCESS) { > - efi_free_pages((uintptr_t) efi_reloc, > - (virt_size + EFI_PAGE_MASK) >> EFI_PAGE_SHIFT); > return EFI_LOAD_ERROR; > } > > /* Flush cache */ > - flush_cache((ulong)efi_reloc, > - ALIGN(virt_size, EFI_CACHELINE_SIZE)); > + flush_cache((ulong)efi, ALIGN(image_size, EFI_CACHELINE_SIZE)); > invalidate_icache_all(); > > /* Populate the loaded image interface bits */ > loaded_image_info->image_base = efi; > loaded_image_info->image_size = image_size; > - handle->reloc_base = efi_reloc; > - handle->reloc_size = virt_size; > + handle->reloc_base = efi; > + handle->reloc_size = image_size; > > return EFI_SUCCESS; > } > -- > 2.20.1 > _______________________________________________ U-Boot mailing list [email protected] https://lists.denx.de/listinfo/u-boot

