Linux系统编程：获取指定目录下指定文件类型的数量

古城微笑少年丶 2022-05-28 21:07 306阅读 0赞

# 分析 #

编程实现对指定目录下的指定文件类型数量的统计。这里涉及到对目录文件的打开和对目录文件下的文件的遍历。

这里主要用的2个库函数，**打开目录的 opendir函数，readdir函数**。对这2个库函数不熟悉？请看man 3 readdir,man 3 opendir，非常详细。**opendir函数 打开失败的时候 返回NULL ,errorno被设置，可以通过perror获取错误信息。readdir函数会遍历目录下的所有文件，注意是一个一个的返回，注意是读一次返回一次，返回的struct dirent 结构体指针，里面包含文件的类型、文件的名称、文件大小等 文件信息**。

## opendir函数说明 ##

![Image 1][]

## readdir函数说明 ##

![Image 1][]

# 实现代码 #

#include <stdio.h>
    #include <sys/types.h>
    #include <dirent.h>
    #include <string.h>
    char* fileType = NULL;
     // [-,d,p,s,l,c,b] 转化为 对应的 文件类型
    unsigned char GetFileType(char* type)
    {
        if(strcmp(type,"b")==0)
        {
            return DT_BLK;
        }else if(strcmp(type,"c")==0)
        {
            return DT_CHR;
        }else if(strcmp(type,"d")==0)
        {
            return DT_DIR;
        }else if(strcmp(type,"p")==0)
        {
            return DT_FIFO;
        }else if(strcmp(type,"l")==0)
        {
            return DT_LNK;
        }else if(strcmp(type,"-")==0)
        {
            return DT_REG;
        }else if(strcmp(type,"s")==0)
        {
            return DT_SOCK;
        }else
        {
            printf("文件类型输入错误,文件类型为[-,d,p,s,l,c,b]\n");
            exit(1);
        }
        
    }
    
    int GetFileNum(const char* path)
    {
        int total = 0;
        DIR* dir = opendir(path);
        if(dir==NULL)
        {
            perror("opendir error");
            exit(1);
        }
        struct dirent* dirObj = NULL;
        char* dirName = NULL;
        //循环读取目录下的文件
        while(dirObj = readdir(dir))
        {
            dirName = dirObj->d_name;
            if(strcmp(dirName,".")==0 || strcmp(dirName,"..")==0)
            {
                continue;
            }
            if(dirObj->d_type == GetFileType(fileType) )
            {
                total++;
            }
            if(dirObj->d_type==DT_DIR)
            {
                char temp[1024] = {0};
                sprintf(temp,"%s/%s",path,dirName);
                //碰到目录文件进行递归遍历
               total+= GetFileNum(temp);
            }
        }
    
        return total;
    }
    int main(int argc,int* argv[])
    {
        if(argc<3)
        {
            printf("./GetFileNum path fileType");
            exit(0);
        }
        char* path = argv[1];
        fileType = argv[2];
        int num = GetFileNum(path);
        printf("%s目录下有%s文件:%d\n",path,fileType,num);
        return 0;
    }

# 运行结果 #

![Image 1][]

[Image 1]: