1caf54c4fSMartin Matuska /*- 2caf54c4fSMartin Matuska * Copyright (c) 2003-2007 Tim Kientzle 3fd082e96SMartin Matuska * Copyright (c) 2012 Michihiro NAKAJIMA 4caf54c4fSMartin Matuska * All rights reserved. 5caf54c4fSMartin Matuska * 6caf54c4fSMartin Matuska * Redistribution and use in source and binary forms, with or without 7caf54c4fSMartin Matuska * modification, are permitted provided that the following conditions 8caf54c4fSMartin Matuska * are met: 9caf54c4fSMartin Matuska * 1. Redistributions of source code must retain the above copyright 10caf54c4fSMartin Matuska * notice, this list of conditions and the following disclaimer. 11caf54c4fSMartin Matuska * 2. Redistributions in binary form must reproduce the above copyright 12caf54c4fSMartin Matuska * notice, this list of conditions and the following disclaimer in the 13caf54c4fSMartin Matuska * documentation and/or other materials provided with the distribution. 14caf54c4fSMartin Matuska * 15caf54c4fSMartin Matuska * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 16caf54c4fSMartin Matuska * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17caf54c4fSMartin Matuska * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18caf54c4fSMartin Matuska * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 19caf54c4fSMartin Matuska * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20caf54c4fSMartin Matuska * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21caf54c4fSMartin Matuska * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22caf54c4fSMartin Matuska * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23caf54c4fSMartin Matuska * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24caf54c4fSMartin Matuska * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25caf54c4fSMartin Matuska */ 26caf54c4fSMartin Matuska 27caf54c4fSMartin Matuska #include "bsdtar_platform.h" 286c22d9efSMartin Matuska __FBSDID("$FreeBSD$"); 29caf54c4fSMartin Matuska 30caf54c4fSMartin Matuska #ifdef HAVE_SYS_TYPES_H 31caf54c4fSMartin Matuska #include <sys/types.h> 32caf54c4fSMartin Matuska #endif 33caf54c4fSMartin Matuska #ifdef HAVE_SYS_STAT_H 34caf54c4fSMartin Matuska #include <sys/stat.h> 35caf54c4fSMartin Matuska #endif 36caf54c4fSMartin Matuska #ifdef HAVE_ATTR_XATTR_H 37caf54c4fSMartin Matuska #include <attr/xattr.h> 38caf54c4fSMartin Matuska #endif 39caf54c4fSMartin Matuska #ifdef HAVE_ERRNO_H 40caf54c4fSMartin Matuska #include <errno.h> 41caf54c4fSMartin Matuska #endif 42caf54c4fSMartin Matuska #ifdef HAVE_FCNTL_H 43caf54c4fSMartin Matuska #include <fcntl.h> 44caf54c4fSMartin Matuska #endif 45caf54c4fSMartin Matuska #ifdef HAVE_GRP_H 46caf54c4fSMartin Matuska #include <grp.h> 47caf54c4fSMartin Matuska #endif 48caf54c4fSMartin Matuska #ifdef HAVE_IO_H 49caf54c4fSMartin Matuska #include <io.h> 50caf54c4fSMartin Matuska #endif 516c95142eSMartin Matuska #ifdef HAVE_LIBGEN_H 526c95142eSMartin Matuska #include <libgen.h> 536c95142eSMartin Matuska #endif 54caf54c4fSMartin Matuska #ifdef HAVE_LIMITS_H 55caf54c4fSMartin Matuska #include <limits.h> 56caf54c4fSMartin Matuska #endif 576c95142eSMartin Matuska #ifdef HAVE_PATHS_H 586c95142eSMartin Matuska #include <paths.h> 596c95142eSMartin Matuska #endif 60caf54c4fSMartin Matuska #ifdef HAVE_PWD_H 61caf54c4fSMartin Matuska #include <pwd.h> 62caf54c4fSMartin Matuska #endif 63caf54c4fSMartin Matuska #ifdef HAVE_STDINT_H 64caf54c4fSMartin Matuska #include <stdint.h> 65caf54c4fSMartin Matuska #endif 66caf54c4fSMartin Matuska #include <stdio.h> 67caf54c4fSMartin Matuska #ifdef HAVE_STDLIB_H 68caf54c4fSMartin Matuska #include <stdlib.h> 69caf54c4fSMartin Matuska #endif 70caf54c4fSMartin Matuska #ifdef HAVE_STRING_H 71caf54c4fSMartin Matuska #include <string.h> 72caf54c4fSMartin Matuska #endif 73caf54c4fSMartin Matuska #ifdef HAVE_UNISTD_H 74caf54c4fSMartin Matuska #include <unistd.h> 75caf54c4fSMartin Matuska #endif 76caf54c4fSMartin Matuska 77caf54c4fSMartin Matuska #include "bsdtar.h" 78caf54c4fSMartin Matuska #include "err.h" 79caf54c4fSMartin Matuska #include "line_reader.h" 80caf54c4fSMartin Matuska 81caf54c4fSMartin Matuska #ifndef O_BINARY 82caf54c4fSMartin Matuska #define O_BINARY 0 83caf54c4fSMartin Matuska #endif 84caf54c4fSMartin Matuska 85caf54c4fSMartin Matuska struct archive_dir_entry { 86caf54c4fSMartin Matuska struct archive_dir_entry *next; 87caf54c4fSMartin Matuska time_t mtime_sec; 88caf54c4fSMartin Matuska int mtime_nsec; 89caf54c4fSMartin Matuska char *name; 90caf54c4fSMartin Matuska }; 91caf54c4fSMartin Matuska 92caf54c4fSMartin Matuska struct archive_dir { 93caf54c4fSMartin Matuska struct archive_dir_entry *head, *tail; 94caf54c4fSMartin Matuska }; 95caf54c4fSMartin Matuska 96caf54c4fSMartin Matuska static int append_archive(struct bsdtar *, struct archive *, 97caf54c4fSMartin Matuska struct archive *ina); 98caf54c4fSMartin Matuska static int append_archive_filename(struct bsdtar *, 99caf54c4fSMartin Matuska struct archive *, const char *fname); 100caf54c4fSMartin Matuska static void archive_names_from_file(struct bsdtar *bsdtar, 101caf54c4fSMartin Matuska struct archive *a); 102fd082e96SMartin Matuska static int copy_file_data_block(struct bsdtar *, 103fd082e96SMartin Matuska struct archive *a, struct archive *, 104fd082e96SMartin Matuska struct archive_entry *); 105fd082e96SMartin Matuska static void excluded_callback(struct archive *, void *, 106fd082e96SMartin Matuska struct archive_entry *); 107caf54c4fSMartin Matuska static void report_write(struct bsdtar *, struct archive *, 108caf54c4fSMartin Matuska struct archive_entry *, int64_t progress); 109caf54c4fSMartin Matuska static void test_for_append(struct bsdtar *); 110fd082e96SMartin Matuska static int metadata_filter(struct archive *, void *, 111fd082e96SMartin Matuska struct archive_entry *); 112caf54c4fSMartin Matuska static void write_archive(struct archive *, struct bsdtar *); 1136c95142eSMartin Matuska static void write_entry(struct bsdtar *, struct archive *, 1146c95142eSMartin Matuska struct archive_entry *); 1156c95142eSMartin Matuska static void write_file(struct bsdtar *, struct archive *, 116caf54c4fSMartin Matuska struct archive_entry *); 117caf54c4fSMartin Matuska static void write_hierarchy(struct bsdtar *, struct archive *, 118caf54c4fSMartin Matuska const char *); 119caf54c4fSMartin Matuska 120caf54c4fSMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 121caf54c4fSMartin Matuska /* Not a full lseek() emulation, but enough for our needs here. */ 122caf54c4fSMartin Matuska static int 123caf54c4fSMartin Matuska seek_file(int fd, int64_t offset, int whence) 124caf54c4fSMartin Matuska { 125caf54c4fSMartin Matuska LARGE_INTEGER distance; 126caf54c4fSMartin Matuska (void)whence; /* UNUSED */ 127caf54c4fSMartin Matuska distance.QuadPart = offset; 128caf54c4fSMartin Matuska return (SetFilePointerEx((HANDLE)_get_osfhandle(fd), 129caf54c4fSMartin Matuska distance, NULL, FILE_BEGIN) ? 1 : -1); 130caf54c4fSMartin Matuska } 131caf54c4fSMartin Matuska #define open _open 132caf54c4fSMartin Matuska #define close _close 133caf54c4fSMartin Matuska #define read _read 134acc60b03SMartin Matuska #ifdef lseek 135acc60b03SMartin Matuska #undef lseek 136acc60b03SMartin Matuska #endif 137caf54c4fSMartin Matuska #define lseek seek_file 138caf54c4fSMartin Matuska #endif 139caf54c4fSMartin Matuska 140acc60b03SMartin Matuska static void 141acc60b03SMartin Matuska set_writer_options(struct bsdtar *bsdtar, struct archive *a) 142acc60b03SMartin Matuska { 143acc60b03SMartin Matuska const char *writer_options; 144acc60b03SMartin Matuska int r; 145acc60b03SMartin Matuska 146acc60b03SMartin Matuska writer_options = getenv(ENV_WRITER_OPTIONS); 147acc60b03SMartin Matuska if (writer_options != NULL) { 1487105995cSMartin Matuska size_t module_len = sizeof(IGNORE_WRONG_MODULE_NAME) - 1; 1497105995cSMartin Matuska size_t opt_len = strlen(writer_options) + 1; 150acc60b03SMartin Matuska char *p; 151acc60b03SMartin Matuska /* Set default write options. */ 1527105995cSMartin Matuska if ((p = malloc(module_len + opt_len)) == NULL) 153acc60b03SMartin Matuska lafe_errc(1, errno, "Out of memory"); 154acc60b03SMartin Matuska /* Prepend magic code to ignore options for 155acc60b03SMartin Matuska * a format or filters which are not added to 156acc60b03SMartin Matuska * the archive write object. */ 1577105995cSMartin Matuska memcpy(p, IGNORE_WRONG_MODULE_NAME, module_len); 1587105995cSMartin Matuska memcpy(p, writer_options, opt_len); 159acc60b03SMartin Matuska r = archive_write_set_options(a, p); 160acc60b03SMartin Matuska free(p); 161acc60b03SMartin Matuska if (r < ARCHIVE_WARN) 162acc60b03SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 163acc60b03SMartin Matuska else 164acc60b03SMartin Matuska archive_clear_error(a); 165acc60b03SMartin Matuska } 166acc60b03SMartin Matuska if (ARCHIVE_OK != archive_write_set_options(a, bsdtar->option_options)) 167acc60b03SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 168acc60b03SMartin Matuska } 169acc60b03SMartin Matuska 170acc60b03SMartin Matuska static void 171acc60b03SMartin Matuska set_reader_options(struct bsdtar *bsdtar, struct archive *a) 172acc60b03SMartin Matuska { 173acc60b03SMartin Matuska const char *reader_options; 174acc60b03SMartin Matuska int r; 175acc60b03SMartin Matuska 176acc60b03SMartin Matuska (void)bsdtar; /* UNUSED */ 177acc60b03SMartin Matuska 178acc60b03SMartin Matuska reader_options = getenv(ENV_READER_OPTIONS); 179acc60b03SMartin Matuska if (reader_options != NULL) { 1807105995cSMartin Matuska size_t module_len = sizeof(IGNORE_WRONG_MODULE_NAME) - 1; 1817105995cSMartin Matuska size_t opt_len = strlen(reader_options) + 1; 182acc60b03SMartin Matuska char *p; 183acc60b03SMartin Matuska /* Set default write options. */ 1847105995cSMartin Matuska if ((p = malloc(module_len + opt_len)) == NULL) 185acc60b03SMartin Matuska if (p == NULL) 186acc60b03SMartin Matuska lafe_errc(1, errno, "Out of memory"); 187acc60b03SMartin Matuska /* Prepend magic code to ignore options for 188acc60b03SMartin Matuska * a format or filters which are not added to 189acc60b03SMartin Matuska * the archive write object. */ 1907105995cSMartin Matuska memcpy(p, IGNORE_WRONG_MODULE_NAME, module_len); 1917105995cSMartin Matuska memcpy(p, reader_options, opt_len); 192acc60b03SMartin Matuska r = archive_read_set_options(a, p); 193acc60b03SMartin Matuska free(p); 194acc60b03SMartin Matuska if (r < ARCHIVE_WARN) 195acc60b03SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 196acc60b03SMartin Matuska else 197acc60b03SMartin Matuska archive_clear_error(a); 198acc60b03SMartin Matuska } 199833a452eSMartin Matuska if (bsdtar->flags & OPTFLAG_IGNORE_ZEROS) 200833a452eSMartin Matuska if (archive_read_set_options(a, 201833a452eSMartin Matuska "read_concatenated_archives") != ARCHIVE_OK) 202833a452eSMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 203acc60b03SMartin Matuska } 204acc60b03SMartin Matuska 205caf54c4fSMartin Matuska void 206caf54c4fSMartin Matuska tar_mode_c(struct bsdtar *bsdtar) 207caf54c4fSMartin Matuska { 208caf54c4fSMartin Matuska struct archive *a; 209acc60b03SMartin Matuska const void *filter_name; 210caf54c4fSMartin Matuska int r; 211caf54c4fSMartin Matuska 212caf54c4fSMartin Matuska if (*bsdtar->argv == NULL && bsdtar->names_from_file == NULL) 213caf54c4fSMartin Matuska lafe_errc(1, 0, "no files or directories specified"); 214caf54c4fSMartin Matuska 215caf54c4fSMartin Matuska a = archive_write_new(); 216caf54c4fSMartin Matuska 217caf54c4fSMartin Matuska /* Support any format that the library supports. */ 218acc60b03SMartin Matuska if (cset_get_format(bsdtar->cset) == NULL) { 219caf54c4fSMartin Matuska r = archive_write_set_format_pax_restricted(a); 220acc60b03SMartin Matuska cset_set_format(bsdtar->cset, "pax restricted"); 221caf54c4fSMartin Matuska } else { 222acc60b03SMartin Matuska r = archive_write_set_format_by_name(a, 223acc60b03SMartin Matuska cset_get_format(bsdtar->cset)); 224caf54c4fSMartin Matuska } 225caf54c4fSMartin Matuska if (r != ARCHIVE_OK) { 226caf54c4fSMartin Matuska fprintf(stderr, "Can't use format %s: %s\n", 227acc60b03SMartin Matuska cset_get_format(bsdtar->cset), 228caf54c4fSMartin Matuska archive_error_string(a)); 229caf54c4fSMartin Matuska usage(); 230caf54c4fSMartin Matuska } 231caf54c4fSMartin Matuska 232caf54c4fSMartin Matuska archive_write_set_bytes_per_block(a, bsdtar->bytes_per_block); 2336c95142eSMartin Matuska archive_write_set_bytes_in_last_block(a, bsdtar->bytes_in_last_block); 234caf54c4fSMartin Matuska 235acc60b03SMartin Matuska r = cset_write_add_filters(bsdtar->cset, a, &filter_name); 236acc60b03SMartin Matuska if (r < ARCHIVE_WARN) { 237acc60b03SMartin Matuska lafe_errc(1, 0, "Unsupported compression option --%s", 238acc60b03SMartin Matuska (const char *)filter_name); 239caf54c4fSMartin Matuska } 240caf54c4fSMartin Matuska 241acc60b03SMartin Matuska set_writer_options(bsdtar, a); 242cdf63a70SMartin Matuska if (bsdtar->passphrase != NULL) 243cdf63a70SMartin Matuska r = archive_write_set_passphrase(a, bsdtar->passphrase); 244cdf63a70SMartin Matuska else 245cdf63a70SMartin Matuska r = archive_write_set_passphrase_callback(a, bsdtar, 246cdf63a70SMartin Matuska &passphrase_callback); 247cdf63a70SMartin Matuska if (r != ARCHIVE_OK) 248cdf63a70SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 249acc60b03SMartin Matuska if (ARCHIVE_OK != archive_write_open_filename(a, bsdtar->filename)) 250caf54c4fSMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 251caf54c4fSMartin Matuska write_archive(a, bsdtar); 252caf54c4fSMartin Matuska } 253caf54c4fSMartin Matuska 254caf54c4fSMartin Matuska /* 255caf54c4fSMartin Matuska * Same as 'c', except we only support tar or empty formats in 256caf54c4fSMartin Matuska * uncompressed files on disk. 257caf54c4fSMartin Matuska */ 258caf54c4fSMartin Matuska void 259caf54c4fSMartin Matuska tar_mode_r(struct bsdtar *bsdtar) 260caf54c4fSMartin Matuska { 261caf54c4fSMartin Matuska int64_t end_offset; 262caf54c4fSMartin Matuska int format; 263caf54c4fSMartin Matuska struct archive *a; 264caf54c4fSMartin Matuska struct archive_entry *entry; 265caf54c4fSMartin Matuska int r; 266caf54c4fSMartin Matuska 267caf54c4fSMartin Matuska /* Sanity-test some arguments and the file. */ 268caf54c4fSMartin Matuska test_for_append(bsdtar); 269caf54c4fSMartin Matuska 270caf54c4fSMartin Matuska format = ARCHIVE_FORMAT_TAR_PAX_RESTRICTED; 271caf54c4fSMartin Matuska 272caf54c4fSMartin Matuska #if defined(__BORLANDC__) 273caf54c4fSMartin Matuska bsdtar->fd = open(bsdtar->filename, O_RDWR | O_CREAT | O_BINARY); 274caf54c4fSMartin Matuska #else 275caf54c4fSMartin Matuska bsdtar->fd = open(bsdtar->filename, O_RDWR | O_CREAT | O_BINARY, 0666); 276caf54c4fSMartin Matuska #endif 277caf54c4fSMartin Matuska if (bsdtar->fd < 0) 278caf54c4fSMartin Matuska lafe_errc(1, errno, 279caf54c4fSMartin Matuska "Cannot open %s", bsdtar->filename); 280caf54c4fSMartin Matuska 281caf54c4fSMartin Matuska a = archive_read_new(); 2826c95142eSMartin Matuska archive_read_support_filter_all(a); 2837ceb9433SMartin Matuska archive_read_support_format_empty(a); 284caf54c4fSMartin Matuska archive_read_support_format_tar(a); 285caf54c4fSMartin Matuska archive_read_support_format_gnutar(a); 286acc60b03SMartin Matuska set_reader_options(bsdtar, a); 287caf54c4fSMartin Matuska r = archive_read_open_fd(a, bsdtar->fd, 10240); 288caf54c4fSMartin Matuska if (r != ARCHIVE_OK) 289caf54c4fSMartin Matuska lafe_errc(1, archive_errno(a), 290caf54c4fSMartin Matuska "Can't read archive %s: %s", bsdtar->filename, 291caf54c4fSMartin Matuska archive_error_string(a)); 292caf54c4fSMartin Matuska while (0 == archive_read_next_header(a, &entry)) { 293acc60b03SMartin Matuska if (archive_filter_code(a, 0) != ARCHIVE_FILTER_NONE) { 2946c95142eSMartin Matuska archive_read_free(a); 295caf54c4fSMartin Matuska close(bsdtar->fd); 296caf54c4fSMartin Matuska lafe_errc(1, 0, 297caf54c4fSMartin Matuska "Cannot append to compressed archive."); 298caf54c4fSMartin Matuska } 299caf54c4fSMartin Matuska /* Keep going until we hit end-of-archive */ 300caf54c4fSMartin Matuska format = archive_format(a); 301caf54c4fSMartin Matuska } 302caf54c4fSMartin Matuska 303caf54c4fSMartin Matuska end_offset = archive_read_header_position(a); 3046c95142eSMartin Matuska archive_read_free(a); 305caf54c4fSMartin Matuska 306caf54c4fSMartin Matuska /* Re-open archive for writing */ 307caf54c4fSMartin Matuska a = archive_write_new(); 308caf54c4fSMartin Matuska /* 309caf54c4fSMartin Matuska * Set the format to be used for writing. To allow people to 310caf54c4fSMartin Matuska * extend empty files, we need to allow them to specify the format, 311caf54c4fSMartin Matuska * which opens the possibility that they will specify a format that 312caf54c4fSMartin Matuska * doesn't match the existing format. Hence, the following bit 313caf54c4fSMartin Matuska * of arcane ugliness. 314caf54c4fSMartin Matuska */ 315caf54c4fSMartin Matuska 316acc60b03SMartin Matuska if (cset_get_format(bsdtar->cset) != NULL) { 317caf54c4fSMartin Matuska /* If the user requested a format, use that, but ... */ 318caf54c4fSMartin Matuska archive_write_set_format_by_name(a, 319acc60b03SMartin Matuska cset_get_format(bsdtar->cset)); 320caf54c4fSMartin Matuska /* ... complain if it's not compatible. */ 321caf54c4fSMartin Matuska format &= ARCHIVE_FORMAT_BASE_MASK; 322caf54c4fSMartin Matuska if (format != (int)(archive_format(a) & ARCHIVE_FORMAT_BASE_MASK) 323caf54c4fSMartin Matuska && format != ARCHIVE_FORMAT_EMPTY) { 324caf54c4fSMartin Matuska lafe_errc(1, 0, 325caf54c4fSMartin Matuska "Format %s is incompatible with the archive %s.", 326acc60b03SMartin Matuska cset_get_format(bsdtar->cset), bsdtar->filename); 327caf54c4fSMartin Matuska } 328caf54c4fSMartin Matuska } else { 329caf54c4fSMartin Matuska /* 330caf54c4fSMartin Matuska * Just preserve the current format, with a little care 331caf54c4fSMartin Matuska * for formats that libarchive can't write. 332caf54c4fSMartin Matuska */ 333caf54c4fSMartin Matuska if (format == ARCHIVE_FORMAT_EMPTY) 334caf54c4fSMartin Matuska format = ARCHIVE_FORMAT_TAR_PAX_RESTRICTED; 335caf54c4fSMartin Matuska archive_write_set_format(a, format); 336caf54c4fSMartin Matuska } 337caf54c4fSMartin Matuska if (lseek(bsdtar->fd, end_offset, SEEK_SET) < 0) 338caf54c4fSMartin Matuska lafe_errc(1, errno, "Could not seek to archive end"); 339acc60b03SMartin Matuska set_writer_options(bsdtar, a); 340caf54c4fSMartin Matuska if (ARCHIVE_OK != archive_write_open_fd(a, bsdtar->fd)) 341caf54c4fSMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 342caf54c4fSMartin Matuska 343caf54c4fSMartin Matuska write_archive(a, bsdtar); /* XXX check return val XXX */ 344caf54c4fSMartin Matuska 345caf54c4fSMartin Matuska close(bsdtar->fd); 346caf54c4fSMartin Matuska bsdtar->fd = -1; 347caf54c4fSMartin Matuska } 348caf54c4fSMartin Matuska 349caf54c4fSMartin Matuska void 350caf54c4fSMartin Matuska tar_mode_u(struct bsdtar *bsdtar) 351caf54c4fSMartin Matuska { 352caf54c4fSMartin Matuska int64_t end_offset; 353caf54c4fSMartin Matuska struct archive *a; 354caf54c4fSMartin Matuska struct archive_entry *entry; 355caf54c4fSMartin Matuska int format; 356caf54c4fSMartin Matuska struct archive_dir_entry *p; 357caf54c4fSMartin Matuska struct archive_dir archive_dir; 358caf54c4fSMartin Matuska 359caf54c4fSMartin Matuska bsdtar->archive_dir = &archive_dir; 360caf54c4fSMartin Matuska memset(&archive_dir, 0, sizeof(archive_dir)); 361caf54c4fSMartin Matuska 362caf54c4fSMartin Matuska format = ARCHIVE_FORMAT_TAR_PAX_RESTRICTED; 363caf54c4fSMartin Matuska 364caf54c4fSMartin Matuska /* Sanity-test some arguments and the file. */ 365caf54c4fSMartin Matuska test_for_append(bsdtar); 366caf54c4fSMartin Matuska 367caf54c4fSMartin Matuska bsdtar->fd = open(bsdtar->filename, O_RDWR | O_BINARY); 368caf54c4fSMartin Matuska if (bsdtar->fd < 0) 369caf54c4fSMartin Matuska lafe_errc(1, errno, 370caf54c4fSMartin Matuska "Cannot open %s", bsdtar->filename); 371caf54c4fSMartin Matuska 372caf54c4fSMartin Matuska a = archive_read_new(); 3736c95142eSMartin Matuska archive_read_support_filter_all(a); 374caf54c4fSMartin Matuska archive_read_support_format_tar(a); 375caf54c4fSMartin Matuska archive_read_support_format_gnutar(a); 376acc60b03SMartin Matuska set_reader_options(bsdtar, a); 3776c95142eSMartin Matuska if (archive_read_open_fd(a, bsdtar->fd, bsdtar->bytes_per_block) 3786c95142eSMartin Matuska != ARCHIVE_OK) { 379caf54c4fSMartin Matuska lafe_errc(1, 0, 380caf54c4fSMartin Matuska "Can't open %s: %s", bsdtar->filename, 381caf54c4fSMartin Matuska archive_error_string(a)); 382caf54c4fSMartin Matuska } 383caf54c4fSMartin Matuska 384caf54c4fSMartin Matuska /* Build a list of all entries and their recorded mod times. */ 385caf54c4fSMartin Matuska while (0 == archive_read_next_header(a, &entry)) { 386acc60b03SMartin Matuska if (archive_filter_code(a, 0) != ARCHIVE_FILTER_NONE) { 3876c95142eSMartin Matuska archive_read_free(a); 388caf54c4fSMartin Matuska close(bsdtar->fd); 389caf54c4fSMartin Matuska lafe_errc(1, 0, 390caf54c4fSMartin Matuska "Cannot append to compressed archive."); 391caf54c4fSMartin Matuska } 392fd082e96SMartin Matuska if (archive_match_exclude_entry(bsdtar->matching, 393fd082e96SMartin Matuska ARCHIVE_MATCH_MTIME | ARCHIVE_MATCH_OLDER | 394fd082e96SMartin Matuska ARCHIVE_MATCH_EQUAL, entry) != ARCHIVE_OK) 395fd082e96SMartin Matuska lafe_errc(1, 0, "Error : %s", 396fd082e96SMartin Matuska archive_error_string(bsdtar->matching)); 397caf54c4fSMartin Matuska /* Record the last format determination we see */ 398caf54c4fSMartin Matuska format = archive_format(a); 399caf54c4fSMartin Matuska /* Keep going until we hit end-of-archive */ 400caf54c4fSMartin Matuska } 401caf54c4fSMartin Matuska 402caf54c4fSMartin Matuska end_offset = archive_read_header_position(a); 4036c95142eSMartin Matuska archive_read_free(a); 404caf54c4fSMartin Matuska 405caf54c4fSMartin Matuska /* Re-open archive for writing. */ 406caf54c4fSMartin Matuska a = archive_write_new(); 407caf54c4fSMartin Matuska /* 4086c95142eSMartin Matuska * Set format to same one auto-detected above. 409caf54c4fSMartin Matuska */ 410caf54c4fSMartin Matuska archive_write_set_format(a, format); 411caf54c4fSMartin Matuska archive_write_set_bytes_per_block(a, bsdtar->bytes_per_block); 4126c95142eSMartin Matuska archive_write_set_bytes_in_last_block(a, bsdtar->bytes_in_last_block); 4136c95142eSMartin Matuska 414caf54c4fSMartin Matuska if (lseek(bsdtar->fd, end_offset, SEEK_SET) < 0) 415caf54c4fSMartin Matuska lafe_errc(1, errno, "Could not seek to archive end"); 416acc60b03SMartin Matuska set_writer_options(bsdtar, a); 417caf54c4fSMartin Matuska if (ARCHIVE_OK != archive_write_open_fd(a, bsdtar->fd)) 418caf54c4fSMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 419caf54c4fSMartin Matuska 420caf54c4fSMartin Matuska write_archive(a, bsdtar); 421caf54c4fSMartin Matuska 422caf54c4fSMartin Matuska close(bsdtar->fd); 423caf54c4fSMartin Matuska bsdtar->fd = -1; 424caf54c4fSMartin Matuska 425caf54c4fSMartin Matuska while (bsdtar->archive_dir->head != NULL) { 426caf54c4fSMartin Matuska p = bsdtar->archive_dir->head->next; 427caf54c4fSMartin Matuska free(bsdtar->archive_dir->head->name); 428caf54c4fSMartin Matuska free(bsdtar->archive_dir->head); 429caf54c4fSMartin Matuska bsdtar->archive_dir->head = p; 430caf54c4fSMartin Matuska } 431caf54c4fSMartin Matuska bsdtar->archive_dir->tail = NULL; 432caf54c4fSMartin Matuska } 433caf54c4fSMartin Matuska 434caf54c4fSMartin Matuska 435caf54c4fSMartin Matuska /* 436caf54c4fSMartin Matuska * Write user-specified files/dirs to opened archive. 437caf54c4fSMartin Matuska */ 438caf54c4fSMartin Matuska static void 439caf54c4fSMartin Matuska write_archive(struct archive *a, struct bsdtar *bsdtar) 440caf54c4fSMartin Matuska { 441caf54c4fSMartin Matuska const char *arg; 442caf54c4fSMartin Matuska struct archive_entry *entry, *sparse_entry; 443caf54c4fSMartin Matuska 4446c95142eSMartin Matuska /* Choose a suitable copy buffer size */ 4456c95142eSMartin Matuska bsdtar->buff_size = 64 * 1024; 4466c95142eSMartin Matuska while (bsdtar->buff_size < (size_t)bsdtar->bytes_per_block) 4476c95142eSMartin Matuska bsdtar->buff_size *= 2; 4486c95142eSMartin Matuska /* Try to compensate for space we'll lose to alignment. */ 4496c95142eSMartin Matuska bsdtar->buff_size += 16 * 1024; 4506c95142eSMartin Matuska 451caf54c4fSMartin Matuska /* Allocate a buffer for file data. */ 4526c95142eSMartin Matuska if ((bsdtar->buff = malloc(bsdtar->buff_size)) == NULL) 453caf54c4fSMartin Matuska lafe_errc(1, 0, "cannot allocate memory"); 454caf54c4fSMartin Matuska 455caf54c4fSMartin Matuska if ((bsdtar->resolver = archive_entry_linkresolver_new()) == NULL) 456caf54c4fSMartin Matuska lafe_errc(1, 0, "cannot create link resolver"); 457caf54c4fSMartin Matuska archive_entry_linkresolver_set_strategy(bsdtar->resolver, 458caf54c4fSMartin Matuska archive_format(a)); 459fd082e96SMartin Matuska 460fd082e96SMartin Matuska /* Create a read_disk object. */ 461caf54c4fSMartin Matuska if ((bsdtar->diskreader = archive_read_disk_new()) == NULL) 462caf54c4fSMartin Matuska lafe_errc(1, 0, "Cannot create read_disk object"); 463fd082e96SMartin Matuska /* Tell the read_disk how handle symlink. */ 464fd082e96SMartin Matuska switch (bsdtar->symlink_mode) { 465fd082e96SMartin Matuska case 'H': 466fd082e96SMartin Matuska archive_read_disk_set_symlink_hybrid(bsdtar->diskreader); 467fd082e96SMartin Matuska break; 468fd082e96SMartin Matuska case 'L': 469fd082e96SMartin Matuska archive_read_disk_set_symlink_logical(bsdtar->diskreader); 470fd082e96SMartin Matuska break; 471fd082e96SMartin Matuska default: 472fd082e96SMartin Matuska archive_read_disk_set_symlink_physical(bsdtar->diskreader); 473fd082e96SMartin Matuska break; 474fd082e96SMartin Matuska } 475fd082e96SMartin Matuska /* Register entry filters. */ 476fd082e96SMartin Matuska archive_read_disk_set_matching(bsdtar->diskreader, 477fd082e96SMartin Matuska bsdtar->matching, excluded_callback, bsdtar); 478fd082e96SMartin Matuska archive_read_disk_set_metadata_filter_callback( 479fd082e96SMartin Matuska bsdtar->diskreader, metadata_filter, bsdtar); 480fd082e96SMartin Matuska /* Set the behavior of archive_read_disk. */ 481fd082e96SMartin Matuska archive_read_disk_set_behavior(bsdtar->diskreader, 482fd082e96SMartin Matuska bsdtar->readdisk_flags); 483caf54c4fSMartin Matuska archive_read_disk_set_standard_lookup(bsdtar->diskreader); 484caf54c4fSMartin Matuska 485caf54c4fSMartin Matuska if (bsdtar->names_from_file != NULL) 486caf54c4fSMartin Matuska archive_names_from_file(bsdtar, a); 487caf54c4fSMartin Matuska 488caf54c4fSMartin Matuska while (*bsdtar->argv) { 489caf54c4fSMartin Matuska arg = *bsdtar->argv; 490caf54c4fSMartin Matuska if (arg[0] == '-' && arg[1] == 'C') { 491caf54c4fSMartin Matuska arg += 2; 492caf54c4fSMartin Matuska if (*arg == '\0') { 493caf54c4fSMartin Matuska bsdtar->argv++; 494caf54c4fSMartin Matuska arg = *bsdtar->argv; 495caf54c4fSMartin Matuska if (arg == NULL) { 496caf54c4fSMartin Matuska lafe_warnc(0, "%s", 497caf54c4fSMartin Matuska "Missing argument for -C"); 498caf54c4fSMartin Matuska bsdtar->return_value = 1; 499caf54c4fSMartin Matuska goto cleanup; 500caf54c4fSMartin Matuska } 5016c95142eSMartin Matuska if (*arg == '\0') { 5026c95142eSMartin Matuska lafe_warnc(0, 5036c95142eSMartin Matuska "Meaningless argument for -C: ''"); 5046c95142eSMartin Matuska bsdtar->return_value = 1; 5056c95142eSMartin Matuska goto cleanup; 5066c95142eSMartin Matuska } 507caf54c4fSMartin Matuska } 508caf54c4fSMartin Matuska set_chdir(bsdtar, arg); 509caf54c4fSMartin Matuska } else { 510f1951fd7SMartin Matuska if (*arg != '/') 511caf54c4fSMartin Matuska do_chdir(bsdtar); /* Handle a deferred -C */ 512caf54c4fSMartin Matuska if (*arg == '@') { 513caf54c4fSMartin Matuska if (append_archive_filename(bsdtar, a, 514caf54c4fSMartin Matuska arg + 1) != 0) 515caf54c4fSMartin Matuska break; 516caf54c4fSMartin Matuska } else 517caf54c4fSMartin Matuska write_hierarchy(bsdtar, a, arg); 518caf54c4fSMartin Matuska } 519caf54c4fSMartin Matuska bsdtar->argv++; 520caf54c4fSMartin Matuska } 521caf54c4fSMartin Matuska 522fd082e96SMartin Matuska archive_read_disk_set_matching(bsdtar->diskreader, NULL, NULL, NULL); 523fd082e96SMartin Matuska archive_read_disk_set_metadata_filter_callback( 524fd082e96SMartin Matuska bsdtar->diskreader, NULL, NULL); 525caf54c4fSMartin Matuska entry = NULL; 526caf54c4fSMartin Matuska archive_entry_linkify(bsdtar->resolver, &entry, &sparse_entry); 527caf54c4fSMartin Matuska while (entry != NULL) { 528fd082e96SMartin Matuska int r; 529fd082e96SMartin Matuska struct archive_entry *entry2; 530fd082e96SMartin Matuska struct archive *disk = bsdtar->diskreader; 531fd082e96SMartin Matuska 532fd082e96SMartin Matuska /* 533007dbc1fSMartin Matuska * This tricky code here is to correctly read the contents 534fd082e96SMartin Matuska * of the entry because the disk reader bsdtar->diskreader 535fd082e96SMartin Matuska * is pointing at does not have any information about the 536fd082e96SMartin Matuska * entry by this time and using archive_read_data_block() 537fd082e96SMartin Matuska * with the disk reader consequently must fail. And we 538fd082e96SMartin Matuska * have to re-open the entry to read the contents. 539fd082e96SMartin Matuska */ 540fd082e96SMartin Matuska /* TODO: Work with -C option as well. */ 541fd082e96SMartin Matuska r = archive_read_disk_open(disk, 542fd082e96SMartin Matuska archive_entry_sourcepath(entry)); 543fd082e96SMartin Matuska if (r != ARCHIVE_OK) { 544fd082e96SMartin Matuska lafe_warnc(archive_errno(disk), 545fd082e96SMartin Matuska "%s", archive_error_string(disk)); 546fd082e96SMartin Matuska bsdtar->return_value = 1; 547276f481dSMartin Matuska goto next_entry; 548fd082e96SMartin Matuska } 549fd082e96SMartin Matuska 550fd082e96SMartin Matuska /* 551fd082e96SMartin Matuska * Invoke archive_read_next_header2() to work 552fd082e96SMartin Matuska * archive_read_data_block(), which is called via write_file(), 553fd082e96SMartin Matuska * without failure. 554fd082e96SMartin Matuska */ 555fd082e96SMartin Matuska entry2 = archive_entry_new(); 556fd082e96SMartin Matuska r = archive_read_next_header2(disk, entry2); 557fd082e96SMartin Matuska archive_entry_free(entry2); 558fd082e96SMartin Matuska if (r != ARCHIVE_OK) { 559fd082e96SMartin Matuska lafe_warnc(archive_errno(disk), 560fd082e96SMartin Matuska "%s", archive_error_string(disk)); 561fd082e96SMartin Matuska if (r == ARCHIVE_FATAL) 562fd082e96SMartin Matuska bsdtar->return_value = 1; 563fd082e96SMartin Matuska archive_read_close(disk); 564276f481dSMartin Matuska goto next_entry; 565fd082e96SMartin Matuska } 566fd082e96SMartin Matuska 5676c95142eSMartin Matuska write_file(bsdtar, a, entry); 568fd082e96SMartin Matuska archive_read_close(disk); 569276f481dSMartin Matuska next_entry: 570276f481dSMartin Matuska archive_entry_free(entry); 571caf54c4fSMartin Matuska entry = NULL; 572caf54c4fSMartin Matuska archive_entry_linkify(bsdtar->resolver, &entry, &sparse_entry); 573caf54c4fSMartin Matuska } 574caf54c4fSMartin Matuska 575caf54c4fSMartin Matuska if (archive_write_close(a)) { 576caf54c4fSMartin Matuska lafe_warnc(0, "%s", archive_error_string(a)); 577caf54c4fSMartin Matuska bsdtar->return_value = 1; 578caf54c4fSMartin Matuska } 579caf54c4fSMartin Matuska 580caf54c4fSMartin Matuska cleanup: 581caf54c4fSMartin Matuska /* Free file data buffer. */ 582caf54c4fSMartin Matuska free(bsdtar->buff); 583caf54c4fSMartin Matuska archive_entry_linkresolver_free(bsdtar->resolver); 584caf54c4fSMartin Matuska bsdtar->resolver = NULL; 5856c95142eSMartin Matuska archive_read_free(bsdtar->diskreader); 586caf54c4fSMartin Matuska bsdtar->diskreader = NULL; 587caf54c4fSMartin Matuska 58864287048SMartin Matuska if (bsdtar->flags & OPTFLAG_TOTALS) { 589caf54c4fSMartin Matuska fprintf(stderr, "Total bytes written: %s\n", 590acc60b03SMartin Matuska tar_i64toa(archive_filter_bytes(a, -1))); 591caf54c4fSMartin Matuska } 592caf54c4fSMartin Matuska 5936c95142eSMartin Matuska archive_write_free(a); 594caf54c4fSMartin Matuska } 595caf54c4fSMartin Matuska 596caf54c4fSMartin Matuska /* 597caf54c4fSMartin Matuska * Archive names specified in file. 598caf54c4fSMartin Matuska * 599caf54c4fSMartin Matuska * Unless --null was specified, a line containing exactly "-C" will 600caf54c4fSMartin Matuska * cause the next line to be a directory to pass to chdir(). If 601caf54c4fSMartin Matuska * --null is specified, then a line "-C" is just another filename. 602caf54c4fSMartin Matuska */ 603caf54c4fSMartin Matuska static void 604caf54c4fSMartin Matuska archive_names_from_file(struct bsdtar *bsdtar, struct archive *a) 605caf54c4fSMartin Matuska { 606caf54c4fSMartin Matuska struct lafe_line_reader *lr; 607caf54c4fSMartin Matuska const char *line; 608caf54c4fSMartin Matuska 609caf54c4fSMartin Matuska bsdtar->next_line_is_dir = 0; 610caf54c4fSMartin Matuska 61164287048SMartin Matuska lr = lafe_line_reader(bsdtar->names_from_file, 61264287048SMartin Matuska (bsdtar->flags & OPTFLAG_NULL)); 613caf54c4fSMartin Matuska while ((line = lafe_line_reader_next(lr)) != NULL) { 614caf54c4fSMartin Matuska if (bsdtar->next_line_is_dir) { 6156c95142eSMartin Matuska if (*line != '\0') 616caf54c4fSMartin Matuska set_chdir(bsdtar, line); 6176c95142eSMartin Matuska else { 6186c95142eSMartin Matuska lafe_warnc(0, 6196c95142eSMartin Matuska "Meaningless argument for -C: ''"); 6206c95142eSMartin Matuska bsdtar->return_value = 1; 6216c95142eSMartin Matuska } 622caf54c4fSMartin Matuska bsdtar->next_line_is_dir = 0; 62364287048SMartin Matuska } else if (((bsdtar->flags & OPTFLAG_NULL) == 0) && 62464287048SMartin Matuska strcmp(line, "-C") == 0) 625caf54c4fSMartin Matuska bsdtar->next_line_is_dir = 1; 626caf54c4fSMartin Matuska else { 627caf54c4fSMartin Matuska if (*line != '/') 628caf54c4fSMartin Matuska do_chdir(bsdtar); /* Handle a deferred -C */ 629caf54c4fSMartin Matuska write_hierarchy(bsdtar, a, line); 630caf54c4fSMartin Matuska } 631caf54c4fSMartin Matuska } 632caf54c4fSMartin Matuska lafe_line_reader_free(lr); 633caf54c4fSMartin Matuska if (bsdtar->next_line_is_dir) 634caf54c4fSMartin Matuska lafe_errc(1, errno, 635caf54c4fSMartin Matuska "Unexpected end of filename list; " 636caf54c4fSMartin Matuska "directory expected after -C"); 637caf54c4fSMartin Matuska } 638caf54c4fSMartin Matuska 639caf54c4fSMartin Matuska /* 640caf54c4fSMartin Matuska * Copy from specified archive to current archive. Returns non-zero 641caf54c4fSMartin Matuska * for write errors (which force us to terminate the entire archiving 642caf54c4fSMartin Matuska * operation). If there are errors reading the input archive, we set 643caf54c4fSMartin Matuska * bsdtar->return_value but return zero, so the overall archiving 644caf54c4fSMartin Matuska * operation will complete and return non-zero. 645caf54c4fSMartin Matuska */ 646caf54c4fSMartin Matuska static int 647caf54c4fSMartin Matuska append_archive_filename(struct bsdtar *bsdtar, struct archive *a, 6486c95142eSMartin Matuska const char *raw_filename) 649caf54c4fSMartin Matuska { 650caf54c4fSMartin Matuska struct archive *ina; 6516c95142eSMartin Matuska const char *filename = raw_filename; 652caf54c4fSMartin Matuska int rc; 653caf54c4fSMartin Matuska 654caf54c4fSMartin Matuska if (strcmp(filename, "-") == 0) 655caf54c4fSMartin Matuska filename = NULL; /* Library uses NULL for stdio. */ 656caf54c4fSMartin Matuska 657caf54c4fSMartin Matuska ina = archive_read_new(); 658caf54c4fSMartin Matuska archive_read_support_format_all(ina); 6596c95142eSMartin Matuska archive_read_support_filter_all(ina); 660cdf63a70SMartin Matuska set_reader_options(bsdtar, ina); 661cdf63a70SMartin Matuska archive_read_set_options(ina, "mtree:checkfs"); 662cdf63a70SMartin Matuska if (bsdtar->passphrase != NULL) 663cdf63a70SMartin Matuska rc = archive_read_add_passphrase(a, bsdtar->passphrase); 664cdf63a70SMartin Matuska else 665cdf63a70SMartin Matuska rc = archive_read_set_passphrase_callback(ina, bsdtar, 666cdf63a70SMartin Matuska &passphrase_callback); 667cdf63a70SMartin Matuska if (rc != ARCHIVE_OK) 668cdf63a70SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 669acc60b03SMartin Matuska if (archive_read_open_filename(ina, filename, 670acc60b03SMartin Matuska bsdtar->bytes_per_block)) { 671caf54c4fSMartin Matuska lafe_warnc(0, "%s", archive_error_string(ina)); 672caf54c4fSMartin Matuska bsdtar->return_value = 1; 673caf54c4fSMartin Matuska return (0); 674caf54c4fSMartin Matuska } 675caf54c4fSMartin Matuska 676caf54c4fSMartin Matuska rc = append_archive(bsdtar, a, ina); 677caf54c4fSMartin Matuska 678caf54c4fSMartin Matuska if (rc != ARCHIVE_OK) { 679caf54c4fSMartin Matuska lafe_warnc(0, "Error reading archive %s: %s", 6806c95142eSMartin Matuska raw_filename, archive_error_string(ina)); 681caf54c4fSMartin Matuska bsdtar->return_value = 1; 682caf54c4fSMartin Matuska } 6836c95142eSMartin Matuska archive_read_free(ina); 684caf54c4fSMartin Matuska 685caf54c4fSMartin Matuska return (rc); 686caf54c4fSMartin Matuska } 687caf54c4fSMartin Matuska 688caf54c4fSMartin Matuska static int 689caf54c4fSMartin Matuska append_archive(struct bsdtar *bsdtar, struct archive *a, struct archive *ina) 690caf54c4fSMartin Matuska { 691caf54c4fSMartin Matuska struct archive_entry *in_entry; 692caf54c4fSMartin Matuska int e; 693caf54c4fSMartin Matuska 694caf54c4fSMartin Matuska while (ARCHIVE_OK == (e = archive_read_next_header(ina, &in_entry))) { 695fd082e96SMartin Matuska if (archive_match_excluded(bsdtar->matching, in_entry)) 696caf54c4fSMartin Matuska continue; 697*64884e0dSMartin Matuska if(edit_pathname(bsdtar, in_entry)) 698*64884e0dSMartin Matuska continue; 69964287048SMartin Matuska if ((bsdtar->flags & OPTFLAG_INTERACTIVE) && 700caf54c4fSMartin Matuska !yes("copy '%s'", archive_entry_pathname(in_entry))) 701caf54c4fSMartin Matuska continue; 702cdf63a70SMartin Matuska if (bsdtar->verbose > 1) { 703cdf63a70SMartin Matuska safe_fprintf(stderr, "a "); 704cdf63a70SMartin Matuska list_item_verbose(bsdtar, stderr, in_entry); 705cdf63a70SMartin Matuska } else if (bsdtar->verbose > 0) 706caf54c4fSMartin Matuska safe_fprintf(stderr, "a %s", 707caf54c4fSMartin Matuska archive_entry_pathname(in_entry)); 708caf54c4fSMartin Matuska if (need_report()) 709caf54c4fSMartin Matuska report_write(bsdtar, a, in_entry, 0); 710caf54c4fSMartin Matuska 711caf54c4fSMartin Matuska e = archive_write_header(a, in_entry); 712caf54c4fSMartin Matuska if (e != ARCHIVE_OK) { 713caf54c4fSMartin Matuska if (!bsdtar->verbose) 714caf54c4fSMartin Matuska lafe_warnc(0, "%s: %s", 715caf54c4fSMartin Matuska archive_entry_pathname(in_entry), 716caf54c4fSMartin Matuska archive_error_string(a)); 717caf54c4fSMartin Matuska else 718caf54c4fSMartin Matuska fprintf(stderr, ": %s", archive_error_string(a)); 719caf54c4fSMartin Matuska } 720caf54c4fSMartin Matuska if (e == ARCHIVE_FATAL) 721caf54c4fSMartin Matuska exit(1); 722caf54c4fSMartin Matuska 723caf54c4fSMartin Matuska if (e >= ARCHIVE_WARN) { 724caf54c4fSMartin Matuska if (archive_entry_size(in_entry) == 0) 725caf54c4fSMartin Matuska archive_read_data_skip(ina); 726fd082e96SMartin Matuska else if (copy_file_data_block(bsdtar, a, ina, in_entry)) 727caf54c4fSMartin Matuska exit(1); 728caf54c4fSMartin Matuska } 729caf54c4fSMartin Matuska 730caf54c4fSMartin Matuska if (bsdtar->verbose) 731caf54c4fSMartin Matuska fprintf(stderr, "\n"); 732caf54c4fSMartin Matuska } 733caf54c4fSMartin Matuska 734caf54c4fSMartin Matuska return (e == ARCHIVE_EOF ? ARCHIVE_OK : e); 735caf54c4fSMartin Matuska } 736caf54c4fSMartin Matuska 737fd082e96SMartin Matuska /* Helper function to copy file to archive. */ 738caf54c4fSMartin Matuska static int 739fd082e96SMartin Matuska copy_file_data_block(struct bsdtar *bsdtar, struct archive *a, 740fd082e96SMartin Matuska struct archive *in_a, struct archive_entry *entry) 741caf54c4fSMartin Matuska { 742fd082e96SMartin Matuska size_t bytes_read; 743caf54c4fSMartin Matuska ssize_t bytes_written; 744fd082e96SMartin Matuska int64_t offset, progress = 0; 745fd082e96SMartin Matuska char *null_buff = NULL; 746fd082e96SMartin Matuska const void *buff; 747fd082e96SMartin Matuska int r; 748caf54c4fSMartin Matuska 749fd082e96SMartin Matuska while ((r = archive_read_data_block(in_a, &buff, 750fd082e96SMartin Matuska &bytes_read, &offset)) == ARCHIVE_OK) { 751caf54c4fSMartin Matuska if (need_report()) 752caf54c4fSMartin Matuska report_write(bsdtar, a, entry, progress); 753caf54c4fSMartin Matuska 754fd082e96SMartin Matuska if (offset > progress) { 755fd082e96SMartin Matuska int64_t sparse = offset - progress; 756fd082e96SMartin Matuska size_t ns; 757fd082e96SMartin Matuska 758fd082e96SMartin Matuska if (null_buff == NULL) { 759fd082e96SMartin Matuska null_buff = bsdtar->buff; 760fd082e96SMartin Matuska memset(null_buff, 0, bsdtar->buff_size); 761fd082e96SMartin Matuska } 762fd082e96SMartin Matuska 763fd082e96SMartin Matuska while (sparse > 0) { 764fd082e96SMartin Matuska if (sparse > (int64_t)bsdtar->buff_size) 765fd082e96SMartin Matuska ns = bsdtar->buff_size; 766fd082e96SMartin Matuska else 767fd082e96SMartin Matuska ns = (size_t)sparse; 768fd082e96SMartin Matuska bytes_written = 769fd082e96SMartin Matuska archive_write_data(a, null_buff, ns); 770fd082e96SMartin Matuska if (bytes_written < 0) { 771fd082e96SMartin Matuska /* Write failed; this is bad */ 772fd082e96SMartin Matuska lafe_warnc(0, "%s", 773fd082e96SMartin Matuska archive_error_string(a)); 774fd082e96SMartin Matuska return (-1); 775fd082e96SMartin Matuska } 776fd082e96SMartin Matuska if ((size_t)bytes_written < ns) { 777fd082e96SMartin Matuska /* Write was truncated; warn but 778fd082e96SMartin Matuska * continue. */ 779fd082e96SMartin Matuska lafe_warnc(0, 780fd082e96SMartin Matuska "%s: Truncated write; file may " 781fd082e96SMartin Matuska "have grown while being archived.", 782fd082e96SMartin Matuska archive_entry_pathname(entry)); 783fd082e96SMartin Matuska return (0); 784fd082e96SMartin Matuska } 785fd082e96SMartin Matuska progress += bytes_written; 786fd082e96SMartin Matuska sparse -= bytes_written; 787fd082e96SMartin Matuska } 788fd082e96SMartin Matuska } 789fd082e96SMartin Matuska 790fd082e96SMartin Matuska bytes_written = archive_write_data(a, buff, bytes_read); 791fd082e96SMartin Matuska if (bytes_written < 0) { 792fd082e96SMartin Matuska /* Write failed; this is bad */ 793caf54c4fSMartin Matuska lafe_warnc(0, "%s", archive_error_string(a)); 794caf54c4fSMartin Matuska return (-1); 795caf54c4fSMartin Matuska } 796fd082e96SMartin Matuska if ((size_t)bytes_written < bytes_read) { 797fd082e96SMartin Matuska /* Write was truncated; warn but continue. */ 798fd082e96SMartin Matuska lafe_warnc(0, 799fd082e96SMartin Matuska "%s: Truncated write; file may have grown " 800fd082e96SMartin Matuska "while being archived.", 801fd082e96SMartin Matuska archive_entry_pathname(entry)); 802fd082e96SMartin Matuska return (0); 803caf54c4fSMartin Matuska } 804fd082e96SMartin Matuska progress += bytes_written; 805fd082e96SMartin Matuska } 806fd082e96SMartin Matuska if (r < ARCHIVE_WARN) { 807fd082e96SMartin Matuska lafe_warnc(archive_errno(a), "%s", archive_error_string(a)); 808fd082e96SMartin Matuska return (-1); 809fd082e96SMartin Matuska } 810caf54c4fSMartin Matuska return (0); 811caf54c4fSMartin Matuska } 812caf54c4fSMartin Matuska 813caf54c4fSMartin Matuska static void 814fd082e96SMartin Matuska excluded_callback(struct archive *a, void *_data, struct archive_entry *entry) 815caf54c4fSMartin Matuska { 816fd082e96SMartin Matuska struct bsdtar *bsdtar = (struct bsdtar *)_data; 817caf54c4fSMartin Matuska 81864287048SMartin Matuska if (bsdtar->flags & OPTFLAG_NO_SUBDIRS) 819fd082e96SMartin Matuska return; 820fd082e96SMartin Matuska if (!archive_read_disk_can_descend(a)) 821fd082e96SMartin Matuska return; 82264287048SMartin Matuska if ((bsdtar->flags & OPTFLAG_INTERACTIVE) && 823fd082e96SMartin Matuska !yes("add '%s'", archive_entry_pathname(entry))) 824fd082e96SMartin Matuska return; 825fd082e96SMartin Matuska archive_read_disk_descend(a); 826fd082e96SMartin Matuska } 827c5d360f8SMartin Matuska 828fd082e96SMartin Matuska static int 829fd082e96SMartin Matuska metadata_filter(struct archive *a, void *_data, struct archive_entry *entry) 830fd082e96SMartin Matuska { 831fd082e96SMartin Matuska struct bsdtar *bsdtar = (struct bsdtar *)_data; 832caf54c4fSMartin Matuska 833caf54c4fSMartin Matuska /* XXX TODO: check whether this filesystem is 834caf54c4fSMartin Matuska * synthetic and/or local. Add a new 835caf54c4fSMartin Matuska * --local-only option to skip non-local 836caf54c4fSMartin Matuska * filesystems. Skip synthetic filesystems 837caf54c4fSMartin Matuska * regardless. 838caf54c4fSMartin Matuska * 839caf54c4fSMartin Matuska * The results should be cached, since 840caf54c4fSMartin Matuska * tree.c doesn't usually visit a directory 841caf54c4fSMartin Matuska * and the directory contents together. A simple 842caf54c4fSMartin Matuska * move-to-front list should perform quite well. 843caf54c4fSMartin Matuska * 844fd082e96SMartin Matuska * Use archive_read_disk_current_filesystem_is_remote(). 845caf54c4fSMartin Matuska */ 846fd082e96SMartin Matuska 847fd082e96SMartin Matuska /* 848fd082e96SMartin Matuska * If the user vetoes this file/directory, skip it. 849fd082e96SMartin Matuska * We want this to be fairly late; if some other 850fd082e96SMartin Matuska * check would veto this file, we shouldn't bother 851fd082e96SMartin Matuska * the user with it. 852fd082e96SMartin Matuska */ 85364287048SMartin Matuska if ((bsdtar->flags & OPTFLAG_INTERACTIVE) && 854fd082e96SMartin Matuska !yes("add '%s'", archive_entry_pathname(entry))) 855fd082e96SMartin Matuska return (0); 856fd082e96SMartin Matuska 857fd082e96SMartin Matuska /* Note: if user vetoes, we won't descend. */ 85864287048SMartin Matuska if (((bsdtar->flags & OPTFLAG_NO_SUBDIRS) == 0) && 85964287048SMartin Matuska archive_read_disk_can_descend(a)) 860fd082e96SMartin Matuska archive_read_disk_descend(a); 861fd082e96SMartin Matuska 862fd082e96SMartin Matuska return (1); 863caf54c4fSMartin Matuska } 864caf54c4fSMartin Matuska 865caf54c4fSMartin Matuska /* 866fd082e96SMartin Matuska * Add the file or dir hierarchy named by 'path' to the archive 867caf54c4fSMartin Matuska */ 868fd082e96SMartin Matuska static void 869fd082e96SMartin Matuska write_hierarchy(struct bsdtar *bsdtar, struct archive *a, const char *path) 870fd082e96SMartin Matuska { 871fd082e96SMartin Matuska struct archive *disk = bsdtar->diskreader; 872fd082e96SMartin Matuska struct archive_entry *entry = NULL, *spare_entry = NULL; 873fd082e96SMartin Matuska int r; 874fd082e96SMartin Matuska 875fd082e96SMartin Matuska r = archive_read_disk_open(disk, path); 876fd082e96SMartin Matuska if (r != ARCHIVE_OK) { 877fd082e96SMartin Matuska lafe_warnc(archive_errno(disk), 878fd082e96SMartin Matuska "%s", archive_error_string(disk)); 879fd082e96SMartin Matuska bsdtar->return_value = 1; 880fd082e96SMartin Matuska return; 881fd082e96SMartin Matuska } 882fd082e96SMartin Matuska bsdtar->first_fs = -1; 883fd082e96SMartin Matuska 884fd082e96SMartin Matuska for (;;) { 885fd082e96SMartin Matuska archive_entry_free(entry); 886fd082e96SMartin Matuska entry = archive_entry_new(); 887fd082e96SMartin Matuska r = archive_read_next_header2(disk, entry); 888fd082e96SMartin Matuska if (r == ARCHIVE_EOF) 889fd082e96SMartin Matuska break; 890fd082e96SMartin Matuska else if (r != ARCHIVE_OK) { 891fd082e96SMartin Matuska lafe_warnc(archive_errno(disk), 892fd082e96SMartin Matuska "%s", archive_error_string(disk)); 893f061a221SMartin Matuska if (r == ARCHIVE_FATAL || r == ARCHIVE_FAILED) { 894fd082e96SMartin Matuska bsdtar->return_value = 1; 895f3e9b21aSMartin Matuska archive_entry_free(entry); 896f3e9b21aSMartin Matuska archive_read_close(disk); 897fd082e96SMartin Matuska return; 898fd082e96SMartin Matuska } else if (r < ARCHIVE_WARN) 899c5d360f8SMartin Matuska continue; 900c5d360f8SMartin Matuska } 901caf54c4fSMartin Matuska 902caf54c4fSMartin Matuska if (bsdtar->uid >= 0) { 903caf54c4fSMartin Matuska archive_entry_set_uid(entry, bsdtar->uid); 904caf54c4fSMartin Matuska if (!bsdtar->uname) 905caf54c4fSMartin Matuska archive_entry_set_uname(entry, 906caf54c4fSMartin Matuska archive_read_disk_uname(bsdtar->diskreader, 907caf54c4fSMartin Matuska bsdtar->uid)); 908caf54c4fSMartin Matuska } 909caf54c4fSMartin Matuska if (bsdtar->gid >= 0) { 910caf54c4fSMartin Matuska archive_entry_set_gid(entry, bsdtar->gid); 911caf54c4fSMartin Matuska if (!bsdtar->gname) 912caf54c4fSMartin Matuska archive_entry_set_gname(entry, 913caf54c4fSMartin Matuska archive_read_disk_gname(bsdtar->diskreader, 914caf54c4fSMartin Matuska bsdtar->gid)); 915caf54c4fSMartin Matuska } 916caf54c4fSMartin Matuska if (bsdtar->uname) 917caf54c4fSMartin Matuska archive_entry_set_uname(entry, bsdtar->uname); 918caf54c4fSMartin Matuska if (bsdtar->gname) 919caf54c4fSMartin Matuska archive_entry_set_gname(entry, bsdtar->gname); 920caf54c4fSMartin Matuska 921caf54c4fSMartin Matuska /* 922caf54c4fSMartin Matuska * Rewrite the pathname to be archived. If rewrite 923caf54c4fSMartin Matuska * fails, skip the entry. 924caf54c4fSMartin Matuska */ 925caf54c4fSMartin Matuska if (edit_pathname(bsdtar, entry)) 926caf54c4fSMartin Matuska continue; 927caf54c4fSMartin Matuska 928cdf63a70SMartin Matuska /* Display entry as we process it. */ 929cdf63a70SMartin Matuska if (bsdtar->verbose > 1) { 930cdf63a70SMartin Matuska safe_fprintf(stderr, "a "); 931cdf63a70SMartin Matuska list_item_verbose(bsdtar, stderr, entry); 932cdf63a70SMartin Matuska } else if (bsdtar->verbose > 0) { 933cdf63a70SMartin Matuska /* This format is required by SUSv2. */ 934caf54c4fSMartin Matuska safe_fprintf(stderr, "a %s", 935caf54c4fSMartin Matuska archive_entry_pathname(entry)); 936cdf63a70SMartin Matuska } 937caf54c4fSMartin Matuska 938caf54c4fSMartin Matuska /* Non-regular files get archived with zero size. */ 939caf54c4fSMartin Matuska if (archive_entry_filetype(entry) != AE_IFREG) 940caf54c4fSMartin Matuska archive_entry_set_size(entry, 0); 941caf54c4fSMartin Matuska 942caf54c4fSMartin Matuska archive_entry_linkify(bsdtar->resolver, &entry, &spare_entry); 943caf54c4fSMartin Matuska 944caf54c4fSMartin Matuska while (entry != NULL) { 9456c95142eSMartin Matuska write_file(bsdtar, a, entry); 946caf54c4fSMartin Matuska archive_entry_free(entry); 947caf54c4fSMartin Matuska entry = spare_entry; 948caf54c4fSMartin Matuska spare_entry = NULL; 949caf54c4fSMartin Matuska } 950caf54c4fSMartin Matuska 951caf54c4fSMartin Matuska if (bsdtar->verbose) 952caf54c4fSMartin Matuska fprintf(stderr, "\n"); 953caf54c4fSMartin Matuska } 954caf54c4fSMartin Matuska archive_entry_free(entry); 955fd082e96SMartin Matuska archive_read_close(disk); 956caf54c4fSMartin Matuska } 957caf54c4fSMartin Matuska 958caf54c4fSMartin Matuska /* 9596c95142eSMartin Matuska * Write a single file (or directory or other filesystem object) to 9606c95142eSMartin Matuska * the archive. 961caf54c4fSMartin Matuska */ 962caf54c4fSMartin Matuska static void 9636c95142eSMartin Matuska write_file(struct bsdtar *bsdtar, struct archive *a, 9646c95142eSMartin Matuska struct archive_entry *entry) 9656c95142eSMartin Matuska { 9666c95142eSMartin Matuska write_entry(bsdtar, a, entry); 9676c95142eSMartin Matuska } 9686c95142eSMartin Matuska 9696c95142eSMartin Matuska /* 9706c95142eSMartin Matuska * Write a single entry to the archive. 9716c95142eSMartin Matuska */ 9726c95142eSMartin Matuska static void 9736c95142eSMartin Matuska write_entry(struct bsdtar *bsdtar, struct archive *a, 974caf54c4fSMartin Matuska struct archive_entry *entry) 975caf54c4fSMartin Matuska { 976caf54c4fSMartin Matuska int e; 977caf54c4fSMartin Matuska 978caf54c4fSMartin Matuska e = archive_write_header(a, entry); 979caf54c4fSMartin Matuska if (e != ARCHIVE_OK) { 980cdf63a70SMartin Matuska if (bsdtar->verbose > 1) { 981cdf63a70SMartin Matuska safe_fprintf(stderr, "a "); 982cdf63a70SMartin Matuska list_item_verbose(bsdtar, stderr, entry); 983cdf63a70SMartin Matuska lafe_warnc(0, ": %s", archive_error_string(a)); 984cdf63a70SMartin Matuska } else if (bsdtar->verbose > 0) { 985caf54c4fSMartin Matuska lafe_warnc(0, "%s: %s", 986caf54c4fSMartin Matuska archive_entry_pathname(entry), 987caf54c4fSMartin Matuska archive_error_string(a)); 988cdf63a70SMartin Matuska } else 989caf54c4fSMartin Matuska fprintf(stderr, ": %s", archive_error_string(a)); 990caf54c4fSMartin Matuska } 991caf54c4fSMartin Matuska 992caf54c4fSMartin Matuska if (e == ARCHIVE_FATAL) 993caf54c4fSMartin Matuska exit(1); 994caf54c4fSMartin Matuska 995caf54c4fSMartin Matuska /* 996caf54c4fSMartin Matuska * If we opened a file earlier, write it out now. Note that 997caf54c4fSMartin Matuska * the format handler might have reset the size field to zero 998caf54c4fSMartin Matuska * to inform us that the archive body won't get stored. In 999caf54c4fSMartin Matuska * that case, just skip the write. 1000caf54c4fSMartin Matuska */ 1001fd082e96SMartin Matuska if (e >= ARCHIVE_WARN && archive_entry_size(entry) > 0) { 1002fd082e96SMartin Matuska if (copy_file_data_block(bsdtar, a, bsdtar->diskreader, entry)) 1003caf54c4fSMartin Matuska exit(1); 1004caf54c4fSMartin Matuska } 1005caf54c4fSMartin Matuska } 1006caf54c4fSMartin Matuska 1007caf54c4fSMartin Matuska static void 1008caf54c4fSMartin Matuska report_write(struct bsdtar *bsdtar, struct archive *a, 1009caf54c4fSMartin Matuska struct archive_entry *entry, int64_t progress) 1010caf54c4fSMartin Matuska { 1011caf54c4fSMartin Matuska uint64_t comp, uncomp; 1012c5d360f8SMartin Matuska int compression; 1013c5d360f8SMartin Matuska 1014caf54c4fSMartin Matuska if (bsdtar->verbose) 1015caf54c4fSMartin Matuska fprintf(stderr, "\n"); 1016acc60b03SMartin Matuska comp = archive_filter_bytes(a, -1); 1017acc60b03SMartin Matuska uncomp = archive_filter_bytes(a, 0); 1018caf54c4fSMartin Matuska fprintf(stderr, "In: %d files, %s bytes;", 1019caf54c4fSMartin Matuska archive_file_count(a), tar_i64toa(uncomp)); 102064287048SMartin Matuska if (comp >= uncomp) 1021c5d360f8SMartin Matuska compression = 0; 1022c5d360f8SMartin Matuska else 1023c5d360f8SMartin Matuska compression = (int)((uncomp - comp) * 100 / uncomp); 1024caf54c4fSMartin Matuska fprintf(stderr, 1025caf54c4fSMartin Matuska " Out: %s bytes, compression %d%%\n", 1026c5d360f8SMartin Matuska tar_i64toa(comp), compression); 1027caf54c4fSMartin Matuska /* Can't have two calls to tar_i64toa() pending, so split the output. */ 1028caf54c4fSMartin Matuska safe_fprintf(stderr, "Current: %s (%s", 1029caf54c4fSMartin Matuska archive_entry_pathname(entry), 1030caf54c4fSMartin Matuska tar_i64toa(progress)); 1031caf54c4fSMartin Matuska fprintf(stderr, "/%s bytes)\n", 1032caf54c4fSMartin Matuska tar_i64toa(archive_entry_size(entry))); 1033caf54c4fSMartin Matuska } 1034caf54c4fSMartin Matuska 1035caf54c4fSMartin Matuska static void 1036caf54c4fSMartin Matuska test_for_append(struct bsdtar *bsdtar) 1037caf54c4fSMartin Matuska { 1038caf54c4fSMartin Matuska struct stat s; 1039caf54c4fSMartin Matuska 1040caf54c4fSMartin Matuska if (*bsdtar->argv == NULL && bsdtar->names_from_file == NULL) 1041caf54c4fSMartin Matuska lafe_errc(1, 0, "no files or directories specified"); 1042caf54c4fSMartin Matuska if (bsdtar->filename == NULL) 1043caf54c4fSMartin Matuska lafe_errc(1, 0, "Cannot append to stdout."); 1044caf54c4fSMartin Matuska 1045caf54c4fSMartin Matuska if (stat(bsdtar->filename, &s) != 0) 1046caf54c4fSMartin Matuska return; 1047caf54c4fSMartin Matuska 1048caf54c4fSMartin Matuska if (!S_ISREG(s.st_mode) && !S_ISBLK(s.st_mode)) 1049caf54c4fSMartin Matuska lafe_errc(1, 0, 1050caf54c4fSMartin Matuska "Cannot append to %s: not a regular file.", 1051caf54c4fSMartin Matuska bsdtar->filename); 1052caf54c4fSMartin Matuska 1053caf54c4fSMartin Matuska /* Is this an appropriate check here on Windows? */ 1054caf54c4fSMartin Matuska /* 1055caf54c4fSMartin Matuska if (GetFileType(handle) != FILE_TYPE_DISK) 1056caf54c4fSMartin Matuska lafe_errc(1, 0, "Cannot append"); 1057caf54c4fSMartin Matuska */ 1058caf54c4fSMartin Matuska 1059caf54c4fSMartin Matuska } 1060