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); 70d192f3d3SKai Wang static void 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 76d192f3d3SKai Wang void 77d192f3d3SKai Wang ar_mode_d(struct bsdar *bsdar) 78d192f3d3SKai Wang { 79d192f3d3SKai Wang 80d192f3d3SKai Wang write_archive(bsdar, 'd'); 81d192f3d3SKai Wang } 82d192f3d3SKai Wang 83d192f3d3SKai Wang void 84d192f3d3SKai Wang ar_mode_m(struct bsdar *bsdar) 85d192f3d3SKai Wang { 86d192f3d3SKai Wang 87d192f3d3SKai Wang write_archive(bsdar, 'm'); 88d192f3d3SKai Wang } 89d192f3d3SKai Wang 90d192f3d3SKai Wang void 91cb0dad38SKai Wang ar_mode_q(struct bsdar *bsdar) 92cb0dad38SKai Wang { 93cb0dad38SKai Wang 94cb0dad38SKai Wang write_archive(bsdar, 'q'); 95cb0dad38SKai Wang } 96cb0dad38SKai Wang 97cb0dad38SKai Wang void 98d192f3d3SKai Wang ar_mode_r(struct bsdar *bsdar) 99d192f3d3SKai Wang { 100d192f3d3SKai Wang 101d192f3d3SKai Wang write_archive(bsdar, 'r'); 102d192f3d3SKai Wang } 103d192f3d3SKai Wang 104d192f3d3SKai Wang void 105d192f3d3SKai Wang ar_mode_s(struct bsdar *bsdar) 106d192f3d3SKai Wang { 107d192f3d3SKai Wang 108d192f3d3SKai Wang write_archive(bsdar, 's'); 109d192f3d3SKai Wang } 110d192f3d3SKai Wang 1110c099281SKai Wang void 1120c099281SKai Wang ar_mode_A(struct bsdar *bsdar) 1130c099281SKai Wang { 1140c099281SKai Wang 1150c099281SKai Wang 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 */ 3810c099281SKai Wang static void 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; 3880c099281SKai Wang int i; 3890c099281SKai Wang 3900c099281SKai Wang TAILQ_INIT(&bsdar->v_obj); 3910c099281SKai Wang nobj = NULL; 3920c099281SKai Wang pos = NULL; 3930c099281SKai Wang memset(&sb, 0, sizeof(sb)); 3940c099281SKai Wang 3950c099281SKai Wang /* 3960c099281SKai Wang * Test if the specified archive exists, to figure out 3970c099281SKai Wang * whether we are creating one here. 3980c099281SKai Wang */ 3990c099281SKai Wang if (stat(bsdar->filename, &sb) != 0) { 4000c099281SKai Wang if (errno != ENOENT) { 4010c099281SKai Wang bsdar_warnc(bsdar, 0, "stat %s failed", 4020c099281SKai Wang bsdar->filename); 4030c099281SKai Wang return; 4040c099281SKai Wang } 4050c099281SKai Wang 4060c099281SKai Wang /* We do not create archive in mode 'd', 'm' and 's'. */ 4070c099281SKai Wang if (mode != 'r' && mode != 'q') { 4080c099281SKai Wang bsdar_warnc(bsdar, 0, "%s: no such file", 4090c099281SKai Wang bsdar->filename); 4100c099281SKai Wang return; 4110c099281SKai Wang } 4120c099281SKai Wang 4130c099281SKai Wang /* Issue a warning if -c is not specified when creating. */ 4140c099281SKai Wang if (!(bsdar->options & AR_C)) 4150c099281SKai Wang bsdar_warnc(bsdar, 0, "creating %s", bsdar->filename); 4160c099281SKai Wang goto new_archive; 4170c099281SKai Wang } 4180c099281SKai Wang 4190c099281SKai Wang /* 4200c099281SKai Wang * First read members from existing archive. 4210c099281SKai Wang */ 4220c099281SKai Wang read_objs(bsdar, bsdar->filename, 0); 423d192f3d3SKai Wang 424d192f3d3SKai Wang /* 425d192f3d3SKai Wang * For mode 's', no member will be moved, deleted or replaced. 426d192f3d3SKai Wang */ 427d192f3d3SKai Wang if (mode == 's') 428d192f3d3SKai Wang goto write_objs; 429d192f3d3SKai Wang 430d192f3d3SKai Wang /* 431cb0dad38SKai Wang * For mode 'q', we don't need to adjust existing members either. 432cb0dad38SKai Wang * Also, -a, -b and -i are ignored in this mode. New members are 433cb0dad38SKai Wang * always inserted at tail. 434cb0dad38SKai Wang */ 435cb0dad38SKai Wang if (mode == 'q') 436cb0dad38SKai Wang goto new_archive; 437cb0dad38SKai Wang 438cb0dad38SKai Wang /* 4390c099281SKai Wang * Mode 'A' adds the contents of another archive to the tail of 4400c099281SKai Wang * current archive. Note that mode 'A' is a special mode for the 4410c099281SKai Wang * ADDLIB command of the ar script mode. Currently there is no 4420c099281SKai Wang * access to this function from the ar command line mode. 4430c099281SKai Wang */ 4440c099281SKai Wang if (mode == 'A') { 4450c099281SKai Wang /* 4460c099281SKai Wang * Read objects from the target archive of ADDLIB command. 4476bef5d28SBenedict Reuschling * If there are members specified in argv, read those members 4480c099281SKai Wang * only, otherwise the entire archive will be read. 4490c099281SKai Wang */ 4500c099281SKai Wang read_objs(bsdar, bsdar->addlib, 1); 4510c099281SKai Wang goto write_objs; 4520c099281SKai Wang } 4530c099281SKai Wang 4540c099281SKai Wang /* 455d192f3d3SKai Wang * Try to find the position member specified by user. 456d192f3d3SKai Wang */ 457d192f3d3SKai Wang if (bsdar->options & AR_A || bsdar->options & AR_B) { 458d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 459d192f3d3SKai Wang if (strcmp(obj->name, bsdar->posarg) == 0) { 460d192f3d3SKai Wang pos = obj; 461d192f3d3SKai Wang break; 462d192f3d3SKai Wang } 463d192f3d3SKai Wang } 464d192f3d3SKai Wang 465d192f3d3SKai Wang /* 466d192f3d3SKai Wang * If can't find `pos' specified by user, 4676bef5d28SBenedict Reuschling * silently insert objects at tail. 468d192f3d3SKai Wang */ 469d192f3d3SKai Wang if (pos == NULL) 470d192f3d3SKai Wang bsdar->options &= ~(AR_A | AR_B); 471d192f3d3SKai Wang } 472d192f3d3SKai Wang 473d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 474d192f3d3SKai Wang av = &bsdar->argv[i]; 475d192f3d3SKai Wang 476d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 477d192f3d3SKai Wang if ((bname = basename(*av)) == NULL) 478d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 479d192f3d3SKai Wang "basename failed"); 480d192f3d3SKai Wang if (bsdar->options & AR_TR) { 481d192f3d3SKai Wang if (strncmp(bname, obj->name, _TRUNCATE_LEN)) 482d192f3d3SKai Wang continue; 483d192f3d3SKai Wang } else 484d192f3d3SKai Wang if (strcmp(bname, obj->name) != 0) 485d192f3d3SKai Wang continue; 486d192f3d3SKai Wang 487d192f3d3SKai Wang if (mode == 'r') { 488d192f3d3SKai Wang /* 489d192f3d3SKai Wang * if the new member is not qualified 490d192f3d3SKai Wang * to replace the old one, skip it. 491d192f3d3SKai Wang */ 492d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 493d192f3d3SKai Wang obj->mtime); 494d192f3d3SKai Wang if (nobj == NULL) 495d192f3d3SKai Wang goto skip_obj; 496d192f3d3SKai Wang } 497d192f3d3SKai Wang 498d192f3d3SKai Wang if (bsdar->options & AR_V) 499d192f3d3SKai Wang (void)fprintf(stdout, "%c - %s\n", mode, 500d192f3d3SKai Wang *av); 501d192f3d3SKai Wang 502d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 503cf933f5bSEd Maste if (mode == 'd' || mode == 'r') 504cf933f5bSEd Maste free_obj(bsdar, obj); 505d192f3d3SKai Wang 506d192f3d3SKai Wang if (mode == 'm') 507d192f3d3SKai Wang insert_obj(bsdar, obj, pos); 508d192f3d3SKai Wang if (mode == 'r') 509d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 510d192f3d3SKai Wang 511d192f3d3SKai Wang skip_obj: 512d192f3d3SKai Wang *av = NULL; 513d192f3d3SKai Wang break; 514d192f3d3SKai Wang } 515d192f3d3SKai Wang 516d192f3d3SKai Wang } 517d192f3d3SKai Wang 518d192f3d3SKai Wang new_archive: 519d192f3d3SKai Wang /* 520d192f3d3SKai Wang * When operating in mode 'r', directly add those user specified 521cb0dad38SKai Wang * objects which do not exist in current archive. When operating 522cb0dad38SKai Wang * in mode 'q', all objects specified in command line args are 523cb0dad38SKai Wang * appended to the archive, without comparing with existing ones. 524d192f3d3SKai Wang */ 525d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 526d192f3d3SKai Wang av = &bsdar->argv[i]; 527cb0dad38SKai Wang if (*av != NULL && (mode == 'r' || mode == 'q')) { 528d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 0); 529d192f3d3SKai Wang if (nobj != NULL) 530d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 531d192f3d3SKai Wang if (bsdar->options & AR_V && nobj != NULL) 532d192f3d3SKai Wang (void)fprintf(stdout, "a - %s\n", *av); 533d192f3d3SKai Wang *av = NULL; 534d192f3d3SKai Wang } 535d192f3d3SKai Wang } 536d192f3d3SKai Wang 537d192f3d3SKai Wang write_objs: 538d192f3d3SKai Wang write_objs(bsdar); 539d192f3d3SKai Wang write_cleanup(bsdar); 540d192f3d3SKai Wang } 541d192f3d3SKai Wang 542d192f3d3SKai Wang /* 543d192f3d3SKai Wang * Memory cleaning up. 544d192f3d3SKai Wang */ 545d192f3d3SKai Wang static void 546d192f3d3SKai Wang write_cleanup(struct bsdar *bsdar) 547d192f3d3SKai Wang { 548d192f3d3SKai Wang struct ar_obj *obj, *obj_temp; 549d192f3d3SKai Wang 550d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 551d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 552cf933f5bSEd Maste free_obj(bsdar, obj); 553d192f3d3SKai Wang } 554d192f3d3SKai Wang 555d192f3d3SKai Wang free(bsdar->as); 556d192f3d3SKai Wang free(bsdar->s_so); 557d192f3d3SKai Wang free(bsdar->s_sn); 558d192f3d3SKai Wang bsdar->as = NULL; 559d192f3d3SKai Wang bsdar->s_so = NULL; 560fabed6b2SEd Maste bsdar->s_so_max = 0; 561d192f3d3SKai Wang bsdar->s_sn = NULL; 562d192f3d3SKai Wang } 563d192f3d3SKai Wang 564d192f3d3SKai Wang /* 56501715d8cSEd Maste * Fault in the buffer prior to writing as a workaround for poor performance 56601715d8cSEd Maste * due to interaction with kernel fs deadlock avoidance code. See the comment 56701715d8cSEd Maste * above vn_io_fault_doio() in sys/kern/vfs_vnops.c for details of the issue. 56801715d8cSEd Maste */ 56901715d8cSEd Maste static void 57001715d8cSEd Maste prefault_buffer(const char *buf, size_t s) 57101715d8cSEd Maste { 57201715d8cSEd Maste volatile const char *p; 57301715d8cSEd Maste size_t page_size; 57401715d8cSEd Maste 57501715d8cSEd Maste if (s == 0) 57601715d8cSEd Maste return; 57701715d8cSEd Maste page_size = sysconf(_SC_PAGESIZE); 57801715d8cSEd Maste for (p = buf; p < buf + s; p += page_size) 57901715d8cSEd Maste *p; 58001715d8cSEd Maste /* 58101715d8cSEd Maste * Ensure we touch the last page as well, in case the buffer is not 58201715d8cSEd Maste * page-aligned. 58301715d8cSEd Maste */ 58401715d8cSEd Maste *(volatile const char *)(buf + s - 1); 58501715d8cSEd Maste } 58601715d8cSEd Maste 58701715d8cSEd Maste /* 588d192f3d3SKai Wang * Wrapper for archive_write_data(). 589d192f3d3SKai Wang */ 590d192f3d3SKai Wang static void 591d192f3d3SKai Wang write_data(struct bsdar *bsdar, struct archive *a, const void *buf, size_t s) 592d192f3d3SKai Wang { 5933f71a0c9SEd Maste ssize_t written; 5943f71a0c9SEd Maste 59501715d8cSEd Maste prefault_buffer(buf, s); 5963f71a0c9SEd Maste while (s > 0) { 5973f71a0c9SEd Maste written = archive_write_data(a, buf, s); 5983f71a0c9SEd Maste if (written < 0) 599ccd8660bSEd Maste bsdar_errc(bsdar, EX_SOFTWARE, archive_errno(a), "%s", 600d192f3d3SKai Wang archive_error_string(a)); 6013f71a0c9SEd Maste buf = (const char *)buf + written; 6023f71a0c9SEd Maste s -= written; 6033f71a0c9SEd Maste } 604d192f3d3SKai Wang } 605d192f3d3SKai Wang 606d192f3d3SKai Wang /* 607d192f3d3SKai Wang * Write the resulting archive members. 608d192f3d3SKai Wang */ 609d192f3d3SKai Wang static void 610d192f3d3SKai Wang write_objs(struct bsdar *bsdar) 611d192f3d3SKai Wang { 612d192f3d3SKai Wang struct ar_obj *obj; 613d192f3d3SKai Wang struct archive *a; 614d192f3d3SKai Wang struct archive_entry *entry; 615d192f3d3SKai Wang size_t s_sz; /* size of archive symbol table. */ 616d192f3d3SKai Wang size_t pm_sz; /* size of pseudo members */ 617fabed6b2SEd Maste size_t w_sz; /* size of words in symbol table */ 618fabed6b2SEd Maste uint64_t nr; 619fabed6b2SEd Maste int i; 620d192f3d3SKai Wang 621d192f3d3SKai Wang if (elf_version(EV_CURRENT) == EV_NONE) 622d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, 623d192f3d3SKai Wang "ELF library initialization failed: %s", elf_errmsg(-1)); 624d192f3d3SKai Wang 625d192f3d3SKai Wang bsdar->rela_off = 0; 626d192f3d3SKai Wang 627d192f3d3SKai Wang /* Create archive symbol table and archive string table, if need. */ 628d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 62947203048SKai Wang if (!(bsdar->options & AR_SS) && obj->maddr != NULL) 630d192f3d3SKai Wang create_symtab_entry(bsdar, obj->maddr, obj->size); 631d192f3d3SKai Wang if (strlen(obj->name) > _MAXNAMELEN_SVR4) 632d192f3d3SKai Wang add_to_ar_str_table(bsdar, obj->name); 633d192f3d3SKai Wang bsdar->rela_off += _ARHDR_LEN + obj->size + obj->size % 2; 634d192f3d3SKai Wang } 635d192f3d3SKai Wang 636d192f3d3SKai Wang /* 637d192f3d3SKai Wang * Pad the symbol name string table. It is treated specially because 638d192f3d3SKai Wang * symbol name table should be padded by a '\0', not the common '\n' 639d192f3d3SKai Wang * for other members. The size of sn table includes the pad bit. 640d192f3d3SKai Wang */ 641d192f3d3SKai Wang if (bsdar->s_cnt != 0 && bsdar->s_sn_sz % 2 != 0) 642d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 643d192f3d3SKai Wang 644d192f3d3SKai Wang /* 645d192f3d3SKai Wang * Archive string table is padded by a "\n" as the normal members. 646517dbe2eSKai Wang * The difference is that the size of archive string table counts 647d192f3d3SKai Wang * in the pad bit, while normal members' size fileds do not. 648d192f3d3SKai Wang */ 649d192f3d3SKai Wang if (bsdar->as != NULL && bsdar->as_sz % 2 != 0) 650d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 651d192f3d3SKai Wang 652d192f3d3SKai Wang /* 653d192f3d3SKai Wang * If there is a symbol table, calculate the size of pseudo members, 654517dbe2eSKai Wang * convert previously stored relative offsets to absolute ones, and 655d192f3d3SKai Wang * then make them Big Endian. 656d192f3d3SKai Wang * 657d192f3d3SKai Wang * absolute_offset = htobe32(relative_offset + size_of_pseudo_members) 658d192f3d3SKai Wang */ 659d192f3d3SKai Wang 660fabed6b2SEd Maste w_sz = sizeof(uint32_t); 661fabed6b2SEd Maste if (bsdar->s_so_max > UINT32_MAX) 662fabed6b2SEd Maste w_sz = sizeof(uint64_t); 663d192f3d3SKai Wang if (bsdar->s_cnt != 0) { 664d192f3d3SKai Wang s_sz = (bsdar->s_cnt + 1) * sizeof(uint32_t) + bsdar->s_sn_sz; 665d192f3d3SKai Wang pm_sz = _ARMAG_LEN + (_ARHDR_LEN + s_sz); 666d192f3d3SKai Wang if (bsdar->as != NULL) 667d192f3d3SKai Wang pm_sz += _ARHDR_LEN + bsdar->as_sz; 668fabed6b2SEd Maste /* Use the 64-bit word size format if necessary. */ 669fabed6b2SEd Maste if (bsdar->s_so_max > UINT32_MAX - pm_sz) { 670fabed6b2SEd Maste w_sz = sizeof(uint64_t); 671fabed6b2SEd Maste pm_sz -= s_sz; 672fabed6b2SEd Maste s_sz = (bsdar->s_cnt + 1) * sizeof(uint64_t) + 673fabed6b2SEd Maste bsdar->s_sn_sz; 674fabed6b2SEd Maste pm_sz += s_sz; 675fabed6b2SEd Maste } 676fabed6b2SEd Maste 677699f1801SEd Maste for (i = 0; (size_t)i < bsdar->s_cnt; i++) { 678fabed6b2SEd Maste if (w_sz == sizeof(uint32_t)) 679fabed6b2SEd Maste *(bsdar->s_so + i) = 680fabed6b2SEd Maste htobe32((uint32_t)(*(bsdar->s_so + i)) + pm_sz); 681fabed6b2SEd Maste else 682fabed6b2SEd Maste *(bsdar->s_so + i) = 683fabed6b2SEd Maste htobe64(*(bsdar->s_so + i) + pm_sz); 684d192f3d3SKai Wang } 685699f1801SEd Maste } 686d192f3d3SKai Wang 687d192f3d3SKai Wang if ((a = archive_write_new()) == NULL) 688d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_write_new failed"); 689d192f3d3SKai Wang 690d192f3d3SKai Wang archive_write_set_format_ar_svr4(a); 691d192f3d3SKai Wang 692d192f3d3SKai Wang AC(archive_write_open_filename(a, bsdar->filename)); 693d192f3d3SKai Wang 694d192f3d3SKai Wang /* 695d192f3d3SKai Wang * write the archive symbol table, if there is one. 696d192f3d3SKai Wang * If options -s is explicitly specified or we are invoked 697d192f3d3SKai Wang * as ranlib, write the symbol table even if it is empty. 698d192f3d3SKai Wang */ 699d192f3d3SKai Wang if ((bsdar->s_cnt != 0 && !(bsdar->options & AR_SS)) || 700d192f3d3SKai Wang bsdar->options & AR_S) { 701d192f3d3SKai Wang entry = archive_entry_new(); 702bd9302d8SBryan Drewery if (entry == NULL) 703bd9302d8SBryan Drewery bsdar_errc(bsdar, EX_SOFTWARE, 0, 704bd9302d8SBryan Drewery "archive_entry_new failed"); 705fabed6b2SEd Maste if (w_sz == sizeof(uint64_t)) 706fabed6b2SEd Maste archive_entry_copy_pathname(entry, "/SYM64/"); 707fabed6b2SEd Maste else 708d192f3d3SKai Wang archive_entry_copy_pathname(entry, "/"); 7090e479ac8STim Kientzle if ((bsdar->options & AR_D) == 0) 710d192f3d3SKai Wang archive_entry_set_mtime(entry, time(NULL), 0); 711fabed6b2SEd Maste archive_entry_set_size(entry, (bsdar->s_cnt + 1) * w_sz + 712fabed6b2SEd Maste bsdar->s_sn_sz); 713d192f3d3SKai Wang AC(archive_write_header(a, entry)); 714fabed6b2SEd Maste if (w_sz == sizeof(uint32_t)) 715fabed6b2SEd Maste nr = (uint64_t)htobe32((uint32_t)bsdar->s_cnt); 716fabed6b2SEd Maste else 717fabed6b2SEd Maste nr = htobe64(bsdar->s_cnt); 718fabed6b2SEd Maste write_data(bsdar, a, &nr, w_sz); 719fabed6b2SEd Maste if (w_sz == sizeof(uint64_t)) 720fabed6b2SEd Maste write_data(bsdar, a, bsdar->s_so, sizeof(uint64_t) * 721d192f3d3SKai Wang bsdar->s_cnt); 722fabed6b2SEd Maste else 723fabed6b2SEd Maste for (i = 0; (size_t)i < bsdar->s_cnt; i++) 724fabed6b2SEd Maste write_data(bsdar, a, 725fabed6b2SEd Maste (uint32_t *)&bsdar->s_so[i], 726fabed6b2SEd Maste sizeof(uint32_t)); 727d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_sn, bsdar->s_sn_sz); 728d192f3d3SKai Wang archive_entry_free(entry); 729d192f3d3SKai Wang } 730d192f3d3SKai Wang 731d192f3d3SKai Wang /* write the archive string table, if any. */ 732d192f3d3SKai Wang if (bsdar->as != NULL) { 733d192f3d3SKai Wang entry = archive_entry_new(); 734bd9302d8SBryan Drewery if (entry == NULL) 735bd9302d8SBryan Drewery bsdar_errc(bsdar, EX_SOFTWARE, 0, 736bd9302d8SBryan Drewery "archive_entry_new failed"); 737d192f3d3SKai Wang archive_entry_copy_pathname(entry, "//"); 738d192f3d3SKai Wang archive_entry_set_size(entry, bsdar->as_sz); 739d192f3d3SKai Wang AC(archive_write_header(a, entry)); 740d192f3d3SKai Wang write_data(bsdar, a, bsdar->as, bsdar->as_sz); 741d192f3d3SKai Wang archive_entry_free(entry); 742d192f3d3SKai Wang } 743d192f3d3SKai Wang 744d192f3d3SKai Wang /* write normal members. */ 745d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 746d192f3d3SKai Wang entry = archive_entry_new(); 747bd9302d8SBryan Drewery if (entry == NULL) 748bd9302d8SBryan Drewery bsdar_errc(bsdar, EX_SOFTWARE, 0, 749bd9302d8SBryan Drewery "archive_entry_new failed"); 750d192f3d3SKai Wang archive_entry_copy_pathname(entry, obj->name); 751d192f3d3SKai Wang archive_entry_set_uid(entry, obj->uid); 752d192f3d3SKai Wang archive_entry_set_gid(entry, obj->gid); 753d192f3d3SKai Wang archive_entry_set_mode(entry, obj->md); 754d192f3d3SKai Wang archive_entry_set_size(entry, obj->size); 755d192f3d3SKai Wang archive_entry_set_mtime(entry, obj->mtime, 0); 756d192f3d3SKai Wang archive_entry_set_dev(entry, obj->dev); 757d192f3d3SKai Wang archive_entry_set_ino(entry, obj->ino); 758d192f3d3SKai Wang archive_entry_set_filetype(entry, AE_IFREG); 759d192f3d3SKai Wang AC(archive_write_header(a, entry)); 760d192f3d3SKai Wang write_data(bsdar, a, obj->maddr, obj->size); 761d192f3d3SKai Wang archive_entry_free(entry); 762d192f3d3SKai Wang } 763d192f3d3SKai Wang 764d192f3d3SKai Wang AC(archive_write_close(a)); 765ebb8fc42SMartin Matuska AC(archive_write_free(a)); 766d192f3d3SKai Wang } 767d192f3d3SKai Wang 768d192f3d3SKai Wang /* 769d192f3d3SKai Wang * Extract global symbols from ELF binary members. 770d192f3d3SKai Wang */ 771d192f3d3SKai Wang static void 772d192f3d3SKai Wang create_symtab_entry(struct bsdar *bsdar, void *maddr, size_t size) 773d192f3d3SKai Wang { 774d192f3d3SKai Wang Elf *e; 775d192f3d3SKai Wang Elf_Scn *scn; 776d192f3d3SKai Wang GElf_Shdr shdr; 777d192f3d3SKai Wang GElf_Sym sym; 778d192f3d3SKai Wang Elf_Data *data; 779d192f3d3SKai Wang char *name; 780d192f3d3SKai Wang size_t n, shstrndx; 781d192f3d3SKai Wang int elferr, tabndx, len, i; 782d192f3d3SKai Wang 783d192f3d3SKai Wang if ((e = elf_memory(maddr, size)) == NULL) { 784d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_memory() failed: %s", 785d192f3d3SKai Wang elf_errmsg(-1)); 786d192f3d3SKai Wang return; 787d192f3d3SKai Wang } 788d192f3d3SKai Wang if (elf_kind(e) != ELF_K_ELF) { 7896bef5d28SBenedict Reuschling /* Silently ignore non-elf member. */ 790d192f3d3SKai Wang elf_end(e); 791d192f3d3SKai Wang return; 792d192f3d3SKai Wang } 793d192f3d3SKai Wang if (elf_getshstrndx(e, &shstrndx) == 0) { 794d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_getshstrndx failed: %s", 795d192f3d3SKai Wang elf_errmsg(-1)); 796d192f3d3SKai Wang elf_end(e); 797d192f3d3SKai Wang return; 798d192f3d3SKai Wang } 799d192f3d3SKai Wang 800d192f3d3SKai Wang tabndx = -1; 801d192f3d3SKai Wang scn = NULL; 802d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 803d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 804d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 805d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 806d192f3d3SKai Wang continue; 807d192f3d3SKai Wang } 808d192f3d3SKai Wang if ((name = elf_strptr(e, shstrndx, shdr.sh_name)) == NULL) { 809d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 810d192f3d3SKai Wang "elf_strptr failed: %s", elf_errmsg(-1)); 811d192f3d3SKai Wang continue; 812d192f3d3SKai Wang } 813d192f3d3SKai Wang if (strcmp(name, ".strtab") == 0) { 814d192f3d3SKai Wang tabndx = elf_ndxscn(scn); 815d192f3d3SKai Wang break; 816d192f3d3SKai Wang } 817d192f3d3SKai Wang } 818d192f3d3SKai Wang elferr = elf_errno(); 819d192f3d3SKai Wang if (elferr != 0) 820d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_nextscn failed: %s", 821d192f3d3SKai Wang elf_errmsg(elferr)); 822d192f3d3SKai Wang if (tabndx == -1) { 823d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "can't find .strtab section"); 824d192f3d3SKai Wang elf_end(e); 825d192f3d3SKai Wang return; 826d192f3d3SKai Wang } 827d192f3d3SKai Wang 828d192f3d3SKai Wang scn = NULL; 829d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 830d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 831d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 832d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 833d192f3d3SKai Wang continue; 834d192f3d3SKai Wang } 835d192f3d3SKai Wang if (shdr.sh_type != SHT_SYMTAB) 836d192f3d3SKai Wang continue; 837d192f3d3SKai Wang 838d192f3d3SKai Wang data = NULL; 839d192f3d3SKai Wang n = 0; 840d192f3d3SKai Wang while (n < shdr.sh_size && 841d192f3d3SKai Wang (data = elf_getdata(scn, data)) != NULL) { 842d192f3d3SKai Wang len = data->d_size / shdr.sh_entsize; 843d192f3d3SKai Wang for (i = 0; i < len; i++) { 844d192f3d3SKai Wang if (gelf_getsym(data, i, &sym) != &sym) { 845d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 846d192f3d3SKai Wang "gelf_getsym failed: %s", 847d192f3d3SKai Wang elf_errmsg(-1)); 848d192f3d3SKai Wang continue; 849d192f3d3SKai Wang } 850d192f3d3SKai Wang 851d192f3d3SKai Wang /* keep only global or weak symbols */ 852d192f3d3SKai Wang if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL && 853d192f3d3SKai Wang GELF_ST_BIND(sym.st_info) != STB_WEAK) 854d192f3d3SKai Wang continue; 855d192f3d3SKai Wang 856d192f3d3SKai Wang /* keep only defined symbols */ 857d192f3d3SKai Wang if (sym.st_shndx == SHN_UNDEF) 858d192f3d3SKai Wang continue; 859d192f3d3SKai Wang 860d192f3d3SKai Wang if ((name = elf_strptr(e, tabndx, 861d192f3d3SKai Wang sym.st_name)) == NULL) { 862d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 863d192f3d3SKai Wang "elf_strptr failed: %s", 864d192f3d3SKai Wang elf_errmsg(-1)); 865d192f3d3SKai Wang continue; 866d192f3d3SKai Wang } 867d192f3d3SKai Wang 868d192f3d3SKai Wang add_to_ar_sym_table(bsdar, name); 869d192f3d3SKai Wang } 870d192f3d3SKai Wang } 871d192f3d3SKai Wang } 872d192f3d3SKai Wang elferr = elf_errno(); 873d192f3d3SKai Wang if (elferr != 0) 874d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_nextscn failed: %s", 875d192f3d3SKai Wang elf_errmsg(elferr)); 876d192f3d3SKai Wang 877d192f3d3SKai Wang elf_end(e); 878d192f3d3SKai Wang } 879d192f3d3SKai Wang 880d192f3d3SKai Wang /* 881d192f3d3SKai Wang * Append to the archive string table buffer. 882d192f3d3SKai Wang */ 883d192f3d3SKai Wang static void 884d192f3d3SKai Wang add_to_ar_str_table(struct bsdar *bsdar, const char *name) 885d192f3d3SKai Wang { 886d192f3d3SKai Wang 887d192f3d3SKai Wang if (bsdar->as == NULL) { 888d192f3d3SKai Wang bsdar->as_cap = _INIT_AS_CAP; 889d192f3d3SKai Wang bsdar->as_sz = 0; 890d192f3d3SKai Wang if ((bsdar->as = malloc(bsdar->as_cap)) == NULL) 891d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 892d192f3d3SKai Wang } 893d192f3d3SKai Wang 894d192f3d3SKai Wang /* 895d192f3d3SKai Wang * The space required for holding one member name in as table includes: 896d192f3d3SKai Wang * strlen(name) + (1 for '/') + (1 for '\n') + (possibly 1 for padding). 897d192f3d3SKai Wang */ 898517dbe2eSKai Wang while (bsdar->as_sz + strlen(name) + 3 > bsdar->as_cap) { 899d192f3d3SKai Wang bsdar->as_cap *= 2; 900d192f3d3SKai Wang bsdar->as = realloc(bsdar->as, bsdar->as_cap); 901d192f3d3SKai Wang if (bsdar->as == NULL) 902d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 903d192f3d3SKai Wang } 904d192f3d3SKai Wang strncpy(&bsdar->as[bsdar->as_sz], name, strlen(name)); 905d192f3d3SKai Wang bsdar->as_sz += strlen(name); 906d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '/'; 907d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 908d192f3d3SKai Wang } 909d192f3d3SKai Wang 910d192f3d3SKai Wang /* 911d192f3d3SKai Wang * Append to the archive symbol table buffer. 912d192f3d3SKai Wang */ 913d192f3d3SKai Wang static void 914d192f3d3SKai Wang add_to_ar_sym_table(struct bsdar *bsdar, const char *name) 915d192f3d3SKai Wang { 916d192f3d3SKai Wang 917d192f3d3SKai Wang if (bsdar->s_so == NULL) { 918d192f3d3SKai Wang if ((bsdar->s_so = malloc(_INIT_SYMOFF_CAP)) == 919d192f3d3SKai Wang NULL) 920d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 921d192f3d3SKai Wang bsdar->s_so_cap = _INIT_SYMOFF_CAP; 922d192f3d3SKai Wang bsdar->s_cnt = 0; 923d192f3d3SKai Wang } 924d192f3d3SKai Wang 925d192f3d3SKai Wang if (bsdar->s_sn == NULL) { 926d192f3d3SKai Wang if ((bsdar->s_sn = malloc(_INIT_SYMNAME_CAP)) == NULL) 927d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 928d192f3d3SKai Wang bsdar->s_sn_cap = _INIT_SYMNAME_CAP; 929d192f3d3SKai Wang bsdar->s_sn_sz = 0; 930d192f3d3SKai Wang } 931d192f3d3SKai Wang 932fabed6b2SEd Maste if (bsdar->s_cnt * sizeof(uint64_t) >= bsdar->s_so_cap) { 933d192f3d3SKai Wang bsdar->s_so_cap *= 2; 934d192f3d3SKai Wang bsdar->s_so = realloc(bsdar->s_so, bsdar->s_so_cap); 935d192f3d3SKai Wang if (bsdar->s_so == NULL) 936d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 937d192f3d3SKai Wang } 938d192f3d3SKai Wang bsdar->s_so[bsdar->s_cnt] = bsdar->rela_off; 939fabed6b2SEd Maste if ((uint64_t)bsdar->rela_off > bsdar->s_so_max) 940fabed6b2SEd Maste bsdar->s_so_max = (uint64_t)bsdar->rela_off; 941d192f3d3SKai Wang bsdar->s_cnt++; 942d192f3d3SKai Wang 943d192f3d3SKai Wang /* 944d192f3d3SKai Wang * The space required for holding one symbol name in sn table includes: 945d192f3d3SKai Wang * strlen(name) + (1 for '\n') + (possibly 1 for padding). 946d192f3d3SKai Wang */ 947517dbe2eSKai Wang while (bsdar->s_sn_sz + strlen(name) + 2 > bsdar->s_sn_cap) { 948d192f3d3SKai Wang bsdar->s_sn_cap *= 2; 949d192f3d3SKai Wang bsdar->s_sn = realloc(bsdar->s_sn, bsdar->s_sn_cap); 950d192f3d3SKai Wang if (bsdar->s_sn == NULL) 951d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 952d192f3d3SKai Wang } 953d192f3d3SKai Wang strncpy(&bsdar->s_sn[bsdar->s_sn_sz], name, strlen(name)); 954d192f3d3SKai Wang bsdar->s_sn_sz += strlen(name); 955d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 956d192f3d3SKai Wang } 957