xref: /freebsd/lib/libc/gen/nftw.c (revision 4d767dfea46dae56082645e4a2939b04323e3553)
1e1bcce4fSTim J. Robbins /*	$OpenBSD: nftw.c,v 1.4 2004/07/07 16:05:23 millert Exp $	*/
2e1bcce4fSTim J. Robbins 
3e1bcce4fSTim J. Robbins /*
4e1bcce4fSTim J. Robbins  * Copyright (c) 2003, 2004 Todd C. Miller <Todd.Miller@courtesan.com>
5e1bcce4fSTim J. Robbins  *
6e1bcce4fSTim J. Robbins  * Permission to use, copy, modify, and distribute this software for any
7e1bcce4fSTim J. Robbins  * purpose with or without fee is hereby granted, provided that the above
8e1bcce4fSTim J. Robbins  * copyright notice and this permission notice appear in all copies.
9e1bcce4fSTim J. Robbins  *
10e1bcce4fSTim J. Robbins  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11e1bcce4fSTim J. Robbins  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12e1bcce4fSTim J. Robbins  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13e1bcce4fSTim J. Robbins  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14e1bcce4fSTim J. Robbins  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15e1bcce4fSTim J. Robbins  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16e1bcce4fSTim J. Robbins  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17e1bcce4fSTim J. Robbins  *
18e1bcce4fSTim J. Robbins  * Sponsored in part by the Defense Advanced Research Projects
19e1bcce4fSTim J. Robbins  * Agency (DARPA) and Air Force Research Laboratory, Air Force
20e1bcce4fSTim J. Robbins  * Materiel Command, USAF, under agreement number F39502-99-1-0512.
21e1bcce4fSTim J. Robbins  */
22e1bcce4fSTim J. Robbins 
23e1bcce4fSTim J. Robbins #if 0
24e1bcce4fSTim J. Robbins #if defined(LIBC_SCCS) && !defined(lint)
25e1bcce4fSTim J. Robbins static const char rcsid[] = "$OpenBSD: nftw.c,v 1.4 2004/07/07 16:05:23 millert Exp $";
26e1bcce4fSTim J. Robbins #endif /* LIBC_SCCS and not lint */
27e1bcce4fSTim J. Robbins #endif
28e1bcce4fSTim J. Robbins 
29e1bcce4fSTim J. Robbins #include <sys/cdefs.h>
30e1bcce4fSTim J. Robbins __FBSDID("$FreeBSD$");
31e1bcce4fSTim J. Robbins 
32e1bcce4fSTim J. Robbins #include <sys/types.h>
33e1bcce4fSTim J. Robbins #include <sys/stat.h>
34e1bcce4fSTim J. Robbins #include <errno.h>
35e1bcce4fSTim J. Robbins #include <fts.h>
36e1bcce4fSTim J. Robbins #include <ftw.h>
37e1bcce4fSTim J. Robbins 
38e1bcce4fSTim J. Robbins int
39e1bcce4fSTim J. Robbins nftw(const char *path, int (*fn)(const char *, const struct stat *, int,
40e1bcce4fSTim J. Robbins      struct FTW *), int nfds, int ftwflags)
41e1bcce4fSTim J. Robbins {
42e1bcce4fSTim J. Robbins 	char * const paths[2] = { (char *)path, NULL };
43e1bcce4fSTim J. Robbins 	struct FTW ftw;
44e1bcce4fSTim J. Robbins 	FTSENT *cur;
45e1bcce4fSTim J. Robbins 	FTS *ftsp;
46e1bcce4fSTim J. Robbins 	int error = 0, ftsflags, fnflag, postorder, sverrno;
47e1bcce4fSTim J. Robbins 
48e1bcce4fSTim J. Robbins 	/* XXX - nfds is currently unused */
49*4d767dfeSJilles Tjoelker 	if (nfds < 1) {
50e1bcce4fSTim J. Robbins 		errno = EINVAL;
51e1bcce4fSTim J. Robbins 		return (-1);
52e1bcce4fSTim J. Robbins 	}
53e1bcce4fSTim J. Robbins 
54e1bcce4fSTim J. Robbins 	ftsflags = FTS_COMFOLLOW;
55e1bcce4fSTim J. Robbins 	if (!(ftwflags & FTW_CHDIR))
56e1bcce4fSTim J. Robbins 		ftsflags |= FTS_NOCHDIR;
57e1bcce4fSTim J. Robbins 	if (ftwflags & FTW_MOUNT)
58e1bcce4fSTim J. Robbins 		ftsflags |= FTS_XDEV;
59e1bcce4fSTim J. Robbins 	if (ftwflags & FTW_PHYS)
60e1bcce4fSTim J. Robbins 		ftsflags |= FTS_PHYSICAL;
61e1bcce4fSTim J. Robbins 	else
62e1bcce4fSTim J. Robbins 		ftsflags |= FTS_LOGICAL;
63e1bcce4fSTim J. Robbins 	postorder = (ftwflags & FTW_DEPTH) != 0;
64e1bcce4fSTim J. Robbins 	ftsp = fts_open(paths, ftsflags, NULL);
65e1bcce4fSTim J. Robbins 	if (ftsp == NULL)
66e1bcce4fSTim J. Robbins 		return (-1);
67e1bcce4fSTim J. Robbins 	while ((cur = fts_read(ftsp)) != NULL) {
68e1bcce4fSTim J. Robbins 		switch (cur->fts_info) {
69e1bcce4fSTim J. Robbins 		case FTS_D:
70e1bcce4fSTim J. Robbins 			if (postorder)
71e1bcce4fSTim J. Robbins 				continue;
72e1bcce4fSTim J. Robbins 			fnflag = FTW_D;
73e1bcce4fSTim J. Robbins 			break;
74e1bcce4fSTim J. Robbins 		case FTS_DNR:
75e1bcce4fSTim J. Robbins 			fnflag = FTW_DNR;
76e1bcce4fSTim J. Robbins 			break;
77e1bcce4fSTim J. Robbins 		case FTS_DP:
78e1bcce4fSTim J. Robbins 			if (!postorder)
79e1bcce4fSTim J. Robbins 				continue;
80e1bcce4fSTim J. Robbins 			fnflag = FTW_DP;
81e1bcce4fSTim J. Robbins 			break;
82e1bcce4fSTim J. Robbins 		case FTS_F:
83e1bcce4fSTim J. Robbins 		case FTS_DEFAULT:
84e1bcce4fSTim J. Robbins 			fnflag = FTW_F;
85e1bcce4fSTim J. Robbins 			break;
86e1bcce4fSTim J. Robbins 		case FTS_NS:
87e1bcce4fSTim J. Robbins 		case FTS_NSOK:
88e1bcce4fSTim J. Robbins 			fnflag = FTW_NS;
89e1bcce4fSTim J. Robbins 			break;
90e1bcce4fSTim J. Robbins 		case FTS_SL:
91e1bcce4fSTim J. Robbins 			fnflag = FTW_SL;
92e1bcce4fSTim J. Robbins 			break;
93e1bcce4fSTim J. Robbins 		case FTS_SLNONE:
94e1bcce4fSTim J. Robbins 			fnflag = FTW_SLN;
95e1bcce4fSTim J. Robbins 			break;
96e1bcce4fSTim J. Robbins 		case FTS_DC:
97e1bcce4fSTim J. Robbins 			errno = ELOOP;
98e1bcce4fSTim J. Robbins 			/* FALLTHROUGH */
99e1bcce4fSTim J. Robbins 		default:
100e1bcce4fSTim J. Robbins 			error = -1;
101e1bcce4fSTim J. Robbins 			goto done;
102e1bcce4fSTim J. Robbins 		}
103e1bcce4fSTim J. Robbins 		ftw.base = cur->fts_pathlen - cur->fts_namelen;
104e1bcce4fSTim J. Robbins 		ftw.level = cur->fts_level;
105e1bcce4fSTim J. Robbins 		error = fn(cur->fts_path, cur->fts_statp, fnflag, &ftw);
106e1bcce4fSTim J. Robbins 		if (error != 0)
107e1bcce4fSTim J. Robbins 			break;
108e1bcce4fSTim J. Robbins 	}
109e1bcce4fSTim J. Robbins done:
110e1bcce4fSTim J. Robbins 	sverrno = errno;
111e1bcce4fSTim J. Robbins 	if (fts_close(ftsp) != 0 && error == 0)
112e1bcce4fSTim J. Robbins 		error = -1;
113e1bcce4fSTim J. Robbins 	else
114e1bcce4fSTim J. Robbins 		errno = sverrno;
115e1bcce4fSTim J. Robbins 	return (error);
116e1bcce4fSTim J. Robbins }
117