1 /*- 2 * Copyright (c) 2007-2013 Kai Wang 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 */ 26 27 #include <sys/param.h> 28 #include <sys/stat.h> 29 30 #include <err.h> 31 #include <errno.h> 32 #include <fcntl.h> 33 #include <getopt.h> 34 #include <libelftc.h> 35 #include <stdio.h> 36 #include <stdlib.h> 37 #include <string.h> 38 #include <unistd.h> 39 40 #include "elfcopy.h" 41 42 ELFTC_VCSID("$Id: main.c 3757 2019-06-28 01:15:28Z emaste $"); 43 44 enum options 45 { 46 ECP_ADD_GNU_DEBUGLINK, 47 ECP_ADD_SECTION, 48 ECP_CHANGE_ADDR, 49 ECP_CHANGE_SEC_ADDR, 50 ECP_CHANGE_SEC_LMA, 51 ECP_CHANGE_SEC_VMA, 52 ECP_CHANGE_START, 53 ECP_CHANGE_WARN, 54 ECP_GAP_FILL, 55 ECP_GLOBALIZE_SYMBOL, 56 ECP_GLOBALIZE_SYMBOLS, 57 ECP_KEEP_SYMBOLS, 58 ECP_KEEP_GLOBAL_SYMBOLS, 59 ECP_LOCALIZE_HIDDEN, 60 ECP_LOCALIZE_SYMBOLS, 61 ECP_NO_CHANGE_WARN, 62 ECP_ONLY_DEBUG, 63 ECP_ONLY_DWO, 64 ECP_PAD_TO, 65 ECP_PREFIX_ALLOC, 66 ECP_PREFIX_SEC, 67 ECP_PREFIX_SYM, 68 ECP_REDEF_SYMBOL, 69 ECP_REDEF_SYMBOLS, 70 ECP_RENAME_SECTION, 71 ECP_SET_OSABI, 72 ECP_SET_SEC_FLAGS, 73 ECP_SET_START, 74 ECP_SREC_FORCE_S3, 75 ECP_SREC_LEN, 76 ECP_STRIP_DWO, 77 ECP_STRIP_SYMBOLS, 78 ECP_STRIP_UNNEEDED, 79 ECP_WEAKEN_ALL, 80 ECP_WEAKEN_SYMBOLS 81 }; 82 83 static struct option mcs_longopts[] = 84 { 85 { "help", no_argument, NULL, 'h' }, 86 { "version", no_argument, NULL, 'V' }, 87 { NULL, 0, NULL, 0 } 88 }; 89 90 static struct option strip_longopts[] = 91 { 92 {"discard-all", no_argument, NULL, 'x'}, 93 {"discard-locals", no_argument, NULL, 'X'}, 94 {"help", no_argument, NULL, 'h'}, 95 {"input-target", required_argument, NULL, 'I'}, 96 {"keep-symbol", required_argument, NULL, 'K'}, 97 {"only-keep-debug", no_argument, NULL, ECP_ONLY_DEBUG}, 98 {"output-file", required_argument, NULL, 'o'}, 99 {"output-target", required_argument, NULL, 'O'}, 100 {"preserve-dates", no_argument, NULL, 'p'}, 101 {"remove-section", required_argument, NULL, 'R'}, 102 {"strip-all", no_argument, NULL, 's'}, 103 {"strip-debug", no_argument, NULL, 'S'}, 104 {"strip-symbol", required_argument, NULL, 'N'}, 105 {"strip-unneeded", no_argument, NULL, ECP_STRIP_UNNEEDED}, 106 {"version", no_argument, NULL, 'V'}, 107 {"wildcard", no_argument, NULL, 'w'}, 108 {NULL, 0, NULL, 0} 109 }; 110 111 static struct option elfcopy_longopts[] = 112 { 113 {"add-gnu-debuglink", required_argument, NULL, ECP_ADD_GNU_DEBUGLINK}, 114 {"add-section", required_argument, NULL, ECP_ADD_SECTION}, 115 {"adjust-section-vma", required_argument, NULL, ECP_CHANGE_SEC_ADDR}, 116 {"adjust-vma", required_argument, NULL, ECP_CHANGE_ADDR}, 117 {"adjust-start", required_argument, NULL, ECP_CHANGE_START}, 118 {"adjust-warnings", no_argument, NULL, ECP_CHANGE_WARN}, 119 {"binary-architecture", required_argument, NULL, 'B'}, 120 {"change-addresses", required_argument, NULL, ECP_CHANGE_ADDR}, 121 {"change-section-address", required_argument, NULL, 122 ECP_CHANGE_SEC_ADDR}, 123 {"change-section-lma", required_argument, NULL, ECP_CHANGE_SEC_LMA}, 124 {"change-section-vma", required_argument, NULL, ECP_CHANGE_SEC_VMA}, 125 {"change-start", required_argument, NULL, ECP_CHANGE_START}, 126 {"change-warnings", no_argument, NULL, ECP_CHANGE_WARN}, 127 {"discard-all", no_argument, NULL, 'x'}, 128 {"discard-locals", no_argument, NULL, 'X'}, 129 {"extract-dwo", no_argument, NULL, ECP_ONLY_DWO}, 130 {"gap-fill", required_argument, NULL, ECP_GAP_FILL}, 131 {"globalize-symbol", required_argument, NULL, ECP_GLOBALIZE_SYMBOL}, 132 {"globalize-symbols", required_argument, NULL, ECP_GLOBALIZE_SYMBOLS}, 133 {"help", no_argument, NULL, 'h'}, 134 {"input-target", required_argument, NULL, 'I'}, 135 {"keep-symbol", required_argument, NULL, 'K'}, 136 {"keep-symbols", required_argument, NULL, ECP_KEEP_SYMBOLS}, 137 {"keep-global-symbol", required_argument, NULL, 'G'}, 138 {"keep-global-symbols", required_argument, NULL, 139 ECP_KEEP_GLOBAL_SYMBOLS}, 140 {"localize-hidden", no_argument, NULL, ECP_LOCALIZE_HIDDEN}, 141 {"localize-symbol", required_argument, NULL, 'L'}, 142 {"localize-symbols", required_argument, NULL, ECP_LOCALIZE_SYMBOLS}, 143 {"no-adjust-warnings", no_argument, NULL, ECP_NO_CHANGE_WARN}, 144 {"no-change-warnings", no_argument, NULL, ECP_NO_CHANGE_WARN}, 145 {"only-keep-debug", no_argument, NULL, ECP_ONLY_DEBUG}, 146 {"only-section", required_argument, NULL, 'j'}, 147 {"osabi", required_argument, NULL, ECP_SET_OSABI}, 148 {"output-target", required_argument, NULL, 'O'}, 149 {"pad-to", required_argument, NULL, ECP_PAD_TO}, 150 {"preserve-dates", no_argument, NULL, 'p'}, 151 {"prefix-alloc-sections", required_argument, NULL, ECP_PREFIX_ALLOC}, 152 {"prefix-sections", required_argument, NULL, ECP_PREFIX_SEC}, 153 {"prefix-symbols", required_argument, NULL, ECP_PREFIX_SYM}, 154 {"redefine-sym", required_argument, NULL, ECP_REDEF_SYMBOL}, 155 {"redefine-syms", required_argument, NULL, ECP_REDEF_SYMBOLS}, 156 {"remove-section", required_argument, NULL, 'R'}, 157 {"rename-section", required_argument, NULL, ECP_RENAME_SECTION}, 158 {"set-section-flags", required_argument, NULL, ECP_SET_SEC_FLAGS}, 159 {"set-start", required_argument, NULL, ECP_SET_START}, 160 {"srec-forceS3", no_argument, NULL, ECP_SREC_FORCE_S3}, 161 {"srec-len", required_argument, NULL, ECP_SREC_LEN}, 162 {"strip-all", no_argument, NULL, 'S'}, 163 {"strip-debug", no_argument, 0, 'g'}, 164 {"strip-dwo", no_argument, NULL, ECP_STRIP_DWO}, 165 {"strip-symbol", required_argument, NULL, 'N'}, 166 {"strip-symbols", required_argument, NULL, ECP_STRIP_SYMBOLS}, 167 {"strip-unneeded", no_argument, NULL, ECP_STRIP_UNNEEDED}, 168 {"version", no_argument, NULL, 'V'}, 169 {"weaken", no_argument, NULL, ECP_WEAKEN_ALL}, 170 {"weaken-symbol", required_argument, NULL, 'W'}, 171 {"weaken-symbols", required_argument, NULL, ECP_WEAKEN_SYMBOLS}, 172 {"wildcard", no_argument, NULL, 'w'}, 173 {NULL, 0, NULL, 0} 174 }; 175 176 static struct { 177 const char *name; 178 int value; 179 } sec_flags[] = { 180 {"alloc", SF_ALLOC}, 181 {"load", SF_LOAD}, 182 {"noload", SF_NOLOAD}, 183 {"readonly", SF_READONLY}, 184 {"debug", SF_DEBUG}, 185 {"code", SF_CODE}, 186 {"data", SF_DATA}, 187 {"rom", SF_ROM}, 188 {"share", SF_SHARED}, 189 {"contents", SF_CONTENTS}, 190 {NULL, 0} 191 }; 192 193 static struct { 194 const char *name; 195 int abi; 196 } osabis[] = { 197 {"sysv", ELFOSABI_SYSV}, 198 {"hpus", ELFOSABI_HPUX}, 199 {"netbsd", ELFOSABI_NETBSD}, 200 {"linux", ELFOSABI_LINUX}, 201 {"hurd", ELFOSABI_HURD}, 202 {"86open", ELFOSABI_86OPEN}, 203 {"solaris", ELFOSABI_SOLARIS}, 204 {"aix", ELFOSABI_AIX}, 205 {"irix", ELFOSABI_IRIX}, 206 {"freebsd", ELFOSABI_FREEBSD}, 207 {"tru64", ELFOSABI_TRU64}, 208 {"modesto", ELFOSABI_MODESTO}, 209 {"openbsd", ELFOSABI_OPENBSD}, 210 {"openvms", ELFOSABI_OPENVMS}, 211 {"nsk", ELFOSABI_NSK}, 212 {"cloudabi", ELFOSABI_CLOUDABI}, 213 {"arm", ELFOSABI_ARM}, 214 {"standalone", ELFOSABI_STANDALONE}, 215 {NULL, 0} 216 }; 217 218 static int copy_from_tempfile(const char *src, const char *dst, 219 int infd, int *outfd, int in_place); 220 static void create_file(struct elfcopy *ecp, const char *src, 221 const char *dst); 222 static void elfcopy_main(struct elfcopy *ecp, int argc, char **argv); 223 static void elfcopy_usage(void); 224 static void mcs_main(struct elfcopy *ecp, int argc, char **argv); 225 static void mcs_usage(void); 226 static void parse_sec_address_op(struct elfcopy *ecp, int optnum, 227 const char *optname, char *s); 228 static void parse_sec_flags(struct sec_action *sac, char *s); 229 static void parse_symlist_file(struct elfcopy *ecp, const char *fn, 230 unsigned int op); 231 static void print_version(void); 232 static void set_input_target(struct elfcopy *ecp, const char *target_name); 233 static void set_osabi(struct elfcopy *ecp, const char *abi); 234 static void set_output_target(struct elfcopy *ecp, const char *target_name); 235 static void strip_main(struct elfcopy *ecp, int argc, char **argv); 236 static void strip_usage(void); 237 238 /* 239 * An ELF object usually has a structure described by the 240 * diagram below. 241 * _____________ 242 * | | 243 * | NULL | <- always a SHT_NULL section 244 * |_____________| 245 * | | 246 * | .interp | 247 * |_____________| 248 * | | 249 * | ... | 250 * |_____________| 251 * | | 252 * | .text | 253 * |_____________| 254 * | | 255 * | ... | 256 * |_____________| 257 * | | 258 * | .comment | <- above(include) this: normal sections 259 * |_____________| 260 * | | 261 * | add sections| <- unloadable sections added by --add-section 262 * |_____________| 263 * | | 264 * | .shstrtab | <- section name string table 265 * |_____________| 266 * | | 267 * | shdrs | <- section header table 268 * |_____________| 269 * | | 270 * | .symtab | <- symbol table, if any 271 * |_____________| 272 * | | 273 * | .strtab | <- symbol name string table, if any 274 * |_____________| 275 * | | 276 * | .rel.text | <- relocation info for .o files. 277 * |_____________| 278 */ 279 void 280 create_elf(struct elfcopy *ecp) 281 { 282 struct section *shtab; 283 GElf_Ehdr ieh; 284 GElf_Ehdr oeh; 285 size_t ishnum; 286 287 ecp->flags |= SYMTAB_INTACT; 288 ecp->flags &= ~SYMTAB_EXIST; 289 290 /* Create EHDR. */ 291 if (gelf_getehdr(ecp->ein, &ieh) == NULL) 292 errx(EXIT_FAILURE, "gelf_getehdr() failed: %s", 293 elf_errmsg(-1)); 294 if ((ecp->iec = gelf_getclass(ecp->ein)) == ELFCLASSNONE) 295 errx(EXIT_FAILURE, "getclass() failed: %s", 296 elf_errmsg(-1)); 297 298 if (ecp->oec == ELFCLASSNONE) 299 ecp->oec = ecp->iec; 300 if (ecp->oed == ELFDATANONE) 301 ecp->oed = ieh.e_ident[EI_DATA]; 302 303 if (gelf_newehdr(ecp->eout, ecp->oec) == NULL) 304 errx(EXIT_FAILURE, "gelf_newehdr failed: %s", 305 elf_errmsg(-1)); 306 if (gelf_getehdr(ecp->eout, &oeh) == NULL) 307 errx(EXIT_FAILURE, "gelf_getehdr() failed: %s", 308 elf_errmsg(-1)); 309 310 memcpy(oeh.e_ident, ieh.e_ident, sizeof(ieh.e_ident)); 311 oeh.e_ident[EI_CLASS] = ecp->oec; 312 oeh.e_ident[EI_DATA] = ecp->oed; 313 if (ecp->abi != -1) 314 oeh.e_ident[EI_OSABI] = ecp->abi; 315 oeh.e_flags = ieh.e_flags; 316 oeh.e_machine = ieh.e_machine; 317 oeh.e_type = ieh.e_type; 318 oeh.e_entry = ieh.e_entry; 319 oeh.e_version = ieh.e_version; 320 321 ecp->flags &= ~(EXECUTABLE | DYNAMIC | RELOCATABLE); 322 if (ieh.e_type == ET_EXEC) 323 ecp->flags |= EXECUTABLE; 324 else if (ieh.e_type == ET_DYN) 325 ecp->flags |= DYNAMIC; 326 else if (ieh.e_type == ET_REL) 327 ecp->flags |= RELOCATABLE; 328 else 329 errx(EXIT_FAILURE, "unsupported e_type"); 330 331 if (!elf_getshnum(ecp->ein, &ishnum)) 332 errx(EXIT_FAILURE, "elf_getshnum failed: %s", 333 elf_errmsg(-1)); 334 if (ishnum > 0 && (ecp->secndx = calloc(ishnum, 335 sizeof(*ecp->secndx))) == NULL) 336 err(EXIT_FAILURE, "calloc failed"); 337 338 /* Read input object program header. */ 339 setup_phdr(ecp); 340 341 /* 342 * Scan of input sections: we iterate through sections from input 343 * object, skip sections need to be stripped, allot Elf_Scn and 344 * create internal section structure for sections we want. 345 * (i.e., determine output sections) 346 */ 347 create_scn(ecp); 348 349 /* Apply section address changes, if any. */ 350 adjust_addr(ecp); 351 352 /* 353 * Determine if the symbol table needs to be changed based on 354 * command line options. 355 */ 356 if (ecp->strip == STRIP_DEBUG || 357 ecp->strip == STRIP_UNNEEDED || 358 ecp->flags & WEAKEN_ALL || 359 ecp->flags & LOCALIZE_HIDDEN || 360 ecp->flags & DISCARD_LOCAL || 361 ecp->flags & DISCARD_LLABEL || 362 ecp->prefix_sym != NULL || 363 !STAILQ_EMPTY(&ecp->v_symop)) 364 ecp->flags &= ~SYMTAB_INTACT; 365 366 /* 367 * Create symbol table. Symbols are filtered or stripped according to 368 * command line args specified by user, and later updated for the new 369 * layout of sections in the output object. 370 */ 371 if ((ecp->flags & SYMTAB_EXIST) != 0) 372 create_symtab(ecp); 373 374 /* 375 * Write the underlying ehdr. Note that it should be called 376 * before elf_setshstrndx() since it will overwrite e->e_shstrndx. 377 */ 378 if (gelf_update_ehdr(ecp->eout, &oeh) == 0) 379 errx(EXIT_FAILURE, "gelf_update_ehdr() failed: %s", 380 elf_errmsg(-1)); 381 382 /* 383 * First processing of output sections: at this stage we copy the 384 * content of each section from input to output object. Section 385 * content will be modified and printed (mcs) if need. Also content of 386 * relocation section probably will be filtered and updated according 387 * to symbol table changes. 388 */ 389 copy_content(ecp); 390 391 /* 392 * Second processing of output sections: Update section headers. 393 * At this stage we set name string index, update st_link and st_info 394 * for output sections. 395 */ 396 update_shdr(ecp, 1); 397 398 /* Renew oeh to get the updated e_shstrndx. */ 399 if (gelf_getehdr(ecp->eout, &oeh) == NULL) 400 errx(EXIT_FAILURE, "gelf_getehdr() failed: %s", 401 elf_errmsg(-1)); 402 403 /* 404 * Insert SHDR table into the internal section list as a "pseudo" 405 * section, so later it will get sorted and resynced just as "normal" 406 * sections. 407 * 408 * Under FreeBSD, Binutils objcopy always put the section header 409 * at the end of all the sections. We want to do the same here. 410 * 411 * However, note that the behaviour is still different with Binutils: 412 * elfcopy checks the FreeBSD OSABI tag to tell whether it needs to 413 * move the section headers, while Binutils is probably configured 414 * this way when it's compiled on FreeBSD. 415 */ 416 if (oeh.e_ident[EI_OSABI] == ELFOSABI_FREEBSD) 417 shtab = insert_shtab(ecp, 1); 418 else 419 shtab = insert_shtab(ecp, 0); 420 421 /* 422 * Resync section offsets in the output object. This is needed 423 * because probably sections are modified or new sections are added, 424 * as a result overlap/gap might appears. 425 */ 426 resync_sections(ecp); 427 428 /* Store SHDR offset in EHDR. */ 429 oeh.e_shoff = shtab->off; 430 431 /* Put program header table immediately after the Elf header. */ 432 if (ecp->ophnum > 0) { 433 oeh.e_phoff = gelf_fsize(ecp->eout, ELF_T_EHDR, 1, EV_CURRENT); 434 if (oeh.e_phoff == 0) 435 errx(EXIT_FAILURE, "gelf_fsize() failed: %s", 436 elf_errmsg(-1)); 437 } 438 439 /* 440 * Update ELF object entry point if requested. 441 */ 442 if (ecp->change_addr != 0) 443 oeh.e_entry += ecp->change_addr; 444 if (ecp->flags & SET_START) 445 oeh.e_entry = ecp->set_start; 446 if (ecp->change_start != 0) 447 oeh.e_entry += ecp->change_start; 448 449 /* 450 * Update ehdr again before we call elf_update(), since we 451 * modified e_shoff and e_phoff. 452 */ 453 if (gelf_update_ehdr(ecp->eout, &oeh) == 0) 454 errx(EXIT_FAILURE, "gelf_update_ehdr() failed: %s", 455 elf_errmsg(-1)); 456 457 if (ecp->ophnum > 0) 458 copy_phdr(ecp); 459 460 /* Write out the output elf object. */ 461 if (elf_update(ecp->eout, ELF_C_WRITE) < 0) 462 errx(EXIT_FAILURE, "elf_update() failed: %s", 463 elf_errmsg(-1)); 464 465 /* Release allocated resource. */ 466 free_elf(ecp); 467 } 468 469 void 470 free_elf(struct elfcopy *ecp) 471 { 472 struct segment *seg, *seg_temp; 473 struct section *sec, *sec_temp; 474 475 /* Free internal segment list. */ 476 if (!STAILQ_EMPTY(&ecp->v_seg)) { 477 STAILQ_FOREACH_SAFE(seg, &ecp->v_seg, seg_list, seg_temp) { 478 STAILQ_REMOVE(&ecp->v_seg, seg, segment, seg_list); 479 free(seg); 480 } 481 } 482 483 /* Free symbol table buffers. */ 484 free_symtab(ecp); 485 486 /* Free section name string table. */ 487 elftc_string_table_destroy(ecp->shstrtab->strtab); 488 489 /* Free internal section list. */ 490 if (!TAILQ_EMPTY(&ecp->v_sec)) { 491 TAILQ_FOREACH_SAFE(sec, &ecp->v_sec, sec_list, sec_temp) { 492 TAILQ_REMOVE(&ecp->v_sec, sec, sec_list); 493 if (sec->buf != NULL) 494 free(sec->buf); 495 if (sec->newname != NULL) 496 free(sec->newname); 497 if (sec->pad != NULL) 498 free(sec->pad); 499 free(sec); 500 } 501 } 502 503 ecp->symtab = NULL; 504 ecp->strtab = NULL; 505 ecp->shstrtab = NULL; 506 507 if (ecp->secndx != NULL) { 508 free(ecp->secndx); 509 ecp->secndx = NULL; 510 } 511 } 512 513 /* 514 * Remove a temporary file, without freeing its filename. 515 * 516 * Safe to pass NULL, will just ignore it. 517 */ 518 int 519 cleanup_tempfile(char *fn) 520 { 521 int errno_save, retval; 522 523 if (fn == NULL) 524 return 0; 525 errno_save = errno; 526 if ((retval = unlink(fn)) < 0) { 527 warn("unlink tempfile %s failed", fn); 528 errno = errno_save; 529 return retval; 530 } 531 return 0; 532 } 533 534 /* Create a temporary file. */ 535 void 536 create_tempfile(const char *src, char **fn, int *fd) 537 { 538 static const char _TEMPDIR[] = "/tmp/"; 539 static const char _TEMPFILE[] = "ecp.XXXXXXXX"; 540 const char *tmpdir; 541 char *tmpf; 542 size_t tlen, slen, plen; 543 544 if (fn == NULL || fd == NULL) 545 return; 546 for (;;) { 547 if (src == NULL) { 548 /* Respect TMPDIR environment variable. */ 549 tmpdir = getenv("TMPDIR"); 550 if (tmpdir == NULL || *tmpdir == '\0') 551 tmpdir = _TEMPDIR; 552 tlen = strlen(tmpdir); 553 slen = tmpdir[tlen - 1] == '/' ? 0 : 1; 554 } else { 555 /* Create temporary file relative to source file. */ 556 if ((tmpdir = strrchr(src, '/')) == NULL) { 557 /* No path, only use a template filename. */ 558 tlen = 0; 559 } else { 560 /* Append the template after the slash. */ 561 tlen = ++tmpdir - src; 562 tmpdir = src; 563 } 564 slen = 0; 565 } 566 plen = strlen(_TEMPFILE) + 1; 567 tmpf = malloc(tlen + slen + plen); 568 if (tmpf == NULL) 569 err(EXIT_FAILURE, "malloc failed"); 570 if (tlen > 0) 571 memcpy(tmpf, tmpdir, tlen); 572 if (slen > 0) 573 tmpf[tlen] = '/'; 574 /* Copy template filename including NUL terminator. */ 575 memcpy(tmpf + tlen + slen, _TEMPFILE, plen); 576 if ((*fd = mkstemp(tmpf)) != -1) 577 break; 578 if (errno != EACCES || src == NULL) 579 err(EXIT_FAILURE, "mkstemp %s failed", tmpf); 580 /* Permission denied, try again using TMPDIR or /tmp. */ 581 free(tmpf); 582 src = NULL; 583 } 584 if (fchmod(*fd, 0644) == -1) 585 err(EXIT_FAILURE, "fchmod %s failed", tmpf); 586 *fn = tmpf; 587 } 588 589 /* 590 * Copy temporary file with path src and file descriptor infd to path dst. 591 * If in_place is set act as if editing the file in place, avoiding rename() 592 * to preserve hard and symbolic links. Output file remains open, with file 593 * descriptor returned in outfd. 594 */ 595 static int 596 copy_from_tempfile(const char *src, const char *dst, int infd, int *outfd, 597 int in_place) 598 { 599 int tmpfd; 600 601 /* 602 * First, check if we can use rename(). 603 */ 604 if (in_place == 0) { 605 if (rename(src, dst) >= 0) { 606 *outfd = infd; 607 return (0); 608 } else if (errno != EXDEV && errno != EACCES) 609 return (-1); 610 611 /* 612 * If the rename() failed due to 'src' and 'dst' residing in 613 * two different file systems, invoke a helper function in 614 * libelftc to do the copy. 615 */ 616 617 if (errno != EACCES && unlink(dst) < 0) 618 return (-1); 619 } 620 621 if ((tmpfd = open(dst, O_CREAT | O_TRUNC | O_WRONLY, 0755)) < 0) 622 return (-1); 623 624 if (elftc_copyfile(infd, tmpfd) < 0) { 625 (void) close(tmpfd); 626 return (-1); 627 } 628 629 /* 630 * Remove the temporary file from the file system 631 * namespace, and close its file descriptor. 632 */ 633 if (unlink(src) < 0) { 634 (void) close(tmpfd); 635 return (-1); 636 } 637 638 (void) close(infd); 639 640 /* 641 * Return the file descriptor for the destination. 642 */ 643 *outfd = tmpfd; 644 645 return (0); 646 } 647 648 static void 649 create_file(struct elfcopy *ecp, const char *src, const char *dst) 650 { 651 struct stat sb; 652 char *tempfile, *elftemp; 653 int efd, ifd, ofd, ofd0, tfd; 654 int in_place; 655 656 tempfile = NULL; 657 658 if (src == NULL) 659 errx(EXIT_FAILURE, "internal: src == NULL"); 660 if ((ifd = open(src, O_RDONLY)) == -1) 661 err(EXIT_FAILURE, "open %s failed", src); 662 663 if (fstat(ifd, &sb) == -1) 664 err(EXIT_FAILURE, "fstat %s failed", src); 665 666 if (dst == NULL) 667 create_tempfile(src, &tempfile, &ofd); 668 else 669 if ((ofd = open(dst, O_RDWR|O_CREAT, 0755)) == -1) 670 err(EXIT_FAILURE, "open %s failed", dst); 671 672 #ifndef LIBELF_AR 673 /* Detect and process ar(1) archive using libarchive. */ 674 if (ac_detect_ar(ifd)) { 675 ac_create_ar(ecp, ifd, ofd); 676 goto copy_done; 677 } 678 #endif 679 680 if (lseek(ifd, 0, SEEK_SET) < 0) { 681 cleanup_tempfile(tempfile); 682 err(EXIT_FAILURE, "lseek failed"); 683 } 684 685 /* 686 * If input object is not ELF file, convert it to an intermediate 687 * ELF object before processing. 688 */ 689 if (ecp->itf != ETF_ELF) { 690 /* 691 * If the output object is not an ELF file, choose an arbitrary 692 * ELF format for the intermediate file. srec, ihex and binary 693 * formats are independent of class, endianness and machine 694 * type so these choices do not affect the output. 695 */ 696 if (ecp->otf != ETF_ELF) { 697 if (ecp->oec == ELFCLASSNONE) 698 ecp->oec = ELFCLASS64; 699 if (ecp->oed == ELFDATANONE) 700 ecp->oed = ELFDATA2LSB; 701 } 702 create_tempfile(src, &elftemp, &efd); 703 if ((ecp->eout = elf_begin(efd, ELF_C_WRITE, NULL)) == NULL) { 704 cleanup_tempfile(elftemp); 705 cleanup_tempfile(tempfile); 706 errx(EXIT_FAILURE, "elf_begin() failed: %s", 707 elf_errmsg(-1)); 708 } 709 elf_flagelf(ecp->eout, ELF_C_SET, ELF_F_LAYOUT); 710 if (ecp->itf == ETF_BINARY) 711 create_elf_from_binary(ecp, ifd, src); 712 else if (ecp->itf == ETF_IHEX) 713 create_elf_from_ihex(ecp, ifd); 714 else if (ecp->itf == ETF_SREC) 715 create_elf_from_srec(ecp, ifd); 716 else { 717 cleanup_tempfile(elftemp); 718 cleanup_tempfile(tempfile); 719 errx(EXIT_FAILURE, "Internal: invalid target flavour"); 720 } 721 elf_end(ecp->eout); 722 723 /* Open intermediate ELF object as new input object. */ 724 close(ifd); 725 if ((ifd = open(elftemp, O_RDONLY)) == -1) { 726 cleanup_tempfile(elftemp); 727 cleanup_tempfile(tempfile); 728 err(EXIT_FAILURE, "open %s failed", src); 729 } 730 close(efd); 731 if (cleanup_tempfile(elftemp) < 0) { 732 cleanup_tempfile(tempfile); 733 err(EXIT_FAILURE, "unlink %s failed", elftemp); 734 } 735 free(elftemp); 736 elftemp = NULL; 737 } 738 739 if ((ecp->ein = elf_begin(ifd, ELF_C_READ, NULL)) == NULL) { 740 cleanup_tempfile(tempfile); 741 if (fstat(ifd, &sb) == 0 && sb.st_size == 0) 742 errx(EXIT_FAILURE, "file format not recognized"); 743 errx(EXIT_FAILURE, "elf_begin() failed: %s", 744 elf_errmsg(-1)); 745 } 746 747 switch (elf_kind(ecp->ein)) { 748 case ELF_K_NONE: 749 cleanup_tempfile(tempfile); 750 errx(EXIT_FAILURE, "file format not recognized"); 751 case ELF_K_ELF: 752 if ((ecp->eout = elf_begin(ofd, ELF_C_WRITE, NULL)) == NULL) { 753 cleanup_tempfile(tempfile); 754 errx(EXIT_FAILURE, "elf_begin() failed: %s", 755 elf_errmsg(-1)); 756 } 757 758 /* elfcopy(1) manage ELF layout by itself. */ 759 elf_flagelf(ecp->eout, ELF_C_SET, ELF_F_LAYOUT); 760 761 /* 762 * Create output ELF object. 763 */ 764 create_elf(ecp); 765 elf_end(ecp->eout); 766 767 /* 768 * Convert the output ELF object to binary/srec/ihex if need. 769 */ 770 if (ecp->otf != ETF_ELF) { 771 /* 772 * Create (another) tempfile for binary/srec/ihex 773 * output object. 774 */ 775 if (cleanup_tempfile(tempfile) < 0) 776 errx(EXIT_FAILURE, "unlink %s failed", 777 tempfile); 778 free(tempfile); 779 create_tempfile(src, &tempfile, &ofd0); 780 781 782 /* 783 * Rewind the file descriptor being processed. 784 */ 785 if (lseek(ofd, 0, SEEK_SET) < 0) { 786 cleanup_tempfile(tempfile); 787 err(EXIT_FAILURE, 788 "lseek failed for the output object"); 789 } 790 791 /* 792 * Call flavour-specific conversion routine. 793 */ 794 switch (ecp->otf) { 795 case ETF_BINARY: 796 create_binary(ofd, ofd0); 797 break; 798 case ETF_IHEX: 799 create_ihex(ofd, ofd0); 800 break; 801 case ETF_SREC: 802 create_srec(ecp, ofd, ofd0, 803 dst != NULL ? dst : src); 804 break; 805 case ETF_PE: 806 case ETF_EFI: 807 #if WITH_PE 808 create_pe(ecp, ofd, ofd0); 809 #else 810 cleanup_tempfile(tempfile); 811 errx(EXIT_FAILURE, "PE/EFI support not enabled" 812 " at compile time"); 813 #endif 814 break; 815 default: 816 cleanup_tempfile(tempfile); 817 errx(EXIT_FAILURE, "Internal: unsupported" 818 " output flavour %d", ecp->oec); 819 } 820 821 close(ofd); 822 ofd = ofd0; 823 } 824 825 break; 826 827 case ELF_K_AR: 828 /* XXX: Not yet supported. */ 829 break; 830 default: 831 cleanup_tempfile(tempfile); 832 errx(EXIT_FAILURE, "file format not supported"); 833 } 834 835 elf_end(ecp->ein); 836 837 #ifndef LIBELF_AR 838 copy_done: 839 #endif 840 841 if (tempfile != NULL) { 842 in_place = 0; 843 if (dst == NULL) { 844 dst = src; 845 if (lstat(dst, &sb) != -1 && 846 (sb.st_nlink > 1 || S_ISLNK(sb.st_mode))) 847 in_place = 1; 848 } 849 850 if (copy_from_tempfile(tempfile, dst, ofd, 851 &tfd, in_place) < 0) { 852 cleanup_tempfile(tempfile); 853 err(EXIT_FAILURE, "creation of %s failed", dst); 854 } 855 856 /* 'tempfile' has been removed by copy_from_tempfile(). */ 857 free(tempfile); 858 tempfile = NULL; 859 860 ofd = tfd; 861 } 862 863 if (strcmp(dst, "/dev/null") && fchmod(ofd, sb.st_mode) == -1) 864 err(EXIT_FAILURE, "fchmod %s failed", dst); 865 866 if ((ecp->flags & PRESERVE_DATE) && 867 elftc_set_timestamps(dst, &sb) < 0) 868 err(EXIT_FAILURE, "setting timestamps failed"); 869 870 close(ifd); 871 close(ofd); 872 } 873 874 static void 875 elfcopy_main(struct elfcopy *ecp, int argc, char **argv) 876 { 877 struct sec_action *sac; 878 const char *infile, *outfile; 879 char *fn, *s; 880 int opt; 881 882 while ((opt = getopt_long(argc, argv, "dB:gG:I:j:K:L:N:O:pR:s:SwW:xXV", 883 elfcopy_longopts, NULL)) != -1) { 884 switch(opt) { 885 case 'B': 886 /* ignored */ 887 break; 888 case 'R': 889 sac = lookup_sec_act(ecp, optarg, 1); 890 if (sac->copy != 0) 891 errx(EXIT_FAILURE, 892 "both copy and remove specified"); 893 sac->remove = 1; 894 ecp->flags |= SEC_REMOVE; 895 break; 896 case 'S': 897 ecp->strip = STRIP_ALL; 898 break; 899 case 'g': 900 ecp->strip = STRIP_DEBUG; 901 break; 902 case 'G': 903 ecp->flags |= KEEP_GLOBAL; 904 add_to_symop_list(ecp, optarg, NULL, SYMOP_KEEPG); 905 break; 906 case 'I': 907 case 's': 908 set_input_target(ecp, optarg); 909 break; 910 case 'j': 911 sac = lookup_sec_act(ecp, optarg, 1); 912 if (sac->remove != 0) 913 errx(EXIT_FAILURE, 914 "both copy and remove specified"); 915 sac->copy = 1; 916 ecp->flags |= SEC_COPY; 917 break; 918 case 'K': 919 add_to_symop_list(ecp, optarg, NULL, SYMOP_KEEP); 920 break; 921 case 'L': 922 add_to_symop_list(ecp, optarg, NULL, SYMOP_LOCALIZE); 923 break; 924 case 'N': 925 add_to_symop_list(ecp, optarg, NULL, SYMOP_STRIP); 926 break; 927 case 'O': 928 set_output_target(ecp, optarg); 929 break; 930 case 'p': 931 ecp->flags |= PRESERVE_DATE; 932 break; 933 case 'V': 934 print_version(); 935 break; 936 case 'w': 937 ecp->flags |= WILDCARD; 938 break; 939 case 'W': 940 add_to_symop_list(ecp, optarg, NULL, SYMOP_WEAKEN); 941 break; 942 case 'x': 943 ecp->flags |= DISCARD_LOCAL; 944 break; 945 case 'X': 946 ecp->flags |= DISCARD_LLABEL; 947 break; 948 case ECP_ADD_GNU_DEBUGLINK: 949 ecp->debuglink = optarg; 950 break; 951 case ECP_ADD_SECTION: 952 add_section(ecp, optarg); 953 break; 954 case ECP_CHANGE_ADDR: 955 ecp->change_addr = (int64_t) strtoll(optarg, NULL, 0); 956 break; 957 case ECP_CHANGE_SEC_ADDR: 958 parse_sec_address_op(ecp, opt, "--change-section-addr", 959 optarg); 960 break; 961 case ECP_CHANGE_SEC_LMA: 962 parse_sec_address_op(ecp, opt, "--change-section-lma", 963 optarg); 964 break; 965 case ECP_CHANGE_SEC_VMA: 966 parse_sec_address_op(ecp, opt, "--change-section-vma", 967 optarg); 968 break; 969 case ECP_CHANGE_START: 970 ecp->change_start = (int64_t) strtoll(optarg, NULL, 0); 971 break; 972 case ECP_CHANGE_WARN: 973 /* default */ 974 break; 975 case ECP_GAP_FILL: 976 ecp->fill = (uint8_t) strtoul(optarg, NULL, 0); 977 ecp->flags |= GAP_FILL; 978 break; 979 case ECP_GLOBALIZE_SYMBOL: 980 add_to_symop_list(ecp, optarg, NULL, SYMOP_GLOBALIZE); 981 break; 982 case ECP_GLOBALIZE_SYMBOLS: 983 parse_symlist_file(ecp, optarg, SYMOP_GLOBALIZE); 984 break; 985 case ECP_KEEP_SYMBOLS: 986 parse_symlist_file(ecp, optarg, SYMOP_KEEP); 987 break; 988 case ECP_KEEP_GLOBAL_SYMBOLS: 989 parse_symlist_file(ecp, optarg, SYMOP_KEEPG); 990 break; 991 case ECP_LOCALIZE_HIDDEN: 992 ecp->flags |= LOCALIZE_HIDDEN; 993 break; 994 case ECP_LOCALIZE_SYMBOLS: 995 parse_symlist_file(ecp, optarg, SYMOP_LOCALIZE); 996 break; 997 case ECP_NO_CHANGE_WARN: 998 ecp->flags |= NO_CHANGE_WARN; 999 break; 1000 case ECP_ONLY_DEBUG: 1001 ecp->strip = STRIP_NONDEBUG; 1002 break; 1003 case ECP_ONLY_DWO: 1004 ecp->strip = STRIP_NONDWO; 1005 break; 1006 case ECP_PAD_TO: 1007 ecp->pad_to = (uint64_t) strtoull(optarg, NULL, 0); 1008 break; 1009 case ECP_PREFIX_ALLOC: 1010 ecp->prefix_alloc = optarg; 1011 break; 1012 case ECP_PREFIX_SEC: 1013 ecp->prefix_sec = optarg; 1014 break; 1015 case ECP_PREFIX_SYM: 1016 ecp->prefix_sym = optarg; 1017 break; 1018 case ECP_REDEF_SYMBOL: 1019 if ((s = strchr(optarg, '=')) == NULL) 1020 errx(EXIT_FAILURE, 1021 "illegal format for --redefine-sym"); 1022 *s++ = '\0'; 1023 add_to_symop_list(ecp, optarg, s, SYMOP_REDEF); 1024 break; 1025 case ECP_REDEF_SYMBOLS: 1026 parse_symlist_file(ecp, optarg, SYMOP_REDEF); 1027 break; 1028 case ECP_RENAME_SECTION: 1029 if ((fn = strchr(optarg, '=')) == NULL) 1030 errx(EXIT_FAILURE, 1031 "illegal format for --rename-section"); 1032 *fn++ = '\0'; 1033 1034 /* Check for optional flags. */ 1035 if ((s = strchr(fn, ',')) != NULL) 1036 *s++ = '\0'; 1037 1038 sac = lookup_sec_act(ecp, optarg, 1); 1039 sac->rename = 1; 1040 sac->newname = fn; 1041 if (s != NULL) 1042 parse_sec_flags(sac, s); 1043 break; 1044 case ECP_SET_OSABI: 1045 set_osabi(ecp, optarg); 1046 break; 1047 case ECP_SET_SEC_FLAGS: 1048 if ((s = strchr(optarg, '=')) == NULL) 1049 errx(EXIT_FAILURE, 1050 "illegal format for --set-section-flags"); 1051 *s++ = '\0'; 1052 sac = lookup_sec_act(ecp, optarg, 1); 1053 parse_sec_flags(sac, s); 1054 break; 1055 case ECP_SET_START: 1056 ecp->flags |= SET_START; 1057 ecp->set_start = (uint64_t) strtoull(optarg, NULL, 0); 1058 break; 1059 case ECP_SREC_FORCE_S3: 1060 ecp->flags |= SREC_FORCE_S3; 1061 break; 1062 case ECP_SREC_LEN: 1063 ecp->flags |= SREC_FORCE_LEN; 1064 ecp->srec_len = strtoul(optarg, NULL, 0); 1065 break; 1066 case ECP_STRIP_DWO: 1067 ecp->strip = STRIP_DWO; 1068 break; 1069 case ECP_STRIP_SYMBOLS: 1070 parse_symlist_file(ecp, optarg, SYMOP_STRIP); 1071 break; 1072 case ECP_STRIP_UNNEEDED: 1073 ecp->strip = STRIP_UNNEEDED; 1074 break; 1075 case ECP_WEAKEN_ALL: 1076 ecp->flags |= WEAKEN_ALL; 1077 break; 1078 case ECP_WEAKEN_SYMBOLS: 1079 parse_symlist_file(ecp, optarg, SYMOP_WEAKEN); 1080 break; 1081 default: 1082 elfcopy_usage(); 1083 } 1084 } 1085 1086 argc -= optind; 1087 argv += optind; 1088 1089 if (argc == 0 || argc > 2) 1090 elfcopy_usage(); 1091 1092 infile = argv[0]; 1093 outfile = NULL; 1094 if (argc > 1) 1095 outfile = argv[1]; 1096 1097 create_file(ecp, infile, outfile); 1098 } 1099 1100 static void 1101 mcs_main(struct elfcopy *ecp, int argc, char **argv) 1102 { 1103 struct sec_action *sac; 1104 const char *string; 1105 int append, delete, compress, name, print; 1106 int opt, i; 1107 1108 append = delete = compress = name = print = 0; 1109 string = NULL; 1110 while ((opt = getopt_long(argc, argv, "a:cdhn:pV", mcs_longopts, 1111 NULL)) != -1) { 1112 switch(opt) { 1113 case 'a': 1114 append = 1; 1115 string = optarg; /* XXX multiple -a not supported */ 1116 break; 1117 case 'c': 1118 compress = 1; 1119 break; 1120 case 'd': 1121 delete = 1; 1122 break; 1123 case 'n': 1124 name = 1; 1125 (void)lookup_sec_act(ecp, optarg, 1); 1126 break; 1127 case 'p': 1128 print = 1; 1129 break; 1130 case 'V': 1131 print_version(); 1132 break; 1133 case 'h': 1134 default: 1135 mcs_usage(); 1136 } 1137 } 1138 1139 argc -= optind; 1140 argv += optind; 1141 1142 if (argc == 0) 1143 mcs_usage(); 1144 1145 /* Must specify one operation at least. */ 1146 if (!append && !compress && !delete && !print) 1147 mcs_usage(); 1148 1149 /* 1150 * If we are going to delete, ignore other operations. This is 1151 * different from the Solaris implementation, which can print 1152 * and delete a section at the same time, for example. Also, this 1153 * implementation do not respect the order between operations that 1154 * user specified, i.e., "mcs -pc a.out" equals to "mcs -cp a.out". 1155 */ 1156 if (delete) { 1157 append = compress = print = 0; 1158 ecp->flags |= SEC_REMOVE; 1159 } 1160 if (append) 1161 ecp->flags |= SEC_APPEND; 1162 if (compress) 1163 ecp->flags |= SEC_COMPRESS; 1164 if (print) 1165 ecp->flags |= SEC_PRINT; 1166 1167 /* .comment is the default section to operate on. */ 1168 if (!name) 1169 (void)lookup_sec_act(ecp, ".comment", 1); 1170 1171 STAILQ_FOREACH(sac, &ecp->v_sac, sac_list) { 1172 sac->append = append; 1173 sac->compress = compress; 1174 sac->print = print; 1175 sac->remove = delete; 1176 sac->string = string; 1177 } 1178 1179 for (i = 0; i < argc; i++) { 1180 /* If only -p is specified, output to /dev/null */ 1181 if (print && !append && !compress && !delete) 1182 create_file(ecp, argv[i], "/dev/null"); 1183 else 1184 create_file(ecp, argv[i], NULL); 1185 } 1186 } 1187 1188 static void 1189 strip_main(struct elfcopy *ecp, int argc, char **argv) 1190 { 1191 struct sec_action *sac; 1192 const char *outfile; 1193 int opt; 1194 int i; 1195 1196 outfile = NULL; 1197 while ((opt = getopt_long(argc, argv, "hI:K:N:o:O:pR:sSdgVxXw", 1198 strip_longopts, NULL)) != -1) { 1199 switch(opt) { 1200 case 'R': 1201 sac = lookup_sec_act(ecp, optarg, 1); 1202 sac->remove = 1; 1203 ecp->flags |= SEC_REMOVE; 1204 break; 1205 case 's': 1206 ecp->strip = STRIP_ALL; 1207 break; 1208 case 'S': 1209 case 'g': 1210 case 'd': 1211 ecp->strip = STRIP_DEBUG; 1212 break; 1213 case 'I': 1214 /* ignored */ 1215 break; 1216 case 'K': 1217 add_to_symop_list(ecp, optarg, NULL, SYMOP_KEEP); 1218 break; 1219 case 'N': 1220 add_to_symop_list(ecp, optarg, NULL, SYMOP_STRIP); 1221 break; 1222 case 'o': 1223 outfile = optarg; 1224 break; 1225 case 'O': 1226 set_output_target(ecp, optarg); 1227 break; 1228 case 'p': 1229 ecp->flags |= PRESERVE_DATE; 1230 break; 1231 case 'V': 1232 print_version(); 1233 break; 1234 case 'w': 1235 ecp->flags |= WILDCARD; 1236 break; 1237 case 'x': 1238 ecp->flags |= DISCARD_LOCAL; 1239 break; 1240 case 'X': 1241 ecp->flags |= DISCARD_LLABEL; 1242 break; 1243 case ECP_ONLY_DEBUG: 1244 ecp->strip = STRIP_NONDEBUG; 1245 break; 1246 case ECP_STRIP_UNNEEDED: 1247 ecp->strip = STRIP_UNNEEDED; 1248 break; 1249 case 'h': 1250 default: 1251 strip_usage(); 1252 } 1253 } 1254 1255 argc -= optind; 1256 argv += optind; 1257 1258 if (ecp->strip == 0 && 1259 ((ecp->flags & DISCARD_LOCAL) == 0) && 1260 ((ecp->flags & DISCARD_LLABEL) == 0) && 1261 lookup_symop_list(ecp, NULL, SYMOP_STRIP) == NULL) 1262 ecp->strip = STRIP_ALL; 1263 if (argc == 0) 1264 strip_usage(); 1265 /* 1266 * Only accept a single input file if an output file had been 1267 * specified. 1268 */ 1269 if (outfile != NULL && argc != 1) 1270 strip_usage(); 1271 1272 for (i = 0; i < argc; i++) 1273 create_file(ecp, argv[i], outfile); 1274 } 1275 1276 static void 1277 parse_sec_flags(struct sec_action *sac, char *s) 1278 { 1279 const char *flag; 1280 int found, i; 1281 1282 for (flag = strtok(s, ","); flag; flag = strtok(NULL, ",")) { 1283 found = 0; 1284 for (i = 0; sec_flags[i].name != NULL; i++) 1285 if (strcasecmp(sec_flags[i].name, flag) == 0) { 1286 sac->flags |= sec_flags[i].value; 1287 found = 1; 1288 break; 1289 } 1290 if (!found) 1291 errx(EXIT_FAILURE, "unrecognized section flag %s", 1292 flag); 1293 } 1294 } 1295 1296 static void 1297 parse_sec_address_op(struct elfcopy *ecp, int optnum, const char *optname, 1298 char *s) 1299 { 1300 struct sec_action *sac; 1301 const char *name; 1302 char *v; 1303 char op; 1304 1305 name = v = s; 1306 do { 1307 v++; 1308 } while (*v != '\0' && *v != '=' && *v != '+' && *v != '-'); 1309 if (*v == '\0' || *(v + 1) == '\0') 1310 errx(EXIT_FAILURE, "invalid format for %s", optname); 1311 op = *v; 1312 *v++ = '\0'; 1313 sac = lookup_sec_act(ecp, name, 1); 1314 switch (op) { 1315 case '=': 1316 if (optnum == ECP_CHANGE_SEC_LMA || 1317 optnum == ECP_CHANGE_SEC_ADDR) { 1318 sac->setlma = 1; 1319 sac->lma = (uint64_t) strtoull(v, NULL, 0); 1320 } 1321 if (optnum == ECP_CHANGE_SEC_VMA || 1322 optnum == ECP_CHANGE_SEC_ADDR) { 1323 sac->setvma = 1; 1324 sac->vma = (uint64_t) strtoull(v, NULL, 0); 1325 } 1326 break; 1327 case '+': 1328 if (optnum == ECP_CHANGE_SEC_LMA || 1329 optnum == ECP_CHANGE_SEC_ADDR) 1330 sac->lma_adjust = (int64_t) strtoll(v, NULL, 0); 1331 if (optnum == ECP_CHANGE_SEC_VMA || 1332 optnum == ECP_CHANGE_SEC_ADDR) 1333 sac->vma_adjust = (int64_t) strtoll(v, NULL, 0); 1334 break; 1335 case '-': 1336 if (optnum == ECP_CHANGE_SEC_LMA || 1337 optnum == ECP_CHANGE_SEC_ADDR) 1338 sac->lma_adjust = (int64_t) -strtoll(v, NULL, 0); 1339 if (optnum == ECP_CHANGE_SEC_VMA || 1340 optnum == ECP_CHANGE_SEC_ADDR) 1341 sac->vma_adjust = (int64_t) -strtoll(v, NULL, 0); 1342 break; 1343 default: 1344 break; 1345 } 1346 } 1347 1348 static void 1349 parse_symlist_file(struct elfcopy *ecp, const char *fn, unsigned int op) 1350 { 1351 struct symfile *sf; 1352 struct stat sb; 1353 FILE *fp; 1354 char *data, *p, *line, *end, *e, *n; 1355 1356 if (stat(fn, &sb) == -1) 1357 err(EXIT_FAILURE, "stat %s failed", fn); 1358 1359 /* Check if we already read and processed this file. */ 1360 STAILQ_FOREACH(sf, &ecp->v_symfile, symfile_list) { 1361 if (sf->dev == sb.st_dev && sf->ino == sb.st_ino) 1362 goto process_symfile; 1363 } 1364 1365 if ((fp = fopen(fn, "r")) == NULL) 1366 err(EXIT_FAILURE, "can not open %s", fn); 1367 if ((data = malloc(sb.st_size + 1)) == NULL) 1368 err(EXIT_FAILURE, "malloc failed"); 1369 if (sb.st_size > 0) 1370 if (fread(data, sb.st_size, 1, fp) != 1) 1371 err(EXIT_FAILURE, "fread failed"); 1372 fclose(fp); 1373 data[sb.st_size] = '\0'; 1374 1375 if ((sf = calloc(1, sizeof(*sf))) == NULL) 1376 err(EXIT_FAILURE, "malloc failed"); 1377 sf->dev = sb.st_dev; 1378 sf->ino = sb.st_ino; 1379 sf->size = sb.st_size + 1; 1380 sf->data = data; 1381 1382 process_symfile: 1383 1384 /* 1385 * Basically what we do here is to convert EOL to '\0', and remove 1386 * leading and trailing whitespaces for each line. 1387 */ 1388 1389 end = sf->data + sf->size; 1390 line = NULL; 1391 for(p = sf->data; p < end; p++) { 1392 if ((*p == '\t' || *p == ' ') && line == NULL) 1393 continue; 1394 if (*p == '\r' || *p == '\n' || *p == '\0') { 1395 *p = '\0'; 1396 if (line == NULL) 1397 continue; 1398 1399 /* Skip comment. */ 1400 if (*line == '#') { 1401 line = NULL; 1402 continue; 1403 } 1404 1405 e = p - 1; 1406 while(e != line && (*e == '\t' || *e == ' ')) 1407 *e-- = '\0'; 1408 if (op != SYMOP_REDEF) 1409 add_to_symop_list(ecp, line, NULL, op); 1410 else { 1411 if (strlen(line) < 3) 1412 errx(EXIT_FAILURE, 1413 "illegal format for" 1414 " --redefine-sym"); 1415 for(n = line + 1; n < e; n++) { 1416 if (*n == ' ' || *n == '\t') { 1417 while(*n == ' ' || *n == '\t') 1418 *n++ = '\0'; 1419 break; 1420 } 1421 } 1422 if (n >= e) 1423 errx(EXIT_FAILURE, 1424 "illegal format for" 1425 " --redefine-sym"); 1426 add_to_symop_list(ecp, line, n, op); 1427 } 1428 line = NULL; 1429 continue; 1430 } 1431 1432 if (line == NULL) 1433 line = p; 1434 } 1435 } 1436 1437 static void 1438 set_input_target(struct elfcopy *ecp, const char *target_name) 1439 { 1440 Elftc_Bfd_Target *tgt; 1441 1442 if ((tgt = elftc_bfd_find_target(target_name)) == NULL) 1443 errx(EXIT_FAILURE, "%s: invalid target name", target_name); 1444 ecp->itf = elftc_bfd_target_flavor(tgt); 1445 } 1446 1447 static void 1448 set_output_target(struct elfcopy *ecp, const char *target_name) 1449 { 1450 Elftc_Bfd_Target *tgt; 1451 1452 if ((tgt = elftc_bfd_find_target(target_name)) == NULL) 1453 errx(EXIT_FAILURE, "%s: invalid target name", target_name); 1454 ecp->otf = elftc_bfd_target_flavor(tgt); 1455 if (ecp->otf == ETF_ELF) { 1456 ecp->oec = elftc_bfd_target_class(tgt); 1457 ecp->oed = elftc_bfd_target_byteorder(tgt); 1458 ecp->oem = elftc_bfd_target_machine(tgt); 1459 ecp->abi = elftc_bfd_target_osabi(tgt); 1460 } 1461 if (ecp->otf == ETF_EFI || ecp->otf == ETF_PE) 1462 ecp->oem = elftc_bfd_target_machine(tgt); 1463 1464 ecp->otgt = target_name; 1465 } 1466 1467 static void 1468 set_osabi(struct elfcopy *ecp, const char *abi) 1469 { 1470 int i, found; 1471 1472 found = 0; 1473 for (i = 0; osabis[i].name != NULL; i++) 1474 if (strcasecmp(osabis[i].name, abi) == 0) { 1475 ecp->abi = osabis[i].abi; 1476 found = 1; 1477 break; 1478 } 1479 if (!found) 1480 errx(EXIT_FAILURE, "unrecognized OSABI %s", abi); 1481 } 1482 1483 #define ELFCOPY_USAGE_MESSAGE "\ 1484 Usage: %s [options] infile [outfile]\n\ 1485 Transform object files.\n\n\ 1486 Options:\n\ 1487 -d | -g | --strip-debug Remove debugging information from the output.\n\ 1488 -j SECTION | --only-section=SECTION\n\ 1489 Copy only the named section to the output.\n\ 1490 -p | --preserve-dates Preserve access and modification times.\n\ 1491 -w | --wildcard Use shell-style patterns to name symbols.\n\ 1492 -x | --discard-all Do not copy non-globals to the output.\n\ 1493 -I FORMAT | --input-target=FORMAT\n\ 1494 Specify object format for the input file.\n\ 1495 -K SYM | --keep-symbol=SYM Copy symbol SYM to the output.\n\ 1496 -L SYM | --localize-symbol=SYM\n\ 1497 Make symbol SYM local to the output file.\n\ 1498 -N SYM | --strip-symbol=SYM Do not copy symbol SYM to the output.\n\ 1499 -O FORMAT | --output-target=FORMAT\n\ 1500 Specify object format for the output file.\n\ 1501 FORMAT should be a target name understood by\n\ 1502 elftc_bfd_find_target(3).\n\ 1503 -R NAME | --remove-section=NAME\n\ 1504 Remove the named section.\n\ 1505 -S | --strip-all Remove all symbol and relocation information\n\ 1506 from the output.\n\ 1507 -V | --version Print a version identifier and exit.\n\ 1508 -W SYM | --weaken-symbol=SYM Mark symbol SYM as weak in the output.\n\ 1509 -X | --discard-locals Do not copy compiler generated symbols to\n\ 1510 the output.\n\ 1511 --add-section NAME=FILE Add the contents of FILE to the ELF object as\n\ 1512 a new section named NAME.\n\ 1513 --adjust-section-vma SECTION{=,+,-}VAL | \\\n\ 1514 --change-section-address SECTION{=,+,-}VAL\n\ 1515 Set or adjust the VMA and the LMA of the\n\ 1516 named section by VAL.\n\ 1517 --adjust-start=INCR | --change-start=INCR\n\ 1518 Add INCR to the start address for the ELF\n\ 1519 object.\n\ 1520 --adjust-vma=INCR | --change-addresses=INCR\n\ 1521 Increase the VMA and LMA of all sections by\n\ 1522 INCR.\n\ 1523 --adjust-warning | --change-warnings\n\ 1524 Issue warnings for non-existent sections.\n\ 1525 --change-section-lma SECTION{=,+,-}VAL\n\ 1526 Set or adjust the LMA address of the named\n\ 1527 section by VAL.\n\ 1528 --change-section-vma SECTION{=,+,-}VAL\n\ 1529 Set or adjust the VMA address of the named\n\ 1530 section by VAL.\n\ 1531 --gap-fill=VAL Fill the gaps between sections with bytes\n\ 1532 of value VAL.\n\ 1533 --localize-hidden Make all hidden symbols local to the output\n\ 1534 file.\n\ 1535 --no-adjust-warning| --no-change-warnings\n\ 1536 Do not issue warnings for non-existent\n\ 1537 sections.\n\ 1538 --only-keep-debug Copy only debugging information.\n\ 1539 --output-target=FORMAT Use the specified format for the output.\n\ 1540 --pad-to=ADDRESS Pad the output object up to the given address.\n\ 1541 --prefix-alloc-sections=STRING\n\ 1542 Prefix the section names of all the allocated\n\ 1543 sections with STRING.\n\ 1544 --prefix-sections=STRING Prefix the section names of all the sections\n\ 1545 with STRING.\n\ 1546 --prefix-symbols=STRING Prefix the symbol names of all the symbols\n\ 1547 with STRING.\n\ 1548 --rename-section OLDNAME=NEWNAME[,FLAGS]\n\ 1549 Rename and optionally change section flags.\n\ 1550 --set-section-flags SECTION=FLAGS\n\ 1551 Set section flags for the named section.\n\ 1552 Supported flags are: 'alloc', 'code',\n\ 1553 'contents', 'data', 'debug', 'load',\n\ 1554 'noload', 'readonly', 'rom', and 'shared'.\n\ 1555 --set-start=ADDRESS Set the start address of the ELF object.\n\ 1556 --srec-forceS3 Only generate S3 S-Records.\n\ 1557 --srec-len=LEN Set the maximum length of a S-Record line.\n\ 1558 --strip-unneeded Do not copy relocation information.\n" 1559 1560 static void 1561 elfcopy_usage(void) 1562 { 1563 (void) fprintf(stderr, ELFCOPY_USAGE_MESSAGE, ELFTC_GETPROGNAME()); 1564 exit(EXIT_FAILURE); 1565 } 1566 1567 #define MCS_USAGE_MESSAGE "\ 1568 Usage: %s [options] file...\n\ 1569 Manipulate the comment section in an ELF object.\n\n\ 1570 Options:\n\ 1571 -a STRING Append 'STRING' to the comment section.\n\ 1572 -c Remove duplicate entries from the comment section.\n\ 1573 -d Delete the comment section.\n\ 1574 -h | --help Print a help message and exit.\n\ 1575 -n NAME Operate on the ELF section with name 'NAME'.\n\ 1576 -p Print the contents of the comment section.\n\ 1577 -V | --version Print a version identifier and exit.\n" 1578 1579 static void 1580 mcs_usage(void) 1581 { 1582 (void) fprintf(stderr, MCS_USAGE_MESSAGE, ELFTC_GETPROGNAME()); 1583 exit(EXIT_FAILURE); 1584 } 1585 1586 #define STRIP_USAGE_MESSAGE "\ 1587 Usage: %s [options] file...\n\ 1588 Discard information from ELF objects.\n\n\ 1589 Options:\n\ 1590 -d | -g | -S | --strip-debug Remove debugging symbols.\n\ 1591 -h | --help Print a help message.\n\ 1592 -o FILE | --output-file FILE Write output to FILE.\n\ 1593 --only-keep-debug Keep debugging information only.\n\ 1594 -p | --preserve-dates Preserve access and modification times.\n\ 1595 -s | --strip-all Remove all symbols.\n\ 1596 --strip-unneeded Remove symbols not needed for relocation\n\ 1597 processing.\n\ 1598 -w | --wildcard Use shell-style patterns to name symbols.\n\ 1599 -x | --discard-all Discard all non-global symbols.\n\ 1600 -I TGT| --input-target=TGT (Accepted, but ignored).\n\ 1601 -K SYM | --keep-symbol=SYM Keep symbol 'SYM' in the output.\n\ 1602 -N SYM | --strip-symbol=SYM Remove symbol 'SYM' from the output.\n\ 1603 -O TGT | --output-target=TGT Set the output file format to 'TGT'.\n\ 1604 -R SEC | --remove-section=SEC Remove the section named 'SEC'.\n\ 1605 -V | --version Print a version identifier and exit.\n\ 1606 -X | --discard-locals Remove compiler-generated local symbols.\n" 1607 1608 static void 1609 strip_usage(void) 1610 { 1611 (void) fprintf(stderr, STRIP_USAGE_MESSAGE, ELFTC_GETPROGNAME()); 1612 exit(EXIT_FAILURE); 1613 } 1614 1615 static void 1616 print_version(void) 1617 { 1618 (void) printf("%s (%s)\n", ELFTC_GETPROGNAME(), elftc_version()); 1619 exit(EXIT_SUCCESS); 1620 } 1621 1622 /* 1623 * Compare the ending of s with end. 1624 */ 1625 static int 1626 strrcmp(const char *s, const char *end) 1627 { 1628 size_t endlen, slen; 1629 1630 slen = strlen(s); 1631 endlen = strlen(end); 1632 1633 if (slen >= endlen) 1634 s += slen - endlen; 1635 return (strcmp(s, end)); 1636 } 1637 1638 int 1639 main(int argc, char **argv) 1640 { 1641 struct elfcopy *ecp; 1642 1643 if (elf_version(EV_CURRENT) == EV_NONE) 1644 errx(EXIT_FAILURE, "ELF library initialization failed: %s", 1645 elf_errmsg(-1)); 1646 1647 ecp = calloc(1, sizeof(*ecp)); 1648 if (ecp == NULL) 1649 err(EXIT_FAILURE, "calloc failed"); 1650 1651 ecp->itf = ecp->otf = ETF_ELF; 1652 ecp->iec = ecp->oec = ELFCLASSNONE; 1653 ecp->oed = ELFDATANONE; 1654 ecp->abi = -1; 1655 /* There is always an empty section. */ 1656 ecp->nos = 1; 1657 ecp->fill = 0; 1658 1659 STAILQ_INIT(&ecp->v_seg); 1660 STAILQ_INIT(&ecp->v_sac); 1661 STAILQ_INIT(&ecp->v_sadd); 1662 STAILQ_INIT(&ecp->v_symop); 1663 STAILQ_INIT(&ecp->v_symfile); 1664 STAILQ_INIT(&ecp->v_arobj); 1665 TAILQ_INIT(&ecp->v_sec); 1666 1667 if ((ecp->progname = ELFTC_GETPROGNAME()) == NULL) 1668 ecp->progname = "elfcopy"; 1669 1670 if (strrcmp(ecp->progname, "strip") == 0) 1671 strip_main(ecp, argc, argv); 1672 else if (strrcmp(ecp->progname, "mcs") == 0) 1673 mcs_main(ecp, argc, argv); 1674 else { 1675 if (strrcmp(ecp->progname, "elfcopy") != 0 && 1676 strrcmp(ecp->progname, "objcopy") != 0) 1677 warnx("program mode not known, defaulting to elfcopy"); 1678 elfcopy_main(ecp, argc, argv); 1679 } 1680 1681 free_sec_add(ecp); 1682 free_sec_act(ecp); 1683 free(ecp); 1684 1685 exit(EXIT_SUCCESS); 1686 } 1687