]> git.ipfire.org Git - thirdparty/systemd.git/blame - src/test/test-cgroup-mask.c
shared/install: do not require /dev/null to be present in chroots
[thirdparty/systemd.git] / src / test / test-cgroup-mask.c
CommitLineData
53e1b683 1/* SPDX-License-Identifier: LGPL-2.1+ */
6414b7c9
DS
2
3#include <stdio.h>
6414b7c9 4
f98c2585
CD
5#include "cgroup.h"
6#include "cgroup-util.h"
6414b7c9 7#include "macro.h"
cf0fbc49 8#include "manager.h"
d2120590 9#include "rm-rf.h"
ec635a2d 10#include "string-util.h"
d2120590 11#include "tests.h"
cf0fbc49 12#include "unit.h"
6414b7c9 13
f98c2585
CD
14#define ASSERT_CGROUP_MASK(got, expected) \
15 log_cgroup_mask(got, expected); \
16 assert_se(got == expected)
17
18#define ASSERT_CGROUP_MASK_JOINED(got, expected) ASSERT_CGROUP_MASK(got, CGROUP_MASK_EXTEND_JOINED(expected))
19
20static void log_cgroup_mask(CGroupMask got, CGroupMask expected) {
21 _cleanup_free_ char *e_store = NULL, *g_store = NULL;
22
23 assert_se(cg_mask_to_string(expected, &e_store) >= 0);
24 log_info("Expected mask: %s\n", e_store);
25 assert_se(cg_mask_to_string(got, &g_store) >= 0);
26 log_info("Got mask: %s\n", g_store);
27}
28
6414b7c9 29static int test_cgroup_mask(void) {
3e29e810 30 _cleanup_(rm_rf_physical_and_freep) char *runtime_dir = NULL;
c70cac54 31 _cleanup_(manager_freep) Manager *m = NULL;
c72703e2 32 Unit *son, *daughter, *parent, *root, *grandchild, *parent_deep, *nomem_parent, *nomem_leaf;
6414b7c9 33 int r;
f98c2585 34 CGroupMask cpu_accounting_mask = get_cpu_accounting_mask();
6414b7c9 35
64ad9e08 36 r = enter_cgroup_subroot(NULL);
317bb217
ZJS
37 if (r == -ENOMEDIUM)
38 return log_tests_skipped("cgroupfs not available");
8c759b33 39
6414b7c9 40 /* Prepare the manager. */
7b432953
ZJS
41 _cleanup_free_ char *unit_dir = NULL;
42 assert_se(get_testdata_dir("units", &unit_dir) >= 0);
43 assert_se(set_unit_path(unit_dir) >= 0);
3e29e810 44 assert_se(runtime_dir = setup_fake_runtime_dir());
e8112e67 45 r = manager_new(UNIT_FILE_USER, MANAGER_TEST_RUN_BASIC, &m);
4c701096 46 if (IN_SET(r, -EPERM, -EACCES)) {
317bb217
ZJS
47 log_error_errno(r, "manager_new: %m");
48 return log_tests_skipped("cannot create manager");
6414b7c9 49 }
317bb217 50
def8b4c5 51 assert_se(r >= 0);
9ded9cd1
LP
52
53 /* Turn off all kinds of default accouning, so that we can
54 * verify the masks resulting of our configuration and nothing
55 * else. */
56 m->default_cpu_accounting =
57 m->default_memory_accounting =
58 m->default_blockio_accounting =
538b4852 59 m->default_io_accounting =
9ded9cd1 60 m->default_tasks_accounting = false;
3a0f06c4 61 m->default_tasks_max = TASKS_MAX_UNSET;
9ded9cd1 62
76752516 63 assert_se(manager_startup(m, NULL, NULL) >= 0);
6414b7c9
DS
64
65 /* Load units and verify hierarchy. */
ba412430
ZJS
66 assert_se(manager_load_startable_unit_or_warn(m, "parent.slice", NULL, &parent) >= 0);
67 assert_se(manager_load_startable_unit_or_warn(m, "son.service", NULL, &son) >= 0);
68 assert_se(manager_load_startable_unit_or_warn(m, "daughter.service", NULL, &daughter) >= 0);
69 assert_se(manager_load_startable_unit_or_warn(m, "grandchild.service", NULL, &grandchild) >= 0);
70 assert_se(manager_load_startable_unit_or_warn(m, "parent-deep.slice", NULL, &parent_deep) >= 0);
c72703e2
CD
71 assert_se(manager_load_startable_unit_or_warn(m, "nomem.slice", NULL, &nomem_parent) >= 0);
72 assert_se(manager_load_startable_unit_or_warn(m, "nomemleaf.service", NULL, &nomem_leaf) >= 0);
bdf7026e
TA
73 assert_se(UNIT_DEREF(son->slice) == parent);
74 assert_se(UNIT_DEREF(daughter->slice) == parent);
75 assert_se(UNIT_DEREF(parent_deep->slice) == parent);
76 assert_se(UNIT_DEREF(grandchild->slice) == parent_deep);
c72703e2 77 assert_se(UNIT_DEREF(nomem_leaf->slice) == nomem_parent);
6414b7c9 78 root = UNIT_DEREF(parent->slice);
c72703e2 79 assert_se(UNIT_DEREF(nomem_parent->slice) == root);
6414b7c9
DS
80
81 /* Verify per-unit cgroups settings. */
f98c2585
CD
82 ASSERT_CGROUP_MASK_JOINED(unit_get_own_mask(son), CGROUP_MASK_CPU);
83 ASSERT_CGROUP_MASK_JOINED(unit_get_own_mask(daughter), cpu_accounting_mask);
84 ASSERT_CGROUP_MASK_JOINED(unit_get_own_mask(grandchild), 0);
85 ASSERT_CGROUP_MASK_JOINED(unit_get_own_mask(parent_deep), CGROUP_MASK_MEMORY);
86 ASSERT_CGROUP_MASK_JOINED(unit_get_own_mask(parent), (CGROUP_MASK_IO | CGROUP_MASK_BLKIO));
c72703e2
CD
87 ASSERT_CGROUP_MASK_JOINED(unit_get_own_mask(nomem_parent), 0);
88 ASSERT_CGROUP_MASK_JOINED(unit_get_own_mask(nomem_leaf), (CGROUP_MASK_IO | CGROUP_MASK_BLKIO));
f98c2585 89 ASSERT_CGROUP_MASK_JOINED(unit_get_own_mask(root), 0);
bc432dc7
LP
90
91 /* Verify aggregation of member masks */
f98c2585
CD
92 ASSERT_CGROUP_MASK_JOINED(unit_get_members_mask(son), 0);
93 ASSERT_CGROUP_MASK_JOINED(unit_get_members_mask(daughter), 0);
94 ASSERT_CGROUP_MASK_JOINED(unit_get_members_mask(grandchild), 0);
95 ASSERT_CGROUP_MASK_JOINED(unit_get_members_mask(parent_deep), 0);
96 ASSERT_CGROUP_MASK_JOINED(unit_get_members_mask(parent), (CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_MEMORY));
c72703e2
CD
97 ASSERT_CGROUP_MASK_JOINED(unit_get_members_mask(nomem_parent), (CGROUP_MASK_IO | CGROUP_MASK_BLKIO));
98 ASSERT_CGROUP_MASK_JOINED(unit_get_members_mask(nomem_leaf), 0);
f98c2585 99 ASSERT_CGROUP_MASK_JOINED(unit_get_members_mask(root), (CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY));
bc432dc7
LP
100
101 /* Verify aggregation of sibling masks. */
f98c2585
CD
102 ASSERT_CGROUP_MASK_JOINED(unit_get_siblings_mask(son), (CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_MEMORY));
103 ASSERT_CGROUP_MASK_JOINED(unit_get_siblings_mask(daughter), (CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_MEMORY));
104 ASSERT_CGROUP_MASK_JOINED(unit_get_siblings_mask(grandchild), 0);
105 ASSERT_CGROUP_MASK_JOINED(unit_get_siblings_mask(parent_deep), (CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_MEMORY));
106 ASSERT_CGROUP_MASK_JOINED(unit_get_siblings_mask(parent), (CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY));
c72703e2
CD
107 ASSERT_CGROUP_MASK_JOINED(unit_get_siblings_mask(nomem_parent), (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY));
108 ASSERT_CGROUP_MASK_JOINED(unit_get_siblings_mask(nomem_leaf), (CGROUP_MASK_IO | CGROUP_MASK_BLKIO));
f98c2585 109 ASSERT_CGROUP_MASK_JOINED(unit_get_siblings_mask(root), (CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY));
bc432dc7
LP
110
111 /* Verify aggregation of target masks. */
f98c2585
CD
112 ASSERT_CGROUP_MASK(unit_get_target_mask(son), (CGROUP_MASK_EXTEND_JOINED(CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_MEMORY) & m->cgroup_supported));
113 ASSERT_CGROUP_MASK(unit_get_target_mask(daughter), (CGROUP_MASK_EXTEND_JOINED(CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_MEMORY) & m->cgroup_supported));
114 ASSERT_CGROUP_MASK(unit_get_target_mask(grandchild), 0);
115 ASSERT_CGROUP_MASK(unit_get_target_mask(parent_deep), (CGROUP_MASK_EXTEND_JOINED(CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_MEMORY) & m->cgroup_supported));
116 ASSERT_CGROUP_MASK(unit_get_target_mask(parent), (CGROUP_MASK_EXTEND_JOINED(CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY) & m->cgroup_supported));
c72703e2
CD
117 ASSERT_CGROUP_MASK(unit_get_target_mask(nomem_parent), (CGROUP_MASK_EXTEND_JOINED(CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_IO | CGROUP_MASK_BLKIO) & m->cgroup_supported));
118 ASSERT_CGROUP_MASK(unit_get_target_mask(nomem_leaf), (CGROUP_MASK_EXTEND_JOINED(CGROUP_MASK_IO | CGROUP_MASK_BLKIO) & m->cgroup_supported));
f98c2585 119 ASSERT_CGROUP_MASK(unit_get_target_mask(root), (CGROUP_MASK_EXTEND_JOINED(CGROUP_MASK_CPU | cpu_accounting_mask | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY) & m->cgroup_supported));
6414b7c9 120
6414b7c9
DS
121 return 0;
122}
123
ec635a2d
LP
124static void test_cg_mask_to_string_one(CGroupMask mask, const char *t) {
125 _cleanup_free_ char *b = NULL;
126
127 assert_se(cg_mask_to_string(mask, &b) >= 0);
128 assert_se(streq_ptr(b, t));
129}
130
131static void test_cg_mask_to_string(void) {
132 test_cg_mask_to_string_one(0, NULL);
047f5d63 133 test_cg_mask_to_string_one(_CGROUP_MASK_ALL, "cpu cpuacct cpuset io blkio memory devices pids bpf-firewall bpf-devices");
ec635a2d
LP
134 test_cg_mask_to_string_one(CGROUP_MASK_CPU, "cpu");
135 test_cg_mask_to_string_one(CGROUP_MASK_CPUACCT, "cpuacct");
047f5d63 136 test_cg_mask_to_string_one(CGROUP_MASK_CPUSET, "cpuset");
ec635a2d
LP
137 test_cg_mask_to_string_one(CGROUP_MASK_IO, "io");
138 test_cg_mask_to_string_one(CGROUP_MASK_BLKIO, "blkio");
139 test_cg_mask_to_string_one(CGROUP_MASK_MEMORY, "memory");
140 test_cg_mask_to_string_one(CGROUP_MASK_DEVICES, "devices");
141 test_cg_mask_to_string_one(CGROUP_MASK_PIDS, "pids");
142 test_cg_mask_to_string_one(CGROUP_MASK_CPU|CGROUP_MASK_CPUACCT, "cpu cpuacct");
143 test_cg_mask_to_string_one(CGROUP_MASK_CPU|CGROUP_MASK_PIDS, "cpu pids");
144 test_cg_mask_to_string_one(CGROUP_MASK_CPUACCT|CGROUP_MASK_PIDS, "cpuacct pids");
145 test_cg_mask_to_string_one(CGROUP_MASK_DEVICES|CGROUP_MASK_PIDS, "devices pids");
146 test_cg_mask_to_string_one(CGROUP_MASK_IO|CGROUP_MASK_BLKIO, "io blkio");
147}
148
6414b7c9 149int main(int argc, char* argv[]) {
317bb217 150 int rc = EXIT_SUCCESS;
d2120590 151
6d7c4033 152 test_setup_logging(LOG_DEBUG);
ec635a2d 153
ec635a2d 154 test_cg_mask_to_string();
317bb217 155 TEST_REQ_RUNNING_SYSTEMD(rc = test_cgroup_mask());
d2120590 156
6414b7c9
DS
157 return rc;
158}