]> git.ipfire.org Git - thirdparty/glibc.git/blame - io/ftw.c
Update copyright dates with scripts/update-copyrights.
[thirdparty/glibc.git] / io / ftw.c
CommitLineData
76b87c03 1/* File tree walker functions.
bfff8b1b 2 Copyright (C) 1996-2017 Free Software Foundation, Inc.
47707456 3 This file is part of the GNU C Library.
76b87c03 4 Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.
28f540f4 5
47707456 6 The GNU C Library is free software; you can redistribute it and/or
41bdb6e2
AJ
7 modify it under the terms of the GNU Lesser General Public
8 License as published by the Free Software Foundation; either
9 version 2.1 of the License, or (at your option) any later version.
28f540f4 10
47707456
UD
11 The GNU C Library is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
41bdb6e2 14 Lesser General Public License for more details.
28f540f4 15
41bdb6e2 16 You should have received a copy of the GNU Lesser General Public
59ba27a6
PE
17 License along with the GNU C Library; if not, see
18 <http://www.gnu.org/licenses/>. */
28f540f4 19
aff4519d
UD
20#ifdef HAVE_CONFIG_H
21# include <config.h>
22#endif
23
ae9ecd08
UD
24#if __GNUC__
25# define alloca __builtin_alloca
26#else
27# if HAVE_ALLOCA_H
28# include <alloca.h>
29# else
30# ifdef _AIX
31 # pragma alloca
32# else
33char *alloca ();
34# endif
35# endif
36#endif
37
becac6c5 38#ifdef _LIBC
ae9ecd08
UD
39# include <dirent.h>
40# define NAMLEN(dirent) _D_EXACT_NAMLEN (dirent)
41#else
42# if HAVE_DIRENT_H
43# include <dirent.h>
44# define NAMLEN(dirent) strlen ((dirent)->d_name)
45# else
46# define dirent direct
47# define NAMLEN(dirent) (dirent)->d_namlen
48# if HAVE_SYS_NDIR_H
49# include <sys/ndir.h>
50# endif
51# if HAVE_SYS_DIR_H
52# include <sys/dir.h>
53# endif
54# if HAVE_NDIR_H
55# include <ndir.h>
56# endif
57# endif
58#endif
59
28f540f4 60#include <errno.h>
becac6c5 61#include <fcntl.h>
76b87c03 62#include <ftw.h>
5049f197 63#include <limits.h>
76b87c03 64#include <search.h>
28f540f4
RM
65#include <stdlib.h>
66#include <string.h>
76b87c03 67#include <unistd.h>
d369ad76 68#include <not-cancel.h>
12f2254b 69#include <sys/param.h>
aff4519d
UD
70#ifdef _LIBC
71# include <include/sys/stat.h>
72#else
73# include <sys/stat.h>
74#endif
28f540f4 75
ae9ecd08
UD
76#if ! _LIBC && !HAVE_DECL_STPCPY && !defined stpcpy
77char *stpcpy ();
78#endif
79
80#if ! _LIBC && ! defined HAVE_MEMPCPY && ! defined mempcpy
81/* Be CAREFUL that there are no side effects in N. */
82# define mempcpy(D, S, N) ((void *) ((char *) memcpy (D, S, N) + (N)))
83#endif
84
76b87c03
UD
85/* #define NDEBUG 1 */
86#include <assert.h>
28f540f4 87
aff4519d
UD
88#ifndef _LIBC
89# undef __chdir
90# define __chdir chdir
91# undef __closedir
92# define __closedir closedir
93# undef __fchdir
94# define __fchdir fchdir
95# undef __getcwd
ae9ecd08
UD
96# define __getcwd(P, N) xgetcwd ()
97extern char *xgetcwd (void);
98# undef __mempcpy
99# define __mempcpy mempcpy
aff4519d
UD
100# undef __opendir
101# define __opendir opendir
102# undef __readdir64
103# define __readdir64 readdir
ae9ecd08
UD
104# undef __stpcpy
105# define __stpcpy stpcpy
aff4519d
UD
106# undef __tdestroy
107# define __tdestroy tdestroy
108# undef __tfind
109# define __tfind tfind
110# undef __tsearch
111# define __tsearch tsearch
112# undef internal_function
113# define internal_function /* empty */
114# undef dirent64
115# define dirent64 dirent
116# undef MAX
117# define MAX(a, b) ((a) > (b) ? (a) : (b))
118#endif
119
ae9ecd08
UD
120/* Arrange to make lstat calls go through the wrapper function
121 on systems with an lstat function that does not dereference symlinks
122 that are specified with a trailing slash. */
123#if ! _LIBC && ! LSTAT_FOLLOWS_SLASHED_SYMLINK
124int rpl_lstat (const char *, struct stat *);
125# undef lstat
126# define lstat(Name, Stat_buf) rpl_lstat(Name, Stat_buf)
127#endif
128
aff4519d
UD
129#ifndef __set_errno
130# define __set_errno(Val) errno = (Val)
131#endif
132
dfd2257a
UD
133/* Support for the LFS API version. */
134#ifndef FTW_NAME
135# define FTW_NAME ftw
136# define NFTW_NAME nftw
ca10f338
UD
137# define NFTW_OLD_NAME __old_nftw
138# define NFTW_NEW_NAME __new_nftw
dfd2257a
UD
139# define INO_T ino_t
140# define STAT stat
ae9ecd08
UD
141# ifdef _LIBC
142# define LXSTAT __lxstat
143# define XSTAT __xstat
d369ad76 144# define FXSTATAT __fxstatat
ae9ecd08
UD
145# else
146# define LXSTAT(V,f,sb) lstat (f,sb)
147# define XSTAT(V,f,sb) stat (f,sb)
d369ad76 148# define FXSTATAT(V,d,f,sb,m) fstatat (d, f, sb, m)
ae9ecd08 149# endif
dfd2257a
UD
150# define FTW_FUNC_T __ftw_func_t
151# define NFTW_FUNC_T __nftw_func_t
152#endif
28f540f4 153
5049f197
UD
154/* We define PATH_MAX if the system does not provide a definition.
155 This does not artificially limit any operation. PATH_MAX is simply
156 used as a guesstimate for the expected maximal path length.
157 Buffers will be enlarged if necessary. */
158#ifndef PATH_MAX
159# define PATH_MAX 1024
160#endif
161
76b87c03
UD
162struct dir_data
163{
164 DIR *stream;
d369ad76 165 int streamfd;
76b87c03
UD
166 char *content;
167};
28f540f4 168
d951286f
UD
169struct known_object
170{
171 dev_t dev;
dfd2257a 172 INO_T ino;
d951286f
UD
173};
174
76b87c03 175struct ftw_data
28f540f4 176{
d951286f 177 /* Array with pointers to open directory streams. */
76b87c03
UD
178 struct dir_data **dirstreams;
179 size_t actdir;
180 size_t maxdir;
28f540f4 181
d951286f 182 /* Buffer containing name of currently processed object. */
76b87c03
UD
183 char *dirbuf;
184 size_t dirbufsize;
d951286f
UD
185
186 /* Passed as fourth argument to `nftw' callback. The `base' member
187 tracks the content of the `dirbuf'. */
76b87c03 188 struct FTW ftw;
28f540f4 189
d951286f 190 /* Flags passed to `nftw' function. 0 for `ftw'. */
76b87c03 191 int flags;
28f540f4 192
d951286f 193 /* Conversion array for flag values. It is the identity mapping for
ae9ecd08 194 `nftw' calls, otherwise it maps the values to those known by
d951286f 195 `ftw'. */
390500b1 196 const int *cvt_arr;
28f540f4 197
d951286f 198 /* Callback function. We always use the `nftw' form. */
dfd2257a 199 NFTW_FUNC_T func;
28f540f4 200
d951286f 201 /* Device of starting point. Needed for FTW_MOUNT. */
76b87c03 202 dev_t dev;
d951286f
UD
203
204 /* Data structure for keeping fingerprints of already processed
205 object. This is needed when not using FTW_PHYS. */
206 void *known_objects;
76b87c03 207};
28f540f4 208
92777700 209
ae9ecd08
UD
210/* Internally we use the FTW_* constants used for `nftw'. When invoked
211 as `ftw', map each flag to the subset of values used by `ftw'. */
390500b1 212static const int nftw_arr[] =
76b87c03
UD
213{
214 FTW_F, FTW_D, FTW_DNR, FTW_NS, FTW_SL, FTW_DP, FTW_SLN
215};
28f540f4 216
390500b1 217static const int ftw_arr[] =
76b87c03
UD
218{
219 FTW_F, FTW_D, FTW_DNR, FTW_NS, FTW_F, FTW_D, FTW_NS
220};
28f540f4 221
76b87c03
UD
222
223/* Forward declarations of local functions. */
ca10f338
UD
224static int ftw_dir (struct ftw_data *data, struct STAT *st,
225 struct dir_data *old_dir) internal_function;
d951286f
UD
226
227
228static int
229object_compare (const void *p1, const void *p2)
230{
231 /* We don't need a sophisticated and useful comparison. We are only
0413b54c
UD
232 interested in equality. However, we must be careful not to
233 accidentally compare `holes' in the structure. */
234 const struct known_object *kp1 = p1, *kp2 = p2;
235 int cmp1;
78e88510 236 cmp1 = (kp1->ino > kp2->ino) - (kp1->ino < kp2->ino);
0413b54c
UD
237 if (cmp1 != 0)
238 return cmp1;
78e88510 239 return (kp1->dev > kp2->dev) - (kp1->dev < kp2->dev);
d951286f
UD
240}
241
242
300ea0ad 243static int
dfd2257a 244add_object (struct ftw_data *data, struct STAT *st)
d951286f
UD
245{
246 struct known_object *newp = malloc (sizeof (struct known_object));
247 if (newp == NULL)
248 return -1;
249 newp->dev = st->st_dev;
250 newp->ino = st->st_ino;
251 return __tsearch (newp, &data->known_objects, object_compare) ? 0 : -1;
252}
253
254
255static inline int
dfd2257a 256find_object (struct ftw_data *data, struct STAT *st)
d951286f 257{
ae9ecd08
UD
258 struct known_object obj;
259 obj.dev = st->st_dev;
260 obj.ino = st->st_ino;
d951286f
UD
261 return __tfind (&obj, &data->known_objects, object_compare) != NULL;
262}
76b87c03
UD
263
264
265static inline int
dd9423a6 266__attribute ((always_inline))
d369ad76 267open_dir_stream (int *dfdp, struct ftw_data *data, struct dir_data *dirp)
76b87c03
UD
268{
269 int result = 0;
270
271 if (data->dirstreams[data->actdir] != NULL)
272 {
273 /* Oh, oh. We must close this stream. Get all remaining
274 entries and store them as a list in the `content' member of
275 the `struct dir_data' variable. */
276 size_t bufsize = 1024;
277 char *buf = malloc (bufsize);
278
279 if (buf == NULL)
280 result = -1;
281 else
28f540f4 282 {
76b87c03 283 DIR *st = data->dirstreams[data->actdir]->stream;
2958e6cc 284 struct dirent64 *d;
76b87c03
UD
285 size_t actsize = 0;
286
2958e6cc 287 while ((d = __readdir64 (st)) != NULL)
76b87c03 288 {
ae9ecd08 289 size_t this_len = NAMLEN (d);
76b87c03
UD
290 if (actsize + this_len + 2 >= bufsize)
291 {
292 char *newp;
293 bufsize += MAX (1024, 2 * this_len);
a5ce5fcf 294 newp = (char *) realloc (buf, bufsize);
76b87c03
UD
295 if (newp == NULL)
296 {
297 /* No more memory. */
298 int save_err = errno;
299 free (buf);
300 __set_errno (save_err);
400cc70a 301 return -1;
76b87c03
UD
302 }
303 buf = newp;
304 }
305
86187531
UD
306 *((char *) __mempcpy (buf + actsize, d->d_name, this_len))
307 = '\0';
308 actsize += this_len + 1;
76b87c03 309 }
28f540f4 310
76b87c03
UD
311 /* Terminate the list with an additional NUL byte. */
312 buf[actsize++] = '\0';
28f540f4 313
76b87c03
UD
314 /* Shrink the buffer to what we actually need. */
315 data->dirstreams[data->actdir]->content = realloc (buf, actsize);
316 if (data->dirstreams[data->actdir]->content == NULL)
317 {
318 int save_err = errno;
319 free (buf);
320 __set_errno (save_err);
321 result = -1;
322 }
28f540f4
RM
323 else
324 {
50304ef0 325 __closedir (st);
76b87c03 326 data->dirstreams[data->actdir]->stream = NULL;
d369ad76 327 data->dirstreams[data->actdir]->streamfd = -1;
76b87c03 328 data->dirstreams[data->actdir] = NULL;
28f540f4
RM
329 }
330 }
76b87c03
UD
331 }
332
333 /* Open the new stream. */
334 if (result == 0)
335 {
336 assert (data->dirstreams[data->actdir] == NULL);
337
d369ad76
UD
338 if (dfdp != NULL && *dfdp != -1)
339 {
340 int fd = openat64_not_cancel_3 (*dfdp, data->dirbuf + data->ftw.base,
341 O_RDONLY | O_DIRECTORY | O_NDELAY);
342 dirp->stream = NULL;
343 if (fd != -1 && (dirp->stream = __fdopendir (fd)) == NULL)
344 close_not_cancel_no_status (fd);
345 }
346 else
347 {
63a2f305
UD
348 const char *name;
349
350 if (data->flags & FTW_CHDIR)
351 {
352 name = data->dirbuf + data->ftw.base;
353 if (name[0] == '\0')
354 name = ".";
355 }
356 else
357 name = data->dirbuf;
358
d369ad76
UD
359 dirp->stream = __opendir (name);
360 }
361
76b87c03
UD
362 if (dirp->stream == NULL)
363 result = -1;
28f540f4 364 else
76b87c03 365 {
d369ad76 366 dirp->streamfd = dirfd (dirp->stream);
76b87c03
UD
367 dirp->content = NULL;
368 data->dirstreams[data->actdir] = dirp;
369
370 if (++data->actdir == data->maxdir)
371 data->actdir = 0;
372 }
373 }
374
375 return result;
376}
377
378
dd9423a6
UD
379static int
380internal_function
76b87c03 381process_entry (struct ftw_data *data, struct dir_data *dir, const char *name,
d369ad76 382 size_t namlen, int d_type)
76b87c03 383{
dfd2257a 384 struct STAT st;
76b87c03 385 int result = 0;
256846bb 386 int flag = 0;
5049f197 387 size_t new_buflen;
76b87c03
UD
388
389 if (name[0] == '.' && (name[1] == '\0'
390 || (name[1] == '.' && name[2] == '\0')))
391 /* Don't process the "." and ".." entries. */
392 return 0;
393
5049f197
UD
394 new_buflen = data->ftw.base + namlen + 2;
395 if (data->dirbufsize < new_buflen)
76b87c03
UD
396 {
397 /* Enlarge the buffer. */
398 char *newp;
399
5049f197 400 data->dirbufsize = 2 * new_buflen;
aff4519d 401 newp = (char *) realloc (data->dirbuf, data->dirbufsize);
76b87c03
UD
402 if (newp == NULL)
403 return -1;
404 data->dirbuf = newp;
405 }
28f540f4 406
86187531 407 *((char *) __mempcpy (data->dirbuf + data->ftw.base, name, namlen)) = '\0';
28f540f4 408
d369ad76
UD
409 int statres;
410 if (dir->streamfd != -1)
411 statres = FXSTATAT (_STAT_VER, dir->streamfd, name, &st,
412 (data->flags & FTW_PHYS) ? AT_SYMLINK_NOFOLLOW : 0);
413 else
414 {
415 if ((data->flags & FTW_CHDIR) == 0)
416 name = data->dirbuf;
417
418 statres = ((data->flags & FTW_PHYS)
419 ? LXSTAT (_STAT_VER, name, &st)
420 : XSTAT (_STAT_VER, name, &st));
421 }
aff4519d 422
d369ad76 423 if (statres < 0)
76b87c03
UD
424 {
425 if (errno != EACCES && errno != ENOENT)
426 result = -1;
bb549088
UD
427 else if (data->flags & FTW_PHYS)
428 flag = FTW_NS;
429 else if (d_type == DT_LNK)
76b87c03
UD
430 flag = FTW_SLN;
431 else
bb549088
UD
432 {
433 if (dir->streamfd != -1)
434 statres = FXSTATAT (_STAT_VER, dir->streamfd, name, &st,
435 AT_SYMLINK_NOFOLLOW);
436 else
437 statres = LXSTAT (_STAT_VER, name, &st);
438 if (statres == 0 && S_ISLNK (st.st_mode))
439 flag = FTW_SLN;
440 else
441 flag = FTW_NS;
442 }
76b87c03
UD
443 }
444 else
445 {
d951286f 446 if (S_ISDIR (st.st_mode))
76b87c03 447 flag = FTW_D;
d951286f 448 else if (S_ISLNK (st.st_mode))
76b87c03
UD
449 flag = FTW_SL;
450 else
451 flag = FTW_F;
452 }
453
454 if (result == 0
b13927da
UD
455 && (flag == FTW_NS
456 || !(data->flags & FTW_MOUNT) || st.st_dev == data->dev))
76b87c03 457 {
eb7c2001 458 if (flag == FTW_D)
28f540f4 459 {
eb7c2001
UD
460 if ((data->flags & FTW_PHYS)
461 || (!find_object (data, &st)
462 /* Remember the object. */
463 && (result = add_object (data, &st)) == 0))
ca10f338 464 result = ftw_dir (data, &st, dir);
28f540f4 465 }
eb7c2001
UD
466 else
467 result = (*data->func) (data->dirbuf, &st, data->cvt_arr[flag],
468 &data->ftw);
76b87c03
UD
469 }
470
ca10f338
UD
471 if ((data->flags & FTW_ACTIONRETVAL) && result == FTW_SKIP_SUBTREE)
472 result = 0;
473
76b87c03
UD
474 return result;
475}
476
477
478static int
ca10f338 479__attribute ((noinline))
dfd2257a 480internal_function
ca10f338 481ftw_dir (struct ftw_data *data, struct STAT *st, struct dir_data *old_dir)
76b87c03
UD
482{
483 struct dir_data dir;
2958e6cc 484 struct dirent64 *d;
76b87c03 485 int previous_base = data->ftw.base;
d951286f 486 int result;
76b87c03
UD
487 char *startp;
488
d951286f
UD
489 /* Open the stream for this directory. This might require that
490 another stream has to be closed. */
d369ad76
UD
491 result = open_dir_stream (old_dir == NULL ? NULL : &old_dir->streamfd,
492 data, &dir);
d951286f
UD
493 if (result != 0)
494 {
495 if (errno == EACCES)
496 /* We cannot read the directory. Signal this with a special flag. */
497 result = (*data->func) (data->dirbuf, st, FTW_DNR, &data->ftw);
498
499 return result;
500 }
501
76b87c03
UD
502 /* First, report the directory (if not depth-first). */
503 if (!(data->flags & FTW_DEPTH))
504 {
d951286f 505 result = (*data->func) (data->dirbuf, st, FTW_D, &data->ftw);
76b87c03 506 if (result != 0)
76b87c03 507 {
93787845
UD
508 int save_err;
509fail:
510 save_err = errno;
50304ef0 511 __closedir (dir.stream);
d369ad76 512 dir.streamfd = -1;
76b87c03
UD
513 __set_errno (save_err);
514
515 if (data->actdir-- == 0)
516 data->actdir = data->maxdir - 1;
517 data->dirstreams[data->actdir] = NULL;
93787845
UD
518 return result;
519 }
520 }
76b87c03 521
93787845
UD
522 /* If necessary, change to this directory. */
523 if (data->flags & FTW_CHDIR)
524 {
525 if (__fchdir (dirfd (dir.stream)) < 0)
526 {
527 result = -1;
528 goto fail;
76b87c03 529 }
28f540f4
RM
530 }
531
76b87c03
UD
532 /* Next, update the `struct FTW' information. */
533 ++data->ftw.level;
e7c8359e 534 startp = __rawmemchr (data->dirbuf, '\0');
25f227b9
UD
535 /* There always must be a directory name. */
536 assert (startp != data->dirbuf);
21a568e2 537 if (startp[-1] != '/')
25f227b9 538 *startp++ = '/';
76b87c03 539 data->ftw.base = startp - data->dirbuf;
28f540f4 540
2958e6cc 541 while (dir.stream != NULL && (d = __readdir64 (dir.stream)) != NULL)
76b87c03 542 {
bea9b193
RM
543 int d_type = DT_UNKNOWN;
544#ifdef _DIRENT_HAVE_D_TYPE
545 d_type = d->d_type;
546#endif
547 result = process_entry (data, &dir, d->d_name, NAMLEN (d), d_type);
76b87c03
UD
548 if (result != 0)
549 break;
550 }
28f540f4 551
76b87c03 552 if (dir.stream != NULL)
28f540f4 553 {
76b87c03
UD
554 /* The stream is still open. I.e., we did not need more
555 descriptors. Simply close the stream now. */
556 int save_err = errno;
557
558 assert (dir.content == NULL);
559
50304ef0 560 __closedir (dir.stream);
d369ad76 561 dir.streamfd = -1;
76b87c03
UD
562 __set_errno (save_err);
563
564 if (data->actdir-- == 0)
565 data->actdir = data->maxdir - 1;
566 data->dirstreams[data->actdir] = NULL;
28f540f4 567 }
76b87c03 568 else
28f540f4 569 {
76b87c03
UD
570 int save_err;
571 char *runp = dir.content;
572
3d73829c 573 while (result == 0 && *runp != '\0')
28f540f4 574 {
76b87c03
UD
575 char *endp = strchr (runp, '\0');
576
d369ad76
UD
577 // XXX Should store the d_type values as well?!
578 result = process_entry (data, &dir, runp, endp - runp, DT_UNKNOWN);
76b87c03
UD
579
580 runp = endp + 1;
28f540f4 581 }
76b87c03
UD
582
583 save_err = errno;
584 free (dir.content);
585 __set_errno (save_err);
28f540f4 586 }
28f540f4 587
ca10f338
UD
588 if ((data->flags & FTW_ACTIONRETVAL) && result == FTW_SKIP_SIBLINGS)
589 result = 0;
590
76b87c03 591 /* Prepare the return, revert the `struct FTW' information. */
d951286f 592 data->dirbuf[data->ftw.base - 1] = '\0';
76b87c03
UD
593 --data->ftw.level;
594 data->ftw.base = previous_base;
595
596 /* Finally, if we process depth-first report the directory. */
597 if (result == 0 && (data->flags & FTW_DEPTH))
d951286f 598 result = (*data->func) (data->dirbuf, st, FTW_DP, &data->ftw);
28f540f4 599
ca10f338
UD
600 if (old_dir
601 && (data->flags & FTW_CHDIR)
602 && (result == 0
603 || ((data->flags & FTW_ACTIONRETVAL)
604 && (result != -1 && result != FTW_STOP))))
605 {
606 /* Change back to the parent directory. */
607 int done = 0;
608 if (old_dir->stream != NULL)
609 if (__fchdir (dirfd (old_dir->stream)) == 0)
610 done = 1;
611
612 if (!done)
613 {
614 if (data->ftw.base == 1)
615 {
616 if (__chdir ("/") < 0)
617 result = -1;
618 }
619 else
620 if (__chdir ("..") < 0)
621 result = -1;
622 }
623 }
624
76b87c03
UD
625 return result;
626}
627
628
629static int
ca10f338 630__attribute ((noinline))
dfd2257a 631internal_function
76b87c03
UD
632ftw_startup (const char *dir, int is_nftw, void *func, int descriptors,
633 int flags)
634{
635 struct ftw_data data;
dfd2257a 636 struct STAT st;
76b87c03
UD
637 int result = 0;
638 int save_err;
becac6c5 639 int cwdfd = -1;
b576fca1 640 char *cwd = NULL;
76b87c03
UD
641 char *cp;
642
643 /* First make sure the parameters are reasonable. */
644 if (dir[0] == '\0')
645 {
a7f775a5 646 __set_errno (ENOENT);
76b87c03
UD
647 return -1;
648 }
28f540f4 649
76b87c03
UD
650 data.maxdir = descriptors < 1 ? 1 : descriptors;
651 data.actdir = 0;
652 data.dirstreams = (struct dir_data **) alloca (data.maxdir
653 * sizeof (struct dir_data *));
654 memset (data.dirstreams, '\0', data.maxdir * sizeof (struct dir_data *));
655
5049f197 656 /* PATH_MAX is always defined when we get here. */
76b87c03
UD
657 data.dirbufsize = MAX (2 * strlen (dir), PATH_MAX);
658 data.dirbuf = (char *) malloc (data.dirbufsize);
659 if (data.dirbuf == NULL)
660 return -1;
d951286f 661 cp = __stpcpy (data.dirbuf, dir);
76b87c03
UD
662 /* Strip trailing slashes. */
663 while (cp > data.dirbuf + 1 && cp[-1] == '/')
664 --cp;
665 *cp = '\0';
666
667 data.ftw.level = 0;
668
669 /* Find basename. */
670 while (cp > data.dirbuf && cp[-1] != '/')
671 --cp;
672 data.ftw.base = cp - data.dirbuf;
673
674 data.flags = flags;
675
676 /* This assignment might seem to be strange but it is what we want.
677 The trick is that the first three arguments to the `ftw' and
678 `nftw' callback functions are equal. Therefore we can call in
679 every case the callback using the format of the `nftw' version
680 and get the correct result since the stack layout for a function
681 call in C allows this. */
dfd2257a 682 data.func = (NFTW_FUNC_T) func;
76b87c03
UD
683
684 /* Since we internally use the complete set of FTW_* values we need
685 to reduce the value range before calling a `ftw' callback. */
686 data.cvt_arr = is_nftw ? nftw_arr : ftw_arr;
687
d951286f
UD
688 /* No object known so far. */
689 data.known_objects = NULL;
690
76b87c03 691 /* Now go to the directory containing the initial file/directory. */
34c86f42 692 if (flags & FTW_CHDIR)
28f540f4 693 {
becac6c5
UD
694 /* We have to be able to go back to the current working
695 directory. The best way to do this is to use a file
696 descriptor. */
697 cwdfd = __open (".", O_RDONLY | O_DIRECTORY);
698 if (cwdfd == -1)
28f540f4 699 {
b576fca1
UD
700 /* Try getting the directory name. This can be needed if
701 the current directory is executable but not readable. */
702 if (errno == EACCES)
703 /* GNU extension ahead. */
704 cwd = __getcwd (NULL, 0);
705
706 if (cwd == NULL)
707 goto out_fail;
708 }
709 else if (data.maxdir > 1)
710 /* Account for the file descriptor we use here. */
711 --data.maxdir;
becac6c5 712
b576fca1
UD
713 if (data.ftw.base > 0)
714 {
715 /* Change to the directory the file is in. In data.dirbuf
716 we have a writable copy of the file name. Just NUL
717 terminate it for now and change the directory. */
718 if (data.ftw.base == 1)
719 /* I.e., the file is in the root directory. */
720 result = __chdir ("/");
721 else
76b87c03 722 {
b576fca1
UD
723 char ch = data.dirbuf[data.ftw.base - 1];
724 data.dirbuf[data.ftw.base - 1] = '\0';
725 result = __chdir (data.dirbuf);
726 data.dirbuf[data.ftw.base - 1] = ch;
76b87c03 727 }
28f540f4
RM
728 }
729 }
730
76b87c03
UD
731 /* Get stat info for start directory. */
732 if (result == 0)
6e4c40ba 733 {
63a2f305
UD
734 const char *name;
735
736 if (data.flags & FTW_CHDIR)
737 {
738 name = data.dirbuf + data.ftw.base;
739 if (name[0] == '\0')
740 name = ".";
741 }
742 else
743 name = data.dirbuf;
b2608c22 744
6e4c40ba 745 if (((flags & FTW_PHYS)
b2608c22
UD
746 ? LXSTAT (_STAT_VER, name, &st)
747 : XSTAT (_STAT_VER, name, &st)) < 0)
6e4c40ba 748 {
0e24e73d
UD
749 if (!(flags & FTW_PHYS)
750 && errno == ENOENT
40212ce0 751 && LXSTAT (_STAT_VER, name, &st) == 0
0e24e73d 752 && S_ISLNK (st.st_mode))
6e4c40ba 753 result = (*data.func) (data.dirbuf, &st, data.cvt_arr[FTW_SLN],
76b87c03 754 &data.ftw);
6e4c40ba
UD
755 else
756 /* No need to call the callback since we cannot say anything
757 about the object. */
758 result = -1;
759 }
760 else
761 {
762 if (S_ISDIR (st.st_mode))
763 {
764 /* Remember the device of the initial directory in case
765 FTW_MOUNT is given. */
766 data.dev = st.st_dev;
767
768 /* We know this directory now. */
769 if (!(flags & FTW_PHYS))
770 result = add_object (&data, &st);
771
772 if (result == 0)
ca10f338 773 result = ftw_dir (&data, &st, NULL);
6e4c40ba
UD
774 }
775 else
776 {
777 int flag = S_ISLNK (st.st_mode) ? FTW_SL : FTW_F;
778
779 result = (*data.func) (data.dirbuf, &st, data.cvt_arr[flag],
780 &data.ftw);
781 }
782 }
ca10f338
UD
783
784 if ((flags & FTW_ACTIONRETVAL)
785 && (result == FTW_SKIP_SUBTREE || result == FTW_SKIP_SIBLINGS))
786 result = 0;
6e4c40ba 787 }
76b87c03
UD
788
789 /* Return to the start directory (if necessary). */
becac6c5 790 if (cwdfd != -1)
76b87c03
UD
791 {
792 int save_err = errno;
becac6c5 793 __fchdir (cwdfd);
247fdf2e 794 close_not_cancel_no_status (cwdfd);
76b87c03
UD
795 __set_errno (save_err);
796 }
b576fca1
UD
797 else if (cwd != NULL)
798 {
799 int save_err = errno;
800 __chdir (cwd);
801 free (cwd);
802 __set_errno (save_err);
803 }
76b87c03
UD
804
805 /* Free all memory. */
b576fca1 806 out_fail:
76b87c03 807 save_err = errno;
d951286f 808 __tdestroy (data.known_objects, free);
76b87c03
UD
809 free (data.dirbuf);
810 __set_errno (save_err);
811
812 return result;
813}
814
815
816
817/* Entry points. */
818
819int
9dd346ff 820FTW_NAME (const char *path, FTW_FUNC_T func, int descriptors)
76b87c03
UD
821{
822 return ftw_startup (path, 0, func, descriptors, 0);
823}
824
ca10f338 825#ifndef _LIBC
76b87c03 826int
9dd346ff 827NFTW_NAME (const char *path, NFTW_FUNC_T func, int descriptors, int flags)
76b87c03
UD
828{
829 return ftw_startup (path, 1, func, descriptors, flags);
28f540f4 830}
ca10f338
UD
831#else
832
b576fca1 833# include <shlib-compat.h>
ca10f338 834
3c0fb574
UD
835int NFTW_NEW_NAME (const char *, NFTW_FUNC_T, int, int);
836
ca10f338 837int
9dd346ff 838NFTW_NEW_NAME (const char *path, NFTW_FUNC_T func, int descriptors, int flags)
ca10f338
UD
839{
840 if (flags
841 & ~(FTW_PHYS | FTW_MOUNT | FTW_CHDIR | FTW_DEPTH | FTW_ACTIONRETVAL))
842 {
843 __set_errno (EINVAL);
844 return -1;
845 }
846 return ftw_startup (path, 1, func, descriptors, flags);
847}
848
849versioned_symbol (libc, NFTW_NEW_NAME, NFTW_NAME, GLIBC_2_3_3);
850
b576fca1 851# if SHLIB_COMPAT(libc, GLIBC_2_1, GLIBC_2_3_3)
ca10f338
UD
852
853/* Older nftw* version just ignored all unknown flags. */
854
3c0fb574
UD
855int NFTW_OLD_NAME (const char *, NFTW_FUNC_T, int, int);
856
ca10f338 857int
4a381a81 858attribute_compat_text_section
9dd346ff 859NFTW_OLD_NAME (const char *path, NFTW_FUNC_T func, int descriptors, int flags)
ca10f338
UD
860{
861 flags &= (FTW_PHYS | FTW_MOUNT | FTW_CHDIR | FTW_DEPTH);
862 return ftw_startup (path, 1, func, descriptors, flags);
863}
864
865compat_symbol (libc, NFTW_OLD_NAME, NFTW_NAME, GLIBC_2_1);
b576fca1 866# endif
ca10f338 867#endif