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