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) { 37150c202c5SJeff Roberson error = ENOSPC; 37250c202c5SJeff Roberson goto out; 37350c202c5SJeff Roberson } 37450c202c5SJeff Roberson memcpy(dpcpu, ef->progtab[pb].addr, 37550c202c5SJeff Roberson ef->progtab[pb].size); 37650c202c5SJeff Roberson dpcpu_copy(dpcpu, shdr[i].sh_size); 37750c202c5SJeff Roberson ef->progtab[pb].addr = dpcpu; 378eddfbb76SRobert Watson #ifdef VIMAGE 379eddfbb76SRobert Watson } else if (ef->progtab[pb].name != NULL && 38017ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 381eddfbb76SRobert Watson void *vnet_data; 382eddfbb76SRobert Watson 383eddfbb76SRobert Watson vnet_data = vnet_data_alloc(shdr[i].sh_size); 384eddfbb76SRobert Watson if (vnet_data == NULL) { 385eddfbb76SRobert Watson error = ENOSPC; 386eddfbb76SRobert Watson goto out; 387eddfbb76SRobert Watson } 388eddfbb76SRobert Watson memcpy(vnet_data, ef->progtab[pb].addr, 389eddfbb76SRobert Watson ef->progtab[pb].size); 390eddfbb76SRobert Watson vnet_data_copy(vnet_data, shdr[i].sh_size); 391eddfbb76SRobert Watson ef->progtab[pb].addr = vnet_data; 392eddfbb76SRobert Watson #endif 3930067051fSMarcel Moolenaar } else if (ef->progtab[pb].name != NULL && 3940067051fSMarcel Moolenaar !strcmp(ef->progtab[pb].name, ".ctors")) { 3950067051fSMarcel Moolenaar lf->ctors_addr = ef->progtab[pb].addr; 3960067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 39750c202c5SJeff Roberson } 39870b7ffeeSIan Dowse 39970b7ffeeSIan Dowse /* Update all symbol values with the offset. */ 40070b7ffeeSIan Dowse for (j = 0; j < ef->ddbsymcnt; j++) { 40170b7ffeeSIan Dowse es = &ef->ddbsymtab[j]; 40270b7ffeeSIan Dowse if (es->st_shndx != i) 40370b7ffeeSIan Dowse continue; 40470b7ffeeSIan Dowse es->st_value += (Elf_Addr)ef->progtab[pb].addr; 40570b7ffeeSIan Dowse } 40670b7ffeeSIan Dowse pb++; 40770b7ffeeSIan Dowse break; 40870b7ffeeSIan Dowse case SHT_REL: 40915746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 41015746ef4SJohn Baldwin break; 41170b7ffeeSIan Dowse ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr; 41270b7ffeeSIan Dowse ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 41370b7ffeeSIan Dowse ef->reltab[rl].sec = shdr[i].sh_info; 41470b7ffeeSIan Dowse rl++; 41570b7ffeeSIan Dowse break; 41670b7ffeeSIan Dowse case SHT_RELA: 41715746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 41815746ef4SJohn Baldwin break; 41970b7ffeeSIan Dowse ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr; 42070b7ffeeSIan Dowse ef->relatab[ra].nrela = 42170b7ffeeSIan Dowse shdr[i].sh_size / sizeof(Elf_Rela); 42270b7ffeeSIan Dowse ef->relatab[ra].sec = shdr[i].sh_info; 42370b7ffeeSIan Dowse ra++; 42470b7ffeeSIan Dowse break; 42570b7ffeeSIan Dowse } 42670b7ffeeSIan Dowse } 4273cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 4283cfce8e4SKonstantin Belousov printf("%s: lost progbits\n", filename); 4293cfce8e4SKonstantin Belousov error = ENOEXEC; 4303cfce8e4SKonstantin Belousov goto out; 4313cfce8e4SKonstantin Belousov } 4323cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 4333cfce8e4SKonstantin Belousov printf("%s: lost reltab\n", filename); 4343cfce8e4SKonstantin Belousov error = ENOEXEC; 4353cfce8e4SKonstantin Belousov goto out; 4363cfce8e4SKonstantin Belousov } 4373cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 4383cfce8e4SKonstantin Belousov printf("%s: lost relatab\n", filename); 4393cfce8e4SKonstantin Belousov error = ENOEXEC; 4403cfce8e4SKonstantin Belousov goto out; 4413cfce8e4SKonstantin Belousov } 44270b7ffeeSIan Dowse 44370b7ffeeSIan Dowse /* Local intra-module relocations */ 444289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 4453cfce8e4SKonstantin Belousov if (error != 0) 4463cfce8e4SKonstantin Belousov goto out; 44770b7ffeeSIan Dowse *result = lf; 44870b7ffeeSIan Dowse return (0); 44970b7ffeeSIan Dowse 45070b7ffeeSIan Dowse out: 45170b7ffeeSIan Dowse /* preload not done this way */ 45270b7ffeeSIan Dowse linker_file_unload(lf, LINKER_UNLOAD_FORCE); 45370b7ffeeSIan Dowse return (error); 45470b7ffeeSIan Dowse } 455de78ca7eSPeter Wemm 4560067051fSMarcel Moolenaar static void 4570067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size) 4580067051fSMarcel Moolenaar { 4590067051fSMarcel Moolenaar void (**ctor)(void); 4600067051fSMarcel Moolenaar size_t i, cnt; 4610067051fSMarcel Moolenaar 4620067051fSMarcel Moolenaar if (addr == NULL || size == 0) 4630067051fSMarcel Moolenaar return; 4640067051fSMarcel Moolenaar cnt = size / sizeof(*ctor); 4650067051fSMarcel Moolenaar ctor = (void *)addr; 4660067051fSMarcel Moolenaar for (i = 0; i < cnt; i++) { 4670067051fSMarcel Moolenaar if (ctor[i] != NULL) 4680067051fSMarcel Moolenaar (*ctor[i])(); 4690067051fSMarcel Moolenaar } 4700067051fSMarcel Moolenaar } 4710067051fSMarcel Moolenaar 472de78ca7eSPeter Wemm static int 47354823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf) 47454823af2SPeter Wemm { 47570b7ffeeSIan Dowse elf_file_t ef; 47670b7ffeeSIan Dowse int error; 47770b7ffeeSIan Dowse 47870b7ffeeSIan Dowse ef = (elf_file_t)lf; 47970b7ffeeSIan Dowse error = relocate_file(ef); 48070b7ffeeSIan Dowse if (error) 481289ead7cSKonstantin Belousov return (error); 48270b7ffeeSIan Dowse 48370b7ffeeSIan Dowse /* Notify MD code that a module is being loaded. */ 48470b7ffeeSIan Dowse error = elf_cpu_load_file(lf); 48570b7ffeeSIan Dowse if (error) 48670b7ffeeSIan Dowse return (error); 48770b7ffeeSIan Dowse 488*8f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 489289ead7cSKonstantin Belousov /* Now ifuncs. */ 490289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 491289ead7cSKonstantin Belousov if (error != 0) 492289ead7cSKonstantin Belousov return (error); 493*8f941950SKonstantin Belousov #endif 494289ead7cSKonstantin Belousov 4950067051fSMarcel Moolenaar /* Invoke .ctors */ 4960067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 49770b7ffeeSIan Dowse return (0); 49854823af2SPeter Wemm } 49954823af2SPeter Wemm 50054823af2SPeter Wemm static int 501c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename, 502c143d6c2SMarcel Moolenaar linker_file_t *result) 503a4f67738SDoug Rabson { 504a344babfSGleb Smirnoff struct nameidata *nd; 505b40ce416SJulian Elischer struct thread *td = curthread; /* XXX */ 506caab6e90SPeter Wemm Elf_Ehdr *hdr; 507e9eabf59SPeter Wemm Elf_Shdr *shdr; 508a8774e39SPeter Wemm Elf_Sym *es; 509a8774e39SPeter Wemm int nbytes, i, j; 51020947801SPeter Wemm vm_offset_t mapbase; 511fe3db7c7SDoug Rabson size_t mapsize; 512a4f67738SDoug Rabson int error = 0; 513526d0bd5SKonstantin Belousov ssize_t resid; 514526d0bd5SKonstantin Belousov int flags; 515a4f67738SDoug Rabson elf_file_t ef; 516a4f67738SDoug Rabson linker_file_t lf; 517ca65d5c7SPeter Wemm int symtabindex; 518ca65d5c7SPeter Wemm int symstrindex; 519e9eabf59SPeter Wemm int shstrindex; 520e9eabf59SPeter Wemm int nsym; 52120947801SPeter Wemm int pb, rl, ra; 522e9eabf59SPeter Wemm int alignmask; 523ca65d5c7SPeter Wemm 524ca65d5c7SPeter Wemm shdr = NULL; 525ca65d5c7SPeter Wemm lf = NULL; 526e9eabf59SPeter Wemm mapsize = 0; 527e9eabf59SPeter Wemm hdr = NULL; 528a4f67738SDoug Rabson 529a344babfSGleb Smirnoff nd = malloc(sizeof(struct nameidata), M_TEMP, M_WAITOK); 530a344babfSGleb Smirnoff NDINIT(nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td); 531e6796b67SKirk McKusick flags = FREAD; 532a344babfSGleb Smirnoff error = vn_open(nd, &flags, 0, NULL); 533a344babfSGleb Smirnoff if (error) { 534a344babfSGleb Smirnoff free(nd, M_TEMP); 535a4f67738SDoug Rabson return error; 536a344babfSGleb Smirnoff } 537a344babfSGleb Smirnoff NDFREE(nd, NDF_ONLY_PNBUF); 538a344babfSGleb Smirnoff if (nd->ni_vp->v_type != VREG) { 5391f49b573SRuslan Ermilov error = ENOEXEC; 5401f49b573SRuslan Ermilov goto out; 5411f49b573SRuslan Ermilov } 542a3df768bSRobert Watson #ifdef MAC 543a344babfSGleb Smirnoff error = mac_kld_check_load(td->td_ucred, nd->ni_vp); 544a3df768bSRobert Watson if (error) { 545a3df768bSRobert Watson goto out; 546a3df768bSRobert Watson } 547a3df768bSRobert Watson #endif 548a4f67738SDoug Rabson 549e9eabf59SPeter Wemm /* Read the elf header from the file. */ 550e9eabf59SPeter Wemm hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK); 551a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)hdr, sizeof(*hdr), 0, 5529ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 5539ca43589SRobert Watson &resid, td); 554a4f67738SDoug Rabson if (error) 555a4f67738SDoug Rabson goto out; 556e9eabf59SPeter Wemm if (resid != 0){ 557e9eabf59SPeter Wemm error = ENOEXEC; 558e9eabf59SPeter Wemm goto out; 559e9eabf59SPeter Wemm } 560a4f67738SDoug Rabson 561caab6e90SPeter Wemm if (!IS_ELF(*hdr)) { 562fe3db7c7SDoug Rabson error = ENOEXEC; 563a4f67738SDoug Rabson goto out; 564fe3db7c7SDoug Rabson } 565fe3db7c7SDoug Rabson 566caab6e90SPeter Wemm if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS 567caab6e90SPeter Wemm || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 568552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file layout"); 569fe3db7c7SDoug Rabson error = ENOEXEC; 570fe3db7c7SDoug Rabson goto out; 571fe3db7c7SDoug Rabson } 572caab6e90SPeter Wemm if (hdr->e_ident[EI_VERSION] != EV_CURRENT 573caab6e90SPeter Wemm || hdr->e_version != EV_CURRENT) { 574552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file version"); 575fe3db7c7SDoug Rabson error = ENOEXEC; 576fe3db7c7SDoug Rabson goto out; 577fe3db7c7SDoug Rabson } 578e9eabf59SPeter Wemm if (hdr->e_type != ET_REL) { 579a1d7ce03SAttilio Rao error = ENOSYS; 580fe3db7c7SDoug Rabson goto out; 581fe3db7c7SDoug Rabson } 582caab6e90SPeter Wemm if (hdr->e_machine != ELF_TARG_MACH) { 583552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported machine"); 584fe3db7c7SDoug Rabson error = ENOEXEC; 585fe3db7c7SDoug Rabson goto out; 586fe3db7c7SDoug Rabson } 587a4f67738SDoug Rabson 588326e27d8SDoug Rabson lf = linker_make_file(filename, &link_elf_class); 589326e27d8SDoug Rabson if (!lf) { 590326e27d8SDoug Rabson error = ENOMEM; 591326e27d8SDoug Rabson goto out; 592326e27d8SDoug Rabson } 593326e27d8SDoug Rabson ef = (elf_file_t) lf; 594e9eabf59SPeter Wemm ef->nprogtab = 0; 595e9eabf59SPeter Wemm ef->e_shdr = 0; 596b474c780SDavid E. O'Brien ef->nreltab = 0; 597b474c780SDavid E. O'Brien ef->nrelatab = 0; 598e9eabf59SPeter Wemm 599e9eabf59SPeter Wemm /* Allocate and read in the section header */ 600e9eabf59SPeter Wemm nbytes = hdr->e_shnum * hdr->e_shentsize; 601e9eabf59SPeter Wemm if (nbytes == 0 || hdr->e_shoff == 0 || 602e9eabf59SPeter Wemm hdr->e_shentsize != sizeof(Elf_Shdr)) { 603e9eabf59SPeter Wemm error = ENOEXEC; 604e9eabf59SPeter Wemm goto out; 605e9eabf59SPeter Wemm } 60620947801SPeter Wemm shdr = malloc(nbytes, M_LINKER, M_WAITOK); 607e9eabf59SPeter Wemm ef->e_shdr = shdr; 608a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (caddr_t)shdr, nbytes, 609a344babfSGleb Smirnoff hdr->e_shoff, UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 610a344babfSGleb Smirnoff NOCRED, &resid, td); 611e9eabf59SPeter Wemm if (error) 612e9eabf59SPeter Wemm goto out; 613e9eabf59SPeter Wemm if (resid) { 614e9eabf59SPeter Wemm error = ENOEXEC; 615e9eabf59SPeter Wemm goto out; 616e9eabf59SPeter Wemm } 617e9eabf59SPeter Wemm 618e9eabf59SPeter Wemm /* Scan the section header for information and table sizing. */ 619e9eabf59SPeter Wemm nsym = 0; 620e9eabf59SPeter Wemm symtabindex = -1; 621e9eabf59SPeter Wemm symstrindex = -1; 622e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 623676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 624676799a0SAndriy Gapon continue; 625e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 626e9eabf59SPeter Wemm case SHT_PROGBITS: 627e9eabf59SPeter Wemm case SHT_NOBITS: 62813f28d96SKonstantin Belousov #ifdef __amd64__ 629b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 63013f28d96SKonstantin Belousov #endif 63158c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 63258c4aee0SJohn Baldwin break; 63320947801SPeter Wemm ef->nprogtab++; 634e9eabf59SPeter Wemm break; 635e9eabf59SPeter Wemm case SHT_SYMTAB: 636e9eabf59SPeter Wemm nsym++; 637e9eabf59SPeter Wemm symtabindex = i; 638e9eabf59SPeter Wemm symstrindex = shdr[i].sh_link; 639e9eabf59SPeter Wemm break; 640e9eabf59SPeter Wemm case SHT_REL: 64115746ef4SJohn Baldwin /* 64215746ef4SJohn Baldwin * Ignore relocation tables for unallocated 64315746ef4SJohn Baldwin * sections. 64415746ef4SJohn Baldwin */ 64515746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 64615746ef4SJohn Baldwin break; 647b474c780SDavid E. O'Brien ef->nreltab++; 648e9eabf59SPeter Wemm break; 649e9eabf59SPeter Wemm case SHT_RELA: 65015746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 65115746ef4SJohn Baldwin break; 652b474c780SDavid E. O'Brien ef->nrelatab++; 653e9eabf59SPeter Wemm break; 654e9eabf59SPeter Wemm case SHT_STRTAB: 655e9eabf59SPeter Wemm break; 656e9eabf59SPeter Wemm } 657e9eabf59SPeter Wemm } 65820947801SPeter Wemm if (ef->nprogtab == 0) { 659552f9f63SEdwin Groothuis link_elf_error(filename, "file has no contents"); 660e9eabf59SPeter Wemm error = ENOEXEC; 661e9eabf59SPeter Wemm goto out; 662e9eabf59SPeter Wemm } 663e9eabf59SPeter Wemm if (nsym != 1) { 664e9eabf59SPeter Wemm /* Only allow one symbol table for now */ 665f912a970SEd Maste link_elf_error(filename, 666f912a970SEd Maste "file must have exactly one symbol table"); 667e9eabf59SPeter Wemm error = ENOEXEC; 668e9eabf59SPeter Wemm goto out; 669e9eabf59SPeter Wemm } 670e9eabf59SPeter Wemm if (symstrindex < 0 || symstrindex > hdr->e_shnum || 671e9eabf59SPeter Wemm shdr[symstrindex].sh_type != SHT_STRTAB) { 672552f9f63SEdwin Groothuis link_elf_error(filename, "file has invalid symbol strings"); 673e9eabf59SPeter Wemm error = ENOEXEC; 674e9eabf59SPeter Wemm goto out; 675e9eabf59SPeter Wemm } 676e9eabf59SPeter Wemm 677e9eabf59SPeter Wemm /* Allocate space for tracking the load chunks */ 678e9eabf59SPeter Wemm if (ef->nprogtab != 0) 67920947801SPeter Wemm ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 68020947801SPeter Wemm M_LINKER, M_WAITOK | M_ZERO); 681b474c780SDavid E. O'Brien if (ef->nreltab != 0) 682b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 683b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 684b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 685b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 686b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 687e9eabf59SPeter Wemm 6883cfce8e4SKonstantin Belousov if (symtabindex == -1) { 6893cfce8e4SKonstantin Belousov link_elf_error(filename, "lost symbol table index"); 6903cfce8e4SKonstantin Belousov error = ENOEXEC; 6913cfce8e4SKonstantin Belousov goto out; 6923cfce8e4SKonstantin Belousov } 69320947801SPeter Wemm /* Allocate space for and load the symbol table */ 694e9eabf59SPeter Wemm ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 695e9eabf59SPeter Wemm ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK); 696a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)ef->ddbsymtab, 69720947801SPeter Wemm shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset, 69820947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 69920947801SPeter Wemm &resid, td); 70020947801SPeter Wemm if (error) 70120947801SPeter Wemm goto out; 70220947801SPeter Wemm if (resid != 0){ 70320947801SPeter Wemm error = EINVAL; 70420947801SPeter Wemm goto out; 70520947801SPeter Wemm } 706e9eabf59SPeter Wemm 7073cfce8e4SKonstantin Belousov if (symstrindex == -1) { 7083cfce8e4SKonstantin Belousov link_elf_error(filename, "lost symbol string index"); 7093cfce8e4SKonstantin Belousov error = ENOEXEC; 7103cfce8e4SKonstantin Belousov goto out; 7113cfce8e4SKonstantin Belousov } 71220947801SPeter Wemm /* Allocate space for and load the symbol strings */ 713e9eabf59SPeter Wemm ef->ddbstrcnt = shdr[symstrindex].sh_size; 714e9eabf59SPeter Wemm ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK); 715a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab, 71620947801SPeter Wemm shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset, 71720947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 71820947801SPeter Wemm &resid, td); 71920947801SPeter Wemm if (error) 72020947801SPeter Wemm goto out; 72120947801SPeter Wemm if (resid != 0){ 72220947801SPeter Wemm error = EINVAL; 72320947801SPeter Wemm goto out; 72420947801SPeter Wemm } 725e9eabf59SPeter Wemm 726e9eabf59SPeter Wemm /* Do we have a string table for the section names? */ 727e9eabf59SPeter Wemm shstrindex = -1; 72820947801SPeter Wemm if (hdr->e_shstrndx != 0 && 72920947801SPeter Wemm shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) { 730e9eabf59SPeter Wemm shstrindex = hdr->e_shstrndx; 731e9eabf59SPeter Wemm ef->shstrcnt = shdr[shstrindex].sh_size; 73220947801SPeter Wemm ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER, 73320947801SPeter Wemm M_WAITOK); 734a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab, 73520947801SPeter Wemm shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset, 736e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 737e9eabf59SPeter Wemm &resid, td); 738e9eabf59SPeter Wemm if (error) 739e9eabf59SPeter Wemm goto out; 74020947801SPeter Wemm if (resid != 0){ 74120947801SPeter Wemm error = EINVAL; 74220947801SPeter Wemm goto out; 74320947801SPeter Wemm } 744e9eabf59SPeter Wemm } 745e9eabf59SPeter Wemm 74620947801SPeter Wemm /* Size up code/data(progbits) and bss(nobits). */ 747e9eabf59SPeter Wemm alignmask = 0; 748e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 749676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 750676799a0SAndriy Gapon continue; 751e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 752e9eabf59SPeter Wemm case SHT_PROGBITS: 753e9eabf59SPeter Wemm case SHT_NOBITS: 75413f28d96SKonstantin Belousov #ifdef __amd64__ 755b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 75613f28d96SKonstantin Belousov #endif 75758c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 75858c4aee0SJohn Baldwin break; 759e9eabf59SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 760e9eabf59SPeter Wemm mapsize += alignmask; 761e9eabf59SPeter Wemm mapsize &= ~alignmask; 762e9eabf59SPeter Wemm mapsize += shdr[i].sh_size; 763e9eabf59SPeter Wemm break; 764e9eabf59SPeter Wemm } 765e9eabf59SPeter Wemm } 766e9eabf59SPeter Wemm 767e9eabf59SPeter Wemm /* 768e9eabf59SPeter Wemm * We know how much space we need for the text/data/bss/etc. 769e9eabf59SPeter Wemm * This stuff needs to be in a single chunk so that profiling etc 770e9eabf59SPeter Wemm * can get the bounds and gdb can associate offsets with modules 771e9eabf59SPeter Wemm */ 77220947801SPeter Wemm ef->object = vm_object_allocate(OBJT_DEFAULT, 77320947801SPeter Wemm round_page(mapsize) >> PAGE_SHIFT); 774fe3db7c7SDoug Rabson if (ef->object == NULL) { 775fe3db7c7SDoug Rabson error = ENOMEM; 776fe3db7c7SDoug Rabson goto out; 777fe3db7c7SDoug Rabson } 778fe3db7c7SDoug Rabson ef->address = (caddr_t) vm_map_min(kernel_map); 779ac68d1c9SAlan Cox 780ac68d1c9SAlan Cox /* 781ac68d1c9SAlan Cox * In order to satisfy amd64's architectural requirements on the 782ac68d1c9SAlan Cox * location of code and data in the kernel's address space, request a 783ac68d1c9SAlan Cox * mapping that is above the kernel. 784ac68d1c9SAlan Cox */ 7854104e835SOleksandr Tymoshenko #ifdef __amd64__ 786ac68d1c9SAlan Cox mapbase = KERNBASE; 7874104e835SOleksandr Tymoshenko #else 7884104e835SOleksandr Tymoshenko mapbase = VM_MIN_KERNEL_ADDRESS; 7894104e835SOleksandr Tymoshenko #endif 79020947801SPeter Wemm error = vm_map_find(kernel_map, ef->object, 0, &mapbase, 791edb572a3SJohn Baldwin round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL, 792edb572a3SJohn Baldwin VM_PROT_ALL, 0); 793fe3db7c7SDoug Rabson if (error) { 794fe3db7c7SDoug Rabson vm_object_deallocate(ef->object); 795326e27d8SDoug Rabson ef->object = 0; 796fe3db7c7SDoug Rabson goto out; 797fe3db7c7SDoug Rabson } 79820947801SPeter Wemm 799e9eabf59SPeter Wemm /* Wire the pages */ 8007f1ef325SAlan Cox error = vm_map_wire(kernel_map, mapbase, 80120947801SPeter Wemm mapbase + round_page(mapsize), 802e9eabf59SPeter Wemm VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES); 8037f1ef325SAlan Cox if (error != KERN_SUCCESS) { 8047f1ef325SAlan Cox error = ENOMEM; 8057f1ef325SAlan Cox goto out; 8067f1ef325SAlan Cox } 807a4f67738SDoug Rabson 80820947801SPeter Wemm /* Inform the kld system about the situation */ 80920947801SPeter Wemm lf->address = ef->address = (caddr_t)mapbase; 81020947801SPeter Wemm lf->size = mapsize; 811e9eabf59SPeter Wemm 812e9eabf59SPeter Wemm /* 81320947801SPeter Wemm * Now load code/data(progbits), zero bss(nobits), allocate space for 81420947801SPeter Wemm * and load relocs 815e9eabf59SPeter Wemm */ 81620947801SPeter Wemm pb = 0; 81720947801SPeter Wemm rl = 0; 81820947801SPeter Wemm ra = 0; 81920947801SPeter Wemm alignmask = 0; 82020947801SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 821676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 822676799a0SAndriy Gapon continue; 82320947801SPeter Wemm switch (shdr[i].sh_type) { 82420947801SPeter Wemm case SHT_PROGBITS: 82520947801SPeter Wemm case SHT_NOBITS: 82613f28d96SKonstantin Belousov #ifdef __amd64__ 827b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 82813f28d96SKonstantin Belousov #endif 82958c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 83058c4aee0SJohn Baldwin break; 83120947801SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 83220947801SPeter Wemm mapbase += alignmask; 83320947801SPeter Wemm mapbase &= ~alignmask; 8340067051fSMarcel Moolenaar if (ef->shstrtab != NULL && shdr[i].sh_name != 0) { 83550c202c5SJeff Roberson ef->progtab[pb].name = 83650c202c5SJeff Roberson ef->shstrtab + shdr[i].sh_name; 8370067051fSMarcel Moolenaar if (!strcmp(ef->progtab[pb].name, ".ctors")) { 8380067051fSMarcel Moolenaar lf->ctors_addr = (caddr_t)mapbase; 8390067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 8400067051fSMarcel Moolenaar } 8410067051fSMarcel Moolenaar } else if (shdr[i].sh_type == SHT_PROGBITS) 84220947801SPeter Wemm ef->progtab[pb].name = "<<PROGBITS>>"; 84313f28d96SKonstantin Belousov #ifdef __amd64__ 844b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 84513f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 84613f28d96SKonstantin Belousov #endif 84750c202c5SJeff Roberson else 84850c202c5SJeff Roberson ef->progtab[pb].name = "<<NOBITS>>"; 84950c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 8505f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 85150c202c5SJeff Roberson ef->progtab[pb].addr = 85250c202c5SJeff Roberson dpcpu_alloc(shdr[i].sh_size); 853eddfbb76SRobert Watson #ifdef VIMAGE 854eddfbb76SRobert Watson else if (ef->progtab[pb].name != NULL && 85517ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 856eddfbb76SRobert Watson ef->progtab[pb].addr = 857eddfbb76SRobert Watson vnet_data_alloc(shdr[i].sh_size); 858eddfbb76SRobert Watson #endif 85950c202c5SJeff Roberson else 86050c202c5SJeff Roberson ef->progtab[pb].addr = 86150c202c5SJeff Roberson (void *)(uintptr_t)mapbase; 86250c202c5SJeff Roberson if (ef->progtab[pb].addr == NULL) { 86350c202c5SJeff Roberson error = ENOSPC; 86450c202c5SJeff Roberson goto out; 86550c202c5SJeff Roberson } 86650c202c5SJeff Roberson ef->progtab[pb].size = shdr[i].sh_size; 86750c202c5SJeff Roberson ef->progtab[pb].sec = i; 86813f28d96SKonstantin Belousov if (shdr[i].sh_type == SHT_PROGBITS 86913f28d96SKonstantin Belousov #ifdef __amd64__ 870b715d9afSKonstantin Belousov || shdr[i].sh_type == SHT_X86_64_UNWIND 87113f28d96SKonstantin Belousov #endif 87213f28d96SKonstantin Belousov ) { 873a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 87420947801SPeter Wemm ef->progtab[pb].addr, 87520947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 87620947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 87720947801SPeter Wemm NOCRED, &resid, td); 87820947801SPeter Wemm if (error) 87920947801SPeter Wemm goto out; 88020947801SPeter Wemm if (resid != 0){ 88120947801SPeter Wemm error = EINVAL; 88220947801SPeter Wemm goto out; 88320947801SPeter Wemm } 884eddfbb76SRobert Watson /* Initialize the per-cpu or vnet area. */ 885eddfbb76SRobert Watson if (ef->progtab[pb].addr != (void *)mapbase && 8865f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 88750c202c5SJeff Roberson dpcpu_copy(ef->progtab[pb].addr, 88850c202c5SJeff Roberson shdr[i].sh_size); 889eddfbb76SRobert Watson #ifdef VIMAGE 890eddfbb76SRobert Watson else if (ef->progtab[pb].addr != 891eddfbb76SRobert Watson (void *)mapbase && 89217ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 893eddfbb76SRobert Watson vnet_data_copy(ef->progtab[pb].addr, 894eddfbb76SRobert Watson shdr[i].sh_size); 895eddfbb76SRobert Watson #endif 89650c202c5SJeff Roberson } else 89720947801SPeter Wemm bzero(ef->progtab[pb].addr, shdr[i].sh_size); 898a8774e39SPeter Wemm 899a8774e39SPeter Wemm /* Update all symbol values with the offset. */ 900a8774e39SPeter Wemm for (j = 0; j < ef->ddbsymcnt; j++) { 901a8774e39SPeter Wemm es = &ef->ddbsymtab[j]; 902a8774e39SPeter Wemm if (es->st_shndx != i) 903a8774e39SPeter Wemm continue; 904a8774e39SPeter Wemm es->st_value += (Elf_Addr)ef->progtab[pb].addr; 905a8774e39SPeter Wemm } 90620947801SPeter Wemm mapbase += shdr[i].sh_size; 90720947801SPeter Wemm pb++; 90820947801SPeter Wemm break; 90920947801SPeter Wemm case SHT_REL: 91015746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 91115746ef4SJohn Baldwin break; 91220947801SPeter Wemm ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER, 91320947801SPeter Wemm M_WAITOK); 91420947801SPeter Wemm ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 91520947801SPeter Wemm ef->reltab[rl].sec = shdr[i].sh_info; 916a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 91720947801SPeter Wemm (void *)ef->reltab[rl].rel, 91820947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 919e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 920e9eabf59SPeter Wemm &resid, td); 921e9eabf59SPeter Wemm if (error) 922e9eabf59SPeter Wemm goto out; 92320947801SPeter Wemm if (resid != 0){ 92420947801SPeter Wemm error = EINVAL; 92520947801SPeter Wemm goto out; 926e9eabf59SPeter Wemm } 92720947801SPeter Wemm rl++; 92820947801SPeter Wemm break; 92920947801SPeter Wemm case SHT_RELA: 93015746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 93115746ef4SJohn Baldwin break; 93220947801SPeter Wemm ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER, 93320947801SPeter Wemm M_WAITOK); 93420947801SPeter Wemm ef->relatab[ra].nrela = 93520947801SPeter Wemm shdr[i].sh_size / sizeof(Elf_Rela); 93620947801SPeter Wemm ef->relatab[ra].sec = shdr[i].sh_info; 937a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 93820947801SPeter Wemm (void *)ef->relatab[ra].rela, 93920947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 940e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 941e9eabf59SPeter Wemm &resid, td); 942e9eabf59SPeter Wemm if (error) 943e9eabf59SPeter Wemm goto out; 94420947801SPeter Wemm if (resid != 0){ 94520947801SPeter Wemm error = EINVAL; 94620947801SPeter Wemm goto out; 947e9eabf59SPeter Wemm } 94820947801SPeter Wemm ra++; 94920947801SPeter Wemm break; 95020947801SPeter Wemm } 95120947801SPeter Wemm } 9523cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 9533cfce8e4SKonstantin Belousov link_elf_error(filename, "lost progbits"); 9543cfce8e4SKonstantin Belousov error = ENOEXEC; 9553cfce8e4SKonstantin Belousov goto out; 9563cfce8e4SKonstantin Belousov } 9573cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 9583cfce8e4SKonstantin Belousov link_elf_error(filename, "lost reltab"); 9593cfce8e4SKonstantin Belousov error = ENOEXEC; 9603cfce8e4SKonstantin Belousov goto out; 9613cfce8e4SKonstantin Belousov } 9623cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 9633cfce8e4SKonstantin Belousov link_elf_error(filename, "lost relatab"); 9643cfce8e4SKonstantin Belousov error = ENOEXEC; 9653cfce8e4SKonstantin Belousov goto out; 9663cfce8e4SKonstantin Belousov } 9673cfce8e4SKonstantin Belousov if (mapbase != (vm_offset_t)ef->address + mapsize) { 9683cfce8e4SKonstantin Belousov printf( 9693cfce8e4SKonstantin Belousov "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n", 9703cfce8e4SKonstantin Belousov filename != NULL ? filename : "<none>", 9716f3c6327SNeel Natu (u_long)mapbase, ef->address, (u_long)mapsize, 9726f3c6327SNeel Natu (u_long)(vm_offset_t)ef->address + mapsize); 9733cfce8e4SKonstantin Belousov error = ENOMEM; 9743cfce8e4SKonstantin Belousov goto out; 9753cfce8e4SKonstantin Belousov } 976e9eabf59SPeter Wemm 977e9eabf59SPeter Wemm /* Local intra-module relocations */ 978289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 9793cfce8e4SKonstantin Belousov if (error != 0) 9803cfce8e4SKonstantin Belousov goto out; 9817251b4bfSJake Burkholder 982e9eabf59SPeter Wemm /* Pull in dependencies */ 983a344babfSGleb Smirnoff VOP_UNLOCK(nd->ni_vp, 0); 9847b9716baSIan Dowse error = linker_load_dependencies(lf); 985a344babfSGleb Smirnoff vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY); 986ca65d5c7SPeter Wemm if (error) 987de78ca7eSPeter Wemm goto out; 988e9eabf59SPeter Wemm 989e9eabf59SPeter Wemm /* External relocations */ 990326e27d8SDoug Rabson error = relocate_file(ef); 991ca65d5c7SPeter Wemm if (error) 992ca65d5c7SPeter Wemm goto out; 993ca65d5c7SPeter Wemm 994e9eabf59SPeter Wemm /* Notify MD code that a module is being loaded. */ 995e9eabf59SPeter Wemm error = elf_cpu_load_file(lf); 996ca65d5c7SPeter Wemm if (error) 997ca65d5c7SPeter Wemm goto out; 998a4f67738SDoug Rabson 999*8f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 1000289ead7cSKonstantin Belousov /* Now ifuncs. */ 1001289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 1002289ead7cSKonstantin Belousov if (error != 0) 1003289ead7cSKonstantin Belousov goto out; 1004*8f941950SKonstantin Belousov #endif 1005289ead7cSKonstantin Belousov 10060067051fSMarcel Moolenaar /* Invoke .ctors */ 10070067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 10080067051fSMarcel Moolenaar 1009a4f67738SDoug Rabson *result = lf; 1010a4f67738SDoug Rabson 1011a4f67738SDoug Rabson out: 1012a344babfSGleb Smirnoff VOP_UNLOCK(nd->ni_vp, 0); 1013a344babfSGleb Smirnoff vn_close(nd->ni_vp, FREAD, td->td_ucred, td); 1014a344babfSGleb Smirnoff free(nd, M_TEMP); 1015ca65d5c7SPeter Wemm if (error && lf) 101665a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 1017e9eabf59SPeter Wemm free(hdr, M_LINKER); 1018a4f67738SDoug Rabson 1019a4f67738SDoug Rabson return error; 1020a4f67738SDoug Rabson } 1021a4f67738SDoug Rabson 1022a4f67738SDoug Rabson static void 1023de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file) 1024a4f67738SDoug Rabson { 1025326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) file; 1026d821d364SPedro F. Giffuni u_int i; 1027a4f67738SDoug Rabson 10281aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being unloaded. */ 10291aeb23cdSMarcel Moolenaar elf_cpu_unload_file(file); 10301aeb23cdSMarcel Moolenaar 103150c202c5SJeff Roberson if (ef->progtab) { 103250c202c5SJeff Roberson for (i = 0; i < ef->nprogtab; i++) { 103350c202c5SJeff Roberson if (ef->progtab[i].size == 0) 103450c202c5SJeff Roberson continue; 103550c202c5SJeff Roberson if (ef->progtab[i].name == NULL) 103650c202c5SJeff Roberson continue; 10375f67450dSDimitry Andric if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME)) 103850c202c5SJeff Roberson dpcpu_free(ef->progtab[i].addr, 103950c202c5SJeff Roberson ef->progtab[i].size); 1040eddfbb76SRobert Watson #ifdef VIMAGE 104117ef1febSRobert Watson else if (!strcmp(ef->progtab[i].name, VNET_SETNAME)) 1042eddfbb76SRobert Watson vnet_data_free(ef->progtab[i].addr, 1043eddfbb76SRobert Watson ef->progtab[i].size); 1044eddfbb76SRobert Watson #endif 104550c202c5SJeff Roberson } 104650c202c5SJeff Roberson } 104770b7ffeeSIan Dowse if (ef->preloaded) { 104870b7ffeeSIan Dowse free(ef->reltab, M_LINKER); 104970b7ffeeSIan Dowse free(ef->relatab, M_LINKER); 105070b7ffeeSIan Dowse free(ef->progtab, M_LINKER); 1051a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1052a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1053a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 105473624a80SMark Johnston if (file->pathname != NULL) 105573624a80SMark Johnston preload_delete_name(file->pathname); 105670b7ffeeSIan Dowse return; 105770b7ffeeSIan Dowse } 105870b7ffeeSIan Dowse 1059b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) 106020947801SPeter Wemm free(ef->reltab[i].rel, M_LINKER); 1061b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) 106220947801SPeter Wemm free(ef->relatab[i].rela, M_LINKER); 106320947801SPeter Wemm free(ef->reltab, M_LINKER); 106420947801SPeter Wemm free(ef->relatab, M_LINKER); 106520947801SPeter Wemm free(ef->progtab, M_LINKER); 106620947801SPeter Wemm 1067fe3db7c7SDoug Rabson if (ef->object) { 1068fe3db7c7SDoug Rabson vm_map_remove(kernel_map, (vm_offset_t) ef->address, 106920947801SPeter Wemm (vm_offset_t) ef->address + 107020947801SPeter Wemm (ef->object->size << PAGE_SHIFT)); 1071fe3db7c7SDoug Rabson } 1072e9eabf59SPeter Wemm free(ef->e_shdr, M_LINKER); 1073e9eabf59SPeter Wemm free(ef->ddbsymtab, M_LINKER); 1074e9eabf59SPeter Wemm free(ef->ddbstrtab, M_LINKER); 107520947801SPeter Wemm free(ef->shstrtab, M_LINKER); 1076a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1077a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1078a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 1079de78ca7eSPeter Wemm } 1080de78ca7eSPeter Wemm 1081fe3db7c7SDoug Rabson static const char * 1082757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info) 1083a4f67738SDoug Rabson { 1084fe3db7c7SDoug Rabson const Elf_Sym *ref; 1085a4f67738SDoug Rabson 1086aa855a59SPeter Wemm if (ELF_R_SYM(r_info)) { 1087e9eabf59SPeter Wemm ref = ef->ddbsymtab + ELF_R_SYM(r_info); 1088e9eabf59SPeter Wemm return ef->ddbstrtab + ref->st_name; 1089fe3db7c7SDoug Rabson } else 1090fe3db7c7SDoug Rabson return NULL; 1091a4f67738SDoug Rabson } 1092a4f67738SDoug Rabson 1093e9eabf59SPeter Wemm static Elf_Addr 1094e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec) 1095e9eabf59SPeter Wemm { 1096e9eabf59SPeter Wemm int i; 1097e9eabf59SPeter Wemm Elf_Addr base = 0; 1098e9eabf59SPeter Wemm 1099e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1100a8774e39SPeter Wemm if (sec == ef->progtab[i].sec) { 1101e9eabf59SPeter Wemm base = (Elf_Addr)ef->progtab[i].addr; 1102a8774e39SPeter Wemm break; 1103e9eabf59SPeter Wemm } 1104a8774e39SPeter Wemm } 1105e9eabf59SPeter Wemm return base; 1106e9eabf59SPeter Wemm } 1107e9eabf59SPeter Wemm 1108a4f67738SDoug Rabson static int 1109326e27d8SDoug Rabson relocate_file(elf_file_t ef) 1110a4f67738SDoug Rabson { 1111fe3db7c7SDoug Rabson const Elf_Rel *rellim; 1112fe3db7c7SDoug Rabson const Elf_Rel *rel; 1113fe3db7c7SDoug Rabson const Elf_Rela *relalim; 1114fe3db7c7SDoug Rabson const Elf_Rela *rela; 1115aa855a59SPeter Wemm const char *symname; 1116e9eabf59SPeter Wemm const Elf_Sym *sym; 1117e9eabf59SPeter Wemm int i; 1118757686b1SMarcel Moolenaar Elf_Size symidx; 1119e9eabf59SPeter Wemm Elf_Addr base; 1120e9eabf59SPeter Wemm 1121a4f67738SDoug Rabson 1122fe3db7c7SDoug Rabson /* Perform relocations without addend if there are any: */ 1123b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1124e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 11253cfce8e4SKonstantin Belousov if (rel == NULL) { 11263cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab!"); 11273cfce8e4SKonstantin Belousov return (ENOEXEC); 11283cfce8e4SKonstantin Belousov } 112920947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1130e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 11313cfce8e4SKonstantin Belousov if (base == 0) { 11323cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 11333cfce8e4SKonstantin Belousov return (ENOEXEC); 11343cfce8e4SKonstantin Belousov } 113520947801SPeter Wemm for ( ; rel < rellim; rel++) { 1136e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 113720947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 113820947801SPeter Wemm continue; 1139e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 114020947801SPeter Wemm /* Local relocs are already done */ 114120947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 114220947801SPeter Wemm continue; 114320947801SPeter Wemm if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL, 114420947801SPeter Wemm elf_obj_lookup)) { 1145aa855a59SPeter Wemm symname = symbol_name(ef, rel->r_info); 114620947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 114720947801SPeter Wemm symname); 11483cfce8e4SKonstantin Belousov return (ENOENT); 1149a13ddfb6SPeter Wemm } 1150aa855a59SPeter Wemm } 1151a4f67738SDoug Rabson } 1152a4f67738SDoug Rabson 1153fe3db7c7SDoug Rabson /* Perform relocations with addend if there are any: */ 1154b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1155e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 11563cfce8e4SKonstantin Belousov if (rela == NULL) { 11573cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 11583cfce8e4SKonstantin Belousov return (ENOEXEC); 11593cfce8e4SKonstantin Belousov } 116020947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1161e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 11623cfce8e4SKonstantin Belousov if (base == 0) { 11633cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, 11643cfce8e4SKonstantin Belousov "lost base for relatab"); 11653cfce8e4SKonstantin Belousov return (ENOEXEC); 11663cfce8e4SKonstantin Belousov } 116720947801SPeter Wemm for ( ; rela < relalim; rela++) { 1168e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 116920947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 117020947801SPeter Wemm continue; 1171e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 117220947801SPeter Wemm /* Local relocs are already done */ 117320947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 117420947801SPeter Wemm continue; 117520947801SPeter Wemm if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA, 117620947801SPeter Wemm elf_obj_lookup)) { 1177aa855a59SPeter Wemm symname = symbol_name(ef, rela->r_info); 117820947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 117920947801SPeter Wemm symname); 11803cfce8e4SKonstantin Belousov return (ENOENT); 1181a13ddfb6SPeter Wemm } 1182e9eabf59SPeter Wemm } 1183e9eabf59SPeter Wemm } 1184a4f67738SDoug Rabson 11852832cd54SKonstantin Belousov /* 1186e3043798SPedro F. Giffuni * Only clean SHN_FBSD_CACHED for successful return. If we 11872832cd54SKonstantin Belousov * modified symbol table for the object but found an 11882832cd54SKonstantin Belousov * unresolved symbol, there is no reason to roll back. 11892832cd54SKonstantin Belousov */ 11902832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(ef); 11912832cd54SKonstantin Belousov 11923cfce8e4SKonstantin Belousov return (0); 1193a4f67738SDoug Rabson } 1194a4f67738SDoug Rabson 119537c84183SPoul-Henning Kamp static int 1196d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym) 1197a4f67738SDoug Rabson { 1198326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 11992d636ab0SPeter Wemm const Elf_Sym *symp; 12002d636ab0SPeter Wemm const char *strp; 1201a4f67738SDoug Rabson int i; 1202a4f67738SDoug Rabson 12032d636ab0SPeter Wemm for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 12042d636ab0SPeter Wemm strp = ef->ddbstrtab + symp->st_name; 1205a8774e39SPeter Wemm if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) { 1206d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 12072d636ab0SPeter Wemm return 0; 12082d636ab0SPeter Wemm } 12092d636ab0SPeter Wemm } 1210a4f67738SDoug Rabson return ENOENT; 1211a4f67738SDoug Rabson } 1212a4f67738SDoug Rabson 1213de78ca7eSPeter Wemm static int 121420947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, 121520947801SPeter Wemm linker_symval_t *symval) 1216a4f67738SDoug Rabson { 12177035cf14SKonstantin Belousov elf_file_t ef; 12187035cf14SKonstantin Belousov const Elf_Sym *es; 12197035cf14SKonstantin Belousov caddr_t val; 1220a4f67738SDoug Rabson 12217035cf14SKonstantin Belousov ef = (elf_file_t) lf; 12227035cf14SKonstantin Belousov es = (const Elf_Sym*) sym; 12237035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 12245cf87418SMarcel Moolenaar if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) { 12252d636ab0SPeter Wemm symval->name = ef->ddbstrtab + es->st_name; 12267035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 12277035cf14SKonstantin Belousov if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC) 12287035cf14SKonstantin Belousov val = ((caddr_t (*)(void))val)(); 12297035cf14SKonstantin Belousov symval->value = val; 12302d636ab0SPeter Wemm symval->size = es->st_size; 12312d636ab0SPeter Wemm return 0; 12322d636ab0SPeter Wemm } 12332d636ab0SPeter Wemm return ENOENT; 12342d636ab0SPeter Wemm } 1235a4f67738SDoug Rabson 1236a4f67738SDoug Rabson static int 1237a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value, 1238d254af07SMatthew Dillon c_linker_sym_t *sym, long *diffp) 1239a4f67738SDoug Rabson { 1240326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1241586453feSBruce Evans u_long off = (uintptr_t) (void *) value; 1242a4f67738SDoug Rabson u_long diff = off; 1243586453feSBruce Evans u_long st_value; 1244fe3db7c7SDoug Rabson const Elf_Sym *es; 1245b85f65afSPedro F. Giffuni const Elf_Sym *best = NULL; 1246a4f67738SDoug Rabson int i; 1247a4f67738SDoug Rabson 12482d636ab0SPeter Wemm for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 1249a4f67738SDoug Rabson if (es->st_name == 0) 1250a4f67738SDoug Rabson continue; 1251a8774e39SPeter Wemm st_value = es->st_value; 1252b5abfb70SPeter Wemm if (off >= st_value) { 1253b5abfb70SPeter Wemm if (off - st_value < diff) { 1254b5abfb70SPeter Wemm diff = off - st_value; 1255a4f67738SDoug Rabson best = es; 1256a4f67738SDoug Rabson if (diff == 0) 1257a4f67738SDoug Rabson break; 1258b5abfb70SPeter Wemm } else if (off - st_value == diff) { 1259a4f67738SDoug Rabson best = es; 1260a4f67738SDoug Rabson } 1261a4f67738SDoug Rabson } 1262a4f67738SDoug Rabson } 1263b85f65afSPedro F. Giffuni if (best == NULL) 1264a4f67738SDoug Rabson *diffp = off; 1265a4f67738SDoug Rabson else 1266a4f67738SDoug Rabson *diffp = diff; 1267d254af07SMatthew Dillon *sym = (c_linker_sym_t) best; 1268a4f67738SDoug Rabson 1269a4f67738SDoug Rabson return 0; 1270a4f67738SDoug Rabson } 1271f41325dbSPeter Wemm 1272f41325dbSPeter Wemm /* 1273f41325dbSPeter Wemm * Look up a linker set on an ELF system. 1274f41325dbSPeter Wemm */ 1275f41325dbSPeter Wemm static int 1276f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name, 1277f41325dbSPeter Wemm void ***startp, void ***stopp, int *countp) 1278f41325dbSPeter Wemm { 1279e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 1280f41325dbSPeter Wemm void **start, **stop; 1281e9eabf59SPeter Wemm int i, count; 1282f41325dbSPeter Wemm 1283e9eabf59SPeter Wemm /* Relative to section number */ 1284e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1285e9eabf59SPeter Wemm if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) && 1286e9eabf59SPeter Wemm strcmp(ef->progtab[i].name + 4, name) == 0) { 1287e9eabf59SPeter Wemm start = (void **)ef->progtab[i].addr; 128820947801SPeter Wemm stop = (void **)((char *)ef->progtab[i].addr + 128920947801SPeter Wemm ef->progtab[i].size); 1290f41325dbSPeter Wemm count = stop - start; 1291f41325dbSPeter Wemm if (startp) 1292f41325dbSPeter Wemm *startp = start; 1293f41325dbSPeter Wemm if (stopp) 1294f41325dbSPeter Wemm *stopp = stop; 1295f41325dbSPeter Wemm if (countp) 1296f41325dbSPeter Wemm *countp = count; 1297e9eabf59SPeter Wemm return (0); 1298e9eabf59SPeter Wemm } 1299e9eabf59SPeter Wemm } 1300e9eabf59SPeter Wemm return (ESRCH); 1301f41325dbSPeter Wemm } 1302bb9fe9ddSBrian Feldman 1303bb9fe9ddSBrian Feldman static int 1304bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file, 1305e9eabf59SPeter Wemm int (*callback)(const char *, void *), void *opaque) 1306e9eabf59SPeter Wemm { 1307bb9fe9ddSBrian Feldman elf_file_t ef = (elf_file_t)file; 1308bb9fe9ddSBrian Feldman const Elf_Sym *symp; 1309bb9fe9ddSBrian Feldman int i, error; 1310bb9fe9ddSBrian Feldman 1311bb9fe9ddSBrian Feldman /* Exhaustive search */ 1312bb9fe9ddSBrian Feldman for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1313bb9fe9ddSBrian Feldman if (symp->st_value != 0 && 13147035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 13157035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 1316bb9fe9ddSBrian Feldman error = callback(ef->ddbstrtab + symp->st_name, opaque); 1317bb9fe9ddSBrian Feldman if (error) 1318bb9fe9ddSBrian Feldman return (error); 1319bb9fe9ddSBrian Feldman } 1320bb9fe9ddSBrian Feldman } 1321bb9fe9ddSBrian Feldman return (0); 1322bb9fe9ddSBrian Feldman } 132384201059SMarcel Moolenaar 1324a2024a3eSJohn Birrell static int 1325a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file, 1326a2024a3eSJohn Birrell linker_function_nameval_callback_t callback, void *opaque) 1327a2024a3eSJohn Birrell { 1328a2024a3eSJohn Birrell linker_symval_t symval; 1329a2024a3eSJohn Birrell elf_file_t ef = (elf_file_t)file; 1330a2024a3eSJohn Birrell const Elf_Sym* symp; 1331a2024a3eSJohn Birrell int i, error; 1332a2024a3eSJohn Birrell 1333a2024a3eSJohn Birrell /* Exhaustive search */ 1334a2024a3eSJohn Birrell for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1335a2024a3eSJohn Birrell if (symp->st_value != 0 && 13367035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 13377035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 13387035cf14SKonstantin Belousov error = link_elf_symbol_values(file, 13397035cf14SKonstantin Belousov (c_linker_sym_t)symp, &symval); 1340a2024a3eSJohn Birrell if (error) 1341a2024a3eSJohn Birrell return (error); 1342a2024a3eSJohn Birrell error = callback(file, i, &symval, opaque); 1343a2024a3eSJohn Birrell if (error) 1344a2024a3eSJohn Birrell return (error); 1345a2024a3eSJohn Birrell } 1346a2024a3eSJohn Birrell } 1347a2024a3eSJohn Birrell return (0); 1348a2024a3eSJohn Birrell } 1349a2024a3eSJohn Birrell 13502832cd54SKonstantin Belousov static void 13512832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef) 13522832cd54SKonstantin Belousov { 13532832cd54SKonstantin Belousov Elf_Sym *sym; 13542832cd54SKonstantin Belousov Elf_Size i; 13552832cd54SKonstantin Belousov 13562832cd54SKonstantin Belousov for (i = 0; i < ef->ddbsymcnt; i++) { 13572832cd54SKonstantin Belousov sym = ef->ddbsymtab + i; 13582832cd54SKonstantin Belousov if (sym->st_shndx == SHN_FBSD_CACHED) { 13592832cd54SKonstantin Belousov sym->st_shndx = SHN_UNDEF; 13602832cd54SKonstantin Belousov sym->st_value = 0; 13612832cd54SKonstantin Belousov } 13622832cd54SKonstantin Belousov } 13632832cd54SKonstantin Belousov } 13642832cd54SKonstantin Belousov 1365d297ad16SMarcel Moolenaar /* 1366d297ad16SMarcel Moolenaar * Symbol lookup function that can be used when the symbol index is known (ie 1367d297ad16SMarcel Moolenaar * in relocations). It uses the symbol index instead of doing a fully fledged 1368d297ad16SMarcel Moolenaar * hash table based lookup when such is valid. For example for local symbols. 1369d297ad16SMarcel Moolenaar * This is not only more efficient, it's also more correct. It's not always 1370d297ad16SMarcel Moolenaar * the case that the symbol can be found through the hash table. 1371d297ad16SMarcel Moolenaar */ 1372cff8c6f2SKonstantin Belousov static int 1373cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res) 1374d297ad16SMarcel Moolenaar { 1375d297ad16SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 13762832cd54SKonstantin Belousov Elf_Sym *sym; 1377d297ad16SMarcel Moolenaar const char *symbol; 1378cff8c6f2SKonstantin Belousov Elf_Addr res1; 1379d297ad16SMarcel Moolenaar 1380d297ad16SMarcel Moolenaar /* Don't even try to lookup the symbol if the index is bogus. */ 1381cff8c6f2SKonstantin Belousov if (symidx >= ef->ddbsymcnt) { 1382cff8c6f2SKonstantin Belousov *res = 0; 1383cff8c6f2SKonstantin Belousov return (EINVAL); 1384cff8c6f2SKonstantin Belousov } 1385d297ad16SMarcel Moolenaar 1386e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 1387d297ad16SMarcel Moolenaar 13884cec6f5dSPeter Wemm /* Quick answer if there is a definition included. */ 1389cff8c6f2SKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) { 1390289ead7cSKonstantin Belousov res1 = (Elf_Addr)sym->st_value; 1391289ead7cSKonstantin Belousov if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) 1392289ead7cSKonstantin Belousov res1 = ((Elf_Addr (*)(void))res1)(); 1393289ead7cSKonstantin Belousov *res = res1; 1394cff8c6f2SKonstantin Belousov return (0); 1395cff8c6f2SKonstantin Belousov } 13964cec6f5dSPeter Wemm 13974cec6f5dSPeter Wemm /* If we get here, then it is undefined and needs a lookup. */ 13984cec6f5dSPeter Wemm switch (ELF_ST_BIND(sym->st_info)) { 13994cec6f5dSPeter Wemm case STB_LOCAL: 14004cec6f5dSPeter Wemm /* Local, but undefined? huh? */ 1401cff8c6f2SKonstantin Belousov *res = 0; 1402cff8c6f2SKonstantin Belousov return (EINVAL); 1403d297ad16SMarcel Moolenaar 1404e9eabf59SPeter Wemm case STB_GLOBAL: 1405cff8c6f2SKonstantin Belousov case STB_WEAK: 1406e9eabf59SPeter Wemm /* Relative to Data or Function name */ 1407e9eabf59SPeter Wemm symbol = ef->ddbstrtab + sym->st_name; 1408d297ad16SMarcel Moolenaar 1409d297ad16SMarcel Moolenaar /* Force a lookup failure if the symbol name is bogus. */ 1410cff8c6f2SKonstantin Belousov if (*symbol == 0) { 1411cff8c6f2SKonstantin Belousov *res = 0; 1412cff8c6f2SKonstantin Belousov return (EINVAL); 1413cff8c6f2SKonstantin Belousov } 1414cff8c6f2SKonstantin Belousov res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps); 14152832cd54SKonstantin Belousov 14162832cd54SKonstantin Belousov /* 14172832cd54SKonstantin Belousov * Cache global lookups during module relocation. The failure 14182832cd54SKonstantin Belousov * case is particularly expensive for callers, who must scan 14192832cd54SKonstantin Belousov * through the entire globals table doing strcmp(). Cache to 14202832cd54SKonstantin Belousov * avoid doing such work repeatedly. 14212832cd54SKonstantin Belousov * 14222832cd54SKonstantin Belousov * After relocation is complete, undefined globals will be 14232832cd54SKonstantin Belousov * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(), 14242832cd54SKonstantin Belousov * above. 14252832cd54SKonstantin Belousov */ 1426cff8c6f2SKonstantin Belousov if (res1 != 0) { 14272832cd54SKonstantin Belousov sym->st_shndx = SHN_FBSD_CACHED; 1428cff8c6f2SKonstantin Belousov sym->st_value = res1; 1429cff8c6f2SKonstantin Belousov *res = res1; 1430e9eabf59SPeter Wemm return (0); 1431cff8c6f2SKonstantin Belousov } else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) { 1432cff8c6f2SKonstantin Belousov sym->st_value = 0; 1433cff8c6f2SKonstantin Belousov *res = 0; 1434cff8c6f2SKonstantin Belousov return (0); 1435cff8c6f2SKonstantin Belousov } 1436cff8c6f2SKonstantin Belousov return (EINVAL); 1437e9eabf59SPeter Wemm 1438e9eabf59SPeter Wemm default: 1439cff8c6f2SKonstantin Belousov return (EINVAL); 1440d297ad16SMarcel Moolenaar } 1441e9eabf59SPeter Wemm } 1442e9eabf59SPeter Wemm 14437251b4bfSJake Burkholder static void 14447226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef) 14457226306eSKonstantin Belousov { 14467226306eSKonstantin Belousov static const char startn[] = "__start_"; 14477226306eSKonstantin Belousov static const char stopn[] = "__stop_"; 14487226306eSKonstantin Belousov Elf_Sym *sym; 14497226306eSKonstantin Belousov const char *sym_name, *linkset_name; 14507226306eSKonstantin Belousov Elf_Addr startp, stopp; 14517226306eSKonstantin Belousov Elf_Size symidx; 14527226306eSKonstantin Belousov int start, i; 14537226306eSKonstantin Belousov 14547226306eSKonstantin Belousov startp = stopp = 0; 14557226306eSKonstantin Belousov for (symidx = 1 /* zero entry is special */; 14567226306eSKonstantin Belousov symidx < ef->ddbsymcnt; symidx++) { 14577226306eSKonstantin Belousov sym = ef->ddbsymtab + symidx; 14587226306eSKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) 14597226306eSKonstantin Belousov continue; 14607226306eSKonstantin Belousov 14617226306eSKonstantin Belousov sym_name = ef->ddbstrtab + sym->st_name; 14627226306eSKonstantin Belousov if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) { 14637226306eSKonstantin Belousov start = 1; 14647226306eSKonstantin Belousov linkset_name = sym_name + sizeof(startn) - 1; 14657226306eSKonstantin Belousov } 14667226306eSKonstantin Belousov else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) { 14677226306eSKonstantin Belousov start = 0; 14687226306eSKonstantin Belousov linkset_name = sym_name + sizeof(stopn) - 1; 14697226306eSKonstantin Belousov } 14707226306eSKonstantin Belousov else 14717226306eSKonstantin Belousov continue; 14727226306eSKonstantin Belousov 14737226306eSKonstantin Belousov for (i = 0; i < ef->nprogtab; i++) { 14747226306eSKonstantin Belousov if (strcmp(ef->progtab[i].name, linkset_name) == 0) { 14757226306eSKonstantin Belousov startp = (Elf_Addr)ef->progtab[i].addr; 14767226306eSKonstantin Belousov stopp = (Elf_Addr)(startp + ef->progtab[i].size); 14777226306eSKonstantin Belousov break; 14787226306eSKonstantin Belousov } 14797226306eSKonstantin Belousov } 14807226306eSKonstantin Belousov if (i == ef->nprogtab) 14817226306eSKonstantin Belousov continue; 14827226306eSKonstantin Belousov 14837226306eSKonstantin Belousov sym->st_value = start ? startp : stopp; 14847226306eSKonstantin Belousov sym->st_shndx = i; 14857226306eSKonstantin Belousov } 14867226306eSKonstantin Belousov } 14877226306eSKonstantin Belousov 14883cfce8e4SKonstantin Belousov static int 1489289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs) 14907251b4bfSJake Burkholder { 1491e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 14927251b4bfSJake Burkholder const Elf_Rel *rellim; 14937251b4bfSJake Burkholder const Elf_Rel *rel; 14947251b4bfSJake Burkholder const Elf_Rela *relalim; 14957251b4bfSJake Burkholder const Elf_Rela *rela; 1496e9eabf59SPeter Wemm const Elf_Sym *sym; 1497e9eabf59SPeter Wemm Elf_Addr base; 1498e9eabf59SPeter Wemm int i; 1499757686b1SMarcel Moolenaar Elf_Size symidx; 1500e9eabf59SPeter Wemm 15017226306eSKonstantin Belousov link_elf_fix_link_set(ef); 15027226306eSKonstantin Belousov 15037251b4bfSJake Burkholder /* Perform relocations without addend if there are any: */ 1504b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1505e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 15063cfce8e4SKonstantin Belousov if (rel == NULL) { 15073cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab"); 15083cfce8e4SKonstantin Belousov return (ENOEXEC); 15093cfce8e4SKonstantin Belousov } 151020947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1511e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 15123cfce8e4SKonstantin Belousov if (base == 0) { 15133cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 15143cfce8e4SKonstantin Belousov return (ENOEXEC); 15153cfce8e4SKonstantin Belousov } 151620947801SPeter Wemm for ( ; rel < rellim; rel++) { 1517e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 151820947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 151920947801SPeter Wemm continue; 1520e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 152120947801SPeter Wemm /* Only do local relocs */ 152220947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 152320947801SPeter Wemm continue; 1524289ead7cSKonstantin Belousov if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) == 1525289ead7cSKonstantin Belousov ifuncs) 152620947801SPeter Wemm elf_reloc_local(lf, base, rel, ELF_RELOC_REL, 152720947801SPeter Wemm elf_obj_lookup); 1528*8f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 1529289ead7cSKonstantin Belousov else if (ifuncs) 1530289ead7cSKonstantin Belousov elf_reloc_ifunc(lf, base, rel, ELF_RELOC_REL, 1531289ead7cSKonstantin Belousov elf_obj_lookup); 1532*8f941950SKonstantin Belousov #endif 15337251b4bfSJake Burkholder } 1534e9eabf59SPeter Wemm } 15357251b4bfSJake Burkholder 15367251b4bfSJake Burkholder /* Perform relocations with addend if there are any: */ 1537b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1538e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 15393cfce8e4SKonstantin Belousov if (rela == NULL) { 15403cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 15413cfce8e4SKonstantin Belousov return (ENOEXEC); 15423cfce8e4SKonstantin Belousov } 154320947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1544e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 15453cfce8e4SKonstantin Belousov if (base == 0) { 15463cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 15473cfce8e4SKonstantin Belousov return (ENOEXEC); 15483cfce8e4SKonstantin Belousov } 154920947801SPeter Wemm for ( ; rela < relalim; rela++) { 1550e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 155120947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 155220947801SPeter Wemm continue; 1553e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 155420947801SPeter Wemm /* Only do local relocs */ 155520947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 155620947801SPeter Wemm continue; 1557289ead7cSKonstantin Belousov if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) == 1558289ead7cSKonstantin Belousov ifuncs) 155920947801SPeter Wemm elf_reloc_local(lf, base, rela, ELF_RELOC_RELA, 156020947801SPeter Wemm elf_obj_lookup); 1561*8f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 1562289ead7cSKonstantin Belousov else if (ifuncs) 1563289ead7cSKonstantin Belousov elf_reloc_ifunc(lf, base, rela, ELF_RELOC_RELA, 1564289ead7cSKonstantin Belousov elf_obj_lookup); 1565*8f941950SKonstantin Belousov #endif 15667251b4bfSJake Burkholder } 15677251b4bfSJake Burkholder } 15683cfce8e4SKonstantin Belousov return (0); 1569e9eabf59SPeter Wemm } 157000a5db46SStacey Son 157100a5db46SStacey Son static long 1572e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab) 157300a5db46SStacey Son { 157400a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 157500a5db46SStacey Son 157600a5db46SStacey Son *symtab = ef->ddbsymtab; 157700a5db46SStacey Son 157800a5db46SStacey Son if (*symtab == NULL) 157900a5db46SStacey Son return (0); 158000a5db46SStacey Son 158100a5db46SStacey Son return (ef->ddbsymcnt); 158200a5db46SStacey Son } 158300a5db46SStacey Son 158400a5db46SStacey Son static long 158500a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab) 158600a5db46SStacey Son { 158700a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 158800a5db46SStacey Son 158900a5db46SStacey Son *strtab = ef->ddbstrtab; 159000a5db46SStacey Son 159100a5db46SStacey Son if (*strtab == NULL) 159200a5db46SStacey Son return (0); 159300a5db46SStacey Son 159400a5db46SStacey Son return (ef->ddbstrcnt); 159500a5db46SStacey Son } 1596