archive_write_disk_posix.c revision 358926
1/*- 2 * Copyright (c) 2003-2010 Tim Kientzle 3 * Copyright (c) 2012 Michihiro NAKAJIMA 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer 11 * in this position and unchanged. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 19 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28#include "archive_platform.h" 29__FBSDID("$FreeBSD$"); 30 31#if !defined(_WIN32) || defined(__CYGWIN__) 32 33#ifdef HAVE_SYS_TYPES_H 34#include <sys/types.h> 35#endif 36#ifdef HAVE_SYS_ACL_H 37#include <sys/acl.h> 38#endif 39#ifdef HAVE_SYS_EXTATTR_H 40#include <sys/extattr.h> 41#endif 42#if HAVE_SYS_XATTR_H 43#include <sys/xattr.h> 44#elif HAVE_ATTR_XATTR_H 45#include <attr/xattr.h> 46#endif 47#ifdef HAVE_SYS_EA_H 48#include <sys/ea.h> 49#endif 50#ifdef HAVE_SYS_IOCTL_H 51#include <sys/ioctl.h> 52#endif 53#ifdef HAVE_SYS_STAT_H 54#include <sys/stat.h> 55#endif 56#ifdef HAVE_SYS_TIME_H 57#include <sys/time.h> 58#endif 59#ifdef HAVE_SYS_UTIME_H 60#include <sys/utime.h> 61#endif 62#ifdef HAVE_COPYFILE_H 63#include <copyfile.h> 64#endif 65#ifdef HAVE_ERRNO_H 66#include <errno.h> 67#endif 68#ifdef HAVE_FCNTL_H 69#include <fcntl.h> 70#endif 71#ifdef HAVE_GRP_H 72#include <grp.h> 73#endif 74#ifdef HAVE_LANGINFO_H 75#include <langinfo.h> 76#endif 77#ifdef HAVE_LINUX_FS_H 78#include <linux/fs.h> /* for Linux file flags */ 79#endif 80/* 81 * Some Linux distributions have both linux/ext2_fs.h and ext2fs/ext2_fs.h. 82 * As the include guards don't agree, the order of include is important. 83 */ 84#ifdef HAVE_LINUX_EXT2_FS_H 85#include <linux/ext2_fs.h> /* for Linux file flags */ 86#endif 87#if defined(HAVE_EXT2FS_EXT2_FS_H) && !defined(__CYGWIN__) 88#include <ext2fs/ext2_fs.h> /* Linux file flags, broken on Cygwin */ 89#endif 90#ifdef HAVE_LIMITS_H 91#include <limits.h> 92#endif 93#ifdef HAVE_PWD_H 94#include <pwd.h> 95#endif 96#include <stdio.h> 97#ifdef HAVE_STDLIB_H 98#include <stdlib.h> 99#endif 100#ifdef HAVE_STRING_H 101#include <string.h> 102#endif 103#ifdef HAVE_UNISTD_H 104#include <unistd.h> 105#endif 106#ifdef HAVE_UTIME_H 107#include <utime.h> 108#endif 109#ifdef F_GETTIMES /* Tru64 specific */ 110#include <sys/fcntl1.h> 111#endif 112 113/* 114 * Macro to cast st_mtime and time_t to an int64 so that 2 numbers can reliably be compared. 115 * 116 * It assumes that the input is an integer type of no more than 64 bits. 117 * If the number is less than zero, t must be a signed type, so it fits in 118 * int64_t. Otherwise, it's a nonnegative value so we can cast it to uint64_t 119 * without loss. But it could be a large unsigned value, so we have to clip it 120 * to INT64_MAX.* 121 */ 122#define to_int64_time(t) \ 123 ((t) < 0 ? (int64_t)(t) : (uint64_t)(t) > (uint64_t)INT64_MAX ? INT64_MAX : (int64_t)(t)) 124 125#if __APPLE__ 126#include <TargetConditionals.h> 127#if TARGET_OS_MAC && !TARGET_OS_EMBEDDED && HAVE_QUARANTINE_H 128#include <quarantine.h> 129#define HAVE_QUARANTINE 1 130#endif 131#endif 132 133#ifdef HAVE_ZLIB_H 134#include <zlib.h> 135#endif 136 137/* TODO: Support Mac OS 'quarantine' feature. This is really just a 138 * standard tag to mark files that have been downloaded as "tainted". 139 * On Mac OS, we should mark the extracted files as tainted if the 140 * archive being read was tainted. Windows has a similar feature; we 141 * should investigate ways to support this generically. */ 142 143#include "archive.h" 144#include "archive_acl_private.h" 145#include "archive_string.h" 146#include "archive_endian.h" 147#include "archive_entry.h" 148#include "archive_private.h" 149#include "archive_write_disk_private.h" 150 151#ifndef O_BINARY 152#define O_BINARY 0 153#endif 154#ifndef O_CLOEXEC 155#define O_CLOEXEC 0 156#endif 157 158/* Ignore non-int O_NOFOLLOW constant. */ 159/* gnulib's fcntl.h does this on AIX, but it seems practical everywhere */ 160#if defined O_NOFOLLOW && !(INT_MIN <= O_NOFOLLOW && O_NOFOLLOW <= INT_MAX) 161#undef O_NOFOLLOW 162#endif 163 164#ifndef O_NOFOLLOW 165#define O_NOFOLLOW 0 166#endif 167 168#ifndef AT_FDCWD 169#define AT_FDCWD -100 170#endif 171 172struct fixup_entry { 173 struct fixup_entry *next; 174 struct archive_acl acl; 175 mode_t mode; 176 int64_t atime; 177 int64_t birthtime; 178 int64_t mtime; 179 int64_t ctime; 180 unsigned long atime_nanos; 181 unsigned long birthtime_nanos; 182 unsigned long mtime_nanos; 183 unsigned long ctime_nanos; 184 unsigned long fflags_set; 185 size_t mac_metadata_size; 186 void *mac_metadata; 187 int fixup; /* bitmask of what needs fixing */ 188 char *name; 189}; 190 191/* 192 * We use a bitmask to track which operations remain to be done for 193 * this file. In particular, this helps us avoid unnecessary 194 * operations when it's possible to take care of one step as a 195 * side-effect of another. For example, mkdir() can specify the mode 196 * for the newly-created object but symlink() cannot. This means we 197 * can skip chmod() if mkdir() succeeded, but we must explicitly 198 * chmod() if we're trying to create a directory that already exists 199 * (mkdir() failed) or if we're restoring a symlink. Similarly, we 200 * need to verify UID/GID before trying to restore SUID/SGID bits; 201 * that verification can occur explicitly through a stat() call or 202 * implicitly because of a successful chown() call. 203 */ 204#define TODO_MODE_FORCE 0x40000000 205#define TODO_MODE_BASE 0x20000000 206#define TODO_SUID 0x10000000 207#define TODO_SUID_CHECK 0x08000000 208#define TODO_SGID 0x04000000 209#define TODO_SGID_CHECK 0x02000000 210#define TODO_APPLEDOUBLE 0x01000000 211#define TODO_MODE (TODO_MODE_BASE|TODO_SUID|TODO_SGID) 212#define TODO_TIMES ARCHIVE_EXTRACT_TIME 213#define TODO_OWNER ARCHIVE_EXTRACT_OWNER 214#define TODO_FFLAGS ARCHIVE_EXTRACT_FFLAGS 215#define TODO_ACLS ARCHIVE_EXTRACT_ACL 216#define TODO_XATTR ARCHIVE_EXTRACT_XATTR 217#define TODO_MAC_METADATA ARCHIVE_EXTRACT_MAC_METADATA 218#define TODO_HFS_COMPRESSION ARCHIVE_EXTRACT_HFS_COMPRESSION_FORCED 219 220struct archive_write_disk { 221 struct archive archive; 222 223 mode_t user_umask; 224 struct fixup_entry *fixup_list; 225 struct fixup_entry *current_fixup; 226 int64_t user_uid; 227 int skip_file_set; 228 int64_t skip_file_dev; 229 int64_t skip_file_ino; 230 time_t start_time; 231 232 int64_t (*lookup_gid)(void *private, const char *gname, int64_t gid); 233 void (*cleanup_gid)(void *private); 234 void *lookup_gid_data; 235 int64_t (*lookup_uid)(void *private, const char *uname, int64_t uid); 236 void (*cleanup_uid)(void *private); 237 void *lookup_uid_data; 238 239 /* 240 * Full path of last file to satisfy symlink checks. 241 */ 242 struct archive_string path_safe; 243 244 /* 245 * Cached stat data from disk for the current entry. 246 * If this is valid, pst points to st. Otherwise, 247 * pst is null. 248 */ 249 struct stat st; 250 struct stat *pst; 251 252 /* Information about the object being restored right now. */ 253 struct archive_entry *entry; /* Entry being extracted. */ 254 char *name; /* Name of entry, possibly edited. */ 255 struct archive_string _name_data; /* backing store for 'name' */ 256 char *tmpname; /* Temporary name * */ 257 struct archive_string _tmpname_data; /* backing store for 'tmpname' */ 258 /* Tasks remaining for this object. */ 259 int todo; 260 /* Tasks deferred until end-of-archive. */ 261 int deferred; 262 /* Options requested by the client. */ 263 int flags; 264 /* Handle for the file we're restoring. */ 265 int fd; 266 /* Current offset for writing data to the file. */ 267 int64_t offset; 268 /* Last offset actually written to disk. */ 269 int64_t fd_offset; 270 /* Total bytes actually written to files. */ 271 int64_t total_bytes_written; 272 /* Maximum size of file, -1 if unknown. */ 273 int64_t filesize; 274 /* Dir we were in before this restore; only for deep paths. */ 275 int restore_pwd; 276 /* Mode we should use for this entry; affected by _PERM and umask. */ 277 mode_t mode; 278 /* UID/GID to use in restoring this entry. */ 279 int64_t uid; 280 int64_t gid; 281 /* 282 * HFS+ Compression. 283 */ 284 /* Xattr "com.apple.decmpfs". */ 285 uint32_t decmpfs_attr_size; 286 unsigned char *decmpfs_header_p; 287 /* ResourceFork set options used for fsetxattr. */ 288 int rsrc_xattr_options; 289 /* Xattr "com.apple.ResourceFork". */ 290 unsigned char *resource_fork; 291 size_t resource_fork_allocated_size; 292 unsigned int decmpfs_block_count; 293 uint32_t *decmpfs_block_info; 294 /* Buffer for compressed data. */ 295 unsigned char *compressed_buffer; 296 size_t compressed_buffer_size; 297 size_t compressed_buffer_remaining; 298 /* The offset of the ResourceFork where compressed data will 299 * be placed. */ 300 uint32_t compressed_rsrc_position; 301 uint32_t compressed_rsrc_position_v; 302 /* Buffer for uncompressed data. */ 303 char *uncompressed_buffer; 304 size_t block_remaining_bytes; 305 size_t file_remaining_bytes; 306#ifdef HAVE_ZLIB_H 307 z_stream stream; 308 int stream_valid; 309 int decmpfs_compression_level; 310#endif 311}; 312 313/* 314 * Default mode for dirs created automatically (will be modified by umask). 315 * Note that POSIX specifies 0777 for implicitly-created dirs, "modified 316 * by the process' file creation mask." 317 */ 318#define DEFAULT_DIR_MODE 0777 319/* 320 * Dir modes are restored in two steps: During the extraction, the permissions 321 * in the archive are modified to match the following limits. During 322 * the post-extract fixup pass, the permissions from the archive are 323 * applied. 324 */ 325#define MINIMUM_DIR_MODE 0700 326#define MAXIMUM_DIR_MODE 0775 327 328/* 329 * Maximum uncompressed size of a decmpfs block. 330 */ 331#define MAX_DECMPFS_BLOCK_SIZE (64 * 1024) 332/* 333 * HFS+ compression type. 334 */ 335#define CMP_XATTR 3/* Compressed data in xattr. */ 336#define CMP_RESOURCE_FORK 4/* Compressed data in resource fork. */ 337/* 338 * HFS+ compression resource fork. 339 */ 340#define RSRC_H_SIZE 260 /* Base size of Resource fork header. */ 341#define RSRC_F_SIZE 50 /* Size of Resource fork footer. */ 342/* Size to write compressed data to resource fork. */ 343#define COMPRESSED_W_SIZE (64 * 1024) 344/* decmpfs definitions. */ 345#define MAX_DECMPFS_XATTR_SIZE 3802 346#ifndef DECMPFS_XATTR_NAME 347#define DECMPFS_XATTR_NAME "com.apple.decmpfs" 348#endif 349#define DECMPFS_MAGIC 0x636d7066 350#define DECMPFS_COMPRESSION_MAGIC 0 351#define DECMPFS_COMPRESSION_TYPE 4 352#define DECMPFS_UNCOMPRESSED_SIZE 8 353#define DECMPFS_HEADER_SIZE 16 354 355#define HFS_BLOCKS(s) ((s) >> 12) 356 357 358static int la_opendirat(int, const char *); 359static int la_mktemp(struct archive_write_disk *); 360static void fsobj_error(int *, struct archive_string *, int, const char *, 361 const char *); 362static int check_symlinks_fsobj(char *, int *, struct archive_string *, 363 int); 364static int check_symlinks(struct archive_write_disk *); 365static int create_filesystem_object(struct archive_write_disk *); 366static struct fixup_entry *current_fixup(struct archive_write_disk *, 367 const char *pathname); 368#if defined(HAVE_FCHDIR) && defined(PATH_MAX) 369static void edit_deep_directories(struct archive_write_disk *ad); 370#endif 371static int cleanup_pathname_fsobj(char *, int *, struct archive_string *, 372 int); 373static int cleanup_pathname(struct archive_write_disk *); 374static int create_dir(struct archive_write_disk *, char *); 375static int create_parent_dir(struct archive_write_disk *, char *); 376static ssize_t hfs_write_data_block(struct archive_write_disk *, 377 const char *, size_t); 378static int fixup_appledouble(struct archive_write_disk *, const char *); 379static int older(struct stat *, struct archive_entry *); 380static int restore_entry(struct archive_write_disk *); 381static int set_mac_metadata(struct archive_write_disk *, const char *, 382 const void *, size_t); 383static int set_xattrs(struct archive_write_disk *); 384static int clear_nochange_fflags(struct archive_write_disk *); 385static int set_fflags(struct archive_write_disk *); 386static int set_fflags_platform(struct archive_write_disk *, int fd, 387 const char *name, mode_t mode, 388 unsigned long fflags_set, unsigned long fflags_clear); 389static int set_ownership(struct archive_write_disk *); 390static int set_mode(struct archive_write_disk *, int mode); 391static int set_time(int, int, const char *, time_t, long, time_t, long); 392static int set_times(struct archive_write_disk *, int, int, const char *, 393 time_t, long, time_t, long, time_t, long, time_t, long); 394static int set_times_from_entry(struct archive_write_disk *); 395static struct fixup_entry *sort_dir_list(struct fixup_entry *p); 396static ssize_t write_data_block(struct archive_write_disk *, 397 const char *, size_t); 398 399static struct archive_vtable *archive_write_disk_vtable(void); 400 401static int _archive_write_disk_close(struct archive *); 402static int _archive_write_disk_free(struct archive *); 403static int _archive_write_disk_header(struct archive *, 404 struct archive_entry *); 405static int64_t _archive_write_disk_filter_bytes(struct archive *, int); 406static int _archive_write_disk_finish_entry(struct archive *); 407static ssize_t _archive_write_disk_data(struct archive *, const void *, 408 size_t); 409static ssize_t _archive_write_disk_data_block(struct archive *, const void *, 410 size_t, int64_t); 411 412static int 413la_mktemp(struct archive_write_disk *a) 414{ 415 int oerrno, fd; 416 mode_t mode; 417 418 archive_string_empty(&a->_tmpname_data); 419 archive_string_sprintf(&a->_tmpname_data, "%s.XXXXXX", a->name); 420 a->tmpname = a->_tmpname_data.s; 421 422 fd = __archive_mkstemp(a->tmpname); 423 if (fd == -1) 424 return -1; 425 426 mode = a->mode & 0777 & ~a->user_umask; 427 if (fchmod(fd, mode) == -1) { 428 oerrno = errno; 429 close(fd); 430 errno = oerrno; 431 return -1; 432 } 433 return fd; 434} 435 436static int 437la_opendirat(int fd, const char *path) { 438 const int flags = O_CLOEXEC 439#if defined(O_BINARY) 440 | O_BINARY 441#endif 442#if defined(O_DIRECTORY) 443 | O_DIRECTORY 444#endif 445#if defined(O_PATH) 446 | O_PATH 447#elif defined(O_SEARCH) 448 | O_SEARCH 449#elif defined(__FreeBSD__) && defined(O_EXEC) 450 | O_EXEC 451#else 452 | O_RDONLY 453#endif 454 ; 455 456#if !defined(HAVE_OPENAT) 457 if (fd != AT_FDCWD) { 458 errno = ENOTSUP; 459 return (-1); 460 } else 461 return (open(path, flags)); 462#else 463 return (openat(fd, path, flags)); 464#endif 465} 466 467static int 468lazy_stat(struct archive_write_disk *a) 469{ 470 if (a->pst != NULL) { 471 /* Already have stat() data available. */ 472 return (ARCHIVE_OK); 473 } 474#ifdef HAVE_FSTAT 475 if (a->fd >= 0 && fstat(a->fd, &a->st) == 0) { 476 a->pst = &a->st; 477 return (ARCHIVE_OK); 478 } 479#endif 480 /* 481 * XXX At this point, symlinks should not be hit, otherwise 482 * XXX a race occurred. Do we want to check explicitly for that? 483 */ 484 if (lstat(a->name, &a->st) == 0) { 485 a->pst = &a->st; 486 return (ARCHIVE_OK); 487 } 488 archive_set_error(&a->archive, errno, "Couldn't stat file"); 489 return (ARCHIVE_WARN); 490} 491 492static struct archive_vtable * 493archive_write_disk_vtable(void) 494{ 495 static struct archive_vtable av; 496 static int inited = 0; 497 498 if (!inited) { 499 av.archive_close = _archive_write_disk_close; 500 av.archive_filter_bytes = _archive_write_disk_filter_bytes; 501 av.archive_free = _archive_write_disk_free; 502 av.archive_write_header = _archive_write_disk_header; 503 av.archive_write_finish_entry 504 = _archive_write_disk_finish_entry; 505 av.archive_write_data = _archive_write_disk_data; 506 av.archive_write_data_block = _archive_write_disk_data_block; 507 inited = 1; 508 } 509 return (&av); 510} 511 512static int64_t 513_archive_write_disk_filter_bytes(struct archive *_a, int n) 514{ 515 struct archive_write_disk *a = (struct archive_write_disk *)_a; 516 (void)n; /* UNUSED */ 517 if (n == -1 || n == 0) 518 return (a->total_bytes_written); 519 return (-1); 520} 521 522 523int 524archive_write_disk_set_options(struct archive *_a, int flags) 525{ 526 struct archive_write_disk *a = (struct archive_write_disk *)_a; 527 528 a->flags = flags; 529 return (ARCHIVE_OK); 530} 531 532 533/* 534 * Extract this entry to disk. 535 * 536 * TODO: Validate hardlinks. According to the standards, we're 537 * supposed to check each extracted hardlink and squawk if it refers 538 * to a file that we didn't restore. I'm not entirely convinced this 539 * is a good idea, but more importantly: Is there any way to validate 540 * hardlinks without keeping a complete list of filenames from the 541 * entire archive?? Ugh. 542 * 543 */ 544static int 545_archive_write_disk_header(struct archive *_a, struct archive_entry *entry) 546{ 547 struct archive_write_disk *a = (struct archive_write_disk *)_a; 548 struct fixup_entry *fe; 549 int ret, r; 550 551 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 552 ARCHIVE_STATE_HEADER | ARCHIVE_STATE_DATA, 553 "archive_write_disk_header"); 554 archive_clear_error(&a->archive); 555 if (a->archive.state & ARCHIVE_STATE_DATA) { 556 r = _archive_write_disk_finish_entry(&a->archive); 557 if (r == ARCHIVE_FATAL) 558 return (r); 559 } 560 561 /* Set up for this particular entry. */ 562 a->pst = NULL; 563 a->current_fixup = NULL; 564 a->deferred = 0; 565 if (a->entry) { 566 archive_entry_free(a->entry); 567 a->entry = NULL; 568 } 569 a->entry = archive_entry_clone(entry); 570 a->fd = -1; 571 a->fd_offset = 0; 572 a->offset = 0; 573 a->restore_pwd = -1; 574 a->uid = a->user_uid; 575 a->mode = archive_entry_mode(a->entry); 576 if (archive_entry_size_is_set(a->entry)) 577 a->filesize = archive_entry_size(a->entry); 578 else 579 a->filesize = -1; 580 archive_strcpy(&(a->_name_data), archive_entry_pathname(a->entry)); 581 a->name = a->_name_data.s; 582 archive_clear_error(&a->archive); 583 584 /* 585 * Clean up the requested path. This is necessary for correct 586 * dir restores; the dir restore logic otherwise gets messed 587 * up by nonsense like "dir/.". 588 */ 589 ret = cleanup_pathname(a); 590 if (ret != ARCHIVE_OK) 591 return (ret); 592 593 /* 594 * Query the umask so we get predictable mode settings. 595 * This gets done on every call to _write_header in case the 596 * user edits their umask during the extraction for some 597 * reason. 598 */ 599 umask(a->user_umask = umask(0)); 600 601 /* Figure out what we need to do for this entry. */ 602 a->todo = TODO_MODE_BASE; 603 if (a->flags & ARCHIVE_EXTRACT_PERM) { 604 a->todo |= TODO_MODE_FORCE; /* Be pushy about permissions. */ 605 /* 606 * SGID requires an extra "check" step because we 607 * cannot easily predict the GID that the system will 608 * assign. (Different systems assign GIDs to files 609 * based on a variety of criteria, including process 610 * credentials and the gid of the enclosing 611 * directory.) We can only restore the SGID bit if 612 * the file has the right GID, and we only know the 613 * GID if we either set it (see set_ownership) or if 614 * we've actually called stat() on the file after it 615 * was restored. Since there are several places at 616 * which we might verify the GID, we need a TODO bit 617 * to keep track. 618 */ 619 if (a->mode & S_ISGID) 620 a->todo |= TODO_SGID | TODO_SGID_CHECK; 621 /* 622 * Verifying the SUID is simpler, but can still be 623 * done in multiple ways, hence the separate "check" bit. 624 */ 625 if (a->mode & S_ISUID) 626 a->todo |= TODO_SUID | TODO_SUID_CHECK; 627 } else { 628 /* 629 * User didn't request full permissions, so don't 630 * restore SUID, SGID bits and obey umask. 631 */ 632 a->mode &= ~S_ISUID; 633 a->mode &= ~S_ISGID; 634 a->mode &= ~S_ISVTX; 635 a->mode &= ~a->user_umask; 636 } 637 if (a->flags & ARCHIVE_EXTRACT_OWNER) 638 a->todo |= TODO_OWNER; 639 if (a->flags & ARCHIVE_EXTRACT_TIME) 640 a->todo |= TODO_TIMES; 641 if (a->flags & ARCHIVE_EXTRACT_ACL) { 642#if ARCHIVE_ACL_DARWIN 643 /* 644 * On MacOS, platform ACLs get stored in mac_metadata, too. 645 * If we intend to extract mac_metadata and it is present 646 * we skip extracting libarchive NFSv4 ACLs. 647 */ 648 size_t metadata_size; 649 650 if ((a->flags & ARCHIVE_EXTRACT_MAC_METADATA) == 0 || 651 archive_entry_mac_metadata(a->entry, 652 &metadata_size) == NULL || metadata_size == 0) 653#endif 654#if ARCHIVE_ACL_LIBRICHACL 655 /* 656 * RichACLs are stored in an extended attribute. 657 * If we intend to extract extended attributes and have this 658 * attribute we skip extracting libarchive NFSv4 ACLs. 659 */ 660 short extract_acls = 1; 661 if (a->flags & ARCHIVE_EXTRACT_XATTR && ( 662 archive_entry_acl_types(a->entry) & 663 ARCHIVE_ENTRY_ACL_TYPE_NFS4)) { 664 const char *attr_name; 665 const void *attr_value; 666 size_t attr_size; 667 int i = archive_entry_xattr_reset(a->entry); 668 while (i--) { 669 archive_entry_xattr_next(a->entry, &attr_name, 670 &attr_value, &attr_size); 671 if (attr_name != NULL && attr_value != NULL && 672 attr_size > 0 && strcmp(attr_name, 673 "trusted.richacl") == 0) { 674 extract_acls = 0; 675 break; 676 } 677 } 678 } 679 if (extract_acls) 680#endif 681#if ARCHIVE_ACL_DARWIN || ARCHIVE_ACL_LIBRICHACL 682 { 683#endif 684 if (archive_entry_filetype(a->entry) == AE_IFDIR) 685 a->deferred |= TODO_ACLS; 686 else 687 a->todo |= TODO_ACLS; 688#if ARCHIVE_ACL_DARWIN || ARCHIVE_ACL_LIBRICHACL 689 } 690#endif 691 } 692 if (a->flags & ARCHIVE_EXTRACT_MAC_METADATA) { 693 if (archive_entry_filetype(a->entry) == AE_IFDIR) 694 a->deferred |= TODO_MAC_METADATA; 695 else 696 a->todo |= TODO_MAC_METADATA; 697 } 698#if defined(__APPLE__) && defined(UF_COMPRESSED) && defined(HAVE_ZLIB_H) 699 if ((a->flags & ARCHIVE_EXTRACT_NO_HFS_COMPRESSION) == 0) { 700 unsigned long set, clear; 701 archive_entry_fflags(a->entry, &set, &clear); 702 if ((set & ~clear) & UF_COMPRESSED) { 703 a->todo |= TODO_HFS_COMPRESSION; 704 a->decmpfs_block_count = (unsigned)-1; 705 } 706 } 707 if ((a->flags & ARCHIVE_EXTRACT_HFS_COMPRESSION_FORCED) != 0 && 708 (a->mode & AE_IFMT) == AE_IFREG && a->filesize > 0) { 709 a->todo |= TODO_HFS_COMPRESSION; 710 a->decmpfs_block_count = (unsigned)-1; 711 } 712 { 713 const char *p; 714 715 /* Check if the current file name is a type of the 716 * resource fork file. */ 717 p = strrchr(a->name, '/'); 718 if (p == NULL) 719 p = a->name; 720 else 721 p++; 722 if (p[0] == '.' && p[1] == '_') { 723 /* Do not compress "._XXX" files. */ 724 a->todo &= ~TODO_HFS_COMPRESSION; 725 if (a->filesize > 0) 726 a->todo |= TODO_APPLEDOUBLE; 727 } 728 } 729#endif 730 731 if (a->flags & ARCHIVE_EXTRACT_XATTR) { 732#if ARCHIVE_XATTR_DARWIN 733 /* 734 * On MacOS, extended attributes get stored in mac_metadata, 735 * too. If we intend to extract mac_metadata and it is present 736 * we skip extracting extended attributes. 737 */ 738 size_t metadata_size; 739 740 if ((a->flags & ARCHIVE_EXTRACT_MAC_METADATA) == 0 || 741 archive_entry_mac_metadata(a->entry, 742 &metadata_size) == NULL || metadata_size == 0) 743#endif 744 a->todo |= TODO_XATTR; 745 } 746 if (a->flags & ARCHIVE_EXTRACT_FFLAGS) 747 a->todo |= TODO_FFLAGS; 748 if (a->flags & ARCHIVE_EXTRACT_SECURE_SYMLINKS) { 749 ret = check_symlinks(a); 750 if (ret != ARCHIVE_OK) 751 return (ret); 752 } 753#if defined(HAVE_FCHDIR) && defined(PATH_MAX) 754 /* If path exceeds PATH_MAX, shorten the path. */ 755 edit_deep_directories(a); 756#endif 757 758 ret = restore_entry(a); 759 760#if defined(__APPLE__) && defined(UF_COMPRESSED) && defined(HAVE_ZLIB_H) 761 /* 762 * Check if the filesystem the file is restoring on supports 763 * HFS+ Compression. If not, cancel HFS+ Compression. 764 */ 765 if (a->todo | TODO_HFS_COMPRESSION) { 766 /* 767 * NOTE: UF_COMPRESSED is ignored even if the filesystem 768 * supports HFS+ Compression because the file should 769 * have at least an extended attribute "com.apple.decmpfs" 770 * before the flag is set to indicate that the file have 771 * been compressed. If the filesystem does not support 772 * HFS+ Compression the system call will fail. 773 */ 774 if (a->fd < 0 || fchflags(a->fd, UF_COMPRESSED) != 0) 775 a->todo &= ~TODO_HFS_COMPRESSION; 776 } 777#endif 778 779 /* 780 * TODO: There are rumours that some extended attributes must 781 * be restored before file data is written. If this is true, 782 * then we either need to write all extended attributes both 783 * before and after restoring the data, or find some rule for 784 * determining which must go first and which last. Due to the 785 * many ways people are using xattrs, this may prove to be an 786 * intractable problem. 787 */ 788 789#ifdef HAVE_FCHDIR 790 /* If we changed directory above, restore it here. */ 791 if (a->restore_pwd >= 0) { 792 r = fchdir(a->restore_pwd); 793 if (r != 0) { 794 archive_set_error(&a->archive, errno, 795 "chdir() failure"); 796 ret = ARCHIVE_FATAL; 797 } 798 close(a->restore_pwd); 799 a->restore_pwd = -1; 800 } 801#endif 802 803 /* 804 * Fixup uses the unedited pathname from archive_entry_pathname(), 805 * because it is relative to the base dir and the edited path 806 * might be relative to some intermediate dir as a result of the 807 * deep restore logic. 808 */ 809 if (a->deferred & TODO_MODE) { 810 fe = current_fixup(a, archive_entry_pathname(entry)); 811 if (fe == NULL) 812 return (ARCHIVE_FATAL); 813 fe->fixup |= TODO_MODE_BASE; 814 fe->mode = a->mode; 815 } 816 817 if ((a->deferred & TODO_TIMES) 818 && (archive_entry_mtime_is_set(entry) 819 || archive_entry_atime_is_set(entry))) { 820 fe = current_fixup(a, archive_entry_pathname(entry)); 821 if (fe == NULL) 822 return (ARCHIVE_FATAL); 823 fe->mode = a->mode; 824 fe->fixup |= TODO_TIMES; 825 if (archive_entry_atime_is_set(entry)) { 826 fe->atime = archive_entry_atime(entry); 827 fe->atime_nanos = archive_entry_atime_nsec(entry); 828 } else { 829 /* If atime is unset, use start time. */ 830 fe->atime = a->start_time; 831 fe->atime_nanos = 0; 832 } 833 if (archive_entry_mtime_is_set(entry)) { 834 fe->mtime = archive_entry_mtime(entry); 835 fe->mtime_nanos = archive_entry_mtime_nsec(entry); 836 } else { 837 /* If mtime is unset, use start time. */ 838 fe->mtime = a->start_time; 839 fe->mtime_nanos = 0; 840 } 841 if (archive_entry_birthtime_is_set(entry)) { 842 fe->birthtime = archive_entry_birthtime(entry); 843 fe->birthtime_nanos = archive_entry_birthtime_nsec( 844 entry); 845 } else { 846 /* If birthtime is unset, use mtime. */ 847 fe->birthtime = fe->mtime; 848 fe->birthtime_nanos = fe->mtime_nanos; 849 } 850 } 851 852 if (a->deferred & TODO_ACLS) { 853 fe = current_fixup(a, archive_entry_pathname(entry)); 854 if (fe == NULL) 855 return (ARCHIVE_FATAL); 856 fe->fixup |= TODO_ACLS; 857 archive_acl_copy(&fe->acl, archive_entry_acl(entry)); 858 } 859 860 if (a->deferred & TODO_MAC_METADATA) { 861 const void *metadata; 862 size_t metadata_size; 863 metadata = archive_entry_mac_metadata(a->entry, &metadata_size); 864 if (metadata != NULL && metadata_size > 0) { 865 fe = current_fixup(a, archive_entry_pathname(entry)); 866 if (fe == NULL) 867 return (ARCHIVE_FATAL); 868 fe->mac_metadata = malloc(metadata_size); 869 if (fe->mac_metadata != NULL) { 870 memcpy(fe->mac_metadata, metadata, 871 metadata_size); 872 fe->mac_metadata_size = metadata_size; 873 fe->fixup |= TODO_MAC_METADATA; 874 } 875 } 876 } 877 878 if (a->deferred & TODO_FFLAGS) { 879 fe = current_fixup(a, archive_entry_pathname(entry)); 880 if (fe == NULL) 881 return (ARCHIVE_FATAL); 882 fe->fixup |= TODO_FFLAGS; 883 /* TODO: Complete this.. defer fflags from below. */ 884 } 885 886 /* We've created the object and are ready to pour data into it. */ 887 if (ret >= ARCHIVE_WARN) 888 a->archive.state = ARCHIVE_STATE_DATA; 889 /* 890 * If it's not open, tell our client not to try writing. 891 * In particular, dirs, links, etc, don't get written to. 892 */ 893 if (a->fd < 0) { 894 archive_entry_set_size(entry, 0); 895 a->filesize = 0; 896 } 897 898 return (ret); 899} 900 901int 902archive_write_disk_set_skip_file(struct archive *_a, la_int64_t d, la_int64_t i) 903{ 904 struct archive_write_disk *a = (struct archive_write_disk *)_a; 905 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 906 ARCHIVE_STATE_ANY, "archive_write_disk_set_skip_file"); 907 a->skip_file_set = 1; 908 a->skip_file_dev = d; 909 a->skip_file_ino = i; 910 return (ARCHIVE_OK); 911} 912 913static ssize_t 914write_data_block(struct archive_write_disk *a, const char *buff, size_t size) 915{ 916 uint64_t start_size = size; 917 ssize_t bytes_written = 0; 918 ssize_t block_size = 0, bytes_to_write; 919 920 if (size == 0) 921 return (ARCHIVE_OK); 922 923 if (a->filesize == 0 || a->fd < 0) { 924 archive_set_error(&a->archive, 0, 925 "Attempt to write to an empty file"); 926 return (ARCHIVE_WARN); 927 } 928 929 if (a->flags & ARCHIVE_EXTRACT_SPARSE) { 930#if HAVE_STRUCT_STAT_ST_BLKSIZE 931 int r; 932 if ((r = lazy_stat(a)) != ARCHIVE_OK) 933 return (r); 934 block_size = a->pst->st_blksize; 935#else 936 /* XXX TODO XXX Is there a more appropriate choice here ? */ 937 /* This needn't match the filesystem allocation size. */ 938 block_size = 16*1024; 939#endif 940 } 941 942 /* If this write would run beyond the file size, truncate it. */ 943 if (a->filesize >= 0 && (int64_t)(a->offset + size) > a->filesize) 944 start_size = size = (size_t)(a->filesize - a->offset); 945 946 /* Write the data. */ 947 while (size > 0) { 948 if (block_size == 0) { 949 bytes_to_write = size; 950 } else { 951 /* We're sparsifying the file. */ 952 const char *p, *end; 953 int64_t block_end; 954 955 /* Skip leading zero bytes. */ 956 for (p = buff, end = buff + size; p < end; ++p) { 957 if (*p != '\0') 958 break; 959 } 960 a->offset += p - buff; 961 size -= p - buff; 962 buff = p; 963 if (size == 0) 964 break; 965 966 /* Calculate next block boundary after offset. */ 967 block_end 968 = (a->offset / block_size + 1) * block_size; 969 970 /* If the adjusted write would cross block boundary, 971 * truncate it to the block boundary. */ 972 bytes_to_write = size; 973 if (a->offset + bytes_to_write > block_end) 974 bytes_to_write = block_end - a->offset; 975 } 976 /* Seek if necessary to the specified offset. */ 977 if (a->offset != a->fd_offset) { 978 if (lseek(a->fd, a->offset, SEEK_SET) < 0) { 979 archive_set_error(&a->archive, errno, 980 "Seek failed"); 981 return (ARCHIVE_FATAL); 982 } 983 a->fd_offset = a->offset; 984 } 985 bytes_written = write(a->fd, buff, bytes_to_write); 986 if (bytes_written < 0) { 987 archive_set_error(&a->archive, errno, "Write failed"); 988 return (ARCHIVE_WARN); 989 } 990 buff += bytes_written; 991 size -= bytes_written; 992 a->total_bytes_written += bytes_written; 993 a->offset += bytes_written; 994 a->fd_offset = a->offset; 995 } 996 return (start_size - size); 997} 998 999#if defined(__APPLE__) && defined(UF_COMPRESSED) && defined(HAVE_SYS_XATTR_H)\ 1000 && defined(HAVE_ZLIB_H) 1001 1002/* 1003 * Set UF_COMPRESSED file flag. 1004 * This have to be called after hfs_write_decmpfs() because if the 1005 * file does not have "com.apple.decmpfs" xattr the flag is ignored. 1006 */ 1007static int 1008hfs_set_compressed_fflag(struct archive_write_disk *a) 1009{ 1010 int r; 1011 1012 if ((r = lazy_stat(a)) != ARCHIVE_OK) 1013 return (r); 1014 1015 a->st.st_flags |= UF_COMPRESSED; 1016 if (fchflags(a->fd, a->st.st_flags) != 0) { 1017 archive_set_error(&a->archive, errno, 1018 "Failed to set UF_COMPRESSED file flag"); 1019 return (ARCHIVE_WARN); 1020 } 1021 return (ARCHIVE_OK); 1022} 1023 1024/* 1025 * HFS+ Compression decmpfs 1026 * 1027 * +------------------------------+ +0 1028 * | Magic(LE 4 bytes) | 1029 * +------------------------------+ 1030 * | Type(LE 4 bytes) | 1031 * +------------------------------+ 1032 * | Uncompressed size(LE 8 bytes)| 1033 * +------------------------------+ +16 1034 * | | 1035 * | Compressed data | 1036 * | (Placed only if Type == 3) | 1037 * | | 1038 * +------------------------------+ +3802 = MAX_DECMPFS_XATTR_SIZE 1039 * 1040 * Type is 3: decmpfs has compressed data. 1041 * Type is 4: Resource Fork has compressed data. 1042 */ 1043/* 1044 * Write "com.apple.decmpfs" 1045 */ 1046static int 1047hfs_write_decmpfs(struct archive_write_disk *a) 1048{ 1049 int r; 1050 uint32_t compression_type; 1051 1052 r = fsetxattr(a->fd, DECMPFS_XATTR_NAME, a->decmpfs_header_p, 1053 a->decmpfs_attr_size, 0, 0); 1054 if (r < 0) { 1055 archive_set_error(&a->archive, errno, 1056 "Cannot restore xattr:%s", DECMPFS_XATTR_NAME); 1057 compression_type = archive_le32dec( 1058 &a->decmpfs_header_p[DECMPFS_COMPRESSION_TYPE]); 1059 if (compression_type == CMP_RESOURCE_FORK) 1060 fremovexattr(a->fd, XATTR_RESOURCEFORK_NAME, 1061 XATTR_SHOWCOMPRESSION); 1062 return (ARCHIVE_WARN); 1063 } 1064 return (ARCHIVE_OK); 1065} 1066 1067/* 1068 * HFS+ Compression Resource Fork 1069 * 1070 * +-----------------------------+ 1071 * | Header(260 bytes) | 1072 * +-----------------------------+ 1073 * | Block count(LE 4 bytes) | 1074 * +-----------------------------+ --+ 1075 * +-- | Offset (LE 4 bytes) | | 1076 * | | [distance from Block count] | | Block 0 1077 * | +-----------------------------+ | 1078 * | | Compressed size(LE 4 bytes) | | 1079 * | +-----------------------------+ --+ 1080 * | | | 1081 * | | .................. | 1082 * | | | 1083 * | +-----------------------------+ --+ 1084 * | | Offset (LE 4 bytes) | | 1085 * | +-----------------------------+ | Block (Block count -1) 1086 * | | Compressed size(LE 4 bytes) | | 1087 * +-> +-----------------------------+ --+ 1088 * | Compressed data(n bytes) | Block 0 1089 * +-----------------------------+ 1090 * | | 1091 * | .................. | 1092 * | | 1093 * +-----------------------------+ 1094 * | Compressed data(n bytes) | Block (Block count -1) 1095 * +-----------------------------+ 1096 * | Footer(50 bytes) | 1097 * +-----------------------------+ 1098 * 1099 */ 1100/* 1101 * Write the header of "com.apple.ResourceFork" 1102 */ 1103static int 1104hfs_write_resource_fork(struct archive_write_disk *a, unsigned char *buff, 1105 size_t bytes, uint32_t position) 1106{ 1107 int ret; 1108 1109 ret = fsetxattr(a->fd, XATTR_RESOURCEFORK_NAME, buff, bytes, 1110 position, a->rsrc_xattr_options); 1111 if (ret < 0) { 1112 archive_set_error(&a->archive, errno, 1113 "Cannot restore xattr: %s at %u pos %u bytes", 1114 XATTR_RESOURCEFORK_NAME, 1115 (unsigned)position, 1116 (unsigned)bytes); 1117 return (ARCHIVE_WARN); 1118 } 1119 a->rsrc_xattr_options &= ~XATTR_CREATE; 1120 return (ARCHIVE_OK); 1121} 1122 1123static int 1124hfs_write_compressed_data(struct archive_write_disk *a, size_t bytes_compressed) 1125{ 1126 int ret; 1127 1128 ret = hfs_write_resource_fork(a, a->compressed_buffer, 1129 bytes_compressed, a->compressed_rsrc_position); 1130 if (ret == ARCHIVE_OK) 1131 a->compressed_rsrc_position += bytes_compressed; 1132 return (ret); 1133} 1134 1135static int 1136hfs_write_resource_fork_header(struct archive_write_disk *a) 1137{ 1138 unsigned char *buff; 1139 uint32_t rsrc_bytes; 1140 uint32_t rsrc_header_bytes; 1141 1142 /* 1143 * Write resource fork header + block info. 1144 */ 1145 buff = a->resource_fork; 1146 rsrc_bytes = a->compressed_rsrc_position - RSRC_F_SIZE; 1147 rsrc_header_bytes = 1148 RSRC_H_SIZE + /* Header base size. */ 1149 4 + /* Block count. */ 1150 (a->decmpfs_block_count * 8);/* Block info */ 1151 archive_be32enc(buff, 0x100); 1152 archive_be32enc(buff + 4, rsrc_bytes); 1153 archive_be32enc(buff + 8, rsrc_bytes - 256); 1154 archive_be32enc(buff + 12, 0x32); 1155 memset(buff + 16, 0, 240); 1156 archive_be32enc(buff + 256, rsrc_bytes - 260); 1157 return hfs_write_resource_fork(a, buff, rsrc_header_bytes, 0); 1158} 1159 1160static size_t 1161hfs_set_resource_fork_footer(unsigned char *buff, size_t buff_size) 1162{ 1163 static const char rsrc_footer[RSRC_F_SIZE] = { 1164 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 1165 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 1166 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 1167 0x00, 0x1c, 0x00, 0x32, 0x00, 0x00, 'c', 'm', 1168 'p', 'f', 0x00, 0x00, 0x00, 0x0a, 0x00, 0x01, 1169 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 1170 0x00, 0x00 1171 }; 1172 if (buff_size < sizeof(rsrc_footer)) 1173 return (0); 1174 memcpy(buff, rsrc_footer, sizeof(rsrc_footer)); 1175 return (sizeof(rsrc_footer)); 1176} 1177 1178static int 1179hfs_reset_compressor(struct archive_write_disk *a) 1180{ 1181 int ret; 1182 1183 if (a->stream_valid) 1184 ret = deflateReset(&a->stream); 1185 else 1186 ret = deflateInit(&a->stream, a->decmpfs_compression_level); 1187 1188 if (ret != Z_OK) { 1189 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 1190 "Failed to initialize compressor"); 1191 return (ARCHIVE_FATAL); 1192 } else 1193 a->stream_valid = 1; 1194 1195 return (ARCHIVE_OK); 1196} 1197 1198static int 1199hfs_decompress(struct archive_write_disk *a) 1200{ 1201 uint32_t *block_info; 1202 unsigned int block_count; 1203 uint32_t data_pos, data_size; 1204 ssize_t r; 1205 ssize_t bytes_written, bytes_to_write; 1206 unsigned char *b; 1207 1208 block_info = (uint32_t *)(a->resource_fork + RSRC_H_SIZE); 1209 block_count = archive_le32dec(block_info++); 1210 while (block_count--) { 1211 data_pos = RSRC_H_SIZE + archive_le32dec(block_info++); 1212 data_size = archive_le32dec(block_info++); 1213 r = fgetxattr(a->fd, XATTR_RESOURCEFORK_NAME, 1214 a->compressed_buffer, data_size, data_pos, 0); 1215 if (r != data_size) { 1216 archive_set_error(&a->archive, 1217 (r < 0)?errno:ARCHIVE_ERRNO_MISC, 1218 "Failed to read resource fork"); 1219 return (ARCHIVE_WARN); 1220 } 1221 if (a->compressed_buffer[0] == 0xff) { 1222 bytes_to_write = data_size -1; 1223 b = a->compressed_buffer + 1; 1224 } else { 1225 uLong dest_len = MAX_DECMPFS_BLOCK_SIZE; 1226 int zr; 1227 1228 zr = uncompress((Bytef *)a->uncompressed_buffer, 1229 &dest_len, a->compressed_buffer, data_size); 1230 if (zr != Z_OK) { 1231 archive_set_error(&a->archive, 1232 ARCHIVE_ERRNO_MISC, 1233 "Failed to decompress resource fork"); 1234 return (ARCHIVE_WARN); 1235 } 1236 bytes_to_write = dest_len; 1237 b = (unsigned char *)a->uncompressed_buffer; 1238 } 1239 do { 1240 bytes_written = write(a->fd, b, bytes_to_write); 1241 if (bytes_written < 0) { 1242 archive_set_error(&a->archive, errno, 1243 "Write failed"); 1244 return (ARCHIVE_WARN); 1245 } 1246 bytes_to_write -= bytes_written; 1247 b += bytes_written; 1248 } while (bytes_to_write > 0); 1249 } 1250 r = fremovexattr(a->fd, XATTR_RESOURCEFORK_NAME, 0); 1251 if (r == -1) { 1252 archive_set_error(&a->archive, errno, 1253 "Failed to remove resource fork"); 1254 return (ARCHIVE_WARN); 1255 } 1256 return (ARCHIVE_OK); 1257} 1258 1259static int 1260hfs_drive_compressor(struct archive_write_disk *a, const char *buff, 1261 size_t size) 1262{ 1263 unsigned char *buffer_compressed; 1264 size_t bytes_compressed; 1265 size_t bytes_used; 1266 int ret; 1267 1268 ret = hfs_reset_compressor(a); 1269 if (ret != ARCHIVE_OK) 1270 return (ret); 1271 1272 if (a->compressed_buffer == NULL) { 1273 size_t block_size; 1274 1275 block_size = COMPRESSED_W_SIZE + RSRC_F_SIZE + 1276 + compressBound(MAX_DECMPFS_BLOCK_SIZE); 1277 a->compressed_buffer = malloc(block_size); 1278 if (a->compressed_buffer == NULL) { 1279 archive_set_error(&a->archive, ENOMEM, 1280 "Can't allocate memory for Resource Fork"); 1281 return (ARCHIVE_FATAL); 1282 } 1283 a->compressed_buffer_size = block_size; 1284 a->compressed_buffer_remaining = block_size; 1285 } 1286 1287 buffer_compressed = a->compressed_buffer + 1288 a->compressed_buffer_size - a->compressed_buffer_remaining; 1289 a->stream.next_in = (Bytef *)(uintptr_t)(const void *)buff; 1290 a->stream.avail_in = size; 1291 a->stream.next_out = buffer_compressed; 1292 a->stream.avail_out = a->compressed_buffer_remaining; 1293 do { 1294 ret = deflate(&a->stream, Z_FINISH); 1295 switch (ret) { 1296 case Z_OK: 1297 case Z_STREAM_END: 1298 break; 1299 default: 1300 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 1301 "Failed to compress data"); 1302 return (ARCHIVE_FAILED); 1303 } 1304 } while (ret == Z_OK); 1305 bytes_compressed = a->compressed_buffer_remaining - a->stream.avail_out; 1306 1307 /* 1308 * If the compressed size is larger than the original size, 1309 * throw away compressed data, use uncompressed data instead. 1310 */ 1311 if (bytes_compressed > size) { 1312 buffer_compressed[0] = 0xFF;/* uncompressed marker. */ 1313 memcpy(buffer_compressed + 1, buff, size); 1314 bytes_compressed = size + 1; 1315 } 1316 a->compressed_buffer_remaining -= bytes_compressed; 1317 1318 /* 1319 * If the compressed size is smaller than MAX_DECMPFS_XATTR_SIZE 1320 * and the block count in the file is only one, store compressed 1321 * data to decmpfs xattr instead of the resource fork. 1322 */ 1323 if (a->decmpfs_block_count == 1 && 1324 (a->decmpfs_attr_size + bytes_compressed) 1325 <= MAX_DECMPFS_XATTR_SIZE) { 1326 archive_le32enc(&a->decmpfs_header_p[DECMPFS_COMPRESSION_TYPE], 1327 CMP_XATTR); 1328 memcpy(a->decmpfs_header_p + DECMPFS_HEADER_SIZE, 1329 buffer_compressed, bytes_compressed); 1330 a->decmpfs_attr_size += bytes_compressed; 1331 a->compressed_buffer_remaining = a->compressed_buffer_size; 1332 /* 1333 * Finish HFS+ Compression. 1334 * - Write the decmpfs xattr. 1335 * - Set the UF_COMPRESSED file flag. 1336 */ 1337 ret = hfs_write_decmpfs(a); 1338 if (ret == ARCHIVE_OK) 1339 ret = hfs_set_compressed_fflag(a); 1340 return (ret); 1341 } 1342 1343 /* Update block info. */ 1344 archive_le32enc(a->decmpfs_block_info++, 1345 a->compressed_rsrc_position_v - RSRC_H_SIZE); 1346 archive_le32enc(a->decmpfs_block_info++, bytes_compressed); 1347 a->compressed_rsrc_position_v += bytes_compressed; 1348 1349 /* 1350 * Write the compressed data to the resource fork. 1351 */ 1352 bytes_used = a->compressed_buffer_size - a->compressed_buffer_remaining; 1353 while (bytes_used >= COMPRESSED_W_SIZE) { 1354 ret = hfs_write_compressed_data(a, COMPRESSED_W_SIZE); 1355 if (ret != ARCHIVE_OK) 1356 return (ret); 1357 bytes_used -= COMPRESSED_W_SIZE; 1358 if (bytes_used > COMPRESSED_W_SIZE) 1359 memmove(a->compressed_buffer, 1360 a->compressed_buffer + COMPRESSED_W_SIZE, 1361 bytes_used); 1362 else 1363 memcpy(a->compressed_buffer, 1364 a->compressed_buffer + COMPRESSED_W_SIZE, 1365 bytes_used); 1366 } 1367 a->compressed_buffer_remaining = a->compressed_buffer_size - bytes_used; 1368 1369 /* 1370 * If the current block is the last block, write the remaining 1371 * compressed data and the resource fork footer. 1372 */ 1373 if (a->file_remaining_bytes == 0) { 1374 size_t rsrc_size; 1375 int64_t bk; 1376 1377 /* Append the resource footer. */ 1378 rsrc_size = hfs_set_resource_fork_footer( 1379 a->compressed_buffer + bytes_used, 1380 a->compressed_buffer_remaining); 1381 ret = hfs_write_compressed_data(a, bytes_used + rsrc_size); 1382 a->compressed_buffer_remaining = a->compressed_buffer_size; 1383 1384 /* If the compressed size is not enough smaller than 1385 * the uncompressed size. cancel HFS+ compression. 1386 * TODO: study a behavior of ditto utility and improve 1387 * the condition to fall back into no HFS+ compression. */ 1388 bk = HFS_BLOCKS(a->compressed_rsrc_position); 1389 bk += bk >> 7; 1390 if (bk > HFS_BLOCKS(a->filesize)) 1391 return hfs_decompress(a); 1392 /* 1393 * Write the resourcefork header. 1394 */ 1395 if (ret == ARCHIVE_OK) 1396 ret = hfs_write_resource_fork_header(a); 1397 /* 1398 * Finish HFS+ Compression. 1399 * - Write the decmpfs xattr. 1400 * - Set the UF_COMPRESSED file flag. 1401 */ 1402 if (ret == ARCHIVE_OK) 1403 ret = hfs_write_decmpfs(a); 1404 if (ret == ARCHIVE_OK) 1405 ret = hfs_set_compressed_fflag(a); 1406 } 1407 return (ret); 1408} 1409 1410static ssize_t 1411hfs_write_decmpfs_block(struct archive_write_disk *a, const char *buff, 1412 size_t size) 1413{ 1414 const char *buffer_to_write; 1415 size_t bytes_to_write; 1416 int ret; 1417 1418 if (a->decmpfs_block_count == (unsigned)-1) { 1419 void *new_block; 1420 size_t new_size; 1421 unsigned int block_count; 1422 1423 if (a->decmpfs_header_p == NULL) { 1424 new_block = malloc(MAX_DECMPFS_XATTR_SIZE 1425 + sizeof(uint32_t)); 1426 if (new_block == NULL) { 1427 archive_set_error(&a->archive, ENOMEM, 1428 "Can't allocate memory for decmpfs"); 1429 return (ARCHIVE_FATAL); 1430 } 1431 a->decmpfs_header_p = new_block; 1432 } 1433 a->decmpfs_attr_size = DECMPFS_HEADER_SIZE; 1434 archive_le32enc(&a->decmpfs_header_p[DECMPFS_COMPRESSION_MAGIC], 1435 DECMPFS_MAGIC); 1436 archive_le32enc(&a->decmpfs_header_p[DECMPFS_COMPRESSION_TYPE], 1437 CMP_RESOURCE_FORK); 1438 archive_le64enc(&a->decmpfs_header_p[DECMPFS_UNCOMPRESSED_SIZE], 1439 a->filesize); 1440 1441 /* Calculate a block count of the file. */ 1442 block_count = 1443 (a->filesize + MAX_DECMPFS_BLOCK_SIZE -1) / 1444 MAX_DECMPFS_BLOCK_SIZE; 1445 /* 1446 * Allocate buffer for resource fork. 1447 * Set up related pointers; 1448 */ 1449 new_size = 1450 RSRC_H_SIZE + /* header */ 1451 4 + /* Block count */ 1452 (block_count * sizeof(uint32_t) * 2) + 1453 RSRC_F_SIZE; /* footer */ 1454 if (new_size > a->resource_fork_allocated_size) { 1455 new_block = realloc(a->resource_fork, new_size); 1456 if (new_block == NULL) { 1457 archive_set_error(&a->archive, ENOMEM, 1458 "Can't allocate memory for ResourceFork"); 1459 return (ARCHIVE_FATAL); 1460 } 1461 a->resource_fork_allocated_size = new_size; 1462 a->resource_fork = new_block; 1463 } 1464 1465 /* Allocate uncompressed buffer */ 1466 if (a->uncompressed_buffer == NULL) { 1467 new_block = malloc(MAX_DECMPFS_BLOCK_SIZE); 1468 if (new_block == NULL) { 1469 archive_set_error(&a->archive, ENOMEM, 1470 "Can't allocate memory for decmpfs"); 1471 return (ARCHIVE_FATAL); 1472 } 1473 a->uncompressed_buffer = new_block; 1474 } 1475 a->block_remaining_bytes = MAX_DECMPFS_BLOCK_SIZE; 1476 a->file_remaining_bytes = a->filesize; 1477 a->compressed_buffer_remaining = a->compressed_buffer_size; 1478 1479 /* 1480 * Set up a resource fork. 1481 */ 1482 a->rsrc_xattr_options = XATTR_CREATE; 1483 /* Get the position where we are going to set a bunch 1484 * of block info. */ 1485 a->decmpfs_block_info = 1486 (uint32_t *)(a->resource_fork + RSRC_H_SIZE); 1487 /* Set the block count to the resource fork. */ 1488 archive_le32enc(a->decmpfs_block_info++, block_count); 1489 /* Get the position where we are going to set compressed 1490 * data. */ 1491 a->compressed_rsrc_position = 1492 RSRC_H_SIZE + 4 + (block_count * 8); 1493 a->compressed_rsrc_position_v = a->compressed_rsrc_position; 1494 a->decmpfs_block_count = block_count; 1495 } 1496 1497 /* Ignore redundant bytes. */ 1498 if (a->file_remaining_bytes == 0) 1499 return ((ssize_t)size); 1500 1501 /* Do not overrun a block size. */ 1502 if (size > a->block_remaining_bytes) 1503 bytes_to_write = a->block_remaining_bytes; 1504 else 1505 bytes_to_write = size; 1506 /* Do not overrun the file size. */ 1507 if (bytes_to_write > a->file_remaining_bytes) 1508 bytes_to_write = a->file_remaining_bytes; 1509 1510 /* For efficiency, if a copy length is full of the uncompressed 1511 * buffer size, do not copy writing data to it. */ 1512 if (bytes_to_write == MAX_DECMPFS_BLOCK_SIZE) 1513 buffer_to_write = buff; 1514 else { 1515 memcpy(a->uncompressed_buffer + 1516 MAX_DECMPFS_BLOCK_SIZE - a->block_remaining_bytes, 1517 buff, bytes_to_write); 1518 buffer_to_write = a->uncompressed_buffer; 1519 } 1520 a->block_remaining_bytes -= bytes_to_write; 1521 a->file_remaining_bytes -= bytes_to_write; 1522 1523 if (a->block_remaining_bytes == 0 || a->file_remaining_bytes == 0) { 1524 ret = hfs_drive_compressor(a, buffer_to_write, 1525 MAX_DECMPFS_BLOCK_SIZE - a->block_remaining_bytes); 1526 if (ret < 0) 1527 return (ret); 1528 a->block_remaining_bytes = MAX_DECMPFS_BLOCK_SIZE; 1529 } 1530 /* Ignore redundant bytes. */ 1531 if (a->file_remaining_bytes == 0) 1532 return ((ssize_t)size); 1533 return (bytes_to_write); 1534} 1535 1536static ssize_t 1537hfs_write_data_block(struct archive_write_disk *a, const char *buff, 1538 size_t size) 1539{ 1540 uint64_t start_size = size; 1541 ssize_t bytes_written = 0; 1542 ssize_t bytes_to_write; 1543 1544 if (size == 0) 1545 return (ARCHIVE_OK); 1546 1547 if (a->filesize == 0 || a->fd < 0) { 1548 archive_set_error(&a->archive, 0, 1549 "Attempt to write to an empty file"); 1550 return (ARCHIVE_WARN); 1551 } 1552 1553 /* If this write would run beyond the file size, truncate it. */ 1554 if (a->filesize >= 0 && (int64_t)(a->offset + size) > a->filesize) 1555 start_size = size = (size_t)(a->filesize - a->offset); 1556 1557 /* Write the data. */ 1558 while (size > 0) { 1559 bytes_to_write = size; 1560 /* Seek if necessary to the specified offset. */ 1561 if (a->offset < a->fd_offset) { 1562 /* Can't support backward move. */ 1563 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 1564 "Seek failed"); 1565 return (ARCHIVE_FATAL); 1566 } else if (a->offset > a->fd_offset) { 1567 int64_t skip = a->offset - a->fd_offset; 1568 char nullblock[1024]; 1569 1570 memset(nullblock, 0, sizeof(nullblock)); 1571 while (skip > 0) { 1572 if (skip > (int64_t)sizeof(nullblock)) 1573 bytes_written = hfs_write_decmpfs_block( 1574 a, nullblock, sizeof(nullblock)); 1575 else 1576 bytes_written = hfs_write_decmpfs_block( 1577 a, nullblock, skip); 1578 if (bytes_written < 0) { 1579 archive_set_error(&a->archive, errno, 1580 "Write failed"); 1581 return (ARCHIVE_WARN); 1582 } 1583 skip -= bytes_written; 1584 } 1585 1586 a->fd_offset = a->offset; 1587 } 1588 bytes_written = 1589 hfs_write_decmpfs_block(a, buff, bytes_to_write); 1590 if (bytes_written < 0) 1591 return (bytes_written); 1592 buff += bytes_written; 1593 size -= bytes_written; 1594 a->total_bytes_written += bytes_written; 1595 a->offset += bytes_written; 1596 a->fd_offset = a->offset; 1597 } 1598 return (start_size - size); 1599} 1600#else 1601static ssize_t 1602hfs_write_data_block(struct archive_write_disk *a, const char *buff, 1603 size_t size) 1604{ 1605 return (write_data_block(a, buff, size)); 1606} 1607#endif 1608 1609static ssize_t 1610_archive_write_disk_data_block(struct archive *_a, 1611 const void *buff, size_t size, int64_t offset) 1612{ 1613 struct archive_write_disk *a = (struct archive_write_disk *)_a; 1614 ssize_t r; 1615 1616 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 1617 ARCHIVE_STATE_DATA, "archive_write_data_block"); 1618 1619 a->offset = offset; 1620 if (a->todo & TODO_HFS_COMPRESSION) 1621 r = hfs_write_data_block(a, buff, size); 1622 else 1623 r = write_data_block(a, buff, size); 1624 if (r < ARCHIVE_OK) 1625 return (r); 1626 if ((size_t)r < size) { 1627 archive_set_error(&a->archive, 0, 1628 "Too much data: Truncating file at %ju bytes", 1629 (uintmax_t)a->filesize); 1630 return (ARCHIVE_WARN); 1631 } 1632#if ARCHIVE_VERSION_NUMBER < 3999000 1633 return (ARCHIVE_OK); 1634#else 1635 return (size); 1636#endif 1637} 1638 1639static ssize_t 1640_archive_write_disk_data(struct archive *_a, const void *buff, size_t size) 1641{ 1642 struct archive_write_disk *a = (struct archive_write_disk *)_a; 1643 1644 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 1645 ARCHIVE_STATE_DATA, "archive_write_data"); 1646 1647 if (a->todo & TODO_HFS_COMPRESSION) 1648 return (hfs_write_data_block(a, buff, size)); 1649 return (write_data_block(a, buff, size)); 1650} 1651 1652static int 1653_archive_write_disk_finish_entry(struct archive *_a) 1654{ 1655 struct archive_write_disk *a = (struct archive_write_disk *)_a; 1656 int ret = ARCHIVE_OK; 1657 1658 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 1659 ARCHIVE_STATE_HEADER | ARCHIVE_STATE_DATA, 1660 "archive_write_finish_entry"); 1661 if (a->archive.state & ARCHIVE_STATE_HEADER) 1662 return (ARCHIVE_OK); 1663 archive_clear_error(&a->archive); 1664 1665 /* Pad or truncate file to the right size. */ 1666 if (a->fd < 0) { 1667 /* There's no file. */ 1668 } else if (a->filesize < 0) { 1669 /* File size is unknown, so we can't set the size. */ 1670 } else if (a->fd_offset == a->filesize) { 1671 /* Last write ended at exactly the filesize; we're done. */ 1672 /* Hopefully, this is the common case. */ 1673#if defined(__APPLE__) && defined(UF_COMPRESSED) && defined(HAVE_ZLIB_H) 1674 } else if (a->todo & TODO_HFS_COMPRESSION) { 1675 char null_d[1024]; 1676 ssize_t r; 1677 1678 if (a->file_remaining_bytes) 1679 memset(null_d, 0, sizeof(null_d)); 1680 while (a->file_remaining_bytes) { 1681 if (a->file_remaining_bytes > sizeof(null_d)) 1682 r = hfs_write_data_block( 1683 a, null_d, sizeof(null_d)); 1684 else 1685 r = hfs_write_data_block( 1686 a, null_d, a->file_remaining_bytes); 1687 if (r < 0) 1688 return ((int)r); 1689 } 1690#endif 1691 } else { 1692#if HAVE_FTRUNCATE 1693 if (ftruncate(a->fd, a->filesize) == -1 && 1694 a->filesize == 0) { 1695 archive_set_error(&a->archive, errno, 1696 "File size could not be restored"); 1697 return (ARCHIVE_FAILED); 1698 } 1699#endif 1700 /* 1701 * Not all platforms implement the XSI option to 1702 * extend files via ftruncate. Stat() the file again 1703 * to see what happened. 1704 */ 1705 a->pst = NULL; 1706 if ((ret = lazy_stat(a)) != ARCHIVE_OK) 1707 return (ret); 1708 /* We can use lseek()/write() to extend the file if 1709 * ftruncate didn't work or isn't available. */ 1710 if (a->st.st_size < a->filesize) { 1711 const char nul = '\0'; 1712 if (lseek(a->fd, a->filesize - 1, SEEK_SET) < 0) { 1713 archive_set_error(&a->archive, errno, 1714 "Seek failed"); 1715 return (ARCHIVE_FATAL); 1716 } 1717 if (write(a->fd, &nul, 1) < 0) { 1718 archive_set_error(&a->archive, errno, 1719 "Write to restore size failed"); 1720 return (ARCHIVE_FATAL); 1721 } 1722 a->pst = NULL; 1723 } 1724 } 1725 1726 /* Restore metadata. */ 1727 1728 /* 1729 * This is specific to Mac OS X. 1730 * If the current file is an AppleDouble file, it should be 1731 * linked with the data fork file and remove it. 1732 */ 1733 if (a->todo & TODO_APPLEDOUBLE) { 1734 int r2 = fixup_appledouble(a, a->name); 1735 if (r2 == ARCHIVE_EOF) { 1736 /* The current file has been successfully linked 1737 * with the data fork file and removed. So there 1738 * is nothing to do on the current file. */ 1739 goto finish_metadata; 1740 } 1741 if (r2 < ret) ret = r2; 1742 } 1743 1744 /* 1745 * Look up the "real" UID only if we're going to need it. 1746 * TODO: the TODO_SGID condition can be dropped here, can't it? 1747 */ 1748 if (a->todo & (TODO_OWNER | TODO_SUID | TODO_SGID)) { 1749 a->uid = archive_write_disk_uid(&a->archive, 1750 archive_entry_uname(a->entry), 1751 archive_entry_uid(a->entry)); 1752 } 1753 /* Look up the "real" GID only if we're going to need it. */ 1754 /* TODO: the TODO_SUID condition can be dropped here, can't it? */ 1755 if (a->todo & (TODO_OWNER | TODO_SGID | TODO_SUID)) { 1756 a->gid = archive_write_disk_gid(&a->archive, 1757 archive_entry_gname(a->entry), 1758 archive_entry_gid(a->entry)); 1759 } 1760 1761 /* 1762 * Restore ownership before set_mode tries to restore suid/sgid 1763 * bits. If we set the owner, we know what it is and can skip 1764 * a stat() call to examine the ownership of the file on disk. 1765 */ 1766 if (a->todo & TODO_OWNER) { 1767 int r2 = set_ownership(a); 1768 if (r2 < ret) ret = r2; 1769 } 1770 1771 /* 1772 * HYPOTHESIS: 1773 * If we're not root, we won't be setting any security 1774 * attributes that may be wiped by the set_mode() routine 1775 * below. We also can't set xattr on non-owner-writable files, 1776 * which may be the state after set_mode(). Perform 1777 * set_xattrs() first based on these constraints. 1778 */ 1779 if (a->user_uid != 0 && 1780 (a->todo & TODO_XATTR)) { 1781 int r2 = set_xattrs(a); 1782 if (r2 < ret) ret = r2; 1783 } 1784 1785 /* 1786 * set_mode must precede ACLs on systems such as Solaris and 1787 * FreeBSD where setting the mode implicitly clears extended ACLs 1788 */ 1789 if (a->todo & TODO_MODE) { 1790 int r2 = set_mode(a, a->mode); 1791 if (r2 < ret) ret = r2; 1792 } 1793 1794 /* 1795 * Security-related extended attributes (such as 1796 * security.capability on Linux) have to be restored last, 1797 * since they're implicitly removed by other file changes. 1798 * We do this last only when root. 1799 */ 1800 if (a->user_uid == 0 && 1801 (a->todo & TODO_XATTR)) { 1802 int r2 = set_xattrs(a); 1803 if (r2 < ret) ret = r2; 1804 } 1805 1806 /* 1807 * Some flags prevent file modification; they must be restored after 1808 * file contents are written. 1809 */ 1810 if (a->todo & TODO_FFLAGS) { 1811 int r2 = set_fflags(a); 1812 if (r2 < ret) ret = r2; 1813 } 1814 1815 /* 1816 * Time must follow most other metadata; 1817 * otherwise atime will get changed. 1818 */ 1819 if (a->todo & TODO_TIMES) { 1820 int r2 = set_times_from_entry(a); 1821 if (r2 < ret) ret = r2; 1822 } 1823 1824 /* 1825 * Mac extended metadata includes ACLs. 1826 */ 1827 if (a->todo & TODO_MAC_METADATA) { 1828 const void *metadata; 1829 size_t metadata_size; 1830 metadata = archive_entry_mac_metadata(a->entry, &metadata_size); 1831 if (metadata != NULL && metadata_size > 0) { 1832 int r2 = set_mac_metadata(a, archive_entry_pathname( 1833 a->entry), metadata, metadata_size); 1834 if (r2 < ret) ret = r2; 1835 } 1836 } 1837 1838 /* 1839 * ACLs must be restored after timestamps because there are 1840 * ACLs that prevent attribute changes (including time). 1841 */ 1842 if (a->todo & TODO_ACLS) { 1843 int r2; 1844 r2 = archive_write_disk_set_acls(&a->archive, a->fd, 1845 archive_entry_pathname(a->entry), 1846 archive_entry_acl(a->entry), 1847 archive_entry_mode(a->entry)); 1848 if (r2 < ret) ret = r2; 1849 } 1850 1851finish_metadata: 1852 /* If there's an fd, we can close it now. */ 1853 if (a->fd >= 0) { 1854 close(a->fd); 1855 a->fd = -1; 1856 if (a->tmpname) { 1857 if (rename(a->tmpname, a->name) == -1) { 1858 archive_set_error(&a->archive, errno, 1859 "Failed to rename temporary file"); 1860 ret = ARCHIVE_FAILED; 1861 unlink(a->tmpname); 1862 } 1863 a->tmpname = NULL; 1864 } 1865 } 1866 /* If there's an entry, we can release it now. */ 1867 archive_entry_free(a->entry); 1868 a->entry = NULL; 1869 a->archive.state = ARCHIVE_STATE_HEADER; 1870 return (ret); 1871} 1872 1873int 1874archive_write_disk_set_group_lookup(struct archive *_a, 1875 void *private_data, 1876 la_int64_t (*lookup_gid)(void *private, const char *gname, la_int64_t gid), 1877 void (*cleanup_gid)(void *private)) 1878{ 1879 struct archive_write_disk *a = (struct archive_write_disk *)_a; 1880 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 1881 ARCHIVE_STATE_ANY, "archive_write_disk_set_group_lookup"); 1882 1883 if (a->cleanup_gid != NULL && a->lookup_gid_data != NULL) 1884 (a->cleanup_gid)(a->lookup_gid_data); 1885 1886 a->lookup_gid = lookup_gid; 1887 a->cleanup_gid = cleanup_gid; 1888 a->lookup_gid_data = private_data; 1889 return (ARCHIVE_OK); 1890} 1891 1892int 1893archive_write_disk_set_user_lookup(struct archive *_a, 1894 void *private_data, 1895 int64_t (*lookup_uid)(void *private, const char *uname, int64_t uid), 1896 void (*cleanup_uid)(void *private)) 1897{ 1898 struct archive_write_disk *a = (struct archive_write_disk *)_a; 1899 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 1900 ARCHIVE_STATE_ANY, "archive_write_disk_set_user_lookup"); 1901 1902 if (a->cleanup_uid != NULL && a->lookup_uid_data != NULL) 1903 (a->cleanup_uid)(a->lookup_uid_data); 1904 1905 a->lookup_uid = lookup_uid; 1906 a->cleanup_uid = cleanup_uid; 1907 a->lookup_uid_data = private_data; 1908 return (ARCHIVE_OK); 1909} 1910 1911int64_t 1912archive_write_disk_gid(struct archive *_a, const char *name, la_int64_t id) 1913{ 1914 struct archive_write_disk *a = (struct archive_write_disk *)_a; 1915 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 1916 ARCHIVE_STATE_ANY, "archive_write_disk_gid"); 1917 if (a->lookup_gid) 1918 return (a->lookup_gid)(a->lookup_gid_data, name, id); 1919 return (id); 1920} 1921 1922int64_t 1923archive_write_disk_uid(struct archive *_a, const char *name, la_int64_t id) 1924{ 1925 struct archive_write_disk *a = (struct archive_write_disk *)_a; 1926 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 1927 ARCHIVE_STATE_ANY, "archive_write_disk_uid"); 1928 if (a->lookup_uid) 1929 return (a->lookup_uid)(a->lookup_uid_data, name, id); 1930 return (id); 1931} 1932 1933/* 1934 * Create a new archive_write_disk object and initialize it with global state. 1935 */ 1936struct archive * 1937archive_write_disk_new(void) 1938{ 1939 struct archive_write_disk *a; 1940 1941 a = (struct archive_write_disk *)calloc(1, sizeof(*a)); 1942 if (a == NULL) 1943 return (NULL); 1944 a->archive.magic = ARCHIVE_WRITE_DISK_MAGIC; 1945 /* We're ready to write a header immediately. */ 1946 a->archive.state = ARCHIVE_STATE_HEADER; 1947 a->archive.vtable = archive_write_disk_vtable(); 1948 a->start_time = time(NULL); 1949 /* Query and restore the umask. */ 1950 umask(a->user_umask = umask(0)); 1951#ifdef HAVE_GETEUID 1952 a->user_uid = geteuid(); 1953#endif /* HAVE_GETEUID */ 1954 if (archive_string_ensure(&a->path_safe, 512) == NULL) { 1955 free(a); 1956 return (NULL); 1957 } 1958#ifdef HAVE_ZLIB_H 1959 a->decmpfs_compression_level = 5; 1960#endif 1961 return (&a->archive); 1962} 1963 1964 1965/* 1966 * If pathname is longer than PATH_MAX, chdir to a suitable 1967 * intermediate dir and edit the path down to a shorter suffix. Note 1968 * that this routine never returns an error; if the chdir() attempt 1969 * fails for any reason, we just go ahead with the long pathname. The 1970 * object creation is likely to fail, but any error will get handled 1971 * at that time. 1972 */ 1973#if defined(HAVE_FCHDIR) && defined(PATH_MAX) 1974static void 1975edit_deep_directories(struct archive_write_disk *a) 1976{ 1977 int ret; 1978 char *tail = a->name; 1979 1980 /* If path is short, avoid the open() below. */ 1981 if (strlen(tail) < PATH_MAX) 1982 return; 1983 1984 /* Try to record our starting dir. */ 1985 a->restore_pwd = la_opendirat(AT_FDCWD, "."); 1986 __archive_ensure_cloexec_flag(a->restore_pwd); 1987 if (a->restore_pwd < 0) 1988 return; 1989 1990 /* As long as the path is too long... */ 1991 while (strlen(tail) >= PATH_MAX) { 1992 /* Locate a dir prefix shorter than PATH_MAX. */ 1993 tail += PATH_MAX - 8; 1994 while (tail > a->name && *tail != '/') 1995 tail--; 1996 /* Exit if we find a too-long path component. */ 1997 if (tail <= a->name) 1998 return; 1999 /* Create the intermediate dir and chdir to it. */ 2000 *tail = '\0'; /* Terminate dir portion */ 2001 ret = create_dir(a, a->name); 2002 if (ret == ARCHIVE_OK && chdir(a->name) != 0) 2003 ret = ARCHIVE_FAILED; 2004 *tail = '/'; /* Restore the / we removed. */ 2005 if (ret != ARCHIVE_OK) 2006 return; 2007 tail++; 2008 /* The chdir() succeeded; we've now shortened the path. */ 2009 a->name = tail; 2010 } 2011 return; 2012} 2013#endif 2014 2015/* 2016 * The main restore function. 2017 */ 2018static int 2019restore_entry(struct archive_write_disk *a) 2020{ 2021 int ret = ARCHIVE_OK, en; 2022 2023 if (a->flags & ARCHIVE_EXTRACT_UNLINK && !S_ISDIR(a->mode)) { 2024 /* 2025 * TODO: Fix this. Apparently, there are platforms 2026 * that still allow root to hose the entire filesystem 2027 * by unlinking a dir. The S_ISDIR() test above 2028 * prevents us from using unlink() here if the new 2029 * object is a dir, but that doesn't mean the old 2030 * object isn't a dir. 2031 */ 2032 if (a->flags & ARCHIVE_EXTRACT_CLEAR_NOCHANGE_FFLAGS) 2033 (void)clear_nochange_fflags(a); 2034 if (unlink(a->name) == 0) { 2035 /* We removed it, reset cached stat. */ 2036 a->pst = NULL; 2037 } else if (errno == ENOENT) { 2038 /* File didn't exist, that's just as good. */ 2039 } else if (rmdir(a->name) == 0) { 2040 /* It was a dir, but now it's gone. */ 2041 a->pst = NULL; 2042 } else { 2043 /* We tried, but couldn't get rid of it. */ 2044 archive_set_error(&a->archive, errno, 2045 "Could not unlink"); 2046 return(ARCHIVE_FAILED); 2047 } 2048 } 2049 2050 /* Try creating it first; if this fails, we'll try to recover. */ 2051 en = create_filesystem_object(a); 2052 2053 if ((en == ENOTDIR || en == ENOENT) 2054 && !(a->flags & ARCHIVE_EXTRACT_NO_AUTODIR)) { 2055 /* If the parent dir doesn't exist, try creating it. */ 2056 create_parent_dir(a, a->name); 2057 /* Now try to create the object again. */ 2058 en = create_filesystem_object(a); 2059 } 2060 2061 if ((en == ENOENT) && (archive_entry_hardlink(a->entry) != NULL)) { 2062 archive_set_error(&a->archive, en, 2063 "Hard-link target '%s' does not exist.", 2064 archive_entry_hardlink(a->entry)); 2065 return (ARCHIVE_FAILED); 2066 } 2067 2068 if ((en == EISDIR || en == EEXIST) 2069 && (a->flags & ARCHIVE_EXTRACT_NO_OVERWRITE)) { 2070 /* If we're not overwriting, we're done. */ 2071 if (S_ISDIR(a->mode)) { 2072 /* Don't overwrite any settings on existing directories. */ 2073 a->todo = 0; 2074 } 2075 archive_entry_unset_size(a->entry); 2076 return (ARCHIVE_OK); 2077 } 2078 2079 /* 2080 * Some platforms return EISDIR if you call 2081 * open(O_WRONLY | O_EXCL | O_CREAT) on a directory, some 2082 * return EEXIST. POSIX is ambiguous, requiring EISDIR 2083 * for open(O_WRONLY) on a dir and EEXIST for open(O_EXCL | O_CREAT) 2084 * on an existing item. 2085 */ 2086 if (en == EISDIR) { 2087 /* A dir is in the way of a non-dir, rmdir it. */ 2088 if (rmdir(a->name) != 0) { 2089 archive_set_error(&a->archive, errno, 2090 "Can't remove already-existing dir"); 2091 return (ARCHIVE_FAILED); 2092 } 2093 a->pst = NULL; 2094 /* Try again. */ 2095 en = create_filesystem_object(a); 2096 } else if (en == EEXIST) { 2097 /* 2098 * We know something is in the way, but we don't know what; 2099 * we need to find out before we go any further. 2100 */ 2101 int r = 0; 2102 /* 2103 * The SECURE_SYMLINKS logic has already removed a 2104 * symlink to a dir if the client wants that. So 2105 * follow the symlink if we're creating a dir. 2106 */ 2107 if (S_ISDIR(a->mode)) 2108 r = la_stat(a->name, &a->st); 2109 /* 2110 * If it's not a dir (or it's a broken symlink), 2111 * then don't follow it. 2112 */ 2113 if (r != 0 || !S_ISDIR(a->mode)) 2114 r = lstat(a->name, &a->st); 2115 if (r != 0) { 2116 archive_set_error(&a->archive, errno, 2117 "Can't stat existing object"); 2118 return (ARCHIVE_FAILED); 2119 } 2120 2121 /* 2122 * NO_OVERWRITE_NEWER doesn't apply to directories. 2123 */ 2124 if ((a->flags & ARCHIVE_EXTRACT_NO_OVERWRITE_NEWER) 2125 && !S_ISDIR(a->st.st_mode)) { 2126 if (!older(&(a->st), a->entry)) { 2127 archive_entry_unset_size(a->entry); 2128 return (ARCHIVE_OK); 2129 } 2130 } 2131 2132 /* If it's our archive, we're done. */ 2133 if (a->skip_file_set && 2134 a->st.st_dev == (dev_t)a->skip_file_dev && 2135 a->st.st_ino == (ino_t)a->skip_file_ino) { 2136 archive_set_error(&a->archive, 0, 2137 "Refusing to overwrite archive"); 2138 return (ARCHIVE_FAILED); 2139 } 2140 2141 if (!S_ISDIR(a->st.st_mode)) { 2142 if (a->flags & ARCHIVE_EXTRACT_CLEAR_NOCHANGE_FFLAGS) 2143 (void)clear_nochange_fflags(a); 2144 2145 if ((a->flags & ARCHIVE_EXTRACT_SAFE_WRITES) && 2146 S_ISREG(a->st.st_mode)) { 2147 /* Use a temporary file to extract */ 2148 if ((a->fd = la_mktemp(a)) == -1) { 2149 archive_set_error(&a->archive, errno, 2150 "Can't create temporary file"); 2151 return ARCHIVE_FAILED; 2152 } 2153 a->pst = NULL; 2154 en = 0; 2155 } else { 2156 /* A non-dir is in the way, unlink it. */ 2157 if (unlink(a->name) != 0) { 2158 archive_set_error(&a->archive, errno, 2159 "Can't unlink already-existing " 2160 "object"); 2161 return (ARCHIVE_FAILED); 2162 } 2163 a->pst = NULL; 2164 /* Try again. */ 2165 en = create_filesystem_object(a); 2166 } 2167 } else if (!S_ISDIR(a->mode)) { 2168 /* A dir is in the way of a non-dir, rmdir it. */ 2169 if (a->flags & ARCHIVE_EXTRACT_CLEAR_NOCHANGE_FFLAGS) 2170 (void)clear_nochange_fflags(a); 2171 if (rmdir(a->name) != 0) { 2172 archive_set_error(&a->archive, errno, 2173 "Can't replace existing directory with non-directory"); 2174 return (ARCHIVE_FAILED); 2175 } 2176 /* Try again. */ 2177 en = create_filesystem_object(a); 2178 } else { 2179 /* 2180 * There's a dir in the way of a dir. Don't 2181 * waste time with rmdir()/mkdir(), just fix 2182 * up the permissions on the existing dir. 2183 * Note that we don't change perms on existing 2184 * dirs unless _EXTRACT_PERM is specified. 2185 */ 2186 if ((a->mode != a->st.st_mode) 2187 && (a->todo & TODO_MODE_FORCE)) 2188 a->deferred |= (a->todo & TODO_MODE); 2189 /* Ownership doesn't need deferred fixup. */ 2190 en = 0; /* Forget the EEXIST. */ 2191 } 2192 } 2193 2194 if (en) { 2195 /* Everything failed; give up here. */ 2196 if ((&a->archive)->error == NULL) 2197 archive_set_error(&a->archive, en, "Can't create '%s'", 2198 a->name); 2199 return (ARCHIVE_FAILED); 2200 } 2201 2202 a->pst = NULL; /* Cached stat data no longer valid. */ 2203 return (ret); 2204} 2205 2206/* 2207 * Returns 0 if creation succeeds, or else returns errno value from 2208 * the failed system call. Note: This function should only ever perform 2209 * a single system call. 2210 */ 2211static int 2212create_filesystem_object(struct archive_write_disk *a) 2213{ 2214 /* Create the entry. */ 2215 const char *linkname; 2216 mode_t final_mode, mode; 2217 int r; 2218 /* these for check_symlinks_fsobj */ 2219 char *linkname_copy; /* non-const copy of linkname */ 2220 struct stat st; 2221 struct archive_string error_string; 2222 int error_number; 2223 2224 /* We identify hard/symlinks according to the link names. */ 2225 /* Since link(2) and symlink(2) don't handle modes, we're done here. */ 2226 linkname = archive_entry_hardlink(a->entry); 2227 if (linkname != NULL) { 2228#if !HAVE_LINK 2229 return (EPERM); 2230#else 2231 archive_string_init(&error_string); 2232 linkname_copy = strdup(linkname); 2233 if (linkname_copy == NULL) { 2234 return (EPERM); 2235 } 2236 /* 2237 * TODO: consider using the cleaned-up path as the link 2238 * target? 2239 */ 2240 r = cleanup_pathname_fsobj(linkname_copy, &error_number, 2241 &error_string, a->flags); 2242 if (r != ARCHIVE_OK) { 2243 archive_set_error(&a->archive, error_number, "%s", 2244 error_string.s); 2245 free(linkname_copy); 2246 archive_string_free(&error_string); 2247 /* 2248 * EPERM is more appropriate than error_number for our 2249 * callers 2250 */ 2251 return (EPERM); 2252 } 2253 r = check_symlinks_fsobj(linkname_copy, &error_number, 2254 &error_string, a->flags); 2255 if (r != ARCHIVE_OK) { 2256 archive_set_error(&a->archive, error_number, "%s", 2257 error_string.s); 2258 free(linkname_copy); 2259 archive_string_free(&error_string); 2260 /* 2261 * EPERM is more appropriate than error_number for our 2262 * callers 2263 */ 2264 return (EPERM); 2265 } 2266 free(linkname_copy); 2267 archive_string_free(&error_string); 2268 /* 2269 * Unlinking and linking here is really not atomic, 2270 * but doing it right, would require us to construct 2271 * an mktemplink() function, and then use rename(2). 2272 */ 2273 if (a->flags & ARCHIVE_EXTRACT_SAFE_WRITES) 2274 unlink(a->name); 2275 r = link(linkname, a->name) ? errno : 0; 2276 /* 2277 * New cpio and pax formats allow hardlink entries 2278 * to carry data, so we may have to open the file 2279 * for hardlink entries. 2280 * 2281 * If the hardlink was successfully created and 2282 * the archive doesn't have carry data for it, 2283 * consider it to be non-authoritative for meta data. 2284 * This is consistent with GNU tar and BSD pax. 2285 * If the hardlink does carry data, let the last 2286 * archive entry decide ownership. 2287 */ 2288 if (r == 0 && a->filesize <= 0) { 2289 a->todo = 0; 2290 a->deferred = 0; 2291 } else if (r == 0 && a->filesize > 0) { 2292#ifdef HAVE_LSTAT 2293 r = lstat(a->name, &st); 2294#else 2295 r = la_stat(a->name, &st); 2296#endif 2297 if (r != 0) 2298 r = errno; 2299 else if ((st.st_mode & AE_IFMT) == AE_IFREG) { 2300 a->fd = open(a->name, O_WRONLY | O_TRUNC | 2301 O_BINARY | O_CLOEXEC | O_NOFOLLOW); 2302 __archive_ensure_cloexec_flag(a->fd); 2303 if (a->fd < 0) 2304 r = errno; 2305 } 2306 } 2307 return (r); 2308#endif 2309 } 2310 linkname = archive_entry_symlink(a->entry); 2311 if (linkname != NULL) { 2312#if HAVE_SYMLINK 2313 /* 2314 * Unlinking and linking here is really not atomic, 2315 * but doing it right, would require us to construct 2316 * an mktempsymlink() function, and then use rename(2). 2317 */ 2318 if (a->flags & ARCHIVE_EXTRACT_SAFE_WRITES) 2319 unlink(a->name); 2320 return symlink(linkname, a->name) ? errno : 0; 2321#else 2322 return (EPERM); 2323#endif 2324 } 2325 2326 /* 2327 * The remaining system calls all set permissions, so let's 2328 * try to take advantage of that to avoid an extra chmod() 2329 * call. (Recall that umask is set to zero right now!) 2330 */ 2331 2332 /* Mode we want for the final restored object (w/o file type bits). */ 2333 final_mode = a->mode & 07777; 2334 /* 2335 * The mode that will actually be restored in this step. Note 2336 * that SUID, SGID, etc, require additional work to ensure 2337 * security, so we never restore them at this point. 2338 */ 2339 mode = final_mode & 0777 & ~a->user_umask; 2340 2341 /* 2342 * Always create writable such that [f]setxattr() works if we're not 2343 * root. 2344 */ 2345 if (a->user_uid != 0 && 2346 a->todo & (TODO_HFS_COMPRESSION | TODO_XATTR)) { 2347 mode |= 0200; 2348 } 2349 2350 switch (a->mode & AE_IFMT) { 2351 default: 2352 /* POSIX requires that we fall through here. */ 2353 /* FALLTHROUGH */ 2354 case AE_IFREG: 2355 a->tmpname = NULL; 2356 a->fd = open(a->name, 2357 O_WRONLY | O_CREAT | O_EXCL | O_BINARY | O_CLOEXEC, mode); 2358 __archive_ensure_cloexec_flag(a->fd); 2359 r = (a->fd < 0); 2360 break; 2361 case AE_IFCHR: 2362#ifdef HAVE_MKNOD 2363 /* Note: we use AE_IFCHR for the case label, and 2364 * S_IFCHR for the mknod() call. This is correct. */ 2365 r = mknod(a->name, mode | S_IFCHR, 2366 archive_entry_rdev(a->entry)); 2367 break; 2368#else 2369 /* TODO: Find a better way to warn about our inability 2370 * to restore a char device node. */ 2371 return (EINVAL); 2372#endif /* HAVE_MKNOD */ 2373 case AE_IFBLK: 2374#ifdef HAVE_MKNOD 2375 r = mknod(a->name, mode | S_IFBLK, 2376 archive_entry_rdev(a->entry)); 2377 break; 2378#else 2379 /* TODO: Find a better way to warn about our inability 2380 * to restore a block device node. */ 2381 return (EINVAL); 2382#endif /* HAVE_MKNOD */ 2383 case AE_IFDIR: 2384 mode = (mode | MINIMUM_DIR_MODE) & MAXIMUM_DIR_MODE; 2385 r = mkdir(a->name, mode); 2386 if (r == 0) { 2387 /* Defer setting dir times. */ 2388 a->deferred |= (a->todo & TODO_TIMES); 2389 a->todo &= ~TODO_TIMES; 2390 /* Never use an immediate chmod(). */ 2391 /* We can't avoid the chmod() entirely if EXTRACT_PERM 2392 * because of SysV SGID inheritance. */ 2393 if ((mode != final_mode) 2394 || (a->flags & ARCHIVE_EXTRACT_PERM)) 2395 a->deferred |= (a->todo & TODO_MODE); 2396 a->todo &= ~TODO_MODE; 2397 } 2398 break; 2399 case AE_IFIFO: 2400#ifdef HAVE_MKFIFO 2401 r = mkfifo(a->name, mode); 2402 break; 2403#else 2404 /* TODO: Find a better way to warn about our inability 2405 * to restore a fifo. */ 2406 return (EINVAL); 2407#endif /* HAVE_MKFIFO */ 2408 } 2409 2410 /* All the system calls above set errno on failure. */ 2411 if (r) 2412 return (errno); 2413 2414 /* If we managed to set the final mode, we've avoided a chmod(). */ 2415 if (mode == final_mode) 2416 a->todo &= ~TODO_MODE; 2417 return (0); 2418} 2419 2420/* 2421 * Cleanup function for archive_extract. Mostly, this involves processing 2422 * the fixup list, which is used to address a number of problems: 2423 * * Dir permissions might prevent us from restoring a file in that 2424 * dir, so we restore the dir with minimum 0700 permissions first, 2425 * then correct the mode at the end. 2426 * * Similarly, the act of restoring a file touches the directory 2427 * and changes the timestamp on the dir, so we have to touch-up dir 2428 * timestamps at the end as well. 2429 * * Some file flags can interfere with the restore by, for example, 2430 * preventing the creation of hardlinks to those files. 2431 * * Mac OS extended metadata includes ACLs, so must be deferred on dirs. 2432 * 2433 * Note that tar/cpio do not require that archives be in a particular 2434 * order; there is no way to know when the last file has been restored 2435 * within a directory, so there's no way to optimize the memory usage 2436 * here by fixing up the directory any earlier than the 2437 * end-of-archive. 2438 * 2439 * XXX TODO: Directory ACLs should be restored here, for the same 2440 * reason we set directory perms here. XXX 2441 */ 2442static int 2443_archive_write_disk_close(struct archive *_a) 2444{ 2445 struct archive_write_disk *a = (struct archive_write_disk *)_a; 2446 struct fixup_entry *next, *p; 2447 int fd, ret; 2448 2449 archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC, 2450 ARCHIVE_STATE_HEADER | ARCHIVE_STATE_DATA, 2451 "archive_write_disk_close"); 2452 ret = _archive_write_disk_finish_entry(&a->archive); 2453 2454 /* Sort dir list so directories are fixed up in depth-first order. */ 2455 p = sort_dir_list(a->fixup_list); 2456 2457 while (p != NULL) { 2458 fd = -1; 2459 a->pst = NULL; /* Mark stat cache as out-of-date. */ 2460 if (p->fixup & 2461 (TODO_TIMES | TODO_MODE_BASE | TODO_ACLS | TODO_FFLAGS)) { 2462 fd = open(p->name, 2463 O_WRONLY | O_BINARY | O_NOFOLLOW | O_CLOEXEC); 2464 } 2465 if (p->fixup & TODO_TIMES) { 2466 set_times(a, fd, p->mode, p->name, 2467 p->atime, p->atime_nanos, 2468 p->birthtime, p->birthtime_nanos, 2469 p->mtime, p->mtime_nanos, 2470 p->ctime, p->ctime_nanos); 2471 } 2472 if (p->fixup & TODO_MODE_BASE) { 2473#ifdef HAVE_FCHMOD 2474 if (fd >= 0) 2475 fchmod(fd, p->mode); 2476 else 2477#endif 2478 chmod(p->name, p->mode); 2479 } 2480 if (p->fixup & TODO_ACLS) 2481 archive_write_disk_set_acls(&a->archive, fd, 2482 p->name, &p->acl, p->mode); 2483 if (p->fixup & TODO_FFLAGS) 2484 set_fflags_platform(a, fd, p->name, 2485 p->mode, p->fflags_set, 0); 2486 if (p->fixup & TODO_MAC_METADATA) 2487 set_mac_metadata(a, p->name, p->mac_metadata, 2488 p->mac_metadata_size); 2489 next = p->next; 2490 archive_acl_clear(&p->acl); 2491 free(p->mac_metadata); 2492 free(p->name); 2493 if (fd >= 0) 2494 close(fd); 2495 free(p); 2496 p = next; 2497 } 2498 a->fixup_list = NULL; 2499 return (ret); 2500} 2501 2502static int 2503_archive_write_disk_free(struct archive *_a) 2504{ 2505 struct archive_write_disk *a; 2506 int ret; 2507 if (_a == NULL) 2508 return (ARCHIVE_OK); 2509 archive_check_magic(_a, ARCHIVE_WRITE_DISK_MAGIC, 2510 ARCHIVE_STATE_ANY | ARCHIVE_STATE_FATAL, "archive_write_disk_free"); 2511 a = (struct archive_write_disk *)_a; 2512 ret = _archive_write_disk_close(&a->archive); 2513 archive_write_disk_set_group_lookup(&a->archive, NULL, NULL, NULL); 2514 archive_write_disk_set_user_lookup(&a->archive, NULL, NULL, NULL); 2515 archive_entry_free(a->entry); 2516 archive_string_free(&a->_name_data); 2517 archive_string_free(&a->_tmpname_data); 2518 archive_string_free(&a->archive.error_string); 2519 archive_string_free(&a->path_safe); 2520 a->archive.magic = 0; 2521 __archive_clean(&a->archive); 2522 free(a->decmpfs_header_p); 2523 free(a->resource_fork); 2524 free(a->compressed_buffer); 2525 free(a->uncompressed_buffer); 2526#if defined(__APPLE__) && defined(UF_COMPRESSED) && defined(HAVE_SYS_XATTR_H)\ 2527 && defined(HAVE_ZLIB_H) 2528 if (a->stream_valid) { 2529 switch (deflateEnd(&a->stream)) { 2530 case Z_OK: 2531 break; 2532 default: 2533 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 2534 "Failed to clean up compressor"); 2535 ret = ARCHIVE_FATAL; 2536 break; 2537 } 2538 } 2539#endif 2540 free(a); 2541 return (ret); 2542} 2543 2544/* 2545 * Simple O(n log n) merge sort to order the fixup list. In 2546 * particular, we want to restore dir timestamps depth-first. 2547 */ 2548static struct fixup_entry * 2549sort_dir_list(struct fixup_entry *p) 2550{ 2551 struct fixup_entry *a, *b, *t; 2552 2553 if (p == NULL) 2554 return (NULL); 2555 /* A one-item list is already sorted. */ 2556 if (p->next == NULL) 2557 return (p); 2558 2559 /* Step 1: split the list. */ 2560 t = p; 2561 a = p->next->next; 2562 while (a != NULL) { 2563 /* Step a twice, t once. */ 2564 a = a->next; 2565 if (a != NULL) 2566 a = a->next; 2567 t = t->next; 2568 } 2569 /* Now, t is at the mid-point, so break the list here. */ 2570 b = t->next; 2571 t->next = NULL; 2572 a = p; 2573 2574 /* Step 2: Recursively sort the two sub-lists. */ 2575 a = sort_dir_list(a); 2576 b = sort_dir_list(b); 2577 2578 /* Step 3: Merge the returned lists. */ 2579 /* Pick the first element for the merged list. */ 2580 if (strcmp(a->name, b->name) > 0) { 2581 t = p = a; 2582 a = a->next; 2583 } else { 2584 t = p = b; 2585 b = b->next; 2586 } 2587 2588 /* Always put the later element on the list first. */ 2589 while (a != NULL && b != NULL) { 2590 if (strcmp(a->name, b->name) > 0) { 2591 t->next = a; 2592 a = a->next; 2593 } else { 2594 t->next = b; 2595 b = b->next; 2596 } 2597 t = t->next; 2598 } 2599 2600 /* Only one list is non-empty, so just splice it on. */ 2601 if (a != NULL) 2602 t->next = a; 2603 if (b != NULL) 2604 t->next = b; 2605 2606 return (p); 2607} 2608 2609/* 2610 * Returns a new, initialized fixup entry. 2611 * 2612 * TODO: Reduce the memory requirements for this list by using a tree 2613 * structure rather than a simple list of names. 2614 */ 2615static struct fixup_entry * 2616new_fixup(struct archive_write_disk *a, const char *pathname) 2617{ 2618 struct fixup_entry *fe; 2619 2620 fe = (struct fixup_entry *)calloc(1, sizeof(struct fixup_entry)); 2621 if (fe == NULL) { 2622 archive_set_error(&a->archive, ENOMEM, 2623 "Can't allocate memory for a fixup"); 2624 return (NULL); 2625 } 2626 fe->next = a->fixup_list; 2627 a->fixup_list = fe; 2628 fe->fixup = 0; 2629 fe->name = strdup(pathname); 2630 return (fe); 2631} 2632 2633/* 2634 * Returns a fixup structure for the current entry. 2635 */ 2636static struct fixup_entry * 2637current_fixup(struct archive_write_disk *a, const char *pathname) 2638{ 2639 if (a->current_fixup == NULL) 2640 a->current_fixup = new_fixup(a, pathname); 2641 return (a->current_fixup); 2642} 2643 2644/* Error helper for new *_fsobj functions */ 2645static void 2646fsobj_error(int *a_eno, struct archive_string *a_estr, 2647 int err, const char *errstr, const char *path) 2648{ 2649 if (a_eno) 2650 *a_eno = err; 2651 if (a_estr) 2652 archive_string_sprintf(a_estr, "%s%s", errstr, path); 2653} 2654 2655/* 2656 * TODO: Someday, integrate this with the deep dir support; they both 2657 * scan the path and both can be optimized by comparing against other 2658 * recent paths. 2659 */ 2660/* 2661 * Checks the given path to see if any elements along it are symlinks. Returns 2662 * ARCHIVE_OK if there are none, otherwise puts an error in errmsg. 2663 */ 2664static int 2665check_symlinks_fsobj(char *path, int *a_eno, struct archive_string *a_estr, 2666 int flags) 2667{ 2668#if !defined(HAVE_LSTAT) && \ 2669 !(defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT)) 2670 /* Platform doesn't have lstat, so we can't look for symlinks. */ 2671 (void)path; /* UNUSED */ 2672 (void)error_number; /* UNUSED */ 2673 (void)error_string; /* UNUSED */ 2674 (void)flags; /* UNUSED */ 2675 return (ARCHIVE_OK); 2676#else 2677 int res = ARCHIVE_OK; 2678 char *tail; 2679 char *head; 2680 int last; 2681 char c; 2682 int r; 2683 struct stat st; 2684 int chdir_fd; 2685#if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT) 2686 int fd; 2687#endif 2688 2689 /* Nothing to do here if name is empty */ 2690 if(path[0] == '\0') 2691 return (ARCHIVE_OK); 2692 2693 /* 2694 * Guard against symlink tricks. Reject any archive entry whose 2695 * destination would be altered by a symlink. 2696 * 2697 * Walk the filename in chunks separated by '/'. For each segment: 2698 * - if it doesn't exist, continue 2699 * - if it's symlink, abort or remove it 2700 * - if it's a directory and it's not the last chunk, cd into it 2701 * As we go: 2702 * head points to the current (relative) path 2703 * tail points to the temporary \0 terminating the segment we're 2704 * currently examining 2705 * c holds what used to be in *tail 2706 * last is 1 if this is the last tail 2707 */ 2708 chdir_fd = la_opendirat(AT_FDCWD, "."); 2709 __archive_ensure_cloexec_flag(chdir_fd); 2710 if (chdir_fd < 0) { 2711 fsobj_error(a_eno, a_estr, errno, 2712 "Could not open ", path); 2713 return (ARCHIVE_FATAL); 2714 } 2715 head = path; 2716 tail = path; 2717 last = 0; 2718 /* TODO: reintroduce a safe cache here? */ 2719 /* Skip the root directory if the path is absolute. */ 2720 if(tail == path && tail[0] == '/') 2721 ++tail; 2722 /* Keep going until we've checked the entire name. 2723 * head, tail, path all alias the same string, which is 2724 * temporarily zeroed at tail, so be careful restoring the 2725 * stashed (c=tail[0]) for error messages. 2726 * Exiting the loop with break is okay; continue is not. 2727 */ 2728 while (!last) { 2729 /* 2730 * Skip the separator we just consumed, plus any adjacent ones 2731 */ 2732 while (*tail == '/') 2733 ++tail; 2734 /* Skip the next path element. */ 2735 while (*tail != '\0' && *tail != '/') 2736 ++tail; 2737 /* is this the last path component? */ 2738 last = (tail[0] == '\0') || (tail[0] == '/' && tail[1] == '\0'); 2739 /* temporarily truncate the string here */ 2740 c = tail[0]; 2741 tail[0] = '\0'; 2742 /* Check that we haven't hit a symlink. */ 2743#if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT) 2744 r = fstatat(chdir_fd, head, &st, AT_SYMLINK_NOFOLLOW); 2745#else 2746 r = lstat(head, &st); 2747#endif 2748 if (r != 0) { 2749 tail[0] = c; 2750 /* We've hit a dir that doesn't exist; stop now. */ 2751 if (errno == ENOENT) { 2752 break; 2753 } else { 2754 /* 2755 * Treat any other error as fatal - best to be 2756 * paranoid here. 2757 * Note: This effectively disables deep 2758 * directory support when security checks are 2759 * enabled. Otherwise, very long pathnames that 2760 * trigger an error here could evade the 2761 * sandbox. 2762 * TODO: We could do better, but it would 2763 * probably require merging the symlink checks 2764 * with the deep-directory editing. 2765 */ 2766 fsobj_error(a_eno, a_estr, errno, 2767 "Could not stat ", path); 2768 res = ARCHIVE_FAILED; 2769 break; 2770 } 2771 } else if (S_ISDIR(st.st_mode)) { 2772 if (!last) { 2773#if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT) 2774 fd = la_opendirat(chdir_fd, head); 2775 if (fd < 0) 2776 r = -1; 2777 else { 2778 r = 0; 2779 close(chdir_fd); 2780 chdir_fd = fd; 2781 } 2782#else 2783 r = chdir(head); 2784#endif 2785 if (r != 0) { 2786 tail[0] = c; 2787 fsobj_error(a_eno, a_estr, errno, 2788 "Could not chdir ", path); 2789 res = (ARCHIVE_FATAL); 2790 break; 2791 } 2792 /* Our view is now from inside this dir: */ 2793 head = tail + 1; 2794 } 2795 } else if (S_ISLNK(st.st_mode)) { 2796 if (last) { 2797 /* 2798 * Last element is symlink; remove it 2799 * so we can overwrite it with the 2800 * item being extracted. 2801 */ 2802#if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT) 2803 r = unlinkat(chdir_fd, head, 0); 2804#else 2805 r = unlink(head); 2806#endif 2807 if (r != 0) { 2808 tail[0] = c; 2809 fsobj_error(a_eno, a_estr, errno, 2810 "Could not remove symlink ", 2811 path); 2812 res = ARCHIVE_FAILED; 2813 break; 2814 } 2815 /* 2816 * Even if we did remove it, a warning 2817 * is in order. The warning is silly, 2818 * though, if we're just replacing one 2819 * symlink with another symlink. 2820 */ 2821 tail[0] = c; 2822 /* 2823 * FIXME: not sure how important this is to 2824 * restore 2825 */ 2826 /* 2827 if (!S_ISLNK(path)) { 2828 fsobj_error(a_eno, a_estr, 0, 2829 "Removing symlink ", path); 2830 } 2831 */ 2832 /* Symlink gone. No more problem! */ 2833 res = ARCHIVE_OK; 2834 break; 2835 } else if (flags & ARCHIVE_EXTRACT_UNLINK) { 2836 /* User asked us to remove problems. */ 2837#if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT) 2838 r = unlinkat(chdir_fd, head, 0); 2839#else 2840 r = unlink(head); 2841#endif 2842 if (r != 0) { 2843 tail[0] = c; 2844 fsobj_error(a_eno, a_estr, 0, 2845 "Cannot remove intervening " 2846 "symlink ", path); 2847 res = ARCHIVE_FAILED; 2848 break; 2849 } 2850 tail[0] = c; 2851 } else if ((flags & 2852 ARCHIVE_EXTRACT_SECURE_SYMLINKS) == 0) { 2853 /* 2854 * We are not the last element and we want to 2855 * follow symlinks if they are a directory. 2856 * 2857 * This is needed to extract hardlinks over 2858 * symlinks. 2859 */ 2860#if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT) 2861 r = fstatat(chdir_fd, head, &st, 0); 2862#else 2863 r = la_stat(head, &st); 2864#endif 2865 if (r != 0) { 2866 tail[0] = c; 2867 if (errno == ENOENT) { 2868 break; 2869 } else { 2870 fsobj_error(a_eno, a_estr, 2871 errno, 2872 "Could not stat ", path); 2873 res = (ARCHIVE_FAILED); 2874 break; 2875 } 2876 } else if (S_ISDIR(st.st_mode)) { 2877#if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT) 2878 fd = la_opendirat(chdir_fd, head); 2879 if (fd < 0) 2880 r = -1; 2881 else { 2882 r = 0; 2883 close(chdir_fd); 2884 chdir_fd = fd; 2885 } 2886#else 2887 r = chdir(head); 2888#endif 2889 if (r != 0) { 2890 tail[0] = c; 2891 fsobj_error(a_eno, a_estr, 2892 errno, 2893 "Could not chdir ", path); 2894 res = (ARCHIVE_FATAL); 2895 break; 2896 } 2897 /* 2898 * Our view is now from inside 2899 * this dir: 2900 */ 2901 head = tail + 1; 2902 } else { 2903 tail[0] = c; 2904 fsobj_error(a_eno, a_estr, 0, 2905 "Cannot extract through " 2906 "symlink ", path); 2907 res = ARCHIVE_FAILED; 2908 break; 2909 } 2910 } else { 2911 tail[0] = c; 2912 fsobj_error(a_eno, a_estr, 0, 2913 "Cannot extract through symlink ", path); 2914 res = ARCHIVE_FAILED; 2915 break; 2916 } 2917 } 2918 /* be sure to always maintain this */ 2919 tail[0] = c; 2920 if (tail[0] != '\0') 2921 tail++; /* Advance to the next segment. */ 2922 } 2923 /* Catches loop exits via break */ 2924 tail[0] = c; 2925#if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT) 2926 /* If we operate with openat(), fstatat() and unlinkat() there was 2927 * no chdir(), so just close the fd */ 2928 if (chdir_fd >= 0) 2929 close(chdir_fd); 2930#elif HAVE_FCHDIR 2931 /* If we changed directory above, restore it here. */ 2932 if (chdir_fd >= 0) { 2933 r = fchdir(chdir_fd); 2934 if (r != 0) { 2935 fsobj_error(a_eno, a_estr, errno, 2936 "chdir() failure", ""); 2937 } 2938 close(chdir_fd); 2939 chdir_fd = -1; 2940 if (r != 0) { 2941 res = (ARCHIVE_FATAL); 2942 } 2943 } 2944#endif 2945 /* TODO: reintroduce a safe cache here? */ 2946 return res; 2947#endif 2948} 2949 2950/* 2951 * Check a->name for symlinks, returning ARCHIVE_OK if its clean, otherwise 2952 * calls archive_set_error and returns ARCHIVE_{FATAL,FAILED} 2953 */ 2954static int 2955check_symlinks(struct archive_write_disk *a) 2956{ 2957 struct archive_string error_string; 2958 int error_number; 2959 int rc; 2960 archive_string_init(&error_string); 2961 rc = check_symlinks_fsobj(a->name, &error_number, &error_string, 2962 a->flags); 2963 if (rc != ARCHIVE_OK) { 2964 archive_set_error(&a->archive, error_number, "%s", 2965 error_string.s); 2966 } 2967 archive_string_free(&error_string); 2968 a->pst = NULL; /* to be safe */ 2969 return rc; 2970} 2971 2972 2973#if defined(__CYGWIN__) 2974/* 2975 * 1. Convert a path separator from '\' to '/' . 2976 * We shouldn't check multibyte character directly because some 2977 * character-set have been using the '\' character for a part of 2978 * its multibyte character code. 2979 * 2. Replace unusable characters in Windows with underscore('_'). 2980 * See also : http://msdn.microsoft.com/en-us/library/aa365247.aspx 2981 */ 2982static void 2983cleanup_pathname_win(char *path) 2984{ 2985 wchar_t wc; 2986 char *p; 2987 size_t alen, l; 2988 int mb, complete, utf8; 2989 2990 alen = 0; 2991 mb = 0; 2992 complete = 1; 2993 utf8 = (strcmp(nl_langinfo(CODESET), "UTF-8") == 0)? 1: 0; 2994 for (p = path; *p != '\0'; p++) { 2995 ++alen; 2996 if (*p == '\\') { 2997 /* If previous byte is smaller than 128, 2998 * this is not second byte of multibyte characters, 2999 * so we can replace '\' with '/'. */ 3000 if (utf8 || !mb) 3001 *p = '/'; 3002 else 3003 complete = 0;/* uncompleted. */ 3004 } else if (*(unsigned char *)p > 127) 3005 mb = 1; 3006 else 3007 mb = 0; 3008 /* Rewrite the path name if its next character is unusable. */ 3009 if (*p == ':' || *p == '*' || *p == '?' || *p == '"' || 3010 *p == '<' || *p == '>' || *p == '|') 3011 *p = '_'; 3012 } 3013 if (complete) 3014 return; 3015 3016 /* 3017 * Convert path separator in wide-character. 3018 */ 3019 p = path; 3020 while (*p != '\0' && alen) { 3021 l = mbtowc(&wc, p, alen); 3022 if (l == (size_t)-1) { 3023 while (*p != '\0') { 3024 if (*p == '\\') 3025 *p = '/'; 3026 ++p; 3027 } 3028 break; 3029 } 3030 if (l == 1 && wc == L'\\') 3031 *p = '/'; 3032 p += l; 3033 alen -= l; 3034 } 3035} 3036#endif 3037 3038/* 3039 * Canonicalize the pathname. In particular, this strips duplicate 3040 * '/' characters, '.' elements, and trailing '/'. It also raises an 3041 * error for an empty path, a trailing '..', (if _SECURE_NODOTDOT is 3042 * set) any '..' in the path or (if ARCHIVE_EXTRACT_SECURE_NOABSOLUTEPATHS 3043 * is set) if the path is absolute. 3044 */ 3045static int 3046cleanup_pathname_fsobj(char *path, int *a_eno, struct archive_string *a_estr, 3047 int flags) 3048{ 3049 char *dest, *src; 3050 char separator = '\0'; 3051 3052 dest = src = path; 3053 if (*src == '\0') { 3054 fsobj_error(a_eno, a_estr, ARCHIVE_ERRNO_MISC, 3055 "Invalid empty ", "pathname"); 3056 return (ARCHIVE_FAILED); 3057 } 3058 3059#if defined(__CYGWIN__) 3060 cleanup_pathname_win(path); 3061#endif 3062 /* Skip leading '/'. */ 3063 if (*src == '/') { 3064 if (flags & ARCHIVE_EXTRACT_SECURE_NOABSOLUTEPATHS) { 3065 fsobj_error(a_eno, a_estr, ARCHIVE_ERRNO_MISC, 3066 "Path is ", "absolute"); 3067 return (ARCHIVE_FAILED); 3068 } 3069 3070 separator = *src++; 3071 } 3072 3073 /* Scan the pathname one element at a time. */ 3074 for (;;) { 3075 /* src points to first char after '/' */ 3076 if (src[0] == '\0') { 3077 break; 3078 } else if (src[0] == '/') { 3079 /* Found '//', ignore second one. */ 3080 src++; 3081 continue; 3082 } else if (src[0] == '.') { 3083 if (src[1] == '\0') { 3084 /* Ignore trailing '.' */ 3085 break; 3086 } else if (src[1] == '/') { 3087 /* Skip './'. */ 3088 src += 2; 3089 continue; 3090 } else if (src[1] == '.') { 3091 if (src[2] == '/' || src[2] == '\0') { 3092 /* Conditionally warn about '..' */ 3093 if (flags 3094 & ARCHIVE_EXTRACT_SECURE_NODOTDOT) { 3095 fsobj_error(a_eno, a_estr, 3096 ARCHIVE_ERRNO_MISC, 3097 "Path contains ", "'..'"); 3098 return (ARCHIVE_FAILED); 3099 } 3100 } 3101 /* 3102 * Note: Under no circumstances do we 3103 * remove '..' elements. In 3104 * particular, restoring 3105 * '/foo/../bar/' should create the 3106 * 'foo' dir as a side-effect. 3107 */ 3108 } 3109 } 3110 3111 /* Copy current element, including leading '/'. */ 3112 if (separator) 3113 *dest++ = '/'; 3114 while (*src != '\0' && *src != '/') { 3115 *dest++ = *src++; 3116 } 3117 3118 if (*src == '\0') 3119 break; 3120 3121 /* Skip '/' separator. */ 3122 separator = *src++; 3123 } 3124 /* 3125 * We've just copied zero or more path elements, not including the 3126 * final '/'. 3127 */ 3128 if (dest == path) { 3129 /* 3130 * Nothing got copied. The path must have been something 3131 * like '.' or '/' or './' or '/././././/./'. 3132 */ 3133 if (separator) 3134 *dest++ = '/'; 3135 else 3136 *dest++ = '.'; 3137 } 3138 /* Terminate the result. */ 3139 *dest = '\0'; 3140 return (ARCHIVE_OK); 3141} 3142 3143static int 3144cleanup_pathname(struct archive_write_disk *a) 3145{ 3146 struct archive_string error_string; 3147 int error_number; 3148 int rc; 3149 archive_string_init(&error_string); 3150 rc = cleanup_pathname_fsobj(a->name, &error_number, &error_string, 3151 a->flags); 3152 if (rc != ARCHIVE_OK) { 3153 archive_set_error(&a->archive, error_number, "%s", 3154 error_string.s); 3155 } 3156 archive_string_free(&error_string); 3157 return rc; 3158} 3159 3160/* 3161 * Create the parent directory of the specified path, assuming path 3162 * is already in mutable storage. 3163 */ 3164static int 3165create_parent_dir(struct archive_write_disk *a, char *path) 3166{ 3167 char *slash; 3168 int r; 3169 3170 /* Remove tail element to obtain parent name. */ 3171 slash = strrchr(path, '/'); 3172 if (slash == NULL) 3173 return (ARCHIVE_OK); 3174 *slash = '\0'; 3175 r = create_dir(a, path); 3176 *slash = '/'; 3177 return (r); 3178} 3179 3180/* 3181 * Create the specified dir, recursing to create parents as necessary. 3182 * 3183 * Returns ARCHIVE_OK if the path exists when we're done here. 3184 * Otherwise, returns ARCHIVE_FAILED. 3185 * Assumes path is in mutable storage; path is unchanged on exit. 3186 */ 3187static int 3188create_dir(struct archive_write_disk *a, char *path) 3189{ 3190 struct stat st; 3191 struct fixup_entry *le; 3192 char *slash, *base; 3193 mode_t mode_final, mode; 3194 int r; 3195 3196 /* Check for special names and just skip them. */ 3197 slash = strrchr(path, '/'); 3198 if (slash == NULL) 3199 base = path; 3200 else 3201 base = slash + 1; 3202 3203 if (base[0] == '\0' || 3204 (base[0] == '.' && base[1] == '\0') || 3205 (base[0] == '.' && base[1] == '.' && base[2] == '\0')) { 3206 /* Don't bother trying to create null path, '.', or '..'. */ 3207 if (slash != NULL) { 3208 *slash = '\0'; 3209 r = create_dir(a, path); 3210 *slash = '/'; 3211 return (r); 3212 } 3213 return (ARCHIVE_OK); 3214 } 3215 3216 /* 3217 * Yes, this should be stat() and not lstat(). Using lstat() 3218 * here loses the ability to extract through symlinks. Also note 3219 * that this should not use the a->st cache. 3220 */ 3221 if (la_stat(path, &st) == 0) { 3222 if (S_ISDIR(st.st_mode)) 3223 return (ARCHIVE_OK); 3224 if ((a->flags & ARCHIVE_EXTRACT_NO_OVERWRITE)) { 3225 archive_set_error(&a->archive, EEXIST, 3226 "Can't create directory '%s'", path); 3227 return (ARCHIVE_FAILED); 3228 } 3229 if (unlink(path) != 0) { 3230 archive_set_error(&a->archive, errno, 3231 "Can't create directory '%s': " 3232 "Conflicting file cannot be removed", 3233 path); 3234 return (ARCHIVE_FAILED); 3235 } 3236 } else if (errno != ENOENT && errno != ENOTDIR) { 3237 /* Stat failed? */ 3238 archive_set_error(&a->archive, errno, 3239 "Can't test directory '%s'", path); 3240 return (ARCHIVE_FAILED); 3241 } else if (slash != NULL) { 3242 *slash = '\0'; 3243 r = create_dir(a, path); 3244 *slash = '/'; 3245 if (r != ARCHIVE_OK) 3246 return (r); 3247 } 3248 3249 /* 3250 * Mode we want for the final restored directory. Per POSIX, 3251 * implicitly-created dirs must be created obeying the umask. 3252 * There's no mention whether this is different for privileged 3253 * restores (which the rest of this code handles by pretending 3254 * umask=0). I've chosen here to always obey the user's umask for 3255 * implicit dirs, even if _EXTRACT_PERM was specified. 3256 */ 3257 mode_final = DEFAULT_DIR_MODE & ~a->user_umask; 3258 /* Mode we want on disk during the restore process. */ 3259 mode = mode_final; 3260 mode |= MINIMUM_DIR_MODE; 3261 mode &= MAXIMUM_DIR_MODE; 3262 if (mkdir(path, mode) == 0) { 3263 if (mode != mode_final) { 3264 le = new_fixup(a, path); 3265 if (le == NULL) 3266 return (ARCHIVE_FATAL); 3267 le->fixup |=TODO_MODE_BASE; 3268 le->mode = mode_final; 3269 } 3270 return (ARCHIVE_OK); 3271 } 3272 3273 /* 3274 * Without the following check, a/b/../b/c/d fails at the 3275 * second visit to 'b', so 'd' can't be created. Note that we 3276 * don't add it to the fixup list here, as it's already been 3277 * added. 3278 */ 3279 if (la_stat(path, &st) == 0 && S_ISDIR(st.st_mode)) 3280 return (ARCHIVE_OK); 3281 3282 archive_set_error(&a->archive, errno, "Failed to create dir '%s'", 3283 path); 3284 return (ARCHIVE_FAILED); 3285} 3286 3287/* 3288 * Note: Although we can skip setting the user id if the desired user 3289 * id matches the current user, we cannot skip setting the group, as 3290 * many systems set the gid based on the containing directory. So 3291 * we have to perform a chown syscall if we want to set the SGID 3292 * bit. (The alternative is to stat() and then possibly chown(); it's 3293 * more efficient to skip the stat() and just always chown().) Note 3294 * that a successful chown() here clears the TODO_SGID_CHECK bit, which 3295 * allows set_mode to skip the stat() check for the GID. 3296 */ 3297static int 3298set_ownership(struct archive_write_disk *a) 3299{ 3300#if !defined(__CYGWIN__) && !defined(__linux__) 3301/* 3302 * On Linux, a process may have the CAP_CHOWN capability. 3303 * On Windows there is no 'root' user with uid 0. 3304 * Elsewhere we can skip calling chown if we are not root and the desired 3305 * user id does not match the current user. 3306 */ 3307 if (a->user_uid != 0 && a->user_uid != a->uid) { 3308 archive_set_error(&a->archive, errno, 3309 "Can't set UID=%jd", (intmax_t)a->uid); 3310 return (ARCHIVE_WARN); 3311 } 3312#endif 3313 3314#ifdef HAVE_FCHOWN 3315 /* If we have an fd, we can avoid a race. */ 3316 if (a->fd >= 0 && fchown(a->fd, a->uid, a->gid) == 0) { 3317 /* We've set owner and know uid/gid are correct. */ 3318 a->todo &= ~(TODO_OWNER | TODO_SGID_CHECK | TODO_SUID_CHECK); 3319 return (ARCHIVE_OK); 3320 } 3321#endif 3322 3323 /* We prefer lchown() but will use chown() if that's all we have. */ 3324 /* Of course, if we have neither, this will always fail. */ 3325#ifdef HAVE_LCHOWN 3326 if (lchown(a->name, a->uid, a->gid) == 0) { 3327 /* We've set owner and know uid/gid are correct. */ 3328 a->todo &= ~(TODO_OWNER | TODO_SGID_CHECK | TODO_SUID_CHECK); 3329 return (ARCHIVE_OK); 3330 } 3331#elif HAVE_CHOWN 3332 if (!S_ISLNK(a->mode) && chown(a->name, a->uid, a->gid) == 0) { 3333 /* We've set owner and know uid/gid are correct. */ 3334 a->todo &= ~(TODO_OWNER | TODO_SGID_CHECK | TODO_SUID_CHECK); 3335 return (ARCHIVE_OK); 3336 } 3337#endif 3338 3339 archive_set_error(&a->archive, errno, 3340 "Can't set user=%jd/group=%jd for %s", 3341 (intmax_t)a->uid, (intmax_t)a->gid, a->name); 3342 return (ARCHIVE_WARN); 3343} 3344 3345/* 3346 * Note: Returns 0 on success, non-zero on failure. 3347 */ 3348static int 3349set_time(int fd, int mode, const char *name, 3350 time_t atime, long atime_nsec, 3351 time_t mtime, long mtime_nsec) 3352{ 3353 /* Select the best implementation for this platform. */ 3354#if defined(HAVE_UTIMENSAT) && defined(HAVE_FUTIMENS) 3355 /* 3356 * utimensat() and futimens() are defined in 3357 * POSIX.1-2008. They support ns resolution and setting times 3358 * on fds and symlinks. 3359 */ 3360 struct timespec ts[2]; 3361 (void)mode; /* UNUSED */ 3362 ts[0].tv_sec = atime; 3363 ts[0].tv_nsec = atime_nsec; 3364 ts[1].tv_sec = mtime; 3365 ts[1].tv_nsec = mtime_nsec; 3366 if (fd >= 0) 3367 return futimens(fd, ts); 3368 return utimensat(AT_FDCWD, name, ts, AT_SYMLINK_NOFOLLOW); 3369 3370#elif HAVE_UTIMES 3371 /* 3372 * The utimes()-family functions support ��s-resolution and 3373 * setting times fds and symlinks. utimes() is documented as 3374 * LEGACY by POSIX, futimes() and lutimes() are not described 3375 * in POSIX. 3376 */ 3377 struct timeval times[2]; 3378 3379 times[0].tv_sec = atime; 3380 times[0].tv_usec = atime_nsec / 1000; 3381 times[1].tv_sec = mtime; 3382 times[1].tv_usec = mtime_nsec / 1000; 3383 3384#ifdef HAVE_FUTIMES 3385 if (fd >= 0) 3386 return (futimes(fd, times)); 3387#else 3388 (void)fd; /* UNUSED */ 3389#endif 3390#ifdef HAVE_LUTIMES 3391 (void)mode; /* UNUSED */ 3392 return (lutimes(name, times)); 3393#else 3394 if (S_ISLNK(mode)) 3395 return (0); 3396 return (utimes(name, times)); 3397#endif 3398 3399#elif defined(HAVE_UTIME) 3400 /* 3401 * utime() is POSIX-standard but only supports 1s resolution and 3402 * does not support fds or symlinks. 3403 */ 3404 struct utimbuf times; 3405 (void)fd; /* UNUSED */ 3406 (void)name; /* UNUSED */ 3407 (void)atime_nsec; /* UNUSED */ 3408 (void)mtime_nsec; /* UNUSED */ 3409 times.actime = atime; 3410 times.modtime = mtime; 3411 if (S_ISLNK(mode)) 3412 return (ARCHIVE_OK); 3413 return (utime(name, ×)); 3414 3415#else 3416 /* 3417 * We don't know how to set the time on this platform. 3418 */ 3419 (void)fd; /* UNUSED */ 3420 (void)mode; /* UNUSED */ 3421 (void)name; /* UNUSED */ 3422 (void)atime_nsec; /* UNUSED */ 3423 (void)mtime_nsec; /* UNUSED */ 3424 return (ARCHIVE_WARN); 3425#endif 3426} 3427 3428#ifdef F_SETTIMES 3429static int 3430set_time_tru64(int fd, int mode, const char *name, 3431 time_t atime, long atime_nsec, 3432 time_t mtime, long mtime_nsec, 3433 time_t ctime, long ctime_nsec) 3434{ 3435 struct attr_timbuf tstamp; 3436 tstamp.atime.tv_sec = atime; 3437 tstamp.mtime.tv_sec = mtime; 3438 tstamp.ctime.tv_sec = ctime; 3439#if defined (__hpux) && defined (__ia64) 3440 tstamp.atime.tv_nsec = atime_nsec; 3441 tstamp.mtime.tv_nsec = mtime_nsec; 3442 tstamp.ctime.tv_nsec = ctime_nsec; 3443#else 3444 tstamp.atime.tv_usec = atime_nsec / 1000; 3445 tstamp.mtime.tv_usec = mtime_nsec / 1000; 3446 tstamp.ctime.tv_usec = ctime_nsec / 1000; 3447#endif 3448 return (fcntl(fd,F_SETTIMES,&tstamp)); 3449} 3450#endif /* F_SETTIMES */ 3451 3452static int 3453set_times(struct archive_write_disk *a, 3454 int fd, int mode, const char *name, 3455 time_t atime, long atime_nanos, 3456 time_t birthtime, long birthtime_nanos, 3457 time_t mtime, long mtime_nanos, 3458 time_t cctime, long ctime_nanos) 3459{ 3460 /* Note: set_time doesn't use libarchive return conventions! 3461 * It uses syscall conventions. So 0 here instead of ARCHIVE_OK. */ 3462 int r1 = 0, r2 = 0; 3463 3464#ifdef F_SETTIMES 3465 /* 3466 * on Tru64 try own fcntl first which can restore even the 3467 * ctime, fall back to default code path below if it fails 3468 * or if we are not running as root 3469 */ 3470 if (a->user_uid == 0 && 3471 set_time_tru64(fd, mode, name, 3472 atime, atime_nanos, mtime, 3473 mtime_nanos, cctime, ctime_nanos) == 0) { 3474 return (ARCHIVE_OK); 3475 } 3476#else /* Tru64 */ 3477 (void)cctime; /* UNUSED */ 3478 (void)ctime_nanos; /* UNUSED */ 3479#endif /* Tru64 */ 3480 3481#ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME 3482 /* 3483 * If you have struct stat.st_birthtime, we assume BSD 3484 * birthtime semantics, in which {f,l,}utimes() updates 3485 * birthtime to earliest mtime. So we set the time twice, 3486 * first using the birthtime, then using the mtime. If 3487 * birthtime == mtime, this isn't necessary, so we skip it. 3488 * If birthtime > mtime, then this won't work, so we skip it. 3489 */ 3490 if (birthtime < mtime 3491 || (birthtime == mtime && birthtime_nanos < mtime_nanos)) 3492 r1 = set_time(fd, mode, name, 3493 atime, atime_nanos, 3494 birthtime, birthtime_nanos); 3495#else 3496 (void)birthtime; /* UNUSED */ 3497 (void)birthtime_nanos; /* UNUSED */ 3498#endif 3499 r2 = set_time(fd, mode, name, 3500 atime, atime_nanos, 3501 mtime, mtime_nanos); 3502 if (r1 != 0 || r2 != 0) { 3503 archive_set_error(&a->archive, errno, 3504 "Can't restore time"); 3505 return (ARCHIVE_WARN); 3506 } 3507 return (ARCHIVE_OK); 3508} 3509 3510static int 3511set_times_from_entry(struct archive_write_disk *a) 3512{ 3513 time_t atime, birthtime, mtime, cctime; 3514 long atime_nsec, birthtime_nsec, mtime_nsec, ctime_nsec; 3515 3516 /* Suitable defaults. */ 3517 atime = birthtime = mtime = cctime = a->start_time; 3518 atime_nsec = birthtime_nsec = mtime_nsec = ctime_nsec = 0; 3519 3520 /* If no time was provided, we're done. */ 3521 if (!archive_entry_atime_is_set(a->entry) 3522#if HAVE_STRUCT_STAT_ST_BIRTHTIME 3523 && !archive_entry_birthtime_is_set(a->entry) 3524#endif 3525 && !archive_entry_mtime_is_set(a->entry)) 3526 return (ARCHIVE_OK); 3527 3528 if (archive_entry_atime_is_set(a->entry)) { 3529 atime = archive_entry_atime(a->entry); 3530 atime_nsec = archive_entry_atime_nsec(a->entry); 3531 } 3532 if (archive_entry_birthtime_is_set(a->entry)) { 3533 birthtime = archive_entry_birthtime(a->entry); 3534 birthtime_nsec = archive_entry_birthtime_nsec(a->entry); 3535 } 3536 if (archive_entry_mtime_is_set(a->entry)) { 3537 mtime = archive_entry_mtime(a->entry); 3538 mtime_nsec = archive_entry_mtime_nsec(a->entry); 3539 } 3540 if (archive_entry_ctime_is_set(a->entry)) { 3541 cctime = archive_entry_ctime(a->entry); 3542 ctime_nsec = archive_entry_ctime_nsec(a->entry); 3543 } 3544 3545 return set_times(a, a->fd, a->mode, a->name, 3546 atime, atime_nsec, 3547 birthtime, birthtime_nsec, 3548 mtime, mtime_nsec, 3549 cctime, ctime_nsec); 3550} 3551 3552static int 3553set_mode(struct archive_write_disk *a, int mode) 3554{ 3555 int r = ARCHIVE_OK; 3556 int r2; 3557 mode &= 07777; /* Strip off file type bits. */ 3558 3559 if (a->todo & TODO_SGID_CHECK) { 3560 /* 3561 * If we don't know the GID is right, we must stat() 3562 * to verify it. We can't just check the GID of this 3563 * process, since systems sometimes set GID from 3564 * the enclosing dir or based on ACLs. 3565 */ 3566 if ((r = lazy_stat(a)) != ARCHIVE_OK) 3567 return (r); 3568 if (a->pst->st_gid != a->gid) { 3569 mode &= ~ S_ISGID; 3570 if (a->flags & ARCHIVE_EXTRACT_OWNER) { 3571 /* 3572 * This is only an error if you 3573 * requested owner restore. If you 3574 * didn't, we'll try to restore 3575 * sgid/suid, but won't consider it a 3576 * problem if we can't. 3577 */ 3578 archive_set_error(&a->archive, -1, 3579 "Can't restore SGID bit"); 3580 r = ARCHIVE_WARN; 3581 } 3582 } 3583 /* While we're here, double-check the UID. */ 3584 if (a->pst->st_uid != a->uid 3585 && (a->todo & TODO_SUID)) { 3586 mode &= ~ S_ISUID; 3587 if (a->flags & ARCHIVE_EXTRACT_OWNER) { 3588 archive_set_error(&a->archive, -1, 3589 "Can't restore SUID bit"); 3590 r = ARCHIVE_WARN; 3591 } 3592 } 3593 a->todo &= ~TODO_SGID_CHECK; 3594 a->todo &= ~TODO_SUID_CHECK; 3595 } else if (a->todo & TODO_SUID_CHECK) { 3596 /* 3597 * If we don't know the UID is right, we can just check 3598 * the user, since all systems set the file UID from 3599 * the process UID. 3600 */ 3601 if (a->user_uid != a->uid) { 3602 mode &= ~ S_ISUID; 3603 if (a->flags & ARCHIVE_EXTRACT_OWNER) { 3604 archive_set_error(&a->archive, -1, 3605 "Can't make file SUID"); 3606 r = ARCHIVE_WARN; 3607 } 3608 } 3609 a->todo &= ~TODO_SUID_CHECK; 3610 } 3611 3612 if (S_ISLNK(a->mode)) { 3613#ifdef HAVE_LCHMOD 3614 /* 3615 * If this is a symlink, use lchmod(). If the 3616 * platform doesn't support lchmod(), just skip it. A 3617 * platform that doesn't provide a way to set 3618 * permissions on symlinks probably ignores 3619 * permissions on symlinks, so a failure here has no 3620 * impact. 3621 */ 3622 if (lchmod(a->name, mode) != 0) { 3623 switch (errno) { 3624 case ENOTSUP: 3625 case ENOSYS: 3626#if ENOTSUP != EOPNOTSUPP 3627 case EOPNOTSUPP: 3628#endif 3629 /* 3630 * if lchmod is defined but the platform 3631 * doesn't support it, silently ignore 3632 * error 3633 */ 3634 break; 3635 default: 3636 archive_set_error(&a->archive, errno, 3637 "Can't set permissions to 0%o", (int)mode); 3638 r = ARCHIVE_WARN; 3639 } 3640 } 3641#endif 3642 } else if (!S_ISDIR(a->mode)) { 3643 /* 3644 * If it's not a symlink and not a dir, then use 3645 * fchmod() or chmod(), depending on whether we have 3646 * an fd. Dirs get their perms set during the 3647 * post-extract fixup, which is handled elsewhere. 3648 */ 3649#ifdef HAVE_FCHMOD 3650 if (a->fd >= 0) 3651 r2 = fchmod(a->fd, mode); 3652 else 3653#endif 3654 /* If this platform lacks fchmod(), then 3655 * we'll just use chmod(). */ 3656 r2 = chmod(a->name, mode); 3657 3658 if (r2 != 0) { 3659 archive_set_error(&a->archive, errno, 3660 "Can't set permissions to 0%o", (int)mode); 3661 r = ARCHIVE_WARN; 3662 } 3663 } 3664 return (r); 3665} 3666 3667static int 3668set_fflags(struct archive_write_disk *a) 3669{ 3670 struct fixup_entry *le; 3671 unsigned long set, clear; 3672 int r; 3673 mode_t mode = archive_entry_mode(a->entry); 3674 /* 3675 * Make 'critical_flags' hold all file flags that can't be 3676 * immediately restored. For example, on BSD systems, 3677 * SF_IMMUTABLE prevents hardlinks from being created, so 3678 * should not be set until after any hardlinks are created. To 3679 * preserve some semblance of portability, this uses #ifdef 3680 * extensively. Ugly, but it works. 3681 * 3682 * Yes, Virginia, this does create a security race. It's mitigated 3683 * somewhat by the practice of creating dirs 0700 until the extract 3684 * is done, but it would be nice if we could do more than that. 3685 * People restoring critical file systems should be wary of 3686 * other programs that might try to muck with files as they're 3687 * being restored. 3688 */ 3689 const int critical_flags = 0 3690#ifdef SF_IMMUTABLE 3691 | SF_IMMUTABLE 3692#endif 3693#ifdef UF_IMMUTABLE 3694 | UF_IMMUTABLE 3695#endif 3696#ifdef SF_APPEND 3697 | SF_APPEND 3698#endif 3699#ifdef UF_APPEND 3700 | UF_APPEND 3701#endif 3702#if defined(FS_APPEND_FL) 3703 | FS_APPEND_FL 3704#elif defined(EXT2_APPEND_FL) 3705 | EXT2_APPEND_FL 3706#endif 3707#if defined(FS_IMMUTABLE_FL) 3708 | FS_IMMUTABLE_FL 3709#elif defined(EXT2_IMMUTABLE_FL) 3710 | EXT2_IMMUTABLE_FL 3711#endif 3712#ifdef FS_JOURNAL_DATA_FL 3713 | FS_JOURNAL_DATA_FL 3714#endif 3715 ; 3716 3717 if (a->todo & TODO_FFLAGS) { 3718 archive_entry_fflags(a->entry, &set, &clear); 3719 3720 /* 3721 * The first test encourages the compiler to eliminate 3722 * all of this if it's not necessary. 3723 */ 3724 if ((critical_flags != 0) && (set & critical_flags)) { 3725 le = current_fixup(a, a->name); 3726 if (le == NULL) 3727 return (ARCHIVE_FATAL); 3728 le->fixup |= TODO_FFLAGS; 3729 le->fflags_set = set; 3730 /* Store the mode if it's not already there. */ 3731 if ((le->fixup & TODO_MODE) == 0) 3732 le->mode = mode; 3733 } else { 3734 r = set_fflags_platform(a, a->fd, 3735 a->name, mode, set, clear); 3736 if (r != ARCHIVE_OK) 3737 return (r); 3738 } 3739 } 3740 return (ARCHIVE_OK); 3741} 3742 3743static int 3744clear_nochange_fflags(struct archive_write_disk *a) 3745{ 3746 mode_t mode = archive_entry_mode(a->entry); 3747 const int nochange_flags = 0 3748#ifdef SF_IMMUTABLE 3749 | SF_IMMUTABLE 3750#endif 3751#ifdef UF_IMMUTABLE 3752 | UF_IMMUTABLE 3753#endif 3754#ifdef SF_APPEND 3755 | SF_APPEND 3756#endif 3757#ifdef UF_APPEND 3758 | UF_APPEND 3759#endif 3760#ifdef EXT2_APPEND_FL 3761 | EXT2_APPEND_FL 3762#endif 3763#ifdef EXT2_IMMUTABLE_FL 3764 | EXT2_IMMUTABLE_FL 3765#endif 3766 ; 3767 3768 return (set_fflags_platform(a, a->fd, a->name, mode, 0, 3769 nochange_flags)); 3770} 3771 3772 3773#if ( defined(HAVE_LCHFLAGS) || defined(HAVE_CHFLAGS) || defined(HAVE_FCHFLAGS) ) && defined(HAVE_STRUCT_STAT_ST_FLAGS) 3774/* 3775 * BSD reads flags using stat() and sets them with one of {f,l,}chflags() 3776 */ 3777static int 3778set_fflags_platform(struct archive_write_disk *a, int fd, const char *name, 3779 mode_t mode, unsigned long set, unsigned long clear) 3780{ 3781 int r; 3782 const int sf_mask = 0 3783#ifdef SF_APPEND 3784 | SF_APPEND 3785#endif 3786#ifdef SF_ARCHIVED 3787 | SF_ARCHIVED 3788#endif 3789#ifdef SF_IMMUTABLE 3790 | SF_IMMUTABLE 3791#endif 3792#ifdef SF_NOUNLINK 3793 | SF_NOUNLINK 3794#endif 3795 ; 3796 (void)mode; /* UNUSED */ 3797 3798 if (set == 0 && clear == 0) 3799 return (ARCHIVE_OK); 3800 3801 /* 3802 * XXX Is the stat here really necessary? Or can I just use 3803 * the 'set' flags directly? In particular, I'm not sure 3804 * about the correct approach if we're overwriting an existing 3805 * file that already has flags on it. XXX 3806 */ 3807 if ((r = lazy_stat(a)) != ARCHIVE_OK) 3808 return (r); 3809 3810 a->st.st_flags &= ~clear; 3811 a->st.st_flags |= set; 3812 3813 /* Only super-user may change SF_* flags */ 3814 3815 if (a->user_uid != 0) 3816 a->st.st_flags &= ~sf_mask; 3817 3818#ifdef HAVE_FCHFLAGS 3819 /* If platform has fchflags() and we were given an fd, use it. */ 3820 if (fd >= 0 && fchflags(fd, a->st.st_flags) == 0) 3821 return (ARCHIVE_OK); 3822#endif 3823 /* 3824 * If we can't use the fd to set the flags, we'll use the 3825 * pathname to set flags. We prefer lchflags() but will use 3826 * chflags() if we must. 3827 */ 3828#ifdef HAVE_LCHFLAGS 3829 if (lchflags(name, a->st.st_flags) == 0) 3830 return (ARCHIVE_OK); 3831#elif defined(HAVE_CHFLAGS) 3832 if (S_ISLNK(a->st.st_mode)) { 3833 archive_set_error(&a->archive, errno, 3834 "Can't set file flags on symlink."); 3835 return (ARCHIVE_WARN); 3836 } 3837 if (chflags(name, a->st.st_flags) == 0) 3838 return (ARCHIVE_OK); 3839#endif 3840 archive_set_error(&a->archive, errno, 3841 "Failed to set file flags"); 3842 return (ARCHIVE_WARN); 3843} 3844 3845#elif (defined(FS_IOC_GETFLAGS) && defined(FS_IOC_SETFLAGS) && \ 3846 defined(HAVE_WORKING_FS_IOC_GETFLAGS)) || \ 3847 (defined(EXT2_IOC_GETFLAGS) && defined(EXT2_IOC_SETFLAGS) && \ 3848 defined(HAVE_WORKING_EXT2_IOC_GETFLAGS)) 3849/* 3850 * Linux uses ioctl() to read and write file flags. 3851 */ 3852static int 3853set_fflags_platform(struct archive_write_disk *a, int fd, const char *name, 3854 mode_t mode, unsigned long set, unsigned long clear) 3855{ 3856 int ret; 3857 int myfd = fd; 3858 int newflags, oldflags; 3859 /* 3860 * Linux has no define for the flags that are only settable by 3861 * the root user. This code may seem a little complex, but 3862 * there seem to be some Linux systems that lack these 3863 * defines. (?) The code below degrades reasonably gracefully 3864 * if sf_mask is incomplete. 3865 */ 3866 const int sf_mask = 0 3867#if defined(FS_IMMUTABLE_FL) 3868 | FS_IMMUTABLE_FL 3869#elif defined(EXT2_IMMUTABLE_FL) 3870 | EXT2_IMMUTABLE_FL 3871#endif 3872#if defined(FS_APPEND_FL) 3873 | FS_APPEND_FL 3874#elif defined(EXT2_APPEND_FL) 3875 | EXT2_APPEND_FL 3876#endif 3877#if defined(FS_JOURNAL_DATA_FL) 3878 | FS_JOURNAL_DATA_FL 3879#endif 3880 ; 3881 3882 if (set == 0 && clear == 0) 3883 return (ARCHIVE_OK); 3884 /* Only regular files and dirs can have flags. */ 3885 if (!S_ISREG(mode) && !S_ISDIR(mode)) 3886 return (ARCHIVE_OK); 3887 3888 /* If we weren't given an fd, open it ourselves. */ 3889 if (myfd < 0) { 3890 myfd = open(name, O_RDONLY | O_NONBLOCK | O_BINARY | O_CLOEXEC); 3891 __archive_ensure_cloexec_flag(myfd); 3892 } 3893 if (myfd < 0) 3894 return (ARCHIVE_OK); 3895 3896 /* 3897 * XXX As above, this would be way simpler if we didn't have 3898 * to read the current flags from disk. XXX 3899 */ 3900 ret = ARCHIVE_OK; 3901 3902 /* Read the current file flags. */ 3903 if (ioctl(myfd, 3904#ifdef FS_IOC_GETFLAGS 3905 FS_IOC_GETFLAGS, 3906#else 3907 EXT2_IOC_GETFLAGS, 3908#endif 3909 &oldflags) < 0) 3910 goto fail; 3911 3912 /* Try setting the flags as given. */ 3913 newflags = (oldflags & ~clear) | set; 3914 if (ioctl(myfd, 3915#ifdef FS_IOC_SETFLAGS 3916 FS_IOC_SETFLAGS, 3917#else 3918 EXT2_IOC_SETFLAGS, 3919#endif 3920 &newflags) >= 0) 3921 goto cleanup; 3922 if (errno != EPERM) 3923 goto fail; 3924 3925 /* If we couldn't set all the flags, try again with a subset. */ 3926 newflags &= ~sf_mask; 3927 oldflags &= sf_mask; 3928 newflags |= oldflags; 3929 if (ioctl(myfd, 3930#ifdef FS_IOC_SETFLAGS 3931 FS_IOC_SETFLAGS, 3932#else 3933 EXT2_IOC_SETFLAGS, 3934#endif 3935 &newflags) >= 0) 3936 goto cleanup; 3937 3938 /* We couldn't set the flags, so report the failure. */ 3939fail: 3940 archive_set_error(&a->archive, errno, 3941 "Failed to set file flags"); 3942 ret = ARCHIVE_WARN; 3943cleanup: 3944 if (fd < 0) 3945 close(myfd); 3946 return (ret); 3947} 3948 3949#else 3950 3951/* 3952 * Of course, some systems have neither BSD chflags() nor Linux' flags 3953 * support through ioctl(). 3954 */ 3955static int 3956set_fflags_platform(struct archive_write_disk *a, int fd, const char *name, 3957 mode_t mode, unsigned long set, unsigned long clear) 3958{ 3959 (void)a; /* UNUSED */ 3960 (void)fd; /* UNUSED */ 3961 (void)name; /* UNUSED */ 3962 (void)mode; /* UNUSED */ 3963 (void)set; /* UNUSED */ 3964 (void)clear; /* UNUSED */ 3965 return (ARCHIVE_OK); 3966} 3967 3968#endif /* __linux */ 3969 3970#ifndef HAVE_COPYFILE_H 3971/* Default is to simply drop Mac extended metadata. */ 3972static int 3973set_mac_metadata(struct archive_write_disk *a, const char *pathname, 3974 const void *metadata, size_t metadata_size) 3975{ 3976 (void)a; /* UNUSED */ 3977 (void)pathname; /* UNUSED */ 3978 (void)metadata; /* UNUSED */ 3979 (void)metadata_size; /* UNUSED */ 3980 return (ARCHIVE_OK); 3981} 3982 3983static int 3984fixup_appledouble(struct archive_write_disk *a, const char *pathname) 3985{ 3986 (void)a; /* UNUSED */ 3987 (void)pathname; /* UNUSED */ 3988 return (ARCHIVE_OK); 3989} 3990#else 3991 3992/* 3993 * On Mac OS, we use copyfile() to unpack the metadata and 3994 * apply it to the target file. 3995 */ 3996 3997#if defined(HAVE_SYS_XATTR_H) 3998static int 3999copy_xattrs(struct archive_write_disk *a, int tmpfd, int dffd) 4000{ 4001 ssize_t xattr_size; 4002 char *xattr_names = NULL, *xattr_val = NULL; 4003 int ret = ARCHIVE_OK, xattr_i; 4004 4005 xattr_size = flistxattr(tmpfd, NULL, 0, 0); 4006 if (xattr_size == -1) { 4007 archive_set_error(&a->archive, errno, 4008 "Failed to read metadata(xattr)"); 4009 ret = ARCHIVE_WARN; 4010 goto exit_xattr; 4011 } 4012 xattr_names = malloc(xattr_size); 4013 if (xattr_names == NULL) { 4014 archive_set_error(&a->archive, ENOMEM, 4015 "Can't allocate memory for metadata(xattr)"); 4016 ret = ARCHIVE_FATAL; 4017 goto exit_xattr; 4018 } 4019 xattr_size = flistxattr(tmpfd, xattr_names, xattr_size, 0); 4020 if (xattr_size == -1) { 4021 archive_set_error(&a->archive, errno, 4022 "Failed to read metadata(xattr)"); 4023 ret = ARCHIVE_WARN; 4024 goto exit_xattr; 4025 } 4026 for (xattr_i = 0; xattr_i < xattr_size; 4027 xattr_i += strlen(xattr_names + xattr_i) + 1) { 4028 char *xattr_val_saved; 4029 ssize_t s; 4030 int f; 4031 4032 s = fgetxattr(tmpfd, xattr_names + xattr_i, NULL, 0, 0, 0); 4033 if (s == -1) { 4034 archive_set_error(&a->archive, errno, 4035 "Failed to get metadata(xattr)"); 4036 ret = ARCHIVE_WARN; 4037 goto exit_xattr; 4038 } 4039 xattr_val_saved = xattr_val; 4040 xattr_val = realloc(xattr_val, s); 4041 if (xattr_val == NULL) { 4042 archive_set_error(&a->archive, ENOMEM, 4043 "Failed to get metadata(xattr)"); 4044 ret = ARCHIVE_WARN; 4045 free(xattr_val_saved); 4046 goto exit_xattr; 4047 } 4048 s = fgetxattr(tmpfd, xattr_names + xattr_i, xattr_val, s, 0, 0); 4049 if (s == -1) { 4050 archive_set_error(&a->archive, errno, 4051 "Failed to get metadata(xattr)"); 4052 ret = ARCHIVE_WARN; 4053 goto exit_xattr; 4054 } 4055 f = fsetxattr(dffd, xattr_names + xattr_i, xattr_val, s, 0, 0); 4056 if (f == -1) { 4057 archive_set_error(&a->archive, errno, 4058 "Failed to get metadata(xattr)"); 4059 ret = ARCHIVE_WARN; 4060 goto exit_xattr; 4061 } 4062 } 4063exit_xattr: 4064 free(xattr_names); 4065 free(xattr_val); 4066 return (ret); 4067} 4068#endif 4069 4070static int 4071copy_acls(struct archive_write_disk *a, int tmpfd, int dffd) 4072{ 4073#ifndef HAVE_SYS_ACL_H 4074 return 0; 4075#else 4076 acl_t acl, dfacl = NULL; 4077 int acl_r, ret = ARCHIVE_OK; 4078 4079 acl = acl_get_fd(tmpfd); 4080 if (acl == NULL) { 4081 if (errno == ENOENT) 4082 /* There are not any ACLs. */ 4083 return (ret); 4084 archive_set_error(&a->archive, errno, 4085 "Failed to get metadata(acl)"); 4086 ret = ARCHIVE_WARN; 4087 goto exit_acl; 4088 } 4089 dfacl = acl_dup(acl); 4090 acl_r = acl_set_fd(dffd, dfacl); 4091 if (acl_r == -1) { 4092 archive_set_error(&a->archive, errno, 4093 "Failed to get metadata(acl)"); 4094 ret = ARCHIVE_WARN; 4095 goto exit_acl; 4096 } 4097exit_acl: 4098 if (acl) 4099 acl_free(acl); 4100 if (dfacl) 4101 acl_free(dfacl); 4102 return (ret); 4103#endif 4104} 4105 4106static int 4107create_tempdatafork(struct archive_write_disk *a, const char *pathname) 4108{ 4109 struct archive_string tmpdatafork; 4110 int tmpfd; 4111 4112 archive_string_init(&tmpdatafork); 4113 archive_strcpy(&tmpdatafork, "tar.md.XXXXXX"); 4114 tmpfd = mkstemp(tmpdatafork.s); 4115 if (tmpfd < 0) { 4116 archive_set_error(&a->archive, errno, 4117 "Failed to mkstemp"); 4118 archive_string_free(&tmpdatafork); 4119 return (-1); 4120 } 4121 if (copyfile(pathname, tmpdatafork.s, 0, 4122 COPYFILE_UNPACK | COPYFILE_NOFOLLOW 4123 | COPYFILE_ACL | COPYFILE_XATTR) < 0) { 4124 archive_set_error(&a->archive, errno, 4125 "Failed to restore metadata"); 4126 close(tmpfd); 4127 tmpfd = -1; 4128 } 4129 unlink(tmpdatafork.s); 4130 archive_string_free(&tmpdatafork); 4131 return (tmpfd); 4132} 4133 4134static int 4135copy_metadata(struct archive_write_disk *a, const char *metadata, 4136 const char *datafork, int datafork_compressed) 4137{ 4138 int ret = ARCHIVE_OK; 4139 4140 if (datafork_compressed) { 4141 int dffd, tmpfd; 4142 4143 tmpfd = create_tempdatafork(a, metadata); 4144 if (tmpfd == -1) 4145 return (ARCHIVE_WARN); 4146 4147 /* 4148 * Do not open the data fork compressed by HFS+ compression 4149 * with at least a writing mode(O_RDWR or O_WRONLY). it 4150 * makes the data fork uncompressed. 4151 */ 4152 dffd = open(datafork, 0); 4153 if (dffd == -1) { 4154 archive_set_error(&a->archive, errno, 4155 "Failed to open the data fork for metadata"); 4156 close(tmpfd); 4157 return (ARCHIVE_WARN); 4158 } 4159 4160#if defined(HAVE_SYS_XATTR_H) 4161 ret = copy_xattrs(a, tmpfd, dffd); 4162 if (ret == ARCHIVE_OK) 4163#endif 4164 ret = copy_acls(a, tmpfd, dffd); 4165 close(tmpfd); 4166 close(dffd); 4167 } else { 4168 if (copyfile(metadata, datafork, 0, 4169 COPYFILE_UNPACK | COPYFILE_NOFOLLOW 4170 | COPYFILE_ACL | COPYFILE_XATTR) < 0) { 4171 archive_set_error(&a->archive, errno, 4172 "Failed to restore metadata"); 4173 ret = ARCHIVE_WARN; 4174 } 4175 } 4176 return (ret); 4177} 4178 4179static int 4180set_mac_metadata(struct archive_write_disk *a, const char *pathname, 4181 const void *metadata, size_t metadata_size) 4182{ 4183 struct archive_string tmp; 4184 ssize_t written; 4185 int fd; 4186 int ret = ARCHIVE_OK; 4187 4188 /* This would be simpler if copyfile() could just accept the 4189 * metadata as a block of memory; then we could sidestep this 4190 * silly dance of writing the data to disk just so that 4191 * copyfile() can read it back in again. */ 4192 archive_string_init(&tmp); 4193 archive_strcpy(&tmp, pathname); 4194 archive_strcat(&tmp, ".XXXXXX"); 4195 fd = mkstemp(tmp.s); 4196 4197 if (fd < 0) { 4198 archive_set_error(&a->archive, errno, 4199 "Failed to restore metadata"); 4200 archive_string_free(&tmp); 4201 return (ARCHIVE_WARN); 4202 } 4203 written = write(fd, metadata, metadata_size); 4204 close(fd); 4205 if ((size_t)written != metadata_size) { 4206 archive_set_error(&a->archive, errno, 4207 "Failed to restore metadata"); 4208 ret = ARCHIVE_WARN; 4209 } else { 4210 int compressed; 4211 4212#if defined(UF_COMPRESSED) 4213 if ((a->todo & TODO_HFS_COMPRESSION) != 0 && 4214 (ret = lazy_stat(a)) == ARCHIVE_OK) 4215 compressed = a->st.st_flags & UF_COMPRESSED; 4216 else 4217#endif 4218 compressed = 0; 4219 ret = copy_metadata(a, tmp.s, pathname, compressed); 4220 } 4221 unlink(tmp.s); 4222 archive_string_free(&tmp); 4223 return (ret); 4224} 4225 4226static int 4227fixup_appledouble(struct archive_write_disk *a, const char *pathname) 4228{ 4229 char buff[8]; 4230 struct stat st; 4231 const char *p; 4232 struct archive_string datafork; 4233 int fd = -1, ret = ARCHIVE_OK; 4234 4235 archive_string_init(&datafork); 4236 /* Check if the current file name is a type of the resource 4237 * fork file. */ 4238 p = strrchr(pathname, '/'); 4239 if (p == NULL) 4240 p = pathname; 4241 else 4242 p++; 4243 if (p[0] != '.' || p[1] != '_') 4244 goto skip_appledouble; 4245 4246 /* 4247 * Check if the data fork file exists. 4248 * 4249 * TODO: Check if this write disk object has handled it. 4250 */ 4251 archive_strncpy(&datafork, pathname, p - pathname); 4252 archive_strcat(&datafork, p + 2); 4253 if (lstat(datafork.s, &st) == -1 || 4254 (st.st_mode & AE_IFMT) != AE_IFREG) 4255 goto skip_appledouble; 4256 4257 /* 4258 * Check if the file is in the AppleDouble form. 4259 */ 4260 fd = open(pathname, O_RDONLY | O_BINARY | O_CLOEXEC); 4261 __archive_ensure_cloexec_flag(fd); 4262 if (fd == -1) { 4263 archive_set_error(&a->archive, errno, 4264 "Failed to open a restoring file"); 4265 ret = ARCHIVE_WARN; 4266 goto skip_appledouble; 4267 } 4268 if (read(fd, buff, 8) == -1) { 4269 archive_set_error(&a->archive, errno, 4270 "Failed to read a restoring file"); 4271 close(fd); 4272 ret = ARCHIVE_WARN; 4273 goto skip_appledouble; 4274 } 4275 close(fd); 4276 /* Check AppleDouble Magic Code. */ 4277 if (archive_be32dec(buff) != 0x00051607) 4278 goto skip_appledouble; 4279 /* Check AppleDouble Version. */ 4280 if (archive_be32dec(buff+4) != 0x00020000) 4281 goto skip_appledouble; 4282 4283 ret = copy_metadata(a, pathname, datafork.s, 4284#if defined(UF_COMPRESSED) 4285 st.st_flags & UF_COMPRESSED); 4286#else 4287 0); 4288#endif 4289 if (ret == ARCHIVE_OK) { 4290 unlink(pathname); 4291 ret = ARCHIVE_EOF; 4292 } 4293skip_appledouble: 4294 archive_string_free(&datafork); 4295 return (ret); 4296} 4297#endif 4298 4299#if ARCHIVE_XATTR_LINUX || ARCHIVE_XATTR_DARWIN || ARCHIVE_XATTR_AIX 4300/* 4301 * Restore extended attributes - Linux, Darwin and AIX implementations: 4302 * AIX' ea interface is syntaxwise identical to the Linux xattr interface. 4303 */ 4304static int 4305set_xattrs(struct archive_write_disk *a) 4306{ 4307 struct archive_entry *entry = a->entry; 4308 struct archive_string errlist; 4309 int ret = ARCHIVE_OK; 4310 int i = archive_entry_xattr_reset(entry); 4311 short fail = 0; 4312 4313 archive_string_init(&errlist); 4314 4315 while (i--) { 4316 const char *name; 4317 const void *value; 4318 size_t size; 4319 int e; 4320 4321 archive_entry_xattr_next(entry, &name, &value, &size); 4322 4323 if (name == NULL) 4324 continue; 4325#if ARCHIVE_XATTR_LINUX 4326 /* Linux: quietly skip POSIX.1e ACL extended attributes */ 4327 if (strncmp(name, "system.", 7) == 0 && 4328 (strcmp(name + 7, "posix_acl_access") == 0 || 4329 strcmp(name + 7, "posix_acl_default") == 0)) 4330 continue; 4331 if (strncmp(name, "trusted.SGI_", 12) == 0 && 4332 (strcmp(name + 12, "ACL_DEFAULT") == 0 || 4333 strcmp(name + 12, "ACL_FILE") == 0)) 4334 continue; 4335 4336 /* Linux: xfsroot namespace is obsolete and unsupported */ 4337 if (strncmp(name, "xfsroot.", 8) == 0) { 4338 fail = 1; 4339 archive_strcat(&errlist, name); 4340 archive_strappend_char(&errlist, ' '); 4341 continue; 4342 } 4343#endif 4344 4345 if (a->fd >= 0) { 4346#if ARCHIVE_XATTR_LINUX 4347 e = fsetxattr(a->fd, name, value, size, 0); 4348#elif ARCHIVE_XATTR_DARWIN 4349 e = fsetxattr(a->fd, name, value, size, 0, 0); 4350#elif ARCHIVE_XATTR_AIX 4351 e = fsetea(a->fd, name, value, size, 0); 4352#endif 4353 } else { 4354#if ARCHIVE_XATTR_LINUX 4355 e = lsetxattr(archive_entry_pathname(entry), 4356 name, value, size, 0); 4357#elif ARCHIVE_XATTR_DARWIN 4358 e = setxattr(archive_entry_pathname(entry), 4359 name, value, size, 0, XATTR_NOFOLLOW); 4360#elif ARCHIVE_XATTR_AIX 4361 e = lsetea(archive_entry_pathname(entry), 4362 name, value, size, 0); 4363#endif 4364 } 4365 if (e == -1) { 4366 ret = ARCHIVE_WARN; 4367 archive_strcat(&errlist, name); 4368 archive_strappend_char(&errlist, ' '); 4369 if (errno != ENOTSUP && errno != ENOSYS) 4370 fail = 1; 4371 } 4372 } 4373 4374 if (ret == ARCHIVE_WARN) { 4375 if (fail && errlist.length > 0) { 4376 errlist.length--; 4377 errlist.s[errlist.length] = '\0'; 4378 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 4379 "Cannot restore extended attributes: %s", 4380 errlist.s); 4381 } else 4382 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 4383 "Cannot restore extended " 4384 "attributes on this file system."); 4385 } 4386 4387 archive_string_free(&errlist); 4388 return (ret); 4389} 4390#elif ARCHIVE_XATTR_FREEBSD 4391/* 4392 * Restore extended attributes - FreeBSD implementation 4393 */ 4394static int 4395set_xattrs(struct archive_write_disk *a) 4396{ 4397 struct archive_entry *entry = a->entry; 4398 struct archive_string errlist; 4399 int ret = ARCHIVE_OK; 4400 int i = archive_entry_xattr_reset(entry); 4401 short fail = 0; 4402 4403 archive_string_init(&errlist); 4404 4405 while (i--) { 4406 const char *name; 4407 const void *value; 4408 size_t size; 4409 archive_entry_xattr_next(entry, &name, &value, &size); 4410 if (name != NULL) { 4411 ssize_t e; 4412 int namespace; 4413 4414 if (strncmp(name, "user.", 5) == 0) { 4415 /* "user." attributes go to user namespace */ 4416 name += 5; 4417 namespace = EXTATTR_NAMESPACE_USER; 4418 } else { 4419 /* Other namespaces are unsupported */ 4420 archive_strcat(&errlist, name); 4421 archive_strappend_char(&errlist, ' '); 4422 fail = 1; 4423 ret = ARCHIVE_WARN; 4424 continue; 4425 } 4426 4427 if (a->fd >= 0) { 4428 e = extattr_set_fd(a->fd, namespace, name, 4429 value, size); 4430 } else { 4431 e = extattr_set_link( 4432 archive_entry_pathname(entry), namespace, 4433 name, value, size); 4434 } 4435 if (e != (ssize_t)size) { 4436 archive_strcat(&errlist, name); 4437 archive_strappend_char(&errlist, ' '); 4438 ret = ARCHIVE_WARN; 4439 if (errno != ENOTSUP && errno != ENOSYS) 4440 fail = 1; 4441 } 4442 } 4443 } 4444 4445 if (ret == ARCHIVE_WARN) { 4446 if (fail && errlist.length > 0) { 4447 errlist.length--; 4448 errlist.s[errlist.length] = '\0'; 4449 4450 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 4451 "Cannot restore extended attributes: %s", 4452 errlist.s); 4453 } else 4454 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 4455 "Cannot restore extended " 4456 "attributes on this file system."); 4457 } 4458 4459 archive_string_free(&errlist); 4460 return (ret); 4461} 4462#else 4463/* 4464 * Restore extended attributes - stub implementation for unsupported systems 4465 */ 4466static int 4467set_xattrs(struct archive_write_disk *a) 4468{ 4469 static int warning_done = 0; 4470 4471 /* If there aren't any extended attributes, then it's okay not 4472 * to extract them, otherwise, issue a single warning. */ 4473 if (archive_entry_xattr_count(a->entry) != 0 && !warning_done) { 4474 warning_done = 1; 4475 archive_set_error(&a->archive, ARCHIVE_ERRNO_FILE_FORMAT, 4476 "Cannot restore extended attributes on this system"); 4477 return (ARCHIVE_WARN); 4478 } 4479 /* Warning was already emitted; suppress further warnings. */ 4480 return (ARCHIVE_OK); 4481} 4482#endif 4483 4484/* 4485 * Test if file on disk is older than entry. 4486 */ 4487static int 4488older(struct stat *st, struct archive_entry *entry) 4489{ 4490 /* First, test the seconds and return if we have a definite answer. */ 4491 /* Definitely older. */ 4492 if (to_int64_time(st->st_mtime) < to_int64_time(archive_entry_mtime(entry))) 4493 return (1); 4494 /* Definitely younger. */ 4495 if (to_int64_time(st->st_mtime) > to_int64_time(archive_entry_mtime(entry))) 4496 return (0); 4497 /* If this platform supports fractional seconds, try those. */ 4498#if HAVE_STRUCT_STAT_ST_MTIMESPEC_TV_NSEC 4499 /* Definitely older. */ 4500 if (st->st_mtimespec.tv_nsec < archive_entry_mtime_nsec(entry)) 4501 return (1); 4502#elif HAVE_STRUCT_STAT_ST_MTIM_TV_NSEC 4503 /* Definitely older. */ 4504 if (st->st_mtim.tv_nsec < archive_entry_mtime_nsec(entry)) 4505 return (1); 4506#elif HAVE_STRUCT_STAT_ST_MTIME_N 4507 /* older. */ 4508 if (st->st_mtime_n < archive_entry_mtime_nsec(entry)) 4509 return (1); 4510#elif HAVE_STRUCT_STAT_ST_UMTIME 4511 /* older. */ 4512 if (st->st_umtime * 1000 < archive_entry_mtime_nsec(entry)) 4513 return (1); 4514#elif HAVE_STRUCT_STAT_ST_MTIME_USEC 4515 /* older. */ 4516 if (st->st_mtime_usec * 1000 < archive_entry_mtime_nsec(entry)) 4517 return (1); 4518#else 4519 /* This system doesn't have high-res timestamps. */ 4520#endif 4521 /* Same age or newer, so not older. */ 4522 return (0); 4523} 4524 4525#ifndef ARCHIVE_ACL_SUPPORT 4526int 4527archive_write_disk_set_acls(struct archive *a, int fd, const char *name, 4528 struct archive_acl *abstract_acl, __LA_MODE_T mode) 4529{ 4530 (void)a; /* UNUSED */ 4531 (void)fd; /* UNUSED */ 4532 (void)name; /* UNUSED */ 4533 (void)abstract_acl; /* UNUSED */ 4534 (void)mode; /* UNUSED */ 4535 return (ARCHIVE_OK); 4536} 4537#endif 4538 4539#endif /* !_WIN32 || __CYGWIN__ */ 4540 4541