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> 44d192f3d3SKai Wang 45d192f3d3SKai Wang #include "ar.h" 46d192f3d3SKai Wang 47d192f3d3SKai Wang #define _ARMAG_LEN 8 /* length of ar magic string */ 48d192f3d3SKai Wang #define _ARHDR_LEN 60 /* length of ar header */ 49d192f3d3SKai Wang #define _INIT_AS_CAP 128 /* initial archive string table size */ 50d192f3d3SKai Wang #define _INIT_SYMOFF_CAP (256*(sizeof(uint32_t))) /* initial so table size */ 51d192f3d3SKai Wang #define _INIT_SYMNAME_CAP 1024 /* initial sn table size */ 52d192f3d3SKai Wang #define _MAXNAMELEN_SVR4 15 /* max member name length in svr4 variant */ 53d192f3d3SKai Wang #define _TRUNCATE_LEN 15 /* number of bytes to keep for member name */ 54d192f3d3SKai Wang 55d192f3d3SKai Wang static void add_to_ar_str_table(struct bsdar *bsdar, const char *name); 56d192f3d3SKai Wang static void add_to_ar_sym_table(struct bsdar *bsdar, const char *name); 57d192f3d3SKai Wang static struct ar_obj *create_obj_from_file(struct bsdar *bsdar, 58d192f3d3SKai Wang const char *name, time_t mtime); 59d192f3d3SKai Wang static void create_symtab_entry(struct bsdar *bsdar, void *maddr, 60d192f3d3SKai Wang size_t size); 61d192f3d3SKai Wang static void insert_obj(struct bsdar *bsdar, struct ar_obj *obj, 62d192f3d3SKai Wang struct ar_obj *pos); 63d192f3d3SKai Wang static void write_archive(struct bsdar *bsdar, char mode); 64d192f3d3SKai Wang static void write_cleanup(struct bsdar *bsdar); 65d192f3d3SKai Wang static void write_data(struct bsdar *bsdar, struct archive *a, 66d192f3d3SKai Wang const void *buf, size_t s); 67d192f3d3SKai Wang static void write_objs(struct bsdar *bsdar); 68d192f3d3SKai Wang 69d192f3d3SKai Wang void 70d192f3d3SKai Wang ar_mode_d(struct bsdar *bsdar) 71d192f3d3SKai Wang { 72d192f3d3SKai Wang 73d192f3d3SKai Wang write_archive(bsdar, 'd'); 74d192f3d3SKai Wang } 75d192f3d3SKai Wang 76d192f3d3SKai Wang void 77d192f3d3SKai Wang ar_mode_m(struct bsdar *bsdar) 78d192f3d3SKai Wang { 79d192f3d3SKai Wang 80d192f3d3SKai Wang write_archive(bsdar, 'm'); 81d192f3d3SKai Wang } 82d192f3d3SKai Wang 83d192f3d3SKai Wang void 84d192f3d3SKai Wang ar_mode_r(struct bsdar *bsdar) 85d192f3d3SKai Wang { 86d192f3d3SKai Wang 87d192f3d3SKai Wang write_archive(bsdar, 'r'); 88d192f3d3SKai Wang } 89d192f3d3SKai Wang 90d192f3d3SKai Wang void 91d192f3d3SKai Wang ar_mode_s(struct bsdar *bsdar) 92d192f3d3SKai Wang { 93d192f3d3SKai Wang 94d192f3d3SKai Wang write_archive(bsdar, 's'); 95d192f3d3SKai Wang } 96d192f3d3SKai Wang 97d192f3d3SKai Wang /* 98d192f3d3SKai Wang * Create object from file, return created obj upon success, or NULL 99d192f3d3SKai Wang * when an error occurs or the member is not newer than existing 100d192f3d3SKai Wang * one while -u is specifed. 101d192f3d3SKai Wang */ 102d192f3d3SKai Wang static struct ar_obj * 103d192f3d3SKai Wang create_obj_from_file(struct bsdar *bsdar, const char *name, time_t mtime) 104d192f3d3SKai Wang { 105d192f3d3SKai Wang struct ar_obj *obj; 106d192f3d3SKai Wang struct stat sb; 107d192f3d3SKai Wang const char *bname; 108d192f3d3SKai Wang 109d192f3d3SKai Wang if (name == NULL) 110d192f3d3SKai Wang return(NULL); 111d192f3d3SKai Wang 112d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 113d192f3d3SKai Wang if (obj == NULL) 114d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 115d192f3d3SKai Wang if ((obj->fd = open(name, O_RDONLY, 0)) < 0) { 116d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't open file: %s", name); 117d192f3d3SKai Wang free(obj); 118d192f3d3SKai Wang return(NULL); 119d192f3d3SKai Wang } 120d192f3d3SKai Wang 121d192f3d3SKai Wang if ((bname = basename(name)) == NULL) 122d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "basename failed"); 123d192f3d3SKai Wang if (bsdar->options & AR_TR && strlen(bname) > _TRUNCATE_LEN) { 124d192f3d3SKai Wang if ((obj->name = malloc(_TRUNCATE_LEN + 1)) == NULL) 125d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 126d192f3d3SKai Wang (void)strncpy(obj->name, bname, _TRUNCATE_LEN); 127d192f3d3SKai Wang obj->name[_TRUNCATE_LEN] = '\0'; 128d192f3d3SKai Wang } else 129d192f3d3SKai Wang if ((obj->name = strdup(bname)) == NULL) 130d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 131d192f3d3SKai Wang 132d192f3d3SKai Wang if (fstat(obj->fd, &sb) < 0) { 133d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't fstat file: %s", obj->name); 134d192f3d3SKai Wang goto giveup; 135d192f3d3SKai Wang } 136d192f3d3SKai Wang if (!S_ISREG(sb.st_mode)) { 137d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s is not an ordinary file", obj->name); 138d192f3d3SKai Wang goto giveup; 139d192f3d3SKai Wang } 140d192f3d3SKai Wang 141d192f3d3SKai Wang /* 142d192f3d3SKai Wang * When option '-u' is specified and member is not newer than the 143d192f3d3SKai Wang * existing one, the replace will not happen. While if mtime == 0, 144d192f3d3SKai Wang * which indicates that this is to "replace a none exist member", 145d192f3d3SKai Wang * the replace will proceed regardless of '-u'. 146d192f3d3SKai Wang */ 147d192f3d3SKai Wang if (mtime != 0 && bsdar->options & AR_U && sb.st_mtime <= mtime) 148d192f3d3SKai Wang goto giveup; 149d192f3d3SKai Wang 150d192f3d3SKai Wang obj->uid = sb.st_uid; 151d192f3d3SKai Wang obj->gid = sb.st_gid; 152d192f3d3SKai Wang obj->md = sb.st_mode; 153d192f3d3SKai Wang obj->size = sb.st_size; 154d192f3d3SKai Wang obj->mtime = sb.st_mtime; 155d192f3d3SKai Wang obj->dev = sb.st_dev; 156d192f3d3SKai Wang obj->ino = sb.st_ino; 15747203048SKai Wang 15847203048SKai Wang if (obj->size == 0) { 15947203048SKai Wang obj->maddr = NULL; 16047203048SKai Wang return(obj); 16147203048SKai Wang } 16247203048SKai Wang 163d192f3d3SKai Wang if ((obj->maddr = mmap(NULL, obj->size, PROT_READ, 164d192f3d3SKai Wang MAP_PRIVATE, obj->fd, (off_t)0)) == MAP_FAILED) { 165d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't mmap file: %s", obj->name); 166d192f3d3SKai Wang goto giveup; 167d192f3d3SKai Wang } 168d192f3d3SKai Wang if (close(obj->fd) < 0) 169d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 170d192f3d3SKai Wang obj->name); 171d192f3d3SKai Wang 172d192f3d3SKai Wang return(obj); 173d192f3d3SKai Wang 174d192f3d3SKai Wang giveup: 175d192f3d3SKai Wang if (close(obj->fd) < 0) 176d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 177d192f3d3SKai Wang obj->name); 178d192f3d3SKai Wang free(obj->name); 179d192f3d3SKai Wang free(obj); 180d192f3d3SKai Wang return(NULL); 181d192f3d3SKai Wang } 182d192f3d3SKai Wang 183d192f3d3SKai Wang /* 184d192f3d3SKai Wang * Insert obj to the tail, or before/after the pos obj. 185d192f3d3SKai Wang */ 186d192f3d3SKai Wang static void 187d192f3d3SKai Wang insert_obj(struct bsdar *bsdar, struct ar_obj *obj, struct ar_obj *pos) 188d192f3d3SKai Wang { 189d192f3d3SKai Wang if (obj == NULL) 190d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "try to insert a null obj"); 191d192f3d3SKai Wang 192d192f3d3SKai Wang if (pos == NULL || obj == pos) 193d192f3d3SKai Wang /* 194d192f3d3SKai Wang * If the object to move happens to be the posistion obj, 195d192f3d3SKai Wang * or if there is not a pos obj, move it to tail. 196d192f3d3SKai Wang */ 197d192f3d3SKai Wang goto tail; 198d192f3d3SKai Wang 199d192f3d3SKai Wang if (bsdar->options & AR_B) { 200d192f3d3SKai Wang TAILQ_INSERT_BEFORE(pos, obj, objs); 201d192f3d3SKai Wang return; 202d192f3d3SKai Wang } 203d192f3d3SKai Wang if (bsdar->options & AR_A) { 204d192f3d3SKai Wang TAILQ_INSERT_AFTER(&bsdar->v_obj, pos, obj, objs); 205d192f3d3SKai Wang return; 206d192f3d3SKai Wang } 207d192f3d3SKai Wang 208d192f3d3SKai Wang tail: 209d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 210d192f3d3SKai Wang 211d192f3d3SKai Wang } 212d192f3d3SKai Wang 213d192f3d3SKai Wang /* 214d192f3d3SKai Wang * Determine the constitution of resulting archive. 215d192f3d3SKai Wang */ 216d192f3d3SKai Wang static void 217d192f3d3SKai Wang write_archive(struct bsdar *bsdar, char mode) 218d192f3d3SKai Wang { 219d192f3d3SKai Wang struct archive *a; 220d192f3d3SKai Wang struct archive_entry *entry; 221d192f3d3SKai Wang struct ar_obj *nobj, *obj, *obj_temp, *pos; 222d192f3d3SKai Wang struct stat sb; 223d192f3d3SKai Wang const char *name; 224d192f3d3SKai Wang const char *bname; 225d192f3d3SKai Wang char *buff; 226d192f3d3SKai Wang char **av; 227d192f3d3SKai Wang size_t size; 228d192f3d3SKai Wang int i, r; 229d192f3d3SKai Wang 230d192f3d3SKai Wang TAILQ_INIT(&bsdar->v_obj); 231d192f3d3SKai Wang nobj = NULL; 232d192f3d3SKai Wang pos = NULL; 233d192f3d3SKai Wang memset(&sb, 0, sizeof(sb)); 234d192f3d3SKai Wang 235d192f3d3SKai Wang /* By default, no compression is assumed. */ 236d192f3d3SKai Wang bsdar->compression = ARCHIVE_COMPRESSION_NONE; 237d192f3d3SKai Wang 238d192f3d3SKai Wang /* 239d192f3d3SKai Wang * Test if the specified archive exists, to figure out 240d192f3d3SKai Wang * whether we are creating one here. 241d192f3d3SKai Wang */ 242d192f3d3SKai Wang if (stat(bsdar->filename, &sb) != 0) { 243d192f3d3SKai Wang if (errno != ENOENT) { 244d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "stat %s failed", 245d192f3d3SKai Wang bsdar->filename); 246d192f3d3SKai Wang return; 247d192f3d3SKai Wang } 248d192f3d3SKai Wang 249d192f3d3SKai Wang /* We do not create archive in mode 'd', 'm' and 's'. */ 250d192f3d3SKai Wang if (mode != 'r') { 251d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s: no such file", 252d192f3d3SKai Wang bsdar->filename); 253d192f3d3SKai Wang return; 254d192f3d3SKai Wang } 255d192f3d3SKai Wang 256d192f3d3SKai Wang /* Issue a warning if -c is not specified when creating. */ 257d192f3d3SKai Wang if (!(bsdar->options & AR_C)) 258d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "creating %s", bsdar->filename); 259d192f3d3SKai Wang goto new_archive; 260d192f3d3SKai Wang } 261d192f3d3SKai Wang 262d192f3d3SKai Wang /* 263d192f3d3SKai Wang * First read members from existing archive. 264d192f3d3SKai Wang */ 265d192f3d3SKai Wang if ((a = archive_read_new()) == NULL) 266d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_read_new failed"); 267d192f3d3SKai Wang archive_read_support_compression_all(a); 268d192f3d3SKai Wang archive_read_support_format_ar(a); 269d192f3d3SKai Wang AC(archive_read_open_filename(a, bsdar->filename, DEF_BLKSZ)); 270d192f3d3SKai Wang for (;;) { 271d192f3d3SKai Wang r = archive_read_next_header(a, &entry); 272d192f3d3SKai Wang if (r == ARCHIVE_FATAL) 273d192f3d3SKai Wang bsdar_errc(bsdar, EX_DATAERR, 0, "%s", 274d192f3d3SKai Wang archive_error_string(a)); 275d192f3d3SKai Wang if (r == ARCHIVE_EOF) 276d192f3d3SKai Wang break; 277d192f3d3SKai Wang if (r == ARCHIVE_WARN || r == ARCHIVE_RETRY) 278d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s", archive_error_string(a)); 279d192f3d3SKai Wang if (r == ARCHIVE_RETRY) { 280d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "Retrying..."); 281d192f3d3SKai Wang continue; 282d192f3d3SKai Wang } 283d192f3d3SKai Wang 284d192f3d3SKai Wang /* 285d192f3d3SKai Wang * Remember the compression mode of existing archive. 286d192f3d3SKai Wang * If neither -j nor -z is specified, this mode will 287d192f3d3SKai Wang * be used for resulting archive. 288d192f3d3SKai Wang */ 289d192f3d3SKai Wang bsdar->compression = archive_compression(a); 290d192f3d3SKai Wang 291d192f3d3SKai Wang name = archive_entry_pathname(entry); 292d192f3d3SKai Wang 293d192f3d3SKai Wang /* 294d192f3d3SKai Wang * skip pseudo members. 295d192f3d3SKai Wang */ 296d192f3d3SKai Wang if (strcmp(name, "/") == 0 || strcmp(name, "//") == 0) 297d192f3d3SKai Wang continue; 298d192f3d3SKai Wang 299d192f3d3SKai Wang size = archive_entry_size(entry); 300d192f3d3SKai Wang 301d192f3d3SKai Wang if ((buff = malloc(size)) == NULL) 302d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 303d192f3d3SKai Wang if (archive_read_data(a, buff, size) != (ssize_t)size) { 304d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s", archive_error_string(a)); 305d192f3d3SKai Wang free(buff); 306d192f3d3SKai Wang continue; 307d192f3d3SKai Wang } 308d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 309d192f3d3SKai Wang if (obj == NULL) 310d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 311d192f3d3SKai Wang obj->maddr = buff; 312d192f3d3SKai Wang if ((obj->name = strdup(name)) == NULL) 313d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 314d192f3d3SKai Wang obj->size = size; 315d192f3d3SKai Wang obj->uid = archive_entry_uid(entry); 316d192f3d3SKai Wang obj->gid = archive_entry_gid(entry); 317d192f3d3SKai Wang obj->md = archive_entry_mode(entry); 318d192f3d3SKai Wang obj->mtime = archive_entry_mtime(entry); 319d192f3d3SKai Wang obj->dev = 0; 320d192f3d3SKai Wang obj->ino = 0; 321d192f3d3SKai Wang 322d192f3d3SKai Wang /* 323d192f3d3SKai Wang * Objects from archive have obj->fd set to -1, 324d192f3d3SKai Wang * for the ease of cleaning up. 325d192f3d3SKai Wang */ 326d192f3d3SKai Wang obj->fd = -1; 327d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 328d192f3d3SKai Wang } 329d192f3d3SKai Wang AC(archive_read_close(a)); 330d192f3d3SKai Wang AC(archive_read_finish(a)); 331d192f3d3SKai Wang 332d192f3d3SKai Wang /* 333d192f3d3SKai Wang * For mode 's', no member will be moved, deleted or replaced. 334d192f3d3SKai Wang */ 335d192f3d3SKai Wang if (mode == 's') 336d192f3d3SKai Wang goto write_objs; 337d192f3d3SKai Wang 338d192f3d3SKai Wang /* 339d192f3d3SKai Wang * Try to find the position member specified by user. 340d192f3d3SKai Wang */ 341d192f3d3SKai Wang if (bsdar->options & AR_A || bsdar->options & AR_B) { 342d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 343d192f3d3SKai Wang if (strcmp(obj->name, bsdar->posarg) == 0) { 344d192f3d3SKai Wang pos = obj; 345d192f3d3SKai Wang break; 346d192f3d3SKai Wang } 347d192f3d3SKai Wang } 348d192f3d3SKai Wang 349d192f3d3SKai Wang /* 350d192f3d3SKai Wang * If can't find `pos' specified by user, 351d192f3d3SKai Wang * sliently insert objects at tail. 352d192f3d3SKai Wang */ 353d192f3d3SKai Wang if (pos == NULL) 354d192f3d3SKai Wang bsdar->options &= ~(AR_A | AR_B); 355d192f3d3SKai Wang } 356d192f3d3SKai Wang 357d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 358d192f3d3SKai Wang av = &bsdar->argv[i]; 359d192f3d3SKai Wang 360d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 361d192f3d3SKai Wang if ((bname = basename(*av)) == NULL) 362d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 363d192f3d3SKai Wang "basename failed"); 364d192f3d3SKai Wang if (bsdar->options & AR_TR) { 365d192f3d3SKai Wang if (strncmp(bname, obj->name, _TRUNCATE_LEN)) 366d192f3d3SKai Wang continue; 367d192f3d3SKai Wang } else 368d192f3d3SKai Wang if (strcmp(bname, obj->name) != 0) 369d192f3d3SKai Wang continue; 370d192f3d3SKai Wang 371d192f3d3SKai Wang if (mode == 'r') { 372d192f3d3SKai Wang /* 373d192f3d3SKai Wang * if the new member is not qualified 374d192f3d3SKai Wang * to replace the old one, skip it. 375d192f3d3SKai Wang */ 376d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 377d192f3d3SKai Wang obj->mtime); 378d192f3d3SKai Wang if (nobj == NULL) 379d192f3d3SKai Wang goto skip_obj; 380d192f3d3SKai Wang } 381d192f3d3SKai Wang 382d192f3d3SKai Wang if (bsdar->options & AR_V) 383d192f3d3SKai Wang (void)fprintf(stdout, "%c - %s\n", mode, 384d192f3d3SKai Wang *av); 385d192f3d3SKai Wang 386d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 387d192f3d3SKai Wang if (mode == 'd' || mode == 'r') { 388d192f3d3SKai Wang free(obj->maddr); 389d192f3d3SKai Wang free(obj->name); 390d192f3d3SKai Wang free(obj); 391d192f3d3SKai Wang } 392d192f3d3SKai Wang 393d192f3d3SKai Wang if (mode == 'm') 394d192f3d3SKai Wang insert_obj(bsdar, obj, pos); 395d192f3d3SKai Wang if (mode == 'r') 396d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 397d192f3d3SKai Wang 398d192f3d3SKai Wang skip_obj: 399d192f3d3SKai Wang *av = NULL; 400d192f3d3SKai Wang break; 401d192f3d3SKai Wang } 402d192f3d3SKai Wang 403d192f3d3SKai Wang } 404d192f3d3SKai Wang 405d192f3d3SKai Wang new_archive: 406d192f3d3SKai Wang /* 407d192f3d3SKai Wang * When operating in mode 'r', directly add those user specified 408d192f3d3SKai Wang * objects which do not exist in current archive. 409d192f3d3SKai Wang */ 410d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 411d192f3d3SKai Wang av = &bsdar->argv[i]; 412d192f3d3SKai Wang if (*av != NULL && mode == 'r') { 413d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 0); 414d192f3d3SKai Wang if (nobj != NULL) 415d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 416d192f3d3SKai Wang if (bsdar->options & AR_V && nobj != NULL) 417d192f3d3SKai Wang (void)fprintf(stdout, "a - %s\n", *av); 418d192f3d3SKai Wang *av = NULL; 419d192f3d3SKai Wang } 420d192f3d3SKai Wang } 421d192f3d3SKai Wang 422d192f3d3SKai Wang write_objs: 423d192f3d3SKai Wang write_objs(bsdar); 424d192f3d3SKai Wang write_cleanup(bsdar); 425d192f3d3SKai Wang } 426d192f3d3SKai Wang 427d192f3d3SKai Wang /* 428d192f3d3SKai Wang * Memory cleaning up. 429d192f3d3SKai Wang */ 430d192f3d3SKai Wang static void 431d192f3d3SKai Wang write_cleanup(struct bsdar *bsdar) 432d192f3d3SKai Wang { 433d192f3d3SKai Wang struct ar_obj *obj, *obj_temp; 434d192f3d3SKai Wang 435d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 436d192f3d3SKai Wang if (obj->fd == -1) 437d192f3d3SKai Wang free(obj->maddr); 438d192f3d3SKai Wang else 43947203048SKai Wang if (obj->maddr != NULL && munmap(obj->maddr, obj->size)) 440d192f3d3SKai Wang bsdar_warnc(bsdar, errno, 441d192f3d3SKai Wang "can't munmap file: %s", obj->name); 442d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 44347203048SKai Wang free(obj->name); 444d192f3d3SKai Wang free(obj); 445d192f3d3SKai Wang } 446d192f3d3SKai Wang 447d192f3d3SKai Wang free(bsdar->as); 448d192f3d3SKai Wang free(bsdar->s_so); 449d192f3d3SKai Wang free(bsdar->s_sn); 450d192f3d3SKai Wang bsdar->as = NULL; 451d192f3d3SKai Wang bsdar->s_so = NULL; 452d192f3d3SKai Wang bsdar->s_sn = NULL; 453d192f3d3SKai Wang } 454d192f3d3SKai Wang 455d192f3d3SKai Wang /* 456d192f3d3SKai Wang * Wrapper for archive_write_data(). 457d192f3d3SKai Wang */ 458d192f3d3SKai Wang static void 459d192f3d3SKai Wang write_data(struct bsdar *bsdar, struct archive *a, const void *buf, size_t s) 460d192f3d3SKai Wang { 461d192f3d3SKai Wang if (archive_write_data(a, buf, s) != (ssize_t)s) 462d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "%s", 463d192f3d3SKai Wang archive_error_string(a)); 464d192f3d3SKai Wang } 465d192f3d3SKai Wang 466d192f3d3SKai Wang /* 467d192f3d3SKai Wang * Write the resulting archive members. 468d192f3d3SKai Wang */ 469d192f3d3SKai Wang static void 470d192f3d3SKai Wang write_objs(struct bsdar *bsdar) 471d192f3d3SKai Wang { 472d192f3d3SKai Wang struct ar_obj *obj; 473d192f3d3SKai Wang struct archive *a; 474d192f3d3SKai Wang struct archive_entry *entry; 475d192f3d3SKai Wang size_t s_sz; /* size of archive symbol table. */ 476d192f3d3SKai Wang size_t pm_sz; /* size of pseudo members */ 477d192f3d3SKai Wang int i, nr; 478d192f3d3SKai Wang 479d192f3d3SKai Wang if (elf_version(EV_CURRENT) == EV_NONE) 480d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, 481d192f3d3SKai Wang "ELF library initialization failed: %s", elf_errmsg(-1)); 482d192f3d3SKai Wang 483d192f3d3SKai Wang bsdar->rela_off = 0; 484d192f3d3SKai Wang 485d192f3d3SKai Wang /* Create archive symbol table and archive string table, if need. */ 486d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 48747203048SKai Wang if (!(bsdar->options & AR_SS) && obj->maddr != NULL) 488d192f3d3SKai Wang create_symtab_entry(bsdar, obj->maddr, obj->size); 489d192f3d3SKai Wang if (strlen(obj->name) > _MAXNAMELEN_SVR4) 490d192f3d3SKai Wang add_to_ar_str_table(bsdar, obj->name); 491d192f3d3SKai Wang bsdar->rela_off += _ARHDR_LEN + obj->size + obj->size % 2; 492d192f3d3SKai Wang } 493d192f3d3SKai Wang 494d192f3d3SKai Wang /* 495d192f3d3SKai Wang * Pad the symbol name string table. It is treated specially because 496d192f3d3SKai Wang * symbol name table should be padded by a '\0', not the common '\n' 497d192f3d3SKai Wang * for other members. The size of sn table includes the pad bit. 498d192f3d3SKai Wang */ 499d192f3d3SKai Wang if (bsdar->s_cnt != 0 && bsdar->s_sn_sz % 2 != 0) 500d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 501d192f3d3SKai Wang 502d192f3d3SKai Wang /* 503d192f3d3SKai Wang * Archive string table is padded by a "\n" as the normal members. 504d192f3d3SKai Wang * The differece is that the size of archive string table counts 505d192f3d3SKai Wang * in the pad bit, while normal members' size fileds do not. 506d192f3d3SKai Wang */ 507d192f3d3SKai Wang if (bsdar->as != NULL && bsdar->as_sz % 2 != 0) 508d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 509d192f3d3SKai Wang 510d192f3d3SKai Wang /* 511d192f3d3SKai Wang * If there is a symbol table, calculate the size of pseudo members, 512d192f3d3SKai Wang * covert previously stored relative offsets to absolute ones, and 513d192f3d3SKai Wang * then make them Big Endian. 514d192f3d3SKai Wang * 515d192f3d3SKai Wang * absolute_offset = htobe32(relative_offset + size_of_pseudo_members) 516d192f3d3SKai Wang */ 517d192f3d3SKai Wang 518d192f3d3SKai Wang if (bsdar->s_cnt != 0) { 519d192f3d3SKai Wang s_sz = (bsdar->s_cnt + 1) * sizeof(uint32_t) + bsdar->s_sn_sz; 520d192f3d3SKai Wang pm_sz = _ARMAG_LEN + (_ARHDR_LEN + s_sz); 521d192f3d3SKai Wang if (bsdar->as != NULL) 522d192f3d3SKai Wang pm_sz += _ARHDR_LEN + bsdar->as_sz; 523d192f3d3SKai Wang for (i = 0; (size_t)i < bsdar->s_cnt; i++) 524d192f3d3SKai Wang *(bsdar->s_so + i) = htobe32(*(bsdar->s_so + i) + 525d192f3d3SKai Wang pm_sz); 526d192f3d3SKai Wang } 527d192f3d3SKai Wang 528d192f3d3SKai Wang if ((a = archive_write_new()) == NULL) 529d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_write_new failed"); 530d192f3d3SKai Wang 531d192f3d3SKai Wang archive_write_set_format_ar_svr4(a); 532d192f3d3SKai Wang 533d192f3d3SKai Wang /* The compression mode of the existing archive is used 534d192f3d3SKai Wang * for the result archive or if creating a new archive, we 535d192f3d3SKai Wang * do not compress archive by default. This default behavior can 536d192f3d3SKai Wang * be overrided by compression mode specified explicitly 537d192f3d3SKai Wang * through command line option `-j' or `-z'. 538d192f3d3SKai Wang */ 539d192f3d3SKai Wang if (bsdar->options & AR_J) 540d192f3d3SKai Wang bsdar->compression = ARCHIVE_COMPRESSION_BZIP2; 541d192f3d3SKai Wang if (bsdar->options & AR_Z) 542d192f3d3SKai Wang bsdar->compression = ARCHIVE_COMPRESSION_GZIP; 543d192f3d3SKai Wang if (bsdar->compression == ARCHIVE_COMPRESSION_BZIP2) 544d192f3d3SKai Wang archive_write_set_compression_bzip2(a); 545d192f3d3SKai Wang else if (bsdar->compression == ARCHIVE_COMPRESSION_GZIP) 546d192f3d3SKai Wang archive_write_set_compression_gzip(a); 547d192f3d3SKai Wang else 548d192f3d3SKai Wang archive_write_set_compression_none(a); 549d192f3d3SKai Wang 550d192f3d3SKai Wang AC(archive_write_open_filename(a, bsdar->filename)); 551d192f3d3SKai Wang 552d192f3d3SKai Wang /* 553d192f3d3SKai Wang * write the archive symbol table, if there is one. 554d192f3d3SKai Wang * If options -s is explicitly specified or we are invoked 555d192f3d3SKai Wang * as ranlib, write the symbol table even if it is empty. 556d192f3d3SKai Wang */ 557d192f3d3SKai Wang if ((bsdar->s_cnt != 0 && !(bsdar->options & AR_SS)) || 558d192f3d3SKai Wang bsdar->options & AR_S) { 559d192f3d3SKai Wang entry = archive_entry_new(); 560d192f3d3SKai Wang archive_entry_copy_pathname(entry, "/"); 561d192f3d3SKai Wang archive_entry_set_mtime(entry, time(NULL), 0); 562d192f3d3SKai Wang archive_entry_set_size(entry, (bsdar->s_cnt + 1) * 563d192f3d3SKai Wang sizeof(uint32_t) + bsdar->s_sn_sz); 564d192f3d3SKai Wang AC(archive_write_header(a, entry)); 565d192f3d3SKai Wang nr = htobe32(bsdar->s_cnt); 566d192f3d3SKai Wang write_data(bsdar, a, &nr, sizeof(uint32_t)); 567d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_so, sizeof(uint32_t) * 568d192f3d3SKai Wang bsdar->s_cnt); 569d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_sn, bsdar->s_sn_sz); 570d192f3d3SKai Wang archive_entry_free(entry); 571d192f3d3SKai Wang } 572d192f3d3SKai Wang 573d192f3d3SKai Wang /* write the archive string table, if any. */ 574d192f3d3SKai Wang if (bsdar->as != NULL) { 575d192f3d3SKai Wang entry = archive_entry_new(); 576d192f3d3SKai Wang archive_entry_copy_pathname(entry, "//"); 577d192f3d3SKai Wang archive_entry_set_size(entry, bsdar->as_sz); 578d192f3d3SKai Wang AC(archive_write_header(a, entry)); 579d192f3d3SKai Wang write_data(bsdar, a, bsdar->as, bsdar->as_sz); 580d192f3d3SKai Wang archive_entry_free(entry); 581d192f3d3SKai Wang } 582d192f3d3SKai Wang 583d192f3d3SKai Wang /* write normal members. */ 584d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 585d192f3d3SKai Wang entry = archive_entry_new(); 586d192f3d3SKai Wang archive_entry_copy_pathname(entry, obj->name); 587d192f3d3SKai Wang archive_entry_set_uid(entry, obj->uid); 588d192f3d3SKai Wang archive_entry_set_gid(entry, obj->gid); 589d192f3d3SKai Wang archive_entry_set_mode(entry, obj->md); 590d192f3d3SKai Wang archive_entry_set_size(entry, obj->size); 591d192f3d3SKai Wang archive_entry_set_mtime(entry, obj->mtime, 0); 592d192f3d3SKai Wang archive_entry_set_dev(entry, obj->dev); 593d192f3d3SKai Wang archive_entry_set_ino(entry, obj->ino); 594d192f3d3SKai Wang archive_entry_set_filetype(entry, AE_IFREG); 595d192f3d3SKai Wang AC(archive_write_header(a, entry)); 596d192f3d3SKai Wang write_data(bsdar, a, obj->maddr, obj->size); 597d192f3d3SKai Wang archive_entry_free(entry); 598d192f3d3SKai Wang } 599d192f3d3SKai Wang 600d192f3d3SKai Wang AC(archive_write_close(a)); 601d192f3d3SKai Wang AC(archive_write_finish(a)); 602d192f3d3SKai Wang } 603d192f3d3SKai Wang 604d192f3d3SKai Wang /* 605d192f3d3SKai Wang * Extract global symbols from ELF binary members. 606d192f3d3SKai Wang */ 607d192f3d3SKai Wang static void 608d192f3d3SKai Wang create_symtab_entry(struct bsdar *bsdar, void *maddr, size_t size) 609d192f3d3SKai Wang { 610d192f3d3SKai Wang Elf *e; 611d192f3d3SKai Wang Elf_Scn *scn; 612d192f3d3SKai Wang GElf_Shdr shdr; 613d192f3d3SKai Wang GElf_Sym sym; 614d192f3d3SKai Wang Elf_Data *data; 615d192f3d3SKai Wang char *name; 616d192f3d3SKai Wang size_t n, shstrndx; 617d192f3d3SKai Wang int elferr, tabndx, len, i; 618d192f3d3SKai Wang 619d192f3d3SKai Wang if ((e = elf_memory(maddr, size)) == NULL) { 620d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_memory() failed: %s", 621d192f3d3SKai Wang elf_errmsg(-1)); 622d192f3d3SKai Wang return; 623d192f3d3SKai Wang } 624d192f3d3SKai Wang if (elf_kind(e) != ELF_K_ELF) { 625d192f3d3SKai Wang /* Sliently ignore non-elf member. */ 626d192f3d3SKai Wang elf_end(e); 627d192f3d3SKai Wang return; 628d192f3d3SKai Wang } 629d192f3d3SKai Wang if (elf_getshstrndx(e, &shstrndx) == 0) { 630d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_getshstrndx failed: %s", 631d192f3d3SKai Wang elf_errmsg(-1)); 632d192f3d3SKai Wang elf_end(e); 633d192f3d3SKai Wang return; 634d192f3d3SKai Wang } 635d192f3d3SKai Wang 636d192f3d3SKai Wang tabndx = -1; 637d192f3d3SKai Wang scn = NULL; 638d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 639d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 640d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 641d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 642d192f3d3SKai Wang continue; 643d192f3d3SKai Wang } 644d192f3d3SKai Wang if ((name = elf_strptr(e, shstrndx, shdr.sh_name)) == NULL) { 645d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 646d192f3d3SKai Wang "elf_strptr failed: %s", elf_errmsg(-1)); 647d192f3d3SKai Wang continue; 648d192f3d3SKai Wang } 649d192f3d3SKai Wang if (strcmp(name, ".strtab") == 0) { 650d192f3d3SKai Wang tabndx = elf_ndxscn(scn); 651d192f3d3SKai Wang break; 652d192f3d3SKai Wang } 653d192f3d3SKai Wang } 654d192f3d3SKai Wang elferr = elf_errno(); 655d192f3d3SKai Wang if (elferr != 0) 656d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_nextscn failed: %s", 657d192f3d3SKai Wang elf_errmsg(elferr)); 658d192f3d3SKai Wang if (tabndx == -1) { 659d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "can't find .strtab section"); 660d192f3d3SKai Wang elf_end(e); 661d192f3d3SKai Wang return; 662d192f3d3SKai Wang } 663d192f3d3SKai Wang 664d192f3d3SKai Wang scn = NULL; 665d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 666d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 667d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 668d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 669d192f3d3SKai Wang continue; 670d192f3d3SKai Wang } 671d192f3d3SKai Wang if (shdr.sh_type != SHT_SYMTAB) 672d192f3d3SKai Wang continue; 673d192f3d3SKai Wang 674d192f3d3SKai Wang data = NULL; 675d192f3d3SKai Wang n = 0; 676d192f3d3SKai Wang while (n < shdr.sh_size && 677d192f3d3SKai Wang (data = elf_getdata(scn, data)) != NULL) { 678d192f3d3SKai Wang len = data->d_size / shdr.sh_entsize; 679d192f3d3SKai Wang for (i = 0; i < len; i++) { 680d192f3d3SKai Wang if (gelf_getsym(data, i, &sym) != &sym) { 681d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 682d192f3d3SKai Wang "gelf_getsym failed: %s", 683d192f3d3SKai Wang elf_errmsg(-1)); 684d192f3d3SKai Wang continue; 685d192f3d3SKai Wang } 686d192f3d3SKai Wang 687d192f3d3SKai Wang /* keep only global or weak symbols */ 688d192f3d3SKai Wang if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL && 689d192f3d3SKai Wang GELF_ST_BIND(sym.st_info) != STB_WEAK) 690d192f3d3SKai Wang continue; 691d192f3d3SKai Wang 692d192f3d3SKai Wang /* keep only defined symbols */ 693d192f3d3SKai Wang if (sym.st_shndx == SHN_UNDEF) 694d192f3d3SKai Wang continue; 695d192f3d3SKai Wang 696d192f3d3SKai Wang if ((name = elf_strptr(e, tabndx, 697d192f3d3SKai Wang sym.st_name)) == NULL) { 698d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 699d192f3d3SKai Wang "elf_strptr failed: %s", 700d192f3d3SKai Wang elf_errmsg(-1)); 701d192f3d3SKai Wang continue; 702d192f3d3SKai Wang } 703d192f3d3SKai Wang 704d192f3d3SKai Wang add_to_ar_sym_table(bsdar, name); 705d192f3d3SKai Wang } 706d192f3d3SKai Wang } 707d192f3d3SKai Wang } 708d192f3d3SKai Wang elferr = elf_errno(); 709d192f3d3SKai Wang if (elferr != 0) 710d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_nextscn failed: %s", 711d192f3d3SKai Wang elf_errmsg(elferr)); 712d192f3d3SKai Wang 713d192f3d3SKai Wang elf_end(e); 714d192f3d3SKai Wang } 715d192f3d3SKai Wang 716d192f3d3SKai Wang /* 717d192f3d3SKai Wang * Append to the archive string table buffer. 718d192f3d3SKai Wang */ 719d192f3d3SKai Wang static void 720d192f3d3SKai Wang add_to_ar_str_table(struct bsdar *bsdar, const char *name) 721d192f3d3SKai Wang { 722d192f3d3SKai Wang 723d192f3d3SKai Wang if (bsdar->as == NULL) { 724d192f3d3SKai Wang bsdar->as_cap = _INIT_AS_CAP; 725d192f3d3SKai Wang bsdar->as_sz = 0; 726d192f3d3SKai Wang if ((bsdar->as = malloc(bsdar->as_cap)) == NULL) 727d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 728d192f3d3SKai Wang } 729d192f3d3SKai Wang 730d192f3d3SKai Wang /* 731d192f3d3SKai Wang * The space required for holding one member name in as table includes: 732d192f3d3SKai Wang * strlen(name) + (1 for '/') + (1 for '\n') + (possibly 1 for padding). 733d192f3d3SKai Wang */ 734d192f3d3SKai Wang if (bsdar->as_sz + strlen(name) + 3 > bsdar->as_cap) { 735d192f3d3SKai Wang bsdar->as_cap *= 2; 736d192f3d3SKai Wang bsdar->as = realloc(bsdar->as, bsdar->as_cap); 737d192f3d3SKai Wang if (bsdar->as == NULL) 738d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 739d192f3d3SKai Wang } 740d192f3d3SKai Wang strncpy(&bsdar->as[bsdar->as_sz], name, strlen(name)); 741d192f3d3SKai Wang bsdar->as_sz += strlen(name); 742d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '/'; 743d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 744d192f3d3SKai Wang } 745d192f3d3SKai Wang 746d192f3d3SKai Wang /* 747d192f3d3SKai Wang * Append to the archive symbol table buffer. 748d192f3d3SKai Wang */ 749d192f3d3SKai Wang static void 750d192f3d3SKai Wang add_to_ar_sym_table(struct bsdar *bsdar, const char *name) 751d192f3d3SKai Wang { 752d192f3d3SKai Wang 753d192f3d3SKai Wang if (bsdar->s_so == NULL) { 754d192f3d3SKai Wang if ((bsdar->s_so = malloc(_INIT_SYMOFF_CAP)) == 755d192f3d3SKai Wang NULL) 756d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 757d192f3d3SKai Wang bsdar->s_so_cap = _INIT_SYMOFF_CAP; 758d192f3d3SKai Wang bsdar->s_cnt = 0; 759d192f3d3SKai Wang } 760d192f3d3SKai Wang 761d192f3d3SKai Wang if (bsdar->s_sn == NULL) { 762d192f3d3SKai Wang if ((bsdar->s_sn = malloc(_INIT_SYMNAME_CAP)) == NULL) 763d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 764d192f3d3SKai Wang bsdar->s_sn_cap = _INIT_SYMNAME_CAP; 765d192f3d3SKai Wang bsdar->s_sn_sz = 0; 766d192f3d3SKai Wang } 767d192f3d3SKai Wang 768d192f3d3SKai Wang if (bsdar->s_cnt * sizeof(uint32_t) >= bsdar->s_so_cap) { 769d192f3d3SKai Wang bsdar->s_so_cap *= 2; 770d192f3d3SKai Wang bsdar->s_so = realloc(bsdar->s_so, bsdar->s_so_cap); 771d192f3d3SKai Wang if (bsdar->s_so == NULL) 772d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 773d192f3d3SKai Wang } 774d192f3d3SKai Wang bsdar->s_so[bsdar->s_cnt] = bsdar->rela_off; 775d192f3d3SKai Wang bsdar->s_cnt++; 776d192f3d3SKai Wang 777d192f3d3SKai Wang /* 778d192f3d3SKai Wang * The space required for holding one symbol name in sn table includes: 779d192f3d3SKai Wang * strlen(name) + (1 for '\n') + (possibly 1 for padding). 780d192f3d3SKai Wang */ 781d192f3d3SKai Wang if (bsdar->s_sn_sz + strlen(name) + 2 > bsdar->s_sn_cap) { 782d192f3d3SKai Wang bsdar->s_sn_cap *= 2; 783d192f3d3SKai Wang bsdar->s_sn = realloc(bsdar->s_sn, bsdar->s_sn_cap); 784d192f3d3SKai Wang if (bsdar->s_sn == NULL) 785d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 786d192f3d3SKai Wang } 787d192f3d3SKai Wang strncpy(&bsdar->s_sn[bsdar->s_sn_sz], name, strlen(name)); 788d192f3d3SKai Wang bsdar->s_sn_sz += strlen(name); 789d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 790d192f3d3SKai Wang } 791