--- a/include/ftw.h +++ b/include/ftw.h @@ -20,6 +20,14 @@ #define FTW_MOUNT 2 #define FTW_CHDIR 4 #define FTW_DEPTH 8 + +#ifdef _GNU_SOURCE +#define FTW_ACTIONRETVAL 16 +#define FTW_CONTINUE 0 +#define FTW_STOP 1 +#define FTW_SKIP_SUBTREE 2 +#define FTW_SKIP_SIBLINGS 3 +#endif struct FTW { int base; --- a/src/misc/nftw.c +++ b/src/misc/nftw.c @@ -1,3 +1,8 @@ +#define FTW_ACTIONRETVAL 16 +#define FTW_CONTINUE 0 +#define FTW_STOP 1 +#define FTW_SKIP_SUBTREE 2 +#define FTW_SKIP_SIBLINGS 3 #include #include #include @@ -58,8 +58,20 @@ lev.level = new.level; lev.base = h ? h->base : (name=strrchr(path, '/')) ? name-path : 0; - if (!(flags & FTW_DEPTH) && (r=fn(path, &st, type, &lev))) - return r; + if (!(flags & FTW_DEPTH) && (r=fn(path, &st, type, &lev))) { + if (flags & FTW_ACTIONRETVAL) + switch (r) { + case FTW_SKIP_SUBTREE: + h = NULL; + case FTW_CONTINUE: + break; + case FTW_SKIP_SIBLINGS: + case FTW_STOP: + return r; + } + else + return r; + } for (; h; h = h->chain) if (h->dev == st.st_dev && h->ino == st.st_ino) @@ -82,7 +94,10 @@ strcpy(path+j+1, de->d_name); if ((r=do_nftw(path, fn, fd_limit-1, flags, &new))) { closedir(d); - return r; + if ((flags & FTW_ACTIONRETVAL) && r == FTW_SKIP_SIBLINGS) + break; + else + return r; } } closedir(d); @@ -93,8 +108,16 @@ } path[l] = 0; - if ((flags & FTW_DEPTH) && (r=fn(path, &st, type, &lev))) - return r; + if ((flags & FTW_DEPTH) && (r=fn(path, &st, type, &lev))) { + if (flags & FTW_ACTIONRETVAL) + switch (r) { + case FTW_SKIP_SIBLINGS: + case FTW_STOP: + return r; + } + else + return r; + } return 0; }