linux下遍历目录

小咪咪 2022-03-28 08:19 225阅读 0赞

项目中需要遍历目录并对目录下的所有文件进行处理分析，于是先实现一下遍历目录的功能

实现代码：

1 #include <unistd.h>
     2 #include <dirent.h>
     3 #include <iostream>
     4 #include <stdio.h>
     5 #include <stdlib.h>
     6 #include <string.h>
     7 #include <sys/stat.h>
     8 
     9 using namespace std;
    10 
    11 void traverseDir(char *pPath, int nDeepth);
    12 
    13 int main(int argc, char **argv)
    14 {
    15     for (int i = 1; i < argc; ++i)
    16     {
    17         traverseDir(argv[i], 0);
    18     }
    19 
    20     return 0;
    21 }
    22 
    23 void traverseDir(char *pPath, int nDeepth)
    24 {
    25     DIR *pDir = NULL;
    26     struct dirent *pSTDirEntry;
    27     char *pChild = NULL;
    28 
    29 
    30     if ((pDir = opendir(pPath)) == NULL)
    31     {
    32         return ;
    33     }
    34     else
    35     {
    36         while ((pSTDirEntry = readdir(pDir)) != NULL)
    37         {
    38             if ((strcmp(pSTDirEntry->d_name, ".") == 0) || (strcmp(pSTDirEntry->d_name, "..") == 0))
    39             {
    40                 continue;
    41             }
    42             else
    43             {
    44                 for (int i = 0; i < nDeepth; i++)
    45                 {
    46                     cout << "\t";
    47                 }
    48                 cout << pSTDirEntry->d_name << endl;
    49 
    50                 if (pSTDirEntry->d_type & DT_DIR)
    51                 {
    52                     pChild = (char*)malloc(sizeof(char) * (NAME_MAX + 1));
    53                     if (pChild == NULL)
    54                     {
    55                         perror("memory not enough.");
    56                         return ;
    57                     }
    58                     memset(pChild, 0, NAME_MAX + 1);
    59                     strcpy(pChild, pPath);
    60                     strcat(pChild, pSTDirEntry->d_name);
    61                     traverseDir(pChild, nDeepth + 1);
    62                     free(pChild);
    63                     pChild = NULL;
    64                 }
    65             }
    66         }
    67         closedir(pDir);
    68     }
    69 }

此代码仅仅实现了遍历的功能，输出界面美观性没有考虑，而且也不太健壮，比如执行以下命令：

1 ./Dir ../
    2 ./Dir ..

第二行的就无法递归遍历文件夹，因为没有对路径结尾的/做处理。

使用到的函数是opendir、readdir、closedir，还有结构体dirent以及DIR

没有什么太难的地方，只是有一点需要注意，判断一个路径是不是目录应该使用pSTDirEntry->d\_type & DT\_DIR 判断相与的值是否等于零，不等于0则为目录

参考：http://blog.csdn.net/pengqianhe/article/details/8567955