Fix large-file macro usage in bits/statvfs.h
[kopensolaris-gnu/glibc.git] / io / ftw.c
index d7ec716..6b5e573 100644 (file)
--- a/io/ftw.c
+++ b/io/ftw.c
-/* Copyright (C) 1992, 1995, 1996, 1997 Free Software Foundation, Inc.
+/* File tree walker functions.
+   Copyright (C) 1996-2003, 2004, 2006, 2007 Free Software Foundation, Inc.
    This file is part of the GNU C Library.
-   Contributed by Ian Lance Taylor (ian@airs.com).
+   Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.
 
    The GNU C Library is free software; you can redistribute it and/or
-   modify it under the terms of the GNU Library General Public License as
-   published by the Free Software Foundation; either version 2 of the
-   License, or (at your option) any later version.
+   modify it under the terms of the GNU Lesser General Public
+   License as published by the Free Software Foundation; either
+   version 2.1 of the License, or (at your option) any later version.
 
    The GNU C Library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
-   Library General Public License for more details.
+   Lesser General Public License for more details.
 
-   You should have received a copy of the GNU Library General Public
-   License along with the GNU C Library; see the file COPYING.LIB.  If not,
-   write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
-   Boston, MA 02111-1307, USA.  */
+   You should have received a copy of the GNU Lesser General Public
+   License along with the GNU C Library; if not, write to the Free
+   Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
+   02111-1307 USA.  */
+
+#ifdef HAVE_CONFIG_H
+# include <config.h>
+#endif
+
+#if __GNUC__
+# define alloca __builtin_alloca
+#else
+# if HAVE_ALLOCA_H
+#  include <alloca.h>
+# else
+#  ifdef _AIX
+ #  pragma alloca
+#  else
+char *alloca ();
+#  endif
+# endif
+#endif
+
+#ifdef _LIBC
+# include <dirent.h>
+# define NAMLEN(dirent) _D_EXACT_NAMLEN (dirent)
+#else
+# if HAVE_DIRENT_H
+#  include <dirent.h>
+#  define NAMLEN(dirent) strlen ((dirent)->d_name)
+# else
+#  define dirent direct
+#  define NAMLEN(dirent) (dirent)->d_namlen
+#  if HAVE_SYS_NDIR_H
+#   include <sys/ndir.h>
+#  endif
+#  if HAVE_SYS_DIR_H
+#   include <sys/dir.h>
+#  endif
+#  if HAVE_NDIR_H
+#   include <ndir.h>
+#  endif
+# endif
+#endif
 
 #include <errno.h>
+#include <fcntl.h>
+#include <ftw.h>
 #include <limits.h>
+#include <search.h>
 #include <stdlib.h>
 #include <string.h>
-#include <dirent.h>
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <ftw.h>
+#include <unistd.h>
+#include <not-cancel.h>
+#if HAVE_SYS_PARAM_H || defined _LIBC
+# include <sys/param.h>
+#endif
+#ifdef _LIBC
+# include <include/sys/stat.h>
+#else
+# include <sys/stat.h>
+#endif
 
+#if ! _LIBC && !HAVE_DECL_STPCPY && !defined stpcpy
+char *stpcpy ();
+#endif
 
+#if ! _LIBC && ! defined HAVE_MEMPCPY && ! defined mempcpy
+/* Be CAREFUL that there are no side effects in N.  */
+# define mempcpy(D, S, N) ((void *) ((char *) memcpy (D, S, N) + (N)))
+#endif
+
+/* #define NDEBUG 1 */
+#include <assert.h>
+
+#ifndef _LIBC
+# undef __chdir
+# define __chdir chdir
+# undef __closedir
+# define __closedir closedir
+# undef __fchdir
+# define __fchdir fchdir
+# undef __getcwd
+# define __getcwd(P, N) xgetcwd ()
+extern char *xgetcwd (void);
+# undef __mempcpy
+# define __mempcpy mempcpy
+# undef __opendir
+# define __opendir opendir
+# undef __readdir64
+# define __readdir64 readdir
+# undef __stpcpy
+# define __stpcpy stpcpy
+# undef __tdestroy
+# define __tdestroy tdestroy
+# undef __tfind
+# define __tfind tfind
+# undef __tsearch
+# define __tsearch tsearch
+# undef internal_function
+# define internal_function /* empty */
+# undef dirent64
+# define dirent64 dirent
+# undef MAX
+# define MAX(a, b) ((a) > (b) ? (a) : (b))
+#endif
+
+/* Arrange to make lstat calls go through the wrapper function
+   on systems with an lstat function that does not dereference symlinks
+   that are specified with a trailing slash.  */
+#if ! _LIBC && ! LSTAT_FOLLOWS_SLASHED_SYMLINK
+int rpl_lstat (const char *, struct stat *);
+# undef lstat
+# define lstat(Name, Stat_buf) rpl_lstat(Name, Stat_buf)
+#endif
+
+#ifndef __set_errno
+# define __set_errno(Val) errno = (Val)
+#endif
+
+/* Support for the LFS API version.  */
+#ifndef FTW_NAME
+# define FTW_NAME ftw
+# define NFTW_NAME nftw
+# define NFTW_OLD_NAME __old_nftw
+# define NFTW_NEW_NAME __new_nftw
+# define INO_T ino_t
+# define STAT stat
+# ifdef _LIBC
+#  define LXSTAT __lxstat
+#  define XSTAT __xstat
+#  define FXSTATAT __fxstatat
+# else
+#  define LXSTAT(V,f,sb) lstat (f,sb)
+#  define XSTAT(V,f,sb) stat (f,sb)
+#  define FXSTATAT(V,d,f,sb,m) fstatat (d, f, sb, m)
+# endif
+# define FTW_FUNC_T __ftw_func_t
+# define NFTW_FUNC_T __nftw_func_t
+#endif
+
+/* We define PATH_MAX if the system does not provide a definition.
+   This does not artificially limit any operation.  PATH_MAX is simply
+   used as a guesstimate for the expected maximal path length.
+   Buffers will be enlarged if necessary.  */
 #ifndef PATH_MAX
-#define PATH_MAX 1024          /* XXX */
+# define PATH_MAX 1024
 #endif
 
+struct dir_data
+{
+  DIR *stream;
+  int streamfd;
+  char *content;
+};
 
-/* Traverse one level of a directory tree.  */
+struct known_object
+{
+  dev_t dev;
+  INO_T ino;
+};
 
-static int
-ftw_dir (DIR **dirs, int level, int descriptors, char *dir, size_t len,
-        int (*func) (const char *file, const struct stat *status, int flag))
+struct ftw_data
 {
-  int got;
-  struct dirent *entry;
+  /* Array with pointers to open directory streams.  */
+  struct dir_data **dirstreams;
+  size_t actdir;
+  size_t maxdir;
 
-  got = 0;
+  /* Buffer containing name of currently processed object.  */
+  char *dirbuf;
+  size_t dirbufsize;
 
-  __set_errno (0);
+  /* Passed as fourth argument to `nftw' callback.  The `base' member
+     tracks the content of the `dirbuf'.  */
+  struct FTW ftw;
 
-  while ((entry = readdir (dirs[level])) != NULL)
-    {
-      struct stat s;
-      int flag, retval, newlev;
-      size_t namlen;
+  /* Flags passed to `nftw' function.  0 for `ftw'.  */
+  int flags;
 
-      ++got;
+  /* Conversion array for flag values.  It is the identity mapping for
+     `nftw' calls, otherwise it maps the values to those known by
+     `ftw'.  */
+  const int *cvt_arr;
 
-      if (entry->d_name[0] == '.'
-         && (entry->d_name[1] == '\0' ||
-             (entry->d_name[1] == '.' && entry->d_name[2] == '\0')))
-       {
-         __set_errno (0);
-         continue;
-       }
+  /* Callback function.  We always use the `nftw' form.  */
+  NFTW_FUNC_T func;
 
-      namlen = _D_EXACT_NAMLEN (entry);
+  /* Device of starting point.  Needed for FTW_MOUNT.  */
+  dev_t dev;
 
-      if (namlen + len + 1 > PATH_MAX)
-       {
-#ifdef ENAMETOOLONG
-         __set_errno (ENAMETOOLONG);
-#else
-         __set_errno (ENOMEM);
-#endif
-         return -1;
-       }
+  /* Data structure for keeping fingerprints of already processed
+     object.  This is needed when not using FTW_PHYS.  */
+  void *known_objects;
+};
 
-      dir[len] = '/';
-      memcpy ((void *) (dir + len + 1), (void *) entry->d_name,
-             namlen + 1);
 
-      if (stat (dir, &s) < 0)
-       {
-         if (errno != EACCES && errno != ENOENT)
-           return -1;
-         flag = FTW_NS;
-       }
-      else if (S_ISDIR (s.st_mode))
-       {
-         newlev = (level + 1) % descriptors;
+/* Internally we use the FTW_* constants used for `nftw'.  When invoked
+   as `ftw', map each flag to the subset of values used by `ftw'.  */
+static const int nftw_arr[] =
+{
+  FTW_F, FTW_D, FTW_DNR, FTW_NS, FTW_SL, FTW_DP, FTW_SLN
+};
 
-         if (dirs[newlev] != NULL)
-           closedir (dirs[newlev]);
+static const int ftw_arr[] =
+{
+  FTW_F, FTW_D, FTW_DNR, FTW_NS, FTW_F, FTW_D, FTW_NS
+};
 
-         dirs[newlev] = opendir (dir);
-         if (dirs[newlev] != NULL)
-           flag = FTW_D;
-         else
-           {
-             if (errno != EACCES)
-               return -1;
-             flag = FTW_DNR;
-           }
-       }
-      else
-       flag = FTW_F;
 
-      retval = (*func) (dir, &s, flag);
+/* Forward declarations of local functions.  */
+static int ftw_dir (struct ftw_data *data, struct STAT *st,
+                   struct dir_data *old_dir) internal_function;
 
-      if (flag == FTW_D)
+
+static int
+object_compare (const void *p1, const void *p2)
+{
+  /* We don't need a sophisticated and useful comparison.  We are only
+     interested in equality.  However, we must be careful not to
+     accidentally compare `holes' in the structure.  */
+  const struct known_object *kp1 = p1, *kp2 = p2;
+  int cmp1;
+  cmp1 = (kp1->ino > kp2->ino) - (kp1->ino < kp2->ino);
+  if (cmp1 != 0)
+    return cmp1;
+  return (kp1->dev > kp2->dev) - (kp1->dev < kp2->dev);
+}
+
+
+static inline int
+add_object (struct ftw_data *data, struct STAT *st)
+{
+  struct known_object *newp = malloc (sizeof (struct known_object));
+  if (newp == NULL)
+    return -1;
+  newp->dev = st->st_dev;
+  newp->ino = st->st_ino;
+  return __tsearch (newp, &data->known_objects, object_compare) ? 0 : -1;
+}
+
+
+static inline int
+find_object (struct ftw_data *data, struct STAT *st)
+{
+  struct known_object obj;
+  obj.dev = st->st_dev;
+  obj.ino = st->st_ino;
+  return __tfind (&obj, &data->known_objects, object_compare) != NULL;
+}
+
+
+static inline int
+__attribute ((always_inline))
+open_dir_stream (int *dfdp, struct ftw_data *data, struct dir_data *dirp)
+{
+  int result = 0;
+
+  if (data->dirstreams[data->actdir] != NULL)
+    {
+      /* Oh, oh.  We must close this stream.  Get all remaining
+        entries and store them as a list in the `content' member of
+        the `struct dir_data' variable.  */
+      size_t bufsize = 1024;
+      char *buf = malloc (bufsize);
+
+      if (buf == NULL)
+       result = -1;
+      else
        {
-         if (retval == 0)
-           retval = ftw_dir (dirs, newlev, descriptors, dir,
-                             namlen + len + 1, func);
-         if (dirs[newlev] != NULL)
+         DIR *st = data->dirstreams[data->actdir]->stream;
+         struct dirent64 *d;
+         size_t actsize = 0;
+
+         while ((d = __readdir64 (st)) != NULL)
            {
-             int save;
+             size_t this_len = NAMLEN (d);
+             if (actsize + this_len + 2 >= bufsize)
+               {
+                 char *newp;
+                 bufsize += MAX (1024, 2 * this_len);
+                 newp = (char *) realloc (buf, bufsize);
+                 if (newp == NULL)
+                   {
+                     /* No more memory.  */
+                     int save_err = errno;
+                     free (buf);
+                     __set_errno (save_err);
+                     return -1;
+                   }
+                 buf = newp;
+               }
+
+             *((char *) __mempcpy (buf + actsize, d->d_name, this_len))
+               = '\0';
+             actsize += this_len + 1;
+           }
+
+         /* Terminate the list with an additional NUL byte.  */
+         buf[actsize++] = '\0';
 
-             save = errno;
-             closedir (dirs[newlev]);
-             __set_errno (save);
-             dirs[newlev] = NULL;
+         /* Shrink the buffer to what we actually need.  */
+         data->dirstreams[data->actdir]->content = realloc (buf, actsize);
+         if (data->dirstreams[data->actdir]->content == NULL)
+           {
+             int save_err = errno;
+             free (buf);
+             __set_errno (save_err);
+             result = -1;
+           }
+         else
+           {
+             __closedir (st);
+             data->dirstreams[data->actdir]->stream = NULL;
+             data->dirstreams[data->actdir]->streamfd = -1;
+             data->dirstreams[data->actdir] = NULL;
            }
        }
+    }
 
-      if (retval != 0)
-       return retval;
+  /* Open the new stream.  */
+  if (result == 0)
+    {
+      assert (data->dirstreams[data->actdir] == NULL);
 
-      if (dirs[level] == NULL)
+      if (dfdp != NULL && *dfdp != -1)
+       {
+         int fd = openat64_not_cancel_3 (*dfdp, data->dirbuf + data->ftw.base,
+                                         O_RDONLY | O_DIRECTORY | O_NDELAY);
+         dirp->stream = NULL;
+         if (fd != -1 && (dirp->stream = __fdopendir (fd)) == NULL)
+           close_not_cancel_no_status (fd);
+       }
+      else
        {
-         int skip;
-
-         dir[len] = '\0';
-         dirs[level] = opendir (dir);
-         if (dirs[level] == NULL)
-           return -1;
-         skip = got;
-         while (skip-- != 0)
+         const char *name;
+
+         if (data->flags & FTW_CHDIR)
            {
-             __set_errno (0);
-             if (readdir (dirs[level]) == NULL)
-               return errno == 0 ? 0 : -1;
+             name = data->dirbuf + data->ftw.base;
+             if (name[0] == '\0')
+               name = ".";
            }
+         else
+           name = data->dirbuf;
+
+         dirp->stream = __opendir (name);
        }
 
-      __set_errno (0);
+      if (dirp->stream == NULL)
+       result = -1;
+      else
+       {
+         dirp->streamfd = dirfd (dirp->stream);
+         dirp->content = NULL;
+         data->dirstreams[data->actdir] = dirp;
+
+         if (++data->actdir == data->maxdir)
+           data->actdir = 0;
+       }
     }
 
-  return errno == 0 ? 0 : -1;
+  return result;
 }
 
-/* Call a function on every element in a directory tree.  */
 
-int
-ftw (const char *dir,
-     int (*func) (const char *file, const struct stat *status, int flag),
-     int descriptors)
+static int
+internal_function
+process_entry (struct ftw_data *data, struct dir_data *dir, const char *name,
+              size_t namlen, int d_type)
 {
-  DIR **dirs;
-  size_t len;
-  char buf[PATH_MAX + 1];
-  struct stat s;
-  int flag, retval;
-  int i;
+  struct STAT st;
+  int result = 0;
+  int flag = 0;
+  size_t new_buflen;
+
+  if (name[0] == '.' && (name[1] == '\0'
+                        || (name[1] == '.' && name[2] == '\0')))
+    /* Don't process the "." and ".." entries.  */
+    return 0;
+
+  new_buflen = data->ftw.base + namlen + 2;
+  if (data->dirbufsize < new_buflen)
+    {
+      /* Enlarge the buffer.  */
+      char *newp;
 
-  if (descriptors <= 0)
-    descriptors = 1;
+      data->dirbufsize = 2 * new_buflen;
+      newp = (char *) realloc (data->dirbuf, data->dirbufsize);
+      if (newp == NULL)
+       return -1;
+      data->dirbuf = newp;
+    }
 
-  dirs = (DIR **) __alloca (descriptors * sizeof (DIR *));
-  i = descriptors;
-  while (i-- > 0)
-    dirs[i] = NULL;
+  *((char *) __mempcpy (data->dirbuf + data->ftw.base, name, namlen)) = '\0';
+
+  int statres;
+  if (dir->streamfd != -1)
+    statres = FXSTATAT (_STAT_VER, dir->streamfd, name, &st,
+                       (data->flags & FTW_PHYS) ? AT_SYMLINK_NOFOLLOW : 0);
+  else
+    {
+      if ((data->flags & FTW_CHDIR) == 0)
+       name = data->dirbuf;
+
+      statres = ((data->flags & FTW_PHYS)
+                ? LXSTAT (_STAT_VER, name, &st)
+                : XSTAT (_STAT_VER, name, &st));
+    }
 
-  if (stat (dir, &s) < 0)
+  if (statres < 0)
     {
       if (errno != EACCES && errno != ENOENT)
-       return -1;
-      flag = FTW_NS;
+       result = -1;
+      else if (data->flags & FTW_PHYS)
+       flag = FTW_NS;
+      else if (d_type == DT_LNK)
+       flag = FTW_SLN;
+      else
+       {
+         if (dir->streamfd != -1)
+           statres = FXSTATAT (_STAT_VER, dir->streamfd, name, &st,
+                               AT_SYMLINK_NOFOLLOW);
+         else
+           statres = LXSTAT (_STAT_VER, name, &st);
+         if (statres == 0 && S_ISLNK (st.st_mode))
+           flag = FTW_SLN;
+         else
+           flag = FTW_NS;
+       }
     }
-  else if (S_ISDIR (s.st_mode))
+  else
     {
-      dirs[0] = opendir (dir);
-      if (dirs[0] != NULL)
+      if (S_ISDIR (st.st_mode))
        flag = FTW_D;
+      else if (S_ISLNK (st.st_mode))
+       flag = FTW_SL;
       else
+       flag = FTW_F;
+    }
+
+  if (result == 0
+      && (flag == FTW_NS
+         || !(data->flags & FTW_MOUNT) || st.st_dev == data->dev))
+    {
+      if (flag == FTW_D)
+       {
+         if ((data->flags & FTW_PHYS)
+             || (!find_object (data, &st)
+                 /* Remember the object.  */
+                 && (result = add_object (data, &st)) == 0))
+           result = ftw_dir (data, &st, dir);
+       }
+      else
+       result = (*data->func) (data->dirbuf, &st, data->cvt_arr[flag],
+                               &data->ftw);
+    }
+
+  if ((data->flags & FTW_ACTIONRETVAL) && result == FTW_SKIP_SUBTREE)
+    result = 0;
+
+  return result;
+}
+
+
+static int
+__attribute ((noinline))
+internal_function
+ftw_dir (struct ftw_data *data, struct STAT *st, struct dir_data *old_dir)
+{
+  struct dir_data dir;
+  struct dirent64 *d;
+  int previous_base = data->ftw.base;
+  int result;
+  char *startp;
+
+  /* Open the stream for this directory.  This might require that
+     another stream has to be closed.  */
+  result = open_dir_stream (old_dir == NULL ? NULL : &old_dir->streamfd,
+                           data, &dir);
+  if (result != 0)
+    {
+      if (errno == EACCES)
+       /* We cannot read the directory.  Signal this with a special flag.  */
+       result = (*data->func) (data->dirbuf, st, FTW_DNR, &data->ftw);
+
+      return result;
+    }
+
+  /* First, report the directory (if not depth-first).  */
+  if (!(data->flags & FTW_DEPTH))
+    {
+      result = (*data->func) (data->dirbuf, st, FTW_D, &data->ftw);
+      if (result != 0)
        {
-         if (errno != EACCES)
-           return -1;
-         flag = FTW_DNR;
+         int save_err;
+fail:
+         save_err = errno;
+         __closedir (dir.stream);
+         dir.streamfd = -1;
+         __set_errno (save_err);
+
+         if (data->actdir-- == 0)
+           data->actdir = data->maxdir - 1;
+         data->dirstreams[data->actdir] = NULL;
+         return result;
        }
     }
+
+  /* If necessary, change to this directory.  */
+  if (data->flags & FTW_CHDIR)
+    {
+      if (__fchdir (dirfd (dir.stream)) < 0)
+       {
+         result = -1;
+         goto fail;
+       }
+    }
+
+  /* Next, update the `struct FTW' information.  */
+  ++data->ftw.level;
+  startp = __rawmemchr (data->dirbuf, '\0');
+  /* There always must be a directory name.  */
+  assert (startp != data->dirbuf);
+  if (startp[-1] != '/')
+    *startp++ = '/';
+  data->ftw.base = startp - data->dirbuf;
+
+  while (dir.stream != NULL && (d = __readdir64 (dir.stream)) != NULL)
+    {
+#ifdef _DIRENT_HAVE_D_TYPE
+      result = process_entry (data, &dir, d->d_name, NAMLEN (d), d->d_type);
+#else
+      result = process_entry (data, &dir, d->d_name, NAMLEN (d), DT_UNKNOWN);
+#endif
+      if (result != 0)
+       break;
+    }
+
+  if (dir.stream != NULL)
+    {
+      /* The stream is still open.  I.e., we did not need more
+        descriptors.  Simply close the stream now.  */
+      int save_err = errno;
+
+      assert (dir.content == NULL);
+
+      __closedir (dir.stream);
+      dir.streamfd = -1;
+      __set_errno (save_err);
+
+      if (data->actdir-- == 0)
+       data->actdir = data->maxdir - 1;
+      data->dirstreams[data->actdir] = NULL;
+    }
   else
-    flag = FTW_F;
+    {
+      int save_err;
+      char *runp = dir.content;
 
-  len = strlen (dir);
-  memcpy ((void *) buf, (void *) dir, len + 1);
+      while (result == 0 && *runp != '\0')
+       {
+         char *endp = strchr (runp, '\0');
+
+         // XXX Should store the d_type values as well?!
+         result = process_entry (data, &dir, runp, endp - runp, DT_UNKNOWN);
+
+         runp = endp + 1;
+       }
 
-  retval = (*func) (buf, &s, flag);
+      save_err = errno;
+      free (dir.content);
+      __set_errno (save_err);
+    }
+
+  if ((data->flags & FTW_ACTIONRETVAL) && result == FTW_SKIP_SIBLINGS)
+    result = 0;
+
+  /* Prepare the return, revert the `struct FTW' information.  */
+  data->dirbuf[data->ftw.base - 1] = '\0';
+  --data->ftw.level;
+  data->ftw.base = previous_base;
+
+  /* Finally, if we process depth-first report the directory.  */
+  if (result == 0 && (data->flags & FTW_DEPTH))
+    result = (*data->func) (data->dirbuf, st, FTW_DP, &data->ftw);
+
+  if (old_dir
+      && (data->flags & FTW_CHDIR)
+      && (result == 0
+         || ((data->flags & FTW_ACTIONRETVAL)
+             && (result != -1 && result != FTW_STOP))))
+    {
+      /* Change back to the parent directory.  */
+      int done = 0;
+      if (old_dir->stream != NULL)
+       if (__fchdir (dirfd (old_dir->stream)) == 0)
+         done = 1;
+
+      if (!done)
+       {
+         if (data->ftw.base == 1)
+           {
+             if (__chdir ("/") < 0)
+               result = -1;
+           }
+         else
+           if (__chdir ("..") < 0)
+             result = -1;
+       }
+    }
+
+  return result;
+}
+
+
+static int
+__attribute ((noinline))
+internal_function
+ftw_startup (const char *dir, int is_nftw, void *func, int descriptors,
+            int flags)
+{
+  struct ftw_data data;
+  struct STAT st;
+  int result = 0;
+  int save_err;
+  int cwdfd = -1;
+  char *cwd = NULL;
+  char *cp;
+
+  /* First make sure the parameters are reasonable.  */
+  if (dir[0] == '\0')
+    {
+      __set_errno (ENOENT);
+      return -1;
+    }
+
+  data.maxdir = descriptors < 1 ? 1 : descriptors;
+  data.actdir = 0;
+  data.dirstreams = (struct dir_data **) alloca (data.maxdir
+                                                * sizeof (struct dir_data *));
+  memset (data.dirstreams, '\0', data.maxdir * sizeof (struct dir_data *));
+
+  /* PATH_MAX is always defined when we get here.  */
+  data.dirbufsize = MAX (2 * strlen (dir), PATH_MAX);
+  data.dirbuf = (char *) malloc (data.dirbufsize);
+  if (data.dirbuf == NULL)
+    return -1;
+  cp = __stpcpy (data.dirbuf, dir);
+  /* Strip trailing slashes.  */
+  while (cp > data.dirbuf + 1 && cp[-1] == '/')
+    --cp;
+  *cp = '\0';
+
+  data.ftw.level = 0;
+
+  /* Find basename.  */
+  while (cp > data.dirbuf && cp[-1] != '/')
+    --cp;
+  data.ftw.base = cp - data.dirbuf;
+
+  data.flags = flags;
+
+  /* This assignment might seem to be strange but it is what we want.
+     The trick is that the first three arguments to the `ftw' and
+     `nftw' callback functions are equal.  Therefore we can call in
+     every case the callback using the format of the `nftw' version
+     and get the correct result since the stack layout for a function
+     call in C allows this.  */
+  data.func = (NFTW_FUNC_T) func;
+
+  /* Since we internally use the complete set of FTW_* values we need
+     to reduce the value range before calling a `ftw' callback.  */
+  data.cvt_arr = is_nftw ? nftw_arr : ftw_arr;
+
+  /* No object known so far.  */
+  data.known_objects = NULL;
+
+  /* Now go to the directory containing the initial file/directory.  */
+  if (flags & FTW_CHDIR)
+    {
+      /* We have to be able to go back to the current working
+        directory.  The best way to do this is to use a file
+        descriptor.  */
+      cwdfd = __open (".", O_RDONLY | O_DIRECTORY);
+      if (cwdfd == -1)
+       {
+         /* Try getting the directory name.  This can be needed if
+            the current directory is executable but not readable.  */
+         if (errno == EACCES)
+           /* GNU extension ahead.  */
+           cwd =  __getcwd (NULL, 0);
+
+         if (cwd == NULL)
+           goto out_fail;
+       }
+      else if (data.maxdir > 1)
+       /* Account for the file descriptor we use here.  */
+       --data.maxdir;
+
+      if (data.ftw.base > 0)
+       {
+         /* Change to the directory the file is in.  In data.dirbuf
+            we have a writable copy of the file name.  Just NUL
+            terminate it for now and change the directory.  */
+         if (data.ftw.base == 1)
+           /* I.e., the file is in the root directory.  */
+           result = __chdir ("/");
+         else
+           {
+             char ch = data.dirbuf[data.ftw.base - 1];
+             data.dirbuf[data.ftw.base - 1] = '\0';
+             result = __chdir (data.dirbuf);
+             data.dirbuf[data.ftw.base - 1] = ch;
+           }
+       }
+    }
 
-  if (flag == FTW_D)
+  /* Get stat info for start directory.  */
+  if (result == 0)
     {
-      if (retval == 0)
-       retval = ftw_dir (dirs, 0, descriptors, buf, len, func);
-      if (dirs[0] != NULL)
+      const char *name;
+
+      if (data.flags & FTW_CHDIR)
        {
-         int save;
+         name = data.dirbuf + data.ftw.base;
+         if (name[0] == '\0')
+           name = ".";
+       }
+      else
+       name = data.dirbuf;
 
-         save = errno;
-         closedir (dirs[0]);
-         __set_errno (save);
+      if (((flags & FTW_PHYS)
+          ? LXSTAT (_STAT_VER, name, &st)
+          : XSTAT (_STAT_VER, name, &st)) < 0)
+       {
+         if (!(flags & FTW_PHYS)
+             && errno == ENOENT
+             && LXSTAT (_STAT_VER, name, &st) == 0
+             && S_ISLNK (st.st_mode))
+           result = (*data.func) (data.dirbuf, &st, data.cvt_arr[FTW_SLN],
+                                  &data.ftw);
+         else
+           /* No need to call the callback since we cannot say anything
+              about the object.  */
+           result = -1;
        }
+      else
+       {
+         if (S_ISDIR (st.st_mode))
+           {
+             /* Remember the device of the initial directory in case
+                FTW_MOUNT is given.  */
+             data.dev = st.st_dev;
+
+             /* We know this directory now.  */
+             if (!(flags & FTW_PHYS))
+               result = add_object (&data, &st);
+
+             if (result == 0)
+               result = ftw_dir (&data, &st, NULL);
+           }
+         else
+           {
+             int flag = S_ISLNK (st.st_mode) ? FTW_SL : FTW_F;
+
+             result = (*data.func) (data.dirbuf, &st, data.cvt_arr[flag],
+                                    &data.ftw);
+           }
+       }
+
+      if ((flags & FTW_ACTIONRETVAL)
+         && (result == FTW_SKIP_SUBTREE || result == FTW_SKIP_SIBLINGS))
+       result = 0;
+    }
+
+  /* Return to the start directory (if necessary).  */
+  if (cwdfd != -1)
+    {
+      int save_err = errno;
+      __fchdir (cwdfd);
+      __set_errno (save_err);
+    }
+  else if (cwd != NULL)
+    {
+      int save_err = errno;
+      __chdir (cwd);
+      free (cwd);
+      __set_errno (save_err);
     }
 
-  return retval;
+  /* Free all memory.  */
+ out_fail:
+  save_err = errno;
+  __tdestroy (data.known_objects, free);
+  free (data.dirbuf);
+  __set_errno (save_err);
+
+  return result;
 }
+
+
+
+/* Entry points.  */
+
+int
+FTW_NAME (path, func, descriptors)
+     const char *path;
+     FTW_FUNC_T func;
+     int descriptors;
+{
+  return ftw_startup (path, 0, func, descriptors, 0);
+}
+
+#ifndef _LIBC
+int
+NFTW_NAME (path, func, descriptors, flags)
+     const char *path;
+     NFTW_FUNC_T func;
+     int descriptors;
+     int flags;
+{
+  return ftw_startup (path, 1, func, descriptors, flags);
+}
+#else
+
+# include <shlib-compat.h>
+
+int NFTW_NEW_NAME (const char *, NFTW_FUNC_T, int, int);
+
+int
+NFTW_NEW_NAME (path, func, descriptors, flags)
+     const char *path;
+     NFTW_FUNC_T func;
+     int descriptors;
+     int flags;
+{
+  if (flags
+      & ~(FTW_PHYS | FTW_MOUNT | FTW_CHDIR | FTW_DEPTH | FTW_ACTIONRETVAL))
+    {
+      __set_errno (EINVAL);
+      return -1;
+    }
+  return ftw_startup (path, 1, func, descriptors, flags);
+}
+
+versioned_symbol (libc, NFTW_NEW_NAME, NFTW_NAME, GLIBC_2_3_3);
+
+# if SHLIB_COMPAT(libc, GLIBC_2_1, GLIBC_2_3_3)
+
+/* Older nftw* version just ignored all unknown flags.  */
+
+int NFTW_OLD_NAME (const char *, NFTW_FUNC_T, int, int);
+
+int
+attribute_compat_text_section
+NFTW_OLD_NAME (path, func, descriptors, flags)
+     const char *path;
+     NFTW_FUNC_T func;
+     int descriptors;
+     int flags;
+{
+  flags &= (FTW_PHYS | FTW_MOUNT | FTW_CHDIR | FTW_DEPTH);
+  return ftw_startup (path, 1, func, descriptors, flags);
+}
+
+compat_symbol (libc, NFTW_OLD_NAME, NFTW_NAME, GLIBC_2_1);
+# endif
+#endif