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