]>
Commit | Line | Data |
---|---|---|
3d20ed6d LP |
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> | |
a84f5192 | 27 | #include <fcntl.h> |
27d340c7 | 28 | #include <sys/file.h> |
3d20ed6d | 29 | |
a9e1f5ec | 30 | #include <libudev.h> |
3d20ed6d LP |
31 | #include <dbus/dbus.h> |
32 | ||
33 | #include "util.h" | |
3d20ed6d LP |
34 | #include "dbus-common.h" |
35 | #include "special.h" | |
ef1de59b | 36 | #include "bus-errors.h" |
5dc4c17f | 37 | #include "virt.h" |
3d20ed6d LP |
38 | |
39 | static bool arg_skip = false; | |
40 | static bool arg_force = false; | |
27d340c7 | 41 | static bool arg_show_progress = false; |
3d20ed6d LP |
42 | |
43 | static void start_target(const char *target, bool isolate) { | |
44 | DBusMessage *m = NULL, *reply = NULL; | |
45 | DBusError error; | |
70f12d37 | 46 | const char *mode, *basic_target = "basic.target"; |
3d20ed6d LP |
47 | DBusConnection *bus = NULL; |
48 | ||
49 | assert(target); | |
50 | ||
51 | dbus_error_init(&error); | |
52 | ||
53 | if (bus_connect(DBUS_BUS_SYSTEM, &bus, NULL, &error) < 0) { | |
54 | log_error("Failed to get D-Bus connection: %s", bus_error_message(&error)); | |
55 | goto finish; | |
56 | } | |
57 | ||
58 | if (isolate) | |
59 | mode = "isolate"; | |
60 | else | |
61 | mode = "replace"; | |
62 | ||
a4c24ff7 | 63 | log_info("Running request %s/start/%s", target, mode); |
3d20ed6d | 64 | |
90bb85e1 | 65 | if (!(m = dbus_message_new_method_call("org.freedesktop.systemd1", "/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "StartUnitReplace"))) { |
3d20ed6d LP |
66 | log_error("Could not allocate message."); |
67 | goto finish; | |
68 | } | |
69 | ||
90bb85e1 LP |
70 | /* Start these units only if we can replace base.target with it */ |
71 | ||
3d20ed6d | 72 | if (!dbus_message_append_args(m, |
70f12d37 | 73 | DBUS_TYPE_STRING, &basic_target, |
3d20ed6d LP |
74 | DBUS_TYPE_STRING, &target, |
75 | DBUS_TYPE_STRING, &mode, | |
76 | DBUS_TYPE_INVALID)) { | |
77 | log_error("Could not attach target and flag information to message."); | |
78 | goto finish; | |
79 | } | |
80 | ||
81 | if (!(reply = dbus_connection_send_with_reply_and_block(bus, m, -1, &error))) { | |
ef1de59b | 82 | |
b9114420 | 83 | /* Don't print a warning if we aren't called during |
ef1de59b LP |
84 | * startup */ |
85 | if (!dbus_error_has_name(&error, BUS_ERROR_NO_SUCH_JOB)) | |
86 | log_error("Failed to start unit: %s", bus_error_message(&error)); | |
87 | ||
3d20ed6d LP |
88 | goto finish; |
89 | } | |
90 | ||
91 | finish: | |
92 | if (m) | |
93 | dbus_message_unref(m); | |
94 | ||
95 | if (reply) | |
96 | dbus_message_unref(reply); | |
97 | ||
98 | if (bus) { | |
99 | dbus_connection_flush(bus); | |
100 | dbus_connection_close(bus); | |
101 | dbus_connection_unref(bus); | |
102 | } | |
103 | ||
104 | dbus_error_free(&error); | |
105 | } | |
106 | ||
107 | static int parse_proc_cmdline(void) { | |
108 | char *line, *w, *state; | |
109 | int r; | |
110 | size_t l; | |
111 | ||
03aea2ae | 112 | if (detect_container(NULL) > 0) |
2fc97846 LP |
113 | return 0; |
114 | ||
3d20ed6d LP |
115 | if ((r = read_one_line_file("/proc/cmdline", &line)) < 0) { |
116 | log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r)); | |
117 | return 0; | |
118 | } | |
119 | ||
120 | FOREACH_WORD_QUOTED(w, l, line, state) { | |
121 | ||
122 | if (strneq(w, "fsck.mode=auto", l)) | |
123 | arg_force = arg_skip = false; | |
124 | else if (strneq(w, "fsck.mode=force", l)) | |
125 | arg_force = true; | |
126 | else if (strneq(w, "fsck.mode=skip", l)) | |
127 | arg_skip = true; | |
128 | else if (startswith(w, "fsck.mode")) | |
129 | log_warning("Invalid fsck.mode= parameter. Ignoring."); | |
6fdae8a6 | 130 | #if defined(TARGET_FEDORA) || defined(TARGET_MANDRIVA) || defined(TARGET_MAGEIA) |
3d20ed6d LP |
131 | else if (strneq(w, "fastboot", l)) |
132 | arg_skip = true; | |
133 | else if (strneq(w, "forcefsck", l)) | |
134 | arg_force = true; | |
135 | #endif | |
136 | } | |
137 | ||
138 | free(line); | |
139 | return 0; | |
140 | } | |
141 | ||
142 | static void test_files(void) { | |
143 | if (access("/fastboot", F_OK) >= 0) | |
144 | arg_skip = true; | |
145 | ||
146 | if (access("/forcefsck", F_OK) >= 0) | |
147 | arg_force = true; | |
27d340c7 LP |
148 | |
149 | if (access("/run/systemd/show-status", F_OK) >= 0 || plymouth_running()) | |
150 | arg_show_progress = true; | |
151 | } | |
152 | ||
153 | static double percent(int pass, unsigned long cur, unsigned long max) { | |
154 | /* Values stolen from e2fsck */ | |
155 | ||
156 | static const int pass_table[] = { | |
157 | 0, 70, 90, 92, 95, 100 | |
158 | }; | |
159 | ||
160 | if (pass <= 0) | |
161 | return 0.0; | |
162 | ||
163 | if ((unsigned) pass >= ELEMENTSOF(pass_table) || max == 0) | |
164 | return 100.0; | |
165 | ||
166 | return (double) pass_table[pass-1] + | |
167 | ((double) pass_table[pass] - (double) pass_table[pass-1]) * | |
168 | (double) cur / (double) max; | |
169 | } | |
170 | ||
171 | static int process_progress(int fd) { | |
172 | FILE *f, *console; | |
173 | usec_t last = 0; | |
174 | bool locked = false; | |
175 | int clear = 0; | |
176 | ||
177 | f = fdopen(fd, "r"); | |
178 | if (!f) { | |
179 | close_nointr_nofail(fd); | |
180 | return -errno; | |
181 | } | |
182 | ||
183 | console = fopen("/dev/console", "w"); | |
184 | if (!console) { | |
185 | fclose(f); | |
186 | return -ENOMEM; | |
187 | } | |
188 | ||
189 | while (!feof(f)) { | |
190 | int pass, m; | |
191 | unsigned long cur, max; | |
192 | char *device; | |
193 | double p; | |
194 | usec_t t; | |
195 | ||
196 | if (fscanf(f, "%i %lu %lu %ms", &pass, &cur, &max, &device) != 4) | |
197 | break; | |
198 | ||
199 | /* Only show one progress counter at max */ | |
200 | if (!locked) { | |
201 | if (flock(fileno(console), LOCK_EX|LOCK_NB) < 0) { | |
202 | free(device); | |
203 | continue; | |
204 | } | |
205 | ||
206 | locked = true; | |
207 | } | |
208 | ||
209 | /* Only update once every 50ms */ | |
210 | t = now(CLOCK_MONOTONIC); | |
211 | if (last + 50 * USEC_PER_MSEC > t) { | |
212 | free(device); | |
213 | continue; | |
214 | } | |
215 | ||
216 | last = t; | |
217 | ||
218 | p = percent(pass, cur, max); | |
219 | fprintf(console, "\r%s: fsck %3.1f%% complete...\r%n", device, p, &m); | |
220 | fflush(console); | |
221 | ||
222 | free(device); | |
223 | ||
224 | if (m > clear) | |
225 | clear = m; | |
226 | } | |
227 | ||
228 | if (clear > 0) { | |
229 | unsigned j; | |
230 | ||
231 | fputc('\r', console); | |
232 | for (j = 0; j < (unsigned) clear; j++) | |
233 | fputc(' ', console); | |
234 | fputc('\r', console); | |
235 | fflush(console); | |
236 | } | |
237 | ||
238 | fclose(f); | |
239 | fclose(console); | |
240 | return 0; | |
3d20ed6d LP |
241 | } |
242 | ||
243 | int main(int argc, char *argv[]) { | |
27d340c7 | 244 | const char *cmdline[9]; |
3d20ed6d LP |
245 | int i = 0, r = EXIT_FAILURE, q; |
246 | pid_t pid; | |
247 | siginfo_t status; | |
a9e1f5ec LP |
248 | struct udev *udev = NULL; |
249 | struct udev_device *udev_device = NULL; | |
250 | const char *device; | |
dc8e15c2 | 251 | bool root_directory; |
27d340c7 LP |
252 | int progress_pipe[2] = { -1, -1 }; |
253 | char dash_c[2+10+1]; | |
3d20ed6d | 254 | |
a9e1f5ec LP |
255 | if (argc > 2) { |
256 | log_error("This program expects one or no arguments."); | |
3d20ed6d LP |
257 | return EXIT_FAILURE; |
258 | } | |
259 | ||
27d340c7 | 260 | log_set_target(LOG_TARGET_AUTO); |
3d20ed6d LP |
261 | log_parse_environment(); |
262 | log_open(); | |
263 | ||
4c12626c LP |
264 | umask(0022); |
265 | ||
3d20ed6d LP |
266 | parse_proc_cmdline(); |
267 | test_files(); | |
268 | ||
a9e1f5ec LP |
269 | if (!arg_force && arg_skip) |
270 | return 0; | |
271 | ||
dc8e15c2 | 272 | if (argc > 1) { |
a9e1f5ec | 273 | device = argv[1]; |
dc8e15c2 LP |
274 | root_directory = false; |
275 | } else { | |
a9e1f5ec | 276 | struct stat st; |
a84f5192 | 277 | struct timespec times[2]; |
3d20ed6d | 278 | |
a9e1f5ec LP |
279 | /* Find root device */ |
280 | ||
281 | if (stat("/", &st) < 0) { | |
282 | log_error("Failed to stat() the root directory: %m"); | |
283 | goto finish; | |
284 | } | |
285 | ||
286 | /* Virtual root devices don't need an fsck */ | |
287 | if (major(st.st_dev) == 0) | |
3d20ed6d | 288 | return 0; |
a9e1f5ec | 289 | |
a84f5192 KS |
290 | /* check if we are already writable */ |
291 | times[0] = st.st_atim; | |
292 | times[1] = st.st_mtim; | |
293 | if (utimensat(AT_FDCWD, "/", times, 0) == 0) { | |
cf1a1055 | 294 | log_info("Root directory is writable, skipping check."); |
0f6aaf1b | 295 | return 0; |
a84f5192 KS |
296 | } |
297 | ||
a9e1f5ec LP |
298 | if (!(udev = udev_new())) { |
299 | log_error("Out of memory"); | |
300 | goto finish; | |
301 | } | |
302 | ||
303 | if (!(udev_device = udev_device_new_from_devnum(udev, 'b', st.st_dev))) { | |
304 | log_error("Failed to detect root device."); | |
305 | goto finish; | |
306 | } | |
307 | ||
308 | if (!(device = udev_device_get_devnode(udev_device))) { | |
309 | log_error("Failed to detect device node of root directory."); | |
310 | goto finish; | |
3e33a44a | 311 | } |
dc8e15c2 LP |
312 | |
313 | root_directory = true; | |
3d20ed6d LP |
314 | } |
315 | ||
27d340c7 LP |
316 | if (arg_show_progress) |
317 | if (pipe(progress_pipe) < 0) { | |
318 | log_error("pipe(): %m"); | |
319 | goto finish; | |
320 | } | |
321 | ||
3d20ed6d LP |
322 | cmdline[i++] = "/sbin/fsck"; |
323 | cmdline[i++] = "-a"; | |
324 | cmdline[i++] = "-T"; | |
e9ecea88 | 325 | cmdline[i++] = "-l"; |
dc8e15c2 LP |
326 | |
327 | if (!root_directory) | |
328 | cmdline[i++] = "-M"; | |
3d20ed6d LP |
329 | |
330 | if (arg_force) | |
331 | cmdline[i++] = "-f"; | |
332 | ||
27d340c7 LP |
333 | if (progress_pipe[1] >= 0) { |
334 | snprintf(dash_c, sizeof(dash_c), "-C%i", progress_pipe[1]); | |
335 | char_array_0(dash_c); | |
336 | cmdline[i++] = dash_c; | |
337 | } | |
338 | ||
a9e1f5ec | 339 | cmdline[i++] = device; |
3d20ed6d LP |
340 | cmdline[i++] = NULL; |
341 | ||
27d340c7 LP |
342 | pid = fork(); |
343 | if (pid < 0) { | |
3d20ed6d LP |
344 | log_error("fork(): %m"); |
345 | goto finish; | |
346 | } else if (pid == 0) { | |
347 | /* Child */ | |
27d340c7 LP |
348 | if (progress_pipe[0] >= 0) |
349 | close_nointr_nofail(progress_pipe[0]); | |
3d20ed6d LP |
350 | execv(cmdline[0], (char**) cmdline); |
351 | _exit(8); /* Operational error */ | |
352 | } | |
353 | ||
27d340c7 LP |
354 | if (progress_pipe[1] >= 0) { |
355 | close_nointr_nofail(progress_pipe[1]); | |
356 | progress_pipe[1] = -1; | |
357 | } | |
358 | ||
359 | if (progress_pipe[0] >= 0) { | |
360 | process_progress(progress_pipe[0]); | |
361 | progress_pipe[0] = -1; | |
362 | } | |
363 | ||
364 | q = wait_for_terminate(pid, &status); | |
365 | if (q < 0) { | |
3d20ed6d LP |
366 | log_error("waitid(): %s", strerror(-q)); |
367 | goto finish; | |
368 | } | |
369 | ||
a9e1f5ec | 370 | if (status.si_code != CLD_EXITED || (status.si_status & ~1)) { |
3d20ed6d | 371 | |
a9e1f5ec LP |
372 | if (status.si_code == CLD_KILLED || status.si_code == CLD_DUMPED) |
373 | log_error("fsck terminated by signal %s.", signal_to_string(status.si_status)); | |
374 | else if (status.si_code == CLD_EXITED) | |
375 | log_error("fsck failed with error code %i.", status.si_status); | |
376 | else | |
377 | log_error("fsck failed due to unknown reason."); | |
3d20ed6d | 378 | |
a4c24ff7 | 379 | if (status.si_code == CLD_EXITED && (status.si_status & 2) && root_directory) |
3d20ed6d LP |
380 | /* System should be rebooted. */ |
381 | start_target(SPECIAL_REBOOT_TARGET, false); | |
a4c24ff7 | 382 | else if (status.si_code == CLD_EXITED && (status.si_status & 6)) |
3d20ed6d LP |
383 | /* Some other problem */ |
384 | start_target(SPECIAL_EMERGENCY_TARGET, true); | |
a4c24ff7 LP |
385 | else { |
386 | r = EXIT_SUCCESS; | |
387 | log_warning("Ignoring error."); | |
388 | } | |
3d20ed6d LP |
389 | |
390 | } else | |
391 | r = EXIT_SUCCESS; | |
392 | ||
a9e1f5ec | 393 | if (status.si_code == CLD_EXITED && (status.si_status & 1)) |
2b583ce6 | 394 | touch("/run/systemd/quotacheck"); |
3d20ed6d LP |
395 | |
396 | finish: | |
a9e1f5ec LP |
397 | if (udev_device) |
398 | udev_device_unref(udev_device); | |
399 | ||
400 | if (udev) | |
401 | udev_unref(udev); | |
402 | ||
27d340c7 LP |
403 | close_pipe(progress_pipe); |
404 | ||
3d20ed6d LP |
405 | return r; |
406 | } |