1cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Generate an index to speed access to archives. 2cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Copyright (C) 2005, 2006, 2007 Red Hat, Inc. 3cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng This file is part of Red Hat elfutils. 4cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Written by Ulrich Drepper <drepper@redhat.com>, 2005. 5cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 6cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Red Hat elfutils is free software; you can redistribute it and/or modify 7cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng it under the terms of the GNU General Public License as published by the 8cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Free Software Foundation; version 2 of the License. 9cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 10cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Red Hat elfutils is distributed in the hope that it will be useful, but 11cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng WITHOUT ANY WARRANTY; without even the implied warranty of 12cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 13cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng General Public License for more details. 14cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 15cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng You should have received a copy of the GNU General Public License along 16cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng with Red Hat elfutils; if not, write to the Free Software Foundation, 17cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301 USA. 18cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 19cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Red Hat elfutils is an included package of the Open Invention Network. 20cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng An included package of the Open Invention Network is a package for which 21cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Open Invention Network licensees cross-license their patents. No patent 22cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng license is granted, either expressly or impliedly, by designation as an 23cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng included package. Should you wish to participate in the Open Invention 24cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Network licensing program, please visit www.openinventionnetwork.com 25cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng <http://www.openinventionnetwork.com>. */ 26cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 27cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#ifdef HAVE_CONFIG_H 28cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng# include <config.h> 29cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#endif 30cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 31cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <ar.h> 32cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <argp.h> 33cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <assert.h> 34cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <errno.h> 35cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <error.h> 36cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <fcntl.h> 37cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <gelf.h> 38cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <libintl.h> 39cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <locale.h> 40cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <mcheck.h> 41cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <obstack.h> 42cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <stdlib.h> 43cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <stdio.h> 44cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <stdio_ext.h> 45cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <unistd.h> 46cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <sys/mman.h> 47cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <sys/param.h> 48cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <sys/stat.h> 49cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 50cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include <system.h> 51cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 52cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include "arlib.h" 53cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 54cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 55cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Prototypes for local functions. */ 56cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengstatic int handle_file (const char *fname); 57cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 58cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 59cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Name and version of program. */ 60cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengstatic void print_version (FILE *stream, struct argp_state *state); 61cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengvoid (*argp_program_version_hook) (FILE *, struct argp_state *) = print_version; 62cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 63cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Bug report address. */ 64cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengconst char *argp_program_bug_address = PACKAGE_BUGREPORT; 65cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 66cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 67cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Definitions of arguments for argp functions. */ 68cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengstatic const struct argp_option options[] = 69cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng{ 70cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { NULL, 0, NULL, 0, NULL, 0 } 71cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng}; 72cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 73cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Short description of program. */ 74cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengstatic const char doc[] = N_("Generate an index to speed access to archives."); 75cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 76cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Strings for arguments in help texts. */ 77cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengstatic const char args_doc[] = N_("ARCHIVE"); 78cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 79cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Data structure to communicate with argp functions. */ 80cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengstatic const struct argp argp = 81cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng{ 82cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng options, NULL, args_doc, doc, NULL, NULL, NULL 83cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng}; 84cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 85cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 86cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengint 87cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengmain (int argc, char *argv[]) 88cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng{ 89cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Make memory leak detection possible. */ 90cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng mtrace (); 91cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 92cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* We use no threads here which can interfere with handling a stream. */ 93cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng (void) __fsetlocking (stdin, FSETLOCKING_BYCALLER); 94cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng (void) __fsetlocking (stdout, FSETLOCKING_BYCALLER); 95cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng (void) __fsetlocking (stderr, FSETLOCKING_BYCALLER); 96cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 97cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Set locale. */ 98cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng (void) setlocale (LC_ALL, ""); 99cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 100cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Make sure the message catalog can be found. */ 101cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng (void) bindtextdomain (PACKAGE_TARNAME, LOCALEDIR); 102cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 103cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Initialize the message catalog. */ 104cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng (void) textdomain (PACKAGE_TARNAME); 105cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 106cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Parse and process arguments. */ 107cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng int remaining; 108cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng (void) argp_parse (&argp, argc, argv, ARGP_IN_ORDER, &remaining, NULL); 109cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 110cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Tell the library which version we are expecting. */ 111cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng (void) elf_version (EV_CURRENT); 112cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 113cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* There must at least be one more parameter specifying the archive. */ 114cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (remaining == argc) 115cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 116cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng error (0, 0, gettext ("Archive name required")); 117cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng argp_help (&argp, stderr, ARGP_HELP_SEE, "ranlib"); 118cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng exit (EXIT_FAILURE); 119cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 120cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 121cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* We accept the names of multiple archives. */ 122cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng int status = 0; 123cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng do 124cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng status |= handle_file (argv[remaining]); 125cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng while (++remaining < argc); 126cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 127cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng return status; 128cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng} 129cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 130cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 131cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Print the version information. */ 132cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengstatic void 133cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengprint_version (FILE *stream, struct argp_state *state __attribute__ ((unused))) 134cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng{ 135cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng fprintf (stream, "ranlib (%s) %s\n", PACKAGE_NAME, PACKAGE_VERSION); 136cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng fprintf (stream, gettext ("\ 137cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben ChengCopyright (C) %s Red Hat, Inc.\n\ 138cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben ChengThis is free software; see the source for copying conditions. There is NO\n\ 139cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengwarranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n\ 140cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng"), "2008"); 141cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng fprintf (stream, gettext ("Written by %s.\n"), "Ulrich Drepper"); 142cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng} 143cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 144cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 145cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengstatic int 146cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengcopy_content (Elf *elf, int newfd, off_t off, size_t n) 147cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng{ 148cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng size_t len; 149cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng char *rawfile = elf_rawfile (elf, &len); 150cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 151cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng assert (off + n <= len); 152cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 153cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Tell the kernel we will read all the pages sequentially. */ 154cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng size_t ps = sysconf (_SC_PAGESIZE); 155cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (n > 2 * ps) 156cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng posix_madvise (rawfile + (off & ~(ps - 1)), n, POSIX_MADV_SEQUENTIAL); 157cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 158cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng return write_retry (newfd, rawfile + off, n) != (ssize_t) n; 159cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng} 160cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 161cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 162cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng/* Handle a file given on the command line. */ 163cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chengstatic int 164cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Chenghandle_file (const char *fname) 165cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng{ 166cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng int fd = open (fname, O_RDONLY); 167cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (fd == -1) 168cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 169cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng error (0, errno, gettext ("cannot open '%s'"), fname); 170cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng return 1; 171cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 172cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 173cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng struct stat st; 174cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (fstat (fd, &st) != 0) 175cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 176cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng error (0, errno, gettext ("cannot stat '%s'"), fname); 177cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng close (fd); 178cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng return 1; 179cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 180cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 181cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* First we walk through the file, looking for all ELF files to 182cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng collect symbols from. */ 183cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Elf *arelf = elf_begin (fd, ELF_C_READ_MMAP, NULL); 184cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (arelf == NULL) 185cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 186cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng error (0, 0, gettext ("cannot create ELF descriptor for '%s': %s"), 187cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng fname, elf_errmsg (-1)); 188cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng close (fd); 189cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng return 1; 190cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 191cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 192cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (elf_kind (arelf) != ELF_K_AR) 193cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 194cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng error (0, 0, gettext ("'%s' is no archive"), fname); 195cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng elf_end (arelf); 196cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng close (fd); 197cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng return 1; 198cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 199cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 200cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng arlib_init (); 201cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 202cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Iterate over the content of the archive. */ 203cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng off_t index_off = -1; 204cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng size_t index_size = 0; 205cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng off_t cur_off = SARMAG; 206cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Elf *elf; 207cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Elf_Cmd cmd = ELF_C_READ_MMAP; 208cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng while ((elf = elf_begin (fd, cmd, arelf)) != NULL) 209cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 210cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng Elf_Arhdr *arhdr = elf_getarhdr (elf); 211cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng assert (arhdr != NULL); 212cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 213cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* If this is the index, remember the location. */ 214cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (strcmp (arhdr->ar_name, "/") == 0) 215cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 216cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng index_off = elf_getaroff (elf); 217cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng index_size = arhdr->ar_size; 218cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 219cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng else 220cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 221cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng arlib_add_symbols (elf, fname, arhdr->ar_name, cur_off); 222cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng cur_off += (((arhdr->ar_size + 1) & ~((off_t) 1)) 223cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng + sizeof (struct ar_hdr)); 224cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 225cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 226cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Get next archive element. */ 227cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng cmd = elf_next (elf); 228cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (elf_end (elf) != 0) 229cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng error (0, 0, gettext ("error while freeing sub-ELF descriptor: %s"), 230cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng elf_errmsg (-1)); 231cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 232cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 233cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng arlib_finalize (); 234cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 235cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* If the file contains no symbols we need not do anything. */ 236cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng int status = 0; 237cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (symtab.symsnamelen != 0 238cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* We have to rewrite the file also if it initially had an index 239cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng but now does not need one anymore. */ 240cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng || (symtab.symsnamelen == 0 && index_size != 0)) 241cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 242cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Create a new, temporary file in the same directory as the 243cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng original file. */ 244cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng char tmpfname[strlen (fname) + 7]; 245cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng strcpy (stpcpy (tmpfname, fname), "XXXXXX"); 246cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng int newfd = mkstemp (tmpfname); 247cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (unlikely (newfd == -1)) 248cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 249cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng nonew: 250cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng error (0, errno, gettext ("cannot create new file")); 251cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng status = 1; 252cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 253cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng else 254cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 255cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Create the header. */ 256cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (unlikely (write_retry (newfd, ARMAG, SARMAG) != SARMAG)) 257cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng { 258cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng // XXX Use /prof/self/fd/%d ??? 259cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng nonew_unlink: 260cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng unlink (tmpfname); 261cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (newfd != -1) 262cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng close (newfd); 263cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng goto nonew; 264cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 265cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 266cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Create the new file. There are three parts as far we are 267cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng concerned: 1. original context before the index, 2. the 268cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng new index, 3. everything after the new index. */ 269cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng off_t rest_off; 270cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if (index_off != -1) 271cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng rest_off = (index_off + sizeof (struct ar_hdr) 272cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng + ((index_size + 1) & ~1ul)); 273cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng else 274cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng rest_off = SARMAG; 275cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 276cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng if ((symtab.symsnamelen != 0 277cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng && ((write_retry (newfd, symtab.symsoff, 278cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng symtab.symsofflen) 279cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng != (ssize_t) symtab.symsofflen) 280cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng || (write_retry (newfd, symtab.symsname, 281cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng symtab.symsnamelen) 282cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng != (ssize_t) symtab.symsnamelen))) 283cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Even if the original file had content before the 284cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng symbol table, we write it in the correct order. */ 285cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng || (index_off > SARMAG 286cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng && copy_content (arelf, newfd, SARMAG, index_off - SARMAG)) 287cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng || copy_content (arelf, newfd, rest_off, st.st_size - rest_off) 288cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Set the mode of the new file to the same values the 289cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng original file has. */ 290cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng || fchmod (newfd, st.st_mode & ALLPERMS) != 0 291cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng /* Never complain about fchown failing. */ 292cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng || (({asm ("" :: "r" (fchown (newfd, st.st_uid, st.st_gid))); }), 293cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng close (newfd) != 0) 294cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng || (newfd = -1, rename (tmpfname, fname) != 0)) 295cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng goto nonew_unlink; 296cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 297cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng } 298cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 299cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng elf_end (arelf); 300cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 301cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng arlib_fini (); 302cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 303cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng close (fd); 304cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 305cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng return status; 306cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng} 307cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 308cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng 309cc6695e2684ce93cdf8bd2da63d55d2cf49ff076Ben Cheng#include "debugpred.h" 310