C:使用nftw指定最大搜索深度

时间:2015-11-20 20:37:16

标签: c nftw

在C中,有没有办法指定nftw将搜索的基目录的最大深度?例如,假设我要搜索的目录dir有一个子子目录,但我只希望nftw搜索subdir而不是sub-subdir,或者低于该值的任何内容。

dir
 \_ subdir
    |__ file1
    |__ file2
     \_ sub-subdir
        |__ file1
        |__ file2
         \_ file3

1 个答案:

答案 0 :(得分:2)

根据手册页(http://man7.org/linux/man-pages/man3/nftw.3.html),您可以停止从函数参数内部进入子目录。

从手册中报告的示例中,限制为子目录中的2个级别,源代码为:

#define _XOPEN_SOURCE 500
#include <ftw.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>


// max num of sub dirs
#define MAXLEVEL  2

static int display_info(const char *fpath, const struct stat *sb,
            int tflag, struct FTW *ftwbuf)
{
   // if the depth is above the max sub dirs, continue to next file
   if (ftwbuf->level > MAXLEVEL) {
       return 0; 
   }
   printf("%-3s %2d %7jd   %-40s %d %s\n",
       (tflag == FTW_D) ?   "d"   : (tflag == FTW_DNR) ? "dnr" :
       (tflag == FTW_DP) ?  "dp"  : (tflag == FTW_F) ?   "f" :
       (tflag == FTW_NS) ?  "ns"  : (tflag == FTW_SL) ?  "sl" :
       (tflag == FTW_SLN) ? "sln" : "???",
       ftwbuf->level, (intmax_t) sb->st_size,
       fpath, ftwbuf->base, fpath + ftwbuf->base);
   return 0;           /* To tell nftw() to continue */
}

int main(int argc, char *argv[])
{
   int flags = 0;

   if (argc > 2 && strchr(argv[2], 'd') != NULL)
       flags |= FTW_DEPTH;
   if (argc > 2 && strchr(argv[2], 'p') != NULL)
       flags |= FTW_PHYS;

   if (nftw((argc < 2) ? "." : argv[1], display_info, 20, flags) == -1) {
       perror("nftw");
       exit(EXIT_FAILURE);
   }
   exit(EXIT_SUCCESS);
}