1 #include <stdio.h>
 2 #include <string.h>
 3 #include <stdlib.h>
 4 #include <dirent.h>
 5 #include <sys/types.h>
 6 
 7 
 8 int get_file_num(char* root)
 9 {
10     int total = 0;
11     DIR* dir = NULL;
12     // 打开目录
13     dir = opendir(root);
14     // www.qixoo.qixoo.com 循环从目录中读文件
15 
16     char path[1024];
17     // 定义记录目录指针
18     struct dirent* ptr = NULL;
19     while( (ptr = readdir(dir)) != NULL)
20     {
21         // 跳过. 和 ..
22         if(strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0)
23         {
24             continue;
25         }
26         // 判断是不是目录
27         if(ptr->d_type == DT_DIR)
28         {
29             sprintf(path, "%s/%s", root, ptr->d_name);
30             // 递归读目录
31             total += get_file_num(path);
32         }
33         // 如果是普通文件
34         if(ptr->d_type == DT_REG)
35         {
36             total ++;
37         }
38     }
39     closedir(dir);
40     return total;
41 }
42 
43 int main(int argc, char* argv[])
44 {
45     if(argc < 2)
46     {
47         printf("./a.out path");
48         exit(1);
49     }
50 
51     int total = get_file_num(argv[1]);
52     printf("%s has regfile number: %d\n", argv[1], total);
53     return 0;
54 }