io/ftwtest.c

   1 #include <ftw.h>
   2 #include <getopt.h>
   3 #include <mcheck.h>
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7 #include <unistd.h>
   8 #include <sys/stat.h>
   9
  10
  11 int do_depth;
  12 int do_chdir;
  13 int do_phys;
  14 int do_exit;
  15 char *skip_subtree;
  16 char *skip_siblings;
  17
  18 struct option options[] =
  19 {
  20   { "depth", no_argument, &do_depth, 1 },
  21   { "chdir", no_argument, &do_chdir, 1 },
  22   { "phys", no_argument, &do_phys, 1 },
  23   { "skip-subtree", required_argument, NULL, 't' },
  24   { "skip-siblings", required_argument, NULL, 's' },
  25   { "early-exit", no_argument, &do_exit, 1 },
  26   { NULL, 0, NULL, 0 }
  27 };
  28
  29 const char *flag2name[] =
  30 {
  31   [FTW_F] = "FTW_F",
  32   [FTW_D] = "FTW_D",
  33   [FTW_DNR] = "FTW_DNR",
  34   [FTW_NS] = "FTW_NS",
  35   [FTW_SL] = "FTW_SL",
  36   [FTW_DP] = "FTW_DP",
  37   [FTW_SLN] = "FTW_SLN"
  38 };
  39
  40
  41 static int
  42 cb (const char *name, const struct stat *st, int flag, struct FTW *f)
  43 {
  44   if (do_exit && strcmp (name + f->base, "file@2"))
  45     return FTW_CONTINUE;
  46
  47   printf ("base = \"%.*s\", file = \"%s\", flag = %s",
  48           f->base, name, name + f->base, flag2name[flag]);
  49   if (do_chdir)
  50     {
  51       char *cwd = getcwd (NULL, 0);
  52       printf (", cwd = %s", cwd);
  53       free (cwd);
  54     }
  55   printf (", level = %d\n", f->level);
  56
  57   if (skip_siblings && strcmp (name + f->base, skip_siblings) == 0)
  58     return FTW_SKIP_SIBLINGS;
  59
  60   if (skip_subtree && strcmp (name + f->base, skip_subtree) == 0)
  61     return FTW_SKIP_SUBTREE;
  62
  63   return do_exit ? 26 : FTW_CONTINUE;
  64 }
  65
  66 int
  67 main (int argc, char *argv[])
  68 {
  69   int opt;
  70   int r;
  71   int flag = 0;
  72   mtrace ();
  73
  74   while ((opt = getopt_long_only (argc, argv, "", options, NULL)) != -1)
  75     {
  76       if (opt == 't')
  77         skip_subtree = optarg;
  78       else if (opt == 's')
  79         skip_siblings = optarg;
  80     }
  81
  82   if (do_chdir)
  83     flag |= FTW_CHDIR;
  84   if (do_depth)
  85     flag |= FTW_DEPTH;
  86   if (do_phys)
  87     flag |= FTW_PHYS;
  88   if (skip_subtree || skip_siblings)
  89     {
  90       flag |= FTW_ACTIONRETVAL;
  91       if (do_exit)
  92         {
  93           printf ("--early-exit cannot be used together with --skip-{siblings,subtree}");
  94           exit (1);
  95         }
  96     }
  97
  98   char *cw1 = getcwd (NULL, 0);
  99
 100   r = nftw (optind < argc ? argv[optind] : ".", cb, do_exit ? 1 : 3, flag);
 101   if (r < 0)
 102     perror ("nftw");
 103
 104   char *cw2 = getcwd (NULL, 0);
 105
 106   if (strcmp (cw1, cw2) != 0)
 107     {
 108       printf ("current working directory before and after nftw call differ:\n"
 109               "before: %s\n"
 110               "after:  %s\n", cw1, cw2);
 111       exit (1);
 112     }
 113
 114   if (do_exit)
 115     {
 116       puts (r == 26 ? "succeeded" : "failed");
 117       return r == 26 ? 0 : 1;
 118     }
 119   return r;
 120 }