用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(constchar *,conststruct stat*,int);
 13static MyFunc myfunc;        //定义处理文件的函数 14staticint myftw(constchar *,MyFunc *);
 15staticint dopath(MyFunc *);
 16 17//定义的全局变量 18staticchar *fullpath;    //存放文件的名称的变量 19staticlong 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    //一个目录不能被读 26int main(int argc,char *argv[])
 27{
 28if(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;
 35if(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 48return0;
 49}
 50staticint myftw(constchar* pathname,MyFunc *pmyfunc)
 51{
 52int ret;
 53 54     fullpath = (char *)malloc(sizeof(char)*PATH_MAX);
 55    strcpy(fullpath,pathname);
 56     ret = dopath(myfunc);
 57    free(fullpath);
 58 59return ret;
 60}
 61staticint dopath(MyFunc *pmyfunc)
 62{
 63int ret;
 64struct stat statbuf;
 65char *ptr;
 66     DIR *dp;
 67struct dirent* dirp;
 68 69if(-1 == lstat(fullpath,&statbuf))
 70    {
 71     ret = pmyfunc(fullpath,&statbuf,FTW_NS);
 72return ret;
 73    }
 74if(S_ISDIR(statbuf.st_mode) != 1)
 75    {
 76     ret = pmyfunc(fullpath,&statbuf,FTW_F);
 77return ret;
 78    }
 79 80//使目录文件++ 81if(0 != (ret=pmyfunc(fullpath,&statbuf,FTW_D)))
 82return ret;
 83 84//如果是目录文件则进入这个目录 85if(-1 == chdir(fullpath))
 86    {
 87     printf("%s[chdir]%s\n",fullpath,strerror(errno));
 88     ret == -1;
 89return ret;
 90    }
 91 92//打开当前目录 93if(NULL == (dp=opendir(".")))
 94    {
 95     ret = pmyfunc(fullpath,&statbuf,FTW_ND);
 96return ret;
 97    }
 98while(NULL != (dirp=readdir(dp)))
 99    {
100//忽略.和..文件(dot)101if(0==strcmp(dirp->d_name,".") || 0==strcmp(dirp->d_name,".."))
102continue;
103     memset(fullpath,0,PATH_MAX);
104     strcpy(fullpath,dirp->d_name);
105106if(0 != (ret=dopath(myfunc)))    //进行递归107break;
108    }
109     chdir("..");    //将当前目录设置为上一级目录
110//对关闭文件进行判断111if(-1 == closedir(dp))
112    {
113     printf("不能关闭%s\nError:%s",fullpath,strerror(errno));
114    }
115116return ret;
117}
118staticint myfunc(constchar * pathname,conststruct stat * statptr,int type)
119{
120switch(type)
121    {
122case FTW_F:
123switch(statptr->st_mode & S_IFMT)
124    {
125case S_IFSOCK:    sock_c++;    break;
126case S_IFLNK:    lnk_c++;    break;
127case S_IFREG:    reg_c++;    break;
128case S_IFBLK:    blk_c++;    break;
129case S_IFCHR:    chr_c++;    break;
130case S_IFIFO:    fifo_c++;    break;
131case S_IFDIR:
132         printf("Error:这里不应该出现目录文件%s!\n\nError:%s\n",pathname,strerror(errno));
133break;
134    }
135break;
136case FTW_D:
137     dir_c++;    break;
138case FTW_ND:
139     printf("不能打开目录%s\nError:%s\n",pathname,strerror(errno));
140break;
141case FTW_NS:
142     printf("不能打开文件%s\nError:%s\n",pathname,strerror(errno));
143break;
144    }
145return0;
146 }