| /* Discard section not used at runtime from object files. |
| Copyright (C) 2000, 2001, 2002, 2003, 2004 Red Hat, Inc. |
| Written by Ulrich Drepper <drepper@redhat.com>, 2000. |
| |
| This program is Open Source software; you can redistribute it and/or |
| modify it under the terms of the Open Software License version 1.0 as |
| published by the Open Source Initiative. |
| |
| You should have received a copy of the Open Software License along |
| with this program; if not, you may obtain a copy of the Open Software |
| License version 1.0 from http://www.opensource.org/licenses/osl.php or |
| by writing the Open Source Initiative c/o Lawrence Rosen, Esq., |
| 3001 King Ranch Road, Ukiah, CA 95482. */ |
| |
| #ifdef HAVE_CONFIG_H |
| # include <config.h> |
| #endif |
| |
| #include <argp.h> |
| #include <assert.h> |
| #include <byteswap.h> |
| #include <endian.h> |
| #include <error.h> |
| #include <fcntl.h> |
| #include <gelf.h> |
| #include <libelf.h> |
| #include <libintl.h> |
| #include <locale.h> |
| #include <mcheck.h> |
| #include <stdbool.h> |
| #include <stdio.h> |
| #include <stdio_ext.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <unistd.h> |
| #include <sys/param.h> |
| #include <sys/time.h> |
| |
| #include <elf-knowledge.h> |
| #include <libebl.h> |
| #include <system.h> |
| |
| |
| /* Name and version of program. */ |
| static void print_version (FILE *stream, struct argp_state *state); |
| void (*argp_program_version_hook) (FILE *, struct argp_state *) = print_version; |
| |
| |
| /* Values for the parameters which have no short form. */ |
| #define OPT_REMOVE_COMMENT 0x100 |
| #define OPT_PERMISSIVE 0x101 |
| |
| |
| /* Definitions of arguments for argp functions. */ |
| static const struct argp_option options[] = |
| { |
| { NULL, 0, NULL, 0, N_("Output selection:") }, |
| { NULL, 'o', "FILE", 0, N_("Place stripped output into FILE") }, |
| { NULL, 'f', "FILE", 0, N_("Extract the removed sections into FILE") }, |
| |
| { NULL, 0, NULL, 0, N_("Output options:") }, |
| { "strip-debug", 'g', NULL, 0, N_("Remove all debugging symbols") }, |
| { "preserve-dates", 'p', NULL, 0, |
| N_("Copy modified/access timestamps to the output") }, |
| { "remove-comment", OPT_REMOVE_COMMENT, NULL, 0, |
| N_("Remove .comment section") }, |
| { "permissive", OPT_PERMISSIVE, NULL, 0, |
| N_("Relax a few rules to handle slightly broken ELF files") }, |
| { NULL, 0, NULL, 0, NULL } |
| }; |
| |
| /* Short description of program. */ |
| static const char doc[] = N_("Discard symbols from object files."); |
| |
| /* Strings for arguments in help texts. */ |
| static const char args_doc[] = N_("[FILE...]"); |
| |
| /* Prototype for option handler. */ |
| static error_t parse_opt (int key, char *arg, struct argp_state *state); |
| |
| /* Function to print some extra text in the help message. */ |
| static char *more_help (int key, const char *text, void *input); |
| |
| /* Data structure to communicate with argp functions. */ |
| static struct argp argp = |
| { |
| options, parse_opt, args_doc, doc, NULL, more_help |
| }; |
| |
| |
| /* Print symbols in file named FNAME. */ |
| static int process_file (const char *fname); |
| |
| /* Handle one ELF file. */ |
| static int handle_elf (int fd, Elf *elf, const char *prefix, |
| const char *fname, mode_t mode, struct timeval tvp[2]); |
| |
| /* Handle all files contained in the archive. */ |
| static int handle_ar (int fd, Elf *elf, const char *prefix, const char *fname, |
| struct timeval tvp[2]); |
| |
| #define INTERNAL_ERROR(fname) \ |
| error (EXIT_FAILURE, 0, gettext ("%s: INTERNAL ERROR %d (%s-%s): %s"), \ |
| fname, __LINE__, VERSION, __DATE__, elf_errmsg (-1)) |
| |
| |
| /* Name of the output file. */ |
| static const char *output_fname; |
| |
| /* Name of the debug output file. */ |
| static const char *debug_fname; |
| |
| /* If true output files shall have same date as the input file. */ |
| static bool preserve_dates; |
| |
| /* If true .comment sections will be removed. */ |
| static bool remove_comment; |
| |
| /* If true remove all debug sections. */ |
| static bool remove_debug; |
| |
| /* If true relax some ELF rules for input files. */ |
| static bool permissive; |
| |
| |
| int |
| main (int argc, char *argv[]) |
| { |
| int remaining; |
| int result = 0; |
| |
| /* Make memory leak detection possible. */ |
| mtrace (); |
| |
| /* We use no threads here which can interfere with handling a stream. */ |
| __fsetlocking (stdin, FSETLOCKING_BYCALLER); |
| __fsetlocking (stdout, FSETLOCKING_BYCALLER); |
| __fsetlocking (stderr, FSETLOCKING_BYCALLER); |
| |
| /* Set locale. */ |
| setlocale (LC_ALL, ""); |
| |
| /* Make sure the message catalog can be found. */ |
| bindtextdomain (PACKAGE, LOCALEDIR); |
| |
| /* Initialize the message catalog. */ |
| textdomain (PACKAGE); |
| |
| /* Parse and process arguments. */ |
| argp_parse (&argp, argc, argv, 0, &remaining, NULL); |
| |
| /* Tell the library which version we are expecting. */ |
| elf_version (EV_CURRENT); |
| |
| if (remaining == argc) |
| /* The user didn't specify a name so we use a.out. */ |
| result = process_file ("a.out"); |
| else |
| { |
| /* If we have seen the `-o' or '-f' option there must be exactly one |
| input file. */ |
| if ((output_fname != NULL || debug_fname != NULL) |
| && remaining + 1 < argc) |
| error (EXIT_FAILURE, 0, gettext ("\ |
| Only one input file allowed together with '-o' and '-f'")); |
| |
| /* Process all the remaining files. */ |
| do |
| result |= process_file (argv[remaining]); |
| while (++remaining < argc); |
| } |
| |
| return result; |
| } |
| |
| |
| /* Print the version information. */ |
| static void |
| print_version (FILE *stream, struct argp_state *state) |
| { |
| fprintf (stream, "strip (%s) %s\n", PACKAGE_NAME, VERSION); |
| fprintf (stream, gettext ("\ |
| Copyright (C) %s Red Hat, Inc.\n\ |
| This is free software; see the source for copying conditions. There is NO\n\ |
| warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n\ |
| "), "2004"); |
| fprintf (stream, gettext ("Written by %s.\n"), "Ulrich Drepper"); |
| } |
| |
| |
| /* Handle program arguments. */ |
| static error_t |
| parse_opt (int key, char *arg, struct argp_state *state) |
| { |
| switch (key) |
| { |
| case 'f': |
| debug_fname = arg; |
| break; |
| |
| case 'o': |
| output_fname = arg; |
| break; |
| |
| case 'p': |
| preserve_dates = true; |
| break; |
| |
| case OPT_REMOVE_COMMENT: |
| remove_comment = true; |
| break; |
| |
| case 'g': |
| remove_debug = true; |
| break; |
| |
| case OPT_PERMISSIVE: |
| permissive = true; |
| break; |
| |
| default: |
| return ARGP_ERR_UNKNOWN; |
| } |
| return 0; |
| } |
| |
| |
| static char * |
| more_help (int key, const char *text, void *input) |
| { |
| char *buf; |
| |
| switch (key) |
| { |
| case ARGP_KEY_HELP_EXTRA: |
| /* We print some extra information. */ |
| if (asprintf (&buf, gettext ("Please report bugs to %s.\n"), |
| PACKAGE_BUGREPORT) < 0) |
| buf = NULL; |
| return buf; |
| |
| default: |
| break; |
| } |
| return (char *) text; |
| } |
| |
| |
| static int |
| process_file (const char *fname) |
| { |
| /* If we have to preserve the modify and access timestamps get them |
| now. We cannot use fstat() after opening the file since the open |
| would change the access time. */ |
| struct stat64 pre_st; |
| struct timeval tv[2]; |
| again: |
| if (preserve_dates) |
| { |
| if (stat64 (fname, &pre_st) != 0) |
| { |
| error (0, errno, gettext ("cannot stat input file \"%s\""), fname); |
| return 1; |
| } |
| |
| /* If we have to preserve the timestamp, we need it in the |
| format utimes() understands. */ |
| TIMESPEC_TO_TIMEVAL (&tv[0], &pre_st.st_atim); |
| TIMESPEC_TO_TIMEVAL (&tv[1], &pre_st.st_mtim); |
| } |
| |
| /* Open the file. */ |
| int fd = open (fname, O_RDWR); |
| if (fd == -1) |
| { |
| error (0, errno, gettext ("while opening \"%s\""), fname); |
| return 1; |
| } |
| |
| /* We always use fstat() even if we called stat() before. This is |
| done to make sure the information returned by stat() is for the |
| same file. */ |
| struct stat64 st; |
| if (fstat64 (fd, &st) != 0) |
| { |
| error (0, errno, gettext ("cannot stat input file \"%s\""), fname); |
| return 1; |
| } |
| /* Paranoid mode on. */ |
| if (preserve_dates |
| && (st.st_ino != pre_st.st_ino || st.st_dev != pre_st.st_dev)) |
| { |
| /* We detected a race. Try again. */ |
| close (fd); |
| goto again; |
| } |
| |
| /* Now get the ELF descriptor. */ |
| Elf *elf = elf_begin (fd, ELF_C_RDWR, NULL); |
| int result; |
| switch (elf_kind (elf)) |
| { |
| case ELF_K_ELF: |
| result = handle_elf (fd, elf, NULL, fname, st.st_mode & ACCESSPERMS, |
| preserve_dates ? tv : NULL); |
| break; |
| |
| case ELF_K_AR: |
| /* It is not possible to strip the content of an archive direct |
| the output to a specific file. */ |
| if (unlikely (output_fname != NULL)) |
| { |
| error (0, 0, gettext ("%s: cannot use -o when stripping archive"), |
| fname); |
| result = 1; |
| } |
| else |
| result = handle_ar (fd, elf, NULL, fname, preserve_dates ? tv : NULL); |
| break; |
| |
| default: |
| error (0, 0, gettext ("%s: File format not recognized"), fname); |
| result = 1; |
| break; |
| } |
| |
| if (unlikely (elf_end (elf) != 0)) |
| INTERNAL_ERROR (fname); |
| |
| close (fd); |
| |
| return result; |
| } |
| |
| |
| /* Maximum size of array allocated on stack. */ |
| #define MAX_STACK_ALLOC (400 * 1024) |
| |
| |
| static uint32_t |
| crc32_file (int fd, uint32_t *resp) |
| { |
| unsigned char buffer[1024 * 8]; |
| uint32_t crc = 0; |
| ssize_t count; |
| |
| /* We have to rewind. */ |
| if (lseek (fd, 0, SEEK_SET) < 0) |
| return 1; |
| |
| while ((count = TEMP_FAILURE_RETRY (read (fd, buffer, sizeof (buffer)))) > 0) |
| crc = crc32 (crc, buffer, count); |
| |
| *resp = crc; |
| |
| return count != 0; |
| } |
| |
| |
| static int |
| handle_elf (int fd, Elf *elf, const char *prefix, const char *fname, |
| mode_t mode, struct timeval tvp[2]) |
| { |
| size_t prefix_len = prefix == NULL ? 0 : strlen (prefix); |
| size_t fname_len = strlen (fname) + 1; |
| char *fullname = alloca (prefix_len + 1 + fname_len); |
| char *cp = fullname; |
| Elf *newelf; |
| Elf *debugelf = NULL; |
| char *tmp_debug_fname = NULL; |
| int result = 0; |
| GElf_Ehdr ehdr_mem; |
| GElf_Ehdr *ehdr; |
| size_t shstrndx; |
| size_t shnum; |
| struct shdr_info |
| { |
| Elf_Scn *scn; |
| GElf_Shdr shdr; |
| Elf_Data *data; |
| const char *name; |
| Elf32_Word idx; /* Index in new file. */ |
| Elf32_Word old_sh_link; /* Original value of shdr.sh_link. */ |
| Elf32_Word symtab_idx; |
| Elf32_Word version_idx; |
| Elf32_Word group_idx; |
| Elf32_Word group_cnt; |
| Elf_Scn *newscn; |
| struct Ebl_Strent *se; |
| Elf32_Word *newsymidx; |
| } *shdr_info = NULL; |
| Elf_Scn *scn; |
| size_t cnt; |
| size_t idx; |
| bool changes; |
| GElf_Ehdr newehdr_mem; |
| GElf_Ehdr *newehdr; |
| GElf_Ehdr debugehdr_mem; |
| GElf_Ehdr *debugehdr; |
| struct Ebl_Strtab *shst = NULL; |
| uint32_t debug_crc; |
| bool any_symtab_changes = false; |
| Elf_Data *shstrtab_data = NULL; |
| |
| /* Create the full name of the file. */ |
| if (prefix != NULL) |
| { |
| cp = mempcpy (cp, prefix, prefix_len); |
| *cp++ = ':'; |
| } |
| memcpy (cp, fname, fname_len); |
| |
| /* If we are not replacing the input file open a new file here. */ |
| if (output_fname != NULL) |
| { |
| fd = open (output_fname, O_RDWR | O_CREAT, mode); |
| if (unlikely (fd == -1)) |
| { |
| error (0, errno, gettext ("cannot open `%s'"), output_fname); |
| return 1; |
| } |
| } |
| |
| int debug_fd = -1; |
| |
| /* Get the EBL handling. The -g option is currently the only reason |
| we need EBL so dont open the backend unless necessary. */ |
| Ebl *ebl = NULL; |
| if (remove_debug) |
| { |
| ebl = ebl_openbackend (elf); |
| if (ebl == NULL) |
| { |
| error (0, errno, gettext ("cannot open EBL backend")); |
| result = 1; |
| goto fail; |
| } |
| } |
| |
| /* Open the additional file the debug information will be stored in. */ |
| if (debug_fname != NULL) |
| { |
| /* Create a temporary file name. We do not want to overwrite |
| the debug file if the file would not contain any |
| information. */ |
| size_t debug_fname_len = strlen (debug_fname); |
| tmp_debug_fname = (char *) alloca (debug_fname_len + sizeof (".XXXXXX")); |
| strcpy (mempcpy (tmp_debug_fname, debug_fname, debug_fname_len), |
| ".XXXXXX"); |
| |
| debug_fd = mkstemp (tmp_debug_fname); |
| if (unlikely (debug_fd == -1)) |
| { |
| error (0, errno, gettext ("cannot open `%s'"), debug_fname); |
| result = 1; |
| goto fail; |
| } |
| } |
| |
| /* Get the information from the old file. */ |
| ehdr = gelf_getehdr (elf, &ehdr_mem); |
| if (ehdr == NULL) |
| INTERNAL_ERROR (fname); |
| |
| /* Get the section header string table index. */ |
| if (unlikely (elf_getshstrndx (elf, &shstrndx) < 0)) |
| error (EXIT_FAILURE, 0, |
| gettext ("cannot get section header string table index")); |
| |
| /* We now create a new ELF descriptor for the same file. We |
| construct it almost exactly in the same way with some information |
| dropped. */ |
| if (output_fname != NULL) |
| newelf = elf_begin (fd, ELF_C_WRITE_MMAP, NULL); |
| else |
| newelf = elf_clone (elf, ELF_C_EMPTY); |
| |
| if (unlikely (gelf_newehdr (newelf, gelf_getclass (elf)) == 0) |
| || (ehdr->e_type != ET_REL |
| && unlikely (gelf_newphdr (newelf, ehdr->e_phnum) == 0))) |
| { |
| error (0, 0, gettext ("cannot create new file `%s': %s"), |
| output_fname, elf_errmsg (-1)); |
| goto fail; |
| } |
| |
| /* Copy over the old program header if needed. */ |
| if (ehdr->e_type != ET_REL) |
| for (cnt = 0; cnt < ehdr->e_phnum; ++cnt) |
| { |
| GElf_Phdr phdr_mem; |
| GElf_Phdr *phdr; |
| |
| phdr = gelf_getphdr (elf, cnt, &phdr_mem); |
| if (phdr == NULL |
| || unlikely (gelf_update_phdr (newelf, cnt, phdr) == 0)) |
| INTERNAL_ERROR (fname); |
| } |
| |
| if (debug_fname != NULL) |
| { |
| /* Also create an ELF descriptor for the debug file */ |
| debugelf = elf_begin (debug_fd, ELF_C_WRITE_MMAP, NULL); |
| if (unlikely (gelf_newehdr (debugelf, gelf_getclass (elf)) == 0) |
| || (ehdr->e_type != ET_REL |
| && unlikely (gelf_newphdr (debugelf, ehdr->e_phnum) == 0))) |
| { |
| error (0, 0, gettext ("cannot create new file `%s': %s"), |
| debug_fname, elf_errmsg (-1)); |
| goto fail_close; |
| } |
| |
| /* Copy over the old program header if needed. */ |
| if (ehdr->e_type != ET_REL) |
| for (cnt = 0; cnt < ehdr->e_phnum; ++cnt) |
| { |
| GElf_Phdr phdr_mem; |
| GElf_Phdr *phdr; |
| |
| phdr = gelf_getphdr (elf, cnt, &phdr_mem); |
| if (phdr == NULL |
| || unlikely (gelf_update_phdr (debugelf, cnt, phdr) == 0)) |
| INTERNAL_ERROR (fname); |
| } |
| } |
| |
| /* Number of sections. */ |
| if (unlikely (elf_getshnum (elf, &shnum) < 0)) |
| { |
| error (0, 0, gettext ("cannot determine number of sections: %s"), |
| elf_errmsg (-1)); |
| goto fail_close; |
| } |
| |
| /* Storage for section information. We leave room for two more |
| entries since we unconditionally create a section header string |
| table. Maybe some weird tool created an ELF file without one. |
| The other one is used for the debug link section. */ |
| if ((shnum + 2) * sizeof (struct shdr_info) > MAX_STACK_ALLOC) |
| shdr_info = (struct shdr_info *) xcalloc (shnum + 2, |
| sizeof (struct shdr_info)); |
| else |
| { |
| shdr_info = (struct shdr_info *) alloca ((shnum + 2) |
| * sizeof (struct shdr_info)); |
| memset (shdr_info, '\0', (shnum + 2) * sizeof (struct shdr_info)); |
| } |
| |
| /* Prepare section information data structure. */ |
| scn = NULL; |
| cnt = 1; |
| while ((scn = elf_nextscn (elf, scn)) != NULL) |
| { |
| /* This should always be true (i.e., there should not be any |
| holes in the numbering). */ |
| assert (elf_ndxscn (scn) == cnt); |
| |
| shdr_info[cnt].scn = scn; |
| |
| /* Get the header. */ |
| if (gelf_getshdr (scn, &shdr_info[cnt].shdr) == NULL) |
| INTERNAL_ERROR (fname); |
| |
| /* Get the name of the section. */ |
| shdr_info[cnt].name = elf_strptr (elf, shstrndx, |
| shdr_info[cnt].shdr.sh_name); |
| if (shdr_info[cnt].name == NULL) |
| { |
| error (0, 0, gettext ("illformed file `%s'"), fname); |
| goto fail_close; |
| } |
| |
| /* Mark them as present but not yet investigated. */ |
| shdr_info[cnt].idx = 1; |
| |
| /* Remember the shdr.sh_link value. */ |
| shdr_info[cnt].old_sh_link = shdr_info[cnt].shdr.sh_link; |
| |
| /* Sections in files other than relocatable object files which |
| are not loaded can be freely moved by us. In relocatable |
| object files everything can be moved. */ |
| if (ehdr->e_type == ET_REL |
| || (shdr_info[cnt].shdr.sh_flags & SHF_ALLOC) == 0) |
| shdr_info[cnt].shdr.sh_offset = 0; |
| |
| /* If this is an extended section index table store an |
| appropriate reference. */ |
| if (unlikely (shdr_info[cnt].shdr.sh_type == SHT_SYMTAB_SHNDX)) |
| { |
| assert (shdr_info[shdr_info[cnt].shdr.sh_link].symtab_idx == 0); |
| shdr_info[shdr_info[cnt].shdr.sh_link].symtab_idx = cnt; |
| } |
| else if (unlikely (shdr_info[cnt].shdr.sh_type == SHT_GROUP)) |
| { |
| Elf32_Word *grpref; |
| size_t inner; |
| |
| /* Cross-reference the sections contained in the section |
| group. */ |
| shdr_info[cnt].data = elf_getdata (shdr_info[cnt].scn, NULL); |
| if (shdr_info[cnt].data == NULL) |
| INTERNAL_ERROR (fname); |
| |
| /* XXX Fix for unaligned access. */ |
| grpref = (Elf32_Word *) shdr_info[cnt].data->d_buf; |
| for (inner = 1; |
| inner < shdr_info[cnt].data->d_size / sizeof (Elf32_Word); |
| ++inner) |
| shdr_info[grpref[inner]].group_idx = cnt; |
| |
| if (inner == 1 || (inner == 2 && (grpref[0] & GRP_COMDAT) == 0)) |
| /* If the section group contains only one element and this |
| is n COMDAT section we can drop it right away. */ |
| shdr_info[cnt].idx = 0; |
| else |
| shdr_info[cnt].group_cnt = inner - 1; |
| } |
| else if (unlikely (shdr_info[cnt].shdr.sh_type == SHT_GNU_versym)) |
| { |
| assert (shdr_info[shdr_info[cnt].shdr.sh_link].version_idx == 0); |
| shdr_info[shdr_info[cnt].shdr.sh_link].version_idx = cnt; |
| } |
| |
| /* If this section is part of a group make sure it is not |
| discarded right away. */ |
| if ((shdr_info[cnt].shdr.sh_flags & SHF_GROUP) != 0) |
| { |
| assert (shdr_info[cnt].group_idx != 0); |
| |
| if (shdr_info[shdr_info[cnt].group_idx].idx == 0) |
| { |
| /* The section group section will be removed. */ |
| shdr_info[cnt].group_idx = 0; |
| shdr_info[cnt].shdr.sh_flags &= ~SHF_GROUP; |
| } |
| } |
| |
| /* Increment the counter. */ |
| ++cnt; |
| } |
| |
| /* Now determine which sections can go away. The general rule is that |
| all sections which are not used at runtime are stripped out. But |
| there are a few exceptions: |
| |
| - special sections named ".comment" and ".note" are kept |
| - OS or architecture specific sections are kept since we might not |
| know how to handle them |
| - if a section is referred to from a section which is not removed |
| in the sh_link or sh_info element it cannot be removed either |
| */ |
| for (cnt = 1; cnt < shnum; ++cnt) |
| /* Check whether the section can be removed. */ |
| if (SECTION_STRIP_P (ebl, elf, ehdr, &shdr_info[cnt].shdr, |
| shdr_info[cnt].name, remove_comment, remove_debug)) |
| { |
| /* For now assume this section will be removed. */ |
| shdr_info[cnt].idx = 0; |
| |
| idx = shdr_info[cnt].group_idx; |
| while (idx != 0) |
| { |
| /* If the references section group is a normal section |
| group and has one element remaining, or if it is an |
| empty COMDAT section group it is removed. */ |
| bool is_comdat; |
| |
| /* The section group data is already loaded. */ |
| assert (shdr_info[idx].data != NULL); |
| |
| is_comdat = (((Elf32_Word *) shdr_info[idx].data->d_buf)[0] |
| & GRP_COMDAT) != 0; |
| |
| --shdr_info[idx].group_cnt; |
| if ((!is_comdat && shdr_info[idx].group_cnt == 1) |
| || (is_comdat && shdr_info[idx].group_cnt == 0)) |
| { |
| shdr_info[idx].idx = 0; |
| /* Continue recursively. */ |
| idx = shdr_info[idx].group_idx; |
| } |
| else |
| break; |
| } |
| } |
| |
| /* Mark the SHT_NULL section as handled. */ |
| shdr_info[0].idx = 2; |
| |
| |
| /* Handle exceptions: section groups and cross-references. We might |
| have to repeat this a few times since the resetting of the flag |
| might propagate. */ |
| do |
| { |
| changes = false; |
| |
| for (cnt = 1; cnt < shnum; ++cnt) |
| { |
| if (shdr_info[cnt].idx == 0) |
| { |
| /* If a relocation section is marked as being removed make |
| sure the section it is relocating is removed, too. */ |
| if ((shdr_info[cnt].shdr.sh_type == SHT_REL |
| || shdr_info[cnt].shdr.sh_type == SHT_RELA) |
| && shdr_info[shdr_info[cnt].shdr.sh_info].idx != 0) |
| shdr_info[cnt].idx = 1; |
| } |
| |
| if (shdr_info[cnt].idx == 1) |
| { |
| /* The content of symbol tables we don't remove must not |
| reference any section which we do remove. Otherwise |
| we cannot remove the section. */ |
| if (shdr_info[cnt].shdr.sh_type == SHT_DYNSYM |
| || shdr_info[cnt].shdr.sh_type == SHT_SYMTAB) |
| { |
| Elf_Data *symdata; |
| Elf_Data *xndxdata; |
| size_t elsize; |
| |
| /* Make sure the data is loaded. */ |
| if (shdr_info[cnt].data == NULL) |
| { |
| shdr_info[cnt].data |
| = elf_getdata (shdr_info[cnt].scn, NULL); |
| if (shdr_info[cnt].data == NULL) |
| INTERNAL_ERROR (fname); |
| } |
| symdata = shdr_info[cnt].data; |
| |
| /* If there is an extended section index table load it |
| as well. */ |
| if (shdr_info[cnt].symtab_idx != 0 |
| && shdr_info[shdr_info[cnt].symtab_idx].data == NULL) |
| { |
| assert (shdr_info[cnt].shdr.sh_type == SHT_SYMTAB); |
| |
| shdr_info[shdr_info[cnt].symtab_idx].data |
| = elf_getdata (shdr_info[shdr_info[cnt].symtab_idx].scn, |
| NULL); |
| if (shdr_info[shdr_info[cnt].symtab_idx].data == NULL) |
| INTERNAL_ERROR (fname); |
| } |
| xndxdata = shdr_info[shdr_info[cnt].symtab_idx].data; |
| |
| /* Go through all symbols and make sure the section they |
| reference is not removed. */ |
| elsize = gelf_fsize (elf, ELF_T_SYM, 1, ehdr->e_version); |
| |
| for (size_t inner = 0; |
| inner < shdr_info[cnt].data->d_size / elsize; |
| ++inner) |
| { |
| GElf_Sym sym_mem; |
| Elf32_Word xndx; |
| GElf_Sym *sym; |
| size_t scnidx; |
| |
| sym = gelf_getsymshndx (symdata, xndxdata, inner, |
| &sym_mem, &xndx); |
| if (sym == NULL) |
| INTERNAL_ERROR (fname); |
| |
| scnidx = sym->st_shndx; |
| if (scnidx == SHN_UNDEF || scnidx >= shnum |
| || (scnidx >= SHN_LORESERVE |
| && scnidx <= SHN_HIRESERVE |
| && scnidx != SHN_XINDEX) |
| /* Don't count in the section symbols. */ |
| || GELF_ST_TYPE (sym->st_info) == STT_SECTION) |
| /* This is no section index, leave it alone. */ |
| continue; |
| else if (scnidx == SHN_XINDEX) |
| scnidx = xndx; |
| |
| if (shdr_info[scnidx].idx == 0) |
| { |
| /* Mark this section as used. */ |
| shdr_info[scnidx].idx = 1; |
| changes |= scnidx < cnt; |
| } |
| } |
| } |
| |
| /* Cross referencing happens: |
| - for the cases the ELF specification says. That are |
| + SHT_DYNAMIC in sh_link to string table |
| + SHT_HASH in sh_link to symbol table |
| + SHT_REL and SHT_RELA in sh_link to symbol table |
| + SHT_SYMTAB and SHT_DYNSYM in sh_link to string table |
| + SHT_GROUP in sh_link to symbol table |
| + SHT_SYMTAB_SHNDX in sh_link to symbol table |
| Other (OS or architecture-specific) sections might as |
| well use this field so we process it unconditionally. |
| - references inside section groups |
| - specially marked references in sh_info if the SHF_INFO_LINK |
| flag is set |
| */ |
| |
| if (shdr_info[shdr_info[cnt].shdr.sh_link].idx == 0) |
| { |
| shdr_info[shdr_info[cnt].shdr.sh_link].idx = 1; |
| changes |= shdr_info[cnt].shdr.sh_link < cnt; |
| } |
| |
| /* Handle references through sh_info. */ |
| if (SH_INFO_LINK_P (&shdr_info[cnt].shdr) |
| && shdr_info[shdr_info[cnt].shdr.sh_info].idx == 0) |
| { |
| shdr_info[shdr_info[cnt].shdr.sh_info].idx = 1; |
| changes |= shdr_info[cnt].shdr.sh_info < cnt; |
| } |
| |
| /* Mark the section as investigated. */ |
| shdr_info[cnt].idx = 2; |
| } |
| } |
| } |
| while (changes); |
| |
| /* Write out a copy of all the sections to the debug output file. |
| The ones that are not removed in the stripped file are SHT_NOBITS */ |
| if (debug_fname != NULL) |
| { |
| for (cnt = 1; cnt < shnum; ++cnt) |
| { |
| Elf_Data *debugdata; |
| GElf_Shdr debugshdr; |
| int discard_section; |
| |
| scn = elf_newscn (debugelf); |
| if (scn == NULL) |
| error (EXIT_FAILURE, 0, |
| gettext ("while generating output file: %s"), |
| elf_errmsg (-1)); |
| |
| discard_section = shdr_info[cnt].idx > 0 && cnt != ehdr->e_shstrndx; |
| |
| /* Set the section header in the new file. */ |
| debugshdr = shdr_info[cnt].shdr; |
| if (discard_section) |
| debugshdr.sh_type = SHT_NOBITS; |
| |
| if (unlikely (gelf_update_shdr (scn, &debugshdr)) == 0) |
| /* There cannot be any overflows. */ |
| INTERNAL_ERROR (fname); |
| |
| /* Get the data from the old file if necessary. */ |
| if (shdr_info[cnt].data == NULL) |
| { |
| shdr_info[cnt].data = elf_getdata (shdr_info[cnt].scn, NULL); |
| if (shdr_info[cnt].data == NULL) |
| INTERNAL_ERROR (fname); |
| } |
| |
| /* Set the data. This is done by copying from the old file. */ |
| debugdata = elf_newdata (scn); |
| if (debugdata == NULL) |
| INTERNAL_ERROR (fname); |
| |
| /* Copy the structure. */ |
| *debugdata = *shdr_info[cnt].data; |
| if (discard_section) |
| debugdata->d_buf = NULL; |
| } |
| |
| /* Finish the ELF header. Fill in the fields not handled by |
| libelf from the old file. */ |
| debugehdr = gelf_getehdr (debugelf, &debugehdr_mem); |
| if (debugehdr == NULL) |
| INTERNAL_ERROR (fname); |
| |
| memcpy (debugehdr->e_ident, ehdr->e_ident, EI_NIDENT); |
| debugehdr->e_type = ehdr->e_type; |
| debugehdr->e_machine = ehdr->e_machine; |
| debugehdr->e_version = ehdr->e_version; |
| debugehdr->e_entry = ehdr->e_entry; |
| debugehdr->e_flags = ehdr->e_flags; |
| debugehdr->e_shstrndx = ehdr->e_shstrndx; |
| |
| if (unlikely (gelf_update_ehdr (debugelf, debugehdr)) == 0) |
| { |
| error (0, 0, gettext ("%s: error while creating ELF header: %s"), |
| debug_fname, elf_errmsg (-1)); |
| result = 1; |
| goto fail_close; |
| } |
| |
| /* Finally write the file. */ |
| if (unlikely (elf_update (debugelf, ELF_C_WRITE)) == -1) |
| { |
| error (0, 0, gettext ("while writing `%s': %s"), |
| debug_fname, elf_errmsg (-1)); |
| result = 1; |
| goto fail_close; |
| } |
| |
| /* Create the real output file. First rename, then change the |
| mode. */ |
| if (rename (tmp_debug_fname, debug_fname) != 0 |
| || fchmod (debug_fd, mode) != 0) |
| { |
| error (0, errno, gettext ("while creating '%s'"), debug_fname); |
| result = 1; |
| goto fail_close; |
| } |
| |
| /* The temporary file does not exist anymore. */ |
| tmp_debug_fname = NULL; |
| |
| /* Compute the checksum which we will add to the executable. */ |
| if (crc32_file (debug_fd, &debug_crc) != 0) |
| { |
| error (0, errno, |
| gettext ("while computing checksum for debug information")); |
| unlink (debug_fname); |
| result = 1; |
| goto fail_close; |
| } |
| |
| } |
| |
| /* Mark the section header string table as unused, we will create |
| a new one. */ |
| shdr_info[shstrndx].idx = 0; |
| |
| /* We need a string table for the section headers. */ |
| shst = ebl_strtabinit (true); |
| if (shst == NULL) |
| error (EXIT_FAILURE, errno, gettext ("while preparing output for `%s'"), |
| output_fname ?: fname); |
| |
| /* Assign new section numbers. */ |
| shdr_info[0].idx = 0; |
| for (cnt = idx = 1; cnt < shnum; ++cnt) |
| if (shdr_info[cnt].idx > 0) |
| { |
| shdr_info[cnt].idx = idx++; |
| |
| /* Create a new section. */ |
| shdr_info[cnt].newscn = elf_newscn (newelf); |
| if (shdr_info[cnt].newscn == NULL) |
| error (EXIT_FAILURE, 0, gettext ("while generating output file: %s"), |
| elf_errmsg (-1)); |
| |
| assert (elf_ndxscn (shdr_info[cnt].newscn) == shdr_info[cnt].idx); |
| |
| /* Add this name to the section header string table. */ |
| shdr_info[cnt].se = ebl_strtabadd (shst, shdr_info[cnt].name, 0); |
| } |
| |
| /* Test whether we are doing anything at all. */ |
| if (cnt == idx) |
| /* Nope, all removable sections are already gone. */ |
| goto fail_close; |
| |
| /* Create the reference to the file with the debug info. */ |
| if (debug_fname != NULL) |
| { |
| char *debug_basename; |
| off_t crc_offset; |
| |
| /* Add the section header string table section name. */ |
| shdr_info[cnt].se = ebl_strtabadd (shst, ".gnu_debuglink", 15); |
| shdr_info[cnt].idx = idx++; |
| |
| /* Create the section header. */ |
| shdr_info[cnt].shdr.sh_type = SHT_PROGBITS; |
| shdr_info[cnt].shdr.sh_flags = 0; |
| shdr_info[cnt].shdr.sh_addr = 0; |
| shdr_info[cnt].shdr.sh_link = SHN_UNDEF; |
| shdr_info[cnt].shdr.sh_info = SHN_UNDEF; |
| shdr_info[cnt].shdr.sh_entsize = 0; |
| shdr_info[cnt].shdr.sh_addralign = 4; |
| /* We set the offset to zero here. Before we write the ELF file the |
| field must have the correct value. This is done in the final |
| loop over all section. Then we have all the information needed. */ |
| shdr_info[cnt].shdr.sh_offset = 0; |
| |
| /* Create the section. */ |
| shdr_info[cnt].newscn = elf_newscn (newelf); |
| if (shdr_info[cnt].newscn == NULL) |
| error (EXIT_FAILURE, 0, |
| gettext ("while create section header section: %s"), |
| elf_errmsg (-1)); |
| assert (elf_ndxscn (shdr_info[cnt].newscn) == shdr_info[cnt].idx); |
| |
| shdr_info[cnt].data = elf_newdata (shdr_info[cnt].newscn); |
| if (shdr_info[cnt].data == NULL) |
| error (EXIT_FAILURE, 0, gettext ("cannot allocate section data: %s"), |
| elf_errmsg (-1)); |
| |
| debug_basename = basename (debug_fname); |
| crc_offset = strlen (debug_basename) + 1; |
| /* Align to 4 byte boundary */ |
| crc_offset = ((crc_offset - 1) & ~3) + 4; |
| |
| shdr_info[cnt].data->d_align = 4; |
| shdr_info[cnt].shdr.sh_size = shdr_info[cnt].data->d_size |
| = crc_offset + 4; |
| shdr_info[cnt].data->d_buf = xcalloc (1, shdr_info[cnt].data->d_size); |
| |
| strcpy (shdr_info[cnt].data->d_buf, debug_basename); |
| /* Store the crc value in the correct byteorder */ |
| if ((__BYTE_ORDER == __LITTLE_ENDIAN |
| && ehdr->e_ident[EI_DATA] == ELFDATA2MSB) |
| || (__BYTE_ORDER == __BIG_ENDIAN |
| && ehdr->e_ident[EI_DATA] == ELFDATA2LSB)) |
| debug_crc = bswap_32 (debug_crc); |
| memcpy ((char *)shdr_info[cnt].data->d_buf + crc_offset, |
| (char *) &debug_crc, 4); |
| |
| /* One more section done. */ |
| ++cnt; |
| } |
| |
| /* Index of the section header table in the shdr_info array. */ |
| size_t shdridx = cnt; |
| |
| /* Add the section header string table section name. */ |
| shdr_info[cnt].se = ebl_strtabadd (shst, ".shstrtab", 10); |
| shdr_info[cnt].idx = idx; |
| |
| /* Create the section header. */ |
| shdr_info[cnt].shdr.sh_type = SHT_STRTAB; |
| shdr_info[cnt].shdr.sh_flags = 0; |
| shdr_info[cnt].shdr.sh_addr = 0; |
| shdr_info[cnt].shdr.sh_link = SHN_UNDEF; |
| shdr_info[cnt].shdr.sh_info = SHN_UNDEF; |
| shdr_info[cnt].shdr.sh_entsize = 0; |
| /* We set the offset to zero here. Before we write the ELF file the |
| field must have the correct value. This is done in the final |
| loop over all section. Then we have all the information needed. */ |
| shdr_info[cnt].shdr.sh_offset = 0; |
| shdr_info[cnt].shdr.sh_addralign = 1; |
| |
| /* Create the section. */ |
| shdr_info[cnt].newscn = elf_newscn (newelf); |
| if (shdr_info[cnt].newscn == NULL) |
| error (EXIT_FAILURE, 0, |
| gettext ("while create section header section: %s"), |
| elf_errmsg (-1)); |
| assert (elf_ndxscn (shdr_info[cnt].newscn) == idx); |
| |
| /* Finalize the string table and fill in the correct indices in the |
| section headers. */ |
| shstrtab_data = elf_newdata (shdr_info[cnt].newscn); |
| if (shstrtab_data == NULL) |
| error (EXIT_FAILURE, 0, |
| gettext ("while create section header string table: %s"), |
| elf_errmsg (-1)); |
| ebl_strtabfinalize (shst, shstrtab_data); |
| |
| /* We have to set the section size. */ |
| shdr_info[cnt].shdr.sh_size = shstrtab_data->d_size; |
| |
| /* Update the section information. */ |
| GElf_Off lastoffset = 0; |
| for (cnt = 1; cnt <= shdridx; ++cnt) |
| if (shdr_info[cnt].idx > 0) |
| { |
| Elf_Data *newdata; |
| |
| scn = elf_getscn (newelf, shdr_info[cnt].idx); |
| assert (scn != NULL); |
| |
| /* Update the name. */ |
| shdr_info[cnt].shdr.sh_name = ebl_strtaboffset (shdr_info[cnt].se); |
| |
| /* Update the section header from the input file. Some fields |
| might be section indeces which now have to be adjusted. */ |
| if (shdr_info[cnt].shdr.sh_link != 0) |
| shdr_info[cnt].shdr.sh_link = |
| shdr_info[shdr_info[cnt].shdr.sh_link].idx; |
| |
| if (shdr_info[cnt].shdr.sh_type == SHT_GROUP) |
| { |
| assert (shdr_info[cnt].data != NULL); |
| |
| Elf32_Word *grpref = (Elf32_Word *) shdr_info[cnt].data->d_buf; |
| for (size_t inner = 0; |
| inner < shdr_info[cnt].data->d_size / sizeof (Elf32_Word); |
| ++inner) |
| grpref[inner] = shdr_info[grpref[inner]].idx; |
| } |
| |
| /* Handle the SHT_REL, SHT_RELA, and SHF_INFO_LINK flag. */ |
| if (SH_INFO_LINK_P (&shdr_info[cnt].shdr)) |
| shdr_info[cnt].shdr.sh_info = |
| shdr_info[shdr_info[cnt].shdr.sh_info].idx; |
| |
| /* Get the data from the old file if necessary. We already |
| created the data for the section header string table. */ |
| if (cnt < shnum) |
| { |
| if (shdr_info[cnt].data == NULL) |
| { |
| shdr_info[cnt].data = elf_getdata (shdr_info[cnt].scn, NULL); |
| if (shdr_info[cnt].data == NULL) |
| INTERNAL_ERROR (fname); |
| } |
| |
| /* Set the data. This is done by copying from the old file. */ |
| newdata = elf_newdata (scn); |
| if (newdata == NULL) |
| INTERNAL_ERROR (fname); |
| |
| /* Copy the structure. */ |
| *newdata = *shdr_info[cnt].data; |
| |
| /* We know the size. */ |
| shdr_info[cnt].shdr.sh_size = shdr_info[cnt].data->d_size; |
| |
| /* We have to adjust symtol tables. The st_shndx member might |
| have to be updated. */ |
| if (shdr_info[cnt].shdr.sh_type == SHT_DYNSYM |
| || shdr_info[cnt].shdr.sh_type == SHT_SYMTAB) |
| { |
| Elf_Data *versiondata = NULL; |
| Elf_Data *shndxdata = NULL; |
| |
| size_t elsize = gelf_fsize (elf, ELF_T_SYM, 1, |
| ehdr->e_version); |
| |
| if (shdr_info[cnt].symtab_idx != 0) |
| { |
| assert (shdr_info[cnt].shdr.sh_type == SHT_SYMTAB_SHNDX); |
| /* This section has extended section information. |
| We have to modify that information, too. */ |
| shndxdata = elf_getdata (shdr_info[shdr_info[cnt].symtab_idx].scn, |
| NULL); |
| |
| assert ((versiondata->d_size / sizeof (Elf32_Word)) |
| >= shdr_info[cnt].data->d_size / elsize); |
| } |
| |
| if (shdr_info[cnt].version_idx != 0) |
| { |
| assert (shdr_info[cnt].shdr.sh_type == SHT_DYNSYM); |
| /* This section has associated version |
| information. We have to modify that |
| information, too. */ |
| versiondata = elf_getdata (shdr_info[shdr_info[cnt].version_idx].scn, |
| NULL); |
| |
| assert ((versiondata->d_size / sizeof (GElf_Versym)) |
| >= shdr_info[cnt].data->d_size / elsize); |
| } |
| |
| shdr_info[cnt].newsymidx |
| = (Elf32_Word *) xcalloc (shdr_info[cnt].data->d_size |
| / elsize, sizeof (Elf32_Word)); |
| |
| bool last_was_local = true; |
| size_t destidx; |
| size_t inner; |
| for (destidx = inner = 1; |
| inner < shdr_info[cnt].data->d_size / elsize; |
| ++inner) |
| { |
| Elf32_Word sec; |
| GElf_Sym sym_mem; |
| Elf32_Word xshndx; |
| GElf_Sym *sym = gelf_getsymshndx (shdr_info[cnt].data, |
| shndxdata, inner, |
| &sym_mem, &xshndx); |
| if (sym == NULL) |
| INTERNAL_ERROR (fname); |
| |
| if (sym->st_shndx == SHN_UNDEF |
| || (sym->st_shndx >= shnum |
| && sym->st_shndx != SHN_XINDEX)) |
| { |
| /* This is no section index, leave it alone |
| unless it is moved. */ |
| if (destidx != inner |
| && gelf_update_symshndx (shdr_info[cnt].data, |
| shndxdata, |
| destidx, sym, |
| xshndx) == 0) |
| INTERNAL_ERROR (fname); |
| |
| shdr_info[cnt].newsymidx[inner] = destidx++; |
| |
| if (last_was_local |
| && GELF_ST_BIND (sym->st_info) != STB_LOCAL) |
| { |
| last_was_local = false; |
| shdr_info[cnt].shdr.sh_info = destidx - 1; |
| } |
| |
| continue; |
| } |
| |
| /* Get the full section index, if necessary from the |
| XINDEX table. */ |
| if (sym->st_shndx != SHN_XINDEX) |
| sec = shdr_info[sym->st_shndx].idx; |
| else |
| { |
| assert (shndxdata != NULL); |
| |
| sec = shdr_info[xshndx].idx; |
| } |
| |
| if (sec != 0) |
| { |
| GElf_Section nshndx; |
| Elf32_Word nxshndx; |
| |
| if (sec < SHN_LORESERVE) |
| { |
| nshndx = sec; |
| nxshndx = 0; |
| } |
| else |
| { |
| nshndx = SHN_XINDEX; |
| nxshndx = sec; |
| } |
| |
| assert (sec < SHN_LORESERVE || shndxdata != NULL); |
| |
| if ((inner != destidx || nshndx != sym->st_shndx |
| || (shndxdata != NULL && nxshndx != xshndx)) |
| && (sym->st_shndx = nshndx, |
| gelf_update_symshndx (shdr_info[cnt].data, |
| shndxdata, |
| destidx, sym, |
| nxshndx) == 0)) |
| INTERNAL_ERROR (fname); |
| |
| shdr_info[cnt].newsymidx[inner] = destidx++; |
| |
| if (last_was_local |
| && GELF_ST_BIND (sym->st_info) != STB_LOCAL) |
| { |
| last_was_local = false; |
| shdr_info[cnt].shdr.sh_info = destidx - 1; |
| } |
| } |
| else |
| /* This is a section symbol for a section which has |
| been removed. */ |
| assert (GELF_ST_TYPE (sym->st_info) == STT_SECTION); |
| } |
| |
| if (destidx != inner) |
| { |
| /* The size of the symbol table changed. */ |
| shdr_info[cnt].shdr.sh_size = newdata->d_size |
| = destidx * elsize; |
| any_symtab_changes = true; |
| } |
| else |
| { |
| /* The symbol table didn't really change. */ |
| free (shdr_info[cnt].newsymidx); |
| shdr_info[cnt].newsymidx = NULL; |
| } |
| } |
| } |
| |
| /* If we have to, compute the offset of the section. */ |
| if (shdr_info[cnt].shdr.sh_offset == 0) |
| shdr_info[cnt].shdr.sh_offset |
| = ((lastoffset + shdr_info[cnt].shdr.sh_addralign - 1) |
| & ~((GElf_Off) (shdr_info[cnt].shdr.sh_addralign - 1))); |
| |
| /* Set the section header in the new file. */ |
| if (unlikely (gelf_update_shdr (scn, &shdr_info[cnt].shdr) == 0)) |
| /* There cannot be any overflows. */ |
| INTERNAL_ERROR (fname); |
| |
| /* Remember the last section written so far. */ |
| GElf_Off filesz = (shdr_info[cnt].shdr.sh_type != SHT_NOBITS |
| ? shdr_info[cnt].shdr.sh_size : 0); |
| if (lastoffset < shdr_info[cnt].shdr.sh_offset + filesz) |
| lastoffset = shdr_info[cnt].shdr.sh_offset + filesz; |
| } |
| |
| /* Adjust symbol references if symbol tables changed. */ |
| if (any_symtab_changes) |
| { |
| /* Find all relocation sections which use this |
| symbol table. */ |
| for (cnt = 1; cnt <= shdridx; ++cnt) |
| { |
| if (shdr_info[cnt].idx == 0) |
| /* Ignore sections which are discarded. */ |
| continue; |
| |
| if (shdr_info[cnt].shdr.sh_type == SHT_REL |
| || shdr_info[cnt].shdr.sh_type == SHT_RELA) |
| { |
| /* If the symbol table hasn't changed, do not do anything. */ |
| if (shdr_info[shdr_info[cnt].old_sh_link].newsymidx == NULL) |
| continue; |
| |
| Elf32_Word *newsymidx |
| = shdr_info[shdr_info[cnt].old_sh_link].newsymidx; |
| Elf_Data *d = elf_getdata (elf_getscn (newelf, |
| shdr_info[cnt].idx), |
| NULL); |
| assert (d != NULL); |
| size_t nrels = (shdr_info[cnt].shdr.sh_size |
| / shdr_info[cnt].shdr.sh_entsize); |
| |
| if (shdr_info[cnt].shdr.sh_type == SHT_REL) |
| for (size_t relidx = 0; relidx < nrels; ++relidx) |
| { |
| GElf_Rel rel_mem; |
| if (gelf_getrel (d, relidx, &rel_mem) == NULL) |
| INTERNAL_ERROR (fname); |
| |
| size_t symidx = GELF_R_SYM (rel_mem.r_info); |
| if (newsymidx[symidx] != symidx) |
| { |
| rel_mem.r_info |
| = GELF_R_INFO (newsymidx[symidx], |
| GELF_R_TYPE (rel_mem.r_info)); |
| |
| if (gelf_update_rel (d, relidx, &rel_mem) == 0) |
| INTERNAL_ERROR (fname); |
| } |
| } |
| else |
| for (size_t relidx = 0; relidx < nrels; ++relidx) |
| { |
| GElf_Rela rel_mem; |
| if (gelf_getrela (d, relidx, &rel_mem) == NULL) |
| INTERNAL_ERROR (fname); |
| |
| size_t symidx = GELF_R_SYM (rel_mem.r_info); |
| if (newsymidx[symidx] != symidx) |
| { |
| rel_mem.r_info |
| = GELF_R_INFO (newsymidx[symidx], |
| GELF_R_TYPE (rel_mem.r_info)); |
| |
| if (gelf_update_rela (d, relidx, &rel_mem) == 0) |
| INTERNAL_ERROR (fname); |
| } |
| } |
| } |
| else if (shdr_info[cnt].shdr.sh_type == SHT_HASH) |
| { |
| /* We have to recompute the hash table. */ |
| Elf32_Word symtabidx = shdr_info[cnt].old_sh_link; |
| |
| /* We do not have to do anything if the symbol table was |
| not changed. */ |
| if (shdr_info[symtabidx].newsymidx == NULL) |
| continue; |
| |
| /* The symbol version section in the new file. */ |
| scn = elf_getscn (newelf, shdr_info[cnt].idx); |
| |
| /* The symbol table data. */ |
| Elf_Data *symd = elf_getdata (elf_getscn (newelf, |
| shdr_info[symtabidx].idx), |
| NULL); |
| assert (symd != NULL); |
| |
| /* The hash table data. */ |
| Elf_Data *hashd = elf_getdata (scn, NULL); |
| assert (hashd != NULL); |
| |
| if (shdr_info[cnt].shdr.sh_entsize == sizeof (Elf32_Word)) |
| { |
| /* Sane arches first. */ |
| Elf32_Word *bucket = (Elf32_Word *) hashd->d_buf; |
| |
| size_t strshndx = shdr_info[symtabidx].old_sh_link; |
| size_t elsize = gelf_fsize (elf, ELF_T_SYM, 1, |
| ehdr->e_version); |
| |
| /* Convert to the correct byte order. */ |
| if (gelf_xlatetom (newelf, hashd, hashd, |
| BYTE_ORDER == LITTLE_ENDIAN |
| ? ELFDATA2LSB : ELFDATA2MSB) == NULL) |
| INTERNAL_ERROR (fname); |
| |
| /* Adjust the nchain value. The symbol table size |
| changed. We keep the same size for the bucket array. */ |
| bucket[1] = symd->d_size / elsize; |
| Elf32_Word nbucket = bucket[0]; |
| bucket += 2; |
| Elf32_Word *chain = bucket + nbucket; |
| |
| /* New size of the section. */ |
| GElf_Shdr shdr_mem; |
| GElf_Shdr *shdr = gelf_getshdr (scn, &shdr_mem); |
| shdr->sh_size = hashd->d_size |
| = (2 + symd->d_size / elsize + nbucket) |
| * sizeof (Elf32_Word); |
| (void) gelf_update_shdr (scn, shdr); |
| |
| /* Clear the arrays. */ |
| memset (bucket, '\0', |
| (symd->d_size / elsize + nbucket) |
| * sizeof (Elf32_Word)); |
| |
| for (size_t inner = shdr_info[symtabidx].shdr.sh_info; |
| inner < symd->d_size / elsize; ++inner) |
| { |
| GElf_Sym sym_mem; |
| GElf_Sym *sym = gelf_getsym (symd, inner, &sym_mem); |
| assert (sym != NULL); |
| |
| const char *name = elf_strptr (elf, strshndx, |
| sym->st_name); |
| assert (name != NULL); |
| size_t hidx = elf_hash (name) % nbucket; |
| |
| if (bucket[hidx] == 0) |
| bucket[hidx] = inner; |
| else |
| { |
| hidx = bucket[hidx]; |
| |
| while (chain[hidx] != 0) |
| hidx = chain[hidx]; |
| |
| chain[hidx] = inner; |
| } |
| } |
| } |
| else |
| { |
| /* Alpha and S390 64-bit use 64-bit SHT_HASH entries. */ |
| assert (shdr_info[cnt].shdr.sh_entsize |
| == sizeof (Elf64_Xword)); |
| |
| Elf64_Xword *bucket = (Elf64_Xword *) hashd->d_buf; |
| |
| size_t strshndx = shdr_info[symtabidx].old_sh_link; |
| size_t elsize = gelf_fsize (elf, ELF_T_SYM, 1, |
| ehdr->e_version); |
| |
| /* Convert to the correct byte order. */ |
| if (gelf_xlatetom (newelf, hashd, hashd, |
| BYTE_ORDER == LITTLE_ENDIAN |
| ? ELFDATA2LSB : ELFDATA2MSB) == NULL) |
| INTERNAL_ERROR (fname); |
| |
| /* Adjust the nchain value. The symbol table size |
| changed. We keep the same size for the bucket array. */ |
| bucket[1] = symd->d_size / elsize; |
| Elf64_Xword nbucket = bucket[0]; |
| bucket += 2; |
| Elf64_Xword *chain = bucket + nbucket; |
| |
| /* New size of the section. */ |
| GElf_Shdr shdr_mem; |
| GElf_Shdr *shdr = gelf_getshdr (scn, &shdr_mem); |
| shdr->sh_size = hashd->d_size |
| = (2 + symd->d_size / elsize + nbucket) |
| * sizeof (Elf64_Xword); |
| (void) gelf_update_shdr (scn, shdr); |
| |
| /* Clear the arrays. */ |
| memset (bucket, '\0', |
| (symd->d_size / elsize + nbucket) |
| * sizeof (Elf64_Xword)); |
| |
| for (size_t inner = shdr_info[symtabidx].shdr.sh_info; |
| inner < symd->d_size / elsize; ++inner) |
| { |
| GElf_Sym sym_mem; |
| GElf_Sym *sym = gelf_getsym (symd, inner, &sym_mem); |
| assert (sym != NULL); |
| |
| const char *name = elf_strptr (elf, strshndx, |
| sym->st_name); |
| assert (name != NULL); |
| size_t hidx = elf_hash (name) % nbucket; |
| |
| if (bucket[hidx] == 0) |
| bucket[hidx] = inner; |
| else |
| { |
| hidx = bucket[hidx]; |
| |
| while (chain[hidx] != 0) |
| hidx = chain[hidx]; |
| |
| chain[hidx] = inner; |
| } |
| } |
| } |
| |
| /* Convert back to the file byte order. */ |
| if (gelf_xlatetof (newelf, hashd, hashd, |
| BYTE_ORDER == LITTLE_ENDIAN |
| ? ELFDATA2LSB : ELFDATA2MSB) == NULL) |
| INTERNAL_ERROR (fname); |
| } |
| else if (shdr_info[cnt].shdr.sh_type == SHT_GNU_versym) |
| { |
| /* If the symbol table changed we have to adjust the |
| entries. */ |
| Elf32_Word symtabidx = shdr_info[cnt].old_sh_link; |
| |
| /* We do not have to do anything if the symbol table was |
| not changed. */ |
| if (shdr_info[symtabidx].newsymidx == NULL) |
| continue; |
| |
| /* The symbol version section in the new file. */ |
| scn = elf_getscn (newelf, shdr_info[cnt].idx); |
| |
| /* The symbol table data. */ |
| Elf_Data *symd = elf_getdata (elf_getscn (newelf, |
| shdr_info[symtabidx].idx), |
| NULL); |
| assert (symd != NULL); |
| |
| /* The version symbol data. */ |
| Elf_Data *verd = elf_getdata (scn, NULL); |
| assert (verd != NULL); |
| |
| /* Convert to the correct byte order. */ |
| if (gelf_xlatetom (newelf, verd, verd, |
| BYTE_ORDER == LITTLE_ENDIAN |
| ? ELFDATA2LSB : ELFDATA2MSB) == NULL) |
| INTERNAL_ERROR (fname); |
| |
| /* The symbol version array. */ |
| GElf_Half *verstab = (GElf_Half *) verd->d_buf; |
| |
| /* New indices of the symbols. */ |
| Elf32_Word *newsymidx = shdr_info[symtabidx].newsymidx; |
| |
| /* Walk through the list and */ |
| size_t elsize = gelf_fsize (elf, verd->d_type, 1, |
| ehdr->e_version); |
| for (size_t inner = 1; inner < verd->d_size / elsize; ++inner) |
| if (newsymidx[inner] != 0) |
| /* Overwriting the same array works since the |
| reordering can only move entries to lower indices |
| in the array. */ |
| verstab[newsymidx[inner]] = verstab[inner]; |
| |
| /* New size of the section. */ |
| GElf_Shdr shdr_mem; |
| GElf_Shdr *shdr = gelf_getshdr (scn, &shdr_mem); |
| shdr->sh_size = verd->d_size |
| = gelf_fsize (newelf, verd->d_type, |
| symd->d_size / gelf_fsize (elf, symd->d_type, 1, |
| ehdr->e_version), |
| ehdr->e_version); |
| (void) gelf_update_shdr (scn, shdr); |
| |
| /* Convert back to the file byte order. */ |
| if (gelf_xlatetof (newelf, verd, verd, |
| BYTE_ORDER == LITTLE_ENDIAN |
| ? ELFDATA2LSB : ELFDATA2MSB) == NULL) |
| INTERNAL_ERROR (fname); |
| } |
| else if (shdr_info[cnt].shdr.sh_type == SHT_GROUP) |
| { |
| /* Check whether the associated symbol table changed. */ |
| if (shdr_info[shdr_info[cnt].old_sh_link].newsymidx != NULL) |
| { |
| /* Yes the symbol table changed. Update the section |
| header of the section group. */ |
| scn = elf_getscn (newelf, shdr_info[cnt].idx); |
| GElf_Shdr shdr_mem; |
| GElf_Shdr *shdr = gelf_getshdr (scn, &shdr_mem); |
| assert (shdr != NULL); |
| |
| size_t stabidx = shdr_info[cnt].old_sh_link; |
| shdr->sh_info = shdr_info[stabidx].newsymidx[shdr->sh_info]; |
| |
| (void) gelf_update_shdr (scn, shdr); |
| } |
| } |
| } |
| } |
| |
| /* Finally finish the ELF header. Fill in the fields not handled by |
| libelf from the old file. */ |
| newehdr = gelf_getehdr (newelf, &newehdr_mem); |
| if (newehdr == NULL) |
| INTERNAL_ERROR (fname); |
| |
| memcpy (newehdr->e_ident, ehdr->e_ident, EI_NIDENT); |
| newehdr->e_type = ehdr->e_type; |
| newehdr->e_machine = ehdr->e_machine; |
| newehdr->e_version = ehdr->e_version; |
| newehdr->e_entry = ehdr->e_entry; |
| newehdr->e_flags = ehdr->e_flags; |
| newehdr->e_phoff = ehdr->e_phoff; |
| /* We need to position the section header table. */ |
| const size_t offsize = gelf_fsize (elf, ELF_T_OFF, 1, EV_CURRENT); |
| newehdr->e_shoff = ((shdr_info[shdridx].shdr.sh_offset |
| + shdr_info[shdridx].shdr.sh_size + offsize - 1) |
| & ~((GElf_Off) (offsize - 1))); |
| newehdr->e_shentsize = gelf_fsize (elf, ELF_T_SHDR, 1, EV_CURRENT); |
| |
| /* The new section header string table index. */ |
| if (likely (idx < SHN_HIRESERVE) && likely (idx != SHN_XINDEX)) |
| newehdr->e_shstrndx = idx; |
| else |
| { |
| /* The index does not fit in the ELF header field. */ |
| shdr_info[0].scn = elf_getscn (elf, 0); |
| |
| if (gelf_getshdr (shdr_info[0].scn, &shdr_info[0].shdr) == NULL) |
| INTERNAL_ERROR (fname); |
| |
| shdr_info[0].shdr.sh_link = idx; |
| (void) gelf_update_shdr (shdr_info[0].scn, &shdr_info[0].shdr); |
| |
| newehdr->e_shstrndx = SHN_XINDEX; |
| } |
| |
| if (gelf_update_ehdr (newelf, newehdr) == 0) |
| { |
| error (0, 0, gettext ("%s: error while creating ELF header: %s"), |
| fname, elf_errmsg (-1)); |
| return 1; |
| } |
| |
| /* We have everything from the old file. */ |
| if (elf_cntl (elf, ELF_C_FDDONE) != 0) |
| { |
| error (0, 0, gettext ("%s: error while reading the file: %s"), |
| fname, elf_errmsg (-1)); |
| return 1; |
| } |
| |
| /* The ELF library better follows our layout when this is not a |
| relocatable object file. */ |
| elf_flagelf (newelf, ELF_C_SET, |
| (ehdr->e_type != ET_REL ? ELF_F_LAYOUT : 0) |
| | (permissive ? ELF_F_PERMISSIVE : 0)); |
| |
| /* Finally write the file. */ |
| if (elf_update (newelf, ELF_C_WRITE) == -1) |
| { |
| error (0, 0, gettext ("while writing `%s': %s"), |
| fname, elf_errmsg (-1)); |
| result = 1; |
| } |
| |
| fail_close: |
| if (shdr_info != NULL) |
| { |
| /* For some sections we might have created an table to map symbol |
| table indices. */ |
| if (any_symtab_changes) |
| for (cnt = 1; cnt <= shdridx; ++cnt) |
| free (shdr_info[cnt].newsymidx); |
| |
| /* Free the memory. */ |
| if ((shnum + 2) * sizeof (struct shdr_info) > MAX_STACK_ALLOC) |
| free (shdr_info); |
| } |
| |
| /* Free other resources. */ |
| if (shstrtab_data != NULL) |
| free (shstrtab_data->d_buf); |
| if (shst != NULL) |
| ebl_strtabfree (shst); |
| |
| /* That was it. Close the descriptors. */ |
| if (elf_end (newelf) != 0) |
| { |
| error (0, 0, gettext ("error while finishing `%s': %s"), fname, |
| elf_errmsg (-1)); |
| result = 1; |
| } |
| |
| if (debugelf != NULL && elf_end (debugelf) != 0) |
| { |
| error (0, 0, gettext ("error while finishing `%s': %s"), debug_fname, |
| elf_errmsg (-1)); |
| result = 1; |
| } |
| |
| fail: |
| /* Close the EBL backend. */ |
| if (ebl != NULL) |
| ebl_closebackend (ebl); |
| |
| /* Close debug file descriptor, if opened */ |
| if (debug_fd >= 0) |
| { |
| if (tmp_debug_fname != NULL) |
| unlink (tmp_debug_fname); |
| close (debug_fd); |
| } |
| |
| /* If requested, preserve the timestamp. */ |
| if (tvp != NULL) |
| { |
| if (futimes (fd, tvp) != 0) |
| { |
| error (0, errno, gettext ("\ |
| cannot set access and modification date of \"%s\""), |
| output_fname ?: fname); |
| result = 1; |
| } |
| } |
| |
| /* Close the file descriptor if we created a new file. */ |
| if (output_fname != NULL) |
| close (fd); |
| |
| return result; |
| } |
| |
| |
| static int |
| handle_ar (int fd, Elf *elf, const char *prefix, const char *fname, |
| struct timeval tvp[2]) |
| { |
| size_t prefix_len = prefix == NULL ? 0 : strlen (prefix); |
| size_t fname_len = strlen (fname) + 1; |
| char new_prefix[prefix_len + 1 + fname_len]; |
| char *cp = new_prefix; |
| |
| /* Create the full name of the file. */ |
| if (prefix != NULL) |
| { |
| cp = mempcpy (cp, prefix, prefix_len); |
| *cp++ = ':'; |
| } |
| memcpy (cp, fname, fname_len); |
| |
| |
| /* Process all the files contained in the archive. */ |
| Elf *subelf; |
| Elf_Cmd cmd = ELF_C_RDWR; |
| int result = 0; |
| while ((subelf = elf_begin (fd, cmd, elf)) != NULL) |
| { |
| /* The the header for this element. */ |
| Elf_Arhdr *arhdr = elf_getarhdr (subelf); |
| |
| if (elf_kind (subelf) == ELF_K_ELF) |
| result |= handle_elf (fd, subelf, new_prefix, arhdr->ar_name, 0, NULL); |
| else if (elf_kind (subelf) == ELF_K_AR) |
| result |= handle_ar (fd, subelf, new_prefix, arhdr->ar_name, NULL); |
| |
| /* Get next archive element. */ |
| cmd = elf_next (subelf); |
| if (unlikely (elf_end (subelf) != 0)) |
| INTERNAL_ERROR (fname); |
| } |
| |
| if (tvp != NULL) |
| { |
| if (unlikely (futimes (fd, tvp) != 0)) |
| { |
| error (0, errno, gettext ("\ |
| cannot set access and modification date of \"%s\""), fname); |
| result = 1; |
| } |
| } |
| |
| if (unlikely (close (fd) != 0)) |
| error (EXIT_FAILURE, errno, gettext ("while closing `%s'"), fname); |
| |
| return result; |
| } |