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