获取目录下的普通文件数目

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107

/**********************************************************  * Author        : oyjb  * Email         : jbouyang@126.com  * Last modified : 2015-09-13 23:01  * Filename      : get_file_count.c  * Description   : copy right for oyjb  * *******************************************************/ #include <stdio.h> #include <stdlib.h> #include <dirent.h> #include <sys/stat.h> #include <errno.h> #include <string.h>   #define MAX 1024   int get_file_count(const char *root);   int main(int argc, char *argv[]) {     if (argc < 2)     {         printf("wrong usage.\n");         exit(1);     }       int total = 0;       total = get_file_count(argv[1]);       printf("%s has %d files.\n", argv[1], total);       exit(0); }   /*  * The parameter for function of get_file_count is a filepath,represents a directory's root  * This function return the numbers of all regular files in this directory,include child directory  */ int get_file_count(const char *root) {     DIR *dir;     struct dirent *ptr;     int total = 0; //numbers of files       char path[MAX];           dir = opendir(root); //open directory     if (dir == NULL)     {         perror("fail to open directory");         exit(1);     }       errno = 0;     while ( (ptr = readdir(dir)) != NULL)     {         //jump the . and ..         if (strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0)             continue;           /*          * 以下的得到子目录全路径的方法还可以是：          * sprintf(path, "%s/%s", root, ptr->d_name);          */         sprintf(path, "%s/%s", root, ptr->d_name);         /*         path[0] = '\0';         strcat(path, root);         strcat(path, "/");  //you must be strcat the character "/"，other than error.         strcat(path, ptr->d_name);         */           /*          * 由于xfs文件系统在读取dirent的d_type时无法读取，因此需要转到stat函数处理，读取文件信息          * 否则下面的代码可换成：          * if (ptr->d_type == DT_DIR)          *        total += get_file_count(path);          * if (ptr->d_type == DT_REG)          *        ++total;          * 正是因为d_type与文件系统有关(应该是)，所以最好还是用stat函数获取文件类型信息          */         struct stat sta;         if (lstat(path, &sta) == -1)         {             printf("%s\n", path);             perror("fail to get file information.");             return 0;         }           if (S_ISDIR(sta.st_mode))             total += get_file_count(path);                   else if (S_ISREG(sta.st_mode) || S_ISLNK(sta.st_mode))             ++total;     }       if (errno != 0)     {         perror("fail to read dir");         exit(1);     }           closedir(dir);       return total; }