]>
Commit | Line | Data |
---|---|---|
53e1b683 | 1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
6414b7c9 | 2 | /*** |
96b2fb93 | 3 | Copyright © 2013 David Strauss |
6414b7c9 DS |
4 | ***/ |
5 | ||
6 | #include <stdio.h> | |
6414b7c9 | 7 | |
6414b7c9 | 8 | #include "macro.h" |
cf0fbc49 | 9 | #include "manager.h" |
d2120590 | 10 | #include "rm-rf.h" |
ec635a2d | 11 | #include "string-util.h" |
6414b7c9 | 12 | #include "test-helper.h" |
d2120590 | 13 | #include "tests.h" |
cf0fbc49 | 14 | #include "unit.h" |
6414b7c9 DS |
15 | |
16 | static int test_cgroup_mask(void) { | |
3e29e810 | 17 | _cleanup_(rm_rf_physical_and_freep) char *runtime_dir = NULL; |
c70cac54 | 18 | _cleanup_(manager_freep) Manager *m = NULL; |
bc432dc7 | 19 | Unit *son, *daughter, *parent, *root, *grandchild, *parent_deep; |
6414b7c9 DS |
20 | int r; |
21 | ||
651d47d1 ZJS |
22 | r = enter_cgroup_subroot(); |
23 | if (r == -ENOMEDIUM) { | |
24 | puts("Skipping test: cgroupfs not available"); | |
25 | return EXIT_TEST_SKIP; | |
26 | } | |
8c759b33 | 27 | |
6414b7c9 | 28 | /* Prepare the manager. */ |
cc100a5a | 29 | assert_se(set_unit_path(get_testdata_dir("")) >= 0); |
3e29e810 | 30 | assert_se(runtime_dir = setup_fake_runtime_dir()); |
e8112e67 | 31 | r = manager_new(UNIT_FILE_USER, MANAGER_TEST_RUN_BASIC, &m); |
4c701096 | 32 | if (IN_SET(r, -EPERM, -EACCES)) { |
6414b7c9 DS |
33 | puts("manager_new: Permission denied. Skipping test."); |
34 | return EXIT_TEST_SKIP; | |
35 | } | |
def8b4c5 | 36 | assert_se(r >= 0); |
9ded9cd1 LP |
37 | |
38 | /* Turn off all kinds of default accouning, so that we can | |
39 | * verify the masks resulting of our configuration and nothing | |
40 | * else. */ | |
41 | m->default_cpu_accounting = | |
42 | m->default_memory_accounting = | |
43 | m->default_blockio_accounting = | |
538b4852 | 44 | m->default_io_accounting = |
9ded9cd1 LP |
45 | m->default_tasks_accounting = false; |
46 | m->default_tasks_max = (uint64_t) -1; | |
47 | ||
bdf7026e | 48 | assert_se(r >= 0); |
76752516 | 49 | assert_se(manager_startup(m, NULL, NULL) >= 0); |
6414b7c9 DS |
50 | |
51 | /* Load units and verify hierarchy. */ | |
ba412430 ZJS |
52 | assert_se(manager_load_startable_unit_or_warn(m, "parent.slice", NULL, &parent) >= 0); |
53 | assert_se(manager_load_startable_unit_or_warn(m, "son.service", NULL, &son) >= 0); | |
54 | assert_se(manager_load_startable_unit_or_warn(m, "daughter.service", NULL, &daughter) >= 0); | |
55 | assert_se(manager_load_startable_unit_or_warn(m, "grandchild.service", NULL, &grandchild) >= 0); | |
56 | assert_se(manager_load_startable_unit_or_warn(m, "parent-deep.slice", NULL, &parent_deep) >= 0); | |
bdf7026e TA |
57 | assert_se(UNIT_DEREF(son->slice) == parent); |
58 | assert_se(UNIT_DEREF(daughter->slice) == parent); | |
59 | assert_se(UNIT_DEREF(parent_deep->slice) == parent); | |
60 | assert_se(UNIT_DEREF(grandchild->slice) == parent_deep); | |
6414b7c9 DS |
61 | root = UNIT_DEREF(parent->slice); |
62 | ||
63 | /* Verify per-unit cgroups settings. */ | |
efdb0237 LP |
64 | assert_se(unit_get_own_mask(son) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT)); |
65 | assert_se(unit_get_own_mask(daughter) == 0); | |
66 | assert_se(unit_get_own_mask(grandchild) == 0); | |
67 | assert_se(unit_get_own_mask(parent_deep) == CGROUP_MASK_MEMORY); | |
538b4852 | 68 | assert_se(unit_get_own_mask(parent) == (CGROUP_MASK_IO | CGROUP_MASK_BLKIO)); |
efdb0237 | 69 | assert_se(unit_get_own_mask(root) == 0); |
bc432dc7 LP |
70 | |
71 | /* Verify aggregation of member masks */ | |
bdf7026e TA |
72 | assert_se(unit_get_members_mask(son) == 0); |
73 | assert_se(unit_get_members_mask(daughter) == 0); | |
74 | assert_se(unit_get_members_mask(grandchild) == 0); | |
75 | assert_se(unit_get_members_mask(parent_deep) == 0); | |
efdb0237 | 76 | assert_se(unit_get_members_mask(parent) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY)); |
538b4852 | 77 | assert_se(unit_get_members_mask(root) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY)); |
bc432dc7 LP |
78 | |
79 | /* Verify aggregation of sibling masks. */ | |
efdb0237 LP |
80 | assert_se(unit_get_siblings_mask(son) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY)); |
81 | assert_se(unit_get_siblings_mask(daughter) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY)); | |
bdf7026e | 82 | assert_se(unit_get_siblings_mask(grandchild) == 0); |
efdb0237 | 83 | assert_se(unit_get_siblings_mask(parent_deep) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY)); |
538b4852 TH |
84 | assert_se(unit_get_siblings_mask(parent) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY)); |
85 | assert_se(unit_get_siblings_mask(root) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY)); | |
bc432dc7 LP |
86 | |
87 | /* Verify aggregation of target masks. */ | |
efdb0237 LP |
88 | assert_se(unit_get_target_mask(son) == ((CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY) & m->cgroup_supported)); |
89 | assert_se(unit_get_target_mask(daughter) == ((CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY) & m->cgroup_supported)); | |
bdf7026e | 90 | assert_se(unit_get_target_mask(grandchild) == 0); |
efdb0237 | 91 | assert_se(unit_get_target_mask(parent_deep) == ((CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY) & m->cgroup_supported)); |
538b4852 TH |
92 | assert_se(unit_get_target_mask(parent) == ((CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY) & m->cgroup_supported)); |
93 | assert_se(unit_get_target_mask(root) == ((CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_IO | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY) & m->cgroup_supported)); | |
6414b7c9 | 94 | |
6414b7c9 DS |
95 | return 0; |
96 | } | |
97 | ||
ec635a2d LP |
98 | static void test_cg_mask_to_string_one(CGroupMask mask, const char *t) { |
99 | _cleanup_free_ char *b = NULL; | |
100 | ||
101 | assert_se(cg_mask_to_string(mask, &b) >= 0); | |
102 | assert_se(streq_ptr(b, t)); | |
103 | } | |
104 | ||
105 | static void test_cg_mask_to_string(void) { | |
106 | test_cg_mask_to_string_one(0, NULL); | |
107 | test_cg_mask_to_string_one(_CGROUP_MASK_ALL, "cpu cpuacct io blkio memory devices pids"); | |
108 | test_cg_mask_to_string_one(CGROUP_MASK_CPU, "cpu"); | |
109 | test_cg_mask_to_string_one(CGROUP_MASK_CPUACCT, "cpuacct"); | |
110 | test_cg_mask_to_string_one(CGROUP_MASK_IO, "io"); | |
111 | test_cg_mask_to_string_one(CGROUP_MASK_BLKIO, "blkio"); | |
112 | test_cg_mask_to_string_one(CGROUP_MASK_MEMORY, "memory"); | |
113 | test_cg_mask_to_string_one(CGROUP_MASK_DEVICES, "devices"); | |
114 | test_cg_mask_to_string_one(CGROUP_MASK_PIDS, "pids"); | |
115 | test_cg_mask_to_string_one(CGROUP_MASK_CPU|CGROUP_MASK_CPUACCT, "cpu cpuacct"); | |
116 | test_cg_mask_to_string_one(CGROUP_MASK_CPU|CGROUP_MASK_PIDS, "cpu pids"); | |
117 | test_cg_mask_to_string_one(CGROUP_MASK_CPUACCT|CGROUP_MASK_PIDS, "cpuacct pids"); | |
118 | test_cg_mask_to_string_one(CGROUP_MASK_DEVICES|CGROUP_MASK_PIDS, "devices pids"); | |
119 | test_cg_mask_to_string_one(CGROUP_MASK_IO|CGROUP_MASK_BLKIO, "io blkio"); | |
120 | } | |
121 | ||
6414b7c9 DS |
122 | int main(int argc, char* argv[]) { |
123 | int rc = 0; | |
d2120590 | 124 | |
ec635a2d LP |
125 | log_parse_environment(); |
126 | log_open(); | |
127 | ||
6414b7c9 | 128 | TEST_REQ_RUNNING_SYSTEMD(rc = test_cgroup_mask()); |
ec635a2d | 129 | test_cg_mask_to_string(); |
d2120590 | 130 | |
6414b7c9 DS |
131 | return rc; |
132 | } |