1d192f3d3SKai Wang /*- 2d192f3d3SKai Wang * Copyright (c) 2007 Kai Wang 3d192f3d3SKai Wang * All rights reserved. 4d192f3d3SKai Wang * 5d192f3d3SKai Wang * Redistribution and use in source and binary forms, with or without 6d192f3d3SKai Wang * modification, are permitted provided that the following conditions 7d192f3d3SKai Wang * are met: 8d192f3d3SKai Wang * 1. Redistributions of source code must retain the above copyright 9d192f3d3SKai Wang * notice, this list of conditions and the following disclaimer 10d192f3d3SKai Wang * in this position and unchanged. 11d192f3d3SKai Wang * 2. Redistributions in binary form must reproduce the above copyright 12d192f3d3SKai Wang * notice, this list of conditions and the following disclaimer in the 13d192f3d3SKai Wang * documentation and/or other materials provided with the distribution. 14d192f3d3SKai Wang * 15d192f3d3SKai Wang * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 16d192f3d3SKai Wang * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17d192f3d3SKai Wang * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18d192f3d3SKai Wang * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 19d192f3d3SKai Wang * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20d192f3d3SKai Wang * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21d192f3d3SKai Wang * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22d192f3d3SKai Wang * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23d192f3d3SKai Wang * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24d192f3d3SKai Wang * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25d192f3d3SKai Wang */ 26d192f3d3SKai Wang 27d192f3d3SKai Wang #include <sys/cdefs.h> 28d192f3d3SKai Wang __FBSDID("$FreeBSD$"); 29d192f3d3SKai Wang 30d192f3d3SKai Wang #include <sys/endian.h> 31d192f3d3SKai Wang #include <sys/mman.h> 32d192f3d3SKai Wang #include <sys/queue.h> 33d192f3d3SKai Wang #include <sys/stat.h> 34d192f3d3SKai Wang #include <archive.h> 35d192f3d3SKai Wang #include <archive_entry.h> 36d192f3d3SKai Wang #include <errno.h> 37d192f3d3SKai Wang #include <fcntl.h> 38d192f3d3SKai Wang #include <gelf.h> 39d192f3d3SKai Wang #include <libgen.h> 40d192f3d3SKai Wang #include <stdio.h> 41d192f3d3SKai Wang #include <stdlib.h> 42d192f3d3SKai Wang #include <string.h> 43d192f3d3SKai Wang #include <sysexits.h> 4401715d8cSEd Maste #include <unistd.h> 45d192f3d3SKai Wang 46d192f3d3SKai Wang #include "ar.h" 47d192f3d3SKai Wang 48d192f3d3SKai Wang #define _ARMAG_LEN 8 /* length of ar magic string */ 49d192f3d3SKai Wang #define _ARHDR_LEN 60 /* length of ar header */ 50d192f3d3SKai Wang #define _INIT_AS_CAP 128 /* initial archive string table size */ 51d192f3d3SKai Wang #define _INIT_SYMOFF_CAP (256*(sizeof(uint32_t))) /* initial so table size */ 52d192f3d3SKai Wang #define _INIT_SYMNAME_CAP 1024 /* initial sn table size */ 53d192f3d3SKai Wang #define _MAXNAMELEN_SVR4 15 /* max member name length in svr4 variant */ 54d192f3d3SKai Wang #define _TRUNCATE_LEN 15 /* number of bytes to keep for member name */ 55d192f3d3SKai Wang 56d192f3d3SKai Wang static void add_to_ar_str_table(struct bsdar *bsdar, const char *name); 57d192f3d3SKai Wang static void add_to_ar_sym_table(struct bsdar *bsdar, const char *name); 58d192f3d3SKai Wang static struct ar_obj *create_obj_from_file(struct bsdar *bsdar, 59d192f3d3SKai Wang const char *name, time_t mtime); 60d192f3d3SKai Wang static void create_symtab_entry(struct bsdar *bsdar, void *maddr, 61d192f3d3SKai Wang size_t size); 62cf933f5bSEd Maste static void free_obj(struct bsdar *bsdar, struct ar_obj *obj); 63d192f3d3SKai Wang static void insert_obj(struct bsdar *bsdar, struct ar_obj *obj, 64d192f3d3SKai Wang struct ar_obj *pos); 6501715d8cSEd Maste static void prefault_buffer(const char *buf, size_t s); 660c099281SKai Wang static void read_objs(struct bsdar *bsdar, const char *archive, 670c099281SKai Wang int checkargv); 68d192f3d3SKai Wang static void write_archive(struct bsdar *bsdar, char mode); 69d192f3d3SKai Wang static void write_cleanup(struct bsdar *bsdar); 70d192f3d3SKai Wang static void write_data(struct bsdar *bsdar, struct archive *a, 71d192f3d3SKai Wang const void *buf, size_t s); 72d192f3d3SKai Wang static void write_objs(struct bsdar *bsdar); 73d192f3d3SKai Wang 74d192f3d3SKai Wang void 75d192f3d3SKai Wang ar_mode_d(struct bsdar *bsdar) 76d192f3d3SKai Wang { 77d192f3d3SKai Wang 78d192f3d3SKai Wang write_archive(bsdar, 'd'); 79d192f3d3SKai Wang } 80d192f3d3SKai Wang 81d192f3d3SKai Wang void 82d192f3d3SKai Wang ar_mode_m(struct bsdar *bsdar) 83d192f3d3SKai Wang { 84d192f3d3SKai Wang 85d192f3d3SKai Wang write_archive(bsdar, 'm'); 86d192f3d3SKai Wang } 87d192f3d3SKai Wang 88d192f3d3SKai Wang void 89cb0dad38SKai Wang ar_mode_q(struct bsdar *bsdar) 90cb0dad38SKai Wang { 91cb0dad38SKai Wang 92cb0dad38SKai Wang write_archive(bsdar, 'q'); 93cb0dad38SKai Wang } 94cb0dad38SKai Wang 95cb0dad38SKai Wang void 96d192f3d3SKai Wang ar_mode_r(struct bsdar *bsdar) 97d192f3d3SKai Wang { 98d192f3d3SKai Wang 99d192f3d3SKai Wang write_archive(bsdar, 'r'); 100d192f3d3SKai Wang } 101d192f3d3SKai Wang 102d192f3d3SKai Wang void 103d192f3d3SKai Wang ar_mode_s(struct bsdar *bsdar) 104d192f3d3SKai Wang { 105d192f3d3SKai Wang 106d192f3d3SKai Wang write_archive(bsdar, 's'); 107d192f3d3SKai Wang } 108d192f3d3SKai Wang 1090c099281SKai Wang void 1100c099281SKai Wang ar_mode_A(struct bsdar *bsdar) 1110c099281SKai Wang { 1120c099281SKai Wang 1130c099281SKai Wang write_archive(bsdar, 'A'); 1140c099281SKai Wang } 1150c099281SKai Wang 116d192f3d3SKai Wang /* 117d192f3d3SKai Wang * Create object from file, return created obj upon success, or NULL 118d192f3d3SKai Wang * when an error occurs or the member is not newer than existing 1196bef5d28SBenedict Reuschling * one while -u is specified. 120d192f3d3SKai Wang */ 121d192f3d3SKai Wang static struct ar_obj * 122d192f3d3SKai Wang create_obj_from_file(struct bsdar *bsdar, const char *name, time_t mtime) 123d192f3d3SKai Wang { 124d192f3d3SKai Wang struct ar_obj *obj; 125d192f3d3SKai Wang struct stat sb; 126d192f3d3SKai Wang const char *bname; 12709a11349SEd Schouten char *tmpname; 128d192f3d3SKai Wang 129d192f3d3SKai Wang if (name == NULL) 130d192f3d3SKai Wang return (NULL); 131d192f3d3SKai Wang 132d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 133d192f3d3SKai Wang if (obj == NULL) 134d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 135d192f3d3SKai Wang if ((obj->fd = open(name, O_RDONLY, 0)) < 0) { 136d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't open file: %s", name); 137d192f3d3SKai Wang free(obj); 138d192f3d3SKai Wang return (NULL); 139d192f3d3SKai Wang } 140d192f3d3SKai Wang 14109a11349SEd Schouten tmpname = strdup(name); 14209a11349SEd Schouten if (tmpname == NULL) 14309a11349SEd Schouten bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 14409a11349SEd Schouten if ((bname = basename(tmpname)) == NULL) 145d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "basename failed"); 146d192f3d3SKai Wang if (bsdar->options & AR_TR && strlen(bname) > _TRUNCATE_LEN) { 147d192f3d3SKai Wang if ((obj->name = malloc(_TRUNCATE_LEN + 1)) == NULL) 148d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 149d192f3d3SKai Wang (void)strncpy(obj->name, bname, _TRUNCATE_LEN); 150d192f3d3SKai Wang obj->name[_TRUNCATE_LEN] = '\0'; 151d192f3d3SKai Wang } else 152d192f3d3SKai Wang if ((obj->name = strdup(bname)) == NULL) 153d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 15409a11349SEd Schouten free(tmpname); 155d192f3d3SKai Wang 156d192f3d3SKai Wang if (fstat(obj->fd, &sb) < 0) { 157d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't fstat file: %s", obj->name); 158d192f3d3SKai Wang goto giveup; 159d192f3d3SKai Wang } 160d192f3d3SKai Wang if (!S_ISREG(sb.st_mode)) { 161d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s is not an ordinary file", obj->name); 162d192f3d3SKai Wang goto giveup; 163d192f3d3SKai Wang } 164d192f3d3SKai Wang 165d192f3d3SKai Wang /* 166d192f3d3SKai Wang * When option '-u' is specified and member is not newer than the 167d192f3d3SKai Wang * existing one, the replace will not happen. While if mtime == 0, 168d192f3d3SKai Wang * which indicates that this is to "replace a none exist member", 169d192f3d3SKai Wang * the replace will proceed regardless of '-u'. 170d192f3d3SKai Wang */ 171d192f3d3SKai Wang if (mtime != 0 && bsdar->options & AR_U && sb.st_mtime <= mtime) 172d192f3d3SKai Wang goto giveup; 173d192f3d3SKai Wang 1740e479ac8STim Kientzle /* 1750e479ac8STim Kientzle * When option '-D' is specified, mtime and UID / GID from the file 1760e479ac8STim Kientzle * will be replaced with 0, and file mode with 644. This ensures that 1770e479ac8STim Kientzle * checksums will match for two archives containing the exact same 1780e479ac8STim Kientzle * files. 1790e479ac8STim Kientzle */ 1800e479ac8STim Kientzle if (bsdar->options & AR_D) { 1810e479ac8STim Kientzle obj->uid = 0; 1820e479ac8STim Kientzle obj->gid = 0; 1830e479ac8STim Kientzle obj->mtime = 0; 1840e479ac8STim Kientzle obj->md = S_IFREG | 0644; 1850e479ac8STim Kientzle } else { 186d192f3d3SKai Wang obj->uid = sb.st_uid; 187d192f3d3SKai Wang obj->gid = sb.st_gid; 188d192f3d3SKai Wang obj->mtime = sb.st_mtime; 1890e479ac8STim Kientzle obj->md = sb.st_mode; 1900e479ac8STim Kientzle } 1910e479ac8STim Kientzle obj->size = sb.st_size; 192d192f3d3SKai Wang obj->dev = sb.st_dev; 193d192f3d3SKai Wang obj->ino = sb.st_ino; 19447203048SKai Wang 19547203048SKai Wang if (obj->size == 0) { 19647203048SKai Wang obj->maddr = NULL; 19747203048SKai Wang return (obj); 19847203048SKai Wang } 19947203048SKai Wang 200d192f3d3SKai Wang if ((obj->maddr = mmap(NULL, obj->size, PROT_READ, 201d192f3d3SKai Wang MAP_PRIVATE, obj->fd, (off_t)0)) == MAP_FAILED) { 202d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't mmap file: %s", obj->name); 203d192f3d3SKai Wang goto giveup; 204d192f3d3SKai Wang } 205d192f3d3SKai Wang if (close(obj->fd) < 0) 206d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 207d192f3d3SKai Wang obj->name); 208d192f3d3SKai Wang 209d192f3d3SKai Wang return (obj); 210d192f3d3SKai Wang 211d192f3d3SKai Wang giveup: 212d192f3d3SKai Wang if (close(obj->fd) < 0) 213d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 214d192f3d3SKai Wang obj->name); 215d192f3d3SKai Wang free(obj->name); 216d192f3d3SKai Wang free(obj); 217d192f3d3SKai Wang return (NULL); 218d192f3d3SKai Wang } 219d192f3d3SKai Wang 220d192f3d3SKai Wang /* 221cf933f5bSEd Maste * Free object itself and its associated allocations. 222cf933f5bSEd Maste */ 223cf933f5bSEd Maste static void 224cf933f5bSEd Maste free_obj(struct bsdar *bsdar, struct ar_obj *obj) 225cf933f5bSEd Maste { 226cf933f5bSEd Maste if (obj->fd == -1) 227cf933f5bSEd Maste free(obj->maddr); 228cf933f5bSEd Maste else 229cf933f5bSEd Maste if (obj->maddr != NULL && munmap(obj->maddr, obj->size)) 230cf933f5bSEd Maste bsdar_warnc(bsdar, errno, 231cf933f5bSEd Maste "can't munmap file: %s", obj->name); 232cf933f5bSEd Maste free(obj->name); 233cf933f5bSEd Maste free(obj); 234cf933f5bSEd Maste } 235cf933f5bSEd Maste 236cf933f5bSEd Maste /* 237d192f3d3SKai Wang * Insert obj to the tail, or before/after the pos obj. 238d192f3d3SKai Wang */ 239d192f3d3SKai Wang static void 240d192f3d3SKai Wang insert_obj(struct bsdar *bsdar, struct ar_obj *obj, struct ar_obj *pos) 241d192f3d3SKai Wang { 242d192f3d3SKai Wang if (obj == NULL) 243d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "try to insert a null obj"); 244d192f3d3SKai Wang 245d192f3d3SKai Wang if (pos == NULL || obj == pos) 246d192f3d3SKai Wang /* 2476bef5d28SBenedict Reuschling * If the object to move happens to be the position obj, 248d192f3d3SKai Wang * or if there is not a pos obj, move it to tail. 249d192f3d3SKai Wang */ 250d192f3d3SKai Wang goto tail; 251d192f3d3SKai Wang 252d192f3d3SKai Wang if (bsdar->options & AR_B) { 253d192f3d3SKai Wang TAILQ_INSERT_BEFORE(pos, obj, objs); 254d192f3d3SKai Wang return; 255d192f3d3SKai Wang } 256d192f3d3SKai Wang if (bsdar->options & AR_A) { 257d192f3d3SKai Wang TAILQ_INSERT_AFTER(&bsdar->v_obj, pos, obj, objs); 258d192f3d3SKai Wang return; 259d192f3d3SKai Wang } 260d192f3d3SKai Wang 261d192f3d3SKai Wang tail: 262d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 263d192f3d3SKai Wang 264d192f3d3SKai Wang } 265d192f3d3SKai Wang 266d192f3d3SKai Wang /* 2670c099281SKai Wang * Read objects from archive into v_obj list. Note that checkargv is 2680c099281SKai Wang * set when read_objs is used to read objects from the target of 2690c099281SKai Wang * ADDLIB command (ar script mode), in this case argv array possibly 2700c099281SKai Wang * specifies the members ADDLIB want. 271d192f3d3SKai Wang */ 272d192f3d3SKai Wang static void 2730c099281SKai Wang read_objs(struct bsdar *bsdar, const char *archive, int checkargv) 274d192f3d3SKai Wang { 275d192f3d3SKai Wang struct archive *a; 276d192f3d3SKai Wang struct archive_entry *entry; 2770c099281SKai Wang struct ar_obj *obj; 278d192f3d3SKai Wang const char *name; 279d192f3d3SKai Wang const char *bname; 280d192f3d3SKai Wang char *buff; 281d192f3d3SKai Wang char **av; 282d192f3d3SKai Wang size_t size; 2830c099281SKai Wang int i, r, find; 284d192f3d3SKai Wang 285d192f3d3SKai Wang if ((a = archive_read_new()) == NULL) 286d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_read_new failed"); 287d192f3d3SKai Wang archive_read_support_format_ar(a); 2880c099281SKai Wang AC(archive_read_open_filename(a, archive, DEF_BLKSZ)); 289d192f3d3SKai Wang for (;;) { 290d192f3d3SKai Wang r = archive_read_next_header(a, &entry); 291d192f3d3SKai Wang if (r == ARCHIVE_FATAL) 292d192f3d3SKai Wang bsdar_errc(bsdar, EX_DATAERR, 0, "%s", 293d192f3d3SKai Wang archive_error_string(a)); 294d192f3d3SKai Wang if (r == ARCHIVE_EOF) 295d192f3d3SKai Wang break; 296d192f3d3SKai Wang if (r == ARCHIVE_WARN || r == ARCHIVE_RETRY) 297d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s", archive_error_string(a)); 298d192f3d3SKai Wang if (r == ARCHIVE_RETRY) { 299d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "Retrying..."); 300d192f3d3SKai Wang continue; 301d192f3d3SKai Wang } 302d192f3d3SKai Wang 303d192f3d3SKai Wang name = archive_entry_pathname(entry); 304d192f3d3SKai Wang 305d192f3d3SKai Wang /* 306d192f3d3SKai Wang * skip pseudo members. 307d192f3d3SKai Wang */ 308d192f3d3SKai Wang if (strcmp(name, "/") == 0 || strcmp(name, "//") == 0) 309d192f3d3SKai Wang continue; 310d192f3d3SKai Wang 3110c099281SKai Wang /* 3120c099281SKai Wang * If checkargv is set, only read those members specified 3130c099281SKai Wang * in argv. 3140c099281SKai Wang */ 3150c099281SKai Wang if (checkargv && bsdar->argc > 0) { 3160c099281SKai Wang find = 0; 3170c099281SKai Wang for(i = 0; i < bsdar->argc; i++) { 3180c099281SKai Wang av = &bsdar->argv[i]; 3190c099281SKai Wang if (*av == NULL) 3200c099281SKai Wang continue; 3210c099281SKai Wang if ((bname = basename(*av)) == NULL) 3220c099281SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 3230c099281SKai Wang "basename failed"); 3240c099281SKai Wang if (strcmp(bname, name) != 0) 3250c099281SKai Wang continue; 3260c099281SKai Wang 3270c099281SKai Wang *av = NULL; 3280c099281SKai Wang find = 1; 3290c099281SKai Wang break; 3300c099281SKai Wang } 3310c099281SKai Wang if (!find) 3320c099281SKai Wang continue; 3330c099281SKai Wang } 3340c099281SKai Wang 335d192f3d3SKai Wang size = archive_entry_size(entry); 336d192f3d3SKai Wang 337517dbe2eSKai Wang if (size > 0) { 338d192f3d3SKai Wang if ((buff = malloc(size)) == NULL) 339517dbe2eSKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 340517dbe2eSKai Wang "malloc failed"); 341d192f3d3SKai Wang if (archive_read_data(a, buff, size) != (ssize_t)size) { 342517dbe2eSKai Wang bsdar_warnc(bsdar, 0, "%s", 343517dbe2eSKai Wang archive_error_string(a)); 344d192f3d3SKai Wang free(buff); 345d192f3d3SKai Wang continue; 346d192f3d3SKai Wang } 347517dbe2eSKai Wang } else 348517dbe2eSKai Wang buff = NULL; 349517dbe2eSKai Wang 350d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 351d192f3d3SKai Wang if (obj == NULL) 352d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 353d192f3d3SKai Wang obj->maddr = buff; 354d192f3d3SKai Wang if ((obj->name = strdup(name)) == NULL) 355d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 356d192f3d3SKai Wang obj->size = size; 357d192f3d3SKai Wang obj->uid = archive_entry_uid(entry); 358d192f3d3SKai Wang obj->gid = archive_entry_gid(entry); 359d192f3d3SKai Wang obj->md = archive_entry_mode(entry); 360d192f3d3SKai Wang obj->mtime = archive_entry_mtime(entry); 361d192f3d3SKai Wang obj->dev = 0; 362d192f3d3SKai Wang obj->ino = 0; 363d192f3d3SKai Wang 364d192f3d3SKai Wang /* 365d192f3d3SKai Wang * Objects from archive have obj->fd set to -1, 366d192f3d3SKai Wang * for the ease of cleaning up. 367d192f3d3SKai Wang */ 368d192f3d3SKai Wang obj->fd = -1; 369d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 370d192f3d3SKai Wang } 371d192f3d3SKai Wang AC(archive_read_close(a)); 372ebb8fc42SMartin Matuska AC(archive_read_free(a)); 3730c099281SKai Wang } 3740c099281SKai Wang 3750c099281SKai Wang /* 3760c099281SKai Wang * Determine the constitution of resulting archive. 3770c099281SKai Wang */ 3780c099281SKai Wang static void 3790c099281SKai Wang write_archive(struct bsdar *bsdar, char mode) 3800c099281SKai Wang { 3810c099281SKai Wang struct ar_obj *nobj, *obj, *obj_temp, *pos; 3820c099281SKai Wang struct stat sb; 3830c099281SKai Wang const char *bname; 3840c099281SKai Wang char **av; 3850c099281SKai Wang int i; 3860c099281SKai Wang 3870c099281SKai Wang TAILQ_INIT(&bsdar->v_obj); 3880c099281SKai Wang nobj = NULL; 3890c099281SKai Wang pos = NULL; 3900c099281SKai Wang memset(&sb, 0, sizeof(sb)); 3910c099281SKai Wang 3920c099281SKai Wang /* 3930c099281SKai Wang * Test if the specified archive exists, to figure out 3940c099281SKai Wang * whether we are creating one here. 3950c099281SKai Wang */ 3960c099281SKai Wang if (stat(bsdar->filename, &sb) != 0) { 3970c099281SKai Wang if (errno != ENOENT) { 3980c099281SKai Wang bsdar_warnc(bsdar, 0, "stat %s failed", 3990c099281SKai Wang bsdar->filename); 4000c099281SKai Wang return; 4010c099281SKai Wang } 4020c099281SKai Wang 4030c099281SKai Wang /* We do not create archive in mode 'd', 'm' and 's'. */ 4040c099281SKai Wang if (mode != 'r' && mode != 'q') { 4050c099281SKai Wang bsdar_warnc(bsdar, 0, "%s: no such file", 4060c099281SKai Wang bsdar->filename); 4070c099281SKai Wang return; 4080c099281SKai Wang } 4090c099281SKai Wang 4100c099281SKai Wang /* Issue a warning if -c is not specified when creating. */ 4110c099281SKai Wang if (!(bsdar->options & AR_C)) 4120c099281SKai Wang bsdar_warnc(bsdar, 0, "creating %s", bsdar->filename); 4130c099281SKai Wang goto new_archive; 4140c099281SKai Wang } 4150c099281SKai Wang 4160c099281SKai Wang /* 4170c099281SKai Wang * First read members from existing archive. 4180c099281SKai Wang */ 4190c099281SKai Wang read_objs(bsdar, bsdar->filename, 0); 420d192f3d3SKai Wang 421d192f3d3SKai Wang /* 422d192f3d3SKai Wang * For mode 's', no member will be moved, deleted or replaced. 423d192f3d3SKai Wang */ 424d192f3d3SKai Wang if (mode == 's') 425d192f3d3SKai Wang goto write_objs; 426d192f3d3SKai Wang 427d192f3d3SKai Wang /* 428cb0dad38SKai Wang * For mode 'q', we don't need to adjust existing members either. 429cb0dad38SKai Wang * Also, -a, -b and -i are ignored in this mode. New members are 430cb0dad38SKai Wang * always inserted at tail. 431cb0dad38SKai Wang */ 432cb0dad38SKai Wang if (mode == 'q') 433cb0dad38SKai Wang goto new_archive; 434cb0dad38SKai Wang 435cb0dad38SKai Wang /* 4360c099281SKai Wang * Mode 'A' adds the contents of another archive to the tail of 4370c099281SKai Wang * current archive. Note that mode 'A' is a special mode for the 4380c099281SKai Wang * ADDLIB command of the ar script mode. Currently there is no 4390c099281SKai Wang * access to this function from the ar command line mode. 4400c099281SKai Wang */ 4410c099281SKai Wang if (mode == 'A') { 4420c099281SKai Wang /* 4430c099281SKai Wang * Read objects from the target archive of ADDLIB command. 4446bef5d28SBenedict Reuschling * If there are members specified in argv, read those members 4450c099281SKai Wang * only, otherwise the entire archive will be read. 4460c099281SKai Wang */ 4470c099281SKai Wang read_objs(bsdar, bsdar->addlib, 1); 4480c099281SKai Wang goto write_objs; 4490c099281SKai Wang } 4500c099281SKai Wang 4510c099281SKai Wang /* 452d192f3d3SKai Wang * Try to find the position member specified by user. 453d192f3d3SKai Wang */ 454d192f3d3SKai Wang if (bsdar->options & AR_A || bsdar->options & AR_B) { 455d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 456d192f3d3SKai Wang if (strcmp(obj->name, bsdar->posarg) == 0) { 457d192f3d3SKai Wang pos = obj; 458d192f3d3SKai Wang break; 459d192f3d3SKai Wang } 460d192f3d3SKai Wang } 461d192f3d3SKai Wang 462d192f3d3SKai Wang /* 463d192f3d3SKai Wang * If can't find `pos' specified by user, 4646bef5d28SBenedict Reuschling * silently insert objects at tail. 465d192f3d3SKai Wang */ 466d192f3d3SKai Wang if (pos == NULL) 467d192f3d3SKai Wang bsdar->options &= ~(AR_A | AR_B); 468d192f3d3SKai Wang } 469d192f3d3SKai Wang 470d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 471d192f3d3SKai Wang av = &bsdar->argv[i]; 472d192f3d3SKai Wang 473d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 474d192f3d3SKai Wang if ((bname = basename(*av)) == NULL) 475d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 476d192f3d3SKai Wang "basename failed"); 477d192f3d3SKai Wang if (bsdar->options & AR_TR) { 478d192f3d3SKai Wang if (strncmp(bname, obj->name, _TRUNCATE_LEN)) 479d192f3d3SKai Wang continue; 480d192f3d3SKai Wang } else 481d192f3d3SKai Wang if (strcmp(bname, obj->name) != 0) 482d192f3d3SKai Wang continue; 483d192f3d3SKai Wang 484d192f3d3SKai Wang if (mode == 'r') { 485d192f3d3SKai Wang /* 486d192f3d3SKai Wang * if the new member is not qualified 487d192f3d3SKai Wang * to replace the old one, skip it. 488d192f3d3SKai Wang */ 489d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 490d192f3d3SKai Wang obj->mtime); 491d192f3d3SKai Wang if (nobj == NULL) 492d192f3d3SKai Wang goto skip_obj; 493d192f3d3SKai Wang } 494d192f3d3SKai Wang 495d192f3d3SKai Wang if (bsdar->options & AR_V) 496d192f3d3SKai Wang (void)fprintf(stdout, "%c - %s\n", mode, 497d192f3d3SKai Wang *av); 498d192f3d3SKai Wang 499d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 500cf933f5bSEd Maste if (mode == 'd' || mode == 'r') 501cf933f5bSEd Maste free_obj(bsdar, obj); 502d192f3d3SKai Wang 503d192f3d3SKai Wang if (mode == 'm') 504d192f3d3SKai Wang insert_obj(bsdar, obj, pos); 505d192f3d3SKai Wang if (mode == 'r') 506d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 507d192f3d3SKai Wang 508d192f3d3SKai Wang skip_obj: 509d192f3d3SKai Wang *av = NULL; 510d192f3d3SKai Wang break; 511d192f3d3SKai Wang } 512d192f3d3SKai Wang 513d192f3d3SKai Wang } 514d192f3d3SKai Wang 515d192f3d3SKai Wang new_archive: 516d192f3d3SKai Wang /* 517d192f3d3SKai Wang * When operating in mode 'r', directly add those user specified 518cb0dad38SKai Wang * objects which do not exist in current archive. When operating 519cb0dad38SKai Wang * in mode 'q', all objects specified in command line args are 520cb0dad38SKai Wang * appended to the archive, without comparing with existing ones. 521d192f3d3SKai Wang */ 522d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 523d192f3d3SKai Wang av = &bsdar->argv[i]; 524cb0dad38SKai Wang if (*av != NULL && (mode == 'r' || mode == 'q')) { 525d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 0); 526d192f3d3SKai Wang if (nobj != NULL) 527d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 528d192f3d3SKai Wang if (bsdar->options & AR_V && nobj != NULL) 529d192f3d3SKai Wang (void)fprintf(stdout, "a - %s\n", *av); 530d192f3d3SKai Wang *av = NULL; 531d192f3d3SKai Wang } 532d192f3d3SKai Wang } 533d192f3d3SKai Wang 534d192f3d3SKai Wang write_objs: 535d192f3d3SKai Wang write_objs(bsdar); 536d192f3d3SKai Wang write_cleanup(bsdar); 537d192f3d3SKai Wang } 538d192f3d3SKai Wang 539d192f3d3SKai Wang /* 540d192f3d3SKai Wang * Memory cleaning up. 541d192f3d3SKai Wang */ 542d192f3d3SKai Wang static void 543d192f3d3SKai Wang write_cleanup(struct bsdar *bsdar) 544d192f3d3SKai Wang { 545d192f3d3SKai Wang struct ar_obj *obj, *obj_temp; 546d192f3d3SKai Wang 547d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 548d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 549cf933f5bSEd Maste free_obj(bsdar, obj); 550d192f3d3SKai Wang } 551d192f3d3SKai Wang 552d192f3d3SKai Wang free(bsdar->as); 553d192f3d3SKai Wang free(bsdar->s_so); 554d192f3d3SKai Wang free(bsdar->s_sn); 555d192f3d3SKai Wang bsdar->as = NULL; 556d192f3d3SKai Wang bsdar->s_so = NULL; 557d192f3d3SKai Wang bsdar->s_sn = NULL; 558d192f3d3SKai Wang } 559d192f3d3SKai Wang 560d192f3d3SKai Wang /* 56101715d8cSEd Maste * Fault in the buffer prior to writing as a workaround for poor performance 56201715d8cSEd Maste * due to interaction with kernel fs deadlock avoidance code. See the comment 56301715d8cSEd Maste * above vn_io_fault_doio() in sys/kern/vfs_vnops.c for details of the issue. 56401715d8cSEd Maste */ 56501715d8cSEd Maste static void 56601715d8cSEd Maste prefault_buffer(const char *buf, size_t s) 56701715d8cSEd Maste { 56801715d8cSEd Maste volatile const char *p; 56901715d8cSEd Maste size_t page_size; 57001715d8cSEd Maste 57101715d8cSEd Maste if (s == 0) 57201715d8cSEd Maste return; 57301715d8cSEd Maste page_size = sysconf(_SC_PAGESIZE); 57401715d8cSEd Maste for (p = buf; p < buf + s; p += page_size) 57501715d8cSEd Maste *p; 57601715d8cSEd Maste /* 57701715d8cSEd Maste * Ensure we touch the last page as well, in case the buffer is not 57801715d8cSEd Maste * page-aligned. 57901715d8cSEd Maste */ 58001715d8cSEd Maste *(volatile const char *)(buf + s - 1); 58101715d8cSEd Maste } 58201715d8cSEd Maste 58301715d8cSEd Maste /* 584d192f3d3SKai Wang * Wrapper for archive_write_data(). 585d192f3d3SKai Wang */ 586d192f3d3SKai Wang static void 587d192f3d3SKai Wang write_data(struct bsdar *bsdar, struct archive *a, const void *buf, size_t s) 588d192f3d3SKai Wang { 58901715d8cSEd Maste prefault_buffer(buf, s); 590d192f3d3SKai Wang if (archive_write_data(a, buf, s) != (ssize_t)s) 591d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "%s", 592d192f3d3SKai Wang archive_error_string(a)); 593d192f3d3SKai Wang } 594d192f3d3SKai Wang 595d192f3d3SKai Wang /* 596d192f3d3SKai Wang * Write the resulting archive members. 597d192f3d3SKai Wang */ 598d192f3d3SKai Wang static void 599d192f3d3SKai Wang write_objs(struct bsdar *bsdar) 600d192f3d3SKai Wang { 601d192f3d3SKai Wang struct ar_obj *obj; 602d192f3d3SKai Wang struct archive *a; 603d192f3d3SKai Wang struct archive_entry *entry; 604d192f3d3SKai Wang size_t s_sz; /* size of archive symbol table. */ 605d192f3d3SKai Wang size_t pm_sz; /* size of pseudo members */ 606d192f3d3SKai Wang int i, nr; 607d192f3d3SKai Wang 608d192f3d3SKai Wang if (elf_version(EV_CURRENT) == EV_NONE) 609d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, 610d192f3d3SKai Wang "ELF library initialization failed: %s", elf_errmsg(-1)); 611d192f3d3SKai Wang 612d192f3d3SKai Wang bsdar->rela_off = 0; 613d192f3d3SKai Wang 614d192f3d3SKai Wang /* Create archive symbol table and archive string table, if need. */ 615d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 61647203048SKai Wang if (!(bsdar->options & AR_SS) && obj->maddr != NULL) 617d192f3d3SKai Wang create_symtab_entry(bsdar, obj->maddr, obj->size); 618d192f3d3SKai Wang if (strlen(obj->name) > _MAXNAMELEN_SVR4) 619d192f3d3SKai Wang add_to_ar_str_table(bsdar, obj->name); 620d192f3d3SKai Wang bsdar->rela_off += _ARHDR_LEN + obj->size + obj->size % 2; 621d192f3d3SKai Wang } 622d192f3d3SKai Wang 623d192f3d3SKai Wang /* 624d192f3d3SKai Wang * Pad the symbol name string table. It is treated specially because 625d192f3d3SKai Wang * symbol name table should be padded by a '\0', not the common '\n' 626d192f3d3SKai Wang * for other members. The size of sn table includes the pad bit. 627d192f3d3SKai Wang */ 628d192f3d3SKai Wang if (bsdar->s_cnt != 0 && bsdar->s_sn_sz % 2 != 0) 629d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 630d192f3d3SKai Wang 631d192f3d3SKai Wang /* 632d192f3d3SKai Wang * Archive string table is padded by a "\n" as the normal members. 633517dbe2eSKai Wang * The difference is that the size of archive string table counts 634d192f3d3SKai Wang * in the pad bit, while normal members' size fileds do not. 635d192f3d3SKai Wang */ 636d192f3d3SKai Wang if (bsdar->as != NULL && bsdar->as_sz % 2 != 0) 637d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 638d192f3d3SKai Wang 639d192f3d3SKai Wang /* 640d192f3d3SKai Wang * If there is a symbol table, calculate the size of pseudo members, 641517dbe2eSKai Wang * convert previously stored relative offsets to absolute ones, and 642d192f3d3SKai Wang * then make them Big Endian. 643d192f3d3SKai Wang * 644d192f3d3SKai Wang * absolute_offset = htobe32(relative_offset + size_of_pseudo_members) 645d192f3d3SKai Wang */ 646d192f3d3SKai Wang 647d192f3d3SKai Wang if (bsdar->s_cnt != 0) { 648d192f3d3SKai Wang s_sz = (bsdar->s_cnt + 1) * sizeof(uint32_t) + bsdar->s_sn_sz; 649d192f3d3SKai Wang pm_sz = _ARMAG_LEN + (_ARHDR_LEN + s_sz); 650d192f3d3SKai Wang if (bsdar->as != NULL) 651d192f3d3SKai Wang pm_sz += _ARHDR_LEN + bsdar->as_sz; 652d192f3d3SKai Wang for (i = 0; (size_t)i < bsdar->s_cnt; i++) 653d192f3d3SKai Wang *(bsdar->s_so + i) = htobe32(*(bsdar->s_so + i) + 654d192f3d3SKai Wang pm_sz); 655d192f3d3SKai Wang } 656d192f3d3SKai Wang 657d192f3d3SKai Wang if ((a = archive_write_new()) == NULL) 658d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_write_new failed"); 659d192f3d3SKai Wang 660d192f3d3SKai Wang archive_write_set_format_ar_svr4(a); 661d192f3d3SKai Wang 662d192f3d3SKai Wang AC(archive_write_open_filename(a, bsdar->filename)); 663d192f3d3SKai Wang 664d192f3d3SKai Wang /* 665d192f3d3SKai Wang * write the archive symbol table, if there is one. 666d192f3d3SKai Wang * If options -s is explicitly specified or we are invoked 667d192f3d3SKai Wang * as ranlib, write the symbol table even if it is empty. 668d192f3d3SKai Wang */ 669d192f3d3SKai Wang if ((bsdar->s_cnt != 0 && !(bsdar->options & AR_SS)) || 670d192f3d3SKai Wang bsdar->options & AR_S) { 671d192f3d3SKai Wang entry = archive_entry_new(); 672bd9302d8SBryan Drewery if (entry == NULL) 673bd9302d8SBryan Drewery bsdar_errc(bsdar, EX_SOFTWARE, 0, 674bd9302d8SBryan Drewery "archive_entry_new failed"); 675d192f3d3SKai Wang archive_entry_copy_pathname(entry, "/"); 6760e479ac8STim Kientzle if ((bsdar->options & AR_D) == 0) 677d192f3d3SKai Wang archive_entry_set_mtime(entry, time(NULL), 0); 678d192f3d3SKai Wang archive_entry_set_size(entry, (bsdar->s_cnt + 1) * 679d192f3d3SKai Wang sizeof(uint32_t) + bsdar->s_sn_sz); 680d192f3d3SKai Wang AC(archive_write_header(a, entry)); 681d192f3d3SKai Wang nr = htobe32(bsdar->s_cnt); 682d192f3d3SKai Wang write_data(bsdar, a, &nr, sizeof(uint32_t)); 683d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_so, sizeof(uint32_t) * 684d192f3d3SKai Wang bsdar->s_cnt); 685d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_sn, bsdar->s_sn_sz); 686d192f3d3SKai Wang archive_entry_free(entry); 687d192f3d3SKai Wang } 688d192f3d3SKai Wang 689d192f3d3SKai Wang /* write the archive string table, if any. */ 690d192f3d3SKai Wang if (bsdar->as != NULL) { 691d192f3d3SKai Wang entry = archive_entry_new(); 692bd9302d8SBryan Drewery if (entry == NULL) 693bd9302d8SBryan Drewery bsdar_errc(bsdar, EX_SOFTWARE, 0, 694bd9302d8SBryan Drewery "archive_entry_new failed"); 695d192f3d3SKai Wang archive_entry_copy_pathname(entry, "//"); 696d192f3d3SKai Wang archive_entry_set_size(entry, bsdar->as_sz); 697d192f3d3SKai Wang AC(archive_write_header(a, entry)); 698d192f3d3SKai Wang write_data(bsdar, a, bsdar->as, bsdar->as_sz); 699d192f3d3SKai Wang archive_entry_free(entry); 700d192f3d3SKai Wang } 701d192f3d3SKai Wang 702d192f3d3SKai Wang /* write normal members. */ 703d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 704d192f3d3SKai Wang entry = archive_entry_new(); 705bd9302d8SBryan Drewery if (entry == NULL) 706bd9302d8SBryan Drewery bsdar_errc(bsdar, EX_SOFTWARE, 0, 707bd9302d8SBryan Drewery "archive_entry_new failed"); 708d192f3d3SKai Wang archive_entry_copy_pathname(entry, obj->name); 709d192f3d3SKai Wang archive_entry_set_uid(entry, obj->uid); 710d192f3d3SKai Wang archive_entry_set_gid(entry, obj->gid); 711d192f3d3SKai Wang archive_entry_set_mode(entry, obj->md); 712d192f3d3SKai Wang archive_entry_set_size(entry, obj->size); 713d192f3d3SKai Wang archive_entry_set_mtime(entry, obj->mtime, 0); 714d192f3d3SKai Wang archive_entry_set_dev(entry, obj->dev); 715d192f3d3SKai Wang archive_entry_set_ino(entry, obj->ino); 716d192f3d3SKai Wang archive_entry_set_filetype(entry, AE_IFREG); 717d192f3d3SKai Wang AC(archive_write_header(a, entry)); 718d192f3d3SKai Wang write_data(bsdar, a, obj->maddr, obj->size); 719d192f3d3SKai Wang archive_entry_free(entry); 720d192f3d3SKai Wang } 721d192f3d3SKai Wang 722d192f3d3SKai Wang AC(archive_write_close(a)); 723ebb8fc42SMartin Matuska AC(archive_write_free(a)); 724d192f3d3SKai Wang } 725d192f3d3SKai Wang 726d192f3d3SKai Wang /* 727d192f3d3SKai Wang * Extract global symbols from ELF binary members. 728d192f3d3SKai Wang */ 729d192f3d3SKai Wang static void 730d192f3d3SKai Wang create_symtab_entry(struct bsdar *bsdar, void *maddr, size_t size) 731d192f3d3SKai Wang { 732d192f3d3SKai Wang Elf *e; 733d192f3d3SKai Wang Elf_Scn *scn; 734d192f3d3SKai Wang GElf_Shdr shdr; 735d192f3d3SKai Wang GElf_Sym sym; 736d192f3d3SKai Wang Elf_Data *data; 737d192f3d3SKai Wang char *name; 738d192f3d3SKai Wang size_t n, shstrndx; 739d192f3d3SKai Wang int elferr, tabndx, len, i; 740d192f3d3SKai Wang 741d192f3d3SKai Wang if ((e = elf_memory(maddr, size)) == NULL) { 742d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_memory() failed: %s", 743d192f3d3SKai Wang elf_errmsg(-1)); 744d192f3d3SKai Wang return; 745d192f3d3SKai Wang } 746d192f3d3SKai Wang if (elf_kind(e) != ELF_K_ELF) { 7476bef5d28SBenedict Reuschling /* Silently ignore non-elf member. */ 748d192f3d3SKai Wang elf_end(e); 749d192f3d3SKai Wang return; 750d192f3d3SKai Wang } 751d192f3d3SKai Wang if (elf_getshstrndx(e, &shstrndx) == 0) { 752d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_getshstrndx failed: %s", 753d192f3d3SKai Wang elf_errmsg(-1)); 754d192f3d3SKai Wang elf_end(e); 755d192f3d3SKai Wang return; 756d192f3d3SKai Wang } 757d192f3d3SKai Wang 758d192f3d3SKai Wang tabndx = -1; 759d192f3d3SKai Wang scn = NULL; 760d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 761d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 762d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 763d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 764d192f3d3SKai Wang continue; 765d192f3d3SKai Wang } 766d192f3d3SKai Wang if ((name = elf_strptr(e, shstrndx, shdr.sh_name)) == NULL) { 767d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 768d192f3d3SKai Wang "elf_strptr failed: %s", elf_errmsg(-1)); 769d192f3d3SKai Wang continue; 770d192f3d3SKai Wang } 771d192f3d3SKai Wang if (strcmp(name, ".strtab") == 0) { 772d192f3d3SKai Wang tabndx = elf_ndxscn(scn); 773d192f3d3SKai Wang break; 774d192f3d3SKai Wang } 775d192f3d3SKai Wang } 776d192f3d3SKai Wang elferr = elf_errno(); 777d192f3d3SKai Wang if (elferr != 0) 778d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_nextscn failed: %s", 779d192f3d3SKai Wang elf_errmsg(elferr)); 780d192f3d3SKai Wang if (tabndx == -1) { 781d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "can't find .strtab section"); 782d192f3d3SKai Wang elf_end(e); 783d192f3d3SKai Wang return; 784d192f3d3SKai Wang } 785d192f3d3SKai Wang 786d192f3d3SKai Wang scn = NULL; 787d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 788d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 789d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 790d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 791d192f3d3SKai Wang continue; 792d192f3d3SKai Wang } 793d192f3d3SKai Wang if (shdr.sh_type != SHT_SYMTAB) 794d192f3d3SKai Wang continue; 795d192f3d3SKai Wang 796d192f3d3SKai Wang data = NULL; 797d192f3d3SKai Wang n = 0; 798d192f3d3SKai Wang while (n < shdr.sh_size && 799d192f3d3SKai Wang (data = elf_getdata(scn, data)) != NULL) { 800d192f3d3SKai Wang len = data->d_size / shdr.sh_entsize; 801d192f3d3SKai Wang for (i = 0; i < len; i++) { 802d192f3d3SKai Wang if (gelf_getsym(data, i, &sym) != &sym) { 803d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 804d192f3d3SKai Wang "gelf_getsym failed: %s", 805d192f3d3SKai Wang elf_errmsg(-1)); 806d192f3d3SKai Wang continue; 807d192f3d3SKai Wang } 808d192f3d3SKai Wang 809d192f3d3SKai Wang /* keep only global or weak symbols */ 810d192f3d3SKai Wang if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL && 811d192f3d3SKai Wang GELF_ST_BIND(sym.st_info) != STB_WEAK) 812d192f3d3SKai Wang continue; 813d192f3d3SKai Wang 814d192f3d3SKai Wang /* keep only defined symbols */ 815d192f3d3SKai Wang if (sym.st_shndx == SHN_UNDEF) 816d192f3d3SKai Wang continue; 817d192f3d3SKai Wang 818d192f3d3SKai Wang if ((name = elf_strptr(e, tabndx, 819d192f3d3SKai Wang sym.st_name)) == NULL) { 820d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 821d192f3d3SKai Wang "elf_strptr failed: %s", 822d192f3d3SKai Wang elf_errmsg(-1)); 823d192f3d3SKai Wang continue; 824d192f3d3SKai Wang } 825d192f3d3SKai Wang 826d192f3d3SKai Wang add_to_ar_sym_table(bsdar, name); 827d192f3d3SKai Wang } 828d192f3d3SKai Wang } 829d192f3d3SKai Wang } 830d192f3d3SKai Wang elferr = elf_errno(); 831d192f3d3SKai Wang if (elferr != 0) 832d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_nextscn failed: %s", 833d192f3d3SKai Wang elf_errmsg(elferr)); 834d192f3d3SKai Wang 835d192f3d3SKai Wang elf_end(e); 836d192f3d3SKai Wang } 837d192f3d3SKai Wang 838d192f3d3SKai Wang /* 839d192f3d3SKai Wang * Append to the archive string table buffer. 840d192f3d3SKai Wang */ 841d192f3d3SKai Wang static void 842d192f3d3SKai Wang add_to_ar_str_table(struct bsdar *bsdar, const char *name) 843d192f3d3SKai Wang { 844d192f3d3SKai Wang 845d192f3d3SKai Wang if (bsdar->as == NULL) { 846d192f3d3SKai Wang bsdar->as_cap = _INIT_AS_CAP; 847d192f3d3SKai Wang bsdar->as_sz = 0; 848d192f3d3SKai Wang if ((bsdar->as = malloc(bsdar->as_cap)) == NULL) 849d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 850d192f3d3SKai Wang } 851d192f3d3SKai Wang 852d192f3d3SKai Wang /* 853d192f3d3SKai Wang * The space required for holding one member name in as table includes: 854d192f3d3SKai Wang * strlen(name) + (1 for '/') + (1 for '\n') + (possibly 1 for padding). 855d192f3d3SKai Wang */ 856517dbe2eSKai Wang while (bsdar->as_sz + strlen(name) + 3 > bsdar->as_cap) { 857d192f3d3SKai Wang bsdar->as_cap *= 2; 858d192f3d3SKai Wang bsdar->as = realloc(bsdar->as, bsdar->as_cap); 859d192f3d3SKai Wang if (bsdar->as == NULL) 860d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 861d192f3d3SKai Wang } 862d192f3d3SKai Wang strncpy(&bsdar->as[bsdar->as_sz], name, strlen(name)); 863d192f3d3SKai Wang bsdar->as_sz += strlen(name); 864d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '/'; 865d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 866d192f3d3SKai Wang } 867d192f3d3SKai Wang 868d192f3d3SKai Wang /* 869d192f3d3SKai Wang * Append to the archive symbol table buffer. 870d192f3d3SKai Wang */ 871d192f3d3SKai Wang static void 872d192f3d3SKai Wang add_to_ar_sym_table(struct bsdar *bsdar, const char *name) 873d192f3d3SKai Wang { 874d192f3d3SKai Wang 875d192f3d3SKai Wang if (bsdar->s_so == NULL) { 876d192f3d3SKai Wang if ((bsdar->s_so = malloc(_INIT_SYMOFF_CAP)) == 877d192f3d3SKai Wang NULL) 878d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 879d192f3d3SKai Wang bsdar->s_so_cap = _INIT_SYMOFF_CAP; 880d192f3d3SKai Wang bsdar->s_cnt = 0; 881d192f3d3SKai Wang } 882d192f3d3SKai Wang 883d192f3d3SKai Wang if (bsdar->s_sn == NULL) { 884d192f3d3SKai Wang if ((bsdar->s_sn = malloc(_INIT_SYMNAME_CAP)) == NULL) 885d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 886d192f3d3SKai Wang bsdar->s_sn_cap = _INIT_SYMNAME_CAP; 887d192f3d3SKai Wang bsdar->s_sn_sz = 0; 888d192f3d3SKai Wang } 889d192f3d3SKai Wang 890d192f3d3SKai Wang if (bsdar->s_cnt * sizeof(uint32_t) >= bsdar->s_so_cap) { 891d192f3d3SKai Wang bsdar->s_so_cap *= 2; 892d192f3d3SKai Wang bsdar->s_so = realloc(bsdar->s_so, bsdar->s_so_cap); 893d192f3d3SKai Wang if (bsdar->s_so == NULL) 894d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 895d192f3d3SKai Wang } 896d192f3d3SKai Wang bsdar->s_so[bsdar->s_cnt] = bsdar->rela_off; 897d192f3d3SKai Wang bsdar->s_cnt++; 898d192f3d3SKai Wang 899d192f3d3SKai Wang /* 900d192f3d3SKai Wang * The space required for holding one symbol name in sn table includes: 901d192f3d3SKai Wang * strlen(name) + (1 for '\n') + (possibly 1 for padding). 902d192f3d3SKai Wang */ 903517dbe2eSKai Wang while (bsdar->s_sn_sz + strlen(name) + 2 > bsdar->s_sn_cap) { 904d192f3d3SKai Wang bsdar->s_sn_cap *= 2; 905d192f3d3SKai Wang bsdar->s_sn = realloc(bsdar->s_sn, bsdar->s_sn_cap); 906d192f3d3SKai Wang if (bsdar->s_sn == NULL) 907d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 908d192f3d3SKai Wang } 909d192f3d3SKai Wang strncpy(&bsdar->s_sn[bsdar->s_sn_sz], name, strlen(name)); 910d192f3d3SKai Wang bsdar->s_sn_sz += strlen(name); 911d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 912d192f3d3SKai Wang } 913