枚举当前进程目录下的文件【改进版】

Inse

昨天那段代码，写得实在是烂啊，今天做了一下改进，麻烦管理员把原帖删了。原帖地址：http://bbs.fishc.com/thread-60923-1-1.html (出处: 鱼C论坛)

改进代码如下：

1. vector<wstring> EnumFilesEx()
   
2. {
   
3.         vector<wstring> vectFilePath;
   
4. 
   
5.         WCHAR szBaseFilePath[MAX_PATH] = {0}; //保存文件路径
   
6.         WCHAR szEnumFilePath[MAX_PATH] = {0}; //保存文件路径和格式
   
7.         WCHAR szAddFilePath[MAX_PATH] = {0};
   
8.         HANDLE hFile = INVALID_HANDLE_VALUE;
   
9.         WIN32_FIND_DATA mFileData = {0};
   
10.         HANDLE        hFileRead            =        0;
    
11. 
    
12.         // 获取当前进程路径
    
13.         if (0 == GetModuleFileName(NULL, szBaseFilePath, sizeof(szBaseFilePath)))
    
14.         {
    
15.                 vectFilePath.clear();
    
16.                 return std::move(vectFilePath);
    
17.         }
    
18.         PathRemoveFileSpec(szBaseFilePath);
    
19.         PathAppend(szBaseFilePath, L"\\resource");
    
20.         RtlCopyMemory(szEnumFilePath, szBaseFilePath, sizeof(szBaseFilePath));
    
21.         PathAppend(szEnumFilePath, L"\\*.txt");
    
22. 
    
23.         // 查找文件下的第一个文件
    
24.         hFile = FindFirstFile(szEnumFilePath, &mFileData);
    
25.         if (INVALID_HANDLE_VALUE == hFile)
    
26.         {
    
27.                 vectFilePath.clear();
    
28.                 return std::move(vectFilePath);
    
29.         }
    
30.         RtlCopyMemory(szAddFilePath, szBaseFilePath, sizeof(szBaseFilePath));
    
31.         PathAppend(szAddFilePath, mFileData.cFileName);
    
32.          vectFilePath.push_back(szAddFilePath);    // 将文件名保存在vector数组中
    
33. 
    
34.         // 循环枚举文件名称
    
35.         for (DWORD i = 0; FALSE != FindNextFile(hFile, &mFileData); i++ )
    
36.         {
    
37.                 RtlZeroMemory(szAddFilePath, sizeof(szAddFilePath));
    
38.                 RtlCopyMemory(szAddFilePath, szBaseFilePath, sizeof(szBaseFilePath));
    
39.                 PathAppend(szAddFilePath, mFileData.cFileName);
    
40.                 vectFilePath.push_back(szAddFilePath);
    
41.         }
    
42.        
    
43.         CloseHandle(hFileRead);
    
44.         return std::move(vectFilePath);
    
45. }

复制代码