1*4882a593SmuzhiyunFrom 1643107aa592271413603601161b47681287065e Mon Sep 17 00:00:00 2001 2*4882a593SmuzhiyunFrom: Jeffy Chen <jeffy.chen@rock-chips.com> 3*4882a593SmuzhiyunDate: Tue, 18 Sep 2018 11:32:04 +0800 4*4882a593SmuzhiyunSubject: [PATCH 5/8] halt: Support rebooting with arg 5*4882a593Smuzhiyun 6*4882a593SmuzhiyunSupport passing reboot arg(e.g. loader, fastboot, etc.) to kernel. 7*4882a593Smuzhiyun 8*4882a593SmuzhiyunSigned-off-by: Jeffy Chen <jeffy.chen@rock-chips.com> 9*4882a593Smuzhiyun--- 10*4882a593Smuzhiyun init/halt.c | 49 +++++++++++++++++++++++++++++++++++++++++++++++++ 11*4882a593Smuzhiyun 1 file changed, 49 insertions(+) 12*4882a593Smuzhiyun 13*4882a593Smuzhiyundiff --git a/init/halt.c b/init/halt.c 14*4882a593Smuzhiyunindex fe3cb9e..2e8a0a5 100644 15*4882a593Smuzhiyun--- a/init/halt.c 16*4882a593Smuzhiyun+++ b/init/halt.c 17*4882a593Smuzhiyun@@ -93,6 +93,8 @@ 18*4882a593Smuzhiyun 19*4882a593Smuzhiyun #include "libbb.h" 20*4882a593Smuzhiyun #include "reboot.h" 21*4882a593Smuzhiyun+#include <linux/reboot.h> 22*4882a593Smuzhiyun+#include <sys/syscall.h> 23*4882a593Smuzhiyun 24*4882a593Smuzhiyun #if ENABLE_FEATURE_WTMP 25*4882a593Smuzhiyun #include <sys/utsname.h> 26*4882a593Smuzhiyun@@ -161,6 +163,48 @@ static int init_was_not_there(void) 27*4882a593Smuzhiyun # define init_was_not_there() 0 28*4882a593Smuzhiyun #endif 29*4882a593Smuzhiyun 30*4882a593Smuzhiyun+static volatile int caught_sigterm = FALSE; 31*4882a593Smuzhiyun+static void signal_handler(int sig) 32*4882a593Smuzhiyun+{ 33*4882a593Smuzhiyun+ bb_error_msg("Caught signal %d", sig); 34*4882a593Smuzhiyun+ 35*4882a593Smuzhiyun+ if (sig == SIGTERM) 36*4882a593Smuzhiyun+ caught_sigterm = TRUE; 37*4882a593Smuzhiyun+} 38*4882a593Smuzhiyun+ 39*4882a593Smuzhiyun+static int reboot_with_arg(const char *arg) 40*4882a593Smuzhiyun+{ 41*4882a593Smuzhiyun+ struct sigaction sa; 42*4882a593Smuzhiyun+ int pid; 43*4882a593Smuzhiyun+ 44*4882a593Smuzhiyun+ /* Fork new thread to handle reboot */ 45*4882a593Smuzhiyun+ if ((pid = fork())) 46*4882a593Smuzhiyun+ return pid < 0 ? pid : 0; 47*4882a593Smuzhiyun+ 48*4882a593Smuzhiyun+ /* Handle signal and reboot in child thread */ 49*4882a593Smuzhiyun+ sigemptyset(&sa.sa_mask); 50*4882a593Smuzhiyun+ sa.sa_flags = 0; 51*4882a593Smuzhiyun+ sa.sa_handler = signal_handler; 52*4882a593Smuzhiyun+ sigaction_set(SIGTERM, &sa); 53*4882a593Smuzhiyun+ 54*4882a593Smuzhiyun+ bb_error_msg("Waiting for SIGTERM"); 55*4882a593Smuzhiyun+ 56*4882a593Smuzhiyun+ /* The init will send SIGTERM to us after SHUTDOWN actions */ 57*4882a593Smuzhiyun+ while (!caught_sigterm) 58*4882a593Smuzhiyun+ usleep(50000); 59*4882a593Smuzhiyun+ 60*4882a593Smuzhiyun+ bb_error_msg("Ready to reboot"); 61*4882a593Smuzhiyun+ 62*4882a593Smuzhiyun+ /* Wait 200ms for other processes to exit */ 63*4882a593Smuzhiyun+ usleep(200000); 64*4882a593Smuzhiyun+ sync(); 65*4882a593Smuzhiyun+ 66*4882a593Smuzhiyun+ bb_error_msg("Rebooting with arg(%s)", arg); 67*4882a593Smuzhiyun+ return syscall(__NR_reboot, LINUX_REBOOT_MAGIC1, 68*4882a593Smuzhiyun+ LINUX_REBOOT_MAGIC2, 69*4882a593Smuzhiyun+ LINUX_REBOOT_CMD_RESTART2, arg); 70*4882a593Smuzhiyun+} 71*4882a593Smuzhiyun+ 72*4882a593Smuzhiyun int halt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; 73*4882a593Smuzhiyun int halt_main(int argc UNUSED_PARAM, char **argv) 74*4882a593Smuzhiyun { 75*4882a593Smuzhiyun@@ -239,6 +283,11 @@ int halt_main(int argc UNUSED_PARAM, char **argv) 76*4882a593Smuzhiyun CONFIG_TELINIT_PATH); 77*4882a593Smuzhiyun } 78*4882a593Smuzhiyun } 79*4882a593Smuzhiyun+ 80*4882a593Smuzhiyun+ /* Handle rebooting with arg */ 81*4882a593Smuzhiyun+ if (signals[which] == SIGTERM && argc > 1 && argv[1][0] != '-') 82*4882a593Smuzhiyun+ rc = reboot_with_arg(argv[1]); 83*4882a593Smuzhiyun+ 84*4882a593Smuzhiyun } else { 85*4882a593Smuzhiyun rc = reboot(magic[which]); 86*4882a593Smuzhiyun } 87*4882a593Smuzhiyun-- 88*4882a593Smuzhiyun2.20.1 89*4882a593Smuzhiyun 90