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 65d9db5225SCraig Rodrigues #include <sys/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), 166326e27d8SDoug Rabson { 0, 0 } 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 " 372*9afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 373*9afc5684SBjoern A. Zeeb "for %s\n", __func__, 374*9afc5684SBjoern 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 " 391*9afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 392*9afc5684SBjoern A. Zeeb "for %s\n", __func__, 393*9afc5684SBjoern 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 7173cfce8e4SKonstantin Belousov if (symstrindex == -1) { 7183cfce8e4SKonstantin Belousov link_elf_error(filename, "lost symbol string index"); 7193cfce8e4SKonstantin Belousov error = ENOEXEC; 7203cfce8e4SKonstantin Belousov goto out; 7213cfce8e4SKonstantin Belousov } 72220947801SPeter Wemm /* Allocate space for and load the symbol strings */ 723e9eabf59SPeter Wemm ef->ddbstrcnt = shdr[symstrindex].sh_size; 724e9eabf59SPeter Wemm ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK); 725a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab, 72620947801SPeter Wemm shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset, 72720947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 72820947801SPeter Wemm &resid, td); 72920947801SPeter Wemm if (error) 73020947801SPeter Wemm goto out; 73120947801SPeter Wemm if (resid != 0){ 73220947801SPeter Wemm error = EINVAL; 73320947801SPeter Wemm goto out; 73420947801SPeter Wemm } 735e9eabf59SPeter Wemm 736e9eabf59SPeter Wemm /* Do we have a string table for the section names? */ 737e9eabf59SPeter Wemm shstrindex = -1; 73820947801SPeter Wemm if (hdr->e_shstrndx != 0 && 73920947801SPeter Wemm shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) { 740e9eabf59SPeter Wemm shstrindex = hdr->e_shstrndx; 741e9eabf59SPeter Wemm ef->shstrcnt = shdr[shstrindex].sh_size; 74220947801SPeter Wemm ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER, 74320947801SPeter Wemm M_WAITOK); 744a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab, 74520947801SPeter Wemm shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset, 746e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 747e9eabf59SPeter Wemm &resid, td); 748e9eabf59SPeter Wemm if (error) 749e9eabf59SPeter Wemm goto out; 75020947801SPeter Wemm if (resid != 0){ 75120947801SPeter Wemm error = EINVAL; 75220947801SPeter Wemm goto out; 75320947801SPeter Wemm } 754e9eabf59SPeter Wemm } 755e9eabf59SPeter Wemm 75620947801SPeter Wemm /* Size up code/data(progbits) and bss(nobits). */ 757e9eabf59SPeter Wemm alignmask = 0; 758e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 759676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 760676799a0SAndriy Gapon continue; 761e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 762e9eabf59SPeter Wemm case SHT_PROGBITS: 763e9eabf59SPeter Wemm case SHT_NOBITS: 76413f28d96SKonstantin Belousov #ifdef __amd64__ 765b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 76613f28d96SKonstantin Belousov #endif 76758c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 76858c4aee0SJohn Baldwin break; 769e9eabf59SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 770e9eabf59SPeter Wemm mapsize += alignmask; 771e9eabf59SPeter Wemm mapsize &= ~alignmask; 772e9eabf59SPeter Wemm mapsize += shdr[i].sh_size; 773e9eabf59SPeter Wemm break; 774e9eabf59SPeter Wemm } 775e9eabf59SPeter Wemm } 776e9eabf59SPeter Wemm 777e9eabf59SPeter Wemm /* 778e9eabf59SPeter Wemm * We know how much space we need for the text/data/bss/etc. 779e9eabf59SPeter Wemm * This stuff needs to be in a single chunk so that profiling etc 780e9eabf59SPeter Wemm * can get the bounds and gdb can associate offsets with modules 781e9eabf59SPeter Wemm */ 78220947801SPeter Wemm ef->object = vm_object_allocate(OBJT_DEFAULT, 78320947801SPeter Wemm round_page(mapsize) >> PAGE_SHIFT); 784fe3db7c7SDoug Rabson if (ef->object == NULL) { 785fe3db7c7SDoug Rabson error = ENOMEM; 786fe3db7c7SDoug Rabson goto out; 787fe3db7c7SDoug Rabson } 788fe3db7c7SDoug Rabson ef->address = (caddr_t) vm_map_min(kernel_map); 789ac68d1c9SAlan Cox 790ac68d1c9SAlan Cox /* 791ac68d1c9SAlan Cox * In order to satisfy amd64's architectural requirements on the 792ac68d1c9SAlan Cox * location of code and data in the kernel's address space, request a 793ac68d1c9SAlan Cox * mapping that is above the kernel. 794ac68d1c9SAlan Cox */ 7954104e835SOleksandr Tymoshenko #ifdef __amd64__ 796ac68d1c9SAlan Cox mapbase = KERNBASE; 7974104e835SOleksandr Tymoshenko #else 7984104e835SOleksandr Tymoshenko mapbase = VM_MIN_KERNEL_ADDRESS; 7994104e835SOleksandr Tymoshenko #endif 80020947801SPeter Wemm error = vm_map_find(kernel_map, ef->object, 0, &mapbase, 801edb572a3SJohn Baldwin round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL, 802edb572a3SJohn Baldwin VM_PROT_ALL, 0); 803fe3db7c7SDoug Rabson if (error) { 804fe3db7c7SDoug Rabson vm_object_deallocate(ef->object); 805326e27d8SDoug Rabson ef->object = 0; 806fe3db7c7SDoug Rabson goto out; 807fe3db7c7SDoug Rabson } 80820947801SPeter Wemm 809e9eabf59SPeter Wemm /* Wire the pages */ 8107f1ef325SAlan Cox error = vm_map_wire(kernel_map, mapbase, 81120947801SPeter Wemm mapbase + round_page(mapsize), 812e9eabf59SPeter Wemm VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES); 8137f1ef325SAlan Cox if (error != KERN_SUCCESS) { 8147f1ef325SAlan Cox error = ENOMEM; 8157f1ef325SAlan Cox goto out; 8167f1ef325SAlan Cox } 817a4f67738SDoug Rabson 81820947801SPeter Wemm /* Inform the kld system about the situation */ 81920947801SPeter Wemm lf->address = ef->address = (caddr_t)mapbase; 82020947801SPeter Wemm lf->size = mapsize; 821e9eabf59SPeter Wemm 822e9eabf59SPeter Wemm /* 82320947801SPeter Wemm * Now load code/data(progbits), zero bss(nobits), allocate space for 82420947801SPeter Wemm * and load relocs 825e9eabf59SPeter Wemm */ 82620947801SPeter Wemm pb = 0; 82720947801SPeter Wemm rl = 0; 82820947801SPeter Wemm ra = 0; 82920947801SPeter Wemm alignmask = 0; 83020947801SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 831676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 832676799a0SAndriy Gapon continue; 83320947801SPeter Wemm switch (shdr[i].sh_type) { 83420947801SPeter Wemm case SHT_PROGBITS: 83520947801SPeter Wemm case SHT_NOBITS: 83613f28d96SKonstantin Belousov #ifdef __amd64__ 837b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 83813f28d96SKonstantin Belousov #endif 83958c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 84058c4aee0SJohn Baldwin break; 84120947801SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 84220947801SPeter Wemm mapbase += alignmask; 84320947801SPeter Wemm mapbase &= ~alignmask; 8440067051fSMarcel Moolenaar if (ef->shstrtab != NULL && shdr[i].sh_name != 0) { 84550c202c5SJeff Roberson ef->progtab[pb].name = 84650c202c5SJeff Roberson ef->shstrtab + shdr[i].sh_name; 8470067051fSMarcel Moolenaar if (!strcmp(ef->progtab[pb].name, ".ctors")) { 8480067051fSMarcel Moolenaar lf->ctors_addr = (caddr_t)mapbase; 8490067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 8500067051fSMarcel Moolenaar } 8510067051fSMarcel Moolenaar } else if (shdr[i].sh_type == SHT_PROGBITS) 85220947801SPeter Wemm ef->progtab[pb].name = "<<PROGBITS>>"; 85313f28d96SKonstantin Belousov #ifdef __amd64__ 854b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 85513f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 85613f28d96SKonstantin Belousov #endif 85750c202c5SJeff Roberson else 85850c202c5SJeff Roberson ef->progtab[pb].name = "<<NOBITS>>"; 85950c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 8600f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 86150c202c5SJeff Roberson ef->progtab[pb].addr = 86250c202c5SJeff Roberson dpcpu_alloc(shdr[i].sh_size); 8630f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 8640f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 865*9afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 866*9afc5684SBjoern A. Zeeb "for %s\n", __func__, 867*9afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 8680f823b64SBjoern A. Zeeb filename); 8690f823b64SBjoern A. Zeeb } 8700f823b64SBjoern A. Zeeb } 871eddfbb76SRobert Watson #ifdef VIMAGE 872eddfbb76SRobert Watson else if (ef->progtab[pb].name != NULL && 8730f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 874eddfbb76SRobert Watson ef->progtab[pb].addr = 875eddfbb76SRobert Watson vnet_data_alloc(shdr[i].sh_size); 8760f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 8770f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 878*9afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 879*9afc5684SBjoern A. Zeeb "for %s\n", __func__, 880*9afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 8810f823b64SBjoern A. Zeeb filename); 8820f823b64SBjoern A. Zeeb } 8830f823b64SBjoern A. Zeeb } 884eddfbb76SRobert Watson #endif 88550c202c5SJeff Roberson else 88650c202c5SJeff Roberson ef->progtab[pb].addr = 88750c202c5SJeff Roberson (void *)(uintptr_t)mapbase; 88850c202c5SJeff Roberson if (ef->progtab[pb].addr == NULL) { 88950c202c5SJeff Roberson error = ENOSPC; 89050c202c5SJeff Roberson goto out; 89150c202c5SJeff Roberson } 89250c202c5SJeff Roberson ef->progtab[pb].size = shdr[i].sh_size; 89350c202c5SJeff Roberson ef->progtab[pb].sec = i; 89413f28d96SKonstantin Belousov if (shdr[i].sh_type == SHT_PROGBITS 89513f28d96SKonstantin Belousov #ifdef __amd64__ 896b715d9afSKonstantin Belousov || shdr[i].sh_type == SHT_X86_64_UNWIND 89713f28d96SKonstantin Belousov #endif 89813f28d96SKonstantin Belousov ) { 899a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 90020947801SPeter Wemm ef->progtab[pb].addr, 90120947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 90220947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 90320947801SPeter Wemm NOCRED, &resid, td); 90420947801SPeter Wemm if (error) 90520947801SPeter Wemm goto out; 90620947801SPeter Wemm if (resid != 0){ 90720947801SPeter Wemm error = EINVAL; 90820947801SPeter Wemm goto out; 90920947801SPeter Wemm } 910eddfbb76SRobert Watson /* Initialize the per-cpu or vnet area. */ 911eddfbb76SRobert Watson if (ef->progtab[pb].addr != (void *)mapbase && 9125f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 91350c202c5SJeff Roberson dpcpu_copy(ef->progtab[pb].addr, 91450c202c5SJeff Roberson shdr[i].sh_size); 915eddfbb76SRobert Watson #ifdef VIMAGE 916eddfbb76SRobert Watson else if (ef->progtab[pb].addr != 917eddfbb76SRobert Watson (void *)mapbase && 91817ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 919eddfbb76SRobert Watson vnet_data_copy(ef->progtab[pb].addr, 920eddfbb76SRobert Watson shdr[i].sh_size); 921eddfbb76SRobert Watson #endif 92250c202c5SJeff Roberson } else 92320947801SPeter Wemm bzero(ef->progtab[pb].addr, shdr[i].sh_size); 924a8774e39SPeter Wemm 925a8774e39SPeter Wemm /* Update all symbol values with the offset. */ 926a8774e39SPeter Wemm for (j = 0; j < ef->ddbsymcnt; j++) { 927a8774e39SPeter Wemm es = &ef->ddbsymtab[j]; 928a8774e39SPeter Wemm if (es->st_shndx != i) 929a8774e39SPeter Wemm continue; 930a8774e39SPeter Wemm es->st_value += (Elf_Addr)ef->progtab[pb].addr; 931a8774e39SPeter Wemm } 93220947801SPeter Wemm mapbase += shdr[i].sh_size; 93320947801SPeter Wemm pb++; 93420947801SPeter Wemm break; 93520947801SPeter Wemm case SHT_REL: 93615746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 93715746ef4SJohn Baldwin break; 93820947801SPeter Wemm ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER, 93920947801SPeter Wemm M_WAITOK); 94020947801SPeter Wemm ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 94120947801SPeter Wemm ef->reltab[rl].sec = shdr[i].sh_info; 942a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 94320947801SPeter Wemm (void *)ef->reltab[rl].rel, 94420947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 945e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 946e9eabf59SPeter Wemm &resid, td); 947e9eabf59SPeter Wemm if (error) 948e9eabf59SPeter Wemm goto out; 94920947801SPeter Wemm if (resid != 0){ 95020947801SPeter Wemm error = EINVAL; 95120947801SPeter Wemm goto out; 952e9eabf59SPeter Wemm } 95320947801SPeter Wemm rl++; 95420947801SPeter Wemm break; 95520947801SPeter Wemm case SHT_RELA: 95615746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 95715746ef4SJohn Baldwin break; 95820947801SPeter Wemm ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER, 95920947801SPeter Wemm M_WAITOK); 96020947801SPeter Wemm ef->relatab[ra].nrela = 96120947801SPeter Wemm shdr[i].sh_size / sizeof(Elf_Rela); 96220947801SPeter Wemm ef->relatab[ra].sec = shdr[i].sh_info; 963a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 96420947801SPeter Wemm (void *)ef->relatab[ra].rela, 96520947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 966e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 967e9eabf59SPeter Wemm &resid, td); 968e9eabf59SPeter Wemm if (error) 969e9eabf59SPeter Wemm goto out; 97020947801SPeter Wemm if (resid != 0){ 97120947801SPeter Wemm error = EINVAL; 97220947801SPeter Wemm goto out; 973e9eabf59SPeter Wemm } 97420947801SPeter Wemm ra++; 97520947801SPeter Wemm break; 97620947801SPeter Wemm } 97720947801SPeter Wemm } 9783cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 9793cfce8e4SKonstantin Belousov link_elf_error(filename, "lost progbits"); 9803cfce8e4SKonstantin Belousov error = ENOEXEC; 9813cfce8e4SKonstantin Belousov goto out; 9823cfce8e4SKonstantin Belousov } 9833cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 9843cfce8e4SKonstantin Belousov link_elf_error(filename, "lost reltab"); 9853cfce8e4SKonstantin Belousov error = ENOEXEC; 9863cfce8e4SKonstantin Belousov goto out; 9873cfce8e4SKonstantin Belousov } 9883cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 9893cfce8e4SKonstantin Belousov link_elf_error(filename, "lost relatab"); 9903cfce8e4SKonstantin Belousov error = ENOEXEC; 9913cfce8e4SKonstantin Belousov goto out; 9923cfce8e4SKonstantin Belousov } 9933cfce8e4SKonstantin Belousov if (mapbase != (vm_offset_t)ef->address + mapsize) { 9943cfce8e4SKonstantin Belousov printf( 9953cfce8e4SKonstantin Belousov "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n", 9963cfce8e4SKonstantin Belousov filename != NULL ? filename : "<none>", 9976f3c6327SNeel Natu (u_long)mapbase, ef->address, (u_long)mapsize, 9986f3c6327SNeel Natu (u_long)(vm_offset_t)ef->address + mapsize); 9993cfce8e4SKonstantin Belousov error = ENOMEM; 10003cfce8e4SKonstantin Belousov goto out; 10013cfce8e4SKonstantin Belousov } 1002e9eabf59SPeter Wemm 1003e9eabf59SPeter Wemm /* Local intra-module relocations */ 1004289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 10053cfce8e4SKonstantin Belousov if (error != 0) 10063cfce8e4SKonstantin Belousov goto out; 10077251b4bfSJake Burkholder 1008e9eabf59SPeter Wemm /* Pull in dependencies */ 1009a344babfSGleb Smirnoff VOP_UNLOCK(nd->ni_vp, 0); 10107b9716baSIan Dowse error = linker_load_dependencies(lf); 1011a344babfSGleb Smirnoff vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY); 1012ca65d5c7SPeter Wemm if (error) 1013de78ca7eSPeter Wemm goto out; 1014e9eabf59SPeter Wemm 1015e9eabf59SPeter Wemm /* External relocations */ 1016326e27d8SDoug Rabson error = relocate_file(ef); 1017ca65d5c7SPeter Wemm if (error) 1018ca65d5c7SPeter Wemm goto out; 1019ca65d5c7SPeter Wemm 1020e9eabf59SPeter Wemm /* Notify MD code that a module is being loaded. */ 1021e9eabf59SPeter Wemm error = elf_cpu_load_file(lf); 1022ca65d5c7SPeter Wemm if (error) 1023ca65d5c7SPeter Wemm goto out; 1024a4f67738SDoug Rabson 10258f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 1026289ead7cSKonstantin Belousov /* Now ifuncs. */ 1027289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 1028289ead7cSKonstantin Belousov if (error != 0) 1029289ead7cSKonstantin Belousov goto out; 10308f941950SKonstantin Belousov #endif 1031289ead7cSKonstantin Belousov 10320067051fSMarcel Moolenaar /* Invoke .ctors */ 10330067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 10340067051fSMarcel Moolenaar 1035a4f67738SDoug Rabson *result = lf; 1036a4f67738SDoug Rabson 1037a4f67738SDoug Rabson out: 1038a344babfSGleb Smirnoff VOP_UNLOCK(nd->ni_vp, 0); 1039a344babfSGleb Smirnoff vn_close(nd->ni_vp, FREAD, td->td_ucred, td); 1040a344babfSGleb Smirnoff free(nd, M_TEMP); 1041ca65d5c7SPeter Wemm if (error && lf) 104265a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 1043e9eabf59SPeter Wemm free(hdr, M_LINKER); 1044a4f67738SDoug Rabson 1045a4f67738SDoug Rabson return error; 1046a4f67738SDoug Rabson } 1047a4f67738SDoug Rabson 1048a4f67738SDoug Rabson static void 1049de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file) 1050a4f67738SDoug Rabson { 1051326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) file; 1052d821d364SPedro F. Giffuni u_int i; 1053a4f67738SDoug Rabson 10541aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being unloaded. */ 10551aeb23cdSMarcel Moolenaar elf_cpu_unload_file(file); 10561aeb23cdSMarcel Moolenaar 105750c202c5SJeff Roberson if (ef->progtab) { 105850c202c5SJeff Roberson for (i = 0; i < ef->nprogtab; i++) { 105950c202c5SJeff Roberson if (ef->progtab[i].size == 0) 106050c202c5SJeff Roberson continue; 106150c202c5SJeff Roberson if (ef->progtab[i].name == NULL) 106250c202c5SJeff Roberson continue; 10635f67450dSDimitry Andric if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME)) 106450c202c5SJeff Roberson dpcpu_free(ef->progtab[i].addr, 106550c202c5SJeff Roberson ef->progtab[i].size); 1066eddfbb76SRobert Watson #ifdef VIMAGE 106717ef1febSRobert Watson else if (!strcmp(ef->progtab[i].name, VNET_SETNAME)) 1068eddfbb76SRobert Watson vnet_data_free(ef->progtab[i].addr, 1069eddfbb76SRobert Watson ef->progtab[i].size); 1070eddfbb76SRobert Watson #endif 107150c202c5SJeff Roberson } 107250c202c5SJeff Roberson } 107370b7ffeeSIan Dowse if (ef->preloaded) { 107470b7ffeeSIan Dowse free(ef->reltab, M_LINKER); 107570b7ffeeSIan Dowse free(ef->relatab, M_LINKER); 107670b7ffeeSIan Dowse free(ef->progtab, M_LINKER); 1077a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1078a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1079a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 108073624a80SMark Johnston if (file->pathname != NULL) 108173624a80SMark Johnston preload_delete_name(file->pathname); 108270b7ffeeSIan Dowse return; 108370b7ffeeSIan Dowse } 108470b7ffeeSIan Dowse 1085b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) 108620947801SPeter Wemm free(ef->reltab[i].rel, M_LINKER); 1087b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) 108820947801SPeter Wemm free(ef->relatab[i].rela, M_LINKER); 108920947801SPeter Wemm free(ef->reltab, M_LINKER); 109020947801SPeter Wemm free(ef->relatab, M_LINKER); 109120947801SPeter Wemm free(ef->progtab, M_LINKER); 109220947801SPeter Wemm 1093fe3db7c7SDoug Rabson if (ef->object) { 1094fe3db7c7SDoug Rabson vm_map_remove(kernel_map, (vm_offset_t) ef->address, 109520947801SPeter Wemm (vm_offset_t) ef->address + 109620947801SPeter Wemm (ef->object->size << PAGE_SHIFT)); 1097fe3db7c7SDoug Rabson } 1098e9eabf59SPeter Wemm free(ef->e_shdr, M_LINKER); 1099e9eabf59SPeter Wemm free(ef->ddbsymtab, M_LINKER); 1100e9eabf59SPeter Wemm free(ef->ddbstrtab, M_LINKER); 110120947801SPeter Wemm free(ef->shstrtab, M_LINKER); 1102a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1103a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1104a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 1105de78ca7eSPeter Wemm } 1106de78ca7eSPeter Wemm 1107fe3db7c7SDoug Rabson static const char * 1108757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info) 1109a4f67738SDoug Rabson { 1110fe3db7c7SDoug Rabson const Elf_Sym *ref; 1111a4f67738SDoug Rabson 1112aa855a59SPeter Wemm if (ELF_R_SYM(r_info)) { 1113e9eabf59SPeter Wemm ref = ef->ddbsymtab + ELF_R_SYM(r_info); 1114e9eabf59SPeter Wemm return ef->ddbstrtab + ref->st_name; 1115fe3db7c7SDoug Rabson } else 1116fe3db7c7SDoug Rabson return NULL; 1117a4f67738SDoug Rabson } 1118a4f67738SDoug Rabson 1119e9eabf59SPeter Wemm static Elf_Addr 1120e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec) 1121e9eabf59SPeter Wemm { 1122e9eabf59SPeter Wemm int i; 1123e9eabf59SPeter Wemm Elf_Addr base = 0; 1124e9eabf59SPeter Wemm 1125e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1126a8774e39SPeter Wemm if (sec == ef->progtab[i].sec) { 1127e9eabf59SPeter Wemm base = (Elf_Addr)ef->progtab[i].addr; 1128a8774e39SPeter Wemm break; 1129e9eabf59SPeter Wemm } 1130a8774e39SPeter Wemm } 1131e9eabf59SPeter Wemm return base; 1132e9eabf59SPeter Wemm } 1133e9eabf59SPeter Wemm 1134a4f67738SDoug Rabson static int 1135326e27d8SDoug Rabson relocate_file(elf_file_t ef) 1136a4f67738SDoug Rabson { 1137fe3db7c7SDoug Rabson const Elf_Rel *rellim; 1138fe3db7c7SDoug Rabson const Elf_Rel *rel; 1139fe3db7c7SDoug Rabson const Elf_Rela *relalim; 1140fe3db7c7SDoug Rabson const Elf_Rela *rela; 1141aa855a59SPeter Wemm const char *symname; 1142e9eabf59SPeter Wemm const Elf_Sym *sym; 1143e9eabf59SPeter Wemm int i; 1144757686b1SMarcel Moolenaar Elf_Size symidx; 1145e9eabf59SPeter Wemm Elf_Addr base; 1146e9eabf59SPeter Wemm 1147a4f67738SDoug Rabson 1148fe3db7c7SDoug Rabson /* Perform relocations without addend if there are any: */ 1149b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1150e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 11513cfce8e4SKonstantin Belousov if (rel == NULL) { 11523cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab!"); 11533cfce8e4SKonstantin Belousov return (ENOEXEC); 11543cfce8e4SKonstantin Belousov } 115520947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1156e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 11573cfce8e4SKonstantin Belousov if (base == 0) { 11583cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 11593cfce8e4SKonstantin Belousov return (ENOEXEC); 11603cfce8e4SKonstantin Belousov } 116120947801SPeter Wemm for ( ; rel < rellim; rel++) { 1162e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 116320947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 116420947801SPeter Wemm continue; 1165e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 116620947801SPeter Wemm /* Local relocs are already done */ 116720947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 116820947801SPeter Wemm continue; 116920947801SPeter Wemm if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL, 117020947801SPeter Wemm elf_obj_lookup)) { 1171aa855a59SPeter Wemm symname = symbol_name(ef, rel->r_info); 117220947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 117320947801SPeter Wemm symname); 11743cfce8e4SKonstantin Belousov return (ENOENT); 1175a13ddfb6SPeter Wemm } 1176aa855a59SPeter Wemm } 1177a4f67738SDoug Rabson } 1178a4f67738SDoug Rabson 1179fe3db7c7SDoug Rabson /* Perform relocations with addend if there are any: */ 1180b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1181e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 11823cfce8e4SKonstantin Belousov if (rela == NULL) { 11833cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 11843cfce8e4SKonstantin Belousov return (ENOEXEC); 11853cfce8e4SKonstantin Belousov } 118620947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1187e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 11883cfce8e4SKonstantin Belousov if (base == 0) { 11893cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, 11903cfce8e4SKonstantin Belousov "lost base for relatab"); 11913cfce8e4SKonstantin Belousov return (ENOEXEC); 11923cfce8e4SKonstantin Belousov } 119320947801SPeter Wemm for ( ; rela < relalim; rela++) { 1194e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 119520947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 119620947801SPeter Wemm continue; 1197e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 119820947801SPeter Wemm /* Local relocs are already done */ 119920947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 120020947801SPeter Wemm continue; 120120947801SPeter Wemm if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA, 120220947801SPeter Wemm elf_obj_lookup)) { 1203aa855a59SPeter Wemm symname = symbol_name(ef, rela->r_info); 120420947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 120520947801SPeter Wemm symname); 12063cfce8e4SKonstantin Belousov return (ENOENT); 1207a13ddfb6SPeter Wemm } 1208e9eabf59SPeter Wemm } 1209e9eabf59SPeter Wemm } 1210a4f67738SDoug Rabson 12112832cd54SKonstantin Belousov /* 1212e3043798SPedro F. Giffuni * Only clean SHN_FBSD_CACHED for successful return. If we 12132832cd54SKonstantin Belousov * modified symbol table for the object but found an 12142832cd54SKonstantin Belousov * unresolved symbol, there is no reason to roll back. 12152832cd54SKonstantin Belousov */ 12162832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(ef); 12172832cd54SKonstantin Belousov 12183cfce8e4SKonstantin Belousov return (0); 1219a4f67738SDoug Rabson } 1220a4f67738SDoug Rabson 122137c84183SPoul-Henning Kamp static int 1222d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym) 1223a4f67738SDoug Rabson { 1224326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 12252d636ab0SPeter Wemm const Elf_Sym *symp; 12262d636ab0SPeter Wemm const char *strp; 1227a4f67738SDoug Rabson int i; 1228a4f67738SDoug Rabson 12292d636ab0SPeter Wemm for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 12302d636ab0SPeter Wemm strp = ef->ddbstrtab + symp->st_name; 1231a8774e39SPeter Wemm if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) { 1232d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 12332d636ab0SPeter Wemm return 0; 12342d636ab0SPeter Wemm } 12352d636ab0SPeter Wemm } 1236a4f67738SDoug Rabson return ENOENT; 1237a4f67738SDoug Rabson } 1238a4f67738SDoug Rabson 1239de78ca7eSPeter Wemm static int 124020947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, 124120947801SPeter Wemm linker_symval_t *symval) 1242a4f67738SDoug Rabson { 12437035cf14SKonstantin Belousov elf_file_t ef; 12447035cf14SKonstantin Belousov const Elf_Sym *es; 12457035cf14SKonstantin Belousov caddr_t val; 1246a4f67738SDoug Rabson 12477035cf14SKonstantin Belousov ef = (elf_file_t) lf; 12487035cf14SKonstantin Belousov es = (const Elf_Sym*) sym; 12497035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 12505cf87418SMarcel Moolenaar if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) { 12512d636ab0SPeter Wemm symval->name = ef->ddbstrtab + es->st_name; 12527035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 12537035cf14SKonstantin Belousov if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC) 12547035cf14SKonstantin Belousov val = ((caddr_t (*)(void))val)(); 12557035cf14SKonstantin Belousov symval->value = val; 12562d636ab0SPeter Wemm symval->size = es->st_size; 12572d636ab0SPeter Wemm return 0; 12582d636ab0SPeter Wemm } 12592d636ab0SPeter Wemm return ENOENT; 12602d636ab0SPeter Wemm } 1261a4f67738SDoug Rabson 1262a4f67738SDoug Rabson static int 1263a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value, 1264d254af07SMatthew Dillon c_linker_sym_t *sym, long *diffp) 1265a4f67738SDoug Rabson { 1266326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1267586453feSBruce Evans u_long off = (uintptr_t) (void *) value; 1268a4f67738SDoug Rabson u_long diff = off; 1269586453feSBruce Evans u_long st_value; 1270fe3db7c7SDoug Rabson const Elf_Sym *es; 1271b85f65afSPedro F. Giffuni const Elf_Sym *best = NULL; 1272a4f67738SDoug Rabson int i; 1273a4f67738SDoug Rabson 12742d636ab0SPeter Wemm for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 1275a4f67738SDoug Rabson if (es->st_name == 0) 1276a4f67738SDoug Rabson continue; 1277a8774e39SPeter Wemm st_value = es->st_value; 1278b5abfb70SPeter Wemm if (off >= st_value) { 1279b5abfb70SPeter Wemm if (off - st_value < diff) { 1280b5abfb70SPeter Wemm diff = off - st_value; 1281a4f67738SDoug Rabson best = es; 1282a4f67738SDoug Rabson if (diff == 0) 1283a4f67738SDoug Rabson break; 1284b5abfb70SPeter Wemm } else if (off - st_value == diff) { 1285a4f67738SDoug Rabson best = es; 1286a4f67738SDoug Rabson } 1287a4f67738SDoug Rabson } 1288a4f67738SDoug Rabson } 1289b85f65afSPedro F. Giffuni if (best == NULL) 1290a4f67738SDoug Rabson *diffp = off; 1291a4f67738SDoug Rabson else 1292a4f67738SDoug Rabson *diffp = diff; 1293d254af07SMatthew Dillon *sym = (c_linker_sym_t) best; 1294a4f67738SDoug Rabson 1295a4f67738SDoug Rabson return 0; 1296a4f67738SDoug Rabson } 1297f41325dbSPeter Wemm 1298f41325dbSPeter Wemm /* 1299f41325dbSPeter Wemm * Look up a linker set on an ELF system. 1300f41325dbSPeter Wemm */ 1301f41325dbSPeter Wemm static int 1302f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name, 1303f41325dbSPeter Wemm void ***startp, void ***stopp, int *countp) 1304f41325dbSPeter Wemm { 1305e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 1306f41325dbSPeter Wemm void **start, **stop; 1307e9eabf59SPeter Wemm int i, count; 1308f41325dbSPeter Wemm 1309e9eabf59SPeter Wemm /* Relative to section number */ 1310e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1311e9eabf59SPeter Wemm if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) && 1312e9eabf59SPeter Wemm strcmp(ef->progtab[i].name + 4, name) == 0) { 1313e9eabf59SPeter Wemm start = (void **)ef->progtab[i].addr; 131420947801SPeter Wemm stop = (void **)((char *)ef->progtab[i].addr + 131520947801SPeter Wemm ef->progtab[i].size); 1316f41325dbSPeter Wemm count = stop - start; 1317f41325dbSPeter Wemm if (startp) 1318f41325dbSPeter Wemm *startp = start; 1319f41325dbSPeter Wemm if (stopp) 1320f41325dbSPeter Wemm *stopp = stop; 1321f41325dbSPeter Wemm if (countp) 1322f41325dbSPeter Wemm *countp = count; 1323e9eabf59SPeter Wemm return (0); 1324e9eabf59SPeter Wemm } 1325e9eabf59SPeter Wemm } 1326e9eabf59SPeter Wemm return (ESRCH); 1327f41325dbSPeter Wemm } 1328bb9fe9ddSBrian Feldman 1329bb9fe9ddSBrian Feldman static int 1330bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file, 1331e9eabf59SPeter Wemm int (*callback)(const char *, void *), void *opaque) 1332e9eabf59SPeter Wemm { 1333bb9fe9ddSBrian Feldman elf_file_t ef = (elf_file_t)file; 1334bb9fe9ddSBrian Feldman const Elf_Sym *symp; 1335bb9fe9ddSBrian Feldman int i, error; 1336bb9fe9ddSBrian Feldman 1337bb9fe9ddSBrian Feldman /* Exhaustive search */ 1338bb9fe9ddSBrian Feldman for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1339bb9fe9ddSBrian Feldman if (symp->st_value != 0 && 13407035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 13417035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 1342bb9fe9ddSBrian Feldman error = callback(ef->ddbstrtab + symp->st_name, opaque); 1343bb9fe9ddSBrian Feldman if (error) 1344bb9fe9ddSBrian Feldman return (error); 1345bb9fe9ddSBrian Feldman } 1346bb9fe9ddSBrian Feldman } 1347bb9fe9ddSBrian Feldman return (0); 1348bb9fe9ddSBrian Feldman } 134984201059SMarcel Moolenaar 1350a2024a3eSJohn Birrell static int 1351a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file, 1352a2024a3eSJohn Birrell linker_function_nameval_callback_t callback, void *opaque) 1353a2024a3eSJohn Birrell { 1354a2024a3eSJohn Birrell linker_symval_t symval; 1355a2024a3eSJohn Birrell elf_file_t ef = (elf_file_t)file; 1356a2024a3eSJohn Birrell const Elf_Sym* symp; 1357a2024a3eSJohn Birrell int i, error; 1358a2024a3eSJohn Birrell 1359a2024a3eSJohn Birrell /* Exhaustive search */ 1360a2024a3eSJohn Birrell for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1361a2024a3eSJohn Birrell if (symp->st_value != 0 && 13627035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 13637035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 13647035cf14SKonstantin Belousov error = link_elf_symbol_values(file, 13657035cf14SKonstantin Belousov (c_linker_sym_t)symp, &symval); 1366a2024a3eSJohn Birrell if (error) 1367a2024a3eSJohn Birrell return (error); 1368a2024a3eSJohn Birrell error = callback(file, i, &symval, opaque); 1369a2024a3eSJohn Birrell if (error) 1370a2024a3eSJohn Birrell return (error); 1371a2024a3eSJohn Birrell } 1372a2024a3eSJohn Birrell } 1373a2024a3eSJohn Birrell return (0); 1374a2024a3eSJohn Birrell } 1375a2024a3eSJohn Birrell 13762832cd54SKonstantin Belousov static void 13772832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef) 13782832cd54SKonstantin Belousov { 13792832cd54SKonstantin Belousov Elf_Sym *sym; 13802832cd54SKonstantin Belousov Elf_Size i; 13812832cd54SKonstantin Belousov 13822832cd54SKonstantin Belousov for (i = 0; i < ef->ddbsymcnt; i++) { 13832832cd54SKonstantin Belousov sym = ef->ddbsymtab + i; 13842832cd54SKonstantin Belousov if (sym->st_shndx == SHN_FBSD_CACHED) { 13852832cd54SKonstantin Belousov sym->st_shndx = SHN_UNDEF; 13862832cd54SKonstantin Belousov sym->st_value = 0; 13872832cd54SKonstantin Belousov } 13882832cd54SKonstantin Belousov } 13892832cd54SKonstantin Belousov } 13902832cd54SKonstantin Belousov 1391d297ad16SMarcel Moolenaar /* 1392d297ad16SMarcel Moolenaar * Symbol lookup function that can be used when the symbol index is known (ie 1393d297ad16SMarcel Moolenaar * in relocations). It uses the symbol index instead of doing a fully fledged 1394d297ad16SMarcel Moolenaar * hash table based lookup when such is valid. For example for local symbols. 1395d297ad16SMarcel Moolenaar * This is not only more efficient, it's also more correct. It's not always 1396d297ad16SMarcel Moolenaar * the case that the symbol can be found through the hash table. 1397d297ad16SMarcel Moolenaar */ 1398cff8c6f2SKonstantin Belousov static int 1399cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res) 1400d297ad16SMarcel Moolenaar { 1401d297ad16SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 14022832cd54SKonstantin Belousov Elf_Sym *sym; 1403d297ad16SMarcel Moolenaar const char *symbol; 1404cff8c6f2SKonstantin Belousov Elf_Addr res1; 1405d297ad16SMarcel Moolenaar 1406d297ad16SMarcel Moolenaar /* Don't even try to lookup the symbol if the index is bogus. */ 1407cff8c6f2SKonstantin Belousov if (symidx >= ef->ddbsymcnt) { 1408cff8c6f2SKonstantin Belousov *res = 0; 1409cff8c6f2SKonstantin Belousov return (EINVAL); 1410cff8c6f2SKonstantin Belousov } 1411d297ad16SMarcel Moolenaar 1412e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 1413d297ad16SMarcel Moolenaar 14144cec6f5dSPeter Wemm /* Quick answer if there is a definition included. */ 1415cff8c6f2SKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) { 1416289ead7cSKonstantin Belousov res1 = (Elf_Addr)sym->st_value; 1417289ead7cSKonstantin Belousov if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) 1418289ead7cSKonstantin Belousov res1 = ((Elf_Addr (*)(void))res1)(); 1419289ead7cSKonstantin Belousov *res = res1; 1420cff8c6f2SKonstantin Belousov return (0); 1421cff8c6f2SKonstantin Belousov } 14224cec6f5dSPeter Wemm 14234cec6f5dSPeter Wemm /* If we get here, then it is undefined and needs a lookup. */ 14244cec6f5dSPeter Wemm switch (ELF_ST_BIND(sym->st_info)) { 14254cec6f5dSPeter Wemm case STB_LOCAL: 14264cec6f5dSPeter Wemm /* Local, but undefined? huh? */ 1427cff8c6f2SKonstantin Belousov *res = 0; 1428cff8c6f2SKonstantin Belousov return (EINVAL); 1429d297ad16SMarcel Moolenaar 1430e9eabf59SPeter Wemm case STB_GLOBAL: 1431cff8c6f2SKonstantin Belousov case STB_WEAK: 1432e9eabf59SPeter Wemm /* Relative to Data or Function name */ 1433e9eabf59SPeter Wemm symbol = ef->ddbstrtab + sym->st_name; 1434d297ad16SMarcel Moolenaar 1435d297ad16SMarcel Moolenaar /* Force a lookup failure if the symbol name is bogus. */ 1436cff8c6f2SKonstantin Belousov if (*symbol == 0) { 1437cff8c6f2SKonstantin Belousov *res = 0; 1438cff8c6f2SKonstantin Belousov return (EINVAL); 1439cff8c6f2SKonstantin Belousov } 1440cff8c6f2SKonstantin Belousov res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps); 14412832cd54SKonstantin Belousov 14422832cd54SKonstantin Belousov /* 14432832cd54SKonstantin Belousov * Cache global lookups during module relocation. The failure 14442832cd54SKonstantin Belousov * case is particularly expensive for callers, who must scan 14452832cd54SKonstantin Belousov * through the entire globals table doing strcmp(). Cache to 14462832cd54SKonstantin Belousov * avoid doing such work repeatedly. 14472832cd54SKonstantin Belousov * 14482832cd54SKonstantin Belousov * After relocation is complete, undefined globals will be 14492832cd54SKonstantin Belousov * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(), 14502832cd54SKonstantin Belousov * above. 14512832cd54SKonstantin Belousov */ 1452cff8c6f2SKonstantin Belousov if (res1 != 0) { 14532832cd54SKonstantin Belousov sym->st_shndx = SHN_FBSD_CACHED; 1454cff8c6f2SKonstantin Belousov sym->st_value = res1; 1455cff8c6f2SKonstantin Belousov *res = res1; 1456e9eabf59SPeter Wemm return (0); 1457cff8c6f2SKonstantin Belousov } else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) { 1458cff8c6f2SKonstantin Belousov sym->st_value = 0; 1459cff8c6f2SKonstantin Belousov *res = 0; 1460cff8c6f2SKonstantin Belousov return (0); 1461cff8c6f2SKonstantin Belousov } 1462cff8c6f2SKonstantin Belousov return (EINVAL); 1463e9eabf59SPeter Wemm 1464e9eabf59SPeter Wemm default: 1465cff8c6f2SKonstantin Belousov return (EINVAL); 1466d297ad16SMarcel Moolenaar } 1467e9eabf59SPeter Wemm } 1468e9eabf59SPeter Wemm 14697251b4bfSJake Burkholder static void 14707226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef) 14717226306eSKonstantin Belousov { 14727226306eSKonstantin Belousov static const char startn[] = "__start_"; 14737226306eSKonstantin Belousov static const char stopn[] = "__stop_"; 14747226306eSKonstantin Belousov Elf_Sym *sym; 14757226306eSKonstantin Belousov const char *sym_name, *linkset_name; 14767226306eSKonstantin Belousov Elf_Addr startp, stopp; 14777226306eSKonstantin Belousov Elf_Size symidx; 14787226306eSKonstantin Belousov int start, i; 14797226306eSKonstantin Belousov 14807226306eSKonstantin Belousov startp = stopp = 0; 14817226306eSKonstantin Belousov for (symidx = 1 /* zero entry is special */; 14827226306eSKonstantin Belousov symidx < ef->ddbsymcnt; symidx++) { 14837226306eSKonstantin Belousov sym = ef->ddbsymtab + symidx; 14847226306eSKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) 14857226306eSKonstantin Belousov continue; 14867226306eSKonstantin Belousov 14877226306eSKonstantin Belousov sym_name = ef->ddbstrtab + sym->st_name; 14887226306eSKonstantin Belousov if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) { 14897226306eSKonstantin Belousov start = 1; 14907226306eSKonstantin Belousov linkset_name = sym_name + sizeof(startn) - 1; 14917226306eSKonstantin Belousov } 14927226306eSKonstantin Belousov else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) { 14937226306eSKonstantin Belousov start = 0; 14947226306eSKonstantin Belousov linkset_name = sym_name + sizeof(stopn) - 1; 14957226306eSKonstantin Belousov } 14967226306eSKonstantin Belousov else 14977226306eSKonstantin Belousov continue; 14987226306eSKonstantin Belousov 14997226306eSKonstantin Belousov for (i = 0; i < ef->nprogtab; i++) { 15007226306eSKonstantin Belousov if (strcmp(ef->progtab[i].name, linkset_name) == 0) { 15017226306eSKonstantin Belousov startp = (Elf_Addr)ef->progtab[i].addr; 15027226306eSKonstantin Belousov stopp = (Elf_Addr)(startp + ef->progtab[i].size); 15037226306eSKonstantin Belousov break; 15047226306eSKonstantin Belousov } 15057226306eSKonstantin Belousov } 15067226306eSKonstantin Belousov if (i == ef->nprogtab) 15077226306eSKonstantin Belousov continue; 15087226306eSKonstantin Belousov 15097226306eSKonstantin Belousov sym->st_value = start ? startp : stopp; 15107226306eSKonstantin Belousov sym->st_shndx = i; 15117226306eSKonstantin Belousov } 15127226306eSKonstantin Belousov } 15137226306eSKonstantin Belousov 15143cfce8e4SKonstantin Belousov static int 1515289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs) 15167251b4bfSJake Burkholder { 1517e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 15187251b4bfSJake Burkholder const Elf_Rel *rellim; 15197251b4bfSJake Burkholder const Elf_Rel *rel; 15207251b4bfSJake Burkholder const Elf_Rela *relalim; 15217251b4bfSJake Burkholder const Elf_Rela *rela; 1522e9eabf59SPeter Wemm const Elf_Sym *sym; 1523e9eabf59SPeter Wemm Elf_Addr base; 1524e9eabf59SPeter Wemm int i; 1525757686b1SMarcel Moolenaar Elf_Size symidx; 1526e9eabf59SPeter Wemm 15277226306eSKonstantin Belousov link_elf_fix_link_set(ef); 15287226306eSKonstantin Belousov 15297251b4bfSJake Burkholder /* Perform relocations without addend if there are any: */ 1530b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1531e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 15323cfce8e4SKonstantin Belousov if (rel == NULL) { 15333cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab"); 15343cfce8e4SKonstantin Belousov return (ENOEXEC); 15353cfce8e4SKonstantin Belousov } 153620947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1537e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 15383cfce8e4SKonstantin Belousov if (base == 0) { 15393cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 15403cfce8e4SKonstantin Belousov return (ENOEXEC); 15413cfce8e4SKonstantin Belousov } 154220947801SPeter Wemm for ( ; rel < rellim; rel++) { 1543e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 154420947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 154520947801SPeter Wemm continue; 1546e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 154720947801SPeter Wemm /* Only do local relocs */ 154820947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 154920947801SPeter Wemm continue; 155036716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 155136716fe2SMark Johnston elf_is_ifunc_reloc(rel->r_info)) == ifuncs) 155220947801SPeter Wemm elf_reloc_local(lf, base, rel, ELF_RELOC_REL, 155320947801SPeter Wemm elf_obj_lookup); 15547251b4bfSJake Burkholder } 1555e9eabf59SPeter Wemm } 15567251b4bfSJake Burkholder 15577251b4bfSJake Burkholder /* Perform relocations with addend if there are any: */ 1558b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1559e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 15603cfce8e4SKonstantin Belousov if (rela == NULL) { 15613cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 15623cfce8e4SKonstantin Belousov return (ENOEXEC); 15633cfce8e4SKonstantin Belousov } 156420947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1565e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 15663cfce8e4SKonstantin Belousov if (base == 0) { 15673cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 15683cfce8e4SKonstantin Belousov return (ENOEXEC); 15693cfce8e4SKonstantin Belousov } 157020947801SPeter Wemm for ( ; rela < relalim; rela++) { 1571e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 157220947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 157320947801SPeter Wemm continue; 1574e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 157520947801SPeter Wemm /* Only do local relocs */ 157620947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 157720947801SPeter Wemm continue; 157836716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 157936716fe2SMark Johnston elf_is_ifunc_reloc(rela->r_info)) == ifuncs) 158020947801SPeter Wemm elf_reloc_local(lf, base, rela, ELF_RELOC_RELA, 158120947801SPeter Wemm elf_obj_lookup); 15827251b4bfSJake Burkholder } 15837251b4bfSJake Burkholder } 15843cfce8e4SKonstantin Belousov return (0); 1585e9eabf59SPeter Wemm } 158600a5db46SStacey Son 158700a5db46SStacey Son static long 1588e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab) 158900a5db46SStacey Son { 159000a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 159100a5db46SStacey Son 159200a5db46SStacey Son *symtab = ef->ddbsymtab; 159300a5db46SStacey Son 159400a5db46SStacey Son if (*symtab == NULL) 159500a5db46SStacey Son return (0); 159600a5db46SStacey Son 159700a5db46SStacey Son return (ef->ddbsymcnt); 159800a5db46SStacey Son } 159900a5db46SStacey Son 160000a5db46SStacey Son static long 160100a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab) 160200a5db46SStacey Son { 160300a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 160400a5db46SStacey Son 160500a5db46SStacey Son *strtab = ef->ddbstrtab; 160600a5db46SStacey Son 160700a5db46SStacey Son if (*strtab == NULL) 160800a5db46SStacey Son return (0); 160900a5db46SStacey Son 161000a5db46SStacey Son return (ef->ddbstrcnt); 161100a5db46SStacey Son } 1612