Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
Eric Andersen | c499601 | 1999-10-20 22:08:37 +0000 | [diff] [blame] | 2 | /* |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 3 | * Poweroff reboot and halt, oh my. |
Eric Andersen | c499601 | 1999-10-20 22:08:37 +0000 | [diff] [blame] | 4 | * |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 5 | * Copyright 2006 by Rob Landley <rob@landley.net> |
Eric Andersen | c499601 | 1999-10-20 22:08:37 +0000 | [diff] [blame] | 6 | * |
Rob Landley | e9a7a62 | 2006-09-22 02:52:41 +0000 | [diff] [blame] | 7 | * Licensed under GPL version 2, see file LICENSE in this tarball for details. |
Eric Andersen | c499601 | 1999-10-20 22:08:37 +0000 | [diff] [blame] | 8 | */ |
| 9 | |
Denis Vlasenko | b6adbf1 | 2007-05-26 19:00:18 +0000 | [diff] [blame] | 10 | #include "libbb.h" |
Eric Andersen | 0246222 | 2003-07-22 09:41:39 +0000 | [diff] [blame] | 11 | #include <sys/reboot.h> |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 12 | |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 13 | #if ENABLE_FEATURE_WTMP |
| 14 | #include <sys/utsname.h> |
| 15 | #include <utmp.h> |
| 16 | #endif |
| 17 | |
Denis Vlasenko | 9b49a5e | 2007-10-11 10:05:36 +0000 | [diff] [blame] | 18 | int halt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 19 | int halt_main(int argc UNUSED_PARAM, char **argv) |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 20 | { |
Bernhard Reutner-Fischer | d765ee5 | 2006-05-26 20:34:02 +0000 | [diff] [blame] | 21 | static const int magic[] = { |
| 22 | #ifdef RB_HALT_SYSTEM |
| 23 | RB_HALT_SYSTEM, |
| 24 | #elif defined RB_HALT |
| 25 | RB_HALT, |
| 26 | #endif |
| 27 | #ifdef RB_POWER_OFF |
| 28 | RB_POWER_OFF, |
| 29 | #elif defined RB_POWERDOWN |
| 30 | RB_POWERDOWN, |
| 31 | #endif |
| 32 | RB_AUTOBOOT |
| 33 | }; |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 34 | static const smallint signals[] = { SIGUSR1, SIGUSR2, SIGTERM }; |
Rob Landley | 0a7c8ef | 2006-02-22 17:01:00 +0000 | [diff] [blame] | 35 | |
Denis Vlasenko | 1d42665 | 2008-03-17 09:09:09 +0000 | [diff] [blame] | 36 | int delay = 0; |
Denis Vlasenko | 1e28f61 | 2008-08-03 18:43:45 +0000 | [diff] [blame] | 37 | int which, flags, rc; |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 38 | #if ENABLE_FEATURE_WTMP |
| 39 | struct utmp utmp; |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 40 | struct utsname uts; |
| 41 | #endif |
Eric Andersen | 0246222 | 2003-07-22 09:41:39 +0000 | [diff] [blame] | 42 | |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 43 | /* Figure out which applet we're running */ |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 44 | for (which = 0; "hpr"[which] != *applet_name; which++) |
| 45 | continue; |
Eric Andersen | 0246222 | 2003-07-22 09:41:39 +0000 | [diff] [blame] | 46 | |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 47 | /* Parse and handle arguments */ |
Denis Vlasenko | 1d42665 | 2008-03-17 09:09:09 +0000 | [diff] [blame] | 48 | opt_complementary = "d+"; /* -d N */ |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 49 | flags = getopt32(argv, "d:nfw", &delay); |
Denis Vlasenko | 1d42665 | 2008-03-17 09:09:09 +0000 | [diff] [blame] | 50 | |
| 51 | sleep(delay); |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 52 | |
| 53 | #if ENABLE_FEATURE_WTMP |
| 54 | if (access(bb_path_wtmp_file, R_OK|W_OK) == -1) { |
| 55 | close(creat(bb_path_wtmp_file, 0664)); |
| 56 | } |
| 57 | memset(&utmp, 0, sizeof(utmp)); |
Bernhard Reutner-Fischer | 62d8503 | 2008-06-01 10:10:22 +0000 | [diff] [blame] | 58 | utmp.ut_tv.tv_sec = time(NULL); |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 59 | safe_strncpy(utmp.ut_user, "shutdown", UT_NAMESIZE); |
| 60 | utmp.ut_type = RUN_LVL; |
| 61 | safe_strncpy(utmp.ut_id, "~~", sizeof(utmp.ut_id)); |
| 62 | safe_strncpy(utmp.ut_line, "~~", UT_LINESIZE); |
| 63 | if (uname(&uts) == 0) |
| 64 | safe_strncpy(utmp.ut_host, uts.release, sizeof(utmp.ut_host)); |
| 65 | updwtmp(bb_path_wtmp_file, &utmp); |
| 66 | #endif /* !ENABLE_FEATURE_WTMP */ |
| 67 | |
| 68 | if (flags & 8) /* -w */ |
Bernhard Reutner-Fischer | ef9876a | 2008-07-21 11:30:51 +0000 | [diff] [blame] | 69 | return EXIT_SUCCESS; |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 70 | if (!(flags & 2)) /* no -n */ |
| 71 | sync(); |
Bernhard Reutner-Fischer | e15d757 | 2006-06-02 20:56:16 +0000 | [diff] [blame] | 72 | |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 73 | /* Perform action. */ |
Denis Vlasenko | 1e28f61 | 2008-08-03 18:43:45 +0000 | [diff] [blame] | 74 | rc = 1; |
| 75 | if (!(flags & 4)) { /* no -f */ |
| 76 | //TODO: I tend to think that signalling linuxrc is wrong |
| 77 | // pity original author didn't comment on it... |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 78 | if (ENABLE_FEATURE_INITRD) { |
Denis Vlasenko | 35fb512 | 2006-11-01 09:16:49 +0000 | [diff] [blame] | 79 | pid_t *pidlist = find_pid_by_name("linuxrc"); |
| 80 | if (pidlist[0] > 0) |
| 81 | rc = kill(pidlist[0], signals[which]); |
| 82 | if (ENABLE_FEATURE_CLEAN_UP) |
| 83 | free(pidlist); |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 84 | } |
Denis Vlasenko | 35fb512 | 2006-11-01 09:16:49 +0000 | [diff] [blame] | 85 | if (rc) |
| 86 | rc = kill(1, signals[which]); |
| 87 | } else |
| 88 | rc = reboot(magic[which]); |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 89 | |
Denis Vlasenko | 35fb512 | 2006-11-01 09:16:49 +0000 | [diff] [blame] | 90 | if (rc) |
| 91 | bb_error_msg("no"); |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 92 | return rc; |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 93 | } |