1a4f67738SDoug Rabson /*- 28a36da99SPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 38a36da99SPedro F. Giffuni * 4326e27d8SDoug Rabson * Copyright (c) 1998-2000 Doug Rabson 5e9eabf59SPeter Wemm * Copyright (c) 2004 Peter Wemm 6a4f67738SDoug Rabson * All rights reserved. 7a4f67738SDoug Rabson * 8a4f67738SDoug Rabson * Redistribution and use in source and binary forms, with or without 9a4f67738SDoug Rabson * modification, are permitted provided that the following conditions 10a4f67738SDoug Rabson * are met: 11a4f67738SDoug Rabson * 1. Redistributions of source code must retain the above copyright 12a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer. 13a4f67738SDoug Rabson * 2. Redistributions in binary form must reproduce the above copyright 14a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer in the 15a4f67738SDoug Rabson * documentation and/or other materials provided with the distribution. 16a4f67738SDoug Rabson * 17a4f67738SDoug Rabson * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 18a4f67738SDoug Rabson * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19a4f67738SDoug Rabson * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20a4f67738SDoug Rabson * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 21a4f67738SDoug Rabson * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22a4f67738SDoug Rabson * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23a4f67738SDoug Rabson * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24a4f67738SDoug Rabson * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25a4f67738SDoug Rabson * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26a4f67738SDoug Rabson * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27a4f67738SDoug Rabson * SUCH DAMAGE. 28a4f67738SDoug Rabson */ 29a4f67738SDoug Rabson 30677b542eSDavid E. O'Brien #include <sys/cdefs.h> 31677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$"); 32677b542eSDavid E. O'Brien 33326e27d8SDoug Rabson #include "opt_ddb.h" 34326e27d8SDoug Rabson 35a4f67738SDoug Rabson #include <sys/param.h> 36a4f67738SDoug Rabson #include <sys/systm.h> 37fb919e4dSMark Murray #include <sys/kernel.h> 38fb919e4dSMark Murray #include <sys/lock.h> 39a4f67738SDoug Rabson #include <sys/malloc.h> 40c0b824f9SDima Dorfman #include <sys/mutex.h> 4162d615d5SJohn Baldwin #include <sys/mount.h> 42a4f67738SDoug Rabson #include <sys/proc.h> 43a4f67738SDoug Rabson #include <sys/namei.h> 44a4f67738SDoug Rabson #include <sys/fcntl.h> 45a4f67738SDoug Rabson #include <sys/vnode.h> 46a4f67738SDoug Rabson #include <sys/linker.h> 47fb919e4dSMark Murray 48a4f67738SDoug Rabson #include <machine/elf.h> 49a4f67738SDoug Rabson 50eddfbb76SRobert Watson #include <net/vnet.h> 51eddfbb76SRobert Watson 52aed55708SRobert Watson #include <security/mac/mac_framework.h> 53aed55708SRobert Watson 54fe3db7c7SDoug Rabson #include <vm/vm.h> 55fe3db7c7SDoug Rabson #include <vm/vm_param.h> 56fe3db7c7SDoug Rabson #include <vm/vm_object.h> 57fe3db7c7SDoug Rabson #include <vm/vm_kern.h> 58fe3db7c7SDoug Rabson #include <vm/vm_extern.h> 59fe3db7c7SDoug Rabson #include <vm/pmap.h> 60fe3db7c7SDoug Rabson #include <vm/vm_map.h> 61fb919e4dSMark Murray 62f9980387SPeter Wemm #include <sys/link_elf.h> 63fe3db7c7SDoug Rabson 64a2024a3eSJohn Birrell #ifdef DDB_CTF 6522bbc4b2SXin LI #include <contrib/zlib/zlib.h> 66a2024a3eSJohn Birrell #endif 67a2024a3eSJohn Birrell 68326e27d8SDoug Rabson #include "linker_if.h" 69a4f67738SDoug Rabson 70e9eabf59SPeter Wemm typedef struct { 71e9eabf59SPeter Wemm void *addr; 7220947801SPeter Wemm Elf_Off size; 73e9eabf59SPeter Wemm int flags; 74e9eabf59SPeter Wemm int sec; /* Original section */ 75e9eabf59SPeter Wemm char *name; 76e9eabf59SPeter Wemm } Elf_progent; 77e9eabf59SPeter Wemm 78e9eabf59SPeter Wemm typedef struct { 79e9eabf59SPeter Wemm Elf_Rel *rel; 8020947801SPeter Wemm int nrel; 81e9eabf59SPeter Wemm int sec; 82e9eabf59SPeter Wemm } Elf_relent; 83e9eabf59SPeter Wemm 84e9eabf59SPeter Wemm typedef struct { 85e9eabf59SPeter Wemm Elf_Rela *rela; 8620947801SPeter Wemm int nrela; 87e9eabf59SPeter Wemm int sec; 88e9eabf59SPeter Wemm } Elf_relaent; 89e9eabf59SPeter Wemm 90e9eabf59SPeter Wemm 91a4f67738SDoug Rabson typedef struct elf_file { 92326e27d8SDoug Rabson struct linker_file lf; /* Common fields */ 9320947801SPeter Wemm 9470b7ffeeSIan Dowse int preloaded; 95fe3db7c7SDoug Rabson caddr_t address; /* Relocation address */ 96fe3db7c7SDoug Rabson vm_object_t object; /* VM object to hold file pages */ 97e9eabf59SPeter Wemm Elf_Shdr *e_shdr; 98e9eabf59SPeter Wemm 99e9eabf59SPeter Wemm Elf_progent *progtab; 100d821d364SPedro F. Giffuni u_int nprogtab; 101e9eabf59SPeter Wemm 102e9eabf59SPeter Wemm Elf_relaent *relatab; 103d821d364SPedro F. Giffuni u_int nrelatab; 104e9eabf59SPeter Wemm 105e9eabf59SPeter Wemm Elf_relent *reltab; 106b474c780SDavid E. O'Brien int nreltab; 107e9eabf59SPeter Wemm 108e9eabf59SPeter Wemm Elf_Sym *ddbsymtab; /* The symbol table we are using */ 1092d636ab0SPeter Wemm long ddbsymcnt; /* Number of symbols */ 1102d636ab0SPeter Wemm caddr_t ddbstrtab; /* String table */ 1112d636ab0SPeter Wemm long ddbstrcnt; /* number of bytes in string table */ 112e9eabf59SPeter Wemm 113e9eabf59SPeter Wemm caddr_t shstrtab; /* Section name string table */ 114e9eabf59SPeter Wemm long shstrcnt; /* number of bytes in string table */ 115e9eabf59SPeter Wemm 116a2024a3eSJohn Birrell caddr_t ctftab; /* CTF table */ 117a2024a3eSJohn Birrell long ctfcnt; /* number of bytes in CTF table */ 118a2024a3eSJohn Birrell caddr_t ctfoff; /* CTF offset table */ 119a2024a3eSJohn Birrell caddr_t typoff; /* Type offset table */ 120a2024a3eSJohn Birrell long typlen; /* Number of type entries. */ 121a2024a3eSJohn Birrell 122a4f67738SDoug Rabson } *elf_file_t; 123a4f67738SDoug Rabson 124a2024a3eSJohn Birrell #include <kern/kern_ctf.c> 125a2024a3eSJohn Birrell 12654823af2SPeter Wemm static int link_elf_link_preload(linker_class_t cls, 127326e27d8SDoug Rabson const char *, linker_file_t *); 12854823af2SPeter Wemm static int link_elf_link_preload_finish(linker_file_t); 12954823af2SPeter Wemm static int link_elf_load_file(linker_class_t, const char *, linker_file_t *); 130326e27d8SDoug Rabson static int link_elf_lookup_symbol(linker_file_t, const char *, 131326e27d8SDoug Rabson c_linker_sym_t *); 132e9eabf59SPeter Wemm static int link_elf_symbol_values(linker_file_t, c_linker_sym_t, 133e9eabf59SPeter Wemm linker_symval_t *); 134326e27d8SDoug Rabson static int link_elf_search_symbol(linker_file_t, caddr_t value, 135326e27d8SDoug Rabson c_linker_sym_t *sym, long *diffp); 136326e27d8SDoug Rabson 137326e27d8SDoug Rabson static void link_elf_unload_file(linker_file_t); 138f41325dbSPeter Wemm static int link_elf_lookup_set(linker_file_t, const char *, 139f41325dbSPeter Wemm void ***, void ***, int *); 140bb9fe9ddSBrian Feldman static int link_elf_each_function_name(linker_file_t, 141e9eabf59SPeter Wemm int (*)(const char *, void *), void *); 142a2024a3eSJohn Birrell static int link_elf_each_function_nameval(linker_file_t, 143a2024a3eSJohn Birrell linker_function_nameval_callback_t, 144a2024a3eSJohn Birrell void *); 145289ead7cSKonstantin Belousov static int link_elf_reloc_local(linker_file_t, bool); 146e76f11f4SAndriy Gapon static long link_elf_symtab_get(linker_file_t, const Elf_Sym **); 14700a5db46SStacey Son static long link_elf_strtab_get(linker_file_t, caddr_t *); 148326e27d8SDoug Rabson 149cff8c6f2SKonstantin Belousov static int elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, 150cff8c6f2SKonstantin Belousov Elf_Addr *); 151e9eabf59SPeter Wemm 152326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = { 153326e27d8SDoug Rabson KOBJMETHOD(linker_lookup_symbol, link_elf_lookup_symbol), 154326e27d8SDoug Rabson KOBJMETHOD(linker_symbol_values, link_elf_symbol_values), 155326e27d8SDoug Rabson KOBJMETHOD(linker_search_symbol, link_elf_search_symbol), 156326e27d8SDoug Rabson KOBJMETHOD(linker_unload, link_elf_unload_file), 15754823af2SPeter Wemm KOBJMETHOD(linker_load_file, link_elf_load_file), 15854823af2SPeter Wemm KOBJMETHOD(linker_link_preload, link_elf_link_preload), 15954823af2SPeter Wemm KOBJMETHOD(linker_link_preload_finish, link_elf_link_preload_finish), 160f41325dbSPeter Wemm KOBJMETHOD(linker_lookup_set, link_elf_lookup_set), 161bb9fe9ddSBrian Feldman KOBJMETHOD(linker_each_function_name, link_elf_each_function_name), 162a2024a3eSJohn Birrell KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval), 163a2024a3eSJohn Birrell KOBJMETHOD(linker_ctf_get, link_elf_ctf_get), 16400a5db46SStacey Son KOBJMETHOD(linker_symtab_get, link_elf_symtab_get), 16500a5db46SStacey Son KOBJMETHOD(linker_strtab_get, link_elf_strtab_get), 1666d775f0bSMark Johnston KOBJMETHOD_END 167326e27d8SDoug Rabson }; 168326e27d8SDoug Rabson 169326e27d8SDoug Rabson static struct linker_class link_elf_class = { 170326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32 171e9eabf59SPeter Wemm "elf32_obj", 172326e27d8SDoug Rabson #else 173e9eabf59SPeter Wemm "elf64_obj", 174326e27d8SDoug Rabson #endif 175326e27d8SDoug Rabson link_elf_methods, sizeof(struct elf_file) 176326e27d8SDoug Rabson }; 177326e27d8SDoug Rabson 178326e27d8SDoug Rabson static int relocate_file(elf_file_t ef); 1792832cd54SKonstantin Belousov static void elf_obj_cleanup_globals_cache(elf_file_t); 180a4f67738SDoug Rabson 181a4f67738SDoug Rabson static void 182552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s) 1831720979bSMarcel Moolenaar { 184552f9f63SEdwin Groothuis if (filename == NULL) 1851720979bSMarcel Moolenaar printf("kldload: %s\n", s); 186552f9f63SEdwin Groothuis else 187552f9f63SEdwin Groothuis printf("kldload: %s: %s\n", filename, s); 1881720979bSMarcel Moolenaar } 1891720979bSMarcel Moolenaar 1901720979bSMarcel Moolenaar static void 191a4f67738SDoug Rabson link_elf_init(void *arg) 192a4f67738SDoug Rabson { 193a4f67738SDoug Rabson 194326e27d8SDoug Rabson linker_add_class(&link_elf_class); 195a4f67738SDoug Rabson } 196a4f67738SDoug Rabson 197891cf3edSEd Maste SYSINIT(link_elf_obj, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, NULL); 198a4f67738SDoug Rabson 199a4f67738SDoug Rabson static int 200e9eabf59SPeter Wemm link_elf_link_preload(linker_class_t cls, const char *filename, 201e9eabf59SPeter Wemm linker_file_t *result) 2022d636ab0SPeter Wemm { 20370b7ffeeSIan Dowse Elf_Ehdr *hdr; 20470b7ffeeSIan Dowse Elf_Shdr *shdr; 20570b7ffeeSIan Dowse Elf_Sym *es; 20670b7ffeeSIan Dowse void *modptr, *baseptr, *sizeptr; 20770b7ffeeSIan Dowse char *type; 20870b7ffeeSIan Dowse elf_file_t ef; 20970b7ffeeSIan Dowse linker_file_t lf; 21070b7ffeeSIan Dowse Elf_Addr off; 21170b7ffeeSIan Dowse int error, i, j, pb, ra, rl, shstrindex, symstrindex, symtabindex; 21270b7ffeeSIan Dowse 21370b7ffeeSIan Dowse /* Look to see if we have the file preloaded */ 21470b7ffeeSIan Dowse modptr = preload_search_by_name(filename); 21570b7ffeeSIan Dowse if (modptr == NULL) 21670b7ffeeSIan Dowse return ENOENT; 21770b7ffeeSIan Dowse 21870b7ffeeSIan Dowse type = (char *)preload_search_info(modptr, MODINFO_TYPE); 21970b7ffeeSIan Dowse baseptr = preload_search_info(modptr, MODINFO_ADDR); 22070b7ffeeSIan Dowse sizeptr = preload_search_info(modptr, MODINFO_SIZE); 22170b7ffeeSIan Dowse hdr = (Elf_Ehdr *)preload_search_info(modptr, MODINFO_METADATA | 22270b7ffeeSIan Dowse MODINFOMD_ELFHDR); 22370b7ffeeSIan Dowse shdr = (Elf_Shdr *)preload_search_info(modptr, MODINFO_METADATA | 22470b7ffeeSIan Dowse MODINFOMD_SHDR); 22570b7ffeeSIan Dowse if (type == NULL || (strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE) 22670b7ffeeSIan Dowse " obj module") != 0 && 22770b7ffeeSIan Dowse strcmp(type, "elf obj module") != 0)) { 228de78ca7eSPeter Wemm return (EFTYPE); 229de78ca7eSPeter Wemm } 23070b7ffeeSIan Dowse if (baseptr == NULL || sizeptr == NULL || hdr == NULL || 23170b7ffeeSIan Dowse shdr == NULL) 23270b7ffeeSIan Dowse return (EINVAL); 23370b7ffeeSIan Dowse 23470b7ffeeSIan Dowse lf = linker_make_file(filename, &link_elf_class); 23570b7ffeeSIan Dowse if (lf == NULL) 23670b7ffeeSIan Dowse return (ENOMEM); 23770b7ffeeSIan Dowse 23870b7ffeeSIan Dowse ef = (elf_file_t)lf; 23970b7ffeeSIan Dowse ef->preloaded = 1; 24070b7ffeeSIan Dowse ef->address = *(caddr_t *)baseptr; 24170b7ffeeSIan Dowse lf->address = *(caddr_t *)baseptr; 24270b7ffeeSIan Dowse lf->size = *(size_t *)sizeptr; 24370b7ffeeSIan Dowse 24470b7ffeeSIan Dowse if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS || 24570b7ffeeSIan Dowse hdr->e_ident[EI_DATA] != ELF_TARG_DATA || 24670b7ffeeSIan Dowse hdr->e_ident[EI_VERSION] != EV_CURRENT || 24770b7ffeeSIan Dowse hdr->e_version != EV_CURRENT || 24870b7ffeeSIan Dowse hdr->e_type != ET_REL || 24970b7ffeeSIan Dowse hdr->e_machine != ELF_TARG_MACH) { 25070b7ffeeSIan Dowse error = EFTYPE; 25170b7ffeeSIan Dowse goto out; 25270b7ffeeSIan Dowse } 25370b7ffeeSIan Dowse ef->e_shdr = shdr; 25470b7ffeeSIan Dowse 25570b7ffeeSIan Dowse /* Scan the section header for information and table sizing. */ 25670b7ffeeSIan Dowse symtabindex = -1; 25770b7ffeeSIan Dowse symstrindex = -1; 25870b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 25970b7ffeeSIan Dowse switch (shdr[i].sh_type) { 26070b7ffeeSIan Dowse case SHT_PROGBITS: 26170b7ffeeSIan Dowse case SHT_NOBITS: 26213f28d96SKonstantin Belousov #ifdef __amd64__ 263b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 26413f28d96SKonstantin Belousov #endif 26558c4aee0SJohn Baldwin /* Ignore sections not loaded by the loader. */ 26658c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 26758c4aee0SJohn Baldwin break; 26870b7ffeeSIan Dowse ef->nprogtab++; 26970b7ffeeSIan Dowse break; 27070b7ffeeSIan Dowse case SHT_SYMTAB: 27170b7ffeeSIan Dowse symtabindex = i; 27270b7ffeeSIan Dowse symstrindex = shdr[i].sh_link; 27370b7ffeeSIan Dowse break; 27470b7ffeeSIan Dowse case SHT_REL: 27515746ef4SJohn Baldwin /* 27615746ef4SJohn Baldwin * Ignore relocation tables for sections not 27715746ef4SJohn Baldwin * loaded by the loader. 27815746ef4SJohn Baldwin */ 27915746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 28015746ef4SJohn Baldwin break; 281b474c780SDavid E. O'Brien ef->nreltab++; 28270b7ffeeSIan Dowse break; 28370b7ffeeSIan Dowse case SHT_RELA: 28415746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 28515746ef4SJohn Baldwin break; 286b474c780SDavid E. O'Brien ef->nrelatab++; 28770b7ffeeSIan Dowse break; 28870b7ffeeSIan Dowse } 28970b7ffeeSIan Dowse } 29070b7ffeeSIan Dowse 29170b7ffeeSIan Dowse shstrindex = hdr->e_shstrndx; 29270b7ffeeSIan Dowse if (ef->nprogtab == 0 || symstrindex < 0 || 29370b7ffeeSIan Dowse symstrindex >= hdr->e_shnum || 29470b7ffeeSIan Dowse shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 || 29570b7ffeeSIan Dowse shstrindex >= hdr->e_shnum || 29670b7ffeeSIan Dowse shdr[shstrindex].sh_type != SHT_STRTAB) { 29770b7ffeeSIan Dowse printf("%s: bad/missing section headers\n", filename); 29870b7ffeeSIan Dowse error = ENOEXEC; 29970b7ffeeSIan Dowse goto out; 30070b7ffeeSIan Dowse } 30170b7ffeeSIan Dowse 30270b7ffeeSIan Dowse /* Allocate space for tracking the load chunks */ 30370b7ffeeSIan Dowse if (ef->nprogtab != 0) 30470b7ffeeSIan Dowse ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 30570b7ffeeSIan Dowse M_LINKER, M_WAITOK | M_ZERO); 306b474c780SDavid E. O'Brien if (ef->nreltab != 0) 307b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 308b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 309b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 310b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 311b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 31270b7ffeeSIan Dowse if ((ef->nprogtab != 0 && ef->progtab == NULL) || 313b474c780SDavid E. O'Brien (ef->nreltab != 0 && ef->reltab == NULL) || 314b474c780SDavid E. O'Brien (ef->nrelatab != 0 && ef->relatab == NULL)) { 31570b7ffeeSIan Dowse error = ENOMEM; 31670b7ffeeSIan Dowse goto out; 31770b7ffeeSIan Dowse } 31870b7ffeeSIan Dowse 31970b7ffeeSIan Dowse /* XXX, relocate the sh_addr fields saved by the loader. */ 32070b7ffeeSIan Dowse off = 0; 32170b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 32270b7ffeeSIan Dowse if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off)) 32370b7ffeeSIan Dowse off = shdr[i].sh_addr; 32470b7ffeeSIan Dowse } 32570b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 32670b7ffeeSIan Dowse if (shdr[i].sh_addr != 0) 32770b7ffeeSIan Dowse shdr[i].sh_addr = shdr[i].sh_addr - off + 32870b7ffeeSIan Dowse (Elf_Addr)ef->address; 32970b7ffeeSIan Dowse } 33070b7ffeeSIan Dowse 33170b7ffeeSIan Dowse ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 33270b7ffeeSIan Dowse ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr; 33370b7ffeeSIan Dowse ef->ddbstrcnt = shdr[symstrindex].sh_size; 33470b7ffeeSIan Dowse ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr; 33570b7ffeeSIan Dowse ef->shstrcnt = shdr[shstrindex].sh_size; 33670b7ffeeSIan Dowse ef->shstrtab = (char *)shdr[shstrindex].sh_addr; 33770b7ffeeSIan Dowse 33870b7ffeeSIan Dowse /* Now fill out progtab and the relocation tables. */ 33970b7ffeeSIan Dowse pb = 0; 34070b7ffeeSIan Dowse rl = 0; 34170b7ffeeSIan Dowse ra = 0; 34270b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 34370b7ffeeSIan Dowse switch (shdr[i].sh_type) { 34470b7ffeeSIan Dowse case SHT_PROGBITS: 34570b7ffeeSIan Dowse case SHT_NOBITS: 34613f28d96SKonstantin Belousov #ifdef __amd64__ 347b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 34813f28d96SKonstantin Belousov #endif 34958c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 35058c4aee0SJohn Baldwin break; 35170b7ffeeSIan Dowse ef->progtab[pb].addr = (void *)shdr[i].sh_addr; 35270b7ffeeSIan Dowse if (shdr[i].sh_type == SHT_PROGBITS) 35370b7ffeeSIan Dowse ef->progtab[pb].name = "<<PROGBITS>>"; 35413f28d96SKonstantin Belousov #ifdef __amd64__ 355b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 35613f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 35713f28d96SKonstantin Belousov #endif 35870b7ffeeSIan Dowse else 35970b7ffeeSIan Dowse ef->progtab[pb].name = "<<NOBITS>>"; 36070b7ffeeSIan Dowse ef->progtab[pb].size = shdr[i].sh_size; 36170b7ffeeSIan Dowse ef->progtab[pb].sec = i; 36270b7ffeeSIan Dowse if (ef->shstrtab && shdr[i].sh_name != 0) 36370b7ffeeSIan Dowse ef->progtab[pb].name = 36470b7ffeeSIan Dowse ef->shstrtab + shdr[i].sh_name; 36550c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 3665f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 36750c202c5SJeff Roberson void *dpcpu; 36850c202c5SJeff Roberson 36950c202c5SJeff Roberson dpcpu = dpcpu_alloc(shdr[i].sh_size); 37050c202c5SJeff Roberson if (dpcpu == NULL) { 3710f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 3729afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 3739afc5684SBjoern A. Zeeb "for %s\n", __func__, 3749afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 3750f823b64SBjoern A. Zeeb filename); 37650c202c5SJeff Roberson error = ENOSPC; 37750c202c5SJeff Roberson goto out; 37850c202c5SJeff Roberson } 37950c202c5SJeff Roberson memcpy(dpcpu, ef->progtab[pb].addr, 38050c202c5SJeff Roberson ef->progtab[pb].size); 38150c202c5SJeff Roberson dpcpu_copy(dpcpu, shdr[i].sh_size); 38250c202c5SJeff Roberson ef->progtab[pb].addr = dpcpu; 383eddfbb76SRobert Watson #ifdef VIMAGE 384eddfbb76SRobert Watson } else if (ef->progtab[pb].name != NULL && 38517ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 386eddfbb76SRobert Watson void *vnet_data; 387eddfbb76SRobert Watson 388eddfbb76SRobert Watson vnet_data = vnet_data_alloc(shdr[i].sh_size); 389eddfbb76SRobert Watson if (vnet_data == NULL) { 3900f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 3919afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 3929afc5684SBjoern A. Zeeb "for %s\n", __func__, 3939afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 3940f823b64SBjoern A. Zeeb filename); 395eddfbb76SRobert Watson error = ENOSPC; 396eddfbb76SRobert Watson goto out; 397eddfbb76SRobert Watson } 398eddfbb76SRobert Watson memcpy(vnet_data, ef->progtab[pb].addr, 399eddfbb76SRobert Watson ef->progtab[pb].size); 400eddfbb76SRobert Watson vnet_data_copy(vnet_data, shdr[i].sh_size); 401eddfbb76SRobert Watson ef->progtab[pb].addr = vnet_data; 402eddfbb76SRobert Watson #endif 4030067051fSMarcel Moolenaar } else if (ef->progtab[pb].name != NULL && 4040067051fSMarcel Moolenaar !strcmp(ef->progtab[pb].name, ".ctors")) { 4050067051fSMarcel Moolenaar lf->ctors_addr = ef->progtab[pb].addr; 4060067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 40750c202c5SJeff Roberson } 40870b7ffeeSIan Dowse 40970b7ffeeSIan Dowse /* Update all symbol values with the offset. */ 41070b7ffeeSIan Dowse for (j = 0; j < ef->ddbsymcnt; j++) { 41170b7ffeeSIan Dowse es = &ef->ddbsymtab[j]; 41270b7ffeeSIan Dowse if (es->st_shndx != i) 41370b7ffeeSIan Dowse continue; 41470b7ffeeSIan Dowse es->st_value += (Elf_Addr)ef->progtab[pb].addr; 41570b7ffeeSIan Dowse } 41670b7ffeeSIan Dowse pb++; 41770b7ffeeSIan Dowse break; 41870b7ffeeSIan Dowse case SHT_REL: 41915746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 42015746ef4SJohn Baldwin break; 42170b7ffeeSIan Dowse ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr; 42270b7ffeeSIan Dowse ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 42370b7ffeeSIan Dowse ef->reltab[rl].sec = shdr[i].sh_info; 42470b7ffeeSIan Dowse rl++; 42570b7ffeeSIan Dowse break; 42670b7ffeeSIan Dowse case SHT_RELA: 42715746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 42815746ef4SJohn Baldwin break; 42970b7ffeeSIan Dowse ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr; 43070b7ffeeSIan Dowse ef->relatab[ra].nrela = 43170b7ffeeSIan Dowse shdr[i].sh_size / sizeof(Elf_Rela); 43270b7ffeeSIan Dowse ef->relatab[ra].sec = shdr[i].sh_info; 43370b7ffeeSIan Dowse ra++; 43470b7ffeeSIan Dowse break; 43570b7ffeeSIan Dowse } 43670b7ffeeSIan Dowse } 4373cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 4383cfce8e4SKonstantin Belousov printf("%s: lost progbits\n", filename); 4393cfce8e4SKonstantin Belousov error = ENOEXEC; 4403cfce8e4SKonstantin Belousov goto out; 4413cfce8e4SKonstantin Belousov } 4423cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 4433cfce8e4SKonstantin Belousov printf("%s: lost reltab\n", filename); 4443cfce8e4SKonstantin Belousov error = ENOEXEC; 4453cfce8e4SKonstantin Belousov goto out; 4463cfce8e4SKonstantin Belousov } 4473cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 4483cfce8e4SKonstantin Belousov printf("%s: lost relatab\n", filename); 4493cfce8e4SKonstantin Belousov error = ENOEXEC; 4503cfce8e4SKonstantin Belousov goto out; 4513cfce8e4SKonstantin Belousov } 45270b7ffeeSIan Dowse 45370b7ffeeSIan Dowse /* Local intra-module relocations */ 454289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 4553cfce8e4SKonstantin Belousov if (error != 0) 4563cfce8e4SKonstantin Belousov goto out; 45770b7ffeeSIan Dowse *result = lf; 45870b7ffeeSIan Dowse return (0); 45970b7ffeeSIan Dowse 46070b7ffeeSIan Dowse out: 46170b7ffeeSIan Dowse /* preload not done this way */ 46270b7ffeeSIan Dowse linker_file_unload(lf, LINKER_UNLOAD_FORCE); 46370b7ffeeSIan Dowse return (error); 46470b7ffeeSIan Dowse } 465de78ca7eSPeter Wemm 4660067051fSMarcel Moolenaar static void 4670067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size) 4680067051fSMarcel Moolenaar { 4690067051fSMarcel Moolenaar void (**ctor)(void); 4700067051fSMarcel Moolenaar size_t i, cnt; 4710067051fSMarcel Moolenaar 4720067051fSMarcel Moolenaar if (addr == NULL || size == 0) 4730067051fSMarcel Moolenaar return; 4740067051fSMarcel Moolenaar cnt = size / sizeof(*ctor); 4750067051fSMarcel Moolenaar ctor = (void *)addr; 4760067051fSMarcel Moolenaar for (i = 0; i < cnt; i++) { 4770067051fSMarcel Moolenaar if (ctor[i] != NULL) 4780067051fSMarcel Moolenaar (*ctor[i])(); 4790067051fSMarcel Moolenaar } 4800067051fSMarcel Moolenaar } 4810067051fSMarcel Moolenaar 482de78ca7eSPeter Wemm static int 48354823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf) 48454823af2SPeter Wemm { 48570b7ffeeSIan Dowse elf_file_t ef; 48670b7ffeeSIan Dowse int error; 48770b7ffeeSIan Dowse 48870b7ffeeSIan Dowse ef = (elf_file_t)lf; 48970b7ffeeSIan Dowse error = relocate_file(ef); 49070b7ffeeSIan Dowse if (error) 491289ead7cSKonstantin Belousov return (error); 49270b7ffeeSIan Dowse 49370b7ffeeSIan Dowse /* Notify MD code that a module is being loaded. */ 49470b7ffeeSIan Dowse error = elf_cpu_load_file(lf); 49570b7ffeeSIan Dowse if (error) 49670b7ffeeSIan Dowse return (error); 49770b7ffeeSIan Dowse 4988f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 499289ead7cSKonstantin Belousov /* Now ifuncs. */ 500289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 501289ead7cSKonstantin Belousov if (error != 0) 502289ead7cSKonstantin Belousov return (error); 5038f941950SKonstantin Belousov #endif 504289ead7cSKonstantin Belousov 5050067051fSMarcel Moolenaar /* Invoke .ctors */ 5060067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 50770b7ffeeSIan Dowse return (0); 50854823af2SPeter Wemm } 50954823af2SPeter Wemm 51054823af2SPeter Wemm static int 511c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename, 512c143d6c2SMarcel Moolenaar linker_file_t *result) 513a4f67738SDoug Rabson { 514a344babfSGleb Smirnoff struct nameidata *nd; 515b40ce416SJulian Elischer struct thread *td = curthread; /* XXX */ 516caab6e90SPeter Wemm Elf_Ehdr *hdr; 517e9eabf59SPeter Wemm Elf_Shdr *shdr; 518a8774e39SPeter Wemm Elf_Sym *es; 519a8774e39SPeter Wemm int nbytes, i, j; 52020947801SPeter Wemm vm_offset_t mapbase; 521fe3db7c7SDoug Rabson size_t mapsize; 522a4f67738SDoug Rabson int error = 0; 523526d0bd5SKonstantin Belousov ssize_t resid; 524526d0bd5SKonstantin Belousov int flags; 525a4f67738SDoug Rabson elf_file_t ef; 526a4f67738SDoug Rabson linker_file_t lf; 527ca65d5c7SPeter Wemm int symtabindex; 528ca65d5c7SPeter Wemm int symstrindex; 529e9eabf59SPeter Wemm int shstrindex; 530e9eabf59SPeter Wemm int nsym; 53120947801SPeter Wemm int pb, rl, ra; 532e9eabf59SPeter Wemm int alignmask; 533ca65d5c7SPeter Wemm 534ca65d5c7SPeter Wemm shdr = NULL; 535ca65d5c7SPeter Wemm lf = NULL; 536e9eabf59SPeter Wemm mapsize = 0; 537e9eabf59SPeter Wemm hdr = NULL; 538a4f67738SDoug Rabson 539a344babfSGleb Smirnoff nd = malloc(sizeof(struct nameidata), M_TEMP, M_WAITOK); 540a344babfSGleb Smirnoff NDINIT(nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td); 541e6796b67SKirk McKusick flags = FREAD; 542a344babfSGleb Smirnoff error = vn_open(nd, &flags, 0, NULL); 543a344babfSGleb Smirnoff if (error) { 544a344babfSGleb Smirnoff free(nd, M_TEMP); 545a4f67738SDoug Rabson return error; 546a344babfSGleb Smirnoff } 547a344babfSGleb Smirnoff NDFREE(nd, NDF_ONLY_PNBUF); 548a344babfSGleb Smirnoff if (nd->ni_vp->v_type != VREG) { 5491f49b573SRuslan Ermilov error = ENOEXEC; 5501f49b573SRuslan Ermilov goto out; 5511f49b573SRuslan Ermilov } 552a3df768bSRobert Watson #ifdef MAC 553a344babfSGleb Smirnoff error = mac_kld_check_load(td->td_ucred, nd->ni_vp); 554a3df768bSRobert Watson if (error) { 555a3df768bSRobert Watson goto out; 556a3df768bSRobert Watson } 557a3df768bSRobert Watson #endif 558a4f67738SDoug Rabson 559e9eabf59SPeter Wemm /* Read the elf header from the file. */ 560e9eabf59SPeter Wemm hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK); 561a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)hdr, sizeof(*hdr), 0, 5629ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 5639ca43589SRobert Watson &resid, td); 564a4f67738SDoug Rabson if (error) 565a4f67738SDoug Rabson goto out; 566e9eabf59SPeter Wemm if (resid != 0){ 567e9eabf59SPeter Wemm error = ENOEXEC; 568e9eabf59SPeter Wemm goto out; 569e9eabf59SPeter Wemm } 570a4f67738SDoug Rabson 571caab6e90SPeter Wemm if (!IS_ELF(*hdr)) { 572fe3db7c7SDoug Rabson error = ENOEXEC; 573a4f67738SDoug Rabson goto out; 574fe3db7c7SDoug Rabson } 575fe3db7c7SDoug Rabson 576caab6e90SPeter Wemm if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS 577caab6e90SPeter Wemm || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 578552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file layout"); 579fe3db7c7SDoug Rabson error = ENOEXEC; 580fe3db7c7SDoug Rabson goto out; 581fe3db7c7SDoug Rabson } 582caab6e90SPeter Wemm if (hdr->e_ident[EI_VERSION] != EV_CURRENT 583caab6e90SPeter Wemm || hdr->e_version != EV_CURRENT) { 584552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file version"); 585fe3db7c7SDoug Rabson error = ENOEXEC; 586fe3db7c7SDoug Rabson goto out; 587fe3db7c7SDoug Rabson } 588e9eabf59SPeter Wemm if (hdr->e_type != ET_REL) { 589a1d7ce03SAttilio Rao error = ENOSYS; 590fe3db7c7SDoug Rabson goto out; 591fe3db7c7SDoug Rabson } 592caab6e90SPeter Wemm if (hdr->e_machine != ELF_TARG_MACH) { 593552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported machine"); 594fe3db7c7SDoug Rabson error = ENOEXEC; 595fe3db7c7SDoug Rabson goto out; 596fe3db7c7SDoug Rabson } 597a4f67738SDoug Rabson 598326e27d8SDoug Rabson lf = linker_make_file(filename, &link_elf_class); 599326e27d8SDoug Rabson if (!lf) { 600326e27d8SDoug Rabson error = ENOMEM; 601326e27d8SDoug Rabson goto out; 602326e27d8SDoug Rabson } 603326e27d8SDoug Rabson ef = (elf_file_t) lf; 604e9eabf59SPeter Wemm ef->nprogtab = 0; 605e9eabf59SPeter Wemm ef->e_shdr = 0; 606b474c780SDavid E. O'Brien ef->nreltab = 0; 607b474c780SDavid E. O'Brien ef->nrelatab = 0; 608e9eabf59SPeter Wemm 609e9eabf59SPeter Wemm /* Allocate and read in the section header */ 610e9eabf59SPeter Wemm nbytes = hdr->e_shnum * hdr->e_shentsize; 611e9eabf59SPeter Wemm if (nbytes == 0 || hdr->e_shoff == 0 || 612e9eabf59SPeter Wemm hdr->e_shentsize != sizeof(Elf_Shdr)) { 613e9eabf59SPeter Wemm error = ENOEXEC; 614e9eabf59SPeter Wemm goto out; 615e9eabf59SPeter Wemm } 61620947801SPeter Wemm shdr = malloc(nbytes, M_LINKER, M_WAITOK); 617e9eabf59SPeter Wemm ef->e_shdr = shdr; 618a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (caddr_t)shdr, nbytes, 619a344babfSGleb Smirnoff hdr->e_shoff, UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 620a344babfSGleb Smirnoff NOCRED, &resid, td); 621e9eabf59SPeter Wemm if (error) 622e9eabf59SPeter Wemm goto out; 623e9eabf59SPeter Wemm if (resid) { 624e9eabf59SPeter Wemm error = ENOEXEC; 625e9eabf59SPeter Wemm goto out; 626e9eabf59SPeter Wemm } 627e9eabf59SPeter Wemm 628e9eabf59SPeter Wemm /* Scan the section header for information and table sizing. */ 629e9eabf59SPeter Wemm nsym = 0; 630e9eabf59SPeter Wemm symtabindex = -1; 631e9eabf59SPeter Wemm symstrindex = -1; 632e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 633676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 634676799a0SAndriy Gapon continue; 635e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 636e9eabf59SPeter Wemm case SHT_PROGBITS: 637e9eabf59SPeter Wemm case SHT_NOBITS: 63813f28d96SKonstantin Belousov #ifdef __amd64__ 639b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 64013f28d96SKonstantin Belousov #endif 64158c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 64258c4aee0SJohn Baldwin break; 64320947801SPeter Wemm ef->nprogtab++; 644e9eabf59SPeter Wemm break; 645e9eabf59SPeter Wemm case SHT_SYMTAB: 646e9eabf59SPeter Wemm nsym++; 647e9eabf59SPeter Wemm symtabindex = i; 648e9eabf59SPeter Wemm symstrindex = shdr[i].sh_link; 649e9eabf59SPeter Wemm break; 650e9eabf59SPeter Wemm case SHT_REL: 65115746ef4SJohn Baldwin /* 65215746ef4SJohn Baldwin * Ignore relocation tables for unallocated 65315746ef4SJohn Baldwin * sections. 65415746ef4SJohn Baldwin */ 65515746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 65615746ef4SJohn Baldwin break; 657b474c780SDavid E. O'Brien ef->nreltab++; 658e9eabf59SPeter Wemm break; 659e9eabf59SPeter Wemm case SHT_RELA: 66015746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 66115746ef4SJohn Baldwin break; 662b474c780SDavid E. O'Brien ef->nrelatab++; 663e9eabf59SPeter Wemm break; 664e9eabf59SPeter Wemm case SHT_STRTAB: 665e9eabf59SPeter Wemm break; 666e9eabf59SPeter Wemm } 667e9eabf59SPeter Wemm } 66820947801SPeter Wemm if (ef->nprogtab == 0) { 669552f9f63SEdwin Groothuis link_elf_error(filename, "file has no contents"); 670e9eabf59SPeter Wemm error = ENOEXEC; 671e9eabf59SPeter Wemm goto out; 672e9eabf59SPeter Wemm } 673e9eabf59SPeter Wemm if (nsym != 1) { 674e9eabf59SPeter Wemm /* Only allow one symbol table for now */ 675f912a970SEd Maste link_elf_error(filename, 676f912a970SEd Maste "file must have exactly one symbol table"); 677e9eabf59SPeter Wemm error = ENOEXEC; 678e9eabf59SPeter Wemm goto out; 679e9eabf59SPeter Wemm } 680e9eabf59SPeter Wemm if (symstrindex < 0 || symstrindex > hdr->e_shnum || 681e9eabf59SPeter Wemm shdr[symstrindex].sh_type != SHT_STRTAB) { 682552f9f63SEdwin Groothuis link_elf_error(filename, "file has invalid symbol strings"); 683e9eabf59SPeter Wemm error = ENOEXEC; 684e9eabf59SPeter Wemm goto out; 685e9eabf59SPeter Wemm } 686e9eabf59SPeter Wemm 687e9eabf59SPeter Wemm /* Allocate space for tracking the load chunks */ 688e9eabf59SPeter Wemm if (ef->nprogtab != 0) 68920947801SPeter Wemm ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 69020947801SPeter Wemm M_LINKER, M_WAITOK | M_ZERO); 691b474c780SDavid E. O'Brien if (ef->nreltab != 0) 692b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 693b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 694b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 695b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 696b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 697e9eabf59SPeter Wemm 6983cfce8e4SKonstantin Belousov if (symtabindex == -1) { 6993cfce8e4SKonstantin Belousov link_elf_error(filename, "lost symbol table index"); 7003cfce8e4SKonstantin Belousov error = ENOEXEC; 7013cfce8e4SKonstantin Belousov goto out; 7023cfce8e4SKonstantin Belousov } 70320947801SPeter Wemm /* Allocate space for and load the symbol table */ 704e9eabf59SPeter Wemm ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 705e9eabf59SPeter Wemm ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK); 706a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)ef->ddbsymtab, 70720947801SPeter Wemm shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset, 70820947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 70920947801SPeter Wemm &resid, td); 71020947801SPeter Wemm if (error) 71120947801SPeter Wemm goto out; 71220947801SPeter Wemm if (resid != 0){ 71320947801SPeter Wemm error = EINVAL; 71420947801SPeter Wemm goto out; 71520947801SPeter Wemm } 716e9eabf59SPeter Wemm 71720947801SPeter Wemm /* Allocate space for and load the symbol strings */ 718e9eabf59SPeter Wemm ef->ddbstrcnt = shdr[symstrindex].sh_size; 719e9eabf59SPeter Wemm ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK); 720a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab, 72120947801SPeter Wemm shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset, 72220947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 72320947801SPeter Wemm &resid, td); 72420947801SPeter Wemm if (error) 72520947801SPeter Wemm goto out; 72620947801SPeter Wemm if (resid != 0){ 72720947801SPeter Wemm error = EINVAL; 72820947801SPeter Wemm goto out; 72920947801SPeter Wemm } 730e9eabf59SPeter Wemm 731e9eabf59SPeter Wemm /* Do we have a string table for the section names? */ 732e9eabf59SPeter Wemm shstrindex = -1; 73320947801SPeter Wemm if (hdr->e_shstrndx != 0 && 73420947801SPeter Wemm shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) { 735e9eabf59SPeter Wemm shstrindex = hdr->e_shstrndx; 736e9eabf59SPeter Wemm ef->shstrcnt = shdr[shstrindex].sh_size; 73720947801SPeter Wemm ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER, 73820947801SPeter Wemm M_WAITOK); 739a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab, 74020947801SPeter Wemm shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset, 741e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 742e9eabf59SPeter Wemm &resid, td); 743e9eabf59SPeter Wemm if (error) 744e9eabf59SPeter Wemm goto out; 74520947801SPeter Wemm if (resid != 0){ 74620947801SPeter Wemm error = EINVAL; 74720947801SPeter Wemm goto out; 74820947801SPeter Wemm } 749e9eabf59SPeter Wemm } 750e9eabf59SPeter Wemm 75120947801SPeter Wemm /* Size up code/data(progbits) and bss(nobits). */ 752e9eabf59SPeter Wemm alignmask = 0; 753e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 754676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 755676799a0SAndriy Gapon continue; 756e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 757e9eabf59SPeter Wemm case SHT_PROGBITS: 758e9eabf59SPeter Wemm case SHT_NOBITS: 75913f28d96SKonstantin Belousov #ifdef __amd64__ 760b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 76113f28d96SKonstantin Belousov #endif 76258c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 76358c4aee0SJohn Baldwin break; 764e9eabf59SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 765e9eabf59SPeter Wemm mapsize += alignmask; 766e9eabf59SPeter Wemm mapsize &= ~alignmask; 767e9eabf59SPeter Wemm mapsize += shdr[i].sh_size; 768e9eabf59SPeter Wemm break; 769e9eabf59SPeter Wemm } 770e9eabf59SPeter Wemm } 771e9eabf59SPeter Wemm 772e9eabf59SPeter Wemm /* 773e9eabf59SPeter Wemm * We know how much space we need for the text/data/bss/etc. 774e9eabf59SPeter Wemm * This stuff needs to be in a single chunk so that profiling etc 775e9eabf59SPeter Wemm * can get the bounds and gdb can associate offsets with modules 776e9eabf59SPeter Wemm */ 7775131cba6SMark Johnston ef->object = vm_object_allocate(OBJT_PHYS, atop(round_page(mapsize))); 778fe3db7c7SDoug Rabson if (ef->object == NULL) { 779fe3db7c7SDoug Rabson error = ENOMEM; 780fe3db7c7SDoug Rabson goto out; 781fe3db7c7SDoug Rabson } 782ac68d1c9SAlan Cox 783ac68d1c9SAlan Cox /* 784ac68d1c9SAlan Cox * In order to satisfy amd64's architectural requirements on the 785ac68d1c9SAlan Cox * location of code and data in the kernel's address space, request a 786ac68d1c9SAlan Cox * mapping that is above the kernel. 787ac68d1c9SAlan Cox */ 7884104e835SOleksandr Tymoshenko #ifdef __amd64__ 789ac68d1c9SAlan Cox mapbase = KERNBASE; 7904104e835SOleksandr Tymoshenko #else 7914104e835SOleksandr Tymoshenko mapbase = VM_MIN_KERNEL_ADDRESS; 7924104e835SOleksandr Tymoshenko #endif 79320947801SPeter Wemm error = vm_map_find(kernel_map, ef->object, 0, &mapbase, 794edb572a3SJohn Baldwin round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL, 795edb572a3SJohn Baldwin VM_PROT_ALL, 0); 796*092bacb2SMark Johnston if (error != KERN_SUCCESS) { 797fe3db7c7SDoug Rabson vm_object_deallocate(ef->object); 798*092bacb2SMark Johnston ef->object = NULL; 799*092bacb2SMark Johnston error = ENOMEM; 800fe3db7c7SDoug Rabson goto out; 801fe3db7c7SDoug Rabson } 80220947801SPeter Wemm 803e9eabf59SPeter Wemm /* Wire the pages */ 8047f1ef325SAlan Cox error = vm_map_wire(kernel_map, mapbase, 80520947801SPeter Wemm mapbase + round_page(mapsize), 806e9eabf59SPeter Wemm VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES); 8077f1ef325SAlan Cox if (error != KERN_SUCCESS) { 8087f1ef325SAlan Cox error = ENOMEM; 8097f1ef325SAlan Cox goto out; 8107f1ef325SAlan Cox } 811a4f67738SDoug Rabson 81220947801SPeter Wemm /* Inform the kld system about the situation */ 81320947801SPeter Wemm lf->address = ef->address = (caddr_t)mapbase; 81420947801SPeter Wemm lf->size = mapsize; 815e9eabf59SPeter Wemm 816e9eabf59SPeter Wemm /* 81720947801SPeter Wemm * Now load code/data(progbits), zero bss(nobits), allocate space for 81820947801SPeter Wemm * and load relocs 819e9eabf59SPeter Wemm */ 82020947801SPeter Wemm pb = 0; 82120947801SPeter Wemm rl = 0; 82220947801SPeter Wemm ra = 0; 82320947801SPeter Wemm alignmask = 0; 82420947801SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 825676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 826676799a0SAndriy Gapon continue; 82720947801SPeter Wemm switch (shdr[i].sh_type) { 82820947801SPeter Wemm case SHT_PROGBITS: 82920947801SPeter Wemm case SHT_NOBITS: 83013f28d96SKonstantin Belousov #ifdef __amd64__ 831b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 83213f28d96SKonstantin Belousov #endif 83358c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 83458c4aee0SJohn Baldwin break; 83520947801SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 83620947801SPeter Wemm mapbase += alignmask; 83720947801SPeter Wemm mapbase &= ~alignmask; 8380067051fSMarcel Moolenaar if (ef->shstrtab != NULL && shdr[i].sh_name != 0) { 83950c202c5SJeff Roberson ef->progtab[pb].name = 84050c202c5SJeff Roberson ef->shstrtab + shdr[i].sh_name; 8410067051fSMarcel Moolenaar if (!strcmp(ef->progtab[pb].name, ".ctors")) { 8420067051fSMarcel Moolenaar lf->ctors_addr = (caddr_t)mapbase; 8430067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 8440067051fSMarcel Moolenaar } 8450067051fSMarcel Moolenaar } else if (shdr[i].sh_type == SHT_PROGBITS) 84620947801SPeter Wemm ef->progtab[pb].name = "<<PROGBITS>>"; 84713f28d96SKonstantin Belousov #ifdef __amd64__ 848b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 84913f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 85013f28d96SKonstantin Belousov #endif 85150c202c5SJeff Roberson else 85250c202c5SJeff Roberson ef->progtab[pb].name = "<<NOBITS>>"; 85350c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 8540f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 85550c202c5SJeff Roberson ef->progtab[pb].addr = 85650c202c5SJeff Roberson dpcpu_alloc(shdr[i].sh_size); 8570f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 8580f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 8599afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 8609afc5684SBjoern A. Zeeb "for %s\n", __func__, 8619afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 8620f823b64SBjoern A. Zeeb filename); 8630f823b64SBjoern A. Zeeb } 8640f823b64SBjoern A. Zeeb } 865eddfbb76SRobert Watson #ifdef VIMAGE 866eddfbb76SRobert Watson else if (ef->progtab[pb].name != NULL && 8670f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 868eddfbb76SRobert Watson ef->progtab[pb].addr = 869eddfbb76SRobert Watson vnet_data_alloc(shdr[i].sh_size); 8700f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 8710f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 8729afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 8739afc5684SBjoern A. Zeeb "for %s\n", __func__, 8749afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 8750f823b64SBjoern A. Zeeb filename); 8760f823b64SBjoern A. Zeeb } 8770f823b64SBjoern A. Zeeb } 878eddfbb76SRobert Watson #endif 87950c202c5SJeff Roberson else 88050c202c5SJeff Roberson ef->progtab[pb].addr = 88150c202c5SJeff Roberson (void *)(uintptr_t)mapbase; 88250c202c5SJeff Roberson if (ef->progtab[pb].addr == NULL) { 88350c202c5SJeff Roberson error = ENOSPC; 88450c202c5SJeff Roberson goto out; 88550c202c5SJeff Roberson } 88650c202c5SJeff Roberson ef->progtab[pb].size = shdr[i].sh_size; 88750c202c5SJeff Roberson ef->progtab[pb].sec = i; 88813f28d96SKonstantin Belousov if (shdr[i].sh_type == SHT_PROGBITS 88913f28d96SKonstantin Belousov #ifdef __amd64__ 890b715d9afSKonstantin Belousov || shdr[i].sh_type == SHT_X86_64_UNWIND 89113f28d96SKonstantin Belousov #endif 89213f28d96SKonstantin Belousov ) { 893a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 89420947801SPeter Wemm ef->progtab[pb].addr, 89520947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 89620947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 89720947801SPeter Wemm NOCRED, &resid, td); 89820947801SPeter Wemm if (error) 89920947801SPeter Wemm goto out; 90020947801SPeter Wemm if (resid != 0){ 90120947801SPeter Wemm error = EINVAL; 90220947801SPeter Wemm goto out; 90320947801SPeter Wemm } 904eddfbb76SRobert Watson /* Initialize the per-cpu or vnet area. */ 905eddfbb76SRobert Watson if (ef->progtab[pb].addr != (void *)mapbase && 9065f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 90750c202c5SJeff Roberson dpcpu_copy(ef->progtab[pb].addr, 90850c202c5SJeff Roberson shdr[i].sh_size); 909eddfbb76SRobert Watson #ifdef VIMAGE 910eddfbb76SRobert Watson else if (ef->progtab[pb].addr != 911eddfbb76SRobert Watson (void *)mapbase && 91217ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 913eddfbb76SRobert Watson vnet_data_copy(ef->progtab[pb].addr, 914eddfbb76SRobert Watson shdr[i].sh_size); 915eddfbb76SRobert Watson #endif 91650c202c5SJeff Roberson } else 91720947801SPeter Wemm bzero(ef->progtab[pb].addr, shdr[i].sh_size); 918a8774e39SPeter Wemm 919a8774e39SPeter Wemm /* Update all symbol values with the offset. */ 920a8774e39SPeter Wemm for (j = 0; j < ef->ddbsymcnt; j++) { 921a8774e39SPeter Wemm es = &ef->ddbsymtab[j]; 922a8774e39SPeter Wemm if (es->st_shndx != i) 923a8774e39SPeter Wemm continue; 924a8774e39SPeter Wemm es->st_value += (Elf_Addr)ef->progtab[pb].addr; 925a8774e39SPeter Wemm } 92620947801SPeter Wemm mapbase += shdr[i].sh_size; 92720947801SPeter Wemm pb++; 92820947801SPeter Wemm break; 92920947801SPeter Wemm case SHT_REL: 93015746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 93115746ef4SJohn Baldwin break; 93220947801SPeter Wemm ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER, 93320947801SPeter Wemm M_WAITOK); 93420947801SPeter Wemm ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 93520947801SPeter Wemm ef->reltab[rl].sec = shdr[i].sh_info; 936a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 93720947801SPeter Wemm (void *)ef->reltab[rl].rel, 93820947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 939e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 940e9eabf59SPeter Wemm &resid, td); 941e9eabf59SPeter Wemm if (error) 942e9eabf59SPeter Wemm goto out; 94320947801SPeter Wemm if (resid != 0){ 94420947801SPeter Wemm error = EINVAL; 94520947801SPeter Wemm goto out; 946e9eabf59SPeter Wemm } 94720947801SPeter Wemm rl++; 94820947801SPeter Wemm break; 94920947801SPeter Wemm case SHT_RELA: 95015746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 95115746ef4SJohn Baldwin break; 95220947801SPeter Wemm ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER, 95320947801SPeter Wemm M_WAITOK); 95420947801SPeter Wemm ef->relatab[ra].nrela = 95520947801SPeter Wemm shdr[i].sh_size / sizeof(Elf_Rela); 95620947801SPeter Wemm ef->relatab[ra].sec = shdr[i].sh_info; 957a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 95820947801SPeter Wemm (void *)ef->relatab[ra].rela, 95920947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 960e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 961e9eabf59SPeter Wemm &resid, td); 962e9eabf59SPeter Wemm if (error) 963e9eabf59SPeter Wemm goto out; 96420947801SPeter Wemm if (resid != 0){ 96520947801SPeter Wemm error = EINVAL; 96620947801SPeter Wemm goto out; 967e9eabf59SPeter Wemm } 96820947801SPeter Wemm ra++; 96920947801SPeter Wemm break; 97020947801SPeter Wemm } 97120947801SPeter Wemm } 9723cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 9733cfce8e4SKonstantin Belousov link_elf_error(filename, "lost progbits"); 9743cfce8e4SKonstantin Belousov error = ENOEXEC; 9753cfce8e4SKonstantin Belousov goto out; 9763cfce8e4SKonstantin Belousov } 9773cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 9783cfce8e4SKonstantin Belousov link_elf_error(filename, "lost reltab"); 9793cfce8e4SKonstantin Belousov error = ENOEXEC; 9803cfce8e4SKonstantin Belousov goto out; 9813cfce8e4SKonstantin Belousov } 9823cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 9833cfce8e4SKonstantin Belousov link_elf_error(filename, "lost relatab"); 9843cfce8e4SKonstantin Belousov error = ENOEXEC; 9853cfce8e4SKonstantin Belousov goto out; 9863cfce8e4SKonstantin Belousov } 9873cfce8e4SKonstantin Belousov if (mapbase != (vm_offset_t)ef->address + mapsize) { 9883cfce8e4SKonstantin Belousov printf( 9893cfce8e4SKonstantin Belousov "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n", 9903cfce8e4SKonstantin Belousov filename != NULL ? filename : "<none>", 9916f3c6327SNeel Natu (u_long)mapbase, ef->address, (u_long)mapsize, 9926f3c6327SNeel Natu (u_long)(vm_offset_t)ef->address + mapsize); 9933cfce8e4SKonstantin Belousov error = ENOMEM; 9943cfce8e4SKonstantin Belousov goto out; 9953cfce8e4SKonstantin Belousov } 996e9eabf59SPeter Wemm 997e9eabf59SPeter Wemm /* Local intra-module relocations */ 998289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 9993cfce8e4SKonstantin Belousov if (error != 0) 10003cfce8e4SKonstantin Belousov goto out; 10017251b4bfSJake Burkholder 1002e9eabf59SPeter Wemm /* Pull in dependencies */ 1003a344babfSGleb Smirnoff VOP_UNLOCK(nd->ni_vp, 0); 10047b9716baSIan Dowse error = linker_load_dependencies(lf); 1005a344babfSGleb Smirnoff vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY); 1006ca65d5c7SPeter Wemm if (error) 1007de78ca7eSPeter Wemm goto out; 1008e9eabf59SPeter Wemm 1009e9eabf59SPeter Wemm /* External relocations */ 1010326e27d8SDoug Rabson error = relocate_file(ef); 1011ca65d5c7SPeter Wemm if (error) 1012ca65d5c7SPeter Wemm goto out; 1013ca65d5c7SPeter Wemm 1014e9eabf59SPeter Wemm /* Notify MD code that a module is being loaded. */ 1015e9eabf59SPeter Wemm error = elf_cpu_load_file(lf); 1016ca65d5c7SPeter Wemm if (error) 1017ca65d5c7SPeter Wemm goto out; 1018a4f67738SDoug Rabson 10198f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 1020289ead7cSKonstantin Belousov /* Now ifuncs. */ 1021289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 1022289ead7cSKonstantin Belousov if (error != 0) 1023289ead7cSKonstantin Belousov goto out; 10248f941950SKonstantin Belousov #endif 1025289ead7cSKonstantin Belousov 10260067051fSMarcel Moolenaar /* Invoke .ctors */ 10270067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 10280067051fSMarcel Moolenaar 1029a4f67738SDoug Rabson *result = lf; 1030a4f67738SDoug Rabson 1031a4f67738SDoug Rabson out: 1032a344babfSGleb Smirnoff VOP_UNLOCK(nd->ni_vp, 0); 1033a344babfSGleb Smirnoff vn_close(nd->ni_vp, FREAD, td->td_ucred, td); 1034a344babfSGleb Smirnoff free(nd, M_TEMP); 1035ca65d5c7SPeter Wemm if (error && lf) 103665a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 1037e9eabf59SPeter Wemm free(hdr, M_LINKER); 1038a4f67738SDoug Rabson 1039a4f67738SDoug Rabson return error; 1040a4f67738SDoug Rabson } 1041a4f67738SDoug Rabson 1042a4f67738SDoug Rabson static void 1043de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file) 1044a4f67738SDoug Rabson { 1045326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) file; 1046d821d364SPedro F. Giffuni u_int i; 1047a4f67738SDoug Rabson 10481aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being unloaded. */ 10491aeb23cdSMarcel Moolenaar elf_cpu_unload_file(file); 10501aeb23cdSMarcel Moolenaar 105150c202c5SJeff Roberson if (ef->progtab) { 105250c202c5SJeff Roberson for (i = 0; i < ef->nprogtab; i++) { 105350c202c5SJeff Roberson if (ef->progtab[i].size == 0) 105450c202c5SJeff Roberson continue; 105550c202c5SJeff Roberson if (ef->progtab[i].name == NULL) 105650c202c5SJeff Roberson continue; 10575f67450dSDimitry Andric if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME)) 105850c202c5SJeff Roberson dpcpu_free(ef->progtab[i].addr, 105950c202c5SJeff Roberson ef->progtab[i].size); 1060eddfbb76SRobert Watson #ifdef VIMAGE 106117ef1febSRobert Watson else if (!strcmp(ef->progtab[i].name, VNET_SETNAME)) 1062eddfbb76SRobert Watson vnet_data_free(ef->progtab[i].addr, 1063eddfbb76SRobert Watson ef->progtab[i].size); 1064eddfbb76SRobert Watson #endif 106550c202c5SJeff Roberson } 106650c202c5SJeff Roberson } 106770b7ffeeSIan Dowse if (ef->preloaded) { 106870b7ffeeSIan Dowse free(ef->reltab, M_LINKER); 106970b7ffeeSIan Dowse free(ef->relatab, M_LINKER); 107070b7ffeeSIan Dowse free(ef->progtab, M_LINKER); 1071a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1072a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1073a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 107473624a80SMark Johnston if (file->pathname != NULL) 107573624a80SMark Johnston preload_delete_name(file->pathname); 107670b7ffeeSIan Dowse return; 107770b7ffeeSIan Dowse } 107870b7ffeeSIan Dowse 1079b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) 108020947801SPeter Wemm free(ef->reltab[i].rel, M_LINKER); 1081b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) 108220947801SPeter Wemm free(ef->relatab[i].rela, M_LINKER); 108320947801SPeter Wemm free(ef->reltab, M_LINKER); 108420947801SPeter Wemm free(ef->relatab, M_LINKER); 108520947801SPeter Wemm free(ef->progtab, M_LINKER); 108620947801SPeter Wemm 1087*092bacb2SMark Johnston if (ef->object != NULL) 1088fe3db7c7SDoug Rabson vm_map_remove(kernel_map, (vm_offset_t)ef->address, 1089*092bacb2SMark Johnston (vm_offset_t)ef->address + ptoa(ef->object->size)); 1090e9eabf59SPeter Wemm free(ef->e_shdr, M_LINKER); 1091e9eabf59SPeter Wemm free(ef->ddbsymtab, M_LINKER); 1092e9eabf59SPeter Wemm free(ef->ddbstrtab, M_LINKER); 109320947801SPeter Wemm free(ef->shstrtab, M_LINKER); 1094a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1095a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1096a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 1097de78ca7eSPeter Wemm } 1098de78ca7eSPeter Wemm 1099fe3db7c7SDoug Rabson static const char * 1100757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info) 1101a4f67738SDoug Rabson { 1102fe3db7c7SDoug Rabson const Elf_Sym *ref; 1103a4f67738SDoug Rabson 1104aa855a59SPeter Wemm if (ELF_R_SYM(r_info)) { 1105e9eabf59SPeter Wemm ref = ef->ddbsymtab + ELF_R_SYM(r_info); 1106e9eabf59SPeter Wemm return ef->ddbstrtab + ref->st_name; 1107fe3db7c7SDoug Rabson } else 1108fe3db7c7SDoug Rabson return NULL; 1109a4f67738SDoug Rabson } 1110a4f67738SDoug Rabson 1111e9eabf59SPeter Wemm static Elf_Addr 1112e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec) 1113e9eabf59SPeter Wemm { 1114e9eabf59SPeter Wemm int i; 1115e9eabf59SPeter Wemm Elf_Addr base = 0; 1116e9eabf59SPeter Wemm 1117e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1118a8774e39SPeter Wemm if (sec == ef->progtab[i].sec) { 1119e9eabf59SPeter Wemm base = (Elf_Addr)ef->progtab[i].addr; 1120a8774e39SPeter Wemm break; 1121e9eabf59SPeter Wemm } 1122a8774e39SPeter Wemm } 1123e9eabf59SPeter Wemm return base; 1124e9eabf59SPeter Wemm } 1125e9eabf59SPeter Wemm 1126a4f67738SDoug Rabson static int 1127326e27d8SDoug Rabson relocate_file(elf_file_t ef) 1128a4f67738SDoug Rabson { 1129fe3db7c7SDoug Rabson const Elf_Rel *rellim; 1130fe3db7c7SDoug Rabson const Elf_Rel *rel; 1131fe3db7c7SDoug Rabson const Elf_Rela *relalim; 1132fe3db7c7SDoug Rabson const Elf_Rela *rela; 1133aa855a59SPeter Wemm const char *symname; 1134e9eabf59SPeter Wemm const Elf_Sym *sym; 1135e9eabf59SPeter Wemm int i; 1136757686b1SMarcel Moolenaar Elf_Size symidx; 1137e9eabf59SPeter Wemm Elf_Addr base; 1138e9eabf59SPeter Wemm 1139a4f67738SDoug Rabson 1140fe3db7c7SDoug Rabson /* Perform relocations without addend if there are any: */ 1141b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1142e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 11433cfce8e4SKonstantin Belousov if (rel == NULL) { 11443cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab!"); 11453cfce8e4SKonstantin Belousov return (ENOEXEC); 11463cfce8e4SKonstantin Belousov } 114720947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1148e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 11493cfce8e4SKonstantin Belousov if (base == 0) { 11503cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 11513cfce8e4SKonstantin Belousov return (ENOEXEC); 11523cfce8e4SKonstantin Belousov } 115320947801SPeter Wemm for ( ; rel < rellim; rel++) { 1154e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 115520947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 115620947801SPeter Wemm continue; 1157e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 115820947801SPeter Wemm /* Local relocs are already done */ 115920947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 116020947801SPeter Wemm continue; 116120947801SPeter Wemm if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL, 116220947801SPeter Wemm elf_obj_lookup)) { 1163aa855a59SPeter Wemm symname = symbol_name(ef, rel->r_info); 116420947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 116520947801SPeter Wemm symname); 11663cfce8e4SKonstantin Belousov return (ENOENT); 1167a13ddfb6SPeter Wemm } 1168aa855a59SPeter Wemm } 1169a4f67738SDoug Rabson } 1170a4f67738SDoug Rabson 1171fe3db7c7SDoug Rabson /* Perform relocations with addend if there are any: */ 1172b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1173e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 11743cfce8e4SKonstantin Belousov if (rela == NULL) { 11753cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 11763cfce8e4SKonstantin Belousov return (ENOEXEC); 11773cfce8e4SKonstantin Belousov } 117820947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1179e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 11803cfce8e4SKonstantin Belousov if (base == 0) { 11813cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, 11823cfce8e4SKonstantin Belousov "lost base for relatab"); 11833cfce8e4SKonstantin Belousov return (ENOEXEC); 11843cfce8e4SKonstantin Belousov } 118520947801SPeter Wemm for ( ; rela < relalim; rela++) { 1186e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 118720947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 118820947801SPeter Wemm continue; 1189e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 119020947801SPeter Wemm /* Local relocs are already done */ 119120947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 119220947801SPeter Wemm continue; 119320947801SPeter Wemm if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA, 119420947801SPeter Wemm elf_obj_lookup)) { 1195aa855a59SPeter Wemm symname = symbol_name(ef, rela->r_info); 119620947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 119720947801SPeter Wemm symname); 11983cfce8e4SKonstantin Belousov return (ENOENT); 1199a13ddfb6SPeter Wemm } 1200e9eabf59SPeter Wemm } 1201e9eabf59SPeter Wemm } 1202a4f67738SDoug Rabson 12032832cd54SKonstantin Belousov /* 1204e3043798SPedro F. Giffuni * Only clean SHN_FBSD_CACHED for successful return. If we 12052832cd54SKonstantin Belousov * modified symbol table for the object but found an 12062832cd54SKonstantin Belousov * unresolved symbol, there is no reason to roll back. 12072832cd54SKonstantin Belousov */ 12082832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(ef); 12092832cd54SKonstantin Belousov 12103cfce8e4SKonstantin Belousov return (0); 1211a4f67738SDoug Rabson } 1212a4f67738SDoug Rabson 121337c84183SPoul-Henning Kamp static int 1214d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym) 1215a4f67738SDoug Rabson { 1216326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 12172d636ab0SPeter Wemm const Elf_Sym *symp; 12182d636ab0SPeter Wemm const char *strp; 1219a4f67738SDoug Rabson int i; 1220a4f67738SDoug Rabson 12212d636ab0SPeter Wemm for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 12222d636ab0SPeter Wemm strp = ef->ddbstrtab + symp->st_name; 1223a8774e39SPeter Wemm if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) { 1224d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 12252d636ab0SPeter Wemm return 0; 12262d636ab0SPeter Wemm } 12272d636ab0SPeter Wemm } 1228a4f67738SDoug Rabson return ENOENT; 1229a4f67738SDoug Rabson } 1230a4f67738SDoug Rabson 1231de78ca7eSPeter Wemm static int 123220947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, 123320947801SPeter Wemm linker_symval_t *symval) 1234a4f67738SDoug Rabson { 12357035cf14SKonstantin Belousov elf_file_t ef; 12367035cf14SKonstantin Belousov const Elf_Sym *es; 12377035cf14SKonstantin Belousov caddr_t val; 1238a4f67738SDoug Rabson 12397035cf14SKonstantin Belousov ef = (elf_file_t) lf; 12407035cf14SKonstantin Belousov es = (const Elf_Sym*) sym; 12417035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 12425cf87418SMarcel Moolenaar if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) { 12432d636ab0SPeter Wemm symval->name = ef->ddbstrtab + es->st_name; 12447035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 12457035cf14SKonstantin Belousov if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC) 12467035cf14SKonstantin Belousov val = ((caddr_t (*)(void))val)(); 12477035cf14SKonstantin Belousov symval->value = val; 12482d636ab0SPeter Wemm symval->size = es->st_size; 12492d636ab0SPeter Wemm return 0; 12502d636ab0SPeter Wemm } 12512d636ab0SPeter Wemm return ENOENT; 12522d636ab0SPeter Wemm } 1253a4f67738SDoug Rabson 1254a4f67738SDoug Rabson static int 1255a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value, 1256d254af07SMatthew Dillon c_linker_sym_t *sym, long *diffp) 1257a4f67738SDoug Rabson { 1258326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1259586453feSBruce Evans u_long off = (uintptr_t) (void *) value; 1260a4f67738SDoug Rabson u_long diff = off; 1261586453feSBruce Evans u_long st_value; 1262fe3db7c7SDoug Rabson const Elf_Sym *es; 1263b85f65afSPedro F. Giffuni const Elf_Sym *best = NULL; 1264a4f67738SDoug Rabson int i; 1265a4f67738SDoug Rabson 12662d636ab0SPeter Wemm for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 1267a4f67738SDoug Rabson if (es->st_name == 0) 1268a4f67738SDoug Rabson continue; 1269a8774e39SPeter Wemm st_value = es->st_value; 1270b5abfb70SPeter Wemm if (off >= st_value) { 1271b5abfb70SPeter Wemm if (off - st_value < diff) { 1272b5abfb70SPeter Wemm diff = off - st_value; 1273a4f67738SDoug Rabson best = es; 1274a4f67738SDoug Rabson if (diff == 0) 1275a4f67738SDoug Rabson break; 1276b5abfb70SPeter Wemm } else if (off - st_value == diff) { 1277a4f67738SDoug Rabson best = es; 1278a4f67738SDoug Rabson } 1279a4f67738SDoug Rabson } 1280a4f67738SDoug Rabson } 1281b85f65afSPedro F. Giffuni if (best == NULL) 1282a4f67738SDoug Rabson *diffp = off; 1283a4f67738SDoug Rabson else 1284a4f67738SDoug Rabson *diffp = diff; 1285d254af07SMatthew Dillon *sym = (c_linker_sym_t) best; 1286a4f67738SDoug Rabson 1287a4f67738SDoug Rabson return 0; 1288a4f67738SDoug Rabson } 1289f41325dbSPeter Wemm 1290f41325dbSPeter Wemm /* 1291f41325dbSPeter Wemm * Look up a linker set on an ELF system. 1292f41325dbSPeter Wemm */ 1293f41325dbSPeter Wemm static int 1294f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name, 1295f41325dbSPeter Wemm void ***startp, void ***stopp, int *countp) 1296f41325dbSPeter Wemm { 1297e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 1298f41325dbSPeter Wemm void **start, **stop; 1299e9eabf59SPeter Wemm int i, count; 1300f41325dbSPeter Wemm 1301e9eabf59SPeter Wemm /* Relative to section number */ 1302e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1303e9eabf59SPeter Wemm if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) && 1304e9eabf59SPeter Wemm strcmp(ef->progtab[i].name + 4, name) == 0) { 1305e9eabf59SPeter Wemm start = (void **)ef->progtab[i].addr; 130620947801SPeter Wemm stop = (void **)((char *)ef->progtab[i].addr + 130720947801SPeter Wemm ef->progtab[i].size); 1308f41325dbSPeter Wemm count = stop - start; 1309f41325dbSPeter Wemm if (startp) 1310f41325dbSPeter Wemm *startp = start; 1311f41325dbSPeter Wemm if (stopp) 1312f41325dbSPeter Wemm *stopp = stop; 1313f41325dbSPeter Wemm if (countp) 1314f41325dbSPeter Wemm *countp = count; 1315e9eabf59SPeter Wemm return (0); 1316e9eabf59SPeter Wemm } 1317e9eabf59SPeter Wemm } 1318e9eabf59SPeter Wemm return (ESRCH); 1319f41325dbSPeter Wemm } 1320bb9fe9ddSBrian Feldman 1321bb9fe9ddSBrian Feldman static int 1322bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file, 1323e9eabf59SPeter Wemm int (*callback)(const char *, void *), void *opaque) 1324e9eabf59SPeter Wemm { 1325bb9fe9ddSBrian Feldman elf_file_t ef = (elf_file_t)file; 1326bb9fe9ddSBrian Feldman const Elf_Sym *symp; 1327bb9fe9ddSBrian Feldman int i, error; 1328bb9fe9ddSBrian Feldman 1329bb9fe9ddSBrian Feldman /* Exhaustive search */ 1330bb9fe9ddSBrian Feldman for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1331bb9fe9ddSBrian Feldman if (symp->st_value != 0 && 13327035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 13337035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 1334bb9fe9ddSBrian Feldman error = callback(ef->ddbstrtab + symp->st_name, opaque); 1335bb9fe9ddSBrian Feldman if (error) 1336bb9fe9ddSBrian Feldman return (error); 1337bb9fe9ddSBrian Feldman } 1338bb9fe9ddSBrian Feldman } 1339bb9fe9ddSBrian Feldman return (0); 1340bb9fe9ddSBrian Feldman } 134184201059SMarcel Moolenaar 1342a2024a3eSJohn Birrell static int 1343a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file, 1344a2024a3eSJohn Birrell linker_function_nameval_callback_t callback, void *opaque) 1345a2024a3eSJohn Birrell { 1346a2024a3eSJohn Birrell linker_symval_t symval; 1347a2024a3eSJohn Birrell elf_file_t ef = (elf_file_t)file; 1348a2024a3eSJohn Birrell const Elf_Sym* symp; 1349a2024a3eSJohn Birrell int i, error; 1350a2024a3eSJohn Birrell 1351a2024a3eSJohn Birrell /* Exhaustive search */ 1352a2024a3eSJohn Birrell for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1353a2024a3eSJohn Birrell if (symp->st_value != 0 && 13547035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 13557035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 13567035cf14SKonstantin Belousov error = link_elf_symbol_values(file, 13577035cf14SKonstantin Belousov (c_linker_sym_t)symp, &symval); 1358a2024a3eSJohn Birrell if (error) 1359a2024a3eSJohn Birrell return (error); 1360a2024a3eSJohn Birrell error = callback(file, i, &symval, opaque); 1361a2024a3eSJohn Birrell if (error) 1362a2024a3eSJohn Birrell return (error); 1363a2024a3eSJohn Birrell } 1364a2024a3eSJohn Birrell } 1365a2024a3eSJohn Birrell return (0); 1366a2024a3eSJohn Birrell } 1367a2024a3eSJohn Birrell 13682832cd54SKonstantin Belousov static void 13692832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef) 13702832cd54SKonstantin Belousov { 13712832cd54SKonstantin Belousov Elf_Sym *sym; 13722832cd54SKonstantin Belousov Elf_Size i; 13732832cd54SKonstantin Belousov 13742832cd54SKonstantin Belousov for (i = 0; i < ef->ddbsymcnt; i++) { 13752832cd54SKonstantin Belousov sym = ef->ddbsymtab + i; 13762832cd54SKonstantin Belousov if (sym->st_shndx == SHN_FBSD_CACHED) { 13772832cd54SKonstantin Belousov sym->st_shndx = SHN_UNDEF; 13782832cd54SKonstantin Belousov sym->st_value = 0; 13792832cd54SKonstantin Belousov } 13802832cd54SKonstantin Belousov } 13812832cd54SKonstantin Belousov } 13822832cd54SKonstantin Belousov 1383d297ad16SMarcel Moolenaar /* 1384d297ad16SMarcel Moolenaar * Symbol lookup function that can be used when the symbol index is known (ie 1385d297ad16SMarcel Moolenaar * in relocations). It uses the symbol index instead of doing a fully fledged 1386d297ad16SMarcel Moolenaar * hash table based lookup when such is valid. For example for local symbols. 1387d297ad16SMarcel Moolenaar * This is not only more efficient, it's also more correct. It's not always 1388d297ad16SMarcel Moolenaar * the case that the symbol can be found through the hash table. 1389d297ad16SMarcel Moolenaar */ 1390cff8c6f2SKonstantin Belousov static int 1391cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res) 1392d297ad16SMarcel Moolenaar { 1393d297ad16SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 13942832cd54SKonstantin Belousov Elf_Sym *sym; 1395d297ad16SMarcel Moolenaar const char *symbol; 1396cff8c6f2SKonstantin Belousov Elf_Addr res1; 1397d297ad16SMarcel Moolenaar 1398d297ad16SMarcel Moolenaar /* Don't even try to lookup the symbol if the index is bogus. */ 1399cff8c6f2SKonstantin Belousov if (symidx >= ef->ddbsymcnt) { 1400cff8c6f2SKonstantin Belousov *res = 0; 1401cff8c6f2SKonstantin Belousov return (EINVAL); 1402cff8c6f2SKonstantin Belousov } 1403d297ad16SMarcel Moolenaar 1404e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 1405d297ad16SMarcel Moolenaar 14064cec6f5dSPeter Wemm /* Quick answer if there is a definition included. */ 1407cff8c6f2SKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) { 1408289ead7cSKonstantin Belousov res1 = (Elf_Addr)sym->st_value; 1409289ead7cSKonstantin Belousov if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) 1410289ead7cSKonstantin Belousov res1 = ((Elf_Addr (*)(void))res1)(); 1411289ead7cSKonstantin Belousov *res = res1; 1412cff8c6f2SKonstantin Belousov return (0); 1413cff8c6f2SKonstantin Belousov } 14144cec6f5dSPeter Wemm 14154cec6f5dSPeter Wemm /* If we get here, then it is undefined and needs a lookup. */ 14164cec6f5dSPeter Wemm switch (ELF_ST_BIND(sym->st_info)) { 14174cec6f5dSPeter Wemm case STB_LOCAL: 14184cec6f5dSPeter Wemm /* Local, but undefined? huh? */ 1419cff8c6f2SKonstantin Belousov *res = 0; 1420cff8c6f2SKonstantin Belousov return (EINVAL); 1421d297ad16SMarcel Moolenaar 1422e9eabf59SPeter Wemm case STB_GLOBAL: 1423cff8c6f2SKonstantin Belousov case STB_WEAK: 1424e9eabf59SPeter Wemm /* Relative to Data or Function name */ 1425e9eabf59SPeter Wemm symbol = ef->ddbstrtab + sym->st_name; 1426d297ad16SMarcel Moolenaar 1427d297ad16SMarcel Moolenaar /* Force a lookup failure if the symbol name is bogus. */ 1428cff8c6f2SKonstantin Belousov if (*symbol == 0) { 1429cff8c6f2SKonstantin Belousov *res = 0; 1430cff8c6f2SKonstantin Belousov return (EINVAL); 1431cff8c6f2SKonstantin Belousov } 1432cff8c6f2SKonstantin Belousov res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps); 14332832cd54SKonstantin Belousov 14342832cd54SKonstantin Belousov /* 14352832cd54SKonstantin Belousov * Cache global lookups during module relocation. The failure 14362832cd54SKonstantin Belousov * case is particularly expensive for callers, who must scan 14372832cd54SKonstantin Belousov * through the entire globals table doing strcmp(). Cache to 14382832cd54SKonstantin Belousov * avoid doing such work repeatedly. 14392832cd54SKonstantin Belousov * 14402832cd54SKonstantin Belousov * After relocation is complete, undefined globals will be 14412832cd54SKonstantin Belousov * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(), 14422832cd54SKonstantin Belousov * above. 14432832cd54SKonstantin Belousov */ 1444cff8c6f2SKonstantin Belousov if (res1 != 0) { 14452832cd54SKonstantin Belousov sym->st_shndx = SHN_FBSD_CACHED; 1446cff8c6f2SKonstantin Belousov sym->st_value = res1; 1447cff8c6f2SKonstantin Belousov *res = res1; 1448e9eabf59SPeter Wemm return (0); 1449cff8c6f2SKonstantin Belousov } else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) { 1450cff8c6f2SKonstantin Belousov sym->st_value = 0; 1451cff8c6f2SKonstantin Belousov *res = 0; 1452cff8c6f2SKonstantin Belousov return (0); 1453cff8c6f2SKonstantin Belousov } 1454cff8c6f2SKonstantin Belousov return (EINVAL); 1455e9eabf59SPeter Wemm 1456e9eabf59SPeter Wemm default: 1457cff8c6f2SKonstantin Belousov return (EINVAL); 1458d297ad16SMarcel Moolenaar } 1459e9eabf59SPeter Wemm } 1460e9eabf59SPeter Wemm 14617251b4bfSJake Burkholder static void 14627226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef) 14637226306eSKonstantin Belousov { 14647226306eSKonstantin Belousov static const char startn[] = "__start_"; 14657226306eSKonstantin Belousov static const char stopn[] = "__stop_"; 14667226306eSKonstantin Belousov Elf_Sym *sym; 14677226306eSKonstantin Belousov const char *sym_name, *linkset_name; 14687226306eSKonstantin Belousov Elf_Addr startp, stopp; 14697226306eSKonstantin Belousov Elf_Size symidx; 14707226306eSKonstantin Belousov int start, i; 14717226306eSKonstantin Belousov 14727226306eSKonstantin Belousov startp = stopp = 0; 14737226306eSKonstantin Belousov for (symidx = 1 /* zero entry is special */; 14747226306eSKonstantin Belousov symidx < ef->ddbsymcnt; symidx++) { 14757226306eSKonstantin Belousov sym = ef->ddbsymtab + symidx; 14767226306eSKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) 14777226306eSKonstantin Belousov continue; 14787226306eSKonstantin Belousov 14797226306eSKonstantin Belousov sym_name = ef->ddbstrtab + sym->st_name; 14807226306eSKonstantin Belousov if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) { 14817226306eSKonstantin Belousov start = 1; 14827226306eSKonstantin Belousov linkset_name = sym_name + sizeof(startn) - 1; 14837226306eSKonstantin Belousov } 14847226306eSKonstantin Belousov else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) { 14857226306eSKonstantin Belousov start = 0; 14867226306eSKonstantin Belousov linkset_name = sym_name + sizeof(stopn) - 1; 14877226306eSKonstantin Belousov } 14887226306eSKonstantin Belousov else 14897226306eSKonstantin Belousov continue; 14907226306eSKonstantin Belousov 14917226306eSKonstantin Belousov for (i = 0; i < ef->nprogtab; i++) { 14927226306eSKonstantin Belousov if (strcmp(ef->progtab[i].name, linkset_name) == 0) { 14937226306eSKonstantin Belousov startp = (Elf_Addr)ef->progtab[i].addr; 14947226306eSKonstantin Belousov stopp = (Elf_Addr)(startp + ef->progtab[i].size); 14957226306eSKonstantin Belousov break; 14967226306eSKonstantin Belousov } 14977226306eSKonstantin Belousov } 14987226306eSKonstantin Belousov if (i == ef->nprogtab) 14997226306eSKonstantin Belousov continue; 15007226306eSKonstantin Belousov 15017226306eSKonstantin Belousov sym->st_value = start ? startp : stopp; 15027226306eSKonstantin Belousov sym->st_shndx = i; 15037226306eSKonstantin Belousov } 15047226306eSKonstantin Belousov } 15057226306eSKonstantin Belousov 15063cfce8e4SKonstantin Belousov static int 1507289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs) 15087251b4bfSJake Burkholder { 1509e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 15107251b4bfSJake Burkholder const Elf_Rel *rellim; 15117251b4bfSJake Burkholder const Elf_Rel *rel; 15127251b4bfSJake Burkholder const Elf_Rela *relalim; 15137251b4bfSJake Burkholder const Elf_Rela *rela; 1514e9eabf59SPeter Wemm const Elf_Sym *sym; 1515e9eabf59SPeter Wemm Elf_Addr base; 1516e9eabf59SPeter Wemm int i; 1517757686b1SMarcel Moolenaar Elf_Size symidx; 1518e9eabf59SPeter Wemm 15197226306eSKonstantin Belousov link_elf_fix_link_set(ef); 15207226306eSKonstantin Belousov 15217251b4bfSJake Burkholder /* Perform relocations without addend if there are any: */ 1522b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1523e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 15243cfce8e4SKonstantin Belousov if (rel == NULL) { 15253cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab"); 15263cfce8e4SKonstantin Belousov return (ENOEXEC); 15273cfce8e4SKonstantin Belousov } 152820947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1529e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 15303cfce8e4SKonstantin Belousov if (base == 0) { 15313cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 15323cfce8e4SKonstantin Belousov return (ENOEXEC); 15333cfce8e4SKonstantin Belousov } 153420947801SPeter Wemm for ( ; rel < rellim; rel++) { 1535e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 153620947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 153720947801SPeter Wemm continue; 1538e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 153920947801SPeter Wemm /* Only do local relocs */ 154020947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 154120947801SPeter Wemm continue; 154236716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 154336716fe2SMark Johnston elf_is_ifunc_reloc(rel->r_info)) == ifuncs) 154420947801SPeter Wemm elf_reloc_local(lf, base, rel, ELF_RELOC_REL, 154520947801SPeter Wemm elf_obj_lookup); 15467251b4bfSJake Burkholder } 1547e9eabf59SPeter Wemm } 15487251b4bfSJake Burkholder 15497251b4bfSJake Burkholder /* Perform relocations with addend if there are any: */ 1550b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1551e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 15523cfce8e4SKonstantin Belousov if (rela == NULL) { 15533cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 15543cfce8e4SKonstantin Belousov return (ENOEXEC); 15553cfce8e4SKonstantin Belousov } 155620947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1557e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 15583cfce8e4SKonstantin Belousov if (base == 0) { 15593cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 15603cfce8e4SKonstantin Belousov return (ENOEXEC); 15613cfce8e4SKonstantin Belousov } 156220947801SPeter Wemm for ( ; rela < relalim; rela++) { 1563e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 156420947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 156520947801SPeter Wemm continue; 1566e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 156720947801SPeter Wemm /* Only do local relocs */ 156820947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 156920947801SPeter Wemm continue; 157036716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 157136716fe2SMark Johnston elf_is_ifunc_reloc(rela->r_info)) == ifuncs) 157220947801SPeter Wemm elf_reloc_local(lf, base, rela, ELF_RELOC_RELA, 157320947801SPeter Wemm elf_obj_lookup); 15747251b4bfSJake Burkholder } 15757251b4bfSJake Burkholder } 15763cfce8e4SKonstantin Belousov return (0); 1577e9eabf59SPeter Wemm } 157800a5db46SStacey Son 157900a5db46SStacey Son static long 1580e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab) 158100a5db46SStacey Son { 158200a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 158300a5db46SStacey Son 158400a5db46SStacey Son *symtab = ef->ddbsymtab; 158500a5db46SStacey Son 158600a5db46SStacey Son if (*symtab == NULL) 158700a5db46SStacey Son return (0); 158800a5db46SStacey Son 158900a5db46SStacey Son return (ef->ddbsymcnt); 159000a5db46SStacey Son } 159100a5db46SStacey Son 159200a5db46SStacey Son static long 159300a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab) 159400a5db46SStacey Son { 159500a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 159600a5db46SStacey Son 159700a5db46SStacey Son *strtab = ef->ddbstrtab; 159800a5db46SStacey Son 159900a5db46SStacey Son if (*strtab == NULL) 160000a5db46SStacey Son return (0); 160100a5db46SStacey Son 160200a5db46SStacey Son return (ef->ddbstrcnt); 160300a5db46SStacey Son } 1604