Branch data Line data Source code
1 : : /* SPDX-License-Identifier: LGPL-2.1+ */
2 : :
3 : : #include <dirent.h>
4 : : #include <errno.h>
5 : : #include <stddef.h>
6 : : #include <stdio.h>
7 : : #include <stdlib.h>
8 : : #include <string.h>
9 : :
10 : : #include "alloc-util.h"
11 : : #include "bus-error.h"
12 : : #include "bus-util.h"
13 : : #include "cgroup-show.h"
14 : : #include "cgroup-util.h"
15 : : #include "env-file.h"
16 : : #include "fd-util.h"
17 : : #include "format-util.h"
18 : : #include "locale-util.h"
19 : : #include "macro.h"
20 : : #include "output-mode.h"
21 : : #include "path-util.h"
22 : : #include "process-util.h"
23 : : #include "sort-util.h"
24 : : #include "string-util.h"
25 : : #include "terminal-util.h"
26 : : #include "unit-name.h"
27 : :
28 : 0 : static void show_pid_array(
29 : : pid_t pids[],
30 : : unsigned n_pids,
31 : : const char *prefix,
32 : : size_t n_columns,
33 : : bool extra,
34 : : bool more,
35 : : OutputFlags flags) {
36 : :
37 : : unsigned i, j, pid_width;
38 : :
39 [ # # ]: 0 : if (n_pids == 0)
40 : 0 : return;
41 : :
42 : 0 : typesafe_qsort(pids, n_pids, pid_compare_func);
43 : :
44 : : /* Filter duplicates */
45 [ # # ]: 0 : for (j = 0, i = 1; i < n_pids; i++) {
46 [ # # ]: 0 : if (pids[i] == pids[j])
47 : 0 : continue;
48 : 0 : pids[++j] = pids[i];
49 : : }
50 : 0 : n_pids = j + 1;
51 [ # # ]: 0 : pid_width = DECIMAL_STR_WIDTH(pids[j]);
52 : :
53 [ # # ]: 0 : if (flags & OUTPUT_FULL_WIDTH)
54 : 0 : n_columns = SIZE_MAX;
55 : : else {
56 [ # # ]: 0 : if (n_columns > pid_width + 3) /* something like "├─1114784 " */
57 : 0 : n_columns -= pid_width + 3;
58 : : else
59 : 0 : n_columns = 20;
60 : : }
61 [ # # ]: 0 : for (i = 0; i < n_pids; i++) {
62 : 0 : _cleanup_free_ char *t = NULL;
63 : :
64 : 0 : (void) get_process_cmdline(pids[i], n_columns,
65 : : PROCESS_CMDLINE_COMM_FALLBACK | PROCESS_CMDLINE_USE_LOCALE,
66 : : &t);
67 : :
68 [ # # ]: 0 : if (extra)
69 : 0 : printf("%s%s ", prefix, special_glyph(SPECIAL_GLYPH_TRIANGULAR_BULLET));
70 : : else
71 [ # # # # ]: 0 : printf("%s%s", prefix, special_glyph(((more || i < n_pids-1) ? SPECIAL_GLYPH_TREE_BRANCH : SPECIAL_GLYPH_TREE_RIGHT)));
72 : :
73 : 0 : printf("%*"PID_PRI" %s\n", pid_width, pids[i], strna(t));
74 : : }
75 : : }
76 : :
77 : 0 : static int show_cgroup_one_by_path(
78 : : const char *path,
79 : : const char *prefix,
80 : : size_t n_columns,
81 : : bool more,
82 : : OutputFlags flags) {
83 : :
84 : : char *fn;
85 : 0 : _cleanup_fclose_ FILE *f = NULL;
86 : 0 : size_t n = 0, n_allocated = 0;
87 : 0 : _cleanup_free_ pid_t *pids = NULL;
88 : 0 : _cleanup_free_ char *p = NULL;
89 : : pid_t pid;
90 : : int r;
91 : :
92 : 0 : r = cg_mangle_path(path, &p);
93 [ # # ]: 0 : if (r < 0)
94 : 0 : return r;
95 : :
96 [ # # # # : 0 : fn = strjoina(p, "/cgroup.procs");
# # # # #
# # # ]
97 : 0 : f = fopen(fn, "re");
98 [ # # ]: 0 : if (!f)
99 : 0 : return -errno;
100 : :
101 [ # # ]: 0 : while ((r = cg_read_pid(f, &pid)) > 0) {
102 : :
103 [ # # # # ]: 0 : if (!(flags & OUTPUT_KERNEL_THREADS) && is_kernel_thread(pid) > 0)
104 : 0 : continue;
105 : :
106 [ # # ]: 0 : if (!GREEDY_REALLOC(pids, n_allocated, n + 1))
107 : 0 : return -ENOMEM;
108 : :
109 [ # # ]: 0 : assert(n < n_allocated);
110 : 0 : pids[n++] = pid;
111 : : }
112 : :
113 [ # # ]: 0 : if (r < 0)
114 : 0 : return r;
115 : :
116 : 0 : show_pid_array(pids, n, prefix, n_columns, false, more, flags);
117 : :
118 : 0 : return 0;
119 : : }
120 : :
121 : 0 : int show_cgroup_by_path(
122 : : const char *path,
123 : : const char *prefix,
124 : : size_t n_columns,
125 : : OutputFlags flags) {
126 : :
127 : 0 : _cleanup_free_ char *fn = NULL, *p1 = NULL, *last = NULL, *p2 = NULL;
128 : 0 : _cleanup_closedir_ DIR *d = NULL;
129 : 0 : char *gn = NULL;
130 : 0 : bool shown_pids = false;
131 : : int r;
132 : :
133 [ # # ]: 0 : assert(path);
134 : :
135 [ # # ]: 0 : if (n_columns <= 0)
136 : 0 : n_columns = columns();
137 : :
138 : 0 : prefix = strempty(prefix);
139 : :
140 : 0 : r = cg_mangle_path(path, &fn);
141 [ # # ]: 0 : if (r < 0)
142 : 0 : return r;
143 : :
144 : 0 : d = opendir(fn);
145 [ # # ]: 0 : if (!d)
146 : 0 : return -errno;
147 : :
148 [ # # ]: 0 : while ((r = cg_read_subgroup(d, &gn)) > 0) {
149 [ # # # ]: 0 : _cleanup_free_ char *k = NULL;
150 : :
151 : 0 : k = path_join(fn, gn);
152 : 0 : free(gn);
153 [ # # ]: 0 : if (!k)
154 : 0 : return -ENOMEM;
155 : :
156 [ # # # # ]: 0 : if (!(flags & OUTPUT_SHOW_ALL) && cg_is_empty_recursive(NULL, k) > 0)
157 : 0 : continue;
158 : :
159 [ # # ]: 0 : if (!shown_pids) {
160 : 0 : show_cgroup_one_by_path(path, prefix, n_columns, true, flags);
161 : 0 : shown_pids = true;
162 : : }
163 : :
164 [ # # ]: 0 : if (last) {
165 : 0 : printf("%s%s%s\n", prefix, special_glyph(SPECIAL_GLYPH_TREE_BRANCH), cg_unescape(basename(last)));
166 : :
167 [ # # ]: 0 : if (!p1) {
168 : 0 : p1 = strjoin(prefix, special_glyph(SPECIAL_GLYPH_TREE_VERTICAL));
169 [ # # ]: 0 : if (!p1)
170 : 0 : return -ENOMEM;
171 : : }
172 : :
173 : 0 : show_cgroup_by_path(last, p1, n_columns-2, flags);
174 : 0 : free(last);
175 : : }
176 : :
177 : 0 : last = TAKE_PTR(k);
178 : : }
179 : :
180 [ # # ]: 0 : if (r < 0)
181 : 0 : return r;
182 : :
183 [ # # ]: 0 : if (!shown_pids)
184 : 0 : show_cgroup_one_by_path(path, prefix, n_columns, !!last, flags);
185 : :
186 [ # # ]: 0 : if (last) {
187 : 0 : printf("%s%s%s\n", prefix, special_glyph(SPECIAL_GLYPH_TREE_RIGHT), cg_unescape(basename(last)));
188 : :
189 [ # # ]: 0 : if (!p2) {
190 : 0 : p2 = strjoin(prefix, " ");
191 [ # # ]: 0 : if (!p2)
192 : 0 : return -ENOMEM;
193 : : }
194 : :
195 : 0 : show_cgroup_by_path(last, p2, n_columns-2, flags);
196 : : }
197 : :
198 : 0 : return 0;
199 : : }
200 : :
201 : 0 : int show_cgroup(const char *controller,
202 : : const char *path,
203 : : const char *prefix,
204 : : size_t n_columns,
205 : : OutputFlags flags) {
206 : 0 : _cleanup_free_ char *p = NULL;
207 : : int r;
208 : :
209 [ # # ]: 0 : assert(path);
210 : :
211 : 0 : r = cg_get_path(controller, path, NULL, &p);
212 [ # # ]: 0 : if (r < 0)
213 : 0 : return r;
214 : :
215 : 0 : return show_cgroup_by_path(p, prefix, n_columns, flags);
216 : : }
217 : :
218 : 0 : static int show_extra_pids(
219 : : const char *controller,
220 : : const char *path,
221 : : const char *prefix,
222 : : size_t n_columns,
223 : : const pid_t pids[],
224 : : unsigned n_pids,
225 : : OutputFlags flags) {
226 : :
227 : 0 : _cleanup_free_ pid_t *copy = NULL;
228 : : unsigned i, j;
229 : : int r;
230 : :
231 [ # # ]: 0 : assert(path);
232 : :
233 [ # # ]: 0 : if (n_pids <= 0)
234 : 0 : return 0;
235 : :
236 [ # # ]: 0 : if (n_columns <= 0)
237 : 0 : n_columns = columns();
238 : :
239 : 0 : prefix = strempty(prefix);
240 : :
241 : 0 : copy = new(pid_t, n_pids);
242 [ # # ]: 0 : if (!copy)
243 : 0 : return -ENOMEM;
244 : :
245 [ # # ]: 0 : for (i = 0, j = 0; i < n_pids; i++) {
246 [ # # # ]: 0 : _cleanup_free_ char *k = NULL;
247 : :
248 : 0 : r = cg_pid_get_path(controller, pids[i], &k);
249 [ # # ]: 0 : if (r < 0)
250 : 0 : return r;
251 : :
252 [ # # ]: 0 : if (path_startswith(k, path))
253 : 0 : continue;
254 : :
255 : 0 : copy[j++] = pids[i];
256 : : }
257 : :
258 : 0 : show_pid_array(copy, j, prefix, n_columns, true, false, flags);
259 : :
260 : 0 : return 0;
261 : : }
262 : :
263 : 0 : int show_cgroup_and_extra(
264 : : const char *controller,
265 : : const char *path,
266 : : const char *prefix,
267 : : size_t n_columns,
268 : : const pid_t extra_pids[],
269 : : unsigned n_extra_pids,
270 : : OutputFlags flags) {
271 : :
272 : : int r;
273 : :
274 [ # # ]: 0 : assert(path);
275 : :
276 : 0 : r = show_cgroup(controller, path, prefix, n_columns, flags);
277 [ # # ]: 0 : if (r < 0)
278 : 0 : return r;
279 : :
280 : 0 : return show_extra_pids(controller, path, prefix, n_columns, extra_pids, n_extra_pids, flags);
281 : : }
282 : :
283 : 0 : int show_cgroup_get_unit_path_and_warn(
284 : : sd_bus *bus,
285 : : const char *unit,
286 : : char **ret) {
287 : :
288 : 0 : _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
289 : 0 : _cleanup_free_ char *path = NULL;
290 : : int r;
291 : :
292 : 0 : path = unit_dbus_path_from_name(unit);
293 [ # # ]: 0 : if (!path)
294 : 0 : return log_oom();
295 : :
296 : 0 : r = sd_bus_get_property_string(
297 : : bus,
298 : : "org.freedesktop.systemd1",
299 : : path,
300 : : unit_dbus_interface_from_name(unit),
301 : : "ControlGroup",
302 : : &error,
303 : : ret);
304 [ # # ]: 0 : if (r < 0)
305 [ # # ]: 0 : return log_error_errno(r, "Failed to query unit control group path: %s",
306 : : bus_error_message(&error, r));
307 : :
308 : 0 : return 0;
309 : : }
310 : :
311 : 0 : int show_cgroup_get_path_and_warn(
312 : : const char *machine,
313 : : const char *prefix,
314 : : char **ret) {
315 : :
316 : : int r;
317 : 0 : _cleanup_free_ char *root = NULL;
318 : :
319 [ # # ]: 0 : if (machine) {
320 [ # # ]: 0 : _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL;
321 [ # # ]: 0 : _cleanup_free_ char *unit = NULL;
322 : : const char *m;
323 : :
324 [ # # # # : 0 : m = strjoina("/run/systemd/machines/", machine);
# # # # #
# # # ]
325 : 0 : r = parse_env_file(NULL, m, "SCOPE", &unit);
326 [ # # ]: 0 : if (r < 0)
327 [ # # ]: 0 : return log_error_errno(r, "Failed to load machine data: %m");
328 : :
329 : 0 : r = bus_connect_transport_systemd(BUS_TRANSPORT_LOCAL, NULL, false, &bus);
330 [ # # ]: 0 : if (r < 0)
331 [ # # ]: 0 : return log_error_errno(r, "Failed to create bus connection: %m");
332 : :
333 : 0 : r = show_cgroup_get_unit_path_and_warn(bus, unit, &root);
334 [ # # ]: 0 : if (r < 0)
335 : 0 : return r;
336 : : } else {
337 : 0 : r = cg_get_root_path(&root);
338 [ # # ]: 0 : if (r == -ENOMEDIUM)
339 [ # # ]: 0 : return log_error_errno(r, "Failed to get root control group path.\n"
340 : : "No cgroup filesystem mounted on /sys/fs/cgroup");
341 [ # # ]: 0 : else if (r < 0)
342 [ # # ]: 0 : return log_error_errno(r, "Failed to get root control group path: %m");
343 : : }
344 : :
345 [ # # ]: 0 : if (prefix) {
346 : : char *t;
347 : :
348 : 0 : t = strjoin(root, prefix);
349 [ # # ]: 0 : if (!t)
350 : 0 : return log_oom();
351 : :
352 : 0 : *ret = t;
353 : : } else
354 : 0 : *ret = TAKE_PTR(root);
355 : :
356 : 0 : return 0;
357 : : }
|
