首页 > 代码库 > 用C语言实现统计一个文件夹中各种文件的比例

用C语言实现统计一个文件夹中各种文件的比例

原文:用C语言实现统计一个文件夹中各种文件的比例

  《UNIX环境高级编程》中的程序清单4-7就介绍了如何实现递归地统计某个目录下面的文件!我刚开始看过它的代码后,觉得照着敲太没意思了,所以就合上书自己写了一遍!为此还写了一篇博文,这是博文地址: 在linux下用C语言实现递归查看某个目录中的所有文件【CSDN】!

  今天做《Unix环境高级编程》的课后题,看到题目4.11这里提供了一种新的实现这个程序的思路,那就是每回读到一个目录,就通过chdir函数进入到这个目录,然后再通过opendir函数和readdir函数来读取这个目录中的文件,然后一个一个分析,如果是目录,则进行递归调用。如果不是目录,则对这个文件进行计数后立刻返回!这样一个一个分析完目录中的所有文件之后再来进行一个chdir(".."),返回到上一级的目录。具体的实现代码如下:

  

  1 #include<stdio.h>  2 #include<string.h>  3 #include<stdlib.h>  4 #include<errno.h>  5 #include<linux/limits.h>  6 #include<unistd.h>  7 #include<sys/types.h>  8 #include<sys/stat.h>  9 #include<dirent.h> 10  11 //所有函数的声明 12 typedef int MyFunc(const char *,const struct stat*,int); 13 static MyFunc myfunc;        //定义处理文件的函数 14 static int myftw(const char *,MyFunc *); 15 static int dopath(MyFunc *); 16  17 //定义的全局变量 18 static char *fullpath;    //存放文件的名称的变量 19 static long sock_c,lnk_c,reg_c,blk_c,dir_c,chr_c,fifo_c,total_c;    //统计各种文件类型的数量 20  21 //myfunc函数中需要定义的宏 22 #define FTW_F 1        //文件类型是文件 23 #define FTW_D 2        //文件类型是目录 24 #define FTW_NS 3    //一个文件不能stat 25 #define FTW_ND 4    //一个目录不能被读 26 int main(int argc,char *argv[]) 27 { 28     if(argc != 2) 29     { 30     printf("Usage:%s pathname\n",argv[0]+2); 31     exit(EXIT_FAILURE); 32     } 33     myftw(argv[1],myfunc); 34     total_c = sock_c+lnk_c+reg_c+blk_c+dir_c+chr_c+fifo_c; 35     if(0 == total_c) 36     { 37     total_c = 1; 38     } 39     printf("socket files    = %7ld, %5.2f%%\n",sock_c,sock_c*100.0/total_c); 40     printf("link files      = %7ld, %5.2f%%\n",lnk_c,lnk_c*100.0/total_c); 41     printf("regular files   = %7ld, %5.2f%%\n",reg_c,reg_c*100.0/total_c); 42     printf("block files     = %7ld, %5.2f%%\n",blk_c,blk_c*100.0/total_c); 43     printf("directory files = %7ld, %5.2f%%\n",dir_c,dir_c*100.0/total_c); 44     printf("character files = %7ld, %5.2f%%\n",chr_c,chr_c*100.0/total_c); 45     printf("FIFO files      = %7ld, %5.2f%%\n",fifo_c,fifo_c*100.0/total_c); 46     printf("total files     = %7ld, %5.2f%%\n",total_c,total_c*100.0/total_c); 47  48     return 0; 49 } 50 static int myftw(const char* pathname,MyFunc *pmyfunc) 51 { 52     int ret; 53  54     fullpath = (char *)malloc(sizeof(char)*PATH_MAX); 55     strcpy(fullpath,pathname); 56     ret = dopath(myfunc); 57     free(fullpath); 58  59     return ret; 60 } 61 static int dopath(MyFunc *pmyfunc) 62 { 63     int ret; 64     struct stat statbuf; 65     char *ptr; 66     DIR *dp; 67     struct dirent* dirp; 68  69     if(-1 == lstat(fullpath,&statbuf)) 70     { 71     ret = pmyfunc(fullpath,&statbuf,FTW_NS); 72     return ret; 73     } 74     if(S_ISDIR(statbuf.st_mode) != 1) 75     { 76     ret = pmyfunc(fullpath,&statbuf,FTW_F); 77     return ret; 78     } 79  80     //使目录文件++ 81     if(0 != (ret=pmyfunc(fullpath,&statbuf,FTW_D))) 82     return ret; 83  84     //如果是目录文件则进入这个目录 85     if(-1 == chdir(fullpath)) 86     { 87     printf("%s[chdir]%s\n",fullpath,strerror(errno)); 88     ret == -1; 89     return ret; 90     } 91  92     //打开当前目录 93     if(NULL == (dp=opendir("."))) 94     { 95     ret = pmyfunc(fullpath,&statbuf,FTW_ND); 96     return ret; 97     } 98     while(NULL != (dirp=readdir(dp))) 99     {100     //忽略.和..文件(dot)101     if(0==strcmp(dirp->d_name,".") || 0==strcmp(dirp->d_name,".."))102         continue;103     memset(fullpath,0,PATH_MAX);104     strcpy(fullpath,dirp->d_name);105 106     if(0 != (ret=dopath(myfunc)))    //进行递归107         break;108     }109     chdir("..");    //将当前目录设置为上一级目录110     //对关闭文件进行判断111     if(-1 == closedir(dp))112     {113     printf("不能关闭%s\nError:%s",fullpath,strerror(errno));114     }115 116     return ret;117 }118 static int myfunc(const char * pathname,const struct stat * statptr,int type)119 {120     switch(type)121     {122     case FTW_F:123     switch(statptr->st_mode & S_IFMT)124     {125     case S_IFSOCK:    sock_c++;    break;126     case S_IFLNK:    lnk_c++;    break;127     case S_IFREG:    reg_c++;    break;128     case S_IFBLK:    blk_c++;    break;129     case S_IFCHR:    chr_c++;    break;130     case S_IFIFO:    fifo_c++;    break;131     case S_IFDIR:132         printf("Error:这里不应该出现目录文件%s!\n\nError:%s\n",pathname,strerror(errno));133         break;134     }135     break;136     case FTW_D:137     dir_c++;    break;138     case FTW_ND:139     printf("不能打开目录%s\nError:%s\n",pathname,strerror(errno));140     break;141     case FTW_NS:142     printf("不能打开文件%s\nError:%s\n",pathname,strerror(errno));143     break;144     }145     return 0;146 }

  我这个代码并不是我自己合上书写的,而是在W.Richard Stevens书中给出的代码的基础上改的!在此要特别感谢这些真正的大神给我们提供了这么优秀的书籍!这是这个程序的运行结果,

  那个第一行是我特意设置的,那个root是一个文件夹的名字,是属于root用户的,所以我这里并不能读取,会报出一个错误!下面是这个改进后的程序和原来书中的程序的一个对比,发现效率还真的是提高不少啊!

  那个descend_hierarchy_p的那个程序是书上给的程序,每回读取都是读取的绝对路径的名称!而那个descend_hierarchy_ch命令就是每回碰到一个目录就进入到那个文件夹中,然后再来读取,这样每回读取的时候读取的就是相对路径的名称了!

用C语言实现统计一个文件夹中各种文件的比例