linux系统编程统计一个目录下的普通文件个数

主要是为了统计linux系统下一个指定目录下面的普通文件个数，运用目录操作的一些函数，配合递归调用来实现该功能。

首先介绍一下函数原型：

　　打开一个空目录
                    DIR *opendir(const char *name);
                    参数: 目录名
                    返回值: 指向目录的指针
        读目录
                    struct dirent *readdir(DIR *dirp);
                    参数: opendir的返回值
                    返回值: 指向目录的指针

　　　关闭目录
                    int closedir(DIR *dirp);

代码实现：readfileNum.c

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <dirent.h>
#include <sys/types.h>


int get_file_num(char* root)
{
    int total = 0;
    DIR* dir = NULL;
    // 打开目录
    dir = opendir(root);
    // 循环从目录中读文件

    char path[1024];
    // 定义记录目录指针
    struct dirent* ptr = NULL;
    while( (ptr = readdir(dir)) != NULL)
    {
        // 跳过. 和 ..
        if(strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0)
        {
            continue;
        }
        // 判断是不是目录
        if(ptr->d_type == DT_DIR)
        {
            sprintf(path, "%s/%s", root, ptr->d_name);
            // 递归读目录
            total += get_file_num(path);
        }
        // 如果是普通文件
        if(ptr->d_type == DT_REG)
        {
            total ++;
        }
    }
    closedir(dir);
    return total;
}

int main(int argc, char* argv[])
{
    if(argc < 2)
    {
        printf("./a.out path");
        exit(1);
    }

    int total = get_file_num(argv[1]);
    printf("%s has regfile number: %d\n", argv[1], total);
    return 0;
}

效果展示：