Commit 9e94b6af by Edward Thomson

iterator: cleanups with symlink dir handling

Perform some error checking when examining symlink directories.
parent e9628e7b
...@@ -23,7 +23,7 @@ ...@@ -23,7 +23,7 @@
#define iterator__has_been_accessed(I) iterator__flag(I,FIRST_ACCESS) #define iterator__has_been_accessed(I) iterator__flag(I,FIRST_ACCESS)
#define iterator__honor_ignores(I) iterator__flag(I,HONOR_IGNORES) #define iterator__honor_ignores(I) iterator__flag(I,HONOR_IGNORES)
#define iterator__ignore_dot_git(I) iterator__flag(I,IGNORE_DOT_GIT) #define iterator__ignore_dot_git(I) iterator__flag(I,IGNORE_DOT_GIT)
#define iterator__symlinksdir(I) iterator__flag(I,INCLUDE_SYMLINK_REFSDIR) #define iterator__descend_symlinks(I) iterator__flag(I,DESCEND_SYMLINKS)
static void iterator_set_ignore_case(git_iterator *iter, bool ignore_case) static void iterator_set_ignore_case(git_iterator *iter, bool ignore_case)
...@@ -1492,29 +1492,41 @@ static int filesystem_iterator_current( ...@@ -1492,29 +1492,41 @@ static int filesystem_iterator_current(
return 0; return 0;
} }
static int is_directory( static int filesystem_iterator_is_dir(
const filesystem_iterator *iter, const filesystem_iterator_entry *entry) bool *is_dir,
const filesystem_iterator *iter,
const filesystem_iterator_entry *entry)
{ {
int isdir = 0;
struct stat st; struct stat st;
git_buf fullpath; git_buf fullpath = GIT_BUF_INIT;
int error = 0;
if (S_ISDIR(entry->st.st_mode)) if (S_ISDIR(entry->st.st_mode)) {
return 1; *is_dir = 1;
if (!iterator__symlinksdir(iter) || !S_ISLNK(entry->st.st_mode)) goto done;
return 0; }
if (!iterator__descend_symlinks(iter) || !S_ISLNK(entry->st.st_mode)) {
*is_dir = 0;
goto done;
}
git_buf_init(&fullpath, 0); if ((error = git_buf_joinpath(&fullpath, iter->root, entry->path)) < 0 ||
git_buf_joinpath(&fullpath, iter->root, entry->path); (error = p_stat(fullpath.ptr, &st)) < 0)
isdir = !p_stat(fullpath.ptr, &st) && S_ISDIR(st.st_mode); goto done;
*is_dir = S_ISDIR(st.st_mode);
done:
git_buf_free(&fullpath); git_buf_free(&fullpath);
return isdir; return error;
} }
static int filesystem_iterator_advance( static int filesystem_iterator_advance(
const git_index_entry **out, git_iterator *i) const git_index_entry **out, git_iterator *i)
{ {
filesystem_iterator *iter = (filesystem_iterator *)i; filesystem_iterator *iter = (filesystem_iterator *)i;
bool is_dir;
int error = 0; int error = 0;
iter->base.flags |= GIT_ITERATOR_FIRST_ACCESS; iter->base.flags |= GIT_ITERATOR_FIRST_ACCESS;
...@@ -1539,7 +1551,10 @@ static int filesystem_iterator_advance( ...@@ -1539,7 +1551,10 @@ static int filesystem_iterator_advance(
entry = frame->entries.contents[frame->next_idx]; entry = frame->entries.contents[frame->next_idx];
frame->next_idx++; frame->next_idx++;
if (is_directory(iter, entry)) { if ((error = filesystem_iterator_is_dir(&is_dir, iter, entry)) < 0)
break;
if (is_dir) {
if (iterator__do_autoexpand(iter)) { if (iterator__do_autoexpand(iter)) {
error = filesystem_iterator_frame_push(iter, entry); error = filesystem_iterator_frame_push(iter, entry);
......
...@@ -39,8 +39,8 @@ typedef enum { ...@@ -39,8 +39,8 @@ typedef enum {
GIT_ITERATOR_DONT_PRECOMPOSE_UNICODE = (1u << 5), GIT_ITERATOR_DONT_PRECOMPOSE_UNICODE = (1u << 5),
/** include conflicts */ /** include conflicts */
GIT_ITERATOR_INCLUDE_CONFLICTS = (1u << 6), GIT_ITERATOR_INCLUDE_CONFLICTS = (1u << 6),
/** descend into symlinked directories when looking for references */ /** descend into symlinked directories */
GIT_ITERATOR_INCLUDE_SYMLINK_REFSDIR = (1u << 7), GIT_ITERATOR_DESCEND_SYMLINKS = (1u << 7),
} git_iterator_flag_t; } git_iterator_flag_t;
typedef enum { typedef enum {
......
...@@ -2035,7 +2035,7 @@ int git_refdb_backend_fs( ...@@ -2035,7 +2035,7 @@ int git_refdb_backend_fs(
if ((!git_repository__cvar(&t, backend->repo, GIT_CVAR_FSYNCOBJECTFILES) && t) || if ((!git_repository__cvar(&t, backend->repo, GIT_CVAR_FSYNCOBJECTFILES) && t) ||
git_repository__fsync_gitdir) git_repository__fsync_gitdir)
backend->fsync = 1; backend->fsync = 1;
backend->iterator_flags |= GIT_ITERATOR_INCLUDE_SYMLINK_REFSDIR; backend->iterator_flags |= GIT_ITERATOR_DESCEND_SYMLINKS;
backend->parent.exists = &refdb_fs_backend__exists; backend->parent.exists = &refdb_fs_backend__exists;
backend->parent.lookup = &refdb_fs_backend__lookup; backend->parent.lookup = &refdb_fs_backend__lookup;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment