c++ 遍历目录下的文件

小鱼儿 2022-12-25 14:54 223阅读 0赞

#include <sys/types.h>
    #include <sys/stat.h>
    #include <unistd.h>
    #include <dirent.h>
    #include <iostream>
    #include <string.h>
    #include <stdio.h>
    #include <fcntl.h>
    #include <stdlib.h>
    #include <vector>
    
    using namespace std;
    
    // 递归列出所有目录及文件
    void recursion_scan_dir_file(const char *dir, int depth=1)
    {
            DIR *p_dir = NULL;
            struct dirent *p_entry = NULL;
            struct stat statbuf;
    
            if((p_dir = opendir(dir)) == NULL)
            {
                    printf("can't open dir.\n");
                    return;
            }
    
            chdir (dir);
    
            while(NULL != (p_entry = readdir(p_dir))) { // 获取下一级目录信息
    
                    lstat(p_entry->d_name, &statbuf);   // 获取下一级成员属性
    
                    if(S_IFDIR & statbuf.st_mode) {      // 判断下一级成员是否是目录  
                            if (strcmp(".", p_entry->d_name) == 0 || strcmp("..", p_entry->d_name) == 0)
                                    continue;
    
                            printf("%*s%s/\n", depth, "", p_entry->d_name);
                            recursion_scan_dir_file(p_entry->d_name, depth+4); // 扫描下一级目录的内容
                    } else {
                            printf("%*s%s\n", depth, "", p_entry->d_name);  // 输出属性不是目录的成员
                    }
            }
            chdir(".."); // 回到上级目录  
            closedir(p_dir);
    }
    
    //列出一个目录下所有文件
    void scan_one_dir( const char * dir_name, std::vector<std::string> &files_vector)
    {  
        if( NULL == dir_name )
        {  
            cout<<" dir_name is null ! "<<endl;
            return;
        }
    
        struct stat s;  
        lstat( dir_name , &s );  
        if( ! S_ISDIR( s.st_mode ) )  
        {
            return;
        }
          
        struct dirent * filename;
        DIR * dir;
        dir = opendir( dir_name );  
        if( NULL == dir )  
        {  
            return;  
        }  
    
        int iName=0;
        while( ( filename = readdir(dir) ) != NULL )  
        {  
            if( strcmp( filename->d_name , "." ) == 0 ||
                strcmp( filename->d_name , "..") == 0)
                continue;
    
            char wholePath[128] = {0};
            sprintf(wholePath, "%s", filename->d_name);
            files_vector.push_back(wholePath);
        }
    }
    
     
    main(int argc, char* argv[])
    {
        string file_dir = "/home/";
        std::vector<std::string> files_vector;
    
        scan_one_dir(file_dir.c_str(), files_vector);
        recursion_scan_dir_file(file_dir.c_str(), 1);
    
        for (auto iters : files_vector)
        {
            cout << iters << std::endl;
        }
    
        return 0;
    }