+++ /dev/null
-// $Id: nattribute.c 2403 2006-11-24 23:06:08Z dhozac $ --*- c -*--
-
-// Copyright (C) 2004 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
-// Copyright (C) 2006 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/vserver.h>
-
-#include <getopt.h>
-#include <stdint.h>
-#include <errno.h>
-
-#define ENSC_WRAPPERS_PREFIX "nattribute: "
-#define ENSC_WRAPPERS_VSERVER 1
-#define ENSC_WRAPPERS_UNISTD 1
-#include <wrappers.h>
-
-#define CMD_HELP 0x1000
-#define CMD_VERSION 0x1001
-#define CMD_NID 0x2000
-#define CMD_SET 0x2001
-#define CMD_CAP 0x2002
-#define CMD_FLAG 0x2003
-#define CMD_SECURE 0x2004
-
-int wrapper_exit_code = 1;
-
-struct option const
-CMDLINE_OPTIONS[] = {
- { "help", no_argument, 0, CMD_HELP },
- { "version", no_argument, 0, CMD_VERSION },
- { "nid", required_argument, 0, CMD_NID },
- { "set", no_argument, 0, CMD_SET },
- { "ncap", required_argument, 0, CMD_CAP },
- { "flag", required_argument, 0, CMD_FLAG },
- { "secure", no_argument, 0, CMD_SECURE },
- {0,0,0,0}
-};
-
-struct Arguments {
- nid_t nid;
- struct vc_net_flags flags;
- struct vc_net_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 [--nid <nid>] [--ncap [~!]<ncap>] [--flag [~!]<flag>] [--secure] -- [<program> <args>*]\n"
- "\n"
- " --ncap <cap> ... network capability to be added\n"
- " --flag <flag> ... network flag to be added\n"
- "\n"
- "Please report bugs to " PACKAGE_BUGREPORT "\n");
-
- exit(res);
-}
-
-static void
-showVersion()
-{
- WRITE_MSG(1,
- "nattribute " VERSION " -- sets attributes of network contexts\n"
- "This program is part of " PACKAGE_STRING "\n\n"
- "Copyright (C) 2004 Enrico Scholz\n"
- "Copyright (C) 2006 Daniel Hokka Zakrisson\n"
- VERSION_COPYRIGHT_DISCLAIMER);
- exit(0);
-}
-
-static void
-parseFlags(char const *str, struct vc_net_flags *flags)
-{
- struct vc_err_listparser err;
- int rc;
-
- rc = vc_list2nflag(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
-parseNCaps(char const *str, struct vc_net_caps *caps)
-{
- struct vc_err_listparser err;
- int rc;
-
- rc = vc_list2ncap(str,0, &err, caps);
-
- if (rc==-1) {
- WRITE_MSG(2, "Unknown ncap '");
- Vwrite(2, err.ptr, err.len);
- WRITE_MSG(2, "'\n");
- exit(wrapper_exit_code);
- }
-}
-
-static void
-parseSecure(struct vc_net_flags * flags,
- struct vc_net_caps * caps)
-{
- // TODO: generalize this
- caps->ncaps = 0ull;
- caps->cmask = 0ull;
-
- flags->flagword = VC_NXF_HIDE_NETIF;
- flags->mask = VC_NXF_HIDE_NETIF;
-}
-
-int main(int argc, char *argv[])
-{
- struct Arguments args = {
- .nid = VC_NOCTX,
- .flags = { .flagword = 0, .mask = 0 },
- .caps = { .ncaps = 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_NID : args.nid = Evc_nidopt2nid(optarg,true); break;
- case CMD_FLAG : parseFlags(optarg, &args.flags); break;
- case CMD_CAP : parseNCaps(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.nid==VC_NOCTX) args.nid = Evc_get_task_nid(0);
-
- if (args.caps.cmask &&
- vc_set_ncaps(args.nid, &args.caps)==-1)
- perror(ENSC_WRAPPERS_PREFIX "vc_set_ncaps()");
- else if (args.flags.mask &&
- vc_set_nflags(args.nid, &args.flags)==-1)
- perror(ENSC_WRAPPERS_PREFIX "vc_set_nflags()");
- else if (optind<argc)
- EexecvpD(argv[optind], argv+optind);
- else
- return EXIT_SUCCESS;
-
- return EXIT_FAILURE;
-}
git://git.onelab.eu / util-vserver.git / blobdiff