1a4f67738SDoug Rabson /*- 28a36da99SPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 38a36da99SPedro F. Giffuni * 4326e27d8SDoug Rabson * Copyright (c) 1998-2000 Doug Rabson 5e9eabf59SPeter Wemm * Copyright (c) 2004 Peter Wemm 6a4f67738SDoug Rabson * All rights reserved. 7a4f67738SDoug Rabson * 8a4f67738SDoug Rabson * Redistribution and use in source and binary forms, with or without 9a4f67738SDoug Rabson * modification, are permitted provided that the following conditions 10a4f67738SDoug Rabson * are met: 11a4f67738SDoug Rabson * 1. Redistributions of source code must retain the above copyright 12a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer. 13a4f67738SDoug Rabson * 2. Redistributions in binary form must reproduce the above copyright 14a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer in the 15a4f67738SDoug Rabson * documentation and/or other materials provided with the distribution. 16a4f67738SDoug Rabson * 17a4f67738SDoug Rabson * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 18a4f67738SDoug Rabson * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19a4f67738SDoug Rabson * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20a4f67738SDoug Rabson * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 21a4f67738SDoug Rabson * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22a4f67738SDoug Rabson * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23a4f67738SDoug Rabson * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24a4f67738SDoug Rabson * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25a4f67738SDoug Rabson * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26a4f67738SDoug Rabson * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27a4f67738SDoug Rabson * SUCH DAMAGE. 28a4f67738SDoug Rabson */ 29a4f67738SDoug Rabson 30677b542eSDavid E. O'Brien #include <sys/cdefs.h> 31677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$"); 32677b542eSDavid E. O'Brien 33326e27d8SDoug Rabson #include "opt_ddb.h" 34326e27d8SDoug Rabson 35a4f67738SDoug Rabson #include <sys/param.h> 36a4f67738SDoug Rabson #include <sys/systm.h> 37fb919e4dSMark Murray #include <sys/kernel.h> 38fb919e4dSMark Murray #include <sys/lock.h> 39a4f67738SDoug Rabson #include <sys/malloc.h> 40c0b824f9SDima Dorfman #include <sys/mutex.h> 4162d615d5SJohn Baldwin #include <sys/mount.h> 42a4f67738SDoug Rabson #include <sys/proc.h> 43a4f67738SDoug Rabson #include <sys/namei.h> 44a4f67738SDoug Rabson #include <sys/fcntl.h> 45a4f67738SDoug Rabson #include <sys/vnode.h> 46a4f67738SDoug Rabson #include <sys/linker.h> 47fb919e4dSMark Murray 48a4f67738SDoug Rabson #include <machine/elf.h> 49a4f67738SDoug Rabson 50eddfbb76SRobert Watson #include <net/vnet.h> 51eddfbb76SRobert Watson 52aed55708SRobert Watson #include <security/mac/mac_framework.h> 53aed55708SRobert Watson 54fe3db7c7SDoug Rabson #include <vm/vm.h> 55fe3db7c7SDoug Rabson #include <vm/vm_param.h> 56fe3db7c7SDoug Rabson #include <vm/vm_object.h> 57fe3db7c7SDoug Rabson #include <vm/vm_kern.h> 58fe3db7c7SDoug Rabson #include <vm/vm_extern.h> 59fe3db7c7SDoug Rabson #include <vm/pmap.h> 60fe3db7c7SDoug Rabson #include <vm/vm_map.h> 61fb919e4dSMark Murray 62f9980387SPeter Wemm #include <sys/link_elf.h> 63fe3db7c7SDoug Rabson 64a2024a3eSJohn Birrell #ifdef DDB_CTF 6522bbc4b2SXin LI #include <contrib/zlib/zlib.h> 66a2024a3eSJohn Birrell #endif 67a2024a3eSJohn Birrell 68326e27d8SDoug Rabson #include "linker_if.h" 69a4f67738SDoug Rabson 70e9eabf59SPeter Wemm typedef struct { 71e9eabf59SPeter Wemm void *addr; 7220947801SPeter Wemm Elf_Off size; 73*1d9eae9fSMark Johnston int flags; /* Section flags. */ 74*1d9eae9fSMark Johnston int sec; /* Original section number. */ 75e9eabf59SPeter Wemm char *name; 76e9eabf59SPeter Wemm } Elf_progent; 77e9eabf59SPeter Wemm 78e9eabf59SPeter Wemm typedef struct { 79e9eabf59SPeter Wemm Elf_Rel *rel; 8020947801SPeter Wemm int nrel; 81e9eabf59SPeter Wemm int sec; 82e9eabf59SPeter Wemm } Elf_relent; 83e9eabf59SPeter Wemm 84e9eabf59SPeter Wemm typedef struct { 85e9eabf59SPeter Wemm Elf_Rela *rela; 8620947801SPeter Wemm int nrela; 87e9eabf59SPeter Wemm int sec; 88e9eabf59SPeter Wemm } Elf_relaent; 89e9eabf59SPeter Wemm 90e9eabf59SPeter Wemm 91a4f67738SDoug Rabson typedef struct elf_file { 92326e27d8SDoug Rabson struct linker_file lf; /* Common fields */ 9320947801SPeter Wemm 9470b7ffeeSIan Dowse int preloaded; 95fe3db7c7SDoug Rabson caddr_t address; /* Relocation address */ 96fe3db7c7SDoug Rabson vm_object_t object; /* VM object to hold file pages */ 97e9eabf59SPeter Wemm Elf_Shdr *e_shdr; 98e9eabf59SPeter Wemm 99e9eabf59SPeter Wemm Elf_progent *progtab; 100d821d364SPedro F. Giffuni u_int nprogtab; 101e9eabf59SPeter Wemm 102e9eabf59SPeter Wemm Elf_relaent *relatab; 103d821d364SPedro F. Giffuni u_int nrelatab; 104e9eabf59SPeter Wemm 105e9eabf59SPeter Wemm Elf_relent *reltab; 106b474c780SDavid E. O'Brien int nreltab; 107e9eabf59SPeter Wemm 108e9eabf59SPeter Wemm Elf_Sym *ddbsymtab; /* The symbol table we are using */ 1092d636ab0SPeter Wemm long ddbsymcnt; /* Number of symbols */ 1102d636ab0SPeter Wemm caddr_t ddbstrtab; /* String table */ 1112d636ab0SPeter Wemm long ddbstrcnt; /* number of bytes in string table */ 112e9eabf59SPeter Wemm 113e9eabf59SPeter Wemm caddr_t shstrtab; /* Section name string table */ 114e9eabf59SPeter Wemm long shstrcnt; /* number of bytes in string table */ 115e9eabf59SPeter Wemm 116a2024a3eSJohn Birrell caddr_t ctftab; /* CTF table */ 117a2024a3eSJohn Birrell long ctfcnt; /* number of bytes in CTF table */ 118a2024a3eSJohn Birrell caddr_t ctfoff; /* CTF offset table */ 119a2024a3eSJohn Birrell caddr_t typoff; /* Type offset table */ 120a2024a3eSJohn Birrell long typlen; /* Number of type entries. */ 121a2024a3eSJohn Birrell 122a4f67738SDoug Rabson } *elf_file_t; 123a4f67738SDoug Rabson 124a2024a3eSJohn Birrell #include <kern/kern_ctf.c> 125a2024a3eSJohn Birrell 12654823af2SPeter Wemm static int link_elf_link_preload(linker_class_t cls, 127326e27d8SDoug Rabson const char *, linker_file_t *); 12854823af2SPeter Wemm static int link_elf_link_preload_finish(linker_file_t); 12954823af2SPeter Wemm static int link_elf_load_file(linker_class_t, const char *, linker_file_t *); 130326e27d8SDoug Rabson static int link_elf_lookup_symbol(linker_file_t, const char *, 131326e27d8SDoug Rabson c_linker_sym_t *); 132e9eabf59SPeter Wemm static int link_elf_symbol_values(linker_file_t, c_linker_sym_t, 133e9eabf59SPeter Wemm linker_symval_t *); 134326e27d8SDoug Rabson static int link_elf_search_symbol(linker_file_t, caddr_t value, 135326e27d8SDoug Rabson c_linker_sym_t *sym, long *diffp); 136326e27d8SDoug Rabson 137326e27d8SDoug Rabson static void link_elf_unload_file(linker_file_t); 138f41325dbSPeter Wemm static int link_elf_lookup_set(linker_file_t, const char *, 139f41325dbSPeter Wemm void ***, void ***, int *); 140bb9fe9ddSBrian Feldman static int link_elf_each_function_name(linker_file_t, 141e9eabf59SPeter Wemm int (*)(const char *, void *), void *); 142a2024a3eSJohn Birrell static int link_elf_each_function_nameval(linker_file_t, 143a2024a3eSJohn Birrell linker_function_nameval_callback_t, 144a2024a3eSJohn Birrell void *); 145289ead7cSKonstantin Belousov static int link_elf_reloc_local(linker_file_t, bool); 146e76f11f4SAndriy Gapon static long link_elf_symtab_get(linker_file_t, const Elf_Sym **); 14700a5db46SStacey Son static long link_elf_strtab_get(linker_file_t, caddr_t *); 148326e27d8SDoug Rabson 149cff8c6f2SKonstantin Belousov static int elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, 150cff8c6f2SKonstantin Belousov Elf_Addr *); 151e9eabf59SPeter Wemm 152326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = { 153326e27d8SDoug Rabson KOBJMETHOD(linker_lookup_symbol, link_elf_lookup_symbol), 154326e27d8SDoug Rabson KOBJMETHOD(linker_symbol_values, link_elf_symbol_values), 155326e27d8SDoug Rabson KOBJMETHOD(linker_search_symbol, link_elf_search_symbol), 156326e27d8SDoug Rabson KOBJMETHOD(linker_unload, link_elf_unload_file), 15754823af2SPeter Wemm KOBJMETHOD(linker_load_file, link_elf_load_file), 15854823af2SPeter Wemm KOBJMETHOD(linker_link_preload, link_elf_link_preload), 15954823af2SPeter Wemm KOBJMETHOD(linker_link_preload_finish, link_elf_link_preload_finish), 160f41325dbSPeter Wemm KOBJMETHOD(linker_lookup_set, link_elf_lookup_set), 161bb9fe9ddSBrian Feldman KOBJMETHOD(linker_each_function_name, link_elf_each_function_name), 162a2024a3eSJohn Birrell KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval), 163a2024a3eSJohn Birrell KOBJMETHOD(linker_ctf_get, link_elf_ctf_get), 16400a5db46SStacey Son KOBJMETHOD(linker_symtab_get, link_elf_symtab_get), 16500a5db46SStacey Son KOBJMETHOD(linker_strtab_get, link_elf_strtab_get), 1666d775f0bSMark Johnston KOBJMETHOD_END 167326e27d8SDoug Rabson }; 168326e27d8SDoug Rabson 169326e27d8SDoug Rabson static struct linker_class link_elf_class = { 170326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32 171e9eabf59SPeter Wemm "elf32_obj", 172326e27d8SDoug Rabson #else 173e9eabf59SPeter Wemm "elf64_obj", 174326e27d8SDoug Rabson #endif 175326e27d8SDoug Rabson link_elf_methods, sizeof(struct elf_file) 176326e27d8SDoug Rabson }; 177326e27d8SDoug Rabson 178326e27d8SDoug Rabson static int relocate_file(elf_file_t ef); 1792832cd54SKonstantin Belousov static void elf_obj_cleanup_globals_cache(elf_file_t); 180a4f67738SDoug Rabson 181a4f67738SDoug Rabson static void 182552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s) 1831720979bSMarcel Moolenaar { 184552f9f63SEdwin Groothuis if (filename == NULL) 1851720979bSMarcel Moolenaar printf("kldload: %s\n", s); 186552f9f63SEdwin Groothuis else 187552f9f63SEdwin Groothuis printf("kldload: %s: %s\n", filename, s); 1881720979bSMarcel Moolenaar } 1891720979bSMarcel Moolenaar 1901720979bSMarcel Moolenaar static void 191a4f67738SDoug Rabson link_elf_init(void *arg) 192a4f67738SDoug Rabson { 193a4f67738SDoug Rabson 194326e27d8SDoug Rabson linker_add_class(&link_elf_class); 195a4f67738SDoug Rabson } 196a4f67738SDoug Rabson 197891cf3edSEd Maste SYSINIT(link_elf_obj, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, NULL); 198a4f67738SDoug Rabson 199*1d9eae9fSMark Johnston static void 200*1d9eae9fSMark Johnston link_elf_protect_range(elf_file_t ef, vm_offset_t start, vm_offset_t end, 201*1d9eae9fSMark Johnston vm_prot_t prot) 202*1d9eae9fSMark Johnston { 203*1d9eae9fSMark Johnston int error __unused; 204*1d9eae9fSMark Johnston 205*1d9eae9fSMark Johnston KASSERT(start <= end && start >= (vm_offset_t)ef->address && 206*1d9eae9fSMark Johnston end <= round_page((vm_offset_t)ef->address + ef->lf.size), 207*1d9eae9fSMark Johnston ("link_elf_protect_range: invalid range %#jx-%#jx", 208*1d9eae9fSMark Johnston (uintmax_t)start, (uintmax_t)end)); 209*1d9eae9fSMark Johnston 210*1d9eae9fSMark Johnston if (start == end) 211*1d9eae9fSMark Johnston return; 212*1d9eae9fSMark Johnston error = vm_map_protect(kernel_map, start, end, prot, FALSE); 213*1d9eae9fSMark Johnston KASSERT(error == KERN_SUCCESS, 214*1d9eae9fSMark Johnston ("link_elf_protect_range: vm_map_protect() returned %d", error)); 215*1d9eae9fSMark Johnston } 216*1d9eae9fSMark Johnston 217*1d9eae9fSMark Johnston /* 218*1d9eae9fSMark Johnston * Restrict permissions on linker file memory based on section flags. 219*1d9eae9fSMark Johnston * Sections need not be page-aligned, so overlap within a page is possible. 220*1d9eae9fSMark Johnston */ 221*1d9eae9fSMark Johnston static void 222*1d9eae9fSMark Johnston link_elf_protect(elf_file_t ef) 223*1d9eae9fSMark Johnston { 224*1d9eae9fSMark Johnston vm_offset_t end, segend, segstart, start; 225*1d9eae9fSMark Johnston vm_prot_t gapprot, prot, segprot; 226*1d9eae9fSMark Johnston int i; 227*1d9eae9fSMark Johnston 228*1d9eae9fSMark Johnston /* 229*1d9eae9fSMark Johnston * If the file was preloaded, the last page may contain other preloaded 230*1d9eae9fSMark Johnston * data which may need to be writeable. ELF files are always 231*1d9eae9fSMark Johnston * page-aligned, but other preloaded data, such as entropy or CPU 232*1d9eae9fSMark Johnston * microcode may be loaded with a smaller alignment. 233*1d9eae9fSMark Johnston */ 234*1d9eae9fSMark Johnston gapprot = ef->preloaded ? VM_PROT_RW : VM_PROT_READ; 235*1d9eae9fSMark Johnston 236*1d9eae9fSMark Johnston start = end = (vm_offset_t)ef->address; 237*1d9eae9fSMark Johnston prot = VM_PROT_READ; 238*1d9eae9fSMark Johnston for (i = 0; i < ef->nprogtab; i++) { 239*1d9eae9fSMark Johnston /* 240*1d9eae9fSMark Johnston * VNET and DPCPU sections have their memory allocated by their 241*1d9eae9fSMark Johnston * respective subsystems. 242*1d9eae9fSMark Johnston */ 243*1d9eae9fSMark Johnston if (ef->progtab[i].name != NULL && ( 244*1d9eae9fSMark Johnston #ifdef VIMAGE 245*1d9eae9fSMark Johnston strcmp(ef->progtab[i].name, VNET_SETNAME) == 0 || 246*1d9eae9fSMark Johnston #endif 247*1d9eae9fSMark Johnston strcmp(ef->progtab[i].name, DPCPU_SETNAME) == 0)) 248*1d9eae9fSMark Johnston continue; 249*1d9eae9fSMark Johnston 250*1d9eae9fSMark Johnston segstart = trunc_page((vm_offset_t)ef->progtab[i].addr); 251*1d9eae9fSMark Johnston segend = round_page((vm_offset_t)ef->progtab[i].addr + 252*1d9eae9fSMark Johnston ef->progtab[i].size); 253*1d9eae9fSMark Johnston segprot = VM_PROT_READ; 254*1d9eae9fSMark Johnston if ((ef->progtab[i].flags & SHF_WRITE) != 0) 255*1d9eae9fSMark Johnston segprot |= VM_PROT_WRITE; 256*1d9eae9fSMark Johnston if ((ef->progtab[i].flags & SHF_EXECINSTR) != 0) 257*1d9eae9fSMark Johnston segprot |= VM_PROT_EXECUTE; 258*1d9eae9fSMark Johnston 259*1d9eae9fSMark Johnston if (end <= segstart) { 260*1d9eae9fSMark Johnston /* 261*1d9eae9fSMark Johnston * Case 1: there is no overlap between the previous 262*1d9eae9fSMark Johnston * segment and this one. Apply protections to the 263*1d9eae9fSMark Johnston * previous segment, and protect the gap between the 264*1d9eae9fSMark Johnston * previous and current segments, if any. 265*1d9eae9fSMark Johnston */ 266*1d9eae9fSMark Johnston link_elf_protect_range(ef, start, end, prot); 267*1d9eae9fSMark Johnston link_elf_protect_range(ef, end, segstart, gapprot); 268*1d9eae9fSMark Johnston 269*1d9eae9fSMark Johnston start = segstart; 270*1d9eae9fSMark Johnston end = segend; 271*1d9eae9fSMark Johnston prot = segprot; 272*1d9eae9fSMark Johnston } else if (start < segstart && end == segend) { 273*1d9eae9fSMark Johnston /* 274*1d9eae9fSMark Johnston * Case 2: the current segment is a subrange of the 275*1d9eae9fSMark Johnston * previous segment. Apply protections to the 276*1d9eae9fSMark Johnston * non-overlapping portion of the previous segment. 277*1d9eae9fSMark Johnston */ 278*1d9eae9fSMark Johnston link_elf_protect_range(ef, start, segstart, prot); 279*1d9eae9fSMark Johnston 280*1d9eae9fSMark Johnston start = segstart; 281*1d9eae9fSMark Johnston prot |= segprot; 282*1d9eae9fSMark Johnston } else if (end < segend) { 283*1d9eae9fSMark Johnston /* 284*1d9eae9fSMark Johnston * Case 3: there is partial overlap between the previous 285*1d9eae9fSMark Johnston * and current segments. Apply protections to the 286*1d9eae9fSMark Johnston * non-overlapping portion of the previous segment, and 287*1d9eae9fSMark Johnston * then the overlap, which must use the union of the two 288*1d9eae9fSMark Johnston * segments' protections. 289*1d9eae9fSMark Johnston */ 290*1d9eae9fSMark Johnston link_elf_protect_range(ef, start, segstart, prot); 291*1d9eae9fSMark Johnston link_elf_protect_range(ef, segstart, end, 292*1d9eae9fSMark Johnston prot | segprot); 293*1d9eae9fSMark Johnston start = end; 294*1d9eae9fSMark Johnston end = segend; 295*1d9eae9fSMark Johnston prot = segprot; 296*1d9eae9fSMark Johnston } else { 297*1d9eae9fSMark Johnston /* 298*1d9eae9fSMark Johnston * Case 4: the two segments reside in the same page. 299*1d9eae9fSMark Johnston */ 300*1d9eae9fSMark Johnston prot |= segprot; 301*1d9eae9fSMark Johnston } 302*1d9eae9fSMark Johnston } 303*1d9eae9fSMark Johnston 304*1d9eae9fSMark Johnston /* 305*1d9eae9fSMark Johnston * Fix up the last unprotected segment and trailing data. 306*1d9eae9fSMark Johnston */ 307*1d9eae9fSMark Johnston link_elf_protect_range(ef, start, end, prot); 308*1d9eae9fSMark Johnston link_elf_protect_range(ef, end, 309*1d9eae9fSMark Johnston round_page((vm_offset_t)ef->address + ef->lf.size), gapprot); 310*1d9eae9fSMark Johnston } 311*1d9eae9fSMark Johnston 312a4f67738SDoug Rabson static int 313e9eabf59SPeter Wemm link_elf_link_preload(linker_class_t cls, const char *filename, 314e9eabf59SPeter Wemm linker_file_t *result) 3152d636ab0SPeter Wemm { 31670b7ffeeSIan Dowse Elf_Ehdr *hdr; 31770b7ffeeSIan Dowse Elf_Shdr *shdr; 31870b7ffeeSIan Dowse Elf_Sym *es; 31970b7ffeeSIan Dowse void *modptr, *baseptr, *sizeptr; 32070b7ffeeSIan Dowse char *type; 32170b7ffeeSIan Dowse elf_file_t ef; 32270b7ffeeSIan Dowse linker_file_t lf; 32370b7ffeeSIan Dowse Elf_Addr off; 32470b7ffeeSIan Dowse int error, i, j, pb, ra, rl, shstrindex, symstrindex, symtabindex; 32570b7ffeeSIan Dowse 32670b7ffeeSIan Dowse /* Look to see if we have the file preloaded */ 32770b7ffeeSIan Dowse modptr = preload_search_by_name(filename); 32870b7ffeeSIan Dowse if (modptr == NULL) 32970b7ffeeSIan Dowse return ENOENT; 33070b7ffeeSIan Dowse 33170b7ffeeSIan Dowse type = (char *)preload_search_info(modptr, MODINFO_TYPE); 33270b7ffeeSIan Dowse baseptr = preload_search_info(modptr, MODINFO_ADDR); 33370b7ffeeSIan Dowse sizeptr = preload_search_info(modptr, MODINFO_SIZE); 33470b7ffeeSIan Dowse hdr = (Elf_Ehdr *)preload_search_info(modptr, MODINFO_METADATA | 33570b7ffeeSIan Dowse MODINFOMD_ELFHDR); 33670b7ffeeSIan Dowse shdr = (Elf_Shdr *)preload_search_info(modptr, MODINFO_METADATA | 33770b7ffeeSIan Dowse MODINFOMD_SHDR); 33870b7ffeeSIan Dowse if (type == NULL || (strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE) 33970b7ffeeSIan Dowse " obj module") != 0 && 34070b7ffeeSIan Dowse strcmp(type, "elf obj module") != 0)) { 341de78ca7eSPeter Wemm return (EFTYPE); 342de78ca7eSPeter Wemm } 34370b7ffeeSIan Dowse if (baseptr == NULL || sizeptr == NULL || hdr == NULL || 34470b7ffeeSIan Dowse shdr == NULL) 34570b7ffeeSIan Dowse return (EINVAL); 34670b7ffeeSIan Dowse 34770b7ffeeSIan Dowse lf = linker_make_file(filename, &link_elf_class); 34870b7ffeeSIan Dowse if (lf == NULL) 34970b7ffeeSIan Dowse return (ENOMEM); 35070b7ffeeSIan Dowse 35170b7ffeeSIan Dowse ef = (elf_file_t)lf; 35270b7ffeeSIan Dowse ef->preloaded = 1; 35370b7ffeeSIan Dowse ef->address = *(caddr_t *)baseptr; 35470b7ffeeSIan Dowse lf->address = *(caddr_t *)baseptr; 35570b7ffeeSIan Dowse lf->size = *(size_t *)sizeptr; 35670b7ffeeSIan Dowse 35770b7ffeeSIan Dowse if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS || 35870b7ffeeSIan Dowse hdr->e_ident[EI_DATA] != ELF_TARG_DATA || 35970b7ffeeSIan Dowse hdr->e_ident[EI_VERSION] != EV_CURRENT || 36070b7ffeeSIan Dowse hdr->e_version != EV_CURRENT || 36170b7ffeeSIan Dowse hdr->e_type != ET_REL || 36270b7ffeeSIan Dowse hdr->e_machine != ELF_TARG_MACH) { 36370b7ffeeSIan Dowse error = EFTYPE; 36470b7ffeeSIan Dowse goto out; 36570b7ffeeSIan Dowse } 36670b7ffeeSIan Dowse ef->e_shdr = shdr; 36770b7ffeeSIan Dowse 36870b7ffeeSIan Dowse /* Scan the section header for information and table sizing. */ 36970b7ffeeSIan Dowse symtabindex = -1; 37070b7ffeeSIan Dowse symstrindex = -1; 37170b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 37270b7ffeeSIan Dowse switch (shdr[i].sh_type) { 37370b7ffeeSIan Dowse case SHT_PROGBITS: 37470b7ffeeSIan Dowse case SHT_NOBITS: 37513f28d96SKonstantin Belousov #ifdef __amd64__ 376b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 37713f28d96SKonstantin Belousov #endif 37858c4aee0SJohn Baldwin /* Ignore sections not loaded by the loader. */ 37958c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 38058c4aee0SJohn Baldwin break; 38170b7ffeeSIan Dowse ef->nprogtab++; 38270b7ffeeSIan Dowse break; 38370b7ffeeSIan Dowse case SHT_SYMTAB: 38470b7ffeeSIan Dowse symtabindex = i; 38570b7ffeeSIan Dowse symstrindex = shdr[i].sh_link; 38670b7ffeeSIan Dowse break; 38770b7ffeeSIan Dowse case SHT_REL: 38815746ef4SJohn Baldwin /* 38915746ef4SJohn Baldwin * Ignore relocation tables for sections not 39015746ef4SJohn Baldwin * loaded by the loader. 39115746ef4SJohn Baldwin */ 39215746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 39315746ef4SJohn Baldwin break; 394b474c780SDavid E. O'Brien ef->nreltab++; 39570b7ffeeSIan Dowse break; 39670b7ffeeSIan Dowse case SHT_RELA: 39715746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 39815746ef4SJohn Baldwin break; 399b474c780SDavid E. O'Brien ef->nrelatab++; 40070b7ffeeSIan Dowse break; 40170b7ffeeSIan Dowse } 40270b7ffeeSIan Dowse } 40370b7ffeeSIan Dowse 40470b7ffeeSIan Dowse shstrindex = hdr->e_shstrndx; 40570b7ffeeSIan Dowse if (ef->nprogtab == 0 || symstrindex < 0 || 40670b7ffeeSIan Dowse symstrindex >= hdr->e_shnum || 40770b7ffeeSIan Dowse shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 || 40870b7ffeeSIan Dowse shstrindex >= hdr->e_shnum || 40970b7ffeeSIan Dowse shdr[shstrindex].sh_type != SHT_STRTAB) { 41070b7ffeeSIan Dowse printf("%s: bad/missing section headers\n", filename); 41170b7ffeeSIan Dowse error = ENOEXEC; 41270b7ffeeSIan Dowse goto out; 41370b7ffeeSIan Dowse } 41470b7ffeeSIan Dowse 41570b7ffeeSIan Dowse /* Allocate space for tracking the load chunks */ 41670b7ffeeSIan Dowse if (ef->nprogtab != 0) 41770b7ffeeSIan Dowse ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 41870b7ffeeSIan Dowse M_LINKER, M_WAITOK | M_ZERO); 419b474c780SDavid E. O'Brien if (ef->nreltab != 0) 420b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 421b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 422b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 423b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 424b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 42570b7ffeeSIan Dowse if ((ef->nprogtab != 0 && ef->progtab == NULL) || 426b474c780SDavid E. O'Brien (ef->nreltab != 0 && ef->reltab == NULL) || 427b474c780SDavid E. O'Brien (ef->nrelatab != 0 && ef->relatab == NULL)) { 42870b7ffeeSIan Dowse error = ENOMEM; 42970b7ffeeSIan Dowse goto out; 43070b7ffeeSIan Dowse } 43170b7ffeeSIan Dowse 43270b7ffeeSIan Dowse /* XXX, relocate the sh_addr fields saved by the loader. */ 43370b7ffeeSIan Dowse off = 0; 43470b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 43570b7ffeeSIan Dowse if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off)) 43670b7ffeeSIan Dowse off = shdr[i].sh_addr; 43770b7ffeeSIan Dowse } 43870b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 43970b7ffeeSIan Dowse if (shdr[i].sh_addr != 0) 44070b7ffeeSIan Dowse shdr[i].sh_addr = shdr[i].sh_addr - off + 44170b7ffeeSIan Dowse (Elf_Addr)ef->address; 44270b7ffeeSIan Dowse } 44370b7ffeeSIan Dowse 44470b7ffeeSIan Dowse ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 44570b7ffeeSIan Dowse ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr; 44670b7ffeeSIan Dowse ef->ddbstrcnt = shdr[symstrindex].sh_size; 44770b7ffeeSIan Dowse ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr; 44870b7ffeeSIan Dowse ef->shstrcnt = shdr[shstrindex].sh_size; 44970b7ffeeSIan Dowse ef->shstrtab = (char *)shdr[shstrindex].sh_addr; 45070b7ffeeSIan Dowse 45170b7ffeeSIan Dowse /* Now fill out progtab and the relocation tables. */ 45270b7ffeeSIan Dowse pb = 0; 45370b7ffeeSIan Dowse rl = 0; 45470b7ffeeSIan Dowse ra = 0; 45570b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 45670b7ffeeSIan Dowse switch (shdr[i].sh_type) { 45770b7ffeeSIan Dowse case SHT_PROGBITS: 45870b7ffeeSIan Dowse case SHT_NOBITS: 45913f28d96SKonstantin Belousov #ifdef __amd64__ 460b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 46113f28d96SKonstantin Belousov #endif 46258c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 46358c4aee0SJohn Baldwin break; 46470b7ffeeSIan Dowse ef->progtab[pb].addr = (void *)shdr[i].sh_addr; 46570b7ffeeSIan Dowse if (shdr[i].sh_type == SHT_PROGBITS) 46670b7ffeeSIan Dowse ef->progtab[pb].name = "<<PROGBITS>>"; 46713f28d96SKonstantin Belousov #ifdef __amd64__ 468b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 46913f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 47013f28d96SKonstantin Belousov #endif 47170b7ffeeSIan Dowse else 47270b7ffeeSIan Dowse ef->progtab[pb].name = "<<NOBITS>>"; 47370b7ffeeSIan Dowse ef->progtab[pb].size = shdr[i].sh_size; 474*1d9eae9fSMark Johnston ef->progtab[pb].flags = shdr[i].sh_flags; 47570b7ffeeSIan Dowse ef->progtab[pb].sec = i; 47670b7ffeeSIan Dowse if (ef->shstrtab && shdr[i].sh_name != 0) 47770b7ffeeSIan Dowse ef->progtab[pb].name = 47870b7ffeeSIan Dowse ef->shstrtab + shdr[i].sh_name; 47950c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 4805f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 48150c202c5SJeff Roberson void *dpcpu; 48250c202c5SJeff Roberson 48350c202c5SJeff Roberson dpcpu = dpcpu_alloc(shdr[i].sh_size); 48450c202c5SJeff Roberson if (dpcpu == NULL) { 4850f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 4869afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 4879afc5684SBjoern A. Zeeb "for %s\n", __func__, 4889afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 4890f823b64SBjoern A. Zeeb filename); 49050c202c5SJeff Roberson error = ENOSPC; 49150c202c5SJeff Roberson goto out; 49250c202c5SJeff Roberson } 49350c202c5SJeff Roberson memcpy(dpcpu, ef->progtab[pb].addr, 49450c202c5SJeff Roberson ef->progtab[pb].size); 49550c202c5SJeff Roberson dpcpu_copy(dpcpu, shdr[i].sh_size); 49650c202c5SJeff Roberson ef->progtab[pb].addr = dpcpu; 497eddfbb76SRobert Watson #ifdef VIMAGE 498eddfbb76SRobert Watson } else if (ef->progtab[pb].name != NULL && 49917ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 500eddfbb76SRobert Watson void *vnet_data; 501eddfbb76SRobert Watson 502eddfbb76SRobert Watson vnet_data = vnet_data_alloc(shdr[i].sh_size); 503eddfbb76SRobert Watson if (vnet_data == NULL) { 5040f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 5059afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 5069afc5684SBjoern A. Zeeb "for %s\n", __func__, 5079afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 5080f823b64SBjoern A. Zeeb filename); 509eddfbb76SRobert Watson error = ENOSPC; 510eddfbb76SRobert Watson goto out; 511eddfbb76SRobert Watson } 512eddfbb76SRobert Watson memcpy(vnet_data, ef->progtab[pb].addr, 513eddfbb76SRobert Watson ef->progtab[pb].size); 514eddfbb76SRobert Watson vnet_data_copy(vnet_data, shdr[i].sh_size); 515eddfbb76SRobert Watson ef->progtab[pb].addr = vnet_data; 516eddfbb76SRobert Watson #endif 5170067051fSMarcel Moolenaar } else if (ef->progtab[pb].name != NULL && 5180067051fSMarcel Moolenaar !strcmp(ef->progtab[pb].name, ".ctors")) { 5190067051fSMarcel Moolenaar lf->ctors_addr = ef->progtab[pb].addr; 5200067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 52150c202c5SJeff Roberson } 52270b7ffeeSIan Dowse 52370b7ffeeSIan Dowse /* Update all symbol values with the offset. */ 52470b7ffeeSIan Dowse for (j = 0; j < ef->ddbsymcnt; j++) { 52570b7ffeeSIan Dowse es = &ef->ddbsymtab[j]; 52670b7ffeeSIan Dowse if (es->st_shndx != i) 52770b7ffeeSIan Dowse continue; 52870b7ffeeSIan Dowse es->st_value += (Elf_Addr)ef->progtab[pb].addr; 52970b7ffeeSIan Dowse } 53070b7ffeeSIan Dowse pb++; 53170b7ffeeSIan Dowse break; 53270b7ffeeSIan Dowse case SHT_REL: 53315746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 53415746ef4SJohn Baldwin break; 53570b7ffeeSIan Dowse ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr; 53670b7ffeeSIan Dowse ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 53770b7ffeeSIan Dowse ef->reltab[rl].sec = shdr[i].sh_info; 53870b7ffeeSIan Dowse rl++; 53970b7ffeeSIan Dowse break; 54070b7ffeeSIan Dowse case SHT_RELA: 54115746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 54215746ef4SJohn Baldwin break; 54370b7ffeeSIan Dowse ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr; 54470b7ffeeSIan Dowse ef->relatab[ra].nrela = 54570b7ffeeSIan Dowse shdr[i].sh_size / sizeof(Elf_Rela); 54670b7ffeeSIan Dowse ef->relatab[ra].sec = shdr[i].sh_info; 54770b7ffeeSIan Dowse ra++; 54870b7ffeeSIan Dowse break; 54970b7ffeeSIan Dowse } 55070b7ffeeSIan Dowse } 5513cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 5523cfce8e4SKonstantin Belousov printf("%s: lost progbits\n", filename); 5533cfce8e4SKonstantin Belousov error = ENOEXEC; 5543cfce8e4SKonstantin Belousov goto out; 5553cfce8e4SKonstantin Belousov } 5563cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 5573cfce8e4SKonstantin Belousov printf("%s: lost reltab\n", filename); 5583cfce8e4SKonstantin Belousov error = ENOEXEC; 5593cfce8e4SKonstantin Belousov goto out; 5603cfce8e4SKonstantin Belousov } 5613cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 5623cfce8e4SKonstantin Belousov printf("%s: lost relatab\n", filename); 5633cfce8e4SKonstantin Belousov error = ENOEXEC; 5643cfce8e4SKonstantin Belousov goto out; 5653cfce8e4SKonstantin Belousov } 56670b7ffeeSIan Dowse 56770b7ffeeSIan Dowse /* Local intra-module relocations */ 568289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 5693cfce8e4SKonstantin Belousov if (error != 0) 5703cfce8e4SKonstantin Belousov goto out; 57170b7ffeeSIan Dowse *result = lf; 57270b7ffeeSIan Dowse return (0); 57370b7ffeeSIan Dowse 57470b7ffeeSIan Dowse out: 57570b7ffeeSIan Dowse /* preload not done this way */ 57670b7ffeeSIan Dowse linker_file_unload(lf, LINKER_UNLOAD_FORCE); 57770b7ffeeSIan Dowse return (error); 57870b7ffeeSIan Dowse } 579de78ca7eSPeter Wemm 5800067051fSMarcel Moolenaar static void 5810067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size) 5820067051fSMarcel Moolenaar { 5830067051fSMarcel Moolenaar void (**ctor)(void); 5840067051fSMarcel Moolenaar size_t i, cnt; 5850067051fSMarcel Moolenaar 5860067051fSMarcel Moolenaar if (addr == NULL || size == 0) 5870067051fSMarcel Moolenaar return; 5880067051fSMarcel Moolenaar cnt = size / sizeof(*ctor); 5890067051fSMarcel Moolenaar ctor = (void *)addr; 5900067051fSMarcel Moolenaar for (i = 0; i < cnt; i++) { 5910067051fSMarcel Moolenaar if (ctor[i] != NULL) 5920067051fSMarcel Moolenaar (*ctor[i])(); 5930067051fSMarcel Moolenaar } 5940067051fSMarcel Moolenaar } 5950067051fSMarcel Moolenaar 596de78ca7eSPeter Wemm static int 59754823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf) 59854823af2SPeter Wemm { 59970b7ffeeSIan Dowse elf_file_t ef; 60070b7ffeeSIan Dowse int error; 60170b7ffeeSIan Dowse 60270b7ffeeSIan Dowse ef = (elf_file_t)lf; 60370b7ffeeSIan Dowse error = relocate_file(ef); 60470b7ffeeSIan Dowse if (error) 605289ead7cSKonstantin Belousov return (error); 60670b7ffeeSIan Dowse 60770b7ffeeSIan Dowse /* Notify MD code that a module is being loaded. */ 60870b7ffeeSIan Dowse error = elf_cpu_load_file(lf); 60970b7ffeeSIan Dowse if (error) 61070b7ffeeSIan Dowse return (error); 61170b7ffeeSIan Dowse 6128f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 613289ead7cSKonstantin Belousov /* Now ifuncs. */ 614289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 615289ead7cSKonstantin Belousov if (error != 0) 616289ead7cSKonstantin Belousov return (error); 6178f941950SKonstantin Belousov #endif 618289ead7cSKonstantin Belousov 6190067051fSMarcel Moolenaar /* Invoke .ctors */ 6200067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 62170b7ffeeSIan Dowse return (0); 62254823af2SPeter Wemm } 62354823af2SPeter Wemm 62454823af2SPeter Wemm static int 625c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename, 626c143d6c2SMarcel Moolenaar linker_file_t *result) 627a4f67738SDoug Rabson { 628a344babfSGleb Smirnoff struct nameidata *nd; 629b40ce416SJulian Elischer struct thread *td = curthread; /* XXX */ 630caab6e90SPeter Wemm Elf_Ehdr *hdr; 631e9eabf59SPeter Wemm Elf_Shdr *shdr; 632a8774e39SPeter Wemm Elf_Sym *es; 633a8774e39SPeter Wemm int nbytes, i, j; 63420947801SPeter Wemm vm_offset_t mapbase; 635fe3db7c7SDoug Rabson size_t mapsize; 636a4f67738SDoug Rabson int error = 0; 637526d0bd5SKonstantin Belousov ssize_t resid; 638526d0bd5SKonstantin Belousov int flags; 639a4f67738SDoug Rabson elf_file_t ef; 640a4f67738SDoug Rabson linker_file_t lf; 641ca65d5c7SPeter Wemm int symtabindex; 642ca65d5c7SPeter Wemm int symstrindex; 643e9eabf59SPeter Wemm int shstrindex; 644e9eabf59SPeter Wemm int nsym; 64520947801SPeter Wemm int pb, rl, ra; 646e9eabf59SPeter Wemm int alignmask; 647ca65d5c7SPeter Wemm 648ca65d5c7SPeter Wemm shdr = NULL; 649ca65d5c7SPeter Wemm lf = NULL; 650e9eabf59SPeter Wemm mapsize = 0; 651e9eabf59SPeter Wemm hdr = NULL; 652a4f67738SDoug Rabson 653a344babfSGleb Smirnoff nd = malloc(sizeof(struct nameidata), M_TEMP, M_WAITOK); 654a344babfSGleb Smirnoff NDINIT(nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td); 655e6796b67SKirk McKusick flags = FREAD; 656a344babfSGleb Smirnoff error = vn_open(nd, &flags, 0, NULL); 657a344babfSGleb Smirnoff if (error) { 658a344babfSGleb Smirnoff free(nd, M_TEMP); 659a4f67738SDoug Rabson return error; 660a344babfSGleb Smirnoff } 661a344babfSGleb Smirnoff NDFREE(nd, NDF_ONLY_PNBUF); 662a344babfSGleb Smirnoff if (nd->ni_vp->v_type != VREG) { 6631f49b573SRuslan Ermilov error = ENOEXEC; 6641f49b573SRuslan Ermilov goto out; 6651f49b573SRuslan Ermilov } 666a3df768bSRobert Watson #ifdef MAC 667a344babfSGleb Smirnoff error = mac_kld_check_load(td->td_ucred, nd->ni_vp); 668a3df768bSRobert Watson if (error) { 669a3df768bSRobert Watson goto out; 670a3df768bSRobert Watson } 671a3df768bSRobert Watson #endif 672a4f67738SDoug Rabson 673e9eabf59SPeter Wemm /* Read the elf header from the file. */ 674e9eabf59SPeter Wemm hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK); 675a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)hdr, sizeof(*hdr), 0, 6769ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 6779ca43589SRobert Watson &resid, td); 678a4f67738SDoug Rabson if (error) 679a4f67738SDoug Rabson goto out; 680e9eabf59SPeter Wemm if (resid != 0){ 681e9eabf59SPeter Wemm error = ENOEXEC; 682e9eabf59SPeter Wemm goto out; 683e9eabf59SPeter Wemm } 684a4f67738SDoug Rabson 685caab6e90SPeter Wemm if (!IS_ELF(*hdr)) { 686fe3db7c7SDoug Rabson error = ENOEXEC; 687a4f67738SDoug Rabson goto out; 688fe3db7c7SDoug Rabson } 689fe3db7c7SDoug Rabson 690caab6e90SPeter Wemm if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS 691caab6e90SPeter Wemm || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 692552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file layout"); 693fe3db7c7SDoug Rabson error = ENOEXEC; 694fe3db7c7SDoug Rabson goto out; 695fe3db7c7SDoug Rabson } 696caab6e90SPeter Wemm if (hdr->e_ident[EI_VERSION] != EV_CURRENT 697caab6e90SPeter Wemm || hdr->e_version != EV_CURRENT) { 698552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file version"); 699fe3db7c7SDoug Rabson error = ENOEXEC; 700fe3db7c7SDoug Rabson goto out; 701fe3db7c7SDoug Rabson } 702e9eabf59SPeter Wemm if (hdr->e_type != ET_REL) { 703a1d7ce03SAttilio Rao error = ENOSYS; 704fe3db7c7SDoug Rabson goto out; 705fe3db7c7SDoug Rabson } 706caab6e90SPeter Wemm if (hdr->e_machine != ELF_TARG_MACH) { 707552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported machine"); 708fe3db7c7SDoug Rabson error = ENOEXEC; 709fe3db7c7SDoug Rabson goto out; 710fe3db7c7SDoug Rabson } 711a4f67738SDoug Rabson 712326e27d8SDoug Rabson lf = linker_make_file(filename, &link_elf_class); 713326e27d8SDoug Rabson if (!lf) { 714326e27d8SDoug Rabson error = ENOMEM; 715326e27d8SDoug Rabson goto out; 716326e27d8SDoug Rabson } 717326e27d8SDoug Rabson ef = (elf_file_t) lf; 718e9eabf59SPeter Wemm ef->nprogtab = 0; 719e9eabf59SPeter Wemm ef->e_shdr = 0; 720b474c780SDavid E. O'Brien ef->nreltab = 0; 721b474c780SDavid E. O'Brien ef->nrelatab = 0; 722e9eabf59SPeter Wemm 723e9eabf59SPeter Wemm /* Allocate and read in the section header */ 724e9eabf59SPeter Wemm nbytes = hdr->e_shnum * hdr->e_shentsize; 725e9eabf59SPeter Wemm if (nbytes == 0 || hdr->e_shoff == 0 || 726e9eabf59SPeter Wemm hdr->e_shentsize != sizeof(Elf_Shdr)) { 727e9eabf59SPeter Wemm error = ENOEXEC; 728e9eabf59SPeter Wemm goto out; 729e9eabf59SPeter Wemm } 73020947801SPeter Wemm shdr = malloc(nbytes, M_LINKER, M_WAITOK); 731e9eabf59SPeter Wemm ef->e_shdr = shdr; 732a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (caddr_t)shdr, nbytes, 733a344babfSGleb Smirnoff hdr->e_shoff, UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 734a344babfSGleb Smirnoff NOCRED, &resid, td); 735e9eabf59SPeter Wemm if (error) 736e9eabf59SPeter Wemm goto out; 737e9eabf59SPeter Wemm if (resid) { 738e9eabf59SPeter Wemm error = ENOEXEC; 739e9eabf59SPeter Wemm goto out; 740e9eabf59SPeter Wemm } 741e9eabf59SPeter Wemm 742e9eabf59SPeter Wemm /* Scan the section header for information and table sizing. */ 743e9eabf59SPeter Wemm nsym = 0; 744e9eabf59SPeter Wemm symtabindex = -1; 745e9eabf59SPeter Wemm symstrindex = -1; 746e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 747676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 748676799a0SAndriy Gapon continue; 749e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 750e9eabf59SPeter Wemm case SHT_PROGBITS: 751e9eabf59SPeter Wemm case SHT_NOBITS: 75213f28d96SKonstantin Belousov #ifdef __amd64__ 753b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 75413f28d96SKonstantin Belousov #endif 75558c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 75658c4aee0SJohn Baldwin break; 75720947801SPeter Wemm ef->nprogtab++; 758e9eabf59SPeter Wemm break; 759e9eabf59SPeter Wemm case SHT_SYMTAB: 760e9eabf59SPeter Wemm nsym++; 761e9eabf59SPeter Wemm symtabindex = i; 762e9eabf59SPeter Wemm symstrindex = shdr[i].sh_link; 763e9eabf59SPeter Wemm break; 764e9eabf59SPeter Wemm case SHT_REL: 76515746ef4SJohn Baldwin /* 76615746ef4SJohn Baldwin * Ignore relocation tables for unallocated 76715746ef4SJohn Baldwin * sections. 76815746ef4SJohn Baldwin */ 76915746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 77015746ef4SJohn Baldwin break; 771b474c780SDavid E. O'Brien ef->nreltab++; 772e9eabf59SPeter Wemm break; 773e9eabf59SPeter Wemm case SHT_RELA: 77415746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 77515746ef4SJohn Baldwin break; 776b474c780SDavid E. O'Brien ef->nrelatab++; 777e9eabf59SPeter Wemm break; 778e9eabf59SPeter Wemm case SHT_STRTAB: 779e9eabf59SPeter Wemm break; 780e9eabf59SPeter Wemm } 781e9eabf59SPeter Wemm } 78220947801SPeter Wemm if (ef->nprogtab == 0) { 783552f9f63SEdwin Groothuis link_elf_error(filename, "file has no contents"); 784e9eabf59SPeter Wemm error = ENOEXEC; 785e9eabf59SPeter Wemm goto out; 786e9eabf59SPeter Wemm } 787e9eabf59SPeter Wemm if (nsym != 1) { 788e9eabf59SPeter Wemm /* Only allow one symbol table for now */ 789f912a970SEd Maste link_elf_error(filename, 790f912a970SEd Maste "file must have exactly one symbol table"); 791e9eabf59SPeter Wemm error = ENOEXEC; 792e9eabf59SPeter Wemm goto out; 793e9eabf59SPeter Wemm } 794e9eabf59SPeter Wemm if (symstrindex < 0 || symstrindex > hdr->e_shnum || 795e9eabf59SPeter Wemm shdr[symstrindex].sh_type != SHT_STRTAB) { 796552f9f63SEdwin Groothuis link_elf_error(filename, "file has invalid symbol strings"); 797e9eabf59SPeter Wemm error = ENOEXEC; 798e9eabf59SPeter Wemm goto out; 799e9eabf59SPeter Wemm } 800e9eabf59SPeter Wemm 801e9eabf59SPeter Wemm /* Allocate space for tracking the load chunks */ 802e9eabf59SPeter Wemm if (ef->nprogtab != 0) 80320947801SPeter Wemm ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 80420947801SPeter Wemm M_LINKER, M_WAITOK | M_ZERO); 805b474c780SDavid E. O'Brien if (ef->nreltab != 0) 806b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 807b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 808b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 809b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 810b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 811e9eabf59SPeter Wemm 8123cfce8e4SKonstantin Belousov if (symtabindex == -1) { 8133cfce8e4SKonstantin Belousov link_elf_error(filename, "lost symbol table index"); 8143cfce8e4SKonstantin Belousov error = ENOEXEC; 8153cfce8e4SKonstantin Belousov goto out; 8163cfce8e4SKonstantin Belousov } 81720947801SPeter Wemm /* Allocate space for and load the symbol table */ 818e9eabf59SPeter Wemm ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 819e9eabf59SPeter Wemm ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK); 820a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)ef->ddbsymtab, 82120947801SPeter Wemm shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset, 82220947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 82320947801SPeter Wemm &resid, td); 82420947801SPeter Wemm if (error) 82520947801SPeter Wemm goto out; 82620947801SPeter Wemm if (resid != 0){ 82720947801SPeter Wemm error = EINVAL; 82820947801SPeter Wemm goto out; 82920947801SPeter Wemm } 830e9eabf59SPeter Wemm 83120947801SPeter Wemm /* Allocate space for and load the symbol strings */ 832e9eabf59SPeter Wemm ef->ddbstrcnt = shdr[symstrindex].sh_size; 833e9eabf59SPeter Wemm ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK); 834a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab, 83520947801SPeter Wemm shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset, 83620947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 83720947801SPeter Wemm &resid, td); 83820947801SPeter Wemm if (error) 83920947801SPeter Wemm goto out; 84020947801SPeter Wemm if (resid != 0){ 84120947801SPeter Wemm error = EINVAL; 84220947801SPeter Wemm goto out; 84320947801SPeter Wemm } 844e9eabf59SPeter Wemm 845e9eabf59SPeter Wemm /* Do we have a string table for the section names? */ 846e9eabf59SPeter Wemm shstrindex = -1; 84720947801SPeter Wemm if (hdr->e_shstrndx != 0 && 84820947801SPeter Wemm shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) { 849e9eabf59SPeter Wemm shstrindex = hdr->e_shstrndx; 850e9eabf59SPeter Wemm ef->shstrcnt = shdr[shstrindex].sh_size; 85120947801SPeter Wemm ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER, 85220947801SPeter Wemm M_WAITOK); 853a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab, 85420947801SPeter Wemm shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset, 855e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 856e9eabf59SPeter Wemm &resid, td); 857e9eabf59SPeter Wemm if (error) 858e9eabf59SPeter Wemm goto out; 85920947801SPeter Wemm if (resid != 0){ 86020947801SPeter Wemm error = EINVAL; 86120947801SPeter Wemm goto out; 86220947801SPeter Wemm } 863e9eabf59SPeter Wemm } 864e9eabf59SPeter Wemm 86520947801SPeter Wemm /* Size up code/data(progbits) and bss(nobits). */ 866e9eabf59SPeter Wemm alignmask = 0; 867e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 868676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 869676799a0SAndriy Gapon continue; 870e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 871e9eabf59SPeter Wemm case SHT_PROGBITS: 872e9eabf59SPeter Wemm case SHT_NOBITS: 87313f28d96SKonstantin Belousov #ifdef __amd64__ 874b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 87513f28d96SKonstantin Belousov #endif 87658c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 87758c4aee0SJohn Baldwin break; 878e9eabf59SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 879e9eabf59SPeter Wemm mapsize += alignmask; 880e9eabf59SPeter Wemm mapsize &= ~alignmask; 881e9eabf59SPeter Wemm mapsize += shdr[i].sh_size; 882e9eabf59SPeter Wemm break; 883e9eabf59SPeter Wemm } 884e9eabf59SPeter Wemm } 885e9eabf59SPeter Wemm 886e9eabf59SPeter Wemm /* 887e9eabf59SPeter Wemm * We know how much space we need for the text/data/bss/etc. 888e9eabf59SPeter Wemm * This stuff needs to be in a single chunk so that profiling etc 889e9eabf59SPeter Wemm * can get the bounds and gdb can associate offsets with modules 890e9eabf59SPeter Wemm */ 8915131cba6SMark Johnston ef->object = vm_object_allocate(OBJT_PHYS, atop(round_page(mapsize))); 892fe3db7c7SDoug Rabson if (ef->object == NULL) { 893fe3db7c7SDoug Rabson error = ENOMEM; 894fe3db7c7SDoug Rabson goto out; 895fe3db7c7SDoug Rabson } 896ac68d1c9SAlan Cox 897ac68d1c9SAlan Cox /* 898ac68d1c9SAlan Cox * In order to satisfy amd64's architectural requirements on the 899ac68d1c9SAlan Cox * location of code and data in the kernel's address space, request a 900ac68d1c9SAlan Cox * mapping that is above the kernel. 901*1d9eae9fSMark Johnston * 902*1d9eae9fSMark Johnston * Protections will be restricted once relocations are applied. 903ac68d1c9SAlan Cox */ 9044104e835SOleksandr Tymoshenko #ifdef __amd64__ 905ac68d1c9SAlan Cox mapbase = KERNBASE; 9064104e835SOleksandr Tymoshenko #else 9074104e835SOleksandr Tymoshenko mapbase = VM_MIN_KERNEL_ADDRESS; 9084104e835SOleksandr Tymoshenko #endif 90920947801SPeter Wemm error = vm_map_find(kernel_map, ef->object, 0, &mapbase, 910edb572a3SJohn Baldwin round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL, 911edb572a3SJohn Baldwin VM_PROT_ALL, 0); 912092bacb2SMark Johnston if (error != KERN_SUCCESS) { 913fe3db7c7SDoug Rabson vm_object_deallocate(ef->object); 914092bacb2SMark Johnston ef->object = NULL; 915092bacb2SMark Johnston error = ENOMEM; 916fe3db7c7SDoug Rabson goto out; 917fe3db7c7SDoug Rabson } 91820947801SPeter Wemm 919e9eabf59SPeter Wemm /* Wire the pages */ 9207f1ef325SAlan Cox error = vm_map_wire(kernel_map, mapbase, 92120947801SPeter Wemm mapbase + round_page(mapsize), 922e9eabf59SPeter Wemm VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES); 9237f1ef325SAlan Cox if (error != KERN_SUCCESS) { 9247f1ef325SAlan Cox error = ENOMEM; 9257f1ef325SAlan Cox goto out; 9267f1ef325SAlan Cox } 927a4f67738SDoug Rabson 92820947801SPeter Wemm /* Inform the kld system about the situation */ 92920947801SPeter Wemm lf->address = ef->address = (caddr_t)mapbase; 93020947801SPeter Wemm lf->size = mapsize; 931e9eabf59SPeter Wemm 932e9eabf59SPeter Wemm /* 93320947801SPeter Wemm * Now load code/data(progbits), zero bss(nobits), allocate space for 93420947801SPeter Wemm * and load relocs 935e9eabf59SPeter Wemm */ 93620947801SPeter Wemm pb = 0; 93720947801SPeter Wemm rl = 0; 93820947801SPeter Wemm ra = 0; 93920947801SPeter Wemm alignmask = 0; 94020947801SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 941676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 942676799a0SAndriy Gapon continue; 94320947801SPeter Wemm switch (shdr[i].sh_type) { 94420947801SPeter Wemm case SHT_PROGBITS: 94520947801SPeter Wemm case SHT_NOBITS: 94613f28d96SKonstantin Belousov #ifdef __amd64__ 947b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 94813f28d96SKonstantin Belousov #endif 94958c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 95058c4aee0SJohn Baldwin break; 95120947801SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 95220947801SPeter Wemm mapbase += alignmask; 95320947801SPeter Wemm mapbase &= ~alignmask; 9540067051fSMarcel Moolenaar if (ef->shstrtab != NULL && shdr[i].sh_name != 0) { 95550c202c5SJeff Roberson ef->progtab[pb].name = 95650c202c5SJeff Roberson ef->shstrtab + shdr[i].sh_name; 9570067051fSMarcel Moolenaar if (!strcmp(ef->progtab[pb].name, ".ctors")) { 9580067051fSMarcel Moolenaar lf->ctors_addr = (caddr_t)mapbase; 9590067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 9600067051fSMarcel Moolenaar } 9610067051fSMarcel Moolenaar } else if (shdr[i].sh_type == SHT_PROGBITS) 96220947801SPeter Wemm ef->progtab[pb].name = "<<PROGBITS>>"; 96313f28d96SKonstantin Belousov #ifdef __amd64__ 964b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 96513f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 96613f28d96SKonstantin Belousov #endif 96750c202c5SJeff Roberson else 96850c202c5SJeff Roberson ef->progtab[pb].name = "<<NOBITS>>"; 96950c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 9700f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 97150c202c5SJeff Roberson ef->progtab[pb].addr = 97250c202c5SJeff Roberson dpcpu_alloc(shdr[i].sh_size); 9730f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 9740f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 9759afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 9769afc5684SBjoern A. Zeeb "for %s\n", __func__, 9779afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 9780f823b64SBjoern A. Zeeb filename); 9790f823b64SBjoern A. Zeeb } 9800f823b64SBjoern A. Zeeb } 981eddfbb76SRobert Watson #ifdef VIMAGE 982eddfbb76SRobert Watson else if (ef->progtab[pb].name != NULL && 9830f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 984eddfbb76SRobert Watson ef->progtab[pb].addr = 985eddfbb76SRobert Watson vnet_data_alloc(shdr[i].sh_size); 9860f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 9870f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 9889afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 9899afc5684SBjoern A. Zeeb "for %s\n", __func__, 9909afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 9910f823b64SBjoern A. Zeeb filename); 9920f823b64SBjoern A. Zeeb } 9930f823b64SBjoern A. Zeeb } 994eddfbb76SRobert Watson #endif 99550c202c5SJeff Roberson else 99650c202c5SJeff Roberson ef->progtab[pb].addr = 99750c202c5SJeff Roberson (void *)(uintptr_t)mapbase; 99850c202c5SJeff Roberson if (ef->progtab[pb].addr == NULL) { 99950c202c5SJeff Roberson error = ENOSPC; 100050c202c5SJeff Roberson goto out; 100150c202c5SJeff Roberson } 100250c202c5SJeff Roberson ef->progtab[pb].size = shdr[i].sh_size; 1003*1d9eae9fSMark Johnston ef->progtab[pb].flags = shdr[i].sh_flags; 100450c202c5SJeff Roberson ef->progtab[pb].sec = i; 100513f28d96SKonstantin Belousov if (shdr[i].sh_type == SHT_PROGBITS 100613f28d96SKonstantin Belousov #ifdef __amd64__ 1007b715d9afSKonstantin Belousov || shdr[i].sh_type == SHT_X86_64_UNWIND 100813f28d96SKonstantin Belousov #endif 100913f28d96SKonstantin Belousov ) { 1010a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 101120947801SPeter Wemm ef->progtab[pb].addr, 101220947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 101320947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 101420947801SPeter Wemm NOCRED, &resid, td); 101520947801SPeter Wemm if (error) 101620947801SPeter Wemm goto out; 101720947801SPeter Wemm if (resid != 0){ 101820947801SPeter Wemm error = EINVAL; 101920947801SPeter Wemm goto out; 102020947801SPeter Wemm } 1021eddfbb76SRobert Watson /* Initialize the per-cpu or vnet area. */ 1022eddfbb76SRobert Watson if (ef->progtab[pb].addr != (void *)mapbase && 10235f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 102450c202c5SJeff Roberson dpcpu_copy(ef->progtab[pb].addr, 102550c202c5SJeff Roberson shdr[i].sh_size); 1026eddfbb76SRobert Watson #ifdef VIMAGE 1027eddfbb76SRobert Watson else if (ef->progtab[pb].addr != 1028eddfbb76SRobert Watson (void *)mapbase && 102917ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 1030eddfbb76SRobert Watson vnet_data_copy(ef->progtab[pb].addr, 1031eddfbb76SRobert Watson shdr[i].sh_size); 1032eddfbb76SRobert Watson #endif 103350c202c5SJeff Roberson } else 103420947801SPeter Wemm bzero(ef->progtab[pb].addr, shdr[i].sh_size); 1035a8774e39SPeter Wemm 1036a8774e39SPeter Wemm /* Update all symbol values with the offset. */ 1037a8774e39SPeter Wemm for (j = 0; j < ef->ddbsymcnt; j++) { 1038a8774e39SPeter Wemm es = &ef->ddbsymtab[j]; 1039a8774e39SPeter Wemm if (es->st_shndx != i) 1040a8774e39SPeter Wemm continue; 1041a8774e39SPeter Wemm es->st_value += (Elf_Addr)ef->progtab[pb].addr; 1042a8774e39SPeter Wemm } 104320947801SPeter Wemm mapbase += shdr[i].sh_size; 104420947801SPeter Wemm pb++; 104520947801SPeter Wemm break; 104620947801SPeter Wemm case SHT_REL: 104715746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 104815746ef4SJohn Baldwin break; 104920947801SPeter Wemm ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER, 105020947801SPeter Wemm M_WAITOK); 105120947801SPeter Wemm ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 105220947801SPeter Wemm ef->reltab[rl].sec = shdr[i].sh_info; 1053a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 105420947801SPeter Wemm (void *)ef->reltab[rl].rel, 105520947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 1056e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 1057e9eabf59SPeter Wemm &resid, td); 1058e9eabf59SPeter Wemm if (error) 1059e9eabf59SPeter Wemm goto out; 106020947801SPeter Wemm if (resid != 0){ 106120947801SPeter Wemm error = EINVAL; 106220947801SPeter Wemm goto out; 1063e9eabf59SPeter Wemm } 106420947801SPeter Wemm rl++; 106520947801SPeter Wemm break; 106620947801SPeter Wemm case SHT_RELA: 106715746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 106815746ef4SJohn Baldwin break; 106920947801SPeter Wemm ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER, 107020947801SPeter Wemm M_WAITOK); 107120947801SPeter Wemm ef->relatab[ra].nrela = 107220947801SPeter Wemm shdr[i].sh_size / sizeof(Elf_Rela); 107320947801SPeter Wemm ef->relatab[ra].sec = shdr[i].sh_info; 1074a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 107520947801SPeter Wemm (void *)ef->relatab[ra].rela, 107620947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 1077e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 1078e9eabf59SPeter Wemm &resid, td); 1079e9eabf59SPeter Wemm if (error) 1080e9eabf59SPeter Wemm goto out; 108120947801SPeter Wemm if (resid != 0){ 108220947801SPeter Wemm error = EINVAL; 108320947801SPeter Wemm goto out; 1084e9eabf59SPeter Wemm } 108520947801SPeter Wemm ra++; 108620947801SPeter Wemm break; 108720947801SPeter Wemm } 108820947801SPeter Wemm } 10893cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 10903cfce8e4SKonstantin Belousov link_elf_error(filename, "lost progbits"); 10913cfce8e4SKonstantin Belousov error = ENOEXEC; 10923cfce8e4SKonstantin Belousov goto out; 10933cfce8e4SKonstantin Belousov } 10943cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 10953cfce8e4SKonstantin Belousov link_elf_error(filename, "lost reltab"); 10963cfce8e4SKonstantin Belousov error = ENOEXEC; 10973cfce8e4SKonstantin Belousov goto out; 10983cfce8e4SKonstantin Belousov } 10993cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 11003cfce8e4SKonstantin Belousov link_elf_error(filename, "lost relatab"); 11013cfce8e4SKonstantin Belousov error = ENOEXEC; 11023cfce8e4SKonstantin Belousov goto out; 11033cfce8e4SKonstantin Belousov } 11043cfce8e4SKonstantin Belousov if (mapbase != (vm_offset_t)ef->address + mapsize) { 11053cfce8e4SKonstantin Belousov printf( 11063cfce8e4SKonstantin Belousov "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n", 11073cfce8e4SKonstantin Belousov filename != NULL ? filename : "<none>", 11086f3c6327SNeel Natu (u_long)mapbase, ef->address, (u_long)mapsize, 11096f3c6327SNeel Natu (u_long)(vm_offset_t)ef->address + mapsize); 11103cfce8e4SKonstantin Belousov error = ENOMEM; 11113cfce8e4SKonstantin Belousov goto out; 11123cfce8e4SKonstantin Belousov } 1113e9eabf59SPeter Wemm 1114e9eabf59SPeter Wemm /* Local intra-module relocations */ 1115289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 11163cfce8e4SKonstantin Belousov if (error != 0) 11173cfce8e4SKonstantin Belousov goto out; 11187251b4bfSJake Burkholder 1119e9eabf59SPeter Wemm /* Pull in dependencies */ 1120a344babfSGleb Smirnoff VOP_UNLOCK(nd->ni_vp, 0); 11217b9716baSIan Dowse error = linker_load_dependencies(lf); 1122a344babfSGleb Smirnoff vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY); 1123ca65d5c7SPeter Wemm if (error) 1124de78ca7eSPeter Wemm goto out; 1125e9eabf59SPeter Wemm 1126e9eabf59SPeter Wemm /* External relocations */ 1127326e27d8SDoug Rabson error = relocate_file(ef); 1128ca65d5c7SPeter Wemm if (error) 1129ca65d5c7SPeter Wemm goto out; 1130ca65d5c7SPeter Wemm 1131e9eabf59SPeter Wemm /* Notify MD code that a module is being loaded. */ 1132e9eabf59SPeter Wemm error = elf_cpu_load_file(lf); 1133ca65d5c7SPeter Wemm if (error) 1134ca65d5c7SPeter Wemm goto out; 1135a4f67738SDoug Rabson 11368f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 1137289ead7cSKonstantin Belousov /* Now ifuncs. */ 1138289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 1139289ead7cSKonstantin Belousov if (error != 0) 1140289ead7cSKonstantin Belousov goto out; 11418f941950SKonstantin Belousov #endif 1142289ead7cSKonstantin Belousov 1143*1d9eae9fSMark Johnston link_elf_protect(ef); 11440067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 1145a4f67738SDoug Rabson *result = lf; 1146a4f67738SDoug Rabson 1147a4f67738SDoug Rabson out: 1148a344babfSGleb Smirnoff VOP_UNLOCK(nd->ni_vp, 0); 1149a344babfSGleb Smirnoff vn_close(nd->ni_vp, FREAD, td->td_ucred, td); 1150a344babfSGleb Smirnoff free(nd, M_TEMP); 1151ca65d5c7SPeter Wemm if (error && lf) 115265a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 1153e9eabf59SPeter Wemm free(hdr, M_LINKER); 1154a4f67738SDoug Rabson 1155a4f67738SDoug Rabson return error; 1156a4f67738SDoug Rabson } 1157a4f67738SDoug Rabson 1158a4f67738SDoug Rabson static void 1159de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file) 1160a4f67738SDoug Rabson { 1161326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) file; 1162d821d364SPedro F. Giffuni u_int i; 1163a4f67738SDoug Rabson 11641aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being unloaded. */ 11651aeb23cdSMarcel Moolenaar elf_cpu_unload_file(file); 11661aeb23cdSMarcel Moolenaar 116750c202c5SJeff Roberson if (ef->progtab) { 116850c202c5SJeff Roberson for (i = 0; i < ef->nprogtab; i++) { 116950c202c5SJeff Roberson if (ef->progtab[i].size == 0) 117050c202c5SJeff Roberson continue; 117150c202c5SJeff Roberson if (ef->progtab[i].name == NULL) 117250c202c5SJeff Roberson continue; 11735f67450dSDimitry Andric if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME)) 117450c202c5SJeff Roberson dpcpu_free(ef->progtab[i].addr, 117550c202c5SJeff Roberson ef->progtab[i].size); 1176eddfbb76SRobert Watson #ifdef VIMAGE 117717ef1febSRobert Watson else if (!strcmp(ef->progtab[i].name, VNET_SETNAME)) 1178eddfbb76SRobert Watson vnet_data_free(ef->progtab[i].addr, 1179eddfbb76SRobert Watson ef->progtab[i].size); 1180eddfbb76SRobert Watson #endif 118150c202c5SJeff Roberson } 118250c202c5SJeff Roberson } 118370b7ffeeSIan Dowse if (ef->preloaded) { 118470b7ffeeSIan Dowse free(ef->reltab, M_LINKER); 118570b7ffeeSIan Dowse free(ef->relatab, M_LINKER); 118670b7ffeeSIan Dowse free(ef->progtab, M_LINKER); 1187a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1188a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1189a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 119073624a80SMark Johnston if (file->pathname != NULL) 119173624a80SMark Johnston preload_delete_name(file->pathname); 119270b7ffeeSIan Dowse return; 119370b7ffeeSIan Dowse } 119470b7ffeeSIan Dowse 1195b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) 119620947801SPeter Wemm free(ef->reltab[i].rel, M_LINKER); 1197b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) 119820947801SPeter Wemm free(ef->relatab[i].rela, M_LINKER); 119920947801SPeter Wemm free(ef->reltab, M_LINKER); 120020947801SPeter Wemm free(ef->relatab, M_LINKER); 120120947801SPeter Wemm free(ef->progtab, M_LINKER); 120220947801SPeter Wemm 1203092bacb2SMark Johnston if (ef->object != NULL) 1204fe3db7c7SDoug Rabson vm_map_remove(kernel_map, (vm_offset_t)ef->address, 1205092bacb2SMark Johnston (vm_offset_t)ef->address + ptoa(ef->object->size)); 1206e9eabf59SPeter Wemm free(ef->e_shdr, M_LINKER); 1207e9eabf59SPeter Wemm free(ef->ddbsymtab, M_LINKER); 1208e9eabf59SPeter Wemm free(ef->ddbstrtab, M_LINKER); 120920947801SPeter Wemm free(ef->shstrtab, M_LINKER); 1210a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1211a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1212a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 1213de78ca7eSPeter Wemm } 1214de78ca7eSPeter Wemm 1215fe3db7c7SDoug Rabson static const char * 1216757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info) 1217a4f67738SDoug Rabson { 1218fe3db7c7SDoug Rabson const Elf_Sym *ref; 1219a4f67738SDoug Rabson 1220aa855a59SPeter Wemm if (ELF_R_SYM(r_info)) { 1221e9eabf59SPeter Wemm ref = ef->ddbsymtab + ELF_R_SYM(r_info); 1222e9eabf59SPeter Wemm return ef->ddbstrtab + ref->st_name; 1223fe3db7c7SDoug Rabson } else 1224fe3db7c7SDoug Rabson return NULL; 1225a4f67738SDoug Rabson } 1226a4f67738SDoug Rabson 1227e9eabf59SPeter Wemm static Elf_Addr 1228e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec) 1229e9eabf59SPeter Wemm { 1230e9eabf59SPeter Wemm int i; 1231e9eabf59SPeter Wemm Elf_Addr base = 0; 1232e9eabf59SPeter Wemm 1233e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1234a8774e39SPeter Wemm if (sec == ef->progtab[i].sec) { 1235e9eabf59SPeter Wemm base = (Elf_Addr)ef->progtab[i].addr; 1236a8774e39SPeter Wemm break; 1237e9eabf59SPeter Wemm } 1238a8774e39SPeter Wemm } 1239e9eabf59SPeter Wemm return base; 1240e9eabf59SPeter Wemm } 1241e9eabf59SPeter Wemm 1242a4f67738SDoug Rabson static int 1243326e27d8SDoug Rabson relocate_file(elf_file_t ef) 1244a4f67738SDoug Rabson { 1245fe3db7c7SDoug Rabson const Elf_Rel *rellim; 1246fe3db7c7SDoug Rabson const Elf_Rel *rel; 1247fe3db7c7SDoug Rabson const Elf_Rela *relalim; 1248fe3db7c7SDoug Rabson const Elf_Rela *rela; 1249aa855a59SPeter Wemm const char *symname; 1250e9eabf59SPeter Wemm const Elf_Sym *sym; 1251e9eabf59SPeter Wemm int i; 1252757686b1SMarcel Moolenaar Elf_Size symidx; 1253e9eabf59SPeter Wemm Elf_Addr base; 1254e9eabf59SPeter Wemm 1255a4f67738SDoug Rabson 1256fe3db7c7SDoug Rabson /* Perform relocations without addend if there are any: */ 1257b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1258e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 12593cfce8e4SKonstantin Belousov if (rel == NULL) { 12603cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab!"); 12613cfce8e4SKonstantin Belousov return (ENOEXEC); 12623cfce8e4SKonstantin Belousov } 126320947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1264e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 12653cfce8e4SKonstantin Belousov if (base == 0) { 12663cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 12673cfce8e4SKonstantin Belousov return (ENOEXEC); 12683cfce8e4SKonstantin Belousov } 126920947801SPeter Wemm for ( ; rel < rellim; rel++) { 1270e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 127120947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 127220947801SPeter Wemm continue; 1273e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 127420947801SPeter Wemm /* Local relocs are already done */ 127520947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 127620947801SPeter Wemm continue; 127720947801SPeter Wemm if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL, 127820947801SPeter Wemm elf_obj_lookup)) { 1279aa855a59SPeter Wemm symname = symbol_name(ef, rel->r_info); 128020947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 128120947801SPeter Wemm symname); 12823cfce8e4SKonstantin Belousov return (ENOENT); 1283a13ddfb6SPeter Wemm } 1284aa855a59SPeter Wemm } 1285a4f67738SDoug Rabson } 1286a4f67738SDoug Rabson 1287fe3db7c7SDoug Rabson /* Perform relocations with addend if there are any: */ 1288b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1289e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 12903cfce8e4SKonstantin Belousov if (rela == NULL) { 12913cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 12923cfce8e4SKonstantin Belousov return (ENOEXEC); 12933cfce8e4SKonstantin Belousov } 129420947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1295e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 12963cfce8e4SKonstantin Belousov if (base == 0) { 12973cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, 12983cfce8e4SKonstantin Belousov "lost base for relatab"); 12993cfce8e4SKonstantin Belousov return (ENOEXEC); 13003cfce8e4SKonstantin Belousov } 130120947801SPeter Wemm for ( ; rela < relalim; rela++) { 1302e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 130320947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 130420947801SPeter Wemm continue; 1305e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 130620947801SPeter Wemm /* Local relocs are already done */ 130720947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 130820947801SPeter Wemm continue; 130920947801SPeter Wemm if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA, 131020947801SPeter Wemm elf_obj_lookup)) { 1311aa855a59SPeter Wemm symname = symbol_name(ef, rela->r_info); 131220947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 131320947801SPeter Wemm symname); 13143cfce8e4SKonstantin Belousov return (ENOENT); 1315a13ddfb6SPeter Wemm } 1316e9eabf59SPeter Wemm } 1317e9eabf59SPeter Wemm } 1318a4f67738SDoug Rabson 13192832cd54SKonstantin Belousov /* 1320e3043798SPedro F. Giffuni * Only clean SHN_FBSD_CACHED for successful return. If we 13212832cd54SKonstantin Belousov * modified symbol table for the object but found an 13222832cd54SKonstantin Belousov * unresolved symbol, there is no reason to roll back. 13232832cd54SKonstantin Belousov */ 13242832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(ef); 13252832cd54SKonstantin Belousov 13263cfce8e4SKonstantin Belousov return (0); 1327a4f67738SDoug Rabson } 1328a4f67738SDoug Rabson 132937c84183SPoul-Henning Kamp static int 1330d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym) 1331a4f67738SDoug Rabson { 1332326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 13332d636ab0SPeter Wemm const Elf_Sym *symp; 13342d636ab0SPeter Wemm const char *strp; 1335a4f67738SDoug Rabson int i; 1336a4f67738SDoug Rabson 13372d636ab0SPeter Wemm for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 13382d636ab0SPeter Wemm strp = ef->ddbstrtab + symp->st_name; 1339a8774e39SPeter Wemm if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) { 1340d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 13412d636ab0SPeter Wemm return 0; 13422d636ab0SPeter Wemm } 13432d636ab0SPeter Wemm } 1344a4f67738SDoug Rabson return ENOENT; 1345a4f67738SDoug Rabson } 1346a4f67738SDoug Rabson 1347de78ca7eSPeter Wemm static int 134820947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, 134920947801SPeter Wemm linker_symval_t *symval) 1350a4f67738SDoug Rabson { 13517035cf14SKonstantin Belousov elf_file_t ef; 13527035cf14SKonstantin Belousov const Elf_Sym *es; 13537035cf14SKonstantin Belousov caddr_t val; 1354a4f67738SDoug Rabson 13557035cf14SKonstantin Belousov ef = (elf_file_t) lf; 13567035cf14SKonstantin Belousov es = (const Elf_Sym*) sym; 13577035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 13585cf87418SMarcel Moolenaar if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) { 13592d636ab0SPeter Wemm symval->name = ef->ddbstrtab + es->st_name; 13607035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 13617035cf14SKonstantin Belousov if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC) 13627035cf14SKonstantin Belousov val = ((caddr_t (*)(void))val)(); 13637035cf14SKonstantin Belousov symval->value = val; 13642d636ab0SPeter Wemm symval->size = es->st_size; 13652d636ab0SPeter Wemm return 0; 13662d636ab0SPeter Wemm } 13672d636ab0SPeter Wemm return ENOENT; 13682d636ab0SPeter Wemm } 1369a4f67738SDoug Rabson 1370a4f67738SDoug Rabson static int 1371a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value, 1372d254af07SMatthew Dillon c_linker_sym_t *sym, long *diffp) 1373a4f67738SDoug Rabson { 1374326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1375586453feSBruce Evans u_long off = (uintptr_t) (void *) value; 1376a4f67738SDoug Rabson u_long diff = off; 1377586453feSBruce Evans u_long st_value; 1378fe3db7c7SDoug Rabson const Elf_Sym *es; 1379b85f65afSPedro F. Giffuni const Elf_Sym *best = NULL; 1380a4f67738SDoug Rabson int i; 1381a4f67738SDoug Rabson 13822d636ab0SPeter Wemm for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 1383a4f67738SDoug Rabson if (es->st_name == 0) 1384a4f67738SDoug Rabson continue; 1385a8774e39SPeter Wemm st_value = es->st_value; 1386b5abfb70SPeter Wemm if (off >= st_value) { 1387b5abfb70SPeter Wemm if (off - st_value < diff) { 1388b5abfb70SPeter Wemm diff = off - st_value; 1389a4f67738SDoug Rabson best = es; 1390a4f67738SDoug Rabson if (diff == 0) 1391a4f67738SDoug Rabson break; 1392b5abfb70SPeter Wemm } else if (off - st_value == diff) { 1393a4f67738SDoug Rabson best = es; 1394a4f67738SDoug Rabson } 1395a4f67738SDoug Rabson } 1396a4f67738SDoug Rabson } 1397b85f65afSPedro F. Giffuni if (best == NULL) 1398a4f67738SDoug Rabson *diffp = off; 1399a4f67738SDoug Rabson else 1400a4f67738SDoug Rabson *diffp = diff; 1401d254af07SMatthew Dillon *sym = (c_linker_sym_t) best; 1402a4f67738SDoug Rabson 1403a4f67738SDoug Rabson return 0; 1404a4f67738SDoug Rabson } 1405f41325dbSPeter Wemm 1406f41325dbSPeter Wemm /* 1407f41325dbSPeter Wemm * Look up a linker set on an ELF system. 1408f41325dbSPeter Wemm */ 1409f41325dbSPeter Wemm static int 1410f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name, 1411f41325dbSPeter Wemm void ***startp, void ***stopp, int *countp) 1412f41325dbSPeter Wemm { 1413e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 1414f41325dbSPeter Wemm void **start, **stop; 1415e9eabf59SPeter Wemm int i, count; 1416f41325dbSPeter Wemm 1417e9eabf59SPeter Wemm /* Relative to section number */ 1418e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1419e9eabf59SPeter Wemm if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) && 1420e9eabf59SPeter Wemm strcmp(ef->progtab[i].name + 4, name) == 0) { 1421e9eabf59SPeter Wemm start = (void **)ef->progtab[i].addr; 142220947801SPeter Wemm stop = (void **)((char *)ef->progtab[i].addr + 142320947801SPeter Wemm ef->progtab[i].size); 1424f41325dbSPeter Wemm count = stop - start; 1425f41325dbSPeter Wemm if (startp) 1426f41325dbSPeter Wemm *startp = start; 1427f41325dbSPeter Wemm if (stopp) 1428f41325dbSPeter Wemm *stopp = stop; 1429f41325dbSPeter Wemm if (countp) 1430f41325dbSPeter Wemm *countp = count; 1431e9eabf59SPeter Wemm return (0); 1432e9eabf59SPeter Wemm } 1433e9eabf59SPeter Wemm } 1434e9eabf59SPeter Wemm return (ESRCH); 1435f41325dbSPeter Wemm } 1436bb9fe9ddSBrian Feldman 1437bb9fe9ddSBrian Feldman static int 1438bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file, 1439e9eabf59SPeter Wemm int (*callback)(const char *, void *), void *opaque) 1440e9eabf59SPeter Wemm { 1441bb9fe9ddSBrian Feldman elf_file_t ef = (elf_file_t)file; 1442bb9fe9ddSBrian Feldman const Elf_Sym *symp; 1443bb9fe9ddSBrian Feldman int i, error; 1444bb9fe9ddSBrian Feldman 1445bb9fe9ddSBrian Feldman /* Exhaustive search */ 1446bb9fe9ddSBrian Feldman for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1447bb9fe9ddSBrian Feldman if (symp->st_value != 0 && 14487035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 14497035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 1450bb9fe9ddSBrian Feldman error = callback(ef->ddbstrtab + symp->st_name, opaque); 1451bb9fe9ddSBrian Feldman if (error) 1452bb9fe9ddSBrian Feldman return (error); 1453bb9fe9ddSBrian Feldman } 1454bb9fe9ddSBrian Feldman } 1455bb9fe9ddSBrian Feldman return (0); 1456bb9fe9ddSBrian Feldman } 145784201059SMarcel Moolenaar 1458a2024a3eSJohn Birrell static int 1459a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file, 1460a2024a3eSJohn Birrell linker_function_nameval_callback_t callback, void *opaque) 1461a2024a3eSJohn Birrell { 1462a2024a3eSJohn Birrell linker_symval_t symval; 1463a2024a3eSJohn Birrell elf_file_t ef = (elf_file_t)file; 1464a2024a3eSJohn Birrell const Elf_Sym* symp; 1465a2024a3eSJohn Birrell int i, error; 1466a2024a3eSJohn Birrell 1467a2024a3eSJohn Birrell /* Exhaustive search */ 1468a2024a3eSJohn Birrell for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1469a2024a3eSJohn Birrell if (symp->st_value != 0 && 14707035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 14717035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 14727035cf14SKonstantin Belousov error = link_elf_symbol_values(file, 14737035cf14SKonstantin Belousov (c_linker_sym_t)symp, &symval); 1474a2024a3eSJohn Birrell if (error) 1475a2024a3eSJohn Birrell return (error); 1476a2024a3eSJohn Birrell error = callback(file, i, &symval, opaque); 1477a2024a3eSJohn Birrell if (error) 1478a2024a3eSJohn Birrell return (error); 1479a2024a3eSJohn Birrell } 1480a2024a3eSJohn Birrell } 1481a2024a3eSJohn Birrell return (0); 1482a2024a3eSJohn Birrell } 1483a2024a3eSJohn Birrell 14842832cd54SKonstantin Belousov static void 14852832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef) 14862832cd54SKonstantin Belousov { 14872832cd54SKonstantin Belousov Elf_Sym *sym; 14882832cd54SKonstantin Belousov Elf_Size i; 14892832cd54SKonstantin Belousov 14902832cd54SKonstantin Belousov for (i = 0; i < ef->ddbsymcnt; i++) { 14912832cd54SKonstantin Belousov sym = ef->ddbsymtab + i; 14922832cd54SKonstantin Belousov if (sym->st_shndx == SHN_FBSD_CACHED) { 14932832cd54SKonstantin Belousov sym->st_shndx = SHN_UNDEF; 14942832cd54SKonstantin Belousov sym->st_value = 0; 14952832cd54SKonstantin Belousov } 14962832cd54SKonstantin Belousov } 14972832cd54SKonstantin Belousov } 14982832cd54SKonstantin Belousov 1499d297ad16SMarcel Moolenaar /* 1500d297ad16SMarcel Moolenaar * Symbol lookup function that can be used when the symbol index is known (ie 1501d297ad16SMarcel Moolenaar * in relocations). It uses the symbol index instead of doing a fully fledged 1502d297ad16SMarcel Moolenaar * hash table based lookup when such is valid. For example for local symbols. 1503d297ad16SMarcel Moolenaar * This is not only more efficient, it's also more correct. It's not always 1504d297ad16SMarcel Moolenaar * the case that the symbol can be found through the hash table. 1505d297ad16SMarcel Moolenaar */ 1506cff8c6f2SKonstantin Belousov static int 1507cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res) 1508d297ad16SMarcel Moolenaar { 1509d297ad16SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 15102832cd54SKonstantin Belousov Elf_Sym *sym; 1511d297ad16SMarcel Moolenaar const char *symbol; 1512cff8c6f2SKonstantin Belousov Elf_Addr res1; 1513d297ad16SMarcel Moolenaar 1514d297ad16SMarcel Moolenaar /* Don't even try to lookup the symbol if the index is bogus. */ 1515cff8c6f2SKonstantin Belousov if (symidx >= ef->ddbsymcnt) { 1516cff8c6f2SKonstantin Belousov *res = 0; 1517cff8c6f2SKonstantin Belousov return (EINVAL); 1518cff8c6f2SKonstantin Belousov } 1519d297ad16SMarcel Moolenaar 1520e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 1521d297ad16SMarcel Moolenaar 15224cec6f5dSPeter Wemm /* Quick answer if there is a definition included. */ 1523cff8c6f2SKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) { 1524289ead7cSKonstantin Belousov res1 = (Elf_Addr)sym->st_value; 1525289ead7cSKonstantin Belousov if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) 1526289ead7cSKonstantin Belousov res1 = ((Elf_Addr (*)(void))res1)(); 1527289ead7cSKonstantin Belousov *res = res1; 1528cff8c6f2SKonstantin Belousov return (0); 1529cff8c6f2SKonstantin Belousov } 15304cec6f5dSPeter Wemm 15314cec6f5dSPeter Wemm /* If we get here, then it is undefined and needs a lookup. */ 15324cec6f5dSPeter Wemm switch (ELF_ST_BIND(sym->st_info)) { 15334cec6f5dSPeter Wemm case STB_LOCAL: 15344cec6f5dSPeter Wemm /* Local, but undefined? huh? */ 1535cff8c6f2SKonstantin Belousov *res = 0; 1536cff8c6f2SKonstantin Belousov return (EINVAL); 1537d297ad16SMarcel Moolenaar 1538e9eabf59SPeter Wemm case STB_GLOBAL: 1539cff8c6f2SKonstantin Belousov case STB_WEAK: 1540e9eabf59SPeter Wemm /* Relative to Data or Function name */ 1541e9eabf59SPeter Wemm symbol = ef->ddbstrtab + sym->st_name; 1542d297ad16SMarcel Moolenaar 1543d297ad16SMarcel Moolenaar /* Force a lookup failure if the symbol name is bogus. */ 1544cff8c6f2SKonstantin Belousov if (*symbol == 0) { 1545cff8c6f2SKonstantin Belousov *res = 0; 1546cff8c6f2SKonstantin Belousov return (EINVAL); 1547cff8c6f2SKonstantin Belousov } 1548cff8c6f2SKonstantin Belousov res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps); 15492832cd54SKonstantin Belousov 15502832cd54SKonstantin Belousov /* 15512832cd54SKonstantin Belousov * Cache global lookups during module relocation. The failure 15522832cd54SKonstantin Belousov * case is particularly expensive for callers, who must scan 15532832cd54SKonstantin Belousov * through the entire globals table doing strcmp(). Cache to 15542832cd54SKonstantin Belousov * avoid doing such work repeatedly. 15552832cd54SKonstantin Belousov * 15562832cd54SKonstantin Belousov * After relocation is complete, undefined globals will be 15572832cd54SKonstantin Belousov * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(), 15582832cd54SKonstantin Belousov * above. 15592832cd54SKonstantin Belousov */ 1560cff8c6f2SKonstantin Belousov if (res1 != 0) { 15612832cd54SKonstantin Belousov sym->st_shndx = SHN_FBSD_CACHED; 1562cff8c6f2SKonstantin Belousov sym->st_value = res1; 1563cff8c6f2SKonstantin Belousov *res = res1; 1564e9eabf59SPeter Wemm return (0); 1565cff8c6f2SKonstantin Belousov } else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) { 1566cff8c6f2SKonstantin Belousov sym->st_value = 0; 1567cff8c6f2SKonstantin Belousov *res = 0; 1568cff8c6f2SKonstantin Belousov return (0); 1569cff8c6f2SKonstantin Belousov } 1570cff8c6f2SKonstantin Belousov return (EINVAL); 1571e9eabf59SPeter Wemm 1572e9eabf59SPeter Wemm default: 1573cff8c6f2SKonstantin Belousov return (EINVAL); 1574d297ad16SMarcel Moolenaar } 1575e9eabf59SPeter Wemm } 1576e9eabf59SPeter Wemm 15777251b4bfSJake Burkholder static void 15787226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef) 15797226306eSKonstantin Belousov { 15807226306eSKonstantin Belousov static const char startn[] = "__start_"; 15817226306eSKonstantin Belousov static const char stopn[] = "__stop_"; 15827226306eSKonstantin Belousov Elf_Sym *sym; 15837226306eSKonstantin Belousov const char *sym_name, *linkset_name; 15847226306eSKonstantin Belousov Elf_Addr startp, stopp; 15857226306eSKonstantin Belousov Elf_Size symidx; 15867226306eSKonstantin Belousov int start, i; 15877226306eSKonstantin Belousov 15887226306eSKonstantin Belousov startp = stopp = 0; 15897226306eSKonstantin Belousov for (symidx = 1 /* zero entry is special */; 15907226306eSKonstantin Belousov symidx < ef->ddbsymcnt; symidx++) { 15917226306eSKonstantin Belousov sym = ef->ddbsymtab + symidx; 15927226306eSKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) 15937226306eSKonstantin Belousov continue; 15947226306eSKonstantin Belousov 15957226306eSKonstantin Belousov sym_name = ef->ddbstrtab + sym->st_name; 15967226306eSKonstantin Belousov if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) { 15977226306eSKonstantin Belousov start = 1; 15987226306eSKonstantin Belousov linkset_name = sym_name + sizeof(startn) - 1; 15997226306eSKonstantin Belousov } 16007226306eSKonstantin Belousov else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) { 16017226306eSKonstantin Belousov start = 0; 16027226306eSKonstantin Belousov linkset_name = sym_name + sizeof(stopn) - 1; 16037226306eSKonstantin Belousov } 16047226306eSKonstantin Belousov else 16057226306eSKonstantin Belousov continue; 16067226306eSKonstantin Belousov 16077226306eSKonstantin Belousov for (i = 0; i < ef->nprogtab; i++) { 16087226306eSKonstantin Belousov if (strcmp(ef->progtab[i].name, linkset_name) == 0) { 16097226306eSKonstantin Belousov startp = (Elf_Addr)ef->progtab[i].addr; 16107226306eSKonstantin Belousov stopp = (Elf_Addr)(startp + ef->progtab[i].size); 16117226306eSKonstantin Belousov break; 16127226306eSKonstantin Belousov } 16137226306eSKonstantin Belousov } 16147226306eSKonstantin Belousov if (i == ef->nprogtab) 16157226306eSKonstantin Belousov continue; 16167226306eSKonstantin Belousov 16177226306eSKonstantin Belousov sym->st_value = start ? startp : stopp; 16187226306eSKonstantin Belousov sym->st_shndx = i; 16197226306eSKonstantin Belousov } 16207226306eSKonstantin Belousov } 16217226306eSKonstantin Belousov 16223cfce8e4SKonstantin Belousov static int 1623289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs) 16247251b4bfSJake Burkholder { 1625e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 16267251b4bfSJake Burkholder const Elf_Rel *rellim; 16277251b4bfSJake Burkholder const Elf_Rel *rel; 16287251b4bfSJake Burkholder const Elf_Rela *relalim; 16297251b4bfSJake Burkholder const Elf_Rela *rela; 1630e9eabf59SPeter Wemm const Elf_Sym *sym; 1631e9eabf59SPeter Wemm Elf_Addr base; 1632e9eabf59SPeter Wemm int i; 1633757686b1SMarcel Moolenaar Elf_Size symidx; 1634e9eabf59SPeter Wemm 16357226306eSKonstantin Belousov link_elf_fix_link_set(ef); 16367226306eSKonstantin Belousov 16377251b4bfSJake Burkholder /* Perform relocations without addend if there are any: */ 1638b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1639e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 16403cfce8e4SKonstantin Belousov if (rel == NULL) { 16413cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab"); 16423cfce8e4SKonstantin Belousov return (ENOEXEC); 16433cfce8e4SKonstantin Belousov } 164420947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1645e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 16463cfce8e4SKonstantin Belousov if (base == 0) { 16473cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 16483cfce8e4SKonstantin Belousov return (ENOEXEC); 16493cfce8e4SKonstantin Belousov } 165020947801SPeter Wemm for ( ; rel < rellim; rel++) { 1651e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 165220947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 165320947801SPeter Wemm continue; 1654e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 165520947801SPeter Wemm /* Only do local relocs */ 165620947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 165720947801SPeter Wemm continue; 165836716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 165936716fe2SMark Johnston elf_is_ifunc_reloc(rel->r_info)) == ifuncs) 166020947801SPeter Wemm elf_reloc_local(lf, base, rel, ELF_RELOC_REL, 166120947801SPeter Wemm elf_obj_lookup); 16627251b4bfSJake Burkholder } 1663e9eabf59SPeter Wemm } 16647251b4bfSJake Burkholder 16657251b4bfSJake Burkholder /* Perform relocations with addend if there are any: */ 1666b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1667e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 16683cfce8e4SKonstantin Belousov if (rela == NULL) { 16693cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 16703cfce8e4SKonstantin Belousov return (ENOEXEC); 16713cfce8e4SKonstantin Belousov } 167220947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1673e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 16743cfce8e4SKonstantin Belousov if (base == 0) { 16753cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 16763cfce8e4SKonstantin Belousov return (ENOEXEC); 16773cfce8e4SKonstantin Belousov } 167820947801SPeter Wemm for ( ; rela < relalim; rela++) { 1679e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 168020947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 168120947801SPeter Wemm continue; 1682e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 168320947801SPeter Wemm /* Only do local relocs */ 168420947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 168520947801SPeter Wemm continue; 168636716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 168736716fe2SMark Johnston elf_is_ifunc_reloc(rela->r_info)) == ifuncs) 168820947801SPeter Wemm elf_reloc_local(lf, base, rela, ELF_RELOC_RELA, 168920947801SPeter Wemm elf_obj_lookup); 16907251b4bfSJake Burkholder } 16917251b4bfSJake Burkholder } 16923cfce8e4SKonstantin Belousov return (0); 1693e9eabf59SPeter Wemm } 169400a5db46SStacey Son 169500a5db46SStacey Son static long 1696e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab) 169700a5db46SStacey Son { 169800a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 169900a5db46SStacey Son 170000a5db46SStacey Son *symtab = ef->ddbsymtab; 170100a5db46SStacey Son 170200a5db46SStacey Son if (*symtab == NULL) 170300a5db46SStacey Son return (0); 170400a5db46SStacey Son 170500a5db46SStacey Son return (ef->ddbsymcnt); 170600a5db46SStacey Son } 170700a5db46SStacey Son 170800a5db46SStacey Son static long 170900a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab) 171000a5db46SStacey Son { 171100a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 171200a5db46SStacey Son 171300a5db46SStacey Son *strtab = ef->ddbstrtab; 171400a5db46SStacey Son 171500a5db46SStacey Son if (*strtab == NULL) 171600a5db46SStacey Son return (0); 171700a5db46SStacey Son 171800a5db46SStacey Son return (ef->ddbstrcnt); 171900a5db46SStacey Son } 1720