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: 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_PWD_H
  54 #include <pwd.h>
  55 #endif
  56 #ifdef HAVE_STDARG_H
  57 #include <stdarg.h>
  58 #endif
  59 #ifdef HAVE_STDINT_H
  60 #include <stdint.h>
  61 #endif
  62 #include <stdio.h>
  63 #ifdef HAVE_STDLIB_H
  64 #include <stdlib.h>
  65 #endif
  66 #ifdef HAVE_STRING_H
  67 #include <string.h>
  68 #endif
  69 #ifdef HAVE_UNISTD_H
  70 #include <unistd.h>
  71 #endif
  72 #ifdef HAVE_SYS_TIME_H
  73 #include <sys/time.h>
  74 #endif
  75 #ifdef HAVE_TIME_H
  76 #include <time.h>
  77 #endif
  78
  79 #include "cpio.h"
  80 #include "err.h"
  81 #include "line_reader.h"
  82 #include "matching.h"
  83
  84 /* Fixed size of uname/gname caches. */
  85 #define name_cache_size 101
  86
  87 #ifndef O_BINARY
  88 #define O_BINARY 0
  89 #endif
  90
  91 struct name_cache {
  92         int     probes;
  93         int     hits;
  94         size_t  size;
  95         struct {
  96                 id_t id;
  97                 char *name;
  98         } cache[name_cache_size];
  99 };
 100
 101 static int      extract_data(struct archive *, struct archive *);
 102 const char *    cpio_i64toa(int64_t);
 103 static const char *cpio_rename(const char *name);
 104 static int      entry_to_archive(struct cpio *, struct archive_entry *);
 105 static int      file_to_archive(struct cpio *, const char *);
 106 static void     free_cache(struct name_cache *cache);
 107 static void     list_item_verbose(struct cpio *, struct archive_entry *);
 108 static void     long_help(void);
 109 static const char *lookup_gname(struct cpio *, gid_t gid);
 110 static int      lookup_gname_helper(struct cpio *,
 111                     const char **name, id_t gid);
 112 static const char *lookup_uname(struct cpio *, uid_t uid);
 113 static int      lookup_uname_helper(struct cpio *,
 114                     const char **name, id_t uid);
 115 static void     mode_in(struct cpio *);
 116 static void     mode_list(struct cpio *);
 117 static void     mode_out(struct cpio *);
 118 static void     mode_pass(struct cpio *, const char *);
 119 static int      restore_time(struct cpio *, struct archive_entry *,
 120                     const char *, int fd);
 121 static void     usage(void);
 122 static void     version(void);
 123
 124 int
 125 main(int argc, char *argv[])
 126 {
 127         static char buff[16384];
 128         struct cpio _cpio; /* Allocated on stack. */
 129         struct cpio *cpio;
 130         const char *errmsg;
 131         int uid, gid;
 132         int opt;
 133
 134         cpio = &_cpio;
 135         memset(cpio, 0, sizeof(*cpio));
 136         cpio->buff = buff;
 137         cpio->buff_size = sizeof(buff);
 138
 139         /* Need lafe_progname before calling lafe_warnc. */
 140         if (*argv == NULL)
 141                 lafe_progname = "bsdcpio";
 142         else {
 143 #if defined(_WIN32) && !defined(__CYGWIN__)
 144                 lafe_progname = strrchr(*argv, '\\');
 145 #else
 146                 lafe_progname = strrchr(*argv, '/');
 147 #endif
 148                 if (lafe_progname != NULL)
 149                         lafe_progname++;
 150                 else
 151                         lafe_progname = *argv;
 152         }
 153
 154         cpio->uid_override = -1;
 155         cpio->gid_override = -1;
 156         cpio->argv = argv;
 157         cpio->argc = argc;
 158         cpio->mode = '\0';
 159         cpio->verbose = 0;
 160         cpio->compress = '\0';
 161         cpio->extract_flags = ARCHIVE_EXTRACT_NO_AUTODIR;
 162         cpio->extract_flags |= ARCHIVE_EXTRACT_NO_OVERWRITE_NEWER;
 163         cpio->extract_flags |= ARCHIVE_EXTRACT_SECURE_SYMLINKS;
 164         cpio->extract_flags |= ARCHIVE_EXTRACT_SECURE_NODOTDOT;
 165         cpio->extract_flags |= ARCHIVE_EXTRACT_PERM;
 166         cpio->extract_flags |= ARCHIVE_EXTRACT_FFLAGS;
 167         cpio->extract_flags |= ARCHIVE_EXTRACT_ACL;
 168 #if !defined(_WIN32) && !defined(__CYGWIN__)
 169         if (geteuid() == 0)
 170                 cpio->extract_flags |= ARCHIVE_EXTRACT_OWNER;
 171 #endif
 172         cpio->bytes_per_block = 512;
 173         cpio->filename = NULL;
 174
 175         while ((opt = cpio_getopt(cpio)) != -1) {
 176                 switch (opt) {
 177                 case '0': /* GNU convention: --null, -0 */
 178                         cpio->option_null = 1;
 179                         break;
 180                 case 'A': /* NetBSD/OpenBSD */
 181                         cpio->option_append = 1;
 182                         break;
 183                 case 'a': /* POSIX 1997 */
 184                         cpio->option_atime_restore = 1;
 185                         break;
 186                 case 'B': /* POSIX 1997 */
 187                         cpio->bytes_per_block = 5120;
 188                         break;
 189                 case 'C': /* NetBSD/OpenBSD */
 190                         cpio->bytes_per_block = atoi(cpio->optarg);
 191                         if (cpio->bytes_per_block <= 0)
 192                                 lafe_errc(1, 0, "Invalid blocksize %s", cpio->optarg);
 193                         break;
 194                 case 'c': /* POSIX 1997 */
 195                         cpio->format = "odc";
 196                         break;
 197                 case 'd': /* POSIX 1997 */
 198                         cpio->extract_flags &= ~ARCHIVE_EXTRACT_NO_AUTODIR;
 199                         break;
 200                 case 'E': /* NetBSD/OpenBSD */
 201                         lafe_include_from_file(&cpio->matching,
 202                             cpio->optarg, cpio->option_null);
 203                         break;
 204                 case 'F': /* NetBSD/OpenBSD/GNU cpio */
 205                         cpio->filename = cpio->optarg;
 206                         break;
 207                 case 'f': /* POSIX 1997 */
 208                         lafe_exclude(&cpio->matching, cpio->optarg);
 209                         break;
 210                 case 'H': /* GNU cpio (also --format) */
 211                         cpio->format = cpio->optarg;
 212                         break;
 213                 case 'h':
 214                         long_help();
 215                         break;
 216                 case 'I': /* NetBSD/OpenBSD */
 217                         cpio->filename = cpio->optarg;
 218                         break;
 219                 case 'i': /* POSIX 1997 */
 220                         if (cpio->mode != '\0')
 221                                 lafe_errc(1, 0,
 222                                     "Cannot use both -i and -%c", cpio->mode);
 223                         cpio->mode = opt;
 224                         break;
 225                 case 'J': /* GNU tar, others */
 226                         cpio->compress = opt;
 227                         break;
 228                 case 'j': /* GNU tar, others */
 229                         cpio->compress = opt;
 230                         break;
 231                 case OPTION_INSECURE:
 232                         cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_SYMLINKS;
 233                         cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_NODOTDOT;
 234                         break;
 235                 case 'L': /* GNU cpio */
 236                         cpio->option_follow_links = 1;
 237                         break;
 238                 case 'l': /* POSIX 1997 */
 239                         cpio->option_link = 1;
 240                         break;
 241                 case OPTION_LZMA: /* GNU tar, others */
 242                         cpio->compress = opt;
 243                         break;
 244                 case 'm': /* POSIX 1997 */
 245                         cpio->extract_flags |= ARCHIVE_EXTRACT_TIME;
 246                         break;
 247                 case 'n': /* GNU cpio */
 248                         cpio->option_numeric_uid_gid = 1;
 249                         break;
 250                 case OPTION_NO_PRESERVE_OWNER: /* GNU cpio */
 251                         cpio->extract_flags &= ~ARCHIVE_EXTRACT_OWNER;
 252                         break;
 253                 case 'O': /* GNU cpio */
 254                         cpio->filename = cpio->optarg;
 255                         break;
 256                 case 'o': /* POSIX 1997 */
 257                         if (cpio->mode != '\0')
 258                                 lafe_errc(1, 0,
 259                                     "Cannot use both -o and -%c", cpio->mode);
 260                         cpio->mode = opt;
 261                         break;
 262                 case 'p': /* POSIX 1997 */
 263                         if (cpio->mode != '\0')
 264                                 lafe_errc(1, 0,
 265                                     "Cannot use both -p and -%c", cpio->mode);
 266                         cpio->mode = opt;
 267                         cpio->extract_flags &= ~ARCHIVE_EXTRACT_SECURE_NODOTDOT;
 268                         break;
 269                 case OPTION_PRESERVE_OWNER:
 270                         cpio->extract_flags |= ARCHIVE_EXTRACT_OWNER;
 271                         break;
 272                 case OPTION_QUIET: /* GNU cpio */
 273                         cpio->quiet = 1;
 274                         break;
 275                 case 'R': /* GNU cpio, also --owner */
 276                         errmsg = owner_parse(cpio->optarg, &uid, &gid);
 277                         if (errmsg) {
 278                                 lafe_warnc(-1, "%s", errmsg);
 279                                 usage();
 280                         }
 281                         if (uid != -1)
 282                                 cpio->uid_override = uid;
 283                         if (gid != -1)
 284                                 cpio->gid_override = gid;
 285                         break;
 286                 case 'r': /* POSIX 1997 */
 287                         cpio->option_rename = 1;
 288                         break;
 289                 case 't': /* POSIX 1997 */
 290                         cpio->option_list = 1;
 291                         break;
 292                 case 'u': /* POSIX 1997 */
 293                         cpio->extract_flags
 294                             &= ~ARCHIVE_EXTRACT_NO_OVERWRITE_NEWER;
 295                         break;
 296                 case 'v': /* POSIX 1997 */
 297                         cpio->verbose++;
 298                         break;
 299                 case OPTION_VERSION: /* GNU convention */
 300                         version();
 301                         break;
 302 #if 0
 303                 /*
 304                  * cpio_getopt() handles -W specially, so it's not
 305                  * available here.
 306                  */
 307                 case 'W': /* Obscure, but useful GNU convention. */
 308                         break;
 309 #endif
 310                 case 'y': /* tar convention */
 311                         cpio->compress = opt;
 312                         break;
 313                 case 'Z': /* tar convention */
 314                         cpio->compress = opt;
 315                         break;
 316                 case 'z': /* tar convention */
 317                         cpio->compress = opt;
 318                         break;
 319                 default:
 320                         usage();
 321                 }
 322         }
 323
 324         /*
 325          * Sanity-check args, error out on nonsensical combinations.
 326          */
 327         /* -t implies -i if no mode was specified. */
 328         if (cpio->option_list && cpio->mode == '\0')
 329                 cpio->mode = 'i';
 330         /* -t requires -i */
 331         if (cpio->option_list && cpio->mode != 'i')
 332                 lafe_errc(1, 0, "Option -t requires -i");
 333         /* -n requires -it */
 334         if (cpio->option_numeric_uid_gid && !cpio->option_list)
 335                 lafe_errc(1, 0, "Option -n requires -it");
 336         /* Can only specify format when writing */
 337         if (cpio->format != NULL && cpio->mode != 'o')
 338                 lafe_errc(1, 0, "Option --format requires -o");
 339         /* -l requires -p */
 340         if (cpio->option_link && cpio->mode != 'p')
 341                 lafe_errc(1, 0, "Option -l requires -p");
 342         /* TODO: Flag other nonsensical combinations. */
 343
 344         switch (cpio->mode) {
 345         case 'o':
 346                 /* TODO: Implement old binary format in libarchive,
 347                    use that here. */
 348                 if (cpio->format == NULL)
 349                         cpio->format = "odc"; /* Default format */
 350
 351                 mode_out(cpio);
 352                 break;
 353         case 'i':
 354                 while (*cpio->argv != NULL) {
 355                         lafe_include(&cpio->matching, *cpio->argv);
 356                         --cpio->argc;
 357                         ++cpio->argv;
 358                 }
 359                 if (cpio->option_list)
 360                         mode_list(cpio);
 361                 else
 362                         mode_in(cpio);
 363                 break;
 364         case 'p':
 365                 if (*cpio->argv == NULL || **cpio->argv == '\0')
 366                         lafe_errc(1, 0,
 367                             "-p mode requires a target directory");
 368                 mode_pass(cpio, *cpio->argv);
 369                 break;
 370         default:
 371                 lafe_errc(1, 0,
 372                     "Must specify at least one of -i, -o, or -p");
 373         }
 374
 375         free_cache(cpio->gname_cache);
 376         free_cache(cpio->uname_cache);
 377         return (cpio->return_value);
 378 }
 379
 380 static void
 381 usage(void)
 382 {
 383         const char      *p;
 384
 385         p = lafe_progname;
 386
 387         fprintf(stderr, "Brief Usage:\n");
 388         fprintf(stderr, "  List:    %s -it < archive\n", p);
 389         fprintf(stderr, "  Extract: %s -i < archive\n", p);
 390         fprintf(stderr, "  Create:  %s -o < filenames > archive\n", p);
 391         fprintf(stderr, "  Help:    %s --help\n", p);
 392         exit(1);
 393 }
 394
 395 static const char *long_help_msg =
 396         "First option must be a mode specifier:\n"
 397         "  -i Input  -o Output  -p Pass\n"
 398         "Common Options:\n"
 399         "  -v    Verbose\n"
 400         "Create: %p -o [options]  < [list of files] > [archive]\n"
 401         "  -J,-y,-z,--lzma  Compress archive with xz/bzip2/gzip/lzma\n"
 402         "  --format {odc|newc|ustar}  Select archive format\n"
 403         "List: %p -it < [archive]\n"
 404         "Extract: %p -i [options] < [archive]\n";
 405
 406
 407 /*
 408  * Note that the word 'bsdcpio' will always appear in the first line
 409  * of output.
 410  *
 411  * In particular, /bin/sh scripts that need to test for the presence
 412  * of bsdcpio can use the following template:
 413  *
 414  * if (cpio --help 2>&1 | grep bsdcpio >/dev/null 2>&1 ) then \
 415  *          echo bsdcpio; else echo not bsdcpio; fi
 416  */
 417 static void
 418 long_help(void)
 419 {
 420         const char      *prog;
 421         const char      *p;
 422
 423         prog = lafe_progname;
 424
 425         fflush(stderr);
 426
 427         p = (strcmp(prog,"bsdcpio") != 0) ? "(bsdcpio)" : "";
 428         printf("%s%s: manipulate archive files\n", prog, p);
 429
 430         for (p = long_help_msg; *p != '\0'; p++) {
 431                 if (*p == '%') {
 432                         if (p[1] == 'p') {
 433                                 fputs(prog, stdout);
 434                                 p++;
 435                         } else
 436                                 putchar('%');
 437                 } else
 438                         putchar(*p);
 439         }
 440         version();
 441 }
 442
 443 static void
 444 version(void)
 445 {
 446         fprintf(stdout,"bsdcpio %s -- %s\n",
 447             BSDCPIO_VERSION_STRING,
 448             archive_version());
 449         exit(0);
 450 }
 451
 452 static void
 453 mode_out(struct cpio *cpio)
 454 {
 455         struct archive_entry *entry, *spare;
 456         struct lafe_line_reader *lr;
 457         const char *p;
 458         int r;
 459
 460         if (cpio->option_append)
 461                 lafe_errc(1, 0, "Append mode not yet supported.");
 462
 463         cpio->archive_read_disk = archive_read_disk_new();
 464         if (cpio->archive_read_disk == NULL)
 465                 lafe_errc(1, 0, "Failed to allocate archive object");
 466         if (cpio->option_follow_links)
 467                 archive_read_disk_set_symlink_logical(cpio->archive_read_disk);
 468         else
 469                 archive_read_disk_set_symlink_physical(cpio->archive_read_disk);
 470         archive_read_disk_set_standard_lookup(cpio->archive_read_disk);
 471
 472         cpio->archive = archive_write_new();
 473         if (cpio->archive == NULL)
 474                 lafe_errc(1, 0, "Failed to allocate archive object");
 475         switch (cpio->compress) {
 476         case 'J':
 477                 r = archive_write_set_compression_xz(cpio->archive);
 478                 break;
 479         case OPTION_LZMA:
 480                 r = archive_write_set_compression_lzma(cpio->archive);
 481                 break;
 482         case 'j': case 'y':
 483                 r = archive_write_set_compression_bzip2(cpio->archive);
 484                 break;
 485         case 'z':
 486                 r = archive_write_set_compression_gzip(cpio->archive);
 487                 break;
 488         case 'Z':
 489                 r = archive_write_set_compression_compress(cpio->archive);
 490                 break;
 491         default:
 492                 r = archive_write_set_compression_none(cpio->archive);
 493                 break;
 494         }
 495         if (r < ARCHIVE_WARN)
 496                 lafe_errc(1, 0, "Requested compression not available");
 497         r = archive_write_set_format_by_name(cpio->archive, cpio->format);
 498         if (r != ARCHIVE_OK)
 499                 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
 500         archive_write_set_bytes_per_block(cpio->archive, cpio->bytes_per_block);
 501         cpio->linkresolver = archive_entry_linkresolver_new();
 502         archive_entry_linkresolver_set_strategy(cpio->linkresolver,
 503             archive_format(cpio->archive));
 504
 505         /*
 506          * The main loop:  Copy each file into the output archive.
 507          */
 508         r = archive_write_open_file(cpio->archive, cpio->filename);
 509         if (r != ARCHIVE_OK)
 510                 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
 511         lr = lafe_line_reader("-", cpio->option_null);
 512         while ((p = lafe_line_reader_next(lr)) != NULL)
 513                 file_to_archive(cpio, p);
 514         lafe_line_reader_free(lr);
 515
 516         /*
 517          * The hardlink detection may have queued up a couple of entries
 518          * that can now be flushed.
 519          */
 520         entry = NULL;
 521         archive_entry_linkify(cpio->linkresolver, &entry, &spare);
 522         while (entry != NULL) {
 523                 entry_to_archive(cpio, entry);
 524                 archive_entry_free(entry);
 525                 entry = NULL;
 526                 archive_entry_linkify(cpio->linkresolver, &entry, &spare);
 527         }
 528
 529         r = archive_write_close(cpio->archive);
 530         if (r != ARCHIVE_OK)
 531                 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
 532
 533         if (!cpio->quiet) {
 534                 int64_t blocks =
 535                         (archive_position_uncompressed(cpio->archive) + 511)
 536                         / 512;
 537                 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
 538                     blocks == 1 ? "block" : "blocks");
 539         }
 540         archive_write_finish(cpio->archive);
 541 }
 542
 543 /*
 544  * This is used by both out mode (to copy objects from disk into
 545  * an archive) and pass mode (to copy objects from disk to
 546  * an archive_write_disk "archive").
 547  */
 548 static int
 549 file_to_archive(struct cpio *cpio, const char *srcpath)
 550 {
 551         const char *destpath;
 552         struct archive_entry *entry, *spare;
 553         size_t len;
 554         const char *p;
 555         int r;
 556
 557         /*
 558          * Create an archive_entry describing the source file.
 559          *
 560          */
 561         entry = archive_entry_new();
 562         if (entry == NULL)
 563                 lafe_errc(1, 0, "Couldn't allocate entry");
 564         archive_entry_copy_sourcepath(entry, srcpath);
 565         r = archive_read_disk_entry_from_file(cpio->archive_read_disk,
 566             entry, -1, NULL);
 567         if (r < ARCHIVE_FAILED)
 568                 lafe_errc(1, 0, "%s",
 569                     archive_error_string(cpio->archive_read_disk));
 570         if (r < ARCHIVE_OK)
 571                 lafe_warnc(0, "%s",
 572                     archive_error_string(cpio->archive_read_disk));
 573         if (r <= ARCHIVE_FAILED) {
 574                 cpio->return_value = 1;
 575                 return (r);
 576         }
 577
 578         if (cpio->uid_override >= 0)
 579                 archive_entry_set_uid(entry, cpio->uid_override);
 580         if (cpio->gid_override >= 0)
 581                 archive_entry_set_gid(entry, cpio->gid_override);
 582
 583         /*
 584          * Generate a destination path for this entry.
 585          * "destination path" is the name to which it will be copied in
 586          * pass mode or the name that will go into the archive in
 587          * output mode.
 588          */
 589         destpath = srcpath;
 590         if (cpio->destdir) {
 591                 len = strlen(cpio->destdir) + strlen(srcpath) + 8;
 592                 if (len >= cpio->pass_destpath_alloc) {
 593                         while (len >= cpio->pass_destpath_alloc) {
 594                                 cpio->pass_destpath_alloc += 512;
 595                                 cpio->pass_destpath_alloc *= 2;
 596                         }
 597                         free(cpio->pass_destpath);
 598                         cpio->pass_destpath = malloc(cpio->pass_destpath_alloc);
 599                         if (cpio->pass_destpath == NULL)
 600                                 lafe_errc(1, ENOMEM,
 601                                     "Can't allocate path buffer");
 602                 }
 603                 strcpy(cpio->pass_destpath, cpio->destdir);
 604                 p = srcpath;
 605                 while (p[0] == '/')
 606                         ++p;
 607                 strcat(cpio->pass_destpath, p);
 608                 destpath = cpio->pass_destpath;
 609         }
 610         if (cpio->option_rename)
 611                 destpath = cpio_rename(destpath);
 612         if (destpath == NULL)
 613                 return (0);
 614         archive_entry_copy_pathname(entry, destpath);
 615
 616         /*
 617          * If we're trying to preserve hardlinks, match them here.
 618          */
 619         spare = NULL;
 620         if (cpio->linkresolver != NULL
 621             && archive_entry_filetype(entry) != AE_IFDIR) {
 622                 archive_entry_linkify(cpio->linkresolver, &entry, &spare);
 623         }
 624
 625         if (entry != NULL) {
 626                 r = entry_to_archive(cpio, entry);
 627                 archive_entry_free(entry);
 628                 if (spare != NULL) {
 629                         if (r == 0)
 630                                 r = entry_to_archive(cpio, spare);
 631                         archive_entry_free(spare);
 632                 }
 633         }
 634         return (r);
 635 }
 636
 637 static int
 638 entry_to_archive(struct cpio *cpio, struct archive_entry *entry)
 639 {
 640         const char *destpath = archive_entry_pathname(entry);
 641         const char *srcpath = archive_entry_sourcepath(entry);
 642         int fd = -1;
 643         ssize_t bytes_read;
 644         int r;
 645
 646         /* Print out the destination name to the user. */
 647         if (cpio->verbose)
 648                 fprintf(stderr,"%s", destpath);
 649
 650         /*
 651          * Option_link only makes sense in pass mode and for
 652          * regular files.  Also note: if a link operation fails
 653          * because of cross-device restrictions, we'll fall back
 654          * to copy mode for that entry.
 655          *
 656          * TODO: Test other cpio implementations to see if they
 657          * hard-link anything other than regular files here.
 658          */
 659         if (cpio->option_link
 660             && archive_entry_filetype(entry) == AE_IFREG)
 661         {
 662                 struct archive_entry *t;
 663                 /* Save the original entry in case we need it later. */
 664                 t = archive_entry_clone(entry);
 665                 if (t == NULL)
 666                         lafe_errc(1, ENOMEM, "Can't create link");
 667                 /* Note: link(2) doesn't create parent directories,
 668                  * so we use archive_write_header() instead as a
 669                  * convenience. */
 670                 archive_entry_set_hardlink(t, srcpath);
 671                 /* This is a straight link that carries no data. */
 672                 archive_entry_set_size(t, 0);
 673                 r = archive_write_header(cpio->archive, t);
 674                 archive_entry_free(t);
 675                 if (r != ARCHIVE_OK)
 676                         lafe_warnc(archive_errno(cpio->archive),
 677                             "%s", archive_error_string(cpio->archive));
 678                 if (r == ARCHIVE_FATAL)
 679                         exit(1);
 680 #ifdef EXDEV
 681                 if (r != ARCHIVE_OK && archive_errno(cpio->archive) == EXDEV) {
 682                         /* Cross-device link:  Just fall through and use
 683                          * the original entry to copy the file over. */
 684                         lafe_warnc(0, "Copying file instead");
 685                 } else
 686 #endif
 687                 return (0);
 688         }
 689
 690         /*
 691          * Make sure we can open the file (if necessary) before
 692          * trying to write the header.
 693          */
 694         if (archive_entry_filetype(entry) == AE_IFREG) {
 695                 if (archive_entry_size(entry) > 0) {
 696                         fd = open(srcpath, O_RDONLY | O_BINARY);
 697                         if (fd < 0) {
 698                                 lafe_warnc(errno,
 699                                     "%s: could not open file", srcpath);
 700                                 goto cleanup;
 701                         }
 702                 }
 703         } else {
 704                 archive_entry_set_size(entry, 0);
 705         }
 706
 707         r = archive_write_header(cpio->archive, entry);
 708
 709         if (r != ARCHIVE_OK)
 710                 lafe_warnc(archive_errno(cpio->archive),
 711                     "%s: %s",
 712                     srcpath,
 713                     archive_error_string(cpio->archive));
 714
 715         if (r == ARCHIVE_FATAL)
 716                 exit(1);
 717
 718         if (r >= ARCHIVE_WARN && fd >= 0) {
 719                 bytes_read = read(fd, cpio->buff, cpio->buff_size);
 720                 while (bytes_read > 0) {
 721                         r = archive_write_data(cpio->archive,
 722                             cpio->buff, bytes_read);
 723                         if (r < 0)
 724                                 lafe_errc(1, archive_errno(cpio->archive),
 725                                     "%s", archive_error_string(cpio->archive));
 726                         if (r < bytes_read) {
 727                                 lafe_warnc(0,
 728                                     "Truncated write; file may have grown while being archived.");
 729                         }
 730                         bytes_read = read(fd, cpio->buff, cpio->buff_size);
 731                 }
 732         }
 733
 734         fd = restore_time(cpio, entry, srcpath, fd);
 735
 736 cleanup:
 737         if (cpio->verbose)
 738                 fprintf(stderr,"\n");
 739         if (fd >= 0)
 740                 close(fd);
 741         return (0);
 742 }
 743
 744 static int
 745 restore_time(struct cpio *cpio, struct archive_entry *entry,
 746     const char *name, int fd)
 747 {
 748 #ifndef HAVE_UTIMES
 749         static int warned = 0;
 750
 751         (void)cpio; /* UNUSED */
 752         (void)entry; /* UNUSED */
 753         (void)name; /* UNUSED */
 754
 755         if (!warned)
 756                 lafe_warnc(0, "Can't restore access times on this platform");
 757         warned = 1;
 758         return (fd);
 759 #else
 760 #if defined(_WIN32) && !defined(__CYGWIN__)
 761         struct __timeval times[2];
 762 #else
 763         struct timeval times[2];
 764 #endif
 765
 766         if (!cpio->option_atime_restore)
 767                 return (fd);
 768
 769         times[1].tv_sec = archive_entry_mtime(entry);
 770         times[1].tv_usec = archive_entry_mtime_nsec(entry) / 1000;
 771
 772         times[0].tv_sec = archive_entry_atime(entry);
 773         times[0].tv_usec = archive_entry_atime_nsec(entry) / 1000;
 774
 775 #if defined(HAVE_FUTIMES) && !defined(__CYGWIN__)
 776         if (fd >= 0 && futimes(fd, times) == 0)
 777                 return (fd);
 778 #endif
 779         /*
 780          * Some platform cannot restore access times if the file descriptor
 781          * is still opened.
 782          */
 783         if (fd >= 0) {
 784                 close(fd);
 785                 fd = -1;
 786         }
 787
 788 #ifdef HAVE_LUTIMES
 789         if (lutimes(name, times) != 0)
 790 #else
 791         if ((AE_IFLNK != archive_entry_filetype(entry))
 792                         && utimes(name, times) != 0)
 793 #endif
 794                 lafe_warnc(errno, "Can't update time for %s", name);
 795 #endif
 796         return (fd);
 797 }
 798
 799
 800 static void
 801 mode_in(struct cpio *cpio)
 802 {
 803         struct archive *a;
 804         struct archive_entry *entry;
 805         struct archive *ext;
 806         const char *destpath;
 807         int r;
 808
 809         ext = archive_write_disk_new();
 810         if (ext == NULL)
 811                 lafe_errc(1, 0, "Couldn't allocate restore object");
 812         r = archive_write_disk_set_options(ext, cpio->extract_flags);
 813         if (r != ARCHIVE_OK)
 814                 lafe_errc(1, 0, "%s", archive_error_string(ext));
 815         a = archive_read_new();
 816         if (a == NULL)
 817                 lafe_errc(1, 0, "Couldn't allocate archive object");
 818         archive_read_support_compression_all(a);
 819         archive_read_support_format_all(a);
 820
 821         if (archive_read_open_file(a, cpio->filename, cpio->bytes_per_block))
 822                 lafe_errc(1, archive_errno(a),
 823                     "%s", archive_error_string(a));
 824         for (;;) {
 825                 r = archive_read_next_header(a, &entry);
 826                 if (r == ARCHIVE_EOF)
 827                         break;
 828                 if (r != ARCHIVE_OK) {
 829                         lafe_errc(1, archive_errno(a),
 830                             "%s", archive_error_string(a));
 831                 }
 832                 if (lafe_excluded(cpio->matching, archive_entry_pathname(entry)))
 833                         continue;
 834                 if (cpio->option_rename) {
 835                         destpath = cpio_rename(archive_entry_pathname(entry));
 836                         archive_entry_set_pathname(entry, destpath);
 837                 } else
 838                         destpath = archive_entry_pathname(entry);
 839                 if (destpath == NULL)
 840                         continue;
 841                 if (cpio->verbose)
 842                         fprintf(stdout, "%s\n", destpath);
 843                 if (cpio->uid_override >= 0)
 844                         archive_entry_set_uid(entry, cpio->uid_override);
 845                 if (cpio->gid_override >= 0)
 846                         archive_entry_set_gid(entry, cpio->gid_override);
 847                 r = archive_write_header(ext, entry);
 848                 if (r != ARCHIVE_OK) {
 849                         fprintf(stderr, "%s: %s\n",
 850                             archive_entry_pathname(entry),
 851                             archive_error_string(ext));
 852                 } else if (archive_entry_size(entry) > 0) {
 853                         r = extract_data(a, ext);
 854                         if (r != ARCHIVE_OK)
 855                                 cpio->return_value = 1;
 856                 }
 857         }
 858         r = archive_read_close(a);
 859         if (r != ARCHIVE_OK)
 860                 lafe_errc(1, 0, "%s", archive_error_string(a));
 861         r = archive_write_close(ext);
 862         if (r != ARCHIVE_OK)
 863                 lafe_errc(1, 0, "%s", archive_error_string(ext));
 864         if (!cpio->quiet) {
 865                 int64_t blocks = (archive_position_uncompressed(a) + 511)
 866                               / 512;
 867                 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
 868                     blocks == 1 ? "block" : "blocks");
 869         }
 870         archive_read_finish(a);
 871         archive_write_finish(ext);
 872         exit(cpio->return_value);
 873 }
 874
 875 /*
 876  * Exits if there's a fatal error.  Returns ARCHIVE_OK
 877  * if everything is kosher.
 878  */
 879 static int
 880 extract_data(struct archive *ar, struct archive *aw)
 881 {
 882         int r;
 883         size_t size;
 884         const void *block;
 885         off_t offset;
 886
 887         for (;;) {
 888                 r = archive_read_data_block(ar, &block, &size, &offset);
 889                 if (r == ARCHIVE_EOF)
 890                         return (ARCHIVE_OK);
 891                 if (r != ARCHIVE_OK) {
 892                         lafe_warnc(archive_errno(ar),
 893                             "%s", archive_error_string(ar));
 894                         exit(1);
 895                 }
 896                 r = archive_write_data_block(aw, block, size, offset);
 897                 if (r != ARCHIVE_OK) {
 898                         lafe_warnc(archive_errno(aw),
 899                             "%s", archive_error_string(aw));
 900                         return (r);
 901                 }
 902         }
 903 }
 904
 905 static void
 906 mode_list(struct cpio *cpio)
 907 {
 908         struct archive *a;
 909         struct archive_entry *entry;
 910         int r;
 911
 912         a = archive_read_new();
 913         if (a == NULL)
 914                 lafe_errc(1, 0, "Couldn't allocate archive object");
 915         archive_read_support_compression_all(a);
 916         archive_read_support_format_all(a);
 917
 918         if (archive_read_open_file(a, cpio->filename, cpio->bytes_per_block))
 919                 lafe_errc(1, archive_errno(a),
 920                     "%s", archive_error_string(a));
 921         for (;;) {
 922                 r = archive_read_next_header(a, &entry);
 923                 if (r == ARCHIVE_EOF)
 924                         break;
 925                 if (r != ARCHIVE_OK) {
 926                         lafe_errc(1, archive_errno(a),
 927                             "%s", archive_error_string(a));
 928                 }
 929                 if (lafe_excluded(cpio->matching, archive_entry_pathname(entry)))
 930                         continue;
 931                 if (cpio->verbose)
 932                         list_item_verbose(cpio, entry);
 933                 else
 934                         fprintf(stdout, "%s\n", archive_entry_pathname(entry));
 935         }
 936         r = archive_read_close(a);
 937         if (r != ARCHIVE_OK)
 938                 lafe_errc(1, 0, "%s", archive_error_string(a));
 939         if (!cpio->quiet) {
 940                 int64_t blocks = (archive_position_uncompressed(a) + 511)
 941                               / 512;
 942                 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
 943                     blocks == 1 ? "block" : "blocks");
 944         }
 945         archive_read_finish(a);
 946         exit(0);
 947 }
 948
 949 /*
 950  * Display information about the current file.
 951  *
 952  * The format here roughly duplicates the output of 'ls -l'.
 953  * This is based on SUSv2, where 'tar tv' is documented as
 954  * listing additional information in an "unspecified format,"
 955  * and 'pax -l' is documented as using the same format as 'ls -l'.
 956  */
 957 static void
 958 list_item_verbose(struct cpio *cpio, struct archive_entry *entry)
 959 {
 960         char                     size[32];
 961         char                     date[32];
 962         char                     uids[16], gids[16];
 963         const char              *uname, *gname;
 964         FILE                    *out = stdout;
 965         const char              *fmt;
 966         time_t                   mtime;
 967         static time_t            now;
 968
 969         if (!now)
 970                 time(&now);
 971
 972         if (cpio->option_numeric_uid_gid) {
 973                 /* Format numeric uid/gid for display. */
 974                 strcpy(uids, cpio_i64toa(archive_entry_uid(entry)));
 975                 uname = uids;
 976                 strcpy(gids, cpio_i64toa(archive_entry_gid(entry)));
 977                 gname = gids;
 978         } else {
 979                 /* Use uname if it's present, else lookup name from uid. */
 980                 uname = archive_entry_uname(entry);
 981                 if (uname == NULL)
 982                         uname = lookup_uname(cpio, archive_entry_uid(entry));
 983                 /* Use gname if it's present, else lookup name from gid. */
 984                 gname = archive_entry_gname(entry);
 985                 if (gname == NULL)
 986                         gname = lookup_gname(cpio, archive_entry_gid(entry));
 987         }
 988
 989         /* Print device number or file size. */
 990         if (archive_entry_filetype(entry) == AE_IFCHR
 991             || archive_entry_filetype(entry) == AE_IFBLK) {
 992                 snprintf(size, sizeof(size), "%lu,%lu",
 993                     (unsigned long)archive_entry_rdevmajor(entry),
 994                     (unsigned long)archive_entry_rdevminor(entry));
 995         } else {
 996                 strcpy(size, cpio_i64toa(archive_entry_size(entry)));
 997         }
 998
 999         /* Format the time using 'ls -l' conventions. */
1000         mtime = archive_entry_mtime(entry);
1001 #if defined(_WIN32) && !defined(__CYGWIN__)
1002         /* Windows' strftime function does not support %e format. */
1003         if (mtime - now > 365*86400/2
1004                 || mtime - now < -365*86400/2)
1005                 fmt = cpio->day_first ? "%d %b  %Y" : "%b %d  %Y";
1006         else
1007                 fmt = cpio->day_first ? "%d %b %H:%M" : "%b %d %H:%M";
1008 #else
1009         if (abs(mtime - now) > (365/2)*86400)
1010                 fmt = cpio->day_first ? "%e %b  %Y" : "%b %e  %Y";
1011         else
1012                 fmt = cpio->day_first ? "%e %b %H:%M" : "%b %e %H:%M";
1013 #endif
1014         strftime(date, sizeof(date), fmt, localtime(&mtime));
1015
1016         fprintf(out, "%s%3d %-8s %-8s %8s %12s %s",
1017             archive_entry_strmode(entry),
1018             archive_entry_nlink(entry),
1019             uname, gname, size, date,
1020             archive_entry_pathname(entry));
1021
1022         /* Extra information for links. */
1023         if (archive_entry_hardlink(entry)) /* Hard link */
1024                 fprintf(out, " link to %s", archive_entry_hardlink(entry));
1025         else if (archive_entry_symlink(entry)) /* Symbolic link */
1026                 fprintf(out, " -> %s", archive_entry_symlink(entry));
1027         fprintf(out, "\n");
1028 }
1029
1030 static void
1031 mode_pass(struct cpio *cpio, const char *destdir)
1032 {
1033         struct lafe_line_reader *lr;
1034         const char *p;
1035         int r;
1036
1037         /* Ensure target dir has a trailing '/' to simplify path surgery. */
1038         cpio->destdir = malloc(strlen(destdir) + 8);
1039         strcpy(cpio->destdir, destdir);
1040         if (destdir[strlen(destdir) - 1] != '/')
1041                 strcat(cpio->destdir, "/");
1042
1043         cpio->archive = archive_write_disk_new();
1044         if (cpio->archive == NULL)
1045                 lafe_errc(1, 0, "Failed to allocate archive object");
1046         r = archive_write_disk_set_options(cpio->archive, cpio->extract_flags);
1047         if (r != ARCHIVE_OK)
1048                 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1049         cpio->linkresolver = archive_entry_linkresolver_new();
1050         archive_write_disk_set_standard_lookup(cpio->archive);
1051
1052         cpio->archive_read_disk = archive_read_disk_new();
1053         if (cpio->archive_read_disk == NULL)
1054                 lafe_errc(1, 0, "Failed to allocate archive object");
1055         if (cpio->option_follow_links)
1056                 archive_read_disk_set_symlink_logical(cpio->archive_read_disk);
1057         else
1058                 archive_read_disk_set_symlink_physical(cpio->archive_read_disk);
1059         archive_read_disk_set_standard_lookup(cpio->archive_read_disk);
1060
1061         lr = lafe_line_reader("-", cpio->option_null);
1062         while ((p = lafe_line_reader_next(lr)) != NULL)
1063                 file_to_archive(cpio, p);
1064         lafe_line_reader_free(lr);
1065
1066         archive_entry_linkresolver_free(cpio->linkresolver);
1067         r = archive_write_close(cpio->archive);
1068         if (r != ARCHIVE_OK)
1069                 lafe_errc(1, 0, "%s", archive_error_string(cpio->archive));
1070
1071         if (!cpio->quiet) {
1072                 int64_t blocks =
1073                         (archive_position_uncompressed(cpio->archive) + 511)
1074                         / 512;
1075                 fprintf(stderr, "%lu %s\n", (unsigned long)blocks,
1076                     blocks == 1 ? "block" : "blocks");
1077         }
1078
1079         archive_write_finish(cpio->archive);
1080 }
1081
1082 /*
1083  * Prompt for a new name for this entry.  Returns a pointer to the
1084  * new name or NULL if the entry should not be copied.  This
1085  * implements the semantics defined in POSIX.1-1996, which specifies
1086  * that an input of '.' means the name should be unchanged.  GNU cpio
1087  * treats '.' as a literal new name.
1088  */
1089 static const char *
1090 cpio_rename(const char *name)
1091 {
1092         static char buff[1024];
1093         FILE *t;
1094         char *p, *ret;
1095
1096         t = fopen("/dev/tty", "r+");
1097         if (t == NULL)
1098                 return (name);
1099         fprintf(t, "%s (Enter/./(new name))? ", name);
1100         fflush(t);
1101
1102         p = fgets(buff, sizeof(buff), t);
1103         fclose(t);
1104         if (p == NULL)
1105                 /* End-of-file is a blank line. */
1106                 return (NULL);
1107
1108         while (*p == ' ' || *p == '\t')
1109                 ++p;
1110         if (*p == '\n' || *p == '\0')
1111                 /* Empty line. */
1112                 return (NULL);
1113         if (*p == '.' && p[1] == '\n')
1114                 /* Single period preserves original name. */
1115                 return (name);
1116         ret = p;
1117         /* Trim the final newline. */
1118         while (*p != '\0' && *p != '\n')
1119                 ++p;
1120         /* Overwrite the final \n with a null character. */
1121         *p = '\0';
1122         return (ret);
1123 }
1124
1125 static void
1126 free_cache(struct name_cache *cache)
1127 {
1128         size_t i;
1129
1130         if (cache != NULL) {
1131                 for (i = 0; i < cache->size; i++)
1132                         free(cache->cache[i].name);
1133                 free(cache);
1134         }
1135 }
1136
1137 /*
1138  * Lookup uname/gname from uid/gid, return NULL if no match.
1139  */
1140 static const char *
1141 lookup_name(struct cpio *cpio, struct name_cache **name_cache_variable,
1142     int (*lookup_fn)(struct cpio *, const char **, id_t), id_t id)
1143 {
1144         char asnum[16];
1145         struct name_cache       *cache;
1146         const char *name;
1147         int slot;
1148
1149
1150         if (*name_cache_variable == NULL) {
1151                 *name_cache_variable = malloc(sizeof(struct name_cache));
1152                 if (*name_cache_variable == NULL)
1153                         lafe_errc(1, ENOMEM, "No more memory");
1154                 memset(*name_cache_variable, 0, sizeof(struct name_cache));
1155                 (*name_cache_variable)->size = name_cache_size;
1156         }
1157
1158         cache = *name_cache_variable;
1159         cache->probes++;
1160
1161         slot = id % cache->size;
1162         if (cache->cache[slot].name != NULL) {
1163                 if (cache->cache[slot].id == id) {
1164                         cache->hits++;
1165                         return (cache->cache[slot].name);
1166                 }
1167                 free(cache->cache[slot].name);
1168                 cache->cache[slot].name = NULL;
1169         }
1170
1171         if (lookup_fn(cpio, &name, id) == 0) {
1172                 if (name == NULL || name[0] == '\0') {
1173                         /* If lookup failed, format it as a number. */
1174                         snprintf(asnum, sizeof(asnum), "%u", (unsigned)id);
1175                         name = asnum;
1176                 }
1177                 cache->cache[slot].name = strdup(name);
1178                 if (cache->cache[slot].name != NULL) {
1179                         cache->cache[slot].id = id;
1180                         return (cache->cache[slot].name);
1181                 }
1182                 /*
1183                  * Conveniently, NULL marks an empty slot, so
1184                  * if the strdup() fails, we've just failed to
1185                  * cache it.  No recovery necessary.
1186                  */
1187         }
1188         return (NULL);
1189 }
1190
1191 static const char *
1192 lookup_uname(struct cpio *cpio, uid_t uid)
1193 {
1194         return (lookup_name(cpio, &cpio->uname_cache,
1195                     &lookup_uname_helper, (id_t)uid));
1196 }
1197
1198 static int
1199 lookup_uname_helper(struct cpio *cpio, const char **name, id_t id)
1200 {
1201         struct passwd   *pwent;
1202
1203         (void)cpio; /* UNUSED */
1204
1205         errno = 0;
1206         pwent = getpwuid((uid_t)id);
1207         if (pwent == NULL) {
1208                 *name = NULL;
1209                 if (errno != 0 && errno != ENOENT)
1210                         lafe_warnc(errno, "getpwuid(%d) failed", id);
1211                 return (errno);
1212         }
1213
1214         *name = pwent->pw_name;
1215         return (0);
1216 }
1217
1218 static const char *
1219 lookup_gname(struct cpio *cpio, gid_t gid)
1220 {
1221         return (lookup_name(cpio, &cpio->gname_cache,
1222                     &lookup_gname_helper, (id_t)gid));
1223 }
1224
1225 static int
1226 lookup_gname_helper(struct cpio *cpio, const char **name, id_t id)
1227 {
1228         struct group    *grent;
1229
1230         (void)cpio; /* UNUSED */
1231
1232         errno = 0;
1233         grent = getgrgid((gid_t)id);
1234         if (grent == NULL) {
1235                 *name = NULL;
1236                 if (errno != 0)
1237                         lafe_warnc(errno, "getgrgid(%d) failed", id);
1238                 return (errno);
1239         }
1240
1241         *name = grent->gr_name;
1242         return (0);
1243 }
1244
1245 /*
1246  * It would be nice to just use printf() for formatting large numbers,
1247  * but the compatibility problems are a big headache.  Hence the
1248  * following simple utility function.
1249  */
1250 const char *
1251 cpio_i64toa(int64_t n0)
1252 {
1253         // 2^64 =~ 1.8 * 10^19, so 20 decimal digits suffice.
1254         // We also need 1 byte for '-' and 1 for '\0'.
1255         static char buff[22];
1256         int64_t n = n0 < 0 ? -n0 : n0;
1257         char *p = buff + sizeof(buff);
1258
1259         *--p = '\0';
1260         do {
1261                 *--p = '0' + (int)(n % 10);
1262                 n /= 10;
1263         } while (n > 0);
1264         if (n0 < 0)
1265                 *--p = '-';
1266         return p;
1267 }