C++ 查找文件夹下的文件
#include <string> #include <vector> #include <cstring> #include <cstdio> #include <dirent.h> #include <sys/stat.h> #include <sys/types.h> using std::strcmp; using std::string; using std::vector; #if __cplusplus < 201103L #define nullptr NULL #endif #define SEARCH_PATH_LENGTH 1024 /* * 搜索一个文件夹下的符合条件的文件 */ bool searchFile(const char* directory, vector<string>& files, const char* ext, int subdir_level) { const char* dir_path = directory; DIR *dir = nullptr; //打开文件夹 if((dir = opendir(dir_path)) == nullptr) { return false; } struct dirent entry; struct dirent* result; struct stat file_stat; char cur_name[SEARCH_PATH_LENGTH]; //读取文件夹下的一个文件 while(readdir_r(dir, &entry, &result) == 0 && result) { const char* name = entry.d_name; //忽略掉.和.. if(strcmp(name, ".") == 0 || strcmp(name, "..") == 0) continue; //检查拼接路径 int count = sprintf(cur_name, "%s", dir_path); count += sprintf(&cur_name[count], "%s%s", cur_name[count - 1] == ‘/‘ ? "" : "/", name); //获取文件的状态 if(lstat(cur_name, &file_stat) != -1) { //文件是普通文件 if(!S_ISDIR(file_stat.st_mode)) { const char* rf = strrchr(cur_name, ‘.‘); //检查文件的后缀名 if(rf != nullptr && strcmp(rf + 1, ext) == 0) files.push_back(string(cur_name, count)); } //文件是文件夹 else if(subdir_level >= 0) { //递归获取下一层目录的文件 searchFile(cur_name, files, subdir_level - 1); } } } closedir(dir); return true; }
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。