Blame io/ftw.c

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