C++遍历目录下所有文件(代码一):
#include <iostream>
#include <io.h>
#include <string>
#include <vector>
void getFiles(const std::string & path, std::vector<std::string> & files)
{
//文件句柄
long hFile = 0;
//文件信息,_finddata_t需要io.h头文件
struct _finddata_t fileinfo;
std::string p;
if((hFile = _findfirst(p.assign(path).append("\\*").c_str(),&fileinfo)) != -1)
{
do
{
//如果是目录,迭代之
//如果不是,加入列表
if((fileinfo.attrib & _A_SUBDIR))
{
if(strcmp(fileinfo.name,".") != 0 && strcmp(fileinfo.name,"..") != 0)
getFiles(p.assign(path).append("\\").append(fileinfo.name), files);
}
else
{
files.push_back(p.assign(path).append("\\").append(fileinfo.name));
}
} while (_findnext(hFile, &fileinfo) == 0);
_findclose(hFile);
}
}
int main()
{
std::vector<std::string> filenames;
std::string path;
std::getline(std::cin, path);
getFiles(path, filenames);
for (auto file : filenames)
{
std::cout << file << std::endl;
}
system("pause");
return 0;
}
以上代码会出现中断的bug
错误原因:_findfirst()返回类型为intptr_t而非long型,从“intptr_t”转换到“long”丢失了数据
解决方案:将long改为intptr_t即可
C++遍历文件夹下的所有文件及其子文件夹中的所有文件:
// 遍历指定文件夹下的所有文件(包括子文件夹)
bool GetFolderFileYolo(const string FloderPath, vector<string>& FileName, string subDir)
{
intptr_t handle;
_finddata_t findData;
string dirPath = FloderPath;
dirPath += "*.*";
handle = _findfirst(dirPath.c_str(), &findData); // 查找目录中的第一个文件
if (handle == -1)
{
cout << "Failed to find first file!\n";
return false;
}
do
{
if (findData.attrib == _A_SUBDIR)
{
if (strcmp(findData.name, ".") == 0
|| strcmp(findData.name, "..") == 0)
continue;
else
{
// cout << findData.name << "\t<dir>\n";
string dirName = findData.name;
dirName += "\\";
string subFolder = subDir + dirName;
GetFolderFileYolo(FloderPath + dirName, FileName, subFolder);
}
}
else
{
// cout << findData.name << "\t" << findData.size << endl;
FileName.push_back(subDir + findData.name);
}
} while (_findnext(handle, &findData) == 0); // 查找目录中的下一个文件
// cout << "Done!\n";
_findclose(handle); // 关闭搜索句柄
}
参考文章(亲测有效):