xref: /freebsd/sys/kern/link_elf_obj.c (revision 9afc56849a2e3a58e853868599faf727e33b87d9)
1a4f67738SDoug Rabson /*-
28a36da99SPedro F. Giffuni  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
38a36da99SPedro F. Giffuni  *
4326e27d8SDoug Rabson  * Copyright (c) 1998-2000 Doug Rabson
5e9eabf59SPeter Wemm  * Copyright (c) 2004 Peter Wemm
6a4f67738SDoug Rabson  * All rights reserved.
7a4f67738SDoug Rabson  *
8a4f67738SDoug Rabson  * Redistribution and use in source and binary forms, with or without
9a4f67738SDoug Rabson  * modification, are permitted provided that the following conditions
10a4f67738SDoug Rabson  * are met:
11a4f67738SDoug Rabson  * 1. Redistributions of source code must retain the above copyright
12a4f67738SDoug Rabson  *    notice, this list of conditions and the following disclaimer.
13a4f67738SDoug Rabson  * 2. Redistributions in binary form must reproduce the above copyright
14a4f67738SDoug Rabson  *    notice, this list of conditions and the following disclaimer in the
15a4f67738SDoug Rabson  *    documentation and/or other materials provided with the distribution.
16a4f67738SDoug Rabson  *
17a4f67738SDoug Rabson  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18a4f67738SDoug Rabson  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19a4f67738SDoug Rabson  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20a4f67738SDoug Rabson  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21a4f67738SDoug Rabson  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22a4f67738SDoug Rabson  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23a4f67738SDoug Rabson  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24a4f67738SDoug Rabson  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25a4f67738SDoug Rabson  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26a4f67738SDoug Rabson  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27a4f67738SDoug Rabson  * SUCH DAMAGE.
28a4f67738SDoug Rabson  */
29a4f67738SDoug Rabson 
30677b542eSDavid E. O'Brien #include <sys/cdefs.h>
31677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$");
32677b542eSDavid E. O'Brien 
33326e27d8SDoug Rabson #include "opt_ddb.h"
34326e27d8SDoug Rabson 
35a4f67738SDoug Rabson #include <sys/param.h>
36a4f67738SDoug Rabson #include <sys/systm.h>
37fb919e4dSMark Murray #include <sys/kernel.h>
38fb919e4dSMark Murray #include <sys/lock.h>
39a4f67738SDoug Rabson #include <sys/malloc.h>
40c0b824f9SDima Dorfman #include <sys/mutex.h>
4162d615d5SJohn Baldwin #include <sys/mount.h>
42a4f67738SDoug Rabson #include <sys/proc.h>
43a4f67738SDoug Rabson #include <sys/namei.h>
44a4f67738SDoug Rabson #include <sys/fcntl.h>
45a4f67738SDoug Rabson #include <sys/vnode.h>
46a4f67738SDoug Rabson #include <sys/linker.h>
47fb919e4dSMark Murray 
48a4f67738SDoug Rabson #include <machine/elf.h>
49a4f67738SDoug Rabson 
50eddfbb76SRobert Watson #include <net/vnet.h>
51eddfbb76SRobert Watson 
52aed55708SRobert Watson #include <security/mac/mac_framework.h>
53aed55708SRobert Watson 
54fe3db7c7SDoug Rabson #include <vm/vm.h>
55fe3db7c7SDoug Rabson #include <vm/vm_param.h>
56fe3db7c7SDoug Rabson #include <vm/vm_object.h>
57fe3db7c7SDoug Rabson #include <vm/vm_kern.h>
58fe3db7c7SDoug Rabson #include <vm/vm_extern.h>
59fe3db7c7SDoug Rabson #include <vm/pmap.h>
60fe3db7c7SDoug Rabson #include <vm/vm_map.h>
61fb919e4dSMark Murray 
62f9980387SPeter Wemm #include <sys/link_elf.h>
63fe3db7c7SDoug Rabson 
64a2024a3eSJohn Birrell #ifdef DDB_CTF
65d9db5225SCraig Rodrigues #include <sys/zlib.h>
66a2024a3eSJohn Birrell #endif
67a2024a3eSJohn Birrell 
68326e27d8SDoug Rabson #include "linker_if.h"
69a4f67738SDoug Rabson 
70e9eabf59SPeter Wemm typedef struct {
71e9eabf59SPeter Wemm 	void		*addr;
7220947801SPeter Wemm 	Elf_Off		size;
73e9eabf59SPeter Wemm 	int		flags;
74e9eabf59SPeter Wemm 	int		sec;	/* Original section */
75e9eabf59SPeter Wemm 	char		*name;
76e9eabf59SPeter Wemm } Elf_progent;
77e9eabf59SPeter Wemm 
78e9eabf59SPeter Wemm typedef struct {
79e9eabf59SPeter Wemm 	Elf_Rel		*rel;
8020947801SPeter Wemm 	int		nrel;
81e9eabf59SPeter Wemm 	int		sec;
82e9eabf59SPeter Wemm } Elf_relent;
83e9eabf59SPeter Wemm 
84e9eabf59SPeter Wemm typedef struct {
85e9eabf59SPeter Wemm 	Elf_Rela	*rela;
8620947801SPeter Wemm 	int		nrela;
87e9eabf59SPeter Wemm 	int		sec;
88e9eabf59SPeter Wemm } Elf_relaent;
89e9eabf59SPeter Wemm 
90e9eabf59SPeter Wemm 
91a4f67738SDoug Rabson typedef struct elf_file {
92326e27d8SDoug Rabson 	struct linker_file lf;		/* Common fields */
9320947801SPeter Wemm 
9470b7ffeeSIan Dowse 	int		preloaded;
95fe3db7c7SDoug Rabson 	caddr_t		address;	/* Relocation address */
96fe3db7c7SDoug Rabson 	vm_object_t	object;		/* VM object to hold file pages */
97e9eabf59SPeter Wemm 	Elf_Shdr	*e_shdr;
98e9eabf59SPeter Wemm 
99e9eabf59SPeter Wemm 	Elf_progent	*progtab;
100d821d364SPedro F. Giffuni 	u_int		nprogtab;
101e9eabf59SPeter Wemm 
102e9eabf59SPeter Wemm 	Elf_relaent	*relatab;
103d821d364SPedro F. Giffuni 	u_int		nrelatab;
104e9eabf59SPeter Wemm 
105e9eabf59SPeter Wemm 	Elf_relent	*reltab;
106b474c780SDavid E. O'Brien 	int		nreltab;
107e9eabf59SPeter Wemm 
108e9eabf59SPeter Wemm 	Elf_Sym		*ddbsymtab;	/* The symbol table we are using */
1092d636ab0SPeter Wemm 	long		ddbsymcnt;	/* Number of symbols */
1102d636ab0SPeter Wemm 	caddr_t		ddbstrtab;	/* String table */
1112d636ab0SPeter Wemm 	long		ddbstrcnt;	/* number of bytes in string table */
112e9eabf59SPeter Wemm 
113e9eabf59SPeter Wemm 	caddr_t		shstrtab;	/* Section name string table */
114e9eabf59SPeter Wemm 	long		shstrcnt;	/* number of bytes in string table */
115e9eabf59SPeter Wemm 
116a2024a3eSJohn Birrell 	caddr_t		ctftab;		/* CTF table */
117a2024a3eSJohn Birrell 	long		ctfcnt;		/* number of bytes in CTF table */
118a2024a3eSJohn Birrell 	caddr_t		ctfoff;		/* CTF offset table */
119a2024a3eSJohn Birrell 	caddr_t		typoff;		/* Type offset table */
120a2024a3eSJohn Birrell 	long		typlen;		/* Number of type entries. */
121a2024a3eSJohn Birrell 
122a4f67738SDoug Rabson } *elf_file_t;
123a4f67738SDoug Rabson 
124a2024a3eSJohn Birrell #include <kern/kern_ctf.c>
125a2024a3eSJohn Birrell 
12654823af2SPeter Wemm static int	link_elf_link_preload(linker_class_t cls,
127326e27d8SDoug Rabson 		    const char *, linker_file_t *);
12854823af2SPeter Wemm static int	link_elf_link_preload_finish(linker_file_t);
12954823af2SPeter Wemm static int	link_elf_load_file(linker_class_t, const char *, linker_file_t *);
130326e27d8SDoug Rabson static int	link_elf_lookup_symbol(linker_file_t, const char *,
131326e27d8SDoug Rabson 		    c_linker_sym_t *);
132e9eabf59SPeter Wemm static int	link_elf_symbol_values(linker_file_t, c_linker_sym_t,
133e9eabf59SPeter Wemm 		    linker_symval_t *);
134326e27d8SDoug Rabson static int	link_elf_search_symbol(linker_file_t, caddr_t value,
135326e27d8SDoug Rabson 		    c_linker_sym_t *sym, long *diffp);
136326e27d8SDoug Rabson 
137326e27d8SDoug Rabson static void	link_elf_unload_file(linker_file_t);
138f41325dbSPeter Wemm static int	link_elf_lookup_set(linker_file_t, const char *,
139f41325dbSPeter Wemm 		    void ***, void ***, int *);
140bb9fe9ddSBrian Feldman static int	link_elf_each_function_name(linker_file_t,
141e9eabf59SPeter Wemm 		    int (*)(const char *, void *), void *);
142a2024a3eSJohn Birrell static int	link_elf_each_function_nameval(linker_file_t,
143a2024a3eSJohn Birrell 				linker_function_nameval_callback_t,
144a2024a3eSJohn Birrell 				void *);
145289ead7cSKonstantin Belousov static int	link_elf_reloc_local(linker_file_t, bool);
146e76f11f4SAndriy Gapon static long	link_elf_symtab_get(linker_file_t, const Elf_Sym **);
14700a5db46SStacey Son static long	link_elf_strtab_get(linker_file_t, caddr_t *);
148326e27d8SDoug Rabson 
149cff8c6f2SKonstantin Belousov static int	elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps,
150cff8c6f2SKonstantin Belousov 		    Elf_Addr *);
151e9eabf59SPeter Wemm 
152326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = {
153326e27d8SDoug Rabson 	KOBJMETHOD(linker_lookup_symbol,	link_elf_lookup_symbol),
154326e27d8SDoug Rabson 	KOBJMETHOD(linker_symbol_values,	link_elf_symbol_values),
155326e27d8SDoug Rabson 	KOBJMETHOD(linker_search_symbol,	link_elf_search_symbol),
156326e27d8SDoug Rabson 	KOBJMETHOD(linker_unload,		link_elf_unload_file),
15754823af2SPeter Wemm 	KOBJMETHOD(linker_load_file,		link_elf_load_file),
15854823af2SPeter Wemm 	KOBJMETHOD(linker_link_preload,		link_elf_link_preload),
15954823af2SPeter Wemm 	KOBJMETHOD(linker_link_preload_finish,	link_elf_link_preload_finish),
160f41325dbSPeter Wemm 	KOBJMETHOD(linker_lookup_set,		link_elf_lookup_set),
161bb9fe9ddSBrian Feldman 	KOBJMETHOD(linker_each_function_name,	link_elf_each_function_name),
162a2024a3eSJohn Birrell 	KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval),
163a2024a3eSJohn Birrell 	KOBJMETHOD(linker_ctf_get,		link_elf_ctf_get),
16400a5db46SStacey Son 	KOBJMETHOD(linker_symtab_get, 		link_elf_symtab_get),
16500a5db46SStacey Son 	KOBJMETHOD(linker_strtab_get, 		link_elf_strtab_get),
166326e27d8SDoug Rabson 	{ 0, 0 }
167326e27d8SDoug Rabson };
168326e27d8SDoug Rabson 
169326e27d8SDoug Rabson static struct linker_class link_elf_class = {
170326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32
171e9eabf59SPeter Wemm 	"elf32_obj",
172326e27d8SDoug Rabson #else
173e9eabf59SPeter Wemm 	"elf64_obj",
174326e27d8SDoug Rabson #endif
175326e27d8SDoug Rabson 	link_elf_methods, sizeof(struct elf_file)
176326e27d8SDoug Rabson };
177326e27d8SDoug Rabson 
178326e27d8SDoug Rabson static int	relocate_file(elf_file_t ef);
1792832cd54SKonstantin Belousov static void	elf_obj_cleanup_globals_cache(elf_file_t);
180a4f67738SDoug Rabson 
181a4f67738SDoug Rabson static void
182552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s)
1831720979bSMarcel Moolenaar {
184552f9f63SEdwin Groothuis 	if (filename == NULL)
1851720979bSMarcel Moolenaar 		printf("kldload: %s\n", s);
186552f9f63SEdwin Groothuis 	else
187552f9f63SEdwin Groothuis 		printf("kldload: %s: %s\n", filename, s);
1881720979bSMarcel Moolenaar }
1891720979bSMarcel Moolenaar 
1901720979bSMarcel Moolenaar static void
191a4f67738SDoug Rabson link_elf_init(void *arg)
192a4f67738SDoug Rabson {
193a4f67738SDoug Rabson 
194326e27d8SDoug Rabson 	linker_add_class(&link_elf_class);
195a4f67738SDoug Rabson }
196a4f67738SDoug Rabson 
197891cf3edSEd Maste SYSINIT(link_elf_obj, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, NULL);
198a4f67738SDoug Rabson 
199a4f67738SDoug Rabson static int
200e9eabf59SPeter Wemm link_elf_link_preload(linker_class_t cls, const char *filename,
201e9eabf59SPeter Wemm     linker_file_t *result)
2022d636ab0SPeter Wemm {
20370b7ffeeSIan Dowse 	Elf_Ehdr *hdr;
20470b7ffeeSIan Dowse 	Elf_Shdr *shdr;
20570b7ffeeSIan Dowse 	Elf_Sym *es;
20670b7ffeeSIan Dowse 	void *modptr, *baseptr, *sizeptr;
20770b7ffeeSIan Dowse 	char *type;
20870b7ffeeSIan Dowse 	elf_file_t ef;
20970b7ffeeSIan Dowse 	linker_file_t lf;
21070b7ffeeSIan Dowse 	Elf_Addr off;
21170b7ffeeSIan Dowse 	int error, i, j, pb, ra, rl, shstrindex, symstrindex, symtabindex;
21270b7ffeeSIan Dowse 
21370b7ffeeSIan Dowse 	/* Look to see if we have the file preloaded */
21470b7ffeeSIan Dowse 	modptr = preload_search_by_name(filename);
21570b7ffeeSIan Dowse 	if (modptr == NULL)
21670b7ffeeSIan Dowse 		return ENOENT;
21770b7ffeeSIan Dowse 
21870b7ffeeSIan Dowse 	type = (char *)preload_search_info(modptr, MODINFO_TYPE);
21970b7ffeeSIan Dowse 	baseptr = preload_search_info(modptr, MODINFO_ADDR);
22070b7ffeeSIan Dowse 	sizeptr = preload_search_info(modptr, MODINFO_SIZE);
22170b7ffeeSIan Dowse 	hdr = (Elf_Ehdr *)preload_search_info(modptr, MODINFO_METADATA |
22270b7ffeeSIan Dowse 	    MODINFOMD_ELFHDR);
22370b7ffeeSIan Dowse 	shdr = (Elf_Shdr *)preload_search_info(modptr, MODINFO_METADATA |
22470b7ffeeSIan Dowse 	    MODINFOMD_SHDR);
22570b7ffeeSIan Dowse 	if (type == NULL || (strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE)
22670b7ffeeSIan Dowse 	    " obj module") != 0 &&
22770b7ffeeSIan Dowse 	    strcmp(type, "elf obj module") != 0)) {
228de78ca7eSPeter Wemm 		return (EFTYPE);
229de78ca7eSPeter Wemm 	}
23070b7ffeeSIan Dowse 	if (baseptr == NULL || sizeptr == NULL || hdr == NULL ||
23170b7ffeeSIan Dowse 	    shdr == NULL)
23270b7ffeeSIan Dowse 		return (EINVAL);
23370b7ffeeSIan Dowse 
23470b7ffeeSIan Dowse 	lf = linker_make_file(filename, &link_elf_class);
23570b7ffeeSIan Dowse 	if (lf == NULL)
23670b7ffeeSIan Dowse 		return (ENOMEM);
23770b7ffeeSIan Dowse 
23870b7ffeeSIan Dowse 	ef = (elf_file_t)lf;
23970b7ffeeSIan Dowse 	ef->preloaded = 1;
24070b7ffeeSIan Dowse 	ef->address = *(caddr_t *)baseptr;
24170b7ffeeSIan Dowse 	lf->address = *(caddr_t *)baseptr;
24270b7ffeeSIan Dowse 	lf->size = *(size_t *)sizeptr;
24370b7ffeeSIan Dowse 
24470b7ffeeSIan Dowse 	if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS ||
24570b7ffeeSIan Dowse 	    hdr->e_ident[EI_DATA] != ELF_TARG_DATA ||
24670b7ffeeSIan Dowse 	    hdr->e_ident[EI_VERSION] != EV_CURRENT ||
24770b7ffeeSIan Dowse 	    hdr->e_version != EV_CURRENT ||
24870b7ffeeSIan Dowse 	    hdr->e_type != ET_REL ||
24970b7ffeeSIan Dowse 	    hdr->e_machine != ELF_TARG_MACH) {
25070b7ffeeSIan Dowse 		error = EFTYPE;
25170b7ffeeSIan Dowse 		goto out;
25270b7ffeeSIan Dowse 	}
25370b7ffeeSIan Dowse 	ef->e_shdr = shdr;
25470b7ffeeSIan Dowse 
25570b7ffeeSIan Dowse 	/* Scan the section header for information and table sizing. */
25670b7ffeeSIan Dowse 	symtabindex = -1;
25770b7ffeeSIan Dowse 	symstrindex = -1;
25870b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
25970b7ffeeSIan Dowse 		switch (shdr[i].sh_type) {
26070b7ffeeSIan Dowse 		case SHT_PROGBITS:
26170b7ffeeSIan Dowse 		case SHT_NOBITS:
26213f28d96SKonstantin Belousov #ifdef __amd64__
263b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
26413f28d96SKonstantin Belousov #endif
26558c4aee0SJohn Baldwin 			/* Ignore sections not loaded by the loader. */
26658c4aee0SJohn Baldwin 			if (shdr[i].sh_addr == 0)
26758c4aee0SJohn Baldwin 				break;
26870b7ffeeSIan Dowse 			ef->nprogtab++;
26970b7ffeeSIan Dowse 			break;
27070b7ffeeSIan Dowse 		case SHT_SYMTAB:
27170b7ffeeSIan Dowse 			symtabindex = i;
27270b7ffeeSIan Dowse 			symstrindex = shdr[i].sh_link;
27370b7ffeeSIan Dowse 			break;
27470b7ffeeSIan Dowse 		case SHT_REL:
27515746ef4SJohn Baldwin 			/*
27615746ef4SJohn Baldwin 			 * Ignore relocation tables for sections not
27715746ef4SJohn Baldwin 			 * loaded by the loader.
27815746ef4SJohn Baldwin 			 */
27915746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
28015746ef4SJohn Baldwin 				break;
281b474c780SDavid E. O'Brien 			ef->nreltab++;
28270b7ffeeSIan Dowse 			break;
28370b7ffeeSIan Dowse 		case SHT_RELA:
28415746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
28515746ef4SJohn Baldwin 				break;
286b474c780SDavid E. O'Brien 			ef->nrelatab++;
28770b7ffeeSIan Dowse 			break;
28870b7ffeeSIan Dowse 		}
28970b7ffeeSIan Dowse 	}
29070b7ffeeSIan Dowse 
29170b7ffeeSIan Dowse 	shstrindex = hdr->e_shstrndx;
29270b7ffeeSIan Dowse 	if (ef->nprogtab == 0 || symstrindex < 0 ||
29370b7ffeeSIan Dowse 	    symstrindex >= hdr->e_shnum ||
29470b7ffeeSIan Dowse 	    shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 ||
29570b7ffeeSIan Dowse 	    shstrindex >= hdr->e_shnum ||
29670b7ffeeSIan Dowse 	    shdr[shstrindex].sh_type != SHT_STRTAB) {
29770b7ffeeSIan Dowse 		printf("%s: bad/missing section headers\n", filename);
29870b7ffeeSIan Dowse 		error = ENOEXEC;
29970b7ffeeSIan Dowse 		goto out;
30070b7ffeeSIan Dowse 	}
30170b7ffeeSIan Dowse 
30270b7ffeeSIan Dowse 	/* Allocate space for tracking the load chunks */
30370b7ffeeSIan Dowse 	if (ef->nprogtab != 0)
30470b7ffeeSIan Dowse 		ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab),
30570b7ffeeSIan Dowse 		    M_LINKER, M_WAITOK | M_ZERO);
306b474c780SDavid E. O'Brien 	if (ef->nreltab != 0)
307b474c780SDavid E. O'Brien 		ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab),
308b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
309b474c780SDavid E. O'Brien 	if (ef->nrelatab != 0)
310b474c780SDavid E. O'Brien 		ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab),
311b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
31270b7ffeeSIan Dowse 	if ((ef->nprogtab != 0 && ef->progtab == NULL) ||
313b474c780SDavid E. O'Brien 	    (ef->nreltab != 0 && ef->reltab == NULL) ||
314b474c780SDavid E. O'Brien 	    (ef->nrelatab != 0 && ef->relatab == NULL)) {
31570b7ffeeSIan Dowse 		error = ENOMEM;
31670b7ffeeSIan Dowse 		goto out;
31770b7ffeeSIan Dowse 	}
31870b7ffeeSIan Dowse 
31970b7ffeeSIan Dowse 	/* XXX, relocate the sh_addr fields saved by the loader. */
32070b7ffeeSIan Dowse 	off = 0;
32170b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
32270b7ffeeSIan Dowse 		if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off))
32370b7ffeeSIan Dowse 			off = shdr[i].sh_addr;
32470b7ffeeSIan Dowse 	}
32570b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
32670b7ffeeSIan Dowse 		if (shdr[i].sh_addr != 0)
32770b7ffeeSIan Dowse 			shdr[i].sh_addr = shdr[i].sh_addr - off +
32870b7ffeeSIan Dowse 			    (Elf_Addr)ef->address;
32970b7ffeeSIan Dowse 	}
33070b7ffeeSIan Dowse 
33170b7ffeeSIan Dowse 	ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
33270b7ffeeSIan Dowse 	ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr;
33370b7ffeeSIan Dowse 	ef->ddbstrcnt = shdr[symstrindex].sh_size;
33470b7ffeeSIan Dowse 	ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr;
33570b7ffeeSIan Dowse 	ef->shstrcnt = shdr[shstrindex].sh_size;
33670b7ffeeSIan Dowse 	ef->shstrtab = (char *)shdr[shstrindex].sh_addr;
33770b7ffeeSIan Dowse 
33870b7ffeeSIan Dowse 	/* Now fill out progtab and the relocation tables. */
33970b7ffeeSIan Dowse 	pb = 0;
34070b7ffeeSIan Dowse 	rl = 0;
34170b7ffeeSIan Dowse 	ra = 0;
34270b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
34370b7ffeeSIan Dowse 		switch (shdr[i].sh_type) {
34470b7ffeeSIan Dowse 		case SHT_PROGBITS:
34570b7ffeeSIan Dowse 		case SHT_NOBITS:
34613f28d96SKonstantin Belousov #ifdef __amd64__
347b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
34813f28d96SKonstantin Belousov #endif
34958c4aee0SJohn Baldwin 			if (shdr[i].sh_addr == 0)
35058c4aee0SJohn Baldwin 				break;
35170b7ffeeSIan Dowse 			ef->progtab[pb].addr = (void *)shdr[i].sh_addr;
35270b7ffeeSIan Dowse 			if (shdr[i].sh_type == SHT_PROGBITS)
35370b7ffeeSIan Dowse 				ef->progtab[pb].name = "<<PROGBITS>>";
35413f28d96SKonstantin Belousov #ifdef __amd64__
355b715d9afSKonstantin Belousov 			else if (shdr[i].sh_type == SHT_X86_64_UNWIND)
35613f28d96SKonstantin Belousov 				ef->progtab[pb].name = "<<UNWIND>>";
35713f28d96SKonstantin Belousov #endif
35870b7ffeeSIan Dowse 			else
35970b7ffeeSIan Dowse 				ef->progtab[pb].name = "<<NOBITS>>";
36070b7ffeeSIan Dowse 			ef->progtab[pb].size = shdr[i].sh_size;
36170b7ffeeSIan Dowse 			ef->progtab[pb].sec = i;
36270b7ffeeSIan Dowse 			if (ef->shstrtab && shdr[i].sh_name != 0)
36370b7ffeeSIan Dowse 				ef->progtab[pb].name =
36470b7ffeeSIan Dowse 				    ef->shstrtab + shdr[i].sh_name;
36550c202c5SJeff Roberson 			if (ef->progtab[pb].name != NULL &&
3665f67450dSDimitry Andric 			    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) {
36750c202c5SJeff Roberson 				void *dpcpu;
36850c202c5SJeff Roberson 
36950c202c5SJeff Roberson 				dpcpu = dpcpu_alloc(shdr[i].sh_size);
37050c202c5SJeff Roberson 				if (dpcpu == NULL) {
3710f823b64SBjoern A. Zeeb 					printf("%s: pcpu module space is out "
372*9afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
373*9afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
374*9afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
3750f823b64SBjoern A. Zeeb 					    filename);
37650c202c5SJeff Roberson 					error = ENOSPC;
37750c202c5SJeff Roberson 					goto out;
37850c202c5SJeff Roberson 				}
37950c202c5SJeff Roberson 				memcpy(dpcpu, ef->progtab[pb].addr,
38050c202c5SJeff Roberson 				    ef->progtab[pb].size);
38150c202c5SJeff Roberson 				dpcpu_copy(dpcpu, shdr[i].sh_size);
38250c202c5SJeff Roberson 				ef->progtab[pb].addr = dpcpu;
383eddfbb76SRobert Watson #ifdef VIMAGE
384eddfbb76SRobert Watson 			} else if (ef->progtab[pb].name != NULL &&
38517ef1febSRobert Watson 			    !strcmp(ef->progtab[pb].name, VNET_SETNAME)) {
386eddfbb76SRobert Watson 				void *vnet_data;
387eddfbb76SRobert Watson 
388eddfbb76SRobert Watson 				vnet_data = vnet_data_alloc(shdr[i].sh_size);
389eddfbb76SRobert Watson 				if (vnet_data == NULL) {
3900f823b64SBjoern A. Zeeb 					printf("%s: vnet module space is out "
391*9afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
392*9afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
393*9afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
3940f823b64SBjoern A. Zeeb 					    filename);
395eddfbb76SRobert Watson 					error = ENOSPC;
396eddfbb76SRobert Watson 					goto out;
397eddfbb76SRobert Watson 				}
398eddfbb76SRobert Watson 				memcpy(vnet_data, ef->progtab[pb].addr,
399eddfbb76SRobert Watson 				    ef->progtab[pb].size);
400eddfbb76SRobert Watson 				vnet_data_copy(vnet_data, shdr[i].sh_size);
401eddfbb76SRobert Watson 				ef->progtab[pb].addr = vnet_data;
402eddfbb76SRobert Watson #endif
4030067051fSMarcel Moolenaar 			} else if (ef->progtab[pb].name != NULL &&
4040067051fSMarcel Moolenaar 			    !strcmp(ef->progtab[pb].name, ".ctors")) {
4050067051fSMarcel Moolenaar 				lf->ctors_addr = ef->progtab[pb].addr;
4060067051fSMarcel Moolenaar 				lf->ctors_size = shdr[i].sh_size;
40750c202c5SJeff Roberson 			}
40870b7ffeeSIan Dowse 
40970b7ffeeSIan Dowse 			/* Update all symbol values with the offset. */
41070b7ffeeSIan Dowse 			for (j = 0; j < ef->ddbsymcnt; j++) {
41170b7ffeeSIan Dowse 				es = &ef->ddbsymtab[j];
41270b7ffeeSIan Dowse 				if (es->st_shndx != i)
41370b7ffeeSIan Dowse 					continue;
41470b7ffeeSIan Dowse 				es->st_value += (Elf_Addr)ef->progtab[pb].addr;
41570b7ffeeSIan Dowse 			}
41670b7ffeeSIan Dowse 			pb++;
41770b7ffeeSIan Dowse 			break;
41870b7ffeeSIan Dowse 		case SHT_REL:
41915746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
42015746ef4SJohn Baldwin 				break;
42170b7ffeeSIan Dowse 			ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr;
42270b7ffeeSIan Dowse 			ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel);
42370b7ffeeSIan Dowse 			ef->reltab[rl].sec = shdr[i].sh_info;
42470b7ffeeSIan Dowse 			rl++;
42570b7ffeeSIan Dowse 			break;
42670b7ffeeSIan Dowse 		case SHT_RELA:
42715746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
42815746ef4SJohn Baldwin 				break;
42970b7ffeeSIan Dowse 			ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr;
43070b7ffeeSIan Dowse 			ef->relatab[ra].nrela =
43170b7ffeeSIan Dowse 			    shdr[i].sh_size / sizeof(Elf_Rela);
43270b7ffeeSIan Dowse 			ef->relatab[ra].sec = shdr[i].sh_info;
43370b7ffeeSIan Dowse 			ra++;
43470b7ffeeSIan Dowse 			break;
43570b7ffeeSIan Dowse 		}
43670b7ffeeSIan Dowse 	}
4373cfce8e4SKonstantin Belousov 	if (pb != ef->nprogtab) {
4383cfce8e4SKonstantin Belousov 		printf("%s: lost progbits\n", filename);
4393cfce8e4SKonstantin Belousov 		error = ENOEXEC;
4403cfce8e4SKonstantin Belousov 		goto out;
4413cfce8e4SKonstantin Belousov 	}
4423cfce8e4SKonstantin Belousov 	if (rl != ef->nreltab) {
4433cfce8e4SKonstantin Belousov 		printf("%s: lost reltab\n", filename);
4443cfce8e4SKonstantin Belousov 		error = ENOEXEC;
4453cfce8e4SKonstantin Belousov 		goto out;
4463cfce8e4SKonstantin Belousov 	}
4473cfce8e4SKonstantin Belousov 	if (ra != ef->nrelatab) {
4483cfce8e4SKonstantin Belousov 		printf("%s: lost relatab\n", filename);
4493cfce8e4SKonstantin Belousov 		error = ENOEXEC;
4503cfce8e4SKonstantin Belousov 		goto out;
4513cfce8e4SKonstantin Belousov 	}
45270b7ffeeSIan Dowse 
45370b7ffeeSIan Dowse 	/* Local intra-module relocations */
454289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, false);
4553cfce8e4SKonstantin Belousov 	if (error != 0)
4563cfce8e4SKonstantin Belousov 		goto out;
45770b7ffeeSIan Dowse 	*result = lf;
45870b7ffeeSIan Dowse 	return (0);
45970b7ffeeSIan Dowse 
46070b7ffeeSIan Dowse out:
46170b7ffeeSIan Dowse 	/* preload not done this way */
46270b7ffeeSIan Dowse 	linker_file_unload(lf, LINKER_UNLOAD_FORCE);
46370b7ffeeSIan Dowse 	return (error);
46470b7ffeeSIan Dowse }
465de78ca7eSPeter Wemm 
4660067051fSMarcel Moolenaar static void
4670067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size)
4680067051fSMarcel Moolenaar {
4690067051fSMarcel Moolenaar 	void (**ctor)(void);
4700067051fSMarcel Moolenaar 	size_t i, cnt;
4710067051fSMarcel Moolenaar 
4720067051fSMarcel Moolenaar 	if (addr == NULL || size == 0)
4730067051fSMarcel Moolenaar 		return;
4740067051fSMarcel Moolenaar 	cnt = size / sizeof(*ctor);
4750067051fSMarcel Moolenaar 	ctor = (void *)addr;
4760067051fSMarcel Moolenaar 	for (i = 0; i < cnt; i++) {
4770067051fSMarcel Moolenaar 		if (ctor[i] != NULL)
4780067051fSMarcel Moolenaar 			(*ctor[i])();
4790067051fSMarcel Moolenaar 	}
4800067051fSMarcel Moolenaar }
4810067051fSMarcel Moolenaar 
482de78ca7eSPeter Wemm static int
48354823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf)
48454823af2SPeter Wemm {
48570b7ffeeSIan Dowse 	elf_file_t ef;
48670b7ffeeSIan Dowse 	int error;
48770b7ffeeSIan Dowse 
48870b7ffeeSIan Dowse 	ef = (elf_file_t)lf;
48970b7ffeeSIan Dowse 	error = relocate_file(ef);
49070b7ffeeSIan Dowse 	if (error)
491289ead7cSKonstantin Belousov 		return (error);
49270b7ffeeSIan Dowse 
49370b7ffeeSIan Dowse 	/* Notify MD code that a module is being loaded. */
49470b7ffeeSIan Dowse 	error = elf_cpu_load_file(lf);
49570b7ffeeSIan Dowse 	if (error)
49670b7ffeeSIan Dowse 		return (error);
49770b7ffeeSIan Dowse 
4988f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__)
499289ead7cSKonstantin Belousov 	/* Now ifuncs. */
500289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, true);
501289ead7cSKonstantin Belousov 	if (error != 0)
502289ead7cSKonstantin Belousov 		return (error);
5038f941950SKonstantin Belousov #endif
504289ead7cSKonstantin Belousov 
5050067051fSMarcel Moolenaar 	/* Invoke .ctors */
5060067051fSMarcel Moolenaar 	link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size);
50770b7ffeeSIan Dowse 	return (0);
50854823af2SPeter Wemm }
50954823af2SPeter Wemm 
51054823af2SPeter Wemm static int
511c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename,
512c143d6c2SMarcel Moolenaar     linker_file_t *result)
513a4f67738SDoug Rabson {
514a344babfSGleb Smirnoff 	struct nameidata *nd;
515b40ce416SJulian Elischer 	struct thread *td = curthread;	/* XXX */
516caab6e90SPeter Wemm 	Elf_Ehdr *hdr;
517e9eabf59SPeter Wemm 	Elf_Shdr *shdr;
518a8774e39SPeter Wemm 	Elf_Sym *es;
519a8774e39SPeter Wemm 	int nbytes, i, j;
52020947801SPeter Wemm 	vm_offset_t mapbase;
521fe3db7c7SDoug Rabson 	size_t mapsize;
522a4f67738SDoug Rabson 	int error = 0;
523526d0bd5SKonstantin Belousov 	ssize_t resid;
524526d0bd5SKonstantin Belousov 	int flags;
525a4f67738SDoug Rabson 	elf_file_t ef;
526a4f67738SDoug Rabson 	linker_file_t lf;
527ca65d5c7SPeter Wemm 	int symtabindex;
528ca65d5c7SPeter Wemm 	int symstrindex;
529e9eabf59SPeter Wemm 	int shstrindex;
530e9eabf59SPeter Wemm 	int nsym;
53120947801SPeter Wemm 	int pb, rl, ra;
532e9eabf59SPeter Wemm 	int alignmask;
533ca65d5c7SPeter Wemm 
534ca65d5c7SPeter Wemm 	shdr = NULL;
535ca65d5c7SPeter Wemm 	lf = NULL;
536e9eabf59SPeter Wemm 	mapsize = 0;
537e9eabf59SPeter Wemm 	hdr = NULL;
538a4f67738SDoug Rabson 
539a344babfSGleb Smirnoff 	nd = malloc(sizeof(struct nameidata), M_TEMP, M_WAITOK);
540a344babfSGleb Smirnoff 	NDINIT(nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td);
541e6796b67SKirk McKusick 	flags = FREAD;
542a344babfSGleb Smirnoff 	error = vn_open(nd, &flags, 0, NULL);
543a344babfSGleb Smirnoff 	if (error) {
544a344babfSGleb Smirnoff 		free(nd, M_TEMP);
545a4f67738SDoug Rabson 		return error;
546a344babfSGleb Smirnoff 	}
547a344babfSGleb Smirnoff 	NDFREE(nd, NDF_ONLY_PNBUF);
548a344babfSGleb Smirnoff 	if (nd->ni_vp->v_type != VREG) {
5491f49b573SRuslan Ermilov 		error = ENOEXEC;
5501f49b573SRuslan Ermilov 		goto out;
5511f49b573SRuslan Ermilov 	}
552a3df768bSRobert Watson #ifdef MAC
553a344babfSGleb Smirnoff 	error = mac_kld_check_load(td->td_ucred, nd->ni_vp);
554a3df768bSRobert Watson 	if (error) {
555a3df768bSRobert Watson 		goto out;
556a3df768bSRobert Watson 	}
557a3df768bSRobert Watson #endif
558a4f67738SDoug Rabson 
559e9eabf59SPeter Wemm 	/* Read the elf header from the file. */
560e9eabf59SPeter Wemm 	hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK);
561a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)hdr, sizeof(*hdr), 0,
5629ca43589SRobert Watson 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
5639ca43589SRobert Watson 	    &resid, td);
564a4f67738SDoug Rabson 	if (error)
565a4f67738SDoug Rabson 		goto out;
566e9eabf59SPeter Wemm 	if (resid != 0){
567e9eabf59SPeter Wemm 		error = ENOEXEC;
568e9eabf59SPeter Wemm 		goto out;
569e9eabf59SPeter Wemm 	}
570a4f67738SDoug Rabson 
571caab6e90SPeter Wemm 	if (!IS_ELF(*hdr)) {
572fe3db7c7SDoug Rabson 		error = ENOEXEC;
573a4f67738SDoug Rabson 		goto out;
574fe3db7c7SDoug Rabson 	}
575fe3db7c7SDoug Rabson 
576caab6e90SPeter Wemm 	if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS
577caab6e90SPeter Wemm 	    || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) {
578552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported file layout");
579fe3db7c7SDoug Rabson 		error = ENOEXEC;
580fe3db7c7SDoug Rabson 		goto out;
581fe3db7c7SDoug Rabson 	}
582caab6e90SPeter Wemm 	if (hdr->e_ident[EI_VERSION] != EV_CURRENT
583caab6e90SPeter Wemm 	    || hdr->e_version != EV_CURRENT) {
584552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported file version");
585fe3db7c7SDoug Rabson 		error = ENOEXEC;
586fe3db7c7SDoug Rabson 		goto out;
587fe3db7c7SDoug Rabson 	}
588e9eabf59SPeter Wemm 	if (hdr->e_type != ET_REL) {
589a1d7ce03SAttilio Rao 		error = ENOSYS;
590fe3db7c7SDoug Rabson 		goto out;
591fe3db7c7SDoug Rabson 	}
592caab6e90SPeter Wemm 	if (hdr->e_machine != ELF_TARG_MACH) {
593552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported machine");
594fe3db7c7SDoug Rabson 		error = ENOEXEC;
595fe3db7c7SDoug Rabson 		goto out;
596fe3db7c7SDoug Rabson 	}
597a4f67738SDoug Rabson 
598326e27d8SDoug Rabson 	lf = linker_make_file(filename, &link_elf_class);
599326e27d8SDoug Rabson 	if (!lf) {
600326e27d8SDoug Rabson 		error = ENOMEM;
601326e27d8SDoug Rabson 		goto out;
602326e27d8SDoug Rabson 	}
603326e27d8SDoug Rabson 	ef = (elf_file_t) lf;
604e9eabf59SPeter Wemm 	ef->nprogtab = 0;
605e9eabf59SPeter Wemm 	ef->e_shdr = 0;
606b474c780SDavid E. O'Brien 	ef->nreltab = 0;
607b474c780SDavid E. O'Brien 	ef->nrelatab = 0;
608e9eabf59SPeter Wemm 
609e9eabf59SPeter Wemm 	/* Allocate and read in the section header */
610e9eabf59SPeter Wemm 	nbytes = hdr->e_shnum * hdr->e_shentsize;
611e9eabf59SPeter Wemm 	if (nbytes == 0 || hdr->e_shoff == 0 ||
612e9eabf59SPeter Wemm 	    hdr->e_shentsize != sizeof(Elf_Shdr)) {
613e9eabf59SPeter Wemm 		error = ENOEXEC;
614e9eabf59SPeter Wemm 		goto out;
615e9eabf59SPeter Wemm 	}
61620947801SPeter Wemm 	shdr = malloc(nbytes, M_LINKER, M_WAITOK);
617e9eabf59SPeter Wemm 	ef->e_shdr = shdr;
618a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, (caddr_t)shdr, nbytes,
619a344babfSGleb Smirnoff 	    hdr->e_shoff, UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred,
620a344babfSGleb Smirnoff 	    NOCRED, &resid, td);
621e9eabf59SPeter Wemm 	if (error)
622e9eabf59SPeter Wemm 		goto out;
623e9eabf59SPeter Wemm 	if (resid) {
624e9eabf59SPeter Wemm 		error = ENOEXEC;
625e9eabf59SPeter Wemm 		goto out;
626e9eabf59SPeter Wemm 	}
627e9eabf59SPeter Wemm 
628e9eabf59SPeter Wemm 	/* Scan the section header for information and table sizing. */
629e9eabf59SPeter Wemm 	nsym = 0;
630e9eabf59SPeter Wemm 	symtabindex = -1;
631e9eabf59SPeter Wemm 	symstrindex = -1;
632e9eabf59SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
633676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
634676799a0SAndriy Gapon 			continue;
635e9eabf59SPeter Wemm 		switch (shdr[i].sh_type) {
636e9eabf59SPeter Wemm 		case SHT_PROGBITS:
637e9eabf59SPeter Wemm 		case SHT_NOBITS:
63813f28d96SKonstantin Belousov #ifdef __amd64__
639b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
64013f28d96SKonstantin Belousov #endif
64158c4aee0SJohn Baldwin 			if ((shdr[i].sh_flags & SHF_ALLOC) == 0)
64258c4aee0SJohn Baldwin 				break;
64320947801SPeter Wemm 			ef->nprogtab++;
644e9eabf59SPeter Wemm 			break;
645e9eabf59SPeter Wemm 		case SHT_SYMTAB:
646e9eabf59SPeter Wemm 			nsym++;
647e9eabf59SPeter Wemm 			symtabindex = i;
648e9eabf59SPeter Wemm 			symstrindex = shdr[i].sh_link;
649e9eabf59SPeter Wemm 			break;
650e9eabf59SPeter Wemm 		case SHT_REL:
65115746ef4SJohn Baldwin 			/*
65215746ef4SJohn Baldwin 			 * Ignore relocation tables for unallocated
65315746ef4SJohn Baldwin 			 * sections.
65415746ef4SJohn Baldwin 			 */
65515746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
65615746ef4SJohn Baldwin 				break;
657b474c780SDavid E. O'Brien 			ef->nreltab++;
658e9eabf59SPeter Wemm 			break;
659e9eabf59SPeter Wemm 		case SHT_RELA:
66015746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
66115746ef4SJohn Baldwin 				break;
662b474c780SDavid E. O'Brien 			ef->nrelatab++;
663e9eabf59SPeter Wemm 			break;
664e9eabf59SPeter Wemm 		case SHT_STRTAB:
665e9eabf59SPeter Wemm 			break;
666e9eabf59SPeter Wemm 		}
667e9eabf59SPeter Wemm 	}
66820947801SPeter Wemm 	if (ef->nprogtab == 0) {
669552f9f63SEdwin Groothuis 		link_elf_error(filename, "file has no contents");
670e9eabf59SPeter Wemm 		error = ENOEXEC;
671e9eabf59SPeter Wemm 		goto out;
672e9eabf59SPeter Wemm 	}
673e9eabf59SPeter Wemm 	if (nsym != 1) {
674e9eabf59SPeter Wemm 		/* Only allow one symbol table for now */
675f912a970SEd Maste 		link_elf_error(filename,
676f912a970SEd Maste 		    "file must have exactly one symbol table");
677e9eabf59SPeter Wemm 		error = ENOEXEC;
678e9eabf59SPeter Wemm 		goto out;
679e9eabf59SPeter Wemm 	}
680e9eabf59SPeter Wemm 	if (symstrindex < 0 || symstrindex > hdr->e_shnum ||
681e9eabf59SPeter Wemm 	    shdr[symstrindex].sh_type != SHT_STRTAB) {
682552f9f63SEdwin Groothuis 		link_elf_error(filename, "file has invalid symbol strings");
683e9eabf59SPeter Wemm 		error = ENOEXEC;
684e9eabf59SPeter Wemm 		goto out;
685e9eabf59SPeter Wemm 	}
686e9eabf59SPeter Wemm 
687e9eabf59SPeter Wemm 	/* Allocate space for tracking the load chunks */
688e9eabf59SPeter Wemm 	if (ef->nprogtab != 0)
68920947801SPeter Wemm 		ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab),
69020947801SPeter Wemm 		    M_LINKER, M_WAITOK | M_ZERO);
691b474c780SDavid E. O'Brien 	if (ef->nreltab != 0)
692b474c780SDavid E. O'Brien 		ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab),
693b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
694b474c780SDavid E. O'Brien 	if (ef->nrelatab != 0)
695b474c780SDavid E. O'Brien 		ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab),
696b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
697e9eabf59SPeter Wemm 
6983cfce8e4SKonstantin Belousov 	if (symtabindex == -1) {
6993cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost symbol table index");
7003cfce8e4SKonstantin Belousov 		error = ENOEXEC;
7013cfce8e4SKonstantin Belousov 		goto out;
7023cfce8e4SKonstantin Belousov 	}
70320947801SPeter Wemm 	/* Allocate space for and load the symbol table */
704e9eabf59SPeter Wemm 	ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
705e9eabf59SPeter Wemm 	ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK);
706a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)ef->ddbsymtab,
70720947801SPeter Wemm 	    shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset,
70820947801SPeter Wemm 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
70920947801SPeter Wemm 	    &resid, td);
71020947801SPeter Wemm 	if (error)
71120947801SPeter Wemm 		goto out;
71220947801SPeter Wemm 	if (resid != 0){
71320947801SPeter Wemm 		error = EINVAL;
71420947801SPeter Wemm 		goto out;
71520947801SPeter Wemm 	}
716e9eabf59SPeter Wemm 
7173cfce8e4SKonstantin Belousov 	if (symstrindex == -1) {
7183cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost symbol string index");
7193cfce8e4SKonstantin Belousov 		error = ENOEXEC;
7203cfce8e4SKonstantin Belousov 		goto out;
7213cfce8e4SKonstantin Belousov 	}
72220947801SPeter Wemm 	/* Allocate space for and load the symbol strings */
723e9eabf59SPeter Wemm 	ef->ddbstrcnt = shdr[symstrindex].sh_size;
724e9eabf59SPeter Wemm 	ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK);
725a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab,
72620947801SPeter Wemm 	    shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset,
72720947801SPeter Wemm 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
72820947801SPeter Wemm 	    &resid, td);
72920947801SPeter Wemm 	if (error)
73020947801SPeter Wemm 		goto out;
73120947801SPeter Wemm 	if (resid != 0){
73220947801SPeter Wemm 		error = EINVAL;
73320947801SPeter Wemm 		goto out;
73420947801SPeter Wemm 	}
735e9eabf59SPeter Wemm 
736e9eabf59SPeter Wemm 	/* Do we have a string table for the section names?  */
737e9eabf59SPeter Wemm 	shstrindex = -1;
73820947801SPeter Wemm 	if (hdr->e_shstrndx != 0 &&
73920947801SPeter Wemm 	    shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) {
740e9eabf59SPeter Wemm 		shstrindex = hdr->e_shstrndx;
741e9eabf59SPeter Wemm 		ef->shstrcnt = shdr[shstrindex].sh_size;
74220947801SPeter Wemm 		ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER,
74320947801SPeter Wemm 		    M_WAITOK);
744a344babfSGleb Smirnoff 		error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab,
74520947801SPeter Wemm 		    shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset,
746e9eabf59SPeter Wemm 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
747e9eabf59SPeter Wemm 		    &resid, td);
748e9eabf59SPeter Wemm 		if (error)
749e9eabf59SPeter Wemm 			goto out;
75020947801SPeter Wemm 		if (resid != 0){
75120947801SPeter Wemm 			error = EINVAL;
75220947801SPeter Wemm 			goto out;
75320947801SPeter Wemm 		}
754e9eabf59SPeter Wemm 	}
755e9eabf59SPeter Wemm 
75620947801SPeter Wemm 	/* Size up code/data(progbits) and bss(nobits). */
757e9eabf59SPeter Wemm 	alignmask = 0;
758e9eabf59SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
759676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
760676799a0SAndriy Gapon 			continue;
761e9eabf59SPeter Wemm 		switch (shdr[i].sh_type) {
762e9eabf59SPeter Wemm 		case SHT_PROGBITS:
763e9eabf59SPeter Wemm 		case SHT_NOBITS:
76413f28d96SKonstantin Belousov #ifdef __amd64__
765b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
76613f28d96SKonstantin Belousov #endif
76758c4aee0SJohn Baldwin 			if ((shdr[i].sh_flags & SHF_ALLOC) == 0)
76858c4aee0SJohn Baldwin 				break;
769e9eabf59SPeter Wemm 			alignmask = shdr[i].sh_addralign - 1;
770e9eabf59SPeter Wemm 			mapsize += alignmask;
771e9eabf59SPeter Wemm 			mapsize &= ~alignmask;
772e9eabf59SPeter Wemm 			mapsize += shdr[i].sh_size;
773e9eabf59SPeter Wemm 			break;
774e9eabf59SPeter Wemm 		}
775e9eabf59SPeter Wemm 	}
776e9eabf59SPeter Wemm 
777e9eabf59SPeter Wemm 	/*
778e9eabf59SPeter Wemm 	 * We know how much space we need for the text/data/bss/etc.
779e9eabf59SPeter Wemm 	 * This stuff needs to be in a single chunk so that profiling etc
780e9eabf59SPeter Wemm 	 * can get the bounds and gdb can associate offsets with modules
781e9eabf59SPeter Wemm 	 */
78220947801SPeter Wemm 	ef->object = vm_object_allocate(OBJT_DEFAULT,
78320947801SPeter Wemm 	    round_page(mapsize) >> PAGE_SHIFT);
784fe3db7c7SDoug Rabson 	if (ef->object == NULL) {
785fe3db7c7SDoug Rabson 		error = ENOMEM;
786fe3db7c7SDoug Rabson 		goto out;
787fe3db7c7SDoug Rabson 	}
788fe3db7c7SDoug Rabson 	ef->address = (caddr_t) vm_map_min(kernel_map);
789ac68d1c9SAlan Cox 
790ac68d1c9SAlan Cox 	/*
791ac68d1c9SAlan Cox 	 * In order to satisfy amd64's architectural requirements on the
792ac68d1c9SAlan Cox 	 * location of code and data in the kernel's address space, request a
793ac68d1c9SAlan Cox 	 * mapping that is above the kernel.
794ac68d1c9SAlan Cox 	 */
7954104e835SOleksandr Tymoshenko #ifdef __amd64__
796ac68d1c9SAlan Cox 	mapbase = KERNBASE;
7974104e835SOleksandr Tymoshenko #else
7984104e835SOleksandr Tymoshenko 	mapbase = VM_MIN_KERNEL_ADDRESS;
7994104e835SOleksandr Tymoshenko #endif
80020947801SPeter Wemm 	error = vm_map_find(kernel_map, ef->object, 0, &mapbase,
801edb572a3SJohn Baldwin 	    round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL,
802edb572a3SJohn Baldwin 	    VM_PROT_ALL, 0);
803fe3db7c7SDoug Rabson 	if (error) {
804fe3db7c7SDoug Rabson 		vm_object_deallocate(ef->object);
805326e27d8SDoug Rabson 		ef->object = 0;
806fe3db7c7SDoug Rabson 		goto out;
807fe3db7c7SDoug Rabson 	}
80820947801SPeter Wemm 
809e9eabf59SPeter Wemm 	/* Wire the pages */
8107f1ef325SAlan Cox 	error = vm_map_wire(kernel_map, mapbase,
81120947801SPeter Wemm 	    mapbase + round_page(mapsize),
812e9eabf59SPeter Wemm 	    VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES);
8137f1ef325SAlan Cox 	if (error != KERN_SUCCESS) {
8147f1ef325SAlan Cox 		error = ENOMEM;
8157f1ef325SAlan Cox 		goto out;
8167f1ef325SAlan Cox 	}
817a4f67738SDoug Rabson 
81820947801SPeter Wemm 	/* Inform the kld system about the situation */
81920947801SPeter Wemm 	lf->address = ef->address = (caddr_t)mapbase;
82020947801SPeter Wemm 	lf->size = mapsize;
821e9eabf59SPeter Wemm 
822e9eabf59SPeter Wemm 	/*
82320947801SPeter Wemm 	 * Now load code/data(progbits), zero bss(nobits), allocate space for
82420947801SPeter Wemm 	 * and load relocs
825e9eabf59SPeter Wemm 	 */
82620947801SPeter Wemm 	pb = 0;
82720947801SPeter Wemm 	rl = 0;
82820947801SPeter Wemm 	ra = 0;
82920947801SPeter Wemm 	alignmask = 0;
83020947801SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
831676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
832676799a0SAndriy Gapon 			continue;
83320947801SPeter Wemm 		switch (shdr[i].sh_type) {
83420947801SPeter Wemm 		case SHT_PROGBITS:
83520947801SPeter Wemm 		case SHT_NOBITS:
83613f28d96SKonstantin Belousov #ifdef __amd64__
837b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
83813f28d96SKonstantin Belousov #endif
83958c4aee0SJohn Baldwin 			if ((shdr[i].sh_flags & SHF_ALLOC) == 0)
84058c4aee0SJohn Baldwin 				break;
84120947801SPeter Wemm 			alignmask = shdr[i].sh_addralign - 1;
84220947801SPeter Wemm 			mapbase += alignmask;
84320947801SPeter Wemm 			mapbase &= ~alignmask;
8440067051fSMarcel Moolenaar 			if (ef->shstrtab != NULL && shdr[i].sh_name != 0) {
84550c202c5SJeff Roberson 				ef->progtab[pb].name =
84650c202c5SJeff Roberson 				    ef->shstrtab + shdr[i].sh_name;
8470067051fSMarcel Moolenaar 				if (!strcmp(ef->progtab[pb].name, ".ctors")) {
8480067051fSMarcel Moolenaar 					lf->ctors_addr = (caddr_t)mapbase;
8490067051fSMarcel Moolenaar 					lf->ctors_size = shdr[i].sh_size;
8500067051fSMarcel Moolenaar 				}
8510067051fSMarcel Moolenaar 			} else if (shdr[i].sh_type == SHT_PROGBITS)
85220947801SPeter Wemm 				ef->progtab[pb].name = "<<PROGBITS>>";
85313f28d96SKonstantin Belousov #ifdef __amd64__
854b715d9afSKonstantin Belousov 			else if (shdr[i].sh_type == SHT_X86_64_UNWIND)
85513f28d96SKonstantin Belousov 				ef->progtab[pb].name = "<<UNWIND>>";
85613f28d96SKonstantin Belousov #endif
85750c202c5SJeff Roberson 			else
85850c202c5SJeff Roberson 				ef->progtab[pb].name = "<<NOBITS>>";
85950c202c5SJeff Roberson 			if (ef->progtab[pb].name != NULL &&
8600f823b64SBjoern A. Zeeb 			    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) {
86150c202c5SJeff Roberson 				ef->progtab[pb].addr =
86250c202c5SJeff Roberson 				    dpcpu_alloc(shdr[i].sh_size);
8630f823b64SBjoern A. Zeeb 				if (ef->progtab[pb].addr == NULL) {
8640f823b64SBjoern A. Zeeb 					printf("%s: pcpu module space is out "
865*9afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
866*9afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
867*9afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
8680f823b64SBjoern A. Zeeb 					    filename);
8690f823b64SBjoern A. Zeeb 				}
8700f823b64SBjoern A. Zeeb 			}
871eddfbb76SRobert Watson #ifdef VIMAGE
872eddfbb76SRobert Watson 			else if (ef->progtab[pb].name != NULL &&
8730f823b64SBjoern A. Zeeb 			    !strcmp(ef->progtab[pb].name, VNET_SETNAME)) {
874eddfbb76SRobert Watson 				ef->progtab[pb].addr =
875eddfbb76SRobert Watson 				    vnet_data_alloc(shdr[i].sh_size);
8760f823b64SBjoern A. Zeeb 				if (ef->progtab[pb].addr == NULL) {
8770f823b64SBjoern A. Zeeb 					printf("%s: vnet module space is out "
878*9afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
879*9afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
880*9afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
8810f823b64SBjoern A. Zeeb 					    filename);
8820f823b64SBjoern A. Zeeb 				}
8830f823b64SBjoern A. Zeeb 			}
884eddfbb76SRobert Watson #endif
88550c202c5SJeff Roberson 			else
88650c202c5SJeff Roberson 				ef->progtab[pb].addr =
88750c202c5SJeff Roberson 				    (void *)(uintptr_t)mapbase;
88850c202c5SJeff Roberson 			if (ef->progtab[pb].addr == NULL) {
88950c202c5SJeff Roberson 				error = ENOSPC;
89050c202c5SJeff Roberson 				goto out;
89150c202c5SJeff Roberson 			}
89250c202c5SJeff Roberson 			ef->progtab[pb].size = shdr[i].sh_size;
89350c202c5SJeff Roberson 			ef->progtab[pb].sec = i;
89413f28d96SKonstantin Belousov 			if (shdr[i].sh_type == SHT_PROGBITS
89513f28d96SKonstantin Belousov #ifdef __amd64__
896b715d9afSKonstantin Belousov 			    || shdr[i].sh_type == SHT_X86_64_UNWIND
89713f28d96SKonstantin Belousov #endif
89813f28d96SKonstantin Belousov 			    ) {
899a344babfSGleb Smirnoff 				error = vn_rdwr(UIO_READ, nd->ni_vp,
90020947801SPeter Wemm 				    ef->progtab[pb].addr,
90120947801SPeter Wemm 				    shdr[i].sh_size, shdr[i].sh_offset,
90220947801SPeter Wemm 				    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred,
90320947801SPeter Wemm 				    NOCRED, &resid, td);
90420947801SPeter Wemm 				if (error)
90520947801SPeter Wemm 					goto out;
90620947801SPeter Wemm 				if (resid != 0){
90720947801SPeter Wemm 					error = EINVAL;
90820947801SPeter Wemm 					goto out;
90920947801SPeter Wemm 				}
910eddfbb76SRobert Watson 				/* Initialize the per-cpu or vnet area. */
911eddfbb76SRobert Watson 				if (ef->progtab[pb].addr != (void *)mapbase &&
9125f67450dSDimitry Andric 				    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME))
91350c202c5SJeff Roberson 					dpcpu_copy(ef->progtab[pb].addr,
91450c202c5SJeff Roberson 					    shdr[i].sh_size);
915eddfbb76SRobert Watson #ifdef VIMAGE
916eddfbb76SRobert Watson 				else if (ef->progtab[pb].addr !=
917eddfbb76SRobert Watson 				    (void *)mapbase &&
91817ef1febSRobert Watson 				    !strcmp(ef->progtab[pb].name, VNET_SETNAME))
919eddfbb76SRobert Watson 					vnet_data_copy(ef->progtab[pb].addr,
920eddfbb76SRobert Watson 					    shdr[i].sh_size);
921eddfbb76SRobert Watson #endif
92250c202c5SJeff Roberson 			} else
92320947801SPeter Wemm 				bzero(ef->progtab[pb].addr, shdr[i].sh_size);
924a8774e39SPeter Wemm 
925a8774e39SPeter Wemm 			/* Update all symbol values with the offset. */
926a8774e39SPeter Wemm 			for (j = 0; j < ef->ddbsymcnt; j++) {
927a8774e39SPeter Wemm 				es = &ef->ddbsymtab[j];
928a8774e39SPeter Wemm 				if (es->st_shndx != i)
929a8774e39SPeter Wemm 					continue;
930a8774e39SPeter Wemm 				es->st_value += (Elf_Addr)ef->progtab[pb].addr;
931a8774e39SPeter Wemm 			}
93220947801SPeter Wemm 			mapbase += shdr[i].sh_size;
93320947801SPeter Wemm 			pb++;
93420947801SPeter Wemm 			break;
93520947801SPeter Wemm 		case SHT_REL:
93615746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
93715746ef4SJohn Baldwin 				break;
93820947801SPeter Wemm 			ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER,
93920947801SPeter Wemm 			    M_WAITOK);
94020947801SPeter Wemm 			ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel);
94120947801SPeter Wemm 			ef->reltab[rl].sec = shdr[i].sh_info;
942a344babfSGleb Smirnoff 			error = vn_rdwr(UIO_READ, nd->ni_vp,
94320947801SPeter Wemm 			    (void *)ef->reltab[rl].rel,
94420947801SPeter Wemm 			    shdr[i].sh_size, shdr[i].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;
952e9eabf59SPeter Wemm 			}
95320947801SPeter Wemm 			rl++;
95420947801SPeter Wemm 			break;
95520947801SPeter Wemm 		case SHT_RELA:
95615746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
95715746ef4SJohn Baldwin 				break;
95820947801SPeter Wemm 			ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER,
95920947801SPeter Wemm 			    M_WAITOK);
96020947801SPeter Wemm 			ef->relatab[ra].nrela =
96120947801SPeter Wemm 			    shdr[i].sh_size / sizeof(Elf_Rela);
96220947801SPeter Wemm 			ef->relatab[ra].sec = shdr[i].sh_info;
963a344babfSGleb Smirnoff 			error = vn_rdwr(UIO_READ, nd->ni_vp,
96420947801SPeter Wemm 			    (void *)ef->relatab[ra].rela,
96520947801SPeter Wemm 			    shdr[i].sh_size, shdr[i].sh_offset,
966e9eabf59SPeter Wemm 			    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
967e9eabf59SPeter Wemm 			    &resid, td);
968e9eabf59SPeter Wemm 			if (error)
969e9eabf59SPeter Wemm 				goto out;
97020947801SPeter Wemm 			if (resid != 0){
97120947801SPeter Wemm 				error = EINVAL;
97220947801SPeter Wemm 				goto out;
973e9eabf59SPeter Wemm 			}
97420947801SPeter Wemm 			ra++;
97520947801SPeter Wemm 			break;
97620947801SPeter Wemm 		}
97720947801SPeter Wemm 	}
9783cfce8e4SKonstantin Belousov 	if (pb != ef->nprogtab) {
9793cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost progbits");
9803cfce8e4SKonstantin Belousov 		error = ENOEXEC;
9813cfce8e4SKonstantin Belousov 		goto out;
9823cfce8e4SKonstantin Belousov 	}
9833cfce8e4SKonstantin Belousov 	if (rl != ef->nreltab) {
9843cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost reltab");
9853cfce8e4SKonstantin Belousov 		error = ENOEXEC;
9863cfce8e4SKonstantin Belousov 		goto out;
9873cfce8e4SKonstantin Belousov 	}
9883cfce8e4SKonstantin Belousov 	if (ra != ef->nrelatab) {
9893cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost relatab");
9903cfce8e4SKonstantin Belousov 		error = ENOEXEC;
9913cfce8e4SKonstantin Belousov 		goto out;
9923cfce8e4SKonstantin Belousov 	}
9933cfce8e4SKonstantin Belousov 	if (mapbase != (vm_offset_t)ef->address + mapsize) {
9943cfce8e4SKonstantin Belousov 		printf(
9953cfce8e4SKonstantin Belousov 		    "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n",
9963cfce8e4SKonstantin Belousov 		    filename != NULL ? filename : "<none>",
9976f3c6327SNeel Natu 		    (u_long)mapbase, ef->address, (u_long)mapsize,
9986f3c6327SNeel Natu 		    (u_long)(vm_offset_t)ef->address + mapsize);
9993cfce8e4SKonstantin Belousov 		error = ENOMEM;
10003cfce8e4SKonstantin Belousov 		goto out;
10013cfce8e4SKonstantin Belousov 	}
1002e9eabf59SPeter Wemm 
1003e9eabf59SPeter Wemm 	/* Local intra-module relocations */
1004289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, false);
10053cfce8e4SKonstantin Belousov 	if (error != 0)
10063cfce8e4SKonstantin Belousov 		goto out;
10077251b4bfSJake Burkholder 
1008e9eabf59SPeter Wemm 	/* Pull in dependencies */
1009a344babfSGleb Smirnoff 	VOP_UNLOCK(nd->ni_vp, 0);
10107b9716baSIan Dowse 	error = linker_load_dependencies(lf);
1011a344babfSGleb Smirnoff 	vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY);
1012ca65d5c7SPeter Wemm 	if (error)
1013de78ca7eSPeter Wemm 		goto out;
1014e9eabf59SPeter Wemm 
1015e9eabf59SPeter Wemm 	/* External relocations */
1016326e27d8SDoug Rabson 	error = relocate_file(ef);
1017ca65d5c7SPeter Wemm 	if (error)
1018ca65d5c7SPeter Wemm 		goto out;
1019ca65d5c7SPeter Wemm 
1020e9eabf59SPeter Wemm 	/* Notify MD code that a module is being loaded. */
1021e9eabf59SPeter Wemm 	error = elf_cpu_load_file(lf);
1022ca65d5c7SPeter Wemm 	if (error)
1023ca65d5c7SPeter Wemm 		goto out;
1024a4f67738SDoug Rabson 
10258f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__)
1026289ead7cSKonstantin Belousov 	/* Now ifuncs. */
1027289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, true);
1028289ead7cSKonstantin Belousov 	if (error != 0)
1029289ead7cSKonstantin Belousov 		goto out;
10308f941950SKonstantin Belousov #endif
1031289ead7cSKonstantin Belousov 
10320067051fSMarcel Moolenaar 	/* Invoke .ctors */
10330067051fSMarcel Moolenaar 	link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size);
10340067051fSMarcel Moolenaar 
1035a4f67738SDoug Rabson 	*result = lf;
1036a4f67738SDoug Rabson 
1037a4f67738SDoug Rabson out:
1038a344babfSGleb Smirnoff 	VOP_UNLOCK(nd->ni_vp, 0);
1039a344babfSGleb Smirnoff 	vn_close(nd->ni_vp, FREAD, td->td_ucred, td);
1040a344babfSGleb Smirnoff 	free(nd, M_TEMP);
1041ca65d5c7SPeter Wemm 	if (error && lf)
104265a311fcSPoul-Henning Kamp 		linker_file_unload(lf, LINKER_UNLOAD_FORCE);
1043e9eabf59SPeter Wemm 	free(hdr, M_LINKER);
1044a4f67738SDoug Rabson 
1045a4f67738SDoug Rabson 	return error;
1046a4f67738SDoug Rabson }
1047a4f67738SDoug Rabson 
1048a4f67738SDoug Rabson static void
1049de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file)
1050a4f67738SDoug Rabson {
1051326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) file;
1052d821d364SPedro F. Giffuni 	u_int i;
1053a4f67738SDoug Rabson 
10541aeb23cdSMarcel Moolenaar 	/* Notify MD code that a module is being unloaded. */
10551aeb23cdSMarcel Moolenaar 	elf_cpu_unload_file(file);
10561aeb23cdSMarcel Moolenaar 
105750c202c5SJeff Roberson 	if (ef->progtab) {
105850c202c5SJeff Roberson 		for (i = 0; i < ef->nprogtab; i++) {
105950c202c5SJeff Roberson 			if (ef->progtab[i].size == 0)
106050c202c5SJeff Roberson 				continue;
106150c202c5SJeff Roberson 			if (ef->progtab[i].name == NULL)
106250c202c5SJeff Roberson 				continue;
10635f67450dSDimitry Andric 			if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME))
106450c202c5SJeff Roberson 				dpcpu_free(ef->progtab[i].addr,
106550c202c5SJeff Roberson 				    ef->progtab[i].size);
1066eddfbb76SRobert Watson #ifdef VIMAGE
106717ef1febSRobert Watson 			else if (!strcmp(ef->progtab[i].name, VNET_SETNAME))
1068eddfbb76SRobert Watson 				vnet_data_free(ef->progtab[i].addr,
1069eddfbb76SRobert Watson 				    ef->progtab[i].size);
1070eddfbb76SRobert Watson #endif
107150c202c5SJeff Roberson 		}
107250c202c5SJeff Roberson 	}
107370b7ffeeSIan Dowse 	if (ef->preloaded) {
107470b7ffeeSIan Dowse 		free(ef->reltab, M_LINKER);
107570b7ffeeSIan Dowse 		free(ef->relatab, M_LINKER);
107670b7ffeeSIan Dowse 		free(ef->progtab, M_LINKER);
1077a2024a3eSJohn Birrell 		free(ef->ctftab, M_LINKER);
1078a2024a3eSJohn Birrell 		free(ef->ctfoff, M_LINKER);
1079a2024a3eSJohn Birrell 		free(ef->typoff, M_LINKER);
108073624a80SMark Johnston 		if (file->pathname != NULL)
108173624a80SMark Johnston 			preload_delete_name(file->pathname);
108270b7ffeeSIan Dowse 		return;
108370b7ffeeSIan Dowse 	}
108470b7ffeeSIan Dowse 
1085b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++)
108620947801SPeter Wemm 		free(ef->reltab[i].rel, M_LINKER);
1087b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++)
108820947801SPeter Wemm 		free(ef->relatab[i].rela, M_LINKER);
108920947801SPeter Wemm 	free(ef->reltab, M_LINKER);
109020947801SPeter Wemm 	free(ef->relatab, M_LINKER);
109120947801SPeter Wemm 	free(ef->progtab, M_LINKER);
109220947801SPeter Wemm 
1093fe3db7c7SDoug Rabson 	if (ef->object) {
1094fe3db7c7SDoug Rabson 		vm_map_remove(kernel_map, (vm_offset_t) ef->address,
109520947801SPeter Wemm 		    (vm_offset_t) ef->address +
109620947801SPeter Wemm 		    (ef->object->size << PAGE_SHIFT));
1097fe3db7c7SDoug Rabson 	}
1098e9eabf59SPeter Wemm 	free(ef->e_shdr, M_LINKER);
1099e9eabf59SPeter Wemm 	free(ef->ddbsymtab, M_LINKER);
1100e9eabf59SPeter Wemm 	free(ef->ddbstrtab, M_LINKER);
110120947801SPeter Wemm 	free(ef->shstrtab, M_LINKER);
1102a2024a3eSJohn Birrell 	free(ef->ctftab, M_LINKER);
1103a2024a3eSJohn Birrell 	free(ef->ctfoff, M_LINKER);
1104a2024a3eSJohn Birrell 	free(ef->typoff, M_LINKER);
1105de78ca7eSPeter Wemm }
1106de78ca7eSPeter Wemm 
1107fe3db7c7SDoug Rabson static const char *
1108757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info)
1109a4f67738SDoug Rabson {
1110fe3db7c7SDoug Rabson 	const Elf_Sym *ref;
1111a4f67738SDoug Rabson 
1112aa855a59SPeter Wemm 	if (ELF_R_SYM(r_info)) {
1113e9eabf59SPeter Wemm 		ref = ef->ddbsymtab + ELF_R_SYM(r_info);
1114e9eabf59SPeter Wemm 		return ef->ddbstrtab + ref->st_name;
1115fe3db7c7SDoug Rabson 	} else
1116fe3db7c7SDoug Rabson 		return NULL;
1117a4f67738SDoug Rabson }
1118a4f67738SDoug Rabson 
1119e9eabf59SPeter Wemm static Elf_Addr
1120e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec)
1121e9eabf59SPeter Wemm {
1122e9eabf59SPeter Wemm 	int i;
1123e9eabf59SPeter Wemm 	Elf_Addr base = 0;
1124e9eabf59SPeter Wemm 
1125e9eabf59SPeter Wemm 	for (i = 0; i < ef->nprogtab; i++) {
1126a8774e39SPeter Wemm 		if (sec == ef->progtab[i].sec) {
1127e9eabf59SPeter Wemm 			base = (Elf_Addr)ef->progtab[i].addr;
1128a8774e39SPeter Wemm 			break;
1129e9eabf59SPeter Wemm 		}
1130a8774e39SPeter Wemm 	}
1131e9eabf59SPeter Wemm 	return base;
1132e9eabf59SPeter Wemm }
1133e9eabf59SPeter Wemm 
1134a4f67738SDoug Rabson static int
1135326e27d8SDoug Rabson relocate_file(elf_file_t ef)
1136a4f67738SDoug Rabson {
1137fe3db7c7SDoug Rabson 	const Elf_Rel *rellim;
1138fe3db7c7SDoug Rabson 	const Elf_Rel *rel;
1139fe3db7c7SDoug Rabson 	const Elf_Rela *relalim;
1140fe3db7c7SDoug Rabson 	const Elf_Rela *rela;
1141aa855a59SPeter Wemm 	const char *symname;
1142e9eabf59SPeter Wemm 	const Elf_Sym *sym;
1143e9eabf59SPeter Wemm 	int i;
1144757686b1SMarcel Moolenaar 	Elf_Size symidx;
1145e9eabf59SPeter Wemm 	Elf_Addr base;
1146e9eabf59SPeter Wemm 
1147a4f67738SDoug Rabson 
1148fe3db7c7SDoug Rabson 	/* Perform relocations without addend if there are any: */
1149b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++) {
1150e9eabf59SPeter Wemm 		rel = ef->reltab[i].rel;
11513cfce8e4SKonstantin Belousov 		if (rel == NULL) {
11523cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a reltab!");
11533cfce8e4SKonstantin Belousov 			return (ENOEXEC);
11543cfce8e4SKonstantin Belousov 		}
115520947801SPeter Wemm 		rellim = rel + ef->reltab[i].nrel;
1156e9eabf59SPeter Wemm 		base = findbase(ef, ef->reltab[i].sec);
11573cfce8e4SKonstantin Belousov 		if (base == 0) {
11583cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost base for reltab");
11593cfce8e4SKonstantin Belousov 			return (ENOEXEC);
11603cfce8e4SKonstantin Belousov 		}
116120947801SPeter Wemm 		for ( ; rel < rellim; rel++) {
1162e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rel->r_info);
116320947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
116420947801SPeter Wemm 				continue;
1165e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
116620947801SPeter Wemm 			/* Local relocs are already done */
116720947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL)
116820947801SPeter Wemm 				continue;
116920947801SPeter Wemm 			if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL,
117020947801SPeter Wemm 			    elf_obj_lookup)) {
1171aa855a59SPeter Wemm 				symname = symbol_name(ef, rel->r_info);
117220947801SPeter Wemm 				printf("link_elf_obj: symbol %s undefined\n",
117320947801SPeter Wemm 				    symname);
11743cfce8e4SKonstantin Belousov 				return (ENOENT);
1175a13ddfb6SPeter Wemm 			}
1176aa855a59SPeter Wemm 		}
1177a4f67738SDoug Rabson 	}
1178a4f67738SDoug Rabson 
1179fe3db7c7SDoug Rabson 	/* Perform relocations with addend if there are any: */
1180b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++) {
1181e9eabf59SPeter Wemm 		rela = ef->relatab[i].rela;
11823cfce8e4SKonstantin Belousov 		if (rela == NULL) {
11833cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a relatab!");
11843cfce8e4SKonstantin Belousov 			return (ENOEXEC);
11853cfce8e4SKonstantin Belousov 		}
118620947801SPeter Wemm 		relalim = rela + ef->relatab[i].nrela;
1187e9eabf59SPeter Wemm 		base = findbase(ef, ef->relatab[i].sec);
11883cfce8e4SKonstantin Belousov 		if (base == 0) {
11893cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename,
11903cfce8e4SKonstantin Belousov 			    "lost base for relatab");
11913cfce8e4SKonstantin Belousov 			return (ENOEXEC);
11923cfce8e4SKonstantin Belousov 		}
119320947801SPeter Wemm 		for ( ; rela < relalim; rela++) {
1194e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rela->r_info);
119520947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
119620947801SPeter Wemm 				continue;
1197e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
119820947801SPeter Wemm 			/* Local relocs are already done */
119920947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL)
120020947801SPeter Wemm 				continue;
120120947801SPeter Wemm 			if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA,
120220947801SPeter Wemm 			    elf_obj_lookup)) {
1203aa855a59SPeter Wemm 				symname = symbol_name(ef, rela->r_info);
120420947801SPeter Wemm 				printf("link_elf_obj: symbol %s undefined\n",
120520947801SPeter Wemm 				    symname);
12063cfce8e4SKonstantin Belousov 				return (ENOENT);
1207a13ddfb6SPeter Wemm 			}
1208e9eabf59SPeter Wemm 		}
1209e9eabf59SPeter Wemm 	}
1210a4f67738SDoug Rabson 
12112832cd54SKonstantin Belousov 	/*
1212e3043798SPedro F. Giffuni 	 * Only clean SHN_FBSD_CACHED for successful return.  If we
12132832cd54SKonstantin Belousov 	 * modified symbol table for the object but found an
12142832cd54SKonstantin Belousov 	 * unresolved symbol, there is no reason to roll back.
12152832cd54SKonstantin Belousov 	 */
12162832cd54SKonstantin Belousov 	elf_obj_cleanup_globals_cache(ef);
12172832cd54SKonstantin Belousov 
12183cfce8e4SKonstantin Belousov 	return (0);
1219a4f67738SDoug Rabson }
1220a4f67738SDoug Rabson 
122137c84183SPoul-Henning Kamp static int
1222d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym)
1223a4f67738SDoug Rabson {
1224326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) lf;
12252d636ab0SPeter Wemm 	const Elf_Sym *symp;
12262d636ab0SPeter Wemm 	const char *strp;
1227a4f67738SDoug Rabson 	int i;
1228a4f67738SDoug Rabson 
12292d636ab0SPeter Wemm 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
12302d636ab0SPeter Wemm 		strp = ef->ddbstrtab + symp->st_name;
1231a8774e39SPeter Wemm 		if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) {
1232d254af07SMatthew Dillon 			*sym = (c_linker_sym_t) symp;
12332d636ab0SPeter Wemm 			return 0;
12342d636ab0SPeter Wemm 		}
12352d636ab0SPeter Wemm 	}
1236a4f67738SDoug Rabson 	return ENOENT;
1237a4f67738SDoug Rabson }
1238a4f67738SDoug Rabson 
1239de78ca7eSPeter Wemm static int
124020947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym,
124120947801SPeter Wemm     linker_symval_t *symval)
1242a4f67738SDoug Rabson {
12437035cf14SKonstantin Belousov 	elf_file_t ef;
12447035cf14SKonstantin Belousov 	const Elf_Sym *es;
12457035cf14SKonstantin Belousov 	caddr_t val;
1246a4f67738SDoug Rabson 
12477035cf14SKonstantin Belousov 	ef = (elf_file_t) lf;
12487035cf14SKonstantin Belousov 	es = (const Elf_Sym*) sym;
12497035cf14SKonstantin Belousov 	val = (caddr_t)es->st_value;
12505cf87418SMarcel Moolenaar 	if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) {
12512d636ab0SPeter Wemm 		symval->name = ef->ddbstrtab + es->st_name;
12527035cf14SKonstantin Belousov 		val = (caddr_t)es->st_value;
12537035cf14SKonstantin Belousov 		if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC)
12547035cf14SKonstantin Belousov 			val = ((caddr_t (*)(void))val)();
12557035cf14SKonstantin Belousov 		symval->value = val;
12562d636ab0SPeter Wemm 		symval->size = es->st_size;
12572d636ab0SPeter Wemm 		return 0;
12582d636ab0SPeter Wemm 	}
12592d636ab0SPeter Wemm 	return ENOENT;
12602d636ab0SPeter Wemm }
1261a4f67738SDoug Rabson 
1262a4f67738SDoug Rabson static int
1263a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value,
1264d254af07SMatthew Dillon     c_linker_sym_t *sym, long *diffp)
1265a4f67738SDoug Rabson {
1266326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) lf;
1267586453feSBruce Evans 	u_long off = (uintptr_t) (void *) value;
1268a4f67738SDoug Rabson 	u_long diff = off;
1269586453feSBruce Evans 	u_long st_value;
1270fe3db7c7SDoug Rabson 	const Elf_Sym *es;
1271b85f65afSPedro F. Giffuni 	const Elf_Sym *best = NULL;
1272a4f67738SDoug Rabson 	int i;
1273a4f67738SDoug Rabson 
12742d636ab0SPeter Wemm 	for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) {
1275a4f67738SDoug Rabson 		if (es->st_name == 0)
1276a4f67738SDoug Rabson 			continue;
1277a8774e39SPeter Wemm 		st_value = es->st_value;
1278b5abfb70SPeter Wemm 		if (off >= st_value) {
1279b5abfb70SPeter Wemm 			if (off - st_value < diff) {
1280b5abfb70SPeter Wemm 				diff = off - st_value;
1281a4f67738SDoug Rabson 				best = es;
1282a4f67738SDoug Rabson 				if (diff == 0)
1283a4f67738SDoug Rabson 					break;
1284b5abfb70SPeter Wemm 			} else if (off - st_value == diff) {
1285a4f67738SDoug Rabson 				best = es;
1286a4f67738SDoug Rabson 			}
1287a4f67738SDoug Rabson 		}
1288a4f67738SDoug Rabson 	}
1289b85f65afSPedro F. Giffuni 	if (best == NULL)
1290a4f67738SDoug Rabson 		*diffp = off;
1291a4f67738SDoug Rabson 	else
1292a4f67738SDoug Rabson 		*diffp = diff;
1293d254af07SMatthew Dillon 	*sym = (c_linker_sym_t) best;
1294a4f67738SDoug Rabson 
1295a4f67738SDoug Rabson 	return 0;
1296a4f67738SDoug Rabson }
1297f41325dbSPeter Wemm 
1298f41325dbSPeter Wemm /*
1299f41325dbSPeter Wemm  * Look up a linker set on an ELF system.
1300f41325dbSPeter Wemm  */
1301f41325dbSPeter Wemm static int
1302f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name,
1303f41325dbSPeter Wemm     void ***startp, void ***stopp, int *countp)
1304f41325dbSPeter Wemm {
1305e9eabf59SPeter Wemm 	elf_file_t ef = (elf_file_t)lf;
1306f41325dbSPeter Wemm 	void **start, **stop;
1307e9eabf59SPeter Wemm 	int i, count;
1308f41325dbSPeter Wemm 
1309e9eabf59SPeter Wemm 	/* Relative to section number */
1310e9eabf59SPeter Wemm 	for (i = 0; i < ef->nprogtab; i++) {
1311e9eabf59SPeter Wemm 		if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) &&
1312e9eabf59SPeter Wemm 		    strcmp(ef->progtab[i].name + 4, name) == 0) {
1313e9eabf59SPeter Wemm 			start  = (void **)ef->progtab[i].addr;
131420947801SPeter Wemm 			stop = (void **)((char *)ef->progtab[i].addr +
131520947801SPeter Wemm 			    ef->progtab[i].size);
1316f41325dbSPeter Wemm 			count = stop - start;
1317f41325dbSPeter Wemm 			if (startp)
1318f41325dbSPeter Wemm 				*startp = start;
1319f41325dbSPeter Wemm 			if (stopp)
1320f41325dbSPeter Wemm 				*stopp = stop;
1321f41325dbSPeter Wemm 			if (countp)
1322f41325dbSPeter Wemm 				*countp = count;
1323e9eabf59SPeter Wemm 			return (0);
1324e9eabf59SPeter Wemm 		}
1325e9eabf59SPeter Wemm 	}
1326e9eabf59SPeter Wemm 	return (ESRCH);
1327f41325dbSPeter Wemm }
1328bb9fe9ddSBrian Feldman 
1329bb9fe9ddSBrian Feldman static int
1330bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file,
1331e9eabf59SPeter Wemm     int (*callback)(const char *, void *), void *opaque)
1332e9eabf59SPeter Wemm {
1333bb9fe9ddSBrian Feldman 	elf_file_t ef = (elf_file_t)file;
1334bb9fe9ddSBrian Feldman 	const Elf_Sym *symp;
1335bb9fe9ddSBrian Feldman 	int i, error;
1336bb9fe9ddSBrian Feldman 
1337bb9fe9ddSBrian Feldman 	/* Exhaustive search */
1338bb9fe9ddSBrian Feldman 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1339bb9fe9ddSBrian Feldman 		if (symp->st_value != 0 &&
13407035cf14SKonstantin Belousov 		    (ELF_ST_TYPE(symp->st_info) == STT_FUNC ||
13417035cf14SKonstantin Belousov 		    ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) {
1342bb9fe9ddSBrian Feldman 			error = callback(ef->ddbstrtab + symp->st_name, opaque);
1343bb9fe9ddSBrian Feldman 			if (error)
1344bb9fe9ddSBrian Feldman 				return (error);
1345bb9fe9ddSBrian Feldman 		}
1346bb9fe9ddSBrian Feldman 	}
1347bb9fe9ddSBrian Feldman 	return (0);
1348bb9fe9ddSBrian Feldman }
134984201059SMarcel Moolenaar 
1350a2024a3eSJohn Birrell static int
1351a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file,
1352a2024a3eSJohn Birrell     linker_function_nameval_callback_t callback, void *opaque)
1353a2024a3eSJohn Birrell {
1354a2024a3eSJohn Birrell 	linker_symval_t symval;
1355a2024a3eSJohn Birrell 	elf_file_t ef = (elf_file_t)file;
1356a2024a3eSJohn Birrell 	const Elf_Sym* symp;
1357a2024a3eSJohn Birrell 	int i, error;
1358a2024a3eSJohn Birrell 
1359a2024a3eSJohn Birrell 	/* Exhaustive search */
1360a2024a3eSJohn Birrell 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1361a2024a3eSJohn Birrell 		if (symp->st_value != 0 &&
13627035cf14SKonstantin Belousov 		    (ELF_ST_TYPE(symp->st_info) == STT_FUNC ||
13637035cf14SKonstantin Belousov 		    ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) {
13647035cf14SKonstantin Belousov 			error = link_elf_symbol_values(file,
13657035cf14SKonstantin Belousov 			    (c_linker_sym_t)symp, &symval);
1366a2024a3eSJohn Birrell 			if (error)
1367a2024a3eSJohn Birrell 				return (error);
1368a2024a3eSJohn Birrell 			error = callback(file, i, &symval, opaque);
1369a2024a3eSJohn Birrell 			if (error)
1370a2024a3eSJohn Birrell 				return (error);
1371a2024a3eSJohn Birrell 		}
1372a2024a3eSJohn Birrell 	}
1373a2024a3eSJohn Birrell 	return (0);
1374a2024a3eSJohn Birrell }
1375a2024a3eSJohn Birrell 
13762832cd54SKonstantin Belousov static void
13772832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef)
13782832cd54SKonstantin Belousov {
13792832cd54SKonstantin Belousov 	Elf_Sym *sym;
13802832cd54SKonstantin Belousov 	Elf_Size i;
13812832cd54SKonstantin Belousov 
13822832cd54SKonstantin Belousov 	for (i = 0; i < ef->ddbsymcnt; i++) {
13832832cd54SKonstantin Belousov 		sym = ef->ddbsymtab + i;
13842832cd54SKonstantin Belousov 		if (sym->st_shndx == SHN_FBSD_CACHED) {
13852832cd54SKonstantin Belousov 			sym->st_shndx = SHN_UNDEF;
13862832cd54SKonstantin Belousov 			sym->st_value = 0;
13872832cd54SKonstantin Belousov 		}
13882832cd54SKonstantin Belousov 	}
13892832cd54SKonstantin Belousov }
13902832cd54SKonstantin Belousov 
1391d297ad16SMarcel Moolenaar /*
1392d297ad16SMarcel Moolenaar  * Symbol lookup function that can be used when the symbol index is known (ie
1393d297ad16SMarcel Moolenaar  * in relocations). It uses the symbol index instead of doing a fully fledged
1394d297ad16SMarcel Moolenaar  * hash table based lookup when such is valid. For example for local symbols.
1395d297ad16SMarcel Moolenaar  * This is not only more efficient, it's also more correct. It's not always
1396d297ad16SMarcel Moolenaar  * the case that the symbol can be found through the hash table.
1397d297ad16SMarcel Moolenaar  */
1398cff8c6f2SKonstantin Belousov static int
1399cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res)
1400d297ad16SMarcel Moolenaar {
1401d297ad16SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
14022832cd54SKonstantin Belousov 	Elf_Sym *sym;
1403d297ad16SMarcel Moolenaar 	const char *symbol;
1404cff8c6f2SKonstantin Belousov 	Elf_Addr res1;
1405d297ad16SMarcel Moolenaar 
1406d297ad16SMarcel Moolenaar 	/* Don't even try to lookup the symbol if the index is bogus. */
1407cff8c6f2SKonstantin Belousov 	if (symidx >= ef->ddbsymcnt) {
1408cff8c6f2SKonstantin Belousov 		*res = 0;
1409cff8c6f2SKonstantin Belousov 		return (EINVAL);
1410cff8c6f2SKonstantin Belousov 	}
1411d297ad16SMarcel Moolenaar 
1412e9eabf59SPeter Wemm 	sym = ef->ddbsymtab + symidx;
1413d297ad16SMarcel Moolenaar 
14144cec6f5dSPeter Wemm 	/* Quick answer if there is a definition included. */
1415cff8c6f2SKonstantin Belousov 	if (sym->st_shndx != SHN_UNDEF) {
1416289ead7cSKonstantin Belousov 		res1 = (Elf_Addr)sym->st_value;
1417289ead7cSKonstantin Belousov 		if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC)
1418289ead7cSKonstantin Belousov 			res1 = ((Elf_Addr (*)(void))res1)();
1419289ead7cSKonstantin Belousov 		*res = res1;
1420cff8c6f2SKonstantin Belousov 		return (0);
1421cff8c6f2SKonstantin Belousov 	}
14224cec6f5dSPeter Wemm 
14234cec6f5dSPeter Wemm 	/* If we get here, then it is undefined and needs a lookup. */
14244cec6f5dSPeter Wemm 	switch (ELF_ST_BIND(sym->st_info)) {
14254cec6f5dSPeter Wemm 	case STB_LOCAL:
14264cec6f5dSPeter Wemm 		/* Local, but undefined? huh? */
1427cff8c6f2SKonstantin Belousov 		*res = 0;
1428cff8c6f2SKonstantin Belousov 		return (EINVAL);
1429d297ad16SMarcel Moolenaar 
1430e9eabf59SPeter Wemm 	case STB_GLOBAL:
1431cff8c6f2SKonstantin Belousov 	case STB_WEAK:
1432e9eabf59SPeter Wemm 		/* Relative to Data or Function name */
1433e9eabf59SPeter Wemm 		symbol = ef->ddbstrtab + sym->st_name;
1434d297ad16SMarcel Moolenaar 
1435d297ad16SMarcel Moolenaar 		/* Force a lookup failure if the symbol name is bogus. */
1436cff8c6f2SKonstantin Belousov 		if (*symbol == 0) {
1437cff8c6f2SKonstantin Belousov 			*res = 0;
1438cff8c6f2SKonstantin Belousov 			return (EINVAL);
1439cff8c6f2SKonstantin Belousov 		}
1440cff8c6f2SKonstantin Belousov 		res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps);
14412832cd54SKonstantin Belousov 
14422832cd54SKonstantin Belousov 		/*
14432832cd54SKonstantin Belousov 		 * Cache global lookups during module relocation. The failure
14442832cd54SKonstantin Belousov 		 * case is particularly expensive for callers, who must scan
14452832cd54SKonstantin Belousov 		 * through the entire globals table doing strcmp(). Cache to
14462832cd54SKonstantin Belousov 		 * avoid doing such work repeatedly.
14472832cd54SKonstantin Belousov 		 *
14482832cd54SKonstantin Belousov 		 * After relocation is complete, undefined globals will be
14492832cd54SKonstantin Belousov 		 * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(),
14502832cd54SKonstantin Belousov 		 * above.
14512832cd54SKonstantin Belousov 		 */
1452cff8c6f2SKonstantin Belousov 		if (res1 != 0) {
14532832cd54SKonstantin Belousov 			sym->st_shndx = SHN_FBSD_CACHED;
1454cff8c6f2SKonstantin Belousov 			sym->st_value = res1;
1455cff8c6f2SKonstantin Belousov 			*res = res1;
1456e9eabf59SPeter Wemm 			return (0);
1457cff8c6f2SKonstantin Belousov 		} else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) {
1458cff8c6f2SKonstantin Belousov 			sym->st_value = 0;
1459cff8c6f2SKonstantin Belousov 			*res = 0;
1460cff8c6f2SKonstantin Belousov 			return (0);
1461cff8c6f2SKonstantin Belousov 		}
1462cff8c6f2SKonstantin Belousov 		return (EINVAL);
1463e9eabf59SPeter Wemm 
1464e9eabf59SPeter Wemm 	default:
1465cff8c6f2SKonstantin Belousov 		return (EINVAL);
1466d297ad16SMarcel Moolenaar 	}
1467e9eabf59SPeter Wemm }
1468e9eabf59SPeter Wemm 
14697251b4bfSJake Burkholder static void
14707226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef)
14717226306eSKonstantin Belousov {
14727226306eSKonstantin Belousov 	static const char startn[] = "__start_";
14737226306eSKonstantin Belousov 	static const char stopn[] = "__stop_";
14747226306eSKonstantin Belousov 	Elf_Sym *sym;
14757226306eSKonstantin Belousov 	const char *sym_name, *linkset_name;
14767226306eSKonstantin Belousov 	Elf_Addr startp, stopp;
14777226306eSKonstantin Belousov 	Elf_Size symidx;
14787226306eSKonstantin Belousov 	int start, i;
14797226306eSKonstantin Belousov 
14807226306eSKonstantin Belousov 	startp = stopp = 0;
14817226306eSKonstantin Belousov 	for (symidx = 1 /* zero entry is special */;
14827226306eSKonstantin Belousov 		symidx < ef->ddbsymcnt; symidx++) {
14837226306eSKonstantin Belousov 		sym = ef->ddbsymtab + symidx;
14847226306eSKonstantin Belousov 		if (sym->st_shndx != SHN_UNDEF)
14857226306eSKonstantin Belousov 			continue;
14867226306eSKonstantin Belousov 
14877226306eSKonstantin Belousov 		sym_name = ef->ddbstrtab + sym->st_name;
14887226306eSKonstantin Belousov 		if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) {
14897226306eSKonstantin Belousov 			start = 1;
14907226306eSKonstantin Belousov 			linkset_name = sym_name + sizeof(startn) - 1;
14917226306eSKonstantin Belousov 		}
14927226306eSKonstantin Belousov 		else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) {
14937226306eSKonstantin Belousov 			start = 0;
14947226306eSKonstantin Belousov 			linkset_name = sym_name + sizeof(stopn) - 1;
14957226306eSKonstantin Belousov 		}
14967226306eSKonstantin Belousov 		else
14977226306eSKonstantin Belousov 			continue;
14987226306eSKonstantin Belousov 
14997226306eSKonstantin Belousov 		for (i = 0; i < ef->nprogtab; i++) {
15007226306eSKonstantin Belousov 			if (strcmp(ef->progtab[i].name, linkset_name) == 0) {
15017226306eSKonstantin Belousov 				startp = (Elf_Addr)ef->progtab[i].addr;
15027226306eSKonstantin Belousov 				stopp = (Elf_Addr)(startp + ef->progtab[i].size);
15037226306eSKonstantin Belousov 				break;
15047226306eSKonstantin Belousov 			}
15057226306eSKonstantin Belousov 		}
15067226306eSKonstantin Belousov 		if (i == ef->nprogtab)
15077226306eSKonstantin Belousov 			continue;
15087226306eSKonstantin Belousov 
15097226306eSKonstantin Belousov 		sym->st_value = start ? startp : stopp;
15107226306eSKonstantin Belousov 		sym->st_shndx = i;
15117226306eSKonstantin Belousov 	}
15127226306eSKonstantin Belousov }
15137226306eSKonstantin Belousov 
15143cfce8e4SKonstantin Belousov static int
1515289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs)
15167251b4bfSJake Burkholder {
1517e9eabf59SPeter Wemm 	elf_file_t ef = (elf_file_t)lf;
15187251b4bfSJake Burkholder 	const Elf_Rel *rellim;
15197251b4bfSJake Burkholder 	const Elf_Rel *rel;
15207251b4bfSJake Burkholder 	const Elf_Rela *relalim;
15217251b4bfSJake Burkholder 	const Elf_Rela *rela;
1522e9eabf59SPeter Wemm 	const Elf_Sym *sym;
1523e9eabf59SPeter Wemm 	Elf_Addr base;
1524e9eabf59SPeter Wemm 	int i;
1525757686b1SMarcel Moolenaar 	Elf_Size symidx;
1526e9eabf59SPeter Wemm 
15277226306eSKonstantin Belousov 	link_elf_fix_link_set(ef);
15287226306eSKonstantin Belousov 
15297251b4bfSJake Burkholder 	/* Perform relocations without addend if there are any: */
1530b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++) {
1531e9eabf59SPeter Wemm 		rel = ef->reltab[i].rel;
15323cfce8e4SKonstantin Belousov 		if (rel == NULL) {
15333cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a reltab");
15343cfce8e4SKonstantin Belousov 			return (ENOEXEC);
15353cfce8e4SKonstantin Belousov 		}
153620947801SPeter Wemm 		rellim = rel + ef->reltab[i].nrel;
1537e9eabf59SPeter Wemm 		base = findbase(ef, ef->reltab[i].sec);
15383cfce8e4SKonstantin Belousov 		if (base == 0) {
15393cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost base for reltab");
15403cfce8e4SKonstantin Belousov 			return (ENOEXEC);
15413cfce8e4SKonstantin Belousov 		}
154220947801SPeter Wemm 		for ( ; rel < rellim; rel++) {
1543e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rel->r_info);
154420947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
154520947801SPeter Wemm 				continue;
1546e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
154720947801SPeter Wemm 			/* Only do local relocs */
154820947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) != STB_LOCAL)
154920947801SPeter Wemm 				continue;
155036716fe2SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
155136716fe2SMark Johnston 			    elf_is_ifunc_reloc(rel->r_info)) == ifuncs)
155220947801SPeter Wemm 				elf_reloc_local(lf, base, rel, ELF_RELOC_REL,
155320947801SPeter Wemm 				    elf_obj_lookup);
15547251b4bfSJake Burkholder 		}
1555e9eabf59SPeter Wemm 	}
15567251b4bfSJake Burkholder 
15577251b4bfSJake Burkholder 	/* Perform relocations with addend if there are any: */
1558b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++) {
1559e9eabf59SPeter Wemm 		rela = ef->relatab[i].rela;
15603cfce8e4SKonstantin Belousov 		if (rela == NULL) {
15613cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a relatab!");
15623cfce8e4SKonstantin Belousov 			return (ENOEXEC);
15633cfce8e4SKonstantin Belousov 		}
156420947801SPeter Wemm 		relalim = rela + ef->relatab[i].nrela;
1565e9eabf59SPeter Wemm 		base = findbase(ef, ef->relatab[i].sec);
15663cfce8e4SKonstantin Belousov 		if (base == 0) {
15673cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost base for reltab");
15683cfce8e4SKonstantin Belousov 			return (ENOEXEC);
15693cfce8e4SKonstantin Belousov 		}
157020947801SPeter Wemm 		for ( ; rela < relalim; rela++) {
1571e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rela->r_info);
157220947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
157320947801SPeter Wemm 				continue;
1574e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
157520947801SPeter Wemm 			/* Only do local relocs */
157620947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) != STB_LOCAL)
157720947801SPeter Wemm 				continue;
157836716fe2SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
157936716fe2SMark Johnston 			    elf_is_ifunc_reloc(rela->r_info)) == ifuncs)
158020947801SPeter Wemm 				elf_reloc_local(lf, base, rela, ELF_RELOC_RELA,
158120947801SPeter Wemm 				    elf_obj_lookup);
15827251b4bfSJake Burkholder 		}
15837251b4bfSJake Burkholder 	}
15843cfce8e4SKonstantin Belousov 	return (0);
1585e9eabf59SPeter Wemm }
158600a5db46SStacey Son 
158700a5db46SStacey Son static long
1588e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab)
158900a5db46SStacey Son {
159000a5db46SStacey Son     elf_file_t ef = (elf_file_t)lf;
159100a5db46SStacey Son 
159200a5db46SStacey Son     *symtab = ef->ddbsymtab;
159300a5db46SStacey Son 
159400a5db46SStacey Son     if (*symtab == NULL)
159500a5db46SStacey Son         return (0);
159600a5db46SStacey Son 
159700a5db46SStacey Son     return (ef->ddbsymcnt);
159800a5db46SStacey Son }
159900a5db46SStacey Son 
160000a5db46SStacey Son static long
160100a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab)
160200a5db46SStacey Son {
160300a5db46SStacey Son     elf_file_t ef = (elf_file_t)lf;
160400a5db46SStacey Son 
160500a5db46SStacey Son     *strtab = ef->ddbstrtab;
160600a5db46SStacey Son 
160700a5db46SStacey Son     if (*strtab == NULL)
160800a5db46SStacey Son         return (0);
160900a5db46SStacey Son 
161000a5db46SStacey Son     return (ef->ddbstrcnt);
161100a5db46SStacey Son }
1612