1d192f3d3SKai Wang /*- 21de7b4b8SPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 31de7b4b8SPedro F. Giffuni * 4d192f3d3SKai Wang * Copyright (c) 2007 Kai Wang 5d192f3d3SKai Wang * All rights reserved. 6d192f3d3SKai Wang * 7d192f3d3SKai Wang * Redistribution and use in source and binary forms, with or without 8d192f3d3SKai Wang * modification, are permitted provided that the following conditions 9d192f3d3SKai Wang * are met: 10d192f3d3SKai Wang * 1. Redistributions of source code must retain the above copyright 1125f82d56SEd Maste * notice, this list of conditions and the following disclaimer. 12d192f3d3SKai Wang * 2. Redistributions in binary form must reproduce the above copyright 13d192f3d3SKai Wang * notice, this list of conditions and the following disclaimer in the 14d192f3d3SKai Wang * documentation and/or other materials provided with the distribution. 15d192f3d3SKai Wang * 1625f82d56SEd Maste * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 1725f82d56SEd Maste * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 1825f82d56SEd Maste * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 1925f82d56SEd Maste * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 2025f82d56SEd Maste * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 2125f82d56SEd Maste * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2225f82d56SEd Maste * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 2325f82d56SEd Maste * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 2425f82d56SEd Maste * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 2525f82d56SEd Maste * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 2625f82d56SEd Maste * SUCH DAMAGE. 27d192f3d3SKai Wang */ 28d192f3d3SKai Wang 29d192f3d3SKai Wang #include <sys/cdefs.h> 30d192f3d3SKai Wang __FBSDID("$FreeBSD$"); 31d192f3d3SKai Wang 32d192f3d3SKai Wang #include <sys/endian.h> 33d192f3d3SKai Wang #include <sys/mman.h> 34d192f3d3SKai Wang #include <sys/queue.h> 35d192f3d3SKai Wang #include <sys/stat.h> 36d192f3d3SKai Wang #include <archive.h> 37d192f3d3SKai Wang #include <archive_entry.h> 38d192f3d3SKai Wang #include <errno.h> 39d192f3d3SKai Wang #include <fcntl.h> 40d192f3d3SKai Wang #include <gelf.h> 41d192f3d3SKai Wang #include <libgen.h> 42d192f3d3SKai Wang #include <stdio.h> 43d192f3d3SKai Wang #include <stdlib.h> 44d192f3d3SKai Wang #include <string.h> 45d192f3d3SKai Wang #include <sysexits.h> 4601715d8cSEd Maste #include <unistd.h> 47d192f3d3SKai Wang 48d192f3d3SKai Wang #include "ar.h" 49d192f3d3SKai Wang 50d192f3d3SKai Wang #define _ARMAG_LEN 8 /* length of ar magic string */ 51d192f3d3SKai Wang #define _ARHDR_LEN 60 /* length of ar header */ 52d192f3d3SKai Wang #define _INIT_AS_CAP 128 /* initial archive string table size */ 53fabed6b2SEd Maste #define _INIT_SYMOFF_CAP (256*(sizeof(uint64_t))) /* initial so table size */ 54d192f3d3SKai Wang #define _INIT_SYMNAME_CAP 1024 /* initial sn table size */ 55d192f3d3SKai Wang #define _MAXNAMELEN_SVR4 15 /* max member name length in svr4 variant */ 56d192f3d3SKai Wang #define _TRUNCATE_LEN 15 /* number of bytes to keep for member name */ 57d192f3d3SKai Wang 58d192f3d3SKai Wang static void add_to_ar_str_table(struct bsdar *bsdar, const char *name); 59d192f3d3SKai Wang static void add_to_ar_sym_table(struct bsdar *bsdar, const char *name); 60d192f3d3SKai Wang static struct ar_obj *create_obj_from_file(struct bsdar *bsdar, 61d192f3d3SKai Wang const char *name, time_t mtime); 62d192f3d3SKai Wang static void create_symtab_entry(struct bsdar *bsdar, void *maddr, 63d192f3d3SKai Wang size_t size); 64cf933f5bSEd Maste static void free_obj(struct bsdar *bsdar, struct ar_obj *obj); 65d192f3d3SKai Wang static void insert_obj(struct bsdar *bsdar, struct ar_obj *obj, 66d192f3d3SKai Wang struct ar_obj *pos); 6701715d8cSEd Maste static void prefault_buffer(const char *buf, size_t s); 680c099281SKai Wang static void read_objs(struct bsdar *bsdar, const char *archive, 690c099281SKai Wang int checkargv); 70*38911b3cSEd Maste static int write_archive(struct bsdar *bsdar, char mode); 71d192f3d3SKai Wang static void write_cleanup(struct bsdar *bsdar); 72d192f3d3SKai Wang static void write_data(struct bsdar *bsdar, struct archive *a, 73d192f3d3SKai Wang const void *buf, size_t s); 74d192f3d3SKai Wang static void write_objs(struct bsdar *bsdar); 75d192f3d3SKai Wang 76*38911b3cSEd Maste int 77d192f3d3SKai Wang ar_mode_d(struct bsdar *bsdar) 78d192f3d3SKai Wang { 79d192f3d3SKai Wang 80*38911b3cSEd Maste return (write_archive(bsdar, 'd')); 81d192f3d3SKai Wang } 82d192f3d3SKai Wang 83*38911b3cSEd Maste int 84d192f3d3SKai Wang ar_mode_m(struct bsdar *bsdar) 85d192f3d3SKai Wang { 86d192f3d3SKai Wang 87*38911b3cSEd Maste return (write_archive(bsdar, 'm')); 88d192f3d3SKai Wang } 89d192f3d3SKai Wang 90*38911b3cSEd Maste int 91cb0dad38SKai Wang ar_mode_q(struct bsdar *bsdar) 92cb0dad38SKai Wang { 93cb0dad38SKai Wang 94*38911b3cSEd Maste return (write_archive(bsdar, 'q')); 95cb0dad38SKai Wang } 96cb0dad38SKai Wang 97*38911b3cSEd Maste int 98d192f3d3SKai Wang ar_mode_r(struct bsdar *bsdar) 99d192f3d3SKai Wang { 100d192f3d3SKai Wang 101*38911b3cSEd Maste return (write_archive(bsdar, 'r')); 102d192f3d3SKai Wang } 103d192f3d3SKai Wang 104*38911b3cSEd Maste int 105d192f3d3SKai Wang ar_mode_s(struct bsdar *bsdar) 106d192f3d3SKai Wang { 107d192f3d3SKai Wang 108*38911b3cSEd Maste return (write_archive(bsdar, 's')); 109d192f3d3SKai Wang } 110d192f3d3SKai Wang 111*38911b3cSEd Maste int 1120c099281SKai Wang ar_mode_A(struct bsdar *bsdar) 1130c099281SKai Wang { 1140c099281SKai Wang 115*38911b3cSEd Maste return (write_archive(bsdar, 'A')); 1160c099281SKai Wang } 1170c099281SKai Wang 118d192f3d3SKai Wang /* 119d192f3d3SKai Wang * Create object from file, return created obj upon success, or NULL 120d192f3d3SKai Wang * when an error occurs or the member is not newer than existing 1216bef5d28SBenedict Reuschling * one while -u is specified. 122d192f3d3SKai Wang */ 123d192f3d3SKai Wang static struct ar_obj * 124d192f3d3SKai Wang create_obj_from_file(struct bsdar *bsdar, const char *name, time_t mtime) 125d192f3d3SKai Wang { 126d192f3d3SKai Wang struct ar_obj *obj; 127d192f3d3SKai Wang struct stat sb; 128d192f3d3SKai Wang const char *bname; 12909a11349SEd Schouten char *tmpname; 130d192f3d3SKai Wang 131d192f3d3SKai Wang if (name == NULL) 132d192f3d3SKai Wang return (NULL); 133d192f3d3SKai Wang 134d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 135d192f3d3SKai Wang if (obj == NULL) 136d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 137d192f3d3SKai Wang if ((obj->fd = open(name, O_RDONLY, 0)) < 0) { 138d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't open file: %s", name); 139d192f3d3SKai Wang free(obj); 140d192f3d3SKai Wang return (NULL); 141d192f3d3SKai Wang } 142d192f3d3SKai Wang 14309a11349SEd Schouten tmpname = strdup(name); 14409a11349SEd Schouten if (tmpname == NULL) 14509a11349SEd Schouten bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 14609a11349SEd Schouten if ((bname = basename(tmpname)) == NULL) 147d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "basename failed"); 148d192f3d3SKai Wang if (bsdar->options & AR_TR && strlen(bname) > _TRUNCATE_LEN) { 149d192f3d3SKai Wang if ((obj->name = malloc(_TRUNCATE_LEN + 1)) == NULL) 150d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 151d192f3d3SKai Wang (void)strncpy(obj->name, bname, _TRUNCATE_LEN); 152d192f3d3SKai Wang obj->name[_TRUNCATE_LEN] = '\0'; 153d192f3d3SKai Wang } else 154d192f3d3SKai Wang if ((obj->name = strdup(bname)) == NULL) 155d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 15609a11349SEd Schouten free(tmpname); 157d192f3d3SKai Wang 158d192f3d3SKai Wang if (fstat(obj->fd, &sb) < 0) { 159d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't fstat file: %s", obj->name); 160d192f3d3SKai Wang goto giveup; 161d192f3d3SKai Wang } 162d192f3d3SKai Wang if (!S_ISREG(sb.st_mode)) { 163d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s is not an ordinary file", obj->name); 164d192f3d3SKai Wang goto giveup; 165d192f3d3SKai Wang } 166d192f3d3SKai Wang 167d192f3d3SKai Wang /* 168d192f3d3SKai Wang * When option '-u' is specified and member is not newer than the 169d192f3d3SKai Wang * existing one, the replace will not happen. While if mtime == 0, 170d192f3d3SKai Wang * which indicates that this is to "replace a none exist member", 171d192f3d3SKai Wang * the replace will proceed regardless of '-u'. 172d192f3d3SKai Wang */ 173d192f3d3SKai Wang if (mtime != 0 && bsdar->options & AR_U && sb.st_mtime <= mtime) 174d192f3d3SKai Wang goto giveup; 175d192f3d3SKai Wang 1760e479ac8STim Kientzle /* 1770e479ac8STim Kientzle * When option '-D' is specified, mtime and UID / GID from the file 1780e479ac8STim Kientzle * will be replaced with 0, and file mode with 644. This ensures that 1790e479ac8STim Kientzle * checksums will match for two archives containing the exact same 1800e479ac8STim Kientzle * files. 1810e479ac8STim Kientzle */ 1820e479ac8STim Kientzle if (bsdar->options & AR_D) { 1830e479ac8STim Kientzle obj->uid = 0; 1840e479ac8STim Kientzle obj->gid = 0; 1850e479ac8STim Kientzle obj->mtime = 0; 1860e479ac8STim Kientzle obj->md = S_IFREG | 0644; 1870e479ac8STim Kientzle } else { 188d192f3d3SKai Wang obj->uid = sb.st_uid; 189d192f3d3SKai Wang obj->gid = sb.st_gid; 190d192f3d3SKai Wang obj->mtime = sb.st_mtime; 1910e479ac8STim Kientzle obj->md = sb.st_mode; 1920e479ac8STim Kientzle } 1930e479ac8STim Kientzle obj->size = sb.st_size; 194d192f3d3SKai Wang obj->dev = sb.st_dev; 195d192f3d3SKai Wang obj->ino = sb.st_ino; 19647203048SKai Wang 19747203048SKai Wang if (obj->size == 0) { 19847203048SKai Wang obj->maddr = NULL; 19947203048SKai Wang return (obj); 20047203048SKai Wang } 20147203048SKai Wang 202d192f3d3SKai Wang if ((obj->maddr = mmap(NULL, obj->size, PROT_READ, 203d192f3d3SKai Wang MAP_PRIVATE, obj->fd, (off_t)0)) == MAP_FAILED) { 204d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't mmap file: %s", obj->name); 205d192f3d3SKai Wang goto giveup; 206d192f3d3SKai Wang } 207d192f3d3SKai Wang if (close(obj->fd) < 0) 208d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 209d192f3d3SKai Wang obj->name); 210d192f3d3SKai Wang 211d192f3d3SKai Wang return (obj); 212d192f3d3SKai Wang 213d192f3d3SKai Wang giveup: 214d192f3d3SKai Wang if (close(obj->fd) < 0) 215d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 216d192f3d3SKai Wang obj->name); 217d192f3d3SKai Wang free(obj->name); 218d192f3d3SKai Wang free(obj); 219d192f3d3SKai Wang return (NULL); 220d192f3d3SKai Wang } 221d192f3d3SKai Wang 222d192f3d3SKai Wang /* 223cf933f5bSEd Maste * Free object itself and its associated allocations. 224cf933f5bSEd Maste */ 225cf933f5bSEd Maste static void 226cf933f5bSEd Maste free_obj(struct bsdar *bsdar, struct ar_obj *obj) 227cf933f5bSEd Maste { 228cf933f5bSEd Maste if (obj->fd == -1) 229cf933f5bSEd Maste free(obj->maddr); 230cf933f5bSEd Maste else 231cf933f5bSEd Maste if (obj->maddr != NULL && munmap(obj->maddr, obj->size)) 232cf933f5bSEd Maste bsdar_warnc(bsdar, errno, 233cf933f5bSEd Maste "can't munmap file: %s", obj->name); 234cf933f5bSEd Maste free(obj->name); 235cf933f5bSEd Maste free(obj); 236cf933f5bSEd Maste } 237cf933f5bSEd Maste 238cf933f5bSEd Maste /* 239d192f3d3SKai Wang * Insert obj to the tail, or before/after the pos obj. 240d192f3d3SKai Wang */ 241d192f3d3SKai Wang static void 242d192f3d3SKai Wang insert_obj(struct bsdar *bsdar, struct ar_obj *obj, struct ar_obj *pos) 243d192f3d3SKai Wang { 244d192f3d3SKai Wang if (obj == NULL) 245d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "try to insert a null obj"); 246d192f3d3SKai Wang 247d192f3d3SKai Wang if (pos == NULL || obj == pos) 248d192f3d3SKai Wang /* 2496bef5d28SBenedict Reuschling * If the object to move happens to be the position obj, 250d192f3d3SKai Wang * or if there is not a pos obj, move it to tail. 251d192f3d3SKai Wang */ 252d192f3d3SKai Wang goto tail; 253d192f3d3SKai Wang 254d192f3d3SKai Wang if (bsdar->options & AR_B) { 255d192f3d3SKai Wang TAILQ_INSERT_BEFORE(pos, obj, objs); 256d192f3d3SKai Wang return; 257d192f3d3SKai Wang } 258d192f3d3SKai Wang if (bsdar->options & AR_A) { 259d192f3d3SKai Wang TAILQ_INSERT_AFTER(&bsdar->v_obj, pos, obj, objs); 260d192f3d3SKai Wang return; 261d192f3d3SKai Wang } 262d192f3d3SKai Wang 263d192f3d3SKai Wang tail: 264d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 265d192f3d3SKai Wang 266d192f3d3SKai Wang } 267d192f3d3SKai Wang 268d192f3d3SKai Wang /* 2690c099281SKai Wang * Read objects from archive into v_obj list. Note that checkargv is 2700c099281SKai Wang * set when read_objs is used to read objects from the target of 2710c099281SKai Wang * ADDLIB command (ar script mode), in this case argv array possibly 2720c099281SKai Wang * specifies the members ADDLIB want. 273d192f3d3SKai Wang */ 274d192f3d3SKai Wang static void 2750c099281SKai Wang read_objs(struct bsdar *bsdar, const char *archive, int checkargv) 276d192f3d3SKai Wang { 277d192f3d3SKai Wang struct archive *a; 278d192f3d3SKai Wang struct archive_entry *entry; 2790c099281SKai Wang struct ar_obj *obj; 280d192f3d3SKai Wang const char *name; 281d192f3d3SKai Wang const char *bname; 282d192f3d3SKai Wang char *buff; 283d192f3d3SKai Wang char **av; 284d192f3d3SKai Wang size_t size; 2850c099281SKai Wang int i, r, find; 286d192f3d3SKai Wang 287d192f3d3SKai Wang if ((a = archive_read_new()) == NULL) 288d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_read_new failed"); 289d192f3d3SKai Wang archive_read_support_format_ar(a); 2900c099281SKai Wang AC(archive_read_open_filename(a, archive, DEF_BLKSZ)); 291d192f3d3SKai Wang for (;;) { 292d192f3d3SKai Wang r = archive_read_next_header(a, &entry); 293d192f3d3SKai Wang if (r == ARCHIVE_FATAL) 294ccd8660bSEd Maste bsdar_errc(bsdar, EX_DATAERR, archive_errno(a), "%s", 295d192f3d3SKai Wang archive_error_string(a)); 296d192f3d3SKai Wang if (r == ARCHIVE_EOF) 297d192f3d3SKai Wang break; 298d192f3d3SKai Wang if (r == ARCHIVE_WARN || r == ARCHIVE_RETRY) 299ccd8660bSEd Maste bsdar_warnc(bsdar, archive_errno(a), "%s", 300ccd8660bSEd Maste archive_error_string(a)); 301d192f3d3SKai Wang if (r == ARCHIVE_RETRY) { 302d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "Retrying..."); 303d192f3d3SKai Wang continue; 304d192f3d3SKai Wang } 305d192f3d3SKai Wang 306d192f3d3SKai Wang name = archive_entry_pathname(entry); 307d192f3d3SKai Wang 308d192f3d3SKai Wang /* 309d192f3d3SKai Wang * skip pseudo members. 310d192f3d3SKai Wang */ 311d192f3d3SKai Wang if (strcmp(name, "/") == 0 || strcmp(name, "//") == 0) 312d192f3d3SKai Wang continue; 313d192f3d3SKai Wang 3140c099281SKai Wang /* 3150c099281SKai Wang * If checkargv is set, only read those members specified 3160c099281SKai Wang * in argv. 3170c099281SKai Wang */ 3180c099281SKai Wang if (checkargv && bsdar->argc > 0) { 3190c099281SKai Wang find = 0; 3200c099281SKai Wang for(i = 0; i < bsdar->argc; i++) { 3210c099281SKai Wang av = &bsdar->argv[i]; 3220c099281SKai Wang if (*av == NULL) 3230c099281SKai Wang continue; 3240c099281SKai Wang if ((bname = basename(*av)) == NULL) 3250c099281SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 3260c099281SKai Wang "basename failed"); 3270c099281SKai Wang if (strcmp(bname, name) != 0) 3280c099281SKai Wang continue; 3290c099281SKai Wang 3300c099281SKai Wang *av = NULL; 3310c099281SKai Wang find = 1; 3320c099281SKai Wang break; 3330c099281SKai Wang } 3340c099281SKai Wang if (!find) 3350c099281SKai Wang continue; 3360c099281SKai Wang } 3370c099281SKai Wang 338d192f3d3SKai Wang size = archive_entry_size(entry); 339d192f3d3SKai Wang 340517dbe2eSKai Wang if (size > 0) { 341d192f3d3SKai Wang if ((buff = malloc(size)) == NULL) 342517dbe2eSKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 343517dbe2eSKai Wang "malloc failed"); 344d192f3d3SKai Wang if (archive_read_data(a, buff, size) != (ssize_t)size) { 345ccd8660bSEd Maste bsdar_warnc(bsdar, archive_errno(a), "%s", 346517dbe2eSKai Wang archive_error_string(a)); 347d192f3d3SKai Wang free(buff); 348d192f3d3SKai Wang continue; 349d192f3d3SKai Wang } 350517dbe2eSKai Wang } else 351517dbe2eSKai Wang buff = NULL; 352517dbe2eSKai Wang 353d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 354d192f3d3SKai Wang if (obj == NULL) 355d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 356d192f3d3SKai Wang obj->maddr = buff; 357d192f3d3SKai Wang if ((obj->name = strdup(name)) == NULL) 358d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 359d192f3d3SKai Wang obj->size = size; 360d192f3d3SKai Wang obj->uid = archive_entry_uid(entry); 361d192f3d3SKai Wang obj->gid = archive_entry_gid(entry); 362d192f3d3SKai Wang obj->md = archive_entry_mode(entry); 363d192f3d3SKai Wang obj->mtime = archive_entry_mtime(entry); 364d192f3d3SKai Wang obj->dev = 0; 365d192f3d3SKai Wang obj->ino = 0; 366d192f3d3SKai Wang 367d192f3d3SKai Wang /* 368d192f3d3SKai Wang * Objects from archive have obj->fd set to -1, 369d192f3d3SKai Wang * for the ease of cleaning up. 370d192f3d3SKai Wang */ 371d192f3d3SKai Wang obj->fd = -1; 372d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 373d192f3d3SKai Wang } 374d192f3d3SKai Wang AC(archive_read_close(a)); 375ebb8fc42SMartin Matuska AC(archive_read_free(a)); 3760c099281SKai Wang } 3770c099281SKai Wang 3780c099281SKai Wang /* 3790c099281SKai Wang * Determine the constitution of resulting archive. 3800c099281SKai Wang */ 381*38911b3cSEd Maste static int 3820c099281SKai Wang write_archive(struct bsdar *bsdar, char mode) 3830c099281SKai Wang { 3840c099281SKai Wang struct ar_obj *nobj, *obj, *obj_temp, *pos; 3850c099281SKai Wang struct stat sb; 3860c099281SKai Wang const char *bname; 3870c099281SKai Wang char **av; 388*38911b3cSEd Maste int exitcode, i; 3890c099281SKai Wang 3900c099281SKai Wang TAILQ_INIT(&bsdar->v_obj); 391*38911b3cSEd Maste exitcode = EXIT_SUCCESS; 3920c099281SKai Wang nobj = NULL; 3930c099281SKai Wang pos = NULL; 3940c099281SKai Wang memset(&sb, 0, sizeof(sb)); 3950c099281SKai Wang 3960c099281SKai Wang /* 3970c099281SKai Wang * Test if the specified archive exists, to figure out 3980c099281SKai Wang * whether we are creating one here. 3990c099281SKai Wang */ 4000c099281SKai Wang if (stat(bsdar->filename, &sb) != 0) { 4010c099281SKai Wang if (errno != ENOENT) { 4020c099281SKai Wang bsdar_warnc(bsdar, 0, "stat %s failed", 4030c099281SKai Wang bsdar->filename); 404*38911b3cSEd Maste return (EXIT_FAILURE); 4050c099281SKai Wang } 4060c099281SKai Wang 4070c099281SKai Wang /* We do not create archive in mode 'd', 'm' and 's'. */ 4080c099281SKai Wang if (mode != 'r' && mode != 'q') { 4090c099281SKai Wang bsdar_warnc(bsdar, 0, "%s: no such file", 4100c099281SKai Wang bsdar->filename); 411*38911b3cSEd Maste return (EXIT_FAILURE); 4120c099281SKai Wang } 4130c099281SKai Wang 4140c099281SKai Wang /* Issue a warning if -c is not specified when creating. */ 4150c099281SKai Wang if (!(bsdar->options & AR_C)) 4160c099281SKai Wang bsdar_warnc(bsdar, 0, "creating %s", bsdar->filename); 4170c099281SKai Wang goto new_archive; 4180c099281SKai Wang } 4190c099281SKai Wang 4200c099281SKai Wang /* 4210c099281SKai Wang * First read members from existing archive. 4220c099281SKai Wang */ 4230c099281SKai Wang read_objs(bsdar, bsdar->filename, 0); 424d192f3d3SKai Wang 425d192f3d3SKai Wang /* 426d192f3d3SKai Wang * For mode 's', no member will be moved, deleted or replaced. 427d192f3d3SKai Wang */ 428d192f3d3SKai Wang if (mode == 's') 429d192f3d3SKai Wang goto write_objs; 430d192f3d3SKai Wang 431d192f3d3SKai Wang /* 432cb0dad38SKai Wang * For mode 'q', we don't need to adjust existing members either. 433cb0dad38SKai Wang * Also, -a, -b and -i are ignored in this mode. New members are 434cb0dad38SKai Wang * always inserted at tail. 435cb0dad38SKai Wang */ 436cb0dad38SKai Wang if (mode == 'q') 437cb0dad38SKai Wang goto new_archive; 438cb0dad38SKai Wang 439cb0dad38SKai Wang /* 4400c099281SKai Wang * Mode 'A' adds the contents of another archive to the tail of 4410c099281SKai Wang * current archive. Note that mode 'A' is a special mode for the 4420c099281SKai Wang * ADDLIB command of the ar script mode. Currently there is no 4430c099281SKai Wang * access to this function from the ar command line mode. 4440c099281SKai Wang */ 4450c099281SKai Wang if (mode == 'A') { 4460c099281SKai Wang /* 4470c099281SKai Wang * Read objects from the target archive of ADDLIB command. 4486bef5d28SBenedict Reuschling * If there are members specified in argv, read those members 4490c099281SKai Wang * only, otherwise the entire archive will be read. 4500c099281SKai Wang */ 4510c099281SKai Wang read_objs(bsdar, bsdar->addlib, 1); 4520c099281SKai Wang goto write_objs; 4530c099281SKai Wang } 4540c099281SKai Wang 4550c099281SKai Wang /* 456d192f3d3SKai Wang * Try to find the position member specified by user. 457d192f3d3SKai Wang */ 458d192f3d3SKai Wang if (bsdar->options & AR_A || bsdar->options & AR_B) { 459d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 460d192f3d3SKai Wang if (strcmp(obj->name, bsdar->posarg) == 0) { 461d192f3d3SKai Wang pos = obj; 462d192f3d3SKai Wang break; 463d192f3d3SKai Wang } 464d192f3d3SKai Wang } 465d192f3d3SKai Wang 466d192f3d3SKai Wang /* 467d192f3d3SKai Wang * If can't find `pos' specified by user, 4686bef5d28SBenedict Reuschling * silently insert objects at tail. 469d192f3d3SKai Wang */ 470d192f3d3SKai Wang if (pos == NULL) 471d192f3d3SKai Wang bsdar->options &= ~(AR_A | AR_B); 472d192f3d3SKai Wang } 473d192f3d3SKai Wang 474d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 475d192f3d3SKai Wang av = &bsdar->argv[i]; 476d192f3d3SKai Wang 477d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 478d192f3d3SKai Wang if ((bname = basename(*av)) == NULL) 479d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 480d192f3d3SKai Wang "basename failed"); 481d192f3d3SKai Wang if (bsdar->options & AR_TR) { 482d192f3d3SKai Wang if (strncmp(bname, obj->name, _TRUNCATE_LEN)) 483d192f3d3SKai Wang continue; 484d192f3d3SKai Wang } else 485d192f3d3SKai Wang if (strcmp(bname, obj->name) != 0) 486d192f3d3SKai Wang continue; 487d192f3d3SKai Wang 488d192f3d3SKai Wang if (mode == 'r') { 489d192f3d3SKai Wang /* 490d192f3d3SKai Wang * if the new member is not qualified 491d192f3d3SKai Wang * to replace the old one, skip it. 492d192f3d3SKai Wang */ 493d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 494d192f3d3SKai Wang obj->mtime); 495*38911b3cSEd Maste if (nobj == NULL) { 496*38911b3cSEd Maste exitcode = EXIT_FAILURE; 497d192f3d3SKai Wang goto skip_obj; 498d192f3d3SKai Wang } 499*38911b3cSEd Maste } 500d192f3d3SKai Wang 501d192f3d3SKai Wang if (bsdar->options & AR_V) 502d192f3d3SKai Wang (void)fprintf(stdout, "%c - %s\n", mode, 503d192f3d3SKai Wang *av); 504d192f3d3SKai Wang 505d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 506cf933f5bSEd Maste if (mode == 'd' || mode == 'r') 507cf933f5bSEd Maste free_obj(bsdar, obj); 508d192f3d3SKai Wang 509d192f3d3SKai Wang if (mode == 'm') 510d192f3d3SKai Wang insert_obj(bsdar, obj, pos); 511d192f3d3SKai Wang if (mode == 'r') 512d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 513d192f3d3SKai Wang 514d192f3d3SKai Wang skip_obj: 515d192f3d3SKai Wang *av = NULL; 516d192f3d3SKai Wang break; 517d192f3d3SKai Wang } 518d192f3d3SKai Wang 519d192f3d3SKai Wang } 520d192f3d3SKai Wang 521d192f3d3SKai Wang new_archive: 522d192f3d3SKai Wang /* 523d192f3d3SKai Wang * When operating in mode 'r', directly add those user specified 524cb0dad38SKai Wang * objects which do not exist in current archive. When operating 525cb0dad38SKai Wang * in mode 'q', all objects specified in command line args are 526cb0dad38SKai Wang * appended to the archive, without comparing with existing ones. 527d192f3d3SKai Wang */ 528d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 529d192f3d3SKai Wang av = &bsdar->argv[i]; 530cb0dad38SKai Wang if (*av != NULL && (mode == 'r' || mode == 'q')) { 531d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 0); 532*38911b3cSEd Maste if (nobj == NULL) { 533*38911b3cSEd Maste exitcode = EXIT_FAILURE; 534*38911b3cSEd Maste *av = NULL; 535*38911b3cSEd Maste continue; 536*38911b3cSEd Maste } 537d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 538d192f3d3SKai Wang if (bsdar->options & AR_V && nobj != NULL) 539d192f3d3SKai Wang (void)fprintf(stdout, "a - %s\n", *av); 540d192f3d3SKai Wang *av = NULL; 541d192f3d3SKai Wang } 542d192f3d3SKai Wang } 543d192f3d3SKai Wang 544d192f3d3SKai Wang write_objs: 545d192f3d3SKai Wang write_objs(bsdar); 546d192f3d3SKai Wang write_cleanup(bsdar); 547*38911b3cSEd Maste 548*38911b3cSEd Maste return (exitcode); 549d192f3d3SKai Wang } 550d192f3d3SKai Wang 551d192f3d3SKai Wang /* 552d192f3d3SKai Wang * Memory cleaning up. 553d192f3d3SKai Wang */ 554d192f3d3SKai Wang static void 555d192f3d3SKai Wang write_cleanup(struct bsdar *bsdar) 556d192f3d3SKai Wang { 557d192f3d3SKai Wang struct ar_obj *obj, *obj_temp; 558d192f3d3SKai Wang 559d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 560d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 561cf933f5bSEd Maste free_obj(bsdar, obj); 562d192f3d3SKai Wang } 563d192f3d3SKai Wang 564d192f3d3SKai Wang free(bsdar->as); 565d192f3d3SKai Wang free(bsdar->s_so); 566d192f3d3SKai Wang free(bsdar->s_sn); 567d192f3d3SKai Wang bsdar->as = NULL; 568d192f3d3SKai Wang bsdar->s_so = NULL; 569fabed6b2SEd Maste bsdar->s_so_max = 0; 570d192f3d3SKai Wang bsdar->s_sn = NULL; 571d192f3d3SKai Wang } 572d192f3d3SKai Wang 573d192f3d3SKai Wang /* 57401715d8cSEd Maste * Fault in the buffer prior to writing as a workaround for poor performance 57501715d8cSEd Maste * due to interaction with kernel fs deadlock avoidance code. See the comment 57601715d8cSEd Maste * above vn_io_fault_doio() in sys/kern/vfs_vnops.c for details of the issue. 57701715d8cSEd Maste */ 57801715d8cSEd Maste static void 57901715d8cSEd Maste prefault_buffer(const char *buf, size_t s) 58001715d8cSEd Maste { 58101715d8cSEd Maste volatile const char *p; 58201715d8cSEd Maste size_t page_size; 58301715d8cSEd Maste 58401715d8cSEd Maste if (s == 0) 58501715d8cSEd Maste return; 58601715d8cSEd Maste page_size = sysconf(_SC_PAGESIZE); 58701715d8cSEd Maste for (p = buf; p < buf + s; p += page_size) 58801715d8cSEd Maste *p; 58901715d8cSEd Maste /* 59001715d8cSEd Maste * Ensure we touch the last page as well, in case the buffer is not 59101715d8cSEd Maste * page-aligned. 59201715d8cSEd Maste */ 59301715d8cSEd Maste *(volatile const char *)(buf + s - 1); 59401715d8cSEd Maste } 59501715d8cSEd Maste 59601715d8cSEd Maste /* 597d192f3d3SKai Wang * Wrapper for archive_write_data(). 598d192f3d3SKai Wang */ 599d192f3d3SKai Wang static void 600d192f3d3SKai Wang write_data(struct bsdar *bsdar, struct archive *a, const void *buf, size_t s) 601d192f3d3SKai Wang { 6023f71a0c9SEd Maste ssize_t written; 6033f71a0c9SEd Maste 60401715d8cSEd Maste prefault_buffer(buf, s); 6053f71a0c9SEd Maste while (s > 0) { 6063f71a0c9SEd Maste written = archive_write_data(a, buf, s); 6073f71a0c9SEd Maste if (written < 0) 608ccd8660bSEd Maste bsdar_errc(bsdar, EX_SOFTWARE, archive_errno(a), "%s", 609d192f3d3SKai Wang archive_error_string(a)); 6103f71a0c9SEd Maste buf = (const char *)buf + written; 6113f71a0c9SEd Maste s -= written; 6123f71a0c9SEd Maste } 613d192f3d3SKai Wang } 614d192f3d3SKai Wang 615d192f3d3SKai Wang /* 616d192f3d3SKai Wang * Write the resulting archive members. 617d192f3d3SKai Wang */ 618d192f3d3SKai Wang static void 619d192f3d3SKai Wang write_objs(struct bsdar *bsdar) 620d192f3d3SKai Wang { 621d192f3d3SKai Wang struct ar_obj *obj; 622d192f3d3SKai Wang struct archive *a; 623d192f3d3SKai Wang struct archive_entry *entry; 624d192f3d3SKai Wang size_t s_sz; /* size of archive symbol table. */ 625d192f3d3SKai Wang size_t pm_sz; /* size of pseudo members */ 626fabed6b2SEd Maste size_t w_sz; /* size of words in symbol table */ 627597d9b40SEd Maste size_t i; 628fabed6b2SEd Maste uint64_t nr; 6291dffcf9fSEd Maste uint32_t nr32; 630d192f3d3SKai Wang 631d192f3d3SKai Wang if (elf_version(EV_CURRENT) == EV_NONE) 632d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, 633d192f3d3SKai Wang "ELF library initialization failed: %s", elf_errmsg(-1)); 634d192f3d3SKai Wang 635d192f3d3SKai Wang bsdar->rela_off = 0; 636d192f3d3SKai Wang 637d192f3d3SKai Wang /* Create archive symbol table and archive string table, if need. */ 638d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 63947203048SKai Wang if (!(bsdar->options & AR_SS) && obj->maddr != NULL) 640d192f3d3SKai Wang create_symtab_entry(bsdar, obj->maddr, obj->size); 641d192f3d3SKai Wang if (strlen(obj->name) > _MAXNAMELEN_SVR4) 642d192f3d3SKai Wang add_to_ar_str_table(bsdar, obj->name); 643d192f3d3SKai Wang bsdar->rela_off += _ARHDR_LEN + obj->size + obj->size % 2; 644d192f3d3SKai Wang } 645d192f3d3SKai Wang 646d192f3d3SKai Wang /* 647d192f3d3SKai Wang * Pad the symbol name string table. It is treated specially because 648d192f3d3SKai Wang * symbol name table should be padded by a '\0', not the common '\n' 649d192f3d3SKai Wang * for other members. The size of sn table includes the pad bit. 650d192f3d3SKai Wang */ 651d192f3d3SKai Wang if (bsdar->s_cnt != 0 && bsdar->s_sn_sz % 2 != 0) 652d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 653d192f3d3SKai Wang 654d192f3d3SKai Wang /* 655d192f3d3SKai Wang * Archive string table is padded by a "\n" as the normal members. 656517dbe2eSKai Wang * The difference is that the size of archive string table counts 6575666643aSGordon Bergling * in the pad bit, while normal members' size fields do not. 658d192f3d3SKai Wang */ 659d192f3d3SKai Wang if (bsdar->as != NULL && bsdar->as_sz % 2 != 0) 660d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 661d192f3d3SKai Wang 662d192f3d3SKai Wang /* 663d192f3d3SKai Wang * If there is a symbol table, calculate the size of pseudo members, 664517dbe2eSKai Wang * convert previously stored relative offsets to absolute ones, and 665d192f3d3SKai Wang * then make them Big Endian. 666d192f3d3SKai Wang * 667d192f3d3SKai Wang * absolute_offset = htobe32(relative_offset + size_of_pseudo_members) 668d192f3d3SKai Wang */ 669fabed6b2SEd Maste w_sz = sizeof(uint32_t); 670d192f3d3SKai Wang if (bsdar->s_cnt != 0) { 671d192f3d3SKai Wang s_sz = (bsdar->s_cnt + 1) * sizeof(uint32_t) + bsdar->s_sn_sz; 672d192f3d3SKai Wang pm_sz = _ARMAG_LEN + (_ARHDR_LEN + s_sz); 673d192f3d3SKai Wang if (bsdar->as != NULL) 674d192f3d3SKai Wang pm_sz += _ARHDR_LEN + bsdar->as_sz; 675fabed6b2SEd Maste /* Use the 64-bit word size format if necessary. */ 676fabed6b2SEd Maste if (bsdar->s_so_max > UINT32_MAX - pm_sz) { 677fabed6b2SEd Maste w_sz = sizeof(uint64_t); 678fabed6b2SEd Maste pm_sz -= s_sz; 679fabed6b2SEd Maste s_sz = (bsdar->s_cnt + 1) * sizeof(uint64_t) + 680fabed6b2SEd Maste bsdar->s_sn_sz; 681fabed6b2SEd Maste pm_sz += s_sz; 6821dffcf9fSEd Maste /* Convert to big-endian. */ 683597d9b40SEd Maste for (i = 0; i < bsdar->s_cnt; i++) 684687c0904SEd Maste bsdar->s_so[i] = 685687c0904SEd Maste htobe64(bsdar->s_so[i] + pm_sz); 6861dffcf9fSEd Maste } else { 6871dffcf9fSEd Maste /* 6881dffcf9fSEd Maste * Convert to big-endian and shuffle in-place to 6891dffcf9fSEd Maste * the front of the allocation. XXX UB 6901dffcf9fSEd Maste */ 691597d9b40SEd Maste for (i = 0; i < bsdar->s_cnt; i++) 6921dffcf9fSEd Maste ((uint32_t *)(bsdar->s_so))[i] = 6931dffcf9fSEd Maste htobe32(bsdar->s_so[i] + pm_sz); 694d192f3d3SKai Wang } 695699f1801SEd Maste } 696d192f3d3SKai Wang 697d192f3d3SKai Wang if ((a = archive_write_new()) == NULL) 698d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_write_new failed"); 699d192f3d3SKai Wang 700d192f3d3SKai Wang archive_write_set_format_ar_svr4(a); 701d192f3d3SKai Wang 702d192f3d3SKai Wang AC(archive_write_open_filename(a, bsdar->filename)); 703d192f3d3SKai Wang 704d192f3d3SKai Wang /* 705d192f3d3SKai Wang * write the archive symbol table, if there is one. 706d192f3d3SKai Wang * If options -s is explicitly specified or we are invoked 707d192f3d3SKai Wang * as ranlib, write the symbol table even if it is empty. 708d192f3d3SKai Wang */ 709d192f3d3SKai Wang if ((bsdar->s_cnt != 0 && !(bsdar->options & AR_SS)) || 710d192f3d3SKai Wang bsdar->options & AR_S) { 711d192f3d3SKai Wang entry = archive_entry_new(); 712bd9302d8SBryan Drewery if (entry == NULL) 713bd9302d8SBryan Drewery bsdar_errc(bsdar, EX_SOFTWARE, 0, 714bd9302d8SBryan Drewery "archive_entry_new failed"); 715fabed6b2SEd Maste if (w_sz == sizeof(uint64_t)) 716fabed6b2SEd Maste archive_entry_copy_pathname(entry, "/SYM64/"); 717fabed6b2SEd Maste else 718d192f3d3SKai Wang archive_entry_copy_pathname(entry, "/"); 7190e479ac8STim Kientzle if ((bsdar->options & AR_D) == 0) 720d192f3d3SKai Wang archive_entry_set_mtime(entry, time(NULL), 0); 721fabed6b2SEd Maste archive_entry_set_size(entry, (bsdar->s_cnt + 1) * w_sz + 722fabed6b2SEd Maste bsdar->s_sn_sz); 723d192f3d3SKai Wang AC(archive_write_header(a, entry)); 7241dffcf9fSEd Maste if (w_sz == sizeof(uint64_t)) { 725fabed6b2SEd Maste nr = htobe64(bsdar->s_cnt); 7261dffcf9fSEd Maste write_data(bsdar, a, &nr, sizeof(nr)); 7271dffcf9fSEd Maste } else { 7281dffcf9fSEd Maste nr32 = htobe32((uint32_t)bsdar->s_cnt); 7291dffcf9fSEd Maste write_data(bsdar, a, &nr32, sizeof(nr32)); 7301dffcf9fSEd Maste } 7311dffcf9fSEd Maste write_data(bsdar, a, bsdar->s_so, w_sz * bsdar->s_cnt); 732d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_sn, bsdar->s_sn_sz); 733d192f3d3SKai Wang archive_entry_free(entry); 734d192f3d3SKai Wang } 735d192f3d3SKai Wang 736d192f3d3SKai Wang /* write the archive string table, if any. */ 737d192f3d3SKai Wang if (bsdar->as != NULL) { 738d192f3d3SKai Wang entry = archive_entry_new(); 739bd9302d8SBryan Drewery if (entry == NULL) 740bd9302d8SBryan Drewery bsdar_errc(bsdar, EX_SOFTWARE, 0, 741bd9302d8SBryan Drewery "archive_entry_new failed"); 742d192f3d3SKai Wang archive_entry_copy_pathname(entry, "//"); 743d192f3d3SKai Wang archive_entry_set_size(entry, bsdar->as_sz); 744d192f3d3SKai Wang AC(archive_write_header(a, entry)); 745d192f3d3SKai Wang write_data(bsdar, a, bsdar->as, bsdar->as_sz); 746d192f3d3SKai Wang archive_entry_free(entry); 747d192f3d3SKai Wang } 748d192f3d3SKai Wang 749d192f3d3SKai Wang /* write normal members. */ 750d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 751d192f3d3SKai Wang entry = archive_entry_new(); 752bd9302d8SBryan Drewery if (entry == NULL) 753bd9302d8SBryan Drewery bsdar_errc(bsdar, EX_SOFTWARE, 0, 754bd9302d8SBryan Drewery "archive_entry_new failed"); 755d192f3d3SKai Wang archive_entry_copy_pathname(entry, obj->name); 756d192f3d3SKai Wang archive_entry_set_uid(entry, obj->uid); 757d192f3d3SKai Wang archive_entry_set_gid(entry, obj->gid); 758d192f3d3SKai Wang archive_entry_set_mode(entry, obj->md); 759d192f3d3SKai Wang archive_entry_set_size(entry, obj->size); 760d192f3d3SKai Wang archive_entry_set_mtime(entry, obj->mtime, 0); 761d192f3d3SKai Wang archive_entry_set_dev(entry, obj->dev); 762d192f3d3SKai Wang archive_entry_set_ino(entry, obj->ino); 763d192f3d3SKai Wang archive_entry_set_filetype(entry, AE_IFREG); 764d192f3d3SKai Wang AC(archive_write_header(a, entry)); 765d192f3d3SKai Wang write_data(bsdar, a, obj->maddr, obj->size); 766d192f3d3SKai Wang archive_entry_free(entry); 767d192f3d3SKai Wang } 768d192f3d3SKai Wang 769d192f3d3SKai Wang AC(archive_write_close(a)); 770ebb8fc42SMartin Matuska AC(archive_write_free(a)); 771d192f3d3SKai Wang } 772d192f3d3SKai Wang 773d192f3d3SKai Wang /* 774d192f3d3SKai Wang * Extract global symbols from ELF binary members. 775d192f3d3SKai Wang */ 776d192f3d3SKai Wang static void 777d192f3d3SKai Wang create_symtab_entry(struct bsdar *bsdar, void *maddr, size_t size) 778d192f3d3SKai Wang { 779d192f3d3SKai Wang Elf *e; 780d192f3d3SKai Wang Elf_Scn *scn; 781d192f3d3SKai Wang GElf_Shdr shdr; 782d192f3d3SKai Wang GElf_Sym sym; 783d192f3d3SKai Wang Elf_Data *data; 784d192f3d3SKai Wang char *name; 785d192f3d3SKai Wang size_t n, shstrndx; 786d192f3d3SKai Wang int elferr, tabndx, len, i; 787d192f3d3SKai Wang 788d192f3d3SKai Wang if ((e = elf_memory(maddr, size)) == NULL) { 789d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_memory() failed: %s", 790d192f3d3SKai Wang elf_errmsg(-1)); 791d192f3d3SKai Wang return; 792d192f3d3SKai Wang } 793d192f3d3SKai Wang if (elf_kind(e) != ELF_K_ELF) { 7946bef5d28SBenedict Reuschling /* Silently ignore non-elf member. */ 795d192f3d3SKai Wang elf_end(e); 796d192f3d3SKai Wang return; 797d192f3d3SKai Wang } 798d192f3d3SKai Wang if (elf_getshstrndx(e, &shstrndx) == 0) { 799d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_getshstrndx failed: %s", 800d192f3d3SKai Wang elf_errmsg(-1)); 801d192f3d3SKai Wang elf_end(e); 802d192f3d3SKai Wang return; 803d192f3d3SKai Wang } 804d192f3d3SKai Wang 805d192f3d3SKai Wang tabndx = -1; 806d192f3d3SKai Wang scn = NULL; 807d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 808d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 809d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 810d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 811d192f3d3SKai Wang continue; 812d192f3d3SKai Wang } 813d192f3d3SKai Wang if ((name = elf_strptr(e, shstrndx, shdr.sh_name)) == NULL) { 814d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 815d192f3d3SKai Wang "elf_strptr failed: %s", elf_errmsg(-1)); 816d192f3d3SKai Wang continue; 817d192f3d3SKai Wang } 818d192f3d3SKai Wang if (strcmp(name, ".strtab") == 0) { 819d192f3d3SKai Wang tabndx = elf_ndxscn(scn); 820d192f3d3SKai Wang break; 821d192f3d3SKai Wang } 822d192f3d3SKai Wang } 823d192f3d3SKai Wang elferr = elf_errno(); 824d192f3d3SKai Wang if (elferr != 0) 825d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_nextscn failed: %s", 826d192f3d3SKai Wang elf_errmsg(elferr)); 827d192f3d3SKai Wang if (tabndx == -1) { 828d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "can't find .strtab section"); 829d192f3d3SKai Wang elf_end(e); 830d192f3d3SKai Wang return; 831d192f3d3SKai Wang } 832d192f3d3SKai Wang 833d192f3d3SKai Wang scn = NULL; 834d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 835d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 836d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 837d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 838d192f3d3SKai Wang continue; 839d192f3d3SKai Wang } 840d192f3d3SKai Wang if (shdr.sh_type != SHT_SYMTAB) 841d192f3d3SKai Wang continue; 842d192f3d3SKai Wang 843d192f3d3SKai Wang data = NULL; 844d192f3d3SKai Wang n = 0; 845d192f3d3SKai Wang while (n < shdr.sh_size && 846d192f3d3SKai Wang (data = elf_getdata(scn, data)) != NULL) { 847d192f3d3SKai Wang len = data->d_size / shdr.sh_entsize; 848d192f3d3SKai Wang for (i = 0; i < len; i++) { 849d192f3d3SKai Wang if (gelf_getsym(data, i, &sym) != &sym) { 850d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 851d192f3d3SKai Wang "gelf_getsym failed: %s", 852d192f3d3SKai Wang elf_errmsg(-1)); 853d192f3d3SKai Wang continue; 854d192f3d3SKai Wang } 855d192f3d3SKai Wang 856d192f3d3SKai Wang /* keep only global or weak symbols */ 857d192f3d3SKai Wang if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL && 858d192f3d3SKai Wang GELF_ST_BIND(sym.st_info) != STB_WEAK) 859d192f3d3SKai Wang continue; 860d192f3d3SKai Wang 861d192f3d3SKai Wang /* keep only defined symbols */ 862d192f3d3SKai Wang if (sym.st_shndx == SHN_UNDEF) 863d192f3d3SKai Wang continue; 864d192f3d3SKai Wang 865d192f3d3SKai Wang if ((name = elf_strptr(e, tabndx, 866d192f3d3SKai Wang sym.st_name)) == NULL) { 867d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 868d192f3d3SKai Wang "elf_strptr failed: %s", 869d192f3d3SKai Wang elf_errmsg(-1)); 870d192f3d3SKai Wang continue; 871d192f3d3SKai Wang } 872d192f3d3SKai Wang 873d192f3d3SKai Wang add_to_ar_sym_table(bsdar, name); 874d192f3d3SKai Wang } 875d192f3d3SKai Wang } 876d192f3d3SKai Wang } 877d192f3d3SKai Wang elferr = elf_errno(); 878d192f3d3SKai Wang if (elferr != 0) 879d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_nextscn failed: %s", 880d192f3d3SKai Wang elf_errmsg(elferr)); 881d192f3d3SKai Wang 882d192f3d3SKai Wang elf_end(e); 883d192f3d3SKai Wang } 884d192f3d3SKai Wang 885d192f3d3SKai Wang /* 886d192f3d3SKai Wang * Append to the archive string table buffer. 887d192f3d3SKai Wang */ 888d192f3d3SKai Wang static void 889d192f3d3SKai Wang add_to_ar_str_table(struct bsdar *bsdar, const char *name) 890d192f3d3SKai Wang { 891d192f3d3SKai Wang 892d192f3d3SKai Wang if (bsdar->as == NULL) { 893d192f3d3SKai Wang bsdar->as_cap = _INIT_AS_CAP; 894d192f3d3SKai Wang bsdar->as_sz = 0; 895d192f3d3SKai Wang if ((bsdar->as = malloc(bsdar->as_cap)) == NULL) 896d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 897d192f3d3SKai Wang } 898d192f3d3SKai Wang 899d192f3d3SKai Wang /* 900d192f3d3SKai Wang * The space required for holding one member name in as table includes: 901d192f3d3SKai Wang * strlen(name) + (1 for '/') + (1 for '\n') + (possibly 1 for padding). 902d192f3d3SKai Wang */ 903517dbe2eSKai Wang while (bsdar->as_sz + strlen(name) + 3 > bsdar->as_cap) { 904d192f3d3SKai Wang bsdar->as_cap *= 2; 905d192f3d3SKai Wang bsdar->as = realloc(bsdar->as, bsdar->as_cap); 906d192f3d3SKai Wang if (bsdar->as == NULL) 907d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 908d192f3d3SKai Wang } 909d192f3d3SKai Wang strncpy(&bsdar->as[bsdar->as_sz], name, strlen(name)); 910d192f3d3SKai Wang bsdar->as_sz += strlen(name); 911d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '/'; 912d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 913d192f3d3SKai Wang } 914d192f3d3SKai Wang 915d192f3d3SKai Wang /* 916d192f3d3SKai Wang * Append to the archive symbol table buffer. 917d192f3d3SKai Wang */ 918d192f3d3SKai Wang static void 919d192f3d3SKai Wang add_to_ar_sym_table(struct bsdar *bsdar, const char *name) 920d192f3d3SKai Wang { 921d192f3d3SKai Wang 922d192f3d3SKai Wang if (bsdar->s_so == NULL) { 923d192f3d3SKai Wang if ((bsdar->s_so = malloc(_INIT_SYMOFF_CAP)) == 924d192f3d3SKai Wang NULL) 925d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 926d192f3d3SKai Wang bsdar->s_so_cap = _INIT_SYMOFF_CAP; 927d192f3d3SKai Wang bsdar->s_cnt = 0; 928d192f3d3SKai Wang } 929d192f3d3SKai Wang 930d192f3d3SKai Wang if (bsdar->s_sn == NULL) { 931d192f3d3SKai Wang if ((bsdar->s_sn = malloc(_INIT_SYMNAME_CAP)) == NULL) 932d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 933d192f3d3SKai Wang bsdar->s_sn_cap = _INIT_SYMNAME_CAP; 934d192f3d3SKai Wang bsdar->s_sn_sz = 0; 935d192f3d3SKai Wang } 936d192f3d3SKai Wang 937fabed6b2SEd Maste if (bsdar->s_cnt * sizeof(uint64_t) >= bsdar->s_so_cap) { 938d192f3d3SKai Wang bsdar->s_so_cap *= 2; 939d192f3d3SKai Wang bsdar->s_so = realloc(bsdar->s_so, bsdar->s_so_cap); 940d192f3d3SKai Wang if (bsdar->s_so == NULL) 941d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 942d192f3d3SKai Wang } 943d192f3d3SKai Wang bsdar->s_so[bsdar->s_cnt] = bsdar->rela_off; 944fabed6b2SEd Maste if ((uint64_t)bsdar->rela_off > bsdar->s_so_max) 945fabed6b2SEd Maste bsdar->s_so_max = (uint64_t)bsdar->rela_off; 946d192f3d3SKai Wang bsdar->s_cnt++; 947d192f3d3SKai Wang 948d192f3d3SKai Wang /* 949d192f3d3SKai Wang * The space required for holding one symbol name in sn table includes: 950d192f3d3SKai Wang * strlen(name) + (1 for '\n') + (possibly 1 for padding). 951d192f3d3SKai Wang */ 952517dbe2eSKai Wang while (bsdar->s_sn_sz + strlen(name) + 2 > bsdar->s_sn_cap) { 953d192f3d3SKai Wang bsdar->s_sn_cap *= 2; 954d192f3d3SKai Wang bsdar->s_sn = realloc(bsdar->s_sn, bsdar->s_sn_cap); 955d192f3d3SKai Wang if (bsdar->s_sn == NULL) 956d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 957d192f3d3SKai Wang } 958d192f3d3SKai Wang strncpy(&bsdar->s_sn[bsdar->s_sn_sz], name, strlen(name)); 959d192f3d3SKai Wang bsdar->s_sn_sz += strlen(name); 960d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 961d192f3d3SKai Wang } 962