commit 66b0dc44402d7708a2b1e37368695f6c71f357e8
parent 846d6fab1b58572eb32f6f944c55c312704b2ab0
Author: leitner <leitner>
Date: Fri, 9 May 2003 15:49:27 +0000
add shutdown
Diffstat:
| M | Makefile | | | 9 | +++++++-- |
| A | shutdown.c | | | 252 | +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ |
2 files changed, 259 insertions(+), 2 deletions(-)
diff --git a/Makefile b/Makefile
@@ -1,4 +1,4 @@
-all: minit msvc pidfilehack hard-reboot write_proc killall5
+all: minit msvc pidfilehack hard-reboot write_proc killall5 shutdown
#CFLAGS=-pipe -march=i386 -fomit-frame-pointer -Os -I../dietlibc/include
DIET=diet
@@ -17,11 +17,15 @@ buffer_flush.o buffer_stubborn.o buffer_putflush.o str_len.o \
str_start.o
$(DIET) $(CROSS)$(CC) $(LDFLAGS) -o msvc $^
+shutdown: shutdown.o split.o openreadclose.o
+ $(DIET) $(CROSS)$(CC) $(LDFLAGS) -o shutdown $^
+
%.o: %.c
$(DIET) $(CROSS)$(CC) $(CFLAGS) -c $^
clean:
- rm -f *.o minit msvc pidfilehack hard-reboot write_proc killall5
+ rm -f *.o minit msvc pidfilehack hard-reboot write_proc killall5 \
+ shutdown
test: test.c
gcc -nostdlib -o $@ $^ -I../dietlibc/include ../dietlibc/start.o ../dietlibc/dietlibc.a
@@ -42,6 +46,7 @@ install-files:
install minit pidfilehack $(DESTDIR)/sbin
install write_proc hard-reboot $(DESTDIR)/sbin
install msvc $(DESTDIR)/bin
+ install -m 4750 shutdown $(DESTDIR)/sbin
test -d $(DESTDIR)/etc/minit || mkdir $(DESTDIR)/etc/minit
install-fifos:
diff --git a/shutdown.c b/shutdown.c
@@ -0,0 +1,252 @@
+/*
+ * Notes:
+ * - uncomment `#define ALLOW_SUID' below if you want users other than
+ * root to be able to shut down the system.
+ * - after compiling, install under /sbin/shutdown with chgrp adm and
+ * chmod 4750 for SUID root, or 0700 for root only
+ * - uncomment `#define USE_MINIT' below if you want to use shutdown
+ * with minit. If defined, shutdown will try to bring down the services
+ * halt (for -h or -o) or reboot (-r) before killing all other processes.
+ * Please make sure that you have a depends in reboot and halt that
+ * will bring down all respawning services. A respawning service during
+ * shutdown might cause you to wait for a fsck during the next boot
+ * - If you do not use minit shutdown will bring your system down similar
+ * to SysV-Inits shutdown with -n
+ *
+ * TODO:
+ * - add a function for wall-messages
+ * - make sure that all drives are _really_ unmounted
+ * - cleanup
+ */
+
+#include <sys/types.h>
+#include <sys/reboot.h>
+#include <sys/stat.h>
+#include <signal.h>
+#include <stdarg.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <utmp.h>
+#include <fcntl.h>
+#include <stdio.h>
+
+#define ALLOW_SUID
+#define USE_MINIT
+
+#ifdef USE_MINIT
+#define MINITROOT "/etc/minit"
+#endif
+
+extern char **environ;
+
+extern int openreadclose(char *fn, char **buf, unsigned long *len);
+extern char **split(char *buf,int c,int *len,int plus,int ofs);
+extern char *optarg;
+
+void sulogin() {
+ char *argv[]={"sulogin",0};
+ execve("/sbin/sulogin",argv,environ);
+ exit(1);
+}
+
+void msg(char *buf) {
+ write(2,buf,strlen(buf));
+}
+
+void wall(char *buf) {
+ msg(buf);
+}
+
+void exec_umount() {
+ char *Argv[]={"umount","-a",0};
+ if (fork() == 0) {
+ execve("/bin/umount", Argv, environ);
+ perror("execve failed");
+ }
+}
+
+#ifdef USE_MINIT
+
+static int infd, outfd;
+static char buf[1500];
+
+int minit_serviceDown(char *service) {
+ char *s=0;
+ unsigned long len;
+ pid_t pid=0;
+
+ if (service < 0) return 0;
+ if (chdir(MINITROOT) || chdir(service)) return -1;
+
+ if (!openreadclose("depends", &s, &len)) {
+ char **deps;
+ int depc, i;
+ deps=split(s, '\n', &depc, 0, 0);
+ for (i=0; i<depc; i++) {
+ if (deps[i][0] == '#') continue;
+ minit_serviceDown(deps[i]);
+ }
+ }
+
+ // get the pid
+ buf[0]='p';
+ strncpy(buf+1, service, 1400);
+ write(infd, buf, strlen(buf));
+ len=read(outfd, buf, 1500);
+ if (len != 0) {
+ buf[len]=0;
+ pid = atoi(buf);
+ }
+
+ if (strcmp("reboot",service) && strcmp("halt",service) && pid != 1) {
+ msg("\t--> "); msg(service);
+ buf[0]='r'; // we want to disable respawning first
+ strncpy(buf+1, service, 1400);
+ buf[1400]=0;
+ write(infd, buf, strlen(buf));
+ read(outfd, buf, 1500);
+ int i=kill(pid, SIGTERM);
+ if (i == 0) msg("\t\tdone\n");
+ else msg("\t\tfailed\n");
+ }
+}
+
+int minit_shutdown(int level) {
+ char* service;
+ msg("Shutting down minit services: \n");
+ infd=open("/etc/minit/in", O_WRONLY);
+ outfd=open("/etc/minit/out", O_RDONLY);
+ if (infd>=0) {
+ while (lockf(infd, F_LOCK, 1)) {
+ msg("no lock");
+ sleep(1);
+ }
+ }
+
+ if (!level)
+ minit_serviceDown("reboot");
+ else
+ minit_serviceDown("halt");
+}
+#endif
+
+void printUsage() {
+ msg("usage: shutdown -[rhosmn] -[t secs]\n"
+ "\t -r: reboot after shutdown\n"
+ "\t -h: halt after shutdown\n"
+ "\t -o: power-off after shutdown\n"
+ "\t -s: single-user console after shutdown\n"
+ "\t -m: only shutdown the minit-part\n"
+ "\t -n: do not shutdown services using minit\n"
+ "\t -t secs: delay between SIGTERM and SIGKILL\n");
+}
+
+main(int argc, char **argv[]) {
+ int c,i;
+ int cfg_downlevel=2;
+ /* 0: reboot
+ * 1: halt
+ * 2: power off
+ */
+ char *cfg_delay = "3";
+ int cfg_downat = 0;
+ int cfg_minitonly = 0;
+ int cfg_sulogin = 0;
+
+ #ifdef ALLOW_SUID
+ setuid(geteuid());
+ #endif
+ if (getuid() != 0) {
+ msg("you are not root, go away!\n");
+ return 1;
+ }
+
+ if (argc<2) {
+ printUsage();
+ return 0;
+ }
+
+ /* parse commandline options */
+ while((c = getopt(argc, argv, "rhosmnt:")) != EOF) {
+ switch(c) {
+ case 'r': /* do we have to reboot... */
+ cfg_downlevel = 0;
+ break;
+ case 'h': /* ...halt.. */
+ cfg_downlevel = 1;
+ break;
+ case 's': /* rescue system */
+ cfg_sulogin = 1;
+ break;
+ case 'm': /* exit after minit down */
+ cfg_minitonly = 1;
+ break;
+ case 'o': /* ..or power off? */
+ cfg_downlevel = 2;
+ break;
+ case 't': /* delay between SIGTERM and SIGKILL */
+ cfg_delay = optarg;
+ break;
+ default:
+ printUsage();
+ return 1;
+ }
+ }
+
+ switch (cfg_downlevel) {
+ case 0:
+ wall("system is going down for reboot NOW\n");
+ break;
+ case 1:
+ wall("system is going down for system halt NOW\n");
+ break;
+ case 2:
+ wall("system is going down for power-off NOW\n");
+ break;
+ }
+
+ /*
+ * catch some signals;
+ * getting killed after killing the controlling terminal wouldn't be
+ * such a great thing...
+ */
+ signal(SIGQUIT, SIG_IGN);
+ signal(SIGCHLD, SIG_IGN);
+ signal(SIGHUP, SIG_IGN);
+ signal(SIGTSTP, SIG_IGN);
+ signal(SIGTTIN, SIG_IGN);
+ signal(SIGTTOU, SIG_IGN);
+
+ // real shutdown? then lets rock..
+ #ifdef USE_MINIT
+ minit_shutdown(cfg_downlevel);
+ if (cfg_minitonly) return 0;
+ #endif
+
+ /* kill all processes still left */
+ msg("sending all processes the TERM signal...\n");
+ kill(-1, SIGTERM);
+ sleep(atoi(cfg_delay));
+
+ msg("sending all processes the KILL signal...\n");
+ kill(-1, SIGKILL);
+
+ if (cfg_sulogin) {
+ sulogin();
+ return 0;
+ }
+
+ /* sync buffers */
+ sync();
+
+ exec_umount();
+
+ /* and finally reboot, halt or power-off the system */
+ if (cfg_downlevel == 0) {
+ reboot(RB_AUTOBOOT);
+ } else if (cfg_downlevel == 1) {
+ reboot(RB_HALT_SYSTEM);
+ } else {
+ reboot(RB_POWER_OFF);
+ }
+}
