xref: /freebsd/sys/kern/link_elf.c (revision a2024a3edff86b418e3e599ee5db65a7e653605d)
1a4f67738SDoug Rabson /*-
2326e27d8SDoug Rabson  * Copyright (c) 1998-2000 Doug Rabson
3a4f67738SDoug Rabson  * All rights reserved.
4a4f67738SDoug Rabson  *
5a4f67738SDoug Rabson  * Redistribution and use in source and binary forms, with or without
6a4f67738SDoug Rabson  * modification, are permitted provided that the following conditions
7a4f67738SDoug Rabson  * are met:
8a4f67738SDoug Rabson  * 1. Redistributions of source code must retain the above copyright
9a4f67738SDoug Rabson  *    notice, this list of conditions and the following disclaimer.
10a4f67738SDoug Rabson  * 2. Redistributions in binary form must reproduce the above copyright
11a4f67738SDoug Rabson  *    notice, this list of conditions and the following disclaimer in the
12a4f67738SDoug Rabson  *    documentation and/or other materials provided with the distribution.
13a4f67738SDoug Rabson  *
14a4f67738SDoug Rabson  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15a4f67738SDoug Rabson  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16a4f67738SDoug Rabson  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17a4f67738SDoug Rabson  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18a4f67738SDoug Rabson  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19a4f67738SDoug Rabson  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20a4f67738SDoug Rabson  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21a4f67738SDoug Rabson  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22a4f67738SDoug Rabson  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23a4f67738SDoug Rabson  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24a4f67738SDoug Rabson  * SUCH DAMAGE.
25a4f67738SDoug Rabson  */
26a4f67738SDoug Rabson 
27677b542eSDavid E. O'Brien #include <sys/cdefs.h>
28677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$");
29677b542eSDavid E. O'Brien 
30a2024a3eSJohn Birrell #include "opt_ddb.h"
31ecb01c64SMarcel Moolenaar #include "opt_gdb.h"
32a3df768bSRobert Watson #include "opt_mac.h"
33326e27d8SDoug Rabson 
34a4f67738SDoug Rabson #include <sys/param.h>
35a4f67738SDoug Rabson #include <sys/systm.h>
365ad6c3b1SBruce Evans #ifdef GPROF
375ad6c3b1SBruce Evans #include <sys/gmon.h>
385ad6c3b1SBruce Evans #endif
39fb919e4dSMark Murray #include <sys/kernel.h>
40fb919e4dSMark Murray #include <sys/lock.h>
41a4f67738SDoug Rabson #include <sys/malloc.h>
42c0b824f9SDima Dorfman #include <sys/mutex.h>
4362d615d5SJohn Baldwin #include <sys/mount.h>
44a4f67738SDoug Rabson #include <sys/proc.h>
45a4f67738SDoug Rabson #include <sys/namei.h>
46a4f67738SDoug Rabson #include <sys/fcntl.h>
47a4f67738SDoug Rabson #include <sys/vnode.h>
48a4f67738SDoug Rabson #include <sys/linker.h>
49fb919e4dSMark Murray 
50a4f67738SDoug Rabson #include <machine/elf.h>
51a4f67738SDoug Rabson 
52aed55708SRobert Watson #include <security/mac/mac_framework.h>
53aed55708SRobert Watson 
54fe3db7c7SDoug Rabson #include <vm/vm.h>
55fe3db7c7SDoug Rabson #include <vm/vm_param.h>
56d1f088daSPeter Wemm #ifdef SPARSE_MAPPING
57fe3db7c7SDoug Rabson #include <vm/vm_object.h>
58fe3db7c7SDoug Rabson #include <vm/vm_kern.h>
59fe3db7c7SDoug Rabson #include <vm/vm_extern.h>
60d1f088daSPeter Wemm #endif
61fe3db7c7SDoug Rabson #include <vm/pmap.h>
62fe3db7c7SDoug Rabson #include <vm/vm_map.h>
63fb919e4dSMark Murray 
64f9980387SPeter Wemm #include <sys/link_elf.h>
65fe3db7c7SDoug Rabson 
66a2024a3eSJohn Birrell #ifdef DDB_CTF
67a2024a3eSJohn Birrell #include <net/zlib.h>
68a2024a3eSJohn Birrell #endif
69a2024a3eSJohn Birrell 
70326e27d8SDoug Rabson #include "linker_if.h"
71a4f67738SDoug Rabson 
721a330eb0SPeter Wemm #define MAXSEGS 4
731a330eb0SPeter Wemm 
74a4f67738SDoug Rabson typedef struct elf_file {
75326e27d8SDoug Rabson     struct linker_file	lf;		/* Common fields */
76326e27d8SDoug Rabson     int			preloaded;	/* Was file pre-loaded */
77fe3db7c7SDoug Rabson     caddr_t		address;	/* Relocation address */
78fe3db7c7SDoug Rabson #ifdef SPARSE_MAPPING
79fe3db7c7SDoug Rabson     vm_object_t		object;		/* VM object to hold file pages */
80fe3db7c7SDoug Rabson #endif
81326e27d8SDoug Rabson     Elf_Dyn*		dynamic;	/* Symbol table etc. */
827e510298SJake Burkholder     Elf_Hashelt		nbuckets;	/* DT_HASH info */
837e510298SJake Burkholder     Elf_Hashelt		nchains;
847e510298SJake Burkholder     const Elf_Hashelt*	buckets;
857e510298SJake Burkholder     const Elf_Hashelt*	chains;
86a4f67738SDoug Rabson     caddr_t		hash;
87a4f67738SDoug Rabson     caddr_t		strtab;		/* DT_STRTAB */
882d636ab0SPeter Wemm     int			strsz;		/* DT_STRSZ */
89fe3db7c7SDoug Rabson     const Elf_Sym*	symtab;		/* DT_SYMTAB */
90fe3db7c7SDoug Rabson     Elf_Addr*		got;		/* DT_PLTGOT */
91fe3db7c7SDoug Rabson     const Elf_Rel*	pltrel;		/* DT_JMPREL */
92fe3db7c7SDoug Rabson     int			pltrelsize;	/* DT_PLTRELSZ */
93fe3db7c7SDoug Rabson     const Elf_Rela*	pltrela;	/* DT_JMPREL */
94fe3db7c7SDoug Rabson     int			pltrelasize;	/* DT_PLTRELSZ */
95fe3db7c7SDoug Rabson     const Elf_Rel*	rel;		/* DT_REL */
96fe3db7c7SDoug Rabson     int			relsize;	/* DT_RELSZ */
97fe3db7c7SDoug Rabson     const Elf_Rela*	rela;		/* DT_RELA */
98fe3db7c7SDoug Rabson     int			relasize;	/* DT_RELASZ */
992d636ab0SPeter Wemm     caddr_t		modptr;
1002d636ab0SPeter Wemm     const Elf_Sym*	ddbsymtab;	/* The symbol table we are using */
1012d636ab0SPeter Wemm     long		ddbsymcnt;	/* Number of symbols */
1022d636ab0SPeter Wemm     caddr_t		ddbstrtab;	/* String table */
1032d636ab0SPeter Wemm     long		ddbstrcnt;	/* number of bytes in string table */
104ca65d5c7SPeter Wemm     caddr_t		symbase;	/* malloc'ed symbold base */
105ca65d5c7SPeter Wemm     caddr_t		strbase;	/* malloc'ed string base */
106a2024a3eSJohn Birrell     caddr_t		ctftab;		/* CTF table */
107a2024a3eSJohn Birrell     long		ctfcnt;		/* number of bytes in CTF table */
108a2024a3eSJohn Birrell     caddr_t		ctfoff;		/* CTF offset table */
109a2024a3eSJohn Birrell     caddr_t		typoff;		/* Type offset table */
110a2024a3eSJohn Birrell     long		typlen;		/* Number of type entries. */
111ecb01c64SMarcel Moolenaar #ifdef GDB
112326e27d8SDoug Rabson     struct link_map	gdb;		/* hooks for gdb */
113326e27d8SDoug Rabson #endif
114a4f67738SDoug Rabson } *elf_file_t;
115a4f67738SDoug Rabson 
116a2024a3eSJohn Birrell #include <kern/kern_ctf.c>
117a2024a3eSJohn Birrell 
118c143d6c2SMarcel Moolenaar static int	link_elf_link_common_finish(linker_file_t);
11954823af2SPeter Wemm static int	link_elf_link_preload(linker_class_t cls,
120326e27d8SDoug Rabson 				      const char*, linker_file_t*);
12154823af2SPeter Wemm static int	link_elf_link_preload_finish(linker_file_t);
12254823af2SPeter Wemm static int	link_elf_load_file(linker_class_t, const char*, linker_file_t*);
123326e27d8SDoug Rabson static int	link_elf_lookup_symbol(linker_file_t, const char*,
124326e27d8SDoug Rabson 				       c_linker_sym_t*);
125326e27d8SDoug Rabson static int	link_elf_symbol_values(linker_file_t, c_linker_sym_t, linker_symval_t*);
126326e27d8SDoug Rabson static int	link_elf_search_symbol(linker_file_t, caddr_t value,
127326e27d8SDoug Rabson 				       c_linker_sym_t* sym, long* diffp);
128326e27d8SDoug Rabson 
129326e27d8SDoug Rabson static void	link_elf_unload_file(linker_file_t);
13054823af2SPeter Wemm static void	link_elf_unload_preload(linker_file_t);
131f41325dbSPeter Wemm static int	link_elf_lookup_set(linker_file_t, const char *,
132f41325dbSPeter Wemm 				    void ***, void ***, int *);
133bb9fe9ddSBrian Feldman static int	link_elf_each_function_name(linker_file_t,
134bb9fe9ddSBrian Feldman 				int (*)(const char *, void *),
135bb9fe9ddSBrian Feldman 				void *);
136a2024a3eSJohn Birrell static int	link_elf_each_function_nameval(linker_file_t,
137a2024a3eSJohn Birrell 				linker_function_nameval_callback_t,
138a2024a3eSJohn Birrell 				void *);
1397251b4bfSJake Burkholder static void	link_elf_reloc_local(linker_file_t);
140757686b1SMarcel Moolenaar static Elf_Addr	elf_lookup(linker_file_t lf, Elf_Size symidx, int deps);
141326e27d8SDoug Rabson 
142326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = {
143326e27d8SDoug Rabson     KOBJMETHOD(linker_lookup_symbol,	link_elf_lookup_symbol),
144326e27d8SDoug Rabson     KOBJMETHOD(linker_symbol_values,	link_elf_symbol_values),
145326e27d8SDoug Rabson     KOBJMETHOD(linker_search_symbol,	link_elf_search_symbol),
146326e27d8SDoug Rabson     KOBJMETHOD(linker_unload,		link_elf_unload_file),
14754823af2SPeter Wemm     KOBJMETHOD(linker_load_file,	link_elf_load_file),
14854823af2SPeter Wemm     KOBJMETHOD(linker_link_preload,	link_elf_link_preload),
14954823af2SPeter Wemm     KOBJMETHOD(linker_link_preload_finish, link_elf_link_preload_finish),
150f41325dbSPeter Wemm     KOBJMETHOD(linker_lookup_set,	link_elf_lookup_set),
151bb9fe9ddSBrian Feldman     KOBJMETHOD(linker_each_function_name, link_elf_each_function_name),
152a2024a3eSJohn Birrell     KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval),
153a2024a3eSJohn Birrell     KOBJMETHOD(linker_ctf_get,          link_elf_ctf_get),
154326e27d8SDoug Rabson     { 0, 0 }
155326e27d8SDoug Rabson };
156326e27d8SDoug Rabson 
157326e27d8SDoug Rabson static struct linker_class link_elf_class = {
158326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32
159326e27d8SDoug Rabson     "elf32",
160326e27d8SDoug Rabson #else
161326e27d8SDoug Rabson     "elf64",
162326e27d8SDoug Rabson #endif
163326e27d8SDoug Rabson     link_elf_methods, sizeof(struct elf_file)
164326e27d8SDoug Rabson };
165326e27d8SDoug Rabson 
166326e27d8SDoug Rabson static int		parse_dynamic(elf_file_t ef);
167326e27d8SDoug Rabson static int		relocate_file(elf_file_t ef);
16854823af2SPeter Wemm static int		link_elf_preload_parse_symbols(elf_file_t ef);
169326e27d8SDoug Rabson 
170ecb01c64SMarcel Moolenaar #ifdef GDB
171ca29467eSJohn Baldwin static void		r_debug_state(struct r_debug *dummy_one,
172ca29467eSJohn Baldwin 				      struct link_map *dummy_two);
173326e27d8SDoug Rabson 
174326e27d8SDoug Rabson /*
175326e27d8SDoug Rabson  * A list of loaded modules for GDB to use for loading symbols.
176326e27d8SDoug Rabson  */
177326e27d8SDoug Rabson struct r_debug r_debug;
178326e27d8SDoug Rabson 
179ca29467eSJohn Baldwin #define GDB_STATE(s)	r_debug.r_state = s; r_debug_state(NULL, NULL);
180326e27d8SDoug Rabson 
181326e27d8SDoug Rabson /*
182326e27d8SDoug Rabson  * Function for the debugger to set a breakpoint on to gain control.
183326e27d8SDoug Rabson  */
18437c84183SPoul-Henning Kamp static void
185ca29467eSJohn Baldwin r_debug_state(struct r_debug *dummy_one __unused,
186ca29467eSJohn Baldwin 	      struct link_map *dummy_two __unused)
187326e27d8SDoug Rabson {
188326e27d8SDoug Rabson }
189326e27d8SDoug Rabson 
1901720979bSMarcel Moolenaar static void
1911720979bSMarcel Moolenaar link_elf_add_gdb(struct link_map *l)
1921720979bSMarcel Moolenaar {
1931720979bSMarcel Moolenaar     struct link_map *prev;
1941720979bSMarcel Moolenaar 
195c143d6c2SMarcel Moolenaar     l->l_next = NULL;
196c143d6c2SMarcel Moolenaar 
197c143d6c2SMarcel Moolenaar     if (r_debug.r_map == NULL) {
198c143d6c2SMarcel Moolenaar 	/* Add first. */
199c143d6c2SMarcel Moolenaar 	l->l_prev = NULL;
200c143d6c2SMarcel Moolenaar 	r_debug.r_map = l;
201c143d6c2SMarcel Moolenaar     } else {
202c143d6c2SMarcel Moolenaar 	/* Append to list. */
2031720979bSMarcel Moolenaar 	for (prev = r_debug.r_map; prev->l_next != NULL; prev = prev->l_next)
2041720979bSMarcel Moolenaar 	    ;
2051720979bSMarcel Moolenaar 	l->l_prev = prev;
2061720979bSMarcel Moolenaar 	prev->l_next = l;
2071720979bSMarcel Moolenaar     }
208c143d6c2SMarcel Moolenaar }
2091720979bSMarcel Moolenaar 
2101720979bSMarcel Moolenaar static void
2111720979bSMarcel Moolenaar link_elf_delete_gdb(struct link_map *l)
2121720979bSMarcel Moolenaar {
2131720979bSMarcel Moolenaar     if (l->l_prev == NULL) {
214c143d6c2SMarcel Moolenaar 	/* Remove first. */
2151720979bSMarcel Moolenaar 	if ((r_debug.r_map = l->l_next) != NULL)
2161720979bSMarcel Moolenaar 	    l->l_next->l_prev = NULL;
217c143d6c2SMarcel Moolenaar     } else {
218c143d6c2SMarcel Moolenaar 	/* Remove any but first. */
2191720979bSMarcel Moolenaar 	if ((l->l_prev->l_next = l->l_next) != NULL)
2201720979bSMarcel Moolenaar 	    l->l_next->l_prev = l->l_prev;
2211720979bSMarcel Moolenaar     }
222c143d6c2SMarcel Moolenaar }
223ecb01c64SMarcel Moolenaar #endif /* GDB */
224a4f67738SDoug Rabson 
22584201059SMarcel Moolenaar #ifdef __ia64__
22684201059SMarcel Moolenaar Elf_Addr link_elf_get_gp(linker_file_t);
22784201059SMarcel Moolenaar #endif
22884201059SMarcel Moolenaar 
229a4f67738SDoug Rabson /*
230a4f67738SDoug Rabson  * The kernel symbol table starts here.
231a4f67738SDoug Rabson  */
232a4f67738SDoug Rabson extern struct _dynamic _DYNAMIC;
233a4f67738SDoug Rabson 
234a4f67738SDoug Rabson static void
2351720979bSMarcel Moolenaar link_elf_error(const char *s)
2361720979bSMarcel Moolenaar {
2371720979bSMarcel Moolenaar     printf("kldload: %s\n", s);
2381720979bSMarcel Moolenaar }
2391720979bSMarcel Moolenaar 
240c143d6c2SMarcel Moolenaar /*
241c143d6c2SMarcel Moolenaar  * Actions performed after linking/loading both the preloaded kernel and any
242c143d6c2SMarcel Moolenaar  * modules; whether preloaded or dynamicly loaded.
243c143d6c2SMarcel Moolenaar  */
244c143d6c2SMarcel Moolenaar static int
245c143d6c2SMarcel Moolenaar link_elf_link_common_finish(linker_file_t lf)
246c143d6c2SMarcel Moolenaar {
247ecb01c64SMarcel Moolenaar #ifdef GDB
248c143d6c2SMarcel Moolenaar     elf_file_t ef = (elf_file_t)lf;
249c143d6c2SMarcel Moolenaar     char *newfilename;
250c143d6c2SMarcel Moolenaar #endif
2511aeb23cdSMarcel Moolenaar     int error;
2521aeb23cdSMarcel Moolenaar 
2531aeb23cdSMarcel Moolenaar     /* Notify MD code that a module is being loaded. */
2541aeb23cdSMarcel Moolenaar     error = elf_cpu_load_file(lf);
2551aeb23cdSMarcel Moolenaar     if (error)
2561aeb23cdSMarcel Moolenaar 	return (error);
257c143d6c2SMarcel Moolenaar 
258ecb01c64SMarcel Moolenaar #ifdef GDB
259c143d6c2SMarcel Moolenaar     GDB_STATE(RT_ADD);
260c143d6c2SMarcel Moolenaar     ef->gdb.l_addr = lf->address;
261a163d034SWarner Losh     newfilename = malloc(strlen(lf->filename) + 1, M_LINKER, M_WAITOK);
262c143d6c2SMarcel Moolenaar     strcpy(newfilename, lf->filename);
263c143d6c2SMarcel Moolenaar     ef->gdb.l_name = newfilename;
264c143d6c2SMarcel Moolenaar     ef->gdb.l_ld = ef->dynamic;
265c143d6c2SMarcel Moolenaar     link_elf_add_gdb(&ef->gdb);
266c143d6c2SMarcel Moolenaar     GDB_STATE(RT_CONSISTENT);
267c143d6c2SMarcel Moolenaar #endif
268c143d6c2SMarcel Moolenaar 
269c143d6c2SMarcel Moolenaar     return (0);
270c143d6c2SMarcel Moolenaar }
271c143d6c2SMarcel Moolenaar 
2721720979bSMarcel Moolenaar static void
273a4f67738SDoug Rabson link_elf_init(void* arg)
274a4f67738SDoug Rabson {
275de78ca7eSPeter Wemm     Elf_Dyn	*dp;
276de78ca7eSPeter Wemm     caddr_t	modptr, baseptr, sizeptr;
277de78ca7eSPeter Wemm     elf_file_t	ef;
278de78ca7eSPeter Wemm     char	*modname;
279a4f67738SDoug Rabson 
280326e27d8SDoug Rabson     linker_add_class(&link_elf_class);
281a4f67738SDoug Rabson 
282de78ca7eSPeter Wemm     dp = (Elf_Dyn*) &_DYNAMIC;
283de78ca7eSPeter Wemm     modname = NULL;
28467db8b23SPeter Wemm     modptr = preload_search_by_type("elf" __XSTRING(__ELF_WORD_SIZE) " kernel");
28567db8b23SPeter Wemm     if (modptr == NULL)
286de78ca7eSPeter Wemm 	modptr = preload_search_by_type("elf kernel");
287de78ca7eSPeter Wemm     if (modptr)
288de78ca7eSPeter Wemm 	modname = (char *)preload_search_info(modptr, MODINFO_NAME);
289de78ca7eSPeter Wemm     if (modname == NULL)
290de78ca7eSPeter Wemm 	modname = "kernel";
291326e27d8SDoug Rabson     linker_kernel_file = linker_make_file(modname, &link_elf_class);
292a4f67738SDoug Rabson     if (linker_kernel_file == NULL)
293a4f67738SDoug Rabson 	panic("link_elf_init: Can't create linker structures for kernel");
294326e27d8SDoug Rabson 
295326e27d8SDoug Rabson     ef = (elf_file_t) linker_kernel_file;
29654823af2SPeter Wemm     ef->preloaded = 1;
297326e27d8SDoug Rabson     ef->address = 0;
298326e27d8SDoug Rabson #ifdef SPARSE_MAPPING
299326e27d8SDoug Rabson     ef->object = 0;
300326e27d8SDoug Rabson #endif
301326e27d8SDoug Rabson     ef->dynamic = dp;
302326e27d8SDoug Rabson 
30300dda5e8SPeter Wemm     if (dp)
304326e27d8SDoug Rabson 	parse_dynamic(ef);
3054173e420SPeter Wemm     linker_kernel_file->address = (caddr_t) KERNBASE;
306d009ccfaSBruce Evans     linker_kernel_file->size = -(intptr_t)linker_kernel_file->address;
307de78ca7eSPeter Wemm 
308de78ca7eSPeter Wemm     if (modptr) {
3092d636ab0SPeter Wemm 	ef->modptr = modptr;
310de78ca7eSPeter Wemm 	baseptr = preload_search_info(modptr, MODINFO_ADDR);
311de78ca7eSPeter Wemm 	if (baseptr)
312de78ca7eSPeter Wemm 	    linker_kernel_file->address = *(caddr_t *)baseptr;
313de78ca7eSPeter Wemm 	sizeptr = preload_search_info(modptr, MODINFO_SIZE);
314de78ca7eSPeter Wemm 	if (sizeptr)
315de78ca7eSPeter Wemm 	    linker_kernel_file->size = *(size_t *)sizeptr;
316de78ca7eSPeter Wemm     }
31754823af2SPeter Wemm     (void)link_elf_preload_parse_symbols(ef);
318326e27d8SDoug Rabson 
319ecb01c64SMarcel Moolenaar #ifdef GDB
320c143d6c2SMarcel Moolenaar     r_debug.r_map = NULL;
321326e27d8SDoug Rabson     r_debug.r_brk = r_debug_state;
322326e27d8SDoug Rabson     r_debug.r_state = RT_CONSISTENT;
323326e27d8SDoug Rabson #endif
324c143d6c2SMarcel Moolenaar 
325c143d6c2SMarcel Moolenaar     (void)link_elf_link_common_finish(linker_kernel_file);
32659800afcSJohn Baldwin     linker_kernel_file->flags |= LINKER_FILE_LINKED;
327a4f67738SDoug Rabson }
328a4f67738SDoug Rabson 
3298120ddb4SJung-uk Kim SYSINIT(link_elf, SI_SUB_KLD, SI_ORDER_THIRD, link_elf_init, 0);
330a4f67738SDoug Rabson 
331a4f67738SDoug Rabson static int
33254823af2SPeter Wemm link_elf_preload_parse_symbols(elf_file_t ef)
3332d636ab0SPeter Wemm {
3342d636ab0SPeter Wemm     caddr_t	pointer;
3352d636ab0SPeter Wemm     caddr_t	ssym, esym, base;
3362d636ab0SPeter Wemm     caddr_t	strtab;
3372d636ab0SPeter Wemm     int		strcnt;
3382d636ab0SPeter Wemm     Elf_Sym*	symtab;
3392d636ab0SPeter Wemm     int		symcnt;
3402d636ab0SPeter Wemm 
341ca65d5c7SPeter Wemm     if (ef->modptr == NULL)
342ca65d5c7SPeter Wemm 	return 0;
3432d636ab0SPeter Wemm     pointer = preload_search_info(ef->modptr, MODINFO_METADATA|MODINFOMD_SSYM);
3442d636ab0SPeter Wemm     if (pointer == NULL)
3452d636ab0SPeter Wemm 	return 0;
3462d636ab0SPeter Wemm     ssym = *(caddr_t *)pointer;
3472d636ab0SPeter Wemm     pointer = preload_search_info(ef->modptr, MODINFO_METADATA|MODINFOMD_ESYM);
3482d636ab0SPeter Wemm     if (pointer == NULL)
3492d636ab0SPeter Wemm 	return 0;
3502d636ab0SPeter Wemm     esym = *(caddr_t *)pointer;
3512d636ab0SPeter Wemm 
3522d636ab0SPeter Wemm     base = ssym;
3532d636ab0SPeter Wemm 
3542d636ab0SPeter Wemm     symcnt = *(long *)base;
3552d636ab0SPeter Wemm     base += sizeof(long);
3562d636ab0SPeter Wemm     symtab = (Elf_Sym *)base;
3572d636ab0SPeter Wemm     base += roundup(symcnt, sizeof(long));
3582d636ab0SPeter Wemm 
3592d636ab0SPeter Wemm     if (base > esym || base < ssym) {
3602d636ab0SPeter Wemm 	printf("Symbols are corrupt!\n");
3612d636ab0SPeter Wemm 	return EINVAL;
3622d636ab0SPeter Wemm     }
3632d636ab0SPeter Wemm 
3642d636ab0SPeter Wemm     strcnt = *(long *)base;
3652d636ab0SPeter Wemm     base += sizeof(long);
3662d636ab0SPeter Wemm     strtab = base;
3672d636ab0SPeter Wemm     base += roundup(strcnt, sizeof(long));
3682d636ab0SPeter Wemm 
3692d636ab0SPeter Wemm     if (base > esym || base < ssym) {
3702d636ab0SPeter Wemm 	printf("Symbols are corrupt!\n");
3712d636ab0SPeter Wemm 	return EINVAL;
3722d636ab0SPeter Wemm     }
3732d636ab0SPeter Wemm 
3742d636ab0SPeter Wemm     ef->ddbsymtab = symtab;
3752d636ab0SPeter Wemm     ef->ddbsymcnt = symcnt / sizeof(Elf_Sym);
3762d636ab0SPeter Wemm     ef->ddbstrtab = strtab;
3772d636ab0SPeter Wemm     ef->ddbstrcnt = strcnt;
3782d636ab0SPeter Wemm 
3792d636ab0SPeter Wemm     return 0;
3802d636ab0SPeter Wemm }
3812d636ab0SPeter Wemm 
3822d636ab0SPeter Wemm static int
383326e27d8SDoug Rabson parse_dynamic(elf_file_t ef)
384a4f67738SDoug Rabson {
385326e27d8SDoug Rabson     Elf_Dyn *dp;
386fe3db7c7SDoug Rabson     int plttype = DT_REL;
387a4f67738SDoug Rabson 
388a4f67738SDoug Rabson     for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) {
389a4f67738SDoug Rabson 	switch (dp->d_tag) {
390a4f67738SDoug Rabson 	case DT_HASH:
391a4f67738SDoug Rabson 	{
392a4f67738SDoug Rabson 	    /* From src/libexec/rtld-elf/rtld.c */
3937e510298SJake Burkholder 	    const Elf_Hashelt *hashtab = (const Elf_Hashelt *)
394a4f67738SDoug Rabson 		(ef->address + dp->d_un.d_ptr);
395a4f67738SDoug Rabson 	    ef->nbuckets = hashtab[0];
396a4f67738SDoug Rabson 	    ef->nchains = hashtab[1];
397a4f67738SDoug Rabson 	    ef->buckets = hashtab + 2;
398a4f67738SDoug Rabson 	    ef->chains = ef->buckets + ef->nbuckets;
399a4f67738SDoug Rabson 	    break;
400a4f67738SDoug Rabson 	}
401a4f67738SDoug Rabson 	case DT_STRTAB:
402fe3db7c7SDoug Rabson 	    ef->strtab = (caddr_t) (ef->address + dp->d_un.d_ptr);
403a4f67738SDoug Rabson 	    break;
4042d636ab0SPeter Wemm 	case DT_STRSZ:
4052d636ab0SPeter Wemm 	    ef->strsz = dp->d_un.d_val;
4062d636ab0SPeter Wemm 	    break;
407a4f67738SDoug Rabson 	case DT_SYMTAB:
408fe3db7c7SDoug Rabson 	    ef->symtab = (Elf_Sym*) (ef->address + dp->d_un.d_ptr);
409a4f67738SDoug Rabson 	    break;
410a4f67738SDoug Rabson 	case DT_SYMENT:
411a4f67738SDoug Rabson 	    if (dp->d_un.d_val != sizeof(Elf_Sym))
412a4f67738SDoug Rabson 		return ENOEXEC;
413fe3db7c7SDoug Rabson 	    break;
414fe3db7c7SDoug Rabson 	case DT_PLTGOT:
415fe3db7c7SDoug Rabson 	    ef->got = (Elf_Addr *) (ef->address + dp->d_un.d_ptr);
416fe3db7c7SDoug Rabson 	    break;
417fe3db7c7SDoug Rabson 	case DT_REL:
418fe3db7c7SDoug Rabson 	    ef->rel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr);
419fe3db7c7SDoug Rabson 	    break;
420fe3db7c7SDoug Rabson 	case DT_RELSZ:
421fe3db7c7SDoug Rabson 	    ef->relsize = dp->d_un.d_val;
422fe3db7c7SDoug Rabson 	    break;
423fe3db7c7SDoug Rabson 	case DT_RELENT:
424fe3db7c7SDoug Rabson 	    if (dp->d_un.d_val != sizeof(Elf_Rel))
425fe3db7c7SDoug Rabson 		return ENOEXEC;
426fe3db7c7SDoug Rabson 	    break;
427fe3db7c7SDoug Rabson 	case DT_JMPREL:
428fe3db7c7SDoug Rabson 	    ef->pltrel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr);
429fe3db7c7SDoug Rabson 	    break;
430fe3db7c7SDoug Rabson 	case DT_PLTRELSZ:
431fe3db7c7SDoug Rabson 	    ef->pltrelsize = dp->d_un.d_val;
432fe3db7c7SDoug Rabson 	    break;
433fe3db7c7SDoug Rabson 	case DT_RELA:
434fe3db7c7SDoug Rabson 	    ef->rela = (const Elf_Rela *) (ef->address + dp->d_un.d_ptr);
435fe3db7c7SDoug Rabson 	    break;
436fe3db7c7SDoug Rabson 	case DT_RELASZ:
437fe3db7c7SDoug Rabson 	    ef->relasize = dp->d_un.d_val;
438fe3db7c7SDoug Rabson 	    break;
439fe3db7c7SDoug Rabson 	case DT_RELAENT:
440fe3db7c7SDoug Rabson 	    if (dp->d_un.d_val != sizeof(Elf_Rela))
441fe3db7c7SDoug Rabson 		return ENOEXEC;
442fe3db7c7SDoug Rabson 	    break;
443fe3db7c7SDoug Rabson 	case DT_PLTREL:
444fe3db7c7SDoug Rabson 	    plttype = dp->d_un.d_val;
445fe3db7c7SDoug Rabson 	    if (plttype != DT_REL && plttype != DT_RELA)
446fe3db7c7SDoug Rabson 		return ENOEXEC;
447fe3db7c7SDoug Rabson 	    break;
448ecb01c64SMarcel Moolenaar #ifdef GDB
449326e27d8SDoug Rabson 	case DT_DEBUG:
450326e27d8SDoug Rabson 	    dp->d_un.d_ptr = (Elf_Addr) &r_debug;
451326e27d8SDoug Rabson 	    break;
452326e27d8SDoug Rabson #endif
453a4f67738SDoug Rabson 	}
454a4f67738SDoug Rabson     }
455fe3db7c7SDoug Rabson 
456fe3db7c7SDoug Rabson     if (plttype == DT_RELA) {
457fe3db7c7SDoug Rabson 	ef->pltrela = (const Elf_Rela *) ef->pltrel;
458fe3db7c7SDoug Rabson 	ef->pltrel = NULL;
459fe3db7c7SDoug Rabson 	ef->pltrelasize = ef->pltrelsize;
460fe3db7c7SDoug Rabson 	ef->pltrelsize = 0;
461fe3db7c7SDoug Rabson     }
462fe3db7c7SDoug Rabson 
4632d636ab0SPeter Wemm     ef->ddbsymtab = ef->symtab;
4642d636ab0SPeter Wemm     ef->ddbsymcnt = ef->nchains;
4652d636ab0SPeter Wemm     ef->ddbstrtab = ef->strtab;
4662d636ab0SPeter Wemm     ef->ddbstrcnt = ef->strsz;
4672d636ab0SPeter Wemm 
468a4f67738SDoug Rabson     return 0;
469a4f67738SDoug Rabson }
470a4f67738SDoug Rabson 
471a4f67738SDoug Rabson static int
47254823af2SPeter Wemm link_elf_link_preload(linker_class_t cls,
473326e27d8SDoug Rabson 		      const char* filename, linker_file_t *result)
474de78ca7eSPeter Wemm {
475de78ca7eSPeter Wemm     caddr_t		modptr, baseptr, sizeptr, dynptr;
476de78ca7eSPeter Wemm     char		*type;
477de78ca7eSPeter Wemm     elf_file_t		ef;
478de78ca7eSPeter Wemm     linker_file_t	lf;
479de78ca7eSPeter Wemm     int			error;
480de78ca7eSPeter Wemm     vm_offset_t		dp;
481de78ca7eSPeter Wemm 
48254823af2SPeter Wemm     /* Look to see if we have the file preloaded */
483de78ca7eSPeter Wemm     modptr = preload_search_by_name(filename);
484de78ca7eSPeter Wemm     if (modptr == NULL)
48554823af2SPeter Wemm 	return ENOENT;
486de78ca7eSPeter Wemm 
487de78ca7eSPeter Wemm     type = (char *)preload_search_info(modptr, MODINFO_TYPE);
488de78ca7eSPeter Wemm     baseptr = preload_search_info(modptr, MODINFO_ADDR);
489de78ca7eSPeter Wemm     sizeptr = preload_search_info(modptr, MODINFO_SIZE);
490de78ca7eSPeter Wemm     dynptr = preload_search_info(modptr, MODINFO_METADATA|MODINFOMD_DYNAMIC);
49167db8b23SPeter Wemm     if (type == NULL ||
49267db8b23SPeter Wemm 	(strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE) " module") != 0 &&
49367db8b23SPeter Wemm 	 strcmp(type, "elf module") != 0))
494de78ca7eSPeter Wemm 	return (EFTYPE);
495de78ca7eSPeter Wemm     if (baseptr == NULL || sizeptr == NULL || dynptr == NULL)
496de78ca7eSPeter Wemm 	return (EINVAL);
497de78ca7eSPeter Wemm 
498326e27d8SDoug Rabson     lf = linker_make_file(filename, &link_elf_class);
499326e27d8SDoug Rabson     if (lf == NULL) {
500326e27d8SDoug Rabson 	return ENOMEM;
501326e27d8SDoug Rabson     }
502326e27d8SDoug Rabson 
503326e27d8SDoug Rabson     ef = (elf_file_t) lf;
504326e27d8SDoug Rabson     ef->preloaded = 1;
505ca65d5c7SPeter Wemm     ef->modptr = modptr;
506de78ca7eSPeter Wemm     ef->address = *(caddr_t *)baseptr;
507de78ca7eSPeter Wemm #ifdef SPARSE_MAPPING
508de78ca7eSPeter Wemm     ef->object = 0;
509de78ca7eSPeter Wemm #endif
510de78ca7eSPeter Wemm     dp = (vm_offset_t)ef->address + *(vm_offset_t *)dynptr;
511de78ca7eSPeter Wemm     ef->dynamic = (Elf_Dyn *)dp;
512de78ca7eSPeter Wemm     lf->address = ef->address;
513de78ca7eSPeter Wemm     lf->size = *(size_t *)sizeptr;
514de78ca7eSPeter Wemm 
515326e27d8SDoug Rabson     error = parse_dynamic(ef);
516de78ca7eSPeter Wemm     if (error) {
51765a311fcSPoul-Henning Kamp 	linker_file_unload(lf, LINKER_UNLOAD_FORCE);
518de78ca7eSPeter Wemm 	return error;
519de78ca7eSPeter Wemm     }
5207251b4bfSJake Burkholder     link_elf_reloc_local(lf);
521de78ca7eSPeter Wemm     *result = lf;
522de78ca7eSPeter Wemm     return (0);
523de78ca7eSPeter Wemm }
524de78ca7eSPeter Wemm 
525de78ca7eSPeter Wemm static int
52654823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf)
52754823af2SPeter Wemm {
52854823af2SPeter Wemm     elf_file_t		ef;
52954823af2SPeter Wemm     int error;
53054823af2SPeter Wemm 
53154823af2SPeter Wemm     ef = (elf_file_t) lf;
53254823af2SPeter Wemm #if 0	/* this will be more trouble than it's worth for now */
53354823af2SPeter Wemm     for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) {
53454823af2SPeter Wemm 	if (dp->d_tag != DT_NEEDED)
53554823af2SPeter Wemm 	    continue;
53654823af2SPeter Wemm 	modname = ef->strtab + dp->d_un.d_val;
53754823af2SPeter Wemm 	error = linker_load_module(modname, lf);
53854823af2SPeter Wemm 	if (error)
53954823af2SPeter Wemm 	    goto out;
54054823af2SPeter Wemm     }
54154823af2SPeter Wemm #endif
54254823af2SPeter Wemm     error = relocate_file(ef);
54354823af2SPeter Wemm     if (error)
54454823af2SPeter Wemm 	return error;
54554823af2SPeter Wemm     (void)link_elf_preload_parse_symbols(ef);
54654823af2SPeter Wemm 
547c143d6c2SMarcel Moolenaar     return (link_elf_link_common_finish(lf));
54854823af2SPeter Wemm }
54954823af2SPeter Wemm 
55054823af2SPeter Wemm static int
551c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char* filename,
552c143d6c2SMarcel Moolenaar 	linker_file_t* result)
553a4f67738SDoug Rabson {
554a4f67738SDoug Rabson     struct nameidata nd;
555b40ce416SJulian Elischer     struct thread* td = curthread;	/* XXX */
556caab6e90SPeter Wemm     Elf_Ehdr *hdr;
557caab6e90SPeter Wemm     caddr_t firstpage;
558fe3db7c7SDoug Rabson     int nbytes, i;
559fe3db7c7SDoug Rabson     Elf_Phdr *phdr;
560fe3db7c7SDoug Rabson     Elf_Phdr *phlimit;
5611a330eb0SPeter Wemm     Elf_Phdr *segs[MAXSEGS];
562fe3db7c7SDoug Rabson     int nsegs;
563fe3db7c7SDoug Rabson     Elf_Phdr *phdyn;
564fe3db7c7SDoug Rabson     Elf_Phdr *phphdr;
565fe3db7c7SDoug Rabson     caddr_t mapbase;
566fe3db7c7SDoug Rabson     size_t mapsize;
567fe3db7c7SDoug Rabson     Elf_Off base_offset;
568fe3db7c7SDoug Rabson     Elf_Addr base_vaddr;
569fe3db7c7SDoug Rabson     Elf_Addr base_vlimit;
570a4f67738SDoug Rabson     int error = 0;
571e6796b67SKirk McKusick     int resid, flags;
572a4f67738SDoug Rabson     elf_file_t ef;
573a4f67738SDoug Rabson     linker_file_t lf;
574ca65d5c7SPeter Wemm     Elf_Shdr *shdr;
575ca65d5c7SPeter Wemm     int symtabindex;
576ca65d5c7SPeter Wemm     int symstrindex;
577ca65d5c7SPeter Wemm     int symcnt;
578ca65d5c7SPeter Wemm     int strcnt;
57962d615d5SJohn Baldwin     int vfslocked;
580ca65d5c7SPeter Wemm 
581ca65d5c7SPeter Wemm     shdr = NULL;
582ca65d5c7SPeter Wemm     lf = NULL;
583a4f67738SDoug Rabson 
58462d615d5SJohn Baldwin     NDINIT(&nd, LOOKUP, FOLLOW | MPSAFE, UIO_SYSSPACE, filename, td);
585e6796b67SKirk McKusick     flags = FREAD;
5869e223287SKonstantin Belousov     error = vn_open(&nd, &flags, 0, NULL);
587a4f67738SDoug Rabson     if (error)
588a4f67738SDoug Rabson 	return error;
58962d615d5SJohn Baldwin     vfslocked = NDHASGIANT(&nd);
590762e6b85SEivind Eklund     NDFREE(&nd, NDF_ONLY_PNBUF);
5911f49b573SRuslan Ermilov     if (nd.ni_vp->v_type != VREG) {
5921f49b573SRuslan Ermilov 	error = ENOEXEC;
5931f49b573SRuslan Ermilov 	firstpage = NULL;
5941f49b573SRuslan Ermilov 	goto out;
5951f49b573SRuslan Ermilov     }
596a3df768bSRobert Watson #ifdef MAC
59730d239bcSRobert Watson     error = mac_kld_check_load(curthread->td_ucred, nd.ni_vp);
598a3df768bSRobert Watson     if (error) {
599a3df768bSRobert Watson 	firstpage = NULL;
600a3df768bSRobert Watson 	goto out;
601a3df768bSRobert Watson     }
602a3df768bSRobert Watson #endif
603a4f67738SDoug Rabson 
604a4f67738SDoug Rabson     /*
605fe3db7c7SDoug Rabson      * Read the elf header from the file.
606a4f67738SDoug Rabson      */
607a163d034SWarner Losh     firstpage = malloc(PAGE_SIZE, M_LINKER, M_WAITOK);
608caab6e90SPeter Wemm     if (firstpage == NULL) {
609caab6e90SPeter Wemm 	error = ENOMEM;
610caab6e90SPeter Wemm 	goto out;
611caab6e90SPeter Wemm     }
612caab6e90SPeter Wemm     hdr = (Elf_Ehdr *)firstpage;
613caab6e90SPeter Wemm     error = vn_rdwr(UIO_READ, nd.ni_vp, firstpage, PAGE_SIZE, 0,
6149ca43589SRobert Watson 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
6159ca43589SRobert Watson 		    &resid, td);
616caab6e90SPeter Wemm     nbytes = PAGE_SIZE - resid;
617a4f67738SDoug Rabson     if (error)
618a4f67738SDoug Rabson 	goto out;
619a4f67738SDoug Rabson 
620caab6e90SPeter Wemm     if (!IS_ELF(*hdr)) {
621fe3db7c7SDoug Rabson 	error = ENOEXEC;
622a4f67738SDoug Rabson 	goto out;
623fe3db7c7SDoug Rabson     }
624fe3db7c7SDoug Rabson 
625caab6e90SPeter Wemm     if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS
626caab6e90SPeter Wemm       || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) {
627fe3db7c7SDoug Rabson 	link_elf_error("Unsupported file layout");
628fe3db7c7SDoug Rabson 	error = ENOEXEC;
629fe3db7c7SDoug Rabson 	goto out;
630fe3db7c7SDoug Rabson     }
631caab6e90SPeter Wemm     if (hdr->e_ident[EI_VERSION] != EV_CURRENT
632caab6e90SPeter Wemm       || hdr->e_version != EV_CURRENT) {
633fe3db7c7SDoug Rabson 	link_elf_error("Unsupported file version");
634fe3db7c7SDoug Rabson 	error = ENOEXEC;
635fe3db7c7SDoug Rabson 	goto out;
636fe3db7c7SDoug Rabson     }
637caab6e90SPeter Wemm     if (hdr->e_type != ET_EXEC && hdr->e_type != ET_DYN) {
638fe3db7c7SDoug Rabson 	link_elf_error("Unsupported file type");
639fe3db7c7SDoug Rabson 	error = ENOEXEC;
640fe3db7c7SDoug Rabson 	goto out;
641fe3db7c7SDoug Rabson     }
642caab6e90SPeter Wemm     if (hdr->e_machine != ELF_TARG_MACH) {
643fe3db7c7SDoug Rabson 	link_elf_error("Unsupported machine");
644fe3db7c7SDoug Rabson 	error = ENOEXEC;
645fe3db7c7SDoug Rabson 	goto out;
646fe3db7c7SDoug Rabson     }
647a4f67738SDoug Rabson 
648a4f67738SDoug Rabson     /*
649fe3db7c7SDoug Rabson      * We rely on the program header being in the first page.  This is
650fe3db7c7SDoug Rabson      * not strictly required by the ABI specification, but it seems to
651fe3db7c7SDoug Rabson      * always true in practice.  And, it simplifies things considerably.
652a4f67738SDoug Rabson      */
653caab6e90SPeter Wemm     if (!((hdr->e_phentsize == sizeof(Elf_Phdr)) &&
654caab6e90SPeter Wemm 	  (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= PAGE_SIZE) &&
655caab6e90SPeter Wemm 	  (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= nbytes)))
656fe3db7c7SDoug Rabson 	link_elf_error("Unreadable program headers");
657fe3db7c7SDoug Rabson 
658fe3db7c7SDoug Rabson     /*
659fe3db7c7SDoug Rabson      * Scan the program header entries, and save key information.
660fe3db7c7SDoug Rabson      *
661fe3db7c7SDoug Rabson      * We rely on there being exactly two load segments, text and data,
662fe3db7c7SDoug Rabson      * in that order.
663fe3db7c7SDoug Rabson      */
664caab6e90SPeter Wemm     phdr = (Elf_Phdr *) (firstpage + hdr->e_phoff);
665caab6e90SPeter Wemm     phlimit = phdr + hdr->e_phnum;
666fe3db7c7SDoug Rabson     nsegs = 0;
667fe3db7c7SDoug Rabson     phdyn = NULL;
668fe3db7c7SDoug Rabson     phphdr = NULL;
669fe3db7c7SDoug Rabson     while (phdr < phlimit) {
670fe3db7c7SDoug Rabson 	switch (phdr->p_type) {
671fe3db7c7SDoug Rabson 
672fe3db7c7SDoug Rabson 	case PT_LOAD:
6731a330eb0SPeter Wemm 	    if (nsegs == MAXSEGS) {
674fe3db7c7SDoug Rabson 		link_elf_error("Too many sections");
675fe3db7c7SDoug Rabson 		error = ENOEXEC;
676fe3db7c7SDoug Rabson 		goto out;
677fe3db7c7SDoug Rabson 	    }
67887b1831fSPoul-Henning Kamp 	    /*
67987b1831fSPoul-Henning Kamp 	     * XXX: We just trust they come in right order ??
68087b1831fSPoul-Henning Kamp 	     */
681fe3db7c7SDoug Rabson 	    segs[nsegs] = phdr;
682fe3db7c7SDoug Rabson 	    ++nsegs;
683fe3db7c7SDoug Rabson 	    break;
684fe3db7c7SDoug Rabson 
685fe3db7c7SDoug Rabson 	case PT_PHDR:
686fe3db7c7SDoug Rabson 	    phphdr = phdr;
687fe3db7c7SDoug Rabson 	    break;
688fe3db7c7SDoug Rabson 
689fe3db7c7SDoug Rabson 	case PT_DYNAMIC:
690fe3db7c7SDoug Rabson 	    phdyn = phdr;
691fe3db7c7SDoug Rabson 	    break;
6929548091bSBoris Popov 
6939548091bSBoris Popov 	case PT_INTERP:
6949548091bSBoris Popov 	    link_elf_error("Unsupported file type");
6959548091bSBoris Popov 	    error = ENOEXEC;
6969548091bSBoris Popov 	    goto out;
697fe3db7c7SDoug Rabson 	}
698fe3db7c7SDoug Rabson 
699fe3db7c7SDoug Rabson 	++phdr;
700fe3db7c7SDoug Rabson     }
701fe3db7c7SDoug Rabson     if (phdyn == NULL) {
702fe3db7c7SDoug Rabson 	link_elf_error("Object is not dynamically-linked");
703fe3db7c7SDoug Rabson 	error = ENOEXEC;
704fe3db7c7SDoug Rabson 	goto out;
705fe3db7c7SDoug Rabson     }
7061a330eb0SPeter Wemm     if (nsegs == 0) {
7071a330eb0SPeter Wemm 	link_elf_error("No sections");
70887b1831fSPoul-Henning Kamp 	error = ENOEXEC;
70987b1831fSPoul-Henning Kamp 	goto out;
71087b1831fSPoul-Henning Kamp     }
711fe3db7c7SDoug Rabson 
712fe3db7c7SDoug Rabson     /*
713fe3db7c7SDoug Rabson      * Allocate the entire address space of the object, to stake out our
714fe3db7c7SDoug Rabson      * contiguous region, and to establish the base address for relocation.
715fe3db7c7SDoug Rabson      */
716fe3db7c7SDoug Rabson     base_offset = trunc_page(segs[0]->p_offset);
717fe3db7c7SDoug Rabson     base_vaddr = trunc_page(segs[0]->p_vaddr);
7181a330eb0SPeter Wemm     base_vlimit = round_page(segs[nsegs - 1]->p_vaddr +
7191a330eb0SPeter Wemm 	segs[nsegs - 1]->p_memsz);
720fe3db7c7SDoug Rabson     mapsize = base_vlimit - base_vaddr;
721fe3db7c7SDoug Rabson 
722326e27d8SDoug Rabson     lf = linker_make_file(filename, &link_elf_class);
723326e27d8SDoug Rabson     if (!lf) {
724326e27d8SDoug Rabson 	error = ENOMEM;
725326e27d8SDoug Rabson 	goto out;
726326e27d8SDoug Rabson     }
727326e27d8SDoug Rabson 
728326e27d8SDoug Rabson     ef = (elf_file_t) lf;
729fe3db7c7SDoug Rabson #ifdef SPARSE_MAPPING
730fe3db7c7SDoug Rabson     ef->object = vm_object_allocate(OBJT_DEFAULT, mapsize >> PAGE_SHIFT);
731fe3db7c7SDoug Rabson     if (ef->object == NULL) {
732fe3db7c7SDoug Rabson 	error = ENOMEM;
733fe3db7c7SDoug Rabson 	goto out;
734fe3db7c7SDoug Rabson     }
735fe3db7c7SDoug Rabson     ef->address = (caddr_t) vm_map_min(kernel_map);
736fe3db7c7SDoug Rabson     error = vm_map_find(kernel_map, ef->object, 0,
737fe3db7c7SDoug Rabson 			(vm_offset_t *) &ef->address,
738fe3db7c7SDoug Rabson 			mapsize, 1,
739fe3db7c7SDoug Rabson 			VM_PROT_ALL, VM_PROT_ALL, 0);
740fe3db7c7SDoug Rabson     if (error) {
741fe3db7c7SDoug Rabson 	vm_object_deallocate(ef->object);
742326e27d8SDoug Rabson 	ef->object = 0;
743fe3db7c7SDoug Rabson 	goto out;
744fe3db7c7SDoug Rabson     }
745fe3db7c7SDoug Rabson #else
746a163d034SWarner Losh     ef->address = malloc(mapsize, M_LINKER, M_WAITOK);
747326e27d8SDoug Rabson     if (!ef->address) {
748326e27d8SDoug Rabson 	error = ENOMEM;
749326e27d8SDoug Rabson 	goto out;
750326e27d8SDoug Rabson     }
751fe3db7c7SDoug Rabson #endif
752fe3db7c7SDoug Rabson     mapbase = ef->address;
753a4f67738SDoug Rabson 
754a4f67738SDoug Rabson     /*
755a4f67738SDoug Rabson      * Read the text and data sections and zero the bss.
756a4f67738SDoug Rabson      */
7571a330eb0SPeter Wemm     for (i = 0; i < nsegs; i++) {
758fe3db7c7SDoug Rabson 	caddr_t segbase = mapbase + segs[i]->p_vaddr - base_vaddr;
759fe3db7c7SDoug Rabson 	error = vn_rdwr(UIO_READ, nd.ni_vp,
760fe3db7c7SDoug Rabson 			segbase, segs[i]->p_filesz, segs[i]->p_offset,
7619ca43589SRobert Watson 			UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
7629ca43589SRobert Watson 			&resid, td);
763fe3db7c7SDoug Rabson 	if (error) {
764a4f67738SDoug Rabson 	    goto out;
765a4f67738SDoug Rabson 	}
766fe3db7c7SDoug Rabson 	bzero(segbase + segs[i]->p_filesz,
767fe3db7c7SDoug Rabson 	      segs[i]->p_memsz - segs[i]->p_filesz);
76841673965SAlan Cox 
76941673965SAlan Cox #ifdef SPARSE_MAPPING
77041673965SAlan Cox 	/*
77141673965SAlan Cox 	 * Wire down the pages
77241673965SAlan Cox 	 */
7735d3043ceSAlan Cox 	error = vm_map_wire(kernel_map,
77441673965SAlan Cox 		    (vm_offset_t) segbase,
77541673965SAlan Cox 		    (vm_offset_t) segbase + segs[i]->p_memsz,
77641673965SAlan Cox 		    VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES);
7775d3043ceSAlan Cox 	if (error != KERN_SUCCESS) {
7785d3043ceSAlan Cox 	    error = ENOMEM;
7795d3043ceSAlan Cox 	    goto out;
7805d3043ceSAlan Cox 	}
78141673965SAlan Cox #endif
782fe3db7c7SDoug Rabson     }
783fe3db7c7SDoug Rabson 
784f99502a4SBrian Feldman #ifdef GPROF
785f99502a4SBrian Feldman     /* Update profiling information with the new text segment. */
78670f37788SJohn Baldwin     mtx_lock(&Giant);
787f99502a4SBrian Feldman     kmupetext((uintfptr_t)(mapbase + segs[0]->p_vaddr - base_vaddr +
788f99502a4SBrian Feldman 	segs[0]->p_memsz));
78970f37788SJohn Baldwin     mtx_unlock(&Giant);
790f99502a4SBrian Feldman #endif
791f99502a4SBrian Feldman 
792326e27d8SDoug Rabson     ef->dynamic = (Elf_Dyn *) (mapbase + phdyn->p_vaddr - base_vaddr);
793a4f67738SDoug Rabson 
794a4f67738SDoug Rabson     lf->address = ef->address;
795fe3db7c7SDoug Rabson     lf->size = mapsize;
796a4f67738SDoug Rabson 
797326e27d8SDoug Rabson     error = parse_dynamic(ef);
798ca65d5c7SPeter Wemm     if (error)
799de78ca7eSPeter Wemm 	goto out;
8007251b4bfSJake Burkholder     link_elf_reloc_local(lf);
8017251b4bfSJake Burkholder 
8027b9716baSIan Dowse     error = linker_load_dependencies(lf);
803ca65d5c7SPeter Wemm     if (error)
804de78ca7eSPeter Wemm 	goto out;
80554823af2SPeter Wemm #if 0	/* this will be more trouble than it's worth for now */
80654823af2SPeter Wemm     for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) {
80754823af2SPeter Wemm 	if (dp->d_tag != DT_NEEDED)
80854823af2SPeter Wemm 	    continue;
80954823af2SPeter Wemm 	modname = ef->strtab + dp->d_un.d_val;
81054823af2SPeter Wemm 	error = linker_load_module(modname, lf);
81154823af2SPeter Wemm 	if (error)
81254823af2SPeter Wemm 	    goto out;
81354823af2SPeter Wemm     }
81454823af2SPeter Wemm #endif
815326e27d8SDoug Rabson     error = relocate_file(ef);
816ca65d5c7SPeter Wemm     if (error)
817ca65d5c7SPeter Wemm 	goto out;
818ca65d5c7SPeter Wemm 
819ca65d5c7SPeter Wemm     /* Try and load the symbol table if it's present.  (you can strip it!) */
820caab6e90SPeter Wemm     nbytes = hdr->e_shnum * hdr->e_shentsize;
821caab6e90SPeter Wemm     if (nbytes == 0 || hdr->e_shoff == 0)
822ca65d5c7SPeter Wemm 	goto nosyms;
823a163d034SWarner Losh     shdr = malloc(nbytes, M_LINKER, M_WAITOK | M_ZERO);
824ca65d5c7SPeter Wemm     if (shdr == NULL) {
825ca65d5c7SPeter Wemm 	error = ENOMEM;
826a4f67738SDoug Rabson 	goto out;
827a4f67738SDoug Rabson     }
828ca65d5c7SPeter Wemm     error = vn_rdwr(UIO_READ, nd.ni_vp,
829caab6e90SPeter Wemm 		    (caddr_t)shdr, nbytes, hdr->e_shoff,
8309ca43589SRobert Watson 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
8319ca43589SRobert Watson 		    &resid, td);
832ca65d5c7SPeter Wemm     if (error)
833ca65d5c7SPeter Wemm 	goto out;
834ca65d5c7SPeter Wemm     symtabindex = -1;
835ca65d5c7SPeter Wemm     symstrindex = -1;
836caab6e90SPeter Wemm     for (i = 0; i < hdr->e_shnum; i++) {
837ca65d5c7SPeter Wemm 	if (shdr[i].sh_type == SHT_SYMTAB) {
838ca65d5c7SPeter Wemm 	    symtabindex = i;
839ca65d5c7SPeter Wemm 	    symstrindex = shdr[i].sh_link;
840ca65d5c7SPeter Wemm 	}
841ca65d5c7SPeter Wemm     }
842ca65d5c7SPeter Wemm     if (symtabindex < 0 || symstrindex < 0)
843ca65d5c7SPeter Wemm 	goto nosyms;
844ca65d5c7SPeter Wemm 
845ca65d5c7SPeter Wemm     symcnt = shdr[symtabindex].sh_size;
846a163d034SWarner Losh     ef->symbase = malloc(symcnt, M_LINKER, M_WAITOK);
847ca65d5c7SPeter Wemm     strcnt = shdr[symstrindex].sh_size;
848a163d034SWarner Losh     ef->strbase = malloc(strcnt, M_LINKER, M_WAITOK);
849ca65d5c7SPeter Wemm 
850ca65d5c7SPeter Wemm     if (ef->symbase == NULL || ef->strbase == NULL) {
851ca65d5c7SPeter Wemm 	error = ENOMEM;
852ca65d5c7SPeter Wemm 	goto out;
853ca65d5c7SPeter Wemm     }
854ca65d5c7SPeter Wemm     error = vn_rdwr(UIO_READ, nd.ni_vp,
855ca65d5c7SPeter Wemm 		    ef->symbase, symcnt, shdr[symtabindex].sh_offset,
8569ca43589SRobert Watson 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
8579ca43589SRobert Watson 		    &resid, td);
858ca65d5c7SPeter Wemm     if (error)
859ca65d5c7SPeter Wemm 	goto out;
860ca65d5c7SPeter Wemm     error = vn_rdwr(UIO_READ, nd.ni_vp,
861ca65d5c7SPeter Wemm 		    ef->strbase, strcnt, shdr[symstrindex].sh_offset,
8629ca43589SRobert Watson 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
8639ca43589SRobert Watson 		    &resid, td);
864ca65d5c7SPeter Wemm     if (error)
865ca65d5c7SPeter Wemm 	goto out;
866ca65d5c7SPeter Wemm 
867ca65d5c7SPeter Wemm     ef->ddbsymcnt = symcnt / sizeof(Elf_Sym);
868ca65d5c7SPeter Wemm     ef->ddbsymtab = (const Elf_Sym *)ef->symbase;
869ca65d5c7SPeter Wemm     ef->ddbstrcnt = strcnt;
870ca65d5c7SPeter Wemm     ef->ddbstrtab = ef->strbase;
871ca65d5c7SPeter Wemm 
872c143d6c2SMarcel Moolenaar     error = link_elf_link_common_finish(lf);
873c143d6c2SMarcel Moolenaar     if (error)
874c143d6c2SMarcel Moolenaar 	goto out;
875326e27d8SDoug Rabson 
876ca65d5c7SPeter Wemm nosyms:
877a4f67738SDoug Rabson 
878a4f67738SDoug Rabson     *result = lf;
879a4f67738SDoug Rabson 
880a4f67738SDoug Rabson out:
881ca65d5c7SPeter Wemm     if (error && lf)
88265a311fcSPoul-Henning Kamp 	linker_file_unload(lf, LINKER_UNLOAD_FORCE);
883ca65d5c7SPeter Wemm     if (shdr)
884ca65d5c7SPeter Wemm 	free(shdr, M_LINKER);
885caab6e90SPeter Wemm     if (firstpage)
886caab6e90SPeter Wemm 	free(firstpage, M_LINKER);
88722db15c0SAttilio Rao     VOP_UNLOCK(nd.ni_vp, 0);
888a854ed98SJohn Baldwin     vn_close(nd.ni_vp, FREAD, td->td_ucred, td);
88962d615d5SJohn Baldwin     VFS_UNLOCK_GIANT(vfslocked);
890a4f67738SDoug Rabson 
891a4f67738SDoug Rabson     return error;
892a4f67738SDoug Rabson }
893a4f67738SDoug Rabson 
894a4f67738SDoug Rabson static void
895de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file)
896a4f67738SDoug Rabson {
897326e27d8SDoug Rabson     elf_file_t ef = (elf_file_t) file;
898a4f67738SDoug Rabson 
899ecb01c64SMarcel Moolenaar #ifdef GDB
900326e27d8SDoug Rabson     if (ef->gdb.l_ld) {
901326e27d8SDoug Rabson 	GDB_STATE(RT_DELETE);
902ed6c3888SPeter Wemm 	free((void *)(uintptr_t)ef->gdb.l_name, M_LINKER);
903326e27d8SDoug Rabson 	link_elf_delete_gdb(&ef->gdb);
904326e27d8SDoug Rabson 	GDB_STATE(RT_CONSISTENT);
905326e27d8SDoug Rabson     }
906326e27d8SDoug Rabson #endif
907326e27d8SDoug Rabson 
9081aeb23cdSMarcel Moolenaar     /* Notify MD code that a module is being unloaded. */
9091aeb23cdSMarcel Moolenaar     elf_cpu_unload_file(file);
9101aeb23cdSMarcel Moolenaar 
911326e27d8SDoug Rabson     if (ef->preloaded) {
91254823af2SPeter Wemm 	link_elf_unload_preload(file);
913326e27d8SDoug Rabson 	return;
914326e27d8SDoug Rabson     }
915c143d6c2SMarcel Moolenaar 
916fe3db7c7SDoug Rabson #ifdef SPARSE_MAPPING
917fe3db7c7SDoug Rabson     if (ef->object) {
918fe3db7c7SDoug Rabson 	vm_map_remove(kernel_map, (vm_offset_t) ef->address,
919fe3db7c7SDoug Rabson 		      (vm_offset_t) ef->address
920fe3db7c7SDoug Rabson 		      + (ef->object->size << PAGE_SHIFT));
921fe3db7c7SDoug Rabson     }
922fe3db7c7SDoug Rabson #else
923ca65d5c7SPeter Wemm     if (ef->address)
924a4f67738SDoug Rabson 	free(ef->address, M_LINKER);
925fe3db7c7SDoug Rabson #endif
926ca65d5c7SPeter Wemm     if (ef->symbase)
927ca65d5c7SPeter Wemm 	free(ef->symbase, M_LINKER);
928ca65d5c7SPeter Wemm     if (ef->strbase)
929ca65d5c7SPeter Wemm 	free(ef->strbase, M_LINKER);
930a2024a3eSJohn Birrell     if (ef->ctftab)
931a2024a3eSJohn Birrell 	free(ef->ctftab, M_LINKER);
932a2024a3eSJohn Birrell     if (ef->ctfoff)
933a2024a3eSJohn Birrell 	free(ef->ctfoff, M_LINKER);
934a2024a3eSJohn Birrell     if (ef->typoff)
935a2024a3eSJohn Birrell 	free(ef->typoff, M_LINKER);
936a4f67738SDoug Rabson }
937a4f67738SDoug Rabson 
938de78ca7eSPeter Wemm static void
93954823af2SPeter Wemm link_elf_unload_preload(linker_file_t file)
940de78ca7eSPeter Wemm {
941de78ca7eSPeter Wemm     if (file->filename)
942de78ca7eSPeter Wemm 	preload_delete_name(file->filename);
943de78ca7eSPeter Wemm }
944de78ca7eSPeter Wemm 
945fe3db7c7SDoug Rabson static const char *
946757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info)
947a4f67738SDoug Rabson {
948fe3db7c7SDoug Rabson     const Elf_Sym *ref;
949a4f67738SDoug Rabson 
950aa855a59SPeter Wemm     if (ELF_R_SYM(r_info)) {
951aa855a59SPeter Wemm 	ref = ef->symtab + ELF_R_SYM(r_info);
952434bf653SPeter Wemm 	return ef->strtab + ref->st_name;
953fe3db7c7SDoug Rabson     } else
954fe3db7c7SDoug Rabson 	return NULL;
955a4f67738SDoug Rabson }
956a4f67738SDoug Rabson 
957a4f67738SDoug Rabson static int
958326e27d8SDoug Rabson relocate_file(elf_file_t ef)
959a4f67738SDoug Rabson {
960fe3db7c7SDoug Rabson     const Elf_Rel *rellim;
961fe3db7c7SDoug Rabson     const Elf_Rel *rel;
962fe3db7c7SDoug Rabson     const Elf_Rela *relalim;
963fe3db7c7SDoug Rabson     const Elf_Rela *rela;
964aa855a59SPeter Wemm     const char *symname;
965a4f67738SDoug Rabson 
966fe3db7c7SDoug Rabson     /* Perform relocations without addend if there are any: */
967aa855a59SPeter Wemm     rel = ef->rel;
968aa855a59SPeter Wemm     if (rel) {
969cacd1f6aSBruce Evans 	rellim = (const Elf_Rel *)((const char *)ef->rel + ef->relsize);
970aa855a59SPeter Wemm 	while (rel < rellim) {
971e8855d4fSPeter Wemm 	    if (elf_reloc(&ef->lf, (Elf_Addr)ef->address, rel, ELF_RELOC_REL,
972e8855d4fSPeter Wemm 			  elf_lookup)) {
973aa855a59SPeter Wemm 		symname = symbol_name(ef, rel->r_info);
974ef83592dSPeter Wemm 		printf("link_elf: symbol %s undefined\n", symname);
975a4f67738SDoug Rabson 		return ENOENT;
976a13ddfb6SPeter Wemm 	    }
977aa855a59SPeter Wemm 	    rel++;
978aa855a59SPeter Wemm 	}
979a4f67738SDoug Rabson     }
980a4f67738SDoug Rabson 
981fe3db7c7SDoug Rabson     /* Perform relocations with addend if there are any: */
982aa855a59SPeter Wemm     rela = ef->rela;
983aa855a59SPeter Wemm     if (rela) {
984cacd1f6aSBruce Evans 	relalim = (const Elf_Rela *)((const char *)ef->rela + ef->relasize);
985aa855a59SPeter Wemm 	while (rela < relalim) {
986e8855d4fSPeter Wemm 	    if (elf_reloc(&ef->lf, (Elf_Addr)ef->address, rela, ELF_RELOC_RELA,
987e8855d4fSPeter Wemm 			  elf_lookup)) {
988aa855a59SPeter Wemm 		symname = symbol_name(ef, rela->r_info);
989ef83592dSPeter Wemm 		printf("link_elf: symbol %s undefined\n", symname);
990fe3db7c7SDoug Rabson 		return ENOENT;
991a13ddfb6SPeter Wemm 	    }
992aa855a59SPeter Wemm 	    rela++;
993aa855a59SPeter Wemm 	}
994a4f67738SDoug Rabson     }
995a4f67738SDoug Rabson 
996fe3db7c7SDoug Rabson     /* Perform PLT relocations without addend if there are any: */
997aa855a59SPeter Wemm     rel = ef->pltrel;
998aa855a59SPeter Wemm     if (rel) {
999cacd1f6aSBruce Evans 	rellim = (const Elf_Rel *)((const char *)ef->pltrel + ef->pltrelsize);
1000aa855a59SPeter Wemm 	while (rel < rellim) {
1001e8855d4fSPeter Wemm 	    if (elf_reloc(&ef->lf, (Elf_Addr)ef->address, rel, ELF_RELOC_REL,
1002e8855d4fSPeter Wemm 			  elf_lookup)) {
1003aa855a59SPeter Wemm 		symname = symbol_name(ef, rel->r_info);
1004ef83592dSPeter Wemm 		printf("link_elf: symbol %s undefined\n", symname);
1005fe3db7c7SDoug Rabson 		return ENOENT;
1006a13ddfb6SPeter Wemm 	    }
1007aa855a59SPeter Wemm 	    rel++;
1008aa855a59SPeter Wemm 	}
1009a4f67738SDoug Rabson     }
1010a4f67738SDoug Rabson 
1011fe3db7c7SDoug Rabson     /* Perform relocations with addend if there are any: */
1012aa855a59SPeter Wemm     rela = ef->pltrela;
1013aa855a59SPeter Wemm     if (rela) {
1014cacd1f6aSBruce Evans 	relalim = (const Elf_Rela *)((const char *)ef->pltrela + ef->pltrelasize);
1015aa855a59SPeter Wemm 	while (rela < relalim) {
1016e8855d4fSPeter Wemm 	    if (elf_reloc(&ef->lf, (Elf_Addr)ef->address, rela, ELF_RELOC_RELA,
1017e8855d4fSPeter Wemm 			  elf_lookup)) {
1018aa855a59SPeter Wemm 		symname = symbol_name(ef, rela->r_info);
1019ef83592dSPeter Wemm 		printf("link_elf: symbol %s undefined\n", symname);
1020fe3db7c7SDoug Rabson 		return ENOENT;
1021a13ddfb6SPeter Wemm 	    }
1022aa855a59SPeter Wemm 	    rela++;
1023aa855a59SPeter Wemm 	}
1024a4f67738SDoug Rabson     }
1025a4f67738SDoug Rabson 
1026a4f67738SDoug Rabson     return 0;
1027a4f67738SDoug Rabson }
1028a4f67738SDoug Rabson 
1029fe3db7c7SDoug Rabson /*
1030fe3db7c7SDoug Rabson  * Hash function for symbol table lookup.  Don't even think about changing
1031fe3db7c7SDoug Rabson  * this.  It is specified by the System V ABI.
1032fe3db7c7SDoug Rabson  */
1033fe3db7c7SDoug Rabson static unsigned long
1034fe3db7c7SDoug Rabson elf_hash(const char *name)
1035a4f67738SDoug Rabson {
1036fe3db7c7SDoug Rabson     const unsigned char *p = (const unsigned char *) name;
1037fe3db7c7SDoug Rabson     unsigned long h = 0;
1038fe3db7c7SDoug Rabson     unsigned long g;
1039a4f67738SDoug Rabson 
1040fe3db7c7SDoug Rabson     while (*p != '\0') {
1041fe3db7c7SDoug Rabson 	h = (h << 4) + *p++;
1042fe3db7c7SDoug Rabson 	if ((g = h & 0xf0000000) != 0)
1043fe3db7c7SDoug Rabson 	    h ^= g >> 24;
1044fe3db7c7SDoug Rabson 	h &= ~g;
1045a4f67738SDoug Rabson     }
1046fe3db7c7SDoug Rabson     return h;
1047fe3db7c7SDoug Rabson }
1048a4f67738SDoug Rabson 
104937c84183SPoul-Henning Kamp static int
1050d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char* name, c_linker_sym_t* sym)
1051a4f67738SDoug Rabson {
1052326e27d8SDoug Rabson     elf_file_t ef = (elf_file_t) lf;
1053fe3db7c7SDoug Rabson     unsigned long symnum;
10542d636ab0SPeter Wemm     const Elf_Sym* symp;
10552d636ab0SPeter Wemm     const char *strp;
1056fe3db7c7SDoug Rabson     unsigned long hash;
1057a4f67738SDoug Rabson     int i;
1058a4f67738SDoug Rabson 
1059f6494f2eSJohn Baldwin     /* If we don't have a hash, bail. */
1060f6494f2eSJohn Baldwin     if (ef->buckets == NULL || ef->nbuckets == 0) {
1061f6494f2eSJohn Baldwin 	printf("link_elf_lookup_symbol: missing symbol hash table\n");
1062f6494f2eSJohn Baldwin 	return ENOENT;
1063f6494f2eSJohn Baldwin     }
1064f6494f2eSJohn Baldwin 
10652d636ab0SPeter Wemm     /* First, search hashed global symbols */
1066fe3db7c7SDoug Rabson     hash = elf_hash(name);
1067fe3db7c7SDoug Rabson     symnum = ef->buckets[hash % ef->nbuckets];
1068fe3db7c7SDoug Rabson 
1069fe3db7c7SDoug Rabson     while (symnum != STN_UNDEF) {
1070fe3db7c7SDoug Rabson 	if (symnum >= ef->nchains) {
1071fe3db7c7SDoug Rabson 	    printf("link_elf_lookup_symbol: corrupt symbol table\n");
1072fe3db7c7SDoug Rabson 	    return ENOENT;
1073a4f67738SDoug Rabson 	}
1074fe3db7c7SDoug Rabson 
1075fe3db7c7SDoug Rabson 	symp = ef->symtab + symnum;
1076fe3db7c7SDoug Rabson 	if (symp->st_name == 0) {
1077fe3db7c7SDoug Rabson 	    printf("link_elf_lookup_symbol: corrupt symbol table\n");
1078fe3db7c7SDoug Rabson 	    return ENOENT;
1079fe3db7c7SDoug Rabson 	}
1080fe3db7c7SDoug Rabson 
1081fe3db7c7SDoug Rabson 	strp = ef->strtab + symp->st_name;
1082fe3db7c7SDoug Rabson 
1083fe3db7c7SDoug Rabson 	if (strcmp(name, strp) == 0) {
1084fe3db7c7SDoug Rabson 	    if (symp->st_shndx != SHN_UNDEF ||
1085fe3db7c7SDoug Rabson 		(symp->st_value != 0 &&
1086fe3db7c7SDoug Rabson 		 ELF_ST_TYPE(symp->st_info) == STT_FUNC)) {
1087d254af07SMatthew Dillon 		*sym = (c_linker_sym_t) symp;
1088fe3db7c7SDoug Rabson 		return 0;
1089fe3db7c7SDoug Rabson 	    } else
1090fe3db7c7SDoug Rabson 		return ENOENT;
1091fe3db7c7SDoug Rabson 	}
1092fe3db7c7SDoug Rabson 
1093fe3db7c7SDoug Rabson 	symnum = ef->chains[symnum];
1094a4f67738SDoug Rabson     }
1095a4f67738SDoug Rabson 
10962d636ab0SPeter Wemm     /* If we have not found it, look at the full table (if loaded) */
10972d636ab0SPeter Wemm     if (ef->symtab == ef->ddbsymtab)
10982d636ab0SPeter Wemm 	return ENOENT;
10992d636ab0SPeter Wemm 
11002d636ab0SPeter Wemm     /* Exhaustive search */
11012d636ab0SPeter Wemm     for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
11022d636ab0SPeter Wemm 	strp = ef->ddbstrtab + symp->st_name;
11032d636ab0SPeter Wemm 	if (strcmp(name, strp) == 0) {
11042d636ab0SPeter Wemm 	    if (symp->st_shndx != SHN_UNDEF ||
11052d636ab0SPeter Wemm 		(symp->st_value != 0 &&
11062d636ab0SPeter Wemm 		 ELF_ST_TYPE(symp->st_info) == STT_FUNC)) {
1107d254af07SMatthew Dillon 		*sym = (c_linker_sym_t) symp;
11082d636ab0SPeter Wemm 		return 0;
11092d636ab0SPeter Wemm 	    } else
11102d636ab0SPeter Wemm 		return ENOENT;
11112d636ab0SPeter Wemm 	}
11122d636ab0SPeter Wemm     }
11132d636ab0SPeter Wemm 
1114a4f67738SDoug Rabson     return ENOENT;
1115a4f67738SDoug Rabson }
1116a4f67738SDoug Rabson 
1117de78ca7eSPeter Wemm static int
1118fe08c21aSMatthew Dillon link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, linker_symval_t* symval)
1119a4f67738SDoug Rabson {
1120326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) lf;
11218aef1712SMatthew Dillon 	const Elf_Sym* es = (const Elf_Sym*) sym;
1122a4f67738SDoug Rabson 
11235cf87418SMarcel Moolenaar 	if (es >= ef->symtab && es < (ef->symtab + ef->nchains)) {
1124a4f67738SDoug Rabson 	    symval->name = ef->strtab + es->st_name;
1125fe3db7c7SDoug Rabson 	    symval->value = (caddr_t) ef->address + es->st_value;
1126a4f67738SDoug Rabson 	    symval->size = es->st_size;
1127de78ca7eSPeter Wemm 	    return 0;
1128a4f67738SDoug Rabson 	}
11292d636ab0SPeter Wemm 	if (ef->symtab == ef->ddbsymtab)
11302d636ab0SPeter Wemm 	    return ENOENT;
11315cf87418SMarcel Moolenaar 	if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) {
11322d636ab0SPeter Wemm 	    symval->name = ef->ddbstrtab + es->st_name;
11332d636ab0SPeter Wemm 	    symval->value = (caddr_t) ef->address + es->st_value;
11342d636ab0SPeter Wemm 	    symval->size = es->st_size;
11352d636ab0SPeter Wemm 	    return 0;
11362d636ab0SPeter Wemm 	}
11372d636ab0SPeter Wemm 	return ENOENT;
11382d636ab0SPeter Wemm }
1139a4f67738SDoug Rabson 
1140a4f67738SDoug Rabson static int
1141a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value,
1142d254af07SMatthew Dillon 		       c_linker_sym_t* sym, long* diffp)
1143a4f67738SDoug Rabson {
1144326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) lf;
1145586453feSBruce Evans 	u_long off = (uintptr_t) (void *) value;
1146a4f67738SDoug Rabson 	u_long diff = off;
1147586453feSBruce Evans 	u_long st_value;
1148fe3db7c7SDoug Rabson 	const Elf_Sym* es;
1149fe3db7c7SDoug Rabson 	const Elf_Sym* best = 0;
1150a4f67738SDoug Rabson 	int i;
1151a4f67738SDoug Rabson 
11522d636ab0SPeter Wemm 	for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) {
1153a4f67738SDoug Rabson 		if (es->st_name == 0)
1154a4f67738SDoug Rabson 			continue;
1155586453feSBruce Evans 		st_value = es->st_value + (uintptr_t) (void *) ef->address;
1156b5abfb70SPeter Wemm 		if (off >= st_value) {
1157b5abfb70SPeter Wemm 			if (off - st_value < diff) {
1158b5abfb70SPeter Wemm 				diff = off - st_value;
1159a4f67738SDoug Rabson 				best = es;
1160a4f67738SDoug Rabson 				if (diff == 0)
1161a4f67738SDoug Rabson 					break;
1162b5abfb70SPeter Wemm 			} else if (off - st_value == diff) {
1163a4f67738SDoug Rabson 				best = es;
1164a4f67738SDoug Rabson 			}
1165a4f67738SDoug Rabson 		}
1166a4f67738SDoug Rabson 	}
1167a4f67738SDoug Rabson 	if (best == 0)
1168a4f67738SDoug Rabson 		*diffp = off;
1169a4f67738SDoug Rabson 	else
1170a4f67738SDoug Rabson 		*diffp = diff;
1171d254af07SMatthew Dillon 	*sym = (c_linker_sym_t) best;
1172a4f67738SDoug Rabson 
1173a4f67738SDoug Rabson 	return 0;
1174a4f67738SDoug Rabson }
1175f41325dbSPeter Wemm 
1176f41325dbSPeter Wemm /*
1177f41325dbSPeter Wemm  * Look up a linker set on an ELF system.
1178f41325dbSPeter Wemm  */
1179f41325dbSPeter Wemm static int
1180f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name,
1181f41325dbSPeter Wemm 		    void ***startp, void ***stopp, int *countp)
1182f41325dbSPeter Wemm {
1183f41325dbSPeter Wemm 	c_linker_sym_t sym;
1184f41325dbSPeter Wemm 	linker_symval_t symval;
1185f41325dbSPeter Wemm 	char *setsym;
1186f41325dbSPeter Wemm 	void **start, **stop;
1187f41325dbSPeter Wemm 	int len, error = 0, count;
1188f41325dbSPeter Wemm 
1189f41325dbSPeter Wemm 	len = strlen(name) + sizeof("__start_set_"); /* sizeof includes \0 */
1190a163d034SWarner Losh 	setsym = malloc(len, M_LINKER, M_WAITOK);
1191f41325dbSPeter Wemm 	if (setsym == NULL)
1192f41325dbSPeter Wemm 		return ENOMEM;
1193f41325dbSPeter Wemm 
1194f41325dbSPeter Wemm 	/* get address of first entry */
1195f41325dbSPeter Wemm 	snprintf(setsym, len, "%s%s", "__start_set_", name);
1196f41325dbSPeter Wemm 	error = link_elf_lookup_symbol(lf, setsym, &sym);
1197f41325dbSPeter Wemm 	if (error)
1198f41325dbSPeter Wemm 		goto out;
1199f41325dbSPeter Wemm 	link_elf_symbol_values(lf, sym, &symval);
1200f41325dbSPeter Wemm 	if (symval.value == 0) {
1201f41325dbSPeter Wemm 		error = ESRCH;
1202f41325dbSPeter Wemm 		goto out;
1203f41325dbSPeter Wemm 	}
1204f41325dbSPeter Wemm 	start = (void **)symval.value;
1205f41325dbSPeter Wemm 
1206f41325dbSPeter Wemm 	/* get address of last entry */
1207f41325dbSPeter Wemm 	snprintf(setsym, len, "%s%s", "__stop_set_", name);
1208f41325dbSPeter Wemm 	error = link_elf_lookup_symbol(lf, setsym, &sym);
1209f41325dbSPeter Wemm 	if (error)
1210f41325dbSPeter Wemm 		goto out;
1211f41325dbSPeter Wemm 	link_elf_symbol_values(lf, sym, &symval);
1212f41325dbSPeter Wemm 	if (symval.value == 0) {
1213f41325dbSPeter Wemm 		error = ESRCH;
1214f41325dbSPeter Wemm 		goto out;
1215f41325dbSPeter Wemm 	}
1216f41325dbSPeter Wemm 	stop = (void **)symval.value;
1217f41325dbSPeter Wemm 
1218f41325dbSPeter Wemm 	/* and the number of entries */
1219f41325dbSPeter Wemm 	count = stop - start;
1220f41325dbSPeter Wemm 
1221f41325dbSPeter Wemm 	/* and copy out */
1222f41325dbSPeter Wemm 	if (startp)
1223f41325dbSPeter Wemm 		*startp = start;
1224f41325dbSPeter Wemm 	if (stopp)
1225f41325dbSPeter Wemm 		*stopp = stop;
1226f41325dbSPeter Wemm 	if (countp)
1227f41325dbSPeter Wemm 		*countp = count;
1228f41325dbSPeter Wemm 
1229f41325dbSPeter Wemm out:
1230f41325dbSPeter Wemm 	free(setsym, M_LINKER);
1231f41325dbSPeter Wemm 	return error;
1232f41325dbSPeter Wemm }
1233bb9fe9ddSBrian Feldman 
1234bb9fe9ddSBrian Feldman static int
1235bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file,
1236bb9fe9ddSBrian Feldman   int (*callback)(const char *, void *), void *opaque) {
1237bb9fe9ddSBrian Feldman     elf_file_t ef = (elf_file_t)file;
1238bb9fe9ddSBrian Feldman     const Elf_Sym* symp;
1239bb9fe9ddSBrian Feldman     int i, error;
1240bb9fe9ddSBrian Feldman 
1241bb9fe9ddSBrian Feldman     /* Exhaustive search */
1242bb9fe9ddSBrian Feldman     for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1243bb9fe9ddSBrian Feldman 	if (symp->st_value != 0 &&
1244bb9fe9ddSBrian Feldman 	    ELF_ST_TYPE(symp->st_info) == STT_FUNC) {
1245bb9fe9ddSBrian Feldman 		error = callback(ef->ddbstrtab + symp->st_name, opaque);
1246bb9fe9ddSBrian Feldman 		if (error)
1247bb9fe9ddSBrian Feldman 		    return (error);
1248bb9fe9ddSBrian Feldman 	}
1249bb9fe9ddSBrian Feldman     }
1250bb9fe9ddSBrian Feldman     return (0);
1251bb9fe9ddSBrian Feldman }
125284201059SMarcel Moolenaar 
1253a2024a3eSJohn Birrell static int
1254a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file,
1255a2024a3eSJohn Birrell     linker_function_nameval_callback_t callback, void *opaque)
1256a2024a3eSJohn Birrell {
1257a2024a3eSJohn Birrell 	linker_symval_t symval;
1258a2024a3eSJohn Birrell 	elf_file_t ef = (elf_file_t)file;
1259a2024a3eSJohn Birrell 	const Elf_Sym* symp;
1260a2024a3eSJohn Birrell 	int i, error;
1261a2024a3eSJohn Birrell 
1262a2024a3eSJohn Birrell 	/* Exhaustive search */
1263a2024a3eSJohn Birrell 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1264a2024a3eSJohn Birrell 		if (symp->st_value != 0 &&
1265a2024a3eSJohn Birrell 		    ELF_ST_TYPE(symp->st_info) == STT_FUNC) {
1266a2024a3eSJohn Birrell 			error = link_elf_symbol_values(file, (c_linker_sym_t) symp, &symval);
1267a2024a3eSJohn Birrell 			if (error)
1268a2024a3eSJohn Birrell 				return (error);
1269a2024a3eSJohn Birrell 			error = callback(file, i, &symval, opaque);
1270a2024a3eSJohn Birrell 			if (error)
1271a2024a3eSJohn Birrell 				return (error);
1272a2024a3eSJohn Birrell 		}
1273a2024a3eSJohn Birrell 	}
1274a2024a3eSJohn Birrell 	return (0);
1275a2024a3eSJohn Birrell }
1276a2024a3eSJohn Birrell 
127784201059SMarcel Moolenaar #ifdef __ia64__
127884201059SMarcel Moolenaar /*
127984201059SMarcel Moolenaar  * Each KLD has its own GP. The GP value for each load module is given by
128084201059SMarcel Moolenaar  * DT_PLTGOT on ia64. We need GP to construct function descriptors, but
128184201059SMarcel Moolenaar  * don't have direct access to the ELF file structure. The link_elf_get_gp()
128284201059SMarcel Moolenaar  * function returns the GP given a pointer to a generic linker file struct.
128384201059SMarcel Moolenaar  */
128484201059SMarcel Moolenaar Elf_Addr
128584201059SMarcel Moolenaar link_elf_get_gp(linker_file_t lf)
128684201059SMarcel Moolenaar {
128784201059SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
128884201059SMarcel Moolenaar 	return (Elf_Addr)ef->got;
128984201059SMarcel Moolenaar }
129084201059SMarcel Moolenaar #endif
1291d297ad16SMarcel Moolenaar 
1292169d513cSJake Burkholder const Elf_Sym *
1293757686b1SMarcel Moolenaar elf_get_sym(linker_file_t lf, Elf_Size symidx)
1294169d513cSJake Burkholder {
1295169d513cSJake Burkholder 	elf_file_t ef = (elf_file_t)lf;
1296169d513cSJake Burkholder 
1297169d513cSJake Burkholder 	if (symidx >= ef->nchains)
1298169d513cSJake Burkholder 		return (NULL);
1299169d513cSJake Burkholder 	return (ef->symtab + symidx);
1300169d513cSJake Burkholder }
1301169d513cSJake Burkholder 
130247f75012SMarcel Moolenaar const char *
1303757686b1SMarcel Moolenaar elf_get_symname(linker_file_t lf, Elf_Size symidx)
130447f75012SMarcel Moolenaar {
130547f75012SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
130647f75012SMarcel Moolenaar 	const Elf_Sym *sym;
130747f75012SMarcel Moolenaar 
130847f75012SMarcel Moolenaar 	if (symidx >= ef->nchains)
130947f75012SMarcel Moolenaar 		return (NULL);
131047f75012SMarcel Moolenaar 	sym = ef->symtab + symidx;
131147f75012SMarcel Moolenaar 	return (ef->strtab + sym->st_name);
131247f75012SMarcel Moolenaar }
131347f75012SMarcel Moolenaar 
1314d297ad16SMarcel Moolenaar /*
1315d297ad16SMarcel Moolenaar  * Symbol lookup function that can be used when the symbol index is known (ie
1316d297ad16SMarcel Moolenaar  * in relocations). It uses the symbol index instead of doing a fully fledged
1317d297ad16SMarcel Moolenaar  * hash table based lookup when such is valid. For example for local symbols.
1318d297ad16SMarcel Moolenaar  * This is not only more efficient, it's also more correct. It's not always
1319d297ad16SMarcel Moolenaar  * the case that the symbol can be found through the hash table.
1320d297ad16SMarcel Moolenaar  */
1321e8855d4fSPeter Wemm static Elf_Addr
1322757686b1SMarcel Moolenaar elf_lookup(linker_file_t lf, Elf_Size symidx, int deps)
1323d297ad16SMarcel Moolenaar {
1324d297ad16SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
1325d297ad16SMarcel Moolenaar 	const Elf_Sym *sym;
1326d297ad16SMarcel Moolenaar 	const char *symbol;
1327d297ad16SMarcel Moolenaar 
1328d297ad16SMarcel Moolenaar 	/* Don't even try to lookup the symbol if the index is bogus. */
1329d297ad16SMarcel Moolenaar 	if (symidx >= ef->nchains)
1330d297ad16SMarcel Moolenaar 		return (0);
1331d297ad16SMarcel Moolenaar 
1332d297ad16SMarcel Moolenaar 	sym = ef->symtab + symidx;
1333d297ad16SMarcel Moolenaar 
1334d297ad16SMarcel Moolenaar 	/*
1335d297ad16SMarcel Moolenaar 	 * Don't do a full lookup when the symbol is local. It may even
1336d297ad16SMarcel Moolenaar 	 * fail because it may not be found through the hash table.
1337d297ad16SMarcel Moolenaar 	 */
1338d297ad16SMarcel Moolenaar 	if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) {
1339d297ad16SMarcel Moolenaar 		/* Force lookup failure when we have an insanity. */
1340d297ad16SMarcel Moolenaar 		if (sym->st_shndx == SHN_UNDEF || sym->st_value == 0)
1341d297ad16SMarcel Moolenaar 			return (0);
1342d297ad16SMarcel Moolenaar 		return ((Elf_Addr)ef->address + sym->st_value);
1343d297ad16SMarcel Moolenaar 	}
1344d297ad16SMarcel Moolenaar 
1345d297ad16SMarcel Moolenaar 	/*
1346d297ad16SMarcel Moolenaar 	 * XXX we can avoid doing a hash table based lookup for global
1347d297ad16SMarcel Moolenaar 	 * symbols as well. This however is not always valid, so we'll
1348d297ad16SMarcel Moolenaar 	 * just do it the hard way for now. Performance tweaks can
1349d297ad16SMarcel Moolenaar 	 * always be added.
1350d297ad16SMarcel Moolenaar 	 */
1351d297ad16SMarcel Moolenaar 
1352d297ad16SMarcel Moolenaar 	symbol = ef->strtab + sym->st_name;
1353d297ad16SMarcel Moolenaar 
1354d297ad16SMarcel Moolenaar 	/* Force a lookup failure if the symbol name is bogus. */
1355d297ad16SMarcel Moolenaar 	if (*symbol == 0)
1356d297ad16SMarcel Moolenaar 		return (0);
1357d297ad16SMarcel Moolenaar 
1358d297ad16SMarcel Moolenaar 	return ((Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps));
1359d297ad16SMarcel Moolenaar }
13607251b4bfSJake Burkholder 
13617251b4bfSJake Burkholder static void
13627251b4bfSJake Burkholder link_elf_reloc_local(linker_file_t lf)
13637251b4bfSJake Burkholder {
13647251b4bfSJake Burkholder     const Elf_Rel *rellim;
13657251b4bfSJake Burkholder     const Elf_Rel *rel;
13667251b4bfSJake Burkholder     const Elf_Rela *relalim;
13677251b4bfSJake Burkholder     const Elf_Rela *rela;
13687251b4bfSJake Burkholder     elf_file_t ef = (elf_file_t)lf;
13697251b4bfSJake Burkholder 
13707251b4bfSJake Burkholder     /* Perform relocations without addend if there are any: */
13717251b4bfSJake Burkholder     if ((rel = ef->rel) != NULL) {
13727251b4bfSJake Burkholder 	rellim = (const Elf_Rel *)((const char *)ef->rel + ef->relsize);
13737251b4bfSJake Burkholder 	while (rel < rellim) {
1374e8855d4fSPeter Wemm 	    elf_reloc_local(lf, (Elf_Addr)ef->address, rel, ELF_RELOC_REL,
1375e8855d4fSPeter Wemm 			    elf_lookup);
13767251b4bfSJake Burkholder 	    rel++;
13777251b4bfSJake Burkholder 	}
13787251b4bfSJake Burkholder     }
13797251b4bfSJake Burkholder 
13807251b4bfSJake Burkholder     /* Perform relocations with addend if there are any: */
13817251b4bfSJake Burkholder     if ((rela = ef->rela) != NULL) {
13827251b4bfSJake Burkholder 	relalim = (const Elf_Rela *)((const char *)ef->rela + ef->relasize);
13837251b4bfSJake Burkholder 	while (rela < relalim) {
1384e8855d4fSPeter Wemm 	    elf_reloc_local(lf, (Elf_Addr)ef->address, rela, ELF_RELOC_RELA,
1385e8855d4fSPeter Wemm 			    elf_lookup);
13867251b4bfSJake Burkholder 	    rela++;
13877251b4bfSJake Burkholder 	}
13887251b4bfSJake Burkholder     }
13897251b4bfSJake Burkholder }
1390