xref: /freebsd/sys/kern/link_elf_obj.c (revision 4d213c595ac3247a85cea5d3ea521db14151a427)
1a4f67738SDoug Rabson /*-
24d846d26SWarner Losh  * SPDX-License-Identifier: BSD-2-Clause
38a36da99SPedro F. Giffuni  *
4326e27d8SDoug Rabson  * Copyright (c) 1998-2000 Doug Rabson
5e9eabf59SPeter Wemm  * Copyright (c) 2004 Peter Wemm
6a4f67738SDoug Rabson  * All rights reserved.
7a4f67738SDoug Rabson  *
8a4f67738SDoug Rabson  * Redistribution and use in source and binary forms, with or without
9a4f67738SDoug Rabson  * modification, are permitted provided that the following conditions
10a4f67738SDoug Rabson  * are met:
11a4f67738SDoug Rabson  * 1. Redistributions of source code must retain the above copyright
12a4f67738SDoug Rabson  *    notice, this list of conditions and the following disclaimer.
13a4f67738SDoug Rabson  * 2. Redistributions in binary form must reproduce the above copyright
14a4f67738SDoug Rabson  *    notice, this list of conditions and the following disclaimer in the
15a4f67738SDoug Rabson  *    documentation and/or other materials provided with the distribution.
16a4f67738SDoug Rabson  *
17a4f67738SDoug Rabson  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18a4f67738SDoug Rabson  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19a4f67738SDoug Rabson  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20a4f67738SDoug Rabson  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21a4f67738SDoug Rabson  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22a4f67738SDoug Rabson  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23a4f67738SDoug Rabson  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24a4f67738SDoug Rabson  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25a4f67738SDoug Rabson  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26a4f67738SDoug Rabson  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27a4f67738SDoug Rabson  * SUCH DAMAGE.
28a4f67738SDoug Rabson  */
29a4f67738SDoug Rabson 
30326e27d8SDoug Rabson #include "opt_ddb.h"
31326e27d8SDoug Rabson 
32a4f67738SDoug Rabson #include <sys/param.h>
33a4f67738SDoug Rabson #include <sys/systm.h>
34fbf2a778SKonstantin Belousov #include <sys/fcntl.h>
35fb919e4dSMark Murray #include <sys/kernel.h>
36fb919e4dSMark Murray #include <sys/lock.h>
37a4f67738SDoug Rabson #include <sys/malloc.h>
38fbf2a778SKonstantin Belousov #include <sys/linker.h>
39c0b824f9SDima Dorfman #include <sys/mutex.h>
4062d615d5SJohn Baldwin #include <sys/mount.h>
41a4f67738SDoug Rabson #include <sys/namei.h>
42fbf2a778SKonstantin Belousov #include <sys/proc.h>
43fbf2a778SKonstantin Belousov #include <sys/rwlock.h>
44ecd8245eSKonstantin Belousov #include <sys/sysctl.h>
45a4f67738SDoug Rabson #include <sys/vnode.h>
46fb919e4dSMark Murray 
47a4f67738SDoug Rabson #include <machine/elf.h>
48a4f67738SDoug Rabson 
49eddfbb76SRobert Watson #include <net/vnet.h>
50eddfbb76SRobert Watson 
51aed55708SRobert Watson #include <security/mac/mac_framework.h>
52aed55708SRobert Watson 
53fe3db7c7SDoug Rabson #include <vm/vm.h>
54fe3db7c7SDoug Rabson #include <vm/vm_param.h>
55fe3db7c7SDoug Rabson #include <vm/pmap.h>
56fbf2a778SKonstantin Belousov #include <vm/vm_extern.h>
57fbf2a778SKonstantin Belousov #include <vm/vm_kern.h>
58fe3db7c7SDoug Rabson #include <vm/vm_map.h>
59fbf2a778SKonstantin Belousov #include <vm/vm_object.h>
60fbf2a778SKonstantin Belousov #include <vm/vm_page.h>
61fbf2a778SKonstantin Belousov #include <vm/vm_pager.h>
62fb919e4dSMark Murray 
63f9980387SPeter Wemm #include <sys/link_elf.h>
64fe3db7c7SDoug Rabson 
65a2024a3eSJohn Birrell #ifdef DDB_CTF
6622bbc4b2SXin LI #include <contrib/zlib/zlib.h>
67a2024a3eSJohn Birrell #endif
68a2024a3eSJohn Birrell 
69326e27d8SDoug Rabson #include "linker_if.h"
70a4f67738SDoug Rabson 
71e9eabf59SPeter Wemm typedef struct {
72e9eabf59SPeter Wemm 	void		*addr;
7320947801SPeter Wemm 	Elf_Off		size;
741d9eae9fSMark Johnston 	int		flags;	/* Section flags. */
751d9eae9fSMark Johnston 	int		sec;	/* Original section number. */
76e9eabf59SPeter Wemm 	char		*name;
77e9eabf59SPeter Wemm } Elf_progent;
78e9eabf59SPeter Wemm 
79e9eabf59SPeter Wemm typedef struct {
80e9eabf59SPeter Wemm 	Elf_Rel		*rel;
8120947801SPeter Wemm 	int		nrel;
82e9eabf59SPeter Wemm 	int		sec;
83e9eabf59SPeter Wemm } Elf_relent;
84e9eabf59SPeter Wemm 
85e9eabf59SPeter Wemm typedef struct {
86e9eabf59SPeter Wemm 	Elf_Rela	*rela;
8720947801SPeter Wemm 	int		nrela;
88e9eabf59SPeter Wemm 	int		sec;
89e9eabf59SPeter Wemm } Elf_relaent;
90e9eabf59SPeter Wemm 
91a4f67738SDoug Rabson typedef struct elf_file {
92326e27d8SDoug Rabson 	struct linker_file lf;		/* Common fields */
9320947801SPeter Wemm 
9470b7ffeeSIan Dowse 	int		preloaded;
95fe3db7c7SDoug Rabson 	caddr_t		address;	/* Relocation address */
96fe3db7c7SDoug Rabson 	vm_object_t	object;		/* VM object to hold file pages */
97e9eabf59SPeter Wemm 	Elf_Shdr	*e_shdr;
98e9eabf59SPeter Wemm 
99e9eabf59SPeter Wemm 	Elf_progent	*progtab;
100d821d364SPedro F. Giffuni 	u_int		nprogtab;
101e9eabf59SPeter Wemm 
102e9eabf59SPeter Wemm 	Elf_relaent	*relatab;
103d821d364SPedro F. Giffuni 	u_int		nrelatab;
104e9eabf59SPeter Wemm 
105e9eabf59SPeter Wemm 	Elf_relent	*reltab;
106b474c780SDavid E. O'Brien 	int		nreltab;
107e9eabf59SPeter Wemm 
108e9eabf59SPeter Wemm 	Elf_Sym		*ddbsymtab;	/* The symbol table we are using */
1092d636ab0SPeter Wemm 	long		ddbsymcnt;	/* Number of symbols */
1102d636ab0SPeter Wemm 	caddr_t		ddbstrtab;	/* String table */
1112d636ab0SPeter Wemm 	long		ddbstrcnt;	/* number of bytes in string table */
112e9eabf59SPeter Wemm 
113e9eabf59SPeter Wemm 	caddr_t		shstrtab;	/* Section name string table */
114e9eabf59SPeter Wemm 	long		shstrcnt;	/* number of bytes in string table */
115e9eabf59SPeter Wemm 
116a2024a3eSJohn Birrell 	caddr_t		ctftab;		/* CTF table */
117a2024a3eSJohn Birrell 	long		ctfcnt;		/* number of bytes in CTF table */
118a2024a3eSJohn Birrell 	caddr_t		ctfoff;		/* CTF offset table */
119a2024a3eSJohn Birrell 	caddr_t		typoff;		/* Type offset table */
120a2024a3eSJohn Birrell 	long		typlen;		/* Number of type entries. */
121a2024a3eSJohn Birrell 
122a4f67738SDoug Rabson } *elf_file_t;
123a4f67738SDoug Rabson 
124a2024a3eSJohn Birrell #include <kern/kern_ctf.c>
125a2024a3eSJohn Birrell 
12654823af2SPeter Wemm static int	link_elf_link_preload(linker_class_t cls,
127326e27d8SDoug Rabson 		    const char *, linker_file_t *);
12854823af2SPeter Wemm static int	link_elf_link_preload_finish(linker_file_t);
12954823af2SPeter Wemm static int	link_elf_load_file(linker_class_t, const char *, linker_file_t *);
130326e27d8SDoug Rabson static int	link_elf_lookup_symbol(linker_file_t, const char *,
131326e27d8SDoug Rabson 		    c_linker_sym_t *);
13295c20fafSKonstantin Belousov static int	link_elf_lookup_debug_symbol(linker_file_t, const char *,
13395c20fafSKonstantin Belousov 		    c_linker_sym_t *);
134c21bc6f3SBojan Novković static int	link_elf_lookup_debug_symbol_ctf(linker_file_t lf,
135c21bc6f3SBojan Novković 		    const char *name, c_linker_sym_t *sym, linker_ctf_t *lc);
136e9eabf59SPeter Wemm static int	link_elf_symbol_values(linker_file_t, c_linker_sym_t,
137e9eabf59SPeter Wemm 		    linker_symval_t *);
13895c20fafSKonstantin Belousov static int	link_elf_debug_symbol_values(linker_file_t, c_linker_sym_t,
13995c20fafSKonstantin Belousov 		    linker_symval_t *);
140326e27d8SDoug Rabson static int	link_elf_search_symbol(linker_file_t, caddr_t value,
141326e27d8SDoug Rabson 		    c_linker_sym_t *sym, long *diffp);
142326e27d8SDoug Rabson 
143326e27d8SDoug Rabson static void	link_elf_unload_file(linker_file_t);
144f41325dbSPeter Wemm static int	link_elf_lookup_set(linker_file_t, const char *,
145f41325dbSPeter Wemm 		    void ***, void ***, int *);
146bb9fe9ddSBrian Feldman static int	link_elf_each_function_name(linker_file_t,
147e9eabf59SPeter Wemm 		    int (*)(const char *, void *), void *);
148a2024a3eSJohn Birrell static int	link_elf_each_function_nameval(linker_file_t,
149a2024a3eSJohn Birrell 				linker_function_nameval_callback_t,
150a2024a3eSJohn Birrell 				void *);
151289ead7cSKonstantin Belousov static int	link_elf_reloc_local(linker_file_t, bool);
152e76f11f4SAndriy Gapon static long	link_elf_symtab_get(linker_file_t, const Elf_Sym **);
15300a5db46SStacey Son static long	link_elf_strtab_get(linker_file_t, caddr_t *);
154110113bcSZhenlei Huang #ifdef VIMAGE
155110113bcSZhenlei Huang static void	link_elf_propagate_vnets(linker_file_t);
156110113bcSZhenlei Huang #endif
157326e27d8SDoug Rabson 
158cff8c6f2SKonstantin Belousov static int	elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps,
159cff8c6f2SKonstantin Belousov 		    Elf_Addr *);
160e9eabf59SPeter Wemm 
161326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = {
162326e27d8SDoug Rabson 	KOBJMETHOD(linker_lookup_symbol,	link_elf_lookup_symbol),
16395c20fafSKonstantin Belousov 	KOBJMETHOD(linker_lookup_debug_symbol,	link_elf_lookup_debug_symbol),
164c21bc6f3SBojan Novković 	KOBJMETHOD(linker_lookup_debug_symbol_ctf, link_elf_lookup_debug_symbol_ctf),
165326e27d8SDoug Rabson 	KOBJMETHOD(linker_symbol_values,	link_elf_symbol_values),
16695c20fafSKonstantin Belousov 	KOBJMETHOD(linker_debug_symbol_values,	link_elf_debug_symbol_values),
167326e27d8SDoug Rabson 	KOBJMETHOD(linker_search_symbol,	link_elf_search_symbol),
168326e27d8SDoug Rabson 	KOBJMETHOD(linker_unload,		link_elf_unload_file),
16954823af2SPeter Wemm 	KOBJMETHOD(linker_load_file,		link_elf_load_file),
17054823af2SPeter Wemm 	KOBJMETHOD(linker_link_preload,		link_elf_link_preload),
17154823af2SPeter Wemm 	KOBJMETHOD(linker_link_preload_finish,	link_elf_link_preload_finish),
172f41325dbSPeter Wemm 	KOBJMETHOD(linker_lookup_set,		link_elf_lookup_set),
173bb9fe9ddSBrian Feldman 	KOBJMETHOD(linker_each_function_name,	link_elf_each_function_name),
174a2024a3eSJohn Birrell 	KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval),
175a2024a3eSJohn Birrell 	KOBJMETHOD(linker_ctf_get,		link_elf_ctf_get),
176c21bc6f3SBojan Novković 	KOBJMETHOD(linker_ctf_lookup_typename,  link_elf_ctf_lookup_typename),
17700a5db46SStacey Son 	KOBJMETHOD(linker_symtab_get, 		link_elf_symtab_get),
17800a5db46SStacey Son 	KOBJMETHOD(linker_strtab_get, 		link_elf_strtab_get),
179110113bcSZhenlei Huang #ifdef VIMAGE
180110113bcSZhenlei Huang 	KOBJMETHOD(linker_propagate_vnets,	link_elf_propagate_vnets),
181110113bcSZhenlei Huang #endif
1826d775f0bSMark Johnston 	KOBJMETHOD_END
183326e27d8SDoug Rabson };
184326e27d8SDoug Rabson 
185326e27d8SDoug Rabson static struct linker_class link_elf_class = {
186326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32
187e9eabf59SPeter Wemm 	"elf32_obj",
188326e27d8SDoug Rabson #else
189e9eabf59SPeter Wemm 	"elf64_obj",
190326e27d8SDoug Rabson #endif
191326e27d8SDoug Rabson 	link_elf_methods, sizeof(struct elf_file)
192326e27d8SDoug Rabson };
193326e27d8SDoug Rabson 
194ecd8245eSKonstantin Belousov static bool link_elf_obj_leak_locals = true;
195ecd8245eSKonstantin Belousov SYSCTL_BOOL(_debug, OID_AUTO, link_elf_obj_leak_locals,
196ecd8245eSKonstantin Belousov     CTLFLAG_RWTUN, &link_elf_obj_leak_locals, 0,
197ecd8245eSKonstantin Belousov     "Allow local symbols to participate in global module symbol resolution");
198ecd8245eSKonstantin Belousov 
199326e27d8SDoug Rabson static int	relocate_file(elf_file_t ef);
2002832cd54SKonstantin Belousov static void	elf_obj_cleanup_globals_cache(elf_file_t);
201a4f67738SDoug Rabson 
202a4f67738SDoug Rabson static void
link_elf_error(const char * filename,const char * s)203552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s)
2041720979bSMarcel Moolenaar {
205552f9f63SEdwin Groothuis 	if (filename == NULL)
2061720979bSMarcel Moolenaar 		printf("kldload: %s\n", s);
207552f9f63SEdwin Groothuis 	else
208552f9f63SEdwin Groothuis 		printf("kldload: %s: %s\n", filename, s);
2091720979bSMarcel Moolenaar }
2101720979bSMarcel Moolenaar 
2111720979bSMarcel Moolenaar static void
link_elf_init(void * arg)212a4f67738SDoug Rabson link_elf_init(void *arg)
213a4f67738SDoug Rabson {
214a4f67738SDoug Rabson 
215326e27d8SDoug Rabson 	linker_add_class(&link_elf_class);
216a4f67738SDoug Rabson }
217891cf3edSEd Maste SYSINIT(link_elf_obj, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, NULL);
218a4f67738SDoug Rabson 
2191d9eae9fSMark Johnston static void
link_elf_protect_range(elf_file_t ef,vm_offset_t start,vm_offset_t end,vm_prot_t prot)2201d9eae9fSMark Johnston link_elf_protect_range(elf_file_t ef, vm_offset_t start, vm_offset_t end,
2211d9eae9fSMark Johnston     vm_prot_t prot)
2221d9eae9fSMark Johnston {
2231d9eae9fSMark Johnston 	int error __unused;
2241d9eae9fSMark Johnston 
2251d9eae9fSMark Johnston 	KASSERT(start <= end && start >= (vm_offset_t)ef->address &&
2261d9eae9fSMark Johnston 	    end <= round_page((vm_offset_t)ef->address + ef->lf.size),
2271d9eae9fSMark Johnston 	    ("link_elf_protect_range: invalid range %#jx-%#jx",
2281d9eae9fSMark Johnston 	    (uintmax_t)start, (uintmax_t)end));
2291d9eae9fSMark Johnston 
2301d9eae9fSMark Johnston 	if (start == end)
2311d9eae9fSMark Johnston 		return;
232f822c9e2SMark Johnston 	if (ef->preloaded) {
233f822c9e2SMark Johnston #ifdef __amd64__
234f822c9e2SMark Johnston 		error = pmap_change_prot(start, end - start, prot);
235f822c9e2SMark Johnston 		KASSERT(error == 0,
236f822c9e2SMark Johnston 		    ("link_elf_protect_range: pmap_change_prot() returned %d",
237f822c9e2SMark Johnston 		    error));
238f822c9e2SMark Johnston #endif
239f822c9e2SMark Johnston 		return;
240f822c9e2SMark Johnston 	}
2410659df6fSKonstantin Belousov 	error = vm_map_protect(kernel_map, start, end, prot, 0,
2420659df6fSKonstantin Belousov 	    VM_MAP_PROTECT_SET_PROT);
2431d9eae9fSMark Johnston 	KASSERT(error == KERN_SUCCESS,
2441d9eae9fSMark Johnston 	    ("link_elf_protect_range: vm_map_protect() returned %d", error));
2451d9eae9fSMark Johnston }
2461d9eae9fSMark Johnston 
2471d9eae9fSMark Johnston /*
2481d9eae9fSMark Johnston  * Restrict permissions on linker file memory based on section flags.
2491d9eae9fSMark Johnston  * Sections need not be page-aligned, so overlap within a page is possible.
2501d9eae9fSMark Johnston  */
2511d9eae9fSMark Johnston static void
link_elf_protect(elf_file_t ef)2521d9eae9fSMark Johnston link_elf_protect(elf_file_t ef)
2531d9eae9fSMark Johnston {
2541d9eae9fSMark Johnston 	vm_offset_t end, segend, segstart, start;
2551d9eae9fSMark Johnston 	vm_prot_t gapprot, prot, segprot;
2561d9eae9fSMark Johnston 	int i;
2571d9eae9fSMark Johnston 
2581d9eae9fSMark Johnston 	/*
2591d9eae9fSMark Johnston 	 * If the file was preloaded, the last page may contain other preloaded
2601d9eae9fSMark Johnston 	 * data which may need to be writeable.  ELF files are always
2611d9eae9fSMark Johnston 	 * page-aligned, but other preloaded data, such as entropy or CPU
2621d9eae9fSMark Johnston 	 * microcode may be loaded with a smaller alignment.
2631d9eae9fSMark Johnston 	 */
2641d9eae9fSMark Johnston 	gapprot = ef->preloaded ? VM_PROT_RW : VM_PROT_READ;
2651d9eae9fSMark Johnston 
2661d9eae9fSMark Johnston 	start = end = (vm_offset_t)ef->address;
2671d9eae9fSMark Johnston 	prot = VM_PROT_READ;
2681d9eae9fSMark Johnston 	for (i = 0; i < ef->nprogtab; i++) {
2691d9eae9fSMark Johnston 		/*
2701d9eae9fSMark Johnston 		 * VNET and DPCPU sections have their memory allocated by their
2711d9eae9fSMark Johnston 		 * respective subsystems.
2721d9eae9fSMark Johnston 		 */
2731d9eae9fSMark Johnston 		if (ef->progtab[i].name != NULL && (
2741d9eae9fSMark Johnston #ifdef VIMAGE
2751d9eae9fSMark Johnston 		    strcmp(ef->progtab[i].name, VNET_SETNAME) == 0 ||
2761d9eae9fSMark Johnston #endif
2771d9eae9fSMark Johnston 		    strcmp(ef->progtab[i].name, DPCPU_SETNAME) == 0))
2781d9eae9fSMark Johnston 			continue;
2791d9eae9fSMark Johnston 
2801d9eae9fSMark Johnston 		segstart = trunc_page((vm_offset_t)ef->progtab[i].addr);
2811d9eae9fSMark Johnston 		segend = round_page((vm_offset_t)ef->progtab[i].addr +
2821d9eae9fSMark Johnston 		    ef->progtab[i].size);
2831d9eae9fSMark Johnston 		segprot = VM_PROT_READ;
2841d9eae9fSMark Johnston 		if ((ef->progtab[i].flags & SHF_WRITE) != 0)
2851d9eae9fSMark Johnston 			segprot |= VM_PROT_WRITE;
2861d9eae9fSMark Johnston 		if ((ef->progtab[i].flags & SHF_EXECINSTR) != 0)
2871d9eae9fSMark Johnston 			segprot |= VM_PROT_EXECUTE;
2881d9eae9fSMark Johnston 
2891d9eae9fSMark Johnston 		if (end <= segstart) {
2901d9eae9fSMark Johnston 			/*
2911d9eae9fSMark Johnston 			 * Case 1: there is no overlap between the previous
2921d9eae9fSMark Johnston 			 * segment and this one.  Apply protections to the
2931d9eae9fSMark Johnston 			 * previous segment, and protect the gap between the
2941d9eae9fSMark Johnston 			 * previous and current segments, if any.
2951d9eae9fSMark Johnston 			 */
2961d9eae9fSMark Johnston 			link_elf_protect_range(ef, start, end, prot);
2971d9eae9fSMark Johnston 			link_elf_protect_range(ef, end, segstart, gapprot);
2981d9eae9fSMark Johnston 
2991d9eae9fSMark Johnston 			start = segstart;
3001d9eae9fSMark Johnston 			end = segend;
3011d9eae9fSMark Johnston 			prot = segprot;
3021d9eae9fSMark Johnston 		} else if (start < segstart && end == segend) {
3031d9eae9fSMark Johnston 			/*
3041d9eae9fSMark Johnston 			 * Case 2: the current segment is a subrange of the
3051d9eae9fSMark Johnston 			 * previous segment.  Apply protections to the
3061d9eae9fSMark Johnston 			 * non-overlapping portion of the previous segment.
3071d9eae9fSMark Johnston 			 */
3081d9eae9fSMark Johnston 			link_elf_protect_range(ef, start, segstart, prot);
3091d9eae9fSMark Johnston 
3101d9eae9fSMark Johnston 			start = segstart;
3111d9eae9fSMark Johnston 			prot |= segprot;
3121d9eae9fSMark Johnston 		} else if (end < segend) {
3131d9eae9fSMark Johnston 			/*
3141d9eae9fSMark Johnston 			 * Case 3: there is partial overlap between the previous
3151d9eae9fSMark Johnston 			 * and current segments.  Apply protections to the
3161d9eae9fSMark Johnston 			 * non-overlapping portion of the previous segment, and
3171d9eae9fSMark Johnston 			 * then the overlap, which must use the union of the two
3181d9eae9fSMark Johnston 			 * segments' protections.
3191d9eae9fSMark Johnston 			 */
3201d9eae9fSMark Johnston 			link_elf_protect_range(ef, start, segstart, prot);
3211d9eae9fSMark Johnston 			link_elf_protect_range(ef, segstart, end,
3221d9eae9fSMark Johnston 			    prot | segprot);
3231d9eae9fSMark Johnston 			start = end;
3241d9eae9fSMark Johnston 			end = segend;
3251d9eae9fSMark Johnston 			prot = segprot;
3261d9eae9fSMark Johnston 		} else {
3271d9eae9fSMark Johnston 			/*
3281d9eae9fSMark Johnston 			 * Case 4: the two segments reside in the same page.
3291d9eae9fSMark Johnston 			 */
3301d9eae9fSMark Johnston 			prot |= segprot;
3311d9eae9fSMark Johnston 		}
3321d9eae9fSMark Johnston 	}
3331d9eae9fSMark Johnston 
3341d9eae9fSMark Johnston 	/*
3351d9eae9fSMark Johnston 	 * Fix up the last unprotected segment and trailing data.
3361d9eae9fSMark Johnston 	 */
3371d9eae9fSMark Johnston 	link_elf_protect_range(ef, start, end, prot);
3381d9eae9fSMark Johnston 	link_elf_protect_range(ef, end,
3391d9eae9fSMark Johnston 	    round_page((vm_offset_t)ef->address + ef->lf.size), gapprot);
3401d9eae9fSMark Johnston }
3411d9eae9fSMark Johnston 
342a4f67738SDoug Rabson static int
link_elf_link_preload(linker_class_t cls,const char * filename,linker_file_t * result)343e9eabf59SPeter Wemm link_elf_link_preload(linker_class_t cls, const char *filename,
344e9eabf59SPeter Wemm     linker_file_t *result)
3452d636ab0SPeter Wemm {
34670b7ffeeSIan Dowse 	Elf_Ehdr *hdr;
34770b7ffeeSIan Dowse 	Elf_Shdr *shdr;
34870b7ffeeSIan Dowse 	Elf_Sym *es;
34970b7ffeeSIan Dowse 	void *modptr, *baseptr, *sizeptr;
35070b7ffeeSIan Dowse 	char *type;
35170b7ffeeSIan Dowse 	elf_file_t ef;
35270b7ffeeSIan Dowse 	linker_file_t lf;
35370b7ffeeSIan Dowse 	Elf_Addr off;
35470b7ffeeSIan Dowse 	int error, i, j, pb, ra, rl, shstrindex, symstrindex, symtabindex;
35570b7ffeeSIan Dowse 
35670b7ffeeSIan Dowse 	/* Look to see if we have the file preloaded */
35770b7ffeeSIan Dowse 	modptr = preload_search_by_name(filename);
35870b7ffeeSIan Dowse 	if (modptr == NULL)
35970b7ffeeSIan Dowse 		return ENOENT;
36070b7ffeeSIan Dowse 
36170b7ffeeSIan Dowse 	type = (char *)preload_search_info(modptr, MODINFO_TYPE);
36270b7ffeeSIan Dowse 	baseptr = preload_search_info(modptr, MODINFO_ADDR);
36370b7ffeeSIan Dowse 	sizeptr = preload_search_info(modptr, MODINFO_SIZE);
36470b7ffeeSIan Dowse 	hdr = (Elf_Ehdr *)preload_search_info(modptr, MODINFO_METADATA |
36570b7ffeeSIan Dowse 	    MODINFOMD_ELFHDR);
36670b7ffeeSIan Dowse 	shdr = (Elf_Shdr *)preload_search_info(modptr, MODINFO_METADATA |
36770b7ffeeSIan Dowse 	    MODINFOMD_SHDR);
368*4d213c59SAhmad Khalifa 	if (type == NULL || strcmp(type, preload_modtype_obj) != 0)
369de78ca7eSPeter Wemm 		return (EFTYPE);
37070b7ffeeSIan Dowse 	if (baseptr == NULL || sizeptr == NULL || hdr == NULL ||
37170b7ffeeSIan Dowse 	    shdr == NULL)
37270b7ffeeSIan Dowse 		return (EINVAL);
37370b7ffeeSIan Dowse 
37470b7ffeeSIan Dowse 	lf = linker_make_file(filename, &link_elf_class);
37570b7ffeeSIan Dowse 	if (lf == NULL)
37670b7ffeeSIan Dowse 		return (ENOMEM);
37770b7ffeeSIan Dowse 
37870b7ffeeSIan Dowse 	ef = (elf_file_t)lf;
37970b7ffeeSIan Dowse 	ef->preloaded = 1;
38070b7ffeeSIan Dowse 	ef->address = *(caddr_t *)baseptr;
38170b7ffeeSIan Dowse 	lf->address = *(caddr_t *)baseptr;
38270b7ffeeSIan Dowse 	lf->size = *(size_t *)sizeptr;
38370b7ffeeSIan Dowse 
38470b7ffeeSIan Dowse 	if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS ||
38570b7ffeeSIan Dowse 	    hdr->e_ident[EI_DATA] != ELF_TARG_DATA ||
38670b7ffeeSIan Dowse 	    hdr->e_ident[EI_VERSION] != EV_CURRENT ||
38770b7ffeeSIan Dowse 	    hdr->e_version != EV_CURRENT ||
38870b7ffeeSIan Dowse 	    hdr->e_type != ET_REL ||
38970b7ffeeSIan Dowse 	    hdr->e_machine != ELF_TARG_MACH) {
39070b7ffeeSIan Dowse 		error = EFTYPE;
39170b7ffeeSIan Dowse 		goto out;
39270b7ffeeSIan Dowse 	}
39370b7ffeeSIan Dowse 	ef->e_shdr = shdr;
39470b7ffeeSIan Dowse 
39570b7ffeeSIan Dowse 	/* Scan the section header for information and table sizing. */
39670b7ffeeSIan Dowse 	symtabindex = -1;
39770b7ffeeSIan Dowse 	symstrindex = -1;
39870b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
39970b7ffeeSIan Dowse 		switch (shdr[i].sh_type) {
40070b7ffeeSIan Dowse 		case SHT_PROGBITS:
40170b7ffeeSIan Dowse 		case SHT_NOBITS:
40213f28d96SKonstantin Belousov #ifdef __amd64__
403b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
40413f28d96SKonstantin Belousov #endif
4055e6989baSMark Johnston 		case SHT_INIT_ARRAY:
4065e6989baSMark Johnston 		case SHT_FINI_ARRAY:
40758c4aee0SJohn Baldwin 			/* Ignore sections not loaded by the loader. */
40858c4aee0SJohn Baldwin 			if (shdr[i].sh_addr == 0)
40958c4aee0SJohn Baldwin 				break;
41070b7ffeeSIan Dowse 			ef->nprogtab++;
41170b7ffeeSIan Dowse 			break;
41270b7ffeeSIan Dowse 		case SHT_SYMTAB:
41370b7ffeeSIan Dowse 			symtabindex = i;
41470b7ffeeSIan Dowse 			symstrindex = shdr[i].sh_link;
41570b7ffeeSIan Dowse 			break;
41670b7ffeeSIan Dowse 		case SHT_REL:
41715746ef4SJohn Baldwin 			/*
41815746ef4SJohn Baldwin 			 * Ignore relocation tables for sections not
41915746ef4SJohn Baldwin 			 * loaded by the loader.
42015746ef4SJohn Baldwin 			 */
42115746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
42215746ef4SJohn Baldwin 				break;
423b474c780SDavid E. O'Brien 			ef->nreltab++;
42470b7ffeeSIan Dowse 			break;
42570b7ffeeSIan Dowse 		case SHT_RELA:
42615746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
42715746ef4SJohn Baldwin 				break;
428b474c780SDavid E. O'Brien 			ef->nrelatab++;
42970b7ffeeSIan Dowse 			break;
43070b7ffeeSIan Dowse 		}
43170b7ffeeSIan Dowse 	}
43270b7ffeeSIan Dowse 
43370b7ffeeSIan Dowse 	shstrindex = hdr->e_shstrndx;
43470b7ffeeSIan Dowse 	if (ef->nprogtab == 0 || symstrindex < 0 ||
43570b7ffeeSIan Dowse 	    symstrindex >= hdr->e_shnum ||
43670b7ffeeSIan Dowse 	    shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 ||
43770b7ffeeSIan Dowse 	    shstrindex >= hdr->e_shnum ||
43870b7ffeeSIan Dowse 	    shdr[shstrindex].sh_type != SHT_STRTAB) {
43970b7ffeeSIan Dowse 		printf("%s: bad/missing section headers\n", filename);
44070b7ffeeSIan Dowse 		error = ENOEXEC;
44170b7ffeeSIan Dowse 		goto out;
44270b7ffeeSIan Dowse 	}
44370b7ffeeSIan Dowse 
44470b7ffeeSIan Dowse 	/* Allocate space for tracking the load chunks */
44570b7ffeeSIan Dowse 	if (ef->nprogtab != 0)
44670b7ffeeSIan Dowse 		ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab),
44770b7ffeeSIan Dowse 		    M_LINKER, M_WAITOK | M_ZERO);
448b474c780SDavid E. O'Brien 	if (ef->nreltab != 0)
449b474c780SDavid E. O'Brien 		ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab),
450b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
451b474c780SDavid E. O'Brien 	if (ef->nrelatab != 0)
452b474c780SDavid E. O'Brien 		ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab),
453b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
45470b7ffeeSIan Dowse 	if ((ef->nprogtab != 0 && ef->progtab == NULL) ||
455b474c780SDavid E. O'Brien 	    (ef->nreltab != 0 && ef->reltab == NULL) ||
456b474c780SDavid E. O'Brien 	    (ef->nrelatab != 0 && ef->relatab == NULL)) {
45770b7ffeeSIan Dowse 		error = ENOMEM;
45870b7ffeeSIan Dowse 		goto out;
45970b7ffeeSIan Dowse 	}
46070b7ffeeSIan Dowse 
46170b7ffeeSIan Dowse 	/* XXX, relocate the sh_addr fields saved by the loader. */
46270b7ffeeSIan Dowse 	off = 0;
46370b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
46470b7ffeeSIan Dowse 		if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off))
46570b7ffeeSIan Dowse 			off = shdr[i].sh_addr;
46670b7ffeeSIan Dowse 	}
46770b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
46870b7ffeeSIan Dowse 		if (shdr[i].sh_addr != 0)
46970b7ffeeSIan Dowse 			shdr[i].sh_addr = shdr[i].sh_addr - off +
47070b7ffeeSIan Dowse 			    (Elf_Addr)ef->address;
47170b7ffeeSIan Dowse 	}
47270b7ffeeSIan Dowse 
47370b7ffeeSIan Dowse 	ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
47470b7ffeeSIan Dowse 	ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr;
47570b7ffeeSIan Dowse 	ef->ddbstrcnt = shdr[symstrindex].sh_size;
47670b7ffeeSIan Dowse 	ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr;
47770b7ffeeSIan Dowse 	ef->shstrcnt = shdr[shstrindex].sh_size;
47870b7ffeeSIan Dowse 	ef->shstrtab = (char *)shdr[shstrindex].sh_addr;
47970b7ffeeSIan Dowse 
48070b7ffeeSIan Dowse 	/* Now fill out progtab and the relocation tables. */
48170b7ffeeSIan Dowse 	pb = 0;
48270b7ffeeSIan Dowse 	rl = 0;
48370b7ffeeSIan Dowse 	ra = 0;
48470b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
48570b7ffeeSIan Dowse 		switch (shdr[i].sh_type) {
48670b7ffeeSIan Dowse 		case SHT_PROGBITS:
48770b7ffeeSIan Dowse 		case SHT_NOBITS:
48813f28d96SKonstantin Belousov #ifdef __amd64__
489b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
49013f28d96SKonstantin Belousov #endif
4914aa157ddSMark Johnston 		case SHT_INIT_ARRAY:
4925e6989baSMark Johnston 		case SHT_FINI_ARRAY:
49358c4aee0SJohn Baldwin 			if (shdr[i].sh_addr == 0)
49458c4aee0SJohn Baldwin 				break;
49570b7ffeeSIan Dowse 			ef->progtab[pb].addr = (void *)shdr[i].sh_addr;
49670b7ffeeSIan Dowse 			if (shdr[i].sh_type == SHT_PROGBITS)
49770b7ffeeSIan Dowse 				ef->progtab[pb].name = "<<PROGBITS>>";
49813f28d96SKonstantin Belousov #ifdef __amd64__
499b715d9afSKonstantin Belousov 			else if (shdr[i].sh_type == SHT_X86_64_UNWIND)
50013f28d96SKonstantin Belousov 				ef->progtab[pb].name = "<<UNWIND>>";
50113f28d96SKonstantin Belousov #endif
5025e6989baSMark Johnston 			else if (shdr[i].sh_type == SHT_INIT_ARRAY)
5035e6989baSMark Johnston 				ef->progtab[pb].name = "<<INIT_ARRAY>>";
5045e6989baSMark Johnston 			else if (shdr[i].sh_type == SHT_FINI_ARRAY)
5055e6989baSMark Johnston 				ef->progtab[pb].name = "<<FINI_ARRAY>>";
50670b7ffeeSIan Dowse 			else
50770b7ffeeSIan Dowse 				ef->progtab[pb].name = "<<NOBITS>>";
50870b7ffeeSIan Dowse 			ef->progtab[pb].size = shdr[i].sh_size;
5091d9eae9fSMark Johnston 			ef->progtab[pb].flags = shdr[i].sh_flags;
51070b7ffeeSIan Dowse 			ef->progtab[pb].sec = i;
51170b7ffeeSIan Dowse 			if (ef->shstrtab && shdr[i].sh_name != 0)
51270b7ffeeSIan Dowse 				ef->progtab[pb].name =
51370b7ffeeSIan Dowse 				    ef->shstrtab + shdr[i].sh_name;
51450c202c5SJeff Roberson 			if (ef->progtab[pb].name != NULL &&
5155f67450dSDimitry Andric 			    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) {
51650c202c5SJeff Roberson 				void *dpcpu;
51750c202c5SJeff Roberson 
51850c202c5SJeff Roberson 				dpcpu = dpcpu_alloc(shdr[i].sh_size);
51950c202c5SJeff Roberson 				if (dpcpu == NULL) {
5200f823b64SBjoern A. Zeeb 					printf("%s: pcpu module space is out "
5219afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
5229afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
5239afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
5240f823b64SBjoern A. Zeeb 					    filename);
52550c202c5SJeff Roberson 					error = ENOSPC;
52650c202c5SJeff Roberson 					goto out;
52750c202c5SJeff Roberson 				}
52850c202c5SJeff Roberson 				memcpy(dpcpu, ef->progtab[pb].addr,
52950c202c5SJeff Roberson 				    ef->progtab[pb].size);
53050c202c5SJeff Roberson 				dpcpu_copy(dpcpu, shdr[i].sh_size);
53150c202c5SJeff Roberson 				ef->progtab[pb].addr = dpcpu;
532eddfbb76SRobert Watson #ifdef VIMAGE
533eddfbb76SRobert Watson 			} else if (ef->progtab[pb].name != NULL &&
53417ef1febSRobert Watson 			    !strcmp(ef->progtab[pb].name, VNET_SETNAME)) {
535eddfbb76SRobert Watson 				void *vnet_data;
536eddfbb76SRobert Watson 
537eddfbb76SRobert Watson 				vnet_data = vnet_data_alloc(shdr[i].sh_size);
538eddfbb76SRobert Watson 				if (vnet_data == NULL) {
5390f823b64SBjoern A. Zeeb 					printf("%s: vnet module space is out "
5409afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
5419afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
5429afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
5430f823b64SBjoern A. Zeeb 					    filename);
544eddfbb76SRobert Watson 					error = ENOSPC;
545eddfbb76SRobert Watson 					goto out;
546eddfbb76SRobert Watson 				}
547eddfbb76SRobert Watson 				memcpy(vnet_data, ef->progtab[pb].addr,
548eddfbb76SRobert Watson 				    ef->progtab[pb].size);
549eddfbb76SRobert Watson 				ef->progtab[pb].addr = vnet_data;
550cf7974fdSZhenlei Huang 				vnet_save_init(ef->progtab[pb].addr,
551cf7974fdSZhenlei Huang 				    ef->progtab[pb].size);
552eddfbb76SRobert Watson #endif
5535e6989baSMark Johnston 			} else if ((ef->progtab[pb].name != NULL &&
5545e6989baSMark Johnston 			    strcmp(ef->progtab[pb].name, ".ctors") == 0) ||
5555e6989baSMark Johnston 			    shdr[i].sh_type == SHT_INIT_ARRAY) {
5565e6989baSMark Johnston 				if (lf->ctors_addr != 0) {
5575e6989baSMark Johnston 					printf(
5585e6989baSMark Johnston 				    "%s: multiple ctor sections in %s\n",
5595e6989baSMark Johnston 					    __func__, filename);
5605e6989baSMark Johnston 				} else {
5610067051fSMarcel Moolenaar 					lf->ctors_addr = ef->progtab[pb].addr;
5620067051fSMarcel Moolenaar 					lf->ctors_size = shdr[i].sh_size;
56350c202c5SJeff Roberson 				}
5649e575fadSMark Johnston 			} else if ((ef->progtab[pb].name != NULL &&
5659e575fadSMark Johnston 			    strcmp(ef->progtab[pb].name, ".dtors") == 0) ||
5669e575fadSMark Johnston 			    shdr[i].sh_type == SHT_FINI_ARRAY) {
5679e575fadSMark Johnston 				if (lf->dtors_addr != 0) {
5689e575fadSMark Johnston 					printf(
5699e575fadSMark Johnston 				    "%s: multiple dtor sections in %s\n",
5709e575fadSMark Johnston 					    __func__, filename);
5719e575fadSMark Johnston 				} else {
5729e575fadSMark Johnston 					lf->dtors_addr = ef->progtab[pb].addr;
5739e575fadSMark Johnston 					lf->dtors_size = shdr[i].sh_size;
5749e575fadSMark Johnston 				}
5755e6989baSMark Johnston 			}
57670b7ffeeSIan Dowse 
57770b7ffeeSIan Dowse 			/* Update all symbol values with the offset. */
57870b7ffeeSIan Dowse 			for (j = 0; j < ef->ddbsymcnt; j++) {
57970b7ffeeSIan Dowse 				es = &ef->ddbsymtab[j];
58070b7ffeeSIan Dowse 				if (es->st_shndx != i)
58170b7ffeeSIan Dowse 					continue;
58270b7ffeeSIan Dowse 				es->st_value += (Elf_Addr)ef->progtab[pb].addr;
58370b7ffeeSIan Dowse 			}
58470b7ffeeSIan Dowse 			pb++;
58570b7ffeeSIan Dowse 			break;
58670b7ffeeSIan Dowse 		case SHT_REL:
58715746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
58815746ef4SJohn Baldwin 				break;
58970b7ffeeSIan Dowse 			ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr;
59070b7ffeeSIan Dowse 			ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel);
59170b7ffeeSIan Dowse 			ef->reltab[rl].sec = shdr[i].sh_info;
59270b7ffeeSIan Dowse 			rl++;
59370b7ffeeSIan Dowse 			break;
59470b7ffeeSIan Dowse 		case SHT_RELA:
59515746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
59615746ef4SJohn Baldwin 				break;
59770b7ffeeSIan Dowse 			ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr;
59870b7ffeeSIan Dowse 			ef->relatab[ra].nrela =
59970b7ffeeSIan Dowse 			    shdr[i].sh_size / sizeof(Elf_Rela);
60070b7ffeeSIan Dowse 			ef->relatab[ra].sec = shdr[i].sh_info;
60170b7ffeeSIan Dowse 			ra++;
60270b7ffeeSIan Dowse 			break;
60370b7ffeeSIan Dowse 		}
60470b7ffeeSIan Dowse 	}
6053cfce8e4SKonstantin Belousov 	if (pb != ef->nprogtab) {
6063cfce8e4SKonstantin Belousov 		printf("%s: lost progbits\n", filename);
6073cfce8e4SKonstantin Belousov 		error = ENOEXEC;
6083cfce8e4SKonstantin Belousov 		goto out;
6093cfce8e4SKonstantin Belousov 	}
6103cfce8e4SKonstantin Belousov 	if (rl != ef->nreltab) {
6113cfce8e4SKonstantin Belousov 		printf("%s: lost reltab\n", filename);
6123cfce8e4SKonstantin Belousov 		error = ENOEXEC;
6133cfce8e4SKonstantin Belousov 		goto out;
6143cfce8e4SKonstantin Belousov 	}
6153cfce8e4SKonstantin Belousov 	if (ra != ef->nrelatab) {
6163cfce8e4SKonstantin Belousov 		printf("%s: lost relatab\n", filename);
6173cfce8e4SKonstantin Belousov 		error = ENOEXEC;
6183cfce8e4SKonstantin Belousov 		goto out;
6193cfce8e4SKonstantin Belousov 	}
62070b7ffeeSIan Dowse 
621f822c9e2SMark Johnston 	/*
622f822c9e2SMark Johnston 	 * The file needs to be writeable and executable while applying
623f822c9e2SMark Johnston 	 * relocations.  Mapping protections are applied once relocation
624f822c9e2SMark Johnston 	 * processing is complete.
625f822c9e2SMark Johnston 	 */
626f822c9e2SMark Johnston 	link_elf_protect_range(ef, (vm_offset_t)ef->address,
627f822c9e2SMark Johnston 	    round_page((vm_offset_t)ef->address + ef->lf.size), VM_PROT_ALL);
628f822c9e2SMark Johnston 
62970b7ffeeSIan Dowse 	/* Local intra-module relocations */
630289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, false);
6313cfce8e4SKonstantin Belousov 	if (error != 0)
6323cfce8e4SKonstantin Belousov 		goto out;
63370b7ffeeSIan Dowse 	*result = lf;
63470b7ffeeSIan Dowse 	return (0);
63570b7ffeeSIan Dowse 
63670b7ffeeSIan Dowse out:
63770b7ffeeSIan Dowse 	/* preload not done this way */
63870b7ffeeSIan Dowse 	linker_file_unload(lf, LINKER_UNLOAD_FORCE);
63970b7ffeeSIan Dowse 	return (error);
64070b7ffeeSIan Dowse }
641de78ca7eSPeter Wemm 
6420067051fSMarcel Moolenaar static void
link_elf_invoke_cbs(caddr_t addr,size_t size)6439e575fadSMark Johnston link_elf_invoke_cbs(caddr_t addr, size_t size)
6440067051fSMarcel Moolenaar {
6450067051fSMarcel Moolenaar 	void (**ctor)(void);
6460067051fSMarcel Moolenaar 	size_t i, cnt;
6470067051fSMarcel Moolenaar 
6480067051fSMarcel Moolenaar 	if (addr == NULL || size == 0)
6490067051fSMarcel Moolenaar 		return;
6500067051fSMarcel Moolenaar 	cnt = size / sizeof(*ctor);
6510067051fSMarcel Moolenaar 	ctor = (void *)addr;
6520067051fSMarcel Moolenaar 	for (i = 0; i < cnt; i++) {
6530067051fSMarcel Moolenaar 		if (ctor[i] != NULL)
6540067051fSMarcel Moolenaar 			(*ctor[i])();
6550067051fSMarcel Moolenaar 	}
6560067051fSMarcel Moolenaar }
6570067051fSMarcel Moolenaar 
6587ef5c19bSMark Johnston static void
link_elf_invoke_ctors(linker_file_t lf)6597ef5c19bSMark Johnston link_elf_invoke_ctors(linker_file_t lf)
6607ef5c19bSMark Johnston {
6617ef5c19bSMark Johnston 	KASSERT(lf->ctors_invoked == LF_NONE,
6627ef5c19bSMark Johnston 	    ("%s: file %s ctor state %d",
6637ef5c19bSMark Johnston 	    __func__, lf->filename, lf->ctors_invoked));
6647ef5c19bSMark Johnston 
6657ef5c19bSMark Johnston 	link_elf_invoke_cbs(lf->ctors_addr, lf->ctors_size);
6667ef5c19bSMark Johnston 	lf->ctors_invoked = LF_CTORS;
6677ef5c19bSMark Johnston }
6687ef5c19bSMark Johnston 
6697ef5c19bSMark Johnston static void
link_elf_invoke_dtors(linker_file_t lf)6707ef5c19bSMark Johnston link_elf_invoke_dtors(linker_file_t lf)
6717ef5c19bSMark Johnston {
6727ef5c19bSMark Johnston 	KASSERT(lf->ctors_invoked != LF_DTORS,
6737ef5c19bSMark Johnston 	    ("%s: file %s ctor state %d",
6747ef5c19bSMark Johnston 	    __func__, lf->filename, lf->ctors_invoked));
6757ef5c19bSMark Johnston 
6767ef5c19bSMark Johnston 	if (lf->ctors_invoked == LF_CTORS) {
6777ef5c19bSMark Johnston 		link_elf_invoke_cbs(lf->dtors_addr, lf->dtors_size);
6787ef5c19bSMark Johnston 		lf->ctors_invoked = LF_DTORS;
6797ef5c19bSMark Johnston 	}
6807ef5c19bSMark Johnston }
6817ef5c19bSMark Johnston 
682de78ca7eSPeter Wemm static int
link_elf_link_preload_finish(linker_file_t lf)68354823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf)
68454823af2SPeter Wemm {
68570b7ffeeSIan Dowse 	elf_file_t ef;
68670b7ffeeSIan Dowse 	int error;
68770b7ffeeSIan Dowse 
68870b7ffeeSIan Dowse 	ef = (elf_file_t)lf;
68970b7ffeeSIan Dowse 	error = relocate_file(ef);
69070b7ffeeSIan Dowse 	if (error)
691289ead7cSKonstantin Belousov 		return (error);
69270b7ffeeSIan Dowse 
69370b7ffeeSIan Dowse 	/* Notify MD code that a module is being loaded. */
69470b7ffeeSIan Dowse 	error = elf_cpu_load_file(lf);
69570b7ffeeSIan Dowse 	if (error)
69670b7ffeeSIan Dowse 		return (error);
69770b7ffeeSIan Dowse 
6988f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__)
699289ead7cSKonstantin Belousov 	/* Now ifuncs. */
700289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, true);
701289ead7cSKonstantin Belousov 	if (error != 0)
702289ead7cSKonstantin Belousov 		return (error);
7038f941950SKonstantin Belousov #endif
704289ead7cSKonstantin Belousov 
705f822c9e2SMark Johnston 	/* Apply protections now that relocation processing is complete. */
706f822c9e2SMark Johnston 	link_elf_protect(ef);
707f822c9e2SMark Johnston 
7087ef5c19bSMark Johnston 	link_elf_invoke_ctors(lf);
70970b7ffeeSIan Dowse 	return (0);
71054823af2SPeter Wemm }
71154823af2SPeter Wemm 
71254823af2SPeter Wemm static int
link_elf_load_file(linker_class_t cls,const char * filename,linker_file_t * result)713c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename,
714c143d6c2SMarcel Moolenaar     linker_file_t *result)
715a4f67738SDoug Rabson {
716a344babfSGleb Smirnoff 	struct nameidata *nd;
717b40ce416SJulian Elischer 	struct thread *td = curthread;	/* XXX */
718caab6e90SPeter Wemm 	Elf_Ehdr *hdr;
719e9eabf59SPeter Wemm 	Elf_Shdr *shdr;
720a8774e39SPeter Wemm 	Elf_Sym *es;
721a8774e39SPeter Wemm 	int nbytes, i, j;
72220947801SPeter Wemm 	vm_offset_t mapbase;
723fe3db7c7SDoug Rabson 	size_t mapsize;
724a4f67738SDoug Rabson 	int error = 0;
725526d0bd5SKonstantin Belousov 	ssize_t resid;
726526d0bd5SKonstantin Belousov 	int flags;
727a4f67738SDoug Rabson 	elf_file_t ef;
728a4f67738SDoug Rabson 	linker_file_t lf;
729ca65d5c7SPeter Wemm 	int symtabindex;
730ca65d5c7SPeter Wemm 	int symstrindex;
731e9eabf59SPeter Wemm 	int shstrindex;
732e9eabf59SPeter Wemm 	int nsym;
73320947801SPeter Wemm 	int pb, rl, ra;
734e9eabf59SPeter Wemm 	int alignmask;
735ca65d5c7SPeter Wemm 
736ca65d5c7SPeter Wemm 	shdr = NULL;
737ca65d5c7SPeter Wemm 	lf = NULL;
738e9eabf59SPeter Wemm 	mapsize = 0;
739e9eabf59SPeter Wemm 	hdr = NULL;
740a4f67738SDoug Rabson 
741a344babfSGleb Smirnoff 	nd = malloc(sizeof(struct nameidata), M_TEMP, M_WAITOK);
7427e1d3eefSMateusz Guzik 	NDINIT(nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename);
743e6796b67SKirk McKusick 	flags = FREAD;
744a344babfSGleb Smirnoff 	error = vn_open(nd, &flags, 0, NULL);
745a344babfSGleb Smirnoff 	if (error) {
746a344babfSGleb Smirnoff 		free(nd, M_TEMP);
747a4f67738SDoug Rabson 		return error;
748a344babfSGleb Smirnoff 	}
749bb92cd7bSMateusz Guzik 	NDFREE_PNBUF(nd);
750a344babfSGleb Smirnoff 	if (nd->ni_vp->v_type != VREG) {
7511f49b573SRuslan Ermilov 		error = ENOEXEC;
7521f49b573SRuslan Ermilov 		goto out;
7531f49b573SRuslan Ermilov 	}
754a3df768bSRobert Watson #ifdef MAC
755a344babfSGleb Smirnoff 	error = mac_kld_check_load(td->td_ucred, nd->ni_vp);
756a3df768bSRobert Watson 	if (error) {
757a3df768bSRobert Watson 		goto out;
758a3df768bSRobert Watson 	}
759a3df768bSRobert Watson #endif
760a4f67738SDoug Rabson 
761e9eabf59SPeter Wemm 	/* Read the elf header from the file. */
762e9eabf59SPeter Wemm 	hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK);
763a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)hdr, sizeof(*hdr), 0,
7649ca43589SRobert Watson 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
7659ca43589SRobert Watson 	    &resid, td);
766a4f67738SDoug Rabson 	if (error)
767a4f67738SDoug Rabson 		goto out;
768e9eabf59SPeter Wemm 	if (resid != 0){
769e9eabf59SPeter Wemm 		error = ENOEXEC;
770e9eabf59SPeter Wemm 		goto out;
771e9eabf59SPeter Wemm 	}
772a4f67738SDoug Rabson 
773caab6e90SPeter Wemm 	if (!IS_ELF(*hdr)) {
774fe3db7c7SDoug Rabson 		error = ENOEXEC;
775a4f67738SDoug Rabson 		goto out;
776fe3db7c7SDoug Rabson 	}
777fe3db7c7SDoug Rabson 
778caab6e90SPeter Wemm 	if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS
779caab6e90SPeter Wemm 	    || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) {
780552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported file layout");
781fe3db7c7SDoug Rabson 		error = ENOEXEC;
782fe3db7c7SDoug Rabson 		goto out;
783fe3db7c7SDoug Rabson 	}
784caab6e90SPeter Wemm 	if (hdr->e_ident[EI_VERSION] != EV_CURRENT
785caab6e90SPeter Wemm 	    || hdr->e_version != EV_CURRENT) {
786552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported file version");
787fe3db7c7SDoug Rabson 		error = ENOEXEC;
788fe3db7c7SDoug Rabson 		goto out;
789fe3db7c7SDoug Rabson 	}
790e9eabf59SPeter Wemm 	if (hdr->e_type != ET_REL) {
791a1d7ce03SAttilio Rao 		error = ENOSYS;
792fe3db7c7SDoug Rabson 		goto out;
793fe3db7c7SDoug Rabson 	}
794caab6e90SPeter Wemm 	if (hdr->e_machine != ELF_TARG_MACH) {
795552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported machine");
796fe3db7c7SDoug Rabson 		error = ENOEXEC;
797fe3db7c7SDoug Rabson 		goto out;
798fe3db7c7SDoug Rabson 	}
799a4f67738SDoug Rabson 
800326e27d8SDoug Rabson 	lf = linker_make_file(filename, &link_elf_class);
801326e27d8SDoug Rabson 	if (!lf) {
802326e27d8SDoug Rabson 		error = ENOMEM;
803326e27d8SDoug Rabson 		goto out;
804326e27d8SDoug Rabson 	}
805326e27d8SDoug Rabson 	ef = (elf_file_t) lf;
806e9eabf59SPeter Wemm 	ef->nprogtab = 0;
807e9eabf59SPeter Wemm 	ef->e_shdr = 0;
808b474c780SDavid E. O'Brien 	ef->nreltab = 0;
809b474c780SDavid E. O'Brien 	ef->nrelatab = 0;
810e9eabf59SPeter Wemm 
811e9eabf59SPeter Wemm 	/* Allocate and read in the section header */
812e9eabf59SPeter Wemm 	nbytes = hdr->e_shnum * hdr->e_shentsize;
813e9eabf59SPeter Wemm 	if (nbytes == 0 || hdr->e_shoff == 0 ||
814e9eabf59SPeter Wemm 	    hdr->e_shentsize != sizeof(Elf_Shdr)) {
815e9eabf59SPeter Wemm 		error = ENOEXEC;
816e9eabf59SPeter Wemm 		goto out;
817e9eabf59SPeter Wemm 	}
81820947801SPeter Wemm 	shdr = malloc(nbytes, M_LINKER, M_WAITOK);
819e9eabf59SPeter Wemm 	ef->e_shdr = shdr;
820a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, (caddr_t)shdr, nbytes,
821a344babfSGleb Smirnoff 	    hdr->e_shoff, UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred,
822a344babfSGleb Smirnoff 	    NOCRED, &resid, td);
823e9eabf59SPeter Wemm 	if (error)
824e9eabf59SPeter Wemm 		goto out;
825e9eabf59SPeter Wemm 	if (resid) {
826e9eabf59SPeter Wemm 		error = ENOEXEC;
827e9eabf59SPeter Wemm 		goto out;
828e9eabf59SPeter Wemm 	}
829e9eabf59SPeter Wemm 
830e9eabf59SPeter Wemm 	/* Scan the section header for information and table sizing. */
831e9eabf59SPeter Wemm 	nsym = 0;
832e9eabf59SPeter Wemm 	symtabindex = -1;
833e9eabf59SPeter Wemm 	symstrindex = -1;
834e9eabf59SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
835676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
836676799a0SAndriy Gapon 			continue;
837e9eabf59SPeter Wemm 		switch (shdr[i].sh_type) {
838e9eabf59SPeter Wemm 		case SHT_PROGBITS:
839e9eabf59SPeter Wemm 		case SHT_NOBITS:
84013f28d96SKonstantin Belousov #ifdef __amd64__
841b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
84213f28d96SKonstantin Belousov #endif
8435e6989baSMark Johnston 		case SHT_INIT_ARRAY:
8445e6989baSMark Johnston 		case SHT_FINI_ARRAY:
84558c4aee0SJohn Baldwin 			if ((shdr[i].sh_flags & SHF_ALLOC) == 0)
84658c4aee0SJohn Baldwin 				break;
84720947801SPeter Wemm 			ef->nprogtab++;
848e9eabf59SPeter Wemm 			break;
849e9eabf59SPeter Wemm 		case SHT_SYMTAB:
850e9eabf59SPeter Wemm 			nsym++;
851e9eabf59SPeter Wemm 			symtabindex = i;
852e9eabf59SPeter Wemm 			symstrindex = shdr[i].sh_link;
853e9eabf59SPeter Wemm 			break;
854e9eabf59SPeter Wemm 		case SHT_REL:
85515746ef4SJohn Baldwin 			/*
85615746ef4SJohn Baldwin 			 * Ignore relocation tables for unallocated
85715746ef4SJohn Baldwin 			 * sections.
85815746ef4SJohn Baldwin 			 */
85915746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
86015746ef4SJohn Baldwin 				break;
861b474c780SDavid E. O'Brien 			ef->nreltab++;
862e9eabf59SPeter Wemm 			break;
863e9eabf59SPeter Wemm 		case SHT_RELA:
86415746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
86515746ef4SJohn Baldwin 				break;
866b474c780SDavid E. O'Brien 			ef->nrelatab++;
867e9eabf59SPeter Wemm 			break;
868e9eabf59SPeter Wemm 		case SHT_STRTAB:
869e9eabf59SPeter Wemm 			break;
870e9eabf59SPeter Wemm 		}
871e9eabf59SPeter Wemm 	}
87220947801SPeter Wemm 	if (ef->nprogtab == 0) {
873552f9f63SEdwin Groothuis 		link_elf_error(filename, "file has no contents");
874e9eabf59SPeter Wemm 		error = ENOEXEC;
875e9eabf59SPeter Wemm 		goto out;
876e9eabf59SPeter Wemm 	}
877e9eabf59SPeter Wemm 	if (nsym != 1) {
878e9eabf59SPeter Wemm 		/* Only allow one symbol table for now */
879f912a970SEd Maste 		link_elf_error(filename,
880f912a970SEd Maste 		    "file must have exactly one symbol table");
881e9eabf59SPeter Wemm 		error = ENOEXEC;
882e9eabf59SPeter Wemm 		goto out;
883e9eabf59SPeter Wemm 	}
884e9eabf59SPeter Wemm 	if (symstrindex < 0 || symstrindex > hdr->e_shnum ||
885e9eabf59SPeter Wemm 	    shdr[symstrindex].sh_type != SHT_STRTAB) {
886552f9f63SEdwin Groothuis 		link_elf_error(filename, "file has invalid symbol strings");
887e9eabf59SPeter Wemm 		error = ENOEXEC;
888e9eabf59SPeter Wemm 		goto out;
889e9eabf59SPeter Wemm 	}
890e9eabf59SPeter Wemm 
891e9eabf59SPeter Wemm 	/* Allocate space for tracking the load chunks */
892e9eabf59SPeter Wemm 	if (ef->nprogtab != 0)
89320947801SPeter Wemm 		ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab),
89420947801SPeter Wemm 		    M_LINKER, M_WAITOK | M_ZERO);
895b474c780SDavid E. O'Brien 	if (ef->nreltab != 0)
896b474c780SDavid E. O'Brien 		ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab),
897b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
898b474c780SDavid E. O'Brien 	if (ef->nrelatab != 0)
899b474c780SDavid E. O'Brien 		ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab),
900b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
901e9eabf59SPeter Wemm 
9023cfce8e4SKonstantin Belousov 	if (symtabindex == -1) {
9033cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost symbol table index");
9043cfce8e4SKonstantin Belousov 		error = ENOEXEC;
9053cfce8e4SKonstantin Belousov 		goto out;
9063cfce8e4SKonstantin Belousov 	}
90720947801SPeter Wemm 	/* Allocate space for and load the symbol table */
908e9eabf59SPeter Wemm 	ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
909e9eabf59SPeter Wemm 	ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK);
910a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)ef->ddbsymtab,
91120947801SPeter Wemm 	    shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset,
91220947801SPeter Wemm 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
91320947801SPeter Wemm 	    &resid, td);
91420947801SPeter Wemm 	if (error)
91520947801SPeter Wemm 		goto out;
91620947801SPeter Wemm 	if (resid != 0){
91720947801SPeter Wemm 		error = EINVAL;
91820947801SPeter Wemm 		goto out;
91920947801SPeter Wemm 	}
920e9eabf59SPeter Wemm 
92120947801SPeter Wemm 	/* Allocate space for and load the symbol strings */
922e9eabf59SPeter Wemm 	ef->ddbstrcnt = shdr[symstrindex].sh_size;
923e9eabf59SPeter Wemm 	ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK);
924a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab,
92520947801SPeter Wemm 	    shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset,
92620947801SPeter Wemm 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
92720947801SPeter Wemm 	    &resid, td);
92820947801SPeter Wemm 	if (error)
92920947801SPeter Wemm 		goto out;
93020947801SPeter Wemm 	if (resid != 0){
93120947801SPeter Wemm 		error = EINVAL;
93220947801SPeter Wemm 		goto out;
93320947801SPeter Wemm 	}
934e9eabf59SPeter Wemm 
935e9eabf59SPeter Wemm 	/* Do we have a string table for the section names?  */
936e9eabf59SPeter Wemm 	shstrindex = -1;
93720947801SPeter Wemm 	if (hdr->e_shstrndx != 0 &&
93820947801SPeter Wemm 	    shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) {
939e9eabf59SPeter Wemm 		shstrindex = hdr->e_shstrndx;
940e9eabf59SPeter Wemm 		ef->shstrcnt = shdr[shstrindex].sh_size;
94120947801SPeter Wemm 		ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER,
94220947801SPeter Wemm 		    M_WAITOK);
943a344babfSGleb Smirnoff 		error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab,
94420947801SPeter Wemm 		    shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset,
945e9eabf59SPeter Wemm 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
946e9eabf59SPeter Wemm 		    &resid, td);
947e9eabf59SPeter Wemm 		if (error)
948e9eabf59SPeter Wemm 			goto out;
94920947801SPeter Wemm 		if (resid != 0){
95020947801SPeter Wemm 			error = EINVAL;
95120947801SPeter Wemm 			goto out;
95220947801SPeter Wemm 		}
953e9eabf59SPeter Wemm 	}
954e9eabf59SPeter Wemm 
95520947801SPeter Wemm 	/* Size up code/data(progbits) and bss(nobits). */
956e9eabf59SPeter Wemm 	alignmask = 0;
957e9eabf59SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
958676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
959676799a0SAndriy Gapon 			continue;
960e9eabf59SPeter Wemm 		switch (shdr[i].sh_type) {
961e9eabf59SPeter Wemm 		case SHT_PROGBITS:
962e9eabf59SPeter Wemm 		case SHT_NOBITS:
96313f28d96SKonstantin Belousov #ifdef __amd64__
964b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
96513f28d96SKonstantin Belousov #endif
9665e6989baSMark Johnston 		case SHT_INIT_ARRAY:
9675e6989baSMark Johnston 		case SHT_FINI_ARRAY:
96858c4aee0SJohn Baldwin 			if ((shdr[i].sh_flags & SHF_ALLOC) == 0)
96958c4aee0SJohn Baldwin 				break;
970e9eabf59SPeter Wemm 			alignmask = shdr[i].sh_addralign - 1;
971e9eabf59SPeter Wemm 			mapsize += alignmask;
972e9eabf59SPeter Wemm 			mapsize &= ~alignmask;
973e9eabf59SPeter Wemm 			mapsize += shdr[i].sh_size;
974e9eabf59SPeter Wemm 			break;
975e9eabf59SPeter Wemm 		}
976e9eabf59SPeter Wemm 	}
977e9eabf59SPeter Wemm 
978e9eabf59SPeter Wemm 	/*
979e9eabf59SPeter Wemm 	 * We know how much space we need for the text/data/bss/etc.
980e9eabf59SPeter Wemm 	 * This stuff needs to be in a single chunk so that profiling etc
981e9eabf59SPeter Wemm 	 * can get the bounds and gdb can associate offsets with modules
982e9eabf59SPeter Wemm 	 */
983fbf2a778SKonstantin Belousov 	ef->object = vm_pager_allocate(OBJT_PHYS, NULL, round_page(mapsize),
984fbf2a778SKonstantin Belousov 	    VM_PROT_ALL, 0, thread0.td_ucred);
985fe3db7c7SDoug Rabson 	if (ef->object == NULL) {
986fe3db7c7SDoug Rabson 		error = ENOMEM;
987fe3db7c7SDoug Rabson 		goto out;
988fe3db7c7SDoug Rabson 	}
989d80126a6SMark Johnston #if VM_NRESERVLEVEL > 0
990d80126a6SMark Johnston 	vm_object_color(ef->object, 0);
991d80126a6SMark Johnston #endif
992ac68d1c9SAlan Cox 
993ac68d1c9SAlan Cox 	/*
994ac68d1c9SAlan Cox 	 * In order to satisfy amd64's architectural requirements on the
995ac68d1c9SAlan Cox 	 * location of code and data in the kernel's address space, request a
996ac68d1c9SAlan Cox 	 * mapping that is above the kernel.
9971d9eae9fSMark Johnston 	 *
9981d9eae9fSMark Johnston 	 * Protections will be restricted once relocations are applied.
999ac68d1c9SAlan Cox 	 */
10004104e835SOleksandr Tymoshenko #ifdef __amd64__
1001ac68d1c9SAlan Cox 	mapbase = KERNBASE;
10024104e835SOleksandr Tymoshenko #else
10034104e835SOleksandr Tymoshenko 	mapbase = VM_MIN_KERNEL_ADDRESS;
10044104e835SOleksandr Tymoshenko #endif
100520947801SPeter Wemm 	error = vm_map_find(kernel_map, ef->object, 0, &mapbase,
1006edb572a3SJohn Baldwin 	    round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL,
1007edb572a3SJohn Baldwin 	    VM_PROT_ALL, 0);
1008092bacb2SMark Johnston 	if (error != KERN_SUCCESS) {
1009fe3db7c7SDoug Rabson 		vm_object_deallocate(ef->object);
1010092bacb2SMark Johnston 		ef->object = NULL;
1011092bacb2SMark Johnston 		error = ENOMEM;
1012fe3db7c7SDoug Rabson 		goto out;
1013fe3db7c7SDoug Rabson 	}
101420947801SPeter Wemm 
1015e9eabf59SPeter Wemm 	/* Wire the pages */
10167f1ef325SAlan Cox 	error = vm_map_wire(kernel_map, mapbase,
101720947801SPeter Wemm 	    mapbase + round_page(mapsize),
1018e9eabf59SPeter Wemm 	    VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES);
10197f1ef325SAlan Cox 	if (error != KERN_SUCCESS) {
10207f1ef325SAlan Cox 		error = ENOMEM;
10217f1ef325SAlan Cox 		goto out;
10227f1ef325SAlan Cox 	}
1023a4f67738SDoug Rabson 
102420947801SPeter Wemm 	/* Inform the kld system about the situation */
102520947801SPeter Wemm 	lf->address = ef->address = (caddr_t)mapbase;
102620947801SPeter Wemm 	lf->size = mapsize;
1027e9eabf59SPeter Wemm 
1028e9eabf59SPeter Wemm 	/*
102920947801SPeter Wemm 	 * Now load code/data(progbits), zero bss(nobits), allocate space for
103020947801SPeter Wemm 	 * and load relocs
1031e9eabf59SPeter Wemm 	 */
103220947801SPeter Wemm 	pb = 0;
103320947801SPeter Wemm 	rl = 0;
103420947801SPeter Wemm 	ra = 0;
103520947801SPeter Wemm 	alignmask = 0;
103620947801SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
1037676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
1038676799a0SAndriy Gapon 			continue;
103920947801SPeter Wemm 		switch (shdr[i].sh_type) {
104020947801SPeter Wemm 		case SHT_PROGBITS:
104120947801SPeter Wemm 		case SHT_NOBITS:
104213f28d96SKonstantin Belousov #ifdef __amd64__
1043b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
104413f28d96SKonstantin Belousov #endif
10455e6989baSMark Johnston 		case SHT_INIT_ARRAY:
10465e6989baSMark Johnston 		case SHT_FINI_ARRAY:
104758c4aee0SJohn Baldwin 			if ((shdr[i].sh_flags & SHF_ALLOC) == 0)
104858c4aee0SJohn Baldwin 				break;
104920947801SPeter Wemm 			alignmask = shdr[i].sh_addralign - 1;
105020947801SPeter Wemm 			mapbase += alignmask;
105120947801SPeter Wemm 			mapbase &= ~alignmask;
10520067051fSMarcel Moolenaar 			if (ef->shstrtab != NULL && shdr[i].sh_name != 0) {
105350c202c5SJeff Roberson 				ef->progtab[pb].name =
105450c202c5SJeff Roberson 				    ef->shstrtab + shdr[i].sh_name;
10555e6989baSMark Johnston 				if (!strcmp(ef->progtab[pb].name, ".ctors") ||
10565e6989baSMark Johnston 				    shdr[i].sh_type == SHT_INIT_ARRAY) {
10575e6989baSMark Johnston 					if (lf->ctors_addr != 0) {
10585e6989baSMark Johnston 						printf(
10595e6989baSMark Johnston 				    "%s: multiple ctor sections in %s\n",
10605e6989baSMark Johnston 						    __func__, filename);
10615e6989baSMark Johnston 					} else {
10625e6989baSMark Johnston 						lf->ctors_addr =
10635e6989baSMark Johnston 						    (caddr_t)mapbase;
10645e6989baSMark Johnston 						lf->ctors_size =
10655e6989baSMark Johnston 						    shdr[i].sh_size;
10665e6989baSMark Johnston 					}
10679e575fadSMark Johnston 				} else if (!strcmp(ef->progtab[pb].name,
10689e575fadSMark Johnston 				    ".dtors") ||
10699e575fadSMark Johnston 				    shdr[i].sh_type == SHT_FINI_ARRAY) {
10709e575fadSMark Johnston 					if (lf->dtors_addr != 0) {
10719e575fadSMark Johnston 						printf(
10729e575fadSMark Johnston 				    "%s: multiple dtor sections in %s\n",
10739e575fadSMark Johnston 						    __func__, filename);
10749e575fadSMark Johnston 					} else {
10759e575fadSMark Johnston 						lf->dtors_addr =
10769e575fadSMark Johnston 						    (caddr_t)mapbase;
10779e575fadSMark Johnston 						lf->dtors_size =
10789e575fadSMark Johnston 						    shdr[i].sh_size;
10799e575fadSMark Johnston 					}
10800067051fSMarcel Moolenaar 				}
10810067051fSMarcel Moolenaar 			} else if (shdr[i].sh_type == SHT_PROGBITS)
108220947801SPeter Wemm 				ef->progtab[pb].name = "<<PROGBITS>>";
108313f28d96SKonstantin Belousov #ifdef __amd64__
1084b715d9afSKonstantin Belousov 			else if (shdr[i].sh_type == SHT_X86_64_UNWIND)
108513f28d96SKonstantin Belousov 				ef->progtab[pb].name = "<<UNWIND>>";
108613f28d96SKonstantin Belousov #endif
108750c202c5SJeff Roberson 			else
108850c202c5SJeff Roberson 				ef->progtab[pb].name = "<<NOBITS>>";
108950c202c5SJeff Roberson 			if (ef->progtab[pb].name != NULL &&
10900f823b64SBjoern A. Zeeb 			    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) {
109150c202c5SJeff Roberson 				ef->progtab[pb].addr =
109250c202c5SJeff Roberson 				    dpcpu_alloc(shdr[i].sh_size);
10930f823b64SBjoern A. Zeeb 				if (ef->progtab[pb].addr == NULL) {
10940f823b64SBjoern A. Zeeb 					printf("%s: pcpu module space is out "
10959afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
10969afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
10979afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
10980f823b64SBjoern A. Zeeb 					    filename);
10990f823b64SBjoern A. Zeeb 				}
11000f823b64SBjoern A. Zeeb 			}
1101eddfbb76SRobert Watson #ifdef VIMAGE
1102eddfbb76SRobert Watson 			else if (ef->progtab[pb].name != NULL &&
11030f823b64SBjoern A. Zeeb 			    !strcmp(ef->progtab[pb].name, VNET_SETNAME)) {
1104eddfbb76SRobert Watson 				ef->progtab[pb].addr =
1105eddfbb76SRobert Watson 				    vnet_data_alloc(shdr[i].sh_size);
11060f823b64SBjoern A. Zeeb 				if (ef->progtab[pb].addr == NULL) {
11070f823b64SBjoern A. Zeeb 					printf("%s: vnet module space is out "
11089afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
11099afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
11109afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
11110f823b64SBjoern A. Zeeb 					    filename);
11120f823b64SBjoern A. Zeeb 				}
11130f823b64SBjoern A. Zeeb 			}
1114eddfbb76SRobert Watson #endif
111550c202c5SJeff Roberson 			else
111650c202c5SJeff Roberson 				ef->progtab[pb].addr =
111750c202c5SJeff Roberson 				    (void *)(uintptr_t)mapbase;
111850c202c5SJeff Roberson 			if (ef->progtab[pb].addr == NULL) {
111950c202c5SJeff Roberson 				error = ENOSPC;
112050c202c5SJeff Roberson 				goto out;
112150c202c5SJeff Roberson 			}
112250c202c5SJeff Roberson 			ef->progtab[pb].size = shdr[i].sh_size;
11231d9eae9fSMark Johnston 			ef->progtab[pb].flags = shdr[i].sh_flags;
112450c202c5SJeff Roberson 			ef->progtab[pb].sec = i;
112513f28d96SKonstantin Belousov 			if (shdr[i].sh_type == SHT_PROGBITS
112613f28d96SKonstantin Belousov #ifdef __amd64__
1127b715d9afSKonstantin Belousov 			    || shdr[i].sh_type == SHT_X86_64_UNWIND
112813f28d96SKonstantin Belousov #endif
112913f28d96SKonstantin Belousov 			    ) {
1130a344babfSGleb Smirnoff 				error = vn_rdwr(UIO_READ, nd->ni_vp,
113120947801SPeter Wemm 				    ef->progtab[pb].addr,
113220947801SPeter Wemm 				    shdr[i].sh_size, shdr[i].sh_offset,
113320947801SPeter Wemm 				    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred,
113420947801SPeter Wemm 				    NOCRED, &resid, td);
113520947801SPeter Wemm 				if (error)
113620947801SPeter Wemm 					goto out;
113720947801SPeter Wemm 				if (resid != 0){
113820947801SPeter Wemm 					error = EINVAL;
113920947801SPeter Wemm 					goto out;
114020947801SPeter Wemm 				}
1141205821a7SZhenlei Huang 				/* Initialize the per-cpu area. */
1142eddfbb76SRobert Watson 				if (ef->progtab[pb].addr != (void *)mapbase &&
11435f67450dSDimitry Andric 				    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME))
114450c202c5SJeff Roberson 					dpcpu_copy(ef->progtab[pb].addr,
114550c202c5SJeff Roberson 					    shdr[i].sh_size);
114650c202c5SJeff Roberson 			} else
114720947801SPeter Wemm 				bzero(ef->progtab[pb].addr, shdr[i].sh_size);
1148a8774e39SPeter Wemm 
1149cf7974fdSZhenlei Huang #ifdef VIMAGE
1150cf7974fdSZhenlei Huang 			if (ef->progtab[pb].addr != (void *)mapbase &&
1151cf7974fdSZhenlei Huang 			    strcmp(ef->progtab[pb].name, VNET_SETNAME) == 0)
1152cf7974fdSZhenlei Huang 				vnet_save_init(ef->progtab[pb].addr,
1153cf7974fdSZhenlei Huang 				    ef->progtab[pb].size);
1154cf7974fdSZhenlei Huang #endif
1155a8774e39SPeter Wemm 			/* Update all symbol values with the offset. */
1156a8774e39SPeter Wemm 			for (j = 0; j < ef->ddbsymcnt; j++) {
1157a8774e39SPeter Wemm 				es = &ef->ddbsymtab[j];
1158a8774e39SPeter Wemm 				if (es->st_shndx != i)
1159a8774e39SPeter Wemm 					continue;
1160a8774e39SPeter Wemm 				es->st_value += (Elf_Addr)ef->progtab[pb].addr;
1161a8774e39SPeter Wemm 			}
116220947801SPeter Wemm 			mapbase += shdr[i].sh_size;
116320947801SPeter Wemm 			pb++;
116420947801SPeter Wemm 			break;
116520947801SPeter Wemm 		case SHT_REL:
116615746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
116715746ef4SJohn Baldwin 				break;
116820947801SPeter Wemm 			ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER,
116920947801SPeter Wemm 			    M_WAITOK);
117020947801SPeter Wemm 			ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel);
117120947801SPeter Wemm 			ef->reltab[rl].sec = shdr[i].sh_info;
1172a344babfSGleb Smirnoff 			error = vn_rdwr(UIO_READ, nd->ni_vp,
117320947801SPeter Wemm 			    (void *)ef->reltab[rl].rel,
117420947801SPeter Wemm 			    shdr[i].sh_size, shdr[i].sh_offset,
1175e9eabf59SPeter Wemm 			    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
1176e9eabf59SPeter Wemm 			    &resid, td);
1177e9eabf59SPeter Wemm 			if (error)
1178e9eabf59SPeter Wemm 				goto out;
117920947801SPeter Wemm 			if (resid != 0){
118020947801SPeter Wemm 				error = EINVAL;
118120947801SPeter Wemm 				goto out;
1182e9eabf59SPeter Wemm 			}
118320947801SPeter Wemm 			rl++;
118420947801SPeter Wemm 			break;
118520947801SPeter Wemm 		case SHT_RELA:
118615746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
118715746ef4SJohn Baldwin 				break;
118820947801SPeter Wemm 			ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER,
118920947801SPeter Wemm 			    M_WAITOK);
119020947801SPeter Wemm 			ef->relatab[ra].nrela =
119120947801SPeter Wemm 			    shdr[i].sh_size / sizeof(Elf_Rela);
119220947801SPeter Wemm 			ef->relatab[ra].sec = shdr[i].sh_info;
1193a344babfSGleb Smirnoff 			error = vn_rdwr(UIO_READ, nd->ni_vp,
119420947801SPeter Wemm 			    (void *)ef->relatab[ra].rela,
119520947801SPeter Wemm 			    shdr[i].sh_size, shdr[i].sh_offset,
1196e9eabf59SPeter Wemm 			    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
1197e9eabf59SPeter Wemm 			    &resid, td);
1198e9eabf59SPeter Wemm 			if (error)
1199e9eabf59SPeter Wemm 				goto out;
120020947801SPeter Wemm 			if (resid != 0){
120120947801SPeter Wemm 				error = EINVAL;
120220947801SPeter Wemm 				goto out;
1203e9eabf59SPeter Wemm 			}
120420947801SPeter Wemm 			ra++;
120520947801SPeter Wemm 			break;
120620947801SPeter Wemm 		}
120720947801SPeter Wemm 	}
12083cfce8e4SKonstantin Belousov 	if (pb != ef->nprogtab) {
12093cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost progbits");
12103cfce8e4SKonstantin Belousov 		error = ENOEXEC;
12113cfce8e4SKonstantin Belousov 		goto out;
12123cfce8e4SKonstantin Belousov 	}
12133cfce8e4SKonstantin Belousov 	if (rl != ef->nreltab) {
12143cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost reltab");
12153cfce8e4SKonstantin Belousov 		error = ENOEXEC;
12163cfce8e4SKonstantin Belousov 		goto out;
12173cfce8e4SKonstantin Belousov 	}
12183cfce8e4SKonstantin Belousov 	if (ra != ef->nrelatab) {
12193cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost relatab");
12203cfce8e4SKonstantin Belousov 		error = ENOEXEC;
12213cfce8e4SKonstantin Belousov 		goto out;
12223cfce8e4SKonstantin Belousov 	}
12233cfce8e4SKonstantin Belousov 	if (mapbase != (vm_offset_t)ef->address + mapsize) {
12243cfce8e4SKonstantin Belousov 		printf(
12253cfce8e4SKonstantin Belousov 		    "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n",
12263cfce8e4SKonstantin Belousov 		    filename != NULL ? filename : "<none>",
12276f3c6327SNeel Natu 		    (u_long)mapbase, ef->address, (u_long)mapsize,
12286f3c6327SNeel Natu 		    (u_long)(vm_offset_t)ef->address + mapsize);
12293cfce8e4SKonstantin Belousov 		error = ENOMEM;
12303cfce8e4SKonstantin Belousov 		goto out;
12313cfce8e4SKonstantin Belousov 	}
1232e9eabf59SPeter Wemm 
1233e9eabf59SPeter Wemm 	/* Local intra-module relocations */
1234289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, false);
12353cfce8e4SKonstantin Belousov 	if (error != 0)
12363cfce8e4SKonstantin Belousov 		goto out;
12377251b4bfSJake Burkholder 
1238e9eabf59SPeter Wemm 	/* Pull in dependencies */
1239b249ce48SMateusz Guzik 	VOP_UNLOCK(nd->ni_vp);
12407b9716baSIan Dowse 	error = linker_load_dependencies(lf);
1241a344babfSGleb Smirnoff 	vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY);
1242ca65d5c7SPeter Wemm 	if (error)
1243de78ca7eSPeter Wemm 		goto out;
1244e9eabf59SPeter Wemm 
1245e9eabf59SPeter Wemm 	/* External relocations */
1246326e27d8SDoug Rabson 	error = relocate_file(ef);
1247ca65d5c7SPeter Wemm 	if (error)
1248ca65d5c7SPeter Wemm 		goto out;
1249ca65d5c7SPeter Wemm 
1250e9eabf59SPeter Wemm 	/* Notify MD code that a module is being loaded. */
1251e9eabf59SPeter Wemm 	error = elf_cpu_load_file(lf);
1252ca65d5c7SPeter Wemm 	if (error)
1253ca65d5c7SPeter Wemm 		goto out;
1254a4f67738SDoug Rabson 
12558f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__)
1256289ead7cSKonstantin Belousov 	/* Now ifuncs. */
1257289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, true);
1258289ead7cSKonstantin Belousov 	if (error != 0)
1259289ead7cSKonstantin Belousov 		goto out;
12608f941950SKonstantin Belousov #endif
1261289ead7cSKonstantin Belousov 
12621d9eae9fSMark Johnston 	link_elf_protect(ef);
12637ef5c19bSMark Johnston 	link_elf_invoke_ctors(lf);
1264a4f67738SDoug Rabson 	*result = lf;
1265a4f67738SDoug Rabson 
1266a4f67738SDoug Rabson out:
1267b249ce48SMateusz Guzik 	VOP_UNLOCK(nd->ni_vp);
1268a344babfSGleb Smirnoff 	vn_close(nd->ni_vp, FREAD, td->td_ucred, td);
1269a344babfSGleb Smirnoff 	free(nd, M_TEMP);
1270ca65d5c7SPeter Wemm 	if (error && lf)
127165a311fcSPoul-Henning Kamp 		linker_file_unload(lf, LINKER_UNLOAD_FORCE);
1272e9eabf59SPeter Wemm 	free(hdr, M_LINKER);
1273a4f67738SDoug Rabson 
1274a4f67738SDoug Rabson 	return error;
1275a4f67738SDoug Rabson }
1276a4f67738SDoug Rabson 
1277a4f67738SDoug Rabson static void
link_elf_unload_file(linker_file_t file)1278de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file)
1279a4f67738SDoug Rabson {
1280326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) file;
1281d821d364SPedro F. Giffuni 	u_int i;
1282a4f67738SDoug Rabson 
12837ef5c19bSMark Johnston 	link_elf_invoke_dtors(file);
12849e575fadSMark Johnston 
12851aeb23cdSMarcel Moolenaar 	/* Notify MD code that a module is being unloaded. */
12861aeb23cdSMarcel Moolenaar 	elf_cpu_unload_file(file);
12871aeb23cdSMarcel Moolenaar 
128850c202c5SJeff Roberson 	if (ef->progtab) {
128950c202c5SJeff Roberson 		for (i = 0; i < ef->nprogtab; i++) {
129050c202c5SJeff Roberson 			if (ef->progtab[i].size == 0)
129150c202c5SJeff Roberson 				continue;
129250c202c5SJeff Roberson 			if (ef->progtab[i].name == NULL)
129350c202c5SJeff Roberson 				continue;
12945f67450dSDimitry Andric 			if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME))
129550c202c5SJeff Roberson 				dpcpu_free(ef->progtab[i].addr,
129650c202c5SJeff Roberson 				    ef->progtab[i].size);
1297eddfbb76SRobert Watson #ifdef VIMAGE
129817ef1febSRobert Watson 			else if (!strcmp(ef->progtab[i].name, VNET_SETNAME))
1299eddfbb76SRobert Watson 				vnet_data_free(ef->progtab[i].addr,
1300eddfbb76SRobert Watson 				    ef->progtab[i].size);
1301eddfbb76SRobert Watson #endif
130250c202c5SJeff Roberson 		}
130350c202c5SJeff Roberson 	}
130470b7ffeeSIan Dowse 	if (ef->preloaded) {
130570b7ffeeSIan Dowse 		free(ef->reltab, M_LINKER);
130670b7ffeeSIan Dowse 		free(ef->relatab, M_LINKER);
130770b7ffeeSIan Dowse 		free(ef->progtab, M_LINKER);
1308a2024a3eSJohn Birrell 		free(ef->ctftab, M_LINKER);
1309a2024a3eSJohn Birrell 		free(ef->ctfoff, M_LINKER);
1310a2024a3eSJohn Birrell 		free(ef->typoff, M_LINKER);
131173624a80SMark Johnston 		if (file->pathname != NULL)
131273624a80SMark Johnston 			preload_delete_name(file->pathname);
131370b7ffeeSIan Dowse 		return;
131470b7ffeeSIan Dowse 	}
131570b7ffeeSIan Dowse 
1316b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++)
131720947801SPeter Wemm 		free(ef->reltab[i].rel, M_LINKER);
1318b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++)
131920947801SPeter Wemm 		free(ef->relatab[i].rela, M_LINKER);
132020947801SPeter Wemm 	free(ef->reltab, M_LINKER);
132120947801SPeter Wemm 	free(ef->relatab, M_LINKER);
132220947801SPeter Wemm 	free(ef->progtab, M_LINKER);
132320947801SPeter Wemm 
1324092bacb2SMark Johnston 	if (ef->object != NULL)
1325fe3db7c7SDoug Rabson 		vm_map_remove(kernel_map, (vm_offset_t)ef->address,
1326092bacb2SMark Johnston 		    (vm_offset_t)ef->address + ptoa(ef->object->size));
1327e9eabf59SPeter Wemm 	free(ef->e_shdr, M_LINKER);
1328e9eabf59SPeter Wemm 	free(ef->ddbsymtab, M_LINKER);
1329e9eabf59SPeter Wemm 	free(ef->ddbstrtab, M_LINKER);
133020947801SPeter Wemm 	free(ef->shstrtab, M_LINKER);
1331a2024a3eSJohn Birrell 	free(ef->ctftab, M_LINKER);
1332a2024a3eSJohn Birrell 	free(ef->ctfoff, M_LINKER);
1333a2024a3eSJohn Birrell 	free(ef->typoff, M_LINKER);
1334de78ca7eSPeter Wemm }
1335de78ca7eSPeter Wemm 
1336fe3db7c7SDoug Rabson static const char *
symbol_name(elf_file_t ef,Elf_Size r_info)1337757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info)
1338a4f67738SDoug Rabson {
1339fe3db7c7SDoug Rabson 	const Elf_Sym *ref;
1340a4f67738SDoug Rabson 
1341aa855a59SPeter Wemm 	if (ELF_R_SYM(r_info)) {
1342e9eabf59SPeter Wemm 		ref = ef->ddbsymtab + ELF_R_SYM(r_info);
1343e9eabf59SPeter Wemm 		return ef->ddbstrtab + ref->st_name;
1344fe3db7c7SDoug Rabson 	} else
1345fe3db7c7SDoug Rabson 		return NULL;
1346a4f67738SDoug Rabson }
1347a4f67738SDoug Rabson 
1348e9eabf59SPeter Wemm static Elf_Addr
findbase(elf_file_t ef,int sec)1349e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec)
1350e9eabf59SPeter Wemm {
1351e9eabf59SPeter Wemm 	int i;
1352e9eabf59SPeter Wemm 	Elf_Addr base = 0;
1353e9eabf59SPeter Wemm 
1354e9eabf59SPeter Wemm 	for (i = 0; i < ef->nprogtab; i++) {
1355a8774e39SPeter Wemm 		if (sec == ef->progtab[i].sec) {
1356e9eabf59SPeter Wemm 			base = (Elf_Addr)ef->progtab[i].addr;
1357a8774e39SPeter Wemm 			break;
1358e9eabf59SPeter Wemm 		}
1359a8774e39SPeter Wemm 	}
1360e9eabf59SPeter Wemm 	return base;
1361e9eabf59SPeter Wemm }
1362e9eabf59SPeter Wemm 
1363a4f67738SDoug Rabson static int
relocate_file1(elf_file_t ef,bool ifuncs)1364b11e6fd7SMark Johnston relocate_file1(elf_file_t ef, bool ifuncs)
1365a4f67738SDoug Rabson {
1366fe3db7c7SDoug Rabson 	const Elf_Rel *rellim;
1367fe3db7c7SDoug Rabson 	const Elf_Rel *rel;
1368fe3db7c7SDoug Rabson 	const Elf_Rela *relalim;
1369fe3db7c7SDoug Rabson 	const Elf_Rela *rela;
1370aa855a59SPeter Wemm 	const char *symname;
1371e9eabf59SPeter Wemm 	const Elf_Sym *sym;
1372e9eabf59SPeter Wemm 	int i;
1373757686b1SMarcel Moolenaar 	Elf_Size symidx;
1374e9eabf59SPeter Wemm 	Elf_Addr base;
1375e9eabf59SPeter Wemm 
1376fe3db7c7SDoug Rabson 	/* Perform relocations without addend if there are any: */
1377b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++) {
1378e9eabf59SPeter Wemm 		rel = ef->reltab[i].rel;
13793cfce8e4SKonstantin Belousov 		if (rel == NULL) {
13803cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a reltab!");
13813cfce8e4SKonstantin Belousov 			return (ENOEXEC);
13823cfce8e4SKonstantin Belousov 		}
138320947801SPeter Wemm 		rellim = rel + ef->reltab[i].nrel;
1384e9eabf59SPeter Wemm 		base = findbase(ef, ef->reltab[i].sec);
13853cfce8e4SKonstantin Belousov 		if (base == 0) {
13863cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost base for reltab");
13873cfce8e4SKonstantin Belousov 			return (ENOEXEC);
13883cfce8e4SKonstantin Belousov 		}
138920947801SPeter Wemm 		for ( ; rel < rellim; rel++) {
1390e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rel->r_info);
139120947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
139220947801SPeter Wemm 				continue;
1393e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
139420947801SPeter Wemm 			/* Local relocs are already done */
139520947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL)
139620947801SPeter Wemm 				continue;
1397b11e6fd7SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
1398b11e6fd7SMark Johnston 			    elf_is_ifunc_reloc(rel->r_info)) != ifuncs)
1399b11e6fd7SMark Johnston 				continue;
140020947801SPeter Wemm 			if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL,
140120947801SPeter Wemm 			    elf_obj_lookup)) {
1402aa855a59SPeter Wemm 				symname = symbol_name(ef, rel->r_info);
140320947801SPeter Wemm 				printf("link_elf_obj: symbol %s undefined\n",
140420947801SPeter Wemm 				    symname);
14053cfce8e4SKonstantin Belousov 				return (ENOENT);
1406a13ddfb6SPeter Wemm 			}
1407aa855a59SPeter Wemm 		}
1408a4f67738SDoug Rabson 	}
1409a4f67738SDoug Rabson 
1410fe3db7c7SDoug Rabson 	/* Perform relocations with addend if there are any: */
1411b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++) {
1412e9eabf59SPeter Wemm 		rela = ef->relatab[i].rela;
14133cfce8e4SKonstantin Belousov 		if (rela == NULL) {
14143cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a relatab!");
14153cfce8e4SKonstantin Belousov 			return (ENOEXEC);
14163cfce8e4SKonstantin Belousov 		}
141720947801SPeter Wemm 		relalim = rela + ef->relatab[i].nrela;
1418e9eabf59SPeter Wemm 		base = findbase(ef, ef->relatab[i].sec);
14193cfce8e4SKonstantin Belousov 		if (base == 0) {
14203cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename,
14213cfce8e4SKonstantin Belousov 			    "lost base for relatab");
14223cfce8e4SKonstantin Belousov 			return (ENOEXEC);
14233cfce8e4SKonstantin Belousov 		}
142420947801SPeter Wemm 		for ( ; rela < relalim; rela++) {
1425e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rela->r_info);
142620947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
142720947801SPeter Wemm 				continue;
1428e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
142920947801SPeter Wemm 			/* Local relocs are already done */
143020947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL)
143120947801SPeter Wemm 				continue;
1432b11e6fd7SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
1433b11e6fd7SMark Johnston 			    elf_is_ifunc_reloc(rela->r_info)) != ifuncs)
1434b11e6fd7SMark Johnston 				continue;
143520947801SPeter Wemm 			if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA,
143620947801SPeter Wemm 			    elf_obj_lookup)) {
1437aa855a59SPeter Wemm 				symname = symbol_name(ef, rela->r_info);
143820947801SPeter Wemm 				printf("link_elf_obj: symbol %s undefined\n",
143920947801SPeter Wemm 				    symname);
14403cfce8e4SKonstantin Belousov 				return (ENOENT);
1441a13ddfb6SPeter Wemm 			}
1442e9eabf59SPeter Wemm 		}
1443e9eabf59SPeter Wemm 	}
1444a4f67738SDoug Rabson 
14452832cd54SKonstantin Belousov 	/*
1446e3043798SPedro F. Giffuni 	 * Only clean SHN_FBSD_CACHED for successful return.  If we
14472832cd54SKonstantin Belousov 	 * modified symbol table for the object but found an
14482832cd54SKonstantin Belousov 	 * unresolved symbol, there is no reason to roll back.
14492832cd54SKonstantin Belousov 	 */
14502832cd54SKonstantin Belousov 	elf_obj_cleanup_globals_cache(ef);
14512832cd54SKonstantin Belousov 
14523cfce8e4SKonstantin Belousov 	return (0);
1453a4f67738SDoug Rabson }
1454a4f67738SDoug Rabson 
145537c84183SPoul-Henning Kamp static int
relocate_file(elf_file_t ef)1456b11e6fd7SMark Johnston relocate_file(elf_file_t ef)
1457b11e6fd7SMark Johnston {
1458b11e6fd7SMark Johnston 	int error;
1459b11e6fd7SMark Johnston 
1460b11e6fd7SMark Johnston 	error = relocate_file1(ef, false);
1461b11e6fd7SMark Johnston 	if (error == 0)
1462b11e6fd7SMark Johnston 		error = relocate_file1(ef, true);
1463b11e6fd7SMark Johnston 	return (error);
1464b11e6fd7SMark Johnston }
1465b11e6fd7SMark Johnston 
1466b11e6fd7SMark Johnston static int
link_elf_lookup_symbol1(linker_file_t lf,const char * name,c_linker_sym_t * sym,bool see_local)146795c20fafSKonstantin Belousov link_elf_lookup_symbol1(linker_file_t lf, const char *name, c_linker_sym_t *sym,
146895c20fafSKonstantin Belousov     bool see_local)
1469a4f67738SDoug Rabson {
1470326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t)lf;
14712d636ab0SPeter Wemm 	const Elf_Sym *symp;
14722d636ab0SPeter Wemm 	const char *strp;
1473a4f67738SDoug Rabson 	int i;
1474a4f67738SDoug Rabson 
14752d636ab0SPeter Wemm 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
14762d636ab0SPeter Wemm 		strp = ef->ddbstrtab + symp->st_name;
1477a8774e39SPeter Wemm 		if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) {
147895c20fafSKonstantin Belousov 			if (see_local ||
147995c20fafSKonstantin Belousov 			    ELF_ST_BIND(symp->st_info) == STB_GLOBAL) {
1480d254af07SMatthew Dillon 				*sym = (c_linker_sym_t) symp;
148195c20fafSKonstantin Belousov 				return (0);
148295c20fafSKonstantin Belousov 			}
148395c20fafSKonstantin Belousov 			return (ENOENT);
14842d636ab0SPeter Wemm 		}
14852d636ab0SPeter Wemm 	}
1486a7e4eb14SKonstantin Belousov 	return (ENOENT);
1487a4f67738SDoug Rabson }
1488a4f67738SDoug Rabson 
1489de78ca7eSPeter Wemm static int
link_elf_lookup_symbol(linker_file_t lf,const char * name,c_linker_sym_t * sym)149095c20fafSKonstantin Belousov link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym)
149195c20fafSKonstantin Belousov {
1492ecd8245eSKonstantin Belousov 	return (link_elf_lookup_symbol1(lf, name, sym,
1493ecd8245eSKonstantin Belousov 	    link_elf_obj_leak_locals));
149495c20fafSKonstantin Belousov }
149595c20fafSKonstantin Belousov 
149695c20fafSKonstantin Belousov static int
link_elf_lookup_debug_symbol(linker_file_t lf,const char * name,c_linker_sym_t * sym)149795c20fafSKonstantin Belousov link_elf_lookup_debug_symbol(linker_file_t lf, const char *name,
149895c20fafSKonstantin Belousov     c_linker_sym_t *sym)
149995c20fafSKonstantin Belousov {
150095c20fafSKonstantin Belousov 	return (link_elf_lookup_symbol1(lf, name, sym, true));
150195c20fafSKonstantin Belousov }
150295c20fafSKonstantin Belousov 
150395c20fafSKonstantin Belousov static int
link_elf_lookup_debug_symbol_ctf(linker_file_t lf,const char * name,c_linker_sym_t * sym,linker_ctf_t * lc)1504c21bc6f3SBojan Novković link_elf_lookup_debug_symbol_ctf(linker_file_t lf, const char *name,
1505c21bc6f3SBojan Novković     c_linker_sym_t *sym, linker_ctf_t *lc)
1506c21bc6f3SBojan Novković {
1507c21bc6f3SBojan Novković 	if (link_elf_lookup_debug_symbol(lf, name, sym))
1508c21bc6f3SBojan Novković 		return (ENOENT);
1509c21bc6f3SBojan Novković 
1510c21bc6f3SBojan Novković 	return (link_elf_ctf_get_ddb(lf, lc));
1511c21bc6f3SBojan Novković }
1512c21bc6f3SBojan Novković 
1513c21bc6f3SBojan Novković static int
link_elf_symbol_values1(linker_file_t lf,c_linker_sym_t sym,linker_symval_t * symval,bool see_local)151495c20fafSKonstantin Belousov link_elf_symbol_values1(linker_file_t lf, c_linker_sym_t sym,
151595c20fafSKonstantin Belousov     linker_symval_t *symval, bool see_local)
1516a4f67738SDoug Rabson {
15177035cf14SKonstantin Belousov 	elf_file_t ef;
15187035cf14SKonstantin Belousov 	const Elf_Sym *es;
15197035cf14SKonstantin Belousov 	caddr_t val;
1520a4f67738SDoug Rabson 
15217035cf14SKonstantin Belousov 	ef = (elf_file_t) lf;
15227035cf14SKonstantin Belousov 	es = (const Elf_Sym*) sym;
15237035cf14SKonstantin Belousov 	val = (caddr_t)es->st_value;
15245cf87418SMarcel Moolenaar 	if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) {
152595c20fafSKonstantin Belousov 		if (!see_local && ELF_ST_BIND(es->st_info) == STB_LOCAL)
152695c20fafSKonstantin Belousov 			return (ENOENT);
15272d636ab0SPeter Wemm 		symval->name = ef->ddbstrtab + es->st_name;
15287035cf14SKonstantin Belousov 		val = (caddr_t)es->st_value;
15297035cf14SKonstantin Belousov 		if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC)
15307035cf14SKonstantin Belousov 			val = ((caddr_t (*)(void))val)();
15317035cf14SKonstantin Belousov 		symval->value = val;
15322d636ab0SPeter Wemm 		symval->size = es->st_size;
1533a7e4eb14SKonstantin Belousov 		return (0);
15342d636ab0SPeter Wemm 	}
1535a7e4eb14SKonstantin Belousov 	return (ENOENT);
15362d636ab0SPeter Wemm }
1537a4f67738SDoug Rabson 
1538a4f67738SDoug Rabson static int
link_elf_symbol_values(linker_file_t lf,c_linker_sym_t sym,linker_symval_t * symval)153995c20fafSKonstantin Belousov link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym,
154095c20fafSKonstantin Belousov     linker_symval_t *symval)
154195c20fafSKonstantin Belousov {
1542ecd8245eSKonstantin Belousov 	return (link_elf_symbol_values1(lf, sym, symval,
1543ecd8245eSKonstantin Belousov 	    link_elf_obj_leak_locals));
154495c20fafSKonstantin Belousov }
154595c20fafSKonstantin Belousov 
154695c20fafSKonstantin Belousov static int
link_elf_debug_symbol_values(linker_file_t lf,c_linker_sym_t sym,linker_symval_t * symval)154795c20fafSKonstantin Belousov link_elf_debug_symbol_values(linker_file_t lf, c_linker_sym_t sym,
154895c20fafSKonstantin Belousov     linker_symval_t *symval)
154995c20fafSKonstantin Belousov {
155095c20fafSKonstantin Belousov 	return (link_elf_symbol_values1(lf, sym, symval, true));
155195c20fafSKonstantin Belousov }
155295c20fafSKonstantin Belousov 
155395c20fafSKonstantin Belousov static int
link_elf_search_symbol(linker_file_t lf,caddr_t value,c_linker_sym_t * sym,long * diffp)1554a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value,
1555d254af07SMatthew Dillon     c_linker_sym_t *sym, long *diffp)
1556a4f67738SDoug Rabson {
1557326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t)lf;
1558586453feSBruce Evans 	u_long off = (uintptr_t)(void *)value;
1559a4f67738SDoug Rabson 	u_long diff = off;
1560586453feSBruce Evans 	u_long st_value;
1561fe3db7c7SDoug Rabson 	const Elf_Sym *es;
1562b85f65afSPedro F. Giffuni 	const Elf_Sym *best = NULL;
1563a4f67738SDoug Rabson 	int i;
1564a4f67738SDoug Rabson 
15652d636ab0SPeter Wemm 	for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) {
1566a4f67738SDoug Rabson 		if (es->st_name == 0)
1567a4f67738SDoug Rabson 			continue;
1568a8774e39SPeter Wemm 		st_value = es->st_value;
1569b5abfb70SPeter Wemm 		if (off >= st_value) {
1570b5abfb70SPeter Wemm 			if (off - st_value < diff) {
1571b5abfb70SPeter Wemm 				diff = off - st_value;
1572a4f67738SDoug Rabson 				best = es;
1573a4f67738SDoug Rabson 				if (diff == 0)
1574a4f67738SDoug Rabson 					break;
1575b5abfb70SPeter Wemm 			} else if (off - st_value == diff) {
1576a4f67738SDoug Rabson 				best = es;
1577a4f67738SDoug Rabson 			}
1578a4f67738SDoug Rabson 		}
1579a4f67738SDoug Rabson 	}
1580b85f65afSPedro F. Giffuni 	if (best == NULL)
1581a4f67738SDoug Rabson 		*diffp = off;
1582a4f67738SDoug Rabson 	else
1583a4f67738SDoug Rabson 		*diffp = diff;
1584d254af07SMatthew Dillon 	*sym = (c_linker_sym_t) best;
1585a4f67738SDoug Rabson 
1586a7e4eb14SKonstantin Belousov 	return (0);
1587a4f67738SDoug Rabson }
1588f41325dbSPeter Wemm 
1589f41325dbSPeter Wemm /*
1590f41325dbSPeter Wemm  * Look up a linker set on an ELF system.
1591f41325dbSPeter Wemm  */
1592f41325dbSPeter Wemm static int
link_elf_lookup_set(linker_file_t lf,const char * name,void *** startp,void *** stopp,int * countp)1593f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name,
1594f41325dbSPeter Wemm     void ***startp, void ***stopp, int *countp)
1595f41325dbSPeter Wemm {
1596e9eabf59SPeter Wemm 	elf_file_t ef = (elf_file_t)lf;
1597f41325dbSPeter Wemm 	void **start, **stop;
1598e9eabf59SPeter Wemm 	int i, count;
1599f41325dbSPeter Wemm 
1600e9eabf59SPeter Wemm 	/* Relative to section number */
1601e9eabf59SPeter Wemm 	for (i = 0; i < ef->nprogtab; i++) {
1602e9eabf59SPeter Wemm 		if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) &&
1603e9eabf59SPeter Wemm 		    strcmp(ef->progtab[i].name + 4, name) == 0) {
1604e9eabf59SPeter Wemm 			start  = (void **)ef->progtab[i].addr;
160520947801SPeter Wemm 			stop = (void **)((char *)ef->progtab[i].addr +
160620947801SPeter Wemm 			    ef->progtab[i].size);
1607f41325dbSPeter Wemm 			count = stop - start;
1608f41325dbSPeter Wemm 			if (startp)
1609f41325dbSPeter Wemm 				*startp = start;
1610f41325dbSPeter Wemm 			if (stopp)
1611f41325dbSPeter Wemm 				*stopp = stop;
1612f41325dbSPeter Wemm 			if (countp)
1613f41325dbSPeter Wemm 				*countp = count;
1614e9eabf59SPeter Wemm 			return (0);
1615e9eabf59SPeter Wemm 		}
1616e9eabf59SPeter Wemm 	}
1617e9eabf59SPeter Wemm 	return (ESRCH);
1618f41325dbSPeter Wemm }
1619bb9fe9ddSBrian Feldman 
1620bb9fe9ddSBrian Feldman static int
link_elf_each_function_name(linker_file_t file,int (* callback)(const char *,void *),void * opaque)1621bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file,
1622e9eabf59SPeter Wemm     int (*callback)(const char *, void *), void *opaque)
1623e9eabf59SPeter Wemm {
1624bb9fe9ddSBrian Feldman 	elf_file_t ef = (elf_file_t)file;
1625bb9fe9ddSBrian Feldman 	const Elf_Sym *symp;
1626bb9fe9ddSBrian Feldman 	int i, error;
1627bb9fe9ddSBrian Feldman 
1628bb9fe9ddSBrian Feldman 	/* Exhaustive search */
1629bb9fe9ddSBrian Feldman 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1630bb9fe9ddSBrian Feldman 		if (symp->st_value != 0 &&
16317035cf14SKonstantin Belousov 		    (ELF_ST_TYPE(symp->st_info) == STT_FUNC ||
16327035cf14SKonstantin Belousov 		    ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) {
1633bb9fe9ddSBrian Feldman 			error = callback(ef->ddbstrtab + symp->st_name, opaque);
1634bb9fe9ddSBrian Feldman 			if (error)
1635bb9fe9ddSBrian Feldman 				return (error);
1636bb9fe9ddSBrian Feldman 		}
1637bb9fe9ddSBrian Feldman 	}
1638bb9fe9ddSBrian Feldman 	return (0);
1639bb9fe9ddSBrian Feldman }
164084201059SMarcel Moolenaar 
1641a2024a3eSJohn Birrell static int
link_elf_each_function_nameval(linker_file_t file,linker_function_nameval_callback_t callback,void * opaque)1642a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file,
1643a2024a3eSJohn Birrell     linker_function_nameval_callback_t callback, void *opaque)
1644a2024a3eSJohn Birrell {
1645a2024a3eSJohn Birrell 	linker_symval_t symval;
1646a2024a3eSJohn Birrell 	elf_file_t ef = (elf_file_t)file;
1647a2024a3eSJohn Birrell 	const Elf_Sym *symp;
1648a2024a3eSJohn Birrell 	int i, error;
1649a2024a3eSJohn Birrell 
1650a2024a3eSJohn Birrell 	/* Exhaustive search */
1651a2024a3eSJohn Birrell 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1652a2024a3eSJohn Birrell 		if (symp->st_value != 0 &&
16537035cf14SKonstantin Belousov 		    (ELF_ST_TYPE(symp->st_info) == STT_FUNC ||
16547035cf14SKonstantin Belousov 		    ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) {
165595c20fafSKonstantin Belousov 			error = link_elf_debug_symbol_values(file,
16567035cf14SKonstantin Belousov 			    (c_linker_sym_t)symp, &symval);
165795c20fafSKonstantin Belousov 			if (error == 0)
1658a2024a3eSJohn Birrell 				error = callback(file, i, &symval, opaque);
165995c20fafSKonstantin Belousov 			if (error != 0)
1660a2024a3eSJohn Birrell 				return (error);
1661a2024a3eSJohn Birrell 		}
1662a2024a3eSJohn Birrell 	}
1663a2024a3eSJohn Birrell 	return (0);
1664a2024a3eSJohn Birrell }
1665a2024a3eSJohn Birrell 
16662832cd54SKonstantin Belousov static void
elf_obj_cleanup_globals_cache(elf_file_t ef)16672832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef)
16682832cd54SKonstantin Belousov {
16692832cd54SKonstantin Belousov 	Elf_Sym *sym;
16702832cd54SKonstantin Belousov 	Elf_Size i;
16712832cd54SKonstantin Belousov 
16722832cd54SKonstantin Belousov 	for (i = 0; i < ef->ddbsymcnt; i++) {
16732832cd54SKonstantin Belousov 		sym = ef->ddbsymtab + i;
16742832cd54SKonstantin Belousov 		if (sym->st_shndx == SHN_FBSD_CACHED) {
16752832cd54SKonstantin Belousov 			sym->st_shndx = SHN_UNDEF;
16762832cd54SKonstantin Belousov 			sym->st_value = 0;
16772832cd54SKonstantin Belousov 		}
16782832cd54SKonstantin Belousov 	}
16792832cd54SKonstantin Belousov }
16802832cd54SKonstantin Belousov 
1681d297ad16SMarcel Moolenaar /*
1682d297ad16SMarcel Moolenaar  * Symbol lookup function that can be used when the symbol index is known (ie
1683d297ad16SMarcel Moolenaar  * in relocations). It uses the symbol index instead of doing a fully fledged
1684d297ad16SMarcel Moolenaar  * hash table based lookup when such is valid. For example for local symbols.
1685d297ad16SMarcel Moolenaar  * This is not only more efficient, it's also more correct. It's not always
1686d297ad16SMarcel Moolenaar  * the case that the symbol can be found through the hash table.
1687d297ad16SMarcel Moolenaar  */
1688cff8c6f2SKonstantin Belousov static int
elf_obj_lookup(linker_file_t lf,Elf_Size symidx,int deps,Elf_Addr * res)1689cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res)
1690d297ad16SMarcel Moolenaar {
1691d297ad16SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
16922832cd54SKonstantin Belousov 	Elf_Sym *sym;
1693d297ad16SMarcel Moolenaar 	const char *symbol;
1694cff8c6f2SKonstantin Belousov 	Elf_Addr res1;
1695d297ad16SMarcel Moolenaar 
1696d297ad16SMarcel Moolenaar 	/* Don't even try to lookup the symbol if the index is bogus. */
1697cff8c6f2SKonstantin Belousov 	if (symidx >= ef->ddbsymcnt) {
1698cff8c6f2SKonstantin Belousov 		*res = 0;
1699cff8c6f2SKonstantin Belousov 		return (EINVAL);
1700cff8c6f2SKonstantin Belousov 	}
1701d297ad16SMarcel Moolenaar 
1702e9eabf59SPeter Wemm 	sym = ef->ddbsymtab + symidx;
1703d297ad16SMarcel Moolenaar 
17044cec6f5dSPeter Wemm 	/* Quick answer if there is a definition included. */
1705cff8c6f2SKonstantin Belousov 	if (sym->st_shndx != SHN_UNDEF) {
1706289ead7cSKonstantin Belousov 		res1 = (Elf_Addr)sym->st_value;
1707289ead7cSKonstantin Belousov 		if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC)
1708289ead7cSKonstantin Belousov 			res1 = ((Elf_Addr (*)(void))res1)();
1709289ead7cSKonstantin Belousov 		*res = res1;
1710cff8c6f2SKonstantin Belousov 		return (0);
1711cff8c6f2SKonstantin Belousov 	}
17124cec6f5dSPeter Wemm 
17134cec6f5dSPeter Wemm 	/* If we get here, then it is undefined and needs a lookup. */
17144cec6f5dSPeter Wemm 	switch (ELF_ST_BIND(sym->st_info)) {
17154cec6f5dSPeter Wemm 	case STB_LOCAL:
17164cec6f5dSPeter Wemm 		/* Local, but undefined? huh? */
1717cff8c6f2SKonstantin Belousov 		*res = 0;
1718cff8c6f2SKonstantin Belousov 		return (EINVAL);
1719d297ad16SMarcel Moolenaar 
1720e9eabf59SPeter Wemm 	case STB_GLOBAL:
1721cff8c6f2SKonstantin Belousov 	case STB_WEAK:
1722e9eabf59SPeter Wemm 		/* Relative to Data or Function name */
1723e9eabf59SPeter Wemm 		symbol = ef->ddbstrtab + sym->st_name;
1724d297ad16SMarcel Moolenaar 
1725d297ad16SMarcel Moolenaar 		/* Force a lookup failure if the symbol name is bogus. */
1726cff8c6f2SKonstantin Belousov 		if (*symbol == 0) {
1727cff8c6f2SKonstantin Belousov 			*res = 0;
1728cff8c6f2SKonstantin Belousov 			return (EINVAL);
1729cff8c6f2SKonstantin Belousov 		}
1730cff8c6f2SKonstantin Belousov 		res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps);
17312832cd54SKonstantin Belousov 
17322832cd54SKonstantin Belousov 		/*
17332832cd54SKonstantin Belousov 		 * Cache global lookups during module relocation. The failure
17342832cd54SKonstantin Belousov 		 * case is particularly expensive for callers, who must scan
17352832cd54SKonstantin Belousov 		 * through the entire globals table doing strcmp(). Cache to
17362832cd54SKonstantin Belousov 		 * avoid doing such work repeatedly.
17372832cd54SKonstantin Belousov 		 *
17382832cd54SKonstantin Belousov 		 * After relocation is complete, undefined globals will be
17392832cd54SKonstantin Belousov 		 * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(),
17402832cd54SKonstantin Belousov 		 * above.
17412832cd54SKonstantin Belousov 		 */
1742cff8c6f2SKonstantin Belousov 		if (res1 != 0) {
17432832cd54SKonstantin Belousov 			sym->st_shndx = SHN_FBSD_CACHED;
1744cff8c6f2SKonstantin Belousov 			sym->st_value = res1;
1745cff8c6f2SKonstantin Belousov 			*res = res1;
1746e9eabf59SPeter Wemm 			return (0);
1747cff8c6f2SKonstantin Belousov 		} else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) {
1748cff8c6f2SKonstantin Belousov 			sym->st_value = 0;
1749cff8c6f2SKonstantin Belousov 			*res = 0;
1750cff8c6f2SKonstantin Belousov 			return (0);
1751cff8c6f2SKonstantin Belousov 		}
1752cff8c6f2SKonstantin Belousov 		return (EINVAL);
1753e9eabf59SPeter Wemm 
1754e9eabf59SPeter Wemm 	default:
1755cff8c6f2SKonstantin Belousov 		return (EINVAL);
1756d297ad16SMarcel Moolenaar 	}
1757e9eabf59SPeter Wemm }
1758e9eabf59SPeter Wemm 
17597251b4bfSJake Burkholder static void
link_elf_fix_link_set(elf_file_t ef)17607226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef)
17617226306eSKonstantin Belousov {
17627226306eSKonstantin Belousov 	static const char startn[] = "__start_";
17637226306eSKonstantin Belousov 	static const char stopn[] = "__stop_";
17647226306eSKonstantin Belousov 	Elf_Sym *sym;
17657226306eSKonstantin Belousov 	const char *sym_name, *linkset_name;
17667226306eSKonstantin Belousov 	Elf_Addr startp, stopp;
17677226306eSKonstantin Belousov 	Elf_Size symidx;
17687226306eSKonstantin Belousov 	int start, i;
17697226306eSKonstantin Belousov 
17707226306eSKonstantin Belousov 	startp = stopp = 0;
17717226306eSKonstantin Belousov 	for (symidx = 1 /* zero entry is special */;
17727226306eSKonstantin Belousov 		symidx < ef->ddbsymcnt; symidx++) {
17737226306eSKonstantin Belousov 		sym = ef->ddbsymtab + symidx;
17747226306eSKonstantin Belousov 		if (sym->st_shndx != SHN_UNDEF)
17757226306eSKonstantin Belousov 			continue;
17767226306eSKonstantin Belousov 
17777226306eSKonstantin Belousov 		sym_name = ef->ddbstrtab + sym->st_name;
17787226306eSKonstantin Belousov 		if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) {
17797226306eSKonstantin Belousov 			start = 1;
17807226306eSKonstantin Belousov 			linkset_name = sym_name + sizeof(startn) - 1;
17817226306eSKonstantin Belousov 		}
17827226306eSKonstantin Belousov 		else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) {
17837226306eSKonstantin Belousov 			start = 0;
17847226306eSKonstantin Belousov 			linkset_name = sym_name + sizeof(stopn) - 1;
17857226306eSKonstantin Belousov 		}
17867226306eSKonstantin Belousov 		else
17877226306eSKonstantin Belousov 			continue;
17887226306eSKonstantin Belousov 
17897226306eSKonstantin Belousov 		for (i = 0; i < ef->nprogtab; i++) {
17907226306eSKonstantin Belousov 			if (strcmp(ef->progtab[i].name, linkset_name) == 0) {
17917226306eSKonstantin Belousov 				startp = (Elf_Addr)ef->progtab[i].addr;
17927226306eSKonstantin Belousov 				stopp = (Elf_Addr)(startp + ef->progtab[i].size);
17937226306eSKonstantin Belousov 				break;
17947226306eSKonstantin Belousov 			}
17957226306eSKonstantin Belousov 		}
17967226306eSKonstantin Belousov 		if (i == ef->nprogtab)
17977226306eSKonstantin Belousov 			continue;
17987226306eSKonstantin Belousov 
17997226306eSKonstantin Belousov 		sym->st_value = start ? startp : stopp;
18007226306eSKonstantin Belousov 		sym->st_shndx = i;
18017226306eSKonstantin Belousov 	}
18027226306eSKonstantin Belousov }
18037226306eSKonstantin Belousov 
18043cfce8e4SKonstantin Belousov static int
link_elf_reloc_local(linker_file_t lf,bool ifuncs)1805289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs)
18067251b4bfSJake Burkholder {
1807e9eabf59SPeter Wemm 	elf_file_t ef = (elf_file_t)lf;
18087251b4bfSJake Burkholder 	const Elf_Rel *rellim;
18097251b4bfSJake Burkholder 	const Elf_Rel *rel;
18107251b4bfSJake Burkholder 	const Elf_Rela *relalim;
18117251b4bfSJake Burkholder 	const Elf_Rela *rela;
1812e9eabf59SPeter Wemm 	const Elf_Sym *sym;
1813e9eabf59SPeter Wemm 	Elf_Addr base;
1814e9eabf59SPeter Wemm 	int i;
1815757686b1SMarcel Moolenaar 	Elf_Size symidx;
1816e9eabf59SPeter Wemm 
18177226306eSKonstantin Belousov 	link_elf_fix_link_set(ef);
18187226306eSKonstantin Belousov 
18197251b4bfSJake Burkholder 	/* Perform relocations without addend if there are any: */
1820b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++) {
1821e9eabf59SPeter Wemm 		rel = ef->reltab[i].rel;
18223cfce8e4SKonstantin Belousov 		if (rel == NULL) {
18233cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a reltab");
18243cfce8e4SKonstantin Belousov 			return (ENOEXEC);
18253cfce8e4SKonstantin Belousov 		}
182620947801SPeter Wemm 		rellim = rel + ef->reltab[i].nrel;
1827e9eabf59SPeter Wemm 		base = findbase(ef, ef->reltab[i].sec);
18283cfce8e4SKonstantin Belousov 		if (base == 0) {
18293cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost base for reltab");
18303cfce8e4SKonstantin Belousov 			return (ENOEXEC);
18313cfce8e4SKonstantin Belousov 		}
183220947801SPeter Wemm 		for ( ; rel < rellim; rel++) {
1833e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rel->r_info);
183420947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
183520947801SPeter Wemm 				continue;
1836e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
183720947801SPeter Wemm 			/* Only do local relocs */
183820947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) != STB_LOCAL)
183920947801SPeter Wemm 				continue;
184036716fe2SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
184144c705cfSMitchell Horne 			    elf_is_ifunc_reloc(rel->r_info)) != ifuncs)
184244c705cfSMitchell Horne 				continue;
184344c705cfSMitchell Horne 			if (elf_reloc_local(lf, base, rel, ELF_RELOC_REL,
184444c705cfSMitchell Horne 			    elf_obj_lookup) != 0)
184544c705cfSMitchell Horne 				return (ENOEXEC);
18467251b4bfSJake Burkholder 		}
1847e9eabf59SPeter Wemm 	}
18487251b4bfSJake Burkholder 
18497251b4bfSJake Burkholder 	/* Perform relocations with addend if there are any: */
1850b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++) {
1851e9eabf59SPeter Wemm 		rela = ef->relatab[i].rela;
18523cfce8e4SKonstantin Belousov 		if (rela == NULL) {
18533cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a relatab!");
18543cfce8e4SKonstantin Belousov 			return (ENOEXEC);
18553cfce8e4SKonstantin Belousov 		}
185620947801SPeter Wemm 		relalim = rela + ef->relatab[i].nrela;
1857e9eabf59SPeter Wemm 		base = findbase(ef, ef->relatab[i].sec);
18583cfce8e4SKonstantin Belousov 		if (base == 0) {
18593cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost base for reltab");
18603cfce8e4SKonstantin Belousov 			return (ENOEXEC);
18613cfce8e4SKonstantin Belousov 		}
186220947801SPeter Wemm 		for ( ; rela < relalim; rela++) {
1863e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rela->r_info);
186420947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
186520947801SPeter Wemm 				continue;
1866e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
186720947801SPeter Wemm 			/* Only do local relocs */
186820947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) != STB_LOCAL)
186920947801SPeter Wemm 				continue;
187036716fe2SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
187144c705cfSMitchell Horne 			    elf_is_ifunc_reloc(rela->r_info)) != ifuncs)
187244c705cfSMitchell Horne 				continue;
187344c705cfSMitchell Horne 			if (elf_reloc_local(lf, base, rela, ELF_RELOC_RELA,
187444c705cfSMitchell Horne 			    elf_obj_lookup) != 0)
187544c705cfSMitchell Horne 				return (ENOEXEC);
18767251b4bfSJake Burkholder 		}
18777251b4bfSJake Burkholder 	}
18783cfce8e4SKonstantin Belousov 	return (0);
1879e9eabf59SPeter Wemm }
188000a5db46SStacey Son 
188100a5db46SStacey Son static long
link_elf_symtab_get(linker_file_t lf,const Elf_Sym ** symtab)1882e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab)
188300a5db46SStacey Son {
188400a5db46SStacey Son 	elf_file_t ef = (elf_file_t)lf;
188500a5db46SStacey Son 
188600a5db46SStacey Son 	*symtab = ef->ddbsymtab;
188700a5db46SStacey Son 	if (*symtab == NULL)
188800a5db46SStacey Son 		return (0);
188900a5db46SStacey Son 	return (ef->ddbsymcnt);
189000a5db46SStacey Son }
189100a5db46SStacey Son 
189200a5db46SStacey Son static long
link_elf_strtab_get(linker_file_t lf,caddr_t * strtab)189300a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab)
189400a5db46SStacey Son {
189500a5db46SStacey Son 	elf_file_t ef = (elf_file_t)lf;
189600a5db46SStacey Son 
189700a5db46SStacey Son 	*strtab = ef->ddbstrtab;
189800a5db46SStacey Son 	if (*strtab == NULL)
189900a5db46SStacey Son 		return (0);
190000a5db46SStacey Son 	return (ef->ddbstrcnt);
190100a5db46SStacey Son }
1902110113bcSZhenlei Huang 
1903110113bcSZhenlei Huang #ifdef VIMAGE
1904110113bcSZhenlei Huang static void
link_elf_propagate_vnets(linker_file_t lf)1905110113bcSZhenlei Huang link_elf_propagate_vnets(linker_file_t lf)
1906110113bcSZhenlei Huang {
1907110113bcSZhenlei Huang 	elf_file_t ef = (elf_file_t) lf;
1908110113bcSZhenlei Huang 
1909110113bcSZhenlei Huang 	if (ef->progtab) {
1910110113bcSZhenlei Huang 		for (int i = 0; i < ef->nprogtab; i++) {
1911110113bcSZhenlei Huang 			if (ef->progtab[i].size == 0)
1912110113bcSZhenlei Huang 				continue;
1913110113bcSZhenlei Huang 			if (ef->progtab[i].name == NULL)
1914110113bcSZhenlei Huang 				continue;
1915110113bcSZhenlei Huang 			if (strcmp(ef->progtab[i].name, VNET_SETNAME) == 0) {
1916110113bcSZhenlei Huang 				vnet_data_copy(ef->progtab[i].addr,
1917110113bcSZhenlei Huang 				    ef->progtab[i].size);
1918110113bcSZhenlei Huang 				break;
1919110113bcSZhenlei Huang 			}
1920110113bcSZhenlei Huang 		}
1921110113bcSZhenlei Huang 	}
1922110113bcSZhenlei Huang }
1923110113bcSZhenlei Huang #endif
1924