fs_path: mock ownership checks

Provide a mock for file ownership for testability.

2 files changed, 56 insertions, 0 deletions

diff --git a/src/util/fs_path.c b/src/util/fs_path.c
index d8ef7ea74..9bd773d27 100644
--- a/src/util/fs_path.c
+++ b/src/util/fs_path.c
@@ -1785,6 +1785,13 @@ done:
 	return supported;
 }
 
+static git_fs_path__mock_owner_t mock_owner = GIT_FS_PATH_MOCK_OWNER_NONE;
+
+void git_fs_path__set_owner(git_fs_path__mock_owner_t owner)
+{
+	mock_owner = owner;
+}
+
 #ifdef GIT_WIN32
 static PSID *sid_dup(PSID sid)
 {
@@ -1877,6 +1884,11 @@ int git_fs_path_owner_is_current_user(bool *out, const char *path)
 	PSID owner_sid = NULL, user_sid = NULL;
 	int error = -1;
 
+	if (mock_owner) {
+		*out = (mock_owner == GIT_FS_PATH_MOCK_OWNER_CURRENT_USER);
+		return 0;
+	}
+
 	if ((error = file_owner_sid(&owner_sid, path)) < 0 ||
 	    (error = current_user_sid(&user_sid)) < 0)
 		goto done;
@@ -1894,6 +1906,11 @@ int git_fs_path_owner_is_system(bool *out, const char *path)
 {
 	PSID owner_sid;
 
+	if (mock_owner) {
+		*out = (mock_owner == GIT_FS_PATH_MOCK_OWNER_SYSTEM);
+		return 0;
+	}
+
 	if (file_owner_sid(&owner_sid, path) < 0)
 		return -1;
 
@@ -1909,6 +1926,12 @@ int git_fs_path_owner_is_system_or_current_user(bool *out, const char *path)
 	PSID owner_sid = NULL, user_sid = NULL;
 	int error = -1;
 
+	if (mock_owner) {
+		*out = (mock_owner == GIT_FS_PATH_MOCK_OWNER_SYSTEM ||
+		        mock_owner == GIT_FS_PATH_MOCK_OWNER_CURRENT_USER);
+		return 0;
+	}
+
 	if (file_owner_sid(&owner_sid, path) < 0)
 		goto done;
 
@@ -1961,18 +1984,37 @@ static int fs_path_owner_is(bool *out, const char *path, uid_t *uids, size_t uid
 int git_fs_path_owner_is_current_user(bool *out, const char *path)
 {
 	uid_t userid = geteuid();
+
+	if (mock_owner) {
+		*out = (mock_owner == GIT_FS_PATH_MOCK_OWNER_CURRENT_USER);
+		return 0;
+	}
+
 	return fs_path_owner_is(out, path, &userid, 1);
 }
 
 int git_fs_path_owner_is_system(bool *out, const char *path)
 {
 	uid_t userid = 0;
+
+	if (mock_owner) {
+		*out = (mock_owner == GIT_FS_PATH_MOCK_OWNER_SYSTEM);
+		return 0;
+	}
+
 	return fs_path_owner_is(out, path, &userid, 1);
 }
 
 int git_fs_path_owner_is_system_or_current_user(bool *out, const char *path)
 {
 	uid_t userids[2] = { geteuid(), 0 };
+
+	if (mock_owner) {
+		*out = (mock_owner == GIT_FS_PATH_MOCK_OWNER_SYSTEM ||
+		        mock_owner == GIT_FS_PATH_MOCK_OWNER_CURRENT_USER);
+		return 0;
+	}
+
 	return fs_path_owner_is(out, path, userids, 2);
 }
 
diff --git a/src/util/fs_path.h b/src/util/fs_path.h
index 0913cb591..cf4664178 100644
--- a/src/util/fs_path.h
+++ b/src/util/fs_path.h
@@ -731,6 +731,20 @@ int git_fs_path_normalize_slashes(git_str *out, const char *path);
 
 bool git_fs_path_supports_symlinks(const char *dir);
 
+typedef enum {
+	GIT_FS_PATH_MOCK_OWNER_NONE = 0, /* do filesystem lookups as normal */
+	GIT_FS_PATH_MOCK_OWNER_SYSTEM = 1,
+	GIT_FS_PATH_MOCK_OWNER_CURRENT_USER = 2,
+	GIT_FS_PATH_MOCK_OWNER_OTHER = 3
+} git_fs_path__mock_owner_t;
+
+/**
+ * Sets the mock ownership for files; subsequent calls to
+ * `git_fs_path_owner_is_*` functions will return this data until cleared
+ * with `GIT_FS_PATH_MOCK_OWNER_NONE`.
+ */
+void git_fs_path__set_owner(git_fs_path__mock_owner_t owner);
+
 /**
  * Verify that the file in question is owned by an administrator or system
  * account.