]>
git.ipfire.org Git - thirdparty/git.git/blob - dir-iterator.c
4 #include "dir-iterator.h"
6 struct dir_iterator_level
{
10 * The length of the directory part of path at this level
11 * (including a trailing '/'):
17 * The full data structure used to manage the internal directory
18 * iteration state. It includes members that are not part of the
21 struct dir_iterator_int
{
22 struct dir_iterator base
;
25 * The number of levels currently on the stack. After the first
26 * call to dir_iterator_begin(), if it succeeds to open the
27 * first level's dir, this will always be at least 1. Then,
28 * when it comes to zero the iteration is ended and this
33 /* The number of levels that have been allocated on the stack */
37 * A stack of levels. levels[0] is the uppermost directory
38 * that will be included in this iteration.
40 struct dir_iterator_level
*levels
;
44 * Push a level in the iter stack and initialize it with information from
45 * the directory pointed by iter->base->path. It is assumed that this
46 * strbuf points to a valid directory path. Return 0 on success and -1
47 * otherwise, leaving the stack unchanged.
49 static int push_level(struct dir_iterator_int
*iter
)
51 struct dir_iterator_level
*level
;
53 ALLOC_GROW(iter
->levels
, iter
->levels_nr
+ 1, iter
->levels_alloc
);
54 level
= &iter
->levels
[iter
->levels_nr
++];
56 if (!is_dir_sep(iter
->base
.path
.buf
[iter
->base
.path
.len
- 1]))
57 strbuf_addch(&iter
->base
.path
, '/');
58 level
->prefix_len
= iter
->base
.path
.len
;
60 level
->dir
= opendir(iter
->base
.path
.buf
);
62 if (errno
!= ENOENT
) {
63 warning_errno("error opening directory '%s'",
74 * Pop the top level on the iter stack, releasing any resources associated
75 * with it. Return the new value of iter->levels_nr.
77 static int pop_level(struct dir_iterator_int
*iter
)
79 struct dir_iterator_level
*level
=
80 &iter
->levels
[iter
->levels_nr
- 1];
82 if (level
->dir
&& closedir(level
->dir
))
83 warning_errno("error closing directory '%s'",
87 return --iter
->levels_nr
;
91 * Populate iter->base with the necessary information on the next iteration
92 * entry, represented by the given dirent de. Return 0 on success and -1
95 static int prepare_next_entry_data(struct dir_iterator_int
*iter
,
98 strbuf_addstr(&iter
->base
.path
, de
->d_name
);
100 * We have to reset these because the path strbuf might have
101 * been realloc()ed at the previous strbuf_addstr().
103 iter
->base
.relative_path
= iter
->base
.path
.buf
+
104 iter
->levels
[0].prefix_len
;
105 iter
->base
.basename
= iter
->base
.path
.buf
+
106 iter
->levels
[iter
->levels_nr
- 1].prefix_len
;
108 if (lstat(iter
->base
.path
.buf
, &iter
->base
.st
)) {
110 warning_errno("failed to stat '%s'", iter
->base
.path
.buf
);
117 int dir_iterator_advance(struct dir_iterator
*dir_iterator
)
119 struct dir_iterator_int
*iter
=
120 (struct dir_iterator_int
*)dir_iterator
;
122 if (S_ISDIR(iter
->base
.st
.st_mode
)) {
123 if (push_level(iter
) && iter
->levels_nr
== 0) {
124 /* Pushing the first level failed */
125 return dir_iterator_abort(dir_iterator
);
129 /* Loop until we find an entry that we can give back to the caller. */
132 struct dir_iterator_level
*level
=
133 &iter
->levels
[iter
->levels_nr
- 1];
135 strbuf_setlen(&iter
->base
.path
, level
->prefix_len
);
137 de
= readdir(level
->dir
);
141 warning_errno("error reading directory '%s'",
142 iter
->base
.path
.buf
);
143 else if (pop_level(iter
) == 0)
144 return dir_iterator_abort(dir_iterator
);
148 if (is_dot_or_dotdot(de
->d_name
))
151 if (prepare_next_entry_data(iter
, de
))
158 int dir_iterator_abort(struct dir_iterator
*dir_iterator
)
160 struct dir_iterator_int
*iter
= (struct dir_iterator_int
*)dir_iterator
;
162 for (; iter
->levels_nr
; iter
->levels_nr
--) {
163 struct dir_iterator_level
*level
=
164 &iter
->levels
[iter
->levels_nr
- 1];
166 if (level
->dir
&& closedir(level
->dir
)) {
167 int saved_errno
= errno
;
168 strbuf_setlen(&iter
->base
.path
, level
->prefix_len
);
170 warning_errno("error closing directory '%s'",
171 iter
->base
.path
.buf
);
176 strbuf_release(&iter
->base
.path
);
181 struct dir_iterator
*dir_iterator_begin(const char *path
)
183 struct dir_iterator_int
*iter
= xcalloc(1, sizeof(*iter
));
184 struct dir_iterator
*dir_iterator
= &iter
->base
;
187 strbuf_init(&iter
->base
.path
, PATH_MAX
);
188 strbuf_addstr(&iter
->base
.path
, path
);
190 ALLOC_GROW(iter
->levels
, 10, iter
->levels_alloc
);
194 * Note: stat already checks for NULL or empty strings and
197 if (stat(iter
->base
.path
.buf
, &iter
->base
.st
) < 0) {
202 if (!S_ISDIR(iter
->base
.st
.st_mode
)) {
203 saved_errno
= ENOTDIR
;
210 dir_iterator_abort(dir_iterator
);