]>
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" |
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 | ||
21 | static 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 | 30 | static 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 |
123 | static 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 | ||
130 | static 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 | 147 | int 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 | } |