1 /*	$OpenBSD: nftw.c,v 1.5 2005/08/08 08:05:34 espie 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 
23 #include <sys/types.h>
24 #include <sys/stat.h>
25 #include <errno.h>
26 #include <fts.h>
27 #include <ftw.h>
28 #include <limits.h>
29 #include <string.h>
30 #include <stdlib.h>
31 
32 __RCSID("$MirOS: src/lib/libc/gen/nftw.c,v 1.2 2009/11/09 21:30:48 tg Exp $");
33 
34 int
nftw(const char * path,int (* fn)(const char *,const struct stat *,int,struct FTW *),int nfds,int ftwflags)35 nftw(const char *path, int (*fn)(const char *, const struct stat *, int,
36      struct FTW *), int nfds, int ftwflags)
37 {
38 	struct FTW ftw_;
39 	FTSENT *cur;
40 	FTS *ftsp;
41 	int error = 0, ftsflags, fnflag, postorder, sverrno;
42 	char *paths[2], *allocd;
43 
44 	/* XXX - nfds is currently unused */
45 	if (nfds < 1 || nfds > OPEN_MAX) {
46 		errno = EINVAL;
47 		return (-1);
48 	}
49 
50 	paths[0] = allocd = strdup(path);
51 	paths[1] = NULL;
52 
53 	ftsflags = FTS_COMFOLLOW;
54 	if (!(ftwflags & FTW_CHDIR))
55 		ftsflags |= FTS_NOCHDIR;
56 	if (ftwflags & FTW_MOUNT)
57 		ftsflags |= FTS_XDEV;
58 	if (ftwflags & FTW_PHYS)
59 		ftsflags |= FTS_PHYSICAL;
60 	else
61 		ftsflags |= FTS_LOGICAL;
62 	postorder = (ftwflags & FTW_DEPTH) != 0;
63 	ftsp = fts_open(paths, ftsflags, NULL);
64 	if (ftsp == NULL) {
65 		free(allocd);
66 		return (-1);
67 	}
68 	while ((cur = fts_read(ftsp)) != NULL) {
69 		switch (cur->fts_info) {
70 		case FTS_D:
71 			if (postorder)
72 				continue;
73 			fnflag = FTW_D;
74 			break;
75 		case FTS_DNR:
76 			fnflag = FTW_DNR;
77 			break;
78 		case FTS_DP:
79 			if (!postorder)
80 				continue;
81 			fnflag = FTW_DP;
82 			break;
83 		case FTS_F:
84 		case FTS_DEFAULT:
85 			fnflag = FTW_F;
86 			break;
87 		case FTS_NS:
88 		case FTS_NSOK:
89 			fnflag = FTW_NS;
90 			break;
91 		case FTS_SL:
92 			fnflag = FTW_SL;
93 			break;
94 		case FTS_SLNONE:
95 			fnflag = FTW_SLN;
96 			break;
97 		case FTS_DC:
98 			errno = ELOOP;
99 			/* FALLTHROUGH */
100 		default:
101 			error = -1;
102 			goto done;
103 		}
104 		ftw_.base = cur->fts_pathlen - cur->fts_namelen;
105 		ftw_.level = cur->fts_level;
106 		error = fn(cur->fts_path, cur->fts_statp, fnflag, &ftw_);
107 		if (error != 0)
108 			break;
109 	}
110  done:
111 	sverrno = errno;
112 	if (fts_close(ftsp) != 0 && error == 0)
113 		error = -1;
114 	else
115 		errno = sverrno;
116 	free(allocd);
117 	return (error);
118 }
119