| 1 | /* Copyright 2013-2016 Free Software Foundation, Inc. |
| 2 | This program is free software; you can redistribute it and/or modify |
| 3 | it under the terms of the GNU General Public License as published by |
| 4 | the Free Software Foundation; either version 3 of the License, or |
| 5 | (at your option) any later version. |
| 6 | |
| 7 | This program is distributed in the hope that it will be useful, |
| 8 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 9 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 10 | GNU General Public License for more details. |
| 11 | |
| 12 | You should have received a copy of the GNU General Public License |
| 13 | along with this program. If not, see <http://www.gnu.org/licenses/>. |
| 14 | */ |
| 15 | |
| 16 | #include <unistd.h> |
| 17 | #include <fcntl.h> |
| 18 | #include <limits.h> |
| 19 | #include <stdio.h> |
| 20 | #include <stdlib.h> |
| 21 | #include <string.h> |
| 22 | #include <sys/mman.h> |
| 23 | |
| 24 | #include "sym-file-loader.h" |
| 25 | |
| 26 | #include <inttypes.h> |
| 27 | #include <ansidecl.h> |
| 28 | #include <elf/common.h> |
| 29 | #include <elf/external.h> |
| 30 | |
| 31 | #ifdef TARGET_LP64 |
| 32 | |
| 33 | typedef Elf64_External_Phdr Elf_External_Phdr; |
| 34 | typedef Elf64_External_Ehdr Elf_External_Ehdr; |
| 35 | typedef Elf64_External_Shdr Elf_External_Shdr; |
| 36 | typedef Elf64_External_Sym Elf_External_Sym; |
| 37 | typedef uint64_t Elf_Addr; |
| 38 | |
| 39 | #elif defined TARGET_ILP32 |
| 40 | |
| 41 | typedef Elf32_External_Phdr Elf_External_Phdr; |
| 42 | typedef Elf32_External_Ehdr Elf_External_Ehdr; |
| 43 | typedef Elf32_External_Shdr Elf_External_Shdr; |
| 44 | typedef Elf32_External_Sym Elf_External_Sym; |
| 45 | typedef uint32_t Elf_Addr; |
| 46 | |
| 47 | #endif |
| 48 | |
| 49 | #define GET(hdr, field) (\ |
| 50 | sizeof ((hdr)->field) == 1 ? (uint64_t) (hdr)->field[0] : \ |
| 51 | sizeof ((hdr)->field) == 2 ? (uint64_t) *(uint16_t *) (hdr)->field : \ |
| 52 | sizeof ((hdr)->field) == 4 ? (uint64_t) *(uint32_t *) (hdr)->field : \ |
| 53 | sizeof ((hdr)->field) == 8 ? *(uint64_t *) (hdr)->field : \ |
| 54 | *(uint64_t *) NULL) |
| 55 | |
| 56 | #define GETADDR(hdr, field) (\ |
| 57 | sizeof ((hdr)->field) == sizeof (Elf_Addr) ? *(Elf_Addr *) (hdr)->field : \ |
| 58 | *(Elf_Addr *) NULL) |
| 59 | |
| 60 | struct segment |
| 61 | { |
| 62 | uint8_t *mapped_addr; |
| 63 | size_t mapped_size; |
| 64 | Elf_External_Phdr *phdr; |
| 65 | struct segment *next; |
| 66 | }; |
| 67 | |
| 68 | struct library |
| 69 | { |
| 70 | int fd; |
| 71 | Elf_External_Ehdr *ehdr; |
| 72 | struct segment *segments; |
| 73 | }; |
| 74 | |
| 75 | static Elf_External_Shdr *find_shdr (Elf_External_Ehdr *ehdr, |
| 76 | const char *section); |
| 77 | static int translate_offset (uint64_t file_offset, struct segment *seg, |
| 78 | void **addr); |
| 79 | |
| 80 | #ifdef TARGET_LP64 |
| 81 | |
| 82 | uint8_t |
| 83 | elf_st_type (uint8_t st_info) |
| 84 | { |
| 85 | return ELF64_ST_TYPE (st_info); |
| 86 | } |
| 87 | |
| 88 | #elif defined TARGET_ILP32 |
| 89 | |
| 90 | uint8_t |
| 91 | elf_st_type (uint8_t st_info) |
| 92 | { |
| 93 | return ELF32_ST_TYPE (st_info); |
| 94 | } |
| 95 | |
| 96 | #endif |
| 97 | |
| 98 | /* Load a program segment. */ |
| 99 | |
| 100 | static struct segment * |
| 101 | load (uint8_t *addr, Elf_External_Phdr *phdr, struct segment *tail_seg) |
| 102 | { |
| 103 | struct segment *seg = NULL; |
| 104 | uint8_t *mapped_addr = NULL; |
| 105 | size_t mapped_size = 0; |
| 106 | void *from = NULL; |
| 107 | void *to = NULL; |
| 108 | |
| 109 | /* For the sake of simplicity all operations are permitted. */ |
| 110 | unsigned perm = PROT_READ | PROT_WRITE | PROT_EXEC; |
| 111 | |
| 112 | mapped_addr = (uint8_t *) mmap ((void *) GETADDR (phdr, p_vaddr), |
| 113 | GET (phdr, p_memsz), perm, |
| 114 | MAP_ANONYMOUS | MAP_PRIVATE, -1, 0); |
| 115 | mapped_size = GET (phdr, p_memsz); |
| 116 | |
| 117 | from = (void *) (addr + GET (phdr, p_offset)); |
| 118 | to = (void *) mapped_addr; |
| 119 | |
| 120 | memcpy (to, from, GET (phdr, p_filesz)); |
| 121 | |
| 122 | seg = (struct segment *) malloc (sizeof (struct segment)); |
| 123 | |
| 124 | if (seg == 0) |
| 125 | return 0; |
| 126 | |
| 127 | seg->mapped_addr = mapped_addr; |
| 128 | seg->mapped_size = mapped_size; |
| 129 | seg->phdr = phdr; |
| 130 | seg->next = 0; |
| 131 | |
| 132 | if (tail_seg != 0) |
| 133 | tail_seg->next = seg; |
| 134 | |
| 135 | return seg; |
| 136 | } |
| 137 | |
| 138 | #ifdef __linux__ |
| 139 | # define SELF_LINK "/proc/self/exe" |
| 140 | #elif defined NETBSD |
| 141 | # define SELF_LINK "/proc/curproc/exe" |
| 142 | #elif defined __OpenBSD__ || defined __FreeBSD__ || defined __DragonFly__ |
| 143 | # define SELF_LINK "/proc/curproc/file" |
| 144 | #elif defined SunOS |
| 145 | # define SELF_LINK "/proc/self/path/a.out" |
| 146 | #endif |
| 147 | |
| 148 | /* Like RPATH=$ORIGIN, return the dirname of the current |
| 149 | executable. */ |
| 150 | |
| 151 | static const char * |
| 152 | get_origin (void) |
| 153 | { |
| 154 | static char self_path[PATH_MAX]; |
| 155 | static ssize_t self_path_len; |
| 156 | |
| 157 | if (self_path_len == 0) |
| 158 | { |
| 159 | #ifdef SELF_LINK |
| 160 | self_path_len = readlink (SELF_LINK, self_path, PATH_MAX - 1); |
| 161 | if (self_path_len != -1) |
| 162 | { |
| 163 | char *dirsep; |
| 164 | |
| 165 | self_path[self_path_len] = '\0'; |
| 166 | dirsep = strrchr (self_path, '/'); |
| 167 | *dirsep = '\0'; |
| 168 | } |
| 169 | #else |
| 170 | self_path_len = -1; |
| 171 | #endif |
| 172 | } |
| 173 | |
| 174 | if (self_path_len == -1) |
| 175 | return NULL; |
| 176 | else |
| 177 | return self_path; |
| 178 | } |
| 179 | |
| 180 | /* Unload/unmap a segment. */ |
| 181 | |
| 182 | static void |
| 183 | unload (struct segment *seg) |
| 184 | { |
| 185 | munmap (seg->mapped_addr, seg->mapped_size); |
| 186 | free (seg); |
| 187 | } |
| 188 | |
| 189 | void |
| 190 | unload_shlib (struct library *lib) |
| 191 | { |
| 192 | struct segment *seg, *next_seg; |
| 193 | |
| 194 | for (seg = lib->segments; seg != NULL; seg = next_seg) |
| 195 | { |
| 196 | next_seg = seg->next; |
| 197 | unload (seg); |
| 198 | } |
| 199 | |
| 200 | close (lib->fd); |
| 201 | free (lib); |
| 202 | } |
| 203 | |
| 204 | /* Mini shared library loader. No reallocation |
| 205 | is performed for the sake of simplicity. */ |
| 206 | |
| 207 | struct library * |
| 208 | load_shlib (const char *file) |
| 209 | { |
| 210 | struct library *lib; |
| 211 | uint64_t i; |
| 212 | int fd = -1; |
| 213 | off_t fsize; |
| 214 | uint8_t *addr; |
| 215 | Elf_External_Ehdr *ehdr; |
| 216 | Elf_External_Phdr *phdr; |
| 217 | struct segment *head_seg = NULL; |
| 218 | struct segment *tail_seg = NULL; |
| 219 | const char *origin; |
| 220 | char *path; |
| 221 | |
| 222 | /* Map the lib in memory for reading. |
| 223 | |
| 224 | If the file name is relative, try looking it up relative to the |
| 225 | main executable's path. I.e., emulate RPATH=$ORIGIN. */ |
| 226 | if (file[0] != '/') |
| 227 | { |
| 228 | origin = get_origin (); |
| 229 | if (origin == NULL) |
| 230 | { |
| 231 | fprintf (stderr, "get_origin not implemented."); |
| 232 | return NULL; |
| 233 | } |
| 234 | |
| 235 | path = alloca (strlen (origin) + 1 + strlen (file) + 1); |
| 236 | sprintf (path, "%s/%s", origin, file); |
| 237 | fd = open (path, O_RDONLY); |
| 238 | } |
| 239 | |
| 240 | if (fd < 0) |
| 241 | fd = open (file, O_RDONLY); |
| 242 | |
| 243 | if (fd < 0) |
| 244 | { |
| 245 | perror ("fopen failed."); |
| 246 | return NULL; |
| 247 | } |
| 248 | |
| 249 | fsize = lseek (fd, 0, SEEK_END); |
| 250 | |
| 251 | if (fsize < 0) |
| 252 | { |
| 253 | perror ("lseek failed."); |
| 254 | return NULL; |
| 255 | } |
| 256 | |
| 257 | addr = (uint8_t *) mmap (NULL, fsize, PROT_READ, MAP_PRIVATE, fd, 0); |
| 258 | if (addr == (uint8_t *) -1) |
| 259 | { |
| 260 | perror ("mmap failed."); |
| 261 | return NULL; |
| 262 | } |
| 263 | |
| 264 | /* Check if the lib is an ELF file. */ |
| 265 | ehdr = (Elf_External_Ehdr *) addr; |
| 266 | if (ehdr->e_ident[EI_MAG0] != ELFMAG0 |
| 267 | || ehdr->e_ident[EI_MAG1] != ELFMAG1 |
| 268 | || ehdr->e_ident[EI_MAG2] != ELFMAG2 |
| 269 | || ehdr->e_ident[EI_MAG3] != ELFMAG3) |
| 270 | { |
| 271 | printf ("Not an ELF file: %x\n", ehdr->e_ident[EI_MAG0]); |
| 272 | return NULL; |
| 273 | } |
| 274 | |
| 275 | if (ehdr->e_ident[EI_CLASS] == ELFCLASS32) |
| 276 | { |
| 277 | if (sizeof (void *) != 4) |
| 278 | { |
| 279 | printf ("Architecture mismatch."); |
| 280 | return NULL; |
| 281 | } |
| 282 | } |
| 283 | else if (ehdr->e_ident[EI_CLASS] == ELFCLASS64) |
| 284 | { |
| 285 | if (sizeof (void *) != 8) |
| 286 | { |
| 287 | printf ("Architecture mismatch."); |
| 288 | return NULL; |
| 289 | } |
| 290 | } |
| 291 | |
| 292 | lib = malloc (sizeof (struct library)); |
| 293 | if (lib == NULL) |
| 294 | { |
| 295 | printf ("malloc failed."); |
| 296 | return NULL; |
| 297 | } |
| 298 | |
| 299 | lib->fd = fd; |
| 300 | |
| 301 | /* Load the program segments. For the sake of simplicity |
| 302 | assume that no reallocation is needed. */ |
| 303 | phdr = (Elf_External_Phdr *) (addr + GET (ehdr, e_phoff)); |
| 304 | for (i = 0; i < GET (ehdr, e_phnum); i++, phdr++) |
| 305 | { |
| 306 | if (GET (phdr, p_type) == PT_LOAD) |
| 307 | { |
| 308 | struct segment *next_seg = load (addr, phdr, tail_seg); |
| 309 | if (next_seg == 0) |
| 310 | continue; |
| 311 | tail_seg = next_seg; |
| 312 | if (head_seg == 0) |
| 313 | head_seg = next_seg; |
| 314 | } |
| 315 | } |
| 316 | lib->ehdr = ehdr; |
| 317 | lib->segments = head_seg; |
| 318 | return lib; |
| 319 | } |
| 320 | |
| 321 | int |
| 322 | get_text_addr (struct library *lib, void **text_addr) |
| 323 | { |
| 324 | Elf_External_Shdr *text; |
| 325 | |
| 326 | /* Get the text section. */ |
| 327 | text = find_shdr (lib->ehdr, ".text"); |
| 328 | if (text == NULL) |
| 329 | return -1; |
| 330 | |
| 331 | if (translate_offset (GET (text, sh_offset), lib->segments, text_addr) |
| 332 | != 0) |
| 333 | return -1; |
| 334 | |
| 335 | return 0; |
| 336 | } |
| 337 | |
| 338 | /* Return the section-header table. */ |
| 339 | |
| 340 | Elf_External_Shdr * |
| 341 | find_shdrtab (Elf_External_Ehdr *ehdr) |
| 342 | { |
| 343 | return (Elf_External_Shdr *) (((uint8_t *) ehdr) + GET (ehdr, e_shoff)); |
| 344 | } |
| 345 | |
| 346 | /* Return the string table of the section headers. */ |
| 347 | |
| 348 | const char * |
| 349 | find_shstrtab (Elf_External_Ehdr *ehdr, uint64_t *size) |
| 350 | { |
| 351 | const Elf_External_Shdr *shdr; |
| 352 | const Elf_External_Shdr *shstr; |
| 353 | |
| 354 | if (GET (ehdr, e_shnum) <= GET (ehdr, e_shstrndx)) |
| 355 | { |
| 356 | printf ("The index of the string table is corrupt."); |
| 357 | return NULL; |
| 358 | } |
| 359 | |
| 360 | shdr = find_shdrtab (ehdr); |
| 361 | |
| 362 | shstr = &shdr[GET (ehdr, e_shstrndx)]; |
| 363 | *size = GET (shstr, sh_size); |
| 364 | return ((const char *) ehdr) + GET (shstr, sh_offset); |
| 365 | } |
| 366 | |
| 367 | /* Return the string table named SECTION. */ |
| 368 | |
| 369 | const char * |
| 370 | find_strtab (Elf_External_Ehdr *ehdr, |
| 371 | const char *section, uint64_t *strtab_size) |
| 372 | { |
| 373 | uint64_t shstrtab_size = 0; |
| 374 | const char *shstrtab; |
| 375 | uint64_t i; |
| 376 | const Elf_External_Shdr *shdr = find_shdrtab (ehdr); |
| 377 | |
| 378 | /* Get the string table of the section headers. */ |
| 379 | shstrtab = find_shstrtab (ehdr, &shstrtab_size); |
| 380 | if (shstrtab == NULL) |
| 381 | return NULL; |
| 382 | |
| 383 | for (i = 0; i < GET (ehdr, e_shnum); i++) |
| 384 | { |
| 385 | uint64_t name = GET (shdr + i, sh_name); |
| 386 | if (GET (shdr + i, sh_type) == SHT_STRTAB && name <= shstrtab_size |
| 387 | && strcmp ((const char *) &shstrtab[name], section) == 0) |
| 388 | { |
| 389 | *strtab_size = GET (shdr + i, sh_size); |
| 390 | return ((const char *) ehdr) + GET (shdr + i, sh_offset); |
| 391 | } |
| 392 | |
| 393 | } |
| 394 | return NULL; |
| 395 | } |
| 396 | |
| 397 | /* Return the section header named SECTION. */ |
| 398 | |
| 399 | static Elf_External_Shdr * |
| 400 | find_shdr (Elf_External_Ehdr *ehdr, const char *section) |
| 401 | { |
| 402 | uint64_t shstrtab_size = 0; |
| 403 | const char *shstrtab; |
| 404 | uint64_t i; |
| 405 | |
| 406 | /* Get the string table of the section headers. */ |
| 407 | shstrtab = find_shstrtab (ehdr, &shstrtab_size); |
| 408 | if (shstrtab == NULL) |
| 409 | return NULL; |
| 410 | |
| 411 | Elf_External_Shdr *shdr = find_shdrtab (ehdr); |
| 412 | for (i = 0; i < GET (ehdr, e_shnum); i++) |
| 413 | { |
| 414 | uint64_t name = GET (shdr + i, sh_name); |
| 415 | if (name <= shstrtab_size) |
| 416 | { |
| 417 | if (strcmp ((const char *) &shstrtab[name], section) == 0) |
| 418 | return &shdr[i]; |
| 419 | } |
| 420 | |
| 421 | } |
| 422 | return NULL; |
| 423 | } |
| 424 | |
| 425 | /* Return the symbol table. */ |
| 426 | |
| 427 | static Elf_External_Sym * |
| 428 | find_symtab (Elf_External_Ehdr *ehdr, uint64_t *symtab_size) |
| 429 | { |
| 430 | uint64_t i; |
| 431 | const Elf_External_Shdr *shdr = find_shdrtab (ehdr); |
| 432 | |
| 433 | for (i = 0; i < GET (ehdr, e_shnum); i++) |
| 434 | { |
| 435 | if (GET (shdr + i, sh_type) == SHT_SYMTAB) |
| 436 | { |
| 437 | *symtab_size = GET (shdr + i, sh_size) / sizeof (Elf_External_Sym); |
| 438 | return (Elf_External_Sym *) (((const char *) ehdr) + |
| 439 | GET (shdr + i, sh_offset)); |
| 440 | } |
| 441 | } |
| 442 | return NULL; |
| 443 | } |
| 444 | |
| 445 | /* Translate a file offset to an address in a loaded segment. */ |
| 446 | |
| 447 | static int |
| 448 | translate_offset (uint64_t file_offset, struct segment *seg, void **addr) |
| 449 | { |
| 450 | while (seg) |
| 451 | { |
| 452 | uint64_t p_from, p_to; |
| 453 | |
| 454 | Elf_External_Phdr *phdr = seg->phdr; |
| 455 | |
| 456 | if (phdr == NULL) |
| 457 | { |
| 458 | seg = seg->next; |
| 459 | continue; |
| 460 | } |
| 461 | |
| 462 | p_from = GET (phdr, p_offset); |
| 463 | p_to = p_from + GET (phdr, p_filesz); |
| 464 | |
| 465 | if (p_from <= file_offset && file_offset < p_to) |
| 466 | { |
| 467 | *addr = (void *) (seg->mapped_addr + (file_offset - p_from)); |
| 468 | return 0; |
| 469 | } |
| 470 | seg = seg->next; |
| 471 | } |
| 472 | |
| 473 | return -1; |
| 474 | } |
| 475 | |
| 476 | /* Lookup the address of FUNC. */ |
| 477 | |
| 478 | int |
| 479 | lookup_function (struct library *lib, const char *func, void **addr) |
| 480 | { |
| 481 | const char *strtab; |
| 482 | uint64_t strtab_size = 0; |
| 483 | Elf_External_Sym *symtab; |
| 484 | uint64_t symtab_size = 0; |
| 485 | uint64_t i; |
| 486 | Elf_External_Ehdr *ehdr = lib->ehdr; |
| 487 | struct segment *seg = lib->segments; |
| 488 | |
| 489 | /* Get the string table for the symbols. */ |
| 490 | strtab = find_strtab (ehdr, ".strtab", &strtab_size); |
| 491 | if (strtab == NULL) |
| 492 | { |
| 493 | printf (".strtab not found."); |
| 494 | return -1; |
| 495 | } |
| 496 | |
| 497 | /* Get the symbol table. */ |
| 498 | symtab = find_symtab (ehdr, &symtab_size); |
| 499 | if (symtab == NULL) |
| 500 | { |
| 501 | printf ("symbol table not found."); |
| 502 | return -1; |
| 503 | } |
| 504 | |
| 505 | for (i = 0; i < symtab_size; i++) |
| 506 | { |
| 507 | Elf_External_Sym *sym = &symtab[i]; |
| 508 | |
| 509 | if (elf_st_type (GET (sym, st_info)) != STT_FUNC) |
| 510 | continue; |
| 511 | |
| 512 | if (GET (sym, st_name) < strtab_size) |
| 513 | { |
| 514 | const char *name = &strtab[GET (sym, st_name)]; |
| 515 | if (strcmp (name, func) == 0) |
| 516 | { |
| 517 | |
| 518 | uint64_t offset = GET (sym, st_value); |
| 519 | return translate_offset (offset, seg, addr); |
| 520 | } |
| 521 | } |
| 522 | } |
| 523 | |
| 524 | return -1; |
| 525 | } |