linux下遍历目录

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

实现代码:

 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

posted @ 2014-10-28 09:43  冷冰若水  阅读(4600)  评论(0编辑  收藏  举报