cpio.c revision 311042
11558Srgrimes/*-
250476Speter * Copyright (c) 2003-2007 Tim Kientzle
31558Srgrimes * All rights reserved.
412481Speter *
51558Srgrimes * Redistribution and use in source and binary forms, with or without
641061Sbde * modification, are permitted provided that the following conditions
774448Ssos * are met:
841061Sbde * 1. Redistributions of source code must retain the above copyright
939271Sphk *    notice, this list of conditions and the following disclaimer
1039255Sgibbs *    in this position and unchanged.
1138653Sgpalmer * 2. Redistributions in binary form must reproduce the above copyright
1238653Sgpalmer *    notice, this list of conditions and the following disclaimer in the
1355980Speter *    documentation and/or other materials provided with the distribution.
1443859Sobrien *
1538653Sgpalmer * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
1638653Sgpalmer * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
1738653Sgpalmer * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
1838653Sgpalmer * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
1938653Sgpalmer * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
2069800Stomsoft * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
2138653Sgpalmer * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
2267105Sadrian * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
2366867Sadrian * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
2438653Sgpalmer * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2538653Sgpalmer */
2669800Stomsoft
2738653Sgpalmer
2838653Sgpalmer#include "cpio_platform.h"
2956815Sshin__FBSDID("$FreeBSD: stable/10/contrib/libarchive/cpio/cpio.c 311042 2017-01-02 01:43:11Z mm $");
3053644Sguido
3154225Sguido#include <sys/types.h>
3238653Sgpalmer#include <archive.h>
3354225Sguido#include <archive_entry.h>
3454225Sguido
3538653Sgpalmer#ifdef HAVE_SYS_MKDEV_H
3638653Sgpalmer#include <sys/mkdev.h>
3738653Sgpalmer#endif
3838843Sjb#ifdef HAVE_SYS_STAT_H
3938653Sgpalmer#include <sys/stat.h>
4070450Sphk#endif
4178448Sdd#ifdef HAVE_SYS_TIME_H
4238653Sgpalmer#include <sys/time.h>
4338653Sgpalmer#endif
4438653Sgpalmer#ifdef HAVE_ERRNO_H
4538653Sgpalmer#include <errno.h>
4667105Sadrian#endif
4777577Sru#ifdef HAVE_FCNTL_H
4838653Sgpalmer#include <fcntl.h>
4943557Ssemenu#endif
5077042Sru#ifdef HAVE_GRP_H
5177042Sru#include <grp.h>
5238653Sgpalmer#endif
5377042Sru#ifdef HAVE_LOCALE_H
5477042Sru#include <locale.h>
5538653Sgpalmer#endif
5644690Sbrian#ifdef HAVE_PWD_H
5738653Sgpalmer#include <pwd.h>
5838653Sgpalmer#endif
5938653Sgpalmer#ifdef HAVE_SIGNAL_H
6038653Sgpalmer#include <signal.h>
6138653Sgpalmer#endif
6238653Sgpalmer#ifdef HAVE_STDARG_H
6338653Sgpalmer#include <stdarg.h>
6455163Sshin#endif
6538653Sgpalmer#ifdef HAVE_STDINT_H
6638653Sgpalmer#include <stdint.h>
6738653Sgpalmer#endif
6838653Sgpalmer#include <stdio.h>
6938653Sgpalmer#ifdef HAVE_STDLIB_H
7055163Sshin#include <stdlib.h>
7138653Sgpalmer#endif
7238653Sgpalmer#ifdef HAVE_STRING_H
7338653Sgpalmer#include <string.h>
7441061Sbde#endif
7538653Sgpalmer#ifdef HAVE_UNISTD_H
7646878Sobrien#include <unistd.h>
7738653Sgpalmer#endif
7838653Sgpalmer#ifdef HAVE_TIME_H
7942117Ssos#include <time.h>
8042117Ssos#endif
8110855Sjoerg
8244317Sjkh#include "cpio.h"
8352252Sbp#include "err.h"
8444317Sjkh#include "line_reader.h"
8544317Sjkh#include "passphrase.h"
8658235Skato
8758235Skato/* Fixed size of uname/gname caches. */
8838458Sjb#define	name_cache_size 101
8938458Sjb
901558Srgrimes#ifndef O_BINARY
91#define O_BINARY 0
92#endif
93
94struct name_cache {
95	int	probes;
96	int	hits;
97	size_t	size;
98	struct {
99		id_t id;
100		char *name;
101	} cache[name_cache_size];
102};
103
104static int	extract_data(struct archive *, struct archive *);
105const char *	cpio_i64toa(int64_t);
106static const char *cpio_rename(const char *name);
107static int	entry_to_archive(struct cpio *, struct archive_entry *);
108static int	file_to_archive(struct cpio *, const char *);
109static void	free_cache(struct name_cache *cache);
110static void	list_item_verbose(struct cpio *, struct archive_entry *);
111static void	long_help(void);
112static const char *lookup_gname(struct cpio *, gid_t gid);
113static int	lookup_gname_helper(struct cpio *,
114		    const char **name, id_t gid);
115static const char *lookup_uname(struct cpio *, uid_t uid);
116static int	lookup_uname_helper(struct cpio *,
117		    const char **name, id_t uid);
118static void	mode_in(struct cpio *);
119static void	mode_list(struct cpio *);
120static void	mode_out(struct cpio *);
121static void	mode_pass(struct cpio *, const char *);
122static const char *remove_leading_slash(const char *);
123static int	restore_time(struct cpio *, struct archive_entry *,
124		    const char *, int fd);
125static void	usage(void);
126static void	version(void);
127static const char * passphrase_callback(struct archive *, void *);
128static void	passphrase_free(char *);
129
130int
131main(int argc, char *argv[])
132{
133	static char buff[16384];
134	struct cpio _cpio; /* Allocated on stack. */
135	struct cpio *cpio;
136	const char *errmsg;
137	int uid, gid;
138	int opt;
139
140	cpio = &_cpio;
141	memset(cpio, 0, sizeof(*cpio));
142	cpio->buff = buff;
143	cpio->buff_size = sizeof(buff);
144
145#if defined(HAVE_SIGACTION) && defined(SIGPIPE)
146	{ /* Ignore SIGPIPE signals. */
147		struct sigaction sa;
148		sigemptyset(&sa.sa_mask);
149		sa.sa_flags = 0;
150		sa.sa_handler = SIG_IGN;
151		sigaction(SIGPIPE, &sa, NULL);
152	}
153#endif
154
155	/* Set lafe_progname before calling lafe_warnc. */
156	lafe_setprogname(*argv, "bsdcpio");
157
158#if HAVE_SETLOCALE
159	if (setlocale(LC_ALL, "") == NULL)
160		lafe_warnc(0, "Failed to set default locale");
161#endif
162
163	cpio->uid_override = -1;
164	cpio->gid_override = -1;
165	cpio->argv = argv;
166	cpio->argc = argc;
167	cpio->mode = '\0';
168	cpio->verbose = 0;
169	cpio->compress = '\0';
170	cpio->extract_flags = ARCHIVE_EXTRACT_NO_AUTODIR;
171	cpio->extract_flags |= ARCHIVE_EXTRACT_NO_OVERWRITE_NEWER;
172	cpio->extract_flags |= ARCHIVE_EXTRACT_SECURE_SYMLINKS;
173	cpio->extract_flags |= ARCHIVE_EXTRACT_SECURE_NODOTDOT;
174	cpio->extract_flags |= ARCHIVE_EXTRACT_SECURE_NOABSOLUTEPATHS;
175	cpio->extract_flags |= ARCHIVE_EXTRACT_PERM;
176	cpio->extract_flags |= ARCHIVE_EXTRACT_FFLAGS;
177	cpio->extract_flags |= ARCHIVE_EXTRACT_ACL;
178#if !defined(_WIN32) && !defined(__CYGWIN__)
179	if (geteuid() == 0)
180		cpio->extract_flags |= ARCHIVE_EXTRACT_OWNER;
181#endif
182	cpio->bytes_per_block = 512;
183	cpio->filename = NULL;
184
185	cpio->matching = archive_match_new();
186	if (cpio->matching == NULL)
187		lafe_errc(1, 0, "Out of memory");
188
189	while ((opt = cpio_getopt(cpio)) != -1) {
190		switch (opt) {
191		case '0': /* GNU convention: --null, -0 */
192			cpio->option_null = 1;
193			break;
194		case 'A': /* NetBSD/OpenBSD */
195			cpio->option_append = 1;
196			break;
197		case 'a': /* POSIX 1997 */
198			cpio->option_atime_restore = 1;
199			break;
200		case 'B': /* POSIX 1997 */
201			cpio->bytes_per_block = 5120;
202			break;
203		case OPTION_B64ENCODE:
204			cpio->add_filter = opt;
205			break;
206		case 'C': /* NetBSD/OpenBSD */
207			cpio->bytes_per_block = atoi(cpio->argument);
208			if (cpio->bytes_per_block <= 0)
209				lafe_errc(1, 0, "Invalid blocksize %s", cpio->argument);
210			break;
211		case 'c': /* POSIX 1997 */
212			cpio->format = "odc";
213			break;
214		case 'd': /* POSIX 1997 */
215			cpio->extract_flags &= ~ARCHIVE_EXTRACT_NO_AUTODIR;
216			break;
217		case 'E': /* NetBSD/OpenBSD */
218			if (archive_match_include_pattern_from_file(
219			    cpio->matching, cpio->argument,
220			    cpio->option_null) != ARCHIVE_OK)
221				lafe_errc(1, 0, "Error : %s",
222				    archive_error_string(cpio->matching));
223			break;
224		case 'F': /* NetBSD/OpenBSD/GNU cpio */
225			cpio->filename = cpio->argument;
226			break;
227		case 'f': /* POSIX 1997 */
228			if (archive_match_exclude_pattern(cpio->matching,
229			    cpio->argument) != ARCHIVE_OK)
230				lafe_errc(1, 0, "Error : %s",
231				    archive_error_string(cpio->matching));
232			break;
233		case OPTION_GRZIP:
234			cpio->compress = opt;
235			break;
236		case 'H': /* GNU cpio (also --format) */
237			cpio->format = cpio->argument;
238			break;
239		case 'h':
240			long_help();
241			break;
242		case 'I': /* NetBSD/OpenBSD */
243			cpio->filename = cpio->argument;
244			break;
245		case 'i': /* POSIX 1997 */
246			if (cpio->mode != '\0')
247				lafe_errc(1, 0,
248				    "Cannot use both -i and -%c", cpio->mode);
249			cpio->mode = opt;
250			break;
251		case 'J': /* GNU tar, others */
252			cpio->compress = opt;
253			break;
254		case 'j': /* GNU tar, others */
255			cpio->compress = opt;
256			break;
257		case OPTION_INSECURE:
258			cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_SYMLINKS;
259			cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_NODOTDOT;
260			cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_NOABSOLUTEPATHS;
261			break;
262		case 'L': /* GNU cpio */
263			cpio->option_follow_links = 1;
264			break;
265		case 'l': /* POSIX 1997 */
266			cpio->option_link = 1;
267			break;
268		case OPTION_LRZIP:
269		case OPTION_LZ4:
270		case OPTION_LZMA: /* GNU tar, others */
271		case OPTION_LZOP: /* GNU tar, others */
272			cpio->compress = opt;
273			break;
274		case 'm': /* POSIX 1997 */
275			cpio->extract_flags |= ARCHIVE_EXTRACT_TIME;
276			break;
277		case 'n': /* GNU cpio */
278			cpio->option_numeric_uid_gid = 1;
279			break;
280		case OPTION_NO_PRESERVE_OWNER: /* GNU cpio */
281			cpio->extract_flags &= ~ARCHIVE_EXTRACT_OWNER;
282			break;
283		case 'O': /* GNU cpio */
284			cpio->filename = cpio->argument;
285			break;
286		case 'o': /* POSIX 1997 */
287			if (cpio->mode != '\0')
288				lafe_errc(1, 0,
289				    "Cannot use both -o and -%c", cpio->mode);
290			cpio->mode = opt;
291			break;
292		case 'p': /* POSIX 1997 */
293			if (cpio->mode != '\0')
294				lafe_errc(1, 0,
295				    "Cannot use both -p and -%c", cpio->mode);
296			cpio->mode = opt;
297			cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_NODOTDOT;
298			cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_NOABSOLUTEPATHS;
299			break;
300		case OPTION_PASSPHRASE:
301			cpio->passphrase = cpio->argument;
302			break;
303		case OPTION_PRESERVE_OWNER:
304			cpio->extract_flags |= ARCHIVE_EXTRACT_OWNER;
305			break;
306		case OPTION_QUIET: /* GNU cpio */
307			cpio->quiet = 1;
308			break;
309		case 'R': /* GNU cpio, also --owner */
310			/* TODO: owner_parse should return uname/gname
311			 * also; use that to set [ug]name_override. */
312			errmsg = owner_parse(cpio->argument, &uid, &gid);
313			if (errmsg) {
314				lafe_warnc(-1, "%s", errmsg);
315				usage();
316			}
317			if (uid != -1) {
318				cpio->uid_override = uid;
319				cpio->uname_override = NULL;
320			}
321			if (gid != -1) {
322				cpio->gid_override = gid;
323				cpio->gname_override = NULL;
324			}
325			break;
326		case 'r': /* POSIX 1997 */
327			cpio->option_rename = 1;
328			break;
329		case 't': /* POSIX 1997 */
330			cpio->option_list = 1;
331			break;
332		case 'u': /* POSIX 1997 */
333			cpio->extract_flags
334			    &= ~ARCHIVE_EXTRACT_NO_OVERWRITE_NEWER;
335			break;
336		case OPTION_UUENCODE:
337			cpio->add_filter = opt;
338			break;
339		case 'v': /* POSIX 1997 */
340			cpio->verbose++;
341			break;
342		case 'V': /* GNU cpio */
343			cpio->dot++;
344			break;
345		case OPTION_VERSION: /* GNU convention */
346			version();
347			break;
348#if 0
349	        /*
350		 * cpio_getopt() handles -W specially, so it's not
351		 * available here.
352		 */
353		case 'W': /* Obscure, but useful GNU convention. */
354			break;
355#endif
356		case 'y': /* tar convention */
357			cpio->compress = opt;
358			break;
359		case 'Z': /* tar convention */
360			cpio->compress = opt;
361			break;
362		case 'z': /* tar convention */
363			cpio->compress = opt;
364			break;
365		default:
366			usage();
367		}
368	}
369
370	/*
371	 * Sanity-check args, error out on nonsensical combinations.
372	 */
373	/* -t implies -i if no mode was specified. */
374	if (cpio->option_list && cpio->mode == '\0')
375		cpio->mode = 'i';
376	/* -t requires -i */
377	if (cpio->option_list && cpio->mode != 'i')
378		lafe_errc(1, 0, "Option -t requires -i");
379	/* -n requires -it */
380	if (cpio->option_numeric_uid_gid && !cpio->option_list)
381		lafe_errc(1, 0, "Option -n requires -it");
382	/* Can only specify format when writing */
383	if (cpio->format != NULL && cpio->mode != 'o')
384		lafe_errc(1, 0, "Option --format requires -o");
385	/* -l requires -p */
386	if (cpio->option_link && cpio->mode != 'p')
387		lafe_errc(1, 0, "Option -l requires -p");
388	/* -v overrides -V */
389	if (cpio->dot && cpio->verbose)
390		cpio->dot = 0;
391	/* TODO: Flag other nonsensical combinations. */
392
393	switch (cpio->mode) {
394	case 'o':
395		/* TODO: Implement old binary format in libarchive,
396		   use that here. */
397		if (cpio->format == NULL)
398			cpio->format = "odc"; /* Default format */
399
400		mode_out(cpio);
401		break;
402	case 'i':
403		while (*cpio->argv != NULL) {
404			if (archive_match_include_pattern(cpio->matching,
405			    *cpio->argv) != ARCHIVE_OK)
406				lafe_errc(1, 0, "Error : %s",
407				    archive_error_string(cpio->matching));
408			--cpio->argc;
409			++cpio->argv;
410		}
411		if (cpio->option_list)
412			mode_list(cpio);
413		else
414			mode_in(cpio);
415		break;
416	case 'p':
417		if (*cpio->argv == NULL || **cpio->argv == '\0')
418			lafe_errc(1, 0,
419			    "-p mode requires a target directory");
420		mode_pass(cpio, *cpio->argv);
421		break;
422	default:
423		lafe_errc(1, 0,
424		    "Must specify at least one of -i, -o, or -p");
425	}
426
427	archive_match_free(cpio->matching);
428	free_cache(cpio->gname_cache);
429	free_cache(cpio->uname_cache);
430	free(cpio->destdir);
431	passphrase_free(cpio->ppbuff);
432	return (cpio->return_value);
433}
434
435static void
436usage(void)
437{
438	const char	*p;
439
440	p = lafe_getprogname();
441
442	fprintf(stderr, "Brief Usage:\n");
443	fprintf(stderr, "  List:    %s -it < archive\n", p);
444	fprintf(stderr, "  Extract: %s -i < archive\n", p);
445	fprintf(stderr, "  Create:  %s -o < filenames > archive\n", p);
446	fprintf(stderr, "  Help:    %s --help\n", p);
447	exit(1);
448}
449
450static const char *long_help_msg =
451	"First option must be a mode specifier:\n"
452	"  -i Input  -o Output  -p Pass\n"
453	"Common Options:\n"
454	"  -v Verbose filenames     -V  one dot per file\n"
455	"Create: %p -o [options]  < [list of files] > [archive]\n"
456	"  -J,-y,-z,--lzma  Compress archive with xz/bzip2/gzip/lzma\n"
457	"  --format {odc|newc|ustar}  Select archive format\n"
458	"List: %p -it < [archive]\n"
459	"Extract: %p -i [options] < [archive]\n";
460
461
462/*
463 * Note that the word 'bsdcpio' will always appear in the first line
464 * of output.
465 *
466 * In particular, /bin/sh scripts that need to test for the presence
467 * of bsdcpio can use the following template:
468 *
469 * if (cpio --help 2>&1 | grep bsdcpio >/dev/null 2>&1 ) then \
470 *          echo bsdcpio; else echo not bsdcpio; fi
471 */
472static void
473long_help(void)
474{
475	const char	*prog;
476	const char	*p;
477
478	prog = lafe_getprogname();
479
480	fflush(stderr);
481
482	p = (strcmp(prog,"bsdcpio") != 0) ? "(bsdcpio)" : "";
483	printf("%s%s: manipulate archive files\n", prog, p);
484
485	for (p = long_help_msg; *p != '\0'; p++) {
486		if (*p == '%') {
487			if (p[1] == 'p') {
488				fputs(prog, stdout);
489				p++;
490			} else
491				putchar('%');
492		} else
493			putchar(*p);
494	}
495	version();
496}
497
498static void
499version(void)
500{
501	fprintf(stdout,"bsdcpio %s - %s\n",
502	    BSDCPIO_VERSION_STRING,
503	    archive_version_details());
504	exit(0);
505}
506
507static void
508mode_out(struct cpio *cpio)
509{
510	struct archive_entry *entry, *spare;
511	struct lafe_line_reader *lr;
512	const char *p;
513	int r;
514
515	if (cpio->option_append)
516		lafe_errc(1, 0, "Append mode not yet supported.");
517
518	cpio->archive_read_disk = archive_read_disk_new();
519	if (cpio->archive_read_disk == NULL)
520		lafe_errc(1, 0, "Failed to allocate archive object");
521	if (cpio->option_follow_links)
522		archive_read_disk_set_symlink_logical(cpio->archive_read_disk);
523	else
524		archive_read_disk_set_symlink_physical(cpio->archive_read_disk);
525	archive_read_disk_set_standard_lookup(cpio->archive_read_disk);
526
527	cpio->archive = archive_write_new();
528	if (cpio->archive == NULL)
529		lafe_errc(1, 0, "Failed to allocate archive object");
530	switch (cpio->compress) {
531	case OPTION_GRZIP:
532		r = archive_write_add_filter_grzip(cpio->archive);
533		break;
534	case 'J':
535		r = archive_write_add_filter_xz(cpio->archive);
536		break;
537	case OPTION_LRZIP:
538		r = archive_write_add_filter_lrzip(cpio->archive);
539		break;
540	case OPTION_LZ4:
541		r = archive_write_add_filter_lz4(cpio->archive);
542		break;
543	case OPTION_LZMA:
544		r = archive_write_add_filter_lzma(cpio->archive);
545		break;
546	case OPTION_LZOP:
547		r = archive_write_add_filter_lzop(cpio->archive);
548		break;
549	case 'j': case 'y':
550		r = archive_write_add_filter_bzip2(cpio->archive);
551		break;
552	case 'z':
553		r = archive_write_add_filter_gzip(cpio->archive);
554		break;
555	case 'Z':
556		r = archive_write_add_filter_compress(cpio->archive);
557		break;
558	default:
559		r = archive_write_add_filter_none(cpio->archive);
560		break;
561	}
562	if (r < ARCHIVE_WARN)
563		lafe_errc(1, 0, "Requested compression not available");
564	switch (cpio->add_filter) {
565	case 0:
566		r = ARCHIVE_OK;
567		break;
568	case OPTION_B64ENCODE:
569		r = archive_write_add_filter_b64encode(cpio->archive);
570		break;
571	case OPTION_UUENCODE:
572		r = archive_write_add_filter_uuencode(cpio->archive);
573		break;
574	}
575	if (r < ARCHIVE_WARN)
576		lafe_errc(1, 0, "Requested filter not available");
577	r = archive_write_set_format_by_name(cpio->archive, cpio->format);
578	if (r != ARCHIVE_OK)
579		lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
580	archive_write_set_bytes_per_block(cpio->archive, cpio->bytes_per_block);
581	cpio->linkresolver = archive_entry_linkresolver_new();
582	archive_entry_linkresolver_set_strategy(cpio->linkresolver,
583	    archive_format(cpio->archive));
584	if (cpio->passphrase != NULL)
585		r = archive_write_set_passphrase(cpio->archive,
586			cpio->passphrase);
587	else
588		r = archive_write_set_passphrase_callback(cpio->archive, cpio,
589			&passphrase_callback);
590	if (r != ARCHIVE_OK)
591		lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
592
593	/*
594	 * The main loop:  Copy each file into the output archive.
595	 */
596	r = archive_write_open_filename(cpio->archive, cpio->filename);
597	if (r != ARCHIVE_OK)
598		lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
599	lr = lafe_line_reader("-", cpio->option_null);
600	while ((p = lafe_line_reader_next(lr)) != NULL)
601		file_to_archive(cpio, p);
602	lafe_line_reader_free(lr);
603
604	/*
605	 * The hardlink detection may have queued up a couple of entries
606	 * that can now be flushed.
607	 */
608	entry = NULL;
609	archive_entry_linkify(cpio->linkresolver, &entry, &spare);
610	while (entry != NULL) {
611		entry_to_archive(cpio, entry);
612		archive_entry_free(entry);
613		entry = NULL;
614		archive_entry_linkify(cpio->linkresolver, &entry, &spare);
615	}
616
617	r = archive_write_close(cpio->archive);
618	if (cpio->dot)
619		fprintf(stderr, "\n");
620	if (r != ARCHIVE_OK)
621		lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
622
623	if (!cpio->quiet) {
624		int64_t blocks =
625			(archive_filter_bytes(cpio->archive, 0) + 511)
626			/ 512;
627		fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
628		    blocks == 1 ? "block" : "blocks");
629	}
630	archive_write_free(cpio->archive);
631}
632
633static const char *
634remove_leading_slash(const char *p)
635{
636	const char *rp;
637
638	/* Remove leading "//./" or "//?/" or "//?/UNC/"
639	 * (absolute path prefixes used by Windows API) */
640	if ((p[0] == '/' || p[0] == '\\') &&
641	    (p[1] == '/' || p[1] == '\\') &&
642	    (p[2] == '.' || p[2] == '?') &&
643	    (p[3] == '/' || p[3] == '\\'))
644	{
645		if (p[2] == '?' &&
646		    (p[4] == 'U' || p[4] == 'u') &&
647		    (p[5] == 'N' || p[5] == 'n') &&
648		    (p[6] == 'C' || p[6] == 'c') &&
649		    (p[7] == '/' || p[7] == '\\'))
650			p += 8;
651		else
652			p += 4;
653	}
654	do {
655		rp = p;
656		/* Remove leading drive letter from archives created
657		 * on Windows. */
658		if (((p[0] >= 'a' && p[0] <= 'z') ||
659		     (p[0] >= 'A' && p[0] <= 'Z')) &&
660			 p[1] == ':') {
661			p += 2;
662		}
663		/* Remove leading "/../", "//", etc. */
664		while (p[0] == '/' || p[0] == '\\') {
665			if (p[1] == '.' && p[2] == '.' &&
666				(p[3] == '/' || p[3] == '\\')) {
667				p += 3; /* Remove "/..", leave "/"
668					 * for next pass. */
669			} else
670				p += 1; /* Remove "/". */
671		}
672	} while (rp != p);
673	return (p);
674}
675
676/*
677 * This is used by both out mode (to copy objects from disk into
678 * an archive) and pass mode (to copy objects from disk to
679 * an archive_write_disk "archive").
680 */
681static int
682file_to_archive(struct cpio *cpio, const char *srcpath)
683{
684	const char *destpath;
685	struct archive_entry *entry, *spare;
686	size_t len;
687	int r;
688
689	/*
690	 * Create an archive_entry describing the source file.
691	 *
692	 */
693	entry = archive_entry_new();
694	if (entry == NULL)
695		lafe_errc(1, 0, "Couldn't allocate entry");
696	archive_entry_copy_sourcepath(entry, srcpath);
697	r = archive_read_disk_entry_from_file(cpio->archive_read_disk,
698	    entry, -1, NULL);
699	if (r < ARCHIVE_FAILED)
700		lafe_errc(1, 0, "%s",
701		    archive_error_string(cpio->archive_read_disk));
702	if (r < ARCHIVE_OK)
703		lafe_warnc(0, "%s",
704		    archive_error_string(cpio->archive_read_disk));
705	if (r <= ARCHIVE_FAILED) {
706		cpio->return_value = 1;
707		return (r);
708	}
709
710	if (cpio->uid_override >= 0) {
711		archive_entry_set_uid(entry, cpio->uid_override);
712		archive_entry_set_uname(entry, cpio->uname_override);
713	}
714	if (cpio->gid_override >= 0) {
715		archive_entry_set_gid(entry, cpio->gid_override);
716		archive_entry_set_gname(entry, cpio->gname_override);
717	}
718
719	/*
720	 * Generate a destination path for this entry.
721	 * "destination path" is the name to which it will be copied in
722	 * pass mode or the name that will go into the archive in
723	 * output mode.
724	 */
725	destpath = srcpath;
726	if (cpio->destdir) {
727		len = strlen(cpio->destdir) + strlen(srcpath) + 8;
728		if (len >= cpio->pass_destpath_alloc) {
729			while (len >= cpio->pass_destpath_alloc) {
730				cpio->pass_destpath_alloc += 512;
731				cpio->pass_destpath_alloc *= 2;
732			}
733			free(cpio->pass_destpath);
734			cpio->pass_destpath = malloc(cpio->pass_destpath_alloc);
735			if (cpio->pass_destpath == NULL)
736				lafe_errc(1, ENOMEM,
737				    "Can't allocate path buffer");
738		}
739		strcpy(cpio->pass_destpath, cpio->destdir);
740		strcat(cpio->pass_destpath, remove_leading_slash(srcpath));
741		destpath = cpio->pass_destpath;
742	}
743	if (cpio->option_rename)
744		destpath = cpio_rename(destpath);
745	if (destpath == NULL)
746		return (0);
747	archive_entry_copy_pathname(entry, destpath);
748
749	/*
750	 * If we're trying to preserve hardlinks, match them here.
751	 */
752	spare = NULL;
753	if (cpio->linkresolver != NULL
754	    && archive_entry_filetype(entry) != AE_IFDIR) {
755		archive_entry_linkify(cpio->linkresolver, &entry, &spare);
756	}
757
758	if (entry != NULL) {
759		r = entry_to_archive(cpio, entry);
760		archive_entry_free(entry);
761		if (spare != NULL) {
762			if (r == 0)
763				r = entry_to_archive(cpio, spare);
764			archive_entry_free(spare);
765		}
766	}
767	return (r);
768}
769
770static int
771entry_to_archive(struct cpio *cpio, struct archive_entry *entry)
772{
773	const char *destpath = archive_entry_pathname(entry);
774	const char *srcpath = archive_entry_sourcepath(entry);
775	int fd = -1;
776	ssize_t bytes_read;
777	int r;
778
779	/* Print out the destination name to the user. */
780	if (cpio->verbose)
781		fprintf(stderr,"%s", destpath);
782	if (cpio->dot)
783		fprintf(stderr, ".");
784
785	/*
786	 * Option_link only makes sense in pass mode and for
787	 * regular files.  Also note: if a link operation fails
788	 * because of cross-device restrictions, we'll fall back
789	 * to copy mode for that entry.
790	 *
791	 * TODO: Test other cpio implementations to see if they
792	 * hard-link anything other than regular files here.
793	 */
794	if (cpio->option_link
795	    && archive_entry_filetype(entry) == AE_IFREG)
796	{
797		struct archive_entry *t;
798		/* Save the original entry in case we need it later. */
799		t = archive_entry_clone(entry);
800		if (t == NULL)
801			lafe_errc(1, ENOMEM, "Can't create link");
802		/* Note: link(2) doesn't create parent directories,
803		 * so we use archive_write_header() instead as a
804		 * convenience. */
805		archive_entry_set_hardlink(t, srcpath);
806		/* This is a straight link that carries no data. */
807		archive_entry_set_size(t, 0);
808		r = archive_write_header(cpio->archive, t);
809		archive_entry_free(t);
810		if (r != ARCHIVE_OK)
811			lafe_warnc(archive_errno(cpio->archive),
812			    "%s", archive_error_string(cpio->archive));
813		if (r == ARCHIVE_FATAL)
814			exit(1);
815#ifdef EXDEV
816		if (r != ARCHIVE_OK && archive_errno(cpio->archive) == EXDEV) {
817			/* Cross-device link:  Just fall through and use
818			 * the original entry to copy the file over. */
819			lafe_warnc(0, "Copying file instead");
820		} else
821#endif
822		return (0);
823	}
824
825	/*
826	 * Make sure we can open the file (if necessary) before
827	 * trying to write the header.
828	 */
829	if (archive_entry_filetype(entry) == AE_IFREG) {
830		if (archive_entry_size(entry) > 0) {
831			fd = open(srcpath, O_RDONLY | O_BINARY);
832			if (fd < 0) {
833				lafe_warnc(errno,
834				    "%s: could not open file", srcpath);
835				goto cleanup;
836			}
837		}
838	} else {
839		archive_entry_set_size(entry, 0);
840	}
841
842	r = archive_write_header(cpio->archive, entry);
843
844	if (r != ARCHIVE_OK)
845		lafe_warnc(archive_errno(cpio->archive),
846		    "%s: %s",
847		    srcpath,
848		    archive_error_string(cpio->archive));
849
850	if (r == ARCHIVE_FATAL)
851		exit(1);
852
853	if (r >= ARCHIVE_WARN && archive_entry_size(entry) > 0 && fd >= 0) {
854		bytes_read = read(fd, cpio->buff, (unsigned)cpio->buff_size);
855		while (bytes_read > 0) {
856			ssize_t bytes_write;
857			bytes_write = archive_write_data(cpio->archive,
858			    cpio->buff, bytes_read);
859			if (bytes_write < 0)
860				lafe_errc(1, archive_errno(cpio->archive),
861				    "%s", archive_error_string(cpio->archive));
862			if (bytes_write < bytes_read) {
863				lafe_warnc(0,
864				    "Truncated write; file may have "
865				    "grown while being archived.");
866			}
867			bytes_read = read(fd, cpio->buff,
868			    (unsigned)cpio->buff_size);
869		}
870	}
871
872	fd = restore_time(cpio, entry, srcpath, fd);
873
874cleanup:
875	if (cpio->verbose)
876		fprintf(stderr,"\n");
877	if (fd >= 0)
878		close(fd);
879	return (0);
880}
881
882static int
883restore_time(struct cpio *cpio, struct archive_entry *entry,
884    const char *name, int fd)
885{
886#ifndef HAVE_UTIMES
887	static int warned = 0;
888
889	(void)cpio; /* UNUSED */
890	(void)entry; /* UNUSED */
891	(void)name; /* UNUSED */
892
893	if (!warned)
894		lafe_warnc(0, "Can't restore access times on this platform");
895	warned = 1;
896	return (fd);
897#else
898#if defined(_WIN32) && !defined(__CYGWIN__)
899	struct __timeval times[2];
900#else
901	struct timeval times[2];
902#endif
903
904	if (!cpio->option_atime_restore)
905		return (fd);
906
907        times[1].tv_sec = archive_entry_mtime(entry);
908        times[1].tv_usec = archive_entry_mtime_nsec(entry) / 1000;
909
910        times[0].tv_sec = archive_entry_atime(entry);
911        times[0].tv_usec = archive_entry_atime_nsec(entry) / 1000;
912
913#if defined(HAVE_FUTIMES) && !defined(__CYGWIN__)
914        if (fd >= 0 && futimes(fd, times) == 0)
915		return (fd);
916#endif
917	/*
918	 * Some platform cannot restore access times if the file descriptor
919	 * is still opened.
920	 */
921	if (fd >= 0) {
922		close(fd);
923		fd = -1;
924	}
925
926#ifdef HAVE_LUTIMES
927        if (lutimes(name, times) != 0)
928#else
929        if ((AE_IFLNK != archive_entry_filetype(entry))
930			&& utimes(name, times) != 0)
931#endif
932                lafe_warnc(errno, "Can't update time for %s", name);
933#endif
934	return (fd);
935}
936
937
938static void
939mode_in(struct cpio *cpio)
940{
941	struct archive *a;
942	struct archive_entry *entry;
943	struct archive *ext;
944	const char *destpath;
945	int r;
946
947	ext = archive_write_disk_new();
948	if (ext == NULL)
949		lafe_errc(1, 0, "Couldn't allocate restore object");
950	r = archive_write_disk_set_options(ext, cpio->extract_flags);
951	if (r != ARCHIVE_OK)
952		lafe_errc(1, 0, "%s", archive_error_string(ext));
953	a = archive_read_new();
954	if (a == NULL)
955		lafe_errc(1, 0, "Couldn't allocate archive object");
956	archive_read_support_filter_all(a);
957	archive_read_support_format_all(a);
958	if (cpio->passphrase != NULL)
959		r = archive_read_add_passphrase(a, cpio->passphrase);
960	else
961		r = archive_read_set_passphrase_callback(a, cpio,
962			&passphrase_callback);
963	if (r != ARCHIVE_OK)
964		lafe_errc(1, 0, "%s", archive_error_string(a));
965
966	if (archive_read_open_filename(a, cpio->filename,
967					cpio->bytes_per_block))
968		lafe_errc(1, archive_errno(a),
969		    "%s", archive_error_string(a));
970	for (;;) {
971		r = archive_read_next_header(a, &entry);
972		if (r == ARCHIVE_EOF)
973			break;
974		if (r != ARCHIVE_OK) {
975			lafe_errc(1, archive_errno(a),
976			    "%s", archive_error_string(a));
977		}
978		if (archive_match_path_excluded(cpio->matching, entry))
979			continue;
980		if (cpio->option_rename) {
981			destpath = cpio_rename(archive_entry_pathname(entry));
982			archive_entry_set_pathname(entry, destpath);
983		} else
984			destpath = archive_entry_pathname(entry);
985		if (destpath == NULL)
986			continue;
987		if (cpio->verbose)
988			fprintf(stderr, "%s\n", destpath);
989		if (cpio->dot)
990			fprintf(stderr, ".");
991		if (cpio->uid_override >= 0)
992			archive_entry_set_uid(entry, cpio->uid_override);
993		if (cpio->gid_override >= 0)
994			archive_entry_set_gid(entry, cpio->gid_override);
995		r = archive_write_header(ext, entry);
996		if (r != ARCHIVE_OK) {
997			fprintf(stderr, "%s: %s\n",
998			    archive_entry_pathname(entry),
999			    archive_error_string(ext));
1000		} else if (!archive_entry_size_is_set(entry)
1001		    || archive_entry_size(entry) > 0) {
1002			r = extract_data(a, ext);
1003			if (r != ARCHIVE_OK)
1004				cpio->return_value = 1;
1005		}
1006	}
1007	r = archive_read_close(a);
1008	if (cpio->dot)
1009		fprintf(stderr, "\n");
1010	if (r != ARCHIVE_OK)
1011		lafe_errc(1, 0, "%s", archive_error_string(a));
1012	r = archive_write_close(ext);
1013	if (r != ARCHIVE_OK)
1014		lafe_errc(1, 0, "%s", archive_error_string(ext));
1015	if (!cpio->quiet) {
1016		int64_t blocks = (archive_filter_bytes(a, 0) + 511)
1017			      / 512;
1018		fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1019		    blocks == 1 ? "block" : "blocks");
1020	}
1021	archive_read_free(a);
1022	archive_write_free(ext);
1023	exit(cpio->return_value);
1024}
1025
1026/*
1027 * Exits if there's a fatal error.  Returns ARCHIVE_OK
1028 * if everything is kosher.
1029 */
1030static int
1031extract_data(struct archive *ar, struct archive *aw)
1032{
1033	int r;
1034	size_t size;
1035	const void *block;
1036	int64_t offset;
1037
1038	for (;;) {
1039		r = archive_read_data_block(ar, &block, &size, &offset);
1040		if (r == ARCHIVE_EOF)
1041			return (ARCHIVE_OK);
1042		if (r != ARCHIVE_OK) {
1043			lafe_warnc(archive_errno(ar),
1044			    "%s", archive_error_string(ar));
1045			exit(1);
1046		}
1047		r = (int)archive_write_data_block(aw, block, size, offset);
1048		if (r != ARCHIVE_OK) {
1049			lafe_warnc(archive_errno(aw),
1050			    "%s", archive_error_string(aw));
1051			return (r);
1052		}
1053	}
1054}
1055
1056static void
1057mode_list(struct cpio *cpio)
1058{
1059	struct archive *a;
1060	struct archive_entry *entry;
1061	int r;
1062
1063	a = archive_read_new();
1064	if (a == NULL)
1065		lafe_errc(1, 0, "Couldn't allocate archive object");
1066	archive_read_support_filter_all(a);
1067	archive_read_support_format_all(a);
1068	if (cpio->passphrase != NULL)
1069		r = archive_read_add_passphrase(a, cpio->passphrase);
1070	else
1071		r = archive_read_set_passphrase_callback(a, cpio,
1072			&passphrase_callback);
1073	if (r != ARCHIVE_OK)
1074		lafe_errc(1, 0, "%s", archive_error_string(a));
1075
1076	if (archive_read_open_filename(a, cpio->filename,
1077					cpio->bytes_per_block))
1078		lafe_errc(1, archive_errno(a),
1079		    "%s", archive_error_string(a));
1080	for (;;) {
1081		r = archive_read_next_header(a, &entry);
1082		if (r == ARCHIVE_EOF)
1083			break;
1084		if (r != ARCHIVE_OK) {
1085			lafe_errc(1, archive_errno(a),
1086			    "%s", archive_error_string(a));
1087		}
1088		if (archive_match_path_excluded(cpio->matching, entry))
1089			continue;
1090		if (cpio->verbose)
1091			list_item_verbose(cpio, entry);
1092		else
1093			fprintf(stdout, "%s\n", archive_entry_pathname(entry));
1094	}
1095	r = archive_read_close(a);
1096	if (r != ARCHIVE_OK)
1097		lafe_errc(1, 0, "%s", archive_error_string(a));
1098	if (!cpio->quiet) {
1099		int64_t blocks = (archive_filter_bytes(a, 0) + 511)
1100			      / 512;
1101		fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1102		    blocks == 1 ? "block" : "blocks");
1103	}
1104	archive_read_free(a);
1105	exit(0);
1106}
1107
1108/*
1109 * Display information about the current file.
1110 *
1111 * The format here roughly duplicates the output of 'ls -l'.
1112 * This is based on SUSv2, where 'tar tv' is documented as
1113 * listing additional information in an "unspecified format,"
1114 * and 'pax -l' is documented as using the same format as 'ls -l'.
1115 */
1116static void
1117list_item_verbose(struct cpio *cpio, struct archive_entry *entry)
1118{
1119	char			 size[32];
1120	char			 date[32];
1121	char			 uids[16], gids[16];
1122	const char 		*uname, *gname;
1123	FILE			*out = stdout;
1124	const char		*fmt;
1125	time_t			 mtime;
1126	static time_t		 now;
1127
1128	if (!now)
1129		time(&now);
1130
1131	if (cpio->option_numeric_uid_gid) {
1132		/* Format numeric uid/gid for display. */
1133		strcpy(uids, cpio_i64toa(archive_entry_uid(entry)));
1134		uname = uids;
1135		strcpy(gids, cpio_i64toa(archive_entry_gid(entry)));
1136		gname = gids;
1137	} else {
1138		/* Use uname if it's present, else lookup name from uid. */
1139		uname = archive_entry_uname(entry);
1140		if (uname == NULL)
1141			uname = lookup_uname(cpio, (uid_t)archive_entry_uid(entry));
1142		/* Use gname if it's present, else lookup name from gid. */
1143		gname = archive_entry_gname(entry);
1144		if (gname == NULL)
1145			gname = lookup_gname(cpio, (uid_t)archive_entry_gid(entry));
1146	}
1147
1148	/* Print device number or file size. */
1149	if (archive_entry_filetype(entry) == AE_IFCHR
1150	    || archive_entry_filetype(entry) == AE_IFBLK) {
1151		snprintf(size, sizeof(size), "%lu,%lu",
1152		    (unsigned long)archive_entry_rdevmajor(entry),
1153		    (unsigned long)archive_entry_rdevminor(entry));
1154	} else {
1155		strcpy(size, cpio_i64toa(archive_entry_size(entry)));
1156	}
1157
1158	/* Format the time using 'ls -l' conventions. */
1159	mtime = archive_entry_mtime(entry);
1160#if defined(_WIN32) && !defined(__CYGWIN__)
1161	/* Windows' strftime function does not support %e format. */
1162	if (mtime - now > 365*86400/2
1163		|| mtime - now < -365*86400/2)
1164		fmt = cpio->day_first ? "%d %b  %Y" : "%b %d  %Y";
1165	else
1166		fmt = cpio->day_first ? "%d %b %H:%M" : "%b %d %H:%M";
1167#else
1168	if (mtime - now > 365*86400/2
1169		|| mtime - now < -365*86400/2)
1170		fmt = cpio->day_first ? "%e %b  %Y" : "%b %e  %Y";
1171	else
1172		fmt = cpio->day_first ? "%e %b %H:%M" : "%b %e %H:%M";
1173#endif
1174	strftime(date, sizeof(date), fmt, localtime(&mtime));
1175
1176	fprintf(out, "%s%3d %-8s %-8s %8s %12s %s",
1177	    archive_entry_strmode(entry),
1178	    archive_entry_nlink(entry),
1179	    uname, gname, size, date,
1180	    archive_entry_pathname(entry));
1181
1182	/* Extra information for links. */
1183	if (archive_entry_hardlink(entry)) /* Hard link */
1184		fprintf(out, " link to %s", archive_entry_hardlink(entry));
1185	else if (archive_entry_symlink(entry)) /* Symbolic link */
1186		fprintf(out, " -> %s", archive_entry_symlink(entry));
1187	fprintf(out, "\n");
1188}
1189
1190static void
1191mode_pass(struct cpio *cpio, const char *destdir)
1192{
1193	struct lafe_line_reader *lr;
1194	const char *p;
1195	int r;
1196
1197	/* Ensure target dir has a trailing '/' to simplify path surgery. */
1198	cpio->destdir = malloc(strlen(destdir) + 8);
1199	strcpy(cpio->destdir, destdir);
1200	if (destdir[strlen(destdir) - 1] != '/')
1201		strcat(cpio->destdir, "/");
1202
1203	cpio->archive = archive_write_disk_new();
1204	if (cpio->archive == NULL)
1205		lafe_errc(1, 0, "Failed to allocate archive object");
1206	r = archive_write_disk_set_options(cpio->archive, cpio->extract_flags);
1207	if (r != ARCHIVE_OK)
1208		lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1209	cpio->linkresolver = archive_entry_linkresolver_new();
1210	archive_write_disk_set_standard_lookup(cpio->archive);
1211
1212	cpio->archive_read_disk = archive_read_disk_new();
1213	if (cpio->archive_read_disk == NULL)
1214		lafe_errc(1, 0, "Failed to allocate archive object");
1215	if (cpio->option_follow_links)
1216		archive_read_disk_set_symlink_logical(cpio->archive_read_disk);
1217	else
1218		archive_read_disk_set_symlink_physical(cpio->archive_read_disk);
1219	archive_read_disk_set_standard_lookup(cpio->archive_read_disk);
1220
1221	lr = lafe_line_reader("-", cpio->option_null);
1222	while ((p = lafe_line_reader_next(lr)) != NULL)
1223		file_to_archive(cpio, p);
1224	lafe_line_reader_free(lr);
1225
1226	archive_entry_linkresolver_free(cpio->linkresolver);
1227	r = archive_write_close(cpio->archive);
1228	if (cpio->dot)
1229		fprintf(stderr, "\n");
1230	if (r != ARCHIVE_OK)
1231		lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1232
1233	if (!cpio->quiet) {
1234		int64_t blocks =
1235			(archive_filter_bytes(cpio->archive, 0) + 511)
1236			/ 512;
1237		fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1238		    blocks == 1 ? "block" : "blocks");
1239	}
1240
1241	archive_write_free(cpio->archive);
1242}
1243
1244/*
1245 * Prompt for a new name for this entry.  Returns a pointer to the
1246 * new name or NULL if the entry should not be copied.  This
1247 * implements the semantics defined in POSIX.1-1996, which specifies
1248 * that an input of '.' means the name should be unchanged.  GNU cpio
1249 * treats '.' as a literal new name.
1250 */
1251static const char *
1252cpio_rename(const char *name)
1253{
1254	static char buff[1024];
1255	FILE *t;
1256	char *p, *ret;
1257#if defined(_WIN32) && !defined(__CYGWIN__)
1258	FILE *to;
1259
1260	t = fopen("CONIN$", "r");
1261	if (t == NULL)
1262		return (name);
1263	to = fopen("CONOUT$", "w");
1264	if (to == NULL) {
1265		fclose(t);
1266		return (name);
1267	}
1268	fprintf(to, "%s (Enter/./(new name))? ", name);
1269	fclose(to);
1270#else
1271	t = fopen("/dev/tty", "r+");
1272	if (t == NULL)
1273		return (name);
1274	fprintf(t, "%s (Enter/./(new name))? ", name);
1275	fflush(t);
1276#endif
1277
1278	p = fgets(buff, sizeof(buff), t);
1279	fclose(t);
1280	if (p == NULL)
1281		/* End-of-file is a blank line. */
1282		return (NULL);
1283
1284	while (*p == ' ' || *p == '\t')
1285		++p;
1286	if (*p == '\n' || *p == '\0')
1287		/* Empty line. */
1288		return (NULL);
1289	if (*p == '.' && p[1] == '\n')
1290		/* Single period preserves original name. */
1291		return (name);
1292	ret = p;
1293	/* Trim the final newline. */
1294	while (*p != '\0' && *p != '\n')
1295		++p;
1296	/* Overwrite the final \n with a null character. */
1297	*p = '\0';
1298	return (ret);
1299}
1300
1301static void
1302free_cache(struct name_cache *cache)
1303{
1304	size_t i;
1305
1306	if (cache != NULL) {
1307		for (i = 0; i < cache->size; i++)
1308			free(cache->cache[i].name);
1309		free(cache);
1310	}
1311}
1312
1313/*
1314 * Lookup uname/gname from uid/gid, return NULL if no match.
1315 */
1316static const char *
1317lookup_name(struct cpio *cpio, struct name_cache **name_cache_variable,
1318    int (*lookup_fn)(struct cpio *, const char **, id_t), id_t id)
1319{
1320	char asnum[16];
1321	struct name_cache	*cache;
1322	const char *name;
1323	int slot;
1324
1325
1326	if (*name_cache_variable == NULL) {
1327		*name_cache_variable = calloc(1, sizeof(struct name_cache));
1328		if (*name_cache_variable == NULL)
1329			lafe_errc(1, ENOMEM, "No more memory");
1330		(*name_cache_variable)->size = name_cache_size;
1331	}
1332
1333	cache = *name_cache_variable;
1334	cache->probes++;
1335
1336	slot = id % cache->size;
1337	if (cache->cache[slot].name != NULL) {
1338		if (cache->cache[slot].id == id) {
1339			cache->hits++;
1340			return (cache->cache[slot].name);
1341		}
1342		free(cache->cache[slot].name);
1343		cache->cache[slot].name = NULL;
1344	}
1345
1346	if (lookup_fn(cpio, &name, id) == 0) {
1347		if (name == NULL || name[0] == '\0') {
1348			/* If lookup failed, format it as a number. */
1349			snprintf(asnum, sizeof(asnum), "%u", (unsigned)id);
1350			name = asnum;
1351		}
1352		cache->cache[slot].name = strdup(name);
1353		if (cache->cache[slot].name != NULL) {
1354			cache->cache[slot].id = id;
1355			return (cache->cache[slot].name);
1356		}
1357		/*
1358		 * Conveniently, NULL marks an empty slot, so
1359		 * if the strdup() fails, we've just failed to
1360		 * cache it.  No recovery necessary.
1361		 */
1362	}
1363	return (NULL);
1364}
1365
1366static const char *
1367lookup_uname(struct cpio *cpio, uid_t uid)
1368{
1369	return (lookup_name(cpio, &cpio->uname_cache,
1370		    &lookup_uname_helper, (id_t)uid));
1371}
1372
1373static int
1374lookup_uname_helper(struct cpio *cpio, const char **name, id_t id)
1375{
1376	struct passwd	*pwent;
1377
1378	(void)cpio; /* UNUSED */
1379
1380	errno = 0;
1381	pwent = getpwuid((uid_t)id);
1382	if (pwent == NULL) {
1383		*name = NULL;
1384		if (errno != 0 && errno != ENOENT)
1385			lafe_warnc(errno, "getpwuid(%s) failed",
1386			    cpio_i64toa((int64_t)id));
1387		return (errno);
1388	}
1389
1390	*name = pwent->pw_name;
1391	return (0);
1392}
1393
1394static const char *
1395lookup_gname(struct cpio *cpio, gid_t gid)
1396{
1397	return (lookup_name(cpio, &cpio->gname_cache,
1398		    &lookup_gname_helper, (id_t)gid));
1399}
1400
1401static int
1402lookup_gname_helper(struct cpio *cpio, const char **name, id_t id)
1403{
1404	struct group	*grent;
1405
1406	(void)cpio; /* UNUSED */
1407
1408	errno = 0;
1409	grent = getgrgid((gid_t)id);
1410	if (grent == NULL) {
1411		*name = NULL;
1412		if (errno != 0)
1413			lafe_warnc(errno, "getgrgid(%s) failed",
1414			    cpio_i64toa((int64_t)id));
1415		return (errno);
1416	}
1417
1418	*name = grent->gr_name;
1419	return (0);
1420}
1421
1422/*
1423 * It would be nice to just use printf() for formatting large numbers,
1424 * but the compatibility problems are a big headache.  Hence the
1425 * following simple utility function.
1426 */
1427const char *
1428cpio_i64toa(int64_t n0)
1429{
1430	/* 2^64 =~ 1.8 * 10^19, so 20 decimal digits suffice.
1431	 * We also need 1 byte for '-' and 1 for '\0'.
1432	 */
1433	static char buff[22];
1434	int64_t n = n0 < 0 ? -n0 : n0;
1435	char *p = buff + sizeof(buff);
1436
1437	*--p = '\0';
1438	do {
1439		*--p = '0' + (int)(n % 10);
1440		n /= 10;
1441	} while (n > 0);
1442	if (n0 < 0)
1443		*--p = '-';
1444	return p;
1445}
1446
1447#define PPBUFF_SIZE 1024
1448static const char *
1449passphrase_callback(struct archive *a, void *_client_data)
1450{
1451	struct cpio *cpio = (struct cpio *)_client_data;
1452	(void)a; /* UNUSED */
1453
1454	if (cpio->ppbuff == NULL) {
1455		cpio->ppbuff = malloc(PPBUFF_SIZE);
1456		if (cpio->ppbuff == NULL)
1457			lafe_errc(1, errno, "Out of memory");
1458	}
1459	return lafe_readpassphrase("Enter passphrase:",
1460		cpio->ppbuff, PPBUFF_SIZE);
1461}
1462
1463static void
1464passphrase_free(char *ppbuff)
1465{
1466	if (ppbuff != NULL) {
1467		memset(ppbuff, 0, PPBUFF_SIZE);
1468		free(ppbuff);
1469	}
1470}
1471