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> 37fbf2a778SKonstantin Belousov #include <sys/fcntl.h> 38fb919e4dSMark Murray #include <sys/kernel.h> 39fb919e4dSMark Murray #include <sys/lock.h> 40a4f67738SDoug Rabson #include <sys/malloc.h> 41fbf2a778SKonstantin Belousov #include <sys/linker.h> 42c0b824f9SDima Dorfman #include <sys/mutex.h> 4362d615d5SJohn Baldwin #include <sys/mount.h> 44a4f67738SDoug Rabson #include <sys/namei.h> 45fbf2a778SKonstantin Belousov #include <sys/proc.h> 46fbf2a778SKonstantin Belousov #include <sys/rwlock.h> 47a4f67738SDoug Rabson #include <sys/vnode.h> 48fb919e4dSMark Murray 49a4f67738SDoug Rabson #include <machine/elf.h> 50a4f67738SDoug Rabson 51eddfbb76SRobert Watson #include <net/vnet.h> 52eddfbb76SRobert Watson 53aed55708SRobert Watson #include <security/mac/mac_framework.h> 54aed55708SRobert Watson 55fe3db7c7SDoug Rabson #include <vm/vm.h> 56fe3db7c7SDoug Rabson #include <vm/vm_param.h> 57fe3db7c7SDoug Rabson #include <vm/pmap.h> 58fbf2a778SKonstantin Belousov #include <vm/vm_extern.h> 59fbf2a778SKonstantin Belousov #include <vm/vm_kern.h> 60fe3db7c7SDoug Rabson #include <vm/vm_map.h> 61fbf2a778SKonstantin Belousov #include <vm/vm_object.h> 62fbf2a778SKonstantin Belousov #include <vm/vm_page.h> 63fbf2a778SKonstantin Belousov #include <vm/vm_pager.h> 64fb919e4dSMark Murray 65f9980387SPeter Wemm #include <sys/link_elf.h> 66fe3db7c7SDoug Rabson 67a2024a3eSJohn Birrell #ifdef DDB_CTF 6822bbc4b2SXin LI #include <contrib/zlib/zlib.h> 69a2024a3eSJohn Birrell #endif 70a2024a3eSJohn Birrell 71326e27d8SDoug Rabson #include "linker_if.h" 72a4f67738SDoug Rabson 73e9eabf59SPeter Wemm typedef struct { 74e9eabf59SPeter Wemm void *addr; 7520947801SPeter Wemm Elf_Off size; 761d9eae9fSMark Johnston int flags; /* Section flags. */ 771d9eae9fSMark Johnston int sec; /* Original section number. */ 78e9eabf59SPeter Wemm char *name; 79e9eabf59SPeter Wemm } Elf_progent; 80e9eabf59SPeter Wemm 81e9eabf59SPeter Wemm typedef struct { 82e9eabf59SPeter Wemm Elf_Rel *rel; 8320947801SPeter Wemm int nrel; 84e9eabf59SPeter Wemm int sec; 85e9eabf59SPeter Wemm } Elf_relent; 86e9eabf59SPeter Wemm 87e9eabf59SPeter Wemm typedef struct { 88e9eabf59SPeter Wemm Elf_Rela *rela; 8920947801SPeter Wemm int nrela; 90e9eabf59SPeter Wemm int sec; 91e9eabf59SPeter Wemm } Elf_relaent; 92e9eabf59SPeter Wemm 93a4f67738SDoug Rabson typedef struct elf_file { 94326e27d8SDoug Rabson struct linker_file lf; /* Common fields */ 9520947801SPeter Wemm 9670b7ffeeSIan Dowse int preloaded; 97fe3db7c7SDoug Rabson caddr_t address; /* Relocation address */ 98fe3db7c7SDoug Rabson vm_object_t object; /* VM object to hold file pages */ 99e9eabf59SPeter Wemm Elf_Shdr *e_shdr; 100e9eabf59SPeter Wemm 101e9eabf59SPeter Wemm Elf_progent *progtab; 102d821d364SPedro F. Giffuni u_int nprogtab; 103e9eabf59SPeter Wemm 104e9eabf59SPeter Wemm Elf_relaent *relatab; 105d821d364SPedro F. Giffuni u_int nrelatab; 106e9eabf59SPeter Wemm 107e9eabf59SPeter Wemm Elf_relent *reltab; 108b474c780SDavid E. O'Brien int nreltab; 109e9eabf59SPeter Wemm 110e9eabf59SPeter Wemm Elf_Sym *ddbsymtab; /* The symbol table we are using */ 1112d636ab0SPeter Wemm long ddbsymcnt; /* Number of symbols */ 1122d636ab0SPeter Wemm caddr_t ddbstrtab; /* String table */ 1132d636ab0SPeter Wemm long ddbstrcnt; /* number of bytes in string table */ 114e9eabf59SPeter Wemm 115e9eabf59SPeter Wemm caddr_t shstrtab; /* Section name string table */ 116e9eabf59SPeter Wemm long shstrcnt; /* number of bytes in string table */ 117e9eabf59SPeter Wemm 118a2024a3eSJohn Birrell caddr_t ctftab; /* CTF table */ 119a2024a3eSJohn Birrell long ctfcnt; /* number of bytes in CTF table */ 120a2024a3eSJohn Birrell caddr_t ctfoff; /* CTF offset table */ 121a2024a3eSJohn Birrell caddr_t typoff; /* Type offset table */ 122a2024a3eSJohn Birrell long typlen; /* Number of type entries. */ 123a2024a3eSJohn Birrell 124a4f67738SDoug Rabson } *elf_file_t; 125a4f67738SDoug Rabson 126a2024a3eSJohn Birrell #include <kern/kern_ctf.c> 127a2024a3eSJohn Birrell 12854823af2SPeter Wemm static int link_elf_link_preload(linker_class_t cls, 129326e27d8SDoug Rabson const char *, linker_file_t *); 13054823af2SPeter Wemm static int link_elf_link_preload_finish(linker_file_t); 13154823af2SPeter Wemm static int link_elf_load_file(linker_class_t, const char *, linker_file_t *); 132326e27d8SDoug Rabson static int link_elf_lookup_symbol(linker_file_t, const char *, 133326e27d8SDoug Rabson c_linker_sym_t *); 134e9eabf59SPeter Wemm static int link_elf_symbol_values(linker_file_t, c_linker_sym_t, 135e9eabf59SPeter Wemm linker_symval_t *); 136326e27d8SDoug Rabson static int link_elf_search_symbol(linker_file_t, caddr_t value, 137326e27d8SDoug Rabson c_linker_sym_t *sym, long *diffp); 138326e27d8SDoug Rabson 139326e27d8SDoug Rabson static void link_elf_unload_file(linker_file_t); 140f41325dbSPeter Wemm static int link_elf_lookup_set(linker_file_t, const char *, 141f41325dbSPeter Wemm void ***, void ***, int *); 142bb9fe9ddSBrian Feldman static int link_elf_each_function_name(linker_file_t, 143e9eabf59SPeter Wemm int (*)(const char *, void *), void *); 144a2024a3eSJohn Birrell static int link_elf_each_function_nameval(linker_file_t, 145a2024a3eSJohn Birrell linker_function_nameval_callback_t, 146a2024a3eSJohn Birrell void *); 147289ead7cSKonstantin Belousov static int link_elf_reloc_local(linker_file_t, bool); 148e76f11f4SAndriy Gapon static long link_elf_symtab_get(linker_file_t, const Elf_Sym **); 14900a5db46SStacey Son static long link_elf_strtab_get(linker_file_t, caddr_t *); 150326e27d8SDoug Rabson 151cff8c6f2SKonstantin Belousov static int elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, 152cff8c6f2SKonstantin Belousov Elf_Addr *); 153e9eabf59SPeter Wemm 154326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = { 155326e27d8SDoug Rabson KOBJMETHOD(linker_lookup_symbol, link_elf_lookup_symbol), 156326e27d8SDoug Rabson KOBJMETHOD(linker_symbol_values, link_elf_symbol_values), 157326e27d8SDoug Rabson KOBJMETHOD(linker_search_symbol, link_elf_search_symbol), 158326e27d8SDoug Rabson KOBJMETHOD(linker_unload, link_elf_unload_file), 15954823af2SPeter Wemm KOBJMETHOD(linker_load_file, link_elf_load_file), 16054823af2SPeter Wemm KOBJMETHOD(linker_link_preload, link_elf_link_preload), 16154823af2SPeter Wemm KOBJMETHOD(linker_link_preload_finish, link_elf_link_preload_finish), 162f41325dbSPeter Wemm KOBJMETHOD(linker_lookup_set, link_elf_lookup_set), 163bb9fe9ddSBrian Feldman KOBJMETHOD(linker_each_function_name, link_elf_each_function_name), 164a2024a3eSJohn Birrell KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval), 165a2024a3eSJohn Birrell KOBJMETHOD(linker_ctf_get, link_elf_ctf_get), 16600a5db46SStacey Son KOBJMETHOD(linker_symtab_get, link_elf_symtab_get), 16700a5db46SStacey Son KOBJMETHOD(linker_strtab_get, link_elf_strtab_get), 1686d775f0bSMark Johnston KOBJMETHOD_END 169326e27d8SDoug Rabson }; 170326e27d8SDoug Rabson 171326e27d8SDoug Rabson static struct linker_class link_elf_class = { 172326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32 173e9eabf59SPeter Wemm "elf32_obj", 174326e27d8SDoug Rabson #else 175e9eabf59SPeter Wemm "elf64_obj", 176326e27d8SDoug Rabson #endif 177326e27d8SDoug Rabson link_elf_methods, sizeof(struct elf_file) 178326e27d8SDoug Rabson }; 179326e27d8SDoug Rabson 180326e27d8SDoug Rabson static int relocate_file(elf_file_t ef); 1812832cd54SKonstantin Belousov static void elf_obj_cleanup_globals_cache(elf_file_t); 182a4f67738SDoug Rabson 183a4f67738SDoug Rabson static void 184552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s) 1851720979bSMarcel Moolenaar { 186552f9f63SEdwin Groothuis if (filename == NULL) 1871720979bSMarcel Moolenaar printf("kldload: %s\n", s); 188552f9f63SEdwin Groothuis else 189552f9f63SEdwin Groothuis printf("kldload: %s: %s\n", filename, s); 1901720979bSMarcel Moolenaar } 1911720979bSMarcel Moolenaar 1921720979bSMarcel Moolenaar static void 193a4f67738SDoug Rabson link_elf_init(void *arg) 194a4f67738SDoug Rabson { 195a4f67738SDoug Rabson 196326e27d8SDoug Rabson linker_add_class(&link_elf_class); 197a4f67738SDoug Rabson } 198891cf3edSEd Maste SYSINIT(link_elf_obj, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, NULL); 199a4f67738SDoug Rabson 2001d9eae9fSMark Johnston static void 2011d9eae9fSMark Johnston link_elf_protect_range(elf_file_t ef, vm_offset_t start, vm_offset_t end, 2021d9eae9fSMark Johnston vm_prot_t prot) 2031d9eae9fSMark Johnston { 2041d9eae9fSMark Johnston int error __unused; 2051d9eae9fSMark Johnston 2061d9eae9fSMark Johnston KASSERT(start <= end && start >= (vm_offset_t)ef->address && 2071d9eae9fSMark Johnston end <= round_page((vm_offset_t)ef->address + ef->lf.size), 2081d9eae9fSMark Johnston ("link_elf_protect_range: invalid range %#jx-%#jx", 2091d9eae9fSMark Johnston (uintmax_t)start, (uintmax_t)end)); 2101d9eae9fSMark Johnston 2111d9eae9fSMark Johnston if (start == end) 2121d9eae9fSMark Johnston return; 213f822c9e2SMark Johnston if (ef->preloaded) { 214f822c9e2SMark Johnston #ifdef __amd64__ 215f822c9e2SMark Johnston error = pmap_change_prot(start, end - start, prot); 216f822c9e2SMark Johnston KASSERT(error == 0, 217f822c9e2SMark Johnston ("link_elf_protect_range: pmap_change_prot() returned %d", 218f822c9e2SMark Johnston error)); 219f822c9e2SMark Johnston #endif 220f822c9e2SMark Johnston return; 221f822c9e2SMark Johnston } 2221d9eae9fSMark Johnston error = vm_map_protect(kernel_map, start, end, prot, FALSE); 2231d9eae9fSMark Johnston KASSERT(error == KERN_SUCCESS, 2241d9eae9fSMark Johnston ("link_elf_protect_range: vm_map_protect() returned %d", error)); 2251d9eae9fSMark Johnston } 2261d9eae9fSMark Johnston 2271d9eae9fSMark Johnston /* 2281d9eae9fSMark Johnston * Restrict permissions on linker file memory based on section flags. 2291d9eae9fSMark Johnston * Sections need not be page-aligned, so overlap within a page is possible. 2301d9eae9fSMark Johnston */ 2311d9eae9fSMark Johnston static void 2321d9eae9fSMark Johnston link_elf_protect(elf_file_t ef) 2331d9eae9fSMark Johnston { 2341d9eae9fSMark Johnston vm_offset_t end, segend, segstart, start; 2351d9eae9fSMark Johnston vm_prot_t gapprot, prot, segprot; 2361d9eae9fSMark Johnston int i; 2371d9eae9fSMark Johnston 2381d9eae9fSMark Johnston /* 2391d9eae9fSMark Johnston * If the file was preloaded, the last page may contain other preloaded 2401d9eae9fSMark Johnston * data which may need to be writeable. ELF files are always 2411d9eae9fSMark Johnston * page-aligned, but other preloaded data, such as entropy or CPU 2421d9eae9fSMark Johnston * microcode may be loaded with a smaller alignment. 2431d9eae9fSMark Johnston */ 2441d9eae9fSMark Johnston gapprot = ef->preloaded ? VM_PROT_RW : VM_PROT_READ; 2451d9eae9fSMark Johnston 2461d9eae9fSMark Johnston start = end = (vm_offset_t)ef->address; 2471d9eae9fSMark Johnston prot = VM_PROT_READ; 2481d9eae9fSMark Johnston for (i = 0; i < ef->nprogtab; i++) { 2491d9eae9fSMark Johnston /* 2501d9eae9fSMark Johnston * VNET and DPCPU sections have their memory allocated by their 2511d9eae9fSMark Johnston * respective subsystems. 2521d9eae9fSMark Johnston */ 2531d9eae9fSMark Johnston if (ef->progtab[i].name != NULL && ( 2541d9eae9fSMark Johnston #ifdef VIMAGE 2551d9eae9fSMark Johnston strcmp(ef->progtab[i].name, VNET_SETNAME) == 0 || 2561d9eae9fSMark Johnston #endif 2571d9eae9fSMark Johnston strcmp(ef->progtab[i].name, DPCPU_SETNAME) == 0)) 2581d9eae9fSMark Johnston continue; 2591d9eae9fSMark Johnston 2601d9eae9fSMark Johnston segstart = trunc_page((vm_offset_t)ef->progtab[i].addr); 2611d9eae9fSMark Johnston segend = round_page((vm_offset_t)ef->progtab[i].addr + 2621d9eae9fSMark Johnston ef->progtab[i].size); 2631d9eae9fSMark Johnston segprot = VM_PROT_READ; 2641d9eae9fSMark Johnston if ((ef->progtab[i].flags & SHF_WRITE) != 0) 2651d9eae9fSMark Johnston segprot |= VM_PROT_WRITE; 2661d9eae9fSMark Johnston if ((ef->progtab[i].flags & SHF_EXECINSTR) != 0) 2671d9eae9fSMark Johnston segprot |= VM_PROT_EXECUTE; 2681d9eae9fSMark Johnston 2691d9eae9fSMark Johnston if (end <= segstart) { 2701d9eae9fSMark Johnston /* 2711d9eae9fSMark Johnston * Case 1: there is no overlap between the previous 2721d9eae9fSMark Johnston * segment and this one. Apply protections to the 2731d9eae9fSMark Johnston * previous segment, and protect the gap between the 2741d9eae9fSMark Johnston * previous and current segments, if any. 2751d9eae9fSMark Johnston */ 2761d9eae9fSMark Johnston link_elf_protect_range(ef, start, end, prot); 2771d9eae9fSMark Johnston link_elf_protect_range(ef, end, segstart, gapprot); 2781d9eae9fSMark Johnston 2791d9eae9fSMark Johnston start = segstart; 2801d9eae9fSMark Johnston end = segend; 2811d9eae9fSMark Johnston prot = segprot; 2821d9eae9fSMark Johnston } else if (start < segstart && end == segend) { 2831d9eae9fSMark Johnston /* 2841d9eae9fSMark Johnston * Case 2: the current segment is a subrange of the 2851d9eae9fSMark Johnston * previous segment. Apply protections to the 2861d9eae9fSMark Johnston * non-overlapping portion of the previous segment. 2871d9eae9fSMark Johnston */ 2881d9eae9fSMark Johnston link_elf_protect_range(ef, start, segstart, prot); 2891d9eae9fSMark Johnston 2901d9eae9fSMark Johnston start = segstart; 2911d9eae9fSMark Johnston prot |= segprot; 2921d9eae9fSMark Johnston } else if (end < segend) { 2931d9eae9fSMark Johnston /* 2941d9eae9fSMark Johnston * Case 3: there is partial overlap between the previous 2951d9eae9fSMark Johnston * and current segments. Apply protections to the 2961d9eae9fSMark Johnston * non-overlapping portion of the previous segment, and 2971d9eae9fSMark Johnston * then the overlap, which must use the union of the two 2981d9eae9fSMark Johnston * segments' protections. 2991d9eae9fSMark Johnston */ 3001d9eae9fSMark Johnston link_elf_protect_range(ef, start, segstart, prot); 3011d9eae9fSMark Johnston link_elf_protect_range(ef, segstart, end, 3021d9eae9fSMark Johnston prot | segprot); 3031d9eae9fSMark Johnston start = end; 3041d9eae9fSMark Johnston end = segend; 3051d9eae9fSMark Johnston prot = segprot; 3061d9eae9fSMark Johnston } else { 3071d9eae9fSMark Johnston /* 3081d9eae9fSMark Johnston * Case 4: the two segments reside in the same page. 3091d9eae9fSMark Johnston */ 3101d9eae9fSMark Johnston prot |= segprot; 3111d9eae9fSMark Johnston } 3121d9eae9fSMark Johnston } 3131d9eae9fSMark Johnston 3141d9eae9fSMark Johnston /* 3151d9eae9fSMark Johnston * Fix up the last unprotected segment and trailing data. 3161d9eae9fSMark Johnston */ 3171d9eae9fSMark Johnston link_elf_protect_range(ef, start, end, prot); 3181d9eae9fSMark Johnston link_elf_protect_range(ef, end, 3191d9eae9fSMark Johnston round_page((vm_offset_t)ef->address + ef->lf.size), gapprot); 3201d9eae9fSMark Johnston } 3211d9eae9fSMark Johnston 322a4f67738SDoug Rabson static int 323e9eabf59SPeter Wemm link_elf_link_preload(linker_class_t cls, const char *filename, 324e9eabf59SPeter Wemm linker_file_t *result) 3252d636ab0SPeter Wemm { 32670b7ffeeSIan Dowse Elf_Ehdr *hdr; 32770b7ffeeSIan Dowse Elf_Shdr *shdr; 32870b7ffeeSIan Dowse Elf_Sym *es; 32970b7ffeeSIan Dowse void *modptr, *baseptr, *sizeptr; 33070b7ffeeSIan Dowse char *type; 33170b7ffeeSIan Dowse elf_file_t ef; 33270b7ffeeSIan Dowse linker_file_t lf; 33370b7ffeeSIan Dowse Elf_Addr off; 33470b7ffeeSIan Dowse int error, i, j, pb, ra, rl, shstrindex, symstrindex, symtabindex; 33570b7ffeeSIan Dowse 33670b7ffeeSIan Dowse /* Look to see if we have the file preloaded */ 33770b7ffeeSIan Dowse modptr = preload_search_by_name(filename); 33870b7ffeeSIan Dowse if (modptr == NULL) 33970b7ffeeSIan Dowse return ENOENT; 34070b7ffeeSIan Dowse 34170b7ffeeSIan Dowse type = (char *)preload_search_info(modptr, MODINFO_TYPE); 34270b7ffeeSIan Dowse baseptr = preload_search_info(modptr, MODINFO_ADDR); 34370b7ffeeSIan Dowse sizeptr = preload_search_info(modptr, MODINFO_SIZE); 34470b7ffeeSIan Dowse hdr = (Elf_Ehdr *)preload_search_info(modptr, MODINFO_METADATA | 34570b7ffeeSIan Dowse MODINFOMD_ELFHDR); 34670b7ffeeSIan Dowse shdr = (Elf_Shdr *)preload_search_info(modptr, MODINFO_METADATA | 34770b7ffeeSIan Dowse MODINFOMD_SHDR); 34870b7ffeeSIan Dowse if (type == NULL || (strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE) 34970b7ffeeSIan Dowse " obj module") != 0 && 35070b7ffeeSIan Dowse strcmp(type, "elf obj module") != 0)) { 351de78ca7eSPeter Wemm return (EFTYPE); 352de78ca7eSPeter Wemm } 35370b7ffeeSIan Dowse if (baseptr == NULL || sizeptr == NULL || hdr == NULL || 35470b7ffeeSIan Dowse shdr == NULL) 35570b7ffeeSIan Dowse return (EINVAL); 35670b7ffeeSIan Dowse 35770b7ffeeSIan Dowse lf = linker_make_file(filename, &link_elf_class); 35870b7ffeeSIan Dowse if (lf == NULL) 35970b7ffeeSIan Dowse return (ENOMEM); 36070b7ffeeSIan Dowse 36170b7ffeeSIan Dowse ef = (elf_file_t)lf; 36270b7ffeeSIan Dowse ef->preloaded = 1; 36370b7ffeeSIan Dowse ef->address = *(caddr_t *)baseptr; 36470b7ffeeSIan Dowse lf->address = *(caddr_t *)baseptr; 36570b7ffeeSIan Dowse lf->size = *(size_t *)sizeptr; 36670b7ffeeSIan Dowse 36770b7ffeeSIan Dowse if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS || 36870b7ffeeSIan Dowse hdr->e_ident[EI_DATA] != ELF_TARG_DATA || 36970b7ffeeSIan Dowse hdr->e_ident[EI_VERSION] != EV_CURRENT || 37070b7ffeeSIan Dowse hdr->e_version != EV_CURRENT || 37170b7ffeeSIan Dowse hdr->e_type != ET_REL || 37270b7ffeeSIan Dowse hdr->e_machine != ELF_TARG_MACH) { 37370b7ffeeSIan Dowse error = EFTYPE; 37470b7ffeeSIan Dowse goto out; 37570b7ffeeSIan Dowse } 37670b7ffeeSIan Dowse ef->e_shdr = shdr; 37770b7ffeeSIan Dowse 37870b7ffeeSIan Dowse /* Scan the section header for information and table sizing. */ 37970b7ffeeSIan Dowse symtabindex = -1; 38070b7ffeeSIan Dowse symstrindex = -1; 38170b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 38270b7ffeeSIan Dowse switch (shdr[i].sh_type) { 38370b7ffeeSIan Dowse case SHT_PROGBITS: 38470b7ffeeSIan Dowse case SHT_NOBITS: 38513f28d96SKonstantin Belousov #ifdef __amd64__ 386b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 38713f28d96SKonstantin Belousov #endif 38858c4aee0SJohn Baldwin /* Ignore sections not loaded by the loader. */ 38958c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 39058c4aee0SJohn Baldwin break; 39170b7ffeeSIan Dowse ef->nprogtab++; 39270b7ffeeSIan Dowse break; 39370b7ffeeSIan Dowse case SHT_SYMTAB: 39470b7ffeeSIan Dowse symtabindex = i; 39570b7ffeeSIan Dowse symstrindex = shdr[i].sh_link; 39670b7ffeeSIan Dowse break; 39770b7ffeeSIan Dowse case SHT_REL: 39815746ef4SJohn Baldwin /* 39915746ef4SJohn Baldwin * Ignore relocation tables for sections not 40015746ef4SJohn Baldwin * loaded by the loader. 40115746ef4SJohn Baldwin */ 40215746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 40315746ef4SJohn Baldwin break; 404b474c780SDavid E. O'Brien ef->nreltab++; 40570b7ffeeSIan Dowse break; 40670b7ffeeSIan Dowse case SHT_RELA: 40715746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 40815746ef4SJohn Baldwin break; 409b474c780SDavid E. O'Brien ef->nrelatab++; 41070b7ffeeSIan Dowse break; 41170b7ffeeSIan Dowse } 41270b7ffeeSIan Dowse } 41370b7ffeeSIan Dowse 41470b7ffeeSIan Dowse shstrindex = hdr->e_shstrndx; 41570b7ffeeSIan Dowse if (ef->nprogtab == 0 || symstrindex < 0 || 41670b7ffeeSIan Dowse symstrindex >= hdr->e_shnum || 41770b7ffeeSIan Dowse shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 || 41870b7ffeeSIan Dowse shstrindex >= hdr->e_shnum || 41970b7ffeeSIan Dowse shdr[shstrindex].sh_type != SHT_STRTAB) { 42070b7ffeeSIan Dowse printf("%s: bad/missing section headers\n", filename); 42170b7ffeeSIan Dowse error = ENOEXEC; 42270b7ffeeSIan Dowse goto out; 42370b7ffeeSIan Dowse } 42470b7ffeeSIan Dowse 42570b7ffeeSIan Dowse /* Allocate space for tracking the load chunks */ 42670b7ffeeSIan Dowse if (ef->nprogtab != 0) 42770b7ffeeSIan Dowse ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 42870b7ffeeSIan Dowse M_LINKER, M_WAITOK | M_ZERO); 429b474c780SDavid E. O'Brien if (ef->nreltab != 0) 430b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 431b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 432b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 433b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 434b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 43570b7ffeeSIan Dowse if ((ef->nprogtab != 0 && ef->progtab == NULL) || 436b474c780SDavid E. O'Brien (ef->nreltab != 0 && ef->reltab == NULL) || 437b474c780SDavid E. O'Brien (ef->nrelatab != 0 && ef->relatab == NULL)) { 43870b7ffeeSIan Dowse error = ENOMEM; 43970b7ffeeSIan Dowse goto out; 44070b7ffeeSIan Dowse } 44170b7ffeeSIan Dowse 44270b7ffeeSIan Dowse /* XXX, relocate the sh_addr fields saved by the loader. */ 44370b7ffeeSIan Dowse off = 0; 44470b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 44570b7ffeeSIan Dowse if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off)) 44670b7ffeeSIan Dowse off = shdr[i].sh_addr; 44770b7ffeeSIan Dowse } 44870b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 44970b7ffeeSIan Dowse if (shdr[i].sh_addr != 0) 45070b7ffeeSIan Dowse shdr[i].sh_addr = shdr[i].sh_addr - off + 45170b7ffeeSIan Dowse (Elf_Addr)ef->address; 45270b7ffeeSIan Dowse } 45370b7ffeeSIan Dowse 45470b7ffeeSIan Dowse ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 45570b7ffeeSIan Dowse ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr; 45670b7ffeeSIan Dowse ef->ddbstrcnt = shdr[symstrindex].sh_size; 45770b7ffeeSIan Dowse ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr; 45870b7ffeeSIan Dowse ef->shstrcnt = shdr[shstrindex].sh_size; 45970b7ffeeSIan Dowse ef->shstrtab = (char *)shdr[shstrindex].sh_addr; 46070b7ffeeSIan Dowse 46170b7ffeeSIan Dowse /* Now fill out progtab and the relocation tables. */ 46270b7ffeeSIan Dowse pb = 0; 46370b7ffeeSIan Dowse rl = 0; 46470b7ffeeSIan Dowse ra = 0; 46570b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 46670b7ffeeSIan Dowse switch (shdr[i].sh_type) { 46770b7ffeeSIan Dowse case SHT_PROGBITS: 46870b7ffeeSIan Dowse case SHT_NOBITS: 46913f28d96SKonstantin Belousov #ifdef __amd64__ 470b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 47113f28d96SKonstantin Belousov #endif 47258c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 47358c4aee0SJohn Baldwin break; 47470b7ffeeSIan Dowse ef->progtab[pb].addr = (void *)shdr[i].sh_addr; 47570b7ffeeSIan Dowse if (shdr[i].sh_type == SHT_PROGBITS) 47670b7ffeeSIan Dowse ef->progtab[pb].name = "<<PROGBITS>>"; 47713f28d96SKonstantin Belousov #ifdef __amd64__ 478b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 47913f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 48013f28d96SKonstantin Belousov #endif 48170b7ffeeSIan Dowse else 48270b7ffeeSIan Dowse ef->progtab[pb].name = "<<NOBITS>>"; 48370b7ffeeSIan Dowse ef->progtab[pb].size = shdr[i].sh_size; 4841d9eae9fSMark Johnston ef->progtab[pb].flags = shdr[i].sh_flags; 48570b7ffeeSIan Dowse ef->progtab[pb].sec = i; 48670b7ffeeSIan Dowse if (ef->shstrtab && shdr[i].sh_name != 0) 48770b7ffeeSIan Dowse ef->progtab[pb].name = 48870b7ffeeSIan Dowse ef->shstrtab + shdr[i].sh_name; 48950c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 4905f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 49150c202c5SJeff Roberson void *dpcpu; 49250c202c5SJeff Roberson 49350c202c5SJeff Roberson dpcpu = dpcpu_alloc(shdr[i].sh_size); 49450c202c5SJeff Roberson if (dpcpu == NULL) { 4950f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 4969afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 4979afc5684SBjoern A. Zeeb "for %s\n", __func__, 4989afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 4990f823b64SBjoern A. Zeeb filename); 50050c202c5SJeff Roberson error = ENOSPC; 50150c202c5SJeff Roberson goto out; 50250c202c5SJeff Roberson } 50350c202c5SJeff Roberson memcpy(dpcpu, ef->progtab[pb].addr, 50450c202c5SJeff Roberson ef->progtab[pb].size); 50550c202c5SJeff Roberson dpcpu_copy(dpcpu, shdr[i].sh_size); 50650c202c5SJeff Roberson ef->progtab[pb].addr = dpcpu; 507eddfbb76SRobert Watson #ifdef VIMAGE 508eddfbb76SRobert Watson } else if (ef->progtab[pb].name != NULL && 50917ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 510eddfbb76SRobert Watson void *vnet_data; 511eddfbb76SRobert Watson 512eddfbb76SRobert Watson vnet_data = vnet_data_alloc(shdr[i].sh_size); 513eddfbb76SRobert Watson if (vnet_data == NULL) { 5140f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 5159afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 5169afc5684SBjoern A. Zeeb "for %s\n", __func__, 5179afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 5180f823b64SBjoern A. Zeeb filename); 519eddfbb76SRobert Watson error = ENOSPC; 520eddfbb76SRobert Watson goto out; 521eddfbb76SRobert Watson } 522eddfbb76SRobert Watson memcpy(vnet_data, ef->progtab[pb].addr, 523eddfbb76SRobert Watson ef->progtab[pb].size); 524eddfbb76SRobert Watson vnet_data_copy(vnet_data, shdr[i].sh_size); 525eddfbb76SRobert Watson ef->progtab[pb].addr = vnet_data; 526eddfbb76SRobert Watson #endif 5270067051fSMarcel Moolenaar } else if (ef->progtab[pb].name != NULL && 5280067051fSMarcel Moolenaar !strcmp(ef->progtab[pb].name, ".ctors")) { 5290067051fSMarcel Moolenaar lf->ctors_addr = ef->progtab[pb].addr; 5300067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 53150c202c5SJeff Roberson } 53270b7ffeeSIan Dowse 53370b7ffeeSIan Dowse /* Update all symbol values with the offset. */ 53470b7ffeeSIan Dowse for (j = 0; j < ef->ddbsymcnt; j++) { 53570b7ffeeSIan Dowse es = &ef->ddbsymtab[j]; 53670b7ffeeSIan Dowse if (es->st_shndx != i) 53770b7ffeeSIan Dowse continue; 53870b7ffeeSIan Dowse es->st_value += (Elf_Addr)ef->progtab[pb].addr; 53970b7ffeeSIan Dowse } 54070b7ffeeSIan Dowse pb++; 54170b7ffeeSIan Dowse break; 54270b7ffeeSIan Dowse case SHT_REL: 54315746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 54415746ef4SJohn Baldwin break; 54570b7ffeeSIan Dowse ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr; 54670b7ffeeSIan Dowse ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 54770b7ffeeSIan Dowse ef->reltab[rl].sec = shdr[i].sh_info; 54870b7ffeeSIan Dowse rl++; 54970b7ffeeSIan Dowse break; 55070b7ffeeSIan Dowse case SHT_RELA: 55115746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 55215746ef4SJohn Baldwin break; 55370b7ffeeSIan Dowse ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr; 55470b7ffeeSIan Dowse ef->relatab[ra].nrela = 55570b7ffeeSIan Dowse shdr[i].sh_size / sizeof(Elf_Rela); 55670b7ffeeSIan Dowse ef->relatab[ra].sec = shdr[i].sh_info; 55770b7ffeeSIan Dowse ra++; 55870b7ffeeSIan Dowse break; 55970b7ffeeSIan Dowse } 56070b7ffeeSIan Dowse } 5613cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 5623cfce8e4SKonstantin Belousov printf("%s: lost progbits\n", filename); 5633cfce8e4SKonstantin Belousov error = ENOEXEC; 5643cfce8e4SKonstantin Belousov goto out; 5653cfce8e4SKonstantin Belousov } 5663cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 5673cfce8e4SKonstantin Belousov printf("%s: lost reltab\n", filename); 5683cfce8e4SKonstantin Belousov error = ENOEXEC; 5693cfce8e4SKonstantin Belousov goto out; 5703cfce8e4SKonstantin Belousov } 5713cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 5723cfce8e4SKonstantin Belousov printf("%s: lost relatab\n", filename); 5733cfce8e4SKonstantin Belousov error = ENOEXEC; 5743cfce8e4SKonstantin Belousov goto out; 5753cfce8e4SKonstantin Belousov } 57670b7ffeeSIan Dowse 577f822c9e2SMark Johnston /* 578f822c9e2SMark Johnston * The file needs to be writeable and executable while applying 579f822c9e2SMark Johnston * relocations. Mapping protections are applied once relocation 580f822c9e2SMark Johnston * processing is complete. 581f822c9e2SMark Johnston */ 582f822c9e2SMark Johnston link_elf_protect_range(ef, (vm_offset_t)ef->address, 583f822c9e2SMark Johnston round_page((vm_offset_t)ef->address + ef->lf.size), VM_PROT_ALL); 584f822c9e2SMark Johnston 58570b7ffeeSIan Dowse /* Local intra-module relocations */ 586289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 5873cfce8e4SKonstantin Belousov if (error != 0) 5883cfce8e4SKonstantin Belousov goto out; 58970b7ffeeSIan Dowse *result = lf; 59070b7ffeeSIan Dowse return (0); 59170b7ffeeSIan Dowse 59270b7ffeeSIan Dowse out: 59370b7ffeeSIan Dowse /* preload not done this way */ 59470b7ffeeSIan Dowse linker_file_unload(lf, LINKER_UNLOAD_FORCE); 59570b7ffeeSIan Dowse return (error); 59670b7ffeeSIan Dowse } 597de78ca7eSPeter Wemm 5980067051fSMarcel Moolenaar static void 5990067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size) 6000067051fSMarcel Moolenaar { 6010067051fSMarcel Moolenaar void (**ctor)(void); 6020067051fSMarcel Moolenaar size_t i, cnt; 6030067051fSMarcel Moolenaar 6040067051fSMarcel Moolenaar if (addr == NULL || size == 0) 6050067051fSMarcel Moolenaar return; 6060067051fSMarcel Moolenaar cnt = size / sizeof(*ctor); 6070067051fSMarcel Moolenaar ctor = (void *)addr; 6080067051fSMarcel Moolenaar for (i = 0; i < cnt; i++) { 6090067051fSMarcel Moolenaar if (ctor[i] != NULL) 6100067051fSMarcel Moolenaar (*ctor[i])(); 6110067051fSMarcel Moolenaar } 6120067051fSMarcel Moolenaar } 6130067051fSMarcel Moolenaar 614de78ca7eSPeter Wemm static int 61554823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf) 61654823af2SPeter Wemm { 61770b7ffeeSIan Dowse elf_file_t ef; 61870b7ffeeSIan Dowse int error; 61970b7ffeeSIan Dowse 62070b7ffeeSIan Dowse ef = (elf_file_t)lf; 62170b7ffeeSIan Dowse error = relocate_file(ef); 62270b7ffeeSIan Dowse if (error) 623289ead7cSKonstantin Belousov return (error); 62470b7ffeeSIan Dowse 62570b7ffeeSIan Dowse /* Notify MD code that a module is being loaded. */ 62670b7ffeeSIan Dowse error = elf_cpu_load_file(lf); 62770b7ffeeSIan Dowse if (error) 62870b7ffeeSIan Dowse return (error); 62970b7ffeeSIan Dowse 6308f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 631289ead7cSKonstantin Belousov /* Now ifuncs. */ 632289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 633289ead7cSKonstantin Belousov if (error != 0) 634289ead7cSKonstantin Belousov return (error); 6358f941950SKonstantin Belousov #endif 636289ead7cSKonstantin Belousov 637f822c9e2SMark Johnston /* Apply protections now that relocation processing is complete. */ 638f822c9e2SMark Johnston link_elf_protect(ef); 639f822c9e2SMark Johnston 6400067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 64170b7ffeeSIan Dowse return (0); 64254823af2SPeter Wemm } 64354823af2SPeter Wemm 64454823af2SPeter Wemm static int 645c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename, 646c143d6c2SMarcel Moolenaar linker_file_t *result) 647a4f67738SDoug Rabson { 648a344babfSGleb Smirnoff struct nameidata *nd; 649b40ce416SJulian Elischer struct thread *td = curthread; /* XXX */ 650caab6e90SPeter Wemm Elf_Ehdr *hdr; 651e9eabf59SPeter Wemm Elf_Shdr *shdr; 652a8774e39SPeter Wemm Elf_Sym *es; 653a8774e39SPeter Wemm int nbytes, i, j; 65420947801SPeter Wemm vm_offset_t mapbase; 655fe3db7c7SDoug Rabson size_t mapsize; 656a4f67738SDoug Rabson int error = 0; 657526d0bd5SKonstantin Belousov ssize_t resid; 658526d0bd5SKonstantin Belousov int flags; 659a4f67738SDoug Rabson elf_file_t ef; 660a4f67738SDoug Rabson linker_file_t lf; 661ca65d5c7SPeter Wemm int symtabindex; 662ca65d5c7SPeter Wemm int symstrindex; 663e9eabf59SPeter Wemm int shstrindex; 664e9eabf59SPeter Wemm int nsym; 66520947801SPeter Wemm int pb, rl, ra; 666e9eabf59SPeter Wemm int alignmask; 667ca65d5c7SPeter Wemm 668ca65d5c7SPeter Wemm shdr = NULL; 669ca65d5c7SPeter Wemm lf = NULL; 670e9eabf59SPeter Wemm mapsize = 0; 671e9eabf59SPeter Wemm hdr = NULL; 672a4f67738SDoug Rabson 673a344babfSGleb Smirnoff nd = malloc(sizeof(struct nameidata), M_TEMP, M_WAITOK); 674a344babfSGleb Smirnoff NDINIT(nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td); 675e6796b67SKirk McKusick flags = FREAD; 676a344babfSGleb Smirnoff error = vn_open(nd, &flags, 0, NULL); 677a344babfSGleb Smirnoff if (error) { 678a344babfSGleb Smirnoff free(nd, M_TEMP); 679a4f67738SDoug Rabson return error; 680a344babfSGleb Smirnoff } 681a344babfSGleb Smirnoff NDFREE(nd, NDF_ONLY_PNBUF); 682a344babfSGleb Smirnoff if (nd->ni_vp->v_type != VREG) { 6831f49b573SRuslan Ermilov error = ENOEXEC; 6841f49b573SRuslan Ermilov goto out; 6851f49b573SRuslan Ermilov } 686a3df768bSRobert Watson #ifdef MAC 687a344babfSGleb Smirnoff error = mac_kld_check_load(td->td_ucred, nd->ni_vp); 688a3df768bSRobert Watson if (error) { 689a3df768bSRobert Watson goto out; 690a3df768bSRobert Watson } 691a3df768bSRobert Watson #endif 692a4f67738SDoug Rabson 693e9eabf59SPeter Wemm /* Read the elf header from the file. */ 694e9eabf59SPeter Wemm hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK); 695a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)hdr, sizeof(*hdr), 0, 6969ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 6979ca43589SRobert Watson &resid, td); 698a4f67738SDoug Rabson if (error) 699a4f67738SDoug Rabson goto out; 700e9eabf59SPeter Wemm if (resid != 0){ 701e9eabf59SPeter Wemm error = ENOEXEC; 702e9eabf59SPeter Wemm goto out; 703e9eabf59SPeter Wemm } 704a4f67738SDoug Rabson 705caab6e90SPeter Wemm if (!IS_ELF(*hdr)) { 706fe3db7c7SDoug Rabson error = ENOEXEC; 707a4f67738SDoug Rabson goto out; 708fe3db7c7SDoug Rabson } 709fe3db7c7SDoug Rabson 710caab6e90SPeter Wemm if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS 711caab6e90SPeter Wemm || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 712552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file layout"); 713fe3db7c7SDoug Rabson error = ENOEXEC; 714fe3db7c7SDoug Rabson goto out; 715fe3db7c7SDoug Rabson } 716caab6e90SPeter Wemm if (hdr->e_ident[EI_VERSION] != EV_CURRENT 717caab6e90SPeter Wemm || hdr->e_version != EV_CURRENT) { 718552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file version"); 719fe3db7c7SDoug Rabson error = ENOEXEC; 720fe3db7c7SDoug Rabson goto out; 721fe3db7c7SDoug Rabson } 722e9eabf59SPeter Wemm if (hdr->e_type != ET_REL) { 723a1d7ce03SAttilio Rao error = ENOSYS; 724fe3db7c7SDoug Rabson goto out; 725fe3db7c7SDoug Rabson } 726caab6e90SPeter Wemm if (hdr->e_machine != ELF_TARG_MACH) { 727552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported machine"); 728fe3db7c7SDoug Rabson error = ENOEXEC; 729fe3db7c7SDoug Rabson goto out; 730fe3db7c7SDoug Rabson } 731a4f67738SDoug Rabson 732326e27d8SDoug Rabson lf = linker_make_file(filename, &link_elf_class); 733326e27d8SDoug Rabson if (!lf) { 734326e27d8SDoug Rabson error = ENOMEM; 735326e27d8SDoug Rabson goto out; 736326e27d8SDoug Rabson } 737326e27d8SDoug Rabson ef = (elf_file_t) lf; 738e9eabf59SPeter Wemm ef->nprogtab = 0; 739e9eabf59SPeter Wemm ef->e_shdr = 0; 740b474c780SDavid E. O'Brien ef->nreltab = 0; 741b474c780SDavid E. O'Brien ef->nrelatab = 0; 742e9eabf59SPeter Wemm 743e9eabf59SPeter Wemm /* Allocate and read in the section header */ 744e9eabf59SPeter Wemm nbytes = hdr->e_shnum * hdr->e_shentsize; 745e9eabf59SPeter Wemm if (nbytes == 0 || hdr->e_shoff == 0 || 746e9eabf59SPeter Wemm hdr->e_shentsize != sizeof(Elf_Shdr)) { 747e9eabf59SPeter Wemm error = ENOEXEC; 748e9eabf59SPeter Wemm goto out; 749e9eabf59SPeter Wemm } 75020947801SPeter Wemm shdr = malloc(nbytes, M_LINKER, M_WAITOK); 751e9eabf59SPeter Wemm ef->e_shdr = shdr; 752a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (caddr_t)shdr, nbytes, 753a344babfSGleb Smirnoff hdr->e_shoff, UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 754a344babfSGleb Smirnoff NOCRED, &resid, td); 755e9eabf59SPeter Wemm if (error) 756e9eabf59SPeter Wemm goto out; 757e9eabf59SPeter Wemm if (resid) { 758e9eabf59SPeter Wemm error = ENOEXEC; 759e9eabf59SPeter Wemm goto out; 760e9eabf59SPeter Wemm } 761e9eabf59SPeter Wemm 762e9eabf59SPeter Wemm /* Scan the section header for information and table sizing. */ 763e9eabf59SPeter Wemm nsym = 0; 764e9eabf59SPeter Wemm symtabindex = -1; 765e9eabf59SPeter Wemm symstrindex = -1; 766e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 767676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 768676799a0SAndriy Gapon continue; 769e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 770e9eabf59SPeter Wemm case SHT_PROGBITS: 771e9eabf59SPeter Wemm case SHT_NOBITS: 77213f28d96SKonstantin Belousov #ifdef __amd64__ 773b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 77413f28d96SKonstantin Belousov #endif 77558c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 77658c4aee0SJohn Baldwin break; 77720947801SPeter Wemm ef->nprogtab++; 778e9eabf59SPeter Wemm break; 779e9eabf59SPeter Wemm case SHT_SYMTAB: 780e9eabf59SPeter Wemm nsym++; 781e9eabf59SPeter Wemm symtabindex = i; 782e9eabf59SPeter Wemm symstrindex = shdr[i].sh_link; 783e9eabf59SPeter Wemm break; 784e9eabf59SPeter Wemm case SHT_REL: 78515746ef4SJohn Baldwin /* 78615746ef4SJohn Baldwin * Ignore relocation tables for unallocated 78715746ef4SJohn Baldwin * sections. 78815746ef4SJohn Baldwin */ 78915746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 79015746ef4SJohn Baldwin break; 791b474c780SDavid E. O'Brien ef->nreltab++; 792e9eabf59SPeter Wemm break; 793e9eabf59SPeter Wemm case SHT_RELA: 79415746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 79515746ef4SJohn Baldwin break; 796b474c780SDavid E. O'Brien ef->nrelatab++; 797e9eabf59SPeter Wemm break; 798e9eabf59SPeter Wemm case SHT_STRTAB: 799e9eabf59SPeter Wemm break; 800e9eabf59SPeter Wemm } 801e9eabf59SPeter Wemm } 80220947801SPeter Wemm if (ef->nprogtab == 0) { 803552f9f63SEdwin Groothuis link_elf_error(filename, "file has no contents"); 804e9eabf59SPeter Wemm error = ENOEXEC; 805e9eabf59SPeter Wemm goto out; 806e9eabf59SPeter Wemm } 807e9eabf59SPeter Wemm if (nsym != 1) { 808e9eabf59SPeter Wemm /* Only allow one symbol table for now */ 809f912a970SEd Maste link_elf_error(filename, 810f912a970SEd Maste "file must have exactly one symbol table"); 811e9eabf59SPeter Wemm error = ENOEXEC; 812e9eabf59SPeter Wemm goto out; 813e9eabf59SPeter Wemm } 814e9eabf59SPeter Wemm if (symstrindex < 0 || symstrindex > hdr->e_shnum || 815e9eabf59SPeter Wemm shdr[symstrindex].sh_type != SHT_STRTAB) { 816552f9f63SEdwin Groothuis link_elf_error(filename, "file has invalid symbol strings"); 817e9eabf59SPeter Wemm error = ENOEXEC; 818e9eabf59SPeter Wemm goto out; 819e9eabf59SPeter Wemm } 820e9eabf59SPeter Wemm 821e9eabf59SPeter Wemm /* Allocate space for tracking the load chunks */ 822e9eabf59SPeter Wemm if (ef->nprogtab != 0) 82320947801SPeter Wemm ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 82420947801SPeter Wemm M_LINKER, M_WAITOK | M_ZERO); 825b474c780SDavid E. O'Brien if (ef->nreltab != 0) 826b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 827b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 828b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 829b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 830b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 831e9eabf59SPeter Wemm 8323cfce8e4SKonstantin Belousov if (symtabindex == -1) { 8333cfce8e4SKonstantin Belousov link_elf_error(filename, "lost symbol table index"); 8343cfce8e4SKonstantin Belousov error = ENOEXEC; 8353cfce8e4SKonstantin Belousov goto out; 8363cfce8e4SKonstantin Belousov } 83720947801SPeter Wemm /* Allocate space for and load the symbol table */ 838e9eabf59SPeter Wemm ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 839e9eabf59SPeter Wemm ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK); 840a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)ef->ddbsymtab, 84120947801SPeter Wemm shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset, 84220947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 84320947801SPeter Wemm &resid, td); 84420947801SPeter Wemm if (error) 84520947801SPeter Wemm goto out; 84620947801SPeter Wemm if (resid != 0){ 84720947801SPeter Wemm error = EINVAL; 84820947801SPeter Wemm goto out; 84920947801SPeter Wemm } 850e9eabf59SPeter Wemm 85120947801SPeter Wemm /* Allocate space for and load the symbol strings */ 852e9eabf59SPeter Wemm ef->ddbstrcnt = shdr[symstrindex].sh_size; 853e9eabf59SPeter Wemm ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK); 854a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab, 85520947801SPeter Wemm shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset, 85620947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 85720947801SPeter Wemm &resid, td); 85820947801SPeter Wemm if (error) 85920947801SPeter Wemm goto out; 86020947801SPeter Wemm if (resid != 0){ 86120947801SPeter Wemm error = EINVAL; 86220947801SPeter Wemm goto out; 86320947801SPeter Wemm } 864e9eabf59SPeter Wemm 865e9eabf59SPeter Wemm /* Do we have a string table for the section names? */ 866e9eabf59SPeter Wemm shstrindex = -1; 86720947801SPeter Wemm if (hdr->e_shstrndx != 0 && 86820947801SPeter Wemm shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) { 869e9eabf59SPeter Wemm shstrindex = hdr->e_shstrndx; 870e9eabf59SPeter Wemm ef->shstrcnt = shdr[shstrindex].sh_size; 87120947801SPeter Wemm ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER, 87220947801SPeter Wemm M_WAITOK); 873a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab, 87420947801SPeter Wemm shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset, 875e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 876e9eabf59SPeter Wemm &resid, td); 877e9eabf59SPeter Wemm if (error) 878e9eabf59SPeter Wemm goto out; 87920947801SPeter Wemm if (resid != 0){ 88020947801SPeter Wemm error = EINVAL; 88120947801SPeter Wemm goto out; 88220947801SPeter Wemm } 883e9eabf59SPeter Wemm } 884e9eabf59SPeter Wemm 88520947801SPeter Wemm /* Size up code/data(progbits) and bss(nobits). */ 886e9eabf59SPeter Wemm alignmask = 0; 887e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 888676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 889676799a0SAndriy Gapon continue; 890e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 891e9eabf59SPeter Wemm case SHT_PROGBITS: 892e9eabf59SPeter Wemm case SHT_NOBITS: 89313f28d96SKonstantin Belousov #ifdef __amd64__ 894b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 89513f28d96SKonstantin Belousov #endif 89658c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 89758c4aee0SJohn Baldwin break; 898e9eabf59SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 899e9eabf59SPeter Wemm mapsize += alignmask; 900e9eabf59SPeter Wemm mapsize &= ~alignmask; 901e9eabf59SPeter Wemm mapsize += shdr[i].sh_size; 902e9eabf59SPeter Wemm break; 903e9eabf59SPeter Wemm } 904e9eabf59SPeter Wemm } 905e9eabf59SPeter Wemm 906e9eabf59SPeter Wemm /* 907e9eabf59SPeter Wemm * We know how much space we need for the text/data/bss/etc. 908e9eabf59SPeter Wemm * This stuff needs to be in a single chunk so that profiling etc 909e9eabf59SPeter Wemm * can get the bounds and gdb can associate offsets with modules 910e9eabf59SPeter Wemm */ 911fbf2a778SKonstantin Belousov ef->object = vm_pager_allocate(OBJT_PHYS, NULL, round_page(mapsize), 912fbf2a778SKonstantin Belousov VM_PROT_ALL, 0, thread0.td_ucred); 913fe3db7c7SDoug Rabson if (ef->object == NULL) { 914fe3db7c7SDoug Rabson error = ENOMEM; 915fe3db7c7SDoug Rabson goto out; 916fe3db7c7SDoug Rabson } 917*d80126a6SMark Johnston #if VM_NRESERVLEVEL > 0 918*d80126a6SMark Johnston vm_object_color(ef->object, 0); 919*d80126a6SMark Johnston #endif 920ac68d1c9SAlan Cox 921ac68d1c9SAlan Cox /* 922ac68d1c9SAlan Cox * In order to satisfy amd64's architectural requirements on the 923ac68d1c9SAlan Cox * location of code and data in the kernel's address space, request a 924ac68d1c9SAlan Cox * mapping that is above the kernel. 9251d9eae9fSMark Johnston * 9261d9eae9fSMark Johnston * Protections will be restricted once relocations are applied. 927ac68d1c9SAlan Cox */ 9284104e835SOleksandr Tymoshenko #ifdef __amd64__ 929ac68d1c9SAlan Cox mapbase = KERNBASE; 9304104e835SOleksandr Tymoshenko #else 9314104e835SOleksandr Tymoshenko mapbase = VM_MIN_KERNEL_ADDRESS; 9324104e835SOleksandr Tymoshenko #endif 93320947801SPeter Wemm error = vm_map_find(kernel_map, ef->object, 0, &mapbase, 934edb572a3SJohn Baldwin round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL, 935edb572a3SJohn Baldwin VM_PROT_ALL, 0); 936092bacb2SMark Johnston if (error != KERN_SUCCESS) { 937fe3db7c7SDoug Rabson vm_object_deallocate(ef->object); 938092bacb2SMark Johnston ef->object = NULL; 939092bacb2SMark Johnston error = ENOMEM; 940fe3db7c7SDoug Rabson goto out; 941fe3db7c7SDoug Rabson } 94220947801SPeter Wemm 943e9eabf59SPeter Wemm /* Wire the pages */ 9447f1ef325SAlan Cox error = vm_map_wire(kernel_map, mapbase, 94520947801SPeter Wemm mapbase + round_page(mapsize), 946e9eabf59SPeter Wemm VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES); 9477f1ef325SAlan Cox if (error != KERN_SUCCESS) { 9487f1ef325SAlan Cox error = ENOMEM; 9497f1ef325SAlan Cox goto out; 9507f1ef325SAlan Cox } 951a4f67738SDoug Rabson 95220947801SPeter Wemm /* Inform the kld system about the situation */ 95320947801SPeter Wemm lf->address = ef->address = (caddr_t)mapbase; 95420947801SPeter Wemm lf->size = mapsize; 955e9eabf59SPeter Wemm 956e9eabf59SPeter Wemm /* 95720947801SPeter Wemm * Now load code/data(progbits), zero bss(nobits), allocate space for 95820947801SPeter Wemm * and load relocs 959e9eabf59SPeter Wemm */ 96020947801SPeter Wemm pb = 0; 96120947801SPeter Wemm rl = 0; 96220947801SPeter Wemm ra = 0; 96320947801SPeter Wemm alignmask = 0; 96420947801SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 965676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 966676799a0SAndriy Gapon continue; 96720947801SPeter Wemm switch (shdr[i].sh_type) { 96820947801SPeter Wemm case SHT_PROGBITS: 96920947801SPeter Wemm case SHT_NOBITS: 97013f28d96SKonstantin Belousov #ifdef __amd64__ 971b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 97213f28d96SKonstantin Belousov #endif 97358c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 97458c4aee0SJohn Baldwin break; 97520947801SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 97620947801SPeter Wemm mapbase += alignmask; 97720947801SPeter Wemm mapbase &= ~alignmask; 9780067051fSMarcel Moolenaar if (ef->shstrtab != NULL && shdr[i].sh_name != 0) { 97950c202c5SJeff Roberson ef->progtab[pb].name = 98050c202c5SJeff Roberson ef->shstrtab + shdr[i].sh_name; 9810067051fSMarcel Moolenaar if (!strcmp(ef->progtab[pb].name, ".ctors")) { 9820067051fSMarcel Moolenaar lf->ctors_addr = (caddr_t)mapbase; 9830067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 9840067051fSMarcel Moolenaar } 9850067051fSMarcel Moolenaar } else if (shdr[i].sh_type == SHT_PROGBITS) 98620947801SPeter Wemm ef->progtab[pb].name = "<<PROGBITS>>"; 98713f28d96SKonstantin Belousov #ifdef __amd64__ 988b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 98913f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 99013f28d96SKonstantin Belousov #endif 99150c202c5SJeff Roberson else 99250c202c5SJeff Roberson ef->progtab[pb].name = "<<NOBITS>>"; 99350c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 9940f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 99550c202c5SJeff Roberson ef->progtab[pb].addr = 99650c202c5SJeff Roberson dpcpu_alloc(shdr[i].sh_size); 9970f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 9980f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 9999afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 10009afc5684SBjoern A. Zeeb "for %s\n", __func__, 10019afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 10020f823b64SBjoern A. Zeeb filename); 10030f823b64SBjoern A. Zeeb } 10040f823b64SBjoern A. Zeeb } 1005eddfbb76SRobert Watson #ifdef VIMAGE 1006eddfbb76SRobert Watson else if (ef->progtab[pb].name != NULL && 10070f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 1008eddfbb76SRobert Watson ef->progtab[pb].addr = 1009eddfbb76SRobert Watson vnet_data_alloc(shdr[i].sh_size); 10100f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 10110f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 10129afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 10139afc5684SBjoern A. Zeeb "for %s\n", __func__, 10149afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 10150f823b64SBjoern A. Zeeb filename); 10160f823b64SBjoern A. Zeeb } 10170f823b64SBjoern A. Zeeb } 1018eddfbb76SRobert Watson #endif 101950c202c5SJeff Roberson else 102050c202c5SJeff Roberson ef->progtab[pb].addr = 102150c202c5SJeff Roberson (void *)(uintptr_t)mapbase; 102250c202c5SJeff Roberson if (ef->progtab[pb].addr == NULL) { 102350c202c5SJeff Roberson error = ENOSPC; 102450c202c5SJeff Roberson goto out; 102550c202c5SJeff Roberson } 102650c202c5SJeff Roberson ef->progtab[pb].size = shdr[i].sh_size; 10271d9eae9fSMark Johnston ef->progtab[pb].flags = shdr[i].sh_flags; 102850c202c5SJeff Roberson ef->progtab[pb].sec = i; 102913f28d96SKonstantin Belousov if (shdr[i].sh_type == SHT_PROGBITS 103013f28d96SKonstantin Belousov #ifdef __amd64__ 1031b715d9afSKonstantin Belousov || shdr[i].sh_type == SHT_X86_64_UNWIND 103213f28d96SKonstantin Belousov #endif 103313f28d96SKonstantin Belousov ) { 1034a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 103520947801SPeter Wemm ef->progtab[pb].addr, 103620947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 103720947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 103820947801SPeter Wemm NOCRED, &resid, td); 103920947801SPeter Wemm if (error) 104020947801SPeter Wemm goto out; 104120947801SPeter Wemm if (resid != 0){ 104220947801SPeter Wemm error = EINVAL; 104320947801SPeter Wemm goto out; 104420947801SPeter Wemm } 1045eddfbb76SRobert Watson /* Initialize the per-cpu or vnet area. */ 1046eddfbb76SRobert Watson if (ef->progtab[pb].addr != (void *)mapbase && 10475f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 104850c202c5SJeff Roberson dpcpu_copy(ef->progtab[pb].addr, 104950c202c5SJeff Roberson shdr[i].sh_size); 1050eddfbb76SRobert Watson #ifdef VIMAGE 1051eddfbb76SRobert Watson else if (ef->progtab[pb].addr != 1052eddfbb76SRobert Watson (void *)mapbase && 105317ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 1054eddfbb76SRobert Watson vnet_data_copy(ef->progtab[pb].addr, 1055eddfbb76SRobert Watson shdr[i].sh_size); 1056eddfbb76SRobert Watson #endif 105750c202c5SJeff Roberson } else 105820947801SPeter Wemm bzero(ef->progtab[pb].addr, shdr[i].sh_size); 1059a8774e39SPeter Wemm 1060a8774e39SPeter Wemm /* Update all symbol values with the offset. */ 1061a8774e39SPeter Wemm for (j = 0; j < ef->ddbsymcnt; j++) { 1062a8774e39SPeter Wemm es = &ef->ddbsymtab[j]; 1063a8774e39SPeter Wemm if (es->st_shndx != i) 1064a8774e39SPeter Wemm continue; 1065a8774e39SPeter Wemm es->st_value += (Elf_Addr)ef->progtab[pb].addr; 1066a8774e39SPeter Wemm } 106720947801SPeter Wemm mapbase += shdr[i].sh_size; 106820947801SPeter Wemm pb++; 106920947801SPeter Wemm break; 107020947801SPeter Wemm case SHT_REL: 107115746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 107215746ef4SJohn Baldwin break; 107320947801SPeter Wemm ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER, 107420947801SPeter Wemm M_WAITOK); 107520947801SPeter Wemm ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 107620947801SPeter Wemm ef->reltab[rl].sec = shdr[i].sh_info; 1077a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 107820947801SPeter Wemm (void *)ef->reltab[rl].rel, 107920947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 1080e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 1081e9eabf59SPeter Wemm &resid, td); 1082e9eabf59SPeter Wemm if (error) 1083e9eabf59SPeter Wemm goto out; 108420947801SPeter Wemm if (resid != 0){ 108520947801SPeter Wemm error = EINVAL; 108620947801SPeter Wemm goto out; 1087e9eabf59SPeter Wemm } 108820947801SPeter Wemm rl++; 108920947801SPeter Wemm break; 109020947801SPeter Wemm case SHT_RELA: 109115746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 109215746ef4SJohn Baldwin break; 109320947801SPeter Wemm ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER, 109420947801SPeter Wemm M_WAITOK); 109520947801SPeter Wemm ef->relatab[ra].nrela = 109620947801SPeter Wemm shdr[i].sh_size / sizeof(Elf_Rela); 109720947801SPeter Wemm ef->relatab[ra].sec = shdr[i].sh_info; 1098a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 109920947801SPeter Wemm (void *)ef->relatab[ra].rela, 110020947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 1101e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 1102e9eabf59SPeter Wemm &resid, td); 1103e9eabf59SPeter Wemm if (error) 1104e9eabf59SPeter Wemm goto out; 110520947801SPeter Wemm if (resid != 0){ 110620947801SPeter Wemm error = EINVAL; 110720947801SPeter Wemm goto out; 1108e9eabf59SPeter Wemm } 110920947801SPeter Wemm ra++; 111020947801SPeter Wemm break; 111120947801SPeter Wemm } 111220947801SPeter Wemm } 11133cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 11143cfce8e4SKonstantin Belousov link_elf_error(filename, "lost progbits"); 11153cfce8e4SKonstantin Belousov error = ENOEXEC; 11163cfce8e4SKonstantin Belousov goto out; 11173cfce8e4SKonstantin Belousov } 11183cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 11193cfce8e4SKonstantin Belousov link_elf_error(filename, "lost reltab"); 11203cfce8e4SKonstantin Belousov error = ENOEXEC; 11213cfce8e4SKonstantin Belousov goto out; 11223cfce8e4SKonstantin Belousov } 11233cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 11243cfce8e4SKonstantin Belousov link_elf_error(filename, "lost relatab"); 11253cfce8e4SKonstantin Belousov error = ENOEXEC; 11263cfce8e4SKonstantin Belousov goto out; 11273cfce8e4SKonstantin Belousov } 11283cfce8e4SKonstantin Belousov if (mapbase != (vm_offset_t)ef->address + mapsize) { 11293cfce8e4SKonstantin Belousov printf( 11303cfce8e4SKonstantin Belousov "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n", 11313cfce8e4SKonstantin Belousov filename != NULL ? filename : "<none>", 11326f3c6327SNeel Natu (u_long)mapbase, ef->address, (u_long)mapsize, 11336f3c6327SNeel Natu (u_long)(vm_offset_t)ef->address + mapsize); 11343cfce8e4SKonstantin Belousov error = ENOMEM; 11353cfce8e4SKonstantin Belousov goto out; 11363cfce8e4SKonstantin Belousov } 1137e9eabf59SPeter Wemm 1138e9eabf59SPeter Wemm /* Local intra-module relocations */ 1139289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 11403cfce8e4SKonstantin Belousov if (error != 0) 11413cfce8e4SKonstantin Belousov goto out; 11427251b4bfSJake Burkholder 1143e9eabf59SPeter Wemm /* Pull in dependencies */ 1144b249ce48SMateusz Guzik VOP_UNLOCK(nd->ni_vp); 11457b9716baSIan Dowse error = linker_load_dependencies(lf); 1146a344babfSGleb Smirnoff vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY); 1147ca65d5c7SPeter Wemm if (error) 1148de78ca7eSPeter Wemm goto out; 1149e9eabf59SPeter Wemm 1150e9eabf59SPeter Wemm /* External relocations */ 1151326e27d8SDoug Rabson error = relocate_file(ef); 1152ca65d5c7SPeter Wemm if (error) 1153ca65d5c7SPeter Wemm goto out; 1154ca65d5c7SPeter Wemm 1155e9eabf59SPeter Wemm /* Notify MD code that a module is being loaded. */ 1156e9eabf59SPeter Wemm error = elf_cpu_load_file(lf); 1157ca65d5c7SPeter Wemm if (error) 1158ca65d5c7SPeter Wemm goto out; 1159a4f67738SDoug Rabson 11608f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 1161289ead7cSKonstantin Belousov /* Now ifuncs. */ 1162289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 1163289ead7cSKonstantin Belousov if (error != 0) 1164289ead7cSKonstantin Belousov goto out; 11658f941950SKonstantin Belousov #endif 1166289ead7cSKonstantin Belousov 11671d9eae9fSMark Johnston link_elf_protect(ef); 11680067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 1169a4f67738SDoug Rabson *result = lf; 1170a4f67738SDoug Rabson 1171a4f67738SDoug Rabson out: 1172b249ce48SMateusz Guzik VOP_UNLOCK(nd->ni_vp); 1173a344babfSGleb Smirnoff vn_close(nd->ni_vp, FREAD, td->td_ucred, td); 1174a344babfSGleb Smirnoff free(nd, M_TEMP); 1175ca65d5c7SPeter Wemm if (error && lf) 117665a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 1177e9eabf59SPeter Wemm free(hdr, M_LINKER); 1178a4f67738SDoug Rabson 1179a4f67738SDoug Rabson return error; 1180a4f67738SDoug Rabson } 1181a4f67738SDoug Rabson 1182a4f67738SDoug Rabson static void 1183de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file) 1184a4f67738SDoug Rabson { 1185326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) file; 1186d821d364SPedro F. Giffuni u_int i; 1187a4f67738SDoug Rabson 11881aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being unloaded. */ 11891aeb23cdSMarcel Moolenaar elf_cpu_unload_file(file); 11901aeb23cdSMarcel Moolenaar 119150c202c5SJeff Roberson if (ef->progtab) { 119250c202c5SJeff Roberson for (i = 0; i < ef->nprogtab; i++) { 119350c202c5SJeff Roberson if (ef->progtab[i].size == 0) 119450c202c5SJeff Roberson continue; 119550c202c5SJeff Roberson if (ef->progtab[i].name == NULL) 119650c202c5SJeff Roberson continue; 11975f67450dSDimitry Andric if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME)) 119850c202c5SJeff Roberson dpcpu_free(ef->progtab[i].addr, 119950c202c5SJeff Roberson ef->progtab[i].size); 1200eddfbb76SRobert Watson #ifdef VIMAGE 120117ef1febSRobert Watson else if (!strcmp(ef->progtab[i].name, VNET_SETNAME)) 1202eddfbb76SRobert Watson vnet_data_free(ef->progtab[i].addr, 1203eddfbb76SRobert Watson ef->progtab[i].size); 1204eddfbb76SRobert Watson #endif 120550c202c5SJeff Roberson } 120650c202c5SJeff Roberson } 120770b7ffeeSIan Dowse if (ef->preloaded) { 120870b7ffeeSIan Dowse free(ef->reltab, M_LINKER); 120970b7ffeeSIan Dowse free(ef->relatab, M_LINKER); 121070b7ffeeSIan Dowse free(ef->progtab, M_LINKER); 1211a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1212a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1213a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 121473624a80SMark Johnston if (file->pathname != NULL) 121573624a80SMark Johnston preload_delete_name(file->pathname); 121670b7ffeeSIan Dowse return; 121770b7ffeeSIan Dowse } 121870b7ffeeSIan Dowse 1219b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) 122020947801SPeter Wemm free(ef->reltab[i].rel, M_LINKER); 1221b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) 122220947801SPeter Wemm free(ef->relatab[i].rela, M_LINKER); 122320947801SPeter Wemm free(ef->reltab, M_LINKER); 122420947801SPeter Wemm free(ef->relatab, M_LINKER); 122520947801SPeter Wemm free(ef->progtab, M_LINKER); 122620947801SPeter Wemm 1227092bacb2SMark Johnston if (ef->object != NULL) 1228fe3db7c7SDoug Rabson vm_map_remove(kernel_map, (vm_offset_t)ef->address, 1229092bacb2SMark Johnston (vm_offset_t)ef->address + ptoa(ef->object->size)); 1230e9eabf59SPeter Wemm free(ef->e_shdr, M_LINKER); 1231e9eabf59SPeter Wemm free(ef->ddbsymtab, M_LINKER); 1232e9eabf59SPeter Wemm free(ef->ddbstrtab, M_LINKER); 123320947801SPeter Wemm free(ef->shstrtab, M_LINKER); 1234a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1235a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1236a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 1237de78ca7eSPeter Wemm } 1238de78ca7eSPeter Wemm 1239fe3db7c7SDoug Rabson static const char * 1240757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info) 1241a4f67738SDoug Rabson { 1242fe3db7c7SDoug Rabson const Elf_Sym *ref; 1243a4f67738SDoug Rabson 1244aa855a59SPeter Wemm if (ELF_R_SYM(r_info)) { 1245e9eabf59SPeter Wemm ref = ef->ddbsymtab + ELF_R_SYM(r_info); 1246e9eabf59SPeter Wemm return ef->ddbstrtab + ref->st_name; 1247fe3db7c7SDoug Rabson } else 1248fe3db7c7SDoug Rabson return NULL; 1249a4f67738SDoug Rabson } 1250a4f67738SDoug Rabson 1251e9eabf59SPeter Wemm static Elf_Addr 1252e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec) 1253e9eabf59SPeter Wemm { 1254e9eabf59SPeter Wemm int i; 1255e9eabf59SPeter Wemm Elf_Addr base = 0; 1256e9eabf59SPeter Wemm 1257e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1258a8774e39SPeter Wemm if (sec == ef->progtab[i].sec) { 1259e9eabf59SPeter Wemm base = (Elf_Addr)ef->progtab[i].addr; 1260a8774e39SPeter Wemm break; 1261e9eabf59SPeter Wemm } 1262a8774e39SPeter Wemm } 1263e9eabf59SPeter Wemm return base; 1264e9eabf59SPeter Wemm } 1265e9eabf59SPeter Wemm 1266a4f67738SDoug Rabson static int 1267326e27d8SDoug Rabson relocate_file(elf_file_t ef) 1268a4f67738SDoug Rabson { 1269fe3db7c7SDoug Rabson const Elf_Rel *rellim; 1270fe3db7c7SDoug Rabson const Elf_Rel *rel; 1271fe3db7c7SDoug Rabson const Elf_Rela *relalim; 1272fe3db7c7SDoug Rabson const Elf_Rela *rela; 1273aa855a59SPeter Wemm const char *symname; 1274e9eabf59SPeter Wemm const Elf_Sym *sym; 1275e9eabf59SPeter Wemm int i; 1276757686b1SMarcel Moolenaar Elf_Size symidx; 1277e9eabf59SPeter Wemm Elf_Addr base; 1278e9eabf59SPeter Wemm 1279fe3db7c7SDoug Rabson /* Perform relocations without addend if there are any: */ 1280b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1281e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 12823cfce8e4SKonstantin Belousov if (rel == NULL) { 12833cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab!"); 12843cfce8e4SKonstantin Belousov return (ENOEXEC); 12853cfce8e4SKonstantin Belousov } 128620947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1287e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 12883cfce8e4SKonstantin Belousov if (base == 0) { 12893cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 12903cfce8e4SKonstantin Belousov return (ENOEXEC); 12913cfce8e4SKonstantin Belousov } 129220947801SPeter Wemm for ( ; rel < rellim; rel++) { 1293e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 129420947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 129520947801SPeter Wemm continue; 1296e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 129720947801SPeter Wemm /* Local relocs are already done */ 129820947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 129920947801SPeter Wemm continue; 130020947801SPeter Wemm if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL, 130120947801SPeter Wemm elf_obj_lookup)) { 1302aa855a59SPeter Wemm symname = symbol_name(ef, rel->r_info); 130320947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 130420947801SPeter Wemm symname); 13053cfce8e4SKonstantin Belousov return (ENOENT); 1306a13ddfb6SPeter Wemm } 1307aa855a59SPeter Wemm } 1308a4f67738SDoug Rabson } 1309a4f67738SDoug Rabson 1310fe3db7c7SDoug Rabson /* Perform relocations with addend if there are any: */ 1311b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1312e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 13133cfce8e4SKonstantin Belousov if (rela == NULL) { 13143cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 13153cfce8e4SKonstantin Belousov return (ENOEXEC); 13163cfce8e4SKonstantin Belousov } 131720947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1318e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 13193cfce8e4SKonstantin Belousov if (base == 0) { 13203cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, 13213cfce8e4SKonstantin Belousov "lost base for relatab"); 13223cfce8e4SKonstantin Belousov return (ENOEXEC); 13233cfce8e4SKonstantin Belousov } 132420947801SPeter Wemm for ( ; rela < relalim; rela++) { 1325e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 132620947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 132720947801SPeter Wemm continue; 1328e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 132920947801SPeter Wemm /* Local relocs are already done */ 133020947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 133120947801SPeter Wemm continue; 133220947801SPeter Wemm if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA, 133320947801SPeter Wemm elf_obj_lookup)) { 1334aa855a59SPeter Wemm symname = symbol_name(ef, rela->r_info); 133520947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 133620947801SPeter Wemm symname); 13373cfce8e4SKonstantin Belousov return (ENOENT); 1338a13ddfb6SPeter Wemm } 1339e9eabf59SPeter Wemm } 1340e9eabf59SPeter Wemm } 1341a4f67738SDoug Rabson 13422832cd54SKonstantin Belousov /* 1343e3043798SPedro F. Giffuni * Only clean SHN_FBSD_CACHED for successful return. If we 13442832cd54SKonstantin Belousov * modified symbol table for the object but found an 13452832cd54SKonstantin Belousov * unresolved symbol, there is no reason to roll back. 13462832cd54SKonstantin Belousov */ 13472832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(ef); 13482832cd54SKonstantin Belousov 13493cfce8e4SKonstantin Belousov return (0); 1350a4f67738SDoug Rabson } 1351a4f67738SDoug Rabson 135237c84183SPoul-Henning Kamp static int 1353d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym) 1354a4f67738SDoug Rabson { 1355326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 13562d636ab0SPeter Wemm const Elf_Sym *symp; 13572d636ab0SPeter Wemm const char *strp; 1358a4f67738SDoug Rabson int i; 1359a4f67738SDoug Rabson 13602d636ab0SPeter Wemm for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 13612d636ab0SPeter Wemm strp = ef->ddbstrtab + symp->st_name; 1362a8774e39SPeter Wemm if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) { 1363d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 13642d636ab0SPeter Wemm return 0; 13652d636ab0SPeter Wemm } 13662d636ab0SPeter Wemm } 1367a4f67738SDoug Rabson return ENOENT; 1368a4f67738SDoug Rabson } 1369a4f67738SDoug Rabson 1370de78ca7eSPeter Wemm static int 137120947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, 137220947801SPeter Wemm linker_symval_t *symval) 1373a4f67738SDoug Rabson { 13747035cf14SKonstantin Belousov elf_file_t ef; 13757035cf14SKonstantin Belousov const Elf_Sym *es; 13767035cf14SKonstantin Belousov caddr_t val; 1377a4f67738SDoug Rabson 13787035cf14SKonstantin Belousov ef = (elf_file_t) lf; 13797035cf14SKonstantin Belousov es = (const Elf_Sym*) sym; 13807035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 13815cf87418SMarcel Moolenaar if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) { 13822d636ab0SPeter Wemm symval->name = ef->ddbstrtab + es->st_name; 13837035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 13847035cf14SKonstantin Belousov if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC) 13857035cf14SKonstantin Belousov val = ((caddr_t (*)(void))val)(); 13867035cf14SKonstantin Belousov symval->value = val; 13872d636ab0SPeter Wemm symval->size = es->st_size; 13882d636ab0SPeter Wemm return 0; 13892d636ab0SPeter Wemm } 13902d636ab0SPeter Wemm return ENOENT; 13912d636ab0SPeter Wemm } 1392a4f67738SDoug Rabson 1393a4f67738SDoug Rabson static int 1394a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value, 1395d254af07SMatthew Dillon c_linker_sym_t *sym, long *diffp) 1396a4f67738SDoug Rabson { 1397326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1398586453feSBruce Evans u_long off = (uintptr_t) (void *) value; 1399a4f67738SDoug Rabson u_long diff = off; 1400586453feSBruce Evans u_long st_value; 1401fe3db7c7SDoug Rabson const Elf_Sym *es; 1402b85f65afSPedro F. Giffuni const Elf_Sym *best = NULL; 1403a4f67738SDoug Rabson int i; 1404a4f67738SDoug Rabson 14052d636ab0SPeter Wemm for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 1406a4f67738SDoug Rabson if (es->st_name == 0) 1407a4f67738SDoug Rabson continue; 1408a8774e39SPeter Wemm st_value = es->st_value; 1409b5abfb70SPeter Wemm if (off >= st_value) { 1410b5abfb70SPeter Wemm if (off - st_value < diff) { 1411b5abfb70SPeter Wemm diff = off - st_value; 1412a4f67738SDoug Rabson best = es; 1413a4f67738SDoug Rabson if (diff == 0) 1414a4f67738SDoug Rabson break; 1415b5abfb70SPeter Wemm } else if (off - st_value == diff) { 1416a4f67738SDoug Rabson best = es; 1417a4f67738SDoug Rabson } 1418a4f67738SDoug Rabson } 1419a4f67738SDoug Rabson } 1420b85f65afSPedro F. Giffuni if (best == NULL) 1421a4f67738SDoug Rabson *diffp = off; 1422a4f67738SDoug Rabson else 1423a4f67738SDoug Rabson *diffp = diff; 1424d254af07SMatthew Dillon *sym = (c_linker_sym_t) best; 1425a4f67738SDoug Rabson 1426a4f67738SDoug Rabson return 0; 1427a4f67738SDoug Rabson } 1428f41325dbSPeter Wemm 1429f41325dbSPeter Wemm /* 1430f41325dbSPeter Wemm * Look up a linker set on an ELF system. 1431f41325dbSPeter Wemm */ 1432f41325dbSPeter Wemm static int 1433f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name, 1434f41325dbSPeter Wemm void ***startp, void ***stopp, int *countp) 1435f41325dbSPeter Wemm { 1436e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 1437f41325dbSPeter Wemm void **start, **stop; 1438e9eabf59SPeter Wemm int i, count; 1439f41325dbSPeter Wemm 1440e9eabf59SPeter Wemm /* Relative to section number */ 1441e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1442e9eabf59SPeter Wemm if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) && 1443e9eabf59SPeter Wemm strcmp(ef->progtab[i].name + 4, name) == 0) { 1444e9eabf59SPeter Wemm start = (void **)ef->progtab[i].addr; 144520947801SPeter Wemm stop = (void **)((char *)ef->progtab[i].addr + 144620947801SPeter Wemm ef->progtab[i].size); 1447f41325dbSPeter Wemm count = stop - start; 1448f41325dbSPeter Wemm if (startp) 1449f41325dbSPeter Wemm *startp = start; 1450f41325dbSPeter Wemm if (stopp) 1451f41325dbSPeter Wemm *stopp = stop; 1452f41325dbSPeter Wemm if (countp) 1453f41325dbSPeter Wemm *countp = count; 1454e9eabf59SPeter Wemm return (0); 1455e9eabf59SPeter Wemm } 1456e9eabf59SPeter Wemm } 1457e9eabf59SPeter Wemm return (ESRCH); 1458f41325dbSPeter Wemm } 1459bb9fe9ddSBrian Feldman 1460bb9fe9ddSBrian Feldman static int 1461bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file, 1462e9eabf59SPeter Wemm int (*callback)(const char *, void *), void *opaque) 1463e9eabf59SPeter Wemm { 1464bb9fe9ddSBrian Feldman elf_file_t ef = (elf_file_t)file; 1465bb9fe9ddSBrian Feldman const Elf_Sym *symp; 1466bb9fe9ddSBrian Feldman int i, error; 1467bb9fe9ddSBrian Feldman 1468bb9fe9ddSBrian Feldman /* Exhaustive search */ 1469bb9fe9ddSBrian Feldman for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1470bb9fe9ddSBrian Feldman if (symp->st_value != 0 && 14717035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 14727035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 1473bb9fe9ddSBrian Feldman error = callback(ef->ddbstrtab + symp->st_name, opaque); 1474bb9fe9ddSBrian Feldman if (error) 1475bb9fe9ddSBrian Feldman return (error); 1476bb9fe9ddSBrian Feldman } 1477bb9fe9ddSBrian Feldman } 1478bb9fe9ddSBrian Feldman return (0); 1479bb9fe9ddSBrian Feldman } 148084201059SMarcel Moolenaar 1481a2024a3eSJohn Birrell static int 1482a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file, 1483a2024a3eSJohn Birrell linker_function_nameval_callback_t callback, void *opaque) 1484a2024a3eSJohn Birrell { 1485a2024a3eSJohn Birrell linker_symval_t symval; 1486a2024a3eSJohn Birrell elf_file_t ef = (elf_file_t)file; 1487a2024a3eSJohn Birrell const Elf_Sym* symp; 1488a2024a3eSJohn Birrell int i, error; 1489a2024a3eSJohn Birrell 1490a2024a3eSJohn Birrell /* Exhaustive search */ 1491a2024a3eSJohn Birrell for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1492a2024a3eSJohn Birrell if (symp->st_value != 0 && 14937035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 14947035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 14957035cf14SKonstantin Belousov error = link_elf_symbol_values(file, 14967035cf14SKonstantin Belousov (c_linker_sym_t)symp, &symval); 1497a2024a3eSJohn Birrell if (error) 1498a2024a3eSJohn Birrell return (error); 1499a2024a3eSJohn Birrell error = callback(file, i, &symval, opaque); 1500a2024a3eSJohn Birrell if (error) 1501a2024a3eSJohn Birrell return (error); 1502a2024a3eSJohn Birrell } 1503a2024a3eSJohn Birrell } 1504a2024a3eSJohn Birrell return (0); 1505a2024a3eSJohn Birrell } 1506a2024a3eSJohn Birrell 15072832cd54SKonstantin Belousov static void 15082832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef) 15092832cd54SKonstantin Belousov { 15102832cd54SKonstantin Belousov Elf_Sym *sym; 15112832cd54SKonstantin Belousov Elf_Size i; 15122832cd54SKonstantin Belousov 15132832cd54SKonstantin Belousov for (i = 0; i < ef->ddbsymcnt; i++) { 15142832cd54SKonstantin Belousov sym = ef->ddbsymtab + i; 15152832cd54SKonstantin Belousov if (sym->st_shndx == SHN_FBSD_CACHED) { 15162832cd54SKonstantin Belousov sym->st_shndx = SHN_UNDEF; 15172832cd54SKonstantin Belousov sym->st_value = 0; 15182832cd54SKonstantin Belousov } 15192832cd54SKonstantin Belousov } 15202832cd54SKonstantin Belousov } 15212832cd54SKonstantin Belousov 1522d297ad16SMarcel Moolenaar /* 1523d297ad16SMarcel Moolenaar * Symbol lookup function that can be used when the symbol index is known (ie 1524d297ad16SMarcel Moolenaar * in relocations). It uses the symbol index instead of doing a fully fledged 1525d297ad16SMarcel Moolenaar * hash table based lookup when such is valid. For example for local symbols. 1526d297ad16SMarcel Moolenaar * This is not only more efficient, it's also more correct. It's not always 1527d297ad16SMarcel Moolenaar * the case that the symbol can be found through the hash table. 1528d297ad16SMarcel Moolenaar */ 1529cff8c6f2SKonstantin Belousov static int 1530cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res) 1531d297ad16SMarcel Moolenaar { 1532d297ad16SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 15332832cd54SKonstantin Belousov Elf_Sym *sym; 1534d297ad16SMarcel Moolenaar const char *symbol; 1535cff8c6f2SKonstantin Belousov Elf_Addr res1; 1536d297ad16SMarcel Moolenaar 1537d297ad16SMarcel Moolenaar /* Don't even try to lookup the symbol if the index is bogus. */ 1538cff8c6f2SKonstantin Belousov if (symidx >= ef->ddbsymcnt) { 1539cff8c6f2SKonstantin Belousov *res = 0; 1540cff8c6f2SKonstantin Belousov return (EINVAL); 1541cff8c6f2SKonstantin Belousov } 1542d297ad16SMarcel Moolenaar 1543e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 1544d297ad16SMarcel Moolenaar 15454cec6f5dSPeter Wemm /* Quick answer if there is a definition included. */ 1546cff8c6f2SKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) { 1547289ead7cSKonstantin Belousov res1 = (Elf_Addr)sym->st_value; 1548289ead7cSKonstantin Belousov if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) 1549289ead7cSKonstantin Belousov res1 = ((Elf_Addr (*)(void))res1)(); 1550289ead7cSKonstantin Belousov *res = res1; 1551cff8c6f2SKonstantin Belousov return (0); 1552cff8c6f2SKonstantin Belousov } 15534cec6f5dSPeter Wemm 15544cec6f5dSPeter Wemm /* If we get here, then it is undefined and needs a lookup. */ 15554cec6f5dSPeter Wemm switch (ELF_ST_BIND(sym->st_info)) { 15564cec6f5dSPeter Wemm case STB_LOCAL: 15574cec6f5dSPeter Wemm /* Local, but undefined? huh? */ 1558cff8c6f2SKonstantin Belousov *res = 0; 1559cff8c6f2SKonstantin Belousov return (EINVAL); 1560d297ad16SMarcel Moolenaar 1561e9eabf59SPeter Wemm case STB_GLOBAL: 1562cff8c6f2SKonstantin Belousov case STB_WEAK: 1563e9eabf59SPeter Wemm /* Relative to Data or Function name */ 1564e9eabf59SPeter Wemm symbol = ef->ddbstrtab + sym->st_name; 1565d297ad16SMarcel Moolenaar 1566d297ad16SMarcel Moolenaar /* Force a lookup failure if the symbol name is bogus. */ 1567cff8c6f2SKonstantin Belousov if (*symbol == 0) { 1568cff8c6f2SKonstantin Belousov *res = 0; 1569cff8c6f2SKonstantin Belousov return (EINVAL); 1570cff8c6f2SKonstantin Belousov } 1571cff8c6f2SKonstantin Belousov res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps); 15722832cd54SKonstantin Belousov 15732832cd54SKonstantin Belousov /* 15742832cd54SKonstantin Belousov * Cache global lookups during module relocation. The failure 15752832cd54SKonstantin Belousov * case is particularly expensive for callers, who must scan 15762832cd54SKonstantin Belousov * through the entire globals table doing strcmp(). Cache to 15772832cd54SKonstantin Belousov * avoid doing such work repeatedly. 15782832cd54SKonstantin Belousov * 15792832cd54SKonstantin Belousov * After relocation is complete, undefined globals will be 15802832cd54SKonstantin Belousov * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(), 15812832cd54SKonstantin Belousov * above. 15822832cd54SKonstantin Belousov */ 1583cff8c6f2SKonstantin Belousov if (res1 != 0) { 15842832cd54SKonstantin Belousov sym->st_shndx = SHN_FBSD_CACHED; 1585cff8c6f2SKonstantin Belousov sym->st_value = res1; 1586cff8c6f2SKonstantin Belousov *res = res1; 1587e9eabf59SPeter Wemm return (0); 1588cff8c6f2SKonstantin Belousov } else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) { 1589cff8c6f2SKonstantin Belousov sym->st_value = 0; 1590cff8c6f2SKonstantin Belousov *res = 0; 1591cff8c6f2SKonstantin Belousov return (0); 1592cff8c6f2SKonstantin Belousov } 1593cff8c6f2SKonstantin Belousov return (EINVAL); 1594e9eabf59SPeter Wemm 1595e9eabf59SPeter Wemm default: 1596cff8c6f2SKonstantin Belousov return (EINVAL); 1597d297ad16SMarcel Moolenaar } 1598e9eabf59SPeter Wemm } 1599e9eabf59SPeter Wemm 16007251b4bfSJake Burkholder static void 16017226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef) 16027226306eSKonstantin Belousov { 16037226306eSKonstantin Belousov static const char startn[] = "__start_"; 16047226306eSKonstantin Belousov static const char stopn[] = "__stop_"; 16057226306eSKonstantin Belousov Elf_Sym *sym; 16067226306eSKonstantin Belousov const char *sym_name, *linkset_name; 16077226306eSKonstantin Belousov Elf_Addr startp, stopp; 16087226306eSKonstantin Belousov Elf_Size symidx; 16097226306eSKonstantin Belousov int start, i; 16107226306eSKonstantin Belousov 16117226306eSKonstantin Belousov startp = stopp = 0; 16127226306eSKonstantin Belousov for (symidx = 1 /* zero entry is special */; 16137226306eSKonstantin Belousov symidx < ef->ddbsymcnt; symidx++) { 16147226306eSKonstantin Belousov sym = ef->ddbsymtab + symidx; 16157226306eSKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) 16167226306eSKonstantin Belousov continue; 16177226306eSKonstantin Belousov 16187226306eSKonstantin Belousov sym_name = ef->ddbstrtab + sym->st_name; 16197226306eSKonstantin Belousov if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) { 16207226306eSKonstantin Belousov start = 1; 16217226306eSKonstantin Belousov linkset_name = sym_name + sizeof(startn) - 1; 16227226306eSKonstantin Belousov } 16237226306eSKonstantin Belousov else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) { 16247226306eSKonstantin Belousov start = 0; 16257226306eSKonstantin Belousov linkset_name = sym_name + sizeof(stopn) - 1; 16267226306eSKonstantin Belousov } 16277226306eSKonstantin Belousov else 16287226306eSKonstantin Belousov continue; 16297226306eSKonstantin Belousov 16307226306eSKonstantin Belousov for (i = 0; i < ef->nprogtab; i++) { 16317226306eSKonstantin Belousov if (strcmp(ef->progtab[i].name, linkset_name) == 0) { 16327226306eSKonstantin Belousov startp = (Elf_Addr)ef->progtab[i].addr; 16337226306eSKonstantin Belousov stopp = (Elf_Addr)(startp + ef->progtab[i].size); 16347226306eSKonstantin Belousov break; 16357226306eSKonstantin Belousov } 16367226306eSKonstantin Belousov } 16377226306eSKonstantin Belousov if (i == ef->nprogtab) 16387226306eSKonstantin Belousov continue; 16397226306eSKonstantin Belousov 16407226306eSKonstantin Belousov sym->st_value = start ? startp : stopp; 16417226306eSKonstantin Belousov sym->st_shndx = i; 16427226306eSKonstantin Belousov } 16437226306eSKonstantin Belousov } 16447226306eSKonstantin Belousov 16453cfce8e4SKonstantin Belousov static int 1646289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs) 16477251b4bfSJake Burkholder { 1648e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 16497251b4bfSJake Burkholder const Elf_Rel *rellim; 16507251b4bfSJake Burkholder const Elf_Rel *rel; 16517251b4bfSJake Burkholder const Elf_Rela *relalim; 16527251b4bfSJake Burkholder const Elf_Rela *rela; 1653e9eabf59SPeter Wemm const Elf_Sym *sym; 1654e9eabf59SPeter Wemm Elf_Addr base; 1655e9eabf59SPeter Wemm int i; 1656757686b1SMarcel Moolenaar Elf_Size symidx; 1657e9eabf59SPeter Wemm 16587226306eSKonstantin Belousov link_elf_fix_link_set(ef); 16597226306eSKonstantin Belousov 16607251b4bfSJake Burkholder /* Perform relocations without addend if there are any: */ 1661b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1662e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 16633cfce8e4SKonstantin Belousov if (rel == NULL) { 16643cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab"); 16653cfce8e4SKonstantin Belousov return (ENOEXEC); 16663cfce8e4SKonstantin Belousov } 166720947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1668e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 16693cfce8e4SKonstantin Belousov if (base == 0) { 16703cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 16713cfce8e4SKonstantin Belousov return (ENOEXEC); 16723cfce8e4SKonstantin Belousov } 167320947801SPeter Wemm for ( ; rel < rellim; rel++) { 1674e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 167520947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 167620947801SPeter Wemm continue; 1677e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 167820947801SPeter Wemm /* Only do local relocs */ 167920947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 168020947801SPeter Wemm continue; 168136716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 168244c705cfSMitchell Horne elf_is_ifunc_reloc(rel->r_info)) != ifuncs) 168344c705cfSMitchell Horne continue; 168444c705cfSMitchell Horne if (elf_reloc_local(lf, base, rel, ELF_RELOC_REL, 168544c705cfSMitchell Horne elf_obj_lookup) != 0) 168644c705cfSMitchell Horne return (ENOEXEC); 16877251b4bfSJake Burkholder } 1688e9eabf59SPeter Wemm } 16897251b4bfSJake Burkholder 16907251b4bfSJake Burkholder /* Perform relocations with addend if there are any: */ 1691b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1692e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 16933cfce8e4SKonstantin Belousov if (rela == NULL) { 16943cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 16953cfce8e4SKonstantin Belousov return (ENOEXEC); 16963cfce8e4SKonstantin Belousov } 169720947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1698e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 16993cfce8e4SKonstantin Belousov if (base == 0) { 17003cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 17013cfce8e4SKonstantin Belousov return (ENOEXEC); 17023cfce8e4SKonstantin Belousov } 170320947801SPeter Wemm for ( ; rela < relalim; rela++) { 1704e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 170520947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 170620947801SPeter Wemm continue; 1707e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 170820947801SPeter Wemm /* Only do local relocs */ 170920947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 171020947801SPeter Wemm continue; 171136716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 171244c705cfSMitchell Horne elf_is_ifunc_reloc(rela->r_info)) != ifuncs) 171344c705cfSMitchell Horne continue; 171444c705cfSMitchell Horne if (elf_reloc_local(lf, base, rela, ELF_RELOC_RELA, 171544c705cfSMitchell Horne elf_obj_lookup) != 0) 171644c705cfSMitchell Horne return (ENOEXEC); 17177251b4bfSJake Burkholder } 17187251b4bfSJake Burkholder } 17193cfce8e4SKonstantin Belousov return (0); 1720e9eabf59SPeter Wemm } 172100a5db46SStacey Son 172200a5db46SStacey Son static long 1723e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab) 172400a5db46SStacey Son { 172500a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 172600a5db46SStacey Son 172700a5db46SStacey Son *symtab = ef->ddbsymtab; 172800a5db46SStacey Son 172900a5db46SStacey Son if (*symtab == NULL) 173000a5db46SStacey Son return (0); 173100a5db46SStacey Son 173200a5db46SStacey Son return (ef->ddbsymcnt); 173300a5db46SStacey Son } 173400a5db46SStacey Son 173500a5db46SStacey Son static long 173600a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab) 173700a5db46SStacey Son { 173800a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 173900a5db46SStacey Son 174000a5db46SStacey Son *strtab = ef->ddbstrtab; 174100a5db46SStacey Son 174200a5db46SStacey Son if (*strtab == NULL) 174300a5db46SStacey Son return (0); 174400a5db46SStacey Son 174500a5db46SStacey Son return (ef->ddbstrcnt); 174600a5db46SStacey Son } 1747