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