chiark / gitweb /
dbus: export ControlGroupPersistent field on the bus again
[elogind.git] / src / quotacheck.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 <stdio.h>
23 #include <stdbool.h>
24 #include <string.h>
25 #include <errno.h>
26 #include <unistd.h>
27
28 #include "util.h"
29 #include "virt.h"
30
31 static bool arg_skip = false;
32 static bool arg_force = false;
33
34 static int parse_proc_cmdline(void) {
35         char *line, *w, *state;
36         int r;
37         size_t l;
38
39         if (detect_container(NULL) > 0)
40                 return 0;
41
42         if ((r = read_one_line_file("/proc/cmdline", &line)) < 0) {
43                 log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r));
44                 return 0;
45         }
46
47         FOREACH_WORD_QUOTED(w, l, line, state) {
48
49                 if (strneq(w, "quotacheck.mode=auto", l))
50                         arg_force = arg_skip = false;
51                 else if (strneq(w, "quotacheck.mode=force", l))
52                         arg_force = true;
53                 else if (strneq(w, "quotacheck.mode=skip", l))
54                         arg_skip = true;
55                 else if (startswith(w, "quotacheck.mode"))
56                         log_warning("Invalid quotacheck.mode= parameter. Ignoring.");
57 #if defined(TARGET_FEDORA) || defined(TARGET_MANDRIVA) || defined(TARGET_MAGEIA)
58                 else if (strneq(w, "forcequotacheck", l))
59                         arg_force = true;
60 #endif
61         }
62
63         free(line);
64         return 0;
65 }
66
67 static void test_files(void) {
68 #if defined(TARGET_FEDORA) || defined(TARGET_MANDRIVA) || defined(TARGET_MAGEIA)
69         /* This exists only on Fedora, Mandriva or Mageia */
70         if (access("/forcequotacheck", F_OK) >= 0)
71                 arg_force = true;
72 #endif
73 }
74
75 int main(int argc, char *argv[]) {
76         static const char * const cmdline[] = {
77                 "/sbin/quotacheck",
78                 "-anug",
79                 NULL
80         };
81
82         int r = EXIT_FAILURE;
83         pid_t pid;
84
85         if (argc > 1) {
86                 log_error("This program takes no arguments.");
87                 return EXIT_FAILURE;
88         }
89
90         log_set_target(LOG_TARGET_AUTO);
91         log_parse_environment();
92         log_open();
93
94         umask(0022);
95
96         parse_proc_cmdline();
97         test_files();
98
99         if (!arg_force) {
100                 if (arg_skip)
101                         return 0;
102
103                 if (access("/run/systemd/quotacheck", F_OK) < 0)
104                         return 0;
105         }
106
107         if ((pid = fork()) < 0) {
108                 log_error("fork(): %m");
109                 goto finish;
110         } else if (pid == 0) {
111                 /* Child */
112                 execv(cmdline[0], (char**) cmdline);
113                 _exit(1); /* Operational error */
114         }
115
116         r = wait_for_terminate_and_warn("quotacheck", pid) == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
117
118 finish:
119         return r;
120 }