X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=kernel%2Fvserver%2Fhelper.c;h=60f570fa152ba5c14270fb23f873aee4277ea646;hb=97bf2856c6014879bd04983a3e9dfcdac1e7fe85;hp=e3ab5814fdb9cd99028bf786168082e2efe9b825;hpb=8e8ece46a861c84343256819eaec77e608ff9217;p=linux-2.6.git

diff --git a/kernel/vserver/helper.c b/kernel/vserver/helper.c
index e3ab5814f..60f570fa1 100644
--- a/kernel/vserver/helper.c
+++ b/kernel/vserver/helper.c
@@ -3,19 +3,19 @@
  *
  *  Virtual Context Support
  *
- *  Copyright (C) 2004-2005  Herbert Pötzl
+ *  Copyright (C) 2004-2007  Herbert Pötzl
  *
  *  V0.01  basic helper
  *
  */
 
-#include <linux/config.h>
 #include <linux/errno.h>
-#include <linux/reboot.h>
 #include <linux/kmod.h>
 #include <linux/sched.h>
+#include <linux/reboot.h>
 #include <linux/vs_context.h>
 #include <linux/vs_network.h>
+#include <linux/vserver/signal.h>
 
 #include <asm/uaccess.h>
 #include <asm/unistd.h>
@@ -24,7 +24,7 @@
 char vshelper_path[255] = "/sbin/vshelper";
 
 
-int do_vshelper(char *name, char *argv[], char *envp[], int sync)
+static int do_vshelper(char *name, char *argv[], char *envp[], int sync)
 {
 	int ret;
 
@@ -34,7 +34,7 @@ int do_vshelper(char *name, char *argv[], char *envp[], int sync)
 			name, argv[1], argv[2],
 			sync?"sync":"async", ret);
 	}
-	vxdprintk(VXD_CBIT(switch, 1),
+	vxdprintk(VXD_CBIT(switch, 4),
 		"%s: (%s %s) returned %s with %d",
 		name, argv[1], argv[2], sync?"sync":"async", ret);
 	return ret;
@@ -52,7 +52,7 @@ int do_vshelper(char *name, char *argv[], char *envp[], int sync)
  *      envp [*] = type-specific parameters
  */
 
-long vs_reboot(unsigned int cmd, void * arg)
+long vs_reboot_helper(struct vx_info *vxi, int cmd, void __user *arg)
 {
 	char id_buf[8], cmd_buf[16];
 	char uid_buf[16], pid_buf[16];
@@ -63,7 +63,11 @@ long vs_reboot(unsigned int cmd, void * arg)
 			"PATH=/sbin:/usr/sbin:/bin:/usr/bin",
 			uid_buf, pid_buf, cmd_buf, 0};
 
-	snprintf(id_buf, sizeof(id_buf)-1, "%d", vx_current_xid());
+	if (vx_info_state(vxi, VXS_HELPER))
+		return -EAGAIN;
+	vxi->vx_state |= VXS_HELPER;
+
+	snprintf(id_buf, sizeof(id_buf)-1, "%d", vxi->vx_id);
 
 	snprintf(cmd_buf, sizeof(cmd_buf)-1, "VS_CMD=%08x", cmd);
 	snprintf(uid_buf, sizeof(uid_buf)-1, "VS_UID=%d", current->uid);
@@ -87,6 +91,7 @@ long vs_reboot(unsigned int cmd, void * arg)
 		break;
 
 	default:
+		vxi->vx_state &= ~VXS_HELPER;
 		return 0;
 	}
 
@@ -95,10 +100,41 @@ long vs_reboot(unsigned int cmd, void * arg)
 #else
 	ret = do_vshelper(vshelper_path, argv, envp, 0);
 #endif
+	vxi->vx_state &= ~VXS_HELPER;
+	__wakeup_vx_info(vxi);
 	return (ret) ? -EPERM : 0;
 }
 
 
+long vs_reboot(unsigned int cmd, void __user * arg)
+{
+	struct vx_info *vxi = current->vx_info;
+	long ret = 0;
+
+	vxdprintk(VXD_CBIT(misc, 5),
+		"vs_reboot(%p[#%d],%d)",
+		vxi, vxi?vxi->vx_id:0, cmd);
+
+	ret = vs_reboot_helper(vxi, cmd, arg);
+	if (ret)
+		return ret;
+
+	vxi->reboot_cmd = cmd;
+	if (vx_info_flags(vxi, VXF_REBOOT_KILL, 0)) {
+		switch (cmd) {
+		case LINUX_REBOOT_CMD_RESTART:
+		case LINUX_REBOOT_CMD_HALT:
+		case LINUX_REBOOT_CMD_POWER_OFF:
+			vx_info_kill(vxi, 0, SIGKILL);
+			vx_info_kill(vxi, 1, SIGKILL);
+		default:
+			break;
+		}
+	}
+	return 0;
+}
+
+
 /*
  *      argv [0] = vshelper_path;
  *      argv [1] = action: "startup", "shutdown"
@@ -114,7 +150,7 @@ long vs_state_change(struct vx_info *vxi, unsigned int cmd)
 	char *envp[] = {"HOME=/", "TERM=linux",
 			"PATH=/sbin:/usr/sbin:/bin:/usr/bin", cmd_buf, 0};
 
-	if (!vx_info_flags(vxi, VXF_STATE_HELPER, 0))
+	if (!vx_info_flags(vxi, VXF_SC_HELPER, 0))
 		return 0;
 
 	snprintf(id_buf, sizeof(id_buf)-1, "%d", vxi->vx_id);
@@ -131,8 +167,7 @@ long vs_state_change(struct vx_info *vxi, unsigned int cmd)
 		return 0;
 	}
 
-	do_vshelper(vshelper_path, argv, envp, 1);
-	return 0;
+	return do_vshelper(vshelper_path, argv, envp, 1);
 }
 
 
@@ -151,7 +186,7 @@ long vs_net_change(struct nx_info *nxi, unsigned int cmd)
 	char *envp[] = {"HOME=/", "TERM=linux",
 			"PATH=/sbin:/usr/sbin:/bin:/usr/bin", cmd_buf, 0};
 
-	if (!nx_info_flags(nxi, NXF_STATE_HELPER, 0))
+	if (!nx_info_flags(nxi, NXF_SC_HELPER, 0))
 		return 0;
 
 	snprintf(id_buf, sizeof(id_buf)-1, "%d", nxi->nx_id);
@@ -168,7 +203,6 @@ long vs_net_change(struct nx_info *nxi, unsigned int cmd)
 		return 0;
 	}
 
-	do_vshelper(vshelper_path, argv, envp, 1);
-	return 0;
+	return do_vshelper(vshelper_path, argv, envp, 1);
 }