xref: /freebsd/sys/kern/link_elf.c (revision eddfbb763ded6b5f6777335142be9a0edab628bb)
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"
32326e27d8SDoug Rabson 
33a4f67738SDoug Rabson #include <sys/param.h>
34a4f67738SDoug Rabson #include <sys/systm.h>
355ad6c3b1SBruce Evans #ifdef GPROF
365ad6c3b1SBruce Evans #include <sys/gmon.h>
375ad6c3b1SBruce Evans #endif
38fb919e4dSMark Murray #include <sys/kernel.h>
39fb919e4dSMark Murray #include <sys/lock.h>
40a4f67738SDoug Rabson #include <sys/malloc.h>
41c0b824f9SDima Dorfman #include <sys/mutex.h>
4262d615d5SJohn Baldwin #include <sys/mount.h>
4350c202c5SJeff Roberson #include <sys/pcpu.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 
52eddfbb76SRobert Watson #include <net/vnet.h>
53eddfbb76SRobert Watson 
54aed55708SRobert Watson #include <security/mac/mac_framework.h>
55aed55708SRobert Watson 
56fe3db7c7SDoug Rabson #include <vm/vm.h>
57fe3db7c7SDoug Rabson #include <vm/vm_param.h>
58d1f088daSPeter Wemm #ifdef SPARSE_MAPPING
59fe3db7c7SDoug Rabson #include <vm/vm_object.h>
60fe3db7c7SDoug Rabson #include <vm/vm_kern.h>
61fe3db7c7SDoug Rabson #include <vm/vm_extern.h>
62d1f088daSPeter Wemm #endif
63fe3db7c7SDoug Rabson #include <vm/pmap.h>
64fe3db7c7SDoug Rabson #include <vm/vm_map.h>
65fb919e4dSMark Murray 
66f9980387SPeter Wemm #include <sys/link_elf.h>
67fe3db7c7SDoug Rabson 
68a2024a3eSJohn Birrell #ifdef DDB_CTF
69a2024a3eSJohn Birrell #include <net/zlib.h>
70a2024a3eSJohn Birrell #endif
71a2024a3eSJohn Birrell 
72326e27d8SDoug Rabson #include "linker_if.h"
73a4f67738SDoug Rabson 
741a330eb0SPeter Wemm #define MAXSEGS 4
751a330eb0SPeter Wemm 
764f2945f8SDavid E. O'Brien typedef struct elf_file {
774f2945f8SDavid E. O'Brien     struct linker_file	lf;		/* Common fields */
784f2945f8SDavid E. O'Brien     int			preloaded;	/* Was file pre-loaded */
794f2945f8SDavid E. O'Brien     caddr_t		address;	/* Relocation address */
804f2945f8SDavid E. O'Brien #ifdef SPARSE_MAPPING
814f2945f8SDavid E. O'Brien     vm_object_t		object;		/* VM object to hold file pages */
824f2945f8SDavid E. O'Brien #endif
834f2945f8SDavid E. O'Brien     Elf_Dyn*		dynamic;	/* Symbol table etc. */
844f2945f8SDavid E. O'Brien     Elf_Hashelt		nbuckets;	/* DT_HASH info */
854f2945f8SDavid E. O'Brien     Elf_Hashelt		nchains;
864f2945f8SDavid E. O'Brien     const Elf_Hashelt*	buckets;
874f2945f8SDavid E. O'Brien     const Elf_Hashelt*	chains;
884f2945f8SDavid E. O'Brien     caddr_t		hash;
894f2945f8SDavid E. O'Brien     caddr_t		strtab;		/* DT_STRTAB */
904f2945f8SDavid E. O'Brien     int			strsz;		/* DT_STRSZ */
914f2945f8SDavid E. O'Brien     const Elf_Sym*	symtab;		/* DT_SYMTAB */
924f2945f8SDavid E. O'Brien     Elf_Addr*		got;		/* DT_PLTGOT */
934f2945f8SDavid E. O'Brien     const Elf_Rel*	pltrel;		/* DT_JMPREL */
944f2945f8SDavid E. O'Brien     int			pltrelsize;	/* DT_PLTRELSZ */
954f2945f8SDavid E. O'Brien     const Elf_Rela*	pltrela;	/* DT_JMPREL */
964f2945f8SDavid E. O'Brien     int			pltrelasize;	/* DT_PLTRELSZ */
974f2945f8SDavid E. O'Brien     const Elf_Rel*	rel;		/* DT_REL */
984f2945f8SDavid E. O'Brien     int			relsize;	/* DT_RELSZ */
994f2945f8SDavid E. O'Brien     const Elf_Rela*	rela;		/* DT_RELA */
1004f2945f8SDavid E. O'Brien     int			relasize;	/* DT_RELASZ */
1014f2945f8SDavid E. O'Brien     caddr_t		modptr;
1024f2945f8SDavid E. O'Brien     const Elf_Sym*	ddbsymtab;	/* The symbol table we are using */
1034f2945f8SDavid E. O'Brien     long		ddbsymcnt;	/* Number of symbols */
1044f2945f8SDavid E. O'Brien     caddr_t		ddbstrtab;	/* String table */
1054f2945f8SDavid E. O'Brien     long		ddbstrcnt;	/* number of bytes in string table */
1064f2945f8SDavid E. O'Brien     caddr_t		symbase;	/* malloc'ed symbold base */
1074f2945f8SDavid E. O'Brien     caddr_t		strbase;	/* malloc'ed string base */
1084f2945f8SDavid E. O'Brien     caddr_t		ctftab;		/* CTF table */
1094f2945f8SDavid E. O'Brien     long		ctfcnt;		/* number of bytes in CTF table */
1104f2945f8SDavid E. O'Brien     caddr_t		ctfoff;		/* CTF offset table */
1114f2945f8SDavid E. O'Brien     caddr_t		typoff;		/* Type offset table */
1124f2945f8SDavid E. O'Brien     long		typlen;		/* Number of type entries. */
11350c202c5SJeff Roberson     Elf_Addr		pcpu_start;	/* Pre-relocation pcpu set start. */
11450c202c5SJeff Roberson     Elf_Addr		pcpu_stop;	/* Pre-relocation pcpu set stop. */
11550c202c5SJeff Roberson     Elf_Addr		pcpu_base;	/* Relocated pcpu set address. */
116eddfbb76SRobert Watson #ifdef VIMAGE
117eddfbb76SRobert Watson     Elf_Addr		vnet_start;	/* Pre-relocation vnet set start. */
118eddfbb76SRobert Watson     Elf_Addr		vnet_stop;	/* Pre-relocation vnet set stop. */
119eddfbb76SRobert Watson     Elf_Addr		vnet_base;	/* Relocated vnet set address. */
120eddfbb76SRobert Watson #endif
1214f2945f8SDavid E. O'Brien #ifdef GDB
1224f2945f8SDavid E. O'Brien     struct link_map	gdb;		/* hooks for gdb */
1234f2945f8SDavid E. O'Brien #endif
1244f2945f8SDavid E. O'Brien } *elf_file_t;
1254f2945f8SDavid E. O'Brien 
126a2024a3eSJohn Birrell #include <kern/kern_ctf.c>
127a2024a3eSJohn Birrell 
128c143d6c2SMarcel Moolenaar static int	link_elf_link_common_finish(linker_file_t);
12954823af2SPeter Wemm static int	link_elf_link_preload(linker_class_t cls,
130326e27d8SDoug Rabson 				      const char*, linker_file_t*);
13154823af2SPeter Wemm static int	link_elf_link_preload_finish(linker_file_t);
13254823af2SPeter Wemm static int	link_elf_load_file(linker_class_t, const char*, linker_file_t*);
133326e27d8SDoug Rabson static int	link_elf_lookup_symbol(linker_file_t, const char*,
134326e27d8SDoug Rabson 				       c_linker_sym_t*);
135326e27d8SDoug Rabson static int	link_elf_symbol_values(linker_file_t, c_linker_sym_t, linker_symval_t*);
136326e27d8SDoug Rabson static int	link_elf_search_symbol(linker_file_t, caddr_t value,
137326e27d8SDoug Rabson 				       c_linker_sym_t* sym, long* diffp);
138326e27d8SDoug Rabson 
139326e27d8SDoug Rabson static void	link_elf_unload_file(linker_file_t);
14054823af2SPeter Wemm static void	link_elf_unload_preload(linker_file_t);
141f41325dbSPeter Wemm static int	link_elf_lookup_set(linker_file_t, const char *,
142f41325dbSPeter Wemm 				    void ***, void ***, int *);
143bb9fe9ddSBrian Feldman static int	link_elf_each_function_name(linker_file_t,
144bb9fe9ddSBrian Feldman 				int (*)(const char *, void *),
145bb9fe9ddSBrian Feldman 				void *);
146a2024a3eSJohn Birrell static int	link_elf_each_function_nameval(linker_file_t,
147a2024a3eSJohn Birrell 				linker_function_nameval_callback_t,
148a2024a3eSJohn Birrell 				void *);
1497251b4bfSJake Burkholder static void	link_elf_reloc_local(linker_file_t);
15000a5db46SStacey Son static long	link_elf_symtab_get(linker_file_t, const Elf_Sym **);
15100a5db46SStacey Son static long	link_elf_strtab_get(linker_file_t, caddr_t *);
152757686b1SMarcel Moolenaar static Elf_Addr	elf_lookup(linker_file_t lf, Elf_Size symidx, int deps);
153326e27d8SDoug Rabson 
154326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = {
155326e27d8SDoug Rabson     KOBJMETHOD(linker_lookup_symbol,	link_elf_lookup_symbol),
156326e27d8SDoug Rabson     KOBJMETHOD(linker_symbol_values,	link_elf_symbol_values),
157326e27d8SDoug Rabson     KOBJMETHOD(linker_search_symbol,	link_elf_search_symbol),
158326e27d8SDoug Rabson     KOBJMETHOD(linker_unload,		link_elf_unload_file),
15954823af2SPeter Wemm     KOBJMETHOD(linker_load_file,	link_elf_load_file),
16054823af2SPeter Wemm     KOBJMETHOD(linker_link_preload,	link_elf_link_preload),
16154823af2SPeter Wemm     KOBJMETHOD(linker_link_preload_finish, link_elf_link_preload_finish),
162f41325dbSPeter Wemm     KOBJMETHOD(linker_lookup_set,	link_elf_lookup_set),
163bb9fe9ddSBrian Feldman     KOBJMETHOD(linker_each_function_name, link_elf_each_function_name),
164a2024a3eSJohn Birrell     KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval),
165a2024a3eSJohn Birrell     KOBJMETHOD(linker_ctf_get,          link_elf_ctf_get),
16600a5db46SStacey Son     KOBJMETHOD(linker_symtab_get, 	link_elf_symtab_get),
16700a5db46SStacey Son     KOBJMETHOD(linker_strtab_get, 	link_elf_strtab_get),
168326e27d8SDoug Rabson     { 0, 0 }
169326e27d8SDoug Rabson };
170326e27d8SDoug Rabson 
171326e27d8SDoug Rabson static struct linker_class link_elf_class = {
172326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32
173326e27d8SDoug Rabson     "elf32",
174326e27d8SDoug Rabson #else
175326e27d8SDoug Rabson     "elf64",
176326e27d8SDoug Rabson #endif
177326e27d8SDoug Rabson     link_elf_methods, sizeof(struct elf_file)
178326e27d8SDoug Rabson };
179326e27d8SDoug Rabson 
180326e27d8SDoug Rabson static int		parse_dynamic(elf_file_t ef);
181326e27d8SDoug Rabson static int		relocate_file(elf_file_t ef);
18254823af2SPeter Wemm static int		link_elf_preload_parse_symbols(elf_file_t ef);
183326e27d8SDoug Rabson 
184ecb01c64SMarcel Moolenaar #ifdef GDB
185ca29467eSJohn Baldwin static void		r_debug_state(struct r_debug *dummy_one,
186ca29467eSJohn Baldwin 				      struct link_map *dummy_two);
187326e27d8SDoug Rabson 
188326e27d8SDoug Rabson /*
189326e27d8SDoug Rabson  * A list of loaded modules for GDB to use for loading symbols.
190326e27d8SDoug Rabson  */
191326e27d8SDoug Rabson struct r_debug r_debug;
192326e27d8SDoug Rabson 
193ca29467eSJohn Baldwin #define GDB_STATE(s)	r_debug.r_state = s; r_debug_state(NULL, NULL);
194326e27d8SDoug Rabson 
195326e27d8SDoug Rabson /*
196326e27d8SDoug Rabson  * Function for the debugger to set a breakpoint on to gain control.
197326e27d8SDoug Rabson  */
19837c84183SPoul-Henning Kamp static void
199ca29467eSJohn Baldwin r_debug_state(struct r_debug *dummy_one __unused,
200ca29467eSJohn Baldwin 	      struct link_map *dummy_two __unused)
201326e27d8SDoug Rabson {
202326e27d8SDoug Rabson }
203326e27d8SDoug Rabson 
2041720979bSMarcel Moolenaar static void
2051720979bSMarcel Moolenaar link_elf_add_gdb(struct link_map *l)
2061720979bSMarcel Moolenaar {
2071720979bSMarcel Moolenaar     struct link_map *prev;
2081720979bSMarcel Moolenaar 
209c143d6c2SMarcel Moolenaar     l->l_next = NULL;
210c143d6c2SMarcel Moolenaar 
211c143d6c2SMarcel Moolenaar     if (r_debug.r_map == NULL) {
212c143d6c2SMarcel Moolenaar 	/* Add first. */
213c143d6c2SMarcel Moolenaar 	l->l_prev = NULL;
214c143d6c2SMarcel Moolenaar 	r_debug.r_map = l;
215c143d6c2SMarcel Moolenaar     } else {
216c143d6c2SMarcel Moolenaar 	/* Append to list. */
2171720979bSMarcel Moolenaar 	for (prev = r_debug.r_map; prev->l_next != NULL; prev = prev->l_next)
2181720979bSMarcel Moolenaar 	    ;
2191720979bSMarcel Moolenaar 	l->l_prev = prev;
2201720979bSMarcel Moolenaar 	prev->l_next = l;
2211720979bSMarcel Moolenaar     }
222c143d6c2SMarcel Moolenaar }
2231720979bSMarcel Moolenaar 
2241720979bSMarcel Moolenaar static void
2251720979bSMarcel Moolenaar link_elf_delete_gdb(struct link_map *l)
2261720979bSMarcel Moolenaar {
2271720979bSMarcel Moolenaar     if (l->l_prev == NULL) {
228c143d6c2SMarcel Moolenaar 	/* Remove first. */
2291720979bSMarcel Moolenaar 	if ((r_debug.r_map = l->l_next) != NULL)
2301720979bSMarcel Moolenaar 	    l->l_next->l_prev = NULL;
231c143d6c2SMarcel Moolenaar     } else {
232c143d6c2SMarcel Moolenaar 	/* Remove any but first. */
2331720979bSMarcel Moolenaar 	if ((l->l_prev->l_next = l->l_next) != NULL)
2341720979bSMarcel Moolenaar 	    l->l_next->l_prev = l->l_prev;
2351720979bSMarcel Moolenaar     }
236c143d6c2SMarcel Moolenaar }
237ecb01c64SMarcel Moolenaar #endif /* GDB */
238a4f67738SDoug Rabson 
23984201059SMarcel Moolenaar #ifdef __ia64__
24084201059SMarcel Moolenaar Elf_Addr link_elf_get_gp(linker_file_t);
24184201059SMarcel Moolenaar #endif
24284201059SMarcel Moolenaar 
2434f2945f8SDavid E. O'Brien /*
2444f2945f8SDavid E. O'Brien  * The kernel symbol table starts here.
2454f2945f8SDavid E. O'Brien  */
2464f2945f8SDavid E. O'Brien extern struct _dynamic _DYNAMIC;
2474f2945f8SDavid E. O'Brien 
248a4f67738SDoug Rabson static void
249552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s)
2501720979bSMarcel Moolenaar {
251552f9f63SEdwin Groothuis 	if (filename == NULL)
2521720979bSMarcel Moolenaar 		printf("kldload: %s\n", s);
253552f9f63SEdwin Groothuis 	else
254552f9f63SEdwin Groothuis 		printf("kldload: %s: %s\n", filename, s);
2551720979bSMarcel Moolenaar }
2561720979bSMarcel Moolenaar 
257c143d6c2SMarcel Moolenaar /*
258c143d6c2SMarcel Moolenaar  * Actions performed after linking/loading both the preloaded kernel and any
259c143d6c2SMarcel Moolenaar  * modules; whether preloaded or dynamicly loaded.
260c143d6c2SMarcel Moolenaar  */
261c143d6c2SMarcel Moolenaar static int
262c143d6c2SMarcel Moolenaar link_elf_link_common_finish(linker_file_t lf)
263c143d6c2SMarcel Moolenaar {
264ecb01c64SMarcel Moolenaar #ifdef GDB
265c143d6c2SMarcel Moolenaar     elf_file_t ef = (elf_file_t)lf;
266c143d6c2SMarcel Moolenaar     char *newfilename;
267c143d6c2SMarcel Moolenaar #endif
2681aeb23cdSMarcel Moolenaar     int error;
2691aeb23cdSMarcel Moolenaar 
2701aeb23cdSMarcel Moolenaar     /* Notify MD code that a module is being loaded. */
2711aeb23cdSMarcel Moolenaar     error = elf_cpu_load_file(lf);
2721aeb23cdSMarcel Moolenaar     if (error)
2731aeb23cdSMarcel Moolenaar 	return (error);
274c143d6c2SMarcel Moolenaar 
275ecb01c64SMarcel Moolenaar #ifdef GDB
276c143d6c2SMarcel Moolenaar     GDB_STATE(RT_ADD);
277c143d6c2SMarcel Moolenaar     ef->gdb.l_addr = lf->address;
278a163d034SWarner Losh     newfilename = malloc(strlen(lf->filename) + 1, M_LINKER, M_WAITOK);
279c143d6c2SMarcel Moolenaar     strcpy(newfilename, lf->filename);
280c143d6c2SMarcel Moolenaar     ef->gdb.l_name = newfilename;
281c143d6c2SMarcel Moolenaar     ef->gdb.l_ld = ef->dynamic;
282c143d6c2SMarcel Moolenaar     link_elf_add_gdb(&ef->gdb);
283c143d6c2SMarcel Moolenaar     GDB_STATE(RT_CONSISTENT);
284c143d6c2SMarcel Moolenaar #endif
285c143d6c2SMarcel Moolenaar 
286c143d6c2SMarcel Moolenaar     return (0);
287c143d6c2SMarcel Moolenaar }
288c143d6c2SMarcel Moolenaar 
2891720979bSMarcel Moolenaar static void
290a4f67738SDoug Rabson link_elf_init(void* arg)
291a4f67738SDoug Rabson {
292de78ca7eSPeter Wemm     Elf_Dyn	*dp;
293de78ca7eSPeter Wemm     caddr_t	modptr, baseptr, sizeptr;
294de78ca7eSPeter Wemm     elf_file_t	ef;
295de78ca7eSPeter Wemm     char	*modname;
296a4f67738SDoug Rabson 
297326e27d8SDoug Rabson     linker_add_class(&link_elf_class);
298a4f67738SDoug Rabson 
299de78ca7eSPeter Wemm     dp = (Elf_Dyn*) &_DYNAMIC;
300de78ca7eSPeter Wemm     modname = NULL;
30167db8b23SPeter Wemm     modptr = preload_search_by_type("elf" __XSTRING(__ELF_WORD_SIZE) " kernel");
30267db8b23SPeter Wemm     if (modptr == NULL)
303de78ca7eSPeter Wemm 	modptr = preload_search_by_type("elf kernel");
304de78ca7eSPeter Wemm     if (modptr)
305de78ca7eSPeter Wemm 	modname = (char *)preload_search_info(modptr, MODINFO_NAME);
306de78ca7eSPeter Wemm     if (modname == NULL)
307de78ca7eSPeter Wemm 	modname = "kernel";
308326e27d8SDoug Rabson     linker_kernel_file = linker_make_file(modname, &link_elf_class);
309a4f67738SDoug Rabson     if (linker_kernel_file == NULL)
310a4f67738SDoug Rabson 	panic("link_elf_init: Can't create linker structures for kernel");
311326e27d8SDoug Rabson 
312326e27d8SDoug Rabson     ef = (elf_file_t) linker_kernel_file;
31354823af2SPeter Wemm     ef->preloaded = 1;
314326e27d8SDoug Rabson     ef->address = 0;
315326e27d8SDoug Rabson #ifdef SPARSE_MAPPING
316326e27d8SDoug Rabson     ef->object = 0;
317326e27d8SDoug Rabson #endif
318326e27d8SDoug Rabson     ef->dynamic = dp;
319326e27d8SDoug Rabson 
32000dda5e8SPeter Wemm     if (dp)
321326e27d8SDoug Rabson 	parse_dynamic(ef);
3224173e420SPeter Wemm     linker_kernel_file->address = (caddr_t) KERNBASE;
323d009ccfaSBruce Evans     linker_kernel_file->size = -(intptr_t)linker_kernel_file->address;
324de78ca7eSPeter Wemm 
325de78ca7eSPeter Wemm     if (modptr) {
3262d636ab0SPeter Wemm 	ef->modptr = modptr;
327de78ca7eSPeter Wemm 	baseptr = preload_search_info(modptr, MODINFO_ADDR);
328de78ca7eSPeter Wemm 	if (baseptr)
329de78ca7eSPeter Wemm 	    linker_kernel_file->address = *(caddr_t *)baseptr;
330de78ca7eSPeter Wemm 	sizeptr = preload_search_info(modptr, MODINFO_SIZE);
331de78ca7eSPeter Wemm 	if (sizeptr)
332de78ca7eSPeter Wemm 	    linker_kernel_file->size = *(size_t *)sizeptr;
333de78ca7eSPeter Wemm     }
33454823af2SPeter Wemm     (void)link_elf_preload_parse_symbols(ef);
335326e27d8SDoug Rabson 
336ecb01c64SMarcel Moolenaar #ifdef GDB
337c143d6c2SMarcel Moolenaar     r_debug.r_map = NULL;
338326e27d8SDoug Rabson     r_debug.r_brk = r_debug_state;
339326e27d8SDoug Rabson     r_debug.r_state = RT_CONSISTENT;
340326e27d8SDoug Rabson #endif
341c143d6c2SMarcel Moolenaar 
342c143d6c2SMarcel Moolenaar     (void)link_elf_link_common_finish(linker_kernel_file);
34359800afcSJohn Baldwin     linker_kernel_file->flags |= LINKER_FILE_LINKED;
344a4f67738SDoug Rabson }
345a4f67738SDoug Rabson 
3468120ddb4SJung-uk Kim SYSINIT(link_elf, SI_SUB_KLD, SI_ORDER_THIRD, link_elf_init, 0);
347a4f67738SDoug Rabson 
348a4f67738SDoug Rabson static int
34954823af2SPeter Wemm link_elf_preload_parse_symbols(elf_file_t ef)
3502d636ab0SPeter Wemm {
3512d636ab0SPeter Wemm     caddr_t	pointer;
3522d636ab0SPeter Wemm     caddr_t	ssym, esym, base;
3532d636ab0SPeter Wemm     caddr_t	strtab;
3542d636ab0SPeter Wemm     int		strcnt;
3552d636ab0SPeter Wemm     Elf_Sym*	symtab;
3562d636ab0SPeter Wemm     int		symcnt;
3572d636ab0SPeter Wemm 
358ca65d5c7SPeter Wemm     if (ef->modptr == NULL)
359ca65d5c7SPeter Wemm 	return 0;
3602d636ab0SPeter Wemm     pointer = preload_search_info(ef->modptr, MODINFO_METADATA|MODINFOMD_SSYM);
3612d636ab0SPeter Wemm     if (pointer == NULL)
3622d636ab0SPeter Wemm 	return 0;
3632d636ab0SPeter Wemm     ssym = *(caddr_t *)pointer;
3642d636ab0SPeter Wemm     pointer = preload_search_info(ef->modptr, MODINFO_METADATA|MODINFOMD_ESYM);
3652d636ab0SPeter Wemm     if (pointer == NULL)
3662d636ab0SPeter Wemm 	return 0;
3672d636ab0SPeter Wemm     esym = *(caddr_t *)pointer;
3682d636ab0SPeter Wemm 
3692d636ab0SPeter Wemm     base = ssym;
3702d636ab0SPeter Wemm 
3712d636ab0SPeter Wemm     symcnt = *(long *)base;
3722d636ab0SPeter Wemm     base += sizeof(long);
3732d636ab0SPeter Wemm     symtab = (Elf_Sym *)base;
3742d636ab0SPeter Wemm     base += roundup(symcnt, sizeof(long));
3752d636ab0SPeter Wemm 
3762d636ab0SPeter Wemm     if (base > esym || base < ssym) {
3772d636ab0SPeter Wemm 	printf("Symbols are corrupt!\n");
3782d636ab0SPeter Wemm 	return EINVAL;
3792d636ab0SPeter Wemm     }
3802d636ab0SPeter Wemm 
3812d636ab0SPeter Wemm     strcnt = *(long *)base;
3822d636ab0SPeter Wemm     base += sizeof(long);
3832d636ab0SPeter Wemm     strtab = base;
3842d636ab0SPeter Wemm     base += roundup(strcnt, sizeof(long));
3852d636ab0SPeter Wemm 
3862d636ab0SPeter Wemm     if (base > esym || base < ssym) {
3872d636ab0SPeter Wemm 	printf("Symbols are corrupt!\n");
3882d636ab0SPeter Wemm 	return EINVAL;
3892d636ab0SPeter Wemm     }
3902d636ab0SPeter Wemm 
3912d636ab0SPeter Wemm     ef->ddbsymtab = symtab;
3922d636ab0SPeter Wemm     ef->ddbsymcnt = symcnt / sizeof(Elf_Sym);
3932d636ab0SPeter Wemm     ef->ddbstrtab = strtab;
3942d636ab0SPeter Wemm     ef->ddbstrcnt = strcnt;
3952d636ab0SPeter Wemm 
3962d636ab0SPeter Wemm     return 0;
3972d636ab0SPeter Wemm }
3982d636ab0SPeter Wemm 
3992d636ab0SPeter Wemm static int
400326e27d8SDoug Rabson parse_dynamic(elf_file_t ef)
401a4f67738SDoug Rabson {
402326e27d8SDoug Rabson     Elf_Dyn *dp;
403fe3db7c7SDoug Rabson     int plttype = DT_REL;
404a4f67738SDoug Rabson 
405a4f67738SDoug Rabson     for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) {
406a4f67738SDoug Rabson 	switch (dp->d_tag) {
407a4f67738SDoug Rabson 	case DT_HASH:
408a4f67738SDoug Rabson 	{
409a4f67738SDoug Rabson 	    /* From src/libexec/rtld-elf/rtld.c */
4107e510298SJake Burkholder 	    const Elf_Hashelt *hashtab = (const Elf_Hashelt *)
411a4f67738SDoug Rabson 		(ef->address + dp->d_un.d_ptr);
412a4f67738SDoug Rabson 	    ef->nbuckets = hashtab[0];
413a4f67738SDoug Rabson 	    ef->nchains = hashtab[1];
414a4f67738SDoug Rabson 	    ef->buckets = hashtab + 2;
415a4f67738SDoug Rabson 	    ef->chains = ef->buckets + ef->nbuckets;
416a4f67738SDoug Rabson 	    break;
417a4f67738SDoug Rabson 	}
418a4f67738SDoug Rabson 	case DT_STRTAB:
419fe3db7c7SDoug Rabson 	    ef->strtab = (caddr_t) (ef->address + dp->d_un.d_ptr);
420a4f67738SDoug Rabson 	    break;
4212d636ab0SPeter Wemm 	case DT_STRSZ:
4222d636ab0SPeter Wemm 	    ef->strsz = dp->d_un.d_val;
4232d636ab0SPeter Wemm 	    break;
424a4f67738SDoug Rabson 	case DT_SYMTAB:
425fe3db7c7SDoug Rabson 	    ef->symtab = (Elf_Sym*) (ef->address + dp->d_un.d_ptr);
426a4f67738SDoug Rabson 	    break;
427a4f67738SDoug Rabson 	case DT_SYMENT:
428a4f67738SDoug Rabson 	    if (dp->d_un.d_val != sizeof(Elf_Sym))
429a4f67738SDoug Rabson 		return ENOEXEC;
430fe3db7c7SDoug Rabson 	    break;
431fe3db7c7SDoug Rabson 	case DT_PLTGOT:
432fe3db7c7SDoug Rabson 	    ef->got = (Elf_Addr *) (ef->address + dp->d_un.d_ptr);
433fe3db7c7SDoug Rabson 	    break;
434fe3db7c7SDoug Rabson 	case DT_REL:
435fe3db7c7SDoug Rabson 	    ef->rel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr);
436fe3db7c7SDoug Rabson 	    break;
437fe3db7c7SDoug Rabson 	case DT_RELSZ:
438fe3db7c7SDoug Rabson 	    ef->relsize = dp->d_un.d_val;
439fe3db7c7SDoug Rabson 	    break;
440fe3db7c7SDoug Rabson 	case DT_RELENT:
441fe3db7c7SDoug Rabson 	    if (dp->d_un.d_val != sizeof(Elf_Rel))
442fe3db7c7SDoug Rabson 		return ENOEXEC;
443fe3db7c7SDoug Rabson 	    break;
444fe3db7c7SDoug Rabson 	case DT_JMPREL:
445fe3db7c7SDoug Rabson 	    ef->pltrel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr);
446fe3db7c7SDoug Rabson 	    break;
447fe3db7c7SDoug Rabson 	case DT_PLTRELSZ:
448fe3db7c7SDoug Rabson 	    ef->pltrelsize = dp->d_un.d_val;
449fe3db7c7SDoug Rabson 	    break;
450fe3db7c7SDoug Rabson 	case DT_RELA:
451fe3db7c7SDoug Rabson 	    ef->rela = (const Elf_Rela *) (ef->address + dp->d_un.d_ptr);
452fe3db7c7SDoug Rabson 	    break;
453fe3db7c7SDoug Rabson 	case DT_RELASZ:
454fe3db7c7SDoug Rabson 	    ef->relasize = dp->d_un.d_val;
455fe3db7c7SDoug Rabson 	    break;
456fe3db7c7SDoug Rabson 	case DT_RELAENT:
457fe3db7c7SDoug Rabson 	    if (dp->d_un.d_val != sizeof(Elf_Rela))
458fe3db7c7SDoug Rabson 		return ENOEXEC;
459fe3db7c7SDoug Rabson 	    break;
460fe3db7c7SDoug Rabson 	case DT_PLTREL:
461fe3db7c7SDoug Rabson 	    plttype = dp->d_un.d_val;
462fe3db7c7SDoug Rabson 	    if (plttype != DT_REL && plttype != DT_RELA)
463fe3db7c7SDoug Rabson 		return ENOEXEC;
464fe3db7c7SDoug Rabson 	    break;
465ecb01c64SMarcel Moolenaar #ifdef GDB
466326e27d8SDoug Rabson 	case DT_DEBUG:
467326e27d8SDoug Rabson 	    dp->d_un.d_ptr = (Elf_Addr) &r_debug;
468326e27d8SDoug Rabson 	    break;
469326e27d8SDoug Rabson #endif
470a4f67738SDoug Rabson 	}
471a4f67738SDoug Rabson     }
472fe3db7c7SDoug Rabson 
473fe3db7c7SDoug Rabson     if (plttype == DT_RELA) {
474fe3db7c7SDoug Rabson 	ef->pltrela = (const Elf_Rela *) ef->pltrel;
475fe3db7c7SDoug Rabson 	ef->pltrel = NULL;
476fe3db7c7SDoug Rabson 	ef->pltrelasize = ef->pltrelsize;
477fe3db7c7SDoug Rabson 	ef->pltrelsize = 0;
478fe3db7c7SDoug Rabson     }
479fe3db7c7SDoug Rabson 
4802d636ab0SPeter Wemm     ef->ddbsymtab = ef->symtab;
4812d636ab0SPeter Wemm     ef->ddbsymcnt = ef->nchains;
4822d636ab0SPeter Wemm     ef->ddbstrtab = ef->strtab;
4832d636ab0SPeter Wemm     ef->ddbstrcnt = ef->strsz;
4842d636ab0SPeter Wemm 
485a4f67738SDoug Rabson     return 0;
486a4f67738SDoug Rabson }
487a4f67738SDoug Rabson 
488a4f67738SDoug Rabson static int
48950c202c5SJeff Roberson parse_dpcpu(elf_file_t ef)
49050c202c5SJeff Roberson {
49150c202c5SJeff Roberson     int count;
49250c202c5SJeff Roberson     int error;
49350c202c5SJeff Roberson 
49450c202c5SJeff Roberson     ef->pcpu_start = 0;
49550c202c5SJeff Roberson     ef->pcpu_stop = 0;
49650c202c5SJeff Roberson     error = link_elf_lookup_set(&ef->lf, "pcpu", (void ***)&ef->pcpu_start,
49750c202c5SJeff Roberson                                (void ***)&ef->pcpu_stop, &count);
49850c202c5SJeff Roberson     /* Error just means there is no pcpu set to relocate. */
49950c202c5SJeff Roberson     if (error)
50050c202c5SJeff Roberson         return (0);
50150c202c5SJeff Roberson     count *= sizeof(void *);
50250c202c5SJeff Roberson     /*
50350c202c5SJeff Roberson      * Allocate space in the primary pcpu area.  Copy in our initialization
50450c202c5SJeff Roberson      * from the data section and then initialize all per-cpu storage from
50550c202c5SJeff Roberson      * that.
50650c202c5SJeff Roberson      */
50750c202c5SJeff Roberson     ef->pcpu_base = (Elf_Addr)(uintptr_t)dpcpu_alloc(count);
50850c202c5SJeff Roberson     if (ef->pcpu_base == (Elf_Addr)NULL)
50950c202c5SJeff Roberson         return (ENOSPC);
51050c202c5SJeff Roberson     memcpy((void *)ef->pcpu_base, (void *)ef->pcpu_start, count);
51150c202c5SJeff Roberson     dpcpu_copy((void *)ef->pcpu_base, count);
51250c202c5SJeff Roberson 
51350c202c5SJeff Roberson     return (0);
51450c202c5SJeff Roberson }
51550c202c5SJeff Roberson 
516eddfbb76SRobert Watson #ifdef VIMAGE
517eddfbb76SRobert Watson static int
518eddfbb76SRobert Watson parse_vnet(elf_file_t ef)
519eddfbb76SRobert Watson {
520eddfbb76SRobert Watson     int count;
521eddfbb76SRobert Watson     int error;
522eddfbb76SRobert Watson 
523eddfbb76SRobert Watson     ef->vnet_start = 0;
524eddfbb76SRobert Watson     ef->vnet_stop = 0;
525eddfbb76SRobert Watson     error = link_elf_lookup_set(&ef->lf, "vnet", (void ***)&ef->vnet_start,
526eddfbb76SRobert Watson                                (void ***)&ef->vnet_stop, &count);
527eddfbb76SRobert Watson     /* Error just means there is no vnet data set to relocate. */
528eddfbb76SRobert Watson     if (error)
529eddfbb76SRobert Watson         return (0);
530eddfbb76SRobert Watson     count *= sizeof(void *);
531eddfbb76SRobert Watson     /*
532eddfbb76SRobert Watson      * Allocate space in the primary vnet area.  Copy in our initialization
533eddfbb76SRobert Watson      * from the data section and then initialize all per-vnet storage from
534eddfbb76SRobert Watson      * that.
535eddfbb76SRobert Watson      */
536eddfbb76SRobert Watson     ef->vnet_base = (Elf_Addr)(uintptr_t)vnet_data_alloc(count);
537eddfbb76SRobert Watson     if (ef->vnet_base == (Elf_Addr)NULL)
538eddfbb76SRobert Watson         return (ENOSPC);
539eddfbb76SRobert Watson     memcpy((void *)ef->vnet_base, (void *)ef->vnet_start, count);
540eddfbb76SRobert Watson     vnet_data_copy((void *)ef->vnet_base, count);
541eddfbb76SRobert Watson 
542eddfbb76SRobert Watson     return (0);
543eddfbb76SRobert Watson }
544eddfbb76SRobert Watson #endif
545eddfbb76SRobert Watson 
54650c202c5SJeff Roberson static int
54754823af2SPeter Wemm link_elf_link_preload(linker_class_t cls,
548326e27d8SDoug Rabson 		      const char* filename, linker_file_t *result)
549de78ca7eSPeter Wemm {
550de78ca7eSPeter Wemm     caddr_t		modptr, baseptr, sizeptr, dynptr;
551de78ca7eSPeter Wemm     char		*type;
552de78ca7eSPeter Wemm     elf_file_t		ef;
553de78ca7eSPeter Wemm     linker_file_t	lf;
554de78ca7eSPeter Wemm     int			error;
555de78ca7eSPeter Wemm     vm_offset_t		dp;
556de78ca7eSPeter Wemm 
55754823af2SPeter Wemm     /* Look to see if we have the file preloaded */
558de78ca7eSPeter Wemm     modptr = preload_search_by_name(filename);
559de78ca7eSPeter Wemm     if (modptr == NULL)
56054823af2SPeter Wemm 	return ENOENT;
561de78ca7eSPeter Wemm 
562de78ca7eSPeter Wemm     type = (char *)preload_search_info(modptr, MODINFO_TYPE);
563de78ca7eSPeter Wemm     baseptr = preload_search_info(modptr, MODINFO_ADDR);
564de78ca7eSPeter Wemm     sizeptr = preload_search_info(modptr, MODINFO_SIZE);
565de78ca7eSPeter Wemm     dynptr = preload_search_info(modptr, MODINFO_METADATA|MODINFOMD_DYNAMIC);
56667db8b23SPeter Wemm     if (type == NULL ||
56767db8b23SPeter Wemm 	(strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE) " module") != 0 &&
56867db8b23SPeter Wemm 	 strcmp(type, "elf module") != 0))
569de78ca7eSPeter Wemm 	return (EFTYPE);
570de78ca7eSPeter Wemm     if (baseptr == NULL || sizeptr == NULL || dynptr == NULL)
571de78ca7eSPeter Wemm 	return (EINVAL);
572de78ca7eSPeter Wemm 
573326e27d8SDoug Rabson     lf = linker_make_file(filename, &link_elf_class);
574326e27d8SDoug Rabson     if (lf == NULL) {
575326e27d8SDoug Rabson 	return ENOMEM;
576326e27d8SDoug Rabson     }
577326e27d8SDoug Rabson 
578326e27d8SDoug Rabson     ef = (elf_file_t) lf;
579326e27d8SDoug Rabson     ef->preloaded = 1;
580ca65d5c7SPeter Wemm     ef->modptr = modptr;
581de78ca7eSPeter Wemm     ef->address = *(caddr_t *)baseptr;
582de78ca7eSPeter Wemm #ifdef SPARSE_MAPPING
583de78ca7eSPeter Wemm     ef->object = 0;
584de78ca7eSPeter Wemm #endif
585de78ca7eSPeter Wemm     dp = (vm_offset_t)ef->address + *(vm_offset_t *)dynptr;
586de78ca7eSPeter Wemm     ef->dynamic = (Elf_Dyn *)dp;
587de78ca7eSPeter Wemm     lf->address = ef->address;
588de78ca7eSPeter Wemm     lf->size = *(size_t *)sizeptr;
589de78ca7eSPeter Wemm 
590326e27d8SDoug Rabson     error = parse_dynamic(ef);
59150c202c5SJeff Roberson     if (error == 0)
59250c202c5SJeff Roberson         error = parse_dpcpu(ef);
593eddfbb76SRobert Watson #ifdef VIMAGE
594eddfbb76SRobert Watson     if (error == 0)
595eddfbb76SRobert Watson 	error = parse_vnet(ef);
596eddfbb76SRobert Watson #endif
597de78ca7eSPeter Wemm     if (error) {
59865a311fcSPoul-Henning Kamp 	linker_file_unload(lf, LINKER_UNLOAD_FORCE);
599de78ca7eSPeter Wemm 	return error;
600de78ca7eSPeter Wemm     }
6017251b4bfSJake Burkholder     link_elf_reloc_local(lf);
602de78ca7eSPeter Wemm     *result = lf;
603de78ca7eSPeter Wemm     return (0);
604de78ca7eSPeter Wemm }
605de78ca7eSPeter Wemm 
606de78ca7eSPeter Wemm static int
60754823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf)
60854823af2SPeter Wemm {
60954823af2SPeter Wemm     elf_file_t		ef;
61054823af2SPeter Wemm     int error;
61154823af2SPeter Wemm 
61254823af2SPeter Wemm     ef = (elf_file_t) lf;
61354823af2SPeter Wemm #if 0	/* this will be more trouble than it's worth for now */
61454823af2SPeter Wemm     for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) {
61554823af2SPeter Wemm 	if (dp->d_tag != DT_NEEDED)
61654823af2SPeter Wemm 	    continue;
61754823af2SPeter Wemm 	modname = ef->strtab + dp->d_un.d_val;
61854823af2SPeter Wemm 	error = linker_load_module(modname, lf);
61954823af2SPeter Wemm 	if (error)
62054823af2SPeter Wemm 	    goto out;
62154823af2SPeter Wemm     }
62254823af2SPeter Wemm #endif
62354823af2SPeter Wemm     error = relocate_file(ef);
62454823af2SPeter Wemm     if (error)
62554823af2SPeter Wemm 	return error;
62654823af2SPeter Wemm     (void)link_elf_preload_parse_symbols(ef);
62754823af2SPeter Wemm 
628c143d6c2SMarcel Moolenaar     return (link_elf_link_common_finish(lf));
62954823af2SPeter Wemm }
63054823af2SPeter Wemm 
63154823af2SPeter Wemm static int
632c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char* filename,
633c143d6c2SMarcel Moolenaar 	linker_file_t* result)
634a4f67738SDoug Rabson {
635a4f67738SDoug Rabson     struct nameidata nd;
636b40ce416SJulian Elischer     struct thread* td = curthread;	/* XXX */
637caab6e90SPeter Wemm     Elf_Ehdr *hdr;
638caab6e90SPeter Wemm     caddr_t firstpage;
639fe3db7c7SDoug Rabson     int nbytes, i;
640fe3db7c7SDoug Rabson     Elf_Phdr *phdr;
641fe3db7c7SDoug Rabson     Elf_Phdr *phlimit;
6421a330eb0SPeter Wemm     Elf_Phdr *segs[MAXSEGS];
643fe3db7c7SDoug Rabson     int nsegs;
644fe3db7c7SDoug Rabson     Elf_Phdr *phdyn;
645fe3db7c7SDoug Rabson     Elf_Phdr *phphdr;
646fe3db7c7SDoug Rabson     caddr_t mapbase;
647fe3db7c7SDoug Rabson     size_t mapsize;
648fe3db7c7SDoug Rabson     Elf_Off base_offset;
649fe3db7c7SDoug Rabson     Elf_Addr base_vaddr;
650fe3db7c7SDoug Rabson     Elf_Addr base_vlimit;
651a4f67738SDoug Rabson     int error = 0;
652e6796b67SKirk McKusick     int resid, flags;
653a4f67738SDoug Rabson     elf_file_t ef;
654a4f67738SDoug Rabson     linker_file_t lf;
655ca65d5c7SPeter Wemm     Elf_Shdr *shdr;
656ca65d5c7SPeter Wemm     int symtabindex;
657ca65d5c7SPeter Wemm     int symstrindex;
658ca65d5c7SPeter Wemm     int symcnt;
659ca65d5c7SPeter Wemm     int strcnt;
66062d615d5SJohn Baldwin     int vfslocked;
661ca65d5c7SPeter Wemm 
662ca65d5c7SPeter Wemm     shdr = NULL;
663ca65d5c7SPeter Wemm     lf = NULL;
664a4f67738SDoug Rabson 
66562d615d5SJohn Baldwin     NDINIT(&nd, LOOKUP, FOLLOW | MPSAFE, UIO_SYSSPACE, filename, td);
666e6796b67SKirk McKusick     flags = FREAD;
6679e223287SKonstantin Belousov     error = vn_open(&nd, &flags, 0, NULL);
668a4f67738SDoug Rabson     if (error)
669a4f67738SDoug Rabson 	return error;
67062d615d5SJohn Baldwin     vfslocked = NDHASGIANT(&nd);
671762e6b85SEivind Eklund     NDFREE(&nd, NDF_ONLY_PNBUF);
6721f49b573SRuslan Ermilov     if (nd.ni_vp->v_type != VREG) {
6731f49b573SRuslan Ermilov 	error = ENOEXEC;
6741f49b573SRuslan Ermilov 	firstpage = NULL;
6751f49b573SRuslan Ermilov 	goto out;
6761f49b573SRuslan Ermilov     }
677a3df768bSRobert Watson #ifdef MAC
67830d239bcSRobert Watson     error = mac_kld_check_load(curthread->td_ucred, nd.ni_vp);
679a3df768bSRobert Watson     if (error) {
680a3df768bSRobert Watson 	firstpage = NULL;
681a3df768bSRobert Watson 	goto out;
682a3df768bSRobert Watson     }
683a3df768bSRobert Watson #endif
684a4f67738SDoug Rabson 
685a4f67738SDoug Rabson     /*
686fe3db7c7SDoug Rabson      * Read the elf header from the file.
687a4f67738SDoug Rabson      */
688a163d034SWarner Losh     firstpage = malloc(PAGE_SIZE, M_LINKER, M_WAITOK);
689caab6e90SPeter Wemm     if (firstpage == NULL) {
690caab6e90SPeter Wemm 	error = ENOMEM;
691caab6e90SPeter Wemm 	goto out;
692caab6e90SPeter Wemm     }
693caab6e90SPeter Wemm     hdr = (Elf_Ehdr *)firstpage;
694caab6e90SPeter Wemm     error = vn_rdwr(UIO_READ, nd.ni_vp, firstpage, PAGE_SIZE, 0,
6959ca43589SRobert Watson 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
6969ca43589SRobert Watson 		    &resid, td);
697caab6e90SPeter Wemm     nbytes = PAGE_SIZE - resid;
698a4f67738SDoug Rabson     if (error)
699a4f67738SDoug Rabson 	goto out;
700a4f67738SDoug Rabson 
701caab6e90SPeter Wemm     if (!IS_ELF(*hdr)) {
702fe3db7c7SDoug Rabson 	error = ENOEXEC;
703a4f67738SDoug Rabson 	goto out;
704fe3db7c7SDoug Rabson     }
705fe3db7c7SDoug Rabson 
706caab6e90SPeter Wemm     if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS
707caab6e90SPeter Wemm       || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) {
708552f9f63SEdwin Groothuis 	link_elf_error(filename, "Unsupported file layout");
709fe3db7c7SDoug Rabson 	error = ENOEXEC;
710fe3db7c7SDoug Rabson 	goto out;
711fe3db7c7SDoug Rabson     }
712caab6e90SPeter Wemm     if (hdr->e_ident[EI_VERSION] != EV_CURRENT
713caab6e90SPeter Wemm       || hdr->e_version != EV_CURRENT) {
714552f9f63SEdwin Groothuis 	link_elf_error(filename, "Unsupported file version");
715fe3db7c7SDoug Rabson 	error = ENOEXEC;
716fe3db7c7SDoug Rabson 	goto out;
717fe3db7c7SDoug Rabson     }
718caab6e90SPeter Wemm     if (hdr->e_type != ET_EXEC && hdr->e_type != ET_DYN) {
719a1d7ce03SAttilio Rao 	error = ENOSYS;
720fe3db7c7SDoug Rabson 	goto out;
721fe3db7c7SDoug Rabson     }
722caab6e90SPeter Wemm     if (hdr->e_machine != ELF_TARG_MACH) {
723552f9f63SEdwin Groothuis 	link_elf_error(filename, "Unsupported machine");
724fe3db7c7SDoug Rabson 	error = ENOEXEC;
725fe3db7c7SDoug Rabson 	goto out;
726fe3db7c7SDoug Rabson     }
727a4f67738SDoug Rabson 
728a4f67738SDoug Rabson     /*
729fe3db7c7SDoug Rabson      * We rely on the program header being in the first page.  This is
730fe3db7c7SDoug Rabson      * not strictly required by the ABI specification, but it seems to
731fe3db7c7SDoug Rabson      * always true in practice.  And, it simplifies things considerably.
732a4f67738SDoug Rabson      */
733caab6e90SPeter Wemm     if (!((hdr->e_phentsize == sizeof(Elf_Phdr)) &&
734caab6e90SPeter Wemm 	  (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= PAGE_SIZE) &&
735caab6e90SPeter Wemm 	  (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= nbytes)))
736552f9f63SEdwin Groothuis 	link_elf_error(filename, "Unreadable program headers");
737fe3db7c7SDoug Rabson 
738fe3db7c7SDoug Rabson     /*
739fe3db7c7SDoug Rabson      * Scan the program header entries, and save key information.
740fe3db7c7SDoug Rabson      *
741fe3db7c7SDoug Rabson      * We rely on there being exactly two load segments, text and data,
742fe3db7c7SDoug Rabson      * in that order.
743fe3db7c7SDoug Rabson      */
744caab6e90SPeter Wemm     phdr = (Elf_Phdr *) (firstpage + hdr->e_phoff);
745caab6e90SPeter Wemm     phlimit = phdr + hdr->e_phnum;
746fe3db7c7SDoug Rabson     nsegs = 0;
747fe3db7c7SDoug Rabson     phdyn = NULL;
748fe3db7c7SDoug Rabson     phphdr = NULL;
749fe3db7c7SDoug Rabson     while (phdr < phlimit) {
750fe3db7c7SDoug Rabson 	switch (phdr->p_type) {
751fe3db7c7SDoug Rabson 
752fe3db7c7SDoug Rabson 	case PT_LOAD:
7531a330eb0SPeter Wemm 	    if (nsegs == MAXSEGS) {
754552f9f63SEdwin Groothuis 		link_elf_error(filename, "Too many sections");
755fe3db7c7SDoug Rabson 		error = ENOEXEC;
756fe3db7c7SDoug Rabson 		goto out;
757fe3db7c7SDoug Rabson 	    }
75887b1831fSPoul-Henning Kamp 	    /*
75987b1831fSPoul-Henning Kamp 	     * XXX: We just trust they come in right order ??
76087b1831fSPoul-Henning Kamp 	     */
761fe3db7c7SDoug Rabson 	    segs[nsegs] = phdr;
762fe3db7c7SDoug Rabson 	    ++nsegs;
763fe3db7c7SDoug Rabson 	    break;
764fe3db7c7SDoug Rabson 
765fe3db7c7SDoug Rabson 	case PT_PHDR:
766fe3db7c7SDoug Rabson 	    phphdr = phdr;
767fe3db7c7SDoug Rabson 	    break;
768fe3db7c7SDoug Rabson 
769fe3db7c7SDoug Rabson 	case PT_DYNAMIC:
770fe3db7c7SDoug Rabson 	    phdyn = phdr;
771fe3db7c7SDoug Rabson 	    break;
7729548091bSBoris Popov 
7739548091bSBoris Popov 	case PT_INTERP:
774a1d7ce03SAttilio Rao 	    error = ENOSYS;
7759548091bSBoris Popov 	    goto out;
776fe3db7c7SDoug Rabson 	}
777fe3db7c7SDoug Rabson 
778fe3db7c7SDoug Rabson 	++phdr;
779fe3db7c7SDoug Rabson     }
780fe3db7c7SDoug Rabson     if (phdyn == NULL) {
781552f9f63SEdwin Groothuis 	link_elf_error(filename, "Object is not dynamically-linked");
782fe3db7c7SDoug Rabson 	error = ENOEXEC;
783fe3db7c7SDoug Rabson 	goto out;
784fe3db7c7SDoug Rabson     }
7851a330eb0SPeter Wemm     if (nsegs == 0) {
786552f9f63SEdwin Groothuis 	link_elf_error(filename, "No sections");
78787b1831fSPoul-Henning Kamp 	error = ENOEXEC;
78887b1831fSPoul-Henning Kamp 	goto out;
78987b1831fSPoul-Henning Kamp     }
790fe3db7c7SDoug Rabson 
791fe3db7c7SDoug Rabson     /*
792fe3db7c7SDoug Rabson      * Allocate the entire address space of the object, to stake out our
793fe3db7c7SDoug Rabson      * contiguous region, and to establish the base address for relocation.
794fe3db7c7SDoug Rabson      */
795fe3db7c7SDoug Rabson     base_offset = trunc_page(segs[0]->p_offset);
796fe3db7c7SDoug Rabson     base_vaddr = trunc_page(segs[0]->p_vaddr);
7971a330eb0SPeter Wemm     base_vlimit = round_page(segs[nsegs - 1]->p_vaddr +
7981a330eb0SPeter Wemm 	segs[nsegs - 1]->p_memsz);
799fe3db7c7SDoug Rabson     mapsize = base_vlimit - base_vaddr;
800fe3db7c7SDoug Rabson 
801326e27d8SDoug Rabson     lf = linker_make_file(filename, &link_elf_class);
802326e27d8SDoug Rabson     if (!lf) {
803326e27d8SDoug Rabson 	error = ENOMEM;
804326e27d8SDoug Rabson 	goto out;
805326e27d8SDoug Rabson     }
806326e27d8SDoug Rabson 
807326e27d8SDoug Rabson     ef = (elf_file_t) lf;
808fe3db7c7SDoug Rabson #ifdef SPARSE_MAPPING
809fe3db7c7SDoug Rabson     ef->object = vm_object_allocate(OBJT_DEFAULT, mapsize >> PAGE_SHIFT);
810fe3db7c7SDoug Rabson     if (ef->object == NULL) {
811fe3db7c7SDoug Rabson 	error = ENOMEM;
812fe3db7c7SDoug Rabson 	goto out;
813fe3db7c7SDoug Rabson     }
814fe3db7c7SDoug Rabson     ef->address = (caddr_t) vm_map_min(kernel_map);
815fe3db7c7SDoug Rabson     error = vm_map_find(kernel_map, ef->object, 0,
816fe3db7c7SDoug Rabson 			(vm_offset_t *) &ef->address,
817fe3db7c7SDoug Rabson 			mapsize, 1,
818fe3db7c7SDoug Rabson 			VM_PROT_ALL, VM_PROT_ALL, 0);
819fe3db7c7SDoug Rabson     if (error) {
820fe3db7c7SDoug Rabson 	vm_object_deallocate(ef->object);
821326e27d8SDoug Rabson 	ef->object = 0;
822fe3db7c7SDoug Rabson 	goto out;
823fe3db7c7SDoug Rabson     }
824fe3db7c7SDoug Rabson #else
825a163d034SWarner Losh     ef->address = malloc(mapsize, M_LINKER, M_WAITOK);
826326e27d8SDoug Rabson     if (!ef->address) {
827326e27d8SDoug Rabson 	error = ENOMEM;
828326e27d8SDoug Rabson 	goto out;
829326e27d8SDoug Rabson     }
830fe3db7c7SDoug Rabson #endif
831fe3db7c7SDoug Rabson     mapbase = ef->address;
832a4f67738SDoug Rabson 
833a4f67738SDoug Rabson     /*
834a4f67738SDoug Rabson      * Read the text and data sections and zero the bss.
835a4f67738SDoug Rabson      */
8361a330eb0SPeter Wemm     for (i = 0; i < nsegs; i++) {
837fe3db7c7SDoug Rabson 	caddr_t segbase = mapbase + segs[i]->p_vaddr - base_vaddr;
838fe3db7c7SDoug Rabson 	error = vn_rdwr(UIO_READ, nd.ni_vp,
839fe3db7c7SDoug Rabson 			segbase, segs[i]->p_filesz, segs[i]->p_offset,
8409ca43589SRobert Watson 			UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
8419ca43589SRobert Watson 			&resid, td);
842fe3db7c7SDoug Rabson 	if (error) {
843a4f67738SDoug Rabson 	    goto out;
844a4f67738SDoug Rabson 	}
845fe3db7c7SDoug Rabson 	bzero(segbase + segs[i]->p_filesz,
846fe3db7c7SDoug Rabson 	      segs[i]->p_memsz - segs[i]->p_filesz);
84741673965SAlan Cox 
84841673965SAlan Cox #ifdef SPARSE_MAPPING
84941673965SAlan Cox 	/*
85041673965SAlan Cox 	 * Wire down the pages
85141673965SAlan Cox 	 */
8525d3043ceSAlan Cox 	error = vm_map_wire(kernel_map,
85341673965SAlan Cox 		    (vm_offset_t) segbase,
85441673965SAlan Cox 		    (vm_offset_t) segbase + segs[i]->p_memsz,
85541673965SAlan Cox 		    VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES);
8565d3043ceSAlan Cox 	if (error != KERN_SUCCESS) {
8575d3043ceSAlan Cox 	    error = ENOMEM;
8585d3043ceSAlan Cox 	    goto out;
8595d3043ceSAlan Cox 	}
86041673965SAlan Cox #endif
861fe3db7c7SDoug Rabson     }
862fe3db7c7SDoug Rabson 
863f99502a4SBrian Feldman #ifdef GPROF
864f99502a4SBrian Feldman     /* Update profiling information with the new text segment. */
86570f37788SJohn Baldwin     mtx_lock(&Giant);
866f99502a4SBrian Feldman     kmupetext((uintfptr_t)(mapbase + segs[0]->p_vaddr - base_vaddr +
867f99502a4SBrian Feldman 	segs[0]->p_memsz));
86870f37788SJohn Baldwin     mtx_unlock(&Giant);
869f99502a4SBrian Feldman #endif
870f99502a4SBrian Feldman 
871326e27d8SDoug Rabson     ef->dynamic = (Elf_Dyn *) (mapbase + phdyn->p_vaddr - base_vaddr);
872a4f67738SDoug Rabson 
873a4f67738SDoug Rabson     lf->address = ef->address;
874fe3db7c7SDoug Rabson     lf->size = mapsize;
875a4f67738SDoug Rabson 
876326e27d8SDoug Rabson     error = parse_dynamic(ef);
877ca65d5c7SPeter Wemm     if (error)
878de78ca7eSPeter Wemm 	goto out;
87950c202c5SJeff Roberson     error = parse_dpcpu(ef);
88050c202c5SJeff Roberson     if (error)
88150c202c5SJeff Roberson         goto out;
882eddfbb76SRobert Watson #ifdef VIMAGE
883eddfbb76SRobert Watson     error = parse_vnet(ef);
884eddfbb76SRobert Watson     if (error)
885eddfbb76SRobert Watson         goto out;
886eddfbb76SRobert Watson #endif
8877251b4bfSJake Burkholder     link_elf_reloc_local(lf);
8887251b4bfSJake Burkholder 
8894f7afc20SKonstantin Belousov     VOP_UNLOCK(nd.ni_vp, 0);
8907b9716baSIan Dowse     error = linker_load_dependencies(lf);
8914f7afc20SKonstantin Belousov     vn_lock(nd.ni_vp, LK_EXCLUSIVE | LK_RETRY);
892ca65d5c7SPeter Wemm     if (error)
893de78ca7eSPeter Wemm 	goto out;
89454823af2SPeter Wemm #if 0	/* this will be more trouble than it's worth for now */
89554823af2SPeter Wemm     for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) {
89654823af2SPeter Wemm 	if (dp->d_tag != DT_NEEDED)
89754823af2SPeter Wemm 	    continue;
89854823af2SPeter Wemm 	modname = ef->strtab + dp->d_un.d_val;
89954823af2SPeter Wemm 	error = linker_load_module(modname, lf);
90054823af2SPeter Wemm 	if (error)
90154823af2SPeter Wemm 	    goto out;
90254823af2SPeter Wemm     }
90354823af2SPeter Wemm #endif
904326e27d8SDoug Rabson     error = relocate_file(ef);
905ca65d5c7SPeter Wemm     if (error)
906ca65d5c7SPeter Wemm 	goto out;
907ca65d5c7SPeter Wemm 
908ca65d5c7SPeter Wemm     /* Try and load the symbol table if it's present.  (you can strip it!) */
909caab6e90SPeter Wemm     nbytes = hdr->e_shnum * hdr->e_shentsize;
910caab6e90SPeter Wemm     if (nbytes == 0 || hdr->e_shoff == 0)
911ca65d5c7SPeter Wemm 	goto nosyms;
912a163d034SWarner Losh     shdr = malloc(nbytes, M_LINKER, M_WAITOK | M_ZERO);
913ca65d5c7SPeter Wemm     if (shdr == NULL) {
914ca65d5c7SPeter Wemm 	error = ENOMEM;
915a4f67738SDoug Rabson 	goto out;
916a4f67738SDoug Rabson     }
917ca65d5c7SPeter Wemm     error = vn_rdwr(UIO_READ, nd.ni_vp,
918caab6e90SPeter Wemm 		    (caddr_t)shdr, nbytes, hdr->e_shoff,
9199ca43589SRobert Watson 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
9209ca43589SRobert Watson 		    &resid, td);
921ca65d5c7SPeter Wemm     if (error)
922ca65d5c7SPeter Wemm 	goto out;
923ca65d5c7SPeter Wemm     symtabindex = -1;
924ca65d5c7SPeter Wemm     symstrindex = -1;
925caab6e90SPeter Wemm     for (i = 0; i < hdr->e_shnum; i++) {
926ca65d5c7SPeter Wemm 	if (shdr[i].sh_type == SHT_SYMTAB) {
927ca65d5c7SPeter Wemm 	    symtabindex = i;
928ca65d5c7SPeter Wemm 	    symstrindex = shdr[i].sh_link;
929ca65d5c7SPeter Wemm 	}
930ca65d5c7SPeter Wemm     }
931ca65d5c7SPeter Wemm     if (symtabindex < 0 || symstrindex < 0)
932ca65d5c7SPeter Wemm 	goto nosyms;
933ca65d5c7SPeter Wemm 
934ca65d5c7SPeter Wemm     symcnt = shdr[symtabindex].sh_size;
935a163d034SWarner Losh     ef->symbase = malloc(symcnt, M_LINKER, M_WAITOK);
936ca65d5c7SPeter Wemm     strcnt = shdr[symstrindex].sh_size;
937a163d034SWarner Losh     ef->strbase = malloc(strcnt, M_LINKER, M_WAITOK);
938ca65d5c7SPeter Wemm 
939ca65d5c7SPeter Wemm     if (ef->symbase == NULL || ef->strbase == NULL) {
940ca65d5c7SPeter Wemm 	error = ENOMEM;
941ca65d5c7SPeter Wemm 	goto out;
942ca65d5c7SPeter Wemm     }
943ca65d5c7SPeter Wemm     error = vn_rdwr(UIO_READ, nd.ni_vp,
944ca65d5c7SPeter Wemm 		    ef->symbase, symcnt, shdr[symtabindex].sh_offset,
9459ca43589SRobert Watson 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
9469ca43589SRobert Watson 		    &resid, td);
947ca65d5c7SPeter Wemm     if (error)
948ca65d5c7SPeter Wemm 	goto out;
949ca65d5c7SPeter Wemm     error = vn_rdwr(UIO_READ, nd.ni_vp,
950ca65d5c7SPeter Wemm 		    ef->strbase, strcnt, shdr[symstrindex].sh_offset,
9519ca43589SRobert Watson 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
9529ca43589SRobert Watson 		    &resid, td);
953ca65d5c7SPeter Wemm     if (error)
954ca65d5c7SPeter Wemm 	goto out;
955ca65d5c7SPeter Wemm 
956ca65d5c7SPeter Wemm     ef->ddbsymcnt = symcnt / sizeof(Elf_Sym);
957ca65d5c7SPeter Wemm     ef->ddbsymtab = (const Elf_Sym *)ef->symbase;
958ca65d5c7SPeter Wemm     ef->ddbstrcnt = strcnt;
959ca65d5c7SPeter Wemm     ef->ddbstrtab = ef->strbase;
960ca65d5c7SPeter Wemm 
961c143d6c2SMarcel Moolenaar     error = link_elf_link_common_finish(lf);
962c143d6c2SMarcel Moolenaar     if (error)
963c143d6c2SMarcel Moolenaar 	goto out;
964326e27d8SDoug Rabson 
965ca65d5c7SPeter Wemm nosyms:
966a4f67738SDoug Rabson 
967a4f67738SDoug Rabson     *result = lf;
968a4f67738SDoug Rabson 
969a4f67738SDoug Rabson out:
970ca65d5c7SPeter Wemm     if (error && lf)
97165a311fcSPoul-Henning Kamp 	linker_file_unload(lf, LINKER_UNLOAD_FORCE);
972ca65d5c7SPeter Wemm     if (shdr)
973ca65d5c7SPeter Wemm 	free(shdr, M_LINKER);
974caab6e90SPeter Wemm     if (firstpage)
975caab6e90SPeter Wemm 	free(firstpage, M_LINKER);
97622db15c0SAttilio Rao     VOP_UNLOCK(nd.ni_vp, 0);
977a854ed98SJohn Baldwin     vn_close(nd.ni_vp, FREAD, td->td_ucred, td);
97862d615d5SJohn Baldwin     VFS_UNLOCK_GIANT(vfslocked);
979a4f67738SDoug Rabson 
980a4f67738SDoug Rabson     return error;
981a4f67738SDoug Rabson }
982a4f67738SDoug Rabson 
98350c202c5SJeff Roberson Elf_Addr
98450c202c5SJeff Roberson elf_relocaddr(linker_file_t lf, Elf_Addr x)
98550c202c5SJeff Roberson {
98650c202c5SJeff Roberson     elf_file_t ef;
98750c202c5SJeff Roberson 
98850c202c5SJeff Roberson     ef = (elf_file_t)lf;
98950c202c5SJeff Roberson     if (x >= ef->pcpu_start && x < ef->pcpu_stop)
99050c202c5SJeff Roberson 	return ((x - ef->pcpu_start) + ef->pcpu_base);
991eddfbb76SRobert Watson #ifdef VIMAGE
992eddfbb76SRobert Watson     if (x >= ef->vnet_start && x < ef->vnet_stop)
993eddfbb76SRobert Watson 	return ((x - ef->vnet_start) + ef->vnet_base);
994eddfbb76SRobert Watson #endif
99550c202c5SJeff Roberson     return (x);
99650c202c5SJeff Roberson }
99750c202c5SJeff Roberson 
99850c202c5SJeff Roberson 
999a4f67738SDoug Rabson static void
1000de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file)
1001a4f67738SDoug Rabson {
1002326e27d8SDoug Rabson     elf_file_t ef = (elf_file_t) file;
1003a4f67738SDoug Rabson 
100450c202c5SJeff Roberson     if (ef->pcpu_base) {
100550c202c5SJeff Roberson         dpcpu_free((void *)ef->pcpu_base, ef->pcpu_stop - ef->pcpu_start);
100650c202c5SJeff Roberson     }
1007eddfbb76SRobert Watson #ifdef VIMAGE
1008eddfbb76SRobert Watson     if (ef->vnet_base) {
1009eddfbb76SRobert Watson         vnet_data_free((void *)ef->vnet_base, ef->vnet_stop - ef->vnet_start);
1010eddfbb76SRobert Watson     }
1011eddfbb76SRobert Watson #endif
1012ecb01c64SMarcel Moolenaar #ifdef GDB
1013326e27d8SDoug Rabson     if (ef->gdb.l_ld) {
1014326e27d8SDoug Rabson 	GDB_STATE(RT_DELETE);
1015ed6c3888SPeter Wemm 	free((void *)(uintptr_t)ef->gdb.l_name, M_LINKER);
1016326e27d8SDoug Rabson 	link_elf_delete_gdb(&ef->gdb);
1017326e27d8SDoug Rabson 	GDB_STATE(RT_CONSISTENT);
1018326e27d8SDoug Rabson     }
1019326e27d8SDoug Rabson #endif
1020326e27d8SDoug Rabson 
10211aeb23cdSMarcel Moolenaar     /* Notify MD code that a module is being unloaded. */
10221aeb23cdSMarcel Moolenaar     elf_cpu_unload_file(file);
10231aeb23cdSMarcel Moolenaar 
1024326e27d8SDoug Rabson     if (ef->preloaded) {
102554823af2SPeter Wemm 	link_elf_unload_preload(file);
1026326e27d8SDoug Rabson 	return;
1027326e27d8SDoug Rabson     }
1028c143d6c2SMarcel Moolenaar 
1029fe3db7c7SDoug Rabson #ifdef SPARSE_MAPPING
1030fe3db7c7SDoug Rabson     if (ef->object) {
1031fe3db7c7SDoug Rabson 	vm_map_remove(kernel_map, (vm_offset_t) ef->address,
1032fe3db7c7SDoug Rabson 		      (vm_offset_t) ef->address
1033fe3db7c7SDoug Rabson 		      + (ef->object->size << PAGE_SHIFT));
1034fe3db7c7SDoug Rabson     }
1035fe3db7c7SDoug Rabson #else
1036ca65d5c7SPeter Wemm     if (ef->address)
1037a4f67738SDoug Rabson 	free(ef->address, M_LINKER);
1038fe3db7c7SDoug Rabson #endif
1039ca65d5c7SPeter Wemm     if (ef->symbase)
1040ca65d5c7SPeter Wemm 	free(ef->symbase, M_LINKER);
1041ca65d5c7SPeter Wemm     if (ef->strbase)
1042ca65d5c7SPeter Wemm 	free(ef->strbase, M_LINKER);
1043a2024a3eSJohn Birrell     if (ef->ctftab)
1044a2024a3eSJohn Birrell 	free(ef->ctftab, M_LINKER);
1045a2024a3eSJohn Birrell     if (ef->ctfoff)
1046a2024a3eSJohn Birrell 	free(ef->ctfoff, M_LINKER);
1047a2024a3eSJohn Birrell     if (ef->typoff)
1048a2024a3eSJohn Birrell 	free(ef->typoff, M_LINKER);
1049a4f67738SDoug Rabson }
1050a4f67738SDoug Rabson 
1051de78ca7eSPeter Wemm static void
105254823af2SPeter Wemm link_elf_unload_preload(linker_file_t file)
1053de78ca7eSPeter Wemm {
1054de78ca7eSPeter Wemm     if (file->filename)
1055de78ca7eSPeter Wemm 	preload_delete_name(file->filename);
1056de78ca7eSPeter Wemm }
1057de78ca7eSPeter Wemm 
1058fe3db7c7SDoug Rabson static const char *
1059757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info)
1060a4f67738SDoug Rabson {
1061fe3db7c7SDoug Rabson     const Elf_Sym *ref;
1062a4f67738SDoug Rabson 
1063aa855a59SPeter Wemm     if (ELF_R_SYM(r_info)) {
1064aa855a59SPeter Wemm 	ref = ef->symtab + ELF_R_SYM(r_info);
1065434bf653SPeter Wemm 	return ef->strtab + ref->st_name;
1066fe3db7c7SDoug Rabson     } else
1067fe3db7c7SDoug Rabson 	return NULL;
1068a4f67738SDoug Rabson }
1069a4f67738SDoug Rabson 
1070a4f67738SDoug Rabson static int
1071326e27d8SDoug Rabson relocate_file(elf_file_t ef)
1072a4f67738SDoug Rabson {
1073fe3db7c7SDoug Rabson     const Elf_Rel *rellim;
1074fe3db7c7SDoug Rabson     const Elf_Rel *rel;
1075fe3db7c7SDoug Rabson     const Elf_Rela *relalim;
1076fe3db7c7SDoug Rabson     const Elf_Rela *rela;
1077aa855a59SPeter Wemm     const char *symname;
1078a4f67738SDoug Rabson 
1079fe3db7c7SDoug Rabson     /* Perform relocations without addend if there are any: */
1080aa855a59SPeter Wemm     rel = ef->rel;
1081aa855a59SPeter Wemm     if (rel) {
1082cacd1f6aSBruce Evans 	rellim = (const Elf_Rel *)((const char *)ef->rel + ef->relsize);
1083aa855a59SPeter Wemm 	while (rel < rellim) {
1084e8855d4fSPeter Wemm 	    if (elf_reloc(&ef->lf, (Elf_Addr)ef->address, rel, ELF_RELOC_REL,
1085e8855d4fSPeter Wemm 			  elf_lookup)) {
1086aa855a59SPeter Wemm 		symname = symbol_name(ef, rel->r_info);
1087ef83592dSPeter Wemm 		printf("link_elf: symbol %s undefined\n", symname);
1088a4f67738SDoug Rabson 		return ENOENT;
1089a13ddfb6SPeter Wemm 	    }
1090aa855a59SPeter Wemm 	    rel++;
1091aa855a59SPeter Wemm 	}
1092a4f67738SDoug Rabson     }
1093a4f67738SDoug Rabson 
1094fe3db7c7SDoug Rabson     /* Perform relocations with addend if there are any: */
1095aa855a59SPeter Wemm     rela = ef->rela;
1096aa855a59SPeter Wemm     if (rela) {
1097cacd1f6aSBruce Evans 	relalim = (const Elf_Rela *)((const char *)ef->rela + ef->relasize);
1098aa855a59SPeter Wemm 	while (rela < relalim) {
1099e8855d4fSPeter Wemm 	    if (elf_reloc(&ef->lf, (Elf_Addr)ef->address, rela, ELF_RELOC_RELA,
1100e8855d4fSPeter Wemm 			  elf_lookup)) {
1101aa855a59SPeter Wemm 		symname = symbol_name(ef, rela->r_info);
1102ef83592dSPeter Wemm 		printf("link_elf: symbol %s undefined\n", symname);
1103fe3db7c7SDoug Rabson 		return ENOENT;
1104a13ddfb6SPeter Wemm 	    }
1105aa855a59SPeter Wemm 	    rela++;
1106aa855a59SPeter Wemm 	}
1107a4f67738SDoug Rabson     }
1108a4f67738SDoug Rabson 
1109fe3db7c7SDoug Rabson     /* Perform PLT relocations without addend if there are any: */
1110aa855a59SPeter Wemm     rel = ef->pltrel;
1111aa855a59SPeter Wemm     if (rel) {
1112cacd1f6aSBruce Evans 	rellim = (const Elf_Rel *)((const char *)ef->pltrel + ef->pltrelsize);
1113aa855a59SPeter Wemm 	while (rel < rellim) {
1114e8855d4fSPeter Wemm 	    if (elf_reloc(&ef->lf, (Elf_Addr)ef->address, rel, ELF_RELOC_REL,
1115e8855d4fSPeter Wemm 			  elf_lookup)) {
1116aa855a59SPeter Wemm 		symname = symbol_name(ef, rel->r_info);
1117ef83592dSPeter Wemm 		printf("link_elf: symbol %s undefined\n", symname);
1118fe3db7c7SDoug Rabson 		return ENOENT;
1119a13ddfb6SPeter Wemm 	    }
1120aa855a59SPeter Wemm 	    rel++;
1121aa855a59SPeter Wemm 	}
1122a4f67738SDoug Rabson     }
1123a4f67738SDoug Rabson 
1124fe3db7c7SDoug Rabson     /* Perform relocations with addend if there are any: */
1125aa855a59SPeter Wemm     rela = ef->pltrela;
1126aa855a59SPeter Wemm     if (rela) {
1127cacd1f6aSBruce Evans 	relalim = (const Elf_Rela *)((const char *)ef->pltrela + ef->pltrelasize);
1128aa855a59SPeter Wemm 	while (rela < relalim) {
1129e8855d4fSPeter Wemm 	    if (elf_reloc(&ef->lf, (Elf_Addr)ef->address, rela, ELF_RELOC_RELA,
1130e8855d4fSPeter Wemm 			  elf_lookup)) {
1131aa855a59SPeter Wemm 		symname = symbol_name(ef, rela->r_info);
1132ef83592dSPeter Wemm 		printf("link_elf: symbol %s undefined\n", symname);
1133fe3db7c7SDoug Rabson 		return ENOENT;
1134a13ddfb6SPeter Wemm 	    }
1135aa855a59SPeter Wemm 	    rela++;
1136aa855a59SPeter Wemm 	}
1137a4f67738SDoug Rabson     }
1138a4f67738SDoug Rabson 
1139a4f67738SDoug Rabson     return 0;
1140a4f67738SDoug Rabson }
1141a4f67738SDoug Rabson 
1142fe3db7c7SDoug Rabson /*
1143fe3db7c7SDoug Rabson  * Hash function for symbol table lookup.  Don't even think about changing
1144fe3db7c7SDoug Rabson  * this.  It is specified by the System V ABI.
1145fe3db7c7SDoug Rabson  */
1146fe3db7c7SDoug Rabson static unsigned long
1147fe3db7c7SDoug Rabson elf_hash(const char *name)
1148a4f67738SDoug Rabson {
1149fe3db7c7SDoug Rabson     const unsigned char *p = (const unsigned char *) name;
1150fe3db7c7SDoug Rabson     unsigned long h = 0;
1151fe3db7c7SDoug Rabson     unsigned long g;
1152a4f67738SDoug Rabson 
1153fe3db7c7SDoug Rabson     while (*p != '\0') {
1154fe3db7c7SDoug Rabson 	h = (h << 4) + *p++;
1155fe3db7c7SDoug Rabson 	if ((g = h & 0xf0000000) != 0)
1156fe3db7c7SDoug Rabson 	    h ^= g >> 24;
1157fe3db7c7SDoug Rabson 	h &= ~g;
1158a4f67738SDoug Rabson     }
1159fe3db7c7SDoug Rabson     return h;
1160fe3db7c7SDoug Rabson }
1161a4f67738SDoug Rabson 
116237c84183SPoul-Henning Kamp static int
1163d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char* name, c_linker_sym_t* sym)
1164a4f67738SDoug Rabson {
1165326e27d8SDoug Rabson     elf_file_t ef = (elf_file_t) lf;
1166fe3db7c7SDoug Rabson     unsigned long symnum;
11672d636ab0SPeter Wemm     const Elf_Sym* symp;
11682d636ab0SPeter Wemm     const char *strp;
1169fe3db7c7SDoug Rabson     unsigned long hash;
1170a4f67738SDoug Rabson     int i;
1171a4f67738SDoug Rabson 
1172f6494f2eSJohn Baldwin     /* If we don't have a hash, bail. */
1173f6494f2eSJohn Baldwin     if (ef->buckets == NULL || ef->nbuckets == 0) {
1174f6494f2eSJohn Baldwin 	printf("link_elf_lookup_symbol: missing symbol hash table\n");
1175f6494f2eSJohn Baldwin 	return ENOENT;
1176f6494f2eSJohn Baldwin     }
1177f6494f2eSJohn Baldwin 
11782d636ab0SPeter Wemm     /* First, search hashed global symbols */
1179fe3db7c7SDoug Rabson     hash = elf_hash(name);
1180fe3db7c7SDoug Rabson     symnum = ef->buckets[hash % ef->nbuckets];
1181fe3db7c7SDoug Rabson 
1182fe3db7c7SDoug Rabson     while (symnum != STN_UNDEF) {
1183fe3db7c7SDoug Rabson 	if (symnum >= ef->nchains) {
1184fe3db7c7SDoug Rabson 	    printf("link_elf_lookup_symbol: corrupt symbol table\n");
1185fe3db7c7SDoug Rabson 	    return ENOENT;
1186a4f67738SDoug Rabson 	}
1187fe3db7c7SDoug Rabson 
1188fe3db7c7SDoug Rabson 	symp = ef->symtab + symnum;
1189fe3db7c7SDoug Rabson 	if (symp->st_name == 0) {
1190fe3db7c7SDoug Rabson 	    printf("link_elf_lookup_symbol: corrupt symbol table\n");
1191fe3db7c7SDoug Rabson 	    return ENOENT;
1192fe3db7c7SDoug Rabson 	}
1193fe3db7c7SDoug Rabson 
1194fe3db7c7SDoug Rabson 	strp = ef->strtab + symp->st_name;
1195fe3db7c7SDoug Rabson 
1196fe3db7c7SDoug Rabson 	if (strcmp(name, strp) == 0) {
1197fe3db7c7SDoug Rabson 	    if (symp->st_shndx != SHN_UNDEF ||
1198fe3db7c7SDoug Rabson 		(symp->st_value != 0 &&
1199fe3db7c7SDoug Rabson 		 ELF_ST_TYPE(symp->st_info) == STT_FUNC)) {
1200d254af07SMatthew Dillon 		*sym = (c_linker_sym_t) symp;
1201fe3db7c7SDoug Rabson 		return 0;
1202fe3db7c7SDoug Rabson 	    } else
1203fe3db7c7SDoug Rabson 		return ENOENT;
1204fe3db7c7SDoug Rabson 	}
1205fe3db7c7SDoug Rabson 
1206fe3db7c7SDoug Rabson 	symnum = ef->chains[symnum];
1207a4f67738SDoug Rabson     }
1208a4f67738SDoug Rabson 
12092d636ab0SPeter Wemm     /* If we have not found it, look at the full table (if loaded) */
12102d636ab0SPeter Wemm     if (ef->symtab == ef->ddbsymtab)
12112d636ab0SPeter Wemm 	return ENOENT;
12122d636ab0SPeter Wemm 
12132d636ab0SPeter Wemm     /* Exhaustive search */
12142d636ab0SPeter Wemm     for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
12152d636ab0SPeter Wemm 	strp = ef->ddbstrtab + symp->st_name;
12162d636ab0SPeter Wemm 	if (strcmp(name, strp) == 0) {
12172d636ab0SPeter Wemm 	    if (symp->st_shndx != SHN_UNDEF ||
12182d636ab0SPeter Wemm 		(symp->st_value != 0 &&
12192d636ab0SPeter Wemm 		 ELF_ST_TYPE(symp->st_info) == STT_FUNC)) {
1220d254af07SMatthew Dillon 		*sym = (c_linker_sym_t) symp;
12212d636ab0SPeter Wemm 		return 0;
12222d636ab0SPeter Wemm 	    } else
12232d636ab0SPeter Wemm 		return ENOENT;
12242d636ab0SPeter Wemm 	}
12252d636ab0SPeter Wemm     }
12262d636ab0SPeter Wemm 
1227a4f67738SDoug Rabson     return ENOENT;
1228a4f67738SDoug Rabson }
1229a4f67738SDoug Rabson 
1230de78ca7eSPeter Wemm static int
1231fe08c21aSMatthew Dillon link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, linker_symval_t* symval)
1232a4f67738SDoug Rabson {
1233326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) lf;
12348aef1712SMatthew Dillon 	const Elf_Sym* es = (const Elf_Sym*) sym;
1235a4f67738SDoug Rabson 
12365cf87418SMarcel Moolenaar 	if (es >= ef->symtab && es < (ef->symtab + ef->nchains)) {
1237a4f67738SDoug Rabson 	    symval->name = ef->strtab + es->st_name;
1238fe3db7c7SDoug Rabson 	    symval->value = (caddr_t) ef->address + es->st_value;
1239a4f67738SDoug Rabson 	    symval->size = es->st_size;
1240de78ca7eSPeter Wemm 	    return 0;
1241a4f67738SDoug Rabson 	}
12422d636ab0SPeter Wemm 	if (ef->symtab == ef->ddbsymtab)
12432d636ab0SPeter Wemm 	    return ENOENT;
12445cf87418SMarcel Moolenaar 	if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) {
12452d636ab0SPeter Wemm 	    symval->name = ef->ddbstrtab + es->st_name;
12462d636ab0SPeter Wemm 	    symval->value = (caddr_t) ef->address + es->st_value;
12472d636ab0SPeter Wemm 	    symval->size = es->st_size;
12482d636ab0SPeter Wemm 	    return 0;
12492d636ab0SPeter Wemm 	}
12502d636ab0SPeter Wemm 	return ENOENT;
12512d636ab0SPeter Wemm }
1252a4f67738SDoug Rabson 
1253a4f67738SDoug Rabson static int
1254a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value,
1255d254af07SMatthew Dillon 		       c_linker_sym_t* sym, long* diffp)
1256a4f67738SDoug Rabson {
1257326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) lf;
1258586453feSBruce Evans 	u_long off = (uintptr_t) (void *) value;
1259a4f67738SDoug Rabson 	u_long diff = off;
1260586453feSBruce Evans 	u_long st_value;
1261fe3db7c7SDoug Rabson 	const Elf_Sym* es;
1262fe3db7c7SDoug Rabson 	const Elf_Sym* best = 0;
1263a4f67738SDoug Rabson 	int i;
1264a4f67738SDoug Rabson 
12652d636ab0SPeter Wemm 	for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) {
1266a4f67738SDoug Rabson 		if (es->st_name == 0)
1267a4f67738SDoug Rabson 			continue;
1268586453feSBruce Evans 		st_value = es->st_value + (uintptr_t) (void *) ef->address;
1269b5abfb70SPeter Wemm 		if (off >= st_value) {
1270b5abfb70SPeter Wemm 			if (off - st_value < diff) {
1271b5abfb70SPeter Wemm 				diff = off - st_value;
1272a4f67738SDoug Rabson 				best = es;
1273a4f67738SDoug Rabson 				if (diff == 0)
1274a4f67738SDoug Rabson 					break;
1275b5abfb70SPeter Wemm 			} else if (off - st_value == diff) {
1276a4f67738SDoug Rabson 				best = es;
1277a4f67738SDoug Rabson 			}
1278a4f67738SDoug Rabson 		}
1279a4f67738SDoug Rabson 	}
1280a4f67738SDoug Rabson 	if (best == 0)
1281a4f67738SDoug Rabson 		*diffp = off;
1282a4f67738SDoug Rabson 	else
1283a4f67738SDoug Rabson 		*diffp = diff;
1284d254af07SMatthew Dillon 	*sym = (c_linker_sym_t) best;
1285a4f67738SDoug Rabson 
1286a4f67738SDoug Rabson 	return 0;
1287a4f67738SDoug Rabson }
1288f41325dbSPeter Wemm 
1289f41325dbSPeter Wemm /*
1290f41325dbSPeter Wemm  * Look up a linker set on an ELF system.
1291f41325dbSPeter Wemm  */
1292f41325dbSPeter Wemm static int
1293f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name,
1294f41325dbSPeter Wemm 		    void ***startp, void ***stopp, int *countp)
1295f41325dbSPeter Wemm {
1296f41325dbSPeter Wemm 	c_linker_sym_t sym;
1297f41325dbSPeter Wemm 	linker_symval_t symval;
1298f41325dbSPeter Wemm 	char *setsym;
1299f41325dbSPeter Wemm 	void **start, **stop;
1300f41325dbSPeter Wemm 	int len, error = 0, count;
1301f41325dbSPeter Wemm 
1302f41325dbSPeter Wemm 	len = strlen(name) + sizeof("__start_set_"); /* sizeof includes \0 */
1303a163d034SWarner Losh 	setsym = malloc(len, M_LINKER, M_WAITOK);
1304f41325dbSPeter Wemm 	if (setsym == NULL)
1305f41325dbSPeter Wemm 		return ENOMEM;
1306f41325dbSPeter Wemm 
1307f41325dbSPeter Wemm 	/* get address of first entry */
1308f41325dbSPeter Wemm 	snprintf(setsym, len, "%s%s", "__start_set_", name);
1309f41325dbSPeter Wemm 	error = link_elf_lookup_symbol(lf, setsym, &sym);
1310f41325dbSPeter Wemm 	if (error)
1311f41325dbSPeter Wemm 		goto out;
1312f41325dbSPeter Wemm 	link_elf_symbol_values(lf, sym, &symval);
1313f41325dbSPeter Wemm 	if (symval.value == 0) {
1314f41325dbSPeter Wemm 		error = ESRCH;
1315f41325dbSPeter Wemm 		goto out;
1316f41325dbSPeter Wemm 	}
1317f41325dbSPeter Wemm 	start = (void **)symval.value;
1318f41325dbSPeter Wemm 
1319f41325dbSPeter Wemm 	/* get address of last entry */
1320f41325dbSPeter Wemm 	snprintf(setsym, len, "%s%s", "__stop_set_", name);
1321f41325dbSPeter Wemm 	error = link_elf_lookup_symbol(lf, setsym, &sym);
1322f41325dbSPeter Wemm 	if (error)
1323f41325dbSPeter Wemm 		goto out;
1324f41325dbSPeter Wemm 	link_elf_symbol_values(lf, sym, &symval);
1325f41325dbSPeter Wemm 	if (symval.value == 0) {
1326f41325dbSPeter Wemm 		error = ESRCH;
1327f41325dbSPeter Wemm 		goto out;
1328f41325dbSPeter Wemm 	}
1329f41325dbSPeter Wemm 	stop = (void **)symval.value;
1330f41325dbSPeter Wemm 
1331f41325dbSPeter Wemm 	/* and the number of entries */
1332f41325dbSPeter Wemm 	count = stop - start;
1333f41325dbSPeter Wemm 
1334f41325dbSPeter Wemm 	/* and copy out */
1335f41325dbSPeter Wemm 	if (startp)
1336f41325dbSPeter Wemm 		*startp = start;
1337f41325dbSPeter Wemm 	if (stopp)
1338f41325dbSPeter Wemm 		*stopp = stop;
1339f41325dbSPeter Wemm 	if (countp)
1340f41325dbSPeter Wemm 		*countp = count;
1341f41325dbSPeter Wemm 
1342f41325dbSPeter Wemm out:
1343f41325dbSPeter Wemm 	free(setsym, M_LINKER);
1344f41325dbSPeter Wemm 	return error;
1345f41325dbSPeter Wemm }
1346bb9fe9ddSBrian Feldman 
1347bb9fe9ddSBrian Feldman static int
1348bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file,
1349bb9fe9ddSBrian Feldman   int (*callback)(const char *, void *), void *opaque) {
1350bb9fe9ddSBrian Feldman     elf_file_t ef = (elf_file_t)file;
1351bb9fe9ddSBrian Feldman     const Elf_Sym* symp;
1352bb9fe9ddSBrian Feldman     int i, error;
1353bb9fe9ddSBrian Feldman 
1354bb9fe9ddSBrian Feldman     /* Exhaustive search */
1355bb9fe9ddSBrian Feldman     for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1356bb9fe9ddSBrian Feldman 	if (symp->st_value != 0 &&
1357bb9fe9ddSBrian Feldman 	    ELF_ST_TYPE(symp->st_info) == STT_FUNC) {
1358bb9fe9ddSBrian Feldman 		error = callback(ef->ddbstrtab + symp->st_name, opaque);
1359bb9fe9ddSBrian Feldman 		if (error)
1360bb9fe9ddSBrian Feldman 		    return (error);
1361bb9fe9ddSBrian Feldman 	}
1362bb9fe9ddSBrian Feldman     }
1363bb9fe9ddSBrian Feldman     return (0);
1364bb9fe9ddSBrian Feldman }
136584201059SMarcel Moolenaar 
1366a2024a3eSJohn Birrell static int
1367a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file,
1368a2024a3eSJohn Birrell     linker_function_nameval_callback_t callback, void *opaque)
1369a2024a3eSJohn Birrell {
1370a2024a3eSJohn Birrell 	linker_symval_t symval;
1371a2024a3eSJohn Birrell 	elf_file_t ef = (elf_file_t)file;
1372a2024a3eSJohn Birrell 	const Elf_Sym* symp;
1373a2024a3eSJohn Birrell 	int i, error;
1374a2024a3eSJohn Birrell 
1375a2024a3eSJohn Birrell 	/* Exhaustive search */
1376a2024a3eSJohn Birrell 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1377a2024a3eSJohn Birrell 		if (symp->st_value != 0 &&
1378a2024a3eSJohn Birrell 		    ELF_ST_TYPE(symp->st_info) == STT_FUNC) {
1379a2024a3eSJohn Birrell 			error = link_elf_symbol_values(file, (c_linker_sym_t) symp, &symval);
1380a2024a3eSJohn Birrell 			if (error)
1381a2024a3eSJohn Birrell 				return (error);
1382a2024a3eSJohn Birrell 			error = callback(file, i, &symval, opaque);
1383a2024a3eSJohn Birrell 			if (error)
1384a2024a3eSJohn Birrell 				return (error);
1385a2024a3eSJohn Birrell 		}
1386a2024a3eSJohn Birrell 	}
1387a2024a3eSJohn Birrell 	return (0);
1388a2024a3eSJohn Birrell }
1389a2024a3eSJohn Birrell 
139084201059SMarcel Moolenaar #ifdef __ia64__
139184201059SMarcel Moolenaar /*
139284201059SMarcel Moolenaar  * Each KLD has its own GP. The GP value for each load module is given by
139384201059SMarcel Moolenaar  * DT_PLTGOT on ia64. We need GP to construct function descriptors, but
139484201059SMarcel Moolenaar  * don't have direct access to the ELF file structure. The link_elf_get_gp()
139584201059SMarcel Moolenaar  * function returns the GP given a pointer to a generic linker file struct.
139684201059SMarcel Moolenaar  */
139784201059SMarcel Moolenaar Elf_Addr
139884201059SMarcel Moolenaar link_elf_get_gp(linker_file_t lf)
139984201059SMarcel Moolenaar {
140084201059SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
140184201059SMarcel Moolenaar 	return (Elf_Addr)ef->got;
140284201059SMarcel Moolenaar }
140384201059SMarcel Moolenaar #endif
1404d297ad16SMarcel Moolenaar 
1405169d513cSJake Burkholder const Elf_Sym *
1406757686b1SMarcel Moolenaar elf_get_sym(linker_file_t lf, Elf_Size symidx)
1407169d513cSJake Burkholder {
1408169d513cSJake Burkholder 	elf_file_t ef = (elf_file_t)lf;
1409169d513cSJake Burkholder 
1410169d513cSJake Burkholder 	if (symidx >= ef->nchains)
1411169d513cSJake Burkholder 		return (NULL);
1412169d513cSJake Burkholder 	return (ef->symtab + symidx);
1413169d513cSJake Burkholder }
1414169d513cSJake Burkholder 
141547f75012SMarcel Moolenaar const char *
1416757686b1SMarcel Moolenaar elf_get_symname(linker_file_t lf, Elf_Size symidx)
141747f75012SMarcel Moolenaar {
141847f75012SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
141947f75012SMarcel Moolenaar 	const Elf_Sym *sym;
142047f75012SMarcel Moolenaar 
142147f75012SMarcel Moolenaar 	if (symidx >= ef->nchains)
142247f75012SMarcel Moolenaar 		return (NULL);
142347f75012SMarcel Moolenaar 	sym = ef->symtab + symidx;
142447f75012SMarcel Moolenaar 	return (ef->strtab + sym->st_name);
142547f75012SMarcel Moolenaar }
142647f75012SMarcel Moolenaar 
1427d297ad16SMarcel Moolenaar /*
1428d297ad16SMarcel Moolenaar  * Symbol lookup function that can be used when the symbol index is known (ie
1429d297ad16SMarcel Moolenaar  * in relocations). It uses the symbol index instead of doing a fully fledged
1430d297ad16SMarcel Moolenaar  * hash table based lookup when such is valid. For example for local symbols.
1431d297ad16SMarcel Moolenaar  * This is not only more efficient, it's also more correct. It's not always
1432d297ad16SMarcel Moolenaar  * the case that the symbol can be found through the hash table.
1433d297ad16SMarcel Moolenaar  */
1434e8855d4fSPeter Wemm static Elf_Addr
1435757686b1SMarcel Moolenaar elf_lookup(linker_file_t lf, Elf_Size symidx, int deps)
1436d297ad16SMarcel Moolenaar {
1437d297ad16SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
1438d297ad16SMarcel Moolenaar 	const Elf_Sym *sym;
1439d297ad16SMarcel Moolenaar 	const char *symbol;
1440d297ad16SMarcel Moolenaar 
1441d297ad16SMarcel Moolenaar 	/* Don't even try to lookup the symbol if the index is bogus. */
1442d297ad16SMarcel Moolenaar 	if (symidx >= ef->nchains)
1443d297ad16SMarcel Moolenaar 		return (0);
1444d297ad16SMarcel Moolenaar 
1445d297ad16SMarcel Moolenaar 	sym = ef->symtab + symidx;
1446d297ad16SMarcel Moolenaar 
1447d297ad16SMarcel Moolenaar 	/*
1448d297ad16SMarcel Moolenaar 	 * Don't do a full lookup when the symbol is local. It may even
1449d297ad16SMarcel Moolenaar 	 * fail because it may not be found through the hash table.
1450d297ad16SMarcel Moolenaar 	 */
1451d297ad16SMarcel Moolenaar 	if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) {
1452d297ad16SMarcel Moolenaar 		/* Force lookup failure when we have an insanity. */
1453d297ad16SMarcel Moolenaar 		if (sym->st_shndx == SHN_UNDEF || sym->st_value == 0)
1454d297ad16SMarcel Moolenaar 			return (0);
1455d297ad16SMarcel Moolenaar 		return ((Elf_Addr)ef->address + sym->st_value);
1456d297ad16SMarcel Moolenaar 	}
1457d297ad16SMarcel Moolenaar 
1458d297ad16SMarcel Moolenaar 	/*
1459d297ad16SMarcel Moolenaar 	 * XXX we can avoid doing a hash table based lookup for global
1460d297ad16SMarcel Moolenaar 	 * symbols as well. This however is not always valid, so we'll
1461d297ad16SMarcel Moolenaar 	 * just do it the hard way for now. Performance tweaks can
1462d297ad16SMarcel Moolenaar 	 * always be added.
1463d297ad16SMarcel Moolenaar 	 */
1464d297ad16SMarcel Moolenaar 
1465d297ad16SMarcel Moolenaar 	symbol = ef->strtab + sym->st_name;
1466d297ad16SMarcel Moolenaar 
1467d297ad16SMarcel Moolenaar 	/* Force a lookup failure if the symbol name is bogus. */
1468d297ad16SMarcel Moolenaar 	if (*symbol == 0)
1469d297ad16SMarcel Moolenaar 		return (0);
1470d297ad16SMarcel Moolenaar 
1471d297ad16SMarcel Moolenaar 	return ((Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps));
1472d297ad16SMarcel Moolenaar }
14737251b4bfSJake Burkholder 
14747251b4bfSJake Burkholder static void
14757251b4bfSJake Burkholder link_elf_reloc_local(linker_file_t lf)
14767251b4bfSJake Burkholder {
14777251b4bfSJake Burkholder     const Elf_Rel *rellim;
14787251b4bfSJake Burkholder     const Elf_Rel *rel;
14797251b4bfSJake Burkholder     const Elf_Rela *relalim;
14807251b4bfSJake Burkholder     const Elf_Rela *rela;
14817251b4bfSJake Burkholder     elf_file_t ef = (elf_file_t)lf;
14827251b4bfSJake Burkholder 
14837251b4bfSJake Burkholder     /* Perform relocations without addend if there are any: */
14847251b4bfSJake Burkholder     if ((rel = ef->rel) != NULL) {
14857251b4bfSJake Burkholder 	rellim = (const Elf_Rel *)((const char *)ef->rel + ef->relsize);
14867251b4bfSJake Burkholder 	while (rel < rellim) {
1487e8855d4fSPeter Wemm 	    elf_reloc_local(lf, (Elf_Addr)ef->address, rel, ELF_RELOC_REL,
1488e8855d4fSPeter Wemm 			    elf_lookup);
14897251b4bfSJake Burkholder 	    rel++;
14907251b4bfSJake Burkholder 	}
14917251b4bfSJake Burkholder     }
14927251b4bfSJake Burkholder 
14937251b4bfSJake Burkholder     /* Perform relocations with addend if there are any: */
14947251b4bfSJake Burkholder     if ((rela = ef->rela) != NULL) {
14957251b4bfSJake Burkholder 	relalim = (const Elf_Rela *)((const char *)ef->rela + ef->relasize);
14967251b4bfSJake Burkholder 	while (rela < relalim) {
1497e8855d4fSPeter Wemm 	    elf_reloc_local(lf, (Elf_Addr)ef->address, rela, ELF_RELOC_RELA,
1498e8855d4fSPeter Wemm 			    elf_lookup);
14997251b4bfSJake Burkholder 	    rela++;
15007251b4bfSJake Burkholder 	}
15017251b4bfSJake Burkholder     }
15027251b4bfSJake Burkholder }
150300a5db46SStacey Son 
150400a5db46SStacey Son static long
150500a5db46SStacey Son link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab)
150600a5db46SStacey Son {
150700a5db46SStacey Son     elf_file_t ef = (elf_file_t)lf;
150800a5db46SStacey Son 
150900a5db46SStacey Son     *symtab = ef->ddbsymtab;
151000a5db46SStacey Son 
151100a5db46SStacey Son     if (*symtab == NULL)
151200a5db46SStacey Son         return (0);
151300a5db46SStacey Son 
151400a5db46SStacey Son     return (ef->ddbsymcnt);
151500a5db46SStacey Son }
151600a5db46SStacey Son 
151700a5db46SStacey Son static long
151800a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab)
151900a5db46SStacey Son {
152000a5db46SStacey Son     elf_file_t ef = (elf_file_t)lf;
152100a5db46SStacey Son 
152200a5db46SStacey Son     *strtab = ef->ddbstrtab;
152300a5db46SStacey Son 
152400a5db46SStacey Son     if (*strtab == NULL)
152500a5db46SStacey Son         return (0);
152600a5db46SStacey Son 
152700a5db46SStacey Son     return (ef->ddbstrcnt);
152800a5db46SStacey Son }
1529