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) __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
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 case OPTION_ZSTD:
273 cpio->compress = opt;
274 break;
275 case 'm': /* POSIX 1997 */
276 cpio->extract_flags |= ARCHIVE_EXTRACT_TIME;
277 break;
278 case 'n': /* GNU cpio */
279 cpio->option_numeric_uid_gid = 1;
280 break;
281 case OPTION_NO_PRESERVE_OWNER: /* GNU cpio */
282 cpio->extract_flags &= ~ARCHIVE_EXTRACT_OWNER;
283 break;
284 case 'O': /* GNU cpio */
285 cpio->filename = cpio->argument;
286 break;
287 case 'o': /* POSIX 1997 */
288 if (cpio->mode != '\0')
289 lafe_errc(1, 0,
290 "Cannot use both -o and -%c", cpio->mode);
291 cpio->mode = opt;
292 break;
293 case 'p': /* POSIX 1997 */
294 if (cpio->mode != '\0')
295 lafe_errc(1, 0,
296 "Cannot use both -p and -%c", cpio->mode);
297 cpio->mode = opt;
298 cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_NODOTDOT;
299 cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_NOABSOLUTEPATHS;
300 break;
301 case OPTION_PASSPHRASE:
302 cpio->passphrase = cpio->argument;
303 break;
304 case OPTION_PRESERVE_OWNER:
305 cpio->extract_flags |= ARCHIVE_EXTRACT_OWNER;
306 break;
307 case OPTION_QUIET: /* GNU cpio */
308 cpio->quiet = 1;
309 break;
310 case 'R': /* GNU cpio, also --owner */
311 /* TODO: owner_parse should return uname/gname
312 * also; use that to set [ug]name_override. */
313 errmsg = owner_parse(cpio->argument, &uid, &gid);
314 if (errmsg) {
315 lafe_warnc(-1, "%s", errmsg);
316 usage();
317 }
318 if (uid != -1) {
319 cpio->uid_override = uid;
320 cpio->uname_override = NULL;
321 }
322 if (gid != -1) {
323 cpio->gid_override = gid;
324 cpio->gname_override = NULL;
325 }
326 break;
327 case 'r': /* POSIX 1997 */
328 cpio->option_rename = 1;
329 break;
330 case 't': /* POSIX 1997 */
331 cpio->option_list = 1;
332 break;
333 case 'u': /* POSIX 1997 */
334 cpio->extract_flags
335 &= ~ARCHIVE_EXTRACT_NO_OVERWRITE_NEWER;
336 break;
337 case OPTION_UUENCODE:
338 cpio->add_filter = opt;
339 break;
340 case 'v': /* POSIX 1997 */
341 cpio->verbose++;
342 break;
343 case 'V': /* GNU cpio */
344 cpio->dot++;
345 break;
346 case OPTION_VERSION: /* GNU convention */
347 version();
348 break;
349 #if 0
350 /*
351 * cpio_getopt() handles -W specially, so it's not
352 * available here.
353 */
354 case 'W': /* Obscure, but useful GNU convention. */
355 break;
356 #endif
357 case 'y': /* tar convention */
358 cpio->compress = opt;
359 break;
360 case 'Z': /* tar convention */
361 cpio->compress = opt;
362 break;
363 case 'z': /* tar convention */
364 cpio->compress = opt;
365 break;
366 default:
367 usage();
368 }
369 }
370
371 /*
372 * Sanity-check args, error out on nonsensical combinations.
373 */
374 /* -t implies -i if no mode was specified. */
375 if (cpio->option_list && cpio->mode == '\0')
376 cpio->mode = 'i';
377 /* -t requires -i */
378 if (cpio->option_list && cpio->mode != 'i')
379 lafe_errc(1, 0, "Option -t requires -i");
380 /* -n requires -it */
381 if (cpio->option_numeric_uid_gid && !cpio->option_list)
382 lafe_errc(1, 0, "Option -n requires -it");
383 /* Can only specify format when writing */
384 if (cpio->format != NULL && cpio->mode != 'o')
385 lafe_errc(1, 0, "Option --format requires -o");
386 /* -l requires -p */
387 if (cpio->option_link && cpio->mode != 'p')
388 lafe_errc(1, 0, "Option -l requires -p");
389 /* -v overrides -V */
390 if (cpio->dot && cpio->verbose)
391 cpio->dot = 0;
392 /* TODO: Flag other nonsensical combinations. */
393
394 switch (cpio->mode) {
395 case 'o':
396 /* TODO: Implement old binary format in libarchive,
397 use that here. */
398 if (cpio->format == NULL)
399 cpio->format = "odc"; /* Default format */
400
401 mode_out(cpio);
402 break;
403 case 'i':
404 while (*cpio->argv != NULL) {
405 if (archive_match_include_pattern(cpio->matching,
406 *cpio->argv) != ARCHIVE_OK)
407 lafe_errc(1, 0, "Error : %s",
408 archive_error_string(cpio->matching));
409 --cpio->argc;
410 ++cpio->argv;
411 }
412 if (cpio->option_list)
413 mode_list(cpio);
414 else
415 mode_in(cpio);
416 break;
417 case 'p':
418 if (*cpio->argv == NULL || **cpio->argv == '\0')
419 lafe_errc(1, 0,
420 "-p mode requires a target directory");
421 mode_pass(cpio, *cpio->argv);
422 break;
423 default:
424 lafe_errc(1, 0,
425 "Must specify at least one of -i, -o, or -p");
426 }
427
428 archive_match_free(cpio->matching);
429 free_cache(cpio->gname_cache);
430 free_cache(cpio->uname_cache);
431 free(cpio->destdir);
432 passphrase_free(cpio->ppbuff);
433 return (cpio->return_value);
434 }
435
436 static void
usage(void)437 usage(void)
438 {
439 const char *p;
440
441 p = lafe_getprogname();
442
443 fprintf(stderr, "Brief Usage:\n");
444 fprintf(stderr, " List: %s -it < archive\n", p);
445 fprintf(stderr, " Extract: %s -i < archive\n", p);
446 fprintf(stderr, " Create: %s -o < filenames > archive\n", p);
447 fprintf(stderr, " Help: %s --help\n", p);
448 exit(1);
449 }
450
451 static const char *long_help_msg =
452 "First option must be a mode specifier:\n"
453 " -i Input -o Output -p Pass\n"
454 "Common Options:\n"
455 " -v Verbose filenames -V one dot per file\n"
456 "Create: %p -o [options] < [list of files] > [archive]\n"
457 " -J,-y,-z,--lzma Compress archive with xz/bzip2/gzip/lzma\n"
458 " --format {odc|newc|ustar} Select archive format\n"
459 "List: %p -it < [archive]\n"
460 "Extract: %p -i [options] < [archive]\n";
461
462
463 /*
464 * Note that the word 'bsdcpio' will always appear in the first line
465 * of output.
466 *
467 * In particular, /bin/sh scripts that need to test for the presence
468 * of bsdcpio can use the following template:
469 *
470 * if (cpio --help 2>&1 | grep bsdcpio >/dev/null 2>&1 ) then \
471 * echo bsdcpio; else echo not bsdcpio; fi
472 */
473 static void
long_help(void)474 long_help(void)
475 {
476 const char *prog;
477 const char *p;
478
479 prog = lafe_getprogname();
480
481 fflush(stderr);
482
483 p = (strcmp(prog,"bsdcpio") != 0) ? "(bsdcpio)" : "";
484 printf("%s%s: manipulate archive files\n", prog, p);
485
486 for (p = long_help_msg; *p != '\0'; p++) {
487 if (*p == '%') {
488 if (p[1] == 'p') {
489 fputs(prog, stdout);
490 p++;
491 } else
492 putchar('%');
493 } else
494 putchar(*p);
495 }
496 version();
497 }
498
499 static void
version(void)500 version(void)
501 {
502 fprintf(stdout,"bsdcpio %s - %s \n",
503 BSDCPIO_VERSION_STRING,
504 archive_version_details());
505 exit(0);
506 }
507
508 static void
mode_out(struct cpio * cpio)509 mode_out(struct cpio *cpio)
510 {
511 struct archive_entry *entry, *spare;
512 struct lafe_line_reader *lr;
513 const char *p;
514 int r;
515
516 if (cpio->option_append)
517 lafe_errc(1, 0, "Append mode not yet supported.");
518
519 cpio->archive_read_disk = archive_read_disk_new();
520 if (cpio->archive_read_disk == NULL)
521 lafe_errc(1, 0, "Failed to allocate archive object");
522 if (cpio->option_follow_links)
523 archive_read_disk_set_symlink_logical(cpio->archive_read_disk);
524 else
525 archive_read_disk_set_symlink_physical(cpio->archive_read_disk);
526 archive_read_disk_set_standard_lookup(cpio->archive_read_disk);
527
528 cpio->archive = archive_write_new();
529 if (cpio->archive == NULL)
530 lafe_errc(1, 0, "Failed to allocate archive object");
531 switch (cpio->compress) {
532 case OPTION_GRZIP:
533 r = archive_write_add_filter_grzip(cpio->archive);
534 break;
535 case 'J':
536 r = archive_write_add_filter_xz(cpio->archive);
537 break;
538 case OPTION_LRZIP:
539 r = archive_write_add_filter_lrzip(cpio->archive);
540 break;
541 case OPTION_LZ4:
542 r = archive_write_add_filter_lz4(cpio->archive);
543 break;
544 case OPTION_LZMA:
545 r = archive_write_add_filter_lzma(cpio->archive);
546 break;
547 case OPTION_LZOP:
548 r = archive_write_add_filter_lzop(cpio->archive);
549 break;
550 case OPTION_ZSTD:
551 r = archive_write_add_filter_zstd(cpio->archive);
552 break;
553 case 'j': case 'y':
554 r = archive_write_add_filter_bzip2(cpio->archive);
555 break;
556 case 'z':
557 r = archive_write_add_filter_gzip(cpio->archive);
558 break;
559 case 'Z':
560 r = archive_write_add_filter_compress(cpio->archive);
561 break;
562 default:
563 r = archive_write_add_filter_none(cpio->archive);
564 break;
565 }
566 if (r < ARCHIVE_WARN)
567 lafe_errc(1, 0, "Requested compression not available");
568 switch (cpio->add_filter) {
569 case 0:
570 r = ARCHIVE_OK;
571 break;
572 case OPTION_B64ENCODE:
573 r = archive_write_add_filter_b64encode(cpio->archive);
574 break;
575 case OPTION_UUENCODE:
576 r = archive_write_add_filter_uuencode(cpio->archive);
577 break;
578 }
579 if (r < ARCHIVE_WARN)
580 lafe_errc(1, 0, "Requested filter not available");
581 r = archive_write_set_format_by_name(cpio->archive, cpio->format);
582 if (r != ARCHIVE_OK)
583 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
584 archive_write_set_bytes_per_block(cpio->archive, cpio->bytes_per_block);
585 cpio->linkresolver = archive_entry_linkresolver_new();
586 archive_entry_linkresolver_set_strategy(cpio->linkresolver,
587 archive_format(cpio->archive));
588 if (cpio->passphrase != NULL)
589 r = archive_write_set_passphrase(cpio->archive,
590 cpio->passphrase);
591 else
592 r = archive_write_set_passphrase_callback(cpio->archive, cpio,
593 &passphrase_callback);
594 if (r != ARCHIVE_OK)
595 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
596
597 /*
598 * The main loop: Copy each file into the output archive.
599 */
600 r = archive_write_open_filename(cpio->archive, cpio->filename);
601 if (r != ARCHIVE_OK)
602 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
603 lr = lafe_line_reader("-", cpio->option_null);
604 while ((p = lafe_line_reader_next(lr)) != NULL)
605 file_to_archive(cpio, p);
606 lafe_line_reader_free(lr);
607
608 /*
609 * The hardlink detection may have queued up a couple of entries
610 * that can now be flushed.
611 */
612 entry = NULL;
613 archive_entry_linkify(cpio->linkresolver, &entry, &spare);
614 while (entry != NULL) {
615 entry_to_archive(cpio, entry);
616 archive_entry_free(entry);
617 entry = NULL;
618 archive_entry_linkify(cpio->linkresolver, &entry, &spare);
619 }
620
621 r = archive_write_close(cpio->archive);
622 if (cpio->dot)
623 fprintf(stderr, "\n");
624 if (r != ARCHIVE_OK)
625 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
626
627 if (!cpio->quiet) {
628 int64_t blocks =
629 (archive_filter_bytes(cpio->archive, 0) + 511)
630 / 512;
631 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
632 blocks == 1 ? "block" : "blocks");
633 }
634 archive_write_free(cpio->archive);
635 archive_entry_linkresolver_free(cpio->linkresolver);
636 }
637
638 static const char *
remove_leading_slash(const char * p)639 remove_leading_slash(const char *p)
640 {
641 const char *rp;
642
643 /* Remove leading "//./" or "//?/" or "//?/UNC/"
644 * (absolute path prefixes used by Windows API) */
645 if ((p[0] == '/' || p[0] == '\\') &&
646 (p[1] == '/' || p[1] == '\\') &&
647 (p[2] == '.' || p[2] == '?') &&
648 (p[3] == '/' || p[3] == '\\'))
649 {
650 if (p[2] == '?' &&
651 (p[4] == 'U' || p[4] == 'u') &&
652 (p[5] == 'N' || p[5] == 'n') &&
653 (p[6] == 'C' || p[6] == 'c') &&
654 (p[7] == '/' || p[7] == '\\'))
655 p += 8;
656 else
657 p += 4;
658 }
659 do {
660 rp = p;
661 /* Remove leading drive letter from archives created
662 * on Windows. */
663 if (((p[0] >= 'a' && p[0] <= 'z') ||
664 (p[0] >= 'A' && p[0] <= 'Z')) &&
665 p[1] == ':') {
666 p += 2;
667 }
668 /* Remove leading "/../", "//", etc. */
669 while (p[0] == '/' || p[0] == '\\') {
670 if (p[1] == '.' && p[2] == '.' &&
671 (p[3] == '/' || p[3] == '\\')) {
672 p += 3; /* Remove "/..", leave "/"
673 * for next pass. */
674 } else
675 p += 1; /* Remove "/". */
676 }
677 } while (rp != p);
678 return (p);
679 }
680
681 /*
682 * This is used by both out mode (to copy objects from disk into
683 * an archive) and pass mode (to copy objects from disk to
684 * an archive_write_disk "archive").
685 */
686 static int
file_to_archive(struct cpio * cpio,const char * srcpath)687 file_to_archive(struct cpio *cpio, const char *srcpath)
688 {
689 const char *destpath;
690 struct archive_entry *entry, *spare;
691 size_t len;
692 int r;
693
694 /*
695 * Create an archive_entry describing the source file.
696 *
697 */
698 entry = archive_entry_new();
699 if (entry == NULL)
700 lafe_errc(1, 0, "Couldn't allocate entry");
701 archive_entry_copy_sourcepath(entry, srcpath);
702 r = archive_read_disk_entry_from_file(cpio->archive_read_disk,
703 entry, -1, NULL);
704 if (r < ARCHIVE_FAILED)
705 lafe_errc(1, 0, "%s",
706 archive_error_string(cpio->archive_read_disk));
707 if (r < ARCHIVE_OK)
708 lafe_warnc(0, "%s",
709 archive_error_string(cpio->archive_read_disk));
710 if (r <= ARCHIVE_FAILED) {
711 archive_entry_free(entry);
712 cpio->return_value = 1;
713 return (r);
714 }
715
716 if (cpio->uid_override >= 0) {
717 archive_entry_set_uid(entry, cpio->uid_override);
718 archive_entry_set_uname(entry, cpio->uname_override);
719 }
720 if (cpio->gid_override >= 0) {
721 archive_entry_set_gid(entry, cpio->gid_override);
722 archive_entry_set_gname(entry, cpio->gname_override);
723 }
724
725 /*
726 * Generate a destination path for this entry.
727 * "destination path" is the name to which it will be copied in
728 * pass mode or the name that will go into the archive in
729 * output mode.
730 */
731 destpath = srcpath;
732 if (cpio->destdir) {
733 len = strlen(cpio->destdir) + strlen(srcpath) + 8;
734 if (len >= cpio->pass_destpath_alloc) {
735 while (len >= cpio->pass_destpath_alloc) {
736 cpio->pass_destpath_alloc += 512;
737 cpio->pass_destpath_alloc *= 2;
738 }
739 free(cpio->pass_destpath);
740 cpio->pass_destpath = malloc(cpio->pass_destpath_alloc);
741 if (cpio->pass_destpath == NULL)
742 lafe_errc(1, ENOMEM,
743 "Can't allocate path buffer");
744 }
745 strcpy(cpio->pass_destpath, cpio->destdir);
746 strcat(cpio->pass_destpath, remove_leading_slash(srcpath));
747 destpath = cpio->pass_destpath;
748 }
749 if (cpio->option_rename)
750 destpath = cpio_rename(destpath);
751 if (destpath == NULL)
752 return (0);
753 archive_entry_copy_pathname(entry, destpath);
754
755 /*
756 * If we're trying to preserve hardlinks, match them here.
757 */
758 spare = NULL;
759 if (cpio->linkresolver != NULL
760 && archive_entry_filetype(entry) != AE_IFDIR) {
761 archive_entry_linkify(cpio->linkresolver, &entry, &spare);
762 }
763
764 if (entry != NULL) {
765 r = entry_to_archive(cpio, entry);
766 archive_entry_free(entry);
767 if (spare != NULL) {
768 if (r == 0)
769 r = entry_to_archive(cpio, spare);
770 archive_entry_free(spare);
771 }
772 }
773 return (r);
774 }
775
776 static int
entry_to_archive(struct cpio * cpio,struct archive_entry * entry)777 entry_to_archive(struct cpio *cpio, struct archive_entry *entry)
778 {
779 const char *destpath = archive_entry_pathname(entry);
780 const char *srcpath = archive_entry_sourcepath(entry);
781 int fd = -1;
782 ssize_t bytes_read;
783 int r;
784
785 /* Print out the destination name to the user. */
786 if (cpio->verbose)
787 fprintf(stderr,"%s", destpath);
788 if (cpio->dot)
789 fprintf(stderr, ".");
790
791 /*
792 * Option_link only makes sense in pass mode and for
793 * regular files. Also note: if a link operation fails
794 * because of cross-device restrictions, we'll fall back
795 * to copy mode for that entry.
796 *
797 * TODO: Test other cpio implementations to see if they
798 * hard-link anything other than regular files here.
799 */
800 if (cpio->option_link
801 && archive_entry_filetype(entry) == AE_IFREG)
802 {
803 struct archive_entry *t;
804 /* Save the original entry in case we need it later. */
805 t = archive_entry_clone(entry);
806 if (t == NULL)
807 lafe_errc(1, ENOMEM, "Can't create link");
808 /* Note: link(2) doesn't create parent directories,
809 * so we use archive_write_header() instead as a
810 * convenience. */
811 archive_entry_set_hardlink(t, srcpath);
812 /* This is a straight link that carries no data. */
813 archive_entry_set_size(t, 0);
814 r = archive_write_header(cpio->archive, t);
815 archive_entry_free(t);
816 if (r != ARCHIVE_OK)
817 lafe_warnc(archive_errno(cpio->archive),
818 "%s", archive_error_string(cpio->archive));
819 if (r == ARCHIVE_FATAL)
820 exit(1);
821 #ifdef EXDEV
822 if (r != ARCHIVE_OK && archive_errno(cpio->archive) == EXDEV) {
823 /* Cross-device link: Just fall through and use
824 * the original entry to copy the file over. */
825 lafe_warnc(0, "Copying file instead");
826 } else
827 #endif
828 return (0);
829 }
830
831 /*
832 * Make sure we can open the file (if necessary) before
833 * trying to write the header.
834 */
835 if (archive_entry_filetype(entry) == AE_IFREG) {
836 if (archive_entry_size(entry) > 0) {
837 fd = open(srcpath, O_RDONLY | O_BINARY);
838 if (fd < 0) {
839 lafe_warnc(errno,
840 "%s: could not open file", srcpath);
841 goto cleanup;
842 }
843 }
844 } else {
845 archive_entry_set_size(entry, 0);
846 }
847
848 r = archive_write_header(cpio->archive, entry);
849
850 if (r != ARCHIVE_OK)
851 lafe_warnc(archive_errno(cpio->archive),
852 "%s: %s",
853 srcpath,
854 archive_error_string(cpio->archive));
855
856 if (r == ARCHIVE_FATAL)
857 exit(1);
858
859 if (r >= ARCHIVE_WARN && archive_entry_size(entry) > 0 && fd >= 0) {
860 bytes_read = read(fd, cpio->buff, (unsigned)cpio->buff_size);
861 while (bytes_read > 0) {
862 ssize_t bytes_write;
863 bytes_write = archive_write_data(cpio->archive,
864 cpio->buff, bytes_read);
865 if (bytes_write < 0)
866 lafe_errc(1, archive_errno(cpio->archive),
867 "%s", archive_error_string(cpio->archive));
868 if (bytes_write < bytes_read) {
869 lafe_warnc(0,
870 "Truncated write; file may have "
871 "grown while being archived.");
872 }
873 bytes_read = read(fd, cpio->buff,
874 (unsigned)cpio->buff_size);
875 }
876 }
877
878 fd = restore_time(cpio, entry, srcpath, fd);
879
880 cleanup:
881 if (cpio->verbose)
882 fprintf(stderr,"\n");
883 if (fd >= 0)
884 close(fd);
885 return (0);
886 }
887
888 static int
restore_time(struct cpio * cpio,struct archive_entry * entry,const char * name,int fd)889 restore_time(struct cpio *cpio, struct archive_entry *entry,
890 const char *name, int fd)
891 {
892 #ifndef HAVE_UTIMES
893 static int warned = 0;
894
895 (void)cpio; /* UNUSED */
896 (void)entry; /* UNUSED */
897 (void)name; /* UNUSED */
898
899 if (!warned)
900 lafe_warnc(0, "Can't restore access times on this platform");
901 warned = 1;
902 return (fd);
903 #else
904 #if defined(_WIN32) && !defined(__CYGWIN__)
905 struct __timeval times[2];
906 #else
907 struct timeval times[2];
908 #endif
909
910 if (!cpio->option_atime_restore)
911 return (fd);
912
913 times[1].tv_sec = archive_entry_mtime(entry);
914 times[1].tv_usec = archive_entry_mtime_nsec(entry) / 1000;
915
916 times[0].tv_sec = archive_entry_atime(entry);
917 times[0].tv_usec = archive_entry_atime_nsec(entry) / 1000;
918
919 #if defined(HAVE_FUTIMES) && !defined(__CYGWIN__)
920 if (fd >= 0 && futimes(fd, times) == 0)
921 return (fd);
922 #endif
923 /*
924 * Some platform cannot restore access times if the file descriptor
925 * is still opened.
926 */
927 if (fd >= 0) {
928 close(fd);
929 fd = -1;
930 }
931
932 #ifdef HAVE_LUTIMES
933 if (lutimes(name, times) != 0)
934 #else
935 if ((AE_IFLNK != archive_entry_filetype(entry))
936 && utimes(name, times) != 0)
937 #endif
938 lafe_warnc(errno, "Can't update time for %s", name);
939 #endif
940 return (fd);
941 }
942
943
944 static void
mode_in(struct cpio * cpio)945 mode_in(struct cpio *cpio)
946 {
947 struct archive *a;
948 struct archive_entry *entry;
949 struct archive *ext;
950 const char *destpath;
951 int r;
952
953 ext = archive_write_disk_new();
954 if (ext == NULL)
955 lafe_errc(1, 0, "Couldn't allocate restore object");
956 r = archive_write_disk_set_options(ext, cpio->extract_flags);
957 if (r != ARCHIVE_OK)
958 lafe_errc(1, 0, "%s", archive_error_string(ext));
959 a = archive_read_new();
960 if (a == NULL)
961 lafe_errc(1, 0, "Couldn't allocate archive object");
962 archive_read_support_filter_all(a);
963 archive_read_support_format_all(a);
964 if (cpio->passphrase != NULL)
965 r = archive_read_add_passphrase(a, cpio->passphrase);
966 else
967 r = archive_read_set_passphrase_callback(a, cpio,
968 &passphrase_callback);
969 if (r != ARCHIVE_OK)
970 lafe_errc(1, 0, "%s", archive_error_string(a));
971
972 if (archive_read_open_filename(a, cpio->filename,
973 cpio->bytes_per_block))
974 lafe_errc(1, archive_errno(a),
975 "%s", archive_error_string(a));
976 for (;;) {
977 r = archive_read_next_header(a, &entry);
978 if (r == ARCHIVE_EOF)
979 break;
980 if (r != ARCHIVE_OK) {
981 lafe_errc(1, archive_errno(a),
982 "%s", archive_error_string(a));
983 }
984 if (archive_match_path_excluded(cpio->matching, entry))
985 continue;
986 if (cpio->option_rename) {
987 destpath = cpio_rename(archive_entry_pathname(entry));
988 archive_entry_set_pathname(entry, destpath);
989 } else
990 destpath = archive_entry_pathname(entry);
991 if (destpath == NULL)
992 continue;
993 if (cpio->verbose)
994 fprintf(stderr, "%s\n", destpath);
995 if (cpio->dot)
996 fprintf(stderr, ".");
997 if (cpio->uid_override >= 0)
998 archive_entry_set_uid(entry, cpio->uid_override);
999 if (cpio->gid_override >= 0)
1000 archive_entry_set_gid(entry, cpio->gid_override);
1001 r = archive_write_header(ext, entry);
1002 if (r != ARCHIVE_OK) {
1003 fprintf(stderr, "%s: %s\n",
1004 archive_entry_pathname(entry),
1005 archive_error_string(ext));
1006 } else if (!archive_entry_size_is_set(entry)
1007 || archive_entry_size(entry) > 0) {
1008 r = extract_data(a, ext);
1009 if (r != ARCHIVE_OK)
1010 cpio->return_value = 1;
1011 }
1012 }
1013 r = archive_read_close(a);
1014 if (cpio->dot)
1015 fprintf(stderr, "\n");
1016 if (r != ARCHIVE_OK)
1017 lafe_errc(1, 0, "%s", archive_error_string(a));
1018 r = archive_write_close(ext);
1019 if (r != ARCHIVE_OK)
1020 lafe_errc(1, 0, "%s", archive_error_string(ext));
1021 if (!cpio->quiet) {
1022 int64_t blocks = (archive_filter_bytes(a, 0) + 511)
1023 / 512;
1024 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1025 blocks == 1 ? "block" : "blocks");
1026 }
1027 archive_read_free(a);
1028 archive_write_free(ext);
1029 exit(cpio->return_value);
1030 }
1031
1032 /*
1033 * Exits if there's a fatal error. Returns ARCHIVE_OK
1034 * if everything is kosher.
1035 */
1036 static int
extract_data(struct archive * ar,struct archive * aw)1037 extract_data(struct archive *ar, struct archive *aw)
1038 {
1039 int r;
1040 size_t size;
1041 const void *block;
1042 int64_t offset;
1043
1044 for (;;) {
1045 r = archive_read_data_block(ar, &block, &size, &offset);
1046 if (r == ARCHIVE_EOF)
1047 return (ARCHIVE_OK);
1048 if (r != ARCHIVE_OK) {
1049 lafe_warnc(archive_errno(ar),
1050 "%s", archive_error_string(ar));
1051 exit(1);
1052 }
1053 r = (int)archive_write_data_block(aw, block, size, offset);
1054 if (r != ARCHIVE_OK) {
1055 lafe_warnc(archive_errno(aw),
1056 "%s", archive_error_string(aw));
1057 return (r);
1058 }
1059 }
1060 }
1061
1062 static void
mode_list(struct cpio * cpio)1063 mode_list(struct cpio *cpio)
1064 {
1065 struct archive *a;
1066 struct archive_entry *entry;
1067 int r;
1068
1069 a = archive_read_new();
1070 if (a == NULL)
1071 lafe_errc(1, 0, "Couldn't allocate archive object");
1072 archive_read_support_filter_all(a);
1073 archive_read_support_format_all(a);
1074 if (cpio->passphrase != NULL)
1075 r = archive_read_add_passphrase(a, cpio->passphrase);
1076 else
1077 r = archive_read_set_passphrase_callback(a, cpio,
1078 &passphrase_callback);
1079 if (r != ARCHIVE_OK)
1080 lafe_errc(1, 0, "%s", archive_error_string(a));
1081
1082 if (archive_read_open_filename(a, cpio->filename,
1083 cpio->bytes_per_block))
1084 lafe_errc(1, archive_errno(a),
1085 "%s", archive_error_string(a));
1086 for (;;) {
1087 r = archive_read_next_header(a, &entry);
1088 if (r == ARCHIVE_EOF)
1089 break;
1090 if (r != ARCHIVE_OK) {
1091 lafe_errc(1, archive_errno(a),
1092 "%s", archive_error_string(a));
1093 }
1094 if (archive_match_path_excluded(cpio->matching, entry))
1095 continue;
1096 if (cpio->verbose)
1097 list_item_verbose(cpio, entry);
1098 else
1099 fprintf(stdout, "%s\n", archive_entry_pathname(entry));
1100 }
1101 r = archive_read_close(a);
1102 if (r != ARCHIVE_OK)
1103 lafe_errc(1, 0, "%s", archive_error_string(a));
1104 if (!cpio->quiet) {
1105 int64_t blocks = (archive_filter_bytes(a, 0) + 511)
1106 / 512;
1107 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1108 blocks == 1 ? "block" : "blocks");
1109 }
1110 archive_read_free(a);
1111 exit(0);
1112 }
1113
1114 /*
1115 * Display information about the current file.
1116 *
1117 * The format here roughly duplicates the output of 'ls -l'.
1118 * This is based on SUSv2, where 'tar tv' is documented as
1119 * listing additional information in an "unspecified format,"
1120 * and 'pax -l' is documented as using the same format as 'ls -l'.
1121 */
1122 static void
list_item_verbose(struct cpio * cpio,struct archive_entry * entry)1123 list_item_verbose(struct cpio *cpio, struct archive_entry *entry)
1124 {
1125 char size[32];
1126 char date[32];
1127 char uids[16], gids[16];
1128 const char *uname, *gname;
1129 FILE *out = stdout;
1130 const char *fmt;
1131 time_t mtime;
1132 static time_t now;
1133
1134 if (!now)
1135 time(&now);
1136
1137 if (cpio->option_numeric_uid_gid) {
1138 /* Format numeric uid/gid for display. */
1139 strcpy(uids, cpio_i64toa(archive_entry_uid(entry)));
1140 uname = uids;
1141 strcpy(gids, cpio_i64toa(archive_entry_gid(entry)));
1142 gname = gids;
1143 } else {
1144 /* Use uname if it's present, else lookup name from uid. */
1145 uname = archive_entry_uname(entry);
1146 if (uname == NULL)
1147 uname = lookup_uname(cpio, (uid_t)archive_entry_uid(entry));
1148 /* Use gname if it's present, else lookup name from gid. */
1149 gname = archive_entry_gname(entry);
1150 if (gname == NULL)
1151 gname = lookup_gname(cpio, (uid_t)archive_entry_gid(entry));
1152 }
1153
1154 /* Print device number or file size. */
1155 if (archive_entry_filetype(entry) == AE_IFCHR
1156 || archive_entry_filetype(entry) == AE_IFBLK) {
1157 snprintf(size, sizeof(size), "%lu,%lu",
1158 (unsigned long)archive_entry_rdevmajor(entry),
1159 (unsigned long)archive_entry_rdevminor(entry));
1160 } else {
1161 strcpy(size, cpio_i64toa(archive_entry_size(entry)));
1162 }
1163
1164 /* Format the time using 'ls -l' conventions. */
1165 mtime = archive_entry_mtime(entry);
1166 #if defined(_WIN32) && !defined(__CYGWIN__)
1167 /* Windows' strftime function does not support %e format. */
1168 if (mtime - now > 365*86400/2
1169 || mtime - now < -365*86400/2)
1170 fmt = cpio->day_first ? "%d %b %Y" : "%b %d %Y";
1171 else
1172 fmt = cpio->day_first ? "%d %b %H:%M" : "%b %d %H:%M";
1173 #else
1174 if (mtime - now > 365*86400/2
1175 || mtime - now < -365*86400/2)
1176 fmt = cpio->day_first ? "%e %b %Y" : "%b %e %Y";
1177 else
1178 fmt = cpio->day_first ? "%e %b %H:%M" : "%b %e %H:%M";
1179 #endif
1180 strftime(date, sizeof(date), fmt, localtime(&mtime));
1181
1182 fprintf(out, "%s%3d %-8s %-8s %8s %12s %s",
1183 archive_entry_strmode(entry),
1184 archive_entry_nlink(entry),
1185 uname, gname, size, date,
1186 archive_entry_pathname(entry));
1187
1188 /* Extra information for links. */
1189 if (archive_entry_hardlink(entry)) /* Hard link */
1190 fprintf(out, " link to %s", archive_entry_hardlink(entry));
1191 else if (archive_entry_symlink(entry)) /* Symbolic link */
1192 fprintf(out, " -> %s", archive_entry_symlink(entry));
1193 fprintf(out, "\n");
1194 }
1195
1196 static void
mode_pass(struct cpio * cpio,const char * destdir)1197 mode_pass(struct cpio *cpio, const char *destdir)
1198 {
1199 struct lafe_line_reader *lr;
1200 const char *p;
1201 int r;
1202 size_t destdir_len;
1203
1204 /* Ensure target dir has a trailing '/' to simplify path surgery. */
1205 destdir_len = strlen(destdir);
1206 cpio->destdir = malloc(destdir_len + 8);
1207 memcpy(cpio->destdir, destdir, destdir_len);
1208 if (destdir_len == 0 || destdir[destdir_len - 1] != '/')
1209 cpio->destdir[destdir_len++] = '/';
1210 cpio->destdir[destdir_len++] = '\0';
1211
1212 cpio->archive = archive_write_disk_new();
1213 if (cpio->archive == NULL)
1214 lafe_errc(1, 0, "Failed to allocate archive object");
1215 r = archive_write_disk_set_options(cpio->archive, cpio->extract_flags);
1216 if (r != ARCHIVE_OK)
1217 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1218 cpio->linkresolver = archive_entry_linkresolver_new();
1219 archive_write_disk_set_standard_lookup(cpio->archive);
1220
1221 cpio->archive_read_disk = archive_read_disk_new();
1222 if (cpio->archive_read_disk == NULL)
1223 lafe_errc(1, 0, "Failed to allocate archive object");
1224 if (cpio->option_follow_links)
1225 archive_read_disk_set_symlink_logical(cpio->archive_read_disk);
1226 else
1227 archive_read_disk_set_symlink_physical(cpio->archive_read_disk);
1228 archive_read_disk_set_standard_lookup(cpio->archive_read_disk);
1229
1230 lr = lafe_line_reader("-", cpio->option_null);
1231 while ((p = lafe_line_reader_next(lr)) != NULL)
1232 file_to_archive(cpio, p);
1233 lafe_line_reader_free(lr);
1234
1235 archive_entry_linkresolver_free(cpio->linkresolver);
1236 r = archive_write_close(cpio->archive);
1237 if (cpio->dot)
1238 fprintf(stderr, "\n");
1239 if (r != ARCHIVE_OK)
1240 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1241
1242 if (!cpio->quiet) {
1243 int64_t blocks =
1244 (archive_filter_bytes(cpio->archive, 0) + 511)
1245 / 512;
1246 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1247 blocks == 1 ? "block" : "blocks");
1248 }
1249
1250 archive_write_free(cpio->archive);
1251 free(cpio->pass_destpath);
1252 }
1253
1254 /*
1255 * Prompt for a new name for this entry. Returns a pointer to the
1256 * new name or NULL if the entry should not be copied. This
1257 * implements the semantics defined in POSIX.1-1996, which specifies
1258 * that an input of '.' means the name should be unchanged. GNU cpio
1259 * treats '.' as a literal new name.
1260 */
1261 static const char *
cpio_rename(const char * name)1262 cpio_rename(const char *name)
1263 {
1264 static char buff[1024];
1265 FILE *t;
1266 char *p, *ret;
1267 #if defined(_WIN32) && !defined(__CYGWIN__)
1268 FILE *to;
1269
1270 t = fopen("CONIN$", "r");
1271 if (t == NULL)
1272 return (name);
1273 to = fopen("CONOUT$", "w");
1274 if (to == NULL) {
1275 fclose(t);
1276 return (name);
1277 }
1278 fprintf(to, "%s (Enter/./(new name))? ", name);
1279 fclose(to);
1280 #else
1281 t = fopen("/dev/tty", "r+");
1282 if (t == NULL)
1283 return (name);
1284 fprintf(t, "%s (Enter/./(new name))? ", name);
1285 fflush(t);
1286 #endif
1287
1288 p = fgets(buff, sizeof(buff), t);
1289 fclose(t);
1290 if (p == NULL)
1291 /* End-of-file is a blank line. */
1292 return (NULL);
1293
1294 while (*p == ' ' || *p == '\t')
1295 ++p;
1296 if (*p == '\n' || *p == '\0')
1297 /* Empty line. */
1298 return (NULL);
1299 if (*p == '.' && p[1] == '\n')
1300 /* Single period preserves original name. */
1301 return (name);
1302 ret = p;
1303 /* Trim the final newline. */
1304 while (*p != '\0' && *p != '\n')
1305 ++p;
1306 /* Overwrite the final \n with a null character. */
1307 *p = '\0';
1308 return (ret);
1309 }
1310
1311 static void
free_cache(struct name_cache * cache)1312 free_cache(struct name_cache *cache)
1313 {
1314 size_t i;
1315
1316 if (cache != NULL) {
1317 for (i = 0; i < cache->size; i++)
1318 free(cache->cache[i].name);
1319 free(cache);
1320 }
1321 }
1322
1323 /*
1324 * Lookup uname/gname from uid/gid, return NULL if no match.
1325 */
1326 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)1327 lookup_name(struct cpio *cpio, struct name_cache **name_cache_variable,
1328 int (*lookup_fn)(struct cpio *, const char **, id_t), id_t id)
1329 {
1330 char asnum[16];
1331 struct name_cache *cache;
1332 const char *name;
1333 int slot;
1334
1335
1336 if (*name_cache_variable == NULL) {
1337 *name_cache_variable = calloc(1, sizeof(struct name_cache));
1338 if (*name_cache_variable == NULL)
1339 lafe_errc(1, ENOMEM, "No more memory");
1340 (*name_cache_variable)->size = name_cache_size;
1341 }
1342
1343 cache = *name_cache_variable;
1344 cache->probes++;
1345
1346 slot = id % cache->size;
1347 if (cache->cache[slot].name != NULL) {
1348 if (cache->cache[slot].id == id) {
1349 cache->hits++;
1350 return (cache->cache[slot].name);
1351 }
1352 free(cache->cache[slot].name);
1353 cache->cache[slot].name = NULL;
1354 }
1355
1356 if (lookup_fn(cpio, &name, id)) {
1357 /* If lookup failed, format it as a number. */
1358 snprintf(asnum, sizeof(asnum), "%u", (unsigned)id);
1359 name = asnum;
1360 }
1361
1362 cache->cache[slot].name = strdup(name);
1363 if (cache->cache[slot].name != NULL) {
1364 cache->cache[slot].id = id;
1365 return (cache->cache[slot].name);
1366 }
1367
1368 /*
1369 * Conveniently, NULL marks an empty slot, so
1370 * if the strdup() fails, we've just failed to
1371 * cache it. No recovery necessary.
1372 */
1373 return (NULL);
1374 }
1375
1376 static const char *
lookup_uname(struct cpio * cpio,uid_t uid)1377 lookup_uname(struct cpio *cpio, uid_t uid)
1378 {
1379 return (lookup_name(cpio, &cpio->uname_cache,
1380 &lookup_uname_helper, (id_t)uid));
1381 }
1382
1383 static int
lookup_uname_helper(struct cpio * cpio,const char ** name,id_t id)1384 lookup_uname_helper(struct cpio *cpio, const char **name, id_t id)
1385 {
1386 struct passwd *pwent;
1387
1388 (void)cpio; /* UNUSED */
1389
1390 errno = 0;
1391 pwent = getpwuid((uid_t)id);
1392 if (pwent == NULL) {
1393 if (errno && errno != ENOENT)
1394 lafe_warnc(errno, "getpwuid(%s) failed",
1395 cpio_i64toa((int64_t)id));
1396 return 1;
1397 }
1398
1399 *name = pwent->pw_name;
1400 return 0;
1401 }
1402
1403 static const char *
lookup_gname(struct cpio * cpio,gid_t gid)1404 lookup_gname(struct cpio *cpio, gid_t gid)
1405 {
1406 return (lookup_name(cpio, &cpio->gname_cache,
1407 &lookup_gname_helper, (id_t)gid));
1408 }
1409
1410 static int
lookup_gname_helper(struct cpio * cpio,const char ** name,id_t id)1411 lookup_gname_helper(struct cpio *cpio, const char **name, id_t id)
1412 {
1413 struct group *grent;
1414
1415 (void)cpio; /* UNUSED */
1416
1417 errno = 0;
1418 grent = getgrgid((gid_t)id);
1419 if (grent == NULL) {
1420 if (errno && errno != ENOENT)
1421 lafe_warnc(errno, "getgrgid(%s) failed",
1422 cpio_i64toa((int64_t)id));
1423 return 1;
1424 }
1425
1426 *name = grent->gr_name;
1427 return 0;
1428 }
1429
1430 /*
1431 * It would be nice to just use printf() for formatting large numbers,
1432 * but the compatibility problems are a big headache. Hence the
1433 * following simple utility function.
1434 */
1435 const char *
cpio_i64toa(int64_t n0)1436 cpio_i64toa(int64_t n0)
1437 {
1438 /* 2^64 =~ 1.8 * 10^19, so 20 decimal digits suffice.
1439 * We also need 1 byte for '-' and 1 for '\0'.
1440 */
1441 static char buff[22];
1442 int64_t n = n0 < 0 ? -n0 : n0;
1443 char *p = buff + sizeof(buff);
1444
1445 *--p = '\0';
1446 do {
1447 *--p = '0' + (int)(n % 10);
1448 n /= 10;
1449 } while (n > 0);
1450 if (n0 < 0)
1451 *--p = '-';
1452 return p;
1453 }
1454
1455 #define PPBUFF_SIZE 1024
1456 static const char *
passphrase_callback(struct archive * a,void * _client_data)1457 passphrase_callback(struct archive *a, void *_client_data)
1458 {
1459 struct cpio *cpio = (struct cpio *)_client_data;
1460 (void)a; /* UNUSED */
1461
1462 if (cpio->ppbuff == NULL) {
1463 cpio->ppbuff = malloc(PPBUFF_SIZE);
1464 if (cpio->ppbuff == NULL)
1465 lafe_errc(1, errno, "Out of memory");
1466 }
1467 return lafe_readpassphrase("Enter passphrase:",
1468 cpio->ppbuff, PPBUFF_SIZE);
1469 }
1470
1471 static void
passphrase_free(char * ppbuff)1472 passphrase_free(char *ppbuff)
1473 {
1474 if (ppbuff != NULL) {
1475 memset(ppbuff, 0, PPBUFF_SIZE);
1476 free(ppbuff);
1477 }
1478 }
1479