xref: /freebsd/usr.bin/ar/write.c (revision 57aaefdceccdd590f81af4b274c7a3ee61ae62f1)
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