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