X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=src%2Fvattribute.c;fp=src%2Fvattribute.c;h=e06bf6002ba1ab49e4663743fc393b0eff601a20;hb=8cf13bb177d92c93eb73dc8939777150536c2d00;hp=0000000000000000000000000000000000000000;hpb=6bf3f95de36c804c97716b2d0bdf10680c559044;p=util-vserver.git

diff --git a/src/vattribute.c b/src/vattribute.c
new file mode 100644
index 0000000..e06bf60
--- /dev/null
+++ b/src/vattribute.c
@@ -0,0 +1,198 @@
+// $Id: vattribute.c,v 1.14 2005/03/24 12:44:17 ensc Exp $    --*- c -*--
+
+// Copyright (C) 2004 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
+//  
+// 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/vserver.h>
+
+#include <getopt.h>
+#include <stdint.h>
+#include <errno.h>
+
+#define ENSC_WRAPPERS_PREFIX	"vattribute: "
+#define ENSC_WRAPPERS_VSERVER	1
+#define ENSC_WRAPPERS_UNISTD	1
+#include <wrappers.h>
+
+#define CMD_HELP		0x1000
+#define CMD_VERSION		0x1001
+#define CMD_XID			0x2000
+#define CMD_SET			0x2001
+#define CMD_CCAP		0x2002
+#define CMD_FLAG		0x2003
+#define CMD_SECURE		0x2004
+#define CMD_BCAP		0x2005
+
+int			wrapper_exit_code = 1;
+
+struct option const
+CMDLINE_OPTIONS[] = {
+  { "help",       no_argument,       0, CMD_HELP },
+  { "version",    no_argument,       0, CMD_VERSION },
+  { "xid",        required_argument, 0, CMD_XID },
+  { "set",        no_argument,       0, CMD_SET },
+  { "ccap",       required_argument, 0, CMD_CCAP },
+  { "bcap",       required_argument, 0, CMD_BCAP },
+  { "flag",       required_argument, 0, CMD_FLAG },
+  { "secure",     no_argument,       0, CMD_SECURE },
+  {0,0,0,0}
+};
+
+struct Arguments {
+    xid_t		xid;
+    struct vc_ctx_flags flags;
+    struct vc_ctx_caps  caps;
+};
+
+static void
+showHelp(int fd, char const *cmd, int res)
+{
+  WRITE_MSG(fd, "Usage:\n    ");
+  WRITE_STR(fd, cmd);
+  WRITE_MSG(fd,
+	    " --set [--xid <xid>] [--bcap [~!]<cap>] [--ccap [~!]<cap>] [--flag [~!]<flag>] [--secure] -- [<program> <args>*]\n"
+	    "\n"
+	    " --bcap <cap>   ...  system  capability to be added\n"
+	    " --cap  <cap>   ...  context capability to be added\n"
+	    "\n"
+	    "Please report bugs to " PACKAGE_BUGREPORT "\n");
+
+  exit(res);
+}
+
+static void
+showVersion()
+{
+  WRITE_MSG(1,
+	    "vattribute " VERSION " -- sets attributes of vservers\n"
+	    "This program is part of " PACKAGE_STRING "\n\n"
+	    "Copyright (C) 2004 Enrico Scholz\n"
+	    VERSION_COPYRIGHT_DISCLAIMER);
+  exit(0);
+}
+
+static void
+parseFlags(char const *str, struct vc_ctx_flags *flags)
+{
+  struct vc_err_listparser	err;
+  int				rc;
+
+  rc = vc_list2cflag(str,0, &err, flags);
+  
+  if (rc==-1) {
+    WRITE_MSG(2, "Unknown flag '");
+    Vwrite(2, err.ptr, err.len);
+    WRITE_MSG(2, "'\n");
+    exit(wrapper_exit_code);
+  }
+}
+
+static void
+parseBCaps(char const *str, struct vc_ctx_caps *caps)
+{
+  struct vc_err_listparser	err;
+  int				rc;
+
+  rc = vc_list2bcap(str,0, &err, caps);
+  
+  if (rc==-1) {
+    WRITE_MSG(2, "Unknown bcap '");
+    Vwrite(2, err.ptr, err.len);
+    WRITE_MSG(2, "'\n");
+    exit(wrapper_exit_code);
+  }
+}
+
+static void
+parseCCaps(char const *str, struct vc_ctx_caps *caps)
+{
+  struct vc_err_listparser	err;
+  int				rc;
+
+  rc = vc_list2ccap(str,0, &err, caps);
+  
+  if (rc==-1) {
+    WRITE_MSG(2, "Unknown ccap '");
+    Vwrite(2, err.ptr, err.len);
+    WRITE_MSG(2, "'\n");
+    exit(wrapper_exit_code);
+  }
+}
+
+static void
+parseSecure(struct vc_ctx_flags UNUSED * flags,
+	    struct vc_ctx_caps  UNUSED * caps)
+{
+  caps->ccaps = ~vc_get_insecureccaps();
+  caps->cmask = ~0ull;
+  caps->bcaps = ~vc_get_insecurebcaps();
+  caps->bmask = ~0ull;
+
+    // TODO: generalize this
+  flags->flagword = VC_VXF_HIDE_NETIF;
+  flags->mask     = VC_VXF_HIDE_NETIF;
+}
+
+int main(int argc, char *argv[])
+{
+  struct Arguments		args = {
+    .xid   = VC_NOCTX,
+    .flags = { .flagword = 0, .mask = 0 },
+    .caps  = { .bcaps = 0, .bmask = 0,.ccaps = 0, .cmask = 0 },
+  };
+  
+  while (1) {
+    int		c = getopt_long(argc, argv, "+", CMDLINE_OPTIONS, 0);
+    if (c==-1) break;
+
+    switch (c) {
+      case CMD_HELP	:  showHelp(1, argv[0], 0);
+      case CMD_VERSION	:  showVersion();
+      case CMD_SET	:  break; // default op currently
+      case CMD_XID	:  args.xid = Evc_xidopt2xid(optarg,true); break;
+      case CMD_FLAG	:  parseFlags(optarg, &args.flags);        break;
+      case CMD_CCAP	:  parseCCaps(optarg, &args.caps);         break;
+      case CMD_BCAP	:  parseBCaps(optarg, &args.caps);         break;
+      case CMD_SECURE	:  parseSecure(&args.flags, &args.caps);   break;
+      default		:
+	WRITE_MSG(2, "Try '");
+	WRITE_STR(2, argv[0]);
+	WRITE_MSG(2, " --help\" for more information.\n");
+	return 255;
+	break;
+    }
+  }
+
+  if (args.xid==VC_NOCTX) args.xid = Evc_get_task_xid(0);
+
+  if ((args.caps.cmask || args.caps.bmask) &&
+      vc_set_ccaps(args.xid, &args.caps)==-1)
+    perror(ENSC_WRAPPERS_PREFIX "vc_set_ccaps()");
+  else if (args.flags.mask &&
+	   vc_set_cflags(args.xid, &args.flags)==-1)
+    perror(ENSC_WRAPPERS_PREFIX "vc_set_flags()");
+  else if (optind<argc)
+    EexecvpD(argv[optind], argv+optind);
+  else
+    return EXIT_SUCCESS;
+
+  return EXIT_FAILURE;
+}