]>
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> | |
23 | #include <stdlib.h> | |
24 | #include <string.h> | |
25 | #include <sys/types.h> | |
26 | #include <pwd.h> | |
27 | ||
28 | #include "manager.h" | |
29 | #include "unit.h" | |
30 | #include "util.h" | |
31 | #include "macro.h" | |
32 | #include "test-helper.h" | |
33 | ||
34 | static int test_cgroup_mask(void) { | |
39883f62 | 35 | Manager *m = NULL; |
bc432dc7 | 36 | Unit *son, *daughter, *parent, *root, *grandchild, *parent_deep; |
6414b7c9 DS |
37 | FILE *serial = NULL; |
38 | FDSet *fdset = NULL; | |
39 | int r; | |
40 | ||
41 | /* Prepare the manager. */ | |
0d8c31ff ZJS |
42 | assert_se(set_unit_path(TEST_DIR) >= 0); |
43 | r = manager_new(SYSTEMD_USER, true, &m); | |
6414b7c9 DS |
44 | if (r == -EPERM || r == -EACCES) { |
45 | puts("manager_new: Permission denied. Skipping test."); | |
46 | return EXIT_TEST_SKIP; | |
47 | } | |
bdf7026e | 48 | assert_se(r >= 0); |
6414b7c9 DS |
49 | assert_se(manager_startup(m, serial, fdset) >= 0); |
50 | ||
51 | /* Load units and verify hierarchy. */ | |
52 | assert_se(manager_load_unit(m, "parent.slice", NULL, NULL, &parent) >= 0); | |
53 | assert_se(manager_load_unit(m, "son.service", NULL, NULL, &son) >= 0); | |
54 | assert_se(manager_load_unit(m, "daughter.service", NULL, NULL, &daughter) >= 0); | |
bc432dc7 LP |
55 | assert_se(manager_load_unit(m, "grandchild.service", NULL, NULL, &grandchild) >= 0); |
56 | assert_se(manager_load_unit(m, "parent-deep.slice", NULL, NULL, &parent_deep) >= 0); | |
bdf7026e TA |
57 | assert_se(parent->load_state == UNIT_LOADED); |
58 | assert_se(son->load_state == UNIT_LOADED); | |
59 | assert_se(daughter->load_state == UNIT_LOADED); | |
60 | assert_se(grandchild->load_state == UNIT_LOADED); | |
61 | assert_se(parent_deep->load_state == UNIT_LOADED); | |
62 | assert_se(UNIT_DEREF(son->slice) == parent); | |
63 | assert_se(UNIT_DEREF(daughter->slice) == parent); | |
64 | assert_se(UNIT_DEREF(parent_deep->slice) == parent); | |
65 | assert_se(UNIT_DEREF(grandchild->slice) == parent_deep); | |
6414b7c9 DS |
66 | root = UNIT_DEREF(parent->slice); |
67 | ||
68 | /* Verify per-unit cgroups settings. */ | |
bdf7026e TA |
69 | assert_se(unit_get_cgroup_mask(son) == (CGROUP_CPU | CGROUP_CPUACCT)); |
70 | assert_se(unit_get_cgroup_mask(daughter) == 0); | |
71 | assert_se(unit_get_cgroup_mask(grandchild) == 0); | |
72 | assert_se(unit_get_cgroup_mask(parent_deep) == CGROUP_MEMORY); | |
73 | assert_se(unit_get_cgroup_mask(parent) == CGROUP_BLKIO); | |
74 | assert_se(unit_get_cgroup_mask(root) == 0); | |
bc432dc7 LP |
75 | |
76 | /* Verify aggregation of member masks */ | |
bdf7026e TA |
77 | assert_se(unit_get_members_mask(son) == 0); |
78 | assert_se(unit_get_members_mask(daughter) == 0); | |
79 | assert_se(unit_get_members_mask(grandchild) == 0); | |
80 | assert_se(unit_get_members_mask(parent_deep) == 0); | |
81 | assert_se(unit_get_members_mask(parent) == (CGROUP_CPU | CGROUP_CPUACCT | CGROUP_MEMORY)); | |
82 | assert_se(unit_get_members_mask(root) == (CGROUP_CPU | CGROUP_CPUACCT | CGROUP_BLKIO | CGROUP_MEMORY)); | |
bc432dc7 LP |
83 | |
84 | /* Verify aggregation of sibling masks. */ | |
bdf7026e TA |
85 | assert_se(unit_get_siblings_mask(son) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_MEMORY) & m->cgroup_supported)); |
86 | assert_se(unit_get_siblings_mask(daughter) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_MEMORY) & m->cgroup_supported)); | |
87 | assert_se(unit_get_siblings_mask(grandchild) == 0); | |
88 | assert_se(unit_get_siblings_mask(parent_deep) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_MEMORY) & m->cgroup_supported)); | |
89 | assert_se(unit_get_siblings_mask(parent) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_BLKIO | CGROUP_MEMORY) & m->cgroup_supported)); | |
90 | assert_se(unit_get_siblings_mask(root) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_BLKIO | CGROUP_MEMORY) & m->cgroup_supported)); | |
bc432dc7 LP |
91 | |
92 | /* Verify aggregation of target masks. */ | |
bdf7026e TA |
93 | assert_se(unit_get_target_mask(son) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_MEMORY) & m->cgroup_supported)); |
94 | assert_se(unit_get_target_mask(daughter) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_MEMORY) & m->cgroup_supported)); | |
95 | assert_se(unit_get_target_mask(grandchild) == 0); | |
96 | assert_se(unit_get_target_mask(parent_deep) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_MEMORY) & m->cgroup_supported)); | |
97 | assert_se(unit_get_target_mask(parent) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_BLKIO | CGROUP_MEMORY) & m->cgroup_supported)); | |
98 | assert_se(unit_get_target_mask(root) == ((CGROUP_CPU | CGROUP_CPUACCT | CGROUP_BLKIO | CGROUP_MEMORY) & m->cgroup_supported)); | |
6414b7c9 DS |
99 | |
100 | manager_free(m); | |
101 | ||
102 | return 0; | |
103 | } | |
104 | ||
105 | int main(int argc, char* argv[]) { | |
106 | int rc = 0; | |
107 | TEST_REQ_RUNNING_SYSTEMD(rc = test_cgroup_mask()); | |
108 | return rc; | |
109 | } |