]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/libarchive/libarchive/archive_write_disk_posix.c
MFC r356212,r356366,r356416,r357785
[FreeBSD/FreeBSD.git] / contrib / libarchive / libarchive / archive_write_disk_posix.c
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
172 struct 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
220 struct 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
358 static int      la_opendirat(int, const char *);
359 static int      la_mktemp(struct archive_write_disk *);
360 static void     fsobj_error(int *, struct archive_string *, int, const char *,
361                     const char *);
362 static int      check_symlinks_fsobj(char *, int *, struct archive_string *,
363                     int);
364 static int      check_symlinks(struct archive_write_disk *);
365 static int      create_filesystem_object(struct archive_write_disk *);
366 static struct fixup_entry *current_fixup(struct archive_write_disk *,
367                     const char *pathname);
368 #if defined(HAVE_FCHDIR) && defined(PATH_MAX)
369 static void     edit_deep_directories(struct archive_write_disk *ad);
370 #endif
371 static int      cleanup_pathname_fsobj(char *, int *, struct archive_string *,
372                     int);
373 static int      cleanup_pathname(struct archive_write_disk *);
374 static int      create_dir(struct archive_write_disk *, char *);
375 static int      create_parent_dir(struct archive_write_disk *, char *);
376 static ssize_t  hfs_write_data_block(struct archive_write_disk *,
377                     const char *, size_t);
378 static int      fixup_appledouble(struct archive_write_disk *, const char *);
379 static int      older(struct stat *, struct archive_entry *);
380 static int      restore_entry(struct archive_write_disk *);
381 static int      set_mac_metadata(struct archive_write_disk *, const char *,
382                                  const void *, size_t);
383 static int      set_xattrs(struct archive_write_disk *);
384 static int      clear_nochange_fflags(struct archive_write_disk *);
385 static int      set_fflags(struct archive_write_disk *);
386 static 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);
389 static int      set_ownership(struct archive_write_disk *);
390 static int      set_mode(struct archive_write_disk *, int mode);
391 static int      set_time(int, int, const char *, time_t, long, time_t, long);
392 static int      set_times(struct archive_write_disk *, int, int, const char *,
393                     time_t, long, time_t, long, time_t, long, time_t, long);
394 static int      set_times_from_entry(struct archive_write_disk *);
395 static struct fixup_entry *sort_dir_list(struct fixup_entry *p);
396 static ssize_t  write_data_block(struct archive_write_disk *,
397                     const char *, size_t);
398
399 static struct archive_vtable *archive_write_disk_vtable(void);
400
401 static int      _archive_write_disk_close(struct archive *);
402 static int      _archive_write_disk_free(struct archive *);
403 static int      _archive_write_disk_header(struct archive *,
404                     struct archive_entry *);
405 static int64_t  _archive_write_disk_filter_bytes(struct archive *, int);
406 static int      _archive_write_disk_finish_entry(struct archive *);
407 static ssize_t  _archive_write_disk_data(struct archive *, const void *,
408                     size_t);
409 static ssize_t  _archive_write_disk_data_block(struct archive *, const void *,
410                     size_t, int64_t);
411
412 static int
413 la_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
436 static int
437 la_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
467 static int
468 lazy_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
492 static struct archive_vtable *
493 archive_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
512 static 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
523 int
524 archive_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  */
544 static 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
901 int
902 archive_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
913 static ssize_t
914 write_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  */
1007 static int
1008 hfs_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  */
1046 static int
1047 hfs_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  */
1103 static int
1104 hfs_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
1123 static int
1124 hfs_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
1135 static int
1136 hfs_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
1160 static size_t
1161 hfs_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
1178 static int
1179 hfs_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
1198 static int
1199 hfs_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
1259 static int
1260 hfs_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
1410 static ssize_t
1411 hfs_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
1536 static ssize_t
1537 hfs_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
1601 static ssize_t
1602 hfs_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
1609 static 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
1639 static 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
1652 static 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
1851 finish_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                                     "rename failed");
1860                                 ret = ARCHIVE_FATAL;
1861                         }
1862                         a->tmpname = NULL;
1863                 }
1864         }
1865         /* If there's an entry, we can release it now. */
1866         archive_entry_free(a->entry);
1867         a->entry = NULL;
1868         a->archive.state = ARCHIVE_STATE_HEADER;
1869         return (ret);
1870 }
1871
1872 int
1873 archive_write_disk_set_group_lookup(struct archive *_a,
1874     void *private_data,
1875     la_int64_t (*lookup_gid)(void *private, const char *gname, la_int64_t gid),
1876     void (*cleanup_gid)(void *private))
1877 {
1878         struct archive_write_disk *a = (struct archive_write_disk *)_a;
1879         archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC,
1880             ARCHIVE_STATE_ANY, "archive_write_disk_set_group_lookup");
1881
1882         if (a->cleanup_gid != NULL && a->lookup_gid_data != NULL)
1883                 (a->cleanup_gid)(a->lookup_gid_data);
1884
1885         a->lookup_gid = lookup_gid;
1886         a->cleanup_gid = cleanup_gid;
1887         a->lookup_gid_data = private_data;
1888         return (ARCHIVE_OK);
1889 }
1890
1891 int
1892 archive_write_disk_set_user_lookup(struct archive *_a,
1893     void *private_data,
1894     int64_t (*lookup_uid)(void *private, const char *uname, int64_t uid),
1895     void (*cleanup_uid)(void *private))
1896 {
1897         struct archive_write_disk *a = (struct archive_write_disk *)_a;
1898         archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC,
1899             ARCHIVE_STATE_ANY, "archive_write_disk_set_user_lookup");
1900
1901         if (a->cleanup_uid != NULL && a->lookup_uid_data != NULL)
1902                 (a->cleanup_uid)(a->lookup_uid_data);
1903
1904         a->lookup_uid = lookup_uid;
1905         a->cleanup_uid = cleanup_uid;
1906         a->lookup_uid_data = private_data;
1907         return (ARCHIVE_OK);
1908 }
1909
1910 int64_t
1911 archive_write_disk_gid(struct archive *_a, const char *name, la_int64_t id)
1912 {
1913        struct archive_write_disk *a = (struct archive_write_disk *)_a;
1914        archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC,
1915            ARCHIVE_STATE_ANY, "archive_write_disk_gid");
1916        if (a->lookup_gid)
1917                return (a->lookup_gid)(a->lookup_gid_data, name, id);
1918        return (id);
1919 }
1920  
1921 int64_t
1922 archive_write_disk_uid(struct archive *_a, const char *name, la_int64_t id)
1923 {
1924         struct archive_write_disk *a = (struct archive_write_disk *)_a;
1925         archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC,
1926             ARCHIVE_STATE_ANY, "archive_write_disk_uid");
1927         if (a->lookup_uid)
1928                 return (a->lookup_uid)(a->lookup_uid_data, name, id);
1929         return (id);
1930 }
1931
1932 /*
1933  * Create a new archive_write_disk object and initialize it with global state.
1934  */
1935 struct archive *
1936 archive_write_disk_new(void)
1937 {
1938         struct archive_write_disk *a;
1939
1940         a = (struct archive_write_disk *)calloc(1, sizeof(*a));
1941         if (a == NULL)
1942                 return (NULL);
1943         a->archive.magic = ARCHIVE_WRITE_DISK_MAGIC;
1944         /* We're ready to write a header immediately. */
1945         a->archive.state = ARCHIVE_STATE_HEADER;
1946         a->archive.vtable = archive_write_disk_vtable();
1947         a->start_time = time(NULL);
1948         /* Query and restore the umask. */
1949         umask(a->user_umask = umask(0));
1950 #ifdef HAVE_GETEUID
1951         a->user_uid = geteuid();
1952 #endif /* HAVE_GETEUID */
1953         if (archive_string_ensure(&a->path_safe, 512) == NULL) {
1954                 free(a);
1955                 return (NULL);
1956         }
1957 #ifdef HAVE_ZLIB_H
1958         a->decmpfs_compression_level = 5;
1959 #endif
1960         return (&a->archive);
1961 }
1962
1963
1964 /*
1965  * If pathname is longer than PATH_MAX, chdir to a suitable
1966  * intermediate dir and edit the path down to a shorter suffix.  Note
1967  * that this routine never returns an error; if the chdir() attempt
1968  * fails for any reason, we just go ahead with the long pathname.  The
1969  * object creation is likely to fail, but any error will get handled
1970  * at that time.
1971  */
1972 #if defined(HAVE_FCHDIR) && defined(PATH_MAX)
1973 static void
1974 edit_deep_directories(struct archive_write_disk *a)
1975 {
1976         int ret;
1977         char *tail = a->name;
1978
1979         /* If path is short, avoid the open() below. */
1980         if (strlen(tail) < PATH_MAX)
1981                 return;
1982
1983         /* Try to record our starting dir. */
1984         a->restore_pwd = la_opendirat(AT_FDCWD, ".");
1985         __archive_ensure_cloexec_flag(a->restore_pwd);
1986         if (a->restore_pwd < 0)
1987                 return;
1988
1989         /* As long as the path is too long... */
1990         while (strlen(tail) >= PATH_MAX) {
1991                 /* Locate a dir prefix shorter than PATH_MAX. */
1992                 tail += PATH_MAX - 8;
1993                 while (tail > a->name && *tail != '/')
1994                         tail--;
1995                 /* Exit if we find a too-long path component. */
1996                 if (tail <= a->name)
1997                         return;
1998                 /* Create the intermediate dir and chdir to it. */
1999                 *tail = '\0'; /* Terminate dir portion */
2000                 ret = create_dir(a, a->name);
2001                 if (ret == ARCHIVE_OK && chdir(a->name) != 0)
2002                         ret = ARCHIVE_FAILED;
2003                 *tail = '/'; /* Restore the / we removed. */
2004                 if (ret != ARCHIVE_OK)
2005                         return;
2006                 tail++;
2007                 /* The chdir() succeeded; we've now shortened the path. */
2008                 a->name = tail;
2009         }
2010         return;
2011 }
2012 #endif
2013
2014 /*
2015  * The main restore function.
2016  */
2017 static int
2018 restore_entry(struct archive_write_disk *a)
2019 {
2020         int ret = ARCHIVE_OK, en;
2021
2022         if (a->flags & ARCHIVE_EXTRACT_UNLINK && !S_ISDIR(a->mode)) {
2023                 /*
2024                  * TODO: Fix this.  Apparently, there are platforms
2025                  * that still allow root to hose the entire filesystem
2026                  * by unlinking a dir.  The S_ISDIR() test above
2027                  * prevents us from using unlink() here if the new
2028                  * object is a dir, but that doesn't mean the old
2029                  * object isn't a dir.
2030                  */
2031                 if (a->flags & ARCHIVE_EXTRACT_CLEAR_NOCHANGE_FFLAGS)
2032                         (void)clear_nochange_fflags(a);
2033                 if (unlink(a->name) == 0) {
2034                         /* We removed it, reset cached stat. */
2035                         a->pst = NULL;
2036                 } else if (errno == ENOENT) {
2037                         /* File didn't exist, that's just as good. */
2038                 } else if (rmdir(a->name) == 0) {
2039                         /* It was a dir, but now it's gone. */
2040                         a->pst = NULL;
2041                 } else {
2042                         /* We tried, but couldn't get rid of it. */
2043                         archive_set_error(&a->archive, errno,
2044                             "Could not unlink");
2045                         return(ARCHIVE_FAILED);
2046                 }
2047         }
2048
2049         /* Try creating it first; if this fails, we'll try to recover. */
2050         en = create_filesystem_object(a);
2051
2052         if ((en == ENOTDIR || en == ENOENT)
2053             && !(a->flags & ARCHIVE_EXTRACT_NO_AUTODIR)) {
2054                 /* If the parent dir doesn't exist, try creating it. */
2055                 create_parent_dir(a, a->name);
2056                 /* Now try to create the object again. */
2057                 en = create_filesystem_object(a);
2058         }
2059
2060         if ((en == ENOENT) && (archive_entry_hardlink(a->entry) != NULL)) {
2061                 archive_set_error(&a->archive, en,
2062                     "Hard-link target '%s' does not exist.",
2063                     archive_entry_hardlink(a->entry));
2064                 return (ARCHIVE_FAILED);
2065         }
2066
2067         if ((en == EISDIR || en == EEXIST)
2068             && (a->flags & ARCHIVE_EXTRACT_NO_OVERWRITE)) {
2069                 /* If we're not overwriting, we're done. */
2070                 if (S_ISDIR(a->mode)) {
2071                         /* Don't overwrite any settings on existing directories. */
2072                         a->todo = 0;
2073                 }
2074                 archive_entry_unset_size(a->entry);
2075                 return (ARCHIVE_OK);
2076         }
2077
2078         /*
2079          * Some platforms return EISDIR if you call
2080          * open(O_WRONLY | O_EXCL | O_CREAT) on a directory, some
2081          * return EEXIST.  POSIX is ambiguous, requiring EISDIR
2082          * for open(O_WRONLY) on a dir and EEXIST for open(O_EXCL | O_CREAT)
2083          * on an existing item.
2084          */
2085         if (en == EISDIR) {
2086                 /* A dir is in the way of a non-dir, rmdir it. */
2087                 if (rmdir(a->name) != 0) {
2088                         archive_set_error(&a->archive, errno,
2089                             "Can't remove already-existing dir");
2090                         return (ARCHIVE_FAILED);
2091                 }
2092                 a->pst = NULL;
2093                 /* Try again. */
2094                 en = create_filesystem_object(a);
2095         } else if (en == EEXIST) {
2096                 /*
2097                  * We know something is in the way, but we don't know what;
2098                  * we need to find out before we go any further.
2099                  */
2100                 int r = 0;
2101                 /*
2102                  * The SECURE_SYMLINKS logic has already removed a
2103                  * symlink to a dir if the client wants that.  So
2104                  * follow the symlink if we're creating a dir.
2105                  */
2106                 if (S_ISDIR(a->mode))
2107                         r = la_stat(a->name, &a->st);
2108                 /*
2109                  * If it's not a dir (or it's a broken symlink),
2110                  * then don't follow it.
2111                  */
2112                 if (r != 0 || !S_ISDIR(a->mode))
2113                         r = lstat(a->name, &a->st);
2114                 if (r != 0) {
2115                         archive_set_error(&a->archive, errno,
2116                             "Can't stat existing object");
2117                         return (ARCHIVE_FAILED);
2118                 }
2119
2120                 /*
2121                  * NO_OVERWRITE_NEWER doesn't apply to directories.
2122                  */
2123                 if ((a->flags & ARCHIVE_EXTRACT_NO_OVERWRITE_NEWER)
2124                     &&  !S_ISDIR(a->st.st_mode)) {
2125                         if (!older(&(a->st), a->entry)) {
2126                                 archive_entry_unset_size(a->entry);
2127                                 return (ARCHIVE_OK);
2128                         }
2129                 }
2130
2131                 /* If it's our archive, we're done. */
2132                 if (a->skip_file_set &&
2133                     a->st.st_dev == (dev_t)a->skip_file_dev &&
2134                     a->st.st_ino == (ino_t)a->skip_file_ino) {
2135                         archive_set_error(&a->archive, 0,
2136                             "Refusing to overwrite archive");
2137                         return (ARCHIVE_FAILED);
2138                 }
2139
2140                 if (!S_ISDIR(a->st.st_mode)) {
2141                         if (a->flags & ARCHIVE_EXTRACT_CLEAR_NOCHANGE_FFLAGS)
2142                                 (void)clear_nochange_fflags(a);
2143
2144                         if ((a->flags & ARCHIVE_EXTRACT_SAFE_WRITES) &&
2145                             S_ISREG(a->st.st_mode)) {
2146                                 /* Use a temporary file to extract */
2147                                 if ((a->fd = la_mktemp(a)) == -1)
2148                                         return ARCHIVE_FAILED;
2149                                 a->pst = NULL;
2150                                 en = 0;
2151                         } else {
2152                                 /* A non-dir is in the way, unlink it. */
2153                                 if (unlink(a->name) != 0) {
2154                                         archive_set_error(&a->archive, errno,
2155                                             "Can't unlink already-existing "
2156                                             "object");
2157                                         return (ARCHIVE_FAILED);
2158                                 }
2159                                 a->pst = NULL;
2160                                 /* Try again. */
2161                                 en = create_filesystem_object(a);
2162                         }
2163                 } else if (!S_ISDIR(a->mode)) {
2164                         /* A dir is in the way of a non-dir, rmdir it. */
2165                         if (a->flags & ARCHIVE_EXTRACT_CLEAR_NOCHANGE_FFLAGS)
2166                                 (void)clear_nochange_fflags(a);
2167                         if (rmdir(a->name) != 0) {
2168                                 archive_set_error(&a->archive, errno,
2169                                     "Can't replace existing directory with non-directory");
2170                                 return (ARCHIVE_FAILED);
2171                         }
2172                         /* Try again. */
2173                         en = create_filesystem_object(a);
2174                 } else {
2175                         /*
2176                          * There's a dir in the way of a dir.  Don't
2177                          * waste time with rmdir()/mkdir(), just fix
2178                          * up the permissions on the existing dir.
2179                          * Note that we don't change perms on existing
2180                          * dirs unless _EXTRACT_PERM is specified.
2181                          */
2182                         if ((a->mode != a->st.st_mode)
2183                             && (a->todo & TODO_MODE_FORCE))
2184                                 a->deferred |= (a->todo & TODO_MODE);
2185                         /* Ownership doesn't need deferred fixup. */
2186                         en = 0; /* Forget the EEXIST. */
2187                 }
2188         }
2189
2190         if (en) {
2191                 /* Everything failed; give up here. */
2192                 if ((&a->archive)->error == NULL)
2193                         archive_set_error(&a->archive, en, "Can't create '%s'",
2194                             a->name);
2195                 return (ARCHIVE_FAILED);
2196         }
2197
2198         a->pst = NULL; /* Cached stat data no longer valid. */
2199         return (ret);
2200 }
2201
2202 /*
2203  * Returns 0 if creation succeeds, or else returns errno value from
2204  * the failed system call.   Note:  This function should only ever perform
2205  * a single system call.
2206  */
2207 static int
2208 create_filesystem_object(struct archive_write_disk *a)
2209 {
2210         /* Create the entry. */
2211         const char *linkname;
2212         mode_t final_mode, mode;
2213         int r;
2214         /* these for check_symlinks_fsobj */
2215         char *linkname_copy;    /* non-const copy of linkname */
2216         struct stat st;
2217         struct archive_string error_string;
2218         int error_number;
2219
2220         /* We identify hard/symlinks according to the link names. */
2221         /* Since link(2) and symlink(2) don't handle modes, we're done here. */
2222         linkname = archive_entry_hardlink(a->entry);
2223         if (linkname != NULL) {
2224 #if !HAVE_LINK
2225                 return (EPERM);
2226 #else
2227                 archive_string_init(&error_string);
2228                 linkname_copy = strdup(linkname);
2229                 if (linkname_copy == NULL) {
2230                     return (EPERM);
2231                 }
2232                 /*
2233                  * TODO: consider using the cleaned-up path as the link
2234                  * target?
2235                  */
2236                 r = cleanup_pathname_fsobj(linkname_copy, &error_number,
2237                     &error_string, a->flags);
2238                 if (r != ARCHIVE_OK) {
2239                         archive_set_error(&a->archive, error_number, "%s",
2240                             error_string.s);
2241                         free(linkname_copy);
2242                         archive_string_free(&error_string);
2243                         /*
2244                          * EPERM is more appropriate than error_number for our
2245                          * callers
2246                          */
2247                         return (EPERM);
2248                 }
2249                 r = check_symlinks_fsobj(linkname_copy, &error_number,
2250                     &error_string, a->flags);
2251                 if (r != ARCHIVE_OK) {
2252                         archive_set_error(&a->archive, error_number, "%s",
2253                             error_string.s);
2254                         free(linkname_copy);
2255                         archive_string_free(&error_string);
2256                         /*
2257                          * EPERM is more appropriate than error_number for our
2258                          * callers
2259                          */
2260                         return (EPERM);
2261                 }
2262                 free(linkname_copy);
2263                 archive_string_free(&error_string);
2264                 /*
2265                  * Unlinking and linking here is really not atomic,
2266                  * but doing it right, would require us to construct
2267                  * an mktemplink() function, and then use rename(2).
2268                  */
2269                 if (a->flags & ARCHIVE_EXTRACT_SAFE_WRITES)
2270                         unlink(a->name);
2271                 r = link(linkname, a->name) ? errno : 0;
2272                 /*
2273                  * New cpio and pax formats allow hardlink entries
2274                  * to carry data, so we may have to open the file
2275                  * for hardlink entries.
2276                  *
2277                  * If the hardlink was successfully created and
2278                  * the archive doesn't have carry data for it,
2279                  * consider it to be non-authoritative for meta data.
2280                  * This is consistent with GNU tar and BSD pax.
2281                  * If the hardlink does carry data, let the last
2282                  * archive entry decide ownership.
2283                  */
2284                 if (r == 0 && a->filesize <= 0) {
2285                         a->todo = 0;
2286                         a->deferred = 0;
2287                 } else if (r == 0 && a->filesize > 0) {
2288 #ifdef HAVE_LSTAT
2289                         r = lstat(a->name, &st);
2290 #else
2291                         r = la_stat(a->name, &st);
2292 #endif
2293                         if (r != 0)
2294                                 r = errno;
2295                         else if ((st.st_mode & AE_IFMT) == AE_IFREG) {
2296                                 a->fd = open(a->name, O_WRONLY | O_TRUNC |
2297                                     O_BINARY | O_CLOEXEC | O_NOFOLLOW);
2298                                 __archive_ensure_cloexec_flag(a->fd);
2299                                 if (a->fd < 0)
2300                                         r = errno;
2301                         }
2302                 }
2303                 return (r);
2304 #endif
2305         }
2306         linkname = archive_entry_symlink(a->entry);
2307         if (linkname != NULL) {
2308 #if HAVE_SYMLINK
2309                 /*
2310                  * Unlinking and linking here is really not atomic,
2311                  * but doing it right, would require us to construct
2312                  * an mktempsymlink() function, and then use rename(2).
2313                  */
2314                 if (a->flags & ARCHIVE_EXTRACT_SAFE_WRITES)
2315                         unlink(a->name);
2316                 return symlink(linkname, a->name) ? errno : 0;
2317 #else
2318                 return (EPERM);
2319 #endif
2320         }
2321
2322         /*
2323          * The remaining system calls all set permissions, so let's
2324          * try to take advantage of that to avoid an extra chmod()
2325          * call.  (Recall that umask is set to zero right now!)
2326          */
2327
2328         /* Mode we want for the final restored object (w/o file type bits). */
2329         final_mode = a->mode & 07777;
2330         /*
2331          * The mode that will actually be restored in this step.  Note
2332          * that SUID, SGID, etc, require additional work to ensure
2333          * security, so we never restore them at this point.
2334          */
2335         mode = final_mode & 0777 & ~a->user_umask;
2336
2337         /* 
2338          * Always create writable such that [f]setxattr() works if we're not
2339          * root.
2340          */
2341         if (a->user_uid != 0 &&
2342             a->todo & (TODO_HFS_COMPRESSION | TODO_XATTR)) {
2343                 mode |= 0200;
2344         }
2345
2346         switch (a->mode & AE_IFMT) {
2347         default:
2348                 /* POSIX requires that we fall through here. */
2349                 /* FALLTHROUGH */
2350         case AE_IFREG:
2351                 a->tmpname = NULL;
2352                 a->fd = open(a->name,
2353                     O_WRONLY | O_CREAT | O_EXCL | O_BINARY | O_CLOEXEC, mode);
2354                 __archive_ensure_cloexec_flag(a->fd);
2355                 r = (a->fd < 0);
2356                 break;
2357         case AE_IFCHR:
2358 #ifdef HAVE_MKNOD
2359                 /* Note: we use AE_IFCHR for the case label, and
2360                  * S_IFCHR for the mknod() call.  This is correct.  */
2361                 r = mknod(a->name, mode | S_IFCHR,
2362                     archive_entry_rdev(a->entry));
2363                 break;
2364 #else
2365                 /* TODO: Find a better way to warn about our inability
2366                  * to restore a char device node. */
2367                 return (EINVAL);
2368 #endif /* HAVE_MKNOD */
2369         case AE_IFBLK:
2370 #ifdef HAVE_MKNOD
2371                 r = mknod(a->name, mode | S_IFBLK,
2372                     archive_entry_rdev(a->entry));
2373                 break;
2374 #else
2375                 /* TODO: Find a better way to warn about our inability
2376                  * to restore a block device node. */
2377                 return (EINVAL);
2378 #endif /* HAVE_MKNOD */
2379         case AE_IFDIR:
2380                 mode = (mode | MINIMUM_DIR_MODE) & MAXIMUM_DIR_MODE;
2381                 r = mkdir(a->name, mode);
2382                 if (r == 0) {
2383                         /* Defer setting dir times. */
2384                         a->deferred |= (a->todo & TODO_TIMES);
2385                         a->todo &= ~TODO_TIMES;
2386                         /* Never use an immediate chmod(). */
2387                         /* We can't avoid the chmod() entirely if EXTRACT_PERM
2388                          * because of SysV SGID inheritance. */
2389                         if ((mode != final_mode)
2390                             || (a->flags & ARCHIVE_EXTRACT_PERM))
2391                                 a->deferred |= (a->todo & TODO_MODE);
2392                         a->todo &= ~TODO_MODE;
2393                 }
2394                 break;
2395         case AE_IFIFO:
2396 #ifdef HAVE_MKFIFO
2397                 r = mkfifo(a->name, mode);
2398                 break;
2399 #else
2400                 /* TODO: Find a better way to warn about our inability
2401                  * to restore a fifo. */
2402                 return (EINVAL);
2403 #endif /* HAVE_MKFIFO */
2404         }
2405
2406         /* All the system calls above set errno on failure. */
2407         if (r)
2408                 return (errno);
2409
2410         /* If we managed to set the final mode, we've avoided a chmod(). */
2411         if (mode == final_mode)
2412                 a->todo &= ~TODO_MODE;
2413         return (0);
2414 }
2415
2416 /*
2417  * Cleanup function for archive_extract.  Mostly, this involves processing
2418  * the fixup list, which is used to address a number of problems:
2419  *   * Dir permissions might prevent us from restoring a file in that
2420  *     dir, so we restore the dir with minimum 0700 permissions first,
2421  *     then correct the mode at the end.
2422  *   * Similarly, the act of restoring a file touches the directory
2423  *     and changes the timestamp on the dir, so we have to touch-up dir
2424  *     timestamps at the end as well.
2425  *   * Some file flags can interfere with the restore by, for example,
2426  *     preventing the creation of hardlinks to those files.
2427  *   * Mac OS extended metadata includes ACLs, so must be deferred on dirs.
2428  *
2429  * Note that tar/cpio do not require that archives be in a particular
2430  * order; there is no way to know when the last file has been restored
2431  * within a directory, so there's no way to optimize the memory usage
2432  * here by fixing up the directory any earlier than the
2433  * end-of-archive.
2434  *
2435  * XXX TODO: Directory ACLs should be restored here, for the same
2436  * reason we set directory perms here. XXX
2437  */
2438 static int
2439 _archive_write_disk_close(struct archive *_a)
2440 {
2441         struct archive_write_disk *a = (struct archive_write_disk *)_a;
2442         struct fixup_entry *next, *p;
2443         int fd, ret;
2444
2445         archive_check_magic(&a->archive, ARCHIVE_WRITE_DISK_MAGIC,
2446             ARCHIVE_STATE_HEADER | ARCHIVE_STATE_DATA,
2447             "archive_write_disk_close");
2448         ret = _archive_write_disk_finish_entry(&a->archive);
2449
2450         /* Sort dir list so directories are fixed up in depth-first order. */
2451         p = sort_dir_list(a->fixup_list);
2452
2453         while (p != NULL) {
2454                 fd = -1;
2455                 a->pst = NULL; /* Mark stat cache as out-of-date. */
2456                 if (p->fixup &
2457                     (TODO_TIMES | TODO_MODE_BASE | TODO_ACLS | TODO_FFLAGS)) {
2458                         fd = open(p->name,
2459                             O_WRONLY | O_BINARY | O_NOFOLLOW | O_CLOEXEC);
2460                 }
2461                 if (p->fixup & TODO_TIMES) {
2462                         set_times(a, fd, p->mode, p->name,
2463                             p->atime, p->atime_nanos,
2464                             p->birthtime, p->birthtime_nanos,
2465                             p->mtime, p->mtime_nanos,
2466                             p->ctime, p->ctime_nanos);
2467                 }
2468                 if (p->fixup & TODO_MODE_BASE) {
2469 #ifdef HAVE_FCHMOD
2470                         if (fd >= 0)
2471                                 fchmod(fd, p->mode);
2472                         else
2473 #endif
2474                         chmod(p->name, p->mode);
2475                 }
2476                 if (p->fixup & TODO_ACLS)
2477                         archive_write_disk_set_acls(&a->archive, fd,
2478                             p->name, &p->acl, p->mode);
2479                 if (p->fixup & TODO_FFLAGS)
2480                         set_fflags_platform(a, fd, p->name,
2481                             p->mode, p->fflags_set, 0);
2482                 if (p->fixup & TODO_MAC_METADATA)
2483                         set_mac_metadata(a, p->name, p->mac_metadata,
2484                                          p->mac_metadata_size);
2485                 next = p->next;
2486                 archive_acl_clear(&p->acl);
2487                 free(p->mac_metadata);
2488                 free(p->name);
2489                 if (fd >= 0)
2490                         close(fd);
2491                 free(p);
2492                 p = next;
2493         }
2494         a->fixup_list = NULL;
2495         return (ret);
2496 }
2497
2498 static int
2499 _archive_write_disk_free(struct archive *_a)
2500 {
2501         struct archive_write_disk *a;
2502         int ret;
2503         if (_a == NULL)
2504                 return (ARCHIVE_OK);
2505         archive_check_magic(_a, ARCHIVE_WRITE_DISK_MAGIC,
2506             ARCHIVE_STATE_ANY | ARCHIVE_STATE_FATAL, "archive_write_disk_free");
2507         a = (struct archive_write_disk *)_a;
2508         ret = _archive_write_disk_close(&a->archive);
2509         archive_write_disk_set_group_lookup(&a->archive, NULL, NULL, NULL);
2510         archive_write_disk_set_user_lookup(&a->archive, NULL, NULL, NULL);
2511         archive_entry_free(a->entry);
2512         archive_string_free(&a->_name_data);
2513         archive_string_free(&a->_tmpname_data);
2514         archive_string_free(&a->archive.error_string);
2515         archive_string_free(&a->path_safe);
2516         a->archive.magic = 0;
2517         __archive_clean(&a->archive);
2518         free(a->decmpfs_header_p);
2519         free(a->resource_fork);
2520         free(a->compressed_buffer);
2521         free(a->uncompressed_buffer);
2522 #if defined(__APPLE__) && defined(UF_COMPRESSED) && defined(HAVE_SYS_XATTR_H)\
2523         && defined(HAVE_ZLIB_H)
2524         if (a->stream_valid) {
2525                 switch (deflateEnd(&a->stream)) {
2526                 case Z_OK:
2527                         break;
2528                 default:
2529                         archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC,
2530                             "Failed to clean up compressor");
2531                         ret = ARCHIVE_FATAL;
2532                         break;
2533                 }
2534         }
2535 #endif
2536         free(a);
2537         return (ret);
2538 }
2539
2540 /*
2541  * Simple O(n log n) merge sort to order the fixup list.  In
2542  * particular, we want to restore dir timestamps depth-first.
2543  */
2544 static struct fixup_entry *
2545 sort_dir_list(struct fixup_entry *p)
2546 {
2547         struct fixup_entry *a, *b, *t;
2548
2549         if (p == NULL)
2550                 return (NULL);
2551         /* A one-item list is already sorted. */
2552         if (p->next == NULL)
2553                 return (p);
2554
2555         /* Step 1: split the list. */
2556         t = p;
2557         a = p->next->next;
2558         while (a != NULL) {
2559                 /* Step a twice, t once. */
2560                 a = a->next;
2561                 if (a != NULL)
2562                         a = a->next;
2563                 t = t->next;
2564         }
2565         /* Now, t is at the mid-point, so break the list here. */
2566         b = t->next;
2567         t->next = NULL;
2568         a = p;
2569
2570         /* Step 2: Recursively sort the two sub-lists. */
2571         a = sort_dir_list(a);
2572         b = sort_dir_list(b);
2573
2574         /* Step 3: Merge the returned lists. */
2575         /* Pick the first element for the merged list. */
2576         if (strcmp(a->name, b->name) > 0) {
2577                 t = p = a;
2578                 a = a->next;
2579         } else {
2580                 t = p = b;
2581                 b = b->next;
2582         }
2583
2584         /* Always put the later element on the list first. */
2585         while (a != NULL && b != NULL) {
2586                 if (strcmp(a->name, b->name) > 0) {
2587                         t->next = a;
2588                         a = a->next;
2589                 } else {
2590                         t->next = b;
2591                         b = b->next;
2592                 }
2593                 t = t->next;
2594         }
2595
2596         /* Only one list is non-empty, so just splice it on. */
2597         if (a != NULL)
2598                 t->next = a;
2599         if (b != NULL)
2600                 t->next = b;
2601
2602         return (p);
2603 }
2604
2605 /*
2606  * Returns a new, initialized fixup entry.
2607  *
2608  * TODO: Reduce the memory requirements for this list by using a tree
2609  * structure rather than a simple list of names.
2610  */
2611 static struct fixup_entry *
2612 new_fixup(struct archive_write_disk *a, const char *pathname)
2613 {
2614         struct fixup_entry *fe;
2615
2616         fe = (struct fixup_entry *)calloc(1, sizeof(struct fixup_entry));
2617         if (fe == NULL) {
2618                 archive_set_error(&a->archive, ENOMEM,
2619                     "Can't allocate memory for a fixup");
2620                 return (NULL);
2621         }
2622         fe->next = a->fixup_list;
2623         a->fixup_list = fe;
2624         fe->fixup = 0;
2625         fe->name = strdup(pathname);
2626         return (fe);
2627 }
2628
2629 /*
2630  * Returns a fixup structure for the current entry.
2631  */
2632 static struct fixup_entry *
2633 current_fixup(struct archive_write_disk *a, const char *pathname)
2634 {
2635         if (a->current_fixup == NULL)
2636                 a->current_fixup = new_fixup(a, pathname);
2637         return (a->current_fixup);
2638 }
2639
2640 /* Error helper for new *_fsobj functions */
2641 static void
2642 fsobj_error(int *a_eno, struct archive_string *a_estr,
2643     int err, const char *errstr, const char *path)
2644 {
2645         if (a_eno)
2646                 *a_eno = err;
2647         if (a_estr)
2648                 archive_string_sprintf(a_estr, "%s%s", errstr, path);
2649 }
2650
2651 /*
2652  * TODO: Someday, integrate this with the deep dir support; they both
2653  * scan the path and both can be optimized by comparing against other
2654  * recent paths.
2655  */
2656 /*
2657  * Checks the given path to see if any elements along it are symlinks.  Returns
2658  * ARCHIVE_OK if there are none, otherwise puts an error in errmsg.
2659  */
2660 static int
2661 check_symlinks_fsobj(char *path, int *a_eno, struct archive_string *a_estr,
2662     int flags)
2663 {
2664 #if !defined(HAVE_LSTAT) && \
2665     !(defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT))
2666         /* Platform doesn't have lstat, so we can't look for symlinks. */
2667         (void)path; /* UNUSED */
2668         (void)error_number; /* UNUSED */
2669         (void)error_string; /* UNUSED */
2670         (void)flags; /* UNUSED */
2671         return (ARCHIVE_OK);
2672 #else
2673         int res = ARCHIVE_OK;
2674         char *tail;
2675         char *head;
2676         int last;
2677         char c;
2678         int r;
2679         struct stat st;
2680         int chdir_fd;
2681 #if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT)
2682         int fd;
2683 #endif
2684
2685         /* Nothing to do here if name is empty */
2686         if(path[0] == '\0')
2687             return (ARCHIVE_OK);
2688
2689         /*
2690          * Guard against symlink tricks.  Reject any archive entry whose
2691          * destination would be altered by a symlink.
2692          *
2693          * Walk the filename in chunks separated by '/'.  For each segment:
2694          *  - if it doesn't exist, continue
2695          *  - if it's symlink, abort or remove it
2696          *  - if it's a directory and it's not the last chunk, cd into it
2697          * As we go:
2698          *  head points to the current (relative) path
2699          *  tail points to the temporary \0 terminating the segment we're
2700          *      currently examining
2701          *  c holds what used to be in *tail
2702          *  last is 1 if this is the last tail
2703          */
2704         chdir_fd = la_opendirat(AT_FDCWD, ".");
2705         __archive_ensure_cloexec_flag(chdir_fd);
2706         if (chdir_fd < 0) {
2707                 fsobj_error(a_eno, a_estr, errno,
2708                     "Could not open ", path);
2709                 return (ARCHIVE_FATAL);
2710         }
2711         head = path;
2712         tail = path;
2713         last = 0;
2714         /* TODO: reintroduce a safe cache here? */
2715         /* Skip the root directory if the path is absolute. */
2716         if(tail == path && tail[0] == '/')
2717                 ++tail;
2718         /* Keep going until we've checked the entire name.
2719          * head, tail, path all alias the same string, which is
2720          * temporarily zeroed at tail, so be careful restoring the
2721          * stashed (c=tail[0]) for error messages.
2722          * Exiting the loop with break is okay; continue is not.
2723          */
2724         while (!last) {
2725                 /*
2726                  * Skip the separator we just consumed, plus any adjacent ones
2727                  */
2728                 while (*tail == '/')
2729                     ++tail;
2730                 /* Skip the next path element. */
2731                 while (*tail != '\0' && *tail != '/')
2732                         ++tail;
2733                 /* is this the last path component? */
2734                 last = (tail[0] == '\0') || (tail[0] == '/' && tail[1] == '\0');
2735                 /* temporarily truncate the string here */
2736                 c = tail[0];
2737                 tail[0] = '\0';
2738                 /* Check that we haven't hit a symlink. */
2739 #if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT)
2740                 r = fstatat(chdir_fd, head, &st, AT_SYMLINK_NOFOLLOW);
2741 #else
2742                 r = lstat(head, &st);
2743 #endif
2744                 if (r != 0) {
2745                         tail[0] = c;
2746                         /* We've hit a dir that doesn't exist; stop now. */
2747                         if (errno == ENOENT) {
2748                                 break;
2749                         } else {
2750                                 /*
2751                                  * Treat any other error as fatal - best to be
2752                                  * paranoid here.
2753                                  * Note: This effectively disables deep
2754                                  * directory support when security checks are
2755                                  * enabled. Otherwise, very long pathnames that
2756                                  * trigger an error here could evade the
2757                                  * sandbox.
2758                                  * TODO: We could do better, but it would
2759                                  * probably require merging the symlink checks
2760                                  * with the deep-directory editing.
2761                                  */
2762                                 fsobj_error(a_eno, a_estr, errno,
2763                                     "Could not stat ", path);
2764                                 res = ARCHIVE_FAILED;
2765                                 break;
2766                         }
2767                 } else if (S_ISDIR(st.st_mode)) {
2768                         if (!last) {
2769 #if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT)
2770                                 fd = la_opendirat(chdir_fd, head);
2771                                 if (fd < 0)
2772                                         r = -1;
2773                                 else {
2774                                         r = 0;
2775                                         close(chdir_fd);
2776                                         chdir_fd = fd;
2777                                 }
2778 #else
2779                                 r = chdir(head);
2780 #endif
2781                                 if (r != 0) {
2782                                         tail[0] = c;
2783                                         fsobj_error(a_eno, a_estr, errno,
2784                                             "Could not chdir ", path);
2785                                         res = (ARCHIVE_FATAL);
2786                                         break;
2787                                 }
2788                                 /* Our view is now from inside this dir: */
2789                                 head = tail + 1;
2790                         }
2791                 } else if (S_ISLNK(st.st_mode)) {
2792                         if (last) {
2793                                 /*
2794                                  * Last element is symlink; remove it
2795                                  * so we can overwrite it with the
2796                                  * item being extracted.
2797                                  */
2798 #if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT)
2799                                 r = unlinkat(chdir_fd, head, 0);
2800 #else
2801                                 r = unlink(head);
2802 #endif
2803                                 if (r != 0) {
2804                                         tail[0] = c;
2805                                         fsobj_error(a_eno, a_estr, errno,
2806                                             "Could not remove symlink ",
2807                                             path);
2808                                         res = ARCHIVE_FAILED;
2809                                         break;
2810                                 }
2811                                 /*
2812                                  * Even if we did remove it, a warning
2813                                  * is in order.  The warning is silly,
2814                                  * though, if we're just replacing one
2815                                  * symlink with another symlink.
2816                                  */
2817                                 tail[0] = c;
2818                                 /*
2819                                  * FIXME:  not sure how important this is to
2820                                  * restore
2821                                  */
2822                                 /*
2823                                 if (!S_ISLNK(path)) {
2824                                         fsobj_error(a_eno, a_estr, 0,
2825                                             "Removing symlink ", path);
2826                                 }
2827                                 */
2828                                 /* Symlink gone.  No more problem! */
2829                                 res = ARCHIVE_OK;
2830                                 break;
2831                         } else if (flags & ARCHIVE_EXTRACT_UNLINK) {
2832                                 /* User asked us to remove problems. */
2833 #if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT)
2834                                 r = unlinkat(chdir_fd, head, 0);
2835 #else
2836                                 r = unlink(head);
2837 #endif
2838                                 if (r != 0) {
2839                                         tail[0] = c;
2840                                         fsobj_error(a_eno, a_estr, 0,
2841                                             "Cannot remove intervening "
2842                                             "symlink ", path);
2843                                         res = ARCHIVE_FAILED;
2844                                         break;
2845                                 }
2846                                 tail[0] = c;
2847                         } else if ((flags &
2848                             ARCHIVE_EXTRACT_SECURE_SYMLINKS) == 0) {
2849                                 /*
2850                                  * We are not the last element and we want to
2851                                  * follow symlinks if they are a directory.
2852                                  * 
2853                                  * This is needed to extract hardlinks over
2854                                  * symlinks.
2855                                  */
2856 #if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT)
2857                                 r = fstatat(chdir_fd, head, &st, 0);
2858 #else
2859                                 r = la_stat(head, &st);
2860 #endif
2861                                 if (r != 0) {
2862                                         tail[0] = c;
2863                                         if (errno == ENOENT) {
2864                                                 break;
2865                                         } else {
2866                                                 fsobj_error(a_eno, a_estr,
2867                                                     errno,
2868                                                     "Could not stat ", path);
2869                                                 res = (ARCHIVE_FAILED);
2870                                                 break;
2871                                         }
2872                                 } else if (S_ISDIR(st.st_mode)) {
2873 #if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT)
2874                                         fd = la_opendirat(chdir_fd, head);
2875                                         if (fd < 0)
2876                                                 r = -1;
2877                                         else {
2878                                                 r = 0;
2879                                                 close(chdir_fd);
2880                                                 chdir_fd = fd;
2881                                         }
2882 #else
2883                                         r = chdir(head);
2884 #endif
2885                                         if (r != 0) {
2886                                                 tail[0] = c;
2887                                                 fsobj_error(a_eno, a_estr,
2888                                                     errno,
2889                                                     "Could not chdir ", path);
2890                                                 res = (ARCHIVE_FATAL);
2891                                                 break;
2892                                         }
2893                                         /*
2894                                          * Our view is now from inside
2895                                          * this dir:
2896                                          */
2897                                         head = tail + 1;
2898                                 } else {
2899                                         tail[0] = c;
2900                                         fsobj_error(a_eno, a_estr, 0,
2901                                             "Cannot extract through "
2902                                             "symlink ", path);
2903                                         res = ARCHIVE_FAILED;
2904                                         break;
2905                                 }
2906                         } else {
2907                                 tail[0] = c;
2908                                 fsobj_error(a_eno, a_estr, 0,
2909                                     "Cannot extract through symlink ", path);
2910                                 res = ARCHIVE_FAILED;
2911                                 break;
2912                         }
2913                 }
2914                 /* be sure to always maintain this */
2915                 tail[0] = c;
2916                 if (tail[0] != '\0')
2917                         tail++; /* Advance to the next segment. */
2918         }
2919         /* Catches loop exits via break */
2920         tail[0] = c;
2921 #if defined(HAVE_OPENAT) && defined(HAVE_FSTATAT) && defined(HAVE_UNLINKAT)
2922         /* If we operate with openat(), fstatat() and unlinkat() there was
2923          * no chdir(), so just close the fd */
2924         if (chdir_fd >= 0)
2925                 close(chdir_fd);
2926 #elif HAVE_FCHDIR
2927         /* If we changed directory above, restore it here. */
2928         if (chdir_fd >= 0) {
2929                 r = fchdir(chdir_fd);
2930                 if (r != 0) {
2931                         fsobj_error(a_eno, a_estr, errno,
2932                             "chdir() failure", "");
2933                 }
2934                 close(chdir_fd);
2935                 chdir_fd = -1;
2936                 if (r != 0) {
2937                         res = (ARCHIVE_FATAL);
2938                 }
2939         }
2940 #endif
2941         /* TODO: reintroduce a safe cache here? */
2942         return res;
2943 #endif
2944 }
2945
2946 /*
2947  * Check a->name for symlinks, returning ARCHIVE_OK if its clean, otherwise
2948  * calls archive_set_error and returns ARCHIVE_{FATAL,FAILED}
2949  */
2950 static int
2951 check_symlinks(struct archive_write_disk *a)
2952 {
2953         struct archive_string error_string;
2954         int error_number;
2955         int rc;
2956         archive_string_init(&error_string);
2957         rc = check_symlinks_fsobj(a->name, &error_number, &error_string,
2958             a->flags);
2959         if (rc != ARCHIVE_OK) {
2960                 archive_set_error(&a->archive, error_number, "%s",
2961                     error_string.s);
2962         }
2963         archive_string_free(&error_string);
2964         a->pst = NULL;  /* to be safe */
2965         return rc;
2966 }
2967
2968
2969 #if defined(__CYGWIN__)
2970 /*
2971  * 1. Convert a path separator from '\' to '/' .
2972  *    We shouldn't check multibyte character directly because some
2973  *    character-set have been using the '\' character for a part of
2974  *    its multibyte character code.
2975  * 2. Replace unusable characters in Windows with underscore('_').
2976  * See also : http://msdn.microsoft.com/en-us/library/aa365247.aspx
2977  */
2978 static void
2979 cleanup_pathname_win(char *path)
2980 {
2981         wchar_t wc;
2982         char *p;
2983         size_t alen, l;
2984         int mb, complete, utf8;
2985
2986         alen = 0;
2987         mb = 0;
2988         complete = 1;
2989         utf8 = (strcmp(nl_langinfo(CODESET), "UTF-8") == 0)? 1: 0;
2990         for (p = path; *p != '\0'; p++) {
2991                 ++alen;
2992                 if (*p == '\\') {
2993                         /* If previous byte is smaller than 128,
2994                          * this is not second byte of multibyte characters,
2995                          * so we can replace '\' with '/'. */
2996                         if (utf8 || !mb)
2997                                 *p = '/';
2998                         else
2999                                 complete = 0;/* uncompleted. */
3000                 } else if (*(unsigned char *)p > 127)
3001                         mb = 1;
3002                 else
3003                         mb = 0;
3004                 /* Rewrite the path name if its next character is unusable. */
3005                 if (*p == ':' || *p == '*' || *p == '?' || *p == '"' ||
3006                     *p == '<' || *p == '>' || *p == '|')
3007                         *p = '_';
3008         }
3009         if (complete)
3010                 return;
3011
3012         /*
3013          * Convert path separator in wide-character.
3014          */
3015         p = path;
3016         while (*p != '\0' && alen) {
3017                 l = mbtowc(&wc, p, alen);
3018                 if (l == (size_t)-1) {
3019                         while (*p != '\0') {
3020                                 if (*p == '\\')
3021                                         *p = '/';
3022                                 ++p;
3023                         }
3024                         break;
3025                 }
3026                 if (l == 1 && wc == L'\\')
3027                         *p = '/';
3028                 p += l;
3029                 alen -= l;
3030         }
3031 }
3032 #endif
3033
3034 /*
3035  * Canonicalize the pathname.  In particular, this strips duplicate
3036  * '/' characters, '.' elements, and trailing '/'.  It also raises an
3037  * error for an empty path, a trailing '..', (if _SECURE_NODOTDOT is
3038  * set) any '..' in the path or (if ARCHIVE_EXTRACT_SECURE_NOABSOLUTEPATHS
3039  * is set) if the path is absolute.
3040  */
3041 static int
3042 cleanup_pathname_fsobj(char *path, int *a_eno, struct archive_string *a_estr,
3043     int flags)
3044 {
3045         char *dest, *src;
3046         char separator = '\0';
3047
3048         dest = src = path;
3049         if (*src == '\0') {
3050                 fsobj_error(a_eno, a_estr, ARCHIVE_ERRNO_MISC,
3051                     "Invalid empty ", "pathname");
3052                 return (ARCHIVE_FAILED);
3053         }
3054
3055 #if defined(__CYGWIN__)
3056         cleanup_pathname_win(path);
3057 #endif
3058         /* Skip leading '/'. */
3059         if (*src == '/') {
3060                 if (flags & ARCHIVE_EXTRACT_SECURE_NOABSOLUTEPATHS) {
3061                         fsobj_error(a_eno, a_estr, ARCHIVE_ERRNO_MISC,
3062                             "Path is ", "absolute");
3063                         return (ARCHIVE_FAILED);
3064                 }
3065
3066                 separator = *src++;
3067         }
3068
3069         /* Scan the pathname one element at a time. */
3070         for (;;) {
3071                 /* src points to first char after '/' */
3072                 if (src[0] == '\0') {
3073                         break;
3074                 } else if (src[0] == '/') {
3075                         /* Found '//', ignore second one. */
3076                         src++;
3077                         continue;
3078                 } else if (src[0] == '.') {
3079                         if (src[1] == '\0') {
3080                                 /* Ignore trailing '.' */
3081                                 break;
3082                         } else if (src[1] == '/') {
3083                                 /* Skip './'. */
3084                                 src += 2;
3085                                 continue;
3086                         } else if (src[1] == '.') {
3087                                 if (src[2] == '/' || src[2] == '\0') {
3088                                         /* Conditionally warn about '..' */
3089                                         if (flags
3090                                             & ARCHIVE_EXTRACT_SECURE_NODOTDOT) {
3091                                                 fsobj_error(a_eno, a_estr,
3092                                                     ARCHIVE_ERRNO_MISC,
3093                                                     "Path contains ", "'..'");
3094                                                 return (ARCHIVE_FAILED);
3095                                         }
3096                                 }
3097                                 /*
3098                                  * Note: Under no circumstances do we
3099                                  * remove '..' elements.  In
3100                                  * particular, restoring
3101                                  * '/foo/../bar/' should create the
3102                                  * 'foo' dir as a side-effect.
3103                                  */
3104                         }
3105                 }
3106
3107                 /* Copy current element, including leading '/'. */
3108                 if (separator)
3109                         *dest++ = '/';
3110                 while (*src != '\0' && *src != '/') {
3111                         *dest++ = *src++;
3112                 }
3113
3114                 if (*src == '\0')
3115                         break;
3116
3117                 /* Skip '/' separator. */
3118                 separator = *src++;
3119         }
3120         /*
3121          * We've just copied zero or more path elements, not including the
3122          * final '/'.
3123          */
3124         if (dest == path) {
3125                 /*
3126                  * Nothing got copied.  The path must have been something
3127                  * like '.' or '/' or './' or '/././././/./'.
3128                  */
3129                 if (separator)
3130                         *dest++ = '/';
3131                 else
3132                         *dest++ = '.';
3133         }
3134         /* Terminate the result. */
3135         *dest = '\0';
3136         return (ARCHIVE_OK);
3137 }
3138
3139 static int
3140 cleanup_pathname(struct archive_write_disk *a)
3141 {
3142         struct archive_string error_string;
3143         int error_number;
3144         int rc;
3145         archive_string_init(&error_string);
3146         rc = cleanup_pathname_fsobj(a->name, &error_number, &error_string,
3147             a->flags);
3148         if (rc != ARCHIVE_OK) {
3149                 archive_set_error(&a->archive, error_number, "%s",
3150                     error_string.s);
3151         }
3152         archive_string_free(&error_string);
3153         return rc;
3154 }
3155
3156 /*
3157  * Create the parent directory of the specified path, assuming path
3158  * is already in mutable storage.
3159  */
3160 static int
3161 create_parent_dir(struct archive_write_disk *a, char *path)
3162 {
3163         char *slash;
3164         int r;
3165
3166         /* Remove tail element to obtain parent name. */
3167         slash = strrchr(path, '/');
3168         if (slash == NULL)
3169                 return (ARCHIVE_OK);
3170         *slash = '\0';
3171         r = create_dir(a, path);
3172         *slash = '/';
3173         return (r);
3174 }
3175
3176 /*
3177  * Create the specified dir, recursing to create parents as necessary.
3178  *
3179  * Returns ARCHIVE_OK if the path exists when we're done here.
3180  * Otherwise, returns ARCHIVE_FAILED.
3181  * Assumes path is in mutable storage; path is unchanged on exit.
3182  */
3183 static int
3184 create_dir(struct archive_write_disk *a, char *path)
3185 {
3186         struct stat st;
3187         struct fixup_entry *le;
3188         char *slash, *base;
3189         mode_t mode_final, mode;
3190         int r;
3191
3192         /* Check for special names and just skip them. */
3193         slash = strrchr(path, '/');
3194         if (slash == NULL)
3195                 base = path;
3196         else
3197                 base = slash + 1;
3198
3199         if (base[0] == '\0' ||
3200             (base[0] == '.' && base[1] == '\0') ||
3201             (base[0] == '.' && base[1] == '.' && base[2] == '\0')) {
3202                 /* Don't bother trying to create null path, '.', or '..'. */
3203                 if (slash != NULL) {
3204                         *slash = '\0';
3205                         r = create_dir(a, path);
3206                         *slash = '/';
3207                         return (r);
3208                 }
3209                 return (ARCHIVE_OK);
3210         }
3211
3212         /*
3213          * Yes, this should be stat() and not lstat().  Using lstat()
3214          * here loses the ability to extract through symlinks.  Also note
3215          * that this should not use the a->st cache.
3216          */
3217         if (la_stat(path, &st) == 0) {
3218                 if (S_ISDIR(st.st_mode))
3219                         return (ARCHIVE_OK);
3220                 if ((a->flags & ARCHIVE_EXTRACT_NO_OVERWRITE)) {
3221                         archive_set_error(&a->archive, EEXIST,
3222                             "Can't create directory '%s'", path);
3223                         return (ARCHIVE_FAILED);
3224                 }
3225                 if (unlink(path) != 0) {
3226                         archive_set_error(&a->archive, errno,
3227                             "Can't create directory '%s': "
3228                             "Conflicting file cannot be removed",
3229                             path);
3230                         return (ARCHIVE_FAILED);
3231                 }
3232         } else if (errno != ENOENT && errno != ENOTDIR) {
3233                 /* Stat failed? */
3234                 archive_set_error(&a->archive, errno,
3235                     "Can't test directory '%s'", path);
3236                 return (ARCHIVE_FAILED);
3237         } else if (slash != NULL) {
3238                 *slash = '\0';
3239                 r = create_dir(a, path);
3240                 *slash = '/';
3241                 if (r != ARCHIVE_OK)
3242                         return (r);
3243         }
3244
3245         /*
3246          * Mode we want for the final restored directory.  Per POSIX,
3247          * implicitly-created dirs must be created obeying the umask.
3248          * There's no mention whether this is different for privileged
3249          * restores (which the rest of this code handles by pretending
3250          * umask=0).  I've chosen here to always obey the user's umask for
3251          * implicit dirs, even if _EXTRACT_PERM was specified.
3252          */
3253         mode_final = DEFAULT_DIR_MODE & ~a->user_umask;
3254         /* Mode we want on disk during the restore process. */
3255         mode = mode_final;
3256         mode |= MINIMUM_DIR_MODE;
3257         mode &= MAXIMUM_DIR_MODE;
3258         if (mkdir(path, mode) == 0) {
3259                 if (mode != mode_final) {
3260                         le = new_fixup(a, path);
3261                         if (le == NULL)
3262                                 return (ARCHIVE_FATAL);
3263                         le->fixup |=TODO_MODE_BASE;
3264                         le->mode = mode_final;
3265                 }
3266                 return (ARCHIVE_OK);
3267         }
3268
3269         /*
3270          * Without the following check, a/b/../b/c/d fails at the
3271          * second visit to 'b', so 'd' can't be created.  Note that we
3272          * don't add it to the fixup list here, as it's already been
3273          * added.
3274          */
3275         if (la_stat(path, &st) == 0 && S_ISDIR(st.st_mode))
3276                 return (ARCHIVE_OK);
3277
3278         archive_set_error(&a->archive, errno, "Failed to create dir '%s'",
3279             path);
3280         return (ARCHIVE_FAILED);
3281 }
3282
3283 /*
3284  * Note: Although we can skip setting the user id if the desired user
3285  * id matches the current user, we cannot skip setting the group, as
3286  * many systems set the gid based on the containing directory.  So
3287  * we have to perform a chown syscall if we want to set the SGID
3288  * bit.  (The alternative is to stat() and then possibly chown(); it's
3289  * more efficient to skip the stat() and just always chown().)  Note
3290  * that a successful chown() here clears the TODO_SGID_CHECK bit, which
3291  * allows set_mode to skip the stat() check for the GID.
3292  */
3293 static int
3294 set_ownership(struct archive_write_disk *a)
3295 {
3296 #if !defined(__CYGWIN__) && !defined(__linux__)
3297 /*
3298  * On Linux, a process may have the CAP_CHOWN capability.
3299  * On Windows there is no 'root' user with uid 0.
3300  * Elsewhere we can skip calling chown if we are not root and the desired
3301  * user id does not match the current user.
3302  */
3303         if (a->user_uid != 0 && a->user_uid != a->uid) {
3304                 archive_set_error(&a->archive, errno,
3305                     "Can't set UID=%jd", (intmax_t)a->uid);
3306                 return (ARCHIVE_WARN);
3307         }
3308 #endif
3309
3310 #ifdef HAVE_FCHOWN
3311         /* If we have an fd, we can avoid a race. */
3312         if (a->fd >= 0 && fchown(a->fd, a->uid, a->gid) == 0) {
3313                 /* We've set owner and know uid/gid are correct. */
3314                 a->todo &= ~(TODO_OWNER | TODO_SGID_CHECK | TODO_SUID_CHECK);
3315                 return (ARCHIVE_OK);
3316         }
3317 #endif
3318
3319         /* We prefer lchown() but will use chown() if that's all we have. */
3320         /* Of course, if we have neither, this will always fail. */
3321 #ifdef HAVE_LCHOWN
3322         if (lchown(a->name, a->uid, a->gid) == 0) {
3323                 /* We've set owner and know uid/gid are correct. */
3324                 a->todo &= ~(TODO_OWNER | TODO_SGID_CHECK | TODO_SUID_CHECK);
3325                 return (ARCHIVE_OK);
3326         }
3327 #elif HAVE_CHOWN
3328         if (!S_ISLNK(a->mode) && chown(a->name, a->uid, a->gid) == 0) {
3329                 /* We've set owner and know uid/gid are correct. */
3330                 a->todo &= ~(TODO_OWNER | TODO_SGID_CHECK | TODO_SUID_CHECK);
3331                 return (ARCHIVE_OK);
3332         }
3333 #endif
3334
3335         archive_set_error(&a->archive, errno,
3336             "Can't set user=%jd/group=%jd for %s",
3337             (intmax_t)a->uid, (intmax_t)a->gid, a->name);
3338         return (ARCHIVE_WARN);
3339 }
3340
3341 /*
3342  * Note: Returns 0 on success, non-zero on failure.
3343  */
3344 static int
3345 set_time(int fd, int mode, const char *name,
3346     time_t atime, long atime_nsec,
3347     time_t mtime, long mtime_nsec)
3348 {
3349         /* Select the best implementation for this platform. */
3350 #if defined(HAVE_UTIMENSAT) && defined(HAVE_FUTIMENS)
3351         /*
3352          * utimensat() and futimens() are defined in
3353          * POSIX.1-2008. They support ns resolution and setting times
3354          * on fds and symlinks.
3355          */
3356         struct timespec ts[2];
3357         (void)mode; /* UNUSED */
3358         ts[0].tv_sec = atime;
3359         ts[0].tv_nsec = atime_nsec;
3360         ts[1].tv_sec = mtime;
3361         ts[1].tv_nsec = mtime_nsec;
3362         if (fd >= 0)
3363                 return futimens(fd, ts);
3364         return utimensat(AT_FDCWD, name, ts, AT_SYMLINK_NOFOLLOW);
3365
3366 #elif HAVE_UTIMES
3367         /*
3368          * The utimes()-family functions support Âµs-resolution and
3369          * setting times fds and symlinks.  utimes() is documented as
3370          * LEGACY by POSIX, futimes() and lutimes() are not described
3371          * in POSIX.
3372          */
3373         struct timeval times[2];
3374
3375         times[0].tv_sec = atime;
3376         times[0].tv_usec = atime_nsec / 1000;
3377         times[1].tv_sec = mtime;
3378         times[1].tv_usec = mtime_nsec / 1000;
3379
3380 #ifdef HAVE_FUTIMES
3381         if (fd >= 0)
3382                 return (futimes(fd, times));
3383 #else
3384         (void)fd; /* UNUSED */
3385 #endif
3386 #ifdef HAVE_LUTIMES
3387         (void)mode; /* UNUSED */
3388         return (lutimes(name, times));
3389 #else
3390         if (S_ISLNK(mode))
3391                 return (0);
3392         return (utimes(name, times));
3393 #endif
3394
3395 #elif defined(HAVE_UTIME)
3396         /*
3397          * utime() is POSIX-standard but only supports 1s resolution and
3398          * does not support fds or symlinks.
3399          */
3400         struct utimbuf times;
3401         (void)fd; /* UNUSED */
3402         (void)name; /* UNUSED */
3403         (void)atime_nsec; /* UNUSED */
3404         (void)mtime_nsec; /* UNUSED */
3405         times.actime = atime;
3406         times.modtime = mtime;
3407         if (S_ISLNK(mode))
3408                 return (ARCHIVE_OK);
3409         return (utime(name, &times));
3410
3411 #else
3412         /*
3413          * We don't know how to set the time on this platform.
3414          */
3415         (void)fd; /* UNUSED */
3416         (void)mode; /* UNUSED */
3417         (void)name; /* UNUSED */
3418         (void)atime_nsec; /* UNUSED */
3419         (void)mtime_nsec; /* UNUSED */
3420         return (ARCHIVE_WARN);
3421 #endif
3422 }
3423
3424 #ifdef F_SETTIMES
3425 static int
3426 set_time_tru64(int fd, int mode, const char *name,
3427     time_t atime, long atime_nsec,
3428     time_t mtime, long mtime_nsec,
3429     time_t ctime, long ctime_nsec)
3430 {
3431         struct attr_timbuf tstamp;
3432         tstamp.atime.tv_sec = atime;
3433         tstamp.mtime.tv_sec = mtime;
3434         tstamp.ctime.tv_sec = ctime;
3435 #if defined (__hpux) && defined (__ia64)
3436         tstamp.atime.tv_nsec = atime_nsec;
3437         tstamp.mtime.tv_nsec = mtime_nsec;
3438         tstamp.ctime.tv_nsec = ctime_nsec;
3439 #else
3440         tstamp.atime.tv_usec = atime_nsec / 1000;
3441         tstamp.mtime.tv_usec = mtime_nsec / 1000;
3442         tstamp.ctime.tv_usec = ctime_nsec / 1000;
3443 #endif
3444         return (fcntl(fd,F_SETTIMES,&tstamp));
3445 }
3446 #endif /* F_SETTIMES */
3447
3448 static int
3449 set_times(struct archive_write_disk *a,
3450     int fd, int mode, const char *name,
3451     time_t atime, long atime_nanos,
3452     time_t birthtime, long birthtime_nanos,
3453     time_t mtime, long mtime_nanos,
3454     time_t cctime, long ctime_nanos)
3455 {
3456         /* Note: set_time doesn't use libarchive return conventions!
3457          * It uses syscall conventions.  So 0 here instead of ARCHIVE_OK. */
3458         int r1 = 0, r2 = 0;
3459
3460 #ifdef F_SETTIMES
3461          /*
3462          * on Tru64 try own fcntl first which can restore even the
3463          * ctime, fall back to default code path below if it fails
3464          * or if we are not running as root
3465          */
3466         if (a->user_uid == 0 &&
3467             set_time_tru64(fd, mode, name,
3468                            atime, atime_nanos, mtime,
3469                            mtime_nanos, cctime, ctime_nanos) == 0) {
3470                 return (ARCHIVE_OK);
3471         }
3472 #else /* Tru64 */
3473         (void)cctime; /* UNUSED */
3474         (void)ctime_nanos; /* UNUSED */
3475 #endif /* Tru64 */
3476
3477 #ifdef HAVE_STRUCT_STAT_ST_BIRTHTIME
3478         /*
3479          * If you have struct stat.st_birthtime, we assume BSD
3480          * birthtime semantics, in which {f,l,}utimes() updates
3481          * birthtime to earliest mtime.  So we set the time twice,
3482          * first using the birthtime, then using the mtime.  If
3483          * birthtime == mtime, this isn't necessary, so we skip it.
3484          * If birthtime > mtime, then this won't work, so we skip it.
3485          */
3486         if (birthtime < mtime
3487             || (birthtime == mtime && birthtime_nanos < mtime_nanos))
3488                 r1 = set_time(fd, mode, name,
3489                               atime, atime_nanos,
3490                               birthtime, birthtime_nanos);
3491 #else
3492         (void)birthtime; /* UNUSED */
3493         (void)birthtime_nanos; /* UNUSED */
3494 #endif
3495         r2 = set_time(fd, mode, name,
3496                       atime, atime_nanos,
3497                       mtime, mtime_nanos);
3498         if (r1 != 0 || r2 != 0) {
3499                 archive_set_error(&a->archive, errno,
3500                                   "Can't restore time");
3501                 return (ARCHIVE_WARN);
3502         }
3503         return (ARCHIVE_OK);
3504 }
3505
3506 static int
3507 set_times_from_entry(struct archive_write_disk *a)
3508 {
3509         time_t atime, birthtime, mtime, cctime;
3510         long atime_nsec, birthtime_nsec, mtime_nsec, ctime_nsec;
3511
3512         /* Suitable defaults. */
3513         atime = birthtime = mtime = cctime = a->start_time;
3514         atime_nsec = birthtime_nsec = mtime_nsec = ctime_nsec = 0;
3515
3516         /* If no time was provided, we're done. */
3517         if (!archive_entry_atime_is_set(a->entry)
3518 #if HAVE_STRUCT_STAT_ST_BIRTHTIME
3519             && !archive_entry_birthtime_is_set(a->entry)
3520 #endif
3521             && !archive_entry_mtime_is_set(a->entry))
3522                 return (ARCHIVE_OK);
3523
3524         if (archive_entry_atime_is_set(a->entry)) {
3525                 atime = archive_entry_atime(a->entry);
3526                 atime_nsec = archive_entry_atime_nsec(a->entry);
3527         }
3528         if (archive_entry_birthtime_is_set(a->entry)) {
3529                 birthtime = archive_entry_birthtime(a->entry);
3530                 birthtime_nsec = archive_entry_birthtime_nsec(a->entry);
3531         }
3532         if (archive_entry_mtime_is_set(a->entry)) {
3533                 mtime = archive_entry_mtime(a->entry);
3534                 mtime_nsec = archive_entry_mtime_nsec(a->entry);
3535         }
3536         if (archive_entry_ctime_is_set(a->entry)) {
3537                 cctime = archive_entry_ctime(a->entry);
3538                 ctime_nsec = archive_entry_ctime_nsec(a->entry);
3539         }
3540
3541         return set_times(a, a->fd, a->mode, a->name,
3542                          atime, atime_nsec,
3543                          birthtime, birthtime_nsec,
3544                          mtime, mtime_nsec,
3545                          cctime, ctime_nsec);
3546 }
3547
3548 static int
3549 set_mode(struct archive_write_disk *a, int mode)
3550 {
3551         int r = ARCHIVE_OK;
3552         int r2;
3553         mode &= 07777; /* Strip off file type bits. */
3554
3555         if (a->todo & TODO_SGID_CHECK) {
3556                 /*
3557                  * If we don't know the GID is right, we must stat()
3558                  * to verify it.  We can't just check the GID of this
3559                  * process, since systems sometimes set GID from
3560                  * the enclosing dir or based on ACLs.
3561                  */
3562                 if ((r = lazy_stat(a)) != ARCHIVE_OK)
3563                         return (r);
3564                 if (a->pst->st_gid != a->gid) {
3565                         mode &= ~ S_ISGID;
3566                         if (a->flags & ARCHIVE_EXTRACT_OWNER) {
3567                                 /*
3568                                  * This is only an error if you
3569                                  * requested owner restore.  If you
3570                                  * didn't, we'll try to restore
3571                                  * sgid/suid, but won't consider it a
3572                                  * problem if we can't.
3573                                  */
3574                                 archive_set_error(&a->archive, -1,
3575                                     "Can't restore SGID bit");
3576                                 r = ARCHIVE_WARN;
3577                         }
3578                 }
3579                 /* While we're here, double-check the UID. */
3580                 if (a->pst->st_uid != a->uid
3581                     && (a->todo & TODO_SUID)) {
3582                         mode &= ~ S_ISUID;
3583                         if (a->flags & ARCHIVE_EXTRACT_OWNER) {
3584                                 archive_set_error(&a->archive, -1,
3585                                     "Can't restore SUID bit");
3586                                 r = ARCHIVE_WARN;
3587                         }
3588                 }
3589                 a->todo &= ~TODO_SGID_CHECK;
3590                 a->todo &= ~TODO_SUID_CHECK;
3591         } else if (a->todo & TODO_SUID_CHECK) {
3592                 /*
3593                  * If we don't know the UID is right, we can just check
3594                  * the user, since all systems set the file UID from
3595                  * the process UID.
3596                  */
3597                 if (a->user_uid != a->uid) {
3598                         mode &= ~ S_ISUID;
3599                         if (a->flags & ARCHIVE_EXTRACT_OWNER) {
3600                                 archive_set_error(&a->archive, -1,
3601                                     "Can't make file SUID");
3602                                 r = ARCHIVE_WARN;
3603                         }
3604                 }
3605                 a->todo &= ~TODO_SUID_CHECK;
3606         }
3607
3608         if (S_ISLNK(a->mode)) {
3609 #ifdef HAVE_LCHMOD
3610                 /*
3611                  * If this is a symlink, use lchmod().  If the
3612                  * platform doesn't support lchmod(), just skip it.  A
3613                  * platform that doesn't provide a way to set
3614                  * permissions on symlinks probably ignores
3615                  * permissions on symlinks, so a failure here has no
3616                  * impact.
3617                  */
3618                 if (lchmod(a->name, mode) != 0) {
3619                         switch (errno) {
3620                         case ENOTSUP:
3621                         case ENOSYS:
3622 #if ENOTSUP != EOPNOTSUPP
3623                         case EOPNOTSUPP:
3624 #endif
3625                                 /*
3626                                  * if lchmod is defined but the platform
3627                                  * doesn't support it, silently ignore
3628                                  * error
3629                                  */
3630                                 break;
3631                         default:
3632                                 archive_set_error(&a->archive, errno,
3633                                     "Can't set permissions to 0%o", (int)mode);
3634                                 r = ARCHIVE_WARN;
3635                         }
3636                 }
3637 #endif
3638         } else if (!S_ISDIR(a->mode)) {
3639                 /*
3640                  * If it's not a symlink and not a dir, then use
3641                  * fchmod() or chmod(), depending on whether we have
3642                  * an fd.  Dirs get their perms set during the
3643                  * post-extract fixup, which is handled elsewhere.
3644                  */
3645 #ifdef HAVE_FCHMOD
3646                 if (a->fd >= 0)
3647                         r2 = fchmod(a->fd, mode);
3648                 else
3649 #endif
3650                 /* If this platform lacks fchmod(), then
3651                  * we'll just use chmod(). */
3652                 r2 = chmod(a->name, mode);
3653
3654                 if (r2 != 0) {
3655                         archive_set_error(&a->archive, errno,
3656                             "Can't set permissions to 0%o", (int)mode);
3657                         r = ARCHIVE_WARN;
3658                 }
3659         }
3660         return (r);
3661 }
3662
3663 static int
3664 set_fflags(struct archive_write_disk *a)
3665 {
3666         struct fixup_entry *le;
3667         unsigned long   set, clear;
3668         int             r;
3669         mode_t          mode = archive_entry_mode(a->entry);
3670         /*
3671          * Make 'critical_flags' hold all file flags that can't be
3672          * immediately restored.  For example, on BSD systems,
3673          * SF_IMMUTABLE prevents hardlinks from being created, so
3674          * should not be set until after any hardlinks are created.  To
3675          * preserve some semblance of portability, this uses #ifdef
3676          * extensively.  Ugly, but it works.
3677          *
3678          * Yes, Virginia, this does create a security race.  It's mitigated
3679          * somewhat by the practice of creating dirs 0700 until the extract
3680          * is done, but it would be nice if we could do more than that.
3681          * People restoring critical file systems should be wary of
3682          * other programs that might try to muck with files as they're
3683          * being restored.
3684          */
3685         const int       critical_flags = 0
3686 #ifdef SF_IMMUTABLE
3687             | SF_IMMUTABLE
3688 #endif
3689 #ifdef UF_IMMUTABLE
3690             | UF_IMMUTABLE
3691 #endif
3692 #ifdef SF_APPEND
3693             | SF_APPEND
3694 #endif
3695 #ifdef UF_APPEND
3696             | UF_APPEND
3697 #endif
3698 #if defined(FS_APPEND_FL)
3699             | FS_APPEND_FL
3700 #elif defined(EXT2_APPEND_FL)
3701             | EXT2_APPEND_FL
3702 #endif
3703 #if defined(FS_IMMUTABLE_FL)
3704             | FS_IMMUTABLE_FL
3705 #elif defined(EXT2_IMMUTABLE_FL)
3706             | EXT2_IMMUTABLE_FL
3707 #endif
3708 #ifdef FS_JOURNAL_DATA_FL
3709             | FS_JOURNAL_DATA_FL
3710 #endif
3711         ;
3712
3713         if (a->todo & TODO_FFLAGS) {
3714                 archive_entry_fflags(a->entry, &set, &clear);
3715
3716                 /*
3717                  * The first test encourages the compiler to eliminate
3718                  * all of this if it's not necessary.
3719                  */
3720                 if ((critical_flags != 0)  &&  (set & critical_flags)) {
3721                         le = current_fixup(a, a->name);
3722                         if (le == NULL)
3723                                 return (ARCHIVE_FATAL);
3724                         le->fixup |= TODO_FFLAGS;
3725                         le->fflags_set = set;
3726                         /* Store the mode if it's not already there. */
3727                         if ((le->fixup & TODO_MODE) == 0)
3728                                 le->mode = mode;
3729                 } else {
3730                         r = set_fflags_platform(a, a->fd,
3731                             a->name, mode, set, clear);
3732                         if (r != ARCHIVE_OK)
3733                                 return (r);
3734                 }
3735         }
3736         return (ARCHIVE_OK);
3737 }
3738
3739 static int
3740 clear_nochange_fflags(struct archive_write_disk *a)
3741 {
3742         mode_t          mode = archive_entry_mode(a->entry);
3743         const int nochange_flags = 0
3744 #ifdef SF_IMMUTABLE
3745             | SF_IMMUTABLE
3746 #endif
3747 #ifdef UF_IMMUTABLE
3748             | UF_IMMUTABLE
3749 #endif
3750 #ifdef SF_APPEND
3751             | SF_APPEND
3752 #endif
3753 #ifdef UF_APPEND
3754             | UF_APPEND
3755 #endif
3756 #ifdef EXT2_APPEND_FL
3757             | EXT2_APPEND_FL
3758 #endif
3759 #ifdef EXT2_IMMUTABLE_FL
3760             | EXT2_IMMUTABLE_FL
3761 #endif
3762         ;
3763
3764         return (set_fflags_platform(a, a->fd, a->name, mode, 0,
3765             nochange_flags));
3766 }
3767
3768
3769 #if ( defined(HAVE_LCHFLAGS) || defined(HAVE_CHFLAGS) || defined(HAVE_FCHFLAGS) ) && defined(HAVE_STRUCT_STAT_ST_FLAGS)
3770 /*
3771  * BSD reads flags using stat() and sets them with one of {f,l,}chflags()
3772  */
3773 static int
3774 set_fflags_platform(struct archive_write_disk *a, int fd, const char *name,
3775     mode_t mode, unsigned long set, unsigned long clear)
3776 {
3777         int r;
3778         const int sf_mask = 0
3779 #ifdef SF_APPEND
3780             | SF_APPEND
3781 #endif
3782 #ifdef SF_ARCHIVED
3783             | SF_ARCHIVED
3784 #endif
3785 #ifdef SF_IMMUTABLE
3786             | SF_IMMUTABLE
3787 #endif
3788 #ifdef SF_NOUNLINK
3789             | SF_NOUNLINK
3790 #endif
3791         ;
3792         (void)mode; /* UNUSED */
3793
3794         if (set == 0  && clear == 0)
3795                 return (ARCHIVE_OK);
3796
3797         /*
3798          * XXX Is the stat here really necessary?  Or can I just use
3799          * the 'set' flags directly?  In particular, I'm not sure
3800          * about the correct approach if we're overwriting an existing
3801          * file that already has flags on it. XXX
3802          */
3803         if ((r = lazy_stat(a)) != ARCHIVE_OK)
3804                 return (r);
3805
3806         a->st.st_flags &= ~clear;
3807         a->st.st_flags |= set;
3808
3809         /* Only super-user may change SF_* flags */
3810
3811         if (a->user_uid != 0)
3812                 a->st.st_flags &= ~sf_mask;
3813
3814 #ifdef HAVE_FCHFLAGS
3815         /* If platform has fchflags() and we were given an fd, use it. */
3816         if (fd >= 0 && fchflags(fd, a->st.st_flags) == 0)
3817                 return (ARCHIVE_OK);
3818 #endif
3819         /*
3820          * If we can't use the fd to set the flags, we'll use the
3821          * pathname to set flags.  We prefer lchflags() but will use
3822          * chflags() if we must.
3823          */
3824 #ifdef HAVE_LCHFLAGS
3825         if (lchflags(name, a->st.st_flags) == 0)
3826                 return (ARCHIVE_OK);
3827 #elif defined(HAVE_CHFLAGS)
3828         if (S_ISLNK(a->st.st_mode)) {
3829                 archive_set_error(&a->archive, errno,
3830                     "Can't set file flags on symlink.");
3831                 return (ARCHIVE_WARN);
3832         }
3833         if (chflags(name, a->st.st_flags) == 0)
3834                 return (ARCHIVE_OK);
3835 #endif
3836         archive_set_error(&a->archive, errno,
3837             "Failed to set file flags");
3838         return (ARCHIVE_WARN);
3839 }
3840
3841 #elif (defined(FS_IOC_GETFLAGS) && defined(FS_IOC_SETFLAGS) && \
3842        defined(HAVE_WORKING_FS_IOC_GETFLAGS)) || \
3843       (defined(EXT2_IOC_GETFLAGS) && defined(EXT2_IOC_SETFLAGS) && \
3844        defined(HAVE_WORKING_EXT2_IOC_GETFLAGS))
3845 /*
3846  * Linux uses ioctl() to read and write file flags.
3847  */
3848 static int
3849 set_fflags_platform(struct archive_write_disk *a, int fd, const char *name,
3850     mode_t mode, unsigned long set, unsigned long clear)
3851 {
3852         int              ret;
3853         int              myfd = fd;
3854         int newflags, oldflags;
3855         /*
3856          * Linux has no define for the flags that are only settable by
3857          * the root user.  This code may seem a little complex, but
3858          * there seem to be some Linux systems that lack these
3859          * defines. (?)  The code below degrades reasonably gracefully
3860          * if sf_mask is incomplete.
3861          */
3862         const int sf_mask = 0
3863 #if defined(FS_IMMUTABLE_FL)
3864             | FS_IMMUTABLE_FL
3865 #elif defined(EXT2_IMMUTABLE_FL)
3866             | EXT2_IMMUTABLE_FL
3867 #endif
3868 #if defined(FS_APPEND_FL)
3869             | FS_APPEND_FL
3870 #elif defined(EXT2_APPEND_FL)
3871             | EXT2_APPEND_FL
3872 #endif
3873 #if defined(FS_JOURNAL_DATA_FL)
3874             | FS_JOURNAL_DATA_FL
3875 #endif
3876         ;
3877
3878         if (set == 0 && clear == 0)
3879                 return (ARCHIVE_OK);
3880         /* Only regular files and dirs can have flags. */
3881         if (!S_ISREG(mode) && !S_ISDIR(mode))
3882                 return (ARCHIVE_OK);
3883
3884         /* If we weren't given an fd, open it ourselves. */
3885         if (myfd < 0) {
3886                 myfd = open(name, O_RDONLY | O_NONBLOCK | O_BINARY | O_CLOEXEC);
3887                 __archive_ensure_cloexec_flag(myfd);
3888         }
3889         if (myfd < 0)
3890                 return (ARCHIVE_OK);
3891
3892         /*
3893          * XXX As above, this would be way simpler if we didn't have
3894          * to read the current flags from disk. XXX
3895          */
3896         ret = ARCHIVE_OK;
3897
3898         /* Read the current file flags. */
3899         if (ioctl(myfd,
3900 #ifdef FS_IOC_GETFLAGS
3901             FS_IOC_GETFLAGS,
3902 #else
3903             EXT2_IOC_GETFLAGS,
3904 #endif
3905             &oldflags) < 0)
3906                 goto fail;
3907
3908         /* Try setting the flags as given. */
3909         newflags = (oldflags & ~clear) | set;
3910         if (ioctl(myfd,
3911 #ifdef FS_IOC_SETFLAGS
3912             FS_IOC_SETFLAGS,
3913 #else
3914             EXT2_IOC_SETFLAGS,
3915 #endif
3916             &newflags) >= 0)
3917                 goto cleanup;
3918         if (errno != EPERM)
3919                 goto fail;
3920
3921         /* If we couldn't set all the flags, try again with a subset. */
3922         newflags &= ~sf_mask;
3923         oldflags &= sf_mask;
3924         newflags |= oldflags;
3925         if (ioctl(myfd,
3926 #ifdef FS_IOC_SETFLAGS
3927             FS_IOC_SETFLAGS,
3928 #else
3929             EXT2_IOC_SETFLAGS,
3930 #endif
3931             &newflags) >= 0)
3932                 goto cleanup;
3933
3934         /* We couldn't set the flags, so report the failure. */
3935 fail:
3936         archive_set_error(&a->archive, errno,
3937             "Failed to set file flags");
3938         ret = ARCHIVE_WARN;
3939 cleanup:
3940         if (fd < 0)
3941                 close(myfd);
3942         return (ret);
3943 }
3944
3945 #else
3946
3947 /*
3948  * Of course, some systems have neither BSD chflags() nor Linux' flags
3949  * support through ioctl().
3950  */
3951 static int
3952 set_fflags_platform(struct archive_write_disk *a, int fd, const char *name,
3953     mode_t mode, unsigned long set, unsigned long clear)
3954 {
3955         (void)a; /* UNUSED */
3956         (void)fd; /* UNUSED */
3957         (void)name; /* UNUSED */
3958         (void)mode; /* UNUSED */
3959         (void)set; /* UNUSED */
3960         (void)clear; /* UNUSED */
3961         return (ARCHIVE_OK);
3962 }
3963
3964 #endif /* __linux */
3965
3966 #ifndef HAVE_COPYFILE_H
3967 /* Default is to simply drop Mac extended metadata. */
3968 static int
3969 set_mac_metadata(struct archive_write_disk *a, const char *pathname,
3970                  const void *metadata, size_t metadata_size)
3971 {
3972         (void)a; /* UNUSED */
3973         (void)pathname; /* UNUSED */
3974         (void)metadata; /* UNUSED */
3975         (void)metadata_size; /* UNUSED */
3976         return (ARCHIVE_OK);
3977 }
3978
3979 static int
3980 fixup_appledouble(struct archive_write_disk *a, const char *pathname)
3981 {
3982         (void)a; /* UNUSED */
3983         (void)pathname; /* UNUSED */
3984         return (ARCHIVE_OK);
3985 }
3986 #else
3987
3988 /*
3989  * On Mac OS, we use copyfile() to unpack the metadata and
3990  * apply it to the target file.
3991  */
3992
3993 #if defined(HAVE_SYS_XATTR_H)
3994 static int
3995 copy_xattrs(struct archive_write_disk *a, int tmpfd, int dffd)
3996 {
3997         ssize_t xattr_size;
3998         char *xattr_names = NULL, *xattr_val = NULL;
3999         int ret = ARCHIVE_OK, xattr_i;
4000
4001         xattr_size = flistxattr(tmpfd, NULL, 0, 0);
4002         if (xattr_size == -1) {
4003                 archive_set_error(&a->archive, errno,
4004                     "Failed to read metadata(xattr)");
4005                 ret = ARCHIVE_WARN;
4006                 goto exit_xattr;
4007         }
4008         xattr_names = malloc(xattr_size);
4009         if (xattr_names == NULL) {
4010                 archive_set_error(&a->archive, ENOMEM,
4011                     "Can't allocate memory for metadata(xattr)");
4012                 ret = ARCHIVE_FATAL;
4013                 goto exit_xattr;
4014         }
4015         xattr_size = flistxattr(tmpfd, xattr_names, xattr_size, 0);
4016         if (xattr_size == -1) {
4017                 archive_set_error(&a->archive, errno,
4018                     "Failed to read metadata(xattr)");
4019                 ret = ARCHIVE_WARN;
4020                 goto exit_xattr;
4021         }
4022         for (xattr_i = 0; xattr_i < xattr_size;
4023             xattr_i += strlen(xattr_names + xattr_i) + 1) {
4024                 char *xattr_val_saved;
4025                 ssize_t s;
4026                 int f;
4027
4028                 s = fgetxattr(tmpfd, xattr_names + xattr_i, NULL, 0, 0, 0);
4029                 if (s == -1) {
4030                         archive_set_error(&a->archive, errno,
4031                             "Failed to get metadata(xattr)");
4032                         ret = ARCHIVE_WARN;
4033                         goto exit_xattr;
4034                 }
4035                 xattr_val_saved = xattr_val;
4036                 xattr_val = realloc(xattr_val, s);
4037                 if (xattr_val == NULL) {
4038                         archive_set_error(&a->archive, ENOMEM,
4039                             "Failed to get metadata(xattr)");
4040                         ret = ARCHIVE_WARN;
4041                         free(xattr_val_saved);
4042                         goto exit_xattr;
4043                 }
4044                 s = fgetxattr(tmpfd, xattr_names + xattr_i, xattr_val, s, 0, 0);
4045                 if (s == -1) {
4046                         archive_set_error(&a->archive, errno,
4047                             "Failed to get metadata(xattr)");
4048                         ret = ARCHIVE_WARN;
4049                         goto exit_xattr;
4050                 }
4051                 f = fsetxattr(dffd, xattr_names + xattr_i, xattr_val, s, 0, 0);
4052                 if (f == -1) {
4053                         archive_set_error(&a->archive, errno,
4054                             "Failed to get metadata(xattr)");
4055                         ret = ARCHIVE_WARN;
4056                         goto exit_xattr;
4057                 }
4058         }
4059 exit_xattr:
4060         free(xattr_names);
4061         free(xattr_val);
4062         return (ret);
4063 }
4064 #endif
4065
4066 static int
4067 copy_acls(struct archive_write_disk *a, int tmpfd, int dffd)
4068 {
4069 #ifndef HAVE_SYS_ACL_H
4070         return 0;
4071 #else
4072         acl_t acl, dfacl = NULL;
4073         int acl_r, ret = ARCHIVE_OK;
4074
4075         acl = acl_get_fd(tmpfd);
4076         if (acl == NULL) {
4077                 if (errno == ENOENT)
4078                         /* There are not any ACLs. */
4079                         return (ret);
4080                 archive_set_error(&a->archive, errno,
4081                     "Failed to get metadata(acl)");
4082                 ret = ARCHIVE_WARN;
4083                 goto exit_acl;
4084         }
4085         dfacl = acl_dup(acl);
4086         acl_r = acl_set_fd(dffd, dfacl);
4087         if (acl_r == -1) {
4088                 archive_set_error(&a->archive, errno,
4089                     "Failed to get metadata(acl)");
4090                 ret = ARCHIVE_WARN;
4091                 goto exit_acl;
4092         }
4093 exit_acl:
4094         if (acl)
4095                 acl_free(acl);
4096         if (dfacl)
4097                 acl_free(dfacl);
4098         return (ret);
4099 #endif
4100 }
4101
4102 static int
4103 create_tempdatafork(struct archive_write_disk *a, const char *pathname)
4104 {
4105         struct archive_string tmpdatafork;
4106         int tmpfd;
4107
4108         archive_string_init(&tmpdatafork);
4109         archive_strcpy(&tmpdatafork, "tar.md.XXXXXX");
4110         tmpfd = mkstemp(tmpdatafork.s);
4111         if (tmpfd < 0) {
4112                 archive_set_error(&a->archive, errno,
4113                     "Failed to mkstemp");
4114                 archive_string_free(&tmpdatafork);
4115                 return (-1);
4116         }
4117         if (copyfile(pathname, tmpdatafork.s, 0,
4118             COPYFILE_UNPACK | COPYFILE_NOFOLLOW
4119             | COPYFILE_ACL | COPYFILE_XATTR) < 0) {
4120                 archive_set_error(&a->archive, errno,
4121                     "Failed to restore metadata");
4122                 close(tmpfd);
4123                 tmpfd = -1;
4124         }
4125         unlink(tmpdatafork.s);
4126         archive_string_free(&tmpdatafork);
4127         return (tmpfd);
4128 }
4129
4130 static int
4131 copy_metadata(struct archive_write_disk *a, const char *metadata,
4132     const char *datafork, int datafork_compressed)
4133 {
4134         int ret = ARCHIVE_OK;
4135
4136         if (datafork_compressed) {
4137                 int dffd, tmpfd;
4138
4139                 tmpfd = create_tempdatafork(a, metadata);
4140                 if (tmpfd == -1)
4141                         return (ARCHIVE_WARN);
4142
4143                 /*
4144                  * Do not open the data fork compressed by HFS+ compression
4145                  * with at least a writing mode(O_RDWR or O_WRONLY). it
4146                  * makes the data fork uncompressed.
4147                  */
4148                 dffd = open(datafork, 0);
4149                 if (dffd == -1) {
4150                         archive_set_error(&a->archive, errno,
4151                             "Failed to open the data fork for metadata");
4152                         close(tmpfd);
4153                         return (ARCHIVE_WARN);
4154                 }
4155
4156 #if defined(HAVE_SYS_XATTR_H)
4157                 ret = copy_xattrs(a, tmpfd, dffd);
4158                 if (ret == ARCHIVE_OK)
4159 #endif
4160                         ret = copy_acls(a, tmpfd, dffd);
4161                 close(tmpfd);
4162                 close(dffd);
4163         } else {
4164                 if (copyfile(metadata, datafork, 0,
4165                     COPYFILE_UNPACK | COPYFILE_NOFOLLOW
4166                     | COPYFILE_ACL | COPYFILE_XATTR) < 0) {
4167                         archive_set_error(&a->archive, errno,
4168                             "Failed to restore metadata");
4169                         ret = ARCHIVE_WARN;
4170                 }
4171         }
4172         return (ret);
4173 }
4174
4175 static int
4176 set_mac_metadata(struct archive_write_disk *a, const char *pathname,
4177                  const void *metadata, size_t metadata_size)
4178 {
4179         struct archive_string tmp;
4180         ssize_t written;
4181         int fd;
4182         int ret = ARCHIVE_OK;
4183
4184         /* This would be simpler if copyfile() could just accept the
4185          * metadata as a block of memory; then we could sidestep this
4186          * silly dance of writing the data to disk just so that
4187          * copyfile() can read it back in again. */
4188         archive_string_init(&tmp);
4189         archive_strcpy(&tmp, pathname);
4190         archive_strcat(&tmp, ".XXXXXX");
4191         fd = mkstemp(tmp.s);
4192
4193         if (fd < 0) {
4194                 archive_set_error(&a->archive, errno,
4195                                   "Failed to restore metadata");
4196                 archive_string_free(&tmp);
4197                 return (ARCHIVE_WARN);
4198         }
4199         written = write(fd, metadata, metadata_size);
4200         close(fd);
4201         if ((size_t)written != metadata_size) {
4202                 archive_set_error(&a->archive, errno,
4203                                   "Failed to restore metadata");
4204                 ret = ARCHIVE_WARN;
4205         } else {
4206                 int compressed;
4207
4208 #if defined(UF_COMPRESSED)
4209                 if ((a->todo & TODO_HFS_COMPRESSION) != 0 &&
4210                     (ret = lazy_stat(a)) == ARCHIVE_OK)
4211                         compressed = a->st.st_flags & UF_COMPRESSED;
4212                 else
4213 #endif
4214                         compressed = 0;
4215                 ret = copy_metadata(a, tmp.s, pathname, compressed);
4216         }
4217         unlink(tmp.s);
4218         archive_string_free(&tmp);
4219         return (ret);
4220 }
4221
4222 static int
4223 fixup_appledouble(struct archive_write_disk *a, const char *pathname)
4224 {
4225         char buff[8];
4226         struct stat st;
4227         const char *p;
4228         struct archive_string datafork;
4229         int fd = -1, ret = ARCHIVE_OK;
4230
4231         archive_string_init(&datafork);
4232         /* Check if the current file name is a type of the resource
4233          * fork file. */
4234         p = strrchr(pathname, '/');
4235         if (p == NULL)
4236                 p = pathname;
4237         else
4238                 p++;
4239         if (p[0] != '.' || p[1] != '_')
4240                 goto skip_appledouble;
4241
4242         /*
4243          * Check if the data fork file exists.
4244          *
4245          * TODO: Check if this write disk object has handled it.
4246          */
4247         archive_strncpy(&datafork, pathname, p - pathname);
4248         archive_strcat(&datafork, p + 2);
4249         if (lstat(datafork.s, &st) == -1 ||
4250             (st.st_mode & AE_IFMT) != AE_IFREG)
4251                 goto skip_appledouble;
4252
4253         /*
4254          * Check if the file is in the AppleDouble form.
4255          */
4256         fd = open(pathname, O_RDONLY | O_BINARY | O_CLOEXEC);
4257         __archive_ensure_cloexec_flag(fd);
4258         if (fd == -1) {
4259                 archive_set_error(&a->archive, errno,
4260                     "Failed to open a restoring file");
4261                 ret = ARCHIVE_WARN;
4262                 goto skip_appledouble;
4263         }
4264         if (read(fd, buff, 8) == -1) {
4265                 archive_set_error(&a->archive, errno,
4266                     "Failed to read a restoring file");
4267                 close(fd);
4268                 ret = ARCHIVE_WARN;
4269                 goto skip_appledouble;
4270         }
4271         close(fd);
4272         /* Check AppleDouble Magic Code. */
4273         if (archive_be32dec(buff) != 0x00051607)
4274                 goto skip_appledouble;
4275         /* Check AppleDouble Version. */
4276         if (archive_be32dec(buff+4) != 0x00020000)
4277                 goto skip_appledouble;
4278
4279         ret = copy_metadata(a, pathname, datafork.s,
4280 #if defined(UF_COMPRESSED)
4281             st.st_flags & UF_COMPRESSED);
4282 #else
4283             0);
4284 #endif
4285         if (ret == ARCHIVE_OK) {
4286                 unlink(pathname);
4287                 ret = ARCHIVE_EOF;
4288         }
4289 skip_appledouble:
4290         archive_string_free(&datafork);
4291         return (ret);
4292 }
4293 #endif
4294
4295 #if ARCHIVE_XATTR_LINUX || ARCHIVE_XATTR_DARWIN || ARCHIVE_XATTR_AIX
4296 /*
4297  * Restore extended attributes -  Linux, Darwin and AIX implementations:
4298  * AIX' ea interface is syntaxwise identical to the Linux xattr interface.
4299  */
4300 static int
4301 set_xattrs(struct archive_write_disk *a)
4302 {
4303         struct archive_entry *entry = a->entry;
4304         struct archive_string errlist;
4305         int ret = ARCHIVE_OK;
4306         int i = archive_entry_xattr_reset(entry);
4307         short fail = 0;
4308
4309         archive_string_init(&errlist);
4310
4311         while (i--) {
4312                 const char *name;
4313                 const void *value;
4314                 size_t size;
4315                 int e;
4316
4317                 archive_entry_xattr_next(entry, &name, &value, &size);
4318
4319                 if (name == NULL)
4320                         continue;
4321 #if ARCHIVE_XATTR_LINUX
4322                 /* Linux: quietly skip POSIX.1e ACL extended attributes */
4323                 if (strncmp(name, "system.", 7) == 0 &&
4324                    (strcmp(name + 7, "posix_acl_access") == 0 ||
4325                     strcmp(name + 7, "posix_acl_default") == 0))
4326                         continue;
4327                 if (strncmp(name, "trusted.SGI_", 12) == 0 &&
4328                    (strcmp(name + 12, "ACL_DEFAULT") == 0 ||
4329                     strcmp(name + 12, "ACL_FILE") == 0))
4330                         continue;
4331
4332                 /* Linux: xfsroot namespace is obsolete and unsupported */
4333                 if (strncmp(name, "xfsroot.", 8) == 0) {
4334                         fail = 1;
4335                         archive_strcat(&errlist, name);
4336                         archive_strappend_char(&errlist, ' ');
4337                         continue;
4338                 }
4339 #endif
4340
4341                 if (a->fd >= 0) {
4342 #if ARCHIVE_XATTR_LINUX
4343                         e = fsetxattr(a->fd, name, value, size, 0);
4344 #elif ARCHIVE_XATTR_DARWIN
4345                         e = fsetxattr(a->fd, name, value, size, 0, 0);
4346 #elif ARCHIVE_XATTR_AIX
4347                         e = fsetea(a->fd, name, value, size, 0);
4348 #endif
4349                 } else {
4350 #if ARCHIVE_XATTR_LINUX
4351                         e = lsetxattr(archive_entry_pathname(entry),
4352                             name, value, size, 0);
4353 #elif ARCHIVE_XATTR_DARWIN
4354                         e = setxattr(archive_entry_pathname(entry),
4355                             name, value, size, 0, XATTR_NOFOLLOW);
4356 #elif ARCHIVE_XATTR_AIX
4357                         e = lsetea(archive_entry_pathname(entry),
4358                             name, value, size, 0);
4359 #endif
4360                 }
4361                 if (e == -1) {
4362                         ret = ARCHIVE_WARN;
4363                         archive_strcat(&errlist, name);
4364                         archive_strappend_char(&errlist, ' ');
4365                         if (errno != ENOTSUP && errno != ENOSYS)
4366                                 fail = 1;
4367                 }
4368         }
4369
4370         if (ret == ARCHIVE_WARN) {
4371                 if (fail && errlist.length > 0) {
4372                         errlist.length--;
4373                         errlist.s[errlist.length] = '\0';
4374                         archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC,
4375                             "Cannot restore extended attributes: %s",
4376                             errlist.s);
4377                 } else
4378                         archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC,
4379                             "Cannot restore extended "
4380                             "attributes on this file system.");
4381         }
4382
4383         archive_string_free(&errlist);
4384         return (ret);
4385 }
4386 #elif ARCHIVE_XATTR_FREEBSD
4387 /*
4388  * Restore extended attributes -  FreeBSD implementation
4389  */
4390 static int
4391 set_xattrs(struct archive_write_disk *a)
4392 {
4393         struct archive_entry *entry = a->entry;
4394         struct archive_string errlist;
4395         int ret = ARCHIVE_OK;
4396         int i = archive_entry_xattr_reset(entry);
4397         short fail = 0;
4398
4399         archive_string_init(&errlist);
4400
4401         while (i--) {
4402                 const char *name;
4403                 const void *value;
4404                 size_t size;
4405                 archive_entry_xattr_next(entry, &name, &value, &size);
4406                 if (name != NULL) {
4407                         ssize_t e;
4408                         int namespace;
4409
4410                         if (strncmp(name, "user.", 5) == 0) {
4411                                 /* "user." attributes go to user namespace */
4412                                 name += 5;
4413                                 namespace = EXTATTR_NAMESPACE_USER;
4414                         } else {
4415                                 /* Other namespaces are unsupported */
4416                                 archive_strcat(&errlist, name);
4417                                 archive_strappend_char(&errlist, ' ');
4418                                 fail = 1;
4419                                 ret = ARCHIVE_WARN;
4420                                 continue;
4421                         }
4422
4423                         if (a->fd >= 0) {
4424                                 e = extattr_set_fd(a->fd, namespace, name,
4425                                     value, size);
4426                         } else {
4427                                 e = extattr_set_link(
4428                                     archive_entry_pathname(entry), namespace,
4429                                     name, value, size);
4430                         }
4431                         if (e != (ssize_t)size) {
4432                                 archive_strcat(&errlist, name);
4433                                 archive_strappend_char(&errlist, ' ');
4434                                 ret = ARCHIVE_WARN;
4435                                 if (errno != ENOTSUP && errno != ENOSYS)
4436                                         fail = 1;
4437                         }
4438                 }
4439         }
4440
4441         if (ret == ARCHIVE_WARN) {
4442                 if (fail && errlist.length > 0) {
4443                         errlist.length--;
4444                         errlist.s[errlist.length] = '\0';
4445
4446                         archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC,
4447                             "Cannot restore extended attributes: %s",
4448                             errlist.s);
4449                 } else
4450                         archive_set_error(&a->archive, ARCHIVE_ERRNO_MISC,
4451                             "Cannot restore extended "
4452                             "attributes on this file system.");
4453         }
4454
4455         archive_string_free(&errlist);
4456         return (ret);
4457 }
4458 #else
4459 /*
4460  * Restore extended attributes - stub implementation for unsupported systems
4461  */
4462 static int
4463 set_xattrs(struct archive_write_disk *a)
4464 {
4465         static int warning_done = 0;
4466
4467         /* If there aren't any extended attributes, then it's okay not
4468          * to extract them, otherwise, issue a single warning. */
4469         if (archive_entry_xattr_count(a->entry) != 0 && !warning_done) {
4470                 warning_done = 1;
4471                 archive_set_error(&a->archive, ARCHIVE_ERRNO_FILE_FORMAT,
4472                     "Cannot restore extended attributes on this system");
4473                 return (ARCHIVE_WARN);
4474         }
4475         /* Warning was already emitted; suppress further warnings. */
4476         return (ARCHIVE_OK);
4477 }
4478 #endif
4479
4480 /*
4481  * Test if file on disk is older than entry.
4482  */
4483 static int
4484 older(struct stat *st, struct archive_entry *entry)
4485 {
4486         /* First, test the seconds and return if we have a definite answer. */
4487         /* Definitely older. */
4488         if (to_int64_time(st->st_mtime) < to_int64_time(archive_entry_mtime(entry)))
4489                 return (1);
4490         /* Definitely younger. */
4491         if (to_int64_time(st->st_mtime) > to_int64_time(archive_entry_mtime(entry)))
4492                 return (0);
4493         /* If this platform supports fractional seconds, try those. */
4494 #if HAVE_STRUCT_STAT_ST_MTIMESPEC_TV_NSEC
4495         /* Definitely older. */
4496         if (st->st_mtimespec.tv_nsec < archive_entry_mtime_nsec(entry))
4497                 return (1);
4498 #elif HAVE_STRUCT_STAT_ST_MTIM_TV_NSEC
4499         /* Definitely older. */
4500         if (st->st_mtim.tv_nsec < archive_entry_mtime_nsec(entry))
4501                 return (1);
4502 #elif HAVE_STRUCT_STAT_ST_MTIME_N
4503         /* older. */
4504         if (st->st_mtime_n < archive_entry_mtime_nsec(entry))
4505                 return (1);
4506 #elif HAVE_STRUCT_STAT_ST_UMTIME
4507         /* older. */
4508         if (st->st_umtime * 1000 < archive_entry_mtime_nsec(entry))
4509                 return (1);
4510 #elif HAVE_STRUCT_STAT_ST_MTIME_USEC
4511         /* older. */
4512         if (st->st_mtime_usec * 1000 < archive_entry_mtime_nsec(entry))
4513                 return (1);
4514 #else
4515         /* This system doesn't have high-res timestamps. */
4516 #endif
4517         /* Same age or newer, so not older. */
4518         return (0);
4519 }
4520
4521 #ifndef ARCHIVE_ACL_SUPPORT
4522 int
4523 archive_write_disk_set_acls(struct archive *a, int fd, const char *name,
4524     struct archive_acl *abstract_acl, __LA_MODE_T mode)
4525 {
4526         (void)a; /* UNUSED */
4527         (void)fd; /* UNUSED */
4528         (void)name; /* UNUSED */
4529         (void)abstract_acl; /* UNUSED */
4530         (void)mode; /* UNUSED */
4531         return (ARCHIVE_OK);
4532 }
4533 #endif
4534
4535 #endif /* !_WIN32 || __CYGWIN__ */
4536