It's really more of a kernel check, but worth having.
Signed-off-by: Serge Hallyn <serge.hal...@ubuntu.com>
---
src/tests/Makefile.am | 4 +-
src/tests/reboot.c | 139 ++++++++++++++++++++++++++++++++++++++++++++++++++
2 files changed, 142 insertions(+), 1 deletion(-)
create mode 100644 src/tests/reboot.c
diff --git a/src/tests/Makefile.am b/src/tests/Makefile.am
index 479facc..d9d5da2 100644
--- a/src/tests/Makefile.am
+++ b/src/tests/Makefile.am
@@ -20,6 +20,7 @@ lxc_usernic_test_CFLAGS = -DISTEST
lxc_test_snapshot_SOURCES = snapshot.c
lxc_test_concurrent_SOURCES = concurrent.c
lxc_test_may_control_SOURCES = may_control.c
+lxc_test_reboot_SOURCES = reboot.c
AM_CFLAGS=-I$(top_srcdir)/src \
-DLXCROOTFSMOUNT=\"$(LXCROOTFSMOUNT)\" \
@@ -32,7 +33,8 @@ bin_PROGRAMS = lxc-test-containertests lxc-test-locktests
lxc-test-startone \
lxc-test-destroytest lxc-test-saveconfig lxc-test-createtest \
lxc-test-shutdowntest lxc-test-get_item lxc-test-getkeys
lxc-test-lxcpath \
lxc-test-cgpath lxc-test-clonetest lxc-test-console lxc-usernic-test \
- lxc-test-snapshot lxc-test-concurrent lxc-test-may-control
+ lxc-test-snapshot lxc-test-concurrent lxc-test-may-control \
+ lxc-test-reboot
bin_SCRIPTS = lxc-test-usernic
diff --git a/src/tests/reboot.c b/src/tests/reboot.c
new file mode 100644
index 0000000..82f0d36
--- /dev/null
+++ b/src/tests/reboot.c
@@ -0,0 +1,139 @@
+/*
+ * Copyright © 2012 Serge Hallyn <serge.hal...@ubuntu.com>.
+ * Copyright © 2012 Canonical Ltd.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+#include <alloca.h>
+#include <stdio.h>
+#include <sched.h>
+#include <unistd.h>
+#include <signal.h>
+#include <sys/reboot.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+
+#define _GNU_SOURCE
+#include <sched.h>
+#include <linux/sched.h>
+#include <linux/reboot.h>
+
+int clone(int (*fn)(void *), void *child_stack, int flags, void *arg);
+
+static int do_reboot(void *arg)
+{
+ int *cmd = arg;
+
+ if (reboot(*cmd))
+ printf("failed to reboot(%d): %m\n", *cmd);
+ return 0;
+}
+
+int test_reboot(int cmd, int sig)
+{
+ long stack_size = 4096;
+ void *stack = alloca(stack_size) + stack_size;
+ int status;
+ pid_t ret;
+
+ ret = clone(do_reboot, stack, CLONE_NEWPID | SIGCHLD, &cmd);
+ if (ret < 0) {
+ printf("failed to clone: %m\n");
+ return -1;
+ }
+
+ if (wait(&status) < 0) {
+ printf("unexpected wait error: %m\n");
+ return -1;
+ }
+
+ if (!WIFSIGNALED(status)) {
+ if (sig != -1)
+ printf("child process exited but was not signaled\n");
+ return -1;
+ }
+
+ if (WTERMSIG(status) != sig) {
+ printf("signal termination is not the one expected\n");
+ return -1;
+ }
+
+ return 0;
+}
+
+static int have_reboot_patch(void)
+{
+ FILE *f = fopen("/proc/sys/kernel/ctrl-alt-del", "r");
+ int ret;
+ int v;
+
+ if (!f)
+ return 0;
+
+ ret = fscanf(f, "%d", &v);
+ fclose(f);
+ if (ret != 1)
+ return 0;
+ ret = reboot(v ? LINUX_REBOOT_CMD_CAD_ON : LINUX_REBOOT_CMD_CAD_OFF);
+ if (ret != -1)
+ return 0;
+ return 1;
+}
+
+int main(int argc, char *argv[])
+{
+ int status;
+
+ if (getuid() != 0) {
+ printf("Must run as root.\n");
+ return 1;
+ }
+
+ status = have_reboot_patch();
+ if (status != 0) {
+ printf("Your kernel does not have the container reboot
patch\n");
+ return 1;
+ }
+
+ status = test_reboot(LINUX_REBOOT_CMD_CAD_ON, -1);
+ if (status >= 0) {
+ printf("reboot(LINUX_REBOOT_CMD_CAD_ON) should have failed\n");
+ return 1;
+ }
+ printf("reboot(LINUX_REBOOT_CMD_CAD_ON) has failed as expected\n");
+
+ status = test_reboot(LINUX_REBOOT_CMD_RESTART, SIGHUP);
+ if (status < 0)
+ return 1;
+ printf("reboot(LINUX_REBOOT_CMD_RESTART) succeed\n");
+
+ status = test_reboot(LINUX_REBOOT_CMD_RESTART2, SIGHUP);
+ if (status < 0)
+ return 1;
+ printf("reboot(LINUX_REBOOT_CMD_RESTART2) succeed\n");
+
+ status = test_reboot(LINUX_REBOOT_CMD_HALT, SIGINT);
+ if (status < 0)
+ return 1;
+ printf("reboot(LINUX_REBOOT_CMD_HALT) succeed\n");
+
+ status = test_reboot(LINUX_REBOOT_CMD_POWER_OFF, SIGINT);
+ if (status < 0)
+ return 1;
+ printf("reboot(LINUX_REBOOT_CMD_POWERR_OFF) succeed\n");
+
+ printf("All tests passed\n");
+ return 0;
+}
--
1.8.1.2
------------------------------------------------------------------------------
October Webinars: Code for Performance
Free Intel webinars can help you accelerate application performance.
Explore tips for MPI, OpenMP, advanced profiling, and more. Get the most from
the latest Intel processors and coprocessors. See abstracts and register >
http://pubads.g.doubleclick.net/gampad/clk?id=60134791&iu=/4140/ostg.clktrk
_______________________________________________
Lxc-devel mailing list
Lxc-devel@lists.sourceforge.net
https://lists.sourceforge.net/lists/listinfo/lxc-devel
