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

Inse

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

改进代码如下：

vector<wstring> EnumFilesEx()
{
        vector<wstring> vectFilePath;

        WCHAR szBaseFilePath[MAX_PATH] = {0}; //保存文件路径
        WCHAR szEnumFilePath[MAX_PATH] = {0}; //保存文件路径和格式 
        WCHAR szAddFilePath[MAX_PATH] = {0};
        HANDLE hFile = INVALID_HANDLE_VALUE;
        WIN32_FIND_DATA mFileData = {0};
        HANDLE        hFileRead            =        0;

        // 获取当前进程路径
        if (0 == GetModuleFileName(NULL, szBaseFilePath, sizeof(szBaseFilePath)))
        {
                vectFilePath.clear();
                return std::move(vectFilePath);
        }
        PathRemoveFileSpec(szBaseFilePath);
        PathAppend(szBaseFilePath, L"\\resource");
        RtlCopyMemory(szEnumFilePath, szBaseFilePath, sizeof(szBaseFilePath));
        PathAppend(szEnumFilePath, L"\\*.txt");

        // 查找文件下的第一个文件
        hFile = FindFirstFile(szEnumFilePath, &mFileData);
        if (INVALID_HANDLE_VALUE == hFile)
        {
                vectFilePath.clear();
                return std::move(vectFilePath);
        }
        RtlCopyMemory(szAddFilePath, szBaseFilePath, sizeof(szBaseFilePath));
        PathAppend(szAddFilePath, mFileData.cFileName);
         vectFilePath.push_back(szAddFilePath);    // 将文件名保存在vector数组中

        // 循环枚举文件名称
        for (DWORD i = 0; FALSE != FindNextFile(hFile, &mFileData); i++ )
        {
                RtlZeroMemory(szAddFilePath, sizeof(szAddFilePath));
                RtlCopyMemory(szAddFilePath, szBaseFilePath, sizeof(szBaseFilePath));
                PathAppend(szAddFilePath, mFileData.cFileName);
                vectFilePath.push_back(szAddFilePath);
        }
        
        CloseHandle(hFileRead);
        return std::move(vectFilePath);
}