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