Merge from vendor branch LIBARCHIVE:
[dragonfly.git] / contrib / libarchive-1.3.1 / libarchive / archive_read_extract.c
1 /*-
2  * Copyright (c) 2003-2005 Tim Kientzle
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer
10  *    in this position and unchanged.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26
27 #include "archive_platform.h"
28 __FBSDID("$FreeBSD: src/lib/libarchive/archive_read_extract.c,v 1.47 2006/09/05 05:59:45 kientzle Exp $");
29
30 #include <sys/types.h>
31 #ifdef HAVE_SYS_ACL_H
32 #include <sys/acl.h>
33 #endif
34 #ifdef HAVE_ATTR_XATTR_H
35 #include <attr/xattr.h>
36 #endif
37 #ifdef HAVE_SYS_IOCTL_H
38 #include <sys/ioctl.h>
39 #endif
40 #include <sys/stat.h>
41 #include <sys/time.h>
42
43 #ifdef HAVE_EXT2FS_EXT2_FS_H
44 #include <ext2fs/ext2_fs.h>     /* for Linux file flags */
45 #endif
46 #include <errno.h>
47 #include <fcntl.h>
48 #include <grp.h>
49 #ifdef HAVE_LINUX_EXT2_FS_H
50 #include <linux/ext2_fs.h>      /* for Linux file flags */
51 #endif
52 #include <limits.h>
53 #include <pwd.h>
54 #include <stdio.h>
55 #include <stdlib.h>
56 #include <string.h>
57 #include <unistd.h>
58
59 #include "archive.h"
60 #include "archive_string.h"
61 #include "archive_entry.h"
62 #include "archive_private.h"
63
64 struct fixup_entry {
65         struct fixup_entry      *next;
66         mode_t                   mode;
67         int64_t                  mtime;
68         int64_t                  atime;
69         unsigned long            mtime_nanos;
70         unsigned long            atime_nanos;
71         unsigned long            fflags_set;
72         int                      fixup; /* bitmask of what needs fixing */
73         char                    *name;
74 };
75
76 #define FIXUP_MODE      1
77 #define FIXUP_TIMES     2
78 #define FIXUP_FFLAGS    4
79
80 struct bucket {
81         char    *name;
82         int      hash;
83         id_t     id;
84 };
85
86 struct extract {
87         mode_t                   umask;
88         mode_t                   default_dir_mode;
89         struct archive_string    create_parent_dir;
90         struct fixup_entry      *fixup_list;
91         struct fixup_entry      *current_fixup;
92
93         struct bucket ucache[127];
94         struct bucket gcache[127];
95
96         /*
97          * Cached stat data from disk for the current entry.
98          * If this is valid, pst points to st.  Otherwise,
99          * pst is null.
100          */
101         struct stat              st;
102         struct stat             *pst;
103 };
104
105 /* Default mode for dirs created automatically (will be modified by umask). */
106 #define DEFAULT_DIR_MODE 0777
107 /*
108  * Mode to use for newly-created dirs during extraction; the correct
109  * mode will be set at the end of the extraction.
110  */
111 #define SECURE_DIR_MODE 0700
112
113 static int      archive_extract_cleanup(struct archive *);
114 static int      extract_block_device(struct archive *,
115                     struct archive_entry *, int);
116 static int      extract_char_device(struct archive *,
117                     struct archive_entry *, int);
118 static int      extract_device(struct archive *,
119                     struct archive_entry *, int flags, mode_t mode);
120 static int      extract_dir(struct archive *, struct archive_entry *, int);
121 static int      extract_fifo(struct archive *, struct archive_entry *, int);
122 static int      extract_file(struct archive *, struct archive_entry *, int);
123 static int      extract_hard_link(struct archive *, struct archive_entry *, int);
124 static int      extract_symlink(struct archive *, struct archive_entry *, int);
125 static unsigned int     hash(const char *);
126 static gid_t    lookup_gid(struct archive *, const char *uname, gid_t);
127 static uid_t    lookup_uid(struct archive *, const char *uname, uid_t);
128 static int      create_dir(struct archive *, const char *, int flags);
129 static int      create_dir_mutable(struct archive *, char *, int flags);
130 static int      create_dir_recursive(struct archive *, char *, int flags);
131 static int      create_parent_dir(struct archive *, const char *, int flags);
132 static int      create_parent_dir_mutable(struct archive *, char *, int flags);
133 static int      restore_metadata(struct archive *, int fd,
134                     struct archive_entry *, int flags);
135 #ifdef HAVE_POSIX_ACL
136 static int      set_acl(struct archive *, int fd, struct archive_entry *,
137                     acl_type_t, int archive_entry_acl_type, const char *tn);
138 #endif
139 static int      set_acls(struct archive *, int fd, struct archive_entry *);
140 static int      set_xattrs(struct archive *, int fd, struct archive_entry *);
141 static int      set_fflags(struct archive *, int fd, const char *name, mode_t,
142                     unsigned long fflags_set, unsigned long fflags_clear);
143 static int      set_ownership(struct archive *, int fd, struct archive_entry *,
144                     int flags);
145 static int      set_perm(struct archive *, int fd, struct archive_entry *,
146                     int mode, int flags);
147 static int      set_time(struct archive *, int fd, struct archive_entry *, int);
148 static struct fixup_entry *sort_dir_list(struct fixup_entry *p);
149
150
151 /*
152  * Extract this entry to disk.
153  *
154  * TODO: Validate hardlinks.  According to the standards, we're
155  * supposed to check each extracted hardlink and squawk if it refers
156  * to a file that we didn't restore.  I'm not entirely convinced this
157  * is a good idea, but more importantly: Is there any way to validate
158  * hardlinks without keeping a complete list of filenames from the
159  * entire archive?? Ugh.
160  *
161  */
162 int
163 archive_read_extract(struct archive *a, struct archive_entry *entry, int flags)
164 {
165         mode_t mode;
166         struct extract *extract;
167         int ret;
168         int restore_pwd;
169         char *original_filename;
170
171         if (a->extract == NULL) {
172                 a->extract = malloc(sizeof(*a->extract));
173                 if (a->extract == NULL) {
174                         archive_set_error(a, ENOMEM, "Can't extract");
175                         return (ARCHIVE_FATAL);
176                 }
177                 a->cleanup_archive_extract = archive_extract_cleanup;
178                 memset(a->extract, 0, sizeof(*a->extract));
179         }
180         extract = a->extract;
181         umask(extract->umask = umask(0)); /* Read the current umask. */
182         extract->default_dir_mode = DEFAULT_DIR_MODE & ~extract->umask;
183         extract->pst = NULL;
184         extract->current_fixup = NULL;
185         restore_pwd = -1;
186         original_filename = NULL;
187
188         /* The following is not possible without fchdir.  <sigh> */
189 #ifdef HAVE_FCHDIR
190         /*
191          * If pathname is longer than PATH_MAX, record starting directory
192          * and chdir to a suitable intermediate dir.
193          */
194         if (strlen(archive_entry_pathname(entry)) > PATH_MAX) {
195                 char *intdir, *tail;
196
197                 restore_pwd = open(".", O_RDONLY);
198                 if (restore_pwd < 0) {
199                                 archive_set_error(a, errno,
200                                     "Unable to restore long pathname");
201                                 return (ARCHIVE_WARN);
202                 }
203
204                 /*
205                  * Yes, the copy here is necessary because we edit
206                  * the pathname in-place to create intermediate dirnames.
207                  */
208                 original_filename = strdup(archive_entry_pathname(entry));
209
210                 /*
211                  * "intdir" points to the initial dir section we're going
212                  * to remove, "tail" points to the remainder of the path.
213                  */
214                 intdir = tail = original_filename;
215                 while (strlen(tail) > PATH_MAX) {
216                         intdir = tail;
217
218                         /* Locate a dir prefix shorter than PATH_MAX. */
219                         tail = intdir + PATH_MAX - 8;
220                         while (tail > intdir && *tail != '/')
221                                 tail--;
222                         if (tail <= intdir) {
223                                 archive_set_error(a, EPERM,
224                                     "Path element too long");
225                                 ret = ARCHIVE_WARN;
226                                 goto cleanup;
227                         }
228
229                         /* Create intdir and chdir to it. */
230                         *tail = '\0'; /* Terminate dir portion */
231                         ret = create_dir(a, intdir, flags);
232                         if (ret == ARCHIVE_OK && chdir(intdir) != 0) {
233                                 archive_set_error(a, errno, "Couldn't chdir");
234                                 ret = ARCHIVE_WARN;
235                         }
236                         *tail = '/'; /* Restore the / we removed. */
237                         if (ret != ARCHIVE_OK)
238                                 goto cleanup;
239                         tail++;
240                 }
241                 archive_entry_set_pathname(entry, tail);
242         }
243 #endif
244
245         if (stat(archive_entry_pathname(entry), &extract->st) == 0)
246                 extract->pst = &extract->st;
247
248         if (extract->pst != NULL &&
249             extract->pst->st_dev == a->skip_file_dev &&
250             extract->pst->st_ino == a->skip_file_ino) {
251                 archive_set_error(a, 0, "Refusing to overwrite archive");
252                 ret = ARCHIVE_WARN;
253         } else if (archive_entry_hardlink(entry) != NULL)
254                 ret = extract_hard_link(a, entry, flags);
255         else {
256                 mode = archive_entry_mode(entry);
257                 switch (mode & S_IFMT) {
258                 default:
259                         /* Fall through, as required by POSIX. */
260                 case S_IFREG:
261                         ret = extract_file(a, entry, flags);
262                         break;
263                 case S_IFLNK:   /* Symlink */
264                         ret = extract_symlink(a, entry, flags);
265                         break;
266                 case S_IFCHR:
267                         ret = extract_char_device(a, entry, flags);
268                         break;
269                 case S_IFBLK:
270                         ret = extract_block_device(a, entry, flags);
271                         break;
272                 case S_IFDIR:
273                         ret = extract_dir(a, entry, flags);
274                         break;
275                 case S_IFIFO:
276                         ret = extract_fifo(a, entry, flags);
277                         break;
278                 }
279         }
280
281
282 cleanup:
283 #ifdef HAVE_FCHDIR
284         /* If we changed directory above, restore it here. */
285         if (restore_pwd >= 0 && original_filename != NULL) {
286                 fchdir(restore_pwd);
287                 close(restore_pwd);
288                 archive_entry_copy_pathname(entry, original_filename);
289                 free(original_filename);
290         }
291 #endif
292
293         return (ret);
294 }
295
296 /*
297  * Cleanup function for archive_extract.  Mostly, this involves processing
298  * the fixup list, which is used to address a number of problems:
299  *   * Dir permissions might prevent us from restoring a file in that
300  *     dir, so we restore the dir 0700 first, then correct the
301  *     mode at the end.
302  *   * Similarly, the act of restoring a file touches the directory
303  *     and changes the timestamp on the dir, so we have to touch-up dir
304  *     timestamps at the end as well.
305  *   * Some file flags can interfere with the restore by, for example,
306  *     preventing the creation of hardlinks to those files.
307  *
308  * Note that tar/cpio do not require that archives be in a particular
309  * order; there is no way to know when the last file has been restored
310  * within a directory, so there's no way to optimize the memory usage
311  * here by fixing up the directory any earlier than the
312  * end-of-archive.
313  *
314  * XXX TODO: Directory ACLs should be restored here, for the same
315  * reason we set directory perms here. XXX
316  *
317  * Registering this function (rather than calling it explicitly by
318  * name from archive_read_finish) reduces static link pollution, since
319  * applications that don't use this API won't get this file linked in.
320  */
321 static int
322 archive_extract_cleanup(struct archive *a)
323 {
324         struct fixup_entry *next, *p;
325         struct extract *extract;
326
327         /* Sort dir list so directories are fixed up in depth-first order. */
328         extract = a->extract;
329         p = sort_dir_list(extract->fixup_list);
330
331         while (p != NULL) {
332                 extract->pst = NULL; /* Mark stat cache as out-of-date. */
333                 if (p->fixup & FIXUP_TIMES) {
334                         struct timeval times[2];
335                         times[1].tv_sec = p->mtime;
336                         times[1].tv_usec = p->mtime_nanos / 1000;
337                         times[0].tv_sec = p->atime;
338                         times[0].tv_usec = p->atime_nanos / 1000;
339                         utimes(p->name, times);
340                 }
341                 if (p->fixup & FIXUP_MODE)
342                         chmod(p->name, p->mode);
343
344                 if (p->fixup & FIXUP_FFLAGS)
345                         set_fflags(a, -1, p->name, p->mode, p->fflags_set, 0);
346
347                 next = p->next;
348                 free(p->name);
349                 free(p);
350                 p = next;
351         }
352         extract->fixup_list = NULL;
353         archive_string_free(&extract->create_parent_dir);
354         free(a->extract);
355         a->extract = NULL;
356         return (ARCHIVE_OK);
357 }
358
359 /*
360  * Simple O(n log n) merge sort to order the fixup list.  In
361  * particular, we want to restore dir timestamps depth-first.
362  */
363 static struct fixup_entry *
364 sort_dir_list(struct fixup_entry *p)
365 {
366         struct fixup_entry *a, *b, *t;
367
368         if (p == NULL)
369                 return (NULL);
370         /* A one-item list is already sorted. */
371         if (p->next == NULL)
372                 return (p);
373
374         /* Step 1: split the list. */
375         t = p;
376         a = p->next->next;
377         while (a != NULL) {
378                 /* Step a twice, t once. */
379                 a = a->next;
380                 if (a != NULL)
381                         a = a->next;
382                 t = t->next;
383         }
384         /* Now, t is at the mid-point, so break the list here. */
385         b = t->next;
386         t->next = NULL;
387         a = p;
388
389         /* Step 2: Recursively sort the two sub-lists. */
390         a = sort_dir_list(a);
391         b = sort_dir_list(b);
392
393         /* Step 3: Merge the returned lists. */
394         /* Pick the first element for the merged list. */
395         if (strcmp(a->name, b->name) > 0) {
396                 t = p = a;
397                 a = a->next;
398         } else {
399                 t = p = b;
400                 b = b->next;
401         }
402
403         /* Always put the later element on the list first. */
404         while (a != NULL && b != NULL) {
405                 if (strcmp(a->name, b->name) > 0) {
406                         t->next = a;
407                         a = a->next;
408                 } else {
409                         t->next = b;
410                         b = b->next;
411                 }
412                 t = t->next;
413         }
414
415         /* Only one list is non-empty, so just splice it on. */
416         if (a != NULL)
417                 t->next = a;
418         if (b != NULL)
419                 t->next = b;
420
421         return (p);
422 }
423
424 /*
425  * Returns a new, initialized fixup entry.
426  *
427  * TODO: Reduce the memory requirements for this list by using a tree
428  * structure rather than a simple list of names.
429  */
430 static struct fixup_entry *
431 new_fixup(struct archive *a, const char *pathname)
432 {
433         struct extract *extract;
434         struct fixup_entry *fe;
435
436         extract = a->extract;
437         fe = malloc(sizeof(struct fixup_entry));
438         if (fe == NULL)
439                 return (NULL);
440         fe->next = extract->fixup_list;
441         extract->fixup_list = fe;
442         fe->fixup = 0;
443         fe->name = strdup(pathname);
444         return (fe);
445 }
446
447 /*
448  * Returns a fixup structure for the current entry.
449  */
450 static struct fixup_entry *
451 current_fixup(struct archive *a, const char *pathname)
452 {
453         struct extract *extract;
454
455         extract = a->extract;
456         if (extract->current_fixup == NULL)
457                 extract->current_fixup = new_fixup(a, pathname);
458         return (extract->current_fixup);
459 }
460
461 static int
462 extract_file(struct archive *a, struct archive_entry *entry, int flags)
463 {
464         struct extract *extract;
465         const char *name;
466         mode_t mode;
467         int fd, r, r2;
468
469         extract = a->extract;
470         name = archive_entry_pathname(entry);
471         mode = archive_entry_mode(entry) & 0777;
472         r = ARCHIVE_OK;
473
474         /*
475          * If we're not supposed to overwrite pre-existing files,
476          * use O_EXCL.  Otherwise, use O_TRUNC.
477          */
478         if (flags & (ARCHIVE_EXTRACT_UNLINK | ARCHIVE_EXTRACT_NO_OVERWRITE))
479                 fd = open(name, O_WRONLY | O_CREAT | O_EXCL, mode);
480         else
481                 fd = open(name, O_WRONLY | O_CREAT | O_TRUNC, mode);
482
483         /* Try removing a pre-existing file. */
484         if (fd < 0 && !(flags & ARCHIVE_EXTRACT_NO_OVERWRITE)) {
485                 unlink(name);
486                 fd = open(name, O_WRONLY | O_CREAT | O_EXCL, mode);
487         }
488
489         /* Might be a non-existent parent dir; try fixing that. */
490         if (fd < 0) {
491                 create_parent_dir(a, name, flags);
492                 fd = open(name, O_WRONLY | O_CREAT | O_EXCL, mode);
493         }
494         if (fd < 0) {
495                 archive_set_error(a, errno, "Can't open '%s'", name);
496                 return (ARCHIVE_WARN);
497         }
498         r = archive_read_data_into_fd(a, fd);
499         extract->pst = NULL; /* Cached stat data no longer valid. */
500         r2 = restore_metadata(a, fd, entry, flags);
501         close(fd);
502         return (err_combine(r, r2));
503 }
504
505 static int
506 extract_dir(struct archive *a, struct archive_entry *entry, int flags)
507 {
508         struct extract *extract;
509         struct fixup_entry *fe;
510         char *path, *p;
511
512         extract = a->extract;
513         extract->pst = NULL; /* Invalidate cached stat data. */
514
515         /* Copy path to mutable storage. */
516         archive_strcpy(&(extract->create_parent_dir),
517             archive_entry_pathname(entry));
518         path = extract->create_parent_dir.s;
519
520         if (*path == '\0') {
521                 archive_set_error(a, ARCHIVE_ERRNO_MISC,
522                     "Invalid empty pathname");
523                 return (ARCHIVE_WARN);
524         }
525
526         /* Deal with any troublesome trailing path elements. */
527         /* TODO: Someday, generalize this to remove '//' or '/./' from
528          * the middle of paths.  But, it should not compress '..' from
529          * the middle of paths.  It's a feature that restoring
530          * "a/../b" creates both 'a' and 'b' directories. */
531         for (;;) {
532                 /* Locate last element. */
533                 p = strrchr(path, '/');
534                 if (p != NULL)
535                         p++;
536                 else
537                         p = path;
538                 /* Trim trailing '/' unless that's the entire path. */
539                 if (p[0] == '\0' && p - 1 > path) {
540                         p[-1] = '\0';
541                         continue;
542                 }
543                 /* Trim trailing '.' unless that's the entire path. */
544                 if (p > path && p[0] == '.' && p[1] == '\0') {
545                         p[0] = '\0';
546                         continue;
547                 }
548                 /* Just exit on trailing '..'. */
549                 if (p[0] == '.' && p[1] == '.' && p[2] == '\0') {
550                         archive_set_error(a, ARCHIVE_ERRNO_MISC,
551                             "Can't restore directory '..'");
552                         return (ARCHIVE_WARN);
553                 }
554                 break;
555         }
556
557         if (mkdir(path, SECURE_DIR_MODE) == 0)
558                 goto success;
559
560         if (extract->pst == NULL && stat(path, &extract->st) == 0)
561                 extract->pst = &extract->st;
562
563         if (extract->pst != NULL) {
564                 extract->pst = &extract->st;
565                 /* If dir already exists, don't reset permissions. */
566                 if (S_ISDIR(extract->pst->st_mode))
567                         return (ARCHIVE_OK);
568                 /* It exists but isn't a dir. */
569                 if ((flags & ARCHIVE_EXTRACT_UNLINK))
570                         unlink(path);
571         } else {
572                 /* Doesn't already exist; try building the parent path. */
573                 if (create_parent_dir_mutable(a, path, flags) != ARCHIVE_OK)
574                         return (ARCHIVE_WARN);
575         }
576
577         /* One final attempt to create the dir. */
578         if (mkdir(path, SECURE_DIR_MODE) != 0) {
579                 archive_set_error(a, errno, "Can't create directory");
580                 return (ARCHIVE_WARN);
581         }
582
583 success:
584         /* Add this dir to the fixup list. */
585         fe = current_fixup(a, path);
586         fe->fixup |= FIXUP_MODE;
587         fe->mode = archive_entry_mode(entry);
588         if ((flags & ARCHIVE_EXTRACT_PERM) == 0)
589                 fe->mode &= ~extract->umask;
590         if (flags & ARCHIVE_EXTRACT_TIME) {
591                 fe->fixup |= FIXUP_TIMES;
592                 fe->mtime = archive_entry_mtime(entry);
593                 fe->mtime_nanos = archive_entry_mtime_nsec(entry);
594                 fe->atime = archive_entry_atime(entry);
595                 fe->atime_nanos = archive_entry_atime_nsec(entry);
596         }
597         /* For now, set the mode to SECURE_DIR_MODE. */
598         archive_entry_set_mode(entry, SECURE_DIR_MODE);
599         return (restore_metadata(a, -1, entry, flags));
600 }
601
602
603 /*
604  * Create the parent of the specified path.  Copy the provided
605  * path into mutable storage first.
606  */
607 static int
608 create_parent_dir(struct archive *a, const char *path, int flags)
609 {
610         int r;
611
612         /* Copy path to mutable storage. */
613         archive_strcpy(&(a->extract->create_parent_dir), path);
614         r = create_parent_dir_mutable(a, a->extract->create_parent_dir.s, flags);
615         return (r);
616 }
617
618 /*
619  * Like create_parent_dir, but creates the dir actually requested, not
620  * the parent.
621  */
622 static int
623 create_dir(struct archive *a, const char *path, int flags)
624 {
625         int r;
626         /* Copy path to mutable storage. */
627         archive_strcpy(&(a->extract->create_parent_dir), path);
628         r = create_dir_mutable(a, a->extract->create_parent_dir.s, flags);
629         return (r);
630 }
631
632 /*
633  * Create the parent directory of the specified path, assuming path
634  * is already in mutable storage.
635  */
636 static int
637 create_parent_dir_mutable(struct archive *a, char *path, int flags)
638 {
639         char *slash;
640         int r;
641
642         /* Remove tail element to obtain parent name. */
643         slash = strrchr(path, '/');
644         if (slash == NULL)
645                 return (ARCHIVE_OK);
646         *slash = '\0';
647         r = create_dir_mutable(a, path, flags);
648         *slash = '/';
649         return (r);
650 }
651
652 /*
653  * Create the specified dir, assuming path is already in
654  * mutable storage.
655  */
656 static int
657 create_dir_mutable(struct archive *a, char *path, int flags)
658 {
659         mode_t old_umask;
660         int r;
661
662         old_umask = umask(~SECURE_DIR_MODE);
663         r = create_dir_recursive(a, path, flags);
664         umask(old_umask);
665         return (r);
666 }
667
668 /*
669  * Create the specified dir, recursing to create parents as necessary.
670  *
671  * Returns ARCHIVE_OK if the path exists when we're done here.
672  * Otherwise, returns ARCHIVE_WARN.
673  */
674 static int
675 create_dir_recursive(struct archive *a, char *path, int flags)
676 {
677         struct stat st;
678         struct extract *extract;
679         struct fixup_entry *le;
680         char *slash, *base;
681         int r;
682
683         extract = a->extract;
684         r = ARCHIVE_OK;
685
686         /* Check for special names and just skip them. */
687         slash = strrchr(path, '/');
688         base = strrchr(path, '/');
689         if (slash == NULL)
690                 base = path;
691         else
692                 base = slash + 1;
693
694         if (base[0] == '\0' ||
695             (base[0] == '.' && base[1] == '\0') ||
696             (base[0] == '.' && base[1] == '.' && base[2] == '\0')) {
697                 /* Don't bother trying to create null path, '.', or '..'. */
698                 if (slash != NULL) {
699                         *slash = '\0';
700                         r = create_dir_recursive(a, path, flags);
701                         *slash = '/';
702                         return (r);
703                 }
704                 return (ARCHIVE_OK);
705         }
706
707         /*
708          * Yes, this should be stat() and not lstat().  Using lstat()
709          * here loses the ability to extract through symlinks.  Also note
710          * that this should not use the extract->st cache.
711          */
712         if (stat(path, &st) == 0) {
713                 if (S_ISDIR(st.st_mode))
714                         return (ARCHIVE_OK);
715                 if ((flags & ARCHIVE_EXTRACT_NO_OVERWRITE)) {
716                         archive_set_error(a, EEXIST,
717                             "Can't create directory '%s'", path);
718                         return (ARCHIVE_WARN);
719                 }
720                 if (unlink(path) != 0) {
721                         archive_set_error(a, errno,
722                             "Can't create directory '%s': "
723                             "Conflicting file cannot be removed");
724                         return (ARCHIVE_WARN);
725                 }
726         } else if (errno != ENOENT && errno != ENOTDIR) {
727                 /* Stat failed? */
728                 archive_set_error(a, errno, "Can't test directory '%s'", path);
729                 return (ARCHIVE_WARN);
730         } else if (slash != NULL) {
731                 *slash = '\0';
732                 r = create_dir_recursive(a, path, flags);
733                 *slash = '/';
734                 if (r != ARCHIVE_OK)
735                         return (r);
736         }
737
738         if (mkdir(path, SECURE_DIR_MODE) == 0) {
739                 le = new_fixup(a, path);
740                 le->fixup |= FIXUP_MODE;
741                 le->mode = extract->default_dir_mode;
742                 return (ARCHIVE_OK);
743         }
744
745         /*
746          * Without the following check, a/b/../b/c/d fails at the
747          * second visit to 'b', so 'd' can't be created.  Note that we
748          * don't add it to the fixup list here, as it's already been
749          * added.
750          */
751         if (stat(path, &st) == 0 && S_ISDIR(st.st_mode))
752                 return (ARCHIVE_OK);
753
754         archive_set_error(a, errno, "Failed to create dir '%s'", path);
755         return (ARCHIVE_WARN);
756 }
757
758 static int
759 extract_hard_link(struct archive *a, struct archive_entry *entry, int flags)
760 {
761         struct extract *extract;
762         int r;
763         const char *pathname;
764         const char *linkname;
765
766         extract = a->extract;
767         pathname = archive_entry_pathname(entry);
768         linkname = archive_entry_hardlink(entry);
769
770         /* Just remove any pre-existing file with this name. */
771         if (!(flags & ARCHIVE_EXTRACT_NO_OVERWRITE))
772                 unlink(pathname);
773
774         r = link(linkname, pathname);
775         extract->pst = NULL; /* Invalidate cached stat data. */
776
777         if (r != 0) {
778                 /* Might be a non-existent parent dir; try fixing that. */
779                 create_parent_dir(a, pathname, flags);
780                 r = link(linkname, pathname);
781         }
782
783         if (r != 0) {
784                 /* XXX Better error message here XXX */
785                 archive_set_error(a, errno,
786                     "Can't restore hardlink to '%s'", linkname);
787                 return (ARCHIVE_WARN);
788         }
789
790         /* Set ownership, time, permission information. */
791         r = restore_metadata(a, -1, entry, flags);
792         return (r);
793 }
794
795 static int
796 extract_symlink(struct archive *a, struct archive_entry *entry, int flags)
797 {
798         struct extract *extract;
799         int r;
800         const char *pathname;
801         const char *linkname;
802
803         extract = a->extract;
804         pathname = archive_entry_pathname(entry);
805         linkname = archive_entry_symlink(entry);
806
807         /* Just remove any pre-existing file with this name. */
808         if (!(flags & ARCHIVE_EXTRACT_NO_OVERWRITE))
809                 unlink(pathname);
810
811         r = symlink(linkname, pathname);
812         extract->pst = NULL; /* Invalidate cached stat data. */
813
814         if (r != 0) {
815                 /* Might be a non-existent parent dir; try fixing that. */
816                 create_parent_dir(a, pathname, flags);
817                 r = symlink(linkname, pathname);
818         }
819
820         if (r != 0) {
821                 /* XXX Better error message here XXX */
822                 archive_set_error(a, errno,
823                     "Can't restore symlink to '%s'", linkname);
824                 return (ARCHIVE_WARN);
825         }
826
827         r = restore_metadata(a, -1, entry, flags);
828         return (r);
829 }
830
831 static int
832 extract_device(struct archive *a, struct archive_entry *entry,
833     int flags, mode_t mode)
834 {
835         struct extract *extract;
836         int r;
837
838         extract = a->extract;
839         /* Just remove any pre-existing file with this name. */
840         if (!(flags & ARCHIVE_EXTRACT_NO_OVERWRITE))
841                 unlink(archive_entry_pathname(entry));
842
843         r = mknod(archive_entry_pathname(entry), mode,
844             archive_entry_rdev(entry));
845         extract->pst = NULL; /* Invalidate cached stat data. */
846
847         /* Might be a non-existent parent dir; try fixing that. */
848         if (r != 0 && errno == ENOENT) {
849                 create_parent_dir(a, archive_entry_pathname(entry), flags);
850                 r = mknod(archive_entry_pathname(entry), mode,
851                     archive_entry_rdev(entry));
852         }
853
854         if (r != 0) {
855                 archive_set_error(a, errno, "Can't restore device node");
856                 return (ARCHIVE_WARN);
857         }
858
859         r = restore_metadata(a, -1, entry, flags);
860         return (r);
861 }
862
863 static int
864 extract_char_device(struct archive *a, struct archive_entry *entry, int flags)
865 {
866         mode_t mode;
867
868         mode = (archive_entry_mode(entry) & ~S_IFMT) | S_IFCHR;
869         return (extract_device(a, entry, flags, mode));
870 }
871
872 static int
873 extract_block_device(struct archive *a, struct archive_entry *entry, int flags)
874 {
875         mode_t mode;
876
877         mode = (archive_entry_mode(entry) & ~S_IFMT) | S_IFBLK;
878         return (extract_device(a, entry, flags, mode));
879 }
880
881 static int
882 extract_fifo(struct archive *a, struct archive_entry *entry, int flags)
883 {
884         struct extract *extract;
885         int r;
886
887         extract = a->extract;
888         /* Just remove any pre-existing file with this name. */
889         if (!(flags & ARCHIVE_EXTRACT_NO_OVERWRITE))
890                 unlink(archive_entry_pathname(entry));
891
892         r = mkfifo(archive_entry_pathname(entry),
893             archive_entry_mode(entry));
894         extract->pst = NULL; /* Invalidate cached stat data. */
895
896         /* Might be a non-existent parent dir; try fixing that. */
897         if (r != 0 && errno == ENOENT) {
898                 create_parent_dir(a, archive_entry_pathname(entry), flags);
899                 r = mkfifo(archive_entry_pathname(entry),
900                     archive_entry_mode(entry));
901         }
902
903         if (r != 0) {
904                 archive_set_error(a, errno, "Can't restore fifo");
905                 return (ARCHIVE_WARN);
906         }
907
908         r = restore_metadata(a, -1, entry, flags);
909         return (r);
910 }
911
912 static int
913 restore_metadata(struct archive *a, int fd, struct archive_entry *entry, int flags)
914 {
915         int r, r2;
916
917         r = set_ownership(a, fd, entry, flags);
918         r2 = set_time(a, fd, entry, flags);
919         r = err_combine(r, r2);
920         r2 = set_perm(a, fd, entry, archive_entry_mode(entry), flags);
921         return (err_combine(r, r2));
922 }
923
924 static int
925 set_ownership(struct archive *a, int fd,
926     struct archive_entry *entry, int flags)
927 {
928         uid_t uid;
929         gid_t gid;
930
931         /* Not changed. */
932         if ((flags & ARCHIVE_EXTRACT_OWNER) == 0)
933                 return (ARCHIVE_OK);
934
935         uid = lookup_uid(a, archive_entry_uname(entry),
936             archive_entry_uid(entry));
937         gid = lookup_gid(a, archive_entry_gname(entry),
938             archive_entry_gid(entry));
939
940         /* If we know we can't change it, don't bother trying. */
941         if (a->user_uid != 0  &&  a->user_uid != uid)
942                 return (ARCHIVE_OK);
943
944 #ifdef HAVE_FCHOWN
945         if (fd >= 0 && fchown(fd, uid, gid) == 0)
946                 return (ARCHIVE_OK);
947 #endif
948
949 #ifdef HAVE_LCHOWN
950         if (lchown(archive_entry_pathname(entry), uid, gid))
951 #else
952         if (!S_ISLNK(archive_entry_mode(entry))
953             && chown(archive_entry_pathname(entry), uid, gid) != 0)
954 #endif
955         {
956                 archive_set_error(a, errno,
957                     "Can't set user=%d/group=%d for %s", uid, gid,
958                     archive_entry_pathname(entry));
959                 return (ARCHIVE_WARN);
960         }
961         return (ARCHIVE_OK);
962 }
963
964 static int
965 set_time(struct archive *a, int fd, struct archive_entry *entry, int flags)
966 {
967         const struct stat *st;
968         struct timeval times[2];
969
970         (void)a; /* UNUSED */
971         st = archive_entry_stat(entry);
972
973         if ((flags & ARCHIVE_EXTRACT_TIME) == 0)
974                 return (ARCHIVE_OK);
975         /* It's a waste of time to mess with dir timestamps here. */
976         if (S_ISDIR(archive_entry_mode(entry)))
977                 return (ARCHIVE_OK);
978
979         times[1].tv_sec = st->st_mtime;
980         times[1].tv_usec = ARCHIVE_STAT_MTIME_NANOS(st) / 1000;
981
982         times[0].tv_sec = st->st_atime;
983         times[0].tv_usec = ARCHIVE_STAT_ATIME_NANOS(st) / 1000;
984
985 #ifdef HAVE_FUTIMES
986         if (fd >= 0 && futimes(fd, times) == 0)
987                 return (ARCHIVE_OK);
988 #endif
989
990 #ifdef HAVE_LUTIMES
991         if (lutimes(archive_entry_pathname(entry), times) != 0) {
992 #else
993         if ((archive_entry_mode(entry) & S_IFMT) != S_IFLNK &&
994             utimes(archive_entry_pathname(entry), times) != 0) {
995 #endif
996                 archive_set_error(a, errno, "Can't update time for %s",
997                     archive_entry_pathname(entry));
998                 return (ARCHIVE_WARN);
999         }
1000
1001         /*
1002          * Note: POSIX does not provide a portable way to restore ctime.
1003          * (Apart from resetting the system clock, which is distasteful.)
1004          * So, any restoration of ctime will necessarily be OS-specific.
1005          */
1006
1007         /* XXX TODO: Can FreeBSD restore ctime? XXX */
1008
1009         return (ARCHIVE_OK);
1010 }
1011
1012 static int
1013 set_perm(struct archive *a, int fd, struct archive_entry *entry,
1014     int mode, int flags)
1015 {
1016         struct extract *extract;
1017         struct fixup_entry *le;
1018         const char *name;
1019         unsigned long    set, clear;
1020         int              r;
1021         int              critical_flags;
1022
1023         extract = a->extract;
1024
1025         /* Obey umask unless ARCHIVE_EXTRACT_PERM. */
1026         if ((flags & ARCHIVE_EXTRACT_PERM) == 0)
1027                 mode &= ~extract->umask; /* Enforce umask. */
1028         name = archive_entry_pathname(entry);
1029
1030         if (mode & (S_ISUID | S_ISGID)) {
1031                 if (extract->pst != NULL) {
1032                         /* Already have stat() data available. */
1033 #ifdef HAVE_FSTAT
1034                 } else if (fd >= 0 && fstat(fd, &extract->st) == 0) {
1035                         extract->pst = &extract->st;
1036 #endif
1037                 } else if (stat(name, &extract->st) == 0) {
1038                         extract->pst = &extract->st;
1039                 } else {
1040                         archive_set_error(a, errno,
1041                             "Couldn't stat file");
1042                         return (ARCHIVE_WARN);
1043                 }
1044
1045                 /*
1046                  * TODO: Use the uid/gid looked up in set_ownership
1047                  * above rather than the uid/gid stored in the entry.
1048                  */
1049                 if (extract->pst->st_uid != archive_entry_uid(entry))
1050                         mode &= ~ S_ISUID;
1051                 if (extract->pst->st_gid != archive_entry_gid(entry))
1052                         mode &= ~ S_ISGID;
1053         }
1054
1055         /*
1056          * Ensure we change permissions on the object we extracted,
1057          * and not any incidental symlink that might have gotten in
1058          * the way.
1059          */
1060         if (!S_ISLNK(archive_entry_mode(entry))) {
1061 #ifdef HAVE_FCHMOD
1062                 if (fd >= 0) {
1063                         if (fchmod(fd, mode) != 0) {
1064                                 archive_set_error(a, errno,
1065                                     "Can't set permissions");
1066                                 return (ARCHIVE_WARN);
1067                         }
1068                 } else
1069 #endif
1070                 if (chmod(name, mode) != 0) {
1071                         archive_set_error(a, errno, "Can't set permissions");
1072                         return (ARCHIVE_WARN);
1073                 }
1074 #ifdef HAVE_LCHMOD
1075         } else {
1076                 /*
1077                  * If lchmod() isn't supported, it's no big deal.
1078                  * Permissions on symlinks are actually ignored on
1079                  * most platforms.
1080                  */
1081                 if (lchmod(name, mode) != 0) {
1082                         archive_set_error(a, errno, "Can't set permissions");
1083                         return (ARCHIVE_WARN);
1084                 }
1085 #endif
1086         }
1087
1088         if (flags & ARCHIVE_EXTRACT_ACL) {
1089                 r = set_acls(a, fd, entry);
1090                 if (r != ARCHIVE_OK)
1091                         return (r);
1092         }
1093
1094         if (flags & ARCHIVE_EXTRACT_XATTR) {
1095                 r = set_xattrs(a, fd, entry);
1096                 if (r != ARCHIVE_OK)
1097                         return (r);
1098         }
1099
1100         /*
1101          * Make 'critical_flags' hold all file flags that can't be
1102          * immediately restored.  For example, on BSD systems,
1103          * SF_IMMUTABLE prevents hardlinks from being created, so
1104          * should not be set until after any hardlinks are created.  To
1105          * preserve some semblance of portability, this uses #ifdef
1106          * extensively.  Ugly, but it works.
1107          *
1108          * Yes, Virginia, this does create a security race.  It's mitigated
1109          * somewhat by the practice of creating dirs 0700 until the extract
1110          * is done, but it would be nice if we could do more than that.
1111          * People restoring critical file systems should be wary of
1112          * other programs that might try to muck with files as they're
1113          * being restored.
1114          */
1115         /* Hopefully, the compiler will optimize this mess into a constant. */
1116         critical_flags = 0;
1117 #ifdef SF_IMMUTABLE
1118         critical_flags |= SF_IMMUTABLE;
1119 #endif
1120 #ifdef UF_IMMUTABLE
1121         critical_flags |= UF_IMMUTABLE;
1122 #endif
1123 #ifdef SF_APPEND
1124         critical_flags |= SF_APPEND;
1125 #endif
1126 #ifdef UF_APPEND
1127         critical_flags |= UF_APPEND;
1128 #endif
1129 #ifdef EXT2_APPEND_FL
1130         critical_flags |= EXT2_APPEND_FL;
1131 #endif
1132 #ifdef EXT2_IMMUTABLE_FL
1133         critical_flags |= EXT2_IMMUTABLE_FL;
1134 #endif
1135
1136         if (flags & ARCHIVE_EXTRACT_FFLAGS) {
1137                 archive_entry_fflags(entry, &set, &clear);
1138
1139                 /*
1140                  * The first test encourages the compiler to eliminate
1141                  * all of this if it's not necessary.
1142                  */
1143                 if ((critical_flags != 0)  &&  (set & critical_flags)) {
1144                         le = current_fixup(a, archive_entry_pathname(entry));
1145                         le->fixup |= FIXUP_FFLAGS;
1146                         le->fflags_set = set;
1147                         /* Store the mode if it's not already there. */
1148                         if ((le->fixup & FIXUP_MODE) == 0)
1149                                 le->mode = mode;
1150                 } else {
1151                         r = set_fflags(a, fd, archive_entry_pathname(entry),
1152                             mode, set, clear);
1153                         if (r != ARCHIVE_OK)
1154                                 return (r);
1155                 }
1156         }
1157         return (ARCHIVE_OK);
1158 }
1159
1160
1161 #if ( defined(HAVE_LCHFLAGS) || defined(HAVE_CHFLAGS) || defined(HAVE_FCHFLAGS) ) && !defined(__linux)
1162 static int
1163 set_fflags(struct archive *a, int fd, const char *name, mode_t mode,
1164     unsigned long set, unsigned long clear)
1165 {
1166         struct extract *extract;
1167
1168         extract = a->extract;
1169         if (set == 0  && clear == 0)
1170                 return (ARCHIVE_OK);
1171
1172         (void)mode; /* UNUSED */
1173         /*
1174          * XXX Is the stat here really necessary?  Or can I just use
1175          * the 'set' flags directly?  In particular, I'm not sure
1176          * about the correct approach if we're overwriting an existing
1177          * file that already has flags on it. XXX
1178          */
1179         if (extract->pst != NULL) {
1180                 /* Already have stat() data available. */
1181         } else if (fd >= 0 && fstat(fd, &extract->st) == 0)
1182                 extract->pst = &extract->st;
1183         else if (stat(name, &extract->st) == 0)
1184                 extract->pst = &extract->st;
1185         else {
1186                 archive_set_error(a, errno,
1187                     "Couldn't stat file");
1188                 return (ARCHIVE_WARN);
1189         }
1190
1191         extract->st.st_flags &= ~clear;
1192         extract->st.st_flags |= set;
1193 #ifdef HAVE_FCHFLAGS
1194         /* If platform has fchflags() and we were given an fd, use it. */
1195         if (fd >= 0 && fchflags(fd, extract->st.st_flags) == 0)
1196                 return (ARCHIVE_OK);
1197 #endif
1198         /*
1199          * If we can't use the fd to set the flags, we'll use the
1200          * pathname to set flags.  We prefer lchflags() but will use
1201          * chflags() if we must.
1202          */
1203 #ifdef HAVE_LCHFLAGS
1204         if (lchflags(name, extract->st.st_flags) == 0)
1205                 return (ARCHIVE_OK);
1206 #elif defined(HAVE_CHFLAGS)
1207         if (chflags(name, extract->st.st_flags) == 0)
1208                 return (ARCHIVE_OK);
1209 #endif
1210         archive_set_error(a, errno,
1211             "Failed to set file flags");
1212         return (ARCHIVE_WARN);
1213 }
1214
1215 #elif defined(__linux) && defined(EXT2_IOC_GETFLAGS) && defined(EXT2_IOC_SETFLAGS)
1216
1217 /*
1218  * Linux has flags too, but uses ioctl() to access them instead of
1219  * having a separate chflags() system call.
1220  */
1221 static int
1222 set_fflags(struct archive *a, int fd, const char *name, mode_t mode,
1223     unsigned long set, unsigned long clear)
1224 {
1225         struct extract *extract;
1226         int              ret;
1227         int              myfd = fd;
1228         unsigned long newflags, oldflags;
1229         unsigned long sf_mask = 0;
1230
1231         extract = a->extract;
1232         if (set == 0  && clear == 0)
1233                 return (ARCHIVE_OK);
1234         /* Only regular files and dirs can have flags. */
1235         if (!S_ISREG(mode) && !S_ISDIR(mode))
1236                 return (ARCHIVE_OK);
1237
1238         /* If we weren't given an fd, open it ourselves. */
1239         if (myfd < 0)
1240                 myfd = open(name, O_RDONLY|O_NONBLOCK);
1241         if (myfd < 0)
1242                 return (ARCHIVE_OK);
1243
1244         /*
1245          * Linux has no define for the flags that are only settable by
1246          * the root user.  This code may seem a little complex, but
1247          * there seem to be some Linux systems that lack these
1248          * defines. (?)  The code below degrades reasonably gracefully
1249          * if sf_mask is incomplete.
1250          */
1251 #ifdef EXT2_IMMUTABLE_FL
1252         sf_mask |= EXT2_IMMUTABLE_FL;
1253 #endif
1254 #ifdef EXT2_APPEND_FL
1255         sf_mask |= EXT2_APPEND_FL;
1256 #endif
1257         /*
1258          * XXX As above, this would be way simpler if we didn't have
1259          * to read the current flags from disk. XXX
1260          */
1261         ret = ARCHIVE_OK;
1262         /* Try setting the flags as given. */
1263         if (ioctl(myfd, EXT2_IOC_GETFLAGS, &oldflags) >= 0) {
1264                 newflags = (oldflags & ~clear) | set;
1265                 if (ioctl(myfd, EXT2_IOC_SETFLAGS, &newflags) >= 0)
1266                         goto cleanup;
1267                 if (errno != EPERM)
1268                         goto fail;
1269         }
1270         /* If we couldn't set all the flags, try again with a subset. */
1271         if (ioctl(myfd, EXT2_IOC_GETFLAGS, &oldflags) >= 0) {
1272                 newflags &= ~sf_mask;
1273                 oldflags &= sf_mask;
1274                 newflags |= oldflags;
1275                 if (ioctl(myfd, EXT2_IOC_SETFLAGS, &newflags) >= 0)
1276                         goto cleanup;
1277         }
1278         /* We couldn't set the flags, so report the failure. */
1279 fail:
1280         archive_set_error(a, errno,
1281             "Failed to set file flags");
1282         ret = ARCHIVE_WARN;
1283 cleanup:
1284         if (fd < 0)
1285                 close(myfd);
1286         return (ret);
1287 }
1288
1289 #else /* Not HAVE_CHFLAGS && Not __linux */
1290
1291 /*
1292  * Of course, some systems have neither BSD chflags() nor Linux' flags
1293  * support through ioctl().
1294  */
1295 static int
1296 set_fflags(struct archive *a, int fd, const char *name, mode_t mode,
1297     unsigned long set, unsigned long clear)
1298 {
1299         (void)a;
1300         (void)fd;
1301         (void)name;
1302         (void)mode;
1303         (void)set;
1304         (void)clear;
1305         return (ARCHIVE_OK);
1306 }
1307
1308 #endif /* __linux */
1309
1310 #ifndef HAVE_POSIX_ACL
1311 /* Default empty function body to satisfy mainline code. */
1312 static int
1313 set_acls(struct archive *a, int fd, struct archive_entry *entry)
1314 {
1315         (void)a;
1316         (void)fd;
1317         (void)entry;
1318
1319         return (ARCHIVE_OK);
1320 }
1321
1322 #else
1323
1324 /*
1325  * XXX TODO: What about ACL types other than ACCESS and DEFAULT?
1326  */
1327 static int
1328 set_acls(struct archive *a, int fd, struct archive_entry *entry)
1329 {
1330         int              ret;
1331
1332         ret = set_acl(a, fd, entry, ACL_TYPE_ACCESS,
1333             ARCHIVE_ENTRY_ACL_TYPE_ACCESS, "access");
1334         if (ret != ARCHIVE_OK)
1335                 return (ret);
1336         ret = set_acl(a, fd, entry, ACL_TYPE_DEFAULT,
1337             ARCHIVE_ENTRY_ACL_TYPE_DEFAULT, "default");
1338         return (ret);
1339 }
1340
1341
1342 static int
1343 set_acl(struct archive *a, int fd, struct archive_entry *entry,
1344     acl_type_t acl_type, int ae_requested_type, const char *typename)
1345 {
1346         acl_t            acl;
1347         acl_entry_t      acl_entry;
1348         acl_permset_t    acl_permset;
1349         int              ret;
1350         int              ae_type, ae_permset, ae_tag, ae_id;
1351         uid_t            ae_uid;
1352         gid_t            ae_gid;
1353         const char      *ae_name;
1354         int              entries;
1355         const char      *name;
1356
1357         ret = ARCHIVE_OK;
1358         entries = archive_entry_acl_reset(entry, ae_requested_type);
1359         if (entries == 0)
1360                 return (ARCHIVE_OK);
1361         acl = acl_init(entries);
1362         while (archive_entry_acl_next(entry, ae_requested_type, &ae_type,
1363                    &ae_permset, &ae_tag, &ae_id, &ae_name) == ARCHIVE_OK) {
1364                 acl_create_entry(&acl, &acl_entry);
1365
1366                 switch (ae_tag) {
1367                 case ARCHIVE_ENTRY_ACL_USER:
1368                         acl_set_tag_type(acl_entry, ACL_USER);
1369                         ae_uid = lookup_uid(a, ae_name, ae_id);
1370                         acl_set_qualifier(acl_entry, &ae_uid);
1371                         break;
1372                 case ARCHIVE_ENTRY_ACL_GROUP:
1373                         acl_set_tag_type(acl_entry, ACL_GROUP);
1374                         ae_gid = lookup_gid(a, ae_name, ae_id);
1375                         acl_set_qualifier(acl_entry, &ae_gid);
1376                         break;
1377                 case ARCHIVE_ENTRY_ACL_USER_OBJ:
1378                         acl_set_tag_type(acl_entry, ACL_USER_OBJ);
1379                         break;
1380                 case ARCHIVE_ENTRY_ACL_GROUP_OBJ:
1381                         acl_set_tag_type(acl_entry, ACL_GROUP_OBJ);
1382                         break;
1383                 case ARCHIVE_ENTRY_ACL_MASK:
1384                         acl_set_tag_type(acl_entry, ACL_MASK);
1385                         break;
1386                 case ARCHIVE_ENTRY_ACL_OTHER:
1387                         acl_set_tag_type(acl_entry, ACL_OTHER);
1388                         break;
1389                 default:
1390                         /* XXX */
1391                         break;
1392                 }
1393
1394                 acl_get_permset(acl_entry, &acl_permset);
1395                 acl_clear_perms(acl_permset);
1396                 if (ae_permset & ARCHIVE_ENTRY_ACL_EXECUTE)
1397                         acl_add_perm(acl_permset, ACL_EXECUTE);
1398                 if (ae_permset & ARCHIVE_ENTRY_ACL_WRITE)
1399                         acl_add_perm(acl_permset, ACL_WRITE);
1400                 if (ae_permset & ARCHIVE_ENTRY_ACL_READ)
1401                         acl_add_perm(acl_permset, ACL_READ);
1402         }
1403
1404         name = archive_entry_pathname(entry);
1405
1406         /* Try restoring the ACL through 'fd' if we can. */
1407 #if HAVE_ACL_SET_FD
1408         if (fd >= 0 && acl_type == ACL_TYPE_ACCESS && acl_set_fd(fd, acl) == 0)
1409                 ret = ARCHIVE_OK;
1410         else
1411 #else
1412 #if HAVE_ACL_SET_FD_NP
1413         if (fd >= 0 && acl_set_fd_np(fd, acl, acl_type) == 0)
1414                 ret = ARCHIVE_OK;
1415         else
1416 #endif
1417 #endif
1418         if (acl_set_file(name, acl_type, acl) != 0) {
1419                 archive_set_error(a, errno, "Failed to set %s acl", typename);
1420                 ret = ARCHIVE_WARN;
1421         }
1422         acl_free(acl);
1423         return (ret);
1424 }
1425 #endif
1426
1427 #if HAVE_LSETXATTR
1428 /*
1429  * Restore extended attributes -  Linux implementation
1430  */
1431 static int
1432 set_xattrs(struct archive *a, int fd, struct archive_entry *entry)
1433 {
1434         static int warning_done = 0;
1435         int ret = ARCHIVE_OK;
1436         int i = archive_entry_xattr_reset(entry);
1437
1438         while (i--) {
1439                 const char *name;
1440                 const void *value;
1441                 size_t size;
1442                 archive_entry_xattr_next(entry, &name, &value, &size);
1443                 if (name != NULL &&
1444                                 strncmp(name, "xfsroot.", 8) != 0 &&
1445                                 strncmp(name, "system.", 7) != 0) {
1446                         int e;
1447 #if HAVE_FSETXATTR
1448                         if (fd >= 0)
1449                                 e = fsetxattr(fd, name, value, size, 0);
1450                         else
1451 #endif
1452                         {
1453                                 e = lsetxattr(archive_entry_pathname(entry),
1454                                     name, value, size, 0);
1455                         }
1456                         if (e == -1) {
1457                                 if (errno == ENOTSUP) {
1458                                         if (!warning_done) {
1459                                                 warning_done = 1;
1460                                                 archive_set_error(a, errno,
1461                                                     "Cannot restore extended "
1462                                                     "attributes on this file "
1463                                                     "system");
1464                                         }
1465                                 } else
1466                                         archive_set_error(a, errno,
1467                                             "Failed to set extended attribute");
1468                                 ret = ARCHIVE_WARN;
1469                         }
1470                 } else {
1471                         archive_set_error(a, ARCHIVE_ERRNO_FILE_FORMAT,
1472                             "Invalid extended attribute encountered");
1473                         ret = ARCHIVE_WARN;
1474                 }
1475         }
1476         return (ret);
1477 }
1478 #else
1479 /*
1480  * Restore extended attributes - stub implementation for unsupported systems
1481  */
1482 static int
1483 set_xattrs(struct archive *a, int fd, struct archive_entry *entry)
1484 {
1485         static int warning_done = 0;
1486         (void)a; /* UNUSED */
1487         (void)fd; /* UNUSED */
1488
1489         /* If there aren't any extended attributes, then it's okay not
1490          * to extract them, otherwise, issue a single warning. */
1491         if (archive_entry_xattr_count(entry) != 0 && !warning_done) {
1492                 warning_done = 1;
1493                 archive_set_error(a, ARCHIVE_ERRNO_FILE_FORMAT,
1494                     "Cannot restore extended attributes on this system");
1495                 return (ARCHIVE_WARN);
1496         }
1497         /* Warning was already emitted; suppress further warnings. */
1498         return (ARCHIVE_OK);
1499 }
1500 #endif
1501
1502 /*
1503  * The following routines do some basic caching of uname/gname
1504  * lookups.  All such lookups go through these routines, including ACL
1505  * conversions.  Even a small cache here provides an enormous speedup,
1506  * especially on systems using NIS, LDAP, or a similar networked
1507  * directory system.
1508  *
1509  * TODO: Provide an API for clients to override these routines.
1510  */
1511 static gid_t
1512 lookup_gid(struct archive *a, const char *gname, gid_t gid)
1513 {
1514         struct group    *grent;
1515         struct extract *extract;
1516         int h;
1517         struct bucket *b;
1518         int cache_size;
1519
1520         extract = a->extract;
1521         cache_size = sizeof(extract->gcache) / sizeof(extract->gcache[0]);
1522
1523         /* If no gname, just use the gid provided. */
1524         if (gname == NULL || *gname == '\0')
1525                 return (gid);
1526
1527         /* Try to find gname in the cache. */
1528         h = hash(gname);
1529         b = &extract->gcache[h % cache_size ];
1530         if (b->name != NULL && b->hash == h && strcmp(gname, b->name) == 0)
1531                 return ((gid_t)b->id);
1532
1533         /* Free the cache slot for a new entry. */
1534         if (b->name != NULL)
1535                 free(b->name);
1536         b->name = strdup(gname);
1537         /* Note: If strdup fails, that's okay; we just won't cache. */
1538         b->hash = h;
1539         grent = getgrnam(gname);
1540         if (grent != NULL)
1541                 gid = grent->gr_gid;
1542         b->id = gid;
1543
1544         return (gid);
1545 }
1546
1547 static uid_t
1548 lookup_uid(struct archive *a, const char *uname, uid_t uid)
1549 {
1550         struct passwd   *pwent;
1551         struct extract *extract;
1552         int h;
1553         struct bucket *b;
1554         int cache_size;
1555
1556         extract = a->extract;
1557         cache_size = sizeof(extract->ucache) / sizeof(extract->ucache[0]);
1558
1559         /* If no uname, just use the uid provided. */
1560         if (uname == NULL || *uname == '\0')
1561                 return (uid);
1562
1563         /* Try to find uname in the cache. */
1564         h = hash(uname);
1565         b = &extract->ucache[h % cache_size ];
1566         if (b->name != NULL && b->hash == h && strcmp(uname, b->name) == 0)
1567                 return ((uid_t)b->id);
1568
1569         /* Free the cache slot for a new entry. */
1570         if (b->name != NULL)
1571                 free(b->name);
1572         b->name = strdup(uname);
1573         /* Note: If strdup fails, that's okay; we just won't cache. */
1574         b->hash = h;
1575         pwent = getpwnam(uname);
1576         if (pwent != NULL)
1577                 uid = pwent->pw_uid;
1578         b->id = uid;
1579
1580         return (uid);
1581 }
1582
1583 static unsigned int
1584 hash(const char *p)
1585 {
1586         /* A 32-bit version of Peter Weinberger's (PJW) hash algorithm,
1587            as used by ELF for hashing function names. */
1588         unsigned g, h = 0;
1589         while (*p != '\0') {
1590                 h = ( h << 4 ) + *p++;
1591                 if (( g = h & 0xF0000000 )) {
1592                         h ^= g >> 24;
1593                         h &= 0x0FFFFFFF;
1594                 }
1595         }
1596         return h;
1597 }
1598
1599 void
1600 archive_read_extract_set_progress_callback(struct archive *a,
1601     void (*progress_func)(void *), void *user_data)
1602 {
1603         a->extract_progress = progress_func;
1604         a->extract_progress_user_data = user_data;
1605 }