1a4f67738SDoug Rabson /*- 28a36da99SPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 38a36da99SPedro F. Giffuni * 4326e27d8SDoug Rabson * Copyright (c) 1998-2000 Doug Rabson 5a4f67738SDoug Rabson * All rights reserved. 6a4f67738SDoug Rabson * 7a4f67738SDoug Rabson * Redistribution and use in source and binary forms, with or without 8a4f67738SDoug Rabson * modification, are permitted provided that the following conditions 9a4f67738SDoug Rabson * are met: 10a4f67738SDoug Rabson * 1. Redistributions of source code must retain the above copyright 11a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer. 12a4f67738SDoug Rabson * 2. Redistributions in binary form must reproduce the above copyright 13a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer in the 14a4f67738SDoug Rabson * documentation and/or other materials provided with the distribution. 15a4f67738SDoug Rabson * 16a4f67738SDoug Rabson * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17a4f67738SDoug Rabson * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18a4f67738SDoug Rabson * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19a4f67738SDoug Rabson * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20a4f67738SDoug Rabson * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21a4f67738SDoug Rabson * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22a4f67738SDoug Rabson * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23a4f67738SDoug Rabson * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24a4f67738SDoug Rabson * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25a4f67738SDoug Rabson * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26a4f67738SDoug Rabson * SUCH DAMAGE. 27a4f67738SDoug Rabson */ 28a4f67738SDoug Rabson 29677b542eSDavid E. O'Brien #include <sys/cdefs.h> 30677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$"); 31677b542eSDavid E. O'Brien 32a2024a3eSJohn Birrell #include "opt_ddb.h" 33ecb01c64SMarcel Moolenaar #include "opt_gdb.h" 34326e27d8SDoug Rabson 35a4f67738SDoug Rabson #include <sys/param.h> 36a4f67738SDoug Rabson #include <sys/systm.h> 375ad6c3b1SBruce Evans #ifdef GPROF 385ad6c3b1SBruce Evans #include <sys/gmon.h> 395ad6c3b1SBruce Evans #endif 40fb919e4dSMark Murray #include <sys/kernel.h> 41fb919e4dSMark Murray #include <sys/lock.h> 42a4f67738SDoug Rabson #include <sys/malloc.h> 431aa696baSMark Johnston #ifdef SPARSE_MAPPING 441aa696baSMark Johnston #include <sys/mman.h> 451aa696baSMark Johnston #endif 46c0b824f9SDima Dorfman #include <sys/mutex.h> 4762d615d5SJohn Baldwin #include <sys/mount.h> 4850c202c5SJeff Roberson #include <sys/pcpu.h> 49a4f67738SDoug Rabson #include <sys/proc.h> 50a4f67738SDoug Rabson #include <sys/namei.h> 51a4f67738SDoug Rabson #include <sys/fcntl.h> 52a4f67738SDoug Rabson #include <sys/vnode.h> 53a4f67738SDoug Rabson #include <sys/linker.h> 54eb5a41cfSLeandro Lupori #include <sys/sysctl.h> 55fb919e4dSMark Murray 56a4f67738SDoug Rabson #include <machine/elf.h> 57a4f67738SDoug Rabson 58eddfbb76SRobert Watson #include <net/vnet.h> 59eddfbb76SRobert Watson 60aed55708SRobert Watson #include <security/mac/mac_framework.h> 61aed55708SRobert Watson 62fe3db7c7SDoug Rabson #include <vm/vm.h> 63fe3db7c7SDoug Rabson #include <vm/vm_param.h> 64d1f088daSPeter Wemm #ifdef SPARSE_MAPPING 65fe3db7c7SDoug Rabson #include <vm/vm_object.h> 66fe3db7c7SDoug Rabson #include <vm/vm_kern.h> 67fe3db7c7SDoug Rabson #include <vm/vm_extern.h> 68d1f088daSPeter Wemm #endif 69fe3db7c7SDoug Rabson #include <vm/pmap.h> 70fe3db7c7SDoug Rabson #include <vm/vm_map.h> 71fb919e4dSMark Murray 72f9980387SPeter Wemm #include <sys/link_elf.h> 73fe3db7c7SDoug Rabson 74326e27d8SDoug Rabson #include "linker_if.h" 75a4f67738SDoug Rabson 761a330eb0SPeter Wemm #define MAXSEGS 4 771a330eb0SPeter Wemm 784f2945f8SDavid E. O'Brien typedef struct elf_file { 794f2945f8SDavid E. O'Brien struct linker_file lf; /* Common fields */ 804f2945f8SDavid E. O'Brien int preloaded; /* Was file pre-loaded */ 814f2945f8SDavid E. O'Brien caddr_t address; /* Relocation address */ 824f2945f8SDavid E. O'Brien #ifdef SPARSE_MAPPING 834f2945f8SDavid E. O'Brien vm_object_t object; /* VM object to hold file pages */ 844f2945f8SDavid E. O'Brien #endif 854f2945f8SDavid E. O'Brien Elf_Dyn *dynamic; /* Symbol table etc. */ 864f2945f8SDavid E. O'Brien Elf_Hashelt nbuckets; /* DT_HASH info */ 874f2945f8SDavid E. O'Brien Elf_Hashelt nchains; 884f2945f8SDavid E. O'Brien const Elf_Hashelt *buckets; 894f2945f8SDavid E. O'Brien const Elf_Hashelt *chains; 904f2945f8SDavid E. O'Brien caddr_t hash; 914f2945f8SDavid E. O'Brien caddr_t strtab; /* DT_STRTAB */ 924f2945f8SDavid E. O'Brien int strsz; /* DT_STRSZ */ 934f2945f8SDavid E. O'Brien const Elf_Sym *symtab; /* DT_SYMTAB */ 944f2945f8SDavid E. O'Brien Elf_Addr *got; /* DT_PLTGOT */ 954f2945f8SDavid E. O'Brien const Elf_Rel *pltrel; /* DT_JMPREL */ 964f2945f8SDavid E. O'Brien int pltrelsize; /* DT_PLTRELSZ */ 974f2945f8SDavid E. O'Brien const Elf_Rela *pltrela; /* DT_JMPREL */ 984f2945f8SDavid E. O'Brien int pltrelasize; /* DT_PLTRELSZ */ 994f2945f8SDavid E. O'Brien const Elf_Rel *rel; /* DT_REL */ 1004f2945f8SDavid E. O'Brien int relsize; /* DT_RELSZ */ 1014f2945f8SDavid E. O'Brien const Elf_Rela *rela; /* DT_RELA */ 1024f2945f8SDavid E. O'Brien int relasize; /* DT_RELASZ */ 1034f2945f8SDavid E. O'Brien caddr_t modptr; 1044f2945f8SDavid E. O'Brien const Elf_Sym *ddbsymtab; /* The symbol table we are using */ 1054f2945f8SDavid E. O'Brien long ddbsymcnt; /* Number of symbols */ 1064f2945f8SDavid E. O'Brien caddr_t ddbstrtab; /* String table */ 1074f2945f8SDavid E. O'Brien long ddbstrcnt; /* number of bytes in string table */ 1084f2945f8SDavid E. O'Brien caddr_t symbase; /* malloc'ed symbold base */ 1094f2945f8SDavid E. O'Brien caddr_t strbase; /* malloc'ed string base */ 1104f2945f8SDavid E. O'Brien caddr_t ctftab; /* CTF table */ 1114f2945f8SDavid E. O'Brien long ctfcnt; /* number of bytes in CTF table */ 1124f2945f8SDavid E. O'Brien caddr_t ctfoff; /* CTF offset table */ 1134f2945f8SDavid E. O'Brien caddr_t typoff; /* Type offset table */ 1144f2945f8SDavid E. O'Brien long typlen; /* Number of type entries. */ 11550c202c5SJeff Roberson Elf_Addr pcpu_start; /* Pre-relocation pcpu set start. */ 11650c202c5SJeff Roberson Elf_Addr pcpu_stop; /* Pre-relocation pcpu set stop. */ 11750c202c5SJeff Roberson Elf_Addr pcpu_base; /* Relocated pcpu set address. */ 118eddfbb76SRobert Watson #ifdef VIMAGE 119eddfbb76SRobert Watson Elf_Addr vnet_start; /* Pre-relocation vnet set start. */ 120eddfbb76SRobert Watson Elf_Addr vnet_stop; /* Pre-relocation vnet set stop. */ 121eddfbb76SRobert Watson Elf_Addr vnet_base; /* Relocated vnet set address. */ 122eddfbb76SRobert Watson #endif 1234f2945f8SDavid E. O'Brien #ifdef GDB 1244f2945f8SDavid E. O'Brien struct link_map gdb; /* hooks for gdb */ 1254f2945f8SDavid E. O'Brien #endif 1264f2945f8SDavid E. O'Brien } *elf_file_t; 1274f2945f8SDavid E. O'Brien 12847813f5dSMikolaj Golub struct elf_set { 12947813f5dSMikolaj Golub Elf_Addr es_start; 13047813f5dSMikolaj Golub Elf_Addr es_stop; 13147813f5dSMikolaj Golub Elf_Addr es_base; 13247813f5dSMikolaj Golub TAILQ_ENTRY(elf_set) es_link; 13347813f5dSMikolaj Golub }; 13447813f5dSMikolaj Golub 13547813f5dSMikolaj Golub TAILQ_HEAD(elf_set_head, elf_set); 13647813f5dSMikolaj Golub 137a2024a3eSJohn Birrell #include <kern/kern_ctf.c> 138a2024a3eSJohn Birrell 139c143d6c2SMarcel Moolenaar static int link_elf_link_common_finish(linker_file_t); 14054823af2SPeter Wemm static int link_elf_link_preload(linker_class_t cls, 141326e27d8SDoug Rabson const char *, linker_file_t *); 14254823af2SPeter Wemm static int link_elf_link_preload_finish(linker_file_t); 1432f22b3ffSMatthew D Fleming static int link_elf_load_file(linker_class_t, const char *, 1442f22b3ffSMatthew D Fleming linker_file_t *); 145326e27d8SDoug Rabson static int link_elf_lookup_symbol(linker_file_t, const char *, 146326e27d8SDoug Rabson c_linker_sym_t *); 1472f22b3ffSMatthew D Fleming static int link_elf_symbol_values(linker_file_t, c_linker_sym_t, 1482f22b3ffSMatthew D Fleming linker_symval_t *); 1492f22b3ffSMatthew D Fleming static int link_elf_search_symbol(linker_file_t, caddr_t, 1502f22b3ffSMatthew D Fleming c_linker_sym_t *, long *); 151326e27d8SDoug Rabson 152326e27d8SDoug Rabson static void link_elf_unload_file(linker_file_t); 15354823af2SPeter Wemm static void link_elf_unload_preload(linker_file_t); 154f41325dbSPeter Wemm static int link_elf_lookup_set(linker_file_t, const char *, 155f41325dbSPeter Wemm void ***, void ***, int *); 156bb9fe9ddSBrian Feldman static int link_elf_each_function_name(linker_file_t, 1572f22b3ffSMatthew D Fleming int (*)(const char *, void *), void *); 158a2024a3eSJohn Birrell static int link_elf_each_function_nameval(linker_file_t, 1592f22b3ffSMatthew D Fleming linker_function_nameval_callback_t, void *); 1607251b4bfSJake Burkholder static void link_elf_reloc_local(linker_file_t); 16100a5db46SStacey Son static long link_elf_symtab_get(linker_file_t, const Elf_Sym **); 16200a5db46SStacey Son static long link_elf_strtab_get(linker_file_t, caddr_t *); 163cff8c6f2SKonstantin Belousov static int elf_lookup(linker_file_t, Elf_Size, int, Elf_Addr *); 164326e27d8SDoug Rabson 165326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = { 166326e27d8SDoug Rabson KOBJMETHOD(linker_lookup_symbol, link_elf_lookup_symbol), 167326e27d8SDoug Rabson KOBJMETHOD(linker_symbol_values, link_elf_symbol_values), 168326e27d8SDoug Rabson KOBJMETHOD(linker_search_symbol, link_elf_search_symbol), 169326e27d8SDoug Rabson KOBJMETHOD(linker_unload, link_elf_unload_file), 17054823af2SPeter Wemm KOBJMETHOD(linker_load_file, link_elf_load_file), 17154823af2SPeter Wemm KOBJMETHOD(linker_link_preload, link_elf_link_preload), 17254823af2SPeter Wemm KOBJMETHOD(linker_link_preload_finish, link_elf_link_preload_finish), 173f41325dbSPeter Wemm KOBJMETHOD(linker_lookup_set, link_elf_lookup_set), 174bb9fe9ddSBrian Feldman KOBJMETHOD(linker_each_function_name, link_elf_each_function_name), 175a2024a3eSJohn Birrell KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval), 176a2024a3eSJohn Birrell KOBJMETHOD(linker_ctf_get, link_elf_ctf_get), 17700a5db46SStacey Son KOBJMETHOD(linker_symtab_get, link_elf_symtab_get), 17800a5db46SStacey Son KOBJMETHOD(linker_strtab_get, link_elf_strtab_get), 1796d775f0bSMark Johnston KOBJMETHOD_END 180326e27d8SDoug Rabson }; 181326e27d8SDoug Rabson 182326e27d8SDoug Rabson static struct linker_class link_elf_class = { 183326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32 184326e27d8SDoug Rabson "elf32", 185326e27d8SDoug Rabson #else 186326e27d8SDoug Rabson "elf64", 187326e27d8SDoug Rabson #endif 188326e27d8SDoug Rabson link_elf_methods, sizeof(struct elf_file) 189326e27d8SDoug Rabson }; 190326e27d8SDoug Rabson 19136716fe2SMark Johnston typedef int (*elf_reloc_fn)(linker_file_t lf, Elf_Addr relocbase, 19236716fe2SMark Johnston const void *data, int type, elf_lookup_fn lookup); 19336716fe2SMark Johnston 1942f22b3ffSMatthew D Fleming static int parse_dynamic(elf_file_t); 1952f22b3ffSMatthew D Fleming static int relocate_file(elf_file_t); 19636716fe2SMark Johnston static int relocate_file1(elf_file_t ef, elf_lookup_fn lookup, 19736716fe2SMark Johnston elf_reloc_fn reloc, bool ifuncs); 1982f22b3ffSMatthew D Fleming static int link_elf_preload_parse_symbols(elf_file_t); 199326e27d8SDoug Rabson 20047813f5dSMikolaj Golub static struct elf_set_head set_pcpu_list; 20147813f5dSMikolaj Golub #ifdef VIMAGE 20247813f5dSMikolaj Golub static struct elf_set_head set_vnet_list; 20347813f5dSMikolaj Golub #endif 20447813f5dSMikolaj Golub 20547813f5dSMikolaj Golub static void 20647813f5dSMikolaj Golub elf_set_add(struct elf_set_head *list, Elf_Addr start, Elf_Addr stop, Elf_Addr base) 20747813f5dSMikolaj Golub { 20847813f5dSMikolaj Golub struct elf_set *set, *iter; 20947813f5dSMikolaj Golub 21047813f5dSMikolaj Golub set = malloc(sizeof(*set), M_LINKER, M_WAITOK); 21147813f5dSMikolaj Golub set->es_start = start; 21247813f5dSMikolaj Golub set->es_stop = stop; 21347813f5dSMikolaj Golub set->es_base = base; 21447813f5dSMikolaj Golub 21547813f5dSMikolaj Golub TAILQ_FOREACH(iter, list, es_link) { 21647813f5dSMikolaj Golub KASSERT((set->es_start < iter->es_start && set->es_stop < iter->es_stop) || 21747813f5dSMikolaj Golub (set->es_start > iter->es_start && set->es_stop > iter->es_stop), 21847813f5dSMikolaj Golub ("linker sets intersection: to insert: 0x%jx-0x%jx; inserted: 0x%jx-0x%jx", 21947813f5dSMikolaj Golub (uintmax_t)set->es_start, (uintmax_t)set->es_stop, 22047813f5dSMikolaj Golub (uintmax_t)iter->es_start, (uintmax_t)iter->es_stop)); 22147813f5dSMikolaj Golub 22247813f5dSMikolaj Golub if (iter->es_start > set->es_start) { 22347813f5dSMikolaj Golub TAILQ_INSERT_BEFORE(iter, set, es_link); 22447813f5dSMikolaj Golub break; 22547813f5dSMikolaj Golub } 22647813f5dSMikolaj Golub } 22747813f5dSMikolaj Golub 22847813f5dSMikolaj Golub if (iter == NULL) 22947813f5dSMikolaj Golub TAILQ_INSERT_TAIL(list, set, es_link); 23047813f5dSMikolaj Golub } 23147813f5dSMikolaj Golub 23247813f5dSMikolaj Golub static int 23347813f5dSMikolaj Golub elf_set_find(struct elf_set_head *list, Elf_Addr addr, Elf_Addr *start, Elf_Addr *base) 23447813f5dSMikolaj Golub { 23547813f5dSMikolaj Golub struct elf_set *set; 23647813f5dSMikolaj Golub 23747813f5dSMikolaj Golub TAILQ_FOREACH(set, list, es_link) { 23847813f5dSMikolaj Golub if (addr < set->es_start) 23947813f5dSMikolaj Golub return (0); 24047813f5dSMikolaj Golub if (addr < set->es_stop) { 24147813f5dSMikolaj Golub *start = set->es_start; 24247813f5dSMikolaj Golub *base = set->es_base; 24347813f5dSMikolaj Golub return (1); 24447813f5dSMikolaj Golub } 24547813f5dSMikolaj Golub } 24647813f5dSMikolaj Golub 24747813f5dSMikolaj Golub return (0); 24847813f5dSMikolaj Golub } 24947813f5dSMikolaj Golub 25047813f5dSMikolaj Golub static void 25147813f5dSMikolaj Golub elf_set_delete(struct elf_set_head *list, Elf_Addr start) 25247813f5dSMikolaj Golub { 25347813f5dSMikolaj Golub struct elf_set *set; 25447813f5dSMikolaj Golub 25547813f5dSMikolaj Golub TAILQ_FOREACH(set, list, es_link) { 25647813f5dSMikolaj Golub if (start < set->es_start) 25747813f5dSMikolaj Golub break; 25847813f5dSMikolaj Golub if (start == set->es_start) { 25947813f5dSMikolaj Golub TAILQ_REMOVE(list, set, es_link); 26047813f5dSMikolaj Golub free(set, M_LINKER); 26147813f5dSMikolaj Golub return; 26247813f5dSMikolaj Golub } 26347813f5dSMikolaj Golub } 26447813f5dSMikolaj Golub KASSERT(0, ("deleting unknown linker set (start = 0x%jx)", 26547813f5dSMikolaj Golub (uintmax_t)start)); 26647813f5dSMikolaj Golub } 26747813f5dSMikolaj Golub 268ecb01c64SMarcel Moolenaar #ifdef GDB 2692f22b3ffSMatthew D Fleming static void r_debug_state(struct r_debug *, struct link_map *); 270326e27d8SDoug Rabson 271326e27d8SDoug Rabson /* 272326e27d8SDoug Rabson * A list of loaded modules for GDB to use for loading symbols. 273326e27d8SDoug Rabson */ 274326e27d8SDoug Rabson struct r_debug r_debug; 275326e27d8SDoug Rabson 2762f22b3ffSMatthew D Fleming #define GDB_STATE(s) do { \ 2772f22b3ffSMatthew D Fleming r_debug.r_state = s; r_debug_state(NULL, NULL); \ 2782f22b3ffSMatthew D Fleming } while (0) 279326e27d8SDoug Rabson 280326e27d8SDoug Rabson /* 281326e27d8SDoug Rabson * Function for the debugger to set a breakpoint on to gain control. 282326e27d8SDoug Rabson */ 28337c84183SPoul-Henning Kamp static void 284ca29467eSJohn Baldwin r_debug_state(struct r_debug *dummy_one __unused, 285ca29467eSJohn Baldwin struct link_map *dummy_two __unused) 286326e27d8SDoug Rabson { 287326e27d8SDoug Rabson } 288326e27d8SDoug Rabson 2891720979bSMarcel Moolenaar static void 2901720979bSMarcel Moolenaar link_elf_add_gdb(struct link_map *l) 2911720979bSMarcel Moolenaar { 2921720979bSMarcel Moolenaar struct link_map *prev; 2931720979bSMarcel Moolenaar 294c143d6c2SMarcel Moolenaar l->l_next = NULL; 295c143d6c2SMarcel Moolenaar 296c143d6c2SMarcel Moolenaar if (r_debug.r_map == NULL) { 297c143d6c2SMarcel Moolenaar /* Add first. */ 298c143d6c2SMarcel Moolenaar l->l_prev = NULL; 299c143d6c2SMarcel Moolenaar r_debug.r_map = l; 300c143d6c2SMarcel Moolenaar } else { 301c143d6c2SMarcel Moolenaar /* Append to list. */ 3022f22b3ffSMatthew D Fleming for (prev = r_debug.r_map; 3032f22b3ffSMatthew D Fleming prev->l_next != NULL; 3042f22b3ffSMatthew D Fleming prev = prev->l_next) 3051720979bSMarcel Moolenaar ; 3061720979bSMarcel Moolenaar l->l_prev = prev; 3071720979bSMarcel Moolenaar prev->l_next = l; 3081720979bSMarcel Moolenaar } 309c143d6c2SMarcel Moolenaar } 3101720979bSMarcel Moolenaar 3111720979bSMarcel Moolenaar static void 3121720979bSMarcel Moolenaar link_elf_delete_gdb(struct link_map *l) 3131720979bSMarcel Moolenaar { 3141720979bSMarcel Moolenaar if (l->l_prev == NULL) { 315c143d6c2SMarcel Moolenaar /* Remove first. */ 3161720979bSMarcel Moolenaar if ((r_debug.r_map = l->l_next) != NULL) 3171720979bSMarcel Moolenaar l->l_next->l_prev = NULL; 318c143d6c2SMarcel Moolenaar } else { 319c143d6c2SMarcel Moolenaar /* Remove any but first. */ 3201720979bSMarcel Moolenaar if ((l->l_prev->l_next = l->l_next) != NULL) 3211720979bSMarcel Moolenaar l->l_next->l_prev = l->l_prev; 3221720979bSMarcel Moolenaar } 323c143d6c2SMarcel Moolenaar } 324ecb01c64SMarcel Moolenaar #endif /* GDB */ 325a4f67738SDoug Rabson 3264f2945f8SDavid E. O'Brien /* 3274f2945f8SDavid E. O'Brien * The kernel symbol table starts here. 3284f2945f8SDavid E. O'Brien */ 3294f2945f8SDavid E. O'Brien extern struct _dynamic _DYNAMIC; 3304f2945f8SDavid E. O'Brien 331a4f67738SDoug Rabson static void 332552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s) 3331720979bSMarcel Moolenaar { 334552f9f63SEdwin Groothuis if (filename == NULL) 3351720979bSMarcel Moolenaar printf("kldload: %s\n", s); 336552f9f63SEdwin Groothuis else 337552f9f63SEdwin Groothuis printf("kldload: %s: %s\n", filename, s); 3381720979bSMarcel Moolenaar } 3391720979bSMarcel Moolenaar 3400067051fSMarcel Moolenaar static void 3410067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size) 3420067051fSMarcel Moolenaar { 3430067051fSMarcel Moolenaar void (**ctor)(void); 3440067051fSMarcel Moolenaar size_t i, cnt; 3450067051fSMarcel Moolenaar 3460067051fSMarcel Moolenaar if (addr == NULL || size == 0) 3470067051fSMarcel Moolenaar return; 3480067051fSMarcel Moolenaar cnt = size / sizeof(*ctor); 3490067051fSMarcel Moolenaar ctor = (void *)addr; 3500067051fSMarcel Moolenaar for (i = 0; i < cnt; i++) { 3510067051fSMarcel Moolenaar if (ctor[i] != NULL) 3520067051fSMarcel Moolenaar (*ctor[i])(); 3530067051fSMarcel Moolenaar } 3540067051fSMarcel Moolenaar } 3550067051fSMarcel Moolenaar 356c143d6c2SMarcel Moolenaar /* 357c143d6c2SMarcel Moolenaar * Actions performed after linking/loading both the preloaded kernel and any 358c143d6c2SMarcel Moolenaar * modules; whether preloaded or dynamicly loaded. 359c143d6c2SMarcel Moolenaar */ 360c143d6c2SMarcel Moolenaar static int 361c143d6c2SMarcel Moolenaar link_elf_link_common_finish(linker_file_t lf) 362c143d6c2SMarcel Moolenaar { 363ecb01c64SMarcel Moolenaar #ifdef GDB 364c143d6c2SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 365c143d6c2SMarcel Moolenaar char *newfilename; 366c143d6c2SMarcel Moolenaar #endif 3671aeb23cdSMarcel Moolenaar int error; 3681aeb23cdSMarcel Moolenaar 3691aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being loaded. */ 3701aeb23cdSMarcel Moolenaar error = elf_cpu_load_file(lf); 3712f22b3ffSMatthew D Fleming if (error != 0) 3721aeb23cdSMarcel Moolenaar return (error); 373c143d6c2SMarcel Moolenaar 374ecb01c64SMarcel Moolenaar #ifdef GDB 375c143d6c2SMarcel Moolenaar GDB_STATE(RT_ADD); 376c143d6c2SMarcel Moolenaar ef->gdb.l_addr = lf->address; 377a163d034SWarner Losh newfilename = malloc(strlen(lf->filename) + 1, M_LINKER, M_WAITOK); 378c143d6c2SMarcel Moolenaar strcpy(newfilename, lf->filename); 379c143d6c2SMarcel Moolenaar ef->gdb.l_name = newfilename; 380c143d6c2SMarcel Moolenaar ef->gdb.l_ld = ef->dynamic; 381c143d6c2SMarcel Moolenaar link_elf_add_gdb(&ef->gdb); 382c143d6c2SMarcel Moolenaar GDB_STATE(RT_CONSISTENT); 383c143d6c2SMarcel Moolenaar #endif 384c143d6c2SMarcel Moolenaar 3850067051fSMarcel Moolenaar /* Invoke .ctors */ 3860067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 387c143d6c2SMarcel Moolenaar return (0); 388c143d6c2SMarcel Moolenaar } 389c143d6c2SMarcel Moolenaar 3909411e24dSBrandon Bergren #ifdef RELOCATABLE_KERNEL 39140b664f6SBrandon Bergren /* 39240b664f6SBrandon Bergren * __startkernel and __endkernel are symbols set up as relocation canaries. 39340b664f6SBrandon Bergren * 39440b664f6SBrandon Bergren * They are defined in locore to reference linker script symbols at the 39540b664f6SBrandon Bergren * beginning and end of the LOAD area. This has the desired side effect of 39640b664f6SBrandon Bergren * giving us variables that have relative relocations pointing at them, so 39740b664f6SBrandon Bergren * relocation of the kernel object will cause the variables to be updated 39840b664f6SBrandon Bergren * automatically by the runtime linker when we initialize. 39940b664f6SBrandon Bergren * 40040b664f6SBrandon Bergren * There are two main reasons to relocate the kernel: 40140b664f6SBrandon Bergren * 1) If the loader needed to load the kernel at an alternate load address. 40240b664f6SBrandon Bergren * 2) If the kernel is switching address spaces on machines like POWER9 40340b664f6SBrandon Bergren * under Radix where the high bits of the effective address are used to 40440b664f6SBrandon Bergren * differentiate between hypervisor, host, guest, and problem state. 40540b664f6SBrandon Bergren */ 406ee900504SNathan Whitehorn extern vm_offset_t __startkernel, __endkernel; 4079411e24dSBrandon Bergren #endif 408c6014c73SNathan Whitehorn 409eb5a41cfSLeandro Lupori static unsigned long kern_relbase = KERNBASE; 410eb5a41cfSLeandro Lupori 411eb5a41cfSLeandro Lupori SYSCTL_ULONG(_kern, OID_AUTO, base_address, CTLFLAG_RD, 412eb5a41cfSLeandro Lupori SYSCTL_NULL_ULONG_PTR, KERNBASE, "Kernel base address"); 413eb5a41cfSLeandro Lupori SYSCTL_ULONG(_kern, OID_AUTO, relbase_address, CTLFLAG_RD, 414eb5a41cfSLeandro Lupori &kern_relbase, 0, "Kernel relocated base address"); 415eb5a41cfSLeandro Lupori 4161720979bSMarcel Moolenaar static void 417a4f67738SDoug Rabson link_elf_init(void* arg) 418a4f67738SDoug Rabson { 419de78ca7eSPeter Wemm Elf_Dyn *dp; 4200067051fSMarcel Moolenaar Elf_Addr *ctors_addrp; 4210067051fSMarcel Moolenaar Elf_Size *ctors_sizep; 422de78ca7eSPeter Wemm caddr_t modptr, baseptr, sizeptr; 423de78ca7eSPeter Wemm elf_file_t ef; 424fe20aaecSRyan Libby const char *modname; 425a4f67738SDoug Rabson 426326e27d8SDoug Rabson linker_add_class(&link_elf_class); 427a4f67738SDoug Rabson 428de78ca7eSPeter Wemm dp = (Elf_Dyn *)&_DYNAMIC; 429de78ca7eSPeter Wemm modname = NULL; 43067db8b23SPeter Wemm modptr = preload_search_by_type("elf" __XSTRING(__ELF_WORD_SIZE) " kernel"); 43167db8b23SPeter Wemm if (modptr == NULL) 432de78ca7eSPeter Wemm modptr = preload_search_by_type("elf kernel"); 433de78ca7eSPeter Wemm modname = (char *)preload_search_info(modptr, MODINFO_NAME); 434de78ca7eSPeter Wemm if (modname == NULL) 435de78ca7eSPeter Wemm modname = "kernel"; 436326e27d8SDoug Rabson linker_kernel_file = linker_make_file(modname, &link_elf_class); 437a4f67738SDoug Rabson if (linker_kernel_file == NULL) 4382f22b3ffSMatthew D Fleming panic("%s: Can't create linker structures for kernel", 4392f22b3ffSMatthew D Fleming __func__); 440326e27d8SDoug Rabson 441326e27d8SDoug Rabson ef = (elf_file_t) linker_kernel_file; 44254823af2SPeter Wemm ef->preloaded = 1; 4439411e24dSBrandon Bergren #ifdef RELOCATABLE_KERNEL 44440b664f6SBrandon Bergren /* Compute relative displacement */ 445c6014c73SNathan Whitehorn ef->address = (caddr_t) (__startkernel - KERNBASE); 446c6014c73SNathan Whitehorn #else 447326e27d8SDoug Rabson ef->address = 0; 448c6014c73SNathan Whitehorn #endif 449326e27d8SDoug Rabson #ifdef SPARSE_MAPPING 4501aa696baSMark Johnston ef->object = NULL; 451326e27d8SDoug Rabson #endif 452326e27d8SDoug Rabson ef->dynamic = dp; 453326e27d8SDoug Rabson 4542f22b3ffSMatthew D Fleming if (dp != NULL) 455326e27d8SDoug Rabson parse_dynamic(ef); 4569411e24dSBrandon Bergren #ifdef RELOCATABLE_KERNEL 457ee900504SNathan Whitehorn linker_kernel_file->address = (caddr_t)__startkernel; 458ee900504SNathan Whitehorn linker_kernel_file->size = (intptr_t)(__endkernel - __startkernel); 459eb5a41cfSLeandro Lupori kern_relbase = (unsigned long)__startkernel; 460ee900504SNathan Whitehorn #else 461c6014c73SNathan Whitehorn linker_kernel_file->address += KERNBASE; 462d009ccfaSBruce Evans linker_kernel_file->size = -(intptr_t)linker_kernel_file->address; 463ee900504SNathan Whitehorn #endif 464de78ca7eSPeter Wemm 4652f22b3ffSMatthew D Fleming if (modptr != NULL) { 4662d636ab0SPeter Wemm ef->modptr = modptr; 467de78ca7eSPeter Wemm baseptr = preload_search_info(modptr, MODINFO_ADDR); 4682f22b3ffSMatthew D Fleming if (baseptr != NULL) 469de78ca7eSPeter Wemm linker_kernel_file->address = *(caddr_t *)baseptr; 470de78ca7eSPeter Wemm sizeptr = preload_search_info(modptr, MODINFO_SIZE); 4712f22b3ffSMatthew D Fleming if (sizeptr != NULL) 472de78ca7eSPeter Wemm linker_kernel_file->size = *(size_t *)sizeptr; 4730067051fSMarcel Moolenaar ctors_addrp = (Elf_Addr *)preload_search_info(modptr, 4740067051fSMarcel Moolenaar MODINFO_METADATA | MODINFOMD_CTORS_ADDR); 4750067051fSMarcel Moolenaar ctors_sizep = (Elf_Size *)preload_search_info(modptr, 4760067051fSMarcel Moolenaar MODINFO_METADATA | MODINFOMD_CTORS_SIZE); 4770067051fSMarcel Moolenaar if (ctors_addrp != NULL && ctors_sizep != NULL) { 478853a17adSEnji Cooper linker_kernel_file->ctors_addr = ef->address + 4790067051fSMarcel Moolenaar *ctors_addrp; 4800067051fSMarcel Moolenaar linker_kernel_file->ctors_size = *ctors_sizep; 4810067051fSMarcel Moolenaar } 482de78ca7eSPeter Wemm } 48354823af2SPeter Wemm (void)link_elf_preload_parse_symbols(ef); 484326e27d8SDoug Rabson 485ecb01c64SMarcel Moolenaar #ifdef GDB 486c143d6c2SMarcel Moolenaar r_debug.r_map = NULL; 487326e27d8SDoug Rabson r_debug.r_brk = r_debug_state; 488326e27d8SDoug Rabson r_debug.r_state = RT_CONSISTENT; 489326e27d8SDoug Rabson #endif 490c143d6c2SMarcel Moolenaar 491c143d6c2SMarcel Moolenaar (void)link_elf_link_common_finish(linker_kernel_file); 49259800afcSJohn Baldwin linker_kernel_file->flags |= LINKER_FILE_LINKED; 49347813f5dSMikolaj Golub TAILQ_INIT(&set_pcpu_list); 49447813f5dSMikolaj Golub #ifdef VIMAGE 49547813f5dSMikolaj Golub TAILQ_INIT(&set_vnet_list); 49647813f5dSMikolaj Golub #endif 497a4f67738SDoug Rabson } 498a4f67738SDoug Rabson 499891cf3edSEd Maste SYSINIT(link_elf, SI_SUB_KLD, SI_ORDER_THIRD, link_elf_init, NULL); 500a4f67738SDoug Rabson 501a4f67738SDoug Rabson static int 50254823af2SPeter Wemm link_elf_preload_parse_symbols(elf_file_t ef) 5032d636ab0SPeter Wemm { 5042d636ab0SPeter Wemm caddr_t pointer; 5052d636ab0SPeter Wemm caddr_t ssym, esym, base; 5062d636ab0SPeter Wemm caddr_t strtab; 5072d636ab0SPeter Wemm int strcnt; 5082d636ab0SPeter Wemm Elf_Sym *symtab; 5092d636ab0SPeter Wemm int symcnt; 5102d636ab0SPeter Wemm 511ca65d5c7SPeter Wemm if (ef->modptr == NULL) 5122f22b3ffSMatthew D Fleming return (0); 5132f22b3ffSMatthew D Fleming pointer = preload_search_info(ef->modptr, 5142f22b3ffSMatthew D Fleming MODINFO_METADATA | MODINFOMD_SSYM); 5152d636ab0SPeter Wemm if (pointer == NULL) 5162f22b3ffSMatthew D Fleming return (0); 5172d636ab0SPeter Wemm ssym = *(caddr_t *)pointer; 5182f22b3ffSMatthew D Fleming pointer = preload_search_info(ef->modptr, 5192f22b3ffSMatthew D Fleming MODINFO_METADATA | MODINFOMD_ESYM); 5202d636ab0SPeter Wemm if (pointer == NULL) 5212f22b3ffSMatthew D Fleming return (0); 5222d636ab0SPeter Wemm esym = *(caddr_t *)pointer; 5232d636ab0SPeter Wemm 5242d636ab0SPeter Wemm base = ssym; 5252d636ab0SPeter Wemm 5262d636ab0SPeter Wemm symcnt = *(long *)base; 5272d636ab0SPeter Wemm base += sizeof(long); 5282d636ab0SPeter Wemm symtab = (Elf_Sym *)base; 5292d636ab0SPeter Wemm base += roundup(symcnt, sizeof(long)); 5302d636ab0SPeter Wemm 5312d636ab0SPeter Wemm if (base > esym || base < ssym) { 5322d636ab0SPeter Wemm printf("Symbols are corrupt!\n"); 5332f22b3ffSMatthew D Fleming return (EINVAL); 5342d636ab0SPeter Wemm } 5352d636ab0SPeter Wemm 5362d636ab0SPeter Wemm strcnt = *(long *)base; 5372d636ab0SPeter Wemm base += sizeof(long); 5382d636ab0SPeter Wemm strtab = base; 5392d636ab0SPeter Wemm base += roundup(strcnt, sizeof(long)); 5402d636ab0SPeter Wemm 5412d636ab0SPeter Wemm if (base > esym || base < ssym) { 5422d636ab0SPeter Wemm printf("Symbols are corrupt!\n"); 5432f22b3ffSMatthew D Fleming return (EINVAL); 5442d636ab0SPeter Wemm } 5452d636ab0SPeter Wemm 5462d636ab0SPeter Wemm ef->ddbsymtab = symtab; 5472d636ab0SPeter Wemm ef->ddbsymcnt = symcnt / sizeof(Elf_Sym); 5482d636ab0SPeter Wemm ef->ddbstrtab = strtab; 5492d636ab0SPeter Wemm ef->ddbstrcnt = strcnt; 5502d636ab0SPeter Wemm 5512f22b3ffSMatthew D Fleming return (0); 5522d636ab0SPeter Wemm } 5532d636ab0SPeter Wemm 5542d636ab0SPeter Wemm static int 555326e27d8SDoug Rabson parse_dynamic(elf_file_t ef) 556a4f67738SDoug Rabson { 557326e27d8SDoug Rabson Elf_Dyn *dp; 558fe3db7c7SDoug Rabson int plttype = DT_REL; 559a4f67738SDoug Rabson 560a4f67738SDoug Rabson for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) { 561a4f67738SDoug Rabson switch (dp->d_tag) { 562a4f67738SDoug Rabson case DT_HASH: 563a4f67738SDoug Rabson { 564a4f67738SDoug Rabson /* From src/libexec/rtld-elf/rtld.c */ 5657e510298SJake Burkholder const Elf_Hashelt *hashtab = (const Elf_Hashelt *) 566a4f67738SDoug Rabson (ef->address + dp->d_un.d_ptr); 567a4f67738SDoug Rabson ef->nbuckets = hashtab[0]; 568a4f67738SDoug Rabson ef->nchains = hashtab[1]; 569a4f67738SDoug Rabson ef->buckets = hashtab + 2; 570a4f67738SDoug Rabson ef->chains = ef->buckets + ef->nbuckets; 571a4f67738SDoug Rabson break; 572a4f67738SDoug Rabson } 573a4f67738SDoug Rabson case DT_STRTAB: 574fe3db7c7SDoug Rabson ef->strtab = (caddr_t) (ef->address + dp->d_un.d_ptr); 575a4f67738SDoug Rabson break; 5762d636ab0SPeter Wemm case DT_STRSZ: 5772d636ab0SPeter Wemm ef->strsz = dp->d_un.d_val; 5782d636ab0SPeter Wemm break; 579a4f67738SDoug Rabson case DT_SYMTAB: 580fe3db7c7SDoug Rabson ef->symtab = (Elf_Sym*) (ef->address + dp->d_un.d_ptr); 581a4f67738SDoug Rabson break; 582a4f67738SDoug Rabson case DT_SYMENT: 583a4f67738SDoug Rabson if (dp->d_un.d_val != sizeof(Elf_Sym)) 5842f22b3ffSMatthew D Fleming return (ENOEXEC); 585fe3db7c7SDoug Rabson break; 586fe3db7c7SDoug Rabson case DT_PLTGOT: 587fe3db7c7SDoug Rabson ef->got = (Elf_Addr *) (ef->address + dp->d_un.d_ptr); 588fe3db7c7SDoug Rabson break; 589fe3db7c7SDoug Rabson case DT_REL: 590fe3db7c7SDoug Rabson ef->rel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr); 591fe3db7c7SDoug Rabson break; 592fe3db7c7SDoug Rabson case DT_RELSZ: 593fe3db7c7SDoug Rabson ef->relsize = dp->d_un.d_val; 594fe3db7c7SDoug Rabson break; 595fe3db7c7SDoug Rabson case DT_RELENT: 596fe3db7c7SDoug Rabson if (dp->d_un.d_val != sizeof(Elf_Rel)) 5972f22b3ffSMatthew D Fleming return (ENOEXEC); 598fe3db7c7SDoug Rabson break; 599fe3db7c7SDoug Rabson case DT_JMPREL: 600fe3db7c7SDoug Rabson ef->pltrel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr); 601fe3db7c7SDoug Rabson break; 602fe3db7c7SDoug Rabson case DT_PLTRELSZ: 603fe3db7c7SDoug Rabson ef->pltrelsize = dp->d_un.d_val; 604fe3db7c7SDoug Rabson break; 605fe3db7c7SDoug Rabson case DT_RELA: 606fe3db7c7SDoug Rabson ef->rela = (const Elf_Rela *) (ef->address + dp->d_un.d_ptr); 607fe3db7c7SDoug Rabson break; 608fe3db7c7SDoug Rabson case DT_RELASZ: 609fe3db7c7SDoug Rabson ef->relasize = dp->d_un.d_val; 610fe3db7c7SDoug Rabson break; 611fe3db7c7SDoug Rabson case DT_RELAENT: 612fe3db7c7SDoug Rabson if (dp->d_un.d_val != sizeof(Elf_Rela)) 6132f22b3ffSMatthew D Fleming return (ENOEXEC); 614fe3db7c7SDoug Rabson break; 615fe3db7c7SDoug Rabson case DT_PLTREL: 616fe3db7c7SDoug Rabson plttype = dp->d_un.d_val; 617fe3db7c7SDoug Rabson if (plttype != DT_REL && plttype != DT_RELA) 6182f22b3ffSMatthew D Fleming return (ENOEXEC); 619fe3db7c7SDoug Rabson break; 620ecb01c64SMarcel Moolenaar #ifdef GDB 621326e27d8SDoug Rabson case DT_DEBUG: 622326e27d8SDoug Rabson dp->d_un.d_ptr = (Elf_Addr)&r_debug; 623326e27d8SDoug Rabson break; 624326e27d8SDoug Rabson #endif 625a4f67738SDoug Rabson } 626a4f67738SDoug Rabson } 627fe3db7c7SDoug Rabson 628fe3db7c7SDoug Rabson if (plttype == DT_RELA) { 629fe3db7c7SDoug Rabson ef->pltrela = (const Elf_Rela *)ef->pltrel; 630fe3db7c7SDoug Rabson ef->pltrel = NULL; 631fe3db7c7SDoug Rabson ef->pltrelasize = ef->pltrelsize; 632fe3db7c7SDoug Rabson ef->pltrelsize = 0; 633fe3db7c7SDoug Rabson } 634fe3db7c7SDoug Rabson 6352d636ab0SPeter Wemm ef->ddbsymtab = ef->symtab; 6362d636ab0SPeter Wemm ef->ddbsymcnt = ef->nchains; 6372d636ab0SPeter Wemm ef->ddbstrtab = ef->strtab; 6382d636ab0SPeter Wemm ef->ddbstrcnt = ef->strsz; 6392d636ab0SPeter Wemm 64030693808SBrandon Bergren return elf_cpu_parse_dynamic(ef->address, ef->dynamic); 641a4f67738SDoug Rabson } 642a4f67738SDoug Rabson 6434c62bffeSBjoern A. Zeeb #define LS_PADDING 0x90909090 644a4f67738SDoug Rabson static int 64550c202c5SJeff Roberson parse_dpcpu(elf_file_t ef) 64650c202c5SJeff Roberson { 6479ae7bc39SBjoern A. Zeeb int error, size; 6484c62bffeSBjoern A. Zeeb #if defined(__i386__) 6494c62bffeSBjoern A. Zeeb uint32_t pad; 6504c62bffeSBjoern A. Zeeb #endif 65150c202c5SJeff Roberson 65250c202c5SJeff Roberson ef->pcpu_start = 0; 65350c202c5SJeff Roberson ef->pcpu_stop = 0; 65450c202c5SJeff Roberson error = link_elf_lookup_set(&ef->lf, "pcpu", (void ***)&ef->pcpu_start, 6559ae7bc39SBjoern A. Zeeb (void ***)&ef->pcpu_stop, NULL); 65650c202c5SJeff Roberson /* Error just means there is no pcpu set to relocate. */ 6572f22b3ffSMatthew D Fleming if (error != 0) 65850c202c5SJeff Roberson return (0); 6599ae7bc39SBjoern A. Zeeb size = (uintptr_t)ef->pcpu_stop - (uintptr_t)ef->pcpu_start; 6609ae7bc39SBjoern A. Zeeb /* Empty set? */ 6619ae7bc39SBjoern A. Zeeb if (size < 1) 6629ae7bc39SBjoern A. Zeeb return (0); 6634c62bffeSBjoern A. Zeeb #if defined(__i386__) 6644c62bffeSBjoern A. Zeeb /* In case we do find __start/stop_set_ symbols double-check. */ 6654c62bffeSBjoern A. Zeeb if (size < 4) { 6664c62bffeSBjoern A. Zeeb uprintf("Kernel module '%s' must be recompiled with " 6674c62bffeSBjoern A. Zeeb "linker script\n", ef->lf.pathname); 6684c62bffeSBjoern A. Zeeb return (ENOEXEC); 6694c62bffeSBjoern A. Zeeb } 6704c62bffeSBjoern A. Zeeb 6714c62bffeSBjoern A. Zeeb /* Padding from linker-script correct? */ 6724c62bffeSBjoern A. Zeeb pad = *(uint32_t *)((uintptr_t)ef->pcpu_stop - sizeof(pad)); 6734c62bffeSBjoern A. Zeeb if (pad != LS_PADDING) { 6744c62bffeSBjoern A. Zeeb uprintf("Kernel module '%s' must be recompiled with " 6754c62bffeSBjoern A. Zeeb "linker script, invalid padding %#04x (%#04x)\n", 6764c62bffeSBjoern A. Zeeb ef->lf.pathname, pad, LS_PADDING); 6774c62bffeSBjoern A. Zeeb return (ENOEXEC); 6784c62bffeSBjoern A. Zeeb } 6794c62bffeSBjoern A. Zeeb /* If we only have valid padding, nothing to do. */ 6804c62bffeSBjoern A. Zeeb if (size == 4) 6814c62bffeSBjoern A. Zeeb return (0); 6824c62bffeSBjoern A. Zeeb #endif 68350c202c5SJeff Roberson /* 6842f22b3ffSMatthew D Fleming * Allocate space in the primary pcpu area. Copy in our 6852f22b3ffSMatthew D Fleming * initialization from the data section and then initialize 6862f22b3ffSMatthew D Fleming * all per-cpu storage from that. 68750c202c5SJeff Roberson */ 6889ae7bc39SBjoern A. Zeeb ef->pcpu_base = (Elf_Addr)(uintptr_t)dpcpu_alloc(size); 6890f823b64SBjoern A. Zeeb if (ef->pcpu_base == 0) { 6900f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out of space; " 6910f823b64SBjoern A. Zeeb "cannot allocate %d for %s\n", 6920f823b64SBjoern A. Zeeb __func__, size, ef->lf.pathname); 69350c202c5SJeff Roberson return (ENOSPC); 6940f823b64SBjoern A. Zeeb } 6959ae7bc39SBjoern A. Zeeb memcpy((void *)ef->pcpu_base, (void *)ef->pcpu_start, size); 6969ae7bc39SBjoern A. Zeeb dpcpu_copy((void *)ef->pcpu_base, size); 69747813f5dSMikolaj Golub elf_set_add(&set_pcpu_list, ef->pcpu_start, ef->pcpu_stop, 69847813f5dSMikolaj Golub ef->pcpu_base); 69950c202c5SJeff Roberson 70050c202c5SJeff Roberson return (0); 70150c202c5SJeff Roberson } 70250c202c5SJeff Roberson 703eddfbb76SRobert Watson #ifdef VIMAGE 704eddfbb76SRobert Watson static int 705eddfbb76SRobert Watson parse_vnet(elf_file_t ef) 706eddfbb76SRobert Watson { 7079ae7bc39SBjoern A. Zeeb int error, size; 7084c62bffeSBjoern A. Zeeb #if defined(__i386__) 7094c62bffeSBjoern A. Zeeb uint32_t pad; 7104c62bffeSBjoern A. Zeeb #endif 711eddfbb76SRobert Watson 712eddfbb76SRobert Watson ef->vnet_start = 0; 713eddfbb76SRobert Watson ef->vnet_stop = 0; 714eddfbb76SRobert Watson error = link_elf_lookup_set(&ef->lf, "vnet", (void ***)&ef->vnet_start, 7159ae7bc39SBjoern A. Zeeb (void ***)&ef->vnet_stop, NULL); 716eddfbb76SRobert Watson /* Error just means there is no vnet data set to relocate. */ 7172f22b3ffSMatthew D Fleming if (error != 0) 718eddfbb76SRobert Watson return (0); 7199ae7bc39SBjoern A. Zeeb size = (uintptr_t)ef->vnet_stop - (uintptr_t)ef->vnet_start; 7209ae7bc39SBjoern A. Zeeb /* Empty set? */ 7219ae7bc39SBjoern A. Zeeb if (size < 1) 7229ae7bc39SBjoern A. Zeeb return (0); 7234c62bffeSBjoern A. Zeeb #if defined(__i386__) 7244c62bffeSBjoern A. Zeeb /* In case we do find __start/stop_set_ symbols double-check. */ 7254c62bffeSBjoern A. Zeeb if (size < 4) { 7264c62bffeSBjoern A. Zeeb uprintf("Kernel module '%s' must be recompiled with " 7274c62bffeSBjoern A. Zeeb "linker script\n", ef->lf.pathname); 7284c62bffeSBjoern A. Zeeb return (ENOEXEC); 7294c62bffeSBjoern A. Zeeb } 7304c62bffeSBjoern A. Zeeb 7314c62bffeSBjoern A. Zeeb /* Padding from linker-script correct? */ 7324c62bffeSBjoern A. Zeeb pad = *(uint32_t *)((uintptr_t)ef->vnet_stop - sizeof(pad)); 7334c62bffeSBjoern A. Zeeb if (pad != LS_PADDING) { 7344c62bffeSBjoern A. Zeeb uprintf("Kernel module '%s' must be recompiled with " 7354c62bffeSBjoern A. Zeeb "linker script, invalid padding %#04x (%#04x)\n", 7364c62bffeSBjoern A. Zeeb ef->lf.pathname, pad, LS_PADDING); 7374c62bffeSBjoern A. Zeeb return (ENOEXEC); 7384c62bffeSBjoern A. Zeeb } 7394c62bffeSBjoern A. Zeeb /* If we only have valid padding, nothing to do. */ 7404c62bffeSBjoern A. Zeeb if (size == 4) 7414c62bffeSBjoern A. Zeeb return (0); 7424c62bffeSBjoern A. Zeeb #endif 743eddfbb76SRobert Watson /* 7442f22b3ffSMatthew D Fleming * Allocate space in the primary vnet area. Copy in our 7452f22b3ffSMatthew D Fleming * initialization from the data section and then initialize 7462f22b3ffSMatthew D Fleming * all per-vnet storage from that. 747eddfbb76SRobert Watson */ 7489ae7bc39SBjoern A. Zeeb ef->vnet_base = (Elf_Addr)(uintptr_t)vnet_data_alloc(size); 7490f823b64SBjoern A. Zeeb if (ef->vnet_base == 0) { 7500f823b64SBjoern A. Zeeb printf("%s: vnet module space is out of space; " 7510f823b64SBjoern A. Zeeb "cannot allocate %d for %s\n", 7520f823b64SBjoern A. Zeeb __func__, size, ef->lf.pathname); 753eddfbb76SRobert Watson return (ENOSPC); 7540f823b64SBjoern A. Zeeb } 7559ae7bc39SBjoern A. Zeeb memcpy((void *)ef->vnet_base, (void *)ef->vnet_start, size); 7569ae7bc39SBjoern A. Zeeb vnet_data_copy((void *)ef->vnet_base, size); 75747813f5dSMikolaj Golub elf_set_add(&set_vnet_list, ef->vnet_start, ef->vnet_stop, 75847813f5dSMikolaj Golub ef->vnet_base); 759eddfbb76SRobert Watson 760eddfbb76SRobert Watson return (0); 761eddfbb76SRobert Watson } 762eddfbb76SRobert Watson #endif 7634c62bffeSBjoern A. Zeeb #undef LS_PADDING 764eddfbb76SRobert Watson 765f822c9e2SMark Johnston /* 766f822c9e2SMark Johnston * Apply the specified protection to the loadable segments of a preloaded linker 767f822c9e2SMark Johnston * file. 768f822c9e2SMark Johnston */ 76950c202c5SJeff Roberson static int 770f822c9e2SMark Johnston preload_protect(elf_file_t ef, vm_prot_t prot) 771f822c9e2SMark Johnston { 772f822c9e2SMark Johnston #ifdef __amd64__ 773f822c9e2SMark Johnston Elf_Ehdr *hdr; 774f822c9e2SMark Johnston Elf_Phdr *phdr, *phlimit; 775f822c9e2SMark Johnston vm_prot_t nprot; 776f822c9e2SMark Johnston int error; 777f822c9e2SMark Johnston 778f822c9e2SMark Johnston error = 0; 779f822c9e2SMark Johnston hdr = (Elf_Ehdr *)ef->address; 780f822c9e2SMark Johnston phdr = (Elf_Phdr *)(ef->address + hdr->e_phoff); 781f822c9e2SMark Johnston phlimit = phdr + hdr->e_phnum; 782f822c9e2SMark Johnston for (; phdr < phlimit; phdr++) { 783f822c9e2SMark Johnston if (phdr->p_type != PT_LOAD) 784f822c9e2SMark Johnston continue; 785f822c9e2SMark Johnston 786f822c9e2SMark Johnston nprot = prot | VM_PROT_READ; 787f822c9e2SMark Johnston if ((phdr->p_flags & PF_W) != 0) 788f822c9e2SMark Johnston nprot |= VM_PROT_WRITE; 789f822c9e2SMark Johnston if ((phdr->p_flags & PF_X) != 0) 790f822c9e2SMark Johnston nprot |= VM_PROT_EXECUTE; 791f822c9e2SMark Johnston error = pmap_change_prot((vm_offset_t)ef->address + 792f822c9e2SMark Johnston phdr->p_vaddr, round_page(phdr->p_memsz), nprot); 793f822c9e2SMark Johnston if (error != 0) 794f822c9e2SMark Johnston break; 795f822c9e2SMark Johnston } 796f822c9e2SMark Johnston return (error); 797f822c9e2SMark Johnston #else 798f822c9e2SMark Johnston return (0); 799f822c9e2SMark Johnston #endif 800f822c9e2SMark Johnston } 801f822c9e2SMark Johnston 802b19c9deaSIan Lepore #ifdef __arm__ 803b19c9deaSIan Lepore /* 804b19c9deaSIan Lepore * Locate the ARM exception/unwind table info for DDB and stack(9) use by 805b19c9deaSIan Lepore * searching for the section header that describes it. There may be no unwind 806b19c9deaSIan Lepore * info, for example in a module containing only data. 807b19c9deaSIan Lepore */ 808b19c9deaSIan Lepore static void 809b19c9deaSIan Lepore link_elf_locate_exidx(linker_file_t lf, Elf_Shdr *shdr, int nhdr) 810b19c9deaSIan Lepore { 811b19c9deaSIan Lepore int i; 812b19c9deaSIan Lepore 813b19c9deaSIan Lepore for (i = 0; i < nhdr; i++) { 814b19c9deaSIan Lepore if (shdr[i].sh_type == SHT_ARM_EXIDX) { 815b19c9deaSIan Lepore lf->exidx_addr = shdr[i].sh_addr + lf->address; 816b19c9deaSIan Lepore lf->exidx_size = shdr[i].sh_size; 817b19c9deaSIan Lepore break; 818b19c9deaSIan Lepore } 819b19c9deaSIan Lepore } 820b19c9deaSIan Lepore } 821b19c9deaSIan Lepore 822b19c9deaSIan Lepore /* 823b19c9deaSIan Lepore * Locate the section headers metadata in a preloaded module, then use it to 824b19c9deaSIan Lepore * locate the exception/unwind table in the module. The size of the metadata 825b19c9deaSIan Lepore * block is stored in a uint32 word immediately before the data itself, and a 826b19c9deaSIan Lepore * comment in preload_search_info() says it is safe to rely on that. 827b19c9deaSIan Lepore */ 828b19c9deaSIan Lepore static void 829b19c9deaSIan Lepore link_elf_locate_exidx_preload(struct linker_file *lf, caddr_t modptr) 830b19c9deaSIan Lepore { 831b19c9deaSIan Lepore uint32_t *modinfo; 832b19c9deaSIan Lepore Elf_Shdr *shdr; 833b19c9deaSIan Lepore uint32_t nhdr; 834b19c9deaSIan Lepore 835b19c9deaSIan Lepore modinfo = (uint32_t *)preload_search_info(modptr, 836b19c9deaSIan Lepore MODINFO_METADATA | MODINFOMD_SHDR); 837b19c9deaSIan Lepore if (modinfo != NULL) { 838b19c9deaSIan Lepore shdr = (Elf_Shdr *)modinfo; 839b19c9deaSIan Lepore nhdr = modinfo[-1] / sizeof(Elf_Shdr); 840b19c9deaSIan Lepore link_elf_locate_exidx(lf, shdr, nhdr); 841b19c9deaSIan Lepore } 842b19c9deaSIan Lepore } 843b19c9deaSIan Lepore 844b19c9deaSIan Lepore #endif /* __arm__ */ 845b19c9deaSIan Lepore 846f822c9e2SMark Johnston static int 847f822c9e2SMark Johnston link_elf_link_preload(linker_class_t cls, const char *filename, 848f822c9e2SMark Johnston linker_file_t *result) 849de78ca7eSPeter Wemm { 8500067051fSMarcel Moolenaar Elf_Addr *ctors_addrp; 8510067051fSMarcel Moolenaar Elf_Size *ctors_sizep; 852de78ca7eSPeter Wemm caddr_t modptr, baseptr, sizeptr, dynptr; 853de78ca7eSPeter Wemm char *type; 854de78ca7eSPeter Wemm elf_file_t ef; 855de78ca7eSPeter Wemm linker_file_t lf; 856de78ca7eSPeter Wemm int error; 857de78ca7eSPeter Wemm vm_offset_t dp; 858de78ca7eSPeter Wemm 85954823af2SPeter Wemm /* Look to see if we have the file preloaded */ 860de78ca7eSPeter Wemm modptr = preload_search_by_name(filename); 861de78ca7eSPeter Wemm if (modptr == NULL) 8622f22b3ffSMatthew D Fleming return (ENOENT); 863de78ca7eSPeter Wemm 864de78ca7eSPeter Wemm type = (char *)preload_search_info(modptr, MODINFO_TYPE); 865de78ca7eSPeter Wemm baseptr = preload_search_info(modptr, MODINFO_ADDR); 866de78ca7eSPeter Wemm sizeptr = preload_search_info(modptr, MODINFO_SIZE); 8672f22b3ffSMatthew D Fleming dynptr = preload_search_info(modptr, 8682f22b3ffSMatthew D Fleming MODINFO_METADATA | MODINFOMD_DYNAMIC); 86967db8b23SPeter Wemm if (type == NULL || 87067db8b23SPeter Wemm (strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE) " module") != 0 && 87167db8b23SPeter Wemm strcmp(type, "elf module") != 0)) 872de78ca7eSPeter Wemm return (EFTYPE); 873de78ca7eSPeter Wemm if (baseptr == NULL || sizeptr == NULL || dynptr == NULL) 874de78ca7eSPeter Wemm return (EINVAL); 875de78ca7eSPeter Wemm 876326e27d8SDoug Rabson lf = linker_make_file(filename, &link_elf_class); 8772f22b3ffSMatthew D Fleming if (lf == NULL) 8782f22b3ffSMatthew D Fleming return (ENOMEM); 879326e27d8SDoug Rabson 880326e27d8SDoug Rabson ef = (elf_file_t) lf; 881326e27d8SDoug Rabson ef->preloaded = 1; 882ca65d5c7SPeter Wemm ef->modptr = modptr; 883de78ca7eSPeter Wemm ef->address = *(caddr_t *)baseptr; 884de78ca7eSPeter Wemm #ifdef SPARSE_MAPPING 8851aa696baSMark Johnston ef->object = NULL; 886de78ca7eSPeter Wemm #endif 887de78ca7eSPeter Wemm dp = (vm_offset_t)ef->address + *(vm_offset_t *)dynptr; 888de78ca7eSPeter Wemm ef->dynamic = (Elf_Dyn *)dp; 889de78ca7eSPeter Wemm lf->address = ef->address; 890de78ca7eSPeter Wemm lf->size = *(size_t *)sizeptr; 891de78ca7eSPeter Wemm 8920067051fSMarcel Moolenaar ctors_addrp = (Elf_Addr *)preload_search_info(modptr, 8930067051fSMarcel Moolenaar MODINFO_METADATA | MODINFOMD_CTORS_ADDR); 8940067051fSMarcel Moolenaar ctors_sizep = (Elf_Size *)preload_search_info(modptr, 8950067051fSMarcel Moolenaar MODINFO_METADATA | MODINFOMD_CTORS_SIZE); 8960067051fSMarcel Moolenaar if (ctors_addrp != NULL && ctors_sizep != NULL) { 8970067051fSMarcel Moolenaar lf->ctors_addr = ef->address + *ctors_addrp; 8980067051fSMarcel Moolenaar lf->ctors_size = *ctors_sizep; 8990067051fSMarcel Moolenaar } 9000067051fSMarcel Moolenaar 901b19c9deaSIan Lepore #ifdef __arm__ 902b19c9deaSIan Lepore link_elf_locate_exidx_preload(lf, modptr); 903b19c9deaSIan Lepore #endif 904b19c9deaSIan Lepore 905326e27d8SDoug Rabson error = parse_dynamic(ef); 90650c202c5SJeff Roberson if (error == 0) 90750c202c5SJeff Roberson error = parse_dpcpu(ef); 908eddfbb76SRobert Watson #ifdef VIMAGE 909eddfbb76SRobert Watson if (error == 0) 910eddfbb76SRobert Watson error = parse_vnet(ef); 911eddfbb76SRobert Watson #endif 912f822c9e2SMark Johnston if (error == 0) 913f822c9e2SMark Johnston error = preload_protect(ef, VM_PROT_ALL); 9142f22b3ffSMatthew D Fleming if (error != 0) { 91565a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 9162f22b3ffSMatthew D Fleming return (error); 917de78ca7eSPeter Wemm } 9187251b4bfSJake Burkholder link_elf_reloc_local(lf); 919de78ca7eSPeter Wemm *result = lf; 920de78ca7eSPeter Wemm return (0); 921de78ca7eSPeter Wemm } 922de78ca7eSPeter Wemm 923de78ca7eSPeter Wemm static int 92454823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf) 92554823af2SPeter Wemm { 92654823af2SPeter Wemm elf_file_t ef; 92754823af2SPeter Wemm int error; 92854823af2SPeter Wemm 92954823af2SPeter Wemm ef = (elf_file_t) lf; 93054823af2SPeter Wemm error = relocate_file(ef); 931f822c9e2SMark Johnston if (error == 0) 932f822c9e2SMark Johnston error = preload_protect(ef, VM_PROT_NONE); 9332f22b3ffSMatthew D Fleming if (error != 0) 9342f22b3ffSMatthew D Fleming return (error); 93554823af2SPeter Wemm (void)link_elf_preload_parse_symbols(ef); 93654823af2SPeter Wemm 937c143d6c2SMarcel Moolenaar return (link_elf_link_common_finish(lf)); 93854823af2SPeter Wemm } 93954823af2SPeter Wemm 94054823af2SPeter Wemm static int 941c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char* filename, 942c143d6c2SMarcel Moolenaar linker_file_t* result) 943a4f67738SDoug Rabson { 944a4f67738SDoug Rabson struct nameidata nd; 945b40ce416SJulian Elischer struct thread* td = curthread; /* XXX */ 946caab6e90SPeter Wemm Elf_Ehdr *hdr; 9471aa696baSMark Johnston caddr_t firstpage, segbase; 948fe3db7c7SDoug Rabson int nbytes, i; 949fe3db7c7SDoug Rabson Elf_Phdr *phdr; 950fe3db7c7SDoug Rabson Elf_Phdr *phlimit; 9511a330eb0SPeter Wemm Elf_Phdr *segs[MAXSEGS]; 952fe3db7c7SDoug Rabson int nsegs; 953fe3db7c7SDoug Rabson Elf_Phdr *phdyn; 954fe3db7c7SDoug Rabson caddr_t mapbase; 955fe3db7c7SDoug Rabson size_t mapsize; 956fe3db7c7SDoug Rabson Elf_Addr base_vaddr; 957fe3db7c7SDoug Rabson Elf_Addr base_vlimit; 958a4f67738SDoug Rabson int error = 0; 959526d0bd5SKonstantin Belousov ssize_t resid; 960526d0bd5SKonstantin Belousov int flags; 961a4f67738SDoug Rabson elf_file_t ef; 962a4f67738SDoug Rabson linker_file_t lf; 963ca65d5c7SPeter Wemm Elf_Shdr *shdr; 964ca65d5c7SPeter Wemm int symtabindex; 965ca65d5c7SPeter Wemm int symstrindex; 9660067051fSMarcel Moolenaar int shstrindex; 967ca65d5c7SPeter Wemm int symcnt; 968ca65d5c7SPeter Wemm int strcnt; 9690067051fSMarcel Moolenaar char *shstrs; 970ca65d5c7SPeter Wemm 971ca65d5c7SPeter Wemm shdr = NULL; 972ca65d5c7SPeter Wemm lf = NULL; 9730067051fSMarcel Moolenaar shstrs = NULL; 974a4f67738SDoug Rabson 9755050aa86SKonstantin Belousov NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td); 976e6796b67SKirk McKusick flags = FREAD; 9779e223287SKonstantin Belousov error = vn_open(&nd, &flags, 0, NULL); 9782f22b3ffSMatthew D Fleming if (error != 0) 9792f22b3ffSMatthew D Fleming return (error); 980762e6b85SEivind Eklund NDFREE(&nd, NDF_ONLY_PNBUF); 9811f49b573SRuslan Ermilov if (nd.ni_vp->v_type != VREG) { 9821f49b573SRuslan Ermilov error = ENOEXEC; 9831f49b573SRuslan Ermilov firstpage = NULL; 9841f49b573SRuslan Ermilov goto out; 9851f49b573SRuslan Ermilov } 986a3df768bSRobert Watson #ifdef MAC 98730d239bcSRobert Watson error = mac_kld_check_load(curthread->td_ucred, nd.ni_vp); 9882f22b3ffSMatthew D Fleming if (error != 0) { 989a3df768bSRobert Watson firstpage = NULL; 990a3df768bSRobert Watson goto out; 991a3df768bSRobert Watson } 992a3df768bSRobert Watson #endif 993a4f67738SDoug Rabson 994a4f67738SDoug Rabson /* 995fe3db7c7SDoug Rabson * Read the elf header from the file. 996a4f67738SDoug Rabson */ 997a163d034SWarner Losh firstpage = malloc(PAGE_SIZE, M_LINKER, M_WAITOK); 998caab6e90SPeter Wemm hdr = (Elf_Ehdr *)firstpage; 999caab6e90SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, firstpage, PAGE_SIZE, 0, 10009ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 10019ca43589SRobert Watson &resid, td); 1002caab6e90SPeter Wemm nbytes = PAGE_SIZE - resid; 10032f22b3ffSMatthew D Fleming if (error != 0) 1004a4f67738SDoug Rabson goto out; 1005a4f67738SDoug Rabson 1006caab6e90SPeter Wemm if (!IS_ELF(*hdr)) { 1007fe3db7c7SDoug Rabson error = ENOEXEC; 1008a4f67738SDoug Rabson goto out; 1009fe3db7c7SDoug Rabson } 1010fe3db7c7SDoug Rabson 10112f22b3ffSMatthew D Fleming if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS || 10122f22b3ffSMatthew D Fleming hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 1013552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file layout"); 1014fe3db7c7SDoug Rabson error = ENOEXEC; 1015fe3db7c7SDoug Rabson goto out; 1016fe3db7c7SDoug Rabson } 10172f22b3ffSMatthew D Fleming if (hdr->e_ident[EI_VERSION] != EV_CURRENT || 10182f22b3ffSMatthew D Fleming hdr->e_version != EV_CURRENT) { 1019552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file version"); 1020fe3db7c7SDoug Rabson error = ENOEXEC; 1021fe3db7c7SDoug Rabson goto out; 1022fe3db7c7SDoug Rabson } 1023caab6e90SPeter Wemm if (hdr->e_type != ET_EXEC && hdr->e_type != ET_DYN) { 1024a1d7ce03SAttilio Rao error = ENOSYS; 1025fe3db7c7SDoug Rabson goto out; 1026fe3db7c7SDoug Rabson } 1027caab6e90SPeter Wemm if (hdr->e_machine != ELF_TARG_MACH) { 1028552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported machine"); 1029fe3db7c7SDoug Rabson error = ENOEXEC; 1030fe3db7c7SDoug Rabson goto out; 1031fe3db7c7SDoug Rabson } 1032a4f67738SDoug Rabson 1033a4f67738SDoug Rabson /* 10342f22b3ffSMatthew D Fleming * We rely on the program header being in the first page. 10352f22b3ffSMatthew D Fleming * This is not strictly required by the ABI specification, but 10362f22b3ffSMatthew D Fleming * it seems to always true in practice. And, it simplifies 10372f22b3ffSMatthew D Fleming * things considerably. 1038a4f67738SDoug Rabson */ 1039caab6e90SPeter Wemm if (!((hdr->e_phentsize == sizeof(Elf_Phdr)) && 1040caab6e90SPeter Wemm (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= PAGE_SIZE) && 1041caab6e90SPeter Wemm (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= nbytes))) 1042552f9f63SEdwin Groothuis link_elf_error(filename, "Unreadable program headers"); 1043fe3db7c7SDoug Rabson 1044fe3db7c7SDoug Rabson /* 1045fe3db7c7SDoug Rabson * Scan the program header entries, and save key information. 1046fe3db7c7SDoug Rabson * 1047fe3db7c7SDoug Rabson * We rely on there being exactly two load segments, text and data, 1048fe3db7c7SDoug Rabson * in that order. 1049fe3db7c7SDoug Rabson */ 1050caab6e90SPeter Wemm phdr = (Elf_Phdr *) (firstpage + hdr->e_phoff); 1051caab6e90SPeter Wemm phlimit = phdr + hdr->e_phnum; 1052fe3db7c7SDoug Rabson nsegs = 0; 1053fe3db7c7SDoug Rabson phdyn = NULL; 1054fe3db7c7SDoug Rabson while (phdr < phlimit) { 1055fe3db7c7SDoug Rabson switch (phdr->p_type) { 1056fe3db7c7SDoug Rabson case PT_LOAD: 10571a330eb0SPeter Wemm if (nsegs == MAXSEGS) { 1058552f9f63SEdwin Groothuis link_elf_error(filename, "Too many sections"); 1059fe3db7c7SDoug Rabson error = ENOEXEC; 1060fe3db7c7SDoug Rabson goto out; 1061fe3db7c7SDoug Rabson } 106287b1831fSPoul-Henning Kamp /* 106387b1831fSPoul-Henning Kamp * XXX: We just trust they come in right order ?? 106487b1831fSPoul-Henning Kamp */ 1065fe3db7c7SDoug Rabson segs[nsegs] = phdr; 1066fe3db7c7SDoug Rabson ++nsegs; 1067fe3db7c7SDoug Rabson break; 1068fe3db7c7SDoug Rabson 1069fe3db7c7SDoug Rabson case PT_DYNAMIC: 1070fe3db7c7SDoug Rabson phdyn = phdr; 1071fe3db7c7SDoug Rabson break; 10729548091bSBoris Popov 10739548091bSBoris Popov case PT_INTERP: 1074a1d7ce03SAttilio Rao error = ENOSYS; 10759548091bSBoris Popov goto out; 1076fe3db7c7SDoug Rabson } 1077fe3db7c7SDoug Rabson 1078fe3db7c7SDoug Rabson ++phdr; 1079fe3db7c7SDoug Rabson } 1080fe3db7c7SDoug Rabson if (phdyn == NULL) { 1081552f9f63SEdwin Groothuis link_elf_error(filename, "Object is not dynamically-linked"); 1082fe3db7c7SDoug Rabson error = ENOEXEC; 1083fe3db7c7SDoug Rabson goto out; 1084fe3db7c7SDoug Rabson } 10851a330eb0SPeter Wemm if (nsegs == 0) { 1086552f9f63SEdwin Groothuis link_elf_error(filename, "No sections"); 108787b1831fSPoul-Henning Kamp error = ENOEXEC; 108887b1831fSPoul-Henning Kamp goto out; 108987b1831fSPoul-Henning Kamp } 1090fe3db7c7SDoug Rabson 1091fe3db7c7SDoug Rabson /* 10922f22b3ffSMatthew D Fleming * Allocate the entire address space of the object, to stake 10932f22b3ffSMatthew D Fleming * out our contiguous region, and to establish the base 10942f22b3ffSMatthew D Fleming * address for relocation. 1095fe3db7c7SDoug Rabson */ 1096fe3db7c7SDoug Rabson base_vaddr = trunc_page(segs[0]->p_vaddr); 10971a330eb0SPeter Wemm base_vlimit = round_page(segs[nsegs - 1]->p_vaddr + 10981a330eb0SPeter Wemm segs[nsegs - 1]->p_memsz); 1099fe3db7c7SDoug Rabson mapsize = base_vlimit - base_vaddr; 1100fe3db7c7SDoug Rabson 1101326e27d8SDoug Rabson lf = linker_make_file(filename, &link_elf_class); 11022f22b3ffSMatthew D Fleming if (lf == NULL) { 1103326e27d8SDoug Rabson error = ENOMEM; 1104326e27d8SDoug Rabson goto out; 1105326e27d8SDoug Rabson } 1106326e27d8SDoug Rabson 1107326e27d8SDoug Rabson ef = (elf_file_t) lf; 1108fe3db7c7SDoug Rabson #ifdef SPARSE_MAPPING 1109fbf2a778SKonstantin Belousov ef->object = vm_pager_allocate(OBJT_PHYS, NULL, mapsize, VM_PROT_ALL, 1110fbf2a778SKonstantin Belousov 0, thread0.td_ucred); 1111fe3db7c7SDoug Rabson if (ef->object == NULL) { 1112fe3db7c7SDoug Rabson error = ENOMEM; 1113fe3db7c7SDoug Rabson goto out; 1114fe3db7c7SDoug Rabson } 11151aa696baSMark Johnston #ifdef __amd64__ 11161aa696baSMark Johnston mapbase = (caddr_t)KERNBASE; 11171aa696baSMark Johnston #else 11181aa696baSMark Johnston mapbase = (caddr_t)vm_map_min(kernel_map); 11191aa696baSMark Johnston #endif 11201aa696baSMark Johnston /* 11211aa696baSMark Johnston * Mapping protections are downgraded after relocation processing. 11221aa696baSMark Johnston */ 1123fe3db7c7SDoug Rabson error = vm_map_find(kernel_map, ef->object, 0, 11241aa696baSMark Johnston (vm_offset_t *)&mapbase, mapsize, 0, VMFS_OPTIMAL_SPACE, 1125fe3db7c7SDoug Rabson VM_PROT_ALL, VM_PROT_ALL, 0); 11262f22b3ffSMatthew D Fleming if (error != 0) { 1127fe3db7c7SDoug Rabson vm_object_deallocate(ef->object); 11281aa696baSMark Johnston ef->object = NULL; 1129fe3db7c7SDoug Rabson goto out; 1130fe3db7c7SDoug Rabson } 1131fe3db7c7SDoug Rabson #else 113282c174a3SMateusz Guzik mapbase = malloc_exec(mapsize, M_LINKER, M_WAITOK); 1133fe3db7c7SDoug Rabson #endif 11341aa696baSMark Johnston ef->address = mapbase; 1135a4f67738SDoug Rabson 1136a4f67738SDoug Rabson /* 1137a4f67738SDoug Rabson * Read the text and data sections and zero the bss. 1138a4f67738SDoug Rabson */ 11391a330eb0SPeter Wemm for (i = 0; i < nsegs; i++) { 11401aa696baSMark Johnston segbase = mapbase + segs[i]->p_vaddr - base_vaddr; 11411aa696baSMark Johnston 11421aa696baSMark Johnston #ifdef SPARSE_MAPPING 11431aa696baSMark Johnston /* 11441aa696baSMark Johnston * Consecutive segments may have different mapping permissions, 11451aa696baSMark Johnston * so be strict and verify that their mappings do not overlap. 11461aa696baSMark Johnston */ 11471aa696baSMark Johnston if (((vm_offset_t)segbase & PAGE_MASK) != 0) { 11481aa696baSMark Johnston error = EINVAL; 11491aa696baSMark Johnston goto out; 11501aa696baSMark Johnston } 11511aa696baSMark Johnston 11521aa696baSMark Johnston error = vm_map_wire(kernel_map, 11531aa696baSMark Johnston (vm_offset_t)segbase, 11541aa696baSMark Johnston (vm_offset_t)segbase + round_page(segs[i]->p_memsz), 11551aa696baSMark Johnston VM_MAP_WIRE_SYSTEM | VM_MAP_WIRE_NOHOLES); 11561aa696baSMark Johnston if (error != KERN_SUCCESS) { 11571aa696baSMark Johnston error = ENOMEM; 11581aa696baSMark Johnston goto out; 11591aa696baSMark Johnston } 11601aa696baSMark Johnston #endif 11611aa696baSMark Johnston 1162fe3db7c7SDoug Rabson error = vn_rdwr(UIO_READ, nd.ni_vp, 1163fe3db7c7SDoug Rabson segbase, segs[i]->p_filesz, segs[i]->p_offset, 11649ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 11659ca43589SRobert Watson &resid, td); 11662f22b3ffSMatthew D Fleming if (error != 0) 1167a4f67738SDoug Rabson goto out; 1168fe3db7c7SDoug Rabson bzero(segbase + segs[i]->p_filesz, 1169fe3db7c7SDoug Rabson segs[i]->p_memsz - segs[i]->p_filesz); 1170fe3db7c7SDoug Rabson } 1171fe3db7c7SDoug Rabson 1172f99502a4SBrian Feldman #ifdef GPROF 1173f99502a4SBrian Feldman /* Update profiling information with the new text segment. */ 117470f37788SJohn Baldwin mtx_lock(&Giant); 1175853a17adSEnji Cooper kmupetext((uintfptr_t)(mapbase + segs[0]->p_vaddr - base_vaddr + 1176f99502a4SBrian Feldman segs[0]->p_memsz)); 117770f37788SJohn Baldwin mtx_unlock(&Giant); 1178f99502a4SBrian Feldman #endif 1179f99502a4SBrian Feldman 1180326e27d8SDoug Rabson ef->dynamic = (Elf_Dyn *) (mapbase + phdyn->p_vaddr - base_vaddr); 1181a4f67738SDoug Rabson 1182a4f67738SDoug Rabson lf->address = ef->address; 1183fe3db7c7SDoug Rabson lf->size = mapsize; 1184a4f67738SDoug Rabson 1185326e27d8SDoug Rabson error = parse_dynamic(ef); 11862f22b3ffSMatthew D Fleming if (error != 0) 1187de78ca7eSPeter Wemm goto out; 118850c202c5SJeff Roberson error = parse_dpcpu(ef); 11892f22b3ffSMatthew D Fleming if (error != 0) 119050c202c5SJeff Roberson goto out; 1191eddfbb76SRobert Watson #ifdef VIMAGE 1192eddfbb76SRobert Watson error = parse_vnet(ef); 11932f22b3ffSMatthew D Fleming if (error != 0) 1194eddfbb76SRobert Watson goto out; 1195eddfbb76SRobert Watson #endif 11967251b4bfSJake Burkholder link_elf_reloc_local(lf); 11977251b4bfSJake Burkholder 1198b249ce48SMateusz Guzik VOP_UNLOCK(nd.ni_vp); 11997b9716baSIan Dowse error = linker_load_dependencies(lf); 12004f7afc20SKonstantin Belousov vn_lock(nd.ni_vp, LK_EXCLUSIVE | LK_RETRY); 12012f22b3ffSMatthew D Fleming if (error != 0) 1202de78ca7eSPeter Wemm goto out; 1203326e27d8SDoug Rabson error = relocate_file(ef); 12042f22b3ffSMatthew D Fleming if (error != 0) 1205ca65d5c7SPeter Wemm goto out; 1206ca65d5c7SPeter Wemm 12071aa696baSMark Johnston #ifdef SPARSE_MAPPING 12081aa696baSMark Johnston /* 12091aa696baSMark Johnston * Downgrade permissions on text segment mappings now that relocation 12101aa696baSMark Johnston * processing is complete. Restrict permissions on read-only segments. 12111aa696baSMark Johnston */ 12121aa696baSMark Johnston for (i = 0; i < nsegs; i++) { 12131aa696baSMark Johnston vm_prot_t prot; 12141aa696baSMark Johnston 12151aa696baSMark Johnston if (segs[i]->p_type != PT_LOAD) 12161aa696baSMark Johnston continue; 12171aa696baSMark Johnston 12181aa696baSMark Johnston prot = VM_PROT_READ; 12191aa696baSMark Johnston if ((segs[i]->p_flags & PF_W) != 0) 12201aa696baSMark Johnston prot |= VM_PROT_WRITE; 12211aa696baSMark Johnston if ((segs[i]->p_flags & PF_X) != 0) 12221aa696baSMark Johnston prot |= VM_PROT_EXECUTE; 12231aa696baSMark Johnston segbase = mapbase + segs[i]->p_vaddr - base_vaddr; 12241aa696baSMark Johnston error = vm_map_protect(kernel_map, 12251aa696baSMark Johnston (vm_offset_t)segbase, 12261aa696baSMark Johnston (vm_offset_t)segbase + round_page(segs[i]->p_memsz), 1227*0659df6fSKonstantin Belousov prot, 0, VM_MAP_PROTECT_SET_PROT); 12281aa696baSMark Johnston if (error != KERN_SUCCESS) { 12291aa696baSMark Johnston error = ENOMEM; 12301aa696baSMark Johnston goto out; 12311aa696baSMark Johnston } 12321aa696baSMark Johnston } 12331aa696baSMark Johnston #endif 12341aa696baSMark Johnston 12352f22b3ffSMatthew D Fleming /* 12362f22b3ffSMatthew D Fleming * Try and load the symbol table if it's present. (you can 12372f22b3ffSMatthew D Fleming * strip it!) 12382f22b3ffSMatthew D Fleming */ 1239caab6e90SPeter Wemm nbytes = hdr->e_shnum * hdr->e_shentsize; 1240caab6e90SPeter Wemm if (nbytes == 0 || hdr->e_shoff == 0) 1241ca65d5c7SPeter Wemm goto nosyms; 1242a163d034SWarner Losh shdr = malloc(nbytes, M_LINKER, M_WAITOK | M_ZERO); 1243ca65d5c7SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, 1244caab6e90SPeter Wemm (caddr_t)shdr, nbytes, hdr->e_shoff, 12459ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 12469ca43589SRobert Watson &resid, td); 12472f22b3ffSMatthew D Fleming if (error != 0) 1248ca65d5c7SPeter Wemm goto out; 12490067051fSMarcel Moolenaar 12500067051fSMarcel Moolenaar /* Read section string table */ 12510067051fSMarcel Moolenaar shstrindex = hdr->e_shstrndx; 12520067051fSMarcel Moolenaar if (shstrindex != 0 && shdr[shstrindex].sh_type == SHT_STRTAB && 12530067051fSMarcel Moolenaar shdr[shstrindex].sh_size != 0) { 12540067051fSMarcel Moolenaar nbytes = shdr[shstrindex].sh_size; 12550067051fSMarcel Moolenaar shstrs = malloc(nbytes, M_LINKER, M_WAITOK | M_ZERO); 12560067051fSMarcel Moolenaar error = vn_rdwr(UIO_READ, nd.ni_vp, (caddr_t)shstrs, nbytes, 12570067051fSMarcel Moolenaar shdr[shstrindex].sh_offset, UIO_SYSSPACE, IO_NODELOCKED, 12580067051fSMarcel Moolenaar td->td_ucred, NOCRED, &resid, td); 12590067051fSMarcel Moolenaar if (error) 12600067051fSMarcel Moolenaar goto out; 12610067051fSMarcel Moolenaar } 12620067051fSMarcel Moolenaar 1263ca65d5c7SPeter Wemm symtabindex = -1; 1264ca65d5c7SPeter Wemm symstrindex = -1; 1265caab6e90SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 1266ca65d5c7SPeter Wemm if (shdr[i].sh_type == SHT_SYMTAB) { 1267ca65d5c7SPeter Wemm symtabindex = i; 1268ca65d5c7SPeter Wemm symstrindex = shdr[i].sh_link; 12690067051fSMarcel Moolenaar } else if (shstrs != NULL && shdr[i].sh_name != 0 && 12700067051fSMarcel Moolenaar strcmp(shstrs + shdr[i].sh_name, ".ctors") == 0) { 12710067051fSMarcel Moolenaar /* Record relocated address and size of .ctors. */ 12720067051fSMarcel Moolenaar lf->ctors_addr = mapbase + shdr[i].sh_addr - base_vaddr; 12730067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 1274ca65d5c7SPeter Wemm } 1275ca65d5c7SPeter Wemm } 1276ca65d5c7SPeter Wemm if (symtabindex < 0 || symstrindex < 0) 1277ca65d5c7SPeter Wemm goto nosyms; 1278ca65d5c7SPeter Wemm 1279ca65d5c7SPeter Wemm symcnt = shdr[symtabindex].sh_size; 1280a163d034SWarner Losh ef->symbase = malloc(symcnt, M_LINKER, M_WAITOK); 1281ca65d5c7SPeter Wemm strcnt = shdr[symstrindex].sh_size; 1282a163d034SWarner Losh ef->strbase = malloc(strcnt, M_LINKER, M_WAITOK); 1283ca65d5c7SPeter Wemm 1284ca65d5c7SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, 1285ca65d5c7SPeter Wemm ef->symbase, symcnt, shdr[symtabindex].sh_offset, 12869ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 12879ca43589SRobert Watson &resid, td); 12882f22b3ffSMatthew D Fleming if (error != 0) 1289ca65d5c7SPeter Wemm goto out; 1290ca65d5c7SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, 1291ca65d5c7SPeter Wemm ef->strbase, strcnt, shdr[symstrindex].sh_offset, 12929ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 12939ca43589SRobert Watson &resid, td); 12942f22b3ffSMatthew D Fleming if (error != 0) 1295ca65d5c7SPeter Wemm goto out; 1296ca65d5c7SPeter Wemm 1297ca65d5c7SPeter Wemm ef->ddbsymcnt = symcnt / sizeof(Elf_Sym); 1298ca65d5c7SPeter Wemm ef->ddbsymtab = (const Elf_Sym *)ef->symbase; 1299ca65d5c7SPeter Wemm ef->ddbstrcnt = strcnt; 1300ca65d5c7SPeter Wemm ef->ddbstrtab = ef->strbase; 1301ca65d5c7SPeter Wemm 13028a710317SMarcel Moolenaar nosyms: 1303b19c9deaSIan Lepore 1304b19c9deaSIan Lepore #ifdef __arm__ 1305b19c9deaSIan Lepore link_elf_locate_exidx(lf, shdr, hdr->e_shnum); 1306b19c9deaSIan Lepore #endif 1307b19c9deaSIan Lepore 1308c143d6c2SMarcel Moolenaar error = link_elf_link_common_finish(lf); 13092f22b3ffSMatthew D Fleming if (error != 0) 1310c143d6c2SMarcel Moolenaar goto out; 1311326e27d8SDoug Rabson 1312a4f67738SDoug Rabson *result = lf; 1313a4f67738SDoug Rabson 1314a4f67738SDoug Rabson out: 1315b249ce48SMateusz Guzik VOP_UNLOCK(nd.ni_vp); 1316f1d2d305SKonstantin Belousov vn_close(nd.ni_vp, FREAD, td->td_ucred, td); 13172f22b3ffSMatthew D Fleming if (error != 0 && lf != NULL) 131865a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 1319ca65d5c7SPeter Wemm free(shdr, M_LINKER); 1320caab6e90SPeter Wemm free(firstpage, M_LINKER); 13210067051fSMarcel Moolenaar free(shstrs, M_LINKER); 1322a4f67738SDoug Rabson 13232f22b3ffSMatthew D Fleming return (error); 1324a4f67738SDoug Rabson } 1325a4f67738SDoug Rabson 132650c202c5SJeff Roberson Elf_Addr 132750c202c5SJeff Roberson elf_relocaddr(linker_file_t lf, Elf_Addr x) 132850c202c5SJeff Roberson { 132950c202c5SJeff Roberson elf_file_t ef; 133050c202c5SJeff Roberson 133138dae42cSMark Johnston KASSERT(lf->ops->cls == (kobj_class_t)&link_elf_class, 133238dae42cSMark Johnston ("elf_relocaddr: unexpected linker file %p", lf)); 133338dae42cSMark Johnston 133450c202c5SJeff Roberson ef = (elf_file_t)lf; 133550c202c5SJeff Roberson if (x >= ef->pcpu_start && x < ef->pcpu_stop) 133650c202c5SJeff Roberson return ((x - ef->pcpu_start) + ef->pcpu_base); 1337eddfbb76SRobert Watson #ifdef VIMAGE 1338eddfbb76SRobert Watson if (x >= ef->vnet_start && x < ef->vnet_stop) 1339eddfbb76SRobert Watson return ((x - ef->vnet_start) + ef->vnet_base); 1340eddfbb76SRobert Watson #endif 134150c202c5SJeff Roberson return (x); 134250c202c5SJeff Roberson } 134350c202c5SJeff Roberson 1344a4f67738SDoug Rabson static void 1345de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file) 1346a4f67738SDoug Rabson { 1347326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) file; 1348a4f67738SDoug Rabson 13492f22b3ffSMatthew D Fleming if (ef->pcpu_base != 0) { 13502f22b3ffSMatthew D Fleming dpcpu_free((void *)ef->pcpu_base, 13512f22b3ffSMatthew D Fleming ef->pcpu_stop - ef->pcpu_start); 135247813f5dSMikolaj Golub elf_set_delete(&set_pcpu_list, ef->pcpu_start); 135350c202c5SJeff Roberson } 1354eddfbb76SRobert Watson #ifdef VIMAGE 13552f22b3ffSMatthew D Fleming if (ef->vnet_base != 0) { 13562f22b3ffSMatthew D Fleming vnet_data_free((void *)ef->vnet_base, 13572f22b3ffSMatthew D Fleming ef->vnet_stop - ef->vnet_start); 135847813f5dSMikolaj Golub elf_set_delete(&set_vnet_list, ef->vnet_start); 1359eddfbb76SRobert Watson } 1360eddfbb76SRobert Watson #endif 1361ecb01c64SMarcel Moolenaar #ifdef GDB 13622f22b3ffSMatthew D Fleming if (ef->gdb.l_ld != NULL) { 1363326e27d8SDoug Rabson GDB_STATE(RT_DELETE); 1364ed6c3888SPeter Wemm free((void *)(uintptr_t)ef->gdb.l_name, M_LINKER); 1365326e27d8SDoug Rabson link_elf_delete_gdb(&ef->gdb); 1366326e27d8SDoug Rabson GDB_STATE(RT_CONSISTENT); 1367326e27d8SDoug Rabson } 1368326e27d8SDoug Rabson #endif 1369326e27d8SDoug Rabson 13701aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being unloaded. */ 13711aeb23cdSMarcel Moolenaar elf_cpu_unload_file(file); 13721aeb23cdSMarcel Moolenaar 1373326e27d8SDoug Rabson if (ef->preloaded) { 137454823af2SPeter Wemm link_elf_unload_preload(file); 1375326e27d8SDoug Rabson return; 1376326e27d8SDoug Rabson } 1377c143d6c2SMarcel Moolenaar 1378fe3db7c7SDoug Rabson #ifdef SPARSE_MAPPING 13792f22b3ffSMatthew D Fleming if (ef->object != NULL) { 1380fe3db7c7SDoug Rabson vm_map_remove(kernel_map, (vm_offset_t) ef->address, 1381fe3db7c7SDoug Rabson (vm_offset_t) ef->address 1382fe3db7c7SDoug Rabson + (ef->object->size << PAGE_SHIFT)); 1383fe3db7c7SDoug Rabson } 1384fe3db7c7SDoug Rabson #else 1385a4f67738SDoug Rabson free(ef->address, M_LINKER); 1386fe3db7c7SDoug Rabson #endif 1387ca65d5c7SPeter Wemm free(ef->symbase, M_LINKER); 1388ca65d5c7SPeter Wemm free(ef->strbase, M_LINKER); 1389a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1390a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1391a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 1392a4f67738SDoug Rabson } 1393a4f67738SDoug Rabson 1394de78ca7eSPeter Wemm static void 139554823af2SPeter Wemm link_elf_unload_preload(linker_file_t file) 1396de78ca7eSPeter Wemm { 1397f822c9e2SMark Johnston 139873624a80SMark Johnston if (file->pathname != NULL) 139973624a80SMark Johnston preload_delete_name(file->pathname); 1400de78ca7eSPeter Wemm } 1401de78ca7eSPeter Wemm 1402fe3db7c7SDoug Rabson static const char * 1403757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info) 1404a4f67738SDoug Rabson { 1405fe3db7c7SDoug Rabson const Elf_Sym *ref; 1406a4f67738SDoug Rabson 1407aa855a59SPeter Wemm if (ELF_R_SYM(r_info)) { 1408aa855a59SPeter Wemm ref = ef->symtab + ELF_R_SYM(r_info); 14092f22b3ffSMatthew D Fleming return (ef->strtab + ref->st_name); 14102f22b3ffSMatthew D Fleming } 14112f22b3ffSMatthew D Fleming return (NULL); 1412a4f67738SDoug Rabson } 1413a4f67738SDoug Rabson 1414a4f67738SDoug Rabson static int 141536716fe2SMark Johnston symbol_type(elf_file_t ef, Elf_Size r_info) 1416a4f67738SDoug Rabson { 141736716fe2SMark Johnston const Elf_Sym *ref; 141836716fe2SMark Johnston 141936716fe2SMark Johnston if (ELF_R_SYM(r_info)) { 142036716fe2SMark Johnston ref = ef->symtab + ELF_R_SYM(r_info); 142136716fe2SMark Johnston return (ELF_ST_TYPE(ref->st_info)); 142236716fe2SMark Johnston } 142336716fe2SMark Johnston return (STT_NOTYPE); 142436716fe2SMark Johnston } 142536716fe2SMark Johnston 142636716fe2SMark Johnston static int 142736716fe2SMark Johnston relocate_file1(elf_file_t ef, elf_lookup_fn lookup, elf_reloc_fn reloc, 142836716fe2SMark Johnston bool ifuncs) 142936716fe2SMark Johnston { 1430fe3db7c7SDoug Rabson const Elf_Rel *rel; 1431fe3db7c7SDoug Rabson const Elf_Rela *rela; 1432aa855a59SPeter Wemm const char *symname; 1433a4f67738SDoug Rabson 143436716fe2SMark Johnston #define APPLY_RELOCS(iter, tbl, tblsize, type) do { \ 143536716fe2SMark Johnston for ((iter) = (tbl); (iter) != NULL && \ 143636716fe2SMark Johnston (iter) < (tbl) + (tblsize) / sizeof(*(iter)); (iter)++) { \ 143736716fe2SMark Johnston if ((symbol_type(ef, (iter)->r_info) == \ 143836716fe2SMark Johnston STT_GNU_IFUNC || \ 143936716fe2SMark Johnston elf_is_ifunc_reloc((iter)->r_info)) != ifuncs) \ 144036716fe2SMark Johnston continue; \ 144136716fe2SMark Johnston if (reloc(&ef->lf, (Elf_Addr)ef->address, \ 144236716fe2SMark Johnston (iter), (type), lookup)) { \ 144336716fe2SMark Johnston symname = symbol_name(ef, (iter)->r_info); \ 144436716fe2SMark Johnston printf("link_elf: symbol %s undefined\n", \ 144536716fe2SMark Johnston symname); \ 144636716fe2SMark Johnston return (ENOENT); \ 144736716fe2SMark Johnston } \ 144836716fe2SMark Johnston } \ 144936716fe2SMark Johnston } while (0) 1450a4f67738SDoug Rabson 145136716fe2SMark Johnston APPLY_RELOCS(rel, ef->rel, ef->relsize, ELF_RELOC_REL); 145236716fe2SMark Johnston APPLY_RELOCS(rela, ef->rela, ef->relasize, ELF_RELOC_RELA); 145336716fe2SMark Johnston APPLY_RELOCS(rel, ef->pltrel, ef->pltrelsize, ELF_RELOC_REL); 145436716fe2SMark Johnston APPLY_RELOCS(rela, ef->pltrela, ef->pltrelasize, ELF_RELOC_RELA); 1455a4f67738SDoug Rabson 145636716fe2SMark Johnston #undef APPLY_RELOCS 1457a4f67738SDoug Rabson 14582f22b3ffSMatthew D Fleming return (0); 1459a4f67738SDoug Rabson } 1460a4f67738SDoug Rabson 14617035cf14SKonstantin Belousov static int 14627035cf14SKonstantin Belousov relocate_file(elf_file_t ef) 14637035cf14SKonstantin Belousov { 146436716fe2SMark Johnston int error; 14657035cf14SKonstantin Belousov 146636716fe2SMark Johnston error = relocate_file1(ef, elf_lookup, elf_reloc, false); 146736716fe2SMark Johnston if (error == 0) 146836716fe2SMark Johnston error = relocate_file1(ef, elf_lookup, elf_reloc, true); 146936716fe2SMark Johnston return (error); 14707035cf14SKonstantin Belousov } 14717035cf14SKonstantin Belousov 1472fe3db7c7SDoug Rabson /* 1473fe3db7c7SDoug Rabson * Hash function for symbol table lookup. Don't even think about changing 1474fe3db7c7SDoug Rabson * this. It is specified by the System V ABI. 1475fe3db7c7SDoug Rabson */ 1476fe3db7c7SDoug Rabson static unsigned long 1477fe3db7c7SDoug Rabson elf_hash(const char *name) 1478a4f67738SDoug Rabson { 1479fe3db7c7SDoug Rabson const unsigned char *p = (const unsigned char *) name; 1480fe3db7c7SDoug Rabson unsigned long h = 0; 1481fe3db7c7SDoug Rabson unsigned long g; 1482a4f67738SDoug Rabson 1483fe3db7c7SDoug Rabson while (*p != '\0') { 1484fe3db7c7SDoug Rabson h = (h << 4) + *p++; 1485fe3db7c7SDoug Rabson if ((g = h & 0xf0000000) != 0) 1486fe3db7c7SDoug Rabson h ^= g >> 24; 1487fe3db7c7SDoug Rabson h &= ~g; 1488a4f67738SDoug Rabson } 14892f22b3ffSMatthew D Fleming return (h); 1490fe3db7c7SDoug Rabson } 1491a4f67738SDoug Rabson 149237c84183SPoul-Henning Kamp static int 1493d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym) 1494a4f67738SDoug Rabson { 1495326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1496fe3db7c7SDoug Rabson unsigned long symnum; 14972d636ab0SPeter Wemm const Elf_Sym* symp; 14982d636ab0SPeter Wemm const char *strp; 1499fe3db7c7SDoug Rabson unsigned long hash; 1500a4f67738SDoug Rabson int i; 1501a4f67738SDoug Rabson 1502f6494f2eSJohn Baldwin /* If we don't have a hash, bail. */ 1503f6494f2eSJohn Baldwin if (ef->buckets == NULL || ef->nbuckets == 0) { 1504f6494f2eSJohn Baldwin printf("link_elf_lookup_symbol: missing symbol hash table\n"); 15052f22b3ffSMatthew D Fleming return (ENOENT); 1506f6494f2eSJohn Baldwin } 1507f6494f2eSJohn Baldwin 15082d636ab0SPeter Wemm /* First, search hashed global symbols */ 1509fe3db7c7SDoug Rabson hash = elf_hash(name); 1510fe3db7c7SDoug Rabson symnum = ef->buckets[hash % ef->nbuckets]; 1511fe3db7c7SDoug Rabson 1512fe3db7c7SDoug Rabson while (symnum != STN_UNDEF) { 1513fe3db7c7SDoug Rabson if (symnum >= ef->nchains) { 15142f22b3ffSMatthew D Fleming printf("%s: corrupt symbol table\n", __func__); 15152f22b3ffSMatthew D Fleming return (ENOENT); 1516a4f67738SDoug Rabson } 1517fe3db7c7SDoug Rabson 1518fe3db7c7SDoug Rabson symp = ef->symtab + symnum; 1519fe3db7c7SDoug Rabson if (symp->st_name == 0) { 15202f22b3ffSMatthew D Fleming printf("%s: corrupt symbol table\n", __func__); 15212f22b3ffSMatthew D Fleming return (ENOENT); 1522fe3db7c7SDoug Rabson } 1523fe3db7c7SDoug Rabson 1524fe3db7c7SDoug Rabson strp = ef->strtab + symp->st_name; 1525fe3db7c7SDoug Rabson 1526fe3db7c7SDoug Rabson if (strcmp(name, strp) == 0) { 1527fe3db7c7SDoug Rabson if (symp->st_shndx != SHN_UNDEF || 1528fe3db7c7SDoug Rabson (symp->st_value != 0 && 15297035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 15307035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC))) { 1531d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 15322f22b3ffSMatthew D Fleming return (0); 15332f22b3ffSMatthew D Fleming } 15342f22b3ffSMatthew D Fleming return (ENOENT); 1535fe3db7c7SDoug Rabson } 1536fe3db7c7SDoug Rabson 1537fe3db7c7SDoug Rabson symnum = ef->chains[symnum]; 1538a4f67738SDoug Rabson } 1539a4f67738SDoug Rabson 15402d636ab0SPeter Wemm /* If we have not found it, look at the full table (if loaded) */ 15412d636ab0SPeter Wemm if (ef->symtab == ef->ddbsymtab) 15422f22b3ffSMatthew D Fleming return (ENOENT); 15432d636ab0SPeter Wemm 15442d636ab0SPeter Wemm /* Exhaustive search */ 15452d636ab0SPeter Wemm for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 15462d636ab0SPeter Wemm strp = ef->ddbstrtab + symp->st_name; 15472d636ab0SPeter Wemm if (strcmp(name, strp) == 0) { 15482d636ab0SPeter Wemm if (symp->st_shndx != SHN_UNDEF || 15492d636ab0SPeter Wemm (symp->st_value != 0 && 15507035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 15517035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC))) { 1552d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 15532f22b3ffSMatthew D Fleming return (0); 15542f22b3ffSMatthew D Fleming } 15552f22b3ffSMatthew D Fleming return (ENOENT); 15562d636ab0SPeter Wemm } 15572d636ab0SPeter Wemm } 15582d636ab0SPeter Wemm 15592f22b3ffSMatthew D Fleming return (ENOENT); 1560a4f67738SDoug Rabson } 1561a4f67738SDoug Rabson 1562de78ca7eSPeter Wemm static int 15632f22b3ffSMatthew D Fleming link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, 15642f22b3ffSMatthew D Fleming linker_symval_t *symval) 1565a4f67738SDoug Rabson { 15667035cf14SKonstantin Belousov elf_file_t ef; 15677035cf14SKonstantin Belousov const Elf_Sym *es; 15687035cf14SKonstantin Belousov caddr_t val; 1569a4f67738SDoug Rabson 15707035cf14SKonstantin Belousov ef = (elf_file_t)lf; 15717035cf14SKonstantin Belousov es = (const Elf_Sym *)sym; 15725cf87418SMarcel Moolenaar if (es >= ef->symtab && es < (ef->symtab + ef->nchains)) { 1573a4f67738SDoug Rabson symval->name = ef->strtab + es->st_name; 15747035cf14SKonstantin Belousov val = (caddr_t)ef->address + es->st_value; 15757035cf14SKonstantin Belousov if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC) 15767035cf14SKonstantin Belousov val = ((caddr_t (*)(void))val)(); 15777035cf14SKonstantin Belousov symval->value = val; 1578a4f67738SDoug Rabson symval->size = es->st_size; 15792f22b3ffSMatthew D Fleming return (0); 1580a4f67738SDoug Rabson } 15812d636ab0SPeter Wemm if (ef->symtab == ef->ddbsymtab) 15822f22b3ffSMatthew D Fleming return (ENOENT); 15835cf87418SMarcel Moolenaar if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) { 15842d636ab0SPeter Wemm symval->name = ef->ddbstrtab + es->st_name; 15857035cf14SKonstantin Belousov val = (caddr_t)ef->address + es->st_value; 15867035cf14SKonstantin Belousov if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC) 15877035cf14SKonstantin Belousov val = ((caddr_t (*)(void))val)(); 15887035cf14SKonstantin Belousov symval->value = val; 15892d636ab0SPeter Wemm symval->size = es->st_size; 15902f22b3ffSMatthew D Fleming return (0); 15912d636ab0SPeter Wemm } 15922f22b3ffSMatthew D Fleming return (ENOENT); 15932d636ab0SPeter Wemm } 1594a4f67738SDoug Rabson 1595a4f67738SDoug Rabson static int 1596a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value, 1597d254af07SMatthew Dillon c_linker_sym_t *sym, long *diffp) 1598a4f67738SDoug Rabson { 1599326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1600586453feSBruce Evans u_long off = (uintptr_t) (void *) value; 1601a4f67738SDoug Rabson u_long diff = off; 1602586453feSBruce Evans u_long st_value; 1603fe3db7c7SDoug Rabson const Elf_Sym* es; 1604b85f65afSPedro F. Giffuni const Elf_Sym* best = NULL; 1605a4f67738SDoug Rabson int i; 1606a4f67738SDoug Rabson 16072d636ab0SPeter Wemm for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 1608a4f67738SDoug Rabson if (es->st_name == 0) 1609a4f67738SDoug Rabson continue; 1610586453feSBruce Evans st_value = es->st_value + (uintptr_t) (void *) ef->address; 1611b5abfb70SPeter Wemm if (off >= st_value) { 1612b5abfb70SPeter Wemm if (off - st_value < diff) { 1613b5abfb70SPeter Wemm diff = off - st_value; 1614a4f67738SDoug Rabson best = es; 1615a4f67738SDoug Rabson if (diff == 0) 1616a4f67738SDoug Rabson break; 1617b5abfb70SPeter Wemm } else if (off - st_value == diff) { 1618a4f67738SDoug Rabson best = es; 1619a4f67738SDoug Rabson } 1620a4f67738SDoug Rabson } 1621a4f67738SDoug Rabson } 1622b85f65afSPedro F. Giffuni if (best == NULL) 1623a4f67738SDoug Rabson *diffp = off; 1624a4f67738SDoug Rabson else 1625a4f67738SDoug Rabson *diffp = diff; 1626d254af07SMatthew Dillon *sym = (c_linker_sym_t) best; 1627a4f67738SDoug Rabson 16282f22b3ffSMatthew D Fleming return (0); 1629a4f67738SDoug Rabson } 1630f41325dbSPeter Wemm 1631f41325dbSPeter Wemm /* 1632f41325dbSPeter Wemm * Look up a linker set on an ELF system. 1633f41325dbSPeter Wemm */ 1634f41325dbSPeter Wemm static int 1635f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name, 1636f41325dbSPeter Wemm void ***startp, void ***stopp, int *countp) 1637f41325dbSPeter Wemm { 1638f41325dbSPeter Wemm c_linker_sym_t sym; 1639f41325dbSPeter Wemm linker_symval_t symval; 1640f41325dbSPeter Wemm char *setsym; 1641f41325dbSPeter Wemm void **start, **stop; 1642f41325dbSPeter Wemm int len, error = 0, count; 1643f41325dbSPeter Wemm 1644f41325dbSPeter Wemm len = strlen(name) + sizeof("__start_set_"); /* sizeof includes \0 */ 1645a163d034SWarner Losh setsym = malloc(len, M_LINKER, M_WAITOK); 1646f41325dbSPeter Wemm 1647f41325dbSPeter Wemm /* get address of first entry */ 1648f41325dbSPeter Wemm snprintf(setsym, len, "%s%s", "__start_set_", name); 1649f41325dbSPeter Wemm error = link_elf_lookup_symbol(lf, setsym, &sym); 16502f22b3ffSMatthew D Fleming if (error != 0) 1651f41325dbSPeter Wemm goto out; 1652f41325dbSPeter Wemm link_elf_symbol_values(lf, sym, &symval); 1653f41325dbSPeter Wemm if (symval.value == 0) { 1654f41325dbSPeter Wemm error = ESRCH; 1655f41325dbSPeter Wemm goto out; 1656f41325dbSPeter Wemm } 1657f41325dbSPeter Wemm start = (void **)symval.value; 1658f41325dbSPeter Wemm 1659f41325dbSPeter Wemm /* get address of last entry */ 1660f41325dbSPeter Wemm snprintf(setsym, len, "%s%s", "__stop_set_", name); 1661f41325dbSPeter Wemm error = link_elf_lookup_symbol(lf, setsym, &sym); 16622f22b3ffSMatthew D Fleming if (error != 0) 1663f41325dbSPeter Wemm goto out; 1664f41325dbSPeter Wemm link_elf_symbol_values(lf, sym, &symval); 1665f41325dbSPeter Wemm if (symval.value == 0) { 1666f41325dbSPeter Wemm error = ESRCH; 1667f41325dbSPeter Wemm goto out; 1668f41325dbSPeter Wemm } 1669f41325dbSPeter Wemm stop = (void **)symval.value; 1670f41325dbSPeter Wemm 1671f41325dbSPeter Wemm /* and the number of entries */ 1672f41325dbSPeter Wemm count = stop - start; 1673f41325dbSPeter Wemm 1674f41325dbSPeter Wemm /* and copy out */ 16752f22b3ffSMatthew D Fleming if (startp != NULL) 1676f41325dbSPeter Wemm *startp = start; 16772f22b3ffSMatthew D Fleming if (stopp != NULL) 1678f41325dbSPeter Wemm *stopp = stop; 16792f22b3ffSMatthew D Fleming if (countp != NULL) 1680f41325dbSPeter Wemm *countp = count; 1681f41325dbSPeter Wemm 1682f41325dbSPeter Wemm out: 1683f41325dbSPeter Wemm free(setsym, M_LINKER); 16842f22b3ffSMatthew D Fleming return (error); 1685f41325dbSPeter Wemm } 1686bb9fe9ddSBrian Feldman 1687bb9fe9ddSBrian Feldman static int 1688bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file, 16892f22b3ffSMatthew D Fleming int (*callback)(const char *, void *), void *opaque) 16902f22b3ffSMatthew D Fleming { 1691bb9fe9ddSBrian Feldman elf_file_t ef = (elf_file_t)file; 1692bb9fe9ddSBrian Feldman const Elf_Sym *symp; 1693bb9fe9ddSBrian Feldman int i, error; 1694bb9fe9ddSBrian Feldman 1695bb9fe9ddSBrian Feldman /* Exhaustive search */ 1696bb9fe9ddSBrian Feldman for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1697bb9fe9ddSBrian Feldman if (symp->st_value != 0 && 16987035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 16997035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 1700bb9fe9ddSBrian Feldman error = callback(ef->ddbstrtab + symp->st_name, opaque); 17012f22b3ffSMatthew D Fleming if (error != 0) 1702bb9fe9ddSBrian Feldman return (error); 1703bb9fe9ddSBrian Feldman } 1704bb9fe9ddSBrian Feldman } 1705bb9fe9ddSBrian Feldman return (0); 1706bb9fe9ddSBrian Feldman } 170784201059SMarcel Moolenaar 1708a2024a3eSJohn Birrell static int 1709a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file, 1710a2024a3eSJohn Birrell linker_function_nameval_callback_t callback, void *opaque) 1711a2024a3eSJohn Birrell { 1712a2024a3eSJohn Birrell linker_symval_t symval; 1713a2024a3eSJohn Birrell elf_file_t ef = (elf_file_t)file; 1714a2024a3eSJohn Birrell const Elf_Sym* symp; 1715a2024a3eSJohn Birrell int i, error; 1716a2024a3eSJohn Birrell 1717a2024a3eSJohn Birrell /* Exhaustive search */ 1718a2024a3eSJohn Birrell for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1719a2024a3eSJohn Birrell if (symp->st_value != 0 && 17207035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 17217035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 17222f22b3ffSMatthew D Fleming error = link_elf_symbol_values(file, 17232f22b3ffSMatthew D Fleming (c_linker_sym_t) symp, &symval); 17242f22b3ffSMatthew D Fleming if (error != 0) 1725a2024a3eSJohn Birrell return (error); 1726a2024a3eSJohn Birrell error = callback(file, i, &symval, opaque); 17272f22b3ffSMatthew D Fleming if (error != 0) 1728a2024a3eSJohn Birrell return (error); 1729a2024a3eSJohn Birrell } 1730a2024a3eSJohn Birrell } 1731a2024a3eSJohn Birrell return (0); 1732a2024a3eSJohn Birrell } 1733a2024a3eSJohn Birrell 1734169d513cSJake Burkholder const Elf_Sym * 1735757686b1SMarcel Moolenaar elf_get_sym(linker_file_t lf, Elf_Size symidx) 1736169d513cSJake Burkholder { 1737169d513cSJake Burkholder elf_file_t ef = (elf_file_t)lf; 1738169d513cSJake Burkholder 1739169d513cSJake Burkholder if (symidx >= ef->nchains) 1740169d513cSJake Burkholder return (NULL); 1741169d513cSJake Burkholder return (ef->symtab + symidx); 1742169d513cSJake Burkholder } 1743169d513cSJake Burkholder 174447f75012SMarcel Moolenaar const char * 1745757686b1SMarcel Moolenaar elf_get_symname(linker_file_t lf, Elf_Size symidx) 174647f75012SMarcel Moolenaar { 174747f75012SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 174847f75012SMarcel Moolenaar const Elf_Sym *sym; 174947f75012SMarcel Moolenaar 175047f75012SMarcel Moolenaar if (symidx >= ef->nchains) 175147f75012SMarcel Moolenaar return (NULL); 175247f75012SMarcel Moolenaar sym = ef->symtab + symidx; 175347f75012SMarcel Moolenaar return (ef->strtab + sym->st_name); 175447f75012SMarcel Moolenaar } 175547f75012SMarcel Moolenaar 1756d297ad16SMarcel Moolenaar /* 1757d297ad16SMarcel Moolenaar * Symbol lookup function that can be used when the symbol index is known (ie 1758d297ad16SMarcel Moolenaar * in relocations). It uses the symbol index instead of doing a fully fledged 1759d297ad16SMarcel Moolenaar * hash table based lookup when such is valid. For example for local symbols. 1760d297ad16SMarcel Moolenaar * This is not only more efficient, it's also more correct. It's not always 1761d297ad16SMarcel Moolenaar * the case that the symbol can be found through the hash table. 1762d297ad16SMarcel Moolenaar */ 1763cff8c6f2SKonstantin Belousov static int 1764cff8c6f2SKonstantin Belousov elf_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res) 1765d297ad16SMarcel Moolenaar { 1766d297ad16SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 1767d297ad16SMarcel Moolenaar const Elf_Sym *sym; 1768d297ad16SMarcel Moolenaar const char *symbol; 176947813f5dSMikolaj Golub Elf_Addr addr, start, base; 1770d297ad16SMarcel Moolenaar 1771d297ad16SMarcel Moolenaar /* Don't even try to lookup the symbol if the index is bogus. */ 1772cff8c6f2SKonstantin Belousov if (symidx >= ef->nchains) { 1773cff8c6f2SKonstantin Belousov *res = 0; 1774cff8c6f2SKonstantin Belousov return (EINVAL); 1775cff8c6f2SKonstantin Belousov } 1776d297ad16SMarcel Moolenaar 1777d297ad16SMarcel Moolenaar sym = ef->symtab + symidx; 1778d297ad16SMarcel Moolenaar 1779d297ad16SMarcel Moolenaar /* 1780d297ad16SMarcel Moolenaar * Don't do a full lookup when the symbol is local. It may even 1781d297ad16SMarcel Moolenaar * fail because it may not be found through the hash table. 1782d297ad16SMarcel Moolenaar */ 1783d297ad16SMarcel Moolenaar if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) { 1784d297ad16SMarcel Moolenaar /* Force lookup failure when we have an insanity. */ 1785cff8c6f2SKonstantin Belousov if (sym->st_shndx == SHN_UNDEF || sym->st_value == 0) { 1786cff8c6f2SKonstantin Belousov *res = 0; 1787cff8c6f2SKonstantin Belousov return (EINVAL); 1788cff8c6f2SKonstantin Belousov } 1789cff8c6f2SKonstantin Belousov *res = ((Elf_Addr)ef->address + sym->st_value); 1790d297ad16SMarcel Moolenaar return (0); 1791d297ad16SMarcel Moolenaar } 1792d297ad16SMarcel Moolenaar 1793d297ad16SMarcel Moolenaar /* 1794d297ad16SMarcel Moolenaar * XXX we can avoid doing a hash table based lookup for global 1795d297ad16SMarcel Moolenaar * symbols as well. This however is not always valid, so we'll 1796d297ad16SMarcel Moolenaar * just do it the hard way for now. Performance tweaks can 1797d297ad16SMarcel Moolenaar * always be added. 1798d297ad16SMarcel Moolenaar */ 1799d297ad16SMarcel Moolenaar 1800d297ad16SMarcel Moolenaar symbol = ef->strtab + sym->st_name; 1801d297ad16SMarcel Moolenaar 1802d297ad16SMarcel Moolenaar /* Force a lookup failure if the symbol name is bogus. */ 1803cff8c6f2SKonstantin Belousov if (*symbol == 0) { 1804cff8c6f2SKonstantin Belousov *res = 0; 1805cff8c6f2SKonstantin Belousov return (EINVAL); 1806cff8c6f2SKonstantin Belousov } 1807d297ad16SMarcel Moolenaar 180847813f5dSMikolaj Golub addr = ((Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps)); 1809e8e0fac5SMark Johnston if (addr == 0 && ELF_ST_BIND(sym->st_info) != STB_WEAK) { 1810e8e0fac5SMark Johnston *res = 0; 1811e8e0fac5SMark Johnston return (EINVAL); 1812e8e0fac5SMark Johnston } 181347813f5dSMikolaj Golub 181447813f5dSMikolaj Golub if (elf_set_find(&set_pcpu_list, addr, &start, &base)) 181547813f5dSMikolaj Golub addr = addr - start + base; 181647813f5dSMikolaj Golub #ifdef VIMAGE 181747813f5dSMikolaj Golub else if (elf_set_find(&set_vnet_list, addr, &start, &base)) 181847813f5dSMikolaj Golub addr = addr - start + base; 181947813f5dSMikolaj Golub #endif 1820cff8c6f2SKonstantin Belousov *res = addr; 1821cff8c6f2SKonstantin Belousov return (0); 1822d297ad16SMarcel Moolenaar } 18237251b4bfSJake Burkholder 18247251b4bfSJake Burkholder static void 18257251b4bfSJake Burkholder link_elf_reloc_local(linker_file_t lf) 18267251b4bfSJake Burkholder { 18277251b4bfSJake Burkholder const Elf_Rel *rellim; 18287251b4bfSJake Burkholder const Elf_Rel *rel; 18297251b4bfSJake Burkholder const Elf_Rela *relalim; 18307251b4bfSJake Burkholder const Elf_Rela *rela; 18317251b4bfSJake Burkholder elf_file_t ef = (elf_file_t)lf; 18327251b4bfSJake Burkholder 18337251b4bfSJake Burkholder /* Perform relocations without addend if there are any: */ 18347251b4bfSJake Burkholder if ((rel = ef->rel) != NULL) { 18357251b4bfSJake Burkholder rellim = (const Elf_Rel *)((const char *)ef->rel + ef->relsize); 18367251b4bfSJake Burkholder while (rel < rellim) { 18372f22b3ffSMatthew D Fleming elf_reloc_local(lf, (Elf_Addr)ef->address, rel, 18382f22b3ffSMatthew D Fleming ELF_RELOC_REL, elf_lookup); 18397251b4bfSJake Burkholder rel++; 18407251b4bfSJake Burkholder } 18417251b4bfSJake Burkholder } 18427251b4bfSJake Burkholder 18437251b4bfSJake Burkholder /* Perform relocations with addend if there are any: */ 18447251b4bfSJake Burkholder if ((rela = ef->rela) != NULL) { 18452f22b3ffSMatthew D Fleming relalim = (const Elf_Rela *) 18462f22b3ffSMatthew D Fleming ((const char *)ef->rela + ef->relasize); 18477251b4bfSJake Burkholder while (rela < relalim) { 18482f22b3ffSMatthew D Fleming elf_reloc_local(lf, (Elf_Addr)ef->address, rela, 18492f22b3ffSMatthew D Fleming ELF_RELOC_RELA, elf_lookup); 18507251b4bfSJake Burkholder rela++; 18517251b4bfSJake Burkholder } 18527251b4bfSJake Burkholder } 18537251b4bfSJake Burkholder } 185400a5db46SStacey Son 185500a5db46SStacey Son static long 185600a5db46SStacey Son link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab) 185700a5db46SStacey Son { 185800a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 185900a5db46SStacey Son 186000a5db46SStacey Son *symtab = ef->ddbsymtab; 186100a5db46SStacey Son 186200a5db46SStacey Son if (*symtab == NULL) 186300a5db46SStacey Son return (0); 186400a5db46SStacey Son 186500a5db46SStacey Son return (ef->ddbsymcnt); 186600a5db46SStacey Son } 186700a5db46SStacey Son 186800a5db46SStacey Son static long 186900a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab) 187000a5db46SStacey Son { 187100a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 187200a5db46SStacey Son 187300a5db46SStacey Son *strtab = ef->ddbstrtab; 187400a5db46SStacey Son 187500a5db46SStacey Son if (*strtab == NULL) 187600a5db46SStacey Son return (0); 187700a5db46SStacey Son 187800a5db46SStacey Son return (ef->ddbstrcnt); 187900a5db46SStacey Son } 18807035cf14SKonstantin Belousov 18819411e24dSBrandon Bergren #if defined(__i386__) || defined(__amd64__) || defined(__aarch64__) || defined(__powerpc__) 188236716fe2SMark Johnston /* 188336716fe2SMark Johnston * Use this lookup routine when performing relocations early during boot. 188436716fe2SMark Johnston * The generic lookup routine depends on kobj, which is not initialized 188536716fe2SMark Johnston * at that point. 188636716fe2SMark Johnston */ 188736716fe2SMark Johnston static int 188836716fe2SMark Johnston elf_lookup_ifunc(linker_file_t lf, Elf_Size symidx, int deps __unused, 188936716fe2SMark Johnston Elf_Addr *res) 189036716fe2SMark Johnston { 189136716fe2SMark Johnston elf_file_t ef; 189236716fe2SMark Johnston const Elf_Sym *symp; 189336716fe2SMark Johnston caddr_t val; 189436716fe2SMark Johnston 189536716fe2SMark Johnston ef = (elf_file_t)lf; 189636716fe2SMark Johnston symp = ef->symtab + symidx; 189736716fe2SMark Johnston if (ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC) { 189836716fe2SMark Johnston val = (caddr_t)ef->address + symp->st_value; 189936716fe2SMark Johnston *res = ((Elf_Addr (*)(void))val)(); 190036716fe2SMark Johnston return (0); 190136716fe2SMark Johnston } 190236716fe2SMark Johnston return (ENOENT); 190336716fe2SMark Johnston } 190436716fe2SMark Johnston 19057035cf14SKonstantin Belousov void 19067035cf14SKonstantin Belousov link_elf_ireloc(caddr_t kmdp) 19077035cf14SKonstantin Belousov { 19087035cf14SKonstantin Belousov struct elf_file eff; 19097035cf14SKonstantin Belousov elf_file_t ef; 19107035cf14SKonstantin Belousov 19117035cf14SKonstantin Belousov ef = &eff; 191255c9d75eSKonstantin Belousov 19133f102f58SMateusz Guzik bzero_early(ef, sizeof(*ef)); 191455c9d75eSKonstantin Belousov 19157035cf14SKonstantin Belousov ef->modptr = kmdp; 19167035cf14SKonstantin Belousov ef->dynamic = (Elf_Dyn *)&_DYNAMIC; 19179411e24dSBrandon Bergren 19189411e24dSBrandon Bergren #ifdef RELOCATABLE_KERNEL 19199411e24dSBrandon Bergren ef->address = (caddr_t) (__startkernel - KERNBASE); 19209411e24dSBrandon Bergren #else 19217035cf14SKonstantin Belousov ef->address = 0; 19229411e24dSBrandon Bergren #endif 19239411e24dSBrandon Bergren parse_dynamic(ef); 19249411e24dSBrandon Bergren 19257035cf14SKonstantin Belousov link_elf_preload_parse_symbols(ef); 192636716fe2SMark Johnston relocate_file1(ef, elf_lookup_ifunc, elf_reloc, true); 19277035cf14SKonstantin Belousov } 1928fcf7a481SAndrew Turner 1929fcf7a481SAndrew Turner #if defined(__aarch64__) || defined(__amd64__) 1930fcf7a481SAndrew Turner void 1931fcf7a481SAndrew Turner link_elf_late_ireloc(void) 1932fcf7a481SAndrew Turner { 1933fcf7a481SAndrew Turner elf_file_t ef; 1934fcf7a481SAndrew Turner 1935fcf7a481SAndrew Turner KASSERT(linker_kernel_file != NULL, 1936fcf7a481SAndrew Turner ("link_elf_late_ireloc: No kernel linker file found")); 1937fcf7a481SAndrew Turner ef = (elf_file_t)linker_kernel_file; 1938fcf7a481SAndrew Turner 1939fcf7a481SAndrew Turner relocate_file1(ef, elf_lookup_ifunc, elf_reloc_late, true); 1940fcf7a481SAndrew Turner } 1941fcf7a481SAndrew Turner #endif 19427035cf14SKonstantin Belousov #endif 1943