1 /*-
2 * Copyright (c) 2003-2007 Tim Kientzle
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer
10 * in this position and unchanged.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 */
26
27
28 #include "cpio_platform.h"
29 __FBSDID("$FreeBSD: src/usr.bin/cpio/cpio.c,v 1.15 2008/12/06 07:30:40 kientzle Exp $");
30
31 #include <sys/types.h>
32 #include <archive.h>
33 #include <archive_entry.h>
34
35 #ifdef HAVE_SYS_MKDEV_H
36 #include <sys/mkdev.h>
37 #endif
38 #ifdef HAVE_SYS_STAT_H
39 #include <sys/stat.h>
40 #endif
41 #ifdef HAVE_SYS_TIME_H
42 #include <sys/time.h>
43 #endif
44 #ifdef HAVE_ERRNO_H
45 #include <errno.h>
46 #endif
47 #ifdef HAVE_FCNTL_H
48 #include <fcntl.h>
49 #endif
50 #ifdef HAVE_GRP_H
51 #include <grp.h>
52 #endif
53 #ifdef HAVE_LOCALE_H
54 #include <locale.h>
55 #endif
56 #ifdef HAVE_PWD_H
57 #include <pwd.h>
58 #endif
59 #ifdef HAVE_SIGNAL_H
60 #include <signal.h>
61 #endif
62 #ifdef HAVE_STDARG_H
63 #include <stdarg.h>
64 #endif
65 #ifdef HAVE_STDINT_H
66 #include <stdint.h>
67 #endif
68 #include <stdio.h>
69 #ifdef HAVE_STDLIB_H
70 #include <stdlib.h>
71 #endif
72 #ifdef HAVE_STRING_H
73 #include <string.h>
74 #endif
75 #ifdef HAVE_UNISTD_H
76 #include <unistd.h>
77 #endif
78 #ifdef HAVE_TIME_H
79 #include <time.h>
80 #endif
81
82 #include "cpio.h"
83 #include "err.h"
84 #include "line_reader.h"
85 #include "passphrase.h"
86
87 /* Fixed size of uname/gname caches. */
88 #define name_cache_size 101
89
90 #ifndef O_BINARY
91 #define O_BINARY 0
92 #endif
93
94 struct 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
104 static int extract_data(struct archive *, struct archive *);
105 const char * cpio_i64toa(int64_t);
106 static const char *cpio_rename(const char *name);
107 static int entry_to_archive(struct cpio *, struct archive_entry *);
108 static int file_to_archive(struct cpio *, const char *);
109 static void free_cache(struct name_cache *cache);
110 static void list_item_verbose(struct cpio *, struct archive_entry *);
111 static void long_help(void);
112 static const char *lookup_gname(struct cpio *, gid_t gid);
113 static int lookup_gname_helper(struct cpio *,
114 const char **name, id_t gid);
115 static const char *lookup_uname(struct cpio *, uid_t uid);
116 static int lookup_uname_helper(struct cpio *,
117 const char **name, id_t uid);
118 static void mode_in(struct cpio *);
119 static void mode_list(struct cpio *);
120 static void mode_out(struct cpio *);
121 static void mode_pass(struct cpio *, const char *);
122 static const char *remove_leading_slash(const char *);
123 static int restore_time(struct cpio *, struct archive_entry *,
124 const char *, int fd);
125 static void usage(void);
126 static void version(void);
127 static const char * passphrase_callback(struct archive *, void *);
128 static void passphrase_free(char *);
129
130 int
main(int argc,char * argv[])131 main(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
435 static void
usage(void)436 usage(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
450 static 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 */
472 static void
long_help(void)473 long_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
498 static void
version(void)499 version(void)
500 {
501 fprintf(stdout,"bsdcpio %s - %s\n",
502 BSDCPIO_VERSION_STRING,
503 archive_version_details());
504 exit(0);
505 }
506
507 static void
mode_out(struct cpio * cpio)508 mode_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
633 static const char *
remove_leading_slash(const char * p)634 remove_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 */
681 static int
file_to_archive(struct cpio * cpio,const char * srcpath)682 file_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 archive_entry_free(entry);
707 cpio->return_value = 1;
708 return (r);
709 }
710
711 if (cpio->uid_override >= 0) {
712 archive_entry_set_uid(entry, cpio->uid_override);
713 archive_entry_set_uname(entry, cpio->uname_override);
714 }
715 if (cpio->gid_override >= 0) {
716 archive_entry_set_gid(entry, cpio->gid_override);
717 archive_entry_set_gname(entry, cpio->gname_override);
718 }
719
720 /*
721 * Generate a destination path for this entry.
722 * "destination path" is the name to which it will be copied in
723 * pass mode or the name that will go into the archive in
724 * output mode.
725 */
726 destpath = srcpath;
727 if (cpio->destdir) {
728 len = strlen(cpio->destdir) + strlen(srcpath) + 8;
729 if (len >= cpio->pass_destpath_alloc) {
730 while (len >= cpio->pass_destpath_alloc) {
731 cpio->pass_destpath_alloc += 512;
732 cpio->pass_destpath_alloc *= 2;
733 }
734 free(cpio->pass_destpath);
735 cpio->pass_destpath = malloc(cpio->pass_destpath_alloc);
736 if (cpio->pass_destpath == NULL)
737 lafe_errc(1, ENOMEM,
738 "Can't allocate path buffer");
739 }
740 strcpy(cpio->pass_destpath, cpio->destdir);
741 strcat(cpio->pass_destpath, remove_leading_slash(srcpath));
742 destpath = cpio->pass_destpath;
743 }
744 if (cpio->option_rename)
745 destpath = cpio_rename(destpath);
746 if (destpath == NULL)
747 return (0);
748 archive_entry_copy_pathname(entry, destpath);
749
750 /*
751 * If we're trying to preserve hardlinks, match them here.
752 */
753 spare = NULL;
754 if (cpio->linkresolver != NULL
755 && archive_entry_filetype(entry) != AE_IFDIR) {
756 archive_entry_linkify(cpio->linkresolver, &entry, &spare);
757 }
758
759 if (entry != NULL) {
760 r = entry_to_archive(cpio, entry);
761 archive_entry_free(entry);
762 if (spare != NULL) {
763 if (r == 0)
764 r = entry_to_archive(cpio, spare);
765 archive_entry_free(spare);
766 }
767 }
768 return (r);
769 }
770
771 static int
entry_to_archive(struct cpio * cpio,struct archive_entry * entry)772 entry_to_archive(struct cpio *cpio, struct archive_entry *entry)
773 {
774 const char *destpath = archive_entry_pathname(entry);
775 const char *srcpath = archive_entry_sourcepath(entry);
776 int fd = -1;
777 ssize_t bytes_read;
778 int r;
779
780 /* Print out the destination name to the user. */
781 if (cpio->verbose)
782 fprintf(stderr,"%s", destpath);
783 if (cpio->dot)
784 fprintf(stderr, ".");
785
786 /*
787 * Option_link only makes sense in pass mode and for
788 * regular files. Also note: if a link operation fails
789 * because of cross-device restrictions, we'll fall back
790 * to copy mode for that entry.
791 *
792 * TODO: Test other cpio implementations to see if they
793 * hard-link anything other than regular files here.
794 */
795 if (cpio->option_link
796 && archive_entry_filetype(entry) == AE_IFREG)
797 {
798 struct archive_entry *t;
799 /* Save the original entry in case we need it later. */
800 t = archive_entry_clone(entry);
801 if (t == NULL)
802 lafe_errc(1, ENOMEM, "Can't create link");
803 /* Note: link(2) doesn't create parent directories,
804 * so we use archive_write_header() instead as a
805 * convenience. */
806 archive_entry_set_hardlink(t, srcpath);
807 /* This is a straight link that carries no data. */
808 archive_entry_set_size(t, 0);
809 r = archive_write_header(cpio->archive, t);
810 archive_entry_free(t);
811 if (r != ARCHIVE_OK)
812 lafe_warnc(archive_errno(cpio->archive),
813 "%s", archive_error_string(cpio->archive));
814 if (r == ARCHIVE_FATAL)
815 exit(1);
816 #ifdef EXDEV
817 if (r != ARCHIVE_OK && archive_errno(cpio->archive) == EXDEV) {
818 /* Cross-device link: Just fall through and use
819 * the original entry to copy the file over. */
820 lafe_warnc(0, "Copying file instead");
821 } else
822 #endif
823 return (0);
824 }
825
826 /*
827 * Make sure we can open the file (if necessary) before
828 * trying to write the header.
829 */
830 if (archive_entry_filetype(entry) == AE_IFREG) {
831 if (archive_entry_size(entry) > 0) {
832 fd = open(srcpath, O_RDONLY | O_BINARY);
833 if (fd < 0) {
834 lafe_warnc(errno,
835 "%s: could not open file", srcpath);
836 goto cleanup;
837 }
838 }
839 } else {
840 archive_entry_set_size(entry, 0);
841 }
842
843 r = archive_write_header(cpio->archive, entry);
844
845 if (r != ARCHIVE_OK)
846 lafe_warnc(archive_errno(cpio->archive),
847 "%s: %s",
848 srcpath,
849 archive_error_string(cpio->archive));
850
851 if (r == ARCHIVE_FATAL)
852 exit(1);
853
854 if (r >= ARCHIVE_WARN && archive_entry_size(entry) > 0 && fd >= 0) {
855 bytes_read = read(fd, cpio->buff, (unsigned)cpio->buff_size);
856 while (bytes_read > 0) {
857 ssize_t bytes_write;
858 bytes_write = archive_write_data(cpio->archive,
859 cpio->buff, bytes_read);
860 if (bytes_write < 0)
861 lafe_errc(1, archive_errno(cpio->archive),
862 "%s", archive_error_string(cpio->archive));
863 if (bytes_write < bytes_read) {
864 lafe_warnc(0,
865 "Truncated write; file may have "
866 "grown while being archived.");
867 }
868 bytes_read = read(fd, cpio->buff,
869 (unsigned)cpio->buff_size);
870 }
871 }
872
873 fd = restore_time(cpio, entry, srcpath, fd);
874
875 cleanup:
876 if (cpio->verbose)
877 fprintf(stderr,"\n");
878 if (fd >= 0)
879 close(fd);
880 return (0);
881 }
882
883 static int
restore_time(struct cpio * cpio,struct archive_entry * entry,const char * name,int fd)884 restore_time(struct cpio *cpio, struct archive_entry *entry,
885 const char *name, int fd)
886 {
887 #ifndef HAVE_UTIMES
888 static int warned = 0;
889
890 (void)cpio; /* UNUSED */
891 (void)entry; /* UNUSED */
892 (void)name; /* UNUSED */
893
894 if (!warned)
895 lafe_warnc(0, "Can't restore access times on this platform");
896 warned = 1;
897 return (fd);
898 #else
899 #if defined(_WIN32) && !defined(__CYGWIN__)
900 struct __timeval times[2];
901 #else
902 struct timeval times[2];
903 #endif
904
905 if (!cpio->option_atime_restore)
906 return (fd);
907
908 times[1].tv_sec = archive_entry_mtime(entry);
909 times[1].tv_usec = archive_entry_mtime_nsec(entry) / 1000;
910
911 times[0].tv_sec = archive_entry_atime(entry);
912 times[0].tv_usec = archive_entry_atime_nsec(entry) / 1000;
913
914 #if defined(HAVE_FUTIMES) && !defined(__CYGWIN__)
915 if (fd >= 0 && futimes(fd, times) == 0)
916 return (fd);
917 #endif
918 /*
919 * Some platform cannot restore access times if the file descriptor
920 * is still opened.
921 */
922 if (fd >= 0) {
923 close(fd);
924 fd = -1;
925 }
926
927 #ifdef HAVE_LUTIMES
928 if (lutimes(name, times) != 0)
929 #else
930 if ((AE_IFLNK != archive_entry_filetype(entry))
931 && utimes(name, times) != 0)
932 #endif
933 lafe_warnc(errno, "Can't update time for %s", name);
934 #endif
935 return (fd);
936 }
937
938
939 static void
mode_in(struct cpio * cpio)940 mode_in(struct cpio *cpio)
941 {
942 struct archive *a;
943 struct archive_entry *entry;
944 struct archive *ext;
945 const char *destpath;
946 int r;
947
948 ext = archive_write_disk_new();
949 if (ext == NULL)
950 lafe_errc(1, 0, "Couldn't allocate restore object");
951 r = archive_write_disk_set_options(ext, cpio->extract_flags);
952 if (r != ARCHIVE_OK)
953 lafe_errc(1, 0, "%s", archive_error_string(ext));
954 a = archive_read_new();
955 if (a == NULL)
956 lafe_errc(1, 0, "Couldn't allocate archive object");
957 archive_read_support_filter_all(a);
958 archive_read_support_format_all(a);
959 if (cpio->passphrase != NULL)
960 r = archive_read_add_passphrase(a, cpio->passphrase);
961 else
962 r = archive_read_set_passphrase_callback(a, cpio,
963 &passphrase_callback);
964 if (r != ARCHIVE_OK)
965 lafe_errc(1, 0, "%s", archive_error_string(a));
966
967 if (archive_read_open_filename(a, cpio->filename,
968 cpio->bytes_per_block))
969 lafe_errc(1, archive_errno(a),
970 "%s", archive_error_string(a));
971 for (;;) {
972 r = archive_read_next_header(a, &entry);
973 if (r == ARCHIVE_EOF)
974 break;
975 if (r != ARCHIVE_OK) {
976 lafe_errc(1, archive_errno(a),
977 "%s", archive_error_string(a));
978 }
979 if (archive_match_path_excluded(cpio->matching, entry))
980 continue;
981 if (cpio->option_rename) {
982 destpath = cpio_rename(archive_entry_pathname(entry));
983 archive_entry_set_pathname(entry, destpath);
984 } else
985 destpath = archive_entry_pathname(entry);
986 if (destpath == NULL)
987 continue;
988 if (cpio->verbose)
989 fprintf(stderr, "%s\n", destpath);
990 if (cpio->dot)
991 fprintf(stderr, ".");
992 if (cpio->uid_override >= 0)
993 archive_entry_set_uid(entry, cpio->uid_override);
994 if (cpio->gid_override >= 0)
995 archive_entry_set_gid(entry, cpio->gid_override);
996 r = archive_write_header(ext, entry);
997 if (r != ARCHIVE_OK) {
998 fprintf(stderr, "%s: %s\n",
999 archive_entry_pathname(entry),
1000 archive_error_string(ext));
1001 } else if (!archive_entry_size_is_set(entry)
1002 || archive_entry_size(entry) > 0) {
1003 r = extract_data(a, ext);
1004 if (r != ARCHIVE_OK)
1005 cpio->return_value = 1;
1006 }
1007 }
1008 r = archive_read_close(a);
1009 if (cpio->dot)
1010 fprintf(stderr, "\n");
1011 if (r != ARCHIVE_OK)
1012 lafe_errc(1, 0, "%s", archive_error_string(a));
1013 r = archive_write_close(ext);
1014 if (r != ARCHIVE_OK)
1015 lafe_errc(1, 0, "%s", archive_error_string(ext));
1016 if (!cpio->quiet) {
1017 int64_t blocks = (archive_filter_bytes(a, 0) + 511)
1018 / 512;
1019 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1020 blocks == 1 ? "block" : "blocks");
1021 }
1022 archive_read_free(a);
1023 archive_write_free(ext);
1024 exit(cpio->return_value);
1025 }
1026
1027 /*
1028 * Exits if there's a fatal error. Returns ARCHIVE_OK
1029 * if everything is kosher.
1030 */
1031 static int
extract_data(struct archive * ar,struct archive * aw)1032 extract_data(struct archive *ar, struct archive *aw)
1033 {
1034 int r;
1035 size_t size;
1036 const void *block;
1037 int64_t offset;
1038
1039 for (;;) {
1040 r = archive_read_data_block(ar, &block, &size, &offset);
1041 if (r == ARCHIVE_EOF)
1042 return (ARCHIVE_OK);
1043 if (r != ARCHIVE_OK) {
1044 lafe_warnc(archive_errno(ar),
1045 "%s", archive_error_string(ar));
1046 exit(1);
1047 }
1048 r = (int)archive_write_data_block(aw, block, size, offset);
1049 if (r != ARCHIVE_OK) {
1050 lafe_warnc(archive_errno(aw),
1051 "%s", archive_error_string(aw));
1052 return (r);
1053 }
1054 }
1055 }
1056
1057 static void
mode_list(struct cpio * cpio)1058 mode_list(struct cpio *cpio)
1059 {
1060 struct archive *a;
1061 struct archive_entry *entry;
1062 int r;
1063
1064 a = archive_read_new();
1065 if (a == NULL)
1066 lafe_errc(1, 0, "Couldn't allocate archive object");
1067 archive_read_support_filter_all(a);
1068 archive_read_support_format_all(a);
1069 if (cpio->passphrase != NULL)
1070 r = archive_read_add_passphrase(a, cpio->passphrase);
1071 else
1072 r = archive_read_set_passphrase_callback(a, cpio,
1073 &passphrase_callback);
1074 if (r != ARCHIVE_OK)
1075 lafe_errc(1, 0, "%s", archive_error_string(a));
1076
1077 if (archive_read_open_filename(a, cpio->filename,
1078 cpio->bytes_per_block))
1079 lafe_errc(1, archive_errno(a),
1080 "%s", archive_error_string(a));
1081 for (;;) {
1082 r = archive_read_next_header(a, &entry);
1083 if (r == ARCHIVE_EOF)
1084 break;
1085 if (r != ARCHIVE_OK) {
1086 lafe_errc(1, archive_errno(a),
1087 "%s", archive_error_string(a));
1088 }
1089 if (archive_match_path_excluded(cpio->matching, entry))
1090 continue;
1091 if (cpio->verbose)
1092 list_item_verbose(cpio, entry);
1093 else
1094 fprintf(stdout, "%s\n", archive_entry_pathname(entry));
1095 }
1096 r = archive_read_close(a);
1097 if (r != ARCHIVE_OK)
1098 lafe_errc(1, 0, "%s", archive_error_string(a));
1099 if (!cpio->quiet) {
1100 int64_t blocks = (archive_filter_bytes(a, 0) + 511)
1101 / 512;
1102 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1103 blocks == 1 ? "block" : "blocks");
1104 }
1105 archive_read_free(a);
1106 exit(0);
1107 }
1108
1109 /*
1110 * Display information about the current file.
1111 *
1112 * The format here roughly duplicates the output of 'ls -l'.
1113 * This is based on SUSv2, where 'tar tv' is documented as
1114 * listing additional information in an "unspecified format,"
1115 * and 'pax -l' is documented as using the same format as 'ls -l'.
1116 */
1117 static void
list_item_verbose(struct cpio * cpio,struct archive_entry * entry)1118 list_item_verbose(struct cpio *cpio, struct archive_entry *entry)
1119 {
1120 char size[32];
1121 char date[32];
1122 char uids[16], gids[16];
1123 const char *uname, *gname;
1124 FILE *out = stdout;
1125 const char *fmt;
1126 time_t mtime;
1127 static time_t now;
1128
1129 if (!now)
1130 time(&now);
1131
1132 if (cpio->option_numeric_uid_gid) {
1133 /* Format numeric uid/gid for display. */
1134 strcpy(uids, cpio_i64toa(archive_entry_uid(entry)));
1135 uname = uids;
1136 strcpy(gids, cpio_i64toa(archive_entry_gid(entry)));
1137 gname = gids;
1138 } else {
1139 /* Use uname if it's present, else lookup name from uid. */
1140 uname = archive_entry_uname(entry);
1141 if (uname == NULL)
1142 uname = lookup_uname(cpio, (uid_t)archive_entry_uid(entry));
1143 /* Use gname if it's present, else lookup name from gid. */
1144 gname = archive_entry_gname(entry);
1145 if (gname == NULL)
1146 gname = lookup_gname(cpio, (uid_t)archive_entry_gid(entry));
1147 }
1148
1149 /* Print device number or file size. */
1150 if (archive_entry_filetype(entry) == AE_IFCHR
1151 || archive_entry_filetype(entry) == AE_IFBLK) {
1152 snprintf(size, sizeof(size), "%lu,%lu",
1153 (unsigned long)archive_entry_rdevmajor(entry),
1154 (unsigned long)archive_entry_rdevminor(entry));
1155 } else {
1156 strcpy(size, cpio_i64toa(archive_entry_size(entry)));
1157 }
1158
1159 /* Format the time using 'ls -l' conventions. */
1160 mtime = archive_entry_mtime(entry);
1161 #if defined(_WIN32) && !defined(__CYGWIN__)
1162 /* Windows' strftime function does not support %e format. */
1163 if (mtime - now > 365*86400/2
1164 || mtime - now < -365*86400/2)
1165 fmt = cpio->day_first ? "%d %b %Y" : "%b %d %Y";
1166 else
1167 fmt = cpio->day_first ? "%d %b %H:%M" : "%b %d %H:%M";
1168 #else
1169 if (mtime - now > 365*86400/2
1170 || mtime - now < -365*86400/2)
1171 fmt = cpio->day_first ? "%e %b %Y" : "%b %e %Y";
1172 else
1173 fmt = cpio->day_first ? "%e %b %H:%M" : "%b %e %H:%M";
1174 #endif
1175 strftime(date, sizeof(date), fmt, localtime(&mtime));
1176
1177 fprintf(out, "%s%3d %-8s %-8s %8s %12s %s",
1178 archive_entry_strmode(entry),
1179 archive_entry_nlink(entry),
1180 uname, gname, size, date,
1181 archive_entry_pathname(entry));
1182
1183 /* Extra information for links. */
1184 if (archive_entry_hardlink(entry)) /* Hard link */
1185 fprintf(out, " link to %s", archive_entry_hardlink(entry));
1186 else if (archive_entry_symlink(entry)) /* Symbolic link */
1187 fprintf(out, " -> %s", archive_entry_symlink(entry));
1188 fprintf(out, "\n");
1189 }
1190
1191 static void
mode_pass(struct cpio * cpio,const char * destdir)1192 mode_pass(struct cpio *cpio, const char *destdir)
1193 {
1194 struct lafe_line_reader *lr;
1195 const char *p;
1196 int r;
1197
1198 /* Ensure target dir has a trailing '/' to simplify path surgery. */
1199 cpio->destdir = malloc(strlen(destdir) + 8);
1200 strcpy(cpio->destdir, destdir);
1201 if (destdir[strlen(destdir) - 1] != '/')
1202 strcat(cpio->destdir, "/");
1203
1204 cpio->archive = archive_write_disk_new();
1205 if (cpio->archive == NULL)
1206 lafe_errc(1, 0, "Failed to allocate archive object");
1207 r = archive_write_disk_set_options(cpio->archive, cpio->extract_flags);
1208 if (r != ARCHIVE_OK)
1209 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1210 cpio->linkresolver = archive_entry_linkresolver_new();
1211 archive_write_disk_set_standard_lookup(cpio->archive);
1212
1213 cpio->archive_read_disk = archive_read_disk_new();
1214 if (cpio->archive_read_disk == NULL)
1215 lafe_errc(1, 0, "Failed to allocate archive object");
1216 if (cpio->option_follow_links)
1217 archive_read_disk_set_symlink_logical(cpio->archive_read_disk);
1218 else
1219 archive_read_disk_set_symlink_physical(cpio->archive_read_disk);
1220 archive_read_disk_set_standard_lookup(cpio->archive_read_disk);
1221
1222 lr = lafe_line_reader("-", cpio->option_null);
1223 while ((p = lafe_line_reader_next(lr)) != NULL)
1224 file_to_archive(cpio, p);
1225 lafe_line_reader_free(lr);
1226
1227 archive_entry_linkresolver_free(cpio->linkresolver);
1228 r = archive_write_close(cpio->archive);
1229 if (cpio->dot)
1230 fprintf(stderr, "\n");
1231 if (r != ARCHIVE_OK)
1232 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1233
1234 if (!cpio->quiet) {
1235 int64_t blocks =
1236 (archive_filter_bytes(cpio->archive, 0) + 511)
1237 / 512;
1238 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1239 blocks == 1 ? "block" : "blocks");
1240 }
1241
1242 archive_write_free(cpio->archive);
1243 }
1244
1245 /*
1246 * Prompt for a new name for this entry. Returns a pointer to the
1247 * new name or NULL if the entry should not be copied. This
1248 * implements the semantics defined in POSIX.1-1996, which specifies
1249 * that an input of '.' means the name should be unchanged. GNU cpio
1250 * treats '.' as a literal new name.
1251 */
1252 static const char *
cpio_rename(const char * name)1253 cpio_rename(const char *name)
1254 {
1255 static char buff[1024];
1256 FILE *t;
1257 char *p, *ret;
1258 #if defined(_WIN32) && !defined(__CYGWIN__)
1259 FILE *to;
1260
1261 t = fopen("CONIN$", "r");
1262 if (t == NULL)
1263 return (name);
1264 to = fopen("CONOUT$", "w");
1265 if (to == NULL) {
1266 fclose(t);
1267 return (name);
1268 }
1269 fprintf(to, "%s (Enter/./(new name))? ", name);
1270 fclose(to);
1271 #else
1272 t = fopen("/dev/tty", "r+");
1273 if (t == NULL)
1274 return (name);
1275 fprintf(t, "%s (Enter/./(new name))? ", name);
1276 fflush(t);
1277 #endif
1278
1279 p = fgets(buff, sizeof(buff), t);
1280 fclose(t);
1281 if (p == NULL)
1282 /* End-of-file is a blank line. */
1283 return (NULL);
1284
1285 while (*p == ' ' || *p == '\t')
1286 ++p;
1287 if (*p == '\n' || *p == '\0')
1288 /* Empty line. */
1289 return (NULL);
1290 if (*p == '.' && p[1] == '\n')
1291 /* Single period preserves original name. */
1292 return (name);
1293 ret = p;
1294 /* Trim the final newline. */
1295 while (*p != '\0' && *p != '\n')
1296 ++p;
1297 /* Overwrite the final \n with a null character. */
1298 *p = '\0';
1299 return (ret);
1300 }
1301
1302 static void
free_cache(struct name_cache * cache)1303 free_cache(struct name_cache *cache)
1304 {
1305 size_t i;
1306
1307 if (cache != NULL) {
1308 for (i = 0; i < cache->size; i++)
1309 free(cache->cache[i].name);
1310 free(cache);
1311 }
1312 }
1313
1314 /*
1315 * Lookup uname/gname from uid/gid, return NULL if no match.
1316 */
1317 static const char *
lookup_name(struct cpio * cpio,struct name_cache ** name_cache_variable,int (* lookup_fn)(struct cpio *,const char **,id_t),id_t id)1318 lookup_name(struct cpio *cpio, struct name_cache **name_cache_variable,
1319 int (*lookup_fn)(struct cpio *, const char **, id_t), id_t id)
1320 {
1321 char asnum[16];
1322 struct name_cache *cache;
1323 const char *name;
1324 int slot;
1325
1326
1327 if (*name_cache_variable == NULL) {
1328 *name_cache_variable = calloc(1, sizeof(struct name_cache));
1329 if (*name_cache_variable == NULL)
1330 lafe_errc(1, ENOMEM, "No more memory");
1331 (*name_cache_variable)->size = name_cache_size;
1332 }
1333
1334 cache = *name_cache_variable;
1335 cache->probes++;
1336
1337 slot = id % cache->size;
1338 if (cache->cache[slot].name != NULL) {
1339 if (cache->cache[slot].id == id) {
1340 cache->hits++;
1341 return (cache->cache[slot].name);
1342 }
1343 free(cache->cache[slot].name);
1344 cache->cache[slot].name = NULL;
1345 }
1346
1347 if (lookup_fn(cpio, &name, id) == 0) {
1348 if (name == NULL || name[0] == '\0') {
1349 /* If lookup failed, format it as a number. */
1350 snprintf(asnum, sizeof(asnum), "%u", (unsigned)id);
1351 name = asnum;
1352 }
1353 cache->cache[slot].name = strdup(name);
1354 if (cache->cache[slot].name != NULL) {
1355 cache->cache[slot].id = id;
1356 return (cache->cache[slot].name);
1357 }
1358 /*
1359 * Conveniently, NULL marks an empty slot, so
1360 * if the strdup() fails, we've just failed to
1361 * cache it. No recovery necessary.
1362 */
1363 }
1364 return (NULL);
1365 }
1366
1367 static const char *
lookup_uname(struct cpio * cpio,uid_t uid)1368 lookup_uname(struct cpio *cpio, uid_t uid)
1369 {
1370 return (lookup_name(cpio, &cpio->uname_cache,
1371 &lookup_uname_helper, (id_t)uid));
1372 }
1373
1374 static int
lookup_uname_helper(struct cpio * cpio,const char ** name,id_t id)1375 lookup_uname_helper(struct cpio *cpio, const char **name, id_t id)
1376 {
1377 struct passwd *pwent;
1378
1379 (void)cpio; /* UNUSED */
1380
1381 errno = 0;
1382 pwent = getpwuid((uid_t)id);
1383 if (pwent == NULL) {
1384 *name = NULL;
1385 if (errno != 0 && errno != ENOENT)
1386 lafe_warnc(errno, "getpwuid(%s) failed",
1387 cpio_i64toa((int64_t)id));
1388 return (errno);
1389 }
1390
1391 *name = pwent->pw_name;
1392 return (0);
1393 }
1394
1395 static const char *
lookup_gname(struct cpio * cpio,gid_t gid)1396 lookup_gname(struct cpio *cpio, gid_t gid)
1397 {
1398 return (lookup_name(cpio, &cpio->gname_cache,
1399 &lookup_gname_helper, (id_t)gid));
1400 }
1401
1402 static int
lookup_gname_helper(struct cpio * cpio,const char ** name,id_t id)1403 lookup_gname_helper(struct cpio *cpio, const char **name, id_t id)
1404 {
1405 struct group *grent;
1406
1407 (void)cpio; /* UNUSED */
1408
1409 errno = 0;
1410 grent = getgrgid((gid_t)id);
1411 if (grent == NULL) {
1412 *name = NULL;
1413 if (errno != 0)
1414 lafe_warnc(errno, "getgrgid(%s) failed",
1415 cpio_i64toa((int64_t)id));
1416 return (errno);
1417 }
1418
1419 *name = grent->gr_name;
1420 return (0);
1421 }
1422
1423 /*
1424 * It would be nice to just use printf() for formatting large numbers,
1425 * but the compatibility problems are a big headache. Hence the
1426 * following simple utility function.
1427 */
1428 const char *
cpio_i64toa(int64_t n0)1429 cpio_i64toa(int64_t n0)
1430 {
1431 /* 2^64 =~ 1.8 * 10^19, so 20 decimal digits suffice.
1432 * We also need 1 byte for '-' and 1 for '\0'.
1433 */
1434 static char buff[22];
1435 int64_t n = n0 < 0 ? -n0 : n0;
1436 char *p = buff + sizeof(buff);
1437
1438 *--p = '\0';
1439 do {
1440 *--p = '0' + (int)(n % 10);
1441 n /= 10;
1442 } while (n > 0);
1443 if (n0 < 0)
1444 *--p = '-';
1445 return p;
1446 }
1447
1448 #define PPBUFF_SIZE 1024
1449 static const char *
passphrase_callback(struct archive * a,void * _client_data)1450 passphrase_callback(struct archive *a, void *_client_data)
1451 {
1452 struct cpio *cpio = (struct cpio *)_client_data;
1453 (void)a; /* UNUSED */
1454
1455 if (cpio->ppbuff == NULL) {
1456 cpio->ppbuff = malloc(PPBUFF_SIZE);
1457 if (cpio->ppbuff == NULL)
1458 lafe_errc(1, errno, "Out of memory");
1459 }
1460 return lafe_readpassphrase("Enter passphrase:",
1461 cpio->ppbuff, PPBUFF_SIZE);
1462 }
1463
1464 static void
passphrase_free(char * ppbuff)1465 passphrase_free(char *ppbuff)
1466 {
1467 if (ppbuff != NULL) {
1468 memset(ppbuff, 0, PPBUFF_SIZE);
1469 free(ppbuff);
1470 }
1471 }
1472