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