io/ftwtest.c

#include <ftw.h>
#include <getopt.h>
#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 }
};

const char *flag2name[] =
{
  [FTW_F] = "FTW_F",
  [FTW_D] = "FTW_D",
  [FTW_DNR] = "FTW_DNR",
  [FTW_NS] = "FTW_NS",
  [FTW_SL] = "FTW_SL",
  [FTW_DP] = "FTW_DP",
  [FTW_SLN] = "FTW_SLN"
};


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)
    {
      char *cwd = getcwd (NULL, 0);
      printf (", cwd = %s", cwd);
      free (cwd);
    }
  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
main (int argc, char *argv[])
{
  int opt;
  int r;
  int flag = 0;
  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;
  if (do_depth)
    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);
	}
    }

  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;
}