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