1d192f3d3SKai Wang /*- 21de7b4b8SPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 31de7b4b8SPedro F. Giffuni * 4d192f3d3SKai Wang * Copyright (c) 2007 Kai Wang 5d192f3d3SKai Wang * All rights reserved. 6d192f3d3SKai Wang * 7d192f3d3SKai Wang * Redistribution and use in source and binary forms, with or without 8d192f3d3SKai Wang * modification, are permitted provided that the following conditions 9d192f3d3SKai Wang * are met: 10d192f3d3SKai Wang * 1. Redistributions of source code must retain the above copyright 1125f82d56SEd Maste * notice, this list of conditions and the following disclaimer. 12d192f3d3SKai Wang * 2. Redistributions in binary form must reproduce the above copyright 13d192f3d3SKai Wang * notice, this list of conditions and the following disclaimer in the 14d192f3d3SKai Wang * documentation and/or other materials provided with the distribution. 15d192f3d3SKai Wang * 1625f82d56SEd Maste * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 1725f82d56SEd Maste * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 1825f82d56SEd Maste * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 1925f82d56SEd Maste * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 2025f82d56SEd Maste * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 2125f82d56SEd Maste * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2225f82d56SEd Maste * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 2325f82d56SEd Maste * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 2425f82d56SEd Maste * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 2525f82d56SEd Maste * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 2625f82d56SEd Maste * SUCH DAMAGE. 27d192f3d3SKai Wang */ 28d192f3d3SKai Wang 29d192f3d3SKai Wang #include <sys/cdefs.h> 30d192f3d3SKai Wang __FBSDID("$FreeBSD$"); 31d192f3d3SKai Wang 32d192f3d3SKai Wang #include <sys/endian.h> 33d192f3d3SKai Wang #include <sys/mman.h> 34d192f3d3SKai Wang #include <sys/queue.h> 35d192f3d3SKai Wang #include <sys/stat.h> 36d192f3d3SKai Wang #include <archive.h> 37d192f3d3SKai Wang #include <archive_entry.h> 38*57aaefdcSEd Maste #include <assert.h> 39d192f3d3SKai Wang #include <errno.h> 40d192f3d3SKai Wang #include <fcntl.h> 41d192f3d3SKai Wang #include <gelf.h> 42d192f3d3SKai Wang #include <libgen.h> 43d192f3d3SKai Wang #include <stdio.h> 44d192f3d3SKai Wang #include <stdlib.h> 45d192f3d3SKai Wang #include <string.h> 4601715d8cSEd Maste #include <unistd.h> 47d192f3d3SKai Wang 48d192f3d3SKai Wang #include "ar.h" 49d192f3d3SKai Wang 50d192f3d3SKai Wang #define _ARMAG_LEN 8 /* length of ar magic string */ 51d192f3d3SKai Wang #define _ARHDR_LEN 60 /* length of ar header */ 52d192f3d3SKai Wang #define _INIT_AS_CAP 128 /* initial archive string table size */ 53fabed6b2SEd Maste #define _INIT_SYMOFF_CAP (256*(sizeof(uint64_t))) /* initial so table size */ 54d192f3d3SKai Wang #define _INIT_SYMNAME_CAP 1024 /* initial sn table size */ 55d192f3d3SKai Wang #define _MAXNAMELEN_SVR4 15 /* max member name length in svr4 variant */ 56d192f3d3SKai Wang #define _TRUNCATE_LEN 15 /* number of bytes to keep for member name */ 57d192f3d3SKai Wang 58d192f3d3SKai Wang static void add_to_ar_str_table(struct bsdar *bsdar, const char *name); 59d192f3d3SKai Wang static void add_to_ar_sym_table(struct bsdar *bsdar, const char *name); 60d192f3d3SKai Wang static struct ar_obj *create_obj_from_file(struct bsdar *bsdar, 61d192f3d3SKai Wang const char *name, time_t mtime); 62d192f3d3SKai Wang static void create_symtab_entry(struct bsdar *bsdar, void *maddr, 63d192f3d3SKai Wang size_t size); 64cf933f5bSEd Maste static void free_obj(struct bsdar *bsdar, struct ar_obj *obj); 65d192f3d3SKai Wang static void insert_obj(struct bsdar *bsdar, struct ar_obj *obj, 66d192f3d3SKai Wang struct ar_obj *pos); 6701715d8cSEd Maste static void prefault_buffer(const char *buf, size_t s); 680c099281SKai Wang static void read_objs(struct bsdar *bsdar, const char *archive, 690c099281SKai Wang int checkargv); 70d192f3d3SKai Wang static void write_cleanup(struct bsdar *bsdar); 71d192f3d3SKai Wang static void write_data(struct bsdar *bsdar, struct archive *a, 72d192f3d3SKai Wang const void *buf, size_t s); 73d192f3d3SKai Wang static void write_objs(struct bsdar *bsdar); 74d192f3d3SKai Wang 75d192f3d3SKai Wang /* 76d192f3d3SKai Wang * Create object from file, return created obj upon success, or NULL 77d192f3d3SKai Wang * when an error occurs or the member is not newer than existing 786bef5d28SBenedict Reuschling * one while -u is specified. 79d192f3d3SKai Wang */ 80d192f3d3SKai Wang static struct ar_obj * 81d192f3d3SKai Wang create_obj_from_file(struct bsdar *bsdar, const char *name, time_t mtime) 82d192f3d3SKai Wang { 83d192f3d3SKai Wang struct ar_obj *obj; 84d192f3d3SKai Wang struct stat sb; 85d192f3d3SKai Wang const char *bname; 8609a11349SEd Schouten char *tmpname; 87d192f3d3SKai Wang 88d192f3d3SKai Wang if (name == NULL) 89d192f3d3SKai Wang return (NULL); 90d192f3d3SKai Wang 91d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 92d192f3d3SKai Wang if (obj == NULL) 93086f0908SEd Maste bsdar_errc(bsdar, errno, "malloc failed"); 94d192f3d3SKai Wang if ((obj->fd = open(name, O_RDONLY, 0)) < 0) { 95d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't open file: %s", name); 96d192f3d3SKai Wang free(obj); 97d192f3d3SKai Wang return (NULL); 98d192f3d3SKai Wang } 99d192f3d3SKai Wang 10009a11349SEd Schouten tmpname = strdup(name); 10109a11349SEd Schouten if (tmpname == NULL) 102086f0908SEd Maste bsdar_errc(bsdar, errno, "strdup failed"); 10309a11349SEd Schouten if ((bname = basename(tmpname)) == NULL) 104086f0908SEd Maste bsdar_errc(bsdar, errno, "basename failed"); 105d192f3d3SKai Wang if (bsdar->options & AR_TR && strlen(bname) > _TRUNCATE_LEN) { 106d192f3d3SKai Wang if ((obj->name = malloc(_TRUNCATE_LEN + 1)) == NULL) 107086f0908SEd Maste bsdar_errc(bsdar, errno, "malloc failed"); 108d192f3d3SKai Wang (void)strncpy(obj->name, bname, _TRUNCATE_LEN); 109d192f3d3SKai Wang obj->name[_TRUNCATE_LEN] = '\0'; 110d192f3d3SKai Wang } else 111d192f3d3SKai Wang if ((obj->name = strdup(bname)) == NULL) 112086f0908SEd Maste bsdar_errc(bsdar, errno, "strdup failed"); 11309a11349SEd Schouten free(tmpname); 114d192f3d3SKai Wang 115d192f3d3SKai Wang if (fstat(obj->fd, &sb) < 0) { 116d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't fstat file: %s", obj->name); 117d192f3d3SKai Wang goto giveup; 118d192f3d3SKai Wang } 119d192f3d3SKai Wang if (!S_ISREG(sb.st_mode)) { 120d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "%s is not an ordinary file", obj->name); 121d192f3d3SKai Wang goto giveup; 122d192f3d3SKai Wang } 123d192f3d3SKai Wang 124d192f3d3SKai Wang /* 125d192f3d3SKai Wang * When option '-u' is specified and member is not newer than the 126d192f3d3SKai Wang * existing one, the replace will not happen. While if mtime == 0, 127d192f3d3SKai Wang * which indicates that this is to "replace a none exist member", 128d192f3d3SKai Wang * the replace will proceed regardless of '-u'. 129d192f3d3SKai Wang */ 130d192f3d3SKai Wang if (mtime != 0 && bsdar->options & AR_U && sb.st_mtime <= mtime) 131d192f3d3SKai Wang goto giveup; 132d192f3d3SKai Wang 1330e479ac8STim Kientzle /* 1340e479ac8STim Kientzle * When option '-D' is specified, mtime and UID / GID from the file 1350e479ac8STim Kientzle * will be replaced with 0, and file mode with 644. This ensures that 1360e479ac8STim Kientzle * checksums will match for two archives containing the exact same 1370e479ac8STim Kientzle * files. 1380e479ac8STim Kientzle */ 1390e479ac8STim Kientzle if (bsdar->options & AR_D) { 1400e479ac8STim Kientzle obj->uid = 0; 1410e479ac8STim Kientzle obj->gid = 0; 1420e479ac8STim Kientzle obj->mtime = 0; 1430e479ac8STim Kientzle obj->md = S_IFREG | 0644; 1440e479ac8STim Kientzle } else { 145d192f3d3SKai Wang obj->uid = sb.st_uid; 146d192f3d3SKai Wang obj->gid = sb.st_gid; 147d192f3d3SKai Wang obj->mtime = sb.st_mtime; 1480e479ac8STim Kientzle obj->md = sb.st_mode; 1490e479ac8STim Kientzle } 1500e479ac8STim Kientzle obj->size = sb.st_size; 151d192f3d3SKai Wang obj->dev = sb.st_dev; 152d192f3d3SKai Wang obj->ino = sb.st_ino; 15347203048SKai Wang 15447203048SKai Wang if (obj->size == 0) { 15547203048SKai Wang obj->maddr = NULL; 15647203048SKai Wang return (obj); 15747203048SKai Wang } 15847203048SKai Wang 159d192f3d3SKai Wang if ((obj->maddr = mmap(NULL, obj->size, PROT_READ, 160d192f3d3SKai Wang MAP_PRIVATE, obj->fd, (off_t)0)) == MAP_FAILED) { 161d192f3d3SKai Wang bsdar_warnc(bsdar, errno, "can't mmap file: %s", obj->name); 162d192f3d3SKai Wang goto giveup; 163d192f3d3SKai Wang } 164d192f3d3SKai Wang if (close(obj->fd) < 0) 165086f0908SEd Maste bsdar_errc(bsdar, errno, "close failed: %s", 166d192f3d3SKai Wang obj->name); 167d192f3d3SKai Wang 168d192f3d3SKai Wang return (obj); 169d192f3d3SKai Wang 170d192f3d3SKai Wang giveup: 171d192f3d3SKai Wang if (close(obj->fd) < 0) 172086f0908SEd Maste bsdar_errc(bsdar, errno, "close failed: %s", 173d192f3d3SKai Wang obj->name); 174d192f3d3SKai Wang free(obj->name); 175d192f3d3SKai Wang free(obj); 176d192f3d3SKai Wang return (NULL); 177d192f3d3SKai Wang } 178d192f3d3SKai Wang 179d192f3d3SKai Wang /* 180cf933f5bSEd Maste * Free object itself and its associated allocations. 181cf933f5bSEd Maste */ 182cf933f5bSEd Maste static void 183cf933f5bSEd Maste free_obj(struct bsdar *bsdar, struct ar_obj *obj) 184cf933f5bSEd Maste { 185cf933f5bSEd Maste if (obj->fd == -1) 186cf933f5bSEd Maste free(obj->maddr); 187cf933f5bSEd Maste else 188cf933f5bSEd Maste if (obj->maddr != NULL && munmap(obj->maddr, obj->size)) 189cf933f5bSEd Maste bsdar_warnc(bsdar, errno, 190cf933f5bSEd Maste "can't munmap file: %s", obj->name); 191cf933f5bSEd Maste free(obj->name); 192cf933f5bSEd Maste free(obj); 193cf933f5bSEd Maste } 194cf933f5bSEd Maste 195cf933f5bSEd Maste /* 196d192f3d3SKai Wang * Insert obj to the tail, or before/after the pos obj. 197d192f3d3SKai Wang */ 198d192f3d3SKai Wang static void 199d192f3d3SKai Wang insert_obj(struct bsdar *bsdar, struct ar_obj *obj, struct ar_obj *pos) 200d192f3d3SKai Wang { 201d192f3d3SKai Wang if (obj == NULL) 202086f0908SEd Maste bsdar_errc(bsdar, 0, "try to insert a null obj"); 203d192f3d3SKai Wang 204d192f3d3SKai Wang if (pos == NULL || obj == pos) 205d192f3d3SKai Wang /* 2066bef5d28SBenedict Reuschling * If the object to move happens to be the position obj, 207d192f3d3SKai Wang * or if there is not a pos obj, move it to tail. 208d192f3d3SKai Wang */ 209d192f3d3SKai Wang goto tail; 210d192f3d3SKai Wang 211d192f3d3SKai Wang if (bsdar->options & AR_B) { 212d192f3d3SKai Wang TAILQ_INSERT_BEFORE(pos, obj, objs); 213d192f3d3SKai Wang return; 214d192f3d3SKai Wang } 215d192f3d3SKai Wang if (bsdar->options & AR_A) { 216d192f3d3SKai Wang TAILQ_INSERT_AFTER(&bsdar->v_obj, pos, obj, objs); 217d192f3d3SKai Wang return; 218d192f3d3SKai Wang } 219d192f3d3SKai Wang 220d192f3d3SKai Wang tail: 221d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 222d192f3d3SKai Wang 223d192f3d3SKai Wang } 224d192f3d3SKai Wang 225d192f3d3SKai Wang /* 2260c099281SKai Wang * Read objects from archive into v_obj list. Note that checkargv is 2270c099281SKai Wang * set when read_objs is used to read objects from the target of 2280c099281SKai Wang * ADDLIB command (ar script mode), in this case argv array possibly 2290c099281SKai Wang * specifies the members ADDLIB want. 230d192f3d3SKai Wang */ 231d192f3d3SKai Wang static void 2320c099281SKai Wang read_objs(struct bsdar *bsdar, const char *archive, int checkargv) 233d192f3d3SKai Wang { 234d192f3d3SKai Wang struct archive *a; 235d192f3d3SKai Wang struct archive_entry *entry; 2360c099281SKai Wang struct ar_obj *obj; 237d192f3d3SKai Wang const char *name; 238d192f3d3SKai Wang const char *bname; 239d192f3d3SKai Wang char *buff; 240d192f3d3SKai Wang char **av; 241d192f3d3SKai Wang size_t size; 2420c099281SKai Wang int i, r, find; 243d192f3d3SKai Wang 244d192f3d3SKai Wang if ((a = archive_read_new()) == NULL) 245086f0908SEd Maste bsdar_errc(bsdar, 0, "archive_read_new failed"); 246d192f3d3SKai Wang archive_read_support_format_ar(a); 2470c099281SKai Wang AC(archive_read_open_filename(a, archive, DEF_BLKSZ)); 248d192f3d3SKai Wang for (;;) { 249d192f3d3SKai Wang r = archive_read_next_header(a, &entry); 250d192f3d3SKai Wang if (r == ARCHIVE_FATAL) 251086f0908SEd Maste bsdar_errc(bsdar, archive_errno(a), "%s", 252d192f3d3SKai Wang archive_error_string(a)); 253d192f3d3SKai Wang if (r == ARCHIVE_EOF) 254d192f3d3SKai Wang break; 255d192f3d3SKai Wang if (r == ARCHIVE_WARN || r == ARCHIVE_RETRY) 256ccd8660bSEd Maste bsdar_warnc(bsdar, archive_errno(a), "%s", 257ccd8660bSEd Maste archive_error_string(a)); 258d192f3d3SKai Wang if (r == ARCHIVE_RETRY) { 259d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "Retrying..."); 260d192f3d3SKai Wang continue; 261d192f3d3SKai Wang } 262d192f3d3SKai Wang 263d192f3d3SKai Wang name = archive_entry_pathname(entry); 264d192f3d3SKai Wang 265d192f3d3SKai Wang /* 266d192f3d3SKai Wang * skip pseudo members. 267d192f3d3SKai Wang */ 268d192f3d3SKai Wang if (strcmp(name, "/") == 0 || strcmp(name, "//") == 0) 269d192f3d3SKai Wang continue; 270d192f3d3SKai Wang 2710c099281SKai Wang /* 2720c099281SKai Wang * If checkargv is set, only read those members specified 2730c099281SKai Wang * in argv. 2740c099281SKai Wang */ 2750c099281SKai Wang if (checkargv && bsdar->argc > 0) { 2760c099281SKai Wang find = 0; 2770c099281SKai Wang for(i = 0; i < bsdar->argc; i++) { 2780c099281SKai Wang av = &bsdar->argv[i]; 2790c099281SKai Wang if (*av == NULL) 2800c099281SKai Wang continue; 2810c099281SKai Wang if ((bname = basename(*av)) == NULL) 282086f0908SEd Maste bsdar_errc(bsdar, errno, 2830c099281SKai Wang "basename failed"); 2840c099281SKai Wang if (strcmp(bname, name) != 0) 2850c099281SKai Wang continue; 2860c099281SKai Wang 2870c099281SKai Wang *av = NULL; 2880c099281SKai Wang find = 1; 2890c099281SKai Wang break; 2900c099281SKai Wang } 2910c099281SKai Wang if (!find) 2920c099281SKai Wang continue; 2930c099281SKai Wang } 2940c099281SKai Wang 295d192f3d3SKai Wang size = archive_entry_size(entry); 296d192f3d3SKai Wang 297517dbe2eSKai Wang if (size > 0) { 298d192f3d3SKai Wang if ((buff = malloc(size)) == NULL) 299086f0908SEd Maste bsdar_errc(bsdar, errno, "malloc failed"); 300d192f3d3SKai Wang if (archive_read_data(a, buff, size) != (ssize_t)size) { 301ccd8660bSEd Maste bsdar_warnc(bsdar, archive_errno(a), "%s", 302517dbe2eSKai Wang archive_error_string(a)); 303d192f3d3SKai Wang free(buff); 304d192f3d3SKai Wang continue; 305d192f3d3SKai Wang } 306517dbe2eSKai Wang } else 307517dbe2eSKai Wang buff = NULL; 308517dbe2eSKai Wang 309d192f3d3SKai Wang obj = malloc(sizeof(struct ar_obj)); 310d192f3d3SKai Wang if (obj == NULL) 311086f0908SEd Maste bsdar_errc(bsdar, errno, "malloc failed"); 312d192f3d3SKai Wang obj->maddr = buff; 313d192f3d3SKai Wang if ((obj->name = strdup(name)) == NULL) 314086f0908SEd Maste bsdar_errc(bsdar, errno, "strdup failed"); 315d192f3d3SKai Wang obj->size = size; 316d192f3d3SKai Wang obj->uid = archive_entry_uid(entry); 317d192f3d3SKai Wang obj->gid = archive_entry_gid(entry); 318d192f3d3SKai Wang obj->md = archive_entry_mode(entry); 319d192f3d3SKai Wang obj->mtime = archive_entry_mtime(entry); 320d192f3d3SKai Wang obj->dev = 0; 321d192f3d3SKai Wang obj->ino = 0; 322d192f3d3SKai Wang 323d192f3d3SKai Wang /* 324d192f3d3SKai Wang * Objects from archive have obj->fd set to -1, 325d192f3d3SKai Wang * for the ease of cleaning up. 326d192f3d3SKai Wang */ 327d192f3d3SKai Wang obj->fd = -1; 328d192f3d3SKai Wang TAILQ_INSERT_TAIL(&bsdar->v_obj, obj, objs); 329d192f3d3SKai Wang } 330d192f3d3SKai Wang AC(archive_read_close(a)); 331ebb8fc42SMartin Matuska AC(archive_read_free(a)); 3320c099281SKai Wang } 3330c099281SKai Wang 3340c099281SKai Wang /* 3350c099281SKai Wang * Determine the constitution of resulting archive. 3360c099281SKai Wang */ 337*57aaefdcSEd Maste int 338*57aaefdcSEd Maste ar_write_archive(struct bsdar *bsdar, int mode) 3390c099281SKai Wang { 3400c099281SKai Wang struct ar_obj *nobj, *obj, *obj_temp, *pos; 3410c099281SKai Wang struct stat sb; 3420c099281SKai Wang const char *bname; 3430c099281SKai Wang char **av; 34438911b3cSEd Maste int exitcode, i; 3450c099281SKai Wang 3460c099281SKai Wang TAILQ_INIT(&bsdar->v_obj); 34738911b3cSEd Maste exitcode = EXIT_SUCCESS; 3480c099281SKai Wang nobj = NULL; 3490c099281SKai Wang pos = NULL; 3500c099281SKai Wang memset(&sb, 0, sizeof(sb)); 3510c099281SKai Wang 352*57aaefdcSEd Maste assert(mode == 'A' || mode == 'd' || mode == 'm' || mode == 'q' || 353*57aaefdcSEd Maste mode == 'r' || mode == 's'); 354*57aaefdcSEd Maste 3550c099281SKai Wang /* 3560c099281SKai Wang * Test if the specified archive exists, to figure out 3570c099281SKai Wang * whether we are creating one here. 3580c099281SKai Wang */ 3590c099281SKai Wang if (stat(bsdar->filename, &sb) != 0) { 3600c099281SKai Wang if (errno != ENOENT) { 3610c099281SKai Wang bsdar_warnc(bsdar, 0, "stat %s failed", 3620c099281SKai Wang bsdar->filename); 36338911b3cSEd Maste return (EXIT_FAILURE); 3640c099281SKai Wang } 3650c099281SKai Wang 3660c099281SKai Wang /* We do not create archive in mode 'd', 'm' and 's'. */ 3670c099281SKai Wang if (mode != 'r' && mode != 'q') { 3680c099281SKai Wang bsdar_warnc(bsdar, 0, "%s: no such file", 3690c099281SKai Wang bsdar->filename); 37038911b3cSEd Maste return (EXIT_FAILURE); 3710c099281SKai Wang } 3720c099281SKai Wang 3730c099281SKai Wang /* Issue a warning if -c is not specified when creating. */ 3740c099281SKai Wang if (!(bsdar->options & AR_C)) 3750c099281SKai Wang bsdar_warnc(bsdar, 0, "creating %s", bsdar->filename); 3760c099281SKai Wang goto new_archive; 3770c099281SKai Wang } 3780c099281SKai Wang 3790c099281SKai Wang /* 3800c099281SKai Wang * First read members from existing archive. 3810c099281SKai Wang */ 3820c099281SKai Wang read_objs(bsdar, bsdar->filename, 0); 383d192f3d3SKai Wang 384d192f3d3SKai Wang /* 385d192f3d3SKai Wang * For mode 's', no member will be moved, deleted or replaced. 386d192f3d3SKai Wang */ 387d192f3d3SKai Wang if (mode == 's') 388d192f3d3SKai Wang goto write_objs; 389d192f3d3SKai Wang 390d192f3d3SKai Wang /* 391cb0dad38SKai Wang * For mode 'q', we don't need to adjust existing members either. 392cb0dad38SKai Wang * Also, -a, -b and -i are ignored in this mode. New members are 393cb0dad38SKai Wang * always inserted at tail. 394cb0dad38SKai Wang */ 395cb0dad38SKai Wang if (mode == 'q') 396cb0dad38SKai Wang goto new_archive; 397cb0dad38SKai Wang 398cb0dad38SKai Wang /* 3990c099281SKai Wang * Mode 'A' adds the contents of another archive to the tail of 4000c099281SKai Wang * current archive. Note that mode 'A' is a special mode for the 4010c099281SKai Wang * ADDLIB command of the ar script mode. Currently there is no 4020c099281SKai Wang * access to this function from the ar command line mode. 4030c099281SKai Wang */ 4040c099281SKai Wang if (mode == 'A') { 4050c099281SKai Wang /* 4060c099281SKai Wang * Read objects from the target archive of ADDLIB command. 4076bef5d28SBenedict Reuschling * If there are members specified in argv, read those members 4080c099281SKai Wang * only, otherwise the entire archive will be read. 4090c099281SKai Wang */ 4100c099281SKai Wang read_objs(bsdar, bsdar->addlib, 1); 4110c099281SKai Wang goto write_objs; 4120c099281SKai Wang } 4130c099281SKai Wang 4140c099281SKai Wang /* 415d192f3d3SKai Wang * Try to find the position member specified by user. 416d192f3d3SKai Wang */ 417d192f3d3SKai Wang if (bsdar->options & AR_A || bsdar->options & AR_B) { 418d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 419d192f3d3SKai Wang if (strcmp(obj->name, bsdar->posarg) == 0) { 420d192f3d3SKai Wang pos = obj; 421d192f3d3SKai Wang break; 422d192f3d3SKai Wang } 423d192f3d3SKai Wang } 424d192f3d3SKai Wang 425d192f3d3SKai Wang /* 426d192f3d3SKai Wang * If can't find `pos' specified by user, 4276bef5d28SBenedict Reuschling * silently insert objects at tail. 428d192f3d3SKai Wang */ 429d192f3d3SKai Wang if (pos == NULL) 430d192f3d3SKai Wang bsdar->options &= ~(AR_A | AR_B); 431d192f3d3SKai Wang } 432d192f3d3SKai Wang 433d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 434d192f3d3SKai Wang av = &bsdar->argv[i]; 435d192f3d3SKai Wang 436d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 437d192f3d3SKai Wang if ((bname = basename(*av)) == NULL) 438086f0908SEd Maste bsdar_errc(bsdar, errno, "basename failed"); 439d192f3d3SKai Wang if (bsdar->options & AR_TR) { 440d192f3d3SKai Wang if (strncmp(bname, obj->name, _TRUNCATE_LEN)) 441d192f3d3SKai Wang continue; 442d192f3d3SKai Wang } else 443d192f3d3SKai Wang if (strcmp(bname, obj->name) != 0) 444d192f3d3SKai Wang continue; 445d192f3d3SKai Wang 446d192f3d3SKai Wang if (mode == 'r') { 447d192f3d3SKai Wang /* 448d192f3d3SKai Wang * if the new member is not qualified 449d192f3d3SKai Wang * to replace the old one, skip it. 450d192f3d3SKai Wang */ 451d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 452d192f3d3SKai Wang obj->mtime); 45338911b3cSEd Maste if (nobj == NULL) { 45438911b3cSEd Maste exitcode = EXIT_FAILURE; 455d192f3d3SKai Wang goto skip_obj; 456d192f3d3SKai Wang } 45738911b3cSEd Maste } 458d192f3d3SKai Wang 459d192f3d3SKai Wang if (bsdar->options & AR_V) 460d192f3d3SKai Wang (void)fprintf(stdout, "%c - %s\n", mode, 461d192f3d3SKai Wang *av); 462d192f3d3SKai Wang 463d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 464cf933f5bSEd Maste if (mode == 'd' || mode == 'r') 465cf933f5bSEd Maste free_obj(bsdar, obj); 466d192f3d3SKai Wang 467d192f3d3SKai Wang if (mode == 'm') 468d192f3d3SKai Wang insert_obj(bsdar, obj, pos); 469d192f3d3SKai Wang if (mode == 'r') 470d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 471d192f3d3SKai Wang 472d192f3d3SKai Wang skip_obj: 473d192f3d3SKai Wang *av = NULL; 474d192f3d3SKai Wang break; 475d192f3d3SKai Wang } 476d192f3d3SKai Wang 477d192f3d3SKai Wang } 478d192f3d3SKai Wang 479d192f3d3SKai Wang new_archive: 480d192f3d3SKai Wang /* 481d192f3d3SKai Wang * When operating in mode 'r', directly add those user specified 482cb0dad38SKai Wang * objects which do not exist in current archive. When operating 483cb0dad38SKai Wang * in mode 'q', all objects specified in command line args are 484cb0dad38SKai Wang * appended to the archive, without comparing with existing ones. 485d192f3d3SKai Wang */ 486d192f3d3SKai Wang for (i = 0; i < bsdar->argc; i++) { 487d192f3d3SKai Wang av = &bsdar->argv[i]; 488cb0dad38SKai Wang if (*av != NULL && (mode == 'r' || mode == 'q')) { 489d192f3d3SKai Wang nobj = create_obj_from_file(bsdar, *av, 0); 49038911b3cSEd Maste if (nobj == NULL) { 49138911b3cSEd Maste exitcode = EXIT_FAILURE; 49238911b3cSEd Maste *av = NULL; 49338911b3cSEd Maste continue; 49438911b3cSEd Maste } 495d192f3d3SKai Wang insert_obj(bsdar, nobj, pos); 496d192f3d3SKai Wang if (bsdar->options & AR_V && nobj != NULL) 497d192f3d3SKai Wang (void)fprintf(stdout, "a - %s\n", *av); 498d192f3d3SKai Wang *av = NULL; 499d192f3d3SKai Wang } 500d192f3d3SKai Wang } 501d192f3d3SKai Wang 502d192f3d3SKai Wang write_objs: 503d192f3d3SKai Wang write_objs(bsdar); 504d192f3d3SKai Wang write_cleanup(bsdar); 50538911b3cSEd Maste 50638911b3cSEd Maste return (exitcode); 507d192f3d3SKai Wang } 508d192f3d3SKai Wang 509d192f3d3SKai Wang /* 510d192f3d3SKai Wang * Memory cleaning up. 511d192f3d3SKai Wang */ 512d192f3d3SKai Wang static void 513d192f3d3SKai Wang write_cleanup(struct bsdar *bsdar) 514d192f3d3SKai Wang { 515d192f3d3SKai Wang struct ar_obj *obj, *obj_temp; 516d192f3d3SKai Wang 517d192f3d3SKai Wang TAILQ_FOREACH_SAFE(obj, &bsdar->v_obj, objs, obj_temp) { 518d192f3d3SKai Wang TAILQ_REMOVE(&bsdar->v_obj, obj, objs); 519cf933f5bSEd Maste free_obj(bsdar, obj); 520d192f3d3SKai Wang } 521d192f3d3SKai Wang 522d192f3d3SKai Wang free(bsdar->as); 523d192f3d3SKai Wang free(bsdar->s_so); 524d192f3d3SKai Wang free(bsdar->s_sn); 525d192f3d3SKai Wang bsdar->as = NULL; 526d192f3d3SKai Wang bsdar->s_so = NULL; 527fabed6b2SEd Maste bsdar->s_so_max = 0; 528d192f3d3SKai Wang bsdar->s_sn = NULL; 529d192f3d3SKai Wang } 530d192f3d3SKai Wang 531d192f3d3SKai Wang /* 53201715d8cSEd Maste * Fault in the buffer prior to writing as a workaround for poor performance 53301715d8cSEd Maste * due to interaction with kernel fs deadlock avoidance code. See the comment 53401715d8cSEd Maste * above vn_io_fault_doio() in sys/kern/vfs_vnops.c for details of the issue. 53501715d8cSEd Maste */ 53601715d8cSEd Maste static void 53701715d8cSEd Maste prefault_buffer(const char *buf, size_t s) 53801715d8cSEd Maste { 53901715d8cSEd Maste volatile const char *p; 54001715d8cSEd Maste size_t page_size; 54101715d8cSEd Maste 54201715d8cSEd Maste if (s == 0) 54301715d8cSEd Maste return; 54401715d8cSEd Maste page_size = sysconf(_SC_PAGESIZE); 54501715d8cSEd Maste for (p = buf; p < buf + s; p += page_size) 54601715d8cSEd Maste *p; 54701715d8cSEd Maste /* 54801715d8cSEd Maste * Ensure we touch the last page as well, in case the buffer is not 54901715d8cSEd Maste * page-aligned. 55001715d8cSEd Maste */ 55101715d8cSEd Maste *(volatile const char *)(buf + s - 1); 55201715d8cSEd Maste } 55301715d8cSEd Maste 55401715d8cSEd Maste /* 555d192f3d3SKai Wang * Wrapper for archive_write_data(). 556d192f3d3SKai Wang */ 557d192f3d3SKai Wang static void 558d192f3d3SKai Wang write_data(struct bsdar *bsdar, struct archive *a, const void *buf, size_t s) 559d192f3d3SKai Wang { 5603f71a0c9SEd Maste ssize_t written; 5613f71a0c9SEd Maste 56201715d8cSEd Maste prefault_buffer(buf, s); 5633f71a0c9SEd Maste while (s > 0) { 5643f71a0c9SEd Maste written = archive_write_data(a, buf, s); 5653f71a0c9SEd Maste if (written < 0) 566086f0908SEd Maste bsdar_errc(bsdar, archive_errno(a), "%s", 567d192f3d3SKai Wang archive_error_string(a)); 5683f71a0c9SEd Maste buf = (const char *)buf + written; 5693f71a0c9SEd Maste s -= written; 5703f71a0c9SEd Maste } 571d192f3d3SKai Wang } 572d192f3d3SKai Wang 573d192f3d3SKai Wang /* 574d192f3d3SKai Wang * Write the resulting archive members. 575d192f3d3SKai Wang */ 576d192f3d3SKai Wang static void 577d192f3d3SKai Wang write_objs(struct bsdar *bsdar) 578d192f3d3SKai Wang { 579d192f3d3SKai Wang struct ar_obj *obj; 580d192f3d3SKai Wang struct archive *a; 581d192f3d3SKai Wang struct archive_entry *entry; 582d192f3d3SKai Wang size_t s_sz; /* size of archive symbol table. */ 583d192f3d3SKai Wang size_t pm_sz; /* size of pseudo members */ 584fabed6b2SEd Maste size_t w_sz; /* size of words in symbol table */ 585597d9b40SEd Maste size_t i; 586fabed6b2SEd Maste uint64_t nr; 5871dffcf9fSEd Maste uint32_t nr32; 588d192f3d3SKai Wang 589d192f3d3SKai Wang if (elf_version(EV_CURRENT) == EV_NONE) 590086f0908SEd Maste bsdar_errc(bsdar, 0, "ELF library initialization failed: %s", 591086f0908SEd Maste elf_errmsg(-1)); 592d192f3d3SKai Wang 593d192f3d3SKai Wang bsdar->rela_off = 0; 594d192f3d3SKai Wang 595d192f3d3SKai Wang /* Create archive symbol table and archive string table, if need. */ 596d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 59747203048SKai Wang if (!(bsdar->options & AR_SS) && obj->maddr != NULL) 598d192f3d3SKai Wang create_symtab_entry(bsdar, obj->maddr, obj->size); 599d192f3d3SKai Wang if (strlen(obj->name) > _MAXNAMELEN_SVR4) 600d192f3d3SKai Wang add_to_ar_str_table(bsdar, obj->name); 601d192f3d3SKai Wang bsdar->rela_off += _ARHDR_LEN + obj->size + obj->size % 2; 602d192f3d3SKai Wang } 603d192f3d3SKai Wang 604d192f3d3SKai Wang /* 605d192f3d3SKai Wang * Pad the symbol name string table. It is treated specially because 606d192f3d3SKai Wang * symbol name table should be padded by a '\0', not the common '\n' 607d192f3d3SKai Wang * for other members. The size of sn table includes the pad bit. 608d192f3d3SKai Wang */ 609d192f3d3SKai Wang if (bsdar->s_cnt != 0 && bsdar->s_sn_sz % 2 != 0) 610d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 611d192f3d3SKai Wang 612d192f3d3SKai Wang /* 613d192f3d3SKai Wang * Archive string table is padded by a "\n" as the normal members. 614517dbe2eSKai Wang * The difference is that the size of archive string table counts 6155666643aSGordon Bergling * in the pad bit, while normal members' size fields do not. 616d192f3d3SKai Wang */ 617d192f3d3SKai Wang if (bsdar->as != NULL && bsdar->as_sz % 2 != 0) 618d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 619d192f3d3SKai Wang 620d192f3d3SKai Wang /* 621d192f3d3SKai Wang * If there is a symbol table, calculate the size of pseudo members, 622517dbe2eSKai Wang * convert previously stored relative offsets to absolute ones, and 623d192f3d3SKai Wang * then make them Big Endian. 624d192f3d3SKai Wang * 625d192f3d3SKai Wang * absolute_offset = htobe32(relative_offset + size_of_pseudo_members) 626d192f3d3SKai Wang */ 627fabed6b2SEd Maste w_sz = sizeof(uint32_t); 628d192f3d3SKai Wang if (bsdar->s_cnt != 0) { 629d192f3d3SKai Wang s_sz = (bsdar->s_cnt + 1) * sizeof(uint32_t) + bsdar->s_sn_sz; 630d192f3d3SKai Wang pm_sz = _ARMAG_LEN + (_ARHDR_LEN + s_sz); 631d192f3d3SKai Wang if (bsdar->as != NULL) 632d192f3d3SKai Wang pm_sz += _ARHDR_LEN + bsdar->as_sz; 633fabed6b2SEd Maste /* Use the 64-bit word size format if necessary. */ 634fabed6b2SEd Maste if (bsdar->s_so_max > UINT32_MAX - pm_sz) { 635fabed6b2SEd Maste w_sz = sizeof(uint64_t); 636fabed6b2SEd Maste pm_sz -= s_sz; 637fabed6b2SEd Maste s_sz = (bsdar->s_cnt + 1) * sizeof(uint64_t) + 638fabed6b2SEd Maste bsdar->s_sn_sz; 639fabed6b2SEd Maste pm_sz += s_sz; 6401dffcf9fSEd Maste /* Convert to big-endian. */ 641597d9b40SEd Maste for (i = 0; i < bsdar->s_cnt; i++) 642687c0904SEd Maste bsdar->s_so[i] = 643687c0904SEd Maste htobe64(bsdar->s_so[i] + pm_sz); 6441dffcf9fSEd Maste } else { 6451dffcf9fSEd Maste /* 6461dffcf9fSEd Maste * Convert to big-endian and shuffle in-place to 6471dffcf9fSEd Maste * the front of the allocation. XXX UB 6481dffcf9fSEd Maste */ 649597d9b40SEd Maste for (i = 0; i < bsdar->s_cnt; i++) 6501dffcf9fSEd Maste ((uint32_t *)(bsdar->s_so))[i] = 6511dffcf9fSEd Maste htobe32(bsdar->s_so[i] + pm_sz); 652d192f3d3SKai Wang } 653699f1801SEd Maste } 654d192f3d3SKai Wang 655d192f3d3SKai Wang if ((a = archive_write_new()) == NULL) 656086f0908SEd Maste bsdar_errc(bsdar, 0, "archive_write_new failed"); 657d192f3d3SKai Wang 658d192f3d3SKai Wang archive_write_set_format_ar_svr4(a); 659d192f3d3SKai Wang 660d192f3d3SKai Wang AC(archive_write_open_filename(a, bsdar->filename)); 661d192f3d3SKai Wang 662d192f3d3SKai Wang /* 663d192f3d3SKai Wang * write the archive symbol table, if there is one. 664d192f3d3SKai Wang * If options -s is explicitly specified or we are invoked 665d192f3d3SKai Wang * as ranlib, write the symbol table even if it is empty. 666d192f3d3SKai Wang */ 667d192f3d3SKai Wang if ((bsdar->s_cnt != 0 && !(bsdar->options & AR_SS)) || 668d192f3d3SKai Wang bsdar->options & AR_S) { 669d192f3d3SKai Wang entry = archive_entry_new(); 670bd9302d8SBryan Drewery if (entry == NULL) 671086f0908SEd Maste bsdar_errc(bsdar, 0, "archive_entry_new failed"); 672fabed6b2SEd Maste if (w_sz == sizeof(uint64_t)) 673fabed6b2SEd Maste archive_entry_copy_pathname(entry, "/SYM64/"); 674fabed6b2SEd Maste else 675d192f3d3SKai Wang archive_entry_copy_pathname(entry, "/"); 6760e479ac8STim Kientzle if ((bsdar->options & AR_D) == 0) 677d192f3d3SKai Wang archive_entry_set_mtime(entry, time(NULL), 0); 678fabed6b2SEd Maste archive_entry_set_size(entry, (bsdar->s_cnt + 1) * w_sz + 679fabed6b2SEd Maste bsdar->s_sn_sz); 680d192f3d3SKai Wang AC(archive_write_header(a, entry)); 6811dffcf9fSEd Maste if (w_sz == sizeof(uint64_t)) { 682fabed6b2SEd Maste nr = htobe64(bsdar->s_cnt); 6831dffcf9fSEd Maste write_data(bsdar, a, &nr, sizeof(nr)); 6841dffcf9fSEd Maste } else { 6851dffcf9fSEd Maste nr32 = htobe32((uint32_t)bsdar->s_cnt); 6861dffcf9fSEd Maste write_data(bsdar, a, &nr32, sizeof(nr32)); 6871dffcf9fSEd Maste } 6881dffcf9fSEd Maste write_data(bsdar, a, bsdar->s_so, w_sz * bsdar->s_cnt); 689d192f3d3SKai Wang write_data(bsdar, a, bsdar->s_sn, bsdar->s_sn_sz); 690d192f3d3SKai Wang archive_entry_free(entry); 691d192f3d3SKai Wang } 692d192f3d3SKai Wang 693d192f3d3SKai Wang /* write the archive string table, if any. */ 694d192f3d3SKai Wang if (bsdar->as != NULL) { 695d192f3d3SKai Wang entry = archive_entry_new(); 696bd9302d8SBryan Drewery if (entry == NULL) 697086f0908SEd Maste bsdar_errc(bsdar, 0, "archive_entry_new failed"); 698d192f3d3SKai Wang archive_entry_copy_pathname(entry, "//"); 699d192f3d3SKai Wang archive_entry_set_size(entry, bsdar->as_sz); 700d192f3d3SKai Wang AC(archive_write_header(a, entry)); 701d192f3d3SKai Wang write_data(bsdar, a, bsdar->as, bsdar->as_sz); 702d192f3d3SKai Wang archive_entry_free(entry); 703d192f3d3SKai Wang } 704d192f3d3SKai Wang 705d192f3d3SKai Wang /* write normal members. */ 706d192f3d3SKai Wang TAILQ_FOREACH(obj, &bsdar->v_obj, objs) { 707d192f3d3SKai Wang entry = archive_entry_new(); 708bd9302d8SBryan Drewery if (entry == NULL) 709086f0908SEd Maste bsdar_errc(bsdar, 0, "archive_entry_new failed"); 710d192f3d3SKai Wang archive_entry_copy_pathname(entry, obj->name); 711d192f3d3SKai Wang archive_entry_set_uid(entry, obj->uid); 712d192f3d3SKai Wang archive_entry_set_gid(entry, obj->gid); 713d192f3d3SKai Wang archive_entry_set_mode(entry, obj->md); 714d192f3d3SKai Wang archive_entry_set_size(entry, obj->size); 715d192f3d3SKai Wang archive_entry_set_mtime(entry, obj->mtime, 0); 716d192f3d3SKai Wang archive_entry_set_dev(entry, obj->dev); 717d192f3d3SKai Wang archive_entry_set_ino(entry, obj->ino); 718d192f3d3SKai Wang archive_entry_set_filetype(entry, AE_IFREG); 719d192f3d3SKai Wang AC(archive_write_header(a, entry)); 720d192f3d3SKai Wang write_data(bsdar, a, obj->maddr, obj->size); 721d192f3d3SKai Wang archive_entry_free(entry); 722d192f3d3SKai Wang } 723d192f3d3SKai Wang 724d192f3d3SKai Wang AC(archive_write_close(a)); 725ebb8fc42SMartin Matuska AC(archive_write_free(a)); 726d192f3d3SKai Wang } 727d192f3d3SKai Wang 728d192f3d3SKai Wang /* 729d192f3d3SKai Wang * Extract global symbols from ELF binary members. 730d192f3d3SKai Wang */ 731d192f3d3SKai Wang static void 732d192f3d3SKai Wang create_symtab_entry(struct bsdar *bsdar, void *maddr, size_t size) 733d192f3d3SKai Wang { 734d192f3d3SKai Wang Elf *e; 735d192f3d3SKai Wang Elf_Scn *scn; 736d192f3d3SKai Wang GElf_Shdr shdr; 737d192f3d3SKai Wang GElf_Sym sym; 738d192f3d3SKai Wang Elf_Data *data; 739d192f3d3SKai Wang char *name; 740d192f3d3SKai Wang size_t n, shstrndx; 741d192f3d3SKai Wang int elferr, tabndx, len, i; 742d192f3d3SKai Wang 743d192f3d3SKai Wang if ((e = elf_memory(maddr, size)) == NULL) { 744d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_memory() failed: %s", 745d192f3d3SKai Wang elf_errmsg(-1)); 746d192f3d3SKai Wang return; 747d192f3d3SKai Wang } 748d192f3d3SKai Wang if (elf_kind(e) != ELF_K_ELF) { 7496bef5d28SBenedict Reuschling /* Silently ignore non-elf member. */ 750d192f3d3SKai Wang elf_end(e); 751d192f3d3SKai Wang return; 752d192f3d3SKai Wang } 753d192f3d3SKai Wang if (elf_getshstrndx(e, &shstrndx) == 0) { 75409319f7dSEd Maste bsdar_warnc(bsdar, 0, "elf_getshstrndx failed: %s", 755d192f3d3SKai Wang elf_errmsg(-1)); 756d192f3d3SKai Wang elf_end(e); 757d192f3d3SKai Wang return; 758d192f3d3SKai Wang } 759d192f3d3SKai Wang 760d192f3d3SKai Wang tabndx = -1; 761d192f3d3SKai Wang scn = NULL; 762d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 763d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 764d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 765d192f3d3SKai Wang "elf_getshdr failed: %s", elf_errmsg(-1)); 766d192f3d3SKai Wang continue; 767d192f3d3SKai Wang } 768d192f3d3SKai Wang if ((name = elf_strptr(e, shstrndx, shdr.sh_name)) == NULL) { 769d192f3d3SKai Wang bsdar_warnc(bsdar, 0, 770d192f3d3SKai Wang "elf_strptr failed: %s", elf_errmsg(-1)); 771d192f3d3SKai Wang continue; 772d192f3d3SKai Wang } 773d192f3d3SKai Wang if (strcmp(name, ".strtab") == 0) { 774d192f3d3SKai Wang tabndx = elf_ndxscn(scn); 775d192f3d3SKai Wang break; 776d192f3d3SKai Wang } 777d192f3d3SKai Wang } 778d192f3d3SKai Wang elferr = elf_errno(); 779d192f3d3SKai Wang if (elferr != 0) 780d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "elf_nextscn failed: %s", 781d192f3d3SKai Wang elf_errmsg(elferr)); 782d192f3d3SKai Wang if (tabndx == -1) { 783d192f3d3SKai Wang bsdar_warnc(bsdar, 0, "can't find .strtab section"); 784d192f3d3SKai Wang elf_end(e); 785d192f3d3SKai Wang return; 786d192f3d3SKai Wang } 787d192f3d3SKai Wang 788d192f3d3SKai Wang scn = NULL; 789d192f3d3SKai Wang while ((scn = elf_nextscn(e, scn)) != NULL) { 790d192f3d3SKai Wang if (gelf_getshdr(scn, &shdr) != &shdr) { 79109319f7dSEd Maste bsdar_warnc(bsdar, 0, "elf_getshdr failed: %s", 79209319f7dSEd Maste elf_errmsg(-1)); 793d192f3d3SKai Wang continue; 794d192f3d3SKai Wang } 795d192f3d3SKai Wang if (shdr.sh_type != SHT_SYMTAB) 796d192f3d3SKai Wang continue; 797d192f3d3SKai Wang 798d192f3d3SKai Wang data = NULL; 799d192f3d3SKai Wang n = 0; 800d192f3d3SKai Wang while (n < shdr.sh_size && 801d192f3d3SKai Wang (data = elf_getdata(scn, data)) != NULL) { 802d192f3d3SKai Wang len = data->d_size / shdr.sh_entsize; 803d192f3d3SKai Wang for (i = 0; i < len; i++) { 804d192f3d3SKai Wang if (gelf_getsym(data, i, &sym) != &sym) { 80509319f7dSEd Maste bsdar_warnc(bsdar, 0, 806d192f3d3SKai Wang "gelf_getsym failed: %s", 807d192f3d3SKai Wang elf_errmsg(-1)); 808d192f3d3SKai Wang continue; 809d192f3d3SKai Wang } 810d192f3d3SKai Wang 811d192f3d3SKai Wang /* keep only global or weak symbols */ 812d192f3d3SKai Wang if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL && 813d192f3d3SKai Wang GELF_ST_BIND(sym.st_info) != STB_WEAK) 814d192f3d3SKai Wang continue; 815d192f3d3SKai Wang 816d192f3d3SKai Wang /* keep only defined symbols */ 817d192f3d3SKai Wang if (sym.st_shndx == SHN_UNDEF) 818d192f3d3SKai Wang continue; 819d192f3d3SKai Wang 820d192f3d3SKai Wang if ((name = elf_strptr(e, tabndx, 821d192f3d3SKai Wang sym.st_name)) == NULL) { 82209319f7dSEd Maste bsdar_warnc(bsdar, 0, 823d192f3d3SKai Wang "elf_strptr failed: %s", 824d192f3d3SKai Wang elf_errmsg(-1)); 825d192f3d3SKai Wang continue; 826d192f3d3SKai Wang } 827d192f3d3SKai Wang 828d192f3d3SKai Wang add_to_ar_sym_table(bsdar, name); 829d192f3d3SKai Wang } 830d192f3d3SKai Wang } 831d192f3d3SKai Wang } 832d192f3d3SKai Wang elferr = elf_errno(); 833d192f3d3SKai Wang if (elferr != 0) 83409319f7dSEd Maste bsdar_warnc(bsdar, 0, "elf_nextscn failed: %s", 835d192f3d3SKai Wang elf_errmsg(elferr)); 836d192f3d3SKai Wang 837d192f3d3SKai Wang elf_end(e); 838d192f3d3SKai Wang } 839d192f3d3SKai Wang 840d192f3d3SKai Wang /* 841d192f3d3SKai Wang * Append to the archive string table buffer. 842d192f3d3SKai Wang */ 843d192f3d3SKai Wang static void 844d192f3d3SKai Wang add_to_ar_str_table(struct bsdar *bsdar, const char *name) 845d192f3d3SKai Wang { 846d192f3d3SKai Wang 847d192f3d3SKai Wang if (bsdar->as == NULL) { 848d192f3d3SKai Wang bsdar->as_cap = _INIT_AS_CAP; 849d192f3d3SKai Wang bsdar->as_sz = 0; 850d192f3d3SKai Wang if ((bsdar->as = malloc(bsdar->as_cap)) == NULL) 851086f0908SEd Maste bsdar_errc(bsdar, errno, "malloc failed"); 852d192f3d3SKai Wang } 853d192f3d3SKai Wang 854d192f3d3SKai Wang /* 855d192f3d3SKai Wang * The space required for holding one member name in as table includes: 856d192f3d3SKai Wang * strlen(name) + (1 for '/') + (1 for '\n') + (possibly 1 for padding). 857d192f3d3SKai Wang */ 858517dbe2eSKai Wang while (bsdar->as_sz + strlen(name) + 3 > bsdar->as_cap) { 859d192f3d3SKai Wang bsdar->as_cap *= 2; 860d192f3d3SKai Wang bsdar->as = realloc(bsdar->as, bsdar->as_cap); 861d192f3d3SKai Wang if (bsdar->as == NULL) 862086f0908SEd Maste bsdar_errc(bsdar, errno, "realloc failed"); 863d192f3d3SKai Wang } 864d192f3d3SKai Wang strncpy(&bsdar->as[bsdar->as_sz], name, strlen(name)); 865d192f3d3SKai Wang bsdar->as_sz += strlen(name); 866d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '/'; 867d192f3d3SKai Wang bsdar->as[bsdar->as_sz++] = '\n'; 868d192f3d3SKai Wang } 869d192f3d3SKai Wang 870d192f3d3SKai Wang /* 871d192f3d3SKai Wang * Append to the archive symbol table buffer. 872d192f3d3SKai Wang */ 873d192f3d3SKai Wang static void 874d192f3d3SKai Wang add_to_ar_sym_table(struct bsdar *bsdar, const char *name) 875d192f3d3SKai Wang { 876d192f3d3SKai Wang 877d192f3d3SKai Wang if (bsdar->s_so == NULL) { 878d192f3d3SKai Wang if ((bsdar->s_so = malloc(_INIT_SYMOFF_CAP)) == 879d192f3d3SKai Wang NULL) 880086f0908SEd Maste bsdar_errc(bsdar, errno, "malloc failed"); 881d192f3d3SKai Wang bsdar->s_so_cap = _INIT_SYMOFF_CAP; 882d192f3d3SKai Wang bsdar->s_cnt = 0; 883d192f3d3SKai Wang } 884d192f3d3SKai Wang 885d192f3d3SKai Wang if (bsdar->s_sn == NULL) { 886d192f3d3SKai Wang if ((bsdar->s_sn = malloc(_INIT_SYMNAME_CAP)) == NULL) 887086f0908SEd Maste bsdar_errc(bsdar, errno, "malloc failed"); 888d192f3d3SKai Wang bsdar->s_sn_cap = _INIT_SYMNAME_CAP; 889d192f3d3SKai Wang bsdar->s_sn_sz = 0; 890d192f3d3SKai Wang } 891d192f3d3SKai Wang 892fabed6b2SEd Maste if (bsdar->s_cnt * sizeof(uint64_t) >= bsdar->s_so_cap) { 893d192f3d3SKai Wang bsdar->s_so_cap *= 2; 894d192f3d3SKai Wang bsdar->s_so = realloc(bsdar->s_so, bsdar->s_so_cap); 895d192f3d3SKai Wang if (bsdar->s_so == NULL) 896086f0908SEd Maste bsdar_errc(bsdar, errno, "realloc failed"); 897d192f3d3SKai Wang } 898d192f3d3SKai Wang bsdar->s_so[bsdar->s_cnt] = bsdar->rela_off; 899fabed6b2SEd Maste if ((uint64_t)bsdar->rela_off > bsdar->s_so_max) 900fabed6b2SEd Maste bsdar->s_so_max = (uint64_t)bsdar->rela_off; 901d192f3d3SKai Wang bsdar->s_cnt++; 902d192f3d3SKai Wang 903d192f3d3SKai Wang /* 904d192f3d3SKai Wang * The space required for holding one symbol name in sn table includes: 905d192f3d3SKai Wang * strlen(name) + (1 for '\n') + (possibly 1 for padding). 906d192f3d3SKai Wang */ 907517dbe2eSKai Wang while (bsdar->s_sn_sz + strlen(name) + 2 > bsdar->s_sn_cap) { 908d192f3d3SKai Wang bsdar->s_sn_cap *= 2; 909d192f3d3SKai Wang bsdar->s_sn = realloc(bsdar->s_sn, bsdar->s_sn_cap); 910d192f3d3SKai Wang if (bsdar->s_sn == NULL) 911086f0908SEd Maste bsdar_errc(bsdar, errno, "realloc failed"); 912d192f3d3SKai Wang } 913d192f3d3SKai Wang strncpy(&bsdar->s_sn[bsdar->s_sn_sz], name, strlen(name)); 914d192f3d3SKai Wang bsdar->s_sn_sz += strlen(name); 915d192f3d3SKai Wang bsdar->s_sn[bsdar->s_sn_sz++] = '\0'; 916d192f3d3SKai Wang } 917