chiark / gitweb /
cgroup: if we couldn't remove a cgroup after killing evertyhing in it then it's fine
[elogind.git] / src / cgroup-util.c
1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
2
3 /***
4   This file is part of systemd.
5
6   Copyright 2010 Lennart Poettering
7
8   systemd is free software; you can redistribute it and/or modify it
9   under the terms of the GNU General Public License as published by
10   the Free Software Foundation; either version 2 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   General Public License for more details.
17
18   You should have received a copy of the GNU General Public License
19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
20 ***/
21
22 #include <errno.h>
23 #include <unistd.h>
24 #include <signal.h>
25 #include <string.h>
26 #include <stdlib.h>
27 #include <dirent.h>
28 #include <sys/stat.h>
29 #include <sys/types.h>
30
31 #include "cgroup-util.h"
32 #include "log.h"
33 #include "set.h"
34 #include "macro.h"
35 #include "util.h"
36
37 int cg_enumerate_processes(const char *controller, const char *path, FILE **_f) {
38         char *fs;
39         int r;
40         FILE *f;
41
42         assert(controller);
43         assert(path);
44         assert(_f);
45
46         if ((r = cg_get_path(controller, path, "cgroup.procs", &fs)) < 0)
47                 return r;
48
49         f = fopen(fs, "re");
50         free(fs);
51
52         if (!f)
53                 return -errno;
54
55         *_f = f;
56         return 0;
57 }
58
59 int cg_enumerate_tasks(const char *controller, const char *path, FILE **_f) {
60         char *fs;
61         int r;
62         FILE *f;
63
64         assert(controller);
65         assert(path);
66         assert(_f);
67
68         if ((r = cg_get_path(controller, path, "tasks", &fs)) < 0)
69                 return r;
70
71         f = fopen(fs, "re");
72         free(fs);
73
74         if (!f)
75                 return -errno;
76
77         *_f = f;
78         return 0;
79 }
80
81 int cg_read_pid(FILE *f, pid_t *_pid) {
82         unsigned long ul;
83
84         /* Note that the cgroup.procs might contain duplicates! See
85          * cgroups.txt for details. */
86
87         errno = 0;
88         if (fscanf(f, "%lu", &ul) != 1) {
89
90                 if (feof(f))
91                         return 0;
92
93                 return errno ? -errno : -EIO;
94         }
95
96         if (ul <= 0)
97                 return -EIO;
98
99         *_pid = (pid_t) ul;
100         return 1;
101 }
102
103 int cg_enumerate_subgroups(const char *controller, const char *path, DIR **_d) {
104         char *fs;
105         int r;
106         DIR *d;
107
108         assert(controller);
109         assert(path);
110         assert(_d);
111
112         /* This is not recursive! */
113
114         if ((r = cg_get_path(controller, path, NULL, &fs)) < 0)
115                 return r;
116
117         d = opendir(fs);
118         free(fs);
119
120         if (!d)
121                 return -errno;
122
123         *_d = d;
124         return 0;
125 }
126
127 int cg_read_subgroup(DIR *d, char **fn) {
128         struct dirent *de;
129
130         assert(d);
131
132         errno = 0;
133         while ((de = readdir(d))) {
134                 char *b;
135
136                 if (de->d_type != DT_DIR)
137                         continue;
138
139                 if (streq(de->d_name, ".") ||
140                     streq(de->d_name, ".."))
141                         continue;
142
143                 if (!(b = strdup(de->d_name)))
144                         return -ENOMEM;
145
146                 *fn = b;
147                 return 1;
148         }
149
150         if (errno)
151                 return -errno;
152
153         return 0;
154 }
155
156 int cg_rmdir(const char *controller, const char *path) {
157         char *p;
158         int r;
159
160         if ((r = cg_get_path(controller, path, NULL, &p)) < 0)
161                 return r;
162
163         r = rmdir(p);
164         free(p);
165
166         return r < 0 ? -errno : 0;
167 }
168
169 int cg_kill(const char *controller, const char *path, int sig, bool ignore_self, Set *s) {
170         bool done = false;
171         int r, ret = 0;
172         pid_t my_pid;
173         FILE *f = NULL;
174         Set *allocated_set = NULL;
175
176         assert(controller);
177         assert(path);
178         assert(sig >= 0);
179
180         /* This goes through the tasks list and kills them all. This
181          * is repeated until no further processes are added to the
182          * tasks list, to properly handle forking processes */
183
184         if (!s)
185                 if (!(s = allocated_set = set_new(trivial_hash_func, trivial_compare_func)))
186                         return -ENOMEM;
187
188         my_pid = getpid();
189
190         do {
191                 pid_t pid;
192                 done = true;
193
194                 if ((r = cg_enumerate_processes(controller, path, &f)) < 0) {
195                         if (ret >= 0 && r != -ENOENT)
196                                 ret = r;
197
198                         goto finish;
199                 }
200
201                 while ((r = cg_read_pid(f, &pid)) > 0) {
202
203                         if (pid == my_pid && ignore_self)
204                                 continue;
205
206                         if (set_get(s, LONG_TO_PTR(pid)) == LONG_TO_PTR(pid))
207                                 continue;
208
209                         /* If we haven't killed this process yet, kill
210                          * it */
211                         if (kill(pid, sig) < 0) {
212                                 if (ret >= 0 && errno != ESRCH)
213                                         ret = -errno;
214                         } else if (ret == 0)
215                                 ret = 1;
216
217                         done = false;
218
219                         if ((r = set_put(s, LONG_TO_PTR(pid))) < 0) {
220                                 if (ret >= 0)
221                                         ret = r;
222
223                                 goto finish;
224                         }
225                 }
226
227                 if (r < 0) {
228                         if (ret >= 0)
229                                 ret = r;
230
231                         goto finish;
232                 }
233
234                 fclose(f);
235                 f = NULL;
236
237                 /* To avoid racing against processes which fork
238                  * quicker than we can kill them we repeat this until
239                  * no new pids need to be killed. */
240
241         } while (!done);
242
243 finish:
244         if (allocated_set)
245                 set_free(allocated_set);
246
247         if (f)
248                 fclose(f);
249
250         return ret;
251 }
252
253 int cg_kill_recursive(const char *controller, const char *path, int sig, bool ignore_self, bool rem, Set *s) {
254         int r, ret = 0;
255         DIR *d = NULL;
256         char *fn;
257         Set *allocated_set = NULL;
258
259         assert(path);
260         assert(controller);
261         assert(sig >= 0);
262
263         if (!s)
264                 if (!(s = allocated_set = set_new(trivial_hash_func, trivial_compare_func)))
265                         return -ENOMEM;
266
267         ret = cg_kill(controller, path, sig, ignore_self, s);
268
269         if ((r = cg_enumerate_subgroups(controller, path, &d)) < 0) {
270                 if (ret >= 0 && r != -ENOENT)
271                         ret = r;
272
273                 goto finish;
274         }
275
276         while ((r = cg_read_subgroup(d, &fn)) > 0) {
277                 char *p = NULL;
278
279                 r = asprintf(&p, "%s/%s", path, fn);
280                 free(fn);
281
282                 if (r < 0) {
283                         if (ret >= 0)
284                                 ret = -ENOMEM;
285
286                         goto finish;
287                 }
288
289                 r = cg_kill_recursive(controller, p, sig, ignore_self, rem, s);
290                 free(p);
291
292                 if (r != 0 && ret >= 0)
293                         ret = r;
294         }
295
296         if (r < 0 && ret >= 0)
297                 ret = r;
298
299         if (rem)
300                 if ((r = cg_rmdir(controller, path)) < 0) {
301                         if (ret >= 0 &&
302                             r != -ENOENT &&
303                             r != -EBUSY)
304                                 ret = r;
305                 }
306
307 finish:
308         if (d)
309                 closedir(d);
310
311         if (allocated_set)
312                 set_free(allocated_set);
313
314         return ret;
315 }
316
317 int cg_kill_recursive_and_wait(const char *controller, const char *path, bool rem) {
318         unsigned i;
319
320         assert(path);
321         assert(controller);
322
323         /* This safely kills all processes; first it sends a SIGTERM,
324          * then checks 8 times after 50ms whether the group is
325          * now empty, and finally kills everything that is left with
326          * SIGKILL */
327
328         for (i = 0; i < 10; i++) {
329                 int sig, r;
330
331                 if (i <= 0)
332                         sig = SIGTERM;
333                 else if (i >= 9)
334                         sig = SIGKILL;
335                 else
336                         sig = 0;
337
338                 if ((r = cg_kill_recursive(controller, path, sig, true, rem, NULL)) <= 0)
339                         return r;
340
341                 usleep(50 * USEC_PER_MSEC);
342         }
343
344         return 0;
345 }
346
347 int cg_migrate(const char *controller, const char *from, const char *to, bool ignore_self) {
348         bool done = false;
349         Set *s;
350         int r, ret = 0;
351         pid_t my_pid;
352         FILE *f = NULL;
353
354         assert(controller);
355         assert(from);
356         assert(to);
357
358         if (!(s = set_new(trivial_hash_func, trivial_compare_func)))
359                 return -ENOMEM;
360
361         my_pid = getpid();
362
363         do {
364                 pid_t pid;
365                 done = true;
366
367                 if ((r = cg_enumerate_tasks(controller, from, &f)) < 0) {
368                         if (ret >= 0 && r != -ENOENT)
369                                 ret = r;
370
371                         goto finish;
372                 }
373
374                 while ((r = cg_read_pid(f, &pid)) > 0) {
375
376                         /* This might do weird stuff if we aren't a
377                          * single-threaded program. However, we
378                          * luckily know we are not */
379                         if (pid == my_pid && ignore_self)
380                                 continue;
381
382                         if (set_get(s, LONG_TO_PTR(pid)) == LONG_TO_PTR(pid))
383                                 continue;
384
385                         if ((r = cg_attach(controller, to, pid)) < 0) {
386                                 if (ret >= 0 && r != -ESRCH)
387                                         ret = r;
388                         } else if (ret == 0)
389                                 ret = 1;
390
391                         done = false;
392
393                         if ((r = set_put(s, LONG_TO_PTR(pid))) < 0) {
394                                 if (ret >= 0)
395                                         ret = r;
396
397                                 goto finish;
398                         }
399                 }
400
401                 if (r < 0) {
402                         if (ret >= 0)
403                                 ret = r;
404
405                         goto finish;
406                 }
407
408                 fclose(f);
409                 f = NULL;
410
411         } while (!done);
412
413 finish:
414         set_free(s);
415
416         if (f)
417                 fclose(f);
418
419         return ret;
420 }
421
422 int cg_migrate_recursive(const char *controller, const char *from, const char *to, bool ignore_self, bool rem) {
423         int r, ret = 0;
424         DIR *d = NULL;
425         char *fn;
426
427         assert(controller);
428         assert(from);
429         assert(to);
430
431         ret = cg_migrate(controller, from, to, ignore_self);
432
433         if ((r = cg_enumerate_subgroups(controller, from, &d)) < 0) {
434                 if (ret >= 0 && r != -ENOENT)
435                         ret = r;
436                 goto finish;
437         }
438
439         while ((r = cg_read_subgroup(d, &fn)) > 0) {
440                 char *p = NULL;
441
442                 r = asprintf(&p, "%s/%s", from, fn);
443                 free(fn);
444
445                 if (r < 0) {
446                         if (ret >= 0)
447                                 ret = -ENOMEM;
448
449                         goto finish;
450                 }
451
452                 r = cg_migrate_recursive(controller, p, to, ignore_self, rem);
453                 free(p);
454
455                 if (r != 0 && ret >= 0)
456                         ret = r;
457         }
458
459         if (r < 0 && ret >= 0)
460                 ret = r;
461
462         if (rem)
463                 if ((r = cg_rmdir(controller, from)) < 0) {
464                         if (ret >= 0 &&
465                             r != -ENOENT &&
466                             r != -EBUSY)
467                                 ret = r;
468                 }
469
470 finish:
471         if (d)
472                 closedir(d);
473
474         return ret;
475 }
476
477 int cg_get_path(const char *controller, const char *path, const char *suffix, char **fs) {
478         const char *p;
479         char *mp;
480         int r;
481
482         assert(controller);
483         assert(fs);
484
485         /* This is a very minimal lookup from controller names to
486          * paths. Since we have mounted most hierarchies ourselves
487          * should be kinda safe, but eventually we might want to
488          * extend this to have a fallback to actually check
489          * /proc/mounts. Might need caching then. */
490
491         if (streq(controller, SYSTEMD_CGROUP_CONTROLLER))
492                 p = "systemd";
493         else if (startswith(controller, "name="))
494                 p = controller + 5;
495         else
496                 p = controller;
497
498         if (asprintf(&mp, "/sys/fs/cgroup/%s", p) < 0)
499                 return -ENOMEM;
500
501         if ((r = path_is_mount_point(mp)) <= 0) {
502                 free(mp);
503                 return r < 0 ? r : -ENOENT;
504         }
505
506         if (path && suffix)
507                 r = asprintf(fs, "%s/%s/%s", mp, path, suffix);
508         else if (path)
509                 r = asprintf(fs, "%s/%s", mp, path);
510         else if (suffix)
511                 r = asprintf(fs, "%s/%s", mp, suffix);
512         else {
513                 path_kill_slashes(mp);
514                 *fs = mp;
515                 return 0;
516         }
517
518         free(mp);
519         path_kill_slashes(*fs);
520         return r < 0 ? -ENOMEM : 0;
521 }
522
523 int cg_trim(const char *controller, const char *path, bool delete_root) {
524         char *fs;
525         int r;
526
527         assert(controller);
528         assert(path);
529
530         if ((r = cg_get_path(controller, path, NULL, &fs)) < 0)
531                 return r;
532
533         r = rm_rf(fs, true, delete_root);
534         free(fs);
535
536         return r == -ENOENT ? 0 : r;
537 }
538
539 int cg_delete(const char *controller, const char *path) {
540         char *parent;
541         int r;
542
543         assert(controller);
544         assert(path);
545
546         if ((r = parent_of_path(path, &parent)) < 0)
547                 return r;
548
549         r = cg_migrate_recursive(controller, path, parent, false, true);
550         free(parent);
551
552         return r == -ENOENT ? 0 : r;
553 }
554
555 int cg_create(const char *controller, const char *path) {
556         char *fs;
557         int r;
558
559         assert(controller);
560         assert(path);
561
562         if ((r = cg_get_path(controller, path, NULL, &fs)) < 0)
563                 return r;
564
565         r = mkdir_parents(fs, 0755);
566
567         if (r >= 0) {
568                 if (mkdir(fs, 0755) >= 0)
569                         r = 1;
570                 else if (errno == EEXIST)
571                         r = 0;
572                 else
573                         r = -errno;
574         }
575
576         free(fs);
577
578         return r;
579 }
580
581 int cg_attach(const char *controller, const char *path, pid_t pid) {
582         char *fs;
583         int r;
584         char c[32];
585
586         assert(controller);
587         assert(path);
588         assert(pid >= 0);
589
590         if ((r = cg_get_path(controller, path, "tasks", &fs)) < 0)
591                 return r;
592
593         if (pid == 0)
594                 pid = getpid();
595
596         snprintf(c, sizeof(c), "%lu\n", (unsigned long) pid);
597         char_array_0(c);
598
599         r = write_one_line_file(fs, c);
600         free(fs);
601
602         return r;
603 }
604
605 int cg_create_and_attach(const char *controller, const char *path, pid_t pid) {
606         int r, q;
607
608         assert(controller);
609         assert(path);
610         assert(pid >= 0);
611
612         if ((r = cg_create(controller, path)) < 0)
613                 return r;
614
615         if ((q = cg_attach(controller, path, pid)) < 0)
616                 return q;
617
618         /* This does not remove the cgroup on failure */
619
620         return r;
621 }
622
623 int cg_set_group_access(const char *controller, const char *path, mode_t mode, uid_t uid, gid_t gid) {
624         char *fs;
625         int r;
626
627         assert(controller);
628         assert(path);
629
630         if ((r = cg_get_path(controller, path, NULL, &fs)) < 0)
631                 return r;
632
633         r = chmod_and_chown(fs, mode, uid, gid);
634         free(fs);
635
636         return r;
637 }
638
639 int cg_set_task_access(const char *controller, const char *path, mode_t mode, uid_t uid, gid_t gid) {
640         char *fs;
641         int r;
642
643         assert(controller);
644         assert(path);
645
646         if ((r = cg_get_path(controller, path, "tasks", &fs)) < 0)
647                 return r;
648
649         r = chmod_and_chown(fs, mode, uid, gid);
650         free(fs);
651
652         return r;
653 }
654
655 int cg_get_by_pid(const char *controller, pid_t pid, char **path) {
656         int r;
657         char *p = NULL;
658         FILE *f;
659         char *fs;
660         size_t cs;
661
662         assert(controller);
663         assert(path);
664         assert(pid >= 0);
665
666         if (pid == 0)
667                 pid = getpid();
668
669         if (asprintf(&fs, "/proc/%lu/cgroup", (unsigned long) pid) < 0)
670                 return -ENOMEM;
671
672         f = fopen(fs, "re");
673         free(fs);
674
675         if (!f)
676                 return errno == ENOENT ? -ESRCH : -errno;
677
678         cs = strlen(controller);
679
680         while (!feof(f)) {
681                 char line[LINE_MAX];
682                 char *l;
683
684                 errno = 0;
685                 if (!(fgets(line, sizeof(line), f))) {
686                         if (feof(f))
687                                 break;
688
689                         r = errno ? -errno : -EIO;
690                         goto finish;
691                 }
692
693                 truncate_nl(line);
694
695                 if (!(l = strchr(line, ':')))
696                         continue;
697
698                 l++;
699                 if (strncmp(l, controller, cs) != 0)
700                         continue;
701
702                 if (l[cs] != ':')
703                         continue;
704
705                 if (!(p = strdup(l + cs + 1))) {
706                         r = -ENOMEM;
707                         goto finish;
708                 }
709
710                 *path = p;
711                 r = 0;
712                 goto finish;
713         }
714
715         r = -ENOENT;
716
717 finish:
718         fclose(f);
719
720         return r;
721 }
722
723 int cg_install_release_agent(const char *controller, const char *agent) {
724         char *fs = NULL, *contents = NULL, *line = NULL, *sc;
725         int r;
726
727         assert(controller);
728         assert(agent);
729
730         if ((r = cg_get_path(controller, NULL, "release_agent", &fs)) < 0)
731                 return r;
732
733         if ((r = read_one_line_file(fs, &contents)) < 0)
734                 goto finish;
735
736         sc = strstrip(contents);
737         if (sc[0] == 0) {
738
739                 if (asprintf(&line, "%s\n", agent) < 0) {
740                         r = -ENOMEM;
741                         goto finish;
742                 }
743
744                 if ((r = write_one_line_file(fs, line)) < 0)
745                         goto finish;
746
747         } else if (!streq(sc, agent)) {
748                 r = -EEXIST;
749                 goto finish;
750         }
751
752         free(fs);
753         fs = NULL;
754         if ((r = cg_get_path(controller, NULL, "notify_on_release", &fs)) < 0)
755                 goto finish;
756
757         free(contents);
758         contents = NULL;
759         if ((r = read_one_line_file(fs, &contents)) < 0)
760                 goto finish;
761
762         sc = strstrip(contents);
763
764         if (streq(sc, "0")) {
765                 if ((r = write_one_line_file(fs, "1\n")) < 0)
766                         goto finish;
767
768                 r = 1;
769         } else if (!streq(sc, "1")) {
770                 r = -EIO;
771                 goto finish;
772         } else
773                 r = 0;
774
775 finish:
776         free(fs);
777         free(contents);
778         free(line);
779
780         return r;
781 }
782
783 int cg_is_empty(const char *controller, const char *path, bool ignore_self) {
784         pid_t pid;
785         int r;
786         FILE *f;
787         bool found = false;
788
789         assert(controller);
790         assert(path);
791
792         if ((r = cg_enumerate_tasks(controller, path, &f)) < 0)
793                 return r == -ENOENT ? 1 : r;
794
795         while ((r = cg_read_pid(f, &pid)) > 0) {
796
797                 if (ignore_self && pid == getpid())
798                         continue;
799
800                 found = true;
801                 break;
802         }
803
804         fclose(f);
805
806         if (r < 0)
807                 return r;
808
809         return !found;
810 }
811
812 int cg_is_empty_recursive(const char *controller, const char *path, bool ignore_self) {
813         int r;
814         DIR *d = NULL;
815         char *fn;
816
817         assert(controller);
818         assert(path);
819
820         if ((r = cg_is_empty(controller, path, ignore_self)) <= 0)
821                 return r;
822
823         if ((r = cg_enumerate_subgroups(controller, path, &d)) < 0)
824                 return r == -ENOENT ? 1 : r;
825
826         while ((r = cg_read_subgroup(d, &fn)) > 0) {
827                 char *p = NULL;
828
829                 r = asprintf(&p, "%s/%s", path, fn);
830                 free(fn);
831
832                 if (r < 0) {
833                         r = -ENOMEM;
834                         goto finish;
835                 }
836
837                 r = cg_is_empty_recursive(controller, p, ignore_self);
838                 free(p);
839
840                 if (r <= 0)
841                         goto finish;
842         }
843
844         if (r >= 0)
845                 r = 1;
846
847 finish:
848
849         if (d)
850                 closedir(d);
851
852         return r;
853 }
854
855 int cg_split_spec(const char *spec, char **controller, char **path) {
856         const char *e;
857         char *t = NULL, *u = NULL;
858
859         assert(spec);
860         assert(controller || path);
861
862         if (*spec == '/') {
863
864                 if (path) {
865                         if (!(t = strdup(spec)))
866                                 return -ENOMEM;
867
868                         *path = t;
869                 }
870
871                 if (controller)
872                         *controller = NULL;
873
874                 return 0;
875         }
876
877         if (!(e = strchr(spec, ':'))) {
878
879                 if (strchr(spec, '/') || spec[0] == 0)
880                         return -EINVAL;
881
882                 if (controller) {
883                         if (!(t = strdup(spec)))
884                                 return -ENOMEM;
885
886                         *controller = t;
887                 }
888
889                 if (path)
890                         *path = NULL;
891
892                 return 0;
893         }
894
895         if (e[1] != '/' ||
896             e == spec ||
897             memchr(spec, '/', e-spec))
898                 return -EINVAL;
899
900         if (controller)
901                 if (!(t = strndup(spec, e-spec)))
902                         return -ENOMEM;
903
904         if (path)
905                 if (!(u = strdup(e+1))) {
906                         free(t);
907                         return -ENOMEM;
908                 }
909
910         if (controller)
911                 *controller = t;
912
913         if (path)
914                 *path = u;
915
916         return 0;
917 }
918
919 int cg_join_spec(const char *controller, const char *path, char **spec) {
920         assert(controller);
921         assert(path);
922
923         if (!path_is_absolute(path) ||
924             controller[0] == 0 ||
925             strchr(controller, ':') ||
926             strchr(controller, '/'))
927                 return -EINVAL;
928
929         if (asprintf(spec, "%s:%s", controller, path) < 0)
930                 return -ENOMEM;
931
932         return 0;
933 }
934
935 int cg_fix_path(const char *path, char **result) {
936         char *t, *c, *p;
937         int r;
938
939         assert(path);
940         assert(result);
941
942         /* First check if it already is a filesystem path */
943         if (path_is_absolute(path) &&
944             path_startswith(path, "/sys/fs/cgroup") &&
945             access(path, F_OK) >= 0) {
946
947                 if (!(t = strdup(path)))
948                         return -ENOMEM;
949
950                 *result = t;
951                 return 0;
952         }
953
954         /* Otherwise treat it as cg spec */
955         if ((r = cg_split_spec(path, &c, &p)) < 0)
956                 return r;
957
958         r = cg_get_path(c ? c : SYSTEMD_CGROUP_CONTROLLER, p ? p : "/", NULL, result);
959         free(c);
960         free(p);
961
962         return r;
963 }