| Joe Damato | f0bd98b | 2010-11-08 15:47:42 -0800 | [diff] [blame] | 1 | #include "config.h" |
| Juan Cespedes | d44c6b8 | 1998-09-25 14:48:42 +0200 | [diff] [blame] | 2 | |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 3 | #include <assert.h> |
| Andrey Zonov | 9d878c9 | 2012-08-05 00:19:51 +0400 | [diff] [blame] | 4 | #ifdef __linux__ |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 5 | #include <endian.h> |
| Andrey Zonov | 9d878c9 | 2012-08-05 00:19:51 +0400 | [diff] [blame] | 6 | #endif |
| Juan Cespedes | 96935a9 | 1997-08-09 23:45:39 +0200 | [diff] [blame] | 7 | #include <errno.h> |
| Juan Cespedes | 96935a9 | 1997-08-09 23:45:39 +0200 | [diff] [blame] | 8 | #include <fcntl.h> |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 9 | #include <gelf.h> |
| Zachary T Welch | bfb26c7 | 2010-12-06 23:21:00 -0800 | [diff] [blame] | 10 | #include <inttypes.h> |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 11 | #include <search.h> |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 12 | #include <stdint.h> |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 13 | #include <stdio.h> |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 14 | #include <stdlib.h> |
| Juan Cespedes | 96935a9 | 1997-08-09 23:45:39 +0200 | [diff] [blame] | 15 | #include <string.h> |
| Juan Cespedes | 5e01f65 | 1998-03-08 22:31:44 +0100 | [diff] [blame] | 16 | #include <unistd.h> |
| Juan Cespedes | 96935a9 | 1997-08-09 23:45:39 +0200 | [diff] [blame] | 17 | |
| Juan Cespedes | f728123 | 2009-06-25 16:11:21 +0200 | [diff] [blame] | 18 | #include "common.h" |
| Petr Machata | 366c2f4 | 2012-02-09 19:34:36 +0100 | [diff] [blame] | 19 | #include "proc.h" |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 20 | #include "library.h" |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 21 | #include "filter.h" |
| Joe Damato | f0bd98b | 2010-11-08 15:47:42 -0800 | [diff] [blame] | 22 | |
| Paul Gilliam | be32077 | 2006-04-24 22:06:23 +0200 | [diff] [blame] | 23 | #ifdef PLT_REINITALISATION_BP |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 24 | extern char *PLTs_initialized_by_here; |
| Paul Gilliam | be32077 | 2006-04-24 22:06:23 +0200 | [diff] [blame] | 25 | #endif |
| Ian Wienand | 9a2ad35 | 2006-02-20 22:44:45 +0100 | [diff] [blame] | 26 | |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 27 | #ifndef ARCH_HAVE_LTELF_DATA |
| 28 | int |
| Petr Machata | e0615ab | 2012-04-17 05:17:48 +0200 | [diff] [blame] | 29 | arch_elf_init(struct ltelf *lte, struct library *lib) |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 30 | { |
| 31 | return 0; |
| 32 | } |
| Petr Machata | c67a6e6 | 2012-03-28 02:39:49 +0200 | [diff] [blame] | 33 | |
| 34 | void |
| 35 | arch_elf_destroy(struct ltelf *lte) |
| 36 | { |
| 37 | } |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 38 | #endif |
| 39 | |
| Petr Machata | e6523e6 | 2012-03-24 04:54:06 +0100 | [diff] [blame] | 40 | int |
| 41 | default_elf_add_plt_entry(struct Process *proc, struct ltelf *lte, |
| Petr Machata | 1be2291 | 2012-03-27 03:11:33 +0200 | [diff] [blame] | 42 | const char *a_name, GElf_Rela *rela, size_t ndx, |
| Petr Machata | e6523e6 | 2012-03-24 04:54:06 +0100 | [diff] [blame] | 43 | struct library_symbol **ret) |
| 44 | { |
| 45 | char *name = strdup(a_name); |
| 46 | if (name == NULL) { |
| 47 | fail: |
| 48 | free(name); |
| 49 | return -1; |
| 50 | } |
| 51 | |
| Petr Machata | 1be2291 | 2012-03-27 03:11:33 +0200 | [diff] [blame] | 52 | GElf_Addr addr = arch_plt_sym_val(lte, ndx, rela); |
| Petr Machata | e6523e6 | 2012-03-24 04:54:06 +0100 | [diff] [blame] | 53 | |
| 54 | struct library_symbol *libsym = malloc(sizeof(*libsym)); |
| 55 | if (libsym == NULL) |
| 56 | goto fail; |
| 57 | |
| Petr Machata | ea8eb9a | 2012-04-17 01:32:07 +0200 | [diff] [blame] | 58 | /* XXX The double cast should be removed when |
| 59 | * target_address_t becomes integral type. */ |
| 60 | target_address_t taddr = (target_address_t) |
| 61 | (uintptr_t)(addr + lte->bias); |
| Petr Machata | bb79060 | 2012-03-25 01:41:59 +0100 | [diff] [blame] | 62 | |
| Petr Machata | e8d9076 | 2012-04-15 04:28:31 +0200 | [diff] [blame] | 63 | if (library_symbol_init(libsym, taddr, name, 1, LS_TOPLT_EXEC) < 0) { |
| 64 | free(libsym); |
| 65 | goto fail; |
| 66 | } |
| 67 | |
| Petr Machata | e6523e6 | 2012-03-24 04:54:06 +0100 | [diff] [blame] | 68 | *ret = libsym; |
| 69 | return 0; |
| 70 | } |
| 71 | |
| 72 | #ifndef ARCH_HAVE_ADD_PLT_ENTRY |
| 73 | enum plt_status |
| 74 | arch_elf_add_plt_entry(struct Process *proc, struct ltelf *lte, |
| Petr Machata | 1be2291 | 2012-03-27 03:11:33 +0200 | [diff] [blame] | 75 | const char *a_name, GElf_Rela *rela, size_t ndx, |
| Petr Machata | e6523e6 | 2012-03-24 04:54:06 +0100 | [diff] [blame] | 76 | struct library_symbol **ret) |
| 77 | { |
| 78 | return plt_default; |
| 79 | } |
| 80 | #endif |
| 81 | |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 82 | Elf_Data * |
| 83 | elf_loaddata(Elf_Scn *scn, GElf_Shdr *shdr) |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 84 | { |
| 85 | Elf_Data *data = elf_getdata(scn, NULL); |
| 86 | if (data == NULL || elf_getdata(scn, data) != NULL |
| 87 | || data->d_off || data->d_size != shdr->sh_size) |
| 88 | return NULL; |
| 89 | return data; |
| 90 | } |
| 91 | |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 92 | static int |
| Petr Machata | ffd5aab | 2012-03-24 02:03:33 +0100 | [diff] [blame] | 93 | elf_get_section_if(struct ltelf *lte, Elf_Scn **tgt_sec, GElf_Shdr *tgt_shdr, |
| 94 | int (*predicate)(Elf_Scn *, GElf_Shdr *, void *data), |
| 95 | void *data) |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 96 | { |
| 97 | int i; |
| 98 | for (i = 1; i < lte->ehdr.e_shnum; ++i) { |
| 99 | Elf_Scn *scn; |
| 100 | GElf_Shdr shdr; |
| 101 | |
| 102 | scn = elf_getscn(lte->elf, i); |
| 103 | if (scn == NULL || gelf_getshdr(scn, &shdr) == NULL) { |
| 104 | debug(1, "Couldn't read section or header."); |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 105 | return -1; |
| 106 | } |
| Petr Machata | ffd5aab | 2012-03-24 02:03:33 +0100 | [diff] [blame] | 107 | if (predicate(scn, &shdr, data)) { |
| 108 | *tgt_sec = scn; |
| 109 | *tgt_shdr = shdr; |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 110 | return 0; |
| Petr Machata | ffd5aab | 2012-03-24 02:03:33 +0100 | [diff] [blame] | 111 | } |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 112 | } |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 113 | return -1; |
| Petr Machata | ffd5aab | 2012-03-24 02:03:33 +0100 | [diff] [blame] | 114 | |
| 115 | } |
| 116 | |
| 117 | static int |
| 118 | inside_p(Elf_Scn *scn, GElf_Shdr *shdr, void *data) |
| 119 | { |
| 120 | GElf_Addr addr = *(GElf_Addr *)data; |
| 121 | return addr >= shdr->sh_addr |
| 122 | && addr < shdr->sh_addr + shdr->sh_size; |
| 123 | } |
| 124 | |
| 125 | int |
| 126 | elf_get_section_covering(struct ltelf *lte, GElf_Addr addr, |
| 127 | Elf_Scn **tgt_sec, GElf_Shdr *tgt_shdr) |
| 128 | { |
| 129 | return elf_get_section_if(lte, tgt_sec, tgt_shdr, |
| 130 | &inside_p, &addr); |
| 131 | } |
| 132 | |
| 133 | static int |
| 134 | type_p(Elf_Scn *scn, GElf_Shdr *shdr, void *data) |
| 135 | { |
| 136 | GElf_Word type = *(GElf_Word *)data; |
| 137 | return shdr->sh_type == type; |
| 138 | } |
| 139 | |
| 140 | int |
| 141 | elf_get_section_type(struct ltelf *lte, GElf_Word type, |
| 142 | Elf_Scn **tgt_sec, GElf_Shdr *tgt_shdr) |
| 143 | { |
| 144 | return elf_get_section_if(lte, tgt_sec, tgt_shdr, |
| 145 | &type_p, &type); |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 146 | } |
| 147 | |
| Petr Machata | 5b3e26a | 2012-04-30 20:53:22 +0200 | [diff] [blame] | 148 | struct section_named_data { |
| 149 | struct ltelf *lte; |
| 150 | const char *name; |
| 151 | }; |
| 152 | |
| 153 | static int |
| 154 | name_p(Elf_Scn *scn, GElf_Shdr *shdr, void *d) |
| 155 | { |
| 156 | struct section_named_data *data = d; |
| 157 | const char *name = elf_strptr(data->lte->elf, |
| 158 | data->lte->ehdr.e_shstrndx, |
| 159 | shdr->sh_name); |
| 160 | return strcmp(name, data->name) == 0; |
| 161 | } |
| 162 | |
| 163 | int |
| 164 | elf_get_section_named(struct ltelf *lte, const char *name, |
| 165 | Elf_Scn **tgt_sec, GElf_Shdr *tgt_shdr) |
| 166 | { |
| 167 | struct section_named_data data = { |
| 168 | .lte = lte, |
| 169 | .name = name, |
| 170 | }; |
| 171 | return elf_get_section_if(lte, tgt_sec, tgt_shdr, |
| 172 | &name_p, &data); |
| 173 | } |
| 174 | |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 175 | static int |
| Petr Machata | 3a01cd7 | 2012-04-30 20:50:20 +0200 | [diff] [blame] | 176 | need_data(Elf_Data *data, GElf_Xword offset, GElf_Xword size) |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 177 | { |
| 178 | assert(data != NULL); |
| 179 | if (data->d_size < size || offset > data->d_size - size) { |
| Petr Machata | a82d322 | 2012-05-01 01:04:27 +0200 | [diff] [blame] | 180 | debug(1, "Not enough data to read %"PRId64"-byte value" |
| 181 | " at offset %"PRId64".", size, offset); |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 182 | return -1; |
| 183 | } |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 184 | return 0; |
| 185 | } |
| 186 | |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 187 | #define DEF_READER(NAME, SIZE) \ |
| 188 | int \ |
| Petr Machata | 3a01cd7 | 2012-04-30 20:50:20 +0200 | [diff] [blame] | 189 | NAME(Elf_Data *data, GElf_Xword offset, uint##SIZE##_t *retp) \ |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 190 | { \ |
| 191 | if (!need_data(data, offset, SIZE / 8) < 0) \ |
| 192 | return -1; \ |
| 193 | \ |
| Petr Machata | 6d8ccb2 | 2012-03-27 03:11:57 +0200 | [diff] [blame] | 194 | if (data->d_buf == NULL) /* NODATA section */ { \ |
| 195 | *retp = 0; \ |
| 196 | return 0; \ |
| 197 | } \ |
| 198 | \ |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 199 | union { \ |
| 200 | uint##SIZE##_t dst; \ |
| 201 | char buf[0]; \ |
| 202 | } u; \ |
| 203 | memcpy(u.buf, data->d_buf + offset, sizeof(u.dst)); \ |
| 204 | *retp = u.dst; \ |
| 205 | return 0; \ |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 206 | } |
| 207 | |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 208 | DEF_READER(elf_read_u16, 16) |
| 209 | DEF_READER(elf_read_u32, 32) |
| 210 | DEF_READER(elf_read_u64, 64) |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 211 | |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 212 | #undef DEF_READER |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 213 | |
| Petr Machata | 1974dbc | 2011-08-19 18:58:01 +0200 | [diff] [blame] | 214 | int |
| Petr Machata | 02bd9ec | 2011-09-21 17:38:59 +0200 | [diff] [blame] | 215 | open_elf(struct ltelf *lte, const char *filename) |
| 216 | { |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 217 | lte->fd = open(filename, O_RDONLY); |
| 218 | if (lte->fd == -1) |
| Petr Machata | 1974dbc | 2011-08-19 18:58:01 +0200 | [diff] [blame] | 219 | return 1; |
| Juan Cespedes | 96935a9 | 1997-08-09 23:45:39 +0200 | [diff] [blame] | 220 | |
| Petr Machata | 02bd9ec | 2011-09-21 17:38:59 +0200 | [diff] [blame] | 221 | elf_version(EV_CURRENT); |
| 222 | |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 223 | #ifdef HAVE_ELF_C_READ_MMAP |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 224 | lte->elf = elf_begin(lte->fd, ELF_C_READ_MMAP, NULL); |
| Juan Cespedes | 5c3fe06 | 2004-06-14 18:08:37 +0200 | [diff] [blame] | 225 | #else |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 226 | lte->elf = elf_begin(lte->fd, ELF_C_READ, NULL); |
| Juan Cespedes | 5c3fe06 | 2004-06-14 18:08:37 +0200 | [diff] [blame] | 227 | #endif |
| Juan Cespedes | 1cd999a | 2001-07-03 00:46:04 +0200 | [diff] [blame] | 228 | |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 229 | if (lte->elf == NULL || elf_kind(lte->elf) != ELF_K_ELF) { |
| 230 | fprintf(stderr, "\"%s\" is not an ELF file\n", filename); |
| 231 | exit(EXIT_FAILURE); |
| 232 | } |
| Juan Cespedes | 1cd999a | 2001-07-03 00:46:04 +0200 | [diff] [blame] | 233 | |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 234 | if (gelf_getehdr(lte->elf, <e->ehdr) == NULL) { |
| 235 | fprintf(stderr, "can't read ELF header of \"%s\": %s\n", |
| 236 | filename, elf_errmsg(-1)); |
| 237 | exit(EXIT_FAILURE); |
| 238 | } |
| Juan Cespedes | 1cd999a | 2001-07-03 00:46:04 +0200 | [diff] [blame] | 239 | |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 240 | if (lte->ehdr.e_type != ET_EXEC && lte->ehdr.e_type != ET_DYN) { |
| 241 | fprintf(stderr, "\"%s\" is neither an ELF executable" |
| 242 | " nor a shared library\n", filename); |
| 243 | exit(EXIT_FAILURE); |
| 244 | } |
| Juan Cespedes | 1cd999a | 2001-07-03 00:46:04 +0200 | [diff] [blame] | 245 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 246 | if ((lte->ehdr.e_ident[EI_CLASS] != LT_ELFCLASS |
| 247 | || lte->ehdr.e_machine != LT_ELF_MACHINE) |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 248 | #ifdef LT_ELF_MACHINE2 |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 249 | && (lte->ehdr.e_ident[EI_CLASS] != LT_ELFCLASS2 |
| 250 | || lte->ehdr.e_machine != LT_ELF_MACHINE2) |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 251 | #endif |
| 252 | #ifdef LT_ELF_MACHINE3 |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 253 | && (lte->ehdr.e_ident[EI_CLASS] != LT_ELFCLASS3 |
| 254 | || lte->ehdr.e_machine != LT_ELF_MACHINE3) |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 255 | #endif |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 256 | ) { |
| 257 | fprintf(stderr, |
| 258 | "\"%s\" is ELF from incompatible architecture\n", |
| 259 | filename); |
| 260 | exit(EXIT_FAILURE); |
| 261 | } |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 262 | |
| Petr Machata | 02bd9ec | 2011-09-21 17:38:59 +0200 | [diff] [blame] | 263 | return 0; |
| 264 | } |
| 265 | |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 266 | static void |
| 267 | read_symbol_table(struct ltelf *lte, const char *filename, |
| 268 | Elf_Scn *scn, GElf_Shdr *shdr, const char *name, |
| 269 | Elf_Data **datap, size_t *countp, const char **strsp) |
| 270 | { |
| 271 | *datap = elf_getdata(scn, NULL); |
| 272 | *countp = shdr->sh_size / shdr->sh_entsize; |
| 273 | if ((*datap == NULL || elf_getdata(scn, *datap) != NULL) |
| 274 | && options.static_filter != NULL) { |
| 275 | fprintf(stderr, "Couldn't get data of section" |
| 276 | " %s from \"%s\": %s\n", |
| 277 | name, filename, elf_errmsg(-1)); |
| 278 | exit(EXIT_FAILURE); |
| 279 | } |
| 280 | |
| 281 | scn = elf_getscn(lte->elf, shdr->sh_link); |
| 282 | GElf_Shdr shdr2; |
| 283 | if (scn == NULL || gelf_getshdr(scn, &shdr2) == NULL) { |
| 284 | fprintf(stderr, "Couldn't get header of section" |
| 285 | " #%d from \"%s\": %s\n", |
| 286 | shdr2.sh_link, filename, elf_errmsg(-1)); |
| 287 | exit(EXIT_FAILURE); |
| 288 | } |
| 289 | |
| 290 | Elf_Data *data = elf_getdata(scn, NULL); |
| 291 | if (data == NULL || elf_getdata(scn, data) != NULL |
| 292 | || shdr2.sh_size != data->d_size || data->d_off) { |
| 293 | fprintf(stderr, "Couldn't get data of section" |
| 294 | " #%d from \"%s\": %s\n", |
| 295 | shdr2.sh_link, filename, elf_errmsg(-1)); |
| 296 | exit(EXIT_FAILURE); |
| 297 | } |
| 298 | |
| 299 | *strsp = data->d_buf; |
| 300 | } |
| 301 | |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 302 | static int |
| 303 | do_init_elf(struct ltelf *lte, const char *filename, GElf_Addr bias) |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 304 | { |
| Petr Machata | 02bd9ec | 2011-09-21 17:38:59 +0200 | [diff] [blame] | 305 | int i; |
| 306 | GElf_Addr relplt_addr = 0; |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 307 | GElf_Addr soname_offset = 0; |
| Petr Machata | 02bd9ec | 2011-09-21 17:38:59 +0200 | [diff] [blame] | 308 | |
| 309 | debug(DEBUG_FUNCTION, "do_init_elf(filename=%s)", filename); |
| 310 | debug(1, "Reading ELF from %s...", filename); |
| 311 | |
| 312 | if (open_elf(lte, filename) < 0) |
| 313 | return -1; |
| 314 | |
| Petr Machata | b120fdf | 2012-03-21 05:05:46 +0100 | [diff] [blame] | 315 | /* Find out the base address. */ |
| Petr Machata | 29add4f | 2012-02-18 16:38:05 +0100 | [diff] [blame] | 316 | { |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 317 | GElf_Phdr phdr; |
| 318 | for (i = 0; gelf_getphdr (lte->elf, i, &phdr) != NULL; ++i) { |
| 319 | if (phdr.p_type == PT_LOAD) { |
| Petr Machata | 49275b0 | 2012-04-03 12:38:51 +0200 | [diff] [blame] | 320 | lte->base_addr = phdr.p_vaddr + bias; |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 321 | break; |
| 322 | } |
| 323 | } |
| 324 | } |
| 325 | |
| Petr Machata | b120fdf | 2012-03-21 05:05:46 +0100 | [diff] [blame] | 326 | if (lte->base_addr == 0) { |
| 327 | fprintf(stderr, "Couldn't determine base address of %s\n", |
| 328 | filename); |
| 329 | return -1; |
| 330 | } |
| 331 | |
| 332 | lte->bias = bias; |
| Petr Machata | 29add4f | 2012-02-18 16:38:05 +0100 | [diff] [blame] | 333 | lte->entry_addr = lte->ehdr.e_entry + lte->bias; |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 334 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 335 | for (i = 1; i < lte->ehdr.e_shnum; ++i) { |
| 336 | Elf_Scn *scn; |
| 337 | GElf_Shdr shdr; |
| 338 | const char *name; |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 339 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 340 | scn = elf_getscn(lte->elf, i); |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 341 | if (scn == NULL || gelf_getshdr(scn, &shdr) == NULL) { |
| 342 | fprintf(stderr, "Couldn't get section #%d from" |
| 343 | " \"%s\": %s\n", i, filename, elf_errmsg(-1)); |
| 344 | exit(EXIT_FAILURE); |
| 345 | } |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 346 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 347 | name = elf_strptr(lte->elf, lte->ehdr.e_shstrndx, shdr.sh_name); |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 348 | if (name == NULL) { |
| 349 | fprintf(stderr, "Couldn't get name of section #%d from" |
| 350 | " \"%s\": %s\n", i, filename, elf_errmsg(-1)); |
| 351 | exit(EXIT_FAILURE); |
| 352 | } |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 353 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 354 | if (shdr.sh_type == SHT_SYMTAB) { |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 355 | read_symbol_table(lte, filename, |
| 356 | scn, &shdr, name, <e->symtab, |
| 357 | <e->symtab_count, <e->strtab); |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 358 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 359 | } else if (shdr.sh_type == SHT_DYNSYM) { |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 360 | read_symbol_table(lte, filename, |
| 361 | scn, &shdr, name, <e->dynsym, |
| 362 | <e->dynsym_count, <e->dynstr); |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 363 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 364 | } else if (shdr.sh_type == SHT_DYNAMIC) { |
| 365 | Elf_Data *data; |
| 366 | size_t j; |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 367 | |
| Joe Damato | 87f4f58 | 2010-11-08 15:47:36 -0800 | [diff] [blame] | 368 | lte->dyn_addr = shdr.sh_addr; |
| 369 | lte->dyn_sz = shdr.sh_size; |
| 370 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 371 | data = elf_getdata(scn, NULL); |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 372 | if (data == NULL || elf_getdata(scn, data) != NULL) { |
| 373 | fprintf(stderr, "Couldn't get .dynamic data" |
| 374 | " from \"%s\": %s\n", |
| 375 | filename, strerror(errno)); |
| 376 | exit(EXIT_FAILURE); |
| 377 | } |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 378 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 379 | for (j = 0; j < shdr.sh_size / shdr.sh_entsize; ++j) { |
| 380 | GElf_Dyn dyn; |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 381 | |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 382 | if (gelf_getdyn(data, j, &dyn) == NULL) { |
| 383 | fprintf(stderr, "Couldn't get .dynamic" |
| 384 | " data from \"%s\": %s\n", |
| 385 | filename, strerror(errno)); |
| 386 | exit(EXIT_FAILURE); |
| 387 | } |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 388 | if (dyn.d_tag == DT_JMPREL) |
| 389 | relplt_addr = dyn.d_un.d_ptr; |
| 390 | else if (dyn.d_tag == DT_PLTRELSZ) |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 391 | lte->relplt_size = dyn.d_un.d_val; |
| 392 | else if (dyn.d_tag == DT_SONAME) |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 393 | soname_offset = dyn.d_un.d_val; |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 394 | } |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 395 | } else if (shdr.sh_type == SHT_PROGBITS |
| 396 | || shdr.sh_type == SHT_NOBITS) { |
| 397 | if (strcmp(name, ".plt") == 0) { |
| 398 | lte->plt_addr = shdr.sh_addr; |
| 399 | lte->plt_size = shdr.sh_size; |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 400 | lte->plt_data = elf_loaddata(scn, &shdr); |
| 401 | if (lte->plt_data == NULL) |
| 402 | fprintf(stderr, |
| 403 | "Can't load .plt data\n"); |
| Petr Machata | 18c801c | 2012-04-07 01:24:08 +0200 | [diff] [blame] | 404 | lte->plt_flags = shdr.sh_flags; |
| Petr Machata | b3f8fef | 2006-11-30 14:45:07 +0100 | [diff] [blame] | 405 | } |
| 406 | #ifdef ARCH_SUPPORTS_OPD |
| 407 | else if (strcmp(name, ".opd") == 0) { |
| Paul Gilliam | 3f1219f | 2006-04-24 18:25:38 +0200 | [diff] [blame] | 408 | lte->opd_addr = (GElf_Addr *) (long) shdr.sh_addr; |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 409 | lte->opd_size = shdr.sh_size; |
| 410 | lte->opd = elf_rawdata(scn, NULL); |
| 411 | } |
| Petr Machata | b3f8fef | 2006-11-30 14:45:07 +0100 | [diff] [blame] | 412 | #endif |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 413 | } |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 414 | } |
| 415 | |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 416 | if (lte->dynsym == NULL || lte->dynstr == NULL) { |
| 417 | fprintf(stderr, "Couldn't find .dynsym or .dynstr in \"%s\"\n", |
| 418 | filename); |
| 419 | exit(EXIT_FAILURE); |
| 420 | } |
| Juan Cespedes | d914a20 | 2004-11-10 00:15:33 +0100 | [diff] [blame] | 421 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 422 | if (!relplt_addr || !lte->plt_addr) { |
| 423 | debug(1, "%s has no PLT relocations", filename); |
| 424 | lte->relplt = NULL; |
| 425 | lte->relplt_count = 0; |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 426 | } else if (lte->relplt_size == 0) { |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 427 | debug(1, "%s has unknown PLT size", filename); |
| 428 | lte->relplt = NULL; |
| 429 | lte->relplt_count = 0; |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 430 | } else { |
| Petr Machata | fe1c171 | 2010-10-27 16:57:34 +0200 | [diff] [blame] | 431 | |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 432 | for (i = 1; i < lte->ehdr.e_shnum; ++i) { |
| 433 | Elf_Scn *scn; |
| 434 | GElf_Shdr shdr; |
| 435 | |
| 436 | scn = elf_getscn(lte->elf, i); |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 437 | if (scn == NULL || gelf_getshdr(scn, &shdr) == NULL) { |
| 438 | fprintf(stderr, "Couldn't get section header" |
| 439 | " from \"%s\": %s\n", |
| 440 | filename, elf_errmsg(-1)); |
| 441 | exit(EXIT_FAILURE); |
| 442 | } |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 443 | if (shdr.sh_addr == relplt_addr |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 444 | && shdr.sh_size == lte->relplt_size) { |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 445 | lte->relplt = elf_getdata(scn, NULL); |
| 446 | lte->relplt_count = |
| 447 | shdr.sh_size / shdr.sh_entsize; |
| 448 | if (lte->relplt == NULL |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 449 | || elf_getdata(scn, lte->relplt) != NULL) { |
| 450 | fprintf(stderr, "Couldn't get .rel*.plt" |
| 451 | " data from \"%s\": %s\n", |
| 452 | filename, elf_errmsg(-1)); |
| 453 | exit(EXIT_FAILURE); |
| 454 | } |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 455 | break; |
| 456 | } |
| 457 | } |
| 458 | |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 459 | if (i == lte->ehdr.e_shnum) { |
| 460 | fprintf(stderr, |
| 461 | "Couldn't find .rel*.plt section in \"%s\"\n", |
| 462 | filename); |
| 463 | exit(EXIT_FAILURE); |
| 464 | } |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 465 | |
| 466 | debug(1, "%s %zd PLT relocations", filename, lte->relplt_count); |
| 467 | } |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 468 | |
| 469 | if (soname_offset != 0) |
| 470 | lte->soname = lte->dynstr + soname_offset; |
| 471 | |
| Petr Machata | 1974dbc | 2011-08-19 18:58:01 +0200 | [diff] [blame] | 472 | return 0; |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 473 | } |
| 474 | |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 475 | /* XXX temporarily non-static */ |
| Joe Damato | 7a2bdf8 | 2010-11-08 15:47:41 -0800 | [diff] [blame] | 476 | void |
| Juan Cespedes | f135052 | 2008-12-16 18:19:58 +0100 | [diff] [blame] | 477 | do_close_elf(struct ltelf *lte) { |
| Juan Cespedes | cd8976d | 2009-05-14 13:47:58 +0200 | [diff] [blame] | 478 | debug(DEBUG_FUNCTION, "do_close_elf()"); |
| Petr Machata | 4d9a91c | 2012-03-24 04:55:03 +0100 | [diff] [blame] | 479 | arch_elf_destroy(lte); |
| Ian Wienand | 2d45b1a | 2006-02-20 22:48:07 +0100 | [diff] [blame] | 480 | elf_end(lte->elf); |
| 481 | close(lte->fd); |
| Juan Cespedes | 1cd999a | 2001-07-03 00:46:04 +0200 | [diff] [blame] | 482 | } |
| 483 | |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 484 | static int |
| 485 | populate_plt(struct Process *proc, const char *filename, |
| 486 | struct ltelf *lte, struct library *lib) |
| 487 | { |
| 488 | size_t i; |
| 489 | for (i = 0; i < lte->relplt_count; ++i) { |
| 490 | GElf_Rel rel; |
| 491 | GElf_Rela rela; |
| 492 | GElf_Sym sym; |
| 493 | void *ret; |
| 494 | |
| 495 | if (lte->relplt->d_type == ELF_T_REL) { |
| 496 | ret = gelf_getrel(lte->relplt, i, &rel); |
| 497 | rela.r_offset = rel.r_offset; |
| 498 | rela.r_info = rel.r_info; |
| 499 | rela.r_addend = 0; |
| 500 | } else { |
| 501 | ret = gelf_getrela(lte->relplt, i, &rela); |
| 502 | } |
| 503 | |
| 504 | if (ret == NULL |
| 505 | || ELF64_R_SYM(rela.r_info) >= lte->dynsym_count |
| 506 | || gelf_getsym(lte->dynsym, ELF64_R_SYM(rela.r_info), |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 507 | &sym) == NULL) { |
| 508 | fprintf(stderr, |
| 509 | "Couldn't get relocation from \"%s\": %s\n", |
| 510 | filename, elf_errmsg(-1)); |
| 511 | exit(EXIT_FAILURE); |
| 512 | } |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 513 | |
| 514 | char const *name = lte->dynstr + sym.st_name; |
| 515 | |
| 516 | if (!filter_matches_symbol(options.plt_filter, name, lib)) |
| 517 | continue; |
| 518 | |
| Petr Machata | 218c5ff | 2012-04-15 04:22:39 +0200 | [diff] [blame] | 519 | struct library_symbol *libsym = NULL; |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 520 | switch (arch_elf_add_plt_entry(proc, lte, name, |
| 521 | &rela, i, &libsym)) { |
| 522 | case plt_default: |
| 523 | if (default_elf_add_plt_entry(proc, lte, name, |
| 524 | &rela, i, &libsym) < 0) |
| Petr Machata | 8eb0d93 | 2012-04-17 05:18:18 +0200 | [diff] [blame] | 525 | /* fall-through */ |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 526 | case plt_fail: |
| 527 | return -1; |
| Petr Machata | 8eb0d93 | 2012-04-17 05:18:18 +0200 | [diff] [blame] | 528 | /* fall-through */ |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 529 | case plt_ok: |
| 530 | if (libsym != NULL) |
| 531 | library_add_symbol(lib, libsym); |
| 532 | } |
| 533 | } |
| 534 | return 0; |
| 535 | } |
| 536 | |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 537 | /* When -x rules result in request to trace several aliases, we only |
| 538 | * want to add such symbol once. The only way that those symbols |
| 539 | * differ in is their name, e.g. in glibc you have __GI___libc_free, |
| 540 | * __cfree, __free, __libc_free, cfree and free all defined on the |
| 541 | * same address. So instead we keep this unique symbol struct for |
| 542 | * each address, and replace name in libsym with a shorter variant if |
| 543 | * we find it. */ |
| 544 | struct unique_symbol { |
| 545 | target_address_t addr; |
| 546 | struct library_symbol *libsym; |
| 547 | }; |
| 548 | |
| 549 | static int |
| 550 | unique_symbol_cmp(const void *key, const void *val) |
| 551 | { |
| 552 | const struct unique_symbol *sym_key = key; |
| 553 | const struct unique_symbol *sym_val = val; |
| 554 | return sym_key->addr != sym_val->addr; |
| 555 | } |
| 556 | |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 557 | static int |
| 558 | populate_this_symtab(struct Process *proc, const char *filename, |
| 559 | struct ltelf *lte, struct library *lib, |
| 560 | Elf_Data *symtab, const char *strtab, size_t size) |
| 561 | { |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 562 | /* Using sorted array would be arguably better, but this |
| 563 | * should be well enough for the number of symbols that we |
| 564 | * typically deal with. */ |
| 565 | size_t num_symbols = 0; |
| 566 | struct unique_symbol *symbols = malloc(sizeof(*symbols) * size); |
| 567 | if (symbols == NULL) { |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 568 | fprintf(stderr, "couldn't insert symbols for -x: %s\n", |
| 569 | strerror(errno)); |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 570 | return -1; |
| 571 | } |
| 572 | |
| Petr Machata | 40cc53b | 2012-04-07 01:25:38 +0200 | [diff] [blame] | 573 | GElf_Word secflags[lte->ehdr.e_shnum]; |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 574 | size_t i; |
| Petr Machata | 40cc53b | 2012-04-07 01:25:38 +0200 | [diff] [blame] | 575 | for (i = 1; i < lte->ehdr.e_shnum; ++i) { |
| 576 | Elf_Scn *scn = elf_getscn(lte->elf, i); |
| 577 | if (scn == NULL) |
| 578 | continue; |
| 579 | GElf_Shdr shdr; |
| 580 | if (gelf_getshdr(scn, &shdr) == NULL) |
| 581 | continue; |
| 582 | secflags[i] = shdr.sh_flags; |
| 583 | } |
| 584 | |
| 585 | size_t lib_len = strlen(lib->soname); |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 586 | for (i = 0; i < size; ++i) { |
| 587 | GElf_Sym sym; |
| Petr Machata | 074f68f | 2012-04-07 01:01:02 +0200 | [diff] [blame] | 588 | if (gelf_getsym(symtab, i, &sym) == NULL) { |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 589 | fail: |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 590 | fprintf(stderr, |
| 591 | "couldn't get symbol #%zd from %s: %s\n", |
| 592 | i, filename, elf_errmsg(-1)); |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 593 | continue; |
| 594 | } |
| 595 | |
| Petr Machata | 4de6b6b | 2012-04-04 14:06:09 +0200 | [diff] [blame] | 596 | /* XXX support IFUNC as well. */ |
| 597 | if (GELF_ST_TYPE(sym.st_info) != STT_FUNC |
| Petr Machata | 6c1c0bb | 2012-05-18 16:49:34 +0200 | [diff] [blame] | 598 | || sym.st_value == 0 |
| 599 | || sym.st_shndx == STN_UNDEF) |
| Petr Machata | 4de6b6b | 2012-04-04 14:06:09 +0200 | [diff] [blame] | 600 | continue; |
| 601 | |
| Petr Machata | 2bbeac4 | 2012-04-30 20:48:34 +0200 | [diff] [blame] | 602 | const char *orig_name = strtab + sym.st_name; |
| 603 | const char *version = strchr(orig_name, '@'); |
| 604 | size_t len = version != NULL ? (assert(version > orig_name), |
| 605 | (size_t)(version - orig_name)) |
| 606 | : strlen(orig_name); |
| 607 | char name[len + 1]; |
| 608 | memcpy(name, orig_name, len); |
| 609 | name[len] = 0; |
| 610 | |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 611 | if (!filter_matches_symbol(options.static_filter, name, lib)) |
| 612 | continue; |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 613 | |
| Petr Machata | ea8eb9a | 2012-04-17 01:32:07 +0200 | [diff] [blame] | 614 | target_address_t addr = (target_address_t) |
| 615 | (uintptr_t)(sym.st_value + lte->bias); |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 616 | target_address_t naddr; |
| Petr Machata | 40cc53b | 2012-04-07 01:25:38 +0200 | [diff] [blame] | 617 | |
| 618 | /* On arches that support OPD, the value of typical |
| 619 | * function symbol will be a pointer to .opd, but some |
| 620 | * will point directly to .text. We don't want to |
| 621 | * translate those. */ |
| 622 | if (secflags[sym.st_shndx] & SHF_EXECINSTR) { |
| 623 | naddr = addr; |
| Petr Machata | b1492df | 2012-04-30 21:01:40 +0200 | [diff] [blame] | 624 | } else if (arch_translate_address(lte, addr, &naddr) < 0) { |
| Petr Machata | cc0e1e4 | 2012-04-25 13:42:07 +0200 | [diff] [blame] | 625 | fprintf(stderr, |
| 626 | "couldn't translate address of %s@%s: %s\n", |
| 627 | name, lib->soname, strerror(errno)); |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 628 | continue; |
| 629 | } |
| Petr Machata | 40cc53b | 2012-04-07 01:25:38 +0200 | [diff] [blame] | 630 | |
| Petr Machata | 3840f68 | 2012-04-06 16:05:41 +0200 | [diff] [blame] | 631 | char *full_name; |
| 632 | if (lib->type != LT_LIBTYPE_MAIN) { |
| 633 | full_name = malloc(strlen(name) + 1 + lib_len + 1); |
| 634 | if (full_name == NULL) |
| 635 | goto fail; |
| 636 | sprintf(full_name, "%s@%s", name, lib->soname); |
| 637 | } else { |
| 638 | full_name = strdup(name); |
| 639 | if (full_name == NULL) |
| 640 | goto fail; |
| 641 | } |
| Petr Machata | 4de6b6b | 2012-04-04 14:06:09 +0200 | [diff] [blame] | 642 | |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 643 | /* Look whether we already have a symbol for this |
| 644 | * address. If not, add this one. */ |
| 645 | struct unique_symbol key = { naddr, NULL }; |
| 646 | struct unique_symbol *unique |
| 647 | = lsearch(&key, symbols, &num_symbols, |
| 648 | sizeof(*symbols), &unique_symbol_cmp); |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 649 | |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 650 | if (unique->libsym == NULL) { |
| 651 | struct library_symbol *libsym = malloc(sizeof(*libsym)); |
| Petr Machata | e8d9076 | 2012-04-15 04:28:31 +0200 | [diff] [blame] | 652 | if (libsym == NULL |
| 653 | || library_symbol_init(libsym, naddr, full_name, |
| 654 | 1, LS_TOPLT_NONE) < 0) { |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 655 | --num_symbols; |
| 656 | goto fail; |
| 657 | } |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 658 | unique->libsym = libsym; |
| 659 | unique->addr = naddr; |
| 660 | |
| 661 | } else if (strlen(full_name) < strlen(unique->libsym->name)) { |
| 662 | library_symbol_set_name(unique->libsym, full_name, 1); |
| 663 | |
| 664 | } else { |
| 665 | free(full_name); |
| 666 | } |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 667 | } |
| Petr Machata | 157cc4d | 2012-04-04 19:00:34 +0200 | [diff] [blame] | 668 | |
| 669 | for (i = 0; i < num_symbols; ++i) { |
| 670 | assert(symbols[i].libsym != NULL); |
| 671 | library_add_symbol(lib, symbols[i].libsym); |
| 672 | } |
| 673 | |
| 674 | free(symbols); |
| 675 | |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 676 | return 0; |
| 677 | } |
| 678 | |
| 679 | static int |
| 680 | populate_symtab(struct Process *proc, const char *filename, |
| 681 | struct ltelf *lte, struct library *lib) |
| 682 | { |
| 683 | if (lte->symtab != NULL && lte->strtab != NULL) |
| 684 | return populate_this_symtab(proc, filename, lte, lib, |
| 685 | lte->symtab, lte->strtab, |
| 686 | lte->symtab_count); |
| 687 | else |
| 688 | return populate_this_symtab(proc, filename, lte, lib, |
| 689 | lte->dynsym, lte->dynstr, |
| 690 | lte->dynsym_count); |
| 691 | } |
| 692 | |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 693 | int |
| 694 | ltelf_read_library(struct library *lib, struct Process *proc, |
| 695 | const char *filename, GElf_Addr bias) |
| Petr Machata | e84fa00 | 2012-02-07 13:43:03 +0100 | [diff] [blame] | 696 | { |
| Petr Machata | 29add4f | 2012-02-18 16:38:05 +0100 | [diff] [blame] | 697 | struct ltelf lte = {}; |
| Petr Machata | b120fdf | 2012-03-21 05:05:46 +0100 | [diff] [blame] | 698 | if (do_init_elf(<e, filename, bias) < 0) |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 699 | return -1; |
| Petr Machata | e0615ab | 2012-04-17 05:17:48 +0200 | [diff] [blame] | 700 | if (arch_elf_init(<e, lib) < 0) { |
| 701 | fprintf(stderr, "Backend initialization failed.\n"); |
| 702 | return -1; |
| 703 | } |
| 704 | |
| Petr Machata | e67635d | 2012-03-21 03:37:39 +0100 | [diff] [blame] | 705 | proc->e_machine = lte.ehdr.e_machine; |
| Joe Damato | f0bd98b | 2010-11-08 15:47:42 -0800 | [diff] [blame] | 706 | |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 707 | int status = 0; |
| Petr Machata | fc6ff18 | 2012-04-04 13:11:50 +0200 | [diff] [blame] | 708 | if (lib == NULL) |
| 709 | goto fail; |
| 710 | |
| 711 | /* Note that we set soname and pathname as soon as they are |
| 712 | * allocated, so in case of further errors, this get released |
| 713 | * when LIB is release, which should happen in the caller when |
| 714 | * we return error. */ |
| 715 | |
| 716 | if (lib->pathname == NULL) { |
| 717 | char *pathname = strdup(filename); |
| 718 | if (pathname == NULL) |
| 719 | goto fail; |
| Petr Machata | f13afd5 | 2012-04-14 02:30:31 +0200 | [diff] [blame] | 720 | library_set_pathname(lib, pathname, 1); |
| Joe Damato | f0bd98b | 2010-11-08 15:47:42 -0800 | [diff] [blame] | 721 | } |
| 722 | |
| Petr Machata | 0b55b58 | 2012-04-02 00:38:46 +0200 | [diff] [blame] | 723 | if (lte.soname != NULL) { |
| Petr Machata | fc6ff18 | 2012-04-04 13:11:50 +0200 | [diff] [blame] | 724 | char *soname = strdup(lte.soname); |
| 725 | if (soname == NULL) |
| 726 | goto fail; |
| 727 | library_set_soname(lib, soname, 1); |
| Petr Machata | 0b55b58 | 2012-04-02 00:38:46 +0200 | [diff] [blame] | 728 | } else { |
| Petr Machata | fc6ff18 | 2012-04-04 13:11:50 +0200 | [diff] [blame] | 729 | const char *soname = rindex(lib->pathname, '/') + 1; |
| 730 | if (soname == NULL) |
| 731 | soname = lib->pathname; |
| 732 | library_set_soname(lib, soname, 0); |
| Petr Machata | 0b55b58 | 2012-04-02 00:38:46 +0200 | [diff] [blame] | 733 | } |
| Joe Damato | fa2aefc | 2010-10-30 19:56:50 -0700 | [diff] [blame] | 734 | |
| Petr Machata | ea8eb9a | 2012-04-17 01:32:07 +0200 | [diff] [blame] | 735 | /* XXX The double cast should be removed when |
| 736 | * target_address_t becomes integral type. */ |
| 737 | target_address_t entry = (target_address_t)(uintptr_t)lte.entry_addr; |
| Petr Machata | b1492df | 2012-04-30 21:01:40 +0200 | [diff] [blame] | 738 | if (arch_translate_address(<e, entry, &entry) < 0) |
| Petr Machata | b120fdf | 2012-03-21 05:05:46 +0100 | [diff] [blame] | 739 | goto fail; |
| 740 | |
| Petr Machata | ea8eb9a | 2012-04-17 01:32:07 +0200 | [diff] [blame] | 741 | /* XXX The double cast should be removed when |
| 742 | * target_address_t becomes integral type. */ |
| 743 | lib->base = (target_address_t)(uintptr_t)lte.base_addr; |
| Petr Machata | b120fdf | 2012-03-21 05:05:46 +0100 | [diff] [blame] | 744 | lib->entry = entry; |
| Petr Machata | ea8eb9a | 2012-04-17 01:32:07 +0200 | [diff] [blame] | 745 | /* XXX The double cast should be removed when |
| 746 | * target_address_t becomes integral type. */ |
| 747 | lib->dyn_addr = (target_address_t)(uintptr_t)lte.dyn_addr; |
| Ian Wienand | 9a2ad35 | 2006-02-20 22:44:45 +0100 | [diff] [blame] | 748 | |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 749 | if (filter_matches_library(options.plt_filter, lib) |
| 750 | && populate_plt(proc, filename, <e, lib) < 0) |
| 751 | goto fail; |
| Joe Damato | e2a8f57 | 2010-11-08 15:47:40 -0800 | [diff] [blame] | 752 | |
| Petr Machata | da3edbf | 2012-04-04 02:20:21 +0200 | [diff] [blame] | 753 | if (filter_matches_library(options.static_filter, lib) |
| 754 | && populate_symtab(proc, filename, <e, lib) < 0) |
| 755 | goto fail; |
| 756 | |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 757 | done: |
| 758 | do_close_elf(<e); |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 759 | return status; |
| Petr Machata | fc6ff18 | 2012-04-04 13:11:50 +0200 | [diff] [blame] | 760 | |
| 761 | fail: |
| 762 | status = -1; |
| 763 | goto done; |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 764 | } |
| Petr Machata | e84fa00 | 2012-02-07 13:43:03 +0100 | [diff] [blame] | 765 | |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 766 | struct library * |
| 767 | ltelf_read_main_binary(struct Process *proc, const char *path) |
| 768 | { |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 769 | struct library *lib = malloc(sizeof(*lib)); |
| 770 | if (lib == NULL) |
| 771 | return NULL; |
| 772 | library_init(lib, LT_LIBTYPE_MAIN); |
| Petr Machata | fc6ff18 | 2012-04-04 13:11:50 +0200 | [diff] [blame] | 773 | library_set_pathname(lib, path, 0); |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 774 | |
| Petr Machata | fc6ff18 | 2012-04-04 13:11:50 +0200 | [diff] [blame] | 775 | /* There is a race between running the process and reading its |
| 776 | * binary for internal consumption. So open the binary from |
| 777 | * the /proc filesystem. XXX Note that there is similar race |
| 778 | * for libraries, but there we don't have a nice answer like |
| 779 | * that. Presumably we could read the DSOs from the process |
| 780 | * memory image, but that's not currently done. */ |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 781 | char *fname = pid2name(proc->pid); |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 782 | if (ltelf_read_library(lib, proc, fname, 0) < 0) { |
| 783 | library_destroy(lib); |
| 784 | free(lib); |
| 785 | return NULL; |
| Petr Machata | 0b55b58 | 2012-04-02 00:38:46 +0200 | [diff] [blame] | 786 | } |
| Petr Machata | b5f80ac | 2012-04-04 01:46:18 +0200 | [diff] [blame] | 787 | |
| Petr Machata | 2b46cfc | 2012-02-18 11:17:29 +0100 | [diff] [blame] | 788 | return lib; |
| Juan Cespedes | 96935a9 | 1997-08-09 23:45:39 +0200 | [diff] [blame] | 789 | } |