158f0484fSRodney W. Grimes /*- 258f0484fSRodney W. Grimes * Copyright (c) 1990, 1993, 1994 358f0484fSRodney W. Grimes * The Regents of the University of California. All rights reserved. 458f0484fSRodney W. Grimes * 558f0484fSRodney W. Grimes * Redistribution and use in source and binary forms, with or without 658f0484fSRodney W. Grimes * modification, are permitted provided that the following conditions 758f0484fSRodney W. Grimes * are met: 858f0484fSRodney W. Grimes * 1. Redistributions of source code must retain the above copyright 958f0484fSRodney W. Grimes * notice, this list of conditions and the following disclaimer. 1058f0484fSRodney W. Grimes * 2. Redistributions in binary form must reproduce the above copyright 1158f0484fSRodney W. Grimes * notice, this list of conditions and the following disclaimer in the 1258f0484fSRodney W. Grimes * documentation and/or other materials provided with the distribution. 1358f0484fSRodney W. Grimes * 3. All advertising materials mentioning features or use of this software 1458f0484fSRodney W. Grimes * must display the following acknowledgement: 1558f0484fSRodney W. Grimes * This product includes software developed by the University of 1658f0484fSRodney W. Grimes * California, Berkeley and its contributors. 1758f0484fSRodney W. Grimes * 4. Neither the name of the University nor the names of its contributors 1858f0484fSRodney W. Grimes * may be used to endorse or promote products derived from this software 1958f0484fSRodney W. Grimes * without specific prior written permission. 2058f0484fSRodney W. Grimes * 2158f0484fSRodney W. Grimes * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 2258f0484fSRodney W. Grimes * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 2358f0484fSRodney W. Grimes * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 2458f0484fSRodney W. Grimes * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 2558f0484fSRodney W. Grimes * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 2658f0484fSRodney W. Grimes * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2758f0484fSRodney W. Grimes * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 2858f0484fSRodney W. Grimes * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 2958f0484fSRodney W. Grimes * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 3058f0484fSRodney W. Grimes * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 3158f0484fSRodney W. Grimes * SUCH DAMAGE. 3258f0484fSRodney W. Grimes */ 3358f0484fSRodney W. Grimes 3458f0484fSRodney W. Grimes #if defined(LIBC_SCCS) && !defined(lint) 35adf6ad9eSPeter Wemm static char sccsid[] = "@(#)fts.c 8.6 (Berkeley) 8/14/94"; 3658f0484fSRodney W. Grimes #endif /* LIBC_SCCS and not lint */ 3758f0484fSRodney W. Grimes 3858f0484fSRodney W. Grimes #include <sys/param.h> 3958f0484fSRodney W. Grimes #include <sys/stat.h> 4058f0484fSRodney W. Grimes 4158f0484fSRodney W. Grimes #include <dirent.h> 4258f0484fSRodney W. Grimes #include <errno.h> 4358f0484fSRodney W. Grimes #include <fcntl.h> 4458f0484fSRodney W. Grimes #include <fts.h> 4558f0484fSRodney W. Grimes #include <stdlib.h> 4658f0484fSRodney W. Grimes #include <string.h> 4758f0484fSRodney W. Grimes #include <unistd.h> 4858f0484fSRodney W. Grimes 4958f0484fSRodney W. Grimes static FTSENT *fts_alloc __P((FTS *, char *, int)); 5058f0484fSRodney W. Grimes static FTSENT *fts_build __P((FTS *, int)); 5158f0484fSRodney W. Grimes static void fts_lfree __P((FTSENT *)); 5258f0484fSRodney W. Grimes static void fts_load __P((FTS *, FTSENT *)); 5358f0484fSRodney W. Grimes static size_t fts_maxarglen __P((char * const *)); 5458f0484fSRodney W. Grimes static void fts_padjust __P((FTS *, void *)); 5558f0484fSRodney W. Grimes static int fts_palloc __P((FTS *, size_t)); 5658f0484fSRodney W. Grimes static FTSENT *fts_sort __P((FTS *, FTSENT *, int)); 5758f0484fSRodney W. Grimes static u_short fts_stat __P((FTS *, FTSENT *, int)); 5858f0484fSRodney W. Grimes 5951295a4dSJordan K. Hubbard #define ISDOT(a) (a[0] == '.' && (!a[1] || (a[1] == '.' && !a[2])) ) 6058f0484fSRodney W. Grimes 6158f0484fSRodney W. Grimes #define ISSET(opt) (sp->fts_options & opt) 6258f0484fSRodney W. Grimes #define SET(opt) (sp->fts_options |= opt) 6358f0484fSRodney W. Grimes 6458f0484fSRodney W. Grimes #define CHDIR(sp, path) (!ISSET(FTS_NOCHDIR) && chdir(path)) 6558f0484fSRodney W. Grimes #define FCHDIR(sp, fd) (!ISSET(FTS_NOCHDIR) && fchdir(fd)) 6658f0484fSRodney W. Grimes 6758f0484fSRodney W. Grimes /* fts_build flags */ 6858f0484fSRodney W. Grimes #define BCHILD 1 /* fts_children */ 6958f0484fSRodney W. Grimes #define BNAMES 2 /* fts_children, names only */ 7058f0484fSRodney W. Grimes #define BREAD 3 /* fts_read */ 7158f0484fSRodney W. Grimes 7258f0484fSRodney W. Grimes FTS * 7358f0484fSRodney W. Grimes fts_open(argv, options, compar) 7458f0484fSRodney W. Grimes char * const *argv; 7558f0484fSRodney W. Grimes register int options; 7658f0484fSRodney W. Grimes int (*compar)(); 7758f0484fSRodney W. Grimes { 7858f0484fSRodney W. Grimes register FTS *sp; 7958f0484fSRodney W. Grimes register FTSENT *p, *root; 8058f0484fSRodney W. Grimes register int nitems; 8158f0484fSRodney W. Grimes FTSENT *parent, *tmp; 8258f0484fSRodney W. Grimes int len; 8358f0484fSRodney W. Grimes 8458f0484fSRodney W. Grimes /* Options check. */ 8558f0484fSRodney W. Grimes if (options & ~FTS_OPTIONMASK) { 8658f0484fSRodney W. Grimes errno = EINVAL; 8758f0484fSRodney W. Grimes return (NULL); 8858f0484fSRodney W. Grimes } 8958f0484fSRodney W. Grimes 9058f0484fSRodney W. Grimes /* Allocate/initialize the stream */ 9158f0484fSRodney W. Grimes if ((sp = malloc((u_int)sizeof(FTS))) == NULL) 9258f0484fSRodney W. Grimes return (NULL); 9358f0484fSRodney W. Grimes memset(sp, 0, sizeof(FTS)); 9458f0484fSRodney W. Grimes sp->fts_compar = compar; 9558f0484fSRodney W. Grimes sp->fts_options = options; 9658f0484fSRodney W. Grimes 9758f0484fSRodney W. Grimes /* Logical walks turn on NOCHDIR; symbolic links are too hard. */ 9858f0484fSRodney W. Grimes if (ISSET(FTS_LOGICAL)) 9958f0484fSRodney W. Grimes SET(FTS_NOCHDIR); 10058f0484fSRodney W. Grimes 10158f0484fSRodney W. Grimes /* 10258f0484fSRodney W. Grimes * Start out with 1K of path space, and enough, in any case, 10358f0484fSRodney W. Grimes * to hold the user's paths. 10458f0484fSRodney W. Grimes */ 10558f0484fSRodney W. Grimes if (fts_palloc(sp, MAX(fts_maxarglen(argv), MAXPATHLEN))) 10658f0484fSRodney W. Grimes goto mem1; 10758f0484fSRodney W. Grimes 10858f0484fSRodney W. Grimes /* Allocate/initialize root's parent. */ 10958f0484fSRodney W. Grimes if ((parent = fts_alloc(sp, "", 0)) == NULL) 11058f0484fSRodney W. Grimes goto mem2; 11158f0484fSRodney W. Grimes parent->fts_level = FTS_ROOTPARENTLEVEL; 11258f0484fSRodney W. Grimes 11358f0484fSRodney W. Grimes /* Allocate/initialize root(s). */ 11458f0484fSRodney W. Grimes for (root = NULL, nitems = 0; *argv; ++argv, ++nitems) { 11558f0484fSRodney W. Grimes /* Don't allow zero-length paths. */ 11658f0484fSRodney W. Grimes if ((len = strlen(*argv)) == 0) { 11758f0484fSRodney W. Grimes errno = ENOENT; 11858f0484fSRodney W. Grimes goto mem3; 11958f0484fSRodney W. Grimes } 12058f0484fSRodney W. Grimes 12158f0484fSRodney W. Grimes p = fts_alloc(sp, *argv, len); 12258f0484fSRodney W. Grimes p->fts_level = FTS_ROOTLEVEL; 12358f0484fSRodney W. Grimes p->fts_parent = parent; 12458f0484fSRodney W. Grimes p->fts_accpath = p->fts_name; 12558f0484fSRodney W. Grimes p->fts_info = fts_stat(sp, p, ISSET(FTS_COMFOLLOW)); 12658f0484fSRodney W. Grimes 12758f0484fSRodney W. Grimes /* Command-line "." and ".." are real directories. */ 12858f0484fSRodney W. Grimes if (p->fts_info == FTS_DOT) 12958f0484fSRodney W. Grimes p->fts_info = FTS_D; 13058f0484fSRodney W. Grimes 13158f0484fSRodney W. Grimes /* 13258f0484fSRodney W. Grimes * If comparison routine supplied, traverse in sorted 13358f0484fSRodney W. Grimes * order; otherwise traverse in the order specified. 13458f0484fSRodney W. Grimes */ 13558f0484fSRodney W. Grimes if (compar) { 13658f0484fSRodney W. Grimes p->fts_link = root; 13758f0484fSRodney W. Grimes root = p; 13858f0484fSRodney W. Grimes } else { 13958f0484fSRodney W. Grimes p->fts_link = NULL; 14058f0484fSRodney W. Grimes if (root == NULL) 14158f0484fSRodney W. Grimes tmp = root = p; 14258f0484fSRodney W. Grimes else { 14358f0484fSRodney W. Grimes tmp->fts_link = p; 14458f0484fSRodney W. Grimes tmp = p; 14558f0484fSRodney W. Grimes } 14658f0484fSRodney W. Grimes } 14758f0484fSRodney W. Grimes } 14858f0484fSRodney W. Grimes if (compar && nitems > 1) 14958f0484fSRodney W. Grimes root = fts_sort(sp, root, nitems); 15058f0484fSRodney W. Grimes 15158f0484fSRodney W. Grimes /* 15258f0484fSRodney W. Grimes * Allocate a dummy pointer and make fts_read think that we've just 15358f0484fSRodney W. Grimes * finished the node before the root(s); set p->fts_info to FTS_INIT 15458f0484fSRodney W. Grimes * so that everything about the "current" node is ignored. 15558f0484fSRodney W. Grimes */ 15658f0484fSRodney W. Grimes if ((sp->fts_cur = fts_alloc(sp, "", 0)) == NULL) 15758f0484fSRodney W. Grimes goto mem3; 15858f0484fSRodney W. Grimes sp->fts_cur->fts_link = root; 15958f0484fSRodney W. Grimes sp->fts_cur->fts_info = FTS_INIT; 16058f0484fSRodney W. Grimes 16158f0484fSRodney W. Grimes /* 16258f0484fSRodney W. Grimes * If using chdir(2), grab a file descriptor pointing to dot to insure 16358f0484fSRodney W. Grimes * that we can get back here; this could be avoided for some paths, 16458f0484fSRodney W. Grimes * but almost certainly not worth the effort. Slashes, symbolic links, 16558f0484fSRodney W. Grimes * and ".." are all fairly nasty problems. Note, if we can't get the 16658f0484fSRodney W. Grimes * descriptor we run anyway, just more slowly. 16758f0484fSRodney W. Grimes */ 16858f0484fSRodney W. Grimes if (!ISSET(FTS_NOCHDIR) && (sp->fts_rfd = open(".", O_RDONLY, 0)) < 0) 16958f0484fSRodney W. Grimes SET(FTS_NOCHDIR); 17058f0484fSRodney W. Grimes 17158f0484fSRodney W. Grimes return (sp); 17258f0484fSRodney W. Grimes 17358f0484fSRodney W. Grimes mem3: fts_lfree(root); 17458f0484fSRodney W. Grimes free(parent); 17558f0484fSRodney W. Grimes mem2: free(sp->fts_path); 17658f0484fSRodney W. Grimes mem1: free(sp); 17758f0484fSRodney W. Grimes return (NULL); 17858f0484fSRodney W. Grimes } 17958f0484fSRodney W. Grimes 18058f0484fSRodney W. Grimes static void 18158f0484fSRodney W. Grimes fts_load(sp, p) 18258f0484fSRodney W. Grimes FTS *sp; 18358f0484fSRodney W. Grimes register FTSENT *p; 18458f0484fSRodney W. Grimes { 18558f0484fSRodney W. Grimes register int len; 18658f0484fSRodney W. Grimes register char *cp; 18758f0484fSRodney W. Grimes 18858f0484fSRodney W. Grimes /* 18958f0484fSRodney W. Grimes * Load the stream structure for the next traversal. Since we don't 19058f0484fSRodney W. Grimes * actually enter the directory until after the preorder visit, set 19158f0484fSRodney W. Grimes * the fts_accpath field specially so the chdir gets done to the right 19258f0484fSRodney W. Grimes * place and the user can access the first node. From fts_open it's 19358f0484fSRodney W. Grimes * known that the path will fit. 19458f0484fSRodney W. Grimes */ 19558f0484fSRodney W. Grimes len = p->fts_pathlen = p->fts_namelen; 19658f0484fSRodney W. Grimes memmove(sp->fts_path, p->fts_name, len + 1); 19758f0484fSRodney W. Grimes if ((cp = strrchr(p->fts_name, '/')) && (cp != p->fts_name || cp[1])) { 19858f0484fSRodney W. Grimes len = strlen(++cp); 19958f0484fSRodney W. Grimes memmove(p->fts_name, cp, len + 1); 20058f0484fSRodney W. Grimes p->fts_namelen = len; 20158f0484fSRodney W. Grimes } 20258f0484fSRodney W. Grimes p->fts_accpath = p->fts_path = sp->fts_path; 20358f0484fSRodney W. Grimes sp->fts_dev = p->fts_dev; 20458f0484fSRodney W. Grimes } 20558f0484fSRodney W. Grimes 20658f0484fSRodney W. Grimes int 20758f0484fSRodney W. Grimes fts_close(sp) 20858f0484fSRodney W. Grimes FTS *sp; 20958f0484fSRodney W. Grimes { 21058f0484fSRodney W. Grimes register FTSENT *freep, *p; 21158f0484fSRodney W. Grimes int saved_errno; 21258f0484fSRodney W. Grimes 21358f0484fSRodney W. Grimes /* 21458f0484fSRodney W. Grimes * This still works if we haven't read anything -- the dummy structure 21558f0484fSRodney W. Grimes * points to the root list, so we step through to the end of the root 21658f0484fSRodney W. Grimes * list which has a valid parent pointer. 21758f0484fSRodney W. Grimes */ 21858f0484fSRodney W. Grimes if (sp->fts_cur) { 21958f0484fSRodney W. Grimes for (p = sp->fts_cur; p->fts_level >= FTS_ROOTLEVEL;) { 22058f0484fSRodney W. Grimes freep = p; 22158f0484fSRodney W. Grimes p = p->fts_link ? p->fts_link : p->fts_parent; 22258f0484fSRodney W. Grimes free(freep); 22358f0484fSRodney W. Grimes } 22458f0484fSRodney W. Grimes free(p); 22558f0484fSRodney W. Grimes } 22658f0484fSRodney W. Grimes 22758f0484fSRodney W. Grimes /* Free up child linked list, sort array, path buffer. */ 22858f0484fSRodney W. Grimes if (sp->fts_child) 22958f0484fSRodney W. Grimes fts_lfree(sp->fts_child); 23058f0484fSRodney W. Grimes if (sp->fts_array) 23158f0484fSRodney W. Grimes free(sp->fts_array); 23258f0484fSRodney W. Grimes free(sp->fts_path); 23358f0484fSRodney W. Grimes 23458f0484fSRodney W. Grimes /* Return to original directory, save errno if necessary. */ 23558f0484fSRodney W. Grimes if (!ISSET(FTS_NOCHDIR)) { 23658f0484fSRodney W. Grimes saved_errno = fchdir(sp->fts_rfd) ? errno : 0; 23758f0484fSRodney W. Grimes (void)close(sp->fts_rfd); 23858f0484fSRodney W. Grimes } 23958f0484fSRodney W. Grimes 24058f0484fSRodney W. Grimes /* Free up the stream pointer. */ 24158f0484fSRodney W. Grimes free(sp); 24258f0484fSRodney W. Grimes 24358f0484fSRodney W. Grimes /* Set errno and return. */ 24458f0484fSRodney W. Grimes if (!ISSET(FTS_NOCHDIR) && saved_errno) { 24558f0484fSRodney W. Grimes errno = saved_errno; 24658f0484fSRodney W. Grimes return (-1); 24758f0484fSRodney W. Grimes } 24858f0484fSRodney W. Grimes return (0); 24958f0484fSRodney W. Grimes } 25058f0484fSRodney W. Grimes 25158f0484fSRodney W. Grimes /* 25258f0484fSRodney W. Grimes * Special case a root of "/" so that slashes aren't appended which would 25358f0484fSRodney W. Grimes * cause paths to be written as "//foo". 25458f0484fSRodney W. Grimes */ 25558f0484fSRodney W. Grimes #define NAPPEND(p) \ 25658f0484fSRodney W. Grimes (p->fts_level == FTS_ROOTLEVEL && p->fts_pathlen == 1 && \ 25758f0484fSRodney W. Grimes p->fts_path[0] == '/' ? 0 : p->fts_pathlen) 25858f0484fSRodney W. Grimes 25958f0484fSRodney W. Grimes FTSENT * 26058f0484fSRodney W. Grimes fts_read(sp) 26158f0484fSRodney W. Grimes register FTS *sp; 26258f0484fSRodney W. Grimes { 26358f0484fSRodney W. Grimes register FTSENT *p, *tmp; 26458f0484fSRodney W. Grimes register int instr; 26558f0484fSRodney W. Grimes register char *t; 26658f0484fSRodney W. Grimes int saved_errno; 26758f0484fSRodney W. Grimes 26858f0484fSRodney W. Grimes /* If finished or unrecoverable error, return NULL. */ 26958f0484fSRodney W. Grimes if (sp->fts_cur == NULL || ISSET(FTS_STOP)) 27058f0484fSRodney W. Grimes return (NULL); 27158f0484fSRodney W. Grimes 27258f0484fSRodney W. Grimes /* Set current node pointer. */ 27358f0484fSRodney W. Grimes p = sp->fts_cur; 27458f0484fSRodney W. Grimes 27558f0484fSRodney W. Grimes /* Save and zero out user instructions. */ 27658f0484fSRodney W. Grimes instr = p->fts_instr; 27758f0484fSRodney W. Grimes p->fts_instr = FTS_NOINSTR; 27858f0484fSRodney W. Grimes 27958f0484fSRodney W. Grimes /* Any type of file may be re-visited; re-stat and re-turn. */ 28058f0484fSRodney W. Grimes if (instr == FTS_AGAIN) { 28158f0484fSRodney W. Grimes p->fts_info = fts_stat(sp, p, 0); 28258f0484fSRodney W. Grimes return (p); 28358f0484fSRodney W. Grimes } 28458f0484fSRodney W. Grimes 28558f0484fSRodney W. Grimes /* 28658f0484fSRodney W. Grimes * Following a symlink -- SLNONE test allows application to see 28758f0484fSRodney W. Grimes * SLNONE and recover. If indirecting through a symlink, have 28858f0484fSRodney W. Grimes * keep a pointer to current location. If unable to get that 28958f0484fSRodney W. Grimes * pointer, follow fails. 29058f0484fSRodney W. Grimes */ 29158f0484fSRodney W. Grimes if (instr == FTS_FOLLOW && 29258f0484fSRodney W. Grimes (p->fts_info == FTS_SL || p->fts_info == FTS_SLNONE)) { 29358f0484fSRodney W. Grimes p->fts_info = fts_stat(sp, p, 1); 29458f0484fSRodney W. Grimes if (p->fts_info == FTS_D && !ISSET(FTS_NOCHDIR)) 29558f0484fSRodney W. Grimes if ((p->fts_symfd = open(".", O_RDONLY, 0)) < 0) { 29658f0484fSRodney W. Grimes p->fts_errno = errno; 29758f0484fSRodney W. Grimes p->fts_info = FTS_ERR; 29858f0484fSRodney W. Grimes } else 29958f0484fSRodney W. Grimes p->fts_flags |= FTS_SYMFOLLOW; 30058f0484fSRodney W. Grimes return (p); 30158f0484fSRodney W. Grimes } 30258f0484fSRodney W. Grimes 30358f0484fSRodney W. Grimes /* Directory in pre-order. */ 30458f0484fSRodney W. Grimes if (p->fts_info == FTS_D) { 30558f0484fSRodney W. Grimes /* If skipped or crossed mount point, do post-order visit. */ 30658f0484fSRodney W. Grimes if (instr == FTS_SKIP || 30751295a4dSJordan K. Hubbard (ISSET(FTS_XDEV) && p->fts_dev != sp->fts_dev) ) { 30858f0484fSRodney W. Grimes if (p->fts_flags & FTS_SYMFOLLOW) 30958f0484fSRodney W. Grimes (void)close(p->fts_symfd); 31058f0484fSRodney W. Grimes if (sp->fts_child) { 31158f0484fSRodney W. Grimes fts_lfree(sp->fts_child); 31258f0484fSRodney W. Grimes sp->fts_child = NULL; 31358f0484fSRodney W. Grimes } 31458f0484fSRodney W. Grimes p->fts_info = FTS_DP; 31558f0484fSRodney W. Grimes return (p); 31658f0484fSRodney W. Grimes } 31758f0484fSRodney W. Grimes 31858f0484fSRodney W. Grimes /* Rebuild if only read the names and now traversing. */ 31958f0484fSRodney W. Grimes if (sp->fts_child && sp->fts_options & FTS_NAMEONLY) { 32058f0484fSRodney W. Grimes sp->fts_options &= ~FTS_NAMEONLY; 32158f0484fSRodney W. Grimes fts_lfree(sp->fts_child); 32258f0484fSRodney W. Grimes sp->fts_child = NULL; 32358f0484fSRodney W. Grimes } 32458f0484fSRodney W. Grimes 32558f0484fSRodney W. Grimes /* 32658f0484fSRodney W. Grimes * Cd to the subdirectory. 32758f0484fSRodney W. Grimes * 32858f0484fSRodney W. Grimes * If have already read and now fail to chdir, whack the list 32958f0484fSRodney W. Grimes * to make the names come out right, and set the parent errno 33058f0484fSRodney W. Grimes * so the application will eventually get an error condition. 33158f0484fSRodney W. Grimes * Set the FTS_DONTCHDIR flag so that when we logically change 33258f0484fSRodney W. Grimes * directories back to the parent we don't do a chdir. 33358f0484fSRodney W. Grimes * 33458f0484fSRodney W. Grimes * If haven't read do so. If the read fails, fts_build sets 33558f0484fSRodney W. Grimes * FTS_STOP or the fts_info field of the node. 33658f0484fSRodney W. Grimes */ 33758f0484fSRodney W. Grimes if (sp->fts_child) { 33858f0484fSRodney W. Grimes if (CHDIR(sp, p->fts_accpath)) { 33958f0484fSRodney W. Grimes p->fts_errno = errno; 34058f0484fSRodney W. Grimes p->fts_flags |= FTS_DONTCHDIR; 34158f0484fSRodney W. Grimes for (p = sp->fts_child; p; p = p->fts_link) 34258f0484fSRodney W. Grimes p->fts_accpath = 34358f0484fSRodney W. Grimes p->fts_parent->fts_accpath; 34458f0484fSRodney W. Grimes } 34558f0484fSRodney W. Grimes } else if ((sp->fts_child = fts_build(sp, BREAD)) == NULL) { 34658f0484fSRodney W. Grimes if (ISSET(FTS_STOP)) 34758f0484fSRodney W. Grimes return (NULL); 34858f0484fSRodney W. Grimes return (p); 34958f0484fSRodney W. Grimes } 35058f0484fSRodney W. Grimes p = sp->fts_child; 35158f0484fSRodney W. Grimes sp->fts_child = NULL; 35258f0484fSRodney W. Grimes goto name; 35358f0484fSRodney W. Grimes } 35458f0484fSRodney W. Grimes 35558f0484fSRodney W. Grimes /* Move to the next node on this level. */ 35658f0484fSRodney W. Grimes next: tmp = p; 35751295a4dSJordan K. Hubbard if ( (p = p->fts_link) ) { 35858f0484fSRodney W. Grimes free(tmp); 35958f0484fSRodney W. Grimes 36058f0484fSRodney W. Grimes /* 36158f0484fSRodney W. Grimes * If reached the top, return to the original directory, and 36258f0484fSRodney W. Grimes * load the paths for the next root. 36358f0484fSRodney W. Grimes */ 36458f0484fSRodney W. Grimes if (p->fts_level == FTS_ROOTLEVEL) { 36558f0484fSRodney W. Grimes if (!ISSET(FTS_NOCHDIR) && FCHDIR(sp, sp->fts_rfd)) { 36658f0484fSRodney W. Grimes SET(FTS_STOP); 36758f0484fSRodney W. Grimes return (NULL); 36858f0484fSRodney W. Grimes } 36958f0484fSRodney W. Grimes fts_load(sp, p); 37058f0484fSRodney W. Grimes return (sp->fts_cur = p); 37158f0484fSRodney W. Grimes } 37258f0484fSRodney W. Grimes 37358f0484fSRodney W. Grimes /* 37458f0484fSRodney W. Grimes * User may have called fts_set on the node. If skipped, 37558f0484fSRodney W. Grimes * ignore. If followed, get a file descriptor so we can 37658f0484fSRodney W. Grimes * get back if necessary. 37758f0484fSRodney W. Grimes */ 37858f0484fSRodney W. Grimes if (p->fts_instr == FTS_SKIP) 37958f0484fSRodney W. Grimes goto next; 38058f0484fSRodney W. Grimes if (p->fts_instr == FTS_FOLLOW) { 38158f0484fSRodney W. Grimes p->fts_info = fts_stat(sp, p, 1); 38258f0484fSRodney W. Grimes if (p->fts_info == FTS_D && !ISSET(FTS_NOCHDIR)) 38358f0484fSRodney W. Grimes if ((p->fts_symfd = 38458f0484fSRodney W. Grimes open(".", O_RDONLY, 0)) < 0) { 38558f0484fSRodney W. Grimes p->fts_errno = errno; 38658f0484fSRodney W. Grimes p->fts_info = FTS_ERR; 38758f0484fSRodney W. Grimes } else 38858f0484fSRodney W. Grimes p->fts_flags |= FTS_SYMFOLLOW; 38958f0484fSRodney W. Grimes p->fts_instr = FTS_NOINSTR; 39058f0484fSRodney W. Grimes } 39158f0484fSRodney W. Grimes 39258f0484fSRodney W. Grimes name: t = sp->fts_path + NAPPEND(p->fts_parent); 39358f0484fSRodney W. Grimes *t++ = '/'; 39458f0484fSRodney W. Grimes memmove(t, p->fts_name, p->fts_namelen + 1); 39558f0484fSRodney W. Grimes return (sp->fts_cur = p); 39658f0484fSRodney W. Grimes } 39758f0484fSRodney W. Grimes 39858f0484fSRodney W. Grimes /* Move up to the parent node. */ 39958f0484fSRodney W. Grimes p = tmp->fts_parent; 40058f0484fSRodney W. Grimes free(tmp); 40158f0484fSRodney W. Grimes 40258f0484fSRodney W. Grimes if (p->fts_level == FTS_ROOTPARENTLEVEL) { 40358f0484fSRodney W. Grimes /* 40458f0484fSRodney W. Grimes * Done; free everything up and set errno to 0 so the user 40558f0484fSRodney W. Grimes * can distinguish between error and EOF. 40658f0484fSRodney W. Grimes */ 40758f0484fSRodney W. Grimes free(p); 40858f0484fSRodney W. Grimes errno = 0; 40958f0484fSRodney W. Grimes return (sp->fts_cur = NULL); 41058f0484fSRodney W. Grimes } 41158f0484fSRodney W. Grimes 41258f0484fSRodney W. Grimes /* Nul terminate the pathname. */ 41358f0484fSRodney W. Grimes sp->fts_path[p->fts_pathlen] = '\0'; 41458f0484fSRodney W. Grimes 41558f0484fSRodney W. Grimes /* 41658f0484fSRodney W. Grimes * Return to the parent directory. If at a root node or came through 41758f0484fSRodney W. Grimes * a symlink, go back through the file descriptor. Otherwise, cd up 41858f0484fSRodney W. Grimes * one directory. 41958f0484fSRodney W. Grimes */ 42058f0484fSRodney W. Grimes if (p->fts_level == FTS_ROOTLEVEL) { 42158f0484fSRodney W. Grimes if (!ISSET(FTS_NOCHDIR) && FCHDIR(sp, sp->fts_rfd)) { 42258f0484fSRodney W. Grimes SET(FTS_STOP); 42358f0484fSRodney W. Grimes return (NULL); 42458f0484fSRodney W. Grimes } 42558f0484fSRodney W. Grimes } else if (p->fts_flags & FTS_SYMFOLLOW) { 42658f0484fSRodney W. Grimes if (FCHDIR(sp, p->fts_symfd)) { 42758f0484fSRodney W. Grimes saved_errno = errno; 42858f0484fSRodney W. Grimes (void)close(p->fts_symfd); 42958f0484fSRodney W. Grimes errno = saved_errno; 43058f0484fSRodney W. Grimes SET(FTS_STOP); 43158f0484fSRodney W. Grimes return (NULL); 43258f0484fSRodney W. Grimes } 43358f0484fSRodney W. Grimes (void)close(p->fts_symfd); 43458f0484fSRodney W. Grimes } else if (!(p->fts_flags & FTS_DONTCHDIR)) { 43558f0484fSRodney W. Grimes if (CHDIR(sp, "..")) { 43658f0484fSRodney W. Grimes SET(FTS_STOP); 43758f0484fSRodney W. Grimes return (NULL); 43858f0484fSRodney W. Grimes } 43958f0484fSRodney W. Grimes } 44058f0484fSRodney W. Grimes p->fts_info = p->fts_errno ? FTS_ERR : FTS_DP; 44158f0484fSRodney W. Grimes return (sp->fts_cur = p); 44258f0484fSRodney W. Grimes } 44358f0484fSRodney W. Grimes 44458f0484fSRodney W. Grimes /* 44558f0484fSRodney W. Grimes * Fts_set takes the stream as an argument although it's not used in this 44658f0484fSRodney W. Grimes * implementation; it would be necessary if anyone wanted to add global 44758f0484fSRodney W. Grimes * semantics to fts using fts_set. An error return is allowed for similar 44858f0484fSRodney W. Grimes * reasons. 44958f0484fSRodney W. Grimes */ 45058f0484fSRodney W. Grimes /* ARGSUSED */ 45158f0484fSRodney W. Grimes int 45258f0484fSRodney W. Grimes fts_set(sp, p, instr) 45358f0484fSRodney W. Grimes FTS *sp; 45458f0484fSRodney W. Grimes FTSENT *p; 45558f0484fSRodney W. Grimes int instr; 45658f0484fSRodney W. Grimes { 45758f0484fSRodney W. Grimes if (instr && instr != FTS_AGAIN && instr != FTS_FOLLOW && 45858f0484fSRodney W. Grimes instr != FTS_NOINSTR && instr != FTS_SKIP) { 45958f0484fSRodney W. Grimes errno = EINVAL; 46058f0484fSRodney W. Grimes return (1); 46158f0484fSRodney W. Grimes } 46258f0484fSRodney W. Grimes p->fts_instr = instr; 46358f0484fSRodney W. Grimes return (0); 46458f0484fSRodney W. Grimes } 46558f0484fSRodney W. Grimes 46658f0484fSRodney W. Grimes FTSENT * 46758f0484fSRodney W. Grimes fts_children(sp, instr) 46858f0484fSRodney W. Grimes register FTS *sp; 46958f0484fSRodney W. Grimes int instr; 47058f0484fSRodney W. Grimes { 47158f0484fSRodney W. Grimes register FTSENT *p; 47258f0484fSRodney W. Grimes int fd; 47358f0484fSRodney W. Grimes 47458f0484fSRodney W. Grimes if (instr && instr != FTS_NAMEONLY) { 47558f0484fSRodney W. Grimes errno = EINVAL; 47658f0484fSRodney W. Grimes return (NULL); 47758f0484fSRodney W. Grimes } 47858f0484fSRodney W. Grimes 47958f0484fSRodney W. Grimes /* Set current node pointer. */ 48058f0484fSRodney W. Grimes p = sp->fts_cur; 48158f0484fSRodney W. Grimes 48258f0484fSRodney W. Grimes /* 48358f0484fSRodney W. Grimes * Errno set to 0 so user can distinguish empty directory from 48458f0484fSRodney W. Grimes * an error. 48558f0484fSRodney W. Grimes */ 48658f0484fSRodney W. Grimes errno = 0; 48758f0484fSRodney W. Grimes 48858f0484fSRodney W. Grimes /* Fatal errors stop here. */ 48958f0484fSRodney W. Grimes if (ISSET(FTS_STOP)) 49058f0484fSRodney W. Grimes return (NULL); 49158f0484fSRodney W. Grimes 49258f0484fSRodney W. Grimes /* Return logical hierarchy of user's arguments. */ 49358f0484fSRodney W. Grimes if (p->fts_info == FTS_INIT) 49458f0484fSRodney W. Grimes return (p->fts_link); 49558f0484fSRodney W. Grimes 49658f0484fSRodney W. Grimes /* 49758f0484fSRodney W. Grimes * If not a directory being visited in pre-order, stop here. Could 49858f0484fSRodney W. Grimes * allow FTS_DNR, assuming the user has fixed the problem, but the 49958f0484fSRodney W. Grimes * same effect is available with FTS_AGAIN. 50058f0484fSRodney W. Grimes */ 50158f0484fSRodney W. Grimes if (p->fts_info != FTS_D /* && p->fts_info != FTS_DNR */) 50258f0484fSRodney W. Grimes return (NULL); 50358f0484fSRodney W. Grimes 50458f0484fSRodney W. Grimes /* Free up any previous child list. */ 50558f0484fSRodney W. Grimes if (sp->fts_child) 50658f0484fSRodney W. Grimes fts_lfree(sp->fts_child); 50758f0484fSRodney W. Grimes 50858f0484fSRodney W. Grimes if (instr == FTS_NAMEONLY) { 50958f0484fSRodney W. Grimes sp->fts_options |= FTS_NAMEONLY; 51058f0484fSRodney W. Grimes instr = BNAMES; 51158f0484fSRodney W. Grimes } else 51258f0484fSRodney W. Grimes instr = BCHILD; 51358f0484fSRodney W. Grimes 51458f0484fSRodney W. Grimes /* 51558f0484fSRodney W. Grimes * If using chdir on a relative path and called BEFORE fts_read does 51658f0484fSRodney W. Grimes * its chdir to the root of a traversal, we can lose -- we need to 51758f0484fSRodney W. Grimes * chdir into the subdirectory, and we don't know where the current 51858f0484fSRodney W. Grimes * directory is, so we can't get back so that the upcoming chdir by 51958f0484fSRodney W. Grimes * fts_read will work. 52058f0484fSRodney W. Grimes */ 52158f0484fSRodney W. Grimes if (p->fts_level != FTS_ROOTLEVEL || p->fts_accpath[0] == '/' || 52258f0484fSRodney W. Grimes ISSET(FTS_NOCHDIR)) 52358f0484fSRodney W. Grimes return (sp->fts_child = fts_build(sp, instr)); 52458f0484fSRodney W. Grimes 52558f0484fSRodney W. Grimes if ((fd = open(".", O_RDONLY, 0)) < 0) 52658f0484fSRodney W. Grimes return (NULL); 52758f0484fSRodney W. Grimes sp->fts_child = fts_build(sp, instr); 52858f0484fSRodney W. Grimes if (fchdir(fd)) 52958f0484fSRodney W. Grimes return (NULL); 53058f0484fSRodney W. Grimes (void)close(fd); 53158f0484fSRodney W. Grimes return (sp->fts_child); 53258f0484fSRodney W. Grimes } 53358f0484fSRodney W. Grimes 53458f0484fSRodney W. Grimes /* 53558f0484fSRodney W. Grimes * This is the tricky part -- do not casually change *anything* in here. The 53658f0484fSRodney W. Grimes * idea is to build the linked list of entries that are used by fts_children 53758f0484fSRodney W. Grimes * and fts_read. There are lots of special cases. 53858f0484fSRodney W. Grimes * 53958f0484fSRodney W. Grimes * The real slowdown in walking the tree is the stat calls. If FTS_NOSTAT is 54058f0484fSRodney W. Grimes * set and it's a physical walk (so that symbolic links can't be directories), 54158f0484fSRodney W. Grimes * we can do things quickly. First, if it's a 4.4BSD file system, the type 54258f0484fSRodney W. Grimes * of the file is in the directory entry. Otherwise, we assume that the number 54358f0484fSRodney W. Grimes * of subdirectories in a node is equal to the number of links to the parent. 54458f0484fSRodney W. Grimes * The former skips all stat calls. The latter skips stat calls in any leaf 54558f0484fSRodney W. Grimes * directories and for any files after the subdirectories in the directory have 54658f0484fSRodney W. Grimes * been found, cutting the stat calls by about 2/3. 54758f0484fSRodney W. Grimes */ 54858f0484fSRodney W. Grimes static FTSENT * 54958f0484fSRodney W. Grimes fts_build(sp, type) 55058f0484fSRodney W. Grimes register FTS *sp; 55158f0484fSRodney W. Grimes int type; 55258f0484fSRodney W. Grimes { 55358f0484fSRodney W. Grimes register struct dirent *dp; 55458f0484fSRodney W. Grimes register FTSENT *p, *head; 55558f0484fSRodney W. Grimes register int nitems; 55658f0484fSRodney W. Grimes FTSENT *cur, *tail; 55758f0484fSRodney W. Grimes DIR *dirp; 55858f0484fSRodney W. Grimes void *adjaddr; 559adf6ad9eSPeter Wemm int cderrno, descend, len, level, maxlen, nlinks, oflag, saved_errno; 56058f0484fSRodney W. Grimes char *cp; 56158f0484fSRodney W. Grimes 56258f0484fSRodney W. Grimes /* Set current node pointer. */ 56358f0484fSRodney W. Grimes cur = sp->fts_cur; 56458f0484fSRodney W. Grimes 56558f0484fSRodney W. Grimes /* 56658f0484fSRodney W. Grimes * Open the directory for reading. If this fails, we're done. 56758f0484fSRodney W. Grimes * If being called from fts_read, set the fts_info field. 56858f0484fSRodney W. Grimes */ 569adf6ad9eSPeter Wemm #ifdef FTS_WHITEOUT 570adf6ad9eSPeter Wemm if (ISSET(FTS_WHITEOUT)) 571adf6ad9eSPeter Wemm oflag = DTF_NODUP|DTF_REWIND; 572adf6ad9eSPeter Wemm else 573adf6ad9eSPeter Wemm oflag = DTF_HIDEW|DTF_NODUP|DTF_REWIND; 574adf6ad9eSPeter Wemm #else 575adf6ad9eSPeter Wemm #define __opendir2(path, flag) opendir(path) 576adf6ad9eSPeter Wemm #endif 577adf6ad9eSPeter Wemm if ((dirp = __opendir2(cur->fts_accpath, oflag)) == NULL) { 57858f0484fSRodney W. Grimes if (type == BREAD) { 57958f0484fSRodney W. Grimes cur->fts_info = FTS_DNR; 58058f0484fSRodney W. Grimes cur->fts_errno = errno; 58158f0484fSRodney W. Grimes } 58258f0484fSRodney W. Grimes return (NULL); 58358f0484fSRodney W. Grimes } 58458f0484fSRodney W. Grimes 58558f0484fSRodney W. Grimes /* 58658f0484fSRodney W. Grimes * Nlinks is the number of possible entries of type directory in the 58758f0484fSRodney W. Grimes * directory if we're cheating on stat calls, 0 if we're not doing 58858f0484fSRodney W. Grimes * any stat calls at all, -1 if we're doing stats on everything. 58958f0484fSRodney W. Grimes */ 59058f0484fSRodney W. Grimes if (type == BNAMES) 59158f0484fSRodney W. Grimes nlinks = 0; 59258f0484fSRodney W. Grimes else if (ISSET(FTS_NOSTAT) && ISSET(FTS_PHYSICAL)) 59358f0484fSRodney W. Grimes nlinks = cur->fts_nlink - (ISSET(FTS_SEEDOT) ? 0 : 2); 59458f0484fSRodney W. Grimes else 59558f0484fSRodney W. Grimes nlinks = -1; 59658f0484fSRodney W. Grimes 59758f0484fSRodney W. Grimes #ifdef notdef 59858f0484fSRodney W. Grimes (void)printf("nlinks == %d (cur: %d)\n", nlinks, cur->fts_nlink); 59958f0484fSRodney W. Grimes (void)printf("NOSTAT %d PHYSICAL %d SEEDOT %d\n", 60058f0484fSRodney W. Grimes ISSET(FTS_NOSTAT), ISSET(FTS_PHYSICAL), ISSET(FTS_SEEDOT)); 60158f0484fSRodney W. Grimes #endif 60258f0484fSRodney W. Grimes /* 60358f0484fSRodney W. Grimes * If we're going to need to stat anything or we want to descend 60458f0484fSRodney W. Grimes * and stay in the directory, chdir. If this fails we keep going, 60558f0484fSRodney W. Grimes * but set a flag so we don't chdir after the post-order visit. 60658f0484fSRodney W. Grimes * We won't be able to stat anything, but we can still return the 60758f0484fSRodney W. Grimes * names themselves. Note, that since fts_read won't be able to 60858f0484fSRodney W. Grimes * chdir into the directory, it will have to return different path 60958f0484fSRodney W. Grimes * names than before, i.e. "a/b" instead of "b". Since the node 61058f0484fSRodney W. Grimes * has already been visited in pre-order, have to wait until the 61158f0484fSRodney W. Grimes * post-order visit to return the error. There is a special case 61258f0484fSRodney W. Grimes * here, if there was nothing to stat then it's not an error to 61358f0484fSRodney W. Grimes * not be able to stat. This is all fairly nasty. If a program 61458f0484fSRodney W. Grimes * needed sorted entries or stat information, they had better be 61558f0484fSRodney W. Grimes * checking FTS_NS on the returned nodes. 61658f0484fSRodney W. Grimes */ 61758f0484fSRodney W. Grimes cderrno = 0; 61858f0484fSRodney W. Grimes if (nlinks || type == BREAD) 61958f0484fSRodney W. Grimes if (FCHDIR(sp, dirfd(dirp))) { 62058f0484fSRodney W. Grimes if (nlinks && type == BREAD) 62158f0484fSRodney W. Grimes cur->fts_errno = errno; 62258f0484fSRodney W. Grimes cur->fts_flags |= FTS_DONTCHDIR; 62358f0484fSRodney W. Grimes descend = 0; 62458f0484fSRodney W. Grimes cderrno = errno; 62558f0484fSRodney W. Grimes } else 62658f0484fSRodney W. Grimes descend = 1; 62758f0484fSRodney W. Grimes else 62858f0484fSRodney W. Grimes descend = 0; 62958f0484fSRodney W. Grimes 63058f0484fSRodney W. Grimes /* 63158f0484fSRodney W. Grimes * Figure out the max file name length that can be stored in the 63258f0484fSRodney W. Grimes * current path -- the inner loop allocates more path as necessary. 63358f0484fSRodney W. Grimes * We really wouldn't have to do the maxlen calculations here, we 63458f0484fSRodney W. Grimes * could do them in fts_read before returning the path, but it's a 63558f0484fSRodney W. Grimes * lot easier here since the length is part of the dirent structure. 63658f0484fSRodney W. Grimes * 63758f0484fSRodney W. Grimes * If not changing directories set a pointer so that can just append 63858f0484fSRodney W. Grimes * each new name into the path. 63958f0484fSRodney W. Grimes */ 64058f0484fSRodney W. Grimes maxlen = sp->fts_pathlen - cur->fts_pathlen - 1; 64158f0484fSRodney W. Grimes len = NAPPEND(cur); 64258f0484fSRodney W. Grimes if (ISSET(FTS_NOCHDIR)) { 64358f0484fSRodney W. Grimes cp = sp->fts_path + len; 64458f0484fSRodney W. Grimes *cp++ = '/'; 64558f0484fSRodney W. Grimes } 64658f0484fSRodney W. Grimes 64758f0484fSRodney W. Grimes level = cur->fts_level + 1; 64858f0484fSRodney W. Grimes 64958f0484fSRodney W. Grimes /* Read the directory, attaching each entry to the `link' pointer. */ 65058f0484fSRodney W. Grimes adjaddr = NULL; 65151295a4dSJordan K. Hubbard for (head = tail = NULL, nitems = 0; (dp = readdir(dirp)); ) { 65258f0484fSRodney W. Grimes if (!ISSET(FTS_SEEDOT) && ISDOT(dp->d_name)) 65358f0484fSRodney W. Grimes continue; 65458f0484fSRodney W. Grimes 65558f0484fSRodney W. Grimes if ((p = fts_alloc(sp, dp->d_name, (int)dp->d_namlen)) == NULL) 65658f0484fSRodney W. Grimes goto mem1; 65758f0484fSRodney W. Grimes if (dp->d_namlen > maxlen) { 65858f0484fSRodney W. Grimes if (fts_palloc(sp, (size_t)dp->d_namlen)) { 65958f0484fSRodney W. Grimes /* 66058f0484fSRodney W. Grimes * No more memory for path or structures. Save 66158f0484fSRodney W. Grimes * errno, free up the current structure and the 66258f0484fSRodney W. Grimes * structures already allocated. 66358f0484fSRodney W. Grimes */ 66458f0484fSRodney W. Grimes mem1: saved_errno = errno; 66558f0484fSRodney W. Grimes if (p) 66658f0484fSRodney W. Grimes free(p); 66758f0484fSRodney W. Grimes fts_lfree(head); 66858f0484fSRodney W. Grimes (void)closedir(dirp); 66958f0484fSRodney W. Grimes errno = saved_errno; 67058f0484fSRodney W. Grimes cur->fts_info = FTS_ERR; 67158f0484fSRodney W. Grimes SET(FTS_STOP); 67258f0484fSRodney W. Grimes return (NULL); 67358f0484fSRodney W. Grimes } 67458f0484fSRodney W. Grimes adjaddr = sp->fts_path; 67558f0484fSRodney W. Grimes maxlen = sp->fts_pathlen - sp->fts_cur->fts_pathlen - 1; 67658f0484fSRodney W. Grimes } 67758f0484fSRodney W. Grimes 67858f0484fSRodney W. Grimes p->fts_pathlen = len + dp->d_namlen + 1; 67958f0484fSRodney W. Grimes p->fts_parent = sp->fts_cur; 68058f0484fSRodney W. Grimes p->fts_level = level; 68158f0484fSRodney W. Grimes 682adf6ad9eSPeter Wemm #ifdef FTS_WHITEOUT 683adf6ad9eSPeter Wemm if (dp->d_type == DT_WHT) 684adf6ad9eSPeter Wemm p->fts_flags |= FTS_ISW; 685adf6ad9eSPeter Wemm #endif 686adf6ad9eSPeter Wemm 68758f0484fSRodney W. Grimes if (cderrno) { 68858f0484fSRodney W. Grimes if (nlinks) { 68958f0484fSRodney W. Grimes p->fts_info = FTS_NS; 69058f0484fSRodney W. Grimes p->fts_errno = cderrno; 69158f0484fSRodney W. Grimes } else 69258f0484fSRodney W. Grimes p->fts_info = FTS_NSOK; 69358f0484fSRodney W. Grimes p->fts_accpath = cur->fts_accpath; 69458f0484fSRodney W. Grimes } else if (nlinks == 0 69558f0484fSRodney W. Grimes #ifdef DT_DIR 69651295a4dSJordan K. Hubbard || (nlinks > 0 && 69751295a4dSJordan K. Hubbard dp->d_type != DT_DIR && dp->d_type != DT_UNKNOWN) 69858f0484fSRodney W. Grimes #endif 69958f0484fSRodney W. Grimes ) { 70058f0484fSRodney W. Grimes p->fts_accpath = 70158f0484fSRodney W. Grimes ISSET(FTS_NOCHDIR) ? p->fts_path : p->fts_name; 70258f0484fSRodney W. Grimes p->fts_info = FTS_NSOK; 70358f0484fSRodney W. Grimes } else { 70458f0484fSRodney W. Grimes /* Build a file name for fts_stat to stat. */ 70558f0484fSRodney W. Grimes if (ISSET(FTS_NOCHDIR)) { 70658f0484fSRodney W. Grimes p->fts_accpath = p->fts_path; 70758f0484fSRodney W. Grimes memmove(cp, p->fts_name, p->fts_namelen + 1); 70858f0484fSRodney W. Grimes } else 70958f0484fSRodney W. Grimes p->fts_accpath = p->fts_name; 71058f0484fSRodney W. Grimes /* Stat it. */ 71158f0484fSRodney W. Grimes p->fts_info = fts_stat(sp, p, 0); 71258f0484fSRodney W. Grimes 71358f0484fSRodney W. Grimes /* Decrement link count if applicable. */ 71458f0484fSRodney W. Grimes if (nlinks > 0 && (p->fts_info == FTS_D || 71558f0484fSRodney W. Grimes p->fts_info == FTS_DC || p->fts_info == FTS_DOT)) 71658f0484fSRodney W. Grimes --nlinks; 71758f0484fSRodney W. Grimes } 71858f0484fSRodney W. Grimes 71958f0484fSRodney W. Grimes /* We walk in directory order so "ls -f" doesn't get upset. */ 72058f0484fSRodney W. Grimes p->fts_link = NULL; 72158f0484fSRodney W. Grimes if (head == NULL) 72258f0484fSRodney W. Grimes head = tail = p; 72358f0484fSRodney W. Grimes else { 72458f0484fSRodney W. Grimes tail->fts_link = p; 72558f0484fSRodney W. Grimes tail = p; 72658f0484fSRodney W. Grimes } 72758f0484fSRodney W. Grimes ++nitems; 72858f0484fSRodney W. Grimes } 72958f0484fSRodney W. Grimes (void)closedir(dirp); 73058f0484fSRodney W. Grimes 73158f0484fSRodney W. Grimes /* 73258f0484fSRodney W. Grimes * If had to realloc the path, adjust the addresses for the rest 73358f0484fSRodney W. Grimes * of the tree. 73458f0484fSRodney W. Grimes */ 73558f0484fSRodney W. Grimes if (adjaddr) 73658f0484fSRodney W. Grimes fts_padjust(sp, adjaddr); 73758f0484fSRodney W. Grimes 73858f0484fSRodney W. Grimes /* 73958f0484fSRodney W. Grimes * If not changing directories, reset the path back to original 74058f0484fSRodney W. Grimes * state. 74158f0484fSRodney W. Grimes */ 74258f0484fSRodney W. Grimes if (ISSET(FTS_NOCHDIR)) { 74358f0484fSRodney W. Grimes if (cp - 1 > sp->fts_path) 74458f0484fSRodney W. Grimes --cp; 74558f0484fSRodney W. Grimes *cp = '\0'; 74658f0484fSRodney W. Grimes } 74758f0484fSRodney W. Grimes 74858f0484fSRodney W. Grimes /* 74958f0484fSRodney W. Grimes * If descended after called from fts_children or after called from 75058f0484fSRodney W. Grimes * fts_read and nothing found, get back. At the root level we use 75158f0484fSRodney W. Grimes * the saved fd; if one of fts_open()'s arguments is a relative path 75258f0484fSRodney W. Grimes * to an empty directory, we wind up here with no other way back. If 75358f0484fSRodney W. Grimes * can't get back, we're done. 75458f0484fSRodney W. Grimes */ 75558f0484fSRodney W. Grimes if (descend && (type == BCHILD || !nitems) && 75658f0484fSRodney W. Grimes (cur->fts_level == FTS_ROOTLEVEL ? 75758f0484fSRodney W. Grimes FCHDIR(sp, sp->fts_rfd) : CHDIR(sp, ".."))) { 75858f0484fSRodney W. Grimes cur->fts_info = FTS_ERR; 75958f0484fSRodney W. Grimes SET(FTS_STOP); 76058f0484fSRodney W. Grimes return (NULL); 76158f0484fSRodney W. Grimes } 76258f0484fSRodney W. Grimes 76358f0484fSRodney W. Grimes /* If didn't find anything, return NULL. */ 76458f0484fSRodney W. Grimes if (!nitems) { 76558f0484fSRodney W. Grimes if (type == BREAD) 76658f0484fSRodney W. Grimes cur->fts_info = FTS_DP; 76758f0484fSRodney W. Grimes return (NULL); 76858f0484fSRodney W. Grimes } 76958f0484fSRodney W. Grimes 77058f0484fSRodney W. Grimes /* Sort the entries. */ 77158f0484fSRodney W. Grimes if (sp->fts_compar && nitems > 1) 77258f0484fSRodney W. Grimes head = fts_sort(sp, head, nitems); 77358f0484fSRodney W. Grimes return (head); 77458f0484fSRodney W. Grimes } 77558f0484fSRodney W. Grimes 77658f0484fSRodney W. Grimes static u_short 77758f0484fSRodney W. Grimes fts_stat(sp, p, follow) 77858f0484fSRodney W. Grimes FTS *sp; 77958f0484fSRodney W. Grimes register FTSENT *p; 78058f0484fSRodney W. Grimes int follow; 78158f0484fSRodney W. Grimes { 78258f0484fSRodney W. Grimes register FTSENT *t; 78358f0484fSRodney W. Grimes register dev_t dev; 78458f0484fSRodney W. Grimes register ino_t ino; 78558f0484fSRodney W. Grimes struct stat *sbp, sb; 78658f0484fSRodney W. Grimes int saved_errno; 78758f0484fSRodney W. Grimes 78858f0484fSRodney W. Grimes /* If user needs stat info, stat buffer already allocated. */ 78958f0484fSRodney W. Grimes sbp = ISSET(FTS_NOSTAT) ? &sb : p->fts_statp; 79058f0484fSRodney W. Grimes 791adf6ad9eSPeter Wemm #ifdef FTS_WHITEOUT 792adf6ad9eSPeter Wemm /* check for whiteout */ 793adf6ad9eSPeter Wemm if (p->fts_flags & FTS_ISW) { 794adf6ad9eSPeter Wemm if (sbp != &sb) { 795adf6ad9eSPeter Wemm memset(sbp, '\0', sizeof (*sbp)); 796adf6ad9eSPeter Wemm sbp->st_mode = S_IFWHT; 797adf6ad9eSPeter Wemm } 798adf6ad9eSPeter Wemm return (FTS_W); 799adf6ad9eSPeter Wemm } 800adf6ad9eSPeter Wemm #endif 801adf6ad9eSPeter Wemm 80258f0484fSRodney W. Grimes /* 80358f0484fSRodney W. Grimes * If doing a logical walk, or application requested FTS_FOLLOW, do 80458f0484fSRodney W. Grimes * a stat(2). If that fails, check for a non-existent symlink. If 80558f0484fSRodney W. Grimes * fail, set the errno from the stat call. 80658f0484fSRodney W. Grimes */ 80758f0484fSRodney W. Grimes if (ISSET(FTS_LOGICAL) || follow) { 80858f0484fSRodney W. Grimes if (stat(p->fts_accpath, sbp)) { 80958f0484fSRodney W. Grimes saved_errno = errno; 81058f0484fSRodney W. Grimes if (!lstat(p->fts_accpath, sbp)) { 81158f0484fSRodney W. Grimes errno = 0; 81258f0484fSRodney W. Grimes return (FTS_SLNONE); 81358f0484fSRodney W. Grimes } 81458f0484fSRodney W. Grimes p->fts_errno = saved_errno; 81558f0484fSRodney W. Grimes goto err; 81658f0484fSRodney W. Grimes } 81758f0484fSRodney W. Grimes } else if (lstat(p->fts_accpath, sbp)) { 81858f0484fSRodney W. Grimes p->fts_errno = errno; 81958f0484fSRodney W. Grimes err: memset(sbp, 0, sizeof(struct stat)); 82058f0484fSRodney W. Grimes return (FTS_NS); 82158f0484fSRodney W. Grimes } 82258f0484fSRodney W. Grimes 82358f0484fSRodney W. Grimes if (S_ISDIR(sbp->st_mode)) { 82458f0484fSRodney W. Grimes /* 82558f0484fSRodney W. Grimes * Set the device/inode. Used to find cycles and check for 82658f0484fSRodney W. Grimes * crossing mount points. Also remember the link count, used 82758f0484fSRodney W. Grimes * in fts_build to limit the number of stat calls. It is 82858f0484fSRodney W. Grimes * understood that these fields are only referenced if fts_info 82958f0484fSRodney W. Grimes * is set to FTS_D. 83058f0484fSRodney W. Grimes */ 83158f0484fSRodney W. Grimes dev = p->fts_dev = sbp->st_dev; 83258f0484fSRodney W. Grimes ino = p->fts_ino = sbp->st_ino; 83358f0484fSRodney W. Grimes p->fts_nlink = sbp->st_nlink; 83458f0484fSRodney W. Grimes 83558f0484fSRodney W. Grimes if (ISDOT(p->fts_name)) 83658f0484fSRodney W. Grimes return (FTS_DOT); 83758f0484fSRodney W. Grimes 83858f0484fSRodney W. Grimes /* 83958f0484fSRodney W. Grimes * Cycle detection is done by brute force when the directory 84058f0484fSRodney W. Grimes * is first encountered. If the tree gets deep enough or the 84158f0484fSRodney W. Grimes * number of symbolic links to directories is high enough, 84258f0484fSRodney W. Grimes * something faster might be worthwhile. 84358f0484fSRodney W. Grimes */ 84458f0484fSRodney W. Grimes for (t = p->fts_parent; 84558f0484fSRodney W. Grimes t->fts_level >= FTS_ROOTLEVEL; t = t->fts_parent) 84658f0484fSRodney W. Grimes if (ino == t->fts_ino && dev == t->fts_dev) { 84758f0484fSRodney W. Grimes p->fts_cycle = t; 84858f0484fSRodney W. Grimes return (FTS_DC); 84958f0484fSRodney W. Grimes } 85058f0484fSRodney W. Grimes return (FTS_D); 85158f0484fSRodney W. Grimes } 85258f0484fSRodney W. Grimes if (S_ISLNK(sbp->st_mode)) 85358f0484fSRodney W. Grimes return (FTS_SL); 85458f0484fSRodney W. Grimes if (S_ISREG(sbp->st_mode)) 85558f0484fSRodney W. Grimes return (FTS_F); 85658f0484fSRodney W. Grimes return (FTS_DEFAULT); 85758f0484fSRodney W. Grimes } 85858f0484fSRodney W. Grimes 85958f0484fSRodney W. Grimes static FTSENT * 86058f0484fSRodney W. Grimes fts_sort(sp, head, nitems) 86158f0484fSRodney W. Grimes FTS *sp; 86258f0484fSRodney W. Grimes FTSENT *head; 86358f0484fSRodney W. Grimes register int nitems; 86458f0484fSRodney W. Grimes { 86558f0484fSRodney W. Grimes register FTSENT **ap, *p; 86658f0484fSRodney W. Grimes 86758f0484fSRodney W. Grimes /* 86858f0484fSRodney W. Grimes * Construct an array of pointers to the structures and call qsort(3). 86958f0484fSRodney W. Grimes * Reassemble the array in the order returned by qsort. If unable to 87058f0484fSRodney W. Grimes * sort for memory reasons, return the directory entries in their 87158f0484fSRodney W. Grimes * current order. Allocate enough space for the current needs plus 87258f0484fSRodney W. Grimes * 40 so don't realloc one entry at a time. 87358f0484fSRodney W. Grimes */ 87458f0484fSRodney W. Grimes if (nitems > sp->fts_nitems) { 87558f0484fSRodney W. Grimes sp->fts_nitems = nitems + 40; 87658f0484fSRodney W. Grimes if ((sp->fts_array = realloc(sp->fts_array, 87758f0484fSRodney W. Grimes (size_t)(sp->fts_nitems * sizeof(FTSENT *)))) == NULL) { 87858f0484fSRodney W. Grimes sp->fts_nitems = 0; 87958f0484fSRodney W. Grimes return (head); 88058f0484fSRodney W. Grimes } 88158f0484fSRodney W. Grimes } 88258f0484fSRodney W. Grimes for (ap = sp->fts_array, p = head; p; p = p->fts_link) 88358f0484fSRodney W. Grimes *ap++ = p; 88458f0484fSRodney W. Grimes qsort((void *)sp->fts_array, nitems, sizeof(FTSENT *), sp->fts_compar); 88558f0484fSRodney W. Grimes for (head = *(ap = sp->fts_array); --nitems; ++ap) 88658f0484fSRodney W. Grimes ap[0]->fts_link = ap[1]; 88758f0484fSRodney W. Grimes ap[0]->fts_link = NULL; 88858f0484fSRodney W. Grimes return (head); 88958f0484fSRodney W. Grimes } 89058f0484fSRodney W. Grimes 89158f0484fSRodney W. Grimes static FTSENT * 89258f0484fSRodney W. Grimes fts_alloc(sp, name, namelen) 89358f0484fSRodney W. Grimes FTS *sp; 89458f0484fSRodney W. Grimes char *name; 89558f0484fSRodney W. Grimes register int namelen; 89658f0484fSRodney W. Grimes { 89758f0484fSRodney W. Grimes register FTSENT *p; 89858f0484fSRodney W. Grimes size_t len; 89958f0484fSRodney W. Grimes 90058f0484fSRodney W. Grimes /* 90158f0484fSRodney W. Grimes * The file name is a variable length array and no stat structure is 90258f0484fSRodney W. Grimes * necessary if the user has set the nostat bit. Allocate the FTSENT 90358f0484fSRodney W. Grimes * structure, the file name and the stat structure in one chunk, but 90458f0484fSRodney W. Grimes * be careful that the stat structure is reasonably aligned. Since the 90558f0484fSRodney W. Grimes * fts_name field is declared to be of size 1, the fts_name pointer is 90658f0484fSRodney W. Grimes * namelen + 2 before the first possible address of the stat structure. 90758f0484fSRodney W. Grimes */ 90858f0484fSRodney W. Grimes len = sizeof(FTSENT) + namelen; 90958f0484fSRodney W. Grimes if (!ISSET(FTS_NOSTAT)) 91058f0484fSRodney W. Grimes len += sizeof(struct stat) + ALIGNBYTES; 91158f0484fSRodney W. Grimes if ((p = malloc(len)) == NULL) 91258f0484fSRodney W. Grimes return (NULL); 91358f0484fSRodney W. Grimes 91458f0484fSRodney W. Grimes /* Copy the name plus the trailing NULL. */ 91558f0484fSRodney W. Grimes memmove(p->fts_name, name, namelen + 1); 91658f0484fSRodney W. Grimes 91758f0484fSRodney W. Grimes if (!ISSET(FTS_NOSTAT)) 91858f0484fSRodney W. Grimes p->fts_statp = (struct stat *)ALIGN(p->fts_name + namelen + 2); 91958f0484fSRodney W. Grimes p->fts_namelen = namelen; 92058f0484fSRodney W. Grimes p->fts_path = sp->fts_path; 92158f0484fSRodney W. Grimes p->fts_errno = 0; 92258f0484fSRodney W. Grimes p->fts_flags = 0; 92358f0484fSRodney W. Grimes p->fts_instr = FTS_NOINSTR; 92458f0484fSRodney W. Grimes p->fts_number = 0; 92558f0484fSRodney W. Grimes p->fts_pointer = NULL; 92658f0484fSRodney W. Grimes return (p); 92758f0484fSRodney W. Grimes } 92858f0484fSRodney W. Grimes 92958f0484fSRodney W. Grimes static void 93058f0484fSRodney W. Grimes fts_lfree(head) 93158f0484fSRodney W. Grimes register FTSENT *head; 93258f0484fSRodney W. Grimes { 93358f0484fSRodney W. Grimes register FTSENT *p; 93458f0484fSRodney W. Grimes 93558f0484fSRodney W. Grimes /* Free a linked list of structures. */ 93651295a4dSJordan K. Hubbard while ( (p = head) ) { 93758f0484fSRodney W. Grimes head = head->fts_link; 93858f0484fSRodney W. Grimes free(p); 93958f0484fSRodney W. Grimes } 94058f0484fSRodney W. Grimes } 94158f0484fSRodney W. Grimes 94258f0484fSRodney W. Grimes /* 94358f0484fSRodney W. Grimes * Allow essentially unlimited paths; find, rm, ls should all work on any tree. 94458f0484fSRodney W. Grimes * Most systems will allow creation of paths much longer than MAXPATHLEN, even 94558f0484fSRodney W. Grimes * though the kernel won't resolve them. Add the size (not just what's needed) 94658f0484fSRodney W. Grimes * plus 256 bytes so don't realloc the path 2 bytes at a time. 94758f0484fSRodney W. Grimes */ 94858f0484fSRodney W. Grimes static int 94958f0484fSRodney W. Grimes fts_palloc(sp, more) 95058f0484fSRodney W. Grimes FTS *sp; 95158f0484fSRodney W. Grimes size_t more; 95258f0484fSRodney W. Grimes { 95358f0484fSRodney W. Grimes sp->fts_pathlen += more + 256; 95458f0484fSRodney W. Grimes sp->fts_path = realloc(sp->fts_path, (size_t)sp->fts_pathlen); 95558f0484fSRodney W. Grimes return (sp->fts_path == NULL); 95658f0484fSRodney W. Grimes } 95758f0484fSRodney W. Grimes 95858f0484fSRodney W. Grimes /* 95958f0484fSRodney W. Grimes * When the path is realloc'd, have to fix all of the pointers in structures 96058f0484fSRodney W. Grimes * already returned. 96158f0484fSRodney W. Grimes */ 96258f0484fSRodney W. Grimes static void 96358f0484fSRodney W. Grimes fts_padjust(sp, addr) 96458f0484fSRodney W. Grimes FTS *sp; 96558f0484fSRodney W. Grimes void *addr; 96658f0484fSRodney W. Grimes { 96758f0484fSRodney W. Grimes FTSENT *p; 96858f0484fSRodney W. Grimes 96958f0484fSRodney W. Grimes #define ADJUST(p) { \ 97058f0484fSRodney W. Grimes (p)->fts_accpath = \ 97158f0484fSRodney W. Grimes (char *)addr + ((p)->fts_accpath - (p)->fts_path); \ 97258f0484fSRodney W. Grimes (p)->fts_path = addr; \ 97358f0484fSRodney W. Grimes } 97458f0484fSRodney W. Grimes /* Adjust the current set of children. */ 97558f0484fSRodney W. Grimes for (p = sp->fts_child; p; p = p->fts_link) 97658f0484fSRodney W. Grimes ADJUST(p); 97758f0484fSRodney W. Grimes 97858f0484fSRodney W. Grimes /* Adjust the rest of the tree. */ 97958f0484fSRodney W. Grimes for (p = sp->fts_cur; p->fts_level >= FTS_ROOTLEVEL;) { 98058f0484fSRodney W. Grimes ADJUST(p); 98158f0484fSRodney W. Grimes p = p->fts_link ? p->fts_link : p->fts_parent; 98258f0484fSRodney W. Grimes } 98358f0484fSRodney W. Grimes } 98458f0484fSRodney W. Grimes 98558f0484fSRodney W. Grimes static size_t 98658f0484fSRodney W. Grimes fts_maxarglen(argv) 98758f0484fSRodney W. Grimes char * const *argv; 98858f0484fSRodney W. Grimes { 98958f0484fSRodney W. Grimes size_t len, max; 99058f0484fSRodney W. Grimes 99158f0484fSRodney W. Grimes for (max = 0; *argv; ++argv) 99258f0484fSRodney W. Grimes if ((len = strlen(*argv)) > max) 99358f0484fSRodney W. Grimes max = len; 99458f0484fSRodney W. Grimes return (max); 99558f0484fSRodney W. Grimes } 996