C++ 获取目录和文件列表

////////////////////////////////////////////////////

//获取目录路径下面所有文件的内容

//适用于linux和windows系统

////////////////////////////////////////////////////

#include <iostream>

#include <fstream>

#include <vector>

#include <string>

#ifdef _WIN32

#include <Windows.h>

#include <strsafe.h>

#else

#include <dirent.h>

#endif

 

using namespace std;

 

//////////////////////////////////////////////////////////////////////////

//获取指定目录下所有文件的文件名，不包括文件夹，在GetFileFromDir中使用

//strDir: 输入，目录路径

//vFileDirList： 输出，文件路径列表

//返回：空

//////////////////////////////////////////////////////////////////////////

void GetFileNameFromDir(string strDir,vector<string>& vFileDirList)

{

#ifdef _WIN32

WIN32_FIND_DATAA ffd;

LARGE_INTEGER filesize;

string szDir;

//size_t length_of_arg;

HANDLE hFind = INVALID_HANDLE_VALUE;

DWORD dwError=0;

 

szDir=strDir+"\\*";

hFind = FindFirstFileA(szDir.c_str(), &ffd);

 

if (INVALID_HANDLE_VALUE == hFind) 

{

cout<<"get file name error"<<endl;

return;

}

do

{

if (!(ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))

{

string filename=ffd.cFileName;//(const char*)

string filedir=strDir+"\\"+filename;

vFileDirList.push_back(filedir);

}

}while (FindNextFileA(hFind, &ffd) != 0);

 

dwError = GetLastError();

if (dwError != ERROR_NO_MORE_FILES) 

{

cout<<"FindFirstFile error"<<endl;

return;

}

FindClose(hFind);

#else

DIR *dir;

struct dirent *ptr;

dir = opendir(strDir.c_str());

while( (ptr = readdir(dir)) != NULL)

{

string path = strDir + string("/") + string(ptr->d_name);

vFileDirList.push_back(path);

}

closedir(dir);

sort(vFileDirList.begin(), vFileDirList.end());

#endif

}

 

//////////////////////////////////////////////////////////////////////////

//获取指定目录下所有文件内容

//strDir: 输入，目录路径

//vFileNameList： 输出，文件内容列表

//返回：空

//////////////////////////////////////////////////////////////////////////

void GetFileFromDir(string strDir,vector<string>& vFileList)

{

//获取目录路径下所有文件的文件名

vector<string> vFileDirList;

GetFileNameFromDir(strDir,vFileDirList);

//读取文件

vector<string>::iterator iterFileDir;

for (iterFileDir=vFileDirList.begin();iterFileDir!=vFileDirList.end();iterFileDir++)

{

ifstream infile((*iterFileDir).c_str());

if (!infile)

{

cout<<"read file error"<<endl;

return;

}

string line="";

string file="";

while (getline(infile,line))

{

file+=line;

}

vFileList.push_back(file);

}

}

 

int main(int argc,char* argv[])

{

if (argc!=2)

{

cout<<"paramater error"<<endl;

return -1;

}

else

{

string dir(argv[1]);

vector<string> files;

GetFileFromDir(dir,files);

vector<string>::iterator iterFile;

for (iterFile=files.begin();iterFile!=files.end();iterFile++)

{

cout<<*iterFile<<endl;

}

cout<<endl;

system("pause");

return 0;

}

}