]>
Commit | Line | Data |
---|---|---|
6414b7c9 DS |
1 | /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ |
2 | ||
3 | /*** | |
4 | This file is part of systemd. | |
5 | ||
6 | Copyright 2013 David Strauss | |
7 | ||
8 | systemd is free software; you can redistribute it and/or modify it | |
9 | under the terms of the GNU Lesser General Public License as published by | |
10 | the Free Software Foundation; either version 2.1 of the License, or | |
11 | (at your option) any later version. | |
12 | ||
13 | systemd is distributed in the hope that it will be useful, but | |
14 | WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
16 | Lesser General Public License for more details. | |
17 | ||
18 | You should have received a copy of the GNU Lesser General Public License | |
19 | along with systemd; If not, see <http://www.gnu.org/licenses/>. | |
20 | ***/ | |
21 | ||
22 | #include <stdio.h> | |
6414b7c9 DS |
23 | |
24 | #include "manager.h" | |
25 | #include "unit.h" | |
6414b7c9 DS |
26 | #include "macro.h" |
27 | #include "test-helper.h" | |
28 | ||
29 | static int test_cgroup_mask(void) { | |
39883f62 | 30 | Manager *m = NULL; |
bc432dc7 | 31 | Unit *son, *daughter, *parent, *root, *grandchild, *parent_deep; |
6414b7c9 DS |
32 | FILE *serial = NULL; |
33 | FDSet *fdset = NULL; | |
34 | int r; | |
35 | ||
36 | /* Prepare the manager. */ | |
0d8c31ff | 37 | assert_se(set_unit_path(TEST_DIR) >= 0); |
b2c23da8 | 38 | r = manager_new(MANAGER_USER, true, &m); |
6414b7c9 DS |
39 | if (r == -EPERM || r == -EACCES) { |
40 | puts("manager_new: Permission denied. Skipping test."); | |
41 | return EXIT_TEST_SKIP; | |
42 | } | |
bdf7026e | 43 | assert_se(r >= 0); |
6414b7c9 DS |
44 | assert_se(manager_startup(m, serial, fdset) >= 0); |
45 | ||
46 | /* Load units and verify hierarchy. */ | |
47 | assert_se(manager_load_unit(m, "parent.slice", NULL, NULL, &parent) >= 0); | |
48 | assert_se(manager_load_unit(m, "son.service", NULL, NULL, &son) >= 0); | |
49 | assert_se(manager_load_unit(m, "daughter.service", NULL, NULL, &daughter) >= 0); | |
bc432dc7 LP |
50 | assert_se(manager_load_unit(m, "grandchild.service", NULL, NULL, &grandchild) >= 0); |
51 | assert_se(manager_load_unit(m, "parent-deep.slice", NULL, NULL, &parent_deep) >= 0); | |
bdf7026e TA |
52 | assert_se(parent->load_state == UNIT_LOADED); |
53 | assert_se(son->load_state == UNIT_LOADED); | |
54 | assert_se(daughter->load_state == UNIT_LOADED); | |
55 | assert_se(grandchild->load_state == UNIT_LOADED); | |
56 | assert_se(parent_deep->load_state == UNIT_LOADED); | |
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); | |
68 | assert_se(unit_get_own_mask(parent) == CGROUP_MASK_BLKIO); | |
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 LP |
76 | assert_se(unit_get_members_mask(parent) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY)); |
77 | assert_se(unit_get_members_mask(root) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | 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 LP |
83 | assert_se(unit_get_siblings_mask(parent_deep) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY)); |
84 | assert_se(unit_get_siblings_mask(parent) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_BLKIO | CGROUP_MASK_MEMORY)); | |
85 | assert_se(unit_get_siblings_mask(root) == (CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | 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 LP |
91 | assert_se(unit_get_target_mask(parent_deep) == ((CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | CGROUP_MASK_MEMORY) & m->cgroup_supported)); |
92 | assert_se(unit_get_target_mask(parent) == ((CGROUP_MASK_CPU | CGROUP_MASK_CPUACCT | 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_BLKIO | CGROUP_MASK_MEMORY) & m->cgroup_supported)); | |
6414b7c9 DS |
94 | |
95 | manager_free(m); | |
96 | ||
97 | return 0; | |
98 | } | |
99 | ||
100 | int main(int argc, char* argv[]) { | |
101 | int rc = 0; | |
102 | TEST_REQ_RUNNING_SYSTEMD(rc = test_cgroup_mask()); | |
103 | return rc; | |
104 | } |