Initial commit

Signed-off-by: David Oberhollenzer <david.oberhollenzer@tele2.at>

2 files changed, 152 insertions, 0 deletions

diff --git a/cmd/Makemodule.am b/cmd/Makemodule.am
new file mode 100644
index 0000000..d26b93b
--- /dev/null
+++ b/cmd/Makemodule.am
@@ -0,0 +1,13 @@
+shutdown_SOURCES = cmd/shutdown.c
+shutdown_CPPFLAGS = $(AM_CPPFLAGS) -DPROGNAME=shutdown
+shutdown_CFLAGS = $(AM_CFLAGS)
+shutdown_LDFLAGS = $(AM_LDFLAGS)
+shutdown_LDADD = libinit.a
+
+reboot_SOURCES = cmd/shutdown.c
+reboot_CPPFLAGS = $(AM_CPPFLAGS) -DPROGNAME=reboot
+reboot_CFLAGS = $(AM_CFLAGS)
+reboot_LDFLAGS = $(AM_LDFLAGS)
+reboot_LDADD = libinit.a
+
+sbin_PROGRAMS += reboot shutdown
diff --git a/cmd/shutdown.c b/cmd/shutdown.c
new file mode 100644
index 0000000..16e0c8e
--- /dev/null
+++ b/cmd/shutdown.c
@@ -0,0 +1,139 @@
+#include <getopt.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <stdio.h>
+#include <errno.h>
+
+#include <sys/reboot.h>
+#include <linux/reboot.h>
+
+#include "telinit.h"
+#include "util.h"
+
+#define STRINIFY(x) #x
+#define STRINIFY_VALUE(x) STRINIFY(x)
+#define PROGRAM_NAME STRINIFY_VALUE(PROGNAME)
+
+#define FL_FORCE 0x01
+#define FL_NOSYNC 0x02
+
+static const struct option options[] = {
+	{ "help", no_argument, NULL, 'h' },
+	{ "version", no_argument, NULL, 'V' },
+	{ "poweroff", no_argument, NULL, 'p' },
+	{ "reboot", no_argument, NULL, 'r' },
+	{ "force", no_argument, NULL, 'f' },
+	{ "no-sync", no_argument, NULL, 'n' },
+	{ NULL, 0, NULL, 0 },
+};
+
+static const char *shortopt = "hVprfn";
+
+static const char *defact_str = "power-off";
+static int defact = TI_SHUTDOWN;
+
+static NORETURN void usage(int status)
+{
+	fprintf(status == EXIT_SUCCESS ? stdout : stderr,
+"%s [OPTIONS...]\n\n"
+"Perform a system shutdown or reboot.\n\n"
+"   -h, --help      Display this help text and exit.\n"
+"   -V, --version   Display version information and exit.\n"
+"   -p, --poweroff  Power-off the machine.\n"
+"   -r, --reboot    Reboot the machine.\n"
+"   -f, --force     Force immediate power-off or reboot. Do not contact the\n"
+"                   init system.\n"
+"   -n, --no-sync   Don't sync storage media before power-off or reboot.\n\n"
+"If no option is specified, the default action is %s.\n",
+	PROGRAM_NAME, defact_str);
+	exit(status);
+}
+
+static NORETURN void version(void)
+{
+	fputs(
+PROGRAM_NAME " (Pygos init) " PACKAGE_VERSION "\n"
+"Copyright (C) 2018 David Oberhollenzer\n"
+"License GPLv3+: GNU GPL version 3 or later <http://gnu.org/licenses/gpl.html>.\n"
+"This is free software: you are free to change and redistribute it.\n"
+"There is NO WARRANTY, to the extent permitted by law.\n",
+	stdout);
+
+	exit(EXIT_SUCCESS);
+}
+
+int main(int argc, char **argv)
+{
+	int c, fd, flags = 0;
+	ti_msg_t msg;
+	ssize_t ret;
+
+	if (!strcmp(PROGRAM_NAME, "reboot")) {
+		defact_str = "reboot";
+		defact = TI_REBOOT;
+	}
+
+	while (1) {
+		c = getopt_long(argc, argv, shortopt, options, NULL);
+		if (c == -1)
+			break;
+
+		switch (c) {
+		case 'f':
+			flags |= FL_FORCE;
+			break;
+		case 'n':
+			flags |= FL_NOSYNC;
+			break;
+		case 'p':
+			defact = TI_SHUTDOWN;
+			break;
+		case 'r':
+			defact = TI_REBOOT;
+			break;
+		case 'V':
+			version();
+		case 'h':
+			usage(EXIT_SUCCESS);
+		default:
+			exit(EXIT_FAILURE);
+		}
+	}
+
+	if (flags & FL_FORCE) {
+		if (!(flags & FL_NOSYNC))
+			sync();
+
+		switch (defact) {
+		case TI_REBOOT:
+			reboot(RB_AUTOBOOT);
+			break;
+		case TI_SHUTDOWN:
+			reboot(RB_POWER_OFF);
+			break;
+		}
+
+		perror("reboot system call");
+		return EXIT_FAILURE;
+	}
+
+	fd = opensock();
+	if (fd < 0)
+		return EXIT_FAILURE;
+
+	msg.type = defact;
+retry:
+	ret = write(fd, &msg, sizeof(msg));
+
+	if (ret < 0) {
+		if (errno == EINTR)
+			goto retry;
+		perror("write on init socket");
+		close(fd);
+		return EXIT_FAILURE;
+	}
+
+	close(fd);
+	return EXIT_SUCCESS;
+}