#include <mcheck.h>
#include <stdio.h>
#include <stdlib.h>
+#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
int do_depth;
int do_chdir;
int do_phys;
+int do_exit;
+char *skip_subtree;
+char *skip_siblings;
struct option options[] =
{
{ "depth", no_argument, &do_depth, 1 },
{ "chdir", no_argument, &do_chdir, 1 },
{ "phys", no_argument, &do_phys, 1 },
+ { "skip-subtree", required_argument, NULL, 't' },
+ { "skip-siblings", required_argument, NULL, 's' },
+ { "early-exit", no_argument, &do_exit, 1 },
{ NULL, 0, NULL, 0 }
};
};
-int
+static int
cb (const char *name, const struct stat *st, int flag, struct FTW *f)
{
+ if (do_exit && strcmp (name + f->base, "file@2"))
+ return FTW_CONTINUE;
+
printf ("base = \"%.*s\", file = \"%s\", flag = %s",
f->base, name, name + f->base, flag2name[flag]);
if (do_chdir)
printf (", cwd = %s", cwd);
free (cwd);
}
- puts ("");
- return 0;
+ printf (", level = %d\n", f->level);
+
+ if (skip_siblings && strcmp (name + f->base, skip_siblings) == 0)
+ return FTW_SKIP_SIBLINGS;
+
+ if (skip_subtree && strcmp (name + f->base, skip_subtree) == 0)
+ return FTW_SKIP_SUBTREE;
+
+ return do_exit ? 26 : FTW_CONTINUE;
}
int
mtrace ();
while ((opt = getopt_long_only (argc, argv, "", options, NULL)) != -1)
- ;
+ {
+ if (opt == 't')
+ skip_subtree = optarg;
+ else if (opt == 's')
+ skip_siblings = optarg;
+ }
if (do_chdir)
flag |= FTW_CHDIR;
flag |= FTW_DEPTH;
if (do_phys)
flag |= FTW_PHYS;
+ if (skip_subtree || skip_siblings)
+ {
+ flag |= FTW_ACTIONRETVAL;
+ if (do_exit)
+ {
+ printf ("--early-exit cannot be used together with --skip-{siblings,subtree}");
+ exit (1);
+ }
+ }
- r = nftw (optind < argc ? argv[optind] : ".", cb, 3, flag);
- if (r)
+ char *cw1 = getcwd (NULL, 0);
+
+ r = nftw (optind < argc ? argv[optind] : ".", cb, do_exit ? 1 : 3, flag);
+ if (r < 0)
perror ("nftw");
+
+ char *cw2 = getcwd (NULL, 0);
+
+ if (strcmp (cw1, cw2) != 0)
+ {
+ printf ("current working directory before and after nftw call differ:\n"
+ "before: %s\n"
+ "after: %s\n", cw1, cw2);
+ exit (1);
+ }
+
+ if (do_exit)
+ {
+ puts (r == 26 ? "succeeded" : "failed");
+ return r == 26 ? 0 : 1;
+ }
return r;
}