c++ 遍历目录下的文件

#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;
}