ryantimwilson / rpms / systemd

Forked from rpms/systemd 2 months ago
Clone
36e8a3
From b53f89d56a5b7528735ddf335f8b47ab3e1a947a Mon Sep 17 00:00:00 2001
36e8a3
From: Lennart Poettering <lennart@poettering.net>
36e8a3
Date: Fri, 19 Oct 2018 11:31:37 +0200
36e8a3
Subject: [PATCH] test: add test case for recursive chown()ing
36e8a3
36e8a3
[msekleta: I removed call to log_test_skipped() and replaced it with older construct log_info() + return EXIT_TEST_SKIP]
36e8a3
36e8a3
(cherry-picked from commit cb9e44db36caefcbb8ee7a12e14217305ed69ff2)
36e8a3
36e8a3
Related: #1643368
36e8a3
---
36e8a3
 src/test/meson.build      |   5 ++
36e8a3
 src/test/test-chown-rec.c | 162 ++++++++++++++++++++++++++++++++++++++
36e8a3
 2 files changed, 167 insertions(+)
36e8a3
 create mode 100644 src/test/test-chown-rec.c
36e8a3
36e8a3
diff --git a/src/test/meson.build b/src/test/meson.build
4bff0a
index 7da7e3a22c..b982251b1f 100644
36e8a3
--- a/src/test/meson.build
36e8a3
+++ b/src/test/meson.build
36e8a3
@@ -60,6 +60,11 @@ tests += [
36e8a3
           libmount,
36e8a3
           libblkid]],
36e8a3
 
36e8a3
+        [['src/test/test-chown-rec.c'],
36e8a3
+         [libcore,
36e8a3
+          libshared],
36e8a3
+         []],
36e8a3
+
36e8a3
         [['src/test/test-job-type.c'],
36e8a3
          [libcore,
36e8a3
           libshared],
36e8a3
diff --git a/src/test/test-chown-rec.c b/src/test/test-chown-rec.c
36e8a3
new file mode 100644
4bff0a
index 0000000000..f16d4d4ba2
36e8a3
--- /dev/null
36e8a3
+++ b/src/test/test-chown-rec.c
36e8a3
@@ -0,0 +1,162 @@
36e8a3
+/* SPDX-License-Identifier: LGPL-2.1+ */
36e8a3
+
36e8a3
+#include <sys/xattr.h>
36e8a3
+
36e8a3
+#include "alloc-util.h"
36e8a3
+#include "chown-recursive.h"
36e8a3
+#include "fileio.h"
36e8a3
+#include "log.h"
36e8a3
+#include "rm-rf.h"
36e8a3
+#include "string-util.h"
36e8a3
+#include "tests.h"
36e8a3
+
36e8a3
+static const uint8_t acl[] = {
36e8a3
+        0x02, 0x00, 0x00, 0x00, 0x01, 0x00, 0x07, 0x00,
36e8a3
+        0xff, 0xff, 0xff, 0xff, 0x02, 0x00, 0x07, 0x00,
36e8a3
+        0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x07, 0x00,
36e8a3
+        0xff, 0xff, 0xff, 0xff, 0x10, 0x00, 0x07, 0x00,
36e8a3
+        0xff, 0xff, 0xff, 0xff, 0x20, 0x00, 0x05, 0x00,
36e8a3
+        0xff, 0xff, 0xff, 0xff,
36e8a3
+};
36e8a3
+
36e8a3
+static const uint8_t default_acl[] = {
36e8a3
+        0x02, 0x00, 0x00, 0x00, 0x01, 0x00, 0x07, 0x00,
36e8a3
+        0xff, 0xff, 0xff, 0xff, 0x04, 0x00, 0x07, 0x00,
36e8a3
+        0xff, 0xff, 0xff, 0xff, 0x08, 0x00, 0x07, 0x00,
36e8a3
+        0x04, 0x00, 0x00, 0x00, 0x10, 0x00, 0x07, 0x00,
36e8a3
+        0xff, 0xff, 0xff, 0xff, 0x20, 0x00, 0x05, 0x00,
36e8a3
+        0xff, 0xff, 0xff, 0xff,
36e8a3
+};
36e8a3
+
36e8a3
+static bool has_xattr(const char *p) {
36e8a3
+        char buffer[sizeof(acl) * 4];
36e8a3
+
36e8a3
+        if (lgetxattr(p, "system.posix_acl_access", buffer, sizeof(buffer)) < 0) {
36e8a3
+                if (IN_SET(errno, EOPNOTSUPP, ENOTTY, ENODATA, ENOSYS))
36e8a3
+                        return false;
36e8a3
+        }
36e8a3
+
36e8a3
+        return true;
36e8a3
+}
36e8a3
+
36e8a3
+static void test_chown_recursive(void) {
36e8a3
+        _cleanup_(rm_rf_physical_and_freep) char *t = NULL;
36e8a3
+        struct stat st;
36e8a3
+        const char *p;
36e8a3
+
36e8a3
+        umask(022);
36e8a3
+        assert_se(mkdtemp_malloc(NULL, &t) >= 0);
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir");
36e8a3
+        assert_se(mkdir(p, 0777) >= 0);
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISDIR(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0755);
36e8a3
+        assert_se(st.st_uid == 0);
36e8a3
+        assert_se(st.st_gid == 0);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir/symlink");
36e8a3
+        assert_se(symlink("../../", p) >= 0);
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISLNK(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0777);
36e8a3
+        assert_se(st.st_uid == 0);
36e8a3
+        assert_se(st.st_gid == 0);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir/reg");
36e8a3
+        assert_se(mknod(p, S_IFREG|0777, 0) >= 0);
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISREG(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0755);
36e8a3
+        assert_se(st.st_uid == 0);
36e8a3
+        assert_se(st.st_gid == 0);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir/sock");
36e8a3
+        assert_se(mknod(p, S_IFSOCK|0777, 0) >= 0);
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISSOCK(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0755);
36e8a3
+        assert_se(st.st_uid == 0);
36e8a3
+        assert_se(st.st_gid == 0);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir/fifo");
36e8a3
+        assert_se(mknod(p, S_IFIFO|0777, 0) >= 0);
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISFIFO(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0755);
36e8a3
+        assert_se(st.st_uid == 0);
36e8a3
+        assert_se(st.st_gid == 0);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+
36e8a3
+        /* We now apply an xattr to the dir, and check it again */
36e8a3
+        p = strjoina(t, "/dir");
36e8a3
+        assert_se(setxattr(p, "system.posix_acl_access", acl, sizeof(acl), 0) >= 0);
36e8a3
+        assert_se(setxattr(p, "system.posix_acl_default", default_acl, sizeof(default_acl), 0) >= 0);
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISDIR(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0775); /* acl change changed the mode too */
36e8a3
+        assert_se(st.st_uid == 0);
36e8a3
+        assert_se(st.st_gid == 0);
36e8a3
+        assert_se(has_xattr(p));
36e8a3
+
36e8a3
+        assert_se(path_chown_recursive(t, 1, 2) >= 0);
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir");
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISDIR(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0775);
36e8a3
+        assert_se(st.st_uid == 1);
36e8a3
+        assert_se(st.st_gid == 2);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir/symlink");
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISLNK(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0777);
36e8a3
+        assert_se(st.st_uid == 1);
36e8a3
+        assert_se(st.st_gid == 2);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir/reg");
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISREG(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0755);
36e8a3
+        assert_se(st.st_uid == 1);
36e8a3
+        assert_se(st.st_gid == 2);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir/sock");
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISSOCK(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0755);
36e8a3
+        assert_se(st.st_uid == 1);
36e8a3
+        assert_se(st.st_gid == 2);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+
36e8a3
+        p = strjoina(t, "/dir/fifo");
36e8a3
+        assert_se(lstat(p, &st) >= 0);
36e8a3
+        assert_se(S_ISFIFO(st.st_mode));
36e8a3
+        assert_se((st.st_mode & 07777) == 0755);
36e8a3
+        assert_se(st.st_uid == 1);
36e8a3
+        assert_se(st.st_gid == 2);
36e8a3
+        assert_se(!has_xattr(p));
36e8a3
+}
36e8a3
+
36e8a3
+int main(int argc, char *argv[]) {
36e8a3
+        log_set_max_level(LOG_DEBUG);
36e8a3
+        log_parse_environment();
36e8a3
+        log_open();
36e8a3
+
36e8a3
+        if (geteuid() != 0) {
36e8a3
+                log_info("not running as root");
36e8a3
+                return EXIT_TEST_SKIP;
36e8a3
+        }
36e8a3
+
36e8a3
+        test_chown_recursive();
36e8a3
+
36e8a3
+        return EXIT_SUCCESS;
36e8a3
+}