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); 630c099281SKai Wang static void read_objs(struct bsdar *bsdar, const char *archive, 640c099281SKai Wang int checkargv); 65d192f3d3SKai Wang static void write_archive(struct bsdar *bsdar, char mode); 66d192f3d3SKai Wang static void write_cleanup(struct bsdar *bsdar); 67d192f3d3SKai Wang static void write_data(struct bsdar *bsdar, struct archive *a, 68d192f3d3SKai Wang const void *buf, size_t s); 69d192f3d3SKai Wang static void write_objs(struct bsdar *bsdar); 70d192f3d3SKai Wang 71d192f3d3SKai Wang void 72d192f3d3SKai Wang ar_mode_d(struct bsdar *bsdar) 73d192f3d3SKai Wang { 74d192f3d3SKai Wang 75d192f3d3SKai Wang write_archive(bsdar, 'd'); 76d192f3d3SKai Wang } 77d192f3d3SKai Wang 78d192f3d3SKai Wang void 79d192f3d3SKai Wang ar_mode_m(struct bsdar *bsdar) 80d192f3d3SKai Wang { 81d192f3d3SKai Wang 82d192f3d3SKai Wang write_archive(bsdar, 'm'); 83d192f3d3SKai Wang } 84d192f3d3SKai Wang 85d192f3d3SKai Wang void 86cb0dad38SKai Wang ar_mode_q(struct bsdar *bsdar) 87cb0dad38SKai Wang { 88cb0dad38SKai Wang 89cb0dad38SKai Wang write_archive(bsdar, 'q'); 90cb0dad38SKai Wang } 91cb0dad38SKai Wang 92cb0dad38SKai Wang void 93d192f3d3SKai Wang ar_mode_r(struct bsdar *bsdar) 94d192f3d3SKai Wang { 95d192f3d3SKai Wang 96d192f3d3SKai Wang write_archive(bsdar, 'r'); 97d192f3d3SKai Wang } 98d192f3d3SKai Wang 99d192f3d3SKai Wang void 100d192f3d3SKai Wang ar_mode_s(struct bsdar *bsdar) 101d192f3d3SKai Wang { 102d192f3d3SKai Wang 103d192f3d3SKai Wang write_archive(bsdar, 's'); 104d192f3d3SKai Wang } 105d192f3d3SKai Wang 1060c099281SKai Wang void 1070c099281SKai Wang ar_mode_A(struct bsdar *bsdar) 1080c099281SKai Wang { 1090c099281SKai Wang 1100c099281SKai Wang write_archive(bsdar, 'A'); 1110c099281SKai Wang } 1120c099281SKai Wang 113d192f3d3SKai Wang /* 114d192f3d3SKai Wang * Create object from file, return created obj upon success, or NULL 115d192f3d3SKai Wang * when an error occurs or the member is not newer than existing 116*6bef5d28SBenedict Reuschling * one while -u is specified. 117d192f3d3SKai Wang */ 118d192f3d3SKai Wang static struct ar_obj * 119d192f3d3SKai Wang create_obj_from_file(struct bsdar *bsdar, const char *name, time_t mtime) 120d192f3d3SKai Wang { 121d192f3d3SKai Wang struct ar_obj *obj; 122d192f3d3SKai Wang struct stat sb; 123d192f3d3SKai Wang const char *bname; 124d192f3d3SKai Wang 125d192f3d3SKai Wang if (name == NULL) 126d192f3d3SKai Wang return (NULL); 127d192f3d3SKai Wang 128d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 129d192f3d3SKai Wang if (obj == NULL) 130d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 131d192f3d3SKai Wang if ((obj->fd = open(name, O_RDONLY, 0)) < 0) { 132d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't open file: %s", name); 133d192f3d3SKai Wang free(obj); 134d192f3d3SKai Wang return (NULL); 135d192f3d3SKai Wang } 136d192f3d3SKai Wang 137d192f3d3SKai Wang if ((bname = basename(name)) == NULL) 138d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "basename failed"); 139d192f3d3SKai Wang if (bsdar->options & AR_TR && strlen(bname) > _TRUNCATE_LEN) { 140d192f3d3SKai Wang if ((obj->name = malloc(_TRUNCATE_LEN + 1)) == NULL) 141d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 142d192f3d3SKai Wang (void)strncpy(obj->name, bname, _TRUNCATE_LEN); 143d192f3d3SKai Wang obj->name[_TRUNCATE_LEN] = '\0'; 144d192f3d3SKai Wang } else 145d192f3d3SKai Wang if ((obj->name = strdup(bname)) == NULL) 146d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 147d192f3d3SKai Wang 148d192f3d3SKai Wang if (fstat(obj->fd, &sb) < 0) { 149d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't fstat file: %s", obj->name); 150d192f3d3SKai Wang goto giveup; 151d192f3d3SKai Wang } 152d192f3d3SKai Wang if (!S_ISREG(sb.st_mode)) { 153d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s is not an ordinary file", obj->name); 154d192f3d3SKai Wang goto giveup; 155d192f3d3SKai Wang } 156d192f3d3SKai Wang 157d192f3d3SKai Wang /* 158d192f3d3SKai Wang * When option '-u' is specified and member is not newer than the 159d192f3d3SKai Wang * existing one, the replace will not happen. While if mtime == 0, 160d192f3d3SKai Wang * which indicates that this is to "replace a none exist member", 161d192f3d3SKai Wang * the replace will proceed regardless of '-u'. 162d192f3d3SKai Wang */ 163d192f3d3SKai Wang if (mtime != 0 && bsdar->options & AR_U && sb.st_mtime <= mtime) 164d192f3d3SKai Wang goto giveup; 165d192f3d3SKai Wang 1660e479ac8STim Kientzle /* 1670e479ac8STim Kientzle * When option '-D' is specified, mtime and UID / GID from the file 1680e479ac8STim Kientzle * will be replaced with 0, and file mode with 644. This ensures that 1690e479ac8STim Kientzle * checksums will match for two archives containing the exact same 1700e479ac8STim Kientzle * files. 1710e479ac8STim Kientzle */ 1720e479ac8STim Kientzle if (bsdar->options & AR_D) { 1730e479ac8STim Kientzle obj->uid = 0; 1740e479ac8STim Kientzle obj->gid = 0; 1750e479ac8STim Kientzle obj->mtime = 0; 1760e479ac8STim Kientzle obj->md = S_IFREG | 0644; 1770e479ac8STim Kientzle } else { 178d192f3d3SKai Wang obj->uid = sb.st_uid; 179d192f3d3SKai Wang obj->gid = sb.st_gid; 180d192f3d3SKai Wang obj->mtime = sb.st_mtime; 1810e479ac8STim Kientzle obj->md = sb.st_mode; 1820e479ac8STim Kientzle } 1830e479ac8STim Kientzle obj->size = sb.st_size; 184d192f3d3SKai Wang obj->dev = sb.st_dev; 185d192f3d3SKai Wang obj->ino = sb.st_ino; 18647203048SKai Wang 18747203048SKai Wang if (obj->size == 0) { 18847203048SKai Wang obj->maddr = NULL; 18947203048SKai Wang return (obj); 19047203048SKai Wang } 19147203048SKai Wang 192d192f3d3SKai Wang if ((obj->maddr = mmap(NULL, obj->size, PROT_READ, 193d192f3d3SKai Wang MAP_PRIVATE, obj->fd, (off_t)0)) == MAP_FAILED) { 194d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't mmap file: %s", obj->name); 195d192f3d3SKai Wang goto giveup; 196d192f3d3SKai Wang } 197d192f3d3SKai Wang if (close(obj->fd) < 0) 198d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 199d192f3d3SKai Wang obj->name); 200d192f3d3SKai Wang 201d192f3d3SKai Wang return (obj); 202d192f3d3SKai Wang 203d192f3d3SKai Wang giveup: 204d192f3d3SKai Wang if (close(obj->fd) < 0) 205d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "close failed: %s", 206d192f3d3SKai Wang obj->name); 207d192f3d3SKai Wang free(obj->name); 208d192f3d3SKai Wang free(obj); 209d192f3d3SKai Wang return (NULL); 210d192f3d3SKai Wang } 211d192f3d3SKai Wang 212d192f3d3SKai Wang /* 213d192f3d3SKai Wang * Insert obj to the tail, or before/after the pos obj. 214d192f3d3SKai Wang */ 215d192f3d3SKai Wang static void 216d192f3d3SKai Wang insert_obj(struct bsdar *bsdar, struct ar_obj *obj, struct ar_obj *pos) 217d192f3d3SKai Wang { 218d192f3d3SKai Wang if (obj == NULL) 219d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "try to insert a null obj"); 220d192f3d3SKai Wang 221d192f3d3SKai Wang if (pos == NULL || obj == pos) 222d192f3d3SKai Wang /* 223*6bef5d28SBenedict Reuschling * If the object to move happens to be the position obj, 224d192f3d3SKai Wang * or if there is not a pos obj, move it to tail. 225d192f3d3SKai Wang */ 226d192f3d3SKai Wang goto tail; 227d192f3d3SKai Wang 228d192f3d3SKai Wang if (bsdar->options & AR_B) { 229d192f3d3SKai Wang TAILQ_INSERT_BEFORE(pos, obj, objs); 230d192f3d3SKai Wang return; 231d192f3d3SKai Wang } 232d192f3d3SKai Wang if (bsdar->options & AR_A) { 233d192f3d3SKai Wang TAILQ_INSERT_AFTER(&bsdar->v_obj, pos, obj, objs); 234d192f3d3SKai Wang return; 235d192f3d3SKai Wang } 236d192f3d3SKai Wang 237d192f3d3SKai Wang tail: 238d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 239d192f3d3SKai Wang 240d192f3d3SKai Wang } 241d192f3d3SKai Wang 242d192f3d3SKai Wang /* 2430c099281SKai Wang * Read objects from archive into v_obj list. Note that checkargv is 2440c099281SKai Wang * set when read_objs is used to read objects from the target of 2450c099281SKai Wang * ADDLIB command (ar script mode), in this case argv array possibly 2460c099281SKai Wang * specifies the members ADDLIB want. 247d192f3d3SKai Wang */ 248d192f3d3SKai Wang static void 2490c099281SKai Wang read_objs(struct bsdar *bsdar, const char *archive, int checkargv) 250d192f3d3SKai Wang { 251d192f3d3SKai Wang struct archive *a; 252d192f3d3SKai Wang struct archive_entry *entry; 2530c099281SKai Wang struct ar_obj *obj; 254d192f3d3SKai Wang const char *name; 255d192f3d3SKai Wang const char *bname; 256d192f3d3SKai Wang char *buff; 257d192f3d3SKai Wang char **av; 258d192f3d3SKai Wang size_t size; 2590c099281SKai Wang int i, r, find; 260d192f3d3SKai Wang 261d192f3d3SKai Wang if ((a = archive_read_new()) == NULL) 262d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_read_new failed"); 2635dfab45bSKai Wang archive_read_support_compression_none(a); 264d192f3d3SKai Wang archive_read_support_format_ar(a); 2650c099281SKai Wang AC(archive_read_open_filename(a, archive, DEF_BLKSZ)); 266d192f3d3SKai Wang for (;;) { 267d192f3d3SKai Wang r = archive_read_next_header(a, &entry); 268d192f3d3SKai Wang if (r == ARCHIVE_FATAL) 269d192f3d3SKai Wang bsdar_errc(bsdar, EX_DATAERR, 0, "%s", 270d192f3d3SKai Wang archive_error_string(a)); 271d192f3d3SKai Wang if (r == ARCHIVE_EOF) 272d192f3d3SKai Wang break; 273d192f3d3SKai Wang if (r == ARCHIVE_WARN || r == ARCHIVE_RETRY) 274d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s", archive_error_string(a)); 275d192f3d3SKai Wang if (r == ARCHIVE_RETRY) { 276d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "Retrying..."); 277d192f3d3SKai Wang continue; 278d192f3d3SKai Wang } 279d192f3d3SKai Wang 280d192f3d3SKai Wang name = archive_entry_pathname(entry); 281d192f3d3SKai Wang 282d192f3d3SKai Wang /* 283d192f3d3SKai Wang * skip pseudo members. 284d192f3d3SKai Wang */ 285d192f3d3SKai Wang if (strcmp(name, "/") == 0 || strcmp(name, "//") == 0) 286d192f3d3SKai Wang continue; 287d192f3d3SKai Wang 2880c099281SKai Wang /* 2890c099281SKai Wang * If checkargv is set, only read those members specified 2900c099281SKai Wang * in argv. 2910c099281SKai Wang */ 2920c099281SKai Wang if (checkargv && bsdar->argc > 0) { 2930c099281SKai Wang find = 0; 2940c099281SKai Wang for(i = 0; i < bsdar->argc; i++) { 2950c099281SKai Wang av = &bsdar->argv[i]; 2960c099281SKai Wang if (*av == NULL) 2970c099281SKai Wang continue; 2980c099281SKai Wang if ((bname = basename(*av)) == NULL) 2990c099281SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 3000c099281SKai Wang "basename failed"); 3010c099281SKai Wang if (strcmp(bname, name) != 0) 3020c099281SKai Wang continue; 3030c099281SKai Wang 3040c099281SKai Wang *av = NULL; 3050c099281SKai Wang find = 1; 3060c099281SKai Wang break; 3070c099281SKai Wang } 3080c099281SKai Wang if (!find) 3090c099281SKai Wang continue; 3100c099281SKai Wang } 3110c099281SKai Wang 312d192f3d3SKai Wang size = archive_entry_size(entry); 313d192f3d3SKai Wang 314517dbe2eSKai Wang if (size > 0) { 315d192f3d3SKai Wang if ((buff = malloc(size)) == NULL) 316517dbe2eSKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 317517dbe2eSKai Wang "malloc failed"); 318d192f3d3SKai Wang if (archive_read_data(a, buff, size) != (ssize_t)size) { 319517dbe2eSKai Wang bsdar_warnc(bsdar, 0, "%s", 320517dbe2eSKai Wang archive_error_string(a)); 321d192f3d3SKai Wang free(buff); 322d192f3d3SKai Wang continue; 323d192f3d3SKai Wang } 324517dbe2eSKai Wang } else 325517dbe2eSKai Wang buff = NULL; 326517dbe2eSKai Wang 327d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 328d192f3d3SKai Wang if (obj == NULL) 329d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 330d192f3d3SKai Wang obj->maddr = buff; 331d192f3d3SKai Wang if ((obj->name = strdup(name)) == NULL) 332d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "strdup failed"); 333d192f3d3SKai Wang obj->size = size; 334d192f3d3SKai Wang obj->uid = archive_entry_uid(entry); 335d192f3d3SKai Wang obj->gid = archive_entry_gid(entry); 336d192f3d3SKai Wang obj->md = archive_entry_mode(entry); 337d192f3d3SKai Wang obj->mtime = archive_entry_mtime(entry); 338d192f3d3SKai Wang obj->dev = 0; 339d192f3d3SKai Wang obj->ino = 0; 340d192f3d3SKai Wang 341d192f3d3SKai Wang /* 342d192f3d3SKai Wang * Objects from archive have obj->fd set to -1, 343d192f3d3SKai Wang * for the ease of cleaning up. 344d192f3d3SKai Wang */ 345d192f3d3SKai Wang obj->fd = -1; 346d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 347d192f3d3SKai Wang } 348d192f3d3SKai Wang AC(archive_read_close(a)); 349d192f3d3SKai Wang AC(archive_read_finish(a)); 3500c099281SKai Wang } 3510c099281SKai Wang 3520c099281SKai Wang /* 3530c099281SKai Wang * Determine the constitution of resulting archive. 3540c099281SKai Wang */ 3550c099281SKai Wang static void 3560c099281SKai Wang write_archive(struct bsdar *bsdar, char mode) 3570c099281SKai Wang { 3580c099281SKai Wang struct ar_obj *nobj, *obj, *obj_temp, *pos; 3590c099281SKai Wang struct stat sb; 3600c099281SKai Wang const char *bname; 3610c099281SKai Wang char **av; 3620c099281SKai Wang int i; 3630c099281SKai Wang 3640c099281SKai Wang TAILQ_INIT(&bsdar->v_obj); 3650c099281SKai Wang nobj = NULL; 3660c099281SKai Wang pos = NULL; 3670c099281SKai Wang memset(&sb, 0, sizeof(sb)); 3680c099281SKai Wang 3690c099281SKai Wang /* 3700c099281SKai Wang * Test if the specified archive exists, to figure out 3710c099281SKai Wang * whether we are creating one here. 3720c099281SKai Wang */ 3730c099281SKai Wang if (stat(bsdar->filename, &sb) != 0) { 3740c099281SKai Wang if (errno != ENOENT) { 3750c099281SKai Wang bsdar_warnc(bsdar, 0, "stat %s failed", 3760c099281SKai Wang bsdar->filename); 3770c099281SKai Wang return; 3780c099281SKai Wang } 3790c099281SKai Wang 3800c099281SKai Wang /* We do not create archive in mode 'd', 'm' and 's'. */ 3810c099281SKai Wang if (mode != 'r' && mode != 'q') { 3820c099281SKai Wang bsdar_warnc(bsdar, 0, "%s: no such file", 3830c099281SKai Wang bsdar->filename); 3840c099281SKai Wang return; 3850c099281SKai Wang } 3860c099281SKai Wang 3870c099281SKai Wang /* Issue a warning if -c is not specified when creating. */ 3880c099281SKai Wang if (!(bsdar->options & AR_C)) 3890c099281SKai Wang bsdar_warnc(bsdar, 0, "creating %s", bsdar->filename); 3900c099281SKai Wang goto new_archive; 3910c099281SKai Wang } 3920c099281SKai Wang 3930c099281SKai Wang /* 3940c099281SKai Wang * First read members from existing archive. 3950c099281SKai Wang */ 3960c099281SKai Wang read_objs(bsdar, bsdar->filename, 0); 397d192f3d3SKai Wang 398d192f3d3SKai Wang /* 399d192f3d3SKai Wang * For mode 's', no member will be moved, deleted or replaced. 400d192f3d3SKai Wang */ 401d192f3d3SKai Wang if (mode == 's') 402d192f3d3SKai Wang goto write_objs; 403d192f3d3SKai Wang 404d192f3d3SKai Wang /* 405cb0dad38SKai Wang * For mode 'q', we don't need to adjust existing members either. 406cb0dad38SKai Wang * Also, -a, -b and -i are ignored in this mode. New members are 407cb0dad38SKai Wang * always inserted at tail. 408cb0dad38SKai Wang */ 409cb0dad38SKai Wang if (mode == 'q') 410cb0dad38SKai Wang goto new_archive; 411cb0dad38SKai Wang 412cb0dad38SKai Wang /* 4130c099281SKai Wang * Mode 'A' adds the contents of another archive to the tail of 4140c099281SKai Wang * current archive. Note that mode 'A' is a special mode for the 4150c099281SKai Wang * ADDLIB command of the ar script mode. Currently there is no 4160c099281SKai Wang * access to this function from the ar command line mode. 4170c099281SKai Wang */ 4180c099281SKai Wang if (mode == 'A') { 4190c099281SKai Wang /* 4200c099281SKai Wang * Read objects from the target archive of ADDLIB command. 421*6bef5d28SBenedict Reuschling * If there are members specified in argv, read those members 4220c099281SKai Wang * only, otherwise the entire archive will be read. 4230c099281SKai Wang */ 4240c099281SKai Wang read_objs(bsdar, bsdar->addlib, 1); 4250c099281SKai Wang goto write_objs; 4260c099281SKai Wang } 4270c099281SKai Wang 4280c099281SKai Wang /* 429d192f3d3SKai Wang * Try to find the position member specified by user. 430d192f3d3SKai Wang */ 431d192f3d3SKai Wang if (bsdar->options & AR_A || bsdar->options & AR_B) { 432d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 433d192f3d3SKai Wang if (strcmp(obj->name, bsdar->posarg) == 0) { 434d192f3d3SKai Wang pos = obj; 435d192f3d3SKai Wang break; 436d192f3d3SKai Wang } 437d192f3d3SKai Wang } 438d192f3d3SKai Wang 439d192f3d3SKai Wang /* 440d192f3d3SKai Wang * If can't find `pos' specified by user, 441*6bef5d28SBenedict Reuschling * silently insert objects at tail. 442d192f3d3SKai Wang */ 443d192f3d3SKai Wang if (pos == NULL) 444d192f3d3SKai Wang bsdar->options &= ~(AR_A | AR_B); 445d192f3d3SKai Wang } 446d192f3d3SKai Wang 447d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 448d192f3d3SKai Wang av = &bsdar->argv[i]; 449d192f3d3SKai Wang 450d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 451d192f3d3SKai Wang if ((bname = basename(*av)) == NULL) 452d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, 453d192f3d3SKai Wang "basename failed"); 454d192f3d3SKai Wang if (bsdar->options & AR_TR) { 455d192f3d3SKai Wang if (strncmp(bname, obj->name, _TRUNCATE_LEN)) 456d192f3d3SKai Wang continue; 457d192f3d3SKai Wang } else 458d192f3d3SKai Wang if (strcmp(bname, obj->name) != 0) 459d192f3d3SKai Wang continue; 460d192f3d3SKai Wang 461d192f3d3SKai Wang if (mode == 'r') { 462d192f3d3SKai Wang /* 463d192f3d3SKai Wang * if the new member is not qualified 464d192f3d3SKai Wang * to replace the old one, skip it. 465d192f3d3SKai Wang */ 466d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 467d192f3d3SKai Wang obj->mtime); 468d192f3d3SKai Wang if (nobj == NULL) 469d192f3d3SKai Wang goto skip_obj; 470d192f3d3SKai Wang } 471d192f3d3SKai Wang 472d192f3d3SKai Wang if (bsdar->options & AR_V) 473d192f3d3SKai Wang (void)fprintf(stdout, "%c - %s\n", mode, 474d192f3d3SKai Wang *av); 475d192f3d3SKai Wang 476d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 477d192f3d3SKai Wang if (mode == 'd' || mode == 'r') { 478d192f3d3SKai Wang free(obj->maddr); 479d192f3d3SKai Wang free(obj->name); 480d192f3d3SKai Wang free(obj); 481d192f3d3SKai Wang } 482d192f3d3SKai Wang 483d192f3d3SKai Wang if (mode == 'm') 484d192f3d3SKai Wang insert_obj(bsdar, obj, pos); 485d192f3d3SKai Wang if (mode == 'r') 486d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 487d192f3d3SKai Wang 488d192f3d3SKai Wang skip_obj: 489d192f3d3SKai Wang *av = NULL; 490d192f3d3SKai Wang break; 491d192f3d3SKai Wang } 492d192f3d3SKai Wang 493d192f3d3SKai Wang } 494d192f3d3SKai Wang 495d192f3d3SKai Wang new_archive: 496d192f3d3SKai Wang /* 497d192f3d3SKai Wang * When operating in mode 'r', directly add those user specified 498cb0dad38SKai Wang * objects which do not exist in current archive. When operating 499cb0dad38SKai Wang * in mode 'q', all objects specified in command line args are 500cb0dad38SKai Wang * appended to the archive, without comparing with existing ones. 501d192f3d3SKai Wang */ 502d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 503d192f3d3SKai Wang av = &bsdar->argv[i]; 504cb0dad38SKai Wang if (*av != NULL && (mode == 'r' || mode == 'q')) { 505d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 0); 506d192f3d3SKai Wang if (nobj != NULL) 507d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 508d192f3d3SKai Wang if (bsdar->options & AR_V && nobj != NULL) 509d192f3d3SKai Wang (void)fprintf(stdout, "a - %s\n", *av); 510d192f3d3SKai Wang *av = NULL; 511d192f3d3SKai Wang } 512d192f3d3SKai Wang } 513d192f3d3SKai Wang 514d192f3d3SKai Wang write_objs: 515d192f3d3SKai Wang write_objs(bsdar); 516d192f3d3SKai Wang write_cleanup(bsdar); 517d192f3d3SKai Wang } 518d192f3d3SKai Wang 519d192f3d3SKai Wang /* 520d192f3d3SKai Wang * Memory cleaning up. 521d192f3d3SKai Wang */ 522d192f3d3SKai Wang static void 523d192f3d3SKai Wang write_cleanup(struct bsdar *bsdar) 524d192f3d3SKai Wang { 525d192f3d3SKai Wang struct ar_obj *obj, *obj_temp; 526d192f3d3SKai Wang 527d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 528d192f3d3SKai Wang if (obj->fd == -1) 529d192f3d3SKai Wang free(obj->maddr); 530d192f3d3SKai Wang else 53147203048SKai Wang if (obj->maddr != NULL && munmap(obj->maddr, obj->size)) 532d192f3d3SKai Wang bsdar_warnc(bsdar, errno, 533d192f3d3SKai Wang "can't munmap file: %s", obj->name); 534d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 53547203048SKai Wang free(obj->name); 536d192f3d3SKai Wang free(obj); 537d192f3d3SKai Wang } 538d192f3d3SKai Wang 539d192f3d3SKai Wang free(bsdar->as); 540d192f3d3SKai Wang free(bsdar->s_so); 541d192f3d3SKai Wang free(bsdar->s_sn); 542d192f3d3SKai Wang bsdar->as = NULL; 543d192f3d3SKai Wang bsdar->s_so = NULL; 544d192f3d3SKai Wang bsdar->s_sn = NULL; 545d192f3d3SKai Wang } 546d192f3d3SKai Wang 547d192f3d3SKai Wang /* 548d192f3d3SKai Wang * Wrapper for archive_write_data(). 549d192f3d3SKai Wang */ 550d192f3d3SKai Wang static void 551d192f3d3SKai Wang write_data(struct bsdar *bsdar, struct archive *a, const void *buf, size_t s) 552d192f3d3SKai Wang { 553d192f3d3SKai Wang if (archive_write_data(a, buf, s) != (ssize_t)s) 554d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "%s", 555d192f3d3SKai Wang archive_error_string(a)); 556d192f3d3SKai Wang } 557d192f3d3SKai Wang 558d192f3d3SKai Wang /* 559d192f3d3SKai Wang * Write the resulting archive members. 560d192f3d3SKai Wang */ 561d192f3d3SKai Wang static void 562d192f3d3SKai Wang write_objs(struct bsdar *bsdar) 563d192f3d3SKai Wang { 564d192f3d3SKai Wang struct ar_obj *obj; 565d192f3d3SKai Wang struct archive *a; 566d192f3d3SKai Wang struct archive_entry *entry; 567d192f3d3SKai Wang size_t s_sz; /* size of archive symbol table. */ 568d192f3d3SKai Wang size_t pm_sz; /* size of pseudo members */ 569d192f3d3SKai Wang int i, nr; 570d192f3d3SKai Wang 571d192f3d3SKai Wang if (elf_version(EV_CURRENT) == EV_NONE) 572d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, 573d192f3d3SKai Wang "ELF library initialization failed: %s", elf_errmsg(-1)); 574d192f3d3SKai Wang 575d192f3d3SKai Wang bsdar->rela_off = 0; 576d192f3d3SKai Wang 577d192f3d3SKai Wang /* Create archive symbol table and archive string table, if need. */ 578d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 57947203048SKai Wang if (!(bsdar->options & AR_SS) && obj->maddr != NULL) 580d192f3d3SKai Wang create_symtab_entry(bsdar, obj->maddr, obj->size); 581d192f3d3SKai Wang if (strlen(obj->name) > _MAXNAMELEN_SVR4) 582d192f3d3SKai Wang add_to_ar_str_table(bsdar, obj->name); 583d192f3d3SKai Wang bsdar->rela_off += _ARHDR_LEN + obj->size + obj->size % 2; 584d192f3d3SKai Wang } 585d192f3d3SKai Wang 586d192f3d3SKai Wang /* 587d192f3d3SKai Wang * Pad the symbol name string table. It is treated specially because 588d192f3d3SKai Wang * symbol name table should be padded by a '\0', not the common '\n' 589d192f3d3SKai Wang * for other members. The size of sn table includes the pad bit. 590d192f3d3SKai Wang */ 591d192f3d3SKai Wang if (bsdar->s_cnt != 0 && bsdar->s_sn_sz % 2 != 0) 592d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 593d192f3d3SKai Wang 594d192f3d3SKai Wang /* 595d192f3d3SKai Wang * Archive string table is padded by a "\n" as the normal members. 596517dbe2eSKai Wang * The difference is that the size of archive string table counts 597d192f3d3SKai Wang * in the pad bit, while normal members' size fileds do not. 598d192f3d3SKai Wang */ 599d192f3d3SKai Wang if (bsdar->as != NULL && bsdar->as_sz % 2 != 0) 600d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 601d192f3d3SKai Wang 602d192f3d3SKai Wang /* 603d192f3d3SKai Wang * If there is a symbol table, calculate the size of pseudo members, 604517dbe2eSKai Wang * convert previously stored relative offsets to absolute ones, and 605d192f3d3SKai Wang * then make them Big Endian. 606d192f3d3SKai Wang * 607d192f3d3SKai Wang * absolute_offset = htobe32(relative_offset + size_of_pseudo_members) 608d192f3d3SKai Wang */ 609d192f3d3SKai Wang 610d192f3d3SKai Wang if (bsdar->s_cnt != 0) { 611d192f3d3SKai Wang s_sz = (bsdar->s_cnt + 1) * sizeof(uint32_t) + bsdar->s_sn_sz; 612d192f3d3SKai Wang pm_sz = _ARMAG_LEN + (_ARHDR_LEN + s_sz); 613d192f3d3SKai Wang if (bsdar->as != NULL) 614d192f3d3SKai Wang pm_sz += _ARHDR_LEN + bsdar->as_sz; 615d192f3d3SKai Wang for (i = 0; (size_t)i < bsdar->s_cnt; i++) 616d192f3d3SKai Wang *(bsdar->s_so + i) = htobe32(*(bsdar->s_so + i) + 617d192f3d3SKai Wang pm_sz); 618d192f3d3SKai Wang } 619d192f3d3SKai Wang 620d192f3d3SKai Wang if ((a = archive_write_new()) == NULL) 621d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, 0, "archive_write_new failed"); 622d192f3d3SKai Wang 623d192f3d3SKai Wang archive_write_set_format_ar_svr4(a); 624d192f3d3SKai Wang archive_write_set_compression_none(a); 625d192f3d3SKai Wang 626d192f3d3SKai Wang AC(archive_write_open_filename(a, bsdar->filename)); 627d192f3d3SKai Wang 628d192f3d3SKai Wang /* 629d192f3d3SKai Wang * write the archive symbol table, if there is one. 630d192f3d3SKai Wang * If options -s is explicitly specified or we are invoked 631d192f3d3SKai Wang * as ranlib, write the symbol table even if it is empty. 632d192f3d3SKai Wang */ 633d192f3d3SKai Wang if ((bsdar->s_cnt != 0 && !(bsdar->options & AR_SS)) || 634d192f3d3SKai Wang bsdar->options & AR_S) { 635d192f3d3SKai Wang entry = archive_entry_new(); 636d192f3d3SKai Wang archive_entry_copy_pathname(entry, "/"); 6370e479ac8STim Kientzle if ((bsdar->options & AR_D) == 0) 638d192f3d3SKai Wang archive_entry_set_mtime(entry, time(NULL), 0); 639d192f3d3SKai Wang archive_entry_set_size(entry, (bsdar->s_cnt + 1) * 640d192f3d3SKai Wang sizeof(uint32_t) + bsdar->s_sn_sz); 641d192f3d3SKai Wang AC(archive_write_header(a, entry)); 642d192f3d3SKai Wang nr = htobe32(bsdar->s_cnt); 643d192f3d3SKai Wang write_data(bsdar, a, &nr, sizeof(uint32_t)); 644d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_so, sizeof(uint32_t) * 645d192f3d3SKai Wang bsdar->s_cnt); 646d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_sn, bsdar->s_sn_sz); 647d192f3d3SKai Wang archive_entry_free(entry); 648d192f3d3SKai Wang } 649d192f3d3SKai Wang 650d192f3d3SKai Wang /* write the archive string table, if any. */ 651d192f3d3SKai Wang if (bsdar->as != NULL) { 652d192f3d3SKai Wang entry = archive_entry_new(); 653d192f3d3SKai Wang archive_entry_copy_pathname(entry, "//"); 654d192f3d3SKai Wang archive_entry_set_size(entry, bsdar->as_sz); 655d192f3d3SKai Wang AC(archive_write_header(a, entry)); 656d192f3d3SKai Wang write_data(bsdar, a, bsdar->as, bsdar->as_sz); 657d192f3d3SKai Wang archive_entry_free(entry); 658d192f3d3SKai Wang } 659d192f3d3SKai Wang 660d192f3d3SKai Wang /* write normal members. */ 661d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 662d192f3d3SKai Wang entry = archive_entry_new(); 663d192f3d3SKai Wang archive_entry_copy_pathname(entry, obj->name); 664d192f3d3SKai Wang archive_entry_set_uid(entry, obj->uid); 665d192f3d3SKai Wang archive_entry_set_gid(entry, obj->gid); 666d192f3d3SKai Wang archive_entry_set_mode(entry, obj->md); 667d192f3d3SKai Wang archive_entry_set_size(entry, obj->size); 668d192f3d3SKai Wang archive_entry_set_mtime(entry, obj->mtime, 0); 669d192f3d3SKai Wang archive_entry_set_dev(entry, obj->dev); 670d192f3d3SKai Wang archive_entry_set_ino(entry, obj->ino); 671d192f3d3SKai Wang archive_entry_set_filetype(entry, AE_IFREG); 672d192f3d3SKai Wang AC(archive_write_header(a, entry)); 673d192f3d3SKai Wang write_data(bsdar, a, obj->maddr, obj->size); 674d192f3d3SKai Wang archive_entry_free(entry); 675d192f3d3SKai Wang } 676d192f3d3SKai Wang 677d192f3d3SKai Wang AC(archive_write_close(a)); 678d192f3d3SKai Wang AC(archive_write_finish(a)); 679d192f3d3SKai Wang } 680d192f3d3SKai Wang 681d192f3d3SKai Wang /* 682d192f3d3SKai Wang * Extract global symbols from ELF binary members. 683d192f3d3SKai Wang */ 684d192f3d3SKai Wang static void 685d192f3d3SKai Wang create_symtab_entry(struct bsdar *bsdar, void *maddr, size_t size) 686d192f3d3SKai Wang { 687d192f3d3SKai Wang Elf *e; 688d192f3d3SKai Wang Elf_Scn *scn; 689d192f3d3SKai Wang GElf_Shdr shdr; 690d192f3d3SKai Wang GElf_Sym sym; 691d192f3d3SKai Wang Elf_Data *data; 692d192f3d3SKai Wang char *name; 693d192f3d3SKai Wang size_t n, shstrndx; 694d192f3d3SKai Wang int elferr, tabndx, len, i; 695d192f3d3SKai Wang 696d192f3d3SKai Wang if ((e = elf_memory(maddr, size)) == NULL) { 697d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_memory() failed: %s", 698d192f3d3SKai Wang elf_errmsg(-1)); 699d192f3d3SKai Wang return; 700d192f3d3SKai Wang } 701d192f3d3SKai Wang if (elf_kind(e) != ELF_K_ELF) { 702*6bef5d28SBenedict Reuschling /* Silently ignore non-elf member. */ 703d192f3d3SKai Wang elf_end(e); 704d192f3d3SKai Wang return; 705d192f3d3SKai Wang } 706d192f3d3SKai Wang if (elf_getshstrndx(e, &shstrndx) == 0) { 707d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_getshstrndx failed: %s", 708d192f3d3SKai Wang elf_errmsg(-1)); 709d192f3d3SKai Wang elf_end(e); 710d192f3d3SKai Wang return; 711d192f3d3SKai Wang } 712d192f3d3SKai Wang 713d192f3d3SKai Wang tabndx = -1; 714d192f3d3SKai Wang scn = NULL; 715d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 716d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 717d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 718d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 719d192f3d3SKai Wang continue; 720d192f3d3SKai Wang } 721d192f3d3SKai Wang if ((name = elf_strptr(e, shstrndx, shdr.sh_name)) == NULL) { 722d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 723d192f3d3SKai Wang "elf_strptr failed: %s", elf_errmsg(-1)); 724d192f3d3SKai Wang continue; 725d192f3d3SKai Wang } 726d192f3d3SKai Wang if (strcmp(name, ".strtab") == 0) { 727d192f3d3SKai Wang tabndx = elf_ndxscn(scn); 728d192f3d3SKai Wang break; 729d192f3d3SKai Wang } 730d192f3d3SKai Wang } 731d192f3d3SKai Wang elferr = elf_errno(); 732d192f3d3SKai Wang if (elferr != 0) 733d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_nextscn failed: %s", 734d192f3d3SKai Wang elf_errmsg(elferr)); 735d192f3d3SKai Wang if (tabndx == -1) { 736d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "can't find .strtab section"); 737d192f3d3SKai Wang elf_end(e); 738d192f3d3SKai Wang return; 739d192f3d3SKai Wang } 740d192f3d3SKai Wang 741d192f3d3SKai Wang scn = NULL; 742d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 743d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 744d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 745d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 746d192f3d3SKai Wang continue; 747d192f3d3SKai Wang } 748d192f3d3SKai Wang if (shdr.sh_type != SHT_SYMTAB) 749d192f3d3SKai Wang continue; 750d192f3d3SKai Wang 751d192f3d3SKai Wang data = NULL; 752d192f3d3SKai Wang n = 0; 753d192f3d3SKai Wang while (n < shdr.sh_size && 754d192f3d3SKai Wang (data = elf_getdata(scn, data)) != NULL) { 755d192f3d3SKai Wang len = data->d_size / shdr.sh_entsize; 756d192f3d3SKai Wang for (i = 0; i < len; i++) { 757d192f3d3SKai Wang if (gelf_getsym(data, i, &sym) != &sym) { 758d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 759d192f3d3SKai Wang "gelf_getsym failed: %s", 760d192f3d3SKai Wang elf_errmsg(-1)); 761d192f3d3SKai Wang continue; 762d192f3d3SKai Wang } 763d192f3d3SKai Wang 764d192f3d3SKai Wang /* keep only global or weak symbols */ 765d192f3d3SKai Wang if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL && 766d192f3d3SKai Wang GELF_ST_BIND(sym.st_info) != STB_WEAK) 767d192f3d3SKai Wang continue; 768d192f3d3SKai Wang 769d192f3d3SKai Wang /* keep only defined symbols */ 770d192f3d3SKai Wang if (sym.st_shndx == SHN_UNDEF) 771d192f3d3SKai Wang continue; 772d192f3d3SKai Wang 773d192f3d3SKai Wang if ((name = elf_strptr(e, tabndx, 774d192f3d3SKai Wang sym.st_name)) == NULL) { 775d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, 776d192f3d3SKai Wang "elf_strptr failed: %s", 777d192f3d3SKai Wang elf_errmsg(-1)); 778d192f3d3SKai Wang continue; 779d192f3d3SKai Wang } 780d192f3d3SKai Wang 781d192f3d3SKai Wang add_to_ar_sym_table(bsdar, name); 782d192f3d3SKai Wang } 783d192f3d3SKai Wang } 784d192f3d3SKai Wang } 785d192f3d3SKai Wang elferr = elf_errno(); 786d192f3d3SKai Wang if (elferr != 0) 787d192f3d3SKai Wang bsdar_warnc(bsdar, EX_SOFTWARE, 0, "elf_nextscn failed: %s", 788d192f3d3SKai Wang elf_errmsg(elferr)); 789d192f3d3SKai Wang 790d192f3d3SKai Wang elf_end(e); 791d192f3d3SKai Wang } 792d192f3d3SKai Wang 793d192f3d3SKai Wang /* 794d192f3d3SKai Wang * Append to the archive string table buffer. 795d192f3d3SKai Wang */ 796d192f3d3SKai Wang static void 797d192f3d3SKai Wang add_to_ar_str_table(struct bsdar *bsdar, const char *name) 798d192f3d3SKai Wang { 799d192f3d3SKai Wang 800d192f3d3SKai Wang if (bsdar->as == NULL) { 801d192f3d3SKai Wang bsdar->as_cap = _INIT_AS_CAP; 802d192f3d3SKai Wang bsdar->as_sz = 0; 803d192f3d3SKai Wang if ((bsdar->as = malloc(bsdar->as_cap)) == NULL) 804d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 805d192f3d3SKai Wang } 806d192f3d3SKai Wang 807d192f3d3SKai Wang /* 808d192f3d3SKai Wang * The space required for holding one member name in as table includes: 809d192f3d3SKai Wang * strlen(name) + (1 for '/') + (1 for '\n') + (possibly 1 for padding). 810d192f3d3SKai Wang */ 811517dbe2eSKai Wang while (bsdar->as_sz + strlen(name) + 3 > bsdar->as_cap) { 812d192f3d3SKai Wang bsdar->as_cap *= 2; 813d192f3d3SKai Wang bsdar->as = realloc(bsdar->as, bsdar->as_cap); 814d192f3d3SKai Wang if (bsdar->as == NULL) 815d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 816d192f3d3SKai Wang } 817d192f3d3SKai Wang strncpy(&bsdar->as[bsdar->as_sz], name, strlen(name)); 818d192f3d3SKai Wang bsdar->as_sz += strlen(name); 819d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '/'; 820d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 821d192f3d3SKai Wang } 822d192f3d3SKai Wang 823d192f3d3SKai Wang /* 824d192f3d3SKai Wang * Append to the archive symbol table buffer. 825d192f3d3SKai Wang */ 826d192f3d3SKai Wang static void 827d192f3d3SKai Wang add_to_ar_sym_table(struct bsdar *bsdar, const char *name) 828d192f3d3SKai Wang { 829d192f3d3SKai Wang 830d192f3d3SKai Wang if (bsdar->s_so == NULL) { 831d192f3d3SKai Wang if ((bsdar->s_so = malloc(_INIT_SYMOFF_CAP)) == 832d192f3d3SKai Wang NULL) 833d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 834d192f3d3SKai Wang bsdar->s_so_cap = _INIT_SYMOFF_CAP; 835d192f3d3SKai Wang bsdar->s_cnt = 0; 836d192f3d3SKai Wang } 837d192f3d3SKai Wang 838d192f3d3SKai Wang if (bsdar->s_sn == NULL) { 839d192f3d3SKai Wang if ((bsdar->s_sn = malloc(_INIT_SYMNAME_CAP)) == NULL) 840d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "malloc failed"); 841d192f3d3SKai Wang bsdar->s_sn_cap = _INIT_SYMNAME_CAP; 842d192f3d3SKai Wang bsdar->s_sn_sz = 0; 843d192f3d3SKai Wang } 844d192f3d3SKai Wang 845d192f3d3SKai Wang if (bsdar->s_cnt * sizeof(uint32_t) >= bsdar->s_so_cap) { 846d192f3d3SKai Wang bsdar->s_so_cap *= 2; 847d192f3d3SKai Wang bsdar->s_so = realloc(bsdar->s_so, bsdar->s_so_cap); 848d192f3d3SKai Wang if (bsdar->s_so == NULL) 849d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 850d192f3d3SKai Wang } 851d192f3d3SKai Wang bsdar->s_so[bsdar->s_cnt] = bsdar->rela_off; 852d192f3d3SKai Wang bsdar->s_cnt++; 853d192f3d3SKai Wang 854d192f3d3SKai Wang /* 855d192f3d3SKai Wang * The space required for holding one symbol name in sn table includes: 856d192f3d3SKai Wang * strlen(name) + (1 for '\n') + (possibly 1 for padding). 857d192f3d3SKai Wang */ 858517dbe2eSKai Wang while (bsdar->s_sn_sz + strlen(name) + 2 > bsdar->s_sn_cap) { 859d192f3d3SKai Wang bsdar->s_sn_cap *= 2; 860d192f3d3SKai Wang bsdar->s_sn = realloc(bsdar->s_sn, bsdar->s_sn_cap); 861d192f3d3SKai Wang if (bsdar->s_sn == NULL) 862d192f3d3SKai Wang bsdar_errc(bsdar, EX_SOFTWARE, errno, "realloc failed"); 863d192f3d3SKai Wang } 864d192f3d3SKai Wang strncpy(&bsdar->s_sn[bsdar->s_sn_sz], name, strlen(name)); 865d192f3d3SKai Wang bsdar->s_sn_sz += strlen(name); 866d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 867d192f3d3SKai Wang } 868