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 301517dbe2eSKai Wang if (size > 0) { 302d192f3d3SKai Wang if ((buff = malloc(size)) == NULL) 303517dbe2eSKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 304517dbe2eSKai Wang "malloc failed"); 305d192f3d3SKai Wang if (archive_read_data(a, buff, size) != (ssize_t)size) { 306517dbe2eSKai Wang bsdar_warnc(bsdar, 0, "%s", 307517dbe2eSKai Wang archive_error_string(a)); 308d192f3d3SKai Wang free(buff); 309d192f3d3SKai Wang continue; 310d192f3d3SKai Wang } 311517dbe2eSKai Wang } else 312517dbe2eSKai Wang buff = NULL; 313517dbe2eSKai Wang 314d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 315d192f3d3SKai Wang if (obj == NULL) 316d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 317d192f3d3SKai Wang obj->maddr = buff; 318d192f3d3SKai Wang if ((obj->name = strdup(name)) == NULL) 319d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 320d192f3d3SKai Wang obj->size = size; 321d192f3d3SKai Wang obj->uid = archive_entry_uid(entry); 322d192f3d3SKai Wang obj->gid = archive_entry_gid(entry); 323d192f3d3SKai Wang obj->md = archive_entry_mode(entry); 324d192f3d3SKai Wang obj->mtime = archive_entry_mtime(entry); 325d192f3d3SKai Wang obj->dev = 0; 326d192f3d3SKai Wang obj->ino = 0; 327d192f3d3SKai Wang 328d192f3d3SKai Wang /* 329d192f3d3SKai Wang * Objects from archive have obj->fd set to -1, 330d192f3d3SKai Wang * for the ease of cleaning up. 331d192f3d3SKai Wang */ 332d192f3d3SKai Wang obj->fd = -1; 333d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 334d192f3d3SKai Wang } 335d192f3d3SKai Wang AC(archive_read_close(a)); 336d192f3d3SKai Wang AC(archive_read_finish(a)); 337d192f3d3SKai Wang 338d192f3d3SKai Wang /* 339d192f3d3SKai Wang * For mode 's', no member will be moved, deleted or replaced. 340d192f3d3SKai Wang */ 341d192f3d3SKai Wang if (mode == 's') 342d192f3d3SKai Wang goto write_objs; 343d192f3d3SKai Wang 344d192f3d3SKai Wang /* 345d192f3d3SKai Wang * Try to find the position member specified by user. 346d192f3d3SKai Wang */ 347d192f3d3SKai Wang if (bsdar->options & AR_A || bsdar->options & AR_B) { 348d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 349d192f3d3SKai Wang if (strcmp(obj->name, bsdar->posarg) == 0) { 350d192f3d3SKai Wang pos = obj; 351d192f3d3SKai Wang break; 352d192f3d3SKai Wang } 353d192f3d3SKai Wang } 354d192f3d3SKai Wang 355d192f3d3SKai Wang /* 356d192f3d3SKai Wang * If can't find `pos' specified by user, 357d192f3d3SKai Wang * sliently insert objects at tail. 358d192f3d3SKai Wang */ 359d192f3d3SKai Wang if (pos == NULL) 360d192f3d3SKai Wang bsdar->options &= ~(AR_A | AR_B); 361d192f3d3SKai Wang } 362d192f3d3SKai Wang 363d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 364d192f3d3SKai Wang av = &bsdar->argv[i]; 365d192f3d3SKai Wang 366d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 367d192f3d3SKai Wang if ((bname = basename(*av)) == NULL) 368d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 369d192f3d3SKai Wang "basename failed"); 370d192f3d3SKai Wang if (bsdar->options & AR_TR) { 371d192f3d3SKai Wang if (strncmp(bname, obj->name, _TRUNCATE_LEN)) 372d192f3d3SKai Wang continue; 373d192f3d3SKai Wang } else 374d192f3d3SKai Wang if (strcmp(bname, obj->name) != 0) 375d192f3d3SKai Wang continue; 376d192f3d3SKai Wang 377d192f3d3SKai Wang if (mode == 'r') { 378d192f3d3SKai Wang /* 379d192f3d3SKai Wang * if the new member is not qualified 380d192f3d3SKai Wang * to replace the old one, skip it. 381d192f3d3SKai Wang */ 382d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 383d192f3d3SKai Wang obj->mtime); 384d192f3d3SKai Wang if (nobj == NULL) 385d192f3d3SKai Wang goto skip_obj; 386d192f3d3SKai Wang } 387d192f3d3SKai Wang 388d192f3d3SKai Wang if (bsdar->options & AR_V) 389d192f3d3SKai Wang (void)fprintf(stdout, "%c - %s\n", mode, 390d192f3d3SKai Wang *av); 391d192f3d3SKai Wang 392d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 393d192f3d3SKai Wang if (mode == 'd' || mode == 'r') { 394d192f3d3SKai Wang free(obj->maddr); 395d192f3d3SKai Wang free(obj->name); 396d192f3d3SKai Wang free(obj); 397d192f3d3SKai Wang } 398d192f3d3SKai Wang 399d192f3d3SKai Wang if (mode == 'm') 400d192f3d3SKai Wang insert_obj(bsdar, obj, pos); 401d192f3d3SKai Wang if (mode == 'r') 402d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 403d192f3d3SKai Wang 404d192f3d3SKai Wang skip_obj: 405d192f3d3SKai Wang *av = NULL; 406d192f3d3SKai Wang break; 407d192f3d3SKai Wang } 408d192f3d3SKai Wang 409d192f3d3SKai Wang } 410d192f3d3SKai Wang 411d192f3d3SKai Wang new_archive: 412d192f3d3SKai Wang /* 413d192f3d3SKai Wang * When operating in mode 'r', directly add those user specified 414d192f3d3SKai Wang * objects which do not exist in current archive. 415d192f3d3SKai Wang */ 416d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 417d192f3d3SKai Wang av = &bsdar->argv[i]; 418d192f3d3SKai Wang if (*av != NULL && mode == 'r') { 419d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 0); 420d192f3d3SKai Wang if (nobj != NULL) 421d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 422d192f3d3SKai Wang if (bsdar->options & AR_V && nobj != NULL) 423d192f3d3SKai Wang (void)fprintf(stdout, "a - %s\n", *av); 424d192f3d3SKai Wang *av = NULL; 425d192f3d3SKai Wang } 426d192f3d3SKai Wang } 427d192f3d3SKai Wang 428d192f3d3SKai Wang write_objs: 429d192f3d3SKai Wang write_objs(bsdar); 430d192f3d3SKai Wang write_cleanup(bsdar); 431d192f3d3SKai Wang } 432d192f3d3SKai Wang 433d192f3d3SKai Wang /* 434d192f3d3SKai Wang * Memory cleaning up. 435d192f3d3SKai Wang */ 436d192f3d3SKai Wang static void 437d192f3d3SKai Wang write_cleanup(struct bsdar *bsdar) 438d192f3d3SKai Wang { 439d192f3d3SKai Wang struct ar_obj *obj, *obj_temp; 440d192f3d3SKai Wang 441d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 442d192f3d3SKai Wang if (obj->fd == -1) 443d192f3d3SKai Wang free(obj->maddr); 444d192f3d3SKai Wang else 44547203048SKai Wang if (obj->maddr != NULL && munmap(obj->maddr, obj->size)) 446d192f3d3SKai Wang bsdar_warnc(bsdar, errno, 447d192f3d3SKai Wang "can't munmap file: %s", obj->name); 448d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 44947203048SKai Wang free(obj->name); 450d192f3d3SKai Wang free(obj); 451d192f3d3SKai Wang } 452d192f3d3SKai Wang 453d192f3d3SKai Wang free(bsdar->as); 454d192f3d3SKai Wang free(bsdar->s_so); 455d192f3d3SKai Wang free(bsdar->s_sn); 456d192f3d3SKai Wang bsdar->as = NULL; 457d192f3d3SKai Wang bsdar->s_so = NULL; 458d192f3d3SKai Wang bsdar->s_sn = NULL; 459d192f3d3SKai Wang } 460d192f3d3SKai Wang 461d192f3d3SKai Wang /* 462d192f3d3SKai Wang * Wrapper for archive_write_data(). 463d192f3d3SKai Wang */ 464d192f3d3SKai Wang static void 465d192f3d3SKai Wang write_data(struct bsdar *bsdar, struct archive *a, const void *buf, size_t s) 466d192f3d3SKai Wang { 467d192f3d3SKai Wang if (archive_write_data(a, buf, s) != (ssize_t)s) 468d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "%s", 469d192f3d3SKai Wang archive_error_string(a)); 470d192f3d3SKai Wang } 471d192f3d3SKai Wang 472d192f3d3SKai Wang /* 473d192f3d3SKai Wang * Write the resulting archive members. 474d192f3d3SKai Wang */ 475d192f3d3SKai Wang static void 476d192f3d3SKai Wang write_objs(struct bsdar *bsdar) 477d192f3d3SKai Wang { 478d192f3d3SKai Wang struct ar_obj *obj; 479d192f3d3SKai Wang struct archive *a; 480d192f3d3SKai Wang struct archive_entry *entry; 481d192f3d3SKai Wang size_t s_sz; /* size of archive symbol table. */ 482d192f3d3SKai Wang size_t pm_sz; /* size of pseudo members */ 483d192f3d3SKai Wang int i, nr; 484d192f3d3SKai Wang 485d192f3d3SKai Wang if (elf_version(EV_CURRENT) == EV_NONE) 486d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, 487d192f3d3SKai Wang "ELF library initialization failed: %s", elf_errmsg(-1)); 488d192f3d3SKai Wang 489d192f3d3SKai Wang bsdar->rela_off = 0; 490d192f3d3SKai Wang 491d192f3d3SKai Wang /* Create archive symbol table and archive string table, if need. */ 492d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 49347203048SKai Wang if (!(bsdar->options & AR_SS) && obj->maddr != NULL) 494d192f3d3SKai Wang create_symtab_entry(bsdar, obj->maddr, obj->size); 495d192f3d3SKai Wang if (strlen(obj->name) > _MAXNAMELEN_SVR4) 496d192f3d3SKai Wang add_to_ar_str_table(bsdar, obj->name); 497d192f3d3SKai Wang bsdar->rela_off += _ARHDR_LEN + obj->size + obj->size % 2; 498d192f3d3SKai Wang } 499d192f3d3SKai Wang 500d192f3d3SKai Wang /* 501d192f3d3SKai Wang * Pad the symbol name string table. It is treated specially because 502d192f3d3SKai Wang * symbol name table should be padded by a '\0', not the common '\n' 503d192f3d3SKai Wang * for other members. The size of sn table includes the pad bit. 504d192f3d3SKai Wang */ 505d192f3d3SKai Wang if (bsdar->s_cnt != 0 && bsdar->s_sn_sz % 2 != 0) 506d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 507d192f3d3SKai Wang 508d192f3d3SKai Wang /* 509d192f3d3SKai Wang * Archive string table is padded by a "\n" as the normal members. 510517dbe2eSKai Wang * The difference is that the size of archive string table counts 511d192f3d3SKai Wang * in the pad bit, while normal members' size fileds do not. 512d192f3d3SKai Wang */ 513d192f3d3SKai Wang if (bsdar->as != NULL && bsdar->as_sz % 2 != 0) 514d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 515d192f3d3SKai Wang 516d192f3d3SKai Wang /* 517d192f3d3SKai Wang * If there is a symbol table, calculate the size of pseudo members, 518517dbe2eSKai Wang * convert previously stored relative offsets to absolute ones, and 519d192f3d3SKai Wang * then make them Big Endian. 520d192f3d3SKai Wang * 521d192f3d3SKai Wang * absolute_offset = htobe32(relative_offset + size_of_pseudo_members) 522d192f3d3SKai Wang */ 523d192f3d3SKai Wang 524d192f3d3SKai Wang if (bsdar->s_cnt != 0) { 525d192f3d3SKai Wang s_sz = (bsdar->s_cnt + 1) * sizeof(uint32_t) + bsdar->s_sn_sz; 526d192f3d3SKai Wang pm_sz = _ARMAG_LEN + (_ARHDR_LEN + s_sz); 527d192f3d3SKai Wang if (bsdar->as != NULL) 528d192f3d3SKai Wang pm_sz += _ARHDR_LEN + bsdar->as_sz; 529d192f3d3SKai Wang for (i = 0; (size_t)i < bsdar->s_cnt; i++) 530d192f3d3SKai Wang *(bsdar->s_so + i) = htobe32(*(bsdar->s_so + i) + 531d192f3d3SKai Wang pm_sz); 532d192f3d3SKai Wang } 533d192f3d3SKai Wang 534d192f3d3SKai Wang if ((a = archive_write_new()) == NULL) 535d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_write_new failed"); 536d192f3d3SKai Wang 537d192f3d3SKai Wang archive_write_set_format_ar_svr4(a); 538d192f3d3SKai Wang 539d192f3d3SKai Wang /* The compression mode of the existing archive is used 540d192f3d3SKai Wang * for the result archive or if creating a new archive, we 541d192f3d3SKai Wang * do not compress archive by default. This default behavior can 542d192f3d3SKai Wang * be overrided by compression mode specified explicitly 543d192f3d3SKai Wang * through command line option `-j' or `-z'. 544d192f3d3SKai Wang */ 545d192f3d3SKai Wang if (bsdar->options & AR_J) 546d192f3d3SKai Wang bsdar->compression = ARCHIVE_COMPRESSION_BZIP2; 547d192f3d3SKai Wang if (bsdar->options & AR_Z) 548d192f3d3SKai Wang bsdar->compression = ARCHIVE_COMPRESSION_GZIP; 549d192f3d3SKai Wang if (bsdar->compression == ARCHIVE_COMPRESSION_BZIP2) 550d192f3d3SKai Wang archive_write_set_compression_bzip2(a); 551d192f3d3SKai Wang else if (bsdar->compression == ARCHIVE_COMPRESSION_GZIP) 552d192f3d3SKai Wang archive_write_set_compression_gzip(a); 553d192f3d3SKai Wang else 554d192f3d3SKai Wang archive_write_set_compression_none(a); 555d192f3d3SKai Wang 556d192f3d3SKai Wang AC(archive_write_open_filename(a, bsdar->filename)); 557d192f3d3SKai Wang 558d192f3d3SKai Wang /* 559d192f3d3SKai Wang * write the archive symbol table, if there is one. 560d192f3d3SKai Wang * If options -s is explicitly specified or we are invoked 561d192f3d3SKai Wang * as ranlib, write the symbol table even if it is empty. 562d192f3d3SKai Wang */ 563d192f3d3SKai Wang if ((bsdar->s_cnt != 0 && !(bsdar->options & AR_SS)) || 564d192f3d3SKai Wang bsdar->options & AR_S) { 565d192f3d3SKai Wang entry = archive_entry_new(); 566d192f3d3SKai Wang archive_entry_copy_pathname(entry, "/"); 567d192f3d3SKai Wang archive_entry_set_mtime(entry, time(NULL), 0); 568d192f3d3SKai Wang archive_entry_set_size(entry, (bsdar->s_cnt + 1) * 569d192f3d3SKai Wang sizeof(uint32_t) + bsdar->s_sn_sz); 570d192f3d3SKai Wang AC(archive_write_header(a, entry)); 571d192f3d3SKai Wang nr = htobe32(bsdar->s_cnt); 572d192f3d3SKai Wang write_data(bsdar, a, &nr, sizeof(uint32_t)); 573d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_so, sizeof(uint32_t) * 574d192f3d3SKai Wang bsdar->s_cnt); 575d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_sn, bsdar->s_sn_sz); 576d192f3d3SKai Wang archive_entry_free(entry); 577d192f3d3SKai Wang } 578d192f3d3SKai Wang 579d192f3d3SKai Wang /* write the archive string table, if any. */ 580d192f3d3SKai Wang if (bsdar->as != NULL) { 581d192f3d3SKai Wang entry = archive_entry_new(); 582d192f3d3SKai Wang archive_entry_copy_pathname(entry, "//"); 583d192f3d3SKai Wang archive_entry_set_size(entry, bsdar->as_sz); 584d192f3d3SKai Wang AC(archive_write_header(a, entry)); 585d192f3d3SKai Wang write_data(bsdar, a, bsdar->as, bsdar->as_sz); 586d192f3d3SKai Wang archive_entry_free(entry); 587d192f3d3SKai Wang } 588d192f3d3SKai Wang 589d192f3d3SKai Wang /* write normal members. */ 590d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 591d192f3d3SKai Wang entry = archive_entry_new(); 592d192f3d3SKai Wang archive_entry_copy_pathname(entry, obj->name); 593d192f3d3SKai Wang archive_entry_set_uid(entry, obj->uid); 594d192f3d3SKai Wang archive_entry_set_gid(entry, obj->gid); 595d192f3d3SKai Wang archive_entry_set_mode(entry, obj->md); 596d192f3d3SKai Wang archive_entry_set_size(entry, obj->size); 597d192f3d3SKai Wang archive_entry_set_mtime(entry, obj->mtime, 0); 598d192f3d3SKai Wang archive_entry_set_dev(entry, obj->dev); 599d192f3d3SKai Wang archive_entry_set_ino(entry, obj->ino); 600d192f3d3SKai Wang archive_entry_set_filetype(entry, AE_IFREG); 601d192f3d3SKai Wang AC(archive_write_header(a, entry)); 602d192f3d3SKai Wang write_data(bsdar, a, obj->maddr, obj->size); 603d192f3d3SKai Wang archive_entry_free(entry); 604d192f3d3SKai Wang } 605d192f3d3SKai Wang 606d192f3d3SKai Wang AC(archive_write_close(a)); 607d192f3d3SKai Wang AC(archive_write_finish(a)); 608d192f3d3SKai Wang } 609d192f3d3SKai Wang 610d192f3d3SKai Wang /* 611d192f3d3SKai Wang * Extract global symbols from ELF binary members. 612d192f3d3SKai Wang */ 613d192f3d3SKai Wang static void 614d192f3d3SKai Wang create_symtab_entry(struct bsdar *bsdar, void *maddr, size_t size) 615d192f3d3SKai Wang { 616d192f3d3SKai Wang Elf *e; 617d192f3d3SKai Wang Elf_Scn *scn; 618d192f3d3SKai Wang GElf_Shdr shdr; 619d192f3d3SKai Wang GElf_Sym sym; 620d192f3d3SKai Wang Elf_Data *data; 621d192f3d3SKai Wang char *name; 622d192f3d3SKai Wang size_t n, shstrndx; 623d192f3d3SKai Wang int elferr, tabndx, len, i; 624d192f3d3SKai Wang 625d192f3d3SKai Wang if ((e = elf_memory(maddr, size)) == NULL) { 626d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_memory() failed: %s", 627d192f3d3SKai Wang elf_errmsg(-1)); 628d192f3d3SKai Wang return; 629d192f3d3SKai Wang } 630d192f3d3SKai Wang if (elf_kind(e) != ELF_K_ELF) { 631d192f3d3SKai Wang /* Sliently ignore non-elf member. */ 632d192f3d3SKai Wang elf_end(e); 633d192f3d3SKai Wang return; 634d192f3d3SKai Wang } 635d192f3d3SKai Wang if (elf_getshstrndx(e, &shstrndx) == 0) { 636d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_getshstrndx failed: %s", 637d192f3d3SKai Wang elf_errmsg(-1)); 638d192f3d3SKai Wang elf_end(e); 639d192f3d3SKai Wang return; 640d192f3d3SKai Wang } 641d192f3d3SKai Wang 642d192f3d3SKai Wang tabndx = -1; 643d192f3d3SKai Wang scn = NULL; 644d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 645d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 646d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 647d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 648d192f3d3SKai Wang continue; 649d192f3d3SKai Wang } 650d192f3d3SKai Wang if ((name = elf_strptr(e, shstrndx, shdr.sh_name)) == NULL) { 651d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 652d192f3d3SKai Wang "elf_strptr failed: %s", elf_errmsg(-1)); 653d192f3d3SKai Wang continue; 654d192f3d3SKai Wang } 655d192f3d3SKai Wang if (strcmp(name, ".strtab") == 0) { 656d192f3d3SKai Wang tabndx = elf_ndxscn(scn); 657d192f3d3SKai Wang break; 658d192f3d3SKai Wang } 659d192f3d3SKai Wang } 660d192f3d3SKai Wang elferr = elf_errno(); 661d192f3d3SKai Wang if (elferr != 0) 662d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_nextscn failed: %s", 663d192f3d3SKai Wang elf_errmsg(elferr)); 664d192f3d3SKai Wang if (tabndx == -1) { 665d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "can't find .strtab section"); 666d192f3d3SKai Wang elf_end(e); 667d192f3d3SKai Wang return; 668d192f3d3SKai Wang } 669d192f3d3SKai Wang 670d192f3d3SKai Wang scn = NULL; 671d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 672d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 673d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 674d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 675d192f3d3SKai Wang continue; 676d192f3d3SKai Wang } 677d192f3d3SKai Wang if (shdr.sh_type != SHT_SYMTAB) 678d192f3d3SKai Wang continue; 679d192f3d3SKai Wang 680d192f3d3SKai Wang data = NULL; 681d192f3d3SKai Wang n = 0; 682d192f3d3SKai Wang while (n < shdr.sh_size && 683d192f3d3SKai Wang (data = elf_getdata(scn, data)) != NULL) { 684d192f3d3SKai Wang len = data->d_size / shdr.sh_entsize; 685d192f3d3SKai Wang for (i = 0; i < len; i++) { 686d192f3d3SKai Wang if (gelf_getsym(data, i, &sym) != &sym) { 687d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 688d192f3d3SKai Wang "gelf_getsym failed: %s", 689d192f3d3SKai Wang elf_errmsg(-1)); 690d192f3d3SKai Wang continue; 691d192f3d3SKai Wang } 692d192f3d3SKai Wang 693d192f3d3SKai Wang /* keep only global or weak symbols */ 694d192f3d3SKai Wang if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL && 695d192f3d3SKai Wang GELF_ST_BIND(sym.st_info) != STB_WEAK) 696d192f3d3SKai Wang continue; 697d192f3d3SKai Wang 698d192f3d3SKai Wang /* keep only defined symbols */ 699d192f3d3SKai Wang if (sym.st_shndx == SHN_UNDEF) 700d192f3d3SKai Wang continue; 701d192f3d3SKai Wang 702d192f3d3SKai Wang if ((name = elf_strptr(e, tabndx, 703d192f3d3SKai Wang sym.st_name)) == NULL) { 704d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 705d192f3d3SKai Wang "elf_strptr failed: %s", 706d192f3d3SKai Wang elf_errmsg(-1)); 707d192f3d3SKai Wang continue; 708d192f3d3SKai Wang } 709d192f3d3SKai Wang 710d192f3d3SKai Wang add_to_ar_sym_table(bsdar, name); 711d192f3d3SKai Wang } 712d192f3d3SKai Wang } 713d192f3d3SKai Wang } 714d192f3d3SKai Wang elferr = elf_errno(); 715d192f3d3SKai Wang if (elferr != 0) 716d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_nextscn failed: %s", 717d192f3d3SKai Wang elf_errmsg(elferr)); 718d192f3d3SKai Wang 719d192f3d3SKai Wang elf_end(e); 720d192f3d3SKai Wang } 721d192f3d3SKai Wang 722d192f3d3SKai Wang /* 723d192f3d3SKai Wang * Append to the archive string table buffer. 724d192f3d3SKai Wang */ 725d192f3d3SKai Wang static void 726d192f3d3SKai Wang add_to_ar_str_table(struct bsdar *bsdar, const char *name) 727d192f3d3SKai Wang { 728d192f3d3SKai Wang 729d192f3d3SKai Wang if (bsdar->as == NULL) { 730d192f3d3SKai Wang bsdar->as_cap = _INIT_AS_CAP; 731d192f3d3SKai Wang bsdar->as_sz = 0; 732d192f3d3SKai Wang if ((bsdar->as = malloc(bsdar->as_cap)) == NULL) 733d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 734d192f3d3SKai Wang } 735d192f3d3SKai Wang 736d192f3d3SKai Wang /* 737d192f3d3SKai Wang * The space required for holding one member name in as table includes: 738d192f3d3SKai Wang * strlen(name) + (1 for '/') + (1 for '\n') + (possibly 1 for padding). 739d192f3d3SKai Wang */ 740517dbe2eSKai Wang while (bsdar->as_sz + strlen(name) + 3 > bsdar->as_cap) { 741d192f3d3SKai Wang bsdar->as_cap *= 2; 742d192f3d3SKai Wang bsdar->as = realloc(bsdar->as, bsdar->as_cap); 743d192f3d3SKai Wang if (bsdar->as == NULL) 744d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 745d192f3d3SKai Wang } 746d192f3d3SKai Wang strncpy(&bsdar->as[bsdar->as_sz], name, strlen(name)); 747d192f3d3SKai Wang bsdar->as_sz += strlen(name); 748d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '/'; 749d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 750d192f3d3SKai Wang } 751d192f3d3SKai Wang 752d192f3d3SKai Wang /* 753d192f3d3SKai Wang * Append to the archive symbol table buffer. 754d192f3d3SKai Wang */ 755d192f3d3SKai Wang static void 756d192f3d3SKai Wang add_to_ar_sym_table(struct bsdar *bsdar, const char *name) 757d192f3d3SKai Wang { 758d192f3d3SKai Wang 759d192f3d3SKai Wang if (bsdar->s_so == NULL) { 760d192f3d3SKai Wang if ((bsdar->s_so = malloc(_INIT_SYMOFF_CAP)) == 761d192f3d3SKai Wang NULL) 762d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 763d192f3d3SKai Wang bsdar->s_so_cap = _INIT_SYMOFF_CAP; 764d192f3d3SKai Wang bsdar->s_cnt = 0; 765d192f3d3SKai Wang } 766d192f3d3SKai Wang 767d192f3d3SKai Wang if (bsdar->s_sn == NULL) { 768d192f3d3SKai Wang if ((bsdar->s_sn = malloc(_INIT_SYMNAME_CAP)) == NULL) 769d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 770d192f3d3SKai Wang bsdar->s_sn_cap = _INIT_SYMNAME_CAP; 771d192f3d3SKai Wang bsdar->s_sn_sz = 0; 772d192f3d3SKai Wang } 773d192f3d3SKai Wang 774d192f3d3SKai Wang if (bsdar->s_cnt * sizeof(uint32_t) >= bsdar->s_so_cap) { 775d192f3d3SKai Wang bsdar->s_so_cap *= 2; 776d192f3d3SKai Wang bsdar->s_so = realloc(bsdar->s_so, bsdar->s_so_cap); 777d192f3d3SKai Wang if (bsdar->s_so == NULL) 778d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 779d192f3d3SKai Wang } 780d192f3d3SKai Wang bsdar->s_so[bsdar->s_cnt] = bsdar->rela_off; 781d192f3d3SKai Wang bsdar->s_cnt++; 782d192f3d3SKai Wang 783d192f3d3SKai Wang /* 784d192f3d3SKai Wang * The space required for holding one symbol name in sn table includes: 785d192f3d3SKai Wang * strlen(name) + (1 for '\n') + (possibly 1 for padding). 786d192f3d3SKai Wang */ 787517dbe2eSKai Wang while (bsdar->s_sn_sz + strlen(name) + 2 > bsdar->s_sn_cap) { 788d192f3d3SKai Wang bsdar->s_sn_cap *= 2; 789d192f3d3SKai Wang bsdar->s_sn = realloc(bsdar->s_sn, bsdar->s_sn_cap); 790d192f3d3SKai Wang if (bsdar->s_sn == NULL) 791d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 792d192f3d3SKai Wang } 793d192f3d3SKai Wang strncpy(&bsdar->s_sn[bsdar->s_sn_sz], name, strlen(name)); 794d192f3d3SKai Wang bsdar->s_sn_sz += strlen(name); 795d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 796d192f3d3SKai Wang } 797