本文介绍如何在Linux系统中使用C语言高效筛选遍历目录。 opendir 和 readdir 函数是目录遍历的常用工具，但若需筛选特定类型文件，则需额外处理。以下代码示例演示如何实现这一功能：
代码示例:
#include <stdio.h>
#include <stdlib.h>
#include <dirent.h>
#include <string.h>
#include <sys/stat.h>
// 函数：检查文件类型是否匹配
int checkFileType(const char *path, const char *type) {
if (strcmp(type, "all") == 0) return 1; // 不筛选
char ext[10];
const char *dot = strrchr(path, '.');
if (dot && dot > path) {
size_t len = dot - path;
if (len >= sizeof(ext)) len = sizeof(ext) - 1;
strncpy(ext, dot + 1, len);
ext[len] = '\0';
} else {
strcpy(ext, "");
}
return strcmp(ext, type) == 0;
}
// 函数：递归遍历目录并筛选
void traverseDir(const char *path, const char *filter) {
DIR *dir = opendir(path);
if (dir == NULL) {
perror("opendir");
return;
}
struct dirent *entry;
while ((entry = readdir(dir)) != NULL) {
if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) continue;
char fullPath[PATH_MAX];
snprintf(fullPath, sizeof(fullPath), "%s/%s", path, entry->d_name);
struct stat st;
if (stat(fullPath, &st) == -1) {
perror("stat");
continue;
}
if (S_ISDIR(st.st_mode)) {
traverseDir(fullPath, filter);
} else if (S_ISREG(st.st_mode)) {
if (checkFileType(fullPath, filter)) {
printf("文件: %s\n", fullPath);
}
}
}
closedir(dir);
}
int main(int argc, char *argv[]) {
if (argc != 3) {
fprintf(stderr, "用法: %s <目录路径> <过滤器>\n", argv[0]);
fprintf(stderr, "过滤器示例:\n");
fprintf(stderr, " all - 所有文件\n");
fprintf(stderr, " txt - .txt文件\n");
fprintf(stderr, " jpg,png - .jpg或.png文件\n");
return EXIT_FAILURE;
}
const char *directory = argv[1];
const char *filter = argv[2];
traverseDir(directory, filter);
return EXIT_SUCCESS;
}
代码说明:
-
checkFileType函数：检查文件扩展名是否与过滤器匹配，支持 "all" (不筛选)，以及多个扩展名用逗号分隔的情况。 -
traverseDir函数：递归遍历目录，使用stat获取文件信息，判断文件类型并调用checkFileType进行筛选。 -
main函数：处理命令行参数，调用traverseDir开始遍历。
编译和运行:
- 将代码保存为
dir_traverse.c。 - 编译：
gcc -o dir_traverse dir_traverse.c - 运行：
./dir_traverse /path/to/directory txt(将/path/to/directory替换为实际路径，txt为过滤器)
改进方向:
- 更强大的过滤器：支持更复杂的模式匹配 (例如正则表达式)。
- 错误处理：更完善的错误处理机制。
- 性能优化：对于大型目录，考虑多线程或异步IO提高效率。
这个改进后的示例提供了更清晰的代码结构和更强大的筛选功能，方便用户根据需求进行调整和扩展。 Python实现方式类似，但可以使用 os.walk 和 fnmatch 模块简化代码。
