]> CyberLeo.Net >> Repos - FreeBSD/stable/10.git/blob - contrib/libarchive/cpio/cpio.c
MFC r317782,318181:
[FreeBSD/stable/10.git] / contrib / libarchive / cpio / cpio.c
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$");
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) __LA_DEAD;
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 *) __LA_DEAD;
119 static void     mode_list(struct cpio *) __LA_DEAD;
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) __LA_DEAD;
126 static void     version(void) __LA_DEAD;
127 static const char * passphrase_callback(struct archive *, void *);
128 static void     passphrase_free(char *);
129
130 int
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
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
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
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
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         archive_entry_linkresolver_free(cpio->linkresolver);
632 }
633
634 static const char *
635 remove_leading_slash(const char *p)
636 {
637         const char *rp;
638
639         /* Remove leading "//./" or "//?/" or "//?/UNC/"
640          * (absolute path prefixes used by Windows API) */
641         if ((p[0] == '/' || p[0] == '\\') &&
642             (p[1] == '/' || p[1] == '\\') &&
643             (p[2] == '.' || p[2] == '?') &&
644             (p[3] == '/' || p[3] == '\\'))
645         {
646                 if (p[2] == '?' &&
647                     (p[4] == 'U' || p[4] == 'u') &&
648                     (p[5] == 'N' || p[5] == 'n') &&
649                     (p[6] == 'C' || p[6] == 'c') &&
650                     (p[7] == '/' || p[7] == '\\'))
651                         p += 8;
652                 else
653                         p += 4;
654         }
655         do {
656                 rp = p;
657                 /* Remove leading drive letter from archives created
658                  * on Windows. */
659                 if (((p[0] >= 'a' && p[0] <= 'z') ||
660                      (p[0] >= 'A' && p[0] <= 'Z')) &&
661                          p[1] == ':') {
662                         p += 2;
663                 }
664                 /* Remove leading "/../", "//", etc. */
665                 while (p[0] == '/' || p[0] == '\\') {
666                         if (p[1] == '.' && p[2] == '.' &&
667                                 (p[3] == '/' || p[3] == '\\')) {
668                                 p += 3; /* Remove "/..", leave "/"
669                                          * for next pass. */
670                         } else
671                                 p += 1; /* Remove "/". */
672                 }
673         } while (rp != p);
674         return (p);
675 }
676
677 /*
678  * This is used by both out mode (to copy objects from disk into
679  * an archive) and pass mode (to copy objects from disk to
680  * an archive_write_disk "archive").
681  */
682 static int
683 file_to_archive(struct cpio *cpio, const char *srcpath)
684 {
685         const char *destpath;
686         struct archive_entry *entry, *spare;
687         size_t len;
688         int r;
689
690         /*
691          * Create an archive_entry describing the source file.
692          *
693          */
694         entry = archive_entry_new();
695         if (entry == NULL)
696                 lafe_errc(1, 0, "Couldn't allocate entry");
697         archive_entry_copy_sourcepath(entry, srcpath);
698         r = archive_read_disk_entry_from_file(cpio->archive_read_disk,
699             entry, -1, NULL);
700         if (r < ARCHIVE_FAILED)
701                 lafe_errc(1, 0, "%s",
702                     archive_error_string(cpio->archive_read_disk));
703         if (r < ARCHIVE_OK)
704                 lafe_warnc(0, "%s",
705                     archive_error_string(cpio->archive_read_disk));
706         if (r <= ARCHIVE_FAILED) {
707                 archive_entry_free(entry);
708                 cpio->return_value = 1;
709                 return (r);
710         }
711
712         if (cpio->uid_override >= 0) {
713                 archive_entry_set_uid(entry, cpio->uid_override);
714                 archive_entry_set_uname(entry, cpio->uname_override);
715         }
716         if (cpio->gid_override >= 0) {
717                 archive_entry_set_gid(entry, cpio->gid_override);
718                 archive_entry_set_gname(entry, cpio->gname_override);
719         }
720
721         /*
722          * Generate a destination path for this entry.
723          * "destination path" is the name to which it will be copied in
724          * pass mode or the name that will go into the archive in
725          * output mode.
726          */
727         destpath = srcpath;
728         if (cpio->destdir) {
729                 len = strlen(cpio->destdir) + strlen(srcpath) + 8;
730                 if (len >= cpio->pass_destpath_alloc) {
731                         while (len >= cpio->pass_destpath_alloc) {
732                                 cpio->pass_destpath_alloc += 512;
733                                 cpio->pass_destpath_alloc *= 2;
734                         }
735                         free(cpio->pass_destpath);
736                         cpio->pass_destpath = malloc(cpio->pass_destpath_alloc);
737                         if (cpio->pass_destpath == NULL)
738                                 lafe_errc(1, ENOMEM,
739                                     "Can't allocate path buffer");
740                 }
741                 strcpy(cpio->pass_destpath, cpio->destdir);
742                 strcat(cpio->pass_destpath, remove_leading_slash(srcpath));
743                 destpath = cpio->pass_destpath;
744         }
745         if (cpio->option_rename)
746                 destpath = cpio_rename(destpath);
747         if (destpath == NULL)
748                 return (0);
749         archive_entry_copy_pathname(entry, destpath);
750
751         /*
752          * If we're trying to preserve hardlinks, match them here.
753          */
754         spare = NULL;
755         if (cpio->linkresolver != NULL
756             && archive_entry_filetype(entry) != AE_IFDIR) {
757                 archive_entry_linkify(cpio->linkresolver, &entry, &spare);
758         }
759
760         if (entry != NULL) {
761                 r = entry_to_archive(cpio, entry);
762                 archive_entry_free(entry);
763                 if (spare != NULL) {
764                         if (r == 0)
765                                 r = entry_to_archive(cpio, spare);
766                         archive_entry_free(spare);
767                 }
768         }
769         return (r);
770 }
771
772 static int
773 entry_to_archive(struct cpio *cpio, struct archive_entry *entry)
774 {
775         const char *destpath = archive_entry_pathname(entry);
776         const char *srcpath = archive_entry_sourcepath(entry);
777         int fd = -1;
778         ssize_t bytes_read;
779         int r;
780
781         /* Print out the destination name to the user. */
782         if (cpio->verbose)
783                 fprintf(stderr,"%s", destpath);
784         if (cpio->dot)
785                 fprintf(stderr, ".");
786
787         /*
788          * Option_link only makes sense in pass mode and for
789          * regular files.  Also note: if a link operation fails
790          * because of cross-device restrictions, we'll fall back
791          * to copy mode for that entry.
792          *
793          * TODO: Test other cpio implementations to see if they
794          * hard-link anything other than regular files here.
795          */
796         if (cpio->option_link
797             && archive_entry_filetype(entry) == AE_IFREG)
798         {
799                 struct archive_entry *t;
800                 /* Save the original entry in case we need it later. */
801                 t = archive_entry_clone(entry);
802                 if (t == NULL)
803                         lafe_errc(1, ENOMEM, "Can't create link");
804                 /* Note: link(2) doesn't create parent directories,
805                  * so we use archive_write_header() instead as a
806                  * convenience. */
807                 archive_entry_set_hardlink(t, srcpath);
808                 /* This is a straight link that carries no data. */
809                 archive_entry_set_size(t, 0);
810                 r = archive_write_header(cpio->archive, t);
811                 archive_entry_free(t);
812                 if (r != ARCHIVE_OK)
813                         lafe_warnc(archive_errno(cpio->archive),
814                             "%s", archive_error_string(cpio->archive));
815                 if (r == ARCHIVE_FATAL)
816                         exit(1);
817 #ifdef EXDEV
818                 if (r != ARCHIVE_OK && archive_errno(cpio->archive) == EXDEV) {
819                         /* Cross-device link:  Just fall through and use
820                          * the original entry to copy the file over. */
821                         lafe_warnc(0, "Copying file instead");
822                 } else
823 #endif
824                 return (0);
825         }
826
827         /*
828          * Make sure we can open the file (if necessary) before
829          * trying to write the header.
830          */
831         if (archive_entry_filetype(entry) == AE_IFREG) {
832                 if (archive_entry_size(entry) > 0) {
833                         fd = open(srcpath, O_RDONLY | O_BINARY);
834                         if (fd < 0) {
835                                 lafe_warnc(errno,
836                                     "%s: could not open file", srcpath);
837                                 goto cleanup;
838                         }
839                 }
840         } else {
841                 archive_entry_set_size(entry, 0);
842         }
843
844         r = archive_write_header(cpio->archive, entry);
845
846         if (r != ARCHIVE_OK)
847                 lafe_warnc(archive_errno(cpio->archive),
848                     "%s: %s",
849                     srcpath,
850                     archive_error_string(cpio->archive));
851
852         if (r == ARCHIVE_FATAL)
853                 exit(1);
854
855         if (r >= ARCHIVE_WARN && archive_entry_size(entry) > 0 && fd >= 0) {
856                 bytes_read = read(fd, cpio->buff, (unsigned)cpio->buff_size);
857                 while (bytes_read > 0) {
858                         ssize_t bytes_write;
859                         bytes_write = archive_write_data(cpio->archive,
860                             cpio->buff, bytes_read);
861                         if (bytes_write < 0)
862                                 lafe_errc(1, archive_errno(cpio->archive),
863                                     "%s", archive_error_string(cpio->archive));
864                         if (bytes_write < bytes_read) {
865                                 lafe_warnc(0,
866                                     "Truncated write; file may have "
867                                     "grown while being archived.");
868                         }
869                         bytes_read = read(fd, cpio->buff,
870                             (unsigned)cpio->buff_size);
871                 }
872         }
873
874         fd = restore_time(cpio, entry, srcpath, fd);
875
876 cleanup:
877         if (cpio->verbose)
878                 fprintf(stderr,"\n");
879         if (fd >= 0)
880                 close(fd);
881         return (0);
882 }
883
884 static int
885 restore_time(struct cpio *cpio, struct archive_entry *entry,
886     const char *name, int fd)
887 {
888 #ifndef HAVE_UTIMES
889         static int warned = 0;
890
891         (void)cpio; /* UNUSED */
892         (void)entry; /* UNUSED */
893         (void)name; /* UNUSED */
894
895         if (!warned)
896                 lafe_warnc(0, "Can't restore access times on this platform");
897         warned = 1;
898         return (fd);
899 #else
900 #if defined(_WIN32) && !defined(__CYGWIN__)
901         struct __timeval times[2];
902 #else
903         struct timeval times[2];
904 #endif
905
906         if (!cpio->option_atime_restore)
907                 return (fd);
908
909         times[1].tv_sec = archive_entry_mtime(entry);
910         times[1].tv_usec = archive_entry_mtime_nsec(entry) / 1000;
911
912         times[0].tv_sec = archive_entry_atime(entry);
913         times[0].tv_usec = archive_entry_atime_nsec(entry) / 1000;
914
915 #if defined(HAVE_FUTIMES) && !defined(__CYGWIN__)
916         if (fd >= 0 && futimes(fd, times) == 0)
917                 return (fd);
918 #endif
919         /*
920          * Some platform cannot restore access times if the file descriptor
921          * is still opened.
922          */
923         if (fd >= 0) {
924                 close(fd);
925                 fd = -1;
926         }
927
928 #ifdef HAVE_LUTIMES
929         if (lutimes(name, times) != 0)
930 #else
931         if ((AE_IFLNK != archive_entry_filetype(entry))
932                         && utimes(name, times) != 0)
933 #endif
934                 lafe_warnc(errno, "Can't update time for %s", name);
935 #endif
936         return (fd);
937 }
938
939
940 static void
941 mode_in(struct cpio *cpio)
942 {
943         struct archive *a;
944         struct archive_entry *entry;
945         struct archive *ext;
946         const char *destpath;
947         int r;
948
949         ext = archive_write_disk_new();
950         if (ext == NULL)
951                 lafe_errc(1, 0, "Couldn't allocate restore object");
952         r = archive_write_disk_set_options(ext, cpio->extract_flags);
953         if (r != ARCHIVE_OK)
954                 lafe_errc(1, 0, "%s", archive_error_string(ext));
955         a = archive_read_new();
956         if (a == NULL)
957                 lafe_errc(1, 0, "Couldn't allocate archive object");
958         archive_read_support_filter_all(a);
959         archive_read_support_format_all(a);
960         if (cpio->passphrase != NULL)
961                 r = archive_read_add_passphrase(a, cpio->passphrase);
962         else
963                 r = archive_read_set_passphrase_callback(a, cpio,
964                         &passphrase_callback);
965         if (r != ARCHIVE_OK)
966                 lafe_errc(1, 0, "%s", archive_error_string(a));
967
968         if (archive_read_open_filename(a, cpio->filename,
969                                         cpio->bytes_per_block))
970                 lafe_errc(1, archive_errno(a),
971                     "%s", archive_error_string(a));
972         for (;;) {
973                 r = archive_read_next_header(a, &entry);
974                 if (r == ARCHIVE_EOF)
975                         break;
976                 if (r != ARCHIVE_OK) {
977                         lafe_errc(1, archive_errno(a),
978                             "%s", archive_error_string(a));
979                 }
980                 if (archive_match_path_excluded(cpio->matching, entry))
981                         continue;
982                 if (cpio->option_rename) {
983                         destpath = cpio_rename(archive_entry_pathname(entry));
984                         archive_entry_set_pathname(entry, destpath);
985                 } else
986                         destpath = archive_entry_pathname(entry);
987                 if (destpath == NULL)
988                         continue;
989                 if (cpio->verbose)
990                         fprintf(stderr, "%s\n", destpath);
991                 if (cpio->dot)
992                         fprintf(stderr, ".");
993                 if (cpio->uid_override >= 0)
994                         archive_entry_set_uid(entry, cpio->uid_override);
995                 if (cpio->gid_override >= 0)
996                         archive_entry_set_gid(entry, cpio->gid_override);
997                 r = archive_write_header(ext, entry);
998                 if (r != ARCHIVE_OK) {
999                         fprintf(stderr, "%s: %s\n",
1000                             archive_entry_pathname(entry),
1001                             archive_error_string(ext));
1002                 } else if (!archive_entry_size_is_set(entry)
1003                     || archive_entry_size(entry) > 0) {
1004                         r = extract_data(a, ext);
1005                         if (r != ARCHIVE_OK)
1006                                 cpio->return_value = 1;
1007                 }
1008         }
1009         r = archive_read_close(a);
1010         if (cpio->dot)
1011                 fprintf(stderr, "\n");
1012         if (r != ARCHIVE_OK)
1013                 lafe_errc(1, 0, "%s", archive_error_string(a));
1014         r = archive_write_close(ext);
1015         if (r != ARCHIVE_OK)
1016                 lafe_errc(1, 0, "%s", archive_error_string(ext));
1017         if (!cpio->quiet) {
1018                 int64_t blocks = (archive_filter_bytes(a, 0) + 511)
1019                               / 512;
1020                 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1021                     blocks == 1 ? "block" : "blocks");
1022         }
1023         archive_read_free(a);
1024         archive_write_free(ext);
1025         exit(cpio->return_value);
1026 }
1027
1028 /*
1029  * Exits if there's a fatal error.  Returns ARCHIVE_OK
1030  * if everything is kosher.
1031  */
1032 static int
1033 extract_data(struct archive *ar, struct archive *aw)
1034 {
1035         int r;
1036         size_t size;
1037         const void *block;
1038         int64_t offset;
1039
1040         for (;;) {
1041                 r = archive_read_data_block(ar, &block, &size, &offset);
1042                 if (r == ARCHIVE_EOF)
1043                         return (ARCHIVE_OK);
1044                 if (r != ARCHIVE_OK) {
1045                         lafe_warnc(archive_errno(ar),
1046                             "%s", archive_error_string(ar));
1047                         exit(1);
1048                 }
1049                 r = (int)archive_write_data_block(aw, block, size, offset);
1050                 if (r != ARCHIVE_OK) {
1051                         lafe_warnc(archive_errno(aw),
1052                             "%s", archive_error_string(aw));
1053                         return (r);
1054                 }
1055         }
1056 }
1057
1058 static void
1059 mode_list(struct cpio *cpio)
1060 {
1061         struct archive *a;
1062         struct archive_entry *entry;
1063         int r;
1064
1065         a = archive_read_new();
1066         if (a == NULL)
1067                 lafe_errc(1, 0, "Couldn't allocate archive object");
1068         archive_read_support_filter_all(a);
1069         archive_read_support_format_all(a);
1070         if (cpio->passphrase != NULL)
1071                 r = archive_read_add_passphrase(a, cpio->passphrase);
1072         else
1073                 r = archive_read_set_passphrase_callback(a, cpio,
1074                         &passphrase_callback);
1075         if (r != ARCHIVE_OK)
1076                 lafe_errc(1, 0, "%s", archive_error_string(a));
1077
1078         if (archive_read_open_filename(a, cpio->filename,
1079                                         cpio->bytes_per_block))
1080                 lafe_errc(1, archive_errno(a),
1081                     "%s", archive_error_string(a));
1082         for (;;) {
1083                 r = archive_read_next_header(a, &entry);
1084                 if (r == ARCHIVE_EOF)
1085                         break;
1086                 if (r != ARCHIVE_OK) {
1087                         lafe_errc(1, archive_errno(a),
1088                             "%s", archive_error_string(a));
1089                 }
1090                 if (archive_match_path_excluded(cpio->matching, entry))
1091                         continue;
1092                 if (cpio->verbose)
1093                         list_item_verbose(cpio, entry);
1094                 else
1095                         fprintf(stdout, "%s\n", archive_entry_pathname(entry));
1096         }
1097         r = archive_read_close(a);
1098         if (r != ARCHIVE_OK)
1099                 lafe_errc(1, 0, "%s", archive_error_string(a));
1100         if (!cpio->quiet) {
1101                 int64_t blocks = (archive_filter_bytes(a, 0) + 511)
1102                               / 512;
1103                 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1104                     blocks == 1 ? "block" : "blocks");
1105         }
1106         archive_read_free(a);
1107         exit(0);
1108 }
1109
1110 /*
1111  * Display information about the current file.
1112  *
1113  * The format here roughly duplicates the output of 'ls -l'.
1114  * This is based on SUSv2, where 'tar tv' is documented as
1115  * listing additional information in an "unspecified format,"
1116  * and 'pax -l' is documented as using the same format as 'ls -l'.
1117  */
1118 static void
1119 list_item_verbose(struct cpio *cpio, struct archive_entry *entry)
1120 {
1121         char                     size[32];
1122         char                     date[32];
1123         char                     uids[16], gids[16];
1124         const char              *uname, *gname;
1125         FILE                    *out = stdout;
1126         const char              *fmt;
1127         time_t                   mtime;
1128         static time_t            now;
1129
1130         if (!now)
1131                 time(&now);
1132
1133         if (cpio->option_numeric_uid_gid) {
1134                 /* Format numeric uid/gid for display. */
1135                 strcpy(uids, cpio_i64toa(archive_entry_uid(entry)));
1136                 uname = uids;
1137                 strcpy(gids, cpio_i64toa(archive_entry_gid(entry)));
1138                 gname = gids;
1139         } else {
1140                 /* Use uname if it's present, else lookup name from uid. */
1141                 uname = archive_entry_uname(entry);
1142                 if (uname == NULL)
1143                         uname = lookup_uname(cpio, (uid_t)archive_entry_uid(entry));
1144                 /* Use gname if it's present, else lookup name from gid. */
1145                 gname = archive_entry_gname(entry);
1146                 if (gname == NULL)
1147                         gname = lookup_gname(cpio, (uid_t)archive_entry_gid(entry));
1148         }
1149
1150         /* Print device number or file size. */
1151         if (archive_entry_filetype(entry) == AE_IFCHR
1152             || archive_entry_filetype(entry) == AE_IFBLK) {
1153                 snprintf(size, sizeof(size), "%lu,%lu",
1154                     (unsigned long)archive_entry_rdevmajor(entry),
1155                     (unsigned long)archive_entry_rdevminor(entry));
1156         } else {
1157                 strcpy(size, cpio_i64toa(archive_entry_size(entry)));
1158         }
1159
1160         /* Format the time using 'ls -l' conventions. */
1161         mtime = archive_entry_mtime(entry);
1162 #if defined(_WIN32) && !defined(__CYGWIN__)
1163         /* Windows' strftime function does not support %e format. */
1164         if (mtime - now > 365*86400/2
1165                 || mtime - now < -365*86400/2)
1166                 fmt = cpio->day_first ? "%d %b  %Y" : "%b %d  %Y";
1167         else
1168                 fmt = cpio->day_first ? "%d %b %H:%M" : "%b %d %H:%M";
1169 #else
1170         if (mtime - now > 365*86400/2
1171                 || mtime - now < -365*86400/2)
1172                 fmt = cpio->day_first ? "%e %b  %Y" : "%b %e  %Y";
1173         else
1174                 fmt = cpio->day_first ? "%e %b %H:%M" : "%b %e %H:%M";
1175 #endif
1176         strftime(date, sizeof(date), fmt, localtime(&mtime));
1177
1178         fprintf(out, "%s%3d %-8s %-8s %8s %12s %s",
1179             archive_entry_strmode(entry),
1180             archive_entry_nlink(entry),
1181             uname, gname, size, date,
1182             archive_entry_pathname(entry));
1183
1184         /* Extra information for links. */
1185         if (archive_entry_hardlink(entry)) /* Hard link */
1186                 fprintf(out, " link to %s", archive_entry_hardlink(entry));
1187         else if (archive_entry_symlink(entry)) /* Symbolic link */
1188                 fprintf(out, " -> %s", archive_entry_symlink(entry));
1189         fprintf(out, "\n");
1190 }
1191
1192 static void
1193 mode_pass(struct cpio *cpio, const char *destdir)
1194 {
1195         struct lafe_line_reader *lr;
1196         const char *p;
1197         int r;
1198         size_t destdir_len;
1199
1200         /* Ensure target dir has a trailing '/' to simplify path surgery. */
1201         destdir_len = strlen(destdir);
1202         cpio->destdir = malloc(destdir_len + 8);
1203         memcpy(cpio->destdir, destdir, destdir_len);
1204         if (destdir_len == 0 || destdir[destdir_len - 1] != '/')
1205                 cpio->destdir[destdir_len++] = '/';
1206         cpio->destdir[destdir_len++] = '\0';
1207
1208         cpio->archive = archive_write_disk_new();
1209         if (cpio->archive == NULL)
1210                 lafe_errc(1, 0, "Failed to allocate archive object");
1211         r = archive_write_disk_set_options(cpio->archive, cpio->extract_flags);
1212         if (r != ARCHIVE_OK)
1213                 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1214         cpio->linkresolver = archive_entry_linkresolver_new();
1215         archive_write_disk_set_standard_lookup(cpio->archive);
1216
1217         cpio->archive_read_disk = archive_read_disk_new();
1218         if (cpio->archive_read_disk == NULL)
1219                 lafe_errc(1, 0, "Failed to allocate archive object");
1220         if (cpio->option_follow_links)
1221                 archive_read_disk_set_symlink_logical(cpio->archive_read_disk);
1222         else
1223                 archive_read_disk_set_symlink_physical(cpio->archive_read_disk);
1224         archive_read_disk_set_standard_lookup(cpio->archive_read_disk);
1225
1226         lr = lafe_line_reader("-", cpio->option_null);
1227         while ((p = lafe_line_reader_next(lr)) != NULL)
1228                 file_to_archive(cpio, p);
1229         lafe_line_reader_free(lr);
1230
1231         archive_entry_linkresolver_free(cpio->linkresolver);
1232         r = archive_write_close(cpio->archive);
1233         if (cpio->dot)
1234                 fprintf(stderr, "\n");
1235         if (r != ARCHIVE_OK)
1236                 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1237
1238         if (!cpio->quiet) {
1239                 int64_t blocks =
1240                         (archive_filter_bytes(cpio->archive, 0) + 511)
1241                         / 512;
1242                 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1243                     blocks == 1 ? "block" : "blocks");
1244         }
1245
1246         archive_write_free(cpio->archive);
1247         free(cpio->pass_destpath);
1248 }
1249
1250 /*
1251  * Prompt for a new name for this entry.  Returns a pointer to the
1252  * new name or NULL if the entry should not be copied.  This
1253  * implements the semantics defined in POSIX.1-1996, which specifies
1254  * that an input of '.' means the name should be unchanged.  GNU cpio
1255  * treats '.' as a literal new name.
1256  */
1257 static const char *
1258 cpio_rename(const char *name)
1259 {
1260         static char buff[1024];
1261         FILE *t;
1262         char *p, *ret;
1263 #if defined(_WIN32) && !defined(__CYGWIN__)
1264         FILE *to;
1265
1266         t = fopen("CONIN$", "r");
1267         if (t == NULL)
1268                 return (name);
1269         to = fopen("CONOUT$", "w");
1270         if (to == NULL) {
1271                 fclose(t);
1272                 return (name);
1273         }
1274         fprintf(to, "%s (Enter/./(new name))? ", name);
1275         fclose(to);
1276 #else
1277         t = fopen("/dev/tty", "r+");
1278         if (t == NULL)
1279                 return (name);
1280         fprintf(t, "%s (Enter/./(new name))? ", name);
1281         fflush(t);
1282 #endif
1283
1284         p = fgets(buff, sizeof(buff), t);
1285         fclose(t);
1286         if (p == NULL)
1287                 /* End-of-file is a blank line. */
1288                 return (NULL);
1289
1290         while (*p == ' ' || *p == '\t')
1291                 ++p;
1292         if (*p == '\n' || *p == '\0')
1293                 /* Empty line. */
1294                 return (NULL);
1295         if (*p == '.' && p[1] == '\n')
1296                 /* Single period preserves original name. */
1297                 return (name);
1298         ret = p;
1299         /* Trim the final newline. */
1300         while (*p != '\0' && *p != '\n')
1301                 ++p;
1302         /* Overwrite the final \n with a null character. */
1303         *p = '\0';
1304         return (ret);
1305 }
1306
1307 static void
1308 free_cache(struct name_cache *cache)
1309 {
1310         size_t i;
1311
1312         if (cache != NULL) {
1313                 for (i = 0; i < cache->size; i++)
1314                         free(cache->cache[i].name);
1315                 free(cache);
1316         }
1317 }
1318
1319 /*
1320  * Lookup uname/gname from uid/gid, return NULL if no match.
1321  */
1322 static const char *
1323 lookup_name(struct cpio *cpio, struct name_cache **name_cache_variable,
1324     int (*lookup_fn)(struct cpio *, const char **, id_t), id_t id)
1325 {
1326         char asnum[16];
1327         struct name_cache       *cache;
1328         const char *name;
1329         int slot;
1330
1331
1332         if (*name_cache_variable == NULL) {
1333                 *name_cache_variable = calloc(1, sizeof(struct name_cache));
1334                 if (*name_cache_variable == NULL)
1335                         lafe_errc(1, ENOMEM, "No more memory");
1336                 (*name_cache_variable)->size = name_cache_size;
1337         }
1338
1339         cache = *name_cache_variable;
1340         cache->probes++;
1341
1342         slot = id % cache->size;
1343         if (cache->cache[slot].name != NULL) {
1344                 if (cache->cache[slot].id == id) {
1345                         cache->hits++;
1346                         return (cache->cache[slot].name);
1347                 }
1348                 free(cache->cache[slot].name);
1349                 cache->cache[slot].name = NULL;
1350         }
1351
1352         if (lookup_fn(cpio, &name, id)) {
1353                 /* If lookup failed, format it as a number. */
1354                 snprintf(asnum, sizeof(asnum), "%u", (unsigned)id);
1355                 name = asnum;
1356         }
1357
1358         cache->cache[slot].name = strdup(name);
1359         if (cache->cache[slot].name != NULL) {
1360                 cache->cache[slot].id = id;
1361                 return (cache->cache[slot].name);
1362         }
1363
1364         /*
1365          * Conveniently, NULL marks an empty slot, so
1366          * if the strdup() fails, we've just failed to
1367          * cache it.  No recovery necessary.
1368          */
1369         return (NULL);
1370 }
1371
1372 static const char *
1373 lookup_uname(struct cpio *cpio, uid_t uid)
1374 {
1375         return (lookup_name(cpio, &cpio->uname_cache,
1376                     &lookup_uname_helper, (id_t)uid));
1377 }
1378
1379 static int
1380 lookup_uname_helper(struct cpio *cpio, const char **name, id_t id)
1381 {
1382         struct passwd   *pwent;
1383
1384         (void)cpio; /* UNUSED */
1385
1386         errno = 0;
1387         pwent = getpwuid((uid_t)id);
1388         if (pwent == NULL) {
1389                 if (errno && errno != ENOENT)
1390                         lafe_warnc(errno, "getpwuid(%s) failed",
1391                             cpio_i64toa((int64_t)id));
1392                 return 1;
1393         }
1394
1395         *name = pwent->pw_name;
1396         return 0;
1397 }
1398
1399 static const char *
1400 lookup_gname(struct cpio *cpio, gid_t gid)
1401 {
1402         return (lookup_name(cpio, &cpio->gname_cache,
1403                     &lookup_gname_helper, (id_t)gid));
1404 }
1405
1406 static int
1407 lookup_gname_helper(struct cpio *cpio, const char **name, id_t id)
1408 {
1409         struct group    *grent;
1410
1411         (void)cpio; /* UNUSED */
1412
1413         errno = 0;
1414         grent = getgrgid((gid_t)id);
1415         if (grent == NULL) {
1416                 if (errno && errno != ENOENT)
1417                         lafe_warnc(errno, "getgrgid(%s) failed",
1418                             cpio_i64toa((int64_t)id));
1419                 return 1;
1420         }
1421
1422         *name = grent->gr_name;
1423         return 0;
1424 }
1425
1426 /*
1427  * It would be nice to just use printf() for formatting large numbers,
1428  * but the compatibility problems are a big headache.  Hence the
1429  * following simple utility function.
1430  */
1431 const char *
1432 cpio_i64toa(int64_t n0)
1433 {
1434         /* 2^64 =~ 1.8 * 10^19, so 20 decimal digits suffice.
1435          * We also need 1 byte for '-' and 1 for '\0'.
1436          */
1437         static char buff[22];
1438         int64_t n = n0 < 0 ? -n0 : n0;
1439         char *p = buff + sizeof(buff);
1440
1441         *--p = '\0';
1442         do {
1443                 *--p = '0' + (int)(n % 10);
1444                 n /= 10;
1445         } while (n > 0);
1446         if (n0 < 0)
1447                 *--p = '-';
1448         return p;
1449 }
1450
1451 #define PPBUFF_SIZE 1024
1452 static const char *
1453 passphrase_callback(struct archive *a, void *_client_data)
1454 {
1455         struct cpio *cpio = (struct cpio *)_client_data;
1456         (void)a; /* UNUSED */
1457
1458         if (cpio->ppbuff == NULL) {
1459                 cpio->ppbuff = malloc(PPBUFF_SIZE);
1460                 if (cpio->ppbuff == NULL)
1461                         lafe_errc(1, errno, "Out of memory");
1462         }
1463         return lafe_readpassphrase("Enter passphrase:",
1464                 cpio->ppbuff, PPBUFF_SIZE);
1465 }
1466
1467 static void
1468 passphrase_free(char *ppbuff)
1469 {
1470         if (ppbuff != NULL) {
1471                 memset(ppbuff, 0, PPBUFF_SIZE);
1472                 free(ppbuff);
1473         }
1474 }