aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorWilliam Hubbs <w.d.hubbs@gmail.com>2017-04-12 17:56:30 -0500
committerWilliam Hubbs <w.d.hubbs@gmail.com>2017-04-12 17:56:36 -0500
commit05738bfce120114037d4f02c67ec740813f94b89 (patch)
tree20e6670fb45dcb5bb8ad70d28a301dbd57ab2351 /src
parent6f88ee4ec6f59e545346a7422facc3e5b6adac04 (diff)
init: add re-exec capability
This will allow the re-execution of the init process after upgrading OpenRC.
Diffstat (limited to 'src')
-rw-r--r--src/rc/openrc-init.c22
-rw-r--r--src/rc/openrc-shutdown.c14
2 files changed, 31 insertions, 5 deletions
diff --git a/src/rc/openrc-init.c b/src/rc/openrc-init.c
index fb3347a4..61052806 100644
--- a/src/rc/openrc-init.c
+++ b/src/rc/openrc-init.c
@@ -20,6 +20,7 @@
#include <errno.h>
#include <signal.h>
+#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
@@ -79,6 +80,12 @@ static void init(const char *default_runlevel)
waitpid(pid, NULL, 0);
}
+static void handle_reexec(char *my_name)
+{
+ execl(my_name, my_name, "reexec", NULL);
+ return;
+}
+
static void handle_shutdown(const char *runlevel, int cmd)
{
pid_t pid;
@@ -123,10 +130,11 @@ static void signal_handler(int sig)
int main(int argc, char **argv)
{
- char *default_runlevel = NULL;
+ char *default_runlevel;
char buf[2048];
int count;
FILE *fifo;
+ bool reexec = false;
struct sigaction sa;
if (getpid() != 1)
@@ -134,16 +142,22 @@ int main(int argc, char **argv)
if (argc > 1)
default_runlevel = argv[1];
+ else
+ default_runlevel = NULL;
+
+ if (default_runlevel && strcmp(default_runlevel, "reexec") == 0)
+ reexec = true;
printf("OpenRC init version %s starting\n", VERSION);
- init(default_runlevel);
+ if (! reexec)
+ init(default_runlevel);
memset(&sa, 0, sizeof(sa));
sa.sa_handler = signal_handler;
sigaction(SIGCHLD, &sa, NULL);
sigaction(SIGINT, &sa, NULL);
reboot(RB_DISABLE_CAD);
- if (mkfifo(RC_INIT_FIFO, 0600) == -1)
+ if (mkfifo(RC_INIT_FIFO, 0600) == -1 && errno != EEXIST)
perror("mkfifo");
for (;;) {
@@ -166,6 +180,8 @@ int main(int argc, char **argv)
handle_shutdown("shutdown", RB_POWER_OFF);
else if (strcmp(buf, "reboot") == 0)
handle_shutdown("reboot", RB_AUTOBOOT);
+ else if (strcmp(buf, "reexec") == 0)
+ handle_reexec(argv[0]);
}
return 0;
}
diff --git a/src/rc/openrc-shutdown.c b/src/rc/openrc-shutdown.c
index 978e8a68..8905d354 100644
--- a/src/rc/openrc-shutdown.c
+++ b/src/rc/openrc-shutdown.c
@@ -35,11 +35,12 @@
const char *applet = NULL;
const char *extraopts = NULL;
-const char *getoptstring = "kpr" getoptstring_COMMON;
+const char *getoptstring = "HkpRr" getoptstring_COMMON;
const struct option longopts[] = {
{ "halt", no_argument, NULL, 'H'},
{ "kexec", no_argument, NULL, 'k'},
{ "poweroff", no_argument, NULL, 'p'},
+ { "reexec", no_argument, NULL, 'R'},
{ "reboot", no_argument, NULL, 'r'},
longopts_COMMON
};
@@ -47,11 +48,13 @@ const char * const longopts_help[] = {
"halt the system",
"reboot the system using kexec",
"power off the system",
+ "re-execute init (use after upgrading)",
"reboot the system",
longopts_help_COMMON
};
const char *usagestring = NULL;
-const char *exclusive = "Select one of --halt, --kexec, --poweroff or --reboot";
+const char *exclusive = "Select one of "
+"--halt, --kexec, --poweroff, --reexec or --reboot";
static void send_cmd(const char *cmd)
{
@@ -79,6 +82,7 @@ int main(int argc, char **argv)
bool do_kexec = false;
bool do_poweroff = false;
bool do_reboot = false;
+ bool do_reexec = false;
applet = basename_c(argv[0]);
if (geteuid() != 0)
@@ -99,6 +103,10 @@ if (geteuid() != 0)
do_poweroff = true;
cmd_count++;
break;
+ case 'R':
+ do_reexec = true;
+ cmd_count++;
+ break;
case 'r':
do_reboot = true;
cmd_count++;
@@ -118,5 +126,7 @@ if (geteuid() != 0)
send_cmd("poweroff");
else if (do_reboot)
send_cmd("reboot");
+ else if (do_reexec)
+ send_cmd("reexec");
return 0;
}