blob: b1af07064d28fce07c5e6a5df5e7a7de6af872fa [file] [log] [blame]
Joe Damatof0bd98b2010-11-08 15:47:42 -08001#include "config.h"
Juan Cespedesd44c6b81998-09-25 14:48:42 +02002
Petr Machata157cc4d2012-04-04 19:00:34 +02003#include <assert.h>
Juan Cespedesd914a202004-11-10 00:15:33 +01004#include <endian.h>
Juan Cespedes96935a91997-08-09 23:45:39 +02005#include <errno.h>
Juan Cespedes96935a91997-08-09 23:45:39 +02006#include <fcntl.h>
Juan Cespedesd914a202004-11-10 00:15:33 +01007#include <gelf.h>
Zachary T Welchbfb26c72010-12-06 23:21:00 -08008#include <inttypes.h>
Petr Machata157cc4d2012-04-04 19:00:34 +02009#include <search.h>
Juan Cespedesd914a202004-11-10 00:15:33 +010010#include <stdint.h>
Petr Machatacc0e1e42012-04-25 13:42:07 +020011#include <stdio.h>
Juan Cespedesd914a202004-11-10 00:15:33 +010012#include <stdlib.h>
Juan Cespedes96935a91997-08-09 23:45:39 +020013#include <string.h>
Juan Cespedes5e01f651998-03-08 22:31:44 +010014#include <unistd.h>
Juan Cespedes96935a91997-08-09 23:45:39 +020015
Juan Cespedesf7281232009-06-25 16:11:21 +020016#include "common.h"
Petr Machata366c2f42012-02-09 19:34:36 +010017#include "proc.h"
Petr Machata2b46cfc2012-02-18 11:17:29 +010018#include "library.h"
Petr Machatab5f80ac2012-04-04 01:46:18 +020019#include "filter.h"
Joe Damatof0bd98b2010-11-08 15:47:42 -080020
Paul Gilliambe320772006-04-24 22:06:23 +020021#ifdef PLT_REINITALISATION_BP
Ian Wienand2d45b1a2006-02-20 22:48:07 +010022extern char *PLTs_initialized_by_here;
Paul Gilliambe320772006-04-24 22:06:23 +020023#endif
Ian Wienand9a2ad352006-02-20 22:44:45 +010024
Petr Machatae67635d2012-03-21 03:37:39 +010025#ifndef ARCH_HAVE_LTELF_DATA
26int
Petr Machatae0615ab2012-04-17 05:17:48 +020027arch_elf_init(struct ltelf *lte, struct library *lib)
Petr Machatae67635d2012-03-21 03:37:39 +010028{
29 return 0;
30}
Petr Machatac67a6e62012-03-28 02:39:49 +020031
32void
33arch_elf_destroy(struct ltelf *lte)
34{
35}
Petr Machatae67635d2012-03-21 03:37:39 +010036#endif
37
Petr Machatae6523e62012-03-24 04:54:06 +010038int
39default_elf_add_plt_entry(struct Process *proc, struct ltelf *lte,
Petr Machata1be22912012-03-27 03:11:33 +020040 const char *a_name, GElf_Rela *rela, size_t ndx,
Petr Machatae6523e62012-03-24 04:54:06 +010041 struct library_symbol **ret)
42{
43 char *name = strdup(a_name);
44 if (name == NULL) {
45 fail:
46 free(name);
47 return -1;
48 }
49
Petr Machata1be22912012-03-27 03:11:33 +020050 GElf_Addr addr = arch_plt_sym_val(lte, ndx, rela);
Petr Machatae6523e62012-03-24 04:54:06 +010051
52 struct library_symbol *libsym = malloc(sizeof(*libsym));
53 if (libsym == NULL)
54 goto fail;
55
Petr Machataea8eb9a2012-04-17 01:32:07 +020056 /* XXX The double cast should be removed when
57 * target_address_t becomes integral type. */
58 target_address_t taddr = (target_address_t)
59 (uintptr_t)(addr + lte->bias);
Petr Machatabb790602012-03-25 01:41:59 +010060
Petr Machatae8d90762012-04-15 04:28:31 +020061 if (library_symbol_init(libsym, taddr, name, 1, LS_TOPLT_EXEC) < 0) {
62 free(libsym);
63 goto fail;
64 }
65
Petr Machatae6523e62012-03-24 04:54:06 +010066 *ret = libsym;
67 return 0;
68}
69
70#ifndef ARCH_HAVE_ADD_PLT_ENTRY
71enum plt_status
72arch_elf_add_plt_entry(struct Process *proc, struct ltelf *lte,
Petr Machata1be22912012-03-27 03:11:33 +020073 const char *a_name, GElf_Rela *rela, size_t ndx,
Petr Machatae6523e62012-03-24 04:54:06 +010074 struct library_symbol **ret)
75{
76 return plt_default;
77}
78#endif
79
Petr Machatae67635d2012-03-21 03:37:39 +010080Elf_Data *
81elf_loaddata(Elf_Scn *scn, GElf_Shdr *shdr)
Petr Machatafe1c1712010-10-27 16:57:34 +020082{
83 Elf_Data *data = elf_getdata(scn, NULL);
84 if (data == NULL || elf_getdata(scn, data) != NULL
85 || data->d_off || data->d_size != shdr->sh_size)
86 return NULL;
87 return data;
88}
89
Petr Machatae67635d2012-03-21 03:37:39 +010090static int
Petr Machataffd5aab2012-03-24 02:03:33 +010091elf_get_section_if(struct ltelf *lte, Elf_Scn **tgt_sec, GElf_Shdr *tgt_shdr,
92 int (*predicate)(Elf_Scn *, GElf_Shdr *, void *data),
93 void *data)
Petr Machatafe1c1712010-10-27 16:57:34 +020094{
95 int i;
96 for (i = 1; i < lte->ehdr.e_shnum; ++i) {
97 Elf_Scn *scn;
98 GElf_Shdr shdr;
99
100 scn = elf_getscn(lte->elf, i);
101 if (scn == NULL || gelf_getshdr(scn, &shdr) == NULL) {
102 debug(1, "Couldn't read section or header.");
Petr Machatae67635d2012-03-21 03:37:39 +0100103 return -1;
104 }
Petr Machataffd5aab2012-03-24 02:03:33 +0100105 if (predicate(scn, &shdr, data)) {
106 *tgt_sec = scn;
107 *tgt_shdr = shdr;
Petr Machatafe1c1712010-10-27 16:57:34 +0200108 return 0;
Petr Machataffd5aab2012-03-24 02:03:33 +0100109 }
Petr Machatafe1c1712010-10-27 16:57:34 +0200110 }
Petr Machatae67635d2012-03-21 03:37:39 +0100111 return -1;
Petr Machataffd5aab2012-03-24 02:03:33 +0100112
113}
114
115static int
116inside_p(Elf_Scn *scn, GElf_Shdr *shdr, void *data)
117{
118 GElf_Addr addr = *(GElf_Addr *)data;
119 return addr >= shdr->sh_addr
120 && addr < shdr->sh_addr + shdr->sh_size;
121}
122
123int
124elf_get_section_covering(struct ltelf *lte, GElf_Addr addr,
125 Elf_Scn **tgt_sec, GElf_Shdr *tgt_shdr)
126{
127 return elf_get_section_if(lte, tgt_sec, tgt_shdr,
128 &inside_p, &addr);
129}
130
131static int
132type_p(Elf_Scn *scn, GElf_Shdr *shdr, void *data)
133{
134 GElf_Word type = *(GElf_Word *)data;
135 return shdr->sh_type == type;
136}
137
138int
139elf_get_section_type(struct ltelf *lte, GElf_Word type,
140 Elf_Scn **tgt_sec, GElf_Shdr *tgt_shdr)
141{
142 return elf_get_section_if(lte, tgt_sec, tgt_shdr,
143 &type_p, &type);
Petr Machatae67635d2012-03-21 03:37:39 +0100144}
145
Petr Machata5b3e26a2012-04-30 20:53:22 +0200146struct section_named_data {
147 struct ltelf *lte;
148 const char *name;
149};
150
151static int
152name_p(Elf_Scn *scn, GElf_Shdr *shdr, void *d)
153{
154 struct section_named_data *data = d;
155 const char *name = elf_strptr(data->lte->elf,
156 data->lte->ehdr.e_shstrndx,
157 shdr->sh_name);
158 return strcmp(name, data->name) == 0;
159}
160
161int
162elf_get_section_named(struct ltelf *lte, const char *name,
163 Elf_Scn **tgt_sec, GElf_Shdr *tgt_shdr)
164{
165 struct section_named_data data = {
166 .lte = lte,
167 .name = name,
168 };
169 return elf_get_section_if(lte, tgt_sec, tgt_shdr,
170 &name_p, &data);
171}
172
Petr Machatae67635d2012-03-21 03:37:39 +0100173static int
Petr Machata3a01cd72012-04-30 20:50:20 +0200174need_data(Elf_Data *data, GElf_Xword offset, GElf_Xword size)
Petr Machatae67635d2012-03-21 03:37:39 +0100175{
176 assert(data != NULL);
177 if (data->d_size < size || offset > data->d_size - size) {
Petr Machataa82d3222012-05-01 01:04:27 +0200178 debug(1, "Not enough data to read %"PRId64"-byte value"
179 " at offset %"PRId64".", size, offset);
Petr Machatae67635d2012-03-21 03:37:39 +0100180 return -1;
181 }
Petr Machatafe1c1712010-10-27 16:57:34 +0200182 return 0;
183}
184
Petr Machatae67635d2012-03-21 03:37:39 +0100185#define DEF_READER(NAME, SIZE) \
186 int \
Petr Machata3a01cd72012-04-30 20:50:20 +0200187 NAME(Elf_Data *data, GElf_Xword offset, uint##SIZE##_t *retp) \
Petr Machatae67635d2012-03-21 03:37:39 +0100188 { \
189 if (!need_data(data, offset, SIZE / 8) < 0) \
190 return -1; \
191 \
Petr Machata6d8ccb22012-03-27 03:11:57 +0200192 if (data->d_buf == NULL) /* NODATA section */ { \
193 *retp = 0; \
194 return 0; \
195 } \
196 \
Petr Machatae67635d2012-03-21 03:37:39 +0100197 union { \
198 uint##SIZE##_t dst; \
199 char buf[0]; \
200 } u; \
201 memcpy(u.buf, data->d_buf + offset, sizeof(u.dst)); \
202 *retp = u.dst; \
203 return 0; \
Petr Machatafe1c1712010-10-27 16:57:34 +0200204 }
205
Petr Machatae67635d2012-03-21 03:37:39 +0100206DEF_READER(elf_read_u16, 16)
207DEF_READER(elf_read_u32, 32)
208DEF_READER(elf_read_u64, 64)
Petr Machatafe1c1712010-10-27 16:57:34 +0200209
Petr Machatae67635d2012-03-21 03:37:39 +0100210#undef DEF_READER
Petr Machatafe1c1712010-10-27 16:57:34 +0200211
Petr Machata1974dbc2011-08-19 18:58:01 +0200212int
Petr Machata02bd9ec2011-09-21 17:38:59 +0200213open_elf(struct ltelf *lte, const char *filename)
214{
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100215 lte->fd = open(filename, O_RDONLY);
216 if (lte->fd == -1)
Petr Machata1974dbc2011-08-19 18:58:01 +0200217 return 1;
Juan Cespedes96935a91997-08-09 23:45:39 +0200218
Petr Machata02bd9ec2011-09-21 17:38:59 +0200219 elf_version(EV_CURRENT);
220
Juan Cespedesd914a202004-11-10 00:15:33 +0100221#ifdef HAVE_ELF_C_READ_MMAP
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100222 lte->elf = elf_begin(lte->fd, ELF_C_READ_MMAP, NULL);
Juan Cespedes5c3fe062004-06-14 18:08:37 +0200223#else
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100224 lte->elf = elf_begin(lte->fd, ELF_C_READ, NULL);
Juan Cespedes5c3fe062004-06-14 18:08:37 +0200225#endif
Juan Cespedes1cd999a2001-07-03 00:46:04 +0200226
Petr Machatacc0e1e42012-04-25 13:42:07 +0200227 if (lte->elf == NULL || elf_kind(lte->elf) != ELF_K_ELF) {
228 fprintf(stderr, "\"%s\" is not an ELF file\n", filename);
229 exit(EXIT_FAILURE);
230 }
Juan Cespedes1cd999a2001-07-03 00:46:04 +0200231
Petr Machatacc0e1e42012-04-25 13:42:07 +0200232 if (gelf_getehdr(lte->elf, &lte->ehdr) == NULL) {
233 fprintf(stderr, "can't read ELF header of \"%s\": %s\n",
234 filename, elf_errmsg(-1));
235 exit(EXIT_FAILURE);
236 }
Juan Cespedes1cd999a2001-07-03 00:46:04 +0200237
Petr Machatacc0e1e42012-04-25 13:42:07 +0200238 if (lte->ehdr.e_type != ET_EXEC && lte->ehdr.e_type != ET_DYN) {
239 fprintf(stderr, "\"%s\" is neither an ELF executable"
240 " nor a shared library\n", filename);
241 exit(EXIT_FAILURE);
242 }
Juan Cespedes1cd999a2001-07-03 00:46:04 +0200243
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100244 if ((lte->ehdr.e_ident[EI_CLASS] != LT_ELFCLASS
245 || lte->ehdr.e_machine != LT_ELF_MACHINE)
Juan Cespedesd914a202004-11-10 00:15:33 +0100246#ifdef LT_ELF_MACHINE2
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100247 && (lte->ehdr.e_ident[EI_CLASS] != LT_ELFCLASS2
248 || lte->ehdr.e_machine != LT_ELF_MACHINE2)
Juan Cespedesd914a202004-11-10 00:15:33 +0100249#endif
250#ifdef LT_ELF_MACHINE3
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100251 && (lte->ehdr.e_ident[EI_CLASS] != LT_ELFCLASS3
252 || lte->ehdr.e_machine != LT_ELF_MACHINE3)
Juan Cespedesd914a202004-11-10 00:15:33 +0100253#endif
Petr Machatacc0e1e42012-04-25 13:42:07 +0200254 ) {
255 fprintf(stderr,
256 "\"%s\" is ELF from incompatible architecture\n",
257 filename);
258 exit(EXIT_FAILURE);
259 }
Juan Cespedesd914a202004-11-10 00:15:33 +0100260
Petr Machata02bd9ec2011-09-21 17:38:59 +0200261 return 0;
262}
263
Petr Machatacc0e1e42012-04-25 13:42:07 +0200264static void
265read_symbol_table(struct ltelf *lte, const char *filename,
266 Elf_Scn *scn, GElf_Shdr *shdr, const char *name,
267 Elf_Data **datap, size_t *countp, const char **strsp)
268{
269 *datap = elf_getdata(scn, NULL);
270 *countp = shdr->sh_size / shdr->sh_entsize;
271 if ((*datap == NULL || elf_getdata(scn, *datap) != NULL)
272 && options.static_filter != NULL) {
273 fprintf(stderr, "Couldn't get data of section"
274 " %s from \"%s\": %s\n",
275 name, filename, elf_errmsg(-1));
276 exit(EXIT_FAILURE);
277 }
278
279 scn = elf_getscn(lte->elf, shdr->sh_link);
280 GElf_Shdr shdr2;
281 if (scn == NULL || gelf_getshdr(scn, &shdr2) == NULL) {
282 fprintf(stderr, "Couldn't get header of section"
283 " #%d from \"%s\": %s\n",
284 shdr2.sh_link, filename, elf_errmsg(-1));
285 exit(EXIT_FAILURE);
286 }
287
288 Elf_Data *data = elf_getdata(scn, NULL);
289 if (data == NULL || elf_getdata(scn, data) != NULL
290 || shdr2.sh_size != data->d_size || data->d_off) {
291 fprintf(stderr, "Couldn't get data of section"
292 " #%d from \"%s\": %s\n",
293 shdr2.sh_link, filename, elf_errmsg(-1));
294 exit(EXIT_FAILURE);
295 }
296
297 *strsp = data->d_buf;
298}
299
Petr Machatae67635d2012-03-21 03:37:39 +0100300static int
301do_init_elf(struct ltelf *lte, const char *filename, GElf_Addr bias)
Petr Machata2b46cfc2012-02-18 11:17:29 +0100302{
Petr Machata02bd9ec2011-09-21 17:38:59 +0200303 int i;
304 GElf_Addr relplt_addr = 0;
Petr Machata2b46cfc2012-02-18 11:17:29 +0100305 GElf_Addr soname_offset = 0;
Petr Machata02bd9ec2011-09-21 17:38:59 +0200306
307 debug(DEBUG_FUNCTION, "do_init_elf(filename=%s)", filename);
308 debug(1, "Reading ELF from %s...", filename);
309
310 if (open_elf(lte, filename) < 0)
311 return -1;
312
Petr Machatab120fdf2012-03-21 05:05:46 +0100313 /* Find out the base address. */
Petr Machata29add4f2012-02-18 16:38:05 +0100314 {
Petr Machata2b46cfc2012-02-18 11:17:29 +0100315 GElf_Phdr phdr;
316 for (i = 0; gelf_getphdr (lte->elf, i, &phdr) != NULL; ++i) {
317 if (phdr.p_type == PT_LOAD) {
Petr Machata49275b02012-04-03 12:38:51 +0200318 lte->base_addr = phdr.p_vaddr + bias;
Petr Machata2b46cfc2012-02-18 11:17:29 +0100319 break;
320 }
321 }
322 }
323
Petr Machatab120fdf2012-03-21 05:05:46 +0100324 if (lte->base_addr == 0) {
325 fprintf(stderr, "Couldn't determine base address of %s\n",
326 filename);
327 return -1;
328 }
329
330 lte->bias = bias;
Petr Machata29add4f2012-02-18 16:38:05 +0100331 lte->entry_addr = lte->ehdr.e_entry + lte->bias;
Petr Machata2b46cfc2012-02-18 11:17:29 +0100332
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100333 for (i = 1; i < lte->ehdr.e_shnum; ++i) {
334 Elf_Scn *scn;
335 GElf_Shdr shdr;
336 const char *name;
Juan Cespedesd914a202004-11-10 00:15:33 +0100337
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100338 scn = elf_getscn(lte->elf, i);
Petr Machatacc0e1e42012-04-25 13:42:07 +0200339 if (scn == NULL || gelf_getshdr(scn, &shdr) == NULL) {
340 fprintf(stderr, "Couldn't get section #%d from"
341 " \"%s\": %s\n", i, filename, elf_errmsg(-1));
342 exit(EXIT_FAILURE);
343 }
Juan Cespedesd914a202004-11-10 00:15:33 +0100344
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100345 name = elf_strptr(lte->elf, lte->ehdr.e_shstrndx, shdr.sh_name);
Petr Machatacc0e1e42012-04-25 13:42:07 +0200346 if (name == NULL) {
347 fprintf(stderr, "Couldn't get name of section #%d from"
348 " \"%s\": %s\n", i, filename, elf_errmsg(-1));
349 exit(EXIT_FAILURE);
350 }
Juan Cespedesd914a202004-11-10 00:15:33 +0100351
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100352 if (shdr.sh_type == SHT_SYMTAB) {
Petr Machatacc0e1e42012-04-25 13:42:07 +0200353 read_symbol_table(lte, filename,
354 scn, &shdr, name, &lte->symtab,
355 &lte->symtab_count, &lte->strtab);
Juan Cespedesd914a202004-11-10 00:15:33 +0100356
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100357 } else if (shdr.sh_type == SHT_DYNSYM) {
Petr Machatacc0e1e42012-04-25 13:42:07 +0200358 read_symbol_table(lte, filename,
359 scn, &shdr, name, &lte->dynsym,
360 &lte->dynsym_count, &lte->dynstr);
Juan Cespedesd914a202004-11-10 00:15:33 +0100361
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100362 } else if (shdr.sh_type == SHT_DYNAMIC) {
363 Elf_Data *data;
364 size_t j;
Juan Cespedesd914a202004-11-10 00:15:33 +0100365
Joe Damato87f4f582010-11-08 15:47:36 -0800366 lte->dyn_addr = shdr.sh_addr;
367 lte->dyn_sz = shdr.sh_size;
368
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100369 data = elf_getdata(scn, NULL);
Petr Machatacc0e1e42012-04-25 13:42:07 +0200370 if (data == NULL || elf_getdata(scn, data) != NULL) {
371 fprintf(stderr, "Couldn't get .dynamic data"
372 " from \"%s\": %s\n",
373 filename, strerror(errno));
374 exit(EXIT_FAILURE);
375 }
Juan Cespedesd914a202004-11-10 00:15:33 +0100376
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100377 for (j = 0; j < shdr.sh_size / shdr.sh_entsize; ++j) {
378 GElf_Dyn dyn;
Juan Cespedesd914a202004-11-10 00:15:33 +0100379
Petr Machatacc0e1e42012-04-25 13:42:07 +0200380 if (gelf_getdyn(data, j, &dyn) == NULL) {
381 fprintf(stderr, "Couldn't get .dynamic"
382 " data from \"%s\": %s\n",
383 filename, strerror(errno));
384 exit(EXIT_FAILURE);
385 }
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100386 if (dyn.d_tag == DT_JMPREL)
387 relplt_addr = dyn.d_un.d_ptr;
388 else if (dyn.d_tag == DT_PLTRELSZ)
Petr Machatae67635d2012-03-21 03:37:39 +0100389 lte->relplt_size = dyn.d_un.d_val;
390 else if (dyn.d_tag == DT_SONAME)
Petr Machata2b46cfc2012-02-18 11:17:29 +0100391 soname_offset = dyn.d_un.d_val;
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100392 }
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100393 } else if (shdr.sh_type == SHT_PROGBITS
394 || shdr.sh_type == SHT_NOBITS) {
395 if (strcmp(name, ".plt") == 0) {
396 lte->plt_addr = shdr.sh_addr;
397 lte->plt_size = shdr.sh_size;
Petr Machatae67635d2012-03-21 03:37:39 +0100398 lte->plt_data = elf_loaddata(scn, &shdr);
399 if (lte->plt_data == NULL)
400 fprintf(stderr,
401 "Can't load .plt data\n");
Petr Machata18c801c2012-04-07 01:24:08 +0200402 lte->plt_flags = shdr.sh_flags;
Petr Machatab3f8fef2006-11-30 14:45:07 +0100403 }
404#ifdef ARCH_SUPPORTS_OPD
405 else if (strcmp(name, ".opd") == 0) {
Paul Gilliam3f1219f2006-04-24 18:25:38 +0200406 lte->opd_addr = (GElf_Addr *) (long) shdr.sh_addr;
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100407 lte->opd_size = shdr.sh_size;
408 lte->opd = elf_rawdata(scn, NULL);
409 }
Petr Machatab3f8fef2006-11-30 14:45:07 +0100410#endif
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100411 }
Juan Cespedesd914a202004-11-10 00:15:33 +0100412 }
413
Petr Machatacc0e1e42012-04-25 13:42:07 +0200414 if (lte->dynsym == NULL || lte->dynstr == NULL) {
415 fprintf(stderr, "Couldn't find .dynsym or .dynstr in \"%s\"\n",
416 filename);
417 exit(EXIT_FAILURE);
418 }
Juan Cespedesd914a202004-11-10 00:15:33 +0100419
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100420 if (!relplt_addr || !lte->plt_addr) {
421 debug(1, "%s has no PLT relocations", filename);
422 lte->relplt = NULL;
423 lte->relplt_count = 0;
Petr Machatae67635d2012-03-21 03:37:39 +0100424 } else if (lte->relplt_size == 0) {
Petr Machatafe1c1712010-10-27 16:57:34 +0200425 debug(1, "%s has unknown PLT size", filename);
426 lte->relplt = NULL;
427 lte->relplt_count = 0;
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100428 } else {
Petr Machatafe1c1712010-10-27 16:57:34 +0200429
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100430 for (i = 1; i < lte->ehdr.e_shnum; ++i) {
431 Elf_Scn *scn;
432 GElf_Shdr shdr;
433
434 scn = elf_getscn(lte->elf, i);
Petr Machatacc0e1e42012-04-25 13:42:07 +0200435 if (scn == NULL || gelf_getshdr(scn, &shdr) == NULL) {
436 fprintf(stderr, "Couldn't get section header"
437 " from \"%s\": %s\n",
438 filename, elf_errmsg(-1));
439 exit(EXIT_FAILURE);
440 }
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100441 if (shdr.sh_addr == relplt_addr
Petr Machatae67635d2012-03-21 03:37:39 +0100442 && shdr.sh_size == lte->relplt_size) {
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100443 lte->relplt = elf_getdata(scn, NULL);
444 lte->relplt_count =
445 shdr.sh_size / shdr.sh_entsize;
446 if (lte->relplt == NULL
Petr Machatacc0e1e42012-04-25 13:42:07 +0200447 || elf_getdata(scn, lte->relplt) != NULL) {
448 fprintf(stderr, "Couldn't get .rel*.plt"
449 " data from \"%s\": %s\n",
450 filename, elf_errmsg(-1));
451 exit(EXIT_FAILURE);
452 }
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100453 break;
454 }
455 }
456
Petr Machatacc0e1e42012-04-25 13:42:07 +0200457 if (i == lte->ehdr.e_shnum) {
458 fprintf(stderr,
459 "Couldn't find .rel*.plt section in \"%s\"\n",
460 filename);
461 exit(EXIT_FAILURE);
462 }
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100463
464 debug(1, "%s %zd PLT relocations", filename, lte->relplt_count);
465 }
Petr Machata2b46cfc2012-02-18 11:17:29 +0100466
467 if (soname_offset != 0)
468 lte->soname = lte->dynstr + soname_offset;
469
Petr Machata1974dbc2011-08-19 18:58:01 +0200470 return 0;
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100471}
472
Petr Machata2b46cfc2012-02-18 11:17:29 +0100473/* XXX temporarily non-static */
Joe Damato7a2bdf82010-11-08 15:47:41 -0800474void
Juan Cespedesf1350522008-12-16 18:19:58 +0100475do_close_elf(struct ltelf *lte) {
Juan Cespedescd8976d2009-05-14 13:47:58 +0200476 debug(DEBUG_FUNCTION, "do_close_elf()");
Petr Machata4d9a91c2012-03-24 04:55:03 +0100477 arch_elf_destroy(lte);
Ian Wienand2d45b1a2006-02-20 22:48:07 +0100478 elf_end(lte->elf);
479 close(lte->fd);
Juan Cespedes1cd999a2001-07-03 00:46:04 +0200480}
481
Petr Machatab5f80ac2012-04-04 01:46:18 +0200482static int
483populate_plt(struct Process *proc, const char *filename,
484 struct ltelf *lte, struct library *lib)
485{
486 size_t i;
487 for (i = 0; i < lte->relplt_count; ++i) {
488 GElf_Rel rel;
489 GElf_Rela rela;
490 GElf_Sym sym;
491 void *ret;
492
493 if (lte->relplt->d_type == ELF_T_REL) {
494 ret = gelf_getrel(lte->relplt, i, &rel);
495 rela.r_offset = rel.r_offset;
496 rela.r_info = rel.r_info;
497 rela.r_addend = 0;
498 } else {
499 ret = gelf_getrela(lte->relplt, i, &rela);
500 }
501
502 if (ret == NULL
503 || ELF64_R_SYM(rela.r_info) >= lte->dynsym_count
504 || gelf_getsym(lte->dynsym, ELF64_R_SYM(rela.r_info),
Petr Machatacc0e1e42012-04-25 13:42:07 +0200505 &sym) == NULL) {
506 fprintf(stderr,
507 "Couldn't get relocation from \"%s\": %s\n",
508 filename, elf_errmsg(-1));
509 exit(EXIT_FAILURE);
510 }
Petr Machatab5f80ac2012-04-04 01:46:18 +0200511
512 char const *name = lte->dynstr + sym.st_name;
513
514 if (!filter_matches_symbol(options.plt_filter, name, lib))
515 continue;
516
Petr Machata218c5ff2012-04-15 04:22:39 +0200517 struct library_symbol *libsym = NULL;
Petr Machatab5f80ac2012-04-04 01:46:18 +0200518 switch (arch_elf_add_plt_entry(proc, lte, name,
519 &rela, i, &libsym)) {
520 case plt_default:
521 if (default_elf_add_plt_entry(proc, lte, name,
522 &rela, i, &libsym) < 0)
Petr Machata8eb0d932012-04-17 05:18:18 +0200523 /* fall-through */
Petr Machatab5f80ac2012-04-04 01:46:18 +0200524 case plt_fail:
525 return -1;
Petr Machata8eb0d932012-04-17 05:18:18 +0200526 /* fall-through */
Petr Machatab5f80ac2012-04-04 01:46:18 +0200527 case plt_ok:
528 if (libsym != NULL)
529 library_add_symbol(lib, libsym);
530 }
531 }
532 return 0;
533}
534
Petr Machata157cc4d2012-04-04 19:00:34 +0200535/* When -x rules result in request to trace several aliases, we only
536 * want to add such symbol once. The only way that those symbols
537 * differ in is their name, e.g. in glibc you have __GI___libc_free,
538 * __cfree, __free, __libc_free, cfree and free all defined on the
539 * same address. So instead we keep this unique symbol struct for
540 * each address, and replace name in libsym with a shorter variant if
541 * we find it. */
542struct unique_symbol {
543 target_address_t addr;
544 struct library_symbol *libsym;
545};
546
547static int
548unique_symbol_cmp(const void *key, const void *val)
549{
550 const struct unique_symbol *sym_key = key;
551 const struct unique_symbol *sym_val = val;
552 return sym_key->addr != sym_val->addr;
553}
554
Petr Machatada3edbf2012-04-04 02:20:21 +0200555static int
556populate_this_symtab(struct Process *proc, const char *filename,
557 struct ltelf *lte, struct library *lib,
558 Elf_Data *symtab, const char *strtab, size_t size)
559{
Petr Machata157cc4d2012-04-04 19:00:34 +0200560 /* Using sorted array would be arguably better, but this
561 * should be well enough for the number of symbols that we
562 * typically deal with. */
563 size_t num_symbols = 0;
564 struct unique_symbol *symbols = malloc(sizeof(*symbols) * size);
565 if (symbols == NULL) {
Petr Machatacc0e1e42012-04-25 13:42:07 +0200566 fprintf(stderr, "couldn't insert symbols for -x: %s\n",
567 strerror(errno));
Petr Machata157cc4d2012-04-04 19:00:34 +0200568 return -1;
569 }
570
Petr Machata40cc53b2012-04-07 01:25:38 +0200571 GElf_Word secflags[lte->ehdr.e_shnum];
Petr Machatada3edbf2012-04-04 02:20:21 +0200572 size_t i;
Petr Machata40cc53b2012-04-07 01:25:38 +0200573 for (i = 1; i < lte->ehdr.e_shnum; ++i) {
574 Elf_Scn *scn = elf_getscn(lte->elf, i);
575 if (scn == NULL)
576 continue;
577 GElf_Shdr shdr;
578 if (gelf_getshdr(scn, &shdr) == NULL)
579 continue;
580 secflags[i] = shdr.sh_flags;
581 }
582
583 size_t lib_len = strlen(lib->soname);
Petr Machatada3edbf2012-04-04 02:20:21 +0200584 for (i = 0; i < size; ++i) {
585 GElf_Sym sym;
Petr Machata074f68f2012-04-07 01:01:02 +0200586 if (gelf_getsym(symtab, i, &sym) == NULL) {
Petr Machatada3edbf2012-04-04 02:20:21 +0200587 fail:
Petr Machatacc0e1e42012-04-25 13:42:07 +0200588 fprintf(stderr,
589 "couldn't get symbol #%zd from %s: %s\n",
590 i, filename, elf_errmsg(-1));
Petr Machatada3edbf2012-04-04 02:20:21 +0200591 continue;
592 }
593
Petr Machata4de6b6b2012-04-04 14:06:09 +0200594 /* XXX support IFUNC as well. */
595 if (GELF_ST_TYPE(sym.st_info) != STT_FUNC
596 || sym.st_value == 0)
597 continue;
598
Petr Machata2bbeac42012-04-30 20:48:34 +0200599 const char *orig_name = strtab + sym.st_name;
600 const char *version = strchr(orig_name, '@');
601 size_t len = version != NULL ? (assert(version > orig_name),
602 (size_t)(version - orig_name))
603 : strlen(orig_name);
604 char name[len + 1];
605 memcpy(name, orig_name, len);
606 name[len] = 0;
607
Petr Machatada3edbf2012-04-04 02:20:21 +0200608 if (!filter_matches_symbol(options.static_filter, name, lib))
609 continue;
Petr Machatada3edbf2012-04-04 02:20:21 +0200610
Petr Machataea8eb9a2012-04-17 01:32:07 +0200611 target_address_t addr = (target_address_t)
612 (uintptr_t)(sym.st_value + lte->bias);
Petr Machatada3edbf2012-04-04 02:20:21 +0200613 target_address_t naddr;
Petr Machata40cc53b2012-04-07 01:25:38 +0200614
615 /* On arches that support OPD, the value of typical
616 * function symbol will be a pointer to .opd, but some
617 * will point directly to .text. We don't want to
618 * translate those. */
619 if (secflags[sym.st_shndx] & SHF_EXECINSTR) {
620 naddr = addr;
Petr Machatab1492df2012-04-30 21:01:40 +0200621 } else if (arch_translate_address(lte, addr, &naddr) < 0) {
Petr Machatacc0e1e42012-04-25 13:42:07 +0200622 fprintf(stderr,
623 "couldn't translate address of %s@%s: %s\n",
624 name, lib->soname, strerror(errno));
Petr Machatada3edbf2012-04-04 02:20:21 +0200625 continue;
626 }
Petr Machata40cc53b2012-04-07 01:25:38 +0200627
Petr Machata3840f682012-04-06 16:05:41 +0200628 char *full_name;
629 if (lib->type != LT_LIBTYPE_MAIN) {
630 full_name = malloc(strlen(name) + 1 + lib_len + 1);
631 if (full_name == NULL)
632 goto fail;
633 sprintf(full_name, "%s@%s", name, lib->soname);
634 } else {
635 full_name = strdup(name);
636 if (full_name == NULL)
637 goto fail;
638 }
Petr Machata4de6b6b2012-04-04 14:06:09 +0200639
Petr Machata157cc4d2012-04-04 19:00:34 +0200640 /* Look whether we already have a symbol for this
641 * address. If not, add this one. */
642 struct unique_symbol key = { naddr, NULL };
643 struct unique_symbol *unique
644 = lsearch(&key, symbols, &num_symbols,
645 sizeof(*symbols), &unique_symbol_cmp);
Petr Machatada3edbf2012-04-04 02:20:21 +0200646
Petr Machata157cc4d2012-04-04 19:00:34 +0200647 if (unique->libsym == NULL) {
648 struct library_symbol *libsym = malloc(sizeof(*libsym));
Petr Machatae8d90762012-04-15 04:28:31 +0200649 if (libsym == NULL
650 || library_symbol_init(libsym, naddr, full_name,
651 1, LS_TOPLT_NONE) < 0) {
Petr Machata157cc4d2012-04-04 19:00:34 +0200652 --num_symbols;
653 goto fail;
654 }
Petr Machata157cc4d2012-04-04 19:00:34 +0200655 unique->libsym = libsym;
656 unique->addr = naddr;
657
658 } else if (strlen(full_name) < strlen(unique->libsym->name)) {
659 library_symbol_set_name(unique->libsym, full_name, 1);
660
661 } else {
662 free(full_name);
663 }
Petr Machatada3edbf2012-04-04 02:20:21 +0200664 }
Petr Machata157cc4d2012-04-04 19:00:34 +0200665
666 for (i = 0; i < num_symbols; ++i) {
667 assert(symbols[i].libsym != NULL);
668 library_add_symbol(lib, symbols[i].libsym);
669 }
670
671 free(symbols);
672
Petr Machatada3edbf2012-04-04 02:20:21 +0200673 return 0;
674}
675
676static int
677populate_symtab(struct Process *proc, const char *filename,
678 struct ltelf *lte, struct library *lib)
679{
680 if (lte->symtab != NULL && lte->strtab != NULL)
681 return populate_this_symtab(proc, filename, lte, lib,
682 lte->symtab, lte->strtab,
683 lte->symtab_count);
684 else
685 return populate_this_symtab(proc, filename, lte, lib,
686 lte->dynsym, lte->dynstr,
687 lte->dynsym_count);
688}
689
Petr Machatab5f80ac2012-04-04 01:46:18 +0200690int
691ltelf_read_library(struct library *lib, struct Process *proc,
692 const char *filename, GElf_Addr bias)
Petr Machatae84fa002012-02-07 13:43:03 +0100693{
Petr Machata29add4f2012-02-18 16:38:05 +0100694 struct ltelf lte = {};
Petr Machatab120fdf2012-03-21 05:05:46 +0100695 if (do_init_elf(&lte, filename, bias) < 0)
Petr Machatab5f80ac2012-04-04 01:46:18 +0200696 return -1;
Petr Machatae0615ab2012-04-17 05:17:48 +0200697 if (arch_elf_init(&lte, lib) < 0) {
698 fprintf(stderr, "Backend initialization failed.\n");
699 return -1;
700 }
701
Petr Machatae67635d2012-03-21 03:37:39 +0100702 proc->e_machine = lte.ehdr.e_machine;
Joe Damatof0bd98b2010-11-08 15:47:42 -0800703
Petr Machatab5f80ac2012-04-04 01:46:18 +0200704 int status = 0;
Petr Machatafc6ff182012-04-04 13:11:50 +0200705 if (lib == NULL)
706 goto fail;
707
708 /* Note that we set soname and pathname as soon as they are
709 * allocated, so in case of further errors, this get released
710 * when LIB is release, which should happen in the caller when
711 * we return error. */
712
713 if (lib->pathname == NULL) {
714 char *pathname = strdup(filename);
715 if (pathname == NULL)
716 goto fail;
Petr Machataf13afd52012-04-14 02:30:31 +0200717 library_set_pathname(lib, pathname, 1);
Joe Damatof0bd98b2010-11-08 15:47:42 -0800718 }
719
Petr Machata0b55b582012-04-02 00:38:46 +0200720 if (lte.soname != NULL) {
Petr Machatafc6ff182012-04-04 13:11:50 +0200721 char *soname = strdup(lte.soname);
722 if (soname == NULL)
723 goto fail;
724 library_set_soname(lib, soname, 1);
Petr Machata0b55b582012-04-02 00:38:46 +0200725 } else {
Petr Machatafc6ff182012-04-04 13:11:50 +0200726 const char *soname = rindex(lib->pathname, '/') + 1;
727 if (soname == NULL)
728 soname = lib->pathname;
729 library_set_soname(lib, soname, 0);
Petr Machata0b55b582012-04-02 00:38:46 +0200730 }
Joe Damatofa2aefc2010-10-30 19:56:50 -0700731
Petr Machataea8eb9a2012-04-17 01:32:07 +0200732 /* XXX The double cast should be removed when
733 * target_address_t becomes integral type. */
734 target_address_t entry = (target_address_t)(uintptr_t)lte.entry_addr;
Petr Machatab1492df2012-04-30 21:01:40 +0200735 if (arch_translate_address(&lte, entry, &entry) < 0)
Petr Machatab120fdf2012-03-21 05:05:46 +0100736 goto fail;
737
Petr Machataea8eb9a2012-04-17 01:32:07 +0200738 /* XXX The double cast should be removed when
739 * target_address_t becomes integral type. */
740 lib->base = (target_address_t)(uintptr_t)lte.base_addr;
Petr Machatab120fdf2012-03-21 05:05:46 +0100741 lib->entry = entry;
Petr Machataea8eb9a2012-04-17 01:32:07 +0200742 /* XXX The double cast should be removed when
743 * target_address_t becomes integral type. */
744 lib->dyn_addr = (target_address_t)(uintptr_t)lte.dyn_addr;
Ian Wienand9a2ad352006-02-20 22:44:45 +0100745
Petr Machatab5f80ac2012-04-04 01:46:18 +0200746 if (filter_matches_library(options.plt_filter, lib)
747 && populate_plt(proc, filename, &lte, lib) < 0)
748 goto fail;
Joe Damatoe2a8f572010-11-08 15:47:40 -0800749
Petr Machatada3edbf2012-04-04 02:20:21 +0200750 if (filter_matches_library(options.static_filter, lib)
751 && populate_symtab(proc, filename, &lte, lib) < 0)
752 goto fail;
753
Petr Machata2b46cfc2012-02-18 11:17:29 +0100754done:
755 do_close_elf(&lte);
Petr Machatab5f80ac2012-04-04 01:46:18 +0200756 return status;
Petr Machatafc6ff182012-04-04 13:11:50 +0200757
758fail:
759 status = -1;
760 goto done;
Petr Machata2b46cfc2012-02-18 11:17:29 +0100761}
Petr Machatae84fa002012-02-07 13:43:03 +0100762
Petr Machata2b46cfc2012-02-18 11:17:29 +0100763struct library *
764ltelf_read_main_binary(struct Process *proc, const char *path)
765{
Petr Machatab5f80ac2012-04-04 01:46:18 +0200766 struct library *lib = malloc(sizeof(*lib));
767 if (lib == NULL)
768 return NULL;
769 library_init(lib, LT_LIBTYPE_MAIN);
Petr Machatafc6ff182012-04-04 13:11:50 +0200770 library_set_pathname(lib, path, 0);
Petr Machatab5f80ac2012-04-04 01:46:18 +0200771
Petr Machatafc6ff182012-04-04 13:11:50 +0200772 /* There is a race between running the process and reading its
773 * binary for internal consumption. So open the binary from
774 * the /proc filesystem. XXX Note that there is similar race
775 * for libraries, but there we don't have a nice answer like
776 * that. Presumably we could read the DSOs from the process
777 * memory image, but that's not currently done. */
Petr Machata2b46cfc2012-02-18 11:17:29 +0100778 char *fname = pid2name(proc->pid);
Petr Machatab5f80ac2012-04-04 01:46:18 +0200779 if (ltelf_read_library(lib, proc, fname, 0) < 0) {
780 library_destroy(lib);
781 free(lib);
782 return NULL;
Petr Machata0b55b582012-04-02 00:38:46 +0200783 }
Petr Machatab5f80ac2012-04-04 01:46:18 +0200784
Petr Machata2b46cfc2012-02-18 11:17:29 +0100785 return lib;
Juan Cespedes96935a91997-08-09 23:45:39 +0200786}