]> git.proxmox.com Git - mirror_lxc.git/blob - src/lxc/cgroups/cgfsng.c
Revert "cgfsng: avoid tiny race window"
[mirror_lxc.git] / src / lxc / cgroups / cgfsng.c
1 /*
2 * lxc: linux Container library
3 *
4 * Copyright © 2016 Canonical Ltd.
5 *
6 * Authors:
7 * Serge Hallyn <serge.hallyn@ubuntu.com>
8 * Christian Brauner <christian.brauner@ubuntu.com>
9 *
10 * This library is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License as published by the Free Software Foundation; either
13 * version 2.1 of the License, or (at your option) any later version.
14 *
15 * This library is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * Lesser General Public License for more details.
19 *
20 * You should have received a copy of the GNU Lesser General Public
21 * License along with this library; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 */
24
25 /*
26 * cgfs-ng.c: this is a new, simplified implementation of a filesystem
27 * cgroup backend. The original cgfs.c was designed to be as flexible
28 * as possible. It would try to find cgroup filesystems no matter where
29 * or how you had them mounted, and deduce the most usable mount for
30 * each controller.
31 *
32 * This new implementation assumes that cgroup filesystems are mounted
33 * under /sys/fs/cgroup/clist where clist is either the controller, or
34 * a comman-separated list of controllers.
35 */
36
37 #ifndef _GNU_SOURCE
38 #define _GNU_SOURCE 1
39 #endif
40 #include <ctype.h>
41 #include <dirent.h>
42 #include <errno.h>
43 #include <grp.h>
44 #include <linux/kdev_t.h>
45 #include <linux/types.h>
46 #include <stdint.h>
47 #include <stdio.h>
48 #include <stdlib.h>
49 #include <string.h>
50 #include <sys/types.h>
51 #include <unistd.h>
52
53 #include "caps.h"
54 #include "cgroup.h"
55 #include "cgroup_utils.h"
56 #include "commands.h"
57 #include "conf.h"
58 #include "config.h"
59 #include "log.h"
60 #include "macro.h"
61 #include "storage/storage.h"
62 #include "utils.h"
63
64 #ifndef HAVE_STRLCPY
65 #include "include/strlcpy.h"
66 #endif
67
68 #ifndef HAVE_STRLCAT
69 #include "include/strlcat.h"
70 #endif
71
72 lxc_log_define(cgfsng, cgroup);
73
74 static void free_string_list(char **clist)
75 {
76 int i;
77
78 if (!clist)
79 return;
80
81 for (i = 0; clist[i]; i++)
82 free(clist[i]);
83
84 free(clist);
85 }
86
87 /* Allocate a pointer, do not fail. */
88 static void *must_alloc(size_t sz)
89 {
90 return must_realloc(NULL, sz);
91 }
92
93 /* Given a pointer to a null-terminated array of pointers, realloc to add one
94 * entry, and point the new entry to NULL. Do not fail. Return the index to the
95 * second-to-last entry - that is, the one which is now available for use
96 * (keeping the list null-terminated).
97 */
98 static int append_null_to_list(void ***list)
99 {
100 int newentry = 0;
101
102 if (*list)
103 for (; (*list)[newentry]; newentry++)
104 ;
105
106 *list = must_realloc(*list, (newentry + 2) * sizeof(void **));
107 (*list)[newentry + 1] = NULL;
108 return newentry;
109 }
110
111 /* Given a null-terminated array of strings, check whether @entry is one of the
112 * strings.
113 */
114 static bool string_in_list(char **list, const char *entry)
115 {
116 int i;
117
118 if (!list)
119 return false;
120
121 for (i = 0; list[i]; i++)
122 if (strcmp(list[i], entry) == 0)
123 return true;
124
125 return false;
126 }
127
128 /* Return a copy of @entry prepending "name=", i.e. turn "systemd" into
129 * "name=systemd". Do not fail.
130 */
131 static char *cg_legacy_must_prefix_named(char *entry)
132 {
133 size_t len;
134 char *prefixed;
135
136 len = strlen(entry);
137 prefixed = must_alloc(len + 6);
138
139 memcpy(prefixed, "name=", STRLITERALLEN("name="));
140 memcpy(prefixed + STRLITERALLEN("name="), entry, len);
141 prefixed[len + 5] = '\0';
142
143 return prefixed;
144 }
145
146 /* Append an entry to the clist. Do not fail. @clist must be NULL the first time
147 * we are called.
148 *
149 * We also handle named subsystems here. Any controller which is not a kernel
150 * subsystem, we prefix "name=". Any which is both a kernel and named subsystem,
151 * we refuse to use because we're not sure which we have here.
152 * (TODO: We could work around this in some cases by just remounting to be
153 * unambiguous, or by comparing mountpoint contents with current cgroup.)
154 *
155 * The last entry will always be NULL.
156 */
157 static void must_append_controller(char **klist, char **nlist, char ***clist,
158 char *entry)
159 {
160 int newentry;
161 char *copy;
162
163 if (string_in_list(klist, entry) && string_in_list(nlist, entry)) {
164 ERROR("Refusing to use ambiguous controller \"%s\"", entry);
165 ERROR("It is both a named and kernel subsystem");
166 return;
167 }
168
169 newentry = append_null_to_list((void ***)clist);
170
171 if (strncmp(entry, "name=", 5) == 0)
172 copy = must_copy_string(entry);
173 else if (string_in_list(klist, entry))
174 copy = must_copy_string(entry);
175 else
176 copy = cg_legacy_must_prefix_named(entry);
177
178 (*clist)[newentry] = copy;
179 }
180
181 /* Given a handler's cgroup data, return the struct hierarchy for the controller
182 * @c, or NULL if there is none.
183 */
184 struct hierarchy *get_hierarchy(struct cgroup_ops *ops, const char *controller)
185 {
186 int i;
187
188 errno = ENOENT;
189
190 if (!ops->hierarchies) {
191 TRACE("There are no useable cgroup controllers");
192 return NULL;
193 }
194
195 for (i = 0; ops->hierarchies[i]; i++) {
196 if (!controller) {
197 /* This is the empty unified hierarchy. */
198 if (ops->hierarchies[i]->controllers &&
199 !ops->hierarchies[i]->controllers[0])
200 return ops->hierarchies[i];
201
202 continue;
203 }
204
205 if (string_in_list(ops->hierarchies[i]->controllers, controller))
206 return ops->hierarchies[i];
207 }
208
209 if (controller)
210 WARN("There is no useable %s controller", controller);
211 else
212 WARN("There is no empty unified cgroup hierarchy");
213
214 return NULL;
215 }
216
217 #define BATCH_SIZE 50
218 static void batch_realloc(char **mem, size_t oldlen, size_t newlen)
219 {
220 int newbatches = (newlen / BATCH_SIZE) + 1;
221 int oldbatches = (oldlen / BATCH_SIZE) + 1;
222
223 if (!*mem || newbatches > oldbatches) {
224 *mem = must_realloc(*mem, newbatches * BATCH_SIZE);
225 }
226 }
227
228 static void append_line(char **dest, size_t oldlen, char *new, size_t newlen)
229 {
230 size_t full = oldlen + newlen;
231
232 batch_realloc(dest, oldlen, full + 1);
233
234 memcpy(*dest + oldlen, new, newlen + 1);
235 }
236
237 /* Slurp in a whole file */
238 static char *read_file(const char *fnam)
239 {
240 FILE *f;
241 char *line = NULL, *buf = NULL;
242 size_t len = 0, fulllen = 0;
243 int linelen;
244
245 f = fopen(fnam, "r");
246 if (!f)
247 return NULL;
248 while ((linelen = getline(&line, &len, f)) != -1) {
249 append_line(&buf, fulllen, line, linelen);
250 fulllen += linelen;
251 }
252 fclose(f);
253 free(line);
254 return buf;
255 }
256
257 /* Taken over modified from the kernel sources. */
258 #define NBITS 32 /* bits in uint32_t */
259 #define DIV_ROUND_UP(n, d) (((n) + (d)-1) / (d))
260 #define BITS_TO_LONGS(nr) DIV_ROUND_UP(nr, NBITS)
261
262 static void set_bit(unsigned bit, uint32_t *bitarr)
263 {
264 bitarr[bit / NBITS] |= (1 << (bit % NBITS));
265 }
266
267 static void clear_bit(unsigned bit, uint32_t *bitarr)
268 {
269 bitarr[bit / NBITS] &= ~(1 << (bit % NBITS));
270 }
271
272 static bool is_set(unsigned bit, uint32_t *bitarr)
273 {
274 return (bitarr[bit / NBITS] & (1 << (bit % NBITS))) != 0;
275 }
276
277 /* Create cpumask from cpulist aka turn:
278 *
279 * 0,2-3
280 *
281 * into bit array
282 *
283 * 1 0 1 1
284 */
285 static uint32_t *lxc_cpumask(char *buf, size_t nbits)
286 {
287 char *token;
288 size_t arrlen;
289 uint32_t *bitarr;
290
291 arrlen = BITS_TO_LONGS(nbits);
292 bitarr = calloc(arrlen, sizeof(uint32_t));
293 if (!bitarr)
294 return NULL;
295
296 lxc_iterate_parts(token, buf, ",") {
297 errno = 0;
298 unsigned end, start;
299 char *range;
300
301 start = strtoul(token, NULL, 0);
302 end = start;
303 range = strchr(token, '-');
304 if (range)
305 end = strtoul(range + 1, NULL, 0);
306
307 if (!(start <= end)) {
308 free(bitarr);
309 return NULL;
310 }
311
312 if (end >= nbits) {
313 free(bitarr);
314 return NULL;
315 }
316
317 while (start <= end)
318 set_bit(start++, bitarr);
319 }
320
321 return bitarr;
322 }
323
324 /* Turn cpumask into simple, comma-separated cpulist. */
325 static char *lxc_cpumask_to_cpulist(uint32_t *bitarr, size_t nbits)
326 {
327 int ret;
328 size_t i;
329 char **cpulist = NULL;
330 char numstr[INTTYPE_TO_STRLEN(size_t)] = {0};
331
332 for (i = 0; i <= nbits; i++) {
333 if (!is_set(i, bitarr))
334 continue;
335
336 ret = snprintf(numstr, sizeof(numstr), "%zu", i);
337 if (ret < 0 || (size_t)ret >= sizeof(numstr)) {
338 lxc_free_array((void **)cpulist, free);
339 return NULL;
340 }
341
342 ret = lxc_append_string(&cpulist, numstr);
343 if (ret < 0) {
344 lxc_free_array((void **)cpulist, free);
345 return NULL;
346 }
347 }
348
349 if (!cpulist)
350 return NULL;
351
352 return lxc_string_join(",", (const char **)cpulist, false);
353 }
354
355 static ssize_t get_max_cpus(char *cpulist)
356 {
357 char *c1, *c2;
358 char *maxcpus = cpulist;
359 size_t cpus = 0;
360
361 c1 = strrchr(maxcpus, ',');
362 if (c1)
363 c1++;
364
365 c2 = strrchr(maxcpus, '-');
366 if (c2)
367 c2++;
368
369 if (!c1 && !c2)
370 c1 = maxcpus;
371 else if (c1 > c2)
372 c2 = c1;
373 else if (c1 < c2)
374 c1 = c2;
375 else if (!c1 && c2)
376 c1 = c2;
377
378 errno = 0;
379 cpus = strtoul(c1, NULL, 0);
380 if (errno != 0)
381 return -1;
382
383 return cpus;
384 }
385
386 #define __ISOL_CPUS "/sys/devices/system/cpu/isolated"
387 static bool cg_legacy_filter_and_set_cpus(char *path, bool am_initialized)
388 {
389 int ret;
390 ssize_t i;
391 char *lastslash, *fpath, oldv;
392 ssize_t maxisol = 0, maxposs = 0;
393 char *cpulist = NULL, *isolcpus = NULL, *posscpus = NULL;
394 uint32_t *isolmask = NULL, *possmask = NULL;
395 bool bret = false, flipped_bit = false;
396
397 lastslash = strrchr(path, '/');
398 if (!lastslash) {
399 ERROR("Failed to detect \"/\" in \"%s\"", path);
400 return bret;
401 }
402 oldv = *lastslash;
403 *lastslash = '\0';
404 fpath = must_make_path(path, "cpuset.cpus", NULL);
405 posscpus = read_file(fpath);
406 if (!posscpus) {
407 SYSERROR("Failed to read file \"%s\"", fpath);
408 goto on_error;
409 }
410
411 /* Get maximum number of cpus found in possible cpuset. */
412 maxposs = get_max_cpus(posscpus);
413 if (maxposs < 0 || maxposs >= INT_MAX - 1)
414 goto on_error;
415
416 if (!file_exists(__ISOL_CPUS)) {
417 /* This system doesn't expose isolated cpus. */
418 DEBUG("The path \""__ISOL_CPUS"\" to read isolated cpus from does not exist");
419 cpulist = posscpus;
420 /* No isolated cpus but we weren't already initialized by
421 * someone. We should simply copy the parents cpuset.cpus
422 * values.
423 */
424 if (!am_initialized) {
425 DEBUG("Copying cpu settings of parent cgroup");
426 goto copy_parent;
427 }
428 /* No isolated cpus but we were already initialized by someone.
429 * Nothing more to do for us.
430 */
431 goto on_success;
432 }
433
434 isolcpus = read_file(__ISOL_CPUS);
435 if (!isolcpus) {
436 SYSERROR("Failed to read file \""__ISOL_CPUS"\"");
437 goto on_error;
438 }
439 if (!isdigit(isolcpus[0])) {
440 TRACE("No isolated cpus detected");
441 cpulist = posscpus;
442 /* No isolated cpus but we weren't already initialized by
443 * someone. We should simply copy the parents cpuset.cpus
444 * values.
445 */
446 if (!am_initialized) {
447 DEBUG("Copying cpu settings of parent cgroup");
448 goto copy_parent;
449 }
450 /* No isolated cpus but we were already initialized by someone.
451 * Nothing more to do for us.
452 */
453 goto on_success;
454 }
455
456 /* Get maximum number of cpus found in isolated cpuset. */
457 maxisol = get_max_cpus(isolcpus);
458 if (maxisol < 0 || maxisol >= INT_MAX - 1)
459 goto on_error;
460
461 if (maxposs < maxisol)
462 maxposs = maxisol;
463 maxposs++;
464
465 possmask = lxc_cpumask(posscpus, maxposs);
466 if (!possmask) {
467 ERROR("Failed to create cpumask for possible cpus");
468 goto on_error;
469 }
470
471 isolmask = lxc_cpumask(isolcpus, maxposs);
472 if (!isolmask) {
473 ERROR("Failed to create cpumask for isolated cpus");
474 goto on_error;
475 }
476
477 for (i = 0; i <= maxposs; i++) {
478 if (!is_set(i, isolmask) || !is_set(i, possmask))
479 continue;
480
481 flipped_bit = true;
482 clear_bit(i, possmask);
483 }
484
485 if (!flipped_bit) {
486 DEBUG("No isolated cpus present in cpuset");
487 goto on_success;
488 }
489 DEBUG("Removed isolated cpus from cpuset");
490
491 cpulist = lxc_cpumask_to_cpulist(possmask, maxposs);
492 if (!cpulist) {
493 ERROR("Failed to create cpu list");
494 goto on_error;
495 }
496
497 copy_parent:
498 *lastslash = oldv;
499 free(fpath);
500 fpath = must_make_path(path, "cpuset.cpus", NULL);
501 ret = lxc_write_to_file(fpath, cpulist, strlen(cpulist), false, 0666);
502 if (ret < 0) {
503 SYSERROR("Failed to write cpu list to \"%s\"", fpath);
504 goto on_error;
505 }
506
507 on_success:
508 bret = true;
509
510 on_error:
511 free(fpath);
512
513 free(isolcpus);
514 free(isolmask);
515
516 if (posscpus != cpulist)
517 free(posscpus);
518 free(possmask);
519
520 free(cpulist);
521 return bret;
522 }
523
524 /* Copy contents of parent(@path)/@file to @path/@file */
525 static bool copy_parent_file(char *path, char *file)
526 {
527 int ret;
528 char *fpath, *lastslash, oldv;
529 int len = 0;
530 char *value = NULL;
531
532 lastslash = strrchr(path, '/');
533 if (!lastslash) {
534 ERROR("Failed to detect \"/\" in \"%s\"", path);
535 return false;
536 }
537 oldv = *lastslash;
538 *lastslash = '\0';
539 fpath = must_make_path(path, file, NULL);
540 len = lxc_read_from_file(fpath, NULL, 0);
541 if (len <= 0)
542 goto on_error;
543
544 value = must_alloc(len + 1);
545 ret = lxc_read_from_file(fpath, value, len);
546 if (ret != len)
547 goto on_error;
548 free(fpath);
549
550 *lastslash = oldv;
551 fpath = must_make_path(path, file, NULL);
552 ret = lxc_write_to_file(fpath, value, len, false, 0666);
553 if (ret < 0)
554 SYSERROR("Failed to write \"%s\" to file \"%s\"", value, fpath);
555 free(fpath);
556 free(value);
557 return ret >= 0;
558
559 on_error:
560 SYSERROR("Failed to read file \"%s\"", fpath);
561 free(fpath);
562 free(value);
563 return false;
564 }
565
566 /* Initialize the cpuset hierarchy in first directory of @gname and set
567 * cgroup.clone_children so that children inherit settings. Since the
568 * h->base_path is populated by init or ourselves, we know it is already
569 * initialized.
570 */
571 static bool cg_legacy_handle_cpuset_hierarchy(struct hierarchy *h, char *cgname)
572 {
573 int ret;
574 char v;
575 char *cgpath, *clonechildrenpath, *slash;
576
577 if (!string_in_list(h->controllers, "cpuset"))
578 return true;
579
580 if (*cgname == '/')
581 cgname++;
582 slash = strchr(cgname, '/');
583 if (slash)
584 *slash = '\0';
585
586 cgpath = must_make_path(h->mountpoint, h->container_base_path, cgname, NULL);
587 if (slash)
588 *slash = '/';
589
590 ret = mkdir(cgpath, 0755);
591 if (ret < 0) {
592 if (errno != EEXIST) {
593 SYSERROR("Failed to create directory \"%s\"", cgpath);
594 free(cgpath);
595 return false;
596 }
597 }
598
599 clonechildrenpath = must_make_path(cgpath, "cgroup.clone_children", NULL);
600 /* unified hierarchy doesn't have clone_children */
601 if (!file_exists(clonechildrenpath)) {
602 free(clonechildrenpath);
603 free(cgpath);
604 return true;
605 }
606
607 ret = lxc_read_from_file(clonechildrenpath, &v, 1);
608 if (ret < 0) {
609 SYSERROR("Failed to read file \"%s\"", clonechildrenpath);
610 free(clonechildrenpath);
611 free(cgpath);
612 return false;
613 }
614
615 /* Make sure any isolated cpus are removed from cpuset.cpus. */
616 if (!cg_legacy_filter_and_set_cpus(cgpath, v == '1')) {
617 SYSERROR("Failed to remove isolated cpus");
618 free(clonechildrenpath);
619 free(cgpath);
620 return false;
621 }
622
623 /* Already set for us by someone else. */
624 if (v == '1') {
625 DEBUG("\"cgroup.clone_children\" was already set to \"1\"");
626 free(clonechildrenpath);
627 free(cgpath);
628 return true;
629 }
630
631 /* copy parent's settings */
632 if (!copy_parent_file(cgpath, "cpuset.mems")) {
633 SYSERROR("Failed to copy \"cpuset.mems\" settings");
634 free(cgpath);
635 free(clonechildrenpath);
636 return false;
637 }
638 free(cgpath);
639
640 ret = lxc_write_to_file(clonechildrenpath, "1", 1, false, 0666);
641 if (ret < 0) {
642 /* Set clone_children so children inherit our settings */
643 SYSERROR("Failed to write 1 to \"%s\"", clonechildrenpath);
644 free(clonechildrenpath);
645 return false;
646 }
647 free(clonechildrenpath);
648 return true;
649 }
650
651 /* Given two null-terminated lists of strings, return true if any string is in
652 * both.
653 */
654 static bool controller_lists_intersect(char **l1, char **l2)
655 {
656 int i;
657
658 if (!l1 || !l2)
659 return false;
660
661 for (i = 0; l1[i]; i++) {
662 if (string_in_list(l2, l1[i]))
663 return true;
664 }
665
666 return false;
667 }
668
669 /* For a null-terminated list of controllers @clist, return true if any of those
670 * controllers is already listed the null-terminated list of hierarchies @hlist.
671 * Realistically, if one is present, all must be present.
672 */
673 static bool controller_list_is_dup(struct hierarchy **hlist, char **clist)
674 {
675 int i;
676
677 if (!hlist)
678 return false;
679
680 for (i = 0; hlist[i]; i++)
681 if (controller_lists_intersect(hlist[i]->controllers, clist))
682 return true;
683
684 return false;
685 }
686
687 /* Return true if the controller @entry is found in the null-terminated list of
688 * hierarchies @hlist.
689 */
690 static bool controller_found(struct hierarchy **hlist, char *entry)
691 {
692 int i;
693
694 if (!hlist)
695 return false;
696
697 for (i = 0; hlist[i]; i++)
698 if (string_in_list(hlist[i]->controllers, entry))
699 return true;
700
701 return false;
702 }
703
704 /* Return true if all of the controllers which we require have been found. The
705 * required list is freezer and anything in lxc.cgroup.use.
706 */
707 static bool all_controllers_found(struct cgroup_ops *ops)
708 {
709 char **cur;
710 struct hierarchy **hlist = ops->hierarchies;
711
712 if (!controller_found(hlist, "freezer")) {
713 ERROR("No freezer controller mountpoint found");
714 return false;
715 }
716
717 if (!ops->cgroup_use)
718 return true;
719
720 for (cur = ops->cgroup_use; cur && *cur; cur++)
721 if (!controller_found(hlist, *cur)) {
722 ERROR("No %s controller mountpoint found", *cur);
723 return false;
724 }
725
726 return true;
727 }
728
729 /* Get the controllers from a mountinfo line There are other ways we could get
730 * this info. For lxcfs, field 3 is /cgroup/controller-list. For cgroupfs, we
731 * could parse the mount options. But we simply assume that the mountpoint must
732 * be /sys/fs/cgroup/controller-list
733 */
734 static char **cg_hybrid_get_controllers(char **klist, char **nlist, char *line,
735 int type)
736 {
737 /* The fourth field is /sys/fs/cgroup/comma-delimited-controller-list
738 * for legacy hierarchies.
739 */
740 int i;
741 char *dup, *p2, *tok;
742 char *p = line, *sep = ",";
743 char **aret = NULL;
744
745 for (i = 0; i < 4; i++) {
746 p = strchr(p, ' ');
747 if (!p)
748 return NULL;
749 p++;
750 }
751
752 /* Note, if we change how mountinfo works, then our caller will need to
753 * verify /sys/fs/cgroup/ in this field.
754 */
755 if (strncmp(p, "/sys/fs/cgroup/", 15) != 0) {
756 ERROR("Found hierarchy not under /sys/fs/cgroup: \"%s\"", p);
757 return NULL;
758 }
759
760 p += 15;
761 p2 = strchr(p, ' ');
762 if (!p2) {
763 ERROR("Corrupt mountinfo");
764 return NULL;
765 }
766 *p2 = '\0';
767
768 if (type == CGROUP_SUPER_MAGIC) {
769 /* strdup() here for v1 hierarchies. Otherwise
770 * lxc_iterate_parts() will destroy mountpoints such as
771 * "/sys/fs/cgroup/cpu,cpuacct".
772 */
773 dup = strdup(p);
774 if (!dup)
775 return NULL;
776
777 lxc_iterate_parts(tok, dup, sep) {
778 must_append_controller(klist, nlist, &aret, tok);
779 }
780
781 free(dup);
782 }
783 *p2 = ' ';
784
785 return aret;
786 }
787
788 static char **cg_unified_make_empty_controller(void)
789 {
790 int newentry;
791 char **aret = NULL;
792
793 newentry = append_null_to_list((void ***)&aret);
794 aret[newentry] = NULL;
795 return aret;
796 }
797
798 static char **cg_unified_get_controllers(const char *file)
799 {
800 char *buf, *tok;
801 char *sep = " \t\n";
802 char **aret = NULL;
803
804 buf = read_file(file);
805 if (!buf)
806 return NULL;
807
808 lxc_iterate_parts(tok, buf, sep) {
809 int newentry;
810 char *copy;
811
812 newentry = append_null_to_list((void ***)&aret);
813 copy = must_copy_string(tok);
814 aret[newentry] = copy;
815 }
816
817 free(buf);
818 return aret;
819 }
820
821 static struct hierarchy *add_hierarchy(struct hierarchy ***h, char **clist, char *mountpoint,
822 char *container_base_path, int type)
823 {
824 struct hierarchy *new;
825 int newentry;
826
827 new = must_alloc(sizeof(*new));
828 new->controllers = clist;
829 new->mountpoint = mountpoint;
830 new->container_base_path = container_base_path;
831 new->container_full_path = NULL;
832 new->monitor_full_path = NULL;
833 new->version = type;
834
835 newentry = append_null_to_list((void ***)h);
836 (*h)[newentry] = new;
837 return new;
838 }
839
840 /* Get a copy of the mountpoint from @line, which is a line from
841 * /proc/self/mountinfo.
842 */
843 static char *cg_hybrid_get_mountpoint(char *line)
844 {
845 int i;
846 size_t len;
847 char *p2;
848 char *p = line, *sret = NULL;
849
850 for (i = 0; i < 4; i++) {
851 p = strchr(p, ' ');
852 if (!p)
853 return NULL;
854 p++;
855 }
856
857 if (strncmp(p, "/sys/fs/cgroup/", 15) != 0)
858 return NULL;
859
860 p2 = strchr(p + 15, ' ');
861 if (!p2)
862 return NULL;
863 *p2 = '\0';
864
865 len = strlen(p);
866 sret = must_alloc(len + 1);
867 memcpy(sret, p, len);
868 sret[len] = '\0';
869 return sret;
870 }
871
872 /* Given a multi-line string, return a null-terminated copy of the current line. */
873 static char *copy_to_eol(char *p)
874 {
875 char *p2 = strchr(p, '\n'), *sret;
876 size_t len;
877
878 if (!p2)
879 return NULL;
880
881 len = p2 - p;
882 sret = must_alloc(len + 1);
883 memcpy(sret, p, len);
884 sret[len] = '\0';
885 return sret;
886 }
887
888 /* cgline: pointer to character after the first ':' in a line in a \n-terminated
889 * /proc/self/cgroup file. Check whether controller c is present.
890 */
891 static bool controller_in_clist(char *cgline, char *c)
892 {
893 char *tok, *eol, *tmp;
894 size_t len;
895
896 eol = strchr(cgline, ':');
897 if (!eol)
898 return false;
899
900 len = eol - cgline;
901 tmp = alloca(len + 1);
902 memcpy(tmp, cgline, len);
903 tmp[len] = '\0';
904
905 lxc_iterate_parts(tok, tmp, ",") {
906 if (strcmp(tok, c) == 0)
907 return true;
908 }
909
910 return false;
911 }
912
913 /* @basecginfo is a copy of /proc/$$/cgroup. Return the current cgroup for
914 * @controller.
915 */
916 static char *cg_hybrid_get_current_cgroup(char *basecginfo, char *controller,
917 int type)
918 {
919 char *p = basecginfo;
920
921 for (;;) {
922 bool is_cgv2_base_cgroup = false;
923
924 /* cgroup v2 entry in "/proc/<pid>/cgroup": "0::/some/path" */
925 if ((type == CGROUP2_SUPER_MAGIC) && (*p == '0'))
926 is_cgv2_base_cgroup = true;
927
928 p = strchr(p, ':');
929 if (!p)
930 return NULL;
931 p++;
932
933 if (is_cgv2_base_cgroup || (controller && controller_in_clist(p, controller))) {
934 p = strchr(p, ':');
935 if (!p)
936 return NULL;
937 p++;
938 return copy_to_eol(p);
939 }
940
941 p = strchr(p, '\n');
942 if (!p)
943 return NULL;
944 p++;
945 }
946 }
947
948 static void must_append_string(char ***list, char *entry)
949 {
950 int newentry;
951 char *copy;
952
953 newentry = append_null_to_list((void ***)list);
954 copy = must_copy_string(entry);
955 (*list)[newentry] = copy;
956 }
957
958 static int get_existing_subsystems(char ***klist, char ***nlist)
959 {
960 FILE *f;
961 char *line = NULL;
962 size_t len = 0;
963
964 f = fopen("/proc/self/cgroup", "r");
965 if (!f)
966 return -1;
967
968 while (getline(&line, &len, f) != -1) {
969 char *p, *p2, *tok;
970 p = strchr(line, ':');
971 if (!p)
972 continue;
973 p++;
974 p2 = strchr(p, ':');
975 if (!p2)
976 continue;
977 *p2 = '\0';
978
979 /* If the kernel has cgroup v2 support, then /proc/self/cgroup
980 * contains an entry of the form:
981 *
982 * 0::/some/path
983 *
984 * In this case we use "cgroup2" as controller name.
985 */
986 if ((p2 - p) == 0) {
987 must_append_string(klist, "cgroup2");
988 continue;
989 }
990
991 lxc_iterate_parts(tok, p, ",") {
992 if (strncmp(tok, "name=", 5) == 0)
993 must_append_string(nlist, tok);
994 else
995 must_append_string(klist, tok);
996 }
997 }
998
999 free(line);
1000 fclose(f);
1001 return 0;
1002 }
1003
1004 static void trim(char *s)
1005 {
1006 size_t len;
1007
1008 len = strlen(s);
1009 while ((len > 1) && (s[len - 1] == '\n'))
1010 s[--len] = '\0';
1011 }
1012
1013 static void lxc_cgfsng_print_hierarchies(struct cgroup_ops *ops)
1014 {
1015 int i;
1016 struct hierarchy **it;
1017
1018 if (!ops->hierarchies) {
1019 TRACE(" No hierarchies found");
1020 return;
1021 }
1022
1023 TRACE(" Hierarchies:");
1024 for (i = 0, it = ops->hierarchies; it && *it; it++, i++) {
1025 int j;
1026 char **cit;
1027
1028 TRACE(" %d: base_cgroup: %s", i, (*it)->container_base_path ? (*it)->container_base_path : "(null)");
1029 TRACE(" mountpoint: %s", (*it)->mountpoint ? (*it)->mountpoint : "(null)");
1030 TRACE(" controllers:");
1031 for (j = 0, cit = (*it)->controllers; cit && *cit; cit++, j++)
1032 TRACE(" %d: %s", j, *cit);
1033 }
1034 }
1035
1036 static void lxc_cgfsng_print_basecg_debuginfo(char *basecginfo, char **klist,
1037 char **nlist)
1038 {
1039 int k;
1040 char **it;
1041
1042 TRACE("basecginfo is:");
1043 TRACE("%s", basecginfo);
1044
1045 for (k = 0, it = klist; it && *it; it++, k++)
1046 TRACE("kernel subsystem %d: %s", k, *it);
1047
1048 for (k = 0, it = nlist; it && *it; it++, k++)
1049 TRACE("named subsystem %d: %s", k, *it);
1050 }
1051
1052 static int cgroup_rmdir(struct hierarchy **hierarchies,
1053 const char *container_cgroup)
1054 {
1055 int i;
1056
1057 if (!container_cgroup || !hierarchies)
1058 return 0;
1059
1060 for (i = 0; hierarchies[i]; i++) {
1061 int ret;
1062 struct hierarchy *h = hierarchies[i];
1063
1064 if (!h->container_full_path)
1065 continue;
1066
1067 ret = recursive_destroy(h->container_full_path);
1068 if (ret < 0)
1069 WARN("Failed to destroy \"%s\"", h->container_full_path);
1070
1071 free(h->container_full_path);
1072 h->container_full_path = NULL;
1073 }
1074
1075 return 0;
1076 }
1077
1078 struct generic_userns_exec_data {
1079 struct hierarchy **hierarchies;
1080 const char *container_cgroup;
1081 struct lxc_conf *conf;
1082 uid_t origuid; /* target uid in parent namespace */
1083 char *path;
1084 };
1085
1086 static int cgroup_rmdir_wrapper(void *data)
1087 {
1088 int ret;
1089 struct generic_userns_exec_data *arg = data;
1090 uid_t nsuid = (arg->conf->root_nsuid_map != NULL) ? 0 : arg->conf->init_uid;
1091 gid_t nsgid = (arg->conf->root_nsgid_map != NULL) ? 0 : arg->conf->init_gid;
1092
1093 ret = setresgid(nsgid, nsgid, nsgid);
1094 if (ret < 0) {
1095 SYSERROR("Failed to setresgid(%d, %d, %d)", (int)nsgid,
1096 (int)nsgid, (int)nsgid);
1097 return -1;
1098 }
1099
1100 ret = setresuid(nsuid, nsuid, nsuid);
1101 if (ret < 0) {
1102 SYSERROR("Failed to setresuid(%d, %d, %d)", (int)nsuid,
1103 (int)nsuid, (int)nsuid);
1104 return -1;
1105 }
1106
1107 ret = setgroups(0, NULL);
1108 if (ret < 0 && errno != EPERM) {
1109 SYSERROR("Failed to setgroups(0, NULL)");
1110 return -1;
1111 }
1112
1113 return cgroup_rmdir(arg->hierarchies, arg->container_cgroup);
1114 }
1115
1116 __cgfsng_ops static void cgfsng_payload_destroy(struct cgroup_ops *ops,
1117 struct lxc_handler *handler)
1118 {
1119 int ret;
1120 struct generic_userns_exec_data wrap;
1121
1122 wrap.origuid = 0;
1123 wrap.container_cgroup = ops->container_cgroup;
1124 wrap.hierarchies = ops->hierarchies;
1125 wrap.conf = handler->conf;
1126
1127 if (handler->conf && !lxc_list_empty(&handler->conf->id_map))
1128 ret = userns_exec_1(handler->conf, cgroup_rmdir_wrapper, &wrap,
1129 "cgroup_rmdir_wrapper");
1130 else
1131 ret = cgroup_rmdir(ops->hierarchies, ops->container_cgroup);
1132 if (ret < 0) {
1133 WARN("Failed to destroy cgroups");
1134 return;
1135 }
1136 }
1137
1138 __cgfsng_ops static void cgfsng_monitor_destroy(struct cgroup_ops *ops,
1139 struct lxc_handler *handler)
1140 {
1141 int len;
1142 char *pivot_path;
1143 struct lxc_conf *conf = handler->conf;
1144 char pidstr[INTTYPE_TO_STRLEN(pid_t)];
1145
1146 if (!ops->hierarchies)
1147 return;
1148
1149 len = snprintf(pidstr, sizeof(pidstr), "%d", handler->monitor_pid);
1150 if (len < 0 || (size_t)len >= sizeof(pidstr))
1151 return;
1152
1153 for (int i = 0; ops->hierarchies[i]; i++) {
1154 int ret;
1155 struct hierarchy *h = ops->hierarchies[i];
1156
1157 if (!h->monitor_full_path)
1158 continue;
1159
1160 if (conf && conf->cgroup_meta.dir)
1161 pivot_path = must_make_path(h->mountpoint,
1162 h->container_base_path,
1163 conf->cgroup_meta.dir,
1164 PIVOT_CGROUP,
1165 "cgroup.procs", NULL);
1166 else
1167 pivot_path = must_make_path(h->mountpoint,
1168 h->container_base_path,
1169 PIVOT_CGROUP,
1170 "cgroup.procs", NULL);
1171
1172 ret = mkdir_p(pivot_path, 0755);
1173 if (ret < 0 && errno != EEXIST)
1174 goto next;
1175
1176 /* Move ourselves into the pivot cgroup to delete our own
1177 * cgroup.
1178 */
1179 ret = lxc_write_to_file(pivot_path, pidstr, len, false, 0666);
1180 if (ret != 0)
1181 goto next;
1182
1183 ret = recursive_destroy(h->monitor_full_path);
1184 if (ret < 0)
1185 WARN("Failed to destroy \"%s\"", h->monitor_full_path);
1186
1187 next:
1188 free(pivot_path);
1189 }
1190 }
1191
1192 static bool cg_unified_create_cgroup(struct hierarchy *h, char *cgname)
1193 {
1194 size_t i, parts_len;
1195 char **it;
1196 size_t full_len = 0;
1197 char *add_controllers = NULL, *cgroup = NULL;
1198 char **parts = NULL;
1199 bool bret = false;
1200
1201 if (h->version != CGROUP2_SUPER_MAGIC)
1202 return true;
1203
1204 if (!h->controllers)
1205 return true;
1206
1207 /* For now we simply enable all controllers that we have detected by
1208 * creating a string like "+memory +pids +cpu +io".
1209 * TODO: In the near future we might want to support "-<controller>"
1210 * etc. but whether supporting semantics like this make sense will need
1211 * some thinking.
1212 */
1213 for (it = h->controllers; it && *it; it++) {
1214 full_len += strlen(*it) + 2;
1215 add_controllers = must_realloc(add_controllers, full_len + 1);
1216
1217 if (h->controllers[0] == *it)
1218 add_controllers[0] = '\0';
1219
1220 (void)strlcat(add_controllers, "+", full_len + 1);
1221 (void)strlcat(add_controllers, *it, full_len + 1);
1222
1223 if ((it + 1) && *(it + 1))
1224 (void)strlcat(add_controllers, " ", full_len + 1);
1225 }
1226
1227 parts = lxc_string_split(cgname, '/');
1228 if (!parts)
1229 goto on_error;
1230
1231 parts_len = lxc_array_len((void **)parts);
1232 if (parts_len > 0)
1233 parts_len--;
1234
1235 cgroup = must_make_path(h->mountpoint, h->container_base_path, NULL);
1236 for (i = 0; i < parts_len; i++) {
1237 int ret;
1238 char *target;
1239
1240 cgroup = must_append_path(cgroup, parts[i], NULL);
1241 target = must_make_path(cgroup, "cgroup.subtree_control", NULL);
1242 ret = lxc_write_to_file(target, add_controllers, full_len, false, 0666);
1243 free(target);
1244 if (ret < 0) {
1245 SYSERROR("Could not enable \"%s\" controllers in the "
1246 "unified cgroup \"%s\"", add_controllers, cgroup);
1247 goto on_error;
1248 }
1249 }
1250
1251 bret = true;
1252
1253 on_error:
1254 lxc_free_array((void **)parts, free);
1255 free(add_controllers);
1256 free(cgroup);
1257 return bret;
1258 }
1259
1260 static bool monitor_create_path_for_hierarchy(struct hierarchy *h, char *cgname)
1261 {
1262 int ret;
1263
1264 h->monitor_full_path = must_make_path(h->mountpoint, h->container_base_path, cgname, NULL);
1265 if (dir_exists(h->monitor_full_path)) {
1266 ERROR("The cgroup \"%s\" already existed", h->monitor_full_path);
1267 return false;
1268 }
1269
1270 if (!cg_legacy_handle_cpuset_hierarchy(h, cgname)) {
1271 ERROR("Failed to handle legacy cpuset controller");
1272 return false;
1273 }
1274
1275 ret = mkdir_p(h->monitor_full_path, 0755);
1276 if (ret < 0) {
1277 ERROR("Failed to create cgroup \"%s\"", h->monitor_full_path);
1278 return false;
1279 }
1280
1281 return cg_unified_create_cgroup(h, cgname);
1282 }
1283
1284 static bool container_create_path_for_hierarchy(struct hierarchy *h, char *cgname)
1285 {
1286 int ret;
1287
1288 h->container_full_path = must_make_path(h->mountpoint, h->container_base_path, cgname, NULL);
1289 if (dir_exists(h->container_full_path)) {
1290 ERROR("The cgroup \"%s\" already existed", h->container_full_path);
1291 return false;
1292 }
1293
1294 if (!cg_legacy_handle_cpuset_hierarchy(h, cgname)) {
1295 ERROR("Failed to handle legacy cpuset controller");
1296 return false;
1297 }
1298
1299 ret = mkdir_p(h->container_full_path, 0755);
1300 if (ret < 0) {
1301 ERROR("Failed to create cgroup \"%s\"", h->container_full_path);
1302 return false;
1303 }
1304
1305 return cg_unified_create_cgroup(h, cgname);
1306 }
1307
1308 static void remove_path_for_hierarchy(struct hierarchy *h, char *cgname, bool monitor)
1309 {
1310 int ret;
1311 char *full_path;
1312
1313 if (monitor)
1314 full_path = h->monitor_full_path;
1315 else
1316 full_path = h->container_full_path;
1317
1318 ret = rmdir(full_path);
1319 if (ret < 0)
1320 SYSERROR("Failed to rmdir(\"%s\") from failed creation attempt", full_path);
1321
1322 free(full_path);
1323
1324 if (monitor)
1325 h->monitor_full_path = NULL;
1326 else
1327 h->container_full_path = NULL;
1328 }
1329
1330 __cgfsng_ops static inline bool cgfsng_monitor_create(struct cgroup_ops *ops,
1331 struct lxc_handler *handler)
1332 {
1333 char *monitor_cgroup, *offset, *tmp;
1334 int i, idx = 0;
1335 size_t len;
1336 bool bret = false;
1337 struct lxc_conf *conf = handler->conf;
1338
1339 if (!conf)
1340 return bret;
1341
1342 if (conf->cgroup_meta.dir)
1343 tmp = lxc_string_join("/",
1344 (const char *[]){conf->cgroup_meta.dir,
1345 ops->monitor_pattern,
1346 handler->name, NULL},
1347 false);
1348 else
1349 tmp = must_make_path(ops->monitor_pattern, handler->name, NULL);
1350 if (!tmp)
1351 return bret;
1352
1353 len = strlen(tmp) + 5; /* leave room for -NNN\0 */
1354 monitor_cgroup = must_alloc(len);
1355 (void)strlcpy(monitor_cgroup, tmp, len);
1356 free(tmp);
1357 offset = monitor_cgroup + len - 5;
1358
1359 do {
1360 if (idx) {
1361 int ret = snprintf(offset, 5, "-%d", idx);
1362 if (ret < 0 || (size_t)ret >= 5)
1363 goto on_error;
1364 }
1365
1366 for (i = 0; ops->hierarchies[i]; i++) {
1367 if (!monitor_create_path_for_hierarchy(ops->hierarchies[i], monitor_cgroup)) {
1368 ERROR("Failed to create cgroup \"%s\"", ops->hierarchies[i]->monitor_full_path);
1369 free(ops->hierarchies[i]->container_full_path);
1370 ops->hierarchies[i]->container_full_path = NULL;
1371
1372 for (int j = 0; j < i; j++)
1373 remove_path_for_hierarchy(ops->hierarchies[j], monitor_cgroup, true);
1374
1375 idx++;
1376 break;
1377 }
1378 }
1379 } while (ops->hierarchies[i] && idx > 0 && idx < 1000);
1380
1381 if (idx < 1000) {
1382 bret = true;
1383 INFO("The monitor process uses \"%s\" as cgroup", monitor_cgroup);
1384 }
1385
1386 on_error:
1387 free(monitor_cgroup);
1388
1389 return bret;
1390 }
1391
1392 /* Try to create the same cgroup in all hierarchies. Start with cgroup_pattern;
1393 * next cgroup_pattern-1, -2, ..., -999.
1394 */
1395 __cgfsng_ops static inline bool cgfsng_payload_create(struct cgroup_ops *ops,
1396 struct lxc_handler *handler)
1397 {
1398 int i;
1399 size_t len;
1400 char *container_cgroup, *offset, *tmp;
1401 int idx = 0;
1402 struct lxc_conf *conf = handler->conf;
1403
1404 if (ops->container_cgroup) {
1405 WARN("cgfsng_create called a second time: %s", ops->container_cgroup);
1406 return false;
1407 }
1408
1409 if (!conf)
1410 return false;
1411
1412 if (conf->cgroup_meta.dir)
1413 tmp = lxc_string_join("/", (const char *[]){conf->cgroup_meta.dir, handler->name, NULL}, false);
1414 else
1415 tmp = lxc_string_replace("%n", handler->name, ops->cgroup_pattern);
1416 if (!tmp) {
1417 ERROR("Failed expanding cgroup name pattern");
1418 return false;
1419 }
1420
1421 len = strlen(tmp) + 5; /* leave room for -NNN\0 */
1422 container_cgroup = must_alloc(len);
1423 (void)strlcpy(container_cgroup, tmp, len);
1424 free(tmp);
1425 offset = container_cgroup + len - 5;
1426
1427 again:
1428 if (idx == 1000) {
1429 ERROR("Too many conflicting cgroup names");
1430 goto out_free;
1431 }
1432
1433 if (idx) {
1434 int ret;
1435
1436 ret = snprintf(offset, 5, "-%d", idx);
1437 if (ret < 0 || (size_t)ret >= 5) {
1438 FILE *f = fopen("/dev/null", "w");
1439 if (f) {
1440 fprintf(f, "Workaround for GCC7 bug: "
1441 "https://gcc.gnu.org/bugzilla/"
1442 "show_bug.cgi?id=78969");
1443 fclose(f);
1444 }
1445 }
1446 }
1447
1448 for (i = 0; ops->hierarchies[i]; i++) {
1449 if (!container_create_path_for_hierarchy(ops->hierarchies[i], container_cgroup)) {
1450 ERROR("Failed to create cgroup \"%s\"", ops->hierarchies[i]->container_full_path);
1451 free(ops->hierarchies[i]->container_full_path);
1452 ops->hierarchies[i]->container_full_path = NULL;
1453 for (int j = 0; j < i; j++)
1454 remove_path_for_hierarchy(ops->hierarchies[j], container_cgroup, false);
1455 idx++;
1456 goto again;
1457 }
1458 }
1459
1460 ops->container_cgroup = container_cgroup;
1461 INFO("The container uses \"%s\" as cgroup", container_cgroup);
1462
1463 return true;
1464
1465 out_free:
1466 free(container_cgroup);
1467
1468 return false;
1469 }
1470
1471 __cgfsng_ops static bool __do_cgroup_enter(struct cgroup_ops *ops, pid_t pid,
1472 bool monitor)
1473 {
1474 int len;
1475 char pidstr[INTTYPE_TO_STRLEN(pid_t)];
1476
1477 len = snprintf(pidstr, sizeof(pidstr), "%d", pid);
1478 if (len < 0 || (size_t)len >= sizeof(pidstr))
1479 return false;
1480
1481 for (int i = 0; ops->hierarchies[i]; i++) {
1482 int ret;
1483 char *path;
1484
1485 if (monitor)
1486 path = must_make_path(ops->hierarchies[i]->monitor_full_path,
1487 "cgroup.procs", NULL);
1488 else
1489 path = must_make_path(ops->hierarchies[i]->container_full_path,
1490 "cgroup.procs", NULL);
1491 ret = lxc_write_to_file(path, pidstr, len, false, 0666);
1492 if (ret != 0) {
1493 SYSERROR("Failed to enter cgroup \"%s\"", path);
1494 free(path);
1495 return false;
1496 }
1497 free(path);
1498 }
1499
1500 return true;
1501 }
1502
1503 __cgfsng_ops static bool cgfsng_monitor_enter(struct cgroup_ops *ops, pid_t pid)
1504 {
1505 return __do_cgroup_enter(ops, pid, true);
1506 }
1507
1508 static bool cgfsng_payload_enter(struct cgroup_ops *ops, pid_t pid)
1509 {
1510 return __do_cgroup_enter(ops, pid, false);
1511 }
1512
1513 static int chowmod(char *path, uid_t chown_uid, gid_t chown_gid,
1514 mode_t chmod_mode)
1515 {
1516 int ret;
1517
1518 ret = chown(path, chown_uid, chown_gid);
1519 if (ret < 0) {
1520 SYSWARN("Failed to chown(%s, %d, %d)", path, (int)chown_uid, (int)chown_gid);
1521 return -1;
1522 }
1523
1524 ret = chmod(path, chmod_mode);
1525 if (ret < 0) {
1526 SYSWARN("Failed to chmod(%s, %d)", path, (int)chmod_mode);
1527 return -1;
1528 }
1529
1530 return 0;
1531 }
1532
1533 /* chgrp the container cgroups to container group. We leave
1534 * the container owner as cgroup owner. So we must make the
1535 * directories 775 so that the container can create sub-cgroups.
1536 *
1537 * Also chown the tasks and cgroup.procs files. Those may not
1538 * exist depending on kernel version.
1539 */
1540 static int chown_cgroup_wrapper(void *data)
1541 {
1542 int i, ret;
1543 uid_t destuid;
1544 struct generic_userns_exec_data *arg = data;
1545 uid_t nsuid = (arg->conf->root_nsuid_map != NULL) ? 0 : arg->conf->init_uid;
1546 gid_t nsgid = (arg->conf->root_nsgid_map != NULL) ? 0 : arg->conf->init_gid;
1547
1548 ret = setresgid(nsgid, nsgid, nsgid);
1549 if (ret < 0) {
1550 SYSERROR("Failed to setresgid(%d, %d, %d)",
1551 (int)nsgid, (int)nsgid, (int)nsgid);
1552 return -1;
1553 }
1554
1555 ret = setresuid(nsuid, nsuid, nsuid);
1556 if (ret < 0) {
1557 SYSERROR("Failed to setresuid(%d, %d, %d)",
1558 (int)nsuid, (int)nsuid, (int)nsuid);
1559 return -1;
1560 }
1561
1562 ret = setgroups(0, NULL);
1563 if (ret < 0 && errno != EPERM) {
1564 SYSERROR("Failed to setgroups(0, NULL)");
1565 return -1;
1566 }
1567
1568 destuid = get_ns_uid(arg->origuid);
1569 if (destuid == LXC_INVALID_UID)
1570 destuid = 0;
1571
1572 for (i = 0; arg->hierarchies[i]; i++) {
1573 char *fullpath;
1574 char *path = arg->hierarchies[i]->container_full_path;
1575
1576 ret = chowmod(path, destuid, nsgid, 0775);
1577 if (ret < 0)
1578 return -1;
1579
1580 /* Failures to chown() these are inconvenient but not
1581 * detrimental We leave these owned by the container launcher,
1582 * so that container root can write to the files to attach. We
1583 * chmod() them 664 so that container systemd can write to the
1584 * files (which systemd in wily insists on doing).
1585 */
1586
1587 if (arg->hierarchies[i]->version == CGROUP_SUPER_MAGIC) {
1588 fullpath = must_make_path(path, "tasks", NULL);
1589 (void)chowmod(fullpath, destuid, nsgid, 0664);
1590 free(fullpath);
1591 }
1592
1593 fullpath = must_make_path(path, "cgroup.procs", NULL);
1594 (void)chowmod(fullpath, destuid, nsgid, 0664);
1595 free(fullpath);
1596
1597 if (arg->hierarchies[i]->version != CGROUP2_SUPER_MAGIC)
1598 continue;
1599
1600 fullpath = must_make_path(path, "cgroup.subtree_control", NULL);
1601 (void)chowmod(fullpath, destuid, nsgid, 0664);
1602 free(fullpath);
1603
1604 fullpath = must_make_path(path, "cgroup.threads", NULL);
1605 (void)chowmod(fullpath, destuid, nsgid, 0664);
1606 free(fullpath);
1607 }
1608
1609 return 0;
1610 }
1611
1612 __cgfsng_ops static bool cgfsng_chown(struct cgroup_ops *ops,
1613 struct lxc_conf *conf)
1614 {
1615 struct generic_userns_exec_data wrap;
1616
1617 if (lxc_list_empty(&conf->id_map))
1618 return true;
1619
1620 wrap.origuid = geteuid();
1621 wrap.path = NULL;
1622 wrap.hierarchies = ops->hierarchies;
1623 wrap.conf = conf;
1624
1625 if (userns_exec_1(conf, chown_cgroup_wrapper, &wrap,
1626 "chown_cgroup_wrapper") < 0) {
1627 ERROR("Error requesting cgroup chown in new user namespace");
1628 return false;
1629 }
1630
1631 return true;
1632 }
1633
1634 /* cgroup-full:* is done, no need to create subdirs */
1635 static bool cg_mount_needs_subdirs(int type)
1636 {
1637 if (type >= LXC_AUTO_CGROUP_FULL_RO)
1638 return false;
1639
1640 return true;
1641 }
1642
1643 /* After $rootfs/sys/fs/container/controller/the/cg/path has been created,
1644 * remount controller ro if needed and bindmount the cgroupfs onto
1645 * controll/the/cg/path.
1646 */
1647 static int cg_legacy_mount_controllers(int type, struct hierarchy *h,
1648 char *controllerpath, char *cgpath,
1649 const char *container_cgroup)
1650 {
1651 int ret, remount_flags;
1652 char *sourcepath;
1653 int flags = MS_BIND;
1654
1655 if (type == LXC_AUTO_CGROUP_RO || type == LXC_AUTO_CGROUP_MIXED) {
1656 ret = mount(controllerpath, controllerpath, "cgroup", MS_BIND, NULL);
1657 if (ret < 0) {
1658 SYSERROR("Failed to bind mount \"%s\" onto \"%s\"",
1659 controllerpath, controllerpath);
1660 return -1;
1661 }
1662
1663 remount_flags = add_required_remount_flags(controllerpath,
1664 controllerpath,
1665 flags | MS_REMOUNT);
1666 ret = mount(controllerpath, controllerpath, "cgroup",
1667 remount_flags | MS_REMOUNT | MS_BIND | MS_RDONLY,
1668 NULL);
1669 if (ret < 0) {
1670 SYSERROR("Failed to remount \"%s\" ro", controllerpath);
1671 return -1;
1672 }
1673
1674 INFO("Remounted %s read-only", controllerpath);
1675 }
1676
1677 sourcepath = must_make_path(h->mountpoint, h->container_base_path,
1678 container_cgroup, NULL);
1679 if (type == LXC_AUTO_CGROUP_RO)
1680 flags |= MS_RDONLY;
1681
1682 ret = mount(sourcepath, cgpath, "cgroup", flags, NULL);
1683 if (ret < 0) {
1684 SYSERROR("Failed to mount \"%s\" onto \"%s\"", h->controllers[0], cgpath);
1685 free(sourcepath);
1686 return -1;
1687 }
1688 INFO("Mounted \"%s\" onto \"%s\"", h->controllers[0], cgpath);
1689
1690 if (flags & MS_RDONLY) {
1691 remount_flags = add_required_remount_flags(sourcepath, cgpath,
1692 flags | MS_REMOUNT);
1693 ret = mount(sourcepath, cgpath, "cgroup", remount_flags, NULL);
1694 if (ret < 0) {
1695 SYSERROR("Failed to remount \"%s\" ro", cgpath);
1696 free(sourcepath);
1697 return -1;
1698 }
1699 INFO("Remounted %s read-only", cgpath);
1700 }
1701
1702 free(sourcepath);
1703 INFO("Completed second stage cgroup automounts for \"%s\"", cgpath);
1704 return 0;
1705 }
1706
1707 /* __cg_mount_direct
1708 *
1709 * Mount cgroup hierarchies directly without using bind-mounts. The main
1710 * uses-cases are mounting cgroup hierarchies in cgroup namespaces and mounting
1711 * cgroups for the LXC_AUTO_CGROUP_FULL option.
1712 */
1713 static int __cg_mount_direct(int type, struct hierarchy *h,
1714 const char *controllerpath)
1715 {
1716 int ret;
1717 char *controllers = NULL;
1718 char *fstype = "cgroup2";
1719 unsigned long flags = 0;
1720
1721 flags |= MS_NOSUID;
1722 flags |= MS_NOEXEC;
1723 flags |= MS_NODEV;
1724 flags |= MS_RELATIME;
1725
1726 if (type == LXC_AUTO_CGROUP_RO || type == LXC_AUTO_CGROUP_FULL_RO)
1727 flags |= MS_RDONLY;
1728
1729 if (h->version != CGROUP2_SUPER_MAGIC) {
1730 controllers = lxc_string_join(",", (const char **)h->controllers, false);
1731 if (!controllers)
1732 return -ENOMEM;
1733 fstype = "cgroup";
1734 }
1735
1736 ret = mount("cgroup", controllerpath, fstype, flags, controllers);
1737 free(controllers);
1738 if (ret < 0) {
1739 SYSERROR("Failed to mount \"%s\" with cgroup filesystem type %s", controllerpath, fstype);
1740 return -1;
1741 }
1742
1743 DEBUG("Mounted \"%s\" with cgroup filesystem type %s", controllerpath, fstype);
1744 return 0;
1745 }
1746
1747 static inline int cg_mount_in_cgroup_namespace(int type, struct hierarchy *h,
1748 const char *controllerpath)
1749 {
1750 return __cg_mount_direct(type, h, controllerpath);
1751 }
1752
1753 static inline int cg_mount_cgroup_full(int type, struct hierarchy *h,
1754 const char *controllerpath)
1755 {
1756 if (type < LXC_AUTO_CGROUP_FULL_RO || type > LXC_AUTO_CGROUP_FULL_MIXED)
1757 return 0;
1758
1759 return __cg_mount_direct(type, h, controllerpath);
1760 }
1761
1762 __cgfsng_ops static bool cgfsng_mount(struct cgroup_ops *ops,
1763 struct lxc_handler *handler,
1764 const char *root, int type)
1765 {
1766 int i, ret;
1767 char *tmpfspath = NULL;
1768 bool has_cgns = false, retval = false, wants_force_mount = false;
1769
1770 if ((type & LXC_AUTO_CGROUP_MASK) == 0)
1771 return true;
1772
1773 if (type & LXC_AUTO_CGROUP_FORCE) {
1774 type &= ~LXC_AUTO_CGROUP_FORCE;
1775 wants_force_mount = true;
1776 }
1777
1778 if (!wants_force_mount){
1779 if (!lxc_list_empty(&handler->conf->keepcaps))
1780 wants_force_mount = !in_caplist(CAP_SYS_ADMIN, &handler->conf->keepcaps);
1781 else
1782 wants_force_mount = in_caplist(CAP_SYS_ADMIN, &handler->conf->caps);
1783 }
1784
1785 has_cgns = cgns_supported();
1786 if (has_cgns && !wants_force_mount)
1787 return true;
1788
1789 if (type == LXC_AUTO_CGROUP_NOSPEC)
1790 type = LXC_AUTO_CGROUP_MIXED;
1791 else if (type == LXC_AUTO_CGROUP_FULL_NOSPEC)
1792 type = LXC_AUTO_CGROUP_FULL_MIXED;
1793
1794 /* Mount tmpfs */
1795 tmpfspath = must_make_path(root, "/sys/fs/cgroup", NULL);
1796 ret = safe_mount(NULL, tmpfspath, "tmpfs",
1797 MS_NOSUID | MS_NODEV | MS_NOEXEC | MS_RELATIME,
1798 "size=10240k,mode=755", root);
1799 if (ret < 0)
1800 goto on_error;
1801
1802 for (i = 0; ops->hierarchies[i]; i++) {
1803 char *controllerpath, *path2;
1804 struct hierarchy *h = ops->hierarchies[i];
1805 char *controller = strrchr(h->mountpoint, '/');
1806
1807 if (!controller)
1808 continue;
1809 controller++;
1810
1811 controllerpath = must_make_path(tmpfspath, controller, NULL);
1812 if (dir_exists(controllerpath)) {
1813 free(controllerpath);
1814 continue;
1815 }
1816
1817 ret = mkdir(controllerpath, 0755);
1818 if (ret < 0) {
1819 SYSERROR("Error creating cgroup path: %s", controllerpath);
1820 free(controllerpath);
1821 goto on_error;
1822 }
1823
1824 if (has_cgns && wants_force_mount) {
1825 /* If cgroup namespaces are supported but the container
1826 * will not have CAP_SYS_ADMIN after it has started we
1827 * need to mount the cgroups manually.
1828 */
1829 ret = cg_mount_in_cgroup_namespace(type, h, controllerpath);
1830 free(controllerpath);
1831 if (ret < 0)
1832 goto on_error;
1833
1834 continue;
1835 }
1836
1837 ret = cg_mount_cgroup_full(type, h, controllerpath);
1838 if (ret < 0) {
1839 free(controllerpath);
1840 goto on_error;
1841 }
1842
1843 if (!cg_mount_needs_subdirs(type)) {
1844 free(controllerpath);
1845 continue;
1846 }
1847
1848 path2 = must_make_path(controllerpath, h->container_base_path,
1849 ops->container_cgroup, NULL);
1850 ret = mkdir_p(path2, 0755);
1851 if (ret < 0) {
1852 free(controllerpath);
1853 free(path2);
1854 goto on_error;
1855 }
1856
1857 ret = cg_legacy_mount_controllers(type, h, controllerpath,
1858 path2, ops->container_cgroup);
1859 free(controllerpath);
1860 free(path2);
1861 if (ret < 0)
1862 goto on_error;
1863 }
1864 retval = true;
1865
1866 on_error:
1867 free(tmpfspath);
1868 return retval;
1869 }
1870
1871 static int recursive_count_nrtasks(char *dirname)
1872 {
1873 struct dirent *direntp;
1874 DIR *dir;
1875 int count = 0, ret;
1876 char *path;
1877
1878 dir = opendir(dirname);
1879 if (!dir)
1880 return 0;
1881
1882 while ((direntp = readdir(dir))) {
1883 struct stat mystat;
1884
1885 if (!strcmp(direntp->d_name, ".") ||
1886 !strcmp(direntp->d_name, ".."))
1887 continue;
1888
1889 path = must_make_path(dirname, direntp->d_name, NULL);
1890
1891 if (lstat(path, &mystat))
1892 goto next;
1893
1894 if (!S_ISDIR(mystat.st_mode))
1895 goto next;
1896
1897 count += recursive_count_nrtasks(path);
1898 next:
1899 free(path);
1900 }
1901
1902 path = must_make_path(dirname, "cgroup.procs", NULL);
1903 ret = lxc_count_file_lines(path);
1904 if (ret != -1)
1905 count += ret;
1906 free(path);
1907
1908 (void)closedir(dir);
1909
1910 return count;
1911 }
1912
1913 __cgfsng_ops static int cgfsng_nrtasks(struct cgroup_ops *ops)
1914 {
1915 int count;
1916 char *path;
1917
1918 if (!ops->container_cgroup || !ops->hierarchies)
1919 return -1;
1920
1921 path = must_make_path(ops->hierarchies[0]->container_full_path, NULL);
1922 count = recursive_count_nrtasks(path);
1923 free(path);
1924 return count;
1925 }
1926
1927 /* Only root needs to escape to the cgroup of its init. */
1928 __cgfsng_ops static bool cgfsng_escape(const struct cgroup_ops *ops,
1929 struct lxc_conf *conf)
1930 {
1931 int i;
1932
1933 if (conf->cgroup_meta.relative || geteuid())
1934 return true;
1935
1936 for (i = 0; ops->hierarchies[i]; i++) {
1937 int ret;
1938 char *fullpath;
1939
1940 fullpath = must_make_path(ops->hierarchies[i]->mountpoint,
1941 ops->hierarchies[i]->container_base_path,
1942 "cgroup.procs", NULL);
1943 ret = lxc_write_to_file(fullpath, "0", 2, false, 0666);
1944 if (ret != 0) {
1945 SYSERROR("Failed to escape to cgroup \"%s\"", fullpath);
1946 free(fullpath);
1947 return false;
1948 }
1949 free(fullpath);
1950 }
1951
1952 return true;
1953 }
1954
1955 __cgfsng_ops static int cgfsng_num_hierarchies(struct cgroup_ops *ops)
1956 {
1957 int i;
1958
1959 for (i = 0; ops->hierarchies[i]; i++)
1960 ;
1961
1962 return i;
1963 }
1964
1965 __cgfsng_ops static bool cgfsng_get_hierarchies(struct cgroup_ops *ops, int n, char ***out)
1966 {
1967 int i;
1968
1969 /* sanity check n */
1970 for (i = 0; i < n; i++)
1971 if (!ops->hierarchies[i])
1972 return false;
1973
1974 *out = ops->hierarchies[i]->controllers;
1975
1976 return true;
1977 }
1978
1979 #define THAWED "THAWED"
1980 #define THAWED_LEN (strlen(THAWED))
1981
1982 /* TODO: If the unified cgroup hierarchy grows a freezer controller this needs
1983 * to be adapted.
1984 */
1985 __cgfsng_ops static bool cgfsng_unfreeze(struct cgroup_ops *ops)
1986 {
1987 int ret;
1988 char *fullpath;
1989 struct hierarchy *h;
1990
1991 h = get_hierarchy(ops, "freezer");
1992 if (!h)
1993 return false;
1994
1995 fullpath = must_make_path(h->container_full_path, "freezer.state", NULL);
1996 ret = lxc_write_to_file(fullpath, THAWED, THAWED_LEN, false, 0666);
1997 free(fullpath);
1998 if (ret < 0)
1999 return false;
2000
2001 return true;
2002 }
2003
2004 __cgfsng_ops static const char *cgfsng_get_cgroup(struct cgroup_ops *ops,
2005 const char *controller)
2006 {
2007 struct hierarchy *h;
2008
2009 h = get_hierarchy(ops, controller);
2010 if (!h) {
2011 WARN("Failed to find hierarchy for controller \"%s\"",
2012 controller ? controller : "(null)");
2013 return NULL;
2014 }
2015
2016 return h->container_full_path ? h->container_full_path + strlen(h->mountpoint) : NULL;
2017 }
2018
2019 /* Given a cgroup path returned from lxc_cmd_get_cgroup_path, build a full path,
2020 * which must be freed by the caller.
2021 */
2022 static inline char *build_full_cgpath_from_monitorpath(struct hierarchy *h,
2023 const char *inpath,
2024 const char *filename)
2025 {
2026 return must_make_path(h->mountpoint, inpath, filename, NULL);
2027 }
2028
2029 /* Technically, we're always at a delegation boundary here (This is especially
2030 * true when cgroup namespaces are available.). The reasoning is that in order
2031 * for us to have been able to start a container in the first place the root
2032 * cgroup must have been a leaf node. Now, either the container's init system
2033 * has populated the cgroup and kept it as a leaf node or it has created
2034 * subtrees. In the former case we will simply attach to the leaf node we
2035 * created when we started the container in the latter case we create our own
2036 * cgroup for the attaching process.
2037 */
2038 static int __cg_unified_attach(const struct hierarchy *h, const char *name,
2039 const char *lxcpath, const char *pidstr,
2040 size_t pidstr_len, const char *controller)
2041 {
2042 int ret;
2043 size_t len;
2044 int fret = -1, idx = 0;
2045 char *base_path = NULL, *container_cgroup = NULL, *full_path = NULL;
2046
2047 container_cgroup = lxc_cmd_get_cgroup_path(name, lxcpath, controller);
2048 /* not running */
2049 if (!container_cgroup)
2050 return 0;
2051
2052 base_path = must_make_path(h->mountpoint, container_cgroup, NULL);
2053 full_path = must_make_path(base_path, "cgroup.procs", NULL);
2054 /* cgroup is populated */
2055 ret = lxc_write_to_file(full_path, pidstr, pidstr_len, false, 0666);
2056 if (ret < 0 && errno != EBUSY)
2057 goto on_error;
2058
2059 if (ret == 0)
2060 goto on_success;
2061
2062 free(full_path);
2063
2064 len = strlen(base_path) + STRLITERALLEN("/lxc-1000") +
2065 STRLITERALLEN("/cgroup-procs");
2066 full_path = must_alloc(len + 1);
2067 do {
2068 if (idx)
2069 ret = snprintf(full_path, len + 1, "%s/lxc-%d",
2070 base_path, idx);
2071 else
2072 ret = snprintf(full_path, len + 1, "%s/lxc", base_path);
2073 if (ret < 0 || (size_t)ret >= len + 1)
2074 goto on_error;
2075
2076 ret = mkdir_p(full_path, 0755);
2077 if (ret < 0 && errno != EEXIST)
2078 goto on_error;
2079
2080 (void)strlcat(full_path, "/cgroup.procs", len + 1);
2081 ret = lxc_write_to_file(full_path, pidstr, len, false, 0666);
2082 if (ret == 0)
2083 goto on_success;
2084
2085 /* this is a non-leaf node */
2086 if (errno != EBUSY)
2087 goto on_error;
2088
2089 } while (++idx > 0 && idx < 1000);
2090
2091 on_success:
2092 if (idx < 1000)
2093 fret = 0;
2094
2095 on_error:
2096 free(base_path);
2097 free(container_cgroup);
2098 free(full_path);
2099
2100 return fret;
2101 }
2102
2103 __cgfsng_ops static bool cgfsng_attach(struct cgroup_ops *ops, const char *name,
2104 const char *lxcpath, pid_t pid)
2105 {
2106 int i, len, ret;
2107 char pidstr[INTTYPE_TO_STRLEN(pid_t)];
2108
2109 len = snprintf(pidstr, sizeof(pidstr), "%d", pid);
2110 if (len < 0 || (size_t)len >= sizeof(pidstr))
2111 return false;
2112
2113 for (i = 0; ops->hierarchies[i]; i++) {
2114 char *path;
2115 char *fullpath = NULL;
2116 struct hierarchy *h = ops->hierarchies[i];
2117
2118 if (h->version == CGROUP2_SUPER_MAGIC) {
2119 ret = __cg_unified_attach(h, name, lxcpath, pidstr, len,
2120 h->controllers[0]);
2121 if (ret < 0)
2122 return false;
2123
2124 continue;
2125 }
2126
2127 path = lxc_cmd_get_cgroup_path(name, lxcpath, h->controllers[0]);
2128 /* not running */
2129 if (!path)
2130 continue;
2131
2132 fullpath = build_full_cgpath_from_monitorpath(h, path, "cgroup.procs");
2133 free(path);
2134 ret = lxc_write_to_file(fullpath, pidstr, len, false, 0666);
2135 if (ret < 0) {
2136 SYSERROR("Failed to attach %d to %s", (int)pid, fullpath);
2137 free(fullpath);
2138 return false;
2139 }
2140 free(fullpath);
2141 }
2142
2143 return true;
2144 }
2145
2146 /* Called externally (i.e. from 'lxc-cgroup') to query cgroup limits. Here we
2147 * don't have a cgroup_data set up, so we ask the running container through the
2148 * commands API for the cgroup path.
2149 */
2150 __cgfsng_ops static int cgfsng_get(struct cgroup_ops *ops, const char *filename,
2151 char *value, size_t len, const char *name,
2152 const char *lxcpath)
2153 {
2154 int ret = -1;
2155 size_t controller_len;
2156 char *controller, *p, *path;
2157 struct hierarchy *h;
2158
2159 controller_len = strlen(filename);
2160 controller = alloca(controller_len + 1);
2161 (void)strlcpy(controller, filename, controller_len + 1);
2162
2163 p = strchr(controller, '.');
2164 if (p)
2165 *p = '\0';
2166
2167 path = lxc_cmd_get_cgroup_path(name, lxcpath, controller);
2168 /* not running */
2169 if (!path)
2170 return -1;
2171
2172 h = get_hierarchy(ops, controller);
2173 if (h) {
2174 char *fullpath;
2175
2176 fullpath = build_full_cgpath_from_monitorpath(h, path, filename);
2177 ret = lxc_read_from_file(fullpath, value, len);
2178 free(fullpath);
2179 }
2180 free(path);
2181
2182 return ret;
2183 }
2184
2185 /* Called externally (i.e. from 'lxc-cgroup') to set new cgroup limits. Here we
2186 * don't have a cgroup_data set up, so we ask the running container through the
2187 * commands API for the cgroup path.
2188 */
2189 __cgfsng_ops static int cgfsng_set(struct cgroup_ops *ops,
2190 const char *filename, const char *value,
2191 const char *name, const char *lxcpath)
2192 {
2193 int ret = -1;
2194 size_t controller_len;
2195 char *controller, *p, *path;
2196 struct hierarchy *h;
2197
2198 controller_len = strlen(filename);
2199 controller = alloca(controller_len + 1);
2200 (void)strlcpy(controller, filename, controller_len + 1);
2201
2202 p = strchr(controller, '.');
2203 if (p)
2204 *p = '\0';
2205
2206 path = lxc_cmd_get_cgroup_path(name, lxcpath, controller);
2207 /* not running */
2208 if (!path)
2209 return -1;
2210
2211 h = get_hierarchy(ops, controller);
2212 if (h) {
2213 char *fullpath;
2214
2215 fullpath = build_full_cgpath_from_monitorpath(h, path, filename);
2216 ret = lxc_write_to_file(fullpath, value, strlen(value), false, 0666);
2217 free(fullpath);
2218 }
2219 free(path);
2220
2221 return ret;
2222 }
2223
2224 /* take devices cgroup line
2225 * /dev/foo rwx
2226 * and convert it to a valid
2227 * type major:minor mode
2228 * line. Return <0 on error. Dest is a preallocated buffer long enough to hold
2229 * the output.
2230 */
2231 static int convert_devpath(const char *invalue, char *dest)
2232 {
2233 int n_parts;
2234 char *p, *path, type;
2235 unsigned long minor, major;
2236 struct stat sb;
2237 int ret = -EINVAL;
2238 char *mode = NULL;
2239
2240 path = must_copy_string(invalue);
2241
2242 /* Read path followed by mode. Ignore any trailing text.
2243 * A ' # comment' would be legal. Technically other text is not
2244 * legal, we could check for that if we cared to.
2245 */
2246 for (n_parts = 1, p = path; *p && n_parts < 3; p++) {
2247 if (*p != ' ')
2248 continue;
2249 *p = '\0';
2250
2251 if (n_parts != 1)
2252 break;
2253 p++;
2254 n_parts++;
2255
2256 while (*p == ' ')
2257 p++;
2258
2259 mode = p;
2260
2261 if (*p == '\0')
2262 goto out;
2263 }
2264
2265 if (n_parts == 1)
2266 goto out;
2267
2268 ret = stat(path, &sb);
2269 if (ret < 0)
2270 goto out;
2271
2272 mode_t m = sb.st_mode & S_IFMT;
2273 switch (m) {
2274 case S_IFBLK:
2275 type = 'b';
2276 break;
2277 case S_IFCHR:
2278 type = 'c';
2279 break;
2280 default:
2281 ERROR("Unsupported device type %i for \"%s\"", m, path);
2282 ret = -EINVAL;
2283 goto out;
2284 }
2285
2286 major = MAJOR(sb.st_rdev);
2287 minor = MINOR(sb.st_rdev);
2288 ret = snprintf(dest, 50, "%c %lu:%lu %s", type, major, minor, mode);
2289 if (ret < 0 || ret >= 50) {
2290 ERROR("Error on configuration value \"%c %lu:%lu %s\" (max 50 "
2291 "chars)", type, major, minor, mode);
2292 ret = -ENAMETOOLONG;
2293 goto out;
2294 }
2295 ret = 0;
2296
2297 out:
2298 free(path);
2299 return ret;
2300 }
2301
2302 /* Called from setup_limits - here we have the container's cgroup_data because
2303 * we created the cgroups.
2304 */
2305 static int cg_legacy_set_data(struct cgroup_ops *ops, const char *filename,
2306 const char *value)
2307 {
2308 size_t len;
2309 char *fullpath, *p;
2310 /* "b|c <2^64-1>:<2^64-1> r|w|m" = 47 chars max */
2311 char converted_value[50];
2312 struct hierarchy *h;
2313 int ret = 0;
2314 char *controller = NULL;
2315
2316 len = strlen(filename);
2317 controller = alloca(len + 1);
2318 (void)strlcpy(controller, filename, len + 1);
2319
2320 p = strchr(controller, '.');
2321 if (p)
2322 *p = '\0';
2323
2324 if (strcmp("devices.allow", filename) == 0 && value[0] == '/') {
2325 ret = convert_devpath(value, converted_value);
2326 if (ret < 0)
2327 return ret;
2328 value = converted_value;
2329 }
2330
2331 h = get_hierarchy(ops, controller);
2332 if (!h) {
2333 ERROR("Failed to setup limits for the \"%s\" controller. "
2334 "The controller seems to be unused by \"cgfsng\" cgroup "
2335 "driver or not enabled on the cgroup hierarchy",
2336 controller);
2337 errno = ENOENT;
2338 return -ENOENT;
2339 }
2340
2341 fullpath = must_make_path(h->container_full_path, filename, NULL);
2342 ret = lxc_write_to_file(fullpath, value, strlen(value), false, 0666);
2343 free(fullpath);
2344 return ret;
2345 }
2346
2347 static bool __cg_legacy_setup_limits(struct cgroup_ops *ops,
2348 struct lxc_list *cgroup_settings,
2349 bool do_devices)
2350 {
2351 struct lxc_list *iterator, *next, *sorted_cgroup_settings;
2352 struct lxc_cgroup *cg;
2353 bool ret = false;
2354
2355 if (lxc_list_empty(cgroup_settings))
2356 return true;
2357
2358 sorted_cgroup_settings = sort_cgroup_settings(cgroup_settings);
2359 if (!sorted_cgroup_settings)
2360 return false;
2361
2362 lxc_list_for_each(iterator, sorted_cgroup_settings) {
2363 cg = iterator->elem;
2364
2365 if (do_devices == !strncmp("devices", cg->subsystem, 7)) {
2366 if (cg_legacy_set_data(ops, cg->subsystem, cg->value)) {
2367 if (do_devices && (errno == EACCES || errno == EPERM)) {
2368 WARN("Failed to set \"%s\" to \"%s\"",
2369 cg->subsystem, cg->value);
2370 continue;
2371 }
2372 WARN("Failed to set \"%s\" to \"%s\"",
2373 cg->subsystem, cg->value);
2374 goto out;
2375 }
2376 DEBUG("Set controller \"%s\" set to \"%s\"",
2377 cg->subsystem, cg->value);
2378 }
2379 }
2380
2381 ret = true;
2382 INFO("Limits for the legacy cgroup hierarchies have been setup");
2383 out:
2384 lxc_list_for_each_safe(iterator, sorted_cgroup_settings, next) {
2385 lxc_list_del(iterator);
2386 free(iterator);
2387 }
2388 free(sorted_cgroup_settings);
2389 return ret;
2390 }
2391
2392 static bool __cg_unified_setup_limits(struct cgroup_ops *ops,
2393 struct lxc_list *cgroup_settings)
2394 {
2395 struct lxc_list *iterator;
2396 struct hierarchy *h = ops->unified;
2397
2398 if (lxc_list_empty(cgroup_settings))
2399 return true;
2400
2401 if (!h)
2402 return false;
2403
2404 lxc_list_for_each(iterator, cgroup_settings) {
2405 int ret;
2406 char *fullpath;
2407 struct lxc_cgroup *cg = iterator->elem;
2408
2409 fullpath = must_make_path(h->container_full_path, cg->subsystem, NULL);
2410 ret = lxc_write_to_file(fullpath, cg->value, strlen(cg->value), false, 0666);
2411 free(fullpath);
2412 if (ret < 0) {
2413 SYSERROR("Failed to set \"%s\" to \"%s\"",
2414 cg->subsystem, cg->value);
2415 return false;
2416 }
2417 TRACE("Set \"%s\" to \"%s\"", cg->subsystem, cg->value);
2418 }
2419
2420 INFO("Limits for the unified cgroup hierarchy have been setup");
2421 return true;
2422 }
2423
2424 __cgfsng_ops static bool cgfsng_setup_limits(struct cgroup_ops *ops,
2425 struct lxc_conf *conf,
2426 bool do_devices)
2427 {
2428 bool bret;
2429
2430 bret = __cg_legacy_setup_limits(ops, &conf->cgroup, do_devices);
2431 if (!bret)
2432 return false;
2433
2434 return __cg_unified_setup_limits(ops, &conf->cgroup2);
2435 }
2436
2437 static bool cgroup_use_wants_controllers(const struct cgroup_ops *ops,
2438 char **controllers)
2439 {
2440 char **cur_ctrl, **cur_use;
2441
2442 if (!ops->cgroup_use)
2443 return true;
2444
2445 for (cur_ctrl = controllers; cur_ctrl && *cur_ctrl; cur_ctrl++) {
2446 bool found = false;
2447
2448 for (cur_use = ops->cgroup_use; cur_use && *cur_use; cur_use++) {
2449 if (strcmp(*cur_use, *cur_ctrl) != 0)
2450 continue;
2451
2452 found = true;
2453 break;
2454 }
2455
2456 if (found)
2457 continue;
2458
2459 return false;
2460 }
2461
2462 return true;
2463 }
2464
2465 /* At startup, parse_hierarchies finds all the info we need about cgroup
2466 * mountpoints and current cgroups, and stores it in @d.
2467 */
2468 static bool cg_hybrid_init(struct cgroup_ops *ops, bool relative)
2469 {
2470 int ret;
2471 char *basecginfo;
2472 FILE *f;
2473 size_t len = 0;
2474 char *line = NULL;
2475 char **klist = NULL, **nlist = NULL;
2476
2477 /* Root spawned containers escape the current cgroup, so use init's
2478 * cgroups as our base in that case.
2479 */
2480 if (!relative && (geteuid() == 0))
2481 basecginfo = read_file("/proc/1/cgroup");
2482 else
2483 basecginfo = read_file("/proc/self/cgroup");
2484 if (!basecginfo)
2485 return false;
2486
2487 ret = get_existing_subsystems(&klist, &nlist);
2488 if (ret < 0) {
2489 ERROR("Failed to retrieve available legacy cgroup controllers");
2490 free(basecginfo);
2491 return false;
2492 }
2493
2494 f = fopen("/proc/self/mountinfo", "r");
2495 if (!f) {
2496 ERROR("Failed to open \"/proc/self/mountinfo\"");
2497 free(basecginfo);
2498 return false;
2499 }
2500
2501 lxc_cgfsng_print_basecg_debuginfo(basecginfo, klist, nlist);
2502
2503 while (getline(&line, &len, f) != -1) {
2504 int type;
2505 bool writeable;
2506 struct hierarchy *new;
2507 char *base_cgroup = NULL, *mountpoint = NULL;
2508 char **controller_list = NULL;
2509
2510 type = get_cgroup_version(line);
2511 if (type == 0)
2512 continue;
2513
2514 if (type == CGROUP2_SUPER_MAGIC && ops->unified)
2515 continue;
2516
2517 if (ops->cgroup_layout == CGROUP_LAYOUT_UNKNOWN) {
2518 if (type == CGROUP2_SUPER_MAGIC)
2519 ops->cgroup_layout = CGROUP_LAYOUT_UNIFIED;
2520 else if (type == CGROUP_SUPER_MAGIC)
2521 ops->cgroup_layout = CGROUP_LAYOUT_LEGACY;
2522 } else if (ops->cgroup_layout == CGROUP_LAYOUT_UNIFIED) {
2523 if (type == CGROUP_SUPER_MAGIC)
2524 ops->cgroup_layout = CGROUP_LAYOUT_HYBRID;
2525 } else if (ops->cgroup_layout == CGROUP_LAYOUT_LEGACY) {
2526 if (type == CGROUP2_SUPER_MAGIC)
2527 ops->cgroup_layout = CGROUP_LAYOUT_HYBRID;
2528 }
2529
2530 controller_list = cg_hybrid_get_controllers(klist, nlist, line, type);
2531 if (!controller_list && type == CGROUP_SUPER_MAGIC)
2532 continue;
2533
2534 if (type == CGROUP_SUPER_MAGIC)
2535 if (controller_list_is_dup(ops->hierarchies, controller_list))
2536 goto next;
2537
2538 mountpoint = cg_hybrid_get_mountpoint(line);
2539 if (!mountpoint) {
2540 ERROR("Failed parsing mountpoint from \"%s\"", line);
2541 goto next;
2542 }
2543
2544 if (type == CGROUP_SUPER_MAGIC)
2545 base_cgroup = cg_hybrid_get_current_cgroup(basecginfo, controller_list[0], CGROUP_SUPER_MAGIC);
2546 else
2547 base_cgroup = cg_hybrid_get_current_cgroup(basecginfo, NULL, CGROUP2_SUPER_MAGIC);
2548 if (!base_cgroup) {
2549 ERROR("Failed to find current cgroup");
2550 goto next;
2551 }
2552
2553 trim(base_cgroup);
2554 prune_init_scope(base_cgroup);
2555 if (type == CGROUP2_SUPER_MAGIC)
2556 writeable = test_writeable_v2(mountpoint, base_cgroup);
2557 else
2558 writeable = test_writeable_v1(mountpoint, base_cgroup);
2559 if (!writeable)
2560 goto next;
2561
2562 if (type == CGROUP2_SUPER_MAGIC) {
2563 char *cgv2_ctrl_path;
2564
2565 cgv2_ctrl_path = must_make_path(mountpoint, base_cgroup,
2566 "cgroup.controllers",
2567 NULL);
2568
2569 controller_list = cg_unified_get_controllers(cgv2_ctrl_path);
2570 free(cgv2_ctrl_path);
2571 if (!controller_list) {
2572 controller_list = cg_unified_make_empty_controller();
2573 TRACE("No controllers are enabled for "
2574 "delegation in the unified hierarchy");
2575 }
2576 }
2577
2578 /* Exclude all controllers that cgroup use does not want. */
2579 if (!cgroup_use_wants_controllers(ops, controller_list))
2580 goto next;
2581
2582 new = add_hierarchy(&ops->hierarchies, controller_list, mountpoint, base_cgroup, type);
2583 if (type == CGROUP2_SUPER_MAGIC && !ops->unified)
2584 ops->unified = new;
2585
2586 continue;
2587
2588 next:
2589 free_string_list(controller_list);
2590 free(mountpoint);
2591 free(base_cgroup);
2592 }
2593
2594 free_string_list(klist);
2595 free_string_list(nlist);
2596
2597 free(basecginfo);
2598
2599 fclose(f);
2600 free(line);
2601
2602 TRACE("Writable cgroup hierarchies:");
2603 lxc_cgfsng_print_hierarchies(ops);
2604
2605 /* verify that all controllers in cgroup.use and all crucial
2606 * controllers are accounted for
2607 */
2608 if (!all_controllers_found(ops))
2609 return false;
2610
2611 return true;
2612 }
2613
2614 static int cg_is_pure_unified(void)
2615 {
2616
2617 int ret;
2618 struct statfs fs;
2619
2620 ret = statfs("/sys/fs/cgroup", &fs);
2621 if (ret < 0)
2622 return -ENOMEDIUM;
2623
2624 if (is_fs_type(&fs, CGROUP2_SUPER_MAGIC))
2625 return CGROUP2_SUPER_MAGIC;
2626
2627 return 0;
2628 }
2629
2630 /* Get current cgroup from /proc/self/cgroup for the cgroupfs v2 hierarchy. */
2631 static char *cg_unified_get_current_cgroup(bool relative)
2632 {
2633 char *basecginfo, *base_cgroup;
2634 char *copy = NULL;
2635
2636 if (!relative && (geteuid() == 0))
2637 basecginfo = read_file("/proc/1/cgroup");
2638 else
2639 basecginfo = read_file("/proc/self/cgroup");
2640 if (!basecginfo)
2641 return NULL;
2642
2643 base_cgroup = strstr(basecginfo, "0::/");
2644 if (!base_cgroup)
2645 goto cleanup_on_err;
2646
2647 base_cgroup = base_cgroup + 3;
2648 copy = copy_to_eol(base_cgroup);
2649 if (!copy)
2650 goto cleanup_on_err;
2651
2652 cleanup_on_err:
2653 free(basecginfo);
2654 if (copy)
2655 trim(copy);
2656
2657 return copy;
2658 }
2659
2660 static int cg_unified_init(struct cgroup_ops *ops, bool relative)
2661 {
2662 int ret;
2663 char *mountpoint, *subtree_path;
2664 char **delegatable;
2665 char *base_cgroup = NULL;
2666
2667 ret = cg_is_pure_unified();
2668 if (ret == -ENOMEDIUM)
2669 return -ENOMEDIUM;
2670
2671 if (ret != CGROUP2_SUPER_MAGIC)
2672 return 0;
2673
2674 base_cgroup = cg_unified_get_current_cgroup(relative);
2675 if (!base_cgroup)
2676 return -EINVAL;
2677 prune_init_scope(base_cgroup);
2678
2679 /* We assume that we have already been given controllers to delegate
2680 * further down the hierarchy. If not it is up to the user to delegate
2681 * them to us.
2682 */
2683 mountpoint = must_copy_string("/sys/fs/cgroup");
2684 subtree_path = must_make_path(mountpoint, base_cgroup,
2685 "cgroup.subtree_control", NULL);
2686 delegatable = cg_unified_get_controllers(subtree_path);
2687 free(subtree_path);
2688 if (!delegatable)
2689 delegatable = cg_unified_make_empty_controller();
2690 if (!delegatable[0])
2691 TRACE("No controllers are enabled for delegation");
2692
2693 /* TODO: If the user requested specific controllers via lxc.cgroup.use
2694 * we should verify here. The reason I'm not doing it right is that I'm
2695 * not convinced that lxc.cgroup.use will be the future since it is a
2696 * global property. I much rather have an option that lets you request
2697 * controllers per container.
2698 */
2699
2700 add_hierarchy(&ops->hierarchies, delegatable, mountpoint, base_cgroup, CGROUP2_SUPER_MAGIC);
2701
2702 ops->cgroup_layout = CGROUP_LAYOUT_UNIFIED;
2703 return CGROUP2_SUPER_MAGIC;
2704 }
2705
2706 static bool cg_init(struct cgroup_ops *ops, struct lxc_conf *conf)
2707 {
2708 int ret;
2709 const char *tmp;
2710 bool relative = conf->cgroup_meta.relative;
2711
2712 tmp = lxc_global_config_value("lxc.cgroup.use");
2713 if (tmp) {
2714 char *chop, *cur, *pin;
2715
2716 pin = must_copy_string(tmp);
2717 chop = pin;
2718
2719 lxc_iterate_parts(cur, chop, ",") {
2720 must_append_string(&ops->cgroup_use, cur);
2721 }
2722
2723 free(pin);
2724 }
2725
2726 ret = cg_unified_init(ops, relative);
2727 if (ret < 0)
2728 return false;
2729
2730 if (ret == CGROUP2_SUPER_MAGIC)
2731 return true;
2732
2733 return cg_hybrid_init(ops, relative);
2734 }
2735
2736 __cgfsng_ops static bool cgfsng_data_init(struct cgroup_ops *ops)
2737 {
2738 const char *cgroup_pattern;
2739
2740 /* copy system-wide cgroup information */
2741 cgroup_pattern = lxc_global_config_value("lxc.cgroup.pattern");
2742 if (!cgroup_pattern) {
2743 /* lxc.cgroup.pattern is only NULL on error. */
2744 ERROR("Failed to retrieve cgroup pattern");
2745 return false;
2746 }
2747 ops->cgroup_pattern = must_copy_string(cgroup_pattern);
2748 ops->monitor_pattern = MONITOR_CGROUP;
2749
2750 return true;
2751 }
2752
2753 struct cgroup_ops *cgfsng_ops_init(struct lxc_conf *conf)
2754 {
2755 struct cgroup_ops *cgfsng_ops;
2756
2757 cgfsng_ops = malloc(sizeof(struct cgroup_ops));
2758 if (!cgfsng_ops)
2759 return NULL;
2760
2761 memset(cgfsng_ops, 0, sizeof(struct cgroup_ops));
2762 cgfsng_ops->cgroup_layout = CGROUP_LAYOUT_UNKNOWN;
2763
2764 if (!cg_init(cgfsng_ops, conf)) {
2765 free(cgfsng_ops);
2766 return NULL;
2767 }
2768
2769 cgfsng_ops->data_init = cgfsng_data_init;
2770 cgfsng_ops->payload_destroy = cgfsng_payload_destroy;
2771 cgfsng_ops->monitor_destroy = cgfsng_monitor_destroy;
2772 cgfsng_ops->monitor_create = cgfsng_monitor_create;
2773 cgfsng_ops->monitor_enter = cgfsng_monitor_enter;
2774 cgfsng_ops->payload_create = cgfsng_payload_create;
2775 cgfsng_ops->payload_enter = cgfsng_payload_enter;
2776 cgfsng_ops->escape = cgfsng_escape;
2777 cgfsng_ops->num_hierarchies = cgfsng_num_hierarchies;
2778 cgfsng_ops->get_hierarchies = cgfsng_get_hierarchies;
2779 cgfsng_ops->get_cgroup = cgfsng_get_cgroup;
2780 cgfsng_ops->get = cgfsng_get;
2781 cgfsng_ops->set = cgfsng_set;
2782 cgfsng_ops->unfreeze = cgfsng_unfreeze;
2783 cgfsng_ops->setup_limits = cgfsng_setup_limits;
2784 cgfsng_ops->driver = "cgfsng";
2785 cgfsng_ops->version = "1.0.0";
2786 cgfsng_ops->attach = cgfsng_attach;
2787 cgfsng_ops->chown = cgfsng_chown;
2788 cgfsng_ops->mount = cgfsng_mount;
2789 cgfsng_ops->nrtasks = cgfsng_nrtasks;
2790
2791 return cgfsng_ops;
2792 }