From: Yu Watanabe <watanabe.yu+github@gmail.com>
Date: Mon, 10 Apr 2023 19:09:55 +0000 (+0900)
Subject: path-util: introduce path_prefix_root_cwd()
X-Git-Tag: v254-rc1~769^2~1
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=75cc8d7c42ad2ad9b793bbdd96302a04e5d40a65;p=thirdparty%2Fsystemd.git

path-util: introduce path_prefix_root_cwd()
---

diff --git a/src/basic/path-util.c b/src/basic/path-util.c
index 0b0f0da7602..fa2e26789f3 100644
--- a/src/basic/path-util.c
+++ b/src/basic/path-util.c
@@ -100,6 +100,34 @@ int path_make_absolute_cwd(const char *p, char **ret) {
         return 0;
 }
 
+int path_prefix_root_cwd(const char *p, const char *root, char **ret) {
+        _cleanup_free_ char *root_abs = NULL;
+        char *c;
+        int r;
+
+        assert(p);
+        assert(ret);
+
+        /* Unlike path_make_absolute(), this always prefixes root path if specified.
+         * The root path is always simplified, but the provided path will not.
+         * This is useful for prefixing the result of chaseat(). */
+
+        if (empty_or_root(root))
+                return path_make_absolute_cwd(p, ret);
+
+        r = path_make_absolute_cwd(root, &root_abs);
+        if (r < 0)
+                return r;
+
+        path_simplify(root_abs);
+        c = path_join(root_abs, p);
+        if (!c)
+                return -ENOMEM;
+
+        *ret = c;
+        return 0;
+}
+
 int path_make_relative(const char *from, const char *to, char **ret) {
         _cleanup_free_ char *result = NULL;
         unsigned n_parents;
diff --git a/src/basic/path-util.h b/src/basic/path-util.h
index 7843599816e..a0af9de6742 100644
--- a/src/basic/path-util.h
+++ b/src/basic/path-util.h
@@ -60,6 +60,7 @@ int path_split_and_make_absolute(const char *p, char ***ret);
 char* path_make_absolute(const char *p, const char *prefix);
 int safe_getcwd(char **ret);
 int path_make_absolute_cwd(const char *p, char **ret);
+int path_prefix_root_cwd(const char *p, const char *root, char **ret);
 int path_make_relative(const char *from, const char *to, char **ret);
 int path_make_relative_parent(const char *from_child, const char *to, char **ret);
 char *path_startswith_full(const char *path, const char *prefix, bool accept_dot_dot) _pure_;
diff --git a/src/test/test-path-util.c b/src/test/test-path-util.c
index e40ffea4d53..22e8f3481af 100644
--- a/src/test/test-path-util.c
+++ b/src/test/test-path-util.c
@@ -494,6 +494,45 @@ TEST(fsck_exists) {
         assert_se(fsck_exists_for_fstype("/../bin/") == 0);
 }
 
+TEST(path_prefix_root_cwd) {
+        _cleanup_free_ char *cwd = NULL, *ret = NULL, *expected = NULL;
+
+        assert_se(safe_getcwd(&cwd) >= 0);
+
+        assert_se(path_prefix_root_cwd("hoge", NULL, &ret) >= 0);
+        assert_se(expected = path_join(cwd, "hoge"));
+        assert_se(streq(ret, expected));
+
+        ret = mfree(ret);
+        expected = mfree(expected);
+
+        assert_se(path_prefix_root_cwd("/hoge", NULL, &ret) >= 0);
+        assert_se(streq(ret, "/hoge"));
+
+        ret = mfree(ret);
+
+        assert_se(path_prefix_root_cwd("hoge", "/a/b//./c///", &ret) >= 0);
+        assert_se(streq(ret, "/a/b/c/hoge"));
+
+        ret = mfree(ret);
+
+        assert_se(path_prefix_root_cwd("hoge", "a/b//./c///", &ret) >= 0);
+        assert_se(expected = path_join(cwd, "a/b/c/hoge"));
+        assert_se(streq(ret, expected));
+
+        ret = mfree(ret);
+        expected = mfree(expected);
+
+        assert_se(path_prefix_root_cwd("/../hoge/aaa/../././b", "/a/b//./c///", &ret) >= 0);
+        assert_se(streq(ret, "/a/b/c/../hoge/aaa/../././b"));
+
+        ret = mfree(ret);
+
+        assert_se(path_prefix_root_cwd("/../hoge/aaa/../././b", "a/b//./c///", &ret) >= 0);
+        assert_se(expected = path_join(cwd, "a/b/c/../hoge/aaa/../././b"));
+        assert_se(streq(ret, expected));
+}
+
 static void test_path_make_relative_one(const char *from, const char *to, const char *expected) {
         _cleanup_free_ char *z = NULL;
         int r;