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> |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 16 | |
Denis Vlasenko | 9725daa | 2008-09-11 09:54:23 +0000 | [diff] [blame] | 17 | static void write_wtmp(void) |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 18 | { |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 19 | struct utmp utmp; |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 20 | struct utsname uts; |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 21 | if (access(bb_path_wtmp_file, R_OK|W_OK) == -1) { |
| 22 | close(creat(bb_path_wtmp_file, 0664)); |
| 23 | } |
| 24 | memset(&utmp, 0, sizeof(utmp)); |
Bernhard Reutner-Fischer | 62d8503 | 2008-06-01 10:10:22 +0000 | [diff] [blame] | 25 | utmp.ut_tv.tv_sec = time(NULL); |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 26 | safe_strncpy(utmp.ut_user, "shutdown", UT_NAMESIZE); |
| 27 | utmp.ut_type = RUN_LVL; |
| 28 | safe_strncpy(utmp.ut_id, "~~", sizeof(utmp.ut_id)); |
| 29 | safe_strncpy(utmp.ut_line, "~~", UT_LINESIZE); |
| 30 | if (uname(&uts) == 0) |
| 31 | safe_strncpy(utmp.ut_host, uts.release, sizeof(utmp.ut_host)); |
| 32 | updwtmp(bb_path_wtmp_file, &utmp); |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 33 | |
Denis Vlasenko | 9725daa | 2008-09-11 09:54:23 +0000 | [diff] [blame] | 34 | } |
| 35 | #else |
| 36 | #define write_wtmp() ((void)0) |
| 37 | #endif |
| 38 | |
| 39 | #ifndef RB_HALT_SYSTEM |
| 40 | #define RB_HALT_SYSTEM RB_HALT |
| 41 | #endif |
| 42 | |
Denys Vlasenko | 9d57a80 | 2009-10-23 14:24:44 +0200 | [diff] [blame] | 43 | #ifndef RB_POWERDOWN |
| 44 | /* Stop system and switch power off if possible. */ |
| 45 | # define RB_POWERDOWN 0x4321fedc |
Denis Vlasenko | 9725daa | 2008-09-11 09:54:23 +0000 | [diff] [blame] | 46 | #endif |
Denys Vlasenko | 9d57a80 | 2009-10-23 14:24:44 +0200 | [diff] [blame] | 47 | #ifndef RB_POWER_OFF |
| 48 | # define RB_POWER_OFF RB_POWERDOWN |
| 49 | #endif |
| 50 | |
Denis Vlasenko | 9725daa | 2008-09-11 09:54:23 +0000 | [diff] [blame] | 51 | |
| 52 | int halt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
| 53 | int halt_main(int argc UNUSED_PARAM, char **argv) |
| 54 | { |
| 55 | static const int magic[] = { |
| 56 | RB_HALT_SYSTEM, |
| 57 | RB_POWER_OFF, |
Denis Vlasenko | ea7c9b3 | 2008-09-25 10:39:10 +0000 | [diff] [blame] | 58 | RB_AUTOBOOT |
Denis Vlasenko | 9725daa | 2008-09-11 09:54:23 +0000 | [diff] [blame] | 59 | }; |
| 60 | static const smallint signals[] = { SIGUSR1, SIGUSR2, SIGTERM }; |
| 61 | |
| 62 | int delay = 0; |
| 63 | int which, flags, rc; |
| 64 | |
| 65 | /* Figure out which applet we're running */ |
| 66 | for (which = 0; "hpr"[which] != applet_name[0]; which++) |
| 67 | continue; |
| 68 | |
| 69 | /* Parse and handle arguments */ |
| 70 | opt_complementary = "d+"; /* -d N */ |
Denis Vlasenko | e12c902 | 2009-04-12 15:59:35 +0000 | [diff] [blame] | 71 | /* We support -w even if !ENABLE_FEATURE_WTMP, |
| 72 | * in order to not break scripts. |
| 73 | * -i (shut down network interfaces) is ignored. |
| 74 | */ |
| 75 | flags = getopt32(argv, "d:nfwi", &delay); |
Denis Vlasenko | 9725daa | 2008-09-11 09:54:23 +0000 | [diff] [blame] | 76 | |
| 77 | sleep(delay); |
| 78 | |
| 79 | write_wtmp(); |
| 80 | |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 81 | if (flags & 8) /* -w */ |
Bernhard Reutner-Fischer | ef9876a | 2008-07-21 11:30:51 +0000 | [diff] [blame] | 82 | return EXIT_SUCCESS; |
Bernhard Reutner-Fischer | d93179f | 2008-09-01 15:24:52 +0000 | [diff] [blame] | 83 | |
Denis Vlasenko | 680b86a | 2008-01-24 02:28:00 +0000 | [diff] [blame] | 84 | if (!(flags & 2)) /* no -n */ |
| 85 | sync(); |
Bernhard Reutner-Fischer | e15d757 | 2006-06-02 20:56:16 +0000 | [diff] [blame] | 86 | |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 87 | /* Perform action. */ |
Denis Vlasenko | 1e28f61 | 2008-08-03 18:43:45 +0000 | [diff] [blame] | 88 | rc = 1; |
| 89 | if (!(flags & 4)) { /* no -f */ |
| 90 | //TODO: I tend to think that signalling linuxrc is wrong |
| 91 | // pity original author didn't comment on it... |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 92 | if (ENABLE_FEATURE_INITRD) { |
Alexander Shishkin | 97af2ff | 2009-07-27 02:49:35 +0200 | [diff] [blame] | 93 | /* talk to linuxrc */ |
| 94 | /* bbox init/linuxrc assumed */ |
Denis Vlasenko | 35fb512 | 2006-11-01 09:16:49 +0000 | [diff] [blame] | 95 | pid_t *pidlist = find_pid_by_name("linuxrc"); |
| 96 | if (pidlist[0] > 0) |
| 97 | rc = kill(pidlist[0], signals[which]); |
| 98 | if (ENABLE_FEATURE_CLEAN_UP) |
| 99 | free(pidlist); |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 100 | } |
Denis Vlasenko | e12c902 | 2009-04-12 15:59:35 +0000 | [diff] [blame] | 101 | if (rc) { |
Alexander Shishkin | 97af2ff | 2009-07-27 02:49:35 +0200 | [diff] [blame] | 102 | /* talk to init */ |
| 103 | if (!ENABLE_FEATURE_CALL_TELINIT) { |
| 104 | /* bbox init assumed */ |
| 105 | rc = kill(1, signals[which]); |
| 106 | } else { |
| 107 | /* SysV style init assumed */ |
| 108 | /* runlevels: |
| 109 | * 0 == shutdown |
| 110 | * 6 == reboot */ |
| 111 | rc = execlp(CONFIG_TELINIT_PATH, |
| 112 | CONFIG_TELINIT_PATH, |
| 113 | which == 2 ? "6" : "0", |
| 114 | (char *)NULL |
| 115 | ); |
| 116 | } |
Denis Vlasenko | e12c902 | 2009-04-12 15:59:35 +0000 | [diff] [blame] | 117 | } |
| 118 | } else { |
Denis Vlasenko | 35fb512 | 2006-11-01 09:16:49 +0000 | [diff] [blame] | 119 | rc = reboot(magic[which]); |
Denis Vlasenko | e12c902 | 2009-04-12 15:59:35 +0000 | [diff] [blame] | 120 | } |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 121 | |
Denis Vlasenko | 35fb512 | 2006-11-01 09:16:49 +0000 | [diff] [blame] | 122 | if (rc) |
Denis Vlasenko | fb1a23d | 2009-03-07 01:54:24 +0000 | [diff] [blame] | 123 | bb_perror_nomsg_and_die(); |
Rob Landley | 6461291 | 2006-01-30 08:31:37 +0000 | [diff] [blame] | 124 | return rc; |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 125 | } |