如何获取目录中的所有文件名？

Question

我正在用 C++ 编写程序。我正在尝试获取程序可执行文件所在文件夹中的所有文件并将它们存储在向量中。我被告知下面的代码应该可以工作，但是 FindFirstFile 操作只能找到一个文件（它应该搜索的文件夹的名称）。如何更改代码以便正确查看文件夹？

std::vector<char*> fileArray;

//Get location of program executable
HMODULE hModule = GetModuleHandleW(NULL);
WCHAR path[MAX_PATH];
GetModuleFileNameW(hModule, path, MAX_PATH);

//Remove the executable file name from 'path' so that it refers to the folder instead
PathCchRemoveFileSpec(path, sizeof(path));

//This code should find the first file in the executable folder, but it fails
//Instead, it obtains the name of the folder that it is searching
WIN32_FIND_DATA ffd;
HANDLE hFind = INVALID_HANDLE_VALUE;
hFind = FindFirstFile(path, &ffd);

do
{
    //The name of the folder is pushed onto the array because of the previous code's mistake
    //e.g. If the folder is "C:\\MyFolder", it stores "MyFolder"
    fileArray.push_back(ffd.cFileName); //Disclaimer: This line of code won't add the file name properly (I'll get to fixing it later), but that's not relevant to the question
} while (FindNextFile(hFind, &ffd) != 0); //This line fails to find anymore files

Answer 1

有人告诉我下面的代码应该可以工作

你被告知错了，因为你提供的代码被严重破坏了。

FindFirstFile 操作只找到一个文件（它应该搜索的文件夹的名称）。

您仅将文件夹路径本身传递给FindFirstFile()，因此只会报告一个条目，描述文件夹本身。您需要做的是在路径末尾附加一个* 或*.* 通配符，然后FindFirstFile()/FindNextFile() 将枚举文件夹内的文件和子文件夹。

除此之外，代码还有其他几个问题。

即使枚举工作正常，您也不会区分文件和子文件夹。

您将错误的值传递给PathCchRemoveFileSpec() 的第二个参数。您正在传递一个 byte 计数，但它需要一个 character 计数。

在进入循环之前，您没有检查 FindFirstFile() 是否失败，并且在循环完成后您没有调用 FindClose()。

最后，您的代码甚至无法按原样编译。您的vector 存储char* 值，但是为了将WCHAR[] 传递给FindFirstFile()，必须定义UNICODE，这意味着FindFirstFile() 将映射到FindFirstFileW() 和WIN32_FIND_DATA 将映射到WIN32_FIND_DATAW，因此cFileName 字段将是WCHAR[]。编译器不允许将WCHAR[]（或WCHAR*) 分配给char*。

试试类似的方法：

std::vector<std::wstring> fileArray;

//Get location of program executable
WCHAR path[MAX_PATH+1] = {0};
GetModuleFileNameW(NULL, path, MAX_PATH);

//Remove the executable file name from 'path' so that it refers to the folder instead
PathCchRemoveFileSpec(path, MAX_PATH);

// append a wildcard to the path
PathCchAppend(path, MAX_PATH, L"*.*");

WIN32_FIND_DATAW ffd;
HANDLE hFind = FindFirstFileW(path, &ffd);
if (hFile != INVALID_HANDLE_VALUE)
{
    do
    {
        if ((ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == 0)
            fileArray.push_back(ffd.cFileName);
    }
    while (FindNextFileW(hFind, &ffd));
    FindClose(hFind);
}