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

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

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

　　

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<errno.h>
#include<linux/limits.h>
#include<unistd.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<dirent.h>

//所有函数的声明
typedef int MyFunc(const char *,const struct stat*,int);
static MyFunc myfunc;        //定义处理文件的函数
static int myftw(const char *,MyFunc *);
static int dopath(MyFunc *);

//定义的全局变量
static char *fullpath;    //存放文件的名称的变量
static long sock_c,lnk_c,reg_c,blk_c,dir_c,chr_c,fifo_c,total_c;    //统计各种文件类型的数量

//myfunc函数中需要定义的宏
#define FTW_F 1        //文件类型是文件
#define FTW_D 2        //文件类型是目录
#define FTW_NS 3    //一个文件不能stat
#define FTW_ND 4    //一个目录不能被读
int main(int argc,char *argv[])
{
    if(argc != 2)
    {
    printf("Usage:%s pathname
",argv[0]+2);
    exit(EXIT_FAILURE);
    }
    myftw(argv[1],myfunc);
    total_c = sock_c+lnk_c+reg_c+blk_c+dir_c+chr_c+fifo_c;
    if(0 == total_c)
    {
    total_c = 1;
    }
    printf("socket files    = %7ld, %5.2f%%
",sock_c,sock_c*100.0/total_c);
    printf("link files      = %7ld, %5.2f%%
",lnk_c,lnk_c*100.0/total_c);
    printf("regular files   = %7ld, %5.2f%%
",reg_c,reg_c*100.0/total_c);
    printf("block files     = %7ld, %5.2f%%
",blk_c,blk_c*100.0/total_c);
    printf("directory files = %7ld, %5.2f%%
",dir_c,dir_c*100.0/total_c);
    printf("character files = %7ld, %5.2f%%
",chr_c,chr_c*100.0/total_c);
    printf("FIFO files      = %7ld, %5.2f%%
",fifo_c,fifo_c*100.0/total_c);
    printf("total files     = %7ld, %5.2f%%
",total_c,total_c*100.0/total_c);

    return 0;
}
static int myftw(const char* pathname,MyFunc *pmyfunc)
{
    int ret;

    fullpath = (char *)malloc(sizeof(char)*PATH_MAX);
    strcpy(fullpath,pathname);
    ret = dopath(myfunc);
    free(fullpath);

    return ret;
}
static int dopath(MyFunc *pmyfunc)
{
    int ret;
    struct stat statbuf;
    char *ptr;
    DIR *dp;
    struct dirent* dirp;

    if(-1 == lstat(fullpath,&statbuf))
    {
    ret = pmyfunc(fullpath,&statbuf,FTW_NS);
    return ret;
    }
    if(S_ISDIR(statbuf.st_mode) != 1)
    {
    ret = pmyfunc(fullpath,&statbuf,FTW_F);
    return ret;
    }

    //使目录文件++
    if(0 != (ret=pmyfunc(fullpath,&statbuf,FTW_D)))
    return ret;

    //如果是目录文件则进入这个目录
    if(-1 == chdir(fullpath))
    {
    printf("%s[chdir]%s
",fullpath,strerror(errno));
    ret == -1;
    return ret;
    }

    //打开当前目录
    if(NULL == (dp=opendir(".")))
    {
    ret = pmyfunc(fullpath,&statbuf,FTW_ND);
    return ret;
    }
    while(NULL != (dirp=readdir(dp)))
    {
    //忽略.和..文件(dot)
    if(0==strcmp(dirp->d_name,".") || 0==strcmp(dirp->d_name,".."))
        continue;
    memset(fullpath,0,PATH_MAX);
    strcpy(fullpath,dirp->d_name);

    if(0 != (ret=dopath(myfunc)))    //进行递归
        break;
    }
    chdir("..");    //将当前目录设置为上一级目录
    //对关闭文件进行判断
    if(-1 == closedir(dp))
    {
    printf("不能关闭%s
Error:%s",fullpath,strerror(errno));
    }

    return ret;
}
static int myfunc(const char * pathname,const struct stat * statptr,int type)
{
    switch(type)
    {
    case FTW_F:
    switch(statptr->st_mode & S_IFMT)
    {
    case S_IFSOCK:    sock_c++;    break;
    case S_IFLNK:    lnk_c++;    break;
    case S_IFREG:    reg_c++;    break;
    case S_IFBLK:    blk_c++;    break;
    case S_IFCHR:    chr_c++;    break;
    case S_IFIFO:    fifo_c++;    break;
    case S_IFDIR:
        printf("Error:这里不应该出现目录文件%s!

Error:%s
",pathname,strerror(errno));
        break;
    }
    break;
    case FTW_D:
    dir_c++;    break;
    case FTW_ND:
    printf("不能打开目录%s
Error:%s
",pathname,strerror(errno));
    break;
    case FTW_NS:
    printf("不能打开文件%s
Error:%s
",pathname,strerror(errno));
    break;
    }
    return 0;
}

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

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

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