xref: /freebsd/libexec/rtld-elf/i386/reloc.c (revision 076ad2f8) 
1 /*-
2  * Copyright 1996, 1997, 1998, 1999 John D. Polstra.
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
15  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
18  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  *
25  * $FreeBSD$
26  */
27 
28 /*
29  * Dynamic linker for ELF.
30  *
31  * John Polstra <jdp@polstra.com>.
32  */
33 
34 #include <sys/param.h>
35 #include <sys/mman.h>
36 #include <machine/segments.h>
37 #include <machine/sysarch.h>
38 
39 #include <dlfcn.h>
40 #include <err.h>
41 #include <errno.h>
42 #include <fcntl.h>
43 #include <stdarg.h>
44 #include <stdio.h>
45 #include <stdlib.h>
46 #include <string.h>
47 #include <unistd.h>
48 
49 #include "debug.h"
50 #include "rtld.h"
51 #include "rtld_tls.h"
52 
53 /*
54  * Process the special R_386_COPY relocations in the main program.  These
55  * copy data from a shared object into a region in the main program's BSS
56  * segment.
57  *
58  * Returns 0 on success, -1 on failure.
59  */
60 int
61 do_copy_relocations(Obj_Entry *dstobj)
62 {
63     const Elf_Rel *rellim;
64     const Elf_Rel *rel;
65 
66     assert(dstobj->mainprog);	/* COPY relocations are invalid elsewhere */
67 
68     rellim = (const Elf_Rel *) ((caddr_t) dstobj->rel + dstobj->relsize);
69     for (rel = dstobj->rel;  rel < rellim;  rel++) {
70 	if (ELF_R_TYPE(rel->r_info) == R_386_COPY) {
71 	    void *dstaddr;
72 	    const Elf_Sym *dstsym;
73 	    const char *name;
74 	    size_t size;
75 	    const void *srcaddr;
76 	    const Elf_Sym *srcsym;
77 	    const Obj_Entry *srcobj, *defobj;
78 	    SymLook req;
79 	    int res;
80 
81 	    dstaddr = (void *) (dstobj->relocbase + rel->r_offset);
82 	    dstsym = dstobj->symtab + ELF_R_SYM(rel->r_info);
83 	    name = dstobj->strtab + dstsym->st_name;
84 	    size = dstsym->st_size;
85 	    symlook_init(&req, name);
86 	    req.ventry = fetch_ventry(dstobj, ELF_R_SYM(rel->r_info));
87 	    req.flags = SYMLOOK_EARLY;
88 
89 	    for (srcobj = globallist_next(dstobj);  srcobj != NULL;
90 	      srcobj = globallist_next(srcobj)) {
91 		res = symlook_obj(&req, srcobj);
92 		if (res == 0) {
93 		    srcsym = req.sym_out;
94 		    defobj = req.defobj_out;
95 		    break;
96 		}
97 	    }
98 
99 	    if (srcobj == NULL) {
100 		_rtld_error("Undefined symbol \"%s\" referenced from COPY"
101 		  " relocation in %s", name, dstobj->path);
102 		return -1;
103 	    }
104 
105 	    srcaddr = (const void *) (defobj->relocbase + srcsym->st_value);
106 	    memcpy(dstaddr, srcaddr, size);
107 	}
108     }
109 
110     return 0;
111 }
112 
113 /* Initialize the special GOT entries. */
114 void
115 init_pltgot(Obj_Entry *obj)
116 {
117     if (obj->pltgot != NULL) {
118 	obj->pltgot[1] = (Elf_Addr) obj;
119 	obj->pltgot[2] = (Elf_Addr) &_rtld_bind_start;
120     }
121 }
122 
123 /* Process the non-PLT relocations. */
124 int
125 reloc_non_plt(Obj_Entry *obj, Obj_Entry *obj_rtld, int flags,
126     RtldLockState *lockstate)
127 {
128 	const Elf_Rel *rellim;
129 	const Elf_Rel *rel;
130 	SymCache *cache;
131 	const Elf_Sym *def;
132 	const Obj_Entry *defobj;
133 	Elf_Addr *where, symval, add;
134 	int r;
135 
136 	r = -1;
137 	/*
138 	 * The dynamic loader may be called from a thread, we have
139 	 * limited amounts of stack available so we cannot use alloca().
140 	 */
141 	if (obj != obj_rtld) {
142 		cache = calloc(obj->dynsymcount, sizeof(SymCache));
143 		/* No need to check for NULL here */
144 	} else
145 		cache = NULL;
146 
147 	rellim = (const Elf_Rel *)((caddr_t) obj->rel + obj->relsize);
148 	for (rel = obj->rel;  rel < rellim;  rel++) {
149 		switch (ELF_R_TYPE(rel->r_info)) {
150 		case R_386_32:
151 		case R_386_PC32:
152 		case R_386_GLOB_DAT:
153 		case R_386_TLS_TPOFF:
154 		case R_386_TLS_TPOFF32:
155 		case R_386_TLS_DTPMOD32:
156 		case R_386_TLS_DTPOFF32:
157 			def = find_symdef(ELF_R_SYM(rel->r_info), obj, &defobj,
158 			    flags, cache, lockstate);
159 			if (def == NULL)
160 				goto done;
161 			if (ELF_ST_TYPE(def->st_info) == STT_GNU_IFUNC) {
162 				switch (ELF_R_TYPE(rel->r_info)) {
163 				case R_386_32:
164 				case R_386_PC32:
165 				case R_386_GLOB_DAT:
166 					if ((flags & SYMLOOK_IFUNC) == 0) {
167 						obj->non_plt_gnu_ifunc = true;
168 						continue;
169 					}
170 					symval = (Elf_Addr)rtld_resolve_ifunc(
171 					    defobj, def);
172 					break;
173 				case R_386_TLS_TPOFF:
174 				case R_386_TLS_TPOFF32:
175 				case R_386_TLS_DTPMOD32:
176 				case R_386_TLS_DTPOFF32:
177 					_rtld_error("%s: IFUNC for TLS reloc",
178 					    obj->path);
179 					goto done;
180 				}
181 			} else {
182 				if ((flags & SYMLOOK_IFUNC) != 0)
183 					continue;
184 				symval = (Elf_Addr)defobj->relocbase +
185 				    def->st_value;
186 			}
187 			break;
188 		default:
189 			if ((flags & SYMLOOK_IFUNC) != 0)
190 				continue;
191 			break;
192 		}
193 		where = (Elf_Addr *)(obj->relocbase + rel->r_offset);
194 
195 		switch (ELF_R_TYPE(rel->r_info)) {
196 		case R_386_NONE:
197 			break;
198 		case R_386_32:
199 			*where += symval;
200 			break;
201 		case R_386_PC32:
202 		    /*
203 		     * I don't think the dynamic linker should ever
204 		     * see this type of relocation.  But the
205 		     * binutils-2.6 tools sometimes generate it.
206 		     */
207 		    *where += symval - (Elf_Addr)where;
208 		    break;
209 		case R_386_COPY:
210 			/*
211 			 * These are deferred until all other
212 			 * relocations have been done.  All we do here
213 			 * is make sure that the COPY relocation is
214 			 * not in a shared library.  They are allowed
215 			 * only in executable files.
216 			 */
217 			if (!obj->mainprog) {
218 				_rtld_error("%s: Unexpected R_386_COPY "
219 				    "relocation in shared library", obj->path);
220 				goto done;
221 			}
222 			break;
223 		case R_386_GLOB_DAT:
224 			*where = symval;
225 			break;
226 		case R_386_RELATIVE:
227 			*where += (Elf_Addr)obj->relocbase;
228 			break;
229 		case R_386_TLS_TPOFF:
230 		case R_386_TLS_TPOFF32:
231 			/*
232 			 * We lazily allocate offsets for static TLS
233 			 * as we see the first relocation that
234 			 * references the TLS block. This allows us to
235 			 * support (small amounts of) static TLS in
236 			 * dynamically loaded modules. If we run out
237 			 * of space, we generate an error.
238 			 */
239 			if (!defobj->tls_done) {
240 				if (!allocate_tls_offset((Obj_Entry*) defobj)) {
241 					_rtld_error("%s: No space available "
242 					    "for static Thread Local Storage",
243 					    obj->path);
244 					goto done;
245 				}
246 			}
247 			add = (Elf_Addr)(def->st_value - defobj->tlsoffset);
248 			if (ELF_R_TYPE(rel->r_info) == R_386_TLS_TPOFF)
249 				*where += add;
250 			else
251 				*where -= add;
252 			break;
253 		case R_386_TLS_DTPMOD32:
254 			*where += (Elf_Addr)defobj->tlsindex;
255 			break;
256 		case R_386_TLS_DTPOFF32:
257 			*where += (Elf_Addr) def->st_value;
258 			break;
259 		default:
260 			_rtld_error("%s: Unsupported relocation type %d"
261 			    " in non-PLT relocations\n", obj->path,
262 			    ELF_R_TYPE(rel->r_info));
263 			goto done;
264 		}
265 	}
266 	r = 0;
267 done:
268 	free(cache);
269 	return (r);
270 }
271 
272 /* Process the PLT relocations. */
273 int
274 reloc_plt(Obj_Entry *obj)
275 {
276     const Elf_Rel *rellim;
277     const Elf_Rel *rel;
278 
279     rellim = (const Elf_Rel *)((char *)obj->pltrel + obj->pltrelsize);
280     for (rel = obj->pltrel;  rel < rellim;  rel++) {
281 	Elf_Addr *where/*, val*/;
282 
283 	switch (ELF_R_TYPE(rel->r_info)) {
284 	case R_386_JMP_SLOT:
285 	  /* Relocate the GOT slot pointing into the PLT. */
286 	  where = (Elf_Addr *)(obj->relocbase + rel->r_offset);
287 	  *where += (Elf_Addr)obj->relocbase;
288 	  break;
289 
290 	case R_386_IRELATIVE:
291 	  obj->irelative = true;
292 	  break;
293 
294 	default:
295 	  _rtld_error("Unknown relocation type %x in PLT",
296 	    ELF_R_TYPE(rel->r_info));
297 	  return (-1);
298 	}
299     }
300     return 0;
301 }
302 
303 /* Relocate the jump slots in an object. */
304 int
305 reloc_jmpslots(Obj_Entry *obj, int flags, RtldLockState *lockstate)
306 {
307     const Elf_Rel *rellim;
308     const Elf_Rel *rel;
309 
310     if (obj->jmpslots_done)
311 	return 0;
312     rellim = (const Elf_Rel *)((char *)obj->pltrel + obj->pltrelsize);
313     for (rel = obj->pltrel;  rel < rellim;  rel++) {
314 	Elf_Addr *where, target;
315 	const Elf_Sym *def;
316 	const Obj_Entry *defobj;
317 
318 	switch (ELF_R_TYPE(rel->r_info)) {
319 	case R_386_JMP_SLOT:
320 	  where = (Elf_Addr *)(obj->relocbase + rel->r_offset);
321 	  def = find_symdef(ELF_R_SYM(rel->r_info), obj, &defobj,
322 		SYMLOOK_IN_PLT | flags, NULL, lockstate);
323 	  if (def == NULL)
324 	      return (-1);
325 	  if (ELF_ST_TYPE(def->st_info) == STT_GNU_IFUNC) {
326 	      obj->gnu_ifunc = true;
327 	      continue;
328 	  }
329 	  target = (Elf_Addr)(defobj->relocbase + def->st_value);
330 	  reloc_jmpslot(where, target, defobj, obj, rel);
331 	  break;
332 
333 	case R_386_IRELATIVE:
334 	  break;
335 
336 	default:
337 	  _rtld_error("Unknown relocation type %x in PLT",
338 	    ELF_R_TYPE(rel->r_info));
339 	  return (-1);
340 	}
341     }
342 
343     obj->jmpslots_done = true;
344     return 0;
345 }
346 
347 int
348 reloc_iresolve(Obj_Entry *obj, RtldLockState *lockstate)
349 {
350     const Elf_Rel *rellim;
351     const Elf_Rel *rel;
352     Elf_Addr *where, target;
353 
354     if (!obj->irelative)
355 	return (0);
356     rellim = (const Elf_Rel *)((char *)obj->pltrel + obj->pltrelsize);
357     for (rel = obj->pltrel;  rel < rellim;  rel++) {
358 	switch (ELF_R_TYPE(rel->r_info)) {
359 	case R_386_IRELATIVE:
360 	  where = (Elf_Addr *)(obj->relocbase + rel->r_offset);
361 	  lock_release(rtld_bind_lock, lockstate);
362 	  target = call_ifunc_resolver(obj->relocbase + *where);
363 	  wlock_acquire(rtld_bind_lock, lockstate);
364 	  *where = target;
365 	  break;
366 	}
367     }
368     obj->irelative = false;
369     return (0);
370 }
371 
372 int
373 reloc_gnu_ifunc(Obj_Entry *obj, int flags, RtldLockState *lockstate)
374 {
375     const Elf_Rel *rellim;
376     const Elf_Rel *rel;
377 
378     if (!obj->gnu_ifunc)
379 	return (0);
380     rellim = (const Elf_Rel *)((char *)obj->pltrel + obj->pltrelsize);
381     for (rel = obj->pltrel;  rel < rellim;  rel++) {
382 	Elf_Addr *where, target;
383 	const Elf_Sym *def;
384 	const Obj_Entry *defobj;
385 
386 	switch (ELF_R_TYPE(rel->r_info)) {
387 	case R_386_JMP_SLOT:
388 	  where = (Elf_Addr *)(obj->relocbase + rel->r_offset);
389 	  def = find_symdef(ELF_R_SYM(rel->r_info), obj, &defobj,
390 		SYMLOOK_IN_PLT | flags, NULL, lockstate);
391 	  if (def == NULL)
392 	      return (-1);
393 	  if (ELF_ST_TYPE(def->st_info) != STT_GNU_IFUNC)
394 	      continue;
395 	  lock_release(rtld_bind_lock, lockstate);
396 	  target = (Elf_Addr)rtld_resolve_ifunc(defobj, def);
397 	  wlock_acquire(rtld_bind_lock, lockstate);
398 	  reloc_jmpslot(where, target, defobj, obj, rel);
399 	  break;
400 	}
401     }
402 
403     obj->gnu_ifunc = false;
404     return (0);
405 }
406 
407 uint32_t cpu_feature, cpu_feature2, cpu_stdext_feature, cpu_stdext_feature2;
408 
409 static void
410 rtld_cpuid_count(int idx, int cnt, u_int *p)
411 {
412 
413 	__asm __volatile(
414 	    "	pushl	%%ebx\n"
415 	    "	cpuid\n"
416 	    "	movl	%%ebx,%1\n"
417 	    "	popl	%%ebx\n"
418 	    : "=a" (p[0]), "=r" (p[1]), "=c" (p[2]), "=d" (p[3])
419 	    :  "0" (idx), "2" (cnt));
420 }
421 
422 void
423 ifunc_init(Elf_Auxinfo aux_info[__min_size(AT_COUNT)] __unused)
424 {
425 	u_int p[4], cpu_high;
426 	int cpuid_supported;
427 
428 	__asm __volatile(
429 	    "	pushfl\n"
430 	    "	popl	%%eax\n"
431 	    "	movl    %%eax,%%ecx\n"
432 	    "	xorl    $0x200000,%%eax\n"
433 	    "	pushl	%%eax\n"
434 	    "	popfl\n"
435 	    "	pushfl\n"
436 	    "	popl    %%eax\n"
437 	    "	xorl    %%eax,%%ecx\n"
438 	    "	je	1f\n"
439 	    "	movl	$1,%0\n"
440 	    "	jmp	2f\n"
441 	    "1:	movl	$0,%0\n"
442 	    "2:\n"
443 	    : "=r" (cpuid_supported) : : "eax", "ecx");
444 	if (!cpuid_supported)
445 		return;
446 
447 	rtld_cpuid_count(1, 0, p);
448 	cpu_feature = p[3];
449 	cpu_feature2 = p[2];
450 	rtld_cpuid_count(0, 0, p);
451 	cpu_high = p[0];
452 	if (cpu_high >= 7) {
453 		rtld_cpuid_count(7, 0, p);
454 		cpu_stdext_feature = p[1];
455 		cpu_stdext_feature2 = p[2];
456 	}
457 }
458 
459 void
460 allocate_initial_tls(Obj_Entry *objs)
461 {
462     void* tls;
463 
464     /*
465      * Fix the size of the static TLS block by using the maximum
466      * offset allocated so far and adding a bit for dynamic modules to
467      * use.
468      */
469     tls_static_space = tls_last_offset + RTLD_STATIC_TLS_EXTRA;
470     tls = allocate_tls(objs, NULL, 3*sizeof(Elf_Addr), sizeof(Elf_Addr));
471     i386_set_gsbase(tls);
472 }
473 
474 /* GNU ABI */
475 __attribute__((__regparm__(1)))
476 void *___tls_get_addr(tls_index *ti)
477 {
478     Elf_Addr** segbase;
479 
480     __asm __volatile("movl %%gs:0, %0" : "=r" (segbase));
481 
482     return tls_get_addr_common(&segbase[1], ti->ti_module, ti->ti_offset);
483 }
484 
485 /* Sun ABI */
486 void *__tls_get_addr(tls_index *ti)
487 {
488     Elf_Addr** segbase;
489 
490     __asm __volatile("movl %%gs:0, %0" : "=r" (segbase));
491 
492     return tls_get_addr_common(&segbase[1], ti->ti_module, ti->ti_offset);
493 }
494