Windows及Linux系统中获取文件夹路径下的所有文件名

最新推荐文章于 2024-05-07 20:40:10 发布

。 7.

最新推荐文章于 2024-05-07 20:40:10 发布

阅读量244

点赞数 2

分类专栏： C++ 文章标签： linux 运维服务器

本文链接：https://blog.csdn.net/dui845819593/article/details/135946005

版权

C++ 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

Linux系统:

#include <string.h>
#include <iostream>
#include <sstream>
#include <fstream>
#include <dirent.h>

bool hasEnding(std::string const &fullString, std::string const &ending)
{
    if (fullString.length() >= ending.length()) {
        return (0 == fullString.compare (fullString.length() - ending.length(), ending.length(), ending));
    } else {
        return false;
    }
}

int getFolderFile(const char *path, std::vector<std::string>& files, const char *suffix = ".bin")
{
    DIR *dir;
    struct dirent *ent;
    if ((dir = opendir(path)) != NULL) {
        while ((ent = readdir (dir)) != NULL) {
            std::string file = ent->d_name;
            if(hasEnding(file, suffix)){
                files.push_back(file.substr(0, file.length()-4));
            }
        }
        closedir(dir);
    } else {
        printf("No such folder: %s.", path);
        exit(EXIT_FAILURE);
    }
    return EXIT_SUCCESS;
}

Windows系统：

#undef UNICODE
#include <windows.h>

#include <string.h>
#include <iostream>
#include <sstream>
#include <fstream>
#include <thread>
#include <filesystem>

int getFolderFile(const char* path, std::vector<std::string>& files, const char* suffix = ".bin")
{
    WIN32_FIND_DATA findData;
    char filePathName[256];
    strcpy(filePathName, path);
    strcat(filePathName, "/*");
    strcat(filePathName, suffix);

    //HANDLE hFind = FindFirstFile((path +"*" + suffix).cstr(), &findData);
    HANDLE hFind = FindFirstFile(filePathName, &findData);

    if (hFind == INVALID_HANDLE_VALUE)
    {
        printf("No such folder or no such files: %s\n", path);
        return EXIT_FAILURE;
    }
    do
    {
        std::string filename = findData.cFileName;

        size_t lasDotPos = filename.find_last_of(".");//查找最后一个点号位
        if (lasDotPos != std::string::npos)
            filename.erase(lasDotPos); //删除从最后一个点号到结尾的部分

        files.push_back(filename);
    } while (FindNextFile(hFind, &findData) != 0);
    FindClose(hFind);
    return EXIT_SUCCESS;
}