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; 127d192f3d3SKai Wang 128d192f3d3SKai Wang if (name == NULL) 129d192f3d3SKai Wang return (NULL); 130d192f3d3SKai Wang 131d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 132d192f3d3SKai Wang if (obj == NULL) 133d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 134d192f3d3SKai Wang if ((obj->fd = open(name, O_RDONLY, 0)) < 0) { 135d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't open file: %s", name); 136d192f3d3SKai Wang free(obj); 137d192f3d3SKai Wang return (NULL); 138d192f3d3SKai Wang } 139d192f3d3SKai Wang 140d192f3d3SKai Wang if ((bname = basename(name)) == NULL) 141d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "basename failed"); 142d192f3d3SKai Wang if (bsdar->options & AR_TR && strlen(bname) > _TRUNCATE_LEN) { 143d192f3d3SKai Wang if ((obj->name = malloc(_TRUNCATE_LEN + 1)) == NULL) 144d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 145d192f3d3SKai Wang (void)strncpy(obj->name, bname, _TRUNCATE_LEN); 146d192f3d3SKai Wang obj->name[_TRUNCATE_LEN] = '\0'; 147d192f3d3SKai Wang } else 148d192f3d3SKai Wang if ((obj->name = strdup(bname)) == NULL) 149d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 150d192f3d3SKai Wang 151d192f3d3SKai Wang if (fstat(obj->fd, &sb) < 0) { 152d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't fstat file: %s", obj->name); 153d192f3d3SKai Wang goto giveup; 154d192f3d3SKai Wang } 155d192f3d3SKai Wang if (!S_ISREG(sb.st_mode)) { 156d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s is not an ordinary file", obj->name); 157d192f3d3SKai Wang goto giveup; 158d192f3d3SKai Wang } 159d192f3d3SKai Wang 160d192f3d3SKai Wang /* 161d192f3d3SKai Wang * When option '-u' is specified and member is not newer than the 162d192f3d3SKai Wang * existing one, the replace will not happen. While if mtime == 0, 163d192f3d3SKai Wang * which indicates that this is to "replace a none exist member", 164d192f3d3SKai Wang * the replace will proceed regardless of '-u'. 165d192f3d3SKai Wang */ 166d192f3d3SKai Wang if (mtime != 0 && bsdar->options & AR_U && sb.st_mtime <= mtime) 167d192f3d3SKai Wang goto giveup; 168d192f3d3SKai Wang 1690e479ac8STim Kientzle /* 1700e479ac8STim Kientzle * When option '-D' is specified, mtime and UID / GID from the file 1710e479ac8STim Kientzle * will be replaced with 0, and file mode with 644. This ensures that 1720e479ac8STim Kientzle * checksums will match for two archives containing the exact same 1730e479ac8STim Kientzle * files. 1740e479ac8STim Kientzle */ 1750e479ac8STim Kientzle if (bsdar->options & AR_D) { 1760e479ac8STim Kientzle obj->uid = 0; 1770e479ac8STim Kientzle obj->gid = 0; 1780e479ac8STim Kientzle obj->mtime = 0; 1790e479ac8STim Kientzle obj->md = S_IFREG | 0644; 1800e479ac8STim Kientzle } else { 181d192f3d3SKai Wang obj->uid = sb.st_uid; 182d192f3d3SKai Wang obj->gid = sb.st_gid; 183d192f3d3SKai Wang obj->mtime = sb.st_mtime; 1840e479ac8STim Kientzle obj->md = sb.st_mode; 1850e479ac8STim Kientzle } 1860e479ac8STim Kientzle obj->size = sb.st_size; 187d192f3d3SKai Wang obj->dev = sb.st_dev; 188d192f3d3SKai Wang obj->ino = sb.st_ino; 18947203048SKai Wang 19047203048SKai Wang if (obj->size == 0) { 19147203048SKai Wang obj->maddr = NULL; 19247203048SKai Wang return (obj); 19347203048SKai Wang } 19447203048SKai Wang 195d192f3d3SKai Wang if ((obj->maddr = mmap(NULL, obj->size, PROT_READ, 196d192f3d3SKai Wang MAP_PRIVATE, obj->fd, (off_t)0)) == MAP_FAILED) { 197d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't mmap file: %s", obj->name); 198d192f3d3SKai Wang goto giveup; 199d192f3d3SKai Wang } 200d192f3d3SKai Wang if (close(obj->fd) < 0) 201d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 202d192f3d3SKai Wang obj->name); 203d192f3d3SKai Wang 204d192f3d3SKai Wang return (obj); 205d192f3d3SKai Wang 206d192f3d3SKai Wang giveup: 207d192f3d3SKai Wang if (close(obj->fd) < 0) 208d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 209d192f3d3SKai Wang obj->name); 210d192f3d3SKai Wang free(obj->name); 211d192f3d3SKai Wang free(obj); 212d192f3d3SKai Wang return (NULL); 213d192f3d3SKai Wang } 214d192f3d3SKai Wang 215d192f3d3SKai Wang /* 216cf933f5bSEd Maste * Free object itself and its associated allocations. 217cf933f5bSEd Maste */ 218cf933f5bSEd Maste static void 219cf933f5bSEd Maste free_obj(struct bsdar *bsdar, struct ar_obj *obj) 220cf933f5bSEd Maste { 221cf933f5bSEd Maste if (obj->fd == -1) 222cf933f5bSEd Maste free(obj->maddr); 223cf933f5bSEd Maste else 224cf933f5bSEd Maste if (obj->maddr != NULL && munmap(obj->maddr, obj->size)) 225cf933f5bSEd Maste bsdar_warnc(bsdar, errno, 226cf933f5bSEd Maste "can't munmap file: %s", obj->name); 227cf933f5bSEd Maste free(obj->name); 228cf933f5bSEd Maste free(obj); 229cf933f5bSEd Maste } 230cf933f5bSEd Maste 231cf933f5bSEd Maste /* 232d192f3d3SKai Wang * Insert obj to the tail, or before/after the pos obj. 233d192f3d3SKai Wang */ 234d192f3d3SKai Wang static void 235d192f3d3SKai Wang insert_obj(struct bsdar *bsdar, struct ar_obj *obj, struct ar_obj *pos) 236d192f3d3SKai Wang { 237d192f3d3SKai Wang if (obj == NULL) 238d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "try to insert a null obj"); 239d192f3d3SKai Wang 240d192f3d3SKai Wang if (pos == NULL || obj == pos) 241d192f3d3SKai Wang /* 2426bef5d28SBenedict Reuschling * If the object to move happens to be the position obj, 243d192f3d3SKai Wang * or if there is not a pos obj, move it to tail. 244d192f3d3SKai Wang */ 245d192f3d3SKai Wang goto tail; 246d192f3d3SKai Wang 247d192f3d3SKai Wang if (bsdar->options & AR_B) { 248d192f3d3SKai Wang TAILQ_INSERT_BEFORE(pos, obj, objs); 249d192f3d3SKai Wang return; 250d192f3d3SKai Wang } 251d192f3d3SKai Wang if (bsdar->options & AR_A) { 252d192f3d3SKai Wang TAILQ_INSERT_AFTER(&bsdar->v_obj, pos, obj, objs); 253d192f3d3SKai Wang return; 254d192f3d3SKai Wang } 255d192f3d3SKai Wang 256d192f3d3SKai Wang tail: 257d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 258d192f3d3SKai Wang 259d192f3d3SKai Wang } 260d192f3d3SKai Wang 261d192f3d3SKai Wang /* 2620c099281SKai Wang * Read objects from archive into v_obj list. Note that checkargv is 2630c099281SKai Wang * set when read_objs is used to read objects from the target of 2640c099281SKai Wang * ADDLIB command (ar script mode), in this case argv array possibly 2650c099281SKai Wang * specifies the members ADDLIB want. 266d192f3d3SKai Wang */ 267d192f3d3SKai Wang static void 2680c099281SKai Wang read_objs(struct bsdar *bsdar, const char *archive, int checkargv) 269d192f3d3SKai Wang { 270d192f3d3SKai Wang struct archive *a; 271d192f3d3SKai Wang struct archive_entry *entry; 2720c099281SKai Wang struct ar_obj *obj; 273d192f3d3SKai Wang const char *name; 274d192f3d3SKai Wang const char *bname; 275d192f3d3SKai Wang char *buff; 276d192f3d3SKai Wang char **av; 277d192f3d3SKai Wang size_t size; 2780c099281SKai Wang int i, r, find; 279d192f3d3SKai Wang 280d192f3d3SKai Wang if ((a = archive_read_new()) == NULL) 281d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_read_new failed"); 282d192f3d3SKai Wang archive_read_support_format_ar(a); 2830c099281SKai Wang AC(archive_read_open_filename(a, archive, DEF_BLKSZ)); 284d192f3d3SKai Wang for (;;) { 285d192f3d3SKai Wang r = archive_read_next_header(a, &entry); 286d192f3d3SKai Wang if (r == ARCHIVE_FATAL) 287d192f3d3SKai Wang bsdar_errc(bsdar, EX_DATAERR, 0, "%s", 288d192f3d3SKai Wang archive_error_string(a)); 289d192f3d3SKai Wang if (r == ARCHIVE_EOF) 290d192f3d3SKai Wang break; 291d192f3d3SKai Wang if (r == ARCHIVE_WARN || r == ARCHIVE_RETRY) 292d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s", archive_error_string(a)); 293d192f3d3SKai Wang if (r == ARCHIVE_RETRY) { 294d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "Retrying..."); 295d192f3d3SKai Wang continue; 296d192f3d3SKai Wang } 297d192f3d3SKai Wang 298d192f3d3SKai Wang name = archive_entry_pathname(entry); 299d192f3d3SKai Wang 300d192f3d3SKai Wang /* 301d192f3d3SKai Wang * skip pseudo members. 302d192f3d3SKai Wang */ 303d192f3d3SKai Wang if (strcmp(name, "/") == 0 || strcmp(name, "//") == 0) 304d192f3d3SKai Wang continue; 305d192f3d3SKai Wang 3060c099281SKai Wang /* 3070c099281SKai Wang * If checkargv is set, only read those members specified 3080c099281SKai Wang * in argv. 3090c099281SKai Wang */ 3100c099281SKai Wang if (checkargv && bsdar->argc > 0) { 3110c099281SKai Wang find = 0; 3120c099281SKai Wang for(i = 0; i < bsdar->argc; i++) { 3130c099281SKai Wang av = &bsdar->argv[i]; 3140c099281SKai Wang if (*av == NULL) 3150c099281SKai Wang continue; 3160c099281SKai Wang if ((bname = basename(*av)) == NULL) 3170c099281SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 3180c099281SKai Wang "basename failed"); 3190c099281SKai Wang if (strcmp(bname, name) != 0) 3200c099281SKai Wang continue; 3210c099281SKai Wang 3220c099281SKai Wang *av = NULL; 3230c099281SKai Wang find = 1; 3240c099281SKai Wang break; 3250c099281SKai Wang } 3260c099281SKai Wang if (!find) 3270c099281SKai Wang continue; 3280c099281SKai Wang } 3290c099281SKai Wang 330d192f3d3SKai Wang size = archive_entry_size(entry); 331d192f3d3SKai Wang 332517dbe2eSKai Wang if (size > 0) { 333d192f3d3SKai Wang if ((buff = malloc(size)) == NULL) 334517dbe2eSKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 335517dbe2eSKai Wang "malloc failed"); 336d192f3d3SKai Wang if (archive_read_data(a, buff, size) != (ssize_t)size) { 337517dbe2eSKai Wang bsdar_warnc(bsdar, 0, "%s", 338517dbe2eSKai Wang archive_error_string(a)); 339d192f3d3SKai Wang free(buff); 340d192f3d3SKai Wang continue; 341d192f3d3SKai Wang } 342517dbe2eSKai Wang } else 343517dbe2eSKai Wang buff = NULL; 344517dbe2eSKai Wang 345d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 346d192f3d3SKai Wang if (obj == NULL) 347d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 348d192f3d3SKai Wang obj->maddr = buff; 349d192f3d3SKai Wang if ((obj->name = strdup(name)) == NULL) 350d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 351d192f3d3SKai Wang obj->size = size; 352d192f3d3SKai Wang obj->uid = archive_entry_uid(entry); 353d192f3d3SKai Wang obj->gid = archive_entry_gid(entry); 354d192f3d3SKai Wang obj->md = archive_entry_mode(entry); 355d192f3d3SKai Wang obj->mtime = archive_entry_mtime(entry); 356d192f3d3SKai Wang obj->dev = 0; 357d192f3d3SKai Wang obj->ino = 0; 358d192f3d3SKai Wang 359d192f3d3SKai Wang /* 360d192f3d3SKai Wang * Objects from archive have obj->fd set to -1, 361d192f3d3SKai Wang * for the ease of cleaning up. 362d192f3d3SKai Wang */ 363d192f3d3SKai Wang obj->fd = -1; 364d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 365d192f3d3SKai Wang } 366d192f3d3SKai Wang AC(archive_read_close(a)); 367ebb8fc42SMartin Matuska AC(archive_read_free(a)); 3680c099281SKai Wang } 3690c099281SKai Wang 3700c099281SKai Wang /* 3710c099281SKai Wang * Determine the constitution of resulting archive. 3720c099281SKai Wang */ 3730c099281SKai Wang static void 3740c099281SKai Wang write_archive(struct bsdar *bsdar, char mode) 3750c099281SKai Wang { 3760c099281SKai Wang struct ar_obj *nobj, *obj, *obj_temp, *pos; 3770c099281SKai Wang struct stat sb; 3780c099281SKai Wang const char *bname; 3790c099281SKai Wang char **av; 3800c099281SKai Wang int i; 3810c099281SKai Wang 3820c099281SKai Wang TAILQ_INIT(&bsdar->v_obj); 3830c099281SKai Wang nobj = NULL; 3840c099281SKai Wang pos = NULL; 3850c099281SKai Wang memset(&sb, 0, sizeof(sb)); 3860c099281SKai Wang 3870c099281SKai Wang /* 3880c099281SKai Wang * Test if the specified archive exists, to figure out 3890c099281SKai Wang * whether we are creating one here. 3900c099281SKai Wang */ 3910c099281SKai Wang if (stat(bsdar->filename, &sb) != 0) { 3920c099281SKai Wang if (errno != ENOENT) { 3930c099281SKai Wang bsdar_warnc(bsdar, 0, "stat %s failed", 3940c099281SKai Wang bsdar->filename); 3950c099281SKai Wang return; 3960c099281SKai Wang } 3970c099281SKai Wang 3980c099281SKai Wang /* We do not create archive in mode 'd', 'm' and 's'. */ 3990c099281SKai Wang if (mode != 'r' && mode != 'q') { 4000c099281SKai Wang bsdar_warnc(bsdar, 0, "%s: no such file", 4010c099281SKai Wang bsdar->filename); 4020c099281SKai Wang return; 4030c099281SKai Wang } 4040c099281SKai Wang 4050c099281SKai Wang /* Issue a warning if -c is not specified when creating. */ 4060c099281SKai Wang if (!(bsdar->options & AR_C)) 4070c099281SKai Wang bsdar_warnc(bsdar, 0, "creating %s", bsdar->filename); 4080c099281SKai Wang goto new_archive; 4090c099281SKai Wang } 4100c099281SKai Wang 4110c099281SKai Wang /* 4120c099281SKai Wang * First read members from existing archive. 4130c099281SKai Wang */ 4140c099281SKai Wang read_objs(bsdar, bsdar->filename, 0); 415d192f3d3SKai Wang 416d192f3d3SKai Wang /* 417d192f3d3SKai Wang * For mode 's', no member will be moved, deleted or replaced. 418d192f3d3SKai Wang */ 419d192f3d3SKai Wang if (mode == 's') 420d192f3d3SKai Wang goto write_objs; 421d192f3d3SKai Wang 422d192f3d3SKai Wang /* 423cb0dad38SKai Wang * For mode 'q', we don't need to adjust existing members either. 424cb0dad38SKai Wang * Also, -a, -b and -i are ignored in this mode. New members are 425cb0dad38SKai Wang * always inserted at tail. 426cb0dad38SKai Wang */ 427cb0dad38SKai Wang if (mode == 'q') 428cb0dad38SKai Wang goto new_archive; 429cb0dad38SKai Wang 430cb0dad38SKai Wang /* 4310c099281SKai Wang * Mode 'A' adds the contents of another archive to the tail of 4320c099281SKai Wang * current archive. Note that mode 'A' is a special mode for the 4330c099281SKai Wang * ADDLIB command of the ar script mode. Currently there is no 4340c099281SKai Wang * access to this function from the ar command line mode. 4350c099281SKai Wang */ 4360c099281SKai Wang if (mode == 'A') { 4370c099281SKai Wang /* 4380c099281SKai Wang * Read objects from the target archive of ADDLIB command. 4396bef5d28SBenedict Reuschling * If there are members specified in argv, read those members 4400c099281SKai Wang * only, otherwise the entire archive will be read. 4410c099281SKai Wang */ 4420c099281SKai Wang read_objs(bsdar, bsdar->addlib, 1); 4430c099281SKai Wang goto write_objs; 4440c099281SKai Wang } 4450c099281SKai Wang 4460c099281SKai Wang /* 447d192f3d3SKai Wang * Try to find the position member specified by user. 448d192f3d3SKai Wang */ 449d192f3d3SKai Wang if (bsdar->options & AR_A || bsdar->options & AR_B) { 450d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 451d192f3d3SKai Wang if (strcmp(obj->name, bsdar->posarg) == 0) { 452d192f3d3SKai Wang pos = obj; 453d192f3d3SKai Wang break; 454d192f3d3SKai Wang } 455d192f3d3SKai Wang } 456d192f3d3SKai Wang 457d192f3d3SKai Wang /* 458d192f3d3SKai Wang * If can't find `pos' specified by user, 4596bef5d28SBenedict Reuschling * silently insert objects at tail. 460d192f3d3SKai Wang */ 461d192f3d3SKai Wang if (pos == NULL) 462d192f3d3SKai Wang bsdar->options &= ~(AR_A | AR_B); 463d192f3d3SKai Wang } 464d192f3d3SKai Wang 465d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 466d192f3d3SKai Wang av = &bsdar->argv[i]; 467d192f3d3SKai Wang 468d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 469d192f3d3SKai Wang if ((bname = basename(*av)) == NULL) 470d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 471d192f3d3SKai Wang "basename failed"); 472d192f3d3SKai Wang if (bsdar->options & AR_TR) { 473d192f3d3SKai Wang if (strncmp(bname, obj->name, _TRUNCATE_LEN)) 474d192f3d3SKai Wang continue; 475d192f3d3SKai Wang } else 476d192f3d3SKai Wang if (strcmp(bname, obj->name) != 0) 477d192f3d3SKai Wang continue; 478d192f3d3SKai Wang 479d192f3d3SKai Wang if (mode == 'r') { 480d192f3d3SKai Wang /* 481d192f3d3SKai Wang * if the new member is not qualified 482d192f3d3SKai Wang * to replace the old one, skip it. 483d192f3d3SKai Wang */ 484d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 485d192f3d3SKai Wang obj->mtime); 486d192f3d3SKai Wang if (nobj == NULL) 487d192f3d3SKai Wang goto skip_obj; 488d192f3d3SKai Wang } 489d192f3d3SKai Wang 490d192f3d3SKai Wang if (bsdar->options & AR_V) 491d192f3d3SKai Wang (void)fprintf(stdout, "%c - %s\n", mode, 492d192f3d3SKai Wang *av); 493d192f3d3SKai Wang 494d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 495cf933f5bSEd Maste if (mode == 'd' || mode == 'r') 496cf933f5bSEd Maste free_obj(bsdar, obj); 497d192f3d3SKai Wang 498d192f3d3SKai Wang if (mode == 'm') 499d192f3d3SKai Wang insert_obj(bsdar, obj, pos); 500d192f3d3SKai Wang if (mode == 'r') 501d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 502d192f3d3SKai Wang 503d192f3d3SKai Wang skip_obj: 504d192f3d3SKai Wang *av = NULL; 505d192f3d3SKai Wang break; 506d192f3d3SKai Wang } 507d192f3d3SKai Wang 508d192f3d3SKai Wang } 509d192f3d3SKai Wang 510d192f3d3SKai Wang new_archive: 511d192f3d3SKai Wang /* 512d192f3d3SKai Wang * When operating in mode 'r', directly add those user specified 513cb0dad38SKai Wang * objects which do not exist in current archive. When operating 514cb0dad38SKai Wang * in mode 'q', all objects specified in command line args are 515cb0dad38SKai Wang * appended to the archive, without comparing with existing ones. 516d192f3d3SKai Wang */ 517d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 518d192f3d3SKai Wang av = &bsdar->argv[i]; 519cb0dad38SKai Wang if (*av != NULL && (mode == 'r' || mode == 'q')) { 520d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 0); 521d192f3d3SKai Wang if (nobj != NULL) 522d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 523d192f3d3SKai Wang if (bsdar->options & AR_V && nobj != NULL) 524d192f3d3SKai Wang (void)fprintf(stdout, "a - %s\n", *av); 525d192f3d3SKai Wang *av = NULL; 526d192f3d3SKai Wang } 527d192f3d3SKai Wang } 528d192f3d3SKai Wang 529d192f3d3SKai Wang write_objs: 530d192f3d3SKai Wang write_objs(bsdar); 531d192f3d3SKai Wang write_cleanup(bsdar); 532d192f3d3SKai Wang } 533d192f3d3SKai Wang 534d192f3d3SKai Wang /* 535d192f3d3SKai Wang * Memory cleaning up. 536d192f3d3SKai Wang */ 537d192f3d3SKai Wang static void 538d192f3d3SKai Wang write_cleanup(struct bsdar *bsdar) 539d192f3d3SKai Wang { 540d192f3d3SKai Wang struct ar_obj *obj, *obj_temp; 541d192f3d3SKai Wang 542d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 543d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 544cf933f5bSEd Maste free_obj(bsdar, obj); 545d192f3d3SKai Wang } 546d192f3d3SKai Wang 547d192f3d3SKai Wang free(bsdar->as); 548d192f3d3SKai Wang free(bsdar->s_so); 549d192f3d3SKai Wang free(bsdar->s_sn); 550d192f3d3SKai Wang bsdar->as = NULL; 551d192f3d3SKai Wang bsdar->s_so = NULL; 552d192f3d3SKai Wang bsdar->s_sn = NULL; 553d192f3d3SKai Wang } 554d192f3d3SKai Wang 555d192f3d3SKai Wang /* 55601715d8cSEd Maste * Fault in the buffer prior to writing as a workaround for poor performance 55701715d8cSEd Maste * due to interaction with kernel fs deadlock avoidance code. See the comment 55801715d8cSEd Maste * above vn_io_fault_doio() in sys/kern/vfs_vnops.c for details of the issue. 55901715d8cSEd Maste */ 56001715d8cSEd Maste static void 56101715d8cSEd Maste prefault_buffer(const char *buf, size_t s) 56201715d8cSEd Maste { 56301715d8cSEd Maste volatile const char *p; 56401715d8cSEd Maste size_t page_size; 56501715d8cSEd Maste 56601715d8cSEd Maste if (s == 0) 56701715d8cSEd Maste return; 56801715d8cSEd Maste page_size = sysconf(_SC_PAGESIZE); 56901715d8cSEd Maste for (p = buf; p < buf + s; p += page_size) 57001715d8cSEd Maste *p; 57101715d8cSEd Maste /* 57201715d8cSEd Maste * Ensure we touch the last page as well, in case the buffer is not 57301715d8cSEd Maste * page-aligned. 57401715d8cSEd Maste */ 57501715d8cSEd Maste *(volatile const char *)(buf + s - 1); 57601715d8cSEd Maste } 57701715d8cSEd Maste 57801715d8cSEd Maste /* 579d192f3d3SKai Wang * Wrapper for archive_write_data(). 580d192f3d3SKai Wang */ 581d192f3d3SKai Wang static void 582d192f3d3SKai Wang write_data(struct bsdar *bsdar, struct archive *a, const void *buf, size_t s) 583d192f3d3SKai Wang { 58401715d8cSEd Maste prefault_buffer(buf, s); 585d192f3d3SKai Wang if (archive_write_data(a, buf, s) != (ssize_t)s) 586d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "%s", 587d192f3d3SKai Wang archive_error_string(a)); 588d192f3d3SKai Wang } 589d192f3d3SKai Wang 590d192f3d3SKai Wang /* 591d192f3d3SKai Wang * Write the resulting archive members. 592d192f3d3SKai Wang */ 593d192f3d3SKai Wang static void 594d192f3d3SKai Wang write_objs(struct bsdar *bsdar) 595d192f3d3SKai Wang { 596d192f3d3SKai Wang struct ar_obj *obj; 597d192f3d3SKai Wang struct archive *a; 598d192f3d3SKai Wang struct archive_entry *entry; 599d192f3d3SKai Wang size_t s_sz; /* size of archive symbol table. */ 600d192f3d3SKai Wang size_t pm_sz; /* size of pseudo members */ 601d192f3d3SKai Wang int i, nr; 602d192f3d3SKai Wang 603d192f3d3SKai Wang if (elf_version(EV_CURRENT) == EV_NONE) 604d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, 605d192f3d3SKai Wang "ELF library initialization failed: %s", elf_errmsg(-1)); 606d192f3d3SKai Wang 607d192f3d3SKai Wang bsdar->rela_off = 0; 608d192f3d3SKai Wang 609d192f3d3SKai Wang /* Create archive symbol table and archive string table, if need. */ 610d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 61147203048SKai Wang if (!(bsdar->options & AR_SS) && obj->maddr != NULL) 612d192f3d3SKai Wang create_symtab_entry(bsdar, obj->maddr, obj->size); 613d192f3d3SKai Wang if (strlen(obj->name) > _MAXNAMELEN_SVR4) 614d192f3d3SKai Wang add_to_ar_str_table(bsdar, obj->name); 615d192f3d3SKai Wang bsdar->rela_off += _ARHDR_LEN + obj->size + obj->size % 2; 616d192f3d3SKai Wang } 617d192f3d3SKai Wang 618d192f3d3SKai Wang /* 619d192f3d3SKai Wang * Pad the symbol name string table. It is treated specially because 620d192f3d3SKai Wang * symbol name table should be padded by a '\0', not the common '\n' 621d192f3d3SKai Wang * for other members. The size of sn table includes the pad bit. 622d192f3d3SKai Wang */ 623d192f3d3SKai Wang if (bsdar->s_cnt != 0 && bsdar->s_sn_sz % 2 != 0) 624d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 625d192f3d3SKai Wang 626d192f3d3SKai Wang /* 627d192f3d3SKai Wang * Archive string table is padded by a "\n" as the normal members. 628517dbe2eSKai Wang * The difference is that the size of archive string table counts 629d192f3d3SKai Wang * in the pad bit, while normal members' size fileds do not. 630d192f3d3SKai Wang */ 631d192f3d3SKai Wang if (bsdar->as != NULL && bsdar->as_sz % 2 != 0) 632d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 633d192f3d3SKai Wang 634d192f3d3SKai Wang /* 635d192f3d3SKai Wang * If there is a symbol table, calculate the size of pseudo members, 636517dbe2eSKai Wang * convert previously stored relative offsets to absolute ones, and 637d192f3d3SKai Wang * then make them Big Endian. 638d192f3d3SKai Wang * 639d192f3d3SKai Wang * absolute_offset = htobe32(relative_offset + size_of_pseudo_members) 640d192f3d3SKai Wang */ 641d192f3d3SKai Wang 642d192f3d3SKai Wang if (bsdar->s_cnt != 0) { 643d192f3d3SKai Wang s_sz = (bsdar->s_cnt + 1) * sizeof(uint32_t) + bsdar->s_sn_sz; 644d192f3d3SKai Wang pm_sz = _ARMAG_LEN + (_ARHDR_LEN + s_sz); 645d192f3d3SKai Wang if (bsdar->as != NULL) 646d192f3d3SKai Wang pm_sz += _ARHDR_LEN + bsdar->as_sz; 647d192f3d3SKai Wang for (i = 0; (size_t)i < bsdar->s_cnt; i++) 648d192f3d3SKai Wang *(bsdar->s_so + i) = htobe32(*(bsdar->s_so + i) + 649d192f3d3SKai Wang pm_sz); 650d192f3d3SKai Wang } 651d192f3d3SKai Wang 652d192f3d3SKai Wang if ((a = archive_write_new()) == NULL) 653d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_write_new failed"); 654d192f3d3SKai Wang 655d192f3d3SKai Wang archive_write_set_format_ar_svr4(a); 656d192f3d3SKai Wang 657d192f3d3SKai Wang AC(archive_write_open_filename(a, bsdar->filename)); 658d192f3d3SKai Wang 659d192f3d3SKai Wang /* 660d192f3d3SKai Wang * write the archive symbol table, if there is one. 661d192f3d3SKai Wang * If options -s is explicitly specified or we are invoked 662d192f3d3SKai Wang * as ranlib, write the symbol table even if it is empty. 663d192f3d3SKai Wang */ 664d192f3d3SKai Wang if ((bsdar->s_cnt != 0 && !(bsdar->options & AR_SS)) || 665d192f3d3SKai Wang bsdar->options & AR_S) { 666d192f3d3SKai Wang entry = archive_entry_new(); 667d192f3d3SKai Wang archive_entry_copy_pathname(entry, "/"); 6680e479ac8STim Kientzle if ((bsdar->options & AR_D) == 0) 669d192f3d3SKai Wang archive_entry_set_mtime(entry, time(NULL), 0); 670d192f3d3SKai Wang archive_entry_set_size(entry, (bsdar->s_cnt + 1) * 671d192f3d3SKai Wang sizeof(uint32_t) + bsdar->s_sn_sz); 672d192f3d3SKai Wang AC(archive_write_header(a, entry)); 673d192f3d3SKai Wang nr = htobe32(bsdar->s_cnt); 674d192f3d3SKai Wang write_data(bsdar, a, &nr, sizeof(uint32_t)); 675d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_so, sizeof(uint32_t) * 676d192f3d3SKai Wang bsdar->s_cnt); 677d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_sn, bsdar->s_sn_sz); 678d192f3d3SKai Wang archive_entry_free(entry); 679d192f3d3SKai Wang } 680d192f3d3SKai Wang 681d192f3d3SKai Wang /* write the archive string table, if any. */ 682d192f3d3SKai Wang if (bsdar->as != NULL) { 683d192f3d3SKai Wang entry = archive_entry_new(); 684d192f3d3SKai Wang archive_entry_copy_pathname(entry, "//"); 685d192f3d3SKai Wang archive_entry_set_size(entry, bsdar->as_sz); 686d192f3d3SKai Wang AC(archive_write_header(a, entry)); 687d192f3d3SKai Wang write_data(bsdar, a, bsdar->as, bsdar->as_sz); 688d192f3d3SKai Wang archive_entry_free(entry); 689d192f3d3SKai Wang } 690d192f3d3SKai Wang 691d192f3d3SKai Wang /* write normal members. */ 692d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 693d192f3d3SKai Wang entry = archive_entry_new(); 694d192f3d3SKai Wang archive_entry_copy_pathname(entry, obj->name); 695d192f3d3SKai Wang archive_entry_set_uid(entry, obj->uid); 696d192f3d3SKai Wang archive_entry_set_gid(entry, obj->gid); 697d192f3d3SKai Wang archive_entry_set_mode(entry, obj->md); 698d192f3d3SKai Wang archive_entry_set_size(entry, obj->size); 699d192f3d3SKai Wang archive_entry_set_mtime(entry, obj->mtime, 0); 700d192f3d3SKai Wang archive_entry_set_dev(entry, obj->dev); 701d192f3d3SKai Wang archive_entry_set_ino(entry, obj->ino); 702d192f3d3SKai Wang archive_entry_set_filetype(entry, AE_IFREG); 703d192f3d3SKai Wang AC(archive_write_header(a, entry)); 704d192f3d3SKai Wang write_data(bsdar, a, obj->maddr, obj->size); 705d192f3d3SKai Wang archive_entry_free(entry); 706d192f3d3SKai Wang } 707d192f3d3SKai Wang 708d192f3d3SKai Wang AC(archive_write_close(a)); 709ebb8fc42SMartin Matuska AC(archive_write_free(a)); 710d192f3d3SKai Wang } 711d192f3d3SKai Wang 712d192f3d3SKai Wang /* 713d192f3d3SKai Wang * Extract global symbols from ELF binary members. 714d192f3d3SKai Wang */ 715d192f3d3SKai Wang static void 716d192f3d3SKai Wang create_symtab_entry(struct bsdar *bsdar, void *maddr, size_t size) 717d192f3d3SKai Wang { 718d192f3d3SKai Wang Elf *e; 719d192f3d3SKai Wang Elf_Scn *scn; 720d192f3d3SKai Wang GElf_Shdr shdr; 721d192f3d3SKai Wang GElf_Sym sym; 722d192f3d3SKai Wang Elf_Data *data; 723d192f3d3SKai Wang char *name; 724d192f3d3SKai Wang size_t n, shstrndx; 725d192f3d3SKai Wang int elferr, tabndx, len, i; 726d192f3d3SKai Wang 727d192f3d3SKai Wang if ((e = elf_memory(maddr, size)) == NULL) { 728d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_memory() failed: %s", 729d192f3d3SKai Wang elf_errmsg(-1)); 730d192f3d3SKai Wang return; 731d192f3d3SKai Wang } 732d192f3d3SKai Wang if (elf_kind(e) != ELF_K_ELF) { 7336bef5d28SBenedict Reuschling /* Silently ignore non-elf member. */ 734d192f3d3SKai Wang elf_end(e); 735d192f3d3SKai Wang return; 736d192f3d3SKai Wang } 737d192f3d3SKai Wang if (elf_getshstrndx(e, &shstrndx) == 0) { 738d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_getshstrndx failed: %s", 739d192f3d3SKai Wang elf_errmsg(-1)); 740d192f3d3SKai Wang elf_end(e); 741d192f3d3SKai Wang return; 742d192f3d3SKai Wang } 743d192f3d3SKai Wang 744d192f3d3SKai Wang tabndx = -1; 745d192f3d3SKai Wang scn = NULL; 746d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 747d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 748d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 749d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 750d192f3d3SKai Wang continue; 751d192f3d3SKai Wang } 752d192f3d3SKai Wang if ((name = elf_strptr(e, shstrndx, shdr.sh_name)) == NULL) { 753d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 754d192f3d3SKai Wang "elf_strptr failed: %s", elf_errmsg(-1)); 755d192f3d3SKai Wang continue; 756d192f3d3SKai Wang } 757d192f3d3SKai Wang if (strcmp(name, ".strtab") == 0) { 758d192f3d3SKai Wang tabndx = elf_ndxscn(scn); 759d192f3d3SKai Wang break; 760d192f3d3SKai Wang } 761d192f3d3SKai Wang } 762d192f3d3SKai Wang elferr = elf_errno(); 763d192f3d3SKai Wang if (elferr != 0) 764d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_nextscn failed: %s", 765d192f3d3SKai Wang elf_errmsg(elferr)); 766d192f3d3SKai Wang if (tabndx == -1) { 767d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "can't find .strtab section"); 768d192f3d3SKai Wang elf_end(e); 769d192f3d3SKai Wang return; 770d192f3d3SKai Wang } 771d192f3d3SKai Wang 772d192f3d3SKai Wang scn = NULL; 773d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 774d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 775d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 776d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 777d192f3d3SKai Wang continue; 778d192f3d3SKai Wang } 779d192f3d3SKai Wang if (shdr.sh_type != SHT_SYMTAB) 780d192f3d3SKai Wang continue; 781d192f3d3SKai Wang 782d192f3d3SKai Wang data = NULL; 783d192f3d3SKai Wang n = 0; 784d192f3d3SKai Wang while (n < shdr.sh_size && 785d192f3d3SKai Wang (data = elf_getdata(scn, data)) != NULL) { 786d192f3d3SKai Wang len = data->d_size / shdr.sh_entsize; 787d192f3d3SKai Wang for (i = 0; i < len; i++) { 788d192f3d3SKai Wang if (gelf_getsym(data, i, &sym) != &sym) { 789d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 790d192f3d3SKai Wang "gelf_getsym failed: %s", 791d192f3d3SKai Wang elf_errmsg(-1)); 792d192f3d3SKai Wang continue; 793d192f3d3SKai Wang } 794d192f3d3SKai Wang 795d192f3d3SKai Wang /* keep only global or weak symbols */ 796d192f3d3SKai Wang if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL && 797d192f3d3SKai Wang GELF_ST_BIND(sym.st_info) != STB_WEAK) 798d192f3d3SKai Wang continue; 799d192f3d3SKai Wang 800d192f3d3SKai Wang /* keep only defined symbols */ 801d192f3d3SKai Wang if (sym.st_shndx == SHN_UNDEF) 802d192f3d3SKai Wang continue; 803d192f3d3SKai Wang 804d192f3d3SKai Wang if ((name = elf_strptr(e, tabndx, 805d192f3d3SKai Wang sym.st_name)) == NULL) { 806d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 807d192f3d3SKai Wang "elf_strptr failed: %s", 808d192f3d3SKai Wang elf_errmsg(-1)); 809d192f3d3SKai Wang continue; 810d192f3d3SKai Wang } 811d192f3d3SKai Wang 812d192f3d3SKai Wang add_to_ar_sym_table(bsdar, name); 813d192f3d3SKai Wang } 814d192f3d3SKai Wang } 815d192f3d3SKai Wang } 816d192f3d3SKai Wang elferr = elf_errno(); 817d192f3d3SKai Wang if (elferr != 0) 818d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_nextscn failed: %s", 819d192f3d3SKai Wang elf_errmsg(elferr)); 820d192f3d3SKai Wang 821d192f3d3SKai Wang elf_end(e); 822d192f3d3SKai Wang } 823d192f3d3SKai Wang 824d192f3d3SKai Wang /* 825d192f3d3SKai Wang * Append to the archive string table buffer. 826d192f3d3SKai Wang */ 827d192f3d3SKai Wang static void 828d192f3d3SKai Wang add_to_ar_str_table(struct bsdar *bsdar, const char *name) 829d192f3d3SKai Wang { 830d192f3d3SKai Wang 831d192f3d3SKai Wang if (bsdar->as == NULL) { 832d192f3d3SKai Wang bsdar->as_cap = _INIT_AS_CAP; 833d192f3d3SKai Wang bsdar->as_sz = 0; 834d192f3d3SKai Wang if ((bsdar->as = malloc(bsdar->as_cap)) == NULL) 835d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 836d192f3d3SKai Wang } 837d192f3d3SKai Wang 838d192f3d3SKai Wang /* 839d192f3d3SKai Wang * The space required for holding one member name in as table includes: 840d192f3d3SKai Wang * strlen(name) + (1 for '/') + (1 for '\n') + (possibly 1 for padding). 841d192f3d3SKai Wang */ 842517dbe2eSKai Wang while (bsdar->as_sz + strlen(name) + 3 > bsdar->as_cap) { 843d192f3d3SKai Wang bsdar->as_cap *= 2; 844d192f3d3SKai Wang bsdar->as = realloc(bsdar->as, bsdar->as_cap); 845d192f3d3SKai Wang if (bsdar->as == NULL) 846d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 847d192f3d3SKai Wang } 848d192f3d3SKai Wang strncpy(&bsdar->as[bsdar->as_sz], name, strlen(name)); 849d192f3d3SKai Wang bsdar->as_sz += strlen(name); 850d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '/'; 851d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 852d192f3d3SKai Wang } 853d192f3d3SKai Wang 854d192f3d3SKai Wang /* 855d192f3d3SKai Wang * Append to the archive symbol table buffer. 856d192f3d3SKai Wang */ 857d192f3d3SKai Wang static void 858d192f3d3SKai Wang add_to_ar_sym_table(struct bsdar *bsdar, const char *name) 859d192f3d3SKai Wang { 860d192f3d3SKai Wang 861d192f3d3SKai Wang if (bsdar->s_so == NULL) { 862d192f3d3SKai Wang if ((bsdar->s_so = malloc(_INIT_SYMOFF_CAP)) == 863d192f3d3SKai Wang NULL) 864d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 865d192f3d3SKai Wang bsdar->s_so_cap = _INIT_SYMOFF_CAP; 866d192f3d3SKai Wang bsdar->s_cnt = 0; 867d192f3d3SKai Wang } 868d192f3d3SKai Wang 869d192f3d3SKai Wang if (bsdar->s_sn == NULL) { 870d192f3d3SKai Wang if ((bsdar->s_sn = malloc(_INIT_SYMNAME_CAP)) == NULL) 871d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 872d192f3d3SKai Wang bsdar->s_sn_cap = _INIT_SYMNAME_CAP; 873d192f3d3SKai Wang bsdar->s_sn_sz = 0; 874d192f3d3SKai Wang } 875d192f3d3SKai Wang 876d192f3d3SKai Wang if (bsdar->s_cnt * sizeof(uint32_t) >= bsdar->s_so_cap) { 877d192f3d3SKai Wang bsdar->s_so_cap *= 2; 878d192f3d3SKai Wang bsdar->s_so = realloc(bsdar->s_so, bsdar->s_so_cap); 879d192f3d3SKai Wang if (bsdar->s_so == NULL) 880d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 881d192f3d3SKai Wang } 882d192f3d3SKai Wang bsdar->s_so[bsdar->s_cnt] = bsdar->rela_off; 883d192f3d3SKai Wang bsdar->s_cnt++; 884d192f3d3SKai Wang 885d192f3d3SKai Wang /* 886d192f3d3SKai Wang * The space required for holding one symbol name in sn table includes: 887d192f3d3SKai Wang * strlen(name) + (1 for '\n') + (possibly 1 for padding). 888d192f3d3SKai Wang */ 889517dbe2eSKai Wang while (bsdar->s_sn_sz + strlen(name) + 2 > bsdar->s_sn_cap) { 890d192f3d3SKai Wang bsdar->s_sn_cap *= 2; 891d192f3d3SKai Wang bsdar->s_sn = realloc(bsdar->s_sn, bsdar->s_sn_cap); 892d192f3d3SKai Wang if (bsdar->s_sn == NULL) 893d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 894d192f3d3SKai Wang } 895d192f3d3SKai Wang strncpy(&bsdar->s_sn[bsdar->s_sn_sz], name, strlen(name)); 896d192f3d3SKai Wang bsdar->s_sn_sz += strlen(name); 897d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 898d192f3d3SKai Wang } 899