获取目录中所有含有特定后缀名的文件

#include <stdio.h>
#include <stdlib.h>
#include <dirent.h>
#include <sys/types.h>
#include <string.h>

#ifndef oj_dir
#define oj_dir "/home/rain/oj"
#endif

//取出目录中.in文件
//参数中,dir是要查询的目录,in_files数组保存每个.in文件,函数返回所有.in文件个数
int printdir(char *dir, char* in_files[])
{
DIR *dp;
struct dirent *entry;

if ((dp = opendir(dir)) == NULL)
{
fprintf(stderr, "cannot open directory: %s\n", dir);
return 0;
}
chdir(dir);
int in_file_num = 0;
char dir_prefix[20], dir_suffix[20];
while((entry = readdir(dp)) != NULL)
{
/* ignore . and .. ,分别代表当前目录和上一级目录*/
if (strcmp(".", entry->d_name) == 0 ||  strcmp("..", entry->d_name) == 0)
{
continue;
}
sscanf(entry->d_name, "%[^.].%[^.]", dir_prefix, dir_suffix);
if (strcmp("in", dir_suffix) == 0)
{
in_files[in_file_num ++] = entry->d_name;
}
}
chdir(oj_dir);
closedir(dp);
return in_file_num;
}

int main(int argc, char const *argv[])
{
char* in_files[10];
int in_file_nums = printdir("/home/rain/data/1001", in_files);
printf("%d\n", in_file_nums);
int i;
for (i = 0; i < in_file_nums; ++i)
{
printf("%s\n", in_files[i]);
}
return 0;
}

目录如下:



rain@HY:~/oj$ gcc get_file_with_same_suffix.c -o get_file_with_same_suffix

rain@HY:~/oj$ ./get_file_with_same_suffix

4

data.in

data3.in

data4.in

data2.in