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

  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 }