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
96
97
98
99
100
101
102
103
104
105
106
107
|
/*-
* Copyright (c) 2014-2016 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_posix_directory_list --
* Get a list of files from a directory, POSIX version.
*/
int
__wt_posix_directory_list(WT_FILE_SYSTEM *file_system,
WT_SESSION *wt_session, const char *directory,
const char *prefix, char ***dirlistp, uint32_t *countp)
{
struct dirent *dp;
DIR *dirp;
WT_DECL_RET;
WT_SESSION_IMPL *session;
size_t dirallocsz;
uint32_t count;
int tret;
char **entries;
session = (WT_SESSION_IMPL *)wt_session;
*dirlistp = NULL;
*countp = 0;
dirp = NULL;
dirallocsz = 0;
entries = NULL;
WT_SYSCALL_RETRY(((dirp = opendir(directory)) == NULL ? -1 : 0), ret);
if (ret != 0)
WT_RET_MSG(session, ret,
"%s: directory-list: opendir", directory);
for (count = 0; (dp = readdir(dirp)) != NULL;) {
/*
* Skip . and ..
*/
if (strcmp(dp->d_name, ".") == 0 ||
strcmp(dp->d_name, "..") == 0)
continue;
/* The list of files is optionally filtered by a prefix. */
if (prefix != NULL && !WT_PREFIX_MATCH(dp->d_name, prefix))
continue;
WT_ERR(__wt_realloc_def(
session, &dirallocsz, count + 1, &entries));
WT_ERR(__wt_strdup(session, dp->d_name, &entries[count]));
++count;
}
*dirlistp = entries;
*countp = count;
err: if (dirp != NULL) {
WT_SYSCALL(closedir(dirp), tret);
if (tret != 0) {
__wt_err(session, tret,
"%s: directory-list: closedir", directory);
if (ret == 0)
ret = tret;
}
}
if (ret == 0)
return (0);
WT_TRET(__wt_posix_directory_list_free(
file_system, wt_session, entries, count));
WT_RET_MSG(session, ret,
"%s: directory-list, prefix \"%s\"",
directory, prefix == NULL ? "" : prefix);
}
/*
* __wt_posix_directory_list_free --
* Free memory returned by __wt_posix_directory_list.
*/
int
__wt_posix_directory_list_free(WT_FILE_SYSTEM *file_system,
WT_SESSION *wt_session, char **dirlist, uint32_t count)
{
WT_SESSION_IMPL *session;
WT_UNUSED(file_system);
session = (WT_SESSION_IMPL *)wt_session;
if (dirlist != NULL) {
while (count > 0)
__wt_free(session, dirlist[--count]);
__wt_free(session, dirlist);
}
return (0);
}
|