1 /*- 2 * Copyright (c) 2003-2009 Tim Kientzle 3 * Copyright (c) 2010-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 /* This is the tree-walking code for POSIX systems. */ 29 #if !defined(_WIN32) || defined(__CYGWIN__) 30 31 #include "archive_platform.h" 32 __FBSDID("$FreeBSD$"); 33 34 #ifdef HAVE_SYS_PARAM_H 35 #include <sys/param.h> 36 #endif 37 #ifdef HAVE_SYS_MOUNT_H 38 #include <sys/mount.h> 39 #endif 40 #ifdef HAVE_SYS_STAT_H 41 #include <sys/stat.h> 42 #endif 43 #ifdef HAVE_SYS_STATFS_H 44 #include <sys/statfs.h> 45 #endif 46 #ifdef HAVE_SYS_STATVFS_H 47 #include <sys/statvfs.h> 48 #endif 49 #ifdef HAVE_SYS_TIME_H 50 #include <sys/time.h> 51 #endif 52 #ifdef HAVE_LINUX_MAGIC_H 53 #include <linux/magic.h> 54 #endif 55 #ifdef HAVE_LINUX_FS_H 56 #include <linux/fs.h> 57 #endif 58 /* 59 * Some Linux distributions have both linux/ext2_fs.h and ext2fs/ext2_fs.h. 60 * As the include guards don't agree, the order of include is important. 61 */ 62 #ifdef HAVE_LINUX_EXT2_FS_H 63 #include <linux/ext2_fs.h> /* for Linux file flags */ 64 #endif 65 #if defined(HAVE_EXT2FS_EXT2_FS_H) && !defined(__CYGWIN__) 66 #include <ext2fs/ext2_fs.h> /* Linux file flags, broken on Cygwin */ 67 #endif 68 #ifdef HAVE_DIRECT_H 69 #include <direct.h> 70 #endif 71 #ifdef HAVE_DIRENT_H 72 #include <dirent.h> 73 #endif 74 #ifdef HAVE_ERRNO_H 75 #include <errno.h> 76 #endif 77 #ifdef HAVE_FCNTL_H 78 #include <fcntl.h> 79 #endif 80 #ifdef HAVE_LIMITS_H 81 #include <limits.h> 82 #endif 83 #ifdef HAVE_STDLIB_H 84 #include <stdlib.h> 85 #endif 86 #ifdef HAVE_STRING_H 87 #include <string.h> 88 #endif 89 #ifdef HAVE_UNISTD_H 90 #include <unistd.h> 91 #endif 92 #ifdef HAVE_SYS_IOCTL_H 93 #include <sys/ioctl.h> 94 #endif 95 96 #include "archive.h" 97 #include "archive_string.h" 98 #include "archive_entry.h" 99 #include "archive_private.h" 100 #include "archive_read_disk_private.h" 101 102 #ifndef HAVE_FCHDIR 103 #error fchdir function required. 104 #endif 105 #ifndef O_BINARY 106 #define O_BINARY 0 107 #endif 108 #ifndef O_CLOEXEC 109 #define O_CLOEXEC 0 110 #endif 111 112 /*- 113 * This is a new directory-walking system that addresses a number 114 * of problems I've had with fts(3). In particular, it has no 115 * pathname-length limits (other than the size of 'int'), handles 116 * deep logical traversals, uses considerably less memory, and has 117 * an opaque interface (easier to modify in the future). 118 * 119 * Internally, it keeps a single list of "tree_entry" items that 120 * represent filesystem objects that require further attention. 121 * Non-directories are not kept in memory: they are pulled from 122 * readdir(), returned to the client, then freed as soon as possible. 123 * Any directory entry to be traversed gets pushed onto the stack. 124 * 125 * There is surprisingly little information that needs to be kept for 126 * each item on the stack. Just the name, depth (represented here as the 127 * string length of the parent directory's pathname), and some markers 128 * indicating how to get back to the parent (via chdir("..") for a 129 * regular dir or via fchdir(2) for a symlink). 130 */ 131 /* 132 * TODO: 133 * 1) Loop checking. 134 * 3) Arbitrary logical traversals by closing/reopening intermediate fds. 135 */ 136 137 struct restore_time { 138 const char *name; 139 time_t mtime; 140 long mtime_nsec; 141 time_t atime; 142 long atime_nsec; 143 mode_t filetype; 144 int noatime; 145 }; 146 147 struct tree_entry { 148 int depth; 149 struct tree_entry *next; 150 struct tree_entry *parent; 151 struct archive_string name; 152 size_t dirname_length; 153 int64_t dev; 154 int64_t ino; 155 int flags; 156 int filesystem_id; 157 /* How to return back to the parent of a symlink. */ 158 int symlink_parent_fd; 159 /* How to restore time of a directory. */ 160 struct restore_time restore_time; 161 }; 162 163 struct filesystem { 164 int64_t dev; 165 int synthetic; 166 int remote; 167 int noatime; 168 #if defined(USE_READDIR_R) 169 size_t name_max; 170 #endif 171 long incr_xfer_size; 172 long max_xfer_size; 173 long min_xfer_size; 174 long xfer_align; 175 176 /* 177 * Buffer used for reading file contents. 178 */ 179 /* Exactly allocated memory pointer. */ 180 unsigned char *allocation_ptr; 181 /* Pointer adjusted to the filesystem alignment . */ 182 unsigned char *buff; 183 size_t buff_size; 184 }; 185 186 /* Definitions for tree_entry.flags bitmap. */ 187 #define isDir 1 /* This entry is a regular directory. */ 188 #define isDirLink 2 /* This entry is a symbolic link to a directory. */ 189 #define needsFirstVisit 4 /* This is an initial entry. */ 190 #define needsDescent 8 /* This entry needs to be previsited. */ 191 #define needsOpen 16 /* This is a directory that needs to be opened. */ 192 #define needsAscent 32 /* This entry needs to be postvisited. */ 193 194 /* 195 * Local data for this package. 196 */ 197 struct tree { 198 struct tree_entry *stack; 199 struct tree_entry *current; 200 DIR *d; 201 #define INVALID_DIR_HANDLE NULL 202 struct dirent *de; 203 #if defined(USE_READDIR_R) 204 struct dirent *dirent; 205 size_t dirent_allocated; 206 #endif 207 int flags; 208 int visit_type; 209 /* Error code from last failed operation. */ 210 int tree_errno; 211 212 /* Dynamically-sized buffer for holding path */ 213 struct archive_string path; 214 215 /* Last path element */ 216 const char *basename; 217 /* Leading dir length */ 218 size_t dirname_length; 219 220 int depth; 221 int openCount; 222 int maxOpenCount; 223 int initial_dir_fd; 224 int working_dir_fd; 225 226 struct stat lst; 227 struct stat st; 228 int descend; 229 int nlink; 230 /* How to restore time of a file. */ 231 struct restore_time restore_time; 232 233 struct entry_sparse { 234 int64_t length; 235 int64_t offset; 236 } *sparse_list, *current_sparse; 237 int sparse_count; 238 int sparse_list_size; 239 240 char initial_symlink_mode; 241 char symlink_mode; 242 struct filesystem *current_filesystem; 243 struct filesystem *filesystem_table; 244 int initial_filesystem_id; 245 int current_filesystem_id; 246 int max_filesystem_id; 247 int allocated_filesystem; 248 249 int entry_fd; 250 int entry_eof; 251 int64_t entry_remaining_bytes; 252 int64_t entry_total; 253 unsigned char *entry_buff; 254 size_t entry_buff_size; 255 }; 256 257 /* Definitions for tree.flags bitmap. */ 258 #define hasStat 16 /* The st entry is valid. */ 259 #define hasLstat 32 /* The lst entry is valid. */ 260 #define onWorkingDir 64 /* We are on the working dir where we are 261 * reading directory entry at this time. */ 262 #define needsRestoreTimes 128 263 #define onInitialDir 256 /* We are on the initial dir. */ 264 265 static int 266 tree_dir_next_posix(struct tree *t); 267 268 #ifdef HAVE_DIRENT_D_NAMLEN 269 /* BSD extension; avoids need for a strlen() call. */ 270 #define D_NAMELEN(dp) (dp)->d_namlen 271 #else 272 #define D_NAMELEN(dp) (strlen((dp)->d_name)) 273 #endif 274 275 /* Initiate/terminate a tree traversal. */ 276 static struct tree *tree_open(const char *, int, int); 277 static struct tree *tree_reopen(struct tree *, const char *, int); 278 static void tree_close(struct tree *); 279 static void tree_free(struct tree *); 280 static void tree_push(struct tree *, const char *, int, int64_t, int64_t, 281 struct restore_time *); 282 static int tree_enter_initial_dir(struct tree *); 283 static int tree_enter_working_dir(struct tree *); 284 static int tree_current_dir_fd(struct tree *); 285 286 /* 287 * tree_next() returns Zero if there is no next entry, non-zero if 288 * there is. Note that directories are visited three times. 289 * Directories are always visited first as part of enumerating their 290 * parent; that is a "regular" visit. If tree_descend() is invoked at 291 * that time, the directory is added to a work list and will 292 * subsequently be visited two more times: once just after descending 293 * into the directory ("postdescent") and again just after ascending 294 * back to the parent ("postascent"). 295 * 296 * TREE_ERROR_DIR is returned if the descent failed (because the 297 * directory couldn't be opened, for instance). This is returned 298 * instead of TREE_POSTDESCENT/TREE_POSTASCENT. TREE_ERROR_DIR is not a 299 * fatal error, but it does imply that the relevant subtree won't be 300 * visited. TREE_ERROR_FATAL is returned for an error that left the 301 * traversal completely hosed. Right now, this is only returned for 302 * chdir() failures during ascent. 303 */ 304 #define TREE_REGULAR 1 305 #define TREE_POSTDESCENT 2 306 #define TREE_POSTASCENT 3 307 #define TREE_ERROR_DIR -1 308 #define TREE_ERROR_FATAL -2 309 310 static int tree_next(struct tree *); 311 312 /* 313 * Return information about the current entry. 314 */ 315 316 /* 317 * The current full pathname, length of the full pathname, and a name 318 * that can be used to access the file. Because tree does use chdir 319 * extensively, the access path is almost never the same as the full 320 * current path. 321 * 322 * TODO: On platforms that support it, use openat()-style operations 323 * to eliminate the chdir() operations entirely while still supporting 324 * arbitrarily deep traversals. This makes access_path troublesome to 325 * support, of course, which means we'll need a rich enough interface 326 * that clients can function without it. (In particular, we'll need 327 * tree_current_open() that returns an open file descriptor.) 328 * 329 */ 330 static const char *tree_current_path(struct tree *); 331 static const char *tree_current_access_path(struct tree *); 332 333 /* 334 * Request the lstat() or stat() data for the current path. Since the 335 * tree package needs to do some of this anyway, and caches the 336 * results, you should take advantage of it here if you need it rather 337 * than make a redundant stat() or lstat() call of your own. 338 */ 339 static const struct stat *tree_current_stat(struct tree *); 340 static const struct stat *tree_current_lstat(struct tree *); 341 static int tree_current_is_symblic_link_target(struct tree *); 342 343 /* The following functions use tricks to avoid a certain number of 344 * stat()/lstat() calls. */ 345 /* "is_physical_dir" is equivalent to S_ISDIR(tree_current_lstat()->st_mode) */ 346 static int tree_current_is_physical_dir(struct tree *); 347 /* "is_dir" is equivalent to S_ISDIR(tree_current_stat()->st_mode) */ 348 static int tree_current_is_dir(struct tree *); 349 static int update_current_filesystem(struct archive_read_disk *a, 350 int64_t dev); 351 static int setup_current_filesystem(struct archive_read_disk *); 352 static int tree_target_is_same_as_parent(struct tree *, const struct stat *); 353 354 static int _archive_read_disk_open(struct archive *, const char *); 355 static int _archive_read_free(struct archive *); 356 static int _archive_read_close(struct archive *); 357 static int _archive_read_data_block(struct archive *, 358 const void **, size_t *, int64_t *); 359 static int _archive_read_next_header(struct archive *, 360 struct archive_entry **); 361 static int _archive_read_next_header2(struct archive *, 362 struct archive_entry *); 363 static const char *trivial_lookup_gname(void *, int64_t gid); 364 static const char *trivial_lookup_uname(void *, int64_t uid); 365 static int setup_sparse(struct archive_read_disk *, struct archive_entry *); 366 static int close_and_restore_time(int fd, struct tree *, 367 struct restore_time *); 368 static int open_on_current_dir(struct tree *, const char *, int); 369 static int tree_dup(int); 370 371 372 static struct archive_vtable * 373 archive_read_disk_vtable(void) 374 { 375 static struct archive_vtable av; 376 static int inited = 0; 377 378 if (!inited) { 379 av.archive_free = _archive_read_free; 380 av.archive_close = _archive_read_close; 381 av.archive_read_data_block = _archive_read_data_block; 382 av.archive_read_next_header = _archive_read_next_header; 383 av.archive_read_next_header2 = _archive_read_next_header2; 384 inited = 1; 385 } 386 return (&av); 387 } 388 389 const char * 390 archive_read_disk_gname(struct archive *_a, la_int64_t gid) 391 { 392 struct archive_read_disk *a = (struct archive_read_disk *)_a; 393 if (ARCHIVE_OK != __archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 394 ARCHIVE_STATE_ANY, "archive_read_disk_gname")) 395 return (NULL); 396 if (a->lookup_gname == NULL) 397 return (NULL); 398 return ((*a->lookup_gname)(a->lookup_gname_data, gid)); 399 } 400 401 const char * 402 archive_read_disk_uname(struct archive *_a, la_int64_t uid) 403 { 404 struct archive_read_disk *a = (struct archive_read_disk *)_a; 405 if (ARCHIVE_OK != __archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 406 ARCHIVE_STATE_ANY, "archive_read_disk_uname")) 407 return (NULL); 408 if (a->lookup_uname == NULL) 409 return (NULL); 410 return ((*a->lookup_uname)(a->lookup_uname_data, uid)); 411 } 412 413 int 414 archive_read_disk_set_gname_lookup(struct archive *_a, 415 void *private_data, 416 const char * (*lookup_gname)(void *private, la_int64_t gid), 417 void (*cleanup_gname)(void *private)) 418 { 419 struct archive_read_disk *a = (struct archive_read_disk *)_a; 420 archive_check_magic(&a->archive, ARCHIVE_READ_DISK_MAGIC, 421 ARCHIVE_STATE_ANY, "archive_read_disk_set_gname_lookup"); 422 423 if (a->cleanup_gname != NULL && a->lookup_gname_data != NULL) 424 (a->cleanup_gname)(a->lookup_gname_data); 425 426 a->lookup_gname = lookup_gname; 427 a->cleanup_gname = cleanup_gname; 428 a->lookup_gname_data = private_data; 429 return (ARCHIVE_OK); 430 } 431 432 int 433 archive_read_disk_set_uname_lookup(struct archive *_a, 434 void *private_data, 435 const char * (*lookup_uname)(void *private, la_int64_t uid), 436 void (*cleanup_uname)(void *private)) 437 { 438 struct archive_read_disk *a = (struct archive_read_disk *)_a; 439 archive_check_magic(&a->archive, ARCHIVE_READ_DISK_MAGIC, 440 ARCHIVE_STATE_ANY, "archive_read_disk_set_uname_lookup"); 441 442 if (a->cleanup_uname != NULL && a->lookup_uname_data != NULL) 443 (a->cleanup_uname)(a->lookup_uname_data); 444 445 a->lookup_uname = lookup_uname; 446 a->cleanup_uname = cleanup_uname; 447 a->lookup_uname_data = private_data; 448 return (ARCHIVE_OK); 449 } 450 451 /* 452 * Create a new archive_read_disk object and initialize it with global state. 453 */ 454 struct archive * 455 archive_read_disk_new(void) 456 { 457 struct archive_read_disk *a; 458 459 a = (struct archive_read_disk *)calloc(1, sizeof(*a)); 460 if (a == NULL) 461 return (NULL); 462 a->archive.magic = ARCHIVE_READ_DISK_MAGIC; 463 a->archive.state = ARCHIVE_STATE_NEW; 464 a->archive.vtable = archive_read_disk_vtable(); 465 a->entry = archive_entry_new2(&a->archive); 466 a->lookup_uname = trivial_lookup_uname; 467 a->lookup_gname = trivial_lookup_gname; 468 a->flags = ARCHIVE_READDISK_MAC_COPYFILE; 469 a->open_on_current_dir = open_on_current_dir; 470 a->tree_current_dir_fd = tree_current_dir_fd; 471 a->tree_enter_working_dir = tree_enter_working_dir; 472 return (&a->archive); 473 } 474 475 static int 476 _archive_read_free(struct archive *_a) 477 { 478 struct archive_read_disk *a = (struct archive_read_disk *)_a; 479 int r; 480 481 if (_a == NULL) 482 return (ARCHIVE_OK); 483 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 484 ARCHIVE_STATE_ANY | ARCHIVE_STATE_FATAL, "archive_read_free"); 485 486 if (a->archive.state != ARCHIVE_STATE_CLOSED) 487 r = _archive_read_close(&a->archive); 488 else 489 r = ARCHIVE_OK; 490 491 tree_free(a->tree); 492 if (a->cleanup_gname != NULL && a->lookup_gname_data != NULL) 493 (a->cleanup_gname)(a->lookup_gname_data); 494 if (a->cleanup_uname != NULL && a->lookup_uname_data != NULL) 495 (a->cleanup_uname)(a->lookup_uname_data); 496 archive_string_free(&a->archive.error_string); 497 archive_entry_free(a->entry); 498 a->archive.magic = 0; 499 __archive_clean(&a->archive); 500 free(a); 501 return (r); 502 } 503 504 static int 505 _archive_read_close(struct archive *_a) 506 { 507 struct archive_read_disk *a = (struct archive_read_disk *)_a; 508 509 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 510 ARCHIVE_STATE_ANY | ARCHIVE_STATE_FATAL, "archive_read_close"); 511 512 if (a->archive.state != ARCHIVE_STATE_FATAL) 513 a->archive.state = ARCHIVE_STATE_CLOSED; 514 515 tree_close(a->tree); 516 517 return (ARCHIVE_OK); 518 } 519 520 static void 521 setup_symlink_mode(struct archive_read_disk *a, char symlink_mode, 522 int follow_symlinks) 523 { 524 a->symlink_mode = symlink_mode; 525 a->follow_symlinks = follow_symlinks; 526 if (a->tree != NULL) { 527 a->tree->initial_symlink_mode = a->symlink_mode; 528 a->tree->symlink_mode = a->symlink_mode; 529 } 530 } 531 532 int 533 archive_read_disk_set_symlink_logical(struct archive *_a) 534 { 535 struct archive_read_disk *a = (struct archive_read_disk *)_a; 536 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 537 ARCHIVE_STATE_ANY, "archive_read_disk_set_symlink_logical"); 538 setup_symlink_mode(a, 'L', 1); 539 return (ARCHIVE_OK); 540 } 541 542 int 543 archive_read_disk_set_symlink_physical(struct archive *_a) 544 { 545 struct archive_read_disk *a = (struct archive_read_disk *)_a; 546 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 547 ARCHIVE_STATE_ANY, "archive_read_disk_set_symlink_physical"); 548 setup_symlink_mode(a, 'P', 0); 549 return (ARCHIVE_OK); 550 } 551 552 int 553 archive_read_disk_set_symlink_hybrid(struct archive *_a) 554 { 555 struct archive_read_disk *a = (struct archive_read_disk *)_a; 556 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 557 ARCHIVE_STATE_ANY, "archive_read_disk_set_symlink_hybrid"); 558 setup_symlink_mode(a, 'H', 1);/* Follow symlinks initially. */ 559 return (ARCHIVE_OK); 560 } 561 562 int 563 archive_read_disk_set_atime_restored(struct archive *_a) 564 { 565 struct archive_read_disk *a = (struct archive_read_disk *)_a; 566 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 567 ARCHIVE_STATE_ANY, "archive_read_disk_restore_atime"); 568 #ifdef HAVE_UTIMES 569 a->flags |= ARCHIVE_READDISK_RESTORE_ATIME; 570 if (a->tree != NULL) 571 a->tree->flags |= needsRestoreTimes; 572 return (ARCHIVE_OK); 573 #else 574 /* Display warning and unset flag */ 575 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 576 "Cannot restore access time on this system"); 577 a->flags &= ~ARCHIVE_READDISK_RESTORE_ATIME; 578 return (ARCHIVE_WARN); 579 #endif 580 } 581 582 int 583 archive_read_disk_set_behavior(struct archive *_a, int flags) 584 { 585 struct archive_read_disk *a = (struct archive_read_disk *)_a; 586 int r = ARCHIVE_OK; 587 588 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 589 ARCHIVE_STATE_ANY, "archive_read_disk_honor_nodump"); 590 591 a->flags = flags; 592 593 if (flags & ARCHIVE_READDISK_RESTORE_ATIME) 594 r = archive_read_disk_set_atime_restored(_a); 595 else { 596 if (a->tree != NULL) 597 a->tree->flags &= ~needsRestoreTimes; 598 } 599 return (r); 600 } 601 602 /* 603 * Trivial implementations of gname/uname lookup functions. 604 * These are normally overridden by the client, but these stub 605 * versions ensure that we always have something that works. 606 */ 607 static const char * 608 trivial_lookup_gname(void *private_data, int64_t gid) 609 { 610 (void)private_data; /* UNUSED */ 611 (void)gid; /* UNUSED */ 612 return (NULL); 613 } 614 615 static const char * 616 trivial_lookup_uname(void *private_data, int64_t uid) 617 { 618 (void)private_data; /* UNUSED */ 619 (void)uid; /* UNUSED */ 620 return (NULL); 621 } 622 623 /* 624 * Allocate memory for the reading buffer adjusted to the filesystem 625 * alignment. 626 */ 627 static int 628 setup_suitable_read_buffer(struct archive_read_disk *a) 629 { 630 struct tree *t = a->tree; 631 struct filesystem *cf = t->current_filesystem; 632 size_t asize; 633 size_t s; 634 635 if (cf->allocation_ptr == NULL) { 636 /* If we couldn't get a filesystem alignment, 637 * we use 4096 as default value but we won't use 638 * O_DIRECT to open() and openat() operations. */ 639 long xfer_align = (cf->xfer_align == -1)?4096:cf->xfer_align; 640 641 if (cf->max_xfer_size != -1) 642 asize = cf->max_xfer_size + xfer_align; 643 else { 644 long incr = cf->incr_xfer_size; 645 /* Some platform does not set a proper value to 646 * incr_xfer_size.*/ 647 if (incr < 0) 648 incr = cf->min_xfer_size; 649 if (cf->min_xfer_size < 0) { 650 incr = xfer_align; 651 asize = xfer_align; 652 } else 653 asize = cf->min_xfer_size; 654 655 /* Increase a buffer size up to 64K bytes in 656 * a proper increment size. */ 657 while (asize < 1024*64) 658 asize += incr; 659 /* Take a margin to adjust to the filesystem 660 * alignment. */ 661 asize += xfer_align; 662 } 663 cf->allocation_ptr = malloc(asize); 664 if (cf->allocation_ptr == NULL) { 665 archive_set_error(&a->archive, ENOMEM, 666 "Couldn't allocate memory"); 667 a->archive.state = ARCHIVE_STATE_FATAL; 668 return (ARCHIVE_FATAL); 669 } 670 671 /* 672 * Calculate proper address for the filesystem. 673 */ 674 s = (uintptr_t)cf->allocation_ptr; 675 s %= xfer_align; 676 if (s > 0) 677 s = xfer_align - s; 678 679 /* 680 * Set a read buffer pointer in the proper alignment of 681 * the current filesystem. 682 */ 683 cf->buff = cf->allocation_ptr + s; 684 cf->buff_size = asize - xfer_align; 685 } 686 return (ARCHIVE_OK); 687 } 688 689 static int 690 _archive_read_data_block(struct archive *_a, const void **buff, 691 size_t *size, int64_t *offset) 692 { 693 struct archive_read_disk *a = (struct archive_read_disk *)_a; 694 struct tree *t = a->tree; 695 int r; 696 ssize_t bytes; 697 int64_t sparse_bytes; 698 size_t buffbytes; 699 int empty_sparse_region = 0; 700 701 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, ARCHIVE_STATE_DATA, 702 "archive_read_data_block"); 703 704 if (t->entry_eof || t->entry_remaining_bytes <= 0) { 705 r = ARCHIVE_EOF; 706 goto abort_read_data; 707 } 708 709 /* 710 * Open the current file. 711 */ 712 if (t->entry_fd < 0) { 713 int flags = O_RDONLY | O_BINARY | O_CLOEXEC; 714 715 /* 716 * Eliminate or reduce cache effects if we can. 717 * 718 * Carefully consider this to be enabled. 719 */ 720 #if defined(O_DIRECT) && 0/* Disabled for now */ 721 if (t->current_filesystem->xfer_align != -1 && 722 t->nlink == 1) 723 flags |= O_DIRECT; 724 #endif 725 #if defined(O_NOATIME) 726 /* 727 * Linux has O_NOATIME flag; use it if we need. 728 */ 729 if ((t->flags & needsRestoreTimes) != 0 && 730 t->restore_time.noatime == 0) 731 flags |= O_NOATIME; 732 #endif 733 t->entry_fd = open_on_current_dir(t, 734 tree_current_access_path(t), flags); 735 __archive_ensure_cloexec_flag(t->entry_fd); 736 #if defined(O_NOATIME) 737 /* 738 * When we did open the file with O_NOATIME flag, 739 * if successful, set 1 to t->restore_time.noatime 740 * not to restore an atime of the file later. 741 * if failed by EPERM, retry it without O_NOATIME flag. 742 */ 743 if (flags & O_NOATIME) { 744 if (t->entry_fd >= 0) 745 t->restore_time.noatime = 1; 746 else if (errno == EPERM) 747 flags &= ~O_NOATIME; 748 } 749 #endif 750 if (t->entry_fd < 0) { 751 archive_set_error(&a->archive, errno, 752 "Couldn't open %s", tree_current_path(t)); 753 r = ARCHIVE_FAILED; 754 tree_enter_initial_dir(t); 755 goto abort_read_data; 756 } 757 tree_enter_initial_dir(t); 758 } 759 760 /* 761 * Allocate read buffer if not allocated. 762 */ 763 if (t->current_filesystem->allocation_ptr == NULL) { 764 r = setup_suitable_read_buffer(a); 765 if (r != ARCHIVE_OK) { 766 a->archive.state = ARCHIVE_STATE_FATAL; 767 goto abort_read_data; 768 } 769 } 770 t->entry_buff = t->current_filesystem->buff; 771 t->entry_buff_size = t->current_filesystem->buff_size; 772 773 buffbytes = t->entry_buff_size; 774 if ((int64_t)buffbytes > t->current_sparse->length) 775 buffbytes = t->current_sparse->length; 776 777 if (t->current_sparse->length == 0) 778 empty_sparse_region = 1; 779 780 /* 781 * Skip hole. 782 * TODO: Should we consider t->current_filesystem->xfer_align? 783 */ 784 if (t->current_sparse->offset > t->entry_total) { 785 if (lseek(t->entry_fd, 786 (off_t)t->current_sparse->offset, SEEK_SET) < 0) { 787 archive_set_error(&a->archive, errno, "Seek error"); 788 r = ARCHIVE_FATAL; 789 a->archive.state = ARCHIVE_STATE_FATAL; 790 goto abort_read_data; 791 } 792 sparse_bytes = t->current_sparse->offset - t->entry_total; 793 t->entry_remaining_bytes -= sparse_bytes; 794 t->entry_total += sparse_bytes; 795 } 796 797 /* 798 * Read file contents. 799 */ 800 if (buffbytes > 0) { 801 bytes = read(t->entry_fd, t->entry_buff, buffbytes); 802 if (bytes < 0) { 803 archive_set_error(&a->archive, errno, "Read error"); 804 r = ARCHIVE_FATAL; 805 a->archive.state = ARCHIVE_STATE_FATAL; 806 goto abort_read_data; 807 } 808 } else 809 bytes = 0; 810 /* 811 * Return an EOF unless we've read a leading empty sparse region, which 812 * is used to represent fully-sparse files. 813 */ 814 if (bytes == 0 && !empty_sparse_region) { 815 /* Get EOF */ 816 t->entry_eof = 1; 817 r = ARCHIVE_EOF; 818 goto abort_read_data; 819 } 820 *buff = t->entry_buff; 821 *size = bytes; 822 *offset = t->entry_total; 823 t->entry_total += bytes; 824 t->entry_remaining_bytes -= bytes; 825 if (t->entry_remaining_bytes == 0) { 826 /* Close the current file descriptor */ 827 close_and_restore_time(t->entry_fd, t, &t->restore_time); 828 t->entry_fd = -1; 829 t->entry_eof = 1; 830 } 831 t->current_sparse->offset += bytes; 832 t->current_sparse->length -= bytes; 833 if (t->current_sparse->length == 0 && !t->entry_eof) 834 t->current_sparse++; 835 return (ARCHIVE_OK); 836 837 abort_read_data: 838 *buff = NULL; 839 *size = 0; 840 *offset = t->entry_total; 841 if (t->entry_fd >= 0) { 842 /* Close the current file descriptor */ 843 close_and_restore_time(t->entry_fd, t, &t->restore_time); 844 t->entry_fd = -1; 845 } 846 return (r); 847 } 848 849 static int 850 next_entry(struct archive_read_disk *a, struct tree *t, 851 struct archive_entry *entry) 852 { 853 const struct stat *st; /* info to use for this entry */ 854 const struct stat *lst;/* lstat() information */ 855 const char *name; 856 int delayed, delayed_errno, descend, r; 857 struct archive_string delayed_str; 858 859 delayed = ARCHIVE_OK; 860 delayed_errno = 0; 861 archive_string_init(&delayed_str); 862 863 st = NULL; 864 lst = NULL; 865 t->descend = 0; 866 do { 867 switch (tree_next(t)) { 868 case TREE_ERROR_FATAL: 869 archive_set_error(&a->archive, t->tree_errno, 870 "%s: Unable to continue traversing directory tree", 871 tree_current_path(t)); 872 a->archive.state = ARCHIVE_STATE_FATAL; 873 tree_enter_initial_dir(t); 874 return (ARCHIVE_FATAL); 875 case TREE_ERROR_DIR: 876 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 877 "%s: Couldn't visit directory", 878 tree_current_path(t)); 879 tree_enter_initial_dir(t); 880 return (ARCHIVE_FAILED); 881 case 0: 882 tree_enter_initial_dir(t); 883 return (ARCHIVE_EOF); 884 case TREE_POSTDESCENT: 885 case TREE_POSTASCENT: 886 break; 887 case TREE_REGULAR: 888 lst = tree_current_lstat(t); 889 if (lst == NULL) { 890 if (errno == ENOENT && t->depth > 0) { 891 delayed = ARCHIVE_WARN; 892 delayed_errno = errno; 893 if (delayed_str.length == 0) { 894 archive_string_sprintf(&delayed_str, 895 "%s", tree_current_path(t)); 896 } else { 897 archive_string_sprintf(&delayed_str, 898 " %s", tree_current_path(t)); 899 } 900 } else { 901 archive_set_error(&a->archive, errno, 902 "%s: Cannot stat", 903 tree_current_path(t)); 904 tree_enter_initial_dir(t); 905 return (ARCHIVE_FAILED); 906 } 907 } 908 break; 909 } 910 } while (lst == NULL); 911 912 #ifdef __APPLE__ 913 if (a->flags & ARCHIVE_READDISK_MAC_COPYFILE) { 914 /* If we're using copyfile(), ignore "._XXX" files. */ 915 const char *bname = strrchr(tree_current_path(t), '/'); 916 if (bname == NULL) 917 bname = tree_current_path(t); 918 else 919 ++bname; 920 if (bname[0] == '.' && bname[1] == '_') 921 return (ARCHIVE_RETRY); 922 } 923 #endif 924 925 archive_entry_copy_pathname(entry, tree_current_path(t)); 926 /* 927 * Perform path matching. 928 */ 929 if (a->matching) { 930 r = archive_match_path_excluded(a->matching, entry); 931 if (r < 0) { 932 archive_set_error(&(a->archive), errno, 933 "Failed : %s", archive_error_string(a->matching)); 934 return (r); 935 } 936 if (r) { 937 if (a->excluded_cb_func) 938 a->excluded_cb_func(&(a->archive), 939 a->excluded_cb_data, entry); 940 return (ARCHIVE_RETRY); 941 } 942 } 943 944 /* 945 * Distinguish 'L'/'P'/'H' symlink following. 946 */ 947 switch(t->symlink_mode) { 948 case 'H': 949 /* 'H': After the first item, rest like 'P'. */ 950 t->symlink_mode = 'P'; 951 /* 'H': First item (from command line) like 'L'. */ 952 /* FALLTHROUGH */ 953 case 'L': 954 /* 'L': Do descend through a symlink to dir. */ 955 descend = tree_current_is_dir(t); 956 /* 'L': Follow symlinks to files. */ 957 a->symlink_mode = 'L'; 958 a->follow_symlinks = 1; 959 /* 'L': Archive symlinks as targets, if we can. */ 960 st = tree_current_stat(t); 961 if (st != NULL && !tree_target_is_same_as_parent(t, st)) 962 break; 963 /* If stat fails, we have a broken symlink; 964 * in that case, don't follow the link. */ 965 /* FALLTHROUGH */ 966 default: 967 /* 'P': Don't descend through a symlink to dir. */ 968 descend = tree_current_is_physical_dir(t); 969 /* 'P': Don't follow symlinks to files. */ 970 a->symlink_mode = 'P'; 971 a->follow_symlinks = 0; 972 /* 'P': Archive symlinks as symlinks. */ 973 st = lst; 974 break; 975 } 976 977 if (update_current_filesystem(a, st->st_dev) != ARCHIVE_OK) { 978 a->archive.state = ARCHIVE_STATE_FATAL; 979 tree_enter_initial_dir(t); 980 return (ARCHIVE_FATAL); 981 } 982 if (t->initial_filesystem_id == -1) 983 t->initial_filesystem_id = t->current_filesystem_id; 984 if (a->flags & ARCHIVE_READDISK_NO_TRAVERSE_MOUNTS) { 985 if (t->initial_filesystem_id != t->current_filesystem_id) 986 descend = 0; 987 } 988 t->descend = descend; 989 990 /* 991 * Honor nodump flag. 992 * If the file is marked with nodump flag, do not return this entry. 993 */ 994 if (a->flags & ARCHIVE_READDISK_HONOR_NODUMP) { 995 #if defined(HAVE_STRUCT_STAT_ST_FLAGS) && defined(UF_NODUMP) 996 if (st->st_flags & UF_NODUMP) 997 return (ARCHIVE_RETRY); 998 #elif (defined(FS_IOC_GETFLAGS) && defined(FS_NODUMP_FL) && \ 999 defined(HAVE_WORKING_FS_IOC_GETFLAGS)) || \ 1000 (defined(EXT2_IOC_GETFLAGS) && defined(EXT2_NODUMP_FL) && \ 1001 defined(HAVE_WORKING_EXT2_IOC_GETFLAGS)) 1002 if (S_ISREG(st->st_mode) || S_ISDIR(st->st_mode)) { 1003 int stflags; 1004 1005 t->entry_fd = open_on_current_dir(t, 1006 tree_current_access_path(t), 1007 O_RDONLY | O_NONBLOCK | O_CLOEXEC); 1008 __archive_ensure_cloexec_flag(t->entry_fd); 1009 if (t->entry_fd >= 0) { 1010 r = ioctl(t->entry_fd, 1011 #ifdef FS_IOC_GETFLAGS 1012 FS_IOC_GETFLAGS, 1013 #else 1014 EXT2_IOC_GETFLAGS, 1015 #endif 1016 &stflags); 1017 #ifdef FS_NODUMP_FL 1018 if (r == 0 && (stflags & FS_NODUMP_FL) != 0) 1019 #else 1020 if (r == 0 && (stflags & EXT2_NODUMP_FL) != 0) 1021 #endif 1022 return (ARCHIVE_RETRY); 1023 } 1024 } 1025 #endif 1026 } 1027 1028 archive_entry_copy_stat(entry, st); 1029 1030 /* Save the times to be restored. This must be in before 1031 * calling archive_read_disk_descend() or any chance of it, 1032 * especially, invoking a callback. */ 1033 t->restore_time.mtime = archive_entry_mtime(entry); 1034 t->restore_time.mtime_nsec = archive_entry_mtime_nsec(entry); 1035 t->restore_time.atime = archive_entry_atime(entry); 1036 t->restore_time.atime_nsec = archive_entry_atime_nsec(entry); 1037 t->restore_time.filetype = archive_entry_filetype(entry); 1038 t->restore_time.noatime = t->current_filesystem->noatime; 1039 1040 /* 1041 * Perform time matching. 1042 */ 1043 if (a->matching) { 1044 r = archive_match_time_excluded(a->matching, entry); 1045 if (r < 0) { 1046 archive_set_error(&(a->archive), errno, 1047 "Failed : %s", archive_error_string(a->matching)); 1048 return (r); 1049 } 1050 if (r) { 1051 if (a->excluded_cb_func) 1052 a->excluded_cb_func(&(a->archive), 1053 a->excluded_cb_data, entry); 1054 return (ARCHIVE_RETRY); 1055 } 1056 } 1057 1058 /* Lookup uname/gname */ 1059 name = archive_read_disk_uname(&(a->archive), archive_entry_uid(entry)); 1060 if (name != NULL) 1061 archive_entry_copy_uname(entry, name); 1062 name = archive_read_disk_gname(&(a->archive), archive_entry_gid(entry)); 1063 if (name != NULL) 1064 archive_entry_copy_gname(entry, name); 1065 1066 /* 1067 * Perform owner matching. 1068 */ 1069 if (a->matching) { 1070 r = archive_match_owner_excluded(a->matching, entry); 1071 if (r < 0) { 1072 archive_set_error(&(a->archive), errno, 1073 "Failed : %s", archive_error_string(a->matching)); 1074 return (r); 1075 } 1076 if (r) { 1077 if (a->excluded_cb_func) 1078 a->excluded_cb_func(&(a->archive), 1079 a->excluded_cb_data, entry); 1080 return (ARCHIVE_RETRY); 1081 } 1082 } 1083 1084 /* 1085 * Invoke a meta data filter callback. 1086 */ 1087 if (a->metadata_filter_func) { 1088 if (!a->metadata_filter_func(&(a->archive), 1089 a->metadata_filter_data, entry)) 1090 return (ARCHIVE_RETRY); 1091 } 1092 1093 /* 1094 * Populate the archive_entry with metadata from the disk. 1095 */ 1096 archive_entry_copy_sourcepath(entry, tree_current_access_path(t)); 1097 r = archive_read_disk_entry_from_file(&(a->archive), entry, 1098 t->entry_fd, st); 1099 1100 if (r == ARCHIVE_OK) { 1101 r = delayed; 1102 if (r != ARCHIVE_OK) { 1103 archive_string_sprintf(&delayed_str, ": %s", 1104 "File removed before we read it"); 1105 archive_set_error(&(a->archive), delayed_errno, 1106 "%s", delayed_str.s); 1107 } 1108 } 1109 archive_string_free(&delayed_str); 1110 1111 return (r); 1112 } 1113 1114 static int 1115 _archive_read_next_header(struct archive *_a, struct archive_entry **entryp) 1116 { 1117 int ret; 1118 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1119 *entryp = NULL; 1120 ret = _archive_read_next_header2(_a, a->entry); 1121 *entryp = a->entry; 1122 return ret; 1123 } 1124 1125 static int 1126 _archive_read_next_header2(struct archive *_a, struct archive_entry *entry) 1127 { 1128 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1129 struct tree *t; 1130 int r; 1131 1132 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 1133 ARCHIVE_STATE_HEADER | ARCHIVE_STATE_DATA, 1134 "archive_read_next_header2"); 1135 1136 t = a->tree; 1137 if (t->entry_fd >= 0) { 1138 close_and_restore_time(t->entry_fd, t, &t->restore_time); 1139 t->entry_fd = -1; 1140 } 1141 1142 archive_entry_clear(entry); 1143 1144 for (;;) { 1145 r = next_entry(a, t, entry); 1146 if (t->entry_fd >= 0) { 1147 close(t->entry_fd); 1148 t->entry_fd = -1; 1149 } 1150 1151 if (r == ARCHIVE_RETRY) { 1152 archive_entry_clear(entry); 1153 continue; 1154 } 1155 break; 1156 } 1157 1158 /* Return to the initial directory. */ 1159 tree_enter_initial_dir(t); 1160 1161 /* 1162 * EOF and FATAL are persistent at this layer. By 1163 * modifying the state, we guarantee that future calls to 1164 * read a header or read data will fail. 1165 */ 1166 switch (r) { 1167 case ARCHIVE_EOF: 1168 a->archive.state = ARCHIVE_STATE_EOF; 1169 break; 1170 case ARCHIVE_OK: 1171 case ARCHIVE_WARN: 1172 /* Overwrite the sourcepath based on the initial directory. */ 1173 archive_entry_copy_sourcepath(entry, tree_current_path(t)); 1174 t->entry_total = 0; 1175 if (archive_entry_filetype(entry) == AE_IFREG) { 1176 t->nlink = archive_entry_nlink(entry); 1177 t->entry_remaining_bytes = archive_entry_size(entry); 1178 t->entry_eof = (t->entry_remaining_bytes == 0)? 1: 0; 1179 if (!t->entry_eof && 1180 setup_sparse(a, entry) != ARCHIVE_OK) 1181 return (ARCHIVE_FATAL); 1182 } else { 1183 t->entry_remaining_bytes = 0; 1184 t->entry_eof = 1; 1185 } 1186 a->archive.state = ARCHIVE_STATE_DATA; 1187 break; 1188 case ARCHIVE_RETRY: 1189 break; 1190 case ARCHIVE_FATAL: 1191 a->archive.state = ARCHIVE_STATE_FATAL; 1192 break; 1193 } 1194 1195 __archive_reset_read_data(&a->archive); 1196 return (r); 1197 } 1198 1199 static int 1200 setup_sparse(struct archive_read_disk *a, struct archive_entry *entry) 1201 { 1202 struct tree *t = a->tree; 1203 int64_t length, offset; 1204 int i; 1205 1206 t->sparse_count = archive_entry_sparse_reset(entry); 1207 if (t->sparse_count+1 > t->sparse_list_size) { 1208 free(t->sparse_list); 1209 t->sparse_list_size = t->sparse_count + 1; 1210 t->sparse_list = malloc(sizeof(t->sparse_list[0]) * 1211 t->sparse_list_size); 1212 if (t->sparse_list == NULL) { 1213 t->sparse_list_size = 0; 1214 archive_set_error(&a->archive, ENOMEM, 1215 "Can't allocate data"); 1216 a->archive.state = ARCHIVE_STATE_FATAL; 1217 return (ARCHIVE_FATAL); 1218 } 1219 } 1220 for (i = 0; i < t->sparse_count; i++) { 1221 archive_entry_sparse_next(entry, &offset, &length); 1222 t->sparse_list[i].offset = offset; 1223 t->sparse_list[i].length = length; 1224 } 1225 if (i == 0) { 1226 t->sparse_list[i].offset = 0; 1227 t->sparse_list[i].length = archive_entry_size(entry); 1228 } else { 1229 t->sparse_list[i].offset = archive_entry_size(entry); 1230 t->sparse_list[i].length = 0; 1231 } 1232 t->current_sparse = t->sparse_list; 1233 1234 return (ARCHIVE_OK); 1235 } 1236 1237 int 1238 archive_read_disk_set_matching(struct archive *_a, struct archive *_ma, 1239 void (*_excluded_func)(struct archive *, void *, struct archive_entry *), 1240 void *_client_data) 1241 { 1242 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1243 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 1244 ARCHIVE_STATE_ANY, "archive_read_disk_set_matching"); 1245 a->matching = _ma; 1246 a->excluded_cb_func = _excluded_func; 1247 a->excluded_cb_data = _client_data; 1248 return (ARCHIVE_OK); 1249 } 1250 1251 int 1252 archive_read_disk_set_metadata_filter_callback(struct archive *_a, 1253 int (*_metadata_filter_func)(struct archive *, void *, 1254 struct archive_entry *), void *_client_data) 1255 { 1256 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1257 1258 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, ARCHIVE_STATE_ANY, 1259 "archive_read_disk_set_metadata_filter_callback"); 1260 1261 a->metadata_filter_func = _metadata_filter_func; 1262 a->metadata_filter_data = _client_data; 1263 return (ARCHIVE_OK); 1264 } 1265 1266 int 1267 archive_read_disk_can_descend(struct archive *_a) 1268 { 1269 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1270 struct tree *t = a->tree; 1271 1272 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 1273 ARCHIVE_STATE_HEADER | ARCHIVE_STATE_DATA, 1274 "archive_read_disk_can_descend"); 1275 1276 return (t->visit_type == TREE_REGULAR && t->descend); 1277 } 1278 1279 /* 1280 * Called by the client to mark the directory just returned from 1281 * tree_next() as needing to be visited. 1282 */ 1283 int 1284 archive_read_disk_descend(struct archive *_a) 1285 { 1286 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1287 struct tree *t = a->tree; 1288 1289 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 1290 ARCHIVE_STATE_HEADER | ARCHIVE_STATE_DATA, 1291 "archive_read_disk_descend"); 1292 1293 if (t->visit_type != TREE_REGULAR || !t->descend) 1294 return (ARCHIVE_OK); 1295 1296 /* 1297 * We must not treat the initial specified path as a physical dir, 1298 * because if we do then we will try and ascend out of it by opening 1299 * ".." which is (a) wrong and (b) causes spurious permissions errors 1300 * if ".." is not readable by us. Instead, treat it as if it were a 1301 * symlink. (This uses an extra fd, but it can only happen once at the 1302 * top level of a traverse.) But we can't necessarily assume t->st is 1303 * valid here (though t->lst is), which complicates the logic a 1304 * little. 1305 */ 1306 if (tree_current_is_physical_dir(t)) { 1307 tree_push(t, t->basename, t->current_filesystem_id, 1308 t->lst.st_dev, t->lst.st_ino, &t->restore_time); 1309 if (t->stack->parent->parent != NULL) 1310 t->stack->flags |= isDir; 1311 else 1312 t->stack->flags |= isDirLink; 1313 } else if (tree_current_is_dir(t)) { 1314 tree_push(t, t->basename, t->current_filesystem_id, 1315 t->st.st_dev, t->st.st_ino, &t->restore_time); 1316 t->stack->flags |= isDirLink; 1317 } 1318 t->descend = 0; 1319 return (ARCHIVE_OK); 1320 } 1321 1322 int 1323 archive_read_disk_open(struct archive *_a, const char *pathname) 1324 { 1325 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1326 1327 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 1328 ARCHIVE_STATE_NEW | ARCHIVE_STATE_CLOSED, 1329 "archive_read_disk_open"); 1330 archive_clear_error(&a->archive); 1331 1332 return (_archive_read_disk_open(_a, pathname)); 1333 } 1334 1335 int 1336 archive_read_disk_open_w(struct archive *_a, const wchar_t *pathname) 1337 { 1338 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1339 struct archive_string path; 1340 int ret; 1341 1342 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, 1343 ARCHIVE_STATE_NEW | ARCHIVE_STATE_CLOSED, 1344 "archive_read_disk_open_w"); 1345 archive_clear_error(&a->archive); 1346 1347 /* Make a char string from a wchar_t string. */ 1348 archive_string_init(&path); 1349 if (archive_string_append_from_wcs(&path, pathname, 1350 wcslen(pathname)) != 0) { 1351 if (errno == ENOMEM) 1352 archive_set_error(&a->archive, ENOMEM, 1353 "Can't allocate memory"); 1354 else 1355 archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC, 1356 "Can't convert a path to a char string"); 1357 a->archive.state = ARCHIVE_STATE_FATAL; 1358 ret = ARCHIVE_FATAL; 1359 } else 1360 ret = _archive_read_disk_open(_a, path.s); 1361 1362 archive_string_free(&path); 1363 return (ret); 1364 } 1365 1366 static int 1367 _archive_read_disk_open(struct archive *_a, const char *pathname) 1368 { 1369 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1370 1371 if (a->tree != NULL) 1372 a->tree = tree_reopen(a->tree, pathname, 1373 a->flags & ARCHIVE_READDISK_RESTORE_ATIME); 1374 else 1375 a->tree = tree_open(pathname, a->symlink_mode, 1376 a->flags & ARCHIVE_READDISK_RESTORE_ATIME); 1377 if (a->tree == NULL) { 1378 archive_set_error(&a->archive, ENOMEM, 1379 "Can't allocate tar data"); 1380 a->archive.state = ARCHIVE_STATE_FATAL; 1381 return (ARCHIVE_FATAL); 1382 } 1383 a->archive.state = ARCHIVE_STATE_HEADER; 1384 1385 return (ARCHIVE_OK); 1386 } 1387 1388 /* 1389 * Return a current filesystem ID which is index of the filesystem entry 1390 * you've visited through archive_read_disk. 1391 */ 1392 int 1393 archive_read_disk_current_filesystem(struct archive *_a) 1394 { 1395 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1396 1397 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, ARCHIVE_STATE_DATA, 1398 "archive_read_disk_current_filesystem"); 1399 1400 return (a->tree->current_filesystem_id); 1401 } 1402 1403 static int 1404 update_current_filesystem(struct archive_read_disk *a, int64_t dev) 1405 { 1406 struct tree *t = a->tree; 1407 int i, fid; 1408 1409 if (t->current_filesystem != NULL && 1410 t->current_filesystem->dev == dev) 1411 return (ARCHIVE_OK); 1412 1413 for (i = 0; i < t->max_filesystem_id; i++) { 1414 if (t->filesystem_table[i].dev == dev) { 1415 /* There is the filesystem ID we've already generated. */ 1416 t->current_filesystem_id = i; 1417 t->current_filesystem = &(t->filesystem_table[i]); 1418 return (ARCHIVE_OK); 1419 } 1420 } 1421 1422 /* 1423 * This is the new filesystem which we have to generate a new ID for. 1424 */ 1425 fid = t->max_filesystem_id++; 1426 if (t->max_filesystem_id > t->allocated_filesystem) { 1427 size_t s; 1428 void *p; 1429 1430 s = t->max_filesystem_id * 2; 1431 p = realloc(t->filesystem_table, 1432 s * sizeof(*t->filesystem_table)); 1433 if (p == NULL) { 1434 archive_set_error(&a->archive, ENOMEM, 1435 "Can't allocate tar data"); 1436 return (ARCHIVE_FATAL); 1437 } 1438 t->filesystem_table = (struct filesystem *)p; 1439 t->allocated_filesystem = s; 1440 } 1441 t->current_filesystem_id = fid; 1442 t->current_filesystem = &(t->filesystem_table[fid]); 1443 t->current_filesystem->dev = dev; 1444 t->current_filesystem->allocation_ptr = NULL; 1445 t->current_filesystem->buff = NULL; 1446 1447 /* Setup the current filesystem properties which depend on 1448 * platform specific. */ 1449 return (setup_current_filesystem(a)); 1450 } 1451 1452 /* 1453 * Returns 1 if current filesystem is generated filesystem, 0 if it is not 1454 * or -1 if it is unknown. 1455 */ 1456 int 1457 archive_read_disk_current_filesystem_is_synthetic(struct archive *_a) 1458 { 1459 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1460 1461 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, ARCHIVE_STATE_DATA, 1462 "archive_read_disk_current_filesystem"); 1463 1464 return (a->tree->current_filesystem->synthetic); 1465 } 1466 1467 /* 1468 * Returns 1 if current filesystem is remote filesystem, 0 if it is not 1469 * or -1 if it is unknown. 1470 */ 1471 int 1472 archive_read_disk_current_filesystem_is_remote(struct archive *_a) 1473 { 1474 struct archive_read_disk *a = (struct archive_read_disk *)_a; 1475 1476 archive_check_magic(_a, ARCHIVE_READ_DISK_MAGIC, ARCHIVE_STATE_DATA, 1477 "archive_read_disk_current_filesystem"); 1478 1479 return (a->tree->current_filesystem->remote); 1480 } 1481 1482 #if defined(_PC_REC_INCR_XFER_SIZE) && defined(_PC_REC_MAX_XFER_SIZE) &&\ 1483 defined(_PC_REC_MIN_XFER_SIZE) && defined(_PC_REC_XFER_ALIGN) 1484 static int 1485 get_xfer_size(struct tree *t, int fd, const char *path) 1486 { 1487 t->current_filesystem->xfer_align = -1; 1488 errno = 0; 1489 if (fd >= 0) { 1490 t->current_filesystem->incr_xfer_size = 1491 fpathconf(fd, _PC_REC_INCR_XFER_SIZE); 1492 t->current_filesystem->max_xfer_size = 1493 fpathconf(fd, _PC_REC_MAX_XFER_SIZE); 1494 t->current_filesystem->min_xfer_size = 1495 fpathconf(fd, _PC_REC_MIN_XFER_SIZE); 1496 t->current_filesystem->xfer_align = 1497 fpathconf(fd, _PC_REC_XFER_ALIGN); 1498 } else if (path != NULL) { 1499 t->current_filesystem->incr_xfer_size = 1500 pathconf(path, _PC_REC_INCR_XFER_SIZE); 1501 t->current_filesystem->max_xfer_size = 1502 pathconf(path, _PC_REC_MAX_XFER_SIZE); 1503 t->current_filesystem->min_xfer_size = 1504 pathconf(path, _PC_REC_MIN_XFER_SIZE); 1505 t->current_filesystem->xfer_align = 1506 pathconf(path, _PC_REC_XFER_ALIGN); 1507 } 1508 /* At least we need an alignment size. */ 1509 if (t->current_filesystem->xfer_align == -1) 1510 return ((errno == EINVAL)?1:-1); 1511 else 1512 return (0); 1513 } 1514 #else 1515 static int 1516 get_xfer_size(struct tree *t, int fd, const char *path) 1517 { 1518 (void)t; /* UNUSED */ 1519 (void)fd; /* UNUSED */ 1520 (void)path; /* UNUSED */ 1521 return (1);/* Not supported */ 1522 } 1523 #endif 1524 1525 #if defined(HAVE_STATVFS) 1526 static inline __LA_UNUSED void 1527 set_statvfs_transfer_size(struct filesystem *fs, const struct statvfs *sfs) 1528 { 1529 fs->xfer_align = sfs->f_frsize > 0 ? (long)sfs->f_frsize : -1; 1530 fs->max_xfer_size = -1; 1531 #if defined(HAVE_STRUCT_STATVFS_F_IOSIZE) 1532 fs->min_xfer_size = sfs->f_iosize > 0 ? (long)sfs->f_iosize : -1; 1533 fs->incr_xfer_size = sfs->f_iosize > 0 ? (long)sfs->f_iosize : -1; 1534 #else 1535 fs->min_xfer_size = sfs->f_bsize > 0 ? (long)sfs->f_bsize : -1; 1536 fs->incr_xfer_size = sfs->f_bsize > 0 ? (long)sfs->f_bsize : -1; 1537 #endif 1538 } 1539 #endif 1540 1541 #if defined(HAVE_STRUCT_STATFS) 1542 static inline __LA_UNUSED void 1543 set_statfs_transfer_size(struct filesystem *fs, const struct statfs *sfs) 1544 { 1545 fs->xfer_align = sfs->f_bsize > 0 ? (long)sfs->f_bsize : -1; 1546 fs->max_xfer_size = -1; 1547 #if defined(HAVE_STRUCT_STATFS_F_IOSIZE) 1548 fs->min_xfer_size = sfs->f_iosize > 0 ? (long)sfs->f_iosize : -1; 1549 fs->incr_xfer_size = sfs->f_iosize > 0 ? (long)sfs->f_iosize : -1; 1550 #else 1551 fs->min_xfer_size = sfs->f_bsize > 0 ? (long)sfs->f_bsize : -1; 1552 fs->incr_xfer_size = sfs->f_bsize > 0 ? (long)sfs->f_bsize : -1; 1553 #endif 1554 } 1555 #endif 1556 1557 #if defined(HAVE_STRUCT_STATFS) && defined(HAVE_STATFS) && \ 1558 defined(HAVE_FSTATFS) && defined(MNT_LOCAL) && !defined(ST_LOCAL) 1559 1560 /* 1561 * Gather current filesystem properties on FreeBSD, OpenBSD and Mac OS X. 1562 */ 1563 static int 1564 setup_current_filesystem(struct archive_read_disk *a) 1565 { 1566 struct tree *t = a->tree; 1567 struct statfs sfs; 1568 #if defined(HAVE_GETVFSBYNAME) && defined(VFCF_SYNTHETIC) 1569 /* TODO: configure should set GETVFSBYNAME_ARG_TYPE to make 1570 * this accurate; some platforms have both and we need the one that's 1571 * used by getvfsbyname() 1572 * 1573 * Then the following would become: 1574 * #if defined(GETVFSBYNAME_ARG_TYPE) 1575 * GETVFSBYNAME_ARG_TYPE vfc; 1576 * #endif 1577 */ 1578 # if defined(HAVE_STRUCT_XVFSCONF) 1579 struct xvfsconf vfc; 1580 # else 1581 struct vfsconf vfc; 1582 # endif 1583 #endif 1584 int r, xr = 0; 1585 #if !defined(HAVE_STRUCT_STATFS_F_NAMEMAX) 1586 long nm; 1587 #endif 1588 1589 t->current_filesystem->synthetic = -1; 1590 t->current_filesystem->remote = -1; 1591 if (tree_current_is_symblic_link_target(t)) { 1592 #if defined(HAVE_OPENAT) 1593 /* 1594 * Get file system statistics on any directory 1595 * where current is. 1596 */ 1597 int fd = openat(tree_current_dir_fd(t), 1598 tree_current_access_path(t), O_RDONLY | O_CLOEXEC); 1599 __archive_ensure_cloexec_flag(fd); 1600 if (fd < 0) { 1601 archive_set_error(&a->archive, errno, 1602 "openat failed"); 1603 return (ARCHIVE_FAILED); 1604 } 1605 r = fstatfs(fd, &sfs); 1606 if (r == 0) 1607 xr = get_xfer_size(t, fd, NULL); 1608 close(fd); 1609 #else 1610 if (tree_enter_working_dir(t) != 0) { 1611 archive_set_error(&a->archive, errno, "fchdir failed"); 1612 return (ARCHIVE_FAILED); 1613 } 1614 r = statfs(tree_current_access_path(t), &sfs); 1615 if (r == 0) 1616 xr = get_xfer_size(t, -1, tree_current_access_path(t)); 1617 #endif 1618 } else { 1619 r = fstatfs(tree_current_dir_fd(t), &sfs); 1620 if (r == 0) 1621 xr = get_xfer_size(t, tree_current_dir_fd(t), NULL); 1622 } 1623 if (r == -1 || xr == -1) { 1624 archive_set_error(&a->archive, errno, "statfs failed"); 1625 return (ARCHIVE_FAILED); 1626 } else if (xr == 1) { 1627 /* pathconf(_PC_REX_*) operations are not supported. */ 1628 set_statfs_transfer_size(t->current_filesystem, &sfs); 1629 } 1630 if (sfs.f_flags & MNT_LOCAL) 1631 t->current_filesystem->remote = 0; 1632 else 1633 t->current_filesystem->remote = 1; 1634 1635 #if defined(HAVE_GETVFSBYNAME) && defined(VFCF_SYNTHETIC) 1636 r = getvfsbyname(sfs.f_fstypename, &vfc); 1637 if (r == -1) { 1638 archive_set_error(&a->archive, errno, "getvfsbyname failed"); 1639 return (ARCHIVE_FAILED); 1640 } 1641 if (vfc.vfc_flags & VFCF_SYNTHETIC) 1642 t->current_filesystem->synthetic = 1; 1643 else 1644 t->current_filesystem->synthetic = 0; 1645 #endif 1646 1647 #if defined(MNT_NOATIME) 1648 if (sfs.f_flags & MNT_NOATIME) 1649 t->current_filesystem->noatime = 1; 1650 else 1651 #endif 1652 t->current_filesystem->noatime = 0; 1653 1654 #if defined(USE_READDIR_R) 1655 /* Set maximum filename length. */ 1656 #if defined(HAVE_STRUCT_STATFS_F_NAMEMAX) 1657 t->current_filesystem->name_max = sfs.f_namemax; 1658 #else 1659 # if defined(_PC_NAME_MAX) 1660 /* Mac OS X does not have f_namemax in struct statfs. */ 1661 if (tree_current_is_symblic_link_target(t)) { 1662 if (tree_enter_working_dir(t) != 0) { 1663 archive_set_error(&a->archive, errno, "fchdir failed"); 1664 return (ARCHIVE_FAILED); 1665 } 1666 nm = pathconf(tree_current_access_path(t), _PC_NAME_MAX); 1667 } else 1668 nm = fpathconf(tree_current_dir_fd(t), _PC_NAME_MAX); 1669 # else 1670 nm = -1; 1671 # endif 1672 if (nm == -1) 1673 t->current_filesystem->name_max = NAME_MAX; 1674 else 1675 t->current_filesystem->name_max = nm; 1676 #endif 1677 #endif /* USE_READDIR_R */ 1678 return (ARCHIVE_OK); 1679 } 1680 1681 #elif (defined(HAVE_STATVFS) || defined(HAVE_FSTATVFS)) && defined(ST_LOCAL) 1682 1683 /* 1684 * Gather current filesystem properties on NetBSD 1685 */ 1686 static int 1687 setup_current_filesystem(struct archive_read_disk *a) 1688 { 1689 struct tree *t = a->tree; 1690 struct statvfs svfs; 1691 int r, xr = 0; 1692 1693 t->current_filesystem->synthetic = -1; 1694 if (tree_enter_working_dir(t) != 0) { 1695 archive_set_error(&a->archive, errno, "fchdir failed"); 1696 return (ARCHIVE_FAILED); 1697 } 1698 if (tree_current_is_symblic_link_target(t)) { 1699 r = statvfs(tree_current_access_path(t), &svfs); 1700 if (r == 0) 1701 xr = get_xfer_size(t, -1, tree_current_access_path(t)); 1702 } else { 1703 #ifdef HAVE_FSTATVFS 1704 r = fstatvfs(tree_current_dir_fd(t), &svfs); 1705 if (r == 0) 1706 xr = get_xfer_size(t, tree_current_dir_fd(t), NULL); 1707 #else 1708 r = statvfs(".", &svfs); 1709 if (r == 0) 1710 xr = get_xfer_size(t, -1, "."); 1711 #endif 1712 } 1713 if (r == -1 || xr == -1) { 1714 t->current_filesystem->remote = -1; 1715 archive_set_error(&a->archive, errno, "statvfs failed"); 1716 return (ARCHIVE_FAILED); 1717 } else if (xr == 1) { 1718 /* Usually come here unless NetBSD supports _PC_REC_XFER_ALIGN 1719 * for pathconf() function. */ 1720 set_statvfs_transfer_size(t->current_filesystem, &svfs); 1721 } 1722 if (svfs.f_flag & ST_LOCAL) 1723 t->current_filesystem->remote = 0; 1724 else 1725 t->current_filesystem->remote = 1; 1726 1727 #if defined(ST_NOATIME) 1728 if (svfs.f_flag & ST_NOATIME) 1729 t->current_filesystem->noatime = 1; 1730 else 1731 #endif 1732 t->current_filesystem->noatime = 0; 1733 1734 /* Set maximum filename length. */ 1735 t->current_filesystem->name_max = svfs.f_namemax; 1736 return (ARCHIVE_OK); 1737 } 1738 1739 #elif defined(HAVE_SYS_STATFS_H) && defined(HAVE_LINUX_MAGIC_H) &&\ 1740 defined(HAVE_STATFS) && defined(HAVE_FSTATFS) 1741 /* 1742 * Note: statfs is deprecated since LSB 3.2 1743 */ 1744 1745 #ifndef CIFS_SUPER_MAGIC 1746 #define CIFS_SUPER_MAGIC 0xFF534D42 1747 #endif 1748 #ifndef DEVFS_SUPER_MAGIC 1749 #define DEVFS_SUPER_MAGIC 0x1373 1750 #endif 1751 1752 /* 1753 * Gather current filesystem properties on Linux 1754 */ 1755 static int 1756 setup_current_filesystem(struct archive_read_disk *a) 1757 { 1758 struct tree *t = a->tree; 1759 struct statfs sfs; 1760 #if defined(HAVE_STATVFS) 1761 struct statvfs svfs; 1762 #endif 1763 int r, vr = 0, xr = 0; 1764 1765 if (tree_current_is_symblic_link_target(t)) { 1766 #if defined(HAVE_OPENAT) 1767 /* 1768 * Get file system statistics on any directory 1769 * where current is. 1770 */ 1771 int fd = openat(tree_current_dir_fd(t), 1772 tree_current_access_path(t), O_RDONLY | O_CLOEXEC); 1773 __archive_ensure_cloexec_flag(fd); 1774 if (fd < 0) { 1775 archive_set_error(&a->archive, errno, 1776 "openat failed"); 1777 return (ARCHIVE_FAILED); 1778 } 1779 #if defined(HAVE_FSTATVFS) 1780 vr = fstatvfs(fd, &svfs);/* for f_flag, mount flags */ 1781 #endif 1782 r = fstatfs(fd, &sfs); 1783 if (r == 0) 1784 xr = get_xfer_size(t, fd, NULL); 1785 close(fd); 1786 #else 1787 if (tree_enter_working_dir(t) != 0) { 1788 archive_set_error(&a->archive, errno, "fchdir failed"); 1789 return (ARCHIVE_FAILED); 1790 } 1791 #if defined(HAVE_STATVFS) 1792 vr = statvfs(tree_current_access_path(t), &svfs); 1793 #endif 1794 r = statfs(tree_current_access_path(t), &sfs); 1795 if (r == 0) 1796 xr = get_xfer_size(t, -1, tree_current_access_path(t)); 1797 #endif 1798 } else { 1799 #ifdef HAVE_FSTATFS 1800 #if defined(HAVE_FSTATVFS) 1801 vr = fstatvfs(tree_current_dir_fd(t), &svfs); 1802 #endif 1803 r = fstatfs(tree_current_dir_fd(t), &sfs); 1804 if (r == 0) 1805 xr = get_xfer_size(t, tree_current_dir_fd(t), NULL); 1806 #else 1807 if (tree_enter_working_dir(t) != 0) { 1808 archive_set_error(&a->archive, errno, "fchdir failed"); 1809 return (ARCHIVE_FAILED); 1810 } 1811 #if defined(HAVE_STATVFS) 1812 vr = statvfs(".", &svfs); 1813 #endif 1814 r = statfs(".", &sfs); 1815 if (r == 0) 1816 xr = get_xfer_size(t, -1, "."); 1817 #endif 1818 } 1819 if (r == -1 || xr == -1 || vr == -1) { 1820 t->current_filesystem->synthetic = -1; 1821 t->current_filesystem->remote = -1; 1822 archive_set_error(&a->archive, errno, "statfs failed"); 1823 return (ARCHIVE_FAILED); 1824 } else if (xr == 1) { 1825 /* pathconf(_PC_REX_*) operations are not supported. */ 1826 #if defined(HAVE_STATVFS) 1827 set_statvfs_transfer_size(t->current_filesystem, &svfs); 1828 #else 1829 set_statfs_transfer_size(t->current_filesystem, &sfs); 1830 #endif 1831 } 1832 switch (sfs.f_type) { 1833 case AFS_SUPER_MAGIC: 1834 case CIFS_SUPER_MAGIC: 1835 case CODA_SUPER_MAGIC: 1836 case NCP_SUPER_MAGIC:/* NetWare */ 1837 case NFS_SUPER_MAGIC: 1838 case SMB_SUPER_MAGIC: 1839 t->current_filesystem->remote = 1; 1840 t->current_filesystem->synthetic = 0; 1841 break; 1842 case DEVFS_SUPER_MAGIC: 1843 case PROC_SUPER_MAGIC: 1844 case USBDEVICE_SUPER_MAGIC: 1845 t->current_filesystem->remote = 0; 1846 t->current_filesystem->synthetic = 1; 1847 break; 1848 default: 1849 t->current_filesystem->remote = 0; 1850 t->current_filesystem->synthetic = 0; 1851 break; 1852 } 1853 1854 #if defined(ST_NOATIME) 1855 #if defined(HAVE_STATVFS) 1856 if (svfs.f_flag & ST_NOATIME) 1857 #else 1858 if (sfs.f_flags & ST_NOATIME) 1859 #endif 1860 t->current_filesystem->noatime = 1; 1861 else 1862 #endif 1863 t->current_filesystem->noatime = 0; 1864 1865 #if defined(USE_READDIR_R) 1866 /* Set maximum filename length. */ 1867 t->current_filesystem->name_max = sfs.f_namelen; 1868 #endif 1869 return (ARCHIVE_OK); 1870 } 1871 1872 #elif defined(HAVE_SYS_STATVFS_H) &&\ 1873 (defined(HAVE_STATVFS) || defined(HAVE_FSTATVFS)) 1874 1875 /* 1876 * Gather current filesystem properties on other posix platform. 1877 */ 1878 static int 1879 setup_current_filesystem(struct archive_read_disk *a) 1880 { 1881 struct tree *t = a->tree; 1882 struct statvfs svfs; 1883 int r, xr = 0; 1884 1885 t->current_filesystem->synthetic = -1;/* Not supported */ 1886 t->current_filesystem->remote = -1;/* Not supported */ 1887 if (tree_current_is_symblic_link_target(t)) { 1888 #if defined(HAVE_OPENAT) 1889 /* 1890 * Get file system statistics on any directory 1891 * where current is. 1892 */ 1893 int fd = openat(tree_current_dir_fd(t), 1894 tree_current_access_path(t), O_RDONLY | O_CLOEXEC); 1895 __archive_ensure_cloexec_flag(fd); 1896 if (fd < 0) { 1897 archive_set_error(&a->archive, errno, 1898 "openat failed"); 1899 return (ARCHIVE_FAILED); 1900 } 1901 r = fstatvfs(fd, &svfs); 1902 if (r == 0) 1903 xr = get_xfer_size(t, fd, NULL); 1904 close(fd); 1905 #else 1906 if (tree_enter_working_dir(t) != 0) { 1907 archive_set_error(&a->archive, errno, "fchdir failed"); 1908 return (ARCHIVE_FAILED); 1909 } 1910 r = statvfs(tree_current_access_path(t), &svfs); 1911 if (r == 0) 1912 xr = get_xfer_size(t, -1, tree_current_access_path(t)); 1913 #endif 1914 } else { 1915 #ifdef HAVE_FSTATVFS 1916 r = fstatvfs(tree_current_dir_fd(t), &svfs); 1917 if (r == 0) 1918 xr = get_xfer_size(t, tree_current_dir_fd(t), NULL); 1919 #else 1920 if (tree_enter_working_dir(t) != 0) { 1921 archive_set_error(&a->archive, errno, "fchdir failed"); 1922 return (ARCHIVE_FAILED); 1923 } 1924 r = statvfs(".", &svfs); 1925 if (r == 0) 1926 xr = get_xfer_size(t, -1, "."); 1927 #endif 1928 } 1929 if (r == -1 || xr == -1) { 1930 t->current_filesystem->synthetic = -1; 1931 t->current_filesystem->remote = -1; 1932 archive_set_error(&a->archive, errno, "statvfs failed"); 1933 return (ARCHIVE_FAILED); 1934 } else if (xr == 1) { 1935 /* pathconf(_PC_REX_*) operations are not supported. */ 1936 set_statvfs_transfer_size(t->current_filesystem, &svfs); 1937 } 1938 1939 #if defined(ST_NOATIME) 1940 if (svfs.f_flag & ST_NOATIME) 1941 t->current_filesystem->noatime = 1; 1942 else 1943 #endif 1944 t->current_filesystem->noatime = 0; 1945 1946 #if defined(USE_READDIR_R) 1947 /* Set maximum filename length. */ 1948 t->current_filesystem->name_max = svfs.f_namemax; 1949 #endif 1950 return (ARCHIVE_OK); 1951 } 1952 1953 #else 1954 1955 /* 1956 * Generic: Gather current filesystem properties. 1957 * TODO: Is this generic function really needed? 1958 */ 1959 static int 1960 setup_current_filesystem(struct archive_read_disk *a) 1961 { 1962 struct tree *t = a->tree; 1963 #if defined(_PC_NAME_MAX) && defined(USE_READDIR_R) 1964 long nm; 1965 #endif 1966 t->current_filesystem->synthetic = -1;/* Not supported */ 1967 t->current_filesystem->remote = -1;/* Not supported */ 1968 t->current_filesystem->noatime = 0; 1969 (void)get_xfer_size(t, -1, ".");/* Dummy call to avoid build error. */ 1970 t->current_filesystem->xfer_align = -1;/* Unknown */ 1971 t->current_filesystem->max_xfer_size = -1; 1972 t->current_filesystem->min_xfer_size = -1; 1973 t->current_filesystem->incr_xfer_size = -1; 1974 1975 #if defined(USE_READDIR_R) 1976 /* Set maximum filename length. */ 1977 # if defined(_PC_NAME_MAX) 1978 if (tree_current_is_symblic_link_target(t)) { 1979 if (tree_enter_working_dir(t) != 0) { 1980 archive_set_error(&a->archive, errno, "fchdir failed"); 1981 return (ARCHIVE_FAILED); 1982 } 1983 nm = pathconf(tree_current_access_path(t), _PC_NAME_MAX); 1984 } else 1985 nm = fpathconf(tree_current_dir_fd(t), _PC_NAME_MAX); 1986 if (nm == -1) 1987 # endif /* _PC_NAME_MAX */ 1988 /* 1989 * Some systems (HP-UX or others?) incorrectly defined 1990 * NAME_MAX macro to be a smaller value. 1991 */ 1992 # if defined(NAME_MAX) && NAME_MAX >= 255 1993 t->current_filesystem->name_max = NAME_MAX; 1994 # else 1995 /* No way to get a trusted value of maximum filename 1996 * length. */ 1997 t->current_filesystem->name_max = PATH_MAX; 1998 # endif /* NAME_MAX */ 1999 # if defined(_PC_NAME_MAX) 2000 else 2001 t->current_filesystem->name_max = nm; 2002 # endif /* _PC_NAME_MAX */ 2003 #endif /* USE_READDIR_R */ 2004 return (ARCHIVE_OK); 2005 } 2006 2007 #endif 2008 2009 static int 2010 close_and_restore_time(int fd, struct tree *t, struct restore_time *rt) 2011 { 2012 #ifndef HAVE_UTIMES 2013 (void)t; /* UNUSED */ 2014 (void)rt; /* UNUSED */ 2015 return (close(fd)); 2016 #else 2017 #if defined(HAVE_FUTIMENS) && !defined(__CYGWIN__) 2018 struct timespec timespecs[2]; 2019 #endif 2020 struct timeval times[2]; 2021 2022 if ((t->flags & needsRestoreTimes) == 0 || rt->noatime) { 2023 if (fd >= 0) 2024 return (close(fd)); 2025 else 2026 return (0); 2027 } 2028 2029 #if defined(HAVE_FUTIMENS) && !defined(__CYGWIN__) 2030 timespecs[1].tv_sec = rt->mtime; 2031 timespecs[1].tv_nsec = rt->mtime_nsec; 2032 2033 timespecs[0].tv_sec = rt->atime; 2034 timespecs[0].tv_nsec = rt->atime_nsec; 2035 /* futimens() is defined in POSIX.1-2008. */ 2036 if (futimens(fd, timespecs) == 0) 2037 return (close(fd)); 2038 #endif 2039 2040 times[1].tv_sec = rt->mtime; 2041 times[1].tv_usec = rt->mtime_nsec / 1000; 2042 2043 times[0].tv_sec = rt->atime; 2044 times[0].tv_usec = rt->atime_nsec / 1000; 2045 2046 #if !defined(HAVE_FUTIMENS) && defined(HAVE_FUTIMES) && !defined(__CYGWIN__) 2047 if (futimes(fd, times) == 0) 2048 return (close(fd)); 2049 #endif 2050 close(fd); 2051 #if defined(HAVE_FUTIMESAT) 2052 if (futimesat(tree_current_dir_fd(t), rt->name, times) == 0) 2053 return (0); 2054 #endif 2055 #ifdef HAVE_LUTIMES 2056 if (lutimes(rt->name, times) != 0) 2057 #else 2058 if (AE_IFLNK != rt->filetype && utimes(rt->name, times) != 0) 2059 #endif 2060 return (-1); 2061 #endif 2062 return (0); 2063 } 2064 2065 static int 2066 open_on_current_dir(struct tree *t, const char *path, int flags) 2067 { 2068 #ifdef HAVE_OPENAT 2069 return (openat(tree_current_dir_fd(t), path, flags)); 2070 #else 2071 if (tree_enter_working_dir(t) != 0) 2072 return (-1); 2073 return (open(path, flags)); 2074 #endif 2075 } 2076 2077 static int 2078 tree_dup(int fd) 2079 { 2080 int new_fd; 2081 #ifdef F_DUPFD_CLOEXEC 2082 static volatile int can_dupfd_cloexec = 1; 2083 2084 if (can_dupfd_cloexec) { 2085 new_fd = fcntl(fd, F_DUPFD_CLOEXEC, 0); 2086 if (new_fd != -1) 2087 return (new_fd); 2088 /* Linux 2.6.18 - 2.6.23 declare F_DUPFD_CLOEXEC, 2089 * but it cannot be used. So we have to try dup(). */ 2090 /* We won't try F_DUPFD_CLOEXEC. */ 2091 can_dupfd_cloexec = 0; 2092 } 2093 #endif /* F_DUPFD_CLOEXEC */ 2094 new_fd = dup(fd); 2095 __archive_ensure_cloexec_flag(new_fd); 2096 return (new_fd); 2097 } 2098 2099 /* 2100 * Add a directory path to the current stack. 2101 */ 2102 static void 2103 tree_push(struct tree *t, const char *path, int filesystem_id, 2104 int64_t dev, int64_t ino, struct restore_time *rt) 2105 { 2106 struct tree_entry *te; 2107 2108 te = calloc(1, sizeof(*te)); 2109 te->next = t->stack; 2110 te->parent = t->current; 2111 if (te->parent) 2112 te->depth = te->parent->depth + 1; 2113 t->stack = te; 2114 archive_string_init(&te->name); 2115 te->symlink_parent_fd = -1; 2116 archive_strcpy(&te->name, path); 2117 te->flags = needsDescent | needsOpen | needsAscent; 2118 te->filesystem_id = filesystem_id; 2119 te->dev = dev; 2120 te->ino = ino; 2121 te->dirname_length = t->dirname_length; 2122 te->restore_time.name = te->name.s; 2123 if (rt != NULL) { 2124 te->restore_time.mtime = rt->mtime; 2125 te->restore_time.mtime_nsec = rt->mtime_nsec; 2126 te->restore_time.atime = rt->atime; 2127 te->restore_time.atime_nsec = rt->atime_nsec; 2128 te->restore_time.filetype = rt->filetype; 2129 te->restore_time.noatime = rt->noatime; 2130 } 2131 } 2132 2133 /* 2134 * Append a name to the current dir path. 2135 */ 2136 static void 2137 tree_append(struct tree *t, const char *name, size_t name_length) 2138 { 2139 size_t size_needed; 2140 2141 t->path.s[t->dirname_length] = '\0'; 2142 t->path.length = t->dirname_length; 2143 /* Strip trailing '/' from name, unless entire name is "/". */ 2144 while (name_length > 1 && name[name_length - 1] == '/') 2145 name_length--; 2146 2147 /* Resize pathname buffer as needed. */ 2148 size_needed = name_length + t->dirname_length + 2; 2149 archive_string_ensure(&t->path, size_needed); 2150 /* Add a separating '/' if it's needed. */ 2151 if (t->dirname_length > 0 && t->path.s[archive_strlen(&t->path)-1] != '/') 2152 archive_strappend_char(&t->path, '/'); 2153 t->basename = t->path.s + archive_strlen(&t->path); 2154 archive_strncat(&t->path, name, name_length); 2155 t->restore_time.name = t->basename; 2156 } 2157 2158 /* 2159 * Open a directory tree for traversal. 2160 */ 2161 static struct tree * 2162 tree_open(const char *path, int symlink_mode, int restore_time) 2163 { 2164 struct tree *t; 2165 2166 if ((t = calloc(1, sizeof(*t))) == NULL) 2167 return (NULL); 2168 archive_string_init(&t->path); 2169 archive_string_ensure(&t->path, 31); 2170 t->initial_symlink_mode = symlink_mode; 2171 return (tree_reopen(t, path, restore_time)); 2172 } 2173 2174 static struct tree * 2175 tree_reopen(struct tree *t, const char *path, int restore_time) 2176 { 2177 #if defined(O_PATH) 2178 /* Linux */ 2179 const int o_flag = O_PATH; 2180 #elif defined(O_SEARCH) 2181 /* SunOS */ 2182 const int o_flag = O_SEARCH; 2183 #elif defined(__FreeBSD__) && defined(O_EXEC) 2184 /* FreeBSD */ 2185 const int o_flag = O_EXEC; 2186 #endif 2187 2188 t->flags = (restore_time != 0)?needsRestoreTimes:0; 2189 t->flags |= onInitialDir; 2190 t->visit_type = 0; 2191 t->tree_errno = 0; 2192 t->dirname_length = 0; 2193 t->depth = 0; 2194 t->descend = 0; 2195 t->current = NULL; 2196 t->d = INVALID_DIR_HANDLE; 2197 t->symlink_mode = t->initial_symlink_mode; 2198 archive_string_empty(&t->path); 2199 t->entry_fd = -1; 2200 t->entry_eof = 0; 2201 t->entry_remaining_bytes = 0; 2202 t->initial_filesystem_id = -1; 2203 2204 /* First item is set up a lot like a symlink traversal. */ 2205 tree_push(t, path, 0, 0, 0, NULL); 2206 t->stack->flags = needsFirstVisit; 2207 t->maxOpenCount = t->openCount = 1; 2208 t->initial_dir_fd = open(".", O_RDONLY | O_CLOEXEC); 2209 #if defined(O_PATH) || defined(O_SEARCH) || \ 2210 (defined(__FreeBSD__) && defined(O_EXEC)) 2211 /* 2212 * Most likely reason to fail opening "." is that it's not readable, 2213 * so try again for execute. The consequences of not opening this are 2214 * unhelpful and unnecessary errors later. 2215 */ 2216 if (t->initial_dir_fd < 0) 2217 t->initial_dir_fd = open(".", o_flag | O_CLOEXEC); 2218 #endif 2219 __archive_ensure_cloexec_flag(t->initial_dir_fd); 2220 t->working_dir_fd = tree_dup(t->initial_dir_fd); 2221 return (t); 2222 } 2223 2224 static int 2225 tree_descent(struct tree *t) 2226 { 2227 int flag, new_fd, r = 0; 2228 2229 t->dirname_length = archive_strlen(&t->path); 2230 flag = O_RDONLY | O_CLOEXEC; 2231 #if defined(O_DIRECTORY) 2232 flag |= O_DIRECTORY; 2233 #endif 2234 new_fd = open_on_current_dir(t, t->stack->name.s, flag); 2235 __archive_ensure_cloexec_flag(new_fd); 2236 if (new_fd < 0) { 2237 t->tree_errno = errno; 2238 r = TREE_ERROR_DIR; 2239 } else { 2240 t->depth++; 2241 /* If it is a link, set up fd for the ascent. */ 2242 if (t->stack->flags & isDirLink) { 2243 t->stack->symlink_parent_fd = t->working_dir_fd; 2244 t->openCount++; 2245 if (t->openCount > t->maxOpenCount) 2246 t->maxOpenCount = t->openCount; 2247 } else 2248 close(t->working_dir_fd); 2249 /* Renew the current working directory. */ 2250 t->working_dir_fd = new_fd; 2251 t->flags &= ~onWorkingDir; 2252 } 2253 return (r); 2254 } 2255 2256 /* 2257 * We've finished a directory; ascend back to the parent. 2258 */ 2259 static int 2260 tree_ascend(struct tree *t) 2261 { 2262 struct tree_entry *te; 2263 int new_fd, r = 0, prev_dir_fd; 2264 2265 te = t->stack; 2266 prev_dir_fd = t->working_dir_fd; 2267 if (te->flags & isDirLink) 2268 new_fd = te->symlink_parent_fd; 2269 else { 2270 new_fd = open_on_current_dir(t, "..", O_RDONLY | O_CLOEXEC); 2271 __archive_ensure_cloexec_flag(new_fd); 2272 } 2273 if (new_fd < 0) { 2274 t->tree_errno = errno; 2275 r = TREE_ERROR_FATAL; 2276 } else { 2277 /* Renew the current working directory. */ 2278 t->working_dir_fd = new_fd; 2279 t->flags &= ~onWorkingDir; 2280 /* Current directory has been changed, we should 2281 * close an fd of previous working directory. */ 2282 close_and_restore_time(prev_dir_fd, t, &te->restore_time); 2283 if (te->flags & isDirLink) { 2284 t->openCount--; 2285 te->symlink_parent_fd = -1; 2286 } 2287 t->depth--; 2288 } 2289 return (r); 2290 } 2291 2292 /* 2293 * Return to the initial directory where tree_open() was performed. 2294 */ 2295 static int 2296 tree_enter_initial_dir(struct tree *t) 2297 { 2298 int r = 0; 2299 2300 if ((t->flags & onInitialDir) == 0) { 2301 r = fchdir(t->initial_dir_fd); 2302 if (r == 0) { 2303 t->flags &= ~onWorkingDir; 2304 t->flags |= onInitialDir; 2305 } 2306 } 2307 return (r); 2308 } 2309 2310 /* 2311 * Restore working directory of directory traversals. 2312 */ 2313 static int 2314 tree_enter_working_dir(struct tree *t) 2315 { 2316 int r = 0; 2317 2318 /* 2319 * Change the current directory if really needed. 2320 * Sometimes this is unneeded when we did not do 2321 * descent. 2322 */ 2323 if (t->depth > 0 && (t->flags & onWorkingDir) == 0) { 2324 r = fchdir(t->working_dir_fd); 2325 if (r == 0) { 2326 t->flags &= ~onInitialDir; 2327 t->flags |= onWorkingDir; 2328 } 2329 } 2330 return (r); 2331 } 2332 2333 static int 2334 tree_current_dir_fd(struct tree *t) 2335 { 2336 return (t->working_dir_fd); 2337 } 2338 2339 /* 2340 * Pop the working stack. 2341 */ 2342 static void 2343 tree_pop(struct tree *t) 2344 { 2345 struct tree_entry *te; 2346 2347 t->path.s[t->dirname_length] = '\0'; 2348 t->path.length = t->dirname_length; 2349 if (t->stack == t->current && t->current != NULL) 2350 t->current = t->current->parent; 2351 te = t->stack; 2352 t->stack = te->next; 2353 t->dirname_length = te->dirname_length; 2354 t->basename = t->path.s + t->dirname_length; 2355 while (t->basename[0] == '/') 2356 t->basename++; 2357 archive_string_free(&te->name); 2358 free(te); 2359 } 2360 2361 /* 2362 * Get the next item in the tree traversal. 2363 */ 2364 static int 2365 tree_next(struct tree *t) 2366 { 2367 int r; 2368 2369 while (t->stack != NULL) { 2370 /* If there's an open dir, get the next entry from there. */ 2371 if (t->d != INVALID_DIR_HANDLE) { 2372 r = tree_dir_next_posix(t); 2373 if (r == 0) 2374 continue; 2375 return (r); 2376 } 2377 2378 if (t->stack->flags & needsFirstVisit) { 2379 /* Top stack item needs a regular visit. */ 2380 t->current = t->stack; 2381 tree_append(t, t->stack->name.s, 2382 archive_strlen(&(t->stack->name))); 2383 /* t->dirname_length = t->path_length; */ 2384 /* tree_pop(t); */ 2385 t->stack->flags &= ~needsFirstVisit; 2386 return (t->visit_type = TREE_REGULAR); 2387 } else if (t->stack->flags & needsDescent) { 2388 /* Top stack item is dir to descend into. */ 2389 t->current = t->stack; 2390 tree_append(t, t->stack->name.s, 2391 archive_strlen(&(t->stack->name))); 2392 t->stack->flags &= ~needsDescent; 2393 r = tree_descent(t); 2394 if (r != 0) { 2395 tree_pop(t); 2396 t->visit_type = r; 2397 } else 2398 t->visit_type = TREE_POSTDESCENT; 2399 return (t->visit_type); 2400 } else if (t->stack->flags & needsOpen) { 2401 t->stack->flags &= ~needsOpen; 2402 r = tree_dir_next_posix(t); 2403 if (r == 0) 2404 continue; 2405 return (r); 2406 } else if (t->stack->flags & needsAscent) { 2407 /* Top stack item is dir and we're done with it. */ 2408 r = tree_ascend(t); 2409 tree_pop(t); 2410 t->visit_type = r != 0 ? r : TREE_POSTASCENT; 2411 return (t->visit_type); 2412 } else { 2413 /* Top item on stack is dead. */ 2414 tree_pop(t); 2415 t->flags &= ~hasLstat; 2416 t->flags &= ~hasStat; 2417 } 2418 } 2419 return (t->visit_type = 0); 2420 } 2421 2422 static int 2423 tree_dir_next_posix(struct tree *t) 2424 { 2425 int r; 2426 const char *name; 2427 size_t namelen; 2428 2429 if (t->d == NULL) { 2430 #if defined(USE_READDIR_R) 2431 size_t dirent_size; 2432 #endif 2433 2434 #if defined(HAVE_FDOPENDIR) 2435 t->d = fdopendir(tree_dup(t->working_dir_fd)); 2436 #else /* HAVE_FDOPENDIR */ 2437 if (tree_enter_working_dir(t) == 0) { 2438 t->d = opendir("."); 2439 #if HAVE_DIRFD || defined(dirfd) 2440 __archive_ensure_cloexec_flag(dirfd(t->d)); 2441 #endif 2442 } 2443 #endif /* HAVE_FDOPENDIR */ 2444 if (t->d == NULL) { 2445 r = tree_ascend(t); /* Undo "chdir" */ 2446 tree_pop(t); 2447 t->tree_errno = errno; 2448 t->visit_type = r != 0 ? r : TREE_ERROR_DIR; 2449 return (t->visit_type); 2450 } 2451 #if defined(USE_READDIR_R) 2452 dirent_size = offsetof(struct dirent, d_name) + 2453 t->filesystem_table[t->current->filesystem_id].name_max + 1; 2454 if (t->dirent == NULL || t->dirent_allocated < dirent_size) { 2455 free(t->dirent); 2456 t->dirent = malloc(dirent_size); 2457 if (t->dirent == NULL) { 2458 closedir(t->d); 2459 t->d = INVALID_DIR_HANDLE; 2460 (void)tree_ascend(t); 2461 tree_pop(t); 2462 t->tree_errno = ENOMEM; 2463 t->visit_type = TREE_ERROR_DIR; 2464 return (t->visit_type); 2465 } 2466 t->dirent_allocated = dirent_size; 2467 } 2468 #endif /* USE_READDIR_R */ 2469 } 2470 for (;;) { 2471 errno = 0; 2472 #if defined(USE_READDIR_R) 2473 r = readdir_r(t->d, t->dirent, &t->de); 2474 #ifdef _AIX 2475 /* Note: According to the man page, return value 9 indicates 2476 * that the readdir_r was not successful and the error code 2477 * is set to the global errno variable. And then if the end 2478 * of directory entries was reached, the return value is 9 2479 * and the third parameter is set to NULL and errno is 2480 * unchanged. */ 2481 if (r == 9) 2482 r = errno; 2483 #endif /* _AIX */ 2484 if (r != 0 || t->de == NULL) { 2485 #else 2486 t->de = readdir(t->d); 2487 if (t->de == NULL) { 2488 r = errno; 2489 #endif 2490 closedir(t->d); 2491 t->d = INVALID_DIR_HANDLE; 2492 if (r != 0) { 2493 t->tree_errno = r; 2494 t->visit_type = TREE_ERROR_DIR; 2495 return (t->visit_type); 2496 } else 2497 return (0); 2498 } 2499 name = t->de->d_name; 2500 namelen = D_NAMELEN(t->de); 2501 t->flags &= ~hasLstat; 2502 t->flags &= ~hasStat; 2503 if (name[0] == '.' && name[1] == '\0') 2504 continue; 2505 if (name[0] == '.' && name[1] == '.' && name[2] == '\0') 2506 continue; 2507 tree_append(t, name, namelen); 2508 return (t->visit_type = TREE_REGULAR); 2509 } 2510 } 2511 2512 2513 /* 2514 * Get the stat() data for the entry just returned from tree_next(). 2515 */ 2516 static const struct stat * 2517 tree_current_stat(struct tree *t) 2518 { 2519 if (!(t->flags & hasStat)) { 2520 #ifdef HAVE_FSTATAT 2521 if (fstatat(tree_current_dir_fd(t), 2522 tree_current_access_path(t), &t->st, 0) != 0) 2523 #else 2524 if (tree_enter_working_dir(t) != 0) 2525 return NULL; 2526 if (la_stat(tree_current_access_path(t), &t->st) != 0) 2527 #endif 2528 return NULL; 2529 t->flags |= hasStat; 2530 } 2531 return (&t->st); 2532 } 2533 2534 /* 2535 * Get the lstat() data for the entry just returned from tree_next(). 2536 */ 2537 static const struct stat * 2538 tree_current_lstat(struct tree *t) 2539 { 2540 if (!(t->flags & hasLstat)) { 2541 #ifdef HAVE_FSTATAT 2542 if (fstatat(tree_current_dir_fd(t), 2543 tree_current_access_path(t), &t->lst, 2544 AT_SYMLINK_NOFOLLOW) != 0) 2545 #else 2546 if (tree_enter_working_dir(t) != 0) 2547 return NULL; 2548 if (lstat(tree_current_access_path(t), &t->lst) != 0) 2549 #endif 2550 return NULL; 2551 t->flags |= hasLstat; 2552 } 2553 return (&t->lst); 2554 } 2555 2556 /* 2557 * Test whether current entry is a dir or link to a dir. 2558 */ 2559 static int 2560 tree_current_is_dir(struct tree *t) 2561 { 2562 const struct stat *st; 2563 /* 2564 * If we already have lstat() info, then try some 2565 * cheap tests to determine if this is a dir. 2566 */ 2567 if (t->flags & hasLstat) { 2568 /* If lstat() says it's a dir, it must be a dir. */ 2569 st = tree_current_lstat(t); 2570 if (st == NULL) 2571 return 0; 2572 if (S_ISDIR(st->st_mode)) 2573 return 1; 2574 /* Not a dir; might be a link to a dir. */ 2575 /* If it's not a link, then it's not a link to a dir. */ 2576 if (!S_ISLNK(st->st_mode)) 2577 return 0; 2578 /* 2579 * It's a link, but we don't know what it's a link to, 2580 * so we'll have to use stat(). 2581 */ 2582 } 2583 2584 st = tree_current_stat(t); 2585 /* If we can't stat it, it's not a dir. */ 2586 if (st == NULL) 2587 return 0; 2588 /* Use the definitive test. Hopefully this is cached. */ 2589 return (S_ISDIR(st->st_mode)); 2590 } 2591 2592 /* 2593 * Test whether current entry is a physical directory. Usually, we 2594 * already have at least one of stat() or lstat() in memory, so we 2595 * use tricks to try to avoid an extra trip to the disk. 2596 */ 2597 static int 2598 tree_current_is_physical_dir(struct tree *t) 2599 { 2600 const struct stat *st; 2601 2602 /* 2603 * If stat() says it isn't a dir, then it's not a dir. 2604 * If stat() data is cached, this check is free, so do it first. 2605 */ 2606 if (t->flags & hasStat) { 2607 st = tree_current_stat(t); 2608 if (st == NULL) 2609 return (0); 2610 if (!S_ISDIR(st->st_mode)) 2611 return (0); 2612 } 2613 2614 /* 2615 * Either stat() said it was a dir (in which case, we have 2616 * to determine whether it's really a link to a dir) or 2617 * stat() info wasn't available. So we use lstat(), which 2618 * hopefully is already cached. 2619 */ 2620 2621 st = tree_current_lstat(t); 2622 /* If we can't stat it, it's not a dir. */ 2623 if (st == NULL) 2624 return 0; 2625 /* Use the definitive test. Hopefully this is cached. */ 2626 return (S_ISDIR(st->st_mode)); 2627 } 2628 2629 /* 2630 * Test whether the same file has been in the tree as its parent. 2631 */ 2632 static int 2633 tree_target_is_same_as_parent(struct tree *t, const struct stat *st) 2634 { 2635 struct tree_entry *te; 2636 2637 for (te = t->current->parent; te != NULL; te = te->parent) { 2638 if (te->dev == (int64_t)st->st_dev && 2639 te->ino == (int64_t)st->st_ino) 2640 return (1); 2641 } 2642 return (0); 2643 } 2644 2645 /* 2646 * Test whether the current file is symbolic link target and 2647 * on the other filesystem. 2648 */ 2649 static int 2650 tree_current_is_symblic_link_target(struct tree *t) 2651 { 2652 static const struct stat *lst, *st; 2653 2654 lst = tree_current_lstat(t); 2655 st = tree_current_stat(t); 2656 return (st != NULL && lst != NULL && 2657 (int64_t)st->st_dev == t->current_filesystem->dev && 2658 st->st_dev != lst->st_dev); 2659 } 2660 2661 /* 2662 * Return the access path for the entry just returned from tree_next(). 2663 */ 2664 static const char * 2665 tree_current_access_path(struct tree *t) 2666 { 2667 return (t->basename); 2668 } 2669 2670 /* 2671 * Return the full path for the entry just returned from tree_next(). 2672 */ 2673 static const char * 2674 tree_current_path(struct tree *t) 2675 { 2676 return (t->path.s); 2677 } 2678 2679 /* 2680 * Terminate the traversal. 2681 */ 2682 static void 2683 tree_close(struct tree *t) 2684 { 2685 2686 if (t == NULL) 2687 return; 2688 if (t->entry_fd >= 0) { 2689 close_and_restore_time(t->entry_fd, t, &t->restore_time); 2690 t->entry_fd = -1; 2691 } 2692 /* Close the handle of readdir(). */ 2693 if (t->d != INVALID_DIR_HANDLE) { 2694 closedir(t->d); 2695 t->d = INVALID_DIR_HANDLE; 2696 } 2697 /* Release anything remaining in the stack. */ 2698 while (t->stack != NULL) { 2699 if (t->stack->flags & isDirLink) 2700 close(t->stack->symlink_parent_fd); 2701 tree_pop(t); 2702 } 2703 if (t->working_dir_fd >= 0) { 2704 close(t->working_dir_fd); 2705 t->working_dir_fd = -1; 2706 } 2707 if (t->initial_dir_fd >= 0) { 2708 close(t->initial_dir_fd); 2709 t->initial_dir_fd = -1; 2710 } 2711 } 2712 2713 /* 2714 * Release any resources. 2715 */ 2716 static void 2717 tree_free(struct tree *t) 2718 { 2719 int i; 2720 2721 if (t == NULL) 2722 return; 2723 archive_string_free(&t->path); 2724 #if defined(USE_READDIR_R) 2725 free(t->dirent); 2726 #endif 2727 free(t->sparse_list); 2728 for (i = 0; i < t->max_filesystem_id; i++) 2729 free(t->filesystem_table[i].allocation_ptr); 2730 free(t->filesystem_table); 2731 free(t); 2732 } 2733 2734 #endif 2735