chiark / gitweb /
70a591883e70cc161165077fba3674e12c5960eb
[elogind.git] / src / fsck / fsck.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   Copyright 2014 Holger Hans Peter Freyther
8
9   systemd is free software; you can redistribute it and/or modify it
10   under the terms of the GNU Lesser General Public License as published by
11   the Free Software Foundation; either version 2.1 of the License, or
12   (at your option) any later version.
13
14   systemd is distributed in the hope that it will be useful, but
15   WITHOUT ANY WARRANTY; without even the implied warranty of
16   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17   Lesser General Public License for more details.
18
19   You should have received a copy of the GNU Lesser General Public License
20   along with systemd; If not, see <http://www.gnu.org/licenses/>.
21 ***/
22
23 #include <stdio.h>
24 #include <stdbool.h>
25 #include <string.h>
26 #include <errno.h>
27 #include <unistd.h>
28 #include <fcntl.h>
29 #include <sys/file.h>
30
31 #include "sd-bus.h"
32 #include "libudev.h"
33
34 #include "util.h"
35 #include "special.h"
36 #include "bus-util.h"
37 #include "bus-error.h"
38 #include "bus-errors.h"
39 #include "fileio.h"
40 #include "udev-util.h"
41 #include "path-util.h"
42
43 static bool arg_skip = false;
44 static bool arg_force = false;
45 static bool arg_show_progress = false;
46 static const char *arg_repair = "-a";
47
48 static void start_target(const char *target) {
49         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
50         _cleanup_bus_close_unref_ sd_bus *bus = NULL;
51         int r;
52
53         assert(target);
54
55         r = bus_open_system_systemd(&bus);
56         if (r < 0) {
57                 log_error("Failed to get D-Bus connection: %s", strerror(-r));
58                 return;
59         }
60
61         log_info("Running request %s/start/replace", target);
62
63         /* Start these units only if we can replace base.target with it */
64         r = sd_bus_call_method(bus,
65                                "org.freedesktop.systemd1",
66                                "/org/freedesktop/systemd1",
67                                "org.freedesktop.systemd1.Manager",
68                                "StartUnitReplace",
69                                &error,
70                                NULL,
71                                "sss", "basic.target", target, "replace");
72
73         /* Don't print a warning if we aren't called during startup */
74         if (r < 0 && !sd_bus_error_has_name(&error, BUS_ERROR_NO_SUCH_JOB))
75                 log_error("Failed to start unit: %s", bus_error_message(&error, -r));
76 }
77
78 static int parse_proc_cmdline_item(const char *key, const char *value) {
79
80         if (streq(key, "fsck.mode") && value) {
81
82                 if (streq(value, "auto"))
83                         arg_force = arg_skip = false;
84                 else if (streq(value, "force"))
85                         arg_force = true;
86                 else if (streq(value, "skip"))
87                         arg_skip = true;
88                 else
89                         log_warning("Invalid fsck.mode= parameter '%s'. Ignoring.", value);
90
91         } else if (streq(key, "fsck.repair") && value) {
92
93                 if (streq(value, "preen"))
94                         arg_repair = "-a";
95                 else if (streq(value, "yes"))
96                         arg_repair = "-y";
97                 else if (streq(value, "no"))
98                         arg_repair = "-n";
99                 else
100                         log_warning("Invalid fsck.repair= parameter '%s'. Ignoring.", value);
101         }
102
103 #ifdef HAVE_SYSV_COMPAT
104         else if (streq(key, "fastboot") && !value) {
105                 log_warning("Please pass 'fsck.mode=skip' rather than 'fastboot' on the kernel command line.");
106                 arg_skip = true;
107
108         } else if (streq(key, "forcefsck") && !value) {
109                 log_warning("Please pass 'fsck.mode=force' rather than 'forcefsck' on the kernel command line.");
110                 arg_force = true;
111         }
112 #endif
113
114         return 0;
115 }
116
117 static void test_files(void) {
118
119 #ifdef HAVE_SYSV_COMPAT
120         if (access("/fastboot", F_OK) >= 0) {
121                 log_error("Please pass 'fsck.mode=skip' on the kernel command line rather than creating /fastboot on the root file system.");
122                 arg_skip = true;
123         }
124
125         if (access("/forcefsck", F_OK) >= 0) {
126                 log_error("Please pass 'fsck.mode=force' on the kernel command line rather than creating /forcefsck on the root file system.");
127                 arg_force = true;
128         }
129 #endif
130
131         if (access("/run/systemd/show-status", F_OK) >= 0 || plymouth_running())
132                 arg_show_progress = true;
133 }
134
135 static double percent(int pass, unsigned long cur, unsigned long max) {
136         /* Values stolen from e2fsck */
137
138         static const int pass_table[] = {
139                 0, 70, 90, 92, 95, 100
140         };
141
142         if (pass <= 0)
143                 return 0.0;
144
145         if ((unsigned) pass >= ELEMENTSOF(pass_table) || max == 0)
146                 return 100.0;
147
148         return (double) pass_table[pass-1] +
149                 ((double) pass_table[pass] - (double) pass_table[pass-1]) *
150                 (double) cur / (double) max;
151 }
152
153 static int process_progress(int fd) {
154         _cleanup_fclose_ FILE *console = NULL, *f = NULL;
155         usec_t last = 0;
156         bool locked = false;
157         int clear = 0;
158
159         f = fdopen(fd, "r");
160         if (!f) {
161                 safe_close(fd);
162                 return -errno;
163         }
164
165         console = fopen("/dev/console", "we");
166         if (!console)
167                 return -ENOMEM;
168
169         while (!feof(f)) {
170                 int pass, m;
171                 unsigned long cur, max;
172                 _cleanup_free_ char *device = NULL;
173                 double p;
174                 usec_t t;
175
176                 if (fscanf(f, "%i %lu %lu %ms", &pass, &cur, &max, &device) != 4)
177                         break;
178
179                 /* Only show one progress counter at max */
180                 if (!locked) {
181                         if (flock(fileno(console), LOCK_EX|LOCK_NB) < 0)
182                                 continue;
183
184                         locked = true;
185                 }
186
187                 /* Only update once every 50ms */
188                 t = now(CLOCK_MONOTONIC);
189                 if (last + 50 * USEC_PER_MSEC > t)
190                         continue;
191
192                 last = t;
193
194                 p = percent(pass, cur, max);
195                 fprintf(console, "\r%s: fsck %3.1f%% complete...\r%n", device, p, &m);
196                 fflush(console);
197
198                 if (m > clear)
199                         clear = m;
200         }
201
202         if (clear > 0) {
203                 unsigned j;
204
205                 fputc('\r', console);
206                 for (j = 0; j < (unsigned) clear; j++)
207                         fputc(' ', console);
208                 fputc('\r', console);
209                 fflush(console);
210         }
211
212         return 0;
213 }
214
215 int main(int argc, char *argv[]) {
216         const char *cmdline[9];
217         int i = 0, r = EXIT_FAILURE, q;
218         pid_t pid;
219         siginfo_t status;
220         _cleanup_udev_unref_ struct udev *udev = NULL;
221         _cleanup_udev_device_unref_ struct udev_device *udev_device = NULL;
222         const char *device, *type;
223         bool root_directory;
224         int progress_pipe[2] = { -1, -1 };
225         char dash_c[2+10+1];
226         struct stat st;
227
228         if (argc > 2) {
229                 log_error("This program expects one or no arguments.");
230                 return EXIT_FAILURE;
231         }
232
233         log_set_target(LOG_TARGET_AUTO);
234         log_parse_environment();
235         log_open();
236
237         umask(0022);
238
239         parse_proc_cmdline(parse_proc_cmdline_item);
240         test_files();
241
242         if (!arg_force && arg_skip)
243                 return 0;
244
245         udev = udev_new();
246         if (!udev) {
247                 log_oom();
248                 return EXIT_FAILURE;
249         }
250
251         if (argc > 1) {
252                 device = argv[1];
253                 root_directory = false;
254
255                 if (stat(device, &st) < 0) {
256                         log_error("Failed to stat '%s': %m", device);
257                         return EXIT_FAILURE;
258                 }
259
260                 udev_device = udev_device_new_from_devnum(udev, 'b', st.st_rdev);
261                 if (!udev_device) {
262                         log_error("Failed to detect device %s", device);
263                         return EXIT_FAILURE;
264                 }
265         } else {
266                 struct timespec times[2];
267
268                 /* Find root device */
269
270                 if (stat("/", &st) < 0) {
271                         log_error("Failed to stat() the root directory: %m");
272                         return EXIT_FAILURE;
273                 }
274
275                 /* Virtual root devices don't need an fsck */
276                 if (major(st.st_dev) == 0)
277                         return EXIT_SUCCESS;
278
279                 /* check if we are already writable */
280                 times[0] = st.st_atim;
281                 times[1] = st.st_mtim;
282                 if (utimensat(AT_FDCWD, "/", times, 0) == 0) {
283                         log_info("Root directory is writable, skipping check.");
284                         return EXIT_SUCCESS;
285                 }
286
287                 udev_device = udev_device_new_from_devnum(udev, 'b', st.st_dev);
288                 if (!udev_device) {
289                         log_error("Failed to detect root device.");
290                         return EXIT_FAILURE;
291                 }
292
293                 device = udev_device_get_devnode(udev_device);
294                 if (!device) {
295                         log_error("Failed to detect device node of root directory.");
296                         return EXIT_FAILURE;
297                 }
298
299                 root_directory = true;
300         }
301
302         type = udev_device_get_property_value(udev_device, "ID_FS_TYPE");
303         if (type) {
304                 r = fsck_exists(type);
305                 if (r == -ENOENT) {
306                         log_info("fsck.%s doesn't exist, not checking file system on %s", type, device);
307                         return EXIT_SUCCESS;
308                 } else if (r < 0)
309                         log_warning("fsck.%s cannot be used for %s: %s", type, device, strerror(-r));
310         }
311
312         if (arg_show_progress)
313                 if (pipe(progress_pipe) < 0) {
314                         log_error("pipe(): %m");
315                         return EXIT_FAILURE;
316                 }
317
318         cmdline[i++] = "/sbin/fsck";
319         cmdline[i++] =  arg_repair;
320         cmdline[i++] = "-T";
321
322         /*
323          * Since util-linux v2.25 fsck uses /run/fsck/<diskname>.lock files.
324          * The previous versions use flock for the device and conflict with
325          * udevd, see https://bugs.freedesktop.org/show_bug.cgi?id=79576#c5
326          */
327         cmdline[i++] = "-l";
328
329         if (!root_directory)
330                 cmdline[i++] = "-M";
331
332         if (arg_force)
333                 cmdline[i++] = "-f";
334
335         if (progress_pipe[1] >= 0) {
336                 snprintf(dash_c, sizeof(dash_c), "-C%i", progress_pipe[1]);
337                 char_array_0(dash_c);
338                 cmdline[i++] = dash_c;
339         }
340
341         cmdline[i++] = device;
342         cmdline[i++] = NULL;
343
344         pid = fork();
345         if (pid < 0) {
346                 log_error("fork(): %m");
347                 goto finish;
348         } else if (pid == 0) {
349                 /* Child */
350                 if (progress_pipe[0] >= 0)
351                         safe_close(progress_pipe[0]);
352                 execv(cmdline[0], (char**) cmdline);
353                 _exit(8); /* Operational error */
354         }
355
356         progress_pipe[1] = safe_close(progress_pipe[1]);
357
358         if (progress_pipe[0] >= 0) {
359                 process_progress(progress_pipe[0]);
360                 progress_pipe[0] = -1;
361         }
362
363         q = wait_for_terminate(pid, &status);
364         if (q < 0) {
365                 log_error("waitid(): %s", strerror(-q));
366                 goto finish;
367         }
368
369         if (status.si_code != CLD_EXITED || (status.si_status & ~1)) {
370
371                 if (status.si_code == CLD_KILLED || status.si_code == CLD_DUMPED)
372                         log_error("fsck terminated by signal %s.", signal_to_string(status.si_status));
373                 else if (status.si_code == CLD_EXITED)
374                         log_error("fsck failed with error code %i.", status.si_status);
375                 else
376                         log_error("fsck failed due to unknown reason.");
377
378                 if (status.si_code == CLD_EXITED && (status.si_status & 2) && root_directory)
379                         /* System should be rebooted. */
380                         start_target(SPECIAL_REBOOT_TARGET);
381                 else if (status.si_code == CLD_EXITED && (status.si_status & 6))
382                         /* Some other problem */
383                         start_target(SPECIAL_EMERGENCY_TARGET);
384                 else {
385                         r = EXIT_SUCCESS;
386                         log_warning("Ignoring error.");
387                 }
388
389         } else
390                 r = EXIT_SUCCESS;
391
392         if (status.si_code == CLD_EXITED && (status.si_status & 1))
393                 touch("/run/systemd/quotacheck");
394
395 finish:
396         safe_close_pair(progress_pipe);
397
398         return r;
399 }