1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
/*-
* Copyright (c) 2014-2015 MongoDB, Inc.
* Copyright (c) 2008-2014 WiredTiger, Inc.
* All rights reserved.
*
* See the file LICENSE for redistribution information.
*/
#include "wt_internal.h"
/* I'm sure we need to config this */
#include <dirent.h>
/*
* __wt_dirlist --
* Get a list of files from a directory, optionally filtered by
* a given prefix.
*/
int
__wt_dirlist(WT_SESSION_IMPL *session, const char *dir, const char *prefix,
uint32_t flags, char ***dirlist, u_int *countp)
{
struct dirent *dp;
DIR *dirp;
WT_DECL_RET;
size_t dirallocsz;
u_int count, dirsz;
int match;
char **entries, *path;
*dirlist = NULL;
*countp = 0;
WT_RET(__wt_filename(session, dir, &path));
dirp = NULL;
dirallocsz = 0;
dirsz = 0;
entries = NULL;
if (flags == 0)
LF_SET(WT_DIRLIST_INCLUDE);
WT_ERR(__wt_verbose(session, WT_VERB_FILEOPS,
"wt_dirlist of %s %s prefix %s",
path, LF_ISSET(WT_DIRLIST_INCLUDE) ? "include" : "exclude",
prefix == NULL ? "all" : prefix));
WT_SYSCALL_RETRY(((dirp = opendir(path)) == NULL ? 1 : 0), ret);
if (ret != 0)
WT_ERR_MSG(session, ret, "%s: opendir", path);
for (dirsz = 0, count = 0; (dp = readdir(dirp)) != NULL;) {
/*
* Skip . and ..
*/
if (strcmp(dp->d_name, ".") == 0 ||
strcmp(dp->d_name, "..") == 0)
continue;
match = 0;
if (prefix != NULL &&
((LF_ISSET(WT_DIRLIST_INCLUDE) &&
WT_PREFIX_MATCH(dp->d_name, prefix)) ||
(LF_ISSET(WT_DIRLIST_EXCLUDE) &&
!WT_PREFIX_MATCH(dp->d_name, prefix))))
match = 1;
if (prefix == NULL || match) {
/*
* We have a file name we want to return.
*/
count++;
if (count > dirsz) {
dirsz += WT_DIR_ENTRY;
WT_ERR(__wt_realloc_def(
session, &dirallocsz, dirsz, &entries));
}
WT_ERR(__wt_strdup(
session, dp->d_name, &entries[count-1]));
}
}
if (count > 0)
*dirlist = entries;
*countp = count;
err:
if (dirp != NULL)
(void)closedir(dirp);
__wt_free(session, path);
if (ret == 0)
return (0);
if (*dirlist != NULL) {
for (count = dirsz; count > 0; count--)
__wt_free(session, entries[count]);
__wt_free(session, entries);
}
WT_RET_MSG(session, ret, "dirlist %s prefix %s", dir, prefix);
}
|
