C++文件体系操作5 - 跨平台列出指定目录下的所有文件和文件夹 ...

• 1. 关键词
  
• 2. fileutil.h
  
• 3. fileutil.cpp
  
• 4. filesystem_win.h
  
• 5. filesystem_win.cpp
  
• 6. filesystem_unix.cpp
  
• 7. 源码地点
  

1. 关键词

C++ 文件体系操作 列出指定目录下的所有文件 列出指定目录下的所有文件夹 跨平台
2. fileutil.h

1. #pragma once
3. #include <string>
4. #include <cstdio>
5. #include <cstdint>
6. #include "filetype.h"
7. #include "filepath.h"
9. namespace cutl
10. {
12.     /**
13.      * @brief The file guard class to manage the FILE pointer automatically.
14.      * file_guard object can close the FILE pointer automatically when his scope is exit.
15.      */
16.     class file_guard
17.     {
18.     public:
19.         /**
20.          * @brief Construct a new file guard object
21.          *
22.          * @param file the pointer of the FILE object
23.          */
24.         explicit file_guard(FILE *file);
26.         /**
27.          * @brief Destroy the file guard object
28.          *
29.          */
30.         ~file_guard();
32.         /**
33.          * @brief Get the FILE pointer.
34.          *
35.          * @return FILE*
36.          */
37.         FILE *getfd() const;
39.     private:
40.         FILE *file_;
41.     };
43.     /**
44.      * @brief List all files in a directory.
45.      *
46.      * @param dirpath the filepath of the directory to be listed
47.      * @param type file type to be listed, default is all types.
48.      * @param recursive whether to list the whole directory recursively, default is false.
49.      * If true, means list the whole directory recursively, like the 'ls -R' command.
50.      * @return filevec the vector of file_entity, filepaths in the directory.
51.      */
52.     filevec list_files(const filepath &dirpath, filetype type = filetype::all, bool recursive = false);
54. } // namespace cutl

复制代码

3. fileutil.cpp

1. #include <cstdio>
2. #include <map>
3. #include <iostream>
4. #include <cstring>
5. #include <sys/stat.h>
6. #include "fileutil.h"
7. #include "inner/logger.h"
8. #include "inner/filesystem.h"
9. #include "strutil.h"
11. namespace cutl
12. {
13.     file_guard::file_guard(FILE *file)
14.         : file_(file)
15.     {
16.     }
18.     file_guard::~file_guard()
19.     {
20.         if (file_)
21.         {
22.             // CUTL_DEBUG("close file");
23.             int ret = fclose(file_);
24.             if (ret != 0)
25.             {
26.                 CUTL_ERROR("fail to close file, ret" + std::to_string(ret));
27.             }
28.             file_ = nullptr;
29.         }
30.         // ROBOLOG_DCHECK(file_ == nullptr);
31.     }
33.     FILE *file_guard::getfd() const
34.     {
35.         return file_;
36.     }
38.     filevec list_files(const filepath &dirpath, filetype type, bool recursive)
39.     {
40.         return list_sub_files(dirpath.str(), type, recursive);
41.     }

复制代码

4. filesystem_win.h

1. #include <vector>
2. #include <string>
4. #pragma once
6. namespace cutl
7. {
8.     filetype get_file_type(const std::string &filepath);
9.     filevec list_sub_files(const std::string &dirpath, filetype type = filetype::all, bool recursive = false);
10. } // namespace cutl

复制代码

5. filesystem_win.cpp

1. #if defined(_WIN32) || defined(__WIN32__)
3. #include <io.h>
4. #include <direct.h>
5. #include <Windows.h>
6. #include <stdlib.h>
7. #include "strutil.h"
8. #include "filesystem.h"
9. #include "logger.h"
11. namespace cutl
12. {
13.     filetype get_file_type(DWORD attributes, const std::string &extension)
14.     {
15.         filetype type = filetype::unknown;
17.         if (attributes == INVALID_FILE_ATTRIBUTES)
18.         {
19.             CUTL_WARN("Failed to get file attributes, error code: " + std::to_string(GetLastError()));
20.             if (extension == ".lnk")
21.             {
22.                 // 注意：测试时发现，有些快捷方式访问会失败，用后缀名判断进行兜底
23.                 type = filetype::symlink;
24.             }
25.             return type;
26.         }
27.         else
28.         {
29.             if (attributes & FILE_ATTRIBUTE_DIRECTORY)
30.             {
31.                 type = filetype::directory; // directory
32.             }
33.             else if (attributes & FILE_ATTRIBUTE_NORMAL || attributes & FILE_ATTRIBUTE_READONLY)
34.             {
35.                 // 普通文件|只读文件
36.                 type = filetype::file; // regular file
37.             }
38.             else if ((attributes & FILE_ATTRIBUTE_ARCHIVE) && extension == ".lnk")
39.             {
40.                 // windows的快捷方式
41.                 type = filetype::symlink; // symbolic link
42.             }
43.         }
45.         return type;
46.     }
48.     std::string get_file_extension(const std::string &filepath)
49.     {
50.         auto pos = filepath.find_last_of('.');
51.         std::string extension = "";
52.         if (pos != std::string::npos)
53.         {
54.             extension = filepath.substr(pos);
55.             extension = cutl::to_lower(extension);
56.         }
57.         return extension;
58.     }
60.     // https://learn.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-getfileattributesa
61.     // https://learn.microsoft.com/en-us/windows/win32/fileio/file-attribute-constants
62.     filetype get_file_type(const std::string &filepath)
63.     {
64.         auto attributes = GetFileAttributesA(filepath.c_str());
65.         auto extension = get_file_extension(filepath);
66.         // CUTL_DEBUG(filepath + ", extension: " + extension + ", attributes: " + std::to_string(attributes));
67.         return get_file_type(attributes, extension);
68.     }
70.     filevec list_sub_files(const std::string &dirpath, filetype type, bool recursive)
71.     {
72.         filevec file_list;
74.         // 使用FindFirstFileA查找第一个文件
75.         auto findpath = dirpath + win_separator + "*.*";
76.         WIN32_FIND_DATAA findData = {0};
77.         HANDLE hFind = FindFirstFileA(findpath.c_str(), &findData);
78.         if (hFind == INVALID_HANDLE_VALUE || hFind == NULL)
79.         {
80.             CUTL_ERROR("FindFirstFileA failed for " + findpath + ", errCode: " + std::to_string(GetLastError()));
81.             return file_list;
82.         }
84.         do
85.         {
86.             auto dwAttrs = findData.dwFileAttributes;
87.             auto filename = std::string(findData.cFileName);
88.             CUTL_DEBUG(filename + ", attributes: " + std::to_string(dwAttrs));
89.             if (is_special_dir(filename))
90.             {
91.                 // “..”和“.”不做处理
92.                 continue;
93.             }
94.             std::string filepath = dirpath + win_separator + filename;
95.             auto extension = get_file_extension(filename);
96.             auto ftype = get_file_type(dwAttrs, extension);
97.             if (ftype & type)
98.             {
99.                 file_entity entity;
100.                 entity.type = ftype;
101.                 entity.filepath = filepath;
102.                 file_list.emplace_back(entity);
103.             }
104.             if ((dwAttrs & FILE_ATTRIBUTE_DIRECTORY) && recursive)
105.             {
106.                 // directory
107.                 auto sub_files = list_sub_files(filepath, type, recursive);
108.                 if (!sub_files.empty())
109.                 {
110.                     file_list.insert(file_list.end(), sub_files.begin(), sub_files.end());
111.                 }
112.             }
113.         } while (FindNextFileA(hFind, &findData));
114.         // 关闭句柄
115.         FindClose(hFind);
117.         return file_list;
118.     }
119. } // namespace cutl
121. #endif // defined(_WIN32) || defined(__WIN32__)

复制代码

6. filesystem_unix.cpp

1. #if defined(_WIN32) || defined(__WIN32__)
2. // do nothing
3. #else
5. #include <unistd.h>
6. #include <sys/stat.h>
7. #include <dirent.h>
8. #include <stack>
9. #include <cstring>
10. #include <utime.h>
11. #include <stdlib.h>
12. #include <sys/time.h>
13. #include "filesystem.h"
14. #include "inner/logger.h"
16. namespace cutl
17. {
18.     filetype get_file_type(const std::string &filepath)
19.     {
20.         struct stat file_stat;
21.         int ret = lstat(filepath.c_str(), &file_stat);
22.         if (0 != ret)
23.         {
24.             CUTL_ERROR("stat error. filepath:" + filepath + ", error:" + strerror(errno));
25.             return filetype::unknown;
26.         }
28.         return get_file_type(file_stat.st_mode);
29.     }
30.     filevec list_sub_files(const std::string &dirpath, filetype type, bool recursive)
31.     {
32.         filevec file_list;
34.         DIR *dir = opendir(dirpath.c_str()); // 打开这个目录
35.         if (dir == NULL)
36.         {
37.             CUTL_ERROR("opendir error. dirpath:" + dirpath + ", error:" + strerror(errno));
38.             return file_list;
39.         }
40.         struct dirent *file_info = NULL;
41.         // 逐个读取目录中的文件到file_info
42.         while ((file_info = readdir(dir)) != NULL)
43.         {
44.             // 系统有个系统文件，名为“..”和“.”,对它不做处理
45.             std::string filename(file_info->d_name);
46.             if (is_special_dir(filename))
47.             {
48.                 continue;
49.             }
50.             struct stat file_stat; // 文件的信息
51.             std::string filepath = dirpath + unix_separator + filename;
52.             int ret = lstat(filepath.c_str(), &file_stat);
53.             if (0 != ret)
54.             {
55.                 CUTL_ERROR("stat error. filepath:" + filepath + ", error:" + strerror(errno));
56.                 closedir(dir);
57.                 return file_list;
58.             }
59.             auto ftype = get_file_type(file_stat.st_mode);
60.             if (ftype & type)
61.             {
62.                 file_entity entity;
63.                 entity.type = ftype;
64.                 entity.filepath = filepath;
65.                 file_list.emplace_back(entity);
66.             }
68.             if (S_ISDIR(file_stat.st_mode) && recursive)
69.             {
70.                 auto sub_files = list_sub_files(filepath, type, recursive);
71.                 if (!sub_files.empty())
72.                 {
73.                     file_list.insert(file_list.end(), sub_files.begin(), sub_files.end());
74.                 }
75.             }
76.         }
77.         closedir(dir);
79.         return file_list;
80.     }
81. } // namespace cutl
83. #endif // defined(_WIN32) || defined(__WIN32__)

复制代码

7. 源码地点

更多详细代码，请检察本人写的C++ 通用工具库: common_util, 本项目已开源，代码简洁，且有详细的文档和Demo。
【SunLogging】

扫码二维码，关注微信公众号，阅读更多出色内容
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。