From c52f2164502544c17811e47fc3236fe87aa7fc5b Mon Sep 17 00:00:00 2001 From: Roy Marples Date: Fri, 20 Apr 2007 13:12:21 +0000 Subject: Added the rc-abort command, #175106 thanks to Daniel Drake. --- src/rc.c | 175 +++++++++++++++++++++++++++++++++++---------------------------- 1 file changed, 98 insertions(+), 77 deletions(-) (limited to 'src/rc.c') diff --git a/src/rc.c b/src/rc.c index d5482d15..375800af 100644 --- a/src/rc.c +++ b/src/rc.c @@ -56,16 +56,14 @@ extern char **environ; static char *applet = NULL; static char **env = NULL; static char **newenv = NULL; -static char **coldplugged_services; +static char **coldplugged_services = NULL; static char **stop_services = NULL; static char **start_services = NULL; static rc_depinfo_t *deptree = NULL; static char **types = NULL; -static char *mycmd = NULL; -static char *myarg = NULL; static char *tmp = NULL; -struct termios *termios_orig; +struct termios *termios_orig = NULL; static void cleanup (void) { @@ -76,24 +74,13 @@ static void cleanup (void) free (termios_orig); } - if (env) - rc_strlist_free (env); - if (newenv) - rc_strlist_free (newenv); - if (coldplugged_services) - rc_strlist_free (coldplugged_services); - if (stop_services) - rc_strlist_free (stop_services); - if (start_services) - rc_strlist_free (start_services); - if (deptree) - rc_free_deptree (deptree); - if (types) - rc_strlist_free (types); - if (mycmd) - free (mycmd); - if (myarg) - free (myarg); + rc_strlist_free (env); + rc_strlist_free (newenv); + rc_strlist_free (coldplugged_services); + rc_strlist_free (stop_services); + rc_strlist_free (start_services); + rc_free_deptree (deptree); + rc_strlist_free (types); /* Clean runlevel start, stop markers */ if (rc_is_dir (RC_SVCDIR "softscripts.new")) @@ -101,8 +88,7 @@ static void cleanup (void) if (rc_is_dir (RC_SVCDIR "softscripts.old")) rc_rm_dir (RC_SVCDIR "softscripts.old", true); - if (applet) - free (applet); + free (applet); } static int do_e (int argc, char **argv) @@ -335,6 +321,10 @@ static char read_key (bool block) struct termios termios; char c = 0; + /* This locks up rc for some reason! + * Why? it used to work fine... */ + return 0; + if (! isatty (STDIN_FILENO)) return (false); @@ -378,33 +368,54 @@ static void mark_interactive (void) static void sulogin (bool cont) { #ifdef __linux__ + char *e = getenv ("RC_SYS"); + + /* VPS systems cannot do an sulogin */ + if (e && strcmp (e, "VPS") == 0) { + execl ("/sbin/halt", "/sbin/halt", "-f", (char *) NULL); + eerrorx ("%s: unable to exec `/sbin/halt': %s", applet, strerror (errno)); + } + if (cont) { int status = 0; - pid_t pid = fork(); + pid_t pid = vfork(); if (pid == -1) - eerrorx ("%s: fork: %s", applet, strerror (errno)); + eerrorx ("%s: vfork: %s", applet, strerror (errno)); if (pid == 0) { newenv = rc_filter_env (); - mycmd = rc_xstrdup ("/sbin/sulogin"); - myarg = rc_xstrdup (getenv ("CONSOLE")); - execle (mycmd, mycmd, myarg, (char *) NULL, newenv); - eerrorx ("%s: unable to exec `/sbin/sulogin': %s", applet, strerror (errno)); + execl ("/sbin/sulogin", "/sbin/sulogin", + getenv ("CONSOLE"), (char *) NULL, newenv); + eerror ("%s: unable to exec `/sbin/sulogin': %s", applet, + strerror (errno)); + _exit (EXIT_FAILURE); } waitpid (pid, &status, 0); } else { newenv = rc_filter_env (); - mycmd = rc_xstrdup ("/sbin/sulogin"); - myarg = rc_xstrdup (getenv ("CONSOLE")); - execle (mycmd, mycmd, myarg, (char *) NULL, newenv); + execl ("/sbin/sulogin", "/sbin/sulogin", + getenv ("CONSOLE"), (char *) NULL, newenv); eerrorx ("%s: unable to exec `/sbin/sulogin': %s", applet, strerror (errno)); } #else - /* Appease gcc */ exit (cont ? EXIT_FAILURE : EXIT_SUCCESS); #endif } +static void single_user (void) +{ +#ifdef __linux__ + execl ("/sbin/telinit", "/sbin/telinit", "S", (char *) NULL); + eerrorx ("%s: unable to exec `/sbin/telinit': %s", + applet, strerror (errno)); +#else + if (kill (1, SIGTERM) != 0) + eerrorx ("%s: unable to send SIGTERM to init (pid 1): %s", + applet, strerror (errno)); + exit (EXIT_SUCCESS); +#endif +} + static void set_ksoftlevel (const char *runlevel) { FILE *fp; @@ -457,6 +468,18 @@ static void handle_signal (int sig) if (! signame[0]) snprintf (signame, sizeof (signame), "SIGQUIT"); eerrorx ("%s: caught %s, aborting", applet, signame); + case SIGUSR1: + eerror ("rc: Aborting!"); + /* Kill any running services we have started */ + signal (SIGTERM, SIG_IGN); + killpg (getpgrp (), SIGTERM); + + /* If we're in boot runlevel then change into single-user mode */ + if (strcmp (rc_get_runlevel (), RC_LEVEL_BOOT) == 0) + single_user (); + + exit (EXIT_FAILURE); + break; default: eerror ("%s: caught unknown signal %d", applet, sig); @@ -480,6 +503,7 @@ int main (int argc, char **argv) bool interactive = false; int depoptions = RC_DEP_STRICT | RC_DEP_TRACE; char ksoftbuffer [PATH_MAX]; + char pidstr[6]; if (argv[0]) applet = rc_xstrdup (basename (argv[0])); @@ -509,6 +533,19 @@ int main (int argc, char **argv) else if (strcmp (applet, "is_runlevel_stop") == 0) exit (rc_runlevel_stopping () ? 0 : 1); + if (strcmp (applet, "rc-abort") == 0) { + char *p = getenv ("RC_PID"); + pid_t pid = 0; + + if (p && sscanf (p, "%d", &pid) == 1) { + if (kill (pid, SIGUSR1) != 0) + eerrorx ("rc-abort: failed to signal parent %d: %s", + pid, strerror (errno)); + exit (EXIT_SUCCESS); + } + exit (EXIT_FAILURE); + } + if (strcmp (applet, "rc" ) != 0) eerrorx ("%s: unknown applet", applet); @@ -520,10 +557,14 @@ int main (int argc, char **argv) atexit (cleanup); newlevel = argv[0]; + /* Start a new process group */ + setpgrp(); + /* Setup a signal handler */ signal (SIGINT, handle_signal); signal (SIGQUIT, handle_signal); signal (SIGTERM, handle_signal); + signal (SIGUSR1, handle_signal); /* Ensure our environment is pure Also, add our configuration to it */ @@ -563,6 +604,10 @@ int main (int argc, char **argv) /* Enable logging */ setenv ("RC_ELOG", "rc", 1); + /* Export our PID */ + snprintf (pidstr, sizeof (pidstr), "%d", getpid ()); + setenv ("RC_PID", pidstr, 1); + interactive = rc_exists (INTERACTIVE); rc_plugin_load (); @@ -599,14 +644,14 @@ int main (int argc, char **argv) setenv ("RC_SOFTLEVEL", newlevel, 1); rc_plugin_run (rc_hook_runlevel_start_in, newlevel); - if ((pid = fork ()) == -1) - eerrorx ("%s: fork: %s", applet, strerror (errno)); + if ((pid = vfork ()) == -1) + eerrorx ("%s: vfork: %s", applet, strerror (errno)); if (pid == 0) { - mycmd = rc_xstrdup (INITSH); - execl (mycmd, mycmd, (char *) NULL); - eerrorx ("%s: unable to exec `" INITSH "': %s", - applet, strerror (errno)); + execl (INITSH, INITSH, (char *) NULL); + eerror ("%s: unable to exec `" INITSH "': %s", + applet, strerror (errno)); + _exit (EXIT_FAILURE); } do { @@ -695,9 +740,7 @@ int main (int argc, char **argv) eerrorx ("%s: couldn't find a runlevel called `%s'", applet, newlevel); } else { - mycmd = rc_xstrdup ("/sbin/telinit"); - myarg = rc_xstrdup (lvl); - execl (mycmd, mycmd, myarg, (char *) NULL); + execl ("/sbin/telinit", "/sbin/telinit", lvl, (char *) NULL); eerrorx ("%s: unable to exec `/sbin/telinit': %s", applet, strerror (errno)); } @@ -712,11 +755,6 @@ int main (int argc, char **argv) rc reboot */ if (newlevel) { - if (myarg) { - free (myarg); - myarg = NULL; - } - if (strcmp (newlevel, RC_LEVEL_SINGLE) == 0) { if (! RUNLEVEL || (strcmp (RUNLEVEL, "S") != 0 && @@ -724,44 +762,29 @@ int main (int argc, char **argv) { /* Remember the current runlevel for when we come back */ set_ksoftlevel (runlevel); -#ifdef __linux__ - mycmd = rc_xstrdup ("/sbin/telinit"); - myarg = rc_xstrdup ("S"); - execl (mycmd, mycmd, myarg, (char *) NULL); - eerrorx ("%s: unable to exec `/%s': %s", - mycmd, applet, strerror (errno)); -#else - if (kill (1, SIGTERM) != 0) - eerrorx ("%s: unable to send SIGTERM to init (pid 1): %s", - applet, strerror (errno)); - exit (EXIT_SUCCESS); -#endif + single_user (); } } else if (strcmp (newlevel, RC_LEVEL_REBOOT) == 0) { if (! RUNLEVEL || strcmp (RUNLEVEL, "6") != 0) { - mycmd = rc_xstrdup ("/sbin/shutdown"); - myarg = rc_xstrdup ("-r"); - tmp = rc_xstrdup ("now"); - execl (mycmd, mycmd, myarg, tmp, (char *) NULL); - eerrorx ("%s: unable to exec `%s': %s", - mycmd, applet, strerror (errno)); + execl ("/sbin/shutdown", "/sbin/shutdown", "-r", "now", (char *) NULL); + eerrorx ("%s: unable to exec `/sbin/shutdown': %s", + applet, strerror (errno)); } } else if (strcmp (newlevel, RC_LEVEL_SHUTDOWN) == 0) { if (! RUNLEVEL || strcmp (RUNLEVEL, "0") != 0) { - mycmd = rc_xstrdup ("/sbin/shutdown"); -#ifdef __linux__ - myarg = rc_xstrdup ("-h"); + execl ("/sbin/shutdown", "/sbin/shutdown", +#ifdef __linux + "-h", #else - myarg = rc_xstrdup ("-p"); + "-p", #endif - tmp = rc_xstrdup ("now"); - execl (mycmd, mycmd, myarg, tmp, (char *) NULL); - eerrorx ("%s: unable to exec `%s': %s", - mycmd, applet, strerror (errno)); + "now", (char *) NULL); + eerrorx ("%s: unable to exec `/sbin/shutdown': %s", + applet, strerror (errno)); } } } @@ -1060,9 +1083,7 @@ int main (int argc, char **argv) if (strcmp (runlevel, RC_LEVEL_SHUTDOWN) == 0 || strcmp (runlevel, RC_LEVEL_REBOOT) == 0) { - mycmd = rc_xstrdup (HALTSH); - myarg = rc_xstrdup (runlevel); - execl (mycmd, mycmd, myarg, (char *) NULL); + execl (HALTSH, HALTSH, runlevel, (char *) NULL); eerrorx ("%s: unable to exec `%s': %s", applet, HALTSH, strerror (errno)); } -- cgit v1.2.3