1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2013 Zbigniew Jędrzejewski-Szmek
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.
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.
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/>.
25 #include "cgroup-util.h"
27 static void check_p_d_u(const char *path, int code, const char *result) {
28 _cleanup_free_ char *unit = NULL;
30 assert_se(cg_path_decode_unit(path, &unit) == code);
31 assert_se(streq_ptr(unit, result));
34 static void test_path_decode_unit(void) {
35 check_p_d_u("getty@.service/getty@tty2.service", 0, "getty@tty2.service");
36 check_p_d_u("getty@.service/getty@tty2.service/xxx", 0, "getty@tty2.service");
37 check_p_d_u("getty@.service/", -EINVAL, NULL);
38 check_p_d_u("getty@.service", -EINVAL, NULL);
39 check_p_d_u("getty.service", 0, "getty.service");
40 check_p_d_u("getty", -EINVAL, NULL);
43 static void check_p_g_u(const char *path, int code, const char *result) {
44 _cleanup_free_ char *unit = NULL;
46 assert_se(cg_path_get_unit(path, &unit) == code);
47 assert_se(streq_ptr(unit, result));
50 static void check_p_g_u_u(const char *path, int code, const char *result) {
51 _cleanup_free_ char *unit = NULL;
53 assert_se(cg_path_get_user_unit(path, &unit) == code);
54 assert_se(streq_ptr(unit, result));
57 static void test_path_get_unit(void) {
58 check_p_g_u("/system/foobar.service/sdfdsaf", 0, "foobar.service");
59 check_p_g_u("/system/getty@.service/getty@tty5.service", 0, "getty@tty5.service");
60 check_p_g_u("/system/getty@.service/getty@tty5.service/aaa/bbb", 0, "getty@tty5.service");
61 check_p_g_u("/system/getty@.service/getty@tty5.service/", 0, "getty@tty5.service");
62 check_p_g_u("/system/getty@tty6.service/tty5", 0, "getty@tty6.service");
63 check_p_g_u("sadfdsafsda", -ENOENT, NULL);
64 check_p_g_u("/system/getty####@tty6.service/tty5", -EINVAL, NULL);
67 static void test_path_get_user_unit(void) {
68 check_p_g_u_u("/user/lennart/2/systemd-21548/foobar.service", 0, "foobar.service");
69 check_p_g_u_u("/user/lennart/2/systemd-21548/foobar.service/waldo", 0, "foobar.service");
70 check_p_g_u_u("/user/lennart/2/systemd-21548/foobar.service/waldo/uuuux", 0, "foobar.service");
71 check_p_g_u_u("/user/lennart/2/systemd-21548/waldo/waldo/uuuux", -EINVAL, NULL);
72 check_p_g_u_u("/user/lennart/2/foobar.service", -ENOENT, NULL);
73 check_p_g_u_u("/user/lennart/2/systemd-21548/foobar@.service/foobar@pie.service/pa/po", 0, "foobar@pie.service");
76 static void test_get_paths(void) {
77 _cleanup_free_ char *a = NULL, *b = NULL, *c = NULL, *d = NULL;
79 assert_se(cg_get_root_path(&a) >= 0);
80 log_info("Root = %s", a);
82 assert_se(cg_get_system_path(&b) >= 0);
83 log_info("System = %s", b);
85 assert_se(cg_get_user_path(&c) >= 0);
86 log_info("User = %s", c);
88 assert_se(cg_get_machine_path("harley", &d) >= 0);
89 log_info("Machine = %s", d);
92 static void test_proc(void) {
93 _cleanup_closedir_ DIR *d = NULL;
100 FOREACH_DIRENT(de, d, break) {
101 _cleanup_free_ char *path = NULL, *path_shifted = NULL, *session = NULL, *unit = NULL, *user_unit = NULL, *machine = NULL, *prefix = NULL;
103 uid_t uid = (uid_t) -1;
105 if (de->d_type != DT_DIR &&
106 de->d_type != DT_UNKNOWN)
109 r = parse_pid(de->d_name, &pid);
113 if (is_kernel_thread(pid))
116 cg_pid_get_path(SYSTEMD_CGROUP_CONTROLLER, pid, &path);
117 cg_pid_get_path_shifted(pid, &prefix, &path_shifted);
118 cg_pid_get_owner_uid(pid, &uid);
119 cg_pid_get_session(pid, &session);
120 cg_pid_get_unit(pid, &unit);
121 cg_pid_get_user_unit(pid, &user_unit);
122 cg_pid_get_machine_name(pid, &machine);
124 printf("%lu\t%s\t%s\t%s\t%lu\t%s\t%s\t%s\t%s\n",
137 static void test_escape_one(const char *s, const char *r) {
138 _cleanup_free_ char *b;
142 assert_se(streq(b, r));
144 assert_se(streq(cg_unescape(b), s));
147 static void test_escape(void) {
148 test_escape_one("foobar", "foobar");
149 test_escape_one(".foobar", "_.foobar");
150 test_escape_one("foobar.service", "foobar.service");
151 test_escape_one("cgroup.service", "_cgroup.service");
152 test_escape_one("cpu.service", "_cpu.service");
153 test_escape_one("tasks", "_tasks");
154 test_escape_one("_foobar", "__foobar");
155 test_escape_one("", "_");
156 test_escape_one("_", "__");
157 test_escape_one(".", "_.");
160 static void test_controller_is_valid(void) {
161 assert_se(cg_controller_is_valid("foobar", false));
162 assert_se(cg_controller_is_valid("foo_bar", false));
163 assert_se(cg_controller_is_valid("name=foo", true));
164 assert_se(!cg_controller_is_valid("", false));
165 assert_se(!cg_controller_is_valid("name=", true));
166 assert_se(!cg_controller_is_valid("=", false));
167 assert_se(!cg_controller_is_valid("cpu,cpuacct", false));
168 assert_se(!cg_controller_is_valid("_", false));
169 assert_se(!cg_controller_is_valid("_foobar", false));
170 assert_se(!cg_controller_is_valid("tatü", false));
174 test_path_decode_unit();
175 test_path_get_unit();
176 test_path_get_user_unit();
180 test_controller_is_valid();