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) { 148acc60b03SMartin Matuska char *p; 149acc60b03SMartin Matuska /* Set default write options. */ 150acc60b03SMartin Matuska p = malloc(sizeof(IGNORE_WRONG_MODULE_NAME) 151acc60b03SMartin Matuska + strlen(writer_options) + 1); 152acc60b03SMartin Matuska if (p == 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. */ 157acc60b03SMartin Matuska strncpy(p, IGNORE_WRONG_MODULE_NAME, 158acc60b03SMartin Matuska sizeof(IGNORE_WRONG_MODULE_NAME) -1); 159acc60b03SMartin Matuska strcpy(p + sizeof(IGNORE_WRONG_MODULE_NAME) -1, writer_options); 160acc60b03SMartin Matuska r = archive_write_set_options(a, p); 161acc60b03SMartin Matuska free(p); 162acc60b03SMartin Matuska if (r < ARCHIVE_WARN) 163acc60b03SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 164acc60b03SMartin Matuska else 165acc60b03SMartin Matuska archive_clear_error(a); 166acc60b03SMartin Matuska } 167acc60b03SMartin Matuska if (ARCHIVE_OK != archive_write_set_options(a, bsdtar->option_options)) 168acc60b03SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 169acc60b03SMartin Matuska } 170acc60b03SMartin Matuska 171acc60b03SMartin Matuska static void 172acc60b03SMartin Matuska set_reader_options(struct bsdtar *bsdtar, struct archive *a) 173acc60b03SMartin Matuska { 174acc60b03SMartin Matuska const char *reader_options; 175acc60b03SMartin Matuska int r; 176acc60b03SMartin Matuska 177acc60b03SMartin Matuska (void)bsdtar; /* UNUSED */ 178acc60b03SMartin Matuska 179acc60b03SMartin Matuska reader_options = getenv(ENV_READER_OPTIONS); 180acc60b03SMartin Matuska if (reader_options != NULL) { 181acc60b03SMartin Matuska char *p; 182acc60b03SMartin Matuska /* Set default write options. */ 183acc60b03SMartin Matuska p = malloc(sizeof(IGNORE_WRONG_MODULE_NAME) 184acc60b03SMartin Matuska + strlen(reader_options) + 1); 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. */ 190acc60b03SMartin Matuska strncpy(p, IGNORE_WRONG_MODULE_NAME, 191acc60b03SMartin Matuska sizeof(IGNORE_WRONG_MODULE_NAME) -1); 192acc60b03SMartin Matuska strcpy(p + sizeof(IGNORE_WRONG_MODULE_NAME) -1, reader_options); 193acc60b03SMartin Matuska r = archive_read_set_options(a, p); 194acc60b03SMartin Matuska free(p); 195acc60b03SMartin Matuska if (r < ARCHIVE_WARN) 196acc60b03SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 197acc60b03SMartin Matuska else 198acc60b03SMartin Matuska archive_clear_error(a); 199acc60b03SMartin Matuska } 200acc60b03SMartin Matuska } 201acc60b03SMartin Matuska 202caf54c4fSMartin Matuska void 203caf54c4fSMartin Matuska tar_mode_c(struct bsdtar *bsdtar) 204caf54c4fSMartin Matuska { 205caf54c4fSMartin Matuska struct archive *a; 206acc60b03SMartin Matuska const void *filter_name; 207caf54c4fSMartin Matuska int r; 208caf54c4fSMartin Matuska 209caf54c4fSMartin Matuska if (*bsdtar->argv == NULL && bsdtar->names_from_file == NULL) 210caf54c4fSMartin Matuska lafe_errc(1, 0, "no files or directories specified"); 211caf54c4fSMartin Matuska 212caf54c4fSMartin Matuska a = archive_write_new(); 213caf54c4fSMartin Matuska 214caf54c4fSMartin Matuska /* Support any format that the library supports. */ 215acc60b03SMartin Matuska if (cset_get_format(bsdtar->cset) == NULL) { 216caf54c4fSMartin Matuska r = archive_write_set_format_pax_restricted(a); 217acc60b03SMartin Matuska cset_set_format(bsdtar->cset, "pax restricted"); 218caf54c4fSMartin Matuska } else { 219acc60b03SMartin Matuska r = archive_write_set_format_by_name(a, 220acc60b03SMartin Matuska cset_get_format(bsdtar->cset)); 221caf54c4fSMartin Matuska } 222caf54c4fSMartin Matuska if (r != ARCHIVE_OK) { 223caf54c4fSMartin Matuska fprintf(stderr, "Can't use format %s: %s\n", 224acc60b03SMartin Matuska cset_get_format(bsdtar->cset), 225caf54c4fSMartin Matuska archive_error_string(a)); 226caf54c4fSMartin Matuska usage(); 227caf54c4fSMartin Matuska } 228caf54c4fSMartin Matuska 229caf54c4fSMartin Matuska archive_write_set_bytes_per_block(a, bsdtar->bytes_per_block); 2306c95142eSMartin Matuska archive_write_set_bytes_in_last_block(a, bsdtar->bytes_in_last_block); 231caf54c4fSMartin Matuska 232acc60b03SMartin Matuska r = cset_write_add_filters(bsdtar->cset, a, &filter_name); 233acc60b03SMartin Matuska if (r < ARCHIVE_WARN) { 234acc60b03SMartin Matuska lafe_errc(1, 0, "Unsupported compression option --%s", 235acc60b03SMartin Matuska (const char *)filter_name); 236caf54c4fSMartin Matuska } 237caf54c4fSMartin Matuska 238acc60b03SMartin Matuska set_writer_options(bsdtar, a); 239cdf63a70SMartin Matuska if (bsdtar->passphrase != NULL) 240cdf63a70SMartin Matuska r = archive_write_set_passphrase(a, bsdtar->passphrase); 241cdf63a70SMartin Matuska else 242cdf63a70SMartin Matuska r = archive_write_set_passphrase_callback(a, bsdtar, 243cdf63a70SMartin Matuska &passphrase_callback); 244cdf63a70SMartin Matuska if (r != ARCHIVE_OK) 245cdf63a70SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 246acc60b03SMartin Matuska if (ARCHIVE_OK != archive_write_open_filename(a, bsdtar->filename)) 247caf54c4fSMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 248caf54c4fSMartin Matuska write_archive(a, bsdtar); 249caf54c4fSMartin Matuska } 250caf54c4fSMartin Matuska 251caf54c4fSMartin Matuska /* 252caf54c4fSMartin Matuska * Same as 'c', except we only support tar or empty formats in 253caf54c4fSMartin Matuska * uncompressed files on disk. 254caf54c4fSMartin Matuska */ 255caf54c4fSMartin Matuska void 256caf54c4fSMartin Matuska tar_mode_r(struct bsdtar *bsdtar) 257caf54c4fSMartin Matuska { 258caf54c4fSMartin Matuska int64_t end_offset; 259caf54c4fSMartin Matuska int format; 260caf54c4fSMartin Matuska struct archive *a; 261caf54c4fSMartin Matuska struct archive_entry *entry; 262caf54c4fSMartin Matuska int r; 263caf54c4fSMartin Matuska 264caf54c4fSMartin Matuska /* Sanity-test some arguments and the file. */ 265caf54c4fSMartin Matuska test_for_append(bsdtar); 266caf54c4fSMartin Matuska 267caf54c4fSMartin Matuska format = ARCHIVE_FORMAT_TAR_PAX_RESTRICTED; 268caf54c4fSMartin Matuska 269caf54c4fSMartin Matuska #if defined(__BORLANDC__) 270caf54c4fSMartin Matuska bsdtar->fd = open(bsdtar->filename, O_RDWR | O_CREAT | O_BINARY); 271caf54c4fSMartin Matuska #else 272caf54c4fSMartin Matuska bsdtar->fd = open(bsdtar->filename, O_RDWR | O_CREAT | O_BINARY, 0666); 273caf54c4fSMartin Matuska #endif 274caf54c4fSMartin Matuska if (bsdtar->fd < 0) 275caf54c4fSMartin Matuska lafe_errc(1, errno, 276caf54c4fSMartin Matuska "Cannot open %s", bsdtar->filename); 277caf54c4fSMartin Matuska 278caf54c4fSMartin Matuska a = archive_read_new(); 2796c95142eSMartin Matuska archive_read_support_filter_all(a); 2807ceb9433SMartin Matuska archive_read_support_format_empty(a); 281caf54c4fSMartin Matuska archive_read_support_format_tar(a); 282caf54c4fSMartin Matuska archive_read_support_format_gnutar(a); 283acc60b03SMartin Matuska set_reader_options(bsdtar, a); 284caf54c4fSMartin Matuska r = archive_read_open_fd(a, bsdtar->fd, 10240); 285caf54c4fSMartin Matuska if (r != ARCHIVE_OK) 286caf54c4fSMartin Matuska lafe_errc(1, archive_errno(a), 287caf54c4fSMartin Matuska "Can't read archive %s: %s", bsdtar->filename, 288caf54c4fSMartin Matuska archive_error_string(a)); 289caf54c4fSMartin Matuska while (0 == archive_read_next_header(a, &entry)) { 290acc60b03SMartin Matuska if (archive_filter_code(a, 0) != ARCHIVE_FILTER_NONE) { 2916c95142eSMartin Matuska archive_read_free(a); 292caf54c4fSMartin Matuska close(bsdtar->fd); 293caf54c4fSMartin Matuska lafe_errc(1, 0, 294caf54c4fSMartin Matuska "Cannot append to compressed archive."); 295caf54c4fSMartin Matuska } 296caf54c4fSMartin Matuska /* Keep going until we hit end-of-archive */ 297caf54c4fSMartin Matuska format = archive_format(a); 298caf54c4fSMartin Matuska } 299caf54c4fSMartin Matuska 300caf54c4fSMartin Matuska end_offset = archive_read_header_position(a); 3016c95142eSMartin Matuska archive_read_free(a); 302caf54c4fSMartin Matuska 303caf54c4fSMartin Matuska /* Re-open archive for writing */ 304caf54c4fSMartin Matuska a = archive_write_new(); 305caf54c4fSMartin Matuska /* 306caf54c4fSMartin Matuska * Set the format to be used for writing. To allow people to 307caf54c4fSMartin Matuska * extend empty files, we need to allow them to specify the format, 308caf54c4fSMartin Matuska * which opens the possibility that they will specify a format that 309caf54c4fSMartin Matuska * doesn't match the existing format. Hence, the following bit 310caf54c4fSMartin Matuska * of arcane ugliness. 311caf54c4fSMartin Matuska */ 312caf54c4fSMartin Matuska 313acc60b03SMartin Matuska if (cset_get_format(bsdtar->cset) != NULL) { 314caf54c4fSMartin Matuska /* If the user requested a format, use that, but ... */ 315caf54c4fSMartin Matuska archive_write_set_format_by_name(a, 316acc60b03SMartin Matuska cset_get_format(bsdtar->cset)); 317caf54c4fSMartin Matuska /* ... complain if it's not compatible. */ 318caf54c4fSMartin Matuska format &= ARCHIVE_FORMAT_BASE_MASK; 319caf54c4fSMartin Matuska if (format != (int)(archive_format(a) & ARCHIVE_FORMAT_BASE_MASK) 320caf54c4fSMartin Matuska && format != ARCHIVE_FORMAT_EMPTY) { 321caf54c4fSMartin Matuska lafe_errc(1, 0, 322caf54c4fSMartin Matuska "Format %s is incompatible with the archive %s.", 323acc60b03SMartin Matuska cset_get_format(bsdtar->cset), bsdtar->filename); 324caf54c4fSMartin Matuska } 325caf54c4fSMartin Matuska } else { 326caf54c4fSMartin Matuska /* 327caf54c4fSMartin Matuska * Just preserve the current format, with a little care 328caf54c4fSMartin Matuska * for formats that libarchive can't write. 329caf54c4fSMartin Matuska */ 330caf54c4fSMartin Matuska if (format == ARCHIVE_FORMAT_EMPTY) 331caf54c4fSMartin Matuska format = ARCHIVE_FORMAT_TAR_PAX_RESTRICTED; 332caf54c4fSMartin Matuska archive_write_set_format(a, format); 333caf54c4fSMartin Matuska } 334caf54c4fSMartin Matuska if (lseek(bsdtar->fd, end_offset, SEEK_SET) < 0) 335caf54c4fSMartin Matuska lafe_errc(1, errno, "Could not seek to archive end"); 336acc60b03SMartin Matuska set_writer_options(bsdtar, a); 337caf54c4fSMartin Matuska if (ARCHIVE_OK != archive_write_open_fd(a, bsdtar->fd)) 338caf54c4fSMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 339caf54c4fSMartin Matuska 340caf54c4fSMartin Matuska write_archive(a, bsdtar); /* XXX check return val XXX */ 341caf54c4fSMartin Matuska 342caf54c4fSMartin Matuska close(bsdtar->fd); 343caf54c4fSMartin Matuska bsdtar->fd = -1; 344caf54c4fSMartin Matuska } 345caf54c4fSMartin Matuska 346caf54c4fSMartin Matuska void 347caf54c4fSMartin Matuska tar_mode_u(struct bsdtar *bsdtar) 348caf54c4fSMartin Matuska { 349caf54c4fSMartin Matuska int64_t end_offset; 350caf54c4fSMartin Matuska struct archive *a; 351caf54c4fSMartin Matuska struct archive_entry *entry; 352caf54c4fSMartin Matuska int format; 353caf54c4fSMartin Matuska struct archive_dir_entry *p; 354caf54c4fSMartin Matuska struct archive_dir archive_dir; 355caf54c4fSMartin Matuska 356caf54c4fSMartin Matuska bsdtar->archive_dir = &archive_dir; 357caf54c4fSMartin Matuska memset(&archive_dir, 0, sizeof(archive_dir)); 358caf54c4fSMartin Matuska 359caf54c4fSMartin Matuska format = ARCHIVE_FORMAT_TAR_PAX_RESTRICTED; 360caf54c4fSMartin Matuska 361caf54c4fSMartin Matuska /* Sanity-test some arguments and the file. */ 362caf54c4fSMartin Matuska test_for_append(bsdtar); 363caf54c4fSMartin Matuska 364caf54c4fSMartin Matuska bsdtar->fd = open(bsdtar->filename, O_RDWR | O_BINARY); 365caf54c4fSMartin Matuska if (bsdtar->fd < 0) 366caf54c4fSMartin Matuska lafe_errc(1, errno, 367caf54c4fSMartin Matuska "Cannot open %s", bsdtar->filename); 368caf54c4fSMartin Matuska 369caf54c4fSMartin Matuska a = archive_read_new(); 3706c95142eSMartin Matuska archive_read_support_filter_all(a); 371caf54c4fSMartin Matuska archive_read_support_format_tar(a); 372caf54c4fSMartin Matuska archive_read_support_format_gnutar(a); 373acc60b03SMartin Matuska set_reader_options(bsdtar, a); 3746c95142eSMartin Matuska if (archive_read_open_fd(a, bsdtar->fd, bsdtar->bytes_per_block) 3756c95142eSMartin Matuska != ARCHIVE_OK) { 376caf54c4fSMartin Matuska lafe_errc(1, 0, 377caf54c4fSMartin Matuska "Can't open %s: %s", bsdtar->filename, 378caf54c4fSMartin Matuska archive_error_string(a)); 379caf54c4fSMartin Matuska } 380caf54c4fSMartin Matuska 381caf54c4fSMartin Matuska /* Build a list of all entries and their recorded mod times. */ 382caf54c4fSMartin Matuska while (0 == archive_read_next_header(a, &entry)) { 383acc60b03SMartin Matuska if (archive_filter_code(a, 0) != ARCHIVE_FILTER_NONE) { 3846c95142eSMartin Matuska archive_read_free(a); 385caf54c4fSMartin Matuska close(bsdtar->fd); 386caf54c4fSMartin Matuska lafe_errc(1, 0, 387caf54c4fSMartin Matuska "Cannot append to compressed archive."); 388caf54c4fSMartin Matuska } 389fd082e96SMartin Matuska if (archive_match_exclude_entry(bsdtar->matching, 390fd082e96SMartin Matuska ARCHIVE_MATCH_MTIME | ARCHIVE_MATCH_OLDER | 391fd082e96SMartin Matuska ARCHIVE_MATCH_EQUAL, entry) != ARCHIVE_OK) 392fd082e96SMartin Matuska lafe_errc(1, 0, "Error : %s", 393fd082e96SMartin Matuska archive_error_string(bsdtar->matching)); 394caf54c4fSMartin Matuska /* Record the last format determination we see */ 395caf54c4fSMartin Matuska format = archive_format(a); 396caf54c4fSMartin Matuska /* Keep going until we hit end-of-archive */ 397caf54c4fSMartin Matuska } 398caf54c4fSMartin Matuska 399caf54c4fSMartin Matuska end_offset = archive_read_header_position(a); 4006c95142eSMartin Matuska archive_read_free(a); 401caf54c4fSMartin Matuska 402caf54c4fSMartin Matuska /* Re-open archive for writing. */ 403caf54c4fSMartin Matuska a = archive_write_new(); 404caf54c4fSMartin Matuska /* 4056c95142eSMartin Matuska * Set format to same one auto-detected above. 406caf54c4fSMartin Matuska */ 407caf54c4fSMartin Matuska archive_write_set_format(a, format); 408caf54c4fSMartin Matuska archive_write_set_bytes_per_block(a, bsdtar->bytes_per_block); 4096c95142eSMartin Matuska archive_write_set_bytes_in_last_block(a, bsdtar->bytes_in_last_block); 4106c95142eSMartin Matuska 411caf54c4fSMartin Matuska if (lseek(bsdtar->fd, end_offset, SEEK_SET) < 0) 412caf54c4fSMartin Matuska lafe_errc(1, errno, "Could not seek to archive end"); 413acc60b03SMartin Matuska set_writer_options(bsdtar, a); 414caf54c4fSMartin Matuska if (ARCHIVE_OK != archive_write_open_fd(a, bsdtar->fd)) 415caf54c4fSMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 416caf54c4fSMartin Matuska 417caf54c4fSMartin Matuska write_archive(a, bsdtar); 418caf54c4fSMartin Matuska 419caf54c4fSMartin Matuska close(bsdtar->fd); 420caf54c4fSMartin Matuska bsdtar->fd = -1; 421caf54c4fSMartin Matuska 422caf54c4fSMartin Matuska while (bsdtar->archive_dir->head != NULL) { 423caf54c4fSMartin Matuska p = bsdtar->archive_dir->head->next; 424caf54c4fSMartin Matuska free(bsdtar->archive_dir->head->name); 425caf54c4fSMartin Matuska free(bsdtar->archive_dir->head); 426caf54c4fSMartin Matuska bsdtar->archive_dir->head = p; 427caf54c4fSMartin Matuska } 428caf54c4fSMartin Matuska bsdtar->archive_dir->tail = NULL; 429caf54c4fSMartin Matuska } 430caf54c4fSMartin Matuska 431caf54c4fSMartin Matuska 432caf54c4fSMartin Matuska /* 433caf54c4fSMartin Matuska * Write user-specified files/dirs to opened archive. 434caf54c4fSMartin Matuska */ 435caf54c4fSMartin Matuska static void 436caf54c4fSMartin Matuska write_archive(struct archive *a, struct bsdtar *bsdtar) 437caf54c4fSMartin Matuska { 438caf54c4fSMartin Matuska const char *arg; 439caf54c4fSMartin Matuska struct archive_entry *entry, *sparse_entry; 440caf54c4fSMartin Matuska 4416c95142eSMartin Matuska /* Choose a suitable copy buffer size */ 4426c95142eSMartin Matuska bsdtar->buff_size = 64 * 1024; 4436c95142eSMartin Matuska while (bsdtar->buff_size < (size_t)bsdtar->bytes_per_block) 4446c95142eSMartin Matuska bsdtar->buff_size *= 2; 4456c95142eSMartin Matuska /* Try to compensate for space we'll lose to alignment. */ 4466c95142eSMartin Matuska bsdtar->buff_size += 16 * 1024; 4476c95142eSMartin Matuska 448caf54c4fSMartin Matuska /* Allocate a buffer for file data. */ 4496c95142eSMartin Matuska if ((bsdtar->buff = malloc(bsdtar->buff_size)) == NULL) 450caf54c4fSMartin Matuska lafe_errc(1, 0, "cannot allocate memory"); 451caf54c4fSMartin Matuska 452caf54c4fSMartin Matuska if ((bsdtar->resolver = archive_entry_linkresolver_new()) == NULL) 453caf54c4fSMartin Matuska lafe_errc(1, 0, "cannot create link resolver"); 454caf54c4fSMartin Matuska archive_entry_linkresolver_set_strategy(bsdtar->resolver, 455caf54c4fSMartin Matuska archive_format(a)); 456fd082e96SMartin Matuska 457fd082e96SMartin Matuska /* Create a read_disk object. */ 458caf54c4fSMartin Matuska if ((bsdtar->diskreader = archive_read_disk_new()) == NULL) 459caf54c4fSMartin Matuska lafe_errc(1, 0, "Cannot create read_disk object"); 460fd082e96SMartin Matuska /* Tell the read_disk how handle symlink. */ 461fd082e96SMartin Matuska switch (bsdtar->symlink_mode) { 462fd082e96SMartin Matuska case 'H': 463fd082e96SMartin Matuska archive_read_disk_set_symlink_hybrid(bsdtar->diskreader); 464fd082e96SMartin Matuska break; 465fd082e96SMartin Matuska case 'L': 466fd082e96SMartin Matuska archive_read_disk_set_symlink_logical(bsdtar->diskreader); 467fd082e96SMartin Matuska break; 468fd082e96SMartin Matuska default: 469fd082e96SMartin Matuska archive_read_disk_set_symlink_physical(bsdtar->diskreader); 470fd082e96SMartin Matuska break; 471fd082e96SMartin Matuska } 472fd082e96SMartin Matuska /* Register entry filters. */ 473fd082e96SMartin Matuska archive_read_disk_set_matching(bsdtar->diskreader, 474fd082e96SMartin Matuska bsdtar->matching, excluded_callback, bsdtar); 475fd082e96SMartin Matuska archive_read_disk_set_metadata_filter_callback( 476fd082e96SMartin Matuska bsdtar->diskreader, metadata_filter, bsdtar); 477fd082e96SMartin Matuska /* Set the behavior of archive_read_disk. */ 478fd082e96SMartin Matuska archive_read_disk_set_behavior(bsdtar->diskreader, 479fd082e96SMartin Matuska bsdtar->readdisk_flags); 480caf54c4fSMartin Matuska archive_read_disk_set_standard_lookup(bsdtar->diskreader); 481caf54c4fSMartin Matuska 482caf54c4fSMartin Matuska if (bsdtar->names_from_file != NULL) 483caf54c4fSMartin Matuska archive_names_from_file(bsdtar, a); 484caf54c4fSMartin Matuska 485caf54c4fSMartin Matuska while (*bsdtar->argv) { 486caf54c4fSMartin Matuska arg = *bsdtar->argv; 487caf54c4fSMartin Matuska if (arg[0] == '-' && arg[1] == 'C') { 488caf54c4fSMartin Matuska arg += 2; 489caf54c4fSMartin Matuska if (*arg == '\0') { 490caf54c4fSMartin Matuska bsdtar->argv++; 491caf54c4fSMartin Matuska arg = *bsdtar->argv; 492caf54c4fSMartin Matuska if (arg == NULL) { 493caf54c4fSMartin Matuska lafe_warnc(0, "%s", 494caf54c4fSMartin Matuska "Missing argument for -C"); 495caf54c4fSMartin Matuska bsdtar->return_value = 1; 496caf54c4fSMartin Matuska goto cleanup; 497caf54c4fSMartin Matuska } 4986c95142eSMartin Matuska if (*arg == '\0') { 4996c95142eSMartin Matuska lafe_warnc(0, 5006c95142eSMartin Matuska "Meaningless argument for -C: ''"); 5016c95142eSMartin Matuska bsdtar->return_value = 1; 5026c95142eSMartin Matuska goto cleanup; 5036c95142eSMartin Matuska } 504caf54c4fSMartin Matuska } 505caf54c4fSMartin Matuska set_chdir(bsdtar, arg); 506caf54c4fSMartin Matuska } else { 507caf54c4fSMartin Matuska if (*arg != '/' && (arg[0] != '@' || arg[1] != '/')) 508caf54c4fSMartin Matuska do_chdir(bsdtar); /* Handle a deferred -C */ 509caf54c4fSMartin Matuska if (*arg == '@') { 510caf54c4fSMartin Matuska if (append_archive_filename(bsdtar, a, 511caf54c4fSMartin Matuska arg + 1) != 0) 512caf54c4fSMartin Matuska break; 513caf54c4fSMartin Matuska } else 514caf54c4fSMartin Matuska write_hierarchy(bsdtar, a, arg); 515caf54c4fSMartin Matuska } 516caf54c4fSMartin Matuska bsdtar->argv++; 517caf54c4fSMartin Matuska } 518caf54c4fSMartin Matuska 519fd082e96SMartin Matuska archive_read_disk_set_matching(bsdtar->diskreader, NULL, NULL, NULL); 520fd082e96SMartin Matuska archive_read_disk_set_metadata_filter_callback( 521fd082e96SMartin Matuska bsdtar->diskreader, NULL, NULL); 522caf54c4fSMartin Matuska entry = NULL; 523caf54c4fSMartin Matuska archive_entry_linkify(bsdtar->resolver, &entry, &sparse_entry); 524caf54c4fSMartin Matuska while (entry != NULL) { 525fd082e96SMartin Matuska int r; 526fd082e96SMartin Matuska struct archive_entry *entry2; 527fd082e96SMartin Matuska struct archive *disk = bsdtar->diskreader; 528fd082e96SMartin Matuska 529fd082e96SMartin Matuska /* 530*007dbc1fSMartin Matuska * This tricky code here is to correctly read the contents 531fd082e96SMartin Matuska * of the entry because the disk reader bsdtar->diskreader 532fd082e96SMartin Matuska * is pointing at does not have any information about the 533fd082e96SMartin Matuska * entry by this time and using archive_read_data_block() 534fd082e96SMartin Matuska * with the disk reader consequently must fail. And we 535fd082e96SMartin Matuska * have to re-open the entry to read the contents. 536fd082e96SMartin Matuska */ 537fd082e96SMartin Matuska /* TODO: Work with -C option as well. */ 538fd082e96SMartin Matuska r = archive_read_disk_open(disk, 539fd082e96SMartin Matuska archive_entry_sourcepath(entry)); 540fd082e96SMartin Matuska if (r != ARCHIVE_OK) { 541fd082e96SMartin Matuska lafe_warnc(archive_errno(disk), 542fd082e96SMartin Matuska "%s", archive_error_string(disk)); 543fd082e96SMartin Matuska bsdtar->return_value = 1; 544fd082e96SMartin Matuska archive_entry_free(entry); 545fd082e96SMartin Matuska continue; 546fd082e96SMartin Matuska } 547fd082e96SMartin Matuska 548fd082e96SMartin Matuska /* 549fd082e96SMartin Matuska * Invoke archive_read_next_header2() to work 550fd082e96SMartin Matuska * archive_read_data_block(), which is called via write_file(), 551fd082e96SMartin Matuska * without failure. 552fd082e96SMartin Matuska */ 553fd082e96SMartin Matuska entry2 = archive_entry_new(); 554fd082e96SMartin Matuska r = archive_read_next_header2(disk, entry2); 555fd082e96SMartin Matuska archive_entry_free(entry2); 556fd082e96SMartin Matuska if (r != ARCHIVE_OK) { 557fd082e96SMartin Matuska lafe_warnc(archive_errno(disk), 558fd082e96SMartin Matuska "%s", archive_error_string(disk)); 559fd082e96SMartin Matuska if (r == ARCHIVE_FATAL) 560fd082e96SMartin Matuska bsdtar->return_value = 1; 561fd082e96SMartin Matuska else 562fd082e96SMartin Matuska archive_read_close(disk); 563fd082e96SMartin Matuska archive_entry_free(entry); 564fd082e96SMartin Matuska continue; 565fd082e96SMartin Matuska } 566fd082e96SMartin Matuska 5676c95142eSMartin Matuska write_file(bsdtar, a, entry); 568caf54c4fSMartin Matuska archive_entry_free(entry); 569fd082e96SMartin Matuska archive_read_close(disk); 570caf54c4fSMartin Matuska entry = NULL; 571caf54c4fSMartin Matuska archive_entry_linkify(bsdtar->resolver, &entry, &sparse_entry); 572caf54c4fSMartin Matuska } 573caf54c4fSMartin Matuska 574caf54c4fSMartin Matuska if (archive_write_close(a)) { 575caf54c4fSMartin Matuska lafe_warnc(0, "%s", archive_error_string(a)); 576caf54c4fSMartin Matuska bsdtar->return_value = 1; 577caf54c4fSMartin Matuska } 578caf54c4fSMartin Matuska 579caf54c4fSMartin Matuska cleanup: 580caf54c4fSMartin Matuska /* Free file data buffer. */ 581caf54c4fSMartin Matuska free(bsdtar->buff); 582caf54c4fSMartin Matuska archive_entry_linkresolver_free(bsdtar->resolver); 583caf54c4fSMartin Matuska bsdtar->resolver = NULL; 5846c95142eSMartin Matuska archive_read_free(bsdtar->diskreader); 585caf54c4fSMartin Matuska bsdtar->diskreader = NULL; 586caf54c4fSMartin Matuska 587caf54c4fSMartin Matuska if (bsdtar->option_totals) { 588caf54c4fSMartin Matuska fprintf(stderr, "Total bytes written: %s\n", 589acc60b03SMartin Matuska tar_i64toa(archive_filter_bytes(a, -1))); 590caf54c4fSMartin Matuska } 591caf54c4fSMartin Matuska 5926c95142eSMartin Matuska archive_write_free(a); 593caf54c4fSMartin Matuska } 594caf54c4fSMartin Matuska 595caf54c4fSMartin Matuska /* 596caf54c4fSMartin Matuska * Archive names specified in file. 597caf54c4fSMartin Matuska * 598caf54c4fSMartin Matuska * Unless --null was specified, a line containing exactly "-C" will 599caf54c4fSMartin Matuska * cause the next line to be a directory to pass to chdir(). If 600caf54c4fSMartin Matuska * --null is specified, then a line "-C" is just another filename. 601caf54c4fSMartin Matuska */ 602caf54c4fSMartin Matuska static void 603caf54c4fSMartin Matuska archive_names_from_file(struct bsdtar *bsdtar, struct archive *a) 604caf54c4fSMartin Matuska { 605caf54c4fSMartin Matuska struct lafe_line_reader *lr; 606caf54c4fSMartin Matuska const char *line; 607caf54c4fSMartin Matuska 608caf54c4fSMartin Matuska bsdtar->next_line_is_dir = 0; 609caf54c4fSMartin Matuska 610caf54c4fSMartin Matuska lr = lafe_line_reader(bsdtar->names_from_file, bsdtar->option_null); 611caf54c4fSMartin Matuska while ((line = lafe_line_reader_next(lr)) != NULL) { 612caf54c4fSMartin Matuska if (bsdtar->next_line_is_dir) { 6136c95142eSMartin Matuska if (*line != '\0') 614caf54c4fSMartin Matuska set_chdir(bsdtar, line); 6156c95142eSMartin Matuska else { 6166c95142eSMartin Matuska lafe_warnc(0, 6176c95142eSMartin Matuska "Meaningless argument for -C: ''"); 6186c95142eSMartin Matuska bsdtar->return_value = 1; 6196c95142eSMartin Matuska } 620caf54c4fSMartin Matuska bsdtar->next_line_is_dir = 0; 621caf54c4fSMartin Matuska } else if (!bsdtar->option_null && strcmp(line, "-C") == 0) 622caf54c4fSMartin Matuska bsdtar->next_line_is_dir = 1; 623caf54c4fSMartin Matuska else { 624caf54c4fSMartin Matuska if (*line != '/') 625caf54c4fSMartin Matuska do_chdir(bsdtar); /* Handle a deferred -C */ 626caf54c4fSMartin Matuska write_hierarchy(bsdtar, a, line); 627caf54c4fSMartin Matuska } 628caf54c4fSMartin Matuska } 629caf54c4fSMartin Matuska lafe_line_reader_free(lr); 630caf54c4fSMartin Matuska if (bsdtar->next_line_is_dir) 631caf54c4fSMartin Matuska lafe_errc(1, errno, 632caf54c4fSMartin Matuska "Unexpected end of filename list; " 633caf54c4fSMartin Matuska "directory expected after -C"); 634caf54c4fSMartin Matuska } 635caf54c4fSMartin Matuska 636caf54c4fSMartin Matuska /* 637caf54c4fSMartin Matuska * Copy from specified archive to current archive. Returns non-zero 638caf54c4fSMartin Matuska * for write errors (which force us to terminate the entire archiving 639caf54c4fSMartin Matuska * operation). If there are errors reading the input archive, we set 640caf54c4fSMartin Matuska * bsdtar->return_value but return zero, so the overall archiving 641caf54c4fSMartin Matuska * operation will complete and return non-zero. 642caf54c4fSMartin Matuska */ 643caf54c4fSMartin Matuska static int 644caf54c4fSMartin Matuska append_archive_filename(struct bsdtar *bsdtar, struct archive *a, 6456c95142eSMartin Matuska const char *raw_filename) 646caf54c4fSMartin Matuska { 647caf54c4fSMartin Matuska struct archive *ina; 6486c95142eSMartin Matuska const char *filename = raw_filename; 649caf54c4fSMartin Matuska int rc; 650caf54c4fSMartin Matuska 651caf54c4fSMartin Matuska if (strcmp(filename, "-") == 0) 652caf54c4fSMartin Matuska filename = NULL; /* Library uses NULL for stdio. */ 653caf54c4fSMartin Matuska 654caf54c4fSMartin Matuska ina = archive_read_new(); 655caf54c4fSMartin Matuska archive_read_support_format_all(ina); 6566c95142eSMartin Matuska archive_read_support_filter_all(ina); 657cdf63a70SMartin Matuska set_reader_options(bsdtar, ina); 658cdf63a70SMartin Matuska archive_read_set_options(ina, "mtree:checkfs"); 659cdf63a70SMartin Matuska if (bsdtar->passphrase != NULL) 660cdf63a70SMartin Matuska rc = archive_read_add_passphrase(a, bsdtar->passphrase); 661cdf63a70SMartin Matuska else 662cdf63a70SMartin Matuska rc = archive_read_set_passphrase_callback(ina, bsdtar, 663cdf63a70SMartin Matuska &passphrase_callback); 664cdf63a70SMartin Matuska if (rc != ARCHIVE_OK) 665cdf63a70SMartin Matuska lafe_errc(1, 0, "%s", archive_error_string(a)); 666acc60b03SMartin Matuska if (archive_read_open_filename(ina, filename, 667acc60b03SMartin Matuska bsdtar->bytes_per_block)) { 668caf54c4fSMartin Matuska lafe_warnc(0, "%s", archive_error_string(ina)); 669caf54c4fSMartin Matuska bsdtar->return_value = 1; 670caf54c4fSMartin Matuska return (0); 671caf54c4fSMartin Matuska } 672caf54c4fSMartin Matuska 673caf54c4fSMartin Matuska rc = append_archive(bsdtar, a, ina); 674caf54c4fSMartin Matuska 675caf54c4fSMartin Matuska if (rc != ARCHIVE_OK) { 676caf54c4fSMartin Matuska lafe_warnc(0, "Error reading archive %s: %s", 6776c95142eSMartin Matuska raw_filename, archive_error_string(ina)); 678caf54c4fSMartin Matuska bsdtar->return_value = 1; 679caf54c4fSMartin Matuska } 6806c95142eSMartin Matuska archive_read_free(ina); 681caf54c4fSMartin Matuska 682caf54c4fSMartin Matuska return (rc); 683caf54c4fSMartin Matuska } 684caf54c4fSMartin Matuska 685caf54c4fSMartin Matuska static int 686caf54c4fSMartin Matuska append_archive(struct bsdtar *bsdtar, struct archive *a, struct archive *ina) 687caf54c4fSMartin Matuska { 688caf54c4fSMartin Matuska struct archive_entry *in_entry; 689caf54c4fSMartin Matuska int e; 690caf54c4fSMartin Matuska 691caf54c4fSMartin Matuska while (ARCHIVE_OK == (e = archive_read_next_header(ina, &in_entry))) { 692fd082e96SMartin Matuska if (archive_match_excluded(bsdtar->matching, in_entry)) 693caf54c4fSMartin Matuska continue; 694caf54c4fSMartin Matuska if (bsdtar->option_interactive && 695caf54c4fSMartin Matuska !yes("copy '%s'", archive_entry_pathname(in_entry))) 696caf54c4fSMartin Matuska continue; 697cdf63a70SMartin Matuska if (bsdtar->verbose > 1) { 698cdf63a70SMartin Matuska safe_fprintf(stderr, "a "); 699cdf63a70SMartin Matuska list_item_verbose(bsdtar, stderr, in_entry); 700cdf63a70SMartin Matuska } else if (bsdtar->verbose > 0) 701caf54c4fSMartin Matuska safe_fprintf(stderr, "a %s", 702caf54c4fSMartin Matuska archive_entry_pathname(in_entry)); 703caf54c4fSMartin Matuska if (need_report()) 704caf54c4fSMartin Matuska report_write(bsdtar, a, in_entry, 0); 705caf54c4fSMartin Matuska 706caf54c4fSMartin Matuska e = archive_write_header(a, in_entry); 707caf54c4fSMartin Matuska if (e != ARCHIVE_OK) { 708caf54c4fSMartin Matuska if (!bsdtar->verbose) 709caf54c4fSMartin Matuska lafe_warnc(0, "%s: %s", 710caf54c4fSMartin Matuska archive_entry_pathname(in_entry), 711caf54c4fSMartin Matuska archive_error_string(a)); 712caf54c4fSMartin Matuska else 713caf54c4fSMartin Matuska fprintf(stderr, ": %s", archive_error_string(a)); 714caf54c4fSMartin Matuska } 715caf54c4fSMartin Matuska if (e == ARCHIVE_FATAL) 716caf54c4fSMartin Matuska exit(1); 717caf54c4fSMartin Matuska 718caf54c4fSMartin Matuska if (e >= ARCHIVE_WARN) { 719caf54c4fSMartin Matuska if (archive_entry_size(in_entry) == 0) 720caf54c4fSMartin Matuska archive_read_data_skip(ina); 721fd082e96SMartin Matuska else if (copy_file_data_block(bsdtar, a, ina, in_entry)) 722caf54c4fSMartin Matuska exit(1); 723caf54c4fSMartin Matuska } 724caf54c4fSMartin Matuska 725caf54c4fSMartin Matuska if (bsdtar->verbose) 726caf54c4fSMartin Matuska fprintf(stderr, "\n"); 727caf54c4fSMartin Matuska } 728caf54c4fSMartin Matuska 729caf54c4fSMartin Matuska return (e == ARCHIVE_EOF ? ARCHIVE_OK : e); 730caf54c4fSMartin Matuska } 731caf54c4fSMartin Matuska 732fd082e96SMartin Matuska /* Helper function to copy file to archive. */ 733caf54c4fSMartin Matuska static int 734fd082e96SMartin Matuska copy_file_data_block(struct bsdtar *bsdtar, struct archive *a, 735fd082e96SMartin Matuska struct archive *in_a, struct archive_entry *entry) 736caf54c4fSMartin Matuska { 737fd082e96SMartin Matuska size_t bytes_read; 738caf54c4fSMartin Matuska ssize_t bytes_written; 739fd082e96SMartin Matuska int64_t offset, progress = 0; 740fd082e96SMartin Matuska char *null_buff = NULL; 741fd082e96SMartin Matuska const void *buff; 742fd082e96SMartin Matuska int r; 743caf54c4fSMartin Matuska 744fd082e96SMartin Matuska while ((r = archive_read_data_block(in_a, &buff, 745fd082e96SMartin Matuska &bytes_read, &offset)) == ARCHIVE_OK) { 746caf54c4fSMartin Matuska if (need_report()) 747caf54c4fSMartin Matuska report_write(bsdtar, a, entry, progress); 748caf54c4fSMartin Matuska 749fd082e96SMartin Matuska if (offset > progress) { 750fd082e96SMartin Matuska int64_t sparse = offset - progress; 751fd082e96SMartin Matuska size_t ns; 752fd082e96SMartin Matuska 753fd082e96SMartin Matuska if (null_buff == NULL) { 754fd082e96SMartin Matuska null_buff = bsdtar->buff; 755fd082e96SMartin Matuska memset(null_buff, 0, bsdtar->buff_size); 756fd082e96SMartin Matuska } 757fd082e96SMartin Matuska 758fd082e96SMartin Matuska while (sparse > 0) { 759fd082e96SMartin Matuska if (sparse > (int64_t)bsdtar->buff_size) 760fd082e96SMartin Matuska ns = bsdtar->buff_size; 761fd082e96SMartin Matuska else 762fd082e96SMartin Matuska ns = (size_t)sparse; 763fd082e96SMartin Matuska bytes_written = 764fd082e96SMartin Matuska archive_write_data(a, null_buff, ns); 765fd082e96SMartin Matuska if (bytes_written < 0) { 766fd082e96SMartin Matuska /* Write failed; this is bad */ 767fd082e96SMartin Matuska lafe_warnc(0, "%s", 768fd082e96SMartin Matuska archive_error_string(a)); 769fd082e96SMartin Matuska return (-1); 770fd082e96SMartin Matuska } 771fd082e96SMartin Matuska if ((size_t)bytes_written < ns) { 772fd082e96SMartin Matuska /* Write was truncated; warn but 773fd082e96SMartin Matuska * continue. */ 774fd082e96SMartin Matuska lafe_warnc(0, 775fd082e96SMartin Matuska "%s: Truncated write; file may " 776fd082e96SMartin Matuska "have grown while being archived.", 777fd082e96SMartin Matuska archive_entry_pathname(entry)); 778fd082e96SMartin Matuska return (0); 779fd082e96SMartin Matuska } 780fd082e96SMartin Matuska progress += bytes_written; 781fd082e96SMartin Matuska sparse -= bytes_written; 782fd082e96SMartin Matuska } 783fd082e96SMartin Matuska } 784fd082e96SMartin Matuska 785fd082e96SMartin Matuska bytes_written = archive_write_data(a, buff, bytes_read); 786fd082e96SMartin Matuska if (bytes_written < 0) { 787fd082e96SMartin Matuska /* Write failed; this is bad */ 788caf54c4fSMartin Matuska lafe_warnc(0, "%s", archive_error_string(a)); 789caf54c4fSMartin Matuska return (-1); 790caf54c4fSMartin Matuska } 791fd082e96SMartin Matuska if ((size_t)bytes_written < bytes_read) { 792fd082e96SMartin Matuska /* Write was truncated; warn but continue. */ 793fd082e96SMartin Matuska lafe_warnc(0, 794fd082e96SMartin Matuska "%s: Truncated write; file may have grown " 795fd082e96SMartin Matuska "while being archived.", 796fd082e96SMartin Matuska archive_entry_pathname(entry)); 797fd082e96SMartin Matuska return (0); 798caf54c4fSMartin Matuska } 799fd082e96SMartin Matuska progress += bytes_written; 800fd082e96SMartin Matuska } 801fd082e96SMartin Matuska if (r < ARCHIVE_WARN) { 802fd082e96SMartin Matuska lafe_warnc(archive_errno(a), "%s", archive_error_string(a)); 803fd082e96SMartin Matuska return (-1); 804fd082e96SMartin Matuska } 805caf54c4fSMartin Matuska return (0); 806caf54c4fSMartin Matuska } 807caf54c4fSMartin Matuska 808caf54c4fSMartin Matuska static void 809fd082e96SMartin Matuska excluded_callback(struct archive *a, void *_data, struct archive_entry *entry) 810caf54c4fSMartin Matuska { 811fd082e96SMartin Matuska struct bsdtar *bsdtar = (struct bsdtar *)_data; 812caf54c4fSMartin Matuska 813c5d360f8SMartin Matuska if (bsdtar->option_no_subdirs) 814fd082e96SMartin Matuska return; 815fd082e96SMartin Matuska if (!archive_read_disk_can_descend(a)) 816fd082e96SMartin Matuska return; 817fd082e96SMartin Matuska if (bsdtar->option_interactive && 818fd082e96SMartin Matuska !yes("add '%s'", archive_entry_pathname(entry))) 819fd082e96SMartin Matuska return; 820fd082e96SMartin Matuska archive_read_disk_descend(a); 821fd082e96SMartin Matuska } 822c5d360f8SMartin Matuska 823fd082e96SMartin Matuska static int 824fd082e96SMartin Matuska metadata_filter(struct archive *a, void *_data, struct archive_entry *entry) 825fd082e96SMartin Matuska { 826fd082e96SMartin Matuska struct bsdtar *bsdtar = (struct bsdtar *)_data; 827caf54c4fSMartin Matuska 828caf54c4fSMartin Matuska /* XXX TODO: check whether this filesystem is 829caf54c4fSMartin Matuska * synthetic and/or local. Add a new 830caf54c4fSMartin Matuska * --local-only option to skip non-local 831caf54c4fSMartin Matuska * filesystems. Skip synthetic filesystems 832caf54c4fSMartin Matuska * regardless. 833caf54c4fSMartin Matuska * 834caf54c4fSMartin Matuska * The results should be cached, since 835caf54c4fSMartin Matuska * tree.c doesn't usually visit a directory 836caf54c4fSMartin Matuska * and the directory contents together. A simple 837caf54c4fSMartin Matuska * move-to-front list should perform quite well. 838caf54c4fSMartin Matuska * 839fd082e96SMartin Matuska * Use archive_read_disk_current_filesystem_is_remote(). 840caf54c4fSMartin Matuska */ 841fd082e96SMartin Matuska 842fd082e96SMartin Matuska /* 843fd082e96SMartin Matuska * If the user vetoes this file/directory, skip it. 844fd082e96SMartin Matuska * We want this to be fairly late; if some other 845fd082e96SMartin Matuska * check would veto this file, we shouldn't bother 846fd082e96SMartin Matuska * the user with it. 847fd082e96SMartin Matuska */ 848fd082e96SMartin Matuska if (bsdtar->option_interactive && 849fd082e96SMartin Matuska !yes("add '%s'", archive_entry_pathname(entry))) 850fd082e96SMartin Matuska return (0); 851fd082e96SMartin Matuska 852fd082e96SMartin Matuska /* Note: if user vetoes, we won't descend. */ 853fd082e96SMartin Matuska if (!bsdtar->option_no_subdirs && archive_read_disk_can_descend(a)) 854fd082e96SMartin Matuska archive_read_disk_descend(a); 855fd082e96SMartin Matuska 856fd082e96SMartin Matuska return (1); 857caf54c4fSMartin Matuska } 858caf54c4fSMartin Matuska 859caf54c4fSMartin Matuska /* 860fd082e96SMartin Matuska * Add the file or dir hierarchy named by 'path' to the archive 861caf54c4fSMartin Matuska */ 862fd082e96SMartin Matuska static void 863fd082e96SMartin Matuska write_hierarchy(struct bsdtar *bsdtar, struct archive *a, const char *path) 864fd082e96SMartin Matuska { 865fd082e96SMartin Matuska struct archive *disk = bsdtar->diskreader; 866fd082e96SMartin Matuska struct archive_entry *entry = NULL, *spare_entry = NULL; 867fd082e96SMartin Matuska int r; 868fd082e96SMartin Matuska 869fd082e96SMartin Matuska r = archive_read_disk_open(disk, path); 870fd082e96SMartin Matuska if (r != ARCHIVE_OK) { 871fd082e96SMartin Matuska lafe_warnc(archive_errno(disk), 872fd082e96SMartin Matuska "%s", archive_error_string(disk)); 873fd082e96SMartin Matuska bsdtar->return_value = 1; 874fd082e96SMartin Matuska return; 875fd082e96SMartin Matuska } 876fd082e96SMartin Matuska bsdtar->first_fs = -1; 877fd082e96SMartin Matuska 878fd082e96SMartin Matuska for (;;) { 879fd082e96SMartin Matuska archive_entry_free(entry); 880fd082e96SMartin Matuska entry = archive_entry_new(); 881fd082e96SMartin Matuska r = archive_read_next_header2(disk, entry); 882fd082e96SMartin Matuska if (r == ARCHIVE_EOF) 883fd082e96SMartin Matuska break; 884fd082e96SMartin Matuska else if (r != ARCHIVE_OK) { 885fd082e96SMartin Matuska lafe_warnc(archive_errno(disk), 886fd082e96SMartin Matuska "%s", archive_error_string(disk)); 887f061a221SMartin Matuska if (r == ARCHIVE_FATAL || r == ARCHIVE_FAILED) { 888fd082e96SMartin Matuska bsdtar->return_value = 1; 889f3e9b21aSMartin Matuska archive_entry_free(entry); 890f3e9b21aSMartin Matuska archive_read_close(disk); 891fd082e96SMartin Matuska return; 892fd082e96SMartin Matuska } else if (r < ARCHIVE_WARN) 893c5d360f8SMartin Matuska continue; 894c5d360f8SMartin Matuska } 895caf54c4fSMartin Matuska 896caf54c4fSMartin Matuska if (bsdtar->uid >= 0) { 897caf54c4fSMartin Matuska archive_entry_set_uid(entry, bsdtar->uid); 898caf54c4fSMartin Matuska if (!bsdtar->uname) 899caf54c4fSMartin Matuska archive_entry_set_uname(entry, 900caf54c4fSMartin Matuska archive_read_disk_uname(bsdtar->diskreader, 901caf54c4fSMartin Matuska bsdtar->uid)); 902caf54c4fSMartin Matuska } 903caf54c4fSMartin Matuska if (bsdtar->gid >= 0) { 904caf54c4fSMartin Matuska archive_entry_set_gid(entry, bsdtar->gid); 905caf54c4fSMartin Matuska if (!bsdtar->gname) 906caf54c4fSMartin Matuska archive_entry_set_gname(entry, 907caf54c4fSMartin Matuska archive_read_disk_gname(bsdtar->diskreader, 908caf54c4fSMartin Matuska bsdtar->gid)); 909caf54c4fSMartin Matuska } 910caf54c4fSMartin Matuska if (bsdtar->uname) 911caf54c4fSMartin Matuska archive_entry_set_uname(entry, bsdtar->uname); 912caf54c4fSMartin Matuska if (bsdtar->gname) 913caf54c4fSMartin Matuska archive_entry_set_gname(entry, bsdtar->gname); 914caf54c4fSMartin Matuska 915caf54c4fSMartin Matuska /* 916caf54c4fSMartin Matuska * Rewrite the pathname to be archived. If rewrite 917caf54c4fSMartin Matuska * fails, skip the entry. 918caf54c4fSMartin Matuska */ 919caf54c4fSMartin Matuska if (edit_pathname(bsdtar, entry)) 920caf54c4fSMartin Matuska continue; 921caf54c4fSMartin Matuska 922cdf63a70SMartin Matuska /* Display entry as we process it. */ 923cdf63a70SMartin Matuska if (bsdtar->verbose > 1) { 924cdf63a70SMartin Matuska safe_fprintf(stderr, "a "); 925cdf63a70SMartin Matuska list_item_verbose(bsdtar, stderr, entry); 926cdf63a70SMartin Matuska } else if (bsdtar->verbose > 0) { 927cdf63a70SMartin Matuska /* This format is required by SUSv2. */ 928caf54c4fSMartin Matuska safe_fprintf(stderr, "a %s", 929caf54c4fSMartin Matuska archive_entry_pathname(entry)); 930cdf63a70SMartin Matuska } 931caf54c4fSMartin Matuska 932caf54c4fSMartin Matuska /* Non-regular files get archived with zero size. */ 933caf54c4fSMartin Matuska if (archive_entry_filetype(entry) != AE_IFREG) 934caf54c4fSMartin Matuska archive_entry_set_size(entry, 0); 935caf54c4fSMartin Matuska 936caf54c4fSMartin Matuska archive_entry_linkify(bsdtar->resolver, &entry, &spare_entry); 937caf54c4fSMartin Matuska 938caf54c4fSMartin Matuska while (entry != NULL) { 9396c95142eSMartin Matuska write_file(bsdtar, a, entry); 940caf54c4fSMartin Matuska archive_entry_free(entry); 941caf54c4fSMartin Matuska entry = spare_entry; 942caf54c4fSMartin Matuska spare_entry = NULL; 943caf54c4fSMartin Matuska } 944caf54c4fSMartin Matuska 945caf54c4fSMartin Matuska if (bsdtar->verbose) 946caf54c4fSMartin Matuska fprintf(stderr, "\n"); 947caf54c4fSMartin Matuska } 948caf54c4fSMartin Matuska archive_entry_free(entry); 949fd082e96SMartin Matuska archive_read_close(disk); 950caf54c4fSMartin Matuska } 951caf54c4fSMartin Matuska 952caf54c4fSMartin Matuska /* 9536c95142eSMartin Matuska * Write a single file (or directory or other filesystem object) to 9546c95142eSMartin Matuska * the archive. 955caf54c4fSMartin Matuska */ 956caf54c4fSMartin Matuska static void 9576c95142eSMartin Matuska write_file(struct bsdtar *bsdtar, struct archive *a, 9586c95142eSMartin Matuska struct archive_entry *entry) 9596c95142eSMartin Matuska { 9606c95142eSMartin Matuska write_entry(bsdtar, a, entry); 9616c95142eSMartin Matuska } 9626c95142eSMartin Matuska 9636c95142eSMartin Matuska /* 9646c95142eSMartin Matuska * Write a single entry to the archive. 9656c95142eSMartin Matuska */ 9666c95142eSMartin Matuska static void 9676c95142eSMartin Matuska write_entry(struct bsdtar *bsdtar, struct archive *a, 968caf54c4fSMartin Matuska struct archive_entry *entry) 969caf54c4fSMartin Matuska { 970caf54c4fSMartin Matuska int e; 971caf54c4fSMartin Matuska 972caf54c4fSMartin Matuska e = archive_write_header(a, entry); 973caf54c4fSMartin Matuska if (e != ARCHIVE_OK) { 974cdf63a70SMartin Matuska if (bsdtar->verbose > 1) { 975cdf63a70SMartin Matuska safe_fprintf(stderr, "a "); 976cdf63a70SMartin Matuska list_item_verbose(bsdtar, stderr, entry); 977cdf63a70SMartin Matuska lafe_warnc(0, ": %s", archive_error_string(a)); 978cdf63a70SMartin Matuska } else if (bsdtar->verbose > 0) { 979caf54c4fSMartin Matuska lafe_warnc(0, "%s: %s", 980caf54c4fSMartin Matuska archive_entry_pathname(entry), 981caf54c4fSMartin Matuska archive_error_string(a)); 982cdf63a70SMartin Matuska } else 983caf54c4fSMartin Matuska fprintf(stderr, ": %s", archive_error_string(a)); 984caf54c4fSMartin Matuska } 985caf54c4fSMartin Matuska 986caf54c4fSMartin Matuska if (e == ARCHIVE_FATAL) 987caf54c4fSMartin Matuska exit(1); 988caf54c4fSMartin Matuska 989caf54c4fSMartin Matuska /* 990caf54c4fSMartin Matuska * If we opened a file earlier, write it out now. Note that 991caf54c4fSMartin Matuska * the format handler might have reset the size field to zero 992caf54c4fSMartin Matuska * to inform us that the archive body won't get stored. In 993caf54c4fSMartin Matuska * that case, just skip the write. 994caf54c4fSMartin Matuska */ 995fd082e96SMartin Matuska if (e >= ARCHIVE_WARN && archive_entry_size(entry) > 0) { 996fd082e96SMartin Matuska if (copy_file_data_block(bsdtar, a, bsdtar->diskreader, entry)) 997caf54c4fSMartin Matuska exit(1); 998caf54c4fSMartin Matuska } 999caf54c4fSMartin Matuska } 1000caf54c4fSMartin Matuska 1001caf54c4fSMartin Matuska static void 1002caf54c4fSMartin Matuska report_write(struct bsdtar *bsdtar, struct archive *a, 1003caf54c4fSMartin Matuska struct archive_entry *entry, int64_t progress) 1004caf54c4fSMartin Matuska { 1005caf54c4fSMartin Matuska uint64_t comp, uncomp; 1006c5d360f8SMartin Matuska int compression; 1007c5d360f8SMartin Matuska 1008caf54c4fSMartin Matuska if (bsdtar->verbose) 1009caf54c4fSMartin Matuska fprintf(stderr, "\n"); 1010acc60b03SMartin Matuska comp = archive_filter_bytes(a, -1); 1011acc60b03SMartin Matuska uncomp = archive_filter_bytes(a, 0); 1012caf54c4fSMartin Matuska fprintf(stderr, "In: %d files, %s bytes;", 1013caf54c4fSMartin Matuska archive_file_count(a), tar_i64toa(uncomp)); 1014c5d360f8SMartin Matuska if (comp > uncomp) 1015c5d360f8SMartin Matuska compression = 0; 1016c5d360f8SMartin Matuska else 1017c5d360f8SMartin Matuska compression = (int)((uncomp - comp) * 100 / uncomp); 1018caf54c4fSMartin Matuska fprintf(stderr, 1019caf54c4fSMartin Matuska " Out: %s bytes, compression %d%%\n", 1020c5d360f8SMartin Matuska tar_i64toa(comp), compression); 1021caf54c4fSMartin Matuska /* Can't have two calls to tar_i64toa() pending, so split the output. */ 1022caf54c4fSMartin Matuska safe_fprintf(stderr, "Current: %s (%s", 1023caf54c4fSMartin Matuska archive_entry_pathname(entry), 1024caf54c4fSMartin Matuska tar_i64toa(progress)); 1025caf54c4fSMartin Matuska fprintf(stderr, "/%s bytes)\n", 1026caf54c4fSMartin Matuska tar_i64toa(archive_entry_size(entry))); 1027caf54c4fSMartin Matuska } 1028caf54c4fSMartin Matuska 1029caf54c4fSMartin Matuska static void 1030caf54c4fSMartin Matuska test_for_append(struct bsdtar *bsdtar) 1031caf54c4fSMartin Matuska { 1032caf54c4fSMartin Matuska struct stat s; 1033caf54c4fSMartin Matuska 1034caf54c4fSMartin Matuska if (*bsdtar->argv == NULL && bsdtar->names_from_file == NULL) 1035caf54c4fSMartin Matuska lafe_errc(1, 0, "no files or directories specified"); 1036caf54c4fSMartin Matuska if (bsdtar->filename == NULL) 1037caf54c4fSMartin Matuska lafe_errc(1, 0, "Cannot append to stdout."); 1038caf54c4fSMartin Matuska 1039caf54c4fSMartin Matuska if (stat(bsdtar->filename, &s) != 0) 1040caf54c4fSMartin Matuska return; 1041caf54c4fSMartin Matuska 1042caf54c4fSMartin Matuska if (!S_ISREG(s.st_mode) && !S_ISBLK(s.st_mode)) 1043caf54c4fSMartin Matuska lafe_errc(1, 0, 1044caf54c4fSMartin Matuska "Cannot append to %s: not a regular file.", 1045caf54c4fSMartin Matuska bsdtar->filename); 1046caf54c4fSMartin Matuska 1047caf54c4fSMartin Matuska /* Is this an appropriate check here on Windows? */ 1048caf54c4fSMartin Matuska /* 1049caf54c4fSMartin Matuska if (GetFileType(handle) != FILE_TYPE_DISK) 1050caf54c4fSMartin Matuska lafe_errc(1, 0, "Cannot append"); 1051caf54c4fSMartin Matuska */ 1052caf54c4fSMartin Matuska 1053caf54c4fSMartin Matuska } 1054