AuroraMiddleware/glibc
1
0
Fork 0
mirror of https://sourceware.org/git/glibc.git synced 2025-01-18 14:40:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update from recent BSD source.

Browse Source
This commit is contained in:
Ulrich Drepper 2001-08-18 06:22:09 +00:00
parent c3151cca37
commit f8efc16a02
1 changed files with 201 additions and 115 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

316
io/fts.c
View File

@ -56,12 +56,13 @@ static char sccsid[] = "@(#)fts.c 8.6 (Berkeley) 8/14/94";
static FTSENT *fts_alloc __P((FTS *, const char *, int)) internal_function; static FTSENT *fts_alloc __P((FTS *, const char *, int)) internal_function;
static FTSENT *fts_build __P((FTS *, int)) internal_function; static FTSENT *fts_build __P((FTS *, int)) internal_function;
static void fts_lfree __P((FTSENT *)) internal_function; static void fts_lfree __P((FTSENT *)) internal_function;
static void fts_load __P((FTS *, FTSENT *)); static void fts_load __P((FTS *, FTSENT *)) internal_function;
static size_t fts_maxarglen __P((char * const *)) internal_function; static size_t fts_maxarglen __P((char * const *)) internal_function;
static void fts_padjust __P((FTS *, void *)) internal_function; static void fts_padjust __P((FTS *, FTSENT *)) internal_function;
static int fts_palloc __P((FTS *, size_t)) internal_function; static int fts_palloc __P((FTS *, size_t)) internal_function;
static FTSENT *fts_sort __P((FTS *, FTSENT *, int)) internal_function; static FTSENT *fts_sort __P((FTS *, FTSENT *, int)) internal_function;
static u_short fts_stat __P((FTS *, struct dirent *, FTSENT *, int)) static u_short fts_stat __P((FTS *, FTSENT *, int)) internal_function;
static int fts_safe_changedir __P((FTS *, FTSENT *, int, const char *))
internal_function; internal_function;
#ifndef MAX #ifndef MAX
@ -72,10 +73,10 @@ static u_short fts_stat __P((FTS *, struct dirent *, FTSENT *, int))
#define ISDOT(a) (a[0] == '.' && (!a[1] || (a[1] == '.' && !a[2]))) #define ISDOT(a) (a[0] == '.' && (!a[1] || (a[1] == '.' && !a[2])))
#define ISSET(opt) (sp->fts_options & opt) #define CLR(opt) (sp->fts_options &= ~(opt))
#define SET(opt) (sp->fts_options |= opt) #define ISSET(opt) (sp->fts_options & (opt))
#define SET(opt) (sp->fts_options |= (opt))
#define CHDIR(sp, path) (!ISSET(FTS_NOCHDIR) && __chdir(path))
#define FCHDIR(sp, fd) (!ISSET(FTS_NOCHDIR) && __fchdir(fd)) #define FCHDIR(sp, fd) (!ISSET(FTS_NOCHDIR) && __fchdir(fd))
/* fts_build flags */ /* fts_build flags */
@ -104,7 +105,7 @@ fts_open(argv, options, compar)
/* Allocate/initialize the stream */ /* Allocate/initialize the stream */
if ((sp = malloc((u_int)sizeof(FTS))) == NULL) if ((sp = malloc((u_int)sizeof(FTS))) == NULL)
return (NULL); return (NULL);
__bzero(sp, sizeof(FTS)); memset(sp, 0, sizeof(FTS));
sp->fts_compar = (int (*) __P((const void *, const void *))) compar; sp->fts_compar = (int (*) __P((const void *, const void *))) compar;
sp->fts_options = options; sp->fts_options = options;
@ -128,7 +129,7 @@ fts_open(argv, options, compar)
parent->fts_level = FTS_ROOTPARENTLEVEL; parent->fts_level = FTS_ROOTPARENTLEVEL;
/* Allocate/initialize root(s). */ /* Allocate/initialize root(s). */
for (root = NULL, nitems = 0; *argv; ++argv, ++nitems) { for (root = NULL, nitems = 0; *argv != NULL; ++argv, ++nitems) {
/* Don't allow zero-length paths. */ /* Don't allow zero-length paths. */
if ((len = strlen(*argv)) == 0) { if ((len = strlen(*argv)) == 0) {
__set_errno (ENOENT); __set_errno (ENOENT);
@ -139,7 +140,7 @@ fts_open(argv, options, compar)
p->fts_level = FTS_ROOTLEVEL; p->fts_level = FTS_ROOTLEVEL;
p->fts_parent = parent; p->fts_parent = parent;
p->fts_accpath = p->fts_name; p->fts_accpath = p->fts_name;
p->fts_info = fts_stat(sp, NULL, p, ISSET(FTS_COMFOLLOW)); p->fts_info = fts_stat(sp, p, ISSET(FTS_COMFOLLOW));
/* Command-line "." and ".." are real directories. */ /* Command-line "." and ".." are real directories. */
if (p->fts_info == FTS_DOT) if (p->fts_info == FTS_DOT)
@ -176,7 +177,7 @@ fts_open(argv, options, compar)
sp->fts_cur->fts_info = FTS_INIT; sp->fts_cur->fts_info = FTS_INIT;
/* /*
* If using chdir(2), grab a file descriptor pointing to dot to insure * If using chdir(2), grab a file descriptor pointing to dot to ensure
* that we can get back here; this could be avoided for some paths, * that we can get back here; this could be avoided for some paths,
* but almost certainly not worth the effort. Slashes, symbolic links, * but almost certainly not worth the effort. Slashes, symbolic links,
* and ".." are all fairly nasty problems. Note, if we can't get the * and ".." are all fairly nasty problems. Note, if we can't get the
@ -196,6 +197,7 @@ mem1: free(sp);
} }
static void static void
internal_function
fts_load(sp, p) fts_load(sp, p)
FTS *sp; FTS *sp;
register FTSENT *p; register FTSENT *p;
@ -211,10 +213,10 @@ fts_load(sp, p)
* known that the path will fit. * known that the path will fit.
*/ */
len = p->fts_pathlen = p->fts_namelen; len = p->fts_pathlen = p->fts_namelen;
bcopy(p->fts_name, sp->fts_path, len + 1); memmove(sp->fts_path, p->fts_name, len + 1);
if ((cp = rindex(p->fts_name, '/')) && (cp != p->fts_name || cp[1])) { if ((cp = strrchr(p->fts_name, '/')) && (cp != p->fts_name || cp[1])) {
len = strlen(++cp); len = strlen(++cp);
bcopy(cp, p->fts_name, len + 1); memmove(p->fts_name, cp, len + 1);
p->fts_namelen = len; p->fts_namelen = len;
} }
p->fts_accpath = p->fts_path = sp->fts_path; p->fts_accpath = p->fts_path = sp->fts_path;
@ -227,7 +229,6 @@ fts_close(sp)
{ {
register FTSENT *freep, *p; register FTSENT *freep, *p;
int saved_errno; int saved_errno;
int retval = 0;
/* /*
* This still works if we haven't read anything -- the dummy structure * This still works if we haven't read anything -- the dummy structure
@ -237,7 +238,7 @@ fts_close(sp)
if (sp->fts_cur) { if (sp->fts_cur) {
for (p = sp->fts_cur; p->fts_level >= FTS_ROOTLEVEL;) { for (p = sp->fts_cur; p->fts_level >= FTS_ROOTLEVEL;) {
freep = p; freep = p;
p = p->fts_link ? p->fts_link : p->fts_parent; p = p->fts_link != NULL ? p->fts_link : p->fts_parent;
free(freep); free(freep);
} }
free(p); free(p);
@ -254,27 +255,28 @@ fts_close(sp)
if (!ISSET(FTS_NOCHDIR)) { if (!ISSET(FTS_NOCHDIR)) {
saved_errno = __fchdir(sp->fts_rfd) ? errno : 0; saved_errno = __fchdir(sp->fts_rfd) ? errno : 0;
(void)__close(sp->fts_rfd); (void)__close(sp->fts_rfd);
}
/* Set errno and return. */ /* Set errno and return. */
if (!ISSET(FTS_NOCHDIR) && saved_errno) { if (saved_errno != 0) {
__set_errno (saved_errno); /* Free up the stream pointer. */
retval = -1; free(sp);
__set_errno (saved_errno);
return (-1);
}
} }
/* Free up the stream pointer. */ /* Free up the stream pointer. */
free (sp); free(sp);
return (0);
return retval;
} }
/* /*
* Special case a root of "/" so that slashes aren't appended which would * Special case of "/" at the end of the path so that slashes aren't
* cause paths to be written as "//foo". * appended which would cause paths to be written as "....//foo".
*/ */
#define NAPPEND(p) \ #define NAPPEND(p) \
(p->fts_level == FTS_ROOTLEVEL && p->fts_pathlen == 1 && \ (p->fts_path[p->fts_pathlen - 1] == '/' \
p->fts_path[0] == '/' ? 0 : p->fts_pathlen) ? p->fts_pathlen - 1 : p->fts_pathlen)
FTSENT * FTSENT *
fts_read(sp) fts_read(sp)
@ -298,7 +300,7 @@ fts_read(sp)
/* Any type of file may be re-visited; re-stat and re-turn. */ /* Any type of file may be re-visited; re-stat and re-turn. */
if (instr == FTS_AGAIN) { if (instr == FTS_AGAIN) {
p->fts_info = fts_stat(sp, NULL, p, 0); p->fts_info = fts_stat(sp, p, 0);
return (p); return (p);
} }
@ -310,7 +312,7 @@ fts_read(sp)
*/ */
if (instr == FTS_FOLLOW && if (instr == FTS_FOLLOW &&
(p->fts_info == FTS_SL || p->fts_info == FTS_SLNONE)) { (p->fts_info == FTS_SL || p->fts_info == FTS_SLNONE)) {
p->fts_info = fts_stat(sp, NULL, p, 1); p->fts_info = fts_stat(sp, p, 1);
if (p->fts_info == FTS_D && !ISSET(FTS_NOCHDIR)) { if (p->fts_info == FTS_D && !ISSET(FTS_NOCHDIR)) {
if ((p->fts_symfd = __open(".", O_RDONLY, 0)) < 0) { if ((p->fts_symfd = __open(".", O_RDONLY, 0)) < 0) {
p->fts_errno = errno; p->fts_errno = errno;
@ -337,8 +339,8 @@ fts_read(sp)
} }
/* Rebuild if only read the names and now traversing. */ /* Rebuild if only read the names and now traversing. */
if (sp->fts_child && sp->fts_options & FTS_NAMEONLY) { if (sp->fts_child != NULL && ISSET(FTS_NAMEONLY)) {
sp->fts_options &= ~FTS_NAMEONLY; CLR(FTS_NAMEONLY);
fts_lfree(sp->fts_child); fts_lfree(sp->fts_child);
sp->fts_child = NULL; sp->fts_child = NULL;
} }
@ -355,11 +357,12 @@ fts_read(sp)
* If haven't read do so. If the read fails, fts_build sets * If haven't read do so. If the read fails, fts_build sets
* FTS_STOP or the fts_info field of the node. * FTS_STOP or the fts_info field of the node.
*/ */
if (sp->fts_child) { if (sp->fts_child != NULL) {
if (CHDIR(sp, p->fts_accpath)) { if (fts_safe_changedir(sp, p, -1, p->fts_accpath)) {
p->fts_errno = errno; p->fts_errno = errno;
p->fts_flags |= FTS_DONTCHDIR; p->fts_flags |= FTS_DONTCHDIR;
for (p = sp->fts_child; p; p = p->fts_link) for (p = sp->fts_child; p != NULL;
p = p->fts_link)
p->fts_accpath = p->fts_accpath =
p->fts_parent->fts_accpath; p->fts_parent->fts_accpath;
} }
@ -375,15 +378,15 @@ fts_read(sp)
/* Move to the next node on this level. */ /* Move to the next node on this level. */
next: tmp = p; next: tmp = p;
if ((p = p->fts_link)) { if ((p = p->fts_link) != NULL) {
free(tmp); free(tmp);
/* /*
* If reached the top, return to the original directory, and * If reached the top, return to the original directory (or
* load the paths for the next root. * the root of the tree), and load the paths for the next root.
*/ */
if (p->fts_level == FTS_ROOTLEVEL) { if (p->fts_level == FTS_ROOTLEVEL) {
if (!ISSET(FTS_NOCHDIR) && FCHDIR(sp, sp->fts_rfd)) { if (FCHDIR(sp, sp->fts_rfd)) {
SET(FTS_STOP); SET(FTS_STOP);
return (NULL); return (NULL);
} }
@ -399,7 +402,7 @@ next: tmp = p;
if (p->fts_instr == FTS_SKIP) if (p->fts_instr == FTS_SKIP)
goto next; goto next;
if (p->fts_instr == FTS_FOLLOW) { if (p->fts_instr == FTS_FOLLOW) {
p->fts_info = fts_stat(sp, NULL, p, 1); p->fts_info = fts_stat(sp, p, 1);
if (p->fts_info == FTS_D && !ISSET(FTS_NOCHDIR)) { if (p->fts_info == FTS_D && !ISSET(FTS_NOCHDIR)) {
if ((p->fts_symfd = if ((p->fts_symfd =
__open(".", O_RDONLY, 0)) < 0) { __open(".", O_RDONLY, 0)) < 0) {
@ -413,7 +416,7 @@ next: tmp = p;
name: t = sp->fts_path + NAPPEND(p->fts_parent); name: t = sp->fts_path + NAPPEND(p->fts_parent);
*t++ = '/'; *t++ = '/';
bcopy(p->fts_name, t, p->fts_namelen + 1); memmove(t, p->fts_name, p->fts_namelen + 1);
return (sp->fts_cur = p); return (sp->fts_cur = p);
} }
@ -431,7 +434,7 @@ name: t = sp->fts_path + NAPPEND(p->fts_parent);
return (sp->fts_cur = NULL); return (sp->fts_cur = NULL);
} }
/* Nul terminate the pathname. */ /* NUL terminate the pathname. */
sp->fts_path[p->fts_pathlen] = '\0'; sp->fts_path[p->fts_pathlen] = '\0';
/* /*
@ -440,7 +443,7 @@ name: t = sp->fts_path + NAPPEND(p->fts_parent);
* one directory. * one directory.
*/ */
if (p->fts_level == FTS_ROOTLEVEL) { if (p->fts_level == FTS_ROOTLEVEL) {
if (!ISSET(FTS_NOCHDIR) && FCHDIR(sp, sp->fts_rfd)) { if (FCHDIR(sp, sp->fts_rfd)) {
SET(FTS_STOP); SET(FTS_STOP);
return (NULL); return (NULL);
} }
@ -453,11 +456,10 @@ name: t = sp->fts_path + NAPPEND(p->fts_parent);
return (NULL); return (NULL);
} }
(void)__close(p->fts_symfd); (void)__close(p->fts_symfd);
} else if (!(p->fts_flags & FTS_DONTCHDIR)) { } else if (!(p->fts_flags & FTS_DONTCHDIR) &&
if (CHDIR(sp, "..")) { fts_safe_changedir(sp, p->fts_parent, -1, "..")) {
SET(FTS_STOP); SET(FTS_STOP);
return (NULL); return (NULL);
}
} }
p->fts_info = p->fts_errno ? FTS_ERR : FTS_DP; p->fts_info = p->fts_errno ? FTS_ERR : FTS_DP;
return (sp->fts_cur = p); return (sp->fts_cur = p);
@ -476,7 +478,7 @@ fts_set(sp, p, instr)
FTSENT *p; FTSENT *p;
int instr; int instr;
{ {
if (instr && instr != FTS_AGAIN && instr != FTS_FOLLOW && if (instr != 0 && instr != FTS_AGAIN && instr != FTS_FOLLOW &&
instr != FTS_NOINSTR && instr != FTS_SKIP) { instr != FTS_NOINSTR && instr != FTS_SKIP) {
__set_errno (EINVAL); __set_errno (EINVAL);
return (1); return (1);
@ -493,7 +495,7 @@ fts_children(sp, instr)
register FTSENT *p; register FTSENT *p;
int fd; int fd;
if (instr && instr != FTS_NAMEONLY) { if (instr != 0 && instr != FTS_NAMEONLY) {
__set_errno (EINVAL); __set_errno (EINVAL);
return (NULL); return (NULL);
} }
@ -524,11 +526,11 @@ fts_children(sp, instr)
return (NULL); return (NULL);
/* Free up any previous child list. */ /* Free up any previous child list. */
if (sp->fts_child) if (sp->fts_child != NULL)
fts_lfree(sp->fts_child); fts_lfree(sp->fts_child);
if (instr == FTS_NAMEONLY) { if (instr == FTS_NAMEONLY) {
sp->fts_options |= FTS_NAMEONLY; SET(FTS_NAMEONLY);
instr = BNAMES; instr = BNAMES;
} else } else
instr = BCHILD; instr = BCHILD;
@ -573,17 +575,15 @@ fts_build(sp, type)
register FTS *sp; register FTS *sp;
int type; int type;
{ {
struct dirent *dp; register struct dirent *dp;
register FTSENT *p, *head; register FTSENT *p, *head;
register int nitems; register int nitems;
FTSENT *cur, *tail; FTSENT *cur, *tail;
DIR *dirp; DIR *dirp;
void *adjaddr; void *oldaddr;
int cderrno, descend, len, level, maxlen, nlinks, saved_errno; int cderrno, descend, len, level, maxlen, nlinks, saved_errno,
nostat, doadjust;
char *cp; char *cp;
#ifdef DTF_HIDEW
int oflag;
#endif
/* Set current node pointer. */ /* Set current node pointer. */
cur = sp->fts_cur; cur = sp->fts_cur;
@ -592,15 +592,15 @@ fts_build(sp, type)
* Open the directory for reading. If this fails, we're done. * Open the directory for reading. If this fails, we're done.
* If being called from fts_read, set the fts_info field. * If being called from fts_read, set the fts_info field.
*/ */
#ifdef DTF_HIDEW #if defined FTS_WHITEOUT && 0
if (ISSET (FTS_WHITEOUT)) if (ISSET(FTS_WHITEOUT))
oflag = DTF_NODUP|DTF_REWIND; oflag = DTF_NODUP|DTF_REWIND;
else else
oflag = DTF_HIDEW|DTF_NODUP|DTF_REWIND; oflag = DTF_HIDEW|DTF_NODUP|DTF_REWIND;
#else #else
# define __opendir2(path, flag) __opendir(path) # define __opendir2(path, flag) __opendir(path)
#endif #endif
if ((dirp = __opendir2 (cur->fts_accpath, oflag)) == NULL) { if ((dirp = __opendir2(cur->fts_accpath, oflag)) == NULL) {
if (type == BREAD) { if (type == BREAD) {
cur->fts_info = FTS_DNR; cur->fts_info = FTS_DNR;
cur->fts_errno = errno; cur->fts_errno = errno;
@ -613,12 +613,17 @@ fts_build(sp, type)
* directory if we're cheating on stat calls, 0 if we're not doing * directory if we're cheating on stat calls, 0 if we're not doing
* any stat calls at all, -1 if we're doing stats on everything. * any stat calls at all, -1 if we're doing stats on everything.
*/ */
if (type == BNAMES) if (type == BNAMES) {
nlinks = 0; nlinks = 0;
else if (ISSET(FTS_NOSTAT) && ISSET(FTS_PHYSICAL)) /* Be quiet about nostat, GCC. */
nostat = 0;
} else if (ISSET(FTS_NOSTAT) && ISSET(FTS_PHYSICAL)) {
nlinks = cur->fts_nlink - (ISSET(FTS_SEEDOT) ? 0 : 2); nlinks = cur->fts_nlink - (ISSET(FTS_SEEDOT) ? 0 : 2);
else nostat = 1;
} else {
nlinks = -1; nlinks = -1;
nostat = 0;
}
#ifdef notdef #ifdef notdef
(void)printf("nlinks == %d (cur: %d)\n", nlinks, cur->fts_nlink); (void)printf("nlinks == %d (cur: %d)\n", nlinks, cur->fts_nlink);
@ -642,12 +647,14 @@ fts_build(sp, type)
*/ */
cderrno = 0; cderrno = 0;
if (nlinks || type == BREAD) { if (nlinks || type == BREAD) {
if (FCHDIR(sp, dirfd(dirp))) { if (fts_safe_changedir(sp, cur, dirfd(dirp), NULL)) {
if (nlinks && type == BREAD) if (nlinks && type == BREAD)
cur->fts_errno = errno; cur->fts_errno = errno;
cur->fts_flags |= FTS_DONTCHDIR; cur->fts_flags |= FTS_DONTCHDIR;
descend = 0; descend = 0;
cderrno = errno; cderrno = errno;
(void)closedir(dirp);
dirp = NULL;
} else } else
descend = 1; descend = 1;
} else } else
@ -663,30 +670,30 @@ fts_build(sp, type)
* If not changing directories set a pointer so that can just append * If not changing directories set a pointer so that can just append
* each new name into the path. * each new name into the path.
*/ */
maxlen = sp->fts_pathlen - cur->fts_pathlen - 1;
len = NAPPEND(cur); len = NAPPEND(cur);
if (ISSET(FTS_NOCHDIR)) { if (ISSET(FTS_NOCHDIR)) {
cp = sp->fts_path + len; cp = sp->fts_path + len;
*cp++ = '/'; *cp++ = '/';
} else {
/* GCC, you're too verbose. */
cp = NULL;
} }
len++;
maxlen = sp->fts_pathlen - len;
level = cur->fts_level + 1; level = cur->fts_level + 1;
/* Read the directory, attaching each entry to the `link' pointer. */ /* Read the directory, attaching each entry to the `link' pointer. */
adjaddr = NULL; doadjust = 0;
head = tail = NULL; for (head = tail = NULL, nitems = 0; dirp && (dp = readdir(dirp));) {
nitems = 0;
while((dp = __readdir(dirp))) {
int namlen;
if (!ISSET(FTS_SEEDOT) && ISDOT(dp->d_name)) if (!ISSET(FTS_SEEDOT) && ISDOT(dp->d_name))
continue; continue;
namlen = _D_EXACT_NAMLEN (dp); if ((p = fts_alloc(sp, dp->d_name, (int)_D_EXACT_NAMLEN (dp))) == NULL)
if ((p = fts_alloc(sp, dp->d_name, namlen)) == NULL)
goto mem1; goto mem1;
if (namlen > maxlen) { if (_D_EXACT_NAMLEN (dp) >= maxlen) {/* include space for NUL */
if (fts_palloc(sp, (size_t)namlen)) { oldaddr = sp->fts_path;
if (fts_palloc(sp, _D_EXACT_NAMLEN (dp) + len + 1)) {
/* /*
* No more memory for path or structures. Save * No more memory for path or structures. Save
* errno, free up the current structure and the * errno, free up the current structure and the
@ -697,18 +704,43 @@ mem1: saved_errno = errno;
free(p); free(p);
fts_lfree(head); fts_lfree(head);
(void)__closedir(dirp); (void)__closedir(dirp);
__set_errno (saved_errno);
cur->fts_info = FTS_ERR; cur->fts_info = FTS_ERR;
SET(FTS_STOP); SET(FTS_STOP);
__set_errno (saved_errno);
return (NULL); return (NULL);
} }
adjaddr = sp->fts_path; /* Did realloc() change the pointer? */
maxlen = sp->fts_pathlen - sp->fts_cur->fts_pathlen - 1; if (oldaddr != sp->fts_path) {
doadjust = 1;
if (ISSET(FTS_NOCHDIR))
cp = sp->fts_path + len;
}
maxlen = sp->fts_pathlen - len;
} }
p->fts_pathlen = len + namlen + 1; if (len + _D_EXACT_NAMLEN (dp) >= USHRT_MAX) {
p->fts_parent = sp->fts_cur; /*
* In an FTSENT, fts_pathlen is a u_short so it is
* possible to wraparound here. If we do, free up
* the current structure and the structures already
* allocated, then error out with ENAMETOOLONG.
*/
free(p);
fts_lfree(head);
(void)__closedir(dirp);
cur->fts_info = FTS_ERR;
SET(FTS_STOP);
__set_errno (ENAMETOOLONG);
return (NULL);
}
p->fts_level = level; p->fts_level = level;
p->fts_parent = sp->fts_cur;
p->fts_pathlen = len + _D_EXACT_NAMLEN (dp);
#if defined FTS_WHITEOUT && 0
if (dp->d_type == DT_WHT)
p->fts_flags |= FTS_ISW;
#endif
if (cderrno) { if (cderrno) {
if (nlinks) { if (nlinks) {
@ -719,8 +751,8 @@ mem1: saved_errno = errno;
p->fts_accpath = cur->fts_accpath; p->fts_accpath = cur->fts_accpath;
} else if (nlinks == 0 } else if (nlinks == 0
#if defined DT_DIR && defined _DIRENT_HAVE_D_TYPE #if defined DT_DIR && defined _DIRENT_HAVE_D_TYPE
|| (nlinks > 0 && || (nostat &&
dp->d_type != DT_DIR && dp->d_type != DT_UNKNOWN) dp->d_type != DT_DIR && dp->d_type != DT_UNKNOWN)
#endif #endif
) { ) {
p->fts_accpath = p->fts_accpath =
@ -730,11 +762,11 @@ mem1: saved_errno = errno;
/* Build a file name for fts_stat to stat. */ /* Build a file name for fts_stat to stat. */
if (ISSET(FTS_NOCHDIR)) { if (ISSET(FTS_NOCHDIR)) {
p->fts_accpath = p->fts_path; p->fts_accpath = p->fts_path;
bcopy(p->fts_name, cp, p->fts_namelen + 1); memmove(cp, p->fts_name, p->fts_namelen + 1);
} else } else
p->fts_accpath = p->fts_name; p->fts_accpath = p->fts_name;
/* Stat it. */ /* Stat it. */
p->fts_info = fts_stat(sp, dp, p, 0); p->fts_info = fts_stat(sp, p, 0);
/* Decrement link count if applicable. */ /* Decrement link count if applicable. */
if (nlinks > 0 && (p->fts_info == FTS_D || if (nlinks > 0 && (p->fts_info == FTS_D ||
@ -752,21 +784,22 @@ mem1: saved_errno = errno;
} }
++nitems; ++nitems;
} }
(void)__closedir(dirp); if (dirp)
(void)__closedir(dirp);
/* /*
* If had to realloc the path, adjust the addresses for the rest * If realloc() changed the address of the path, adjust the
* of the tree. * addresses for the rest of the tree and the dir list.
*/ */
if (adjaddr) if (doadjust)
fts_padjust(sp, adjaddr); fts_padjust(sp, head);
/* /*
* If not changing directories, reset the path back to original * If not changing directories, reset the path back to original
* state. * state.
*/ */
if (ISSET(FTS_NOCHDIR)) { if (ISSET(FTS_NOCHDIR)) {
if (cp - 1 > sp->fts_path) if (len == sp->fts_pathlen || nitems == 0)
--cp; --cp;
*cp = '\0'; *cp = '\0';
} }
@ -780,7 +813,8 @@ mem1: saved_errno = errno;
*/ */
if (descend && (type == BCHILD || !nitems) && if (descend && (type == BCHILD || !nitems) &&
(cur->fts_level == FTS_ROOTLEVEL ? (cur->fts_level == FTS_ROOTLEVEL ?
FCHDIR (sp, sp->fts_rfd) : CHDIR (sp, ".."))) { FCHDIR(sp, sp->fts_rfd) :
fts_safe_changedir(sp, cur->fts_parent, -1, ".."))) {
cur->fts_info = FTS_ERR; cur->fts_info = FTS_ERR;
SET(FTS_STOP); SET(FTS_STOP);
return (NULL); return (NULL);
@ -801,10 +835,9 @@ mem1: saved_errno = errno;
static u_short static u_short
internal_function internal_function
fts_stat(sp, dp, p, follow) fts_stat(sp, p, follow)
FTS *sp; FTS *sp;
register FTSENT *p; register FTSENT *p;
struct dirent *dp;
int follow; int follow;
{ {
register FTSENT *t; register FTSENT *t;
@ -816,15 +849,13 @@ fts_stat(sp, dp, p, follow)
/* If user needs stat info, stat buffer already allocated. */ /* If user needs stat info, stat buffer already allocated. */
sbp = ISSET(FTS_NOSTAT) ? &sb : p->fts_statp; sbp = ISSET(FTS_NOSTAT) ? &sb : p->fts_statp;
#if defined DT_WHT && defined S_IFWHT #if defined FTS_WHITEOUT && 0
/* /* check for whiteout */
* Whited-out files don't really exist. However, there's stat(2) file if (p->fts_flags & FTS_ISW) {
* mask for them, so we set it so that programs (i.e., find) don't have if (sbp != &sb) {
* to test FTS_W separately from other file types. memset(sbp, '\0', sizeof (*sbp));
*/ sbp->st_mode = S_IFWHT;
if (dp != NULL && dp->d_type == DT_WHT) { }
memset(sbp, 0, sizeof(struct stat));
sbp->st_mode = S_IFWHT;
return (FTS_W); return (FTS_W);
} }
#endif #endif
@ -846,7 +877,7 @@ fts_stat(sp, dp, p, follow)
} }
} else if (lstat(p->fts_accpath, sbp)) { } else if (lstat(p->fts_accpath, sbp)) {
p->fts_errno = errno; p->fts_errno = errno;
err: __bzero(sbp, sizeof(struct stat)); err: memset(sbp, 0, sizeof(struct stat));
return (FTS_NS); return (FTS_NS);
} }
@ -943,8 +974,9 @@ fts_alloc(sp, name, namelen)
if ((p = malloc(len)) == NULL) if ((p = malloc(len)) == NULL)
return (NULL); return (NULL);
/* Copy the name plus the trailing NULL. */ /* Copy the name and guarantee NUL termination. */
bcopy(name, p->fts_name, namelen + 1); memmove(p->fts_name, name, namelen);
p->fts_name[namelen] = '\0';
if (!ISSET(FTS_NOSTAT)) if (!ISSET(FTS_NOSTAT))
p->fts_statp = (struct stat *)ALIGN(p->fts_name + namelen + 2); p->fts_statp = (struct stat *)ALIGN(p->fts_name + namelen + 2);
@ -985,7 +1017,19 @@ fts_palloc(sp, more)
size_t more; size_t more;
{ {
sp->fts_pathlen += more + 256; sp->fts_pathlen += more + 256;
sp->fts_path = realloc(sp->fts_path, (size_t)sp->fts_pathlen); /*
* Check for possible wraparound. In an FTS, fts_pathlen is
* a signed int but in an FTSENT it is an unsigned short.
* We limit fts_pathlen to USHRT_MAX to be safe in both cases.
*/
if (sp->fts_pathlen < 0 || sp->fts_pathlen >= USHRT_MAX) {
if (sp->fts_path)
free(sp->fts_path);
sp->fts_path = NULL;
__set_errno (ENAMETOOLONG);
return (1);
}
sp->fts_path = realloc(sp->fts_path, sp->fts_pathlen);
return (sp->fts_path == NULL); return (sp->fts_path == NULL);
} }
@ -995,23 +1039,26 @@ fts_palloc(sp, more)
*/ */
static void static void
internal_function internal_function
fts_padjust(sp, addr) fts_padjust(sp, head)
FTS *sp; FTS *sp;
void *addr; FTSENT *head;
{ {
FTSENT *p; FTSENT *p;
char *addr = sp->fts_path;
#define ADJUST(p) { \ #define ADJUST(p) do { \
(p)->fts_accpath = \ if ((p)->fts_accpath != (p)->fts_name) { \
(char *)addr + ((p)->fts_accpath - (p)->fts_path); \ (p)->fts_accpath = \
(char *)addr + ((p)->fts_accpath - (p)->fts_path); \
} \
(p)->fts_path = addr; \ (p)->fts_path = addr; \
} } while (0)
/* Adjust the current set of children. */ /* Adjust the current set of children. */
for (p = sp->fts_child; p; p = p->fts_link) for (p = sp->fts_child; p; p = p->fts_link)
ADJUST(p); ADJUST(p);
/* Adjust the rest of the tree. */ /* Adjust the rest of the tree, including the current level. */
for (p = sp->fts_cur; p->fts_level >= FTS_ROOTLEVEL;) { for (p = head; p->fts_level >= FTS_ROOTLEVEL;) {
ADJUST(p); ADJUST(p);
p = p->fts_link ? p->fts_link : p->fts_parent; p = p->fts_link ? p->fts_link : p->fts_parent;
} }
@ -1027,5 +1074,44 @@ fts_maxarglen(argv)
for (max = 0; *argv; ++argv) for (max = 0; *argv; ++argv)
if ((len = strlen(*argv)) > max) if ((len = strlen(*argv)) > max)
max = len; max = len;
return (max); return (max + 1);
}
/*
* Change to dir specified by fd or p->fts_accpath without getting
* tricked by someone changing the world out from underneath us.
* Assumes p->fts_dev and p->fts_ino are filled in.
*/
static int
internal_function
fts_safe_changedir(sp, p, fd, path)
FTS *sp;
FTSENT *p;
int fd;
const char *path;
{
int ret, oerrno, newfd;
struct stat sb;
newfd = fd;
if (ISSET(FTS_NOCHDIR))
return (0);
if (fd < 0 && (newfd = __open(path, O_RDONLY, 0)) < 0)
return (-1);
if (__fstat(newfd, &sb)) {
ret = -1;
goto bail;
}
if (p->fts_dev != sb.st_dev || p->fts_ino != sb.st_ino) {
__set_errno (ENOENT); /* disinformation */
ret = -1;
goto bail;
}
ret = __fchdir(newfd);
bail:
oerrno = errno;
if (fd < 0)
(void)__close(newfd);
__set_errno (oerrno);
return (ret);
} }