--- /dev/null
+// $Id: vspace.c 2693 2008-03-01 00:26:31Z dhozac $ --*- c -*--
+
+// Copyright (C) 2004 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
+// Copyright (C) 2007 Daniel Hokka Zakrisson
+//
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation; version 2 of the License.
+//
+// This program 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 General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+
+
+#ifdef HAVE_CONFIG_H
+# include <config.h>
+#endif
+
+#include "util.h"
+#include <lib_internal/sys_clone.h>
+
+#include <vserver.h>
+
+#include <getopt.h>
+#include <libgen.h>
+#include <errno.h>
+#include <signal.h>
+#include <sched.h>
+
+#define ENSC_WRAPPERS_PREFIX "vspace: "
+#define ENSC_WRAPPERS_UNISTD 1
+#define ENSC_WRAPPERS_VSERVER 1
+#include <wrappers.h>
+
+#define CMD_HELP 0x1000
+#define CMD_VERSION 0x1001
+
+int wrapper_exit_code = 255;
+
+struct option const
+CMDLINE_OPTIONS[] = {
+ { "help", no_argument, 0, CMD_HELP },
+ { "version", no_argument, 0, CMD_VERSION },
+ { "new", no_argument, 0, 'n' },
+ { "enter", required_argument, 0, 'e' },
+ { "set", no_argument, 0, 's' },
+ { "mask", required_argument, 0, 'm' },
+
+ { "mount", no_argument, 0, 'M' },
+ { "fs", no_argument, 0, 'F' },
+ { "ipc", no_argument, 0, 'I' },
+ { "uts", no_argument, 0, 'U' },
+ { "user", no_argument, 0, 'S' },
+ { "pid", no_argument, 0, 'P' },
+ { "net", no_argument, 0, 'N' },
+ {0,0,0,0}
+};
+
+static void
+showHelp(int fd, char const *cmd, int res)
+{
+ WRITE_MSG(fd, "Usage: ");
+ WRITE_STR(fd, cmd);
+ WRITE_MSG(fd,
+ " <operation> <spaces>* [--] [<program> <args>*]\n"
+ "\n"
+ "<operation> can be one of:\n"
+ " --new|-n ... create new spaces and execute <program> there;\n"
+ " <program> is mandatory in this case\n"
+ " --enter|-e <xid> ... enter the spaces of context <xid> and execute\n"
+ " <program> there; <program> is mandatory in this\n"
+ " case\n"
+ " --set|-s ... assign the current spaces to the current context\n"
+ "\n"
+ "<spaces>* specifies the spaces to manipulate.\n"
+ "It can be any combination of:\n"
+ " --mask <mask> ... specify a mask of spaces\n"
+ " --mount ... the mount namespace\n"
+ " --fs ... the fs_struct\n"
+ " --ipc ... the IPC namespace\n"
+ " --uts ... the uts namespace\n"
+ " --user ... the user namespace\n"
+ " --pid ... the pid namespace\n"
+ " --net ... the network namespace\n"
+ "\n"
+ "Please report bugs to " PACKAGE_BUGREPORT "\n");
+
+ exit(res);
+}
+
+static void
+showVersion()
+{
+ WRITE_MSG(1,
+ "vspace " VERSION " -- manages spaces\n"
+ "This program is part of " PACKAGE_STRING "\n\n"
+ "Copyright (C) 2004 Enrico Scholz\n"
+ "Copyright (C) 2007 Daniel Hokka Zakrisson\n"
+ VERSION_COPYRIGHT_DISCLAIMER);
+ exit(0);
+}
+
+static void
+newSpaces(uint_least64_t mask, const char *cmd)
+{
+ pid_t pid;
+
+ /* optimize default case */
+ if (mask == 0)
+ return;
+
+ signal(SIGCHLD, SIG_DFL);
+
+#ifdef NDEBUG
+ pid = sys_clone((int) mask | CLONE_VFORK|SIGCHLD, 0);
+#else
+ pid = sys_clone((int) mask | SIGCHLD, 0);
+#endif
+
+ switch (pid) {
+ case -1 :
+ perror(ENSC_WRAPPERS_PREFIX "clone()");
+ exit(wrapper_exit_code);
+ case 0 :
+ break;
+ default :
+ exitLikeProcess(pid, cmd, wrapper_exit_code);
+ }
+}
+
+static void
+enterSpaces(xid_t xid, uint_least64_t mask)
+{
+ if (vc_enter_namespace(xid, mask)==-1) {
+ perror(ENSC_WRAPPERS_PREFIX "vc_enter_namespace()");
+ exit(wrapper_exit_code);
+ }
+}
+
+static void
+setSpaces(xid_t xid, uint_least64_t mask)
+{
+ if (vc_set_namespace(xid, mask)==-1) {
+ perror(ENSC_WRAPPERS_PREFIX "vc_set_namespace()");
+ exit(wrapper_exit_code);
+ }
+}
+
+int main(int argc, char *argv[])
+{
+ bool do_new = false;
+ bool do_enter = false;
+ bool do_set = false;
+ uint_least64_t mask = 0;
+ xid_t xid = VC_NOCTX;
+ int sum = 0;
+
+ while (1) {
+ int c = getopt_long(argc, argv, "+nsce:m:" "MFIUSPN", CMDLINE_OPTIONS, 0);
+ if (c==-1) break;
+
+ switch (c) {
+ case CMD_HELP : showHelp(1, argv[0], 0);
+ case CMD_VERSION : showVersion();
+ case 'n' : do_new = true; break;
+ case 's' : do_set = true; break;
+ case 'e' :
+ do_enter = true;
+ xid = Evc_xidopt2xid(optarg,true);
+ break;
+ case 'm' : {
+ unsigned long mask_l;
+ if (!isNumberUnsigned(optarg, &mask_l, true)) {
+ WRITE_MSG(2, "Invalid mask '");
+ WRITE_STR(2, optarg);
+ WRITE_MSG(2, "'; try '--help' for more information\n");
+ return wrapper_exit_code;
+ }
+ mask = mask_l;
+ break;
+ }
+ case 'M' : mask |= CLONE_NEWNS; break;
+ case 'F' : mask |= CLONE_FS; break;
+ case 'I' : mask |= CLONE_NEWIPC; break;
+ case 'U' : mask |= CLONE_NEWUTS; break;
+ case 'S' : mask |= CLONE_NEWUSER; break;
+ case 'P' : mask |= CLONE_NEWPID; break;
+ case 'N' : mask |= CLONE_NEWNET; break;
+
+ default :
+ WRITE_MSG(2, "Try '");
+ WRITE_STR(2, argv[0]);
+ WRITE_MSG(2, " --help' for more information.\n");
+ return 255;
+ break;
+ }
+ }
+
+ sum = ((do_new ? 1 : 0) + (do_enter ? 1 : 0) +
+ (do_set ? 1 : 0));
+
+ if (sum==0)
+ WRITE_MSG(2, "No operation was specified; try '--help' for more information\n");
+ else if (sum>1)
+ WRITE_MSG(2, "Can not specify multiple operations; try '--help' for more information\n");
+ else if (optind==argc && (do_new || do_enter))
+ WRITE_MSG(2, "No command specified; try '--help' for more information\n");
+ else {
+ if (do_new) newSpaces(mask, argv[optind]);
+ else if (do_set) setSpaces(VC_SAMECTX, mask);
+ else if (do_enter) enterSpaces(xid, mask);
+
+ if (optind<argc)
+ EexecvpD(argv[optind], argv+optind);
+
+ return EXIT_SUCCESS;
+ }
+
+ return 255;
+}
git://git.onelab.eu / util-vserver.git / blobdiff