]>
Commit | Line | Data |
---|---|---|
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 | ||
20 | static 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 | 29 | static 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 |
124 | static 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 | ||
131 | static 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 | 149 | int 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 | } |